diff --git a/.ci/slurm/config.ini b/.ci/slurm/config.ini index 3aeb77cba5f..89432701553 100644 --- a/.ci/slurm/config.ini +++ b/.ci/slurm/config.ini @@ -491,7 +491,7 @@ runner = autotest_gpu [case.080] suite = 01_PW -name = 035_PW_15_SO +name = scf_spin4 resource = pw_gpu1 runner = autotest_gpu diff --git a/.github/instructions/abacus-governance.instructions.md b/.github/instructions/abacus-governance.instructions.md index 6e0e09df09e..d79be8d7c24 100644 --- a/.github/instructions/abacus-governance.instructions.md +++ b/.github/instructions/abacus-governance.instructions.md @@ -22,6 +22,8 @@ Apply these instructions when reviewing or changing ABACUS code: usually warnings unless the PR records a narrow reason. - Require LF line endings for text files. `.bat` and `.cmd` files are the CRLF exceptions. +- Flag new `#pragma once` in header files as a blocker-level governance issue. + Require traditional `#ifndef`/`#define`/`#endif` include guards instead. - For INPUT parameter behavior changes, require synchronized updates to `docs/parameters.yaml` and `docs/advanced/input_files/input-main.md`, or a clear no-update explanation in the PR. diff --git a/.github/workflows/ase_plugin_test.yml b/.github/workflows/ase_plugin_test.yml index e9a586e3b08..822afcd957c 100644 --- a/.github/workflows/ase_plugin_test.yml +++ b/.github/workflows/ase_plugin_test.yml @@ -74,3 +74,17 @@ jobs: cd interfaces/ASE_interface/tests chmod +x xtest.sh ./xtest.sh + + # When a test fails, every integration test script (scf/relax/md/band/ + # magnetic) preserves its whole ABACUS run directory under + # /tmp/abacus_ase_failure (see preserved_tmpdir in tests/common.py, + # overridable via ASE_ABACUS_KEEP_DIR). Upload it so remote-CI-only + # failures (e.g. issue #7794) can be root-caused without runner access. + - name: Upload ASE failure artifacts + if: failure() + uses: actions/upload-artifact@v7 + with: + name: ase-failure-${{ github.run_id }}-${{ github.run_attempt }} + path: /tmp/abacus_ase_failure/** + if-no-files-found: ignore + retention-days: 14 diff --git a/.github/workflows/build_test_cmake.yml b/.github/workflows/build_test_cmake.yml index c14ad0ccdf9..ed760753f58 100644 --- a/.github/workflows/build_test_cmake.yml +++ b/.github/workflows/build_test_cmake.yml @@ -24,11 +24,11 @@ jobs: - tag: gnu external_toolchain_args: "" - build_args: "-DENABLE_LIBXC=ON -DENABLE_MLALGO=ON -DENABLE_LIBRI=ON -DENABLE_DFTD4=ON -DENABLE_PEXSI=ON" + build_args: "-DENABLE_LIBXC=ON -DENABLE_MLALGO=ON -DENABLE_LIBRI=ON -DENABLE_DFTD4=ON -DENABLE_PEXSI=ON -DENABLE_JSON=ON" name: "Build extra components with GNU toolchain" - tag: intel external_toolchain_args: "--with-intel" - build_args: "-DENABLE_LIBXC=ON -DENABLE_MLALGO=ON -DENABLE_LIBRI=ON -DENABLE_DFTD4=ON" + build_args: "-DENABLE_LIBXC=ON -DENABLE_MLALGO=ON -DENABLE_LIBRI=ON -DENABLE_DFTD4=ON -DENABLE_JSON=ON" name: "Build extra components with Intel toolchain" - tag: cuda diff --git a/.github/workflows/build_test_makefile.yml b/.github/workflows/build_test_makefile.yml index 70f5dd5f6cc..bb6f25c206d 100644 --- a/.github/workflows/build_test_makefile.yml +++ b/.github/workflows/build_test_makefile.yml @@ -22,7 +22,9 @@ jobs: - name: Checkout uses: actions/checkout@v7.0.1 - name: Build + # TODO: Remove nlohmann-json installation here since Dockerfile will have installed it. run: | + apt update && apt install -y nlohmann-json3-dev export I_MPI_CXX=icpx chmod a+x generate_build_info.sh cd source diff --git a/.github/workflows/coverage.yml b/.github/workflows/coverage.yml index f6dcf32a9eb..01c89ab05b2 100644 --- a/.github/workflows/coverage.yml +++ b/.github/workflows/coverage.yml @@ -62,8 +62,8 @@ jobs: -DENABLE_MLALGO=ON \ -DENABLE_LIBXC=ON \ -DENABLE_LIBRI=ON \ - -DENABLE_GOOGLEBENCH=ON \ - -DENABLE_RAPIDJSON=ON \ + -DENABLE_GOOGLEBENCH=OFF \ + -DENABLE_JSON=ON \ -DCMAKE_BUILD_TYPE=Debug \ -DCMAKE_CXX_FLAGS="--coverage -fprofile-arcs -ftest-coverage" \ -DCMAKE_EXE_LINKER_FLAGS="--coverage" @@ -83,7 +83,7 @@ jobs: lcov --directory . --capture --output-file coverage.info - lcov --remove coverage.info '/usr/*' '*/test/*' '*/external/*' '*/build/*' --output-file coverage.filtered.info + lcov --remove coverage.info '/usr/*' '*/test/*' '*/unittests/*' '*/external/*' '*/build/*' --output-file coverage.filtered.info genhtml coverage.filtered.info --output-directory coverage-report diff --git a/.github/workflows/cuda.yml b/.github/workflows/cuda.yml index 98ad724a950..9a0f8fe8ce3 100644 --- a/.github/workflows/cuda.yml +++ b/.github/workflows/cuda.yml @@ -28,6 +28,12 @@ jobs: CCACHE_BASEDIR: ${{ github.workspace }} CCACHE_COMPRESS: "true" CCACHE_MAXSIZE: 8G + # Test cases to run side by side. The runner has 12 cores and one + # V100-32GB; a GPU case is a few atoms and takes about 2 s, so these + # suites are bound by process startup, not by the device. Four leaves + # every rank a core of its own and keeps the concurrent CUDA contexts + # far inside the card's memory. + GPU_TEST_JOBS: "4" steps: - name: Checkout uses: actions/checkout@v7.0.1 @@ -46,7 +52,7 @@ jobs: echo "fingerprint=${fingerprint}" >> "${GITHUB_OUTPUT}" - name: Restore compiler cache - uses: actions/cache@v6 + uses: actions/cache/restore@v6 with: path: ${{ env.CCACHE_DIR }} key: abacus-cuda-ccache-v1-${{ runner.os }}-${{ runner.arch }}-${{ steps.compiler.outputs.fingerprint }}-${{ github.sha }} @@ -72,6 +78,7 @@ jobs: source toolchain/install/setup rm -rf build cmake -B build -G Ninja -DUSE_CUDA=ON -DBUILD_TESTING=ON -DENABLE_FLOAT_FFTW=ON \ + -DENABLE_GOOGLEBENCH=OFF \ -DCMAKE_CUDA_ARCHITECTURES=70 cmake --build build -j "$(nproc)" cmake --install build @@ -82,6 +89,17 @@ jobs: ccache --show-stats ccache --show-stats >> "${GITHUB_STEP_SUMMARY}" + # Only develop writes the cache. A cache saved from a pull request is + # scoped to that pull request's ref, so no other branch can ever read + # it, while it still counts against the repository's 10 GB budget and + # evicts the develop entry that every pull request does restore from. + - name: Save compiler cache + if: github.ref == 'refs/heads/develop' + uses: actions/cache/save@v6 + with: + path: ${{ env.CCACHE_DIR }} + key: abacus-cuda-ccache-v1-${{ runner.os }}-${{ runner.arch }}-${{ steps.compiler.outputs.fingerprint }}-${{ github.sha }} + - name: Module_LCAO CUDA Unittests env: GTEST_COLOR: 'yes' @@ -92,17 +110,17 @@ jobs: - name: Test 11_PW_GPU run: | cd tests/11_PW_GPU - bash ../integrate/Autotest.sh -n 2 -f CASES_GPU.txt + bash ../integrate/Autotest.sh -n 2 -j "$GPU_TEST_JOBS" -f CASES_GPU.txt - name: Test 12_NAO_Gamma_GPU run: | cd tests/12_NAO_Gamma_GPU - bash ../integrate/Autotest.sh -n 2 -f CASES_GPU.txt + bash ../integrate/Autotest.sh -n 2 -j "$GPU_TEST_JOBS" -f CASES_GPU.txt - name: Test 13_NAO_multik_GPU run: | cd tests/13_NAO_multik_GPU - bash ../integrate/Autotest.sh -n 2 -f CASES_GPU.txt + bash ../integrate/Autotest.sh -n 2 -j "$GPU_TEST_JOBS" -f CASES_GPU.txt - name: Test 02_NAO_Gamma on GPU run: | @@ -114,7 +132,7 @@ jobs: echo "device gpu" >> "$f" fi done - bash ../integrate/Autotest.sh -n 2 -f CASES_GPU.txt + bash ../integrate/Autotest.sh -n 2 -j "$GPU_TEST_JOBS" -f CASES_GPU.txt - name: Test 03_NAO_multik on GPU run: | @@ -126,17 +144,17 @@ jobs: echo "device gpu" >> "$f" fi done - bash ../integrate/Autotest.sh -n 2 -f CASES_GPU.txt + bash ../integrate/Autotest.sh -n 2 -j "$GPU_TEST_JOBS" -f CASES_GPU.txt - name: Test 15_rtTDDFT_GPU run: | cd tests/15_rtTDDFT_GPU - bash ../integrate/Autotest.sh -n 2 -f CASES_GPU.txt + bash ../integrate/Autotest.sh -n 2 -j "$GPU_TEST_JOBS" -f CASES_GPU.txt - name: Test 16_SDFT_GPU run: | cd tests/16_SDFT_GPU - bash ../integrate/Autotest.sh -n 2 -f CASES_GPU.txt + bash ../integrate/Autotest.sh -n 2 -j "$GPU_TEST_JOBS" -f CASES_GPU.txt - name: Test 01_PW on GPU run: | @@ -148,4 +166,4 @@ jobs: echo "device gpu" >> "$f" fi done - bash ../integrate/Autotest.sh -n 1 -a abacus -f CASES_GPU.txt + bash ../integrate/Autotest.sh -n 1 -j "$GPU_TEST_JOBS" -a abacus -f CASES_GPU.txt diff --git a/.github/workflows/performance.yml b/.github/workflows/performance.yml deleted file mode 100644 index 80274314baa..00000000000 --- a/.github/workflows/performance.yml +++ /dev/null @@ -1,33 +0,0 @@ -name: Performance test - -on: - workflow_dispatch: - -defaults: - run: - shell: bash - -jobs: - test: - name: Performance test - runs-on: X64 - strategy: - matrix: - tag: ["gnu", "intel"] - container: ghcr.io/deepmodeling/abacus-${{ matrix.tag }} - timeout-minutes: 2880 - steps: - - name: Checkout - uses: actions/checkout@v7.0.1 - - name: Install Requirements - run: | - sudo apt install -y time - - name: Test - run: | - . /opt/intel/oneapi/setvars.sh || : - cd tests/performance/ - bash run.sh - - name: Show Result - if: always() - run: | - cat tests/performance/sumall.dat diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml index 7ed069dc499..b9cdf22f4ad 100644 --- a/.github/workflows/test.yml +++ b/.github/workflows/test.yml @@ -63,7 +63,7 @@ jobs: echo "fingerprint=${fingerprint}" >> "${GITHUB_OUTPUT}" - name: Restore compiler cache - uses: actions/cache@v6 + uses: actions/cache/restore@v6 with: path: ${{ env.CCACHE_DIR }} key: abacus-gnu-ccache-v1-${{ runner.os }}-${{ runner.arch }}-${{ steps.compiler.outputs.fingerprint }}-${{ github.sha }} @@ -92,8 +92,8 @@ jobs: -DENABLE_MLALGO=ON \ -DENABLE_LIBXC=ON \ -DENABLE_LIBRI=ON \ - -DENABLE_GOOGLEBENCH=ON \ - -DENABLE_RAPIDJSON=ON \ + -DENABLE_GOOGLEBENCH=OFF \ + -DENABLE_JSON=ON \ -DENABLE_FLOAT_FFTW=ON \ -DENABLE_DFTD4=ON \ -DENABLE_PEXSI=ON \ @@ -121,6 +121,17 @@ jobs: ccache --show-stats ccache --show-stats >> "${GITHUB_STEP_SUMMARY}" + # Only develop writes the cache. A cache saved from a pull request is + # scoped to that pull request's ref, so no other branch can ever read + # it, while it still counts against the repository's 10 GB budget and + # evicts the develop entry that every pull request does restore from. + - name: Save compiler cache + if: github.ref == 'refs/heads/develop' + uses: actions/cache/save@v6 + with: + path: ${{ env.CCACHE_DIR }} + key: abacus-gnu-ccache-v1-${{ runner.os }}-${{ runner.arch }}-${{ steps.compiler.outputs.fingerprint }}-${{ github.sha }} + - name: Check documentation consistency run: | ABACUS_BIN=$(find build -name "abacus_*" -type f -executable | head -1) @@ -175,7 +186,7 @@ jobs: GTEST_COLOR: 'yes' OMP_NUM_THREADS: '2' run: | - ctest --test-dir build -V --timeout 1700 -R MODULE_HSOLVER -E PERF_MODULE_HSOLVER_KERNELS + ctest --test-dir build -V --timeout 1700 -R MODULE_HSOLVER - name: Module_Cell Unittests env: @@ -212,6 +223,13 @@ jobs: run: | ctest --test-dir build -V --timeout 1700 -R MODULE_ESTATE + - name: Module_Charge Unittests + env: + GTEST_COLOR: 'yes' + OMP_NUM_THREADS: '2' + run: | + ctest --test-dir build -V --timeout 1700 -R MODULE_CHARGE + - name: Module_Hamilt Unittests env: GTEST_COLOR: 'yes' @@ -310,12 +328,12 @@ jobs: run: | ctest --test-dir build -V --timeout 1700 -R 07_OFDFT - - name: 08_EXX Test + - name: 08_RI Test env: GTEST_COLOR: 'yes' OMP_NUM_THREADS: '2' run: | - ctest --test-dir build -V --timeout 1700 -R 08_EXX + ctest --test-dir build -V --timeout 1700 -R 08_RI - name: 09_DeePKS Test env: @@ -343,4 +361,4 @@ jobs: GTEST_COLOR: 'yes' OMP_NUM_THREADS: '2' run: | - ctest --test-dir build -V --timeout 1700 -E 'integrate_test|01_PW|02_NAO_Gamma|03_NAO_multik|04_FF|05_rtTDDFT|06_SDFT|07_OFDFT|08_EXX|09_DeePKS|10_others|11_PW_GPU|12_NAO_Gamma_GPU|13_NAO_multik_GPU|15_rtTDDFT_GPU|16_SDFT_GPU|17_DS_DFTU|MODULE_BASE|MODULE_IO|MODULE_HSOLVER|MODULE_CELL|MODULE_MD|MODULE_PSI|MODULE_ESTATE|MODULE_RI|MODULE_HAMILT|MODULE_PW|MODULE_LCAO|MODULE_AO|MODULE_NAO|MODULE_RELAX|MODULE_LR' + ctest --test-dir build -V --timeout 1700 -E 'integrated_test|01_PW|02_NAO_Gamma|03_NAO_multik|04_FF|05_rtTDDFT|06_SDFT|07_OFDFT|08_RI|09_DeePKS|10_others|11_PW_GPU|12_NAO_Gamma_GPU|13_NAO_multik_GPU|15_rtTDDFT_GPU|16_SDFT_GPU|17_DS_DFTU|MODULE_BASE|MODULE_IO|MODULE_HSOLVER|MODULE_CELL|MODULE_MD|MODULE_PSI|MODULE_ESTATE|MODULE_CHARGE|MODULE_RI|MODULE_HAMILT|MODULE_PW|MODULE_LCAO|MODULE_AO|MODULE_NAO|MODULE_RELAX|MODULE_LR' diff --git a/.readthedocs.yaml b/.readthedocs.yaml index db95d725eba..9086711962c 100644 --- a/.readthedocs.yaml +++ b/.readthedocs.yaml @@ -9,17 +9,32 @@ build: os: ubuntu-22.04 tools: python: "3.11" + # Toolchain for the `latexpdf` output (latex_engine = 'xelatex' in conf.py). + # Note: none of these come with the default image — each package below + # fixed a real build failure encountered during local verification. + jobs: + pre_build: + # RTD builds PR (external) versions as HTML only, so PDF failures would + # not surface until post-merge. Run the same latex->PDF chain here as a + # pre-merge gate. `latexpdf` is a make-mode target, not a -b builder, + # hence the two-step latex build + make all-pdf. + - | + if [ "$READTHEDOCS_VERSION_TYPE" = "external" ]; then + sphinx-build -b latex docs/ _latex_prcheck --fail-on-warning + fi + # Build documentation in the "docs/" directory with Sphinx sphinx: configuration: docs/conf.py # You can configure Sphinx to use a different builder, for instance use the dirhtml builder for simpler URLs # builder: "dirhtml" # Fail on all warnings to avoid broken references - # fail_on_warning: true + fail_on_warning: true # Optionally build your docs in additional formats such as PDF and ePub -formats: all +formats: + - pdf # Optional but recommended, declare the Python requirements required # to build your documentation diff --git a/AGENTS.md b/AGENTS.md index 5d05669e67e..8f157fecfdb 100644 --- a/AGENTS.md +++ b/AGENTS.md @@ -50,6 +50,14 @@ rules. Read the complete governance document before making or reviewing changes: initializer list to direct parameter passthrough. Perform multi-step computations in the constructor body instead, so failures are easy to debug and each intermediate result is inspectable. + 14. Do not write conditional or computed expressions in a function's + argument list; first assign the value to a named local variable, then + pass that variable. This keeps each argument inspectable and makes the + call site self-documenting. For example, prefer + `const int isk = (nspin == 2 && isk_ptr) ? isk_ptr[ik] : 0;` + followed by `f(..., isk, ...)` over inlining the ternary into the call. + 15. Do not use `#pragma once` in header files; use traditional + `#ifndef`/`#define`/`#endif` include guards instead. - Use LF line endings for text files. Only `.bat` and `.cmd` files may use CRLF. - Keep source file additions deterministic: update the relevant `CMakeLists.txt` or explain why the file is generated or included indirectly. @@ -157,6 +165,12 @@ full mixed-line-ending hook only for intentional repository-wide normalization. - Pull requests: https://github.com/deepmodeling/abacus-develop/pulls - Upstream PRs are opened from personal fork branches (`:` into `develop`). +- When synchronizing a local branch with `upstream/develop`, prefer + `git merge upstream/develop` over `git rebase upstream/develop`. Merge + preserves the original commit SHAs of the local work, keeps the branch + pushable without `--force`, and avoids re-applying a long patch series + when conflicts arise. Use rebase only when the user explicitly asks for + a linear history. - `workflow_dispatch`-only workflows (e.g. `.github/workflows/interface.yml`) are not triggered by push/PR events; PR CI cannot verify such fixes, so state "manual dispatch run required" in the PR verification notes. diff --git a/CMakeLists.txt b/CMakeLists.txt index 33b852f0cd1..62843068f6d 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -35,6 +35,9 @@ project( option(ENABLE_MPI "Enable MPI" ON) option(ENABLE_OPENMP "Enable OpenMP" ON) option(USE_CUDA "Enable CUDA" OFF) +cmake_dependent_option(ENABLE_GPU_NEIGHBOR_LIST + "Enable the CUDA GPU neighbor-list implementation" + OFF "USE_CUDA" OFF) option(USE_ROCM "Enable ROCm" OFF) option(USE_DSP "Enable DSP" OFF) option(USE_CUDA_ON_DCU "Enable CUDA on DCU" OFF) @@ -47,6 +50,7 @@ option(ENABLE_FLOAT_FFTW "Enable using single-precision FFTW library." OFF) option(ENABLE_LCAO "Enable LCAO algorithm" ON) option(ENABLE_DFTD4 "Enable DFT-D4 dispersion correction" OFF) +option(ENABLE_PLUMED "Enable using the PLUMED enhanced-sampling library" OFF) option(BUILD_TESTING "Build unittests" OFF) option(DEBUG_INFO "Print message to debug" OFF) @@ -61,7 +65,7 @@ option(ENABLE_NATIVE_OPTIMIZATION option(COMMIT_INFO "Print commit information in log" ON) option(ENABLE_GOOGLEBENCH "Enable GOOGLE-benchmark usage" OFF) -option(ENABLE_RAPIDJSON "Enable rapid-json usage" OFF) +option(ENABLE_JSON "Enable JSON output" OFF) option(ENABLE_CNPY "Enable cnpy usage" OFF) # Options requiring MPI and LCAO @@ -72,9 +76,6 @@ cmake_dependent_option(ENABLE_PEXSI "Enable PEXSI for LCAO" OFF "ENABLE_LCAO;ENA cmake_dependent_option(ENABLE_MLALGO "Enable the machine learning algorithms" OFF "ENABLE_LCAO;ENABLE_MPI" OFF) -# EXX_DEV works with libRI PR#10 which is not merged into main branch; disabling it -# cmake_dependent_option(ENABLE_EXX_DEV "Enable LibRI developing features" OFF "ENABLE_LIBRI" OFF) - # Two-center FFT is only used in LCAO cmake_dependent_option(ENABLE_FFT_TWO_CENTER "Enable FFT-based two-center integral method" ON "ENABLE_LCAO" OFF) @@ -110,6 +111,14 @@ abacus_rename_option(USE_OPENMP ENABLE_OPENMP) abacus_rename_option(USE_ABACUS_LIBM ENABLE_ABACUS_LIBM) abacus_rename_option(INFO MATH_INFO) +if(DEFINED CACHE{ENABLE_RAPIDJSON}) + message( + FATAL_ERROR + "ENABLE_RAPIDJSON is no longer supported. " + "ABACUS now uses nlohmann-json as its JSON backend; " + "please use -DENABLE_JSON=ON instead.") +endif() + if(DEFINED CACHE{ENABLE_LIBCOMM}) message( DEPRECATION @@ -139,15 +148,8 @@ if(NOT DEFINED NVHPC_ROOT_DIR AND DEFINED ENV{NVHPC_ROOT}) endif() # enable json support -if(ENABLE_RAPIDJSON) - find_package(RapidJSON CONFIG REQUIRED) - if(NOT TARGET RapidJSON) - message( - FATAL_ERROR - "RapidJSON was found, but target RapidJSON is missing. " - "Check if your RapidJSON installation provides a complete exported CMake configuration." - ) - endif() +if(ENABLE_JSON) + find_package(nlohmann_json REQUIRED) endif() # get commit info @@ -284,8 +286,8 @@ if(CCACHE) set(CMAKE_CUDA_COMPILER_LAUNCHER ${CCACHE} ${CMAKE_CUDA_COMPILER_LAUNCHER}) endif() -# Choose build type from: Debug Release RelWithDebInfo MinSizeRel Select -# 'Release' configuration for best performance; this will disable all +# Choose build type from: Debug Release RelWithDebInfo MinSizeRel +# Select 'Release' configuration for best performance; this will disable all # assertions. Other default configurations are also available, see: # https://cmake.org/cmake/help/latest/manual/cmake-buildsystem.7.html#default-and-custom-configurations # For default flags, see: @@ -310,7 +312,7 @@ endif() if(ENABLE_PEXSI) find_package(PEXSI REQUIRED CONFIG) if(PEXSI_VERSION VERSION_LESS "2.0.0") - message(FATAL_ERROR "PEXSI >= 2.0.0 is required") + message(FATAL_ERROR "PEXSI >= 2.0.0 is required; found version ${PEXSI_VERSION}") endif() set(CMAKE_CXX_STANDARD 14) endif() @@ -368,7 +370,10 @@ if(NOT USE_KML AND NOT MKL_FOUND AND NOT USE_SW) endif() if(ENABLE_DFTD4) - find_package(dftd4 4.2.0 REQUIRED) + find_package(dftd4 REQUIRED) + if(dftd4_VERSION VERSION_LESS "4.3.0") + message(FATAL_ERROR "dftd4 >= 4.3.0 is required; found version ${dftd4_VERSION}") + endif() endif() include(CheckLanguage) @@ -609,7 +614,7 @@ endif() if(ENABLE_LIBXC) find_package(Libxc CONFIG REQUIRED) if(Libxc_VERSION VERSION_LESS "5.1.7") - message(FATAL_ERROR "Libxc >= 5.1.7 is required") + message(FATAL_ERROR "Libxc >= 5.1.7 is required; found version ${Libxc_VERSION}") endif() endif() @@ -625,6 +630,15 @@ if(DEFINED TensorFlow_DIR) find_package(TensorFlow REQUIRED) endif() +if(ENABLE_PLUMED) + # PLUMED plugin for collective variables and enhanced sampling + # (https://www.plumed.org). The installation is discovered with + # FindPLUMED.cmake, which prefers pkg-config (plumed.pc) and falls back to + # a manual -DPLUMED_ROOT= hint. + find_package(PLUMED REQUIRED) + message(STATUS "PLUMED support is enabled: version ${PLUMED_VERSION}, include ${PLUMED_INCLUDE_DIRS}") +endif() + if(MATH_INFO) message(STATUS "Will gather math lib info.") # modifications on blas_connector and lapack_connector diff --git a/Dockerfile.cuda b/Dockerfile.cuda index e5ecd76bde1..ac8a09596ce 100644 --- a/Dockerfile.cuda +++ b/Dockerfile.cuda @@ -2,7 +2,7 @@ FROM nvidia/cuda:12.2.0-devel-ubuntu22.04 RUN apt update && apt install -y --no-install-recommends \ libopenblas-openmp-dev liblapack-dev libscalapack-mpi-dev libfftw3-dev libcereal-dev \ - libxc-dev libgtest-dev libgmock-dev libbenchmark-dev python3-numpy ca-certificates \ + libxc-dev libgtest-dev libgmock-dev libbenchmark-dev python3-numpy ca-certificates nlohmann-json3-dev \ bc cmake git g++ make time sudo unzip vim wget libopenmpi-dev gfortran libtool-bin && \ rm -rf /var/lib/apt/lists/* @@ -25,19 +25,11 @@ RUN cd /tmp && \ ln -s /usr/local/include/elpa_openmp-$ELPA_VER/elpa /usr/local/include/ && \ cd /tmp && rm -rf elpa-$ELPA_VER -# RapidJSON -RUN cd /tmp && \ - wget --quiet https://codeload.github.com/Tencent/rapidjson/tar.gz/24b5e7a -O rapidjson-24b5e7a.tar.gz && \ - tar -xzf rapidjson-24b5e7a.tar.gz && cd rapidjson-24b5e7a && \ - cmake -B build -DRAPIDJSON_BUILD_DOC=OFF -DRAPIDJSON_BUILD_EXAMPLES=OFF -DRAPIDJSON_BUILD_TESTS=OFF && \ - cmake --build build --target install && \ - cd /tmp && rm -rf rapidjson-24b5e7a rapidjson-24b5e7a.tar.gz - ADD https://api.github.com/repos/deepmodeling/abacus-develop/git/refs/heads/develop /dev/null RUN git clone https://github.com/deepmodeling/abacus-develop.git --depth 1 && \ cd abacus-develop && \ - cmake -B build -DUSE_CUDA=ON -DENABLE_RAPIDJSON=ON && \ + cmake -B build -DUSE_CUDA=ON -DENABLE_JSON=ON && \ cmake --build build -j $(nproc) && \ cmake --install build && \ rm -rf build && \ diff --git a/Dockerfile.gnu b/Dockerfile.gnu index 943ecf5e7ab..2f0cd53e818 100644 --- a/Dockerfile.gnu +++ b/Dockerfile.gnu @@ -10,7 +10,7 @@ FROM ubuntu:22.04 RUN apt update && apt install -y --no-install-recommends \ libopenblas-openmp-dev liblapack-dev libscalapack-mpi-dev libelpa-dev libfftw3-dev libcereal-dev \ - libxc-dev libgtest-dev libgmock-dev libbenchmark-dev python3-numpy \ + libxc-dev libgtest-dev libgmock-dev libbenchmark-dev python3-numpy nlohmann-json3-dev \ libmetis-dev libparmetis-dev libsuperlu-dist-dev \ bc cmake git g++ make time sudo unzip vim wget gfortran ca-certificates && \ rm -rf /var/lib/apt/lists/* @@ -28,14 +28,6 @@ RUN wget https://download.pytorch.org/libtorch/cpu/libtorch-cxx11-abi-shared-wit --quiet -O libtorch.zip && \ unzip -q libtorch.zip -d /opt && rm libtorch.zip -# RapidJSON -RUN cd /tmp && \ - wget --quiet https://codeload.github.com/Tencent/rapidjson/tar.gz/24b5e7a -O rapidjson-24b5e7a.tar.gz && \ - tar -xzf rapidjson-24b5e7a.tar.gz && cd rapidjson-24b5e7a && \ - cmake -B build -DRAPIDJSON_BUILD_DOC=OFF -DRAPIDJSON_BUILD_EXAMPLES=OFF -DRAPIDJSON_BUILD_TESTS=OFF && \ - cmake --build build --target install && \ - cd /tmp && rm -r rapidjson-24b5e7a rapidjson-24b5e7a.tar.gz - ENV CMAKE_PREFIX_PATH=/opt/libtorch/share/cmake # PEXSI @@ -56,7 +48,7 @@ ADD https://api.github.com/repos/deepmodeling/abacus-develop/git/refs/heads/deve RUN git clone https://github.com/deepmodeling/abacus-develop.git --depth 1 && \ cd abacus-develop && \ - cmake -B build -DENABLE_MLALGO=ON -DENABLE_PEXSI=ON -DENABLE_LIBXC=ON -DENABLE_LIBRI=ON -DENABLE_RAPIDJSON=ON && \ + cmake -B build -DENABLE_MLALGO=ON -DENABLE_PEXSI=ON -DENABLE_LIBXC=ON -DENABLE_LIBRI=ON -DENABLE_JSON=ON && \ cmake --build build -j $(nproc) && \ cmake --install build && \ rm -rf build diff --git a/Dockerfile.intel b/Dockerfile.intel index 6cfcfc4daa3..b9ac0f8821c 100644 --- a/Dockerfile.intel +++ b/Dockerfile.intel @@ -2,7 +2,7 @@ FROM intel/oneapi-hpckit:2025.2.2-0-devel-ubuntu22.04 RUN apt-get update && apt-get install -y \ bc cmake git gnupg gcc g++ python3-numpy sudo wget vim unzip \ - libcereal-dev libxc-dev libgtest-dev libgmock-dev libbenchmark-dev \ + libcereal-dev libxc-dev libgtest-dev libgmock-dev libbenchmark-dev nlohmann-json3-dev \ pkg-config build-essential autoconf automake libtool ca-certificates && \ rm -rf /var/lib/apt/lists/* @@ -22,14 +22,6 @@ RUN cd /tmp && \ ln -s /usr/local/include/elpa_openmp-$ELPA_VER/elpa /usr/local/include/ && \ cd /tmp && rm -rf elpa-$ELPA_VER -# RapidJSON -RUN cd /tmp && \ - wget --quiet https://codeload.github.com/Tencent/rapidjson/tar.gz/24b5e7a -O rapidjson-24b5e7a.tar.gz && \ - tar -xzf rapidjson-24b5e7a.tar.gz && cd rapidjson-24b5e7a && \ - cmake -B build -DRAPIDJSON_BUILD_DOC=OFF -DRAPIDJSON_BUILD_EXAMPLES=OFF -DRAPIDJSON_BUILD_TESTS=OFF && \ - cmake --build build --target install && \ - cd /tmp && rm -r rapidjson-24b5e7a rapidjson-24b5e7a.tar.gz - # LibTorch (Note: Using pre-built Torch library with MKL might cause issues) RUN wget -q https://download.pytorch.org/libtorch/cpu/libtorch-cxx11-abi-shared-with-deps-2.0.0%2Bcpu.zip -O /tmp/libtorch.zip && \ unzip -q /tmp/libtorch.zip -d /opt && rm -f /tmp/libtorch.zip @@ -43,7 +35,7 @@ RUN cd /tmp && git clone https://github.com/deepmodeling/abacus-develop.git --de -DENABLE_MLALGO=ON \ -DENABLE_LIBXC=ON \ -DENABLE_LIBRI=ON \ - -DENABLE_RAPIDJSON=ON \ + -DENABLE_JSON=ON \ -DCMAKE_BUILD_TYPE=Release && \ cmake --build build -j $(nproc) && \ cmake --install build && \ diff --git a/cmake/CollectBuildInfoVars.cmake b/cmake/CollectBuildInfoVars.cmake index 8fc2603c32f..740c50ca4b0 100644 --- a/cmake/CollectBuildInfoVars.cmake +++ b/cmake/CollectBuildInfoVars.cmake @@ -155,19 +155,7 @@ else() if(FFTW3_VERSION) set(ABACUS_FFTW_VERSION "yes (v${FFTW3_VERSION})") else() - if(FFTW3_INCLUDE_DIR AND EXISTS "${FFTW3_INCLUDE_DIR}/fftw3.h") - file(STRINGS "${FFTW3_INCLUDE_DIR}/fftw3.h" _fftw_ver_line - REGEX "^#define[\t ]+FFTW_VERSION[\t ]+\"[^\"]+\"") - if(_fftw_ver_line) - string(REGEX REPLACE "^#define[\t ]+FFTW_VERSION[\t ]+\"([^\"]+)\"" "\\1" - FFTW3_VERSION "${_fftw_ver_line}") - set(ABACUS_FFTW_VERSION "yes (v${FFTW3_VERSION})") - else() - set(ABACUS_FFTW_VERSION "yes (version unknown)") - endif() - else() - set(ABACUS_FFTW_VERSION "yes (version unknown)") - endif() + set(ABACUS_FFTW_VERSION "yes (version unknown)") endif() endif() endif() @@ -240,7 +228,7 @@ endif() if(ENABLE_NCCL_PARALLEL_DEVICE) set(ABACUS_NCCL_PARA "Enabled") else() - set(ABACUS_NCCL_PARA "Enabled") + set(ABACUS_NCCL_PARA "Disabled") endif() # EXX Libraries @@ -361,12 +349,10 @@ else() set(ABACUS_GOOGLEBENCH_VERSION "Disabled") endif() -if(DEFINED RapidJSON_DIR AND ENABLE_RAPIDJSON) - set(ABACUS_RAPIDJSON_VERSION "yes (path: ${RapidJSON_DIR})") -elseif(ENABLE_RAPIDJSON) - set(ABACUS_RAPIDJSON_VERSION "yes (from git origin/master)") +if(ENABLE_JSON) + set(ABACUS_JSON_VERSION "yes (nlohmann-json v${nlohmann_json_VERSION})") else() - set(ABACUS_RAPIDJSON_VERSION "no") + set(ABACUS_JSON_VERSION "no") endif() if(ENABLE_MLALGO) @@ -454,11 +440,8 @@ message(" DeePMD = ${ABACUS_DEEPMD_VERSION}") message(" NEP = ${ABACUS_NEP_VERSION}") message(" Features : Libxc = ${ABACUS_LIBXC_VERSION}") message(" LibRI = ${ABACUS_LIBRI_VERSION}") -if(ENABLE_EXX_DEV) - message(" (EXX developing features enabled)") -endif() message(" DFT-D4 = ${ABACUS_DFTD4_VERSION}") -message(" RAPIDJSON = ${ABACUS_RAPIDJSON_VERSION}") +message(" JSON = ${ABACUS_JSON_VERSION}") message(" PEXSI = ${ABACUS_PEXSI_VERSION}") message(" CNPY = ${ABACUS_CNPY_VERSION}") message(" -------------------------------------------------------------------") diff --git a/cmake/modules/FindFFTW3.cmake b/cmake/modules/FindFFTW3.cmake index 2cf8625f72a..350a7fa1250 100644 --- a/cmake/modules/FindFFTW3.cmake +++ b/cmake/modules/FindFFTW3.cmake @@ -4,6 +4,7 @@ # FFTW3_INCLUDE_DIRS - Where to find FFTW3 headers. # FFTW3_LIBRARIES - List of libraries when using FFTW3. # FFTW3_FOUND - True if FFTW3 is found. +# FFTW3_VERSION - Version from the selected library's pkgconfig/fftw3.pc, if available. # find_path(FFTW3_INCLUDE_DIR @@ -47,6 +48,7 @@ endif() find_package_handle_standard_args(FFTW3 DEFAULT_MSG ${_fftw3_required_vars}) # Copy the results to the output variables and target. +set(FFTW3_VERSION "") if(FFTW3_FOUND) set(FFTW3_LIBRARIES ${FFTW3_LIBRARY}) if (ENABLE_OPENMP) @@ -55,13 +57,16 @@ if(FFTW3_FOUND) set(FFTW3_INCLUDE_DIRS ${FFTW3_INCLUDE_DIR}) - # Try to extract FFTW version from header - if(FFTW3_INCLUDE_DIR AND EXISTS "${FFTW3_INCLUDE_DIR}/fftw3.h") - file(STRINGS "${FFTW3_INCLUDE_DIR}/fftw3.h" _fftw_ver_line REGEX "^#define[\t ]+FFTW_VERSION[\t ]+\"[^\"]+\"") - if(_fftw_ver_line) - string(REGEX REPLACE "^#define[\t ]+FFTW_VERSION[\t ]+\"([^\"]+)\"" "\\1" FFTW3_VERSION "${_fftw_ver_line}") - endif() + # Read the literal version from metadata beside the selected library. + get_filename_component(_fftw3_library_dir "${FFTW3_LIBRARY}" DIRECTORY) + set(_fftw3_pc "${_fftw3_library_dir}/pkgconfig/fftw3.pc") + if(EXISTS "${_fftw3_pc}") + file(STRINGS "${_fftw3_pc}" _fftw3_version_line REGEX "^Version:[ \t]*[0-9]") + if(_fftw3_version_line) + string(REGEX REPLACE "^Version:[ \t]*" "" FFTW3_VERSION "${_fftw3_version_line}") + string(STRIP "${FFTW3_VERSION}" FFTW3_VERSION) endif() + endif() if(NOT TARGET FFTW3::FFTW3) add_library(FFTW3::FFTW3 UNKNOWN IMPORTED) diff --git a/cmake/modules/FindPLUMED.cmake b/cmake/modules/FindPLUMED.cmake new file mode 100644 index 00000000000..d32760e7a3f --- /dev/null +++ b/cmake/modules/FindPLUMED.cmake @@ -0,0 +1,51 @@ +############################################################################### +# - Find PLUMED +# Find the native PLUMED headers and libraries. +# +# PLUMED is built with Autotools and installs a pkg-config file (plumed.pc), so +# pkg-config is the preferred discovery mechanism; a manual PLUMED_ROOT hint is +# kept as a fallback for non-standard installations. +# +# This module provides the imported target +# PLUMED::plumed +# which carries the include directories and the wrapper library (libplumed) of +# a PLUMED installation. +# + +find_package(PkgConfig QUIET) +if(PkgConfig_FOUND) + pkg_check_modules(PLUMED_PKG QUIET IMPORTED_TARGET GLOBAL plumed) +endif() + +if(PLUMED_PKG_FOUND) + set(PLUMED_INCLUDE_DIRS ${PLUMED_PKG_INCLUDE_DIRS}) + set(PLUMED_LIBRARIES ${PLUMED_PKG_LINK_LIBRARIES}) + set(PLUMED_VERSION ${PLUMED_PKG_VERSION}) +else() + # Fallback: manual discovery, e.g. -DPLUMED_ROOT=/path/to/prefix + find_path(PLUMED_INCLUDE_DIRS + plumed/wrapper/Plumed.h + HINTS ${PLUMED_ROOT} + PATH_SUFFIXES "include" + ) + find_library(PLUMED_LIBRARIES + NAMES plumed + HINTS ${PLUMED_ROOT} + PATH_SUFFIXES "lib" + ) +endif() + +include(FindPackageHandleStandardArgs) +find_package_handle_standard_args(PLUMED + REQUIRED_VARS PLUMED_LIBRARIES PLUMED_INCLUDE_DIRS + VERSION_VAR PLUMED_VERSION) + +if(PLUMED_FOUND AND NOT TARGET PLUMED::plumed) + add_library(PLUMED::plumed UNKNOWN IMPORTED) + set_target_properties(PLUMED::plumed PROPERTIES + IMPORTED_LOCATION "${PLUMED_LIBRARIES}" + INTERFACE_INCLUDE_DIRECTORIES "${PLUMED_INCLUDE_DIRS}" + ) +endif() + +mark_as_advanced(PLUMED_INCLUDE_DIRS PLUMED_LIBRARIES) diff --git a/docs/CONTRIBUTING.md b/docs/CONTRIBUTING.md index 428190292f5..c53846664c5 100644 --- a/docs/CONTRIBUTING.md +++ b/docs/CONTRIBUTING.md @@ -95,7 +95,7 @@ To help us reproduce and confirm a bug, please provide a test case and building ## Agent governance and automated review Before coding or requesting review, read the repository governance entry point -[`AGENTS.md`](../AGENTS.md) and the full +[`AGENTS.md`](https://github.com/deepmodeling/abacus-develop/blob/develop/AGENTS.md) and the full [ABACUS Agent Governance](./developers_guide/agent_governance.md) guide. These rules apply to human contributors, AI agents, GitHub CI, and CodeRabbit. diff --git a/docs/Makefile b/docs/Makefile index d7b9e870f7a..c1d81852ef6 100644 --- a/docs/Makefile +++ b/docs/Makefile @@ -3,7 +3,7 @@ # You can set these variables from the command line, and also # from the environment for the first two. -SPHINXOPTS ?= +SPHINXOPTS ?= -W -n --keep-going SPHINXBUILD ?= sphinx-build SOURCEDIR = $(PWD) BUILDDIR = build diff --git a/docs/advanced/acceleration/cuda.md b/docs/advanced/acceleration/cuda.md index ef92a72f02e..782e0990ee8 100644 --- a/docs/advanced/acceleration/cuda.md +++ b/docs/advanced/acceleration/cuda.md @@ -49,7 +49,7 @@ In `INPUT` file we need to set the input parameter [device](../input_files/input - **multi-cards**: ABACUS allows for multi-GPU acceleration. If you have multiple GPU cards, you can run ABACUS with several MPI processes, and each process will utilize one GPU card. For example, the command `mpirun -n 2 abacus` will by default launch two GPUs for computation. If you only have one card, this command will only start one GPU. *note: the number of MPI processes SHOULD be equal to the number of GPU cards, unless you are using MPS in your computer.* ## Examples -We provides [examples](https://github.com/deepmodeling/abacus-develop/tree/develop/examples/gpu) of gpu calculations. +We provides [examples](https://github.com/deepmodeling/abacus-develop/tree/develop/examples/34_gpu) of gpu calculations. ## Known limitations PW basis: diff --git a/docs/advanced/elec_properties/Berry_phase.md b/docs/advanced/elec_properties/Berry_phase.md index 7f85689a5ff..37d81518dab 100644 --- a/docs/advanced/elec_properties/Berry_phase.md +++ b/docs/advanced/elec_properties/Berry_phase.md @@ -1,6 +1,6 @@ # Berry Phase Calculation -From version 2.0.0, ABACUS is capable of calculating macroscopic polarization of insulators by using the Berry phase method, known as the ["modern theory of polarization"](https://www.sciencedirect.com/science/article/abs/pii/S0022459612003234). To calculate the polarization, you need first to do a self-consistent calculation to get the converged charge density. Then, do a non-self-consistent calculation with berry_phase setting to 1. You need also to specify the direction of the polarization you want to calculate. An example is given in the directory [examples/berryphase/lcao_PbTiO3](https://github.com/deepmodeling/abacus-develop/tree/develop/examples/berryphase/lcao_PbTiO3). +From version 2.0.0, ABACUS is capable of calculating macroscopic polarization of insulators by using the Berry phase method, known as the ["modern theory of polarization"](https://www.sciencedirect.com/science/article/abs/pii/S0022459612003234). To calculate the polarization, you need first to do a self-consistent calculation to get the converged charge density. Then, do a non-self-consistent calculation with berry_phase setting to 1. You need also to specify the direction of the polarization you want to calculate. An example is given in the directory [examples/berryphase/lcao_PbTiO3](https://github.com/deepmodeling/abacus-develop/tree/develop/examples/26_berryphase/02_lcao_PbTiO3). To run this example, first do a self-consistent calculation: ``` diff --git a/docs/advanced/elec_properties/Mulliken.md b/docs/advanced/elec_properties/Mulliken.md index a34bca5d07a..c768e83c7ff 100644 --- a/docs/advanced/elec_properties/Mulliken.md +++ b/docs/advanced/elec_properties/Mulliken.md @@ -1,7 +1,7 @@ # Mulliken Charge Analysis -From version 2.1.0, ABACUS has the function of Mulliken population analysis. The example can be found in [examples/mulliken](https://github.com/deepmodeling/abacus-develop/tree/develop/examples/mulliken). \ -To use this function, set [out_mul](./input-main.md#out_mul) to `1` in the INPUT file. After calculation, there will be an output file named `mulliken.txt` in the output directory. In MD calculations, the output interval is controlled by the keyword [out_freq_ion](./input-main.md#out_freq_ion). In the file, there are contents like (`nspin 1`): +From version 2.1.0, ABACUS has the function of Mulliken population analysis. The example can be found in [examples/mulliken](https://github.com/deepmodeling/abacus-develop/tree/develop/examples/14_mulliken). \ +To use this function, set [out_mul](../input_files/input-main.md#out_mul) to `1` in the INPUT file. After calculation, there will be an output file named `mulliken.txt` in the output directory. In MD calculations, the output interval is controlled by the keyword [out_freq_ion](../input_files/input-main.md#out_freq_ion). In the file, there are contents like (`nspin 1`): ``` STEP: 0 diff --git a/docs/advanced/elec_properties/band.md b/docs/advanced/elec_properties/band.md index d6b66fa2c2a..19bcca6ff11 100644 --- a/docs/advanced/elec_properties/band.md +++ b/docs/advanced/elec_properties/band.md @@ -1,6 +1,6 @@ # Extracting Band Structure -In ABACUS, in order to obtain the eigenvalues of Hamiltonian, or generally called band structure, examples can be found in [examples/band](https://github.com/deepmodeling/abacus-develop/tree/develop/examples/band). +In ABACUS, in order to obtain the eigenvalues of Hamiltonian, or generally called band structure, examples can be found in [examples/band](https://github.com/deepmodeling/abacus-develop/tree/develop/examples/12_band). Similar to the [DOS case](https://abacus-rtd.readthedocs.io/en/latest/advanced/elec_properties/dos.html), one first needs to perform a ground-state energy calculation ***with one additional keyword "[out_chg](https://abacus-rtd.readthedocs.io/en/latest/advanced/input_files/input-main.html#out-chg)" in the INPUT file***: ``` diff --git a/docs/advanced/elec_properties/charge.md b/docs/advanced/elec_properties/charge.md index f957f7bf3a7..393f5cdaa9c 100644 --- a/docs/advanced/elec_properties/charge.md +++ b/docs/advanced/elec_properties/charge.md @@ -40,4 +40,4 @@ The following 4 lines are the informations of lattice, in order:\  the number of lattice points along lattice vector a3 (nz), a3/nz, in Bohr.\ The following lines are about the elements and coordinates, in order: the atom number of each atoms, the electron number in pseudopotential, the Cartesian coordinates, in Bohr.\ The rest lines are the value of charge density at each grid. Note that the inner loop is z index, followed by y index, x index in turn.\ -The examples can be found in [examples/charge_density](https://github.com/deepmodeling/abacus-develop/tree/develop/examples/charge_density) +The examples can be found in [examples/charge_density](https://github.com/deepmodeling/abacus-develop/tree/develop/examples/08_charge_density) diff --git a/docs/advanced/elec_properties/density_matrix.md b/docs/advanced/elec_properties/density_matrix.md index 5d6b8ab2e12..2155af8a239 100644 --- a/docs/advanced/elec_properties/density_matrix.md +++ b/docs/advanced/elec_properties/density_matrix.md @@ -45,7 +45,7 @@ The following lines are about the elements and coordinates, in order: all elemen After a blank line, the output is the values of NSPIN and fermi energy.\ The following line is dimension of the density matrix, and the rest lines are the value of each matrix element. -The examples can be found in [examples/density_matrix](https://github.com/deepmodeling/abacus-develop/tree/develop/examples/density_matrix) +The examples can be found in [examples/density_matrix](https://github.com/deepmodeling/abacus-develop/tree/develop/examples/09_density_matrix) - Note: Version difference summary: - develop: `out_dmk` supports both gamma-only and multi-k-point output. diff --git a/docs/advanced/elec_properties/dos.md b/docs/advanced/elec_properties/dos.md index 82777abf0f9..d9f78bc4cc1 100644 --- a/docs/advanced/elec_properties/dos.md +++ b/docs/advanced/elec_properties/dos.md @@ -2,14 +2,14 @@ ## DOS -ABACUS can calculate the density of states (DOS) of the system, and the examples can be found in [examples/dos](https://github.com/deepmodeling/abacus-develop/tree/develop/examples/dos). +ABACUS can calculate the density of states (DOS) of the system, and the examples can be found in [examples/dos](https://github.com/deepmodeling/abacus-develop/tree/develop/examples/13_dos). We first, do a ground-state energy calculation ***with one additional keyword "[out_chg](https://abacus-rtd.readthedocs.io/en/latest/advanced/input_files/input-main.html#out-chg)" in the INPUT file***: ``` out_chg 1 ``` -this will produce the converged charge density, which is contained in the file SPIN1_CHG.cube. +this will produce the converged charge density, which is contained in the file chg.cube. Then, use the same `STRU` file, pseudopotential file and atomic orbital file (and the local density matrix file dm_onsite.txt if DFT+U is used) to do a non-self-consistent calculation. In this example, the potential is constructed from the ground-state charge density from the proceeding calculation. Now the INPUT file is like: ``` @@ -48,7 +48,7 @@ Some parameters in the INPUT file are explained: For LCAO calculations, this parameter will be neglected ! - init_chg - the type of starting density. When doing scf calculation, this variable can be set ”atomic”. When doing nscf calculation, the charge density already exists(eg. in SPIN1_CHG.cube), and the variable should be set as ”file”. It means the density will be read from the existing file SPIN1_CHG.cube. For its more information please see the [here](https://abacus-rtd--1282.org.readthedocs.build/en/1282/advanced/input_files/input-main.html#init_chg). + the type of starting density. When doing scf calculation, this variable can be set ”atomic”. When doing nscf calculation, the charge density already exists(eg. in chg.cube), and the variable should be set as ”file”. It means the density will be read from the existing file chg.cube. For its more information please see the [here](https://abacus-rtd--1282.org.readthedocs.build/en/1282/advanced/input_files/input-main.html#init_chg). - out_dos output density of state(DOS). The unit of DOS is `(number of states)/(eV * unitcell)`. For its more information please see the [here](https://abacus-rtd--1282.org.readthedocs.build/en/1282/advanced/input_files/input-main.html#out_dos). @@ -68,52 +68,30 @@ Gamma 8 8 8 0 0 0 ``` -Run the program, and you will see a file named DOS1_smearing.dat in the output directory. The first two columns in the file are the energy and DOS, respectively, and the third column is the sum of DOS. Plot file DOS1_smearing.dat with graphing software, and you’ll get the DOS. +Run the program, and you will see a file named doss1g1_nao.txt in the output directory. The columns are: energy(eV), dos (number of states in each energy bin), dos_int (integrated DOS), dos_smear(1/eV) (Gaussian-smeared DOS), dos_smear_int (integrated smeared DOS). Plot the file with graphing software, and you'll get the DOS. ``` - -5.49311 0.0518133 0.0518133 - -5.48311 0.0641955 0.116009 - -5.47311 0.0779299 0.193939 - -5.46311 0.0926918 0.28663 - -5.45311 0.108023 0.394653 - -5.44311 0.123346 0.517999 - ... +# energy(eV) dos dos_int dos_smear(1/eV) dos_smear_int + -5.49311 0.0518133 0.0518133 0.0518133 0.0518133 + -5.48311 0.0641955 0.116009 0.0641955 0.116009 + -5.47311 0.0779299 0.193939 0.0779299 0.193939 + ... ``` ## PDOS -Along with the DOS1_smearing.dat file, we also produce the projected density of states (PDOS) in a file called PDOS. +Along with the DOS files, we also produce the projected density of states (PDOS) in files named pdoss{spin}g{geom}_{basis}.txt (e.g., pdoss1g1_nao.txt). -The PDOS file starts with number of atomic orbitals in the system, then a list of energy values, such as: +The PDOS file uses a plain-text format. Each row corresponds to one energy point and one atom. Columns: energy(eV), atom (1-based), species, then pdos values ordered as s(1), p(3), d(5), f(7), etc. Zeta components are summed, and values below 1e-6 are zeroed out. ``` - -1 -26 - - -5.50311 - -5.49311 - -5.48311 - -5.47311 -... - +# istep: 1 +# npoints: 2736 +# energy(eV) atom species pdos(1/eV), columns: s(m=0) p(m=0,+1,-1) d(m=0,+1,-1,+2,-2) f(m=0,+1,-1,+2,-2,+3,-3) + -55.607730 1 Fe 0.000000 0.000000 0.000000 0.000000 0.000000 0.000000 0.000000 0.000000 0.000000 + ... ``` -The rest of the fileis arranged in sections, each section with a header such as below: - -``` - - -... - - -``` +The `# istep:` line is only present for geometry-optimization (istep) output. Within each angular momentum `l`, the `2l+1` columns follow the magnetic-quantum-number order `m = 0, +1, -1, +2, -2, ...`. -which tells the atom and symmetry of the current atomic orbital, and followed by the PDOS values. The values can thus be plotted against the energies. The unit of PDOS is also `(number of states)/(eV * unitcell)`. +For nspin=2, two files are written (pdoss1* and pdoss2*), one per spin channel. For nspin=4, the two spinor components are summed into a single file. The unit of PDOS is also `(number of states)/(eV * unitcell)`. diff --git a/docs/advanced/elec_properties/hs_matrix.md b/docs/advanced/elec_properties/hs_matrix.md index 3683b333075..06cdc78786f 100644 --- a/docs/advanced/elec_properties/hs_matrix.md +++ b/docs/advanced/elec_properties/hs_matrix.md @@ -45,6 +45,8 @@ Each output block starts with a comment header containing the one-based ionic-st For multi-k calculations, the matrices are Hermitian and each matrix element is written as `(real,imag)`. For gamma-only calculations, the matrices are symmetric and the matrix elements are written as real numbers. +> **Note (3.10-LTS):** In the 3.10-LTS version, the corresponding keyword is `out_mat_hs`, and the output file names are `data-0-H` and `data-0-S`, etc. + ### Native Binary Format For `out_hsk 2`, the filenames in the table above use `.dat` instead of `.txt`. Each matrix record is written without padding or a self-describing header: @@ -74,6 +76,8 @@ For a multi-k calculation, the files contain the individual real-space blocks st The folded gamma-only output is sufficient to inspect the matrix used by the gamma-only real-space container, but it does not retain the original lattice-vector resolution and cannot be used to interpolate matrices at arbitrary k points. Terms that are added only while constructing $H(k)$, rather than stored in the internal $H(R)$ container, are not guaranteed to be present. Use [out_hsk](../input_files/input-main.md#out_hsk) when the final $H(\Gamma)$ and $S(\Gamma)$ matrices are required. +> **Note (3.10-LTS):** In the 3.10-LTS version, the corresponding keyword is `out_mat_hs2`, and the output file names are `data-HR-sparse_SPIN0.csr` and `data-SR-sparse_SPIN0.csr`, etc. + ### Text CSR Format The H(R) and S(R) matrices are output in standard Compressed Sparse Row (CSR) format, matching the format used by `out_dmr`. diff --git a/docs/advanced/elec_properties/index.rst b/docs/advanced/elec_properties/index.rst index ad032d392a4..d75e28e28c1 100644 --- a/docs/advanced/elec_properties/index.rst +++ b/docs/advanced/elec_properties/index.rst @@ -11,4 +11,5 @@ Electronic Properties and Outputs charge hs_matrix density_matrix + position_matrix Berry_phase diff --git a/docs/advanced/elec_properties/potential.md b/docs/advanced/elec_properties/potential.md index 00732f407b5..960ceda0c70 100644 --- a/docs/advanced/elec_properties/potential.md +++ b/docs/advanced/elec_properties/potential.md @@ -1,6 +1,6 @@ # Extracting Electrostatic Potential -From version 2.1.0, ABACUS has the function of outputing electrostatic potential, which consists of Hartree potential and the local pseudopotential. To use this function, set ‘[out_pot](https://abacus-rtd.readthedocs.io/en/latest/advanced/input_files/input-main.html#out-pot)’ to ‘2’ in the INPUT file. Here is an example for the [Si-111 surface](https://github.com/deepmodeling/abacus-develop/tree/develop/examples/electrostatic_potential/lcao_Si), and the INPUT file is: +From version 2.1.0, ABACUS has the function of outputing electrostatic potential, which consists of Hartree potential and the local pseudopotential. To use this function, set ‘[out_pot](https://abacus-rtd.readthedocs.io/en/latest/advanced/input_files/input-main.html#out-pot)’ to ‘2’ in the INPUT file. Here is an example for the [Si-111 surface](https://github.com/deepmodeling/abacus-develop/tree/develop/examples/30_elec_pot/01_lcao_Si), and the INPUT file is: ``` INPUT_PARAMETERS diff --git a/docs/advanced/elec_properties/wfc.md b/docs/advanced/elec_properties/wfc.md index a3f6d52f6c8..354d8844fc4 100644 --- a/docs/advanced/elec_properties/wfc.md +++ b/docs/advanced/elec_properties/wfc.md @@ -14,7 +14,7 @@ The files are stored in `OUT.${suffix}/`. Their pattern is `wfk{k}[s{spin}][g{ge With `out_freq_ion=0`, files are written only when the electronic calculation converges or reaches `scf_nmax`, and the names contain neither `g*` nor `e*`. During structural relaxation or molecular dynamics, each later ionic step overwrites the same files. With `out_freq_ion>0`, output is restricted to the ionic steps selected by `out_freq_ion` and occurs at multiples of `out_freq_elec`, at convergence, or at `scf_nmax`; both `g*` and `e*` are included in the file names. A static `calculation=scf` or `calculation=nscf` run also receives `g1e*` indices when `out_freq_ion>0`. -The [`init_wfc file binary`](../scf/initialization.md#wave-function) path reads only unindexed binary `wf*_pw.dat` files from `read_file_dir`. Generate directly reusable files with `out_wfc_pw=2` and normally `out_freq_ion=0`. Text `wf*_pw.txt` files and files containing `g*` or `e*` indices are not matched automatically. +The [`init_wfc file binary`](../scf/initialization.md#wavefunction) path reads only unindexed binary `wf*_pw.dat` files from `read_file_dir`. Generate directly reusable files with `out_wfc_pw=2` and normally `out_freq_ion=0`. Text `wf*_pw.txt` files and files containing `g*` or `e*` indices are not matched automatically. For `basis_type=lcao`, [`out_wfc_lcao`](https://abacus-rtd.readthedocs.io/en/latest/advanced/input_files/input-main.html#out-wfc-lcao) controls the output of numerical-atomic-orbital coefficients: `1` writes text files with the `.txt` suffix and `2` writes binary files with the `.dat` suffix. Multi-k calculations generate `wfs{spin}k{k-point}_nao.txt` or `.dat`, while Gamma-only calculations generate `wfs{spin}_nao.txt` or `.dat`. diff --git a/docs/advanced/input_files/input-main.md b/docs/advanced/input_files/input-main.md index 428f38fd01d..c268d1c4e6d 100644 --- a/docs/advanced/input_files/input-main.md +++ b/docs/advanced/input_files/input-main.md @@ -95,6 +95,7 @@ - [pseudo\_rcut](#pseudo_rcut) - [pseudo\_mesh](#pseudo_mesh) - [nspin](#nspin) + - [gga\_grad](#gga_grad) - [smearing\_method](#smearing_method) - [smearing\_sigma](#smearing_sigma) - [smearing\_sigma\_temp](#smearing_sigma_temp) @@ -155,7 +156,6 @@ - [fixed\_atoms](#fixed_atoms) - [Output information](#output-information) - [out\_freq\_ion](#out_freq_ion) - - [out\_freq\_td](#out_freq_td) - [out\_freq\_elec](#out_freq_elec) - [out\_chg](#out_chg) - [out\_pot](#out_pot) @@ -182,7 +182,6 @@ - [out\_mat\_dh\_vnl](#out_mat_dh_vnl) - [out\_mat\_dh\_vh](#out_mat_dh_vh) - [out\_mat\_dh\_vxc](#out_mat_dh_vxc) - - [out\_mat\_dh\_exx](#out_mat_dh_exx) - [out\_mat\_h\_t](#out_mat_h_t) - [out\_mat\_h\_vnl](#out_mat_h_vnl) - [out\_mat\_h\_vl](#out_mat_h_vl) @@ -343,8 +342,9 @@ - [out\_ri\_cv](#out_ri_cv) - [Exact Exchange (PW)](#exact-exchange-pw) - [exxace](#exxace) - - [exx\_gamma\_extrapolation](#exx_gamma_extrapolation) + - [exx\_gamma\_extra](#exx_gamma_extra) - [ecutexx](#ecutexx) + - [exx\_batch\_size](#exx_batch_size) - [exx\_thr\_type](#exx_thr_type) - [exx\_ene\_thr](#exx_ene_thr) - [Molecular dynamics](#molecular-dynamics) @@ -393,6 +393,8 @@ - [md\_nraise](#md_nraise) - [cal\_syns](#cal_syns) - [dmax](#dmax) + - [plumed](#plumed) + - [plumed\_file](#plumed_file) - [DFT+U correction](#dftu-correction) - [dft\_plus\_u](#dft_plus_u) - [dft\_plus\_dmft](#dft_plus_dmft) @@ -452,12 +454,9 @@ - [out\_wannier\_eig](#out_wannier_eig) - [out\_wannier\_unk](#out_wannier_unk) - [out\_wannier\_wvfn\_formatted](#out_wannier_wvfn_formatted) - - [RT-TDDFT: Real-Time Time-Dependent Density Functional Theory](#rt-tddft-real-time-time-dependent-density-functional-theory) + - [Real-Time TDDFT (Common)](#real-time-tddft-common) - [estep\_per\_md](#estep_per_md) - [td\_dt](#td_dt) - - [td\_edm](#td_edm) - - [td\_print\_eij](#td_print_eij) - - [td\_propagator](#td_propagator) - [td\_vext](#td_vext) - [td\_vext\_dire](#td_vext_dire) - [td\_stype](#td_stype) @@ -490,14 +489,29 @@ - [td\_supsine\_sigma](#td_supsine_sigma) - [td\_supsine\_tstart](#td_supsine_tstart) - [td\_supsine\_tend](#td_supsine_tend) - - [init\_vecpot\_file](#init_vecpot_file) - [ocp](#ocp) - [ocp\_set](#ocp_set) + - [out\_freq\_td](#out_freq_td) - [out\_dipole](#out_dipole) - [out\_current](#out_current) - [out\_current\_k](#out_current_k) - [out\_efield](#out_efield) + - [Real-Time TDDFT (LCAO)](#real-time-tddft-lcao) + - [td\_edm](#td_edm) + - [td\_print\_eij](#td_print_eij) + - [td\_propagator](#td_propagator) + - [init\_vecpot\_file](#init_vecpot_file) - [out\_vecpot](#out_vecpot) + - [Real-Time TDDFT (PW)](#real-time-tddft-pw) + - [td\_orthonormal](#td_orthonormal) + - [td\_out\_stat](#td_out_stat) + - [lin\_solver](#lin_solver) + - [lin\_precond](#lin_precond) + - [lin\_thr](#lin_thr) + - [lin\_maxiter](#lin_maxiter) + - [lin\_gmres\_restart](#lin_gmres_restart) + - [td\_cn\_init](#td_cn_init) + - [lin\_reconstruct](#lin_reconstruct) - [Variables useful for debugging](#variables-useful-for-debugging) - [nurse](#nurse) - [t\_in\_h](#t_in_h) @@ -1351,8 +1365,21 @@ - 1: Spin degeneracy - 2: Collinear spin polarized. - 4: Noncollinear or spin-orbit calculations. Set nspin to 4 explicitly when noncolin or lspinorb is enabled. + - Note: With nspin=2 and no initial magnetization in STRU, a moment of 1.0 is autoset for every atom, unless symmetry is 1. + - Note: With nspin=4 no moment is ever autoset. The calculation starts from zero magnetic moment and a warning is printed; set 'mag' explicitly in STRU for the magnetic atoms if a magnetic ground state is expected. - **Default**: 1 +### gga_grad + +- **Type**: Integer +- **Description**: Selects the local spin mapping for LDA/GGA functionals in magnetic nspin=4 calculations. + - 0: preserves the original algorithm (default). + - 1: uses the local magnetization magnitude instead of the global quantization axis in the built-in GGA gradient correction. For LIBXC functionals, 0 and 1 are equivalent. + - 2: uses a C2-regularized magnetization magnitude with eta = 1e-3 in atomic density units. The spin densities are (abs(n + rho_core) +/- min(S_eta(m), abs(n + rho_core)))/2. GGA gradients are the local-map Jacobian applied to the FFT gradients of the four density channels. The potential reverses this same discrete energy graph, including the radial Hessian and density/sigma clipping branches; the GGA stress uses the corresponding metric derivative. + For r = |m| and x = r/eta, S_eta = eta*x^3*(3*x^2 - 8*x + 6) for r < eta, and S_eta = r otherwise. The regularization is part of the functional definition, including its first and second derivatives. + Mode 2 also uses this local map for the LDA contribution. Other spin configurations retain their existing behavior. +- **Default**: 0 + ### smearing_method - **Type**: String @@ -1581,6 +1608,8 @@ - noncolin=0, lspinorb=1: SOC with z-axis magnetism only (for non-magnetic materials with SOC) - noncolin=1, lspinorb=0: Non-collinear magnetism without SOC - noncolin=1, lspinorb=1: Both non-collinear magnetism and SOC + - Note: When nspin=4 and noncolin=0, only the z component of the initial magnetization in STRU is used; x/y components are ignored and a warning is printed. + - Note: When nspin=4 and no initial magnetization is set in STRU, the calculation starts from zero magnetic moment; no automatic magnetization is assigned. - **Default**: False ### soc_lambda @@ -1878,14 +1907,6 @@ > Note: In RT-TDDFT calculations, this parameter is inactive; output frequency is instead controlled by out_freq_td. - **Default**: 0 -### out_freq_td - -- **Type**: Integer -- **Description**: Controls the output interval in completed electronic evolution steps during RT-TDDFT calculations. When set to a positive integer n, detailed information (see out_freq_ion) is printed every n electron time-evolution steps (i.e., every STEP OF ELECTRON EVOLVE). For example, if you wish to output information once per ionic step, you should set out_freq_td equal to estep_per_md, since one ionic step corresponds to estep_per_md electronic evolution steps. - - > Note: This parameter is only active in RT-TDDFT mode (esolver_type = tddft). It has no effect in ground-state calculations. -- **Default**: 0 - ### out_freq_elec - **Type**: Integer @@ -2027,13 +2048,12 @@ ### out_dos - **Type**: Integer -- **Description**: Whether to output the density of states (DOS). For more information, refer to the dos.md. +- **Description**: Whether to output the density of states (DOS) and projected density of states (PDOS). For more information, refer to the dos.md. - 0: no output - - 1: output the density of states (DOS) - - nspin=1 or 4: doss1g{geom}_{basis}.txt, where geom is the geometry index when cell changes or ions move while basis is either pw or nao. - - nspin=2: doss1g{geom}_{basis}.txt and doss2g{geom}_{basis}.txt for two spin channles. - - 2: (LCAO) output the density of states (DOS) and the projected density of states (PDOS) - - 3: output the Fermi surface file (fermi.bxsf) in BXSF format that can be visualized by XCrySDen + - 1: output the density of states (DOS) and projected density of states (PDOS, LCAO only) + - nspin=1 or 4: doss1g{geom}_{basis}.txt and pdoss1g{geom}_{basis}.txt, where geom is the geometry index when cell changes or ions move while basis is either pw or nao. + - nspin=2: doss1/doss2 and pdoss1/pdoss2 files for two spin channels. + Note: values 2 and 3 are no longer supported. Setting out_dos to 2 or 3 will raise an error. - **Default**: 0 ### out_ldos @@ -2064,7 +2084,7 @@ ### out_stru - **Type**: Integer -- **Description**: Controls the output of structure files per ionic step in geometry relaxation calculations. The files are written to the OUT.{suffix}/ directory. Each file corresponds to the structure at RELAX STEP ${istep}, i.e., the structure for which that step's energy was computed (before the relax move), and includes a header comment with the ABACUS version, timestamp, energy, and stress tensor. When out_freq_ion is positive, the numbered files STRU{istep} (or STRU{istep}.cif) are written every out_freq_ion steps; when out_freq_ion is 0, no numbered files are output. +- **Description**: Controls the output of structure files per ionic step. The files are written to the OUT.{suffix}/ directory. Each file corresponds to the structure at RELAX STEP ${istep} (for scf/nscf this is the single step), i.e., the structure for which that step's energy was computed (before the relax move), and includes a header comment with the ABACUS version, timestamp, energy, and stress tensor. When out_freq_ion is positive, the numbered files STRU{istep} (or STRU{istep}.cif) are written every out_freq_ion steps during geometry relaxation; when out_freq_ion is 0, no numbered files are output. This parameter is effective for scf/nscf/relax/cell-relax; for scf/nscf only STRU_FINAL (or STRU_FINAL.cif) is written, and structure output is disabled by default unless out_stru is set explicitly. Molecular dynamics structure output is instead controlled by md_restartfreq (STRU_MD_*). - 0: No structure files are output. - 1: ABACUS STRU format files are output. The latest structure is written to STRU_NOW (overwritten each step), the numbered file STRU{istep} (e.g., STRU1, STRU2) is written every out_freq_ion steps (when out_freq_ion is positive), and the final converged structure is written to STRU_FINAL. No CIF files are output. - 2: CIF format files are output. The latest structure is written to STRU_NOW.cif (overwritten each step), the numbered file STRU{istep}.cif (e.g., STRU1.cif, STRU2.cif) is written every out_freq_ion steps (when out_freq_ion is positive), and the final converged structure is written to STRU_FINAL.cif. No non-CIF files are output. @@ -2164,7 +2184,7 @@ - **Type**: Boolean \[Integer\](optional) - **Availability**: *[`basis_type`](#basis_type)==lcao and [`gamma_only`](#gamma_only)==0* -- **Description**: Generate files containing the kinetic energy matrix. The optional second parameter controls text output precision. The format will be the same as the Hamiltonian matrix and overlap matrix as mentioned in out_hsr. The name of the files will be trs1_nao.csr and so on. Also controled by out_freq_ion and out_app_flag. +- **Description**: Generate files containing the kinetic energy matrix. The optional second parameter controls text output precision. The format will be the same as the Hamiltonian matrix and overlap matrix as mentioned in out_hsr. The name of the files will be tr_nao.csr and so on. Also controled by out_freq_ion and out_app_flag. > Note: In the 3.10-LTS version, the file name is data-TR-sparse_SPIN0.csr. - **Default**: False 8 @@ -2227,15 +2247,6 @@ - **Default**: 0 8 - **Unit**: Ry/Bohr -### out_mat_dh_exx - -- **Type**: Integer -- **Description**: Whether to print files containing the derivatives of the exact-exchange matrix dV^EXX/dR. - - See out_mat_dh for format details. -- **Default**: 0 8 -- **Unit**: Ry/Bohr - ### out_mat_h_t - **Type**: Integer @@ -2324,7 +2335,7 @@ - **Type**: Boolean \[Integer\](optional) - **Availability**: *[`basis_type`](#basis_type)==lcao* -- **Description**: Whether to print the expectation value of the angular momentum operator , , and in the basis of the localized atomic orbitals. The files are named OUT.{suffix}_Lx.dat, OUT.{suffix}_Ly.dat, and OUT.{suffix}_Lz.dat. The second integer controls the precision of the output. +- **Description**: Whether to print the expectation value of the angular momentum operator , , and in the basis of the localized atomic orbitals. The files are named lx_nao.txt, ly_nao.txt, and lz_nao.txt (or lxg{step+1}_nao.txt etc. when out_freq_ion is set). The second integer controls the precision of the output. - **Default**: False 8 ### out_xc_r @@ -2512,8 +2523,8 @@ ### dos_edelta_ev - **Type**: Real -- **Description**: The step size in writing Density of States (DOS) -- **Default**: 0.01 +- **Description**: The step size in writing Density of States (DOS). The default value was changed from 0.01 to 0.03 eV. +- **Default**: 0.03 - **Unit**: eV ### dos_sigma @@ -3441,7 +3452,7 @@ - False: Use the traditional method to calculate the Fock exchange operator. - **Default**: True -### exx_gamma_extrapolation +### exx_gamma_extra - **Type**: Boolean - **Description**: Whether to use the gamma point extrapolation method to calculate the Fock exchange operator. See https://doi.org/10.1103/PhysRevB.79.205114 for details. Should be set to true most of the time. @@ -3450,10 +3461,16 @@ ### ecutexx - **Type**: Real -- **Description**: The energy cutoff for EXX (Fock) exchange operator in plane wave basis calculations. Reducing ecutexx below ecutrho may significantly accelerate EXX computations. This speed improvement comes with a reduced numerical accuracy in the exchange energy calculation. +- **Description**: The energy cutoff for EXX (Fock) exchange operator in plane wave basis calculations. The pair-density G-sphere of the exchange operator, the EXX energy, and the EXX stress are all truncated at this value. If ecutexx yields a smaller FFT box and every |k+G|^2 of the wavefunctions fits inside it (i.e. ecutexx should not be smaller than ecutwfc), all EXX FFTs run on that smaller grid (QE ecutfock-style), which can significantly accelerate EXX computations. If the small grid is not usable (box not smaller, wavefunctions do not fit, or the FFT box is distributed over MPI), a warning is printed and the full grid is used. Reducing ecutexx below ecutrho reduces the numerical accuracy of the exchange contribution. - **Default**: same as ecutrho - **Unit**: Ry +### exx_batch_size + +- **Type**: Integer +- **Description**: Number of bands processed per round of the EXX batched FFT path. 0 (the default) processes all bands in one round, which is fastest but needs nbands * nxyz work buffers; a positive value processes the bands in chunks of that width, trading some performance for a proportionally smaller memory footprint. The result is independent of the chunking. +- **Default**: 0 + ### exx_thr_type - **Type**: String @@ -3800,7 +3817,7 @@ ### cal_syns -- **Type**: Boolean [Integer](optional) +- **Type**: Boolean \[Integer\](optional) - **Description**: Whether to calculate and output asynchronous overlap matrix for Hefei-NAMD interface. When enabled, calculates <phi(t-1)|phi(t)> by computing overlap between basis functions at atomic positions from previous time step and current time step. The overlap is calculated by shifting atom positions backward by velocity x md_dt. Output file: OUT.*/syns_nao.csr in CSR format. - 0 or false: disable @@ -3817,6 +3834,19 @@ - **Default**: 0.01 - **Unit**: bohr +### plumed + +- **Type**: Boolean +- **Description**: Whether to use PLUMED (https://www.plumed.org) to compute collective variables, biasing potentials and free-energy methods during an MD run. ABACUS must be compiled with -DENABLE_PLUMED=ON and linked against a PLUMED installation (discovered through pkg-config or PLUMED_ROOT) to enable this feature. The PLUMED interface currently supports a single MPI rank only. +- **Default**: False + +### plumed_file + +- **Type**: String +- **Availability**: *[`plumed`](#plumed)==true* +- **Description**: The input file of PLUMED, read when `plumed` is enabled. Relative paths are resolved against the working directory of the run. +- **Default**: plumed.dat + [back to top](#full-list-of-input-keywords) ## DFT+U correction @@ -4284,7 +4314,7 @@ [back to top](#full-list-of-input-keywords) -## RT-TDDFT: Real-Time Time-Dependent Density Functional Theory +## Real-Time TDDFT (Common) ### estep_per_md @@ -4299,35 +4329,6 @@ - **Default**: md_dt / estep_per_md - **Unit**: fs -### td_edm - -- **Type**: Integer -- **Description**: Method used to calculate the energy-density matrix for the overlap contribution to forces in LCAO RT-TDDFT. - - 0: Use $\mathrm{EDM}_{\boldsymbol{k}}=\frac{1}{2}\left(S_{\boldsymbol{k}}^{-1}H_{\boldsymbol{k}}\rho_{\boldsymbol{k}}+\rho_{\boldsymbol{k}}H_{\boldsymbol{k}}S_{\boldsymbol{k}}^{-1}\right)$. - - 1: Use the ground-state eigenvalue-weighted expression $\mathrm{EDM}_{\mu\nu,\boldsymbol{k}}=\sum_i w_{i\boldsymbol{k}}\epsilon_{i\boldsymbol{k}}C_{\mu i,\boldsymbol{k}}C_{\nu i,\boldsymbol{k}}^*$. This expression is deprecated for RT-TDDFT and is generally not valid when the propagated wave functions are not Hamiltonian eigenstates. -- **Default**: 0 - -### td_print_eij - -- **Type**: Real -- **Description**: Controls output of the propagated-state Hamiltonian matrix elements $E_{ij}=\Braket{\psi_i | \hat{H} | \psi_j}$ to the running log. The printed band indices $i$ and $j$ are one-based global indices. Both the threshold and the printed matrix elements are in Ry. - - $\lt 0$: Disable the output. - - $\geqslant 0$: Print an element when either $\left|\operatorname{Re}E_{ij}\right|$ or $\left|\operatorname{Im}E_{ij}\right|$ is greater than or equal to td_print_eij. -- **Default**: -1 -- **Unit**: Ry - -### td_propagator - -- **Type**: Integer -- **Description**: Method used to propagate the electronic states in a nonorthogonal LCAO basis. The formulas below use Hartree atomic units, with $S$, $H$, and $\Delta t=\mathtt{td\_dt}$ evaluated as required by each approximation. - - 0: Crank-Nicolson through an explicitly constructed evolution matrix, $U=\left[S+\mathrm{i}H\Delta t/2\right]^{-1}\left[S-\mathrm{i}H\Delta t/2\right]$. - - 1: Fourth-order Taylor approximation to the exponential. With $\mathcal{A}=-\mathrm{i}S^{-1}H\Delta t$, $U=I+\mathcal{A}+\mathcal{A}^2/2+\mathcal{A}^3/6+\mathcal{A}^4/24$. - - 2: Enforced time-reversal symmetry (ETRS), $U(t+\Delta t,t)=\exp\left[-\mathrm{i}S^{-1}H(t+\Delta t)\Delta t/2\right]\exp\left[-\mathrm{i}S^{-1}H(t)\Delta t/2\right]$. In the implementation, each exponential is replaced by the fourth-order Taylor polynomial from method 1 evaluated with a half time step. - - 3: Crank-Nicolson by directly solving $\left[S+\mathrm{i}H\Delta t/2\right]\psi(t+\Delta t)=\left[S-\mathrm{i}H\Delta t/2\right]\psi(t)$. - - > Note: GPU execution currently supports only method 0 in both single-GPU and multi-GPU solver configurations. CPU execution supports methods 0 through 3. -- **Default**: 0 - ### td_vext - **Type**: Boolean @@ -4349,9 +4350,9 @@ - **Type**: Integer - **Description**: Type of electric field in the space domain, i.e. the gauge of the electric field. - - 0: Length gauge. - - 1: Velocity gauge. - - 2: Hybrid gauge. See J. Chem. Theory Comput. 2025, 21, 3335-3341 for more information. + - 0: Length gauge, available for PW and LCAO. + - 1: Velocity gauge, available for PW and LCAO. + - 2: Hybrid gauge, available only for LCAO. See J. Chem. Theory Comput. 2025, 21, 3335-3341 for more information. - **Default**: 0 ### td_ttype @@ -4453,21 +4454,21 @@ - **Type**: Vector of Real - **Availability**: *[`td_ttype`](#td_ttype) contains 1* -- **Description**: Electronic step defining the end of the linear rise, $t_1=\mathtt{td\_trape\_t1}\Delta t$. Each field must satisfy td_trape_t1 <= td_trape_t2 <= td_trape_t3. Supply exactly one value for each td_ttype 1 occurrence, in occurrence order. +- **Description**: Electronic step defining the end of the linear rise, $t_1=\mathtt{td\_trape\_t1}\Delta t$. The value must be a nonnegative integer; integer-valued real inputs such as 2.0 are accepted. Each field must satisfy td_trape_t1 <= td_trape_t2 <= td_trape_t3. Supply exactly one value for each td_ttype 1 occurrence, in occurrence order. - **Default**: 1875 ### td_trape_t2 - **Type**: Vector of Real - **Availability**: *[`td_ttype`](#td_ttype) contains 1* -- **Description**: Electronic step defining the end of the plateau, $t_2=\mathtt{td\_trape\_t2}\Delta t$. Each field must satisfy td_trape_t1 <= td_trape_t2 <= td_trape_t3. Supply exactly one value for each td_ttype 1 occurrence, in occurrence order. +- **Description**: Electronic step defining the end of the plateau, $t_2=\mathtt{td\_trape\_t2}\Delta t$. The value must be a nonnegative integer; integer-valued real inputs such as 2.0 are accepted. Each field must satisfy td_trape_t1 <= td_trape_t2 <= td_trape_t3. Supply exactly one value for each td_ttype 1 occurrence, in occurrence order. - **Default**: 5625 ### td_trape_t3 - **Type**: Vector of Real - **Availability**: *[`td_ttype`](#td_ttype) contains 1* -- **Description**: Electronic step defining the end of the linear fall, $t_3=\mathtt{td\_trape\_t3}\Delta t$. Each field must satisfy td_trape_t1 <= td_trape_t2 <= td_trape_t3. Supply exactly one value for each td_ttype 1 occurrence, in occurrence order. +- **Description**: Electronic step defining the end of the linear fall, $t_3=\mathtt{td\_trape\_t3}\Delta t$. The value must be a nonnegative integer; integer-valued real inputs such as 2.0 are accepted. Each field must satisfy td_trape_t1 <= td_trape_t2 <= td_trape_t3. Supply exactly one value for each td_ttype 1 occurrence, in occurrence order. - **Default**: 7500 ### td_trape_amp @@ -4578,31 +4579,31 @@ - **Description**: Integer electronic step at the right, exactly zero boundary of each supersine pulse, defining $t_{\mathrm{e}}=\mathtt{td\_supsine\_tend}\Delta t$. Supply exactly one integer or default token for each td_ttype 4 occurrence, in occurrence order; each default token inherits td_tend. The complete pulse support must lie inside the inclusive global td_tstart to td_tend interval; hard truncation of a supersine pulse is rejected. - **Default**: default -### init_vecpot_file - -- **Type**: Boolean -- **Description**: Selects the source of the Cartesian vector potential used by LCAO RT-TDDFT. - - True: Read vector_pot.txt from the calculation working directory. Each non-comment line must contain four columns: a conventionally one-based electronic-step label followed by $A_x$, $A_y$, and $A_z$ in atomic units. Rows are consumed sequentially; the first column is read as a label and is not used for lookup. If propagation continues beyond the available rows, the last row is reused. - - False: Obtain the vector potential by integrating the configured electric field. -- **Default**: False - ### ocp - **Type**: Boolean -- **Description**: Controls fixed band occupations. In calculations other than LCAO RT-TDDFT, fixed values are applied during electronic-state setup. In LCAO RT-TDDFT, the initial ground-state SCF determines occupations normally, and fixed values from ocp_set are applied during the subsequent real-time propagation steps. - - True: Use the fixed occupations specified by ocp_set during propagation. - - False: Keep the occupations determined by the initial SCF. +- **Description**: Controls fixed band occupations. In PW RT-TDDFT and calculations other than LCAO RT-TDDFT, fixed values from ocp_set are applied during electronic-state setup. PW real-time propagation preserves these initial occupations. In LCAO RT-TDDFT, the initial ground-state SCF determines occupations normally, and fixed values are applied during the subsequent real-time propagation steps. + - True: Use the fixed occupations specified by ocp_set at the stage described above. + - False: Determine occupations normally; real-time propagation preserves the initial SCF occupations. - **Default**: False ### ocp_set - **Type**: String -- **Description**: Fixed occupation weights used when ocp is true. Values are assigned in band order for each k-point, following k-point order. In LCAO RT-TDDFT, the initial ground-state SCF uses its normally determined occupations, and this array is applied only during subsequent real-time propagation steps. The repetition syntax N*x expands to N copies of x. +- **Description**: Fixed occupation weights used when ocp is true. Values are assigned in band order for each k-point, following k-point order. In PW RT-TDDFT and other calculations outside LCAO RT-TDDFT, this array is applied during electronic-state setup. PW propagation preserves these occupations. In LCAO RT-TDDFT, the initial ground-state SCF uses its normally determined occupations, and this array is applied only during subsequent real-time propagation steps. The repetition syntax N*x expands to N copies of x. - Example: 1 10*1 0 1 expands to 13 values, with the 12th value equal to 0 and all other values equal to 1. - After expansion, provide one block of nbands values for each k-point. If nspin is 2, provide all k-point blocks for spin up followed by all k-point blocks for spin down; otherwise, provide one block per k-point. - The sum of all weights must equal nelec; otherwise the calculation terminates with an error. - **Default**: None +### out_freq_td + +- **Type**: Integer +- **Description**: Controls the output interval in completed electronic evolution steps during RT-TDDFT calculations. When set to a positive integer n, detailed information (see out_freq_ion) is printed every n electron time-evolution steps (i.e., every STEP OF ELECTRON EVOLVE). For example, if you wish to output information once per ionic step, you should set out_freq_td equal to estep_per_md, since one ionic step corresponds to estep_per_md electronic evolution steps. + + > Note: This parameter is only active in RT-TDDFT mode (esolver_type = tddft). It has no effect in ground-state calculations. +- **Default**: 0 + ### out_dipole - **Type**: Boolean @@ -4614,19 +4615,19 @@ ### out_current - **Type**: Integer -- **Availability**: *[`basis_type`](#basis_type)==lcao and [`esolver_type`](#esolver_type)==tddft* -- **Description**: Controls the current-density output method for LCAO RT-TDDFT. Output rows contain the one-based electronic-step index followed by $J_x$, $J_y$, and $J_z$ in atomic units. +- **Availability**: *[`basis_type`](#basis_type) in [pw, lcao] and [`esolver_type`](#esolver_type)==tddft* +- **Description**: Controls the current-density output method for RT-TDDFT. Each row contains the one-based electronic-step index followed by $J_x$, $J_y$, and $J_z$ in atomic units; the initial ground state is step 1 and subsequent steps increase by one. - 0: Do not output current. - - 1: Explicitly construct the velocity operator from the momentum, vector-potential, and KB nonlocal-pseudopotential terms using two-center and spherical-grid integrals: $\hat{v}_{\alpha}=-\mathrm{i}\nabla_{\alpha}+A_{\alpha}(t)+\mathrm{i}\left[\widetilde{V}_{\mathrm{NL}}^{\mathrm{KB}},r_{\alpha}\right]$, where $\widetilde{V}_{\mathrm{NL}}^{\mathrm{KB}}=\mathrm{e}^{-\mathrm{i}\boldsymbol{A}(t)\cdot\boldsymbol{r}}\hat{V}_{\mathrm{NL}}^{\mathrm{KB}}\mathrm{e}^{\mathrm{i}\boldsymbol{A}(t)\cdot\boldsymbol{r}}$. $\boldsymbol{A}(t)$ is nonzero only for the velocity gauge (td_stype=1); otherwise $\boldsymbol{A}(t)=0$. Other nonlocal Hamiltonian terms, such as EXX, are not included explicitly. The total current is written to OUT.{suffix}/current_tot.txt. - - 2: Use the full Hamiltonian to construct the generalized velocity matrix in a nonorthogonal NAO basis, $\widetilde{v}_{\alpha}=\partial_{\alpha}H+\mathrm{i}HS^{-1}\mathcal{R}_{\alpha}-\mathrm{i}\mathcal{R}_{\alpha}S^{-1}H-HS^{-1}\partial_{\alpha}S$. This includes all contributions available in the real-space Hamiltonian matrix when enabled. This method is more general but more expensive. The total current is written to OUT.{suffix}/current_tot_comm.txt. + - 1: Available for PW and LCAO. PW evaluates the occupied-state expectation values of the velocity operator, using plane-wave momentum and nonlocal-projector derivatives; in the velocity gauge these are evaluated at the vector-potential-shifted momentum. In the length gauge, a kinetic-energy-density functional also contributes its velocity term. LCAO explicitly constructs the velocity operator from the momentum, vector-potential, and KB nonlocal-pseudopotential terms using two-center and spherical-grid integrals: $\hat{v}_{\alpha}=-\mathrm{i}\nabla_{\alpha}+A_{\alpha}(t)+\mathrm{i}\left[\widetilde{V}_{\mathrm{NL}}^{\mathrm{KB}},r_{\alpha}\right]$, where $\widetilde{V}_{\mathrm{NL}}^{\mathrm{KB}}=\mathrm{e}^{-\mathrm{i}\boldsymbol{A}(t)\cdot\boldsymbol{r}}\hat{V}_{\mathrm{NL}}^{\mathrm{KB}}\mathrm{e}^{\mathrm{i}\boldsymbol{A}(t)\cdot\boldsymbol{r}}$. $\boldsymbol{A}(t)$ is nonzero only for the velocity gauge (td_stype=1); otherwise $\boldsymbol{A}(t)=0$. Other nonlocal Hamiltonian terms, such as EXX, are not included explicitly. The total current is written to OUT.{suffix}/current_tot.txt. + - 2: Available only for LCAO. Use the full Hamiltonian to construct the generalized velocity matrix in a nonorthogonal NAO basis, $\widetilde{v}_{\alpha}=\partial_{\alpha}H+\mathrm{i}HS^{-1}\mathcal{R}_{\alpha}-\mathrm{i}\mathcal{R}_{\alpha}S^{-1}H-HS^{-1}\partial_{\alpha}S$. This includes all contributions available in the real-space Hamiltonian matrix when enabled. This method is more general but more expensive. The total current is written to OUT.{suffix}/current_tot_comm.txt. - **Default**: 0 ### out_current_k - **Type**: Boolean -- **Availability**: *[`basis_type`](#basis_type)==lcao and [`esolver_type`](#esolver_type)==tddft and [`out_current`](#out_current)>0* -- **Description**: Controls whether LCAO RT-TDDFT current density is also resolved by spin and k-point. The total-current file is always written when out_current is 1 or 2. - - True: In addition to the total, out_current=1 writes OUT.{suffix}/current_s[spin]k[kpoint].txt; out_current=2 writes OUT.{suffix}/current_s[spin]k[kpoint]_comm.txt. Both use one-based spin and k-point numbers, with k-points numbered independently within each spin channel. Each row contains the one-based electronic-step index followed by $J_x$, $J_y$, and $J_z$ in atomic units. +- **Availability**: *[`basis_type`](#basis_type) in [pw, lcao] and [`esolver_type`](#esolver_type)==tddft and [`out_current`](#out_current)>0* +- **Description**: Controls whether RT-TDDFT current density is also resolved by spin and k-point. PW supports out_current=1; LCAO supports out_current=1 or 2. The total-current file is always written when current output is enabled. + - True: In addition to the total, out_current=1 writes OUT.{suffix}/current_s[spin]k[kpoint].txt; out_current=2 writes OUT.{suffix}/current_s[spin]k[kpoint]_comm.txt. Both use one-based spin and k-point numbers, with k-points numbered independently within each spin channel. Rows follow the step numbering, atomic units, precision, and append convention described in out_current. These weighted contributions sum to the total current. - False: Output only current_tot.txt for out_current=1 or current_tot_comm.txt for out_current=2. - **Default**: False @@ -4639,17 +4640,156 @@ - False: Do not output electric-field values. - **Default**: False +[back to top](#full-list-of-input-keywords) + +## Real-Time TDDFT (LCAO) + +### td_edm + +- **Type**: Integer +- **Availability**: *[`basis_type`](#basis_type)==lcao and [`esolver_type`](#esolver_type)==tddft* +- **Description**: Method used to calculate the energy-density matrix for the overlap contribution to forces in LCAO RT-TDDFT. + - 0: Use $\mathrm{EDM}_{\boldsymbol{k}}=\frac{1}{2}\left(S_{\boldsymbol{k}}^{-1}H_{\boldsymbol{k}}\rho_{\boldsymbol{k}}+\rho_{\boldsymbol{k}}H_{\boldsymbol{k}}S_{\boldsymbol{k}}^{-1}\right)$. + - 1: Use the ground-state eigenvalue-weighted expression $\mathrm{EDM}_{\mu\nu,\boldsymbol{k}}=\sum_i w_{i\boldsymbol{k}}\epsilon_{i\boldsymbol{k}}C_{\mu i,\boldsymbol{k}}C_{\nu i,\boldsymbol{k}}^*$. This expression is deprecated for RT-TDDFT and is generally not valid when the propagated wave functions are not Hamiltonian eigenstates. +- **Default**: 0 + +### td_print_eij + +- **Type**: Real +- **Availability**: *[`basis_type`](#basis_type)==lcao and [`esolver_type`](#esolver_type)==tddft* +- **Description**: Controls output of the propagated-state Hamiltonian matrix elements $E_{ij}=\Braket{\psi_i | \hat{H} | \psi_j}$ to the running log. The printed band indices $i$ and $j$ are one-based global indices. Both the threshold and the printed matrix elements are in Ry. + - $\lt 0$: Disable the output. + - $\geqslant 0$: Print an element when either $\left|\operatorname{Re}E_{ij}\right|$ or $\left|\operatorname{Im}E_{ij}\right|$ is greater than or equal to td_print_eij. +- **Default**: -1 +- **Unit**: Ry + +### td_propagator + +- **Type**: Integer +- **Availability**: *[`basis_type`](#basis_type)==lcao and [`esolver_type`](#esolver_type)==tddft* +- **Description**: Method used to propagate the electronic states in a nonorthogonal LCAO basis. The formulas below use Hartree atomic units, with $S$, $H$, and $\Delta t=\mathtt{td\_dt}$ evaluated as required by each approximation. + - 0: Crank-Nicolson through an explicitly constructed evolution matrix, $U=\left[S+\mathrm{i}H\Delta t/2\right]^{-1}\left[S-\mathrm{i}H\Delta t/2\right]$. + - 1: Fourth-order Taylor approximation to the exponential. With $\mathcal{A}=-\mathrm{i}S^{-1}H\Delta t$, $U=I+\mathcal{A}+\mathcal{A}^2/2+\mathcal{A}^3/6+\mathcal{A}^4/24$. + - 2: Enforced time-reversal symmetry (ETRS), $U(t+\Delta t,t)=\exp\left[-\mathrm{i}S^{-1}H(t+\Delta t)\Delta t/2\right]\exp\left[-\mathrm{i}S^{-1}H(t)\Delta t/2\right]$. In the implementation, each exponential is replaced by the fourth-order Taylor polynomial from method 1 evaluated with a half time step. + - 3: Crank-Nicolson by directly solving $\left[S+\mathrm{i}H\Delta t/2\right]\psi(t+\Delta t)=\left[S-\mathrm{i}H\Delta t/2\right]\psi(t)$. + + > Note: GPU execution currently supports only method 0 in both single-GPU and multi-GPU solver configurations. CPU execution supports methods 0 through 3. +- **Default**: 0 + +### init_vecpot_file + +- **Type**: Boolean +- **Availability**: *[`basis_type`](#basis_type)==lcao and [`esolver_type`](#esolver_type)==tddft* +- **Description**: Selects the source of the Cartesian vector potential used by LCAO RT-TDDFT. + - True: Read vector_pot.txt from the calculation working directory. Each non-comment line must contain four columns: a conventionally one-based electronic-step label followed by $A_x$, $A_y$, and $A_z$ in Hartree atomic units. These are propagation values, not endpoint samples. Rows are consumed sequentially; the first column is read as a label and is not used for lookup. If propagation continues beyond the available rows, the last row is reused. + - False: Obtain the vector potential by integrating the configured electric field. +- **Default**: False + ### out_vecpot - **Type**: Boolean - **Availability**: *[`basis_type`](#basis_type)==lcao and [`esolver_type`](#esolver_type)==tddft* -- **Description**: Controls Cartesian vector-potential output for LCAO RT-TDDFT. OUT.{suffix}/vector_pot.txt contains four columns: the one-based electronic-step index followed by $A_x$, $A_y$, and $A_z$ in atomic units. At initialization, a fresh calculation with md_restart=False truncates the file and writes a new header, whereas a calculation with md_restart=True preserves a nonempty existing file and appends new samples. If the restart output file is missing or empty, a new file with a header is created. +- **Description**: Controls Cartesian vector-potential output for LCAO RT-TDDFT. OUT.{suffix}/vector_pot.txt contains four columns: the one-based electronic-step index followed by $A_x$, $A_y$, and $A_z$ in Hartree atomic units. The vector potential is the propagation value: the average of interval endpoints for an integrated electric field, or the supplied sample for init_vecpot_file=True. At initialization, a fresh calculation with md_restart=False truncates the file and writes a new header, whereas a calculation with md_restart=True preserves a nonempty existing file and appends new samples. If the restart output file is missing or empty, a new file with a header is created. - True: Write vector-potential samples on electronic propagation steps. - False: Do not output the vector potential. - **Default**: False [back to top](#full-list-of-input-keywords) +## Real-Time TDDFT (PW) + +### td_orthonormal + +- **Type**: String +- **Availability**: *[`basis_type`](#basis_type)==pw and [`esolver_type`](#esolver_type)==tddft* +- **Description**: Orthonormalize each propagated wavefunction before constructing its density. With the wavefunctions as columns of $\boldsymbol{\Psi}$, define the Gram matrix $\boldsymbol{S}=\boldsymbol{\Psi}^{\dagger}\boldsymbol{\Psi}$ and apply $\boldsymbol{\Psi}\leftarrow\boldsymbol{\Psi}\boldsymbol{C}$. + - `cholesky`: Cholesky orthonormalization. Factor $\boldsymbol{S}=\boldsymbol{R}^{\dagger}\boldsymbol{R}$ with upper-triangular $\boldsymbol{R}$ and use $\boldsymbol{C}=\boldsymbol{R}^{-1}$. + - `lowdin`: Löwdin symmetric orthonormalization. Compute $\boldsymbol{C}=\boldsymbol{S}^{-1/2}$ by eigendecomposition. + - `newton_schulz`: Newton-Schulz iteration for a Gram matrix close to the identity. Starting from $\boldsymbol{C}_0=\boldsymbol{I}$, approximate $\boldsymbol{S}^{-1/2}$ using $\boldsymbol{C}_{j+1}=\boldsymbol{C}_j(3\boldsymbol{I}-\boldsymbol{S}\boldsymbol{C}_j^2)/2$. + - `none`: Disable orthonormalization. + + > Note: Orthonormalization is applied after the Crank-Nicolson linear solve and uses an orthogonality tolerance independent of `lin_thr`. With orthonormalization enabled, the propagated Gram matrix must differ from the identity by at most `1e-6` (single precision) or `1e-12` (double precision) per element; unsuccessful corrections stop the calculation. Wavefunctions within this tolerance are left unchanged. Nonfinite or nonpositive orbital norms are rejected even with `none`. +- **Default**: cholesky + +### td_out_stat + +- **Type**: Boolean +- **Availability**: *[`basis_type`](#basis_type)==pw and [`esolver_type`](#esolver_type)==tddft* +- **Description**: Write wavefunction and density electron counts, their changes from the initial state, and orthogonality errors to the running log after each electronic evolution step. Electron counts describe the end-of-step state; orthogonality errors before and after correction are separate maxima over all SCF iterations and k points within that step. Orthogonality errors are omitted when `td_orthonormal=none`. + + > Note: Enabling this output adds density integration and diagnostic communication. +- **Default**: false + +### lin_solver + +- **Type**: String +- **Availability**: *[`basis_type`](#basis_type)==pw and [`esolver_type`](#esolver_type)==tddft* +- **Description**: Iterative linear solver used for PW real-time propagation. + - `bicgstab`: Biconjugate gradient stabilized (BiCGSTAB) method. + - `cgs`: Conjugate gradient squared (CGS) method. + - `gmres`: Restarted generalized minimal residual (GMRES) method, controlled by `lin_gmres_restart`. + + The initial ground-state diagonalization is controlled by `ks_solver`. +- **Default**: gmres + +### lin_precond + +- **Type**: String +- **Availability**: *[`basis_type`](#basis_type)==pw and [`esolver_type`](#esolver_type)==tddft* +- **Description**: Right preconditioner used by PW real-time linear solvers. The Crank-Nicolson operator is $\boldsymbol{L}=\boldsymbol{I}+\mathrm{i}\Delta t\,\boldsymbol{H}/2$, with the propagation Hamiltonian $\boldsymbol{H}$ in Hartree and time step $\Delta t$ in atomic units. Below, $\boldsymbol{M}^{-1}$ denotes the inverse preconditioner and $\boldsymbol{D}$ the kinetic diagonal inverse. + - `kinetic`: Apply $\boldsymbol{M}^{-1}=\boldsymbol{D}$, with diagonal entries $D_{\boldsymbol{G}\boldsymbol{G}}=(1+\mathrm{i}\Delta t\,T_{\boldsymbol{G}}/2)^{-1}$, where $T_{\boldsymbol{G}}=\lVert\boldsymbol{k}+\boldsymbol{G}+\boldsymbol{A}_{\mathrm{mid}}\rVert^2/2$ is the kinetic energy in Hartree. In the velocity gauge, $\boldsymbol{A}_{\mathrm{mid}}=(\boldsymbol{A}_n+\boldsymbol{A}_{n+1})/2$ is the propagation vector potential in Hartree atomic units; in the length gauge, set $\boldsymbol{A}_{\mathrm{mid}}=\boldsymbol{0}$. + - `kinetic_recycle`: Apply $\boldsymbol{M}^{-1}=\boldsymbol{D}+(\boldsymbol{Z}_{\mathrm{hist}}-\boldsymbol{D}\boldsymbol{W}_{\mathrm{hist}})\boldsymbol{W}_{\mathrm{hist}}^{\dagger}$, where $\boldsymbol{Z}_{\mathrm{hist}}$ and $\boldsymbol{W}_{\mathrm{hist}}$ are paired response directions and approximate operator images from the preceding successful solve at the same k point, with $\boldsymbol{W}_{\mathrm{hist}}^{\dagger}\boldsymbol{W}_{\mathrm{hist}}\approx\boldsymbol{I}$. The images belong to the historical solve, not necessarily the current $\boldsymbol{L}$. Falls back to `kinetic` when no reliable history is available. + - `kinetic_subspace`: Apply $\boldsymbol{M}^{-1}=\boldsymbol{D}+(\boldsymbol{U}-\boldsymbol{D}\boldsymbol{L}\boldsymbol{U})(\boldsymbol{U}^{\dagger}\boldsymbol{L}\boldsymbol{U})^{-1}\boldsymbol{U}^{\dagger}$, where the columns of $\boldsymbol{U}$ are the previous-time wavefunctions defining the coarse subspace. + - `none`: Disable preconditioning. + + Preconditioning changes the convergence rate, while `lin_thr` still controls the residual of the original equation. +- **Default**: kinetic_recycle + +### lin_thr + +- **Type**: Real +- **Availability**: *[`basis_type`](#basis_type)==pw and [`esolver_type`](#esolver_type)==tddft* +- **Description**: Nonnegative finite residual tolerance for each band in a PW real-time Crank-Nicolson solve $\boldsymbol{L}\boldsymbol{x}=\boldsymbol{b}$, with residual $\boldsymbol{r}=\boldsymbol{b}-\boldsymbol{L}\boldsymbol{x}$. A value of 0 selects $\tau=\max(10^{-10},100\epsilon)$, where $\epsilon$ is machine epsilon for the wavefunction precision. This gives a tolerance of approximately $1.19209\times10^{-5}$ in single precision and $10^{-10}$ in double precision. A positive value specifies $\tau$ directly. + - `bicgstab` and `gmres`: Require $\lVert\boldsymbol{r}\rVert\leqslant\tau\max(1,\lVert\boldsymbol{b}\rVert)$. + - `cgs`: Require $\lVert\boldsymbol{r}\rVert\leqslant\tau\lVert\boldsymbol{b}\rVert$ for nonzero $\boldsymbol{b}$, or $\lVert\boldsymbol{r}\rVert\leqslant\tau$ for zero $\boldsymbol{b}$. + + All methods check the residual at the end of the linear solve, before orthonormalization. GMRES can use explicit residual reconstruction with periodic independent checks when `lin_reconstruct` is enabled. Subsequent orthonormalization uses a separate orthogonality tolerance and does not guarantee that the corrected wavefunctions satisfy the same linear residual tolerance. +- **Default**: 0 + +### lin_maxiter + +- **Type**: Integer +- **Availability**: *[`basis_type`](#basis_type)==pw and [`esolver_type`](#esolver_type)==tddft* +- **Description**: Positive maximum number of iterations for each PW real-time linear solve. Failure to converge stops the calculation. The number of self-consistency iterations is controlled separately by `scf_nmax`. +- **Default**: 500 + +### lin_gmres_restart + +- **Type**: Integer +- **Availability**: *[`basis_type`](#basis_type)==pw and [`esolver_type`](#esolver_type)==tddft and [`lin_solver`](#lin_solver)==gmres* +- **Description**: Positive maximum Arnoldi dimension per GMRES cycle. `lin_maxiter` limits the total iterations across all cycles. +- **Default**: 20 + +### td_cn_init + +- **Type**: Boolean +- **Availability**: *[`basis_type`](#basis_type)==pw and [`esolver_type`](#esolver_type)==tddft* +- **Description**: Project the Crank-Nicolson equation $\boldsymbol{L}\boldsymbol{x}=\boldsymbol{b}$ onto the previous-time wavefunction subspace to initialize the first propagation solve of each time step. With those wavefunctions as the columns of $\boldsymbol{U}$, the projected initial guess is $\boldsymbol{x}_0=\boldsymbol{U}(\boldsymbol{U}^{\dagger}\boldsymbol{L}\boldsymbol{U})^{-1}\boldsymbol{U}^{\dagger}\boldsymbol{b}$. Evaluate and reuse the initial residual $\boldsymbol{r}_0=\boldsymbol{b}-\boldsymbol{L}\boldsymbol{x}_0$ using the stored subspace operator images $\boldsymbol{L}\boldsymbol{U}$, without an additional Hamiltonian application. Available with all linear solvers and preconditioners; later self-consistency iterations retain their current wavefunction guess. +- **Default**: true + +### lin_reconstruct + +- **Type**: Boolean +- **Availability**: *[`basis_type`](#basis_type)==pw and [`esolver_type`](#esolver_type)==tddft and [`lin_solver`](#lin_solver)==gmres* +- **Description**: Reconstruct the GMRES residual for $\boldsymbol{L}\boldsymbol{x}=\boldsymbol{b}$ as $\boldsymbol{r}=\boldsymbol{r}_0-(\boldsymbol{L}\boldsymbol{Z})\boldsymbol{y}$ for the update $\boldsymbol{x}=\boldsymbol{x}_0+\boldsymbol{Z}\boldsymbol{y}$, where $\boldsymbol{r}_0=\boldsymbol{b}-\boldsymbol{L}\boldsymbol{x}_0$ is the initial residual, $\boldsymbol{Z}$ contains the current preconditioned search directions, and $\boldsymbol{y}$ contains their update coefficients. Reusing the stored, unmodified operator images $\boldsymbol{L}\boldsymbol{Z}$ reduces Hamiltonian applications. + + Use an internal tolerance of 0.8 times the effective `lin_thr` and independently verify the first solve at each k point and every 16 solves thereafter, with additional independent checks when needed. Failed reconstruction checks trigger a true-residual restart within `lin_maxiter`. + + Only effective for `lin_solver=gmres`; ignored otherwise. +- **Default**: true + +[back to top](#full-list-of-input-keywords) + ## Variables useful for debugging ### nurse diff --git a/docs/advanced/input_files/stru.md b/docs/advanced/input_files/stru.md index 37ae64d8712..885a1e833fc 100644 --- a/docs/advanced/input_files/stru.md +++ b/docs/advanced/input_files/stru.md @@ -4,13 +4,13 @@ - [no latname](#no-latname) - [latname fcc](#latname-fcc) - [Structure of the file](#structure-of-the-file) - - [ATOMIC_SPECIES](#ATOMIC_SPECIES) - - [NUMERICAL_ORBITAL](#NUMERICAL_ORBITAL) - - [LATTICE_CONSTANT](#LATTICE_CONSTANT) - - [LATTICE_VECTORS](#LATTICE_VECTORS) - - [LATTICE_PARAMETERS](#LATTICE_PARAMETERS) - - [ATOMIC_POSITIONS](#ATOMIC_POSITIONS) - - [More Key Words](#More-Key-Words) + - [ATOMIC_SPECIES](#atomic_species) + - [NUMERICAL_ORBITAL](#numerical_orbital) + - [LATTICE_CONSTANT](#lattice_constant) + - [LATTICE_VECTORS](#lattice_vectors) + - [LATTICE_PARAMETERS](#lattice_parameters) + - [ATOMIC_POSITIONS](#atomic_positions) + - [More Key Words](#more-key-words) ## Examples @@ -281,7 +281,7 @@ For general usage requirements, the APNSv1.0 pseudopotential and orbital set is 0.0 0.0 0.0 m 0 0 0 0.5 0.5 0.5 m 1 1 1 ``` - For `nspin==2`, we will autoset atomic magmon is `1.0`: + For `nspin==2`, the atomic magnetic moment is autoset to `1.0`: ``` Fe 1.0 @@ -295,26 +295,39 @@ For general usage requirements, the APNSv1.0 pseudopotential and orbital set is 0.0 0.0 0.0 m 0 0 0 0.5 0.5 0.5 m 1 1 1 ``` - For `nspin==4`, we will autoset atomic magmon as follow: + This autoset is skipped as soon as `STRU` specifies a finite moment for + any single atom, and also when [`symmetry`](input-main.md#symmetry) is + `1`, because an all-zero moment is a legitimate nonmagnetic choice under + the full point group. + + For `nspin==4`, **no** moment is autoset: the calculation starts from a + zero magnetic moment and a warning is printed. Zero magnetization is a + stationary point, so such a run stays nonmagnetic. Set `mag` explicitly + on the magnetic atoms if a magnetic ground state is expected, e.g.: ``` Fe 0.0 2 - 0.0 0.0 0.0 m 0 0 0 mag 1 1 1 - 0.5 0.5 0.5 m 1 1 1 mag 1 1 1 + 0.0 0.0 0.0 m 0 0 0 mag 0.0 0.0 2.0 + 0.5 0.5 0.5 m 1 1 1 mag 0.0 0.0 2.0 O 0.0 2 - 0.0 0.0 0.0 m 0 0 0 mag 1 1 1 - 0.5 0.5 0.5 m 1 1 1 mag 1 1 1 + 0.0 0.0 0.0 m 0 0 0 + 0.5 0.5 0.5 m 1 1 1 ``` - However, this autoset will not be vaild once `STRU` specalize a finite magnetic for any single atom. + With [`noncolin`](input-main.md#noncolin) `0`, only the z component of + `mag` is used; the x/y components are discarded and the number of + affected atoms is reported per atom type. A moment given purely along + x/y therefore also leaves a zero starting moment. - `lambda`: Lagrange multiplier vector for spin constraint method. Can specify one value (z-component) or three values for x, y, z components (e.g., `lambda 0.5` or `lambda 0.1 0.2 0.3`). Values are in eV and will be converted to Rydberg internally. Used with spin-constrained DFT (enable with `sc_mag_switch` in INPUT file). - `sc`: set the spin constraint target magnetization for each atom. Can specify one value (z-component) or three values for x, y, z components (e.g., `sc 1.0` or `sc 0.5 0.5 1.0`). Used with spin-constrained DFT (enable with `sc_mag_switch` in INPUT file). + - `f` or `force` or `forces`: three numbers giving the per-atom force (fx, fy, fz) in **eV/Angstrom**. This field is *output-only*: it is written into structure files (STRU{istep}, STRU_NOW, STRU_FINAL) when [`out_stru`](input-main.md#out_stru) is enabled, [`cal_force`](input-main.md#cal_force) is true, and the geometry has been evaluated. When such a file is read back as an input STRU, the `f` values are consumed but ignored -- they are informational only and do not affect the calculation. + ### Important Notes for ATOMIC_POSITIONS 1. **Coordinate System Selection**: Choose the appropriate coordinate system based on your needs: @@ -340,4 +353,31 @@ For general usage requirements, the APNSv1.0 pseudopotential and orbital set is - Don't mix Direct and Cartesian coordinates in the same STRU file - Ensure the number of atoms specified matches the actual number of coordinate lines provided - When using vector magnetization (`mag x y z`), don't also specify angles for the same atom - - Remember that angles are in degrees, not radians \ No newline at end of file + - Remember that angles are in degrees, not radians + +### Header of structure files written by ABACUS + +When [`out_stru`](input-main.md#out_stru) writes STRU{istep}, STRU_NOW or STRU_FINAL, the file begins with a fixed **7-line** comment header so that downstream parsers can skip a constant-size block: + +``` +# ABACUS version: +# Written at +# RELAX STEP [ (FINAL)], Energy: eV +# Stress (kbar): sxx sxy sxz +# Stress (kbar): syx syy syz +# Stress (kbar): szx szy szz +# NOTE: +``` + +Rules: + +- Line 3 carries the total energy in eV only when that energy was computed for the geometry written below. If the geometry was proposed by the optimizer but never evaluated (early relaxation exit), line 3 is written as `# RELAX STEP [ (FINAL)], Energy: N/A` so the energy label and the coordinates always describe the same frame. +- Lines 4-6 always carry the stress tensor in **kbar**, one row per line. If stress was not computed ([`cal_stress`](input-main.md#cal_stress) = 0) or the geometry was proposed by the optimizer but not yet evaluated (early relaxation exit), all three rows are written as `# Stress (kbar): N/A N/A N/A` with no extra inline comment, so every stress line has the same token shape. +- Line 7 is a single `# NOTE:` line describing the state: + - `# NOTE: stress and forces computed for this geometry` (normal case) + - `# NOTE: stress not computed (cal_stress=0)` + - `# NOTE: forces not computed (cal_force=0); per-atom f fields omitted intentionally` + - `# NOTE: stress not computed (cal_stress=0); forces not computed (cal_force=0); per-atom f fields omitted intentionally` + - `# NOTE: geometry proposed by optimizer but not evaluated; energy N/A; stress N/A; forces omitted` +- Per-atom `f fx fy fz` fields (forces in eV/Angstrom) are appended to the atom lines only when [`cal_force`](input-main.md#cal_force) is true **and** the geometry was evaluated. They are omitted on early relaxation exit so the geometry and the property data always come from the same configuration. +- The coordinate format line of the ATOMIC_POSITIONS section is always `Cartesian_angstrom`, independent of force availability; `has_force` only controls the per-atom `f` fields. `Direct` and the other coordinate keywords remain fully supported as **input** formats. diff --git a/docs/advanced/install.md b/docs/advanced/install.md index a69352dfb97..ab1939181be 100644 --- a/docs/advanced/install.md +++ b/docs/advanced/install.md @@ -99,6 +99,10 @@ vdw_d4_model d4 # or d4s for the smooth D4S model If `vdw_d4_xc` is set to `default`, ABACUS will infer the functional name from `dft_functional` or pseudopotential metadata and pass it to the DFT-D4 library. The `vdw_d4_model` keyword selects the dispersion model inside the DFT-D4 library; the default is `d4`, while `d4s` enables the smooth D4S model. +## Build with JSON support + +ABACUS uses [nlohmann-json](https://github.com/nlohmann/json) for optional JSON output. To build ABACUS with JSON support, pass `-DENABLE_JSON=ON` to CMake. + ## Build Unit Tests To build tests for ABACUS, define `BUILD_TESTING` flag. You can also specify path to local installation of [Googletest](https://github.com/google/googletest) by setting `GTEST_DIR` flags. If not found in local, the configuration process will try to download it automatically. @@ -283,7 +287,7 @@ When `CXX=mpicxx`, a parallel version will be compiled. When `CXX=g++`, a sequen Except modifying `Makefile.vars`, you can also directly use -```makefile +```bash make CXX=mpiicpc ELPA_DIR=/public/soft/elpa_21.05.002 \ ELPA_INCLUDE_DIR=${ELPA_DIR}/include/elpa-2021.05.002 \ CEREAL_DIR=/public/soft/cereal @@ -301,7 +305,7 @@ The program compiled using the above instructions do not link with LIBXC and use To compile ABACUS with LIBXC, you need to define `LIBXC_DIR` in the file `Makefile.vars` or use -```makefile +```bash make LIBXC_DIR=/pulic/soft/libxc ``` @@ -311,7 +315,7 @@ directly. To compile ABACUS with machine learning algorithms, you need to set `ENABLE_MLALGO = ON`, and define `LIBTORCH_DIR` and `LIBNPY_DIR` in the file `Makefile.vars` or use -```makefile +```bash make ENABLE_MLALGO=ON LIBTORCH_DIR=/opt/libtorch/ LIBNPY_DIR=/opt/libnpy/ ``` @@ -325,22 +329,22 @@ directly. To compile ABACUS with DeePMD-kit, you need to define `DeePMD_DIR` and `TensorFlow_DIR` (TensorFlow Backend, optional) and/or `LIBTORCH_DIR` (PyTorch Backend, optional) in the file `Makefile.vars`. Or the `tensorflow_cc` and `torch` libraries are in the same directory as the `deepmd_c`/`deepmd_cc` libraries, then -```makefile +```bash make DeePMD_DIR=/dir_to_deepmd-kit ``` DeePMD-kit supports TensorFlow backend but its libraries are placed at another directory, then -```makefile +```bash make DeePMD_DIR=/dir_to_deepmd-kit TensorFlow_DIR=/dir_to_tensorflow ``` Similarly, DeePMD-kit supports PyTorch backend but its libraries are placed at another directory, then -```makefile +```bash make DeePMD_DIR=/dir_to_deepmd-kit Torch_DIR=/dir_to_pytorch ``` ### Add LibRI Support To use new EXX, you need two libraries: [LibRI](https://github.com/abacusmodeling/LibRI) and [LibComm](https://github.com/abacusmodeling/LibComm) and need to define `LIBRI_DIR` and `LIBCOMM_DIR` in the file `Makefile.vars` or use -```makefile +```bash make LIBRI_DIR=/public/software/LibRI LIBCOMM_DIR=/public/software/LibComm ``` directly. diff --git a/docs/advanced/interface/dpgen.md b/docs/advanced/interface/dpgen.md index bcfb1b1cb79..9d1d926bf25 100644 --- a/docs/advanced/interface/dpgen.md +++ b/docs/advanced/interface/dpgen.md @@ -14,7 +14,8 @@ Secondly, for the "dpgen init_bulk" step, an `init.json` file should be provided ```json { - "init_fp_style": "ABACUS", # abacus interface + "init_fp_style": "ABACUS", + "_comment": "abacus interface", "stages": [1,2,3,4], "cell_type": "fcc", "super_cell": [2, 1, 1], @@ -49,7 +50,8 @@ Next, for the "dpgen run" step, the following `run_param.json` should be provide "init_data_sys": [ "Al.STRU.01x01x01/02.md/sys-0004/deepmd" ], - "sys_format": "abacus/stru", # the initial structures are in ABACUS/STRU formate + "_comment": "the initial structures are in ABACUS/STRU format", + "sys_format": "abacus/stru", "sys_configs_prefix": "./", "sys_configs": [ [ @@ -162,11 +164,15 @@ Next, for the "dpgen run" step, the following `run_param.json` should be provide "fp_task_max": 20, "fp_task_min": 5, "fp_pp_path": "./", - "fp_pp_files": ["Al_ONCV_PBE-1.0.upf"], # the pseudopotential file - "fp_orb_files": ["Al_gga_9au_100Ry_4s4p1d.orb"], # the orbital file (use only in LCAO calculation) - "k_points":[2, 2, 2, 0, 0, 0], # k-mesh setting - "user_fp_params":{ # All the ABACUS input paramters are defined here - "ntype": 1, # defining input parameters from INPUT files is not supported yet. + "_comment": "the pseudopotential file", + "fp_pp_files": ["Al_ONCV_PBE-1.0.upf"], + "_comment": "the orbital file is used only in LCAO calculation", + "fp_orb_files": ["Al_gga_9au_100Ry_4s4p1d.orb"], + "_comment": "k-mesh setting", + "k_points":[2, 2, 2, 0, 0, 0], + "_comment": "All the ABACUS input paramters are defined through user_fp_params, and defining input parameters from INPUT files is not supported yet.", + "user_fp_params":{ + "ntype": 1, "ecutwfc": 80, "mixing_type": "broyden", "mixing_beta": 0.8, @@ -176,8 +182,10 @@ Next, for the "dpgen run" step, the following `run_param.json` should be provide "smearing_method": "mp", "smearing_sigma": 0.002, "scf_thr":1e-8, - "cal_force":1, # calculate force must be set to 1 in dpgen calculation - "kspacing": 0.01 # when KSPACING is set, the above k_points setting becomes invalid. + "_comment": "calculate force must be set to 1 in dpgen calculation", + "cal_force":1, + "_comment": "when KSPACING is set, the above k_points setting becomes invalid", + "kspacing": 0.01 } } ``` @@ -264,7 +272,7 @@ This example can be found in examples/dpgen-example/autotest directory. "remote_profile":{ "email": "xxx@xxx.xxx", "password": "xxx", - "program_id": 000, + "program_id": "000", "input_data":{ "api_version":2, "job_type": "indicate", @@ -298,7 +306,7 @@ This example can be found in examples/dpgen-example/autotest directory. "remote_profile":{ "email": "xxx@xxx.xxx", "password": "xxx", - "program_id": 000, + "program_id": "000", "input_data":{ "api_version":2, "job_type": "indicate", @@ -332,7 +340,7 @@ This example can be found in examples/dpgen-example/autotest directory. "remote_profile":{ "email": "xxx@xxx.xxx", "password": "xxx", - "program_id": 000, + "program_id": "000", "input_data":{ "api_version":2, "job_type": "indicate", diff --git a/docs/advanced/interface/index.rst b/docs/advanced/interface/index.rst index 6622e821b7e..1033502a281 100644 --- a/docs/advanced/interface/index.rst +++ b/docs/advanced/interface/index.rst @@ -19,3 +19,4 @@ Interfaces to Other Softwares ShengBTE candela TB2J + migration-guide-csr-format diff --git a/docs/advanced/interface/migration-guide-csr-format.md b/docs/advanced/interface/migration-guide-csr-format.md index d1300690b56..0703ed22980 100644 --- a/docs/advanced/interface/migration-guide-csr-format.md +++ b/docs/advanced/interface/migration-guide-csr-format.md @@ -1,10 +1,10 @@ -# Migration Guide: New CSR Format for H(R) and S(R) Matrices +# Migration Guide: New CSR Format for H\(R\) and S\(R\) Matrices ## Overview -Starting from ABACUS v3.9.0.25, the output format for Hamiltonian H(R) and overlap S(R) matrices has been unified to use standard CSR (Compressed Sparse Row) format, matching the format used by `out_dmr` for density matrices. +Starting from ABACUS v3.9.0.25, the output format for Hamiltonian H\(R\) and overlap S\(R\) matrices has been unified to use standard CSR (Compressed Sparse Row) format, matching the format used by `out_dmr` for density matrices. -This change affects downstream tools that read H(R) and S(R) matrices, including: +This change affects downstream tools that read H\(R\) and S\(R\) matrices, including: - TB2J (magnetic exchange parameters) - DeepH (machine learning Hamiltonian) - pyATB (tight-binding analysis) diff --git a/docs/advanced/interface/pyabacus.md b/docs/advanced/interface/pyabacus.md index ffe32d43502..36d06fe9ea4 100644 --- a/docs/advanced/interface/pyabacus.md +++ b/docs/advanced/interface/pyabacus.md @@ -192,10 +192,10 @@ if bands_file: |------|-------------| | `running_scf.log` | Main calculation log | | `BANDS_1.dat` | Band structure data | -| `PDOS` | Projected density of states | -| `CHARGE.cube` | Charge density in cube format | -| `SPIN1_CHG.cube` | Spin-up charge density | -| `SPIN2_CHG.cube` | Spin-down charge density | +| `pdoss1_*.txt` | Projected density of states (one file per spin channel) | +| `chg.cube` | Charge density in cube format | +| `chgs1.cube` | Spin-up charge density | +| `chgs2.cube` | Spin-down charge density | | `istate.info` | Band eigenvalues and occupations | | `kpoints` | K-point information | diff --git a/docs/advanced/json/index.rst b/docs/advanced/json/index.rst new file mode 100644 index 00000000000..172481781d8 --- /dev/null +++ b/docs/advanced/json/index.rst @@ -0,0 +1,7 @@ +================================================= +JSON Usage +================================================= + +.. toctree:: + json_add + json_para diff --git a/docs/advanced/json/json_add.md b/docs/advanced/json/json_add.md index 826b77119f1..2ec0ee9d2c5 100644 --- a/docs/advanced/json/json_add.md +++ b/docs/advanced/json/json_add.md @@ -1,219 +1,134 @@ -# Table of Contents +# ABACUS JSON Development Guide -1. [Abacus-Json Usage Instructions](#1-abacus-json-usage-instructions) - - [Normal Usage](#normal-usage) - - [Add/Modify a value to object json node](#addmodify-a-value-to-object-json-node-key2-is-a-object-node) - - [Pushback a value to array json node](#pushback-a-value-to-array-json-node-key2-is-a-array-node) - - [Initialization and Assignment Functions for Different Value Types in Arrays](#initialization-and-assignment-functions-for-different-value-types-in-arrays) - - [Object Type](#object-type) - - [Array Type](#array-type) - - [Array Modification Instructions](#array-modification-instructions) -2. [Json Codes Addition Guidelines](#2-json-codes-addition-guidelines) - - [Abacus JSON Functionality Code Structure](#abacus-json-functionality-code-structure) - - [Add JSON code principles](#add-json-code-principles) +## Overview +ABACUS uses [nlohmann-json](https://github.com/nlohmann/json) for its optional JSON output. The implementation lives in `source/source_io/module_json` and uses `Json::jsonValue`, an alias for `nlohmann::ordered_json`, to retain object-key insertion order. +`AbacusJson` provides access to the shared document and writes it to a file. Its declarations are in namespace `Json`: -# 1. Abacus-Json Usage Instructions +```cpp +using jsonValue = nlohmann::ordered_json; -In Abacus, the main utility functions for manipulating JSON trees are outlined below. These functions are used to add objects to Abacus JSON trees. +class AbacusJson +{ + public: + static jsonValue& document(); + static void write_to_json(const std::string& filename); -Function signature: -void AbacusJson::add_json (std::vector keys, const T& value,bool IsArray) -Where: -- `keys` is a vector of string dictionaries, representing the node paths where values are to be added in the JSON tree. -- `value` is a generic value type, including int, bool, double, string, or rapidjson value type, indicating the value to be added to the Abacus JSON. -- `IsArray` is a boolean object, indicating whether the current node being added is an array. `true` represents an array node, while `false` represents a non-array node. + private: + static jsonValue doc; +}; +``` -Example usage: -const std::string version = "v3.5.2"; -AbacusJson::add_json({"general_info", "version"}, version, false); +Keep the document root an object. Its state remains shared within each process; this change does not introduce independent output contexts or make concurrent writes safe. The mutable accessor is for the schema generators and tests in `module_json`. Other modules should continue to pass data to functions such as `add_output_energy()` instead of directly editing the document. +The old path-component type and generic set/append interface have been removed. Use native object assignment, shallow `update()`, and array `push_back()` inside the schema generators; do not introduce another generic path wrapper. +`abacusjson.h` includes only `nlohmann/json_fwd.hpp`. Source files that construct or manipulate JSON values must include `nlohmann/json.hpp` under `__JSON`. The existing CMake option `ENABLE_JSON` controls this feature. Callers using only the higher-level declarations in `init_info.h` or `output_info.h` do not need the backend header. -## Normal usage +## Constructing metadata -### Add/Modify a value to object json node (key2 is a object node): -```cpp -Json::AbacusJson::add_json({"key1","key2"}, 3.1415,false); -``` +`gen_general_info()` owns the whole `general_info` section and assigns it as a complete object: -### Pushback a value to array json node (key2 is a array node): ```cpp -Json::AbacusJson::add_json({"key1","key2"}, 3.1415,true); +AbacusJson::document()["general_info"] = { + {"version", version}, + {"commit", commit}, + {"device", param.inp.device}, + {"mpi_num", mpi_num}, + {"omp_num", omp_num}, + {"pseudo_dir", param.inp.pseudo_dir}, + {"orbital_dir", param.inp.orbital_dir}, + {"stru_file", param.globalv.global_in_stru}, + {"kpt_file", param.inp.kpoint_file}, + {"start_time", start_time_str}, + {"end_time", end_time_str}}; ``` -Through this function alone, the addition of the majority of JSON parameters can be achieved. However, for complex array types, additional operations are required. - - -## Initialization and Assignment Functions for Different Value Types in Arrays - -### Object Type: -Since the object type consists of key-value pairs, four member functions are divided based on whether key and val are of type std::string. - -- JaddStringV(str,val): key is not string, val is string -- JaddStringK(str,val): key is string, val is not string -- JaddStringKV(str,val): both key and val are string -- JaddNormal(str,val): both key and val are not string - - -### Array Type: -For array types, the following member functions are used directly. - -- JPushBack(val): val is not string -- JPushBackString(val): val is string - -For example, to add nodes to a JSON tree with multiple arrays in Abacus, the following code is needed: +The `init` section is shared by `gen_stru()`, `gen_init()`, and `add_nkstot()`. The first two construct the fields they own in a local object, then apply a **shallow** update: ```cpp -// add key-val to an object array -for(int i=0;i<1;i++){ - Json::jsonValue object(JobjectType); - std::string str = std::to_string(i*100); - - object.JaddNormal("int",i); - object.JaddStringV("string", str); - Json::AbacusJson::add_json({"array"}, object,true); -} +// Inside init_info.cpp; init_section() is local to this source file. +init_section().update(info); ``` -```cpp -// add array in array -Json::jsonValue object0(JarrayType); +The local helper creates a missing `init` object but rejects an existing non-object, including `null`. The update preserves fields supplied by the other generators and replaces each supplied value as a whole. In particular, per-species maps and coordinate arrays must not retain stale entries or accumulate on repeated generation. Do not assign a newly generated object to the entire `init` section, and do not enable recursive object merging here. -object0.JPushBack(1); -object0.JPushBack(2); -object0.JPushBack(3); +`add_nkstot()` only sets its own field: -Json::AbacusJson::add_json({"Darray"}, object0,true); +```cpp +init_section()["nkstot"] = nkstot; ``` +## Output-record lifecycle +The workflow starts each record with `init_output_array_obj()` **before** the corresponding solver writes SCF or other result data. That function alone creates the `output` array and appends the initial record. It rejects an existing `output` value that is not an array; an explicit `null` is not treated as a missing field. +The existing workflow entry points own this initialization: +| Workflow | Record initialization | +| --- | --- | +| SCF/relaxation | `Relax_Driver::iter_info()` starts the record, except for the first `ks-lr` step described below. | +| `ks-lr` | `ESolver_LR::before_all_runners()` starts the record before its embedded KS calculation; the first relaxation-driver step reuses it. | +| UnitCell-backed MD | `Run_MD::md_line()` starts a record at the beginning of each MD iteration when `mdcell.has_backing_unitcell()` is true. | +| Socket/i-PI | `SocketHandlers::handle_posdata()` starts a record before running the solver for the received `POSDATA` frame. | -## Array Modification Instructions +Do not move record creation into individual field writers, create a second record for the same step, or reset the whole document to start a new step. -For values that need to be modified in arrays, the following method can be used: -- The index number of the array starts at 0, if it's negative, it's going from back to front. eg. If the index is -1, it means that the last element of the array is modified: -- If the path contains an array, use the array index directly. +The result writers use `current_output()`, a helper local to `output_info.cpp`. It rejects a missing or non-array `output`, an empty array, or a final element that is not an object. It never creates a record as a side effect of writing a result. -```cpp -AbacusJson::add_json({"path",index }, value, is_array); -``` +For example, inside namespace `Json` in `output_info.cpp`: -Here, index is a number. index >= 0 indicates the index from the beginning of the array, while index < 0 indicates traversal from the end of the array. - -For example, to modify the value of "vasp" to "cp2k" in the following JSON tree: - -```json -"Json":{ - "key6": { - "key7": [ - { - "a":1, - "new":2 - } - "vasp", - "abacus" - ] - } +```cpp +void add_output_energy(const double energy) +{ + current_output()["energy"] = energy; } ``` -The relative path of "vasp" in layman's terms is Json - key6 - key7[0]. To use the JSON modification method in abacus, simply change the index [0] to "0". -```cpp -AbacusJson::add_json({"Json","key6","key7",1}, "cp2k" , false); -``` +Coordinate, force, stress, magnetic-moment, and cell arrays are built locally and assigned as complete arrays. Repeatedly updating the same record must replace these arrays rather than append rows. -If traversal is done from the end: -```cpp -AbacusJson::add_json({"Json","key6","key7",-2}, "cp2k", false); -``` +SCF iterations are different: they form a history and must be appended. `add_output_scf_mag()` creates a missing `scf` array, rejects an existing non-array history, and appends one iteration object. Its implementation uses: -An error is reported if index exceeds the array length! ```cpp -AbacusJson::add_json({"Json","key6","key7",3}, "cp2k", false); +jsonValue& output = current_output(); +output["total_mag"] = total_mag; +output["absolute_mag"] = absolute_mag; +jsonValue& scf = *output.emplace("scf", jsonValue::array()).first; +if (!scf.is_array()) +{ + throw std::invalid_argument("JSON SCF history must be an array"); +} +scf.push_back({{"energy", energy}, {"ediff", ediff}, + {"drho", drho}, {"time", time}}); ``` +`ordered_json` may invalidate references to child values when new members are inserted into their parent object. Acquire the `scf` reference after inserting `total_mag` and `absolute_mag`, and do not retain a record reference across appending another `output` record. The same caution applies to references to root sections when new root keys are inserted. +## Serialization and tests +`document()` and `write_to_json()` do not perform MPI rank filtering. The existing `json_output()` wrapper writes `abacus.json` only on rank 0 in MPI builds; callers outside `module_json` should retain the existing integration wrappers. -# 2. Abacus Json Codes Addition Guidelines - -## Abacus JSON Functionality Code Structure +`write_to_json()` preserves the existing four-space formatting and reports file-open and write/close failures. It serializes the document before opening the destination, so a serialization error does not first truncate the file. Non-finite numbers serialize as JSON `null`; decimal versus scientific float notation is not part of the schema contract. -The current code structure of JSON functionality in Abacus is roughly as follows: +The tests reset the shared document through `document()` in their fixture; no access-control macro or friend accessor is needed. Focus coverage on ABACUS behavior: generated fields and units, repeated metadata updates, record initialization and SCF accumulation, invalid section types, insertion order, escaping and non-finite values through the real writer, and file errors. Do not replace removed path-walker tests with tests of nlohmann-json's generic container API. -- source/source_io - - para_json.cpp: Contains JSON generation and output interfaces directly called by the device in Abacus. - - json_output/: Contains the functionality encapsulation class `abacusjson.cpp` of RapidJSON in Abacus and code classes for parameter generation in various JSON modules. - - test: Code testing files in `json_output`. +## Code structure +```text +source/source_io/module_json/ +├── abacusjson.cpp/.h # shared document and file output +├── general_info.cpp/.h # general_info section +├── init_info.cpp/.h # comment and init sections +├── output_info.cpp/.h # output records and lifecycle checks +├── para_json.cpp/.h # integration-facing wrappers +└── test/ # focused unit tests +``` -## Add JSON code principles: -In Abacus JSON addition, the following principles need to be followed: - -1. Whenever possible, code to be added in the module should be written in the `json_output` module (there may also be cases where parameters cannot be directly obtained through parameter passing in `json_output`), and then called in the path `para_json.cpp` -> `device.cpp` or other main execution paths. (Ensure minimal impact on other modules as much as possible) - -2. For parameters that can be obtained without depending on other modules, do not reference parameter values saved in other modules. (Such as `mpi_num`, `start_time`) - -3. Use classes as function parameters as much as possible instead of using global classes for obtained parameters. (For example, in `gen_general_info`, `Input`) +`init_section()` and `current_output()` are file-local helpers, not public interfaces for workflow callers. -4. After adding parameters, supplement test code in `source_io/json_output/test`. +## Guidelines for extending JSON output -For the current JSON file, there are two JSON modules: `init` and `general_info`, `output_info`. -Taking `general_info` as an example, the code to be added is as follows: +Keep construction in the existing schema generator, pass its required data explicitly, and avoid adding `GlobalV`, `GlobalC`, or `PARAM` access. Preserve field names, value types, units, and order unless a schema change is intentional. Add focused tests for new fields and lifecycle behavior, and update the [JSON output reference](json_para.md) when the public schema changes. -```cpp -namespace Json -{ - -#ifdef __RAPIDJSON -void gen_general_info(const Parameter& param) -{ - -#ifdef VERSION - const std::string version = VERSION; -#else - const std::string version = "unknown"; -#endif -#ifdef COMMIT - const std::string commit = COMMIT; -#else - const std::string commit = "unknown"; -#endif - - // start_time - std::time_t start_time = input->get_start_time(); - std::string start_time_str; - convert_time(start_time, start_time_str); - - // end_time - std::time_t time_now = std::time(NULL); - std::string end_time_str; - convert_time(time_now, end_time_str); - -#ifdef __MPI - int mpi_num = Parallel_Global::mpi_number; - int omp_num = Parallel_Global::omp_number; -#elif - int mpi_num = 1; - int omp_num = 1; -#endif - - AbacusJson::add_json({"general_info", "version"}, version,false); - AbacusJson::add_json({"general_info", "commit"}, commit,false); - AbacusJson::add_json({"general_info", "device"}, input->device,false); - AbacusJson::add_json({"general_info", "mpi_num"}, mpi_num,false); - AbacusJson::add_json({"general_info", "omp_num"}, omp_num,false); - AbacusJson::add_json({"general_info", "pseudo_dir"}, input->pseudo_dir,false); - AbacusJson::add_json({"general_info", "orbital_dir"}, input->orbital_dir,false); - AbacusJson::add_json({"general_info", "stru_file"}, input->stru_file,false); - AbacusJson::add_json({"general_info", "kpt_file"}, input->kpoint_file,false); - AbacusJson::add_json({"general_info", "start_time"}, start_time_str,false); - AbacusJson::add_json({"general_info", "end_time"}, end_time_str,false); -} -#endif -} // namespace Json -``` \ No newline at end of file +Keep examples and new implementation code compatible with the C++11 baseline. Include complete domain-type definitions in the source or test file that needs them, keep public header dependencies minimal, and do not reintroduce access-control macros for testing. diff --git a/docs/advanced/json/json_para.md b/docs/advanced/json/json_para.md index 36e5cbebc9b..eba7f35022a 100644 --- a/docs/advanced/json/json_para.md +++ b/docs/advanced/json/json_para.md @@ -1,85 +1,88 @@ -# JSON Configuration Parameters Documentation +# JSON Output Documentation -- [JSON Configuration Parameters Documentation](#json-configuration-parameters-documentation) +- [JSON Output Documentation](#json-output-documentation) - [Overview](#overview) - [General Information](#general-information) - - [Input](#input) - - [Init](#init) + - [Initialization Information](#initialization-information) - [Output](#output) - - [Final Structure](#final-structure) + - [Serialization](#serialization) ## Overview -This JSON template provides input and output configurations for ABACUS. It contains parameters for the execution of the program and the output of results, primarily for recording computational processes and outcomes for post processing. - -Notice: one need to add the option `-DENABLE_RAPIDJSON=ON` when compiling ABACUS to enable the output of "abacus.json". +When JSON support is enabled with the CMake option `ENABLE_JSON`, ABACUS writes calculation metadata and results to `abacus.json` for post-processing using nlohmann-json. In MPI builds, the output wrapper writes this file only on rank 0. +The current top-level JSON members are `comment`, `init`, `output`, and `general_info`. Some fields are populated only when the corresponding calculation data are available. The native-schema refactor changes the internal construction API, not these field names or their units. See the [JSON development guide](json_add.md) for implementation details. ## General Information -- `version` - [str] The version number of ABACUS. -- `commit` - [str] The commit hash of ABACUS code at the time of computation. -- `start_time` - [str] The start time of the computation. -- `end_time` - [str] The end time of the computation. -- `device` - [str] The name of the hardware device on which the computation was run. -- `omp_number` - [int] The number of OpenMP threads. -- `mpi_number` - [int] The number of MPI processes. -- `out_dir` - [str] The output directory, e.g., "OUT.ABACUS". -- `log_file` - [str] The name of the log file, e.g., "running_scf.log". -- `pseudo_dir` - [str] The directory where pseudopotential files are stored. -- `orbital_dir` - [str] The directory where atomic orbital files are stored. -- `stru_file` - [str] The name of the structure file. -- `kpt_file` - [str] The name of the k-point file. +The `general_info` object records basic build and runtime metadata: + +- `version` - [string] ABACUS version. +- `commit` - [string] Git commit information when available at build time. +- `device` - [string] Hardware device selected for the calculation. +- `mpi_num` - [int] Number of MPI processes. +- `omp_num` - [int] Number of OpenMP threads. +- `pseudo_dir` - [string] Pseudopotential directory. +- `orbital_dir` - [string] Numerical atomic orbital directory. +- `stru_file` - [string] Structure input file. +- `kpt_file` - [string] K-point input file. +- `start_time` - [string] Calculation start time. +- `end_time` - [string] Time at which the JSON output is finalized. + +## Initialization Information + +The top-level `comment` describes the default units used by the JSON output. The `init` object records the initial structure and calculation settings. Depending on the calculation path, it can contain: + +- `element` - [object(string:string)] Element/pseudopotential element information keyed by atom label. +- `orb` - [object(string:string/null)] Numerical orbital path for each atom type, formed by concatenating the configured orbital-directory string and the per-type filename; `null` when that combined string is empty. +- `pp` - [object(string:string)] Pseudopotential file for each atom type. +- `coordinate` - [array(array(double))] Initial Cartesian coordinates in Angstrom. +- `mag` - [array(double)] Initial magnetic moment for each atom. +- `label` - [array(string)] Atomic labels. +- `cell` - [array(array(double))] Initial lattice vectors in Angstrom. +- `point_group` - [string] Schoenflies name of the point group. +- `point_group_in_space` - [string] Schoenflies name of the point group in the space group. +- `natom` - [int] Total number of atoms. +- `nband` - [int] Number of bands. +- `natom_each_type` - [object(string:int)] Number of atoms of each type. +- `nelectron_each_type` - [object(string:double)] Number of valence electrons for each atom type. +- `nelectron` - [int] Total number of electrons. +- `ecutwfc` - [double] Wavefunction energy cutoff. +- `ecutwfc_unit` - [string] Unit of `ecutwfc`, currently `Ry`. +- `smearing_method` - [string] Smearing method. +- `smearing_sigma` - [double] Smearing width. +- `smearing_sigma_unit` - [string] Unit of `smearing_sigma`, currently `Ry`. +- `kmesh_type` - [string] K-point mesh type. +- `kspacing` - [array(double)] K-point spacing parameters. +- `koffset` - [array(double)] K-point mesh offsets. +- `nkstot` - [int] Total number of k-points, when available on the calculation path. -## Input -- A dictionary of parameters and their values as defined by the user in the INPUT file. (This part of the content will not be output in the current version yet.) +## Output +`output` is an array. Each element represents one calculation/ionic-step output record, initialized before its results are written. A newly initialized record has `null` values for `e_fermi`, `energy`, `scf_converge`, `force`, and `stress`, and empty arrays for `coordinate`, `mag`, and `cell`. The `total_mag`, `absolute_mag`, and `scf` members are added by the SCF writer. -## Init +Fields are filled as the corresponding results become available; not every workflow populates all of them: +- `energy` - [double/null] Total energy in eV. +- `e_fermi` - [double/null] Fermi energy in eV. +- `scf_converge` - [bool/null] Whether the SCF calculation converged. +- `force` - [array(array(double))/null] Atomic forces in eV/Angstrom when force calculation is enabled. +- `stress` - [array(array(double))/null] Stress tensor in kbar when stress calculation is enabled. +- `coordinate` - [array(array(double))] Cartesian coordinates in Angstrom. +- `mag` - [array(double)] Magnetic moment for each atom. +- `cell` - [array(array(double))] Lattice vectors in Angstrom. +- `total_mag` - [double] Total magnetic moment when available. +- `absolute_mag` - [double] Absolute magnetic moment when available. +- `scf` - [array(object)] SCF iteration history. Each entry contains: + - `energy` - [double] Total energy in eV. + - `ediff` - [double] Energy change from the previous SCF step in eV. + - `drho` - [double] Charge-density difference. + - `time` - [double] Time used by the SCF step in seconds. -- `Input` - Lists the value of all input parameters. (This part of the content will not be output in the current version yet.) -- `point_group` - [str] the Schoenflies name of the point group. -- `point_group_in_space` - [str] the Schoenflies name of the point group in the space group. -- `nkstot`, `nkstot_ibz` - [int] Total number of k-points and total number of irreducible k-points. -- `nelectron_each_type` - [object(str-int)] The number of valence electron for each atom type, e.g., `{"C": 2, "H":1}`. -- `nelectron` - [int] Total number of electrons. -- `nband` - [int] Number of bands. -- `natom` - [int] Total number of atoms. -- `natom_each_type` - [object(str-int)] The atom number of each atom type, e.g., `{"C": 2, "H":1}`. -- `label` - [array(str)] An array of atomic labels. -- `element` - [array(object(str:str))] The element of each atom type. -- `cell` - [array(array(double))] The lattice vector. Unit in Angstrom. -- `coordinate` - [array(array(double))] The cartesian coordinates of each atom. Unit in Angstrom. -- `mag` - [array(double)] The magnetic moments for each atom. -- `pp` - [object(str-str)] The pseudopotential file of each atom type. -- `orb` - [object(str-str)] The orbital file of each atom type. +Updating geometry data for an existing record replaces its coordinate, magnetic-moment, and cell arrays, together with force and stress arrays when requested; it does not append duplicate rows. SCF iterations are appended to that record's `scf` history, while starting a new calculation/ionic step appends a new `output` record. +## Serialization -## Output +The writer uses four-space indentation and retains object-key insertion order. Non-finite floating-point values (NaN and positive or negative infinity) are serialized as `null`, not as nonstandard JSON numeric tokens. A `null` numeric field can therefore mean either that no value has been written or that the stored value was non-finite; it should not be interpreted as zero. -An array of dicts, including information about each self-consistent field (SCF) step, such as energy, convergence, and configuration: - -- `energy`, `e_fermi` - [double] The total energy and Fermi energy. Unit in eV. -- `force` - [array(array(double))] The forces calculated on each atom. Unit in eV/Angstrom. -- `stress` - [array(array(double))] The stress tensor. Unit in Kbar. -- `cell` - [array(array(double))] The cell parameters. Unit in Angstrom. -- `coordinate` - [array(array(double))] The coordinates of the atoms in the box after the simulation. Unit in Angstrom. -- `total_mag` , `absolute_mag` , `mag` - [double] The total magnetic moment; total absolute magnetic moment; and a list of magnetic moments for each atom, respectively. -- `scf_converge` - [bool] A boolean indicating whether the scf optimization has converged. -- `scf` - [array(object(str:double)] A list of each scf step, each item contains: - - `energy` - [double] The total energy. Unit in eV. - - `ediff` - [double] The energy difference between the current and previous step. Unit in eV. - - `drho` - [double] The charge density difference between the current and previous step. - - `time` - [double] The time used for the current step. Unit in seconds. - -## Final Structure -Parameters regarding the final converged results and the optimized geometry: - -- `energy` - [double] The final energy. -- `label` - [str] An array of atomic labels. -- `cell` - [array(array(double))] The resulting cell parameters. -- `coordinate` - [array(array(double))] The final atomic coordinates. -- `relax_converge` - [bool] A boolean indicating whether the geometry optimization has converged. -- `dos` - [array(array(array(double)))] The state energy, and the dimension is NSPIN\*NKPOINT\*NBAND. -- `dos_weight` - [array(array(array(double)))] The weight of each state, and the dimension is same as `dos`. \ No newline at end of file +JSON numbers are intended to be consumed as numeric values. Their textual representation (for example, decimal versus scientific notation) is not part of the output schema. diff --git a/docs/advanced/md.md b/docs/advanced/md.md index c66243a362e..25c01b1c8a4 100644 --- a/docs/advanced/md.md +++ b/docs/advanced/md.md @@ -30,7 +30,7 @@ Furthermore, ABACUS also provides a [list of keywords](./input_files/input-main. The MD output information will be written into the file `MD_dump`， in which the atomic forces, atomic velocities, and lattice virial are controlled by keyword [dump_force](./input_files/input-main.md#dump_force), [dump_vel](./input_files/input-main.md#dump_vel), and [dump_virial](./input_files/input-main.md#dump_virial), respectively. -[Examples](../../examples/md/lcao_gammaonly_Si8/) of MD simulations are also provided. +[Examples](https://github.com/deepmodeling/abacus-develop/tree/develop/examples/18_md/01_LCAO_NVE_Si8) of MD simulations are also provided. There are eight INPUT files corresponding to eight different MD evolution methods in the directory. For examlpe, `INPUT_0` shows how to employ the NVE simulation. diff --git a/docs/advanced/opt.md b/docs/advanced/opt.md index 05322ce5be3..eddd1a0d181 100644 --- a/docs/advanced/opt.md +++ b/docs/advanced/opt.md @@ -8,7 +8,7 @@ ABACUS provides two CG implementations for variable-cell relaxation, selected by - **CG variant 1** (`relax_method = cg 1`): Follows a nested procedure where fixed-cell structural relaxation is performed first, followed by an update of the cell parameters, and the process is repeated until convergence is achieved. -An example of the variable cell relaxation can be found in our [repository](https://github.com/deepmodeling/abacus-develop/tree/develop/examples/relax/pw_al), which is provided with the reference output file log.ref. When using CG variant 1, each ionic step is labelled in the following manner: +An example of the variable cell relaxation can be found in our [repository](https://github.com/deepmodeling/abacus-develop/tree/develop/examples/17_relax/01_pw_cell_relax_BFGS_Al), which is provided with the reference output file log.ref. When using CG variant 1, each ionic step is labelled in the following manner: ``` ------------------------------------------- RELAX CELL : 3 diff --git a/docs/advanced/output_files/running_scf.log.md b/docs/advanced/output_files/running_scf.log.md index 07d16a4f545..be80ffc44ca 100644 --- a/docs/advanced/output_files/running_scf.log.md +++ b/docs/advanced/output_files/running_scf.log.md @@ -1,6 +1,6 @@ # The running_scf.log file -- [The running\_scf.log file](#the-running_scf.log-file) +- [The running\_scf.log file](#the-running_scflog-file) - [Reading information](#reading-information) - [Reading version information](#reading-version-information) - [Reading general information](#reading-general-information) @@ -1289,8 +1289,8 @@ TIME STATISTICS HSolverLCAO hamiltSolvePsiK 90.50 17 5.32 18.07 ElecStateLCAO psiToRho 132.45 17 7.79 26.45 elecstate cal_dm 2.02 18 0.11 0.40 - psiMulPsiMpi pdgemm 2.01 18 0.11 0.40 - DensityMatrix cal_DMR 0.13 18 0.01 0.03 + cal_dmk_psi pdgemm 2.01 18 0.11 0.40 + DensityMatrix cal_dmr 0.13 18 0.01 0.03 Gint transfer_DMR 2.98 17 0.18 0.60 Gint_interface cal_gint_rho 127.22 17 7.48 25.40 Charge_Mixing get_drho 0.07 17 0.00 0.01 diff --git a/docs/advanced/pp_orb.md b/docs/advanced/pp_orb.md index 4cfe750096b..8cc56839b9e 100644 --- a/docs/advanced/pp_orb.md +++ b/docs/advanced/pp_orb.md @@ -38,7 +38,7 @@ Stable orbital generation programs can be found in guidelines above, there is al For treating BSSE(Basis Set Superposition Error), we allow for the inclusion of "empty" or "ghost" atoms in the calculation. Namely, when expanding the Hamiltonian, basis sets on the atoms are used, while the ionic potentials on those atoms are not included when constructing the Hamiltonian. -An empty atom is defined in the `STRU` file when an element name contains the "empty" suffix, such as "H_empty", "O_empty" and so on. Here we provide an [example](https://github.com/deepmodeling/abacus-develop/tree/develop/examples/bsse/water) of calculating the molecular formation energy of $H_2O$ with BSSE correction. +An empty atom is defined in the `STRU` file when an element name contains the "empty" suffix, such as "H_empty", "O_empty" and so on. Here we provide an [example](https://github.com/deepmodeling/abacus-develop/tree/develop/examples/32_bsse/01_water) of calculating the molecular formation energy of $H_2O$ with BSSE correction. In the example, we provide four STRU files: @@ -76,11 +76,10 @@ When performing spin-orbit coupling (SOC) calculations with `lspinorb=1`, specif Full-relativistic pseudopotentials suitable for SOC calculations can be identified by checking the UPF file header (`PP_HEADER` section): ```xml + ``` diff --git a/docs/advanced/scf/advanced.md b/docs/advanced/scf/advanced.md index 8b8b24f51a4..a01280344bb 100644 --- a/docs/advanced/scf/advanced.md +++ b/docs/advanced/scf/advanced.md @@ -17,7 +17,7 @@ sigma_k 0.6 nc_k 0.00037 ``` -Example of running DFT calculation with the implicit solvation model is provided in this [directory](https://github.com/deepmodeling/abacus-develop/tree/develop/examples/implicit_solvation_model/Pt-slab). +Example of running DFT calculation with the implicit solvation model is provided in this [directory](https://github.com/deepmodeling/abacus-develop/tree/develop/examples/27_imp_sol/01_pw_Pt_slab). ## External Electric Field @@ -34,7 +34,7 @@ efield_pos_dec 0.1 efield_amp 0.001 ``` -Example of running DFT calculation with added external electric field is provided in this [directory](https://github.com/deepmodeling/abacus-develop/tree/develop/examples/electric_field/Pt-slab). +Example of running DFT calculation with added external electric field is provided in this [directory](https://github.com/deepmodeling/abacus-develop/tree/develop/examples/28_efield/01_Pt_slab). ## Dipole Correction @@ -60,7 +60,7 @@ efield_pos_dec 0.1 efield_amp 0.001 ``` -Examples of running DFT calculations with dipole correction are provided in this [directory](https://github.com/deepmodeling/abacus-develop/tree/develop/examples/dipole_correction/Pt-slab). There are two input files, where `INPUT1` considers only the dipole correction without no applied external field, while `INPUT2` considers the dipole correction under an applied external field. +Examples of running DFT calculations with dipole correction are provided in this [directory](https://github.com/deepmodeling/abacus-develop/tree/develop/examples/29_dipole_corr/01_Pt_slab). There are two input files, where `INPUT1` considers only the dipole correction without no applied external field, while `INPUT2` considers the dipole correction under an applied external field. To run any of the two cases, users may enter the directory, copy the corresponding input file to `INPUT`, and run ABACUS. @@ -82,7 +82,7 @@ block_up 0.55 block_height 0.1 ``` -Example of running DFT calculation with the compensating charge is provided in this [directory](https://github.com/deepmodeling/abacus-develop/tree/develop/examples/compensating_charge/Pt-slab). +Example of running DFT calculation with the compensating charge is provided in this [directory](https://github.com/deepmodeling/abacus-develop/tree/develop/examples/31_comp_charge/01_Pt_slab). ## Van-der-Waals Correction Conventional DFT functionals often suffer from an inadequate treatment of long-range dispersion, or Van der Waals (VdW) interactions. In order to describe materials where VdW interactions are prominent, one simple and popular approach is to add a Lennard-Jones type term. The resulting VdW-corrected DFT has been proved to be a very effective method for description of both short-range chemical bonding and long-range dispersive interactions. @@ -99,7 +99,7 @@ To use VdW-correction, users need to supply value to the `vdw_method` keyword in Furthermore, ABACUS also provides a [list of keywords](../input_files/input-main.md#vdw-correction) to control relevant parmeters used in calculating the VdW correction, such as the scale factor (s6) term. Recommended values of such parameters can be found on the [webpage](https://www.chemie.uni-bonn.de/pctc/mulliken-center/software/dft-d3). The default values of the parameters in ABACUS are set to be the recommended values for PBE. -Examples of VdW-corrected DFT calculations are provided in this [directory](https://github.com/deepmodeling/abacus-develop/tree/develop/examples/vdw/si2). There are two input files, where `INPUT1` shows how to apply D2 correction with user-specified $C_6$ parameter, and `INPUT2` shows how to apply D3(BJ) correction with default VdW parameters. +Examples of VdW-corrected DFT calculations are provided in this [directory](https://github.com/deepmodeling/abacus-develop/tree/develop/examples/25_vdw). There are two input files, where `INPUT1` shows how to apply D2 correction with user-specified $C_6$ parameter, and `INPUT2` shows how to apply D3(BJ) correction with default VdW parameters. To run any of the two cases, users may enter the directory, copy the corresponding input file to `INPUT`, and run ABACUS. diff --git a/docs/advanced/scf/construct_H.md b/docs/advanced/scf/construct_H.md index 69a22ad80e9..31e42b859d0 100644 --- a/docs/advanced/scf/construct_H.md +++ b/docs/advanced/scf/construct_H.md @@ -4,9 +4,9 @@ In our package, the XC functional can be set explicitly using the `dft_functional` keyword in `INPUT` file. If `dft_functional` is not specified, ABACUS will use the xc functional indicated in the pseudopotential file. -Several common functionals are implemented in ABACUS, such as PZ and PBE. Users can check out this [file](../../../source/source_hamilt/module_xc/xc_funcs.h) for a complete list of functionals implemented in ABACUS. Furthermore, if ABACUS is compiled with LIBXC, we also support all the LDA, GGA and meta-GGA functionals provided therein. +Several common functionals are implemented in ABACUS, such as PZ and PBE. Users can check out this [file](https://github.com/deepmodeling/abacus-develop/blob/develop/source/source_hamilt/module_xc/xc_funcs.h) for a complete list of functionals implemented in ABACUS. Furthermore, if ABACUS is compiled with LIBXC, we also support all the LDA, GGA and meta-GGA functionals provided therein. -Here, we use a simple [example calculation](https://github.com/deepmodeling/abacus-develop/tree/develop/examples/scf/lcao_Si2) for illustration. +Here, we use a simple [example calculation](https://github.com/deepmodeling/abacus-develop/tree/develop/examples/02_scf/02_lcao_Si2) for illustration. 1. **Default setting:** @@ -69,9 +69,9 @@ Here, we use a simple [example calculation](https://github.com/deepmodeling/abac ABACUS supports functionals with exact Hartree-Fock exchange in LCAO basis set only. The old INPUT parameter exx_hybrid_type for hybrid functionals has been absorbed into `dft_functional`. Options are `hf` (pure Hartree-Fock), `pbe0`(PBE0), `hse`, and `scan0`(SCAN0) (Note: in order to use HSE or SCAN0 functional, LIBXC is required). Note also that only HSE has been tested while other hybrid functionals have NOT been fully tested yet, and the maximum parallel cpus for running exx is N^4, with N being the number of atoms. - More information on the hybrid functional can be found from the section [Exact Exchange](../input_files/input-main.md#exact-exchange) in the list of input variables for more information. + More information on the hybrid functional can be found from the section [Exact Exchange](../input_files/input-main.md#exact-exchange-common) in the list of input variables for more information. - An example HSE calculation is provided in this [directory](https://github.com/deepmodeling/abacus-develop/tree/develop/examples/hse/lcao_Si2). Apart from the input files (`INPUT`, `STRU`, `KPT`), we further provide two files: running_scf.log_ref and log_ref, which contains reference for running_scf.log and standard output from the program, respectively. + An example HSE calculation is provided in this [directory](https://github.com/deepmodeling/abacus-develop/tree/develop/examples/20_hybrid/02_lcao_Si2). Apart from the input files (`INPUT`, `STRU`, `KPT`), we further provide two files: running_scf.log_ref and log_ref, which contains reference for running_scf.log and standard output from the program, respectively. ## DFT+*U* @@ -79,4 +79,4 @@ Conventional functionals, e.g., L(S)DA and GGAs, encounter failures in strongly Now the DFT+*U* method is accessible in ABACUS. The details of the DFT+*U* method could be found in this [paper](https://doi.org/10.1063/5.0090122). It should be noted that the DFT+*U* works only within the NAO scheme, which means that the value of the keyword `basis_type` must be lcao when DFT+*U* is called. To turn on DFT+*U*, users need to set the value of the `dft_plus_u` keyword in the `INPUT` file to be 1. All relevant parmeters used in DFT+*U* calculations are listed in the [DFT+*U* correction](../input_files/input-main.md#dftu-correction) part of the [list of keywords](../input_files/input-main.md). -Examples of DFT+*U* calculations are provided in this [directory](https://github.com/deepmodeling/abacus-develop/tree/develop/examples/dft_plus_u). +Examples of DFT+*U* calculations are provided in this [directory](https://github.com/deepmodeling/abacus-develop/tree/develop/examples/19_dftu). diff --git a/docs/advanced/scf/converge.md b/docs/advanced/scf/converge.md index 4f77f2f3b7c..fca624d9aec 100644 --- a/docs/advanced/scf/converge.md +++ b/docs/advanced/scf/converge.md @@ -16,7 +16,7 @@ For non-spin-polarized calculations, the default choices usually achieve converg For magnetic calculations, `mixing_beta_mag` and `mixing_gg0_mag` are activated. Considering collinear calculations, you can rely on the default value for most cases. If convergence issue arises, you can try to reduce `mixing_beta` and `mixing_beta_mag` together. For non-collinear calculations, tradtional broyden usually works, especially for a given magnetic configuration. If one is not interested in the energies of a given magnetic configuration but wants to determine the ground state by relaxing the magnetic moments’ directions, the standard Broyden mixing algorithm sometimes fails to find the correct magnetic configuration. If so, we can set [mixing_angle=1.0](../input_files/input-main.md#mixing_angle), which is a promising mixing method proposed by J. Phys. Soc. Jpn. 82 (2013) 114706. -An example showcasing different charge mixing methods can be found in our [repository](https://github.com/deepmodeling/abacus-develop/tree/develop/examples/charge_mixing/pw_Al). Four INPUT files are provided, with description given in README. +An example showcasing different charge mixing methods can be found in our [repository](https://github.com/deepmodeling/abacus-develop/tree/develop/examples/07_charge_mixing/01_pw_Al). Four INPUT files are provided, with description given in README. As for DFT+U calculations, where the hamiltonian is not only dependent on charge density, but also dependent on density matrix. You can try `mixing_restart>0` and `mixing_dmr=1` to improve convergence. For case extremely hard to converge, you can use so-called U-Ramping method by setting a finite positive `uramping` with `mixing_restart>0` and `mixing_dmr=1`. @@ -28,4 +28,4 @@ In ABACUS, we provide a few smearing methods, which can be controlled using the > Note : The two keywords `smearing_sigma` and `smearing_sigma_temp` should not be used concurrently. -We provide an example showing the importance of smearing in our [repository](https://github.com/deepmodeling/abacus-develop/tree/develop/examples/smearing/lcao_fe). Two INPUT fiels rae provided, with description given in README. \ No newline at end of file +We provide an example showing the importance of smearing in our [repository](https://github.com/deepmodeling/abacus-develop/tree/develop/examples/06_smearing/01_lcao_Fe). Two INPUT fiels rae provided, with description given in README. \ No newline at end of file diff --git a/docs/advanced/scf/hsolver.md b/docs/advanced/scf/hsolver.md index 0cb02ce9dd9..cc85b4d87be 100644 --- a/docs/advanced/scf/hsolver.md +++ b/docs/advanced/scf/hsolver.md @@ -9,7 +9,7 @@ When "basis_type = pw", `ks_solver` can be `cg`, `bpcg` or `dav`. The default se When "basis_type = lcao", `ks_solver` can be `genelpa` or `scalapack_gvx`. The default setting `genelpa` is recommended, which is based on ELPA (EIGENVALUE SOLVERS FOR PETAFLOP APPLICATIONS) (https://elpa.mpcdf.mpg.de/) and the kernel is auto choosed by GENELPA(https://github.com/pplab/GenELPA), usually faster than the setting of "scalapack_gvx", which is based on ScaLAPACK(Scalable Linear Algebra PACKage) ## Stochasic DFT -We support stochastic DFT calculation (SDFT) or mixed stochastic-deterministic DFT (MDFT) with plane-wave basis [[Phys. Rev. B 106, 125132 (2022)](https://doi.org/10.1103/PhysRevB.106.125132)]. Different from traditional KSDFT with the explicit diagonalization method, SDFT and MDFT calculate physical quantities with trace of the corresponding operators. The advantages of SDFT and MDFT compared to the traditional KSDFT are the ability to simulate larger sizes and higher temperatures. In our package, SDFT and MDFT can be used by setting the `esolver_type` parameter to `sdft` for SCF calculations or MD calculations. To start with, you can refer to two [examples](https://github.com/deepmodeling/abacus-develop/tree/develop/examples/stochastic) and an explanation of the [input variables](../input_files/input-main.md#electronic-structure-sdft). +We support stochastic DFT calculation (SDFT) or mixed stochastic-deterministic DFT (MDFT) with plane-wave basis [[Phys. Rev. B 106, 125132 (2022)](https://doi.org/10.1103/PhysRevB.106.125132)]. Different from traditional KSDFT with the explicit diagonalization method, SDFT and MDFT calculate physical quantities with trace of the corresponding operators. The advantages of SDFT and MDFT compared to the traditional KSDFT are the ability to simulate larger sizes and higher temperatures. In our package, SDFT and MDFT can be used by setting the `esolver_type` parameter to `sdft` for SCF calculations or MD calculations. To start with, you can refer to two [examples](https://github.com/deepmodeling/abacus-develop/tree/develop/examples/23_sdft) and an explanation of the [input variables](../input_files/input-main.md#electronic-structure-sdft). When we have a hamiltonian, the electronic density can be calculated with: diff --git a/docs/advanced/scf/performance.md b/docs/advanced/scf/performance.md index cf3ebb907f3..49e9d38d4dd 100644 --- a/docs/advanced/scf/performance.md +++ b/docs/advanced/scf/performance.md @@ -20,7 +20,7 @@ Inclusion of k-point symmetry helps increasing the efficiency of calculations by For LCAO calculation, the matrix elements of the local potential is evaluated using grid integration. In grid integration, we group real-space FFT grid points into boxes of dimension bx * by * bz, and then proceed with the boxes as the basis unit of calculation. -Setting [bx, by, bz](../input_files/input-main.md#bx-by-bz) to be values other than default might help with the efficiency of grid integration. +Setting [bx](../input_files/input-main.md#bx), [by](../input_files/input-main.md#by), and [bz](../input_files/input-main.md#bz) to values other than default might help with the efficiency of grid integration. > Note: the choice of bx, by, bz should be integer factors of the dimension of the real space FFT grid in each direction. diff --git a/docs/advanced/scf/spin.md b/docs/advanced/scf/spin.md index 0da2ad7cbcc..40f349a43e2 100644 --- a/docs/advanced/scf/spin.md +++ b/docs/advanced/scf/spin.md @@ -49,9 +49,9 @@ Note: different settings for "noncolin" and "lspinorb" correspond to different c **Special case**: `noncolin=0, lspinorb=1` is commonly used for non-magnetic materials with SOC effects (e.g., topological insulators, semiconductors with spin-orbit splitting). In this case, the magnetization is NOT automatically set, implying no magnetic moments in the system. ## For the continuation job -- Continuation job for "nspin 1" need file "SPIN1_CHG.cube" which is generated by setting "out_chg=1" in task before. By setting "init_chg file" in new job's INPUT file, charge density will start from file but not atomic. -- Continuation job for "nspin 2" need files "SPIN1_CHG.cube" and "SPIN2_CHG.cube" which are generated by "out_chg 1" with "nspin 2", and refer to spin-up and spin-down charge densities respectively. It should be note that reading "SPIN1_CHG.cube" only for the continuation target magnetic moment job is not supported now. -- Continuation job for "nspin 4" need files "SPIN%s_CHG.cube", where %s in {1,2,3,4}, which are generated by "out_chg 1" with any variable setting leading to 'nspin'=4, and refer to charge densities in Pauli spin matrixes. It should be note that reading charge density files printing by 'nspin'=2 case is supported, which means only $\sigma_{tot}$ and $\sigma_{z}$ are read. +- Continuation job for "nspin 1" need file "chg.cube" which is generated by setting "out_chg=1" in task before. By setting "init_chg file" in new job's INPUT file, charge density will start from file but not atomic. +- Continuation job for "nspin 2" need files "chgs1.cube" and "chgs2.cube" which are generated by "out_chg 1" with "nspin 2", and refer to spin-up and spin-down charge densities respectively. It should be note that reading "chgs1.cube" only for the continuation target magnetic moment job is not supported now. +- Continuation job for "nspin 4" need files "chgs%s.cube", where %s in {1,2,3,4}, which are generated by "out_chg 1" with any variable setting leading to 'nspin'=4, and refer to charge densities in Pauli spin matrixes. It should be note that reading charge density files printing by 'nspin'=2 case is supported, which means only $\sigma_{tot}$ and $\sigma_{z}$ are read. # SOC Effects ## SOC diff --git a/docs/community/contribution_guide.md b/docs/community/contribution_guide.md index 33fc82e8400..7f37664cde4 100644 --- a/docs/community/contribution_guide.md +++ b/docs/community/contribution_guide.md @@ -11,7 +11,7 @@ We assume you already have a good idea on what to do, otherwise the [issue track It is suggested to [submit new issues](https://github.com/deepmodeling/abacus-develop/issues/new/choose) before coding out changes to involve more discussions and suggestions from development team. Refer to the technical guide in [Contributing to ABACUS](../CONTRIBUTING.md) when needed. - **Follow governance rules.** -Before implementation and review, read [`AGENTS.md`](../../AGENTS.md) and the +Before implementation and review, read [`AGENTS.md`](https://github.com/deepmodeling/abacus-develop/blob/develop/AGENTS.md) and the [ABACUS Agent Governance](../developers_guide/agent_governance.md) guide. New code is reviewed against the shared rules for global dependencies, default parameters, header dependencies, tests, INPUT documentation linkage, LF line diff --git a/docs/community/faq.md b/docs/community/faq.md index c5b4e4c9a75..6896f8961a1 100644 --- a/docs/community/faq.md +++ b/docs/community/faq.md @@ -108,7 +108,7 @@ This is a [known issue](https://github.com/open-mpi/ompi/issues/4948) of OpenMPI **1. How to visualize charge density file?** -The output file SPIN1_CHG.cube can be visualized by using VESTA. +The output file chg.cube can be visualized by using VESTA. **2. How to change cif file directly to STRU file?** diff --git a/docs/conf.py b/docs/conf.py index bf96df8cb7b..999468b3661 100644 --- a/docs/conf.py +++ b/docs/conf.py @@ -18,11 +18,18 @@ # -- Project information ----------------------------------------------------- project = 'ABACUS' -copyright = '2024, ABACUS' +copyright = '%Y, ABACUS' author = 'ABACUS' -# The full version, including alpha/beta/rc tags -# release = '2.3.5' +# The full version, including alpha/beta/rc tags, read from the C++ source so +# it stays in sync with releases. The EPUB3 builder warns when it is empty, +# which fails the Read the Docs build under fail_on_warning. +import re +from pathlib import Path + +_version_h = Path(__file__).resolve().parent.parent / 'source' / 'source_main' / 'version.h' +release = re.search(r'#define\s+VERSION\s+"v?([^"]+)"', _version_h.read_text()).group(1) +version = release # -- General configuration --------------------------------------------------- @@ -33,7 +40,13 @@ extensions = [ 'myst_parser', 'deepmodeling_sphinx', + 'sphinx.ext.intersphinx', ] +intersphinx_mapping = { + "python": ("https://docs.python.org/3" , None), + "numpy": ("https://numpy.org/doc/stable/" , None), + "scipy": ("https://docs.scipy.org/doc/scipy/" , None), +} myst_enable_extensions = [ "amsmath", "colon_fence", @@ -50,6 +63,25 @@ "tasklist", ] myst_heading_anchors = 4 +nitpicky = True +# nitpick_ignore_regex = [ +# (r"py:.*", r"abacus\..*"), +# +# (r"py:class", r"numpy\..*"), +# (r"py:class", r"scipy\..*"), +# (r"py:class", r"ase\..*"), +# (r"py:.*", r"cython\..*"), +# +# (r"cpp:.*", r"ModuleBase::.*"), +# (r"cpp:.*", r"ModuleIO::.*"), +# (r"cpp:identifier", r".*"), +# +# (r"std:ref", r"input-.*"), +# (r"std:doc", r"advanced/input_files/.*"), +# +# (r"myst", r".*"), +# (r"std:term", r".*"), +# ] # Add any paths that contain templates here, relative to this directory. templates_path = ['_templates'] @@ -94,10 +126,23 @@ html_static_path = ['_static'] latex_engine = 'xelatex' +latex_use_xindy = False mathjax_path = 'https://cdnjs.cloudflare.com/ajax/libs/mathjax/3.2.0/es5/tex-mml-chtml.min.js' # deepmodeling_current_site = 'Tutorials' latex_elements = { - 'extraclassoptions':'openany,oneside' + 'extraclassoptions':'openany,oneside', + # LaTeX-side shims for MathJax-dialect macros used in the markdown + # sources. MathJax (HTML rendering) provides these natively, which is + # why the gap only surfaces in the PDF build. + 'preamble': r''' +% Dirac notation: MathJax has \ket/\bra built in; LaTeX needs braket. +\usepackage{braket} +\providecommand{\bra}[1]{\langle #1\rvert} +\providecommand{\ket}[1]{\lvert #1\rangle} +% MathJax alias macros that plain LaTeX lacks: \lt / \gt for < / > +\providecommand{\lt}{<} +\providecommand{\gt}{>} +''', } diff --git a/docs/developers_guide/agent_governance.md b/docs/developers_guide/agent_governance.md index a02af72aef6..f708fcc73f5 100644 --- a/docs/developers_guide/agent_governance.md +++ b/docs/developers_guide/agent_governance.md @@ -54,6 +54,8 @@ focused cleanup. not around isolated statements inside the function body. Use the enclosing function name (or constructor name) as the timer label so the timer scopes the whole unit of work. +- Do not use `#pragma once` in header files; use traditional + `#ifndef`/`#define`/`#endif` include guards instead. AI agents have additional workflow obligations: @@ -72,6 +74,7 @@ decisions. | Rule category | Typical rule | Phase-one status | Default executor | Severity | Default action | Detection scope | Notes | | --- | --- | --- | --- | --- | --- | --- | --- | | Basic text format | LF line endings | phase-one mechanical | hook + CI | medium | block | full changed text file | `.bat` and `.cmd` keep CRLF | +| Header include guard | New `#pragma once` in a header | phase-one mechanical | hook + CI | high | block | added header lines (`.h`/`.hh`/`.hpp`/`.hxx`/`.cuh`) | Use `#ifndef`/`#define`/`#endif` guards; no exception allowed | | Language baseline | C++11 compatibility | build/toolchain | CI | high | block | build/static tooling | Actual compiler/toolchain result wins | | Global dependency budget | Net increase of `GlobalV`/`GlobalC`/`PARAM` references in code diff | phase-one mechanical + AI review | CI + AI review | high | block on net increase, warn on non-increasing added usage | added and removed code lines | Historical untouched usage and documentation mentions are not blocked; migration-neutral moves require reviewer rationale | | New default parameter | Header declaration adds a default argument | phase-one mechanical + AI review | CI + AI review | high | block | header diff | High misuse risk | diff --git a/docs/generate_input_main.py b/docs/generate_input_main.py old mode 100644 new mode 100755 index 62c660d6793..89a74521045 --- a/docs/generate_input_main.py +++ b/docs/generate_input_main.py @@ -56,7 +56,9 @@ "Spin-Constrained DFT", "vdW correction", "Berry phase and wannier90 interface", - "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory", + "Real-Time TDDFT (Common)", + "Real-Time TDDFT (LCAO)", + "Real-Time TDDFT (PW)", "Variables useful for debugging", "Electronic conductivities", "Implicit solvation model", diff --git a/docs/index.rst b/docs/index.rst index dd0b5968c4c..f0d34d3242e 100644 --- a/docs/index.rst +++ b/docs/index.rst @@ -45,6 +45,7 @@ the developments and applications of novel machine-learning-assisted DFT methods advanced/interface/index advanced/input_files/index advanced/windows_installer + advanced/json/index .. toctree:: :maxdepth: 2 diff --git a/docs/parameters.yaml b/docs/parameters.yaml index a0f76236e86..9cf9dd62dee 100644 --- a/docs/parameters.yaml +++ b/docs/parameters.yaml @@ -709,9 +709,24 @@ parameters: * 1: Spin degeneracy * 2: Collinear spin polarized. * 4: Noncollinear or spin-orbit calculations. Set nspin to 4 explicitly when noncolin or lspinorb is enabled. + * Note: With nspin=2 and no initial magnetization in STRU, a moment of 1.0 is autoset for every atom, unless symmetry is 1. + * Note: With nspin=4 no moment is ever autoset. The calculation starts from zero magnetic moment and a warning is printed; set 'mag' explicitly in STRU for the magnetic atoms if a magnetic ground state is expected. default_value: "1" unit: "" availability: "" + - name: gga_grad + category: Electronic structure + type: Integer + description: | + Selects the local spin mapping for LDA/GGA functionals in magnetic nspin=4 calculations. + * 0: preserves the original algorithm (default). + * 1: uses the local magnetization magnitude instead of the global quantization axis in the built-in GGA gradient correction. For LIBXC functionals, 0 and 1 are equivalent. + * 2: uses a C2-regularized magnetization magnitude with eta = 1e-3 in atomic density units. The spin densities are (abs(n + rho_core) +/- min(S_eta(m), abs(n + rho_core)))/2. GGA gradients are the local-map Jacobian applied to the FFT gradients of the four density channels. The potential reverses this same discrete energy graph, including the radial Hessian and density/sigma clipping branches; the GGA stress uses the corresponding metric derivative. + For r = |m| and x = r/eta, S_eta = eta*x^3*(3*x^2 - 8*x + 6) for r < eta, and S_eta = r otherwise. The regularization is part of the functional definition, including its first and second derivatives. + Mode 2 also uses this local map for the LDA contribution. Other spin configurations retain their existing behavior. + default_value: "0" + unit: "" + availability: "" - name: smearing_method category: Electronic structure type: String @@ -983,6 +998,8 @@ parameters: * noncolin=0, lspinorb=1: SOC with z-axis magnetism only (for non-magnetic materials with SOC) * noncolin=1, lspinorb=0: Non-collinear magnetism without SOC * noncolin=1, lspinorb=1: Both non-collinear magnetism and SOC + * Note: When nspin=4 and noncolin=0, only the z component of the initial magnetization in STRU is used; x/y components are ignored and a warning is printed. + * Note: When nspin=4 and no initial magnetization is set in STRU, the calculation starts from zero magnetic moment; no automatic magnetization is assigned. default_value: "False" unit: "" availability: "" @@ -1780,7 +1797,7 @@ parameters: availability: "" - name: cal_syns category: Molecular dynamics - type: "Boolean [Integer](optional)" + type: "Boolean \\[Integer\\](optional)" description: | Whether to calculate and output asynchronous overlap matrix for Hefei-NAMD interface. When enabled, calculates by computing overlap between basis functions at atomic positions from previous time step and current time step. The overlap is calculated by shifting atom positions backward by velocity x md_dt. Output file: OUT.*/syns_nao.csr in CSR format. @@ -1800,6 +1817,22 @@ parameters: default_value: "0.01" unit: bohr availability: "" + - name: plumed + category: Molecular dynamics + type: Boolean + description: | + Whether to use PLUMED (https://www.plumed.org) to compute collective variables, biasing potentials and free-energy methods during an MD run. ABACUS must be compiled with -DENABLE_PLUMED=ON and linked against a PLUMED installation (discovered through pkg-config or PLUMED_ROOT) to enable this feature. The PLUMED interface currently supports a single MPI rank only. + default_value: "False" + unit: "" + availability: "" + - name: plumed_file + category: Molecular dynamics + type: String + description: | + The input file of PLUMED, read when `plumed` is enabled. Relative paths are resolved against the working directory of the run. + default_value: plumed.dat + unit: "" + availability: plumed==true - name: of_kinetic category: "OFDFT: orbital free density functional theory" type: String @@ -2460,8 +2493,107 @@ parameters: default_value: "False" unit: "" availability: "" + - name: td_orthonormal + category: Real-Time TDDFT (PW) + type: String + description: | + Orthonormalize each propagated wavefunction before constructing its density. With the wavefunctions as columns of $\boldsymbol{\Psi}$, define the Gram matrix $\boldsymbol{S}=\boldsymbol{\Psi}^{\dagger}\boldsymbol{\Psi}$ and apply $\boldsymbol{\Psi}\leftarrow\boldsymbol{\Psi}\boldsymbol{C}$. + * `cholesky`: Cholesky orthonormalization. Factor $\boldsymbol{S}=\boldsymbol{R}^{\dagger}\boldsymbol{R}$ with upper-triangular $\boldsymbol{R}$ and use $\boldsymbol{C}=\boldsymbol{R}^{-1}$. + * `lowdin`: Löwdin symmetric orthonormalization. Compute $\boldsymbol{C}=\boldsymbol{S}^{-1/2}$ by eigendecomposition. + * `newton_schulz`: Newton-Schulz iteration for a Gram matrix close to the identity. Starting from $\boldsymbol{C}_0=\boldsymbol{I}$, approximate $\boldsymbol{S}^{-1/2}$ using $\boldsymbol{C}_{j+1}=\boldsymbol{C}_j(3\boldsymbol{I}-\boldsymbol{S}\boldsymbol{C}_j^2)/2$. + * `none`: Disable orthonormalization. + + [NOTE] Orthonormalization is applied after the Crank-Nicolson linear solve and uses an orthogonality tolerance independent of `lin_thr`. With orthonormalization enabled, the propagated Gram matrix must differ from the identity by at most `1e-6` (single precision) or `1e-12` (double precision) per element; unsuccessful corrections stop the calculation. Wavefunctions within this tolerance are left unchanged. Nonfinite or nonpositive orbital norms are rejected even with `none`. + default_value: cholesky + unit: "" + availability: basis_type==pw and esolver_type==tddft + - name: td_out_stat + category: Real-Time TDDFT (PW) + type: Boolean + description: | + Write wavefunction and density electron counts, their changes from the initial state, and orthogonality errors to the running log after each electronic evolution step. Electron counts describe the end-of-step state; orthogonality errors before and after correction are separate maxima over all SCF iterations and k points within that step. Orthogonality errors are omitted when `td_orthonormal=none`. + + [NOTE] Enabling this output adds density integration and diagnostic communication. + default_value: "false" + unit: "" + availability: basis_type==pw and esolver_type==tddft + - name: lin_solver + category: Real-Time TDDFT (PW) + type: String + description: | + Iterative linear solver used for PW real-time propagation. + * `bicgstab`: Biconjugate gradient stabilized (BiCGSTAB) method. + * `cgs`: Conjugate gradient squared (CGS) method. + * `gmres`: Restarted generalized minimal residual (GMRES) method, controlled by `lin_gmres_restart`. + + The initial ground-state diagonalization is controlled by `ks_solver`. + default_value: gmres + unit: "" + availability: basis_type==pw and esolver_type==tddft + - name: lin_precond + category: Real-Time TDDFT (PW) + type: String + description: | + Right preconditioner used by PW real-time linear solvers. The Crank-Nicolson operator is $\boldsymbol{L}=\boldsymbol{I}+\mathrm{i}\Delta t\,\boldsymbol{H}/2$, with the propagation Hamiltonian $\boldsymbol{H}$ in Hartree and time step $\Delta t$ in atomic units. Below, $\boldsymbol{M}^{-1}$ denotes the inverse preconditioner and $\boldsymbol{D}$ the kinetic diagonal inverse. + * `kinetic`: Apply $\boldsymbol{M}^{-1}=\boldsymbol{D}$, with diagonal entries $D_{\boldsymbol{G}\boldsymbol{G}}=(1+\mathrm{i}\Delta t\,T_{\boldsymbol{G}}/2)^{-1}$, where $T_{\boldsymbol{G}}=\lVert\boldsymbol{k}+\boldsymbol{G}+\boldsymbol{A}_{\mathrm{mid}}\rVert^2/2$ is the kinetic energy in Hartree. In the velocity gauge, $\boldsymbol{A}_{\mathrm{mid}}=(\boldsymbol{A}_n+\boldsymbol{A}_{n+1})/2$ is the propagation vector potential in Hartree atomic units; in the length gauge, set $\boldsymbol{A}_{\mathrm{mid}}=\boldsymbol{0}$. + * `kinetic_recycle`: Apply $\boldsymbol{M}^{-1}=\boldsymbol{D}+(\boldsymbol{Z}_{\mathrm{hist}}-\boldsymbol{D}\boldsymbol{W}_{\mathrm{hist}})\boldsymbol{W}_{\mathrm{hist}}^{\dagger}$, where $\boldsymbol{Z}_{\mathrm{hist}}$ and $\boldsymbol{W}_{\mathrm{hist}}$ are paired response directions and approximate operator images from the preceding successful solve at the same k point, with $\boldsymbol{W}_{\mathrm{hist}}^{\dagger}\boldsymbol{W}_{\mathrm{hist}}\approx\boldsymbol{I}$. The images belong to the historical solve, not necessarily the current $\boldsymbol{L}$. Falls back to `kinetic` when no reliable history is available. + * `kinetic_subspace`: Apply $\boldsymbol{M}^{-1}=\boldsymbol{D}+(\boldsymbol{U}-\boldsymbol{D}\boldsymbol{L}\boldsymbol{U})(\boldsymbol{U}^{\dagger}\boldsymbol{L}\boldsymbol{U})^{-1}\boldsymbol{U}^{\dagger}$, where the columns of $\boldsymbol{U}$ are the previous-time wavefunctions defining the coarse subspace. + * `none`: Disable preconditioning. + + Preconditioning changes the convergence rate, while `lin_thr` still controls the residual of the original equation. + default_value: kinetic_recycle + unit: "" + availability: basis_type==pw and esolver_type==tddft + - name: lin_thr + category: Real-Time TDDFT (PW) + type: Real + description: | + Nonnegative finite residual tolerance for each band in a PW real-time Crank-Nicolson solve $\boldsymbol{L}\boldsymbol{x}=\boldsymbol{b}$, with residual $\boldsymbol{r}=\boldsymbol{b}-\boldsymbol{L}\boldsymbol{x}$. A value of 0 selects $\tau=\max(10^{-10},100\epsilon)$, where $\epsilon$ is machine epsilon for the wavefunction precision. This gives a tolerance of approximately $1.19209\times10^{-5}$ in single precision and $10^{-10}$ in double precision. A positive value specifies $\tau$ directly. + * `bicgstab` and `gmres`: Require $\lVert\boldsymbol{r}\rVert\leqslant\tau\max(1,\lVert\boldsymbol{b}\rVert)$. + * `cgs`: Require $\lVert\boldsymbol{r}\rVert\leqslant\tau\lVert\boldsymbol{b}\rVert$ for nonzero $\boldsymbol{b}$, or $\lVert\boldsymbol{r}\rVert\leqslant\tau$ for zero $\boldsymbol{b}$. + + All methods check the residual at the end of the linear solve, before orthonormalization. GMRES can use explicit residual reconstruction with periodic independent checks when `lin_reconstruct` is enabled. Subsequent orthonormalization uses a separate orthogonality tolerance and does not guarantee that the corrected wavefunctions satisfy the same linear residual tolerance. + default_value: "0" + unit: "" + availability: basis_type==pw and esolver_type==tddft + - name: lin_maxiter + category: Real-Time TDDFT (PW) + type: Integer + description: | + Positive maximum number of iterations for each PW real-time linear solve. Failure to converge stops the calculation. The number of self-consistency iterations is controlled separately by `scf_nmax`. + default_value: "500" + unit: "" + availability: basis_type==pw and esolver_type==tddft + - name: lin_gmres_restart + category: Real-Time TDDFT (PW) + type: Integer + description: | + Positive maximum Arnoldi dimension per GMRES cycle. `lin_maxiter` limits the total iterations across all cycles. + default_value: "20" + unit: "" + availability: basis_type==pw and esolver_type==tddft and lin_solver==gmres + - name: td_cn_init + category: Real-Time TDDFT (PW) + type: Boolean + description: | + Project the Crank-Nicolson equation $\boldsymbol{L}\boldsymbol{x}=\boldsymbol{b}$ onto the previous-time wavefunction subspace to initialize the first propagation solve of each time step. With those wavefunctions as the columns of $\boldsymbol{U}$, the projected initial guess is $\boldsymbol{x}_0=\boldsymbol{U}(\boldsymbol{U}^{\dagger}\boldsymbol{L}\boldsymbol{U})^{-1}\boldsymbol{U}^{\dagger}\boldsymbol{b}$. Evaluate and reuse the initial residual $\boldsymbol{r}_0=\boldsymbol{b}-\boldsymbol{L}\boldsymbol{x}_0$ using the stored subspace operator images $\boldsymbol{L}\boldsymbol{U}$, without an additional Hamiltonian application. Available with all linear solvers and preconditioners; later self-consistency iterations retain their current wavefunction guess. + default_value: "true" + unit: "" + availability: basis_type==pw and esolver_type==tddft + - name: lin_reconstruct + category: Real-Time TDDFT (PW) + type: Boolean + description: | + Reconstruct the GMRES residual for $\boldsymbol{L}\boldsymbol{x}=\boldsymbol{b}$ as $\boldsymbol{r}=\boldsymbol{r}_0-(\boldsymbol{L}\boldsymbol{Z})\boldsymbol{y}$ for the update $\boldsymbol{x}=\boldsymbol{x}_0+\boldsymbol{Z}\boldsymbol{y}$, where $\boldsymbol{r}_0=\boldsymbol{b}-\boldsymbol{L}\boldsymbol{x}_0$ is the initial residual, $\boldsymbol{Z}$ contains the current preconditioned search directions, and $\boldsymbol{y}$ contains their update coefficients. Reusing the stored, unmodified operator images $\boldsymbol{L}\boldsymbol{Z}$ reduces Hamiltonian applications. + + Use an internal tolerance of 0.8 times the effective `lin_thr` and independently verify the first solve at each k point and every 16 solves thereafter, with additional independent checks when needed. Failed reconstruction checks trigger a true-residual restart within `lin_maxiter`. + + Only effective for `lin_solver=gmres`; ignored otherwise. + default_value: "true" + unit: "" + availability: basis_type==pw and esolver_type==tddft and lin_solver==gmres - name: estep_per_md - category: "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory" + category: Real-Time TDDFT (Common) type: Integer description: | The number of electronic propagation steps between two ionic steps. @@ -2469,7 +2601,7 @@ parameters: unit: "" availability: "" - name: td_dt - category: "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory" + category: Real-Time TDDFT (Common) type: Real description: | The time step used for electronic propagation. If td_dt is not specified, it is set to md_dt / estep_per_md. If td_dt is specified explicitly, md_dt is reset to td_dt * estep_per_md. @@ -2477,7 +2609,7 @@ parameters: unit: fs availability: "" - name: td_edm - category: "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory" + category: Real-Time TDDFT (LCAO) type: Integer description: | Method used to calculate the energy-density matrix for the overlap contribution to forces in LCAO RT-TDDFT. @@ -2485,9 +2617,9 @@ parameters: * 1: Use the ground-state eigenvalue-weighted expression $\mathrm{EDM}_{\mu\nu,\boldsymbol{k}}=\sum_i w_{i\boldsymbol{k}}\epsilon_{i\boldsymbol{k}}C_{\mu i,\boldsymbol{k}}C_{\nu i,\boldsymbol{k}}^*$. This expression is deprecated for RT-TDDFT and is generally not valid when the propagated wave functions are not Hamiltonian eigenstates. default_value: "0" unit: "" - availability: "" + availability: basis_type==lcao and esolver_type==tddft - name: td_print_eij - category: "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory" + category: Real-Time TDDFT (LCAO) type: Real description: | Controls output of the propagated-state Hamiltonian matrix elements $E_{ij}=\Braket{\psi_i | \hat{H} | \psi_j}$ to the running log. The printed band indices $i$ and $j$ are one-based global indices. Both the threshold and the printed matrix elements are in Ry. @@ -2495,9 +2627,9 @@ parameters: * $\geqslant 0$: Print an element when either $\left|\operatorname{Re}E_{ij}\right|$ or $\left|\operatorname{Im}E_{ij}\right|$ is greater than or equal to td_print_eij. default_value: "-1" unit: Ry - availability: "" + availability: basis_type==lcao and esolver_type==tddft - name: td_propagator - category: "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory" + category: Real-Time TDDFT (LCAO) type: Integer description: | Method used to propagate the electronic states in a nonorthogonal LCAO basis. The formulas below use Hartree atomic units, with $S$, $H$, and $\Delta t=\mathtt{td\_dt}$ evaluated as required by each approximation. @@ -2509,9 +2641,9 @@ parameters: [NOTE] GPU execution currently supports only method 0 in both single-GPU and multi-GPU solver configurations. CPU execution supports methods 0 through 3. default_value: "0" unit: "" - availability: "" + availability: basis_type==lcao and esolver_type==tddft - name: td_vext - category: "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory" + category: Real-Time TDDFT (Common) type: Boolean description: | Controls whether a time-dependent external electric field is applied. @@ -2521,7 +2653,7 @@ parameters: unit: "" availability: "" - name: td_vext_dire - category: "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory" + category: Real-Time TDDFT (Common) type: Vector of Integer description: | Specifies one absolute Cartesian direction for each external electric field when td_vext is enabled. Unlike the ground-state efield_dir parameter, these directions are not defined by lattice or reciprocal-lattice vectors. The number of values must equal that of td_ttype, and repeated directions are allowed; fields assigned to the same direction are added. For example, td_vext_dire 1 2 applies one field along Cartesian x and one along Cartesian y. @@ -2532,18 +2664,18 @@ parameters: unit: "" availability: "" - name: td_stype - category: "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory" + category: Real-Time TDDFT (Common) type: Integer description: | Type of electric field in the space domain, i.e. the gauge of the electric field. - * 0: Length gauge. - * 1: Velocity gauge. - * 2: Hybrid gauge. See J. Chem. Theory Comput. 2025, 21, 3335-3341 for more information. + * 0: Length gauge, available for PW and LCAO. + * 1: Velocity gauge, available for PW and LCAO. + * 2: Hybrid gauge, available only for LCAO. See J. Chem. Theory Comput. 2025, 21, 3335-3341 for more information. default_value: "0" unit: "" availability: "" - name: td_ttype - category: "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory" + category: Real-Time TDDFT (Common) type: Vector of Integer description: | Specifies one time-domain type for each external electric field. Its number of values must equal that of td_vext_dire. Parameters belonging to each type must provide exactly one value for every occurrence of that type, in occurrence order; fields with a repeated direction are added. @@ -2560,7 +2692,7 @@ parameters: unit: "" availability: "" - name: td_tstart - category: "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory" + category: Real-Time TDDFT (Common) type: Integer description: | First electronic step at which the time-dependent electric field is active. The interval from td_tstart through td_tend includes both endpoints. On each active step $n$, the velocity and hybrid gauges integrate the field over $[n\Delta t,(n+1)\Delta t]$, where $\Delta t=\mathtt{td\_dt}$. @@ -2568,7 +2700,7 @@ parameters: unit: "" availability: "" - name: td_tend - category: "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory" + category: Real-Time TDDFT (Common) type: Integer description: | Last electronic step at which the time-dependent electric field is active. The interval from td_tstart through td_tend includes both endpoints. On each active step $n$, the velocity and hybrid gauges integrate the field over $[n\Delta t,(n+1)\Delta t]$, where $\Delta t=\mathtt{td\_dt}$. @@ -2576,7 +2708,7 @@ parameters: unit: "" availability: "" - name: td_lcut1 - category: "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory" + category: Real-Time TDDFT (Common) type: Real description: | Lower fractional-coordinate cutoff for the periodic spatial modulation used in the length gauge. Let $c_1=\mathtt{td\_lcut1}$, $c_2=\mathtt{td\_lcut2}$, $D=c_2-c_1$, and $G=c_1+1-c_2$. For a fractional coordinate $x$, the field factor is $\eta(x)=1$ when $c_1\leqslant x\lt c_2$ and $\eta(x)=-D/G$ elsewhere. The reversed outer interval makes the potential periodic and continuous and gives the field zero cell average. @@ -2584,7 +2716,7 @@ parameters: unit: "" availability: "" - name: td_lcut2 - category: "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory" + category: Real-Time TDDFT (Common) type: Real description: | Upper fractional-coordinate cutoff for the periodic spatial modulation used in the length gauge. Let $c_1=\mathtt{td\_lcut1}$, $c_2=\mathtt{td\_lcut2}$, $D=c_2-c_1$, and $G=c_1+1-c_2$. For a fractional coordinate $x$, the field factor is $\eta(x)=1$ when $c_1\leqslant x\lt c_2$ and $\eta(x)=-D/G$ elsewhere. The reversed outer interval makes the potential periodic and continuous and gives the field zero cell average. @@ -2592,7 +2724,7 @@ parameters: unit: "" availability: "" - name: td_gauss_freq - category: "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory" + category: Real-Time TDDFT (Common) type: Vector of Real description: | Ordinary frequency $f$ in the Gaussian-pulse formula, with $\omega=2\pi f$. Supply exactly one value for each td_ttype 0 occurrence, in occurrence order. @@ -2600,7 +2732,7 @@ parameters: unit: 1/fs availability: td_ttype contains 0 - name: td_gauss_phase - category: "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory" + category: Real-Time TDDFT (Common) type: Vector of Real description: | Carrier phase $\varphi$ in the Gaussian-pulse formula. Supply exactly one value for each td_ttype 0 occurrence, in occurrence order. @@ -2608,7 +2740,7 @@ parameters: unit: rad availability: td_ttype contains 0 - name: td_gauss_sigma - category: "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory" + category: Real-Time TDDFT (Common) type: Vector of Real description: | Nonzero standard deviation $\sigma$ of the Gaussian envelope. Supply exactly one value for each td_ttype 0 occurrence, in occurrence order. @@ -2616,7 +2748,7 @@ parameters: unit: fs availability: td_ttype contains 0 - name: td_gauss_t0 - category: "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory" + category: Real-Time TDDFT (Common) type: Vector of Real description: | Electronic-step position of the Gaussian center, which defines $t_0=\mathtt{td\_gauss\_t0}\Delta t$. Supply exactly one value for each td_ttype 0 occurrence, in occurrence order. @@ -2624,7 +2756,7 @@ parameters: unit: "" availability: td_ttype contains 0 - name: td_gauss_amp - category: "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory" + category: Real-Time TDDFT (Common) type: Vector of Real description: | Electric-field scale $E_0$ in the Gaussian-pulse formula. Supply exactly one value for each td_ttype 0 occurrence, in occurrence order. @@ -2632,7 +2764,7 @@ parameters: unit: V/Angstrom availability: td_ttype contains 0 - name: td_trape_freq - category: "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory" + category: Real-Time TDDFT (Common) type: Vector of Real description: | Ordinary carrier frequency $f$ in the trapezoid-pulse formula, with $\omega=2\pi f$. Supply exactly one value for each td_ttype 1 occurrence, in occurrence order. @@ -2640,7 +2772,7 @@ parameters: unit: 1/fs availability: td_ttype contains 1 - name: td_trape_phase - category: "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory" + category: Real-Time TDDFT (Common) type: Vector of Real description: | Carrier phase $\varphi$ in the trapezoid-pulse formula. Supply exactly one value for each td_ttype 1 occurrence, in occurrence order. @@ -2648,31 +2780,31 @@ parameters: unit: rad availability: td_ttype contains 1 - name: td_trape_t1 - category: "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory" + category: Real-Time TDDFT (Common) type: Vector of Real description: | - Electronic step defining the end of the linear rise, $t_1=\mathtt{td\_trape\_t1}\Delta t$. Each field must satisfy td_trape_t1 <= td_trape_t2 <= td_trape_t3. Supply exactly one value for each td_ttype 1 occurrence, in occurrence order. + Electronic step defining the end of the linear rise, $t_1=\mathtt{td\_trape\_t1}\Delta t$. The value must be a nonnegative integer; integer-valued real inputs such as 2.0 are accepted. Each field must satisfy td_trape_t1 <= td_trape_t2 <= td_trape_t3. Supply exactly one value for each td_ttype 1 occurrence, in occurrence order. default_value: "1875" unit: "" availability: td_ttype contains 1 - name: td_trape_t2 - category: "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory" + category: Real-Time TDDFT (Common) type: Vector of Real description: | - Electronic step defining the end of the plateau, $t_2=\mathtt{td\_trape\_t2}\Delta t$. Each field must satisfy td_trape_t1 <= td_trape_t2 <= td_trape_t3. Supply exactly one value for each td_ttype 1 occurrence, in occurrence order. + Electronic step defining the end of the plateau, $t_2=\mathtt{td\_trape\_t2}\Delta t$. The value must be a nonnegative integer; integer-valued real inputs such as 2.0 are accepted. Each field must satisfy td_trape_t1 <= td_trape_t2 <= td_trape_t3. Supply exactly one value for each td_ttype 1 occurrence, in occurrence order. default_value: "5625" unit: "" availability: td_ttype contains 1 - name: td_trape_t3 - category: "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory" + category: Real-Time TDDFT (Common) type: Vector of Real description: | - Electronic step defining the end of the linear fall, $t_3=\mathtt{td\_trape\_t3}\Delta t$. Each field must satisfy td_trape_t1 <= td_trape_t2 <= td_trape_t3. Supply exactly one value for each td_ttype 1 occurrence, in occurrence order. + Electronic step defining the end of the linear fall, $t_3=\mathtt{td\_trape\_t3}\Delta t$. The value must be a nonnegative integer; integer-valued real inputs such as 2.0 are accepted. Each field must satisfy td_trape_t1 <= td_trape_t2 <= td_trape_t3. Supply exactly one value for each td_ttype 1 occurrence, in occurrence order. default_value: "7500" unit: "" availability: td_ttype contains 1 - name: td_trape_amp - category: "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory" + category: Real-Time TDDFT (Common) type: Vector of Real description: | Electric-field scale $E_0$ in the trapezoid-pulse formula. Supply exactly one value for each td_ttype 1 occurrence, in occurrence order. @@ -2680,7 +2812,7 @@ parameters: unit: V/Angstrom availability: td_ttype contains 1 - name: td_trigo_freq1 - category: "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory" + category: Real-Time TDDFT (Common) type: Vector of Real description: | First ordinary frequency $f_1$ in the trigonometric-pulse formula, with $\omega_1=2\pi f_1$. Supply exactly one value for each td_ttype 2 occurrence, in occurrence order. @@ -2688,7 +2820,7 @@ parameters: unit: 1/fs availability: td_ttype contains 2 - name: td_trigo_freq2 - category: "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory" + category: Real-Time TDDFT (Common) type: Vector of Real description: | Second ordinary frequency $f_2$ in the trigonometric-pulse formula, with $\omega_2=2\pi f_2$. Supply exactly one value for each td_ttype 2 occurrence, in occurrence order. @@ -2696,7 +2828,7 @@ parameters: unit: 1/fs availability: td_ttype contains 2 - name: td_trigo_phase1 - category: "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory" + category: Real-Time TDDFT (Common) type: Vector of Real description: | Carrier phase $\varphi_1$ in the cosine factor of the trigonometric-pulse formula. Supply exactly one value for each td_ttype 2 occurrence, in occurrence order. @@ -2704,7 +2836,7 @@ parameters: unit: rad availability: td_ttype contains 2 - name: td_trigo_phase2 - category: "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory" + category: Real-Time TDDFT (Common) type: Vector of Real description: | Envelope phase $\varphi_2$ in the sine-squared factor of the trigonometric-pulse formula. Supply exactly one value for each td_ttype 2 occurrence, in occurrence order. @@ -2712,7 +2844,7 @@ parameters: unit: rad availability: td_ttype contains 2 - name: td_trigo_amp - category: "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory" + category: Real-Time TDDFT (Common) type: Vector of Real description: | Electric-field scale $E_0$ in the trigonometric-pulse formula. Supply exactly one value for each td_ttype 2 occurrence, in occurrence order. @@ -2720,7 +2852,7 @@ parameters: unit: V/Angstrom availability: td_ttype contains 2 - name: td_heavi_t0 - category: "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory" + category: Real-Time TDDFT (Common) type: Vector of Real description: | Electronic switch step $n_0$ in the Heaviside-pulse definition. The field is $E_0$ for $n\lt n_0$ and zero for $n\geqslant n_0$. Supply exactly one value for each td_ttype 3 occurrence, in occurrence order. @@ -2728,7 +2860,7 @@ parameters: unit: "" availability: td_ttype contains 3 - name: td_heavi_amp - category: "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory" + category: Real-Time TDDFT (Common) type: Vector of Real description: | Electric-field scale $E_0$ in the Heaviside-pulse definition. Supply exactly one value for each td_ttype 3 occurrence, in occurrence order. @@ -2736,7 +2868,7 @@ parameters: unit: V/Angstrom availability: td_ttype contains 3 - name: td_supsine_amp - category: "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory" + category: Real-Time TDDFT (Common) type: Vector of Real description: | Carrier electric-field scale $E_0$ of each supersine pulse. This is not a normalization of the complete waveform maximum, because the envelope-derivative term also contributes. Supply exactly one value for each td_ttype 4 occurrence, in occurrence order. @@ -2744,7 +2876,7 @@ parameters: unit: V/Angstrom availability: td_ttype contains 4 - name: td_supsine_freq - category: "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory" + category: Real-Time TDDFT (Common) type: Vector of Real description: | Nonzero ordinary carrier frequency $f$ of each supersine pulse, with $\omega=2\pi f$. Supply exactly one value for each td_ttype 4 occurrence, in occurrence order. @@ -2752,7 +2884,7 @@ parameters: unit: 1/fs availability: td_ttype contains 4 - name: td_supsine_phase - category: "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory" + category: Real-Time TDDFT (Common) type: Vector of Real description: | Electric-field carrier phase $\varphi$ at the center of each supersine envelope. A value of 0 places a cosine carrier maximum at the envelope center. Supply exactly one value for each td_ttype 4 occurrence, in occurrence order. @@ -2760,7 +2892,7 @@ parameters: unit: rad availability: td_ttype contains 4 - name: td_supsine_sigma - category: "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory" + category: Real-Time TDDFT (Common) type: Vector of Real description: | Dimensionless shape parameter $\sigma$ of each supersine envelope. It must satisfy $0\lt\sigma\lt\pi/2$ so that the electric field approaches zero at the pulse boundaries. Supply exactly one value for each td_ttype 4 occurrence, in occurrence order. @@ -2768,7 +2900,7 @@ parameters: unit: "" availability: td_ttype contains 4 - name: td_supsine_tstart - category: "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory" + category: Real-Time TDDFT (Common) type: Vector of String description: | Integer electronic step at the left, exactly zero boundary of each supersine pulse, defining $t_{\mathrm{s}}=\mathtt{td\_supsine\_tstart}\Delta t$. Supply exactly one integer or default token for each td_ttype 4 occurrence, in occurrence order; each default token inherits td_tstart. The complete pulse support must lie inside the inclusive global td_tstart to td_tend interval; hard truncation of a supersine pulse is rejected. @@ -2776,7 +2908,7 @@ parameters: unit: "" availability: td_ttype contains 4 - name: td_supsine_tend - category: "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory" + category: Real-Time TDDFT (Common) type: Vector of String description: | Integer electronic step at the right, exactly zero boundary of each supersine pulse, defining $t_{\mathrm{e}}=\mathtt{td\_supsine\_tend}\Delta t$. Supply exactly one integer or default token for each td_ttype 4 occurrence, in occurrence order; each default token inherits td_tend. The complete pulse support must lie inside the inclusive global td_tstart to td_tend interval; hard truncation of a supersine pulse is rejected. @@ -2784,30 +2916,30 @@ parameters: unit: "" availability: td_ttype contains 4 - name: init_vecpot_file - category: "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory" + category: Real-Time TDDFT (LCAO) type: Boolean description: | Selects the source of the Cartesian vector potential used by LCAO RT-TDDFT. - * True: Read vector_pot.txt from the calculation working directory. Each non-comment line must contain four columns: a conventionally one-based electronic-step label followed by $A_x$, $A_y$, and $A_z$ in atomic units. Rows are consumed sequentially; the first column is read as a label and is not used for lookup. If propagation continues beyond the available rows, the last row is reused. + * True: Read vector_pot.txt from the calculation working directory. Each non-comment line must contain four columns: a conventionally one-based electronic-step label followed by $A_x$, $A_y$, and $A_z$ in Hartree atomic units. These are propagation values, not endpoint samples. Rows are consumed sequentially; the first column is read as a label and is not used for lookup. If propagation continues beyond the available rows, the last row is reused. * False: Obtain the vector potential by integrating the configured electric field. default_value: "False" unit: "" - availability: "" + availability: basis_type==lcao and esolver_type==tddft - name: ocp - category: "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory" + category: Real-Time TDDFT (Common) type: Boolean description: | - Controls fixed band occupations. In calculations other than LCAO RT-TDDFT, fixed values are applied during electronic-state setup. In LCAO RT-TDDFT, the initial ground-state SCF determines occupations normally, and fixed values from ocp_set are applied during the subsequent real-time propagation steps. - * True: Use the fixed occupations specified by ocp_set during propagation. - * False: Keep the occupations determined by the initial SCF. + Controls fixed band occupations. In PW RT-TDDFT and calculations other than LCAO RT-TDDFT, fixed values from ocp_set are applied during electronic-state setup. PW real-time propagation preserves these initial occupations. In LCAO RT-TDDFT, the initial ground-state SCF determines occupations normally, and fixed values are applied during the subsequent real-time propagation steps. + * True: Use the fixed occupations specified by ocp_set at the stage described above. + * False: Determine occupations normally; real-time propagation preserves the initial SCF occupations. default_value: "False" unit: "" availability: "" - name: ocp_set - category: "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory" + category: Real-Time TDDFT (Common) type: String description: | - Fixed occupation weights used when ocp is true. Values are assigned in band order for each k-point, following k-point order. In LCAO RT-TDDFT, the initial ground-state SCF uses its normally determined occupations, and this array is applied only during subsequent real-time propagation steps. The repetition syntax N*x expands to N copies of x. + Fixed occupation weights used when ocp is true. Values are assigned in band order for each k-point, following k-point order. In PW RT-TDDFT and other calculations outside LCAO RT-TDDFT, this array is applied during electronic-state setup. PW propagation preserves these occupations. In LCAO RT-TDDFT, the initial ground-state SCF uses its normally determined occupations, and this array is applied only during subsequent real-time propagation steps. The repetition syntax N*x expands to N copies of x. * Example: 1 10*1 0 1 expands to 13 values, with the 12th value equal to 0 and all other values equal to 1. * After expansion, provide one block of nbands values for each k-point. If nspin is 2, provide all k-point blocks for spin up followed by all k-point blocks for spin down; otherwise, provide one block per k-point. * The sum of all weights must equal nelec; otherwise the calculation terminates with an error. @@ -3140,7 +3272,7 @@ parameters: unit: "" availability: "" - name: out_freq_td - category: Output information + category: Real-Time TDDFT (Common) type: Integer description: | Controls the output interval in completed electronic evolution steps during RT-TDDFT calculations. When set to a positive integer n, detailed information (see out_freq_ion) is printed every n electron time-evolution steps (i.e., every STEP OF ELECTRON EVOLVE). For example, if you wish to output information once per ionic step, you should set out_freq_td equal to estep_per_md, since one ionic step corresponds to estep_per_md electronic evolution steps. @@ -3300,13 +3432,12 @@ parameters: category: Output information type: Integer description: | - Whether to output the density of states (DOS). For more information, refer to the dos.md. + Whether to output the density of states (DOS) and projected density of states (PDOS). For more information, refer to the dos.md. * 0: no output - * 1: output the density of states (DOS) - * nspin=1 or 4: doss1g{geom}_{basis}.txt, where geom is the geometry index when cell changes or ions move while basis is either pw or nao. - * nspin=2: doss1g{geom}_{basis}.txt and doss2g{geom}_{basis}.txt for two spin channles. - * 2: (LCAO) output the density of states (DOS) and the projected density of states (PDOS) - * 3: output the Fermi surface file (fermi.bxsf) in BXSF format that can be visualized by XCrySDen + * 1: output the density of states (DOS) and projected density of states (PDOS, LCAO only) + * nspin=1 or 4: doss1g{geom}_{basis}.txt and pdoss1g{geom}_{basis}.txt, where geom is the geometry index when cell changes or ions move while basis is either pw or nao. + * nspin=2: doss1/doss2 and pdoss1/pdoss2 files for two spin channels. + Note: values 2 and 3 are no longer supported. Setting out_dos to 2 or 3 will raise an error. default_value: "0" unit: "" availability: "" @@ -3345,7 +3476,7 @@ parameters: category: Output information type: Integer description: | - Controls the output of structure files per ionic step in geometry relaxation calculations. The files are written to the OUT.{suffix}/ directory. Each file corresponds to the structure at RELAX STEP ${istep}, i.e., the structure for which that step's energy was computed (before the relax move), and includes a header comment with the ABACUS version, timestamp, energy, and stress tensor. When out_freq_ion is positive, the numbered files STRU{istep} (or STRU{istep}.cif) are written every out_freq_ion steps; when out_freq_ion is 0, no numbered files are output. + Controls the output of structure files per ionic step. The files are written to the OUT.{suffix}/ directory. Each file corresponds to the structure at RELAX STEP ${istep} (for scf/nscf this is the single step), i.e., the structure for which that step's energy was computed (before the relax move), and includes a header comment with the ABACUS version, timestamp, energy, and stress tensor. When out_freq_ion is positive, the numbered files STRU{istep} (or STRU{istep}.cif) are written every out_freq_ion steps during geometry relaxation; when out_freq_ion is 0, no numbered files are output. This parameter is effective for scf/nscf/relax/cell-relax; for scf/nscf only STRU_FINAL (or STRU_FINAL.cif) is written, and structure output is disabled by default unless out_stru is set explicitly. Molecular dynamics structure output is instead controlled by md_restartfreq (STRU_MD_*). - 0: No structure files are output. - 1: ABACUS STRU format files are output. The latest structure is written to STRU_NOW (overwritten each step), the numbered file STRU{istep} (e.g., STRU1, STRU2) is written every out_freq_ion steps (when out_freq_ion is positive), and the final converged structure is written to STRU_FINAL. No CIF files are output. - 2: CIF format files are output. The latest structure is written to STRU_NOW.cif (overwritten each step), the numbered file STRU{istep}.cif (e.g., STRU1.cif, STRU2.cif) is written every out_freq_ion steps (when out_freq_ion is positive), and the final converged structure is written to STRU_FINAL.cif. No non-CIF files are output. @@ -3447,7 +3578,7 @@ parameters: category: Output information type: "Boolean \\[Integer\\](optional)" description: | - Generate files containing the kinetic energy matrix. The optional second parameter controls text output precision. The format will be the same as the Hamiltonian matrix and overlap matrix as mentioned in out_hsr. The name of the files will be trs1_nao.csr and so on. Also controled by out_freq_ion and out_app_flag. + Generate files containing the kinetic energy matrix. The optional second parameter controls text output precision. The format will be the same as the Hamiltonian matrix and overlap matrix as mentioned in out_hsr. The name of the files will be tr_nao.csr and so on. Also controled by out_freq_ion and out_app_flag. [NOTE] In the 3.10-LTS version, the file name is data-TR-sparse_SPIN0.csr. default_value: False 8 @@ -3515,16 +3646,6 @@ parameters: default_value: 0 8 unit: Ry/Bohr availability: "" - - name: out_mat_dh_exx - category: Output information - type: Integer - description: | - Whether to print files containing the derivatives of the exact-exchange matrix dV^EXX/dR. - - See out_mat_dh for format details. - default_value: 0 8 - unit: Ry/Bohr - availability: "" - name: out_mat_h_t category: Output information type: Integer @@ -3619,7 +3740,7 @@ parameters: category: Output information type: "Boolean \\[Integer\\](optional)" description: | - Whether to print the expectation value of the angular momentum operator , , and in the basis of the localized atomic orbitals. The files are named OUT.{suffix}_Lx.dat, OUT.{suffix}_Ly.dat, and OUT.{suffix}_Lz.dat. The second integer controls the precision of the output. + Whether to print the expectation value of the angular momentum operator , , and in the basis of the localized atomic orbitals. The files are named lx_nao.txt, ly_nao.txt, and lz_nao.txt (or lxg{step+1}_nao.txt etc. when out_freq_ion is set). The second integer controls the precision of the output. default_value: False 8 unit: "" availability: basis_type==lcao @@ -3813,7 +3934,7 @@ parameters: unit: "" availability: esolver_type==ksdft and basis_type==pw - name: out_dipole - category: "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory" + category: Real-Time TDDFT (Common) type: Boolean description: | Controls electric-dipole output. In RT-TDDFT, each enabled spin channel is written to OUT.{suffix}/dipole_s[spin].txt using a one-based spin number. Every row contains the one-based electronic-step index followed by the Cartesian electronic-dipole components $P_x$, $P_y$, and $P_z$ in atomic units. The running log additionally reports the electronic, ionic, and total dipoles and the norm of the total dipole. @@ -3823,28 +3944,28 @@ parameters: unit: "" availability: "" - name: out_current - category: "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory" + category: Real-Time TDDFT (Common) type: Integer description: | - Controls the current-density output method for LCAO RT-TDDFT. Output rows contain the one-based electronic-step index followed by $J_x$, $J_y$, and $J_z$ in atomic units. + Controls the current-density output method for RT-TDDFT. Each row contains the one-based electronic-step index followed by $J_x$, $J_y$, and $J_z$ in atomic units; the initial ground state is step 1 and subsequent steps increase by one. * 0: Do not output current. - * 1: Explicitly construct the velocity operator from the momentum, vector-potential, and KB nonlocal-pseudopotential terms using two-center and spherical-grid integrals: $\hat{v}_{\alpha}=-\mathrm{i}\nabla_{\alpha}+A_{\alpha}(t)+\mathrm{i}\left[\widetilde{V}_{\mathrm{NL}}^{\mathrm{KB}},r_{\alpha}\right]$, where $\widetilde{V}_{\mathrm{NL}}^{\mathrm{KB}}=\mathrm{e}^{-\mathrm{i}\boldsymbol{A}(t)\cdot\boldsymbol{r}}\hat{V}_{\mathrm{NL}}^{\mathrm{KB}}\mathrm{e}^{\mathrm{i}\boldsymbol{A}(t)\cdot\boldsymbol{r}}$. $\boldsymbol{A}(t)$ is nonzero only for the velocity gauge (td_stype=1); otherwise $\boldsymbol{A}(t)=0$. Other nonlocal Hamiltonian terms, such as EXX, are not included explicitly. The total current is written to OUT.{suffix}/current_tot.txt. - * 2: Use the full Hamiltonian to construct the generalized velocity matrix in a nonorthogonal NAO basis, $\widetilde{v}_{\alpha}=\partial_{\alpha}H+\mathrm{i}HS^{-1}\mathcal{R}_{\alpha}-\mathrm{i}\mathcal{R}_{\alpha}S^{-1}H-HS^{-1}\partial_{\alpha}S$. This includes all contributions available in the real-space Hamiltonian matrix when enabled. This method is more general but more expensive. The total current is written to OUT.{suffix}/current_tot_comm.txt. + * 1: Available for PW and LCAO. PW evaluates the occupied-state expectation values of the velocity operator, using plane-wave momentum and nonlocal-projector derivatives; in the velocity gauge these are evaluated at the vector-potential-shifted momentum. In the length gauge, a kinetic-energy-density functional also contributes its velocity term. LCAO explicitly constructs the velocity operator from the momentum, vector-potential, and KB nonlocal-pseudopotential terms using two-center and spherical-grid integrals: $\hat{v}_{\alpha}=-\mathrm{i}\nabla_{\alpha}+A_{\alpha}(t)+\mathrm{i}\left[\widetilde{V}_{\mathrm{NL}}^{\mathrm{KB}},r_{\alpha}\right]$, where $\widetilde{V}_{\mathrm{NL}}^{\mathrm{KB}}=\mathrm{e}^{-\mathrm{i}\boldsymbol{A}(t)\cdot\boldsymbol{r}}\hat{V}_{\mathrm{NL}}^{\mathrm{KB}}\mathrm{e}^{\mathrm{i}\boldsymbol{A}(t)\cdot\boldsymbol{r}}$. $\boldsymbol{A}(t)$ is nonzero only for the velocity gauge (td_stype=1); otherwise $\boldsymbol{A}(t)=0$. Other nonlocal Hamiltonian terms, such as EXX, are not included explicitly. The total current is written to OUT.{suffix}/current_tot.txt. + * 2: Available only for LCAO. Use the full Hamiltonian to construct the generalized velocity matrix in a nonorthogonal NAO basis, $\widetilde{v}_{\alpha}=\partial_{\alpha}H+\mathrm{i}HS^{-1}\mathcal{R}_{\alpha}-\mathrm{i}\mathcal{R}_{\alpha}S^{-1}H-HS^{-1}\partial_{\alpha}S$. This includes all contributions available in the real-space Hamiltonian matrix when enabled. This method is more general but more expensive. The total current is written to OUT.{suffix}/current_tot_comm.txt. default_value: "0" unit: "" - availability: basis_type==lcao and esolver_type==tddft + availability: "basis_type in [pw, lcao] and esolver_type==tddft" - name: out_current_k - category: "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory" + category: Real-Time TDDFT (Common) type: Boolean description: | - Controls whether LCAO RT-TDDFT current density is also resolved by spin and k-point. The total-current file is always written when out_current is 1 or 2. - * True: In addition to the total, out_current=1 writes OUT.{suffix}/current_s[spin]k[kpoint].txt; out_current=2 writes OUT.{suffix}/current_s[spin]k[kpoint]_comm.txt. Both use one-based spin and k-point numbers, with k-points numbered independently within each spin channel. Each row contains the one-based electronic-step index followed by $J_x$, $J_y$, and $J_z$ in atomic units. + Controls whether RT-TDDFT current density is also resolved by spin and k-point. PW supports out_current=1; LCAO supports out_current=1 or 2. The total-current file is always written when current output is enabled. + * True: In addition to the total, out_current=1 writes OUT.{suffix}/current_s[spin]k[kpoint].txt; out_current=2 writes OUT.{suffix}/current_s[spin]k[kpoint]_comm.txt. Both use one-based spin and k-point numbers, with k-points numbered independently within each spin channel. Rows follow the step numbering, atomic units, precision, and append convention described in out_current. These weighted contributions sum to the total current. * False: Output only current_tot.txt for out_current=1 or current_tot_comm.txt for out_current=2. default_value: "False" unit: "" - availability: "basis_type==lcao and esolver_type==tddft and out_current>0" + availability: "basis_type in [pw, lcao] and esolver_type==tddft and out_current>0" - name: out_efield - category: "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory" + category: Real-Time TDDFT (Common) type: Boolean description: | Controls time-dependent electric-field output. For each configured field, OUT.{suffix}/efield_[index].txt contains two columns: physical time in fs and the field value in V/Angstrom. The one-based field index follows the occurrence order shared by td_ttype and td_vext_dire, so fields assigned to the same direction remain in separate files. At initialization, a fresh calculation with md_restart=False truncates the files corresponding to the currently configured fields, whereas a calculation with md_restart=True preserves them and appends new samples. @@ -3854,10 +3975,10 @@ parameters: unit: "" availability: esolver_type==tddft and td_vext==true - name: out_vecpot - category: "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory" + category: Real-Time TDDFT (LCAO) type: Boolean description: | - Controls Cartesian vector-potential output for LCAO RT-TDDFT. OUT.{suffix}/vector_pot.txt contains four columns: the one-based electronic-step index followed by $A_x$, $A_y$, and $A_z$ in atomic units. At initialization, a fresh calculation with md_restart=False truncates the file and writes a new header, whereas a calculation with md_restart=True preserves a nonempty existing file and appends new samples. If the restart output file is missing or empty, a new file with a header is created. + Controls Cartesian vector-potential output for LCAO RT-TDDFT. OUT.{suffix}/vector_pot.txt contains four columns: the one-based electronic-step index followed by $A_x$, $A_y$, and $A_z$ in Hartree atomic units. The vector potential is the propagation value: the average of interval endpoints for an integrated electric field, or the supplied sample for init_vecpot_file=True. At initialization, a fresh calculation with md_restart=False truncates the file and writes a new header, whereas a calculation with md_restart=True preserves a nonempty existing file and appends new samples. If the restart output file is missing or empty, a new file with a header is created. * True: Write vector-potential samples on electronic propagation steps. * False: Do not output the vector potential. default_value: "False" @@ -3883,8 +4004,8 @@ parameters: category: Density of states type: Real description: | - The step size in writing Density of States (DOS) - default_value: "0.01" + The step size in writing Density of States (DOS). The default value was changed from 0.01 to 0.03 eV. + default_value: "0.03" unit: eV availability: "" - name: dos_sigma @@ -5267,7 +5388,7 @@ parameters: default_value: "True" unit: "" availability: exx_separate_loop==true - - name: exx_gamma_extrapolation + - name: exx_gamma_extra category: Exact Exchange (PW) type: Boolean description: | @@ -5279,10 +5400,18 @@ parameters: category: Exact Exchange (PW) type: Real description: | - The energy cutoff for EXX (Fock) exchange operator in plane wave basis calculations. Reducing ecutexx below ecutrho may significantly accelerate EXX computations. This speed improvement comes with a reduced numerical accuracy in the exchange energy calculation. + The energy cutoff for EXX (Fock) exchange operator in plane wave basis calculations. The pair-density G-sphere of the exchange operator, the EXX energy, and the EXX stress are all truncated at this value. If ecutexx yields a smaller FFT box and every |k+G|^2 of the wavefunctions fits inside it (i.e. ecutexx should not be smaller than ecutwfc), all EXX FFTs run on that smaller grid (QE ecutfock-style), which can significantly accelerate EXX computations. If the small grid is not usable (box not smaller, wavefunctions do not fit, or the FFT box is distributed over MPI), a warning is printed and the full grid is used. Reducing ecutexx below ecutrho reduces the numerical accuracy of the exchange contribution. default_value: same as ecutrho unit: Ry availability: "" + - name: exx_batch_size + category: Exact Exchange (PW) + type: Integer + description: | + Number of bands processed per round of the EXX batched FFT path. 0 (the default) processes all bands in one round, which is fastest but needs nbands * nxyz work buffers; a positive value processes the bands in chunks of that width, trading some performance for a proportionally smaller memory footprint. The result is independent of the chunking. + default_value: "0" + unit: "" + availability: "" - name: exx_thr_type category: Exact Exchange (PW) type: String diff --git a/docs/quick_start/easy_install.md b/docs/quick_start/easy_install.md index b8db35a5fd8..637b4bb54fb 100644 --- a/docs/quick_start/easy_install.md +++ b/docs/quick_start/easy_install.md @@ -1,6 +1,6 @@ # Easy Installation -This guide helps you install ABACUS with basic features. **For DeePKS, DeePMD and Libxc support, or building with `make`, please refer to [the advanced installation guide](../advanced/install.md)** after going through this page. We recommend building ABACUS with `cmake` to avoid dependency issues. We recommend compiling ABACUS (and possibly its requirements) from the source code using the latest compiler for the best performace. You can use [toolchain](#install-by-toolchain) to install ABACUS and dependencies in a source-code compilation way with convience. You can also deploy ABACUS **without building** by [Docker](#container-deployment) or [conda](#install-by-conda). Please note that ABACUS only supports Linux; for Windows users, please consider using [WSL](https://learn.microsoft.com/en-us/windows/wsl/) or docker. For a scripted one-click setup that provisions WSL2 and installs ABACUS inside it automatically, see [Windows One-Click Installer](./advanced/windows_installer.md). +This guide helps you install ABACUS with basic features. **For DeePKS, DeePMD and Libxc support, or building with `make`, please refer to [the advanced installation guide](../advanced/install.md)** after going through this page. We recommend building ABACUS with `cmake` to avoid dependency issues. We recommend compiling ABACUS (and possibly its requirements) from the source code using the latest compiler for the best performace. You can use [toolchain](#install-by-toolchain) to install ABACUS and dependencies in a source-code compilation way with convience. You can also deploy ABACUS **without building** by [Docker](#container-deployment) or [conda](#install-by-conda). Please note that ABACUS only supports Linux; for Windows users, please consider using [WSL](https://learn.microsoft.com/en-us/windows/wsl/) or docker. For a scripted one-click setup that provisions WSL2 and installs ABACUS inside it automatically, see [Windows One-Click Installer](../advanced/windows_installer.md). ## Get ABACUS source code @@ -124,7 +124,7 @@ Possible command: `cmake -B build -DENABLE_MLALGO=ON -DENABLE_LIBXC=ON -DENABLE_ Some of these packages can be installed with popular package management system via root permission if you have, such as `apt` and `yum`. For example: ```bash -sudo apt update && sudo apt install -y libopenmpi-dev libopenblas-openmp-dev libscalapack-openmpi-dev libelpa-dev libfftw3-dev libcereal-dev libxc-dev g++ make cmake bc git pkgconf +sudo apt update && sudo apt install -y libopenmpi-dev libopenblas-openmp-dev libscalapack-openmpi-dev libelpa-dev libfftw3-dev libcereal-dev libxc-dev nlohmann-json3-dev g++ make cmake bc git pkgconf ``` > Installing ELPA by apt only matches requirements on Ubuntu 22.04. For earlier linux distributions, you should build ELPA from source. @@ -164,13 +164,14 @@ Here, 'build' is the path for building ABACUS; and '-D' is used for setting up s - `LIBCOMM_DIR`: (Optional) Path to LibComm when `ENABLE_LIBRI=ON`. ```{important} -For some dependencies built with CMake, such as Libxc, dftd4, cereal, and RapidJSON, you'll have to add their prefix paths to the environment variable `CMAKE_PREFIX_PATH` so that CMake can correctly find and use their CMake configuration files. A non-general variable such as `PKG_DIR` is discouraged for these packages. +For some dependencies built with CMake, such as Libxc, dftd4, cereal, and nlohmann-json, you'll have to add their prefix paths to the environment variable `CMAKE_PREFIX_PATH` so that CMake can correctly find and use their CMake configuration files. A non-general variable such as `PKG_DIR` is discouraged for these packages. ``` - Components: The values of these variables should be 'ON', '1' or 'OFF', '0'. The default values are given below. - `ENABLE_LCAO=ON`: Enable LCAO calculation. If SCALAPACK, ELPA or CEREAL is absent and only require plane-wave calculations, the feature of calculating LCAO basis can be turned off. - `ENABLE_LIBXC=OFF`: [Enable Libxc](../advanced/install.md#add-libxc-support) to suppport variety of functionals. - `ENABLE_LIBRI=OFF`: [Enable LibRI](../advanced/install.md#add-libri-support) and its LibComm dependency for hybrid-functional calculations. Set `LIBRI_DIR` and `LIBCOMM_DIR` to use manually installed libraries. + - `ENABLE_JSON=OFF`: [Enable JSON output](../advanced/install.md#build-with-json-support) through nlohmann-json. - `ENABLE_OPENMP=ON`: Enable OpenMP support. Building ABACUS without OpenMP is not fully tested yet. - `BUILD_TESTING=OFF`: [Build unit tests](../advanced/install.md#build-unit-tests). - `ENABLE_GOOGLEBENCH=OFF`: [Build performance tests](../advanced/install.md#build-performance-tests) @@ -271,7 +272,7 @@ Usually a CPU has a feature named Hyper-Threading(HT). This feature enables one > Please note that containers target at developing and testing, but not massively parallel computing for production. Docker has a bad support to MPI, which may cause performance degradation. -We've built a ready-for-use version of ABACUS with docker [here](https://github.com/deepmodeling/abacus-develop/pkgs/container/abacus). For a quick start: pull the image, prepare the data, run container. Instructions on using the image can be accessed in [Dockerfile](../../Dockerfile). A mirror is available by `docker pull registry.dp.tech/deepmodeling/abacus`. +We've built a ready-for-use version of ABACUS with docker [here](https://github.com/deepmodeling/abacus-develop/pkgs/container/abacus). For a quick start: pull the image, prepare the data, run container. Instructions on using the image can be accessed in [Dockerfile](https://github.com/deepmodeling/abacus-develop/blob/develop/Dockerfile). A mirror is available by `docker pull registry.dp.tech/deepmodeling/abacus`. We also offer a pre-built docker image containing all the requirements for development. Please refer to our [Package Page](https://github.com/orgs/deepmodeling/packages?repo_name=abacus-develop). diff --git a/docs/quick_start/input.md b/docs/quick_start/input.md index f4dfe068afc..1f0ac33c640 100644 --- a/docs/quick_start/input.md +++ b/docs/quick_start/input.md @@ -8,7 +8,7 @@ The `INPUT` file contains parameters that control the type of calculation as wel Below is an example `INPUT` file with some of the most important parameters that need to be set: -```plaintext +```text INPUT_PARAMETERS suffix MgO # the output files will be in OUT.{suffix} directory pseudo_dir ./ # where the pseudopotential for each element is @@ -103,7 +103,7 @@ The structure file contains structural information about the system, e.g., latti An example of the `STRU` file is given as follows : -```plaintext +```text #This is the atom file containing all the information #about the lattice structure. @@ -152,7 +152,7 @@ This file contains information of the kpoint grid setting for the Brillouin zone An example of the `KPT` file is given below: -```plaintext +```text K_POINTS 0 Gamma diff --git a/examples/03_spin_polarized/01_H_atom/INPUT b/examples/03_spin_polarized/01_H_atom/INPUT index 18a2794e88e..a14d3b8fda4 100644 --- a/examples/03_spin_polarized/01_H_atom/INPUT +++ b/examples/03_spin_polarized/01_H_atom/INPUT @@ -51,7 +51,7 @@ out_stru 1 # print STRU in OUT out_chg 0 # print CHG or not out_mul 1 # print Mulliken charge and mag of atom in mulliken.txt # out_wfc_lcao 1 -# out_interval 1 +# out_freq_ion 1 # restart setting # restart_save auto # false, auto, other # restart_load false diff --git a/examples/03_spin_polarized/01_H_atom/reference/running_scf.log.ref b/examples/03_spin_polarized/01_H_atom/reference/running_scf.log.ref index 68f568f337b..8db0b1d2aed 100644 --- a/examples/03_spin_polarized/01_H_atom/reference/running_scf.log.ref +++ b/examples/03_spin_polarized/01_H_atom/reference/running_scf.log.ref @@ -638,8 +638,8 @@ DiagoElpa elpa_solve 0.00 14 0.00 0. ElecStateLCAO psiToRho 0.34 7 0.05 2.06 ElecStateLCAO cal_dm_2d 0.00 7 0.00 0.00 elecstate cal_dm 0.00 8 0.00 0.00 -psiMulPsiMpi pdgemm 0.00 16 0.00 0.00 -DensityMatrix cal_DMR 0.00 7 0.00 0.00 +cal_dmk_psi pdgemm 0.00 16 0.00 0.00 +DensityMatrix cal_dmr 0.00 7 0.00 0.00 Gint transfer_DMR 0.00 7 0.00 0.01 Gint_interface cal_gint_rho 0.07 7 0.01 0.45 Charge_Mixing get_drho 0.02 7 0.00 0.09 diff --git a/examples/03_spin_polarized/01_H_atom/reference/scf.output.ref b/examples/03_spin_polarized/01_H_atom/reference/scf.output.ref index 41fe4824a0a..7e7eeb5a074 100644 --- a/examples/03_spin_polarized/01_H_atom/reference/scf.output.ref +++ b/examples/03_spin_polarized/01_H_atom/reference/scf.output.ref @@ -125,8 +125,8 @@ DiagoElpa elpa_solve 0.00 14 0.00 0. ElecStateLCAO psiToRho 0.34 7 0.05 2.06 ElecStateLCAO cal_dm_2d 0.00 7 0.00 0.00 elecstate cal_dm 0.00 8 0.00 0.00 -psiMulPsiMpi pdgemm 0.00 16 0.00 0.00 -DensityMatrix cal_DMR 0.00 7 0.00 0.00 +cal_dmk_psi pdgemm 0.00 16 0.00 0.00 +DensityMatrix cal_dmr 0.00 7 0.00 0.00 Gint transfer_DMR 0.00 7 0.00 0.01 Gint_interface cal_gint_rho 0.07 7 0.01 0.45 Charge_Mixing get_drho 0.02 7 0.00 0.09 diff --git a/examples/03_spin_polarized/02_Fe_FM/reference/running_scf.log.ref b/examples/03_spin_polarized/02_Fe_FM/reference/running_scf.log.ref index a2b17a38ac0..543ffba72f2 100644 --- a/examples/03_spin_polarized/02_Fe_FM/reference/running_scf.log.ref +++ b/examples/03_spin_polarized/02_Fe_FM/reference/running_scf.log.ref @@ -5343,8 +5343,8 @@ HSolverLCAO hamiltSolvePsiK 3.90 1764 0.00 16. DiagoElpa elpa_solve 3.33 1764 0.00 13.86 ElecStateLCAO psiToRho 8.12 14 0.58 33.78 elecstate cal_dm 0.07 14 0.01 0.31 -psiMulPsiMpi pdgemm 0.07 1764 0.00 0.29 -DensityMatrix cal_DMR 0.40 14 0.03 1.68 +cal_dmk_psi pdgemm 0.07 1764 0.00 0.29 +DensityMatrix cal_dmr 0.40 14 0.03 1.68 Local_Orbital_wfc wfc_2d_to_grid 0.12 1890 0.00 0.50 Gint transfer_DMR 0.23 14 0.02 0.94 Gint_interface cal_gint_rho 7.10 14 0.51 29.56 diff --git a/examples/03_spin_polarized/02_Fe_FM/reference/scf.output.ref b/examples/03_spin_polarized/02_Fe_FM/reference/scf.output.ref index 77aa7a6488d..d676d388306 100644 --- a/examples/03_spin_polarized/02_Fe_FM/reference/scf.output.ref +++ b/examples/03_spin_polarized/02_Fe_FM/reference/scf.output.ref @@ -130,8 +130,8 @@ HSolverLCAO hamiltSolvePsiK 3.90 1764 0.00 16. DiagoElpa elpa_solve 3.33 1764 0.00 13.86 ElecStateLCAO psiToRho 8.12 14 0.58 33.78 elecstate cal_dm 0.07 14 0.01 0.31 -psiMulPsiMpi pdgemm 0.07 1764 0.00 0.29 -DensityMatrix cal_DMR 0.40 14 0.03 1.68 +cal_dmk_psi pdgemm 0.07 1764 0.00 0.29 +DensityMatrix cal_dmr 0.40 14 0.03 1.68 Local_Orbital_wfc wfc_2d_to_grid 0.12 1890 0.00 0.50 Gint transfer_DMR 0.23 14 0.02 0.94 Gint_interface cal_gint_rho 7.10 14 0.51 29.56 diff --git a/examples/03_spin_polarized/03_Fe_AFM/reference/running_scf.log.ref b/examples/03_spin_polarized/03_Fe_AFM/reference/running_scf.log.ref index 0125faebfc4..38236c35b92 100644 --- a/examples/03_spin_polarized/03_Fe_AFM/reference/running_scf.log.ref +++ b/examples/03_spin_polarized/03_Fe_AFM/reference/running_scf.log.ref @@ -4849,8 +4849,8 @@ HSolverLCAO hamiltSolvePsiK 6.35 2898 0.00 16. DiagoElpa elpa_solve 5.46 2898 0.00 13.93 ElecStateLCAO psiToRho 13.41 23 0.58 34.20 elecstate cal_dm 0.12 23 0.01 0.30 -psiMulPsiMpi pdgemm 0.11 2898 0.00 0.28 -DensityMatrix cal_DMR 0.66 23 0.03 1.70 +cal_dmk_psi pdgemm 0.11 2898 0.00 0.28 +DensityMatrix cal_dmr 0.66 23 0.03 1.70 Local_Orbital_wfc wfc_2d_to_grid 0.19 3024 0.00 0.47 Gint transfer_DMR 0.38 23 0.02 0.98 Gint_interface cal_gint_rho 11.97 23 0.52 30.53 diff --git a/examples/03_spin_polarized/03_Fe_AFM/reference/scf.output.ref b/examples/03_spin_polarized/03_Fe_AFM/reference/scf.output.ref index d34ff54d901..1d76dd81c70 100644 --- a/examples/03_spin_polarized/03_Fe_AFM/reference/scf.output.ref +++ b/examples/03_spin_polarized/03_Fe_AFM/reference/scf.output.ref @@ -139,8 +139,8 @@ HSolverLCAO hamiltSolvePsiK 6.35 2898 0.00 16. DiagoElpa elpa_solve 5.46 2898 0.00 13.93 ElecStateLCAO psiToRho 13.41 23 0.58 34.20 elecstate cal_dm 0.12 23 0.01 0.30 -psiMulPsiMpi pdgemm 0.11 2898 0.00 0.28 -DensityMatrix cal_DMR 0.66 23 0.03 1.70 +cal_dmk_psi pdgemm 0.11 2898 0.00 0.28 +DensityMatrix cal_dmr 0.66 23 0.03 1.70 Local_Orbital_wfc wfc_2d_to_grid 0.19 3024 0.00 0.47 Gint transfer_DMR 0.38 23 0.02 0.98 Gint_interface cal_gint_rho 11.97 23 0.52 30.53 diff --git a/examples/04_noncollinear/01_Fe_noncollinear/reference/scf.output.ref b/examples/04_noncollinear/01_Fe_noncollinear/reference/scf.output.ref index 7882dc68ff2..8d1d0f2024c 100644 --- a/examples/04_noncollinear/01_Fe_noncollinear/reference/scf.output.ref +++ b/examples/04_noncollinear/01_Fe_noncollinear/reference/scf.output.ref @@ -140,8 +140,8 @@ HSolverLCAO hamiltSolvePsiK 1.08 192 0.01 0. DiagoElpa elpa_solve 0.90 192 0.00 0.79 ElecStateLCAO psiToRho 47.23 24 1.97 41.47 elecstate cal_dm 0.02 24 0.00 0.02 -psiMulPsiMpi pdgemm 0.02 192 0.00 0.02 -DensityMatrix cal_DMR 0.11 24 0.00 0.09 +cal_dmk_psi pdgemm 0.02 192 0.00 0.02 +DensityMatrix cal_dmr 0.11 24 0.00 0.09 Local_Orbital_wfc wfc_2d_to_grid 0.04 200 0.00 0.04 Gint transfer_DMR 1.34 24 0.06 1.17 Gint_interface cal_gint_rho 44.89 24 1.87 39.41 diff --git a/examples/13_dos/01_pw_Al/run.sh b/examples/13_dos/01_pw_Al/run.sh index 9bdfbc75a58..618ea429607 100755 --- a/examples/13_dos/01_pw_Al/run.sh +++ b/examples/13_dos/01_pw_Al/run.sh @@ -18,9 +18,8 @@ if [[ ! -f scf.output ]] || [[ ! -f nscf.output ]] || [[ ! -f OUT.ABACUS/running_scf.log ]] || [[ ! -f OUT.ABACUS/running_nscf.log ]] || - [[ ! -f OUT.ABACUS/DOS1 ]] || - [[ ! -f OUT.ABACUS/DOS1_smearing.dat ]] || - [[ ! -f OUT.ABACUS/SPIN1_CHG.cube ]] || + [[ ! -f OUT.ABACUS/dos.txt ]] || + [[ ! -f OUT.ABACUS/chg.cube ]] || [[ ! ( "$(tail -1 OUT.ABACUS/running_scf.log)" == " Total Time :"* ) ]] || [[ ! ( "$(tail -1 OUT.ABACUS/running_nscf.log)" == " Total Time :"* ) ]] then diff --git a/examples/13_dos/02_lcao_Si2/run.sh b/examples/13_dos/02_lcao_Si2/run.sh index 9bdfbc75a58..fa91c0046cf 100755 --- a/examples/13_dos/02_lcao_Si2/run.sh +++ b/examples/13_dos/02_lcao_Si2/run.sh @@ -18,9 +18,8 @@ if [[ ! -f scf.output ]] || [[ ! -f nscf.output ]] || [[ ! -f OUT.ABACUS/running_scf.log ]] || [[ ! -f OUT.ABACUS/running_nscf.log ]] || - [[ ! -f OUT.ABACUS/DOS1 ]] || - [[ ! -f OUT.ABACUS/DOS1_smearing.dat ]] || - [[ ! -f OUT.ABACUS/SPIN1_CHG.cube ]] || + [[ ! -f OUT.ABACUS/doss1g1_nao.txt ]] || + [[ ! -f OUT.ABACUS/chg.cube ]] || [[ ! ( "$(tail -1 OUT.ABACUS/running_scf.log)" == " Total Time :"* ) ]] || [[ ! ( "$(tail -1 OUT.ABACUS/running_nscf.log)" == " Total Time :"* ) ]] then diff --git a/examples/13_dos/README b/examples/13_dos/README index bdd25b93611..fb4cbe244fd 100644 --- a/examples/13_dos/README +++ b/examples/13_dos/README @@ -16,11 +16,10 @@ It requires two steps of calculation: - set 'out_dos' to '1' - set 'pw_diag_thr' to a suitable value when basis_type is 'pw' -Finish! Then you can see DOS1_smearing.dat in OUT.ABACUS. The first column is energy, -and the second is density of states. Plot it! +Finish! Then you can see the DOS file in OUT.ABACUS (dos.txt for PW basis, doss1g{geom}_nao.txt for LCAO basis). The columns are: energy(eV), dos(1/eV), dos_int (integrated DOS), dos_smear(1/eV), dos_smear_int (integrated smeared DOS). Plot it! Note: -1)If the "out_dos" is used, the DOS1_smearing.dat will appear in the directory. +1)If the "out_dos" is used, the DOS file will appear in the directory. 2)The charge density needed to be output in scf calculations and be read in non-self consistent calculations. diff --git a/examples/17_relax/03_relax_with_output_pw/INPUT b/examples/17_relax/03_relax_with_output_pw/INPUT index d2b8b5660d1..305dea867fd 100644 --- a/examples/17_relax/03_relax_with_output_pw/INPUT +++ b/examples/17_relax/03_relax_with_output_pw/INPUT @@ -34,4 +34,4 @@ out_band 1 out_stru 1 out_app_flag 0 -out_interval 1 +out_freq_ion 1 diff --git a/examples/17_relax/04_relax_with_output_lcao/INPUT b/examples/17_relax/04_relax_with_output_lcao/INPUT index e83244b6b0a..05125b6fa1e 100644 --- a/examples/17_relax/04_relax_with_output_lcao/INPUT +++ b/examples/17_relax/04_relax_with_output_lcao/INPUT @@ -24,7 +24,7 @@ relax_method cg 1 pseudo_dir ../../../tests/PP_ORB orbital_dir ../../../tests/PP_ORB -nspin 4 +nspin 2 out_chg 1 out_pot 1 @@ -36,4 +36,9 @@ out_band 1 out_stru 1 out_app_flag 0 -out_interval 1 +out_mat_l 1 +out_mat_r 1 +out_mat_t 1 +#out_mat_dh 1 + +out_freq_ion 2 diff --git a/examples/20_hybrid/02_lcao_Si2/log_ref b/examples/20_hybrid/02_lcao_Si2/log_ref index 526b4cdb230..4b734af3230 100644 --- a/examples/20_hybrid/02_lcao_Si2/log_ref +++ b/examples/20_hybrid/02_lcao_Si2/log_ref @@ -143,7 +143,7 @@ Please make sure this is what you need DiagoElpa elpa_solve 13.171 25088 0.00052 4.9 % ElecStateLCAO psiToRho 3.2479 49 0.066 1.2 % elecstate cal_dm 0.23405 49 0.0048 0.087 % - psiMulPsiMpi pdgemm 0.10221 25088 4.1e-06 0.038 % + cal_dmk_psi pdgemm 0.10221 25088 4.1e-06 0.038 % LCAO_Charge cal_dk_k 2.3764 49 0.048 0.88 % Gint_interface cal_gint_rho 0.57094 49 0.012 0.21 % Exx_LRI cal_exx_elec 128.96 9 14 48 % diff --git a/examples/20_hybrid/02_lcao_Si2/running_scf.log_ref b/examples/20_hybrid/02_lcao_Si2/running_scf.log_ref index 507b7bb9a7b..816025bfaf5 100644 --- a/examples/20_hybrid/02_lcao_Si2/running_scf.log_ref +++ b/examples/20_hybrid/02_lcao_Si2/running_scf.log_ref @@ -6012,7 +6012,7 @@ DiagoElpa elpa_solve 13.171 25088 0.00052 4.9% ElecStateLCAO psiToRho 3.2479 49 0.066 1.2% elecstate cal_dm 0.23405 49 0.0048 0.087% - psiMulPsiMpi pdgemm 0.10221 25088 4.1e-06 0.038% + cal_dmk_psi pdgemm 0.10221 25088 4.1e-06 0.038% LCAO_Charge cal_dk_k 2.3764 49 0.048 0.88% Gint_interface cal_gint_rho 0.57094 49 0.012 0.21% Exx_LRI cal_exx_elec 128.96 9 14. 48.% diff --git a/interfaces/ASE_interface/abacuslite/core.py b/interfaces/ASE_interface/abacuslite/core.py index fffa7b63dab..528ba73a558 100644 --- a/interfaces/ASE_interface/abacuslite/core.py +++ b/interfaces/ASE_interface/abacuslite/core.py @@ -372,11 +372,34 @@ def execute(self, outputfile=self.outputname, errorfile=self.errorname) except SubprocessError: + # Surface only the tails of each log: abacus.out already contains + # the SCF iteration table, so full stdout/stderr plus 200 lines of + # running log would make the exception message unreadably long. + # The run dir is preserved separately by the test harness + # (ASE_ABACUS_KEEP_DIR) when post-mortem of the full logs is + # needed. + def tail(p: Path, n: int) -> str: + if not p.exists(): + return '' + lines = p.read_text(errors='replace').splitlines() + header = f'=== {p.name} (last {min(n, len(lines))} lines) ===' + return header + '\n' + '\n'.join(lines[-n:]) + + directory = Path(directory) message = ['ABACUS Lite calculation failed'] - with open(directory / self.outputname, 'r') as f: - message.append(f.read()) - with open(directory / self.errorname, 'r') as f: - message.append(f.read()) + for fn, n in ((self.outputname, 200), + (self.errorname, 200)): + chunk = tail(directory / fn, n) + if chunk: + message.append(chunk) + # The tail of the SCF running log holds the per-step drho + # history, usually the key evidence for issues like #7794 + # ("Can't find even an electron"). + scf_log = (directory / f'OUT.{self.suffix}' + / f'running_{self.calculation}.log') + chunk = tail(scf_log, 100) + if chunk: + message.append(chunk) raise SubprocessError('\n'.join(message)) def read_results(self, directory) -> Dict: diff --git a/interfaces/ASE_interface/abacuslite/io/generalio.py b/interfaces/ASE_interface/abacuslite/io/generalio.py index 053b42203a4..c690964a5bb 100644 --- a/interfaces/ASE_interface/abacuslite/io/generalio.py +++ b/interfaces/ASE_interface/abacuslite/io/generalio.py @@ -495,7 +495,7 @@ def _trim(line): for line in blocks['LATTICE_VECTORS']] elif 'LATTICE_PARAMETER' in blocks: stru['lat']['param'] = [float(x) - for x in blocks['LATTICE_PARAMETERS'].split()] + for x in blocks['LATTICE_PARAMETER'][0].split()] #============ ATOMIC_SPECIES ============ stru['species'] = [dict(zip(['symbol', 'mass', 'pp_file', 'pp_type'], @@ -731,6 +731,38 @@ def test_stru_io(self): self.assertEqual(stru_['species'][0]['orb_file'], 'Na_gga_6au_100Ry_2s2p1d.orb') self.assertEqual(stru_['species'][1]['orb_file'], 'Cl_gga_6au_100Ry_2s2p1d.orb') + def test_read_stru_with_lattice_parameter(self): + # STRU that uses LATTICE_PARAMETER instead of LATTICE_VECTORS + # (issue #7555: previously raised KeyError on 'LATTICE_PARAMETERS') + content = ( + "ATOMIC_SPECIES\n" + "Si 28.085 Si.upf\n" + "\n" + "LATTICE_CONSTANT\n" + "10.2\n" + "\n" + "LATTICE_PARAMETER\n" + "5.43 5.43 5.43 90 90 90\n" + "\n" + "ATOMIC_POSITIONS\n" + "Direct\n" + "\n" + "Si\n" + "0.0\n" + "2\n" + "0.0 0.0 0.0 1 1 1\n" + "0.25 0.25 0.25 1 1 1\n" + ) + with tempfile.TemporaryDirectory() as tmpdir: + fn = Path(tmpdir) / 'STRU' + fn.write_text(content) + stru = read_stru(fn) + self.assertEqual(stru['lat']['const'], 10.2) + self.assertNotIn('vec', stru['lat']) + self.assertEqual(stru['lat']['param'], [5.43, 5.43, 5.43, 90.0, 90.0, 90.0]) + self.assertEqual(stru['species'][0]['symbol'], 'Si') + self.assertEqual(stru['species'][0]['natom'], 2) + def test_write_stru_preserves_first_occurrence_species_order(self): atoms = Atoms( symbols=['C', 'C', 'Pt', 'H', 'H'], diff --git a/interfaces/ASE_interface/tests/band.py b/interfaces/ASE_interface/tests/band.py index 78b80922111..1ba7799c83e 100644 --- a/interfaces/ASE_interface/tests/band.py +++ b/interfaces/ASE_interface/tests/band.py @@ -1,11 +1,11 @@ import unittest -import tempfile from pathlib import Path here = Path(__file__).parent from ase.build import bulk from abacuslite.io.generalio import load_pseudo, load_orbital from abacuslite import AbacusProfile, Abacus from abacuslite.utils.ksampling import kpathgen +from common import preserved_tmpdir class TestSCFFollowedNSCF(unittest.TestCase): @@ -28,7 +28,7 @@ def test(self): omp_num_threads=1, ) - with tempfile.TemporaryDirectory() as tmpdir: + with preserved_tmpdir('abacus_ase_band_') as tmpdir: abacus = Abacus( profile=aprof, directory=tmpdir, diff --git a/interfaces/ASE_interface/tests/common.py b/interfaces/ASE_interface/tests/common.py new file mode 100644 index 00000000000..dab62b072ab --- /dev/null +++ b/interfaces/ASE_interface/tests/common.py @@ -0,0 +1,31 @@ +"""Shared helpers for the ASE interface integration tests.""" + +import contextlib +import os +import shutil +import tempfile +from pathlib import Path + + +@contextlib.contextmanager +def preserved_tmpdir(prefix): + """Temporary ABACUS run directory kept on failure for post-mortem CI. + + On success the directory is removed like tempfile.TemporaryDirectory. + On exception it is copied into ASE_ABACUS_KEEP_DIR (default + /tmp/abacus_ase_failure) so remote-CI-only failures (e.g. issue #7794) + can be root-caused from the uploaded artifact. + """ + tmpdir = tempfile.mkdtemp(prefix=prefix) + try: + yield tmpdir + except Exception: + keep_root = Path(os.environ.get( + 'ASE_ABACUS_KEEP_DIR', '/tmp/abacus_ase_failure')) + keep_root.mkdir(parents=True, exist_ok=True) + keep = keep_root / Path(tmpdir).name + shutil.copytree(tmpdir, keep, dirs_exist_ok=True) + print(f'[ase-test] failure, run dir preserved at {keep}') + raise + finally: + shutil.rmtree(tmpdir, ignore_errors=True) diff --git a/interfaces/ASE_interface/tests/magnetic.py b/interfaces/ASE_interface/tests/magnetic.py index 3fbacbdc998..080961c3e2e 100644 --- a/interfaces/ASE_interface/tests/magnetic.py +++ b/interfaces/ASE_interface/tests/magnetic.py @@ -3,12 +3,12 @@ and ferromagnetic phases calculation on the BCC Fe ''' import unittest -import tempfile import numpy as np from pathlib import Path here = Path(__file__).parent from ase.atoms import Atoms from abacuslite import AbacusProfile, Abacus +from common import preserved_tmpdir class TestMagneticSCF(unittest.TestCase): @@ -28,7 +28,7 @@ def test_antiferromagnetic(self): cell=np.eye(3) * 5.2, pbc=True) - with tempfile.TemporaryDirectory() as tmpdir: + with preserved_tmpdir('abacus_ase_afm_') as tmpdir: abacus = Abacus( profile=self.aprof, directory=tmpdir, @@ -58,7 +58,7 @@ def test_ferromagnetic(self): cell=np.eye(3) * 5.2, pbc=True) - with tempfile.TemporaryDirectory() as tmpdir: + with preserved_tmpdir('abacus_ase_fm_') as tmpdir: abacus = Abacus( profile=self.aprof, directory=tmpdir, @@ -96,7 +96,7 @@ def test_noncolinear(self): cell=np.eye(3) * 5.2, pbc=True) - with tempfile.TemporaryDirectory() as tmpdir: + with preserved_tmpdir('abacus_ase_nc_') as tmpdir: abacus = Abacus( profile=self.aprof, directory=tmpdir, diff --git a/interfaces/ASE_interface/tests/md.py b/interfaces/ASE_interface/tests/md.py index 2b771eb3df3..f00b9e3c965 100644 --- a/interfaces/ASE_interface/tests/md.py +++ b/interfaces/ASE_interface/tests/md.py @@ -1,5 +1,4 @@ import unittest -import tempfile from pathlib import Path here = Path(__file__).parent from ase.build import bulk @@ -8,6 +7,7 @@ from ase.md.velocitydistribution import MaxwellBoltzmannDistribution from abacuslite.io.generalio import load_pseudo, load_orbital from abacuslite import AbacusProfile, Abacus +from common import preserved_tmpdir class TestLangevinMolecularDynamics(unittest.TestCase): @@ -22,7 +22,7 @@ def test(self): omp_num_threads=1, ) - with tempfile.TemporaryDirectory() as tmpdir: + with preserved_tmpdir('abacus_ase_md_') as tmpdir: abacus = Abacus( profile=aprof, directory=tmpdir, diff --git a/interfaces/ASE_interface/tests/relax.py b/interfaces/ASE_interface/tests/relax.py index dc8e9e9fd70..1b3c2a98751 100644 --- a/interfaces/ASE_interface/tests/relax.py +++ b/interfaces/ASE_interface/tests/relax.py @@ -1,11 +1,11 @@ import unittest -import tempfile from pathlib import Path here = Path(__file__).parent from ase.build import bulk from ase.optimize import BFGS from abacuslite.io.generalio import load_pseudo, load_orbital from abacuslite import AbacusProfile, Abacus +from common import preserved_tmpdir class TestIonicRelaxationWithStress(unittest.TestCase): @@ -20,7 +20,7 @@ def test(self): omp_num_threads=1, ) - with tempfile.TemporaryDirectory() as tmpdir: + with preserved_tmpdir('abacus_ase_relax_') as tmpdir: abacus = Abacus( profile=aprof, directory=tmpdir, diff --git a/interfaces/ASE_interface/tests/scf.py b/interfaces/ASE_interface/tests/scf.py index 4d792e8eab2..d9f3e52e150 100644 --- a/interfaces/ASE_interface/tests/scf.py +++ b/interfaces/ASE_interface/tests/scf.py @@ -1,10 +1,10 @@ import unittest -import tempfile from pathlib import Path here = Path(__file__).parent from ase.build import bulk from abacuslite.io.generalio import load_pseudo, load_orbital from abacuslite import AbacusProfile, Abacus +from common import preserved_tmpdir class TestSCF(unittest.TestCase): @@ -19,7 +19,9 @@ def test(self): omp_num_threads=1, ) - with tempfile.TemporaryDirectory() as tmpdir: + # preserved_tmpdir keeps the run dir under ASE_ABACUS_KEEP_DIR on + # failure so remote-CI-only failures (issue #7794) can be root-caused. + with preserved_tmpdir('abacus_ase_scf_') as tmpdir: abacus = Abacus( profile=aprof, directory=tmpdir, @@ -36,4 +38,4 @@ def test(self): print('Silicon :', silicon.get_potential_energy()) if __name__ == '__main__': - unittest.main() \ No newline at end of file + unittest.main() diff --git a/interfaces/PLUMED_interface/README.md b/interfaces/PLUMED_interface/README.md new file mode 100644 index 00000000000..7648f715da4 --- /dev/null +++ b/interfaces/PLUMED_interface/README.md @@ -0,0 +1,68 @@ +# ABACUS + PLUMED interface + +[PLUMED](https://www.plumed.org) is a library for enhanced-sampling methods +(metadynamics, umbrella sampling, ...). During an ABACUS molecular-dynamics run +it can compute collective variables and apply biasing potentials, controlled by +the `plumed` and `plumed_file` INPUT keywords. + +## Building ABACUS with PLUMED + +Install PLUMED 2.x first, for example: + + ./configure --prefix=/path/to/plumed && make -j && make install + +Then configure ABACUS with `-DENABLE_PLUMED=ON`: + + cmake -B build -DENABLE_PLUMED=ON ... + cmake --build build -j + +The PLUMED installation is located with `pkg-config` (through the `plumed.pc` +file shipped by PLUMED); if PLUMED is installed in a non-standard prefix, add +`-DPLUMED_ROOT=/path/to/plumed` so that it can be found without pkg-config. +When ABACUS is built without `-DENABLE_PLUMED=ON`, setting `plumed` in INPUT +stops the run with an explicit error message. + +## Usage + +In the `INPUT` file: + + plumed 1 # enable the PLUMED interface + plumed_file plumed.dat # PLUMED input file (default: plumed.dat) + +The PLUMED input file defines the collective variables and the biasing +potential, e.g. for a dihedral angle: + + UNITS LENGTH=A TIME=fs + t: TORSION ATOMS=3,1,2,7 + metad: METAD ARG=t PACE=50 HEIGHT=2.0 SIGMA=0.15 FILE=HILLS BIASFACTOR=20 TEMP=600 + PRINT ARG=t,metad.bias FILE=COLVAR STRIDE=1 + +ABACUS hands the atomic positions (Bohr), masses (amu), cell, potential energy +and virial over to PLUMED after every MD step, and PLUMED adds the biasing +forces in place. + +Note: the interface currently supports a single MPI rank only, because PLUMED +must see the whole system while ABACUS distributes the atoms of an MD run over +the ranks. OpenMP threads inside that single rank are fine, e.g. +`OMP_NUM_THREADS=32 abacus > log`. + +## Example + +`example01` runs a well-tempered metadynamics of the ethane (C2H6) H-C-C-H +torsion at 600 K (LCAO). Running ABACUS in that directory produces + + - `COLVAR`: time, the torsion angle and the metadynamics bias at every MD step; + - `HILLS`: the Gaussian hills deposited by metadynamics. + +The free-energy surface can be reconstructed with the tools shipped with PLUMED: + + plumed sum_hills --hills HILLS --mintozero --outfile fes.dat + +`fes.dat` contains the torsion angle (rad) in the first column and the free +energy (kJ/mol) in the second one. + +## References + + - PLUMED manual: + - ABACUS `plumed` / `plumed_file` keywords: + diff --git a/interfaces/PLUMED_interface/example01/INPUT b/interfaces/PLUMED_interface/example01/INPUT new file mode 100644 index 00000000000..527fcdf7eb9 --- /dev/null +++ b/interfaces/PLUMED_interface/example01/INPUT @@ -0,0 +1,48 @@ +INPUT_PARAMETERS +#Parameters (1.General) +suffix ethane_plumed +calculation md +symmetry 0 +gamma_only 1 +precision double +pseudo_dir ../../../tests/PP_ORB +orbital_dir ../../../tests/PP_ORB + +#Parameters (2.Plane waves) +ecutwfc 40 + +#Parameters (3.Basis) +basis_type lcao +ks_solver genelpa + +#Parameters (4.Smearing) +smearing_method gaussian +smearing_sigma 0.01 + +#Parameters (5.Mixing) +mixing_type broyden +mixing_beta 0.8 + +#Parameters (6.Iteration) +scf_nmax 100 +scf_thr 1e-5 + +#Parameters (7.MD) +md_type nvt +md_nstep 200 +md_dt 0.6 +md_tfirst 600 +md_tlast 600 +md_thermostat nhc +md_dumpfreq 100 +md_restartfreq 100 + +#Parameters (8.PLUMED) +# Set plumed to 1 and point plumed_file to the PLUMED input file to enable +# collective variables and biasing potentials (see plumed.dat and README). +plumed 1 +plumed_file plumed.dat + +#Parameters (9.Output) +out_level m +vdw_method d3_0 diff --git a/tests/01_PW/033_PW_CF_CS_S1_smallg/KPT b/interfaces/PLUMED_interface/example01/KPT similarity index 100% rename from tests/01_PW/033_PW_CF_CS_S1_smallg/KPT rename to interfaces/PLUMED_interface/example01/KPT diff --git a/interfaces/PLUMED_interface/example01/STRU b/interfaces/PLUMED_interface/example01/STRU new file mode 100644 index 00000000000..eec042fd694 --- /dev/null +++ b/interfaces/PLUMED_interface/example01/STRU @@ -0,0 +1,34 @@ +ATOMIC_SPECIES +C 12.011000 C_ONCV_PBE-1.0.upf +H 1.008000 H_ONCV_PBE-1.0.upf + +NUMERICAL_ORBITAL +C_gga_8au_100Ry_2s2p1d.orb +H_gga_8au_100Ry_2s1p.orb + +LATTICE_CONSTANT +1.889727 + +LATTICE_VECTORS + 12.0000000000 0.0000000000 0.0000000000 + 0.0000000000 12.0000000000 0.0000000000 + 0.0000000000 0.0000000000 12.0000000000 + +ATOMIC_POSITIONS +Direct + +C +0.000000 +2 + 0.5000000000 0.5000000000 0.4362080000 m 1 1 1 + 0.5000000000 0.5000000000 0.5637920000 m 1 1 1 + +H +0.000000 +6 + 0.5855610000 0.5000000000 0.4059600000 m 1 1 1 + 0.4572200000 0.5740980000 0.4059600000 m 1 1 1 + 0.4572200000 0.4259020000 0.4059600000 m 1 1 1 + 0.5427800000 0.5740980000 0.5940400000 m 1 1 1 + 0.4144390000 0.5000000000 0.5940400000 m 1 1 1 + 0.5427800000 0.4259020000 0.5940400000 m 1 1 1 diff --git a/interfaces/PLUMED_interface/example01/plumed.dat b/interfaces/PLUMED_interface/example01/plumed.dat new file mode 100644 index 00000000000..551a5b18443 --- /dev/null +++ b/interfaces/PLUMED_interface/example01/plumed.dat @@ -0,0 +1,16 @@ +# Ethane (C2H6) H-C-C-H torsion metadynamics for ABACUS + PLUMED. +# The dihedral angle involves the atoms H3-C1-C2-H7 (1-based indices as in +# STRU); the initial structure is staggered (t = pi). +# +# ABACUS runs in atomic units, so the units of the configuration passed to +# PLUMED are declared below with UNITS. +UNITS LENGTH=A TIME=fs + +# Collective variable: the H-C-C-H dihedral angle +t: TORSION ATOMS=3,1,2,7 + +# Well-tempered metadynamics at 600 K (same temperature as the NVT run) +metad: METAD ARG=t PACE=50 HEIGHT=2.0 SIGMA=0.15 FILE=HILLS BIASFACTOR=20 TEMP=600 + +# Write the collective variable and the bias at every MD step +PRINT ARG=t,metad.bias FILE=COLVAR STRIDE=1 diff --git a/interfaces/Wannier90_interface/abacusw90/interface.py b/interfaces/Wannier90_interface/abacusw90/interface.py index 2c4e793c08e..c2d13ee1fcb 100644 --- a/interfaces/Wannier90_interface/abacusw90/interface.py +++ b/interfaces/Wannier90_interface/abacusw90/interface.py @@ -230,7 +230,7 @@ def step0_run_scf( for d in [self.scf_dir, self.scf_dir / "OUT.ABACUS"]: if not d.exists(): continue - for pat in ["SPIN*_CHG.cube", "SPIN*_CHG", "*CHARGE-DENSITY.restart"]: + for pat in ["chg*.cube", "chgs*.cube", "*CHARGE-DENSITY.restart"]: if list(d.glob(pat)): chg_found = True break @@ -283,11 +283,8 @@ def _prepare_scf_files(self): found = set() patterns = [ - "SPIN*_CHG.cube", - "SPIN*_CHG", "chg*.cube", - "CHG*.cube", - "CHG*", + "chgs*.cube", "*CHARGE-DENSITY.restart", ] for pat in patterns: @@ -468,7 +465,7 @@ def run(self, run_scf: bool = True): for d in [self.scf_dir, self.scf_dir / "OUT.ABACUS"]: if not d.exists(): continue - for pat in ["SPIN*_CHG.cube", "SPIN*_CHG", "*CHARGE-DENSITY.restart"]: + for pat in ["chg*.cube", "chgs*.cube", "*CHARGE-DENSITY.restart"]: if list(d.glob(pat)): has_chg = True break diff --git a/python/pyabacus/src/ModuleESolver/py_esolver_lcao.cpp b/python/pyabacus/src/ModuleESolver/py_esolver_lcao.cpp index e95d07a1c3f..cc899b55c20 100644 --- a/python/pyabacus/src/ModuleESolver/py_esolver_lcao.cpp +++ b/python/pyabacus/src/ModuleESolver/py_esolver_lcao.cpp @@ -346,7 +346,7 @@ template class PyHamiltonianAccessor, double>; // ============================================================================ template -void PyDensityMatrixAccessor::set_from_dm(elecstate::DensityMatrix* dm) +void PyDensityMatrixAccessor::set_from_dm(module_dm::DensityMatrix* dm) { dm_ptr_ = dm; @@ -358,9 +358,9 @@ void PyDensityMatrixAccessor::set_from_dm(elecstate::DensityMatrixget_DMK_nks(); - nrow_ = dm->get_DMK_nrow(); - ncol_ = dm->get_DMK_ncol(); + nks_ = dm->get_dmk_nks(); + nrow_ = dm->get_dmk_nrow(); + ncol_ = dm->get_dmk_ncol(); // Initialize pointer arrays for compatibility mode dmk_ptrs_.resize(nks_, nullptr); @@ -385,7 +385,7 @@ void PyDensityMatrixAccessor::set_DMK_data(int ik, const TK* data) } template -py::array_t PyDensityMatrixAccessor::get_DMK(int ik) const +py::array_t PyDensityMatrixAccessor::get_dmk(int ik) const { if (!is_valid() || ik < 0 || ik >= nks_) { @@ -414,7 +414,7 @@ std::vector> PyDensityMatrixAccessor::get_DMK_all() cons std::vector> result; for (int ik = 0; ik < nks_; ++ik) { - result.push_back(get_DMK(ik)); + result.push_back(get_dmk(ik)); } return result; } @@ -797,7 +797,7 @@ void bind_density_matrix_accessor(py::module& m, const std::string& suffix) "Number of rows in density matrix") .def_property_readonly("ncol", &DMAccessor::get_ncol, "Number of columns in density matrix") - .def("get_DMK", &DMAccessor::get_DMK, + .def("get_dmk", &DMAccessor::get_dmk, R"pbdoc( Get DM(k) for specific k-point. diff --git a/python/pyabacus/src/ModuleESolver/py_esolver_lcao.hpp b/python/pyabacus/src/ModuleESolver/py_esolver_lcao.hpp index 61e6b24cbe2..1f6d5ce78b7 100644 --- a/python/pyabacus/src/ModuleESolver/py_esolver_lcao.hpp +++ b/python/pyabacus/src/ModuleESolver/py_esolver_lcao.hpp @@ -27,6 +27,8 @@ class Parallel_Orbitals; namespace elecstate { struct fenergy; class ElecState; +} +namespace module_dm { template class DensityMatrix; } namespace hamilt { @@ -225,7 +227,7 @@ class PyDensityMatrixAccessor PyDensityMatrixAccessor() = default; /// Set from DensityMatrix object - void set_from_dm(elecstate::DensityMatrix* dm); + void set_from_dm(module_dm::DensityMatrix* dm); /// Set dimensions directly (for compatibility) void set_dimensions(int nks, int nrow, int ncol); @@ -234,7 +236,7 @@ class PyDensityMatrixAccessor void set_DMK_data(int ik, const TK* data); /// Get DM(k) for specific k-point - py::array_t get_DMK(int ik) const; + py::array_t get_dmk(int ik) const; /// Get all DM(k) matrices std::vector> get_DMK_all() const; @@ -255,7 +257,7 @@ class PyDensityMatrixAccessor bool is_valid() const { return (dm_ptr_ != nullptr || nks_ > 0); } private: - elecstate::DensityMatrix* dm_ptr_ = nullptr; + module_dm::DensityMatrix* dm_ptr_ = nullptr; int nks_ = 0; int nrow_ = 0; int ncol_ = 0; diff --git a/python/pyabacus/src/hsolver/diago_adapter.hpp b/python/pyabacus/src/hsolver/diago_adapter.hpp index d57c8fd82a5..0493e474a69 100644 --- a/python/pyabacus/src/hsolver/diago_adapter.hpp +++ b/python/pyabacus/src/hsolver/diago_adapter.hpp @@ -86,8 +86,7 @@ class PyDiagoDavidAdapter int max_iter, ::hsolver::diag_comm_info comm_info) { - auto hpsi_func = make_hpsi_func_fstyle(mm_op); - auto spsi_func = make_spsi_func_identity(); + const PyHSOperator op(mm_op); solver_ = std::make_unique( precond_vec.data(), @@ -98,8 +97,7 @@ class PyDiagoDavidAdapter ); return solver_->diag( - hpsi_func, - spsi_func, + op, nbasis_, storage_.psi_ptr(), storage_.eigenvalue_ptr(), @@ -178,8 +176,7 @@ class PyDiagoDavSubspaceAdapter int diag_subspace, int nb2d) { - auto hpsi_func = make_hpsi_func_fstyle(mm_op); - auto spsi_func = make_spsi_func_identity(); + const PyHSOperator op(mm_op); solver_ = std::make_unique( precond_vec, @@ -194,8 +191,7 @@ class PyDiagoDavSubspaceAdapter ); return solver_->diag( - hpsi_func, - spsi_func, + op, storage_.psi_ptr(), nbasis_, storage_.eigenvalue_ptr(), diff --git a/python/pyabacus/src/hsolver/diago_traits.hpp b/python/pyabacus/src/hsolver/diago_traits.hpp index c23d9f3a9f2..c73b6f3bea9 100644 --- a/python/pyabacus/src/hsolver/diago_traits.hpp +++ b/python/pyabacus/src/hsolver/diago_traits.hpp @@ -22,6 +22,7 @@ #include #include "source_hsolver/diago_david.h" +#include "source_hsolver/hs_operator.h" #include "source_hsolver/diago_dav_subspace.h" #include "source_hsolver/diago_cg.h" #include "source_base/module_device/memory_op.h" @@ -384,52 +385,53 @@ struct DiagoCGTraits #endif // __ENABLE_ATEN // ============================================================================ -// Helper Functions for Creating HPsi/SPsi Lambdas +// HSOperator view of a Python callable, plus tensor helpers for the ATen CG path // ============================================================================ /** - * @brief Create hpsi_func lambda for raw pointer interface (F-style) + * @brief The Python matrix-vector callable seen through hsolver::HSOperator. * - * Wraps a Python callable to work with ABACUS raw pointer interface. - * Handles array layout conversion between Python (row-major) and - * ABACUS (column-major for Davidson methods). + * H is applied by the Python callable mm_op on a column-major (ld_psi x nvec) + * array; S is the identity. This is all the Davidson-type solvers need. */ template -auto make_hpsi_func_fstyle( - std::function(py::array_t)> mm_op) +class PyHSOperator : public ::hsolver::HSOperator { - return [mm_op](T* psi_in, T* hpsi_out, const int ld_psi, const int nvec) { +public: + explicit PyHSOperator(std::function(py::array_t)> mm_op) + : mm_op_(std::move(mm_op)) + { + } + + void update_k(const int ik) override + { + } + + void hpsi(const T* x, T* hx, const int ld, const int nvec) const override + { // Create F-style numpy array (column-major) - py::array_t psi({ld_psi, nvec}); + py::array_t psi({ld, nvec}); py::buffer_info buf = psi.request(); T* ptr = static_cast(buf.ptr); - std::copy(psi_in, psi_in + nvec * ld_psi, ptr); + std::copy(x, x + static_cast(nvec) * ld, ptr); // Call Python function - py::array_t hpsi = mm_op(psi); + py::array_t hpsi_arr = mm_op_(psi); // Copy result back - py::buffer_info hpsi_buf = hpsi.request(); - T* hpsi_ptr = static_cast(hpsi_buf.ptr); - std::copy(hpsi_ptr, hpsi_ptr + nvec * ld_psi, hpsi_out); - }; -} + py::buffer_info hpsi_buf = hpsi_arr.request(); + const T* hpsi_ptr = static_cast(hpsi_buf.ptr); + std::copy(hpsi_ptr, hpsi_ptr + static_cast(nvec) * ld, hx); + } -/** - * @brief Create spsi_func lambda for raw pointer interface (identity) - * - * For non-orthogonal basis, S*psi = psi (identity operation). - */ -template -auto make_spsi_func_identity() -{ - using T = typename Traits::T; - using syncmem_op = typename Traits::syncmem_op; + void spsi(const T* x, T* sx, const int ld, const int nvec) const override + { + std::copy(x, x + static_cast(nvec) * ld, sx); + } - return [](const T* psi_in, T* spsi_out, const int nrow, const int nbands) { - syncmem_op()(spsi_out, psi_in, static_cast(nbands * nrow)); - }; -} +private: + std::function(py::array_t)> mm_op_; +}; #ifdef __ENABLE_ATEN /** diff --git a/python/pyabacus/src/pyabacus/cell.py b/python/pyabacus/src/pyabacus/cell.py index 3b22fde5689..a9c7dde84db 100644 --- a/python/pyabacus/src/pyabacus/cell.py +++ b/python/pyabacus/src/pyabacus/cell.py @@ -209,24 +209,24 @@ def _save_stru(self, file_path: str) -> None: for symbol, pos, props in self._atoms: if symbol not in species_dict: pp_info = self.pseudo_potentials.get(symbol, {}) - - # Ensure we have a clean filename (remove './' if present) - pp_file = pp_info.get('pseudo_file', f"{symbol}.UPF") - if pp_file.startswith('./'): - pp_file = pp_file[2:] - - species_dict[symbol] = { - 'symbol': symbol, - 'mass': pp_info.get('mass', 1.0), - 'pp_file': pp_file, # Store clean filename - 'natom': 1, - 'mag_each': props.get('mag', 0.0), - 'atom': [] - } - if 'pp_type' in pp_info: - species_dict[symbol]['pp_type'] = pp_info['pp_type'] - else: - species_dict[symbol]['natom'] += 1 + + # Ensure we have a clean filename (remove './' if present) + pp_file = pp_info.get('pseudo_file', f"{symbol}.UPF") + if pp_file.startswith('./'): + pp_file = pp_file[2:] + + species_dict[symbol] = { + 'symbol': symbol, + 'mass': pp_info.get('mass', 1.0), + 'pp_file': pp_file, # Store clean filename + 'natom': 0, + 'mag_each': props.get('mag', 0.0), + 'atom': [] + } + if 'pp_type' in pp_info: + species_dict[symbol]['pp_type'] = pp_info['pp_type'] + + species_dict[symbol]['natom'] += 1 # Convert coordinates if needed coord = pos diff --git a/python/pyabacus/tests/test_cell.py b/python/pyabacus/tests/test_cell.py index a664af4fa35..a2ae950e576 100644 --- a/python/pyabacus/tests/test_cell.py +++ b/python/pyabacus/tests/test_cell.py @@ -186,6 +186,27 @@ def test_add_atom(self): cell.add_atom("Fe", [1, 1, 1]) self.assertFalse(cell._built) + def test_save_stru_repeated_species(self): + """Regression test: saving a cell with multiple atoms of the same species + must keep all atoms and report a consistent natom.""" + # Start from an existing STRU so lattice_constant and pp info exist + cell = Cell.from_file(self.stru_file) + n_orig = len(cell.atoms) + # Add two more atoms of an already-present species + existing_species = cell.species[0] + cell.add_atom(existing_species, [0.10, 0.10, 0.10]) + cell.add_atom(existing_species, [0.20, 0.20, 0.20]) + + out = os.path.join(self.temp_dir, 'STRU') + cell.to_file(out, 'stru') + + reloaded = Cell.from_file(out) + self.assertEqual(len(reloaded.atoms), n_orig + 2) + self.assertEqual( + reloaded.species.count(existing_species), + cell.species.count(existing_species), + ) + def test_k_points(self): """Test k-points generation.""" cell = Cell() diff --git a/python/pyabacus/tests/test_m_nao.py b/python/pyabacus/tests/test_m_nao.py index 9a13d1b3fd9..26bfc7b5a21 100644 --- a/python/pyabacus/tests/test_m_nao.py +++ b/python/pyabacus/tests/test_m_nao.py @@ -28,9 +28,12 @@ def test_nr(): assert chi.nr == sz assert chi.nk == 0 assert chi.rmax == grid[sz-1] + # Each property returns a copy of the entire radial array. + rgrid = chi.rgrid + rvalue = chi.rvalue for i in range(sz): - assert(chi.rgrid[i] == grid[i]) - assert(chi.rvalue[i] == f[i]) + assert(rgrid[i] == grid[i]) + assert(rvalue[i] == f[i]) assert chi.pr == pr assert chi.pk == 0 assert chi.is_fft_compliant == False diff --git a/source/CMakeLists.txt b/source/CMakeLists.txt index ffd4272f71f..05ee4118ee5 100644 --- a/source/CMakeLists.txt +++ b/source/CMakeLists.txt @@ -31,7 +31,7 @@ include(CompilerConfiguration) set(_abacus_feature_definitions __FFTW3 - $<$:__RAPIDJSON> + $<$:__JSON> $<$:__CUDA_ON_DCU> $<$:__CUDA_MPI> $<$:__ABACUS_LIBM> @@ -65,7 +65,6 @@ set(_abacus_feature_definitions $<$:EXX_H_COMM=2> $<$:TEST_EXX_LCAO=0> $<$:TEST_EXX_RADIAL=1> - $<$:__EXX_DEV> $<$:__LIBXC> $<$:__DFTD4> $<$:GATHER_INFO>) @@ -148,6 +147,20 @@ target_link_libraries(abacus_link_libs INTERFACE set(_abacus_linalg_libs) set(_abacus_linalg_include_dirs) +if(USE_DSP) + if(NOT MT_CPACK_DIR) + message(FATAL_ERROR + "USE_DSP requires MT_CPACK_DIR to be set to the DSP C_PACK install prefix " + "(the directory containing lib/libc_pack.a).") + endif() + set(_abacus_cpack_library "${MT_CPACK_DIR}/lib/libc_pack.a") + if(NOT EXISTS "${_abacus_cpack_library}") + message(FATAL_ERROR + "MT_CPACK_DIR is set to '${MT_CPACK_DIR}' but '${_abacus_cpack_library}' does not exist.") + endif() + list(APPEND _abacus_linalg_libs ${_abacus_cpack_library}) +endif() + if(USE_KML) if(ENABLE_MPI) list(APPEND _abacus_linalg_libs KML::ScaLAPACK) @@ -211,6 +224,16 @@ if(NOT MSVC) list(APPEND _abacus_linalg_libs m) endif() +# The mtblas archives bundle their own OpenBLAS objects. They must come after +# the standalone BLAS/LAPACK/ScaLAPACK above so that the linker resolves the +# BLAS symbols from the standalone BLAS instead of the bundled copy. +if(USE_DSP) + list(APPEND _abacus_linalg_libs + ${MTBLAS_FFT_DIR}/libmtblas/lib/libmtblas.a + ${MTBLAS_FFT_DIR}/libmtblas/lib/libmtblasdev.a + ${MTBLAS_FFT_DIR}/libmtblas/lib/libmtfft.a) +endif() + target_link_libraries(abacus_linalg_libs INTERFACE ${_abacus_linalg_libs}) target_include_directories(abacus_linalg_libs INTERFACE ${_abacus_linalg_include_dirs}) @@ -240,13 +263,14 @@ endif() # ------------------------------------------------------------------------------ set(_abacus_feature_libs - $<$:RapidJSON> + $<$:nlohmann_json::nlohmann_json> $<$:ELPA::ELPA> $<$:PEXSI::PEXSI> $<$:cnpy> $<$:ZLIB::ZLIB> $<$:cereal::cereal> - $<$:Libxc::xc>) + $<$:Libxc::xc> + $<$:PLUMED::plumed>) set(_abacus_feature_include_dirs) set(_abacus_feature_compile_options) @@ -364,7 +388,7 @@ target_compile_options(abacus_compile_requirements INTERFACE ${_abacus_feature_compile_options}) foreach(_abacus_feature_target IN ITEMS - RapidJSON + nlohmann_json::nlohmann_json cereal::cereal ELPA::ELPA Libxc::xc @@ -517,7 +541,6 @@ set(ABACUS_BIN_PATH ${PROJECT_BINARY_DIR}/${ABACUS_BIN_NAME}) if(USE_CUDA) set_property(TARGET ${ABACUS_BIN_NAME} PROPERTY CUDA_ARCHITECTURES ${CMAKE_CUDA_ARCHITECTURES}) - target_compile_definitions(${ABACUS_BIN_NAME} PRIVATE __USE_NVTX) endif() if(ENABLE_COVERAGE) @@ -553,7 +576,29 @@ add_library( source_main/driver.cpp source_main/driver_run.cpp) +# __USE_NVTX gates the NVTX code in source_base/timer.cpp (target base) and +# source_main/driver.cpp (target driver), so it must be defined on those +# OBJECT libraries, not only on the final executable (whose sole TU, main.cpp, +# contains no NVTX code). Defining it only on the executable silently dropped +# every NVTX range and kept timer_enable_nvtx ineffective. +if(USE_CUDA) + target_compile_definitions(base PRIVATE __USE_NVTX) + target_compile_definitions(driver PRIVATE __USE_NVTX) + # NVTX is header-only since CUDA 12.9; older toolkits need libnvToolsExt. + if(CUDAToolkit_VERSION VERSION_LESS 12.9) + # timer.cpp (in target base) references nvtxRangePushA/nvtxRangePop, so + # every consumer of base's object files needs libnvToolsExt on its link + # line. Attach it to base itself so the dependency propagates through + # base's link interface to the main executable and to every unit test + # that links base; linking it only to the executable leaves those tests + # with undefined NVTX references (e.g. MODULE_CELL_SYMMETRY_analysis on + # CUDA 12.2 CI). + target_link_libraries(base INTERFACE CUDA::nvToolsExt) + endif() +endif() + list(APPEND device_srcs + source_pw/module_pwdft/kernels/projector_gradient_op.cpp source_pw/module_pwdft/kernels/nonlocal_op.cpp source_pw/module_pwdft/kernels/veff_op.cpp source_pw/module_pwdft/kernels/ekinetic_op.cpp @@ -562,6 +607,7 @@ list(APPEND device_srcs source_basis/module_pw/kernels/pw_op.cpp source_hsolver/kernels/hegvd_op.cpp source_hsolver/kernels/bpcg_kernel_op.cpp + source_hsolver/kernels/linear_op.cpp source_estate/kernels/elecstate_op.cpp source_base/module_device/device.cpp @@ -585,6 +631,7 @@ list(APPEND device_srcs if(USE_CUDA) list(APPEND device_srcs + source_pw/module_pwdft/kernels/cuda/projector_gradient_op.cu source_pw/module_pwdft/kernels/cuda/nonlocal_op.cu source_pw/module_pwdft/kernels/cuda/veff_op.cu source_pw/module_pwdft/kernels/cuda/ekinetic_op.cu @@ -594,6 +641,7 @@ if(USE_CUDA) source_basis/module_pw/kernels/cuda/pw_op.cu source_hsolver/kernels/cuda/hegvd_op.cu source_hsolver/kernels/cuda/bpcg_kernel_op.cu + source_hsolver/kernels/cuda/linear_op.cu source_estate/kernels/cuda/elecstate_op.cu source_base/module_device/cuda/memory_op.cu source_pw/module_pwdft/kernels/cuda/force_op.cu @@ -609,11 +657,19 @@ if(USE_CUDA) source_pw/module_pwdft/kernels/cuda/cal_density_real_op.cu source_pw/module_pwdft/kernels/cuda/mul_potential_op.cu source_pw/module_pwdft/kernels/cuda/vec_mul_vec_complex.cu - source_pw/module_pwdft/kernels/cuda/exx_cal_energy_op.cu) + source_pw/module_pwdft/kernels/cuda/exx_cal_energy_op.cu + source_pw/module_pwdft/kernels/cuda/exx_batch_op.cu) + if(USE_CUDA AND ENABLE_GPU_NEIGHBOR_LIST) + list(APPEND device_srcs + source_cell/kernels/cuda/gpu_neighlist_kernels.cu + source_cell/kernels/cuda/gpu_neighlist_host.cu + source_cell/kernels/cuda/gpu_neighlist_device.cu) + endif() endif() if(USE_ROCM) hip_add_library(device_rocm STATIC + source_pw/module_pwdft/kernels/rocm/projector_gradient_op.hip.cu source_pw/module_pwdft/kernels/rocm/nonlocal_op.hip.cu source_pw/module_pwdft/kernels/rocm/veff_op.hip.cu source_pw/module_pwdft/kernels/rocm/ekinetic_op.hip.cu @@ -623,6 +679,7 @@ if(USE_ROCM) source_basis/module_pw/kernels/rocm/pw_op.hip.cu source_hsolver/kernels/rocm/hegvd_op.hip.cu source_hsolver/kernels/rocm/bpcg_kernel_op.hip.cu + source_hsolver/kernels/rocm/linear_op.hip.cu source_estate/kernels/rocm/elecstate_op.hip.cu source_base/module_device/rocm/memory_op.hip.cu source_pw/module_pwdft/kernels/rocm/force_op.hip.cu @@ -675,6 +732,7 @@ target_link_libraries( hamilt_general elecstate module_pwdft + module_proj module_ofdft module_stodft module_dfpt @@ -686,8 +744,10 @@ target_link_libraries( io_input surchem neighbor_search + domain_decomposition neighbor md + module_plumed planewave symmetry cell @@ -734,7 +794,7 @@ endif() if(ENABLE_LIBRI) target_link_libraries(${ABACUS_BIN_NAME} PRIVATE ri module_exx_symmetry) endif() -if(ENABLE_RAPIDJSON) +if(ENABLE_JSON) target_link_libraries(${ABACUS_BIN_NAME} PRIVATE json_output) endif() diff --git a/source/Makefile b/source/Makefile index 1f29676b210..92dfe1aa8c9 100644 --- a/source/Makefile +++ b/source/Makefile @@ -199,6 +199,15 @@ ifdef PEXSI_DIR HONG += -D__PEXSI endif +ifdef PLUMED_DIR + #========================== + # PLUMED (https://www.plumed.org) + #========================== + INCLUDES += -I${PLUMED_DIR}/include + LIBS += -L${PLUMED_DIR}/lib -Wl,--no-as-needed -lplumed -lplumedKernel -Wl,--as-needed -Wl,-rpath=${PLUMED_DIR}/lib + HONG += -DUSE_PLUMED +endif + include $(ABACUS_ROOT)Makefile.Objects VPATH := $(patsubst ./%,$(ABACUS_ROOT)%,$(VPATH)) HEADERS := $(patsubst %,$(ABACUS_ROOT)%,$(HEADERS)) @@ -217,7 +226,7 @@ HONG += -D__SELINV -DMETIS ifeq ($(MPI), ON) ifdef LIBRI_DIR - OBJS_ABACUS += $(OBJS_MODULE_RI) ${OBJS_SRC_RI} + OBJS_ABACUS += $(OBJS_MODULE_RI) $(OBJS_BSE) ${OBJS_SRC_RI} endif endif @@ -259,14 +268,6 @@ ${OBJ_DIR}/parse_args.o: $(ABACUS_ROOT)source_io/parse_args.cpp $(BUILD_INFO_DIR @mkdir -p $(dir $@) ${CXX} ${OPTS} ${OPTS_MPI} -c ${HONG} $< -o $@ -# Explicit rule: source_cell/module_symmetry and source_estate/module_charge -# both contain a symm_rho.cpp. VPATH resolves symm_rho.o to the source_cell -# one (listed first), so the Symmetry_rho implementation needs an explicit -# object name to be compiled from the correct source. -${OBJ_DIR}/symm_rho_charge.o: $(ABACUS_ROOT)source_estate/module_charge/symm_rho.cpp - @mkdir -p $(dir $@) - ${CXX} ${OPTS} ${OPTS_MPI} -c ${HONG} $< -o $@ - ###### END of ABACUS INFO PART ###### #========================== diff --git a/source/Makefile.Objects b/source/Makefile.Objects index 5f47f240cfe..f34c7ca601f 100644 --- a/source/Makefile.Objects +++ b/source/Makefile.Objects @@ -31,6 +31,7 @@ VPATH=./src_global:\ ./source_base/module_mixing:\ ./source_base/module_out:\ ./source_md:\ +./source_md/module_plumed:\ ./source_basis/module_pw:\ ./source_base/module_fft:\ ./source_esolver:\ @@ -50,6 +51,7 @@ VPATH=./src_global:\ ./source_hamilt/module_surchem:\ ./source_hamilt/module_xc:\ ./source_hamilt/module_xc/kernels:\ +./source_pw/module_proj:\ ./source_pw/module_pwdft:\ ./source_pw/module_ofdft:\ ./source_pw/module_stodft:\ @@ -70,9 +72,11 @@ VPATH=./src_global:\ ./source_hamilt/module_vdw:\ ./source_io:\ ./source_io/module_json:\ +./source_io/module_hs:\ ./src_ri:\ ./source_lcao/module_ri:\ ./source_lcao/module_ri/module_exx_symmetry:\ +./source_lcao/module_bse:\ ./source_io/module_parameter:\ ./source_io/module_efield:\ ./source_lcao/module_lr:\ @@ -175,6 +179,7 @@ OBJS_BASE=assoc_laguerre.o\ ylm.o\ opt_cg.o\ opt_dcsrch.o\ + opt_tn.o\ cubic_spline.o\ sph_bessel_tf.o\ mixing_data.o\ @@ -216,6 +221,7 @@ OBJS_CELL=atom_pseudo.o\ reciprocal_grid.o\ cell_index.o\ cell_tools.o\ + record_adj.o\ check_atomic_stru.o\ update_cell.o\ bcast_cell.o\ @@ -229,6 +235,7 @@ OBJS_CELL=atom_pseudo.o\ cal_ux.o\ mdcell_reader.o\ mdcell.o\ + domain_decomposition.o\ cif_io.o\ ucell_io.o\ @@ -269,6 +276,7 @@ OBJS_ELECSTAT=elecstate.o\ gatefield.o\ potential_new.o\ potential_types.o\ + pot_surchem.o\ pot_sep.o\ pot_local.o\ h_hartree_pw.o\ @@ -285,19 +293,29 @@ OBJS_ELECSTAT=elecstate.o\ setup_estate_pw.o\ update_pot.o\ occ_matrix.o\ - occ_mixer.o + occ_mixer.o\ + occ_comput.o OBJS_ELECSTAT_LCAO=elecstate_lcao.o\ - init_dm.o\ + dm_routine.o\ density_matrix.o\ - density_matrix_io.o\ - cal_dm_psi.o\ - cal_edm_tddft.o\ + dmr_init.o\ + dm_setter.o\ + dm_getter.o\ + dmr_gamma.o\ + dmr_k.o\ + dmr_td.o\ + dmr_full.o\ + dm_tools.o\ + dm_from_psi.o\ + edm_tddft.o\ + edm_tddft_lapack.o\ OBJS_ESOLVER=esolver_factory.o\ esolver_ks.o\ esolver_fp.o\ esolver_ks_pw.o\ + esolver_ks_pw_tddft.o\ esolver_ks_lcaopw.o\ esolver_sdft_pw.o\ esolver_lj.o\ @@ -330,6 +348,7 @@ OBJS_GINT=batch_biggrid.o\ gint_fvl.o\ gint_info.o\ gint_interface.o\ + gint_prec_ctrl.o\ gint_rho.o\ gint_drho.o\ gint_tau.o\ @@ -344,19 +363,27 @@ OBJS_GINT=batch_biggrid.o\ unitcell_info.o\ OBJS_HAMILT=hamilt_pw.o\ + td_history_pw.o\ + td_pw.o\ hs_matrix_k.o\ - hamilt_sdft_pw.o\ + sto_hamilt_pw.o\ operator.o\ op_pw.o\ op_pw_exx.o\ op_pw_ekin.o\ + op_pw_ekin_td.o\ ekinetic_op.o\ op_pw_exx_ace.o\ op_pw_exx_pot.o\ hpsi_norm_op.o\ op_pw_veff.o\ veff_op.o\ + projector_gradient.o\ + velocity_workspace.o\ + nonlocal_workspace.o\ op_pw_nl.o\ + op_pw_nl_td.o\ + projector_gradient_op.o\ nonlocal_op.o\ op_pw_meta.o\ meta_op.o\ @@ -368,6 +395,8 @@ OBJS_HAMILT=hamilt_pw.o\ exx_cal_energy_op.o\ cal_density_real_op.o\ mul_potential_op.o\ + exx_batch_op.o\ + exx_batch_op_float_stub.o\ OBJS_HAMILT_OF=kedf_tf.o\ kedf_extwt.o\ @@ -399,6 +428,7 @@ OBJS_DFPT=dfpt_hamilt_shift.o\ dfpt_stern.o OBJS_HAMILT_LCAO=hamilt_lcao.o\ + hamilt_lcao_factory.o\ operator_lcao.o\ ekinetic.o\ ekinetic_fs.o\ @@ -440,14 +470,25 @@ OBJS_HSOLVER=diago_cg.o\ para_lin_tf.o\ hsolver.o\ hsolver_pw.o\ + hsolver_pw_tddft.o\ + orthonormal.o\ + orthonormal_small.o\ hsolver_lcaopw.o\ - hsolver_pw_sdft.o\ diago_iter_assist.o\ hegvd_op.o\ bpcg_kernel_op.o\ diag_const_nums.o\ diag_hs_para.o\ diago_pxxxgvx.o\ + linear_bicgstab.o\ + linear_cgs.o\ + linear_gmres.o\ + linear_algebra.o\ + cn_subspace.o\ + linear_low_rank.o\ + linear_workspace.o\ + hsolver_linear.o\ + linear_op.o\ OBJS_HSOLVER_LCAO=hsolver_lcao.o\ diago_scalapack.o\ @@ -474,6 +515,7 @@ OBJS_MD=fire.o\ md_func.o\ msst.o\ nhchain.o\ + plumed_interface.o\ run_md.o\ verlet.o\ @@ -484,7 +526,6 @@ OBJS_NEIGHBOR=sltk_atom.o\ OBJS_NEIGHBOR_SEARCH=neighbor_search.o\ bin_manager.o\ - domain_decomposition.o\ page_allocator.o\ @@ -545,6 +586,7 @@ OBJS_RELAXATION=relax_data.o\ cg_base.o\ bfgs_basic.o\ relax_driver.o\ + relax_stru_io.o\ ions_move_basic.o\ ions_move_bfgs.o\ ions_move_bfgs2.o\ @@ -587,8 +629,13 @@ OBJS_SYMMETRY=symm_other.o\ symm_rho.o\ little_group.o\ symmetry.o\ + irreducible_sector.o\ + irred_sec_bvk.o\ + symm_rotation_k.o\ OBJS_XC=xc_functional.o\ + xc_functional_ncgga_sf.o\ + xc_ncgga_radial.o\ xc_functional_op.o\ xc_pot.o\ xc_grad.o\ @@ -620,11 +667,11 @@ OBJS_IO=module_parameter/input_conv.o\ module_bessel/bessel_basis.o\ module_output/cal_test.o\ module_dos/write_dos_pw.o\ - module_energy/nscf_fermi_surf.o\ module_energy/write_bands.o\ module_dos/cal_dos.o\ module_dos/cal_pdos_gamma.o\ module_dos/cal_pdos_multik.o\ + module_dos/write_pdos_text.o\ module_dos/cal_ldos.o\ module_bessel/numerical_descriptor.o\ module_bessel/numerical_basis.o\ @@ -660,6 +707,7 @@ OBJS_IO=module_parameter/input_conv.o\ module_dipole/write_dipole.o\ module_current/td_current_io.o\ module_current/td_current_io_comm.o\ + module_current/td_current_pw.o\ td_efield_io.o\ td_vector_pot_io.o\ module_chgpot/write_libxc_r.o\ @@ -676,7 +724,6 @@ OBJS_IO=module_parameter/input_conv.o\ abacusjson.o\ general_info.o\ init_info.o\ - readin_info.o\ output_info.o\ parse_args.o\ module_parameter/availability.o\ @@ -698,12 +745,18 @@ OBJS_IO=module_parameter/input_conv.o\ module_parameter/read_inp_other.o\ module_parameter/read_inp_out.o\ module_parameter/read_set_globalv.o\ - module_hs/cal_plpr.o\ + module_hs/angmom_op_mat.o\ -OBJS_IO_LCAO=module_hs/cal_r_overlap_r.o\ +OBJS_IO_LCAO=module_hs/pos_op_mat.o\ + module_hs/pos_op_basis.o\ + module_hs/pos_op_calc.o\ + module_hs/pos_op_writer.o\ + module_hs/vxc_op_mat.o\ + module_hs/vxc_op_tools.o\ module_output/write_orb_info.o\ module_dos/write_dos_lcao.o\ module_energy/write_proj_band_lcao.o\ + module_energy/write_eband_terms.o\ module_energy/write_eig_occ.o\ module_chgpot/get_pchg_lcao.o\ module_wf/get_wf_lcao.o\ @@ -711,9 +764,13 @@ OBJS_IO_LCAO=module_hs/cal_r_overlap_r.o\ module_unk/unk_overlap_lcao.o\ module_wf/read_wfc_nao.o\ module_wf/write_wfc_nao.o\ - module_hs/write_hs_sparse.o\ - module_hs/single_r_io.o\ - module_hs/write_hs_r.o\ + module_hs/hs_dense_io.o\ + module_hs/hs_sparse_io.o\ + module_hs/dhs_sparse_writer.o\ + module_hs/lat_r_csr.o\ + module_hs/hsk_writer.o\ + module_hs/hsr_writer.o\ + module_hs/hsr_legacy.o\ module_dm/write_dmr.o\ module_mulliken/output_mulliken.o\ module_mulliken/output_sk.o\ @@ -721,8 +778,10 @@ OBJS_IO_LCAO=module_hs/cal_r_overlap_r.o\ module_ml/io_npz.o\ module_dhs/write_dh.o\ module_dhs/write_dh_terms.o\ - module_hs/write_h_terms.o\ - module_hs/rr_sparse_writer.o\ + module_hs/hterm_writer.o\ + module_hs/pos_op_csr.o\ + module_hs/vxc_op_r.o\ + module_hs/vxc_op_lip.o\ OBJS_LCAO=evolve_elec.o\ force_rt_overlap.o\ @@ -746,8 +805,8 @@ OBJS_LCAO=evolve_elec.o\ boundary_fix.o\ upsi.o\ force_stress_lcao.o\ - force_lcao_gamma.o\ - force_lcao_k.o\ + force_stress_assemble.o\ + force_stress_terms.o\ stress_tools.o\ edm.o\ pulay_fs_center2.o\ @@ -765,17 +824,16 @@ OBJS_LCAO=evolve_elec.o\ lcao_init_basis.o\ setup_exx.o\ setup_deepks.o\ - setup_dm.o\ + allocate_dm.o\ rho_tau_lcao.o\ center2orb.o\ center2orb_orb11.o\ center2orb_orb21.o\ center2orb_orb22.o\ - record_adj.o\ OBJS_MODULE_RI=conv_coulomb_pot_k.o\ - exx_abfs-abfs_index.o\ exx_abfs_jle.o\ + module_restart/restart_exx_csr.o\ exx_abfs_io.o\ exx_abfs_ctor_orbs.o\ abfs_construct_pca.o\ @@ -789,8 +847,16 @@ OBJS_MODULE_RI=conv_coulomb_pot_k.o\ symm_rotation.o\ symm_rot_out.o\ gaussian_abfs.o\ + exx_lri_detail.o\ + ri_util.o\ singular_value.o\ +OBJS_BSE=esolver_lr_lcao_bse.o\ + hamilt_bse.o\ + hamilt_bse_solver.o\ + bse_util.o\ + utils/lr_io_krlist.o + OBJS_PARALLEL=parallel_common.o\ parallel_cell.o\ parallel_global.o\ @@ -821,20 +887,27 @@ OBJS_SRCPW=h_ewald_pw.o\ vnl_pw_getvnl.o\ vnl_pw_grad.o\ vnl_pw_init_vnl.o\ + vnl_pw_table.o\ vnl_pw_qrad.o\ - chgmixing.o\ - gint_prec_ctrl.o\ + chg_routine.o\ charge.o\ - charge_init.o\ - charge_mpi.o\ - charge_extra.o\ - charge_mixing.o\ - charge_mixing_dmr.o\ - mix_resid.o\ - mix_precond.o\ - charge_mixing_rho.o\ - charge_mixing_uspp.o\ - rhog_io.o\ + chg_atomic.o\ + chg_atomic_inner.o\ + chg_init.o\ + chg_tools.o\ + chg_parallel.o\ + chg_extra.o\ + chg_mix.o\ + chg_mix_rho.o\ + chg_dmr.o\ + chg_drho.o\ + chg_drho_inner.o\ + chg_precond.o\ + chg_tau.o\ + chg_symm.o\ + chg_symm_detail.o\ + chg_uspp.o\ + chg_rhog_io.o\ write_elecstat_pot.o\ write_init.o\ fp_energy.o\ @@ -861,6 +934,7 @@ OBJS_SRCPW=h_ewald_pw.o\ fs_nonlocal_tools_stress.o\ fs_nonlocal_tools_vkb.o\ fs_kin_tools.o\ + nonlocal_maths.o\ force_op.o\ stress_op.o\ onsite_op.o\ @@ -871,6 +945,7 @@ OBJS_SRCPW=h_ewald_pw.o\ stru_fac.o\ stru_fac_k.o\ soc.o\ + sto_hsolver_pw.o\ sto_iter.o\ sto_che.o\ sto_wf.o\ @@ -891,8 +966,6 @@ OBJS_SRCPW=h_ewald_pw.o\ stress_pw.o\ of_stress_pw.o\ of_print_info.o\ - symm_rho_charge.o\ - symm_rhog.o\ setup_psi_pw.o\ setup_psi.o\ psi_prepare.o\ @@ -924,6 +997,7 @@ OBJS_VDW=vdw.o\ OBJS_DFTU=dftu_nao_fs_k.o\ dftu_nao_for_r.o\ dftu_nao_fs_r.o\ + dftu_nao_fs_reduce.o\ dftu_nao_str_r.o\ dftu_nao_folding.o\ dftu_nao_pots.o\ diff --git a/source/Makefile.vars b/source/Makefile.vars index 564246f8006..656337ca69e 100644 --- a/source/Makefile.vars +++ b/source/Makefile.vars @@ -60,6 +60,7 @@ CEREAL_DIR = /usr/local/include/cereal ## To use DeePMD: set DeePMD_DIR LIBTORCH_DIR and TensorFlow_DIR ## To use LibRI: set LIBRI_DIR and LIBCOMM_DIR ## To use PEXSI: set PEXSI_DIR DSUPERLU_DIR and PARMETIS_DIR +## To use PLUMED: set PLUMED_DIR ##--------------------------------------------------------------------- # LIBTORCH_DIR = /usr/local @@ -78,6 +79,8 @@ ENABLE_MLALGO ?= OFF # DSUPERLU_DIR = /public/software/superlu_dist # PARMETIS_DIR = /public/software/parmetis +# PLUMED_DIR = /public/software/plumed + ##--------------------------------------------------------------------- # NP = 14 # It is not supported. use make -j14 or make -j to parallelly compile # DEBUG = OFF diff --git a/source/source_base/CMakeLists.txt b/source/source_base/CMakeLists.txt index 2b4f67bd95a..4d305ad41d3 100644 --- a/source/source_base/CMakeLists.txt +++ b/source/source_base/CMakeLists.txt @@ -40,6 +40,7 @@ add_library( mymath.cpp opt_cg.cpp opt_dcsrch.cpp + opt_tn.cpp output.cpp para_gemm.cpp realarray.cpp @@ -89,10 +90,6 @@ add_library( ) target_link_libraries(base PUBLIC container) -if (USE_DSP) - target_link_libraries(base PUBLIC ${MTBLAS_FFT_DIR}/libmtblas/lib/libmtblas.a) - target_link_libraries(base PUBLIC ${MTBLAS_FFT_DIR}/libmtblas/lib/libmtblasdev.a) -endif() add_subdirectory(module_container) if(ENABLE_COVERAGE) diff --git a/source/source_base/complexarray.h b/source/source_base/complexarray.h index 831f42ee02d..26ae234680c 100644 --- a/source/source_base/complexarray.h +++ b/source/source_base/complexarray.h @@ -14,94 +14,94 @@ namespace ModuleBase class ComplexArray { public: - std::complex *ptr=nullptr; // data array - - ComplexArray(const int bnd1=0, const int bnd2=1, const int bnd3=1, const int bnd4=1); - - ~ComplexArray(); - - void freemem(); - - void create(const int bnd1=0, const int bnd2=1, const int bnd3=1, const int bnd4=1); - - ComplexArray(const ComplexArray &cd); - ComplexArray(ComplexArray &&cd); - - /**************************************************** - * OPERATOR FUNCTIONS - ***************************************************/ - ComplexArray& operator=(ComplexArray &&cd); - ComplexArray &operator=(const ComplexArray &cd); - /// Assignment of scalar: all entries set to c. - void operator=(std::complex c); - /// Add two ComplexArray - ComplexArray operator+(const ComplexArray &cd) const; - /// Accumulate sum of ComplexArray - void operator+=(const ComplexArray &cd); - /// Subtract two ComplexArray - ComplexArray operator-(const ComplexArray &cd) const; - /// Accumulate difference of arrays - void operator-=(const ComplexArray &cd); - /// Scale a ComplexArray by real r - ComplexArray operator*(const double r) const; - /// Scale a ComplexArray by a std::complex number c - ComplexArray operator*(const std::complex c) const; - /// Scale a ComplexArray by real number in place - void operator*=(const double r); - /// Scale a ComplexArray by std::complex c in place - void operator*=(const std::complex c); - /// accumulate pointwise multiply - void operator*=(const ComplexArray &cd); - /// Judge if two ComplexArray is equal - bool operator== (const ComplexArray &cd2)const; - /// Judge if two ComplexArray is not equal - bool operator!= (const ComplexArray &cd2)const; - - /// overloaded subscript operator for non-const std::complex Array const reference return creates an lvakue - std::complex &operator() - (const int ind1=0, const int ind2=0, const int ind3=0, const int ind4=0) - { - assert(ind1>=0); assert(ind1=0); assert(ind2=0); assert(ind3=0); assert(ind4 &operator()(int, int, int, int, int); - /// overloaded subscript operator for const std::complex Array const reference return creates an cvakue - const std::complex &operator() - (const int ind1=0, const int ind2=0, const int ind3=0, const int ind4=0) const - { - assert(ind1>=0); assert(ind1=0); assert(ind2=0); assert(ind3=0); assert(ind4 &operator()(int, int, int, int, int)const; - - /**************************************************** - * MEMBER FUNCTIONS - ***************************************************/ - /// set all elements to be {0.0,0.0} - void zero_out(void); - - /// Negates all the entries in the array - void negate(void); - - /// set all elements to a random number whose real/image is between [-0.5,0.5). - void randomize(void); - int getBound1()const{ return bound1; } - int getBound2()const{ return bound2; } - int getBound3()const{ return bound3; } - int getBound4()const{ return bound4; } - int getSize()const{ return bound1*bound2*bound3*bound4; } + std::complex *ptr=nullptr; // data array + + ComplexArray(const int bnd1=0, const int bnd2=1, const int bnd3=1, const int bnd4=1); + + ~ComplexArray(); + + void freemem(); + + void create(const int bnd1=0, const int bnd2=1, const int bnd3=1, const int bnd4=1); + + ComplexArray(const ComplexArray &cd); + ComplexArray(ComplexArray &&cd); + + /**************************************************** + * OPERATOR FUNCTIONS + ***************************************************/ + ComplexArray& operator=(ComplexArray &&cd); + ComplexArray &operator=(const ComplexArray &cd); + /// Assignment of scalar: all entries set to c. + void operator=(std::complex c); + /// Add two ComplexArray + ComplexArray operator+(const ComplexArray &cd) const; + /// Accumulate sum of ComplexArray + void operator+=(const ComplexArray &cd); + /// Subtract two ComplexArray + ComplexArray operator-(const ComplexArray &cd) const; + /// Accumulate difference of arrays + void operator-=(const ComplexArray &cd); + /// Scale a ComplexArray by real r + ComplexArray operator*(const double r) const; + /// Scale a ComplexArray by a std::complex number c + ComplexArray operator*(const std::complex c) const; + /// Scale a ComplexArray by real number in place + void operator*=(const double r); + /// Scale a ComplexArray by std::complex c in place + void operator*=(const std::complex c); + /// accumulate pointwise multiply + void operator*=(const ComplexArray &cd); + /// Judge if two ComplexArray is equal + bool operator== (const ComplexArray &cd2)const; + /// Judge if two ComplexArray is not equal + bool operator!= (const ComplexArray &cd2)const; + + /// overloaded subscript operator for non-const std::complex Array const reference return creates an lvakue + std::complex &operator() + (const int ind1=0, const int ind2=0, const int ind3=0, const int ind4=0) + { + assert(ind1>=0); assert(ind1=0); assert(ind2=0); assert(ind3=0); assert(ind4 &operator()(int, int, int, int, int); + /// overloaded subscript operator for const std::complex Array const reference return creates an cvakue + const std::complex &operator() + (const int ind1=0, const int ind2=0, const int ind3=0, const int ind4=0) const + { + assert(ind1>=0); assert(ind1=0); assert(ind2=0); assert(ind3=0); assert(ind4 &operator()(int, int, int, int, int)const; + + /**************************************************** + * MEMBER FUNCTIONS + ***************************************************/ + /// set all elements to be {0.0,0.0} + void zero_out(void); + + /// Negates all the entries in the array + void negate(void); + + /// set all elements to a random number whose real/image is between [-0.5,0.5). + void randomize(void); + int getBound1()const{ return bound1; } + int getBound2()const{ return bound2; } + int getBound3()const{ return bound3; } + int getBound4()const{ return bound4; } + int getSize()const{ return bound1*bound2*bound3*bound4; } private: - int bound1, bound2, bound3, bound4; - void init(const int size); + int bound1, bound2, bound3, bound4; + void init(const int size); }; /// Scale a ComplexArray cd by real r ComplexArray operator*(const double r, const ComplexArray &cd); @@ -140,16 +140,16 @@ void point_mult(ComplexArray &a1, ComplexArray &in2, ComplexArray &out); template void zeros(std::complex *u, int n) { - if (n == 0 || u == 0) - { - std::cout << "\n error in zeros(),n or u = 0"; - return; - } - - for (int i = 0;i < n;i++) - { - u[i] = std::complex (0.0, 0.0); - } + if (n == 0 || u == 0) + { + std::cout << "\n error in zeros(),n or u = 0"; + return; + } + + for (int i = 0;i < n;i++) + { + u[i] = std::complex (0.0, 0.0); + } } } diff --git a/source/source_base/complexmatrix.cpp b/source/source_base/complexmatrix.cpp index 217ebaf0dd7..6c07014ee50 100644 --- a/source/source_base/complexmatrix.cpp +++ b/source/source_base/complexmatrix.cpp @@ -14,268 +14,268 @@ namespace ModuleBase { // constructor with sizes ComplexMatrix::ComplexMatrix(const int nrows, const int ncols, const bool flag_zero) - :nr(nrows), - nc(ncols), - size(nrows*ncols), - c(nullptr) + :nr(nrows), + nc(ncols), + size(nrows*ncols), + c(nullptr) { - if( size ) - { - c = new std::complex[size]; - if(flag_zero) zero_out(); - } + if( size ) + { + c = new std::complex[size]; + if(flag_zero) zero_out(); + } } // zero out the ComplexMatrix void ComplexMatrix::zero_out(void) { - for (int i=0; i(0.0,0.0); + for (int i=0; i(0.0,0.0); } /* void need_more_memory() { - std::cout << "\n Sorry to crash... but the running need more momory! Exit." << std::endl; - exit(0); + std::cout << "\n Sorry to crash... but the running need more momory! Exit." << std::endl; + exit(0); } */ // Copy constructor ComplexMatrix::ComplexMatrix(const ComplexMatrix &m1) - :nr(m1.nr), - nc(m1.nc), - size(m1.size), - c(nullptr) + :nr(m1.nr), + nc(m1.nc), + size(m1.size), + c(nullptr) { - if(size) - { - c = new std::complex[size]; - memcpy( c, m1.c, size*sizeof(std::complex) ); - } + if(size) + { + c = new std::complex[size]; + memcpy( c, m1.c, size*sizeof(std::complex) ); + } } // Peize Lin add 2016-08-05 ComplexMatrix::ComplexMatrix( ComplexMatrix && m1 ) - :nr(m1.nr), - nc(m1.nc), - size(m1.size), - c(m1.c) + :nr(m1.nr), + nc(m1.nc), + size(m1.size), + c(m1.c) { - m1.nr = m1.nc = m1.size = 0; - m1.c = nullptr; + m1.nr = m1.nc = m1.size = 0; + m1.c = nullptr; } // Peize Lin add 2017-03-29 ComplexMatrix::ComplexMatrix(const matrix &m) - :nr(m.nr), - nc(m.nc), - size(m.nr*m.nc), - c(nullptr) + :nr(m.nr), + nc(m.nc), + size(m.nr*m.nc), + c(nullptr) { - if( size ) - { - c = new std::complex[size]; - for( int i=0; i[size]; + for( int i=0; i[size_in]; - } - } - else - { - c = new std::complex[nr_in * nc_in]; - } - - nr = nr_in; - nc = nc_in; - size = nr*nc; - if(flag_zero) zero_out(); - } - else - { - if(c) delete[] c; - c = nullptr; - nr = nr_in; - nc = nc_in; - size = nr*nc; - } + if( nr_in && nc_in ) + { + if(c) + { + const int size_in=nr_in*nc_in; + if( size_in!=nr*nc ) + { + delete[] c; + c = new std::complex[size_in]; + } + } + else + { + c = new std::complex[nr_in * nc_in]; + } + + nr = nr_in; + nc = nc_in; + size = nr*nc; + if(flag_zero) zero_out(); + } + else + { + if(c) delete[] c; + c = nullptr; + nr = nr_in; + nc = nc_in; + size = nr*nc; + } } void ComplexMatrix::set_as_identity_matrix(void) { - for(int i=0; i(1.0, 0.0); - else c[nc * i + j] = std::complex(0.0, 0.0); - } - } - return; + for(int i=0; i(1.0, 0.0); + else c[nc * i + j] = std::complex(0.0, 0.0); + } + } + return; } // Adding matrices, as a friend ComplexMatrix operator+(const ComplexMatrix &m1, const ComplexMatrix &m2) { - assert(m1.nr == m2.nr); - assert(m2.nc == m2.nc); - - ComplexMatrix tm(m1); - tm+=m2; - return tm; + assert(m1.nr == m2.nr); + assert(m2.nc == m2.nc); + + ComplexMatrix tm(m1); + tm+=m2; + return tm; } // Subtracting matrices, as a friend ComplexMatrix operator-(const ComplexMatrix &m1, const ComplexMatrix &m2) { - assert(m1.nr == m2.nr); - assert(m2.nc == m2.nc); - - ComplexMatrix tm(m1); - tm-=m2; - return tm; + assert(m1.nr == m2.nr); + assert(m2.nc == m2.nc); + + ComplexMatrix tm(m1); + tm-=m2; + return tm; } // Multiplying matrices, as a friend // mprod = m1 * m2 ComplexMatrix operator*(const ComplexMatrix &m1, const ComplexMatrix &m2) { - assert(m1.nc == m2.nr); - ComplexMatrix mprod(m1.nr, m2.nc); + assert(m1.nc == m2.nr); + ComplexMatrix mprod(m1.nr, m2.nc); // mohan add 2021-04-05 #ifdef __NORMAL - std::complex z; - for (int i = 0;i < m1.nr;i++) - { - for (int j = 0;j < m2.nc;j++) - { - z = std::complex(0,0); - for (int k = 0;k < m1.nc;k++) - { - z += m1(i, k) * m2(k, j); - } - mprod(i, j) = z; - } - } + std::complex z; + for (int i = 0;i < m1.nr;i++) + { + for (int j = 0;j < m2.nc;j++) + { + z = std::complex(0,0); + for (int k = 0;k < m1.nc;k++) + { + z += m1(i, k) * m2(k, j); + } + mprod(i, j) = z; + } + } #else - // Peize Lin accelerate 2017-10-27 - BlasConnector::gemm('N', 'N', m1.nr, m2.nc, m1.nc, - 1, m1.c, m1.nc, m2.c, m2.nc, - 0, mprod.c, mprod.nc); + // Peize Lin accelerate 2017-10-27 + BlasConnector::gemm('N', 'N', m1.nr, m2.nc, m1.nc, + 1, m1.c, m1.nc, m2.c, m2.nc, + 0, mprod.c, mprod.nc); #endif - return mprod; + return mprod; } // Scale a ComplexMatrix ComplexMatrix operator*(const std::complex &c,const ComplexMatrix &m) { - ComplexMatrix sm(m); - for (int i=0 ;i &c) { - ComplexMatrix sm(m); - for (int i = 0;i < m.size;i++) sm.c[i] *= c; - return sm; + ComplexMatrix sm(m); + for (int i = 0;i < m.size;i++) sm.c[i] *= c; + return sm; } ComplexMatrix operator*(const double &r,const ComplexMatrix &m) { - ComplexMatrix sm(m); - for(int i=0; icreate(m.nr, m.nc, false); - memcpy( c, m.c, size*sizeof(std::complex) ); - return *this; + this->create(m.nr, m.nc, false); + memcpy( c, m.c, size*sizeof(std::complex) ); + return *this; } // Peize Lin add 2016-08-05 ComplexMatrix& ComplexMatrix::operator=( ComplexMatrix && m ) { - nr = m.nr; nc = m.nc; size = m.size; - if(c) delete[] c; - c = m.c; - m.nr = m.nc = m.size = 0; - m.c = nullptr; - return *this; + nr = m.nr; nc = m.nc; size = m.size; + if(c) delete[] c; + c = m.c; + m.nr = m.nc = m.size = 0; + m.c = nullptr; + return *this; } ComplexMatrix& ComplexMatrix::operator*=(const std::complex &s) { - for (int i = 0;i < this->size;i++) c[i] *= s; - return *this; + for (int i = 0;i < this->size;i++) c[i] *= s; + return *this; } // Accumulate to a ComplexMatrix in place ComplexMatrix& ComplexMatrix::operator+=(const ComplexMatrix &m) { - for(int i=0; ic[i] += m.c[i]; - return *this; + for(int i=0; ic[i] += m.c[i]; + return *this; } // decumulate to a ComplexMatrix in place ComplexMatrix& ComplexMatrix::operator-=(const ComplexMatrix &m) { - for(int i=0; ic[i] -= m.c[i]; - return *this; + for(int i=0; ic[i] -= m.c[i]; + return *this; } // Peize Lin add 2017-03-29 matrix ComplexMatrix::real() const { - matrix m(nr,nc,false); - for( int i=0; isize; ++i) m.c[i] = c[i].real(); - return m; + matrix m(nr,nc,false); + for( int i=0; isize; ++i) m.c[i] = c[i].real(); + return m; } // Returns trace of ComplexMatrix std::complex trace(const ComplexMatrix &m) { - std::complex tr=std::complex(0,0); - assert(m.nr == m.nc); - for (int i=0; i tr=std::complex(0,0); + assert(m.nr == m.nc); + for (int i=0; i &s, const ComplexMatrix &min, ComplexMatrix &mout) { - assert(min.nr == mout.nr); - assert(min.nc == mout.nc); - for (int j=0; j=0); - for (int i=0; i=0); + for (int i=0; i &s1, const ComplexMatrix &m2, ComplexMatrix &mout) { - assert(m1.nr == m2.nr); - assert(m1.nr == mout.nr); - assert(m1.nc == m2.nc); - assert(m1.nc == mout.nc); + assert(m1.nr == m2.nr); + assert(m1.nr == mout.nr); + assert(m1.nc == m2.nc); + assert(m1.nc == mout.nc); - for(int i=0; i0); - for(int i=0; i0); + for(int i=0; i z; - for(int ic=0;ic z; + for(int ic=0;ic z; - for(int ir=0;ir z; + for(int ir=0;ir z; + double r=0.0; + std::complex z; - for (int i = 0;i < m.size;i++) - { - z = m.c[i]; - r += z.real() * z.real() + z.imag() * z.imag(); - } - return r; + for (int i = 0;i < m.size;i++) + { + z = m.c[i]; + r += z.real() * z.real() + z.imag() * z.imag(); + } + return r; } // Same for an array of matrices double abs2(const int nmat, ComplexMatrix **m) { - double r = 0.0; - for (int i = 0;i < nmat;i++) - { - r += abs2(*m[i]); - } - return r; + double r = 0.0; + for (int i = 0;i < nmat;i++) + { + r += abs2(*m[i]); + } + return r; } ComplexMatrix transpose(const ComplexMatrix &m, const bool &conjugate) { - ComplexMatrix tm(m.nc, m.nr, false); - if(conjugate) - for (int i = 0;i < m.nr;i++) - for (int j = 0;j < m.nc;j++) - tm(j, i) = conj ( m(i, j) ); - else - for (int i = 0;i < m.nr;i++) - for (int j = 0;j < m.nc;j++) - tm(j, i) = m(i, j); - return tm; + ComplexMatrix tm(m.nc, m.nr, false); + if(conjugate) + for (int i = 0;i < m.nr;i++) + for (int j = 0;j < m.nc;j++) + tm(j, i) = conj ( m(i, j) ); + else + for (int i = 0;i < m.nr;i++) + for (int j = 0;j < m.nc;j++) + tm(j, i) = m(i, j); + return tm; } ComplexMatrix conj(const ComplexMatrix &m) { - ComplexMatrix cm( m.nr, m.nc, false ); - for(int i=0; i!=m.size; ++i) - cm.c[i] = conj(m.c[i]); - return cm; + ComplexMatrix cm( m.nr, m.nc, false ); + for(int i=0; i!=m.size; ++i) + cm.c[i] = conj(m.c[i]); + return cm; } // Peize Lin add 2021.09.08 std::ostream & ComplexMatrix::print( std::ostream & os, const double threshold_abs, const double threshold_imag ) const { - for( int ir=0; ir!=this->nr; ++ir ) - { - for( int ic=0; ic!=this->nc; ++ic ) - { - const std::complex & data = (*this)(ir,ic); - if(std::abs(data)>threshold_abs) - { - if(std::abs(std::imag(data))>threshold_imag) - os<nr; ++ir ) + { + for( int ic=0; ic!=this->nc; ++ic ) + { + const std::complex & data = (*this)(ir,ic); + if(std::abs(data)>threshold_abs) + { + if(std::abs(std::imag(data))>threshold_imag) + os<nr;i++) - { - for(int j=0;jnc;j++) - { - if(std::imag((*this)(i,j)) > tiny) - { - return 0; - } - } - } - return 1; + const double tiny = 1e-12; + for(int i=0;inr;i++) + { + for(int j=0;jnc;j++) + { + if(std::imag((*this)(i,j)) > tiny) + { + return 0; + } + } + } + return 1; } } \ No newline at end of file diff --git a/source/source_base/complexmatrix.h b/source/source_base/complexmatrix.h index df1585e4c18..9dc890198d3 100644 --- a/source/source_base/complexmatrix.h +++ b/source/source_base/complexmatrix.h @@ -14,56 +14,56 @@ class ComplexMatrix public: - int nr=0; - int nc=0; - int size=0; - std::complex *c=nullptr; - - ComplexMatrix(): nr(0), nc(0), size(0), c(nullptr){} - ComplexMatrix(const int nrows,const int ncols,const bool flag_zero=true); // Peize Lin add flag_zero 2019-05-13 - ComplexMatrix(const ComplexMatrix &m1); - ComplexMatrix(ComplexMatrix && m1); // Peize Lin add 2016-08-05 - explicit ComplexMatrix(const matrix &m); // Peize Lin add 2017-03-29 - ~ComplexMatrix(); - - void create(const int nrow,const int ncol,const bool flag_zero=true); // Peize Lin add flag_zero 2019-05-13 - ComplexMatrix& operator=(const ComplexMatrix &m); - ComplexMatrix& operator=(ComplexMatrix && m); // Peize Lin add 2016-08-05 - - //============ - // Operators - //============ - std::complex &operator()(const int ir,const int ic) - { - assert(ir>=0); assert(ir=0); assert(ic &operator()(const int ir,const int ic)const - { - assert(ir>=0); assert(ir=0); assert(ic &s); - ComplexMatrix& operator+=(const ComplexMatrix &m); - ComplexMatrix& operator-=(const ComplexMatrix &m); - //return a matrix whose element is the real part of element of the ComplexMatrix. - matrix real() const; // Peize Lin add 2017-03-29 - - //================== - // member function: - //================== - //set all elements to be std::complex {0.0,0.0} - void zero_out(void); - //set to be a unit matrix, - void set_as_identity_matrix(void); - - std::ostream & print( std::ostream & os, const double threshold_abs=0.0, const double threshold_imag=0.0 ) const; // Peize Lin add 2021.09.08 - - // check if all the elements are real - bool checkreal(void) const; - - using type=std::complex; // Peiae Lin add 2022.08.08 for template + int nr=0; + int nc=0; + int size=0; + std::complex *c=nullptr; + + ComplexMatrix(): nr(0), nc(0), size(0), c(nullptr){} + ComplexMatrix(const int nrows,const int ncols,const bool flag_zero=true); // Peize Lin add flag_zero 2019-05-13 + ComplexMatrix(const ComplexMatrix &m1); + ComplexMatrix(ComplexMatrix && m1); // Peize Lin add 2016-08-05 + explicit ComplexMatrix(const matrix &m); // Peize Lin add 2017-03-29 + ~ComplexMatrix(); + + void create(const int nrow,const int ncol,const bool flag_zero=true); // Peize Lin add flag_zero 2019-05-13 + ComplexMatrix& operator=(const ComplexMatrix &m); + ComplexMatrix& operator=(ComplexMatrix && m); // Peize Lin add 2016-08-05 + + //============ + // Operators + //============ + std::complex &operator()(const int ir,const int ic) + { + assert(ir>=0); assert(ir=0); assert(ic &operator()(const int ir,const int ic)const + { + assert(ir>=0); assert(ir=0); assert(ic &s); + ComplexMatrix& operator+=(const ComplexMatrix &m); + ComplexMatrix& operator-=(const ComplexMatrix &m); + //return a matrix whose element is the real part of element of the ComplexMatrix. + matrix real() const; // Peize Lin add 2017-03-29 + + //================== + // member function: + //================== + //set all elements to be std::complex {0.0,0.0} + void zero_out(void); + //set to be a unit matrix, + void set_as_identity_matrix(void); + + std::ostream & print( std::ostream & os, const double threshold_abs=0.0, const double threshold_imag=0.0 ) const; // Peize Lin add 2021.09.08 + + // check if all the elements are real + bool checkreal(void) const; + + using type=std::complex; // Peiae Lin add 2022.08.08 for template }; ComplexMatrix operator+(const ComplexMatrix &m1, const ComplexMatrix &m2); @@ -94,32 +94,32 @@ ComplexMatrix conj(const ComplexMatrix &m); // Peize Lin add 2019-05-13 //do mout += s*min void scale_accumulate( - const std::complex &s, - const ComplexMatrix &min, - ComplexMatrix &mout); + const std::complex &s, + const ComplexMatrix &min, + ComplexMatrix &mout); //do (*mout[i]) += s * (*min[i]); int i &s, - ComplexMatrix **min, - ComplexMatrix **mout); + const int &nmat, + const std::complex &s, + ComplexMatrix **min, + ComplexMatrix **mout); // Do mout = s1*m1 + s2*m2 void scaled_sum( - const std::complex &s1, - const ComplexMatrix &m1, - const std::complex &s2, - const ComplexMatrix &m2, - ComplexMatrix &mout); + const std::complex &s1, + const ComplexMatrix &m1, + const std::complex &s2, + const ComplexMatrix &m2, + ComplexMatrix &mout); // Do (*mout[i]) = s1 * (*m1[i]) + s2 * (*m2[i]) void scaled_sum( - const int &nmat, - const std::complex &s1, - ComplexMatrix **m1, - const std::complex &s2, - ComplexMatrix **m2, - ComplexMatrix **mout); + const int &nmat, + const std::complex &s1, + ComplexMatrix **m1, + const std::complex &s2, + ComplexMatrix **m2, + ComplexMatrix **mout); } #endif diff --git a/source/source_base/container_operator.h b/source/source_base/container_operator.h index ec1f3fbc237..fb9fc16ed10 100644 --- a/source/source_base/container_operator.h +++ b/source/source_base/container_operator.h @@ -8,59 +8,59 @@ template< typename T> std::vector operator + ( const std::vector & x1, const std::vector & x2 ) { - assert(x1.size()==x2.size()); - std::vector x; - for(std::size_t i=0; i!=x1.size(); ++i ) - x.push_back(x1[i]+x2[i]); - return x; + assert(x1.size()==x2.size()); + std::vector x; + for(std::size_t i=0; i!=x1.size(); ++i ) + x.push_back(x1[i]+x2[i]); + return x; } template< typename T> std::vector operator - ( const std::vector & x1, const std::vector & x2 ) { - assert(x1.size()==x2.size()); - std::vector x; - for(std::size_t i=0; i!=x1.size(); ++i ) - x.push_back(x1[i]-x2[i]); - return x; + assert(x1.size()==x2.size()); + std::vector x; + for(std::size_t i=0; i!=x1.size(); ++i ) + x.push_back(x1[i]-x2[i]); + return x; } template< typename T1, typename T2 > std::map operator + ( const std::map & x1, const std::map & x2 ) { - assert(x1.size()==x2.size()); - std::map x; - for( const auto &x1i : x1 ) - x.insert(std::make_pair( x1i.first, x1i.second + x2.at(x1i.first) )); - return x; + assert(x1.size()==x2.size()); + std::map x; + for( const auto &x1i : x1 ) + x.insert(std::make_pair( x1i.first, x1i.second + x2.at(x1i.first) )); + return x; } template< typename T1, typename T2 > std::map operator - ( const std::map & x1, const std::map & x2 ) { - assert(x1.size()==x2.size()); - std::map x; - for( const auto &x1i : x1 ) - x.insert(std::make_pair( x1i.first, x1i.second - x2.at(x1i.first) )); - return x; + assert(x1.size()==x2.size()); + std::map x; + for( const auto &x1i : x1 ) + x.insert(std::make_pair( x1i.first, x1i.second - x2.at(x1i.first) )); + return x; } template< typename T1, typename T2 > std::vector operator * ( const T1 & x1, const std::vector & x2 ) { - std::vector x; - for(std::size_t i=0; i!=x2.size(); ++i ) - x.push_back(x1*x2[i]); - return x; + std::vector x; + for(std::size_t i=0; i!=x2.size(); ++i ) + x.push_back(x1*x2[i]); + return x; } template< typename T1, typename T21, typename T22 > std::map operator * ( const T1 & x1, const std::map & x2 ) { - std::map x; - for( const auto & x2i : x2 ) - x.insert(std::make_pair( x2i.first, x1*x2i.second )); - return x; + std::map x; + for( const auto & x2i : x2 ) + x.insert(std::make_pair( x2i.first, x1*x2i.second )); + return x; } #endif // CONTAINER_OPERATOR_H diff --git a/source/source_base/element_basis_index.cpp b/source/source_base/element_basis_index.cpp index 32eafd01e1d..46fe192c97f 100644 --- a/source/source_base/element_basis_index.cpp +++ b/source/source_base/element_basis_index.cpp @@ -11,30 +11,30 @@ namespace ModuleBase Element_Basis_Index::IndexLNM Element_Basis_Index::construct_index( const Range &range ) { - IndexLNM index; - index.resize( range.size() ); - for( std::size_t T=0; T!=range.size(); ++T ) - { - std::size_t count=0; - index[T].resize( range[T].size() ); - for( std::size_t L=0; L!=range[T].size(); ++L ) - { - index[T][L].resize( range[T][L].N ); - for( std::size_t N=0; N!=range[T][L].N; ++N ) - { - index[T][L][N].resize( range[T][L].M ); - for( std::size_t M=0; M!=range[T][L].M; ++M ) - { - index[T][L][N][M] = count; - ++count; - } - } - index[T][L].N = range[T][L].N; - index[T][L].M = range[T][L].M; - } - index[T].count_size = count; - } - return index; + IndexLNM index; + index.resize( range.size() ); + for( std::size_t T=0; T!=range.size(); ++T ) + { + std::size_t count=0; + index[T].resize( range[T].size() ); + for( std::size_t L=0; L!=range[T].size(); ++L ) + { + index[T][L].resize( range[T][L].N ); + for( std::size_t N=0; N!=range[T][L].N; ++N ) + { + index[T][L][N].resize( range[T][L].M ); + for( std::size_t M=0; M!=range[T][L].M; ++M ) + { + index[T][L][N][M] = count; + ++count; + } + } + index[T][L].N = range[T][L].N; + index[T][L].M = range[T][L].M; + } + index[T].count_size = count; + } + return index; } } \ No newline at end of file diff --git a/source/source_base/element_basis_index.h b/source/source_base/element_basis_index.h index bb81ebf40e9..22bfd2c7f0a 100644 --- a/source/source_base/element_basis_index.h +++ b/source/source_base/element_basis_index.h @@ -16,32 +16,32 @@ namespace Element_Basis_Index { //private: - struct NM - { - public: - std::size_t N; - std::size_t M; - }; - - class Index_TL: public std::vector> - { - public: - std::size_t N; - std::size_t M; - }; - - class Index_T: public std::vector - { - public: - std::size_t count_size; - }; + struct NM + { + public: + std::size_t N; + std::size_t M; + }; + + class Index_TL: public std::vector> + { + public: + std::size_t N; + std::size_t M; + }; + + class Index_T: public std::vector + { + public: + std::size_t count_size; + }; //public: - typedef std::vector> Range; // range[T][L] - typedef std::vector IndexLNM; // index[T][L][N][M] + typedef std::vector> Range; // range[T][L] + typedef std::vector IndexLNM; // index[T][L][N][M] - extern IndexLNM construct_index( const Range &range ); + extern IndexLNM construct_index( const Range &range ); } } diff --git a/source/source_base/export.cpp b/source/source_base/export.cpp index 413673a8495..947f12ffb91 100644 --- a/source/source_base/export.cpp +++ b/source/source_base/export.cpp @@ -13,20 +13,20 @@ /* void ModuleBase::IF_MATCH(const std::string &name,const std::string &name2) { - if(name!=name2) - { - if(GlobalV::MY_RANK == 0) - { - std::cout<<"\n Can not match : "< void ModuleBase::GlobalFunc::AUTO_SET(std::ofstream &ofs,const std::string &name,const T &a) { - ofs<<" AUTO_SET "< void IF_MATCH(const T &a,const T &b) { - if(a!=b) - { - if(GlobalV::MY_RANK == 0) - { - std::cout<<"\n Can not match : "<0) {break; + if(make_dir>0) {break; } - ++times; - } + ++times; + } #ifdef __MPI - if(make_dir==0) - { - std::cout << " CAN NOT MAKE THE OUT DIR......." << std::endl; - ModuleBase::QUIT(); - } - MPI_Barrier(MPI_COMM_WORLD); + if(make_dir==0) + { + std::cout << " CAN NOT MAKE THE OUT DIR......." << std::endl; + ModuleBase::QUIT(); + } + MPI_Barrier(MPI_COMM_WORLD); #endif if(calculation == "md") @@ -287,20 +287,20 @@ void ModuleBase::Global_File::make_dir_out( // mohan add 2010-09-12 if(out_alllog) { - open_log(GlobalV::ofs_running, log_file, calculation, restart, global_out_dir); + open_log(GlobalV::ofs_running, log_file, calculation, restart, global_out_dir); #if defined(__CUDA) || defined(__ROCM) open_log(GlobalV::ofs_device, "device" + std::to_string(rank) + ".log", calculation, restart, global_out_dir); #endif } else { - if(rank==0) - { - open_log(GlobalV::ofs_running, log_file, calculation, restart, global_out_dir); + if(rank==0) + { + open_log(GlobalV::ofs_running, log_file, calculation, restart, global_out_dir); #if defined(__CUDA) || defined(__ROCM) open_log(GlobalV::ofs_device, "device.log", calculation, restart, global_out_dir); #endif - } + } } if(rank==0) @@ -350,10 +350,10 @@ void ModuleBase::Global_File::open_log(std::ofstream &ofs, const std::string &fn void ModuleBase::Global_File::close_log( std::ofstream &ofs,const std::string &fn) { - if(ofs) - { - ofs.close(); - } + if(ofs) + { + ofs.close(); + } //ofs << "CLOSE "<> SearchName) - { - if (SearchName == TargetName) - { - find = true; - //std::cout << " search name = " << SearchName << std::endl; - return find; - } - } - } + { + if (SearchName == TargetName) + { + find = true; + //std::cout << " search name = " << SearchName << std::endl; + return find; + } + } + } if (!find && ifwarn) { @@ -225,16 +225,16 @@ void OUT_TIME(const std::string &name, time_t &start, time_t &end) double mini = difftime(end, start) / 60.0; if (mini > 0.1) { - if(GlobalV::ofs_warning) - { - GlobalV::ofs_warning << std::setprecision(2); - GlobalV::ofs_warning << " -------------------------------------------------------" << std::endl; - GlobalV::ofs_warning << " NAME < " << name << " > = " << std::endl; - GlobalV::ofs_warning << " -> " << ctime(&start) << " -> " << ctime(&end); - GlobalV::ofs_warning << " TIME = " << mini << " [Minutes]" << std::endl; - GlobalV::ofs_warning << " -------------------------------------------------------" << std::endl; - GlobalV::ofs_warning << std::setprecision(6); - } + if(GlobalV::ofs_warning) + { + GlobalV::ofs_warning << std::setprecision(2); + GlobalV::ofs_warning << " -------------------------------------------------------" << std::endl; + GlobalV::ofs_warning << " NAME < " << name << " > = " << std::endl; + GlobalV::ofs_warning << " -> " << ctime(&start) << " -> " << ctime(&end); + GlobalV::ofs_warning << " TIME = " << mini << " [Minutes]" << std::endl; + GlobalV::ofs_warning << " -------------------------------------------------------" << std::endl; + GlobalV::ofs_warning << std::setprecision(6); + } } } diff --git a/source/source_base/gram_schmidt_orth.h b/source/source_base/gram_schmidt_orth.h index 897e3c7a98d..bc69dfa6283 100644 --- a/source/source_base/gram_schmidt_orth.h +++ b/source/source_base/gram_schmidt_orth.h @@ -16,21 +16,21 @@ class Gram_Schmidt_Orth { public: - enum class Coordinate { Cartesian, Sphere }; - - Gram_Schmidt_Orth( const std::vector &rab, const Coordinate &coordinate ); - - std::vector> cal_orth( - const std::vector> &func, - const Func_Type norm_threshold = std::numeric_limits::min() ); - + enum class Coordinate { Cartesian, Sphere }; + + Gram_Schmidt_Orth( const std::vector &rab, const Coordinate &coordinate ); + + std::vector> cal_orth( + const std::vector> &func, + const Func_Type norm_threshold = std::numeric_limits::min() ); + private: - const Coordinate coordinate; - const std::vector &rab; - std::vector radial_2; + const Coordinate coordinate; + const std::vector &rab; + std::vector radial_2; - Func_Type cal_norm( const std::vector &f ); + Func_Type cal_norm( const std::vector &f ); }; diff --git a/source/source_base/gram_schmidt_orth_inl.h b/source/source_base/gram_schmidt_orth_inl.h index 809f9fbb391..90eb08cb6e4 100644 --- a/source/source_base/gram_schmidt_orth_inl.h +++ b/source/source_base/gram_schmidt_orth_inl.h @@ -16,90 +16,90 @@ namespace ModuleBase template Gram_Schmidt_Orth::Gram_Schmidt_Orth( const std::vector &rab_in, const Coordinate &coordinate_in ) - :rab(rab_in), - coordinate(coordinate_in) + :rab(rab_in), + coordinate(coordinate_in) { - if( Coordinate::Sphere == coordinate ) - { - std::vector radial( rab.size() ); - radial[0] = 0; - for( int ir=1; ir!=radial.size(); ++ir ) - radial[ir] = radial[ir-1] + rab[ir-1]; - this->radial_2 = Mathzone::Pointwise_Product( radial, radial ); - } + if( Coordinate::Sphere == coordinate ) + { + std::vector radial( rab.size() ); + radial[0] = 0; + for( int ir=1; ir!=radial.size(); ++ir ) + radial[ir] = radial[ir-1] + rab[ir-1]; + this->radial_2 = Mathzone::Pointwise_Product( radial, radial ); + } } template std::vector> Gram_Schmidt_Orth::cal_orth( - const std::vector> &func, - const Func_Type norm_threshold ) + const std::vector> &func, + const Func_Type norm_threshold ) { - // Schmidt: hn to en - // e1 = h1 / ||h1|| - // gn = hn - \sum{i=1 to n-1}(hn,ei)ei - // en = gn / ||gn|| - - std::vector> func_new; - - for( size_t if1=0; if1!=func.size(); ++if1 ) - { - //use CGS2 algorithm to do twice orthogonalization - //DOI 10.1007/s00211-005-0615-4 - std::vector func_try = func[if1]; - for(int niter=0;niter<3;niter++) - { - std::vector func_tmp = func_try; - for( size_t if_minus=0; if_minus!=func_new.size(); ++if_minus ) - { - // (hn,ei) - const std::vector && mul_func = Mathzone::Pointwise_Product( func_tmp, func_new[if_minus] ); - const Func_Type in_product = cal_norm(mul_func); + // Schmidt: hn to en + // e1 = h1 / ||h1|| + // gn = hn - \sum{i=1 to n-1}(hn,ei)ei + // en = gn / ||gn|| + + std::vector> func_new; + + for( size_t if1=0; if1!=func.size(); ++if1 ) + { + //use CGS2 algorithm to do twice orthogonalization + //DOI 10.1007/s00211-005-0615-4 + std::vector func_try = func[if1]; + for(int niter=0;niter<3;niter++) + { + std::vector func_tmp = func_try; + for( size_t if_minus=0; if_minus!=func_new.size(); ++if_minus ) + { + // (hn,ei) + const std::vector && mul_func = Mathzone::Pointwise_Product( func_tmp, func_new[if_minus] ); + const Func_Type in_product = cal_norm(mul_func); - // hn - (hn,ei)ei - BlasConnector::axpy( mul_func.size(), -in_product, ModuleBase::GlobalFunc::VECTOR_TO_PTR(func_new[if_minus]), 1, ModuleBase::GlobalFunc::VECTOR_TO_PTR(func_try), 1); - } - } - - // ||gn|| - const std::vector && func_2 = Mathzone::Pointwise_Product( func_try, func_try ); - const Func_Type norm = sqrt(cal_norm(func_2)); - - // en = gn / ||gn|| - // if ||gn|| too small, filter out - if( norm >= norm_threshold ) - { - BlasConnector::scal( func_try.size(), 1.0/norm, ModuleBase::GlobalFunc::VECTOR_TO_PTR(func_try), 1 ); - func_new.push_back( func_try ); - } - } - return func_new; + // hn - (hn,ei)ei + BlasConnector::axpy( mul_func.size(), -in_product, ModuleBase::GlobalFunc::VECTOR_TO_PTR(func_new[if_minus]), 1, ModuleBase::GlobalFunc::VECTOR_TO_PTR(func_try), 1); + } + } + + // ||gn|| + const std::vector && func_2 = Mathzone::Pointwise_Product( func_try, func_try ); + const Func_Type norm = sqrt(cal_norm(func_2)); + + // en = gn / ||gn|| + // if ||gn|| too small, filter out + if( norm >= norm_threshold ) + { + BlasConnector::scal( func_try.size(), 1.0/norm, ModuleBase::GlobalFunc::VECTOR_TO_PTR(func_try), 1 ); + func_new.push_back( func_try ); + } + } + return func_new; } // cal ||f|| template Func_Type Gram_Schmidt_Orth::cal_norm( const std::vector &f ) { - Func_Type norm = 0.0; - switch( this->coordinate ) - { - case Coordinate::Cartesian: - { - Integral::Simpson_Integral( f.size(), ModuleBase::GlobalFunc::VECTOR_TO_PTR(f), ModuleBase::GlobalFunc::VECTOR_TO_PTR(rab), norm); - break; - } - case Coordinate::Sphere: - { - const std::vector &&tmp_func = Mathzone::Pointwise_Product( f, radial_2 ); - Integral::Simpson_Integral( f.size(), ModuleBase::GlobalFunc::VECTOR_TO_PTR(tmp_func), ModuleBase::GlobalFunc::VECTOR_TO_PTR(rab), norm); - break; - } - default: - { - throw std::invalid_argument("coordinate must be Cartesian or Sphere "+std::string(__FILE__)+" line "+std::to_string(__LINE__)); - break; - } - } - return norm; + Func_Type norm = 0.0; + switch( this->coordinate ) + { + case Coordinate::Cartesian: + { + Integral::Simpson_Integral( f.size(), ModuleBase::GlobalFunc::VECTOR_TO_PTR(f), ModuleBase::GlobalFunc::VECTOR_TO_PTR(rab), norm); + break; + } + case Coordinate::Sphere: + { + const std::vector &&tmp_func = Mathzone::Pointwise_Product( f, radial_2 ); + Integral::Simpson_Integral( f.size(), ModuleBase::GlobalFunc::VECTOR_TO_PTR(tmp_func), ModuleBase::GlobalFunc::VECTOR_TO_PTR(rab), norm); + break; + } + default: + { + throw std::invalid_argument("coordinate must be Cartesian or Sphere "+std::string(__FILE__)+" line "+std::to_string(__LINE__)); + break; + } + } + return norm; } } diff --git a/source/source_base/inverse_matrix.cpp b/source/source_base/inverse_matrix.cpp index 66ced1c46e9..89916533778 100644 --- a/source/source_base/inverse_matrix.cpp +++ b/source/source_base/inverse_matrix.cpp @@ -9,68 +9,68 @@ namespace ModuleBase Inverse_Matrix_Complex::Inverse_Matrix_Complex() { - allocate=false; + allocate=false; } Inverse_Matrix_Complex::~Inverse_Matrix_Complex() { - if(allocate) - { - delete[] e; //mohan fix bug 2012-04-02 - delete[] work2; - delete[] rwork; - allocate=false; - } + if(allocate) + { + delete[] e; //mohan fix bug 2012-04-02 + delete[] work2; + delete[] rwork; + allocate=false; + } } void Inverse_Matrix_Complex::init(const int &dim_in) { // GlobalV::ofs_running << " allocate=" << allocate << std::endl; - if(allocate) - { - delete[] e; //mohan fix bug 2012-04-02 - delete[] work2; - delete[] rwork; - allocate=false; - } + if(allocate) + { + delete[] e; //mohan fix bug 2012-04-02 + delete[] work2; + delete[] rwork; + allocate=false; + } - this->dim = dim_in; + this->dim = dim_in; - assert(dim>0); - this->e = new double[dim]; - this->lwork = 2*dim; + assert(dim>0); + this->e = new double[dim]; + this->lwork = 2*dim; - assert(lwork>0); - this->work2 = new std::complex[lwork]; + assert(lwork>0); + this->work2 = new std::complex[lwork]; - assert(3*dim-2>0); - this->rwork = new double[3*dim-2]; - this->info = 0; - this->A.create(dim, dim); - this->EA.create(dim, dim); + assert(3*dim-2>0); + this->rwork = new double[3*dim-2]; + this->info = 0; + this->A.create(dim, dim); + this->EA.create(dim, dim); - this->allocate = true; + this->allocate = true; - return; + return; } void Inverse_Matrix_Complex::using_zheev( const ModuleBase::ComplexMatrix &Sin, ModuleBase::ComplexMatrix &Sout) { - ModuleBase::timer::start("Inverse","using_zheev"); - this->A = Sin; + ModuleBase::timer::start("Inverse","using_zheev"); + this->A = Sin; LapackConnector::zheev('V', 'U', dim, this->A, dim, e, work2, lwork, rwork, &info); - - for(int i=0; iA(j,i) ) / e[i] ; - } - } + + for(int i=0; iA(j,i) ) / e[i] ; + } + } Sout = this->A * this->EA; - ModuleBase::timer::end("Inverse","using_zheev"); + ModuleBase::timer::end("Inverse","using_zheev"); return; } diff --git a/source/source_base/inverse_matrix.h b/source/source_base/inverse_matrix.h index bac2e5b979a..e9213ba9a34 100644 --- a/source/source_base/inverse_matrix.h +++ b/source/source_base/inverse_matrix.h @@ -8,26 +8,26 @@ namespace ModuleBase class Inverse_Matrix_Complex { - public: + public: - Inverse_Matrix_Complex(); - ~Inverse_Matrix_Complex(); - - ModuleBase::ComplexMatrix A; + Inverse_Matrix_Complex(); + ~Inverse_Matrix_Complex(); + + ModuleBase::ComplexMatrix A; - void using_zheev(const ModuleBase::ComplexMatrix &in, ModuleBase::ComplexMatrix &out); - void init( const int &dim_in); + void using_zheev(const ModuleBase::ComplexMatrix &in, ModuleBase::ComplexMatrix &out); + void init( const int &dim_in); - private: - int dim=0; - double *e=nullptr; - int lwork=0; - std::complex *work2=nullptr; - double* rwork=nullptr; - int info=0; - bool allocate=false; //mohan add 2012-04-02 + private: + int dim=0; + double *e=nullptr; + int lwork=0; + std::complex *work2=nullptr; + double* rwork=nullptr; + int info=0; + bool allocate=false; //mohan add 2012-04-02 - ModuleBase::ComplexMatrix EA; + ModuleBase::ComplexMatrix EA; }; /** diff --git a/source/source_base/kernels/cuda/math_ylm_op.cu b/source/source_base/kernels/cuda/math_ylm_op.cu index 01b3676ff7c..6be1da3bc5a 100644 --- a/source/source_base/kernels/cuda/math_ylm_op.cu +++ b/source/source_base/kernels/cuda/math_ylm_op.cu @@ -1,4 +1,6 @@ #include + +#include #include "source_base/kernels/math_ylm_op.h" #include @@ -41,6 +43,7 @@ __global__ void cal_ylm_real( FPTYPE * p, FPTYPE * ylm) { + const std::int64_t stride = ng; int ig = blockIdx.x * blockDim.x + threadIdx.x; if (ig >= ng) {return;} @@ -49,7 +52,7 @@ __global__ void cal_ylm_real( // EXPLAIN : if lmax = 1,only use Y00 , output result. //---------------------------------------------------------- if (lmax == 0) { - ylm[0 * ng + ig] = SQRT_INVERSE_FOUR_PI; + ylm[0 * stride + ig] = SQRT_INVERSE_FOUR_PI; return; } //---------------------------------------------------------- @@ -57,17 +60,18 @@ __global__ void cal_ylm_real( // NAME : cost = cos(theta),theta and phi are polar angles // NAME : phi //---------------------------------------------------------- - const FPTYPE gmod = sqrt(g[ig * 3 + 0] * g[ig * 3 + 0] + g[ig * 3 + 1] * g[ig * 3 + 1] + g[ig * 3 + 2] * g[ig * 3 + 2]); - cost = gmod < 1.0e-9 ? 0.0 : g[ig * 3 + 2] / gmod; + const std::int64_t g_offset = static_cast(ig) * 3; + const FPTYPE gmod = sqrt(g[g_offset + 0] * g[g_offset + 0] + g[g_offset + 1] * g[g_offset + 1] + g[g_offset + 2] * g[g_offset + 2]); + cost = gmod < 1.0e-9 ? 0.0 : g[g_offset + 2] / gmod; // beware the arc tan, it is defined modulo pi - if (g[ig * 3 + 0] > 1.0e-9) { - phi = atan(g[ig * 3 + 1] / g[ig * 3 + 0]); + if (g[g_offset + 0] > 1.0e-9) { + phi = atan(g[g_offset + 1] / g[g_offset + 0]); } - else if (g[ig * 3 + 0] < -1.e-9) { - phi = atan(g[ig * 3 + 1] / g[ig * 3 + 0]) + PI; + else if (g[g_offset + 0] < -1.e-9) { + phi = atan(g[g_offset + 1] / g[g_offset + 0]) + PI; } else { - phi = PI_HALF * ((g[ig * 3 + 1] >= 0.0) ? 1.0 : -1.0); //HLX: modified on 10/13/2006 + phi = PI_HALF * ((g[g_offset + 1] >= 0.0) ? 1.0 : -1.0); //HLX: modified on 10/13/2006 } // end if //========================================================== // NAME : p(Legendre Polynomials) (0 <= m <= l) @@ -76,12 +80,12 @@ __global__ void cal_ylm_real( for (int l = 0; l <= lmax; l++) { const FPTYPE c = sqrt((2 * l + 1) / FOUR_PI); if (l == 0) { - p[0 * (lmax + 1) * ng + 0 * ng + ig] = 1.0; + p[0 * (lmax + 1) * stride + 0 * stride + ig] = 1.0; } else if (l == 1) { - p[0 * (lmax + 1) * ng + 1 * ng + ig] = cost; + p[0 * (lmax + 1) * stride + 1 * stride + ig] = cost; FPTYPE var = (1.0 - cost * cost) > 0.0 ? (1.0 - cost * cost) : 0.0; - p[1 * (lmax + 1) * ng + 1 * ng + ig] = -sqrt(var); + p[1 * (lmax + 1) * stride + 1 * stride + ig] = -sqrt(var); } else { const int l1 = l - 1, @@ -89,22 +93,22 @@ __global__ void cal_ylm_real( l3 = 2 * l - 1; // recursion on l for P(:,l,m) for (int m = 0; m <= l2; m++) { // do m = 0, l - 2//mohan modify 2007-10-13 - p[m * (lmax + 1) * ng + l * ng + ig] = - (cost * l3 * p[m * (lmax + 1) * ng + l1 * ng + ig] - - (l1 + m) * p[m * (lmax + 1) * ng + l2 * ng + ig]) / (l - m); + p[m * (lmax + 1) * stride + l * stride + ig] = + (cost * l3 * p[m * (lmax + 1) * stride + l1 * stride + ig] - + (l1 + m) * p[m * (lmax + 1) * stride + l2 * stride + ig]) / (l - m); } // end do - p[l1 * (lmax + 1) * ng + l * ng + ig] = - cost * l3 * p[l1 * (lmax + 1) * ng + l1 * ng + ig]; + p[l1 * (lmax + 1) * stride + l * stride + ig] = + cost * l3 * p[l1 * (lmax + 1) * stride + l1 * stride + ig]; FPTYPE x2 = (1.0 - cost * cost) > 0.0 ? (1.0 - cost * cost) : 0.0; - p[l * (lmax + 1) * ng + l * ng + ig] = __semi_fact(l3) * pow(x2, static_cast(l) / 2.0);//mohan modify 2007-10-13 + p[l * (lmax + 1) * stride + l * stride + ig] = __semi_fact(l3) * pow(x2, static_cast(l) / 2.0);//mohan modify 2007-10-13 if (l % 2 == 1) { - p[l * (lmax + 1) * ng + l * ng + ig] *= -1; + p[l * (lmax + 1) * stride + l * stride + ig] *= -1; } } // end if // Y_lm, m = 0 ++lm; - ylm[lm * ng + ig] = c * p[0 * (lmax + 1) * ng + l * ng + ig]; + ylm[lm * stride + ig] = c * p[0 * (lmax + 1) * stride + l * stride + ig]; for (int m = 1; m <= l; m++) { // Y_lm, m > 0 @@ -113,11 +117,11 @@ __global__ void cal_ylm_real( __fact(l + m)) * SQRT2; ++lm; - ylm[lm * ng + ig] = same * p[m * (lmax + 1) * ng + l * ng + ig] * cos(m * phi); + ylm[lm * stride + ig] = same * p[m * (lmax + 1) * stride + l * stride + ig] * cos(m * phi); // Y_lm, m < 0 ++lm; - ylm[lm * ng + ig] = same * p[m * (lmax + 1) * ng + l * ng + ig] * sin(m * phi); + ylm[lm * stride + ig] = same * p[m * (lmax + 1) * stride + l * stride + ig] * sin(m * phi); } }// end do } @@ -135,7 +139,7 @@ void cal_ylm_real_op::operator()(const base_dev FPTYPE* p, FPTYPE* ylm) { - int block = (ng + THREADS_PER_BLOCK - 1) / THREADS_PER_BLOCK; + int block = ng / THREADS_PER_BLOCK + (ng % THREADS_PER_BLOCK != 0); cal_ylm_real<<>>( ng, lmax, diff --git a/source/source_base/kernels/cuda/sph_harm_gpu.cuh b/source/source_base/kernels/cuda/sph_harm_gpu.cuh index caf29795d61..63bd8673b44 100644 --- a/source/source_base/kernels/cuda/sph_harm_gpu.cuh +++ b/source/source_base/kernels/cuda/sph_harm_gpu.cuh @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_BASE_KERNELS_CUDA_SPH_HARM_GPU_CUH +#define ABACUS_SOURCE_BASE_KERNELS_CUDA_SPH_HARM_GPU_CUH + #include "source_base/ylmcoef.h" @@ -174,224 +176,226 @@ __device__ static void grad_rl_sph_harm( double tz = z * 2; //begin calculation - /*************************** - L = 0 - ***************************/ - rly[0] = ylmcoef[0]; //l=0, m=0 - grly[0] = grly[1] = grly[2] = 0.0; - if (nwl == 0) return; - - /*************************** - L = 1 - ***************************/ - rly[1] = ylmcoef[1]*z; //l=1, m=0 - grly[3] = grly[4] = 0.0; - grly[5] = ylmcoef[1]; - - rly[2] = -ylmcoef[1]*x; //l=1, m=1 - grly[7] = grly[8] = 0.0; - grly[6] = -ylmcoef[1]; - - rly[3] = -ylmcoef[1]*y; //l=1, m=-1 - grly[9] = grly[11] = 0.0; - grly[10] = -ylmcoef[1]; - - if (nwl == 1) return; - - /*************************** - L = 2 - ***************************/ - rly[4] = ylmcoef[2]*z*rly[1]-ylmcoef[3]*rly[0]*r2;//l=2, m=0 - grly[12] = ylmcoef[2]*z*grly[3]-ylmcoef[3]*(grly[0]*r2+rly[0]*tx);//l=2, m=0 - grly[13] = ylmcoef[2]*z*grly[4]-ylmcoef[3]*(grly[1]*r2+rly[0]*ty);//l=2, m=0 - grly[14] = ylmcoef[2]*(z*grly[5]+rly[1])-ylmcoef[3]*(grly[2]*r2+rly[0]*tz);//l=2, m=0 - - - double tmp0 = ylmcoef[4]*z; - rly[5] = tmp0*rly[2];//l=2,m=1 - grly[15] = tmp0*grly[6]; - grly[16] = tmp0*grly[7]; - grly[17] = ylmcoef[4]*(rly[2]+z*grly[8]); - - rly[6] = tmp0*rly[3];//l=2,m=-1 - grly[18] = tmp0*grly[9]; - grly[19] = tmp0*grly[10]; - grly[20] = ylmcoef[4]*(rly[3]+z*grly[11]); - - double tmp2 = ylmcoef[4]*x; - rly[7]= ylmcoef[5]*rly[4]-ylmcoef[6]*rly[0]*r2 - tmp2*rly[2];//l=2,m=2 - grly[21] = ylmcoef[5]*grly[12]-ylmcoef[6]*(rly[0]*tx+grly[0]*r2)-ylmcoef[4]*(x*grly[6]+rly[2]); - - grly[22] = ylmcoef[5]*grly[13]-ylmcoef[6]*(rly[0]*ty+grly[1]*r2)-tmp2*grly[7]; - grly[23] = ylmcoef[5]*grly[14]-ylmcoef[6]*(rly[0]*tz+grly[2]*r2)-tmp2*grly[8]; - - rly[8] = -tmp2*rly[3]; - grly[24] = -ylmcoef[4]*(rly[3]+x*grly[9]); - grly[25] = -tmp2*grly[10]; - grly[26] = -tmp2*grly[11]; - if (nwl == 2) return; - - /*************************** - L = 3 - ***************************/ - rly[9] = ylmcoef[7]*z*rly[4]-ylmcoef[8]*rly[1]*r2; //l=3, m=0 - grly[27] = ylmcoef[7]*z*grly[12]-ylmcoef[8]*(rly[1]*tx+grly[3]*r2); - grly[28] = ylmcoef[7]*z*grly[13]-ylmcoef[8]*(rly[1]*ty+grly[4]*r2); - grly[29] = ylmcoef[7]*(rly[4]+z*grly[14])-ylmcoef[8]*(rly[1]*tz+grly[5]*r2); - - double tmp3 = ylmcoef[9]*z; - rly[10] = tmp3*rly[5]-ylmcoef[10]*rly[2]*r2;//l=3,m=1 - grly[30] = tmp3*grly[15]-ylmcoef[10]*(grly[6]*r2+rly[2]*tx); - grly[31] = tmp3*grly[16]-ylmcoef[10]*(grly[7]*r2+rly[2]*ty); - grly[32] = ylmcoef[9]*(z*grly[17]+rly[5])-ylmcoef[10]*(grly[8]*r2+rly[2]*tz); - - rly[11] = tmp3*rly[6]-ylmcoef[10]*rly[3]*r2;//l=3,m=-1 - grly[33] = tmp3*grly[18]-ylmcoef[10]*(grly[9]*r2+rly[3]*tx); - grly[34] = tmp3*grly[19]-ylmcoef[10]*(grly[10]*r2+rly[3]*ty); - grly[35] = ylmcoef[9]*(z*grly[20]+rly[6])-ylmcoef[10]*(grly[11]*r2+rly[3]*tz); - - double tmp4 = ylmcoef[11]*z; - rly[12] = tmp4*rly[7];//l=3,m=2 - grly[36] = tmp4*grly[21]; - grly[37] = tmp4*grly[22]; - grly[38] = ylmcoef[11]*(z*grly[23]+rly[7]); - - rly[13] = tmp4*rly[8];//l=3,m=-2 - grly[39] = tmp4*grly[24]; - grly[40] = tmp4*grly[25]; - grly[41] = ylmcoef[11]*(z*grly[26]+rly[8]); - - double tmp5 = ylmcoef[14]*x; - rly[14] = ylmcoef[12]*rly[10]-ylmcoef[13]*rly[2]*r2-tmp5*rly[7];//l=3,m=3 - grly[42] = ylmcoef[12]*grly[30]-ylmcoef[13]*(rly[2]*tx+grly[6]*r2)-ylmcoef[14]*(rly[7]+x*grly[21]); - grly[43] = ylmcoef[12]*grly[31]-ylmcoef[13]*(rly[2]*ty+grly[7]*r2)-tmp5*grly[22]; - grly[44] = ylmcoef[12]*grly[32]-ylmcoef[13]*(rly[2]*tz+grly[8]*r2)-tmp5*grly[23]; - - rly[15] = ylmcoef[12]*rly[11]-ylmcoef[13]*rly[3]*r2-tmp5*rly[8];//l=3,m=-3 - grly[45] = ylmcoef[12]*grly[33]-ylmcoef[13]*(rly[3]*tx+grly[9]*r2)-ylmcoef[14]*(rly[8]+x*grly[24]); - grly[46] = ylmcoef[12]*grly[34]-ylmcoef[13]*(rly[3]*ty+grly[10]*r2)-tmp5*grly[25]; - grly[47] = ylmcoef[12]*grly[35]-ylmcoef[13]*(rly[3]*tz+grly[11]*r2)-tmp5*grly[26]; - if (nwl == 3) return; - - /*************************** - L = 4 - ***************************/ - rly[16] = ylmcoef[15]*z*rly[9]-ylmcoef[16]*rly[4]*r2;//l=4,m=0 - grly[48] = ylmcoef[15]*z*grly[27]-ylmcoef[16]*(rly[4]*tx+grly[12]*r2); - grly[49] = ylmcoef[15]*z*grly[28]-ylmcoef[16]*(rly[4]*ty+grly[13]*r2); - grly[50] = ylmcoef[15]*(z*grly[29]+rly[9])-ylmcoef[16]*(rly[4]*tz+grly[14]*r2); - - double tmp6 = ylmcoef[17]*z; - rly[17] = tmp6*rly[10]-ylmcoef[18]*rly[5]*r2;//l=4,m=1 - grly[51] = tmp6*grly[30]-ylmcoef[18]*(rly[5]*tx+grly[15]*r2); - grly[52] = tmp6*grly[31]-ylmcoef[18]*(rly[5]*ty+grly[16]*r2); - grly[53] = ylmcoef[17]*(z*grly[32]+rly[10])-ylmcoef[18]*(rly[5]*tz+grly[17]*r2); - - rly[18] = tmp6*rly[11]-ylmcoef[18]*rly[6]*r2;//l=4,m=-1 - grly[54] = tmp6*grly[33]-ylmcoef[18]*(rly[6]*tx+grly[18]*r2); - grly[55] = tmp6*grly[34]-ylmcoef[18]*(rly[6]*ty+grly[19]*r2); - grly[56] = ylmcoef[17]*(z*grly[35]+rly[11])-ylmcoef[18]*(rly[6]*tz+grly[20]*r2); - - double tmp7 = ylmcoef[19]*z; - rly[19] = tmp7*rly[12]-ylmcoef[20]*rly[7]*r2;//l=4,m=2 - grly[57] = tmp7*grly[36]-ylmcoef[20]*(rly[7]*tx+grly[21]*r2); - grly[58] = tmp7*grly[37]-ylmcoef[20]*(rly[7]*ty+grly[22]*r2); - grly[59] = ylmcoef[19]*(z*grly[38]+rly[12])-ylmcoef[20]*(rly[7]*tz+grly[23]*r2); - - rly[20] = tmp7*rly[13]-ylmcoef[20]*rly[8]*r2;//l=4,m=-2 - grly[60] = tmp7*grly[39]-ylmcoef[20]*(rly[8]*tx+grly[24]*r2); - grly[61] = tmp7*grly[40]-ylmcoef[20]*(rly[8]*ty+grly[25]*r2); - grly[62] = ylmcoef[19]*(z*grly[41]+rly[13])-ylmcoef[20]*(rly[8]*tz+grly[26]*r2); - - double tmp8 = 3.0*z; - rly[21] = tmp8*rly[14];//l=4,m=3 - grly[63] = tmp8*grly[42]; - grly[64] = tmp8*grly[43]; - grly[65] = 3.0*(z*grly[44]+rly[14]); - - - rly[22] = tmp8*rly[15];//l=4,m=-3 - grly[66] = tmp8*grly[45]; - grly[67] = tmp8*grly[46]; - grly[68] = 3.0*(z*grly[47]+rly[15]); - - double tmp9 = ylmcoef[23]*x; - rly[23] = ylmcoef[21]*rly[19]-ylmcoef[22]*rly[7]*r2-tmp9*rly[14];//l=4,m=4 - grly[69] = ylmcoef[21]*grly[57]-ylmcoef[22]*(rly[7]*tx+grly[21]*r2)-ylmcoef[23]*(x*grly[42]+rly[14]); - grly[70] = ylmcoef[21]*grly[58]-ylmcoef[22]*(rly[7]*ty+grly[22]*r2)-tmp9*grly[43]; - grly[71] = ylmcoef[21]*grly[59]-ylmcoef[22]*(rly[7]*tz+grly[23]*r2)-tmp9*grly[44]; - - rly[24] = ylmcoef[21]*rly[20]-ylmcoef[22]*rly[8]*r2-tmp9*rly[15];//l=4,m=-4 - grly[72] = ylmcoef[21]*grly[60]-ylmcoef[22]*(rly[8]*tx+grly[24]*r2)-ylmcoef[23]*(x*grly[45]+rly[15]); - grly[73] = ylmcoef[21]*grly[61]-ylmcoef[22]*(rly[8]*ty+grly[25]*r2)-tmp9*grly[46]; - grly[74] = ylmcoef[21]*grly[62]-ylmcoef[22]*(rly[8]*tz+grly[26]*r2)-tmp9*grly[47]; - - if (nwl == 4) return; - - /*************************** - L = 5 - ***************************/ - rly[25] = ylmcoef[24]*z*rly[16]-ylmcoef[25]*rly[9]*r2;//l=5,m=0 - grly[75] = ylmcoef[24]*z*grly[48]-ylmcoef[25]*(rly[9]*tx+grly[27]*r2); - grly[76] = ylmcoef[24]*z*grly[49]-ylmcoef[25]*(rly[9]*ty+grly[28]*r2); - grly[77] = ylmcoef[24]*(z*grly[50]+rly[16])-ylmcoef[25]*(rly[9]*tz+grly[29]*r2); - - double tmp10 = ylmcoef[26]*z; - rly[26] = tmp10*rly[17]-ylmcoef[27]*rly[10]*r2;//l=5,m=1 - grly[78] = tmp10*grly[51]-ylmcoef[27]*(rly[10]*tx+grly[30]*r2); - grly[79] = tmp10*grly[52]-ylmcoef[27]*(rly[10]*ty+grly[31]*r2); - grly[80] = ylmcoef[26]*(z*grly[53]+rly[17])-ylmcoef[27]*(rly[10]*tz+grly[32]*r2); - - rly[27] = tmp10*rly[18]-ylmcoef[27]*rly[11]*r2;//l=5,m=-1 - grly[81] = tmp10*grly[54]-ylmcoef[27]*(rly[11]*tx+grly[33]*r2); - grly[82] = tmp10*grly[55]-ylmcoef[27]*(rly[11]*ty+grly[34]*r2); - grly[83] = ylmcoef[26]*(z*grly[56]+rly[18])-ylmcoef[27]*(rly[11]*tz+grly[35]*r2); - - double tmp11 = ylmcoef[28]*z; - rly[28] = tmp11*rly[19]-ylmcoef[29]*rly[12]*r2;//l=5,m=2 - grly[84] = tmp11*grly[57]-ylmcoef[29]*(rly[12]*tx+grly[36]*r2); - grly[85] = tmp11*grly[58]-ylmcoef[29]*(rly[12]*ty+grly[37]*r2); - grly[86] = ylmcoef[28]*(z*grly[59]+rly[19])-ylmcoef[29]*(rly[12]*tz+grly[38]*r2); - - rly[29] = tmp11*rly[20]-ylmcoef[29]*rly[13]*r2;//l=5,m=-2 - grly[87] = tmp11*grly[60]-ylmcoef[29]*(rly[13]*tx+grly[39]*r2); - grly[88] = tmp11*grly[61]-ylmcoef[29]*(rly[13]*ty+grly[40]*r2); - grly[89] = ylmcoef[28]*(z*grly[62]+rly[20])-ylmcoef[29]*(rly[13]*tz+grly[41]*r2); - - double tmp12 = ylmcoef[30]*z; - rly[30] = tmp12*rly[21]-ylmcoef[31]*rly[14]*r2;//l=5,m=3 - grly[90] = tmp12*grly[63]-ylmcoef[31]*(grly[42]*r2+rly[14]*tx); - grly[91] = tmp12*grly[64]-ylmcoef[31]*(grly[43]*r2+rly[14]*ty); - grly[92] = ylmcoef[30]*(z*grly[65]+rly[21])-ylmcoef[31]*(grly[44]*r2+rly[14]*tz); - - rly[31] = tmp12*rly[22]-ylmcoef[31]*rly[15]*r2;//l=5,m=-3 - grly[93] = tmp12*grly[66]-ylmcoef[31]*(grly[45]*r2+rly[15]*tx); - grly[94] = tmp12*grly[67]-ylmcoef[31]*(grly[46]*r2+rly[15]*ty); - grly[95] = ylmcoef[30]*(z*grly[68]+rly[22])-ylmcoef[31]*(grly[47]*r2+rly[15]*tz); - - double tmp13 = ylmcoef[32]*z; - rly[32] = tmp13*rly[23];//l=5,m=4 - grly[96] = tmp13*grly[69]; - grly[97] = tmp13*grly[70]; - grly[98] = ylmcoef[32]*(rly[23]+z*grly[71]); - - rly[33] = tmp13*rly[24];//l=5,m=-4 - grly[99] = tmp13*grly[72]; - grly[100] = tmp13*grly[73]; - grly[101] = ylmcoef[32]*(rly[24]+z*grly[74]); - - double tmp14 = ylmcoef[35]*x; - rly[34] = ylmcoef[33]*rly[30]-ylmcoef[34]*rly[14]*r2-tmp14*rly[23];//l=5,m=5 - grly[102] = ylmcoef[33]*grly[90]-ylmcoef[34]*(rly[14]*tx+grly[42]*r2)-ylmcoef[35]*(x*grly[69]+rly[23]); - grly[103] = ylmcoef[33]*grly[91]-ylmcoef[34]*(rly[14]*ty+grly[43]*r2)-tmp14*grly[70]; - grly[104] = ylmcoef[33]*grly[92]-ylmcoef[34]*(rly[14]*tz+grly[44]*r2)-tmp14*grly[71]; - - rly[35] = ylmcoef[33]*rly[31]-ylmcoef[34]*rly[15]*r2-tmp14*rly[24];//l=5,m=-5 - grly[105] = ylmcoef[33]*grly[93]-ylmcoef[34]*(rly[15]*tx+grly[45]*r2)-ylmcoef[35]*(x*grly[72]+rly[24]); - grly[106] = ylmcoef[33]*grly[94]-ylmcoef[34]*(rly[15]*ty+grly[46]*r2)-tmp14*grly[73]; - grly[107] = ylmcoef[33]*grly[95]-ylmcoef[34]*(rly[15]*tz+grly[47]*r2)-tmp14*grly[74]; - - if (nwl == 5) return; + /*************************** + L = 0 + ***************************/ + rly[0] = ylmcoef[0]; //l=0, m=0 + grly[0] = grly[1] = grly[2] = 0.0; + if (nwl == 0) return; + + /*************************** + L = 1 + ***************************/ + rly[1] = ylmcoef[1]*z; //l=1, m=0 + grly[3] = grly[4] = 0.0; + grly[5] = ylmcoef[1]; + + rly[2] = -ylmcoef[1]*x; //l=1, m=1 + grly[7] = grly[8] = 0.0; + grly[6] = -ylmcoef[1]; + + rly[3] = -ylmcoef[1]*y; //l=1, m=-1 + grly[9] = grly[11] = 0.0; + grly[10] = -ylmcoef[1]; + + if (nwl == 1) return; + + /*************************** + L = 2 + ***************************/ + rly[4] = ylmcoef[2]*z*rly[1]-ylmcoef[3]*rly[0]*r2;//l=2, m=0 + grly[12] = ylmcoef[2]*z*grly[3]-ylmcoef[3]*(grly[0]*r2+rly[0]*tx);//l=2, m=0 + grly[13] = ylmcoef[2]*z*grly[4]-ylmcoef[3]*(grly[1]*r2+rly[0]*ty);//l=2, m=0 + grly[14] = ylmcoef[2]*(z*grly[5]+rly[1])-ylmcoef[3]*(grly[2]*r2+rly[0]*tz);//l=2, m=0 + + + double tmp0 = ylmcoef[4]*z; + rly[5] = tmp0*rly[2];//l=2,m=1 + grly[15] = tmp0*grly[6]; + grly[16] = tmp0*grly[7]; + grly[17] = ylmcoef[4]*(rly[2]+z*grly[8]); + + rly[6] = tmp0*rly[3];//l=2,m=-1 + grly[18] = tmp0*grly[9]; + grly[19] = tmp0*grly[10]; + grly[20] = ylmcoef[4]*(rly[3]+z*grly[11]); + + double tmp2 = ylmcoef[4]*x; + rly[7]= ylmcoef[5]*rly[4]-ylmcoef[6]*rly[0]*r2 - tmp2*rly[2];//l=2,m=2 + grly[21] = ylmcoef[5]*grly[12]-ylmcoef[6]*(rly[0]*tx+grly[0]*r2)-ylmcoef[4]*(x*grly[6]+rly[2]); + + grly[22] = ylmcoef[5]*grly[13]-ylmcoef[6]*(rly[0]*ty+grly[1]*r2)-tmp2*grly[7]; + grly[23] = ylmcoef[5]*grly[14]-ylmcoef[6]*(rly[0]*tz+grly[2]*r2)-tmp2*grly[8]; + + rly[8] = -tmp2*rly[3]; + grly[24] = -ylmcoef[4]*(rly[3]+x*grly[9]); + grly[25] = -tmp2*grly[10]; + grly[26] = -tmp2*grly[11]; + if (nwl == 2) return; + + /*************************** + L = 3 + ***************************/ + rly[9] = ylmcoef[7]*z*rly[4]-ylmcoef[8]*rly[1]*r2; //l=3, m=0 + grly[27] = ylmcoef[7]*z*grly[12]-ylmcoef[8]*(rly[1]*tx+grly[3]*r2); + grly[28] = ylmcoef[7]*z*grly[13]-ylmcoef[8]*(rly[1]*ty+grly[4]*r2); + grly[29] = ylmcoef[7]*(rly[4]+z*grly[14])-ylmcoef[8]*(rly[1]*tz+grly[5]*r2); + + double tmp3 = ylmcoef[9]*z; + rly[10] = tmp3*rly[5]-ylmcoef[10]*rly[2]*r2;//l=3,m=1 + grly[30] = tmp3*grly[15]-ylmcoef[10]*(grly[6]*r2+rly[2]*tx); + grly[31] = tmp3*grly[16]-ylmcoef[10]*(grly[7]*r2+rly[2]*ty); + grly[32] = ylmcoef[9]*(z*grly[17]+rly[5])-ylmcoef[10]*(grly[8]*r2+rly[2]*tz); + + rly[11] = tmp3*rly[6]-ylmcoef[10]*rly[3]*r2;//l=3,m=-1 + grly[33] = tmp3*grly[18]-ylmcoef[10]*(grly[9]*r2+rly[3]*tx); + grly[34] = tmp3*grly[19]-ylmcoef[10]*(grly[10]*r2+rly[3]*ty); + grly[35] = ylmcoef[9]*(z*grly[20]+rly[6])-ylmcoef[10]*(grly[11]*r2+rly[3]*tz); + + double tmp4 = ylmcoef[11]*z; + rly[12] = tmp4*rly[7];//l=3,m=2 + grly[36] = tmp4*grly[21]; + grly[37] = tmp4*grly[22]; + grly[38] = ylmcoef[11]*(z*grly[23]+rly[7]); + + rly[13] = tmp4*rly[8];//l=3,m=-2 + grly[39] = tmp4*grly[24]; + grly[40] = tmp4*grly[25]; + grly[41] = ylmcoef[11]*(z*grly[26]+rly[8]); + + double tmp5 = ylmcoef[14]*x; + rly[14] = ylmcoef[12]*rly[10]-ylmcoef[13]*rly[2]*r2-tmp5*rly[7];//l=3,m=3 + grly[42] = ylmcoef[12]*grly[30]-ylmcoef[13]*(rly[2]*tx+grly[6]*r2)-ylmcoef[14]*(rly[7]+x*grly[21]); + grly[43] = ylmcoef[12]*grly[31]-ylmcoef[13]*(rly[2]*ty+grly[7]*r2)-tmp5*grly[22]; + grly[44] = ylmcoef[12]*grly[32]-ylmcoef[13]*(rly[2]*tz+grly[8]*r2)-tmp5*grly[23]; + + rly[15] = ylmcoef[12]*rly[11]-ylmcoef[13]*rly[3]*r2-tmp5*rly[8];//l=3,m=-3 + grly[45] = ylmcoef[12]*grly[33]-ylmcoef[13]*(rly[3]*tx+grly[9]*r2)-ylmcoef[14]*(rly[8]+x*grly[24]); + grly[46] = ylmcoef[12]*grly[34]-ylmcoef[13]*(rly[3]*ty+grly[10]*r2)-tmp5*grly[25]; + grly[47] = ylmcoef[12]*grly[35]-ylmcoef[13]*(rly[3]*tz+grly[11]*r2)-tmp5*grly[26]; + if (nwl == 3) return; + + /*************************** + L = 4 + ***************************/ + rly[16] = ylmcoef[15]*z*rly[9]-ylmcoef[16]*rly[4]*r2;//l=4,m=0 + grly[48] = ylmcoef[15]*z*grly[27]-ylmcoef[16]*(rly[4]*tx+grly[12]*r2); + grly[49] = ylmcoef[15]*z*grly[28]-ylmcoef[16]*(rly[4]*ty+grly[13]*r2); + grly[50] = ylmcoef[15]*(z*grly[29]+rly[9])-ylmcoef[16]*(rly[4]*tz+grly[14]*r2); + + double tmp6 = ylmcoef[17]*z; + rly[17] = tmp6*rly[10]-ylmcoef[18]*rly[5]*r2;//l=4,m=1 + grly[51] = tmp6*grly[30]-ylmcoef[18]*(rly[5]*tx+grly[15]*r2); + grly[52] = tmp6*grly[31]-ylmcoef[18]*(rly[5]*ty+grly[16]*r2); + grly[53] = ylmcoef[17]*(z*grly[32]+rly[10])-ylmcoef[18]*(rly[5]*tz+grly[17]*r2); + + rly[18] = tmp6*rly[11]-ylmcoef[18]*rly[6]*r2;//l=4,m=-1 + grly[54] = tmp6*grly[33]-ylmcoef[18]*(rly[6]*tx+grly[18]*r2); + grly[55] = tmp6*grly[34]-ylmcoef[18]*(rly[6]*ty+grly[19]*r2); + grly[56] = ylmcoef[17]*(z*grly[35]+rly[11])-ylmcoef[18]*(rly[6]*tz+grly[20]*r2); + + double tmp7 = ylmcoef[19]*z; + rly[19] = tmp7*rly[12]-ylmcoef[20]*rly[7]*r2;//l=4,m=2 + grly[57] = tmp7*grly[36]-ylmcoef[20]*(rly[7]*tx+grly[21]*r2); + grly[58] = tmp7*grly[37]-ylmcoef[20]*(rly[7]*ty+grly[22]*r2); + grly[59] = ylmcoef[19]*(z*grly[38]+rly[12])-ylmcoef[20]*(rly[7]*tz+grly[23]*r2); + + rly[20] = tmp7*rly[13]-ylmcoef[20]*rly[8]*r2;//l=4,m=-2 + grly[60] = tmp7*grly[39]-ylmcoef[20]*(rly[8]*tx+grly[24]*r2); + grly[61] = tmp7*grly[40]-ylmcoef[20]*(rly[8]*ty+grly[25]*r2); + grly[62] = ylmcoef[19]*(z*grly[41]+rly[13])-ylmcoef[20]*(rly[8]*tz+grly[26]*r2); + + double tmp8 = 3.0*z; + rly[21] = tmp8*rly[14];//l=4,m=3 + grly[63] = tmp8*grly[42]; + grly[64] = tmp8*grly[43]; + grly[65] = 3.0*(z*grly[44]+rly[14]); + + + rly[22] = tmp8*rly[15];//l=4,m=-3 + grly[66] = tmp8*grly[45]; + grly[67] = tmp8*grly[46]; + grly[68] = 3.0*(z*grly[47]+rly[15]); + + double tmp9 = ylmcoef[23]*x; + rly[23] = ylmcoef[21]*rly[19]-ylmcoef[22]*rly[7]*r2-tmp9*rly[14];//l=4,m=4 + grly[69] = ylmcoef[21]*grly[57]-ylmcoef[22]*(rly[7]*tx+grly[21]*r2)-ylmcoef[23]*(x*grly[42]+rly[14]); + grly[70] = ylmcoef[21]*grly[58]-ylmcoef[22]*(rly[7]*ty+grly[22]*r2)-tmp9*grly[43]; + grly[71] = ylmcoef[21]*grly[59]-ylmcoef[22]*(rly[7]*tz+grly[23]*r2)-tmp9*grly[44]; + + rly[24] = ylmcoef[21]*rly[20]-ylmcoef[22]*rly[8]*r2-tmp9*rly[15];//l=4,m=-4 + grly[72] = ylmcoef[21]*grly[60]-ylmcoef[22]*(rly[8]*tx+grly[24]*r2)-ylmcoef[23]*(x*grly[45]+rly[15]); + grly[73] = ylmcoef[21]*grly[61]-ylmcoef[22]*(rly[8]*ty+grly[25]*r2)-tmp9*grly[46]; + grly[74] = ylmcoef[21]*grly[62]-ylmcoef[22]*(rly[8]*tz+grly[26]*r2)-tmp9*grly[47]; + + if (nwl == 4) return; + + /*************************** + L = 5 + ***************************/ + rly[25] = ylmcoef[24]*z*rly[16]-ylmcoef[25]*rly[9]*r2;//l=5,m=0 + grly[75] = ylmcoef[24]*z*grly[48]-ylmcoef[25]*(rly[9]*tx+grly[27]*r2); + grly[76] = ylmcoef[24]*z*grly[49]-ylmcoef[25]*(rly[9]*ty+grly[28]*r2); + grly[77] = ylmcoef[24]*(z*grly[50]+rly[16])-ylmcoef[25]*(rly[9]*tz+grly[29]*r2); + + double tmp10 = ylmcoef[26]*z; + rly[26] = tmp10*rly[17]-ylmcoef[27]*rly[10]*r2;//l=5,m=1 + grly[78] = tmp10*grly[51]-ylmcoef[27]*(rly[10]*tx+grly[30]*r2); + grly[79] = tmp10*grly[52]-ylmcoef[27]*(rly[10]*ty+grly[31]*r2); + grly[80] = ylmcoef[26]*(z*grly[53]+rly[17])-ylmcoef[27]*(rly[10]*tz+grly[32]*r2); + + rly[27] = tmp10*rly[18]-ylmcoef[27]*rly[11]*r2;//l=5,m=-1 + grly[81] = tmp10*grly[54]-ylmcoef[27]*(rly[11]*tx+grly[33]*r2); + grly[82] = tmp10*grly[55]-ylmcoef[27]*(rly[11]*ty+grly[34]*r2); + grly[83] = ylmcoef[26]*(z*grly[56]+rly[18])-ylmcoef[27]*(rly[11]*tz+grly[35]*r2); + + double tmp11 = ylmcoef[28]*z; + rly[28] = tmp11*rly[19]-ylmcoef[29]*rly[12]*r2;//l=5,m=2 + grly[84] = tmp11*grly[57]-ylmcoef[29]*(rly[12]*tx+grly[36]*r2); + grly[85] = tmp11*grly[58]-ylmcoef[29]*(rly[12]*ty+grly[37]*r2); + grly[86] = ylmcoef[28]*(z*grly[59]+rly[19])-ylmcoef[29]*(rly[12]*tz+grly[38]*r2); + + rly[29] = tmp11*rly[20]-ylmcoef[29]*rly[13]*r2;//l=5,m=-2 + grly[87] = tmp11*grly[60]-ylmcoef[29]*(rly[13]*tx+grly[39]*r2); + grly[88] = tmp11*grly[61]-ylmcoef[29]*(rly[13]*ty+grly[40]*r2); + grly[89] = ylmcoef[28]*(z*grly[62]+rly[20])-ylmcoef[29]*(rly[13]*tz+grly[41]*r2); + + double tmp12 = ylmcoef[30]*z; + rly[30] = tmp12*rly[21]-ylmcoef[31]*rly[14]*r2;//l=5,m=3 + grly[90] = tmp12*grly[63]-ylmcoef[31]*(grly[42]*r2+rly[14]*tx); + grly[91] = tmp12*grly[64]-ylmcoef[31]*(grly[43]*r2+rly[14]*ty); + grly[92] = ylmcoef[30]*(z*grly[65]+rly[21])-ylmcoef[31]*(grly[44]*r2+rly[14]*tz); + + rly[31] = tmp12*rly[22]-ylmcoef[31]*rly[15]*r2;//l=5,m=-3 + grly[93] = tmp12*grly[66]-ylmcoef[31]*(grly[45]*r2+rly[15]*tx); + grly[94] = tmp12*grly[67]-ylmcoef[31]*(grly[46]*r2+rly[15]*ty); + grly[95] = ylmcoef[30]*(z*grly[68]+rly[22])-ylmcoef[31]*(grly[47]*r2+rly[15]*tz); + + double tmp13 = ylmcoef[32]*z; + rly[32] = tmp13*rly[23];//l=5,m=4 + grly[96] = tmp13*grly[69]; + grly[97] = tmp13*grly[70]; + grly[98] = ylmcoef[32]*(rly[23]+z*grly[71]); + + rly[33] = tmp13*rly[24];//l=5,m=-4 + grly[99] = tmp13*grly[72]; + grly[100] = tmp13*grly[73]; + grly[101] = ylmcoef[32]*(rly[24]+z*grly[74]); + + double tmp14 = ylmcoef[35]*x; + rly[34] = ylmcoef[33]*rly[30]-ylmcoef[34]*rly[14]*r2-tmp14*rly[23];//l=5,m=5 + grly[102] = ylmcoef[33]*grly[90]-ylmcoef[34]*(rly[14]*tx+grly[42]*r2)-ylmcoef[35]*(x*grly[69]+rly[23]); + grly[103] = ylmcoef[33]*grly[91]-ylmcoef[34]*(rly[14]*ty+grly[43]*r2)-tmp14*grly[70]; + grly[104] = ylmcoef[33]*grly[92]-ylmcoef[34]*(rly[14]*tz+grly[44]*r2)-tmp14*grly[71]; + + rly[35] = ylmcoef[33]*rly[31]-ylmcoef[34]*rly[15]*r2-tmp14*rly[24];//l=5,m=-5 + grly[105] = ylmcoef[33]*grly[93]-ylmcoef[34]*(rly[15]*tx+grly[45]*r2)-ylmcoef[35]*(x*grly[72]+rly[24]); + grly[106] = ylmcoef[33]*grly[94]-ylmcoef[34]*(rly[15]*ty+grly[46]*r2)-tmp14*grly[73]; + grly[107] = ylmcoef[33]*grly[95]-ylmcoef[34]*(rly[15]*tz+grly[47]*r2)-tmp14*grly[74]; + + if (nwl == 5) return; } } // namespace ModuleBase + +#endif // ABACUS_SOURCE_BASE_KERNELS_CUDA_SPH_HARM_GPU_CUH diff --git a/source/source_base/kernels/math_ylm_op.cpp b/source/source_base/kernels/math_ylm_op.cpp index 833f84d4086..48ecaebbedf 100644 --- a/source/source_base/kernels/math_ylm_op.cpp +++ b/source/source_base/kernels/math_ylm_op.cpp @@ -1,4 +1,6 @@ #include "source_base/kernels/math_ylm_op.h" + +#include #include "source_base/libm/libm.h" namespace ModuleBase { @@ -39,6 +41,7 @@ struct cal_ylm_real_op FPTYPE* p, FPTYPE* ylm) { + const std::int64_t stride = ng; #ifdef _OPENMP #pragma omp parallel for #endif @@ -47,7 +50,7 @@ struct cal_ylm_real_op // EXPLAIN : if lmax = 1,only use Y00 , output result. //---------------------------------------------------------- if (lmax == 0) { - ylm[0 * ng + ig] = SQRT_INVERSE_FOUR_PI; + ylm[0 * stride + ig] = SQRT_INVERSE_FOUR_PI; continue; } //---------------------------------------------------------- @@ -55,18 +58,19 @@ struct cal_ylm_real_op // NAME : cost = cos(theta),theta and phi are polar angles // NAME : phi //---------------------------------------------------------- - const FPTYPE gmod = sqrt(g[ig * 3 + 0] * g[ig * 3 + 0] + g[ig * 3 + 1] * g[ig * 3 + 1] + g[ig * 3 + 2] * g[ig * 3 + 2]); - FPTYPE cost = gmod < 1.0e-9 ? 0.0 : g[ig * 3 + 2] / gmod; + const std::int64_t g_offset = static_cast(ig) * 3; + const FPTYPE gmod = sqrt(g[g_offset + 0] * g[g_offset + 0] + g[g_offset + 1] * g[g_offset + 1] + g[g_offset + 2] * g[g_offset + 2]); + FPTYPE cost = gmod < 1.0e-9 ? 0.0 : g[g_offset + 2] / gmod; FPTYPE phi; // beware the arc tan, it is defined modulo pi - if (g[ig * 3 + 0] > 1.0e-9) { - phi = atan(g[ig * 3 + 1] / g[ig * 3 + 0]); + if (g[g_offset + 0] > 1.0e-9) { + phi = atan(g[g_offset + 1] / g[g_offset + 0]); } - else if (g[ig * 3 + 0] < -1.e-9) { - phi = atan(g[ig * 3 + 1] / g[ig * 3 + 0]) + PI; + else if (g[g_offset + 0] < -1.e-9) { + phi = atan(g[g_offset + 1] / g[g_offset + 0]) + PI; } else { - phi = PI_HALF * ((g[ig * 3 + 1] >= 0.0) ? 1.0 : -1.0); //HLX: modified on 10/13/2006 + phi = PI_HALF * ((g[g_offset + 1] >= 0.0) ? 1.0 : -1.0); //HLX: modified on 10/13/2006 } // end if //========================================================== // NAME : p(Legendre Polynomials) (0 <= m <= l) @@ -75,11 +79,11 @@ struct cal_ylm_real_op for (int l = 0; l <= lmax; l++) { const FPTYPE c = sqrt((2 * l + 1) / FOUR_PI); if (l == 0) { - p[0 * (lmax + 1) * ng + 0 * ng + ig] = 1.0; + p[0 * (lmax + 1) * stride + 0 * stride + ig] = 1.0; } else if (l == 1) { - p[0 * (lmax + 1) * ng + 1 * ng + ig] = cost; - p[1 * (lmax + 1) * ng + 1 * ng + ig] = -sqrt(std::max(0.0, 1.0 - cost * cost)); + p[0 * (lmax + 1) * stride + 1 * stride + ig] = cost; + p[1 * (lmax + 1) * stride + 1 * stride + ig] = -sqrt(std::max(0.0, 1.0 - cost * cost)); } else { const int l1 = l - 1, @@ -87,22 +91,22 @@ struct cal_ylm_real_op l3 = 2 * l - 1; // recursion on l for P(:,l,m) for (int m = 0; m <= l2; m++) { // do m = 0, l - 2//mohan modify 2007-10-13 - p[m * (lmax + 1) * ng + l * ng + ig] = - (cost * l3 * p[m * (lmax + 1) * ng + l1 * ng + ig] - - (l1 + m) * p[m * (lmax + 1) * ng + l2 * ng + ig]) / (l - m); + p[m * (lmax + 1) * stride + l * stride + ig] = + (cost * l3 * p[m * (lmax + 1) * stride + l1 * stride + ig] - + (l1 + m) * p[m * (lmax + 1) * stride + l2 * stride + ig]) / (l - m); } // end do - p[l1 * (lmax + 1) * ng + l * ng + ig] = - cost * l3 * p[l1 * (lmax + 1) * ng + l1 * ng + ig]; + p[l1 * (lmax + 1) * stride + l * stride + ig] = + cost * l3 * p[l1 * (lmax + 1) * stride + l1 * stride + ig]; FPTYPE x2 = std::max(0.0, 1.0 - cost * cost); - p[l * (lmax + 1) * ng + l * ng + ig] = __semi_fact(l3) * pow(x2, static_cast(l) / 2.0);//mohan modify 2007-10-13 + p[l * (lmax + 1) * stride + l * stride + ig] = __semi_fact(l3) * pow(x2, static_cast(l) / 2.0);//mohan modify 2007-10-13 if (l % 2 == 1) { - p[l * (lmax + 1) * ng + l * ng + ig] *= -1; + p[l * (lmax + 1) * stride + l * stride + ig] *= -1; } } // end if // Y_lm, m = 0 ++lm; - ylm[lm * ng + ig] = c * p[0 * (lmax + 1) * ng + l * ng + ig]; + ylm[lm * stride + ig] = c * p[0 * (lmax + 1) * stride + l * stride + ig]; for (int m = 1; m <= l; m++) { // Y_lm, m > 0 @@ -112,11 +116,11 @@ struct cal_ylm_real_op FPTYPE sinp, cosp; ModuleBase::libm::sincos(m * phi, &sinp, &cosp); ++lm; - ylm[lm * ng + ig] = same * p[m * (lmax + 1) * ng + l * ng + ig] * cosp; + ylm[lm * stride + ig] = same * p[m * (lmax + 1) * stride + l * stride + ig] * cosp; // Y_lm, m < 0 ++lm; - ylm[lm * ng + ig] = same * p[m * (lmax + 1) * ng + l * ng + ig] * sinp; + ylm[lm * stride + ig] = same * p[m * (lmax + 1) * stride + l * stride + ig] * sinp; } }// end do } @@ -127,4 +131,3 @@ template struct cal_ylm_real_op; template struct cal_ylm_real_op; } // namespace ModuleBase - diff --git a/source/source_base/kernels/rocm/math_ylm_op.hip.cu b/source/source_base/kernels/rocm/math_ylm_op.hip.cu index db64d9c9de3..85b6af8be56 100644 --- a/source/source_base/kernels/rocm/math_ylm_op.hip.cu +++ b/source/source_base/kernels/rocm/math_ylm_op.hip.cu @@ -1,5 +1,7 @@ #include "source_base/kernels/math_ylm_op.h" +#include + #include #include @@ -41,6 +43,7 @@ __global__ void cal_ylm_real( FPTYPE * p, FPTYPE * ylm) { + const std::int64_t stride = ng; int ig = blockIdx.x * blockDim.x + threadIdx.x; if (ig >= ng) {return;} @@ -49,7 +52,7 @@ __global__ void cal_ylm_real( // EXPLAIN : if lmax = 1,only use Y00 , output result. //---------------------------------------------------------- if (lmax == 0) { - ylm[0 * ng + ig] = SQRT_INVERSE_FOUR_PI; + ylm[0 * stride + ig] = SQRT_INVERSE_FOUR_PI; return; } //---------------------------------------------------------- @@ -57,17 +60,18 @@ __global__ void cal_ylm_real( // NAME : cost = cos(theta),theta and phi are polar angles // NAME : phi //---------------------------------------------------------- - const FPTYPE gmod = sqrt(g[ig * 3 + 0] * g[ig * 3 + 0] + g[ig * 3 + 1] * g[ig * 3 + 1] + g[ig * 3 + 2] * g[ig * 3 + 2]); - cost = gmod < 1.0e-9 ? 0.0 : g[ig * 3 + 2] / gmod; + const std::int64_t g_offset = static_cast(ig) * 3; + const FPTYPE gmod = sqrt(g[g_offset + 0] * g[g_offset + 0] + g[g_offset + 1] * g[g_offset + 1] + g[g_offset + 2] * g[g_offset + 2]); + cost = gmod < 1.0e-9 ? 0.0 : g[g_offset + 2] / gmod; // beware the arc tan, it is defined modulo pi - if (g[ig * 3 + 0] > 1.0e-9) { - phi = atan(g[ig * 3 + 1] / g[ig * 3 + 0]); + if (g[g_offset + 0] > 1.0e-9) { + phi = atan(g[g_offset + 1] / g[g_offset + 0]); } - else if (g[ig * 3 + 0] < -1.e-9) { - phi = atan(g[ig * 3 + 1] / g[ig * 3 + 0]) + PI; + else if (g[g_offset + 0] < -1.e-9) { + phi = atan(g[g_offset + 1] / g[g_offset + 0]) + PI; } else { - phi = PI_HALF * ((g[ig * 3 + 1] >= 0.0) ? 1.0 : -1.0); //HLX: modified on 10/13/2006 + phi = PI_HALF * ((g[g_offset + 1] >= 0.0) ? 1.0 : -1.0); //HLX: modified on 10/13/2006 } // end if //========================================================== // NAME : p(Legendre Polynomials) (0 <= m <= l) @@ -76,12 +80,12 @@ __global__ void cal_ylm_real( for (int l = 0; l <= lmax; l++) { const FPTYPE c = sqrt((2 * l + 1) / FOUR_PI); if (l == 0) { - p[0 * (lmax + 1) * ng + 0 * ng + ig] = 1.0; + p[0 * (lmax + 1) * stride + 0 * stride + ig] = 1.0; } else if (l == 1) { - p[0 * (lmax + 1) * ng + 1 * ng + ig] = cost; + p[0 * (lmax + 1) * stride + 1 * stride + ig] = cost; FPTYPE var = (1.0 - cost * cost) > 0.0 ? (1.0 - cost * cost) : 0.0; - p[1 * (lmax + 1) * ng + 1 * ng + ig] = -sqrt(var); + p[1 * (lmax + 1) * stride + 1 * stride + ig] = -sqrt(var); } else { const int l1 = l - 1, @@ -89,22 +93,22 @@ __global__ void cal_ylm_real( l3 = 2 * l - 1; // recursion on l for P(:,l,m) for (int m = 0; m <= l2; m++) { // do m = 0, l - 2//mohan modify 2007-10-13 - p[m * (lmax + 1) * ng + l * ng + ig] = - (cost * l3 * p[m * (lmax + 1) * ng + l1 * ng + ig] - - (l1 + m) * p[m * (lmax + 1) * ng + l2 * ng + ig]) / (l - m); + p[m * (lmax + 1) * stride + l * stride + ig] = + (cost * l3 * p[m * (lmax + 1) * stride + l1 * stride + ig] - + (l1 + m) * p[m * (lmax + 1) * stride + l2 * stride + ig]) / (l - m); } // end do - p[l1 * (lmax + 1) * ng + l * ng + ig] = - cost * l3 * p[l1 * (lmax + 1) * ng + l1 * ng + ig]; + p[l1 * (lmax + 1) * stride + l * stride + ig] = + cost * l3 * p[l1 * (lmax + 1) * stride + l1 * stride + ig]; FPTYPE x2 = (1.0 - cost * cost) > 0.0 ? (1.0 - cost * cost) : 0.0; - p[l * (lmax + 1) * ng + l * ng + ig] = __semi_fact(l3) * pow(x2, static_cast(l) / 2.0);//mohan modify 2007-10-13 + p[l * (lmax + 1) * stride + l * stride + ig] = __semi_fact(l3) * pow(x2, static_cast(l) / 2.0);//mohan modify 2007-10-13 if (l % 2 == 1) { - p[l * (lmax + 1) * ng + l * ng + ig] *= -1; + p[l * (lmax + 1) * stride + l * stride + ig] *= -1; } } // end if // Y_lm, m = 0 ++lm; - ylm[lm * ng + ig] = c * p[0 * (lmax + 1) * ng + l * ng + ig]; + ylm[lm * stride + ig] = c * p[0 * (lmax + 1) * stride + l * stride + ig]; for (int m = 1; m <= l; m++) { // Y_lm, m > 0 @@ -113,11 +117,11 @@ __global__ void cal_ylm_real( __fact(l + m)) * SQRT2; ++lm; - ylm[lm * ng + ig] = same * p[m * (lmax + 1) * ng + l * ng + ig] * cos(m * phi); + ylm[lm * stride + ig] = same * p[m * (lmax + 1) * stride + l * stride + ig] * cos(m * phi); // Y_lm, m < 0 ++lm; - ylm[lm * ng + ig] = same * p[m * (lmax + 1) * ng + l * ng + ig] * sin(m * phi); + ylm[lm * stride + ig] = same * p[m * (lmax + 1) * stride + l * stride + ig] * sin(m * phi); } }// end do } @@ -135,7 +139,7 @@ void cal_ylm_real_op::operator()(const base_dev FPTYPE* p, FPTYPE* ylm) { - int block = (ng + THREADS_PER_BLOCK - 1) / THREADS_PER_BLOCK; + int block = ng / THREADS_PER_BLOCK + (ng % THREADS_PER_BLOCK != 0); hipLaunchKernelGGL(HIP_KERNEL_NAME(cal_ylm_real), dim3(block), dim3(THREADS_PER_BLOCK), 0, 0, ng, lmax, diff --git a/source/source_base/main.cpp b/source/source_base/main.cpp index 9a32f11d289..857b99423ed 100644 --- a/source/source_base/main.cpp +++ b/source/source_base/main.cpp @@ -9,11 +9,11 @@ void calculate(); int main(int argc, char **argv) { - std::cout << "Hello, this is the 'base' module of ABACUS." << std::endl; + std::cout << "Hello, this is the 'base' module of ABACUS." << std::endl; - std::cout << "The module searchs for the neighboring atoms for a given atomic position" << std::endl; + std::cout << "The module searchs for the neighboring atoms for a given atomic position" << std::endl; - std::cout << "Right now, the module is still empty, soon we will have more tests." << std::endl; + std::cout << "Right now, the module is still empty, soon we will have more tests." << std::endl; calculate(); @@ -23,41 +23,41 @@ int main(int argc, char **argv) void calculate() { - std::ofstream ofs("log.txt"); + std::ofstream ofs("log.txt"); // ooo.set_orb_tables(); - ofs.close(); + ofs.close(); - std::cout << "--------------------" << std::endl; - std::cout << " Have a great day! " << std::endl; - std::cout << "--------------------" << std::endl; + std::cout << "--------------------" << std::endl; + std::cout << " Have a great day! " << std::endl; + std::cout << "--------------------" << std::endl; /* - time_t time_start = std::time(NULL); + time_t time_start = std::time(NULL); // ModuleBase::timer::start(); - //---------------------------------------------------------- - // main program for doing electronic structure calculations - //---------------------------------------------------------- + //---------------------------------------------------------- + // main program for doing electronic structure calculations + //---------------------------------------------------------- // Driver DD; // DD.init(); - time_t time_finish= std::time(NULL); + time_t time_finish= std::time(NULL); - // print out information before ABACUS ends - std::cout << "\n START Time : " << ctime(&time_start); - std::cout << " FINISH Time : " << ctime(&time_finish); - std::cout << " TOTAL Time : " << difftime(time_finish, time_start) << std::endl; + // print out information before ABACUS ends + std::cout << "\n START Time : " << ctime(&time_start); + std::cout << " FINISH Time : " << ctime(&time_finish); + std::cout << " TOTAL Time : " << difftime(time_finish, time_start) << std::endl; - double total_time = difftime(time_finish, time_start); - int hour = total_time / 3600; - int mins = ( total_time - 3600 * hour ) / 60; - int secs = total_time - 3600 * hour - 60 * mins ; - std::cout << " Total Time : " << hour << " h " - << mins << " mins " - << secs << " secs "<< std::endl; + double total_time = difftime(time_finish, time_start); + int hour = total_time / 3600; + int mins = ( total_time - 3600 * hour ) / 60; + int secs = total_time - 3600 * hour - 60 * mins ; + std::cout << " Total Time : " << hour << " h " + << mins << " mins " + << secs << " secs "<< std::endl; */ return; diff --git a/source/source_base/math_integral.h b/source/source_base/math_integral.h index 4e32889e5f0..3163a1dce76 100644 --- a/source/source_base/math_integral.h +++ b/source/source_base/math_integral.h @@ -8,7 +8,7 @@ namespace ModuleBase class Integral { - public: + public: Integral(); ~Integral(); @@ -41,13 +41,13 @@ class Integral * @author Peize Lin * @date 2017-10-02 */ - static void Simpson_Integral - ( - const int mesh, - const double * const func, - const double dr, - double &asum - ); + static void Simpson_Integral + ( + const int mesh, + const double * const func, + const double dr, + double &asum + ); /** * @brief simpson integral. diff --git a/source/source_base/math_polyint.cpp b/source/source_base/math_polyint.cpp index ca62ebb72c2..6f09e8703ed 100644 --- a/source/source_base/math_polyint.cpp +++ b/source/source_base/math_polyint.cpp @@ -19,11 +19,11 @@ void PolyInt::Polynomial_Interpolation assert(table_interval>0.0); const double position = x / table_interval; const int iq = static_cast(position); - if(iq>=table_length-4) - { - std::cout << "\n x = " << x; - std::cout << "\n iq = " << iq << " table_length = " << table_length << std::endl; - } + if(iq>=table_length-4) + { + std::cout << "\n x = " << x; + std::cout << "\n iq = " << iq << " table_length = " << table_length << std::endl; + } assert(iq < table_length-4); const double x0 = position - static_cast(iq); @@ -86,7 +86,7 @@ double PolyInt::Polynomial_Interpolation // pengfei Li 2018-3-23 const ModuleBase::realArray &table, const int &dim1, const int &dim2, - const int &dim3, + const int &dim3, const int &table_length, const double &table_interval, const double &x // input value @@ -97,13 +97,13 @@ double PolyInt::Polynomial_Interpolation // pengfei Li 2018-3-23 const double position = x / table_interval; const int iq = static_cast(position); - if(iq>table_length-4) - { - std::cout << "\n x = " << x; - std::cout << "\n table_interval = " << table_interval; - std::cout << "\n iq=" << iq << " table_length = " << table_length << std::endl; - } - assert(iq < table_length-4); + if(iq>table_length-4) + { + std::cout << "\n x = " << x; + std::cout << "\n table_interval = " << table_interval; + std::cout << "\n iq=" << iq << " table_length = " << table_length << std::endl; + } + assert(iq < table_length-4); const double x0 = position - static_cast(iq); const double x1 = 1.0 - x0; const double x2 = 2.0 - x0; @@ -144,11 +144,11 @@ double PolyInt::Polynomial_Interpolation /* const double y= - table[iq] * x1 * x2 * x3 / 6.0 + - table[iq+1] * x0 * x2 * x3 / 2.0 - - table[iq+2] * x1 * x0 * x3 / 2.0 + - table[iq+3] * x1 * x2 * x0 / 6.0 ; - */ + table[iq] * x1 * x2 * x3 / 6.0 + + table[iq+1] * x0 * x2 * x3 / 2.0 - + table[iq+2] * x1 * x0 * x3 / 2.0 + + table[iq+3] * x1 * x2 * x0 / 6.0 ; + */ return x1*x2*(table[iq]*x3+table[iq+3]*x0)/6.0 + x0*x3*(table[iq+1]*x2-table[iq+2]*x1)/2.0; diff --git a/source/source_base/math_polyint.h b/source/source_base/math_polyint.h index b15f50678c3..da6f33c7dcb 100644 --- a/source/source_base/math_polyint.h +++ b/source/source_base/math_polyint.h @@ -10,10 +10,10 @@ namespace ModuleBase class PolyInt { - public: + public: - PolyInt(); - ~PolyInt(); + PolyInt(); + ~PolyInt(); //======================================================== // Polynomial_Interpolation @@ -98,8 +98,8 @@ class PolyInt * @param x [in] the position to be predicted * @return double the predicted value */ - static double Polynomial_Interpolation - ( + static double Polynomial_Interpolation + ( const double *table, const int &table_length, const double &table_interval, diff --git a/source/source_base/math_sphbes.cpp b/source/source_base/math_sphbes.cpp index 43b65073de2..832f1191a62 100644 --- a/source/source_base/math_sphbes.cpp +++ b/source/source_base/math_sphbes.cpp @@ -2,6 +2,7 @@ #include "constants.h" #include #include +#include #include @@ -41,9 +42,9 @@ void Sphbes::BESSJY(double x, double xnu, double *rj, double *ry, double *rjp, d if (x <= 0.0 || xnu < 0.0) { - std::cout << "Sphbes::BESSJY, bad arguments" << std::endl; + std::cout << "Sphbes::BESSJY, bad arguments" << std::endl; //ModuleBase::WARNING_QUIT("Sphbes::BESSJY","bad arguments"); - exit(0); // mohan add 2021-05-06 + exit(0); // mohan add 2021-05-06 } @@ -57,9 +58,9 @@ void Sphbes::BESSJY(double x, double xnu, double *rj, double *ry, double *rjp, d h = xnu * xi; if (h < FPMIN) - { - h = FPMIN; - } + { + h = FPMIN; + } b = xi2 * xnu; @@ -90,9 +91,9 @@ void Sphbes::BESSJY(double x, double xnu, double *rj, double *ry, double *rjp, d } if (i > MAXIT) - { - std::cout << "x too large in bessjy; try asymptotic expansion" << std::endl; - } + { + std::cout << "x too large in bessjy; try asymptotic expansion" << std::endl; + } rjl = isign * FPMIN; @@ -113,9 +114,9 @@ void Sphbes::BESSJY(double x, double xnu, double *rj, double *ry, double *rjp, d } if (rjl == 0.0) - { - rjl = EPS; - } + { + rjl = EPS; + } f = rjpl / rjl; @@ -285,16 +286,16 @@ void Sphbes::BESCHB(double x, double *gam1, double *gam2, double *gampl, double double Sphbes::CHEBEV(double a, double b, double c[], int m, double x) { double d = 0.0; - double dd = 0.0; - double sv = 0.0; - double y = 0.0; - double y2 = 0.0; + double dd = 0.0; + double sv = 0.0; + double y = 0.0; + double y2 = 0.0; int j=0; if ((x - a)*(x - b) > 0.0) - { - std::cout << "x not in range in routine chebev" << std::endl; - } + { + std::cout << "x not in range in routine chebev" << std::endl; + } y2 = 2.0 * (y = (2.0 * x - a - b) / (b - a)); @@ -320,9 +321,9 @@ double Sphbes::Spherical_Bessel_7(const int n, const double &x) if (n < 0 || x <= 0.0) { - std::cout << "Spherical_Bessel_7, bad arguments in sphbes" << std::endl; + std::cout << "Spherical_Bessel_7, bad arguments in sphbes" << std::endl; //ModuleBase::WARNING_QUIT("Sphbes::Spherical_Bessel_7","bad arguments in sphbes"); - exit(0); + exit(0); } order = n + 0.5; @@ -349,17 +350,17 @@ void Sphbes::Spherical_Bessel_Roots { //ModuleBase::TITLE("Sphbes","Spherical_Bessel_Roots"); if (num<=0) - { - std::cout << "Spherical_Bessel_Roots, num<=0" << std::endl; - //ModuleBase::WARNING_QUIT("Sphbes::Spherical_Bessel_Roots","num<=0"); - exit(0); - } + { + std::cout << "Spherical_Bessel_Roots, num<=0" << std::endl; + //ModuleBase::WARNING_QUIT("Sphbes::Spherical_Bessel_Roots","num<=0"); + exit(0); + } if (rcut<=0.0) - { - std::cout << "Spherical_Bessel_Roots, rcut<=0" << std::endl; - //ModuleBase::WARNING_QUIT("Sphbes::Spherical_Bessel_Roots","rcut<=0.0"); - exit(0); - } + { + std::cout << "Spherical_Bessel_Roots, rcut<=0" << std::endl; + //ModuleBase::WARNING_QUIT("Sphbes::Spherical_Bessel_Roots","rcut<=0.0"); + exit(0); + } double min = 0.0; double max = 2*ModuleBase::PI + (num + (l+0.5)/2 + 0.75)*ModuleBase::PI/2 + @@ -443,8 +444,8 @@ void Sphbes::Spherical_Bessel double x1=0.0; int i=0; - int ir=0; - int ir0=0; + int ir=0; + int ir0=0; if (l>=7) { @@ -608,7 +609,7 @@ void Sphbes::Spherical_Bessel else { std::cout << "\n error in sph_bes, l out of {-1 ... 6},l = " << l ; - exit(0); + exit(0); } } @@ -618,23 +619,23 @@ void Sphbes::Spherical_Bessel void Sphbes::Spherical_Bessel ( - const int &msh, //number of grid points - const double *r,//radial grid - const double &q, // - const int &l, //angular momentum - double *sj, //jl(1:msh) = j_l(q*r(i)),spherical bessel function - double *sjp + const int &msh, //number of grid points + const double *r,//radial grid + const double &q, // + const int &l, //angular momentum + double *sj, //jl(1:msh) = j_l(q*r(i)),spherical bessel function + double *sjp ) { - //calculate jlx first - Spherical_Bessel (msh, r, q, l, sj); + //calculate jlx first + Spherical_Bessel (msh, r, q, l, sj); - for (int ir = 0; ir < msh; ir++) - { - sjp[ir] = 1.0; - } - return; + for (int ir = 0; ir < msh; ir++) + { + sjp[ir] = 1.0; + } + return; } void Sphbes::dSpherical_Bessel_dx @@ -648,8 +649,8 @@ void Sphbes::dSpherical_Bessel_dx { if (l < 0 ) { - std::cout << "We temporarily only calculate derivative of l >= 0." << std::endl; - exit(0); + std::cout << "We temporarily only calculate derivative of l >= 0." << std::endl; + exit(0); } double djl0 = 0; @@ -844,7 +845,8 @@ void Sphbes::sphbes_zeros(const int l, const int n, double* const zeros, const b // to ensure n zeros of j_l; otherwise with a stride of 2 one only // needs to start with n+(l+1)/2 zeros of j_0 int nz = n + ( return_all ? l : (l+1)/2 ); - double* buffer = new double[nz]; + std::vector buffer_storage(nz); + double* buffer = buffer_storage.data(); // zeros of j_0 = sin(x)/x is just n*pi double PI = std::acos(-1.0); @@ -881,7 +883,6 @@ void Sphbes::sphbes_zeros(const int l, const int n, double* const zeros, const b } std::copy(buffer, buffer + n, zeros + offset); - delete[] buffer; } double Sphbes::illinois(std::function func, double x0, double x1, const double tol, const int max_iter) diff --git a/source/source_base/math_sphbes.h b/source/source_base/math_sphbes.h index 7aa9c78a48c..c14829af56f 100644 --- a/source/source_base/math_sphbes.h +++ b/source/source_base/math_sphbes.h @@ -11,7 +11,7 @@ namespace ModuleBase class Sphbes { - public: + public: Sphbes(); ~Sphbes(); @@ -62,25 +62,25 @@ class Sphbes * @param jl [out] jl spherical bessel function * @param sjp [out] sjp[i] is assigned to be 1.0. i < msh. */ - static void Spherical_Bessel - ( - const int &msh, - const double *r, - const double &q, - const int &l, - double *sj, - double *sjp - ); - - /** - * @brief return num eigenvalues of spherical bessel function - * - * @param num [in] the number of eigenvalues - * @param l [in] angular number - * @param epsilon [in] the accuracy - * @param eigenvalue [out] the calculated eigenvalues - * @param rcut [in] the cutoff the radial function - */ + static void Spherical_Bessel + ( + const int &msh, + const double *r, + const double &q, + const int &l, + double *sj, + double *sjp + ); + + /** + * @brief return num eigenvalues of spherical bessel function + * + * @param num [in] the number of eigenvalues + * @param l [in] angular number + * @param epsilon [in] the accuracy + * @param eigenvalue [out] the calculated eigenvalues + * @param rcut [in] the cutoff the radial function + */ static void Spherical_Bessel_Roots ( const int &num, @@ -144,7 +144,7 @@ class Sphbes static double Spherical_Bessel_7(const int n, const double &x); - // Peize Lin change double to void 2019-05-01 + // Peize Lin change double to void 2019-05-01 static void BESSJY(double x, double xnu, double *rj, double *ry, double *rjp, double *ryp); static void BESCHB(double x, double *gam1, double *gam2, double *gampl, double *gammi); diff --git a/source/source_base/math_ylmreal.cpp b/source/source_base/math_ylmreal.cpp index 47415de44d1..02ad16fc9b4 100644 --- a/source/source_base/math_ylmreal.cpp +++ b/source/source_base/math_ylmreal.cpp @@ -10,6 +10,7 @@ #include "ylm.h" #include +#include #include namespace ModuleBase @@ -23,220 +24,220 @@ void YlmReal::rlylm const int lmax, const double& x, const double& y, - const double& z, // g_cartesian_vec(x,y,z) + const double& z, // g_cartesian_vec(x,y,z) double* rly // output ) { - ModuleBase::timer::start("YlmReal","rlylm"); - - assert(lmax >= 0); - - //get xy_dependence - assert(lmax <= 19); - - double Am[20]; - double Bm[20]; - - // mohan add 2021-05-07 - for(int i=0; i<20; ++i) - { - Am[i]=0.0; - Bm[i]=0.0; - } - - //ZEROS(Am, 20); - //ZEROS(Bm, 20); - - double x2, x3, x4, x5; - double y2, y3, y4, y5; - - x2 = x * x; - x3 = x2 * x; - x4 = x3 * x; - x5 = x4 * x; - - y2 = y * y; - y3 = y2 * y; - y4 = y3 * y; - y5 = y4 * y; - - //x-y dependence - //Am - //Bm - for(int im = 0; im < lmax+1; im++) - { - if(im == 0) - { - Am[0] = 1.0; - Bm[0] = 0.0; - } - else if(im == 1) - { - Am[1] = x; - Bm[1] = y; - } - else if(im == 2) - { - Am[2] = x2- y2; - Bm[2] = 2.0 * x * y; - } - else if(im == 3) - { - Am[3] = x3 - 3.0 * x * y2; - Bm[3] = 3.0 * x2 * y - y3; - } - else if(im == 4) - { - Am[4] = x4 - 6.0 * x2 * y2 + y4; - Bm[4] = 4.0 * (x3 * y - x * y3); - } - else if(im == 5) - { - Am[5] = x5 - 10.0 * x3 * y2 + 5.0 * x * y4; - Bm[5] = 5.0 * x4 * y - 10.0 * x2 * y3 + y5; - } - else - { - for(int ip = 0; ip <= im; ip++) - { - double aux = Fact(im) / Fact(ip) / Fact(im - ip); - Am[im] += aux * pow(x, ip) * pow(y, im-ip) * cos( (im-ip) * ModuleBase::PI / 2.0 ); - Bm[im] += aux * pow(x, ip) * pow(y, im-ip) * sin( (im-ip) * ModuleBase::PI / 2.0 ); - } - } - } - - //z dependence - double zdep[20][20]; - - for(int il = 0; il < 20; il++) - { - for(int jl=0; jl < 20; jl++) - { - zdep[il][jl]=0.0; // mohan add 2021-05-07 - } + ModuleBase::timer::start("YlmReal","rlylm"); + + assert(lmax >= 0); + + //get xy_dependence + assert(lmax <= 19); + + double Am[20]; + double Bm[20]; + + // mohan add 2021-05-07 + for(int i=0; i<20; ++i) + { + Am[i]=0.0; + Bm[i]=0.0; + } + + //ZEROS(Am, 20); + //ZEROS(Bm, 20); + + double x2, x3, x4, x5; + double y2, y3, y4, y5; + + x2 = x * x; + x3 = x2 * x; + x4 = x3 * x; + x5 = x4 * x; + + y2 = y * y; + y3 = y2 * y; + y4 = y3 * y; + y5 = y4 * y; + + //x-y dependence + //Am + //Bm + for(int im = 0; im < lmax+1; im++) + { + if(im == 0) + { + Am[0] = 1.0; + Bm[0] = 0.0; + } + else if(im == 1) + { + Am[1] = x; + Bm[1] = y; + } + else if(im == 2) + { + Am[2] = x2- y2; + Bm[2] = 2.0 * x * y; + } + else if(im == 3) + { + Am[3] = x3 - 3.0 * x * y2; + Bm[3] = 3.0 * x2 * y - y3; + } + else if(im == 4) + { + Am[4] = x4 - 6.0 * x2 * y2 + y4; + Bm[4] = 4.0 * (x3 * y - x * y3); + } + else if(im == 5) + { + Am[5] = x5 - 10.0 * x3 * y2 + 5.0 * x * y4; + Bm[5] = 5.0 * x4 * y - 10.0 * x2 * y3 + y5; + } + else + { + for(int ip = 0; ip <= im; ip++) + { + double aux = Fact(im) / Fact(ip) / Fact(im - ip); + Am[im] += aux * pow(x, ip) * pow(y, im-ip) * cos( (im-ip) * ModuleBase::PI / 2.0 ); + Bm[im] += aux * pow(x, ip) * pow(y, im-ip) * sin( (im-ip) * ModuleBase::PI / 2.0 ); + } + } + } + + //z dependence + double zdep[20][20]; + + for(int il = 0; il < 20; il++) + { + for(int jl=0; jl < 20; jl++) + { + zdep[il][jl]=0.0; // mohan add 2021-05-07 + } // ZEROS(zdep[il], 20); - } - - double z2 = z * z; - double z3 = z2 * z; - double z4 = z3 * z; - //double z5 = z4 * z; - - double r = sqrt(x*x + y*y + z*z); - double r2 = r * r; - double r3 = r2 * r; - double r4 = r3 * r; - - for(int il = 0; il < lmax + 1; il++) - { - if(il == 0) - { - zdep[0][0] = 1.0; - } - else if(il == 1) - { - zdep[1][0] = z; - zdep[1][1] = 1.0; - } - else if(il == 2) - { - zdep[2][0] = 0.5 * (3.0 * z2 - r2); - zdep[2][1] = sqrt(3.0) * z; - zdep[2][2] = sqrt(3.0) * 0.5; - } - else if(il == 3) - { - zdep[3][0] = 2.5 * z3 - 1.5 * z * r2; - zdep[3][1] = 0.25 * sqrt(6.0) * (5.0 * z2 - r2); - zdep[3][2] = 0.5 * sqrt(15.0) * z; - zdep[3][3] = 0.25 * sqrt(10.0); - } - else if(il == 4) - { - zdep[4][0] = 0.125 * (35.0 * z4 - 30.0 * r2 * z2 + 3.0 * r4); - zdep[4][1] = sqrt(10.0) * 0.25 * z * (7.0 * z2 - 3.0 * r2); - zdep[4][2] = sqrt(5.0) * 0.25 * (7.0 * z2 - r2); - zdep[4][3] = sqrt(70.0) * 0.25 * z; - zdep[4][4] = sqrt(35.0) * 0.125; - } - else if(il == 5) - { - zdep[5][0] = 0.125 * z *( 63.0 * z4 - 70.0 * z2 * r2 + 15.0 * r4); - zdep[5][1] = 0.125 * sqrt(15.0) * (21.0 * z4 - 14.0 * z2 * r2 + r4); - zdep[5][2] = 0.25 * sqrt(105.0) * z * (3.0 * z2 - r2); - zdep[5][3] = 0.0625 * sqrt(70.0) * (9.0 * z2 - r2); - zdep[5][4] = 0.375 * sqrt(35.0) * z; - zdep[5][5] = 0.1875 * sqrt(14.0); - } - else - { - for(int im = 0; im <= il; im++) - { - int kmax = static_cast( (il - im) / 2 ); - for(int ik = 0; ik <= kmax; ik++) - { - int twok = 2 * ik; - - double gamma = 0.0; - double aux0, aux1, aux2, aux3; - - aux0 = pow(-1.0, ik) * pow(2.0, -il); - aux1 = Fact(il) / Fact(ik) / Fact(il-ik); - aux2 = Fact(2*il - twok) / Fact(il) / Fact(il - twok); - aux3 = Fact(il - twok) / Fact(il - twok - im); - - gamma = aux0 * aux1 * aux2 * aux3; - - assert(il - twok - im >= 0); - zdep[il][im] += pow(r, twok) * pow(z, il-twok-im) * gamma; - } - - if(im >= 1) - { - zdep[il][im] *= sqrt(2 * Fact(il - im) / Fact(il + im)); - - } - } - } - } - - //calc - int ic = 0; - - //special case for r=0 - double rpi = r; - const double tiny = 1.0E-10; - if (rpi < tiny) rpi += tiny; - - for(int il = 0; il <= lmax; il++) - { - double fac = sqrt( (2.0 * il + 1.0) / ModuleBase::FOUR_PI ); - - double rl = pow(rpi, il); - - //m=0 - rly[ic] = Am[0] * zdep[il][0] * fac / rl; - - ic++; - - //m ! = 0 - for(int im = 1; im <= il; im++) - { - //m>0 - rly[ic] = Am[im] * zdep[il][im] * pow(-1.0, im) * fac / rl; - - ic++; - - //m<0 - rly[ic] = Bm[im] * zdep[il][im] * pow(-1.0, im) * fac / rl; - - ic++; - } - } - - ModuleBase::timer::end("YlmReal","rlylm"); - return; + } + + double z2 = z * z; + double z3 = z2 * z; + double z4 = z3 * z; + //double z5 = z4 * z; + + double r = sqrt(x*x + y*y + z*z); + double r2 = r * r; + double r3 = r2 * r; + double r4 = r3 * r; + + for(int il = 0; il < lmax + 1; il++) + { + if(il == 0) + { + zdep[0][0] = 1.0; + } + else if(il == 1) + { + zdep[1][0] = z; + zdep[1][1] = 1.0; + } + else if(il == 2) + { + zdep[2][0] = 0.5 * (3.0 * z2 - r2); + zdep[2][1] = sqrt(3.0) * z; + zdep[2][2] = sqrt(3.0) * 0.5; + } + else if(il == 3) + { + zdep[3][0] = 2.5 * z3 - 1.5 * z * r2; + zdep[3][1] = 0.25 * sqrt(6.0) * (5.0 * z2 - r2); + zdep[3][2] = 0.5 * sqrt(15.0) * z; + zdep[3][3] = 0.25 * sqrt(10.0); + } + else if(il == 4) + { + zdep[4][0] = 0.125 * (35.0 * z4 - 30.0 * r2 * z2 + 3.0 * r4); + zdep[4][1] = sqrt(10.0) * 0.25 * z * (7.0 * z2 - 3.0 * r2); + zdep[4][2] = sqrt(5.0) * 0.25 * (7.0 * z2 - r2); + zdep[4][3] = sqrt(70.0) * 0.25 * z; + zdep[4][4] = sqrt(35.0) * 0.125; + } + else if(il == 5) + { + zdep[5][0] = 0.125 * z *( 63.0 * z4 - 70.0 * z2 * r2 + 15.0 * r4); + zdep[5][1] = 0.125 * sqrt(15.0) * (21.0 * z4 - 14.0 * z2 * r2 + r4); + zdep[5][2] = 0.25 * sqrt(105.0) * z * (3.0 * z2 - r2); + zdep[5][3] = 0.0625 * sqrt(70.0) * (9.0 * z2 - r2); + zdep[5][4] = 0.375 * sqrt(35.0) * z; + zdep[5][5] = 0.1875 * sqrt(14.0); + } + else + { + for(int im = 0; im <= il; im++) + { + int kmax = static_cast( (il - im) / 2 ); + for(int ik = 0; ik <= kmax; ik++) + { + int twok = 2 * ik; + + double gamma = 0.0; + double aux0, aux1, aux2, aux3; + + aux0 = pow(-1.0, ik) * pow(2.0, -il); + aux1 = Fact(il) / Fact(ik) / Fact(il-ik); + aux2 = Fact(2*il - twok) / Fact(il) / Fact(il - twok); + aux3 = Fact(il - twok) / Fact(il - twok - im); + + gamma = aux0 * aux1 * aux2 * aux3; + + assert(il - twok - im >= 0); + zdep[il][im] += pow(r, twok) * pow(z, il-twok-im) * gamma; + } + + if(im >= 1) + { + zdep[il][im] *= sqrt(2 * Fact(il - im) / Fact(il + im)); + + } + } + } + } + + //calc + int ic = 0; + + //special case for r=0 + double rpi = r; + const double tiny = 1.0E-10; + if (rpi < tiny) rpi += tiny; + + for(int il = 0; il <= lmax; il++) + { + double fac = sqrt( (2.0 * il + 1.0) / ModuleBase::FOUR_PI ); + + double rl = pow(rpi, il); + + //m=0 + rly[ic] = Am[0] * zdep[il][0] * fac / rl; + + ic++; + + //m ! = 0 + for(int im = 1; im <= il; im++) + { + //m>0 + rly[ic] = Am[im] * zdep[il][im] * pow(-1.0, im) * fac / rl; + + ic++; + + //m<0 + rly[ic] = Bm[im] * zdep[il][im] * pow(-1.0, im) * fac / rl; + + ic++; + } + } + + ModuleBase::timer::end("YlmReal","rlylm"); + return; } @@ -276,19 +277,19 @@ void YlmReal::Ylm_Real2 //---------------------------------------------------------- // Start CALC //---------------------------------------------------------- - std::vector rly(lmax2); - - for (int ig = 0; ig < ng; ig++) - { - rlylm (lmax, g[ig].x, g[ig].y, g[ig].z, rly.data()); - - for (int lm = 0; lm < lmax2; lm++) - { - ylm (lm, ig) = rly[lm]; - } - } - - return; + std::vector rly(lmax2); + + for (int ig = 0; ig < ng; ig++) + { + rlylm (lmax, g[ig].x, g[ig].y, g[ig].z, rly.data()); + + for (int lm = 0; lm < lmax2; lm++) + { + ylm (lm, ig) = rly[lm]; + } + } + + return; } //========================================================== @@ -326,7 +327,8 @@ void YlmReal::Ylm_Real(Device * ctx, const int lmax2, const int ng, const FPTYPE ModuleBase::WARNING_QUIT("YLM_REAL","l>30 or l<0"); } FPTYPE * p = nullptr, * phi = nullptr, * cost = nullptr; - resmem_var_op()(p, (lmax + 1) * (lmax + 1) * ng, "YlmReal::Ylm_Real"); + const std::int64_t p_elements = static_cast(lmax2) * ng; + resmem_var_op()(p, p_elements, "YlmReal::Ylm_Real"); cal_ylm_real_op()( ctx, @@ -406,8 +408,8 @@ void YlmReal::Ylm_Real // NAME : cost = cos(theta),theta and phi are polar angles // NAME : phi //---------------------------------------------------------- - std::vector cost(ng); - std::vector phi(ng); + std::vector cost(ng); + std::vector phi(ng); #ifdef _OPENMP #pragma omp parallel for @@ -529,10 +531,10 @@ void YlmReal::Ylm_Real #endif for (int i=0;i *g, - matrix &ylm, + matrix &ylm, matrix &dylmx, - matrix &dylmy, - matrix &dylmz + matrix &dylmy, + matrix &dylmz ) { - ModuleBase::Ylm::set_coefficients(); - const int lmax = int(sqrt( double(lmax2) ) + 0.1) - 1; - std::vector tmpylm((lmax2+1) * (lmax2+1)); - std::vector tmpgylm((lmax2+1) * (lmax2+1) * 3); + ModuleBase::Ylm::set_coefficients(); + const int lmax = int(sqrt( double(lmax2) ) + 0.1) - 1; + std::vector tmpylm((lmax2+1) * (lmax2+1)); + std::vector tmpgylm((lmax2+1) * (lmax2+1) * 3); - for (int ig = 0;ig < ng;ig++) + for (int ig = 0;ig < ng;ig++) { - ModuleBase::Vector3 gg = g[ig]; + ModuleBase::Vector3 gg = g[ig]; double gmod = gg.norm(); if (gmod < 1.0e-9) { - for(int lm = 0 ; lm < lmax2 ; ++lm) - { - if(lm == 0) - ylm(lm,ig) = ModuleBase::SQRT_INVERSE_FOUR_PI; - else - ylm(lm,ig) = 0; - dylmx(lm,ig) = dylmy(lm,ig) = dylmz(lm,ig) = 0; - } - } - else - { - Ylm::grad_rl_sph_harm(lmax2, gg.x, gg.y, gg.z, tmpylm.data(), tmpgylm.data()); - int lm = 0; - for(int il = 0 ; il <= lmax ; ++il) - { - for(int im = 0; im < 2*il+1; ++im, ++lm) - { - double rlylm = tmpylm[lm]; - ylm(lm,ig) = rlylm / pow(gmod,il); - dylmx(lm,ig) = ( tmpgylm[lm*3] - il*rlylm * gg.x / pow(gmod,2) )/pow(gmod,il); - dylmy(lm,ig) = ( tmpgylm[lm*3 + 1] - il*rlylm * gg.y / pow(gmod,2) )/pow(gmod,il); - dylmz(lm,ig) = ( tmpgylm[lm*3 + 2] - il*rlylm * gg.z / pow(gmod,2) )/pow(gmod,il); - } - } - - } - } - return; + for(int lm = 0 ; lm < lmax2 ; ++lm) + { + if(lm == 0) + ylm(lm,ig) = ModuleBase::SQRT_INVERSE_FOUR_PI; + else + ylm(lm,ig) = 0; + dylmx(lm,ig) = dylmy(lm,ig) = dylmz(lm,ig) = 0; + } + } + else + { + Ylm::grad_rl_sph_harm(lmax2, gg.x, gg.y, gg.z, tmpylm.data(), tmpgylm.data()); + int lm = 0; + for(int il = 0 ; il <= lmax ; ++il) + { + for(int im = 0; im < 2*il+1; ++im, ++lm) + { + double rlylm = tmpylm[lm]; + ylm(lm,ig) = rlylm / pow(gmod,il); + dylmx(lm,ig) = ( tmpgylm[lm*3] - il*rlylm * gg.x / pow(gmod,2) )/pow(gmod,il); + dylmy(lm,ig) = ( tmpgylm[lm*3 + 1] - il*rlylm * gg.y / pow(gmod,2) )/pow(gmod,il); + dylmz(lm,ig) = ( tmpgylm[lm*3 + 2] - il*rlylm * gg.z / pow(gmod,2) )/pow(gmod,il); + } + } + + } + } + return; } diff --git a/source/source_base/math_ylmreal.h b/source/source_base/math_ylmreal.h index 9643196a21e..b84dc911260 100644 --- a/source/source_base/math_ylmreal.h +++ b/source/source_base/math_ylmreal.h @@ -9,19 +9,19 @@ namespace ModuleBase class YlmReal { - public: + public: YlmReal(); ~YlmReal(); - /** - * @brief spherical harmonic function (real form) an array of vectors - * - * @param lmax2 [in] lmax2 = (lmax + 1)^2 ; lmax = angular quantum number - * @param ng [in] the number of vectors - * @param g [in] an array of vectors - * @param ylm [out] Ylm; column index represent vector, row index represent Y00, Y10, Y11, Y1-1, Y20,Y21,Y2-1,Y22.Y2-2,...; - */ + /** + * @brief spherical harmonic function (real form) an array of vectors + * + * @param lmax2 [in] lmax2 = (lmax + 1)^2 ; lmax = angular quantum number + * @param ng [in] the number of vectors + * @param g [in] an array of vectors + * @param ylm [out] Ylm; column index represent vector, row index represent Y00, Y10, Y11, Y1-1, Y20,Y21,Y2-1,Y22.Y2-2,...; + */ static void Ylm_Real ( const int lmax2, @@ -31,71 +31,71 @@ class YlmReal ); /** - * @brief spherical harmonic function (real form) an array - * - * @param lmax2 [in] lmax2 = (lmax + 1)^2 ; lmax = angular quantum number - * @param ng [in] the number of vectors - * @param g [in] an array of vectors - * @param ylm [out] Ylm; column index represent vector, row index represent Y00, Y10, Y11, Y1-1, Y20,Y21,Y2-1,Y22.Y2-2,...; - */ - template + * @brief spherical harmonic function (real form) an array + * + * @param lmax2 [in] lmax2 = (lmax + 1)^2 ; lmax = angular quantum number + * @param ng [in] the number of vectors + * @param g [in] an array of vectors + * @param ylm [out] Ylm; column index represent vector, row index represent Y00, Y10, Y11, Y1-1, Y20,Y21,Y2-1,Y22.Y2-2,...; + */ + template static void Ylm_Real(Device * ctx, const int lmax2, const int ng, const FPTYPE *g, FPTYPE * ylm); - /** - * @brief gradient of spherical harmonic function (real form) an array of vectors - * - * @param lmax2 [in] lmax2 = (lmax + 1)^2 ; lmax = angular quantum number - * @param ng [in] the number of vectors - * @param g [in] an array of vectors - * @param ylm [out] Ylm; column index represent vector, row index represent Y00, Y10, Y11, Y1-1, Y20,Y21,Y2-1,Y22.Y2-2,...; - * @param dylmx/dylmy/dylmz [out] \nabla Ylm; column index represent vector, row index represent dY00/dxyz, dY10/dxyz,...; - */ + /** + * @brief gradient of spherical harmonic function (real form) an array of vectors + * + * @param lmax2 [in] lmax2 = (lmax + 1)^2 ; lmax = angular quantum number + * @param ng [in] the number of vectors + * @param g [in] an array of vectors + * @param ylm [out] Ylm; column index represent vector, row index represent Y00, Y10, Y11, Y1-1, Y20,Y21,Y2-1,Y22.Y2-2,...; + * @param dylmx/dylmy/dylmz [out] \nabla Ylm; column index represent vector, row index represent dY00/dxyz, dY10/dxyz,...; + */ static void grad_Ylm_Real ( const int lmax2, const int ng, const ModuleBase::Vector3 *g, - matrix &ylm, + matrix &ylm, matrix &dylmx, - matrix &dylmy, - matrix &dylmz + matrix &dylmy, + matrix &dylmz + ); + + /** + * @brief spherical harmonic function (Herglotz generating form) of an array of vectors + * + * @param lmax2 [in] lmax2 = (lmax + 1)^2 ; lmax = angular quantum number + * @param ng [in] the number of vectors + * @param g [in] an array of vectors + * @param ylm [out] Ylm; column index represent vector, row index represent Y00, Y10, Y11, Y1-1, Y20,Y21,Y2-1,Y22.Y2-2,...; + */ + static void Ylm_Real2 + ( + const int lmax2, + const int ng, + const ModuleBase::Vector3 *g, + matrix &ylm ); - - /** - * @brief spherical harmonic function (Herglotz generating form) of an array of vectors - * - * @param lmax2 [in] lmax2 = (lmax + 1)^2 ; lmax = angular quantum number - * @param ng [in] the number of vectors - * @param g [in] an array of vectors - * @param ylm [out] Ylm; column index represent vector, row index represent Y00, Y10, Y11, Y1-1, Y20,Y21,Y2-1,Y22.Y2-2,...; - */ - static void Ylm_Real2 - ( - const int lmax2, - const int ng, - const ModuleBase::Vector3 *g, - matrix &ylm - ); - /** - * @brief spherical harmonic function (Herglotz generating form) of a vector - * - * @param lmax [in] maximum angular quantum number - * @param x [in] x part of the vector - * @param y [in] y part of the vector - * @param z [in] z part of the vector - * @param rly [in] Ylm, Y00, Y10, Y11, Y1-1, Y20,Y21,Y2-1,Y22.Y2-2,... - */ - static void rlylm - ( - const int lmax, - const double& x, - const double& y, - const double& z, - double* rly - ); + /** + * @brief spherical harmonic function (Herglotz generating form) of a vector + * + * @param lmax [in] maximum angular quantum number + * @param x [in] x part of the vector + * @param y [in] y part of the vector + * @param z [in] z part of the vector + * @param rly [in] Ylm, Y00, Y10, Y11, Y1-1, Y20,Y21,Y2-1,Y22.Y2-2,... + */ + static void rlylm + ( + const int lmax, + const double& x, + const double& y, + const double& z, + double* rly + ); - private: + private: static long double Fact(const int n); static int Semi_Fact(const int n); diff --git a/source/source_base/mathzone.h b/source/source_base/mathzone.h index 80239d58a18..fd06cd59381 100644 --- a/source/source_base/mathzone.h +++ b/source/source_base/mathzone.h @@ -2,7 +2,9 @@ #define MATHZONE_H #include "global_function.h" +#include "matrix.h" #include "matrix3.h" +#include "tool_quit.h" #include "vector3.h" #include "realarray.h" @@ -171,9 +173,42 @@ class Mathzone proj.y = std::abs( latvec[1] * (latvec[2] ^ latvec[0]).normalize() ); proj.z = std::abs( latvec[2] * (latvec[0] ^ latvec[1]).normalize() ); return proj; - } + } }; +/** + * @brief Remove the uniform net-force component in place (Newton's 3rd law). + * + * Subtract the per-component mean force so the total force sums to zero. This + * is a pure matrix operation, independent of symmetry. Shared by the PW and + * LCAO force paths. Not applied when an external field (gate/efield) is present. + * + * @param[in] nat number of atoms + * @param[in,out] force per-atom Cartesian forces, nat x 3, modified in place + */ +inline void remove_net_force(const int nat, ModuleBase::matrix& force) +{ + if (nat <= 0 || force.nr < nat || force.nc < 3) + { + ModuleBase::WARNING_QUIT("remove_net_force", + "nat must be positive and force must have at least nat x 3 elements"); + } + for (int i = 0; i < 3; i++) + { + double sum = 0.0; + for (int iat = 0; iat < nat; iat++) + { + sum += force(iat, i); + } + const double compen = sum / nat; + for (int iat = 0; iat < nat; ++iat) + { + force(iat, i) -= compen; + } + } + return; +} + } // namespace ModuleBase #endif diff --git a/source/source_base/mathzone_add1.cpp b/source/source_base/mathzone_add1.cpp index 7f05961265e..67f1afff785 100644 --- a/source/source_base/mathzone_add1.cpp +++ b/source/source_base/mathzone_add1.cpp @@ -40,376 +40,376 @@ Mathzone_Add1::~Mathzone_Add1() double Mathzone_Add1::factorial (const int& l) { - if (l == 0 || l == 1) return 1.0; - else return l*factorial(l-1); + if (l == 0 || l == 1) return 1.0; + else return l*factorial(l-1); } double Mathzone_Add1::dualfac (const int& l) { - if (l == -1 || l == 0) return 1.0; - else return l * dualfac (l-2); + if (l == -1 || l == 0) return 1.0; + else return l * dualfac (l-2); } void Mathzone_Add1::SplineD2 // modified by pengfei 13-8-8 add second derivative as a condition ( - const double *rad, - const double *rad_f, - const int& mesh, - const double &yp1, // if yp1 > ypmax, consider the second derivative - const double &ypn, - double* y2 + const double *rad, + const double *rad_f, + const int& mesh, + const double &yp1, // if yp1 > ypmax, consider the second derivative + const double &ypn, + double* y2 ) { - ModuleBase::timer::start("Mathzone_Add1","SplineD2"); - - double dx1 = 0.0, dx2 = 0.0, dy1 = 0.0, dy2 = 0.0, p = 0.0, qn = 0.0, sig = 0.0, un = 0.0; - double * u = nullptr; - - u = new double[mesh-1]; - const double ypmax = 99999.00; - - if (yp1 > ypmax) - { - y2[0] = 0.0; - u[0] = 0.0; - } - else - { - y2[0] = -0.5; - - dx1 = rad[1] - rad[0]; - dy1 = rad_f[1] - rad_f[0]; - - u[0] = ( 3.0 / dx1 ) * (dy1 / dx1 - yp1); - } - - for(int i = 1; i < mesh-1 ; i++) - { - dx1 = rad[i] - rad[i-1]; - dx2 = rad[i+1] - rad[i-1]; - dy1 = rad_f[i+1] - rad_f[i]; - dy2 = rad_f[i] - rad_f[i-1]; - - sig = dx1 / dx2; - - p = sig * y2[i-1] + 2.0; - y2[i] = (sig - 1.0) / p; - - u[i] = dy1 / (dx2 - dx1) - dy2 / dx1; - u[i] = (6.0 * u[i] / dx2 - sig * u[i-1]) / p; - } - - if (ypn > ypmax) - { - qn = un = 0.0; - } - else - { - dx1 = rad[mesh-1] - rad[mesh-2]; - dy1 = rad_f[mesh-1] - rad_f[mesh-2]; - - qn = 0.5; - un = 3.0 / dx1 * (ypn - dy1 / dx1); - } - - y2[mesh-1] = (un - qn * u[mesh-2]) / (qn * y2[mesh-2] + 1.0); - - for(int i = mesh-2; i >= 0; i--) - { - y2[i] = y2[i] * y2[i+1] + u[i]; - } - - delete[] u; - - ModuleBase::timer::end("Mathzone_Add1","SplineD2"); + ModuleBase::timer::start("Mathzone_Add1","SplineD2"); + + double dx1 = 0.0, dx2 = 0.0, dy1 = 0.0, dy2 = 0.0, p = 0.0, qn = 0.0, sig = 0.0, un = 0.0; + double * u = nullptr; + + u = new double[mesh-1]; + const double ypmax = 99999.00; + + if (yp1 > ypmax) + { + y2[0] = 0.0; + u[0] = 0.0; + } + else + { + y2[0] = -0.5; + + dx1 = rad[1] - rad[0]; + dy1 = rad_f[1] - rad_f[0]; + + u[0] = ( 3.0 / dx1 ) * (dy1 / dx1 - yp1); + } + + for(int i = 1; i < mesh-1 ; i++) + { + dx1 = rad[i] - rad[i-1]; + dx2 = rad[i+1] - rad[i-1]; + dy1 = rad_f[i+1] - rad_f[i]; + dy2 = rad_f[i] - rad_f[i-1]; + + sig = dx1 / dx2; + + p = sig * y2[i-1] + 2.0; + y2[i] = (sig - 1.0) / p; + + u[i] = dy1 / (dx2 - dx1) - dy2 / dx1; + u[i] = (6.0 * u[i] / dx2 - sig * u[i-1]) / p; + } + + if (ypn > ypmax) + { + qn = un = 0.0; + } + else + { + dx1 = rad[mesh-1] - rad[mesh-2]; + dy1 = rad_f[mesh-1] - rad_f[mesh-2]; + + qn = 0.5; + un = 3.0 / dx1 * (ypn - dy1 / dx1); + } + + y2[mesh-1] = (un - qn * u[mesh-2]) / (qn * y2[mesh-2] + 1.0); + + for(int i = mesh-2; i >= 0; i--) + { + y2[i] = y2[i] * y2[i+1] + u[i]; + } + + delete[] u; + + ModuleBase::timer::end("Mathzone_Add1","SplineD2"); } // Peize Lin add openmp 2019-12-13 void Mathzone_Add1::Cubic_Spline_Interpolation ( - const double * const rad, - const double * const rad_f, - const double * const y2, - const int& mesh, - const double * const r, - const int& rsize, - double * const y, - double * const dy + const double * const rad, + const double * const rad_f, + const double * const y2, + const int& mesh, + const double * const r, + const int& rsize, + double * const y, + double * const dy ) { - ModuleBase::timer::start("Mathzone","cubic_spline"); + ModuleBase::timer::start("Mathzone","cubic_spline"); #ifdef _OPENMP - #pragma omp parallel for schedule(static) + #pragma omp parallel for schedule(static) #endif - for(int m = 0; m < rsize ; m++) - { - int klo = 0; - int khi = mesh-1; - while (khi - klo > 1) - { - const int k = (khi + klo) / 2 ; - if(rad[k] > r[m]) khi = k; - else klo = k; - } - - const double h = rad[khi] - rad[klo]; - if(h == 0.0) - { - std::cout << "Cubic_Spline_Interpolation, h == 0.0 so that cannot be divided" << std::endl; - //ModuleBase::WARNING_QUIT("Cubic_Spline_Interpolation","h == 0.0 so that cannot be divided"); - exit(0); - } - - const double a = (rad[khi] - r[m]) / h; - const double b = (r[m] - rad[klo]) / h; - - const double dy_tmp = (rad_f[khi] - rad_f[klo]) / h - - (3.0 * a * a - 1.0) / 6.0 * h * y2[klo] + - ( 3.0 * b * b - 1.0) / 6.0 * h * y2[khi]; - dy[m] = dy_tmp; - const double y_tmp = a * rad_f[klo] + b * rad_f[khi] + ((a*a*a - a) * y2[klo] + (b*b*b - b) * y2[khi]) * (h*h) / 6.0; - y[m] = y_tmp; - //const double ddy_tmp = a * y2[klo] + b * y2 [khi]; - //ddy[m] = ddy_tmp; - } - - ModuleBase::timer::end("Mathzone","cubic_spline"); + for(int m = 0; m < rsize ; m++) + { + int klo = 0; + int khi = mesh-1; + while (khi - klo > 1) + { + const int k = (khi + klo) / 2 ; + if(rad[k] > r[m]) khi = k; + else klo = k; + } + + const double h = rad[khi] - rad[klo]; + if(h == 0.0) + { + std::cout << "Cubic_Spline_Interpolation, h == 0.0 so that cannot be divided" << std::endl; + //ModuleBase::WARNING_QUIT("Cubic_Spline_Interpolation","h == 0.0 so that cannot be divided"); + exit(0); + } + + const double a = (rad[khi] - r[m]) / h; + const double b = (r[m] - rad[klo]) / h; + + const double dy_tmp = (rad_f[khi] - rad_f[klo]) / h - + (3.0 * a * a - 1.0) / 6.0 * h * y2[klo] + + ( 3.0 * b * b - 1.0) / 6.0 * h * y2[khi]; + dy[m] = dy_tmp; + const double y_tmp = a * rad_f[klo] + b * rad_f[khi] + ((a*a*a - a) * y2[klo] + (b*b*b - b) * y2[khi]) * (h*h) / 6.0; + y[m] = y_tmp; + //const double ddy_tmp = a * y2[klo] + b * y2 [khi]; + //ddy[m] = ddy_tmp; + } + + ModuleBase::timer::end("Mathzone","cubic_spline"); } /// Interpolation for Numerical Orbitals double Mathzone_Add1::Uni_RadialF ( - const double* old_phi, - const int& msh, - const double& dr, - const double& newr + const double* old_phi, + const int& msh, + const double& dr, + const double& newr ) { - double h1 = 0.0, h2 = 0.0, h3 = 0.0, f1 = 0.0, f2 = 0.0, f3 = 0.0, f4 = 0.0; - double g1 = 0.0, g2 = 0.0, x1 = 0.0, x2 = 0.0, y1 = 0.0, y2 = 0.0, f = 0.0; - double result = 0.0; - double rmax = (msh-1) * dr; - - if (newr < 0.0) - { - throw std::runtime_error("newr should >= 0. "+std::string(__FILE__)+" line "+std::to_string(__LINE__)); - - } - else if ( rmax <= newr ) - { - f = 0.0; - } - else - { - // the old position. - const int m = static_cast (newr / dr) + 1; + double h1 = 0.0, h2 = 0.0, h3 = 0.0, f1 = 0.0, f2 = 0.0, f3 = 0.0, f4 = 0.0; + double g1 = 0.0, g2 = 0.0, x1 = 0.0, x2 = 0.0, y1 = 0.0, y2 = 0.0, f = 0.0; + double result = 0.0; + double rmax = (msh-1) * dr; + + if (newr < 0.0) + { + throw std::runtime_error("newr should >= 0. "+std::string(__FILE__)+" line "+std::to_string(__LINE__)); + + } + else if ( rmax <= newr ) + { + f = 0.0; + } + else + { + // the old position. + const int m = static_cast (newr / dr) + 1; // Spline like interpolation - if (m == 1) - { - h2 = dr; - h3 = dr; - - f2 = old_phi[m-1]; - f3 = old_phi[m]; - f4 = old_phi[m+1]; - - h1 = -(h2+h3); - f1 = f4; - } - else if (m == (msh-1)) - { - h1 = dr; - h2 = dr; - - f1 = old_phi[m-2]; - f2 = old_phi[m-1]; - f3 = old_phi[m]; - - h3 = -(h1+h2); - f4 = f1; - } - else - { - h1 = dr; - h2 = dr; - h3 = dr; - - f1 = old_phi[m-2]; - f2 = old_phi[m-1]; - f3 = old_phi[m]; - f4 = old_phi[m+1]; - } + if (m == 1) + { + h2 = dr; + h3 = dr; + + f2 = old_phi[m-1]; + f3 = old_phi[m]; + f4 = old_phi[m+1]; + + h1 = -(h2+h3); + f1 = f4; + } + else if (m == (msh-1)) + { + h1 = dr; + h2 = dr; + + f1 = old_phi[m-2]; + f2 = old_phi[m-1]; + f3 = old_phi[m]; + + h3 = -(h1+h2); + f4 = f1; + } + else + { + h1 = dr; + h2 = dr; + h3 = dr; + + f1 = old_phi[m-2]; + f2 = old_phi[m-1]; + f3 = old_phi[m]; + f4 = old_phi[m+1]; + } // Calculate the value at newr - g1 = ((f3-f2)*h1/h2 + (f2-f1)*h2/h1)/(h1+h2); - g2 = ((f4-f3)*h2/h3 + (f3-f2)*h3/h2)/(h2+h3); + g1 = ((f3-f2)*h1/h2 + (f2-f1)*h2/h1)/(h1+h2); + g2 = ((f4-f3)*h2/h3 + (f3-f2)*h3/h2)/(h2+h3); - x1 = newr - (m-1)*dr; - x2 = newr - m*dr; - y1 = x1/h2; - y2 = x2/h2; + x1 = newr - (m-1)*dr; + x2 = newr - m*dr; + y1 = x1/h2; + y2 = x2/h2; - f = y2*y2*(3.0*f2 + h2*g1 + (2.0*f2 + h2*g1)*y2) - + y1*y1*(3.0*f3 - h2*g2 - (2.0*f3 - h2*g2)*y1); - } - result = f; + f = y2*y2*(3.0*f2 + h2*g1 + (2.0*f2 + h2*g1)*y2) + + y1*y1*(3.0*f3 - h2*g2 - (2.0*f3 - h2*g2)*y1); + } + result = f; - return result; + return result; } void Mathzone_Add1::Uni_Deriv_Phi ( - const double *radf, - const int &mesh, - const double &dr, - const int &nd, - double *phind + const double *radf, + const int &mesh, + const double &dr, + const int &nd, + double *phind ) { - ModuleBase::timer::start("Mathzone_Add1", "Uni_Deriv_Phi"); - int FFT_NR = 2*mesh-1; // FFT_NR = 16019 - // std::cout << "\n mesh=" << mesh << ", radf[8010]=" << radf[8010] << ", radf[8009]=" << radf[8009] ; - // mesh=8010, radf[8010]=4.396478951532926e-01, radf[8009]=0.000000000000000e+00 + ModuleBase::timer::start("Mathzone_Add1", "Uni_Deriv_Phi"); + int FFT_NR = 2*mesh-1; // FFT_NR = 16019 + // std::cout << "\n mesh=" << mesh << ", radf[8010]=" << radf[8010] << ", radf[8009]=" << radf[8009] ; + // mesh=8010, radf[8010]=4.396478951532926e-01, radf[8009]=0.000000000000000e+00 fftw_complex *fft_phir = new fftw_complex[FFT_NR]; fftw_complex *fft_phik = new fftw_complex[FFT_NR]; fftw_complex *fft_ndphik = new fftw_complex[FFT_NR]; fftw_complex *fft_ndphir = new fftw_complex[FFT_NR]; - fftw_plan p1; - fftw_plan p2; - - ////CAREFUL: POINT 0 is OF GOOD IMPORTANCE - //for (int ir = 0; ir < FFT_NR/2; ++ir) - //{ - //fft_phir[ir].re = radf[ir]; - //fft_phir[ir].im = 0.0; - //} - - //for (int ir = FFT_NR/2; ir < FFT_NR; ++ir) - //{ - //int jr = FFT_NR - ir; - //fft_phir[ir].re = radf[jr]; - //fft_phir[ir].im = 0.0; - //} - // - // second call: different value at [8010]; FFT_NR = 16019, FFT_NR/2=8009 :: - // CAREFUL: POINT 0 is OF GOOD IMPORTANCE - for (int ir = 0; ir < FFT_NR/2; ++ir) // ik = 0 1 ... 8008 - { - c_re(fft_phir[ir]) = radf[ir]; - c_im(fft_phir[ir]) = 0.0; - } - - for (int ir = FFT_NR/2; ir < FFT_NR; ++ir) // ir = 8009 8010 ... 16018 - { - //int jr = FFT_NR - ir ; // jr = 8010 8009 ... 1 - int jr = FFT_NR - ir -1 ; // -> 8009 8008 ... 0 - c_re(fft_phir[ir]) = radf[jr]; - c_im(fft_phir[ir]) = 0.0; - } - - // FFTW + fftw_plan p1; + fftw_plan p2; + + ////CAREFUL: POINT 0 is OF GOOD IMPORTANCE + //for (int ir = 0; ir < FFT_NR/2; ++ir) + //{ + //fft_phir[ir].re = radf[ir]; + //fft_phir[ir].im = 0.0; + //} + + //for (int ir = FFT_NR/2; ir < FFT_NR; ++ir) + //{ + //int jr = FFT_NR - ir; + //fft_phir[ir].re = radf[jr]; + //fft_phir[ir].im = 0.0; + //} + // + // second call: different value at [8010]; FFT_NR = 16019, FFT_NR/2=8009 :: + // CAREFUL: POINT 0 is OF GOOD IMPORTANCE + for (int ir = 0; ir < FFT_NR/2; ++ir) // ik = 0 1 ... 8008 + { + c_re(fft_phir[ir]) = radf[ir]; + c_im(fft_phir[ir]) = 0.0; + } + + for (int ir = FFT_NR/2; ir < FFT_NR; ++ir) // ir = 8009 8010 ... 16018 + { + //int jr = FFT_NR - ir ; // jr = 8010 8009 ... 1 + int jr = FFT_NR - ir -1 ; // -> 8009 8008 ... 0 + c_re(fft_phir[ir]) = radf[jr]; + c_im(fft_phir[ir]) = 0.0; + } + + // FFTW #if defined __FFTW3 - //std::cout << "\n Call FFTW3 "; - p1 = fftw_plan_dft_1d(FFT_NR, fft_phir, fft_phik, FFTW_FORWARD, FFTW_ESTIMATE); - fftw_execute(p1); - //fftw_destroy_plan(p1); + //std::cout << "\n Call FFTW3 "; + p1 = fftw_plan_dft_1d(FFT_NR, fft_phir, fft_phik, FFTW_FORWARD, FFTW_ESTIMATE); + fftw_execute(p1); + //fftw_destroy_plan(p1); #elif defined __FFTW2 - //std::cout << "\n Call FFTW2 "; - p1 = fftw_create_plan(FFT_NR, FFTW_FORWARD, FFTW_ESTIMATE); - fftw_one(p1, fft_phir, fft_phik); - //fftw_destroy_plan(p1); + //std::cout << "\n Call FFTW2 "; + p1 = fftw_create_plan(FFT_NR, FFTW_FORWARD, FFTW_ESTIMATE); + fftw_one(p1, fft_phir, fft_phik); + //fftw_destroy_plan(p1); #endif - - double dk_uniform = ModuleBase::TWO_PI / FFT_NR / dr; - - //for (int ik = 0; ik < FFT_NR/2; ik++) - //{ - //double kp = ik * dk_uniform; - //fft_ndphik[ik].re = pow(kp, nd) * fft_phik[ik].re; - //fft_ndphik[ik].im = 0.0; - //} - - //for (int ik = FFT_NR/2; ik < FFT_NR; ik++) - //{ - //double kp = -(FFT_NR - ik)* dk_uniform; - //fft_ndphik[ik].re = pow(kp, nd) * fft_phik[ik].re; - //fft_ndphik[ik].im = 0.0; - //} - - for (int ik = 0; ik < FFT_NR/2; ik++) // ik = 0 1 ... 8008 - { - double kp = ik * dk_uniform; - c_re(fft_ndphik[ik]) = pow(kp, nd) * c_re(fft_phik[ik]); - c_im(fft_ndphik[ik]) = 0.0; - } - for (int ik = FFT_NR/2; ik < FFT_NR; ik++) // ik = 8009 8010 ... 16018 - { - //double kp = -(FFT_NR - ik )* dk_uniform; //(...) = 8010 8009 ... 1 - double kp = -(FFT_NR - ik -1)* dk_uniform; //(...) -> 8009 8008 ... 0 - c_re(fft_ndphik[ik]) = pow(kp, nd) * c_re(fft_phik[ik]); - c_im(fft_ndphik[ik]) = 0.0; - } + + double dk_uniform = ModuleBase::TWO_PI / FFT_NR / dr; + + //for (int ik = 0; ik < FFT_NR/2; ik++) + //{ + //double kp = ik * dk_uniform; + //fft_ndphik[ik].re = pow(kp, nd) * fft_phik[ik].re; + //fft_ndphik[ik].im = 0.0; + //} + + //for (int ik = FFT_NR/2; ik < FFT_NR; ik++) + //{ + //double kp = -(FFT_NR - ik)* dk_uniform; + //fft_ndphik[ik].re = pow(kp, nd) * fft_phik[ik].re; + //fft_ndphik[ik].im = 0.0; + //} + + for (int ik = 0; ik < FFT_NR/2; ik++) // ik = 0 1 ... 8008 + { + double kp = ik * dk_uniform; + c_re(fft_ndphik[ik]) = pow(kp, nd) * c_re(fft_phik[ik]); + c_im(fft_ndphik[ik]) = 0.0; + } + for (int ik = FFT_NR/2; ik < FFT_NR; ik++) // ik = 8009 8010 ... 16018 + { + //double kp = -(FFT_NR - ik )* dk_uniform; //(...) = 8010 8009 ... 1 + double kp = -(FFT_NR - ik -1)* dk_uniform; //(...) -> 8009 8008 ... 0 + c_re(fft_ndphik[ik]) = pow(kp, nd) * c_re(fft_phik[ik]); + c_im(fft_ndphik[ik]) = 0.0; + } #if defined __FFTW3 - p2 = fftw_plan_dft_1d(FFT_NR, fft_ndphik, fft_ndphir, FFTW_BACKWARD, FFTW_ESTIMATE); - fftw_execute(p2); - //fftw_destroy_plan(p2); + p2 = fftw_plan_dft_1d(FFT_NR, fft_ndphik, fft_ndphir, FFTW_BACKWARD, FFTW_ESTIMATE); + fftw_execute(p2); + //fftw_destroy_plan(p2); #elif defined __FFTW2 - p2 = fftw_create_plan(FFT_NR, FFTW_BACKWARD, FFTW_ESTIMATE); - fftw_one(p2, fft_ndphik, fft_ndphir); - //fftw_destroy_plan(p2); + p2 = fftw_create_plan(FFT_NR, FFTW_BACKWARD, FFTW_ESTIMATE); + fftw_one(p2, fft_ndphik, fft_ndphir); + //fftw_destroy_plan(p2); #endif - bool is_re = true; - double fac = 0.0; - if (nd % 4 == 0) - { - is_re = true; - fac = 1.0; - } - else if (nd % 4 == 1) - { - is_re = false; - fac = -1.0; - } - else if (nd % 4 == 2) - { - is_re = true; - fac = -1.0; - } - else - { - is_re = false; - fac = 1.0; - } - - for (int ir = 0; ir < mesh; ir++) - { - if (is_re) - { - phind[ir] = fac * c_re(fft_ndphir[ir]) / FFT_NR; - } - else - { - phind[ir] = fac * c_im(fft_ndphir[ir]) / FFT_NR; - } - } - - fftw_destroy_plan (p1); - fftw_destroy_plan (p2); + bool is_re = true; + double fac = 0.0; + if (nd % 4 == 0) + { + is_re = true; + fac = 1.0; + } + else if (nd % 4 == 1) + { + is_re = false; + fac = -1.0; + } + else if (nd % 4 == 2) + { + is_re = true; + fac = -1.0; + } + else + { + is_re = false; + fac = 1.0; + } + + for (int ir = 0; ir < mesh; ir++) + { + if (is_re) + { + phind[ir] = fac * c_re(fft_ndphir[ir]) / FFT_NR; + } + else + { + phind[ir] = fac * c_im(fft_ndphir[ir]) / FFT_NR; + } + } + + fftw_destroy_plan (p1); + fftw_destroy_plan (p2); delete [] fft_phir; delete [] fft_phik; delete [] fft_ndphik; delete [] fft_ndphir; - - ModuleBase::timer::end("Mathzone_Add1", "Uni_Deriv_Phi"); + + ModuleBase::timer::end("Mathzone_Add1", "Uni_Deriv_Phi"); } } diff --git a/source/source_base/matrix.cpp b/source/source_base/matrix.cpp index eaa7c304061..fb036438ce5 100644 --- a/source/source_base/matrix.cpp +++ b/source/source_base/matrix.cpp @@ -34,73 +34,73 @@ void matrixAlloc() /*create a matrix with nrows*ncols size */ matrix::matrix( const int nrows, const int ncols, const bool flag_zero ) - :nr(nrows), - nc(ncols), - c(nullptr) + :nr(nrows), + nc(ncols), + c(nullptr) { - if( nr && nc ) - { - auto handler_old = std::set_new_handler(matrixAlloc); - c = new double[nr*nc]; - std::set_new_handler(handler_old); - if(flag_zero) this->zero_out(); - } + if( nr && nc ) + { + auto handler_old = std::set_new_handler(matrixAlloc); + c = new double[nr*nc]; + std::set_new_handler(handler_old); + if(flag_zero) this->zero_out(); + } } matrix::matrix( const matrix &m_in ) - :nr(m_in.nr), - nc(m_in.nc), - c(nullptr) + :nr(m_in.nr), + nc(m_in.nc), + c(nullptr) { - if( nr && nc ) - { - auto handler_old = std::set_new_handler(matrixAlloc); - c = new double[nr*nc]; - std::set_new_handler(handler_old); - memcpy( c, m_in.c, nr*nc*sizeof(double) ); - } + if( nr && nc ) + { + auto handler_old = std::set_new_handler(matrixAlloc); + c = new double[nr*nc]; + std::set_new_handler(handler_old); + memcpy( c, m_in.c, nr*nc*sizeof(double) ); + } } // Peize Lin add 2016-08-05 matrix::matrix( matrix && m_in ) - :nr(m_in.nr), - nc(m_in.nc) + :nr(m_in.nr), + nc(m_in.nc) { - c = m_in.c; - m_in.nr = m_in.nc = 0; - m_in.c = nullptr; + c = m_in.c; + m_in.nr = m_in.nc = 0; + m_in.c = nullptr; } // Peize Lin change 2018-07-02 matrix& matrix::operator=( const matrix & m_in ) { - this->create( m_in.nr, m_in.nc, false ); - memcpy( c, m_in.c, nr*nc*sizeof(double) ); - return *this; + this->create( m_in.nr, m_in.nc, false ); + memcpy( c, m_in.c, nr*nc*sizeof(double) ); + return *this; } // Peize Lin add 2016-08-05 matrix& matrix::operator=( matrix && m_in ) { - nr = m_in.nr; nc = m_in.nc; - if(c) delete[] c; - c = m_in.c; - m_in.nr = m_in.nc = 0; - m_in.c = nullptr; - return *this; + nr = m_in.nr; nc = m_in.nc; + if(c) delete[] c; + c = m_in.c; + m_in.nr = m_in.nc = 0; + m_in.c = nullptr; + return *this; } /* double & matrix::operator()(const int ir,const int ic) { - assert(ir>=0); assert(ir=0); assert(ic=0); assert(ir=0); assert(ic=0); assert(ir=0); assert(ic=0); assert(ir=0); assert(ic zero_out(); + if(flag_zero) this-> zero_out(); } double trace_on(const matrix &A, const matrix &B) { - assert(A.nr == B.nc); - assert(A.nc == B.nr); + assert(A.nr == B.nc); + assert(A.nc == B.nr); double tr = 0.0; for (int i = 0; i < A.nr; ++i) @@ -376,18 +376,18 @@ double mdot(const matrix &A, const matrix &B) // Peize Lin add 2016-09-08 std::ostream & operator<<( std::ostream & os, const matrix & m ) { - for( int ir=0; ir!=m.nr; ++ir ) - { - for( int ic=0; ic!=m.nc; ++ic ) - { - if(std::abs(m(ir,ic))>1E-10) - os<1E-10) + os<nr; ++ir ) - { - for( int ic=0; ic!=this->nc; ++ic ) - { - if(std::abs((*this)(ir,ic))>threshold) - os<<(*this)(ir,ic)<<"\t"; - else - os<<0<<"\t"; - } - os<nr; ++ir ) + { + for( int ic=0; ic!=this->nc; ++ic ) + { + if(std::abs((*this)(ir,ic))>threshold) + os<<(*this)(ir,ic)<<"\t"; + else + os<<0<<"\t"; + } + os<::min(); - const int size = nr * nc; - for( int i=0; i::min(); + const int size = nr * nc; + for( int i=0; i::max(); - const int size = nr * nc; - for( int i=0; i::max(); + const int size = nr * nc; + for( int i=0; i=0); assert(ir=0); assert(ic=0); assert(ir=0); assert(ic=0); assert(ir=0); assert(ic=0); assert(ir=0); assert(iczero_out(); - } + if(flag_zero) this->zero_out(); + } } inline matrix::matrix( const matrix &m_in ) { std::cout<<__FILE__<<__LINE__<create( m_in.nr, m_in.nc, false ); + this->create( m_in.nr, m_in.nc, false ); std::cout<<__FILE__<<__LINE__<=0); assert(ir=0); assert(ic=0); assert(ir=0); assert(ic=0); assert(ir=0); assert(ic=0); assert(ir=0); assert(ic #ifdef WIN32 - #define __FUNCTION__ __FILE__ + #define __FUNCTION__ __FILE__ #endif #ifndef MCD_QUIET - #ifndef WIN32 - #warning - Building with memory checking. - #warning expect lower performance. - - #endif + #ifndef WIN32 + #warning - Building with memory checking. + #warning expect lower performance. - + #endif #endif //warn about redefinitions (sometimes these can be #defined) #ifdef malloc #undef malloc - #ifndef WIN32 - #warning ------ Redefining malloc() ------ - #endif + #ifndef WIN32 + #warning ------ Redefining malloc() ------ + #endif #endif #ifdef calloc #undef calloc - #ifndef WIN32 - #warning ------ Redefining calloc() ------ - #endif + #ifndef WIN32 + #warning ------ Redefining calloc() ------ + #endif #endif #ifdef realloc #undef realloc - #ifndef WIN32 - #warning ------ Redefining realloc() ------ - #endif + #ifndef WIN32 + #warning ------ Redefining realloc() ------ + #endif #endif #ifdef free #undef free - #ifndef WIN32 - #warning ------ Redefining free() ------ - #endif + #ifndef WIN32 + #warning ------ Redefining free() ------ + #endif #endif #ifdef strdup #undef strdup - #ifndef WIN32 - #warning ------- Redefining strdup() -------- - #endif + #ifndef WIN32 + #warning ------- Redefining strdup() -------- + #endif #endif #ifdef strndup #undef strndup - #ifndef WIN32 - #warning ------- Redefining strndup() ------- - #endif + #ifndef WIN32 + #warning ------- Redefining strndup() ------- + #endif #endif #ifdef asprintf #undef asprintf - #ifndef WIN32 - #warning ------ Redefining asprintf() ------ - #endif + #ifndef WIN32 + #warning ------ Redefining asprintf() ------ + #endif #endif #ifdef vasprintf #undef vasprintf - #ifndef WIN32 - #warning ------ Redefining vasprintf() ------ - #endif + #ifndef WIN32 + #warning ------ Redefining vasprintf() ------ + #endif #endif #ifdef scanf #undef scanf - #ifndef WIN32 - #warning ------ Redefining scanf() ------ - #endif + #ifndef WIN32 + #warning ------ Redefining scanf() ------ + #endif #endif #ifdef fscanf #undef fscanf - #ifndef WIN32 - #warning ------ Redefining fscanf() ------ - #endif + #ifndef WIN32 + #warning ------ Redefining fscanf() ------ + #endif #endif #ifdef sscanf #undef sscanf - #ifndef WIN32 - #warning ------ Redefining sscanf() ------ - #endif + #ifndef WIN32 + #warning ------ Redefining sscanf() ------ + #endif #endif #ifdef getcwd #undef getcwd - #ifndef WIN32 - #warning ------ Redefining getcwd() ------ - #endif + #ifndef WIN32 + #warning ------ Redefining getcwd() ------ + #endif #endif #define strdup(p) MCD_strdup(p,__FUNCTION__,__FILE__,__LINE__) @@ -180,11 +180,11 @@ void MCD_free(void *p,char*,char*,int); void showMemStats(void); /* --- to send realtime stats somewhere other than stderr, - put an opened fp in here --- */ + put an opened fp in here --- */ void _MCD_RealTimeLog(FILE*); /* --- to send showMemStats() somewhere other than stdout, - put an opened fp in here --- */ + put an opened fp in here --- */ void _MCD_MemStatLog(FILE*); #else //MCD is gone @@ -204,15 +204,15 @@ void _MCD_MemStatLog(FILE*); #ifdef __cplusplus /* Some C++ new/delete operator overloading */ #ifndef WIN32 - #warning C++ Extentions Enabled + #warning C++ Extentions Enabled #endif #ifdef new - #undef new + #undef new #endif #ifdef delete - #undef delete + #undef delete #endif extern char *_MCD_LastSetFile,*_MCD_LastSetFun; @@ -220,21 +220,21 @@ extern int _MCD_LastSetLine; inline void setFileFunLineState(char*file,char*fun,int line) { - _MCD_LastSetLine=line; - _MCD_LastSetFile=file; - _MCD_LastSetFun=fun; + _MCD_LastSetLine=line; + _MCD_LastSetFile=file; + _MCD_LastSetFun=fun; } inline void* operator new (unsigned int size,char *file, char*fun,int line) { - return MCD_malloc(size,file,fun,line); + return MCD_malloc(size,file,fun,line); } inline void* operator new[] (unsigned int size,char*file, char*fun,int line) { - return MCD_malloc(size,file,fun,line); + return MCD_malloc(size,file,fun,line); } // currently, passing args to delete operator is not working at all... @@ -246,7 +246,7 @@ char*fun=__FILE__,int line=__LINE__) char*fun=__FUNCTION__,int line=__LINE__) #endif { - MCD_free(buf,_MCD_LastSetFile,_MCD_LastSetFun,_MCD_LastSetLine); + MCD_free(buf,_MCD_LastSetFile,_MCD_LastSetFun,_MCD_LastSetLine); } // ...so we have these here for the day they work, which isn't today. inline void operator delete[] (void * buf,char*file=__FILE__, @@ -256,18 +256,18 @@ char*fun=__FILE__,int line=__LINE__) char*fun=__FUNCTION__,int line=__LINE__) #endif { - MCD_free(buf,_MCD_LastSetFile,_MCD_LastSetFun,_MCD_LastSetLine); + MCD_free(buf,_MCD_LastSetFile,_MCD_LastSetFun,_MCD_LastSetLine); } #endif // win32 default delete params inline void operator delete (void * buf) { - MCD_free(buf,0,0,0); + MCD_free(buf,0,0,0); } inline void operator delete[] (void * buf) { - MCD_free(buf,0,0,0); + MCD_free(buf,0,0,0); } #ifdef WIN32 diff --git a/source/source_base/memory_recorder.cpp b/source/source_base/memory_recorder.cpp index f122a6156a3..773063db816 100644 --- a/source/source_base/memory_recorder.cpp +++ b/source/source_base/memory_recorder.cpp @@ -51,451 +51,451 @@ Memory::~Memory() double Memory::calculate_mem(const long &n_in,const std::string &type) { - double n = static_cast(n_in); - double mem = 0.0; - - double factor = 1.0/1024.0/1024.0; - double complex_matrix_mem = complex_matrix_memory * factor; - double double_mem = double_memory * factor; - double int_mem = int_memory * factor; - double bool_mem = bool_memory * factor; - double float_mem = float_memory * factor; - double short_mem = short_memory * factor; - - if(type=="ModuleBase::ComplexMatrix" || type=="complexmatrix" || type=="cdouble") - { - mem = complex_matrix_mem; - } - else if(type=="real" || type=="double") - { - mem = double_mem; - } - else if(type=="int") - { - mem = int_mem; - } - else if(type=="bool") - { - mem = bool_mem; - } - else if(type=="short") - { - mem = short_mem; - } - else if(type=="float") - { - mem = float_mem; - } - else if(type=="AtomLink") - { - mem = int_mem * 2 + double_mem * 3; - } - else if(type=="ModuleBase::Vector3") - { - mem = 3 * double_mem; - } - else - { - std::cout<<"not this type in memory storage : "<(n_in); + double mem = 0.0; + + double factor = 1.0/1024.0/1024.0; + double complex_matrix_mem = complex_matrix_memory * factor; + double double_mem = double_memory * factor; + double int_mem = int_memory * factor; + double bool_mem = bool_memory * factor; + double float_mem = float_memory * factor; + double short_mem = short_memory * factor; + + if(type=="ModuleBase::ComplexMatrix" || type=="complexmatrix" || type=="cdouble") + { + mem = complex_matrix_mem; + } + else if(type=="real" || type=="double") + { + mem = double_mem; + } + else if(type=="int") + { + mem = int_mem; + } + else if(type=="bool") + { + mem = bool_mem; + } + else if(type=="short") + { + mem = short_mem; + } + else if(type=="float") + { + mem = float_mem; + } + else if(type=="AtomLink") + { + mem = int_mem * 2 + double_mem * 3; + } + else if(type=="ModuleBase::Vector3") + { + mem = 3 * double_mem; + } + else + { + std::cout<<"not this type in memory storage : "<= n_memory) - { - std::cout<<" Error! Too many memories required."; - return 0.0; - } - - consume[find] = Memory::calculate_mem(n_in,type); - - if(consume[find] > memory_warning_threshold_mb) - { - print(name[find], consume[find]); - } - return consume[find]; + if(!Memory::init_flag) + { + name = new std::string[n_memory]; + class_name = new std::string[n_memory]; + consume = new double[n_memory]; + for(int i=0;i= n_memory) + { + std::cout<<" Error! Too many memories required."; + return 0.0; + } + + consume[find] = Memory::calculate_mem(n_in,type); + + if(consume[find] > memory_warning_threshold_mb) + { + print(name[find], consume[find]); + } + return consume[find]; } void Memory::record ( - const std::string &name_in, - const long &n_in, - const bool accumulate + const std::string &name_in, + const long &n_in, + const bool accumulate ) { - if(!Memory::init_flag) - { - name = new std::string[n_memory]; - class_name = new std::string[n_memory]; - consume = new double[n_memory]; - for(int i=0;i= n_memory) - { - std::cout<<" Error! Too many memories has been recorded."; - return; - } - - const double factor = 1.0/1024.0/1024.0; - double size_mb = n_in * factor; - - if(accumulate) - { - consume[find] += size_mb; - Memory::total += size_mb; - } - else - { - if(consume[find] < size_mb) - { - Memory::total += size_mb - consume[find]; - consume[find] = size_mb; - if(consume[find] > memory_warning_threshold_mb) - { - print(name[find], consume[find]); - } - } - } - - return; + if(!Memory::init_flag) + { + name = new std::string[n_memory]; + class_name = new std::string[n_memory]; + consume = new double[n_memory]; + for(int i=0;i= n_memory) + { + std::cout<<" Error! Too many memories has been recorded."; + return; + } + + const double factor = 1.0/1024.0/1024.0; + double size_mb = n_in * factor; + + if(accumulate) + { + consume[find] += size_mb; + Memory::total += size_mb; + } + else + { + if(consume[find] < size_mb) + { + Memory::total += size_mb - consume[find]; + consume[find] = size_mb; + if(consume[find] > memory_warning_threshold_mb) + { + print(name[find], consume[find]); + } + } + } + + return; } #if defined(__CUDA) || defined(__ROCM) double Memory::record_gpu ( - const std::string &class_name_in, - const std::string &name_in, - const long &n_in, - const std::string &type, - const bool accumulate + const std::string &class_name_in, + const std::string &name_in, + const long &n_in, + const std::string &type, + const bool accumulate ) { - if(!Memory::init_flag_gpu) - { - name_gpu = new std::string[n_memory]; - class_name_gpu = new std::string[n_memory]; - consume_gpu = new double[n_memory]; - for(int i=0;i= n_memory) - { - std::cout<<" Error! Too many gpu memories required."; - return 0.0; - } - - consume_gpu[find] = Memory::calculate_mem(n_in,type); - - if(consume_gpu[find] > memory_warning_threshold_mb) - { - print(name_gpu[find], consume_gpu[find]); - } - return consume_gpu[find]; + if(!Memory::init_flag_gpu) + { + name_gpu = new std::string[n_memory]; + class_name_gpu = new std::string[n_memory]; + consume_gpu = new double[n_memory]; + for(int i=0;i= n_memory) + { + std::cout<<" Error! Too many gpu memories required."; + return 0.0; + } + + consume_gpu[find] = Memory::calculate_mem(n_in,type); + + if(consume_gpu[find] > memory_warning_threshold_mb) + { + print(name_gpu[find], consume_gpu[find]); + } + return consume_gpu[find]; } void Memory::record_gpu ( - const std::string &name_in, - const size_t &n_in, - const bool accumulate + const std::string &name_in, + const size_t &n_in, + const bool accumulate ) { - if(!Memory::init_flag_gpu) - { - name_gpu = new std::string[n_memory]; - class_name_gpu = new std::string[n_memory]; - consume_gpu = new double[n_memory]; - for(int i=0;i= n_memory) - { - std::cout<<" Error! Too many gpu memories has been recorded."; - return; - } - - const double factor = 1.0/1024.0/1024.0; - double size_mb = n_in * factor; - - if(accumulate) - { - consume_gpu[find] += size_mb; - Memory::total_gpu += size_mb; - } - else - { - if(consume_gpu[find] < size_mb) - { - Memory::total_gpu += size_mb - consume_gpu[find]; - consume_gpu[find] = size_mb; - if(consume_gpu[find] > memory_warning_threshold_mb) - { - print(name_gpu[find], consume_gpu[find]); - } - } - } - return; + if(!Memory::init_flag_gpu) + { + name_gpu = new std::string[n_memory]; + class_name_gpu = new std::string[n_memory]; + consume_gpu = new double[n_memory]; + for(int i=0;i= n_memory) + { + std::cout<<" Error! Too many gpu memories has been recorded."; + return; + } + + const double factor = 1.0/1024.0/1024.0; + double size_mb = n_in * factor; + + if(accumulate) + { + consume_gpu[find] += size_mb; + Memory::total_gpu += size_mb; + } + else + { + if(consume_gpu[find] < size_mb) + { + Memory::total_gpu += size_mb - consume_gpu[find]; + consume_gpu[find] = size_mb; + if(consume_gpu[find] > memory_warning_threshold_mb) + { + print(name_gpu[find], consume_gpu[find]); + } + } + } + return; } #endif void Memory::print(const std::string& mem_name, double size_mb) { - GlobalV::ofs_running <<"\n *** Memory Allocation Warning *** " - <<" "<< mem_name <<" "<< size_mb <<" MB" << std::endl; - return; + GlobalV::ofs_running <<"\n *** Memory Allocation Warning *** " + <<" "<< mem_name <<" "<< size_mb <<" MB" << std::endl; + return; } void Memory::finish(std::ofstream &ofs) { - print_all(ofs); - if(init_flag) - { - delete[] name; - delete[] class_name; - delete[] consume; - init_flag = false; - } + print_all(ofs); + if(init_flag) + { + delete[] name; + delete[] class_name; + delete[] consume; + init_flag = false; + } #if defined(__CUDA) || defined(__ROCM) - if(init_flag_gpu) - { - delete[] name_gpu; - delete[] class_name_gpu; - delete[] consume_gpu; - init_flag_gpu = false; - } + if(init_flag_gpu) + { + delete[] name_gpu; + delete[] class_name_gpu; + delete[] consume_gpu; + init_flag_gpu = false; + } #endif - return; + return; } void Memory::print_all(std::ofstream &ofs) { - if(!init_flag) - { - return; - } + if(!init_flag) + { + return; + } - const double small = 1.0; // unit is MB + const double small = 1.0; // unit is MB #ifdef __MPI - Parallel_Reduce::reduce_all(Memory::total); + Parallel_Reduce::reduce_all(Memory::total); #if defined(__CUDA) || defined(__ROCM) - Parallel_Reduce::reduce_all(Memory::total_gpu); + Parallel_Reduce::reduce_all(Memory::total_gpu); #endif #endif - ofs <<"\n NAME-------------------------|MEMORY(MB)------------------" << std::endl; + ofs <<"\n NAME-------------------------|MEMORY(MB)------------------" << std::endl; ofs << std::right; - ofs << std::setw(30)<< "total" << std::setw(15) <0); - bool *print_flag = new bool[n_memory]; + bool *print_flag = new bool[n_memory]; - for(int i=0; i0); - bool *print_flag_gpu = new bool[n_memory]; + bool *print_flag_gpu = new bool[n_memory]; - for(int i=0; i, base_device::DEVICE_CPU template struct delete_memory_op_mt, base_device::DEVICE_CPU>; #endif -template -void resize_memory(FPTYPE* arr, const size_t size, base_device::AbacusDevice_t device_type) -{ - if (device_type == base_device::AbacusDevice_t::CpuDevice){ - resize_memory_op()(arr, size); - } - else if (device_type == base_device::AbacusDevice_t::GpuDevice){ - resize_memory_op()(arr, size); - } -} - -template -void set_memory(FPTYPE* arr, const int var, const size_t size, base_device::AbacusDevice_t device_type){ - if (device_type == base_device::AbacusDevice_t::CpuDevice){ - set_memory_op()(arr, var, size); - } - else if (device_type == base_device::AbacusDevice_t::GpuDevice){ - set_memory_op()(arr, var, size); - } -} - -template -void synchronize_memory(FPTYPE* arr_out, const FPTYPE* arr_in, const size_t size, base_device::AbacusDevice_t device_type_out, base_device::AbacusDevice_t device_type_in){ - // The four source/destination combinations are mutually exclusive, so each - // branch must test BOTH devices. Using `||` here made the first branch match - // whenever either side was the CPU, which routed host<->device transfers to - // the host-to-host specialization. - if (device_type_out == base_device::AbacusDevice_t::CpuDevice && device_type_in == base_device::AbacusDevice_t::CpuDevice){ - synchronize_memory_op()(arr_out, arr_in, size); - } -#if __CUDA || __UT_USE_CUDA || __ROCM || __UT_USE_ROCM - else if (device_type_out == base_device::AbacusDevice_t::CpuDevice && device_type_in == base_device::AbacusDevice_t::GpuDevice){ - synchronize_memory_op()(arr_out, arr_in, size); - } - else if (device_type_out == base_device::AbacusDevice_t::GpuDevice && device_type_in == base_device::AbacusDevice_t::CpuDevice){ - synchronize_memory_op()(arr_out, arr_in, size); - } - else if (device_type_out == base_device::AbacusDevice_t::GpuDevice && device_type_in == base_device::AbacusDevice_t::GpuDevice){ - synchronize_memory_op()(arr_out, arr_in, size); - } -#endif - else { - ModuleBase::WARNING_QUIT("base_device::memory::synchronize_memory", - "unsupported source/destination device combination"); - } -} - -template -void cast_memory(FPTYPE_out* arr_out, const FPTYPE_in* arr_in, const size_t size, base_device::AbacusDevice_t device_type_out, base_device::AbacusDevice_t device_type_in) -{ - // See synchronize_memory() above: dispatch on the exact (out, in) device pair. - if (device_type_out == base_device::AbacusDevice_t::CpuDevice && device_type_in == base_device::AbacusDevice_t::CpuDevice){ - cast_memory_op()(arr_out, arr_in, size); - } -#if __CUDA || __UT_USE_CUDA || __ROCM || __UT_USE_ROCM - else if (device_type_out == base_device::AbacusDevice_t::CpuDevice && device_type_in == base_device::AbacusDevice_t::GpuDevice){ - cast_memory_op()(arr_out, arr_in, size); - } - else if (device_type_out == base_device::AbacusDevice_t::GpuDevice && device_type_in == base_device::AbacusDevice_t::CpuDevice){ - cast_memory_op()(arr_out, arr_in, size); - } - else if (device_type_out == base_device::AbacusDevice_t::GpuDevice && device_type_in == base_device::AbacusDevice_t::GpuDevice){ - cast_memory_op()(arr_out, arr_in, size); - } -#endif - else { - ModuleBase::WARNING_QUIT("base_device::memory::cast_memory", - "unsupported source/destination device combination"); - } -} - -template -void delete_memory(FPTYPE* arr, base_device::AbacusDevice_t device_type) -{ - if (device_type == base_device::AbacusDevice_t::CpuDevice){ - delete_memory_op()(arr); - } - else if (device_type == base_device::AbacusDevice_t::GpuDevice){ - delete_memory_op()(arr); - } -} - -// Explicit instantiations of the runtime-dispatch wrappers, so that the -// declarations in memory_op.h can actually be linked from another translation -// unit (and covered by unit tests). cast_memory is instantiated only for the -// type pairs that cast_memory_op provides for all four device combinations. -template void synchronize_memory(int*, const int*, const size_t, base_device::AbacusDevice_t, base_device::AbacusDevice_t); -template void synchronize_memory(float*, const float*, const size_t, base_device::AbacusDevice_t, base_device::AbacusDevice_t); -template void synchronize_memory(double*, const double*, const size_t, base_device::AbacusDevice_t, base_device::AbacusDevice_t); -template void synchronize_memory>(std::complex*, const std::complex*, const size_t, base_device::AbacusDevice_t, base_device::AbacusDevice_t); -template void synchronize_memory>(std::complex*, const std::complex*, const size_t, base_device::AbacusDevice_t, base_device::AbacusDevice_t); - -template void cast_memory(float*, const float*, const size_t, base_device::AbacusDevice_t, base_device::AbacusDevice_t); -template void cast_memory(double*, const double*, const size_t, base_device::AbacusDevice_t, base_device::AbacusDevice_t); -template void cast_memory(float*, const double*, const size_t, base_device::AbacusDevice_t, base_device::AbacusDevice_t); -template void cast_memory(double*, const float*, const size_t, base_device::AbacusDevice_t, base_device::AbacusDevice_t); -template void cast_memory, std::complex>(std::complex*, const std::complex*, const size_t, base_device::AbacusDevice_t, base_device::AbacusDevice_t); -template void cast_memory, std::complex>(std::complex*, const std::complex*, const size_t, base_device::AbacusDevice_t, base_device::AbacusDevice_t); -template void cast_memory, std::complex>(std::complex*, const std::complex*, const size_t, base_device::AbacusDevice_t, base_device::AbacusDevice_t); -template void cast_memory, std::complex>(std::complex*, const std::complex*, const size_t, base_device::AbacusDevice_t, base_device::AbacusDevice_t); - } // namespace memory } // namespace base_device \ No newline at end of file diff --git a/source/source_base/module_device/memory_op.h b/source/source_base/module_device/memory_op.h index 5e292c57a65..ef8863fb92c 100644 --- a/source/source_base/module_device/memory_op.h +++ b/source/source_base/module_device/memory_op.h @@ -120,21 +120,6 @@ struct delete_memory_op void operator()(FPTYPE* arr); }; -template -void resize_memory(FPTYPE* arr, const size_t size, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); - -template -void set_memory(FPTYPE* arr, const int var, const size_t size, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); - -template -void synchronize_memory(FPTYPE* arr_out, const FPTYPE* arr_in, const size_t size, base_device::AbacusDevice_t device_type_out, base_device::AbacusDevice_t device_type_in); - -template -void cast_memory(FPTYPE_out* arr_out, const FPTYPE_in* arr_in, const size_t size, base_device::AbacusDevice_t device_type_out, base_device::AbacusDevice_t device_type_in); - -template -void delete_memory(FPTYPE* arr, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); - #if __CUDA || __UT_USE_CUDA || __ROCM || __UT_USE_ROCM // Partially specialize operator for base_device::GpuDevice. template diff --git a/source/source_base/module_device/test/memory_test.cpp b/source/source_base/module_device/test/memory_test.cpp index a6912beffe3..a963c68c532 100644 --- a/source/source_base/module_device/test/memory_test.cpp +++ b/source/source_base/module_device/test/memory_test.cpp @@ -170,59 +170,6 @@ TEST_F(TestModulePsiMemory, delete_memory_op_complex_double_cpu) delete_memory_complex_double_cpu_op()(hz_xx); } -// --------------------------------------------------------------------------- -// Runtime-dispatch wrappers (issue #7553). -// -// synchronize_memory()/cast_memory() pick a compile-time specialization from a -// pair of runtime AbacusDevice_t values. The branches used to be joined with -// `||`, so the first one matched whenever EITHER side was the CPU and every -// host<->device transfer was served by the host-to-host specialization. The -// checks below pin the exact-pair dispatch for all combinations available in -// the current build. -// --------------------------------------------------------------------------- - -TEST_F(TestModulePsiMemory, synchronize_memory_dispatch_cpu_to_cpu) -{ - std::vector h_xx(xx.size(), 0); - base_device::memory::synchronize_memory(h_xx.data(), - xx.data(), - xx.size(), - base_device::AbacusDevice_t::CpuDevice, - base_device::AbacusDevice_t::CpuDevice); - for (int ii = 0; ii < xx.size(); ii++) - { - EXPECT_EQ(h_xx[ii], xx[ii]); - } -} - -TEST_F(TestModulePsiMemory, synchronize_memory_dispatch_complex_cpu_to_cpu) -{ - std::vector> hz_xx(z_xx.size(), std::complex(0, 0)); - base_device::memory::synchronize_memory(hz_xx.data(), - z_xx.data(), - z_xx.size(), - base_device::AbacusDevice_t::CpuDevice, - base_device::AbacusDevice_t::CpuDevice); - for (int ii = 0; ii < z_xx.size(); ii++) - { - EXPECT_EQ(hz_xx[ii], z_xx[ii]); - } -} - -TEST_F(TestModulePsiMemory, cast_memory_dispatch_cpu_to_cpu) -{ - std::vector h_xx(xx.size(), 0); - base_device::memory::cast_memory(h_xx.data(), - xx.data(), - xx.size(), - base_device::AbacusDevice_t::CpuDevice, - base_device::AbacusDevice_t::CpuDevice); - for (int ii = 0; ii < xx.size(); ii++) - { - EXPECT_FLOAT_EQ(h_xx[ii], static_cast(xx[ii])); - } -} - #if __UT_USE_CUDA || __UT_USE_ROCM TEST_F(TestModulePsiMemory, set_memory_op_double_gpu) { @@ -400,90 +347,4 @@ TEST_F(TestModulePsiMemory, delete_memory_op_complex_double_gpu) delete_memory_complex_double_gpu_op()(thrust::raw_pointer_cast(dz_xx)); } - -// Exact-pair dispatch across the host/device boundary (issue #7553). Before the -// fix these three cases all reached synchronize_memory_op<..., CPU, CPU>, i.e. a -// plain host memcpy on a device pointer. -TEST_F(TestModulePsiMemory, synchronize_memory_dispatch_cpu_to_gpu) -{ - thrust::device_ptr d_xx = thrust::device_malloc(xx.size()); - std::vector hv_xx(xx.size(), 0); - thrust::copy(hv_xx.begin(), hv_xx.end(), d_xx); - base_device::memory::synchronize_memory(thrust::raw_pointer_cast(d_xx), - xx.data(), - xx.size(), - base_device::AbacusDevice_t::GpuDevice, - base_device::AbacusDevice_t::CpuDevice); - - thrust::host_vector h_xx(xx.size()); - thrust::copy(d_xx, d_xx + xx.size(), h_xx.begin()); - for (int ii = 0; ii < xx.size(); ii++) - { - EXPECT_EQ(h_xx[ii], xx[ii]); - } - thrust::device_free(d_xx); -} - -TEST_F(TestModulePsiMemory, synchronize_memory_dispatch_gpu_to_cpu) -{ - thrust::device_ptr d_xx = thrust::device_malloc(xx.size()); - thrust::copy(xx.begin(), xx.end(), d_xx); - thrust::host_vector h_xx(xx.size()); - base_device::memory::synchronize_memory(thrust::raw_pointer_cast(h_xx.data()), - thrust::raw_pointer_cast(d_xx), - xx.size(), - base_device::AbacusDevice_t::CpuDevice, - base_device::AbacusDevice_t::GpuDevice); - - for (int ii = 0; ii < xx.size(); ii++) - { - EXPECT_EQ(h_xx[ii], xx[ii]); - } - thrust::device_free(d_xx); -} - -TEST_F(TestModulePsiMemory, synchronize_memory_dispatch_gpu_to_gpu) -{ - thrust::device_ptr d1_xx = thrust::device_malloc(xx.size()); - thrust::device_ptr d2_xx = thrust::device_malloc(xx.size()); - thrust::copy(xx.begin(), xx.end(), d1_xx); - base_device::memory::synchronize_memory(thrust::raw_pointer_cast(d2_xx), - thrust::raw_pointer_cast(d1_xx), - xx.size(), - base_device::AbacusDevice_t::GpuDevice, - base_device::AbacusDevice_t::GpuDevice); - - thrust::host_vector h_xx(xx.size()); - thrust::copy(d2_xx, d2_xx + xx.size(), h_xx.begin()); - for (int ii = 0; ii < xx.size(); ii++) - { - EXPECT_EQ(h_xx[ii], xx[ii]); - } - thrust::device_free(d1_xx); - thrust::device_free(d2_xx); -} - -TEST_F(TestModulePsiMemory, cast_memory_dispatch_cpu_to_gpu_and_back) -{ - thrust::device_ptr d_xx = thrust::device_malloc(xx.size()); - base_device::memory::cast_memory(thrust::raw_pointer_cast(d_xx), - xx.data(), - xx.size(), - base_device::AbacusDevice_t::GpuDevice, - base_device::AbacusDevice_t::CpuDevice); - - std::vector h_xx(xx.size(), 0); - base_device::memory::cast_memory(h_xx.data(), - thrust::raw_pointer_cast(d_xx), - xx.size(), - base_device::AbacusDevice_t::CpuDevice, - base_device::AbacusDevice_t::GpuDevice); - - for (int ii = 0; ii < xx.size(); ii++) - { - EXPECT_FLOAT_EQ(static_cast(h_xx[ii]), static_cast(xx[ii])); - } - thrust::device_free(d_xx); -} - #endif // __UT_USE_CUDA || __UT_USE_ROCM diff --git a/source/source_base/module_external/blacs_connector.h b/source/source_base/module_external/blacs_connector.h index 61c67324e87..1aa23a479d9 100644 --- a/source/source_base/module_external/blacs_connector.h +++ b/source/source_base/module_external/blacs_connector.h @@ -33,16 +33,16 @@ extern "C" { - void Cblacs_pinfo(int *myid, int *nprocs); - void Cblacs_get(int icontxt, int what, int *val); - void Cblacs_gridmap(int* icontxt, int *usermap, int ldumap, int nprow, int npcol); - // Informational and Miscellaneous - void Cblacs_gridinfo(int icontxt, int* nprow, int *npcol, int *myprow, int *mypcol); + void Cblacs_pinfo(int *myid, int *nprocs); + void Cblacs_get(int icontxt, int what, int *val); + void Cblacs_gridmap(int* icontxt, int *usermap, int ldumap, int nprow, int npcol); + // Informational and Miscellaneous + void Cblacs_gridinfo(int icontxt, int* nprow, int *npcol, int *myprow, int *mypcol); void Cblacs_gridinit(int* icontxt, char* layout, int nprow, int npcol); void Cblacs_gridexit(int icontxt); int Cblacs_pnum(int icontxt, int prow, int pcol); void Cblacs_pcoord(int icontxt, int pnum, int *prow, int *pcol); - void Cblacs_exit(int icontxt); + void Cblacs_exit(int icontxt); // broadcast (send/recv) void Cigebs2d(int ConTxt, char *scope, char *top, int m, int n, int *A, int lda); @@ -73,19 +73,19 @@ void Cxgebs2d(int ConTxt, char *scope, char *top, int m, int n, T *A, int lda) std::is_same>::value, "Type not supported"); - if (std::is_same::value) { + if (std::is_same::value) { Cigebs2d(ConTxt, scope, top, m, n, reinterpret_cast(A), lda); } - if (std::is_same::value) { + if (std::is_same::value) { Csgebs2d(ConTxt, scope, top, m, n, reinterpret_cast(A), lda); } - if (std::is_same::value) { + if (std::is_same::value) { Cdgebs2d(ConTxt, scope, top, m, n, reinterpret_cast(A), lda); } - if (std::is_same>::value) { + if (std::is_same>::value) { Ccgebs2d(ConTxt, scope, top, m, n, reinterpret_cast*>(A), lda); } - if (std::is_same>::value) { + if (std::is_same>::value) { Czgebs2d(ConTxt, scope, top, m, n, reinterpret_cast*>(A), lda); } } @@ -101,19 +101,19 @@ void Cxgebr2d(int ConTxt, char *scope, char *top, int m, int n, T *A, int lda, i std::is_same>::value, "Type not supported"); - if (std::is_same::value) { + if (std::is_same::value) { Cigebr2d(ConTxt, scope, top, m, n, reinterpret_cast(A), lda, rsrc, csrc); } - if (std::is_same::value) { + if (std::is_same::value) { Csgebr2d(ConTxt, scope, top, m, n, reinterpret_cast(A), lda, rsrc, csrc); } - if (std::is_same::value) { + if (std::is_same::value) { Cdgebr2d(ConTxt, scope, top, m, n, reinterpret_cast(A), lda, rsrc, csrc); } - if (std::is_same>::value) { + if (std::is_same>::value) { Ccgebr2d(ConTxt, scope, top, m, n, reinterpret_cast*>(A), lda, rsrc, csrc); } - if (std::is_same>::value) { + if (std::is_same>::value) { Czgebr2d(ConTxt, scope, top, m, n, reinterpret_cast*>(A), lda, rsrc, csrc); } } diff --git a/source/source_base/module_external/blas_connector.h b/source/source_base/module_external/blas_connector.h index 0e34c42f08e..22f9f99212d 100644 --- a/source/source_base/module_external/blas_connector.h +++ b/source/source_base/module_external/blas_connector.h @@ -203,165 +203,165 @@ class BlasConnector { public: - // Peize Lin add 2016-08-04 - // y=a*x+y - static - void axpy( const int n, const float alpha, const float *X, const int incX, float *Y, const int incY, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); + // Peize Lin add 2016-08-04 + // y=a*x+y + static + void axpy( const int n, const float alpha, const float *X, const int incX, float *Y, const int incY, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); - static - void axpy( const int n, const double alpha, const double *X, const int incX, double *Y, const int incY, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); + static + void axpy( const int n, const double alpha, const double *X, const int incX, double *Y, const int incY, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); - static - void axpy( const int n, const std::complex alpha, const std::complex *X, const int incX, std::complex *Y, const int incY, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); + static + void axpy( const int n, const std::complex alpha, const std::complex *X, const int incX, std::complex *Y, const int incY, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); - static - void axpy( const int n, const std::complex alpha, const std::complex *X, const int incX, std::complex *Y, const int incY, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); + static + void axpy( const int n, const std::complex alpha, const std::complex *X, const int incX, std::complex *Y, const int incY, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); - // Peize Lin add 2016-08-04 - // x=a*x - static - void scal( const int n, const float alpha, float *X, const int incX, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); + // Peize Lin add 2016-08-04 + // x=a*x + static + void scal( const int n, const float alpha, float *X, const int incX, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); - static - void scal( const int n, const double alpha, double *X, const int incX, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); + static + void scal( const int n, const double alpha, double *X, const int incX, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); - static - void scal( const int n, const std::complex alpha, std::complex *X, const int incX, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); + static + void scal( const int n, const std::complex alpha, std::complex *X, const int incX, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); - static - void scal( const int n, const std::complex alpha, std::complex *X, const int incX, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); + static + void scal( const int n, const std::complex alpha, std::complex *X, const int incX, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); - // Peize Lin add 2017-10-27 - // d=x*y - static - float dot( const int n, const float*const X, const int incX, const float*const Y, const int incY, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); + // Peize Lin add 2017-10-27 + // d=x*y + static + float dot( const int n, const float*const X, const int incX, const float*const Y, const int incY, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); - static - double dot( const int n, const double*const X, const int incX, const double*const Y, const int incY, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); + static + double dot( const int n, const double*const X, const int incX, const double*const Y, const int incY, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); - // d=x*y - static - float dotu( const int n, const float*const X, const int incX, const float*const Y, const int incY, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); + // d=x*y + static + float dotu( const int n, const float*const X, const int incX, const float*const Y, const int incY, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); - static - double dotu( const int n, const double*const X, const int incX, const double*const Y, const int incY, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); + static + double dotu( const int n, const double*const X, const int incX, const double*const Y, const int incY, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); - static - std::complex dotu( const int n, const std::complex*const X, const int incX, const std::complex*const Y, const int incY, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); + static + std::complex dotu( const int n, const std::complex*const X, const int incX, const std::complex*const Y, const int incY, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); - static - std::complex dotu( const int n, const std::complex*const X, const int incX, const std::complex*const Y, const int incY, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); + static + std::complex dotu( const int n, const std::complex*const X, const int incX, const std::complex*const Y, const int incY, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); - // d=x.conj()*y - static - float dotc( const int n, const float*const X, const int incX, const float*const Y, const int incY, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); + // d=x.conj()*y + static + float dotc( const int n, const float*const X, const int incX, const float*const Y, const int incY, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); - static - double dotc( const int n, const double*const X, const int incX, const double*const Y, const int incY, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); + static + double dotc( const int n, const double*const X, const int incX, const double*const Y, const int incY, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); - static - std::complex dotc( const int n, const std::complex*const X, const int incX, const std::complex*const Y, const int incY, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); + static + std::complex dotc( const int n, const std::complex*const X, const int incX, const std::complex*const Y, const int incY, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); - static - std::complex dotc( const int n, const std::complex*const X, const int incX, const std::complex*const Y, const int incY, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); + static + std::complex dotc( const int n, const std::complex*const X, const int incX, const std::complex*const Y, const int incY, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); - // Peize Lin add 2017-10-27, fix bug trans 2019-01-17 - // C = a * A.? * B.? + b * C - // Row Major by default - static - void gemm(const char transa, const char transb, const int m, const int n, const int k, - const float alpha, const float *a, const int lda, const float *b, const int ldb, - const float beta, float *c, const int ldc, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); + // Peize Lin add 2017-10-27, fix bug trans 2019-01-17 + // C = a * A.? * B.? + b * C + // Row Major by default + static + void gemm(const char transa, const char transb, const int m, const int n, const int k, + const float alpha, const float *a, const int lda, const float *b, const int ldb, + const float beta, float *c, const int ldc, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); - static - void gemm(const char transa, const char transb, const int m, const int n, const int k, - const double alpha, const double *a, const int lda, const double *b, const int ldb, - const double beta, double *c, const int ldc, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); + static + void gemm(const char transa, const char transb, const int m, const int n, const int k, + const double alpha, const double *a, const int lda, const double *b, const int ldb, + const double beta, double *c, const int ldc, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); static void gemm(const char transa, const char transb, const int m, const int n, const int k, const std::complex alpha, const std::complex *a, const int lda, const std::complex *b, const int ldb, const std::complex beta, std::complex *c, const int ldc, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); - static - void gemm(const char transa, const char transb, const int m, const int n, const int k, - const std::complex alpha, const std::complex *a, const int lda, const std::complex *b, const int ldb, - const std::complex beta, std::complex *c, const int ldc, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); + static + void gemm(const char transa, const char transb, const int m, const int n, const int k, + const std::complex alpha, const std::complex *a, const int lda, const std::complex *b, const int ldb, + const std::complex beta, std::complex *c, const int ldc, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); - // Col-Major if you need to use it + // Col-Major if you need to use it - static - void gemm_cm(const char transa, const char transb, const int m, const int n, const int k, - const float alpha, const float *a, const int lda, const float *b, const int ldb, - const float beta, float *c, const int ldc, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); + static + void gemm_cm(const char transa, const char transb, const int m, const int n, const int k, + const float alpha, const float *a, const int lda, const float *b, const int ldb, + const float beta, float *c, const int ldc, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); - static - void gemm_cm(const char transa, const char transb, const int m, const int n, const int k, - const double alpha, const double *a, const int lda, const double *b, const int ldb, - const double beta, double *c, const int ldc, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); + static + void gemm_cm(const char transa, const char transb, const int m, const int n, const int k, + const double alpha, const double *a, const int lda, const double *b, const int ldb, + const double beta, double *c, const int ldc, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); static void gemm_cm(const char transa, const char transb, const int m, const int n, const int k, const std::complex alpha, const std::complex *a, const int lda, const std::complex *b, const int ldb, const std::complex beta, std::complex *c, const int ldc, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); - static - void gemm_cm(const char transa, const char transb, const int m, const int n, const int k, - const std::complex alpha, const std::complex *a, const int lda, const std::complex *b, const int ldb, - const std::complex beta, std::complex *c, const int ldc, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); + static + void gemm_cm(const char transa, const char transb, const int m, const int n, const int k, + const std::complex alpha, const std::complex *a, const int lda, const std::complex *b, const int ldb, + const std::complex beta, std::complex *c, const int ldc, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); - // side=='L': C = a * A * B + b * C. - // side=='R': C = a * B * A + b * C. - // A == A^T - // Because you cannot pack symm or hemm into a row-major kernel by exchanging parameters, so only col-major functions are provided. - static - void symm_cm(const char side, const char uplo, const int m, const int n, - const float alpha, const float *a, const int lda, const float *b, const int ldb, - const float beta, float *c, const int ldc, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); + // side=='L': C = a * A * B + b * C. + // side=='R': C = a * B * A + b * C. + // A == A^T + // Because you cannot pack symm or hemm into a row-major kernel by exchanging parameters, so only col-major functions are provided. + static + void symm_cm(const char side, const char uplo, const int m, const int n, + const float alpha, const float *a, const int lda, const float *b, const int ldb, + const float beta, float *c, const int ldc, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); - static - void symm_cm(const char side, const char uplo, const int m, const int n, - const double alpha, const double *a, const int lda, const double *b, const int ldb, - const double beta, double *c, const int ldc, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); + static + void symm_cm(const char side, const char uplo, const int m, const int n, + const double alpha, const double *a, const int lda, const double *b, const int ldb, + const double beta, double *c, const int ldc, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); static void symm_cm(const char side, const char uplo, const int m, const int n, const std::complex alpha, const std::complex *a, const int lda, const std::complex *b, const int ldb, const std::complex beta, std::complex *c, const int ldc, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); - static - void symm_cm(const char side, const char uplo, const int m, const int n, - const std::complex alpha, const std::complex *a, const int lda, const std::complex *b, const int ldb, - const std::complex beta, std::complex *c, const int ldc, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); - - // side=='L': C = a * A * B + b * C. - // side=='R': C = a * B * A + b * C. - // A == A^H - static - void hemm_cm(const char side, const char uplo, const int m, const int n, - const float alpha, const float *a, const int lda, const float *b, const int ldb, - const float beta, float *c, const int ldc, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); - - static - void hemm_cm(const char side, const char uplo, const int m, const int n, - const double alpha, const double *a, const int lda, const double *b, const int ldb, - const double beta, double *c, const int ldc, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); - - static + static + void symm_cm(const char side, const char uplo, const int m, const int n, + const std::complex alpha, const std::complex *a, const int lda, const std::complex *b, const int ldb, + const std::complex beta, std::complex *c, const int ldc, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); + + // side=='L': C = a * A * B + b * C. + // side=='R': C = a * B * A + b * C. + // A == A^H + static + void hemm_cm(const char side, const char uplo, const int m, const int n, + const float alpha, const float *a, const int lda, const float *b, const int ldb, + const float beta, float *c, const int ldc, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); + + static + void hemm_cm(const char side, const char uplo, const int m, const int n, + const double alpha, const double *a, const int lda, const double *b, const int ldb, + const double beta, double *c, const int ldc, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); + + static void hemm_cm(char side, char uplo, int m, int n, std::complex alpha, std::complex *a, int lda, std::complex *b, int ldb, std::complex beta, std::complex *c, int ldc, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); - static - void hemm_cm(char side, char uplo, int m, int n, - std::complex alpha, std::complex *a, int lda, std::complex *b, int ldb, - std::complex beta, std::complex *c, int ldc, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); + static + void hemm_cm(char side, char uplo, int m, int n, + std::complex alpha, std::complex *a, int lda, std::complex *b, int ldb, + std::complex beta, std::complex *c, int ldc, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); - // y = A*x + beta*y - static - void gemv(const char trans, const int m, const int n, + // y = A*x + beta*y + static + void gemv(const char trans, const int m, const int n, const float alpha, const float* A, const int lda, const float* X, const int incx, const float beta, float* Y, const int incy, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); @@ -380,67 +380,67 @@ class BlasConnector const std::complex alpha, const std::complex *A, const int lda, const std::complex *X, const int incx, const std::complex beta, std::complex *Y, const int incy, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); - // Peize Lin add 2018-06-12 - // out = ||x||_2 - static - float nrm2( const int n, const float *X, const int, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice ); - - static - double nrm2( const int n, const double *X, const int incX, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice ); - - static - double nrm2( const int n, const std::complex *X, const int incX, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice ); - - - // copies a into b - static - void copy(const int n, const double *a, const int incx, double *b, const int incy, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); - - static - void copy(const int n, const float *a, const int incx, float *b, const int incy, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); - - static - void copy(const int n, const std::complex *a, const int incx, std::complex *b, const int incy, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); - - static - void copy(const int n, const std::complex *a, const int incx, std::complex *b, const int incy, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); - - // There is some other operators needed, so implemented manually here - template - static - void vector_mul_vector(const int& dim, - T* result, - const T* vector1, - const Operand* vector2, - base_device::AbacusDevice_t device_type); - - template - static - void vector_div_vector(const int& dim, - T* result, - const T* vector1, - const Operand* vector2, - base_device::AbacusDevice_t device_type); - - // result = constant1 * vector1 + constant2 * vector2 - template - static - void vector_add_vector(const int& dim, - T* result, - const T* vector1, - const Scalar constant1, - const T* vector2, - const Scalar constant2, - base_device::AbacusDevice_t device_type); + // Peize Lin add 2018-06-12 + // out = ||x||_2 + static + float nrm2( const int n, const float *X, const int, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice ); + + static + double nrm2( const int n, const double *X, const int incX, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice ); + + static + double nrm2( const int n, const std::complex *X, const int incX, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice ); + + + // copies a into b + static + void copy(const int n, const double *a, const int incx, double *b, const int incy, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); + + static + void copy(const int n, const float *a, const int incx, float *b, const int incy, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); + + static + void copy(const int n, const std::complex *a, const int incx, std::complex *b, const int incy, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); + + static + void copy(const int n, const std::complex *a, const int incx, std::complex *b, const int incy, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice); + + // There is some other operators needed, so implemented manually here + template + static + void vector_mul_vector(const int& dim, + T* result, + const T* vector1, + const Operand* vector2, + base_device::AbacusDevice_t device_type); + + template + static + void vector_div_vector(const int& dim, + T* result, + const T* vector1, + const Operand* vector2, + base_device::AbacusDevice_t device_type); + + // result = constant1 * vector1 + constant2 * vector2 + template + static + void vector_add_vector(const int& dim, + T* result, + const T* vector1, + const Scalar constant1, + const T* vector2, + const Scalar constant2, + base_device::AbacusDevice_t device_type); #ifdef __DSP - /// @brief Inject the DSP cluster id used by mt-allocator BLAS kernels. - /// Caller-injected (typically once after input parameters are read). - /// Defaults to 0 if never set. - static void set_dsp_cluster_id(int id); + /// @brief Inject the DSP cluster id used by mt-allocator BLAS kernels. + /// Caller-injected (typically once after input parameters are read). + /// Defaults to 0 if never set. + static void set_dsp_cluster_id(int id); private: - static int dsp_cluster_id_; + static int dsp_cluster_id_; #endif }; @@ -454,17 +454,17 @@ class BlasConnector namespace BlasUtils{ - static cublasHandle_t cublas_handle = nullptr; + static cublasHandle_t cublas_handle = nullptr; - void createGpuBlasHandle(); // Create a cublas/hipblas handle. + void createGpuBlasHandle(); // Create a cublas/hipblas handle. - void destoryBLAShandle(); // Destroy the cublas/hipblas handle. Do this when the software is about to end. + void destoryBLAShandle(); // Destroy the cublas/hipblas handle. Do this when the software is about to end. - cublasOperation_t judge_trans(bool is_complex, const char& trans, const char* name); // Translate a normal transpose parameter to a cublas/hipblas type. + cublasOperation_t judge_trans(bool is_complex, const char& trans, const char* name); // Translate a normal transpose parameter to a cublas/hipblas type. - cublasSideMode_t judge_side(const char& trans); // Translate a normal side parameter to a cublas/hipblas type. + cublasSideMode_t judge_side(const char& trans); // Translate a normal side parameter to a cublas/hipblas type. - cublasFillMode_t judge_fill(const char& trans); // Translate a normal fill parameter to a cublas/hipblas type. + cublasFillMode_t judge_fill(const char& trans); // Translate a normal fill parameter to a cublas/hipblas type. } diff --git a/source/source_base/module_external/blas_connector_base.cpp b/source/source_base/module_external/blas_connector_base.cpp index b07f8a3a780..be0bc342c00 100644 --- a/source/source_base/module_external/blas_connector_base.cpp +++ b/source/source_base/module_external/blas_connector_base.cpp @@ -11,66 +11,66 @@ namespace BlasUtils{ - void createGpuBlasHandle(){ - if (cublas_handle == nullptr) { - CHECK_CUBLAS(cublasCreate(&cublas_handle)); - } - } + void createGpuBlasHandle(){ + if (cublas_handle == nullptr) { + CHECK_CUBLAS(cublasCreate(&cublas_handle)); + } + } - void destoryBLAShandle(){ - if (cublas_handle != nullptr) { - CHECK_CUBLAS(cublasDestroy(cublas_handle)); - cublas_handle = nullptr; - } - } + void destoryBLAShandle(){ + if (cublas_handle != nullptr) { + CHECK_CUBLAS(cublasDestroy(cublas_handle)); + cublas_handle = nullptr; + } + } - cublasOperation_t judge_trans(bool is_complex, const char& trans, const char* name) - { - if (trans == 'N') - { - return CUBLAS_OP_N; - } - else if(trans == 'T') - { - return CUBLAS_OP_T; - } - else if(is_complex && trans == 'C') - { - return CUBLAS_OP_C; - } - return CUBLAS_OP_N; - } + cublasOperation_t judge_trans(bool is_complex, const char& trans, const char* name) + { + if (trans == 'N') + { + return CUBLAS_OP_N; + } + else if(trans == 'T') + { + return CUBLAS_OP_T; + } + else if(is_complex && trans == 'C') + { + return CUBLAS_OP_C; + } + return CUBLAS_OP_N; + } - cublasSideMode_t judge_side(const char& trans) - { - if (trans == 'L') - { - return CUBLAS_SIDE_LEFT; - } - else if (trans == 'R') - { - return CUBLAS_SIDE_RIGHT; - } - return CUBLAS_SIDE_LEFT; - } + cublasSideMode_t judge_side(const char& trans) + { + if (trans == 'L') + { + return CUBLAS_SIDE_LEFT; + } + else if (trans == 'R') + { + return CUBLAS_SIDE_RIGHT; + } + return CUBLAS_SIDE_LEFT; + } - cublasFillMode_t judge_fill(const char& trans) - { - if (trans == 'F') - { - return CUBLAS_FILL_MODE_FULL; - } - else if (trans == 'U') - { - return CUBLAS_FILL_MODE_UPPER; - } - else if (trans == 'D') - { - return CUBLAS_FILL_MODE_LOWER; - } - return CUBLAS_FILL_MODE_FULL; - } + cublasFillMode_t judge_fill(const char& trans) + { + if (trans == 'F') + { + return CUBLAS_FILL_MODE_FULL; + } + else if (trans == 'U') + { + return CUBLAS_FILL_MODE_UPPER; + } + else if (trans == 'D') + { + return CUBLAS_FILL_MODE_LOWER; + } + return CUBLAS_FILL_MODE_FULL; + } } // namespace BlasUtils diff --git a/source/source_base/module_external/blas_connector_matrix.cpp b/source/source_base/module_external/blas_connector_matrix.cpp index 1e819d56ae0..964ce5cd9db 100644 --- a/source/source_base/module_external/blas_connector_matrix.cpp +++ b/source/source_base/module_external/blas_connector_matrix.cpp @@ -24,31 +24,31 @@ void BlasConnector::set_dsp_cluster_id(int id) // C = a * A.? * B.? + b * C // Row-Major part void BlasConnector::gemm(const char transa, const char transb, const int m, const int n, const int k, - const float alpha, const float *a, const int lda, const float *b, const int ldb, - const float beta, float *c, const int ldc, base_device::AbacusDevice_t device_type) + const float alpha, const float *a, const int lda, const float *b, const int ldb, + const float beta, float *c, const int ldc, base_device::AbacusDevice_t device_type) { - if (device_type == base_device::AbacusDevice_t::CpuDevice) { - sgemm_(&transb, &transa, &n, &m, &k, - &alpha, b, &ldb, a, &lda, - &beta, c, &ldc); - } + if (device_type == base_device::AbacusDevice_t::CpuDevice) { + sgemm_(&transb, &transa, &n, &m, &k, + &alpha, b, &ldb, a, &lda, + &beta, c, &ldc); + } #ifdef __DSP - else if (device_type == base_device::AbacusDevice_t::DspDevice){ - mtfunc::sgemm_mth_(&transb, &transa, &n, &m, &k, - &alpha, b, &ldb, a, &lda, - &beta, c, &ldc, BlasConnector::dsp_cluster_id_); - } + else if (device_type == base_device::AbacusDevice_t::DspDevice){ + mtfunc::sgemm_mth_(&transb, &transa, &n, &m, &k, + &alpha, b, &ldb, a, &lda, + &beta, c, &ldc, BlasConnector::dsp_cluster_id_); + } #endif #ifdef __CUDA - else if (device_type == base_device::AbacusDevice_t::GpuDevice) { - cublasOperation_t cutransA = BlasUtils::judge_trans(false, transa, "gemm_op"); - cublasOperation_t cutransB = BlasUtils::judge_trans(false, transb, "gemm_op"); - CHECK_CUBLAS(cublasSgemm(BlasUtils::cublas_handle, cutransA, cutransB, n, m, k, &alpha, b, ldb, a, lda, &beta, c, ldc)); - } -#endif - else { - throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); - } + else if (device_type == base_device::AbacusDevice_t::GpuDevice) { + cublasOperation_t cutransA = BlasUtils::judge_trans(false, transa, "gemm_op"); + cublasOperation_t cutransB = BlasUtils::judge_trans(false, transb, "gemm_op"); + CHECK_CUBLAS(cublasSgemm(BlasUtils::cublas_handle, cutransA, cutransB, n, m, k, &alpha, b, ldb, a, lda, &beta, c, ldc)); + } +#endif + else { + throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); + } } void BlasConnector::gemm(const char transa, @@ -85,9 +85,9 @@ void BlasConnector::gemm(const char transa, cublasDgemm(BlasUtils::cublas_handle, cutransA, cutransB, n, m, k, &alpha, b, ldb, a, lda, &beta, c, ldc)); #endif } - else { - throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); - } + else { + throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); + } } void BlasConnector::gemm(const char transa, @@ -138,9 +138,9 @@ void BlasConnector::gemm(const char transa, ldc)); #endif } - else { - throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); - } + else { + throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); + } } void BlasConnector::gemm(const char transa, @@ -191,38 +191,38 @@ void BlasConnector::gemm(const char transa, ldc)); #endif } - else { - throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); - } + else { + throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); + } } // Col-Major part void BlasConnector::gemm_cm(const char transa, const char transb, const int m, const int n, const int k, - const float alpha, const float *a, const int lda, const float *b, const int ldb, - const float beta, float *c, const int ldc, base_device::AbacusDevice_t device_type) + const float alpha, const float *a, const int lda, const float *b, const int ldb, + const float beta, float *c, const int ldc, base_device::AbacusDevice_t device_type) { - if (device_type == base_device::AbacusDevice_t::CpuDevice) { - sgemm_(&transa, &transb, &m, &n, &k, - &alpha, a, &lda, b, &ldb, - &beta, c, &ldc); - } + if (device_type == base_device::AbacusDevice_t::CpuDevice) { + sgemm_(&transa, &transb, &m, &n, &k, + &alpha, a, &lda, b, &ldb, + &beta, c, &ldc); + } #ifdef __DSP - else if (device_type == base_device::AbacusDevice_t::DspDevice){ - mtfunc::sgemm_mth_(&transb, &transa, &m, &n, &k, - &alpha, a, &lda, b, &ldb, - &beta, c, &ldc, BlasConnector::dsp_cluster_id_); - } + else if (device_type == base_device::AbacusDevice_t::DspDevice){ + mtfunc::sgemm_mth_(&transb, &transa, &m, &n, &k, + &alpha, a, &lda, b, &ldb, + &beta, c, &ldc, BlasConnector::dsp_cluster_id_); + } #endif #ifdef __CUDA - else if (device_type == base_device::AbacusDevice_t::GpuDevice) { - cublasOperation_t cutransA = BlasUtils::judge_trans(false, transa, "gemm_op"); - cublasOperation_t cutransB = BlasUtils::judge_trans(false, transb, "gemm_op"); - CHECK_CUBLAS(cublasSgemm(BlasUtils::cublas_handle, cutransA, cutransB, m, n, k, &alpha, a, lda, b, ldb, &beta, c, ldc)); - } -#endif - else { - throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); - } + else if (device_type == base_device::AbacusDevice_t::GpuDevice) { + cublasOperation_t cutransA = BlasUtils::judge_trans(false, transa, "gemm_op"); + cublasOperation_t cutransB = BlasUtils::judge_trans(false, transb, "gemm_op"); + CHECK_CUBLAS(cublasSgemm(BlasUtils::cublas_handle, cutransA, cutransB, m, n, k, &alpha, a, lda, b, ldb, &beta, c, ldc)); + } +#endif + else { + throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); + } } void BlasConnector::gemm_cm(const char transa, @@ -259,9 +259,9 @@ void BlasConnector::gemm_cm(const char transa, cublasDgemm(BlasUtils::cublas_handle, cutransA, cutransB, m, n, k, &alpha, a, lda, b, ldb, &beta, c, ldc)); } #endif - else { - throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); - } + else { + throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); + } } void BlasConnector::gemm_cm(const char transa, @@ -312,9 +312,9 @@ void BlasConnector::gemm_cm(const char transa, ldc)); } #endif - else { - throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); - } + else { + throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); + } } void BlasConnector::gemm_cm(const char transa, @@ -365,160 +365,160 @@ void BlasConnector::gemm_cm(const char transa, ldc)); } #endif - else { - throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); - } + else { + throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); + } } // Symm and Hemm part. Only col-major is supported. void BlasConnector::symm_cm(const char side, const char uplo, const int m, const int n, - const float alpha, const float *a, const int lda, const float *b, const int ldb, - const float beta, float *c, const int ldc, base_device::AbacusDevice_t device_type) + const float alpha, const float *a, const int lda, const float *b, const int ldb, + const float beta, float *c, const int ldc, base_device::AbacusDevice_t device_type) { - if (device_type == base_device::AbacusDevice_t::CpuDevice) { - ssymm_(&side, &uplo, &m, &n, - &alpha, a, &lda, b, &ldb, - &beta, c, &ldc); - } + if (device_type == base_device::AbacusDevice_t::CpuDevice) { + ssymm_(&side, &uplo, &m, &n, + &alpha, a, &lda, b, &ldb, + &beta, c, &ldc); + } #ifdef __CUDA - else if (device_type == base_device::AbacusDevice_t::GpuDevice) { - cublasSideMode_t sideMode = BlasUtils::judge_side(side); - cublasFillMode_t fillMode = BlasUtils::judge_fill(uplo); - CHECK_CUBLAS(cublasSsymm(BlasUtils::cublas_handle, sideMode, fillMode, m, n, &alpha, a, lda, b, ldb, &beta, c, ldc)); - } -#endif - else { - throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); - } + else if (device_type == base_device::AbacusDevice_t::GpuDevice) { + cublasSideMode_t sideMode = BlasUtils::judge_side(side); + cublasFillMode_t fillMode = BlasUtils::judge_fill(uplo); + CHECK_CUBLAS(cublasSsymm(BlasUtils::cublas_handle, sideMode, fillMode, m, n, &alpha, a, lda, b, ldb, &beta, c, ldc)); + } +#endif + else { + throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); + } } void BlasConnector::symm_cm(const char side, const char uplo, const int m, const int n, - const double alpha, const double *a, const int lda, const double *b, const int ldb, - const double beta, double *c, const int ldc, base_device::AbacusDevice_t device_type) + const double alpha, const double *a, const int lda, const double *b, const int ldb, + const double beta, double *c, const int ldc, base_device::AbacusDevice_t device_type) { - if (device_type == base_device::AbacusDevice_t::CpuDevice) { - dsymm_(&side, &uplo, &m, &n, - &alpha, a, &lda, b, &ldb, - &beta, c, &ldc); - } + if (device_type == base_device::AbacusDevice_t::CpuDevice) { + dsymm_(&side, &uplo, &m, &n, + &alpha, a, &lda, b, &ldb, + &beta, c, &ldc); + } #ifdef __CUDA - else if (device_type == base_device::AbacusDevice_t::GpuDevice) { - cublasSideMode_t sideMode = BlasUtils::judge_side(side); - cublasFillMode_t fillMode = BlasUtils::judge_fill(uplo); - CHECK_CUBLAS(cublasDsymm(BlasUtils::cublas_handle, sideMode, fillMode, m, n, &alpha, a, lda, b, ldb, &beta, c, ldc)); - } -#endif - else { - throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); - } + else if (device_type == base_device::AbacusDevice_t::GpuDevice) { + cublasSideMode_t sideMode = BlasUtils::judge_side(side); + cublasFillMode_t fillMode = BlasUtils::judge_fill(uplo); + CHECK_CUBLAS(cublasDsymm(BlasUtils::cublas_handle, sideMode, fillMode, m, n, &alpha, a, lda, b, ldb, &beta, c, ldc)); + } +#endif + else { + throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); + } } void BlasConnector::symm_cm(const char side, const char uplo, const int m, const int n, const std::complex alpha, const std::complex *a, const int lda, const std::complex *b, const int ldb, const std::complex beta, std::complex *c, const int ldc, base_device::AbacusDevice_t device_type) { - if (device_type == base_device::AbacusDevice_t::CpuDevice) { - csymm_(&side, &uplo, &m, &n, - &alpha, a, &lda, b, &ldb, - &beta, c, &ldc); - } + if (device_type == base_device::AbacusDevice_t::CpuDevice) { + csymm_(&side, &uplo, &m, &n, + &alpha, a, &lda, b, &ldb, + &beta, c, &ldc); + } #ifdef __CUDA - else if (device_type == base_device::AbacusDevice_t::GpuDevice) { - cublasSideMode_t sideMode = BlasUtils::judge_side(side); - cublasFillMode_t fillMode = BlasUtils::judge_fill(uplo); - CHECK_CUBLAS(cublasCsymm(BlasUtils::cublas_handle, sideMode, fillMode, m, n, (float2*)&alpha, (float2*)a, lda, (float2*)b, ldb, (float2*)&beta, (float2*)c, ldc)); - } -#endif - else { - throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); - } + else if (device_type == base_device::AbacusDevice_t::GpuDevice) { + cublasSideMode_t sideMode = BlasUtils::judge_side(side); + cublasFillMode_t fillMode = BlasUtils::judge_fill(uplo); + CHECK_CUBLAS(cublasCsymm(BlasUtils::cublas_handle, sideMode, fillMode, m, n, (float2*)&alpha, (float2*)a, lda, (float2*)b, ldb, (float2*)&beta, (float2*)c, ldc)); + } +#endif + else { + throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); + } } void BlasConnector::symm_cm(const char side, const char uplo, const int m, const int n, - const std::complex alpha, const std::complex *a, const int lda, const std::complex *b, const int ldb, - const std::complex beta, std::complex *c, const int ldc, base_device::AbacusDevice_t device_type) + const std::complex alpha, const std::complex *a, const int lda, const std::complex *b, const int ldb, + const std::complex beta, std::complex *c, const int ldc, base_device::AbacusDevice_t device_type) { - if (device_type == base_device::AbacusDevice_t::CpuDevice) { - zsymm_(&side, &uplo, &m, &n, - &alpha, a, &lda, b, &ldb, - &beta, c, &ldc); - } + if (device_type == base_device::AbacusDevice_t::CpuDevice) { + zsymm_(&side, &uplo, &m, &n, + &alpha, a, &lda, b, &ldb, + &beta, c, &ldc); + } #ifdef __CUDA - else if (device_type == base_device::AbacusDevice_t::GpuDevice) { - cublasSideMode_t sideMode = BlasUtils::judge_side(side); - cublasFillMode_t fillMode = BlasUtils::judge_fill(uplo); - CHECK_CUBLAS(cublasZsymm(BlasUtils::cublas_handle, sideMode, fillMode, m, n, (double2*)&alpha, (double2*)a, lda, (double2*)b, ldb, (double2*)&beta, (double2*)c, ldc)); - } -#endif - else { - throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); - } + else if (device_type == base_device::AbacusDevice_t::GpuDevice) { + cublasSideMode_t sideMode = BlasUtils::judge_side(side); + cublasFillMode_t fillMode = BlasUtils::judge_fill(uplo); + CHECK_CUBLAS(cublasZsymm(BlasUtils::cublas_handle, sideMode, fillMode, m, n, (double2*)&alpha, (double2*)a, lda, (double2*)b, ldb, (double2*)&beta, (double2*)c, ldc)); + } +#endif + else { + throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); + } } void BlasConnector::hemm_cm(const char side, const char uplo, const int m, const int n, - const float alpha, const float *a, const int lda, const float *b, const int ldb, - const float beta, float *c, const int ldc, base_device::AbacusDevice_t device_type) + const float alpha, const float *a, const int lda, const float *b, const int ldb, + const float beta, float *c, const int ldc, base_device::AbacusDevice_t device_type) { - symm_cm(side, uplo, m, n, alpha, a, lda, b, ldb, beta, c, ldc, device_type); + symm_cm(side, uplo, m, n, alpha, a, lda, b, ldb, beta, c, ldc, device_type); } void BlasConnector::hemm_cm(const char side, const char uplo, const int m, const int n, - const double alpha, const double *a, const int lda, const double *b, const int ldb, - const double beta, double *c, const int ldc, base_device::AbacusDevice_t device_type) + const double alpha, const double *a, const int lda, const double *b, const int ldb, + const double beta, double *c, const int ldc, base_device::AbacusDevice_t device_type) { - symm_cm(side, uplo, m, n, alpha, a, lda, b, ldb, beta, c, ldc, device_type); + symm_cm(side, uplo, m, n, alpha, a, lda, b, ldb, beta, c, ldc, device_type); } void BlasConnector::hemm_cm(char side, char uplo, int m, int n, std::complex alpha, std::complex *a, int lda, std::complex *b, int ldb, std::complex beta, std::complex *c, int ldc, base_device::AbacusDevice_t device_type) { - if (device_type == base_device::AbacusDevice_t::CpuDevice) { - chemm_(&side, &uplo, &m, &n, - &alpha, a, &lda, b, &ldb, - &beta, c, &ldc); - } + if (device_type == base_device::AbacusDevice_t::CpuDevice) { + chemm_(&side, &uplo, &m, &n, + &alpha, a, &lda, b, &ldb, + &beta, c, &ldc); + } #ifdef __CUDA - else if (device_type == base_device::AbacusDevice_t::GpuDevice) { - cublasSideMode_t sideMode = BlasUtils::judge_side(side); - cublasFillMode_t fillMode = BlasUtils::judge_fill(uplo); - CHECK_CUBLAS(cublasChemm(BlasUtils::cublas_handle, sideMode, fillMode, m, n, (float2*)&alpha, (float2*)a, lda, (float2*)b, ldb, (float2*)&beta, (float2*)c, ldc)); - } -#endif - else { - throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); - } + else if (device_type == base_device::AbacusDevice_t::GpuDevice) { + cublasSideMode_t sideMode = BlasUtils::judge_side(side); + cublasFillMode_t fillMode = BlasUtils::judge_fill(uplo); + CHECK_CUBLAS(cublasChemm(BlasUtils::cublas_handle, sideMode, fillMode, m, n, (float2*)&alpha, (float2*)a, lda, (float2*)b, ldb, (float2*)&beta, (float2*)c, ldc)); + } +#endif + else { + throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); + } } void BlasConnector::hemm_cm(char side, char uplo, int m, int n, - std::complex alpha, std::complex *a, int lda, std::complex *b, int ldb, - std::complex beta, std::complex *c, int ldc, base_device::AbacusDevice_t device_type) + std::complex alpha, std::complex *a, int lda, std::complex *b, int ldb, + std::complex beta, std::complex *c, int ldc, base_device::AbacusDevice_t device_type) { - if (device_type == base_device::AbacusDevice_t::CpuDevice) { - zhemm_(&side, &uplo, &m, &n, - &alpha, a, &lda, b, &ldb, - &beta, c, &ldc); - } + if (device_type == base_device::AbacusDevice_t::CpuDevice) { + zhemm_(&side, &uplo, &m, &n, + &alpha, a, &lda, b, &ldb, + &beta, c, &ldc); + } #ifdef __CUDA - else if (device_type == base_device::AbacusDevice_t::GpuDevice) { - cublasSideMode_t sideMode = BlasUtils::judge_side(side); - cublasFillMode_t fillMode = BlasUtils::judge_fill(uplo); - CHECK_CUBLAS(cublasZhemm(BlasUtils::cublas_handle, sideMode, fillMode, m, n, (double2*)&alpha, (double2*)a, lda, (double2*)b, ldb, (double2*)&beta, (double2*)c, ldc)); - } -#endif - else { - throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); - } + else if (device_type == base_device::AbacusDevice_t::GpuDevice) { + cublasSideMode_t sideMode = BlasUtils::judge_side(side); + cublasFillMode_t fillMode = BlasUtils::judge_fill(uplo); + CHECK_CUBLAS(cublasZhemm(BlasUtils::cublas_handle, sideMode, fillMode, m, n, (double2*)&alpha, (double2*)a, lda, (double2*)b, ldb, (double2*)&beta, (double2*)c, ldc)); + } +#endif + else { + throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); + } } void BlasConnector::gemv(const char trans, const int m, const int n, const float alpha, const float* A, const int lda, const float* X, const int incx, const float beta, float* Y, const int incy, base_device::AbacusDevice_t device_type) { - if (device_type == base_device::AbacusDevice_t::CpuDevice) { - sgemv_(&trans, &m, &n, &alpha, A, &lda, X, &incx, &beta, Y, &incy); - } + if (device_type == base_device::AbacusDevice_t::CpuDevice) { + sgemv_(&trans, &m, &n, &alpha, A, &lda, X, &incx, &beta, Y, &incy); + } #ifdef __DSP else if (device_type == base_device::AbacusDevice_t::DspDevice) { mtfunc::sgemv_mth_(&trans, @@ -536,23 +536,23 @@ void BlasConnector::gemv(const char trans, const int m, const int n, } #endif #ifdef __CUDA - else if (device_type == base_device::AbacusDevice_t::GpuDevice) { - cublasOperation_t cutransA = BlasUtils::judge_trans(false, trans, "gemv_op"); - CHECK_CUBLAS(cublasSgemv(BlasUtils::cublas_handle, cutransA, m, n, &alpha, A, lda, X, incx, &beta, Y, incy)); - } -#endif - else { - throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); - } + else if (device_type == base_device::AbacusDevice_t::GpuDevice) { + cublasOperation_t cutransA = BlasUtils::judge_trans(false, trans, "gemv_op"); + CHECK_CUBLAS(cublasSgemv(BlasUtils::cublas_handle, cutransA, m, n, &alpha, A, lda, X, incx, &beta, Y, incy)); + } +#endif + else { + throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); + } } void BlasConnector::gemv(const char trans, const int m, const int n, const double alpha, const double* A, const int lda, const double* X, const int incx, const double beta, double* Y, const int incy, base_device::AbacusDevice_t device_type) { - if (device_type == base_device::AbacusDevice_t::CpuDevice) { - dgemv_(&trans, &m, &n, &alpha, A, &lda, X, &incx, &beta, Y, &incy); - } + if (device_type == base_device::AbacusDevice_t::CpuDevice) { + dgemv_(&trans, &m, &n, &alpha, A, &lda, X, &incx, &beta, Y, &incy); + } #ifdef __DSP else if (device_type == base_device::AbacusDevice_t::DspDevice) { mtfunc::dgemv_mth_(&trans, @@ -570,23 +570,23 @@ void BlasConnector::gemv(const char trans, const int m, const int n, } #endif #ifdef __CUDA - else if (device_type == base_device::AbacusDevice_t::GpuDevice) { - cublasOperation_t cutransA = BlasUtils::judge_trans(false, trans, "gemv_op"); - CHECK_CUBLAS(cublasDgemv(BlasUtils::cublas_handle, cutransA, m, n, &alpha, A, lda, X, incx, &beta, Y, incy)); - } -#endif - else { - throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); - } + else if (device_type == base_device::AbacusDevice_t::GpuDevice) { + cublasOperation_t cutransA = BlasUtils::judge_trans(false, trans, "gemv_op"); + CHECK_CUBLAS(cublasDgemv(BlasUtils::cublas_handle, cutransA, m, n, &alpha, A, lda, X, incx, &beta, Y, incy)); + } +#endif + else { + throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); + } } void BlasConnector::gemv(const char trans, const int m, const int n, const std::complex alpha, const std::complex *A, const int lda, const std::complex *X, const int incx, const std::complex beta, std::complex *Y, const int incy, base_device::AbacusDevice_t device_type) { - if (device_type == base_device::AbacusDevice_t::CpuDevice) { - cgemv_(&trans, &m, &n, &alpha, A, &lda, X, &incx, &beta, Y, &incy); - } + if (device_type == base_device::AbacusDevice_t::CpuDevice) { + cgemv_(&trans, &m, &n, &alpha, A, &lda, X, &incx, &beta, Y, &incy); + } #ifdef __DSP else if (device_type == base_device::AbacusDevice_t::DspDevice) { mtfunc::cgemv_mth_(&trans, @@ -604,25 +604,25 @@ void BlasConnector::gemv(const char trans, const int m, const int n, } #endif #ifdef __CUDA - else if (device_type == base_device::AbacusDevice_t::GpuDevice) { - cuFloatComplex alpha_cu = make_cuFloatComplex(alpha.real(), alpha.imag()); - cuFloatComplex beta_cu = make_cuFloatComplex(beta.real(), beta.imag()); - cublasOperation_t cutransA = BlasUtils::judge_trans(true, trans, "gemv_op"); - CHECK_CUBLAS(cublasCgemv(BlasUtils::cublas_handle, cutransA, m, n, &alpha_cu, (cuFloatComplex*)A, lda, (cuFloatComplex*)X, incx, &beta_cu, (cuFloatComplex*)Y, incy)); - } -#endif - else { - throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); - } + else if (device_type == base_device::AbacusDevice_t::GpuDevice) { + cuFloatComplex alpha_cu = make_cuFloatComplex(alpha.real(), alpha.imag()); + cuFloatComplex beta_cu = make_cuFloatComplex(beta.real(), beta.imag()); + cublasOperation_t cutransA = BlasUtils::judge_trans(true, trans, "gemv_op"); + CHECK_CUBLAS(cublasCgemv(BlasUtils::cublas_handle, cutransA, m, n, &alpha_cu, (cuFloatComplex*)A, lda, (cuFloatComplex*)X, incx, &beta_cu, (cuFloatComplex*)Y, incy)); + } +#endif + else { + throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); + } } void BlasConnector::gemv(const char trans, const int m, const int n, const std::complex alpha, const std::complex *A, const int lda, const std::complex *X, const int incx, const std::complex beta, std::complex *Y, const int incy, base_device::AbacusDevice_t device_type) { - if (device_type == base_device::AbacusDevice_t::CpuDevice) { - zgemv_(&trans, &m, &n, &alpha, A, &lda, X, &incx, &beta, Y, &incy); - } + if (device_type == base_device::AbacusDevice_t::CpuDevice) { + zgemv_(&trans, &m, &n, &alpha, A, &lda, X, &incx, &beta, Y, &incy); + } #ifdef __DSP else if (device_type == base_device::AbacusDevice_t::DspDevice) { mtfunc::zgemv_mth_(&trans, @@ -640,14 +640,14 @@ void BlasConnector::gemv(const char trans, const int m, const int n, } #endif #ifdef __CUDA - else if (device_type == base_device::AbacusDevice_t::GpuDevice) { - cuDoubleComplex alpha_cu = make_cuDoubleComplex(alpha.real(), alpha.imag()); - cuDoubleComplex beta_cu = make_cuDoubleComplex(beta.real(), beta.imag()); - cublasOperation_t cutransA = BlasUtils::judge_trans(true, trans, "gemv_op"); - CHECK_CUBLAS(cublasZgemv(BlasUtils::cublas_handle, cutransA, m, n, &alpha_cu, (cuDoubleComplex*)A, lda, (cuDoubleComplex*)X, incx, &beta_cu, (cuDoubleComplex*)Y, incy)); - } -#endif - else { - throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); - } + else if (device_type == base_device::AbacusDevice_t::GpuDevice) { + cuDoubleComplex alpha_cu = make_cuDoubleComplex(alpha.real(), alpha.imag()); + cuDoubleComplex beta_cu = make_cuDoubleComplex(beta.real(), beta.imag()); + cublasOperation_t cutransA = BlasUtils::judge_trans(true, trans, "gemv_op"); + CHECK_CUBLAS(cublasZgemv(BlasUtils::cublas_handle, cutransA, m, n, &alpha_cu, (cuDoubleComplex*)A, lda, (cuDoubleComplex*)X, incx, &beta_cu, (cuDoubleComplex*)Y, incy)); + } +#endif + else { + throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); + } } \ No newline at end of file diff --git a/source/source_base/module_external/blas_connector_vector.cpp b/source/source_base/module_external/blas_connector_vector.cpp index 7aa7296a96b..37b4c327660 100644 --- a/source/source_base/module_external/blas_connector_vector.cpp +++ b/source/source_base/module_external/blas_connector_vector.cpp @@ -17,349 +17,349 @@ void BlasConnector::axpy( const int n, const float alpha, const float *X, const int incX, float *Y, const int incY, base_device::AbacusDevice_t device_type) { - if (device_type == base_device::AbacusDevice_t::CpuDevice) { - saxpy_(&n, &alpha, X, &incX, Y, &incY); - } + if (device_type == base_device::AbacusDevice_t::CpuDevice) { + saxpy_(&n, &alpha, X, &incX, Y, &incY); + } #ifdef __CUDA - else if (device_type == base_device::AbacusDevice_t::GpuDevice) { - CHECK_CUBLAS(cublasSaxpy(BlasUtils::cublas_handle, n, &alpha, X, incX, Y, incY)); - } + else if (device_type == base_device::AbacusDevice_t::GpuDevice) { + CHECK_CUBLAS(cublasSaxpy(BlasUtils::cublas_handle, n, &alpha, X, incX, Y, incY)); + } #endif - else { - throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); - } + else { + throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); + } } void BlasConnector::axpy( const int n, const double alpha, const double *X, const int incX, double *Y, const int incY, base_device::AbacusDevice_t device_type) { - if (device_type == base_device::AbacusDevice_t::CpuDevice) { - daxpy_(&n, &alpha, X, &incX, Y, &incY); - } + if (device_type == base_device::AbacusDevice_t::CpuDevice) { + daxpy_(&n, &alpha, X, &incX, Y, &incY); + } #ifdef __CUDA - else if (device_type == base_device::AbacusDevice_t::GpuDevice) { - CHECK_CUBLAS(cublasDaxpy(BlasUtils::cublas_handle, n, &alpha, X, incX, Y, incY)); - } + else if (device_type == base_device::AbacusDevice_t::GpuDevice) { + CHECK_CUBLAS(cublasDaxpy(BlasUtils::cublas_handle, n, &alpha, X, incX, Y, incY)); + } #endif - else { - throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); - } + else { + throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); + } } void BlasConnector::axpy( const int n, const std::complex alpha, const std::complex *X, const int incX, std::complex *Y, const int incY, base_device::AbacusDevice_t device_type) { - if (device_type == base_device::AbacusDevice_t::CpuDevice) { - caxpy_(&n, &alpha, X, &incX, Y, &incY); - } + if (device_type == base_device::AbacusDevice_t::CpuDevice) { + caxpy_(&n, &alpha, X, &incX, Y, &incY); + } #ifdef __CUDA - else if (device_type == base_device::AbacusDevice_t::GpuDevice) { - CHECK_CUBLAS(cublasCaxpy(BlasUtils::cublas_handle, n, (float2*)&alpha, (float2*)X, incX, (float2*)Y, incY)); - } + else if (device_type == base_device::AbacusDevice_t::GpuDevice) { + CHECK_CUBLAS(cublasCaxpy(BlasUtils::cublas_handle, n, (float2*)&alpha, (float2*)X, incX, (float2*)Y, incY)); + } #endif - else { - throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); - } + else { + throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); + } } void BlasConnector::axpy( const int n, const std::complex alpha, const std::complex *X, const int incX, std::complex *Y, const int incY, base_device::AbacusDevice_t device_type) { - if (device_type == base_device::AbacusDevice_t::CpuDevice) { - zaxpy_(&n, &alpha, X, &incX, Y, &incY); - } + if (device_type == base_device::AbacusDevice_t::CpuDevice) { + zaxpy_(&n, &alpha, X, &incX, Y, &incY); + } #ifdef __CUDA - else if (device_type == base_device::AbacusDevice_t::GpuDevice) { - CHECK_CUBLAS(cublasZaxpy(BlasUtils::cublas_handle, n, (double2*)&alpha, (double2*)X, incX, (double2*)Y, incY)); - } + else if (device_type == base_device::AbacusDevice_t::GpuDevice) { + CHECK_CUBLAS(cublasZaxpy(BlasUtils::cublas_handle, n, (double2*)&alpha, (double2*)X, incX, (double2*)Y, incY)); + } #endif - else { - throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); - } + else { + throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); + } } // x=a*x void BlasConnector::scal( const int n, const float alpha, float *X, const int incX, base_device::AbacusDevice_t device_type) { - if (device_type == base_device::AbacusDevice_t::CpuDevice) { - sscal_(&n, &alpha, X, &incX); - } + if (device_type == base_device::AbacusDevice_t::CpuDevice) { + sscal_(&n, &alpha, X, &incX); + } #ifdef __CUDA - else if (device_type == base_device::AbacusDevice_t::GpuDevice) { - CHECK_CUBLAS(cublasSscal(BlasUtils::cublas_handle, n, &alpha, X, incX)); - } + else if (device_type == base_device::AbacusDevice_t::GpuDevice) { + CHECK_CUBLAS(cublasSscal(BlasUtils::cublas_handle, n, &alpha, X, incX)); + } #endif - else { - throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); - } + else { + throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); + } } void BlasConnector::scal( const int n, const double alpha, double *X, const int incX, base_device::AbacusDevice_t device_type) { - if (device_type == base_device::AbacusDevice_t::CpuDevice) { - dscal_(&n, &alpha, X, &incX); - } + if (device_type == base_device::AbacusDevice_t::CpuDevice) { + dscal_(&n, &alpha, X, &incX); + } #ifdef __CUDA - else if (device_type == base_device::AbacusDevice_t::GpuDevice) { - CHECK_CUBLAS(cublasDscal(BlasUtils::cublas_handle, n, &alpha, X, incX)); - } + else if (device_type == base_device::AbacusDevice_t::GpuDevice) { + CHECK_CUBLAS(cublasDscal(BlasUtils::cublas_handle, n, &alpha, X, incX)); + } #endif - else { - throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); - } + else { + throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); + } } void BlasConnector::scal( const int n, const std::complex alpha, std::complex *X, const int incX, base_device::AbacusDevice_t device_type) { - if (device_type == base_device::AbacusDevice_t::CpuDevice) { - cscal_(&n, &alpha, X, &incX); - } + if (device_type == base_device::AbacusDevice_t::CpuDevice) { + cscal_(&n, &alpha, X, &incX); + } #ifdef __CUDA - else if (device_type == base_device::AbacusDevice_t::GpuDevice) { - CHECK_CUBLAS(cublasCscal(BlasUtils::cublas_handle, n, (float2*)&alpha, (float2*)X, incX)); - } + else if (device_type == base_device::AbacusDevice_t::GpuDevice) { + CHECK_CUBLAS(cublasCscal(BlasUtils::cublas_handle, n, (float2*)&alpha, (float2*)X, incX)); + } #endif - else { - throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); - } + else { + throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); + } } void BlasConnector::scal( const int n, const std::complex alpha, std::complex *X, const int incX, base_device::AbacusDevice_t device_type) { - if (device_type == base_device::AbacusDevice_t::CpuDevice) { - zscal_(&n, &alpha, X, &incX); - } + if (device_type == base_device::AbacusDevice_t::CpuDevice) { + zscal_(&n, &alpha, X, &incX); + } #ifdef __CUDA - else if (device_type == base_device::AbacusDevice_t::GpuDevice) { - CHECK_CUBLAS(cublasZscal(BlasUtils::cublas_handle, n, (double2*)&alpha, (double2*)X, incX)); - } + else if (device_type == base_device::AbacusDevice_t::GpuDevice) { + CHECK_CUBLAS(cublasZscal(BlasUtils::cublas_handle, n, (double2*)&alpha, (double2*)X, incX)); + } #endif - else { - throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); - } + else { + throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); + } } // d=x*y float BlasConnector::dot( const int n, const float*const X, const int incX, const float*const Y, const int incY, base_device::AbacusDevice_t device_type) { - if (device_type == base_device::AbacusDevice_t::CpuDevice) { - return sdot_(&n, X, &incX, Y, &incY); - } + if (device_type == base_device::AbacusDevice_t::CpuDevice) { + return sdot_(&n, X, &incX, Y, &incY); + } #ifdef __CUDA - else if (device_type == base_device::AbacusDevice_t::GpuDevice){ - float result = 0.0; - CHECK_CUBLAS(cublasSdot(BlasUtils::cublas_handle, n, X, incX, Y, incY, &result)); - return result; - } + else if (device_type == base_device::AbacusDevice_t::GpuDevice){ + float result = 0.0; + CHECK_CUBLAS(cublasSdot(BlasUtils::cublas_handle, n, X, incX, Y, incY, &result)); + return result; + } #endif - else { - throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); - } + else { + throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); + } } double BlasConnector::dot( const int n, const double*const X, const int incX, const double*const Y, const int incY, base_device::AbacusDevice_t device_type) { - if (device_type == base_device::AbacusDevice_t::CpuDevice) { - return ddot_(&n, X, &incX, Y, &incY); - } + if (device_type == base_device::AbacusDevice_t::CpuDevice) { + return ddot_(&n, X, &incX, Y, &incY); + } #ifdef __CUDA - else if (device_type == base_device::AbacusDevice_t::GpuDevice){ - double result = 0.0; - CHECK_CUBLAS(cublasDdot(BlasUtils::cublas_handle, n, X, incX, Y, incY, &result)); - return result; - } + else if (device_type == base_device::AbacusDevice_t::GpuDevice){ + double result = 0.0; + CHECK_CUBLAS(cublasDdot(BlasUtils::cublas_handle, n, X, incX, Y, incY, &result)); + return result; + } #endif - else { - throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); - } + else { + throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); + } } // d=x*y float BlasConnector::dotu(const int n, const float*const X, const int incX, const float*const Y, const int incY, base_device::AbacusDevice_t device_type) { - return BlasConnector::dot(n, X, incX, Y, incY, device_type); + return BlasConnector::dot(n, X, incX, Y, incY, device_type); } double BlasConnector::dotu(const int n, const double*const X, const int incX, const double*const Y, const int incY, base_device::AbacusDevice_t device_type) { - return BlasConnector::dot(n, X, incX, Y, incY, device_type); + return BlasConnector::dot(n, X, incX, Y, incY, device_type); } std::complex BlasConnector::dotu(const int n, const std::complex*const X, const int incX, const std::complex*const Y, const int incY, base_device::AbacusDevice_t device_type) { - if (device_type == base_device::AbacusDevice_t::CpuDevice) { - const int incX2 = 2 * incX; - const int incY2 = 2 * incY; - const float*const x = reinterpret_cast(X); - const float*const y = reinterpret_cast(Y); - //Re(result)=Re(x)*Re(y)-Im(x)*Im(y) - //Im(result)=Re(x)*Im(y)+Im(x)*Re(y) - return std::complex( - BlasConnector::dot(n, x, incX2, y, incY2, device_type) - dot(n, x+1, incX2, y+1, incY2, device_type), - BlasConnector::dot(n, x, incX2, y+1, incY2, device_type) + dot(n, x+1, incX2, y, incY2, device_type)); - } - else { - throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); - } + if (device_type == base_device::AbacusDevice_t::CpuDevice) { + const int incX2 = 2 * incX; + const int incY2 = 2 * incY; + const float*const x = reinterpret_cast(X); + const float*const y = reinterpret_cast(Y); + //Re(result)=Re(x)*Re(y)-Im(x)*Im(y) + //Im(result)=Re(x)*Im(y)+Im(x)*Re(y) + return std::complex( + BlasConnector::dot(n, x, incX2, y, incY2, device_type) - dot(n, x+1, incX2, y+1, incY2, device_type), + BlasConnector::dot(n, x, incX2, y+1, incY2, device_type) + dot(n, x+1, incX2, y, incY2, device_type)); + } + else { + throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); + } } std::complex BlasConnector::dotu(const int n, const std::complex*const X, const int incX, const std::complex*const Y, const int incY, base_device::AbacusDevice_t device_type) { - if (device_type == base_device::AbacusDevice_t::CpuDevice) { - const int incX2 = 2 * incX; - const int incY2 = 2 * incY; - const double*const x = reinterpret_cast(X); - const double*const y = reinterpret_cast(Y); - //Re(result)=Re(x)*Re(y)-Im(x)*Im(y) - //Im(result)=Re(x)*Im(y)+Im(x)*Re(y) - return std::complex( - BlasConnector::dot(n, x, incX2, y, incY2, device_type) - dot(n, x+1, incX2, y+1, incY2, device_type), - BlasConnector::dot(n, x, incX2, y+1, incY2, device_type) + dot(n, x+1, incX2, y, incY2, device_type)); - } - else { - throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); - } + if (device_type == base_device::AbacusDevice_t::CpuDevice) { + const int incX2 = 2 * incX; + const int incY2 = 2 * incY; + const double*const x = reinterpret_cast(X); + const double*const y = reinterpret_cast(Y); + //Re(result)=Re(x)*Re(y)-Im(x)*Im(y) + //Im(result)=Re(x)*Im(y)+Im(x)*Re(y) + return std::complex( + BlasConnector::dot(n, x, incX2, y, incY2, device_type) - dot(n, x+1, incX2, y+1, incY2, device_type), + BlasConnector::dot(n, x, incX2, y+1, incY2, device_type) + dot(n, x+1, incX2, y, incY2, device_type)); + } + else { + throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); + } } // d = x.conj() * Vy float BlasConnector::dotc(const int n, const float*const X, const int incX, const float*const Y, const int incY, base_device::AbacusDevice_t device_type) { - return BlasConnector::dot(n, X, incX, Y, incY, device_type); + return BlasConnector::dot(n, X, incX, Y, incY, device_type); } double BlasConnector::dotc(const int n, const double*const X, const int incX, const double*const Y, const int incY, base_device::AbacusDevice_t device_type) { - return BlasConnector::dot(n, X, incX, Y, incY, device_type); + return BlasConnector::dot(n, X, incX, Y, incY, device_type); } std::complex BlasConnector::dotc(const int n, const std::complex*const X, const int incX, const std::complex*const Y, const int incY, base_device::AbacusDevice_t device_type) { - if (device_type == base_device::AbacusDevice_t::CpuDevice) { - const int incX2 = 2 * incX; - const int incY2 = 2 * incY; - const float*const x = reinterpret_cast(X); - const float*const y = reinterpret_cast(Y); - // Re(result)=Re(X)*Re(Y)+Im(X)*Im(Y) - // Im(result)=Re(X)*Im(Y)-Im(X)*Re(Y) - return std::complex( - BlasConnector::dot(n, x, incX2, y, incY2, device_type) + dot(n, x+1, incX2, y+1, incY2, device_type), - BlasConnector::dot(n, x, incX2, y+1, incY2, device_type) - dot(n, x+1, incX2, y, incY2, device_type)); - } - else { - throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); - } + if (device_type == base_device::AbacusDevice_t::CpuDevice) { + const int incX2 = 2 * incX; + const int incY2 = 2 * incY; + const float*const x = reinterpret_cast(X); + const float*const y = reinterpret_cast(Y); + // Re(result)=Re(X)*Re(Y)+Im(X)*Im(Y) + // Im(result)=Re(X)*Im(Y)-Im(X)*Re(Y) + return std::complex( + BlasConnector::dot(n, x, incX2, y, incY2, device_type) + dot(n, x+1, incX2, y+1, incY2, device_type), + BlasConnector::dot(n, x, incX2, y+1, incY2, device_type) - dot(n, x+1, incX2, y, incY2, device_type)); + } + else { + throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); + } } std::complex BlasConnector::dotc(const int n, const std::complex*const X, const int incX, const std::complex*const Y, const int incY, base_device::AbacusDevice_t device_type) { - if (device_type == base_device::AbacusDevice_t::CpuDevice) { - const int incX2 = 2 * incX; - const int incY2 = 2 * incY; - const double*const x = reinterpret_cast(X); - const double*const y = reinterpret_cast(Y); - // Re(result)=Re(X)*Re(Y)+Im(X)*Im(Y) - // Im(result)=Re(X)*Im(Y)-Im(X)*Re(Y) - return std::complex( - BlasConnector::dot(n, x, incX2, y, incY2, device_type) + dot(n, x+1, incX2, y+1, incY2, device_type), - BlasConnector::dot(n, x, incX2, y+1, incY2, device_type) - dot(n, x+1, incX2, y, incY2, device_type)); - } - else { - throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); - } + if (device_type == base_device::AbacusDevice_t::CpuDevice) { + const int incX2 = 2 * incX; + const int incY2 = 2 * incY; + const double*const x = reinterpret_cast(X); + const double*const y = reinterpret_cast(Y); + // Re(result)=Re(X)*Re(Y)+Im(X)*Im(Y) + // Im(result)=Re(X)*Im(Y)-Im(X)*Re(Y) + return std::complex( + BlasConnector::dot(n, x, incX2, y, incY2, device_type) + dot(n, x+1, incX2, y+1, incY2, device_type), + BlasConnector::dot(n, x, incX2, y+1, incY2, device_type) - dot(n, x+1, incX2, y, incY2, device_type)); + } + else { + throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); + } } // out = ||x||_2 float BlasConnector::nrm2( const int n, const float *X, const int incX, base_device::AbacusDevice_t device_type ) { - if (device_type == base_device::AbacusDevice_t::CpuDevice) { - return snrm2_( &n, X, &incX ); - } + if (device_type == base_device::AbacusDevice_t::CpuDevice) { + return snrm2_( &n, X, &incX ); + } #ifdef __CUDA - else if (device_type == base_device::AbacusDevice_t::GpuDevice){ - float result = 0.0; - CHECK_CUBLAS(cublasSnrm2(BlasUtils::cublas_handle, n, X, incX, &result)); - return result; - } + else if (device_type == base_device::AbacusDevice_t::GpuDevice){ + float result = 0.0; + CHECK_CUBLAS(cublasSnrm2(BlasUtils::cublas_handle, n, X, incX, &result)); + return result; + } #endif - else { - throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); - } + else { + throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); + } } double BlasConnector::nrm2( const int n, const double *X, const int incX, base_device::AbacusDevice_t device_type ) { - if (device_type == base_device::AbacusDevice_t::CpuDevice) { - return dnrm2_( &n, X, &incX ); - } + if (device_type == base_device::AbacusDevice_t::CpuDevice) { + return dnrm2_( &n, X, &incX ); + } #ifdef __CUDA - else if (device_type == base_device::AbacusDevice_t::GpuDevice){ - double result = 0.0; - CHECK_CUBLAS(cublasDnrm2(BlasUtils::cublas_handle, n, X, incX, &result)); - return result; - } + else if (device_type == base_device::AbacusDevice_t::GpuDevice){ + double result = 0.0; + CHECK_CUBLAS(cublasDnrm2(BlasUtils::cublas_handle, n, X, incX, &result)); + return result; + } #endif - else { - throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); - } + else { + throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); + } } double BlasConnector::nrm2( const int n, const std::complex *X, const int incX, base_device::AbacusDevice_t device_type ) { - if (device_type == base_device::AbacusDevice_t::CpuDevice) { - return dznrm2_( &n, X, &incX ); - } + if (device_type == base_device::AbacusDevice_t::CpuDevice) { + return dznrm2_( &n, X, &incX ); + } #ifdef __CUDA - else if (device_type == base_device::AbacusDevice_t::GpuDevice){ - double result = 0.0; - CHECK_CUBLAS(cublasDznrm2(BlasUtils::cublas_handle, n, (double2*)X, incX, &result)); - return result; - } + else if (device_type == base_device::AbacusDevice_t::GpuDevice){ + double result = 0.0; + CHECK_CUBLAS(cublasDznrm2(BlasUtils::cublas_handle, n, (double2*)X, incX, &result)); + return result; + } #endif - else { - throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); - } + else { + throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); + } } // copies a into b void BlasConnector::copy(const int n, const float *a, const int incx, float *b, const int incy, base_device::AbacusDevice_t device_type) { - if (device_type == base_device::AbacusDevice_t::CpuDevice) { - scopy_(&n, a, &incx, b, &incy); - } - else { - throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); - } + if (device_type == base_device::AbacusDevice_t::CpuDevice) { + scopy_(&n, a, &incx, b, &incy); + } + else { + throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); + } } void BlasConnector::copy(const int n, const double *a, const int incx, double *b, const int incy, base_device::AbacusDevice_t device_type) { - if (device_type == base_device::AbacusDevice_t::CpuDevice) { - dcopy_(&n, a, &incx, b, &incy); - } - else { - throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); - } + if (device_type == base_device::AbacusDevice_t::CpuDevice) { + dcopy_(&n, a, &incx, b, &incy); + } + else { + throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); + } } void BlasConnector::copy(const int n, const std::complex *a, const int incx, std::complex *b, const int incy, base_device::AbacusDevice_t device_type) { - if (device_type == base_device::AbacusDevice_t::CpuDevice) { - ccopy_(&n, a, &incx, b, &incy); - } - else { - throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); - } + if (device_type == base_device::AbacusDevice_t::CpuDevice) { + ccopy_(&n, a, &incx, b, &incy); + } + else { + throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); + } } void BlasConnector::copy(const int n, const std::complex *a, const int incx, std::complex *b, const int incy, base_device::AbacusDevice_t device_type) { - if (device_type == base_device::AbacusDevice_t::CpuDevice) { - zcopy_(&n, a, &incx, b, &incy); - } - else { - throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); - } + if (device_type == base_device::AbacusDevice_t::CpuDevice) { + zcopy_(&n, a, &incx, b, &incy); + } + else { + throw std::invalid_argument("device_type = " + std::to_string(device_type) + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); + } } diff --git a/source/source_base/module_external/lapack_connector.h b/source/source_base/module_external/lapack_connector.h index c71e926eb84..1b895bac904 100644 --- a/source/source_base/module_external/lapack_connector.h +++ b/source/source_base/module_external/lapack_connector.h @@ -160,6 +160,20 @@ void zhegvx_(const int* itype, const char* jobz, const char* range, const char* double* rwork, int* iwork, int* ifail, int* info); +#ifdef __DSP +void cpack_zhegvx_driver(const int* itype, const char* jobz, const char* range, const char* uplo, + const int* n, + std::complex* a, const int* lda, + std::complex* b, const int* ldb, + const double* vl, const double* vu, + const int* il, const int* iu, + const double* abstol, + int* m, double* w, std::complex* z, const int* ldz, + std::complex* work, const int* lwork, + double* rwork, int* iwork, int* ifail, + int* info); +#endif + // === Generalized Hermitian: all eigenvalues (simple driver) === void dsygv_(const int* itype, const char* jobz, const char* uplo, const int* n, @@ -439,29 +453,29 @@ namespace LapackConnector } } - // Peize Lin add 2015-12-27 - static inline - char change_uplo(const char &uplo) - { - switch(uplo) - { - case 'U': return 'L'; - case 'L': return 'U'; - default: throw std::invalid_argument("uplo must be 'U' or 'L'"); - } - } - - // Peize Lin add 2019-04-14 - static inline - char change_trans_NC(const char &trans) - { - switch(trans) - { - case 'N': return 'C'; - case 'C': return 'N'; - default: throw std::invalid_argument("trans must be 'N' or 'C'"); - } - } + // Peize Lin add 2015-12-27 + static inline + char change_uplo(const char &uplo) + { + switch(uplo) + { + case 'U': return 'L'; + case 'L': return 'U'; + default: throw std::invalid_argument("uplo must be 'U' or 'L'"); + } + } + + // Peize Lin add 2019-04-14 + static inline + char change_trans_NC(const char &trans) + { + switch(trans) + { + case 'N': return 'C'; + case 'C': return 'N'; + default: throw std::invalid_argument("trans must be 'N' or 'C'"); + } + } // wrap function of fortran lapack routine zheev. static inline @@ -504,95 +518,95 @@ namespace LapackConnector return; } - // Peize Lin add 2016-07-09 - static inline - void potrf( const char &uplo, const int &n, float*const A, const int &lda, int &info ) - { - const char uplo_changed = change_uplo(uplo); - spotrf_( &uplo_changed, &n, A, &lda, &info ); - } - static inline - void potrf( const char &uplo, const int &n, double*const A, const int &lda, int &info ) - { - const char uplo_changed = change_uplo(uplo); - dpotrf_( &uplo_changed, &n, A, &lda, &info ); - } - static inline - void potrf( const char &uplo, const int &n, std::complex*const A, const int &lda, int &info ) - { - const char uplo_changed = change_uplo(uplo); - cpotrf_( &uplo_changed, &n, A, &lda, &info ); - } - static inline - void potrf( const char &uplo, const int &n, std::complex*const A, const int &lda, int &info ) - { - const char uplo_changed = change_uplo(uplo); - zpotrf_( &uplo_changed, &n, A, &lda, &info ); - } - - - // Peize Lin add 2016-07-09 - static inline - void potri( const char &uplo, const int &n, float*const A, const int &lda, int &info ) - { - const char uplo_changed = change_uplo(uplo); - spotri_( &uplo_changed, &n, A, &lda, &info); - } - static inline - void potri( const char &uplo, const int &n, double*const A, const int &lda, int &info ) - { - const char uplo_changed = change_uplo(uplo); - dpotri_( &uplo_changed, &n, A, &lda, &info); - } - static inline - void potri( const char &uplo, const int &n, std::complex*const A, const int &lda, int &info ) - { - const char uplo_changed = change_uplo(uplo); - cpotri_( &uplo_changed, &n, A, &lda, &info); - } - static inline - void potri( const char &uplo, const int &n, std::complex*const A, const int &lda, int &info ) - { - const char uplo_changed = change_uplo(uplo); - zpotri_( &uplo_changed, &n, A, &lda, &info); - } - - // Peize Lin add 2016-07-09 - static inline - void potrf( const char &uplo, const int &n, ModuleBase::matrix &A, const int &lda, int &info ) - { - potrf( uplo, n, A.c, lda, info ); - } - static inline - void potrf( const char &uplo, const int &n, ModuleBase::ComplexMatrix &A, const int &lda, int &info ) - { - potrf( uplo, n, A.c, lda, info ); - } - - // Peize Lin add 2016-07-09 - static inline - void potri( const char &uplo, const int &n, ModuleBase::matrix &A, const int &lda, int &info ) - { - potri( uplo, n, A.c, lda, info); - } - static inline - void potri( const char &uplo, const int &n, ModuleBase::ComplexMatrix &A, const int &lda, int &info ) - { - potri( uplo, n, A.c, lda, info); - } - - // Peize Lin add 2019-04-14 - // if trans=='N': C = a * A * A.H + b * C - // if trans=='C': C = a * A.H * A + b * C - static inline + // Peize Lin add 2016-07-09 + static inline + void potrf( const char &uplo, const int &n, float*const A, const int &lda, int &info ) + { + const char uplo_changed = change_uplo(uplo); + spotrf_( &uplo_changed, &n, A, &lda, &info ); + } + static inline + void potrf( const char &uplo, const int &n, double*const A, const int &lda, int &info ) + { + const char uplo_changed = change_uplo(uplo); + dpotrf_( &uplo_changed, &n, A, &lda, &info ); + } + static inline + void potrf( const char &uplo, const int &n, std::complex*const A, const int &lda, int &info ) + { + const char uplo_changed = change_uplo(uplo); + cpotrf_( &uplo_changed, &n, A, &lda, &info ); + } + static inline + void potrf( const char &uplo, const int &n, std::complex*const A, const int &lda, int &info ) + { + const char uplo_changed = change_uplo(uplo); + zpotrf_( &uplo_changed, &n, A, &lda, &info ); + } + + + // Peize Lin add 2016-07-09 + static inline + void potri( const char &uplo, const int &n, float*const A, const int &lda, int &info ) + { + const char uplo_changed = change_uplo(uplo); + spotri_( &uplo_changed, &n, A, &lda, &info); + } + static inline + void potri( const char &uplo, const int &n, double*const A, const int &lda, int &info ) + { + const char uplo_changed = change_uplo(uplo); + dpotri_( &uplo_changed, &n, A, &lda, &info); + } + static inline + void potri( const char &uplo, const int &n, std::complex*const A, const int &lda, int &info ) + { + const char uplo_changed = change_uplo(uplo); + cpotri_( &uplo_changed, &n, A, &lda, &info); + } + static inline + void potri( const char &uplo, const int &n, std::complex*const A, const int &lda, int &info ) + { + const char uplo_changed = change_uplo(uplo); + zpotri_( &uplo_changed, &n, A, &lda, &info); + } + + // Peize Lin add 2016-07-09 + static inline + void potrf( const char &uplo, const int &n, ModuleBase::matrix &A, const int &lda, int &info ) + { + potrf( uplo, n, A.c, lda, info ); + } + static inline + void potrf( const char &uplo, const int &n, ModuleBase::ComplexMatrix &A, const int &lda, int &info ) + { + potrf( uplo, n, A.c, lda, info ); + } + + // Peize Lin add 2016-07-09 + static inline + void potri( const char &uplo, const int &n, ModuleBase::matrix &A, const int &lda, int &info ) + { + potri( uplo, n, A.c, lda, info); + } + static inline + void potri( const char &uplo, const int &n, ModuleBase::ComplexMatrix &A, const int &lda, int &info ) + { + potri( uplo, n, A.c, lda, info); + } + + // Peize Lin add 2019-04-14 + // if trans=='N': C = a * A * A.H + b * C + // if trans=='C': C = a * A.H * A + b * C + static inline void herk(const char uplo, const char trans, const int n, const int k, - const double alpha, const std::complex *A, const int lda, - const double beta, std::complex *C, const int ldc) - { - const char uplo_changed = change_uplo(uplo); - const char trans_changed = change_trans_NC(trans); - zherk_(&uplo_changed, &trans_changed, &n, &k, &alpha, A, &lda, &beta, C, &ldc); - } + const double alpha, const std::complex *A, const int lda, + const double beta, std::complex *C, const int ldc) + { + const char uplo_changed = change_uplo(uplo); + const char trans_changed = change_trans_NC(trans); + zherk_(&uplo_changed, &trans_changed, &n, &k, &alpha, A, &lda, &beta, C, &ldc); + } static inline void herk(const char uplo, const char trans, const int n, const int k, const float alpha, const std::complex* A, const int lda, diff --git a/source/source_base/module_external/scalapack_connector.h b/source/source_base/module_external/scalapack_connector.h index a6e2ef5ea44..35673385b67 100644 --- a/source/source_base/module_external/scalapack_connector.h +++ b/source/source_base/module_external/scalapack_connector.h @@ -24,243 +24,243 @@ extern "C" { - int numroc_( const int *n, const int *nb, const int *iproc, const int *srcproc, const int *nprocs ); - void descinit_( - int *desc, - const int *m, const int *n, const int *mb, const int *nb, const int *irsrc, const int *icsrc, - const int *ictxt, const int *lld, int *info); - - void pddot_(int* n, double* dot, double* x, int* ix, int* jx, int* descx, int* incx, - double* y, int* iy, int* jy, int* descy, int* incy); - void pzdotc_(int* n, std::complex* dot, std::complex* x, int* ix, int* jx, int* descx, int* incx, - std::complex* y, int* iy, int* jy, int* descy, int* incy); - - void pdpotrf_(char *uplo, int *n, double *a, int *ia, int *ja, int *desca, int *info); + int numroc_( const int *n, const int *nb, const int *iproc, const int *srcproc, const int *nprocs ); + void descinit_( + int *desc, + const int *m, const int *n, const int *mb, const int *nb, const int *irsrc, const int *icsrc, + const int *ictxt, const int *lld, int *info); + + void pddot_(int* n, double* dot, double* x, int* ix, int* jx, int* descx, int* incx, + double* y, int* iy, int* jy, int* descy, int* incy); + void pzdotc_(int* n, std::complex* dot, std::complex* x, int* ix, int* jx, int* descx, int* incx, + std::complex* y, int* iy, int* jy, int* descy, int* incy); + + void pdpotrf_(char *uplo, int *n, double *a, int *ia, int *ja, int *desca, int *info); // void pzpotrf_(char *uplo, int *n, double _Complex *a, int *ia, int *ja, int *desca, int *info); - void pzpotrf_(char *uplo, int *n, std::complex *a, int *ia, int *ja, int *desca, int *info); + void pzpotrf_(char *uplo, int *n, std::complex *a, int *ia, int *ja, int *desca, int *info); void pdtran_(const int* m, const int* n, const double* alpha, const double* a, const int* ia, const int* ja, const int* desca, const double* beta, double* c, const int* ic, const int* jc, const int* descc); - void pztranu_(const int *m,const int*n, + void pztranu_(const int *m,const int*n, const std::complex* alpha, const std::complex* a, const int* ia, const int* ja, const int* desca, - const std::complex *beta , std::complex *c, const int *ic, const int *jc, const int *descc); - void pztranc_(const int *m, const int *n, - const std::complex *alpha, const std::complex *a, const int *ia, const int *ja, const int *desca, - const std::complex *beta, std::complex *c, const int *ic, const int *jc, const int *descc); - - double pdlange_(const char* norm, - const int* m, const int* n, - const double* a, const int* ia, const int* ja, const int* desca, - double* work); - - double pzlange_(const char* norm, - const int* m, const int* n, - const std::complex* a, const int* ia, const int* ja, const int* desca, - double* work); - - void pzgemv_( - const char *transa, - const int *M, const int *N, - const double *alpha, - const std::complex *A, const int *IA, const int *JA, const int *DESCA, - const std::complex *B, const int *IB, const int *JB, const int *DESCB, const int *K, - const double *beta, std::complex *C, const int *IC, const int *JC, const int *DESCC,const int *L); - void pdgemv_( - const char *transa, - const int *M, const int *N, - const double *alpha, - const double *A, const int *IA, const int *JA, const int *DESCA, - const double *B, const int *IB, const int *JB, const int *DESCB, const int *K, - const double *beta, double *C, const int *IC, const int *JC, const int *DESCC,const int *L); - // C = a * A.? * B.? + b * C - void pdgemm_( - const char *transa, const char *transb, - const int *M, const int *N, const int *K, - const double *alpha, - const double *A, const int *IA, const int *JA, const int *DESCA, - const double *B, const int *IB, const int *JB, const int *DESCB, - const double *beta, - double *C, const int *IC, const int *JC, const int *DESCC); - void pzgemm_( - const char *transa, const char *transb, - const int *M, const int *N, const int *K, - const std::complex *alpha, - const std::complex *A, const int *IA, const int *JA, const int *DESCA, - const std::complex *B, const int *IB, const int *JB, const int *DESCB, - const std::complex *beta, - std::complex *C, const int *IC, const int *JC, const int *DESCC); - void pdsymm_(char *side , char *uplo , int *m , int *n , - double *alpha , double *a , int *ia , int *ja , int *desca , - double *b , int *ib , int *jb , int *descb , - double *beta , double *c , int *ic , int *jc , int *descc ); - void pdtrmm_(char *side , char *uplo , char *transa , char *diag , int *m , int *n , - double *alpha , double *a , int *ia , int *ja , int *desca , - double *b , int *ib , int *jb , int *descb ); + const std::complex *beta , std::complex *c, const int *ic, const int *jc, const int *descc); + void pztranc_(const int *m, const int *n, + const std::complex *alpha, const std::complex *a, const int *ia, const int *ja, const int *desca, + const std::complex *beta, std::complex *c, const int *ic, const int *jc, const int *descc); + + double pdlange_(const char* norm, + const int* m, const int* n, + const double* a, const int* ia, const int* ja, const int* desca, + double* work); + + double pzlange_(const char* norm, + const int* m, const int* n, + const std::complex* a, const int* ia, const int* ja, const int* desca, + double* work); + + void pzgemv_( + const char *transa, + const int *M, const int *N, + const double *alpha, + const std::complex *A, const int *IA, const int *JA, const int *DESCA, + const std::complex *B, const int *IB, const int *JB, const int *DESCB, const int *K, + const double *beta, std::complex *C, const int *IC, const int *JC, const int *DESCC,const int *L); + void pdgemv_( + const char *transa, + const int *M, const int *N, + const double *alpha, + const double *A, const int *IA, const int *JA, const int *DESCA, + const double *B, const int *IB, const int *JB, const int *DESCB, const int *K, + const double *beta, double *C, const int *IC, const int *JC, const int *DESCC,const int *L); + // C = a * A.? * B.? + b * C + void pdgemm_( + const char *transa, const char *transb, + const int *M, const int *N, const int *K, + const double *alpha, + const double *A, const int *IA, const int *JA, const int *DESCA, + const double *B, const int *IB, const int *JB, const int *DESCB, + const double *beta, + double *C, const int *IC, const int *JC, const int *DESCC); + void pzgemm_( + const char *transa, const char *transb, + const int *M, const int *N, const int *K, + const std::complex *alpha, + const std::complex *A, const int *IA, const int *JA, const int *DESCA, + const std::complex *B, const int *IB, const int *JB, const int *DESCB, + const std::complex *beta, + std::complex *C, const int *IC, const int *JC, const int *DESCC); + void pdsymm_(char *side , char *uplo , int *m , int *n , + double *alpha , double *a , int *ia , int *ja , int *desca , + double *b , int *ib , int *jb , int *descb , + double *beta , double *c , int *ic , int *jc , int *descc ); + void pdtrmm_(char *side , char *uplo , char *transa , char *diag , int *m , int *n , + double *alpha , double *a , int *ia , int *ja , int *desca , + double *b , int *ib , int *jb , int *descb ); // void pztrmm_(char *side , char *uplo , char *transa , char *diag , int *m , int *n , // double *alpha , double _Complex *a , int *ia , int *ja , int *desca , // double _Complex *b , int *ib , int *jb , int *descb ); - void pztrmm_(char *side , char *uplo , char *transa , char *diag , int *m , int *n , - std::complex *alpha , std::complex *a , int *ia , int *ja , int *desca , - std::complex *b , int *ib , int *jb , int *descb ); - void pzhemm_(char* side , char* uplo , int* m , int* n , - std::complex* alpha , std::complex* a , int* ia , int* ja , int* desca , - std::complex* b , int* ib , int* jb , int* descb , - std::complex* beta , std::complex* c , int* ic , int* jc , int* descc ); - void pzgetrf_( - const int *M, const int *N, - std::complex *A, const int *IA, const int *JA, const int *DESCA, - int *ipiv, int *info); - - void pzgesv_( - const int *n, const int *nrhs, - const std::complex *A, const int *ia, const int *ja, const int *desca, - int *ipiv, std::complex* B, const int* ib, const int* jb, const int*descb, const int *info - ); - - void pdsygvx_(const int* itype, const char* jobz, const char* range, const char* uplo, - const int* n, double* A, const int* ia, const int* ja, const int*desca, double* B, const int* ib, const int* jb, const int*descb, - const double* vl, const double* vu, const int* il, const int* iu, - const double* abstol, int* m, int* nz, double* w, const double*orfac, double* Z, const int* iz, const int* jz, const int*descz, - double* work, int* lwork, int*iwork, int*liwork, int* ifail, int*iclustr, double*gap, int* info); - - void pzhegvx_(const int* itype, const char* jobz, const char* range, const char* uplo, - const int* n, std::complex* A, const int* ia, const int* ja, const int*desca, std::complex* B, const int* ib, const int* jb, const int*descb, - const double* vl, const double* vu, const int* il, const int* iu, - const double* abstol, int* m, int* nz, double* w, const double*orfac, std::complex* Z, const int* iz, const int* jz, const int*descz, - std::complex* work, int* lwork, double* rwork, int* lrwork, int*iwork, int*liwork, int* ifail, int*iclustr, double*gap, int* info); - - void pssygvx_(const int* itype, const char* jobz, const char* range, const char* uplo, - const int* n, float* A, const int* ia, const int* ja, const int*desca, float* B, const int* ib, const int* jb, const int*descb, - const float* vl, const float* vu, const int* il, const int* iu, - const float* abstol, int* m, int* nz, float* w, const float*orfac, float* Z, const int* iz, const int* jz, const int*descz, - float* work, int* lwork, int*iwork, int*liwork, int* ifail, int*iclustr, float*gap, int* info); - - void pchegvx_(const int* itype, const char* jobz, const char* range, const char* uplo, - const int* n, std::complex* A, const int* ia, const int* ja, const int*desca, std::complex* B, const int* ib, const int* jb, const int*descb, - const float* vl, const float* vu, const int* il, const int* iu, - const float* abstol, int* m, int* nz, float* w, const float*orfac, std::complex* Z, const int* iz, const int* jz, const int*descz, - std::complex* work, int* lwork, float* rwork, int* lrwork, int*iwork, int*liwork, int* ifail, int*iclustr, float*gap, int* info); - - - void pzgetri_( - const int *n, - const std::complex *A, const int *ia, const int *ja, const int *desca, - int *ipiv, const std::complex *work, const int *lwork, const int *iwork, const int *liwork, const int *info); + void pztrmm_(char *side , char *uplo , char *transa , char *diag , int *m , int *n , + std::complex *alpha , std::complex *a , int *ia , int *ja , int *desca , + std::complex *b , int *ib , int *jb , int *descb ); + void pzhemm_(char* side , char* uplo , int* m , int* n , + std::complex* alpha , std::complex* a , int* ia , int* ja , int* desca , + std::complex* b , int* ib , int* jb , int* descb , + std::complex* beta , std::complex* c , int* ic , int* jc , int* descc ); + void pzgetrf_( + const int *M, const int *N, + std::complex *A, const int *IA, const int *JA, const int *DESCA, + int *ipiv, int *info); + + void pzgesv_( + const int *n, const int *nrhs, + const std::complex *A, const int *ia, const int *ja, const int *desca, + int *ipiv, std::complex* B, const int* ib, const int* jb, const int*descb, const int *info + ); + + void pdsygvx_(const int* itype, const char* jobz, const char* range, const char* uplo, + const int* n, double* A, const int* ia, const int* ja, const int*desca, double* B, const int* ib, const int* jb, const int*descb, + const double* vl, const double* vu, const int* il, const int* iu, + const double* abstol, int* m, int* nz, double* w, const double*orfac, double* Z, const int* iz, const int* jz, const int*descz, + double* work, int* lwork, int*iwork, int*liwork, int* ifail, int*iclustr, double*gap, int* info); + + void pzhegvx_(const int* itype, const char* jobz, const char* range, const char* uplo, + const int* n, std::complex* A, const int* ia, const int* ja, const int*desca, std::complex* B, const int* ib, const int* jb, const int*descb, + const double* vl, const double* vu, const int* il, const int* iu, + const double* abstol, int* m, int* nz, double* w, const double*orfac, std::complex* Z, const int* iz, const int* jz, const int*descz, + std::complex* work, int* lwork, double* rwork, int* lrwork, int*iwork, int*liwork, int* ifail, int*iclustr, double*gap, int* info); + + void pssygvx_(const int* itype, const char* jobz, const char* range, const char* uplo, + const int* n, float* A, const int* ia, const int* ja, const int*desca, float* B, const int* ib, const int* jb, const int*descb, + const float* vl, const float* vu, const int* il, const int* iu, + const float* abstol, int* m, int* nz, float* w, const float*orfac, float* Z, const int* iz, const int* jz, const int*descz, + float* work, int* lwork, int*iwork, int*liwork, int* ifail, int*iclustr, float*gap, int* info); + + void pchegvx_(const int* itype, const char* jobz, const char* range, const char* uplo, + const int* n, std::complex* A, const int* ia, const int* ja, const int*desca, std::complex* B, const int* ib, const int* jb, const int*descb, + const float* vl, const float* vu, const int* il, const int* iu, + const float* abstol, int* m, int* nz, float* w, const float*orfac, std::complex* Z, const int* iz, const int* jz, const int*descz, + std::complex* work, int* lwork, float* rwork, int* lrwork, int*iwork, int*liwork, int* ifail, int*iclustr, float*gap, int* info); + + + void pzgetri_( + const int *n, + const std::complex *A, const int *ia, const int *ja, const int *desca, + int *ipiv, const std::complex *work, const int *lwork, const int *iwork, const int *liwork, const int *info); void pzgeadd_( - const char *transa, - const int *m, const int *n, - const std::complex *alpha, - const std::complex *a, const int *ia, const int *ja, const int *desca, - const std::complex *beta, - const std::complex *c, const int *ic, const int *jc, const int *descc); + const char *transa, + const int *m, const int *n, + const std::complex *alpha, + const std::complex *a, const int *ia, const int *ja, const int *desca, + const std::complex *beta, + const std::complex *c, const int *ic, const int *jc, const int *descc); void pdgemr2d_(const int *M, const int *N, - double *A, const int *IA, const int *JA, const int *DESCA, - double *B, const int *IB, const int *JB, const int *DESCB, - const int *ICTXT); - + double *A, const int *IA, const int *JA, const int *DESCA, + double *B, const int *IB, const int *JB, const int *DESCB, + const int *ICTXT); + void pzgemr2d_(const int *M, const int *N, - std::complex *A, const int *IA, const int *JA, const int *DESCA, - std::complex *B, const int *IB, const int *JB, const int *DESCB, - const int *ICTXT); - - // Scalapack wrappers to copy 2D blocks of data - // more info: - // https://netlib.org/scalapack/explore-html/da/db5/pigemr_8c.html - // https://netlib.org/scalapack/explore-html/dd/dcd/pdgemr_8c.html - // https://netlib.org/scalapack/explore-html/d5/dd4/pzgemr_8c.html - // https://netlib.org/scalapack/explore-html/d5/deb/psgemr_8c.html - // https://netlib.org/scalapack/explore-html/d4/dad/pcgemr_8c.html - void Cpigemr2d (int m, int n, int *ptrmyblock, int ia, int ja, int *ma, int *ptrmynewblock, int ib, int jb, int *mb, int globcontext); - void Cpdgemr2d (int m, int n, double *ptrmyblock, int ia, int ja, int *ma, double *ptrmynewblock, int ib, int jb, int *mb, int globcontext); - void Cpzgemr2d (int m, int n, std::complex *ptrmyblock, int ia, int ja, int *ma, std::complex *ptrmynewblock, int ib, int jb, int *mb, int globcontext); - void Cpsgemr2d (int m, int n, float *ptrmyblock, int ia, int ja, int *ma, float *ptrmynewblock, int ib, int jb, int *mb, int globcontext); - void Cpcgemr2d (int m, int n, std::complex *ptrmyblock, int ia, int ja, int *ma, std::complex *ptrmynewblock, int ib, int jb, int *mb, int globcontext); + std::complex *A, const int *IA, const int *JA, const int *DESCA, + std::complex *B, const int *IB, const int *JB, const int *DESCB, + const int *ICTXT); + + // Scalapack wrappers to copy 2D blocks of data + // more info: + // https://netlib.org/scalapack/explore-html/da/db5/pigemr_8c.html + // https://netlib.org/scalapack/explore-html/dd/dcd/pdgemr_8c.html + // https://netlib.org/scalapack/explore-html/d5/dd4/pzgemr_8c.html + // https://netlib.org/scalapack/explore-html/d5/deb/psgemr_8c.html + // https://netlib.org/scalapack/explore-html/d4/dad/pcgemr_8c.html + void Cpigemr2d (int m, int n, int *ptrmyblock, int ia, int ja, int *ma, int *ptrmynewblock, int ib, int jb, int *mb, int globcontext); + void Cpdgemr2d (int m, int n, double *ptrmyblock, int ia, int ja, int *ma, double *ptrmynewblock, int ib, int jb, int *mb, int globcontext); + void Cpzgemr2d (int m, int n, std::complex *ptrmyblock, int ia, int ja, int *ma, std::complex *ptrmynewblock, int ib, int jb, int *mb, int globcontext); + void Cpsgemr2d (int m, int n, float *ptrmyblock, int ia, int ja, int *ma, float *ptrmynewblock, int ib, int jb, int *mb, int globcontext); + void Cpcgemr2d (int m, int n, std::complex *ptrmyblock, int ia, int ja, int *ma, std::complex *ptrmynewblock, int ib, int jb, int *mb, int globcontext); } - template - struct block2d_data_type - { - constexpr static bool value = std::is_same::value || std::is_same>::value || std::is_same::value || std::is_same>::value || std::is_same::value; - }; - - - /** - * Copies a 2D block of data from matrix A to matrix B using the Scalapack library. - * This function supports different data types: double, std::complex, float, std::complex, and int. - * - * @tparam T The data type of the matrices A and B. - * @param M The number of rows of matrix A. - * @param N The number of columns of matrix A. - * @param A Pointer to the source matrix A. - * @param IA The starting row index of the block in matrix A. - * @param JA The starting column index of the block in matrix A. - * @param DESCA Descriptor array for matrix A. - * @param B Pointer to the destination matrix B. - * @param IB The starting row index of the block in matrix B. - * @param JB The starting column index of the block in matrix B. - * @param DESCB Descriptor array for matrix B. - * @param ICTXT The context identifier. - */ - template - typename std::enable_if::value,void>::type Cpxgemr2d(int M, int N, T *A, int IA, int JA, int *DESCA, T *B, int IB, int JB, int *DESCB, int ICTXT) - { - if (std::is_same::value) Cpdgemr2d(M, N, reinterpret_cast(A),IA, JA, DESCA,reinterpret_cast(B),IB,JB, DESCB,ICTXT); - if (std::is_same>::value) Cpzgemr2d(M, N, reinterpret_cast*>(A),IA, JA, DESCA,reinterpret_cast*>(B),IB,JB, DESCB,ICTXT); - if (std::is_same::value) Cpsgemr2d(M, N, reinterpret_cast(A),IA, JA, DESCA,reinterpret_cast(B),IB,JB, DESCB,ICTXT); - if (std::is_same>::value) Cpcgemr2d(M, N, reinterpret_cast*>(A),IA, JA, DESCA,reinterpret_cast*>(B),IB,JB, DESCB,ICTXT); - if (std::is_same::value) Cpigemr2d(M, N, reinterpret_cast(A),IA, JA, DESCA,reinterpret_cast(B),IB,JB, DESCB,ICTXT); - }; - + template + struct block2d_data_type + { + constexpr static bool value = std::is_same::value || std::is_same>::value || std::is_same::value || std::is_same>::value || std::is_same::value; + }; + + + /** + * Copies a 2D block of data from matrix A to matrix B using the Scalapack library. + * This function supports different data types: double, std::complex, float, std::complex, and int. + * + * @tparam T The data type of the matrices A and B. + * @param M The number of rows of matrix A. + * @param N The number of columns of matrix A. + * @param A Pointer to the source matrix A. + * @param IA The starting row index of the block in matrix A. + * @param JA The starting column index of the block in matrix A. + * @param DESCA Descriptor array for matrix A. + * @param B Pointer to the destination matrix B. + * @param IB The starting row index of the block in matrix B. + * @param JB The starting column index of the block in matrix B. + * @param DESCB Descriptor array for matrix B. + * @param ICTXT The context identifier. + */ + template + typename std::enable_if::value,void>::type Cpxgemr2d(int M, int N, T *A, int IA, int JA, int *DESCA, T *B, int IB, int JB, int *DESCB, int ICTXT) + { + if (std::is_same::value) Cpdgemr2d(M, N, reinterpret_cast(A),IA, JA, DESCA,reinterpret_cast(B),IB,JB, DESCB,ICTXT); + if (std::is_same>::value) Cpzgemr2d(M, N, reinterpret_cast*>(A),IA, JA, DESCA,reinterpret_cast*>(B),IB,JB, DESCB,ICTXT); + if (std::is_same::value) Cpsgemr2d(M, N, reinterpret_cast(A),IA, JA, DESCA,reinterpret_cast(B),IB,JB, DESCB,ICTXT); + if (std::is_same>::value) Cpcgemr2d(M, N, reinterpret_cast*>(A),IA, JA, DESCA,reinterpret_cast*>(B),IB,JB, DESCB,ICTXT); + if (std::is_same::value) Cpigemr2d(M, N, reinterpret_cast(A),IA, JA, DESCA,reinterpret_cast(B),IB,JB, DESCB,ICTXT); + }; + class ScalapackConnector { public: - static inline - void geadd( - const char transa, - const int m, const int n, - const std::complex alpha, - const std::complex *a, const int ia, const int ja, const int *desca, - const std::complex beta, - const std::complex *c, const int ic, const int jc, const int *descc) - { - pzgeadd_(&transa, &m, &n, &alpha, a, &ia, &ja, desca, &beta, c, &ic, &jc, descc); - } - - static inline - void dot(int n, - double& dot, - double* a, - int ia, - int ja, - int inca, - double* b, - int ib, - int jb, - int incb, - int* desc) - { - pddot_(&n, &dot, a, &ia, &ja, desc, &inca, b, &ib, &jb, desc, &incb); - } - - static inline - void dot(int n, - std::complex& dotc, - std::complex* a, - int ia, - int ja, - int inca, - std::complex* b, - int ib, - int jb, - int incb, - int* desc) - { - pzdotc_(&n, &dotc, a, &ia, &ja, desc, &inca, b, &ib, &jb, desc, &incb); - } + static inline + void geadd( + const char transa, + const int m, const int n, + const std::complex alpha, + const std::complex *a, const int ia, const int ja, const int *desca, + const std::complex beta, + const std::complex *c, const int ic, const int jc, const int *descc) + { + pzgeadd_(&transa, &m, &n, &alpha, a, &ia, &ja, desca, &beta, c, &ic, &jc, descc); + } + + static inline + void dot(int n, + double& dot, + double* a, + int ia, + int ja, + int inca, + double* b, + int ib, + int jb, + int incb, + int* desc) + { + pddot_(&n, &dot, a, &ia, &ja, desc, &inca, b, &ib, &jb, desc, &incb); + } + + static inline + void dot(int n, + std::complex& dotc, + std::complex* a, + int ia, + int ja, + int inca, + std::complex* b, + int ib, + int jb, + int incb, + int* desc) + { + pzdotc_(&n, &dotc, a, &ia, &ja, desc, &inca, b, &ib, &jb, desc, &incb); + } static inline void gemm( @@ -277,157 +277,157 @@ class ScalapackConnector } static inline - void gemm( - const char transa, const char transb, - const int M, const int N, const int K, - const std::complex alpha, - const std::complex *A, const int IA, const int JA, const int *DESCA, - const std::complex *B, const int IB, const int JB, const int *DESCB, - const std::complex beta, - std::complex *C, const int IC, const int JC, const int *DESCC) - { - pzgemm_(&transa, &transb, &M, &N, &K, &alpha, A, &IA, &JA, DESCA, - B, &IB, &JB, DESCB, &beta, C, &IC, &JC, DESCC); - } - - static inline - void gemm(char transa, char transb, int M, int N, int K, - double alpha, - double* A, - double* B, - double beta, - double* C, - int* DESC) - { - int isrc = 1; - pdgemm_(&transa, - &transb, - &M, - &N, - &K, - &alpha, - A, - &isrc, - &isrc, - DESC, - B, - &isrc, - &isrc, - DESC, - &beta, - C, - &isrc, - &isrc, - DESC); - } + void gemm( + const char transa, const char transb, + const int M, const int N, const int K, + const std::complex alpha, + const std::complex *A, const int IA, const int JA, const int *DESCA, + const std::complex *B, const int IB, const int JB, const int *DESCB, + const std::complex beta, + std::complex *C, const int IC, const int JC, const int *DESCC) + { + pzgemm_(&transa, &transb, &M, &N, &K, &alpha, A, &IA, &JA, DESCA, + B, &IB, &JB, DESCB, &beta, C, &IC, &JC, DESCC); + } + + static inline + void gemm(char transa, char transb, int M, int N, int K, + double alpha, + double* A, + double* B, + double beta, + double* C, + int* DESC) + { + int isrc = 1; + pdgemm_(&transa, + &transb, + &M, + &N, + &K, + &alpha, + A, + &isrc, + &isrc, + DESC, + B, + &isrc, + &isrc, + DESC, + &beta, + C, + &isrc, + &isrc, + DESC); + } + + static inline + void gemm(char transa, char transb, int M, int N, int K, + std::complex alpha, + std::complex* A, + std::complex* B, + std::complex beta, + std::complex* C, + int* DESC) + { + + int isrc = 1; + pzgemm_(&transa, + &transb, + &M, + &N, + &K, + &alpha, + A, + &isrc, + &isrc, + DESC, + B, + &isrc, + &isrc, + DESC, + &beta, + C, + &isrc, + &isrc, + DESC); + } + + static inline + void symm(char side, + char uplo, + int m, + int n, + double alpha, + double* a, + double* b, + double beta, + double* c, + int* desc) + { + int isrc = 1; + pdsymm_(&side, &uplo, &m, &n, &alpha, a, &isrc, &isrc, desc, b, &isrc, &isrc, desc, &beta, c, &isrc, &isrc, desc); + } + + static inline + void getrf( + const int M, const int N, + std::complex *A, const int IA, const int JA, const int *DESCA, + int *ipiv, int *info) //fix a bug: info is output and we must use int* + { + pzgetrf_(&M, &N, A, &IA, &JA, DESCA, ipiv, info); + } + + static inline + void getri( + const int n, + const std::complex *A, const int ia, const int ja, const int *desca, int *ipiv, + const std::complex *work, const int *lwork, const int *iwork, const int *liwork, int *info) + { + pzgetri_(&n, A, &ia, &ja, desca, ipiv, work, lwork, iwork, liwork, info); + } static inline - void gemm(char transa, char transb, int M, int N, int K, - std::complex alpha, - std::complex* A, - std::complex* B, - std::complex beta, - std::complex* C, - int* DESC) - { - - int isrc = 1; - pzgemm_(&transa, - &transb, - &M, - &N, - &K, - &alpha, - A, - &isrc, - &isrc, - DESC, - B, - &isrc, - &isrc, - DESC, - &beta, - C, - &isrc, - &isrc, - DESC); - } - - static inline - void symm(char side, - char uplo, - int m, - int n, - double alpha, - double* a, - double* b, - double beta, - double* c, - int* desc) - { - int isrc = 1; - pdsymm_(&side, &uplo, &m, &n, &alpha, a, &isrc, &isrc, desc, b, &isrc, &isrc, desc, &beta, c, &isrc, &isrc, desc); - } + void gesv( + const int n, const int nrhs, + const std::complex *A, const int ia, const int ja, const int *desca, + int *ipiv, std::complex* B, const int ib, const int jb, const int*descb, int *info) + { + pzgesv_(&n, &nrhs, A, &ia, &ja, desca, ipiv, B, &ib, &jb, descb, info); + } static inline - void getrf( - const int M, const int N, - std::complex *A, const int IA, const int JA, const int *DESCA, - int *ipiv, int *info) //fix a bug: info is output and we must use int* - { - pzgetrf_(&M, &N, A, &IA, &JA, DESCA, ipiv, info); - } - - static inline - void getri( - const int n, - const std::complex *A, const int ia, const int ja, const int *desca, int *ipiv, - const std::complex *work, const int *lwork, const int *iwork, const int *liwork, int *info) - { - pzgetri_(&n, A, &ia, &ja, desca, ipiv, work, lwork, iwork, liwork, info); - } - - static inline - void gesv( - const int n, const int nrhs, - const std::complex *A, const int ia, const int ja, const int *desca, - int *ipiv, std::complex* B, const int ib, const int jb, const int*descb, int *info) - { - pzgesv_(&n, &nrhs, A, &ia, &ja, desca, ipiv, B, &ib, &jb, descb, info); - } - - static inline - void tranu( - const int m, const int n, - const double alpha, double *a, const int ia, const int ja, const int *desca, - const double beta, double *c, const int ic, const int jc, const int *descc) - { - pdtran_(&m, &n, &alpha, a, &ia, &ja, desca, &beta, c, &ic, &jc, descc); - } - static inline - void tranu( - const int m, const int n, - const std::complex alpha, std::complex *a, const int ia, const int ja, const int *desca, - const std::complex beta, std::complex *c, const int ic, const int jc, const int *descc) - { - pztranu_(&m, &n, &alpha, a, &ia, &ja, desca, &beta, c, &ic, &jc, descc); - } - static inline - void tranc( - const int m, const int n, - const double alpha, double *a, const int ia, const int ja, const int *desca, - const double beta, double *c, const int ic, const int jc, const int *descc) - { - pdtran_(&m, &n, &alpha, a, &ia, &ja, desca, &beta, c, &ic, &jc, descc); - } - static inline - void tranc( - const int m, const int n, - const std::complex alpha, std::complex *a, const int ia, const int ja, const int *desca, - const std::complex beta, std::complex *c, const int ic, const int jc, const int *descc) - { - pztranc_(&m, &n, &alpha, a, &ia, &ja, desca, &beta, c, &ic, &jc, descc); - } + void tranu( + const int m, const int n, + const double alpha, double *a, const int ia, const int ja, const int *desca, + const double beta, double *c, const int ic, const int jc, const int *descc) + { + pdtran_(&m, &n, &alpha, a, &ia, &ja, desca, &beta, c, &ic, &jc, descc); + } + static inline + void tranu( + const int m, const int n, + const std::complex alpha, std::complex *a, const int ia, const int ja, const int *desca, + const std::complex beta, std::complex *c, const int ic, const int jc, const int *descc) + { + pztranu_(&m, &n, &alpha, a, &ia, &ja, desca, &beta, c, &ic, &jc, descc); + } + static inline + void tranc( + const int m, const int n, + const double alpha, double *a, const int ia, const int ja, const int *desca, + const double beta, double *c, const int ic, const int jc, const int *descc) + { + pdtran_(&m, &n, &alpha, a, &ia, &ja, desca, &beta, c, &ic, &jc, descc); + } + static inline + void tranc( + const int m, const int n, + const std::complex alpha, std::complex *a, const int ia, const int ja, const int *desca, + const std::complex beta, std::complex *c, const int ic, const int jc, const int *descc) + { + pztranc_(&m, &n, &alpha, a, &ia, &ja, desca, &beta, c, &ic, &jc, descc); + } static inline double lange( @@ -448,87 +448,87 @@ class ScalapackConnector return pzlange_(&norm, &m, &n, a, &ia, &ja, desca, work); } - static inline - int potrf(char uplo, int na, double* U, int* desc) - { - int isrc = 1; - int info; - pdpotrf_(&uplo, &na, U, &isrc, &isrc, desc, &info); - return info; - } - - static inline - int potrf(char uplo, int na, std::complex* U, int* desc) - { - int isrc = 1; - int info; - pzpotrf_(&uplo, &na, U, &isrc, &isrc, desc, &info); - return info; - } - - static inline - void trmm(char side, - char uplo, - char trans, - char diag, - int m, - int n, - double alpha, - double* a, - double* b, - int* desc) - { - int isrc = 1; - pdtrmm_(&side, &uplo, &trans, &diag, &m, &n, &alpha, a, &isrc, &isrc, desc, b, &isrc, &isrc, desc); - } - - static inline - void trmm(char side, - char uplo, - char trans, - char diag, - int m, - int n, - std::complex alpha, - std::complex* a, - std::complex* b, - int* desc) - { - int isrc = 1; - pztrmm_(&side, &uplo, &trans, &diag, &m, &n, &alpha, a, &isrc, &isrc, desc, b, &isrc, &isrc, desc); - } - - static inline - void hemm(char side, - char uplo, - int na, - std::complex alpha, - std::complex* a, - std::complex* b, - std::complex beta, - std::complex* c, - int* desc) - { - int isrc = 1; - pzhemm_(&side, - &uplo, - &na, - &na, - &alpha, - a, - &isrc, - &isrc, - desc, - b, - &isrc, - &isrc, - desc, - &beta, - c, - &isrc, - &isrc, - desc); - } + static inline + int potrf(char uplo, int na, double* U, int* desc) + { + int isrc = 1; + int info; + pdpotrf_(&uplo, &na, U, &isrc, &isrc, desc, &info); + return info; + } + + static inline + int potrf(char uplo, int na, std::complex* U, int* desc) + { + int isrc = 1; + int info; + pzpotrf_(&uplo, &na, U, &isrc, &isrc, desc, &info); + return info; + } + + static inline + void trmm(char side, + char uplo, + char trans, + char diag, + int m, + int n, + double alpha, + double* a, + double* b, + int* desc) + { + int isrc = 1; + pdtrmm_(&side, &uplo, &trans, &diag, &m, &n, &alpha, a, &isrc, &isrc, desc, b, &isrc, &isrc, desc); + } + + static inline + void trmm(char side, + char uplo, + char trans, + char diag, + int m, + int n, + std::complex alpha, + std::complex* a, + std::complex* b, + int* desc) + { + int isrc = 1; + pztrmm_(&side, &uplo, &trans, &diag, &m, &n, &alpha, a, &isrc, &isrc, desc, b, &isrc, &isrc, desc); + } + + static inline + void hemm(char side, + char uplo, + int na, + std::complex alpha, + std::complex* a, + std::complex* b, + std::complex beta, + std::complex* c, + int* desc) + { + int isrc = 1; + pzhemm_(&side, + &uplo, + &na, + &na, + &alpha, + a, + &isrc, + &isrc, + desc, + b, + &isrc, + &isrc, + desc, + &beta, + c, + &isrc, + &isrc, + desc); + } }; #endif // __MPI diff --git a/source/source_base/module_fft/fft_cpu.cpp b/source/source_base/module_fft/fft_cpu.cpp index 75854a93980..281ec21895b 100644 --- a/source/source_base/module_fft/fft_cpu.cpp +++ b/source/source_base/module_fft/fft_cpu.cpp @@ -11,7 +11,7 @@ void FFT_CPU::initfft(int nx_in, int rixy_in, int ns_in, int nplane_in, - int nproc_in, + int nproc_in, bool gamma_only_in, bool xprime_in) { diff --git a/source/source_base/module_fft/fft_cpu.h b/source/source_base/module_fft/fft_cpu.h index ec47768d8e9..5c6f18b5c8d 100644 --- a/source/source_base/module_fft/fft_cpu.h +++ b/source/source_base/module_fft/fft_cpu.h @@ -40,10 +40,10 @@ class FFT_CPU : public FFT_BASE bool gamma_only_in, bool xprime_in = true) override; - ABACUS_FFT_WEAK + ABACUS_FFT_WEAK void setupFFT() override; - // void initplan(const unsigned int& flag = 0); + // void initplan(const unsigned int& flag = 0); ABACUS_FFT_WEAK void cleanFFT() override; diff --git a/source/source_base/module_fft/fft_cuda.h b/source/source_base/module_fft/fft_cuda.h index 7734caffa9d..0b7a7aacf6e 100644 --- a/source/source_base/module_fft/fft_cuda.h +++ b/source/source_base/module_fft/fft_cuda.h @@ -13,7 +13,7 @@ class FFT_CUDA : public FFT_BASE FFT_CUDA(){}; ~FFT_CUDA(){}; - void setupFFT() override; + void setupFFT() override; void clear() override; diff --git a/source/source_base/module_fft/fft_dsp.h b/source/source_base/module_fft/fft_dsp.h index a735fd1c500..f29ed85f8e3 100644 --- a/source/source_base/module_fft/fft_dsp.h +++ b/source/source_base/module_fft/fft_dsp.h @@ -20,7 +20,7 @@ class FFT_DSP : public FFT_BASE FFT_DSP(){}; ~FFT_DSP(){}; - void setupFFT() override; + void setupFFT() override; void clear() override; diff --git a/source/source_base/module_mixing/broyden_mixing.cpp b/source/source_base/module_mixing/broyden_mixing.cpp index e189010a464..f8211412387 100644 --- a/source/source_base/module_mixing/broyden_mixing.cpp +++ b/source/source_base/module_mixing/broyden_mixing.cpp @@ -106,14 +106,14 @@ void Broyden_Mixing::tem_cal_coef(const Mixing_Data& mdata, std::function(dF); FPTYPE* FP_F = static_cast(F); diff --git a/source/source_base/module_out/binstream.cpp b/source/source_base/module_out/binstream.cpp index f5eed1830f0..dce826c3e02 100644 --- a/source/source_base/module_out/binstream.cpp +++ b/source/source_base/module_out/binstream.cpp @@ -1,22 +1,23 @@ #include #include #include "source_base/module_out/binstream.h" +#include "source_base/tool_quit.h" namespace { // Binstream is always a *binary* stream. On Windows, fopen mode "r"/"w"/"a" // opens in text mode, which translates CRLF and treats 0x1A as EOF, corrupting -// binary data (e.g. wavefunction / charge files) -> "Some data didn't be read". +// binary data (e.g. wavefunction / charge files) -> "Some data couldn't be read". // Append "b" if the caller didn't, so binary mode is always used. On POSIX the // "b" flag is a harmless no-op, so the Linux behaviour is unchanged. std::string ensure_binary_mode(const char* op) { - std::string mode(op ? op : ""); - if (mode.find('b') == std::string::npos) - { - mode += 'b'; - } - return mode; + std::string mode(op ? op : ""); + if (mode.find('b') == std::string::npos) + { + mode += 'b'; + } + return mode; } } // namespace @@ -28,46 +29,79 @@ std::string ensure_binary_mode(const char* op) * "a": add * "w": write */ -Binstream::Binstream(const std::string filename,const char *op) +Binstream::Binstream(const std::string filename, const char *op) { - fileptr=fopen(filename.c_str(),ensure_binary_mode(op).c_str()); + fileptr = fopen(filename.c_str(), ensure_binary_mode(op).c_str()); } Binstream::~Binstream() { - if(fileptr != NULL) fclose(fileptr); + if(fileptr != NULL) + { + // A destructor must not terminate the program, so a delayed write + // failure surfacing here can only be reported, not quit on. Callers + // that need the guarantee should use close(), which checks the result. + if(fclose(fileptr) != 0) + { + ModuleBase::WARNING("Binstream::~Binstream", + "fclose failed: some buffered data may not have been written."); + } + fileptr = NULL; + } } // close file -void Binstream:: close() +void Binstream::close() { - fclose(fileptr); - fileptr = NULL; - return; + if(fileptr == NULL) + { + return; + } + // fclose() flushes the buffer; its failure is the last chance to detect + // delayed write errors (e.g. RLIMIT_FSIZE, full disk). + if(fclose(fileptr) != 0) + { + fileptr = NULL; + ModuleBase::WARNING_QUIT("Binstream::close", + "fclose failed: some buffered data may not have been written."); + } + fileptr = NULL; + return; } // open a file -void Binstream::open(const std::string filename,const char *op) +void Binstream::open(const std::string filename, const char *op) { - fileptr=fopen(filename.c_str(),ensure_binary_mode(op).c_str()); + // Close any previously opened file first; overwriting fileptr directly + // would leak the old handle together with its buffered data. + close(); + fileptr = fopen(filename.c_str(), ensure_binary_mode(op).c_str()); } // ! operator // we can use if(!Binstream) ... bool Binstream::operator!() const { - if (fileptr==NULL) - return true; - else - return false; + if (fileptr == NULL) + { + return true; + } + else + { + return false; + } } // bool operator // we can use if(Binstream) ... Binstream::operator bool() const { - if (fileptr==NULL) - return false; - else - return true; + if (fileptr == NULL) + { + return false; + } + else + { + return true; + } } diff --git a/source/source_base/module_out/binstream.h b/source/source_base/module_out/binstream.h index c7fbf128ed5..01f59344efe 100644 --- a/source/source_base/module_out/binstream.h +++ b/source/source_base/module_out/binstream.h @@ -6,6 +6,7 @@ #include #include +#include "source_base/tool_quit.h" /** * @brief A stream to read or write binary data. @@ -13,111 +14,112 @@ */ class Binstream { - public: - Binstream(){ - fileptr=NULL; //we should use NULL (not nullptr) here because FILE use NULL. - }; - Binstream(const std::string,const char*); - ~Binstream(); - FILE* fileptr = nullptr; - void close(); - void open(const std::string,const char*); - bool operator!() const; - operator bool() const; - - template - Binstream& operator>>( T& data); - - template - Binstream& operator<<(const T& data); - - template - Binstream& read(T* data,const int n); - - template - Binstream& write(const T* data,const int n); - + public: + Binstream() = default; + Binstream(const std::string, const char*); + ~Binstream(); + // The destructor closes the owned FILE*, so copying would double-close. + Binstream(const Binstream&) = delete; + Binstream& operator=(const Binstream&) = delete; + FILE* fileptr = nullptr; + void close(); + void open(const std::string, const char*); + bool operator!() const; + operator bool() const; + + template + Binstream& operator>>(T& data); + + template + Binstream& operator<<(const T& data); + + template + Binstream& read(T* data, const int n); + + template + Binstream& write(const T* data, const int n); }; // read a data from file template -Binstream& Binstream:: operator>>(T& data) +Binstream& Binstream::operator>>(T& data) { - const int size=sizeof(T); - size_t ch = fread(&data,size,1,this->fileptr); - if(ch<1) - { - std::cout<<"Error in Binstream: Some data didn't be read."<fileptr == NULL) + { + ModuleBase::WARNING_QUIT("Binstream::operator>>", + "cannot read from an unopened file."); + } + size_t ch = fread(&data, size, 1, this->fileptr); + if(ch < 1) + { + ModuleBase::WARNING_QUIT("Binstream::operator>>", + "Some data couldn't be read. Please make sure you are using op: \"r\"."); + } + return *this; } // write a data into file template -Binstream& Binstream:: operator<<(const T& data) +Binstream& Binstream::operator<<(const T& data) { - const int size=sizeof(T); - fwrite(&data,size,1,this->fileptr); + const int size = sizeof(T); + if(this->fileptr == NULL) + { + ModuleBase::WARNING_QUIT("Binstream::operator<<", + "cannot write to an unopened file."); + } + size_t ch = fwrite(&data, size, 1, this->fileptr); + // fwrite() reports success while data is still buffered; fflush() forces + // the buffer out so delayed write errors (e.g. RLIMIT_FSIZE, full disk) + // are detected here instead of being silently dropped by fclose(). + if(ch < 1 || fflush(this->fileptr) != 0) + { + ModuleBase::WARNING_QUIT("Binstream::operator<<", + "Some data couldn't be written."); + } return *this; } -//read an array of data +// read an array of data template Binstream& Binstream::read(T* data, const int n) { - const int size=sizeof(T); - size_t ch = fread(data,size,n,this->fileptr); - if(chfileptr == NULL) + { + ModuleBase::WARNING_QUIT("Binstream::read", + "cannot read from an unopened file."); + } + size_t ch = fread(data, size, n, this->fileptr); + if(ch < static_cast(n)) { - std::cout<<"Error in Binstream: Some dynamic memory didn't be read."< Binstream& Binstream::write(const T* data, const int n) { - const int size=sizeof(T); - fwrite(data,size,n,this->fileptr); - return *this; -} - - - -/*//for dynamic memory -//malloc_usable_size has problem! -template -Binstream& operator<<(Binstream& wstream,T* &data) -{ - int size=sizeof(T); - int n=malloc_usable_size(data)/sizeof(T); - fwrite(data,size,n,wstream.fileptr); - return wstream; -} - - -//for dynamic memory -template -Binstream& operator>>(Binstream& rstream,T* &data) -{ - int size=sizeof(T); - int n=malloc_usable_size(data)/sizeof(T); - std::cout<fileptr == NULL) + { + ModuleBase::WARNING_QUIT("Binstream::write", + "cannot write to an unopened file."); } - return rstream; + size_t ch = fwrite(data, size, n, this->fileptr); + // fwrite() reports success while data is still buffered; fflush() forces + // the buffer out so delayed write errors (e.g. RLIMIT_FSIZE, full disk) + // are detected here instead of being silently dropped by fclose(). + if(ch < static_cast(n) || fflush(this->fileptr) != 0) + { + ModuleBase::WARNING_QUIT("Binstream::write", + "Some array elements couldn't be written."); + } + return *this; } -*/ #endif - diff --git a/source/source_base/module_out/csr_reader.cpp b/source/source_base/module_out/csr_reader.cpp index ad38a574227..7697f057bbf 100644 --- a/source/source_base/module_out/csr_reader.cpp +++ b/source/source_base/module_out/csr_reader.cpp @@ -131,54 +131,54 @@ void csrFileReader::parseFile() // std::cout << " ss1: " << ss.str() << std::endl; readLine(); - size_t count1 = 0; + size_t count1 = 0; while (count1 < nonZero) { if (ss.eof() || ss.fail()) { readLine(); - } + } if (ss >> csr_values[count1]) { count1++; } - } + } // std::cout << "count1=" << count1 << std::endl; // read CSR column indices readLine(); // std::cout << " ss2: " << ss.str() << std::endl; - size_t count2 = 0; + size_t count2 = 0; while (count2 < nonZero) { if (ss.eof() || ss.fail()) { readLine(); - } + } if (ss >> csr_col_ind[count2]) { count2++; } - } + } // std::cout << "count2=" << count2 << std::endl; // read row pointers readLine(); // std::cout << " ss3: " << ss.str() << std::endl; - size_t count3 = 0; + size_t count3 = 0; while (count3 < matrixDimension + 1) { if (ss.eof() || ss.fail()) { readLine(); - } + } if (ss >> csr_row_ptr[count3]) { count3++; } - } + } // std::cout << "count3=" << count3 << std::endl; // create sparse matrix diff --git a/source/source_base/module_out/file_reader.cpp b/source/source_base/module_out/file_reader.cpp index dd36731652c..7211ac44456 100644 --- a/source/source_base/module_out/file_reader.cpp +++ b/source/source_base/module_out/file_reader.cpp @@ -11,7 +11,7 @@ FileReader::FileReader(std::string filename) ifs.open(filename.c_str()); if (!ifs.is_open()) { - ModuleBase::WARNING_QUIT("FileReader::FileReader", "Error opening file"); + ModuleBase::WARNING_QUIT("FileReader::FileReader", "Error opening file: " + filename); } } diff --git a/source/source_base/module_out/filename.cpp b/source/source_base/module_out/filename.cpp index c7818642f6f..4dd86dc2bea 100644 --- a/source/source_base/module_out/filename.cpp +++ b/source/source_base/module_out/filename.cpp @@ -6,17 +6,17 @@ namespace ModuleIO { std::string filename_output( - const std::string &directory, - const std::string &property, - const std::string &basis, - const int ik_local, // the ik index within each pool - const std::vector &ik2iktot, - const int nspin, - const int nkstot, - const int out_type, - const bool out_app_flag, - const bool gamma_only, - const int istep, + const std::string &directory, + const std::string &property, + const std::string &basis, + const int ik_local, // the ik index within each pool + const std::vector &ik2iktot, + const int nspin, + const int nkstot, + const int out_type, + const bool out_app_flag, + const bool gamma_only, + const int istep, const int iter) { // output filename = "{global_out_dir}/property{s}{spin index} @@ -24,57 +24,57 @@ std::string filename_output( // + {".txt"/".dat"}" std::set valid_properties = { "wf", "chg", "hk", "sk", "tk", "vxc", "vxck", "vlk", "vnlk", "vhk", "vexxk" }; - if (valid_properties.find(property) == valid_properties.end()) - { - ModuleBase::WARNING_QUIT("ModuleIO::filename_output", "unknown property in filename function"); - } + if (valid_properties.find(property) == valid_properties.end()) + { + ModuleBase::WARNING_QUIT("ModuleIO::filename_output", "unknown property in filename function"); + } - std::set valid_basis = {"pw", "nao"}; - if (valid_basis.find(basis) == valid_basis.end()) - { - ModuleBase::WARNING_QUIT("ModuleIO::filename_output", "unknown basis in filename function"); - } + std::set valid_basis = {"pw", "nao"}; + if (valid_basis.find(basis) == valid_basis.end()) + { + ModuleBase::WARNING_QUIT("ModuleIO::filename_output", "unknown basis in filename function"); + } assert(ik_local>=0); // mohan update 2025.05.07, if KPAR>1, "<" works - assert(ik2iktot.size() <= nkstot); + assert(ik2iktot.size() <= nkstot); assert(nspin>0); - // spin index - int is0 = -1; - // ik0 is the k-point index, starting from 0 - int ik0 = ik2iktot[ik_local]; + // spin index + int is0 = -1; + // ik0 is the k-point index, starting from 0 + int ik0 = ik2iktot[ik_local]; - // spin part - std::string spin_block; + // spin part + std::string spin_block; // mohan add 2026-01-04, overlap matrix is the same for any spin - if(property != "sk") - { - if(nspin == 1) - { - // do nothing - } - else if(nspin == 2) - { - const int half_k = nkstot/2; - if(ik0 >= half_k) - { - is0 = 2; - ik0 -= half_k; - } - else - { - is0 = 1; - } - spin_block = "s" + std::to_string(is0); - } - else if(nspin==4) - { - is0 = 4; - spin_block = "s" + std::to_string(is0); - } - } + if(property != "sk") + { + if(nspin == 1) + { + // do nothing + } + else if(nspin == 2) + { + const int half_k = nkstot/2; + if(ik0 >= half_k) + { + is0 = 2; + ik0 -= half_k; + } + else + { + is0 = 1; + } + spin_block = "s" + std::to_string(is0); + } + else if(nspin==4) + { + is0 = 4; + spin_block = "s" + std::to_string(is0); + } + } // k-point part @@ -86,14 +86,14 @@ std::string filename_output( else { // mohan add 20250921 - if(property=="hk" || property=="sk" || property=="tk") - { - kpoint_block = std::to_string(ik0+1); - } - else - { - kpoint_block = "k" + std::to_string(ik0+1); - } + if(property=="hk" || property=="sk" || property=="tk") + { + kpoint_block = std::to_string(ik0+1); + } + else + { + kpoint_block = "k" + std::to_string(ik0+1); + } } std::string istep_block diff --git a/source/source_base/module_out/filename.h b/source/source_base/module_out/filename.h index 895bf29ef05..d76970c769f 100644 --- a/source/source_base/module_out/filename.h +++ b/source/source_base/module_out/filename.h @@ -26,14 +26,14 @@ std::string filename_output( const std::string &directory, const std::string &property, const std::string &basis, - const int ik_local, - const std::vector &ik2iktot, - const int nspin, - const int nkstot, + const int ik_local, + const std::vector &ik2iktot, + const int nspin, + const int nkstot, const int out_type, - const bool out_app_flag, - const bool gamma_only, - const int istep=-1, + const bool out_app_flag, + const bool gamma_only, + const int istep=-1, const int iter=-1); } diff --git a/source/source_base/module_out/sparse_matrix.cpp b/source/source_base/module_out/sparse_matrix.cpp index 535f2bb93b6..c0b3357762b 100644 --- a/source/source_base/module_out/sparse_matrix.cpp +++ b/source/source_base/module_out/sparse_matrix.cpp @@ -41,8 +41,8 @@ void SparseMatrix::printToCSR(std::ostream& ofs, int precision) size_t count1 = 0; for (const auto &element : elements) { - if(count1%6==0) ofs << '\n'; - count1++; + if(count1%6==0) ofs << '\n'; + count1++; ofs << " " << element.second; } ofs << '\n'; @@ -51,8 +51,8 @@ void SparseMatrix::printToCSR(std::ostream& ofs, int precision) size_t count2 = 0; for (const auto &element : elements) { - if(count2%16==0) ofs << '\n'; - count2++; + if(count2%16==0) ofs << '\n'; + count2++; ofs << " " << element.first.second; int row = element.first.first; csr_row_ptr[row + 1]++; diff --git a/source/source_base/module_parallel/para_rgrid_world.cpp b/source/source_base/module_parallel/para_rgrid_world.cpp index 961eb22d2e2..27f0ec8881f 100644 --- a/source/source_base/module_parallel/para_rgrid_world.cpp +++ b/source/source_base/module_parallel/para_rgrid_world.cpp @@ -77,7 +77,10 @@ void ParaRgridWorld::reduce_across_pools(double* data, const ParaWorld& kmesh_wo if (!kmesh_world.valid()) return; if (kmesh_world.size() <= 1) return; - assert(data != nullptr); + // A rank may own zero real-space grid points (nrxx == 0); the buffer is + // legitimately null in that case. MPI_Allreduce below uses count 0 and + // ignores the buffer. Only a null buffer with a non-zero nrxx is a bug. + assert(data != nullptr || nrxx() == 0); // Equal-sized pools: corresponding ranks have identical z-slab layouts, // so local buffers can be summed directly without redistribution. diff --git a/source/source_base/name_angular.h b/source/source_base/name_angular.h index 48b9dae1520..75bfb453164 100644 --- a/source/source_base/name_angular.h +++ b/source/source_base/name_angular.h @@ -3,14 +3,14 @@ namespace ModuleBase { - const std::string Name_Angular[5][11] = - { - {"s"}, - {"pz", "px", "py"}, - {"dz^2", "dxz", "dyz", "dx^2-y^2", "dxy"}, - {"fz^3", "fxz^2", "fyz^2", "fzx^2-zy^2", "fxyz", "fx^3-3*xy^2", "f3yx^2-y^3"}, - {"g1", "g2", "g3", "g4", "g5", "g6", "g7", "g8", "g9"} - }; // name of atomic orbital jiyy add 2022-05-10 + const std::string Name_Angular[5][11] = + { + {"s"}, + {"pz", "px", "py"}, + {"dz^2", "dxz", "dyz", "dx^2-y^2", "dxy"}, + {"fz^3", "fxz^2", "fyz^2", "fzx^2-zy^2", "fxyz", "fx^3-3*xy^2", "f3yx^2-y^3"}, + {"g1", "g2", "g3", "g4", "g5", "g6", "g7", "g8", "g9"} + }; // name of atomic orbital jiyy add 2022-05-10 } #endif diff --git a/source/source_base/ndarray.h b/source/source_base/ndarray.h index 5f76fb9e282..b4da1feea52 100644 --- a/source/source_base/ndarray.h +++ b/source/source_base/ndarray.h @@ -1,5 +1,5 @@ /** - * @file NDArray.h + * @file ndarray.h * @author your name (you@domain.com) * @brief under the restriction of C++11, a simple alternative to std::vector + std::mdspan. In source_base/module_container/ATen/tensor.h, tensor class provides a cross-device container, but std::string is not supported. Therefore, this class is to provide a general (but CPU-only) container for multi-dimensional data. It can easily convert to ontainer::Tensor. * @version 0.1 diff --git a/source/source_base/opt_dcsrch.cpp b/source/source_base/opt_dcsrch.cpp index 303ac658181..189ed554cdd 100644 --- a/source/source_base/opt_dcsrch.cpp +++ b/source/source_base/opt_dcsrch.cpp @@ -1,10 +1,10 @@ #include "opt_dcsrch.h" #include -#include // This file is translated from fortran codes dcstep.f of scipy. -// The structure and all annotation of the original file have been retained. +// The numerical algorithm and original Fortran annotations are retained; +// status storage uses std::string. // See original source at https://github.com/scipy/scipy/blob/main/scipy/optimize/minpack2/dcstep.f. // sunliang 2022-05-30 @@ -16,7 +16,7 @@ int dcsrch(double& stp, double& ftol, double& gtol, double& xtol, - char* task, + std::string& task, double& stpmin, double& stpmax, int* isave, @@ -115,7 +115,7 @@ int dcsrch(double& stp, // c is less than xtol. // c On exit xtol is unchanged. // c - // c task is a character variable of length at least 60. + // c task is a status string updated in place. // c On initial entry task must be set to 'START'. // c On exit task indicates the required action: // c @@ -184,45 +184,45 @@ int dcsrch(double& stp, double&, double&); // c Initialization block. - if (strncmp(task, "START", 5) == 0) + if (task.compare(0, 5, "START") == 0) { // c Check the input arguments for errors. if (stp < stpmin) { - strcpy(task, "ERROR: STP .LT. STPMIN"); + task = "ERROR: STP .LT. STPMIN"; } if (stp > stpmax) { - strcpy(task, "ERROR: STP .GT. STPMAX"); + task = "ERROR: STP .GT. STPMAX"; } if (g >= 0.) { - strcpy(task, "ERROR: INITIAL G .GE. ZERO"); + task = "ERROR: INITIAL G .GE. ZERO"; } if (ftol < 0.) { - strcpy(task, "ERROR: FTOL .LT. ZERO"); + task = "ERROR: FTOL .LT. ZERO"; } if (gtol < 0.) { - strcpy(task, "ERROR: GTOL .LT. ZERO"); + task = "ERROR: GTOL .LT. ZERO"; } if (xtol < 0.) { - strcpy(task, "ERROR: XTOL .LT. ZERO"); + task = "ERROR: XTOL .LT. ZERO"; } if (stpmin < 0.) { - strcpy(task, "ERROR: STPMIN .LT. ZERO"); + task = "ERROR: STPMIN .LT. ZERO"; } if (stpmax < stpmin) { - strcpy(task, "ERROR: STPMAX .LT. STPMIN"); + task = "ERROR: STPMAX .LT. STPMIN"; } // c Exit if there are errors on input. - if (strncmp(task, "ERROR", 5) == 0) + if (task.compare(0, 5, "ERROR") == 0) { return 0; } @@ -250,7 +250,7 @@ int dcsrch(double& stp, gy = ginit; stmin = zero; stmax = stp + stp * xtrapu; - strcpy(task, "FG"); + task = "FG"; goto L10; } else @@ -293,32 +293,31 @@ int dcsrch(double& stp, if (brackt && (stp <= stmin || stp >= stmax)) { - strcpy(task, "WARNING: ROUNDING ERRORS PREVENT PROGRESS"); + task = "WARNING: ROUNDING ERRORS PREVENT PROGRESS"; } if (brackt && stmax - stmin <= xtol * stmax) { - strcpy(task, "WARNING: XTOL TEST SATISFIED"); + task = "WARNING: XTOL TEST SATISFIED"; } if (stp == stpmax && f <= ftest && g <= gtest) { - strcpy(task, "WARNING: STP = STPMAX"); + task = "WARNING: STP = STPMAX"; } if (stp == stpmin && (f > ftest || g >= gtest)) { - strcpy(task, "WARNING: STP = STPMIN"); + task = "WARNING: STP = STPMIN"; } // c Test for convergence. if (f <= ftest && std::abs(g) <= gtol * (-ginit)) { - strcpy(task, "CONVERGENCE"); - // strcpy(task, "CONVERGENCE", 11); + task = "CONVERGENCE"; } // c Test for termination. - if (strncmp(task, "WARN", 4) == 0 || strncmp(task, "CONV", 4) == 0) + if (task.compare(0, 4, "WARN") == 0 || task.compare(0, 4, "CONV") == 0) { goto L10; } @@ -389,7 +388,7 @@ int dcsrch(double& stp, } // c Obtain another function and derivative. - strcpy(task, "FG"); + task = "FG"; L10: // c Save local variables. if (brackt) @@ -715,7 +714,7 @@ int dcsrch(double& stp, stp = stpf; } -void Opt_DCsrch::dcSrch(double& f, double& g, double& rstp, char* rtask) +void Opt_DCsrch::dcSrch(double& f, double& g, double& rstp, std::string& rtask) { dcsrch(rstp, f, diff --git a/source/source_base/opt_dcsrch.h b/source/source_base/opt_dcsrch.h index 6432833350c..cbfbfb4bbc3 100644 --- a/source/source_base/opt_dcsrch.h +++ b/source/source_base/opt_dcsrch.h @@ -2,6 +2,7 @@ #define OPT_DCSRCH_H #include +#include #include "constants.h" @@ -64,7 +65,7 @@ class Opt_DCsrch * @param g the derivative of the function at 0 on initial entry. * On subsequent entries g is the derivative of the function at x + stp * d. * @param rstp the optimized step length, assert the initial value is larger than zero. - * @param rtask a character variable of length at least 60. + * @param rtask the line-search status string, updated in place. * On initial entry task must be set to 'START'. * On exit task indicates the required action: * If task(1:2) = 'FG' then evaluate the function and derivative at stp and call dcsrch again. @@ -73,7 +74,7 @@ class Opt_DCsrch * The exit value of stp contains the best point found during the search. * If task(1:5) = 'ERROR' then there is an error in the input arguments. */ - void dcSrch(double& f, double& g, double& rstp, char* rtask); + void dcSrch(double& f, double& g, double& rstp, std::string& rtask); private: double ftol_ = 1e-4; // nonnegative tolerance for the sufficient decrease condition. @@ -87,4 +88,4 @@ class Opt_DCsrch }; } // namespace ModuleBase -#endif \ No newline at end of file +#endif diff --git a/source/source_base/opt_tn.cpp b/source/source_base/opt_tn.cpp new file mode 100644 index 00000000000..a93a815c0d8 --- /dev/null +++ b/source/source_base/opt_tn.cpp @@ -0,0 +1,53 @@ +#include "opt_tn.h" + +namespace ModuleBase +{ + +Opt_TN::Opt_TN() +{ + this->mach_prec_ = std::numeric_limits::epsilon(); // get machine precise +} + +void Opt_TN::allocate(int nx) +{ + this->nx_ = nx; + this->cg_.allocate(this->nx_); +} + +void Opt_TN::set_para(double dV) +{ + this->dV_ = dV; + this->cg_.set_para(this->dV_); +} + +void Opt_TN::refresh(int nx_new) +{ + this->iter_ = 0; + if (nx_new != 0) + { + this->nx_ = nx_new; + } + this->cg_.refresh(nx_new); +} + +double Opt_TN::inner_product(double* pa, double* pb, int length) +{ + double innerproduct = BlasConnector::dot(length, pa, 1, pb, 1); + innerproduct *= this->dV_; + return innerproduct; +} + +double Opt_TN::get_epsilon(double* px, double* pcg_direction) +{ + double epsilon = 0.; + double xx = this->inner_product(px, px, this->nx_); + Parallel_Reduce::reduce_all(xx); + double dd = this->inner_product(pcg_direction, pcg_direction, this->nx_); + Parallel_Reduce::reduce_all(dd); + epsilon = 2 * sqrt(this->mach_prec_) * (1 + sqrt(xx)) / sqrt(dd); + // epsilon = 2 * sqrt(this->mach_prec_) * (1 + sqrt(this->inner_product(px, px, this->nx_))) + // / sqrt(this->inner_product(pcg_direction, pcg_direction, this->nx_)); + return epsilon; +} + +} // namespace ModuleBase diff --git a/source/source_base/opt_tn.h b/source/source_base/opt_tn.h new file mode 100644 index 00000000000..32b9afdcd48 --- /dev/null +++ b/source/source_base/opt_tn.h @@ -0,0 +1,204 @@ +#ifndef OPT_TN_H +#define OPT_TN_H + +#include "opt_cg.h" + +#include + +namespace ModuleBase +{ +/** + * @brief A class designed to deal with optimization problems with Truncated-Newton (TN) method. + * At each step, the optimization direction d is determined by roughly + * solving linear equation Hd=-g, where H is Hessian matrix of f(x), and g is the gradient df(x)/dx. + * In this section, we get Hd by interpolation, and solve Hd=-g with CG method. + * We set three truncated conditions: + * 1) the residual of CG method has decreased more than 90%; + * 2) the number of CG iteration is more than 50; + * 3) the residual of CG method doesn't decrease and the CG iteration is larger than 9. + * @author sunliang + */ +class Opt_TN +{ + public: + Opt_TN(); + ~Opt_TN() {}; + + /** + * @brief Allocate the space for the arrays in cg_. + * + * @param nx length of the solution array x + */ + void allocate(int nx); + + void set_para(double dV); + + /** + * @brief Refresh the class. + * If nx changes, reallocate space in cg_. + * + * @param nx_new length of new x, default 0 means the length doesn't change + */ + void refresh(int nx_new = 0); + + template + void next_direct( + double* px, // current x + double* pgradient, // df(x)/dx + int& flag, // record which truncated condition was triggered, 0 for cond.1, 1 for cond.2, and 2 for cond.3 + double* rdirect, // next optimization direction + T* t, // point of class T, which contains the gradient function + void (T::*p_calGradient)( + double* ptemp_x, + double* rtemp_gradient) // a function point, which calculates the gradient at provided x + ); + + int get_iter() + { + return this->iter_; + } + + private: + Opt_CG cg_; + double dV_ = 1.; + + int nx_ = 0; // length of the solution array x + int iter_ = 0; // number of the iteration + double mach_prec_ = 0.; // machine precision + + double inner_product(double* pa, double* pb, int length); + + /** + * @brief Get epsilon used in interpolation. + * epsilon = 2*sqrt(mach_prec_) * (1+|x|) / |d|. + * || means modulu. + * @param px x + * @param pcg_direction the direction of cg_ + * @return epsilon + */ + double get_epsilon(double* px, double* pcg_direction); +}; + +/** + * @brief Get the next direction d with TN method. + * + * @tparam T for example, ESolver_OF + * @param [in] px current x. + * @param [in] pgradient df(x)/dx. + * @param [out] flag record which truncated condition was triggered, 0 for cond.1, 1 for cond.2, and 2 for cond.3. + * @param [out] rdirect next optimization direction. + * @param [in] t pointer of class T, which contains the gradient function. + * @param [in] p_calGradient a function pointer, which calculates gradient at provided x. + */ +template +void Opt_TN::next_direct(double* px, + double* pgradient, + int& flag, + double* rdirect, + T* t, + void (T::*p_calGradient)(double* px, double* rgradient)) +{ + // initialize arrays and parameters + ModuleBase::GlobalFunc::ZEROS(rdirect, this->nx_); // very important + + double* minus_gradient = new double[this->nx_]; // b=-g, which will be used in CG + double* temp_x = new double[this->nx_]; // temp_x = x + step * cg_direct, used in interpolation + double* temp_gradient = new double[this->nx_]; // df(temp_x)/dx + double* cg_direct = new double[this->nx_]; // rdirect += cg_alpha * cg_direct at each step + double* temp_Hcgd = new double[this->nx_]; // Hessian * cg_direct + for (int i = 0; i < this->nx_; ++i) + { + minus_gradient[i] = -pgradient[i]; + } + ModuleBase::GlobalFunc::ZEROS(cg_direct, this->nx_); + ModuleBase::GlobalFunc::ZEROS(temp_x, this->nx_); + ModuleBase::GlobalFunc::ZEROS(temp_gradient, this->nx_); + ModuleBase::GlobalFunc::ZEROS(temp_Hcgd, this->nx_); + + cg_.refresh(0, minus_gradient); + int cg_iter = 0; + int cg_ifPD = 0; + + double epsilon = 0.; // step length in interpolation + double cg_alpha = 0.; // step length got by CG + double init_residual = 0.; // initial residual of CG + double last_residual = 0.; // last residual of CG + double curr_residual = 0.; // current residual of CG + + while (true) + { + cg_.next_direct(temp_Hcgd, 0, cg_direct); + + // get temp_Hcgd with interpolation + // Hcgd = (df(temp_x)/dx - df(x)/x) / epsilon, where temp_x = x + step * cg_direct + epsilon = this->get_epsilon(px, cg_direct); + // epsilon = 1e-9; + for (int i = 0; i < this->nx_; ++i) + { + temp_x[i] = px[i] + epsilon * cg_direct[i]; + } + (t->*p_calGradient)(temp_x, temp_gradient); + for (int i = 0; i < this->nx_; ++i) + { + temp_Hcgd[i] = (temp_gradient[i] - pgradient[i]) / epsilon; + } + + // get CG step length and update rdirect + cg_alpha = cg_.step_length(temp_Hcgd, cg_direct, cg_ifPD); + if (cg_ifPD == -1) // Hessian is not positive definite, and cgiter = 1. + { + for (int i = 0; i < this->nx_; ++i) + { + rdirect[i] += cg_alpha * cg_direct[i]; + } + flag = -1; + break; + } + else if (cg_ifPD == -2) // Hessian is not positive definite, and cgiter > 1. + { + flag = -2; + break; + } + + for (int i = 0; i < this->nx_; ++i) + { + rdirect[i] += cg_alpha * cg_direct[i]; + } + + // store residuals used in truncated conditions + last_residual = curr_residual; + curr_residual = cg_.get_residual(); + cg_iter = cg_.get_iter(); + if (cg_iter == 1) + { + init_residual = curr_residual; + } + + // check truncated conditions + // if (curr_residual < 1e-12) + if (curr_residual < 0.1 * init_residual) + { + flag = 0; + // std::cout << "cg_ iter_ = " << cg_iter << "\n"; + break; + } + else if (cg_iter > 50) + { + flag = 1; + break; + } + else if ((fabs(curr_residual - last_residual) / curr_residual) < 0.01 && cg_iter > 9) + { + flag = 2; + break; + } + } + this->iter_++; + delete[] minus_gradient; + delete[] temp_gradient; + delete[] temp_x; + delete[] temp_Hcgd; + delete[] cg_direct; +} +} // namespace ModuleBase +#endif diff --git a/source/source_base/opt_tn.hpp b/source/source_base/opt_tn.hpp deleted file mode 100644 index 37c8199f286..00000000000 --- a/source/source_base/opt_tn.hpp +++ /dev/null @@ -1,239 +0,0 @@ -#ifndef OPT_TN_H -#define OPT_TN_H - -#include "opt_cg.h" - -#include - -namespace ModuleBase -{ -/** - * @brief A class designed to deal with optimization problems with Truncated-Newton (TN) method. - * At each step, the optimization direction d is determined by roughly - * solving linear equation Hd=-g, where H is Hessian matrix of f(x), and g is the gradient df(x)/dx. - * In this section, we get Hd by interpolation, and solve Hd=-g with CG method. - * We set three truncated conditions: - * 1) the residual of CG method has decreased more than 90%; - * 2) the number of CG iteration is more than 50; - * 3) the residual of CG method doesn't decrease and the CG iteration is larger than 9. - * @author sunliang - */ -class Opt_TN -{ - public: - Opt_TN() - { - this->mach_prec_ = std::numeric_limits::epsilon(); // get machine precise - } - ~Opt_TN() {}; - - /** - * @brief Allocate the space for the arrays in cg_. - * - * @param nx length of the solution array x - */ - void allocate(int nx) - { - this->nx_ = nx; - this->cg_.allocate(this->nx_); - } - - void set_para(double dV) - { - this->dV_ = dV; - this->cg_.set_para(this->dV_); - } - - /** - * @brief Refresh the class. - * If nx changes, reallocate space in cg_. - * - * @param nx_new length of new x, default 0 means the length doesn't change - */ - void refresh(int nx_new = 0) - { - this->iter_ = 0; - if (nx_new != 0) - { - this->nx_ = nx_new; - } - this->cg_.refresh(nx_new); - } - - template - void next_direct( - double* px, // current x - double* pgradient, // df(x)/dx - int& flag, // record which truncated condition was triggered, 0 for cond.1, 1 for cond.2, and 2 for cond.3 - double* rdirect, // next optimization direction - T* t, // point of class T, which contains the gradient function - void (T::*p_calGradient)( - double* ptemp_x, - double* rtemp_gradient) // a function point, which calculates the gradient at provided x - ); - - int get_iter() - { - return this->iter_; - } - - private: - Opt_CG cg_; - double dV_ = 1.; - - int nx_ = 0; // length of the solution array x - int iter_ = 0; // number of the iteration - double mach_prec_ = 0.; // machine precision - - double inner_product(double* pa, double* pb, int length) - { - double innerproduct = BlasConnector::dot(length, pa, 1, pb, 1); - innerproduct *= this->dV_; - return innerproduct; - } - - /** - * @brief Get epsilon used in interpolation. - * epsilon = 2*sqrt(mach_prec_) * (1+|x|) / |d|. - * || means modulu. - * @param px x - * @param pcg_direction the direction of cg_ - * @return epsilon - */ - double get_epsilon(double* px, double* pcg_direction) - { - double epsilon = 0.; - double xx = this->inner_product(px, px, this->nx_); - Parallel_Reduce::reduce_all(xx); - double dd = this->inner_product(pcg_direction, pcg_direction, this->nx_); - Parallel_Reduce::reduce_all(dd); - epsilon = 2 * sqrt(this->mach_prec_) * (1 + sqrt(xx)) / sqrt(dd); - // epsilon = 2 * sqrt(this->mach_prec_) * (1 + sqrt(this->inner_product(px, px, this->nx_))) - // / sqrt(this->inner_product(pcg_direction, pcg_direction, this->nx_)); - return epsilon; - } -}; - -/** - * @brief Get the next direction d with TN method. - * - * @tparam T for example, ESolver_OF - * @param [in] px current x. - * @param [in] pgradient df(x)/dx. - * @param [out] flag record which truncated condition was triggered, 0 for cond.1, 1 for cond.2, and 2 for cond.3. - * @param [out] rdirect next optimization direction. - * @param [in] t pointer of class T, which contains the gradient function. - * @param [in] p_calGradient a function pointer, which calculates gradient at provided x. - */ -template -void Opt_TN::next_direct(double* px, - double* pgradient, - int& flag, - double* rdirect, - T* t, - void (T::*p_calGradient)(double* px, double* rgradient)) -{ - // initialize arrays and parameters - ModuleBase::GlobalFunc::ZEROS(rdirect, this->nx_); // very important - - double* minus_gradient = new double[this->nx_]; // b=-g, which will be used in CG - double* temp_x = new double[this->nx_]; // temp_x = x + step * cg_direct, used in interpolation - double* temp_gradient = new double[this->nx_]; // df(temp_x)/dx - double* cg_direct = new double[this->nx_]; // rdirect += cg_alpha * cg_direct at each step - double* temp_Hcgd = new double[this->nx_]; // Hessian * cg_direct - for (int i = 0; i < this->nx_; ++i) - { - minus_gradient[i] = -pgradient[i]; - } - ModuleBase::GlobalFunc::ZEROS(cg_direct, this->nx_); - ModuleBase::GlobalFunc::ZEROS(temp_x, this->nx_); - ModuleBase::GlobalFunc::ZEROS(temp_gradient, this->nx_); - ModuleBase::GlobalFunc::ZEROS(temp_Hcgd, this->nx_); - - cg_.refresh(0, minus_gradient); - int cg_iter = 0; - int cg_ifPD = 0; - - double epsilon = 0.; // step length in interpolation - double cg_alpha = 0.; // step length got by CG - double init_residual = 0.; // initial residual of CG - double last_residual = 0.; // last residual of CG - double curr_residual = 0.; // current residual of CG - - while (true) - { - cg_.next_direct(temp_Hcgd, 0, cg_direct); - - // get temp_Hcgd with interpolation - // Hcgd = (df(temp_x)/dx - df(x)/x) / epsilon, where temp_x = x + step * cg_direct - epsilon = this->get_epsilon(px, cg_direct); - // epsilon = 1e-9; - for (int i = 0; i < this->nx_; ++i) - { - temp_x[i] = px[i] + epsilon * cg_direct[i]; - } - (t->*p_calGradient)(temp_x, temp_gradient); - for (int i = 0; i < this->nx_; ++i) - { - temp_Hcgd[i] = (temp_gradient[i] - pgradient[i]) / epsilon; - } - - // get CG step length and update rdirect - cg_alpha = cg_.step_length(temp_Hcgd, cg_direct, cg_ifPD); - if (cg_ifPD == -1) // Hessian is not positive definite, and cgiter = 1. - { - for (int i = 0; i < this->nx_; ++i) - { - rdirect[i] += cg_alpha * cg_direct[i]; - } - flag = -1; - break; - } - else if (cg_ifPD == -2) // Hessian is not positive definite, and cgiter > 1. - { - flag = -2; - break; - } - - for (int i = 0; i < this->nx_; ++i) - { - rdirect[i] += cg_alpha * cg_direct[i]; - } - - // store residuals used in truncated conditions - last_residual = curr_residual; - curr_residual = cg_.get_residual(); - cg_iter = cg_.get_iter(); - if (cg_iter == 1) - { - init_residual = curr_residual; - } - - // check truncated conditions - // if (curr_residual < 1e-12) - if (curr_residual < 0.1 * init_residual) - { - flag = 0; - // std::cout << "cg_ iter_ = " << cg_iter << "\n"; - break; - } - else if (cg_iter > 50) - { - flag = 1; - break; - } - else if ((fabs(curr_residual - last_residual) / curr_residual) < 0.01 && cg_iter > 9) - { - flag = 2; - break; - } - } - this->iter_++; - delete[] minus_gradient; - delete[] temp_gradient; - delete[] temp_x; - delete[] temp_Hcgd; - delete[] cg_direct; -} -} // namespace ModuleBase -#endif \ No newline at end of file diff --git a/source/source_base/output.cpp b/source/source_base/output.cpp index 7512e0bfe2e..2de3882c7aa 100644 --- a/source/source_base/output.cpp +++ b/source/source_base/output.cpp @@ -14,17 +14,17 @@ void output::printrm(std::ofstream &ofs,const std::string &s, const ModuleBase:: { if (j % 8 == 0) ofs << "\n "; - if (std::abs(m(i,j)) > limit) - { - ofs << std::setprecision(6) << std::setw(12) << m(i,j); + if (std::abs(m(i,j)) > limit) + { + ofs << std::setprecision(6) << std::setw(12) << m(i,j); + } + else + { + ofs << std::setw(12) << "0"; } - else - { - ofs << std::setw(12) << "0"; - } } } - ofs << std::endl; + ofs << std::endl; return; } @@ -37,16 +37,16 @@ void output::printrm(const std::string &s, const ModuleBase::matrix &m, const do for (int i = 0;i < b1;i++) { - //std::cout << "\n row=" << i; + //std::cout << "\n row=" << i; for (int j = 0;j < b2;j++) { - if (j % 8 == 0) std::cout << "\n "; + if (j % 8 == 0) std::cout << "\n "; if (std::abs(m(i,j)) > limit) std::cout << std::setprecision(6) << std::setw(12) << m(i,j); else std::cout< #include +#include + +Parallel_2D::~Parallel_2D() +{ +#ifdef __MPI + release_blacs_grid(); +#endif +} + +Parallel_2D::Parallel_2D(Parallel_2D&& rhs) noexcept +{ + *this = std::move(rhs); +} + +Parallel_2D& Parallel_2D::operator=(Parallel_2D&& rhs) noexcept +{ + if (this != &rhs) + { +#ifdef __MPI + release_blacs_grid(); + owns_blacs_ctxt_ = rhs.owns_blacs_ctxt_; + rhs.owns_blacs_ctxt_ = false; + blacs_ctxt = rhs.blacs_ctxt; + rhs.blacs_ctxt = -1; +#endif + nrow = rhs.nrow; + ncol = rhs.ncol; + nloc = rhs.nloc; + nb = rhs.nb; + dim0 = rhs.dim0; + dim1 = rhs.dim1; + coord[0] = rhs.coord[0]; + coord[1] = rhs.coord[1]; + is_serial = rhs.is_serial; + global2local_row_ = std::move(rhs.global2local_row_); + global2local_col_ = std::move(rhs.global2local_col_); + local2global_row_ = std::move(rhs.local2global_row_); + local2global_col_ = std::move(rhs.local2global_col_); +#ifdef __MPI + std::copy(rhs.desc, rhs.desc + 9, desc); +#endif + } + return *this; +} bool Parallel_2D::in_this_processor(const std::size_t iw1_all, const std::size_t iw2_all) const { @@ -86,6 +130,16 @@ void Parallel_2D::_init_proc_grid(const MPI_Comm comm, const bool mode) Cblacs_gridinfo(blacs_ctxt, &dim0, &dim1, &coord[0], &coord[1]); } +void Parallel_2D::release_blacs_grid() +{ + if (owns_blacs_ctxt_ && blacs_ctxt >= 0) + { + Cblacs_gridexit(blacs_ctxt); + blacs_ctxt = -1; + owns_blacs_ctxt_ = false; + } +} + void Parallel_2D::_set_dist_info(const int mg, const int ng, const int nb) { this->nb = nb; @@ -120,14 +174,26 @@ void Parallel_2D::_set_dist_info(const int mg, const int ng, const int nb) int Parallel_2D::init(const int mg, const int ng, const int nb, const MPI_Comm comm, const bool mode) { + release_blacs_grid(); _init_proc_grid(comm, mode); + owns_blacs_ctxt_ = true; _set_dist_info(mg, ng, nb); return nrow == 0 || ncol == 0; } int Parallel_2D::set(const int mg, const int ng, const int nb, const int blacs_ctxt) { + // Reusing the context owned by this object must not destroy the grid or + // drop ownership; otherwise the grid about to be reused is released. + if (blacs_ctxt == this->blacs_ctxt && owns_blacs_ctxt_) + { + Cblacs_gridinfo(blacs_ctxt, &dim0, &dim1, &coord[0], &coord[1]); + _set_dist_info(mg, ng, nb); + return nrow == 0 || ncol == 0; + } + release_blacs_grid(); this->blacs_ctxt = blacs_ctxt; + owns_blacs_ctxt_ = false; Cblacs_gridinfo(blacs_ctxt, &dim0, &dim1, &coord[0], &coord[1]); _set_dist_info(mg, ng, nb); return nrow == 0 || ncol == 0; @@ -152,6 +218,11 @@ void Parallel_2D::set_serial(const int mg, const int ng) global2local_col_ = local2global_col_; is_serial = true; #ifdef __MPI + release_blacs_grid(); + // A serial layout must not reference any BLACS grid: for borrowers + // release_blacs_grid() is a no-op, so clear the handle explicitly; + // otherwise the object keeps a reference to the old grid, which also + // dangles once the owner destroys it. blacs_ctxt = -1; #endif } diff --git a/source/source_base/parallel_2d.h b/source/source_base/parallel_2d.h index 08fb5f25f2e..e6a4987206a 100644 --- a/source/source_base/parallel_2d.h +++ b/source/source_base/parallel_2d.h @@ -15,10 +15,10 @@ class Parallel_2D { public: Parallel_2D() = default; - ~Parallel_2D() = default; + ~Parallel_2D(); - Parallel_2D& operator=(Parallel_2D&& rhs) = default; - Parallel_2D(Parallel_2D&& rhs) = default; + Parallel_2D(Parallel_2D&& rhs) noexcept; + Parallel_2D& operator=(Parallel_2D&& rhs) noexcept; /// number of local rows int get_row_size() const @@ -195,6 +195,12 @@ class Parallel_2D #ifdef __MPI void _init_proc_grid(const MPI_Comm comm, const bool mode); void _set_dist_info(const int mg, const int ng, const int nb); + + /// Release the BLACS grid if owned by this object + void release_blacs_grid(); + + /// Whether this object owns the BLACS context (created by init()) + bool owns_blacs_ctxt_ = false; #endif }; #endif diff --git a/source/source_base/parallel_grid.cpp b/source/source_base/parallel_grid.cpp index c8a058e52e9..d293533545c 100644 --- a/source/source_base/parallel_grid.cpp +++ b/source/source_base/parallel_grid.cpp @@ -17,17 +17,17 @@ Parallel_Grid::~Parallel_Grid() { } -void Parallel_Grid::init(const int& ncx_in, - const int& ncy_in, - const int& ncz_in, - const int& nczp_in, - const int& nrxx_in, - const int& nbz_in, - const int& bz_in, - const int nprocgroup) -{ - ModuleBase::TITLE("Parallel_Grid", "init"); + +void Parallel_Grid::init_serial(const int& ncx_in, + const int& ncy_in, + const int& ncz_in, + const int& nczp_in, + const int& nrxx_in, + const int& nbz_in, + const int& bz_in) +{ + ModuleBase::TITLE("Parallel_Grid", "init_serial"); this->ncx = ncx_in; this->ncy = ncy_in; @@ -49,10 +49,11 @@ void Parallel_Grid::init(const int& ncx_in, this->ncxy = ncx * ncy; this->ncxyz = ncxy * ncz; - -#ifndef __MPI - return; -#endif +} +#ifdef __MPI +void Parallel_Grid::init_parallel(const int nprocgroup) +{ + ModuleBase::TITLE("Parallel_Grid", "init_parallel"); // enable to call this function again liuyu 2023-03-10 if (!this->numz.empty()) @@ -64,7 +65,6 @@ void Parallel_Grid::init(const int& ncx_in, this->whichpro_loc.clear(); } - // (2) assert(this->numz.empty()); assert(GlobalV::KPAR > 0); @@ -94,18 +94,32 @@ void Parallel_Grid::init(const int& ncx_in, this->whichpro_loc[ip].assign(this->ncz, 0); } - this->z_distribution(); - - return; + this->z_distribution(GlobalV::KPAR); } +#endif +void Parallel_Grid::init(const int& ncx_in, + const int& ncy_in, + const int& ncz_in, + const int& nczp_in, + const int& nrxx_in, + const int& nbz_in, + const int& bz_in, + const int nprocgroup) +{ + this->init_serial(ncx_in, ncy_in, ncz_in, nczp_in, nrxx_in, nbz_in, bz_in); -void Parallel_Grid::z_distribution() +#ifdef __MPI + this->init_parallel(nprocgroup); +#endif +} +void Parallel_Grid::z_distribution(const int kpar) { assert(!this->numz.empty()); + assert(kpar > 0); - std::vector startp(GlobalV::KPAR); + std::vector startp(kpar); startp[0] = 0; - for (int ip = 0; ip < GlobalV::KPAR; ip++) + for (int ip = 0; ip < kpar; ip++) { const int nproc = nproc_in_pool[ip]; @@ -160,7 +174,10 @@ void Parallel_Grid::reduce_across_pools(double* data) const return; } - assert(data != nullptr); + // A rank may own zero real-space grid points (nrxx == 0); the buffer is + // legitimately null in that case. MPI_Allreduce below uses count 0 and + // ignores the buffer. Only a null buffer with a non-zero nrxx is a bug. + assert(data != nullptr || this->nrxx == 0); if (KP_WORLD != MPI_COMM_NULL) { // Equal-sized pools give corresponding ranks identical z-slab layouts, @@ -352,7 +369,12 @@ void Parallel_Grid::reduce(double* rhotot, const double* const rhoin, const bool return; } - assert(rhoin != nullptr); + // A rank may own zero real-space grid points (nrxx == 0) when the grid is + // decomposed across more processes than it has slabs. In that case the + // source buffer is legitimately null: MPI_Gatherv is called with + // sendcount 0 below and ignores the send buffer. Only a null buffer with a + // non-zero nrxx is a genuine bug. + assert(rhoin != nullptr || this->nrxx == 0); assert(this->nrxx == this->ncxy * this->nczp); int pool_size = 0; diff --git a/source/source_base/parallel_grid.h b/source/source_base/parallel_grid.h index 8cc7592384c..495021cc83b 100644 --- a/source/source_base/parallel_grid.h +++ b/source/source_base/parallel_grid.h @@ -45,7 +45,19 @@ class Parallel_Grid private: - void z_distribution(void); + void z_distribution(const int kpar); + + void init_serial(const int& ncx_in, + const int& ncy_in, + const int& ncz_in, + const int& nczp_in, + const int& nrxx_in, + const int& nbz_in, + const int& bz_in); + +#ifdef __MPI + void init_parallel(const int nprocgroup); +#endif #ifdef __MPI void zpiece_distribute(double* zpiece, const int& iz, double* rho, const bool is_sdft) const; diff --git a/source/source_base/parallel_reduce.cpp b/source/source_base/parallel_reduce.cpp index 62ac884f0a2..741b38f36d5 100644 --- a/source/source_base/parallel_reduce.cpp +++ b/source/source_base/parallel_reduce.cpp @@ -73,6 +73,13 @@ void Parallel_Reduce::reduce_min_pool(const int& nproc_in_pool, T& v) #endif } +void Parallel_Reduce::reduce_max(double* values, const int count) +{ +#ifdef __MPI + MPI_Allreduce(MPI_IN_PLACE, values, count, MPI_DOUBLE, MPI_MAX, MPI_COMM_WORLD); +#endif +} + template void Parallel_Reduce::reduce_max_pool(const int& nproc_in_pool, T& v) { @@ -170,4 +177,4 @@ void Parallel_Reduce::gather_int_all(int& v, int* all) MPI_Allgather(&v, 1, MPI_INT, all, 1, MPI_INT, MPI_COMM_WORLD); #endif return; -} \ No newline at end of file +} diff --git a/source/source_base/parallel_reduce.h b/source/source_base/parallel_reduce.h index 620620f9827..2e2f842f3a2 100644 --- a/source/source_base/parallel_reduce.h +++ b/source/source_base/parallel_reduce.h @@ -65,6 +65,8 @@ template void reduce_min(T& v); template void reduce_max(T& v); +/** @brief Reduce an array of maxima across all ranks. */ +void reduce_max(double* values, const int count); template void reduce_min_pool(const int& nproc_in_pool, T& v); template diff --git a/source/source_base/random.h b/source/source_base/random.h index 42c526866ed..9ae87c58dc8 100644 --- a/source/source_base/random.h +++ b/source/source_base/random.h @@ -8,42 +8,42 @@ namespace ModuleBase class Random { - public: - Random(); - ~Random(); - - static void between0and1( double *v, const int &num ) - { - assert( v!= NULL); - assert( num > 1); - for(int i=0; i( std::rand() ) / RAND_MAX; - } - } - - static double betweenMinus2and2(void) - { - return 2.0*betweenMinus1and1(); - } - - static double betweenMinus1and1(void) - { - const int a = std::rand() % 2; - if(a==0) return between0and1(); - else if(a==1) return betweenMinus1and0(); - else throw(std::string(__FILE__)+" line "+std::to_string(__LINE__)); // Peize Lin add to fix warning 2019-05-01 - } - - static double between0and1(void) - { - return static_cast( std::rand() )/RAND_MAX; - } - - static double betweenMinus1and0(void) - { - return -static_cast( std::rand() )/RAND_MAX; - } + public: + Random(); + ~Random(); + + static void between0and1( double *v, const int &num ) + { + assert( v!= NULL); + assert( num > 1); + for(int i=0; i( std::rand() ) / RAND_MAX; + } + } + + static double betweenMinus2and2(void) + { + return 2.0*betweenMinus1and1(); + } + + static double betweenMinus1and1(void) + { + const int a = std::rand() % 2; + if(a==0) return between0and1(); + else if(a==1) return betweenMinus1and0(); + else throw(std::string(__FILE__)+" line "+std::to_string(__LINE__)); // Peize Lin add to fix warning 2019-05-01 + } + + static double between0and1(void) + { + return static_cast( std::rand() )/RAND_MAX; + } + + static double betweenMinus1and0(void) + { + return -static_cast( std::rand() )/RAND_MAX; + } }; diff --git a/source/source_base/realarray.cpp b/source/source_base/realarray.cpp index b9983ed804e..e179a582d46 100644 --- a/source/source_base/realarray.cpp +++ b/source/source_base/realarray.cpp @@ -13,60 +13,60 @@ int realArray::arrayCount = 0; void realArrayAlloc() { - std::cout << "\n Allocation error for realArray " << std::endl; - exit(0); + std::cout << "\n Allocation error for realArray " << std::endl; + exit(0); } realArray::realArray(const int d1,const int d2,const int d3) { - dim = 3; - bound1 = (d1 <= 0) ? 1 : d1; - bound2 = (d2 <= 0) ? 1 : d2; - bound3 = (d3 <= 0) ? 1 : d3; - bound4 = 0; + dim = 3; + bound1 = (d1 <= 0) ? 1 : d1; + bound2 = (d2 <= 0) ? 1 : d2; + bound3 = (d3 <= 0) ? 1 : d3; + bound4 = 0; - size = bound1 * bound2 * bound3 ; //* sizeof(float); + size = bound1 * bound2 * bound3 ; //* sizeof(float); - auto handler_old = std::set_new_handler(realArrayAlloc); - ptr = new double[size]; - std::set_new_handler(handler_old); - zero_out(); - assert(ptr != 0); + auto handler_old = std::set_new_handler(realArrayAlloc); + ptr = new double[size]; + std::set_new_handler(handler_old); + zero_out(); + assert(ptr != 0); - ++arrayCount; + ++arrayCount; } realArray::realArray(const int d1,const int d2,const int d3,const int d4) { - dim = 4; - bound1 = (d1 <= 0) ? 1 : d1; - bound2 = (d2 <= 0) ? 1 : d2; - bound3 = (d3 <= 0) ? 1 : d3; - bound4 = (d4 <= 0) ? 1 : d4; + dim = 4; + bound1 = (d1 <= 0) ? 1 : d1; + bound2 = (d2 <= 0) ? 1 : d2; + bound3 = (d3 <= 0) ? 1 : d3; + bound4 = (d4 <= 0) ? 1 : d4; - size = bound1 * bound2 * bound3 * bound4 ; //* sizeof(float); + size = bound1 * bound2 * bound3 * bound4 ; //* sizeof(float); - auto handler_old = std::set_new_handler(realArrayAlloc); - ptr = new double[size]; - std::set_new_handler(handler_old); - zero_out(); + auto handler_old = std::set_new_handler(realArrayAlloc); + ptr = new double[size]; + std::set_new_handler(handler_old); + zero_out(); - ++arrayCount; + ++arrayCount; } realArray::realArray(const realArray &cd) { - this->size = cd.getSize(); - this->ptr = new double[size]; - for (int i = 0; i < size; i++) - this->ptr[i] = cd.ptr[i]; - this->dim = cd.dim; - this->bound1 = cd.bound1; - this->bound2 = cd.bound2; - this->bound3 = cd.bound3; - this->bound4 = cd.bound4; + this->size = cd.getSize(); + this->ptr = new double[size]; + for (int i = 0; i < size; i++) + this->ptr[i] = cd.ptr[i]; + this->dim = cd.dim; + this->bound1 = cd.bound1; + this->bound2 = cd.bound2; + this->bound3 = cd.bound3; + this->bound4 = cd.bound4; - ++arrayCount; + ++arrayCount; } @@ -82,43 +82,43 @@ realArray ::~realArray() void realArray::freemem() { - delete [] ptr; - ptr = NULL; + delete [] ptr; + ptr = NULL; } void realArray::create(const int d1,const int d2,const int d3,const int d4) { - size = d1 * d2 * d3 * d4; - assert(size>0); + size = d1 * d2 * d3 * d4; + assert(size>0); - dim = 4; + dim = 4; - bound1 = d1; - bound2 = d2; - bound3 = d3; - bound4 = d4; + bound1 = d1; + bound2 = d2; + bound3 = d3; + bound4 = d4; - delete [] ptr; - ptr = new double[size]; + delete [] ptr; + ptr = new double[size]; - zero_out(); // mohan modify 2009-09-17 - - assert(ptr != 0); + zero_out(); // mohan modify 2009-09-17 + + assert(ptr != 0); } void realArray::create(const int d1,const int d2,const int d3) { - size = d1 * d2 * d3; - assert(size>0); + size = d1 * d2 * d3; + assert(size>0); - dim = 3; + dim = 3; - bound1 = d1; - bound2 = d2; - bound3 = d3; - bound4 = 1; + bound1 = d1; + bound2 = d2; + bound3 = d3; + bound4 = 1; - delete [] ptr; + delete [] ptr; ptr = new double[size]; zero_out(); assert(ptr != 0); @@ -126,14 +126,14 @@ void realArray::create(const int d1,const int d2,const int d3) const realArray &realArray::operator=(const realArray &right) { - for (int i = 0;i < size;i++) ptr[i] = right.ptr[i]; - return *this;// enables x = y = z; + for (int i = 0;i < size;i++) ptr[i] = right.ptr[i]; + return *this;// enables x = y = z; } const realArray &realArray::operator=(const double &right) { - for (int i = 0;i < size;i++) ptr[i] = right; - return *this;// enables x = y = z; + for (int i = 0;i < size;i++) ptr[i] = right; + return *this;// enables x = y = z; } //******************************************************** @@ -144,20 +144,20 @@ const realArray &realArray::operator=(const double &right) const double &realArray::operator() (const int ind1,const int ind2,const int ind3)const { - assert(ind1>=0); assert(ind1=0); assert(ind2=0); assert(ind3=0); assert(ind1=0); assert(ind2=0); assert(ind3=0); assert(ind1=0); assert(ind2=0); assert(ind3=0); assert(ind4=0); assert(ind1=0); assert(ind2=0); assert(ind3=0); assert(ind4=0); assert(ind1=0); assert(ind2=0); assert(ind3=0); assert(ind1=0); assert(ind2=0); assert(ind3=0); assert(ind1=0); assert(ind2=0); assert(ind3=0); assert(ind4=0); assert(ind1=0); assert(ind2=0); assert(ind3=0); assert(ind4 Sph_Bessel_Recursive_Pool::D1::sb_pool; void Sph_Bessel_Recursive::D1::set_dx( const double dx_in ) { - if(finish_set_dx && dx_in!=dx) - throw std::runtime_error("Sph_Bessel_Recursive::set_dx, dx can only set once"); - else - { - dx = dx_in; - finish_set_dx = true; - } + if(finish_set_dx && dx_in!=dx) + throw std::runtime_error("Sph_Bessel_Recursive::set_dx, dx can only set once"); + else + { + dx = dx_in; + finish_set_dx = true; + } } const std::vector> & Sph_Bessel_Recursive::D1::cal_jlx( const int lmax, const size_t ix_size ) { - if(lmax<0) - throw std::invalid_argument("Sph_Bessel_Recursive::jlx l<0"); - cal_jlx_0( lmax+1 ); - cal_jlx_smallx( lmax+1, ix_size ); - cal_jlx_recursive( lmax+1, ix_size ); - return jlx; + if(lmax<0) + throw std::invalid_argument("Sph_Bessel_Recursive::jlx l<0"); + cal_jlx_0( lmax+1 ); + cal_jlx_smallx( lmax+1, ix_size ); + cal_jlx_recursive( lmax+1, ix_size ); + return jlx; } void Sph_Bessel_Recursive::D1::cal_jlx_0( const int l_size ) { - if(jlx.size() < static_cast(l_size)) - jlx.resize(l_size); - - for( int l=0; l!=l_size; ++l ) - { - if(jlx[l].size()<1) - { - jlx[l].resize(1); - if(0==l) - jlx[l][0] = 1.0; - else - jlx[l][0] = 0.0; - } - } + if(jlx.size() < static_cast(l_size)) + jlx.resize(l_size); + + for( int l=0; l!=l_size; ++l ) + { + if(jlx[l].size()<1) + { + jlx[l].resize(1); + if(0==l) + jlx[l][0] = 1.0; + else + jlx[l][0] = 0.0; + } + } } void Sph_Bessel_Recursive::D1::cal_jlx_smallx( const int l_size, const size_t ix_size ) { - if(jlx.size() < static_cast(l_size)) - jlx.resize(l_size); - - for( int l=0; l!=l_size; ++l ) - { - if(jlx[l].size()threshold/coeff*(l+1.5)*4, 1.0/(l+2) ); - - const size_t ix_size_begin = static_cast(jlx[l].size()); - const size_t ix_size_end = std::min( ix_size, static_cast(smallx_range/dx) ); - if(jlx[l].size()(l_size)) + jlx.resize(l_size); + + for( int l=0; l!=l_size; ++l ) + { + if(jlx[l].size()threshold/coeff*(l+1.5)*4, 1.0/(l+2) ); + + const size_t ix_size_begin = static_cast(jlx[l].size()); + const size_t ix_size_end = std::min( ix_size, static_cast(smallx_range/dx) ); + if(jlx[l].size()(l_size)) - jlx.resize(l_size); - - for( int l=0; l!=l_size; ++l ) - { - if(jlx[l].size()(jlx[l].size()); - jlx[l].resize(ix_size); - switch(l) - { - case 0: - for( size_t ix=ix_size_begin; ix(l_size)) + jlx.resize(l_size); + + for( int l=0; l!=l_size; ++l ) + { + if(jlx[l].size()(jlx[l].size()); + jlx[l].resize(ix_size); + switch(l) + { + case 0: + for( size_t ix=ix_size_begin; ix(jlx[l].size()); - jlx[l].resize(ix_size); - switch(l) - { - case 0: - { - for( size_t ix=ix_size_begin; ix(0.14/dx)+1 ); - for( size_t ix=ix_size_begin; ix(0.29/dx)+1 ); - for( size_t ix=ix_size_begin; ix(0.29/dx)+1 ); - for( size_t ix=ix_size_begin; ix(0.29/dx)+1 ); - for( size_t ix=ix_size_begin; ix(0.29/dx)+1 ); - for( size_t ix=ix_size_begin; ix(0.29/dx)+1 ); - for( size_t ix=ix_size_begin; ix(0.29/dx)+1 ); - for( size_t ix=ix_size_begin; ix(0.29/dx)+1 ); - for( size_t ix=ix_size_begin; ix(jlx[l].size()); + jlx[l].resize(ix_size); + switch(l) + { + case 0: + { + for( size_t ix=ix_size_begin; ix(0.14/dx)+1 ); + for( size_t ix=ix_size_begin; ix(0.29/dx)+1 ); + for( size_t ix=ix_size_begin; ix(0.29/dx)+1 ); + for( size_t ix=ix_size_begin; ix(0.29/dx)+1 ); + for( size_t ix=ix_size_begin; ix(0.29/dx)+1 ); + for( size_t ix=ix_size_begin; ix(0.29/dx)+1 ); + for( size_t ix=ix_size_begin; ix(0.29/dx)+1 ); + for( size_t ix=ix_size_begin; ix(0.29/dx)+1 ); + for( size_t ix=ix_size_begin; ix6"); - } - } + jlx[l][ix] = sin(x1)*( (1.0 +(-3003.0 +(1351350.0/x5 +(-192972780.0 +(9820936125.0 +(-151242416325.0 +316234143225.0/x2)/x2)/x2)/x2)/x2)/x2)/x1 ) + + cos(x1)*( (78.0 +(-75075.0 +(18378360.0 +(-1571349780.0 +(45831035250.0 -316234143225.0/x2)/x2)/x2)/x2)/x2)/x2 ); + } + break; + } + default: + throw std::invalid_argument("Sph_Bessel_Recursive::jlx l<0 or l>6"); + } + } //timeval t_end; gettimeofday( &t_end, NULL); //std::cout< Sph_Bessel_Recursive_Pool::D2::sb_pool; void Sph_Bessel_Recursive::D2::set_dx( const double dx_in ) { - if(finish_set_dx && dx_in!=dx) - throw std::runtime_error("Sph_Bessel_Recursive::set_dx, dx can only set once"); - else - { - dx = dx_in; - finish_set_dx = true; - } + if(finish_set_dx && dx_in!=dx) + throw std::runtime_error("Sph_Bessel_Recursive::set_dx, dx can only set once"); + else + { + dx = dx_in; + finish_set_dx = true; + } } const std::vector>> & Sph_Bessel_Recursive::D2::cal_jlx( const int lmax, const size_t ix1_size, const size_t ix2_size ) { - if(lmax<0) - throw std::invalid_argument("Sph_Bessel_Recursive::jlx l<0"); - cal_jlx_0( lmax+1, ix1_size, ix2_size ); - cal_jlx_smallx( lmax+1, ix1_size, ix2_size ); - cal_jlx_recursive( lmax+1, ix1_size, ix2_size ); - ModuleBase::Memory::record("ORB::Jl(x)", sizeof(double) * (lmax+1) * ix1_size * ix2_size); - return jlx; + if(lmax<0) + throw std::invalid_argument("Sph_Bessel_Recursive::jlx l<0"); + cal_jlx_0( lmax+1, ix1_size, ix2_size ); + cal_jlx_smallx( lmax+1, ix1_size, ix2_size ); + cal_jlx_recursive( lmax+1, ix1_size, ix2_size ); + ModuleBase::Memory::record("ORB::Jl(x)", sizeof(double) * (lmax+1) * ix1_size * ix2_size); + return jlx; } void Sph_Bessel_Recursive::D2::cal_jlx_0( const int l_size, const size_t ix1_size, const size_t ix2_size ) { - if(jlx.size() < static_cast(l_size)) - jlx.resize(l_size); - - for( int l=0; l(l_size)) + jlx.resize(l_size); + + for( int l=0; l(1,jlx0)); - - if( jlx[l][0].size()(1,jlx0)); + + if( jlx[l][0].size()(l_size)) - jlx.resize(l_size); - - for( int l=0; l!=l_size; ++l ) - { - const double coeff = sqrt(ModuleBase::PI)/tgamma(l+1.5)/pow(2,l+1); - const double smallx_range = pow( this->threshold/coeff*(l+1.5)*4, 1.0/(l+2) ); - - if(jlx[l].size()(1), jlx[l][ix1].size() ); - const size_t ix2_size_end = std::min( ix2_size, static_cast(smallx_range/dx)/ix1 ); - if(jlx[l][ix1].size()(l_size)) + jlx.resize(l_size); + + for( int l=0; l!=l_size; ++l ) + { + const double coeff = sqrt(ModuleBase::PI)/tgamma(l+1.5)/pow(2,l+1); + const double smallx_range = pow( this->threshold/coeff*(l+1.5)*4, 1.0/(l+2) ); + + if(jlx[l].size()(1), jlx[l][ix1].size() ); + const size_t ix2_size_end = std::min( ix2_size, static_cast(smallx_range/dx)/ix1 ); + if(jlx[l][ix1].size()(l_size)) - jlx.resize(l_size); - - for( int l=0; l!=l_size; ++l ) - { - switch(l) - { - case 0: - if(jlx[l].size()(1), jlx[l][ix1].size() ); - if(jlx[l][ix1].size()(1), jlx[l][ix1].size() ); - if(jlx[l][ix1].size()(1), jlx[l][ix1].size() ); - if(jlx[l][ix1].size()(l_size)) + jlx.resize(l_size); + + for( int l=0; l!=l_size; ++l ) + { + switch(l) + { + case 0: + if(jlx[l].size()(1), jlx[l][ix1].size() ); + if(jlx[l][ix1].size()(1), jlx[l][ix1].size() ); + if(jlx[l][ix1].size()(1), jlx[l][ix1].size() ); + if(jlx[l][ix1].size()(jlx[l].size()); - jlx[l].resize(ix_size); - switch(l) - { - case 0: - { - for( size_t ix=ix_size_begin; ix(0.14/dx)+1 ); - for( size_t ix=ix_size_begin; ix(0.29/dx)+1 ); - for( size_t ix=ix_size_begin; ix(0.29/dx)+1 ); - for( size_t ix=ix_size_begin; ix(0.29/dx)+1 ); - for( size_t ix=ix_size_begin; ix(0.29/dx)+1 ); - for( size_t ix=ix_size_begin; ix(0.29/dx)+1 ); - for( size_t ix=ix_size_begin; ix(0.29/dx)+1 ); - for( size_t ix=ix_size_begin; ix(0.29/dx)+1 ); - for( size_t ix=ix_size_begin; ix(jlx[l].size()); + jlx[l].resize(ix_size); + switch(l) + { + case 0: + { + for( size_t ix=ix_size_begin; ix(0.14/dx)+1 ); + for( size_t ix=ix_size_begin; ix(0.29/dx)+1 ); + for( size_t ix=ix_size_begin; ix(0.29/dx)+1 ); + for( size_t ix=ix_size_begin; ix(0.29/dx)+1 ); + for( size_t ix=ix_size_begin; ix(0.29/dx)+1 ); + for( size_t ix=ix_size_begin; ix(0.29/dx)+1 ); + for( size_t ix=ix_size_begin; ix(0.29/dx)+1 ); + for( size_t ix=ix_size_begin; ix(0.29/dx)+1 ); + for( size_t ix=ix_size_begin; ix6"); - } - } + jlx[l][ix] = sin(x1)*( (1.0 +(-3003.0 +(1351350.0/x5 +(-192972780.0 +(9820936125.0 +(-151242416325.0 +316234143225.0/x2)/x2)/x2)/x2)/x2)/x2)/x1 ) + + cos(x1)*( (78.0 +(-75075.0 +(18378360.0 +(-1571349780.0 +(45831035250.0 -316234143225.0/x2)/x2)/x2)/x2)/x2)/x2 ); + } + break; + } + default: + throw std::invalid_argument("Sph_Bessel_Recursive::jlx l<0 or l>6"); + } + } //timeval t_end; gettimeofday( &t_end, NULL); //std::cout<> & cal_jlx( const int lmax, const size_t ix_size ); - const std::vector> & get_jlx()const{ return jlx; } - - void set_dx(const double dx_in); - double get_dx()const{ return dx; } + const std::vector> & cal_jlx( const int lmax, const size_t ix_size ); + const std::vector> & get_jlx()const{ return jlx; } + + void set_dx(const double dx_in); + double get_dx()const{ return dx; } private: - std::vector> jlx; // jlx[l][x] - double dx = 0.0; - bool finish_set_dx = false; - - void cal_jlx_0( const int l_size ); - void cal_jlx_smallx( const int l_size, const size_t ix_size ); - void cal_jlx_recursive( const int l_size, const size_t ix_size ); - - - double threshold = 1e-8; // Peize Lin test + std::vector> jlx; // jlx[l][x] + double dx = 0.0; + bool finish_set_dx = false; + + void cal_jlx_0( const int l_size ); + void cal_jlx_smallx( const int l_size, const size_t ix_size ); + void cal_jlx_recursive( const int l_size, const size_t ix_size ); + + + double threshold = 1e-8; // Peize Lin test }; @@ -48,23 +48,23 @@ class Sph_Bessel_Recursive::D1 class Sph_Bessel_Recursive::D2 { public: - const std::vector>> & cal_jlx( const int lmax, const size_t ix1_size, const size_t ix2_size ); - const std::vector>> & get_jlx()const{ return jlx; } - - void set_dx(const double dx_in); - double get_dx()const{ return dx; } + const std::vector>> & cal_jlx( const int lmax, const size_t ix1_size, const size_t ix2_size ); + const std::vector>> & get_jlx()const{ return jlx; } + + void set_dx(const double dx_in); + double get_dx()const{ return dx; } private: - std::vector>> jlx; // jlx[l][x1][x2] - double dx = 0.0; - bool finish_set_dx = false; - - void cal_jlx_0( const int l_size, const size_t ix1_size, const size_t ix2_size ); - void cal_jlx_smallx( const int l_size, const size_t ix1_size, const size_t ix2_size ); - void cal_jlx_recursive( const int l_size, const size_t ix1_size, const size_t ix2_size ); - - - double threshold = 1e-8; // Peize Lin test + std::vector>> jlx; // jlx[l][x1][x2] + double dx = 0.0; + bool finish_set_dx = false; + + void cal_jlx_0( const int l_size, const size_t ix1_size, const size_t ix2_size ); + void cal_jlx_smallx( const int l_size, const size_t ix1_size, const size_t ix2_size ); + void cal_jlx_recursive( const int l_size, const size_t ix1_size, const size_t ix2_size ); + + + double threshold = 1e-8; // Peize Lin test }; @@ -72,16 +72,16 @@ class Sph_Bessel_Recursive::D2 class Sph_Bessel_Recursive_Pool { public: - class D1 - { - public: - static std::vector sb_pool; - }; - class D2 - { - public: - static std::vector sb_pool; - }; + class D1 + { + public: + static std::vector sb_pool; + }; + class D2 + { + public: + static std::vector sb_pool; + }; }; } diff --git a/source/source_base/test/CMakeLists.txt b/source/source_base/test/CMakeLists.txt index 3b8fb64deee..ce9df8138ba 100644 --- a/source/source_base/test/CMakeLists.txt +++ b/source/source_base/test/CMakeLists.txt @@ -153,6 +153,11 @@ AddTest( SOURCES opt_cg_test.cpp opt_test_tools.cpp mpi_test_main.cpp ) +AddTest( + TARGET MODULE_BASE_opt_dcsrch + SOURCES test_opt_dcsrch.cpp ../opt_dcsrch.cpp +) + AddTest( TARGET MODULE_BASE_opt_tn LIBS parameter base device diff --git a/source/source_base/test/binstream_test.cpp b/source/source_base/test/binstream_test.cpp index fc0d5b93383..3381b9e1f8f 100644 --- a/source/source_base/test/binstream_test.cpp +++ b/source/source_base/test/binstream_test.cpp @@ -1,5 +1,9 @@ #include "gtest/gtest.h" #include "gmock/gmock.h" +#ifdef __unix__ +#include +#include +#endif /************************************************ * unit test of binstream.cpp ***********************************************/ @@ -21,73 +25,151 @@ class BinstreamTest : public testing::Test TEST_F(BinstreamTest, variable) { - int a1,a2,a3; - Binstream wfc("wfc","w"); - EXPECT_EQ(!wfc, false); - wfc << 10 ; - wfc.close(); - EXPECT_EQ(bool(wfc), false); - wfc.open("wfc","a"); - EXPECT_EQ(bool(wfc), true); - wfc << 100; - wfc.close(); - EXPECT_EQ(!wfc, true); - - wfc.open("wfc","r"); - EXPECT_EQ(bool(wfc), true); - wfc >> a1; - wfc >> a2; - testing::internal::CaptureStdout(); - EXPECT_EXIT(wfc >> a3, ::testing::ExitedWithCode(0), ""); + int a1, a2, a3; + Binstream wfc("wfc", "w"); + EXPECT_EQ(!wfc, false); + wfc << 10; + wfc.close(); + EXPECT_EQ(bool(wfc), false); + wfc.open("wfc", "a"); + EXPECT_EQ(bool(wfc), true); + wfc << 100; + wfc.close(); + EXPECT_EQ(!wfc, true); + + wfc.open("wfc", "r"); + EXPECT_EQ(bool(wfc), true); + wfc >> a1; + wfc >> a2; + testing::internal::CaptureStdout(); + EXPECT_EXIT(wfc >> a3, ::testing::ExitedWithCode(1), ""); std::string output = testing::internal::GetCapturedStdout(); - EXPECT_THAT(output,testing::HasSubstr("Binstream")); - EXPECT_EQ(bool(wfc), true); - wfc.close(); - EXPECT_EQ(bool(wfc), false); + EXPECT_THAT(output, testing::HasSubstr("!NOTICE!")); + EXPECT_EQ(bool(wfc), true); + wfc.close(); + EXPECT_EQ(bool(wfc), false); - EXPECT_EQ(a1,10); - EXPECT_EQ(a2,100); + EXPECT_EQ(a1, 10); + EXPECT_EQ(a2, 100); - wfc.open("wfc","w"); - testing::internal::CaptureStdout(); - EXPECT_EXIT(wfc >> a3, ::testing::ExitedWithCode(0), ""); + wfc.open("wfc", "w"); + testing::internal::CaptureStdout(); + EXPECT_EXIT(wfc >> a3, ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); - EXPECT_THAT(output,testing::HasSubstr("Binstream")); - remove("wfc"); + EXPECT_THAT(output, testing::HasSubstr("!NOTICE!")); + remove("wfc"); - wfc.open("wfc", "r"); - EXPECT_EQ(bool(wfc), false); //If file is not open, return false. + wfc.open("wfc", "r"); + EXPECT_EQ(bool(wfc), false); // If file is not open, return false. - Binstream *p = new Binstream("wfc" , "r"); - delete p; + Binstream *p = new Binstream("wfc", "r"); + delete p; - remove("wfc"); // mohan add 2025-06-22 + remove("wfc"); // mohan add 2025-06-22 } TEST_F(BinstreamTest, array) { - - int a[10], b[11]; - for(int i = 0 ; i < 10 ; ++i) a[i] = i; - Binstream wwfc("wfc","w"); - wwfc.write(a, 10); - wwfc.close(); - - Binstream rwfc("wfc","r"); - testing::internal::CaptureStdout(); - EXPECT_EXIT(rwfc.read(b,11);, ::testing::ExitedWithCode(0), ""); + int a[10], b[11]; + for(int i = 0; i < 10; ++i) + { + a[i] = i; + } + Binstream wwfc("wfc", "w"); + wwfc.write(a, 10); + wwfc.close(); + + Binstream rwfc("wfc", "r"); + testing::internal::CaptureStdout(); + EXPECT_EXIT(rwfc.read(b, 11);, ::testing::ExitedWithCode(1), ""); std::string output = testing::internal::GetCapturedStdout(); - EXPECT_THAT(output,testing::HasSubstr("Binstream")); - rwfc.close(); - rwfc.open("wfc","r"); - rwfc.read(b, 10); - rwfc.close(); - remove("wfc"); - - for(int i = 0 ; i < 10 ; ++i) - { - EXPECT_EQ(a[i], b[i]); - } - - wwfc.open("wfc", "w"); + EXPECT_THAT(output, testing::HasSubstr("!NOTICE!")); + rwfc.close(); + rwfc.open("wfc", "r"); + rwfc.read(b, 10); + rwfc.close(); + remove("wfc"); + + for(int i = 0; i < 10; ++i) + { + EXPECT_EQ(a[i], b[i]); + } + + wwfc.open("wfc", "w"); +} + +TEST_F(BinstreamTest, WriteToUnopenedFile) +{ + Binstream ofs; + testing::internal::CaptureStdout(); + EXPECT_EXIT(ofs << 42, ::testing::ExitedWithCode(1), ""); + std::string output = testing::internal::GetCapturedStdout(); + EXPECT_THAT(output, testing::HasSubstr("!NOTICE!")); + + int a[3] = {1, 2, 3}; + testing::internal::CaptureStdout(); + EXPECT_EXIT(ofs.write(a, 3), ::testing::ExitedWithCode(1), ""); + output = testing::internal::GetCapturedStdout(); + EXPECT_THAT(output, testing::HasSubstr("!NOTICE!")); +} + +TEST_F(BinstreamTest, ReadFromUnopenedFile) +{ + Binstream ifs; + int v = 0; + testing::internal::CaptureStdout(); + EXPECT_EXIT(ifs >> v, ::testing::ExitedWithCode(1), ""); + std::string output = testing::internal::GetCapturedStdout(); + EXPECT_THAT(output, testing::HasSubstr("!NOTICE!")); + + int a[3] = {0, 0, 0}; + testing::internal::CaptureStdout(); + EXPECT_EXIT(ifs.read(a, 3), ::testing::ExitedWithCode(1), ""); + output = testing::internal::GetCapturedStdout(); + EXPECT_THAT(output, testing::HasSubstr("!NOTICE!")); +} + +// fwrite() may report success while the data is still buffered; with +// RLIMIT_FSIZE=1 the write() call must detect the failure (via fflush) +// instead of returning normally with only one byte on disk. +TEST_F(BinstreamTest, DelayedWriteFailureDetected) +{ +#ifdef __unix__ + // Ignore SIGXFSZ so the file-size limit surfaces as EFBIG from + // fflush/fwrite instead of killing the process. + struct sigaction old_act; + ASSERT_EQ(sigaction(SIGXFSZ, NULL, &old_act), 0); + struct sigaction ign_act = old_act; + ign_act.sa_handler = SIG_IGN; + ASSERT_EQ(sigaction(SIGXFSZ, &ign_act, NULL), 0); + + rlimit lim; + lim.rlim_cur = 1; // allow at most 1 byte per file + lim.rlim_max = RLIM_INFINITY; + ASSERT_EQ(setrlimit(RLIMIT_FSIZE, &lim), 0); + + int a[4] = {1, 2, 3, 4}; + EXPECT_EXIT( + { + Binstream wfc("wfc_rlimit", "w"); + wfc.write(a, 4); + wfc.close(); + }, + ::testing::ExitedWithCode(1), ""); + + lim.rlim_cur = RLIM_INFINITY; + ASSERT_EQ(setrlimit(RLIMIT_FSIZE, &lim), 0); + ASSERT_EQ(sigaction(SIGXFSZ, &old_act, NULL), 0); + remove("wfc_rlimit"); +#endif +} + +// A close() on a stream whose earlier buffered writes failed at flush time +// must also be caught (fclose returns EOF). +TEST_F(BinstreamTest, CloseFailureDetected) +{ + // close() on an unopened stream must be a safe no-op. + Binstream empty; + empty.close(); + EXPECT_EQ(bool(empty), false); } diff --git a/source/source_base/test/math_sphbes_test.cpp b/source/source_base/test/math_sphbes_test.cpp index f6a68529891..171d3eac694 100644 --- a/source/source_base/test/math_sphbes_test.cpp +++ b/source/source_base/test/math_sphbes_test.cpp @@ -4,6 +4,7 @@ #include // Dependency for memset initialization #include #include +#include #ifdef __MPI #include "mpi.h" @@ -287,12 +288,15 @@ TEST_F(Sphbes, SphericalBesselPrecisionGrid) r[i] = (i + 1) * dr; } - // test for new sphbesj + // Check both scalar and batch interfaces against the same reference data. + std::vector jl_new(nr); for (int l = l_lo; l <= l_hi; ++l) { + ModuleBase::Sphbes::sphbesj(nr, r, q, l, jl_new.data()); for (int i = 0; i < nr; ++i) { EXPECT_NEAR(ModuleBase::Sphbes::sphbesj(l, r[i] * q), Y[l * nr + i], 1e-12); + EXPECT_NEAR(jl_new[i], Y[l * nr + i], 1e-12); double tmp = std::abs(Y[l * nr + i] - ModuleBase::Sphbes::sphbesj(l, r[i] * q)); file_n.write(reinterpret_cast(&tmp), sizeof(double)); } diff --git a/source/source_base/test/memory_test.cpp b/source/source_base/test/memory_test.cpp index a533f3898c3..f9c65fecb56 100644 --- a/source/source_base/test/memory_test.cpp +++ b/source/source_base/test/memory_test.cpp @@ -25,9 +25,7 @@ namespace GlobalV * - std::ofstream file */ -#define private public #include "../memory_recorder.h" -#undef private class MemoryTest : public testing::Test { @@ -141,14 +139,11 @@ TEST_F(MemoryTest, printall) TEST_F(MemoryTest, finish) { - *ModuleBase::Memory::name = "tmp_name"; - *ModuleBase::Memory::class_name = "tmp_class_name"; - *ModuleBase::Memory::consume = 100.0; - ModuleBase::Memory::init_flag = true; ofs.open("tmp"); - // total memory is an internal parameter and added inside the class Memory + // record() allocates the tables and adds the entry that finish() then prints + // and releases; total memory is an internal parameter added inside Memory ModuleBase::Memory::record("Charge_Mixing","Rrho",1024*1024,"ModuleBase::Vector3"); EXPECT_NO_THROW(ModuleBase::Memory::finish(ofs)); ofs.close(); - EXPECT_FALSE(ModuleBase::Memory::init_flag); + EXPECT_FALSE(ModuleBase::Memory::is_initialized()); } diff --git a/source/source_base/test/opt_cg_test.cpp b/source/source_base/test/opt_cg_test.cpp index c2793f292d5..d6b1d38a72c 100644 --- a/source/source_base/test/opt_cg_test.cpp +++ b/source/source_base/test/opt_cg_test.cpp @@ -21,7 +21,7 @@ class CG_test : public testing::Test double residual = 10.; double tol = 1e-5; int final_iter = 0; - char *task = nullptr; + std::string task; double *Ap = nullptr; double *p = nullptr; double *x = nullptr; @@ -31,7 +31,6 @@ class CG_test : public testing::Test cg.set_para(1.); cg.allocate(tools.nx); cg.init_b(tools.le.b); - task = new char[60]; Ap = new double[tools.nx]; p = new double[tools.nx]; x = new double[tools.nx]; @@ -39,7 +38,6 @@ class CG_test : public testing::Test void TearDown() { - delete[] task; delete[] Ap; delete[] p; delete[] x; @@ -116,27 +114,27 @@ class CG_test : public testing::Test cg.next_direct(gradient, cg_label, p); for (int i = 0; i < 3; ++i) { temp_x[i] = x[i]; } - task[0] = 'S'; task[1] = 'T'; task[2] = 'A'; task[3] = 'R'; task[4] = 'T'; + task = "START"; while (true) { f = tools.func(temp_x, func_label); g = tools.dfuncdstp(temp_x, p, func_label); ds.dcSrch(f, g, step, task); - if (task[0] == 'F' && task[1] == 'G') + if (task.compare(0, 2, "FG") == 0) { for (int j = 0; j < 3; ++j) { temp_x[j] = x[j] + step * p[j]; } continue; } - else if (task[0] == 'C' && task[1] == 'O') + else if (task.compare(0, 2, "CO") == 0) { break; } - else if (task[0] == 'W' && task[1] == 'A') + else if (task.compare(0, 2, "WA") == 0) { break; } - else if (task[0] == 'E' && task[1] == 'R') + else if (task.compare(0, 2, "ER") == 0) { break; } diff --git a/source/source_base/test/opt_tn_test.cpp b/source/source_base/test/opt_tn_test.cpp index 8eaa7229bb5..aad28200f5f 100644 --- a/source/source_base/test/opt_tn_test.cpp +++ b/source/source_base/test/opt_tn_test.cpp @@ -1,5 +1,5 @@ #include "gtest/gtest.h" -#include "../opt_tn.hpp" +#include "../opt_tn.h" #include "../opt_dcsrch.h" #include "./opt_test_tools.h" @@ -17,7 +17,7 @@ class TN_test : public testing::Test double tol = 1e-5; int final_iter = 0; int flag = 0; - char *task = nullptr; + std::string task; double *p = nullptr; double *x = nullptr; @@ -25,14 +25,12 @@ class TN_test : public testing::Test { tn.set_para(1.); tn.allocate(tools.nx); - task = new char[60]; p = new double[tools.nx]; x = new double[tools.nx]; } void TearDown() { - delete[] task; delete[] p; delete[] x; } @@ -78,27 +76,27 @@ class TN_test : public testing::Test } for (int i = 0; i < 3; ++i) { temp_x[i] = x[i]; } - task[0] = 'S'; task[1] = 'T'; task[2] = 'A'; task[3] = 'R'; task[4] = 'T'; + task = "START"; while (true) { f = tools.func(temp_x, func_label); g = tools.dfuncdstp(temp_x, p, func_label); ds.dcSrch(f, g, step, task); - if (task[0] == 'F' && task[1] == 'G') + if (task.compare(0, 2, "FG") == 0) { for (int j = 0; j < 3; ++j) { temp_x[j] = x[j] + step * p[j]; } continue; } - else if (task[0] == 'C' && task[1] == 'O') + else if (task.compare(0, 2, "CO") == 0) { break; } - else if (task[0] == 'W' && task[1] == 'A') + else if (task.compare(0, 2, "WA") == 0) { break; } - else if (task[0] == 'E' && task[1] == 'R') + else if (task.compare(0, 2, "ER") == 0) { break; } diff --git a/source/source_base/test/test_opt_dcsrch.cpp b/source/source_base/test/test_opt_dcsrch.cpp new file mode 100644 index 00000000000..5fc786fbba2 --- /dev/null +++ b/source/source_base/test/test_opt_dcsrch.cpp @@ -0,0 +1,61 @@ +#include "../opt_dcsrch.h" +#include "gtest/gtest.h" + +TEST(OptDCsrch, ConvergenceAndRestart) +{ + ModuleBase::Opt_DCsrch search; + search.set_paras(); + std::string task; + for (int run = 0; run < 2; ++run) + { + double f = 1.0; + double g = -2.0; + double step = 1.0; + task = "START"; + search.dcSrch(f, g, step, task); + ASSERT_EQ(task, "FG"); + EXPECT_EQ(task.size(), 2u); + f = (step - 1.0) * (step - 1.0); + g = 2.0 * (step - 1.0); + search.dcSrch(f, g, step, task); + EXPECT_EQ(task, "CONVERGENCE"); + EXPECT_EQ(task.size(), 11u); + EXPECT_DOUBLE_EQ(step, 1.0); + } +} + +TEST(OptDCsrch, ErrorReplacesStatus) +{ + ModuleBase::Opt_DCsrch search; + search.set_paras(); + double f = 1.0; + double g = 1.0; + double step = 1.0; + std::string task = "START"; + search.dcSrch(f, g, step, task); + EXPECT_EQ(task, "ERROR: INITIAL G .GE. ZERO"); + EXPECT_EQ(task.size(), std::string("ERROR: INITIAL G .GE. ZERO").size()); + + task = "START"; + g = -2.0; + search.dcSrch(f, g, step, task); + EXPECT_EQ(task, "FG"); + EXPECT_EQ(task.size(), 2u); +} + +TEST(OptDCsrch, WarningAtMaximumStep) +{ + ModuleBase::Opt_DCsrch search; + search.set_paras(1e-4, 0.2, 1e-12, 0.0, 0.1); + double f = 1.0; + double g = -2.0; + double step = 0.1; + std::string task = "START"; + search.dcSrch(f, g, step, task); + ASSERT_EQ(task, "FG"); + f = (step - 1.0) * (step - 1.0); + g = 2.0 * (step - 1.0); + search.dcSrch(f, g, step, task); + EXPECT_EQ(task, "WARNING: STP = STPMAX"); + EXPECT_DOUBLE_EQ(step, 0.1); +} diff --git a/source/source_base/test_parallel/blacs_connector_test.cpp b/source/source_base/test_parallel/blacs_connector_test.cpp index 4e92bfd44d9..24f77ed30c6 100644 --- a/source/source_base/test_parallel/blacs_connector_test.cpp +++ b/source/source_base/test_parallel/blacs_connector_test.cpp @@ -68,6 +68,9 @@ TEST_F(BLACSTest, WorldGrid) // two BLACS grids should have difference context index EXPECT_NE(ictxt_row, ictxt_col); + + Cblacs_gridexit(ictxt_row); + Cblacs_gridexit(ictxt_col); } TEST_F(BLACSTest, SplitGrid) @@ -93,6 +96,8 @@ TEST_F(BLACSTest, SplitGrid) // verifies that the BLACS grid is created based on comm_sub instead of MPI_COMM_WORLD EXPECT_EQ(iprow, 0); EXPECT_EQ(ipcol, rank_sub); + + Cblacs_gridexit(ctxt_sub); } int main(int argc, char** argv) diff --git a/source/source_base/test_parallel/parallel_2d_test.cpp b/source/source_base/test_parallel/parallel_2d_test.cpp index 4dae05a9bf6..40e20fa1e2e 100644 --- a/source/source_base/test_parallel/parallel_2d_test.cpp +++ b/source/source_base/test_parallel/parallel_2d_test.cpp @@ -138,6 +138,57 @@ TEST_F(test_para2d, DescReuseCtxt) EXPECT_NE(p1.desc[1], p3.desc[1]); } } +TEST_F(test_para2d, SetWithOwnCtxt) +{ + // Reusing the context owned by the object itself (as in the block-size + // fallback in lcao_init_basis.cpp) must keep the grid alive and valid: + // the descriptor must remain usable and ownership must be preserved so + // that the destructor releases the grid exactly once. + for (auto nb: nbs) + { + Parallel_2D p2d; + const int gr = sizes[0].first; + const int gc = sizes[0].second; + p2d.init(gr, gc, nb, MPI_COMM_WORLD); + + const int ctxt = p2d.blacs_ctxt; + p2d.set(gr, gc, 1, p2d.blacs_ctxt); + + // the grid must survive: context unchanged, descriptor valid + EXPECT_EQ(p2d.blacs_ctxt, ctxt); + EXPECT_EQ(p2d.desc[0], 1); + EXPECT_EQ(p2d.desc[1], ctxt); + EXPECT_EQ(p2d.desc[2], gr); + EXPECT_EQ(p2d.desc[3], gc); + EXPECT_EQ(p2d.desc[4], 1); + EXPECT_EQ(p2d.get_block_size(), 1); + EXPECT_EQ(p2d.get_row_size() * p2d.get_col_size(), p2d.get_local_size()); + EXPECT_EQ(p2d.dim0 * p2d.dim1, dsize); + } +} + +TEST_F(test_para2d, SetSerialClearsBorrowedCtxt) +{ + // Switching to serial mode must drop any reference to a BLACS grid, + // including a borrowed one: release_blacs_grid() is a no-op for + // borrowers, so without an explicit reset the serial object would keep + // a handle that dangles once the owner destroys the grid. + Parallel_2D owner; + owner.init(sizes[0].first, sizes[0].second, 1, MPI_COMM_WORLD); + + Parallel_2D borrower; + borrower.set(sizes[0].first, sizes[0].second, 1, owner.blacs_ctxt); + EXPECT_EQ(borrower.blacs_ctxt, owner.blacs_ctxt); + + borrower.set_serial(3, 4); + EXPECT_EQ(borrower.blacs_ctxt, -1); + EXPECT_EQ(borrower.comm(), MPI_COMM_NULL); + + // the owner's grid must be unaffected by the borrower's mode switch + EXPECT_GE(owner.blacs_ctxt, 0); + EXPECT_NE(owner.comm(), MPI_COMM_NULL); +} + TEST_F(test_para2d, SerialLayoutInMpiBuild) { Parallel_2D p2d; diff --git a/source/source_base/tool_check.cpp b/source/source_base/tool_check.cpp index dab7dd5f213..1549fdbebeb 100644 --- a/source/source_base/tool_check.cpp +++ b/source/source_base/tool_check.cpp @@ -10,81 +10,81 @@ void CHECK_NAME(std::ifstream &ifs,const std::string &name_in,bool quit) ifs >> name; if ( name != name_in) { - if(quit) - { - //GlobalV::ofs_warning << "\n name = " <> v_in; - if( v!= v_in) - { - if(quit) - { - std::cout << "\n value = " << v_in; - std::cout << "\n should be = " << v; - WARNING_QUIT("CHECK_INT","Some parameter name is wrong!"); - } - else - { - std::cout <<"\n Can not match well: "<> v_in; + if( v!= v_in) + { + if(quit) + { + std::cout << "\n value = " << v_in; + std::cout << "\n should be = " << v; + WARNING_QUIT("CHECK_INT","Some parameter name is wrong!"); + } + else + { + std::cout <<"\n Can not match well: "<> v_in; - if( fabs(v - v_in) > tiny ) - { - if(quit) - { - std::cout << " read in value = " << v_in << std::endl; - std::cout << " the value should be = " << v << std::endl; - WARNING_QUIT("CHECK_DOUBLE","the name of parameter wrong!"); - } - else - { - std::cout <<" can not match well (1.0e-5): "<< v_in <<"(readin) "<> v_in; + if( fabs(v - v_in) > tiny ) + { + if(quit) + { + std::cout << " read in value = " << v_in << std::endl; + std::cout << " the value should be = " << v << std::endl; + WARNING_QUIT("CHECK_DOUBLE","the name of parameter wrong!"); + } + else + { + std::cout <<" can not match well (1.0e-5): "<< v_in <<"(readin) "<> v_in; - if( v_in != v ) - { - if(quit) - { - std::cout << " read in value = " << v_in << std::endl; - std::cout << " the value should be = " << v << std::endl; - WARNING_QUIT("CHECK_STRING","the name of parameter wrong!"); - } - else - { - std::cout <<" can not match well : "<> v_in; + if( v_in != v ) + { + if(quit) + { + std::cout << " read in value = " << v_in << std::endl; + std::cout << " the value should be = " << v << std::endl; + WARNING_QUIT("CHECK_STRING","the name of parameter wrong!"); + } + else + { + std::cout <<" can not match well : "< "< "< "< "< "< "< &vec, double ylmr[] ) { - //ModuleBase::timer::start("Ylm","get_ylm_real"); - //1e-9 is too large - const double cut0 = 1e-12; - // allocate space. - Ylm::nlm = Lmax * Lmax; - if(Lmax==1) - { - for(int i=0; i 1.0 - cut0) - { - cost = sgn(cost) * (1.0 - cut0); - // std::cout << "\n" << "cost = " << cost << std::endl; - } - sint = sqrt(1.0 - cost*cost); - - if(vec.x > cut0) - { - phi = std::atan( vec.y / vec.x ); - } - else if( vec.x < -cut0 ) - { - phi = std::atan( vec.y / vec.x ) + ModuleBase::PI; - } - else - { - phi = ModuleBase::PI_HALF * ((vec.y >= 0.0) ? 1.0 : -1.0); - } - - //=============================== - // NAME : p(Legendre Polynomials) - //=============================== - double p[20][20]; - assert(Lmax <= 20); - - int m=0; - double x1=0.0; // x2; - int lm = -1; // must initialized! - - for (int l=0; l(l)) ;//liaochen modify 2009-09-06 - if (l%2 == 1) - { - x1 = -x1; - } - p[l][l] = x1; - } // end if - - // Y_lm, m = 0 - ++lm; - ylmr[lm] = c*p[0][l]; - - for(m=1;m<=l;m++) - { - // Y_lm, m > 0 - const double same = c * sqrt - ( - static_cast( Ylm::Fact(l - m)) / - static_cast( Ylm::Fact(l + m)) - ) - *ModuleBase::SQRT2; - - ++lm; - ylmr[lm] = same * p[m][l] * cos(m * phi); - - // Y_lm, m < 0 - ++lm; - ylmr[lm] = same * p[m][l] * sin(m * phi); - } - }// end do - - //ModuleBase::timer::end("Ylm", "get_ylm_real"); - return; + //ModuleBase::timer::start("Ylm","get_ylm_real"); + //1e-9 is too large + const double cut0 = 1e-12; + // allocate space. + Ylm::nlm = Lmax * Lmax; + if(Lmax==1) + { + for(int i=0; i 1.0 - cut0) + { + cost = sgn(cost) * (1.0 - cut0); + // std::cout << "\n" << "cost = " << cost << std::endl; + } + sint = sqrt(1.0 - cost*cost); + + if(vec.x > cut0) + { + phi = std::atan( vec.y / vec.x ); + } + else if( vec.x < -cut0 ) + { + phi = std::atan( vec.y / vec.x ) + ModuleBase::PI; + } + else + { + phi = ModuleBase::PI_HALF * ((vec.y >= 0.0) ? 1.0 : -1.0); + } + + //=============================== + // NAME : p(Legendre Polynomials) + //=============================== + double p[20][20]; + assert(Lmax <= 20); + + int m=0; + double x1=0.0; // x2; + int lm = -1; // must initialized! + + for (int l=0; l(l)) ;//liaochen modify 2009-09-06 + if (l%2 == 1) + { + x1 = -x1; + } + p[l][l] = x1; + } // end if + + // Y_lm, m = 0 + ++lm; + ylmr[lm] = c*p[0][l]; + + for(m=1;m<=l;m++) + { + // Y_lm, m > 0 + const double same = c * sqrt + ( + static_cast( Ylm::Fact(l - m)) / + static_cast( Ylm::Fact(l + m)) + ) + *ModuleBase::SQRT2; + + ++lm; + ylmr[lm] = same * p[m][l] * cos(m * phi); + + // Y_lm, m < 0 + ++lm; + ylmr[lm] = same * p[m][l] * sin(m * phi); + } + }// end do + + //ModuleBase::timer::end("Ylm", "get_ylm_real"); + return; } void Ylm::get_ylm_real( const int &Lmax, const ModuleBase::Vector3 &vec, double ylmr[], double dylmdr[][3] ) { - //1e-9 is too large - const double cut0 = 1e-12; - // allocate space. - Ylm::nlm = Lmax * Lmax; - if(Lmax==1) - { - for(int i=0; i 1.0-cut0) - { - cost = sgn(cost) * (1.0 - cut0); - } - sint = sqrt(1.0 - cost*cost); - - if(vec.x > cut0) - { - phi = std::atan( vec.y / vec.x ); - } - else if( vec.x < -cut0 ) - { - phi = std::atan( vec.y / vec.x ) + ModuleBase::PI; - } - else - { - phi = ModuleBase::PI_HALF * ((vec.y >= 0.0) ? 1.0 : -1.0); - } - - //=============================== - // NAME : p(Legendre Polynomials) - //=============================== - double p[20][20]; - double dp[20][20]; - assert(Lmax <= 20); - - int m = 0; - int lm = -1; // must initialized! - for (int l=0; l(l)) ; - if (l%2 == 1) - { - p[l][l] = -p[l][l]; - } - } - - for(m=0; m <= l; m++) - { - if( m == l ) - { - dp[l][l] = l * cost * p[l][l] / sint; - } - else - { - dp[m][l] = (l * cost * p[m][l] - (l+m) * p[m][l-1]) / sint; - } - } - - // Y_lm, m = 0 - ++lm; - ylmr[lm] = c*p[0][l]; - - dylmdr[lm][0] = c * dp[0][l] * cost * cos(phi) / vnorm; - dylmdr[lm][1] = c * dp[0][l] * cost * sin(phi) / vnorm; - dylmdr[lm][2] = -c * dp[0][l] * sint / vnorm; - - for(m=1; m <= l; m++) - { - // Y_lm, m > 0 - const double same = c * sqrt - ( - static_cast( Ylm::Fact(l - m)) / - static_cast( Ylm::Fact(l + m)) - ) - *ModuleBase::SQRT2; - - ++lm; - ylmr[lm] = same * p[m][l] * cos(m * phi); - - dylmdr[lm][0] = same * dp[m][l] * cos(m * phi) * cost * cos(phi) / vnorm - + same * p[m][l] * m * sin(m * phi) * sin(phi) / sint / vnorm; - dylmdr[lm][1] = same * dp[m][l] * cos(m * phi) * cost * sin(phi) / vnorm - - same * p[m][l] * m * sin(m * phi) * cos(phi) / sint / vnorm; - dylmdr[lm][2] = -same * dp[m][l] * cos(m * phi) * sint / vnorm; - - // Y_lm, m < 0 - ++lm; - ylmr[lm] = same * p[m][l] * sin(m * phi); - - dylmdr[lm][0] = same * dp[m][l] * sin(m * phi) * cost * cos(phi) / vnorm - - same * p[m][l] * m * cos(m * phi) * sin(phi) / sint / vnorm; - dylmdr[lm][1] = same * dp[m][l] * sin(m * phi) * cost * sin(phi) / vnorm - + same * p[m][l] * m * cos(m * phi) * cos(phi) / sint / vnorm; - dylmdr[lm][2] = -same * dp[m][l] * sin(m * phi) * sint / vnorm; - } - }// end do - - return; + //1e-9 is too large + const double cut0 = 1e-12; + // allocate space. + Ylm::nlm = Lmax * Lmax; + if(Lmax==1) + { + for(int i=0; i 1.0-cut0) + { + cost = sgn(cost) * (1.0 - cut0); + } + sint = sqrt(1.0 - cost*cost); + + if(vec.x > cut0) + { + phi = std::atan( vec.y / vec.x ); + } + else if( vec.x < -cut0 ) + { + phi = std::atan( vec.y / vec.x ) + ModuleBase::PI; + } + else + { + phi = ModuleBase::PI_HALF * ((vec.y >= 0.0) ? 1.0 : -1.0); + } + + //=============================== + // NAME : p(Legendre Polynomials) + //=============================== + double p[20][20]; + double dp[20][20]; + assert(Lmax <= 20); + + int m = 0; + int lm = -1; // must initialized! + for (int l=0; l(l)) ; + if (l%2 == 1) + { + p[l][l] = -p[l][l]; + } + } + + for(m=0; m <= l; m++) + { + if( m == l ) + { + dp[l][l] = l * cost * p[l][l] / sint; + } + else + { + dp[m][l] = (l * cost * p[m][l] - (l+m) * p[m][l-1]) / sint; + } + } + + // Y_lm, m = 0 + ++lm; + ylmr[lm] = c*p[0][l]; + + dylmdr[lm][0] = c * dp[0][l] * cost * cos(phi) / vnorm; + dylmdr[lm][1] = c * dp[0][l] * cost * sin(phi) / vnorm; + dylmdr[lm][2] = -c * dp[0][l] * sint / vnorm; + + for(m=1; m <= l; m++) + { + // Y_lm, m > 0 + const double same = c * sqrt + ( + static_cast( Ylm::Fact(l - m)) / + static_cast( Ylm::Fact(l + m)) + ) + *ModuleBase::SQRT2; + + ++lm; + ylmr[lm] = same * p[m][l] * cos(m * phi); + + dylmdr[lm][0] = same * dp[m][l] * cos(m * phi) * cost * cos(phi) / vnorm + + same * p[m][l] * m * sin(m * phi) * sin(phi) / sint / vnorm; + dylmdr[lm][1] = same * dp[m][l] * cos(m * phi) * cost * sin(phi) / vnorm + - same * p[m][l] * m * sin(m * phi) * cos(phi) / sint / vnorm; + dylmdr[lm][2] = -same * dp[m][l] * cos(m * phi) * sint / vnorm; + + // Y_lm, m < 0 + ++lm; + ylmr[lm] = same * p[m][l] * sin(m * phi); + + dylmdr[lm][0] = same * dp[m][l] * sin(m * phi) * cost * cos(phi) / vnorm + - same * p[m][l] * m * cos(m * phi) * sin(phi) / sint / vnorm; + dylmdr[lm][1] = same * dp[m][l] * sin(m * phi) * cost * sin(phi) / vnorm + + same * p[m][l] * m * cos(m * phi) * cos(phi) / sint / vnorm; + dylmdr[lm][2] = -same * dp[m][l] * sin(m * phi) * sint / vnorm; + } + }// end do + + return; } /*************************** @@ -286,1739 +286,1739 @@ void Ylm::get_ylm_real( const int &Lmax, const ModuleBase::Vector3 &vec, * *************************/ void Ylm::rlylm ( - const int Lmax, //max momentum of l + 1 - const double x, - const double y, - const double z, - double rly[] + const int Lmax, //max momentum of l + 1 + const double x, + const double y, + const double z, + double rly[] ) { // ModuleBase::TITLE("Ylm","rlylm"); // ModuleBase::timer::start("Ylm","rlylm"); - int MaxL = Lmax - 1; + int MaxL = Lmax - 1; - assert(MaxL >= 0); + assert(MaxL >= 0); - //get xy_dependence - assert(MaxL <= 19); + //get xy_dependence + assert(MaxL <= 19); - double Am[20]; - double Bm[20]; + double Am[20]; + double Bm[20]; // ZEROS(Am, 20); // ZEROS(Bm, 20); - double x2, x3, x4, x5; - double y2, y3, y4, y5; - - x2 = x * x; - x3 = x2 * x; - x4 = x3 * x; - x5 = x4 * x; - - y2 = y * y; - y3 = y2 * y; - y4 = y3 * y; - y5 = y4 * y; - - //x-y dependence - //Am - //Bm - for(int im = 0; im < MaxL+1; im++) - { - if(im == 0) - { - Am[0] = 1.0; - Bm[0] = 0.0; - } - else if(im == 1) - { - Am[1] = x; - Bm[1] = y; - } - else if(im == 2) - { - Am[2] = x2- y2; - Bm[2] = 2.0 * x * y; - } - else if(im == 3) - { - Am[3] = x3 - 3.0 * x * y2; - Bm[3] = 3.0 * x2 * y - y3; - } - else if(im == 4) - { - Am[4] = x4 - 6.0 * x2 * y2 + y4; - Bm[4] = 4.0 * (x3 * y - x * y3); - } - else if(im == 5) - { - Am[5] = x5 - 10.0 * x3 * y2 + 5.0 * x * y4; - Bm[5] = 5.0 * x4 * y - 10.0 * x2 * y3 + y5; - } - else - { - for(int ip = 0; ip <= im; ip++) - { - double aux = Fact(im) / Fact(ip) / Fact(im - ip); - Am[im] += aux * pow(x, ip) * pow(y, im-ip) * cos( (im-ip) * ModuleBase::PI / 2.0 ); - Bm[im] += aux * pow(x, ip) * pow(y, im-ip) * sin( (im-ip) * ModuleBase::PI / 2.0 ); - } - } - } - - //z dependence - double zdep[20][20]; + double x2, x3, x4, x5; + double y2, y3, y4, y5; + + x2 = x * x; + x3 = x2 * x; + x4 = x3 * x; + x5 = x4 * x; + + y2 = y * y; + y3 = y2 * y; + y4 = y3 * y; + y5 = y4 * y; + + //x-y dependence + //Am + //Bm + for(int im = 0; im < MaxL+1; im++) + { + if(im == 0) + { + Am[0] = 1.0; + Bm[0] = 0.0; + } + else if(im == 1) + { + Am[1] = x; + Bm[1] = y; + } + else if(im == 2) + { + Am[2] = x2- y2; + Bm[2] = 2.0 * x * y; + } + else if(im == 3) + { + Am[3] = x3 - 3.0 * x * y2; + Bm[3] = 3.0 * x2 * y - y3; + } + else if(im == 4) + { + Am[4] = x4 - 6.0 * x2 * y2 + y4; + Bm[4] = 4.0 * (x3 * y - x * y3); + } + else if(im == 5) + { + Am[5] = x5 - 10.0 * x3 * y2 + 5.0 * x * y4; + Bm[5] = 5.0 * x4 * y - 10.0 * x2 * y3 + y5; + } + else + { + for(int ip = 0; ip <= im; ip++) + { + double aux = Fact(im) / Fact(ip) / Fact(im - ip); + Am[im] += aux * pow(x, ip) * pow(y, im-ip) * cos( (im-ip) * ModuleBase::PI / 2.0 ); + Bm[im] += aux * pow(x, ip) * pow(y, im-ip) * sin( (im-ip) * ModuleBase::PI / 2.0 ); + } + } + } + + //z dependence + double zdep[20][20]; // for(int il = 0; il < 20; il++) // { // ZEROS(zdep[il], 20); // } - double z2 = z * z; - double z3 = z2 * z; - double z4 = z3 * z; - //double z5 = z4 * z; - - double r = sqrt(x*x + y*y + z*z); - double r2 = r * r; - double r3 = r2 * r; - double r4 = r3 * r; - - for(int il = 0; il < MaxL+1; il++) - { - if(il == 0) - { - zdep[0][0] = 1.0; - } - else if(il == 1) - { - zdep[1][0] = z; - zdep[1][1] = 1.0; - } - else if(il == 2) - { - zdep[2][0] = 0.5 * (3.0 * z2 - r2); - zdep[2][1] = sqrt(3.0) * z; - zdep[2][2] = sqrt(3.0) * 0.5; - } - else if(il == 3) - { - zdep[3][0] = 2.5 * z3 - 1.5 * z * r2; - zdep[3][1] = 0.25 * sqrt(6.0) * (5.0 * z2 - r2); - zdep[3][2] = 0.5 * sqrt(15.0) * z; - zdep[3][3] = 0.25 * sqrt(10.0); - } - else if(il == 4) - { - zdep[4][0] = 0.125 * (35.0 * z4 - 30.0 * r2 * z2 + 3.0 * r4); - zdep[4][1] = sqrt(10.0) * 0.25 * z * (7.0 * z2 - 3.0 * r2); - zdep[4][2] = sqrt(5.0) * 0.25 * (7.0 * z2 - r2); - zdep[4][3] = sqrt(70.0) * 0.25 * z; - zdep[4][4] = sqrt(35.0) * 0.125; - } - else if(il == 5) - { - zdep[5][0] = 0.125 * z *( 63.0 * z4 - 70.0 * z2 * r2 + 15.0 * r4); - zdep[5][1] = 0.125 * sqrt(15.0) * (21.0 * z4 - 14.0 * z2 * r2 + r4); - zdep[5][2] = 0.25 * sqrt(105.0) * z * (3.0 * z2 - r2); - zdep[5][3] = 0.0625 * sqrt(70.0) * (9.0 * z2 - r2); - zdep[5][4] = 0.375 * sqrt(35.0) * z; - zdep[5][5] = 0.1875 * sqrt(14.0); - } - else - { - for(int im = 0; im <= il; im++) - { - int kmax = static_cast( (il - im) / 2 ); - for(int ik = 0; ik <= kmax; ik++) - { - int twok = 2 * ik; - - double gamma = 0.0; - double aux0, aux1, aux2, aux3; - - aux0 = pow(-1.0, ik) * pow(2.0, -il); - aux1 = Fact(il) / Fact(ik) / Fact(il-ik); - aux2 = Fact(2*il - twok) / Fact(il) / Fact(il - twok); - aux3 = Fact(il - twok) / Fact(il - twok - im); - - gamma = aux0 * aux1 * aux2 * aux3; - - assert(il - twok - im >= 0); - zdep[il][im] += pow(r, twok) * pow(z, il-twok-im) * gamma; - } - - if(im >= 1) - { - zdep[il][im] *= sqrt(2 * Fact(il - im) / Fact(il + im)); - } - } - } - } - - //calc - int ic = 0; - for(int il = 0; il <= MaxL; il++) - { - double fac = sqrt( (2.0 * il + 1.0) / ModuleBase::FOUR_PI ); - - //m=0 - rly[ic] = Am[0] * zdep[il][0] * fac; - - ic++; - - //m ! = 0 - for(int im = 1; im <= il; im++) - { - //m>0 - rly[ic] = Am[im] * zdep[il][im] * pow(-1.0, im) * fac; - - ic++; - - //m<0 - rly[ic] = Bm[im] * zdep[il][im] * pow(-1.0, im) * fac; - - ic++; - } - } + double z2 = z * z; + double z3 = z2 * z; + double z4 = z3 * z; + //double z5 = z4 * z; + + double r = sqrt(x*x + y*y + z*z); + double r2 = r * r; + double r3 = r2 * r; + double r4 = r3 * r; + + for(int il = 0; il < MaxL+1; il++) + { + if(il == 0) + { + zdep[0][0] = 1.0; + } + else if(il == 1) + { + zdep[1][0] = z; + zdep[1][1] = 1.0; + } + else if(il == 2) + { + zdep[2][0] = 0.5 * (3.0 * z2 - r2); + zdep[2][1] = sqrt(3.0) * z; + zdep[2][2] = sqrt(3.0) * 0.5; + } + else if(il == 3) + { + zdep[3][0] = 2.5 * z3 - 1.5 * z * r2; + zdep[3][1] = 0.25 * sqrt(6.0) * (5.0 * z2 - r2); + zdep[3][2] = 0.5 * sqrt(15.0) * z; + zdep[3][3] = 0.25 * sqrt(10.0); + } + else if(il == 4) + { + zdep[4][0] = 0.125 * (35.0 * z4 - 30.0 * r2 * z2 + 3.0 * r4); + zdep[4][1] = sqrt(10.0) * 0.25 * z * (7.0 * z2 - 3.0 * r2); + zdep[4][2] = sqrt(5.0) * 0.25 * (7.0 * z2 - r2); + zdep[4][3] = sqrt(70.0) * 0.25 * z; + zdep[4][4] = sqrt(35.0) * 0.125; + } + else if(il == 5) + { + zdep[5][0] = 0.125 * z *( 63.0 * z4 - 70.0 * z2 * r2 + 15.0 * r4); + zdep[5][1] = 0.125 * sqrt(15.0) * (21.0 * z4 - 14.0 * z2 * r2 + r4); + zdep[5][2] = 0.25 * sqrt(105.0) * z * (3.0 * z2 - r2); + zdep[5][3] = 0.0625 * sqrt(70.0) * (9.0 * z2 - r2); + zdep[5][4] = 0.375 * sqrt(35.0) * z; + zdep[5][5] = 0.1875 * sqrt(14.0); + } + else + { + for(int im = 0; im <= il; im++) + { + int kmax = static_cast( (il - im) / 2 ); + for(int ik = 0; ik <= kmax; ik++) + { + int twok = 2 * ik; + + double gamma = 0.0; + double aux0, aux1, aux2, aux3; + + aux0 = pow(-1.0, ik) * pow(2.0, -il); + aux1 = Fact(il) / Fact(ik) / Fact(il-ik); + aux2 = Fact(2*il - twok) / Fact(il) / Fact(il - twok); + aux3 = Fact(il - twok) / Fact(il - twok - im); + + gamma = aux0 * aux1 * aux2 * aux3; + + assert(il - twok - im >= 0); + zdep[il][im] += pow(r, twok) * pow(z, il-twok-im) * gamma; + } + + if(im >= 1) + { + zdep[il][im] *= sqrt(2 * Fact(il - im) / Fact(il + im)); + } + } + } + } + + //calc + int ic = 0; + for(int il = 0; il <= MaxL; il++) + { + double fac = sqrt( (2.0 * il + 1.0) / ModuleBase::FOUR_PI ); + + //m=0 + rly[ic] = Am[0] * zdep[il][0] * fac; + + ic++; + + //m ! = 0 + for(int im = 1; im <= il; im++) + { + //m>0 + rly[ic] = Am[im] * zdep[il][im] * pow(-1.0, im) * fac; + + ic++; + + //m<0 + rly[ic] = Bm[im] * zdep[il][im] * pow(-1.0, im) * fac; + + ic++; + } + } // ModuleBase::timer::end("Ylm", "rlylm"); - return; + return; } //return ylm, not rlylm void Ylm::sph_harm ( - const int Lmax, //max momentum of l - const double xdr, - const double ydr, - const double zdr, - std::vector &rly + const int Lmax, //max momentum of l + const double xdr, + const double ydr, + const double zdr, + std::vector &rly ) { - rly.resize( (Lmax+1)*(Lmax+1) ); - - //begin calculation - /*************************** - L = 0 - ***************************/ - rly[0] = ylmcoef[0]; //l=0, m=0 - if (Lmax == 0) return; - - /*************************** - L = 1 - ***************************/ - rly[1] = ylmcoef[1]*zdr; //l=1, m=0 - rly[2] = -ylmcoef[1]*xdr; //l=1, m=1 - rly[3] = -ylmcoef[1]*ydr; //l=1, m=-1 - if (Lmax == 1) return; - - /*************************** - L = 2 - ***************************/ - rly[4] = ylmcoef[2]*zdr*rly[1]-ylmcoef[3]*rly[0];//l=2, m=0 - - double tmp0 = ylmcoef[4]*zdr; - rly[5] = tmp0*rly[2];//l=2,m=1 - rly[6] = tmp0*rly[3];//l=2,m=-1 - - double tmp2 = ylmcoef[4]*xdr; - rly[7]= ylmcoef[5]*rly[4]-ylmcoef[6]*rly[0] - tmp2*rly[2];//l=2,m=2 - rly[8] = -tmp2*rly[3]; + rly.resize( (Lmax+1)*(Lmax+1) ); + + //begin calculation + /*************************** + L = 0 + ***************************/ + rly[0] = ylmcoef[0]; //l=0, m=0 + if (Lmax == 0) return; + + /*************************** + L = 1 + ***************************/ + rly[1] = ylmcoef[1]*zdr; //l=1, m=0 + rly[2] = -ylmcoef[1]*xdr; //l=1, m=1 + rly[3] = -ylmcoef[1]*ydr; //l=1, m=-1 + if (Lmax == 1) return; + + /*************************** + L = 2 + ***************************/ + rly[4] = ylmcoef[2]*zdr*rly[1]-ylmcoef[3]*rly[0];//l=2, m=0 + + double tmp0 = ylmcoef[4]*zdr; + rly[5] = tmp0*rly[2];//l=2,m=1 + rly[6] = tmp0*rly[3];//l=2,m=-1 + + double tmp2 = ylmcoef[4]*xdr; + rly[7]= ylmcoef[5]*rly[4]-ylmcoef[6]*rly[0] - tmp2*rly[2];//l=2,m=2 + rly[8] = -tmp2*rly[3]; // rly[8] = tmp1+tmp2*rly[3];//l=2,m=-2 - if (Lmax == 2) return; - - /*************************** - L = 3 - ***************************/ - rly[9] = ylmcoef[7]*zdr*rly[4]-ylmcoef[8]*rly[1]; //l=3, m=0 - - double tmp3 = ylmcoef[9]*zdr; - rly[10] = tmp3*rly[5]-ylmcoef[10]*rly[2];//l=3,m=1 - rly[11] = tmp3*rly[6]-ylmcoef[10]*rly[3];//l=3,m=-1 - - double tmp4 = ylmcoef[11]*zdr; - rly[12] = tmp4*rly[7];//l=3,m=2 - rly[13] = tmp4*rly[8];//l=3,m=-2 - - double tmp5 = ylmcoef[14]*xdr; - rly[14] = ylmcoef[12]*rly[10]-ylmcoef[13]*rly[2]-tmp5*rly[7];//l=3,m=3 - rly[15] = ylmcoef[12]*rly[11]-ylmcoef[13]*rly[3]-tmp5*rly[8];//l=3,m=-3 - if (Lmax == 3) return; - - /*************************** - L = 4 - ***************************/ - rly[16] = ylmcoef[15]*zdr*rly[9]-ylmcoef[16]*rly[4];//l=4,m=0 - - double tmp6 = ylmcoef[17]*zdr; - rly[17] = tmp6*rly[10]-ylmcoef[18]*rly[5];//l=4,m=1 - rly[18] = tmp6*rly[11]-ylmcoef[18]*rly[6];//l=4,m=-1 - - double tmp7 = ylmcoef[19]*zdr; - rly[19] = tmp7*rly[12]-ylmcoef[20]*rly[7];//l=4,m=2 - rly[20] = tmp7*rly[13]-ylmcoef[20]*rly[8];//l=4,m=-2 - - double tmp8 = 3.0*zdr; - rly[21] = tmp8*rly[14];//l=4,m=3 - rly[22] = tmp8*rly[15];//l=4,m=-3 - - double tmp9 = ylmcoef[23]*xdr; - rly[23] = ylmcoef[21]*rly[19]-ylmcoef[22]*rly[7]-tmp9*rly[14];//l=4,m=4 - rly[24] = ylmcoef[21]*rly[20]-ylmcoef[22]*rly[8]-tmp9*rly[15];//l=4,m=-4 - if (Lmax == 4) return; - - /*************************** - L = 5 - ***************************/ - rly[25] = ylmcoef[24]*zdr*rly[16]-ylmcoef[25]*rly[9];//l=5,m=0 - - double tmp10 = ylmcoef[26]*zdr; - rly[26] = tmp10*rly[17]-ylmcoef[27]*rly[10];//l=5,m=1 - rly[27] = tmp10*rly[18]-ylmcoef[27]*rly[11];//l=5,m=-1 - - double tmp11 = ylmcoef[28]*zdr; - rly[28] = tmp11*rly[19]-ylmcoef[29]*rly[12];//l=5,m=2 - rly[29] = tmp11*rly[20]-ylmcoef[29]*rly[13];//l=5,m=-2 - - double tmp12 = ylmcoef[30]*zdr; - rly[30] = tmp12*rly[21]-ylmcoef[31]*rly[14];//l=5,m=3 - rly[31] = tmp12*rly[22]-ylmcoef[31]*rly[15];//l=5,m=-3 - - double tmp13 = ylmcoef[32]*zdr; - rly[32] = tmp13*rly[23];//l=5,m=4 - rly[33] = tmp13*rly[24];//l=5,m=-4 - - double tmp14 = ylmcoef[35]*xdr; - rly[34] = ylmcoef[33]*rly[30]-ylmcoef[34]*rly[14]-tmp14*rly[23];//l=5,m=5 - rly[35] = ylmcoef[33]*rly[31]-ylmcoef[34]*rly[15]-tmp14*rly[24];//l=5,m=-5 - if (Lmax == 5) return; - - //if Lmax > 5 - for (int il = 6; il <= Lmax; il++) - { - int istart = il*il; - int istart1 = (il-1)*(il-1); - int istart2 = (il-2)*(il-2); - - double fac2 = sqrt(4.0*istart-1.0); - double fac4 = sqrt(4.0*istart1-1.0); - - for (int im = 0; im < 2*il-1; im++) - { - int imm = (im+1)/2; + if (Lmax == 2) return; + + /*************************** + L = 3 + ***************************/ + rly[9] = ylmcoef[7]*zdr*rly[4]-ylmcoef[8]*rly[1]; //l=3, m=0 + + double tmp3 = ylmcoef[9]*zdr; + rly[10] = tmp3*rly[5]-ylmcoef[10]*rly[2];//l=3,m=1 + rly[11] = tmp3*rly[6]-ylmcoef[10]*rly[3];//l=3,m=-1 + + double tmp4 = ylmcoef[11]*zdr; + rly[12] = tmp4*rly[7];//l=3,m=2 + rly[13] = tmp4*rly[8];//l=3,m=-2 + + double tmp5 = ylmcoef[14]*xdr; + rly[14] = ylmcoef[12]*rly[10]-ylmcoef[13]*rly[2]-tmp5*rly[7];//l=3,m=3 + rly[15] = ylmcoef[12]*rly[11]-ylmcoef[13]*rly[3]-tmp5*rly[8];//l=3,m=-3 + if (Lmax == 3) return; + + /*************************** + L = 4 + ***************************/ + rly[16] = ylmcoef[15]*zdr*rly[9]-ylmcoef[16]*rly[4];//l=4,m=0 + + double tmp6 = ylmcoef[17]*zdr; + rly[17] = tmp6*rly[10]-ylmcoef[18]*rly[5];//l=4,m=1 + rly[18] = tmp6*rly[11]-ylmcoef[18]*rly[6];//l=4,m=-1 + + double tmp7 = ylmcoef[19]*zdr; + rly[19] = tmp7*rly[12]-ylmcoef[20]*rly[7];//l=4,m=2 + rly[20] = tmp7*rly[13]-ylmcoef[20]*rly[8];//l=4,m=-2 + + double tmp8 = 3.0*zdr; + rly[21] = tmp8*rly[14];//l=4,m=3 + rly[22] = tmp8*rly[15];//l=4,m=-3 + + double tmp9 = ylmcoef[23]*xdr; + rly[23] = ylmcoef[21]*rly[19]-ylmcoef[22]*rly[7]-tmp9*rly[14];//l=4,m=4 + rly[24] = ylmcoef[21]*rly[20]-ylmcoef[22]*rly[8]-tmp9*rly[15];//l=4,m=-4 + if (Lmax == 4) return; + + /*************************** + L = 5 + ***************************/ + rly[25] = ylmcoef[24]*zdr*rly[16]-ylmcoef[25]*rly[9];//l=5,m=0 + + double tmp10 = ylmcoef[26]*zdr; + rly[26] = tmp10*rly[17]-ylmcoef[27]*rly[10];//l=5,m=1 + rly[27] = tmp10*rly[18]-ylmcoef[27]*rly[11];//l=5,m=-1 + + double tmp11 = ylmcoef[28]*zdr; + rly[28] = tmp11*rly[19]-ylmcoef[29]*rly[12];//l=5,m=2 + rly[29] = tmp11*rly[20]-ylmcoef[29]*rly[13];//l=5,m=-2 + + double tmp12 = ylmcoef[30]*zdr; + rly[30] = tmp12*rly[21]-ylmcoef[31]*rly[14];//l=5,m=3 + rly[31] = tmp12*rly[22]-ylmcoef[31]*rly[15];//l=5,m=-3 + + double tmp13 = ylmcoef[32]*zdr; + rly[32] = tmp13*rly[23];//l=5,m=4 + rly[33] = tmp13*rly[24];//l=5,m=-4 + + double tmp14 = ylmcoef[35]*xdr; + rly[34] = ylmcoef[33]*rly[30]-ylmcoef[34]*rly[14]-tmp14*rly[23];//l=5,m=5 + rly[35] = ylmcoef[33]*rly[31]-ylmcoef[34]*rly[15]-tmp14*rly[24];//l=5,m=-5 + if (Lmax == 5) return; + + //if Lmax > 5 + for (int il = 6; il <= Lmax; il++) + { + int istart = il*il; + int istart1 = (il-1)*(il-1); + int istart2 = (il-2)*(il-2); + + double fac2 = sqrt(4.0*istart-1.0); + double fac4 = sqrt(4.0*istart1-1.0); + + for (int im = 0; im < 2*il-1; im++) + { + int imm = (im+1)/2; // if (im % 2 == 0) imm *= -1; - rly[istart+im] = fac2/sqrt((double)istart-imm*imm)* - (zdr*rly[istart1+im] - sqrt((double)istart1-imm*imm)/fac4*rly[istart2+im]); - } + rly[istart+im] = fac2/sqrt((double)istart-imm*imm)* + (zdr*rly[istart1+im] - sqrt((double)istart1-imm*imm)/fac4*rly[istart2+im]); + } - double bl1 = sqrt(2.0*il/(2.0*il+1.0)); - double bl2 = sqrt((2.0*il-2.0)/(2.0*il-1.0)); - double bl3 = sqrt(2.0)/fac2; + double bl1 = sqrt(2.0*il/(2.0*il+1.0)); + double bl2 = sqrt((2.0*il-2.0)/(2.0*il-1.0)); + double bl3 = sqrt(2.0)/fac2; - rly[istart+2*il-1] = (bl3*rly[istart+2*il-5]-bl2*rly[istart2+2*il-5]-2.0*xdr*rly[istart1+2*il-3]) / bl1; - rly[istart+2*il] = (bl3*rly[istart+2*il-4]-bl2*rly[istart2+2*il-4]-2.0*xdr*rly[istart1+2*il-2]) / bl1; - } + rly[istart+2*il-1] = (bl3*rly[istart+2*il-5]-bl2*rly[istart2+2*il-5]-2.0*xdr*rly[istart1+2*il-3]) / bl1; + rly[istart+2*il] = (bl3*rly[istart+2*il-4]-bl2*rly[istart2+2*il-4]-2.0*xdr*rly[istart1+2*il-2]) / bl1; + } - return; + return; } // Peize Lin change rly 2016-08-26 void Ylm::rl_sph_harm ( - const int Lmax, //max momentum of L - const double x, - const double y, - const double z, - std::vector& rly + const int Lmax, //max momentum of L + const double x, + const double y, + const double z, + std::vector& rly ) { - rly.resize( (Lmax+1)*(Lmax+1) ); - - double radius2 = x*x+y*y+z*z; - - //begin calculation - /*************************** - L = 0 - ***************************/ - rly[0] = ylmcoef[0]; //l=0, m=0 - if (Lmax == 0) return; - - /*************************** - L = 1 - ***************************/ - rly[1] = ylmcoef[1]*z; //l=1, m=0 - rly[2] = -ylmcoef[1]*x; //l=1, m=1 - rly[3] = -ylmcoef[1]*y; //l=1, m=-1 - if (Lmax == 1) return; - - /*************************** - L = 2 - ***************************/ - rly[4] = ylmcoef[2]*z*rly[1]-ylmcoef[3]*rly[0]*radius2;//l=2, m=0 - - double tmp0 = ylmcoef[4]*z; - rly[5] = tmp0*rly[2];//l=2,m=1 - rly[6] = tmp0*rly[3];//l=2,m=-1 - - double tmp2 = ylmcoef[4]*x; - rly[7]= ylmcoef[5]*rly[4]-ylmcoef[6]*rly[0]*radius2 - tmp2*rly[2];//l=2,m=2 - rly[8] = -tmp2*rly[3]; + rly.resize( (Lmax+1)*(Lmax+1) ); + + double radius2 = x*x+y*y+z*z; + + //begin calculation + /*************************** + L = 0 + ***************************/ + rly[0] = ylmcoef[0]; //l=0, m=0 + if (Lmax == 0) return; + + /*************************** + L = 1 + ***************************/ + rly[1] = ylmcoef[1]*z; //l=1, m=0 + rly[2] = -ylmcoef[1]*x; //l=1, m=1 + rly[3] = -ylmcoef[1]*y; //l=1, m=-1 + if (Lmax == 1) return; + + /*************************** + L = 2 + ***************************/ + rly[4] = ylmcoef[2]*z*rly[1]-ylmcoef[3]*rly[0]*radius2;//l=2, m=0 + + double tmp0 = ylmcoef[4]*z; + rly[5] = tmp0*rly[2];//l=2,m=1 + rly[6] = tmp0*rly[3];//l=2,m=-1 + + double tmp2 = ylmcoef[4]*x; + rly[7]= ylmcoef[5]*rly[4]-ylmcoef[6]*rly[0]*radius2 - tmp2*rly[2];//l=2,m=2 + rly[8] = -tmp2*rly[3]; // rly[8] = tmp1+tmp2*rly[3];//l=2,m=-2 - if (Lmax == 2) return; - - /*************************** - L = 3 - ***************************/ - rly[9] = ylmcoef[7]*z*rly[4]-ylmcoef[8]*rly[1]*radius2; //l=3, m=0 - - double tmp3 = ylmcoef[9]*z; - rly[10] = tmp3*rly[5]-ylmcoef[10]*rly[2]*radius2;//l=3,m=1 - rly[11] = tmp3*rly[6]-ylmcoef[10]*rly[3]*radius2;//l=3,m=-1 - - double tmp4 = ylmcoef[11]*z; - rly[12] = tmp4*rly[7];//l=3,m=2 - rly[13] = tmp4*rly[8];//l=3,m=-2 - - double tmp5 = ylmcoef[14]*x; - rly[14] = ylmcoef[12]*rly[10]-ylmcoef[13]*rly[2]*radius2-tmp5*rly[7];//l=3,m=3 - rly[15] = ylmcoef[12]*rly[11]-ylmcoef[13]*rly[3]*radius2-tmp5*rly[8];//l=3,m=-3 - if (Lmax == 3) return; - - /*************************** - L = 4 - ***************************/ - rly[16] = ylmcoef[15]*z*rly[9]-ylmcoef[16]*rly[4]*radius2;//l=4,m=0 - - double tmp6 = ylmcoef[17]*z; - rly[17] = tmp6*rly[10]-ylmcoef[18]*rly[5]*radius2;//l=4,m=1 - rly[18] = tmp6*rly[11]-ylmcoef[18]*rly[6]*radius2;//l=4,m=-1 - - double tmp7 = ylmcoef[19]*z; - rly[19] = tmp7*rly[12]-ylmcoef[20]*rly[7]*radius2;//l=4,m=2 - rly[20] = tmp7*rly[13]-ylmcoef[20]*rly[8]*radius2;//l=4,m=-2 - - double tmp8 = 3.0*z; - rly[21] = tmp8*rly[14];//l=4,m=3 - rly[22] = tmp8*rly[15];//l=4,m=-3 - - double tmp9 = ylmcoef[23]*x; - rly[23] = ylmcoef[21]*rly[19]-ylmcoef[22]*rly[7]*radius2-tmp9*rly[14];//l=4,m=4 - rly[24] = ylmcoef[21]*rly[20]-ylmcoef[22]*rly[8]*radius2-tmp9*rly[15];//l=4,m=-4 - if (Lmax == 4) return; - - /*************************** - L = 5 - ***************************/ - rly[25] = ylmcoef[24]*z*rly[16]-ylmcoef[25]*rly[9]*radius2;//l=5,m=0 - - double tmp10 = ylmcoef[26]*z; - rly[26] = tmp10*rly[17]-ylmcoef[27]*rly[10]*radius2;//l=5,m=1 - rly[27] = tmp10*rly[18]-ylmcoef[27]*rly[11]*radius2;//l=5,m=-1 - - double tmp11 = ylmcoef[28]*z; - rly[28] = tmp11*rly[19]-ylmcoef[29]*rly[12]*radius2;//l=5,m=2 - rly[29] = tmp11*rly[20]-ylmcoef[29]*rly[13]*radius2;//l=5,m=-2 - - double tmp12 = ylmcoef[30]*z; - rly[30] = tmp12*rly[21]-ylmcoef[31]*rly[14]*radius2;//l=5,m=3 - rly[31] = tmp12*rly[22]-ylmcoef[31]*rly[15]*radius2;//l=5,m=-3 - - double tmp13 = ylmcoef[32]*z; - rly[32] = tmp13*rly[23];//l=5,m=4 - rly[33] = tmp13*rly[24];//l=5,m=-4 - - double tmp14 = ylmcoef[35]*x; - rly[34] = ylmcoef[33]*rly[30]-ylmcoef[34]*rly[14]*radius2-tmp14*rly[23];//l=5,m=5 - rly[35] = ylmcoef[33]*rly[31]-ylmcoef[34]*rly[15]*radius2-tmp14*rly[24];//l=5,m=-5 - if (Lmax == 5) return; - - //if Lmax > 5 - for (int il = 6; il <= Lmax; il++) - { - int istart = il*il; - int istart1 = (il-1)*(il-1); - int istart2 = (il-2)*(il-2); - - double fac2 = sqrt(4.0*istart-1); - double fac4 = sqrt(4.0*istart1-1); - - for (int im = 0; im < 2*il-1; im++) - { - int imm = (im+1)/2; + if (Lmax == 2) return; + + /*************************** + L = 3 + ***************************/ + rly[9] = ylmcoef[7]*z*rly[4]-ylmcoef[8]*rly[1]*radius2; //l=3, m=0 + + double tmp3 = ylmcoef[9]*z; + rly[10] = tmp3*rly[5]-ylmcoef[10]*rly[2]*radius2;//l=3,m=1 + rly[11] = tmp3*rly[6]-ylmcoef[10]*rly[3]*radius2;//l=3,m=-1 + + double tmp4 = ylmcoef[11]*z; + rly[12] = tmp4*rly[7];//l=3,m=2 + rly[13] = tmp4*rly[8];//l=3,m=-2 + + double tmp5 = ylmcoef[14]*x; + rly[14] = ylmcoef[12]*rly[10]-ylmcoef[13]*rly[2]*radius2-tmp5*rly[7];//l=3,m=3 + rly[15] = ylmcoef[12]*rly[11]-ylmcoef[13]*rly[3]*radius2-tmp5*rly[8];//l=3,m=-3 + if (Lmax == 3) return; + + /*************************** + L = 4 + ***************************/ + rly[16] = ylmcoef[15]*z*rly[9]-ylmcoef[16]*rly[4]*radius2;//l=4,m=0 + + double tmp6 = ylmcoef[17]*z; + rly[17] = tmp6*rly[10]-ylmcoef[18]*rly[5]*radius2;//l=4,m=1 + rly[18] = tmp6*rly[11]-ylmcoef[18]*rly[6]*radius2;//l=4,m=-1 + + double tmp7 = ylmcoef[19]*z; + rly[19] = tmp7*rly[12]-ylmcoef[20]*rly[7]*radius2;//l=4,m=2 + rly[20] = tmp7*rly[13]-ylmcoef[20]*rly[8]*radius2;//l=4,m=-2 + + double tmp8 = 3.0*z; + rly[21] = tmp8*rly[14];//l=4,m=3 + rly[22] = tmp8*rly[15];//l=4,m=-3 + + double tmp9 = ylmcoef[23]*x; + rly[23] = ylmcoef[21]*rly[19]-ylmcoef[22]*rly[7]*radius2-tmp9*rly[14];//l=4,m=4 + rly[24] = ylmcoef[21]*rly[20]-ylmcoef[22]*rly[8]*radius2-tmp9*rly[15];//l=4,m=-4 + if (Lmax == 4) return; + + /*************************** + L = 5 + ***************************/ + rly[25] = ylmcoef[24]*z*rly[16]-ylmcoef[25]*rly[9]*radius2;//l=5,m=0 + + double tmp10 = ylmcoef[26]*z; + rly[26] = tmp10*rly[17]-ylmcoef[27]*rly[10]*radius2;//l=5,m=1 + rly[27] = tmp10*rly[18]-ylmcoef[27]*rly[11]*radius2;//l=5,m=-1 + + double tmp11 = ylmcoef[28]*z; + rly[28] = tmp11*rly[19]-ylmcoef[29]*rly[12]*radius2;//l=5,m=2 + rly[29] = tmp11*rly[20]-ylmcoef[29]*rly[13]*radius2;//l=5,m=-2 + + double tmp12 = ylmcoef[30]*z; + rly[30] = tmp12*rly[21]-ylmcoef[31]*rly[14]*radius2;//l=5,m=3 + rly[31] = tmp12*rly[22]-ylmcoef[31]*rly[15]*radius2;//l=5,m=-3 + + double tmp13 = ylmcoef[32]*z; + rly[32] = tmp13*rly[23];//l=5,m=4 + rly[33] = tmp13*rly[24];//l=5,m=-4 + + double tmp14 = ylmcoef[35]*x; + rly[34] = ylmcoef[33]*rly[30]-ylmcoef[34]*rly[14]*radius2-tmp14*rly[23];//l=5,m=5 + rly[35] = ylmcoef[33]*rly[31]-ylmcoef[34]*rly[15]*radius2-tmp14*rly[24];//l=5,m=-5 + if (Lmax == 5) return; + + //if Lmax > 5 + for (int il = 6; il <= Lmax; il++) + { + int istart = il*il; + int istart1 = (il-1)*(il-1); + int istart2 = (il-2)*(il-2); + + double fac2 = sqrt(4.0*istart-1); + double fac4 = sqrt(4.0*istart1-1); + + for (int im = 0; im < 2*il-1; im++) + { + int imm = (im+1)/2; // if (im % 2 == 0) imm *= -1; - rly[istart+im] = fac2/sqrt((double)istart-imm*imm)* - (z*rly[istart1+im] - sqrt((double)istart1-imm*imm)/fac4*rly[istart2+im]*radius2); - } + rly[istart+im] = fac2/sqrt((double)istart-imm*imm)* + (z*rly[istart1+im] - sqrt((double)istart1-imm*imm)/fac4*rly[istart2+im]*radius2); + } - double bl1 = sqrt(2.0*il/(2.0*il+1.0)); - double bl2 = sqrt((2.0*il-2.0)/(2.0*il-1.0)); - double bl3 = sqrt(2.0)/fac2; + double bl1 = sqrt(2.0*il/(2.0*il+1.0)); + double bl2 = sqrt((2.0*il-2.0)/(2.0*il-1.0)); + double bl3 = sqrt(2.0)/fac2; - rly[istart+2*il-1] = (bl3*rly[istart+2*il-5]-bl2*rly[istart2+2*il-5]*radius2-2.0*x*rly[istart1+2*il-3]) / bl1; - rly[istart+2*il] = (bl3*rly[istart+2*il-4]-bl2*rly[istart2+2*il-4]*radius2-2.0*x*rly[istart1+2*il-2]) / bl1; - } + rly[istart+2*il-1] = (bl3*rly[istart+2*il-5]-bl2*rly[istart2+2*il-5]*radius2-2.0*x*rly[istart1+2*il-3]) / bl1; + rly[istart+2*il] = (bl3*rly[istart+2*il-4]-bl2*rly[istart2+2*il-4]*radius2-2.0*x*rly[istart1+2*il-2]) / bl1; + } - return; + return; } void Ylm::grad_rl_sph_harm ( - const int Lmax, //max momentum of L - const double x, - const double y, - const double z, - double* rly, - double* grly_flat + const int Lmax, //max momentum of L + const double x, + const double y, + const double z, + double* rly, + double* grly_flat ) { - // Alias the flat buffer as a pointer-to-array-of-3-doubles so the body - // below can continue to use the natural grly[lm][xyz] indexing without - // any performance penalty — the memory layout is unchanged. - double (*grly)[3] = reinterpret_cast(grly_flat); - double radius2 = x*x+y*y+z*z; - double tx = 2.0*x; - double ty = 2.0*y; - double tz = 2.0*z; - - //begin calculation - /*************************** - L = 0 - ***************************/ - rly[0] = ylmcoef[0]; //l=0, m=0 - grly[0][0] = grly[0][1] = grly[0][2] = 0.0; - if (Lmax == 0) return; - - /*************************** - L = 1 - ***************************/ - rly[1] = ylmcoef[1]*z; //l=1, m=0 - grly[1][0] = grly[1][1] = 0.0; - grly[1][2] = ylmcoef[1]; - - rly[2] = -ylmcoef[1]*x; //l=1, m=1 - grly[2][1] = grly[2][2] = 0.0; - grly[2][0] = -ylmcoef[1]; - - rly[3] = -ylmcoef[1]*y; //l=1, m=-1 - grly[3][0] = grly[3][2] = 0.0; - grly[3][1] = -ylmcoef[1]; - - if (Lmax == 1) return; - - /*************************** - L = 2 - ***************************/ - rly[4] = ylmcoef[2]*z*rly[1]-ylmcoef[3]*rly[0]*radius2;//l=2, m=0 - grly[4][0] = ylmcoef[2]*z*grly[1][0]-ylmcoef[3]*(grly[0][0]*radius2+rly[0]*tx);//l=2, m=0 - grly[4][1] = ylmcoef[2]*z*grly[1][1]-ylmcoef[3]*(grly[0][1]*radius2+rly[0]*ty);//l=2, m=0 - grly[4][2] = ylmcoef[2]*(z*grly[1][2]+rly[1])-ylmcoef[3]*(grly[0][2]*radius2+rly[0]*tz);//l=2, m=0 - - - double tmp0 = ylmcoef[4]*z; - rly[5] = tmp0*rly[2];//l=2,m=1 - grly[5][0] = tmp0*grly[2][0]; - grly[5][1] = tmp0*grly[2][1]; - grly[5][2] = ylmcoef[4]*(rly[2]+z*grly[2][2]); - - rly[6] = tmp0*rly[3];//l=2,m=-1 - grly[6][0] = tmp0*grly[3][0]; - grly[6][1] = tmp0*grly[3][1]; - grly[6][2] = ylmcoef[4]*(rly[3]+z*grly[3][2]); - - double tmp2 = ylmcoef[4]*x; - rly[7]= ylmcoef[5]*rly[4]-ylmcoef[6]*rly[0]*radius2 - tmp2*rly[2];//l=2,m=2 - grly[7][0] = ylmcoef[5]*grly[4][0]-ylmcoef[6]*(rly[0]*tx+grly[0][0]*radius2)-ylmcoef[4]*(x*grly[2][0]+rly[2]); + // Alias the flat buffer as a pointer-to-array-of-3-doubles so the body + // below can continue to use the natural grly[lm][xyz] indexing without + // any performance penalty — the memory layout is unchanged. + double (*grly)[3] = reinterpret_cast(grly_flat); + double radius2 = x*x+y*y+z*z; + double tx = 2.0*x; + double ty = 2.0*y; + double tz = 2.0*z; + + //begin calculation + /*************************** + L = 0 + ***************************/ + rly[0] = ylmcoef[0]; //l=0, m=0 + grly[0][0] = grly[0][1] = grly[0][2] = 0.0; + if (Lmax == 0) return; + + /*************************** + L = 1 + ***************************/ + rly[1] = ylmcoef[1]*z; //l=1, m=0 + grly[1][0] = grly[1][1] = 0.0; + grly[1][2] = ylmcoef[1]; + + rly[2] = -ylmcoef[1]*x; //l=1, m=1 + grly[2][1] = grly[2][2] = 0.0; + grly[2][0] = -ylmcoef[1]; + + rly[3] = -ylmcoef[1]*y; //l=1, m=-1 + grly[3][0] = grly[3][2] = 0.0; + grly[3][1] = -ylmcoef[1]; + + if (Lmax == 1) return; + + /*************************** + L = 2 + ***************************/ + rly[4] = ylmcoef[2]*z*rly[1]-ylmcoef[3]*rly[0]*radius2;//l=2, m=0 + grly[4][0] = ylmcoef[2]*z*grly[1][0]-ylmcoef[3]*(grly[0][0]*radius2+rly[0]*tx);//l=2, m=0 + grly[4][1] = ylmcoef[2]*z*grly[1][1]-ylmcoef[3]*(grly[0][1]*radius2+rly[0]*ty);//l=2, m=0 + grly[4][2] = ylmcoef[2]*(z*grly[1][2]+rly[1])-ylmcoef[3]*(grly[0][2]*radius2+rly[0]*tz);//l=2, m=0 + + + double tmp0 = ylmcoef[4]*z; + rly[5] = tmp0*rly[2];//l=2,m=1 + grly[5][0] = tmp0*grly[2][0]; + grly[5][1] = tmp0*grly[2][1]; + grly[5][2] = ylmcoef[4]*(rly[2]+z*grly[2][2]); + + rly[6] = tmp0*rly[3];//l=2,m=-1 + grly[6][0] = tmp0*grly[3][0]; + grly[6][1] = tmp0*grly[3][1]; + grly[6][2] = ylmcoef[4]*(rly[3]+z*grly[3][2]); + + double tmp2 = ylmcoef[4]*x; + rly[7]= ylmcoef[5]*rly[4]-ylmcoef[6]*rly[0]*radius2 - tmp2*rly[2];//l=2,m=2 + grly[7][0] = ylmcoef[5]*grly[4][0]-ylmcoef[6]*(rly[0]*tx+grly[0][0]*radius2)-ylmcoef[4]*(x*grly[2][0]+rly[2]); // std::cout << "\np1 = "<< ylmcoef[5]*grly[4][0] << " p2 = " << -ylmcoef[6]*rly[0]*tx // << " p3 = " << -ylmcoef[4]*x*grly[2][0] << " p4 = " << -ylmcoef[4]*rly[2] << std::endl; - grly[7][1] = ylmcoef[5]*grly[4][1]-ylmcoef[6]*(rly[0]*ty+grly[0][1]*radius2)-tmp2*grly[2][1]; - grly[7][2] = ylmcoef[5]*grly[4][2]-ylmcoef[6]*(rly[0]*tz+grly[0][2]*radius2)-tmp2*grly[2][2]; + grly[7][1] = ylmcoef[5]*grly[4][1]-ylmcoef[6]*(rly[0]*ty+grly[0][1]*radius2)-tmp2*grly[2][1]; + grly[7][2] = ylmcoef[5]*grly[4][2]-ylmcoef[6]*(rly[0]*tz+grly[0][2]*radius2)-tmp2*grly[2][2]; - rly[8] = -tmp2*rly[3]; - grly[8][0] = -ylmcoef[4]*(rly[3]+x*grly[3][0]); - grly[8][1] = -tmp2*grly[3][1]; - grly[8][2] = -tmp2*grly[3][2]; + rly[8] = -tmp2*rly[3]; + grly[8][0] = -ylmcoef[4]*(rly[3]+x*grly[3][0]); + grly[8][1] = -tmp2*grly[3][1]; + grly[8][2] = -tmp2*grly[3][2]; // rly[8] = tmp1+tmp2*rly[3];//l=2,m=-2 - if (Lmax == 2) return; - - /*************************** - L = 3 - ***************************/ - rly[9] = ylmcoef[7]*z*rly[4]-ylmcoef[8]*rly[1]*radius2; //l=3, m=0 - grly[9][0] = ylmcoef[7]*z*grly[4][0]-ylmcoef[8]*(rly[1]*tx+grly[1][0]*radius2); - grly[9][1] = ylmcoef[7]*z*grly[4][1]-ylmcoef[8]*(rly[1]*ty+grly[1][1]*radius2); - grly[9][2] = ylmcoef[7]*(rly[4]+z*grly[4][2])-ylmcoef[8]*(rly[1]*tz+grly[1][2]*radius2); - - double tmp3 = ylmcoef[9]*z; - rly[10] = tmp3*rly[5]-ylmcoef[10]*rly[2]*radius2;//l=3,m=1 - grly[10][0] = tmp3*grly[5][0]-ylmcoef[10]*(grly[2][0]*radius2+rly[2]*tx); - grly[10][1] = tmp3*grly[5][1]-ylmcoef[10]*(grly[2][1]*radius2+rly[2]*ty); - grly[10][2] = ylmcoef[9]*(z*grly[5][2]+rly[5])-ylmcoef[10]*(grly[2][2]*radius2+rly[2]*tz); - - rly[11] = tmp3*rly[6]-ylmcoef[10]*rly[3]*radius2;//l=3,m=-1 - grly[11][0] = tmp3*grly[6][0]-ylmcoef[10]*(grly[3][0]*radius2+rly[3]*tx); - grly[11][1] = tmp3*grly[6][1]-ylmcoef[10]*(grly[3][1]*radius2+rly[3]*ty); - grly[11][2] = ylmcoef[9]*(z*grly[6][2]+rly[6])-ylmcoef[10]*(grly[3][2]*radius2+rly[3]*tz); - - double tmp4 = ylmcoef[11]*z; - rly[12] = tmp4*rly[7];//l=3,m=2 - grly[12][0] = tmp4*grly[7][0]; - grly[12][1] = tmp4*grly[7][1]; - grly[12][2] = ylmcoef[11]*(z*grly[7][2]+rly[7]); - - rly[13] = tmp4*rly[8];//l=3,m=-2 - grly[13][0] = tmp4*grly[8][0]; - grly[13][1] = tmp4*grly[8][1]; - grly[13][2] = ylmcoef[11]*(z*grly[8][2]+rly[8]); - - double tmp5 = ylmcoef[14]*x; - rly[14] = ylmcoef[12]*rly[10]-ylmcoef[13]*rly[2]*radius2-tmp5*rly[7];//l=3,m=3 - grly[14][0] = ylmcoef[12]*grly[10][0]-ylmcoef[13]*(rly[2]*tx+grly[2][0]*radius2)-ylmcoef[14]*(rly[7]+x*grly[7][0]); - grly[14][1] = ylmcoef[12]*grly[10][1]-ylmcoef[13]*(rly[2]*ty+grly[2][1]*radius2)-tmp5*grly[7][1]; - grly[14][2] = ylmcoef[12]*grly[10][2]-ylmcoef[13]*(rly[2]*tz+grly[2][2]*radius2)-tmp5*grly[7][2]; - - rly[15] = ylmcoef[12]*rly[11]-ylmcoef[13]*rly[3]*radius2-tmp5*rly[8];//l=3,m=-3 - grly[15][0] = ylmcoef[12]*grly[11][0]-ylmcoef[13]*(rly[3]*tx+grly[3][0]*radius2)-ylmcoef[14]*(rly[8]+x*grly[8][0]); - grly[15][1] = ylmcoef[12]*grly[11][1]-ylmcoef[13]*(rly[3]*ty+grly[3][1]*radius2)-tmp5*grly[8][1]; - grly[15][2] = ylmcoef[12]*grly[11][2]-ylmcoef[13]*(rly[3]*tz+grly[3][2]*radius2)-tmp5*grly[8][2]; - if (Lmax == 3) return; - - /*************************** - L = 4 - ***************************/ - rly[16] = ylmcoef[15]*z*rly[9]-ylmcoef[16]*rly[4]*radius2;//l=4,m=0 - grly[16][0] = ylmcoef[15]*z*grly[9][0]-ylmcoef[16]*(rly[4]*tx+grly[4][0]*radius2); - grly[16][1] = ylmcoef[15]*z*grly[9][1]-ylmcoef[16]*(rly[4]*ty+grly[4][1]*radius2); - grly[16][2] = ylmcoef[15]*(z*grly[9][2]+rly[9])-ylmcoef[16]*(rly[4]*tz+grly[4][2]*radius2); - - double tmp6 = ylmcoef[17]*z; - rly[17] = tmp6*rly[10]-ylmcoef[18]*rly[5]*radius2;//l=4,m=1 - grly[17][0] = tmp6*grly[10][0]-ylmcoef[18]*(rly[5]*tx+grly[5][0]*radius2); - grly[17][1] = tmp6*grly[10][1]-ylmcoef[18]*(rly[5]*ty+grly[5][1]*radius2); - grly[17][2] = ylmcoef[17]*(z*grly[10][2]+rly[10])-ylmcoef[18]*(rly[5]*tz+grly[5][2]*radius2); - - rly[18] = tmp6*rly[11]-ylmcoef[18]*rly[6]*radius2;//l=4,m=-1 - grly[18][0] = tmp6*grly[11][0]-ylmcoef[18]*(rly[6]*tx+grly[6][0]*radius2); - grly[18][1] = tmp6*grly[11][1]-ylmcoef[18]*(rly[6]*ty+grly[6][1]*radius2); - grly[18][2] = ylmcoef[17]*(z*grly[11][2]+rly[11])-ylmcoef[18]*(rly[6]*tz+grly[6][2]*radius2); - - double tmp7 = ylmcoef[19]*z; - rly[19] = tmp7*rly[12]-ylmcoef[20]*rly[7]*radius2;//l=4,m=2 - grly[19][0] = tmp7*grly[12][0]-ylmcoef[20]*(rly[7]*tx+grly[7][0]*radius2); - grly[19][1] = tmp7*grly[12][1]-ylmcoef[20]*(rly[7]*ty+grly[7][1]*radius2); - grly[19][2] = ylmcoef[19]*(z*grly[12][2]+rly[12])-ylmcoef[20]*(rly[7]*tz+grly[7][2]*radius2); - - rly[20] = tmp7*rly[13]-ylmcoef[20]*rly[8]*radius2;//l=4,m=-2 - grly[20][0] = tmp7*grly[13][0]-ylmcoef[20]*(rly[8]*tx+grly[8][0]*radius2); - grly[20][1] = tmp7*grly[13][1]-ylmcoef[20]*(rly[8]*ty+grly[8][1]*radius2); - grly[20][2] = ylmcoef[19]*(z*grly[13][2]+rly[13])-ylmcoef[20]*(rly[8]*tz+grly[8][2]*radius2); - - double tmp8 = 3.0*z; - rly[21] = tmp8*rly[14];//l=4,m=3 - grly[21][0] = tmp8*grly[14][0]; - grly[21][1] = tmp8*grly[14][1]; - grly[21][2] = 3.0*(z*grly[14][2]+rly[14]); - - - rly[22] = tmp8*rly[15];//l=4,m=-3 - grly[22][0] = tmp8*grly[15][0]; - grly[22][1] = tmp8*grly[15][1]; - grly[22][2] = 3.0*(z*grly[15][2]+rly[15]); - - double tmp9 = ylmcoef[23]*x; - rly[23] = ylmcoef[21]*rly[19]-ylmcoef[22]*rly[7]*radius2-tmp9*rly[14];//l=4,m=4 - grly[23][0] = ylmcoef[21]*grly[19][0]-ylmcoef[22]*(rly[7]*tx+grly[7][0]*radius2)-ylmcoef[23]*(x*grly[14][0]+rly[14]); - grly[23][1] = ylmcoef[21]*grly[19][1]-ylmcoef[22]*(rly[7]*ty+grly[7][1]*radius2)-tmp9*grly[14][1]; - grly[23][2] = ylmcoef[21]*grly[19][2]-ylmcoef[22]*(rly[7]*tz+grly[7][2]*radius2)-tmp9*grly[14][2]; - - rly[24] = ylmcoef[21]*rly[20]-ylmcoef[22]*rly[8]*radius2-tmp9*rly[15];//l=4,m=-4 - grly[24][0] = ylmcoef[21]*grly[20][0]-ylmcoef[22]*(rly[8]*tx+grly[8][0]*radius2)-ylmcoef[23]*(x*grly[15][0]+rly[15]); - grly[24][1] = ylmcoef[21]*grly[20][1]-ylmcoef[22]*(rly[8]*ty+grly[8][1]*radius2)-tmp9*grly[15][1]; - grly[24][2] = ylmcoef[21]*grly[20][2]-ylmcoef[22]*(rly[8]*tz+grly[8][2]*radius2)-tmp9*grly[15][2]; - - if (Lmax == 4) return; - - /*************************** - L = 5 - ***************************/ - rly[25] = ylmcoef[24]*z*rly[16]-ylmcoef[25]*rly[9]*radius2;//l=5,m=0 - grly[25][0] = ylmcoef[24]*z*grly[16][0]-ylmcoef[25]*(rly[9]*tx+grly[9][0]*radius2); - grly[25][1] = ylmcoef[24]*z*grly[16][1]-ylmcoef[25]*(rly[9]*ty+grly[9][1]*radius2); - grly[25][2] = ylmcoef[24]*(z*grly[16][2]+rly[16])-ylmcoef[25]*(rly[9]*tz+grly[9][2]*radius2); - - double tmp10 = ylmcoef[26]*z; - rly[26] = tmp10*rly[17]-ylmcoef[27]*rly[10]*radius2;//l=5,m=1 - grly[26][0] = tmp10*grly[17][0]-ylmcoef[27]*(rly[10]*tx+grly[10][0]*radius2); - grly[26][1] = tmp10*grly[17][1]-ylmcoef[27]*(rly[10]*ty+grly[10][1]*radius2); - grly[26][2] = ylmcoef[26]*(z*grly[17][2]+rly[17])-ylmcoef[27]*(rly[10]*tz+grly[10][2]*radius2); - - rly[27] = tmp10*rly[18]-ylmcoef[27]*rly[11]*radius2;//l=5,m=-1 - grly[27][0] = tmp10*grly[18][0]-ylmcoef[27]*(rly[11]*tx+grly[11][0]*radius2); - grly[27][1] = tmp10*grly[18][1]-ylmcoef[27]*(rly[11]*ty+grly[11][1]*radius2); - grly[27][2] = ylmcoef[26]*(z*grly[18][2]+rly[18])-ylmcoef[27]*(rly[11]*tz+grly[11][2]*radius2); - - double tmp11 = ylmcoef[28]*z; - rly[28] = tmp11*rly[19]-ylmcoef[29]*rly[12]*radius2;//l=5,m=2 - grly[28][0] = tmp11*grly[19][0]-ylmcoef[29]*(rly[12]*tx+grly[12][0]*radius2); - grly[28][1] = tmp11*grly[19][1]-ylmcoef[29]*(rly[12]*ty+grly[12][1]*radius2); - grly[28][2] = ylmcoef[28]*(z*grly[19][2]+rly[19])-ylmcoef[29]*(rly[12]*tz+grly[12][2]*radius2); - - rly[29] = tmp11*rly[20]-ylmcoef[29]*rly[13]*radius2;//l=5,m=-2 - grly[29][0] = tmp11*grly[20][0]-ylmcoef[29]*(rly[13]*tx+grly[13][0]*radius2); - grly[29][1] = tmp11*grly[20][1]-ylmcoef[29]*(rly[13]*ty+grly[13][1]*radius2); - grly[29][2] = ylmcoef[28]*(z*grly[20][2]+rly[20])-ylmcoef[29]*(rly[13]*tz+grly[13][2]*radius2); - - double tmp12 = ylmcoef[30]*z; - rly[30] = tmp12*rly[21]-ylmcoef[31]*rly[14]*radius2;//l=5,m=3 - grly[30][0] = tmp12*grly[21][0]-ylmcoef[31]*(grly[14][0]*radius2+rly[14]*tx); - grly[30][1] = tmp12*grly[21][1]-ylmcoef[31]*(grly[14][1]*radius2+rly[14]*ty); - grly[30][2] = ylmcoef[30]*(z*grly[21][2]+rly[21])-ylmcoef[31]*(grly[14][2]*radius2+rly[14]*tz); - - rly[31] = tmp12*rly[22]-ylmcoef[31]*rly[15]*radius2;//l=5,m=-3 - grly[31][0] = tmp12*grly[22][0]-ylmcoef[31]*(grly[15][0]*radius2+rly[15]*tx); - grly[31][1] = tmp12*grly[22][1]-ylmcoef[31]*(grly[15][1]*radius2+rly[15]*ty); - grly[31][2] = ylmcoef[30]*(z*grly[22][2]+rly[22])-ylmcoef[31]*(grly[15][2]*radius2+rly[15]*tz); - - double tmp13 = ylmcoef[32]*z; - rly[32] = tmp13*rly[23];//l=5,m=4 - grly[32][0] = tmp13*grly[23][0]; - grly[32][1] = tmp13*grly[23][1]; - grly[32][2] = ylmcoef[32]*(rly[23]+z*grly[23][2]); - - rly[33] = tmp13*rly[24];//l=5,m=-4 - grly[33][0] = tmp13*grly[24][0]; - grly[33][1] = tmp13*grly[24][1]; - grly[33][2] = ylmcoef[32]*(rly[24]+z*grly[24][2]); - - double tmp14 = ylmcoef[35]*x; - rly[34] = ylmcoef[33]*rly[30]-ylmcoef[34]*rly[14]*radius2-tmp14*rly[23];//l=5,m=5 - grly[34][0] = ylmcoef[33]*grly[30][0]-ylmcoef[34]*(rly[14]*tx+grly[14][0]*radius2)-ylmcoef[35]*(x*grly[23][0]+rly[23]); - grly[34][1] = ylmcoef[33]*grly[30][1]-ylmcoef[34]*(rly[14]*ty+grly[14][1]*radius2)-tmp14*grly[23][1]; - grly[34][2] = ylmcoef[33]*grly[30][2]-ylmcoef[34]*(rly[14]*tz+grly[14][2]*radius2)-tmp14*grly[23][2]; - - rly[35] = ylmcoef[33]*rly[31]-ylmcoef[34]*rly[15]*radius2-tmp14*rly[24];//l=5,m=-5 - grly[35][0] = ylmcoef[33]*grly[31][0]-ylmcoef[34]*(rly[15]*tx+grly[15][0]*radius2)-ylmcoef[35]*(x*grly[24][0]+rly[24]); - grly[35][1] = ylmcoef[33]*grly[31][1]-ylmcoef[34]*(rly[15]*ty+grly[15][1]*radius2)-tmp14*grly[24][1]; - grly[35][2] = ylmcoef[33]*grly[31][2]-ylmcoef[34]*(rly[15]*tz+grly[15][2]*radius2)-tmp14*grly[24][2]; - - if (Lmax == 5) return; - - //if Lmax > 5 - for (int il = 6; il <= Lmax; il++) - { - int istart = il*il; - int istart1 = (il-1)*(il-1); - int istart2 = (il-2)*(il-2); - - double fac2 = sqrt(4.0*istart-1.0); - double fac4 = sqrt(4.0*istart1-1.0); - - for (int im = 0; im < 2*il-1; im++) - { - int imm = (im+1)/2; + if (Lmax == 2) return; + + /*************************** + L = 3 + ***************************/ + rly[9] = ylmcoef[7]*z*rly[4]-ylmcoef[8]*rly[1]*radius2; //l=3, m=0 + grly[9][0] = ylmcoef[7]*z*grly[4][0]-ylmcoef[8]*(rly[1]*tx+grly[1][0]*radius2); + grly[9][1] = ylmcoef[7]*z*grly[4][1]-ylmcoef[8]*(rly[1]*ty+grly[1][1]*radius2); + grly[9][2] = ylmcoef[7]*(rly[4]+z*grly[4][2])-ylmcoef[8]*(rly[1]*tz+grly[1][2]*radius2); + + double tmp3 = ylmcoef[9]*z; + rly[10] = tmp3*rly[5]-ylmcoef[10]*rly[2]*radius2;//l=3,m=1 + grly[10][0] = tmp3*grly[5][0]-ylmcoef[10]*(grly[2][0]*radius2+rly[2]*tx); + grly[10][1] = tmp3*grly[5][1]-ylmcoef[10]*(grly[2][1]*radius2+rly[2]*ty); + grly[10][2] = ylmcoef[9]*(z*grly[5][2]+rly[5])-ylmcoef[10]*(grly[2][2]*radius2+rly[2]*tz); + + rly[11] = tmp3*rly[6]-ylmcoef[10]*rly[3]*radius2;//l=3,m=-1 + grly[11][0] = tmp3*grly[6][0]-ylmcoef[10]*(grly[3][0]*radius2+rly[3]*tx); + grly[11][1] = tmp3*grly[6][1]-ylmcoef[10]*(grly[3][1]*radius2+rly[3]*ty); + grly[11][2] = ylmcoef[9]*(z*grly[6][2]+rly[6])-ylmcoef[10]*(grly[3][2]*radius2+rly[3]*tz); + + double tmp4 = ylmcoef[11]*z; + rly[12] = tmp4*rly[7];//l=3,m=2 + grly[12][0] = tmp4*grly[7][0]; + grly[12][1] = tmp4*grly[7][1]; + grly[12][2] = ylmcoef[11]*(z*grly[7][2]+rly[7]); + + rly[13] = tmp4*rly[8];//l=3,m=-2 + grly[13][0] = tmp4*grly[8][0]; + grly[13][1] = tmp4*grly[8][1]; + grly[13][2] = ylmcoef[11]*(z*grly[8][2]+rly[8]); + + double tmp5 = ylmcoef[14]*x; + rly[14] = ylmcoef[12]*rly[10]-ylmcoef[13]*rly[2]*radius2-tmp5*rly[7];//l=3,m=3 + grly[14][0] = ylmcoef[12]*grly[10][0]-ylmcoef[13]*(rly[2]*tx+grly[2][0]*radius2)-ylmcoef[14]*(rly[7]+x*grly[7][0]); + grly[14][1] = ylmcoef[12]*grly[10][1]-ylmcoef[13]*(rly[2]*ty+grly[2][1]*radius2)-tmp5*grly[7][1]; + grly[14][2] = ylmcoef[12]*grly[10][2]-ylmcoef[13]*(rly[2]*tz+grly[2][2]*radius2)-tmp5*grly[7][2]; + + rly[15] = ylmcoef[12]*rly[11]-ylmcoef[13]*rly[3]*radius2-tmp5*rly[8];//l=3,m=-3 + grly[15][0] = ylmcoef[12]*grly[11][0]-ylmcoef[13]*(rly[3]*tx+grly[3][0]*radius2)-ylmcoef[14]*(rly[8]+x*grly[8][0]); + grly[15][1] = ylmcoef[12]*grly[11][1]-ylmcoef[13]*(rly[3]*ty+grly[3][1]*radius2)-tmp5*grly[8][1]; + grly[15][2] = ylmcoef[12]*grly[11][2]-ylmcoef[13]*(rly[3]*tz+grly[3][2]*radius2)-tmp5*grly[8][2]; + if (Lmax == 3) return; + + /*************************** + L = 4 + ***************************/ + rly[16] = ylmcoef[15]*z*rly[9]-ylmcoef[16]*rly[4]*radius2;//l=4,m=0 + grly[16][0] = ylmcoef[15]*z*grly[9][0]-ylmcoef[16]*(rly[4]*tx+grly[4][0]*radius2); + grly[16][1] = ylmcoef[15]*z*grly[9][1]-ylmcoef[16]*(rly[4]*ty+grly[4][1]*radius2); + grly[16][2] = ylmcoef[15]*(z*grly[9][2]+rly[9])-ylmcoef[16]*(rly[4]*tz+grly[4][2]*radius2); + + double tmp6 = ylmcoef[17]*z; + rly[17] = tmp6*rly[10]-ylmcoef[18]*rly[5]*radius2;//l=4,m=1 + grly[17][0] = tmp6*grly[10][0]-ylmcoef[18]*(rly[5]*tx+grly[5][0]*radius2); + grly[17][1] = tmp6*grly[10][1]-ylmcoef[18]*(rly[5]*ty+grly[5][1]*radius2); + grly[17][2] = ylmcoef[17]*(z*grly[10][2]+rly[10])-ylmcoef[18]*(rly[5]*tz+grly[5][2]*radius2); + + rly[18] = tmp6*rly[11]-ylmcoef[18]*rly[6]*radius2;//l=4,m=-1 + grly[18][0] = tmp6*grly[11][0]-ylmcoef[18]*(rly[6]*tx+grly[6][0]*radius2); + grly[18][1] = tmp6*grly[11][1]-ylmcoef[18]*(rly[6]*ty+grly[6][1]*radius2); + grly[18][2] = ylmcoef[17]*(z*grly[11][2]+rly[11])-ylmcoef[18]*(rly[6]*tz+grly[6][2]*radius2); + + double tmp7 = ylmcoef[19]*z; + rly[19] = tmp7*rly[12]-ylmcoef[20]*rly[7]*radius2;//l=4,m=2 + grly[19][0] = tmp7*grly[12][0]-ylmcoef[20]*(rly[7]*tx+grly[7][0]*radius2); + grly[19][1] = tmp7*grly[12][1]-ylmcoef[20]*(rly[7]*ty+grly[7][1]*radius2); + grly[19][2] = ylmcoef[19]*(z*grly[12][2]+rly[12])-ylmcoef[20]*(rly[7]*tz+grly[7][2]*radius2); + + rly[20] = tmp7*rly[13]-ylmcoef[20]*rly[8]*radius2;//l=4,m=-2 + grly[20][0] = tmp7*grly[13][0]-ylmcoef[20]*(rly[8]*tx+grly[8][0]*radius2); + grly[20][1] = tmp7*grly[13][1]-ylmcoef[20]*(rly[8]*ty+grly[8][1]*radius2); + grly[20][2] = ylmcoef[19]*(z*grly[13][2]+rly[13])-ylmcoef[20]*(rly[8]*tz+grly[8][2]*radius2); + + double tmp8 = 3.0*z; + rly[21] = tmp8*rly[14];//l=4,m=3 + grly[21][0] = tmp8*grly[14][0]; + grly[21][1] = tmp8*grly[14][1]; + grly[21][2] = 3.0*(z*grly[14][2]+rly[14]); + + + rly[22] = tmp8*rly[15];//l=4,m=-3 + grly[22][0] = tmp8*grly[15][0]; + grly[22][1] = tmp8*grly[15][1]; + grly[22][2] = 3.0*(z*grly[15][2]+rly[15]); + + double tmp9 = ylmcoef[23]*x; + rly[23] = ylmcoef[21]*rly[19]-ylmcoef[22]*rly[7]*radius2-tmp9*rly[14];//l=4,m=4 + grly[23][0] = ylmcoef[21]*grly[19][0]-ylmcoef[22]*(rly[7]*tx+grly[7][0]*radius2)-ylmcoef[23]*(x*grly[14][0]+rly[14]); + grly[23][1] = ylmcoef[21]*grly[19][1]-ylmcoef[22]*(rly[7]*ty+grly[7][1]*radius2)-tmp9*grly[14][1]; + grly[23][2] = ylmcoef[21]*grly[19][2]-ylmcoef[22]*(rly[7]*tz+grly[7][2]*radius2)-tmp9*grly[14][2]; + + rly[24] = ylmcoef[21]*rly[20]-ylmcoef[22]*rly[8]*radius2-tmp9*rly[15];//l=4,m=-4 + grly[24][0] = ylmcoef[21]*grly[20][0]-ylmcoef[22]*(rly[8]*tx+grly[8][0]*radius2)-ylmcoef[23]*(x*grly[15][0]+rly[15]); + grly[24][1] = ylmcoef[21]*grly[20][1]-ylmcoef[22]*(rly[8]*ty+grly[8][1]*radius2)-tmp9*grly[15][1]; + grly[24][2] = ylmcoef[21]*grly[20][2]-ylmcoef[22]*(rly[8]*tz+grly[8][2]*radius2)-tmp9*grly[15][2]; + + if (Lmax == 4) return; + + /*************************** + L = 5 + ***************************/ + rly[25] = ylmcoef[24]*z*rly[16]-ylmcoef[25]*rly[9]*radius2;//l=5,m=0 + grly[25][0] = ylmcoef[24]*z*grly[16][0]-ylmcoef[25]*(rly[9]*tx+grly[9][0]*radius2); + grly[25][1] = ylmcoef[24]*z*grly[16][1]-ylmcoef[25]*(rly[9]*ty+grly[9][1]*radius2); + grly[25][2] = ylmcoef[24]*(z*grly[16][2]+rly[16])-ylmcoef[25]*(rly[9]*tz+grly[9][2]*radius2); + + double tmp10 = ylmcoef[26]*z; + rly[26] = tmp10*rly[17]-ylmcoef[27]*rly[10]*radius2;//l=5,m=1 + grly[26][0] = tmp10*grly[17][0]-ylmcoef[27]*(rly[10]*tx+grly[10][0]*radius2); + grly[26][1] = tmp10*grly[17][1]-ylmcoef[27]*(rly[10]*ty+grly[10][1]*radius2); + grly[26][2] = ylmcoef[26]*(z*grly[17][2]+rly[17])-ylmcoef[27]*(rly[10]*tz+grly[10][2]*radius2); + + rly[27] = tmp10*rly[18]-ylmcoef[27]*rly[11]*radius2;//l=5,m=-1 + grly[27][0] = tmp10*grly[18][0]-ylmcoef[27]*(rly[11]*tx+grly[11][0]*radius2); + grly[27][1] = tmp10*grly[18][1]-ylmcoef[27]*(rly[11]*ty+grly[11][1]*radius2); + grly[27][2] = ylmcoef[26]*(z*grly[18][2]+rly[18])-ylmcoef[27]*(rly[11]*tz+grly[11][2]*radius2); + + double tmp11 = ylmcoef[28]*z; + rly[28] = tmp11*rly[19]-ylmcoef[29]*rly[12]*radius2;//l=5,m=2 + grly[28][0] = tmp11*grly[19][0]-ylmcoef[29]*(rly[12]*tx+grly[12][0]*radius2); + grly[28][1] = tmp11*grly[19][1]-ylmcoef[29]*(rly[12]*ty+grly[12][1]*radius2); + grly[28][2] = ylmcoef[28]*(z*grly[19][2]+rly[19])-ylmcoef[29]*(rly[12]*tz+grly[12][2]*radius2); + + rly[29] = tmp11*rly[20]-ylmcoef[29]*rly[13]*radius2;//l=5,m=-2 + grly[29][0] = tmp11*grly[20][0]-ylmcoef[29]*(rly[13]*tx+grly[13][0]*radius2); + grly[29][1] = tmp11*grly[20][1]-ylmcoef[29]*(rly[13]*ty+grly[13][1]*radius2); + grly[29][2] = ylmcoef[28]*(z*grly[20][2]+rly[20])-ylmcoef[29]*(rly[13]*tz+grly[13][2]*radius2); + + double tmp12 = ylmcoef[30]*z; + rly[30] = tmp12*rly[21]-ylmcoef[31]*rly[14]*radius2;//l=5,m=3 + grly[30][0] = tmp12*grly[21][0]-ylmcoef[31]*(grly[14][0]*radius2+rly[14]*tx); + grly[30][1] = tmp12*grly[21][1]-ylmcoef[31]*(grly[14][1]*radius2+rly[14]*ty); + grly[30][2] = ylmcoef[30]*(z*grly[21][2]+rly[21])-ylmcoef[31]*(grly[14][2]*radius2+rly[14]*tz); + + rly[31] = tmp12*rly[22]-ylmcoef[31]*rly[15]*radius2;//l=5,m=-3 + grly[31][0] = tmp12*grly[22][0]-ylmcoef[31]*(grly[15][0]*radius2+rly[15]*tx); + grly[31][1] = tmp12*grly[22][1]-ylmcoef[31]*(grly[15][1]*radius2+rly[15]*ty); + grly[31][2] = ylmcoef[30]*(z*grly[22][2]+rly[22])-ylmcoef[31]*(grly[15][2]*radius2+rly[15]*tz); + + double tmp13 = ylmcoef[32]*z; + rly[32] = tmp13*rly[23];//l=5,m=4 + grly[32][0] = tmp13*grly[23][0]; + grly[32][1] = tmp13*grly[23][1]; + grly[32][2] = ylmcoef[32]*(rly[23]+z*grly[23][2]); + + rly[33] = tmp13*rly[24];//l=5,m=-4 + grly[33][0] = tmp13*grly[24][0]; + grly[33][1] = tmp13*grly[24][1]; + grly[33][2] = ylmcoef[32]*(rly[24]+z*grly[24][2]); + + double tmp14 = ylmcoef[35]*x; + rly[34] = ylmcoef[33]*rly[30]-ylmcoef[34]*rly[14]*radius2-tmp14*rly[23];//l=5,m=5 + grly[34][0] = ylmcoef[33]*grly[30][0]-ylmcoef[34]*(rly[14]*tx+grly[14][0]*radius2)-ylmcoef[35]*(x*grly[23][0]+rly[23]); + grly[34][1] = ylmcoef[33]*grly[30][1]-ylmcoef[34]*(rly[14]*ty+grly[14][1]*radius2)-tmp14*grly[23][1]; + grly[34][2] = ylmcoef[33]*grly[30][2]-ylmcoef[34]*(rly[14]*tz+grly[14][2]*radius2)-tmp14*grly[23][2]; + + rly[35] = ylmcoef[33]*rly[31]-ylmcoef[34]*rly[15]*radius2-tmp14*rly[24];//l=5,m=-5 + grly[35][0] = ylmcoef[33]*grly[31][0]-ylmcoef[34]*(rly[15]*tx+grly[15][0]*radius2)-ylmcoef[35]*(x*grly[24][0]+rly[24]); + grly[35][1] = ylmcoef[33]*grly[31][1]-ylmcoef[34]*(rly[15]*ty+grly[15][1]*radius2)-tmp14*grly[24][1]; + grly[35][2] = ylmcoef[33]*grly[31][2]-ylmcoef[34]*(rly[15]*tz+grly[15][2]*radius2)-tmp14*grly[24][2]; + + if (Lmax == 5) return; + + //if Lmax > 5 + for (int il = 6; il <= Lmax; il++) + { + int istart = il*il; + int istart1 = (il-1)*(il-1); + int istart2 = (il-2)*(il-2); + + double fac2 = sqrt(4.0*istart-1.0); + double fac4 = sqrt(4.0*istart1-1.0); + + for (int im = 0; im < 2*il-1; im++) + { + int imm = (im+1)/2; // if (im % 2 == 0) imm *= -1; - double var1 = fac2/sqrt((double)istart-imm*imm); - double var2 = sqrt((double)istart1-imm*imm)/fac4; + double var1 = fac2/sqrt((double)istart-imm*imm); + double var2 = sqrt((double)istart1-imm*imm)/fac4; - rly[istart+im] = var1*(z*rly[istart1+im] - var2*rly[istart2+im]*radius2); + rly[istart+im] = var1*(z*rly[istart1+im] - var2*rly[istart2+im]*radius2); - grly[istart+im][0]=var1*(z*grly[istart1+im][0]-var2*(rly[istart2+im]*tx+grly[istart2+im][0]*radius2)); - grly[istart+im][1]=var1*(z*grly[istart1+im][1]-var2*(rly[istart2+im]*ty+grly[istart2+im][1]*radius2)); - grly[istart+im][2]=var1*(z*grly[istart1+im][2]+rly[istart1+im]-var2*(rly[istart2+im]*tz+grly[istart2+im][2]*radius2)); + grly[istart+im][0]=var1*(z*grly[istart1+im][0]-var2*(rly[istart2+im]*tx+grly[istart2+im][0]*radius2)); + grly[istart+im][1]=var1*(z*grly[istart1+im][1]-var2*(rly[istart2+im]*ty+grly[istart2+im][1]*radius2)); + grly[istart+im][2]=var1*(z*grly[istart1+im][2]+rly[istart1+im]-var2*(rly[istart2+im]*tz+grly[istart2+im][2]*radius2)); - } + } - double bl1 = sqrt(2.0*il/(2.0*il+1.0)); - double bl2 = sqrt((2.0*il-2.0)/(2.0*il-1.0)); - double bl3 = sqrt(2.0)/fac2; + double bl1 = sqrt(2.0*il/(2.0*il+1.0)); + double bl2 = sqrt((2.0*il-2.0)/(2.0*il-1.0)); + double bl3 = sqrt(2.0)/fac2; - int id1 = istart+2*il-1; - int id2 = istart+2*il-5; - int id3 = istart2+2*il-5; - int id4 = istart1+2*il-3; + int id1 = istart+2*il-1; + int id2 = istart+2*il-5; + int id3 = istart2+2*il-5; + int id4 = istart1+2*il-3; - rly[id1] = (bl3*rly[id2]-bl2*rly[id3]*radius2-2.0*x*rly[id4]) / bl1; - grly[id1][0] = (bl3*grly[id2][0]-bl2*(grly[id3][0]*radius2+rly[id3]*tx)-2.0*(rly[id4]+x*grly[id4][0]))/bl1; - grly[id1][1] = (bl3*grly[id2][1]-bl2*(grly[id3][1]*radius2+rly[id3]*ty)-2.0*x*grly[id4][1])/bl1; - grly[id1][2] = (bl3*grly[id2][2]-bl2*(grly[id3][2]*radius2+rly[id3]*tz)-2.0*x*grly[id4][2])/bl1; + rly[id1] = (bl3*rly[id2]-bl2*rly[id3]*radius2-2.0*x*rly[id4]) / bl1; + grly[id1][0] = (bl3*grly[id2][0]-bl2*(grly[id3][0]*radius2+rly[id3]*tx)-2.0*(rly[id4]+x*grly[id4][0]))/bl1; + grly[id1][1] = (bl3*grly[id2][1]-bl2*(grly[id3][1]*radius2+rly[id3]*ty)-2.0*x*grly[id4][1])/bl1; + grly[id1][2] = (bl3*grly[id2][2]-bl2*(grly[id3][2]*radius2+rly[id3]*tz)-2.0*x*grly[id4][2])/bl1; - rly[id1+1] = (bl3*rly[id2+1]-bl2*rly[id3+1]*radius2-2.0*x*rly[id4+1]) / bl1; - grly[id1+1][0] = (bl3*grly[id2+1][0]-bl2*(grly[id3+1][0]*radius2+rly[id3+1]*tx)-2.0*(rly[id4+1]+x*grly[id4+1][0]))/bl1; - grly[id1+1][1] = (bl3*grly[id2+1][1]-bl2*(grly[id3+1][1]*radius2+rly[id3+1]*ty)-2.0*x*grly[id4+1][1])/bl1; - grly[id1+1][2] = (bl3*grly[id2+1][2]-bl2*(grly[id3+1][2]*radius2+rly[id3+1]*tz)-2.0*x*grly[id4+1][2])/bl1; - } + rly[id1+1] = (bl3*rly[id2+1]-bl2*rly[id3+1]*radius2-2.0*x*rly[id4+1]) / bl1; + grly[id1+1][0] = (bl3*grly[id2+1][0]-bl2*(grly[id3+1][0]*radius2+rly[id3+1]*tx)-2.0*(rly[id4+1]+x*grly[id4+1][0]))/bl1; + grly[id1+1][1] = (bl3*grly[id2+1][1]-bl2*(grly[id3+1][1]*radius2+rly[id3+1]*ty)-2.0*x*grly[id4+1][1])/bl1; + grly[id1+1][2] = (bl3*grly[id2+1][2]-bl2*(grly[id3+1][2]*radius2+rly[id3+1]*tz)-2.0*x*grly[id4+1][2])/bl1; + } - return; + return; } void Ylm::hes_rl_sph_harm ( - const int Lmax, //max momentum of L - const double x, - const double y, - const double z, - std::vector>& hrly + const int Lmax, //max momentum of L + const double x, + const double y, + const double z, + std::vector>& hrly ) { - hrly.resize( (Lmax+1)*(Lmax+1), std::vector(6) ); - - double radius2 = x*x+y*y+z*z; - double coeff = 0.0; - - //begin calculation - /*************************** - L = 0 - ***************************/ - hrly[0][0] = hrly[0][1] = hrly[0][2] = 0.0; - hrly[0][3] = hrly[0][4] = hrly[0][5] = 0.0; - if (Lmax == 0) return; - - /*************************** - L = 1 - ***************************/ - hrly[1][0] = hrly[1][1] = hrly[1][2] = 0.0; - hrly[1][3] = hrly[1][4] = hrly[1][5] = 0.0; - - hrly[2][0] = hrly[2][1] = hrly[2][2] = 0.0; - hrly[2][3] = hrly[2][4] = hrly[2][5] = 0.0; - - hrly[3][0] = hrly[3][1] = hrly[3][2] = 0.0; - hrly[3][3] = hrly[3][4] = hrly[3][5] = 0.0; - - if (Lmax == 1) return; - - /*************************** - L = 2 - ***************************/ - //m=0 : 3z^2-r^2 - coeff = sqrt(5.0 / ModuleBase::PI) / 4.0; - hrly[4][0] = hrly[4][3] = -2.0 * coeff; - hrly[4][5] = 4.0 * coeff; - hrly[4][1] = hrly[4][2] = hrly[4][4] = 0.0; - - //m=1 : xz - coeff = sqrt(15.0 / ModuleBase::PI) / 2.0; - hrly[5][2] = coeff; - hrly[5][0] = hrly[5][1] = 0.0; - hrly[5][3] = hrly[5][4] = hrly[5][5] = 0.0; - - //m=-1 : yz - hrly[6][4] = coeff; - hrly[6][0] = hrly[6][1] = 0.0; - hrly[6][2] = hrly[6][3] = hrly[6][5] = 0.0; - - //m=-2 : xy - hrly[8][1] = coeff; - hrly[8][0] = hrly[8][2] = 0.0; - hrly[8][3] = hrly[8][4] = hrly[8][5] = 0.0; - - //m=2 : (x^2-y^2) - coeff = sqrt(15.0 / ModuleBase::PI) / 4.0; - hrly[7][0] = 2.0 * coeff; - hrly[7][3] = -2.0 * coeff; - hrly[7][1] = hrly[7][2] = 0.0; - hrly[7][4] = hrly[7][5] = 0.0; - - if (Lmax == 2) return; - - /*************************** - L = 3 - ***************************/ - //m=0 : (5z^3-3zr^2) - coeff = sqrt(7.0 / ModuleBase::PI) / 4.0; - hrly[9][0] = hrly[9][3] = -6.0 * z * coeff; - hrly[9][1] = 0.0; - hrly[9][2] = -6.0 * x * coeff; - hrly[9][4] = -6.0 * y * coeff; - hrly[9][5] = 12.0 * z * coeff; - - //m=1 : x(5z^2-r^2) - coeff = sqrt(21.0 / 2.0 / ModuleBase::PI) / 4.0; - hrly[10][0] = -6.0 * x * coeff; - hrly[10][1] = -2.0 * y * coeff; - hrly[10][2] = 8.0 * z * coeff; - hrly[10][3] = -2.0 * x * coeff; - hrly[10][4] = 0.0; - hrly[10][5] = 8.0 * x * coeff; - - //m=-1 : y(5z^2-r^2) - hrly[11][0] = -2.0 * y * coeff; - hrly[11][1] = -2.0 * x * coeff; - hrly[11][2] = 0.0; - hrly[11][3] = -6.0 * y * coeff; - hrly[11][4] = 8.0 * z * coeff; - hrly[11][5] = 8.0 * y * coeff; - - //m=2 : (x^2-y^2)z - coeff = sqrt(105.0 / ModuleBase::PI) / 4.0; - hrly[12][0] = 2.0 * z * coeff; - hrly[12][1] = 0.0; - hrly[12][2] = 2.0 * x * coeff; - hrly[12][3] = -hrly[12][0]; - hrly[12][4] = -2.0 * y * coeff; - hrly[12][5] = 0.0; - - //m=-2 : xyz - coeff = sqrt(105.0 / ModuleBase::PI) / 2.0; - hrly[13][0] = 0.0; - hrly[13][1] = z * coeff; - hrly[13][2] = y * coeff; - hrly[13][3] = 0.0; - hrly[13][4] = x * coeff; - hrly[13][5] = 0.0; - - //m=3 : x(x^2-3y^2) - coeff = sqrt(35.0 / 2.0 / ModuleBase::PI) / 4.0; - hrly[14][0] = 6.0 * x * coeff; - hrly[14][1] = -6.0 * y * coeff; - hrly[14][2] = 0.0; - hrly[14][3] = -hrly[14][0]; - hrly[14][4] = 0.0; - hrly[14][5] = 0.0; - - //m=-3 : y(3x^2-y^2) - hrly[15][0] = 6.0 * y * coeff; - hrly[15][1] = 6.0 * x * coeff; - hrly[15][2] = 0.0; - hrly[15][3] = -hrly[15][0]; - hrly[15][4] = 0.0; - hrly[15][5] = 0.0; - - if (Lmax == 3) return; - - /*************************** - L = 4 - ***************************/ - //m=0 : (35z^4 - 30z^2r^2 + 3r^4) - coeff = sqrt(1.0 / ModuleBase::PI) * 3.0 / 16.0; - hrly[16][0] = 12.0 * (3.0 * x*x + y*y - 4.0 * z*z) * coeff; - hrly[16][1] = 24.0 * x * y * coeff; - hrly[16][2] = -96.0 * x * z * coeff; - hrly[16][3] = 12.0 * (x*x + 3.0 * y*y - 4.0 * z*z) * coeff; - hrly[16][4] = -96.0 * y * z * coeff; - hrly[16][5] = -48.0 * (x*x + y*y -2.0 * z*z) * coeff; - - //m=1 : x(7z^3 - 3zr^2) - coeff = 3.0 / 4.0 * sqrt(5.0 / 2.0 / ModuleBase::PI); - hrly[17][0] = -18.0 * x * z * coeff; - hrly[17][1] = -6.0 * y * z * coeff; - hrly[17][2] = -3.0 * (3.0 * x*x + y*y - 4.0 * z*z) * coeff; - hrly[17][3] = -6.0 * x * z * coeff; - hrly[17][4] = -6.0 * x * y * coeff; - hrly[17][5] = 24.0 * x * z * coeff; - - //m=-1 : y(7z^3 - 3zr^2) - hrly[18][0] = -6.0 * y * z * coeff; - hrly[18][1] = -6.0 * x * z * coeff; - hrly[18][2] = -6.0 * x * y * coeff; - hrly[18][3] = -18.0 * y * z * coeff; - hrly[18][4] = -3.0 * (x*x + 3.0 * y*y - 4.0 * z*z) * coeff; - hrly[18][5] = 24.0 * y * z * coeff; - - //m=2 : (x^2 - y^2)(7z^2 - r^2) - coeff = 3.0 / 8.0 * sqrt(5.0 / ModuleBase::PI); - hrly[19][0] = -12.0 * (x*x - z*z) * coeff; - hrly[19][1] = 0.0; - hrly[19][2] = 24.0 * x * z * coeff; - hrly[19][3] = 12.0 * (y*y - z*z) * coeff; - hrly[19][4] = -24.0 * y * z * coeff; - hrly[19][5] = 12.0 * (x*x - y*y) * coeff; - - //m=-2 : xy(7z^2 - r^2) - coeff = 3.0 / 4.0 * sqrt(5.0 / ModuleBase::PI); - hrly[20][0] = -6.0 * x * y * coeff; - hrly[20][1] = -3.0 * (x*x + y*y - 2.0 * z*z) * coeff; - hrly[20][2] = 2.0 * y * z * coeff; - hrly[20][3] = hrly[20][0]; - hrly[20][4] = 12.0 * x * z * coeff; - hrly[20][5] = 12.0 * x * y * coeff; - - //m=3 : x(x^2-3y^2)z - coeff = 3.0 / 4.0 * sqrt(35.0 / 2.0 / ModuleBase::PI); - hrly[21][0] = 6.0 * x * z * coeff; - hrly[21][1] = -6.0 * y * z * coeff; - hrly[21][2] = 3.0 * (x*x - y*y) * coeff; - hrly[21][3] = -6.0 * x * z * coeff; - hrly[21][4] = -6.0 * x * y * coeff; - hrly[21][5] = 0.0; - - //m=-3 : y(3x^2-y^2)z - hrly[22][0] = 6.0 * y * z * coeff; - hrly[22][1] = 6.0 * x * z * coeff; - hrly[22][2] = 6.0 * x * y * coeff; - hrly[22][3] = -6.0 * y * z * coeff; - hrly[22][4] = 3.0 * (x*x - y*y) * coeff; - hrly[22][5] = 0.0; - - //m=4 : x^4 + y^4 - 6 x^2y^2 - coeff = 3.0 / 16.0 * sqrt(35.0 / ModuleBase::PI); - hrly[23][0] = 12.0 * (x*x - y*y) * coeff; - hrly[23][1] = -24.0 * x * y * coeff; - hrly[23][2] = 0.0; - hrly[23][3] = -hrly[23][0]; - hrly[23][4] = 0.0; - hrly[23][5] = 0.0; - - //m=-4 : xy(x^2 - y^2) - coeff = 3.0 / 4.0 * sqrt(35.0 / ModuleBase::PI); - hrly[24][0] = 6.0 * x * y * coeff; - hrly[24][1] = 3.0 * (x*x - y*y) * coeff; - hrly[24][2] = 0.0; - hrly[24][3] = -hrly[24][0]; - hrly[24][4] = 0.0; - hrly[24][5] = 0.0; - - if (Lmax == 4) return; - - /*************************** - L = 5 - ***************************/ - //m=0 : (63z^5 - 70z^3*r^2 + 15z*r^4) - coeff = sqrt(11.0 / ModuleBase::PI) / 16.0; - hrly[25][0] = (180*x*x*z + 60*y*y*z - 80*z*z*z) * coeff; - hrly[25][1] = (120*x*y*z) * coeff; - hrly[25][2] = (60*x*x*x + 60*x*y*y - 240*x*z*z) * coeff; - hrly[25][3] = (60*x*x*z + 180*y*y*z - 80*z*z*z) * coeff; - hrly[25][4] = (60*x*x*y + 60*y*y*y - 240*y*z*z) * coeff; - hrly[25][5] = (-240*x*x*z - 240*y*y*z + 160*z*z*z) * coeff; - - //m=1 : x(21z^4 - 14z^2*r^2 + r^4) - coeff = sqrt(165.0 / 2.0 / ModuleBase::PI) / 16.0; - hrly[26][0] = (20*x*x*x + 12*x*y*y - 72*x*z*z) * coeff; - hrly[26][1] = (12*x*x*y + 4*y*y*y - 24*y*z*z) * coeff; - hrly[26][2] = (-72*x*x*z - 24*y*y*z + 32*z*z*z) * coeff; - hrly[26][3] = (4*x*x*x + 12*x*y*y - 24*x*z*z) * coeff; - hrly[26][4] = (-48*x*y*z) * coeff; - hrly[26][5] = (-24*x*x*x - 24*x*y*y + 96*x*z*z) * coeff; - - //m=-1 : y(21z^4 - 14z^2*r^2 + r^4) - hrly[27][0] = (12*x*x*y + 4*y*y*y - 24*y*z*z) * coeff; - hrly[27][1] = (4*x*x*x + 12*x*y*y - 24*x*z*z) * coeff; - hrly[27][2] = (-48*x*y*z) * coeff; - hrly[27][3] = (12*x*x*y + 20*y*y*y - 72*y*z*z) * coeff; - hrly[27][4] = (-24*x*x*z - 72*y*y*z + 32*z*z*z) * coeff; - hrly[27][5] = (-24*x*x*y - 24*y*y*y + 96*y*z*z) * coeff; - - //m=2 : (x^2 - y^2)(3z^3 - z*r^2) - coeff = sqrt(1155.0 / ModuleBase::PI) / 8.0; - hrly[28][0] = (-12*x*x*z + 4*z*z*z) * coeff; - hrly[28][1] = 0.0; - hrly[28][2] = (-4*x*x*x + 12*x*z*z) * coeff; - hrly[28][3] = (12*y*y*z - 4*z*z*z) * coeff; - hrly[28][4] = (4*y*y*y - 12*y*z*z) * coeff; - hrly[28][5] = (12*x*x*z - 12*y*y*z) * coeff; - - //m=-2 : xy(3z^3 - z*r^2) - hrly[29][0] = (-6*x*y*z) * coeff; - hrly[29][1] = (-3*x*x*z - 3*y*y*z + 2*z*z*z) * coeff; - hrly[29][2] = (-3*x*x*y - y*y*y + 6*y*z*z) * coeff; - hrly[29][3] = (-6*x*y*z) * coeff; - hrly[29][4] = (-x*x*x - 3*x*y*y + 6*x*z*z) * coeff; - hrly[29][5] = (12*x*y*z) * coeff; - - //m=3 : x(x^2 - 3y^2)(9z^2 - r^2) - coeff = sqrt(385.0 / 2.0 / ModuleBase::PI) / 16.0; - hrly[30][0] = (-20*x*x*x + 12*x*y*y + 48*x*z*z) * coeff; - hrly[30][1] = (12*x*x*y + 12*y*y*y - 48*y*z*z) * coeff; - hrly[30][2] = (48*x*x*z - 48*y*y*z) * coeff; - hrly[30][3] = (4*x*x*x + 36*x*y*y - 48*x*z*z) * coeff; - hrly[30][4] = (-96*x*y*z) * coeff; - hrly[30][5] = (16*x*x*x - 48*x*y*y) * coeff; - - //m=-3 : y(3x^2 - y^2)(9z^2 - r^2) - hrly[31][0] = (-36*x*x*y - 4*y*y*y + 48*y*z*z) * coeff; - hrly[31][1] = (-12*x*x*x - 12*x*y*y + 48*x*z*z) * coeff; - hrly[31][2] = (96*x*y*z) * coeff; - hrly[31][3] = (-12*x*x*y + 20*y*y*y - 48*y*z*z) * coeff; - hrly[31][4] = (48*x*x*z - 48*y*y*z) * coeff; - hrly[31][5] = (48*x*x*y - 16*y*y*y) * coeff; - - //m=4 : (x^4 - 6x^2*y^2 + y^4) * z - coeff = sqrt(385.0 / ModuleBase::PI) / 16.0; - hrly[32][0] = (12*x*x*z - 12*y*y*z) * coeff; - hrly[32][1] = (-24*x*y*z) * coeff; - hrly[32][2] = (4*x*x*x - 12*x*y*y) * coeff; - hrly[32][3] = (-12*x*x*z + 12*y*y*z) * coeff; - hrly[32][4] = (-12*x*x*y + 4*y*y*y) * coeff; - hrly[32][5] = 0.0; - - //m=-4 : xy(x^2 - y^2) * z - hrly[33][0] = (6*x*y*z) * coeff; - hrly[33][1] = (3*x*x*z - 3*y*y*z) * coeff; - hrly[33][2] = (3*x*x*y - y*y*y) * coeff; - hrly[33][3] = (-6*x*y*z) * coeff; - hrly[33][4] = (x*x*x - 3*x*y*y) * coeff; - hrly[33][5] = 0.0; - - //m=5 : x(x^4 - 10x^2*y^2 + 5y^4) - coeff = sqrt(77.0 / 2.0 / ModuleBase::PI) / 16.0; - hrly[34][0] = (20.0 * x*x*x - 60.0 * x * y*y) * coeff; - hrly[34][1] = (-60.0 * x*x * y + 20.0 * y*y*y) * coeff; - hrly[34][2] = 0.0; - hrly[34][3] = (-20.0 * x*x*x + 60.0 * x * y*y) * coeff; - hrly[34][4] = 0.0; - hrly[34][5] = 0.0; - - //m=-5 : y(5x^4 - 10x^2*y^2 + y^4) - hrly[35][0] = (60.0 * x*x * y - 20.0 * y*y*y) * coeff; - hrly[35][1] = (20.0 * x*x*x - 60.0 * x * y*y) * coeff; - hrly[35][2] = 0.0; - hrly[35][3] = (-60.0 * x*x * y + 20.0 * y*y*y) * coeff; - hrly[35][4] = 0.0; - hrly[35][5] = 0.0; - - if (Lmax == 5) return; - - /*************************** - L = 6 - ***************************/ - //m=0 : (231z^6 - 315z^4*r^2 + 105z^2*r^4 - 5r^6) - coeff = sqrt(13.0 / ModuleBase::PI) / 32.0; - hrly[36][0] = (-150*x*x*x*x - 180*x*x*y*y + 1080*x*x*z*z - 30*y*y*y*y + 360*y*y*z*z - 240*z*z*z*z) * coeff; - hrly[36][1] = (-120*x*x*x*y - 120*x*y*y*y + 720*x*y*z*z) * coeff; - hrly[36][2] = (720*x*x*x*z + 720*x*y*y*z - 960*x*z*z*z) * coeff; - hrly[36][3] = (-30*x*x*x*x - 180*x*x*y*y + 360*x*x*z*z - 150*y*y*y*y + 1080*y*y*z*z - 240*z*z*z*z) * coeff; - hrly[36][4] = (720*x*x*y*z + 720*y*y*y*z - 960*y*z*z*z) * coeff; - hrly[36][5] = (180*x*x*x*x + 360*x*x*y*y - 1440*x*x*z*z + 180*y*y*y*y - 1440*y*y*z*z + 480*z*z*z*z) * coeff; - - //m=1 : x(33z^5 - 30z^3*r^2 + 5z*r^4) - coeff = sqrt(273.0 / 2.0 / ModuleBase::PI) / 16.0; - hrly[37][0] = (100*x*x*x*z + 60*x*y*y*z - 120*x*z*z*z) * coeff; - hrly[37][1] = (60*x*x*y*z + 20*y*y*y*z - 40*y*z*z*z) * coeff; - hrly[37][2] = (25*x*x*x*x + 30*x*x*y*y - 180*x*x*z*z + 5*y*y*y*y - 60*y*y*z*z + 40*z*z*z*z) * coeff; - hrly[37][3] = (20*x*x*x*z + 60*x*y*y*z - 40*x*z*z*z) * coeff; - hrly[37][4] = (20*x*x*x*y + 20*x*y*y*y - 120*x*y*z*z) * coeff; - hrly[37][5] = (-120*x*x*x*z - 120*x*y*y*z + 160*x*z*z*z) * coeff; - - //m=-1 : y(33z^5 - 30z^3*r^2 + 5z*r^4) - hrly[38][0] = (60*x*x*y*z + 20*y*y*y*z - 40*y*z*z*z) * coeff; - hrly[38][1] = (20*x*x*x*z + 60*x*y*y*z - 40*x*z*z*z) * coeff; - hrly[38][2] = (20*x*x*x*y + 20*x*y*y*y - 120*x*y*z*z) * coeff; - hrly[38][3] = (60*x*x*y*z + 100*y*y*y*z - 120*y*z*z*z) * coeff; - hrly[38][4] = (5*x*x*x*x + 30*x*x*y*y - 60*x*x*z*z + 25*y*y*y*y - 180*y*y*z*z + 40*z*z*z*z) * coeff; - hrly[38][5] = (-120*x*x*y*z - 120*y*y*y*z + 160*y*z*z*z) * coeff; - - //m=2 : (x^2 - y^2)(33z^4 - 18z^2*r^2 + r^4) - coeff = sqrt(1365.0 / ModuleBase::PI) / 32.0; - hrly[39][0] = (30*x*x*x*x + 12*x*x*y*y - 192*x*x*z*z - 2*y*y*y*y + 32*z*z*z*z) * coeff; - hrly[39][1] = (8*x*x*x*y - 8*x*y*y*y) * coeff; - hrly[39][2] = (-128*x*x*x*z + 128*x*z*z*z) * coeff; - hrly[39][3] = (2*x*x*x*x - 12*x*x*y*y - 30*y*y*y*y + 192*y*y*z*z - 32*z*z*z*z) * coeff; - hrly[39][4] = (128*y*y*y*z - 128*y*z*z*z) * coeff; - hrly[39][5] = (-32*x*x*x*x + 192*x*x*z*z + 32*y*y*y*y - 192*y*y*z*z) * coeff; - - //m=-2 : xy(33z^4 - 18z^2*r^2 + r^4) - hrly[40][0] = (20*x*x*x*y + 12*x*y*y*y - 96*x*y*z*z) * coeff; - hrly[40][1] = (20*x*x*x*x + 36*x*x*y*y - 96*x*x*z*z + 20*y*y*y*y - 96*y*y*z*z + 32*z*z*z*z) * coeff; - hrly[40][2] = (-96*x*x*y*z - 32*y*y*y*z + 64*y*z*z*z) * coeff; - hrly[40][3] = (12*x*x*x*y + 20*x*y*y*y - 96*x*y*z*z) * coeff; - hrly[40][4] = (-32*x*x*x*z - 96*x*y*y*z + 64*x*z*z*z) * coeff; - hrly[40][5] = (-32*x*x*x*y - 32*x*y*y*y + 192*x*y*z*z) * coeff; - - //m=3 : x(x^2 - 3y^2)(11z^3 - 3z*r^2) - coeff = sqrt(1365.0 / ModuleBase::PI) / 16.0; - hrly[41][0] = (-60*x*x*x*z + 36*x*y*y*z + 48*x*z*z*z) * coeff; - hrly[41][1] = (36*x*x*y*z + 36*y*y*y*z - 48*y*z*z*z) * coeff; - hrly[41][2] = (-30*x*x*x*x + 36*x*x*y*y + 72*x*x*z*z + 18*y*y*y*y - 72*y*y*z*z) * coeff; - hrly[41][3] = (12*x*x*x*z + 108*x*y*y*z - 48*x*z*z*z) * coeff; - hrly[41][4] = (12*x*x*x*y + 36*x*y*y*y - 144*x*y*z*z) * coeff; - hrly[41][5] = (48*x*x*x*z - 144*x*y*y*z) * coeff; - - //m=-3 : y(3x^2 - y^2)(11z^3 - 3z*r^2) - hrly[42][0] = (-108*x*x*y*z - 12*y*y*y*z + 48*y*z*z*z) * coeff; - hrly[42][1] = (-36*x*x*x*z - 36*x*y*y*z + 48*x*z*z*z) * coeff; - hrly[42][2] = (-36*x*x*x*y - 12*x*y*y*y + 144*x*y*z*z) * coeff; - hrly[42][3] = (-36*x*x*y*z + 60*y*y*y*z - 48*y*z*z*z) * coeff; - hrly[42][4] = (-18*x*x*x*x - 36*x*x*y*y + 72*x*x*z*z + 30*y*y*y*y - 72*y*y*z*z) * coeff; - hrly[42][5] = (144*x*x*y*z - 48*y*y*y*z) * coeff; - - //m=4 : (x^4 - 6x^2*y^2 + y^4)(11z^2 - r^2) - coeff = sqrt(91.0 / ModuleBase::PI) / 32.0; - hrly[43][0] = (-30*x*x*x*x + 60*x*x*y*y + 120*x*x*z*z + 10*y*y*y*y - 120*y*y*z*z) * coeff; - hrly[43][1] = (40*x*x*x*y + 40*x*y*y*y - 240*x*y*z*z) * coeff; - hrly[43][2] = (80*x*x*x*z - 240*x*y*y*z) * coeff; - hrly[43][3] = (10*x*x*x*x + 60*x*x*y*y - 120*x*x*z*z - 30*y*y*y*y + 120*y*y*z*z) * coeff; - hrly[43][4] = (-240*x*x*y*z + 80*y*y*y*z) * coeff; - hrly[43][5] = (20*x*x*x*x - 120*x*x*y*y + 20*y*y*y*y) * coeff; - - //m=-4 : xy(x^2 - y^2)(11z^2 - r^2) - hrly[44][0] = (-20*x*x*x*y + 60*x*y*z*z) * coeff; - hrly[44][1] = (-5*x*x*x*x + 30*x*x*z*z + 5*y*y*y*y - 30*y*y*z*z) * coeff; - hrly[44][2] = (60*x*x*y*z - 20*y*y*y*z) * coeff; - hrly[44][3] = (20*x*y*y*y - 60*x*y*z*z) * coeff; - hrly[44][4] = (20*x*x*x*z - 60*x*y*y*z) * coeff; - hrly[44][5] = (20*x*x*x*y - 20*x*y*y*y) * coeff; - - //m=5 : x(x^4 - 10x^2*y^2 + 5y^4) * z - coeff = sqrt(1001.0 / 2.0 / ModuleBase::PI) / 16.0; - hrly[45][0] = (20*x*x*x*z - 60*x*y*y*z) * coeff; - hrly[45][1] = (-60*x*x*y*z + 20*y*y*y*z) * coeff; - hrly[45][2] = (5*x*x*x*x - 30*x*x*y*y + 5*y*y*y*y) * coeff; - hrly[45][3] = (-20*x*x*x*z + 60*x*y*y*z) * coeff; - hrly[45][4] = (-20*x*x*x*y + 20*x*y*y*y) * coeff; - hrly[45][5] = 0.0; - - //m=-5 : y(5x^4 - 10x^2*y^2 + y^4) * z - hrly[46][0] = (60*x*x*y*z - 20*y*y*y*z) * coeff; - hrly[46][1] = (20*x*x*x*z - 60*x*y*y*z) * coeff; - hrly[46][2] = (20*x*x*x*y - 20*x*y*y*y) * coeff; - hrly[46][3] = (-60*x*x*y*z + 20*y*y*y*z) * coeff; - hrly[46][4] = (5*x*x*x*x - 30*x*x*y*y + 5*y*y*y*y) * coeff; - hrly[46][5] = 0.0; - - //m=6 : (x^6 - 15x^4*y^2 + 15x^2*y^4 - y^6) - coeff = sqrt(3003.0 / ModuleBase::PI) / 32.0; - hrly[47][0] = (30*x*x*x*x - 180*x*x*y*y + 30*y*y*y*y) * coeff; - hrly[47][1] = (-120*x*x*x*y + 120*x*y*y*y) * coeff; - hrly[47][2] = 0.0; - hrly[47][3] = (-30*x*x*x*x + 180*x*x*y*y - 30*y*y*y*y) * coeff; - hrly[47][4] = 0.0; - hrly[47][5] = 0.0; - - //m=-6 : xy(3x^4 - 10x^2*y^2 + 3y^4) - hrly[48][0] = (60*x*x*x*y - 60*x*y*y*y) * coeff; - hrly[48][1] = (15*x*x*x*x - 90*x*x*y*y + 15*y*y*y*y) * coeff; - hrly[48][2] = 0.0; - hrly[48][3] = (-60*x*x*x*y + 60*x*y*y*y) * coeff; - hrly[48][4] = 0.0; - hrly[48][5] = 0.0; - - if (Lmax == 6) return; - - /*************************** - L > 6 - ***************************/ - ModuleBase::WARNING_QUIT("hes_rl_sph_harm","l>6 not implemented!"); - - - return; + hrly.resize( (Lmax+1)*(Lmax+1), std::vector(6) ); + + double radius2 = x*x+y*y+z*z; + double coeff = 0.0; + + //begin calculation + /*************************** + L = 0 + ***************************/ + hrly[0][0] = hrly[0][1] = hrly[0][2] = 0.0; + hrly[0][3] = hrly[0][4] = hrly[0][5] = 0.0; + if (Lmax == 0) return; + + /*************************** + L = 1 + ***************************/ + hrly[1][0] = hrly[1][1] = hrly[1][2] = 0.0; + hrly[1][3] = hrly[1][4] = hrly[1][5] = 0.0; + + hrly[2][0] = hrly[2][1] = hrly[2][2] = 0.0; + hrly[2][3] = hrly[2][4] = hrly[2][5] = 0.0; + + hrly[3][0] = hrly[3][1] = hrly[3][2] = 0.0; + hrly[3][3] = hrly[3][4] = hrly[3][5] = 0.0; + + if (Lmax == 1) return; + + /*************************** + L = 2 + ***************************/ + //m=0 : 3z^2-r^2 + coeff = sqrt(5.0 / ModuleBase::PI) / 4.0; + hrly[4][0] = hrly[4][3] = -2.0 * coeff; + hrly[4][5] = 4.0 * coeff; + hrly[4][1] = hrly[4][2] = hrly[4][4] = 0.0; + + //m=1 : xz + coeff = sqrt(15.0 / ModuleBase::PI) / 2.0; + hrly[5][2] = coeff; + hrly[5][0] = hrly[5][1] = 0.0; + hrly[5][3] = hrly[5][4] = hrly[5][5] = 0.0; + + //m=-1 : yz + hrly[6][4] = coeff; + hrly[6][0] = hrly[6][1] = 0.0; + hrly[6][2] = hrly[6][3] = hrly[6][5] = 0.0; + + //m=-2 : xy + hrly[8][1] = coeff; + hrly[8][0] = hrly[8][2] = 0.0; + hrly[8][3] = hrly[8][4] = hrly[8][5] = 0.0; + + //m=2 : (x^2-y^2) + coeff = sqrt(15.0 / ModuleBase::PI) / 4.0; + hrly[7][0] = 2.0 * coeff; + hrly[7][3] = -2.0 * coeff; + hrly[7][1] = hrly[7][2] = 0.0; + hrly[7][4] = hrly[7][5] = 0.0; + + if (Lmax == 2) return; + + /*************************** + L = 3 + ***************************/ + //m=0 : (5z^3-3zr^2) + coeff = sqrt(7.0 / ModuleBase::PI) / 4.0; + hrly[9][0] = hrly[9][3] = -6.0 * z * coeff; + hrly[9][1] = 0.0; + hrly[9][2] = -6.0 * x * coeff; + hrly[9][4] = -6.0 * y * coeff; + hrly[9][5] = 12.0 * z * coeff; + + //m=1 : x(5z^2-r^2) + coeff = sqrt(21.0 / 2.0 / ModuleBase::PI) / 4.0; + hrly[10][0] = -6.0 * x * coeff; + hrly[10][1] = -2.0 * y * coeff; + hrly[10][2] = 8.0 * z * coeff; + hrly[10][3] = -2.0 * x * coeff; + hrly[10][4] = 0.0; + hrly[10][5] = 8.0 * x * coeff; + + //m=-1 : y(5z^2-r^2) + hrly[11][0] = -2.0 * y * coeff; + hrly[11][1] = -2.0 * x * coeff; + hrly[11][2] = 0.0; + hrly[11][3] = -6.0 * y * coeff; + hrly[11][4] = 8.0 * z * coeff; + hrly[11][5] = 8.0 * y * coeff; + + //m=2 : (x^2-y^2)z + coeff = sqrt(105.0 / ModuleBase::PI) / 4.0; + hrly[12][0] = 2.0 * z * coeff; + hrly[12][1] = 0.0; + hrly[12][2] = 2.0 * x * coeff; + hrly[12][3] = -hrly[12][0]; + hrly[12][4] = -2.0 * y * coeff; + hrly[12][5] = 0.0; + + //m=-2 : xyz + coeff = sqrt(105.0 / ModuleBase::PI) / 2.0; + hrly[13][0] = 0.0; + hrly[13][1] = z * coeff; + hrly[13][2] = y * coeff; + hrly[13][3] = 0.0; + hrly[13][4] = x * coeff; + hrly[13][5] = 0.0; + + //m=3 : x(x^2-3y^2) + coeff = sqrt(35.0 / 2.0 / ModuleBase::PI) / 4.0; + hrly[14][0] = 6.0 * x * coeff; + hrly[14][1] = -6.0 * y * coeff; + hrly[14][2] = 0.0; + hrly[14][3] = -hrly[14][0]; + hrly[14][4] = 0.0; + hrly[14][5] = 0.0; + + //m=-3 : y(3x^2-y^2) + hrly[15][0] = 6.0 * y * coeff; + hrly[15][1] = 6.0 * x * coeff; + hrly[15][2] = 0.0; + hrly[15][3] = -hrly[15][0]; + hrly[15][4] = 0.0; + hrly[15][5] = 0.0; + + if (Lmax == 3) return; + + /*************************** + L = 4 + ***************************/ + //m=0 : (35z^4 - 30z^2r^2 + 3r^4) + coeff = sqrt(1.0 / ModuleBase::PI) * 3.0 / 16.0; + hrly[16][0] = 12.0 * (3.0 * x*x + y*y - 4.0 * z*z) * coeff; + hrly[16][1] = 24.0 * x * y * coeff; + hrly[16][2] = -96.0 * x * z * coeff; + hrly[16][3] = 12.0 * (x*x + 3.0 * y*y - 4.0 * z*z) * coeff; + hrly[16][4] = -96.0 * y * z * coeff; + hrly[16][5] = -48.0 * (x*x + y*y -2.0 * z*z) * coeff; + + //m=1 : x(7z^3 - 3zr^2) + coeff = 3.0 / 4.0 * sqrt(5.0 / 2.0 / ModuleBase::PI); + hrly[17][0] = -18.0 * x * z * coeff; + hrly[17][1] = -6.0 * y * z * coeff; + hrly[17][2] = -3.0 * (3.0 * x*x + y*y - 4.0 * z*z) * coeff; + hrly[17][3] = -6.0 * x * z * coeff; + hrly[17][4] = -6.0 * x * y * coeff; + hrly[17][5] = 24.0 * x * z * coeff; + + //m=-1 : y(7z^3 - 3zr^2) + hrly[18][0] = -6.0 * y * z * coeff; + hrly[18][1] = -6.0 * x * z * coeff; + hrly[18][2] = -6.0 * x * y * coeff; + hrly[18][3] = -18.0 * y * z * coeff; + hrly[18][4] = -3.0 * (x*x + 3.0 * y*y - 4.0 * z*z) * coeff; + hrly[18][5] = 24.0 * y * z * coeff; + + //m=2 : (x^2 - y^2)(7z^2 - r^2) + coeff = 3.0 / 8.0 * sqrt(5.0 / ModuleBase::PI); + hrly[19][0] = -12.0 * (x*x - z*z) * coeff; + hrly[19][1] = 0.0; + hrly[19][2] = 24.0 * x * z * coeff; + hrly[19][3] = 12.0 * (y*y - z*z) * coeff; + hrly[19][4] = -24.0 * y * z * coeff; + hrly[19][5] = 12.0 * (x*x - y*y) * coeff; + + //m=-2 : xy(7z^2 - r^2) + coeff = 3.0 / 4.0 * sqrt(5.0 / ModuleBase::PI); + hrly[20][0] = -6.0 * x * y * coeff; + hrly[20][1] = -3.0 * (x*x + y*y - 2.0 * z*z) * coeff; + hrly[20][2] = 2.0 * y * z * coeff; + hrly[20][3] = hrly[20][0]; + hrly[20][4] = 12.0 * x * z * coeff; + hrly[20][5] = 12.0 * x * y * coeff; + + //m=3 : x(x^2-3y^2)z + coeff = 3.0 / 4.0 * sqrt(35.0 / 2.0 / ModuleBase::PI); + hrly[21][0] = 6.0 * x * z * coeff; + hrly[21][1] = -6.0 * y * z * coeff; + hrly[21][2] = 3.0 * (x*x - y*y) * coeff; + hrly[21][3] = -6.0 * x * z * coeff; + hrly[21][4] = -6.0 * x * y * coeff; + hrly[21][5] = 0.0; + + //m=-3 : y(3x^2-y^2)z + hrly[22][0] = 6.0 * y * z * coeff; + hrly[22][1] = 6.0 * x * z * coeff; + hrly[22][2] = 6.0 * x * y * coeff; + hrly[22][3] = -6.0 * y * z * coeff; + hrly[22][4] = 3.0 * (x*x - y*y) * coeff; + hrly[22][5] = 0.0; + + //m=4 : x^4 + y^4 - 6 x^2y^2 + coeff = 3.0 / 16.0 * sqrt(35.0 / ModuleBase::PI); + hrly[23][0] = 12.0 * (x*x - y*y) * coeff; + hrly[23][1] = -24.0 * x * y * coeff; + hrly[23][2] = 0.0; + hrly[23][3] = -hrly[23][0]; + hrly[23][4] = 0.0; + hrly[23][5] = 0.0; + + //m=-4 : xy(x^2 - y^2) + coeff = 3.0 / 4.0 * sqrt(35.0 / ModuleBase::PI); + hrly[24][0] = 6.0 * x * y * coeff; + hrly[24][1] = 3.0 * (x*x - y*y) * coeff; + hrly[24][2] = 0.0; + hrly[24][3] = -hrly[24][0]; + hrly[24][4] = 0.0; + hrly[24][5] = 0.0; + + if (Lmax == 4) return; + + /*************************** + L = 5 + ***************************/ + //m=0 : (63z^5 - 70z^3*r^2 + 15z*r^4) + coeff = sqrt(11.0 / ModuleBase::PI) / 16.0; + hrly[25][0] = (180*x*x*z + 60*y*y*z - 80*z*z*z) * coeff; + hrly[25][1] = (120*x*y*z) * coeff; + hrly[25][2] = (60*x*x*x + 60*x*y*y - 240*x*z*z) * coeff; + hrly[25][3] = (60*x*x*z + 180*y*y*z - 80*z*z*z) * coeff; + hrly[25][4] = (60*x*x*y + 60*y*y*y - 240*y*z*z) * coeff; + hrly[25][5] = (-240*x*x*z - 240*y*y*z + 160*z*z*z) * coeff; + + //m=1 : x(21z^4 - 14z^2*r^2 + r^4) + coeff = sqrt(165.0 / 2.0 / ModuleBase::PI) / 16.0; + hrly[26][0] = (20*x*x*x + 12*x*y*y - 72*x*z*z) * coeff; + hrly[26][1] = (12*x*x*y + 4*y*y*y - 24*y*z*z) * coeff; + hrly[26][2] = (-72*x*x*z - 24*y*y*z + 32*z*z*z) * coeff; + hrly[26][3] = (4*x*x*x + 12*x*y*y - 24*x*z*z) * coeff; + hrly[26][4] = (-48*x*y*z) * coeff; + hrly[26][5] = (-24*x*x*x - 24*x*y*y + 96*x*z*z) * coeff; + + //m=-1 : y(21z^4 - 14z^2*r^2 + r^4) + hrly[27][0] = (12*x*x*y + 4*y*y*y - 24*y*z*z) * coeff; + hrly[27][1] = (4*x*x*x + 12*x*y*y - 24*x*z*z) * coeff; + hrly[27][2] = (-48*x*y*z) * coeff; + hrly[27][3] = (12*x*x*y + 20*y*y*y - 72*y*z*z) * coeff; + hrly[27][4] = (-24*x*x*z - 72*y*y*z + 32*z*z*z) * coeff; + hrly[27][5] = (-24*x*x*y - 24*y*y*y + 96*y*z*z) * coeff; + + //m=2 : (x^2 - y^2)(3z^3 - z*r^2) + coeff = sqrt(1155.0 / ModuleBase::PI) / 8.0; + hrly[28][0] = (-12*x*x*z + 4*z*z*z) * coeff; + hrly[28][1] = 0.0; + hrly[28][2] = (-4*x*x*x + 12*x*z*z) * coeff; + hrly[28][3] = (12*y*y*z - 4*z*z*z) * coeff; + hrly[28][4] = (4*y*y*y - 12*y*z*z) * coeff; + hrly[28][5] = (12*x*x*z - 12*y*y*z) * coeff; + + //m=-2 : xy(3z^3 - z*r^2) + hrly[29][0] = (-6*x*y*z) * coeff; + hrly[29][1] = (-3*x*x*z - 3*y*y*z + 2*z*z*z) * coeff; + hrly[29][2] = (-3*x*x*y - y*y*y + 6*y*z*z) * coeff; + hrly[29][3] = (-6*x*y*z) * coeff; + hrly[29][4] = (-x*x*x - 3*x*y*y + 6*x*z*z) * coeff; + hrly[29][5] = (12*x*y*z) * coeff; + + //m=3 : x(x^2 - 3y^2)(9z^2 - r^2) + coeff = sqrt(385.0 / 2.0 / ModuleBase::PI) / 16.0; + hrly[30][0] = (-20*x*x*x + 12*x*y*y + 48*x*z*z) * coeff; + hrly[30][1] = (12*x*x*y + 12*y*y*y - 48*y*z*z) * coeff; + hrly[30][2] = (48*x*x*z - 48*y*y*z) * coeff; + hrly[30][3] = (4*x*x*x + 36*x*y*y - 48*x*z*z) * coeff; + hrly[30][4] = (-96*x*y*z) * coeff; + hrly[30][5] = (16*x*x*x - 48*x*y*y) * coeff; + + //m=-3 : y(3x^2 - y^2)(9z^2 - r^2) + hrly[31][0] = (-36*x*x*y - 4*y*y*y + 48*y*z*z) * coeff; + hrly[31][1] = (-12*x*x*x - 12*x*y*y + 48*x*z*z) * coeff; + hrly[31][2] = (96*x*y*z) * coeff; + hrly[31][3] = (-12*x*x*y + 20*y*y*y - 48*y*z*z) * coeff; + hrly[31][4] = (48*x*x*z - 48*y*y*z) * coeff; + hrly[31][5] = (48*x*x*y - 16*y*y*y) * coeff; + + //m=4 : (x^4 - 6x^2*y^2 + y^4) * z + coeff = sqrt(385.0 / ModuleBase::PI) / 16.0; + hrly[32][0] = (12*x*x*z - 12*y*y*z) * coeff; + hrly[32][1] = (-24*x*y*z) * coeff; + hrly[32][2] = (4*x*x*x - 12*x*y*y) * coeff; + hrly[32][3] = (-12*x*x*z + 12*y*y*z) * coeff; + hrly[32][4] = (-12*x*x*y + 4*y*y*y) * coeff; + hrly[32][5] = 0.0; + + //m=-4 : xy(x^2 - y^2) * z + hrly[33][0] = (6*x*y*z) * coeff; + hrly[33][1] = (3*x*x*z - 3*y*y*z) * coeff; + hrly[33][2] = (3*x*x*y - y*y*y) * coeff; + hrly[33][3] = (-6*x*y*z) * coeff; + hrly[33][4] = (x*x*x - 3*x*y*y) * coeff; + hrly[33][5] = 0.0; + + //m=5 : x(x^4 - 10x^2*y^2 + 5y^4) + coeff = sqrt(77.0 / 2.0 / ModuleBase::PI) / 16.0; + hrly[34][0] = (20.0 * x*x*x - 60.0 * x * y*y) * coeff; + hrly[34][1] = (-60.0 * x*x * y + 20.0 * y*y*y) * coeff; + hrly[34][2] = 0.0; + hrly[34][3] = (-20.0 * x*x*x + 60.0 * x * y*y) * coeff; + hrly[34][4] = 0.0; + hrly[34][5] = 0.0; + + //m=-5 : y(5x^4 - 10x^2*y^2 + y^4) + hrly[35][0] = (60.0 * x*x * y - 20.0 * y*y*y) * coeff; + hrly[35][1] = (20.0 * x*x*x - 60.0 * x * y*y) * coeff; + hrly[35][2] = 0.0; + hrly[35][3] = (-60.0 * x*x * y + 20.0 * y*y*y) * coeff; + hrly[35][4] = 0.0; + hrly[35][5] = 0.0; + + if (Lmax == 5) return; + + /*************************** + L = 6 + ***************************/ + //m=0 : (231z^6 - 315z^4*r^2 + 105z^2*r^4 - 5r^6) + coeff = sqrt(13.0 / ModuleBase::PI) / 32.0; + hrly[36][0] = (-150*x*x*x*x - 180*x*x*y*y + 1080*x*x*z*z - 30*y*y*y*y + 360*y*y*z*z - 240*z*z*z*z) * coeff; + hrly[36][1] = (-120*x*x*x*y - 120*x*y*y*y + 720*x*y*z*z) * coeff; + hrly[36][2] = (720*x*x*x*z + 720*x*y*y*z - 960*x*z*z*z) * coeff; + hrly[36][3] = (-30*x*x*x*x - 180*x*x*y*y + 360*x*x*z*z - 150*y*y*y*y + 1080*y*y*z*z - 240*z*z*z*z) * coeff; + hrly[36][4] = (720*x*x*y*z + 720*y*y*y*z - 960*y*z*z*z) * coeff; + hrly[36][5] = (180*x*x*x*x + 360*x*x*y*y - 1440*x*x*z*z + 180*y*y*y*y - 1440*y*y*z*z + 480*z*z*z*z) * coeff; + + //m=1 : x(33z^5 - 30z^3*r^2 + 5z*r^4) + coeff = sqrt(273.0 / 2.0 / ModuleBase::PI) / 16.0; + hrly[37][0] = (100*x*x*x*z + 60*x*y*y*z - 120*x*z*z*z) * coeff; + hrly[37][1] = (60*x*x*y*z + 20*y*y*y*z - 40*y*z*z*z) * coeff; + hrly[37][2] = (25*x*x*x*x + 30*x*x*y*y - 180*x*x*z*z + 5*y*y*y*y - 60*y*y*z*z + 40*z*z*z*z) * coeff; + hrly[37][3] = (20*x*x*x*z + 60*x*y*y*z - 40*x*z*z*z) * coeff; + hrly[37][4] = (20*x*x*x*y + 20*x*y*y*y - 120*x*y*z*z) * coeff; + hrly[37][5] = (-120*x*x*x*z - 120*x*y*y*z + 160*x*z*z*z) * coeff; + + //m=-1 : y(33z^5 - 30z^3*r^2 + 5z*r^4) + hrly[38][0] = (60*x*x*y*z + 20*y*y*y*z - 40*y*z*z*z) * coeff; + hrly[38][1] = (20*x*x*x*z + 60*x*y*y*z - 40*x*z*z*z) * coeff; + hrly[38][2] = (20*x*x*x*y + 20*x*y*y*y - 120*x*y*z*z) * coeff; + hrly[38][3] = (60*x*x*y*z + 100*y*y*y*z - 120*y*z*z*z) * coeff; + hrly[38][4] = (5*x*x*x*x + 30*x*x*y*y - 60*x*x*z*z + 25*y*y*y*y - 180*y*y*z*z + 40*z*z*z*z) * coeff; + hrly[38][5] = (-120*x*x*y*z - 120*y*y*y*z + 160*y*z*z*z) * coeff; + + //m=2 : (x^2 - y^2)(33z^4 - 18z^2*r^2 + r^4) + coeff = sqrt(1365.0 / ModuleBase::PI) / 32.0; + hrly[39][0] = (30*x*x*x*x + 12*x*x*y*y - 192*x*x*z*z - 2*y*y*y*y + 32*z*z*z*z) * coeff; + hrly[39][1] = (8*x*x*x*y - 8*x*y*y*y) * coeff; + hrly[39][2] = (-128*x*x*x*z + 128*x*z*z*z) * coeff; + hrly[39][3] = (2*x*x*x*x - 12*x*x*y*y - 30*y*y*y*y + 192*y*y*z*z - 32*z*z*z*z) * coeff; + hrly[39][4] = (128*y*y*y*z - 128*y*z*z*z) * coeff; + hrly[39][5] = (-32*x*x*x*x + 192*x*x*z*z + 32*y*y*y*y - 192*y*y*z*z) * coeff; + + //m=-2 : xy(33z^4 - 18z^2*r^2 + r^4) + hrly[40][0] = (20*x*x*x*y + 12*x*y*y*y - 96*x*y*z*z) * coeff; + hrly[40][1] = (20*x*x*x*x + 36*x*x*y*y - 96*x*x*z*z + 20*y*y*y*y - 96*y*y*z*z + 32*z*z*z*z) * coeff; + hrly[40][2] = (-96*x*x*y*z - 32*y*y*y*z + 64*y*z*z*z) * coeff; + hrly[40][3] = (12*x*x*x*y + 20*x*y*y*y - 96*x*y*z*z) * coeff; + hrly[40][4] = (-32*x*x*x*z - 96*x*y*y*z + 64*x*z*z*z) * coeff; + hrly[40][5] = (-32*x*x*x*y - 32*x*y*y*y + 192*x*y*z*z) * coeff; + + //m=3 : x(x^2 - 3y^2)(11z^3 - 3z*r^2) + coeff = sqrt(1365.0 / ModuleBase::PI) / 16.0; + hrly[41][0] = (-60*x*x*x*z + 36*x*y*y*z + 48*x*z*z*z) * coeff; + hrly[41][1] = (36*x*x*y*z + 36*y*y*y*z - 48*y*z*z*z) * coeff; + hrly[41][2] = (-30*x*x*x*x + 36*x*x*y*y + 72*x*x*z*z + 18*y*y*y*y - 72*y*y*z*z) * coeff; + hrly[41][3] = (12*x*x*x*z + 108*x*y*y*z - 48*x*z*z*z) * coeff; + hrly[41][4] = (12*x*x*x*y + 36*x*y*y*y - 144*x*y*z*z) * coeff; + hrly[41][5] = (48*x*x*x*z - 144*x*y*y*z) * coeff; + + //m=-3 : y(3x^2 - y^2)(11z^3 - 3z*r^2) + hrly[42][0] = (-108*x*x*y*z - 12*y*y*y*z + 48*y*z*z*z) * coeff; + hrly[42][1] = (-36*x*x*x*z - 36*x*y*y*z + 48*x*z*z*z) * coeff; + hrly[42][2] = (-36*x*x*x*y - 12*x*y*y*y + 144*x*y*z*z) * coeff; + hrly[42][3] = (-36*x*x*y*z + 60*y*y*y*z - 48*y*z*z*z) * coeff; + hrly[42][4] = (-18*x*x*x*x - 36*x*x*y*y + 72*x*x*z*z + 30*y*y*y*y - 72*y*y*z*z) * coeff; + hrly[42][5] = (144*x*x*y*z - 48*y*y*y*z) * coeff; + + //m=4 : (x^4 - 6x^2*y^2 + y^4)(11z^2 - r^2) + coeff = sqrt(91.0 / ModuleBase::PI) / 32.0; + hrly[43][0] = (-30*x*x*x*x + 60*x*x*y*y + 120*x*x*z*z + 10*y*y*y*y - 120*y*y*z*z) * coeff; + hrly[43][1] = (40*x*x*x*y + 40*x*y*y*y - 240*x*y*z*z) * coeff; + hrly[43][2] = (80*x*x*x*z - 240*x*y*y*z) * coeff; + hrly[43][3] = (10*x*x*x*x + 60*x*x*y*y - 120*x*x*z*z - 30*y*y*y*y + 120*y*y*z*z) * coeff; + hrly[43][4] = (-240*x*x*y*z + 80*y*y*y*z) * coeff; + hrly[43][5] = (20*x*x*x*x - 120*x*x*y*y + 20*y*y*y*y) * coeff; + + //m=-4 : xy(x^2 - y^2)(11z^2 - r^2) + hrly[44][0] = (-20*x*x*x*y + 60*x*y*z*z) * coeff; + hrly[44][1] = (-5*x*x*x*x + 30*x*x*z*z + 5*y*y*y*y - 30*y*y*z*z) * coeff; + hrly[44][2] = (60*x*x*y*z - 20*y*y*y*z) * coeff; + hrly[44][3] = (20*x*y*y*y - 60*x*y*z*z) * coeff; + hrly[44][4] = (20*x*x*x*z - 60*x*y*y*z) * coeff; + hrly[44][5] = (20*x*x*x*y - 20*x*y*y*y) * coeff; + + //m=5 : x(x^4 - 10x^2*y^2 + 5y^4) * z + coeff = sqrt(1001.0 / 2.0 / ModuleBase::PI) / 16.0; + hrly[45][0] = (20*x*x*x*z - 60*x*y*y*z) * coeff; + hrly[45][1] = (-60*x*x*y*z + 20*y*y*y*z) * coeff; + hrly[45][2] = (5*x*x*x*x - 30*x*x*y*y + 5*y*y*y*y) * coeff; + hrly[45][3] = (-20*x*x*x*z + 60*x*y*y*z) * coeff; + hrly[45][4] = (-20*x*x*x*y + 20*x*y*y*y) * coeff; + hrly[45][5] = 0.0; + + //m=-5 : y(5x^4 - 10x^2*y^2 + y^4) * z + hrly[46][0] = (60*x*x*y*z - 20*y*y*y*z) * coeff; + hrly[46][1] = (20*x*x*x*z - 60*x*y*y*z) * coeff; + hrly[46][2] = (20*x*x*x*y - 20*x*y*y*y) * coeff; + hrly[46][3] = (-60*x*x*y*z + 20*y*y*y*z) * coeff; + hrly[46][4] = (5*x*x*x*x - 30*x*x*y*y + 5*y*y*y*y) * coeff; + hrly[46][5] = 0.0; + + //m=6 : (x^6 - 15x^4*y^2 + 15x^2*y^4 - y^6) + coeff = sqrt(3003.0 / ModuleBase::PI) / 32.0; + hrly[47][0] = (30*x*x*x*x - 180*x*x*y*y + 30*y*y*y*y) * coeff; + hrly[47][1] = (-120*x*x*x*y + 120*x*y*y*y) * coeff; + hrly[47][2] = 0.0; + hrly[47][3] = (-30*x*x*x*x + 180*x*x*y*y - 30*y*y*y*y) * coeff; + hrly[47][4] = 0.0; + hrly[47][5] = 0.0; + + //m=-6 : xy(3x^4 - 10x^2*y^2 + 3y^4) + hrly[48][0] = (60*x*x*x*y - 60*x*y*y*y) * coeff; + hrly[48][1] = (15*x*x*x*x - 90*x*x*y*y + 15*y*y*y*y) * coeff; + hrly[48][2] = 0.0; + hrly[48][3] = (-60*x*x*x*y + 60*x*y*y*y) * coeff; + hrly[48][4] = 0.0; + hrly[48][5] = 0.0; + + if (Lmax == 6) return; + + /*************************** + L > 6 + ***************************/ + ModuleBase::WARNING_QUIT("hes_rl_sph_harm","l>6 not implemented!"); + + + return; } void Ylm::set_coefficients (){}; /* void Ylm::test1 (void) { - ModuleBase::Vector3 R (20.0, 0.0, 0.0); - double xdr = R.x/R.norm(); - double ydr = R.y/R.norm(); - double zdr = R.z/R.norm(); - const int L = 9; - const double rl = std::pow( R.norm(), L); - std::cout << " rl=" << rl << std::endl; - Ylm::set_coefficients(); - - int nu = 100; - - // Peize Lin change rlya 2016-08-26 - std::vector rlya; - double rlyb[400]; - ZEROS( rlyb, 400); + ModuleBase::Vector3 R (20.0, 0.0, 0.0); + double xdr = R.x/R.norm(); + double ydr = R.y/R.norm(); + double zdr = R.z/R.norm(); + const int L = 9; + const double rl = std::pow( R.norm(), L); + std::cout << " rl=" << rl << std::endl; + Ylm::set_coefficients(); + + int nu = 100; + + // Peize Lin change rlya 2016-08-26 + std::vector rlya; + double rlyb[400]; + ZEROS( rlyb, 400); // Ylm::sph_harm (9, xdr, ydr, zdr, rlya); - Ylm::rl_sph_harm (L, xdr, ydr, zdr, rlya); + Ylm::rl_sph_harm (L, xdr, ydr, zdr, rlya); // Ylm::rlylm (10, R.x, R.y, R.z, rlyb); - Ylm::get_ylm_real (L+1, R, rlyb); - - for (int i=0; i < nu; i++) - { - // std::cout << "\ni= " << i << " rlya = " << rlya[i] << " rlyb = " << rlyb[i] << std::endl; - double diff = fabs(rlya[i]-rlyb[i]); - if (diff > 1e-8) - { - std::cout << "Ylm::test1, error is too large!" << std::endl; - //WARNING_QUIT ("Ylm::test1","error is too large!"); - exit(0); - } - } - return; + Ylm::get_ylm_real (L+1, R, rlyb); + + for (int i=0; i < nu; i++) + { + // std::cout << "\ni= " << i << " rlya = " << rlya[i] << " rlyb = " << rlyb[i] << std::endl; + double diff = fabs(rlya[i]-rlyb[i]); + if (diff > 1e-8) + { + std::cout << "Ylm::test1, error is too large!" << std::endl; + //WARNING_QUIT ("Ylm::test1","error is too large!"); + exit(0); + } + } + return; } */ /* void Ylm::test2 (void) { - ModuleBase::Vector3 R (0.1,-0.2,0.5); - Ylm::set_coefficients(); - - //int nu = 100; - - std::vector rlya; - double rlyb[400]; - - std::vector> grlya; - double grlyb[400][3]; - - Ylm::grad_rl_sph_harm (9, R.x, R.y, R.z, rlya, grlya); - Ylm::rlylm (10, R.x, R.y, R.z, rlyb, grlyb); - - for (int i = 0; i < 100; i++) - { - double diffx = fabs(grlya[i][2]-grlyb[i][2]); - if (diffx > 1e-8) - { - std::cout << "Ylm::test2, Large error in Direv X!" << std::endl; - //WARNING_QUIT ("Ylm::test2","Large error in Direv X!"); - exit(0); - } - } - return; + ModuleBase::Vector3 R (0.1,-0.2,0.5); + Ylm::set_coefficients(); + + //int nu = 100; + + std::vector rlya; + double rlyb[400]; + + std::vector> grlya; + double grlyb[400][3]; + + Ylm::grad_rl_sph_harm (9, R.x, R.y, R.z, rlya, grlya); + Ylm::rlylm (10, R.x, R.y, R.z, rlyb, grlyb); + + for (int i = 0; i < 100; i++) + { + double diffx = fabs(grlya[i][2]-grlyb[i][2]); + if (diffx > 1e-8) + { + std::cout << "Ylm::test2, Large error in Direv X!" << std::endl; + //WARNING_QUIT ("Ylm::test2","Large error in Direv X!"); + exit(0); + } + } + return; } */ void Ylm::rlylm ( - const int Lmax, //max momentum of l + 1 - const double x, - const double y, - const double z, - double rly[], - double grly[][3] + const int Lmax, //max momentum of l + 1 + const double x, + const double y, + const double z, + double rly[], + double grly[][3] ) { - int MaxL = Lmax - 1; - - assert(MaxL >= 0); - - //get xy_dependence - assert(MaxL <= 19); - - double Am[20]; - double Bm[20]; - double Gx_Am[20]; - double Gx_Bm[20]; - double Gy_Am[20]; - double Gy_Bm[20]; - - ZEROS(Am, 20); - ZEROS(Bm, 20); - ZEROS(Gx_Am, 20); - ZEROS(Gy_Am, 20); - - double x2, x3, x4, x5; - double y2, y3, y4, y5; - - x2 = x * x; - x3 = x2 * x; - x4 = x3 * x; - x5 = x4 * x; - - y2 = y * y; - y3 = y2 * y; - y4 = y3 * y; - y5 = y4 * y; - - //x-y dependence - //Am - //Bm - for(int im = 0; im < MaxL+1; im++) - { - if(im == 0) - { - Am[0] = 1.0; - Bm[0] = 0.0; - - Gx_Am[0] = 0.0; - Gy_Am[0] = 0.0; - - Gx_Bm[0] = 0.0; - Gy_Bm[0] = 0.0; - } - else if(im == 1) - { - Am[1] = x; - Bm[1] = y; - - Gx_Am[1] = 1.0; - Gy_Am[1] = 0.0; - - Gx_Bm[1] = 0.0; - Gy_Bm[1] = 1.0; - } - else if(im == 2) - { - Am[2] = x2- y2; - Bm[2] = 2.0 * x * y; - - Gx_Am[2] = 2.0 * x; - Gy_Am[2] = -2.0 * y; - - Gx_Bm[2] = 2.0 * y; - Gy_Bm[2] = 2.0 * x; - } - else if(im == 3) - { - Am[3] = x3 - 3.0 * x * y2; - Bm[3] = 3.0 * x2 * y - y3; - - Gx_Am[3] = 3.0 * (x2 - y2); - Gy_Am[3] = -6.0 * x * y; - - Gx_Bm[3] = 6.0 * x * y; - Gy_Bm[3] = 3.0 * (x2 - y2); - } - else if(im == 4) - { - Am[4] = x4 - 6.0 * x2 * y2 + y4; - Bm[4] = 4.0 * (x3 * y - x * y3); - - Gx_Am[4] = 4.0 * x3 - 12.0 * x * y2; - Gy_Am[4] = -12.0 * x2 * y + 4.0 * y3; - - Gx_Bm[4] = 12.0 * x2 * y - 4.0 * y3; - Gy_Bm[4] = 4.0 * x3 - 12.0 * x * y2; - } - else if(im == 5) - { - Am[5] = x5 - 10.0 * x3 * y2 + 5.0 * x * y4; - Bm[5] = 5.0 * x4 * y - 10.0 * x2 * y3 + y5; - - Gx_Am[5] = 5.0 * x4 - 30.0 * x2 * y2 + 5.0 * y4; - Gy_Am[5] = 20.0 * (x * y3 - x3 * y); - - Gx_Bm[5] = 20.0 * (x3 * y - x * y3); - Gy_Bm[5] = 5.0 * x4 - 30.0 * x2 * y2 + 5.0 * y4; - } - else - { - for(int ip = 0; ip <= im; ip++) - { - double aux = Fact(im) / Fact(ip) / Fact(im - ip); - Am[im] += aux * pow(x, ip) * pow(y, im-ip) * cos( (im-ip) * ModuleBase::PI / 2.0 ); - Bm[im] += aux * pow(x, ip) * pow(y, im-ip) * sin( (im-ip) * ModuleBase::PI / 2.0 ); - - if(ip > 0) - { - Gx_Am[im] += aux * ip * pow(x, ip-1) * pow(y, im-ip) * cos( (im-ip) * ModuleBase::PI / 2.0 ); - Gx_Bm[im] += aux * ip * pow(x, ip-1) * pow(y, im-ip) * sin( (im-ip) * ModuleBase::PI / 2.0 ); - } - - if(ip < im) - { - Gy_Am[im] += aux * pow(x, ip) * (im - ip) * pow(y, im-ip-1) * cos( (im-ip) * ModuleBase::PI / 2.0 ); - Gy_Bm[im] += aux * pow(x, ip) * (im - ip) * pow(y, im-ip-1) * sin( (im-ip) * ModuleBase::PI / 2.0 ); - } - } - } - } - - //z dependence - double zdep[20][20]; - double Gx_dep[20][20]; - double Gy_dep[20][20]; - double Gz_dep[20][20]; - - for(int il = 0; il < 20; il++) - { - ZEROS(zdep[il], 20); - ZEROS(Gx_dep[il], 20); - ZEROS(Gy_dep[il], 20); - ZEROS(Gz_dep[il], 20); - } - - double z2 = z * z; - double z3 = z2 * z; - double z4 = z3 * z; - //double z5 = z4 * z; - - double r = sqrt(x*x + y*y + z*z); - double r2 = r * r; - double r3 = r2 * r; - double r4 = r3 * r; - - for(int il = 0; il < MaxL+1; il++) - { - if(il == 0) - { - zdep[0][0] = 1.0; - } - else if(il == 1) - { - zdep[1][0] = z; - zdep[1][1] = 1.0; - - Gz_dep[1][0] = 1.0; - } - else if(il == 2) - { - zdep[2][0] = 0.5 * (3.0 * z2 - r2); - Gx_dep[2][0] = -x; - Gy_dep[2][0] = -y; - Gz_dep[2][0] = 2.0 * z; - - zdep[2][1] = sqrt(3.0) * z; - Gz_dep[2][1] = sqrt(3.0); - - zdep[2][2] = sqrt(3.0) * 0.5; - } - else if(il == 3) - { - zdep[3][0] = 2.5 * z3 - 1.5 * z * r2; - Gx_dep[3][0] = -3.0 * x * z; - Gy_dep[3][0] = -3.0 * y * z; - Gz_dep[3][0] = 1.5 * (3.0 * z2 - r2); - - zdep[3][1] = 0.25 * sqrt(6.0) * (5.0 * z2 - r2); - Gx_dep[3][1] = -0.5 * sqrt(6.0) * x; - Gy_dep[3][1] = -0.5 * sqrt(6.0) * y; - Gz_dep[3][1] = sqrt(6.0) * 2.0 * z; - - zdep[3][2] = 0.5 * sqrt(15.0) * z; - Gz_dep[3][2] = 0.5 * sqrt(15.0); - - zdep[3][3] = 0.25 * sqrt(10.0); - } - else if(il == 4) - { - zdep[4][0] = 0.125 * (35.0 * z4 - 30.0 * r2 * z2 + 3.0 * r4); - Gx_dep[4][0] = -7.5 * x * z2 + 1.5 * x * r2; - Gy_dep[4][0] = -7.5 * y * z2 + 1.5 * y * r2; - Gz_dep[4][0] = 10.0 * z3 - 6.0 * r2 * z; - - zdep[4][1] = sqrt(10.0) * 0.25 * z * (7.0 * z2 - 3.0 * r2); - Gx_dep[4][1] = -1.5 * sqrt(10.0) * x * z; - Gy_dep[4][1] = -1.5 * sqrt(10.0) * y * z; - Gz_dep[4][1] = 0.75 * sqrt(10.0) * (5.0 * z2 - r2); - - zdep[4][2] = sqrt(5.0) * 0.25 * (7.0 * z2 - r2); - Gx_dep[4][2] = -0.5 * sqrt(5.0) * x; - Gy_dep[4][2] = -0.5 * sqrt(5.0) * y; - Gz_dep[4][2] = 3.0 * sqrt(5.0) * z; - - zdep[4][3] = sqrt(70.0) * 0.25 * z; - Gz_dep[4][3] = 0.25 * sqrt(70.0); - - zdep[4][4] = sqrt(35.0) * 0.125; - } - else if(il == 5) - { - zdep[5][0] = 0.125 * z *( 63.0 * z4 - 70.0 * z2 * r2 + 15.0 * r4); - Gx_dep[5][0] = -17.5 * x * z3 + 7.5 * x * z * r2; - Gy_dep[5][0] = -17.5 * y * z3 + 7.5 * y * z * r2; - Gz_dep[5][0] = 175.0 * 0.125 * z4 + 15.0 * 0.125 * r4 - 150.0 * 0.125 * r2 * z2; - - zdep[5][1] = 0.125 * sqrt(15.0) * (21.0 * z4 - 14.0 * z2 * r2 + r4); - Gx_dep[5][1] = -3.5 * sqrt(15.0) * x * z2 + 0.5 * sqrt(15.0) * x * r2; - Gy_dep[5][1] = -3.5 * sqrt(15.0) * y * z2 + 0.5 * sqrt(15.0) * y * r2; - Gz_dep[5][1] = 7.0 * sqrt(15.0) * z3 - 3.0 * sqrt(15.0) * r2 * z; - - zdep[5][2] = 0.25 * sqrt(105.0) * z * (3.0 * z2 - r2); - Gx_dep[5][2] = -0.5 * sqrt(105.0) * x * z; - Gy_dep[5][2] = -0.5 * sqrt(105.0) * y * z; - Gz_dep[5][2] = 0.25 * sqrt(105.0) * (7.0 * z2 - r2); - - zdep[5][3] = 0.0625 * sqrt(70.0) * (9.0 * z2 - r2); - Gx_dep[5][3] = -0.125 * sqrt(70.0) * x; - Gy_dep[5][3] = -0.125 * sqrt(70.0) * y; - Gz_dep[5][3] = sqrt(70.0) * z; - - zdep[5][4] = 0.375 * sqrt(35.0) * z; - Gz_dep[5][4] = 0.375 * sqrt(35.0); - - zdep[5][5] = 0.1875 * sqrt(14.0); - } - else - { - for(int im = 0; im <= il; im++) - { - int kmax = static_cast( (il - im) / 2 ); - for(int ik = 0; ik <= kmax; ik++) - { - int twok = 2 * ik; - - double gamma = 0.0; - double aux0, aux1, aux2, aux3; - - aux0 = pow(-1.0, ik) * pow(2.0, -il); - aux1 = Fact(il) / Fact(ik) / Fact(il-ik); - aux2 = Fact(2*il - twok) / Fact(il) / Fact(il - twok); - aux3 = Fact(il - twok) / Fact(il - twok - im); - - gamma = aux0 * aux1 * aux2 * aux3; - - assert(il - twok - im >= 0); - zdep[il][im] += pow(r, twok) * pow(z, il-twok-im) * gamma; - - if(ik > 0) - { - Gx_dep[il][im] += (ik * pow(r2, ik-1) * 2.0 * x) * pow(z, il-twok-im) * gamma; - Gy_dep[il][im] += (ik * pow(r2, ik-1) * 2.0 * y) * pow(z, il-twok-im) * gamma; - } - - if(ik == 0) - { - if(il > im) - { - Gz_dep[il][im] += (il-im) * pow(z, il-im-1) * gamma; - } - } - else - { - if(il - twok - im == 0) - { - Gz_dep[il][im] += gamma * ik * pow(r2, ik-1) * 2.0 * z; - } - else - { - Gz_dep[il][im] += gamma * (ik * pow(r2, ik-1) * 2.0 * z * pow(z, il-twok-im) - + pow(r, twok) * (il-twok-im) * pow(z, il-twok-im-1)); - } - } - } - - if(im >= 1) - { - zdep[il][im] *= sqrt(2 * Fact(il - im) / Fact(il + im)); - Gx_dep[il][im] *= sqrt(2 * Fact(il - im) / Fact(il + im)); - Gy_dep[il][im] *= sqrt(2 * Fact(il - im) / Fact(il + im)); - Gz_dep[il][im] *= sqrt(2 * Fact(il - im) / Fact(il + im)); - - } - } - } - } - - //calc - int ic = 0; - for(int il = 0; il <= MaxL; il++) - { - double fac = sqrt( (2.0 * il + 1.0) / ModuleBase::FOUR_PI ); - - //m=0 - rly[ic] = Am[0] * zdep[il][0] * fac; - grly[ic][0] = (Gx_dep[il][0] * Am[0] + zdep[il][0] * Gx_Am[0]) * fac; - grly[ic][1] = (Gy_dep[il][0] * Am[0] + zdep[il][0] * Gy_Am[0]) * fac; - grly[ic][2] = Gz_dep[il][0] * Am[0] * fac; - - ic++; - - //m ! = 0 - for(int im = 1; im <= il; im++) - { - //m>0 - rly[ic] = Am[im] * zdep[il][im] * pow(-1.0, im) * fac; - grly[ic][0] = (Gx_dep[il][im] * Am[im] + zdep[il][im] * Gx_Am[im]) * pow(-1.0, im) * fac; - grly[ic][1] = (Gy_dep[il][im] * Am[im] + zdep[il][im] * Gy_Am[im]) * pow(-1.0, im) * fac; - grly[ic][2] = Gz_dep[il][im] * Am[im] * pow(-1.0, im) * fac; - - ic++; - - //m<0 - rly[ic] = Bm[im] * zdep[il][im] * pow(-1.0, im) * fac; - grly[ic][0] = (Gx_dep[il][im] * Bm[im] + zdep[il][im] * Gx_Bm[im]) * pow(-1.0, im) * fac; - grly[ic][1] = (Gy_dep[il][im] * Bm[im] + zdep[il][im] * Gy_Bm[im]) * pow(-1.0, im) * fac; - grly[ic][2] = Gz_dep[il][im] * Bm[im] * pow(-1.0, im) * fac; - - ic++; - } - } - - return; + int MaxL = Lmax - 1; + + assert(MaxL >= 0); + + //get xy_dependence + assert(MaxL <= 19); + + double Am[20]; + double Bm[20]; + double Gx_Am[20]; + double Gx_Bm[20]; + double Gy_Am[20]; + double Gy_Bm[20]; + + ZEROS(Am, 20); + ZEROS(Bm, 20); + ZEROS(Gx_Am, 20); + ZEROS(Gy_Am, 20); + + double x2, x3, x4, x5; + double y2, y3, y4, y5; + + x2 = x * x; + x3 = x2 * x; + x4 = x3 * x; + x5 = x4 * x; + + y2 = y * y; + y3 = y2 * y; + y4 = y3 * y; + y5 = y4 * y; + + //x-y dependence + //Am + //Bm + for(int im = 0; im < MaxL+1; im++) + { + if(im == 0) + { + Am[0] = 1.0; + Bm[0] = 0.0; + + Gx_Am[0] = 0.0; + Gy_Am[0] = 0.0; + + Gx_Bm[0] = 0.0; + Gy_Bm[0] = 0.0; + } + else if(im == 1) + { + Am[1] = x; + Bm[1] = y; + + Gx_Am[1] = 1.0; + Gy_Am[1] = 0.0; + + Gx_Bm[1] = 0.0; + Gy_Bm[1] = 1.0; + } + else if(im == 2) + { + Am[2] = x2- y2; + Bm[2] = 2.0 * x * y; + + Gx_Am[2] = 2.0 * x; + Gy_Am[2] = -2.0 * y; + + Gx_Bm[2] = 2.0 * y; + Gy_Bm[2] = 2.0 * x; + } + else if(im == 3) + { + Am[3] = x3 - 3.0 * x * y2; + Bm[3] = 3.0 * x2 * y - y3; + + Gx_Am[3] = 3.0 * (x2 - y2); + Gy_Am[3] = -6.0 * x * y; + + Gx_Bm[3] = 6.0 * x * y; + Gy_Bm[3] = 3.0 * (x2 - y2); + } + else if(im == 4) + { + Am[4] = x4 - 6.0 * x2 * y2 + y4; + Bm[4] = 4.0 * (x3 * y - x * y3); + + Gx_Am[4] = 4.0 * x3 - 12.0 * x * y2; + Gy_Am[4] = -12.0 * x2 * y + 4.0 * y3; + + Gx_Bm[4] = 12.0 * x2 * y - 4.0 * y3; + Gy_Bm[4] = 4.0 * x3 - 12.0 * x * y2; + } + else if(im == 5) + { + Am[5] = x5 - 10.0 * x3 * y2 + 5.0 * x * y4; + Bm[5] = 5.0 * x4 * y - 10.0 * x2 * y3 + y5; + + Gx_Am[5] = 5.0 * x4 - 30.0 * x2 * y2 + 5.0 * y4; + Gy_Am[5] = 20.0 * (x * y3 - x3 * y); + + Gx_Bm[5] = 20.0 * (x3 * y - x * y3); + Gy_Bm[5] = 5.0 * x4 - 30.0 * x2 * y2 + 5.0 * y4; + } + else + { + for(int ip = 0; ip <= im; ip++) + { + double aux = Fact(im) / Fact(ip) / Fact(im - ip); + Am[im] += aux * pow(x, ip) * pow(y, im-ip) * cos( (im-ip) * ModuleBase::PI / 2.0 ); + Bm[im] += aux * pow(x, ip) * pow(y, im-ip) * sin( (im-ip) * ModuleBase::PI / 2.0 ); + + if(ip > 0) + { + Gx_Am[im] += aux * ip * pow(x, ip-1) * pow(y, im-ip) * cos( (im-ip) * ModuleBase::PI / 2.0 ); + Gx_Bm[im] += aux * ip * pow(x, ip-1) * pow(y, im-ip) * sin( (im-ip) * ModuleBase::PI / 2.0 ); + } + + if(ip < im) + { + Gy_Am[im] += aux * pow(x, ip) * (im - ip) * pow(y, im-ip-1) * cos( (im-ip) * ModuleBase::PI / 2.0 ); + Gy_Bm[im] += aux * pow(x, ip) * (im - ip) * pow(y, im-ip-1) * sin( (im-ip) * ModuleBase::PI / 2.0 ); + } + } + } + } + + //z dependence + double zdep[20][20]; + double Gx_dep[20][20]; + double Gy_dep[20][20]; + double Gz_dep[20][20]; + + for(int il = 0; il < 20; il++) + { + ZEROS(zdep[il], 20); + ZEROS(Gx_dep[il], 20); + ZEROS(Gy_dep[il], 20); + ZEROS(Gz_dep[il], 20); + } + + double z2 = z * z; + double z3 = z2 * z; + double z4 = z3 * z; + //double z5 = z4 * z; + + double r = sqrt(x*x + y*y + z*z); + double r2 = r * r; + double r3 = r2 * r; + double r4 = r3 * r; + + for(int il = 0; il < MaxL+1; il++) + { + if(il == 0) + { + zdep[0][0] = 1.0; + } + else if(il == 1) + { + zdep[1][0] = z; + zdep[1][1] = 1.0; + + Gz_dep[1][0] = 1.0; + } + else if(il == 2) + { + zdep[2][0] = 0.5 * (3.0 * z2 - r2); + Gx_dep[2][0] = -x; + Gy_dep[2][0] = -y; + Gz_dep[2][0] = 2.0 * z; + + zdep[2][1] = sqrt(3.0) * z; + Gz_dep[2][1] = sqrt(3.0); + + zdep[2][2] = sqrt(3.0) * 0.5; + } + else if(il == 3) + { + zdep[3][0] = 2.5 * z3 - 1.5 * z * r2; + Gx_dep[3][0] = -3.0 * x * z; + Gy_dep[3][0] = -3.0 * y * z; + Gz_dep[3][0] = 1.5 * (3.0 * z2 - r2); + + zdep[3][1] = 0.25 * sqrt(6.0) * (5.0 * z2 - r2); + Gx_dep[3][1] = -0.5 * sqrt(6.0) * x; + Gy_dep[3][1] = -0.5 * sqrt(6.0) * y; + Gz_dep[3][1] = sqrt(6.0) * 2.0 * z; + + zdep[3][2] = 0.5 * sqrt(15.0) * z; + Gz_dep[3][2] = 0.5 * sqrt(15.0); + + zdep[3][3] = 0.25 * sqrt(10.0); + } + else if(il == 4) + { + zdep[4][0] = 0.125 * (35.0 * z4 - 30.0 * r2 * z2 + 3.0 * r4); + Gx_dep[4][0] = -7.5 * x * z2 + 1.5 * x * r2; + Gy_dep[4][0] = -7.5 * y * z2 + 1.5 * y * r2; + Gz_dep[4][0] = 10.0 * z3 - 6.0 * r2 * z; + + zdep[4][1] = sqrt(10.0) * 0.25 * z * (7.0 * z2 - 3.0 * r2); + Gx_dep[4][1] = -1.5 * sqrt(10.0) * x * z; + Gy_dep[4][1] = -1.5 * sqrt(10.0) * y * z; + Gz_dep[4][1] = 0.75 * sqrt(10.0) * (5.0 * z2 - r2); + + zdep[4][2] = sqrt(5.0) * 0.25 * (7.0 * z2 - r2); + Gx_dep[4][2] = -0.5 * sqrt(5.0) * x; + Gy_dep[4][2] = -0.5 * sqrt(5.0) * y; + Gz_dep[4][2] = 3.0 * sqrt(5.0) * z; + + zdep[4][3] = sqrt(70.0) * 0.25 * z; + Gz_dep[4][3] = 0.25 * sqrt(70.0); + + zdep[4][4] = sqrt(35.0) * 0.125; + } + else if(il == 5) + { + zdep[5][0] = 0.125 * z *( 63.0 * z4 - 70.0 * z2 * r2 + 15.0 * r4); + Gx_dep[5][0] = -17.5 * x * z3 + 7.5 * x * z * r2; + Gy_dep[5][0] = -17.5 * y * z3 + 7.5 * y * z * r2; + Gz_dep[5][0] = 175.0 * 0.125 * z4 + 15.0 * 0.125 * r4 - 150.0 * 0.125 * r2 * z2; + + zdep[5][1] = 0.125 * sqrt(15.0) * (21.0 * z4 - 14.0 * z2 * r2 + r4); + Gx_dep[5][1] = -3.5 * sqrt(15.0) * x * z2 + 0.5 * sqrt(15.0) * x * r2; + Gy_dep[5][1] = -3.5 * sqrt(15.0) * y * z2 + 0.5 * sqrt(15.0) * y * r2; + Gz_dep[5][1] = 7.0 * sqrt(15.0) * z3 - 3.0 * sqrt(15.0) * r2 * z; + + zdep[5][2] = 0.25 * sqrt(105.0) * z * (3.0 * z2 - r2); + Gx_dep[5][2] = -0.5 * sqrt(105.0) * x * z; + Gy_dep[5][2] = -0.5 * sqrt(105.0) * y * z; + Gz_dep[5][2] = 0.25 * sqrt(105.0) * (7.0 * z2 - r2); + + zdep[5][3] = 0.0625 * sqrt(70.0) * (9.0 * z2 - r2); + Gx_dep[5][3] = -0.125 * sqrt(70.0) * x; + Gy_dep[5][3] = -0.125 * sqrt(70.0) * y; + Gz_dep[5][3] = sqrt(70.0) * z; + + zdep[5][4] = 0.375 * sqrt(35.0) * z; + Gz_dep[5][4] = 0.375 * sqrt(35.0); + + zdep[5][5] = 0.1875 * sqrt(14.0); + } + else + { + for(int im = 0; im <= il; im++) + { + int kmax = static_cast( (il - im) / 2 ); + for(int ik = 0; ik <= kmax; ik++) + { + int twok = 2 * ik; + + double gamma = 0.0; + double aux0, aux1, aux2, aux3; + + aux0 = pow(-1.0, ik) * pow(2.0, -il); + aux1 = Fact(il) / Fact(ik) / Fact(il-ik); + aux2 = Fact(2*il - twok) / Fact(il) / Fact(il - twok); + aux3 = Fact(il - twok) / Fact(il - twok - im); + + gamma = aux0 * aux1 * aux2 * aux3; + + assert(il - twok - im >= 0); + zdep[il][im] += pow(r, twok) * pow(z, il-twok-im) * gamma; + + if(ik > 0) + { + Gx_dep[il][im] += (ik * pow(r2, ik-1) * 2.0 * x) * pow(z, il-twok-im) * gamma; + Gy_dep[il][im] += (ik * pow(r2, ik-1) * 2.0 * y) * pow(z, il-twok-im) * gamma; + } + + if(ik == 0) + { + if(il > im) + { + Gz_dep[il][im] += (il-im) * pow(z, il-im-1) * gamma; + } + } + else + { + if(il - twok - im == 0) + { + Gz_dep[il][im] += gamma * ik * pow(r2, ik-1) * 2.0 * z; + } + else + { + Gz_dep[il][im] += gamma * (ik * pow(r2, ik-1) * 2.0 * z * pow(z, il-twok-im) + + pow(r, twok) * (il-twok-im) * pow(z, il-twok-im-1)); + } + } + } + + if(im >= 1) + { + zdep[il][im] *= sqrt(2 * Fact(il - im) / Fact(il + im)); + Gx_dep[il][im] *= sqrt(2 * Fact(il - im) / Fact(il + im)); + Gy_dep[il][im] *= sqrt(2 * Fact(il - im) / Fact(il + im)); + Gz_dep[il][im] *= sqrt(2 * Fact(il - im) / Fact(il + im)); + + } + } + } + } + + //calc + int ic = 0; + for(int il = 0; il <= MaxL; il++) + { + double fac = sqrt( (2.0 * il + 1.0) / ModuleBase::FOUR_PI ); + + //m=0 + rly[ic] = Am[0] * zdep[il][0] * fac; + grly[ic][0] = (Gx_dep[il][0] * Am[0] + zdep[il][0] * Gx_Am[0]) * fac; + grly[ic][1] = (Gy_dep[il][0] * Am[0] + zdep[il][0] * Gy_Am[0]) * fac; + grly[ic][2] = Gz_dep[il][0] * Am[0] * fac; + + ic++; + + //m ! = 0 + for(int im = 1; im <= il; im++) + { + //m>0 + rly[ic] = Am[im] * zdep[il][im] * pow(-1.0, im) * fac; + grly[ic][0] = (Gx_dep[il][im] * Am[im] + zdep[il][im] * Gx_Am[im]) * pow(-1.0, im) * fac; + grly[ic][1] = (Gy_dep[il][im] * Am[im] + zdep[il][im] * Gy_Am[im]) * pow(-1.0, im) * fac; + grly[ic][2] = Gz_dep[il][im] * Am[im] * pow(-1.0, im) * fac; + + ic++; + + //m<0 + rly[ic] = Bm[im] * zdep[il][im] * pow(-1.0, im) * fac; + grly[ic][0] = (Gx_dep[il][im] * Bm[im] + zdep[il][im] * Gx_Bm[im]) * pow(-1.0, im) * fac; + grly[ic][1] = (Gy_dep[il][im] * Bm[im] + zdep[il][im] * Gy_Bm[im]) * pow(-1.0, im) * fac; + grly[ic][2] = Gz_dep[il][im] * Bm[im] * pow(-1.0, im) * fac; + + ic++; + } + } + + return; } /* void Ylm::test(void) { - ModuleBase::Vector3 R(0.0, 0.0, 1.0); - - double r,r2,r3,r4,r5,r6,r7; - r = R.norm(); - r2 = r * r; - r3 = r2 * r; - r4 = r3 * r; - r5 = r4 * r; - r6 = r5 * r; - r7 = r6 * r; - - //Max L = 7; - double ylm[64]; - double dylmdr[64][3]; - - double rly[64]; - double grly[64][3]; + ModuleBase::Vector3 R(0.0, 0.0, 1.0); + + double r,r2,r3,r4,r5,r6,r7; + r = R.norm(); + r2 = r * r; + r3 = r2 * r; + r4 = r3 * r; + r5 = r4 * r; + r6 = r5 * r; + r7 = r6 * r; + + //Max L = 7; + double ylm[64]; + double dylmdr[64][3]; + + double rly[64]; + double grly[64][3]; // std::cout << R.x << " " << R.y << " " << R.z << std::endl; - get_ylm_real(8, R, ylm, dylmdr); - rlylm(8, R.x, R.y, R.z, rly, grly); + get_ylm_real(8, R, ylm, dylmdr); + rlylm(8, R.x, R.y, R.z, rly, grly); // std::cout << R.x << " " << R.y << " " << R.z << std::endl; - for(int i = 0; i < 64; i++) - { - if(i >= 1 && i <= 3) - { - dylmdr[i][0] = dylmdr[i][0] * r + ylm[i] * R.x / r; - dylmdr[i][1] = dylmdr[i][1] * r + ylm[i] * R.y / r; - dylmdr[i][2] = dylmdr[i][2] * r + ylm[i] * R.z / r; - - ylm[i] *= r; - } - if(i >= 4 && i <= 8) - { - dylmdr[i][0] = dylmdr[i][0] * r2 + ylm[i] * R.x * 2.0; - dylmdr[i][1] = dylmdr[i][1] * r2 + ylm[i] * R.y * 2.0; - dylmdr[i][2] = dylmdr[i][2] * r2 + ylm[i] * R.z * 2.0; - - ylm[i] *= r2; - } - if(i >= 9 && i <= 15) - { - dylmdr[i][0] = dylmdr[i][0] * r3 + ylm[i] * R.x * 3.0 * r; - dylmdr[i][1] = dylmdr[i][1] * r3 + ylm[i] * R.y * 3.0 * r; - dylmdr[i][2] = dylmdr[i][2] * r3 + ylm[i] * R.z * 3.0 * r; - - ylm[i] *= pow(R.norm(),3); - } - if(i >= 16 && i <=24) - { - dylmdr[i][0] = dylmdr[i][0] * r4 + ylm[i] * R.x * 4.0 * r2; - dylmdr[i][1] = dylmdr[i][1] * r4 + ylm[i] * R.y * 4.0 * r2; - dylmdr[i][2] = dylmdr[i][2] * r4 + ylm[i] * R.z * 4.0 * r2; - - ylm[i] *= pow(R.norm(), 4); - } - if(i >= 25 && i <= 35) - { - dylmdr[i][0] = dylmdr[i][0] * r5 + ylm[i] * R.x * 5.0 * r3; - dylmdr[i][1] = dylmdr[i][1] * r5 + ylm[i] * R.y * 5.0 * r3; - dylmdr[i][2] = dylmdr[i][2] * r5 + ylm[i] * R.z * 5.0 * r3; - - ylm[i] *= pow(R.norm(), 5); - } - if(i >= 36 && i <= 48) - { - dylmdr[i][0] = dylmdr[i][0] * r6 + ylm[i] * R.x * 6.0 * r4; - dylmdr[i][1] = dylmdr[i][1] * r6 + ylm[i] * R.y * 6.0 * r4; - dylmdr[i][2] = dylmdr[i][2] * r6 + ylm[i] * R.z * 6.0 * r4; - ylm[i] *= pow(R.norm(), 6); - } - if(i >= 49 && i <= 63) - { - dylmdr[i][0] = dylmdr[i][0] * r7 + ylm[i] * R.x * 7.0 * r5; - dylmdr[i][1] = dylmdr[i][1] * r7 + ylm[i] * R.y * 7.0 * r5; - dylmdr[i][2] = dylmdr[i][2] * r7 + ylm[i] * R.z * 7.0 * r5; - ylm[i] *= pow(R.norm(), 7); - } - - std::cout << grly[i][0] << std::setw(20) << grly[i][1] << std::setw(20) << grly[i][2] << std::endl; - } - - return; + for(int i = 0; i < 64; i++) + { + if(i >= 1 && i <= 3) + { + dylmdr[i][0] = dylmdr[i][0] * r + ylm[i] * R.x / r; + dylmdr[i][1] = dylmdr[i][1] * r + ylm[i] * R.y / r; + dylmdr[i][2] = dylmdr[i][2] * r + ylm[i] * R.z / r; + + ylm[i] *= r; + } + if(i >= 4 && i <= 8) + { + dylmdr[i][0] = dylmdr[i][0] * r2 + ylm[i] * R.x * 2.0; + dylmdr[i][1] = dylmdr[i][1] * r2 + ylm[i] * R.y * 2.0; + dylmdr[i][2] = dylmdr[i][2] * r2 + ylm[i] * R.z * 2.0; + + ylm[i] *= r2; + } + if(i >= 9 && i <= 15) + { + dylmdr[i][0] = dylmdr[i][0] * r3 + ylm[i] * R.x * 3.0 * r; + dylmdr[i][1] = dylmdr[i][1] * r3 + ylm[i] * R.y * 3.0 * r; + dylmdr[i][2] = dylmdr[i][2] * r3 + ylm[i] * R.z * 3.0 * r; + + ylm[i] *= pow(R.norm(),3); + } + if(i >= 16 && i <=24) + { + dylmdr[i][0] = dylmdr[i][0] * r4 + ylm[i] * R.x * 4.0 * r2; + dylmdr[i][1] = dylmdr[i][1] * r4 + ylm[i] * R.y * 4.0 * r2; + dylmdr[i][2] = dylmdr[i][2] * r4 + ylm[i] * R.z * 4.0 * r2; + + ylm[i] *= pow(R.norm(), 4); + } + if(i >= 25 && i <= 35) + { + dylmdr[i][0] = dylmdr[i][0] * r5 + ylm[i] * R.x * 5.0 * r3; + dylmdr[i][1] = dylmdr[i][1] * r5 + ylm[i] * R.y * 5.0 * r3; + dylmdr[i][2] = dylmdr[i][2] * r5 + ylm[i] * R.z * 5.0 * r3; + + ylm[i] *= pow(R.norm(), 5); + } + if(i >= 36 && i <= 48) + { + dylmdr[i][0] = dylmdr[i][0] * r6 + ylm[i] * R.x * 6.0 * r4; + dylmdr[i][1] = dylmdr[i][1] * r6 + ylm[i] * R.y * 6.0 * r4; + dylmdr[i][2] = dylmdr[i][2] * r6 + ylm[i] * R.z * 6.0 * r4; + ylm[i] *= pow(R.norm(), 6); + } + if(i >= 49 && i <= 63) + { + dylmdr[i][0] = dylmdr[i][0] * r7 + ylm[i] * R.x * 7.0 * r5; + dylmdr[i][1] = dylmdr[i][1] * r7 + ylm[i] * R.y * 7.0 * r5; + dylmdr[i][2] = dylmdr[i][2] * r7 + ylm[i] * R.z * 7.0 * r5; + ylm[i] *= pow(R.norm(), 7); + } + + std::cout << grly[i][0] << std::setw(20) << grly[i][1] << std::setw(20) << grly[i][2] << std::endl; + } + + return; } */ void Ylm::ZEROS(double u[], const int& n) { - for(int i = 0; i < n; i++) - { - u[i] = 0.0; - } - return; + for(int i = 0; i < n; i++) + { + u[i] = 0.0; + } + return; } @@ -2029,31 +2029,31 @@ void Ylm::ZEROS(double u[], const int& n) //========================================================== long double Ylm::Fact(const int n) { - long double f = 1; - for(int i=n; i>1; i--) - { - f *= i; - } - return f; + long double f = 1; + for(int i=n; i>1; i--) + { + f *= i; + } + return f; } int Ylm::Semi_Fact(const int n) { - int semif = 1; - for(int i=n; i>2; i -= 2) - { - semif *= i; - } - return semif; + int semif = 1; + for(int i=n; i>2; i -= 2) + { + semif *= i; + } + return semif; } double Ylm::sgn(const double x) { - if(x < 0.0) return -1.0; - if(x > 0.0) return 1.0; - return 0.0; + if(x < 0.0) return -1.0; + if(x > 0.0) return 1.0; + return 0.0; } } diff --git a/source/source_base/ylm.h b/source/source_base/ylm.h index ff25a56a912..45a0d0baec7 100644 --- a/source/source_base/ylm.h +++ b/source/source_base/ylm.h @@ -10,157 +10,157 @@ namespace ModuleBase class Ylm { - public: - Ylm(){}; - ~Ylm(){}; - - static int nlm; - - - /** - * @brief Get the ylm real object - * - * @param Lmax [in] maximum angular quantum number + 1 - * @param vec [in] the vector to be calculated - * @param ylmr [out] calculated Ylm, Y00, Y10, Y11, Y1-1, Y20, Y21, Y2-1, Y22, Y2-2... - */ - static void get_ylm_real( - const int &Lmax , - const ModuleBase::Vector3 &vec, - double ylmr[]); - - /** - * @brief Get the ylm real object and the gradient - * - * @param Lmax [in] maximum angular quantum number + l - * @param vec [in] the vector to be calculated - * @param ylmr [out] calculated Ylm, Y00, Y10, Y11, Y1-1, Y20, Y21, Y2-1, Y22, Y2-2... - * @param dylmdr [out] gradient of Ylm, [dY00/dx, dY00/dy, dY00/dz], [dY10/dx, dY10/dy, dY10/dz], [dY11/dx, dY11/dy, dY11/dz],... - */ - static void get_ylm_real( - const int &Lmax , - const ModuleBase::Vector3 &vec, - double ylmr[], - double dylmdr[][3]); - - /** - * @brief Get the ylm real (solid) object (not used anymore) - * - * @param Lmax [in] maximum angular quantum number + l - * @param x [in] x - * @param y [in] y - * @param z [in] z - * @param rly [in] calculated Ylm, Y00, Y10, Y11, Y1-1, Y20, Y21, Y2-1, Y22, Y2-2... - */ - static void rlylm( - const int Lmax, - const double x, - const double y, - const double z, - double rly[]); - - /** - * @brief Get the ylm real (solid) object and the gradient (not used anymore) - * - * @param Lmax [in] maximum angular quantum number + 1 - * @param x [in] x - * @param y [in] y - * @param z [in] z - * @param rly [in] calculated Ylm, Y00, Y10, Y11, Y1-1, Y20, Y21, Y2-1, Y22, Y2-2... - * @param grly [out] gradient of Ylm, [dY00/dx, dY00/dy, dY00/dz], [dY10/dx, dY10/dy, dY10/dz], [dY11/dx, dY11/dy, dY11/dz],... - */ - static void rlylm( - const int Lmax, - const double x, - const double y, - const double z, - double rly[], - double grly[][3]); - - /** - * @brief Get the ylm real object (used in grid integration) - * - * @param Lmax [in] maximum angular quantum number - * @param xdr [in] x/r - * @param ydr [in] y/r - * @param zdr [in] z/r - * @param rly [in] calculated Ylm, Y00, Y10, Y11, Y1-1, Y20, Y21, Y2-1, Y22, Y2-2... - */ - static void sph_harm( - const int Lmax, - const double xdr, - const double ydr, - const double zdr, - std::vector &rly); - - /** - * @brief Get the ylm real object (used in getting overlap) - * - * @param Lmax [in] maximum angular quantum number - * @param x [in] x/r - * @param y [in] y/r - * @param z [in] z/r - * @param rly [in] calculated Ylm, Y00, Y10, Y11, Y1-1, Y20, Y21, Y2-1, Y22, Y2-2... - * @author Peize Lin - * @date 2016-08-26 - */ - static void rl_sph_harm( - const int Lmax, - const double x, - const double y, - const double z, - std::vector& rly); - - /** - * @brief Get the ylm real object and the gradient (used in getting derivative of overlap) - * - * @param Lmax [in] maximum angular quantum number - * @param x [in] x/r - * @param y [in] y/r - * @param z [in] z/r - * @param rly [in] calculated Ylm, Y00, Y10, Y11, Y1-1, Y20, Y21, Y2-1, Y22, Y2-2... - * @param grly [out] gradient of Ylm, stored as a contiguous flat array of - * size (Lmax+1)^2 * 3 in row-major order: - * [dY00/dx, dY00/dy, dY00/dz, dY10/dx, dY10/dy, dY10/dz, ...] - */ - static void grad_rl_sph_harm( - const int Lmax, - const double x, - const double y, - const double z, - double* rly, - double* grly); - - /** - * @brief Get the hessian of r^l Ylm (used in getting derivative of overlap) - * - * @param Lmax [in] maximum angular quantum number - * @param x [in] x - * @param y [in] y - * @param z [in] z - * @param hrly [out] hessian of Ylm, [dY00/dx2, dY00/dxy, dY00/dxz, dY00/dyy, dY00/dyz, dY00/dzz] , ... - */ - static void hes_rl_sph_harm( - const int Lmax, - const double x, - const double y, - const double z, - std::vector>& hrly); - - //calculate the coefficient of Ylm, ylmcoef. - static void set_coefficients (); - - //static void test(); - //static void test1(); - //static void test2(); - - //set the first n elements of u to be 0.0 - static void ZEROS(double u[], const int& n); - - private: - static long double Fact(const int n); - static int Semi_Fact(const int n); - static double sgn(const double x); + public: + Ylm(){}; + ~Ylm(){}; + + static int nlm; + + + /** + * @brief Get the ylm real object + * + * @param Lmax [in] maximum angular quantum number + 1 + * @param vec [in] the vector to be calculated + * @param ylmr [out] calculated Ylm, Y00, Y10, Y11, Y1-1, Y20, Y21, Y2-1, Y22, Y2-2... + */ + static void get_ylm_real( + const int &Lmax , + const ModuleBase::Vector3 &vec, + double ylmr[]); + + /** + * @brief Get the ylm real object and the gradient + * + * @param Lmax [in] maximum angular quantum number + l + * @param vec [in] the vector to be calculated + * @param ylmr [out] calculated Ylm, Y00, Y10, Y11, Y1-1, Y20, Y21, Y2-1, Y22, Y2-2... + * @param dylmdr [out] gradient of Ylm, [dY00/dx, dY00/dy, dY00/dz], [dY10/dx, dY10/dy, dY10/dz], [dY11/dx, dY11/dy, dY11/dz],... + */ + static void get_ylm_real( + const int &Lmax , + const ModuleBase::Vector3 &vec, + double ylmr[], + double dylmdr[][3]); + + /** + * @brief Get the ylm real (solid) object (not used anymore) + * + * @param Lmax [in] maximum angular quantum number + l + * @param x [in] x + * @param y [in] y + * @param z [in] z + * @param rly [in] calculated Ylm, Y00, Y10, Y11, Y1-1, Y20, Y21, Y2-1, Y22, Y2-2... + */ + static void rlylm( + const int Lmax, + const double x, + const double y, + const double z, + double rly[]); + + /** + * @brief Get the ylm real (solid) object and the gradient (not used anymore) + * + * @param Lmax [in] maximum angular quantum number + 1 + * @param x [in] x + * @param y [in] y + * @param z [in] z + * @param rly [in] calculated Ylm, Y00, Y10, Y11, Y1-1, Y20, Y21, Y2-1, Y22, Y2-2... + * @param grly [out] gradient of Ylm, [dY00/dx, dY00/dy, dY00/dz], [dY10/dx, dY10/dy, dY10/dz], [dY11/dx, dY11/dy, dY11/dz],... + */ + static void rlylm( + const int Lmax, + const double x, + const double y, + const double z, + double rly[], + double grly[][3]); + + /** + * @brief Get the ylm real object (used in grid integration) + * + * @param Lmax [in] maximum angular quantum number + * @param xdr [in] x/r + * @param ydr [in] y/r + * @param zdr [in] z/r + * @param rly [in] calculated Ylm, Y00, Y10, Y11, Y1-1, Y20, Y21, Y2-1, Y22, Y2-2... + */ + static void sph_harm( + const int Lmax, + const double xdr, + const double ydr, + const double zdr, + std::vector &rly); + + /** + * @brief Get the ylm real object (used in getting overlap) + * + * @param Lmax [in] maximum angular quantum number + * @param x [in] x/r + * @param y [in] y/r + * @param z [in] z/r + * @param rly [in] calculated Ylm, Y00, Y10, Y11, Y1-1, Y20, Y21, Y2-1, Y22, Y2-2... + * @author Peize Lin + * @date 2016-08-26 + */ + static void rl_sph_harm( + const int Lmax, + const double x, + const double y, + const double z, + std::vector& rly); + + /** + * @brief Get the ylm real object and the gradient (used in getting derivative of overlap) + * + * @param Lmax [in] maximum angular quantum number + * @param x [in] x/r + * @param y [in] y/r + * @param z [in] z/r + * @param rly [in] calculated Ylm, Y00, Y10, Y11, Y1-1, Y20, Y21, Y2-1, Y22, Y2-2... + * @param grly [out] gradient of Ylm, stored as a contiguous flat array of + * size (Lmax+1)^2 * 3 in row-major order: + * [dY00/dx, dY00/dy, dY00/dz, dY10/dx, dY10/dy, dY10/dz, ...] + */ + static void grad_rl_sph_harm( + const int Lmax, + const double x, + const double y, + const double z, + double* rly, + double* grly); + + /** + * @brief Get the hessian of r^l Ylm (used in getting derivative of overlap) + * + * @param Lmax [in] maximum angular quantum number + * @param x [in] x + * @param y [in] y + * @param z [in] z + * @param hrly [out] hessian of Ylm, [dY00/dx2, dY00/dxy, dY00/dxz, dY00/dyy, dY00/dyz, dY00/dzz] , ... + */ + static void hes_rl_sph_harm( + const int Lmax, + const double x, + const double y, + const double z, + std::vector>& hrly); + + //calculate the coefficient of Ylm, ylmcoef. + static void set_coefficients (); + + //static void test(); + //static void test1(); + //static void test2(); + + //set the first n elements of u to be 0.0 + static void ZEROS(double u[], const int& n); + + private: + static long double Fact(const int n); + static int Semi_Fact(const int n); + static double sgn(const double x); }; } diff --git a/source/source_base/ylmcoef.h b/source/source_base/ylmcoef.h index 72a756ee74f..9cd9a264309 100644 --- a/source/source_base/ylmcoef.h +++ b/source/source_base/ylmcoef.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_BASE_YLMCOEF_H +#define ABACUS_SOURCE_BASE_YLMCOEF_H + #include #include "constants.h" @@ -54,3 +56,5 @@ static constexpr double ylmcoef[YLMCOEF_SIZE] = { }; } // namespace ModuleBase + +#endif // ABACUS_SOURCE_BASE_YLMCOEF_H diff --git a/source/source_basis/module_ao/elem_basis_idx_orb.cpp b/source/source_basis/module_ao/elem_basis_idx_orb.cpp index a05da56f4f8..c3a441cf2a1 100644 --- a/source/source_basis/module_ao/elem_basis_idx_orb.cpp +++ b/source/source_basis/module_ao/elem_basis_idx_orb.cpp @@ -9,36 +9,36 @@ namespace ModuleBase ModuleBase::Element_Basis_Index::Range Element_Basis_Index::construct_range( const LCAO_Orbitals &orb ) { - ModuleBase::Element_Basis_Index::Range range; - range.resize( orb.get_ntype() ); - for( std::size_t T=0; T!=range.size(); ++T ) - { - range[T].resize( orb.Phi[T].getLmax()+1 ); - for( std::size_t L=0; L!=range[T].size(); ++L ) - { - range[T][L].N = orb.Phi[T].getNchi(L); - range[T][L].M = 2*L+1; - } - } - return range; + ModuleBase::Element_Basis_Index::Range range; + range.resize( orb.get_ntype() ); + for( std::size_t T=0; T!=range.size(); ++T ) + { + range[T].resize( orb.Phi[T].getLmax()+1 ); + for( std::size_t L=0; L!=range[T].size(); ++L ) + { + range[T][L].N = orb.Phi[T].getNchi(L); + range[T][L].M = 2*L+1; + } + } + return range; } ModuleBase::Element_Basis_Index::Range Element_Basis_Index::construct_range( const std::vector>> &orb ) { - ModuleBase::Element_Basis_Index::Range range; - range.resize( orb.size() ); - for( std::size_t T=0; T!=range.size(); ++T ) - { - range[T].resize( orb[T].size() ); - for( std::size_t L=0; L!=range[T].size(); ++L ) - { - range[T][L].N = orb[T][L].size(); - range[T][L].M = 2*L+1; - } - } - return range; + ModuleBase::Element_Basis_Index::Range range; + range.resize( orb.size() ); + for( std::size_t T=0; T!=range.size(); ++T ) + { + range[T].resize( orb[T].size() ); + for( std::size_t L=0; L!=range[T].size(); ++L ) + { + range[T][L].N = orb[T][L].size(); + range[T][L].M = 2*L+1; + } + } + return range; } } \ No newline at end of file diff --git a/source/source_basis/module_ao/elem_basis_idx_orb.h b/source/source_basis/module_ao/elem_basis_idx_orb.h index 7da100183ea..93468abc898 100644 --- a/source/source_basis/module_ao/elem_basis_idx_orb.h +++ b/source/source_basis/module_ao/elem_basis_idx_orb.h @@ -4,17 +4,17 @@ #include "../../source_base/element_basis_index.h" #include - class Numerical_Orbital_Lm; - class LCAO_Orbitals; + class Numerical_Orbital_Lm; + class LCAO_Orbitals; namespace ModuleBase { namespace Element_Basis_Index { - extern Range construct_range( const LCAO_Orbitals &orb ); + extern Range construct_range( const LCAO_Orbitals &orb ); - extern Range construct_range( const std::vector>> &orb ); // orb[T][L][N] + extern Range construct_range( const std::vector>> &orb ); // orb[T][L][N] } } diff --git a/source/source_basis/module_ao/orb_atomic.h b/source/source_basis/module_ao/orb_atomic.h index 4dea9ef3384..7905f3bdb89 100644 --- a/source/source_basis/module_ao/orb_atomic.h +++ b/source/source_basis/module_ao/orb_atomic.h @@ -10,34 +10,34 @@ class Numerical_Orbital_AtomRelation { public: - //========================================================== - // It's about two atoms relations, thread-safe interface - //========================================================== - double distance; - ModuleBase::Vector3 R1; - ModuleBase::Vector3 R2; //three-dimesion-coordinate of R - ModuleBase::Vector3 dR; // R1-R2 + //========================================================== + // It's about two atoms relations, thread-safe interface + //========================================================== + double distance; + ModuleBase::Vector3 R1; + ModuleBase::Vector3 R2; //three-dimesion-coordinate of R + ModuleBase::Vector3 dR; // R1-R2 - double& get_distance() - { - if(distance < 0.0) ModuleBase::WARNING_QUIT("NUMERICAL_ORBITAL","distance should be above zero!"); - return distance; - } - - double getX() { return R2.x - R1.x ; } - double getY() { return R2.y - R1.y ; } - double getZ() { return R2.z - R1.z ; } - ModuleBase::Vector3& getR1() { return R1; } - ModuleBase::Vector3& getR2() { return R2; } - ModuleBase::Vector3& getdR() { return dR; } + double& get_distance() + { + if(distance < 0.0) ModuleBase::WARNING_QUIT("NUMERICAL_ORBITAL","distance should be above zero!"); + return distance; + } + + double getX() { return R2.x - R1.x ; } + double getY() { return R2.y - R1.y ; } + double getZ() { return R2.z - R1.z ; } + ModuleBase::Vector3& getR1() { return R1; } + ModuleBase::Vector3& getR2() { return R2; } + ModuleBase::Vector3& getdR() { return dR; } - void set_position(const ModuleBase::Vector3 &R1_in, const ModuleBase::Vector3 &R2_in) - { - R1 = R1_in; - R2 = R2_in; - dR = R1-R2; - distance = dR.norm(); - } + void set_position(const ModuleBase::Vector3 &R1_in, const ModuleBase::Vector3 &R2_in) + { + R1 = R1_in; + R2 = R2_in; + dR = R1-R2; + distance = dR.norm(); + } }; /// @@ -50,82 +50,82 @@ class Numerical_Orbital_AtomRelation /// class Numerical_Orbital { - friend class LCAO_Orbitals; + friend class LCAO_Orbitals; public: - Numerical_Orbital(); - ~Numerical_Orbital(); + Numerical_Orbital(); + ~Numerical_Orbital(); - const int& getLmax() const { return this->lmax; } - const double& getRcut () const {return this->rcut; } - const int& getType() const { return this->type; } - const int& getTotal_nchi() const { return this->total_nchi; } - const int& getNchi(const int l) const { return this->nchi[l]; } - const std::string& getLabel() const { return this->label; } - - const inline Numerical_Orbital_Lm& PhiLN( const int &L, const int &N)const - { - return this->phiLN[ this->find_chi(L, N) ]; - } - - /// about the distance between two atoms. - static double& get_distance() - { - return NOAR.get_distance(); - } - - static double getX() { return NOAR.getX() ; } - static double getY() { return NOAR.getY() ; } - static double getZ() { return NOAR.getZ() ; } - static ModuleBase::Vector3& getR1() { return NOAR.getR1(); } - static ModuleBase::Vector3& getR2() { return NOAR.getR2(); } - static ModuleBase::Vector3& getdR() { return NOAR.getdR(); } + const int& getLmax() const { return this->lmax; } + const double& getRcut () const {return this->rcut; } + const int& getType() const { return this->type; } + const int& getTotal_nchi() const { return this->total_nchi; } + const int& getNchi(const int l) const { return this->nchi[l]; } + const std::string& getLabel() const { return this->label; } + + const inline Numerical_Orbital_Lm& PhiLN( const int &L, const int &N)const + { + return this->phiLN[ this->find_chi(L, N) ]; + } + + /// about the distance between two atoms. + static double& get_distance() + { + return NOAR.get_distance(); + } + + static double getX() { return NOAR.getX() ; } + static double getY() { return NOAR.getY() ; } + static double getZ() { return NOAR.getZ() ; } + static ModuleBase::Vector3& getR1() { return NOAR.getR1(); } + static ModuleBase::Vector3& getR2() { return NOAR.getR2(); } + static ModuleBase::Vector3& getdR() { return NOAR.getdR(); } - /// - /// set information about Numerical Orbital - /// - void set_orbital_info( - const int& type_in, - const std::string& label_in, - const int& lmax_in, - const int* nchi_in, - const int& total_nchi); + /// + /// set information about Numerical Orbital + /// + void set_orbital_info( + const int& type_in, + const std::string& label_in, + const int& lmax_in, + const int* nchi_in, + const int& total_nchi); - static void set_position(const ModuleBase::Vector3 &R1_in, const ModuleBase::Vector3 &R2_in) - { - NOAR.set_position(R1_in, R2_in); - } + static void set_position(const ModuleBase::Vector3 &R1_in, const ModuleBase::Vector3 &R2_in) + { + NOAR.set_position(R1_in, R2_in); + } - std::vector& chi() { return this->phiLN; } - + std::vector& chi() { return this->phiLN; } + private: - - //========================================================== - // MEMBER FUNCTION : - // NAME : label (atom type) - // NAME : lmax( max value of L angular momentum) - // NAME : nchi( number of chi for each L) - // NAME : total_nchi(total chi for this type of atom, total number of NAOs) - // NAME : max_nchi( max chi for certain L) - // NAME : find_chi(lmax+1, max_nchi). - //=========================================================== - std::string label; - - int type; - int lmax; - std::vector nchi; - int total_nchi; - int max_nchi; - ModuleBase::IntArray find_chi; - double rcut; + + //========================================================== + // MEMBER FUNCTION : + // NAME : label (atom type) + // NAME : lmax( max value of L angular momentum) + // NAME : nchi( number of chi for each L) + // NAME : total_nchi(total chi for this type of atom, total number of NAOs) + // NAME : max_nchi( max chi for certain L) + // NAME : find_chi(lmax+1, max_nchi). + //=========================================================== + std::string label; + + int type; + int lmax; + std::vector nchi; + int total_nchi; + int max_nchi; + ModuleBase::IntArray find_chi; + double rcut; - std::vector phiLN;// length: total_nchi (only store radial function ) + std::vector phiLN;// length: total_nchi (only store radial function ) - //========================================================== - // Keep the old interface - //========================================================== - static Numerical_Orbital_AtomRelation NOAR; + //========================================================== + // Keep the old interface + //========================================================== + static Numerical_Orbital_AtomRelation NOAR; }; #endif diff --git a/source/source_basis/module_ao/orb_atomic_lm.cpp b/source/source_basis/module_ao/orb_atomic_lm.cpp index 1ebd3e043c5..9d5e40a8612 100644 --- a/source/source_basis/module_ao/orb_atomic_lm.cpp +++ b/source/source_basis/module_ao/orb_atomic_lm.cpp @@ -14,21 +14,21 @@ Numerical_Orbital_Lm::Numerical_Orbital_Lm() { - label = ""; - index_atom_type = 0; - angular_momentum_l = 0; - index_chi = 0; + label = ""; + index_atom_type = 0; + angular_momentum_l = 0; + index_chi = 0; - nr=1; - nk=1; + nr=1; + nk=1; - rcut=0.0; - kcut=0.0; - dk=0.0; + rcut=0.0; + kcut=0.0; + dk=0.0; - nr_uniform = 1; - dr_uniform = -1.0; - zty = 0.0; + nr_uniform = 1; + dr_uniform = -1.0; + zty = 0.0; } Numerical_Orbital_Lm::~Numerical_Orbital_Lm() @@ -36,140 +36,140 @@ Numerical_Orbital_Lm::~Numerical_Orbital_Lm() void Numerical_Orbital_Lm::set_orbital_info ( - const std::string &label_in, - const int &index_atom_type_in, - const int &angular_momentum_l_in, - const int &index_chi_in, - const int &nr_in, - const double *rab_in, - const double *r_radial_in, - const Psi_Type &psi_type, // Peize Lin add 2017-12-12 - const double *psi_in, - const int &nk_in, - const double &dk_in, - // Peize Lin delete lat0 2016-02-03 - const double &dr_uniform_in, - bool flag_plot, // Peize Lin add flag_plot 2016-08-31 - bool flag_sbpool, // Peize Lin add flag_sbpool 2017-10-02 - const bool &force_flag // mohan add 2021-05-07 + const std::string &label_in, + const int &index_atom_type_in, + const int &angular_momentum_l_in, + const int &index_chi_in, + const int &nr_in, + const double *rab_in, + const double *r_radial_in, + const Psi_Type &psi_type, // Peize Lin add 2017-12-12 + const double *psi_in, + const int &nk_in, + const double &dk_in, + // Peize Lin delete lat0 2016-02-03 + const double &dr_uniform_in, + bool flag_plot, // Peize Lin add flag_plot 2016-08-31 + bool flag_sbpool, // Peize Lin add flag_sbpool 2017-10-02 + const bool &force_flag // mohan add 2021-05-07 ) { - copy_parameter( - label_in, - index_atom_type_in, - angular_momentum_l_in, - index_chi_in, - nr_in, - rab_in, - r_radial_in, - nk_in, - dk_in, - dr_uniform_in); - - switch(psi_type) - { - case Psi_Type::Psi: - for (int ir = 0; ir < nr; ir++) - { - this->psi[ir] = psi_in[ir]; - this->psir[ir] = psi[ir] * r_radial[ir]; //mohan 2010-04-19 - } - break; - case Psi_Type::Psif: - for( int ik=0; ik!=nk; ++ik ) - { - this->psif[ik] = psi_in[ik]; - this->psik[ik] = psif[ik] * k_radial[ik]; - this->psik2[ik] = psik[ik] * k_radial[ik]; - } - break; - case Psi_Type::Psik: - psif.resize(0); - for( int ik=0; ik!=nk; ++ik ) - { - this->psik[ik] = psi_in[ik]; - this->psik2[ik] = psik[ik] * k_radial[ik]; - } - break; - case Psi_Type::Psik2: - psif.resize(0); - psik.resize(0); - for( int ik=0; ik!=nk; ++ik ) - this->psik2[ik] = psi_in[ik]; - break; - default: - throw std::domain_error(std::string(__FILE__)+" line "+std::to_string(__LINE__)); - } - - switch(psi_type) - { - case Psi_Type::Psif: - case Psi_Type::Psik: - case Psi_Type::Psik2: - if( flag_sbpool ) - { - this->cal_rradial_sbpool(); - } - else - { - throw std::domain_error("flag_sbpool false not finished in Numerical_Orbital_Lm::set_orbital_info_k. "+std::string(__FILE__)+" line "+std::to_string(__LINE__)); - } - break; - default: break; - } - - //liaochen modify on 2010/4/7 - //we do SBT on regular mesh - //so we first generate psi_uniform first - //we put uniform in ahead of cal_kradial - - /* - bool uni = true; - if (uni) - { - this->extra_uniform(dr_uniform, force_flag); - } - else - { - this->use_uniform(dr_uniform); - } - */ - this->extra_uniform(dr_uniform, force_flag); - - switch(psi_type) - { - case Psi_Type::Psi: - if( flag_sbpool ) - { - this->cal_kradial_sbpool(); - } - else - { - this->cal_kradial(); - } - break; - default: break; - } + copy_parameter( + label_in, + index_atom_type_in, + angular_momentum_l_in, + index_chi_in, + nr_in, + rab_in, + r_radial_in, + nk_in, + dk_in, + dr_uniform_in); + + switch(psi_type) + { + case Psi_Type::Psi: + for (int ir = 0; ir < nr; ir++) + { + this->psi[ir] = psi_in[ir]; + this->psir[ir] = psi[ir] * r_radial[ir]; //mohan 2010-04-19 + } + break; + case Psi_Type::Psif: + for( int ik=0; ik!=nk; ++ik ) + { + this->psif[ik] = psi_in[ik]; + this->psik[ik] = psif[ik] * k_radial[ik]; + this->psik2[ik] = psik[ik] * k_radial[ik]; + } + break; + case Psi_Type::Psik: + psif.resize(0); + for( int ik=0; ik!=nk; ++ik ) + { + this->psik[ik] = psi_in[ik]; + this->psik2[ik] = psik[ik] * k_radial[ik]; + } + break; + case Psi_Type::Psik2: + psif.resize(0); + psik.resize(0); + for( int ik=0; ik!=nk; ++ik ) + this->psik2[ik] = psi_in[ik]; + break; + default: + throw std::domain_error(std::string(__FILE__)+" line "+std::to_string(__LINE__)); + } + + switch(psi_type) + { + case Psi_Type::Psif: + case Psi_Type::Psik: + case Psi_Type::Psik2: + if( flag_sbpool ) + { + this->cal_rradial_sbpool(); + } + else + { + throw std::domain_error("flag_sbpool false not finished in Numerical_Orbital_Lm::set_orbital_info_k. "+std::string(__FILE__)+" line "+std::to_string(__LINE__)); + } + break; + default: break; + } + + //liaochen modify on 2010/4/7 + //we do SBT on regular mesh + //so we first generate psi_uniform first + //we put uniform in ahead of cal_kradial + + /* + bool uni = true; + if (uni) + { + this->extra_uniform(dr_uniform, force_flag); + } + else + { + this->use_uniform(dr_uniform); + } + */ + this->extra_uniform(dr_uniform, force_flag); + + switch(psi_type) + { + case Psi_Type::Psi: + if( flag_sbpool ) + { + this->cal_kradial_sbpool(); + } + else + { + this->cal_kradial(); + } + break; + default: break; + } // this->norm_test(); // Peize Lin delete 2016-08-31 - if( flag_plot ) - { - this->plot(); // Peize Lin add flag_plot 2016-08-31 - } - return; + if( flag_plot ) + { + this->plot(); // Peize Lin add flag_plot 2016-08-31 + } + return; } void Numerical_Orbital_Lm::copy_parameter( - const std::string &label_in, - const int &index_atom_type_in, - const int &angular_momentum_l_in, - const int &index_chi_in, - const int &nr_in, - const double *rab_in, - const double *r_radial_in, - const int &nk_in, - const double &dk_in, - const double &dr_uniform_in) + const std::string &label_in, + const int &index_atom_type_in, + const int &angular_momentum_l_in, + const int &index_chi_in, + const int &nr_in, + const double *rab_in, + const double *r_radial_in, + const int &nk_in, + const double &dk_in, + const double &dr_uniform_in) { this->label = label_in; this->index_atom_type = index_atom_type_in; @@ -188,216 +188,216 @@ void Numerical_Orbital_Lm::copy_parameter( assert(nk%2!=0); assert(dk_in>0); this->dk = dk_in; - this->dr_uniform=dr_uniform_in; + this->dr_uniform=dr_uniform_in; - /*********************************************************** - be careful! LiaoChen modify on 2010/4/21 - ************************************************************/ + /*********************************************************** + be careful! LiaoChen modify on 2010/4/21 + ************************************************************/ // this->dk = ModuleBase::PI / rcut / 2.0; // this->nk = this->nr; - r_radial.resize(nr); - rab.resize(nr); - psi.resize(nr); - psir.resize(nr); - for (int ir = 0; ir < nr; ir++) - { - this->r_radial[ir] = r_radial_in[ir]; - this->rab[ir] = rab_in[ir]; - } - - k_radial.resize(nk); - psif.resize(nk); - psik.resize(nk); - psik2.resize(nk); - for (int ik = 0; ik < nk; ik++) - { - this->k_radial[ik] = ik * this->dk; - } - this->kcut = (nk-1) * this->dk; + r_radial.resize(nr); + rab.resize(nr); + psi.resize(nr); + psir.resize(nr); + for (int ir = 0; ir < nr; ir++) + { + this->r_radial[ir] = r_radial_in[ir]; + this->rab[ir] = rab_in[ir]; + } + + k_radial.resize(nk); + psif.resize(nk); + psik.resize(nk); + psik2.resize(nk); + for (int ik = 0; ik < nk; ik++) + { + this->k_radial[ik] = ik * this->dk; + } + this->kcut = (nk-1) * this->dk; } #include "source_base/mathzone_add1.h" void Numerical_Orbital_Lm::extra_uniform(const double &dr_uniform_in, const bool &force_flag) { - ModuleBase::timer::start("NOrbital_Lm", "extra_uniform"); - - //--------------------------------------------- - // set the dr, fixed by liaochen. - // calculate the number of radial mesh points. - //--------------------------------------------- - assert(dr_uniform>0.0); - this->dr_uniform = dr_uniform_in; - this->nr_uniform = static_cast(rcut/dr_uniform) + 10; - - this->psi_uniform.resize(nr_uniform,0); - - // do interpolation here to make grid more dense + ModuleBase::timer::start("NOrbital_Lm", "extra_uniform"); + + //--------------------------------------------- + // set the dr, fixed by liaochen. + // calculate the number of radial mesh points. + //--------------------------------------------- + assert(dr_uniform>0.0); + this->dr_uniform = dr_uniform_in; + this->nr_uniform = static_cast(rcut/dr_uniform) + 10; + + this->psi_uniform.resize(nr_uniform,0); + + // do interpolation here to make grid more dense #ifdef _OPENMP - #pragma omp parallel for schedule(static) + #pragma omp parallel for schedule(static) #endif - for (int ir = 0; ir < this->nr_uniform; ir++) - { - const double psi_uniform_tmp = - ModuleBase::Mathzone_Add1::Uni_RadialF(ModuleBase::GlobalFunc::VECTOR_TO_PTR(this->psi), this->nr, this->rab[0], ir * dr_uniform); - this->psi_uniform[ir] = psi_uniform_tmp; + for (int ir = 0; ir < this->nr_uniform; ir++) + { + const double psi_uniform_tmp = + ModuleBase::Mathzone_Add1::Uni_RadialF(ModuleBase::GlobalFunc::VECTOR_TO_PTR(this->psi), this->nr, this->rab[0], ir * dr_uniform); + this->psi_uniform[ir] = psi_uniform_tmp; // this->psi_uniform[ir] = ModuleBase::Mathzone::Polynomial_Interpolation(this->psi, this->nr, this->rab[0], ir * dr_uniform); } - - //---------------------------------------------- - // calculate the dpsi_uniform - //---------------------------------------------- - this->dpsi_uniform.resize(this->nr_uniform); - this->ddpsi_uniform.resize(this->nr_uniform); - - double* y2 = new double[nr]; - - //-------------------------------------------------------------------------- - // old code to calculate the derivate dpsi/dr, - // has problem that the derivatives of orbitals oscillate a lot - // around r=0 - //-------------------------------------------------------------------------- - //ModuleBase::Mathzone_Add1::SplineD2 (r_radial, psi, nr, 100000.0, 100000.0, y2); - //double yp1=(this->psi[1]-this->psi[0])/this->r_radial[1]; - //std::cout<<"psi0="<<" "<psi[0]<<" "<<"psi1="<<" "<psi[1]<<" "<<"r1="<<" "<r_radial[1]<angular_momentum_l ) // added by pengfei 13-8-8 different l has different boundary conditions - { - case 0: ModuleBase::Mathzone_Add1::SplineD2 (ModuleBase::GlobalFunc::VECTOR_TO_PTR(r_radial), ModuleBase::GlobalFunc::VECTOR_TO_PTR(psi), nr, 0.0, 0.0, y2); break; - case 1: ModuleBase::Mathzone_Add1::SplineD2 (ModuleBase::GlobalFunc::VECTOR_TO_PTR(r_radial), ModuleBase::GlobalFunc::VECTOR_TO_PTR(psi), nr, 100000.0, 100000.0, y2); break; - case 2: ModuleBase::Mathzone_Add1::SplineD2 (ModuleBase::GlobalFunc::VECTOR_TO_PTR(r_radial), ModuleBase::GlobalFunc::VECTOR_TO_PTR(psi), nr, 0.0, 0.0, y2); break; - case 3: ModuleBase::Mathzone_Add1::SplineD2 (ModuleBase::GlobalFunc::VECTOR_TO_PTR(r_radial), ModuleBase::GlobalFunc::VECTOR_TO_PTR(psi), nr, 100000.0, 100000.0, y2); break; - case 4: ModuleBase::Mathzone_Add1::SplineD2 (ModuleBase::GlobalFunc::VECTOR_TO_PTR(r_radial), ModuleBase::GlobalFunc::VECTOR_TO_PTR(psi), nr, 0.0, 0.0, y2); break; - default: - //GlobalV::ofs_warning << " The angular momentum larger than 4 (g orbitals) may be error about eggbox. " << std::endl; - //GlobalV::ofs_warning << " Check file " << __FILE__ << " line " << __LINE__ <angular_momentum_l<psi_uniform), - this->nr_uniform, - dr_uniform, - angular_momentum_l, - tmp); - - this->zty = tmp[0]/ModuleBase::Mathzone_Add1::factorial (angular_momentum_l); - } - - delete [] y2; - delete [] rad; - delete [] tmp; - ModuleBase::timer::end("NOrbital_Lm", "extra_uniform"); + + //---------------------------------------------- + // calculate the dpsi_uniform + //---------------------------------------------- + this->dpsi_uniform.resize(this->nr_uniform); + this->ddpsi_uniform.resize(this->nr_uniform); + + double* y2 = new double[nr]; + + //-------------------------------------------------------------------------- + // old code to calculate the derivate dpsi/dr, + // has problem that the derivatives of orbitals oscillate a lot + // around r=0 + //-------------------------------------------------------------------------- + //ModuleBase::Mathzone_Add1::SplineD2 (r_radial, psi, nr, 100000.0, 100000.0, y2); + //double yp1=(this->psi[1]-this->psi[0])/this->r_radial[1]; + //std::cout<<"psi0="<<" "<psi[0]<<" "<<"psi1="<<" "<psi[1]<<" "<<"r1="<<" "<r_radial[1]<angular_momentum_l ) // added by pengfei 13-8-8 different l has different boundary conditions + { + case 0: ModuleBase::Mathzone_Add1::SplineD2 (ModuleBase::GlobalFunc::VECTOR_TO_PTR(r_radial), ModuleBase::GlobalFunc::VECTOR_TO_PTR(psi), nr, 0.0, 0.0, y2); break; + case 1: ModuleBase::Mathzone_Add1::SplineD2 (ModuleBase::GlobalFunc::VECTOR_TO_PTR(r_radial), ModuleBase::GlobalFunc::VECTOR_TO_PTR(psi), nr, 100000.0, 100000.0, y2); break; + case 2: ModuleBase::Mathzone_Add1::SplineD2 (ModuleBase::GlobalFunc::VECTOR_TO_PTR(r_radial), ModuleBase::GlobalFunc::VECTOR_TO_PTR(psi), nr, 0.0, 0.0, y2); break; + case 3: ModuleBase::Mathzone_Add1::SplineD2 (ModuleBase::GlobalFunc::VECTOR_TO_PTR(r_radial), ModuleBase::GlobalFunc::VECTOR_TO_PTR(psi), nr, 100000.0, 100000.0, y2); break; + case 4: ModuleBase::Mathzone_Add1::SplineD2 (ModuleBase::GlobalFunc::VECTOR_TO_PTR(r_radial), ModuleBase::GlobalFunc::VECTOR_TO_PTR(psi), nr, 0.0, 0.0, y2); break; + default: + //GlobalV::ofs_warning << " The angular momentum larger than 4 (g orbitals) may be error about eggbox. " << std::endl; + //GlobalV::ofs_warning << " Check file " << __FILE__ << " line " << __LINE__ <angular_momentum_l<psi_uniform), + this->nr_uniform, + dr_uniform, + angular_momentum_l, + tmp); + + this->zty = tmp[0]/ModuleBase::Mathzone_Add1::factorial (angular_momentum_l); + } + + delete [] y2; + delete [] rad; + delete [] tmp; + ModuleBase::timer::end("NOrbital_Lm", "extra_uniform"); } /* void Numerical_Orbital_Lm::use_uniform(const double &dr_uniform_in) { - assert(dr_uniform_in>0.0); - this->dr_uniform = dr_uniform_in; - // for save: +10, because in real space interpolation, - // there may be "one grid point" more than the cutoff. - this->nr_uniform = static_cast(rcut/dr_uniform)+10; - - this->psi_uniform.resize(nr_uniform,0); - - std::string orbital_type; - // Peize Lin update 2016-08-31 - if( 0==this->angular_momentum_l ) - { - orbital_type = 's'; - } - else if( 1==this->angular_momentum_l ) - { - orbital_type = 'p'; - } - else if( 2==this->angular_momentum_l ) - { - orbital_type = 'd'; - } - else if( 3<=this->angular_momentum_l && this->angular_momentum_l<=6 ) - { - orbital_type = 'f'+this->angular_momentum_l-3; - } - else if( 7<=this->angular_momentum_l && this->angular_momentum_l<=11 ) - { - orbital_type = 'k'+this->angular_momentum_l-7; - } - else - { - orbital_type = "L" + ModuleBase::GlobalFunc::TO_STRING(this->angular_momentum_l); - } - - std::cout << "===========================================================" << std::endl; - for(int i=0; ipsi_uniform[i] = - ModuleBase::Mathzone_Add1::Uni_RadialF(ModuleBase::GlobalFunc::VECTOR_TO_PTR(psi), this->nr, this->rab[0], i*dr_uniform); - } - - this->dpsi_uniform.resize(nr_uniform); - - ModuleBase::Mathzone_Add1::Uni_Deriv_Phi ( - ModuleBase::GlobalFunc::VECTOR_TO_PTR(psi_uniform), - nr_uniform, dr_uniform, - 1, - ModuleBase::GlobalFunc::VECTOR_TO_PTR(dpsi_uniform)); + assert(dr_uniform_in>0.0); + this->dr_uniform = dr_uniform_in; + // for save: +10, because in real space interpolation, + // there may be "one grid point" more than the cutoff. + this->nr_uniform = static_cast(rcut/dr_uniform)+10; + + this->psi_uniform.resize(nr_uniform,0); + + std::string orbital_type; + // Peize Lin update 2016-08-31 + if( 0==this->angular_momentum_l ) + { + orbital_type = 's'; + } + else if( 1==this->angular_momentum_l ) + { + orbital_type = 'p'; + } + else if( 2==this->angular_momentum_l ) + { + orbital_type = 'd'; + } + else if( 3<=this->angular_momentum_l && this->angular_momentum_l<=6 ) + { + orbital_type = 'f'+this->angular_momentum_l-3; + } + else if( 7<=this->angular_momentum_l && this->angular_momentum_l<=11 ) + { + orbital_type = 'k'+this->angular_momentum_l-7; + } + else + { + orbital_type = "L" + ModuleBase::GlobalFunc::TO_STRING(this->angular_momentum_l); + } + + std::cout << "===========================================================" << std::endl; + for(int i=0; ipsi_uniform[i] = + ModuleBase::Mathzone_Add1::Uni_RadialF(ModuleBase::GlobalFunc::VECTOR_TO_PTR(psi), this->nr, this->rab[0], i*dr_uniform); + } + + this->dpsi_uniform.resize(nr_uniform); + + ModuleBase::Mathzone_Add1::Uni_Deriv_Phi ( + ModuleBase::GlobalFunc::VECTOR_TO_PTR(psi_uniform), + nr_uniform, dr_uniform, + 1, + ModuleBase::GlobalFunc::VECTOR_TO_PTR(dpsi_uniform)); #ifdef __NORMAL #else - if(GlobalV::MY_RANK==0) - { - std::stringstream ss; - ss << ModuleBase::get_global_out_dir() << this->label << "/" - << this->label << "-" << orbital_type << ".ORBITAL_NOR_uniform.txt"; - - std::ofstream ofs(ss.str().c_str()); - - for(int i=0; ilabel << "/" + << this->label << "-" << orbital_type << ".ORBITAL_NOR_uniform.txt"; + + std::ofstream ofs(ss.str().c_str()); + + for(int i=0; inr > 0); - assert( this->nr_uniform > 0); - double *jl = new double[nr]; - double *integrated_func = new double[nr]; - - const double pref = sqrt( 2.0 / ModuleBase::PI ); - //Sbt method - - /* - double* rad = new double[nr_uniform]; - for (int ir = 0; ir < nr_uniform; ir++) - { - rad[ir] = dr_uniform * ir; - } - - //liaochen add - ModuleBase::Mathzone_Add1::Sbt_new (3, angular_momentum_l, - k_radial, dk, nk, - rad, dr_uniform, nr_uniform, - psi_uniform, 0, this->psik); - - for (int ik = 0; ik < nk; ik++) this->psik[ik] *= (pref*k_radial[ik]); - delete [] rad; - */ - - //integration directly - for (int ik = 0; ik < nk; ik++) - { - ModuleBase::Sphbes::Spherical_Bessel( - this->nr, - ModuleBase::GlobalFunc::VECTOR_TO_PTR(this->r_radial), - this->k_radial[ik], - this->angular_momentum_l, - jl); - - for (int ir = 0; ir < nr; ir++) - { - integrated_func[ir] = this->psir[ir] * this->r_radial[ir] * jl[ir]; - } - - ModuleBase::Integral::Simpson_Integral( - this->nr, - integrated_func, - ModuleBase::GlobalFunc::VECTOR_TO_PTR(this->rab), - this->psif[ik]); - this->psif[ik] *= pref; - this->psik[ik] = this->psif[ik] * k_radial[ik]; - this->psik2[ik] = this->psik[ik] * k_radial[ik]; - } - - delete[] integrated_func; - delete[] jl; + assert( this->nr > 0); + assert( this->nr_uniform > 0); + double *jl = new double[nr]; + double *integrated_func = new double[nr]; + + const double pref = sqrt( 2.0 / ModuleBase::PI ); + //Sbt method + + /* + double* rad = new double[nr_uniform]; + for (int ir = 0; ir < nr_uniform; ir++) + { + rad[ir] = dr_uniform * ir; + } + + //liaochen add + ModuleBase::Mathzone_Add1::Sbt_new (3, angular_momentum_l, + k_radial, dk, nk, + rad, dr_uniform, nr_uniform, + psi_uniform, 0, this->psik); + + for (int ik = 0; ik < nk; ik++) this->psik[ik] *= (pref*k_radial[ik]); + delete [] rad; + */ + + //integration directly + for (int ik = 0; ik < nk; ik++) + { + ModuleBase::Sphbes::Spherical_Bessel( + this->nr, + ModuleBase::GlobalFunc::VECTOR_TO_PTR(this->r_radial), + this->k_radial[ik], + this->angular_momentum_l, + jl); + + for (int ir = 0; ir < nr; ir++) + { + integrated_func[ir] = this->psir[ir] * this->r_radial[ir] * jl[ir]; + } + + ModuleBase::Integral::Simpson_Integral( + this->nr, + integrated_func, + ModuleBase::GlobalFunc::VECTOR_TO_PTR(this->rab), + this->psif[ik]); + this->psif[ik] *= pref; + this->psik[ik] = this->psif[ik] * k_radial[ik]; + this->psik2[ik] = this->psik[ik] * k_radial[ik]; + } + + delete[] integrated_func; + delete[] jl; } /* // Peize Lin add 2017-10-02 void Numerical_Orbital_Lm::cal_kradial_sbpool(void) { - assert( this->nr > 0); - assert( this->nr_uniform > 0); - - // dr must be all the same for Sph_Bessel_Recursive_Pool and Simpson_Integral - const double dr = this->rab[0]; - for( size_t ir=1; irnr; ++ir ) - assert( dr == this->rab[ir] ); - - ModuleBase::Sph_Bessel_Recursive::D2* pSB = nullptr; - for( auto & sb : Sph_Bessel_Recursive_Pool::D2::sb_pool ) - if( this->dk * dr == sb.get_dx() ) - { - pSB = &sb; - break; - } - if(!pSB) - { - Sph_Bessel_Recursive_Pool::D2::sb_pool.push_back({}); - pSB = &Sph_Bessel_Recursive_Pool::D2::sb_pool.back(); - } - pSB->set_dx( this->dk * dr ); - pSB->cal_jlx( this->angular_momentum_l, this->nk, this->nr ); - const std::vector> &jl = pSB->get_jlx()[this->angular_momentum_l]; - - std::vector integrated_func( this->nr ); - const double pref = sqrt( 2.0 / ModuleBase::PI ); - - std::vector psir2(nr); - for( size_t ir=0; ir!=nr; ++ir ) - psir2[ir] = this->psir[ir] * this->r_radial[ir]; - - for (int ik = 0; ik < nk; ik++) - { - const std::vector &jlk = jl[ik]; - for (int ir = 0; ir < nr; ir++) - integrated_func[ir] = psir2[ir] * jlk[ir]; - ModuleBase::Integral::Simpson_Integral( - this->nr, - ModuleBase::GlobalFunc::VECTOR_TO_PTR(integrated_func), - dr, - this->psik[ik]); - this->psik[ik] *= ( pref * k_radial[ik]); - } + assert( this->nr > 0); + assert( this->nr_uniform > 0); + + // dr must be all the same for Sph_Bessel_Recursive_Pool and Simpson_Integral + const double dr = this->rab[0]; + for( size_t ir=1; irnr; ++ir ) + assert( dr == this->rab[ir] ); + + ModuleBase::Sph_Bessel_Recursive::D2* pSB = nullptr; + for( auto & sb : Sph_Bessel_Recursive_Pool::D2::sb_pool ) + if( this->dk * dr == sb.get_dx() ) + { + pSB = &sb; + break; + } + if(!pSB) + { + Sph_Bessel_Recursive_Pool::D2::sb_pool.push_back({}); + pSB = &Sph_Bessel_Recursive_Pool::D2::sb_pool.back(); + } + pSB->set_dx( this->dk * dr ); + pSB->cal_jlx( this->angular_momentum_l, this->nk, this->nr ); + const std::vector> &jl = pSB->get_jlx()[this->angular_momentum_l]; + + std::vector integrated_func( this->nr ); + const double pref = sqrt( 2.0 / ModuleBase::PI ); + + std::vector psir2(nr); + for( size_t ir=0; ir!=nr; ++ir ) + psir2[ir] = this->psir[ir] * this->r_radial[ir]; + + for (int ik = 0; ik < nk; ik++) + { + const std::vector &jlk = jl[ik]; + for (int ir = 0; ir < nr; ir++) + integrated_func[ir] = psir2[ir] * jlk[ir]; + ModuleBase::Integral::Simpson_Integral( + this->nr, + ModuleBase::GlobalFunc::VECTOR_TO_PTR(integrated_func), + dr, + this->psik[ik]); + this->psik[ik] *= ( pref * k_radial[ik]); + } } */ // Peize Lin add 2017-10-27 void Numerical_Orbital_Lm::cal_kradial_sbpool(void) { - assert( this->nr > 0); - assert( this->nr_uniform > 0); - - // dr must be all the same for Sph_Bessel_Recursive_Pool - const double dr = this->rab[0]; - - for( int ir=1; irnr; ++ir ) - { - assert( dr == this->rab[ir] ); - } - - ModuleBase::Sph_Bessel_Recursive::D2* pSB = nullptr; - for( auto & sb : ModuleBase::Sph_Bessel_Recursive_Pool::D2::sb_pool ) - { - if( this->dk * dr == sb.get_dx() ) - { - pSB = &sb; - break; - } - } - - if(!pSB) - { - ModuleBase::Sph_Bessel_Recursive_Pool::D2::sb_pool.push_back({}); - pSB = &ModuleBase::Sph_Bessel_Recursive_Pool::D2::sb_pool.back(); - } - pSB->set_dx( this->dk * dr ); - pSB->cal_jlx( this->angular_momentum_l, this->nk, this->nr ); - const std::vector> &jl = pSB->get_jlx()[this->angular_momentum_l]; - - const double pref = sqrt( 2.0 / ModuleBase::PI ); - - std::vector r_tmp(nr); - for( int ir=0; ir!=nr; ++ir ) - { - r_tmp[ir] = this->psir[ir] * this->r_radial[ir] * this->rab[ir]; - } - - constexpr double one_three=1.0/3.0, two_three=2.0/3.0, four_three=4.0/3.0; - r_tmp[0]*=one_three; - r_tmp[nr-1]*=one_three; - - for( int ir=1; ir!=nr-1; ++ir ) - { - r_tmp[ir] *= (ir&1) ? four_three : two_three; - } + assert( this->nr > 0); + assert( this->nr_uniform > 0); + + // dr must be all the same for Sph_Bessel_Recursive_Pool + const double dr = this->rab[0]; + + for( int ir=1; irnr; ++ir ) + { + assert( dr == this->rab[ir] ); + } + + ModuleBase::Sph_Bessel_Recursive::D2* pSB = nullptr; + for( auto & sb : ModuleBase::Sph_Bessel_Recursive_Pool::D2::sb_pool ) + { + if( this->dk * dr == sb.get_dx() ) + { + pSB = &sb; + break; + } + } + + if(!pSB) + { + ModuleBase::Sph_Bessel_Recursive_Pool::D2::sb_pool.push_back({}); + pSB = &ModuleBase::Sph_Bessel_Recursive_Pool::D2::sb_pool.back(); + } + pSB->set_dx( this->dk * dr ); + pSB->cal_jlx( this->angular_momentum_l, this->nk, this->nr ); + const std::vector> &jl = pSB->get_jlx()[this->angular_momentum_l]; + + const double pref = sqrt( 2.0 / ModuleBase::PI ); + + std::vector r_tmp(nr); + for( int ir=0; ir!=nr; ++ir ) + { + r_tmp[ir] = this->psir[ir] * this->r_radial[ir] * this->rab[ir]; + } + + constexpr double one_three=1.0/3.0, two_three=2.0/3.0, four_three=4.0/3.0; + r_tmp[0]*=one_three; + r_tmp[nr-1]*=one_three; + + for( int ir=1; ir!=nr-1; ++ir ) + { + r_tmp[ir] *= (ir&1) ? four_three : two_three; + } #ifdef _OPENMP - #pragma omp parallel for schedule(static) + #pragma omp parallel for schedule(static) #endif - for (int ik = 0; ik < nk; ik++) - { + for (int ik = 0; ik < nk; ik++) + { #ifdef __NORMAL - double psi_f_tmp = 0.0; - for(int ir=0; irnr, ModuleBase::GlobalFunc::VECTOR_TO_PTR(r_tmp), 1, ModuleBase::GlobalFunc::VECTOR_TO_PTR(jl[ik]), 1 ) ; + const double psi_f_tmp = + pref * BlasConnector::dot( this->nr, ModuleBase::GlobalFunc::VECTOR_TO_PTR(r_tmp), 1, ModuleBase::GlobalFunc::VECTOR_TO_PTR(jl[ik]), 1 ) ; #endif - this->psif[ik] = psi_f_tmp; - this->psik[ik] = psi_f_tmp * k_radial[ik]; - this->psik2[ik] = this->psik[ik] * k_radial[ik]; - } - return; + this->psif[ik] = psi_f_tmp; + this->psik[ik] = psi_f_tmp * k_radial[ik]; + this->psik2[ik] = this->psik[ik] * k_radial[ik]; + } + return; } // Peize Lin add 2017-12-11 void Numerical_Orbital_Lm::cal_rradial_sbpool(void) { - // dr must be all the same for Sph_Bessel_Recursive_Pool - const double dr = this->rab[0]; + // dr must be all the same for Sph_Bessel_Recursive_Pool + const double dr = this->rab[0]; - for( int ir=1; irnr; ++ir ) - { - assert( dr == this->rab[ir] ); - } + for( int ir=1; irnr; ++ir ) + { + assert( dr == this->rab[ir] ); + } - ModuleBase::Sph_Bessel_Recursive::D2* pSB = nullptr; - for( auto & sb : ModuleBase::Sph_Bessel_Recursive_Pool::D2::sb_pool ) - { - if( dr * dk == sb.get_dx() ) - { - pSB = &sb; - break; - } - } + ModuleBase::Sph_Bessel_Recursive::D2* pSB = nullptr; + for( auto & sb : ModuleBase::Sph_Bessel_Recursive_Pool::D2::sb_pool ) + { + if( dr * dk == sb.get_dx() ) + { + pSB = &sb; + break; + } + } - if(!pSB) - { - ModuleBase::Sph_Bessel_Recursive_Pool::D2::sb_pool.push_back({}); - pSB = &ModuleBase::Sph_Bessel_Recursive_Pool::D2::sb_pool.back(); - } + if(!pSB) + { + ModuleBase::Sph_Bessel_Recursive_Pool::D2::sb_pool.push_back({}); + pSB = &ModuleBase::Sph_Bessel_Recursive_Pool::D2::sb_pool.back(); + } - pSB->set_dx( dr * dk ); - pSB->cal_jlx( this->angular_momentum_l, this->nr, this->nk ); + pSB->set_dx( dr * dk ); + pSB->cal_jlx( this->angular_momentum_l, this->nr, this->nk ); - const std::vector> &jl = pSB->get_jlx()[this->angular_momentum_l]; + const std::vector> &jl = pSB->get_jlx()[this->angular_momentum_l]; - const double pref = sqrt(2.0/ModuleBase::PI); + const double pref = sqrt(2.0/ModuleBase::PI); - std::vector k_tmp(nk); + std::vector k_tmp(nk); - for( int ik=0; ik!=nk; ++ik ) - { - k_tmp[ik] = this->psik2[ik] * dk; - } + for( int ik=0; ik!=nk; ++ik ) + { + k_tmp[ik] = this->psik2[ik] * dk; + } - constexpr double one_three=1.0/3.0, two_three=2.0/3.0, four_three=4.0/3.0; + constexpr double one_three=1.0/3.0, two_three=2.0/3.0, four_three=4.0/3.0; - k_tmp[0]*=one_three; - k_tmp[nk-1]*=one_three; + k_tmp[0]*=one_three; + k_tmp[nk-1]*=one_three; - for( int ik=1; ik!=nk-1; ++ik ) - { - k_tmp[ik] *= (ik&1) ? four_three : two_three; - } + for( int ik=1; ik!=nk-1; ++ik ) + { + k_tmp[ik] *= (ik&1) ? four_three : two_three; + } - for( int ir = 0; ir!=nr; ++ir ) - { + for( int ir = 0; ir!=nr; ++ir ) + { #ifdef __NORMAL - // mohan add 2021-05-08, test needed - double kj_dot = 0.0; - for( int ik=0; ikpsi[ir] = pref * kj_dot; + // mohan add 2021-05-08, test needed + double kj_dot = 0.0; + for( int ik=0; ikpsi[ir] = pref * kj_dot; #else - this->psi[ir] = pref * BlasConnector::dot( this->nk, ModuleBase::GlobalFunc::VECTOR_TO_PTR(k_tmp), 1, ModuleBase::GlobalFunc::VECTOR_TO_PTR(jl[ir]), 1 ); + this->psi[ir] = pref * BlasConnector::dot( this->nk, ModuleBase::GlobalFunc::VECTOR_TO_PTR(k_tmp), 1, ModuleBase::GlobalFunc::VECTOR_TO_PTR(jl[ir]), 1 ); #endif - this->psir[ir] = this->psi[ir] * r_radial[ir]; - } + this->psir[ir] = this->psi[ir] * r_radial[ir]; + } } //=============================================== @@ -657,118 +657,118 @@ void Numerical_Orbital_Lm::cal_rradial_sbpool(void) void Numerical_Orbital_Lm::norm_test(void)const { // ModuleBase::TITLE(ofs_onscaling, "Numerical_Orbital_Lm", "norm_test"); - //double asum_r = 0.0; - //double asum_k = 0.0; - - // note here psir = psi * r - double *f = new double[nr]; - for(int ir=0; irpsir[ir] * this->psir[ir]; - } + //double asum_r = 0.0; + //double asum_k = 0.0; + + // note here psir = psi * r + double *f = new double[nr]; + for(int ir=0; irpsir[ir] * this->psir[ir]; + } - double sumr = 0.0; - //double sumk = 0.0; + double sumr = 0.0; + //double sumk = 0.0; - ModuleBase::Integral::Simpson_Integral(this->nr, f, ModuleBase::GlobalFunc::VECTOR_TO_PTR(this->rab), sumr); + ModuleBase::Integral::Simpson_Integral(this->nr, f, ModuleBase::GlobalFunc::VECTOR_TO_PTR(this->rab), sumr); - delete[] f; - f = new double[nk]; - for(int ik=0; ikpsik[ik] * this->psik[ik]; - } + delete[] f; + f = new double[nk]; + for(int ik=0; ikpsik[ik] * this->psik[ik]; + } // ModuleBase::Integral::Simpson_Integral(this->nk, f, this->k_radial, sumk); - - //means nothing. - //GlobalV::ofs_running << std::setw(12) << sumk << std::endl; + + //means nothing. + //GlobalV::ofs_running << std::setw(12) << sumk << std::endl; - delete[] f; - return; + delete[] f; + return; } */ void Numerical_Orbital_Lm::plot(void)const { - ModuleBase::TITLE("Numerical_Orbital_Lm","plot"); - - std::string orbital_type; - // Peize Lin update 2016-08-31 - if( 0==this->angular_momentum_l ) - { - orbital_type = 's'; - } - else if( 1==this->angular_momentum_l ) - { - orbital_type = 'p'; - } - else if( 2==this->angular_momentum_l ) - { - orbital_type = 'd'; - } - else if( 3<=this->angular_momentum_l && this->angular_momentum_l<=6 ) - { - orbital_type = 'f' + this->angular_momentum_l - 3; - } - else if( 7<=this->angular_momentum_l && this->angular_momentum_l<=11 ) - { - orbital_type = 'k' + this->angular_momentum_l - 7; - } - else - { - orbital_type = "L" + ModuleBase::GlobalFunc::TO_STRING(this->angular_momentum_l); - } - - if(GlobalV::MY_RANK==0) - { - std::stringstream ssr, ssk, ssru ,ssdru; // 2013-08-10 pengfei - ssr << ModuleBase::get_global_out_dir() << this->label << "/" - << this->label << "-"<< orbital_type << index_chi+1 << "-orbital-r.dat"; - - ssk << ModuleBase::get_global_out_dir() << this->label << "/" - << this->label << "-" << orbital_type << index_chi+1 << "-orbital-k.dat"; - - ssru << ModuleBase::get_global_out_dir() << this->label << "/" - << this->label << "-" << orbital_type << index_chi+1 << "-orbital-ru.dat"; - - ssdru << ModuleBase::get_global_out_dir() << this->label << "/" // 2013-08-10 pengfei - << this->label << "-" << orbital_type << index_chi+1 << "-orbital-dru.dat"; - - std::ofstream ofsr(ssr.str().c_str()); - std::ofstream ofsk(ssk.str().c_str()); - std::ofstream ofsru(ssru.str().c_str()); - std::ofstream ofsdru(ssdru.str().c_str()); // 2013-08-10 pengfei - - if (!ofsk || !ofsr || !ofsru || !ofsdru) // 2013-08-10 pengfei - { - ModuleBase::WARNING("Numerical_Orbital_Lm : plot", "Can't open files !"); - } - - for (int i = 0; i < this->nr; i++) - { - ofsr << this->r_radial[i] << " " << psi[i] << std::endl; - } - - for (int i = 0; i < this->nk; i++) - { - ofsk << this->k_radial[i] << " " << psik[i] << std::endl; - } - - for (int i = 0; i < this->nr_uniform; i++) - { - ofsru << this->dr_uniform * i << " " << psi_uniform[i] << std::endl; - } - - for (int i = 0; i < this->nr_uniform; i++) - { - ofsdru << this->dr_uniform * i << " " << dpsi_uniform[i] << std::endl;// output dphi/dr 2013-08-10 pengfei - } - ofsr.close(); - ofsk.close(); - ofsru.close(); - ofsdru.close(); // 13-08-10 pengfei - } - - return; + ModuleBase::TITLE("Numerical_Orbital_Lm","plot"); + + std::string orbital_type; + // Peize Lin update 2016-08-31 + if( 0==this->angular_momentum_l ) + { + orbital_type = 's'; + } + else if( 1==this->angular_momentum_l ) + { + orbital_type = 'p'; + } + else if( 2==this->angular_momentum_l ) + { + orbital_type = 'd'; + } + else if( 3<=this->angular_momentum_l && this->angular_momentum_l<=6 ) + { + orbital_type = 'f' + this->angular_momentum_l - 3; + } + else if( 7<=this->angular_momentum_l && this->angular_momentum_l<=11 ) + { + orbital_type = 'k' + this->angular_momentum_l - 7; + } + else + { + orbital_type = "L" + ModuleBase::GlobalFunc::TO_STRING(this->angular_momentum_l); + } + + if(GlobalV::MY_RANK==0) + { + std::stringstream ssr, ssk, ssru ,ssdru; // 2013-08-10 pengfei + ssr << ModuleBase::get_global_out_dir() << this->label << "/" + << this->label << "-"<< orbital_type << index_chi+1 << "-orbital-r.dat"; + + ssk << ModuleBase::get_global_out_dir() << this->label << "/" + << this->label << "-" << orbital_type << index_chi+1 << "-orbital-k.dat"; + + ssru << ModuleBase::get_global_out_dir() << this->label << "/" + << this->label << "-" << orbital_type << index_chi+1 << "-orbital-ru.dat"; + + ssdru << ModuleBase::get_global_out_dir() << this->label << "/" // 2013-08-10 pengfei + << this->label << "-" << orbital_type << index_chi+1 << "-orbital-dru.dat"; + + std::ofstream ofsr(ssr.str().c_str()); + std::ofstream ofsk(ssk.str().c_str()); + std::ofstream ofsru(ssru.str().c_str()); + std::ofstream ofsdru(ssdru.str().c_str()); // 2013-08-10 pengfei + + if (!ofsk || !ofsr || !ofsru || !ofsdru) // 2013-08-10 pengfei + { + ModuleBase::WARNING("Numerical_Orbital_Lm : plot", "Can't open files !"); + } + + for (int i = 0; i < this->nr; i++) + { + ofsr << this->r_radial[i] << " " << psi[i] << std::endl; + } + + for (int i = 0; i < this->nk; i++) + { + ofsk << this->k_radial[i] << " " << psik[i] << std::endl; + } + + for (int i = 0; i < this->nr_uniform; i++) + { + ofsru << this->dr_uniform * i << " " << psi_uniform[i] << std::endl; + } + + for (int i = 0; i < this->nr_uniform; i++) + { + ofsdru << this->dr_uniform * i << " " << dpsi_uniform[i] << std::endl;// output dphi/dr 2013-08-10 pengfei + } + ofsr.close(); + ofsk.close(); + ofsru.close(); + ofsdru.close(); // 13-08-10 pengfei + } + + return; } diff --git a/source/source_basis/module_ao/orb_atomic_lm.h b/source/source_basis/module_ao/orb_atomic_lm.h index e984092f74a..c4bf2b3360d 100644 --- a/source/source_basis/module_ao/orb_atomic_lm.h +++ b/source/source_basis/module_ao/orb_atomic_lm.h @@ -19,140 +19,140 @@ using std::vector; class Numerical_Orbital_Lm { - friend class Numerical_Orbital; - /// the unit test drives cal_kradial(), cal_kradial_sbpool(), - /// cal_rradial_sbpool() and plot() directly - friend class NumericalOrbitalLmTest; - - public: - - std::vector psi_uniform;// mohan add 2009-5-10 - std::vector dpsi_uniform; //liaochen add 2010/5/11 - std::vector ddpsi_uniform; //wenfei add 2022/7/13 - - int nr_uniform;// mohan add 2009-5-10 - double dr_uniform;// mohan add 2009-5-10 - double zty; ///< the valus of psi at 0. - - Numerical_Orbital_Lm(); - ~Numerical_Orbital_Lm(); - - // Peize Lin add 2017-12-12 - enum class Psi_Type{ Psi, Psif, Psik, Psik2 }; - - /// EXPLAIN : set information about Numerical_Orbital_Lm - void set_orbital_info - ( - const std::string &label_in, - const int &index_atom_type_in, - const int &angular_momentum_l_in, - const int &index_chi_in, - const int &nr_in, - const double *rab_in, - const double *r_radial_in, - const Psi_Type &psi_type, // Peize Lin add 2017-12-12 - const double *psi_in, - const int &nk_in, - const double &dk_in, - // Peize Lin delete lat0 2016-02-03 - const double &dr_uniform, - bool flag_plot, // Peize Lin add flag_plot 2016-08-31 - bool flag_sbpool, // Peize Lin add flag_sbpool 2017-10-02 - const bool &force_flag // mohan add 2021-05-07 - ); + friend class Numerical_Orbital; + /// the unit test drives cal_kradial(), cal_kradial_sbpool(), + /// cal_rradial_sbpool() and plot() directly + friend class NumericalOrbitalLmTest; + + public: + + std::vector psi_uniform;// mohan add 2009-5-10 + std::vector dpsi_uniform; //liaochen add 2010/5/11 + std::vector ddpsi_uniform; //wenfei add 2022/7/13 + + int nr_uniform;// mohan add 2009-5-10 + double dr_uniform;// mohan add 2009-5-10 + double zty; ///< the valus of psi at 0. + + Numerical_Orbital_Lm(); + ~Numerical_Orbital_Lm(); + + // Peize Lin add 2017-12-12 + enum class Psi_Type{ Psi, Psif, Psik, Psik2 }; + + /// EXPLAIN : set information about Numerical_Orbital_Lm + void set_orbital_info + ( + const std::string &label_in, + const int &index_atom_type_in, + const int &angular_momentum_l_in, + const int &index_chi_in, + const int &nr_in, + const double *rab_in, + const double *r_radial_in, + const Psi_Type &psi_type, // Peize Lin add 2017-12-12 + const double *psi_in, + const int &nk_in, + const double &dk_in, + // Peize Lin delete lat0 2016-02-03 + const double &dr_uniform, + bool flag_plot, // Peize Lin add flag_plot 2016-08-31 + bool flag_sbpool, // Peize Lin add flag_sbpool 2017-10-02 + const bool &force_flag // mohan add 2021-05-07 + ); private: - void copy_parameter( - const std::string &label_in, - const int &index_atom_type_in, - const int &angular_momentum_l_in, - const int &index_chi_in, - const int &nr_in, - const double *rab_in, - const double *r_radial_in, - const int &nk_in, - const double &dk_in, - const double &dr_uniform_in); - - void cal_kradial(void); - void cal_kradial_sbpool(void); - void cal_rradial_sbpool(void); - //void norm_test()const; - void plot()const; - //void use_uniform(const double &dr_uniform_in); - void extra_uniform(const double &dr_uniform_in, const bool &force_flag); - - std::string label; - int index_atom_type; - int angular_momentum_l; - int index_chi; - - int nr; - int nk; - - double rcut; - double kcut; - double dk; - - std::vector r_radial; /// k_radial; - - std::vector rab; - - std::vector psi; ///< psi(r) - std::vector psir; ///< psi(r) * r - std::vector psif; ///< psi(k) - std::vector psik; ///< psi(k) * k - std::vector psik2; ///< psi(k) * k^2 + void copy_parameter( + const std::string &label_in, + const int &index_atom_type_in, + const int &angular_momentum_l_in, + const int &index_chi_in, + const int &nr_in, + const double *rab_in, + const double *r_radial_in, + const int &nk_in, + const double &dk_in, + const double &dr_uniform_in); + + void cal_kradial(void); + void cal_kradial_sbpool(void); + void cal_rradial_sbpool(void); + //void norm_test()const; + void plot()const; + //void use_uniform(const double &dr_uniform_in); + void extra_uniform(const double &dr_uniform_in, const bool &force_flag); + + std::string label; + int index_atom_type; + int angular_momentum_l; + int index_chi; + + int nr; + int nk; + + double rcut; + double kcut; + double dk; + + std::vector r_radial; /// k_radial; + + std::vector rab; + + std::vector psi; ///< psi(r) + std::vector psir; ///< psi(r) * r + std::vector psif; ///< psi(k) + std::vector psik; ///< psi(k) * k + std::vector psik2; ///< psi(k) * k^2 public: - const std::string& getLabel() const { return label; } - const int& getType() const { return index_atom_type; } - const int& getL() const { return angular_momentum_l; } - const int& getChi() const { return index_chi; } - - const double* getPsiuniform() const { return ModuleBase::GlobalFunc::VECTOR_TO_PTR(psi_uniform); } - const double* getDpsiuniform() const { return ModuleBase::GlobalFunc::VECTOR_TO_PTR(dpsi_uniform); } - const int& getNruniform() const { return nr_uniform; } - const double& getDruniform() const { return dr_uniform; } - - const int& getNr() const { return nr; } - const int& getNk() const { return nk; } - - const double& getRcut() const { return rcut; } - const double& getKcut() const { return kcut; } - - const double* getRadial() const { return ModuleBase::GlobalFunc::VECTOR_TO_PTR(r_radial); } - const std::vector& get_r_radial() const { return r_radial; } - const double& getRadial(const int ir) const { return r_radial[ir]; } - - const double* getRab() const { return ModuleBase::GlobalFunc::VECTOR_TO_PTR(rab); } - const std::vector& get_rab() const { return rab; } - const double& getRab(const int ir) const { return rab[ir]; } - - const double& getDk()const { return dk; } - const double* getKpoint() const { return ModuleBase::GlobalFunc::VECTOR_TO_PTR(k_radial); } - const double& getKpoint(const int ik) const { return k_radial[ik]; } - const std::vector& get_k_radial() const { return k_radial; } - - const double* getPsi() const { return ModuleBase::GlobalFunc::VECTOR_TO_PTR(psi);} - const double& getPsi(const int ir) const { return psi[ir];} - const std::vector& get_psi() const { return psi; } - const double* getPsi_r() const { return ModuleBase::GlobalFunc::VECTOR_TO_PTR(psir); } - const double& getPsi_r(const int ir) const { return psir[ir]; } - const std::vector& get_psir() const { return psir; } - - const double* getPsif() const { return ModuleBase::GlobalFunc::VECTOR_TO_PTR(psif); } - const double& getPsif(const int ik) const { return psif[ik]; } - const std::vector& get_psif() const { return psif; } - const double* getPsi_k() const { return ModuleBase::GlobalFunc::VECTOR_TO_PTR(psik); } - const double& getPsi_k(const int ik) const { return psik[ik]; } - const std::vector& get_psi_k() const { return psik; } - const double* getPsi_k2() const { return ModuleBase::GlobalFunc::VECTOR_TO_PTR(psik2); } - const double& getPsi_k2(const int ik) const { return psik2[ik]; } - const std::vector& get_psi_k2() const { return psik2; } + const std::string& getLabel() const { return label; } + const int& getType() const { return index_atom_type; } + const int& getL() const { return angular_momentum_l; } + const int& getChi() const { return index_chi; } + + const double* getPsiuniform() const { return ModuleBase::GlobalFunc::VECTOR_TO_PTR(psi_uniform); } + const double* getDpsiuniform() const { return ModuleBase::GlobalFunc::VECTOR_TO_PTR(dpsi_uniform); } + const int& getNruniform() const { return nr_uniform; } + const double& getDruniform() const { return dr_uniform; } + + const int& getNr() const { return nr; } + const int& getNk() const { return nk; } + + const double& getRcut() const { return rcut; } + const double& getKcut() const { return kcut; } + + const double* getRadial() const { return ModuleBase::GlobalFunc::VECTOR_TO_PTR(r_radial); } + const std::vector& get_r_radial() const { return r_radial; } + const double& getRadial(const int ir) const { return r_radial[ir]; } + + const double* getRab() const { return ModuleBase::GlobalFunc::VECTOR_TO_PTR(rab); } + const std::vector& get_rab() const { return rab; } + const double& getRab(const int ir) const { return rab[ir]; } + + const double& getDk()const { return dk; } + const double* getKpoint() const { return ModuleBase::GlobalFunc::VECTOR_TO_PTR(k_radial); } + const double& getKpoint(const int ik) const { return k_radial[ik]; } + const std::vector& get_k_radial() const { return k_radial; } + + const double* getPsi() const { return ModuleBase::GlobalFunc::VECTOR_TO_PTR(psi);} + const double& getPsi(const int ir) const { return psi[ir];} + const std::vector& get_psi() const { return psi; } + const double* getPsi_r() const { return ModuleBase::GlobalFunc::VECTOR_TO_PTR(psir); } + const double& getPsi_r(const int ir) const { return psir[ir]; } + const std::vector& get_psir() const { return psir; } + + const double* getPsif() const { return ModuleBase::GlobalFunc::VECTOR_TO_PTR(psif); } + const double& getPsif(const int ik) const { return psif[ik]; } + const std::vector& get_psif() const { return psif; } + const double* getPsi_k() const { return ModuleBase::GlobalFunc::VECTOR_TO_PTR(psik); } + const double& getPsi_k(const int ik) const { return psik[ik]; } + const std::vector& get_psi_k() const { return psik; } + const double* getPsi_k2() const { return ModuleBase::GlobalFunc::VECTOR_TO_PTR(psik2); } + const double& getPsi_k2(const int ik) const { return psik2[ik]; } + const std::vector& get_psi_k2() const { return psik2; } }; #endif diff --git a/source/source_basis/module_ao/orb_gaunt_table.cpp b/source/source_basis/module_ao/orb_gaunt_table.cpp index 8caddfc4743..26c3e9329fe 100644 --- a/source/source_basis/module_ao/orb_gaunt_table.cpp +++ b/source/source_basis/module_ao/orb_gaunt_table.cpp @@ -20,14 +20,14 @@ void ORB_gaunt_table::init_Gaunt(const int &lmax) ModuleBase::TITLE("ORB_gaunt_table", "init_Gaunt"); ModuleBase::timer::start("ORB_gaunt_table", "init_Gaunt"); - this->Lmax_Gaunt_Coefficients = lmax; - const int nlm = (lmax * 2 + 1) * (lmax * 2 + 1); - this->Gaunt_Coefficients.create(nlm, nlm, nlm); + this->Lmax_Gaunt_Coefficients = lmax; + const int nlm = (lmax * 2 + 1) * (lmax * 2 + 1); + this->Gaunt_Coefficients.create(nlm, nlm, nlm); - // Becaful! , L ends at 2*lmax+1 + // Becaful! , L ends at 2*lmax+1 for (int L = 0; L < 2*lmax + 1; L++) { - // m order is ( 0,0,1,-1,0,1,-1,2,-2...) + // m order is ( 0,0,1,-1,0,1,-1,2,-2...) for (int m = 0; m < 2*L + 1 ; m++) { const int dim = this->get_lm_index(L,m); @@ -44,10 +44,10 @@ void ORB_gaunt_table::init_Gaunt(const int &lmax) ///////////////////// ///call Calculate::Cal_G //////////////////// - - Gaunt_Coefficients(dim1, dim2, dim) = - this->Get_Gaunt_SH (L1, m1, L2, m2, L, m); - }// m2 + + Gaunt_Coefficients(dim1, dim2, dim) = + this->Get_Gaunt_SH (L1, m1, L2, m2, L, m); + }// m2 }// L2 }// m1 }// L1 @@ -62,96 +62,96 @@ void ORB_gaunt_table::init_Gaunt(const int &lmax) double ORB_gaunt_table::Cal_Gaunt_single ( const int &L1, - const int &m1, + const int &m1, const int &L2, - const int &m2, + const int &m2, const int &L, - const int &m, + const int &m, const double &s1, - const double &e1, + const double &e1, const double &s2, - const double &e2 + const double &e2 ) { - ModuleBase::timer::start("ORB_gaunt_table", "Cal_Gaunt_single"); - if ((L1 - L2 - L) % 2 != 0) - { - return 0.0; - } - - double result = 0.0; - static double absc[16] = { - -0.9894009349916499, -0.9445750230732326, -0.8656312023878318, -0.755404408355003, - -0.6178762444026438, -0.4580167776572274, -0.2816035507792589, -0.09501250983763744, - 0.09501250983763744, 0.2816035507792589, 0.4580167776572274, 0.6178762444026438, - 0.755404408355003, 0.8656312023878318, 0.9445750230732326, 0.9894009349916499 }; - - static double weight[16] = { - 0.02715245941175406, 0.06225352393864778, 0.0951585116824929, 0.1246289712555339, - 0.1495959888165768, 0.1691565193950026, 0.1826034150449236, 0.1894506104550685, - 0.1894506104550685, 0.1826034150449236, 0.1691565193950026, 0.1495959888165768, - 0.1246289712555339, 0.0951585116824929, 0.06225352393864778, 0.02715245941175406 }; - - for (int i = 0;i < 16;i++) - { - for (int j = 0;j < 16;j++) - { - double theta = ((s1 + e1) + (e1 - s1) * absc[i]) / 2; - - result += weight[i] * weight[j] * sin(theta) * - this->Ylm_Gaunt( this->get_lm_index(L1, m1), 16 * i + j) * - this->Ylm_Gaunt( this->get_lm_index(L2, m2), 16 * i + j) * - this->Ylm_Gaunt( this->get_lm_index(L, m), 16 * i + j); - } - } - - result *= ((e1 - s1) / 2) * ((e2 - s2) / 2); - ModuleBase::timer::end("ORB_gaunt_table", "Cal_Gaunt_single"); - return result; + ModuleBase::timer::start("ORB_gaunt_table", "Cal_Gaunt_single"); + if ((L1 - L2 - L) % 2 != 0) + { + return 0.0; + } + + double result = 0.0; + static double absc[16] = { + -0.9894009349916499, -0.9445750230732326, -0.8656312023878318, -0.755404408355003, + -0.6178762444026438, -0.4580167776572274, -0.2816035507792589, -0.09501250983763744, + 0.09501250983763744, 0.2816035507792589, 0.4580167776572274, 0.6178762444026438, + 0.755404408355003, 0.8656312023878318, 0.9445750230732326, 0.9894009349916499 }; + + static double weight[16] = { + 0.02715245941175406, 0.06225352393864778, 0.0951585116824929, 0.1246289712555339, + 0.1495959888165768, 0.1691565193950026, 0.1826034150449236, 0.1894506104550685, + 0.1894506104550685, 0.1826034150449236, 0.1691565193950026, 0.1495959888165768, + 0.1246289712555339, 0.0951585116824929, 0.06225352393864778, 0.02715245941175406 }; + + for (int i = 0;i < 16;i++) + { + for (int j = 0;j < 16;j++) + { + double theta = ((s1 + e1) + (e1 - s1) * absc[i]) / 2; + + result += weight[i] * weight[j] * sin(theta) * + this->Ylm_Gaunt( this->get_lm_index(L1, m1), 16 * i + j) * + this->Ylm_Gaunt( this->get_lm_index(L2, m2), 16 * i + j) * + this->Ylm_Gaunt( this->get_lm_index(L, m), 16 * i + j); + } + } + + result *= ((e1 - s1) / 2) * ((e2 - s2) / 2); + ModuleBase::timer::end("ORB_gaunt_table", "Cal_Gaunt_single"); + return result; } */ /* void ORB_gaunt_table::init_Ylm_Gaunt ( - const int &lmax, + const int &lmax, const double &s1, - const double &e1, + const double &e1, const double &s2, - const double &e2 + const double &e2 ) { - ModuleBase::TITLE("ORB_gaunt_table", "init_Ylm_Gaunt"); - ModuleBase::timer::start("ORB_gaunt_table", "inite_Ylm_Gaunt"); + ModuleBase::TITLE("ORB_gaunt_table", "init_Ylm_Gaunt"); + ModuleBase::timer::start("ORB_gaunt_table", "inite_Ylm_Gaunt"); - const int nlm = (2*lmax+1) * (2*lmax+1); + const int nlm = (2*lmax+1) * (2*lmax+1); - static double absc[16] = { - -0.9894009349916499, -0.9445750230732326, -0.8656312023878318, -0.755404408355003, - -0.6178762444026438, -0.4580167776572274, -0.2816035507792589, -0.09501250983763744, - 0.09501250983763744, 0.2816035507792589, 0.4580167776572274, 0.6178762444026438, - 0.755404408355003, 0.8656312023878318, 0.9445750230732326, 0.9894009349916499 }; + static double absc[16] = { + -0.9894009349916499, -0.9445750230732326, -0.8656312023878318, -0.755404408355003, + -0.6178762444026438, -0.4580167776572274, -0.2816035507792589, -0.09501250983763744, + 0.09501250983763744, 0.2816035507792589, 0.4580167776572274, 0.6178762444026438, + 0.755404408355003, 0.8656312023878318, 0.9445750230732326, 0.9894009349916499 }; - //initialization of ylm_map + //initialization of ylm_map - ModuleBase::Vector3 g_gaunt[256]; + ModuleBase::Vector3 g_gaunt[256]; - this->Ylm_Gaunt.create(nlm , 256); + this->Ylm_Gaunt.create(nlm , 256); - for (int i = 0; i < 16; i++) - { - for (int j = 0; j < 16; j++) - { - const double theta = ((s1 + e1) + (e1 - s1) * absc[i]) / 2; - const double phi = ((s2 + e2) + (e2 - s2) * absc[j]) / 2; - ModuleBase::Vector3 u(sin(theta) * cos(phi), sin(theta) * sin(phi), cos(theta)); - g_gaunt[16*i+j] = u; - } - } + for (int i = 0; i < 16; i++) + { + for (int j = 0; j < 16; j++) + { + const double theta = ((s1 + e1) + (e1 - s1) * absc[i]) / 2; + const double phi = ((s2 + e2) + (e2 - s2) * absc[j]) / 2; + ModuleBase::Vector3 u(sin(theta) * cos(phi), sin(theta) * sin(phi), cos(theta)); + g_gaunt[16*i+j] = u; + } + } - ModuleBase::YlmReal::Ylm_Real(nlm, 256, &g_gaunt[0], this->Ylm_Gaunt); + ModuleBase::YlmReal::Ylm_Real(nlm, 256, &g_gaunt[0], this->Ylm_Gaunt); - ModuleBase::timer::start("ORB_gaunt_table", "init_Ylm_Gaunt"); - return; + ModuleBase::timer::start("ORB_gaunt_table", "init_Ylm_Gaunt"); + return; } */ @@ -159,294 +159,294 @@ void ORB_gaunt_table::init_Ylm_Gaunt ///effective pointers int ORB_gaunt_table::EP_EL(const int& L) { - if(L % 2 == 0) return (L+2) * (L+4) * (3*L*L+14*L+24) / 192; - else return (L+1) * (L+3) * (L+5) * (3*L+5) / 192; + if(L % 2 == 0) return (L+2) * (L+4) * (3*L*L+14*L+24) / 192; + else return (L+1) * (L+3) * (L+5) * (3*L+5) / 192; } int ORB_gaunt_table::index_func ( - const int& l1, - const int& l2, - const int& l3, - const int& m3 + const int& l1, + const int& l2, + const int& l3, + const int& m3 ) { - const int aux1 = l1*(l1*l1*l1+6*l1*l1+11*l1+6)/24; - const int aux2 = l2*(l2*l2+3*l2+2)/6; - const int aux3 = l3*(l3+1)/2; - - return aux1 + aux2 + aux3 + m3; + const int aux1 = l1*(l1*l1*l1+6*l1*l1+11*l1+6)/24; + const int aux2 = l2*(l2*l2+3*l2+2)/6; + const int aux3 = l3*(l3+1)/2; + + return aux1 + aux2 + aux3 + m3; } void ORB_gaunt_table::init_Gaunt_CH(const int& Lmax) { - ModuleBase::TITLE("ORB_gaunt_table","init_Gaunt_CH"); - ModuleBase::timer::start("ORB_gaunt_table","init_Gaunt_CH"); - - this->Lmax_Gaunt_CH = Lmax; - int L = 2*Lmax + 1; - int Eff_Np = this->EP_EL(L); - - ModuleBase::Memory::record("ORB::Gaunt_CH", sizeof(double) * Eff_Np * 30); - - int ic1 = 0; - for(int l1 = 0; l1 <= L; l1++) - { - for(int l2 = 0; l2 <= l1; l2++) - { - for(int l3 = 0; l3 <= l2; l3++) - { - for(int m3 = 0; m3 <= l3; m3++) - { - int idx = index_func(l1, l2, l3, m3); - assert(ic1 == idx); - - int l_sum = l1 + l2 + l3; - if((l_sum % 2 == 0) && (l2 + l3 >= l1)) - { - int uplmt_m2 = l1 - m3 > l2 ? l2 : l1 - m3; - - int ic2 = 0; - for(int m2 = -l2; m2 <= uplmt_m2; m2++) - { - //m1 + m2 + m3 == 0 - int m1 = -m2 - m3; - assert(std::abs(m1) <= l1); - - Gaunt_CH[ic1][ic2] = Calc_Gaunt_CH(l1, m1, l2, m2, l3, m3); - ic2++; - } - } - - ic1++; - }// m3 - }// l3 - }// l2 - } // l1 - - ModuleBase::timer::end("ORB_gaunt_table","init_Gaunt_CH"); - return; + ModuleBase::TITLE("ORB_gaunt_table","init_Gaunt_CH"); + ModuleBase::timer::start("ORB_gaunt_table","init_Gaunt_CH"); + + this->Lmax_Gaunt_CH = Lmax; + int L = 2*Lmax + 1; + int Eff_Np = this->EP_EL(L); + + ModuleBase::Memory::record("ORB::Gaunt_CH", sizeof(double) * Eff_Np * 30); + + int ic1 = 0; + for(int l1 = 0; l1 <= L; l1++) + { + for(int l2 = 0; l2 <= l1; l2++) + { + for(int l3 = 0; l3 <= l2; l3++) + { + for(int m3 = 0; m3 <= l3; m3++) + { + int idx = index_func(l1, l2, l3, m3); + assert(ic1 == idx); + + int l_sum = l1 + l2 + l3; + if((l_sum % 2 == 0) && (l2 + l3 >= l1)) + { + int uplmt_m2 = l1 - m3 > l2 ? l2 : l1 - m3; + + int ic2 = 0; + for(int m2 = -l2; m2 <= uplmt_m2; m2++) + { + //m1 + m2 + m3 == 0 + int m1 = -m2 - m3; + assert(std::abs(m1) <= l1); + + Gaunt_CH[ic1][ic2] = Calc_Gaunt_CH(l1, m1, l2, m2, l3, m3); + ic2++; + } + } + + ic1++; + }// m3 + }// l3 + }// l2 + } // l1 + + ModuleBase::timer::end("ORB_gaunt_table","init_Gaunt_CH"); + return; } //using wigner 3j expression double ORB_gaunt_table::Calc_Gaunt_CH ( - const int& l1, - const int& m1, - const int& l2, - const int& m2, - const int& l3, - const int& m3 + const int& l1, + const int& m1, + const int& l2, + const int& m2, + const int& l3, + const int& m3 ) { - ModuleBase::timer::start("ORB_gaunt_table","Calc_Gaunt_CH"); - - double fac = sqrt((2*l1+1) * (2*l2+1) * (2*l3+1) / ModuleBase::FOUR_PI); + ModuleBase::timer::start("ORB_gaunt_table","Calc_Gaunt_CH"); + + double fac = sqrt((2*l1+1) * (2*l2+1) * (2*l3+1) / ModuleBase::FOUR_PI); - int g = (l1+l2+l3)/2; - double triangle_f = sqrt( Fact(l1+l2-l3) * Fact(l1-l2+l3) * Fact(-l1+l2+l3) / Fact(2*g+1) ); + int g = (l1+l2+l3)/2; + double triangle_f = sqrt( Fact(l1+l2-l3) * Fact(l1-l2+l3) * Fact(-l1+l2+l3) / Fact(2*g+1) ); - fac *= pow(-1.0, g) * triangle_f * Fact(g) / Fact(g-l1) / Fact(g-l2) / Fact(g-l3); + fac *= pow(-1.0, g) * triangle_f * Fact(g) / Fact(g-l1) / Fact(g-l2) / Fact(g-l3); - double aux1 = sqrt(Fact(l1+m1) * Fact(l1-m1) * Fact(l2+m2) * Fact(l2-m2) * Fact(l3+m3) * Fact(l3-m3)); + double aux1 = sqrt(Fact(l1+m1) * Fact(l1-m1) * Fact(l2+m2) * Fact(l2-m2) * Fact(l3+m3) * Fact(l3-m3)); - int kmin = 0; + int kmin = 0; int kmax = 0; - - kmin = (l2-l3-m1) > (l1-l3+m2) ? (l2-l3-m1) : (l1-l3+m2); - kmin = kmin > 0 ? kmin : 0; - - kmax = (l1+l2-l3) > (l1-m1) ? (l1-m1) : (l1+l2-l3); - kmax = kmax > (l2+m2) ? (l2+m2) : kmax; - - double aux2 = 0.0; - for(int k = kmin; k <= kmax; k++) - { - aux2 += pow(-1.0, k) / Fact(k) / Fact(l1+l2-l3-k) / Fact(l1-m1-k) / Fact(l2+m2-k) - / Fact(l3-l2+m1+k) / Fact(l3-l1+k-m2); - } - - ModuleBase::timer::end("ORB_gaunt_table","Calc_Gaunt_CH"); - return fac * pow(-1.0, l1-l2-m3) * triangle_f * aux1 * aux2; + + kmin = (l2-l3-m1) > (l1-l3+m2) ? (l2-l3-m1) : (l1-l3+m2); + kmin = kmin > 0 ? kmin : 0; + + kmax = (l1+l2-l3) > (l1-m1) ? (l1-m1) : (l1+l2-l3); + kmax = kmax > (l2+m2) ? (l2+m2) : kmax; + + double aux2 = 0.0; + for(int k = kmin; k <= kmax; k++) + { + aux2 += pow(-1.0, k) / Fact(k) / Fact(l1+l2-l3-k) / Fact(l1-m1-k) / Fact(l2+m2-k) + / Fact(l3-l2+m1+k) / Fact(l3-l1+k-m2); + } + + ModuleBase::timer::end("ORB_gaunt_table","Calc_Gaunt_CH"); + return fac * pow(-1.0, l1-l2-m3) * triangle_f * aux1 * aux2; } - + double ORB_gaunt_table::Get_Gaunt_CH ( - const int& l1, - const int& m1, - const int& l2, - const int& m2, - const int& l3, - const int& m3 + const int& l1, + const int& m1, + const int& l2, + const int& m2, + const int& l3, + const int& m3 ) { - assert(l1 >= 0); - assert(l2 >= 0); - assert(l3 >= 0); - - int l_sum = l1 + l2 + l3; - - if(l_sum % 2 == 1) return 0.0; - - if(std::abs(m1) > l1 || std::abs(m2) > l2 || std::abs(m3) > l3) return 0.0; - - if( (m1 + m2 + m3) != 0) return 0.0; + assert(l1 >= 0); + assert(l2 >= 0); + assert(l3 >= 0); + + int l_sum = l1 + l2 + l3; + + if(l_sum % 2 == 1) return 0.0; + + if(std::abs(m1) > l1 || std::abs(m2) > l2 || std::abs(m3) > l3) return 0.0; + + if( (m1 + m2 + m3) != 0) return 0.0; int L1 = l1; - int M1 = m1; - int L2 = l2; + int M1 = m1; + int L2 = l2; int M2 = m2; - Swap(L1, M1, L2, M2); - - int L3 = l3; - int M3 = m3; - Swap(L1, M1, L3, M3); - - Swap(L2, M2, L3, M3); - - if(M3 < 0) - { - M1 = -M1; - M2 = -M2; - M3 = -M3; - } - - int ic1 = index_func(L1, L2, L3, M3); - int ic2 = M2 + L2; - - try - { - return Gaunt_CH.at(ic1).at(ic2); - } // Peize Lin add 2016-08-26 - catch( std::out_of_range ) - { - return 0; - } + Swap(L1, M1, L2, M2); + + int L3 = l3; + int M3 = m3; + Swap(L1, M1, L3, M3); + + Swap(L2, M2, L3, M3); + + if(M3 < 0) + { + M1 = -M1; + M2 = -M2; + M3 = -M3; + } + + int ic1 = index_func(L1, L2, L3, M3); + int ic2 = M2 + L2; + + try + { + return Gaunt_CH.at(ic1).at(ic2); + } // Peize Lin add 2016-08-26 + catch( std::out_of_range ) + { + return 0; + } } - + ///Input value, ///m1, m2, m3 are restricted within 0 to 2l+1, ///and should be transformed first. double ORB_gaunt_table::Get_Gaunt_SH ( - const int& l1, - const int& mm1, - const int& l2, - const int& mm2, - const int& l3, - const int& mm3 + const int& l1, + const int& mm1, + const int& l2, + const int& mm2, + const int& l3, + const int& mm3 ) { - - //Tranform M index - int m1 = Index_M(mm1); - int m2 = Index_M(mm2); - int m3 = Index_M(mm3); - - if(m1 >= 0 && m2 >= 0 && m3 >= 0) - { - if(m1 * m2 * m3 > 0) - { - if(m1 == m2 + m3) return pow(-1.0, m1) * sqrt(2.0) / 2.0 - * Get_Gaunt_CH(l1, -m1, l2, m2, l3, m3); - else if(m2 == m1 + m3) return pow(-1.0, m2) * sqrt(2.0) / 2.0 - * Get_Gaunt_CH(l1, m1, l2, -m2, l3, m3); - else if(m3 == m1 + m2) return pow(-1.0, m3) * sqrt(2.0) / 2.0 - * Get_Gaunt_CH(l1, m1, l2, m2, l3, -m3); - else return 0.0; - } - else - { - if(m1 == 0 && m2 == 0 && m3 == 0) return Get_Gaunt_CH(l1, 0, l2, 0, l3, 0); - else if( (m1 == 0) && (m2 == m3)) return pow(-1.0, m2) * Get_Gaunt_CH(l1, 0, l2, m2, l3, -m2); - else if( (m2 == 0) && (m3 == m1)) return pow(-1.0, m1) * Get_Gaunt_CH(l2, 0, l1, m1, l3, -m1); - else if( (m3 == 0) && (m1 == m2)) return pow(-1.0, m2) * Get_Gaunt_CH(l3, 0, l2, m2, l1, -m2); - else return 0.0; - } - } - else - { - if(m1 >= 0 && m2 < 0 && m3 < 0) - { - if((m1 == 0) && (m2 == m3)) return pow(-1.0, m2) * Get_Gaunt_CH(l1, 0, l2, m2, l3, -m2); - else if(m1 > 0 && (m2 == m1+m3)) - return pow(-1.0, m3) * sqrt(2.0) / 2.0 * Get_Gaunt_CH(l1, m1, l2, -m2, l3, m3); - else if(m1 > 0 && (m3 == m1+m2)) - return pow(-1.0, m2) * sqrt(2.0) / 2.0 * Get_Gaunt_CH(l1, m1, l2, m2, l3, -m3); - else if(m1 > 0 && ( (m1+m2+m3) == 0)) - return pow(-1.0, m1+1) * sqrt(2.0) / 2.0 * Get_Gaunt_CH(l1, m1, l2, m2, l3, m3); - else return 0.0; - } - else if(m2 >= 0 && m1 < 0 && m3 < 0) - { - if((m2 == 0) && (m1 == m3)) return pow(-1.0, m1) * Get_Gaunt_CH(l2, 0, l1, m1, l3, -m1); - else if(m2 > 0 && (m1 == (m2 + m3))) - return pow(-1.0, m3) * sqrt(2.0) / 2.0 * Get_Gaunt_CH(l2, m2, l1, -m1, l3, m3); - else if(m2 > 0 && (m3 == (m2 + m1))) - return pow(-1.0, m1) * sqrt(2.0) / 2.0 * Get_Gaunt_CH(l2, m2, l1, m1, l3, -m3); - else if(m2 > 0 && ((m1+m2+m3) == 0)) - return pow(-1.0, m2+1) * sqrt(2.0) / 2.0 * Get_Gaunt_CH(l2, m2, l1, m1, l3, m3); - else return 0.0; - - } - else if(m3 >= 0 && m1 < 0 && m2 < 0) - { - if((m3 == 0) && (m1 == m2)) return pow(-1.0, m1) * Get_Gaunt_CH(l3, 0, l1, m1, l2, -m1); - else if(m3 > 0 && (m1 == m3+m2)) - return pow(-1.0, m2) * sqrt(2.0) / 2.0 * Get_Gaunt_CH(l3, m3, l1, -m1, l2, m2); - else if(m3 > 0 && (m2 == m3+m1)) - return pow(-1.0, m1) * sqrt(2.0) / 2.0 * Get_Gaunt_CH(l3, m3, l1, m1, l2, -m2); - else if(m3 > 0 && ( (m1+m2+m3) == 0)) - return pow(-1.0, m3+1) * sqrt(2.0) / 2.0 * Get_Gaunt_CH(l3, m3, l1, m1, l2, m2); - else return 0.0; - } - else return 0.0; - } + + //Tranform M index + int m1 = Index_M(mm1); + int m2 = Index_M(mm2); + int m3 = Index_M(mm3); + + if(m1 >= 0 && m2 >= 0 && m3 >= 0) + { + if(m1 * m2 * m3 > 0) + { + if(m1 == m2 + m3) return pow(-1.0, m1) * sqrt(2.0) / 2.0 + * Get_Gaunt_CH(l1, -m1, l2, m2, l3, m3); + else if(m2 == m1 + m3) return pow(-1.0, m2) * sqrt(2.0) / 2.0 + * Get_Gaunt_CH(l1, m1, l2, -m2, l3, m3); + else if(m3 == m1 + m2) return pow(-1.0, m3) * sqrt(2.0) / 2.0 + * Get_Gaunt_CH(l1, m1, l2, m2, l3, -m3); + else return 0.0; + } + else + { + if(m1 == 0 && m2 == 0 && m3 == 0) return Get_Gaunt_CH(l1, 0, l2, 0, l3, 0); + else if( (m1 == 0) && (m2 == m3)) return pow(-1.0, m2) * Get_Gaunt_CH(l1, 0, l2, m2, l3, -m2); + else if( (m2 == 0) && (m3 == m1)) return pow(-1.0, m1) * Get_Gaunt_CH(l2, 0, l1, m1, l3, -m1); + else if( (m3 == 0) && (m1 == m2)) return pow(-1.0, m2) * Get_Gaunt_CH(l3, 0, l2, m2, l1, -m2); + else return 0.0; + } + } + else + { + if(m1 >= 0 && m2 < 0 && m3 < 0) + { + if((m1 == 0) && (m2 == m3)) return pow(-1.0, m2) * Get_Gaunt_CH(l1, 0, l2, m2, l3, -m2); + else if(m1 > 0 && (m2 == m1+m3)) + return pow(-1.0, m3) * sqrt(2.0) / 2.0 * Get_Gaunt_CH(l1, m1, l2, -m2, l3, m3); + else if(m1 > 0 && (m3 == m1+m2)) + return pow(-1.0, m2) * sqrt(2.0) / 2.0 * Get_Gaunt_CH(l1, m1, l2, m2, l3, -m3); + else if(m1 > 0 && ( (m1+m2+m3) == 0)) + return pow(-1.0, m1+1) * sqrt(2.0) / 2.0 * Get_Gaunt_CH(l1, m1, l2, m2, l3, m3); + else return 0.0; + } + else if(m2 >= 0 && m1 < 0 && m3 < 0) + { + if((m2 == 0) && (m1 == m3)) return pow(-1.0, m1) * Get_Gaunt_CH(l2, 0, l1, m1, l3, -m1); + else if(m2 > 0 && (m1 == (m2 + m3))) + return pow(-1.0, m3) * sqrt(2.0) / 2.0 * Get_Gaunt_CH(l2, m2, l1, -m1, l3, m3); + else if(m2 > 0 && (m3 == (m2 + m1))) + return pow(-1.0, m1) * sqrt(2.0) / 2.0 * Get_Gaunt_CH(l2, m2, l1, m1, l3, -m3); + else if(m2 > 0 && ((m1+m2+m3) == 0)) + return pow(-1.0, m2+1) * sqrt(2.0) / 2.0 * Get_Gaunt_CH(l2, m2, l1, m1, l3, m3); + else return 0.0; + + } + else if(m3 >= 0 && m1 < 0 && m2 < 0) + { + if((m3 == 0) && (m1 == m2)) return pow(-1.0, m1) * Get_Gaunt_CH(l3, 0, l1, m1, l2, -m1); + else if(m3 > 0 && (m1 == m3+m2)) + return pow(-1.0, m2) * sqrt(2.0) / 2.0 * Get_Gaunt_CH(l3, m3, l1, -m1, l2, m2); + else if(m3 > 0 && (m2 == m3+m1)) + return pow(-1.0, m1) * sqrt(2.0) / 2.0 * Get_Gaunt_CH(l3, m3, l1, m1, l2, -m2); + else if(m3 > 0 && ( (m1+m2+m3) == 0)) + return pow(-1.0, m3+1) * sqrt(2.0) / 2.0 * Get_Gaunt_CH(l3, m3, l1, m1, l2, m2); + else return 0.0; + } + else return 0.0; + } } double ORB_gaunt_table::Fact(const int& n) { - double val = 1.0; - for(int i = 1; i <= n; i++) - { - val *= static_cast(i); - } - return val; + double val = 1.0; + for(int i = 1; i <= n; i++) + { + val *= static_cast(i); + } + return val; } void ORB_gaunt_table::Swap( - int& l1, - int& m1, - int& l2, - int & m2) + int& l1, + int& m1, + int& l2, + int & m2) { - int tmp1=0, tmp2=0; - if(l1 >= l2) return; - else - { - tmp1 = l2; - tmp2 = m2; - - l2 = l1; - m2 = m1; - - l1 = tmp1; - m1 = tmp2; - } - return; + int tmp1=0, tmp2=0; + if(l1 >= l2) return; + else + { + tmp1 = l2; + tmp2 = m2; + + l2 = l1; + m2 = m1; + + l1 = tmp1; + m1 = tmp2; + } + return; } int ORB_gaunt_table::Index_M(const int& m) { - if(m % 2 == 0) return (- m / 2); - else return ((m+1) / 2); + if(m % 2 == 0) return (- m / 2); + else return ((m+1) / 2); } diff --git a/source/source_basis/module_ao/orb_gaunt_table.h b/source/source_basis/module_ao/orb_gaunt_table.h index 0e9aefdb701..a9d9b5c2f7e 100644 --- a/source/source_basis/module_ao/orb_gaunt_table.h +++ b/source/source_basis/module_ao/orb_gaunt_table.h @@ -7,126 +7,126 @@ class ORB_gaunt_table { - public: - - ORB_gaunt_table(); - ~ORB_gaunt_table(); - - /** - * Method 2: - * using WIgner 3j symbols - * \f$ Y(l1,m1), Y(l2,m2), Y(L,M) \f$ - */ - - void init_Gaunt_CH(const int& Lmax); - double Get_Gaunt_CH( - const int& l1, - const int& m1, - const int& l2, - const int& m2, - const int& l3, - const int& m3 ); - - ///M defined here are restricted within 0 to 2l+1 - /// - ///should be transformed first - double Get_Gaunt_SH( - const int& l1, - const int& mm1, - const int& l2, - const int& mm2, - const int& l3, - const int& mm3 ); - - double Calc_Gaunt_CH( - const int& l1, - const int& m1, - const int& l2, - const int& m2, - const int& l3, - const int& m3 ); - - - /** - * MEthod 2 - * - * Directly Calculate integral of - * \f$ S(l_1,m_1), S(l_2,m_2), S(L,M) \f$ - */ - ModuleBase::realArray Gaunt_Coefficients; - - /// (1) Make Ylm_Gaunt Table. - ///---------------- + public: + + ORB_gaunt_table(); + ~ORB_gaunt_table(); + + /** + * Method 2: + * using WIgner 3j symbols + * \f$ Y(l1,m1), Y(l2,m2), Y(L,M) \f$ + */ + + void init_Gaunt_CH(const int& Lmax); + double Get_Gaunt_CH( + const int& l1, + const int& m1, + const int& l2, + const int& m2, + const int& l3, + const int& m3 ); + + ///M defined here are restricted within 0 to 2l+1 + /// + ///should be transformed first + double Get_Gaunt_SH( + const int& l1, + const int& mm1, + const int& l2, + const int& mm2, + const int& l3, + const int& mm3 ); + + double Calc_Gaunt_CH( + const int& l1, + const int& m1, + const int& l2, + const int& m2, + const int& l3, + const int& m3 ); + + + /** + * MEthod 2 + * + * Directly Calculate integral of + * \f$ S(l_1,m_1), S(l_2,m_2), S(L,M) \f$ + */ + ModuleBase::realArray Gaunt_Coefficients; + + /// (1) Make Ylm_Gaunt Table. + ///---------------- /* - void init_Ylm_Gaunt( - const int &lmax, - const double &s1, - const double &e1, - const double &s2, - const double &e2); + void init_Ylm_Gaunt( + const int &lmax, + const double &s1, + const double &e1, + const double &s2, + const double &e2); */ - /// (2) Use Ylm_Gaunt to calculate Gaunt Coefficinets element - ///------ + /// (2) Use Ylm_Gaunt to calculate Gaunt Coefficinets element + ///------ /* - double Cal_Gaunt_single( - const int &l1, - const int &m1, - const int &l2, - const int &m2, - const int &l, - const int &m, - const double &s1, - const double &e1, - const double &s2, - const double &e2); + double Cal_Gaunt_single( + const int &l1, + const int &m1, + const int &l2, + const int &m2, + const int &l, + const int &m, + const double &s1, + const double &e1, + const double &s2, + const double &e2); */ - /// (3) Make the whole Gaunt Coefficients table - /// ------------------------------------ - void init_Gaunt(const int &lmax); - - static inline int get_lm_index(const int l, const int m) - { - return l*l+m; - } - - static int Index_M(const int& m); - - int get_Lmax_Gaunt_Coefficients() const { return Lmax_Gaunt_Coefficients; } - int get_Lmax_Gaunt_CH() const { return Lmax_Gaunt_CH; } - - private: - - // Index Function - // Yu's mehtod - // Peize Lin delete void ModuleBase::GlobalFunc::ZEROS(); 2016-08-26 - - //int P_EL(const int& L); - - int EP_EL(const int& L); - - int index_func( - const int& l1, - const int& l2, - const int& l3, - const int& m3 ); - - double Fact(const int& n); - - void Swap( - int& l1, - int& m1, - int& l2, - int& m2 ); - - //2*Lmax+1 - std::map> Gaunt_CH; // Peize Lin update 2016-08-26 - - //direct integral - ModuleBase::matrix Ylm_Gaunt; - - int Lmax_Gaunt_Coefficients = -1; - int Lmax_Gaunt_CH = -1; + /// (3) Make the whole Gaunt Coefficients table + /// ------------------------------------ + void init_Gaunt(const int &lmax); + + static inline int get_lm_index(const int l, const int m) + { + return l*l+m; + } + + static int Index_M(const int& m); + + int get_Lmax_Gaunt_Coefficients() const { return Lmax_Gaunt_Coefficients; } + int get_Lmax_Gaunt_CH() const { return Lmax_Gaunt_CH; } + + private: + + // Index Function + // Yu's mehtod + // Peize Lin delete void ModuleBase::GlobalFunc::ZEROS(); 2016-08-26 + + //int P_EL(const int& L); + + int EP_EL(const int& L); + + int index_func( + const int& l1, + const int& l2, + const int& l3, + const int& m3 ); + + double Fact(const int& n); + + void Swap( + int& l1, + int& m1, + int& l2, + int& m2 ); + + //2*Lmax+1 + std::map> Gaunt_CH; // Peize Lin update 2016-08-26 + + //direct integral + ModuleBase::matrix Ylm_Gaunt; + + int Lmax_Gaunt_Coefficients = -1; + int Lmax_Gaunt_CH = -1; }; #endif diff --git a/source/source_basis/module_ao/orb_nonlocal.cpp b/source/source_basis/module_ao/orb_nonlocal.cpp index 45398deef34..03fcee98624 100644 --- a/source/source_basis/module_ao/orb_nonlocal.cpp +++ b/source/source_basis/module_ao/orb_nonlocal.cpp @@ -3,66 +3,66 @@ Numerical_Nonlocal::Numerical_Nonlocal() { - //make std::pair of new and delete - //question remains - this->type = 0; - this->lmax = 0; - this->rcut_max = 0.0; - this->Proj = new Numerical_Nonlocal_Lm[1]; - this->nproj = -1; - //zhengdy-soc, for optimize nonlocal part + //make std::pair of new and delete + //question remains + this->type = 0; + this->lmax = 0; + this->rcut_max = 0.0; + this->Proj = new Numerical_Nonlocal_Lm[1]; + this->nproj = -1; + //zhengdy-soc, for optimize nonlocal part } Numerical_Nonlocal::~Numerical_Nonlocal() { - delete[] Proj; + delete[] Proj; } void Numerical_Nonlocal::set_type_info ( - const int& type_in, - const std::string& label_in, - const std::string& type_ps_in, - const int& lmax_in, - const int& nproj_in, - const Numerical_Nonlocal_Lm* Proj_in + const int& type_in, + const std::string& label_in, + const std::string& type_ps_in, + const int& lmax_in, + const int& nproj_in, + const Numerical_Nonlocal_Lm* Proj_in ) { - //ModuleBase::TITLE("Numerical_Nonlocal","set_type_info"); + //ModuleBase::TITLE("Numerical_Nonlocal","set_type_info"); - this->type = type_in; - this->label = label_in; - this->type_ps = type_ps_in; + this->type = type_in; + this->label = label_in; + this->type_ps = type_ps_in; - if (lmax_in < -1 || lmax_in > 20) - { - ModuleBase::WARNING_QUIT("Numerical_Nonlocal", "bad input of lmax : should be between -1 and 20"); - } + if (lmax_in < -1 || lmax_in > 20) + { + ModuleBase::WARNING_QUIT("Numerical_Nonlocal", "bad input of lmax : should be between -1 and 20"); + } - this->lmax = lmax_in; + this->lmax = lmax_in; - this->nproj = nproj_in; + this->nproj = nproj_in; - assert(nproj >= 0); + assert(nproj >= 0); //---------------------------------------------------------- // EXPLAIN : non_local pseudopotential projector for each l //---------------------------------------------------------- - //only store radial function - delete[] Proj; - this->Proj = nullptr; - this->Proj = new Numerical_Nonlocal_Lm[this->nproj]; + //only store radial function + delete[] Proj; + this->Proj = nullptr; + this->Proj = new Numerical_Nonlocal_Lm[this->nproj]; - for (int p1=0; p1Proj[p1] = Proj_in[p1]; - } + for (int p1=0; p1Proj[p1] = Proj_in[p1]; + } - this->rcut_max = 0.0; - for(int p1=0; p1rcut_max = std::max( this->Proj[p1].getRcut(), rcut_max ); - } - return; + this->rcut_max = 0.0; + for(int p1=0; p1rcut_max = std::max( this->Proj[p1].getRcut(), rcut_max ); + } + return; } diff --git a/source/source_basis/module_ao/orb_nonlocal.h b/source/source_basis/module_ao/orb_nonlocal.h index a4ed2886532..28ec152fb27 100644 --- a/source/source_basis/module_ao/orb_nonlocal.h +++ b/source/source_basis/module_ao/orb_nonlocal.h @@ -21,44 +21,49 @@ class Numerical_Nonlocal { public: - Numerical_Nonlocal(); - ~Numerical_Nonlocal(); + Numerical_Nonlocal(); + ~Numerical_Nonlocal(); - const int& getLmax() const { return this->lmax; } + // This class owns the raw Proj array; copying it would alias the buffer + // and cause double free, so copy semantics are explicitly forbidden. + Numerical_Nonlocal(const Numerical_Nonlocal&) = delete; + Numerical_Nonlocal& operator=(const Numerical_Nonlocal&) = delete; - const int& getType() const { return this->type; } + const int& getLmax() const { return this->lmax; } - const std::string& getLabel() const { return this->label; } + const int& getType() const { return this->type; } - const std::string& getType_ps() const { return this->type_ps; } + const std::string& getLabel() const { return this->label; } + const std::string& getType_ps() const { return this->type_ps; } - void set_type_info( - const int& type_in, - const std::string& label_in, - const std::string& type_ps_in, - const int& lmax_in, - const int& nproj_in, - const Numerical_Nonlocal_Lm* ps_orbital_in); - Numerical_Nonlocal_Lm* Proj = nullptr; ///< length: nproj(only store radial function ) + void set_type_info( + const int& type_in, + const std::string& label_in, + const std::string& type_ps_in, + const int& lmax_in, + const int& nproj_in, + const Numerical_Nonlocal_Lm* ps_orbital_in); - const double& get_rcut_max() const { return rcut_max; } + Numerical_Nonlocal_Lm* Proj = nullptr; ///< length: nproj(only store radial function ) + + const double& get_rcut_max() const { return rcut_max; } const int& get_nproj() const { return nproj; } - private: - - std::string label=""; /// renew(); + label = ""; + index_atom_type = 0; + angular_momentum_l = 0; + index_proj = 0; + + nr = 1; + nk = 1; + + rcut = 0.0; + kcut = 0.0; + dk = 0.0; + + nr_uniform = 1; + dr_uniform = -1.0; + this->renew(); } Numerical_Nonlocal_Lm::~Numerical_Nonlocal_Lm() { - this->freemem(); + this->freemem(); } void Numerical_Nonlocal_Lm::renew() { - assert(nr_uniform>0); - assert(nr>0); - assert(nk>0); - this->r_radial = new double[nr]; - this->rab = new double[nr]; - this->beta_r = new double[nr]; - this->beta_uniform = new double[nr_uniform]; - this->dbeta_uniform = new double[nr_uniform]; - this->k_radial = new double[nk]; - this->beta_k = new double[nk]; - ModuleBase::GlobalFunc::ZEROS(r_radial, nr); - ModuleBase::GlobalFunc::ZEROS(rab, nr); - ModuleBase::GlobalFunc::ZEROS(beta_r, nr); - ModuleBase::GlobalFunc::ZEROS(beta_uniform, nr_uniform); - ModuleBase::GlobalFunc::ZEROS(dbeta_uniform, nr_uniform); - ModuleBase::GlobalFunc::ZEROS(k_radial, nk); - ModuleBase::GlobalFunc::ZEROS(beta_k, nk); + assert(nr_uniform>0); + assert(nr>0); + assert(nk>0); + this->r_radial = new double[nr]; + this->rab = new double[nr]; + this->beta_r = new double[nr]; + this->beta_uniform = new double[nr_uniform]; + this->dbeta_uniform = new double[nr_uniform]; + this->k_radial = new double[nk]; + this->beta_k = new double[nk]; + ModuleBase::GlobalFunc::ZEROS(r_radial, nr); + ModuleBase::GlobalFunc::ZEROS(rab, nr); + ModuleBase::GlobalFunc::ZEROS(beta_r, nr); + ModuleBase::GlobalFunc::ZEROS(beta_uniform, nr_uniform); + ModuleBase::GlobalFunc::ZEROS(dbeta_uniform, nr_uniform); + ModuleBase::GlobalFunc::ZEROS(k_radial, nk); + ModuleBase::GlobalFunc::ZEROS(beta_k, nk); } void Numerical_Nonlocal_Lm::freemem() { - delete[] this->r_radial; - delete[] this->rab; - delete[] this->beta_r; - delete[] this->beta_uniform; - delete[] this->dbeta_uniform; - delete[] this->k_radial; - delete[] this->beta_k; + delete[] this->r_radial; + delete[] this->rab; + delete[] this->beta_r; + delete[] this->beta_uniform; + delete[] this->dbeta_uniform; + delete[] this->k_radial; + delete[] this->beta_k; r_radial = nullptr; rab = nullptr; @@ -83,49 +83,49 @@ Numerical_Nonlocal_Lm& Numerical_Nonlocal_Lm::operator= const Numerical_Nonlocal_Lm & nol ) { - this->label = nol.label; - this->index_atom_type = nol.index_atom_type; - this->angular_momentum_l = nol.angular_momentum_l; + this->label = nol.label; + this->index_atom_type = nol.index_atom_type; + this->angular_momentum_l = nol.angular_momentum_l; this->index_proj = nol.index_proj; - this->nr = nol.nr; - this->nk = nol.nk; + this->nr = nol.nr; + this->nk = nol.nk; - this->nr_uniform = nol.nr_uniform; - this->dr_uniform = nol.dr_uniform; + this->nr_uniform = nol.nr_uniform; + this->dr_uniform = nol.dr_uniform; - this->rcut = nol.rcut; - this->kcut = nol.kcut; + this->rcut = nol.rcut; + this->kcut = nol.kcut; - this->dk = nol.dk; + this->dk = nol.dk; - this->freemem(); - this->renew(); + this->freemem(); + this->renew(); - for (int ir = 0; ir < nol.nr; ir++) - { - this->r_radial[ir] = nol.r_radial[ir]; - this->rab[ir] = nol.rab[ir]; - this->beta_r[ir] = nol.beta_r[ir]; - } + for (int ir = 0; ir < nol.nr; ir++) + { + this->r_radial[ir] = nol.r_radial[ir]; + this->rab[ir] = nol.rab[ir]; + this->beta_r[ir] = nol.beta_r[ir]; + } - for (int ir = 0; ir < nr_uniform; ir++) - { - this->beta_uniform[ir] = nol.beta_uniform[ir]; - this->dbeta_uniform[ir] = nol.dbeta_uniform[ir]; - } + for (int ir = 0; ir < nr_uniform; ir++) + { + this->beta_uniform[ir] = nol.beta_uniform[ir]; + this->dbeta_uniform[ir] = nol.dbeta_uniform[ir]; + } - for (int ik = 0; ik < nol.nk; ik++) - { - this->k_radial[ik] = nol.k_radial[ik]; - this->beta_k[ik] = nol.beta_k[ik]; - } + for (int ik = 0; ik < nol.nk; ik++) + { + this->k_radial[ik] = nol.k_radial[ik]; + this->beta_k[ik] = nol.beta_k[ik]; + } - return *this; + return *this; } void Numerical_Nonlocal_Lm::set_NL_proj( - const std::string &label_in, + const std::string &label_in, const int &index_atom_type_in, const int &angular_momentum_l_in, const int &nr_in, @@ -134,51 +134,51 @@ void Numerical_Nonlocal_Lm::set_NL_proj( const double *beta_r_in, const int &nk_in, const double &dk_in, - const double &dr_uniform_in) + const double &dr_uniform_in) { - this->label = label_in; - this->index_atom_type = index_atom_type_in; - - this->angular_momentum_l = angular_momentum_l_in; - assert(angular_momentum_l_in>=-1); // -1 means no angular momentum. - - this->dr_uniform = dr_uniform_in; - - this->nr = nr_in; - assert(nr_in>1 && nr_in <10000); - assert(nr%2!=0); + this->label = label_in; + this->index_atom_type = index_atom_type_in; + + this->angular_momentum_l = angular_momentum_l_in; + assert(angular_momentum_l_in>=-1); // -1 means no angular momentum. + + this->dr_uniform = dr_uniform_in; + + this->nr = nr_in; + assert(nr_in>1 && nr_in <10000); + assert(nr%2!=0); - this->rcut = r_radial_in[nr-1]; + this->rcut = r_radial_in[nr-1]; assert(rcut>=0.0); - this->nk = nk_in; - assert(nk%2!=0); + this->nk = nk_in; + assert(nk%2!=0); - this->dk = dk_in; - assert(dk>0.0); + this->dk = dk_in; + assert(dk>0.0); this->freemem(); this->renew(); - - for (int ir = 0; ir < nr; ir++) - { - this->r_radial[ir] = r_radial_in[ir]; - this->rab[ir] = rab_in[ir]; - this->beta_r[ir] = beta_r_in[ir]; - } - - for (int ik = 0; ik < nk; ik++) - { - this->k_radial[ik] = ik * this->dk; - } - this->kcut = (nk-1) * this->dk; - - // (1) extra the uniform mesh - //this->extra_uniform(dr_uniform); - // (2) get the beta_k - this->get_kradial(); - - return; + + for (int ir = 0; ir < nr; ir++) + { + this->r_radial[ir] = r_radial_in[ir]; + this->rab[ir] = rab_in[ir]; + this->beta_r[ir] = beta_r_in[ir]; + } + + for (int ik = 0; ik < nk; ik++) + { + this->k_radial[ik] = ik * this->dk; + } + this->kcut = (nk-1) * this->dk; + + // (1) extra the uniform mesh + //this->extra_uniform(dr_uniform); + // (2) get the beta_k + this->get_kradial(); + + return; } void Numerical_Nonlocal_Lm::get_kradial() @@ -200,7 +200,7 @@ void Numerical_Nonlocal_Lm::get_kradial() for (int ir = 0; ir < nr; ir++) { - // beta_r is beta*r; + // beta_r is beta*r; integrated_func[ir] = this->beta_r[ir] * this->r_radial[ir] * jl[ir]; } @@ -220,60 +220,60 @@ void Numerical_Nonlocal_Lm::get_kradial() void Numerical_Nonlocal_Lm::plot(const int &my_rank)const { - std::string orbital_type; - switch( this->angular_momentum_l ) - { - case 0: orbital_type = "s"; break; - case 1: orbital_type = "p"; break; - case 2: orbital_type = "d"; break; - case 3: orbital_type = "f"; break; - case 4: orbital_type = "g"; break; - default: ModuleBase::WARNING_QUIT("Numerical_Orbital_Lm::plot","Please check in functoin."); - } + std::string orbital_type; + switch( this->angular_momentum_l ) + { + case 0: orbital_type = "s"; break; + case 1: orbital_type = "p"; break; + case 2: orbital_type = "d"; break; + case 3: orbital_type = "f"; break; + case 4: orbital_type = "g"; break; + default: ModuleBase::WARNING_QUIT("Numerical_Orbital_Lm::plot","Please check in functoin."); + } #ifdef __NORMAL #else - if(my_rank==0) - { - std::stringstream ssr, ssk, ssru; - ssr << ModuleBase::get_global_out_dir() << this->label << "/" - << this->label << "-" << orbital_type << "-proj-r.dat"; - - ssk << ModuleBase::get_global_out_dir() << this->label << "/" - << this->label << "-" << orbital_type << "-proj-k.dat"; - - ssru << ModuleBase::get_global_out_dir() << this->label << "/" - << this->label << "-" << orbital_type << "-proj-ru.dat"; - - std::ofstream ofsr(ssr.str().c_str()); - std::ofstream ofsk(ssk.str().c_str()); - std::ofstream ofsru(ssru.str().c_str()); - - if (!ofsk || !ofsr || !ofsru) - { - ModuleBase::WARNING_QUIT("Numerical_Orbital_Lm::plot", "Can't open files!"); - } - - for (int i = 0; i < this->nr; i++) - { - ofsr << this->r_radial[i] << " " << this->beta_r[i] << std::endl; - } - - for (int i = 0; i < this->nk; i++) - { - ofsk << this->k_radial[i] << " " << this->beta_k[i] << std::endl; - } - - for (int i = 0; i < this->nr_uniform; i++) - { - ofsru << i * this->dr_uniform << " " << this->beta_uniform[i] << std::endl; - } - - ofsr.close(); - ofsk.close(); - ofsru.close(); - } + if(my_rank==0) + { + std::stringstream ssr, ssk, ssru; + ssr << ModuleBase::get_global_out_dir() << this->label << "/" + << this->label << "-" << orbital_type << "-proj-r.dat"; + + ssk << ModuleBase::get_global_out_dir() << this->label << "/" + << this->label << "-" << orbital_type << "-proj-k.dat"; + + ssru << ModuleBase::get_global_out_dir() << this->label << "/" + << this->label << "-" << orbital_type << "-proj-ru.dat"; + + std::ofstream ofsr(ssr.str().c_str()); + std::ofstream ofsk(ssk.str().c_str()); + std::ofstream ofsru(ssru.str().c_str()); + + if (!ofsk || !ofsr || !ofsru) + { + ModuleBase::WARNING_QUIT("Numerical_Orbital_Lm::plot", "Can't open files!"); + } + + for (int i = 0; i < this->nr; i++) + { + ofsr << this->r_radial[i] << " " << this->beta_r[i] << std::endl; + } + + for (int i = 0; i < this->nk; i++) + { + ofsk << this->k_radial[i] << " " << this->beta_k[i] << std::endl; + } + + for (int i = 0; i < this->nr_uniform; i++) + { + ofsru << i * this->dr_uniform << " " << this->beta_uniform[i] << std::endl; + } + + ofsr.close(); + ofsk.close(); + ofsru.close(); + } #endif - return; + return; } diff --git a/source/source_basis/module_ao/orb_nonlocal_lm.h b/source/source_basis/module_ao/orb_nonlocal_lm.h index 4a95d3aaca8..914683022e2 100644 --- a/source/source_basis/module_ao/orb_nonlocal_lm.h +++ b/source/source_basis/module_ao/orb_nonlocal_lm.h @@ -19,83 +19,83 @@ class NumericalNonlocalLmTest; class Numerical_Nonlocal_Lm { - /// the unit test drives freemem(), renew() and get_kradial() directly, and - /// reads label/kcut/index_proj/rab, which have no public accessor - friend class NumericalNonlocalLmTest; + /// the unit test drives freemem(), renew() and get_kradial() directly, and + /// reads label/kcut/index_proj/rab, which have no public accessor + friend class NumericalNonlocalLmTest; - public: + public: - double* beta_uniform = nullptr; - double* dbeta_uniform = nullptr; - int nr_uniform; - double dr_uniform; + double* beta_uniform = nullptr; + double* dbeta_uniform = nullptr; + int nr_uniform; + double dr_uniform; - public: + public: - Numerical_Nonlocal_Lm(); - ~Numerical_Nonlocal_Lm(); + Numerical_Nonlocal_Lm(); + ~Numerical_Nonlocal_Lm(); - const int& getL() const { return this->angular_momentum_l; } - const int& getType() const { return this->index_atom_type; } - const double& getRcut() const { return this->rcut; } + const int& getL() const { return this->angular_momentum_l; } + const int& getType() const { return this->index_atom_type; } + const double& getRcut() const { return this->rcut; } const int& getNr() const { return this->nr; } - const double* getRadial() const { return this->r_radial; } - const double& getRadial(const int &ir) const { return this->r_radial[ir]; } - const double* getBeta_r() const { return this->beta_r; } - const double& getBeta_r(const int &ir) const { return this->beta_r[ir]; } - - const double& getDk()const { return this->dk; } - const double* getKpoint()const { return this->k_radial; } - const double& getKpoint(const int &ik) const { return this->k_radial[ik]; } - const double* getBeta_k() const { return this->beta_k; } - const double& getBeta_k(const int &ik) const { return this->beta_k[ik]; } - - const int& getNk() const { return nk; } - const double& getDruniform() const { return dr_uniform; } - + const double* getRadial() const { return this->r_radial; } + const double& getRadial(const int &ir) const { return this->r_radial[ir]; } + const double* getBeta_r() const { return this->beta_r; } + const double& getBeta_r(const int &ir) const { return this->beta_r[ir]; } + + const double& getDk()const { return this->dk; } + const double* getKpoint()const { return this->k_radial; } + const double& getKpoint(const int &ik) const { return this->k_radial[ik]; } + const double* getBeta_k() const { return this->beta_k; } + const double& getBeta_k(const int &ik) const { return this->beta_k[ik]; } + + const int& getNk() const { return nk; } + const double& getDruniform() const { return dr_uniform; } + // enables deep copy - Numerical_Nonlocal_Lm& operator= (const Numerical_Nonlocal_Lm& nol ); - - void set_NL_proj( - const std::string &label, - const int &index_atom_type_in, - const int &angular_momentum_l_in, - const int &nr_in, - const double *rab_in, - const double *r_radial_in, - const double *beta_r_in, - const int &nk_in, - const double &dk_in, - const double &dr_uniform_in); - - void plot(const int &my_rank)const; - - private: - - void freemem(void); - void renew(void); - //void extra_uniform(const double &dr_uniform); - void get_kradial(void); - - std::string label; - int index_atom_type; - int angular_momentum_l; - int index_proj; - - int nr; - int nk; - - double rcut; - double kcut; - double dk; - - double* r_radial = nullptr; //points of r - double* k_radial = nullptr; - - double* rab = nullptr; - double* beta_r = nullptr; // |beta(r) * r> - double* beta_k = nullptr; + Numerical_Nonlocal_Lm& operator= (const Numerical_Nonlocal_Lm& nol ); + + void set_NL_proj( + const std::string &label, + const int &index_atom_type_in, + const int &angular_momentum_l_in, + const int &nr_in, + const double *rab_in, + const double *r_radial_in, + const double *beta_r_in, + const int &nk_in, + const double &dk_in, + const double &dr_uniform_in); + + void plot(const int &my_rank)const; + + private: + + void freemem(void); + void renew(void); + //void extra_uniform(const double &dr_uniform); + void get_kradial(void); + + std::string label; + int index_atom_type; + int angular_momentum_l; + int index_proj; + + int nr; + int nk; + + double rcut; + double kcut; + double dk; + + double* r_radial = nullptr; //points of r + double* k_radial = nullptr; + + double* rab = nullptr; + double* beta_r = nullptr; // |beta(r) * r> + double* beta_k = nullptr; }; #endif diff --git a/source/source_basis/module_ao/orb_read.h b/source/source_basis/module_ao/orb_read.h index bc46359f02d..f12772fad2b 100644 --- a/source/source_basis/module_ao/orb_read.h +++ b/source/source_basis/module_ao/orb_read.h @@ -16,10 +16,10 @@ class LCAO_Orbitals { - public: + public: - LCAO_Orbitals(); - ~LCAO_Orbitals(); + LCAO_Orbitals(); + ~LCAO_Orbitals(); void init( std::ofstream& ofs_in, @@ -39,98 +39,98 @@ class LCAO_Orbitals const int& my_rank ); - void Read_Orbitals( - std::ofstream &ofs_in, // mohan add 2021-05-07 - const int &ntype_in, - const int &lmax_in, - const bool &deepks_setorb, // mohan add 2021-04-25 - const int &out_mat_r, // mohan add 2021-04-26 - const bool &out_element_info, - const bool &force_flag, // mohan add 2021-05-07 - const int &my_rank); // mohan add 2021-04-26 + void Read_Orbitals( + std::ofstream &ofs_in, // mohan add 2021-05-07 + const int &ntype_in, + const int &lmax_in, + const bool &deepks_setorb, // mohan add 2021-04-25 + const int &out_mat_r, // mohan add 2021-04-26 + const bool &out_element_info, + const bool &force_flag, // mohan add 2021-05-07 + const int &my_rank); // mohan add 2021-04-26 - void Read_PAO( - std::ofstream &ofs_in, - const int& it, - const bool &out_element_info, - const bool &force_flag, // mohan add 2021-05-07 - const int& my_rank); // mohan add 2021-04-26 + void Read_PAO( + std::ofstream &ofs_in, + const int& it, + const bool &out_element_info, + const bool &force_flag, // mohan add 2021-05-07 + const int& my_rank); // mohan add 2021-04-26 - void Read_Descriptor( - std::ofstream &ofs_in, - const bool &out_element_info, - const bool &force_flag, // mohan add 2021-05-07 - const int &my_rank); //caoyu add 2020-3-16 + void Read_Descriptor( + std::ofstream &ofs_in, + const bool &out_element_info, + const bool &force_flag, // mohan add 2021-05-07 + const int &my_rank); //caoyu add 2020-3-16 #ifdef __MPI - void bcast_files(const int &ntype_in, const int &my_rank); + void bcast_files(const int &ntype_in, const int &my_rank); #endif - const double& get_ecutwfc() const {return ecutwfc;} - const int& get_kmesh() const{return kmesh;} - const double& get_dk() const {return dk;} - const double& get_dR() const {return dR;} - const double& get_Rmax() const {return Rmax;} - const int& get_lmax() const {return lmax;} - const int& get_lmax_d() const { return lmax_d; } /// cutoffs() const; - /// numerical atomic orbitals - Numerical_Orbital* Phi = nullptr; - - - //caoyu add 2021-3-10 - /// descriptor bases, saved as one-type atom orbital - Numerical_Orbital* Alpha = nullptr; - - // initialized in input.cpp - double ecutwfc; - double dk; - double dR; - double Rmax; - - double dr_uniform; - - // initalized in UnitCell - // assume ntype < 20. - bool read_in_flag; - std::vector orbital_file; - std::vector nonlocal_file; - std::string descriptor_file; //caoyu add 2020-3-16 + /// numerical atomic orbitals + Numerical_Orbital* Phi = nullptr; + + + //caoyu add 2021-3-10 + /// descriptor bases, saved as one-type atom orbital + Numerical_Orbital* Alpha = nullptr; + + // initialized in input.cpp + double ecutwfc; + double dk; + double dR; + double Rmax; + + double dr_uniform; + + // initalized in UnitCell + // assume ntype < 20. + bool read_in_flag; + std::vector orbital_file; + std::vector nonlocal_file; + std::string descriptor_file; //caoyu add 2020-3-16 private: - int ntype; // number of elements - int kmesh; // number of points on kmesh + int ntype; // number of elements + int kmesh; // number of points on kmesh - int lmax; - int nchimax; + int lmax; + int nchimax; - int lmax_d; //max l of descriptor orbitals - int nchimax_d; //max number of descriptor orbitals per l + int lmax_d; //max l of descriptor orbitals + int nchimax_d; //max number of descriptor orbitals per l - double rcutmax_Phi; //caoyu add 2021-05-24 + double rcutmax_Phi; //caoyu add 2021-05-24 - void read_orb_file( - std::ofstream &ofs_in, - std::ifstream &ifs, - const int &it, - int &lmax, - int &nchimax, - Numerical_Orbital* ao, - const bool &out_element_info, - const bool &force_flag, // mohan add 2021-05-07 - const int &my_rank); //caoyu add 2021-04-26 + void read_orb_file( + std::ofstream &ofs_in, + std::ifstream &ifs, + const int &it, + int &lmax, + int &nchimax, + Numerical_Orbital* ao, + const bool &out_element_info, + const bool &force_flag, // mohan add 2021-05-07 + const int &my_rank); //caoyu add 2021-04-26 friend class TwoCenterBundle; // for the sake of TwoCenterBundle::to_LCAO_Orbitals }; diff --git a/source/source_basis/module_ao/parallel_orbitals.h b/source/source_basis/module_ao/parallel_orbitals.h index aecde685b5a..048c2cd9df5 100644 --- a/source/source_basis/module_ao/parallel_orbitals.h +++ b/source/source_basis/module_ao/parallel_orbitals.h @@ -31,8 +31,8 @@ class Parallel_Orbitals : public Parallel_2D /// on all adjacent atoms-pairs(2D division) ///------------------------------------- int nnr=1; - std::vector nlocdim; - std::vector nlocstart; + std::vector nlocdim; + std::vector nlocstart; #ifdef __MPI int desc_wfc[9]; //for wfc, nlocal*nbands diff --git a/source/source_basis/module_nao/two_center_integrator.cpp b/source/source_basis/module_nao/two_center_integrator.cpp index 436f3885ed8..4c4e61abedd 100644 --- a/source/source_basis/module_nao/two_center_integrator.cpp +++ b/source/source_basis/module_nao/two_center_integrator.cpp @@ -29,7 +29,7 @@ void TwoCenterIntegrator::calculate(const int itype1, const int l2, const int izeta2, const int m2, - const ModuleBase::Vector3& vR, // R = R2 - R1 + const ModuleBase::Vector3& vR, // R = R2 - R1 double* out, double* grad_out, double* hess_out) const @@ -67,8 +67,8 @@ void TwoCenterIntegrator::calculate(const int itype1, // generate all necessary real (solid) spherical harmonics const int lmax = l1 + l2; - std::vector Rl_Y((lmax+1) * (lmax+1)); - std::vector grad_Rl_Y((lmax+1) * (lmax+1) * 3); + std::vector Rl_Y((lmax+1) * (lmax+1)); + std::vector grad_Rl_Y((lmax+1) * (lmax+1) * 3); std::vector> hess_Rl_Y; // R^l * Y is necessary anyway @@ -88,7 +88,7 @@ void TwoCenterIntegrator::calculate(const int itype1, // look up S/R^l, (d/dR)(S/R^l), and (d²/dR²)(S/R^l) from the radial table table_.lookup(itype1, l1, izeta1, itype2, l2, izeta2, l, R, S_by_Rl, d_S_by_Rl, d2_S_by_Rl); - for (int m = -l; m <= l; ++m) + for (int m = -l; m <= l; ++m) { double G = RealGauntTable::instance()(l1, l2, l, m1, m2, m); int lm_idx = ylm_index(l, m); @@ -150,7 +150,7 @@ void TwoCenterIntegrator::snap(const int itype1, const int izeta1, const int m1, const int itype2, - const ModuleBase::Vector3& vR, + const ModuleBase::Vector3& vR, const bool deriv, std::vector>& out) const { @@ -172,11 +172,11 @@ void TwoCenterIntegrator::snap(const int itype1, return; } - for(size_t i = 0; i < out.size(); ++i) - { - out[i].resize(num_ket); + for(size_t i = 0; i < out.size(); ++i) + { + out[i].resize(num_ket); std::fill(out[i].begin(), out[i].end(), 0.0); - } + } int index = 0; double tmp[3] = {0.0, 0.0, 0.0}; diff --git a/source/source_basis/module_nao/two_center_integrator.h b/source/source_basis/module_nao/two_center_integrator.h index 285ffe8728c..9955ef22551 100644 --- a/source/source_basis/module_nao/two_center_integrator.h +++ b/source/source_basis/module_nao/two_center_integrator.h @@ -97,7 +97,7 @@ class TwoCenterIntegrator const int l2, const int izeta2, const int m2, - const ModuleBase::Vector3& vR, // vR = R2 - R1 + const ModuleBase::Vector3& vR, // vR = R2 - R1 double* out = nullptr, double* grad_out = nullptr, double* hess_out = nullptr @@ -114,7 +114,7 @@ class TwoCenterIntegrator const int izeta1, const int m1, const int itype2, - const ModuleBase::Vector3& vR, // vR = R2 - R1 + const ModuleBase::Vector3& vR, // vR = R2 - R1 const bool deriv, std::vector>& out ) const; diff --git a/source/source_basis/module_pw/CMakeLists.txt b/source/source_basis/module_pw/CMakeLists.txt index ebb9f8673a1..d2abf09b418 100644 --- a/source/source_basis/module_pw/CMakeLists.txt +++ b/source/source_basis/module_pw/CMakeLists.txt @@ -44,10 +44,8 @@ add_library( ) if (USE_DSP) -target_link_libraries(planewave PRIVATE -${MTBLAS_FFT_DIR}/libmtblas/lib/libmtfft.a) -target_compile_definitions( planewave PUBLIC -FFT_DAT_DIR="${MTBLAS_FFT_DIR}/datfile/mt_fft_blas.dat") +target_compile_definitions( planewave PUBLIC +FFT_DAT_DIR="${MTBLAS_FFT_DIR}/datfile/mt_fft_blas.dat") endif() if(ENABLE_COVERAGE) add_coverage(planewave) diff --git a/source/source_basis/module_pw/pw_basis.cpp b/source/source_basis/module_pw/pw_basis.cpp index 549fec8e5a4..a079a3bdd4c 100644 --- a/source/source_basis/module_pw/pw_basis.cpp +++ b/source/source_basis/module_pw/pw_basis.cpp @@ -85,42 +85,42 @@ void PW_Basis::getstartgr() } //--------------------------------------------- - // sum : starting plane of FFT box. - //--------------------------------------------- + // sum : starting plane of FFT box. + //--------------------------------------------- delete[] this->numg; this->numg = new int[poolnproc]; - delete[] this->startg; this->startg = new int[poolnproc]; - delete[] this->startr; this->startr = new int[poolnproc]; - delete[] this->numr; this->numr = new int[poolnproc]; + delete[] this->startg; this->startg = new int[poolnproc]; + delete[] this->startr; this->startr = new int[poolnproc]; + delete[] this->numr; this->numr = new int[poolnproc]; - // Each processor has a set of full sticks, - // 'rank_use' processor send a piece(npps[ip]) of these sticks(nst_per[rank_use]) - // to all the other processors in this pool - for (int ip = 0;ip < poolnproc; ++ip) + // Each processor has a set of full sticks, + // 'rank_use' processor send a piece(npps[ip]) of these sticks(nst_per[rank_use]) + // to all the other processors in this pool + for (int ip = 0;ip < poolnproc; ++ip) { this->numg[ip] = this->nst_per[poolrank] * this->numz[ip]; } - // Each processor in a pool send a piece of each stick(nst_per[ip]) to - // other processors in this pool - // rank_use processor receive datas in npps[rank_p] planes. - for (int ip = 0;ip < poolnproc; ++ip) + // Each processor in a pool send a piece of each stick(nst_per[ip]) to + // other processors in this pool + // rank_use processor receive datas in npps[rank_p] planes. + for (int ip = 0;ip < poolnproc; ++ip) { this->numr[ip] = this->nst_per[ip] * this->numz[poolrank]; } - // startg record the starting 'numg' position in each processor. - this->startg[0] = 0; - for (int ip = 1;ip < poolnproc; ++ip) + // startg record the starting 'numg' position in each processor. + this->startg[0] = 0; + for (int ip = 1;ip < poolnproc; ++ip) { this->startg[ip] = this->startg[ip-1] + this->numg[ip-1]; } - // startr record the starting 'numr' position - this->startr[0] = 0; - for (int ip = 1;ip < poolnproc; ++ip) + // startr record the starting 'numr' position + this->startr[0] = 0; + for (int ip = 1;ip < poolnproc; ++ip) { this->startr[ip] = this->startr[ip-1] + this->numr[ip-1]; } diff --git a/source/source_basis/module_pw/pw_basis.h b/source/source_basis/module_pw/pw_basis.h index 6a1f93c80d3..d476e59a542 100644 --- a/source/source_basis/module_pw/pw_basis.h +++ b/source/source_basis/module_pw/pw_basis.h @@ -3,6 +3,8 @@ #include "source_base/macros.h" #include "source_base/module_device/memory_op.h" +#include "source_base/constants.h" +#include "source_base/global_function.h" #include "source_base/matrix.h" #include "source_base/matrix3.h" #include "source_base/vector3.h" @@ -124,7 +126,7 @@ class PW_Basis //real space int nrxx=0; //num. of real space grids int *startz=nullptr; //startz[ip]: starting z plane in the ip-th proc. in current POOL_WORLD - int *numz=nullptr; //numz[ip]: num. of z planes in the ip-th proc. in current POOL_WORLD + int *numz=nullptr; //numz[ip]: num. of z planes in the ip-th proc. in current POOL_WORLD int *numg=nullptr; //numg[ip] : nst_per[poolrank] * numz[ip] int *numr=nullptr; //numr[ip] : numz[poolrank] * nst_per[ip] int *startg=nullptr; // startg[ip] = numg[ip-1] + startg[ip-1] @@ -136,7 +138,7 @@ class PW_Basis ModuleBase::Vector3 *gcar=nullptr; //G vectors in cartesian corrdinate double *gg=nullptr; // modulus (G^2) of G vectors [npw] //gg[ng]=ig[ng]*GGT*ig[ng]/(lat0*lat0)=g[ng]*g[ng] (/lat0*lat0) - // gg_global dimension: [cutgg_num_now] (save memory skill is used) + // gg_global dimension: [cutgg_num_now] (save memory skill is used) int ig_gge0=-1; //ig when gg == 0 //distribute plane waves and grids and set up fft @@ -239,8 +241,8 @@ class PW_Basis // FFT //=============================================== public: - // FFT dimensions for wave functions. - int fftnx=0, fftny=0, fftnz=0, fftnxyz=0, fftnxy=0; + // FFT dimensions for wave functions. + int fftnx=0, fftny=0, fftnz=0, fftnxyz=0, fftnxy=0; int nx=0, ny=0, nz=0, nxyz=0, nxy=0; // Gamma_only: fftny = int(ny/2)-1 , others: fftny = ny int liy=0, riy=0;// liy: the left edge of the pw ball; riy: the right edge of the pw ball in the y direction int lix=0, rix=0;// lix: the left edge of the pw ball; rix: the right edge of the pw ball in the x direction @@ -449,5 +451,3 @@ class PW_Basis }; } #endif // PWBASIS_H -#include "pw_basis_sup.h" -#include "pw_basis_big.h" //temporary it will be removed diff --git a/source/source_basis/module_pw/pw_basis_big.h b/source/source_basis/module_pw/pw_basis_big.h index 987af787b3f..701b7cb69d5 100644 --- a/source/source_basis/module_pw/pw_basis_big.h +++ b/source/source_basis/module_pw/pw_basis_big.h @@ -1,12 +1,6 @@ #ifndef PW_BASIS_BIG_H #define PW_BASIS_BIG_H -#include "source_base/constants.h" -#include "source_base/global_function.h" - -#ifdef __MPI -#include "mpi.h" -#endif - +#include "pw_basis_sup.h" // temporary class, because previous ABACUS consider big grid for fft grids // which are used for grid integration in LCAO. // In fact, it is unnecessary. It will be moved after grid integration is refactored. @@ -113,8 +107,8 @@ class PW_Basis_Big : public PW_Basis_Sup this->latvec = latvec_in; this->omega = std::abs(latvec.Det()) * lat0 * lat0 * lat0; this->GT = latvec.Inverse(); - this->G = GT.Transpose(); - this->GGT = G * GT; + this->G = GT.Transpose(); + this->GGT = G * GT; //------------------------------------------------------------ //-------------------------init grids------------------------- @@ -174,45 +168,45 @@ class PW_Basis_Big : public PW_Basis_Sup // Find the minimal FFT box size the factors into the primes (2,3,5,7). for (int i = 0; i < 3; i++) { - int b = 0; + int b = 0; int n2 = 0; int n3 = 0; int n5 = 0; //int n7 = 0; bool done_factoring = false; - + // increase ibox[i] by 1 until it is totally factorizable by (2,3,5,7) do { - b = ibox[i]; + b = ibox[i]; - //n2 = n3 = n5 = n7 = 0; - n2 = n3 = n5 = 0; - done_factoring = false; + //n2 = n3 = n5 = n7 = 0; + n2 = n3 = n5 = 0; + done_factoring = false; if ((this->full_pw && this->full_pw_dim == 2) && b % 2 != 0) done_factoring = true; // full_pw_dim = 2 means FFT dimensions should be even. - while (!done_factoring) - { - if (b % 2 == 0 && (!this->full_pw || this->full_pw_dim != 1)) // full_pw_dim = 1 means FFT dimension should be odd. - { - n2++; - b /= 2; - continue; - } - if (b % 3 == 0) - { - n3++; - b /= 3; - continue; - } - if (b % 5 == 0) - { - n5++; - b /= 5; - continue; - } - //if (b%7==0) { n7++; b /= 7; continue; } - done_factoring = true; - } + while (!done_factoring) + { + if (b % 2 == 0 && (!this->full_pw || this->full_pw_dim != 1)) // full_pw_dim = 1 means FFT dimension should be odd. + { + n2++; + b /= 2; + continue; + } + if (b % 3 == 0) + { + n3++; + b /= 3; + continue; + } + if (b % 5 == 0) + { + n5++; + b /= 5; + continue; + } + //if (b%7==0) { n7++; b /= 7; continue; } + done_factoring = true; + } ibox[i] += 1; } while (b != 1); @@ -279,8 +273,8 @@ class PW_Basis_Big : public PW_Basis_Sup this->latvec = latvec_in; this->omega = std::abs(latvec.Det()) * lat0 * lat0 * lat0; this->GT = latvec.Inverse(); - this->G = GT.Transpose(); - this->GGT = G * GT; + this->G = GT.Transpose(); + this->GGT = G * GT; this->nx = nx_in; this->ny = ny_in; this->nz = nz_in; diff --git a/source/source_basis/module_pw/pw_basis_k.h b/source/source_basis/module_pw/pw_basis_k.h index f633a30769d..65e332b8a25 100644 --- a/source/source_basis/module_pw/pw_basis_k.h +++ b/source/source_basis/module_pw/pw_basis_k.h @@ -279,5 +279,3 @@ class PW_Basis_K : public PW_Basis } #endif //PlaneWave_K class -#include "./pw_basis_k_big.h" //temporary it will be removed - diff --git a/source/source_basis/module_pw/pw_basis_k_big.h b/source/source_basis/module_pw/pw_basis_k_big.h index 128c1d937d2..7e3a9aea0ec 100644 --- a/source/source_basis/module_pw/pw_basis_k_big.h +++ b/source/source_basis/module_pw/pw_basis_k_big.h @@ -1,8 +1,6 @@ #ifndef PW_BASIS_K_BIG_H #define PW_BASIS_K_BIG_H -#include "source_base/constants.h" -#include "source_base/global_function.h" - +#include "pw_basis_k.h" // temporary class, because previous ABACUS consider big grid for fft grids // which are used for grid integration in LCAO. // In fact, it is unnecessary. It will be moved after grid integration is refactored. @@ -14,8 +12,8 @@ class PW_Basis_K_Big: public PW_Basis_K public: // combine [bx,by,bz] FFT grids into a big one - // typical values are bx=2, by=2, bz=2 - // nbx=nx/bx, nby=ny/by, nbz=nz/bz, + // typical values are bx=2, by=2, bz=2 + // nbx=nx/bx, nby=ny/by, nbz=nz/bz, // Note: this class can only use initgrids(lat0_in, latvec_in, PW_Basis_Big::nx, PW_Basis_Big::ny, PW_Basis_Big::nz)!!! PW_Basis_K_Big(){ bx = 1; diff --git a/source/source_basis/module_pw/pw_basis_sup.cpp b/source/source_basis/module_pw/pw_basis_sup.cpp index 2a0d464ce0e..0184bfec92b 100644 --- a/source/source_basis/module_pw/pw_basis_sup.cpp +++ b/source/source_basis/module_pw/pw_basis_sup.cpp @@ -1,5 +1,5 @@ #include "source_base/timer.h" -#include "pw_basis.h" +#include "pw_basis_sup.h" namespace ModulePW { diff --git a/source/source_basis/module_pw/pw_basis_sup.h b/source/source_basis/module_pw/pw_basis_sup.h index b77dfa6bec0..4184759ead8 100644 --- a/source/source_basis/module_pw/pw_basis_sup.h +++ b/source/source_basis/module_pw/pw_basis_sup.h @@ -1,6 +1,6 @@ #ifndef PWBASIS_SUP_H #define PWBASIS_SUP_H - +#include "pw_basis.h" namespace ModulePW { @@ -58,4 +58,4 @@ class PW_Basis_Sup : public PW_Basis }; // class PW_Basis_Sup } // namespace ModulePW -#endif // PWBASIS_SUP_H \ No newline at end of file +#endif // PWBASIS_SUP_H diff --git a/source/source_basis/module_pw/pw_init.cpp b/source/source_basis/module_pw/pw_init.cpp index 08c676d39f3..5240d35b12d 100644 --- a/source/source_basis/module_pw/pw_init.cpp +++ b/source/source_basis/module_pw/pw_init.cpp @@ -33,8 +33,8 @@ void PW_Basis:: initgrids( this->latvec = latvec_in; this->omega = std::abs(latvec.Det()) * lat0 * lat0 * lat0; this->GT = latvec.Inverse(); - this->G = GT.Transpose(); - this->GGT = G * GT; + this->G = GT.Transpose(); + this->GGT = G * GT; //------------------------------------------------------------ //-------------------------init grids------------------------- @@ -92,45 +92,45 @@ void PW_Basis:: initgrids( // Find the minimal FFT box size the factors into the primes (2,3,5,7). for (int i = 0; i < 3; i++) { - int b = 0; + int b = 0; int n2 = 0; int n3 = 0; int n5 = 0; //int n7 = 0; bool done_factoring = false; - + // increase ibox[i] by 1 until it is totally factorizable by (2,3,5,7) do { - b = ibox[i]; + b = ibox[i]; - //n2 = n3 = n5 = n7 = 0; - n2 = n3 = n5 = 0; - done_factoring = false; + //n2 = n3 = n5 = n7 = 0; + n2 = n3 = n5 = 0; + done_factoring = false; if ((this->full_pw && this->full_pw_dim == 2) && b % 2 != 0) done_factoring = true; // full_pw_dim = 2 means FFT dimensions should be even. - while (!done_factoring) - { - if (b % 2 == 0 && (!this->full_pw || this->full_pw_dim != 1)) // full_pw_dim = 1 means FFT dimension should be odd. - { - n2++; - b /= 2; - continue; - } - if (b % 3 == 0) - { - n3++; - b /= 3; - continue; - } - if (b % 5 == 0) - { - n5++; - b /= 5; - continue; - } - //if (b%7==0) { n7++; b /= 7; continue; } - done_factoring = true; - } + while (!done_factoring) + { + if (b % 2 == 0 && (!this->full_pw || this->full_pw_dim != 1)) // full_pw_dim = 1 means FFT dimension should be odd. + { + n2++; + b /= 2; + continue; + } + if (b % 3 == 0) + { + n3++; + b /= 3; + continue; + } + if (b % 5 == 0) + { + n5++; + b /= 5; + continue; + } + //if (b%7==0) { n7++; b /= 7; continue; } + done_factoring = true; + } ibox[i] += 1; } while (b != 1); @@ -164,8 +164,8 @@ void PW_Basis:: initgrids( this->latvec = latvec_in; this->omega = std::abs(latvec.Det()) * lat0 * lat0 * lat0; this->GT = latvec.Inverse(); - this->G = GT.Transpose(); - this->GGT = G * GT; + this->G = GT.Transpose(); + this->GGT = G * GT; this->nx = nx_in; this->ny = ny_in; this->nz = nz_in; diff --git a/source/source_basis/module_pw/pw_momentum.h b/source/source_basis/module_pw/pw_momentum.h new file mode 100644 index 00000000000..e1bb6d6826c --- /dev/null +++ b/source/source_basis/module_pw/pw_momentum.h @@ -0,0 +1,26 @@ +#ifndef MODULE_PW_MOMENTUM_H +#define MODULE_PW_MOMENTUM_H + +#include "source_basis/module_pw/pw_basis_k.h" + +namespace ModulePW +{ +/** @brief Return the kinetic correction 2*p*A + A^2 in Ry for a Hartree vector potential. + * @note Evaluate geometry in double precision; callers convert the result to their native precision. + */ +inline double kinetic_shift(const PW_Basis_K& basis, const int ik, const int ig, + const ModuleBase::Vector3& A_ha) +{ + const ModuleBase::Vector3 p = basis.getgpluskcar(ik, ig) * basis.tpiba; + return 2.0 * (p.x * A_ha.x + p.y * A_ha.y + p.z * A_ha.z) + A_ha.norm2(); +} + +/** @brief Return the shifted kinetic energy in Ry, retaining the cached unshifted squared momentum. */ +inline double shifted_kinetic(const PW_Basis_K& basis, const int ik, const int ig, + const ModuleBase::Vector3& A_ha) +{ + return basis.getgk2(ik, ig) * basis.tpiba2 + kinetic_shift(basis, ik, ig, A_ha); +} +} // namespace ModulePW + +#endif diff --git a/source/source_basis/module_pw/test/test_big.cpp b/source/source_basis/module_pw/test/test_big.cpp index eca02c91a22..15ba51f63bc 100644 --- a/source/source_basis/module_pw/test/test_big.cpp +++ b/source/source_basis/module_pw/test/test_big.cpp @@ -2,6 +2,8 @@ // TEST for PW_Basis_big & PW_Basis_big_k //--------------------------------------------- #include "../pw_basis_k.h" +#include "../pw_basis_big.h" +#include "../pw_basis_k_big.h" #ifdef __MPI #include "test_tool.h" #include "source_base/parallel_global.h" diff --git a/source/source_basis/module_pw/test/test_sup.cpp b/source/source_basis/module_pw/test/test_sup.cpp index 188aabc4eda..76446a237ed 100644 --- a/source/source_basis/module_pw/test/test_sup.cpp +++ b/source/source_basis/module_pw/test/test_sup.cpp @@ -1,4 +1,4 @@ -#include "../pw_basis.h" +#include "../pw_basis_sup.h" #ifdef __MPI #include "source_base/parallel_global.h" #include "mpi.h" diff --git a/source/source_cell/CMakeLists.txt b/source/source_cell/CMakeLists.txt index 46eacaec8f0..b3e574e7144 100644 --- a/source/source_cell/CMakeLists.txt +++ b/source/source_cell/CMakeLists.txt @@ -2,6 +2,12 @@ add_subdirectory(module_symmetry) add_subdirectory(module_neighbor) add_subdirectory(module_neighlist) +add_library( + domain_decomposition + OBJECT + domain_decomposition.cpp +) + add_library( cell OBJECT @@ -27,6 +33,7 @@ add_library( parallel_kpoints.cpp cell_index.cpp cell_tools.cpp + record_adj.cpp check_atomic_stru.cpp update_cell.cpp magnetism.cpp diff --git a/source/source_cell/cal_ux.cpp b/source/source_cell/cal_ux.cpp index 06d7b0558a7..38638c31afc 100644 --- a/source/source_cell/cal_ux.cpp +++ b/source/source_cell/cal_ux.cpp @@ -2,6 +2,19 @@ namespace unitcell { +// Compute the global spin quantization axis ux_ for non-collinear (nspin=4) +// calculations. ux_ is a single axis shared by the whole system (NOT one axis +// per atom); it is consumed by XC_Functional::noncolin_rho to split the spin +// density matrix into spin-up/down scalars for the LSDA/GGA kernel. +// +// The axis is derived from the *initial* moments m_loc_ read from STRU +// (direction of the first atom with |m| > threshold), not from the converged +// density. Consequently the STRU initial moments silently select the XC +// projection axis: with all-zero initial moments lsign_ stays false and the +// XC projection becomes direction-blind (neg = +1 everywhere in +// noncolin_rho), which can follow a different SCF path than a run started +// with an explicit non-zero moment even when the starting density (e.g. from +// init_chg = wfc/file) is identical. void cal_ux(UnitCell& ucell, const int nspin) { if (nspin != 4) @@ -22,10 +35,10 @@ void cal_ux(UnitCell& ucell, const int nspin) { ucell.magnet.lsign_ = false; ModuleBase::GlobalFunc::ZEROS(ucell.magnet.ux_, 3); - for (int it = 0; it < ucell.ntype; it++) - { - for (int ia = 0; ia < ucell.atoms[it].na; ia++) - { + for (int it = 0; it < ucell.ntype; it++) + { + for (int ia = 0; ia < ucell.atoms[it].na; ia++) + { // m_loc_: local magnetization vector for each atom amag = pow(ucell.atoms[it].m_loc_[ia].x, 2) + pow(ucell.atoms[it].m_loc_[ia].y, 2) @@ -33,61 +46,61 @@ void cal_ux(UnitCell& ucell, const int nspin) { // find the first atom (it,ia) whose magnetism is not zero // compute ux - if (amag > absolute_mag_thr) - { - ucell.magnet.ux_[0] = ucell.atoms[it].m_loc_[ia].x; - ucell.magnet.ux_[1] = ucell.atoms[it].m_loc_[ia].y; - ucell.magnet.ux_[2] = ucell.atoms[it].m_loc_[ia].z; + if (amag > absolute_mag_thr) + { + ucell.magnet.ux_[0] = ucell.atoms[it].m_loc_[ia].x; + ucell.magnet.ux_[1] = ucell.atoms[it].m_loc_[ia].y; + ucell.magnet.ux_[2] = ucell.atoms[it].m_loc_[ia].z; - starting_it = it; - starting_ia = ia; + starting_it = it; + starting_ia = ia; - ucell.magnet.lsign_ = true; - break; - } - } + ucell.magnet.lsign_ = true; + break; + } + } // if any atom has magnetism, then break the for iteration - if (ucell.magnet.lsign_) - { - break; - } - } + if (ucell.magnet.lsign_) + { + break; + } + } // whether the initial magnetizations is parallel - for (int it = starting_it; it < ucell.ntype; it++) - { - for (int ia = 0; ia < ucell.atoms[it].na; ia++) - { - if (it > starting_it || ia > starting_ia) - { - ucell.magnet.lsign_ - = ucell.magnet.lsign_ - && judge_parallel(ucell.magnet.ux_, ucell.atoms[it].m_loc_[ia]); - } - } - } + for (int it = starting_it; it < ucell.ntype; it++) + { + for (int ia = 0; ia < ucell.atoms[it].na; ia++) + { + if (it > starting_it || ia > starting_ia) + { + ucell.magnet.lsign_ + = ucell.magnet.lsign_ + && judge_parallel(ucell.magnet.ux_, ucell.atoms[it].m_loc_[ia]); + } + } + } // if all of the atoms have the same parallel magnetism direction, // then set the direction to a unit vector - if (ucell.magnet.lsign_) - { - uxmod = pow(ucell.magnet.ux_[0], 2) - + pow(ucell.magnet.ux_[1], 2) - + pow(ucell.magnet.ux_[2], 2); + if (ucell.magnet.lsign_) + { + uxmod = pow(ucell.magnet.ux_[0], 2) + + pow(ucell.magnet.ux_[1], 2) + + pow(ucell.magnet.ux_[2], 2); - if (uxmod < absolute_mag_thr) - { - ModuleBase::WARNING_QUIT("unitcell::cal_ux", "wrong uxmod"); - } + if (uxmod < absolute_mag_thr) + { + ModuleBase::WARNING_QUIT("unitcell::cal_ux", "wrong uxmod"); + } // reset the magnetism for each direction - for (int i = 0; i < 3; i++) - { - ucell.magnet.ux_[i] *= 1 / sqrt(uxmod); - } - } - return; + for (int i = 0; i < 3; i++) + { + ucell.magnet.ux_[i] *= 1 / sqrt(uxmod); + } + } + return; } bool judge_parallel(double a[3], ModuleBase::Vector3 b) { diff --git a/source/source_cell/cal_wfc.cpp b/source/source_cell/cal_wfc.cpp index bd9b2022ff4..b6307e5fc9e 100644 --- a/source/source_cell/cal_wfc.cpp +++ b/source/source_cell/cal_wfc.cpp @@ -133,38 +133,38 @@ namespace unitcell void cal_natomwfc(std::ofstream& log,int& natomwfc,const int ntype,const Atom* atoms,const int nspin) { natomwfc = 0; - for (int it = 0; it < ntype; it++) - { - //============================ - // Use pseudo-atomic orbitals - //============================ - int tmp = 0; - for (int l = 0; l < atoms[it].ncpp.nchi; l++) - { - if (atoms[it].ncpp.oc[l] >= 0) - { - if (nspin == 4) - { - if (atoms[it].ncpp.has_so) - { - tmp += 2 * atoms[it].ncpp.lchi[l]; - if (fabs(atoms[it].ncpp.jchi[l] - atoms[it].ncpp.lchi[l] - 0.5)< 1e-6) - { - tmp += 2; - } - } else - { - tmp += 2 * (2 * atoms[it].ncpp.lchi[l] + 1); - } - } else - { - tmp += 2 * atoms[it].ncpp.lchi[l] + 1; - } - } - } - natomwfc += tmp * atoms[it].na; - } - ModuleBase::GlobalFunc::OUT(log, "Number of pseudo atomic orbitals", natomwfc); - return; + for (int it = 0; it < ntype; it++) + { + //============================ + // Use pseudo-atomic orbitals + //============================ + int tmp = 0; + for (int l = 0; l < atoms[it].ncpp.nchi; l++) + { + if (atoms[it].ncpp.oc[l] >= 0) + { + if (nspin == 4) + { + if (atoms[it].ncpp.has_so) + { + tmp += 2 * atoms[it].ncpp.lchi[l]; + if (fabs(atoms[it].ncpp.jchi[l] - atoms[it].ncpp.lchi[l] - 0.5)< 1e-6) + { + tmp += 2; + } + } else + { + tmp += 2 * (2 * atoms[it].ncpp.lchi[l] + 1); + } + } else + { + tmp += 2 * atoms[it].ncpp.lchi[l] + 1; + } + } + } + natomwfc += tmp * atoms[it].na; + } + ModuleBase::GlobalFunc::OUT(log, "Number of pseudo atomic orbitals", natomwfc); + return; } } diff --git a/source/source_cell/module_neighlist/domain_decomposition.cpp b/source/source_cell/domain_decomposition.cpp similarity index 91% rename from source/source_cell/module_neighlist/domain_decomposition.cpp rename to source/source_cell/domain_decomposition.cpp index 6a4accf2f68..b48a75dd9a6 100644 --- a/source/source_cell/module_neighlist/domain_decomposition.cpp +++ b/source/source_cell/domain_decomposition.cpp @@ -1,4 +1,4 @@ -#include "source_cell/module_neighlist/domain_decomposition.h" +#include "source_cell/domain_decomposition.h" #include "source_base/parallel_cell.h" #include "source_cell/unitcell.h" #include "source_cell/mdcell.h" @@ -264,6 +264,16 @@ void DomainDecomposition::exchange_ghost_atoms(MDCell& cell) { GhostExchangeSlot& slot = slots[islot]; const std::vector& send_atoms = send_buffers[islot]; + slot.previous_fracs.clear(); + slot.dynamic_image_shifts.clear(); + slot.previous_fracs.reserve(slot.send_atom_indices.size()); + slot.dynamic_image_shifts.reserve(slot.send_atom_indices.size()); + for (const int index : slot.send_atom_indices) + { + const LocalAtom& atom = owned_atoms[static_cast(index)]; + slot.previous_fracs.push_back({{atom.frac.x, atom.frac.y, atom.frac.z}}); + slot.dynamic_image_shifts.push_back(slot.image_shift); + } slot.ghost_begin = ghost_atoms.size(); if (send_atoms.size() > static_cast(std::numeric_limits::max())) @@ -318,16 +328,25 @@ void DomainDecomposition::update_ghost_atom_positions(MDCell& cell) std::vector& ghost_atoms = cell.ghost_atoms_; for (std::size_t islot = 0; islot < ghost_slots_.size(); ++islot) { - const GhostExchangeSlot& slot = ghost_slots_[islot]; - std::vector send_frac(3 * slot.send_atom_indices.size(), 0.0); + GhostExchangeSlot& slot = ghost_slots_[islot]; + // Keep wrapped fractional coordinates for metadata and send the + // dynamically selected image coordinates in the same message. + std::vector send_frac(6 * slot.send_atom_indices.size(), 0.0); for (std::size_t i = 0; i < slot.send_atom_indices.size(); ++i) { const LocalAtom& atom = owned_atoms[static_cast(slot.send_atom_indices[i])]; + const std::array image_shift = image_shift_for_update( + atom, slot.previous_fracs[i], slot.dynamic_image_shifts[i], slot); + slot.previous_fracs[i] = {{atom.frac.x, atom.frac.y, atom.frac.z}}; + slot.dynamic_image_shifts[i] = image_shift; send_frac[3 * i] = atom.frac.x; send_frac[3 * i + 1] = atom.frac.y; send_frac[3 * i + 2] = atom.frac.z; + send_frac[3 * slot.send_atom_indices.size() + 3 * i] = atom.frac.x + image_shift[0]; + send_frac[3 * slot.send_atom_indices.size() + 3 * i + 1] = atom.frac.y + image_shift[1]; + send_frac[3 * slot.send_atom_indices.size() + 3 * i + 2] = atom.frac.z + image_shift[2]; } - std::vector recv_frac(3 * static_cast(slot.ghost_count), 0.0); + std::vector recv_frac(6 * static_cast(slot.ghost_count), 0.0); exchange_bytes_(send_frac.data(), send_frac.size() * sizeof(double), recv_frac.data(), recv_frac.size() * sizeof(double), slot.send_rank, slot.recv_rank, tag_ghost_positions); @@ -335,12 +354,10 @@ void DomainDecomposition::update_ghost_atom_positions(MDCell& cell) { LocalAtom& ghost = ghost_atoms[slot.ghost_begin + static_cast(i)]; ghost.frac.set(recv_frac[3 * i], recv_frac[3 * i + 1], recv_frac[3 * i + 2]); - const std::array& image_shift = slot.send_rank == rank_ && slot.recv_rank == rank_ - ? slot.image_shift - : slot.recv_image_shift; - const ModuleBase::Vector3 image_frac(ghost.frac.x + image_shift[0], - ghost.frac.y + image_shift[1], - ghost.frac.z + image_shift[2]); + const ModuleBase::Vector3 image_frac( + recv_frac[3 * static_cast(slot.ghost_count) + 3 * i], + recv_frac[3 * static_cast(slot.ghost_count) + 3 * i + 1], + recv_frac[3 * static_cast(slot.ghost_count) + 3 * i + 2]); ghost.cart = image_frac * latvec_; ghost.force.set(0.0, 0.0, 0.0); } @@ -470,18 +487,25 @@ void DomainDecomposition::prepare_neighbors(MDCell& cell) bool rebuild = !cell.neighbor_layout_valid_ || cell.neighbor_reference_frac_.size() != cell.owned_atoms_.size(); rebuild = domain.max(rebuild || !ghost_layout_valid_ ? 1 : 0) != 0; double local_max_displacement = 0.0; + bool local_periodic_crossing = false; if (!rebuild) { for (std::size_t i = 0; i < cell.owned_atoms_.size(); ++i) { ModuleBase::Vector3 delta = cell.owned_atoms_[i].frac - cell.neighbor_reference_frac_[i]; + if (std::nearbyint(delta.x) != 0.0 || std::nearbyint(delta.y) != 0.0 + || std::nearbyint(delta.z) != 0.0) + { + local_periodic_crossing = true; + } delta.x -= std::nearbyint(delta.x); delta.y -= std::nearbyint(delta.y); delta.z -= std::nearbyint(delta.z); local_max_displacement = std::max(local_max_displacement, (delta * cell.latvec_).norm() * cell.lat0_); } local_max_displacement = domain.max(local_max_displacement); - rebuild = local_max_displacement >= cell.skin_ * 0.5; + rebuild = local_max_displacement >= cell.skin_ * 0.5 + || domain.max(local_periodic_crossing ? 1 : 0) != 0; } if (rebuild) @@ -775,6 +799,35 @@ void DomainDecomposition::target_for_offset(const std::array& offset, } } +std::array DomainDecomposition::image_shift_for_update( + const LocalAtom& atom, + const std::array& previous_frac, + const std::array& previous_image_shift, + const GhostExchangeSlot& slot) const +{ + std::array image_shift = previous_image_shift; + const double frac[3] = {atom.frac.x, atom.frac.y, atom.frac.z}; + for (int idim = 0; idim < 3; ++idim) + { + if (slot.offset[idim] == 0) + { + continue; + } + + const int crossing = static_cast(std::nearbyint(frac[idim] - previous_frac[idim])); + if (crossing == 0) + { + continue; + } + + // A positive wrapped displacement changes the image index in the + // opposite direction for a positive offset, and vice versa. This + // preserves the same physical image continuously across a boundary. + image_shift[idim] -= slot.offset[idim] > 0 ? crossing : -crossing; + } + return image_shift; +} + int DomainDecomposition::neighbor_layer(int dim) const { return std::max(1, static_cast(std::ceil(margin_[dim] * dims_[dim]))); diff --git a/source/source_cell/module_neighlist/domain_decomposition.h b/source/source_cell/domain_decomposition.h similarity index 90% rename from source/source_cell/module_neighlist/domain_decomposition.h rename to source/source_cell/domain_decomposition.h index cd1b8436f9e..8c503525bad 100644 --- a/source/source_cell/module_neighlist/domain_decomposition.h +++ b/source/source_cell/domain_decomposition.h @@ -76,6 +76,8 @@ class DomainDecomposition int send_rank; int recv_rank; std::vector send_atom_indices; + std::vector> previous_fracs; + std::vector> dynamic_image_shifts; std::size_t ghost_begin; int ghost_count; }; @@ -110,6 +112,10 @@ class DomainDecomposition void target_for_offset(const std::array& offset, std::array& target_coords, std::array& image_shift) const; + std::array image_shift_for_update(const LocalAtom& atom, + const std::array& previous_frac, + const std::array& previous_image_shift, + const GhostExchangeSlot& slot) const; void build_ghost_exchange_slots(std::vector& slots) const; PackedAtom pack_atom(const LocalAtom& atom, const std::array& image_shift) const; LocalAtom unpack_ghost_atom(const PackedAtom& packed) const; diff --git a/source/source_cell/kernels/cuda/gpu_neighlist_cuda.h b/source/source_cell/kernels/cuda/gpu_neighlist_cuda.h new file mode 100644 index 00000000000..8a9a8b97677 --- /dev/null +++ b/source/source_cell/kernels/cuda/gpu_neighlist_cuda.h @@ -0,0 +1,37 @@ +#ifndef SOURCE_CELL_KERNELS_CUDA_GPU_NEIGHLIST_CUDA_H +#define SOURCE_CELL_KERNELS_CUDA_GPU_NEIGHLIST_CUDA_H + +#include + +namespace gpu_neighlist_detail +{ + +int set_message(char* message, int message_size, const char* text); +int check_cuda(cudaError_t status, char* message, int message_size, const char* where); + +__global__ void count_bins_kernel(int nall, const double* position, double x_min, double y_min, + double z_min, double bin_size, int nbinx, int nbiny, + int nbinz, int* bin_counts); +__global__ void fill_bins_kernel(int nall, const double* position, double x_min, double y_min, + double z_min, double bin_size, int nbinx, int nbiny, + int nbinz, int* bin_cursor, int* bin_atoms); +__global__ void count_neighbors_kernel(int nall, const double* position, double x_min, + double y_min, double z_min, double bin_size, + double cutoff2, int nbinx, int nbiny, int nbinz, + const int* bin_offsets, const int* bin_atoms, + int* neighbor_count); +__global__ void fill_neighbors_kernel(int nall, int max_neighbors, const double* position, + double x_min, double y_min, double z_min, double bin_size, + double cutoff2, int nbinx, int nbiny, int nbinz, + const int* bin_offsets, const int* bin_atoms, + const int* neighbor_count, int* neighbor_indices); +__global__ void filter_neighbors_kernel(int nall, int candidate_max_neighbors, + const double* position, double cutoff2, + const int* candidate_count, const int* candidate_indices, + int* neighbor_count, int* neighbor_indices); +__global__ void max_neighbor_count_kernel(int nall, const int* neighbor_count, int* max_neighbors); + +} // namespace gpu_neighlist_detail + +#endif + diff --git a/source/source_cell/kernels/cuda/gpu_neighlist_device.cu b/source/source_cell/kernels/cuda/gpu_neighlist_device.cu new file mode 100644 index 00000000000..1d4a45b2408 --- /dev/null +++ b/source/source_cell/kernels/cuda/gpu_neighlist_device.cu @@ -0,0 +1,291 @@ +#include "gpu_neighlist_cuda.h" + +#include +#include +#include +#include + +namespace gpu_neighlist_detail +{ + +int build_device_impl(int nall, + double cutoff, + const double* position, + void** device_neighbor_count, + void** device_neighbor_indices, + int* max_neighbors, + char* message, + int message_size) +{ + if (nall <= 0 || cutoff <= 0.0 || position == nullptr + || device_neighbor_count == nullptr || device_neighbor_indices == nullptr + || max_neighbors == nullptr) + { + return set_message(message, message_size, "invalid GPU device neighbor-list arguments"); + } + *device_neighbor_count = nullptr; + *device_neighbor_indices = nullptr; + *max_neighbors = 0; + + double x_min = position[0]; + double x_max = position[0]; + double y_min = position[nall]; + double y_max = position[nall]; + double z_min = position[2 * nall]; + double z_max = position[2 * nall]; + for (int i = 1; i < nall; ++i) + { + x_min = std::min(x_min, position[i]); + x_max = std::max(x_max, position[i]); + y_min = std::min(y_min, position[nall + i]); + y_max = std::max(y_max, position[nall + i]); + z_min = std::min(z_min, position[2 * nall + i]); + z_max = std::max(z_max, position[2 * nall + i]); + } + const double bin_size = cutoff; + const int nbinx = std::max(1, static_cast(std::ceil((x_max - x_min) / bin_size)) + 1); + const int nbiny = std::max(1, static_cast(std::ceil((y_max - y_min) / bin_size)) + 1); + const int nbinz = std::max(1, static_cast(std::ceil((z_max - z_min) / bin_size)) + 1); + const long long total_bins_ll = static_cast(nbinx) * nbiny * nbinz; + if (total_bins_ll > std::numeric_limits::max()) + { + return set_message(message, message_size, "GPU neighbor-list bin count exceeds int range"); + } + const int total_bins = static_cast(total_bins_ll); + const int threads = 256; + const int blocks = (nall + threads - 1) / threads; + double* d_position = nullptr; + int* d_bin_counts = nullptr; + int* d_bin_offsets = nullptr; + int* d_bin_cursor = nullptr; + int* d_bin_atoms = nullptr; + int* d_neighbor_count = nullptr; + int* d_neighbor_indices = nullptr; + int* d_max_neighbors = nullptr; + std::vector bin_counts(static_cast(total_bins), 0); + std::vector bin_offsets(static_cast(total_bins + 1), 0); + int status = 0; + status = check_cuda(cudaMalloc(reinterpret_cast(&d_position), sizeof(double) * 3 * nall), + message, message_size, "cudaMalloc device neighbor position"); + if (status != 0) goto cleanup; + status = check_cuda(cudaMalloc(reinterpret_cast(&d_bin_counts), sizeof(int) * total_bins), + message, message_size, "cudaMalloc device bin counts"); + if (status != 0) goto cleanup; + status = check_cuda(cudaMalloc(reinterpret_cast(&d_bin_offsets), + sizeof(int) * (total_bins + 1)), + message, message_size, "cudaMalloc device bin offsets"); + if (status != 0) goto cleanup; + status = check_cuda(cudaMalloc(reinterpret_cast(&d_bin_cursor), sizeof(int) * total_bins), + message, message_size, "cudaMalloc device bin cursor"); + if (status != 0) goto cleanup; + status = check_cuda(cudaMalloc(reinterpret_cast(&d_bin_atoms), sizeof(int) * nall), + message, message_size, "cudaMalloc device bin atoms"); + if (status != 0) goto cleanup; + status = check_cuda(cudaMalloc(reinterpret_cast(&d_neighbor_count), sizeof(int) * nall), + message, message_size, "cudaMalloc device neighbor count"); + if (status != 0) goto cleanup; + status = check_cuda(cudaMalloc(reinterpret_cast(&d_max_neighbors), sizeof(int)), + message, message_size, "cudaMalloc device max neighbor count"); + if (status != 0) goto cleanup; + status = check_cuda(cudaMemcpy(d_position, position, sizeof(double) * 3 * nall, + cudaMemcpyHostToDevice), + message, message_size, "copy device neighbor position"); + if (status != 0) goto cleanup; + status = check_cuda(cudaMemset(d_bin_counts, 0, sizeof(int) * total_bins), + message, message_size, "clear device bin counts"); + if (status != 0) goto cleanup; + count_bins_kernel<<>>(nall, d_position, x_min, y_min, z_min, bin_size, + nbinx, nbiny, nbinz, d_bin_counts); + status = check_cuda(cudaGetLastError(), message, message_size, "count device bins"); + if (status != 0) goto cleanup; + status = check_cuda(cudaDeviceSynchronize(), message, message_size, "synchronize device bins"); + if (status != 0) goto cleanup; + status = check_cuda(cudaMemcpy(bin_counts.data(), d_bin_counts, sizeof(int) * total_bins, + cudaMemcpyDeviceToHost), + message, message_size, "copy device bin counts"); + if (status != 0) goto cleanup; + for (int bin = 0; bin < total_bins; ++bin) + { + bin_offsets[static_cast(bin + 1)] + = bin_offsets[static_cast(bin)] + bin_counts[static_cast(bin)]; + } + status = check_cuda(cudaMemcpy(d_bin_offsets, bin_offsets.data(), + sizeof(int) * (total_bins + 1), cudaMemcpyHostToDevice), + message, message_size, "copy device bin offsets"); + if (status != 0) goto cleanup; + status = check_cuda(cudaMemcpy(d_bin_cursor, bin_offsets.data(), sizeof(int) * total_bins, + cudaMemcpyHostToDevice), + message, message_size, "copy device bin cursor"); + if (status != 0) goto cleanup; + fill_bins_kernel<<>>(nall, d_position, x_min, y_min, z_min, bin_size, + nbinx, nbiny, nbinz, d_bin_cursor, d_bin_atoms); + status = check_cuda(cudaGetLastError(), message, message_size, "fill device bins"); + if (status != 0) goto cleanup; + count_neighbors_kernel<<>>(nall, d_position, x_min, y_min, z_min, bin_size, + cutoff * cutoff, nbinx, nbiny, nbinz, + d_bin_offsets, d_bin_atoms, d_neighbor_count); + status = check_cuda(cudaGetLastError(), message, message_size, "count device neighbors"); + if (status != 0) goto cleanup; + status = check_cuda(cudaMemset(d_max_neighbors, 0, sizeof(int)), + message, message_size, "clear device max neighbors"); + if (status != 0) goto cleanup; + max_neighbor_count_kernel<<>>(nall, d_neighbor_count, d_max_neighbors); + status = check_cuda(cudaGetLastError(), message, message_size, "reduce device max neighbors"); + if (status != 0) goto cleanup; + status = check_cuda(cudaMemcpy(max_neighbors, d_max_neighbors, sizeof(int), cudaMemcpyDeviceToHost), + message, message_size, "copy device max neighbors"); + if (status != 0) goto cleanup; + if (*max_neighbors > 0) + { + const std::size_t list_size = static_cast(nall) + * static_cast(*max_neighbors); + status = check_cuda(cudaMalloc(reinterpret_cast(&d_neighbor_indices), + sizeof(int) * list_size), + message, message_size, "cudaMalloc device neighbor indices"); + if (status != 0) goto cleanup; + fill_neighbors_kernel<<>>(nall, *max_neighbors, d_position, + x_min, y_min, z_min, bin_size, + cutoff * cutoff, nbinx, nbiny, nbinz, + d_bin_offsets, d_bin_atoms, d_neighbor_count, + d_neighbor_indices); + status = check_cuda(cudaGetLastError(), message, message_size, "fill device neighbors"); + if (status != 0) goto cleanup; + } + *device_neighbor_count = d_neighbor_count; + *device_neighbor_indices = d_neighbor_indices; + d_neighbor_count = nullptr; + d_neighbor_indices = nullptr; + +cleanup: + cudaFree(d_position); + cudaFree(d_bin_counts); + cudaFree(d_bin_offsets); + cudaFree(d_bin_cursor); + cudaFree(d_bin_atoms); + cudaFree(d_neighbor_count); + cudaFree(d_neighbor_indices); + cudaFree(d_max_neighbors); + return status; +} + +int filter_device_impl(int nall, + double cutoff, + const double* position, + int candidate_max_neighbors, + const int* device_candidate_count, + const int* device_candidate_indices, + void** device_neighbor_count, + void** device_neighbor_indices, + int* max_neighbors, + char* message, + int message_size) +{ + if (nall <= 0 || cutoff <= 0.0 || position == nullptr + || candidate_max_neighbors < 0 || device_candidate_count == nullptr + || device_neighbor_count == nullptr || device_neighbor_indices == nullptr + || (candidate_max_neighbors > 0 && device_candidate_indices == nullptr) + || max_neighbors == nullptr) + { + return set_message(message, message_size, "invalid GPU device filter arguments"); + } + *device_neighbor_count = nullptr; + *device_neighbor_indices = nullptr; + *max_neighbors = 0; + const int threads = 256; + const int blocks = (nall + threads - 1) / threads; + double* d_position = nullptr; + int* d_neighbor_count = nullptr; + int* d_neighbor_indices = nullptr; + int* d_max_neighbors = nullptr; + int status = check_cuda(cudaMalloc(reinterpret_cast(&d_position), sizeof(double) * 3 * nall), + message, message_size, "cudaMalloc filter device position"); + if (status != 0) goto cleanup; + status = check_cuda(cudaMalloc(reinterpret_cast(&d_neighbor_count), sizeof(int) * nall), + message, message_size, "cudaMalloc filtered device count"); + if (status != 0) goto cleanup; + status = check_cuda(cudaMalloc(reinterpret_cast(&d_max_neighbors), sizeof(int)), + message, message_size, "cudaMalloc filtered max neighbors"); + if (status != 0) goto cleanup; + if (candidate_max_neighbors > 0) + { + const std::size_t list_size = static_cast(nall) + * static_cast(candidate_max_neighbors); + status = check_cuda(cudaMalloc(reinterpret_cast(&d_neighbor_indices), + sizeof(int) * list_size), + message, message_size, "cudaMalloc filtered device indices"); + if (status != 0) goto cleanup; + } + status = check_cuda(cudaMemcpy(d_position, position, sizeof(double) * 3 * nall, + cudaMemcpyHostToDevice), + message, message_size, "copy filter device position"); + if (status != 0) goto cleanup; + filter_neighbors_kernel<<>>(nall, + candidate_max_neighbors, + d_position, + cutoff * cutoff, + device_candidate_count, + device_candidate_indices, + d_neighbor_count, + d_neighbor_indices); + status = check_cuda(cudaGetLastError(), message, message_size, "filter device neighbors"); + if (status != 0) goto cleanup; + status = check_cuda(cudaMemset(d_max_neighbors, 0, sizeof(int)), + message, message_size, "clear filtered max neighbors"); + if (status != 0) goto cleanup; + max_neighbor_count_kernel<<>>(nall, d_neighbor_count, d_max_neighbors); + status = check_cuda(cudaGetLastError(), message, message_size, "reduce filtered max neighbors"); + if (status != 0) goto cleanup; + status = check_cuda(cudaMemcpy(max_neighbors, d_max_neighbors, sizeof(int), cudaMemcpyDeviceToHost), + message, message_size, "copy filtered max neighbors"); + if (status != 0) goto cleanup; + // The filter kernel writes with candidate_max_neighbors as its row stride. + // Keep that stride for the NEP kernels; the actual count remains in d_neighbor_count. + *max_neighbors = candidate_max_neighbors; + *device_neighbor_count = d_neighbor_count; + *device_neighbor_indices = d_neighbor_indices; + d_neighbor_count = nullptr; + d_neighbor_indices = nullptr; + +cleanup: + cudaFree(d_position); + cudaFree(d_neighbor_count); + cudaFree(d_neighbor_indices); + cudaFree(d_max_neighbors); + return status; +} + + +} // namespace gpu_neighlist_detail + +int gpu_build_neighbor_list_device(int nall, + double cutoff, + const double* position, + void** device_neighbor_count, + void** device_neighbor_indices, + int* max_neighbors, + char* message, + int message_size) +{ + return gpu_neighlist_detail::build_device_impl(nall, cutoff, position, device_neighbor_count, + device_neighbor_indices, max_neighbors, message, message_size); +} + +int gpu_filter_neighbor_list_device(int nall, + double cutoff, + const double* position, + int candidate_max_neighbors, + const void* device_candidate_count, + const void* device_candidate_indices, + void** device_neighbor_count, + void** device_neighbor_indices, + int* max_neighbors, + char* message, + int message_size) +{ + return gpu_neighlist_detail::filter_device_impl(nall, cutoff, position, candidate_max_neighbors, + static_cast(device_candidate_count), + static_cast(device_candidate_indices), + device_neighbor_count, device_neighbor_indices, + max_neighbors, message, message_size); +} + diff --git a/source/source_cell/kernels/cuda/gpu_neighlist_host.cu b/source/source_cell/kernels/cuda/gpu_neighlist_host.cu new file mode 100644 index 00000000000..c03cd522d54 --- /dev/null +++ b/source/source_cell/kernels/cuda/gpu_neighlist_host.cu @@ -0,0 +1,371 @@ +#include "gpu_neighlist_cuda.h" + +#include +#include +#include +#include +#include +#include + +namespace gpu_neighlist_detail +{ + +int build_impl(int nall, + double cutoff, + const double* position, + int* neighbor_count, + int* neighbor_indices, + int* max_neighbors, + char* message, + int message_size) +{ + if (nall <= 0 || cutoff <= 0.0 || position == nullptr || neighbor_count == nullptr + || max_neighbors == nullptr) + { + return set_message(message, message_size, "invalid GPU neighbor-list arguments"); + } + + double x_min = position[0]; + double x_max = position[0]; + double y_min = position[nall]; + double y_max = position[nall]; + double z_min = position[2 * nall]; + double z_max = position[2 * nall]; + for (int i = 1; i < nall; ++i) + { + x_min = std::min(x_min, position[i]); + x_max = std::max(x_max, position[i]); + y_min = std::min(y_min, position[nall + i]); + y_max = std::max(y_max, position[nall + i]); + z_min = std::min(z_min, position[2 * nall + i]); + z_max = std::max(z_max, position[2 * nall + i]); + } + + const double bin_size = cutoff; + const int nbinx = std::max(1, static_cast(std::ceil((x_max - x_min) / bin_size)) + 1); + const int nbiny = std::max(1, static_cast(std::ceil((y_max - y_min) / bin_size)) + 1); + const int nbinz = std::max(1, static_cast(std::ceil((z_max - z_min) / bin_size)) + 1); + const long long total_bins_ll = static_cast(nbinx) * nbiny * nbinz; + if (total_bins_ll > std::numeric_limits::max()) + { + return set_message(message, message_size, "GPU neighbor-list bin count exceeds int range"); + } + const int total_bins = static_cast(total_bins_ll); + const int threads = 256; + const int blocks = (nall + threads - 1) / threads; + + double* d_position = nullptr; + int* d_bin_counts = nullptr; + int* d_bin_offsets = nullptr; + int* d_bin_cursor = nullptr; + int* d_bin_atoms = nullptr; + int* d_neighbor_count = nullptr; + int* d_neighbor_indices = nullptr; + std::vector bin_counts(static_cast(total_bins), 0); + std::vector bin_offsets(static_cast(total_bins + 1), 0); + int status = 0; + status = check_cuda(cudaMalloc(reinterpret_cast(&d_position), sizeof(double) * 3 * nall), + message, message_size, "cudaMalloc position"); + if (status != 0) goto cleanup; + status = check_cuda(cudaMalloc(reinterpret_cast(&d_bin_counts), sizeof(int) * total_bins), + message, message_size, "cudaMalloc bin counts"); + if (status != 0) goto cleanup; + status = check_cuda(cudaMalloc(reinterpret_cast(&d_bin_offsets), sizeof(int) * (total_bins + 1)), + message, message_size, "cudaMalloc bin offsets"); + if (status != 0) goto cleanup; + status = check_cuda(cudaMalloc(reinterpret_cast(&d_bin_cursor), sizeof(int) * total_bins), + message, message_size, "cudaMalloc bin cursor"); + if (status != 0) goto cleanup; + status = check_cuda(cudaMalloc(reinterpret_cast(&d_bin_atoms), sizeof(int) * nall), + message, message_size, "cudaMalloc bin atoms"); + if (status != 0) goto cleanup; + status = check_cuda(cudaMalloc(reinterpret_cast(&d_neighbor_count), sizeof(int) * nall), + message, message_size, "cudaMalloc neighbor count"); + if (status != 0) goto cleanup; + status = check_cuda(cudaMemcpy(d_position, position, sizeof(double) * 3 * nall, cudaMemcpyHostToDevice), + message, message_size, "cudaMemcpy position"); + if (status != 0) goto cleanup; + status = check_cuda(cudaMemset(d_bin_counts, 0, sizeof(int) * total_bins), + message, message_size, "cudaMemset bin counts"); + if (status != 0) goto cleanup; + count_bins_kernel<<>>(nall, d_position, x_min, y_min, z_min, bin_size, + nbinx, nbiny, nbinz, d_bin_counts); + status = check_cuda(cudaGetLastError(), message, message_size, "count bins kernel"); + if (status != 0) goto cleanup; + status = check_cuda(cudaDeviceSynchronize(), message, message_size, "count bins synchronize"); + if (status != 0) goto cleanup; + status = check_cuda(cudaMemcpy(bin_counts.data(), d_bin_counts, sizeof(int) * total_bins, + cudaMemcpyDeviceToHost), + message, message_size, "copy bin counts"); + if (status != 0) goto cleanup; + for (int bin = 0; bin < total_bins; ++bin) + { + bin_offsets[static_cast(bin + 1)] + = bin_offsets[static_cast(bin)] + bin_counts[static_cast(bin)]; + } + status = check_cuda(cudaMemcpy(d_bin_offsets, bin_offsets.data(), sizeof(int) * (total_bins + 1), + cudaMemcpyHostToDevice), + message, message_size, "copy bin offsets"); + if (status != 0) goto cleanup; + status = check_cuda(cudaMemcpy(d_bin_cursor, bin_offsets.data(), sizeof(int) * total_bins, + cudaMemcpyHostToDevice), + message, message_size, "copy bin cursor"); + if (status != 0) goto cleanup; + fill_bins_kernel<<>>(nall, d_position, x_min, y_min, z_min, bin_size, + nbinx, nbiny, nbinz, d_bin_cursor, d_bin_atoms); + status = check_cuda(cudaGetLastError(), message, message_size, "fill bins kernel"); + if (status != 0) goto cleanup; + status = check_cuda(cudaDeviceSynchronize(), message, message_size, "fill bins synchronize"); + if (status != 0) goto cleanup; + count_neighbors_kernel<<>>(nall, d_position, x_min, y_min, z_min, bin_size, + cutoff * cutoff, nbinx, nbiny, nbinz, + d_bin_offsets, d_bin_atoms, d_neighbor_count); + status = check_cuda(cudaGetLastError(), message, message_size, "count neighbors kernel"); + if (status != 0) goto cleanup; + status = check_cuda(cudaDeviceSynchronize(), message, message_size, "count neighbors synchronize"); + if (status != 0) goto cleanup; + status = check_cuda(cudaMemcpy(neighbor_count, d_neighbor_count, sizeof(int) * nall, + cudaMemcpyDeviceToHost), + message, message_size, "copy neighbor counts"); + if (status != 0) goto cleanup; + *max_neighbors = 0; + for (int i = 0; i < nall; ++i) + { + *max_neighbors = std::max(*max_neighbors, neighbor_count[i]); + } + if (neighbor_indices != nullptr && *max_neighbors > 0) + { + const std::size_t list_size = static_cast(nall) + * static_cast(*max_neighbors); + status = check_cuda(cudaMalloc(reinterpret_cast(&d_neighbor_indices), + sizeof(int) * list_size), + message, message_size, "cudaMalloc neighbor indices"); + if (status != 0) goto cleanup; + fill_neighbors_kernel<<>>(nall, *max_neighbors, d_position, + x_min, y_min, z_min, bin_size, + cutoff * cutoff, nbinx, nbiny, nbinz, + d_bin_offsets, d_bin_atoms, d_neighbor_count, + d_neighbor_indices); + status = check_cuda(cudaGetLastError(), message, message_size, "fill neighbors kernel"); + if (status != 0) goto cleanup; + status = check_cuda(cudaDeviceSynchronize(), message, message_size, "fill neighbors synchronize"); + if (status != 0) goto cleanup; + status = check_cuda(cudaMemcpy(neighbor_indices, d_neighbor_indices, sizeof(int) * list_size, + cudaMemcpyDeviceToHost), + message, message_size, "copy neighbor indices"); + } + +cleanup: + cudaFree(d_position); + cudaFree(d_bin_counts); + cudaFree(d_bin_offsets); + cudaFree(d_bin_cursor); + cudaFree(d_bin_atoms); + cudaFree(d_neighbor_count); + cudaFree(d_neighbor_indices); + return status; +} + +int filter_impl(int nall, + double cutoff, + const double* position, + const int* candidate_count, + const int* candidate_indices, + int candidate_max_neighbors, + int* neighbor_count, + int* neighbor_indices, + int* max_neighbors, + char* message, + int message_size) +{ + if (nall <= 0 || cutoff <= 0.0 || position == nullptr || candidate_count == nullptr + || max_neighbors == nullptr || candidate_max_neighbors < 0 + || (candidate_max_neighbors > 0 && (candidate_indices == nullptr || neighbor_indices == nullptr))) + { + return set_message(message, message_size, "invalid GPU neighbor-list filter arguments"); + } + + const int threads = 256; + const int blocks = (nall + threads - 1) / threads; + double* d_position = nullptr; + int* d_candidate_count = nullptr; + int* d_candidate_indices = nullptr; + int* d_neighbor_count = nullptr; + int* d_neighbor_indices = nullptr; + int status = 0; + status = check_cuda(cudaMalloc(reinterpret_cast(&d_position), sizeof(double) * 3 * nall), + message, message_size, "cudaMalloc filter position"); + if (status != 0) goto cleanup; + status = check_cuda(cudaMalloc(reinterpret_cast(&d_candidate_count), sizeof(int) * nall), + message, message_size, "cudaMalloc candidate count"); + if (status != 0) goto cleanup; + status = check_cuda(cudaMalloc(reinterpret_cast(&d_neighbor_count), sizeof(int) * nall), + message, message_size, "cudaMalloc filtered count"); + if (status != 0) goto cleanup; + if (candidate_max_neighbors > 0) + { + const std::size_t list_size = static_cast(nall) + * static_cast(candidate_max_neighbors); + status = check_cuda(cudaMalloc(reinterpret_cast(&d_candidate_indices), + sizeof(int) * list_size), + message, message_size, "cudaMalloc candidate indices"); + if (status != 0) goto cleanup; + status = check_cuda(cudaMalloc(reinterpret_cast(&d_neighbor_indices), + sizeof(int) * list_size), + message, message_size, "cudaMalloc filtered indices"); + if (status != 0) goto cleanup; + status = check_cuda(cudaMemcpy(d_candidate_indices, + candidate_indices, + sizeof(int) * list_size, + cudaMemcpyHostToDevice), + message, message_size, "copy candidate indices"); + if (status != 0) goto cleanup; + } + status = check_cuda(cudaMemcpy(d_position, position, sizeof(double) * 3 * nall, + cudaMemcpyHostToDevice), + message, message_size, "copy filter position"); + if (status != 0) goto cleanup; + status = check_cuda(cudaMemcpy(d_candidate_count, candidate_count, sizeof(int) * nall, + cudaMemcpyHostToDevice), + message, message_size, "copy candidate count"); + if (status != 0) goto cleanup; + filter_neighbors_kernel<<>>(nall, + candidate_max_neighbors, + d_position, + cutoff * cutoff, + d_candidate_count, + d_candidate_indices, + d_neighbor_count, + d_neighbor_indices); + status = check_cuda(cudaGetLastError(), message, message_size, "filter neighbors kernel"); + if (status != 0) goto cleanup; + status = check_cuda(cudaDeviceSynchronize(), message, message_size, "filter neighbors synchronize"); + if (status != 0) goto cleanup; + status = check_cuda(cudaMemcpy(neighbor_count, d_neighbor_count, sizeof(int) * nall, + cudaMemcpyDeviceToHost), + message, message_size, "copy filtered count"); + if (status != 0) goto cleanup; + *max_neighbors = 0; + for (int i = 0; i < nall; ++i) + { + *max_neighbors = std::max(*max_neighbors, neighbor_count[i]); + } + if (candidate_max_neighbors > 0) + { + const std::size_t list_size = static_cast(nall) + * static_cast(candidate_max_neighbors); + status = check_cuda(cudaMemcpy(neighbor_indices, + d_neighbor_indices, + sizeof(int) * list_size, + cudaMemcpyDeviceToHost), + message, message_size, "copy filtered indices"); + } + +cleanup: + cudaFree(d_position); + cudaFree(d_candidate_count); + cudaFree(d_candidate_indices); + cudaFree(d_neighbor_count); + cudaFree(d_neighbor_indices); + return status; +} + + +} // namespace gpu_neighlist_detail + +int gpu_build_neighbor_list(int nall, + double cutoff, + const double* position, + int* neighbor_count, + int* neighbor_indices, + int* max_neighbors, + char* message, + int message_size) +{ + return gpu_neighlist_detail::build_impl(nall, cutoff, position, neighbor_count, neighbor_indices, + max_neighbors, message, message_size); +} + +int gpu_filter_neighbor_list(int nall, + double cutoff, + const double* position, + const int* candidate_count, + const int* candidate_indices, + int candidate_max_neighbors, + int* neighbor_count, + int* neighbor_indices, + int* max_neighbors, + char* message, + int message_size) +{ + return gpu_neighlist_detail::filter_impl(nall, cutoff, position, candidate_count, candidate_indices, + candidate_max_neighbors, neighbor_count, neighbor_indices, + max_neighbors, message, message_size); +} + +int gpu_upload_neighbor_list(int nall, + int max_neighbors, + const int* neighbor_count, + const int* neighbor_indices, + void** device_neighbor_count, + void** device_neighbor_indices, + char* message, + int message_size) +{ + if (nall <= 0 || max_neighbors < 0 || neighbor_count == nullptr + || device_neighbor_count == nullptr || device_neighbor_indices == nullptr + || (max_neighbors > 0 && neighbor_indices == nullptr)) + { + return gpu_neighlist_detail::set_message(message, message_size, "invalid GPU neighbor-list upload arguments"); + } + *device_neighbor_count = nullptr; + *device_neighbor_indices = nullptr; + int* d_count = nullptr; + int* d_indices = nullptr; + int status = gpu_neighlist_detail::check_cuda(cudaMalloc(reinterpret_cast(&d_count), sizeof(int) * nall), + message, message_size, "cudaMalloc uploaded neighbor count"); + if (status != 0) + { + return status; + } + status = gpu_neighlist_detail::check_cuda(cudaMemcpy(d_count, neighbor_count, sizeof(int) * nall, + cudaMemcpyHostToDevice), + message, message_size, "copy uploaded neighbor count"); + if (status != 0) + { + cudaFree(d_count); + return status; + } + if (max_neighbors > 0) + { + const std::size_t list_size = static_cast(nall) + * static_cast(max_neighbors); + status = gpu_neighlist_detail::check_cuda(cudaMalloc(reinterpret_cast(&d_indices), + sizeof(int) * list_size), + message, message_size, "cudaMalloc uploaded neighbor indices"); + if (status != 0) + { + cudaFree(d_count); + return status; + } + status = gpu_neighlist_detail::check_cuda(cudaMemcpy(d_indices, neighbor_indices, sizeof(int) * list_size, + cudaMemcpyHostToDevice), + message, message_size, "copy uploaded neighbor indices"); + if (status != 0) + { + cudaFree(d_count); + cudaFree(d_indices); + return status; + } + } + *device_neighbor_count = d_count; + *device_neighbor_indices = d_indices; + return 0; +} + +void gpu_release_neighbor_list(void* device_neighbor_count, + void* device_neighbor_indices) +{ + cudaFree(device_neighbor_count); + cudaFree(device_neighbor_indices); +} diff --git a/source/source_cell/kernels/cuda/gpu_neighlist_kernels.cu b/source/source_cell/kernels/cuda/gpu_neighlist_kernels.cu new file mode 100644 index 00000000000..332deaf5d09 --- /dev/null +++ b/source/source_cell/kernels/cuda/gpu_neighlist_kernels.cu @@ -0,0 +1,255 @@ +#include + +#include +#include +#include +#include +#include +#include + +namespace gpu_neighlist_detail +{ + +int set_message(char* message, int message_size, const char* text) +{ + if (message != nullptr && message_size > 0) + { + std::snprintf(message, message_size, "%s", text); + } + return 1; +} + +int check_cuda(cudaError_t status, char* message, int message_size, const char* where) +{ + if (status == cudaSuccess) + { + return 0; + } + if (message != nullptr && message_size > 0) + { + std::snprintf(message, message_size, "%s: %s", where, cudaGetErrorString(status)); + } + return 1; +} + +__global__ void count_bins_kernel(int nall, + const double* position, + double x_min, + double y_min, + double z_min, + double bin_size, + int nbinx, + int nbiny, + int nbinz, + int* bin_counts) +{ + const int atom = blockIdx.x * blockDim.x + threadIdx.x; + if (atom >= nall) + { + return; + } + const int ix = min(max(static_cast((position[atom] - x_min) / bin_size), 0), nbinx - 1); + const int iy = min(max(static_cast((position[nall + atom] - y_min) / bin_size), 0), nbiny - 1); + const int iz = min(max(static_cast((position[2 * nall + atom] - z_min) / bin_size), 0), nbinz - 1); + const int bin = ix * nbiny * nbinz + iy * nbinz + iz; + atomicAdd(&bin_counts[bin], 1); +} + +__global__ void fill_bins_kernel(int nall, + const double* position, + double x_min, + double y_min, + double z_min, + double bin_size, + int nbinx, + int nbiny, + int nbinz, + int* bin_cursor, + int* bin_atoms) +{ + const int atom = blockIdx.x * blockDim.x + threadIdx.x; + if (atom >= nall) + { + return; + } + const int ix = min(max(static_cast((position[atom] - x_min) / bin_size), 0), nbinx - 1); + const int iy = min(max(static_cast((position[nall + atom] - y_min) / bin_size), 0), nbiny - 1); + const int iz = min(max(static_cast((position[2 * nall + atom] - z_min) / bin_size), 0), nbinz - 1); + const int bin = ix * nbiny * nbinz + iy * nbinz + iz; + const int slot = atomicAdd(&bin_cursor[bin], 1); + bin_atoms[slot] = atom; +} + +__global__ void count_neighbors_kernel(int nall, + const double* position, + double x_min, + double y_min, + double z_min, + double bin_size, + double cutoff2, + int nbinx, + int nbiny, + int nbinz, + const int* bin_offsets, + const int* bin_atoms, + int* neighbor_count) +{ + const int center = blockIdx.x * blockDim.x + threadIdx.x; + if (center >= nall) + { + return; + } + const int ix = min(max(static_cast((position[center] - x_min) / bin_size), 0), nbinx - 1); + const int iy = min(max(static_cast((position[nall + center] - y_min) / bin_size), 0), nbiny - 1); + const int iz = min(max(static_cast((position[2 * nall + center] - z_min) / bin_size), 0), nbinz - 1); + const double x = position[center]; + const double y = position[nall + center]; + const double z = position[2 * nall + center]; + int count = 0; + for (int dx = -1; dx <= 1; ++dx) + { + for (int dy = -1; dy <= 1; ++dy) + { + for (int dz = -1; dz <= 1; ++dz) + { + const int jx = ix + dx; + const int jy = iy + dy; + const int jz = iz + dz; + if (jx < 0 || jx >= nbinx || jy < 0 || jy >= nbiny || jz < 0 || jz >= nbinz) + { + continue; + } + const int bin = jx * nbiny * nbinz + jy * nbinz + jz; + for (int slot = bin_offsets[bin]; slot < bin_offsets[bin + 1]; ++slot) + { + const int neighbor = bin_atoms[slot]; + if (neighbor == center) + { + continue; + } + const double dx12 = x - position[neighbor]; + const double dy12 = y - position[nall + neighbor]; + const double dz12 = z - position[2 * nall + neighbor]; + if (dx12 * dx12 + dy12 * dy12 + dz12 * dz12 <= cutoff2) + { + ++count; + } + } + } + } + } + neighbor_count[center] = count; +} + +__global__ void fill_neighbors_kernel(int nall, + int max_neighbors, + const double* position, + double x_min, + double y_min, + double z_min, + double bin_size, + double cutoff2, + int nbinx, + int nbiny, + int nbinz, + const int* bin_offsets, + const int* bin_atoms, + const int* neighbor_count, + int* neighbor_indices) +{ + const int center = blockIdx.x * blockDim.x + threadIdx.x; + if (center >= nall) + { + return; + } + const int ix = min(max(static_cast((position[center] - x_min) / bin_size), 0), nbinx - 1); + const int iy = min(max(static_cast((position[nall + center] - y_min) / bin_size), 0), nbiny - 1); + const int iz = min(max(static_cast((position[2 * nall + center] - z_min) / bin_size), 0), nbinz - 1); + const double x = position[center]; + const double y = position[nall + center]; + const double z = position[2 * nall + center]; + int count = 0; + for (int dx = -1; dx <= 1; ++dx) + { + for (int dy = -1; dy <= 1; ++dy) + { + for (int dz = -1; dz <= 1; ++dz) + { + const int jx = ix + dx; + const int jy = iy + dy; + const int jz = iz + dz; + if (jx < 0 || jx >= nbinx || jy < 0 || jy >= nbiny || jz < 0 || jz >= nbinz) + { + continue; + } + const int bin = jx * nbiny * nbinz + jy * nbinz + jz; + for (int slot = bin_offsets[bin]; slot < bin_offsets[bin + 1]; ++slot) + { + const int neighbor = bin_atoms[slot]; + if (neighbor == center) + { + continue; + } + const double dx12 = x - position[neighbor]; + const double dy12 = y - position[nall + neighbor]; + const double dz12 = z - position[2 * nall + neighbor]; + if (dx12 * dx12 + dy12 * dy12 + dz12 * dz12 <= cutoff2) + { + neighbor_indices[center + nall * count] = neighbor; + ++count; + } + } + } + } + } + static_cast(neighbor_count); +} + +__global__ void filter_neighbors_kernel(int nall, + int candidate_max_neighbors, + const double* position, + double cutoff2, + const int* candidate_count, + const int* candidate_indices, + int* neighbor_count, + int* neighbor_indices) +{ + const int center = blockIdx.x * blockDim.x + threadIdx.x; + if (center >= nall) + { + return; + } + const double x = position[center]; + const double y = position[nall + center]; + const double z = position[2 * nall + center]; + int count = 0; + for (int slot = 0; slot < candidate_count[center]; ++slot) + { + const int neighbor = candidate_indices[center + nall * slot]; + const double dx = x - position[neighbor]; + const double dy = y - position[nall + neighbor]; + const double dz = z - position[2 * nall + neighbor]; + if (dx * dx + dy * dy + dz * dz <= cutoff2) + { + neighbor_indices[center + nall * count] = neighbor; + ++count; + } + } + neighbor_count[center] = count; + static_cast(candidate_max_neighbors); +} + +__global__ void max_neighbor_count_kernel(int nall, + const int* neighbor_count, + int* max_neighbors) +{ + const int atom = blockIdx.x * blockDim.x + threadIdx.x; + if (atom < nall) + { + atomicMax(max_neighbors, neighbor_count[atom]); + } +} + + +} // namespace gpu_neighlist_detail + diff --git a/source/source_cell/klist.cpp b/source/source_cell/klist.cpp index fd8080076ad..c0be2ebc946 100644 --- a/source/source_cell/klist.cpp +++ b/source/source_cell/klist.cpp @@ -421,8 +421,9 @@ void K_Vectors::update_use_ibz(const int& nkstot_ibz, ModuleBase::GlobalFunc::OUT(ofs_running, "nkstot now", nkstot); - // qianrui fix a bug 2021-7-13: size for the spin_mult=2 doubling in set_kup_and_kdw() - this->kvec_d.resize(this->nkstot * this->spin_mult); + // qianrui fix a bug 2021-7-13: shrink kvec_d to the (now smaller) ibz count; + // the spin_mult=2 doubling for set_kup_and_kdw() is reserved there, on demand. + this->kvec_d.resize(this->nkstot); for (int i = 0; i < this->nkstot; ++i) { @@ -445,6 +446,18 @@ void K_Vectors::set_kup_and_kdw(std::ofstream& ofs_running) { ModuleBase::TITLE("K_Vectors", "setup_kup_and_kdw"); + // grow the containers expand_spin_kpoints() is about to append the down-spin + // copy into (indices [nkstot, 2*nkstot)). Resize only these; NOT via renew(), + // which would also resize kvec_c_full -- that one must keep holding the + // original, un-doubled, un-symmetry-reduced full-BZ mesh for later consumers + // (e.g. Ewald_Vq) regardless of what nkstot has become by this point. + const int nkstot_spin = this->nkstot * this->spin_mult; + this->kvec_c.resize(nkstot_spin); + this->kvec_d.resize(nkstot_spin); + this->wk.resize(nkstot_spin); + this->ngk.resize(nkstot_spin); + this->isk.resize(nkstot_spin); + KListIO::expand_spin_kpoints(this->spin_mult, this->kvec_c, this->kvec_d, @@ -523,8 +536,9 @@ void K_Vectors::reduce_by_symmetry(const UnitCell& ucell, ibz2bz); const int nkstot_ibz = kvec_d_ibz.size(); -#ifdef __EXX // setup kstars according to the final (max-norm) kvec_d_ibz + // (used by both EXX and DFT+U's crystal-symmetry density-matrix restoration; + // no LibRI dependency, so this must not be gated behind __EXX) if (ModuleSymmetry::Symmetry::symm_flag == 1) { KListIO::build_kstars(this->kvec_d, @@ -535,7 +549,6 @@ void K_Vectors::reduce_by_symmetry(const UnitCell& ucell, [&symm](double a, double b) { return symm.equal(a, b); }, this->kstars); } -#endif // output in kpoints file skpt = KListIO::ibz_kpt_table(this->nkstot, this->kvec_d, this->ibz_index, kvec_d_ibz); @@ -665,12 +678,12 @@ void K_Vectors::mpi_k(std::ofstream& ofs_running, const int my_rank, const int m this->kvec_d, this->kvec_c_full); -#ifdef __EXX // bcast kstars (rank 0 holds the filled maps; other ranks rebuild them) + // (no LibRI dependency; needed by DFT+U's symmetry restoration too, so this + // must not be gated behind __EXX) if (ModuleSymmetry::Symmetry::symm_flag == 1) { KListIO::bcast_kstars(this->kstars, this->nkstot, my_rank); } -#endif } // END SUBROUTINE mpi_k #endif diff --git a/source/source_cell/klist.h b/source/source_cell/klist.h index 956a2be794f..f02f18a7d18 100644 --- a/source/source_cell/klist.h +++ b/source/source_cell/klist.h @@ -120,6 +120,13 @@ class K_Vectors : public ModuleCell::ReciprocalGrid this->nkstot_nospin = value; } + /// @brief Set the spin multiplicity: 1 (no doubling, also for non-collinear + /// nspin=4) or 2 (LSDA, k points split into up/down). + void set_spin_mult(int value) + { + this->spin_mult = value; + } + bool get_is_mp() const { return is_mp; @@ -164,6 +171,61 @@ class K_Vectors : public ModuleCell::ReciprocalGrid */ void set_after_vc(const ModuleBase::Matrix3& G, std::ofstream& ofs_running); + //==================================================================== + // Test seam. + // + // set() is the single production entry point: it drives read_kpoints(), + // reduce_by_symmetry(), set_kup_and_kdw() and renew() in order. The unit + // tests exercise those stages one at a time -- most of the KPT-file parsing + // paths are only reachable that way -- so each one is called through the + // wrappers here rather than by reinterpreting the access specifiers. + // + // Production code must keep going through set(); nothing outside the tests + // should call the *_for_testing() wrappers. + //==================================================================== + + bool read_kpoints_for_testing(const UnitCell& ucell, + const std::string& fn, + const bool gamma_only_local, + const double kspacing[3], + const std::string& kmesh_type, + const double koffset_in[3], + std::ofstream& ofs_running, + std::ofstream& ofs_warning, + const int my_rank) + { + return read_kpoints(ucell, + fn, + gamma_only_local, + kspacing, + kmesh_type, + koffset_in, + ofs_running, + ofs_warning, + my_rank); + } + + void renew_for_testing(const int& kpoint_number) + { + renew(kpoint_number); + } + + void reduce_by_symmetry_for_testing(const UnitCell& ucell, + const ModuleSymmetry::Symmetry& symm, + bool use_symm, + std::string& skpt, + bool& match, + const int my_rank, + std::ofstream& ofs_running) + { + reduce_by_symmetry(ucell, symm, use_symm, skpt, match, my_rank, ofs_running); + } + + void set_kup_and_kdw_for_testing(std::ofstream& ofs_running) + { + set_kup_and_kdw(ofs_running); + } + private: /// Spin multiplicity used to size the k-point list: 1 for input nspin 1 /// or 4 (non-collinear k points are not doubled) and 2 for input nspin 2 diff --git a/source/source_cell/mdcell_reader.cpp b/source/source_cell/mdcell_reader.cpp index 3665e77a433..0a2113a720d 100644 --- a/source/source_cell/mdcell_reader.cpp +++ b/source/source_cell/mdcell_reader.cpp @@ -5,7 +5,7 @@ #include "source_base/vector3.h" #include "source_cell/mdcell.h" -#include "source_cell/module_neighlist/domain_decomposition.h" +#include "source_cell/domain_decomposition.h" #include #include diff --git a/source/source_cell/module_neighbor/sltk_grid_driver.h b/source/source_cell/module_neighbor/sltk_grid_driver.h index 5020af9af1d..8ba04f64ece 100644 --- a/source/source_cell/module_neighbor/sltk_grid_driver.h +++ b/source/source_cell/module_neighbor/sltk_grid_driver.h @@ -166,6 +166,15 @@ class Grid_Driver : public Grid return adj_info.box[i]; } + /** + * @brief Get the whole adjacent atom information. + * @return adjacent atom information + */ + const AdjacentAtomInfo& getAdjacentInfo() const + { + return adj_info; + } + private: mutable AdjacentAtomInfo adj_info; ///< adjacent atom information bool test_deconstructor; ///< test deconstructor flag diff --git a/source/source_cell/module_neighlist/CMakeLists.txt b/source/source_cell/module_neighlist/CMakeLists.txt index 3e6c282f024..119eba4dfd4 100644 --- a/source/source_cell/module_neighlist/CMakeLists.txt +++ b/source/source_cell/module_neighlist/CMakeLists.txt @@ -1,10 +1,13 @@ set(neighbor_search_sources bin_manager.cpp - domain_decomposition.cpp neighbor_search.cpp page_allocator.cpp ) +if(USE_CUDA AND ENABLE_GPU_NEIGHBOR_LIST) + list(APPEND neighbor_search_sources gpu_neighbor_list.cpp) +endif() + add_library( neighbor_search OBJECT diff --git a/source/source_cell/module_neighlist/gpu_neighbor_list.cpp b/source/source_cell/module_neighlist/gpu_neighbor_list.cpp new file mode 100644 index 00000000000..65b11c12ba2 --- /dev/null +++ b/source/source_cell/module_neighlist/gpu_neighbor_list.cpp @@ -0,0 +1,366 @@ +#include "gpu_neighbor_list.h" + +#include "source_base/timer.h" + +#include + +int gpu_build_neighbor_list(int nall, + double cutoff, + const double* position, + int* neighbor_count, + int* neighbor_indices, + int* max_neighbors, + char* message, + int message_size); + +int gpu_filter_neighbor_list(int nall, + double cutoff, + const double* position, + const int* candidate_count, + const int* candidate_indices, + int candidate_max_neighbors, + int* neighbor_count, + int* neighbor_indices, + int* max_neighbors, + char* message, + int message_size); + +int gpu_build_neighbor_list_device(int nall, + double cutoff, + const double* position, + void** device_neighbor_count, + void** device_neighbor_indices, + int* max_neighbors, + char* message, + int message_size); + +int gpu_filter_neighbor_list_device(int nall, + double cutoff, + const double* position, + int candidate_max_neighbors, + const void* device_candidate_count, + const void* device_candidate_indices, + void** device_neighbor_count, + void** device_neighbor_indices, + int* max_neighbors, + char* message, + int message_size); + +int gpu_upload_neighbor_list(int nall, + int max_neighbors, + const int* neighbor_count, + const int* neighbor_indices, + void** device_neighbor_count, + void** device_neighbor_indices, + char* message, + int message_size); + +void gpu_release_neighbor_list(void* device_neighbor_count, + void* device_neighbor_indices); + +namespace ModuleNeighList +{ + +GPU_NeighborList::~GPU_NeighborList() +{ + const bool aliases_candidate = device_neighbor_count_ == candidate_device_count_ + && device_neighbor_indices_ == candidate_device_indices_; + const bool aliases_filtered = device_neighbor_count_ == filtered_device_count_ + && device_neighbor_indices_ == filtered_device_indices_; + if (!aliases_candidate && !aliases_filtered) + { + gpu_release_neighbor_list(device_neighbor_count_, device_neighbor_indices_); + } + gpu_release_neighbor_list(candidate_device_count_, candidate_device_indices_); + gpu_release_neighbor_list(filtered_device_count_, filtered_device_indices_); +} + +bool GPU_NeighborList::build_device(int nall, + double cutoff, + const std::vector& position, + int& max_neighbors, + std::string& error) +{ + ModuleBase::timer::start("GPU_NeighborList", "build_device"); + if (nall <= 0 || position.size() != static_cast(3 * nall)) + { + error = "invalid GPU device neighbor-list input"; + ModuleBase::timer::end("GPU_NeighborList", "build_device"); + return false; + } + + char message[512] = {0}; + void* new_count = nullptr; + void* new_indices = nullptr; + const int status = gpu_build_neighbor_list_device(nall, + cutoff, + position.data(), + &new_count, + &new_indices, + &max_neighbors, + message, + sizeof(message)); + if (status != 0) + { + error = message; + ModuleBase::timer::end("GPU_NeighborList", "build_device"); + return false; + } + gpu_release_neighbor_list(candidate_device_count_, candidate_device_indices_); + gpu_release_neighbor_list(filtered_device_count_, filtered_device_indices_); + candidate_device_count_ = new_count; + candidate_device_indices_ = new_indices; + candidate_device_nall_ = nall; + candidate_device_max_neighbors_ = max_neighbors; + filtered_device_count_ = nullptr; + filtered_device_indices_ = nullptr; + filtered_device_nall_ = 0; + filtered_device_max_neighbors_ = 0; + device_neighbor_count_ = candidate_device_count_; + device_neighbor_indices_ = candidate_device_indices_; + device_nall_ = nall; + device_max_neighbors_ = max_neighbors; + ModuleBase::timer::end("GPU_NeighborList", "build_device"); + return true; +} + +bool GPU_NeighborList::filter_device(int nall, + double cutoff, + const std::vector& position, + int candidate_max_neighbors, + int& max_neighbors, + std::string& error) +{ + ModuleBase::timer::start("GPU_NeighborList", "filter_device"); + if (nall <= 0 || position.size() != static_cast(3 * nall) + || candidate_device_nall_ != nall + || candidate_device_count_ == nullptr + || (candidate_max_neighbors > 0 && candidate_device_indices_ == nullptr)) + { + error = "invalid GPU device neighbor-list filter input"; + ModuleBase::timer::end("GPU_NeighborList", "filter_device"); + return false; + } + + char message[512] = {0}; + void* new_count = nullptr; + void* new_indices = nullptr; + const int status = gpu_filter_neighbor_list_device( + nall, + cutoff, + position.data(), + candidate_max_neighbors, + candidate_device_count_, + candidate_device_indices_, + &new_count, + &new_indices, + &max_neighbors, + message, + sizeof(message)); + if (status != 0) + { + error = message; + ModuleBase::timer::end("GPU_NeighborList", "filter_device"); + return false; + } + gpu_release_neighbor_list(filtered_device_count_, filtered_device_indices_); + filtered_device_count_ = new_count; + filtered_device_indices_ = new_indices; + filtered_device_nall_ = nall; + filtered_device_max_neighbors_ = max_neighbors; + device_neighbor_count_ = filtered_device_count_; + device_neighbor_indices_ = filtered_device_indices_; + device_nall_ = nall; + device_max_neighbors_ = max_neighbors; + ModuleBase::timer::end("GPU_NeighborList", "filter_device"); + return true; +} + +void GPU_NeighborList::use_candidate_device() +{ + gpu_release_neighbor_list(filtered_device_count_, filtered_device_indices_); + filtered_device_count_ = nullptr; + filtered_device_indices_ = nullptr; + filtered_device_nall_ = 0; + filtered_device_max_neighbors_ = 0; + device_neighbor_count_ = candidate_device_count_; + device_neighbor_indices_ = candidate_device_indices_; + device_nall_ = candidate_device_nall_; + device_max_neighbors_ = candidate_device_max_neighbors_; +} + +bool GPU_NeighborList::upload(int nall, + int max_neighbors, + const std::vector& neighbor_count, + const std::vector& neighbor_indices, + std::string& error) +{ + ModuleBase::timer::start("GPU_NeighborList", "upload"); + if (nall <= 0 || max_neighbors < 0 + || neighbor_count.size() != static_cast(nall) + || neighbor_indices.size() != static_cast(nall) + * static_cast(max_neighbors)) + { + error = "invalid GPU neighbor-list upload input"; + ModuleBase::timer::end("GPU_NeighborList", "upload"); + return false; + } + + char message[512] = {0}; + void* new_device_count = nullptr; + void* new_device_indices = nullptr; + const int status = gpu_upload_neighbor_list(nall, + max_neighbors, + neighbor_count.data(), + neighbor_indices.empty() ? NULL : neighbor_indices.data(), + &new_device_count, + &new_device_indices, + message, + sizeof(message)); + if (status != 0) + { + error = message; + ModuleBase::timer::end("GPU_NeighborList", "upload"); + return false; + } + gpu_release_neighbor_list(device_neighbor_count_, device_neighbor_indices_); + device_neighbor_count_ = new_device_count; + device_neighbor_indices_ = new_device_indices; + device_nall_ = nall; + device_max_neighbors_ = max_neighbors; + ModuleBase::timer::end("GPU_NeighborList", "upload"); + return true; +} + +const void* GPU_NeighborList::device_neighbor_count() const +{ + return device_neighbor_count_; +} + +const void* GPU_NeighborList::device_neighbor_indices() const +{ + return device_neighbor_indices_; +} + +int GPU_NeighborList::device_nall() const +{ + return device_nall_; +} + +int GPU_NeighborList::device_max_neighbors() const +{ + return device_max_neighbors_; +} + +bool GPU_NeighborList::build(int nall, + double cutoff, + const std::vector& position, + std::vector& neighbor_count, + std::vector& neighbor_indices, + int& max_neighbors, + std::string& error) const +{ + ModuleBase::timer::start("GPU_NeighborList", "build"); + if (nall <= 0 || cutoff <= 0.0 + || position.size() != static_cast(3 * nall)) + { + error = "invalid GPU neighbor-list input"; + ModuleBase::timer::end("GPU_NeighborList", "build"); + return false; + } + + neighbor_count.assign(static_cast(nall), 0); + neighbor_indices.clear(); + max_neighbors = 0; + char message[512] = {0}; + const int status = gpu_build_neighbor_list(nall, + cutoff, + position.data(), + neighbor_count.data(), + NULL, + &max_neighbors, + message, + sizeof(message)); + if (status != 0) + { + error = message; + ModuleBase::timer::end("GPU_NeighborList", "build"); + return false; + } + + neighbor_indices.assign(static_cast(nall) + * static_cast(max_neighbors), + 0); + if (max_neighbors > 0) + { + const int fill_status = gpu_build_neighbor_list(nall, + cutoff, + position.data(), + neighbor_count.data(), + neighbor_indices.data(), + &max_neighbors, + message, + sizeof(message)); + if (fill_status != 0) + { + error = message; + ModuleBase::timer::end("GPU_NeighborList", "build"); + return false; + } + } + ModuleBase::timer::end("GPU_NeighborList", "build"); + return true; +} + +bool GPU_NeighborList::filter(int nall, + double cutoff, + const std::vector& position, + const std::vector& candidate_count, + const std::vector& candidate_indices, + int candidate_max_neighbors, + std::vector& neighbor_count, + std::vector& neighbor_indices, + int& max_neighbors, + std::string& error) const +{ + ModuleBase::timer::start("GPU_NeighborList", "filter"); + if (nall <= 0 || cutoff <= 0.0 + || position.size() != static_cast(3 * nall) + || candidate_count.size() != static_cast(nall) + || candidate_max_neighbors < 0 + || candidate_indices.size() != static_cast(nall) + * static_cast(candidate_max_neighbors)) + { + error = "invalid GPU neighbor-list filter input"; + ModuleBase::timer::end("GPU_NeighborList", "filter"); + return false; + } + + neighbor_count.assign(static_cast(nall), 0); + neighbor_indices.assign(candidate_indices.size(), 0); + max_neighbors = 0; + char message[512] = {0}; + const int status = gpu_filter_neighbor_list(nall, + cutoff, + position.data(), + candidate_count.data(), + candidate_indices.data(), + candidate_max_neighbors, + neighbor_count.data(), + neighbor_indices.empty() ? NULL : neighbor_indices.data(), + &max_neighbors, + message, + sizeof(message)); + if (status != 0) + { + error = message; + ModuleBase::timer::end("GPU_NeighborList", "filter"); + return false; + } + neighbor_indices.resize(static_cast(nall) + * static_cast(max_neighbors)); + ModuleBase::timer::end("GPU_NeighborList", "filter"); + return true; +} + +} // namespace ModuleNeighList diff --git a/source/source_cell/module_neighlist/gpu_neighbor_list.h b/source/source_cell/module_neighlist/gpu_neighbor_list.h new file mode 100644 index 00000000000..7ebf5f77b8b --- /dev/null +++ b/source/source_cell/module_neighlist/gpu_neighbor_list.h @@ -0,0 +1,81 @@ +#ifndef SOURCE_CELL_MODULE_NEIGHLIST_GPU_NEIGHBOR_LIST_H +#define SOURCE_CELL_MODULE_NEIGHLIST_GPU_NEIGHBOR_LIST_H + +#include +#include + +namespace ModuleNeighList +{ + +class GPU_NeighborList +{ + public: + GPU_NeighborList() = default; + ~GPU_NeighborList(); + + GPU_NeighborList(const GPU_NeighborList&) = delete; + GPU_NeighborList& operator=(const GPU_NeighborList&) = delete; + + bool build(int nall, + double cutoff, + const std::vector& position, + std::vector& neighbor_count, + std::vector& neighbor_indices, + int& max_neighbors, + std::string& error) const; + + bool filter(int nall, + double cutoff, + const std::vector& position, + const std::vector& candidate_count, + const std::vector& candidate_indices, + int candidate_max_neighbors, + std::vector& neighbor_count, + std::vector& neighbor_indices, + int& max_neighbors, + std::string& error) const; + + bool build_device(int nall, + double cutoff, + const std::vector& position, + int& max_neighbors, + std::string& error); + + bool filter_device(int nall, + double cutoff, + const std::vector& position, + int candidate_max_neighbors, + int& max_neighbors, + std::string& error); + + void use_candidate_device(); + + bool upload(int nall, + int max_neighbors, + const std::vector& neighbor_count, + const std::vector& neighbor_indices, + std::string& error); + + const void* device_neighbor_count() const; + const void* device_neighbor_indices() const; + int device_nall() const; + int device_max_neighbors() const; + + private: + void* candidate_device_count_ = nullptr; + void* candidate_device_indices_ = nullptr; + int candidate_device_nall_ = 0; + int candidate_device_max_neighbors_ = 0; + void* filtered_device_count_ = nullptr; + void* filtered_device_indices_ = nullptr; + int filtered_device_nall_ = 0; + int filtered_device_max_neighbors_ = 0; + void* device_neighbor_count_ = nullptr; + void* device_neighbor_indices_ = nullptr; + int device_nall_ = 0; + int device_max_neighbors_ = 0; +}; + +} // namespace ModuleNeighList + +#endif diff --git a/source/source_cell/module_neighlist/test/CMakeLists.txt b/source/source_cell/module_neighlist/test/CMakeLists.txt index 234100a5ef9..80270b213d3 100644 --- a/source/source_cell/module_neighlist/test/CMakeLists.txt +++ b/source/source_cell/module_neighlist/test/CMakeLists.txt @@ -8,11 +8,11 @@ abacus_disable_feature_definitions(__EXX) AddTest( TARGET MODULE_CELL_NEIGHBOR_neighbor_search - LIBS parameter base device cell symmetry + LIBS parameter base device cell neighbor symmetry SOURCES - neighbor_search_test.cpp + test_neighbor_search.cpp ../neighbor_search.cpp - ../domain_decomposition.cpp + ../../domain_decomposition.cpp ../bin_manager.cpp ../page_allocator.cpp ) @@ -21,7 +21,7 @@ AddTest( TARGET MODULE_CELL_NEIGHBOR_bin_manager LIBS parameter base device SOURCES - bin_manager_test.cpp + test_bin_manager.cpp ../bin_manager.cpp ../page_allocator.cpp ) @@ -30,16 +30,35 @@ AddTest( TARGET MODULE_CELL_NEIGHBOR_allocator_and_list LIBS parameter base device SOURCES - neighbor_list_test.cpp + test_neighbor_list.cpp ../page_allocator.cpp ) +AddTest( + TARGET MODULE_CELL_NEIGHBOR_page_allocator + LIBS parameter base device + SOURCES + test_page_allocator.cpp + ../page_allocator.cpp +) + +if(USE_CUDA AND ENABLE_GPU_NEIGHBOR_LIST) + AddTest( + TARGET MODULE_CELL_NEIGHBOR_gpu_neighbor_list + LIBS parameter base device + SOURCES + test_gpu_neighbor_list.cpp + ../gpu_neighbor_list.cpp + ) + target_compile_definitions(MODULE_CELL_NEIGHBOR_gpu_neighbor_list PRIVATE __CUDA) +endif() + if(ENABLE_MPI) add_executable(MODULE_CELL_NEIGHBOR_mdcell_migrate_mpi - mdcell_migrate_mpi_test.cpp + test_mdcell_migrate_mpi.cpp test_domain_decomposition.cpp ../../mdcell.cpp - ../domain_decomposition.cpp + ../../domain_decomposition.cpp ../neighbor_search.cpp ../bin_manager.cpp ../page_allocator.cpp @@ -72,11 +91,11 @@ if(ENABLE_MPI) PROPERTIES ENVIRONMENT "OMP_NUM_THREADS=1" TIMEOUT 60) add_executable(MODULE_CELL_NEIGHBOR_mdcell_reader - mdcell_reader_test.cpp + test_mdcell_reader.cpp ../../mdcell_reader.cpp ../../mdcell.cpp ../../print_cell.cpp - ../domain_decomposition.cpp + ../../domain_decomposition.cpp ../neighbor_search.cpp ../bin_manager.cpp ../page_allocator.cpp diff --git a/source/source_cell/module_neighlist/test/mdcell_migrate_mpi_test.cpp b/source/source_cell/module_neighlist/test/mdcell_migrate_mpi_test.cpp deleted file mode 100644 index 817968d84b9..00000000000 --- a/source/source_cell/module_neighlist/test/mdcell_migrate_mpi_test.cpp +++ /dev/null @@ -1,228 +0,0 @@ -#include "source_cell/module_neighlist/domain_decomposition.h" -#include - -#include "source_cell/mdcell.h" -#include "source_base/parallel_cell.h" - -#include - -#include -#include -#include -#include - -namespace -{ -void ensure_mpi_initialized() -{ - int initialized = 0; - MPI_Initialized(&initialized); - if (!initialized) - { - int provided = 0; - MPI_Init_thread(NULL, NULL, MPI_THREAD_SINGLE, &provided); - } -} - -ModuleBase::Matrix3 make_lattice() -{ - ModuleBase::Matrix3 latvec; - latvec.e11 = 1.0; - latvec.e22 = 1.0; - latvec.e33 = 1.0; - return latvec; -} -} - -TEST(MdCellMigrateMpiTest, AtomCrossingDomainMigratesToNewOwner) -{ - int rank = 0; - int size = 1; - MPI_Comm_rank(MPI_COMM_WORLD, &rank); - MPI_Comm_size(MPI_COMM_WORLD, &size); - ASSERT_GE(size, 2); - - const ModuleBase::Matrix3 latvec = make_lattice(); - std::vector owned_atoms; - if (rank < 2) - { - const ModuleBase::Vector3 frac(rank == 0 ? 0.2 : 0.7, 0.2, 0.2); - owned_atoms.push_back(LocalAtom(frac, - frac, - ModuleBase::Vector3(0.0, 0.0, 0.0), - ModuleBase::Vector3(0.0, 0.0, 0.0), - ModuleBase::Vector3(1, 1, 1), - 1.0, - 0, - rank, - rank)); - } - MDCell mdcell; - DomainDecomposition decomp; - decomp.init(ModuleBase::world_comm_domain(), latvec, 1.0, 0.0, 0.0); - mdcell.initialize_from_owned_atoms(latvec, - latvec.Inverse(), - 1.0, - 1.0, - 2, - owned_atoms, - std::vector(1, "X"), - std::vector(1, 1.0), - std::vector(1, 2), - 0.0, - ModuleBase::world_comm_domain()); - mdcell.set_neighbor_cutoff(0.1); - decomp.migrate_owned_atoms(mdcell); - - ASSERT_EQ(mdcell.mpi_size(), size); - if (size == 2) - { - ASSERT_EQ(mdcell.owned_atoms().size(), 1); - mdcell.owned_atoms()[0].vel.x = static_cast(rank + 1); - mdcell.owned_atoms()[0].force.y = static_cast(rank + 3); - decomp.migrate_owned_atoms(mdcell); - ASSERT_EQ(mdcell.owned_atoms().size(), 1); - EXPECT_EQ(mdcell.owned_atoms()[0].owner_rank, rank); - EXPECT_EQ(mdcell.owned_atoms()[0].vel.x, static_cast(rank + 1)); - EXPECT_EQ(mdcell.owned_atoms()[0].force.y, static_cast(rank + 3)); - - if (rank == 0 && mdcell.owned_atoms().size() == 1) - { - mdcell.owned_atoms()[0].cart.x = 0.8; - } - if (rank == 1 && mdcell.owned_atoms().size() == 1) - { - mdcell.owned_atoms()[0].cart.x = 0.3; - } - decomp.migrate_owned_atoms(mdcell); - - long long local_count = mdcell.owned_atoms().size(); - long long global_count = 0; - MPI_Allreduce(&local_count, &global_count, 1, MPI_LONG_LONG, MPI_SUM, MPI_COMM_WORLD); - EXPECT_EQ(global_count, 2); - - for (int i = 0; i < mdcell.owned_atoms().size(); ++i) - { - EXPECT_EQ(mdcell.owned_atoms()[static_cast(i)].owner_rank, rank); - } - } -} - -TEST(MdCellMigrateMpiTest, GhostForcesReturnToOwners) -{ - int rank = 0; - int size = 1; - MPI_Comm_rank(MPI_COMM_WORLD, &rank); - MPI_Comm_size(MPI_COMM_WORLD, &size); - ASSERT_EQ(size, 2); - - const ModuleBase::Matrix3 latvec = make_lattice(); - std::vector owned_atoms; - const ModuleBase::Vector3 frac(rank == 0 ? 0.2 : 0.7, 0.2, 0.2); - owned_atoms.push_back(LocalAtom(frac, - frac, - ModuleBase::Vector3(0.0, 0.0, 0.0), - ModuleBase::Vector3(0.0, 0.0, 0.0), - ModuleBase::Vector3(1, 1, 1), - 1.0, - 0, - rank, - rank)); - MDCell mdcell; - DomainDecomposition decomp; - decomp.init(ModuleBase::world_comm_domain(), latvec, 1.0, 0.0, 0.0); - mdcell.initialize_from_owned_atoms(latvec, - latvec.Inverse(), - 1.0, - 1.0, - 2, - owned_atoms, - std::vector(1, "X"), - std::vector(1, 1.0), - std::vector(1, 2), - 0.0, - ModuleBase::world_comm_domain()); - mdcell.set_neighbor_cutoff(0.6); - decomp.migrate_owned_atoms(mdcell); - - long long local_copies[2] = {0, 0}; - for (std::size_t iat = 0; iat < mdcell.ghost_atoms().size(); ++iat) - { - ++local_copies[mdcell.ghost_atoms()[iat].owner_rank]; - } - long long global_copies[2] = {0, 0}; - MPI_Allreduce(local_copies, global_copies, 2, MPI_LONG_LONG, MPI_SUM, MPI_COMM_WORLD); - - for (std::size_t iat = 0; iat < mdcell.ghost_atoms().size(); ++iat) - { - LocalAtom& ghost = mdcell.ghost_atoms()[iat]; - const double value = static_cast(ghost.owner_rank + 1); - ghost.force.set(value, 2.0 * value, 3.0 * value); - } - decomp.accumulate_ghost_forces(mdcell); - - ASSERT_EQ(mdcell.owned_atoms().size(), 1); - const double expected = static_cast(global_copies[rank] * (rank + 1)); - EXPECT_DOUBLE_EQ(mdcell.owned_atoms()[0].force.x, expected); - EXPECT_DOUBLE_EQ(mdcell.owned_atoms()[0].force.y, 2.0 * expected); - EXPECT_DOUBLE_EQ(mdcell.owned_atoms()[0].force.z, 3.0 * expected); -} - -TEST(MdCellMigrateMpiTest, SkinUpdatesFixedGhostLayoutBeforeRebuild) -{ - int rank = 0; - int size = 1; - MPI_Comm_rank(MPI_COMM_WORLD, &rank); - MPI_Comm_size(MPI_COMM_WORLD, &size); - ASSERT_EQ(size, 2); - - const ModuleBase::Matrix3 latvec = make_lattice(); - const ModuleBase::Vector3 frac(rank == 0 ? 0.20 : 0.70, 0.20, 0.20); - std::vector owned_atoms(1, LocalAtom(frac, - frac, - ModuleBase::Vector3(0.0, 0.0, 0.0), - ModuleBase::Vector3(0.0, 0.0, 0.0), - ModuleBase::Vector3(1, 1, 1), - 1.0, - 0, - rank, - rank)); - MDCell mdcell; - DomainDecomposition decomp; - decomp.init(ModuleBase::world_comm_domain(), latvec, 1.0, 0.0, 0.0); - mdcell.initialize_from_owned_atoms(latvec, - latvec.Inverse(), - 1.0, - 1.0, - 2, - owned_atoms, - std::vector(1, "X"), - std::vector(1, 1.0), - std::vector(1, 2), - 0.2, - ModuleBase::world_comm_domain()); - mdcell.set_neighbor_cutoff(0.1); - decomp.migrate_owned_atoms(mdcell); - - decomp.prepare_neighbors(mdcell); - mdcell.owned_atoms()[0].frac.x += rank == 0 ? 0.05 : -0.05; - mdcell.owned_atoms()[0].cart = mdcell.owned_atoms()[0].frac * latvec; - decomp.prepare_neighbors(mdcell); - - ASSERT_EQ(mdcell.owned_atoms().size(), 1); - for (std::size_t i = 0; i < mdcell.ghost_atoms().size(); ++i) - { - const LocalAtom& ghost = mdcell.ghost_atoms()[i]; - const double expected_frac = ghost.owner_rank == 0 ? 0.25 : 0.65; - EXPECT_DOUBLE_EQ(ghost.frac.x, expected_frac); - } -} - -int main(int argc, char** argv) -{ - MPI_Init(&argc, &argv); - ::testing::InitGoogleTest(&argc, argv); - const int result = RUN_ALL_TESTS(); - MPI_Finalize(); - return result; -} diff --git a/source/source_cell/module_neighlist/test/mdcell_reader_test.cpp b/source/source_cell/module_neighlist/test/mdcell_reader_test.cpp deleted file mode 100644 index 0911d8a4709..00000000000 --- a/source/source_cell/module_neighlist/test/mdcell_reader_test.cpp +++ /dev/null @@ -1,331 +0,0 @@ -#include "source_cell/module_neighlist/domain_decomposition.h" -#include - -#include "source_cell/mdcell_reader.h" -#include "source_cell/mdcell.h" -#include "source_cell/print_cell.h" -#include "source_base/constants.h" -#include "source_base/parallel_cell.h" -#include "source_base/global_variable.h" - -#include -#include -#include -#include -#include -#include -#include -#include - -static_assert(!std::is_copy_constructible::value, "MDCell must not be copy constructible."); -static_assert(!std::is_copy_assignable::value, "MDCell must not be copy assignable."); -static_assert(std::is_default_constructible::value, "MDCell must be default constructible."); -static_assert(std::is_move_constructible::value, "MDCell must be move constructible."); - -namespace -{ -void write_cartesian_stru_case(const std::string& stru_file) -{ - std::ofstream ofs(stru_file.c_str()); - ofs << "ATOMIC_SPECIES\n"; - ofs << "He 4.0026 auto auto\n\n"; - ofs << "LATTICE_CONSTANT\n"; - ofs << "1.0\n\n"; - ofs << "LATTICE_VECTORS\n"; - ofs << "4.0 0.0 0.0\n"; - ofs << "0.0 4.0 0.0\n"; - ofs << "0.0 0.0 4.0\n\n"; - ofs << "ATOMIC_POSITIONS\n"; - ofs << "Cartesian\n\n"; - ofs << "He\n"; - ofs << "0.0\n"; - ofs << "4\n"; - ofs << "0.40 0.40 0.40 m 1 1 1 v 0.01 0.00 0.00\n"; - ofs << "2.40 0.40 0.40 m 1 0 1 v 0.02 0.00 0.00\n"; - ofs << "0.40 2.40 0.40 m 0 1 1 v 0.03 0.00 0.00\n"; - ofs << "2.40 2.40 0.40 m 1 1 0 v 0.04 0.00 0.00\n"; -} - -ModuleBase::Matrix3 make_lattice() -{ - ModuleBase::Matrix3 latvec; - latvec.e11 = 4.0; - latvec.e12 = 0.0; - latvec.e13 = 0.0; - latvec.e21 = 0.0; - latvec.e22 = 4.0; - latvec.e23 = 0.0; - latvec.e31 = 0.0; - latvec.e32 = 0.0; - latvec.e33 = 4.0; - return latvec; -} -} // namespace - -TEST(MDCellReaderTest, ReadOwnedAtomsFromSTRUWithoutUnitCell) -{ - int world_rank = 0; - MPI_Comm_rank(MPI_COMM_WORLD, &world_rank); - - const std::string stru_file = "mdcell_reader_cartesian.STRU"; - if (world_rank == 0) - { - write_cartesian_stru_case(stru_file); - } - MPI_Barrier(MPI_COMM_WORLD); - - MPI_Comm md_comm = MPI_COMM_NULL; - MPI_Comm_split(MPI_COMM_WORLD, world_rank % 2, world_rank, &md_comm); - ModuleBase::CommunicationDomain comm_domain; - comm_domain.initialize(md_comm); - - DomainDecomposition reader_decomp; - MDCell mdcell = MDCellReader::read_stru(stru_file, - std::vector{1, 1, 1}, - 0.0, - comm_domain, reader_decomp); - - EXPECT_EQ(mdcell.type_labels().size(), 1U); - EXPECT_EQ(mdcell.type_labels()[0], "He"); - ASSERT_EQ(mdcell.type_masses().size(), 1U); - EXPECT_DOUBLE_EQ(mdcell.type_masses()[0], 4.0026); - ASSERT_EQ(mdcell.type_atom_counts().size(), 1U); - EXPECT_EQ(mdcell.type_atom_counts()[0], 4); - ASSERT_EQ(mdcell.stru_meta().species.size(), 1U); - EXPECT_EQ(mdcell.stru_meta().species[0].pseudo_file, "auto"); - EXPECT_EQ(mdcell.stru_meta().species[0].pseudo_type, "auto"); - EXPECT_EQ(mdcell.nat(), 4); - - DomainDecomposition decomp; - decomp.init(comm_domain, make_lattice(), 1.0, 1.0 * ModuleBase::ANGSTROM_AU, 0.0); - - long long local_count = static_cast(mdcell.owned_atoms().size()); - long long global_count = 0; - MPI_Allreduce(&local_count, &global_count, 1, MPI_LONG_LONG, MPI_SUM, md_comm); - EXPECT_EQ(global_count, 4); - - std::set > local_ids; - for (std::size_t iat = 0; iat < mdcell.owned_atoms().size(); ++iat) - { - const LocalAtom& atom = mdcell.owned_atoms()[iat]; - EXPECT_EQ(decomp.owner_rank_from_frac(atom.frac), comm_domain.rank()); - local_ids.insert(std::make_pair(atom.type, atom.type_index)); - EXPECT_GE(atom.type, 0); - EXPECT_DOUBLE_EQ(atom.force.x, 0.0); - EXPECT_DOUBLE_EQ(atom.force.y, 0.0); - EXPECT_DOUBLE_EQ(atom.force.z, 0.0); - } - EXPECT_EQ(local_ids.size(), mdcell.owned_atoms().size()); - - bool saw_v01 = false; - bool saw_v04 = false; - for (std::size_t iat = 0; iat < mdcell.owned_atoms().size(); ++iat) - { - const LocalAtom& atom = mdcell.owned_atoms()[iat]; - if (atom.type == 0 && atom.type_index == 0) - { - saw_v01 = true; - EXPECT_DOUBLE_EQ(atom.vel.x, 0.01); - EXPECT_EQ(atom.mbl.x, 1); - EXPECT_EQ(atom.mbl.y, 1); - EXPECT_EQ(atom.mbl.z, 1); - EXPECT_DOUBLE_EQ(atom.mass, 4.0026 / ModuleBase::AU_to_MASS); - } - if (atom.type == 0 && atom.type_index == 3) - { - saw_v04 = true; - EXPECT_DOUBLE_EQ(atom.vel.x, 0.04); - EXPECT_EQ(atom.mbl.x, 1); - EXPECT_EQ(atom.mbl.y, 1); - EXPECT_EQ(atom.mbl.z, 0); - } - } - - const int saw_flags[2] = {saw_v01 ? 1 : 0, saw_v04 ? 1 : 0}; - int reduced_flags[2] = {0, 0}; - MPI_Allreduce(saw_flags, reduced_flags, 2, MPI_INT, MPI_MAX, md_comm); - EXPECT_EQ(reduced_flags[0], 1); - EXPECT_EQ(reduced_flags[1], 1); - - MPI_Comm_free(&md_comm); -} - -TEST(MDCellReaderTest, RestartStruPreservesAtomRecordsAcrossRanks) -{ - int rank = 0; - int size = 1; - MPI_Comm_rank(MPI_COMM_WORLD, &rank); - MPI_Comm_size(MPI_COMM_WORLD, &size); - ASSERT_GE(size, 4); - - std::vector owned_atoms; - if (rank == 0) - { - owned_atoms.push_back(LocalAtom(ModuleBase::Vector3(11.0, 0.0, 0.0), - ModuleBase::Vector3(0.0, 0.0, 0.0), - ModuleBase::Vector3(0.11, 0.0, 0.0), - ModuleBase::Vector3(0.0, 0.0, 0.0), - ModuleBase::Vector3(1, 0, 1), - 1.0, - 1, - 1, - rank)); - } - if (rank == 1) - { - owned_atoms.push_back(LocalAtom(ModuleBase::Vector3(1.0, 0.0, 0.0), - ModuleBase::Vector3(0.0, 0.0, 0.0), - ModuleBase::Vector3(0.01, 0.0, 0.0), - ModuleBase::Vector3(0.0, 0.0, 0.0), - ModuleBase::Vector3(0, 1, 1), - 1.0, - 0, - 1, - rank)); - } - if (rank == 2) - { - owned_atoms.push_back(LocalAtom(ModuleBase::Vector3(10.0, 0.0, 0.0), - ModuleBase::Vector3(0.0, 0.0, 0.0), - ModuleBase::Vector3(0.10, 0.0, 0.0), - ModuleBase::Vector3(0.0, 0.0, 0.0), - ModuleBase::Vector3(1, 1, 0), - 1.0, - 1, - 0, - rank)); - } - if (rank == 3) - { - owned_atoms.push_back(LocalAtom(ModuleBase::Vector3(0.0, 0.0, 0.0), - ModuleBase::Vector3(0.0, 0.0, 0.0), - ModuleBase::Vector3(0.0, 0.0, 0.0), - ModuleBase::Vector3(0.0, 0.0, 0.0), - ModuleBase::Vector3(0, 0, 1), - 1.0, - 0, - 0, - rank)); - } - - ModuleBase::Matrix3 lattice = make_lattice(); - lattice.e11 = 20.0; - lattice.e22 = 20.0; - lattice.e33 = 20.0; - MDCell mdcell; - mdcell.initialize_from_owned_atoms(lattice, - lattice.Inverse(), - 1.0, - 1.0, - 4, - owned_atoms, - std::vector{"A", "B"}, - std::vector{1.0, 1.0}, - std::vector{2, 2}, - 0.0, - ModuleBase::world_comm_domain()); - StruMeta metadata; - metadata.species.resize(2); - const std::string output_file = "distributed_mdcell_restart.STRU"; - mdcell::print_stru_file(mdcell, metadata, output_file); - - DomainDecomposition reader_decomp; - MDCell round_trip = MDCellReader::read_stru(output_file, - std::vector{1, 1, 1}, - 0.0, - ModuleBase::world_comm_domain(), reader_decomp); - double local_positions[4] = {0.0, 0.0, 0.0, 0.0}; - double local_velocities[4] = {0.0, 0.0, 0.0, 0.0}; - int local_mbl_x[4] = {0, 0, 0, 0}; - int local_owners[4] = {0, 0, 0, 0}; - for (std::size_t iat = 0; iat < round_trip.owned_atoms().size(); ++iat) - { - const LocalAtom& atom = round_trip.owned_atoms()[iat]; - const int index = atom.type == 0 ? static_cast(atom.cart.x) - : 2 + static_cast(atom.cart.x) - 10; - local_positions[index] = atom.cart.x; - local_velocities[index] = atom.vel.x; - local_mbl_x[index] = atom.mbl.x; - local_owners[index] = 1; - } - double global_positions[4] = {0.0, 0.0, 0.0, 0.0}; - double global_velocities[4] = {0.0, 0.0, 0.0, 0.0}; - int global_mbl_x[4] = {0, 0, 0, 0}; - int global_owners[4] = {0, 0, 0, 0}; - MPI_Allreduce(local_positions, global_positions, 4, MPI_DOUBLE, MPI_SUM, MPI_COMM_WORLD); - MPI_Allreduce(local_velocities, global_velocities, 4, MPI_DOUBLE, MPI_SUM, MPI_COMM_WORLD); - MPI_Allreduce(local_mbl_x, global_mbl_x, 4, MPI_INT, MPI_SUM, MPI_COMM_WORLD); - MPI_Allreduce(local_owners, global_owners, 4, MPI_INT, MPI_SUM, MPI_COMM_WORLD); - for (int iat = 0; iat < 4; ++iat) EXPECT_EQ(global_owners[iat], 1); - EXPECT_DOUBLE_EQ(global_positions[0], 0.0); - EXPECT_DOUBLE_EQ(global_positions[1], 1.0); - EXPECT_DOUBLE_EQ(global_positions[2], 10.0); - EXPECT_DOUBLE_EQ(global_positions[3], 11.0); - EXPECT_DOUBLE_EQ(global_velocities[0], 0.0); - EXPECT_DOUBLE_EQ(global_velocities[1], 0.01); - EXPECT_DOUBLE_EQ(global_velocities[2], 0.10); - EXPECT_DOUBLE_EQ(global_velocities[3], 0.11); - EXPECT_EQ(global_mbl_x[0], 0); - EXPECT_EQ(global_mbl_x[1], 0); - EXPECT_EQ(global_mbl_x[2], 1); - EXPECT_EQ(global_mbl_x[3], 1); - - if (rank == 0) - { - std::ifstream input(output_file.c_str()); - ASSERT_TRUE(input.good()); - std::vector coordinates_a; - std::vector coordinates_b; - std::string line; - int current_type = -1; - int atoms_remaining = 0; - while (std::getline(input, line)) - { - if (line == "A #label") - { - current_type = 0; - continue; - } - if (line == "B #label") - { - current_type = 1; - continue; - } - if (current_type >= 0 && line.find("#number of atoms") != std::string::npos) - { - std::istringstream count_stream(line); - count_stream >> atoms_remaining; - continue; - } - if (current_type < 0 || atoms_remaining == 0) continue; - std::istringstream values(line); - double x = 0.0; - values >> x; - --atoms_remaining; - if (values) - { - if (current_type == 0) coordinates_a.push_back(x); - else coordinates_b.push_back(x); - } - } - ASSERT_EQ(coordinates_a.size(), 2U); - ASSERT_EQ(coordinates_b.size(), 2U); - std::set expected_a = {0.0, 1.0}; - std::set expected_b = {10.0, 11.0}; - EXPECT_EQ(std::set(coordinates_a.begin(), coordinates_a.end()), expected_a); - EXPECT_EQ(std::set(coordinates_b.begin(), coordinates_b.end()), expected_b); - } - MPI_Barrier(MPI_COMM_WORLD); - if (rank == 0) std::remove(output_file.c_str()); -} - -int main(int argc, char** argv) -{ - MPI_Init(&argc, &argv); - MPI_Comm_rank(MPI_COMM_WORLD, &GlobalV::MY_RANK); - MPI_Comm_size(MPI_COMM_WORLD, &GlobalV::NPROC); - ::testing::InitGoogleTest(&argc, argv); - const int result = RUN_ALL_TESTS(); - MPI_Finalize(); - return result; -} diff --git a/source/source_cell/module_neighlist/test/bin_manager_test.cpp b/source/source_cell/module_neighlist/test/test_bin_manager.cpp similarity index 100% rename from source/source_cell/module_neighlist/test/bin_manager_test.cpp rename to source/source_cell/module_neighlist/test/test_bin_manager.cpp diff --git a/source/source_cell/module_neighlist/test/test_domain_decomposition.cpp b/source/source_cell/module_neighlist/test/test_domain_decomposition.cpp index fe8de476c5e..e7ba6e1ec49 100644 --- a/source/source_cell/module_neighlist/test/test_domain_decomposition.cpp +++ b/source/source_cell/module_neighlist/test/test_domain_decomposition.cpp @@ -1,4 +1,4 @@ -#include "source_cell/module_neighlist/domain_decomposition.h" +#include "source_cell/domain_decomposition.h" #include "source_cell/module_neighlist/neighbor_search.h" #include "source_cell/mdcell.h" #include "source_base/parallel_cell.h" @@ -234,6 +234,91 @@ TEST_F(DomainDecompositionTest, CrossingHaloEdgePreservesCachedGhostSlots) } } +TEST_F(DomainDecompositionTest, PeriodicGhostImageRebuildsOnBoundaryCrossing) +{ + if (domain.size() != 2) + { + return; + } + for (LocalAtom& atom : cell.owned_atoms()) + { + if (atom.type_index == 1) + { + atom.frac.x = 0.99; + atom.cart = atom.frac * cell.latvec(); + } + } + decomp.prepare_neighbors(cell); + const NeighborSearch* search = &cell.neighbor_search(); + for (LocalAtom& atom : cell.owned_atoms()) + { + if (atom.type_index == 1) + { + atom.frac.x = 0.01; + atom.cart = atom.frac * cell.latvec(); + } + } + decomp.prepare_neighbors(cell); + EXPECT_NE(&cell.neighbor_search(), search); + + long long local_type_one = 0; + for (const LocalAtom& atom : cell.owned_atoms()) + { + if (atom.type_index == 1) + { + EXPECT_NEAR(atom.cart.x, 0.04, 1.0e-12); + ++local_type_one; + } + } + Parallel_Reduce::reduce_all(local_type_one); + EXPECT_EQ(local_type_one, 1); +} + +TEST_F(DomainDecompositionTest, PeriodicGhostImageUpdatesAcrossRepeatedCrossings) +{ + if (domain.size() != 1) + { + return; + } + for (LocalAtom& atom : cell.owned_atoms()) + { + if (atom.type_index == 1) + { + atom.frac.x = 0.99; + atom.cart = atom.frac * cell.latvec(); + } + } + decomp.prepare_neighbors(cell); + const NeighborSearch* previous_search = &cell.neighbor_search(); + + const double positions[4] = {0.01, 0.99, 0.01, 0.99}; + for (double position : positions) + { + for (LocalAtom& atom : cell.owned_atoms()) + { + if (atom.type_index == 1) + { + atom.frac.x = position; + atom.cart = atom.frac * cell.latvec(); + } + } + decomp.prepare_neighbors(cell); + EXPECT_NE(&cell.neighbor_search(), previous_search); + previous_search = &cell.neighbor_search(); + + bool found_owned_atom = false; + for (const LocalAtom& atom : cell.owned_atoms()) + { + if (atom.type_index == 1) + { + EXPECT_NEAR(atom.cart.x, position * 4.0, 1.0e-12); + found_owned_atom = true; + } + } + EXPECT_TRUE(found_owned_atom); + } +} + TEST_F(DomainDecompositionTest, SkewCellMultipleImagesMatchBruteForce) { ModuleBase::Matrix3 lattice = cell.latvec(); diff --git a/source/source_cell/module_neighlist/test/test_gpu_neighbor_list.cpp b/source/source_cell/module_neighlist/test/test_gpu_neighbor_list.cpp new file mode 100644 index 00000000000..6203067c0fd --- /dev/null +++ b/source/source_cell/module_neighlist/test/test_gpu_neighbor_list.cpp @@ -0,0 +1,75 @@ +#include + +#include "source_cell/module_neighlist/gpu_neighbor_list.h" + +#include +#include + +using ModuleNeighList::GPU_NeighborList; + +TEST(GPUNeighborListTest, RejectsInvalidBuildInput) +{ + GPU_NeighborList neighbor_list; + std::vector neighbor_count; + std::vector neighbor_indices; + std::string error; + int max_neighbors = -1; + + EXPECT_FALSE(neighbor_list.build(0, + 1.0, + std::vector(), + neighbor_count, + neighbor_indices, + max_neighbors, + error)); + EXPECT_EQ(error, "invalid GPU neighbor-list input"); + EXPECT_TRUE(neighbor_count.empty()); + EXPECT_TRUE(neighbor_indices.empty()); +} + +TEST(GPUNeighborListTest, RejectsInvalidFilterInput) +{ + GPU_NeighborList neighbor_list; + std::vector neighbor_count; + std::vector neighbor_indices; + std::string error; + int max_neighbors = -1; + const std::vector position(3, 0.0); + const std::vector candidate_count(1, 0); + + EXPECT_FALSE(neighbor_list.filter(1, + 1.0, + position, + candidate_count, + std::vector(), + 1, + neighbor_count, + neighbor_indices, + max_neighbors, + error)); + EXPECT_EQ(error, "invalid GPU neighbor-list filter input"); +} + +TEST(GPUNeighborListTest, RejectsInvalidDeviceInput) +{ + GPU_NeighborList neighbor_list; + std::string error; + int max_neighbors = -1; + + EXPECT_FALSE(neighbor_list.build_device(0, 1.0, std::vector(), max_neighbors, error)); + EXPECT_FALSE(neighbor_list.filter_device(0, 1.0, std::vector(), 0, max_neighbors, error)); + EXPECT_EQ(neighbor_list.device_nall(), 0); + EXPECT_EQ(neighbor_list.device_max_neighbors(), 0); + EXPECT_EQ(neighbor_list.device_neighbor_count(), nullptr); + EXPECT_EQ(neighbor_list.device_neighbor_indices(), nullptr); +} + +TEST(GPUNeighborListTest, RejectsInvalidUploadInput) +{ + GPU_NeighborList neighbor_list; + std::string error; + + EXPECT_FALSE(neighbor_list.upload(1, 1, std::vector(), std::vector(), error)); + EXPECT_EQ(neighbor_list.device_nall(), 0); + EXPECT_EQ(neighbor_list.device_max_neighbors(), 0); +} diff --git a/source/source_cell/module_neighlist/test/test_mdcell_migrate_mpi.cpp b/source/source_cell/module_neighlist/test/test_mdcell_migrate_mpi.cpp new file mode 100644 index 00000000000..865ef7f8254 --- /dev/null +++ b/source/source_cell/module_neighlist/test/test_mdcell_migrate_mpi.cpp @@ -0,0 +1,228 @@ +#include "source_cell/domain_decomposition.h" +#include + +#include "source_cell/mdcell.h" +#include "source_base/parallel_cell.h" + +#include + +#include +#include +#include +#include + +namespace +{ +void ensure_mpi_initialized() +{ + int initialized = 0; + MPI_Initialized(&initialized); + if (!initialized) + { + int provided = 0; + MPI_Init_thread(NULL, NULL, MPI_THREAD_SINGLE, &provided); + } +} + +ModuleBase::Matrix3 make_lattice() +{ + ModuleBase::Matrix3 latvec; + latvec.e11 = 1.0; + latvec.e22 = 1.0; + latvec.e33 = 1.0; + return latvec; +} +} + +TEST(MdCellMigrateMpiTest, AtomCrossingDomainMigratesToNewOwner) +{ + int rank = 0; + int size = 1; + MPI_Comm_rank(MPI_COMM_WORLD, &rank); + MPI_Comm_size(MPI_COMM_WORLD, &size); + ASSERT_GE(size, 2); + + const ModuleBase::Matrix3 latvec = make_lattice(); + std::vector owned_atoms; + if (rank < 2) + { + const ModuleBase::Vector3 frac(rank == 0 ? 0.2 : 0.7, 0.2, 0.2); + owned_atoms.push_back(LocalAtom(frac, + frac, + ModuleBase::Vector3(0.0, 0.0, 0.0), + ModuleBase::Vector3(0.0, 0.0, 0.0), + ModuleBase::Vector3(1, 1, 1), + 1.0, + 0, + rank, + rank)); + } + MDCell mdcell; + DomainDecomposition decomp; + decomp.init(ModuleBase::world_comm_domain(), latvec, 1.0, 0.0, 0.0); + mdcell.initialize_from_owned_atoms(latvec, + latvec.Inverse(), + 1.0, + 1.0, + 2, + owned_atoms, + std::vector(1, "X"), + std::vector(1, 1.0), + std::vector(1, 2), + 0.0, + ModuleBase::world_comm_domain()); + mdcell.set_neighbor_cutoff(0.1); + decomp.migrate_owned_atoms(mdcell); + + ASSERT_EQ(mdcell.mpi_size(), size); + if (size == 2) + { + ASSERT_EQ(mdcell.owned_atoms().size(), 1); + mdcell.owned_atoms()[0].vel.x = static_cast(rank + 1); + mdcell.owned_atoms()[0].force.y = static_cast(rank + 3); + decomp.migrate_owned_atoms(mdcell); + ASSERT_EQ(mdcell.owned_atoms().size(), 1); + EXPECT_EQ(mdcell.owned_atoms()[0].owner_rank, rank); + EXPECT_EQ(mdcell.owned_atoms()[0].vel.x, static_cast(rank + 1)); + EXPECT_EQ(mdcell.owned_atoms()[0].force.y, static_cast(rank + 3)); + + if (rank == 0 && mdcell.owned_atoms().size() == 1) + { + mdcell.owned_atoms()[0].cart.x = 0.8; + } + if (rank == 1 && mdcell.owned_atoms().size() == 1) + { + mdcell.owned_atoms()[0].cart.x = 0.3; + } + decomp.migrate_owned_atoms(mdcell); + + long long local_count = mdcell.owned_atoms().size(); + long long global_count = 0; + MPI_Allreduce(&local_count, &global_count, 1, MPI_LONG_LONG, MPI_SUM, MPI_COMM_WORLD); + EXPECT_EQ(global_count, 2); + + for (int i = 0; i < mdcell.owned_atoms().size(); ++i) + { + EXPECT_EQ(mdcell.owned_atoms()[static_cast(i)].owner_rank, rank); + } + } +} + +TEST(MdCellMigrateMpiTest, GhostForcesReturnToOwners) +{ + int rank = 0; + int size = 1; + MPI_Comm_rank(MPI_COMM_WORLD, &rank); + MPI_Comm_size(MPI_COMM_WORLD, &size); + ASSERT_EQ(size, 2); + + const ModuleBase::Matrix3 latvec = make_lattice(); + std::vector owned_atoms; + const ModuleBase::Vector3 frac(rank == 0 ? 0.2 : 0.7, 0.2, 0.2); + owned_atoms.push_back(LocalAtom(frac, + frac, + ModuleBase::Vector3(0.0, 0.0, 0.0), + ModuleBase::Vector3(0.0, 0.0, 0.0), + ModuleBase::Vector3(1, 1, 1), + 1.0, + 0, + rank, + rank)); + MDCell mdcell; + DomainDecomposition decomp; + decomp.init(ModuleBase::world_comm_domain(), latvec, 1.0, 0.0, 0.0); + mdcell.initialize_from_owned_atoms(latvec, + latvec.Inverse(), + 1.0, + 1.0, + 2, + owned_atoms, + std::vector(1, "X"), + std::vector(1, 1.0), + std::vector(1, 2), + 0.0, + ModuleBase::world_comm_domain()); + mdcell.set_neighbor_cutoff(0.6); + decomp.migrate_owned_atoms(mdcell); + + long long local_copies[2] = {0, 0}; + for (std::size_t iat = 0; iat < mdcell.ghost_atoms().size(); ++iat) + { + ++local_copies[mdcell.ghost_atoms()[iat].owner_rank]; + } + long long global_copies[2] = {0, 0}; + MPI_Allreduce(local_copies, global_copies, 2, MPI_LONG_LONG, MPI_SUM, MPI_COMM_WORLD); + + for (std::size_t iat = 0; iat < mdcell.ghost_atoms().size(); ++iat) + { + LocalAtom& ghost = mdcell.ghost_atoms()[iat]; + const double value = static_cast(ghost.owner_rank + 1); + ghost.force.set(value, 2.0 * value, 3.0 * value); + } + decomp.accumulate_ghost_forces(mdcell); + + ASSERT_EQ(mdcell.owned_atoms().size(), 1); + const double expected = static_cast(global_copies[rank] * (rank + 1)); + EXPECT_DOUBLE_EQ(mdcell.owned_atoms()[0].force.x, expected); + EXPECT_DOUBLE_EQ(mdcell.owned_atoms()[0].force.y, 2.0 * expected); + EXPECT_DOUBLE_EQ(mdcell.owned_atoms()[0].force.z, 3.0 * expected); +} + +TEST(MdCellMigrateMpiTest, SkinUpdatesFixedGhostLayoutBeforeRebuild) +{ + int rank = 0; + int size = 1; + MPI_Comm_rank(MPI_COMM_WORLD, &rank); + MPI_Comm_size(MPI_COMM_WORLD, &size); + ASSERT_EQ(size, 2); + + const ModuleBase::Matrix3 latvec = make_lattice(); + const ModuleBase::Vector3 frac(rank == 0 ? 0.20 : 0.70, 0.20, 0.20); + std::vector owned_atoms(1, LocalAtom(frac, + frac, + ModuleBase::Vector3(0.0, 0.0, 0.0), + ModuleBase::Vector3(0.0, 0.0, 0.0), + ModuleBase::Vector3(1, 1, 1), + 1.0, + 0, + rank, + rank)); + MDCell mdcell; + DomainDecomposition decomp; + decomp.init(ModuleBase::world_comm_domain(), latvec, 1.0, 0.0, 0.0); + mdcell.initialize_from_owned_atoms(latvec, + latvec.Inverse(), + 1.0, + 1.0, + 2, + owned_atoms, + std::vector(1, "X"), + std::vector(1, 1.0), + std::vector(1, 2), + 0.2, + ModuleBase::world_comm_domain()); + mdcell.set_neighbor_cutoff(0.1); + decomp.migrate_owned_atoms(mdcell); + + decomp.prepare_neighbors(mdcell); + mdcell.owned_atoms()[0].frac.x += rank == 0 ? 0.05 : -0.05; + mdcell.owned_atoms()[0].cart = mdcell.owned_atoms()[0].frac * latvec; + decomp.prepare_neighbors(mdcell); + + ASSERT_EQ(mdcell.owned_atoms().size(), 1); + for (std::size_t i = 0; i < mdcell.ghost_atoms().size(); ++i) + { + const LocalAtom& ghost = mdcell.ghost_atoms()[i]; + const double expected_frac = ghost.owner_rank == 0 ? 0.25 : 0.65; + EXPECT_DOUBLE_EQ(ghost.frac.x, expected_frac); + } +} + +int main(int argc, char** argv) +{ + MPI_Init(&argc, &argv); + ::testing::InitGoogleTest(&argc, argv); + const int result = RUN_ALL_TESTS(); + MPI_Finalize(); + return result; +} diff --git a/source/source_cell/module_neighlist/test/test_mdcell_reader.cpp b/source/source_cell/module_neighlist/test/test_mdcell_reader.cpp new file mode 100644 index 00000000000..50c907ab4f0 --- /dev/null +++ b/source/source_cell/module_neighlist/test/test_mdcell_reader.cpp @@ -0,0 +1,331 @@ +#include "source_cell/domain_decomposition.h" +#include + +#include "source_cell/mdcell_reader.h" +#include "source_cell/mdcell.h" +#include "source_cell/print_cell.h" +#include "source_base/constants.h" +#include "source_base/parallel_cell.h" +#include "source_base/global_variable.h" + +#include +#include +#include +#include +#include +#include +#include +#include + +static_assert(!std::is_copy_constructible::value, "MDCell must not be copy constructible."); +static_assert(!std::is_copy_assignable::value, "MDCell must not be copy assignable."); +static_assert(std::is_default_constructible::value, "MDCell must be default constructible."); +static_assert(std::is_move_constructible::value, "MDCell must be move constructible."); + +namespace +{ +void write_cartesian_stru_case(const std::string& stru_file) +{ + std::ofstream ofs(stru_file.c_str()); + ofs << "ATOMIC_SPECIES\n"; + ofs << "He 4.0026 auto auto\n\n"; + ofs << "LATTICE_CONSTANT\n"; + ofs << "1.0\n\n"; + ofs << "LATTICE_VECTORS\n"; + ofs << "4.0 0.0 0.0\n"; + ofs << "0.0 4.0 0.0\n"; + ofs << "0.0 0.0 4.0\n\n"; + ofs << "ATOMIC_POSITIONS\n"; + ofs << "Cartesian\n\n"; + ofs << "He\n"; + ofs << "0.0\n"; + ofs << "4\n"; + ofs << "0.40 0.40 0.40 m 1 1 1 v 0.01 0.00 0.00\n"; + ofs << "2.40 0.40 0.40 m 1 0 1 v 0.02 0.00 0.00\n"; + ofs << "0.40 2.40 0.40 m 0 1 1 v 0.03 0.00 0.00\n"; + ofs << "2.40 2.40 0.40 m 1 1 0 v 0.04 0.00 0.00\n"; +} + +ModuleBase::Matrix3 make_lattice() +{ + ModuleBase::Matrix3 latvec; + latvec.e11 = 4.0; + latvec.e12 = 0.0; + latvec.e13 = 0.0; + latvec.e21 = 0.0; + latvec.e22 = 4.0; + latvec.e23 = 0.0; + latvec.e31 = 0.0; + latvec.e32 = 0.0; + latvec.e33 = 4.0; + return latvec; +} +} // namespace + +TEST(MDCellReaderTest, ReadOwnedAtomsFromSTRUWithoutUnitCell) +{ + int world_rank = 0; + MPI_Comm_rank(MPI_COMM_WORLD, &world_rank); + + const std::string stru_file = "mdcell_reader_cartesian.STRU"; + if (world_rank == 0) + { + write_cartesian_stru_case(stru_file); + } + MPI_Barrier(MPI_COMM_WORLD); + + MPI_Comm md_comm = MPI_COMM_NULL; + MPI_Comm_split(MPI_COMM_WORLD, world_rank % 2, world_rank, &md_comm); + ModuleBase::CommunicationDomain comm_domain; + comm_domain.initialize(md_comm); + + DomainDecomposition reader_decomp; + MDCell mdcell = MDCellReader::read_stru(stru_file, + std::vector{1, 1, 1}, + 0.0, + comm_domain, reader_decomp); + + EXPECT_EQ(mdcell.type_labels().size(), 1U); + EXPECT_EQ(mdcell.type_labels()[0], "He"); + ASSERT_EQ(mdcell.type_masses().size(), 1U); + EXPECT_DOUBLE_EQ(mdcell.type_masses()[0], 4.0026); + ASSERT_EQ(mdcell.type_atom_counts().size(), 1U); + EXPECT_EQ(mdcell.type_atom_counts()[0], 4); + ASSERT_EQ(mdcell.stru_meta().species.size(), 1U); + EXPECT_EQ(mdcell.stru_meta().species[0].pseudo_file, "auto"); + EXPECT_EQ(mdcell.stru_meta().species[0].pseudo_type, "auto"); + EXPECT_EQ(mdcell.nat(), 4); + + DomainDecomposition decomp; + decomp.init(comm_domain, make_lattice(), 1.0, 1.0 * ModuleBase::ANGSTROM_AU, 0.0); + + long long local_count = static_cast(mdcell.owned_atoms().size()); + long long global_count = 0; + MPI_Allreduce(&local_count, &global_count, 1, MPI_LONG_LONG, MPI_SUM, md_comm); + EXPECT_EQ(global_count, 4); + + std::set > local_ids; + for (std::size_t iat = 0; iat < mdcell.owned_atoms().size(); ++iat) + { + const LocalAtom& atom = mdcell.owned_atoms()[iat]; + EXPECT_EQ(decomp.owner_rank_from_frac(atom.frac), comm_domain.rank()); + local_ids.insert(std::make_pair(atom.type, atom.type_index)); + EXPECT_GE(atom.type, 0); + EXPECT_DOUBLE_EQ(atom.force.x, 0.0); + EXPECT_DOUBLE_EQ(atom.force.y, 0.0); + EXPECT_DOUBLE_EQ(atom.force.z, 0.0); + } + EXPECT_EQ(local_ids.size(), mdcell.owned_atoms().size()); + + bool saw_v01 = false; + bool saw_v04 = false; + for (std::size_t iat = 0; iat < mdcell.owned_atoms().size(); ++iat) + { + const LocalAtom& atom = mdcell.owned_atoms()[iat]; + if (atom.type == 0 && atom.type_index == 0) + { + saw_v01 = true; + EXPECT_DOUBLE_EQ(atom.vel.x, 0.01); + EXPECT_EQ(atom.mbl.x, 1); + EXPECT_EQ(atom.mbl.y, 1); + EXPECT_EQ(atom.mbl.z, 1); + EXPECT_DOUBLE_EQ(atom.mass, 4.0026 / ModuleBase::AU_to_MASS); + } + if (atom.type == 0 && atom.type_index == 3) + { + saw_v04 = true; + EXPECT_DOUBLE_EQ(atom.vel.x, 0.04); + EXPECT_EQ(atom.mbl.x, 1); + EXPECT_EQ(atom.mbl.y, 1); + EXPECT_EQ(atom.mbl.z, 0); + } + } + + const int saw_flags[2] = {saw_v01 ? 1 : 0, saw_v04 ? 1 : 0}; + int reduced_flags[2] = {0, 0}; + MPI_Allreduce(saw_flags, reduced_flags, 2, MPI_INT, MPI_MAX, md_comm); + EXPECT_EQ(reduced_flags[0], 1); + EXPECT_EQ(reduced_flags[1], 1); + + MPI_Comm_free(&md_comm); +} + +TEST(MDCellReaderTest, RestartStruPreservesAtomRecordsAcrossRanks) +{ + int rank = 0; + int size = 1; + MPI_Comm_rank(MPI_COMM_WORLD, &rank); + MPI_Comm_size(MPI_COMM_WORLD, &size); + ASSERT_GE(size, 4); + + std::vector owned_atoms; + if (rank == 0) + { + owned_atoms.push_back(LocalAtom(ModuleBase::Vector3(11.0, 0.0, 0.0), + ModuleBase::Vector3(0.0, 0.0, 0.0), + ModuleBase::Vector3(0.11, 0.0, 0.0), + ModuleBase::Vector3(0.0, 0.0, 0.0), + ModuleBase::Vector3(1, 0, 1), + 1.0, + 1, + 1, + rank)); + } + if (rank == 1) + { + owned_atoms.push_back(LocalAtom(ModuleBase::Vector3(1.0, 0.0, 0.0), + ModuleBase::Vector3(0.0, 0.0, 0.0), + ModuleBase::Vector3(0.01, 0.0, 0.0), + ModuleBase::Vector3(0.0, 0.0, 0.0), + ModuleBase::Vector3(0, 1, 1), + 1.0, + 0, + 1, + rank)); + } + if (rank == 2) + { + owned_atoms.push_back(LocalAtom(ModuleBase::Vector3(10.0, 0.0, 0.0), + ModuleBase::Vector3(0.0, 0.0, 0.0), + ModuleBase::Vector3(0.10, 0.0, 0.0), + ModuleBase::Vector3(0.0, 0.0, 0.0), + ModuleBase::Vector3(1, 1, 0), + 1.0, + 1, + 0, + rank)); + } + if (rank == 3) + { + owned_atoms.push_back(LocalAtom(ModuleBase::Vector3(0.0, 0.0, 0.0), + ModuleBase::Vector3(0.0, 0.0, 0.0), + ModuleBase::Vector3(0.0, 0.0, 0.0), + ModuleBase::Vector3(0.0, 0.0, 0.0), + ModuleBase::Vector3(0, 0, 1), + 1.0, + 0, + 0, + rank)); + } + + ModuleBase::Matrix3 lattice = make_lattice(); + lattice.e11 = 20.0; + lattice.e22 = 20.0; + lattice.e33 = 20.0; + MDCell mdcell; + mdcell.initialize_from_owned_atoms(lattice, + lattice.Inverse(), + 1.0, + 1.0, + 4, + owned_atoms, + std::vector{"A", "B"}, + std::vector{1.0, 1.0}, + std::vector{2, 2}, + 0.0, + ModuleBase::world_comm_domain()); + StruMeta metadata; + metadata.species.resize(2); + const std::string output_file = "distributed_mdcell_restart.STRU"; + mdcell::print_stru_file(mdcell, metadata, output_file); + + DomainDecomposition reader_decomp; + MDCell round_trip = MDCellReader::read_stru(output_file, + std::vector{1, 1, 1}, + 0.0, + ModuleBase::world_comm_domain(), reader_decomp); + double local_positions[4] = {0.0, 0.0, 0.0, 0.0}; + double local_velocities[4] = {0.0, 0.0, 0.0, 0.0}; + int local_mbl_x[4] = {0, 0, 0, 0}; + int local_owners[4] = {0, 0, 0, 0}; + for (std::size_t iat = 0; iat < round_trip.owned_atoms().size(); ++iat) + { + const LocalAtom& atom = round_trip.owned_atoms()[iat]; + const int index = atom.type == 0 ? static_cast(atom.cart.x) + : 2 + static_cast(atom.cart.x) - 10; + local_positions[index] = atom.cart.x; + local_velocities[index] = atom.vel.x; + local_mbl_x[index] = atom.mbl.x; + local_owners[index] = 1; + } + double global_positions[4] = {0.0, 0.0, 0.0, 0.0}; + double global_velocities[4] = {0.0, 0.0, 0.0, 0.0}; + int global_mbl_x[4] = {0, 0, 0, 0}; + int global_owners[4] = {0, 0, 0, 0}; + MPI_Allreduce(local_positions, global_positions, 4, MPI_DOUBLE, MPI_SUM, MPI_COMM_WORLD); + MPI_Allreduce(local_velocities, global_velocities, 4, MPI_DOUBLE, MPI_SUM, MPI_COMM_WORLD); + MPI_Allreduce(local_mbl_x, global_mbl_x, 4, MPI_INT, MPI_SUM, MPI_COMM_WORLD); + MPI_Allreduce(local_owners, global_owners, 4, MPI_INT, MPI_SUM, MPI_COMM_WORLD); + for (int iat = 0; iat < 4; ++iat) EXPECT_EQ(global_owners[iat], 1); + EXPECT_DOUBLE_EQ(global_positions[0], 0.0); + EXPECT_DOUBLE_EQ(global_positions[1], 1.0); + EXPECT_DOUBLE_EQ(global_positions[2], 10.0); + EXPECT_DOUBLE_EQ(global_positions[3], 11.0); + EXPECT_DOUBLE_EQ(global_velocities[0], 0.0); + EXPECT_DOUBLE_EQ(global_velocities[1], 0.01); + EXPECT_DOUBLE_EQ(global_velocities[2], 0.10); + EXPECT_DOUBLE_EQ(global_velocities[3], 0.11); + EXPECT_EQ(global_mbl_x[0], 0); + EXPECT_EQ(global_mbl_x[1], 0); + EXPECT_EQ(global_mbl_x[2], 1); + EXPECT_EQ(global_mbl_x[3], 1); + + if (rank == 0) + { + std::ifstream input(output_file.c_str()); + ASSERT_TRUE(input.good()); + std::vector coordinates_a; + std::vector coordinates_b; + std::string line; + int current_type = -1; + int atoms_remaining = 0; + while (std::getline(input, line)) + { + if (line == "A #label") + { + current_type = 0; + continue; + } + if (line == "B #label") + { + current_type = 1; + continue; + } + if (current_type >= 0 && line.find("#number of atoms") != std::string::npos) + { + std::istringstream count_stream(line); + count_stream >> atoms_remaining; + continue; + } + if (current_type < 0 || atoms_remaining == 0) continue; + std::istringstream values(line); + double x = 0.0; + values >> x; + --atoms_remaining; + if (values) + { + if (current_type == 0) coordinates_a.push_back(x); + else coordinates_b.push_back(x); + } + } + ASSERT_EQ(coordinates_a.size(), 2U); + ASSERT_EQ(coordinates_b.size(), 2U); + std::set expected_a = {0.0, 1.0}; + std::set expected_b = {10.0, 11.0}; + EXPECT_EQ(std::set(coordinates_a.begin(), coordinates_a.end()), expected_a); + EXPECT_EQ(std::set(coordinates_b.begin(), coordinates_b.end()), expected_b); + } + MPI_Barrier(MPI_COMM_WORLD); + if (rank == 0) std::remove(output_file.c_str()); +} + +int main(int argc, char** argv) +{ + MPI_Init(&argc, &argv); + MPI_Comm_rank(MPI_COMM_WORLD, &GlobalV::MY_RANK); + MPI_Comm_size(MPI_COMM_WORLD, &GlobalV::NPROC); + ::testing::InitGoogleTest(&argc, argv); + const int result = RUN_ALL_TESTS(); + MPI_Finalize(); + return result; +} diff --git a/source/source_cell/module_neighlist/test/neighbor_list_test.cpp b/source/source_cell/module_neighlist/test/test_neighbor_list.cpp similarity index 100% rename from source/source_cell/module_neighlist/test/neighbor_list_test.cpp rename to source/source_cell/module_neighlist/test/test_neighbor_list.cpp diff --git a/source/source_cell/module_neighlist/test/neighbor_search_test.cpp b/source/source_cell/module_neighlist/test/test_neighbor_search.cpp similarity index 100% rename from source/source_cell/module_neighlist/test/neighbor_search_test.cpp rename to source/source_cell/module_neighlist/test/test_neighbor_search.cpp diff --git a/source/source_cell/module_neighlist/test/test_page_allocator.cpp b/source/source_cell/module_neighlist/test/test_page_allocator.cpp new file mode 100644 index 00000000000..b3409dd18ac --- /dev/null +++ b/source/source_cell/module_neighlist/test/test_page_allocator.cpp @@ -0,0 +1,52 @@ +#include + +#include "source_cell/module_neighlist/page_allocator.h" + +#include + +TEST(PageAllocatorTest, RejectsNonPositivePageSize) +{ + PageAllocator allocator; + + EXPECT_THROW(allocator.initialize(0), std::invalid_argument); + EXPECT_THROW(allocator.initialize(-1), std::invalid_argument); +} + +TEST(PageAllocatorTest, AllocatesContiguousBlocksWithinPage) +{ + PageAllocator allocator; + allocator.initialize(8); + + int* first = allocator.allocate(2); + int* second = allocator.allocate(3); + + ASSERT_NE(first, nullptr); + ASSERT_NE(second, nullptr); + EXPECT_EQ(second, first + 2); + EXPECT_EQ(allocator.get_pgsize(), 8); +} + +TEST(PageAllocatorTest, StartsNewPageAndResetReusesFirstPage) +{ + PageAllocator allocator; + allocator.initialize(2); + + int* first = allocator.allocate(2); + int* second_page = allocator.allocate(1); + + ASSERT_NE(first, nullptr); + ASSERT_NE(second_page, nullptr); + EXPECT_NE(second_page, first); + + allocator.reset(); + EXPECT_EQ(allocator.allocate(1), first); +} + +TEST(PageAllocatorTest, NonPositiveAllocationReturnsNull) +{ + PageAllocator allocator; + allocator.initialize(4); + + EXPECT_EQ(allocator.allocate(0), nullptr); + EXPECT_EQ(allocator.allocate(-1), nullptr); +} diff --git a/source/source_cell/module_symmetry/CMakeLists.txt b/source/source_cell/module_symmetry/CMakeLists.txt index 7cdd95d129b..5b2a03cd967 100644 --- a/source/source_cell/module_symmetry/CMakeLists.txt +++ b/source/source_cell/module_symmetry/CMakeLists.txt @@ -17,6 +17,9 @@ add_library( symmetry.cpp symm_rot_spin.cpp little_group.cpp + irreducible_sector.cpp + irred_sec_bvk.cpp + symm_rotation_k.cpp ) if(ENABLE_COVERAGE) diff --git a/source/source_lcao/module_ri/module_exx_symmetry/irred_sec_bvk.cpp b/source/source_cell/module_symmetry/irred_sec_bvk.cpp similarity index 99% rename from source/source_lcao/module_ri/module_exx_symmetry/irred_sec_bvk.cpp rename to source/source_cell/module_symmetry/irred_sec_bvk.cpp index b97f95a990b..5d506cab434 100644 --- a/source/source_lcao/module_ri/module_exx_symmetry/irred_sec_bvk.cpp +++ b/source/source_cell/module_symmetry/irred_sec_bvk.cpp @@ -75,7 +75,7 @@ namespace ModuleSymmetry return; } - // extern lattice to minimal BvK lattice, and set direct coordinates in min BvK lattice + // extern lattice to minimal BvK lattice, and set direct coordinates in min BvK lattice int bvk_gcd = gcd(bvk_period[0], gcd(bvk_period[1], bvk_period[2])); const TC bvk_min_period = TC({ bvk_period[0] / bvk_gcd, bvk_period[1] / bvk_gcd, bvk_period[2] / bvk_gcd }); const int bvk_nat = st.nat * bvk_min_period[0] * bvk_min_period[1] * bvk_min_period[2]; @@ -199,4 +199,4 @@ namespace ModuleSymmetry // return in_plain; // } -}; \ No newline at end of file +}; diff --git a/source/source_lcao/module_ri/module_exx_symmetry/irreducible_sector.cpp b/source/source_cell/module_symmetry/irreducible_sector.cpp similarity index 97% rename from source/source_lcao/module_ri/module_exx_symmetry/irreducible_sector.cpp rename to source/source_cell/module_symmetry/irreducible_sector.cpp index 26157460c3c..c6521c7ab52 100644 --- a/source/source_lcao/module_ri/module_exx_symmetry/irreducible_sector.cpp +++ b/source/source_cell/module_symmetry/irreducible_sector.cpp @@ -1,5 +1,4 @@ -#include "source_lcao/module_ri/module_exx_symmetry/irreducible_sector.h" -#include "source_io/module_parameter/parameter.h" +#include "source_cell/module_symmetry/irreducible_sector.h" namespace ModuleSymmetry { // Raw-index dispatch shared by the real-space sector helpers, matching the convention used @@ -62,7 +61,7 @@ namespace ModuleSymmetry } } - // Perfoming {R|t} to atom position r in the R=0 lattice, we get Rr+t, which may get out of R=0 lattice, + // Perfoming {R|t} to atom position r in the R=0 lattice, we get Rr+t, which may get out of R=0 lattice, // whose image in R=0 lattice is r'=Rr+t-O. This function is to get O for each atom and each symmetry operation. // the range of direct position is [-0.5, 0.5). TCdouble Irreducible_Sector::get_return_lattice(const Symmetry& symm, @@ -109,7 +108,7 @@ namespace ModuleSymmetry ModuleBase::TITLE("Symmetry_rotation", "cal_return_lattice_all"); // Columns [0, nrotk) are the unitary operations; columns [nrotk, nrotk+nrotk_anti) are the // spatial parts of the antiunitary elements Theta*g of the Shubnikov group (nspin=4 magnetic), - // so that Symmetry_rotation can address both with one raw index. + // so that Symmetry_rotation can address both with one raw index. this->return_lattice_.resize(st.nat, std::vector(symm.nrotk + symm.nrotk_anti)); for (int iat1 = 0;iat1 < st.nat;++iat1) { @@ -170,12 +169,12 @@ namespace ModuleSymmetry std::cout << std::endl; } } - void Irreducible_Sector::write_irreducible_sector() + void Irreducible_Sector::write_irreducible_sector(const std::string& output_dir) { - if(GlobalV::MY_RANK == 0) + if(GlobalV::MY_RANK == 0 && !output_dir.empty()) { std::ofstream ofs; - ofs.open(PARAM.globalv.global_out_dir + "irreducible_sector.txt"); + ofs.open(output_dir + "irreducible_sector.txt"); for (auto& irap_irR : this->irreducible_sector_) { for (auto& irR : irap_irR.second){ofs << "atompair (" << irap_irR.first.first << ", " << irap_irR.first.second << "), R = (" << irR[0] << ", " << irR[1] << ", " << irR[2] << ") \n";} @@ -184,7 +183,7 @@ namespace ModuleSymmetry } } - void Irreducible_Sector::find_irreducible_sector(const Symmetry& symm, const Atom* atoms, const Statistics& st, const std::vector& Rs, const TC& period, const Lattice& lat) + void Irreducible_Sector::find_irreducible_sector(const Symmetry& symm, const Atom* atoms, const Statistics& st, const std::vector& Rs, const TC& period, const Lattice& lat, const std::string& output_dir) { this->full_map_to_irreducible_sector_.clear(); this->irreducible_sector_.clear(); @@ -277,6 +276,6 @@ namespace ModuleSymmetry assert(total_apR_in_star == this->full_map_to_irreducible_sector_.size()); // this->output_full_map_to_irreducible_sector(st.nat); // this->output_sector_star(); - this->write_irreducible_sector(); + this->write_irreducible_sector(output_dir); } } diff --git a/source/source_lcao/module_ri/module_exx_symmetry/irreducible_sector.h b/source/source_cell/module_symmetry/irreducible_sector.h similarity index 87% rename from source/source_lcao/module_ri/module_exx_symmetry/irreducible_sector.h rename to source/source_cell/module_symmetry/irreducible_sector.h index 7bfdd1d3eae..f35be6a9d2f 100644 --- a/source/source_lcao/module_ri/module_exx_symmetry/irreducible_sector.h +++ b/source/source_cell/module_symmetry/irreducible_sector.h @@ -1,8 +1,10 @@ -#pragma once +#ifndef ABACUS_SOURCE_CELL_MODULE_SYMMETRY_IRREDUCIBLE_SECTOR_H +#define ABACUS_SOURCE_CELL_MODULE_SYMMETRY_IRREDUCIBLE_SECTOR_H + #include #include #include -#include "source_lcao/module_ri/abfs_vector3_order.h" +#include "source_base/vector3.h" #include "source_base/matrix3.h" #include "source_cell/module_symmetry/symmetry.h" #include "source_cell/klist.h" @@ -12,7 +14,7 @@ namespace ModuleSymmetry using Tap = std::pair; using TC = std::array; using TapR = std::pair; - using TCdouble = Abfs::Vector3_Order; + using TCdouble = ModuleBase::Vector3; class Irreducible_Sector { @@ -54,13 +56,18 @@ namespace ModuleSymmetry //-------------------------------------------------------------------------------- /// The main function to find irreducible sector: {abR} + /// @param output_dir if non-empty, write_irreducible_sector() dumps a debug listing to + /// /irreducible_sector.txt (callers pass the run's global + /// output directory setting); kept as an explicit argument rather than + /// reading the global config directly so this LibRI-free class has no + /// module_parameter link dependency. void find_irreducible_sector(const Symmetry& symm, const Atom* atoms, const Statistics& st, - const std::vector& Rs, const TC& period, const Lattice& lat); + const std::vector& Rs, const TC& period, const Lattice& lat, const std::string& output_dir = ""); const std::map>& get_irreducible_sector()const { return this->irreducible_sector_; } // const std::map>> convirt_irreducible_sector() {}; //-------------------------------------------------------------------------------- - /// Perfoming {R|t} to atom position r in the R=0 lattice, we get Rr+t, which may get out of R=0 lattice, + /// Perfoming {R|t} to atom position r in the R=0 lattice, we get Rr+t, which may get out of R=0 lattice, /// whose image in R=0 lattice is r'=Rr+t-O. This function is to get O for each atom and each symmetry operation. /// the range of direct position is [-0.5, 0.5). TCdouble get_return_lattice(const Symmetry& symm, @@ -104,7 +111,7 @@ namespace ModuleSymmetry void output_full_map_to_irreducible_sector(const int nat); void output_sector_star(); - void write_irreducible_sector(); + void write_irreducible_sector(const std::string& output_dir); //-------------------------------------------------------------------------------- /// The sub functions judge special symmetry @@ -140,5 +147,8 @@ namespace ModuleSymmetry int bvk_nsym_; friend class Symmetry_rotation; + friend class Symmetry_rotation_k; }; -} \ No newline at end of file +} + +#endif // ABACUS_SOURCE_CELL_MODULE_SYMMETRY_IRREDUCIBLE_SECTOR_H diff --git a/source/source_cell/module_symmetry/symm_rho.cpp b/source/source_cell/module_symmetry/symm_rho.cpp index 52f208d03dc..cbada24b38f 100644 --- a/source/source_cell/module_symmetry/symm_rho.cpp +++ b/source/source_cell/module_symmetry/symm_rho.cpp @@ -5,109 +5,109 @@ using namespace ModuleSymmetry; namespace { - // ------------------------------------------------------------------------ - // Rotating reciprocal-space FFT-grid vector (with PBC) - // The rotated vector is returned via ii, jj, kk. - // ------------------------------------------------------------------------ - //rotate function (different from real space, without scaling gmatrix) - static inline void rotate_recip(const ModuleBase::Matrix3& g, const ModuleBase::Vector3& g0, int& ii, int& jj, int& kk, - const int& nx, const int& ny, const int& nz) - { - ii = int(g.e11 * g0.x + g.e21 * g0.y + g.e31 * g0.z) ; - if (ii < 0) - { - ii += 10 * nx; - } - ii = ii%nx; - jj = int(g.e12 * g0.x + g.e22 * g0.y + g.e32 * g0.z) ; - if (jj < 0) - { - jj += 10 * ny; - } - jj = jj%ny; - kk = int(g.e13 * g0.x + g.e23 * g0.y + g.e33 * g0.z); - if (kk < 0) - { - kk += 10 * nz; - } - kk = kk%nz; - return; - } - - // ------------------------------------------------------------------------ - // Trying to group fft grids first. - // It iterates over each FFT-grid point and checks if it is within the - // PW-sphere. If it is, put all the FFT-grid points connected by the - // rotation operation into one group( the index is stored in int(*table_xyz)). - // The code marks the point as processed to avoid redundant calculations - // by using int* symflag. - // This grouping is purely spatial (depends only on kgmatrix/invmap and the - // FFT-grid geometry), so it is shared between rhog_symmetry and rhog_symmetry_nspin4; - // the differing spin/phase accumulation happens after this call. - // ------------------------------------------------------------------------ - static void group_fft_grids(const int& nrotk, const ModuleBase::Matrix3* kgmatrix, const std::vector& invmap, - int* ixyz2ipw, const int& nx, const int& ny, const int& nz, - const int& fftnx, const int& fftny, const int& fftnz, const bool gamma_only_pw, - int* symflag, int (*isymflag)[48], int (*table_xyz)[48], int* count_xyz, int& group_index) - { - ModuleBase::timer::start("Symmetry","group_fft_grids"); - for (int i = 0; i< fftnx; ++i) - { - //tmp variable - ModuleBase::Vector3 tmp_gdirect0(0, 0, 0); - tmp_gdirect0.x=(i>int(nx/2)+1)?(i-nx):i; - for (int j = 0; j< fftny; ++j) - { - tmp_gdirect0.y=(j>int(ny/2)+1)?(j-ny):j; - for (int k = 0; k< fftnz; ++k) - { - int ixyz0=(i*fftny+j)*fftnz+k; - if (symflag[ixyz0] == -1) - { - int ipw0=ixyz2ipw[ixyz0]; - //if a fft-grid is not in pw-sphere, just do not consider it. - if (ipw0 == -1) { - continue; - } - tmp_gdirect0.z=(k>int(nz/2)+1)?(k-nz):k; - int rot_count=0; - for (int isym = 0; isym < nrotk; ++isym) - { - if (invmap[isym] < 0 || invmap[isym] > nrotk) { continue; } - //tmp variables - int ii, jj, kk=0; - rotate_recip(kgmatrix[invmap[isym]], tmp_gdirect0, ii, jj, kk, nx, ny, nz); - if(ii>=fftnx || jj>=fftny || kk>= fftnz) - { - if(!gamma_only_pw) - { - std::cout << " ROTATE OUT OF FFT-GRID IN RHOG_SYMMETRY !" << std::endl; - ModuleBase::QUIT(); - } - // for gamma_only_pw, just do not consider this rotation. - continue; - } - int ixyz=(ii*fftny+jj)*fftnz+kk; - //fft-grid index to (ip, ig) - int ipw=ixyz2ipw[ixyz]; - if(ipw==-1) //not in pw-sphere - { - continue; //else, just skip it - } - symflag[ixyz] = group_index; - isymflag[group_index][rot_count] = invmap[isym]; - table_xyz[group_index][rot_count] = ixyz; - ++rot_count; - assert(rot_count <= nrotk); - count_xyz[group_index] = rot_count; - } - group_index++; - } - } - } - } - ModuleBase::timer::end("Symmetry","group_fft_grids"); - } + // ------------------------------------------------------------------------ + // Rotating reciprocal-space FFT-grid vector (with PBC) + // The rotated vector is returned via ii, jj, kk. + // ------------------------------------------------------------------------ + //rotate function (different from real space, without scaling gmatrix) + static inline void rotate_recip(const ModuleBase::Matrix3& g, const ModuleBase::Vector3& g0, int& ii, int& jj, int& kk, + const int& nx, const int& ny, const int& nz) + { + ii = int(g.e11 * g0.x + g.e21 * g0.y + g.e31 * g0.z) ; + if (ii < 0) + { + ii += 10 * nx; + } + ii = ii%nx; + jj = int(g.e12 * g0.x + g.e22 * g0.y + g.e32 * g0.z) ; + if (jj < 0) + { + jj += 10 * ny; + } + jj = jj%ny; + kk = int(g.e13 * g0.x + g.e23 * g0.y + g.e33 * g0.z); + if (kk < 0) + { + kk += 10 * nz; + } + kk = kk%nz; + return; + } + + // ------------------------------------------------------------------------ + // Trying to group fft grids first. + // It iterates over each FFT-grid point and checks if it is within the + // PW-sphere. If it is, put all the FFT-grid points connected by the + // rotation operation into one group( the index is stored in int(*table_xyz)). + // The code marks the point as processed to avoid redundant calculations + // by using int* symflag. + // This grouping is purely spatial (depends only on kgmatrix/invmap and the + // FFT-grid geometry), so it is shared between rhog_symmetry and rhog_symmetry_nspin4; + // the differing spin/phase accumulation happens after this call. + // ------------------------------------------------------------------------ + static void group_fft_grids(const int& nrotk, const ModuleBase::Matrix3* kgmatrix, const std::vector& invmap, + int* ixyz2ipw, const int& nx, const int& ny, const int& nz, + const int& fftnx, const int& fftny, const int& fftnz, const bool gamma_only_pw, + int* symflag, int (*isymflag)[48], int (*table_xyz)[48], int* count_xyz, int& group_index) + { + ModuleBase::timer::start("Symmetry","group_fft_grids"); + for (int i = 0; i< fftnx; ++i) + { + //tmp variable + ModuleBase::Vector3 tmp_gdirect0(0, 0, 0); + tmp_gdirect0.x=(i>int(nx/2)+1)?(i-nx):i; + for (int j = 0; j< fftny; ++j) + { + tmp_gdirect0.y=(j>int(ny/2)+1)?(j-ny):j; + for (int k = 0; k< fftnz; ++k) + { + int ixyz0=(i*fftny+j)*fftnz+k; + if (symflag[ixyz0] == -1) + { + int ipw0=ixyz2ipw[ixyz0]; + //if a fft-grid is not in pw-sphere, just do not consider it. + if (ipw0 == -1) { + continue; + } + tmp_gdirect0.z=(k>int(nz/2)+1)?(k-nz):k; + int rot_count=0; + for (int isym = 0; isym < nrotk; ++isym) + { + if (invmap[isym] < 0 || invmap[isym] > nrotk) { continue; } + //tmp variables + int ii, jj, kk=0; + rotate_recip(kgmatrix[invmap[isym]], tmp_gdirect0, ii, jj, kk, nx, ny, nz); + if(ii>=fftnx || jj>=fftny || kk>= fftnz) + { + if(!gamma_only_pw) + { + std::cout << " ROTATE OUT OF FFT-GRID IN RHOG_SYMMETRY !" << std::endl; + ModuleBase::QUIT(); + } + // for gamma_only_pw, just do not consider this rotation. + continue; + } + int ixyz=(ii*fftny+jj)*fftnz+kk; + //fft-grid index to (ip, ig) + int ipw=ixyz2ipw[ixyz]; + if(ipw==-1) //not in pw-sphere + { + continue; //else, just skip it + } + symflag[ixyz] = group_index; + isymflag[group_index][rot_count] = invmap[isym]; + table_xyz[group_index][rot_count] = ixyz; + ++rot_count; + assert(rot_count <= nrotk); + count_xyz[group_index] = rot_count; + } + group_index++; + } + } + } + } + ModuleBase::timer::end("Symmetry","group_fft_grids"); + } } // namespace void Symmetry::rho_symmetry( double *rho, @@ -336,41 +336,41 @@ void Symmetry::rhog_symmetry_nspin4(std::complex* rhogtot_x, std::comple const double* trs_inv, const ModuleBase::Matrix3* kgmatrix_in, const ModuleBase::Vector3* gtrans_in, const int nop) { - // Operation set: default = the nrotk unitary members; - // the nspin=4 magnetic caller passes the full Shubnikov list from density_sym_ops(). - // `trs_inv` is the time-reversal sign of each operation: Theta reverses the magnetization, - // so an antiunitary element contributes m -> -W(g) m. - const ModuleBase::Matrix3* kgmatrix_use = (kgmatrix_in != nullptr) ? kgmatrix_in : this->kgmatrix; - const ModuleBase::Vector3* gtrans_use = (gtrans_in != nullptr) ? gtrans_in : this->gtrans; - const int nrot_use = (nop > 0) ? nop : this->nrotk; - std::vector trs_inv_default; - if (trs_inv == nullptr) { trs_inv_default.assign(nrot_use, 1.0); } - const double* trs_invp = (trs_inv != nullptr) ? trs_inv : trs_inv_default.data(); - ModuleBase::timer::start("Symmetry","rhog_symmetry_nspin4"); - // The grouping of FFT grid points into symmetry-connected orbits is purely spatial and - // therefore identical to rhog_symmetry. Only the accumulation/write-back is changed: - // the three spin components are mixed by W(g) (rotated to the orbit-representative frame - // with W(g)^T on the way in, and back with W(g) on the way out), exactly as the scalar - // version uses the phase factor gphase. + // Operation set: default = the nrotk unitary members; + // the nspin=4 magnetic caller passes the full Shubnikov list from density_sym_ops(). + // `trs_inv` is the time-reversal sign of each operation: Theta reverses the magnetization, + // so an antiunitary element contributes m -> -W(g) m. + const ModuleBase::Matrix3* kgmatrix_use = (kgmatrix_in != nullptr) ? kgmatrix_in : this->kgmatrix; + const ModuleBase::Vector3* gtrans_use = (gtrans_in != nullptr) ? gtrans_in : this->gtrans; + const int nrot_use = (nop > 0) ? nop : this->nrotk; + std::vector trs_inv_default; + if (trs_inv == nullptr) { trs_inv_default.assign(nrot_use, 1.0); } + const double* trs_invp = (trs_inv != nullptr) ? trs_inv : trs_inv_default.data(); + ModuleBase::timer::start("Symmetry","rhog_symmetry_nspin4"); + // The grouping of FFT grid points into symmetry-connected orbits is purely spatial and + // therefore identical to rhog_symmetry. Only the accumulation/write-back is changed: + // the three spin components are mixed by W(g) (rotated to the orbit-representative frame + // with W(g)^T on the way in, and back with W(g) on the way out), exactly as the scalar + // version uses the phase factor gphase. const int nxyz = fftnx*fftny*fftnz; assert(nxyz>0); - int* symflag = new int[nxyz]; - int(*isymflag)[48] = new int[nxyz][48]; - int(*table_xyz)[48] = new int[nxyz][48]; - int* count_xyz = new int[nxyz]; + int* symflag = new int[nxyz]; + int(*isymflag)[48] = new int[nxyz][48]; + int(*table_xyz)[48] = new int[nxyz][48]; + int* count_xyz = new int[nxyz]; - for (int i = 0; i < nxyz; i++) - { - symflag[i] = -1; - } - int group_index = 0; + for (int i = 0; i < nxyz; i++) + { + symflag[i] = -1; + } + int group_index = 0; - assert(nrot_use >0 ); - assert(nrot_use <=48 ); + assert(nrot_use >0 ); + assert(nrot_use <=48 ); - //map the gmatrix to inv + //map the gmatrix to inv std::vectorinvmap(nrot_use, -1); this->gmatrix_invmap(kgmatrix_use, nrot_use, invmap.data()); @@ -383,110 +383,110 @@ void Symmetry::rhog_symmetry_nspin4(std::complex* rhogtot_x, std::comple #ifdef _OPENMP #pragma omp parallel for schedule(static) #endif - for (int g_index = 0; g_index < group_index; g_index++) - { - int *ipw_record = new int[nrot_use]; - int *ixyz_record = new int[nrot_use]; - int *sym_record = new int[nrot_use]; - std::complex* gphase_record = new std::complex [nrot_use]; - // orbit-representative-frame spin vector accumulated over the symmetry operations - std::complex sum_x(0, 0), sum_y(0, 0), sum_z(0, 0); - int rot_count=0; - - for (int c_index = 0; c_index < count_xyz[g_index]; ++c_index) - { - int ixyz0 = table_xyz[g_index][c_index]; - int ipw0 = ixyz2ipw[ixyz0]; - - if (symflag[ixyz0] == g_index) - { - int k = ixyz0%fftnz; - int j = ((ixyz0-k)/fftnz)%fftny; - int i = ((ixyz0-k)/fftnz-j)/fftny; - - ModuleBase::Vector3 tmp_gdirect_double(0.0, 0.0, 0.0); - tmp_gdirect_double.x=static_cast((i>int(nx/2)+1)?(i-nx):i); - tmp_gdirect_double.y=static_cast((j>int(ny/2)+1)?(j-ny):j); - tmp_gdirect_double.z=static_cast((k>int(nz/2)+1)?(k-nz):k); - - tmp_gdirect_double = tmp_gdirect_double * ModuleBase::TWO_PI; - - double cos_arg = 0.0, sin_arg = 0.0; - double arg_gtrans = tmp_gdirect_double * gtrans_use[isymflag[g_index][c_index]]; - - std::complex phase_gtrans (ModuleBase::libm::cos(arg_gtrans), - ModuleBase::libm::sin(arg_gtrans)); - - for (int ipt = 0;ipt < ((ModuleSymmetry::Symmetry::pricell_loop) ? this->ncell : 1);++ipt) - { - double arg = tmp_gdirect_double * ptrans[ipt]; - double tmp_cos = 0.0, tmp_sin = 0.0; - ModuleBase::libm::sincos(arg, &tmp_sin, &tmp_cos); - cos_arg += tmp_cos; - sin_arg += tmp_sin; - } - - cos_arg/=static_cast(ncell); - sin_arg/=static_cast(ncell); - - if (equal(cos_arg, 0.0) && equal(sin_arg, 0.0)) - { - continue; - } - - std::complex gphase(cos_arg, sin_arg); - gphase = phase_gtrans * gphase; - - if (equal(gphase.real(), 1.0) && equal(gphase.imag(), 0)) - { - gphase = std::complex(1.0, 0.0); - } - - // pull this orbit member back to the representative frame: multiply by gphase - // (removes the translation/phase, as in the scalar version) then by W(g)^T. - const int isym = isymflag[g_index][c_index]; - const ModuleBase::Matrix3& W = wspin[isym]; - const std::complex vx = rhogtot_x[ipw0] * gphase; - const std::complex vy = rhogtot_y[ipw0] * gphase; - const std::complex vz = rhogtot_z[ipw0] * gphase; - const double trs_sign = trs_invp[isym]; - sum_x += trs_sign * (W.e11 * vx + W.e21 * vy + W.e31 * vz); // trs_inv * (W^T v)_x - sum_y += trs_sign * (W.e12 * vx + W.e22 * vy + W.e32 * vz); // trs_inv * (W^T v)_y - sum_z += trs_sign * (W.e13 * vx + W.e23 * vy + W.e33 * vz); // trs_inv * (W^T v)_z - - gphase_record[rot_count]=gphase; - ipw_record[rot_count]=ipw0; - ixyz_record[rot_count]=ixyz0; - sym_record[rot_count]=isym; - ++rot_count; - }//end if section - }//end c_index loop - if (rot_count!=0) - { - sum_x/= rot_count; - sum_y/= rot_count; - sum_z/= rot_count; - } - for (int ir = 0; ir < rot_count; ++ir) - { - // push the representative-frame value back out to this member: W(g) * S / gphase. - const ModuleBase::Matrix3& W = wspin[sym_record[ir]]; - const std::complex inv_gphase = 1.0 / gphase_record[ir]; - const double trs_sign = trs_invp[sym_record[ir]]; - rhogtot_x[ipw_record[ir]] = trs_sign * (W.e11 * sum_x + W.e12 * sum_y + W.e13 * sum_z) * inv_gphase; - rhogtot_y[ipw_record[ir]] = trs_sign * (W.e21 * sum_x + W.e22 * sum_y + W.e23 * sum_z) * inv_gphase; - rhogtot_z[ipw_record[ir]] = trs_sign * (W.e31 * sum_x + W.e32 * sum_y + W.e33 * sum_z) * inv_gphase; - } - - delete[] ipw_record; - delete[] ixyz_record; - delete[] sym_record; - delete[] gphase_record; - }//end g_index loop - - delete[] symflag; - delete[] isymflag; - delete[] table_xyz; - delete[] count_xyz; - ModuleBase::timer::end("Symmetry","rhog_symmetry_nspin4"); + for (int g_index = 0; g_index < group_index; g_index++) + { + int *ipw_record = new int[nrot_use]; + int *ixyz_record = new int[nrot_use]; + int *sym_record = new int[nrot_use]; + std::complex* gphase_record = new std::complex [nrot_use]; + // orbit-representative-frame spin vector accumulated over the symmetry operations + std::complex sum_x(0, 0), sum_y(0, 0), sum_z(0, 0); + int rot_count=0; + + for (int c_index = 0; c_index < count_xyz[g_index]; ++c_index) + { + int ixyz0 = table_xyz[g_index][c_index]; + int ipw0 = ixyz2ipw[ixyz0]; + + if (symflag[ixyz0] == g_index) + { + int k = ixyz0%fftnz; + int j = ((ixyz0-k)/fftnz)%fftny; + int i = ((ixyz0-k)/fftnz-j)/fftny; + + ModuleBase::Vector3 tmp_gdirect_double(0.0, 0.0, 0.0); + tmp_gdirect_double.x=static_cast((i>int(nx/2)+1)?(i-nx):i); + tmp_gdirect_double.y=static_cast((j>int(ny/2)+1)?(j-ny):j); + tmp_gdirect_double.z=static_cast((k>int(nz/2)+1)?(k-nz):k); + + tmp_gdirect_double = tmp_gdirect_double * ModuleBase::TWO_PI; + + double cos_arg = 0.0, sin_arg = 0.0; + double arg_gtrans = tmp_gdirect_double * gtrans_use[isymflag[g_index][c_index]]; + + std::complex phase_gtrans (ModuleBase::libm::cos(arg_gtrans), + ModuleBase::libm::sin(arg_gtrans)); + + for (int ipt = 0;ipt < ((ModuleSymmetry::Symmetry::pricell_loop) ? this->ncell : 1);++ipt) + { + double arg = tmp_gdirect_double * ptrans[ipt]; + double tmp_cos = 0.0, tmp_sin = 0.0; + ModuleBase::libm::sincos(arg, &tmp_sin, &tmp_cos); + cos_arg += tmp_cos; + sin_arg += tmp_sin; + } + + cos_arg/=static_cast(ncell); + sin_arg/=static_cast(ncell); + + if (equal(cos_arg, 0.0) && equal(sin_arg, 0.0)) + { + continue; + } + + std::complex gphase(cos_arg, sin_arg); + gphase = phase_gtrans * gphase; + + if (equal(gphase.real(), 1.0) && equal(gphase.imag(), 0)) + { + gphase = std::complex(1.0, 0.0); + } + + // pull this orbit member back to the representative frame: multiply by gphase + // (removes the translation/phase, as in the scalar version) then by W(g)^T. + const int isym = isymflag[g_index][c_index]; + const ModuleBase::Matrix3& W = wspin[isym]; + const std::complex vx = rhogtot_x[ipw0] * gphase; + const std::complex vy = rhogtot_y[ipw0] * gphase; + const std::complex vz = rhogtot_z[ipw0] * gphase; + const double trs_sign = trs_invp[isym]; + sum_x += trs_sign * (W.e11 * vx + W.e21 * vy + W.e31 * vz); // trs_inv * (W^T v)_x + sum_y += trs_sign * (W.e12 * vx + W.e22 * vy + W.e32 * vz); // trs_inv * (W^T v)_y + sum_z += trs_sign * (W.e13 * vx + W.e23 * vy + W.e33 * vz); // trs_inv * (W^T v)_z + + gphase_record[rot_count]=gphase; + ipw_record[rot_count]=ipw0; + ixyz_record[rot_count]=ixyz0; + sym_record[rot_count]=isym; + ++rot_count; + }//end if section + }//end c_index loop + if (rot_count!=0) + { + sum_x/= rot_count; + sum_y/= rot_count; + sum_z/= rot_count; + } + for (int ir = 0; ir < rot_count; ++ir) + { + // push the representative-frame value back out to this member: W(g) * S / gphase. + const ModuleBase::Matrix3& W = wspin[sym_record[ir]]; + const std::complex inv_gphase = 1.0 / gphase_record[ir]; + const double trs_sign = trs_invp[sym_record[ir]]; + rhogtot_x[ipw_record[ir]] = trs_sign * (W.e11 * sum_x + W.e12 * sum_y + W.e13 * sum_z) * inv_gphase; + rhogtot_y[ipw_record[ir]] = trs_sign * (W.e21 * sum_x + W.e22 * sum_y + W.e23 * sum_z) * inv_gphase; + rhogtot_z[ipw_record[ir]] = trs_sign * (W.e31 * sum_x + W.e32 * sum_y + W.e33 * sum_z) * inv_gphase; + } + + delete[] ipw_record; + delete[] ixyz_record; + delete[] sym_record; + delete[] gphase_record; + }//end g_index loop + + delete[] symflag; + delete[] isymflag; + delete[] table_xyz; + delete[] count_xyz; + ModuleBase::timer::end("Symmetry","rhog_symmetry_nspin4"); } diff --git a/source/source_cell/module_symmetry/symm_rotation_k.cpp b/source/source_cell/module_symmetry/symm_rotation_k.cpp new file mode 100644 index 00000000000..f645419b05f --- /dev/null +++ b/source/source_cell/module_symmetry/symm_rotation_k.cpp @@ -0,0 +1,602 @@ +#include "symm_rotation_k.h" +#include "source_base/constants.h" +#include +#include "source_base/parallel_reduce.h" +#include "source_base/parallel_global.h" +#include "source_base/module_external/scalapack_connector.h" +#include "source_base/module_external/blas_connector.h" +#include "source_base/tool_title.h" +#include "source_base/timer.h" + +namespace ModuleSymmetry +{ + std::vector Symmetry_rotation_k::get_bvk_cells(const TC& period) + { + std::vector cells; + cells.reserve(static_cast(period[0]) * period[1] * period[2]); + for (int ix = 0; ix < period[0]; ++ix) { + for (int iy = 0; iy < period[1]; ++iy) { + for (int iz = 0; iz < period[2]; ++iz) { + cells.push_back({ix, iy, iz}); + } } } + return cells; + } + + void Symmetry_rotation_k::cal_Ms(const K_Vectors& kv, + const UnitCell& ucell, const Parallel_2D& pv, const int nspin) + { + ModuleBase::TITLE("Symmetry_rotation_k", "cal_Ms"); + ModuleBase::timer::start("Symmetry_rotation_k", "cal_Ms"); + + this->nspin_ = nspin; + this->nsym_ = ucell.symm.nrotk; + this->nanti_ = ucell.symm.nrotk_anti; + this->magnetic_nspin4_ = ucell.symm.magnetic_nspin4; + this->eps_ = ucell.symm.epsilon; + if (this->irs_.invmap_.empty()) + { + this->irs_.invmap_.resize(ucell.symm.nrotk); + ucell.symm.gmatrix_invmap(ucell.symm.gmatrix, ucell.symm.nrotk, this->irs_.invmap_.data()); + } + // 1. calculate the rotation matrix in real spherical harmonics representation for each symmetry operation: [T_l (isym)]_mm' + const int nop_tot = this->nsym_ + this->nanti_; + std::vector gmatc(nop_tot); + for (int i = 0;i < nsym_;++i) { gmatc[i] = this->irs_.direct_to_cartesian(ucell.symm.gmatrix[i], ucell.latvec); } + for (int j = 0;j < this->nanti_;++j) + { gmatc[nsym_ + j] = this->irs_.direct_to_cartesian(ucell.symm.gmatrix_anti[j], ucell.latvec); } + this->cal_rotmat_Slm(gmatc.data(), std::max(this->abfs_Lmax_, ucell.lmax), nop_tot); + + // 1.5 (nspin=4) the SU(2) spin-1/2 rotation U(isym) for each symmetry operation. The AO + // rotation matrix M becomes the spinor operator T(isym) (x) U(isym) so that the same + // gemm D(k)=M^dagger D(k_ibz) M rotates both the orbital and the spin part at once. + // For an antiunitary element Theta*g only the spatial part g enters M here; the Theta + // (sigma_y (.)^* sigma_y) is applied afterwards in restore_dm. + std::vector spin_U(nop_tot, SpinRotation::Su2{ 1.0, 0.0, 0.0, 1.0 }); + if (this->nspin_ == 4) + { + for (int i = 0;i < nop_tot;++i) { spin_U[i] = SpinRotation::so3_to_su2(gmatc[i]); } + } + this->spin_U_ = spin_U; // keep for restore_HR_nspin4 (real-space EXX H(R) spin mixing) + + // 2. calculate the rotation matrix in AO-representation for each ibz_kpoint and symmetry operation: M(k, isym) + int nks_ibz = kv.kstars.size(); // kv.nks = 2 * kv.nks_ibz when nspin=2 + this->Ms_.assign(nks_ibz, {}); + this->little_groups_.assign(nks_ibz, {}); + + // (k-point pools, KPAR>1) kv.kvec_d only holds the k-points owned by this pool, so + // kv.kvec_d[ik_ibz] is only valid for ik_ibz < kv.para_k.nks_np and is otherwise either + // out of range or (after a caller's spin-doubling resize) a meaningless zero placeholder. + // Gather the (small, size nks_ibz) global ibz-representative k-vector list once so every + // pool builds the correct rotation matrix for every ibz k, not just the ones it owns. +#ifdef __MPI + // inlined equivalent of Parallel_Kpoints::gatherkvec (avoided as a direct call so this + // class does not pull in a link dependency on parallel_kpoints.cpp for every target that + // links the "symmetry" library): every rank in the owning pool holds the same local + // k-vectors, so only the pool root contributes to the MPI_Allreduce, matching gatherkvec. + int world_rank = 0; + MPI_Comm_rank(MPI_COMM_WORLD, &world_rank); + const bool is_pool_root = (world_rank == kv.para_k.get_startpro_pool(kv.para_k.my_pool)); + std::vector> kvec_d_ibz_global(nks_ibz, ModuleBase::Vector3(0.0, 0.0, 0.0)); + for (int i = 0; i < kv.para_k.nks_np; ++i) + { + if (is_pool_root) { kvec_d_ibz_global[i + kv.para_k.startk_pool[kv.para_k.my_pool]] = kv.kvec_d[i]; } + } + MPI_Allreduce(MPI_IN_PLACE, kvec_d_ibz_global.data(), 3 * nks_ibz, MPI_DOUBLE, MPI_SUM, MPI_COMM_WORLD); +#else + const std::vector>& kvec_d_ibz_global = kv.kvec_d; +#endif + + // A k-star contains only one operation per distinct k point. The other + // operations fixing k (modulo a reciprocal lattice vector) must still + // be averaged: a finite-grid SCF density need not respect this little group. + for (int ik_ibz = 0; ik_ibz < nks_ibz; ++ik_ibz) + { + std::set needed; + for (const std::pair>& member : kv.kstars[ik_ibz]) + { + const int op = (!this->magnetic_nspin4_ && member.first >= nsym_) + ? member.first - nsym_ : member.first; + needed.insert(op); + } + for (int op = 0; op < nsym_; ++op) + { + const ModuleBase::Vector3 delta = kvec_d_ibz_global[ik_ibz] * ucell.symm.kgmatrix[op] - kvec_d_ibz_global[ik_ibz]; + if (std::abs(delta.x - std::round(delta.x)) < this->eps_ + && std::abs(delta.y - std::round(delta.y)) < this->eps_ + && std::abs(delta.z - std::round(delta.z)) < this->eps_) + { + this->little_groups_[ik_ibz].push_back(op); + needed.insert(op); + } + } + for (const int op : needed) + { + this->Ms_[ik_ibz][op] = this->contruct_2d_rot_mat_ao( + ucell.symm, ucell.atoms, ucell.st, kvec_d_ibz_global[ik_ibz], op, pv, spin_U[op]); + } + } + + ModuleBase::timer::end("Symmetry_rotation_k", "cal_Ms"); + } + + std::vector>> Symmetry_rotation_k::restore_dm(const K_Vectors& kv, + const std::vector>>& dm_k_ibz, const Parallel_2D& pv)const + { + ModuleBase::TITLE("Symmetry_rotation_k", "restore_dm"); + ModuleBase::timer::start("Symmetry_rotation_k", "restore_dm"); + std::vector>> dm_k_full; + int nspin0 = this->nspin_ == 2 ? 2 : 1; + // (k-point pools, KPAR>1) dm_k_ibz (module_dm::DensityMatrix::dmk) only ever holds + // the irreducible k-points owned by THIS pool (_nk = kv.get_nks()/nspin, see + // allocate_dm.cpp), never the global set -- so nk here must be the local count, and + // kv.kstars (which is global, identical on every pool) must be indexed via the + // local-to-global map kv.ik2iktot, not via the local loop variable directly. + // This is safe: D(k) -> D(R) (or, for DFT+U, the occupation matrix built from it) + // is a linear sum over k, so each pool returning only the stars of its own local + // irreducible k-points, to be combined by the caller's existing cross-pool + // reduction (e.g. compute_occ_from_dmr's Parallel_Reduce::reduce_all), gives the + // exact same total as if every pool held the full global k-set -- no pool needs + // (or has to pay for gathering) the complete global D(k) at any point. + int nk = kv.get_nks() / nspin0; + const int nks_ibz_global = static_cast(this->little_groups_.size()); + + // (nspin=4) Sigma_y = I (x) sigma_y for the time-reversal spin flip; k-independent, build once. + std::vector> sigma_y; + if (this->nspin_ == 4) { sigma_y = this->set_sigma_y_2d(pv); } + + for (int is = 0;is < nspin0;++is) + { + for (int ik_local = 0;ik_local < nk;++ik_local) + { + const int ik_ibz = kv.ik2iktot[ik_local + is * nk] % nks_ibz_global; + // P_k D = |G_k|^{-1} sum_g M_g^T D M_g^*. This preserves + // Hermiticity and makes restoration independent of the chosen + // star representative; rotating just one arbitrary D does not. + const std::vector& little_group = this->little_groups_.at(ik_ibz); + assert(!little_group.empty()); + std::vector> projected = dm_k_ibz[ik_local + is * nk]; + if (little_group.size() > 1) + { + std::fill(projected.begin(), projected.end(), 0.0); + for (const int op : little_group) + { + const std::vector> rotated = this->rot_matrix_ao( + dm_k_ibz[ik_local + is * nk], ik_ibz, little_group.size(), op, pv); + for (size_t i = 0; i < projected.size(); ++i) + { + projected[i] += rotated[i]; + } + } + } + for (const std::pair>& isym_kvd : kv.kstars[ik_ibz]) + { + if (isym_kvd.first == 0) + { + double factor = 1.0 / static_cast(kv.kstars[ik_ibz].size()); + std::vector> dm_scaled(pv.get_local_size()); + for (int i = 0;i < pv.get_local_size();++i) { dm_scaled[i] = factor * projected[i]; } + dm_k_full.push_back(dm_scaled); + } + else if (isym_kvd.first < nsym_) + { //space group operations + dm_k_full.push_back(this->rot_matrix_ao(projected, ik_ibz, kv.kstars[ik_ibz].size(), isym_kvd.first, pv)); + } + else + { // antiunitary elements: Theta * (spatial operation) + // D(Theta*g k_ibz) = sigma_y [D(g k_ibz)]^* sigma_y with D(g k_ibz) = M^dagger D M. + // For nspin=4, first do the (non-conjugated) spatial rotation, then the spin flip; + // for nspin<4 (Theta=K) the original TRS_conj path already gives the conjugate. + // + // Which spatial operation the index denotes depends on the regime, matching + // how the k-reduction filled kgmatrix[] (see K_Vectors::reduce_by_symmetry): + // - nspin=4 magnetic (Shubnikov): index j+nsym_ is the antiunitary element + // Theta*gmatrix_anti[j]; its Ms is stored under the RAW key j+nsym_. + // - otherwise (grey group / nspin<4): index i+nsym_ is Theta*gmatrix[i], + // i.e. the unitary operation i, whose Ms is stored under key i. + const int isym_M = this->magnetic_nspin4_ ? isym_kvd.first : (isym_kvd.first - nsym_); + if (this->nspin_ == 4) + { + // m=0: gray group: the space-group part of anti-unitary elements are the same of the unitary elements, isym_M < nsym_ + // m!=0: Shubnikov group: using different space-group part of anti-unitary elements stored in gmatrix_anti with isym_M >= nsym_ + dm_k_full.push_back(this->trs_spin_rotate( + this->rot_matrix_ao(projected, ik_ibz, kv.kstars[ik_ibz].size(), isym_M, pv, false), + sigma_y, pv, 1.0)); + } + else + { + dm_k_full.push_back(this->rot_matrix_ao(projected, ik_ibz, kv.kstars[ik_ibz].size(), isym_M, pv, true)); + } + } + } + } + } + ModuleBase::timer::end("Symmetry_rotation_k", "restore_dm"); + return dm_k_full; + } + std::vector> Symmetry_rotation_k::restore_dm(const K_Vectors& kv, + const std::vector>& dm_k_ibz, const Parallel_2D& pv)const + { + return dm_k_ibz;// do nothing for gamma_only + } + + // calculate Wigner D matrix + double Symmetry_rotation_k::wigner_d(const double beta, const int l, const int m1, const int m2) const + { + auto factorial = [](int n) -> int { + int result = 1; + for (int i = 1;i <= n;++i) { result *= i; +} + return result; + }; + double result = 0.0; + for (int i = std::max(0, m2 - m1);i <= std::min(l - m1, l + m2);++i) { + result += std::pow(-1, i) * std::sqrt(factorial(l + m1) * factorial(l - m1) * factorial(l + m2) * factorial(l - m2)) + * std::pow(std::cos(beta / 2), 2 * l + m2 - m1 - 2 * i) * std::pow(-std::sin(beta / 2), m1 - m2 + 2 * i) + / (factorial(i) * factorial(l - m1 - i) * factorial(l + m2 - i) * factorial(i - m2 + m1)); +} + return result; + } + + std::complex Symmetry_rotation_k::wigner_D(const TCdouble& euler_angle, const int l, const int m1, const int m2, const bool inv) const + { + std::complex prefac(inv ? std::pow(-1, l) : 1, 0); + return std::exp(-ModuleBase::IMAG_UNIT * static_cast(m1) * euler_angle.x) + * std::exp(-ModuleBase::IMAG_UNIT * static_cast(m2) * euler_angle.z) + * wigner_d(euler_angle.y, l, m1, m2) * prefac; + } + + // c^l_{m1, m2}= + std::complex Symmetry_rotation_k::ovlp_Ylm_Slm(const int l, const int m1, const int m2) const + { + if (m1 == m2) + { + if (m1 == 0) { return 1.0; +} + if (m1 > 0) { return 1 / std::sqrt(2); +} + if (m1 < 0) { return std::pow(-1, m1) * ModuleBase::IMAG_UNIT / std::sqrt(2); +} + } + else if (m1 == -m2) + { + if (m1 > 0) { return -ModuleBase::IMAG_UNIT / std::sqrt(2); +} + if (m1 < 0) { return std::pow(-1, m1) / std::sqrt(2); +} + } + return 0.0; + } + + // reference: https://github.com/minyez/abf_trans/blob/f9e68e68069a94610d89e077bfe6e8ffac0b097d/src/rotate.cpp#L118 + // because the atom position here is row vector, the original gmatrix(eular angle) is transposed. + // gmatc: the rotation matrix under the basis of cartesian coordinates + // gmatc should be a rotation matrix, i.e. det(gmatc)=1 + TCdouble Symmetry_rotation_k::get_euler_angle(const ModuleBase::Matrix3& gmatc) const + { + double threshold = this->eps_; + double alpha = 0.0, beta = 0.0, gamma = 0.0; + if (std::fabs(gmatc.e32) > threshold || std::fabs(gmatc.e31) > threshold) // sin(beta) is not zero + { + // use the 2-angle elements to get alpha and gamma + alpha = std::atan2(gmatc.e32, gmatc.e31); + if (alpha < 0) { alpha += 2 * ModuleBase::PI; +} + gamma = std::atan2(gmatc.e23, -gmatc.e13); + if (gamma < 0) { gamma += 2 * ModuleBase::PI; +} + // use the larger one of 2-angle elements to calculate beta + if (std::fabs(gmatc.e32) > std::fabs(gmatc.e31)) { + beta = std::atan2(gmatc.e32 / std::sin(alpha), gmatc.e33); + } else { + beta = std::atan2(gmatc.e31 / std::cos(alpha), gmatc.e33); +} + } + else + {//sin(beta)=0, beta = 0 or pi, only (alpha+gamma) or (alpha-gamma) is important. now assign this to alpha. + alpha = std::atan2(gmatc.e12, gmatc.e11); + if (alpha < 0) { alpha += 2 * ModuleBase::PI; +} + // if beta=0, gmatc.e11=cos(alpha+gamma), gmatc.e21=sin(alpha+gamma) + // if beta=pi, gmatc.e11=cos(pi+alpha-gamma), gmatc.e21=sin(pi+alpha-gamma) + if (gmatc.e33 > 0) + { + beta = 0; + gamma = 0; //alpha+gamma=alpha => gamma=0 + } + else + { + beta = ModuleBase::PI; + gamma = ModuleBase::PI;// pi+alpha-gamma=alpha => gamma=pi + } + } + return TCdouble(alpha, beta, gamma); + } + + // in: the real value of m in range {-l, -l+1, ..., 0, ..., l-1, l} + // out: the index of the orbital in a fixed {n， l}, i.e. the index in array [0, 1, -1, 2, -2, ...] + inline int m2im_k(int m) + { + return (m > 0 ? 2 * m - 1 : -2 * m); + } + + /// T_mm' = [c^\dagger D c]_mm' + void Symmetry_rotation_k::cal_rotmat_Slm(const ModuleBase::Matrix3* gmatc, const int lmax, const int nop) + { + ++this->rotmat_Slm_version_; + const int nop_tot = (nop < 0) ? this->nsym_ : nop; + this->rotmat_Slm_.resize(nop_tot); + // c matrix is independent on isym + std::vector c_mm(lmax + 1); + for (int l = 0;l <= lmax;++l) { + c_mm[l].create(2 * l + 1, 2 * l + 1); +} + for (int l = 0;l <= lmax;++l) { + for (int m1 = -l;m1 <= l;++m1) { + for (int m2 = -l;m2 <= l;++m2) { + c_mm[l](m2im_k(m1), m2im_k(m2)) = ovlp_Ylm_Slm(l, m1, m2); +} +} +} + + for (int isym = 0;isym < nop_tot;++isym) + { + // if R is a reflection operation, calculate D^l(R)=(-1)^l*D^l(IR), so the euler angle of (IR) is needed. + TCdouble euler_angle = get_euler_angle(gmatc[isym].Det() > 0 ? + gmatc[isym] : gmatc[isym] * ModuleBase::Matrix3(-1, 0, 0, 0, -1, 0, 0, 0, -1)); + + this->rotmat_Slm_[isym].resize(lmax + 1); + for (int l = 0;l <= lmax;++l) + {// wigner D matrix + ModuleBase::ComplexMatrix D_mm(2 * l + 1, 2 * l + 1); + for (int m1 = -l;m1 <= l;++m1) { + for (int m2 = -l;m2 <= l;++m2) { + D_mm(m2im_k(m1), m2im_k(m2)) = wigner_D(euler_angle, l, m1, m2, (gmatc[isym].Det() < 0)); +} +} + this->rotmat_Slm_[isym][l] = transpose(c_mm[l], true) * D_mm * c_mm[l]; + } + } + } + + void Symmetry_rotation_k::set_block_to_mat2d(const int starti, const int startj, const ModuleBase::ComplexMatrix& block, + std::vector>& obj_mat, const Parallel_2D& pv, const bool trans) const + { // caution: ComplaxMatrix is row-major(col-continuous), but obj_mat is col-major(row-continuous) + for (int j = 0;j < block.nr;++j) {//outside dimension + for (int i = 0;i < block.nc;++i) { //inside dimension + if (pv.in_this_processor(starti + i, startj + j)) + { + int index = pv.global2local_col(startj + j) * pv.get_row_size() + pv.global2local_row(starti + i); + obj_mat[index] = trans ? block(i, j) : block(j, i); + } +} +} + } + + void Symmetry_rotation_k::set_block_to_mat2d(const int starti, const int startj, const ModuleBase::ComplexMatrix& block, + std::vector& obj_mat, const Parallel_2D& pv, const bool trans) const + { // caution: ComplaxMatrix is row-major(col-continuous), but obj_mat is col-major(row-continuous) + for (int j = 0;j < block.nr;++j) {//outside dimension + for (int i = 0;i < block.nc;++i) { //inside dimension + if (pv.in_this_processor(starti + i, startj + j)) + { + int index = pv.global2local_col(startj + j) * pv.get_row_size() + pv.global2local_row(starti + i); + obj_mat[index] = trans ? block(i, j).real() : block(j, i).real(); + } +} +} + } + + // 2d-block parallized rotation matrix in AO-representation, denoted as M. + // finally we will use D(k)=M(R, k)^\dagger*D(Rk)*M(R, k) to D(k) from D(Rk) in cal_Ms. + std::vector> Symmetry_rotation_k::contruct_2d_rot_mat_ao(const Symmetry& symm, const Atom* atoms, const Statistics& cell_st, + const TCdouble& kvec_d_ibz, int isym, const Parallel_2D& pv, const SpinRotation::Su2& spin_U) const + { + const bool soc = (this->nspin_ == 4); + const int npol = soc ? 2 : 1; // spinor: global AO index is spin-fast interleaved, I = npol*iw_orb + s + std::vector> M_isym(pv.get_local_size(), 0.0); + // isym >= symm.nrotk addresses the antiunitary coset (spatial part gmatrix_anti[isym-nrotk]), + // whose atom map lives in a separate table. + const int nrotk_u = symm.nrotk; + auto rotated_atom = [&symm, nrotk_u](const int is, const int iat) -> int + { + return (is < nrotk_u) ? symm.get_rotated_atom(is, iat) + : symm.get_rotated_atom_anti(is - nrotk_u, iat); + }; + for (int iat1 = 0;iat1 < cell_st.nat;++iat1) + { + int it = cell_st.iat2it[iat1]; // it1=it2 + int ia1 = cell_st.iat2ia[iat1]; + int iat2 = rotated_atom(isym, iat1); //iat2=rot(iat1) + int ia2 = cell_st.iat2ia[iat2]; + // cal phase factor from return lattice: exp(-ik_ibz*O) + double arg = -2 * ModuleBase::PI * kvec_d_ibz * this->irs_.return_lattice_[iat1][isym]; + std::complexphase_factor = std::complex(std::cos(arg), std::sin(arg)); + int iw1start = atoms[it].stapos_wf + ia1 * atoms[it].nw; + int iw2start = atoms[it].stapos_wf + ia2 * atoms[it].nw; + int iw = 0; + while (iw < atoms[it].nw) + { + int l = atoms[it].iw2l[iw]; + int nm = 2 * l + 1; + //caution: the order of m in orbitals may be different from increasing + if (!soc) + { + set_block_to_mat2d(iw2start + iw, iw1start + iw, + phase_factor * this->rotmat_Slm_[isym][l], M_isym, pv, true); + } + else + { + // M = T(isym) (x) U(isym): scatter phase * T_l(m,m') * U(a,b) to the interleaved + // spinor positions (row = rotated atom/spin, col = original atom/spin). For nspin=4 + // stapos_wf already carries the npol factor, so the per-atom offset is ia*nw*npol + // and the within-atom spinor index is (iw_orb)*npol + spin (spin is the fast index). + const int base2 = atoms[it].stapos_wf + ia2 * atoms[it].nw * npol; + const int base1 = atoms[it].stapos_wf + ia1 * atoms[it].nw * npol; + const ModuleBase::ComplexMatrix& Tl = this->rotmat_Slm_[isym][l]; + for (int m = 0;m < nm;++m) + { + for (int mp = 0;mp < nm;++mp) + { + const std::complex t = phase_factor * Tl(m, mp); + for (int a = 0;a < npol;++a) + { + for (int b = 0;b < npol;++b) + { + const int gi = base2 + (iw + m) * npol + a; + const int gj = base1 + (iw + mp) * npol + b; + if (pv.in_this_processor(gi, gj)) + { + const int index = pv.global2local_col(gj) * pv.get_row_size() + pv.global2local_row(gi); + // M(isym) = T_l (x) U is the spinor rep, with U = so3_to_su2 placed as-is: + // M[(m,a),(m',b)] = phase * T_l(m,m') * U_{ab}, U_{ab} = spin_U[a*npol + b]. + // Both T_l (rotmat_Slm) and U are ANTI-homomorphisms here (row-vector / R^T convention: + // rotmat_Slm(g)=R_orb(g)^{-1}, so3_to_su2 likewise), so this M is a consistent rep + // and rot_matrix_ao's stored-DM rotation M^T D M^* is exact for ALL ops. + M_isym[index] = t * spin_U[a * npol + b]; + } + } + } + } + } + } + iw += nm; + } + } + return M_isym; + } + + // D(k) = M^T(R, k) D(k_ibz) M^*(R, k), if D(k) is col-maj + // D^T(k) = M^\dagger(R, k) D^T(k_ibz) M(R, k), if D(k) is row-maj + // Ds from RI_2D_Comm are row-maj + // the link ik_ibz-isym-ik can be found in kstars. + std::vector> Symmetry_rotation_k::rot_matrix_ao(const std::vector>& DMkibz, + const int ik_ibz, const int kstar_size, const int isym, const Parallel_2D& pv, const bool TRS_conj) const + { + std::vector> DMk(pv.nloc, 0.0); + std::vector> DMkibz_M(pv.nloc, 0.0); // intermediate result + const char dagger = 'C'; + const char transpose = 'T'; + const char notrans = 'N'; + std::complex alpha(1.0, 0.0); + const std::complex beta(0.0, 0.0); + const int nbasis = pv.get_global_row_size(); + const int i1 = 1; + if (TRS_conj) + { + // D^T* = M^T [M^T (D^T)^T]^\dagger +#ifdef __MPI + ScalapackConnector::gemm(transpose, transpose, nbasis, nbasis, nbasis, + alpha, this->Ms_[ik_ibz].at(isym).data(), i1, i1, pv.desc, DMkibz.data(), i1, i1, pv.desc, + beta, DMkibz_M.data(), i1, i1, pv.desc); +#else + // without MPI, pv holds the whole (non-block-cyclic) dense matrix locally, + // so the 2D-block-cyclic pdgemm/pzgemm degenerates to a plain col-major gemm. + BlasConnector::gemm_cm(transpose, transpose, nbasis, nbasis, nbasis, + alpha, this->Ms_[ik_ibz].at(isym).data(), nbasis, DMkibz.data(), nbasis, + beta, DMkibz_M.data(), nbasis); +#endif + alpha.real(1.0 / static_cast(kstar_size)); +#ifdef __MPI + ScalapackConnector::gemm(transpose, dagger, nbasis, nbasis, nbasis, + alpha, this->Ms_[ik_ibz].at(isym).data(), i1, i1, pv.desc, DMkibz_M.data(), i1, i1, pv.desc, + beta, DMk.data(), i1, i1, pv.desc); +#else + BlasConnector::gemm_cm(transpose, dagger, nbasis, nbasis, nbasis, + alpha, this->Ms_[ik_ibz].at(isym).data(), nbasis, DMkibz_M.data(), nbasis, + beta, DMk.data(), nbasis); +#endif + } + else + { + // Physical DM rotation D(k) = M^dagger D(k_ibz) M, with M = T (x) U is the anti-homomorphism rep in row-major convention. + // ABACUS stores the DM transposed (S = D^T), for which this becomes S(gk) = M^T S(k_ibz) M^* = (conj M)^dagger S (conj M) + // For nspin<4 the orbital-only M is real, so Mc = M and this is bit-identical to the old M^dagger D M. + const std::vector>& Mref = this->Ms_[ik_ibz].at(isym); + std::vector> Mc(Mref.size()); + for (size_t i = 0; i < Mref.size(); ++i) { Mc[i] = std::conj(Mref[i]); } +#ifdef __MPI + ScalapackConnector::gemm(dagger, notrans, nbasis, nbasis, nbasis, + alpha, Mc.data(), i1, i1, pv.desc, DMkibz.data(), i1, i1, pv.desc, + beta, DMkibz_M.data(), i1, i1, pv.desc); +#else + BlasConnector::gemm_cm(dagger, notrans, nbasis, nbasis, nbasis, + alpha, Mc.data(), nbasis, DMkibz.data(), nbasis, + beta, DMkibz_M.data(), nbasis); +#endif + alpha.real(1.0 / static_cast(kstar_size)); +#ifdef __MPI + ScalapackConnector::gemm(notrans, notrans, nbasis, nbasis, nbasis, + alpha, DMkibz_M.data(), i1, i1, pv.desc, Mc.data(), i1, i1, pv.desc, + beta, DMk.data(), i1, i1, pv.desc); +#else + BlasConnector::gemm_cm(notrans, notrans, nbasis, nbasis, nbasis, + alpha, DMkibz_M.data(), nbasis, Mc.data(), nbasis, + beta, DMk.data(), nbasis); +#endif + } + return DMk; + } + + std::vector> Symmetry_rotation_k::set_sigma_y_2d(const Parallel_2D& pv) const + { + std::vector> sigma_y(pv.get_local_size(), 0.0); + const int nlocal = pv.get_global_row_size(); // = 2*nao for nspin=4 + // sigma_y = [[0, -i], [i, 0]] on the interleaved spin index (I = 2*iorb + spin) + const std::complex sy[2][2] = { {std::complex(0.0, 0.0), std::complex(0.0, -1.0)}, + {std::complex(0.0, 1.0), std::complex(0.0, 0.0)} }; + for (int iorb = 0; 2 * iorb < nlocal; ++iorb) + { + for (int a = 0; a < 2; ++a) + { + const int b = 1 - a; // only the off-diagonal spin entries are non-zero + const int gi = 2 * iorb + a; + const int gj = 2 * iorb + b; + if (pv.in_this_processor(gi, gj)) + { + const int index = pv.global2local_col(gj) * pv.get_row_size() + pv.global2local_row(gi); + sigma_y[index] = sy[a][b]; + } + } + } + return sigma_y; + } + + std::vector> Symmetry_rotation_k::trs_spin_rotate(const std::vector>& X, + const std::vector>& sigma_y, const Parallel_2D& pv, const double scale) const + { + // stored (transposed 2d-block) form of D_new = sigma_y * conj(D) * sigma_y is + // Sigma_y * conj(X) * Sigma_y (Sigma_y^T = -Sigma_y, the two minus signs cancel). + const char notrans = 'N'; + const int nbasis = pv.get_global_row_size(); + const int i1 = 1; + const std::complex one(1.0, 0.0); + const std::complex beta(0.0, 0.0); + std::vector> Xc(X.size()); + for (size_t i = 0; i < X.size(); ++i) { Xc[i] = std::conj(X[i]); } + std::vector> tmp(pv.get_local_size(), 0.0); + std::vector> out(pv.get_local_size(), 0.0); + // tmp = Sigma_y * conj(X) +#ifdef __MPI + ScalapackConnector::gemm(notrans, notrans, nbasis, nbasis, nbasis, + one, sigma_y.data(), i1, i1, pv.desc, Xc.data(), i1, i1, pv.desc, + beta, tmp.data(), i1, i1, pv.desc); +#else + // without MPI, pv holds the whole (non-block-cyclic) dense matrix locally, + // so the 2D-block-cyclic pzgemm degenerates to a plain col-major gemm. + BlasConnector::gemm_cm(notrans, notrans, nbasis, nbasis, nbasis, + one, sigma_y.data(), nbasis, Xc.data(), nbasis, + beta, tmp.data(), nbasis); +#endif + // out = scale * tmp * Sigma_y +#ifdef __MPI + ScalapackConnector::gemm(notrans, notrans, nbasis, nbasis, nbasis, + std::complex(scale, 0.0), tmp.data(), i1, i1, pv.desc, sigma_y.data(), i1, i1, pv.desc, + beta, out.data(), i1, i1, pv.desc); +#else + BlasConnector::gemm_cm(notrans, notrans, nbasis, nbasis, nbasis, + std::complex(scale, 0.0), tmp.data(), nbasis, sigma_y.data(), nbasis, + beta, out.data(), nbasis); +#endif + return out; + } +} diff --git a/source/source_cell/module_symmetry/symm_rotation_k.h b/source/source_cell/module_symmetry/symm_rotation_k.h new file mode 100644 index 00000000000..48ee839f4ff --- /dev/null +++ b/source/source_cell/module_symmetry/symm_rotation_k.h @@ -0,0 +1,177 @@ +#ifndef SYMM_ROTATION_K_H +#define SYMM_ROTATION_K_H +#include "irreducible_sector.h" +#include "source_base/parallel_2d.h" +#include "source_base/complexmatrix.h" +#include "source_cell/module_symmetry/symm_rot_spin.h" + +namespace ModuleSymmetry +{ + /// @brief k-space AO-representation symmetry restoration: reconstructs D(k) at every + /// k-star member from D(k_ibz), for crystal-symmetry-reduced BZ sampling. + /// + /// This is the LibRI-independent subset of what used to be a single + /// source_lcao/module_ri/module_exx_symmetry/Symmetry_rotation class: everything needed + /// to go from an irreducible-k-point density matrix to the full-BZ one, so callers that + /// only need that (e.g. DFT+U's occupation-matrix/DMR restoration) do not have to depend + /// on module_ri or LibRI. EXX/RPA's own real-space H(R)/RI-coefficient restoration (which + /// does need RI::Tensor) is built on top of this class in + /// source_lcao/module_ri/module_exx_symmetry/symm_rotation.h (ModuleSymmetry::Symmetry_rotation, + /// which inherits from this one). + class Symmetry_rotation_k + { + public: + Symmetry_rotation_k() {}; + virtual ~Symmetry_rotation_k() {}; + + //-------------------------------------------------------------------------------- + // getters + const std::map>& get_irreducible_sector()const { return this->irs_.get_irreducible_sector(); } + TCdouble get_return_lattice(const Symmetry& symm, + const ModuleBase::Matrix3& gmatd, const TCdouble gtransd, + const TCdouble& posd_a1, const TCdouble& posd_a2)const + { + return this->irs_.get_return_lattice(symm, gmatd, gtransd, posd_a1, posd_a2); + } + TCdouble get_return_lattice(const int iat, const int isym) const + { + return this->irs_.get_return_lattice(iat, isym); + } + /// the rotation matrix under the basis of S_l^m. size: [nsym][lmax][nm*nm] + const std::vector>& rotmat_Slm = this->rotmat_Slm_; + const int& abfs_Lmax = this->abfs_Lmax_; + //-------------------------------------------------------------------------------- + // setters + void find_irreducible_sector(const Symmetry& symm, const Atom* atoms, const Statistics& st, + const std::vector& Rs, const TC& period, const Lattice& lat, const std::string& output_dir = "") + { + this->irs_.find_irreducible_sector(symm, atoms, st, Rs, period, lat, output_dir); + } + void set_abfs_Lmax(const int l) { this->abfs_Lmax_ = l; } + //-------------------------------------------------------------------------------- + /// functions to contruct rotation matrix in AO-representation + + /// The top-level calculation interface of this class. calculate the rotation matrix in AO representation: M + /// only need once call in each ion step (decided by the configuration) + /// @param kstars equal k points to each ibz-kpont, corresponding to a certain symmetry operations. + /// @param nspin stored as a member so restore_dm()/contruct_2d_rot_mat_ao() do not each + /// need to read the global nspin config setting (keeps this LibRI-free class + /// free of a module_parameter link dependency; every existing caller already + /// has nspin in scope). + void cal_Ms(const K_Vectors& kv, + const UnitCell& ucell, const Parallel_2D& pv, const int nspin); + + /// Use calculated M matrix to recover D(k) from D(k_ibz): D(k) = M(R, k)^\dagger D(k_ibz) M(R, k) + /// the link "ik_ibz-isym-ik" can be found in kstars: k_bz = gmat[isym](k) + std::vector>>restore_dm(const K_Vectors& kv, + const std::vector>>& dm_k_ibz, + const Parallel_2D& pv)const; + std::vector>restore_dm(const K_Vectors& kv, + const std::vector>& dm_k_ibz, + const Parallel_2D& pv)const; + std::vector> rot_matrix_ao(const std::vector>& DMkibz, + const int ik_ibz, const int kstar_size, const int isym, const Parallel_2D& pv, const bool TRS_conj = false) const; + + /// (nspin=4) build the 2*nao spin operator Sigma_y = I_nao (x) sigma_y in 2d-block layout. + std::vector> set_sigma_y_2d(const Parallel_2D& pv) const; + + /// (nspin=4) time-reversal on the spin density matrix: D(k) = sigma_y D^*(-k) sigma_y, + /// realized distribution-safely as scale * Sigma_y * conj(X) * Sigma_y (X is the already + /// space-group-rotated D(-k) stored in the transposed 2d-block convention). + std::vector> trs_spin_rotate(const std::vector>& X, + const std::vector>& sigma_y, const Parallel_2D& pv, const double scale) const; + + /// calculate Wigner D matrix + double wigner_d(const double beta, const int l, const int m1, const int m2) const; + std::complex wigner_D(const TCdouble& euler_angle, const int l, const int m1, const int m2, const bool inv) const; + + /// c^l_{m1, m2}= + std::complex ovlp_Ylm_Slm(const int l, const int m1, const int m2) const; + + /// calculate euler angle from rotation matrix + TCdouble get_euler_angle(const ModuleBase::Matrix3& gmatc) const; + + /// T_mm' = [c^\dagger D c]_mm', the rotation matrix in the representation of real sphere harmonics + /// @param nop number of operations in gmatc; <0 means nsym_ (the unitary ones only). + /// Pass nsym_+nanti_ to also build the antiunitary operations' T_l. + void cal_rotmat_Slm(const ModuleBase::Matrix3* gmatc, const int lmax, const int nop); + + /// set a block matrix onto a 2d-parallelized matrix(col-maj), at the position (starti, startj) + /// if trans=true, the block matrix is transposed before setting + void set_block_to_mat2d(const int starti, const int startj, const ModuleBase::ComplexMatrix& block, + std::vector>& obj_mat, const Parallel_2D& pv, const bool trans = false) const; + void set_block_to_mat2d(const int starti, const int startj, const ModuleBase::ComplexMatrix& block, + std::vector& obj_mat, const Parallel_2D& pv, const bool trans = false) const; + + /// 2d-block parallized rotation matrix in AO-representation, denoted as M. + /// finally we will use D(k)=M(R, k)^\dagger*D(Rk)*M(R, k) to recover D(k) from D(Rk). + std::vector> contruct_2d_rot_mat_ao(const Symmetry& symm, const Atom* atoms, const Statistics& cell_st, + const TCdouble& kvec_d_ibz, int isym, const Parallel_2D& pv, + const SpinRotation::Su2& spin_U /*= SpinRotation::Su2{ 1.0, 0.0, 0.0, 1.0 }*/) const; + + std::vector>& get_rotmat_Slm() { return this->rotmat_Slm_; } + + /// test-only: inject Ms_/little_groups_/nsym_ directly, bypassing cal_Ms(), so restore_dm() + /// can be unit-tested against synthetic k-stars without a real UnitCell/K_Vectors setup. + void set_density_rotations_for_testing(const std::vector>>>& Ms, + const std::vector>& little_groups, const int nsym, const int nspin) + { + this->Ms_ = Ms; + this->little_groups_ = little_groups; + this->nsym_ = nsym; + this->nspin_ = nspin; + } + + //-------------------------------------------------------------------------------- + /// list all cells in a Born-von-Karman supercell of the given period (no LibRI dependency, + /// unlike RI_Util::get_Born_von_Karmen_cells which this mirrors for 3D periods). + static std::vector get_bvk_cells(const TC& period); + + protected: + /// set by cal_Ms() (or set_density_rotations_for_testing()); avoids reading the global + /// nspin config setting in restore_dm()/contruct_2d_rot_mat_ao(), which would otherwise + /// pull a module_parameter link dependency into every target that links this LibRI-free class. + int nspin_ = 1; + + int nsym_ = 1; + /// (nspin=4, magnetic) number of ANTIUNITARY elements Theta*g of the Shubnikov group. + /// Their orbital rotations / return lattices / Ms are appended after the nsym_ unitary + /// ones, so the raw index isym in [nsym_, nsym_+nanti_) addresses gmatrix_anti[isym-nsym_]. + int nanti_ = 0; + /// (nspin=4) true when the configuration carries a non-zero local moment. Then pure time + /// reversal is NOT a symmetry (it reverses m) and the k-star must be restored with the + /// Shubnikov elements Theta*gmatrix_anti[] instead of the generic -k shortcut. + bool magnetic_nspin4_ = false; + + double eps_ = 1e-6; + + int abfs_Lmax_ = 0; + + /// the rotation matrix under the basis of S_l^m. size: [nsym][lmax][nm*nm] + std::vector> rotmat_Slm_; + + /// bumped every time cal_rotmat_Slm() (re)fills rotmat_Slm_, so derived classes caching a + /// converted copy of rotmat_Slm_ (e.g. EXX's RI::Tensor mirror) can detect staleness + /// without recomparing the whole matrix. + int rotmat_Slm_version_ = 0; + + /// The unitary matrix associate D(Rk) with D(k) for each ibz-kpoint Rk and each symmetry operation. + /// size: [nks_ibz][nsym][nbasis*nbasis], only need to calculate once. + std::vector>>> Ms_; + + /// The little group of each ibz-kpoint: the subset of unitary space-group operations that + /// fix kvec_d_ibz modulo a reciprocal lattice vector. D(k_ibz) is averaged over this group + /// before star-expansion, since a finite-grid SCF density need not exactly respect it. + /// size: [nks_ibz][<=nsym_], always non-empty (identity is always a member). + std::vector> little_groups_; + + /// (nspin=4) the SU(2) spin-1/2 rotation U(isym) for each symmetry operation, size [nsym]. + /// The spinor AO rotation is T(isym) (x) U(isym); restore_HR_nspin4 (EXX) uses it to mix + /// the 4 spin channels of the real-space H(R). Filled in cal_Ms (identity for nspin<4). + std::vector spin_U_; + + /// irreducible sector + Irreducible_Sector irs_; + }; +} +#endif // SYMM_ROTATION_K_H diff --git a/source/source_cell/module_symmetry/symmetry.cpp b/source/source_cell/module_symmetry/symmetry.cpp index e01e9cd97dc..c22774aeb02 100644 --- a/source/source_cell/module_symmetry/symmetry.cpp +++ b/source/source_cell/module_symmetry/symmetry.cpp @@ -116,6 +116,46 @@ void Symmetry::symmetrize_vec3_nat(double* v)const // pengfei 2016-12-20 return; } +void symmetrize_force_cartesian(Symmetry* symm, + const int nat, + const ModuleBase::Vector3& a1, + const ModuleBase::Vector3& a2, + const ModuleBase::Vector3& a3, + ModuleBase::matrix& force) +{ + ModuleBase::TITLE("Symmetry", "symmetrize_force_cartesian"); + if (symm == nullptr || nat <= 0 || force.nr < nat || force.nc < 3) + { + ModuleBase::WARNING_QUIT("symmetrize_force_cartesian", + "symm must be non-null and force must have at least nat x 3 elements"); + } + double d1; + double d2; + double d3; + for (int iat = 0; iat < nat; iat++) + { + ModuleBase::Mathzone::Cartesian_to_Direct(force(iat, 0), force(iat, 1), force(iat, 2), + a1.x, a1.y, a1.z, a2.x, a2.y, a2.z, + a3.x, a3.y, a3.z, d1, d2, d3); + + force(iat, 0) = d1; + force(iat, 1) = d2; + force(iat, 2) = d3; + } + symm->symmetrize_vec3_nat(force.c); + for (int iat = 0; iat < nat; iat++) + { + ModuleBase::Mathzone::Direct_to_Cartesian(force(iat, 0), force(iat, 1), force(iat, 2), + a1.x, a1.y, a1.z, a2.x, a2.y, a2.z, + a3.x, a3.y, a3.z, d1, d2, d3); + + force(iat, 0) = d1; + force(iat, 1) = d2; + force(iat, 2) = d3; + } + return; +} + void Symmetry::symmetrize_mat3(ModuleBase::matrix& sigma, const Lattice& lat)const //zhengdy added 2017 { ModuleBase::matrix A = lat.latvec.to_matrix(); diff --git a/source/source_cell/module_symmetry/symmetry.h b/source/source_cell/module_symmetry/symmetry.h index 2905b86da2a..e9308399448 100644 --- a/source/source_cell/module_symmetry/symmetry.h +++ b/source/source_cell/module_symmetry/symmetry.h @@ -377,6 +377,29 @@ class Symmetry : public Symmetry_Basic /// Non-magnetic (m_i=0) keeps all operations. void analyze_magnetic_group_nspin4(const Atom* atoms, const Statistics& st, const ModuleBase::Matrix3& latvec); }; + +/** + * @brief Symmetrize a per-atom Cartesian force vector in place. + * + * Convert Cartesian -> direct, apply symmetrize_vec3_nat, then convert back to + * Cartesian. The lattice vectors are passed explicitly by the caller (the + * current cell), because Symmetry::a1/a2/a3 may have been replaced by the + * symmetry-optimized lattice during the analysis. Shared by the PW and LCAO + * force paths. + * + * @param symm symmetry handler + * @param nat number of atoms + * @param a1 first lattice vector of the current cell + * @param a2 second lattice vector of the current cell + * @param a3 third lattice vector of the current cell + * @param force per-atom Cartesian forces, nat x 3, modified in place + */ +void symmetrize_force_cartesian(Symmetry* symm, + const int nat, + const ModuleBase::Vector3& a1, + const ModuleBase::Vector3& a2, + const ModuleBase::Vector3& a3, + ModuleBase::matrix& force); } #endif diff --git a/source/source_cell/module_symmetry/test/CMakeLists.txt b/source/source_cell/module_symmetry/test/CMakeLists.txt index 4401d65abf6..eaddcc85d9e 100644 --- a/source/source_cell/module_symmetry/test/CMakeLists.txt +++ b/source/source_cell/module_symmetry/test/CMakeLists.txt @@ -22,6 +22,11 @@ AddTest( LIBS parameter base ${math_libs} device symmetry SOURCES symm_rho_soc_test.cpp ${ABACUS_SOURCE_DIR}/source_estate/module_dm/density_matrix.cpp + ${ABACUS_SOURCE_DIR}/source_estate/module_dm/dmr_gamma.cpp + ${ABACUS_SOURCE_DIR}/source_estate/module_dm/dm_tools.cpp + ${ABACUS_SOURCE_DIR}/source_estate/module_dm/dmr_k.cpp + ${ABACUS_SOURCE_DIR}/source_estate/module_dm/dmr_td.cpp + ${ABACUS_SOURCE_DIR}/source_estate/module_dm/dmr_full.cpp ${ABACUS_SOURCE_DIR}/source_hamilt/module_hcontainer/base_matrix.cpp ${ABACUS_SOURCE_DIR}/source_hamilt/module_hcontainer/hcontainer.cpp ${ABACUS_SOURCE_DIR}/source_hamilt/module_hcontainer/atom_pair.cpp diff --git a/source/source_cell/module_symmetry/test/symm_rho_soc_test.cpp b/source/source_cell/module_symmetry/test/symm_rho_soc_test.cpp index 01258305d67..d850a3e1872 100644 --- a/source/source_cell/module_symmetry/test/symm_rho_soc_test.cpp +++ b/source/source_cell/module_symmetry/test/symm_rho_soc_test.cpp @@ -6,7 +6,7 @@ #include "../symmetry.h" #include "../symm_rot_spin.h" #include "source_cell/unitcell.h" -#include "source_estate/module_dm/density_matrix.h" // real func_xyz_to_updown +#include "source_estate/module_dm/density_matrix.h" // real xyz_to_updown /************************************************ * unit test of Symmetry::rhog_symmetry_nspin4 @@ -190,8 +190,8 @@ TEST(RhogSymmetrySoc, GroupInvariance) // --------------------------------------------------------------------------- // Coupling test (nonzero m_y): the spin-density rotation W=spin_so3 used by psymmg_soc for the // grid symmetrization MUST agree with the SU(2) rotation of the physical spinor state followed by -// the REAL func_xyz_to_updown extraction (which reads the conj-first stored DM, DM=conj(P), and -// uses the bare +Im(ud)-Im(du)). This test now calls the actual func_xyz_to_updown rather than a +// the REAL xyz_to_updown extraction (which reads the conj-first stored DM, DM=conj(P), and +// uses the bare +Im(ud)-Im(du)). This test now calls the actual xyz_to_updown rather than a // local re-implementation, so the grid-rotation and DM-extraction conventions cannot drift apart // silently (it fails on the #7664 m_y flip). The self-referential GroupInvariance test above // cannot catch this because it uses the same wspin as its own oracle. @@ -204,20 +204,20 @@ ModuleSymmetry::SpinRotation::Su2 block_from_pauli(double r0, double mx, double { return {cd(r0 + mz, 0.0), cd(mx, -my), cd(mx, my), cd(r0 - mz, 0.0)}; } -// The runtime stores the DM conj-first (DM = conj(P), cal_dm_psi); this is what func_xyz_to_updown +// The runtime stores the DM conj-first (DM = conj(P), dm_from_psi); this is what xyz_to_updown // actually consumes. Given a physical block P, the stored block is its element-wise conjugate. ModuleSymmetry::SpinRotation::Su2 stored_dm_from_phys(const ModuleSymmetry::SpinRotation::Su2& P) { return {std::conj(P[0]), std::conj(P[1]), std::conj(P[2]), std::conj(P[3])}; } -// call the REAL func_xyz_to_updown on a 2x2 stored-DM block; return (m_x, m_y, m_z) +// call the REAL xyz_to_updown on a 2x2 stored-DM block; return (m_x, m_y, m_z) ModuleBase::Vector3 real_extract(const ModuleSymmetry::SpinRotation::Su2& Dstored) { const cd tmp[4] = {Dstored[0], Dstored[1], Dstored[2], Dstored[3]}; // {uu,ud,du,dd} const int col_size = 2; const int step_trace[4] = {0, 1, col_size, col_size + 1}; double out[4] = {0.0, 0.0, 0.0, 0.0}; // rho0/x/y/z written at icol=0 - elecstate::DensityMatrix_Tools::func_xyz_to_updown(tmp, 0, step_trace, out); + module_dm::xyz_to_updown(tmp, 0, step_trace, out); return ModuleBase::Vector3(out[step_trace[1]], out[step_trace[2]], out[step_trace[3]]); } } // namespace @@ -241,12 +241,12 @@ TEST(RhogSymmetrySoc, SpinConventionCoupling) EXPECT_NEAR(Wgrid.e31, Wpauli.e31, TOL) << "g=" << g; EXPECT_NEAR(Wgrid.e32, Wpauli.e32, TOL) << "g=" << g; EXPECT_NEAR(Wgrid.e33, Wpauli.e33, TOL) << "g=" << g; - // (2) End-to-end with the REAL func_xyz_to_updown, exactly the runtime data flow: - // physical block P(m) --conj--> stored DM (conj-first) --func_xyz_to_updown--> grid m. + // (2) End-to-end with the REAL xyz_to_updown, exactly the runtime data flow: + // physical block P(m) --conj--> stored DM (conj-first) --xyz_to_updown--> grid m. // Rotate the PHYSICAL block by the spinor SU(2) U (U P U^dagger, i.e. the physical state // rotation), conj to the stored block, extract again -> m'. psymmg_soc rotates the grid // components with Wgrid=spin_so3, so we must have m' == Wgrid * m. This catches any - // mismatch (e.g. the #7664 m_y flip) between func_xyz_to_updown and spin_so3. + // mismatch (e.g. the #7664 m_y flip) between xyz_to_updown and spin_so3. for (const auto& m : mtest) { const ModuleSymmetry::SpinRotation::Su2 P = block_from_pauli(2.0, m[0], m[1], m[2]); diff --git a/source/source_cell/module_symmetry/test/symm_test.h b/source/source_cell/module_symmetry/test/symm_test.h index b49091b665e..40ffd009c7d 100644 --- a/source/source_cell/module_symmetry/test/symm_test.h +++ b/source/source_cell/module_symmetry/test/symm_test.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_CELL_MODULE_SYMMETRY_TEST_SYMM_TEST_H +#define ABACUS_SOURCE_CELL_MODULE_SYMMETRY_TEST_SYMM_TEST_H + #include "source_base/mathzone.h" #include "source_cell/unitcell.h" #include "gtest/gtest.h" @@ -34,4 +36,6 @@ class SymmetryTest : public testing::Test std::ofstream ofs_running; void construct_ucell(stru_& stru); void ClearUcell(); -}; \ No newline at end of file +}; + +#endif // ABACUS_SOURCE_CELL_MODULE_SYMMETRY_TEST_SYMM_TEST_H diff --git a/source/source_cell/print_cell.cpp b/source/source_cell/print_cell.cpp index 62a47c7dc84..f98b3ab8a99 100644 --- a/source/source_cell/print_cell.cpp +++ b/source/source_cell/print_cell.cpp @@ -11,6 +11,7 @@ #include "source_cell/mdcell.h" #include "source_base/formatter.h" #include "source_base/tool_title.h" +#include "source_base/tool_quit.h" #include "source_base/global_variable.h" #include "source_base/output.h" @@ -94,12 +95,13 @@ namespace unitcell const std::string& fn, const std::string& header, const int& nspin, - const bool& direct, const bool& vel, const bool& magmom, const bool& orb, const bool& dpks_desc, - const int& iproc) + const int& iproc, + const ModuleBase::matrix& force, + const bool& has_force) { ModuleBase::TITLE("UnitCell","print_stru_file"); if (iproc != 0) @@ -136,45 +138,98 @@ namespace unitcell { str += "\nNUMERICAL_DESCRIPTOR\n" + ucell.descriptor_file + "\n"; } - // LATTICE_CONSTANT - str += "\nLATTICE_CONSTANT\n" + FmtCore::format("%-.10f", ucell.lat0) + " # in Bohr\n"; - // LATTICE_VECTORS - str += "\nLATTICE_VECTORS # in units of lat0\n"; - str += FmtCore::format("%24.16f%24.16f%24.16f\n", latvec.e11, latvec.e12, latvec.e13); - str += FmtCore::format("%24.16f%24.16f%24.16f\n", latvec.e21, latvec.e22, latvec.e23); - str += FmtCore::format("%24.16f%24.16f%24.16f\n", latvec.e31, latvec.e32, latvec.e33); + // LATTICE_CONSTANT: fixed to one Angstrom expressed in Bohr, so that the + // lattice vectors below can be written directly in Angstrom. + const double lat0_angstrom = 1.0 / ModuleBase::BOHR_TO_A; + str += "\nLATTICE_CONSTANT\n" + + FmtCore::format("%-.10f", lat0_angstrom) + + " # in Bohr (= 1 Angstrom); lattice vectors below are in Angstrom\n"; + // LATTICE_VECTORS: internal vectors are dimensionless multiples of ucell.lat0; + // multiply by ucell.lat0 * BOHR_TO_A to get the physical vectors in Angstrom. + const double lat_scale = ucell.lat0 * ModuleBase::BOHR_TO_A; + str += "\nLATTICE_VECTORS # in Angstrom\n"; + str += FmtCore::format("%.16f %.16f %.16f\n", + latvec.e11 * lat_scale, latvec.e12 * lat_scale, latvec.e13 * lat_scale); + str += FmtCore::format("%.16f %.16f %.16f\n", + latvec.e21 * lat_scale, latvec.e22 * lat_scale, latvec.e23 * lat_scale); + str += FmtCore::format("%.16f %.16f %.16f\n", + latvec.e31 * lat_scale, latvec.e32 * lat_scale, latvec.e33 * lat_scale); // ATOMIC_POSITIONS str += "\nATOMIC_POSITIONS\n"; - const std::string scale = direct? "Direct": "Cartesian"; int nat_ = 0; // counter iat, for printing out Mulliken magmom who is indexed by iat - str += scale + "\n"; + // Force output is controlled by the explicit has_force flag rather than + // inferred from matrix dimensions, so a zero-allocated matrix is not + // mistaken for computed forces. + if (has_force && (force.nr != ucell.nat || force.nc != 3)) + { + ModuleBase::WARNING_QUIT("print_stru_file", + "has_force is true but force matrix dimensions do not match nat x 3"); + } + // Positions are always written as Cartesian_angstrom so that the + // coordinate format is independent of force availability; has_force + // only controls the per-atom f fields and the unit annotation. + const std::string unit_note = has_force ? " # positions in Angstrom, forces in eV/Angstrom\n" + : " # positions in Angstrom\n"; + str += "Cartesian_angstrom" + unit_note; + // Internal Cartesian tau is in units of lat0 (Bohr); convert to Angstrom. + const double pos_conv = ucell.lat0 * ModuleBase::BOHR_TO_A; + const double force_conv = ModuleBase::Ry_to_eV / ModuleBase::BOHR_TO_A; // Ry/Bohr to eV/Angstrom for(int it = 0; it < ucell.ntype; it++) { str += "\n" + ucell.atoms[it].label + " #label\n"; - str += FmtCore::format("%-8.4f #magnetism\n", ucell.magnet.start_mag[it]); + // Output real initial magnetism: for nspin=2 use mag[0], for nspin=4 use norm of m_loc_[0] + double start_mag = ucell.magnet.start_mag[it]; + if (atoms[it].na > 0) { + if (nspin == 2) { + start_mag = atoms[it].mag[0]; + } else if (nspin == 4) { + start_mag = std::sqrt(std::pow(atoms[it].m_loc_[0].x, 2) + + std::pow(atoms[it].m_loc_[0].y, 2) + + std::pow(atoms[it].m_loc_[0].z, 2)); + } + } + str += FmtCore::format("%.4f #magnetism (default, overridden by per-atom mag below)\n", start_mag); str += FmtCore::format("%d #number of atoms\n", atoms[it].na); for(int ia = 0; ia < atoms[it].na; ia++) { // output position - const double& x = direct? atoms[it].taud[ia].x: atoms[it].tau[ia].x; - const double& y = direct? atoms[it].taud[ia].y: atoms[it].tau[ia].y; - const double& z = direct? atoms[it].taud[ia].z: atoms[it].tau[ia].z; - str += FmtCore::format("%20.10f%20.10f%20.10f", x, y, z); + const double& x = atoms[it].tau[ia].x; + const double& y = atoms[it].tau[ia].y; + const double& z = atoms[it].tau[ia].z; + str += FmtCore::format("%.10f %.10f %.10f", x*pos_conv, y*pos_conv, z*pos_conv); str += FmtCore::format(" m%2d%2d%2d", atoms[it].mbl[ia].x, atoms[it].mbl[ia].y, atoms[it].mbl[ia].z); if (vel) // output velocity { - str += FmtCore::format(" v%20.10f%20.10f%20.10f", atoms[it].vel[ia].x, atoms[it].vel[ia].y, atoms[it].vel[ia].z); + str += FmtCore::format(" v %.10f %.10f %.10f", atoms[it].vel[ia].x, atoms[it].vel[ia].y, atoms[it].vel[ia].z); + } + if (has_force) // output force + { + str += FmtCore::format(" f %.6f %.6f %.6f", + force(nat_, 0)*force_conv, + force(nat_, 1)*force_conv, + force(nat_, 2)*force_conv); } - if (nspin == 2 && magmom) // output magnetic information + if (nspin == 2) // output magnetic information { - str += FmtCore::format(" mag%8.4f", ucell.atom_mulliken[nat_][1]); + if (magmom && !ucell.atom_mulliken.empty()) { + str += FmtCore::format(" mag %.4f", ucell.atom_mulliken[nat_][1]); + } else { + str += FmtCore::format(" mag %.4f", atoms[it].mag[ia]); + } } - else if (nspin == 4 && magmom) // output magnetic information + else if (nspin == 4) // output magnetic information { - str += FmtCore::format(" mag%8.4f%8.4f%8.4f", - ucell.atom_mulliken[nat_][1], - ucell.atom_mulliken[nat_][2], - ucell.atom_mulliken[nat_][3]); + if (magmom && !ucell.atom_mulliken.empty()) { + str += FmtCore::format(" mag %.4f %.4f %.4f", + ucell.atom_mulliken[nat_][1], + ucell.atom_mulliken[nat_][2], + ucell.atom_mulliken[nat_][3]); + } else { + str += FmtCore::format(" mag %.4f %.4f %.4f", + atoms[it].m_loc_[ia].x, + atoms[it].m_loc_[ia].y, + atoms[it].m_loc_[ia].z); + } } str += "\n"; nat_++; diff --git a/source/source_cell/print_cell.h b/source/source_cell/print_cell.h index 458f2d74a30..1370629b0cd 100644 --- a/source/source_cell/print_cell.h +++ b/source/source_cell/print_cell.h @@ -8,6 +8,7 @@ #include "atom_spec.h" #include "source_cell/strumeta.h" #include "source_cell/unitcell.h" +#include "source_base/matrix.h" class MDCell; @@ -38,26 +39,32 @@ namespace unitcell * @param latvec lattice parameter vector [in] * @param fn STRU file name [in] * @param header comment lines written at the top of the file [in] + * @note Positions are always written in Cartesian_angstrom, independent of + * force availability, so that all STRU outputs share one coordinate + * format. has_force only controls the per-atom f fields. + * * @param nspin number of spin channels [in] - * @param direct true for direct coords, false for cartesian coords [in] * @param vel true for printing velocities [in] * @param magmom true for printing Mulliken population analysis produced magmom [in] * @param orb true for printing NUMERICAL_ORBITAL section [in] * @param dpks_desc true for printing NUMERICAL_DESCRIPTOR section [in] * @param iproc GlobalV::MY_RANK [in] + * @param force atomic forces in Ry/Bohr, used only when has_force is true [in] + * @param has_force true to emit per-atom f fields; force matrix must match nat [in] */ void print_stru_file(const UnitCell& ucell, const Atom* atoms, const ModuleBase::Matrix3& latvec, const std::string& fn, const std::string& header, - const int& nspin = 1, - const bool& direct = false, - const bool& vel = false, - const bool& magmom = false, - const bool& orb = false, - const bool& dpks_desc = false, - const int& iproc = 0); + const int& nspin, + const bool& vel, + const bool& magmom, + const bool& orb, + const bool& dpks_desc, + const int& iproc, + const ModuleBase::matrix& force, + const bool& has_force); /** * @brief Print basic unitcell information to output stream. diff --git a/source/source_cell/read_atoms.cpp b/source/source_cell/read_atoms.cpp index d3991d662eb..f5223a7a162 100644 --- a/source/source_cell/read_atoms.cpp +++ b/source/source_cell/read_atoms.cpp @@ -70,6 +70,11 @@ bool unitcell::read_atom_positions(UnitCell& ucell, int na = ucell.atoms[it].na; ucell.nat += na; + // Number of atoms of this type whose x/y moment was discarded by + // the collinear projection (nspin=4 with noncolin=0). Reported + // once per type after the atom loop instead of once per atom. + int n_xy_discarded = 0; + if (na > 0) { unitcell::allocate_atom_properties(ucell.atoms[it], na); @@ -103,9 +108,14 @@ bool unitcell::read_atom_positions(UnitCell& ucell, } // Process magnetization - unitcell::process_magnetization(ucell.atoms[it], it, ia, nspin, - input_vec_mag, input_angle_mag, ofs_running, - noncolin); + const bool xy_discarded + = unitcell::process_magnetization(ucell.atoms[it], it, ia, nspin, + input_vec_mag, input_angle_mag, ofs_running, + noncolin); + if (xy_discarded) + { + ++n_xy_discarded; + } // Transform coordinates unitcell::transform_atom_coordinates(ucell.atoms[it], ia, Coordinate, @@ -117,6 +127,9 @@ bool unitcell::read_atom_positions(UnitCell& ucell, ucell.atoms[it].dis[ia].set(0, 0, 0); }//endj } // end na + + unitcell::warn_xy_magnetization_ignored(ucell.atoms[it].label, + n_xy_discarded, na, ofs_running); // reset some useless parameters if (set_element_mag_zero) { @@ -130,15 +143,29 @@ bool unitcell::read_atom_positions(UnitCell& ucell, // so do not override it with an autoset seed. Warn instead. if (symmetry == 1) { - ofs_running << "\n WARNING: initial magmom is all zero and symmetry=1; " - << "autoset magnetism is SKIPPED to preserve the symmetry of the initial (nonmagnetic) structure.\n" - << " If spontaneous magnetism is expected, set magmom explicitly " - << "in STRU, or use symmetry = 0 or -1." << std::endl; + // Only report the skipped autoset when there is in fact + // nothing set: this message used to be printed even for an + // explicitly magnetized STRU. + if (unitcell::is_magnetization_all_zero(ucell)) + { + ofs_running << "\n WARNING: initial magmom is all zero and symmetry=1; " + << "autoset magnetism is SKIPPED to preserve the symmetry of the initial (nonmagnetic) structure.\n" + << " If spontaneous magnetism is expected, set magmom explicitly " + << "in STRU, or use symmetry = 0 or -1." << std::endl; + } } else { unitcell::autoset_magnetization(ucell, nspin, ofs_running); } + + // nspin=4 never autosets a moment, so the zero-moment start has to + // be reported for every symmetry setting, not only on the autoset + // path that symmetry=1 skips. + if (nspin == 4) + { + unitcell::warn_zero_magnetization_nspin4(ucell, ofs_running); + } } // end scan_begin // Final validation and output diff --git a/source/source_cell/read_atoms_helper.cpp b/source/source_cell/read_atoms_helper.cpp index 8e5e75e6c9a..58ce696db68 100644 --- a/source/source_cell/read_atoms_helper.cpp +++ b/source/source_cell/read_atoms_helper.cpp @@ -17,6 +17,9 @@ namespace { constexpr char LOWER_A = 'a'; constexpr char LOWER_Z = 'z'; constexpr char MINUS_SIGN = '-'; + + // Below this value a magnetic moment component is treated as zero. + constexpr double mag_threshold = 1e-5; } namespace unitcell { @@ -79,56 +82,97 @@ void set_atom_movement_flags(Atom& atom, int ia, } } -void autoset_magnetization(UnitCell& ucell, int nspin, - std::ofstream& ofs_running) +bool is_magnetization_all_zero(const UnitCell& ucell) { - const int ntype = ucell.ntype; - - // Check if any atom has non-zero magnetization - int autoset_mag = 1; - for (int it = 0; it < ntype; it++) + for (int it = 0; it < ucell.ntype; it++) { for (int ia = 0; ia < ucell.atoms[it].na; ia++) { - if(std::abs(ucell.atoms[it].mag[ia]) > 1e-5) + if (std::abs(ucell.atoms[it].mag[ia]) > mag_threshold) { - autoset_mag = 0; - break; + return false; } } } + return true; +} - if (autoset_mag) +void autoset_magnetization(UnitCell& ucell, int nspin, + std::ofstream& ofs_running) +{ + // nspin=4 never autosets a moment. An all-zero moment in STRU is a + // legitimate nonmagnetic start; warn_zero_magnetization_nspin4() reports + // it instead, independently of the symmetry switch that gates this call. + if (nspin != 2) + { + return; + } + + if (!is_magnetization_all_zero(ucell)) + { + return; + } + + const int ntype = ucell.ntype; + for (int it = 0; it < ntype; it++) { - if(nspin==4) + for (int ia = 0; ia < ucell.atoms[it].na; ia++) { - for (int it = 0; it < ntype; it++) - { - for (int ia = 0; ia < ucell.atoms[it].na; ia++) - { - ucell.atoms[it].m_loc_[ia].x = 1.0; - ucell.atoms[it].m_loc_[ia].y = 1.0; - ucell.atoms[it].m_loc_[ia].z = 1.0; - ucell.atoms[it].mag[ia] = sqrt(pow(ucell.atoms[it].m_loc_[ia].x,2) - +pow(ucell.atoms[it].m_loc_[ia].y,2) - +pow(ucell.atoms[it].m_loc_[ia].z,2)); - ModuleBase::GlobalFunc::OUT(ofs_running,"Autoset magnetism for this atom", 1.0, 1.0, 1.0); - } - } + ucell.atoms[it].mag[ia] = 1.0; + ucell.atoms[it].m_loc_[ia].z = ucell.atoms[it].mag[ia]; + ModuleBase::GlobalFunc::OUT(ofs_running,"Autoset magnetism for this atom", 1.0); } - else if(nspin==2) + } +} + +void warn_zero_magnetization_nspin4(const UnitCell& ucell, + std::ofstream& ofs_running) +{ + const double threshold_sq = mag_threshold * mag_threshold; + for (int it = 0; it < ucell.ntype; it++) + { + for (int ia = 0; ia < ucell.atoms[it].na; ia++) { - for (int it = 0; it < ntype; it++) + const ModuleBase::Vector3& m = ucell.atoms[it].m_loc_[ia]; + const double norm_sq = m.x * m.x + m.y * m.y + m.z * m.z; + if (norm_sq > threshold_sq) { - for (int ia = 0; ia < ucell.atoms[it].na; ia++) - { - ucell.atoms[it].mag[ia] = 1.0; - ucell.atoms[it].m_loc_[ia].z = ucell.atoms[it].mag[ia]; - ModuleBase::GlobalFunc::OUT(ofs_running,"Autoset magnetism for this atom", 1.0); - } + return; } } } + + const std::string summary = "nspin=4 but no initial magnetization is set in STRU; " + "all atoms start from zero magnetic moment."; + std::cout << " Warning: " << summary << std::endl; + ModuleBase::WARNING("read_atom_positions", summary); + ofs_running << "\n WARNING: " << summary << std::endl; + ofs_running << " If a magnetic ground state is expected, set 'mag' explicitly " + << "in STRU for the magnetic atoms." << std::endl; + ofs_running << " With noncolin=0 the x/y components of 'mag' are discarded, so a " + << "moment given purely along x/y also leaves a zero starting moment." + << std::endl; +} + +void warn_xy_magnetization_ignored(const std::string& label, + int n_discarded, + int na, + std::ofstream& ofs_running) +{ + if (n_discarded <= 0) + { + return; + } + + std::stringstream ss; + ss << "atom type " << label << ": " << n_discarded << " of " << na + << " atoms give non-zero x/y magnetization in STRU, but nspin=4 with noncolin=0 " + << "is a collinear calculation; only the z component is used. " + << "x/y components are IGNORED. Set 'noncolin 1' to use the full vector."; + const std::string summary = ss.str(); + std::cout << " Warning: " << summary << std::endl; + ModuleBase::WARNING("read_atom_positions", summary); + ofs_running << "\n WARNING: " << summary << std::endl; } bool finalize_atom_positions(UnitCell& ucell, @@ -265,12 +309,14 @@ void transform_atom_coordinates(Atom& atom, int ia, } } -void process_magnetization(Atom& atom, int it, int ia, +bool process_magnetization(Atom& atom, int it, int ia, int nspin, bool input_vec_mag, bool input_angle_mag, std::ofstream& ofs_running, const bool noncolin) { + bool xy_discarded = false; + // Recalculate mag and m_loc_ from read in angle1, angle2 and mag or mx, my, mz if(input_angle_mag) { @@ -307,6 +353,23 @@ void process_magnetization(Atom& atom, int it, int ia, if(!noncolin) { // collinear case with nspin = 4, only z component is used + // Note: a scalar "mag " in STRU is stored purely in + // m_loc_.z (x/y stay zero), so it passes the check below without + // flagging a discard and still defines a non-zero z moment. + // cal_ux() then turns that into the global XC quantization axis + // ux_ = (0,0,1) with lsign_ = true. Any positive scalar suffices; + // only the *direction* enters ux_, the magnitude (e.g. 1.7320508 + // vs 1.0) is irrelevant. + // + // The discard is only reported back to the caller, which + // aggregates it per atom type: emitting a message here would + // print one warning per atom for a cell where every atom tilts + // the moment. + if(std::abs(atom.m_loc_[ia].x) > mag_threshold + || std::abs(atom.m_loc_[ia].y) > mag_threshold) + { + xy_discarded = true; + } atom.m_loc_[ia].x = 0; atom.m_loc_[ia].y = 0; } @@ -347,6 +410,8 @@ void process_magnetization(Atom& atom, int it, int ia, ModuleBase::GlobalFunc::OUT(ofs_running, ss.str(),atom.mag[ia]); } } + + return xy_discarded; } bool parse_atom_properties(std::ifstream& ifpos, @@ -457,6 +522,17 @@ bool parse_atom_properties(std::ifstream& ifpos, atom.lambda[ia].y /= ModuleBase::Ry_to_eV; atom.lambda[ia].z /= ModuleBase::Ry_to_eV; } + else if ( tmpid == "f" || tmpid == "force" || tmpid == "forces" ) + { + // Consume the three force components emitted by print_stru_file. + // Forces are intentionally not stored in Atom; reading them keeps + // STRU round-trip parsing intact and prevents the legacy + // movement-flag parser from misinterpreting force digits. + double fx = 0.0; + double fy = 0.0; + double fz = 0.0; + ifpos >> fx >> fy >> fz; + } else if ( tmpid == "sc") { double tmplam=0; diff --git a/source/source_cell/read_atoms_helper.h b/source/source_cell/read_atoms_helper.h index 05d6091cd81..7cb39af3e30 100644 --- a/source/source_cell/read_atoms_helper.h +++ b/source/source_cell/read_atoms_helper.h @@ -45,6 +45,39 @@ void set_atom_movement_flags(Atom& atom, int ia, void autoset_magnetization(UnitCell& ucell, int nspin, std::ofstream& ofs_running); +/** + * @brief Check whether every atom carries a zero collinear magnetic moment + * @param ucell Unit cell object + * @return true if |mag| of every atom is below the magnetization threshold + */ +bool is_magnetization_all_zero(const UnitCell& ucell); + +/** + * @brief Warn when a nspin=4 run starts from an all-zero magnetization vector + * + * The check uses m_loc_, the vector actually consumed by cal_ux(), and not + * mag: the collinear projection applied when noncolin=0 can leave mag + * non-zero while m_loc_ is already zero, so keying the check on mag would + * miss exactly the silent zero-moment start this warning reports. + * + * @param ucell Unit cell object + * @param ofs_running Output stream for running information + */ +void warn_zero_magnetization_nspin4(const UnitCell& ucell, + std::ofstream& ofs_running); + +/** + * @brief Report, once per atom type, that x/y magnetization was discarded + * @param label Element label of the atom type + * @param n_discarded Number of atoms of this type whose x/y moment was dropped + * @param na Number of atoms of this type + * @param ofs_running Output stream for running information + */ +void warn_xy_magnetization_ignored(const std::string& label, + int n_discarded, + int na, + std::ofstream& ofs_running); + /** * @brief Perform final validation and output * @param ucell Unit cell object @@ -94,8 +127,10 @@ void transform_atom_coordinates(Atom& atom, int ia, * @param input_vec_mag Whether vector magnetization was input * @param input_angle_mag Whether angle magnetization was input * @param ofs_running Output stream for running information + * @return true if the x/y components were discarded by the collinear + * projection applied when nspin=4 and noncolin=0 */ -void process_magnetization(Atom& atom, int it, int ia, +bool process_magnetization(Atom& atom, int it, int ia, int nspin, bool input_vec_mag, bool input_angle_mag, std::ofstream& ofs_running, diff --git a/source/source_cell/read_pp_ucell.cpp b/source/source_cell/read_pp_ucell.cpp index b5f2fb61d93..633fc918f50 100644 --- a/source/source_cell/read_pp_ucell.cpp +++ b/source/source_cell/read_pp_ucell.cpp @@ -55,26 +55,26 @@ AtomsInfoResult read_pseudo(std::ofstream& ofs, UnitCell& ucell, const std::string dft_functional_ = dft_functional; read_cell_pseudopots(pseudo_dir_, ofs, ucell, global_out_dir_, dft_functional_, lspinorb, pseudo_rcut, soc_lambda); - if (GlobalV::MY_RANK == 0) - { - for (int it = 0; it < ucell.ntype; it++) - { - Atom* atom = &ucell.atoms[it]; - if (!(atom->label_orb.empty())) - { + if (GlobalV::MY_RANK == 0) + { + for (int it = 0; it < ucell.ntype; it++) + { + Atom* atom = &ucell.atoms[it]; + if (!(atom->label_orb.empty())) + { unitcell::compare_atom_labels(atom->label_orb, atom->ncpp.psd); } } - if (out_element_info_) - { - for (int i = 0; i < ucell.ntype; i++) - { - ModuleBase::Global_File::make_dir_atom(ucell.atoms[i].label, global_out_dir_); + if (out_element_info_) + { + for (int i = 0; i < ucell.ntype; i++) + { + ModuleBase::Global_File::make_dir_atom(ucell.atoms[i].label, global_out_dir_); } - for (int it = 0; it < ucell.ntype; it++) - { - Atom* atom = &ucell.atoms[it]; + for (int it = 0; it < ucell.ntype; it++) + { + Atom* atom = &ucell.atoms[it]; std::stringstream ss; ss << global_out_dir_ << atom->label << "/" << atom->label << ".NONLOCAL"; @@ -181,20 +181,20 @@ AtomsInfoResult read_pseudo(std::ofstream& ofs, UnitCell& ucell, basis_type, esolver_type, init_wfc, nbands); // Check whether the number of valence is minimum - if (GlobalV::MY_RANK == 0) - { - int abtype = 0; - for (int it = 0; it < ucell.ntype; it++) - { - if (ModuleBase::MinZval.find(ucell.atoms[it].ncpp.psd) - != ModuleBase::MinZval.end()) - { - if (ucell.atoms[it].ncpp.zv - > ModuleBase::MinZval.at(ucell.atoms[it].ncpp.psd)) - { + if (GlobalV::MY_RANK == 0) + { + int abtype = 0; + for (int it = 0; it < ucell.ntype; it++) + { + if (ModuleBase::MinZval.find(ucell.atoms[it].ncpp.psd) + != ModuleBase::MinZval.end()) + { + if (ucell.atoms[it].ncpp.zv + > ModuleBase::MinZval.at(ucell.atoms[it].ncpp.psd)) + { abtype += 1; - if (abtype == 1) - { + if (abtype == 1) + { std::cout << "\n%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%" "%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%" "%%%%%%%%%%%%%%%%%%%%%%%%%%" @@ -219,9 +219,9 @@ AtomsInfoResult read_pseudo(std::ofstream& ofs, UnitCell& ucell, } } } - if (abtype > 0) - { - std::cout << " Pseudopotentials with additional electrons can " + if (abtype > 0) + { + std::cout << " Pseudopotentials with additional electrons can " "yield (more) accurate outcomes, but may be " "less efficient." << std::endl; @@ -347,7 +347,7 @@ void read_cell_pseudopots(const std::string& pp_dir, std::ofstream& log, UnitCel if (GlobalV::MY_RANK == 0) { - upf.complete_default(ucell.atoms[i].ncpp, pseudo_rcut_); + upf.complete_default(ucell.atoms[i].ncpp, pseudo_rcut_); log << std::endl; ModuleBase::GlobalFunc::OUT(log, "Pseudopotential file", ucell.pseudo_fn[i]); diff --git a/source/source_cell/reciprocal_grid.cpp b/source/source_cell/reciprocal_grid.cpp index 43015c4f265..10f5e1a32cc 100644 --- a/source/source_cell/reciprocal_grid.cpp +++ b/source/source_cell/reciprocal_grid.cpp @@ -71,8 +71,10 @@ void ReciprocalGrid::Monkhorst_Pack(const int* nmp_in, const double* koffset_in, const int mpnz = nmp_in[2]; this->nkstot = mpnx * mpny * mpnz; - // only can renew after nkstot is estimated. - this->renew(nkstot * spin_factor()); + // only can renew after nkstot is estimated. The spin_factor() doubling (for + // K_Vectors) is applied later, on demand, by set_kup_and_kdw() itself, so it + // covers the post-symmetry-reduction k-count rather than this pre-reduction one. + this->renew(nkstot); for (int x = 1; x <= mpnx; x++) { diff --git a/source/source_cell/record_adj.cpp b/source/source_cell/record_adj.cpp new file mode 100644 index 00000000000..b20708975c4 --- /dev/null +++ b/source/source_cell/record_adj.cpp @@ -0,0 +1,243 @@ +#include "record_adj.h" +#include "source_base/timer.h" +#include "source_cell/module_neighbor/sltk_grid_driver.h" + +Record_adj::Record_adj() +{ +} +Record_adj::~Record_adj() +{ +} + +void Record_adj::delete_grid() +{ + info.clear(); + info_offset.clear(); + na_each.clear(); + iat2ca.clear(); + na_proc = 0; +} + +//-------------------------------------------- +// Check whether atom (T2, tau2) is adjacent to +// atom (T1, tau1). Two atoms are adjacent if +// their orbital cutoff spheres overlap, or if +// both overlap with the nonlocal-beta cutoff +// sphere of a common third atom (k-point case). +//-------------------------------------------- +static bool is_adjacent(const UnitCell& ucell, + const int T1, + const int T2, + const ModuleBase::Vector3& tau1, + const ModuleBase::Vector3& tau2, + const AdjacentAtomInfo& adjs, + const std::vector& orb_cutoff) +{ + const ModuleBase::Vector3 dtau = tau2 - tau1; + const double distance = dtau.norm() * ucell.lat0; + const double rcut = orb_cutoff[T1] + orb_cutoff[T2]; + + if (distance < rcut) + { + return true; + } + + // there is another possibility that i and j are adjacent atoms. + // which is that are adjacents while are also + // adjacents, these considerations are only considered in k-point + // algorithm, + for (int ad0 = 0; ad0 < adjs.adj_num + 1; ++ad0) + { + const int T0 = adjs.ntype[ad0]; + const ModuleBase::Vector3 tau0 = adjs.adjacent_tau[ad0]; + + const ModuleBase::Vector3 dtau1 = tau0 - tau1; + const double distance1 = dtau1.norm() * ucell.lat0; + const double rcut1 = orb_cutoff[T1] + ucell.infoNL->get_rcut_max(T0); + + const ModuleBase::Vector3 dtau2 = tau0 - tau2; + const double distance2 = dtau2.norm() * ucell.lat0; + const double rcut2 = orb_cutoff[T2] + ucell.infoNL->get_rcut_max(T0); + + if (distance1 < rcut1 && distance2 < rcut2) + { + return true; + } // dis1, dis2 + } + + return false; +} + +//-------------------------------------------- +// This will record the orbitals according to +// HPSEPS's 2D block division. +// If multi-k, calculate nnr at the same time. +// be called only once in an ion-step. +//-------------------------------------------- +void Record_adj::for_2d(const UnitCell& ucell, + const Grid_Driver& grid_d, + Parallel_Orbitals& pv, + bool gamma_only, + const int npol, + const std::vector& orb_cutoff) +{ + ModuleBase::TITLE("Record_adj", "for_2d"); + ModuleBase::timer::start("Record_adj", "for_2d"); + + assert(ucell.nat > 0); + if (!gamma_only) + { + // Record_adj should not modify members of pv, need refactor! mohan add 2025-03-10 + pv.nlocdim.assign(ucell.nat, 0); + pv.nlocstart.assign(ucell.nat, 0); + pv.nnr = 0; + } + + this->count_adjacent(ucell, grid_d, pv, gamma_only, npol, orb_cutoff); + + this->allocate_info(); + + this->fill_info(ucell, grid_d, orb_cutoff); + + ModuleBase::timer::end("Record_adj", "for_2d"); +} + +//-------------------------------------------- +// (1) find the adjacent atoms of each atom and +// count na_each; for multi-k, accumulate +// nlocdim / nlocstart / nnr of pv. +//-------------------------------------------- +void Record_adj::count_adjacent(const UnitCell& ucell, + const Grid_Driver& grid_d, + Parallel_Orbitals& pv, + bool gamma_only, + const int npol, + const std::vector& orb_cutoff) +{ + this->na_proc = ucell.nat; + + // number of adjacents for each atom. + this->na_each.assign(na_proc, 0); + int iat = 0; + + for (int T1 = 0; T1 < ucell.ntype; ++T1) + { + const Atom* atom1 = &ucell.atoms[T1]; + for (int I1 = 0; I1 < atom1->na; ++I1) + { + const ModuleBase::Vector3 tau1 = atom1->tau[I1]; + grid_d.Find_atom(ucell, T1, I1); + const int start1 = ucell.itiaiw2iwt(T1, I1, 0); + if (!gamma_only) + { + pv.nlocstart[iat] = pv.nnr; + } + + // (2) search among all adjacent atoms. + for (int ad = 0; ad < grid_d.getAdjacentNum() + 1; ++ad) + { + const int T2 = grid_d.getType(ad); + const int I2 = grid_d.getNatom(ad); + const int start2 = ucell.itiaiw2iwt(T2, I2, 0); + const ModuleBase::Vector3 tau2 = grid_d.getAdjacentTau(ad); + + if (!is_adjacent(ucell, T1, T2, tau1, tau2, grid_d.getAdjacentInfo(), orb_cutoff)) + { + continue; + } + + ++na_each[iat]; + if (!gamma_only) + { + for (int ii = 0; ii < atom1->nw * npol; ++ii) + { + // the index of orbitals in this processor + const int iw1_all = start1 + ii; + const int mu = pv.global2local_row(iw1_all); + if (mu < 0) + { + continue; + } + + for (int jj = 0; jj < ucell.atoms[T2].nw * npol; ++jj) + { + const int iw2_all = start2 + jj; + const int nu = pv.global2local_col(iw2_all); + if (nu < 0) + { + continue; + } + + pv.nlocdim[iat]++; + ++(pv.nnr); + } + } + } + } // end ad + ++iat; + } // end I1 + } // end T1 +} + +//-------------------------------------------- +// allocate info[na_proc][na_each[i]][5] +//-------------------------------------------- +void Record_adj::allocate_info() +{ + // lay out all adjacent records flat: the records of + // atom iat start at info_offset[iat]. + info_offset.resize(na_proc); + int total = 0; + for (int i = 0; i < na_proc; i++) + { + info_offset[i] = total; + total += na_each[i]; + } + // each record holds (Rx, Ry, Rz, T, I), zero-initialized + info.resize(total); +} + +//-------------------------------------------- +// fill info with (Rx, Ry, Rz, T, I) of each +// adjacent atom. +//-------------------------------------------- +void Record_adj::fill_info(const UnitCell& ucell, + const Grid_Driver& grid_d, + const std::vector& orb_cutoff) +{ +#ifdef _OPENMP +#pragma omp parallel for schedule(dynamic) +#endif + for (int iat = 0; iat < ucell.nat; ++iat) + { + const int T1 = ucell.iat2it[iat]; + const Atom* atom1 = &ucell.atoms[T1]; + const int I1 = ucell.iat2ia[iat]; + const ModuleBase::Vector3 tau1 = atom1->tau[I1]; + + AdjacentAtomInfo adjs; + grid_d.Find_atom(ucell, T1, I1, &adjs); + + // (2) search among all adjacent atoms. + int cb = 0; + for (int ad = 0; ad < adjs.adj_num + 1; ++ad) + { + const int T2 = adjs.ntype[ad]; + const int I2 = adjs.natom[ad]; + const ModuleBase::Vector3 tau2 = adjs.adjacent_tau[ad]; + + if (!is_adjacent(ucell, T1, T2, tau1, tau2, adjs, orb_cutoff)) + { + continue; + } + + std::array& rec = info[info_offset[iat] + cb]; + rec[0] = adjs.box[ad].x; + rec[1] = adjs.box[ad].y; + rec[2] = adjs.box[ad].z; + rec[3] = T2; + rec[4] = I2; + ++cb; + } // end ad + } // end iat +} diff --git a/source/source_cell/record_adj.h b/source/source_cell/record_adj.h new file mode 100644 index 00000000000..aa2e63158f4 --- /dev/null +++ b/source/source_cell/record_adj.h @@ -0,0 +1,89 @@ +#ifndef RECORD_ADJ_H +#define RECORD_ADJ_H + +#include "source_basis/module_ao/parallel_orbitals.h" +#include "source_cell/unitcell.h" +#include "source_cell/module_neighbor/sltk_grid_driver.h" + +#include +#include + +//--------------------------------------------------- +// FUNCTION: record the adjacent atoms for each atom +//--------------------------------------------------- +class Record_adj +{ + public: + Record_adj(); + ~Record_adj(); + + //-------------------------------------------- + // This will record the orbitals according to + // HPSEPS's 2D block division. + //-------------------------------------------- + void for_2d(const UnitCell& ucell, + const Grid_Driver& grid_d, + Parallel_Orbitals& pv, + bool gamma_only, + const int npol, + const std::vector& orb_cutoff); + + + void delete_grid(); + + private: + // (1) count adjacent atoms for each atom and, + // for multi-k, accumulate nlocdim / nlocstart / nnr of pv. + void count_adjacent(const UnitCell& ucell, + const Grid_Driver& grid_d, + Parallel_Orbitals& pv, + bool gamma_only, + const int npol, + const std::vector& orb_cutoff); + + // allocate info[na_proc][na_each[i]][5] + void allocate_info(); + + // fill info with (Rx, Ry, Rz, T, I) of each adjacent atom. + void fill_info(const UnitCell& ucell, + const Grid_Driver& grid_d, + const std::vector& orb_cutoff); + + public: + int na_proc=0; + std::vector na_each; + + //-------------------------------------------- + // record sparse atom index in for_grid(); + // Map iat(dense atom index) to sparse atom index + // Mainly removing the index dependency for OpenMP parallel loop + // + // Meaning: + // 1. if iat2ca[iat] > 0, it contains the sparse atom index + // 2. if iat2ca[iat] < 0, the sparse atom index of iat does not exist + // + // Usage: + // 1. iat2ca[iat] > 0 ? na_each[iat2ca[iat]] : 0 + // 2. iat2ca[iat] > 0 ? info[iat2ca[iat]] : nullptr + //-------------------------------------------- + std::vector iat2ca; + + //------------------------------------------------ + // info identifies each adjacent atom in each + // unitcell. All adjacent records are stored flat: + // the records of atom iat occupy + // info[info_offset[iat], info_offset[iat]+na_each[iat]). + // Each record holds (Rx, Ry, Rz, T, I). + //------------------------------------------------ + std::vector> info; + std::vector info_offset; + + // Access the (Rx, Ry, Rz, T, I) record of the cb-th + // adjacent atom of atom iat. + const std::array& get_info(const int iat, const int cb) const + { + return info[info_offset[iat] + cb]; + } +}; + +#endif diff --git a/source/source_cell/test/CMakeLists.txt b/source/source_cell/test/CMakeLists.txt index 653eaa6d3f1..d681868cc7b 100644 --- a/source/source_cell/test/CMakeLists.txt +++ b/source/source_cell/test/CMakeLists.txt @@ -173,8 +173,49 @@ add_test(NAME MODULE_CELL_parallel_kpoints_test AddTest( TARGET MODULE_CELL_unitcell_test LIBS base device cell_info symmetry - SOURCES unitcell_test.cpp ../cal_ux.cpp + SOURCES unitcell_test.cpp +) + +AddTest( + TARGET MODULE_CELL_cal_ux_test + LIBS base device cell_info + SOURCES test_cal_ux.cpp ../cal_ux.cpp +) + +AddTest( + TARGET MODULE_CELL_read_orb_test + LIBS base device cell_info + SOURCES test_read_orb.cpp +) +AddTest( + TARGET MODULE_CELL_update_cell_test + LIBS base device cell_info + SOURCES test_update_cell.cpp +) + +AddTest( + TARGET MODULE_CELL_cell_tools_test + LIBS base device cell_info + SOURCES test_cell_tools.cpp +) + +AddTest( + TARGET MODULE_CELL_read_atom_species_test + LIBS base device cell_info + SOURCES test_read_atom_species.cpp +) + +AddTest( + TARGET MODULE_CELL_read_atoms_test + LIBS base device cell_info + SOURCES test_read_atoms.cpp +) + +AddTest( + TARGET MODULE_CELL_print_cell_test + LIBS base device cell_info symmetry + SOURCES test_print_cell.cpp ) AddTest( diff --git a/source/source_cell/test/klist_test.cpp b/source/source_cell/test/klist_test.cpp index 4f25675cb4d..a9a546b2196 100644 --- a/source/source_cell/test/klist_test.cpp +++ b/source/source_cell/test/klist_test.cpp @@ -2,7 +2,6 @@ #include "gtest/gtest.h" #include #include -#define private public #include "source_cell/atom_pseudo.h" #include "source_cell/atom_spec.h" #include "source_cell/klist.h" @@ -11,7 +10,6 @@ #include "source_cell/unitcell.h" #include "source_cell/magnetism.h" -#undef private #include "source_base/mathzone.h" #include "source_base/parallel_global.h" #include "source_base/global_variable.h" @@ -216,7 +214,7 @@ TEST_F(KlistTest, Construct) { EXPECT_EQ(kv->get_nks(), 0); EXPECT_EQ(kv->get_nkstot(), 0); - EXPECT_EQ(kv->spin_mult, 0); + EXPECT_EQ(kv->get_spin_mult(), 0); EXPECT_EQ(kv->k_nkstot, 0); EXPECT_FALSE(kv->kc_done); EXPECT_FALSE(kv->kd_done); @@ -228,12 +226,11 @@ TEST_F(KlistTest, MP) kv->nmp[0] = 2; kv->nmp[1] = 2; kv->nmp[2] = 2; - kv->koffset[0] = 0; - kv->koffset[1] = 0; - kv->koffset[2] = 0; - kv->spin_mult = 1; + // Monkhorst_Pack() takes the offset as an argument, so the test owns it + const double koffset_zero[3] = {0.0, 0.0, 0.0}; + kv->set_spin_mult(1); int k_type = 0; - kv->Monkhorst_Pack(kv->nmp, kv->koffset, k_type); + kv->Monkhorst_Pack(kv->nmp, koffset_zero, k_type); /* std::cout << " " <get_nkstot();ik++) @@ -245,12 +242,10 @@ TEST_F(KlistTest, MP) kv1->nmp[0] = 2; kv1->nmp[1] = 2; kv1->nmp[2] = 2; - kv1->koffset[0] = 1; - kv1->koffset[1] = 1; - kv1->koffset[2] = 1; - kv1->spin_mult = 1; + const double koffset_one[3] = {1.0, 1.0, 1.0}; + kv1->set_spin_mult(1); k_type = 1; - kv1->Monkhorst_Pack(kv1->nmp, kv1->koffset, k_type); + kv1->Monkhorst_Pack(kv1->nmp, koffset_one, k_type); // std::cout << " " <nkstot; ik++) { @@ -268,8 +263,8 @@ TEST_F(KlistTest, ReadKpointsGammaOnlyLocal) const std::string kmesh_type = "gamma"; const double koffset[3] = {0.0, 0.0, 0.0}; std::string k_file = "KPT_GO"; - kv->spin_mult = 1; - kv->read_kpoints(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank); + kv->set_spin_mult(1); + kv->read_kpoints_for_testing(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank); ifs.open("KPT_GO"); std::string str((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); EXPECT_THAT(str, testing::HasSubstr("Gamma")); @@ -280,46 +275,46 @@ TEST_F(KlistTest, ReadKpointsGammaOnlyLocal) TEST_F(KlistTest, ReadKpointsKspacing) { - kv->spin_mult = 1; + kv->set_spin_mult(1); const bool gamma_only_local = false; const double kspacing[3] = {0.052918, 0.052918, 0.052918}; // 0.52918/Bohr = 1/A const std::string kmesh_type = "gamma"; const double koffset[3] = {0.0, 0.0, 0.0}; setucell(); std::string k_file = "./support/KPT3"; - kv->read_kpoints(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank); + kv->read_kpoints_for_testing(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank); EXPECT_EQ(kv->get_nkstot(), 343); } TEST_F(KlistTest, ReadKpointsKspacing3values) { - kv->spin_mult = 1; + kv->set_spin_mult(1); const bool gamma_only_local = false; const double kspacing[3] = {0.052918, 0.06, 0.07}; // 0.52918/Bohr = 1/A const std::string kmesh_type = "gamma"; const double koffset[3] = {0.0, 0.0, 0.0}; setucell(); std::string k_file = "./support/KPT3"; - kv->read_kpoints(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank); + kv->read_kpoints_for_testing(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank); EXPECT_EQ(kv->get_nkstot(), 210); } TEST_F(KlistTest, ReadKpointsInvalidKspacing3values) { - kv->spin_mult = 1; + kv->set_spin_mult(1); const bool gamma_only_local = false; const double kspacing[3] = {0.052918, 0.0, 0.07}; // 0.52918/Bohr = 1/A const std::string kmesh_type = "gamma"; const double koffset[3] = {0.0, 0.0, 0.0}; std::string k_file = "./support/KPT3"; testing::internal::CaptureStdout(); - EXPECT_EXIT(kv->read_kpoints(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank), ::testing::ExitedWithCode(1), ""); + EXPECT_EXIT(kv->read_kpoints_for_testing(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); } TEST_F(KlistTest, ReadKpointsKspacingShiftedGamma) { - kv->spin_mult = 1; + kv->set_spin_mult(1); const bool gamma_only_local = false; const double kspacing[3] = {0.052918, 0.052918, 0.052918}; // 0.52918/Bohr = 1/A const std::string kmesh_type = "gamma"; @@ -327,7 +322,7 @@ TEST_F(KlistTest, ReadKpointsKspacingShiftedGamma) setucell(); std::string k_file = "./support/KPT3"; - kv->read_kpoints(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank); + kv->read_kpoints_for_testing(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank); EXPECT_EQ(kv->get_nkstot(), 343); EXPECT_EQ(kv->get_k_kword(), "Gamma"); @@ -341,7 +336,7 @@ TEST_F(KlistTest, ReadKpointsKspacingShiftedGamma) TEST_F(KlistTest, ReadKpointsKspacingShiftedMP) { - kv->spin_mult = 1; + kv->set_spin_mult(1); const bool gamma_only_local = false; const double kspacing[3] = {0.052918, 0.052918, 0.052918}; // 0.52918/Bohr = 1/A const std::string kmesh_type = "mp"; @@ -349,7 +344,7 @@ TEST_F(KlistTest, ReadKpointsKspacingShiftedMP) setucell(); std::string k_file = "./support/KPT3"; - kv->read_kpoints(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank); + kv->read_kpoints_for_testing(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank); EXPECT_EQ(kv->get_nkstot(), 343); EXPECT_EQ(kv->get_k_kword(), "Monkhorst-Pack"); @@ -368,8 +363,8 @@ TEST_F(KlistTest, ReadKpointsGamma) const std::string kmesh_type = "gamma"; const double koffset[3] = {0.0, 0.0, 0.0}; std::string k_file = "./support/KPT"; - kv->spin_mult = 1; - kv->read_kpoints(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank); + kv->set_spin_mult(1); + kv->read_kpoints_for_testing(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank); EXPECT_EQ(kv->get_nkstot(), 512); } @@ -380,8 +375,8 @@ TEST_F(KlistTest, ReadKpointsMP) const std::string kmesh_type = "gamma"; const double koffset[3] = {0.0, 0.0, 0.0}; std::string k_file = "./support/KPT1"; - kv->spin_mult = 1; - kv->read_kpoints(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank); + kv->set_spin_mult(1); + kv->read_kpoints_for_testing(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank); EXPECT_EQ(kv->get_nkstot(), 512); } @@ -393,8 +388,8 @@ TEST_F(KlistTest, ReadKpointsLine) const std::string kmesh_type = "gamma"; const double koffset[3] = {0.0, 0.0, 0.0}; std::string k_file = "./support/KPT2"; - kv->spin_mult = 1; - kv->read_kpoints(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank); + kv->set_spin_mult(1); + kv->read_kpoints_for_testing(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank); EXPECT_EQ(kv->get_nkstot(), 122); } @@ -406,9 +401,9 @@ TEST_F(KlistTest, ReadKpointsLineRejectsZeroInterpolationCount) const std::string kmesh_type = "gamma"; const double koffset[3] = {0.0, 0.0, 0.0}; const std::string k_file = "./support/KPT_ZERO_LINE_COUNT"; - kv->spin_mult = 1; + kv->set_spin_mult(1); - EXPECT_EXIT(kv->read_kpoints(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank), + EXPECT_EXIT(kv->read_kpoints_for_testing(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank), ::testing::ExitedWithCode(1), ""); } @@ -421,12 +416,12 @@ TEST_F(KlistTest, ReadKpointsCartesian) const double koffset[3] = {0.0, 0.0, 0.0}; std::string k_file = "./support/KPT4"; // Cartesian: non-spin case nspin=1 - kv->spin_mult = 1; - kv->read_kpoints(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank); + kv->set_spin_mult(1); + kv->read_kpoints_for_testing(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank); EXPECT_EQ(kv->kvec_c.size(), 5); // spin case nspin=2 - kv->spin_mult = 2; - kv->read_kpoints(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank); + kv->set_spin_mult(2); + kv->read_kpoints_for_testing(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank); EXPECT_EQ(kv->kvec_c.size(), 10); } @@ -438,14 +433,14 @@ TEST_F(KlistTest, ReadKpointsLineCartesian) const double koffset[3] = {0.0, 0.0, 0.0}; std::string k_file = "./support/KPT5"; // Line Cartesian: non-spin case nspin=1 - kv->spin_mult = 1; - kv->set_kup_and_kdw(ofs_running); - kv->read_kpoints(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank); + kv->set_spin_mult(1); + kv->set_kup_and_kdw_for_testing(ofs_running); + kv->read_kpoints_for_testing(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank); EXPECT_EQ(kv->get_nkstot(), 51); EXPECT_EQ(kv->kvec_c.size(), 51); // Line Cartesian: spin case nspin=2 - kv->spin_mult = 2; - kv->read_kpoints(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank); + kv->set_spin_mult(2); + kv->read_kpoints_for_testing(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank); EXPECT_EQ(kv->get_nkstot(), 51); EXPECT_EQ(kv->kvec_c.size(), 102); } @@ -457,9 +452,9 @@ TEST_F(KlistTest, ReadKpointsDirect) const std::string kmesh_type = "gamma"; const double koffset[3] = {0.0, 0.0, 0.0}; std::string k_file = "./support/KPT6"; - kv->spin_mult = 1; - kv->set_kup_and_kdw(ofs_running); - kv->read_kpoints(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank); + kv->set_spin_mult(1); + kv->set_kup_and_kdw_for_testing(ofs_running); + kv->read_kpoints_for_testing(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank); EXPECT_EQ(kv->get_nkstot(), 6); EXPECT_TRUE(kv->kd_done); } @@ -471,9 +466,9 @@ TEST_F(KlistTest, ReadKpointsWarning1) const std::string kmesh_type = "gamma"; const double koffset[3] = {0.0, 0.0, 0.0}; std::string k_file = "arbitrary_1"; - kv->spin_mult = 1; + kv->set_spin_mult(1); ofs_warning.open("klist_tmp_warning_1"); - EXPECT_NO_THROW(kv->read_kpoints(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank)); + EXPECT_NO_THROW(kv->read_kpoints_for_testing(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank)); ofs_warning.close(); ifs.open("klist_tmp_warning_1"); std::string str((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); @@ -492,9 +487,9 @@ TEST_F(KlistTest, ReadKpointsWarning2) ofs.open(k_file.c_str()); ofs << "ARBITRARY"; ofs.close(); - kv->spin_mult = 1; + kv->set_spin_mult(1); ofs_warning.open("klist_tmp_warning_2"); - EXPECT_NO_THROW(kv->read_kpoints(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank)); + EXPECT_NO_THROW(kv->read_kpoints_for_testing(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank)); ofs_warning.close(); ifs.open("klist_tmp_warning_2"); std::string str((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); @@ -515,9 +510,9 @@ TEST_F(KlistTest, ReadKpointsWarning3) ofs << "KPOINTS" << std::endl; ofs << "100001" << std::endl; ofs.close(); - kv->spin_mult = 1; + kv->set_spin_mult(1); ofs_warning.open("klist_tmp_warning_3"); - EXPECT_NO_THROW(kv->read_kpoints(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank)); + EXPECT_NO_THROW(kv->read_kpoints_for_testing(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank)); ofs_warning.close(); ifs.open("klist_tmp_warning_3"); std::string str((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); @@ -539,9 +534,9 @@ TEST_F(KlistTest, ReadKpointsWarning4) ofs << "0" << std::endl; ofs << "arbitrary" << std::endl; ofs.close(); - kv->spin_mult = 1; + kv->set_spin_mult(1); ofs_warning.open("klist_tmp_warning_4"); - EXPECT_NO_THROW(kv->read_kpoints(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank)); + EXPECT_NO_THROW(kv->read_kpoints_for_testing(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank)); ofs_warning.close(); ifs.open("klist_tmp_warning_4"); std::string str((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); @@ -563,9 +558,9 @@ TEST_F(KlistTest, ReadKpointsWarning5) ofs << "100000" << std::endl; ofs << "arbitrary" << std::endl; ofs.close(); - kv->spin_mult = 1; + kv->set_spin_mult(1); ofs_warning.open("klist_tmp_warning_5"); - EXPECT_NO_THROW(kv->read_kpoints(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank)); + EXPECT_NO_THROW(kv->read_kpoints_for_testing(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank)); ofs_warning.close(); ifs.open("klist_tmp_warning_5"); std::string str((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); @@ -587,10 +582,10 @@ TEST_F(KlistTest, ReadKpointsWarning6) ofs << "100000" << std::endl; ofs << "Line_Cartesian" << std::endl; ofs.close(); - kv->spin_mult = 1; + kv->set_spin_mult(1); ModuleSymmetry::Symmetry::symm_flag = 1; ofs_warning.open("klist_tmp_warning_6"); - EXPECT_NO_THROW(kv->read_kpoints(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank)); + EXPECT_NO_THROW(kv->read_kpoints_for_testing(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank)); ofs_warning.close(); ifs.open("klist_tmp_warning_6"); std::string str((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); @@ -613,10 +608,10 @@ TEST_F(KlistTest, ReadKpointsWarning7) ofs << "100000" << std::endl; ofs << "Line_Direct" << std::endl; ofs.close(); - kv->spin_mult = 1; + kv->set_spin_mult(1); ModuleSymmetry::Symmetry::symm_flag = 1; ofs_warning.open("klist_tmp_warning_7"); - EXPECT_NO_THROW(kv->read_kpoints(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank)); + EXPECT_NO_THROW(kv->read_kpoints_for_testing(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank)); ofs_warning.close(); ifs.open("klist_tmp_warning_7"); std::string str((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); @@ -636,9 +631,9 @@ TEST_F(KlistTest, SetKupKdown) std::string k_file = "./support/KPT4"; // case A: physical nspin=1 -> spin_mult=1 (no doubling). - kv->spin_mult = 1; - kv->read_kpoints(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank); - kv->set_kup_and_kdw(ofs_running); + kv->set_spin_mult(1); + kv->read_kpoints_for_testing(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank); + kv->set_kup_and_kdw_for_testing(ofs_running); for (int ik = 0; ik < 5; ik++) { EXPECT_EQ(kv->isk[ik], 0); @@ -648,16 +643,16 @@ TEST_F(KlistTest, SetKupKdown) // K_Vectors::set() time; non-collinear does not double the k-point list, // so the correct spin_mult is still 1. We bypass set() here, so set the // mapped value directly. - kv->spin_mult = 1; - kv->read_kpoints(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank); - kv->set_kup_and_kdw(ofs_running); + kv->set_spin_mult(1); + kv->read_kpoints_for_testing(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank); + kv->set_kup_and_kdw_for_testing(ofs_running); for (int ik = 0; ik < 5; ik++) { EXPECT_EQ(kv->isk[ik], 0); } - kv->spin_mult = 2; - kv->read_kpoints(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank); - kv->set_kup_and_kdw(ofs_running); + kv->set_spin_mult(2); + kv->read_kpoints_for_testing(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank); + kv->set_kup_and_kdw_for_testing(ofs_running); for (int ik = 0; ik < 5; ik++) { EXPECT_EQ(kv->isk[ik], 0); @@ -667,10 +662,10 @@ TEST_F(KlistTest, SetKupKdown) TEST_F(KlistTest, SetAfterVC) { - kv->spin_mult = 1; + kv->set_spin_mult(1); kv->set_nkstot(1); ofs_running.open("tmp_klist_1"); - kv->renew(kv->get_nkstot()); + kv->renew_for_testing(kv->get_nkstot()); kv->kvec_c[0].x = 0; kv->kvec_c[0].y = 0; kv->kvec_c[0].z = 0; @@ -687,11 +682,11 @@ TEST_F(KlistTest, SetAfterVC) TEST_F(KlistTest, PrintKlists) { - kv->spin_mult = 1; + kv->set_spin_mult(1); kv->set_nkstot(1); kv->set_nks(1); ofs_running.open("tmp_klist_2"); - kv->renew(kv->get_nkstot()); + kv->renew_for_testing(kv->get_nkstot()); kv->kvec_c[0].x = 0; kv->kvec_c[0].y = 0; kv->kvec_c[0].z = 0; @@ -704,10 +699,10 @@ TEST_F(KlistTest, PrintKlists) TEST_F(KlistTest, PrintKlistsWarnigQuit) { - kv->spin_mult = 1; + kv->set_spin_mult(1); kv->set_nkstot(1); kv->set_nks(2); - kv->renew(kv->get_nkstot()); + kv->renew_for_testing(kv->get_nkstot()); kv->kvec_c[0].x = 0; kv->kvec_c[0].y = 0; kv->kvec_c[0].z = 0; @@ -719,10 +714,10 @@ TEST_F(KlistTest, PrintKlistsWarnigQuit) TEST_F(KlistTest, SetBothKvecFlagsFromFile) { - kv->spin_mult = 1; + kv->set_spin_mult(1); kv->set_nkstot(1); kv->set_nks(1); - kv->renew(kv->get_nkstot()); + kv->renew_for_testing(kv->get_nkstot()); kv->kvec_d[0].x = 0.0; kv->kvec_d[0].y = 0.0; kv->kvec_d[0].z = 0.0; @@ -766,10 +761,10 @@ TEST_F(KlistTest, SetBothKvecFlagsFromFile) TEST_F(KlistTest, SetBothKvec) { - kv->spin_mult = 1; + kv->set_spin_mult(1); kv->set_nkstot(1); kv->set_nks(1); - kv->renew(kv->get_nkstot()); + kv->renew_for_testing(kv->get_nkstot()); kv->kvec_d[0].x = 0.0; kv->kvec_d[0].y = 0.0; kv->kvec_d[0].z = 0.0; @@ -788,10 +783,10 @@ TEST_F(KlistTest, SetBothKvec) TEST_F(KlistTest, NormalizeWk) { - kv->spin_mult = 1; + kv->set_spin_mult(1); kv->set_nkstot(2); kv->set_nks(2); - kv->renew(kv->get_nkstot()); + kv->renew_for_testing(kv->get_nkstot()); kv->wk[0] = 1.0; kv->wk[1] = 1.0; int deg = 2; @@ -803,10 +798,10 @@ TEST_F(KlistTest, NormalizeWk) TEST_F(KlistTest, NormalizeWkZeroWeights) { // Test that zero weights are handled correctly - kv->spin_mult = 1; + kv->set_spin_mult(1); kv->set_nkstot(3); kv->set_nks(3); - kv->renew(kv->get_nkstot()); + kv->renew_for_testing(kv->get_nkstot()); kv->wk[0] = 0.0; kv->wk[1] = 0.0; kv->wk[2] = 0.0; @@ -828,10 +823,10 @@ TEST_F(KlistTest, NormalizeWkZeroWeights) TEST_F(KlistTest, UpdateUseIBZ) { - kv->spin_mult = 1; + kv->set_spin_mult(1); kv->set_nkstot(3); kv->set_nks(3); - kv->renew(kv->get_nkstot()); + kv->renew_for_testing(kv->get_nkstot()); kv->update_use_ibz(2, std::vector>(2, {0, 0, 0}), std::vector(2, 0.0), ofs_running, my_rank); EXPECT_EQ(kv->get_nkstot(), 2); EXPECT_EQ(kv->kvec_d.size(), 2); @@ -851,14 +846,14 @@ TEST_F(KlistTest, IbzKpoint) const int cal_symm_repr[2] = {0, 6}; symm.analy_sys(ucell.lat, ucell.st, ucell.atoms, ofs_running, 1e-6, 1, "scf", cal_symm_repr); std::string k_file = "./support/KPT1"; - kv->spin_mult = 1; - kv->read_kpoints(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank); + kv->set_spin_mult(1); + kv->read_kpoints_for_testing(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank); EXPECT_EQ(kv->get_nkstot(), 512); // calculate ibz_kpoint std::string skpt; ModuleSymmetry::Symmetry::symm_flag = 1; bool match = true; - kv->reduce_by_symmetry(ucell, symm, ModuleSymmetry::Symmetry::symm_flag, skpt, match, my_rank, ofs_running); + kv->reduce_by_symmetry_for_testing(ucell, symm, ModuleSymmetry::Symmetry::symm_flag, skpt, match, my_rank, ofs_running); EXPECT_EQ(kv->get_nkstot(), 35); ofs_running << skpt << std::endl; ofs_running.close(); @@ -878,15 +873,15 @@ TEST_F(KlistTest, IbzKpointIsMP) const int cal_symm_repr[2] = {0, 6}; symm.analy_sys(ucell.lat, ucell.st, ucell.atoms, ofs_running, 1e-6, 1, "scf", cal_symm_repr); std::string k_file = "./support/KPT1"; - kv->spin_mult = 1; - kv->read_kpoints(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank); + kv->set_spin_mult(1); + kv->read_kpoints_for_testing(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank); EXPECT_EQ(kv->get_nkstot(), 512); EXPECT_TRUE(kv->is_mp); // calculate ibz_kpoint std::string skpt; ModuleSymmetry::Symmetry::symm_flag = 0; bool match = true; - kv->reduce_by_symmetry(ucell, symm, ModuleSymmetry::Symmetry::symm_flag, skpt, match, my_rank, ofs_running); + kv->reduce_by_symmetry_for_testing(ucell, symm, ModuleSymmetry::Symmetry::symm_flag, skpt, match, my_rank, ofs_running); EXPECT_EQ(kv->get_nks(), 260); ofs_running << skpt << std::endl; ofs_running.close(); @@ -910,8 +905,8 @@ TEST_F(KlistTest, IbzKpointCustomWeights) { K_Vectors kv_test1; std::string k_file = "./support/KPT4"; - kv_test1.spin_mult = 1; - kv_test1.read_kpoints(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank); + kv_test1.set_spin_mult(1); + kv_test1.read_kpoints_for_testing(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank); EXPECT_EQ(kv_test1.get_nkstot(), 5); EXPECT_FALSE(kv_test1.is_mp); // Should be non-MP @@ -922,7 +917,7 @@ TEST_F(KlistTest, IbzKpointCustomWeights) std::string skpt; ModuleSymmetry::Symmetry::symm_flag = 1; bool match = true; - kv_test1.reduce_by_symmetry(ucell, symm, ModuleSymmetry::Symmetry::symm_flag, skpt, match, my_rank, ofs_running); + kv_test1.reduce_by_symmetry_for_testing(ucell, symm, ModuleSymmetry::Symmetry::symm_flag, skpt, match, my_rank, ofs_running); // Verify that weights are preserved (not overwritten with 1/nkstot) // After IBZ reduction, weights should still reflect the input weights @@ -939,8 +934,8 @@ TEST_F(KlistTest, IbzKpointCustomWeights) { K_Vectors kv_test2; std::string k_file = "./support/KPT_custom_weights"; - kv_test2.spin_mult = 1; - kv_test2.read_kpoints(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank); + kv_test2.set_spin_mult(1); + kv_test2.read_kpoints_for_testing(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank); EXPECT_EQ(kv_test2.get_nkstot(), 5); EXPECT_FALSE(kv_test2.is_mp); // Should be non-MP @@ -963,7 +958,7 @@ TEST_F(KlistTest, IbzKpointCustomWeights) std::string skpt; ModuleSymmetry::Symmetry::symm_flag = 1; bool match = true; - kv_test2.reduce_by_symmetry(ucell, symm, ModuleSymmetry::Symmetry::symm_flag, skpt, match, my_rank, ofs_running); + kv_test2.reduce_by_symmetry_for_testing(ucell, symm, ModuleSymmetry::Symmetry::symm_flag, skpt, match, my_rank, ofs_running); // After IBZ reduction, the weights should be based on the custom input weights, // not uniform 1/nkstot weights. The total weight should be preserved. @@ -996,8 +991,8 @@ TEST_F(KlistTest, IbzKpointCustomWeights) { K_Vectors kv_test3; std::string k_file = "./support/KPT1"; - kv_test3.spin_mult = 1; - kv_test3.read_kpoints(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank); + kv_test3.set_spin_mult(1); + kv_test3.read_kpoints_for_testing(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank); EXPECT_EQ(kv_test3.get_nkstot(), 512); EXPECT_TRUE(kv_test3.is_mp); // Should be MP @@ -1005,7 +1000,7 @@ TEST_F(KlistTest, IbzKpointCustomWeights) std::string skpt; ModuleSymmetry::Symmetry::symm_flag = 1; bool match = true; - kv_test3.reduce_by_symmetry(ucell, symm, ModuleSymmetry::Symmetry::symm_flag, skpt, match, my_rank, ofs_running); + kv_test3.reduce_by_symmetry_for_testing(ucell, symm, ModuleSymmetry::Symmetry::symm_flag, skpt, match, my_rank, ofs_running); // For MP grids, all weights should be uniform after IBZ reduction EXPECT_EQ(kv_test3.get_nkstot(), 35); // Known result from existing test @@ -1023,17 +1018,17 @@ TEST_F(KlistTest, IbzKpointCustomWeights) { K_Vectors kv_test4; std::string k_file = "./support/KPT_custom_weights"; - kv_test4.spin_mult = 1; - kv_test4.read_kpoints(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank); + kv_test4.set_spin_mult(1); + kv_test4.read_kpoints_for_testing(ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, ofs_running, ofs_warning, my_rank); // Apply IBZ reduction std::string skpt; ModuleSymmetry::Symmetry::symm_flag = 1; bool match = true; - kv_test4.reduce_by_symmetry(ucell, symm, ModuleSymmetry::Symmetry::symm_flag, skpt, match, my_rank, ofs_running); + kv_test4.reduce_by_symmetry_for_testing(ucell, symm, ModuleSymmetry::Symmetry::symm_flag, skpt, match, my_rank, ofs_running); // Normalize weights - int degspin = (kv_test4.spin_mult == 2) ? 1 : 2; + int degspin = (kv_test4.get_spin_mult() == 2) ? 1 : 2; kv_test4.normalize_wk(degspin); // After normalization, weights should sum to degspin diff --git a/source/source_cell/test/klist_test_para.cpp b/source/source_cell/test/klist_test_para.cpp index 2ebc0c609b5..0bd7552a853 100644 --- a/source/source_cell/test/klist_test_para.cpp +++ b/source/source_cell/test/klist_test_para.cpp @@ -9,7 +9,6 @@ #include "gtest/gtest.h" #include #include -#define private public #include "source_cell/klist.h" #include "source_cell/atom_pseudo.h" #include "source_cell/atom_spec.h" @@ -18,7 +17,6 @@ #include "source_cell/unitcell.h" #include "source_cell/magnetism.h" -#undef private pseudo::pseudo() { diff --git a/source/source_cell/test/qlist_test.cpp b/source/source_cell/test/qlist_test.cpp index c9c07e6dffb..637898edf44 100644 --- a/source/source_cell/test/qlist_test.cpp +++ b/source/source_cell/test/qlist_test.cpp @@ -5,14 +5,12 @@ #include #include #include -#define private public #include "source_cell/atom_pseudo.h" #include "source_cell/atom_spec.h" #include "source_cell/pseudo.h" #include "source_cell/qlist.h" #include "source_cell/unitcell.h" #include "source_cell/magnetism.h" -#undef private #include "source_base/mathzone.h" #include "source_base/parallel_global.h" #include "source_base/global_variable.h" diff --git a/source/source_cell/test/read_atoms_hlp_test.cpp b/source/source_cell/test/read_atoms_hlp_test.cpp index 77e6c49dc11..c536fa4eabf 100644 --- a/source/source_cell/test/read_atoms_hlp_test.cpp +++ b/source/source_cell/test/read_atoms_hlp_test.cpp @@ -1,11 +1,13 @@ #include "gtest/gtest.h" #include "gmock/gmock.h" #include "source_cell/read_atoms_helper.h" +#include "source_cell/print_cell.h" #include "source_base/vector3.h" #include "source_base/matrix3.h" #include "source_base/output.h" #include #include +#include // Mock implementations for missing functions that are not in the linked sources namespace elecstate { @@ -473,6 +475,590 @@ TEST_F(ReadAtomsHelperTest, ParseAtomPropertiesVectorMag) std::remove("test_input.tmp"); } +// Test parse_atom_properties with force field (round-trip compatibility) +TEST_F(ReadAtomsHelperTest, ParseAtomPropertiesForce) +{ + std::string input_str = "1.0 2.0 3.0 m 1 1 1 f 0.5 -0.3 0.2 mag 0.8333\n"; + + std::ofstream temp_file("test_input.tmp"); + temp_file << input_str; + temp_file.close(); + + std::ifstream ifpos("test_input.tmp"); + + Atom atom; + atom.label = "C"; + atom.vel.resize(1); + atom.mag.resize(1); + atom.m_loc_.resize(1); + atom.angle1.resize(1); + atom.angle2.resize(1); + atom.lambda.resize(1); + atom.constrain.resize(1); + + ModuleBase::Vector3 mv(1, 1, 1); + bool input_vec_mag = false; + bool input_angle_mag = false; + bool set_element_mag_zero = false; + + // Skip the position coordinates + double x, y, z; + ifpos >> x >> y >> z; + + bool result = unitcell::parse_atom_properties(ifpos, atom, 0, mv, + input_vec_mag, input_angle_mag, + set_element_mag_zero); + + EXPECT_TRUE(result); + EXPECT_EQ(mv.x, 1); + EXPECT_EQ(mv.y, 1); + EXPECT_EQ(mv.z, 1); + EXPECT_DOUBLE_EQ(atom.mag[0], 0.8333); + EXPECT_TRUE(set_element_mag_zero); + EXPECT_FALSE(ifpos.fail()); + + ifpos.close(); + std::remove("test_input.tmp"); +} + +// Test parse_atom_properties with negative force values +TEST_F(ReadAtomsHelperTest, ParseAtomPropertiesNegativeForce) +{ + std::string input_str = "1.0 2.0 3.0 m 1 0 1 f -1.0 -2.0 -3.0\n"; + + std::ofstream temp_file("test_input.tmp"); + temp_file << input_str; + temp_file.close(); + + std::ifstream ifpos("test_input.tmp"); + + Atom atom; + atom.label = "C"; + atom.vel.resize(1); + atom.mag.resize(1); + atom.m_loc_.resize(1); + atom.angle1.resize(1); + atom.angle2.resize(1); + atom.lambda.resize(1); + atom.constrain.resize(1); + + ModuleBase::Vector3 mv(1, 1, 1); + bool input_vec_mag = false; + bool input_angle_mag = false; + bool set_element_mag_zero = false; + + // Skip the position coordinates + double x, y, z; + ifpos >> x >> y >> z; + + bool result = unitcell::parse_atom_properties(ifpos, atom, 0, mv, + input_vec_mag, input_angle_mag, + set_element_mag_zero); + + EXPECT_TRUE(result); + EXPECT_EQ(mv.x, 1); + EXPECT_EQ(mv.y, 0); + EXPECT_EQ(mv.z, 1); + EXPECT_FALSE(ifpos.fail()); + + ifpos.close(); + std::remove("test_input.tmp"); +} + +// Round-trip integration: a STRU atom line in the exact format produced by +// print_stru_file (positions + m + optional f) must be parseable by +// parse_atom_properties without leaving the stream in a fail state. +// This catches the round-trip regression reported in issue #8051. +TEST_F(ReadAtomsHelperTest, RoundTripWriterReaderForce) +{ + // Line as written by print_stru_file with has_force=true + std::string input_str = "0.000000000000 0.000000000000 0.000000000000 m 1 1 1 f -0.123456000000 0.234567000000 -0.345678000000\n"; + + std::ofstream temp_file("test_input.tmp"); + temp_file << input_str; + temp_file.close(); + + std::ifstream ifpos("test_input.tmp"); + + Atom atom; + atom.label = "Fe"; + atom.vel.resize(1); + atom.mag.resize(1); + atom.m_loc_.resize(1); + atom.angle1.resize(1); + atom.angle2.resize(1); + atom.lambda.resize(1); + atom.constrain.resize(1); + + ModuleBase::Vector3 mv(0, 0, 0); + bool input_vec_mag = false; + bool input_angle_mag = false; + bool set_element_mag_zero = false; + + double x, y, z; + ifpos >> x >> y >> z; + + bool result = unitcell::parse_atom_properties(ifpos, atom, 0, mv, + input_vec_mag, input_angle_mag, + set_element_mag_zero); + + EXPECT_TRUE(result); + EXPECT_EQ(mv.x, 1); + EXPECT_EQ(mv.y, 1); + EXPECT_EQ(mv.z, 1); + // Stream must not be in a fail state -- this was the bug in issue #8051: + // the reader did not know "f" and consumed the force values as the next + // keyword, leaving the stream corrupted for subsequent atoms. + EXPECT_FALSE(ifpos.fail()); + EXPECT_TRUE(ifpos.good() || ifpos.eof()); + + ifpos.close(); + std::remove("test_input.tmp"); +} + +// Multi-atom round-trip: write a real 3-atom spin-polarized cell with +// print_stru_file (has_force=true, nspin=2) and parse the produced file +// back through parse_atom_properties. Positions, movement flags and the +// per-atom mag value are asserted for every ia, which catches stream +// corruption that only appears when na > 1 (issue #8051). +TEST_F(ReadAtomsHelperTest, RoundTripMultiAtomMixedFields) +{ + const int nat = 3; + + // Reference data: Cartesian positions in Bohr (units of lat0), + // movement flags and initial magnetic moments, one row per atom. + const double pos_bohr[3][3] = { + {0.0, 0.0, 0.0}, + {1.0, 0.0, 0.0}, + {0.5, 0.5, 0.5} + }; + const int mbl_int[3][3] = { + {1, 1, 1}, + {0, 1, 0}, + {1, 0, 1} + }; + const double mag_expected[3] = {1.5, -0.7, 2.25}; + + UnitCell ucell; + ucell.ntype = 1; + ucell.nat = nat; + ucell.atoms = new Atom[1]; + ucell.lat0 = 1.0; + ucell.omega = 1.0; + ucell.latvec.Identity(); + ucell.pseudo_fn.resize(1); + ucell.pseudo_type.resize(1); + ucell.pseudo_fn[0] = "Fe.upf"; + ucell.pseudo_type[0] = "uspp"; + ucell.magnet.start_mag.resize(1); + ucell.magnet.start_mag[0] = 0.0; + + Atom& fe = ucell.atoms[0]; + fe.na = nat; + fe.label = "Fe"; + fe.mass = 55.847; + fe.tau.resize(nat); + fe.mbl.resize(nat); + fe.mag.resize(nat); + fe.vel.resize(nat); + fe.m_loc_.resize(nat); + fe.angle1.resize(nat); + fe.angle2.resize(nat); + fe.lambda.resize(nat); + fe.constrain.resize(nat); + for (int ia = 0; ia < nat; ++ia) + { + fe.tau[ia] = ModuleBase::Vector3(pos_bohr[ia][0], pos_bohr[ia][1], pos_bohr[ia][2]); + fe.mbl[ia] = ModuleBase::Vector3(mbl_int[ia][0], mbl_int[ia][1], mbl_int[ia][2]); + fe.mag[ia] = mag_expected[ia]; + } + + // Forces in Ry/Bohr; the writer converts them to eV/Angstrom. + ModuleBase::matrix force(nat, 3); + for (int ia = 0; ia < nat; ++ia) + { + force(ia, 0) = 0.01 * (ia + 1); + force(ia, 1) = -0.02 * (ia + 1); + force(ia, 2) = 0.03 * (ia + 1); + } + + const std::string filename = "test_stru_roundtrip.tmp"; + unitcell::print_stru_file(ucell, ucell.atoms, ucell.latvec, filename, "", + 2, false, false, false, false, 0, force, true); + + std::ifstream ifpos(filename.c_str()); + + // Locate the ATOMIC_POSITIONS section. + std::string keyword; + while (ifpos >> keyword) + { + if (keyword == "ATOMIC_POSITIONS") + { + break; + } + } + ASSERT_FALSE(ifpos.fail()); + + // Coordinate descriptor line. + ifpos >> keyword; + EXPECT_EQ(keyword, "Cartesian_angstrom"); + ifpos.ignore(std::numeric_limits::max(), '\n'); + + // Type block header: label, default magnetism, atom count. + std::string label; + ifpos >> label; + EXPECT_EQ(label, "Fe"); + ifpos.ignore(std::numeric_limits::max(), '\n'); + double header_mag = 0.0; + ifpos >> header_mag; + ifpos.ignore(std::numeric_limits::max(), '\n'); + int na_read = 0; + ifpos >> na_read; + ASSERT_EQ(na_read, nat); + ifpos.ignore(std::numeric_limits::max(), '\n'); + + // Parser target: every indexed buffer must cover all ia. + Atom parse_atom; + parse_atom.label = "Fe"; + parse_atom.vel.resize(nat); + parse_atom.mag.resize(nat); + parse_atom.m_loc_.resize(nat); + parse_atom.angle1.resize(nat); + parse_atom.angle2.resize(nat); + parse_atom.lambda.resize(nat); + parse_atom.constrain.resize(nat); + + const double pos_conv = ucell.lat0 * ModuleBase::BOHR_TO_A; + for (int ia = 0; ia < nat; ++ia) + { + double x = 0.0; + double y = 0.0; + double z = 0.0; + ifpos >> x >> y >> z; + ASSERT_FALSE(ifpos.fail()); + + ModuleBase::Vector3 mv(0, 0, 0); + bool input_vec_mag = false; + bool input_angle_mag = false; + bool set_element_mag_zero = false; + + const bool ok = unitcell::parse_atom_properties(ifpos, parse_atom, ia, mv, + input_vec_mag, input_angle_mag, + set_element_mag_zero); + EXPECT_TRUE(ok); + + const double expected_x = pos_bohr[ia][0] * pos_conv; + const double expected_y = pos_bohr[ia][1] * pos_conv; + const double expected_z = pos_bohr[ia][2] * pos_conv; + EXPECT_NEAR(x, expected_x, 1e-8); + EXPECT_NEAR(y, expected_y, 1e-8); + EXPECT_NEAR(z, expected_z, 1e-8); + + EXPECT_EQ(mv.x, mbl_int[ia][0]); + EXPECT_EQ(mv.y, mbl_int[ia][1]); + EXPECT_EQ(mv.z, mbl_int[ia][2]); + + EXPECT_NEAR(parse_atom.mag[ia], mag_expected[ia], 1e-10); + } + + // One read past the last consumed newline reaches end-of-file. + ifpos.get(); + EXPECT_TRUE(ifpos.eof()); + ifpos.close(); + std::remove(filename.c_str()); + delete[] ucell.atoms; +} + +// ============================================================ +// Tests for issue #5021: nspin=4 with noncolin=0 discards x/y +// ============================================================ + +TEST_F(ReadAtomsHelperTest, ProcessMagnetizationNspin4CollinearReportsXYDiscard) +{ + Atom atom; + atom.label = "Fe"; + atom.mag.resize(1); + atom.m_loc_.resize(1); + atom.angle1.resize(1); + atom.angle2.resize(1); + + atom.m_loc_[0].set(1.0, 0.5, 0.3); + atom.mag[0] = sqrt(1.0 + 0.25 + 0.09); + + const int nspin = 4; + const bool input_vec_mag = true; + const bool input_angle_mag = false; + const bool noncolin = false; + + // The discard is reported to the caller, not printed once per atom. + testing::internal::CaptureStdout(); + const bool xy_discarded = unitcell::process_magnetization( + atom, 0, 0, nspin, input_vec_mag, input_angle_mag, ofs_running, noncolin); + const std::string stdout_output = testing::internal::GetCapturedStdout(); + + EXPECT_TRUE(xy_discarded); + EXPECT_DOUBLE_EQ(atom.m_loc_[0].x, 0.0); + EXPECT_DOUBLE_EQ(atom.m_loc_[0].y, 0.0); + EXPECT_DOUBLE_EQ(atom.m_loc_[0].z, 0.3); + EXPECT_EQ(stdout_output.find("Warning"), std::string::npos); +} + +TEST_F(ReadAtomsHelperTest, ProcessMagnetizationNspin4CollinearNoDiscardOnZeroXY) +{ + Atom atom; + atom.label = "Fe"; + atom.mag.resize(1); + atom.m_loc_.resize(1); + atom.angle1.resize(1); + atom.angle2.resize(1); + + atom.m_loc_[0].set(0.0, 0.0, 0.3); + atom.mag[0] = 0.3; + + const int nspin = 4; + const bool input_vec_mag = true; + const bool input_angle_mag = false; + const bool noncolin = false; + + const bool xy_discarded = unitcell::process_magnetization( + atom, 0, 0, nspin, input_vec_mag, input_angle_mag, ofs_running, noncolin); + + EXPECT_FALSE(xy_discarded); + EXPECT_DOUBLE_EQ(atom.m_loc_[0].x, 0.0); + EXPECT_DOUBLE_EQ(atom.m_loc_[0].y, 0.0); + EXPECT_DOUBLE_EQ(atom.m_loc_[0].z, 0.3); +} + +TEST_F(ReadAtomsHelperTest, ProcessMagnetizationNspin4NoncolinKeepsXY) +{ + Atom atom; + atom.label = "Fe"; + atom.mag.resize(1); + atom.m_loc_.resize(1); + atom.angle1.resize(1); + atom.angle2.resize(1); + + atom.m_loc_[0].set(1.0, 0.5, 0.3); + atom.mag[0] = sqrt(1.0 + 0.25 + 0.09); + + const int nspin = 4; + const bool input_vec_mag = true; + const bool input_angle_mag = false; + const bool noncolin = true; + + const bool xy_discarded = unitcell::process_magnetization( + atom, 0, 0, nspin, input_vec_mag, input_angle_mag, ofs_running, noncolin); + + EXPECT_FALSE(xy_discarded); + EXPECT_DOUBLE_EQ(atom.m_loc_[0].x, 1.0); + EXPECT_DOUBLE_EQ(atom.m_loc_[0].y, 0.5); + EXPECT_DOUBLE_EQ(atom.m_loc_[0].z, 0.3); +} + +TEST_F(ReadAtomsHelperTest, WarnXYMagnetizationIgnoredAggregatesPerType) +{ + testing::internal::CaptureStdout(); + unitcell::warn_xy_magnetization_ignored("Fe", 3, 4, ofs_running); + const std::string stdout_output = testing::internal::GetCapturedStdout(); + + // A single message for the whole type, carrying the affected-atom count. + const std::string::size_type first = stdout_output.find("Warning"); + EXPECT_NE(first, std::string::npos); + EXPECT_NE(stdout_output.find("atom type Fe: 3 of 4 atoms"), std::string::npos); + EXPECT_NE(stdout_output.find("x/y components are IGNORED"), std::string::npos); + EXPECT_EQ(stdout_output.find("Warning", first + 1), std::string::npos); + + ofs_running.flush(); + std::ifstream ifs("test_running.log"); + std::stringstream buffer; + buffer << ifs.rdbuf(); + const std::string log_content = buffer.str(); + ifs.close(); + EXPECT_NE(log_content.find("x/y components are IGNORED"), std::string::npos); +} + +TEST_F(ReadAtomsHelperTest, WarnXYMagnetizationIgnoredSilentWhenNoneDiscarded) +{ + testing::internal::CaptureStdout(); + unitcell::warn_xy_magnetization_ignored("Fe", 0, 4, ofs_running); + const std::string stdout_output = testing::internal::GetCapturedStdout(); + + EXPECT_EQ(stdout_output.find("Warning"), std::string::npos); +} + +// ============================================================ +// Tests for issue #5939: nspin=4 no longer autosets (1,1,1) +// ============================================================ + +TEST_F(ReadAtomsHelperTest, IsMagnetizationAllZero) +{ + UnitCell ucell; + ucell.ntype = 1; + ucell.atoms = new Atom[1]; + ucell.atoms[0].label = "Fe"; + ucell.atoms[0].na = 2; + ucell.atoms[0].mag.resize(2, 0.0); + ucell.atoms[0].m_loc_.resize(2); + + EXPECT_TRUE(unitcell::is_magnetization_all_zero(ucell)); + + ucell.atoms[0].mag[1] = 1.5; + EXPECT_FALSE(unitcell::is_magnetization_all_zero(ucell)); + + delete[] ucell.atoms; +} + +TEST_F(ReadAtomsHelperTest, AutosetMagnetizationNspin4NoAutosetAndSilent) +{ + UnitCell ucell; + ucell.ntype = 1; + ucell.atoms = new Atom[1]; + ucell.atoms[0].label = "Fe"; + ucell.atoms[0].na = 1; + ucell.atoms[0].mag.resize(1, 0.0); + ucell.atoms[0].m_loc_.resize(1); + ucell.atoms[0].m_loc_[0].set(0.0, 0.0, 0.0); + + const int nspin = 4; + + testing::internal::CaptureStdout(); + unitcell::autoset_magnetization(ucell, nspin, ofs_running); + const std::string stdout_output = testing::internal::GetCapturedStdout(); + + // Magnetization stays zero (NOT autoset to (1,1,1)), and this function + // stays silent: reporting belongs to warn_zero_magnetization_nspin4(), + // which read_atom_positions() calls for every symmetry setting. + EXPECT_DOUBLE_EQ(ucell.atoms[0].m_loc_[0].x, 0.0); + EXPECT_DOUBLE_EQ(ucell.atoms[0].m_loc_[0].y, 0.0); + EXPECT_DOUBLE_EQ(ucell.atoms[0].m_loc_[0].z, 0.0); + EXPECT_DOUBLE_EQ(ucell.atoms[0].mag[0], 0.0); + EXPECT_EQ(stdout_output.find("Warning"), std::string::npos); + + delete[] ucell.atoms; +} + +TEST_F(ReadAtomsHelperTest, WarnZeroMagnetizationNspin4) +{ + UnitCell ucell; + ucell.ntype = 1; + ucell.atoms = new Atom[1]; + ucell.atoms[0].label = "Fe"; + ucell.atoms[0].na = 1; + ucell.atoms[0].mag.resize(1, 0.0); + ucell.atoms[0].m_loc_.resize(1); + ucell.atoms[0].m_loc_[0].set(0.0, 0.0, 0.0); + + testing::internal::CaptureStdout(); + unitcell::warn_zero_magnetization_nspin4(ucell, ofs_running); + const std::string stdout_output = testing::internal::GetCapturedStdout(); + + EXPECT_NE(stdout_output.find("Warning"), std::string::npos); + EXPECT_NE(stdout_output.find("no initial magnetization is set in STRU"), + std::string::npos); + + ofs_running.flush(); + std::ifstream ifs("test_running.log"); + std::stringstream buffer; + buffer << ifs.rdbuf(); + const std::string log_content = buffer.str(); + ifs.close(); + EXPECT_NE(log_content.find("no initial magnetization is set in STRU"), + std::string::npos); + + delete[] ucell.atoms; +} + +TEST_F(ReadAtomsHelperTest, WarnZeroMagnetizationNspin4KeysOnMLoc) +{ + // Regression guard: a three-component magmom lying purely in x/y leaves + // mag at the full norm (2.0) while m_loc_ has already been projected to + // zero by the noncolin=0 branch. Keying this check on mag would silence + // the warning in exactly that case. + UnitCell ucell; + ucell.ntype = 1; + ucell.atoms = new Atom[1]; + ucell.atoms[0].label = "Fe"; + ucell.atoms[0].na = 1; + ucell.atoms[0].mag.resize(1, 2.0); + ucell.atoms[0].m_loc_.resize(1); + ucell.atoms[0].m_loc_[0].set(0.0, 0.0, 0.0); + + EXPECT_FALSE(unitcell::is_magnetization_all_zero(ucell)); + + testing::internal::CaptureStdout(); + unitcell::warn_zero_magnetization_nspin4(ucell, ofs_running); + const std::string stdout_output = testing::internal::GetCapturedStdout(); + + EXPECT_NE(stdout_output.find("no initial magnetization is set in STRU"), + std::string::npos); + + delete[] ucell.atoms; +} + +TEST_F(ReadAtomsHelperTest, WarnZeroMagnetizationNspin4SilentWhenMagSet) +{ + UnitCell ucell; + ucell.ntype = 1; + ucell.atoms = new Atom[1]; + ucell.atoms[0].label = "Fe"; + ucell.atoms[0].na = 1; + ucell.atoms[0].mag.resize(1, 1.5); + ucell.atoms[0].m_loc_.resize(1); + ucell.atoms[0].m_loc_[0].set(0.0, 0.0, 1.5); + + testing::internal::CaptureStdout(); + unitcell::warn_zero_magnetization_nspin4(ucell, ofs_running); + const std::string stdout_output = testing::internal::GetCapturedStdout(); + + EXPECT_EQ(stdout_output.find("Warning"), std::string::npos); + + delete[] ucell.atoms; +} + +TEST_F(ReadAtomsHelperTest, AutosetMagnetizationNspin2StillAutoset) +{ + UnitCell ucell; + ucell.ntype = 1; + ucell.atoms = new Atom[1]; + ucell.atoms[0].label = "Fe"; + ucell.atoms[0].na = 1; + ucell.atoms[0].mag.resize(1, 0.0); + ucell.atoms[0].m_loc_.resize(1); + ucell.atoms[0].m_loc_[0].set(0.0, 0.0, 0.0); + + const int nspin = 2; + + testing::internal::CaptureStdout(); + unitcell::autoset_magnetization(ucell, nspin, ofs_running); + const std::string stdout_output = testing::internal::GetCapturedStdout(); + + // nspin=2 still autosets to 1.0 (VASP-compatible behavior) + EXPECT_DOUBLE_EQ(ucell.atoms[0].mag[0], 1.0); + EXPECT_DOUBLE_EQ(ucell.atoms[0].m_loc_[0].z, 1.0); + EXPECT_EQ(stdout_output.find("Warning"), std::string::npos); + + delete[] ucell.atoms; +} + +TEST_F(ReadAtomsHelperTest, AutosetMagnetizationNspin2SkippedWhenMagSet) +{ + UnitCell ucell; + ucell.ntype = 1; + ucell.atoms = new Atom[1]; + ucell.atoms[0].label = "Fe"; + ucell.atoms[0].na = 1; + ucell.atoms[0].mag.resize(1, 1.5); + ucell.atoms[0].m_loc_.resize(1); + ucell.atoms[0].m_loc_[0].set(0.0, 0.0, 1.5); + + unitcell::autoset_magnetization(ucell, 2, ofs_running); + + EXPECT_DOUBLE_EQ(ucell.atoms[0].mag[0], 1.5); + EXPECT_DOUBLE_EQ(ucell.atoms[0].m_loc_[0].z, 1.5); + + delete[] ucell.atoms; +} + int main(int argc, char **argv) { ::testing::InitGoogleTest(&argc, argv); diff --git a/source/source_cell/test/sepcell_test.cpp b/source/source_cell/test/sepcell_test.cpp index 811c1277a2d..f06237779dd 100644 --- a/source/source_cell/test/sepcell_test.cpp +++ b/source/source_cell/test/sepcell_test.cpp @@ -9,10 +9,8 @@ #include #endif -#define private public #include "source_cell/sep_cell.h" #include "source_cell/unitcell.h" -#undef private pseudo::pseudo() { } diff --git a/source/source_cell/test/test_cal_ux.cpp b/source/source_cell/test/test_cal_ux.cpp new file mode 100644 index 00000000000..4ee3c864456 --- /dev/null +++ b/source/source_cell/test/test_cal_ux.cpp @@ -0,0 +1,87 @@ +#include "gmock/gmock.h" +#include "gtest/gtest.h" + +#include +#include +#include + +#include "source_base/global_variable.h" +#include "source_base/mathzone.h" +#include "source_cell/unitcell.h" +#include "source_cell/cal_ux.h" +#include "prepare_unitcell.h" + +// The test-only cell_info object library does not contain magnetism.cpp, +// so the Magnetism constructor/destructor must be provided locally. +Magnetism::Magnetism() +{ + this->tot_mag = 0.0; + this->abs_mag = 0.0; +} +Magnetism::~Magnetism() +{ +} + +/************************************************ + * unit test of cal_ux.cpp + ***********************************************/ + +/** + * - Tested Functions: + * - JudgeParallel + * - judge_parallel: judge if two vectors a[3] and Vector3 b are parallel + * - CalUx1 + * - cal_ux: non-parallel initial moments, no common direction + * - CalUx2 + * - cal_ux: parallel moments, normalized common direction (1,1,1)/sqrt(3) + */ + +class CalUxTest : public ::testing::Test +{ + protected: + std::unique_ptr ucell; +}; + +TEST_F(CalUxTest, JudgeParallel) +{ + ModuleBase::Vector3 b(1.0, 1.0, 1.0); + double a[3] = {1.0, 1.0, 1.0}; + EXPECT_TRUE(unitcell::judge_parallel(a, b)); + + // the negative case, moved here from MagnetismTest.JudgeParallel when the + // duplicate Magnetism::judge_parallel was deleted + double c[3] = {1.0, 0.0, 0.0}; + ModuleBase::Vector3 d(0.0, 1.0, 0.0); + EXPECT_FALSE(unitcell::judge_parallel(c, d)); +} + +TEST_F(CalUxTest, CalUx1) +{ + UcellTestPrepare utp = UcellTestLib["C1H2-Read"]; + ucell = utp.SetUcellInfo(); + ucell->atoms[0].m_loc_[0].set(0, -1, 0); + ucell->atoms[1].m_loc_[0].set(1, 1, 1); + ucell->atoms[1].m_loc_[1].set(0, 0, 0); + const int nspin = 4; + unitcell::cal_ux(*ucell, nspin); + EXPECT_FALSE(ucell->magnet.lsign_); + EXPECT_DOUBLE_EQ(ucell->magnet.ux_[0], 0); + EXPECT_DOUBLE_EQ(ucell->magnet.ux_[1], -1); + EXPECT_DOUBLE_EQ(ucell->magnet.ux_[2], 0); +} + +TEST_F(CalUxTest, CalUx2) +{ + UcellTestPrepare utp = UcellTestLib["C1H2-Read"]; + ucell = utp.SetUcellInfo(); + ucell->atoms[0].m_loc_[0].set(0, 0, 0); + ucell->atoms[1].m_loc_[0].set(1, 1, 1); + ucell->atoms[1].m_loc_[1].set(0, 0, 0); + //(0,0,0) is also parallel to (1,1,1) + const int nspin = 4; + unitcell::cal_ux(*ucell, nspin); + EXPECT_TRUE(ucell->magnet.lsign_); + EXPECT_NEAR(ucell->magnet.ux_[0], 0.57735, 1e-5); + EXPECT_NEAR(ucell->magnet.ux_[1], 0.57735, 1e-5); + EXPECT_NEAR(ucell->magnet.ux_[2], 0.57735, 1e-5); +} diff --git a/source/source_cell/test/test_cell_tools.cpp b/source/source_cell/test/test_cell_tools.cpp new file mode 100644 index 00000000000..267849f2b57 --- /dev/null +++ b/source/source_cell/test/test_cell_tools.cpp @@ -0,0 +1,91 @@ +#include "gmock/gmock.h" +#include "gtest/gtest.h" + +#include +#include +#include + +#include "source_base/global_variable.h" +#include "source_base/mathzone.h" +#include "source_cell/unitcell.h" +#include "source_cell/cell_tools.h" +#include "prepare_unitcell.h" + +// The test-only cell_info object library does not contain magnetism.cpp, +// so the Magnetism constructor/destructor must be provided locally. +Magnetism::Magnetism() +{ + this->tot_mag = 0.0; + this->abs_mag = 0.0; +} +Magnetism::~Magnetism() +{ +} + +/************************************************ + * unit test of cell_tools.cpp + ***********************************************/ + +/** + * - Tested Functions: + * - IfCellCanChange + * - if_cell_can_change(): truth table over the three lattice-axis flags + * - GetAtomCounts + * - get_atomCounts(): number of atoms per type as a vector + * - GetLnchiCounts + * - get_lnchiCounts(): number of chi functions per L per type + * - SelectiveDynamics + * - if_atoms_can_move(): true if any atom has a movable coordinate + */ + +class CellToolsTest : public ::testing::Test +{ + protected: + std::unique_ptr ucell{new UnitCell}; +}; + +TEST_F(CellToolsTest, IfCellCanChange) +{ + // Mirror the fixed_axes -> lat_axis_free mapping produced by + // UnitCell::setup_from_input: the cell can change whenever at least one + // lattice axis is free; only "abc" (all axes fixed) returns false. + std::vector> axes_free = { + {1, 1, 1}, {0, 1, 1}, {1, 0, 1}, {1, 1, 0}, + {0, 0, 1}, {0, 1, 0}, {1, 0, 0}, {0, 0, 0}}; + for (int i = 0; i < 7; ++i) + { + EXPECT_TRUE(unitcell::if_cell_can_change(axes_free[i])); + } + EXPECT_FALSE(unitcell::if_cell_can_change(axes_free[7])); +} + +TEST_F(CellToolsTest, GetAtomCounts) +{ + UcellTestPrepare utp = UcellTestLib["C1H2-Index"]; + ucell = utp.SetUcellInfo(); + ucell->set_iat2itia(); + std::vector atomCounts = unitcell::get_atomCounts(ucell->atoms, ucell->ntype); + EXPECT_EQ(atomCounts[0], 1); + EXPECT_EQ(atomCounts[1], 2); +} + +TEST_F(CellToolsTest, GetLnchiCounts) +{ + UcellTestPrepare utp = UcellTestLib["C1H2-Index"]; + ucell = utp.SetUcellInfo(); + ucell->set_iat2itia(); + std::vector> lnchiCounts = unitcell::get_lnchiCounts(ucell->atoms, ucell->ntype); + EXPECT_EQ(lnchiCounts[0][0], 1); + EXPECT_EQ(lnchiCounts[0][1], 1); + EXPECT_EQ(lnchiCounts[0][2], 1); + EXPECT_EQ(lnchiCounts[1][0], 1); + EXPECT_EQ(lnchiCounts[1][1], 1); + EXPECT_EQ(lnchiCounts[1][2], 1); +} + +TEST_F(CellToolsTest, SelectiveDynamics) +{ + UcellTestPrepare utp = UcellTestLib["C1H2-SD"]; + ucell = utp.SetUcellInfo(); + EXPECT_TRUE(unitcell::if_atoms_can_move(ucell->atoms, ucell->ntype)); +} diff --git a/source/source_cell/test/test_print_cell.cpp b/source/source_cell/test/test_print_cell.cpp new file mode 100644 index 00000000000..b1d131ffcb4 --- /dev/null +++ b/source/source_cell/test/test_print_cell.cpp @@ -0,0 +1,392 @@ +#include "gmock/gmock.h" +#include "gtest/gtest.h" + +#include "source_cell/cal_ux.h" +#include "source_cell/read_orb.h" +#include "source_cell/read_pp_ucell.h" +#include "source_cell/read_stru.h" +#include "source_cell/cell_tools.h" +#include "source_cell/print_cell.h" +#include "memory" +#include "source_base/global_variable.h" +#include "source_base/mathzone.h" +#include "prepare_unitcell.h" +#include "source_cell/update_cell.h" +#include +#include +#include +#include +#include + +Magnetism::Magnetism() +{ + this->tot_mag = 0.0; + this->abs_mag = 0.0; +} +Magnetism::~Magnetism() +{ +} + +class PrintCellTest : public testing::Test +{ +protected: + std::unique_ptr ucell; + void SetUp() override + { + // nothing to do here, each test sets up its own ucell + } + void TearDown() override + { + // cleanup generated files + } +}; + +TEST_F(PrintCellTest, PrintSTRU_nspin1) +{ + UcellTestPrepare utp = UcellTestLib["C1H2-Index"]; + ucell = utp.SetUcellInfo(); + std::string fn = "C1H2_STRU_nspin1"; + + unitcell::print_stru_file(*ucell, ucell->atoms, ucell->latvec, + fn, "", 1, false, false, false, false, 0, ModuleBase::matrix(), false); + std::ifstream ifs; + ifs.open(fn); + std::string str((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + EXPECT_THAT(str, testing::HasSubstr("ATOMIC_SPECIES")); + EXPECT_THAT(str, testing::HasSubstr("C 12.0000 C.upf upf201")); + EXPECT_THAT(str, testing::HasSubstr("H 1.0000 H.upf upf201")); + EXPECT_THAT(str, testing::HasSubstr("LATTICE_CONSTANT")); + // lat0 is always printed as one Angstrom in Bohr + EXPECT_THAT(str, testing::HasSubstr( + "1.8897268778 # in Bohr (= 1 Angstrom); lattice vectors below are in Angstrom")); + EXPECT_THAT(str, testing::HasSubstr("LATTICE_VECTORS # in Angstrom")); + EXPECT_THAT(str, testing::HasSubstr("ATOMIC_POSITIONS")); + // positions are always emitted as Cartesian_angstrom + EXPECT_THAT(str, testing::HasSubstr("Cartesian_angstrom # positions in Angstrom")); + EXPECT_THAT(str, testing::HasSubstr("C #label")); + EXPECT_THAT(str, testing::HasSubstr("0.0000 #magnetism (default, overridden by per-atom mag below)")); + EXPECT_THAT(str, testing::HasSubstr("1 #number of atoms")); + // single-space separated coordinates + EXPECT_THAT(str, testing::HasSubstr("0.9999996019 0.9999996019 0.9999996019 m 1 1 1")); + EXPECT_THAT(str, testing::HasSubstr("H #label")); + EXPECT_THAT(str, testing::HasSubstr("0.0000 #magnetism (default, overridden by per-atom mag below)")); + EXPECT_THAT(str, testing::HasSubstr("2 #number of atoms")); + EXPECT_THAT(str, testing::HasSubstr("1.4999994028 1.4999994028 1.4999994028 m 0 0 0")); + EXPECT_THAT(str, testing::HasSubstr("0.4999998009 0.4999998009 0.4999998009 m 0 0 1")); + // No force output when force matrix is empty + EXPECT_THAT(str, testing::Not(testing::HasSubstr(" f "))); + ifs.close(); + remove(fn.c_str()); +} + +TEST_F(PrintCellTest, PrintSTRU_zero_force_computed) +{ + UcellTestPrepare utp = UcellTestLib["C1H2-Index"]; + ucell = utp.SetUcellInfo(); + std::string fn = "C1H2_STRU_zero_force"; + + // has_force=true with an all-zero force matrix: genuinely computed zero + // forces must still be emitted, unlike "not computed" which omits them. + ModuleBase::matrix force(3, 3); + force.zero_out(); + + unitcell::print_stru_file(*ucell, ucell->atoms, ucell->latvec, + fn, "", 2, false, false, false, false, 0, force, true); + std::ifstream ifs; + ifs.open(fn); + std::string str((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + + EXPECT_THAT(str, testing::HasSubstr("Cartesian_angstrom # positions in Angstrom, forces in eV/Angstrom")); + EXPECT_THAT(str, testing::HasSubstr(" f 0.000000 0.000000 0.000000")); + ifs.close(); + remove(fn.c_str()); +} + +TEST_F(PrintCellTest, PrintSTRU_force_matrix_without_flag) +{ + UcellTestPrepare utp = UcellTestLib["C1H2-Index"]; + ucell = utp.SetUcellInfo(); + std::string fn = "C1H2_STRU_flag_off"; + + // Force matrix is correctly sized but has_force=false: the flag takes + // priority and no f fields are emitted. + ModuleBase::matrix force(3, 3); + force(0, 0) = 0.1; force(0, 1) = 0.2; force(0, 2) = 0.3; + + unitcell::print_stru_file(*ucell, ucell->atoms, ucell->latvec, + fn, "", 2, false, false, false, false, 0, force, false); + std::ifstream ifs; + ifs.open(fn); + std::string str((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + + EXPECT_THAT(str, testing::Not(testing::HasSubstr(" f "))); + ifs.close(); + remove(fn.c_str()); +} + +TEST_F(PrintCellTest, PrintSTRU_nspin2_no_force) +{ + UcellTestPrepare utp = UcellTestLib["C1H2-Index"]; + ucell = utp.SetUcellInfo(); + std::string fn = "C1H2_STRU_nspin2"; + + unitcell::print_stru_file(*ucell, ucell->atoms, ucell->latvec, + fn, "", 2, true, false, false, false, 0, ModuleBase::matrix(), false); + std::ifstream ifs; + ifs.open(fn); + std::string str((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + EXPECT_THAT(str, testing::HasSubstr("ATOMIC_POSITIONS")); + // positions are always Cartesian_angstrom, even without forces + EXPECT_THAT(str, testing::HasSubstr("Cartesian_angstrom # positions in Angstrom\n")); + EXPECT_THAT(str, testing::HasSubstr("C #label")); + EXPECT_THAT(str, testing::HasSubstr("0.0000 #magnetism (default, overridden by per-atom mag below)")); + EXPECT_THAT(str, + testing::HasSubstr("0.9999996019 0.9999996019 0.9999996019 m 1 1 1 v 0.1000000000 0.1000000000 0.1000000000 mag 0.0000")); + EXPECT_THAT(str, testing::HasSubstr("H #label")); + EXPECT_THAT(str, + testing::HasSubstr("1.4999994028 1.4999994028 1.4999994028 m 0 0 0 v 0.1000000000 0.1000000000 0.1000000000 mag 0.0000")); + EXPECT_THAT(str, + testing::HasSubstr("0.4999998009 0.4999998009 0.4999998009 m 0 0 1 v 0.1000000000 0.1000000000 0.1000000000 mag 0.0000")); + // No force output when force matrix is empty + EXPECT_THAT(str, testing::Not(testing::HasSubstr(" f "))); + ifs.close(); + remove(fn.c_str()); +} + +TEST_F(PrintCellTest, PrintSTRU_nspin2_with_mag) +{ + UcellTestPrepare utp = UcellTestLib["C1H2-Index"]; + ucell = utp.SetUcellInfo(); + std::string fn = "C1H2_STRU_nspin2_mag"; + + ucell->atoms[0].mag[0] = 1.5; // C + ucell->atoms[1].mag[0] = -0.5; // H1 + ucell->atoms[1].mag[1] = 2.0; // H2 + unitcell::print_stru_file(*ucell, ucell->atoms, ucell->latvec, + fn, "", 2, false, false, false, false, 0, ModuleBase::matrix(), false); + std::ifstream ifs; + ifs.open(fn); + std::string str((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + EXPECT_THAT(str, testing::HasSubstr("C #label")); + EXPECT_THAT(str, testing::HasSubstr("1.5000 #magnetism (default, overridden by per-atom mag below)")); + EXPECT_THAT(str, + testing::HasSubstr("0.9999996019 0.9999996019 0.9999996019 m 1 1 1 mag 1.5000")); + EXPECT_THAT(str, testing::HasSubstr("H #label")); + EXPECT_THAT(str, testing::HasSubstr("-0.5000 #magnetism (default, overridden by per-atom mag below)")); + EXPECT_THAT(str, + testing::HasSubstr("1.4999994028 1.4999994028 1.4999994028 m 0 0 0 mag -0.5000")); + EXPECT_THAT(str, + testing::HasSubstr("0.4999998009 0.4999998009 0.4999998009 m 0 0 1 mag 2.0000")); + ifs.close(); + remove(fn.c_str()); +} + +TEST_F(PrintCellTest, PrintSTRU_nspin2_mulliken) +{ + UcellTestPrepare utp = UcellTestLib["C1H2-Index"]; + ucell = utp.SetUcellInfo(); + std::string fn = "C1H2_STRU_nspin2_mulliken"; + + ucell->descriptor_file = "__unittest_numerical_descriptor__"; + ucell->orbital_fn[0] = "__unittest_orbital_fn_0__"; + ucell->orbital_fn[1] = "__unittest_orbital_fn_1__"; + ucell->atom_mulliken = {{-1, 0.5}, {-1, 0.4}, {-1, 0.3}}; + unitcell::print_stru_file(*ucell, ucell->atoms, ucell->latvec, + fn, "", 2, false, true, true, true, 0, ModuleBase::matrix(), false); + std::ifstream ifs; + ifs.open(fn); + std::string str((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + EXPECT_THAT(str, testing::HasSubstr("NUMERICAL_ORBITAL")); + EXPECT_THAT(str, testing::HasSubstr("__unittest_orbital_fn_0__")); + EXPECT_THAT(str, testing::HasSubstr("__unittest_orbital_fn_1__")); + EXPECT_THAT(str, testing::HasSubstr("NUMERICAL_DESCRIPTOR")); + EXPECT_THAT(str, testing::HasSubstr("__unittest_numerical_descriptor__")); + EXPECT_THAT(str, + testing::HasSubstr("0.9999996019 0.9999996019 0.9999996019 m 1 1 1 mag 0.5000")); + EXPECT_THAT(str, + testing::HasSubstr("1.4999994028 1.4999994028 1.4999994028 m 0 0 0 mag 0.4000")); + EXPECT_THAT(str, + testing::HasSubstr("0.4999998009 0.4999998009 0.4999998009 m 0 0 1 mag 0.3000")); + ifs.close(); + remove(fn.c_str()); +} + +TEST_F(PrintCellTest, PrintSTRU_nspin4_initial_mag) +{ + UcellTestPrepare utp = UcellTestLib["C1H2-Index"]; + ucell = utp.SetUcellInfo(); + std::string fn = "C1H2_STRU_nspin4"; + + ucell->atoms[0].m_loc_[0].set(1.0, 0.0, 0.0); // C + ucell->atoms[1].m_loc_[0].set(0.0, 1.0, 0.0); // H1 + ucell->atoms[1].m_loc_[1].set(0.0, 0.0, 1.0); // H2 + unitcell::print_stru_file(*ucell, ucell->atoms, ucell->latvec, + fn, "", 4, false, false, false, false, 0, ModuleBase::matrix(), false); + std::ifstream ifs; + ifs.open(fn); + std::string str((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + EXPECT_THAT(str, testing::HasSubstr("C #label")); + EXPECT_THAT(str, testing::HasSubstr("1.0000 #magnetism (default, overridden by per-atom mag below)")); + EXPECT_THAT(str, + testing::HasSubstr("0.9999996019 0.9999996019 0.9999996019 m 1 1 1 mag 1.0000 0.0000 0.0000")); + EXPECT_THAT(str, testing::HasSubstr("H #label")); + EXPECT_THAT(str, testing::HasSubstr("1.0000 #magnetism (default, overridden by per-atom mag below)")); + EXPECT_THAT(str, + testing::HasSubstr("1.4999994028 1.4999994028 1.4999994028 m 0 0 0 mag 0.0000 1.0000 0.0000")); + EXPECT_THAT(str, + testing::HasSubstr("0.4999998009 0.4999998009 0.4999998009 m 0 0 1 mag 0.0000 0.0000 1.0000")); + ifs.close(); + remove(fn.c_str()); +} + +TEST_F(PrintCellTest, PrintSTRU_nspin4_mulliken) +{ + UcellTestPrepare utp = UcellTestLib["C1H2-Index"]; + ucell = utp.SetUcellInfo(); + std::string fn = "C1H2_STRU_nspin4_mulliken"; + + ucell->atom_mulliken = {{-1, 0.5, 0.1, 0.2}, {-1, 0.4, 0.3, 0.4}, {-1, 0.3, 0.5, 0.6}}; + unitcell::print_stru_file(*ucell, ucell->atoms, ucell->latvec, + fn, "", 4, false, true, false, false, 0, ModuleBase::matrix(), false); + std::ifstream ifs; + ifs.open(fn); + std::string str((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + EXPECT_THAT(str, + testing::HasSubstr("0.9999996019 0.9999996019 0.9999996019 m 1 1 1 mag 0.5000 0.1000 0.2000")); + EXPECT_THAT(str, + testing::HasSubstr("1.4999994028 1.4999994028 1.4999994028 m 0 0 0 mag 0.4000 0.3000 0.4000")); + EXPECT_THAT(str, + testing::HasSubstr("0.4999998009 0.4999998009 0.4999998009 m 0 0 1 mag 0.3000 0.5000 0.6000")); + ifs.close(); + remove(fn.c_str()); +} + +TEST_F(PrintCellTest, PrintSTRU_with_force) +{ + UcellTestPrepare utp = UcellTestLib["C1H2-Index"]; + ucell = utp.SetUcellInfo(); + std::string fn = "C1H2_STRU_force"; + + // Create force matrix: nat=3, nc=3 (internal unit Ry/Bohr) + ModuleBase::matrix force(3, 3); + force(0, 0) = 0.1; force(0, 1) = 0.2; force(0, 2) = 0.3; // C + force(1, 0) = -0.1; force(1, 1) = -0.2; force(1, 2) = -0.3; // H1 + force(2, 0) = 0.05; force(2, 1) = 0.15; force(2, 2) = -0.25; // H2 + + unitcell::print_stru_file(*ucell, ucell->atoms, ucell->latvec, + fn, "", 2, false, false, false, false, 0, force, true); + std::ifstream ifs; + ifs.open(fn); + std::string str((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + + // Positions are always Cartesian_angstrom; forces add the f fields + EXPECT_THAT(str, testing::HasSubstr("Cartesian_angstrom # positions in Angstrom, forces in eV/Angstrom")); + + // Internal Cartesian tau is in units of lat0 (Bohr); expected Angstrom = tau * lat0 * BOHR_TO_A + const double pos_conv = ucell->lat0 * ModuleBase::BOHR_TO_A; + // Internal force is Ry/Bohr; expected eV/Angstrom = f * Ry_to_eV / BOHR_TO_A + const double force_conv = ModuleBase::Ry_to_eV / ModuleBase::BOHR_TO_A; + + auto fmt3 = [](double a, double b, double c) { + char buf[128]; + std::snprintf(buf, sizeof(buf), "%.10f %.10f %.10f", a, b, c); + return std::string(buf); + }; + auto fmtf = [](double a, double b, double c) { + char buf[128]; + std::snprintf(buf, sizeof(buf), " f %.6f %.6f %.6f", a, b, c); + return std::string(buf); + }; + + // C (iat=0): m 1 1 1 + EXPECT_THAT(str, testing::HasSubstr( + fmt3(ucell->atoms[0].tau[0].x * pos_conv, + ucell->atoms[0].tau[0].y * pos_conv, + ucell->atoms[0].tau[0].z * pos_conv) + " m 1 1 1")); + EXPECT_THAT(str, testing::HasSubstr( + fmtf(force(0, 0) * force_conv, force(0, 1) * force_conv, force(0, 2) * force_conv))); + + // H1 (iat=1): m 0 0 0 + EXPECT_THAT(str, testing::HasSubstr( + fmt3(ucell->atoms[1].tau[0].x * pos_conv, + ucell->atoms[1].tau[0].y * pos_conv, + ucell->atoms[1].tau[0].z * pos_conv) + " m 0 0 0")); + EXPECT_THAT(str, testing::HasSubstr( + fmtf(force(1, 0) * force_conv, force(1, 1) * force_conv, force(1, 2) * force_conv))); + + // H2 (iat=2): m 0 0 1 + EXPECT_THAT(str, testing::HasSubstr( + fmt3(ucell->atoms[1].tau[1].x * pos_conv, + ucell->atoms[1].tau[1].y * pos_conv, + ucell->atoms[1].tau[1].z * pos_conv) + " m 0 0 1")); + EXPECT_THAT(str, testing::HasSubstr( + fmtf(force(2, 0) * force_conv, force(2, 1) * force_conv, force(2, 2) * force_conv))); + + ifs.close(); + remove(fn.c_str()); +} + +TEST_F(PrintCellTest, PrintCell) +{ + UcellTestPrepare utp = UcellTestLib["C1H2-Index"]; + ucell = utp.SetUcellInfo(); + std::ofstream ofs; + ofs.open("print_cell_test.log"); + unitcell::print_cell(*ucell, ofs); + ofs.close(); + std::ifstream ifs; + ifs.open("print_cell_test.log"); + std::string str((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + EXPECT_THAT(str, testing::HasSubstr("latName = bcc")); + EXPECT_THAT(str, testing::HasSubstr("ntype = 2")); + EXPECT_THAT(str, testing::HasSubstr("nat = 3")); + EXPECT_THAT(str, testing::HasSubstr("GGT :")); + EXPECT_THAT(str, testing::HasSubstr("omega = 6748.33")); + remove("print_cell_test.log"); +} +TEST_F(PrintCellTest, PrintTauDirect) +{ + UcellTestPrepare utp = UcellTestLib["C1H2-Index"]; + ucell = utp.SetUcellInfo(); + EXPECT_EQ(ucell->Coordinate, "Direct"); + + // open a file + std::ofstream ofs("print_tau_direct"); + unitcell::print_tau(ucell->atoms,ucell->Coordinate,ucell->ntype,ucell->lat0,ofs); + ofs.close(); + + // readin the data + std::ifstream ifs; + ifs.open("print_tau_direct"); + std::string str((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + EXPECT_THAT(str, testing::HasSubstr("DIRECT COORDINATES")); + EXPECT_THAT(str, testing::HasSubstr(" C 0.100000000000 0.100000000000 0.100000000000 0.0000")); + EXPECT_THAT(str, testing::HasSubstr(" H 0.150000000000 0.150000000000 0.150000000000 0.0000")); + ifs.close(); + + remove("print_tau_direct"); +} + +TEST_F(PrintCellTest, PrintTauCartesian) +{ + UcellTestPrepare utp = UcellTestLib["C1H2-Cartesian"]; + ucell = utp.SetUcellInfo(); + EXPECT_EQ(ucell->Coordinate, "Cartesian"); + + // open a file + std::ofstream ofs("print_tau_Cartesian"); + unitcell::print_tau(ucell->atoms,ucell->Coordinate,ucell->ntype,ucell->lat0,ofs); + ofs.close(); + + // readin the data + std::ifstream ifs; + ifs.open("print_tau_Cartesian"); + std::string str((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + EXPECT_THAT(str, testing::HasSubstr("CARTESIAN COORDINATES")); + EXPECT_THAT(str, testing::HasSubstr(" C 1.000000000000 1.000000000000 1.000000000000 0.0000")); + EXPECT_THAT(str, testing::HasSubstr(" H 1.500000000000 1.500000000000 1.500000000000 0.0000")); + ifs.close(); + + // remove the file + remove("print_tau_Cartesian"); +} diff --git a/source/source_cell/test/test_read_atom_species.cpp b/source/source_cell/test/test_read_atom_species.cpp new file mode 100644 index 00000000000..104697b614d --- /dev/null +++ b/source/source_cell/test/test_read_atom_species.cpp @@ -0,0 +1,236 @@ +#include "gmock/gmock.h" +#include "gtest/gtest.h" + +#include +#include +#include +#include + +#include "source_base/global_variable.h" +#include "source_base/mathzone.h" +#include "source_cell/unitcell.h" +#include "source_cell/read_stru.h" + +// The test-only cell_info object library does not contain magnetism.cpp, +// so the Magnetism constructor/destructor must be provided locally. +Magnetism::Magnetism() +{ + this->tot_mag = 0.0; + this->abs_mag = 0.0; +} +Magnetism::~Magnetism() +{ +} + +#ifdef __LCAO +/************************************************ + * unit test of read_atom_species.cpp + ***********************************************/ + +/** + * - Tested Functions: + * - ReadAtomSpecies + * - read_atom_species(): a successful case + * - ReadAtomSpeciesWarning1 + * - read_atom_species(): unrecognized pseudopotential type + * - ReadAtomSpeciesLatName + * - read_lattice_constant(): all supported lattice names + * - ReadLatticeConstantWarning1 + * - read_lattice_constant(): lattice constant <= 0 + * - ReadLatticeConstantWarning2 + * - read_lattice_constant(): LATTICE_PARAMETERS without lattice type + * - ReadLatticeConstantWarning3 + * - read_lattice_constant(): LATTICE_VECTORS with explicit lattice type + * - ReadAtomSpeciesWarning5 + * - read_lattice_constant(): unsupported lattice name + */ + +class ReadAtomSpeciesTest : public ::testing::Test +{ + protected: + std::unique_ptr ucell{new UnitCell}; + std::string output; + void SetUp() override + { + ucell->ntype = 2; + ucell->pseudo_fn.resize(ucell->ntype); + ucell->pseudo_type.resize(ucell->ntype); + ucell->orbital_fn.resize(ucell->ntype); + } + void TearDown() override + { + ucell->orbital_fn.shrink_to_fit(); + } +}; + +using ReadAtomSpeciesDeathTest = ReadAtomSpeciesTest; + +TEST_F(ReadAtomSpeciesTest, ReadAtomSpecies) +{ + std::string fn = "./support/STRU_MgO"; + std::ifstream ifa(fn.c_str()); + std::ofstream ofs_running; + ofs_running.open("read_atom_species.tmp"); + ucell->ntype = 2; + ucell->atoms = new Atom[ucell->ntype]; + ucell->set_atom_flag = true; + const std::string basis_type = "lcao"; + const std::string orbital_dir = ""; + const std::string init_wfc = ""; + const double onsite_radius = 0.0; + const bool deepks_setorb = true; + const bool rpa = false; + EXPECT_NO_THROW(unitcell::read_atom_species(ifa, ofs_running, *ucell, + basis_type, orbital_dir, init_wfc, + onsite_radius, deepks_setorb, rpa)); + EXPECT_NO_THROW(unitcell::read_lattice_constant(ifa, ofs_running, ucell->lat)); + EXPECT_DOUBLE_EQ(ucell->latvec.e11, 4.27957); + EXPECT_DOUBLE_EQ(ucell->latvec.e22, 4.27957); + EXPECT_DOUBLE_EQ(ucell->latvec.e33, 4.27957); + ofs_running.close(); + ifa.close(); + remove("read_atom_species.tmp"); +} + +TEST_F(ReadAtomSpeciesTest, ReadAtomSpeciesWarning1) +{ + std::string fn = "./support/STRU_MgO_Warning1"; + std::ifstream ifa(fn.c_str()); + std::ofstream ofs_running; + ofs_running.open("read_atom_species.txt"); + ucell->ntype = 2; + ucell->atoms = new Atom[ucell->ntype]; + ucell->set_atom_flag = true; + const std::string basis_type = "lcao"; + const std::string orbital_dir = ""; + const std::string init_wfc = ""; + const double onsite_radius = 0.0; + const bool deepks_setorb = true; + const bool rpa = false; + testing::internal::CaptureStdout(); + EXPECT_EXIT(unitcell::read_atom_species(ifa, ofs_running, *ucell, + basis_type, orbital_dir, init_wfc, + onsite_radius, deepks_setorb, rpa), ::testing::ExitedWithCode(1), ""); + output = testing::internal::GetCapturedStdout(); + EXPECT_THAT(output, testing::HasSubstr("unrecognized pseudopotential type.")); + ofs_running.close(); + ifa.close(); + //remove("read_atom_species.txt"); +} + +TEST_F(ReadAtomSpeciesTest, ReadLatticeConstantWarning1) +{ + std::string fn = "./support/STRU_MgO_Warning2"; + std::ifstream ifa(fn.c_str()); + std::ofstream ofs_running; + ofs_running.open("read_atom_species1.tmp"); + ucell->ntype = 2; + ucell->atoms = new Atom[ucell->ntype]; + ucell->set_atom_flag = true; + testing::internal::CaptureStdout(); + EXPECT_EXIT(unitcell::read_lattice_constant(ifa, ofs_running,ucell->lat), ::testing::ExitedWithCode(1), ""); + output = testing::internal::GetCapturedStdout(); + EXPECT_THAT(output, testing::HasSubstr("Lattice constant <= 0.0")); + ofs_running.close(); + ifa.close(); + remove("read_atom_species1.tmp"); +} + +TEST_F(ReadAtomSpeciesTest, ReadLatticeConstantWarning2) +{ + std::string fn = "./support/STRU_MgO_Warning3"; + std::ifstream ifa(fn.c_str()); + std::ofstream ofs_running; + ofs_running.open("read_atom_species.tmp"); + ucell->ntype = 2; + ucell->atoms = new Atom[ucell->ntype]; + ucell->set_atom_flag = true; + testing::internal::CaptureStdout(); + EXPECT_EXIT(unitcell::read_lattice_constant(ifa, ofs_running,ucell->lat), ::testing::ExitedWithCode(1), ""); + output = testing::internal::GetCapturedStdout(); + EXPECT_THAT(output, + testing::HasSubstr("do not use LATTICE_PARAMETERS without explicit specification of lattice type")); + ofs_running.close(); + ifa.close(); + remove("read_atom_species.tmp"); +} + +TEST_F(ReadAtomSpeciesTest, ReadLatticeConstantWarning3) +{ + std::string fn = "./support/STRU_MgO_Warning4"; + std::ifstream ifa(fn.c_str()); + std::ofstream ofs_running; + ofs_running.open("read_atom_species.tmp"); + ucell->ntype = 2; + ucell->atoms = new Atom[ucell->ntype]; + ucell->set_atom_flag = true; + ucell->latName = "bcc"; + testing::internal::CaptureStdout(); + EXPECT_EXIT(unitcell::read_lattice_constant(ifa, ofs_running,ucell->lat), ::testing::ExitedWithCode(1), ""); + output = testing::internal::GetCapturedStdout(); + EXPECT_THAT(output, + testing::HasSubstr("do not use LATTICE_VECTORS along with explicit specification of lattice type")); + ofs_running.close(); + ifa.close(); + remove("read_atom_species.tmp"); +} + +TEST_F(ReadAtomSpeciesTest, ReadAtomSpeciesLatName) +{ + ucell->ntype = 2; + ucell->atoms = new Atom[ucell->ntype]; + ucell->set_atom_flag = true; + std::vector latName_in = {"sc", + "fcc", + "bcc", + "hexagonal", + "trigonal", + "st", + "bct", + "so", + "baco", + "fco", + "bco", + "sm", + "bacm", + "triclinic"}; + for (int i = 0; i < latName_in.size(); ++i) + { + std::string fn = "./support/STRU_MgO_LatName"; + std::ifstream ifa(fn.c_str()); + std::ofstream ofs_running; + ofs_running.open("read_atom_species.tmp"); + ucell->latName = latName_in[i]; + EXPECT_NO_THROW(unitcell::read_lattice_constant(ifa, ofs_running,ucell->lat)); + if (ucell->latName == "sc") + { + EXPECT_DOUBLE_EQ(ucell->latvec.e11, 1.0); + EXPECT_DOUBLE_EQ(ucell->latvec.e22, 1.0); + EXPECT_DOUBLE_EQ(ucell->latvec.e33, 1.0); + } + ofs_running.close(); + ifa.close(); + remove("read_atom_species.tmp"); + } +} + +TEST_F(ReadAtomSpeciesDeathTest, ReadAtomSpeciesWarning5) +{ + std::string fn = "./support/STRU_MgO_LatName"; + std::ifstream ifa(fn.c_str()); + std::ofstream ofs_running; + ofs_running.open("read_atom_species.tmp"); + ucell->ntype = 2; + ucell->atoms = new Atom[ucell->ntype]; + ucell->set_atom_flag = true; + ucell->latName = "arbitrary"; + testing::internal::CaptureStdout(); + EXPECT_EXIT(unitcell::read_lattice_constant(ifa, ofs_running,ucell->lat), ::testing::ExitedWithCode(1), ""); + output = testing::internal::GetCapturedStdout(); + EXPECT_THAT(output, testing::HasSubstr("latname not supported")); + ofs_running.close(); + ifa.close(); + remove("read_atom_species.tmp"); +} + +#endif // __LCAO diff --git a/source/source_cell/test/test_read_atoms.cpp b/source/source_cell/test/test_read_atoms.cpp new file mode 100644 index 00000000000..d70c267e8ab --- /dev/null +++ b/source/source_cell/test/test_read_atoms.cpp @@ -0,0 +1,860 @@ +#include "gmock/gmock.h" +#include "gtest/gtest.h" + +#include +#include +#include +#include + +#include "source_base/global_variable.h" +#include "source_base/mathzone.h" +#include "source_cell/unitcell.h" +#include "source_cell/read_stru.h" + +// The test-only cell_info object library does not contain magnetism.cpp, +// so the Magnetism constructor/destructor must be provided locally. +Magnetism::Magnetism() +{ + this->tot_mag = 0.0; + this->abs_mag = 0.0; +} +Magnetism::~Magnetism() +{ +} + +#ifdef __LCAO +/************************************************ + * unit test of read_atoms.cpp + ***********************************************/ + +/** + * - Tested Functions: + * - ReadAtomPositionsS1 + * - read_atom_positions(): spin 1 case + * - ReadAtomPositionsS2 + * - read_atom_positions(): spin 2 case + * - ReadAtomPositionsS4Noncolin + * - read_atom_positions(): spin 4 noncolinear case + * - ReadAtomPositionsS4Colin + * - read_atom_positions(): spin 4 colinear case + * - ReadAtomPositionsC + * - read_atom_positions(): Cartesian coordinates + * - ReadAtomPositionsCA + * - read_atom_positions(): Cartesian_angstrom coordinates + * - ReadAtomPositionsCACXY + * - read_atom_positions(): Cartesian_angstrom_center_xy coordinates + * - ReadAtomPositionsCACXZ + * - read_atom_positions(): Cartesian_angstrom_center_xz coordinates + * - ReadAtomPositionsCACYZ + * - read_atom_positions(): Cartesian_angstrom_center_yz coordinates + * - ReadAtomPositionsCACXYZ + * - read_atom_positions(): Cartesian_angstrom_center_xyz coordinates + * - ReadAtomPositionsCAU + * - read_atom_positions(): Cartesian_au coordinates + * - ReadAtomPositionsAutosetMag + * - read_atom_positions(): zero-moment start with warning for nspin=4 + * - ReadAtomPositionsWarning1 + * - read_atom_positions(): unknown type of coordinates + * - ReadAtomPositionsWarning2 + * - read_atom_positions(): atomic label inconsistency between ATOM_POSITIONS + * and ATOM_SPECIES + * - ReadAtomPositionsWarning3 + * - read_atom_positions(): warning : atom number < 0 + * - ReadAtomPositionsWarning4 + * - read_atom_positions(): mismatch in atom number for atom type + * - ReadAtomPositionsWarning5 + * - read_atom_positions(): no atoms can move in MD simulations! + */ + +class ReadAtomsTest : public ::testing::Test +{ + protected: + std::unique_ptr ucell{new UnitCell}; + std::string output; + void SetUp() override + { + ucell->ntype = 2; + ucell->pseudo_fn.resize(ucell->ntype); + ucell->pseudo_type.resize(ucell->ntype); + ucell->orbital_fn.resize(ucell->ntype); + } + void TearDown() override + { + ucell->orbital_fn.shrink_to_fit(); + } +}; + +TEST_F(ReadAtomsTest, ReadAtomPositionsS1) +{ + std::string fn = "./support/STRU_MgO"; + std::ifstream ifa(fn.c_str()); + std::ofstream ofs_running; + std::ofstream ofs_warning; + ofs_running.open("read_atom_positions.tmp"); + ofs_warning.open("read_atom_positions.warn"); + // mandatory preliminaries + ucell->ntype = 2; + ucell->atoms = new Atom[ucell->ntype]; + ucell->set_atom_flag = true; + const std::string basis_type = "lcao"; + const std::string orbital_dir = ""; + const std::string init_wfc = ""; + const double onsite_radius = 0.0; + const bool deepks_setorb = true; + const bool rpa = false; + const int nspin = 1; + const bool fixed_atoms = false; + const bool noncolin = false; + const std::string calculation = "scf"; + const std::string esolver_type = "ksdft"; + EXPECT_NO_THROW(unitcell::read_atom_species(ifa, ofs_running, *ucell, + basis_type, orbital_dir, init_wfc, + onsite_radius, deepks_setorb, rpa)); + EXPECT_NO_THROW(unitcell::read_lattice_constant(ifa, ofs_running,ucell->lat)); + EXPECT_DOUBLE_EQ(ucell->latvec.e11, 4.27957); + EXPECT_DOUBLE_EQ(ucell->latvec.e22, 4.27957); + EXPECT_DOUBLE_EQ(ucell->latvec.e33, 4.27957); + unitcell::read_atom_positions(*ucell, ifa, ofs_running, ofs_warning, nspin, + basis_type, orbital_dir, init_wfc, + onsite_radius, fixed_atoms, noncolin, + calculation, esolver_type, 0); + ofs_running.close(); + ofs_warning.close(); + ifa.close(); + remove("read_atom_positions.tmp"); + remove("read_atom_positions.warn"); +} + +TEST_F(ReadAtomsTest, ReadAtomPositionsS2) +{ + std::string fn = "./support/STRU_MgO"; + std::ifstream ifa(fn.c_str()); + std::ofstream ofs_running; + std::ofstream ofs_warning; + ofs_running.open("read_atom_positions.tmp"); + ofs_warning.open("read_atom_positions.warn"); + // mandatory preliminaries + ucell->ntype = 2; + ucell->atoms = new Atom[ucell->ntype]; + ucell->set_atom_flag = true; + const std::string basis_type = "lcao"; + const std::string orbital_dir = ""; + const std::string init_wfc = ""; + const double onsite_radius = 0.0; + const bool deepks_setorb = true; + const bool rpa = false; + const int nspin = 2; + const bool fixed_atoms = false; + const bool noncolin = false; + const std::string calculation = "scf"; + const std::string esolver_type = "ksdft"; + EXPECT_NO_THROW(unitcell::read_atom_species(ifa, ofs_running, *ucell, + basis_type, orbital_dir, init_wfc, + onsite_radius, deepks_setorb, rpa)); + EXPECT_NO_THROW(unitcell::read_lattice_constant(ifa, ofs_running,ucell->lat)); + EXPECT_DOUBLE_EQ(ucell->latvec.e11, 4.27957); + EXPECT_DOUBLE_EQ(ucell->latvec.e22, 4.27957); + EXPECT_DOUBLE_EQ(ucell->latvec.e33, 4.27957); + unitcell::read_atom_positions(*ucell, ifa, ofs_running, ofs_warning, nspin, + basis_type, orbital_dir, init_wfc, + onsite_radius, fixed_atoms, noncolin, + calculation, esolver_type, 0); + ofs_running.close(); + ofs_warning.close(); + ifa.close(); + remove("read_atom_positions.tmp"); + remove("read_atom_positions.warn"); +} + +TEST_F(ReadAtomsTest, ReadAtomPositionsS4Noncolin) +{ + std::string fn = "./support/STRU_MgO"; + std::ifstream ifa(fn.c_str()); + std::ofstream ofs_running; + std::ofstream ofs_warning; + ofs_running.open("read_atom_positions.tmp"); + ofs_warning.open("read_atom_positions.warn"); + // mandatory preliminaries + ucell->ntype = 2; + ucell->atoms = new Atom[ucell->ntype]; + ucell->set_atom_flag = true; + const std::string basis_type = "lcao"; + const std::string orbital_dir = ""; + const std::string init_wfc = ""; + const double onsite_radius = 0.0; + const bool deepks_setorb = true; + const bool rpa = false; + const int nspin = 4; + const bool fixed_atoms = false; + const bool noncolin = true; + const std::string calculation = "scf"; + const std::string esolver_type = "ksdft"; + EXPECT_NO_THROW(unitcell::read_atom_species(ifa, ofs_running, *ucell, + basis_type, orbital_dir, init_wfc, + onsite_radius, deepks_setorb, rpa)); + EXPECT_NO_THROW(unitcell::read_lattice_constant(ifa, ofs_running,ucell->lat)); + EXPECT_DOUBLE_EQ(ucell->latvec.e11, 4.27957); + EXPECT_DOUBLE_EQ(ucell->latvec.e22, 4.27957); + EXPECT_DOUBLE_EQ(ucell->latvec.e33, 4.27957); + unitcell::read_atom_positions(*ucell, ifa, ofs_running, ofs_warning, nspin, + basis_type, orbital_dir, init_wfc, + onsite_radius, fixed_atoms, noncolin, + calculation, esolver_type, 0); + ofs_running.close(); + ofs_warning.close(); + ifa.close(); + remove("read_atom_positions.tmp"); + remove("read_atom_positions.warn"); +} + +TEST_F(ReadAtomsTest, ReadAtomPositionsS4Colin) +{ + std::string fn = "./support/STRU_MgO"; + std::ifstream ifa(fn.c_str()); + std::ofstream ofs_running; + std::ofstream ofs_warning; + ofs_running.open("read_atom_positions.tmp"); + ofs_warning.open("read_atom_positions.warn"); + // mandatory preliminaries + ucell->ntype = 2; + ucell->atoms = new Atom[ucell->ntype]; + ucell->set_atom_flag = true; + const std::string basis_type = "lcao"; + const std::string orbital_dir = ""; + const std::string init_wfc = ""; + const double onsite_radius = 0.0; + const bool deepks_setorb = true; + const bool rpa = false; + const int nspin = 4; + const bool fixed_atoms = false; + const bool noncolin = false; + const std::string calculation = "scf"; + const std::string esolver_type = "ksdft"; + EXPECT_NO_THROW(unitcell::read_atom_species(ifa, ofs_running, *ucell, + basis_type, orbital_dir, init_wfc, + onsite_radius, deepks_setorb, rpa)); + EXPECT_NO_THROW(unitcell::read_lattice_constant(ifa, ofs_running,ucell->lat)); + EXPECT_DOUBLE_EQ(ucell->latvec.e11, 4.27957); + EXPECT_DOUBLE_EQ(ucell->latvec.e22, 4.27957); + EXPECT_DOUBLE_EQ(ucell->latvec.e33, 4.27957); + unitcell::read_atom_positions(*ucell, ifa, ofs_running, ofs_warning, nspin, + basis_type, orbital_dir, init_wfc, + onsite_radius, fixed_atoms, noncolin, + calculation, esolver_type, 0); + ofs_running.close(); + ofs_warning.close(); + ifa.close(); + remove("read_atom_positions.tmp"); + remove("read_atom_positions.warn"); +} + +TEST_F(ReadAtomsTest, ReadAtomPositionsC) +{ + std::string fn = "./support/STRU_MgO_c"; + std::ifstream ifa(fn.c_str()); + std::ofstream ofs_running; + std::ofstream ofs_warning; + ofs_running.open("read_atom_positions.tmp"); + ofs_warning.open("read_atom_positions.warn"); + // mandatory preliminaries + ucell->ntype = 2; + ucell->atoms = new Atom[ucell->ntype]; + ucell->set_atom_flag = true; + const std::string basis_type = "lcao"; + const std::string orbital_dir = ""; + const std::string init_wfc = ""; + const double onsite_radius = 0.0; + const bool deepks_setorb = true; + const bool rpa = false; + const int nspin = 1; + const bool fixed_atoms = false; + const bool noncolin = false; + const std::string calculation = "scf"; + const std::string esolver_type = "ksdft"; + EXPECT_NO_THROW(unitcell::read_atom_species(ifa, ofs_running, *ucell, + basis_type, orbital_dir, init_wfc, + onsite_radius, deepks_setorb, rpa)); + EXPECT_NO_THROW(unitcell::read_lattice_constant(ifa, ofs_running,ucell->lat)); + EXPECT_DOUBLE_EQ(ucell->latvec.e11, 4.27957); + EXPECT_DOUBLE_EQ(ucell->latvec.e22, 4.27957); + EXPECT_DOUBLE_EQ(ucell->latvec.e33, 4.27957); + unitcell::read_atom_positions(*ucell, ifa, ofs_running, ofs_warning, nspin, + basis_type, orbital_dir, init_wfc, + onsite_radius, fixed_atoms, noncolin, + calculation, esolver_type, 0); + ofs_running.close(); + ofs_warning.close(); + ifa.close(); + remove("read_atom_positions.tmp"); + remove("read_atom_positions.warn"); +} + +TEST_F(ReadAtomsTest, ReadAtomPositionsCA) +{ + std::string fn = "./support/STRU_MgO_ca"; + std::ifstream ifa(fn.c_str()); + std::ofstream ofs_running; + std::ofstream ofs_warning; + ofs_running.open("read_atom_positions.tmp"); + ofs_warning.open("read_atom_positions.warn"); + // mandatory preliminaries + ucell->ntype = 2; + ucell->atoms = new Atom[ucell->ntype]; + ucell->set_atom_flag = true; + const std::string basis_type = "lcao"; + const std::string orbital_dir = ""; + const std::string init_wfc = ""; + const double onsite_radius = 0.0; + const bool deepks_setorb = true; + const bool rpa = false; + const int nspin = 1; + const bool fixed_atoms = false; + const bool noncolin = false; + const std::string calculation = "scf"; + const std::string esolver_type = "ksdft"; + EXPECT_NO_THROW(unitcell::read_atom_species(ifa, ofs_running, *ucell, + basis_type, orbital_dir, init_wfc, + onsite_radius, deepks_setorb, rpa)); + EXPECT_NO_THROW(unitcell::read_lattice_constant(ifa, ofs_running,ucell->lat)); + EXPECT_DOUBLE_EQ(ucell->latvec.e11, 4.27957); + EXPECT_DOUBLE_EQ(ucell->latvec.e22, 4.27957); + EXPECT_DOUBLE_EQ(ucell->latvec.e33, 4.27957); + unitcell::read_atom_positions(*ucell, ifa, ofs_running, ofs_warning, nspin, + basis_type, orbital_dir, init_wfc, + onsite_radius, fixed_atoms, noncolin, + calculation, esolver_type, 0); + ofs_running.close(); + ofs_warning.close(); + ifa.close(); + remove("read_atom_positions.tmp"); + remove("read_atom_positions.warn"); +} + +TEST_F(ReadAtomsTest, ReadAtomPositionsCACXY) +{ + std::string fn = "./support/STRU_MgO_cacxy"; + std::ifstream ifa(fn.c_str()); + std::ofstream ofs_running; + std::ofstream ofs_warning; + ofs_running.open("read_atom_positions.tmp"); + ofs_warning.open("read_atom_positions.warn"); + // mandatory preliminaries + ucell->ntype = 2; + ucell->atoms = new Atom[ucell->ntype]; + ucell->set_atom_flag = true; + const std::string basis_type = "lcao"; + const std::string orbital_dir = ""; + const std::string init_wfc = ""; + const double onsite_radius = 0.0; + const bool deepks_setorb = true; + const bool rpa = false; + const int nspin = 1; + const bool fixed_atoms = false; + const bool noncolin = false; + const std::string calculation = "scf"; + const std::string esolver_type = "ksdft"; + EXPECT_NO_THROW(unitcell::read_atom_species(ifa, ofs_running, *ucell, + basis_type, orbital_dir, init_wfc, + onsite_radius, deepks_setorb, rpa)); + EXPECT_NO_THROW(unitcell::read_lattice_constant(ifa, ofs_running,ucell->lat)); + EXPECT_DOUBLE_EQ(ucell->latvec.e11, 4.27957); + EXPECT_DOUBLE_EQ(ucell->latvec.e22, 4.27957); + EXPECT_DOUBLE_EQ(ucell->latvec.e33, 4.27957); + unitcell::read_atom_positions(*ucell, ifa, ofs_running, ofs_warning, nspin, + basis_type, orbital_dir, init_wfc, + onsite_radius, fixed_atoms, noncolin, + calculation, esolver_type, 0); + ofs_running.close(); + ofs_warning.close(); + ifa.close(); + remove("read_atom_positions.tmp"); + remove("read_atom_positions.warn"); +} + +TEST_F(ReadAtomsTest, ReadAtomPositionsCACXZ) +{ + std::string fn = "./support/STRU_MgO_cacxz"; + std::ifstream ifa(fn.c_str()); + std::ofstream ofs_running; + std::ofstream ofs_warning; + ofs_running.open("read_atom_positions.tmp"); + ofs_warning.open("read_atom_positions.warn"); + // mandatory preliminaries + ucell->ntype = 2; + ucell->atoms = new Atom[ucell->ntype]; + ucell->set_atom_flag = true; + const std::string basis_type = "lcao"; + const std::string orbital_dir = ""; + const std::string init_wfc = ""; + const double onsite_radius = 0.0; + const bool deepks_setorb = true; + const bool rpa = false; + const int nspin = 1; + const bool fixed_atoms = false; + const bool noncolin = false; + const std::string calculation = "scf"; + const std::string esolver_type = "ksdft"; + EXPECT_NO_THROW(unitcell::read_atom_species(ifa, ofs_running, *ucell, + basis_type, orbital_dir, init_wfc, + onsite_radius, deepks_setorb, rpa)); + EXPECT_NO_THROW(unitcell::read_lattice_constant(ifa, ofs_running,ucell->lat)); + EXPECT_DOUBLE_EQ(ucell->latvec.e11, 4.27957); + EXPECT_DOUBLE_EQ(ucell->latvec.e22, 4.27957); + EXPECT_DOUBLE_EQ(ucell->latvec.e33, 4.27957); + unitcell::read_atom_positions(*ucell, ifa, ofs_running, ofs_warning, nspin, + basis_type, orbital_dir, init_wfc, + onsite_radius, fixed_atoms, noncolin, + calculation, esolver_type, 0); + ofs_running.close(); + ofs_warning.close(); + ifa.close(); + remove("read_atom_positions.tmp"); + remove("read_atom_positions.warn"); +} + +TEST_F(ReadAtomsTest, ReadAtomPositionsCACYZ) +{ + std::string fn = "./support/STRU_MgO_cacyz"; + std::ifstream ifa(fn.c_str()); + std::ofstream ofs_running; + std::ofstream ofs_warning; + ofs_running.open("read_atom_positions.tmp"); + ofs_warning.open("read_atom_positions.warn"); + // mandatory preliminaries + ucell->ntype = 2; + ucell->atoms = new Atom[ucell->ntype]; + ucell->set_atom_flag = true; + const std::string basis_type = "lcao"; + const std::string orbital_dir = ""; + const std::string init_wfc = ""; + const double onsite_radius = 0.0; + const bool deepks_setorb = true; + const bool rpa = false; + const int nspin = 1; + const bool fixed_atoms = false; + const bool noncolin = false; + const std::string calculation = "scf"; + const std::string esolver_type = "ksdft"; + EXPECT_NO_THROW(unitcell::read_atom_species(ifa, ofs_running, *ucell, + basis_type, orbital_dir, init_wfc, + onsite_radius, deepks_setorb, rpa)); + EXPECT_NO_THROW(unitcell::read_lattice_constant(ifa, ofs_running,ucell->lat)); + EXPECT_DOUBLE_EQ(ucell->latvec.e11, 4.27957); + EXPECT_DOUBLE_EQ(ucell->latvec.e22, 4.27957); + EXPECT_DOUBLE_EQ(ucell->latvec.e33, 4.27957); + unitcell::read_atom_positions(*ucell, ifa, ofs_running, ofs_warning, nspin, + basis_type, orbital_dir, init_wfc, + onsite_radius, fixed_atoms, noncolin, + calculation, esolver_type, 0); + ofs_running.close(); + ofs_warning.close(); + ifa.close(); + remove("read_atom_positions.tmp"); + remove("read_atom_positions.warn"); +} + +TEST_F(ReadAtomsTest, ReadAtomPositionsCACXYZ) +{ + std::string fn = "./support/STRU_MgO_cacxyz"; + std::ifstream ifa(fn.c_str()); + std::ofstream ofs_running; + std::ofstream ofs_warning; + ofs_running.open("read_atom_positions.tmp"); + ofs_warning.open("read_atom_positions.warn"); + // mandatory preliminaries + ucell->ntype = 2; + ucell->atoms = new Atom[ucell->ntype]; + ucell->set_atom_flag = true; + const std::string basis_type = "lcao"; + const std::string orbital_dir = ""; + const std::string init_wfc = ""; + const double onsite_radius = 0.0; + const bool deepks_setorb = true; + const bool rpa = false; + const int nspin = 1; + const bool fixed_atoms = false; + const bool noncolin = false; + const std::string calculation = "scf"; + const std::string esolver_type = "ksdft"; + EXPECT_NO_THROW(unitcell::read_atom_species(ifa, ofs_running, *ucell, + basis_type, orbital_dir, init_wfc, + onsite_radius, deepks_setorb, rpa)); + EXPECT_NO_THROW(unitcell::read_lattice_constant(ifa, ofs_running,ucell->lat)); + EXPECT_DOUBLE_EQ(ucell->latvec.e11, 4.27957); + EXPECT_DOUBLE_EQ(ucell->latvec.e22, 4.27957); + EXPECT_DOUBLE_EQ(ucell->latvec.e33, 4.27957); + unitcell::read_atom_positions(*ucell, ifa, ofs_running, ofs_warning, nspin, + basis_type, orbital_dir, init_wfc, + onsite_radius, fixed_atoms, noncolin, + calculation, esolver_type, 0); + ofs_running.close(); + ofs_warning.close(); + ifa.close(); + remove("read_atom_positions.tmp"); + remove("read_atom_positions.warn"); +} + +TEST_F(ReadAtomsTest, ReadAtomPositionsCAU) +{ + std::string fn = "./support/STRU_MgO_cau"; + std::ifstream ifa(fn.c_str()); + std::ofstream ofs_running; + std::ofstream ofs_warning; + ofs_running.open("read_atom_positions.tmp"); + ofs_warning.open("read_atom_positions.warn"); + // mandatory preliminaries + ucell->ntype = 2; + ucell->atoms = new Atom[ucell->ntype]; + ucell->set_atom_flag = true; + const std::string basis_type = "lcao"; + const std::string orbital_dir = ""; + const std::string init_wfc = ""; + const double onsite_radius = 0.0; + const bool deepks_setorb = true; + const bool rpa = false; + const int nspin = 1; + const bool fixed_atoms = true; + const bool noncolin = false; + const std::string calculation = "scf"; + const std::string esolver_type = "ksdft"; + EXPECT_NO_THROW(unitcell::read_atom_species(ifa, ofs_running, *ucell, + basis_type, orbital_dir, init_wfc, + onsite_radius, deepks_setorb, rpa)); + EXPECT_NO_THROW(unitcell::read_lattice_constant(ifa, ofs_running,ucell->lat)); + EXPECT_DOUBLE_EQ(ucell->latvec.e11, 4.27957); + EXPECT_DOUBLE_EQ(ucell->latvec.e22, 4.27957); + EXPECT_DOUBLE_EQ(ucell->latvec.e33, 4.27957); + unitcell::read_atom_positions(*ucell, ifa, ofs_running, ofs_warning, nspin, + basis_type, orbital_dir, init_wfc, + onsite_radius, fixed_atoms, noncolin, + calculation, esolver_type, 0); + ofs_running.close(); + ofs_warning.close(); + ifa.close(); + remove("read_atom_positions.tmp"); + remove("read_atom_positions.warn"); +} + +TEST_F(ReadAtomsTest, ReadAtomPositionsAutosetMag) +{ + std::string fn = "./support/STRU_MgO"; + std::ifstream ifa(fn.c_str()); + std::ofstream ofs_running; + std::ofstream ofs_warning; + ofs_running.open("read_atom_positions.tmp"); + ofs_warning.open("read_atom_positions.warn"); + // mandatory preliminaries + ucell->ntype = 2; + ucell->atoms = new Atom[ucell->ntype]; + ucell->set_atom_flag = true; + const std::string basis_type = "lcao"; + const std::string orbital_dir = ""; + const std::string init_wfc = ""; + const double onsite_radius = 0.0; + const bool deepks_setorb = true; + const bool rpa = false; + const bool fixed_atoms = false; + const bool noncolin = false; + const std::string calculation = "scf"; + const std::string esolver_type = "ksdft"; + int nspin = 2; + EXPECT_NO_THROW(unitcell::read_atom_species(ifa, ofs_running, *ucell, + basis_type, orbital_dir, init_wfc, + onsite_radius, deepks_setorb, rpa)); + EXPECT_NO_THROW(unitcell::read_lattice_constant(ifa, ofs_running,ucell->lat)); + EXPECT_DOUBLE_EQ(ucell->latvec.e11, 4.27957); + EXPECT_DOUBLE_EQ(ucell->latvec.e22, 4.27957); + EXPECT_DOUBLE_EQ(ucell->latvec.e33, 4.27957); + unitcell::read_atom_positions(*ucell, ifa, ofs_running, ofs_warning, nspin, + basis_type, orbital_dir, init_wfc, + onsite_radius, fixed_atoms, noncolin, + calculation, esolver_type, 0); + for (int it = 0; it < ucell->ntype; it++) + { + for (int ia = 0; ia < ucell->atoms[it].na; ia++) + { + EXPECT_DOUBLE_EQ(ucell->atoms[it].mag[ia], 1.0); + EXPECT_DOUBLE_EQ(ucell->atoms[it].m_loc_[ia].z, 1.0); + } + } + // for nspin == 4 + // Issue #5939: nspin=4 with no mag in STRU no longer autosets (1,1,1); + // all moments stay zero and a warning is emitted instead. + nspin = 4; + testing::internal::CaptureStdout(); + unitcell::read_atom_positions(*ucell, ifa, ofs_running, ofs_warning, nspin, + basis_type, orbital_dir, init_wfc, + onsite_radius, fixed_atoms, noncolin, + calculation, esolver_type, 0); + const std::string stdout_output = testing::internal::GetCapturedStdout(); + for (int it = 0; it < ucell->ntype; it++) + { + for (int ia = 0; ia < ucell->atoms[it].na; ia++) + { + EXPECT_DOUBLE_EQ(ucell->atoms[it].mag[ia], 0.0); + EXPECT_DOUBLE_EQ(ucell->atoms[it].m_loc_[ia].x, 0.0); + EXPECT_DOUBLE_EQ(ucell->atoms[it].m_loc_[ia].y, 0.0); + EXPECT_DOUBLE_EQ(ucell->atoms[it].m_loc_[ia].z, 0.0); + } + } + // The zero-moment warning must reach both stdout and the running log. + EXPECT_NE(stdout_output.find("no initial magnetization is set in STRU"), + std::string::npos); + ofs_running.flush(); + std::ifstream ifs_log("read_atom_positions.tmp"); + std::string log_content; + std::string log_line; + while (std::getline(ifs_log, log_line)) + { + log_content += log_line; + } + ifs_log.close(); + EXPECT_NE(log_content.find("no initial magnetization is set in STRU"), + std::string::npos); + ofs_running.close(); + ofs_warning.close(); + ifa.close(); + remove("read_atom_positions.tmp"); + remove("read_atom_positions.warn"); +} + +TEST_F(ReadAtomsTest, ReadAtomPositionsWarning1) +{ + std::string fn = "./support/STRU_MgO_WarningC1"; + std::ifstream ifa(fn.c_str()); + std::ofstream ofs_running; + std::ofstream ofs_warning; + ofs_running.open("read_atom_positions.tmp"); + ofs_warning.open("read_atom_positions.warn"); + // mandatory preliminaries + ucell->ntype = 2; + ucell->atoms = new Atom[ucell->ntype]; + ucell->set_atom_flag = true; + const std::string basis_type = "lcao"; + const std::string orbital_dir = ""; + const std::string init_wfc = ""; + const double onsite_radius = 0.0; + const bool deepks_setorb = true; + const bool rpa = false; + const int nspin = 1; + const bool fixed_atoms = false; + const bool noncolin = false; + const std::string calculation = "scf"; + const std::string esolver_type = "ksdft"; + EXPECT_NO_THROW(unitcell::read_atom_species(ifa, ofs_running, *ucell, + basis_type, orbital_dir, init_wfc, + onsite_radius, deepks_setorb, rpa)); + EXPECT_NO_THROW(unitcell::read_lattice_constant(ifa, ofs_running,ucell->lat)); + EXPECT_DOUBLE_EQ(ucell->latvec.e11, 4.27957); + EXPECT_DOUBLE_EQ(ucell->latvec.e22, 4.27957); + EXPECT_DOUBLE_EQ(ucell->latvec.e33, 4.27957); + EXPECT_NO_THROW(unitcell::read_atom_positions(*ucell, ifa, ofs_running, ofs_warning, nspin, + basis_type, orbital_dir, init_wfc, + onsite_radius, fixed_atoms, noncolin, + calculation, esolver_type, 0)); + ofs_running.close(); + ofs_warning.close(); + ifa.close(); + // check warning file + std::ifstream ifs_tmp; + ifs_tmp.open("read_atom_positions.warn"); + std::string str((std::istreambuf_iterator(ifs_tmp)), std::istreambuf_iterator()); + EXPECT_THAT(str, testing::HasSubstr("There are several options for you:")); + EXPECT_THAT(str, testing::HasSubstr("Direct")); + EXPECT_THAT(str, testing::HasSubstr("Cartesian_angstrom")); + EXPECT_THAT(str, testing::HasSubstr("Cartesian_au")); + EXPECT_THAT(str, testing::HasSubstr("Cartesian_angstrom_center_xy")); + EXPECT_THAT(str, testing::HasSubstr("Cartesian_angstrom_center_xz")); + EXPECT_THAT(str, testing::HasSubstr("Cartesian_angstrom_center_yz")); + EXPECT_THAT(str, testing::HasSubstr("Cartesian_angstrom_center_xyz")); + ifs_tmp.close(); + remove("read_atom_positions.tmp"); + remove("read_atom_positions.warn"); +} + +TEST_F(ReadAtomsTest, ReadAtomPositionsWarning2) +{ + std::string fn = "./support/STRU_MgO_WarningC2"; + std::ifstream ifa(fn.c_str()); + std::ofstream ofs_running; + std::ofstream ofs_warning; + ofs_running.open("read_atom_positions.tmp"); + ofs_warning.open("read_atom_positions.warn"); + // mandatory preliminaries + ucell->ntype = 2; + ucell->atoms = new Atom[ucell->ntype]; + ucell->set_atom_flag = true; + const std::string basis_type = "lcao"; + const std::string orbital_dir = ""; + const std::string init_wfc = ""; + const double onsite_radius = 0.0; + const bool deepks_setorb = true; + const bool rpa = false; + const int nspin = 1; + const bool fixed_atoms = false; + const bool noncolin = false; + const std::string calculation = "scf"; + const std::string esolver_type = "ksdft"; + EXPECT_NO_THROW(unitcell::read_atom_species(ifa, ofs_running, *ucell, + basis_type, orbital_dir, init_wfc, + onsite_radius, deepks_setorb, rpa)); + EXPECT_NO_THROW(unitcell::read_lattice_constant(ifa, ofs_running,ucell->lat)); + EXPECT_DOUBLE_EQ(ucell->latvec.e11, 4.27957); + EXPECT_DOUBLE_EQ(ucell->latvec.e22, 4.27957); + EXPECT_DOUBLE_EQ(ucell->latvec.e33, 4.27957); + EXPECT_NO_THROW(unitcell::read_atom_positions(*ucell, ifa, ofs_running, ofs_warning, nspin, + basis_type, orbital_dir, init_wfc, + onsite_radius, fixed_atoms, noncolin, + calculation, esolver_type, 0)); + ofs_running.close(); + ofs_warning.close(); + ifa.close(); + // check warning file + std::ifstream ifs_tmp; + ifs_tmp.open("read_atom_positions.warn"); + std::string str((std::istreambuf_iterator(ifs_tmp)), std::istreambuf_iterator()); + EXPECT_THAT(str, testing::HasSubstr("Label read from ATOMIC_POSITIONS is Mo")); + EXPECT_THAT(str, testing::HasSubstr("Label from ATOMIC_SPECIES is Mg")); + ifs_tmp.close(); + remove("read_atom_positions.tmp"); + remove("read_atom_positions.warn"); +} + +TEST_F(ReadAtomsTest, ReadAtomPositionsWarning3) +{ + std::string fn = "./support/STRU_MgO_WarningC3"; + std::ifstream ifa(fn.c_str()); + std::ofstream ofs_running; + ofs_running.open("read_atom_positions.tmp"); + GlobalV::ofs_warning.open("read_atom_positions.warn"); + // mandatory preliminaries + ucell->ntype = 2; + ucell->atoms = new Atom[ucell->ntype]; + ucell->set_atom_flag = true; + const std::string basis_type = "lcao"; + const std::string orbital_dir = ""; + const std::string init_wfc = ""; + const double onsite_radius = 0.0; + const bool deepks_setorb = true; + const bool rpa = false; + const int nspin = 1; + const bool fixed_atoms = false; + const bool noncolin = false; + const std::string calculation = "scf"; + const std::string esolver_type = "ksdft"; + EXPECT_NO_THROW(unitcell::read_atom_species(ifa, ofs_running, *ucell, + basis_type, orbital_dir, init_wfc, + onsite_radius, deepks_setorb, rpa)); + EXPECT_NO_THROW(unitcell::read_lattice_constant(ifa, ofs_running,ucell->lat)); + EXPECT_DOUBLE_EQ(ucell->latvec.e11, 4.27957); + EXPECT_DOUBLE_EQ(ucell->latvec.e22, 4.27957); + EXPECT_DOUBLE_EQ(ucell->latvec.e33, 4.27957); + EXPECT_NO_THROW(unitcell::read_atom_positions(*ucell, ifa, ofs_running, GlobalV::ofs_warning, nspin, + basis_type, orbital_dir, init_wfc, + onsite_radius, fixed_atoms, noncolin, + calculation, esolver_type, 0)); + ofs_running.close(); + GlobalV::ofs_warning.close(); + ifa.close(); + // check warning file + std::ifstream ifs_tmp; + ifs_tmp.open("read_atom_positions.warn"); + std::string str((std::istreambuf_iterator(ifs_tmp)), std::istreambuf_iterator()); + EXPECT_THAT(str, testing::HasSubstr("read_atom_positions warning : atom number < 0.")); + ifs_tmp.close(); + remove("read_atom_positions.tmp"); + remove("read_atom_positions.warn"); +} + +TEST_F(ReadAtomsTest, ReadAtomPositionsWarning4) +{ + std::string fn = "./support/STRU_MgO_WarningC4"; + std::ifstream ifa(fn.c_str()); + std::ofstream ofs_running; + std::ofstream ofs_warning; + ofs_running.open("read_atom_positions.tmp"); + ofs_warning.open("read_atom_positions.warn"); + // mandatory preliminaries + ucell->ntype = 2; + ucell->atoms = new Atom[ucell->ntype]; + ucell->orbital_fn.resize(ucell->ntype); + ucell->set_atom_flag = true; + const std::string basis_type = "lcao"; + const std::string orbital_dir = ""; + const std::string init_wfc = ""; + const double onsite_radius = 0.0; + const bool deepks_setorb = true; + const bool rpa = false; + const int nspin = 1; + const bool fixed_atoms = false; + const bool noncolin = false; + const std::string calculation = "scf"; + const std::string esolver_type = "ksdft"; + EXPECT_NO_THROW(unitcell::read_atom_species(ifa, ofs_running, *ucell, + basis_type, orbital_dir, init_wfc, + onsite_radius, deepks_setorb, rpa)); + EXPECT_NO_THROW(unitcell::read_lattice_constant(ifa, ofs_running,ucell->lat)); + EXPECT_DOUBLE_EQ(ucell->latvec.e11, 4.27957); + EXPECT_DOUBLE_EQ(ucell->latvec.e22, 4.27957); + EXPECT_DOUBLE_EQ(ucell->latvec.e33, 4.27957); + testing::internal::CaptureStdout(); + EXPECT_EXIT(unitcell::read_atom_positions(*ucell, ifa, ofs_running, ofs_warning, nspin, + basis_type, orbital_dir, init_wfc, + onsite_radius, fixed_atoms, noncolin, + calculation, esolver_type, 0), ::testing::ExitedWithCode(1), ""); + output = testing::internal::GetCapturedStdout(); + EXPECT_THAT(output, testing::HasSubstr("read_atom_positions, mismatch in atom number for atom type: Mg")); + ofs_running.close(); + ofs_warning.close(); + ifa.close(); + remove("read_atom_positions.tmp"); + remove("read_atom_positions.warn"); +} + +TEST_F(ReadAtomsTest, ReadAtomPositionsWarning5) +{ + std::string fn = "./support/STRU_MgO"; + std::ifstream ifa(fn.c_str()); + std::ofstream ofs_running; + ofs_running.open("read_atom_positions.tmp"); + GlobalV::ofs_warning.open("read_atom_positions.warn"); + // mandatory preliminaries + ucell->ntype = 2; + ucell->atoms = new Atom[ucell->ntype]; + ucell->set_atom_flag = true; + const std::string basis_type = "lcao"; + const std::string orbital_dir = ""; + const std::string init_wfc = ""; + const double onsite_radius = 0.0; + const bool deepks_setorb = true; + const bool rpa = false; + const int nspin = 1; + const bool fixed_atoms = true; + const bool noncolin = false; + const std::string calculation = "md"; + const std::string esolver_type = "ksdft"; + EXPECT_NO_THROW(unitcell::read_atom_species(ifa, ofs_running, *ucell, + basis_type, orbital_dir, init_wfc, + onsite_radius, deepks_setorb, rpa)); + EXPECT_NO_THROW(unitcell::read_lattice_constant(ifa, ofs_running,ucell->lat)); + EXPECT_DOUBLE_EQ(ucell->latvec.e11, 4.27957); + EXPECT_DOUBLE_EQ(ucell->latvec.e22, 4.27957); + EXPECT_DOUBLE_EQ(ucell->latvec.e33, 4.27957); + EXPECT_NO_THROW(unitcell::read_atom_positions(*ucell, ifa, ofs_running, GlobalV::ofs_warning, nspin, + basis_type, orbital_dir, init_wfc, + onsite_radius, fixed_atoms, noncolin, + calculation, esolver_type, 0)); + ofs_running.close(); + GlobalV::ofs_warning.close(); + ifa.close(); + // check warning file + std::ifstream ifs_tmp; + ifs_tmp.open("read_atom_positions.warn"); + std::string str((std::istreambuf_iterator(ifs_tmp)), std::istreambuf_iterator()); + EXPECT_THAT(str, testing::HasSubstr("read_atoms warning : no atoms can move in MD simulations!")); + ifs_tmp.close(); + remove("read_atom_positions.tmp"); + remove("read_atom_positions.warn"); +} + +#endif // __LCAO diff --git a/source/source_cell/test/test_read_orb.cpp b/source/source_cell/test/test_read_orb.cpp new file mode 100644 index 00000000000..1d8281268b0 --- /dev/null +++ b/source/source_cell/test/test_read_orb.cpp @@ -0,0 +1,79 @@ +#include "gmock/gmock.h" +#include "gtest/gtest.h" + +#include +#include +#include +#include +#include +#include + +#include "source_base/mathzone.h" +#include "source_cell/unitcell.h" +#include "source_cell/read_orb.h" +#include "prepare_unitcell.h" + +// The test-only cell_info object library does not contain magnetism.cpp, +// so the Magnetism constructor/destructor must be provided locally. +Magnetism::Magnetism() +{ + this->tot_mag = 0.0; + this->abs_mag = 0.0; +} +Magnetism::~Magnetism() +{ +} + +/************************************************ + * unit test of read_orb.cpp + ***********************************************/ + +/** + * - Tested Functions: + * - ReadOrbFile + * - read_orb_file(): read header part of orbital file + * - ReadOrbFileWarning + * - read_orb_file(): ABACUS cannot find the ORBITAL file + */ + +class ReadOrbTest : public ::testing::Test +{ + protected: + std::unique_ptr ucell; + std::string output; +}; + +#ifdef __LCAO +TEST_F(ReadOrbTest, ReadOrbFile) +{ + UcellTestPrepare utp = UcellTestLib["C1H2-Read"]; + ucell = utp.SetUcellInfo(); + std::string orb_file = "./support/C.orb"; + std::ofstream ofs_running; + ofs_running.open("tmp_readorbfile"); + bool result = unitcell::read_orb_file(0, orb_file, ofs_running, &(ucell->atoms[0])); + ofs_running << " result=" << result << std::endl; + EXPECT_TRUE(result); + ofs_running.close(); + EXPECT_EQ(ucell->atoms[0].nw, 25); + remove("tmp_readorbfile"); +} +#endif + +TEST_F(ReadOrbTest, ReadOrbFileWarning) +{ + UcellTestPrepare utp = UcellTestLib["C1H2-Read"]; + ucell = utp.SetUcellInfo(); + std::string orb_file = "./support/CC.orb"; + std::ofstream ofs_running; + ofs_running.open("tmp_readorbfilewarning"); + testing::internal::CaptureStdout(); + bool result = unitcell::read_orb_file(0, orb_file, ofs_running, &(ucell->atoms[0])); + output = testing::internal::GetCapturedStdout(); + ofs_running << output << std::endl; + EXPECT_FALSE(result); + EXPECT_THAT(output, testing::HasSubstr("Element index 1")); + EXPECT_THAT(output, testing::HasSubstr("orbital file: ./support/CC.orb")); + ofs_running.close(); + remove("tmp_readorbfilewarning"); +} diff --git a/source/source_cell/test/test_update_cell.cpp b/source/source_cell/test/test_update_cell.cpp new file mode 100644 index 00000000000..df888e989c8 --- /dev/null +++ b/source/source_cell/test/test_update_cell.cpp @@ -0,0 +1,383 @@ +#include "gmock/gmock.h" +#include "gtest/gtest.h" + +#include +#include +#include +#include + +#include "source_base/global_variable.h" +#include "source_base/mathzone.h" +#include "source_cell/unitcell.h" +#include "source_cell/update_cell.h" +#include "prepare_unitcell.h" + +// The test-only cell_info object library does not contain magnetism.cpp, +// so the Magnetism constructor/destructor must be provided locally. +Magnetism::Magnetism() +{ + this->tot_mag = 0.0; + this->abs_mag = 0.0; +} +Magnetism::~Magnetism() +{ +} + +/************************************************ + * unit test of update_cell.cpp + ***********************************************/ + +/** + * - Tested Functions: + * - RemakeCell + * - remake_cell(): rebuild cell according to its latName + * - RemakeCellWarnings + * - remake_cell(): deliver warnings when find wrong latname or cos12 + * - PeriodicBoundaryAdjustment + * - periodic_boundary_adjustment(): move atoms inside the unitcell after relaxation + * - UpdateVel + * - update_vel(const ModuleBase::Vector3* vel_in) + */ + +class UpdateCellTest : public ::testing::Test +{ + protected: + std::unique_ptr ucell{new UnitCell}; + std::string output; +}; + +using UpdateCellDeathTest = UpdateCellTest; + + +TEST_F(UpdateCellTest, RemakeCell) +{ + std::vector latname_in = {"sc", + "fcc", + "bcc", + "hexagonal", + "trigonal", + "st", + "bct", + "so", + "baco", + "fco", + "bco", + "sm", + "bacm", + "triclinic"}; + for (int i = 0; i < latname_in.size(); ++i) + { + ucell->latvec.e11 = 10.0; + ucell->latvec.e12 = 0.00; + ucell->latvec.e13 = 0.00; + ucell->latvec.e21 = 0.00; + ucell->latvec.e22 = 10.0; + ucell->latvec.e23 = 0.00; + ucell->latvec.e31 = 0.00; + ucell->latvec.e32 = 0.00; + ucell->latvec.e33 = 10.0; + ucell->latName = latname_in[i]; + unitcell::remake_cell(ucell->lat); + if (latname_in[i] == "sc") + { + double celldm + = std::sqrt(pow(ucell->latvec.e11, 2) + pow(ucell->latvec.e12, 2) + pow(ucell->latvec.e13, 2)); + EXPECT_DOUBLE_EQ(ucell->latvec.e11, celldm); + } + else if (latname_in[i] == "fcc") + { + double celldm = std::sqrt(pow(ucell->latvec.e11, 2) + pow(ucell->latvec.e12, 2) + pow(ucell->latvec.e13, 2)) + / std::sqrt(2.0); + EXPECT_DOUBLE_EQ(ucell->latvec.e11, -celldm); + EXPECT_DOUBLE_EQ(ucell->latvec.e12, 0.0); + EXPECT_DOUBLE_EQ(ucell->latvec.e13, celldm); + EXPECT_DOUBLE_EQ(ucell->latvec.e21, 0.0); + EXPECT_DOUBLE_EQ(ucell->latvec.e22, celldm); + EXPECT_DOUBLE_EQ(ucell->latvec.e23, celldm); + EXPECT_DOUBLE_EQ(ucell->latvec.e31, -celldm); + EXPECT_DOUBLE_EQ(ucell->latvec.e32, celldm); + EXPECT_DOUBLE_EQ(ucell->latvec.e33, 0.0); + } + else if (latname_in[i] == "bcc") + { + double celldm = std::sqrt(pow(ucell->latvec.e11, 2) + pow(ucell->latvec.e12, 2) + pow(ucell->latvec.e13, 2)) + / std::sqrt(3.0); + EXPECT_DOUBLE_EQ(ucell->latvec.e11, celldm); + EXPECT_DOUBLE_EQ(ucell->latvec.e12, celldm); + EXPECT_DOUBLE_EQ(ucell->latvec.e13, celldm); + EXPECT_DOUBLE_EQ(ucell->latvec.e21, -celldm); + EXPECT_DOUBLE_EQ(ucell->latvec.e22, celldm); + EXPECT_DOUBLE_EQ(ucell->latvec.e23, celldm); + EXPECT_DOUBLE_EQ(ucell->latvec.e31, -celldm); + EXPECT_DOUBLE_EQ(ucell->latvec.e32, -celldm); + EXPECT_DOUBLE_EQ(ucell->latvec.e33, celldm); + } + else if (latname_in[i] == "hexagonal") + { + double celldm1 + = std::sqrt(pow(ucell->latvec.e11, 2) + pow(ucell->latvec.e12, 2) + pow(ucell->latvec.e13, 2)); + double celldm3 + = std::sqrt(pow(ucell->latvec.e31, 2) + pow(ucell->latvec.e32, 2) + pow(ucell->latvec.e33, 2)); + double mathfoo = sqrt(3.0) / 2.0; + EXPECT_DOUBLE_EQ(ucell->latvec.e11, celldm1); + EXPECT_DOUBLE_EQ(ucell->latvec.e12, 0.0); + EXPECT_DOUBLE_EQ(ucell->latvec.e13, 0.0); + EXPECT_DOUBLE_EQ(ucell->latvec.e21, -0.5 * celldm1); + EXPECT_DOUBLE_EQ(ucell->latvec.e22, celldm1 * mathfoo); + EXPECT_DOUBLE_EQ(ucell->latvec.e23, 0.0); + EXPECT_DOUBLE_EQ(ucell->latvec.e31, 0.0); + EXPECT_DOUBLE_EQ(ucell->latvec.e32, 0.0); + EXPECT_DOUBLE_EQ(ucell->latvec.e33, celldm3); + } + else if (latname_in[i] == "trigonal") + { + double a1 = std::sqrt(pow(ucell->latvec.e11, 2) + pow(ucell->latvec.e12, 2) + pow(ucell->latvec.e13, 2)); + double a2 = std::sqrt(pow(ucell->latvec.e21, 2) + pow(ucell->latvec.e22, 2) + pow(ucell->latvec.e23, 2)); + double a1da2 = (ucell->latvec.e11 * ucell->latvec.e21 + ucell->latvec.e12 * ucell->latvec.e22 + + ucell->latvec.e13 * ucell->latvec.e23); + double cosgamma = a1da2 / (a1 * a2); + double tx = std::sqrt((1.0 - cosgamma) / 2.0); + double ty = std::sqrt((1.0 - cosgamma) / 6.0); + double tz = std::sqrt((1.0 + 2.0 * cosgamma) / 3.0); + EXPECT_DOUBLE_EQ(ucell->latvec.e11, a1 * tx); + EXPECT_DOUBLE_EQ(ucell->latvec.e12, -a1 * ty); + EXPECT_DOUBLE_EQ(ucell->latvec.e13, a1 * tz); + EXPECT_DOUBLE_EQ(ucell->latvec.e21, 0.0); + EXPECT_DOUBLE_EQ(ucell->latvec.e22, 2.0 * a1 * ty); + EXPECT_DOUBLE_EQ(ucell->latvec.e23, a1 * tz); + EXPECT_DOUBLE_EQ(ucell->latvec.e31, -a1 * tx); + EXPECT_DOUBLE_EQ(ucell->latvec.e32, -a1 * ty); + EXPECT_DOUBLE_EQ(ucell->latvec.e33, a1 * tz); + } + else if (latname_in[i] == "st") + { + double a1 = std::sqrt(pow(ucell->latvec.e11, 2) + pow(ucell->latvec.e12, 2) + pow(ucell->latvec.e13, 2)); + double a3 = std::sqrt(pow(ucell->latvec.e31, 2) + pow(ucell->latvec.e32, 2) + pow(ucell->latvec.e33, 2)); + EXPECT_DOUBLE_EQ(ucell->latvec.e11, a1); + EXPECT_DOUBLE_EQ(ucell->latvec.e12, 0.0); + EXPECT_DOUBLE_EQ(ucell->latvec.e13, 0.0); + EXPECT_DOUBLE_EQ(ucell->latvec.e21, 0.0); + EXPECT_DOUBLE_EQ(ucell->latvec.e22, a1); + EXPECT_DOUBLE_EQ(ucell->latvec.e23, 0.0); + EXPECT_DOUBLE_EQ(ucell->latvec.e31, 0.0); + EXPECT_DOUBLE_EQ(ucell->latvec.e32, 0.0); + EXPECT_DOUBLE_EQ(ucell->latvec.e33, a3); + } + else if (latname_in[i] == "bct") + { + double d1 = std::abs(ucell->latvec.e11); + double d2 = std::abs(ucell->latvec.e13); + EXPECT_DOUBLE_EQ(ucell->latvec.e11, d1); + EXPECT_DOUBLE_EQ(ucell->latvec.e12, -d1); + EXPECT_DOUBLE_EQ(ucell->latvec.e13, d2); + EXPECT_DOUBLE_EQ(ucell->latvec.e21, d1); + EXPECT_DOUBLE_EQ(ucell->latvec.e22, d1); + EXPECT_DOUBLE_EQ(ucell->latvec.e23, d2); + EXPECT_DOUBLE_EQ(ucell->latvec.e31, -d1); + EXPECT_DOUBLE_EQ(ucell->latvec.e32, -d1); + EXPECT_DOUBLE_EQ(ucell->latvec.e33, d2); + } + else if (latname_in[i] == "so") + { + double a1 = std::sqrt(pow(ucell->latvec.e11, 2) + pow(ucell->latvec.e12, 2) + pow(ucell->latvec.e13, 2)); + double a2 = std::sqrt(pow(ucell->latvec.e21, 2) + pow(ucell->latvec.e22, 2) + pow(ucell->latvec.e23, 2)); + double a3 = std::sqrt(pow(ucell->latvec.e31, 2) + pow(ucell->latvec.e32, 2) + pow(ucell->latvec.e33, 2)); + EXPECT_DOUBLE_EQ(ucell->latvec.e11, a1); + EXPECT_DOUBLE_EQ(ucell->latvec.e12, 0.0); + EXPECT_DOUBLE_EQ(ucell->latvec.e13, 0.0); + EXPECT_DOUBLE_EQ(ucell->latvec.e21, 0.0); + EXPECT_DOUBLE_EQ(ucell->latvec.e22, a2); + EXPECT_DOUBLE_EQ(ucell->latvec.e23, 0.0); + EXPECT_DOUBLE_EQ(ucell->latvec.e31, 0.0); + EXPECT_DOUBLE_EQ(ucell->latvec.e32, 0.0); + EXPECT_DOUBLE_EQ(ucell->latvec.e33, a3); + } + else if (latname_in[i] == "baco") + { + double d1 = std::abs(ucell->latvec.e11); + double d2 = std::abs(ucell->latvec.e22); + double d3 = std::abs(ucell->latvec.e33); + EXPECT_DOUBLE_EQ(ucell->latvec.e11, d1); + EXPECT_DOUBLE_EQ(ucell->latvec.e12, d2); + EXPECT_DOUBLE_EQ(ucell->latvec.e13, 0.0); + EXPECT_DOUBLE_EQ(ucell->latvec.e21, -d1); + EXPECT_DOUBLE_EQ(ucell->latvec.e22, d2); + EXPECT_DOUBLE_EQ(ucell->latvec.e23, 0.0); + EXPECT_DOUBLE_EQ(ucell->latvec.e31, 0.0); + EXPECT_DOUBLE_EQ(ucell->latvec.e32, 0.0); + EXPECT_DOUBLE_EQ(ucell->latvec.e33, d3); + } + else if (latname_in[i] == "fco") + { + double d1 = std::abs(ucell->latvec.e11); + double d2 = std::abs(ucell->latvec.e22); + double d3 = std::abs(ucell->latvec.e33); + EXPECT_DOUBLE_EQ(ucell->latvec.e11, d1); + EXPECT_DOUBLE_EQ(ucell->latvec.e12, 0.0); + EXPECT_DOUBLE_EQ(ucell->latvec.e13, d3); + EXPECT_DOUBLE_EQ(ucell->latvec.e21, d1); + EXPECT_DOUBLE_EQ(ucell->latvec.e22, d2); + EXPECT_DOUBLE_EQ(ucell->latvec.e23, 0.0); + EXPECT_DOUBLE_EQ(ucell->latvec.e31, 0.0); + EXPECT_DOUBLE_EQ(ucell->latvec.e32, d2); + EXPECT_DOUBLE_EQ(ucell->latvec.e33, d3); + } + else if (latname_in[i] == "bco") + { + double d1 = std::abs(ucell->latvec.e11); + double d2 = std::abs(ucell->latvec.e22); + double d3 = std::abs(ucell->latvec.e33); + EXPECT_DOUBLE_EQ(ucell->latvec.e11, d1); + EXPECT_DOUBLE_EQ(ucell->latvec.e12, d2); + EXPECT_DOUBLE_EQ(ucell->latvec.e13, d3); + EXPECT_DOUBLE_EQ(ucell->latvec.e21, -d1); + EXPECT_DOUBLE_EQ(ucell->latvec.e22, d2); + EXPECT_DOUBLE_EQ(ucell->latvec.e23, d3); + EXPECT_DOUBLE_EQ(ucell->latvec.e31, -d1); + EXPECT_DOUBLE_EQ(ucell->latvec.e32, -d2); + EXPECT_DOUBLE_EQ(ucell->latvec.e33, d3); + } + else if (latname_in[i] == "sm") + { + double a1 = std::sqrt(pow(ucell->latvec.e11, 2) + pow(ucell->latvec.e12, 2) + pow(ucell->latvec.e13, 2)); + double a2 = std::sqrt(pow(ucell->latvec.e21, 2) + pow(ucell->latvec.e22, 2) + pow(ucell->latvec.e23, 2)); + double a3 = std::sqrt(pow(ucell->latvec.e31, 2) + pow(ucell->latvec.e32, 2) + pow(ucell->latvec.e33, 2)); + double a1da2 = (ucell->latvec.e11 * ucell->latvec.e21 + ucell->latvec.e12 * ucell->latvec.e22 + + ucell->latvec.e13 * ucell->latvec.e23); + double cosgamma = a1da2 / (a1 * a2); + double d1 = a2 * cosgamma; + double d2 = a2 * std::sqrt(1.0 - cosgamma * cosgamma); + EXPECT_DOUBLE_EQ(ucell->latvec.e11, a1); + EXPECT_DOUBLE_EQ(ucell->latvec.e12, 0.0); + EXPECT_DOUBLE_EQ(ucell->latvec.e13, 0.0); + EXPECT_DOUBLE_EQ(ucell->latvec.e21, d1); + EXPECT_DOUBLE_EQ(ucell->latvec.e22, d2); + EXPECT_DOUBLE_EQ(ucell->latvec.e23, 0.0); + EXPECT_DOUBLE_EQ(ucell->latvec.e31, 0.0); + EXPECT_DOUBLE_EQ(ucell->latvec.e32, 0.0); + EXPECT_DOUBLE_EQ(ucell->latvec.e33, a3); + } + else if (latname_in[i] == "bacm") + { + double d1 = std::abs(ucell->latvec.e11); + double a2 = std::sqrt(pow(ucell->latvec.e21, 2) + pow(ucell->latvec.e22, 2) + pow(ucell->latvec.e23, 2)); + double d3 = std::abs(ucell->latvec.e13); + double cosgamma = ucell->latvec.e21 / a2; + double f1 = a2 * cosgamma; + double f2 = a2 * std::sqrt(1.0 - cosgamma * cosgamma); + EXPECT_DOUBLE_EQ(ucell->latvec.e11, d1); + EXPECT_DOUBLE_EQ(ucell->latvec.e12, 0.0); + EXPECT_DOUBLE_EQ(ucell->latvec.e13, -d3); + EXPECT_DOUBLE_EQ(ucell->latvec.e21, f1); + EXPECT_DOUBLE_EQ(ucell->latvec.e22, f2); + EXPECT_DOUBLE_EQ(ucell->latvec.e23, 0.0); + EXPECT_DOUBLE_EQ(ucell->latvec.e31, d1); + EXPECT_DOUBLE_EQ(ucell->latvec.e32, 0.0); + EXPECT_DOUBLE_EQ(ucell->latvec.e33, d3); + } + else if (latname_in[i] == "triclinic") + { + double a1 = std::sqrt(pow(ucell->latvec.e11, 2) + pow(ucell->latvec.e12, 2) + pow(ucell->latvec.e13, 2)); + double a2 = std::sqrt(pow(ucell->latvec.e21, 2) + pow(ucell->latvec.e22, 2) + pow(ucell->latvec.e23, 2)); + double a3 = std::sqrt(pow(ucell->latvec.e31, 2) + pow(ucell->latvec.e32, 2) + pow(ucell->latvec.e33, 2)); + double a1da2 = (ucell->latvec.e11 * ucell->latvec.e21 + ucell->latvec.e12 * ucell->latvec.e22 + + ucell->latvec.e13 * ucell->latvec.e23); + double a1da3 = (ucell->latvec.e11 * ucell->latvec.e31 + ucell->latvec.e12 * ucell->latvec.e32 + + ucell->latvec.e13 * ucell->latvec.e33); + double a2da3 = (ucell->latvec.e21 * ucell->latvec.e31 + ucell->latvec.e22 * ucell->latvec.e32 + + ucell->latvec.e23 * ucell->latvec.e33); + double cosgamma = a1da2 / a1 / a2; + double singamma = std::sqrt(1.0 - cosgamma * cosgamma); + double cosbeta = a1da3 / a1 / a3; + double cosalpha = a2da3 / a2 / a3; + double d1 = std::sqrt(1.0 + 2.0 * cosgamma * cosbeta * cosalpha - cosgamma * cosgamma - cosbeta * cosbeta + - cosalpha * cosalpha) + / singamma; + EXPECT_DOUBLE_EQ(ucell->latvec.e11, a1); + EXPECT_DOUBLE_EQ(ucell->latvec.e12, 0.0); + EXPECT_DOUBLE_EQ(ucell->latvec.e13, 0.0); + EXPECT_DOUBLE_EQ(ucell->latvec.e21, a2 * cosgamma); + EXPECT_DOUBLE_EQ(ucell->latvec.e22, a2 * singamma); + EXPECT_DOUBLE_EQ(ucell->latvec.e23, 0.0); + EXPECT_DOUBLE_EQ(ucell->latvec.e31, a3 * cosbeta); + EXPECT_DOUBLE_EQ(ucell->latvec.e32, a3 * (cosalpha - cosbeta * cosgamma) / singamma); + EXPECT_DOUBLE_EQ(ucell->latvec.e33, a3 * d1); + } + } +} + +TEST_F(UpdateCellDeathTest, RemakeCellWarnings) +{ + std::vector latname_in = {"user_defined_lattice", "trigonal", "bacm", "triclinic", "arbitrary"}; + for (int i = 0; i < latname_in.size(); ++i) + { + ucell->latvec.e11 = 10.0; + ucell->latvec.e12 = 0.00; + ucell->latvec.e13 = 0.00; + ucell->latvec.e21 = 10.0; + ucell->latvec.e22 = 0.00; + ucell->latvec.e23 = 0.00; + ucell->latvec.e31 = 0.00; + ucell->latvec.e32 = 0.00; + ucell->latvec.e33 = 10.0; + ucell->latName = latname_in[i]; + testing::internal::CaptureStdout(); + EXPECT_EXIT(unitcell::remake_cell(ucell->lat), ::testing::ExitedWithCode(1), ""); + std::string output = testing::internal::GetCapturedStdout(); + if (latname_in[i] == "user_defined_lattice") + { + EXPECT_THAT(output, testing::HasSubstr("to use fixed_ibrav, latname must be provided")); + } + else if (latname_in[i] == "trigonal" || latname_in[i] == "bacm" || latname_in[i] == "triclinic") + { + EXPECT_THAT(output, testing::HasSubstr("wrong cos12!")); + } + else + { + EXPECT_THAT(output, testing::HasSubstr("latname type not supported!")); + } + } +} + +// mohan comment out 2025-07-14 +/* +TEST_F(UpdateCellDeathTest, PeriodicBoundaryAdjustment1) +{ + UcellTestPrepare utp = UcellTestLib["C1H2-PBA"]; + ucell = utp.SetUcellInfo(); + testing::internal::CaptureStdout(); + EXPECT_EXIT(unitcell::periodic_boundary_adjustment( + ucell->atoms,ucell->latvec,ucell->ntype), + ::testing::ExitedWithCode(1), ""); + std::string output = testing::internal::GetCapturedStdout(); + EXPECT_THAT(output, testing::HasSubstr("the movement of atom is larger than the length of cell")); +} +*/ + +TEST_F(UpdateCellTest, PeriodicBoundaryAdjustment2) +{ + UcellTestPrepare utp = UcellTestLib["C1H2-Index"]; + ucell = utp.SetUcellInfo(); + EXPECT_NO_THROW(unitcell::periodic_boundary_adjustment( + ucell->atoms,ucell->latvec,ucell->ntype)); +} + +TEST_F(UpdateCellTest, UpdateVel) +{ + UcellTestPrepare utp = UcellTestLib["C1H2-Index"]; + ucell = utp.SetUcellInfo(); + ModuleBase::Vector3* vel_in = new ModuleBase::Vector3[ucell->nat]; + for (int iat = 0; iat < ucell->nat; ++iat) + { + vel_in[iat].set(iat * 0.1, iat * 0.1, iat * 0.1); + } + unitcell::update_vel(vel_in,ucell->ntype,ucell->nat,ucell->atoms); + for (int iat = 0; iat < ucell->nat; ++iat) + { + EXPECT_DOUBLE_EQ(vel_in[iat].x, 0.1 * iat); + EXPECT_DOUBLE_EQ(vel_in[iat].y, 0.1 * iat); + EXPECT_DOUBLE_EQ(vel_in[iat].z, 0.1 * iat); + } + delete[] vel_in; +} + diff --git a/source/source_cell/test/unitcell_test.cpp b/source/source_cell/test/unitcell_test.cpp index ece30f6f748..630390c5b7d 100644 --- a/source/source_cell/test/unitcell_test.cpp +++ b/source/source_cell/test/unitcell_test.cpp @@ -1,19 +1,11 @@ #include "gmock/gmock.h" #include "gtest/gtest.h" -#include "source_cell/cal_ux.h" -#include "source_cell/read_orb.h" -#include "source_cell/read_pp_ucell.h" #include "source_cell/read_stru.h" -#include "source_cell/cell_tools.h" -#include "source_cell/print_cell.h" #include "memory" -#include "source_cell/read_stru.h" #include "source_base/global_variable.h" #include "source_base/mathzone.h" #include "prepare_unitcell.h" -#include "source_cell/update_cell.h" -#include "source_cell/read_stru.h" #include #include #include @@ -38,92 +30,19 @@ Magnetism::~Magnetism() * - UnitCell() and ~UnitCell() * - Setup: * - setup_from_input(): to set latname, ntype, lmaxmax, init_vel, and lc - * - if_cell_can_change(): judge if any lattice vector can change - * - RemakeCell - * - remake_cell(): rebuild cell according to its latName - * - RemakeCellWarnings - * - remake_cell(): deliver warnings when find wrong latname or cos12 - * - JudgeParallel - * - judge_parallel: judge if two vectors a[3] and Vector3 b are parallel * - Index * - set_iat2iait(): set index relations in two arrays of Unitcell: iat2it[nat], iat2ia[nat] * - iat2iait(): depends on the above function, but can find both ia & it from iat * - ijat2iaitjajt(): find ia, it, ja, jt from ijat (ijat_max = nat*nat) * which collapses it, ia, jt, ja loop into a single loop * - GetAtomCounts - * - get_atomCounts(): get atomCounts, which is a map from atom type to atom number + * - get_atom_Counts(): get atomCounts, which is a map from atom type to atom number * - GetOrbitalCounts * - get_orbitalCounts(): get orbitalCounts, which is a map from atom type to orbital number * - CheckDTau * - check_dtau(): move all atomic coordinates into the first unitcell, i.e. in between [0,1) * - CheckTau * - check_tau(): check if any "two atoms are too close" - * - SelectiveDynamics - * - if_atoms_can_move():it is true if any coordinates of any atom can move, i.e. mbl = 1 - * - PeriodicBoundaryAdjustment - * - periodic_boundary_adjustment(): move atoms inside the unitcell after relaxation - * - PrintCell - * - print_cell(ofs): print basic cell info into ofs - * - PrintSTRU - * - print_stru_file(): print STRU file of ABACUS - * - PrintTauDirect - * - PrintTauCartesian - * - print_tau(): print atomic coordinates, magmom and initial velocities - * - PrintUnitcellPseudo - * - Actually an integrated function to call unitcell::print_cell and Atom::print_Atom - * - UpdateVel - * - update_vel(const ModuleBase::Vector3* vel_in) - * - CalUx - * - cal_ux(UnitCell& ucell): calculate magnetic moments of cell - * - ReadOrbFile - * - read_orb_file(): read header part of orbital file - * - ReadOrbFileWarning - * - read_orb_file(): ABACUS Cannot find the ORBITAL file - * - ReadAtomSpecies - * - read_atom_species(): a successful case - * - ReadAtomSpeciesWarning1 - * - read_atom_species(): unrecognized pseudopotential type. - * - ReadAtomSpeciesWarning2 - * - read_atom_species(): lat0<=0.0 - * - ReadAtomSpeciesWarning3 - * - read_atom_species(): do not use LATTICE_PARAMETERS without explicit specification of lattice type - * - ReadAtomSpeciesWarning4 - * - read_atom_species():do not use LATTICE_VECTORS along with explicit specification of lattice type - * - ReadAtomSpeciesWarning5 - * - read_atom_species():latname not supported - * - ReadAtomSpeciesLatName - * - read_atom_species(): various latname - * - ReadAtomPositionsS1 - * - read_atom_positions(): spin 1 case - * - ReadAtomPositionsS2 - * - read_atom_positions(): spin 2 case - * - ReadAtomPositionsS4Noncolin - * - read_atom_positions(): spin 4 noncolinear case - * - ReadAtomPositionsS4Colin - * - read_atom_positions(): spin 4 colinear case - * - ReadAtomPositionsC - * - read_atom_positions(): Cartesian coordinates - * - ReadAtomPositionsCA - * - read_atom_positions(): Cartesian_angstrom coordinates - * - ReadAtomPositionsCACXY - * - read_atom_positions(): Cartesian_angstrom_center_xy coordinates - * - ReadAtomPositionsCACXZ - * - read_atom_positions(): Cartesian_angstrom_center_xz coordinates - * - ReadAtomPositionsCACXYZ - * - read_atom_positions(): Cartesian_angstrom_center_xyz coordinates - * - ReadAtomPositionsCAU - * - read_atom_positions(): Cartesian_au coordinates - * - ReadAtomPositionsWarning1 - * - read_atom_positions(): unknown type of coordinates - * - ReadAtomPositionsWarning2 - * - read_atom_positions(): atomic label inconsistency between ATOM_POSITIONS - * and ATOM_SPECIES - * - ReadAtomPositionsWarning3 - * - read_atom_positions(): warning : atom number < 0 - * - ReadAtomPositionsWarning4 - * - read_atom_positions(): mismatch in atom number for atom type - * - ReadAtomPositionsWarning5 - * - read_atom_positions(): no atoms can move in MD simulations! */ class UcellTest : public ::testing::Test @@ -174,384 +93,52 @@ TEST_F(UcellTest, Setup) EXPECT_EQ(ucell->lat_axis_free[0], 1); EXPECT_EQ(ucell->lat_axis_free[1], 1); EXPECT_EQ(ucell->lat_axis_free[2], 1); - EXPECT_TRUE(unitcell::if_cell_can_change(ucell->lat_axis_free)); } else if (fixed_axes_in[i] == "a") { EXPECT_EQ(ucell->lat_axis_free[0], 0); EXPECT_EQ(ucell->lat_axis_free[1], 1); EXPECT_EQ(ucell->lat_axis_free[2], 1); - EXPECT_TRUE(unitcell::if_cell_can_change(ucell->lat_axis_free)); } else if (fixed_axes_in[i] == "b") { EXPECT_EQ(ucell->lat_axis_free[0], 1); EXPECT_EQ(ucell->lat_axis_free[1], 0); EXPECT_EQ(ucell->lat_axis_free[2], 1); - EXPECT_TRUE(unitcell::if_cell_can_change(ucell->lat_axis_free)); } else if (fixed_axes_in[i] == "c") { EXPECT_EQ(ucell->lat_axis_free[0], 1); EXPECT_EQ(ucell->lat_axis_free[1], 1); EXPECT_EQ(ucell->lat_axis_free[2], 0); - EXPECT_TRUE(unitcell::if_cell_can_change(ucell->lat_axis_free)); } else if (fixed_axes_in[i] == "ab") { EXPECT_EQ(ucell->lat_axis_free[0], 0); EXPECT_EQ(ucell->lat_axis_free[1], 0); EXPECT_EQ(ucell->lat_axis_free[2], 1); - EXPECT_TRUE(unitcell::if_cell_can_change(ucell->lat_axis_free)); } else if (fixed_axes_in[i] == "ac") { EXPECT_EQ(ucell->lat_axis_free[0], 0); EXPECT_EQ(ucell->lat_axis_free[1], 1); EXPECT_EQ(ucell->lat_axis_free[2], 0); - EXPECT_TRUE(unitcell::if_cell_can_change(ucell->lat_axis_free)); } else if (fixed_axes_in[i] == "bc") { EXPECT_EQ(ucell->lat_axis_free[0], 1); EXPECT_EQ(ucell->lat_axis_free[1], 0); EXPECT_EQ(ucell->lat_axis_free[2], 0); - EXPECT_TRUE(unitcell::if_cell_can_change(ucell->lat_axis_free)); } else if (fixed_axes_in[i] == "abc") { EXPECT_EQ(ucell->lat_axis_free[0], 0); EXPECT_EQ(ucell->lat_axis_free[1], 0); EXPECT_EQ(ucell->lat_axis_free[2], 0); - EXPECT_FALSE(unitcell::if_cell_can_change(ucell->lat_axis_free)); } } } -TEST_F(UcellDeathTest, CompareAatomLabel) -{ - std::string stru_label[] - = {"Ag", "Ag", "Ag", "47", "47", "47", "Silver", "Silver", "Silver", "Ag", "Ag", "Ag", "Ag_empty"}; - std::string pseudo_label[] - = {"Ag", "47", "Silver", "Ag", "47", "Silver", "Ag", "47", "Silver", "Ag1", "ag", "ag_locpsp", "Ag"}; - for (int it = 0; it < 12; it++) - { - unitcell::compare_atom_labels(stru_label[it], pseudo_label[it]); - } - stru_label[0] = "Fe"; - pseudo_label[0] = "O"; - std::string atom_label_in_orbtial = "atom label in orbital file "; - std::string mismatch_with_pseudo = " mismatch with pseudo file of "; - testing::internal::CaptureStdout(); - EXPECT_EXIT(unitcell::compare_atom_labels(stru_label[0], pseudo_label[0]), ::testing::ExitedWithCode(1), ""); - output = testing::internal::GetCapturedStdout(); - EXPECT_THAT(output, - testing::HasSubstr(atom_label_in_orbtial + stru_label[0] + mismatch_with_pseudo + pseudo_label[0])); -} - -TEST_F(UcellTest, RemakeCell) -{ - std::vector latname_in = {"sc", - "fcc", - "bcc", - "hexagonal", - "trigonal", - "st", - "bct", - "so", - "baco", - "fco", - "bco", - "sm", - "bacm", - "triclinic"}; - for (int i = 0; i < latname_in.size(); ++i) - { - ucell->latvec.e11 = 10.0; - ucell->latvec.e12 = 0.00; - ucell->latvec.e13 = 0.00; - ucell->latvec.e21 = 0.00; - ucell->latvec.e22 = 10.0; - ucell->latvec.e23 = 0.00; - ucell->latvec.e31 = 0.00; - ucell->latvec.e32 = 0.00; - ucell->latvec.e33 = 10.0; - ucell->latName = latname_in[i]; - unitcell::remake_cell(ucell->lat); - if (latname_in[i] == "sc") - { - double celldm - = std::sqrt(pow(ucell->latvec.e11, 2) + pow(ucell->latvec.e12, 2) + pow(ucell->latvec.e13, 2)); - EXPECT_DOUBLE_EQ(ucell->latvec.e11, celldm); - } - else if (latname_in[i] == "fcc") - { - double celldm = std::sqrt(pow(ucell->latvec.e11, 2) + pow(ucell->latvec.e12, 2) + pow(ucell->latvec.e13, 2)) - / std::sqrt(2.0); - EXPECT_DOUBLE_EQ(ucell->latvec.e11, -celldm); - EXPECT_DOUBLE_EQ(ucell->latvec.e12, 0.0); - EXPECT_DOUBLE_EQ(ucell->latvec.e13, celldm); - EXPECT_DOUBLE_EQ(ucell->latvec.e21, 0.0); - EXPECT_DOUBLE_EQ(ucell->latvec.e22, celldm); - EXPECT_DOUBLE_EQ(ucell->latvec.e23, celldm); - EXPECT_DOUBLE_EQ(ucell->latvec.e31, -celldm); - EXPECT_DOUBLE_EQ(ucell->latvec.e32, celldm); - EXPECT_DOUBLE_EQ(ucell->latvec.e33, 0.0); - } - else if (latname_in[i] == "bcc") - { - double celldm = std::sqrt(pow(ucell->latvec.e11, 2) + pow(ucell->latvec.e12, 2) + pow(ucell->latvec.e13, 2)) - / std::sqrt(3.0); - EXPECT_DOUBLE_EQ(ucell->latvec.e11, celldm); - EXPECT_DOUBLE_EQ(ucell->latvec.e12, celldm); - EXPECT_DOUBLE_EQ(ucell->latvec.e13, celldm); - EXPECT_DOUBLE_EQ(ucell->latvec.e21, -celldm); - EXPECT_DOUBLE_EQ(ucell->latvec.e22, celldm); - EXPECT_DOUBLE_EQ(ucell->latvec.e23, celldm); - EXPECT_DOUBLE_EQ(ucell->latvec.e31, -celldm); - EXPECT_DOUBLE_EQ(ucell->latvec.e32, -celldm); - EXPECT_DOUBLE_EQ(ucell->latvec.e33, celldm); - } - else if (latname_in[i] == "hexagonal") - { - double celldm1 - = std::sqrt(pow(ucell->latvec.e11, 2) + pow(ucell->latvec.e12, 2) + pow(ucell->latvec.e13, 2)); - double celldm3 - = std::sqrt(pow(ucell->latvec.e31, 2) + pow(ucell->latvec.e32, 2) + pow(ucell->latvec.e33, 2)); - double mathfoo = sqrt(3.0) / 2.0; - EXPECT_DOUBLE_EQ(ucell->latvec.e11, celldm1); - EXPECT_DOUBLE_EQ(ucell->latvec.e12, 0.0); - EXPECT_DOUBLE_EQ(ucell->latvec.e13, 0.0); - EXPECT_DOUBLE_EQ(ucell->latvec.e21, -0.5 * celldm1); - EXPECT_DOUBLE_EQ(ucell->latvec.e22, celldm1 * mathfoo); - EXPECT_DOUBLE_EQ(ucell->latvec.e23, 0.0); - EXPECT_DOUBLE_EQ(ucell->latvec.e31, 0.0); - EXPECT_DOUBLE_EQ(ucell->latvec.e32, 0.0); - EXPECT_DOUBLE_EQ(ucell->latvec.e33, celldm3); - } - else if (latname_in[i] == "trigonal") - { - double a1 = std::sqrt(pow(ucell->latvec.e11, 2) + pow(ucell->latvec.e12, 2) + pow(ucell->latvec.e13, 2)); - double a2 = std::sqrt(pow(ucell->latvec.e21, 2) + pow(ucell->latvec.e22, 2) + pow(ucell->latvec.e23, 2)); - double a1da2 = (ucell->latvec.e11 * ucell->latvec.e21 + ucell->latvec.e12 * ucell->latvec.e22 - + ucell->latvec.e13 * ucell->latvec.e23); - double cosgamma = a1da2 / (a1 * a2); - double tx = std::sqrt((1.0 - cosgamma) / 2.0); - double ty = std::sqrt((1.0 - cosgamma) / 6.0); - double tz = std::sqrt((1.0 + 2.0 * cosgamma) / 3.0); - EXPECT_DOUBLE_EQ(ucell->latvec.e11, a1 * tx); - EXPECT_DOUBLE_EQ(ucell->latvec.e12, -a1 * ty); - EXPECT_DOUBLE_EQ(ucell->latvec.e13, a1 * tz); - EXPECT_DOUBLE_EQ(ucell->latvec.e21, 0.0); - EXPECT_DOUBLE_EQ(ucell->latvec.e22, 2.0 * a1 * ty); - EXPECT_DOUBLE_EQ(ucell->latvec.e23, a1 * tz); - EXPECT_DOUBLE_EQ(ucell->latvec.e31, -a1 * tx); - EXPECT_DOUBLE_EQ(ucell->latvec.e32, -a1 * ty); - EXPECT_DOUBLE_EQ(ucell->latvec.e33, a1 * tz); - } - else if (latname_in[i] == "st") - { - double a1 = std::sqrt(pow(ucell->latvec.e11, 2) + pow(ucell->latvec.e12, 2) + pow(ucell->latvec.e13, 2)); - double a3 = std::sqrt(pow(ucell->latvec.e31, 2) + pow(ucell->latvec.e32, 2) + pow(ucell->latvec.e33, 2)); - EXPECT_DOUBLE_EQ(ucell->latvec.e11, a1); - EXPECT_DOUBLE_EQ(ucell->latvec.e12, 0.0); - EXPECT_DOUBLE_EQ(ucell->latvec.e13, 0.0); - EXPECT_DOUBLE_EQ(ucell->latvec.e21, 0.0); - EXPECT_DOUBLE_EQ(ucell->latvec.e22, a1); - EXPECT_DOUBLE_EQ(ucell->latvec.e23, 0.0); - EXPECT_DOUBLE_EQ(ucell->latvec.e31, 0.0); - EXPECT_DOUBLE_EQ(ucell->latvec.e32, 0.0); - EXPECT_DOUBLE_EQ(ucell->latvec.e33, a3); - } - else if (latname_in[i] == "bct") - { - double d1 = std::abs(ucell->latvec.e11); - double d2 = std::abs(ucell->latvec.e13); - EXPECT_DOUBLE_EQ(ucell->latvec.e11, d1); - EXPECT_DOUBLE_EQ(ucell->latvec.e12, -d1); - EXPECT_DOUBLE_EQ(ucell->latvec.e13, d2); - EXPECT_DOUBLE_EQ(ucell->latvec.e21, d1); - EXPECT_DOUBLE_EQ(ucell->latvec.e22, d1); - EXPECT_DOUBLE_EQ(ucell->latvec.e23, d2); - EXPECT_DOUBLE_EQ(ucell->latvec.e31, -d1); - EXPECT_DOUBLE_EQ(ucell->latvec.e32, -d1); - EXPECT_DOUBLE_EQ(ucell->latvec.e33, d2); - } - else if (latname_in[i] == "so") - { - double a1 = std::sqrt(pow(ucell->latvec.e11, 2) + pow(ucell->latvec.e12, 2) + pow(ucell->latvec.e13, 2)); - double a2 = std::sqrt(pow(ucell->latvec.e21, 2) + pow(ucell->latvec.e22, 2) + pow(ucell->latvec.e23, 2)); - double a3 = std::sqrt(pow(ucell->latvec.e31, 2) + pow(ucell->latvec.e32, 2) + pow(ucell->latvec.e33, 2)); - EXPECT_DOUBLE_EQ(ucell->latvec.e11, a1); - EXPECT_DOUBLE_EQ(ucell->latvec.e12, 0.0); - EXPECT_DOUBLE_EQ(ucell->latvec.e13, 0.0); - EXPECT_DOUBLE_EQ(ucell->latvec.e21, 0.0); - EXPECT_DOUBLE_EQ(ucell->latvec.e22, a2); - EXPECT_DOUBLE_EQ(ucell->latvec.e23, 0.0); - EXPECT_DOUBLE_EQ(ucell->latvec.e31, 0.0); - EXPECT_DOUBLE_EQ(ucell->latvec.e32, 0.0); - EXPECT_DOUBLE_EQ(ucell->latvec.e33, a3); - } - else if (latname_in[i] == "baco") - { - double d1 = std::abs(ucell->latvec.e11); - double d2 = std::abs(ucell->latvec.e22); - double d3 = std::abs(ucell->latvec.e33); - EXPECT_DOUBLE_EQ(ucell->latvec.e11, d1); - EXPECT_DOUBLE_EQ(ucell->latvec.e12, d2); - EXPECT_DOUBLE_EQ(ucell->latvec.e13, 0.0); - EXPECT_DOUBLE_EQ(ucell->latvec.e21, -d1); - EXPECT_DOUBLE_EQ(ucell->latvec.e22, d2); - EXPECT_DOUBLE_EQ(ucell->latvec.e23, 0.0); - EXPECT_DOUBLE_EQ(ucell->latvec.e31, 0.0); - EXPECT_DOUBLE_EQ(ucell->latvec.e32, 0.0); - EXPECT_DOUBLE_EQ(ucell->latvec.e33, d3); - } - else if (latname_in[i] == "fco") - { - double d1 = std::abs(ucell->latvec.e11); - double d2 = std::abs(ucell->latvec.e22); - double d3 = std::abs(ucell->latvec.e33); - EXPECT_DOUBLE_EQ(ucell->latvec.e11, d1); - EXPECT_DOUBLE_EQ(ucell->latvec.e12, 0.0); - EXPECT_DOUBLE_EQ(ucell->latvec.e13, d3); - EXPECT_DOUBLE_EQ(ucell->latvec.e21, d1); - EXPECT_DOUBLE_EQ(ucell->latvec.e22, d2); - EXPECT_DOUBLE_EQ(ucell->latvec.e23, 0.0); - EXPECT_DOUBLE_EQ(ucell->latvec.e31, 0.0); - EXPECT_DOUBLE_EQ(ucell->latvec.e32, d2); - EXPECT_DOUBLE_EQ(ucell->latvec.e33, d3); - } - else if (latname_in[i] == "bco") - { - double d1 = std::abs(ucell->latvec.e11); - double d2 = std::abs(ucell->latvec.e22); - double d3 = std::abs(ucell->latvec.e33); - EXPECT_DOUBLE_EQ(ucell->latvec.e11, d1); - EXPECT_DOUBLE_EQ(ucell->latvec.e12, d2); - EXPECT_DOUBLE_EQ(ucell->latvec.e13, d3); - EXPECT_DOUBLE_EQ(ucell->latvec.e21, -d1); - EXPECT_DOUBLE_EQ(ucell->latvec.e22, d2); - EXPECT_DOUBLE_EQ(ucell->latvec.e23, d3); - EXPECT_DOUBLE_EQ(ucell->latvec.e31, -d1); - EXPECT_DOUBLE_EQ(ucell->latvec.e32, -d2); - EXPECT_DOUBLE_EQ(ucell->latvec.e33, d3); - } - else if (latname_in[i] == "sm") - { - double a1 = std::sqrt(pow(ucell->latvec.e11, 2) + pow(ucell->latvec.e12, 2) + pow(ucell->latvec.e13, 2)); - double a2 = std::sqrt(pow(ucell->latvec.e21, 2) + pow(ucell->latvec.e22, 2) + pow(ucell->latvec.e23, 2)); - double a3 = std::sqrt(pow(ucell->latvec.e31, 2) + pow(ucell->latvec.e32, 2) + pow(ucell->latvec.e33, 2)); - double a1da2 = (ucell->latvec.e11 * ucell->latvec.e21 + ucell->latvec.e12 * ucell->latvec.e22 - + ucell->latvec.e13 * ucell->latvec.e23); - double cosgamma = a1da2 / (a1 * a2); - double d1 = a2 * cosgamma; - double d2 = a2 * std::sqrt(1.0 - cosgamma * cosgamma); - EXPECT_DOUBLE_EQ(ucell->latvec.e11, a1); - EXPECT_DOUBLE_EQ(ucell->latvec.e12, 0.0); - EXPECT_DOUBLE_EQ(ucell->latvec.e13, 0.0); - EXPECT_DOUBLE_EQ(ucell->latvec.e21, d1); - EXPECT_DOUBLE_EQ(ucell->latvec.e22, d2); - EXPECT_DOUBLE_EQ(ucell->latvec.e23, 0.0); - EXPECT_DOUBLE_EQ(ucell->latvec.e31, 0.0); - EXPECT_DOUBLE_EQ(ucell->latvec.e32, 0.0); - EXPECT_DOUBLE_EQ(ucell->latvec.e33, a3); - } - else if (latname_in[i] == "bacm") - { - double d1 = std::abs(ucell->latvec.e11); - double a2 = std::sqrt(pow(ucell->latvec.e21, 2) + pow(ucell->latvec.e22, 2) + pow(ucell->latvec.e23, 2)); - double d3 = std::abs(ucell->latvec.e13); - double cosgamma = ucell->latvec.e21 / a2; - double f1 = a2 * cosgamma; - double f2 = a2 * std::sqrt(1.0 - cosgamma * cosgamma); - EXPECT_DOUBLE_EQ(ucell->latvec.e11, d1); - EXPECT_DOUBLE_EQ(ucell->latvec.e12, 0.0); - EXPECT_DOUBLE_EQ(ucell->latvec.e13, -d3); - EXPECT_DOUBLE_EQ(ucell->latvec.e21, f1); - EXPECT_DOUBLE_EQ(ucell->latvec.e22, f2); - EXPECT_DOUBLE_EQ(ucell->latvec.e23, 0.0); - EXPECT_DOUBLE_EQ(ucell->latvec.e31, d1); - EXPECT_DOUBLE_EQ(ucell->latvec.e32, 0.0); - EXPECT_DOUBLE_EQ(ucell->latvec.e33, d3); - } - else if (latname_in[i] == "triclinic") - { - double a1 = std::sqrt(pow(ucell->latvec.e11, 2) + pow(ucell->latvec.e12, 2) + pow(ucell->latvec.e13, 2)); - double a2 = std::sqrt(pow(ucell->latvec.e21, 2) + pow(ucell->latvec.e22, 2) + pow(ucell->latvec.e23, 2)); - double a3 = std::sqrt(pow(ucell->latvec.e31, 2) + pow(ucell->latvec.e32, 2) + pow(ucell->latvec.e33, 2)); - double a1da2 = (ucell->latvec.e11 * ucell->latvec.e21 + ucell->latvec.e12 * ucell->latvec.e22 - + ucell->latvec.e13 * ucell->latvec.e23); - double a1da3 = (ucell->latvec.e11 * ucell->latvec.e31 + ucell->latvec.e12 * ucell->latvec.e32 - + ucell->latvec.e13 * ucell->latvec.e33); - double a2da3 = (ucell->latvec.e21 * ucell->latvec.e31 + ucell->latvec.e22 * ucell->latvec.e32 - + ucell->latvec.e23 * ucell->latvec.e33); - double cosgamma = a1da2 / a1 / a2; - double singamma = std::sqrt(1.0 - cosgamma * cosgamma); - double cosbeta = a1da3 / a1 / a3; - double cosalpha = a2da3 / a2 / a3; - double d1 = std::sqrt(1.0 + 2.0 * cosgamma * cosbeta * cosalpha - cosgamma * cosgamma - cosbeta * cosbeta - - cosalpha * cosalpha) - / singamma; - EXPECT_DOUBLE_EQ(ucell->latvec.e11, a1); - EXPECT_DOUBLE_EQ(ucell->latvec.e12, 0.0); - EXPECT_DOUBLE_EQ(ucell->latvec.e13, 0.0); - EXPECT_DOUBLE_EQ(ucell->latvec.e21, a2 * cosgamma); - EXPECT_DOUBLE_EQ(ucell->latvec.e22, a2 * singamma); - EXPECT_DOUBLE_EQ(ucell->latvec.e23, 0.0); - EXPECT_DOUBLE_EQ(ucell->latvec.e31, a3 * cosbeta); - EXPECT_DOUBLE_EQ(ucell->latvec.e32, a3 * (cosalpha - cosbeta * cosgamma) / singamma); - EXPECT_DOUBLE_EQ(ucell->latvec.e33, a3 * d1); - } - } -} - -TEST_F(UcellDeathTest, RemakeCellWarnings) -{ - std::vector latname_in = {"user_defined_lattice", "trigonal", "bacm", "triclinic", "arbitrary"}; - for (int i = 0; i < latname_in.size(); ++i) - { - ucell->latvec.e11 = 10.0; - ucell->latvec.e12 = 0.00; - ucell->latvec.e13 = 0.00; - ucell->latvec.e21 = 10.0; - ucell->latvec.e22 = 0.00; - ucell->latvec.e23 = 0.00; - ucell->latvec.e31 = 0.00; - ucell->latvec.e32 = 0.00; - ucell->latvec.e33 = 10.0; - ucell->latName = latname_in[i]; - testing::internal::CaptureStdout(); - EXPECT_EXIT(unitcell::remake_cell(ucell->lat), ::testing::ExitedWithCode(1), ""); - std::string output = testing::internal::GetCapturedStdout(); - if (latname_in[i] == "user_defined_lattice") - { - EXPECT_THAT(output, testing::HasSubstr("to use fixed_ibrav, latname must be provided")); - } - else if (latname_in[i] == "trigonal" || latname_in[i] == "bacm" || latname_in[i] == "triclinic") - { - EXPECT_THAT(output, testing::HasSubstr("wrong cos12!")); - } - else - { - EXPECT_THAT(output, testing::HasSubstr("latname type not supported!")); - } - } -} - -TEST_F(UcellTest, JudgeParallel) -{ - ModuleBase::Vector3 b(1.0, 1.0, 1.0); - double a[3] = {1.0, 1.0, 1.0}; - EXPECT_TRUE(unitcell::judge_parallel(a, b)); - - // the negative case, moved here from MagnetismTest.JudgeParallel when the - // duplicate Magnetism::judge_parallel was deleted - double c[3] = {1.0, 0.0, 0.0}; - ModuleBase::Vector3 d(0.0, 1.0, 0.0); - EXPECT_FALSE(unitcell::judge_parallel(c, d)); -} - TEST_F(UcellTest, Index) { UcellTestPrepare utp = UcellTestLib["C1H2-Index"]; @@ -613,10 +200,6 @@ TEST_F(UcellTest, GetAtomCounts) std::map atomCounts = ucell->get_atom_Counts(); EXPECT_EQ(atomCounts[0], 1); EXPECT_EQ(atomCounts[1], 2); - /// atomCounts as vector - std::vector atomCounts2 = unitcell::get_atomCounts(ucell->atoms, ucell->ntype); - EXPECT_EQ(atomCounts2[0], 1); - EXPECT_EQ(atomCounts2[1], 2); } TEST_F(UcellTest, GetOrbitalCounts) @@ -643,14 +226,6 @@ TEST_F(UcellTest, GetLnchiCounts) EXPECT_EQ(LnchiCounts[1][0], 1); EXPECT_EQ(LnchiCounts[1][1], 1); EXPECT_EQ(LnchiCounts[1][2], 1); - /// LnchiCounts as vector - std::vector> LnchiCounts2 = unitcell::get_lnchiCounts(ucell->atoms, ucell->ntype); - EXPECT_EQ(LnchiCounts2[0][0], 1); - EXPECT_EQ(LnchiCounts2[0][1], 1); - EXPECT_EQ(LnchiCounts2[0][2], 1); - EXPECT_EQ(LnchiCounts2[1][0], 1); - EXPECT_EQ(LnchiCounts2[1][1], 1); - EXPECT_EQ(LnchiCounts2[1][2], 1); } TEST_F(UcellTest, CheckDTau) @@ -712,1280 +287,3 @@ TEST_F(UcellTest, CheckTauTrue) EXPECT_EQ(unitcell::check_tau(ucell->atoms ,ucell->ntype, ucell->lat0),true); GlobalV::ofs_warning.close(); } - -TEST_F(UcellTest, SelectiveDynamics) -{ - UcellTestPrepare utp = UcellTestLib["C1H2-SD"]; - ucell = utp.SetUcellInfo(); - EXPECT_TRUE(unitcell::if_atoms_can_move(ucell->atoms, ucell->ntype)); -} - - -// mohan comment out 2025-07-14 -/* -TEST_F(UcellDeathTest, PeriodicBoundaryAdjustment1) -{ - UcellTestPrepare utp = UcellTestLib["C1H2-PBA"]; - ucell = utp.SetUcellInfo(); - testing::internal::CaptureStdout(); - EXPECT_EXIT(unitcell::periodic_boundary_adjustment( - ucell->atoms,ucell->latvec,ucell->ntype), - ::testing::ExitedWithCode(1), ""); - std::string output = testing::internal::GetCapturedStdout(); - EXPECT_THAT(output, testing::HasSubstr("the movement of atom is larger than the length of cell")); -} -*/ - -TEST_F(UcellTest, PeriodicBoundaryAdjustment2) -{ - UcellTestPrepare utp = UcellTestLib["C1H2-Index"]; - ucell = utp.SetUcellInfo(); - EXPECT_NO_THROW(unitcell::periodic_boundary_adjustment( - ucell->atoms,ucell->latvec,ucell->ntype)); -} - -TEST_F(UcellTest, PrintCell) -{ - UcellTestPrepare utp = UcellTestLib["C1H2-Index"]; - ucell = utp.SetUcellInfo(); - std::ofstream ofs; - ofs.open("printcell.log"); - unitcell::print_cell(*ucell, ofs); - ofs.close(); - std::ifstream ifs; - ifs.open("printcell.log"); - std::string str((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); - EXPECT_THAT(str, testing::HasSubstr("latName = bcc")); - EXPECT_THAT(str, testing::HasSubstr("ntype = 2")); - EXPECT_THAT(str, testing::HasSubstr("nat = 3")); - EXPECT_THAT(str, testing::HasSubstr("GGT :")); - EXPECT_THAT(str, testing::HasSubstr("omega = 6748.33")); - remove("printcell.log"); -} - -TEST_F(UcellTest, PrintUnitcellPseudo) -{ - UcellTestPrepare utp = UcellTestLib["C1H2-Index"]; - ucell = utp.SetUcellInfo(); - std::string fn = "printcell.log"; - unitcell::print_unitcell_pseudo(fn, *ucell); - std::ifstream ifs; - ifs.open("printcell.log"); - std::string str((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); - EXPECT_THAT(str, testing::HasSubstr("latName = bcc")); - EXPECT_THAT(str, testing::HasSubstr("ntype = 2")); - EXPECT_THAT(str, testing::HasSubstr("nat = 3")); - EXPECT_THAT(str, testing::HasSubstr("GGT :")); - EXPECT_THAT(str, testing::HasSubstr("omega = 6748.33")); - EXPECT_THAT(str, testing::HasSubstr("label = C")); - EXPECT_THAT(str, testing::HasSubstr("mass = 12")); - EXPECT_THAT(str, testing::HasSubstr("atom_position(cartesian) Dimension = 1")); - EXPECT_THAT(str, testing::HasSubstr("label = H")); - EXPECT_THAT(str, testing::HasSubstr("mass = 1")); - EXPECT_THAT(str, testing::HasSubstr("atom_position(cartesian) Dimension = 2")); - remove("printcell.log"); -} - -// Comments and suggestions on the refactor of UnitCell class -// the test of this function may relies on a ABACUS STRU parser to fully rationally proceed. -// however the parser is not easy to be ready cause the structure of STRU may need to be -// re-designed for being better-orgnized. -// based on present situation, the unittest can only be feasibly written with substr check, -// but will time and time again raise error if format change even a little bit. -// if allow user to change the precision of quantities printed out, then this unittest cannot -// cover these kind of cases. -// In summmary, there are two cents: -// 1. STRU file needed to be re-designed to be more well-organized -// 2. STRU file parser can therefore be programmed more succinctly -TEST_F(UcellTest, PrintSTRU) -{ - UcellTestPrepare utp = UcellTestLib["C1H2-Index"]; - ucell = utp.SetUcellInfo(); - // Cartesian type of coordinates - std::string fn = "C1H2_STRU"; - - /** - * CASE: nspin1|Cartesian|no vel|no mag|no orb|no dpks_desc|rank0 - * - */ - unitcell::print_stru_file(*ucell,ucell->atoms,ucell->latvec, - fn, "", 1, false, false, false, false, false, 0); - std::ifstream ifs; - ifs.open("C1H2_STRU"); - std::string str((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); - EXPECT_THAT(str, testing::HasSubstr("ATOMIC_SPECIES")); - EXPECT_THAT(str, testing::HasSubstr("C 12.0000 C.upf upf201")); - EXPECT_THAT(str, testing::HasSubstr("H 1.0000 H.upf upf201")); - EXPECT_THAT(str, testing::HasSubstr("LATTICE_CONSTANT")); - EXPECT_THAT(str, testing::HasSubstr("1.8897261255")); - EXPECT_THAT(str, testing::HasSubstr("LATTICE_VECTORS")); - EXPECT_THAT(str, testing::HasSubstr("10.0000000000000000 0.0000000000000000 0.0000000000000000")); - EXPECT_THAT(str, testing::HasSubstr("0.0000000000000000 10.0000000000000000 0.0000000000000000")); - EXPECT_THAT(str, testing::HasSubstr("0.0000000000000000 0.0000000000000000 10.0000000000000000")); - EXPECT_THAT(str, testing::HasSubstr("ATOMIC_POSITIONS")); - EXPECT_THAT(str, testing::HasSubstr("Cartesian")); - EXPECT_THAT(str, testing::HasSubstr("C #label")); - EXPECT_THAT(str, testing::HasSubstr("0.0000 #magnetism")); - EXPECT_THAT(str, testing::HasSubstr("1 #number of atoms")); - EXPECT_THAT(str, testing::HasSubstr(" 1.0000000000 1.0000000000 1.0000000000 m 1 1 1")); - EXPECT_THAT(str, testing::HasSubstr("H #label")); - EXPECT_THAT(str, testing::HasSubstr("0.0000 #magnetism")); - EXPECT_THAT(str, testing::HasSubstr("2 #number of atoms")); - EXPECT_THAT(str, testing::HasSubstr(" 1.5000000000 1.5000000000 1.5000000000 m 0 0 0")); - EXPECT_THAT(str, testing::HasSubstr(" 0.5000000000 0.5000000000 0.5000000000 m 0 0 1")); - str.clear(); - ifs.close(); - remove("C1H2_STRU"); - /** - * CASE: nspin2|Direct|vel|no mag|no orb|no dpks_desc|rank0 - * - */ - unitcell::print_stru_file(*ucell,ucell->atoms,ucell->latvec, - fn, "", 2, true, true, false, false, false, 0); - ifs.open("C1H2_STRU"); - str = {(std::istreambuf_iterator(ifs)), std::istreambuf_iterator()}; - EXPECT_THAT(str, testing::HasSubstr("ATOMIC_SPECIES")); - EXPECT_THAT(str, testing::HasSubstr("C 12.0000 C.upf upf201")); - EXPECT_THAT(str, testing::HasSubstr("H 1.0000 H.upf upf201")); - EXPECT_THAT(str, testing::HasSubstr("LATTICE_CONSTANT")); - EXPECT_THAT(str, testing::HasSubstr("1.8897261255")); - EXPECT_THAT(str, testing::HasSubstr("LATTICE_VECTORS")); - EXPECT_THAT(str, testing::HasSubstr("10.0000000000000000 0.0000000000000000 0.0000000000000000")); - EXPECT_THAT(str, testing::HasSubstr("0.0000000000000000 10.0000000000000000 0.0000000000000000")); - EXPECT_THAT(str, testing::HasSubstr("0.0000000000000000 0.0000000000000000 10.0000000000000000")); - EXPECT_THAT(str, testing::HasSubstr("ATOMIC_POSITIONS")); - EXPECT_THAT(str, testing::HasSubstr("Direct")); - EXPECT_THAT(str, testing::HasSubstr("C #label")); - EXPECT_THAT(str, testing::HasSubstr("0.0000 #magnetism")); - EXPECT_THAT(str, testing::HasSubstr("1 #number of atoms")); - EXPECT_THAT(str, - testing::HasSubstr(" 0.1000000000 0.1000000000 0.1000000000 m 1 1 1 v " - "0.1000000000 0.1000000000 0.1000000000")); - EXPECT_THAT(str, testing::HasSubstr("H #label")); - EXPECT_THAT(str, testing::HasSubstr("0.0000 #magnetism")); - EXPECT_THAT(str, testing::HasSubstr("2 #number of atoms")); - EXPECT_THAT(str, - testing::HasSubstr(" 0.1500000000 0.1500000000 0.1500000000 m 0 0 0 v " - "0.1000000000 0.1000000000 0.1000000000")); - EXPECT_THAT(str, - testing::HasSubstr(" 0.0500000000 0.0500000000 0.0500000000 m 0 0 1 v " - "0.1000000000 0.1000000000 0.1000000000")); - str.clear(); - ifs.close(); - remove("C1H2_STRU"); - /** - * CASE: nspin2|Direct|no vel|mag|orb|dpks_desc|rank0 - * - */ - ucell->descriptor_file = "__unittest_numerical_descriptor__"; - ucell->orbital_fn[0] = "__unittest_orbital_fn_0__"; - ucell->orbital_fn[1] = "__unittest_orbital_fn_1__"; - ucell->atom_mulliken - = {{-1, 0.5}, {-1, 0.4}, {-1, 0.3}}; // first index is iat, the second is components, starts seems from 1 - unitcell::print_stru_file(*ucell,ucell->atoms,ucell->latvec, - fn, "", 2, true, false, true, true, true, 0); - ifs.open("C1H2_STRU"); - str = {(std::istreambuf_iterator(ifs)), std::istreambuf_iterator()}; - EXPECT_THAT(str, testing::HasSubstr("ATOMIC_SPECIES")); - EXPECT_THAT(str, testing::HasSubstr("C 12.0000 C.upf upf201")); - EXPECT_THAT(str, testing::HasSubstr("H 1.0000 H.upf upf201")); - EXPECT_THAT(str, testing::HasSubstr("NUMERICAL_ORBITAL")); - EXPECT_THAT(str, testing::HasSubstr("__unittest_orbital_fn_0__")); - EXPECT_THAT(str, testing::HasSubstr("__unittest_orbital_fn_1__")); - EXPECT_THAT(str, testing::HasSubstr("NUMERICAL_DESCRIPTOR")); - EXPECT_THAT(str, testing::HasSubstr("__unittest_numerical_descriptor__")); - EXPECT_THAT(str, testing::HasSubstr("LATTICE_CONSTANT")); - EXPECT_THAT(str, testing::HasSubstr("1.8897261255")); - EXPECT_THAT(str, testing::HasSubstr("LATTICE_VECTORS")); - EXPECT_THAT(str, testing::HasSubstr("10.0000000000000000 0.0000000000000000 0.0000000000000000")); - EXPECT_THAT(str, testing::HasSubstr("0.0000000000000000 10.0000000000000000 0.0000000000000000")); - EXPECT_THAT(str, testing::HasSubstr("0.0000000000000000 0.0000000000000000 10.0000000000000000")); - EXPECT_THAT(str, testing::HasSubstr("ATOMIC_POSITIONS")); - EXPECT_THAT(str, testing::HasSubstr("Direct")); - EXPECT_THAT(str, testing::HasSubstr("C #label")); - EXPECT_THAT(str, testing::HasSubstr("0.0000 #magnetism")); - EXPECT_THAT(str, testing::HasSubstr("1 #number of atoms")); - EXPECT_THAT(str, - testing::HasSubstr(" 0.1000000000 0.1000000000 0.1000000000 m 1 1 1 mag 0.5000")); - EXPECT_THAT(str, testing::HasSubstr("H #label")); - EXPECT_THAT(str, testing::HasSubstr("0.0000 #magnetism")); - EXPECT_THAT(str, testing::HasSubstr("2 #number of atoms")); - EXPECT_THAT(str, - testing::HasSubstr(" 0.1500000000 0.1500000000 0.1500000000 m 0 0 0 mag 0.4000")); - EXPECT_THAT(str, - testing::HasSubstr(" 0.0500000000 0.0500000000 0.0500000000 m 0 0 1 mag 0.3000")); - ifs.close(); - remove("C1H2_STRU"); -} - -TEST_F(UcellTest, PrintTauDirect) -{ - UcellTestPrepare utp = UcellTestLib["C1H2-Index"]; - ucell = utp.SetUcellInfo(); - EXPECT_EQ(ucell->Coordinate, "Direct"); - - // open a file - std::ofstream ofs("print_tau_direct"); - unitcell::print_tau(ucell->atoms,ucell->Coordinate,ucell->ntype,ucell->lat0,ofs); - ofs.close(); - - // readin the data - std::ifstream ifs; - ifs.open("print_tau_direct"); - std::string str((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); - EXPECT_THAT(str, testing::HasSubstr("DIRECT COORDINATES")); - EXPECT_THAT(str, testing::HasSubstr(" C 0.100000000000 0.100000000000 0.100000000000 0.0000")); - EXPECT_THAT(str, testing::HasSubstr(" H 0.150000000000 0.150000000000 0.150000000000 0.0000")); - ifs.close(); - - remove("print_tau_direct"); -} - -TEST_F(UcellTest, PrintTauCartesian) -{ - UcellTestPrepare utp = UcellTestLib["C1H2-Cartesian"]; - ucell = utp.SetUcellInfo(); - EXPECT_EQ(ucell->Coordinate, "Cartesian"); - - // open a file - std::ofstream ofs("print_tau_Cartesian"); - unitcell::print_tau(ucell->atoms,ucell->Coordinate,ucell->ntype,ucell->lat0,ofs); - ofs.close(); - - // readin the data - std::ifstream ifs; - ifs.open("print_tau_Cartesian"); - std::string str((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); - EXPECT_THAT(str, testing::HasSubstr("CARTESIAN COORDINATES")); - EXPECT_THAT(str, testing::HasSubstr(" C 1.000000000000 1.000000000000 1.000000000000 0.0000")); - EXPECT_THAT(str, testing::HasSubstr(" H 1.500000000000 1.500000000000 1.500000000000 0.0000")); - ifs.close(); - - // remove the file - remove("print_tau_Cartesian"); -} - -TEST_F(UcellTest, UpdateVel) -{ - UcellTestPrepare utp = UcellTestLib["C1H2-Index"]; - ucell = utp.SetUcellInfo(); - ModuleBase::Vector3* vel_in = new ModuleBase::Vector3[ucell->nat]; - for (int iat = 0; iat < ucell->nat; ++iat) - { - vel_in[iat].set(iat * 0.1, iat * 0.1, iat * 0.1); - } - unitcell::update_vel(vel_in,ucell->ntype,ucell->nat,ucell->atoms); - for (int iat = 0; iat < ucell->nat; ++iat) - { - EXPECT_DOUBLE_EQ(vel_in[iat].x, 0.1 * iat); - EXPECT_DOUBLE_EQ(vel_in[iat].y, 0.1 * iat); - EXPECT_DOUBLE_EQ(vel_in[iat].z, 0.1 * iat); - } - delete[] vel_in; -} - -TEST_F(UcellTest, CalUx1) -{ - UcellTestPrepare utp = UcellTestLib["C1H2-Read"]; - ucell = utp.SetUcellInfo(); - ucell->atoms[0].m_loc_[0].set(0, -1, 0); - ucell->atoms[1].m_loc_[0].set(1, 1, 1); - ucell->atoms[1].m_loc_[1].set(0, 0, 0); - const int nspin = 4; - unitcell::cal_ux(*ucell, nspin); - EXPECT_FALSE(ucell->magnet.lsign_); - EXPECT_DOUBLE_EQ(ucell->magnet.ux_[0], 0); - EXPECT_DOUBLE_EQ(ucell->magnet.ux_[1], -1); - EXPECT_DOUBLE_EQ(ucell->magnet.ux_[2], 0); -} - -TEST_F(UcellTest, CalUx2) -{ - UcellTestPrepare utp = UcellTestLib["C1H2-Read"]; - ucell = utp.SetUcellInfo(); - ucell->atoms[0].m_loc_[0].set(0, 0, 0); - ucell->atoms[1].m_loc_[0].set(1, 1, 1); - ucell->atoms[1].m_loc_[1].set(0, 0, 0); - //(0,0,0) is also parallel to (1,1,1) - const int nspin = 4; - unitcell::cal_ux(*ucell, nspin); - EXPECT_TRUE(ucell->magnet.lsign_); - EXPECT_NEAR(ucell->magnet.ux_[0], 0.57735, 1e-5); - EXPECT_NEAR(ucell->magnet.ux_[1], 0.57735, 1e-5); - EXPECT_NEAR(ucell->magnet.ux_[2], 0.57735, 1e-5); -} - -#ifdef __LCAO -TEST_F(UcellTest, ReadOrbFile) -{ - UcellTestPrepare utp = UcellTestLib["C1H2-Read"]; - ucell = utp.SetUcellInfo(); - std::string orb_file = "./support/C.orb"; - std::ofstream ofs_running; - ofs_running.open("tmp_readorbfile"); - bool result = unitcell::read_orb_file(0, orb_file, ofs_running, &(ucell->atoms[0])); - ofs_running << " result=" << result << std::endl; - EXPECT_TRUE(result); - ofs_running.close(); - EXPECT_EQ(ucell->atoms[0].nw, 25); - remove("tmp_readorbfile"); -} - -class UcellTestReadStru : public ::testing::Test -{ - protected: - std::unique_ptr ucell{new UnitCell}; - std::string output; - void SetUp() override - { - ucell->ntype = 2; - ucell->pseudo_fn.resize(ucell->ntype); - ucell->pseudo_type.resize(ucell->ntype); - ucell->orbital_fn.resize(ucell->ntype); - } - void TearDown() override - { - ucell->orbital_fn.shrink_to_fit(); - } -}; - -TEST_F(UcellTestReadStru, ReadAtomSpecies) -{ - std::string fn = "./support/STRU_MgO"; - std::ifstream ifa(fn.c_str()); - std::ofstream ofs_running; - ofs_running.open("read_atom_species.tmp"); - ucell->ntype = 2; - ucell->atoms = new Atom[ucell->ntype]; - ucell->set_atom_flag = true; - const std::string basis_type = "lcao"; - const std::string orbital_dir = ""; - const std::string init_wfc = ""; - const double onsite_radius = 0.0; - const bool deepks_setorb = true; - const bool rpa = false; - EXPECT_NO_THROW(unitcell::read_atom_species(ifa, ofs_running, *ucell, - basis_type, orbital_dir, init_wfc, - onsite_radius, deepks_setorb, rpa)); - EXPECT_NO_THROW(unitcell::read_lattice_constant(ifa, ofs_running, ucell->lat)); - EXPECT_DOUBLE_EQ(ucell->latvec.e11, 4.27957); - EXPECT_DOUBLE_EQ(ucell->latvec.e22, 4.27957); - EXPECT_DOUBLE_EQ(ucell->latvec.e33, 4.27957); - ofs_running.close(); - ifa.close(); - remove("read_atom_species.tmp"); -} - -TEST_F(UcellTestReadStru, ReadAtomSpeciesWarning1) -{ - std::string fn = "./support/STRU_MgO_Warning1"; - std::ifstream ifa(fn.c_str()); - std::ofstream ofs_running; - ofs_running.open("read_atom_species.txt"); - ucell->ntype = 2; - ucell->atoms = new Atom[ucell->ntype]; - ucell->set_atom_flag = true; - const std::string basis_type = "lcao"; - const std::string orbital_dir = ""; - const std::string init_wfc = ""; - const double onsite_radius = 0.0; - const bool deepks_setorb = true; - const bool rpa = false; - testing::internal::CaptureStdout(); - EXPECT_EXIT(unitcell::read_atom_species(ifa, ofs_running, *ucell, - basis_type, orbital_dir, init_wfc, - onsite_radius, deepks_setorb, rpa), ::testing::ExitedWithCode(1), ""); - output = testing::internal::GetCapturedStdout(); - EXPECT_THAT(output, testing::HasSubstr("unrecognized pseudopotential type.")); - ofs_running.close(); - ifa.close(); - //remove("read_atom_species.txt"); -} - -TEST_F(UcellTestReadStru, ReadLatticeConstantWarning1) -{ - std::string fn = "./support/STRU_MgO_Warning2"; - std::ifstream ifa(fn.c_str()); - std::ofstream ofs_running; - ofs_running.open("read_atom_species1.tmp"); - ucell->ntype = 2; - ucell->atoms = new Atom[ucell->ntype]; - ucell->set_atom_flag = true; - testing::internal::CaptureStdout(); - EXPECT_EXIT(unitcell::read_lattice_constant(ifa, ofs_running,ucell->lat), ::testing::ExitedWithCode(1), ""); - output = testing::internal::GetCapturedStdout(); - EXPECT_THAT(output, testing::HasSubstr("Lattice constant <= 0.0")); - ofs_running.close(); - ifa.close(); - remove("read_atom_species1.tmp"); -} - -TEST_F(UcellTestReadStru, ReadLatticeConstantWarning2) -{ - std::string fn = "./support/STRU_MgO_Warning3"; - std::ifstream ifa(fn.c_str()); - std::ofstream ofs_running; - ofs_running.open("read_atom_species.tmp"); - ucell->ntype = 2; - ucell->atoms = new Atom[ucell->ntype]; - ucell->set_atom_flag = true; - testing::internal::CaptureStdout(); - EXPECT_EXIT(unitcell::read_lattice_constant(ifa, ofs_running,ucell->lat), ::testing::ExitedWithCode(1), ""); - output = testing::internal::GetCapturedStdout(); - EXPECT_THAT(output, - testing::HasSubstr("do not use LATTICE_PARAMETERS without explicit specification of lattice type")); - ofs_running.close(); - ifa.close(); - remove("read_atom_species.tmp"); -} - -TEST_F(UcellTestReadStru, ReadLatticeConstantWarning3) -{ - std::string fn = "./support/STRU_MgO_Warning4"; - std::ifstream ifa(fn.c_str()); - std::ofstream ofs_running; - ofs_running.open("read_atom_species.tmp"); - ucell->ntype = 2; - ucell->atoms = new Atom[ucell->ntype]; - ucell->set_atom_flag = true; - ucell->latName = "bcc"; - testing::internal::CaptureStdout(); - EXPECT_EXIT(unitcell::read_lattice_constant(ifa, ofs_running,ucell->lat), ::testing::ExitedWithCode(1), ""); - output = testing::internal::GetCapturedStdout(); - EXPECT_THAT(output, - testing::HasSubstr("do not use LATTICE_VECTORS along with explicit specification of lattice type")); - ofs_running.close(); - ifa.close(); - remove("read_atom_species.tmp"); -} - -TEST_F(UcellTestReadStru, ReadAtomSpeciesLatName) -{ - ucell->ntype = 2; - ucell->atoms = new Atom[ucell->ntype]; - ucell->set_atom_flag = true; - std::vector latName_in = {"sc", - "fcc", - "bcc", - "hexagonal", - "trigonal", - "st", - "bct", - "so", - "baco", - "fco", - "bco", - "sm", - "bacm", - "triclinic"}; - for (int i = 0; i < latName_in.size(); ++i) - { - std::string fn = "./support/STRU_MgO_LatName"; - std::ifstream ifa(fn.c_str()); - std::ofstream ofs_running; - ofs_running.open("read_atom_species.tmp"); - ucell->latName = latName_in[i]; - EXPECT_NO_THROW(unitcell::read_lattice_constant(ifa, ofs_running,ucell->lat)); - if (ucell->latName == "sc") - { - EXPECT_DOUBLE_EQ(ucell->latvec.e11, 1.0); - EXPECT_DOUBLE_EQ(ucell->latvec.e22, 1.0); - EXPECT_DOUBLE_EQ(ucell->latvec.e33, 1.0); - } - ofs_running.close(); - ifa.close(); - remove("read_atom_species.tmp"); - } -} - -TEST_F(UcellDeathTest, ReadAtomSpeciesWarning5) -{ - std::string fn = "./support/STRU_MgO_LatName"; - std::ifstream ifa(fn.c_str()); - std::ofstream ofs_running; - ofs_running.open("read_atom_species.tmp"); - ucell->ntype = 2; - ucell->atoms = new Atom[ucell->ntype]; - ucell->set_atom_flag = true; - ucell->latName = "arbitrary"; - testing::internal::CaptureStdout(); - EXPECT_EXIT(unitcell::read_lattice_constant(ifa, ofs_running,ucell->lat), ::testing::ExitedWithCode(1), ""); - output = testing::internal::GetCapturedStdout(); - EXPECT_THAT(output, testing::HasSubstr("latname not supported")); - ofs_running.close(); - ifa.close(); - remove("read_atom_species.tmp"); -} - -TEST_F(UcellTestReadStru, ReadAtomPositionsS1) -{ - std::string fn = "./support/STRU_MgO"; - std::ifstream ifa(fn.c_str()); - std::ofstream ofs_running; - std::ofstream ofs_warning; - ofs_running.open("read_atom_positions.tmp"); - ofs_warning.open("read_atom_positions.warn"); - // mandatory preliminaries - ucell->ntype = 2; - ucell->atoms = new Atom[ucell->ntype]; - ucell->set_atom_flag = true; - const std::string basis_type = "lcao"; - const std::string orbital_dir = ""; - const std::string init_wfc = ""; - const double onsite_radius = 0.0; - const bool deepks_setorb = true; - const bool rpa = false; - const int nspin = 1; - const bool fixed_atoms = false; - const bool noncolin = false; - const std::string calculation = "scf"; - const std::string esolver_type = "ksdft"; - EXPECT_NO_THROW(unitcell::read_atom_species(ifa, ofs_running, *ucell, - basis_type, orbital_dir, init_wfc, - onsite_radius, deepks_setorb, rpa)); - EXPECT_NO_THROW(unitcell::read_lattice_constant(ifa, ofs_running,ucell->lat)); - EXPECT_DOUBLE_EQ(ucell->latvec.e11, 4.27957); - EXPECT_DOUBLE_EQ(ucell->latvec.e22, 4.27957); - EXPECT_DOUBLE_EQ(ucell->latvec.e33, 4.27957); - unitcell::read_atom_positions(*ucell, ifa, ofs_running, ofs_warning, nspin, - basis_type, orbital_dir, init_wfc, - onsite_radius, fixed_atoms, noncolin, - calculation, esolver_type, 0); - ofs_running.close(); - ofs_warning.close(); - ifa.close(); - remove("read_atom_positions.tmp"); - remove("read_atom_positions.warn"); -} - -TEST_F(UcellTestReadStru, ReadAtomPositionsS2) -{ - std::string fn = "./support/STRU_MgO"; - std::ifstream ifa(fn.c_str()); - std::ofstream ofs_running; - std::ofstream ofs_warning; - ofs_running.open("read_atom_positions.tmp"); - ofs_warning.open("read_atom_positions.warn"); - // mandatory preliminaries - ucell->ntype = 2; - ucell->atoms = new Atom[ucell->ntype]; - ucell->set_atom_flag = true; - const std::string basis_type = "lcao"; - const std::string orbital_dir = ""; - const std::string init_wfc = ""; - const double onsite_radius = 0.0; - const bool deepks_setorb = true; - const bool rpa = false; - const int nspin = 2; - const bool fixed_atoms = false; - const bool noncolin = false; - const std::string calculation = "scf"; - const std::string esolver_type = "ksdft"; - EXPECT_NO_THROW(unitcell::read_atom_species(ifa, ofs_running, *ucell, - basis_type, orbital_dir, init_wfc, - onsite_radius, deepks_setorb, rpa)); - EXPECT_NO_THROW(unitcell::read_lattice_constant(ifa, ofs_running,ucell->lat)); - EXPECT_DOUBLE_EQ(ucell->latvec.e11, 4.27957); - EXPECT_DOUBLE_EQ(ucell->latvec.e22, 4.27957); - EXPECT_DOUBLE_EQ(ucell->latvec.e33, 4.27957); - unitcell::read_atom_positions(*ucell, ifa, ofs_running, ofs_warning, nspin, - basis_type, orbital_dir, init_wfc, - onsite_radius, fixed_atoms, noncolin, - calculation, esolver_type, 0); - ofs_running.close(); - ofs_warning.close(); - ifa.close(); - remove("read_atom_positions.tmp"); - remove("read_atom_positions.warn"); -} - -TEST_F(UcellTestReadStru, ReadAtomPositionsS4Noncolin) -{ - std::string fn = "./support/STRU_MgO"; - std::ifstream ifa(fn.c_str()); - std::ofstream ofs_running; - std::ofstream ofs_warning; - ofs_running.open("read_atom_positions.tmp"); - ofs_warning.open("read_atom_positions.warn"); - // mandatory preliminaries - ucell->ntype = 2; - ucell->atoms = new Atom[ucell->ntype]; - ucell->set_atom_flag = true; - const std::string basis_type = "lcao"; - const std::string orbital_dir = ""; - const std::string init_wfc = ""; - const double onsite_radius = 0.0; - const bool deepks_setorb = true; - const bool rpa = false; - const int nspin = 4; - const bool fixed_atoms = false; - const bool noncolin = true; - const std::string calculation = "scf"; - const std::string esolver_type = "ksdft"; - EXPECT_NO_THROW(unitcell::read_atom_species(ifa, ofs_running, *ucell, - basis_type, orbital_dir, init_wfc, - onsite_radius, deepks_setorb, rpa)); - EXPECT_NO_THROW(unitcell::read_lattice_constant(ifa, ofs_running,ucell->lat)); - EXPECT_DOUBLE_EQ(ucell->latvec.e11, 4.27957); - EXPECT_DOUBLE_EQ(ucell->latvec.e22, 4.27957); - EXPECT_DOUBLE_EQ(ucell->latvec.e33, 4.27957); - unitcell::read_atom_positions(*ucell, ifa, ofs_running, ofs_warning, nspin, - basis_type, orbital_dir, init_wfc, - onsite_radius, fixed_atoms, noncolin, - calculation, esolver_type, 0); - ofs_running.close(); - ofs_warning.close(); - ifa.close(); - remove("read_atom_positions.tmp"); - remove("read_atom_positions.warn"); -} - -TEST_F(UcellTestReadStru, ReadAtomPositionsS4Colin) -{ - std::string fn = "./support/STRU_MgO"; - std::ifstream ifa(fn.c_str()); - std::ofstream ofs_running; - std::ofstream ofs_warning; - ofs_running.open("read_atom_positions.tmp"); - ofs_warning.open("read_atom_positions.warn"); - // mandatory preliminaries - ucell->ntype = 2; - ucell->atoms = new Atom[ucell->ntype]; - ucell->set_atom_flag = true; - const std::string basis_type = "lcao"; - const std::string orbital_dir = ""; - const std::string init_wfc = ""; - const double onsite_radius = 0.0; - const bool deepks_setorb = true; - const bool rpa = false; - const int nspin = 4; - const bool fixed_atoms = false; - const bool noncolin = false; - const std::string calculation = "scf"; - const std::string esolver_type = "ksdft"; - EXPECT_NO_THROW(unitcell::read_atom_species(ifa, ofs_running, *ucell, - basis_type, orbital_dir, init_wfc, - onsite_radius, deepks_setorb, rpa)); - EXPECT_NO_THROW(unitcell::read_lattice_constant(ifa, ofs_running,ucell->lat)); - EXPECT_DOUBLE_EQ(ucell->latvec.e11, 4.27957); - EXPECT_DOUBLE_EQ(ucell->latvec.e22, 4.27957); - EXPECT_DOUBLE_EQ(ucell->latvec.e33, 4.27957); - unitcell::read_atom_positions(*ucell, ifa, ofs_running, ofs_warning, nspin, - basis_type, orbital_dir, init_wfc, - onsite_radius, fixed_atoms, noncolin, - calculation, esolver_type, 0); - ofs_running.close(); - ofs_warning.close(); - ifa.close(); - remove("read_atom_positions.tmp"); - remove("read_atom_positions.warn"); -} - -TEST_F(UcellTestReadStru, ReadAtomPositionsC) -{ - std::string fn = "./support/STRU_MgO_c"; - std::ifstream ifa(fn.c_str()); - std::ofstream ofs_running; - std::ofstream ofs_warning; - ofs_running.open("read_atom_positions.tmp"); - ofs_warning.open("read_atom_positions.warn"); - // mandatory preliminaries - ucell->ntype = 2; - ucell->atoms = new Atom[ucell->ntype]; - ucell->set_atom_flag = true; - const std::string basis_type = "lcao"; - const std::string orbital_dir = ""; - const std::string init_wfc = ""; - const double onsite_radius = 0.0; - const bool deepks_setorb = true; - const bool rpa = false; - const int nspin = 1; - const bool fixed_atoms = false; - const bool noncolin = false; - const std::string calculation = "scf"; - const std::string esolver_type = "ksdft"; - EXPECT_NO_THROW(unitcell::read_atom_species(ifa, ofs_running, *ucell, - basis_type, orbital_dir, init_wfc, - onsite_radius, deepks_setorb, rpa)); - EXPECT_NO_THROW(unitcell::read_lattice_constant(ifa, ofs_running,ucell->lat)); - EXPECT_DOUBLE_EQ(ucell->latvec.e11, 4.27957); - EXPECT_DOUBLE_EQ(ucell->latvec.e22, 4.27957); - EXPECT_DOUBLE_EQ(ucell->latvec.e33, 4.27957); - unitcell::read_atom_positions(*ucell, ifa, ofs_running, ofs_warning, nspin, - basis_type, orbital_dir, init_wfc, - onsite_radius, fixed_atoms, noncolin, - calculation, esolver_type, 0); - ofs_running.close(); - ofs_warning.close(); - ifa.close(); - remove("read_atom_positions.tmp"); - remove("read_atom_positions.warn"); -} - -TEST_F(UcellTestReadStru, ReadAtomPositionsCA) -{ - std::string fn = "./support/STRU_MgO_ca"; - std::ifstream ifa(fn.c_str()); - std::ofstream ofs_running; - std::ofstream ofs_warning; - ofs_running.open("read_atom_positions.tmp"); - ofs_warning.open("read_atom_positions.warn"); - // mandatory preliminaries - ucell->ntype = 2; - ucell->atoms = new Atom[ucell->ntype]; - ucell->set_atom_flag = true; - const std::string basis_type = "lcao"; - const std::string orbital_dir = ""; - const std::string init_wfc = ""; - const double onsite_radius = 0.0; - const bool deepks_setorb = true; - const bool rpa = false; - const int nspin = 1; - const bool fixed_atoms = false; - const bool noncolin = false; - const std::string calculation = "scf"; - const std::string esolver_type = "ksdft"; - EXPECT_NO_THROW(unitcell::read_atom_species(ifa, ofs_running, *ucell, - basis_type, orbital_dir, init_wfc, - onsite_radius, deepks_setorb, rpa)); - EXPECT_NO_THROW(unitcell::read_lattice_constant(ifa, ofs_running,ucell->lat)); - EXPECT_DOUBLE_EQ(ucell->latvec.e11, 4.27957); - EXPECT_DOUBLE_EQ(ucell->latvec.e22, 4.27957); - EXPECT_DOUBLE_EQ(ucell->latvec.e33, 4.27957); - unitcell::read_atom_positions(*ucell, ifa, ofs_running, ofs_warning, nspin, - basis_type, orbital_dir, init_wfc, - onsite_radius, fixed_atoms, noncolin, - calculation, esolver_type, 0); - ofs_running.close(); - ofs_warning.close(); - ifa.close(); - remove("read_atom_positions.tmp"); - remove("read_atom_positions.warn"); -} - -TEST_F(UcellTestReadStru, ReadAtomPositionsCACXY) -{ - std::string fn = "./support/STRU_MgO_cacxy"; - std::ifstream ifa(fn.c_str()); - std::ofstream ofs_running; - std::ofstream ofs_warning; - ofs_running.open("read_atom_positions.tmp"); - ofs_warning.open("read_atom_positions.warn"); - // mandatory preliminaries - ucell->ntype = 2; - ucell->atoms = new Atom[ucell->ntype]; - ucell->set_atom_flag = true; - const std::string basis_type = "lcao"; - const std::string orbital_dir = ""; - const std::string init_wfc = ""; - const double onsite_radius = 0.0; - const bool deepks_setorb = true; - const bool rpa = false; - const int nspin = 1; - const bool fixed_atoms = false; - const bool noncolin = false; - const std::string calculation = "scf"; - const std::string esolver_type = "ksdft"; - EXPECT_NO_THROW(unitcell::read_atom_species(ifa, ofs_running, *ucell, - basis_type, orbital_dir, init_wfc, - onsite_radius, deepks_setorb, rpa)); - EXPECT_NO_THROW(unitcell::read_lattice_constant(ifa, ofs_running,ucell->lat)); - EXPECT_DOUBLE_EQ(ucell->latvec.e11, 4.27957); - EXPECT_DOUBLE_EQ(ucell->latvec.e22, 4.27957); - EXPECT_DOUBLE_EQ(ucell->latvec.e33, 4.27957); - unitcell::read_atom_positions(*ucell, ifa, ofs_running, ofs_warning, nspin, - basis_type, orbital_dir, init_wfc, - onsite_radius, fixed_atoms, noncolin, - calculation, esolver_type, 0); - ofs_running.close(); - ofs_warning.close(); - ifa.close(); - remove("read_atom_positions.tmp"); - remove("read_atom_positions.warn"); -} - -TEST_F(UcellTestReadStru, ReadAtomPositionsCACXZ) -{ - std::string fn = "./support/STRU_MgO_cacxz"; - std::ifstream ifa(fn.c_str()); - std::ofstream ofs_running; - std::ofstream ofs_warning; - ofs_running.open("read_atom_positions.tmp"); - ofs_warning.open("read_atom_positions.warn"); - // mandatory preliminaries - ucell->ntype = 2; - ucell->atoms = new Atom[ucell->ntype]; - ucell->set_atom_flag = true; - const std::string basis_type = "lcao"; - const std::string orbital_dir = ""; - const std::string init_wfc = ""; - const double onsite_radius = 0.0; - const bool deepks_setorb = true; - const bool rpa = false; - const int nspin = 1; - const bool fixed_atoms = false; - const bool noncolin = false; - const std::string calculation = "scf"; - const std::string esolver_type = "ksdft"; - EXPECT_NO_THROW(unitcell::read_atom_species(ifa, ofs_running, *ucell, - basis_type, orbital_dir, init_wfc, - onsite_radius, deepks_setorb, rpa)); - EXPECT_NO_THROW(unitcell::read_lattice_constant(ifa, ofs_running,ucell->lat)); - EXPECT_DOUBLE_EQ(ucell->latvec.e11, 4.27957); - EXPECT_DOUBLE_EQ(ucell->latvec.e22, 4.27957); - EXPECT_DOUBLE_EQ(ucell->latvec.e33, 4.27957); - unitcell::read_atom_positions(*ucell, ifa, ofs_running, ofs_warning, nspin, - basis_type, orbital_dir, init_wfc, - onsite_radius, fixed_atoms, noncolin, - calculation, esolver_type, 0); - ofs_running.close(); - ofs_warning.close(); - ifa.close(); - remove("read_atom_positions.tmp"); - remove("read_atom_positions.warn"); -} - -TEST_F(UcellTestReadStru, ReadAtomPositionsCACYZ) -{ - std::string fn = "./support/STRU_MgO_cacyz"; - std::ifstream ifa(fn.c_str()); - std::ofstream ofs_running; - std::ofstream ofs_warning; - ofs_running.open("read_atom_positions.tmp"); - ofs_warning.open("read_atom_positions.warn"); - // mandatory preliminaries - ucell->ntype = 2; - ucell->atoms = new Atom[ucell->ntype]; - ucell->set_atom_flag = true; - const std::string basis_type = "lcao"; - const std::string orbital_dir = ""; - const std::string init_wfc = ""; - const double onsite_radius = 0.0; - const bool deepks_setorb = true; - const bool rpa = false; - const int nspin = 1; - const bool fixed_atoms = false; - const bool noncolin = false; - const std::string calculation = "scf"; - const std::string esolver_type = "ksdft"; - EXPECT_NO_THROW(unitcell::read_atom_species(ifa, ofs_running, *ucell, - basis_type, orbital_dir, init_wfc, - onsite_radius, deepks_setorb, rpa)); - EXPECT_NO_THROW(unitcell::read_lattice_constant(ifa, ofs_running,ucell->lat)); - EXPECT_DOUBLE_EQ(ucell->latvec.e11, 4.27957); - EXPECT_DOUBLE_EQ(ucell->latvec.e22, 4.27957); - EXPECT_DOUBLE_EQ(ucell->latvec.e33, 4.27957); - unitcell::read_atom_positions(*ucell, ifa, ofs_running, ofs_warning, nspin, - basis_type, orbital_dir, init_wfc, - onsite_radius, fixed_atoms, noncolin, - calculation, esolver_type, 0); - ofs_running.close(); - ofs_warning.close(); - ifa.close(); - remove("read_atom_positions.tmp"); - remove("read_atom_positions.warn"); -} - -TEST_F(UcellTestReadStru, ReadAtomPositionsCACXYZ) -{ - std::string fn = "./support/STRU_MgO_cacxyz"; - std::ifstream ifa(fn.c_str()); - std::ofstream ofs_running; - std::ofstream ofs_warning; - ofs_running.open("read_atom_positions.tmp"); - ofs_warning.open("read_atom_positions.warn"); - // mandatory preliminaries - ucell->ntype = 2; - ucell->atoms = new Atom[ucell->ntype]; - ucell->set_atom_flag = true; - const std::string basis_type = "lcao"; - const std::string orbital_dir = ""; - const std::string init_wfc = ""; - const double onsite_radius = 0.0; - const bool deepks_setorb = true; - const bool rpa = false; - const int nspin = 1; - const bool fixed_atoms = false; - const bool noncolin = false; - const std::string calculation = "scf"; - const std::string esolver_type = "ksdft"; - EXPECT_NO_THROW(unitcell::read_atom_species(ifa, ofs_running, *ucell, - basis_type, orbital_dir, init_wfc, - onsite_radius, deepks_setorb, rpa)); - EXPECT_NO_THROW(unitcell::read_lattice_constant(ifa, ofs_running,ucell->lat)); - EXPECT_DOUBLE_EQ(ucell->latvec.e11, 4.27957); - EXPECT_DOUBLE_EQ(ucell->latvec.e22, 4.27957); - EXPECT_DOUBLE_EQ(ucell->latvec.e33, 4.27957); - unitcell::read_atom_positions(*ucell, ifa, ofs_running, ofs_warning, nspin, - basis_type, orbital_dir, init_wfc, - onsite_radius, fixed_atoms, noncolin, - calculation, esolver_type, 0); - ofs_running.close(); - ofs_warning.close(); - ifa.close(); - remove("read_atom_positions.tmp"); - remove("read_atom_positions.warn"); -} - -TEST_F(UcellTestReadStru, ReadAtomPositionsCAU) -{ - std::string fn = "./support/STRU_MgO_cau"; - std::ifstream ifa(fn.c_str()); - std::ofstream ofs_running; - std::ofstream ofs_warning; - ofs_running.open("read_atom_positions.tmp"); - ofs_warning.open("read_atom_positions.warn"); - // mandatory preliminaries - ucell->ntype = 2; - ucell->atoms = new Atom[ucell->ntype]; - ucell->set_atom_flag = true; - const std::string basis_type = "lcao"; - const std::string orbital_dir = ""; - const std::string init_wfc = ""; - const double onsite_radius = 0.0; - const bool deepks_setorb = true; - const bool rpa = false; - const int nspin = 1; - const bool fixed_atoms = true; - const bool noncolin = false; - const std::string calculation = "scf"; - const std::string esolver_type = "ksdft"; - EXPECT_NO_THROW(unitcell::read_atom_species(ifa, ofs_running, *ucell, - basis_type, orbital_dir, init_wfc, - onsite_radius, deepks_setorb, rpa)); - EXPECT_NO_THROW(unitcell::read_lattice_constant(ifa, ofs_running,ucell->lat)); - EXPECT_DOUBLE_EQ(ucell->latvec.e11, 4.27957); - EXPECT_DOUBLE_EQ(ucell->latvec.e22, 4.27957); - EXPECT_DOUBLE_EQ(ucell->latvec.e33, 4.27957); - unitcell::read_atom_positions(*ucell, ifa, ofs_running, ofs_warning, nspin, - basis_type, orbital_dir, init_wfc, - onsite_radius, fixed_atoms, noncolin, - calculation, esolver_type, 0); - ofs_running.close(); - ofs_warning.close(); - ifa.close(); - remove("read_atom_positions.tmp"); - remove("read_atom_positions.warn"); -} - -TEST_F(UcellTestReadStru, ReadAtomPositionsAutosetMag) -{ - std::string fn = "./support/STRU_MgO"; - std::ifstream ifa(fn.c_str()); - std::ofstream ofs_running; - std::ofstream ofs_warning; - ofs_running.open("read_atom_positions.tmp"); - ofs_warning.open("read_atom_positions.warn"); - // mandatory preliminaries - ucell->ntype = 2; - ucell->atoms = new Atom[ucell->ntype]; - ucell->set_atom_flag = true; - const std::string basis_type = "lcao"; - const std::string orbital_dir = ""; - const std::string init_wfc = ""; - const double onsite_radius = 0.0; - const bool deepks_setorb = true; - const bool rpa = false; - const bool fixed_atoms = false; - const bool noncolin = false; - const std::string calculation = "scf"; - const std::string esolver_type = "ksdft"; - int nspin = 2; - EXPECT_NO_THROW(unitcell::read_atom_species(ifa, ofs_running, *ucell, - basis_type, orbital_dir, init_wfc, - onsite_radius, deepks_setorb, rpa)); - EXPECT_NO_THROW(unitcell::read_lattice_constant(ifa, ofs_running,ucell->lat)); - EXPECT_DOUBLE_EQ(ucell->latvec.e11, 4.27957); - EXPECT_DOUBLE_EQ(ucell->latvec.e22, 4.27957); - EXPECT_DOUBLE_EQ(ucell->latvec.e33, 4.27957); - unitcell::read_atom_positions(*ucell, ifa, ofs_running, ofs_warning, nspin, - basis_type, orbital_dir, init_wfc, - onsite_radius, fixed_atoms, noncolin, - calculation, esolver_type, 0); - for (int it = 0; it < ucell->ntype; it++) - { - for (int ia = 0; ia < ucell->atoms[it].na; ia++) - { - EXPECT_DOUBLE_EQ(ucell->atoms[it].mag[ia], 1.0); - EXPECT_DOUBLE_EQ(ucell->atoms[it].m_loc_[ia].z, 1.0); - } - } - // for nspin == 4 - nspin = 4; - unitcell::read_atom_positions(*ucell, ifa, ofs_running, ofs_warning, nspin, - basis_type, orbital_dir, init_wfc, - onsite_radius, fixed_atoms, noncolin, - calculation, esolver_type, 0); - for (int it = 0; it < ucell->ntype; it++) - { - for (int ia = 0; ia < ucell->atoms[it].na; ia++) - { - EXPECT_DOUBLE_EQ(ucell->atoms[it].mag[ia], sqrt(pow(1.0, 2) + pow(1.0, 2) + pow(1.0, 2))); - EXPECT_DOUBLE_EQ(ucell->atoms[it].m_loc_[ia].x, 1.0); - EXPECT_DOUBLE_EQ(ucell->atoms[it].m_loc_[ia].y, 1.0); - EXPECT_DOUBLE_EQ(ucell->atoms[it].m_loc_[ia].z, 1.0); - } - } - ofs_running.close(); - ofs_warning.close(); - ifa.close(); - remove("read_atom_positions.tmp"); - remove("read_atom_positions.warn"); -} - -TEST_F(UcellTestReadStru, ReadAtomPositionsWarning1) -{ - std::string fn = "./support/STRU_MgO_WarningC1"; - std::ifstream ifa(fn.c_str()); - std::ofstream ofs_running; - std::ofstream ofs_warning; - ofs_running.open("read_atom_positions.tmp"); - ofs_warning.open("read_atom_positions.warn"); - // mandatory preliminaries - ucell->ntype = 2; - ucell->atoms = new Atom[ucell->ntype]; - ucell->set_atom_flag = true; - const std::string basis_type = "lcao"; - const std::string orbital_dir = ""; - const std::string init_wfc = ""; - const double onsite_radius = 0.0; - const bool deepks_setorb = true; - const bool rpa = false; - const int nspin = 1; - const bool fixed_atoms = false; - const bool noncolin = false; - const std::string calculation = "scf"; - const std::string esolver_type = "ksdft"; - EXPECT_NO_THROW(unitcell::read_atom_species(ifa, ofs_running, *ucell, - basis_type, orbital_dir, init_wfc, - onsite_radius, deepks_setorb, rpa)); - EXPECT_NO_THROW(unitcell::read_lattice_constant(ifa, ofs_running,ucell->lat)); - EXPECT_DOUBLE_EQ(ucell->latvec.e11, 4.27957); - EXPECT_DOUBLE_EQ(ucell->latvec.e22, 4.27957); - EXPECT_DOUBLE_EQ(ucell->latvec.e33, 4.27957); - EXPECT_NO_THROW(unitcell::read_atom_positions(*ucell, ifa, ofs_running, ofs_warning, nspin, - basis_type, orbital_dir, init_wfc, - onsite_radius, fixed_atoms, noncolin, - calculation, esolver_type, 0)); - ofs_running.close(); - ofs_warning.close(); - ifa.close(); - // check warning file - std::ifstream ifs_tmp; - ifs_tmp.open("read_atom_positions.warn"); - std::string str((std::istreambuf_iterator(ifs_tmp)), std::istreambuf_iterator()); - EXPECT_THAT(str, testing::HasSubstr("There are several options for you:")); - EXPECT_THAT(str, testing::HasSubstr("Direct")); - EXPECT_THAT(str, testing::HasSubstr("Cartesian_angstrom")); - EXPECT_THAT(str, testing::HasSubstr("Cartesian_au")); - EXPECT_THAT(str, testing::HasSubstr("Cartesian_angstrom_center_xy")); - EXPECT_THAT(str, testing::HasSubstr("Cartesian_angstrom_center_xz")); - EXPECT_THAT(str, testing::HasSubstr("Cartesian_angstrom_center_yz")); - EXPECT_THAT(str, testing::HasSubstr("Cartesian_angstrom_center_xyz")); - ifs_tmp.close(); - remove("read_atom_positions.tmp"); - remove("read_atom_positions.warn"); -} - -TEST_F(UcellTestReadStru, ReadAtomPositionsWarning2) -{ - std::string fn = "./support/STRU_MgO_WarningC2"; - std::ifstream ifa(fn.c_str()); - std::ofstream ofs_running; - std::ofstream ofs_warning; - ofs_running.open("read_atom_positions.tmp"); - ofs_warning.open("read_atom_positions.warn"); - // mandatory preliminaries - ucell->ntype = 2; - ucell->atoms = new Atom[ucell->ntype]; - ucell->set_atom_flag = true; - const std::string basis_type = "lcao"; - const std::string orbital_dir = ""; - const std::string init_wfc = ""; - const double onsite_radius = 0.0; - const bool deepks_setorb = true; - const bool rpa = false; - const int nspin = 1; - const bool fixed_atoms = false; - const bool noncolin = false; - const std::string calculation = "scf"; - const std::string esolver_type = "ksdft"; - EXPECT_NO_THROW(unitcell::read_atom_species(ifa, ofs_running, *ucell, - basis_type, orbital_dir, init_wfc, - onsite_radius, deepks_setorb, rpa)); - EXPECT_NO_THROW(unitcell::read_lattice_constant(ifa, ofs_running,ucell->lat)); - EXPECT_DOUBLE_EQ(ucell->latvec.e11, 4.27957); - EXPECT_DOUBLE_EQ(ucell->latvec.e22, 4.27957); - EXPECT_DOUBLE_EQ(ucell->latvec.e33, 4.27957); - EXPECT_NO_THROW(unitcell::read_atom_positions(*ucell, ifa, ofs_running, ofs_warning, nspin, - basis_type, orbital_dir, init_wfc, - onsite_radius, fixed_atoms, noncolin, - calculation, esolver_type, 0)); - ofs_running.close(); - ofs_warning.close(); - ifa.close(); - // check warning file - std::ifstream ifs_tmp; - ifs_tmp.open("read_atom_positions.warn"); - std::string str((std::istreambuf_iterator(ifs_tmp)), std::istreambuf_iterator()); - EXPECT_THAT(str, testing::HasSubstr("Label read from ATOMIC_POSITIONS is Mo")); - EXPECT_THAT(str, testing::HasSubstr("Label from ATOMIC_SPECIES is Mg")); - ifs_tmp.close(); - remove("read_atom_positions.tmp"); - remove("read_atom_positions.warn"); -} - -TEST_F(UcellTestReadStru, ReadAtomPositionsWarning3) -{ - std::string fn = "./support/STRU_MgO_WarningC3"; - std::ifstream ifa(fn.c_str()); - std::ofstream ofs_running; - ofs_running.open("read_atom_positions.tmp"); - GlobalV::ofs_warning.open("read_atom_positions.warn"); - // mandatory preliminaries - ucell->ntype = 2; - ucell->atoms = new Atom[ucell->ntype]; - ucell->set_atom_flag = true; - const std::string basis_type = "lcao"; - const std::string orbital_dir = ""; - const std::string init_wfc = ""; - const double onsite_radius = 0.0; - const bool deepks_setorb = true; - const bool rpa = false; - const int nspin = 1; - const bool fixed_atoms = false; - const bool noncolin = false; - const std::string calculation = "scf"; - const std::string esolver_type = "ksdft"; - EXPECT_NO_THROW(unitcell::read_atom_species(ifa, ofs_running, *ucell, - basis_type, orbital_dir, init_wfc, - onsite_radius, deepks_setorb, rpa)); - EXPECT_NO_THROW(unitcell::read_lattice_constant(ifa, ofs_running,ucell->lat)); - EXPECT_DOUBLE_EQ(ucell->latvec.e11, 4.27957); - EXPECT_DOUBLE_EQ(ucell->latvec.e22, 4.27957); - EXPECT_DOUBLE_EQ(ucell->latvec.e33, 4.27957); - EXPECT_NO_THROW(unitcell::read_atom_positions(*ucell, ifa, ofs_running, GlobalV::ofs_warning, nspin, - basis_type, orbital_dir, init_wfc, - onsite_radius, fixed_atoms, noncolin, - calculation, esolver_type, 0)); - ofs_running.close(); - GlobalV::ofs_warning.close(); - ifa.close(); - // check warning file - std::ifstream ifs_tmp; - ifs_tmp.open("read_atom_positions.warn"); - std::string str((std::istreambuf_iterator(ifs_tmp)), std::istreambuf_iterator()); - EXPECT_THAT(str, testing::HasSubstr("read_atom_positions warning : atom number < 0.")); - ifs_tmp.close(); - remove("read_atom_positions.tmp"); - remove("read_atom_positions.warn"); -} - -TEST_F(UcellTestReadStru, ReadAtomPositionsWarning4) -{ - std::string fn = "./support/STRU_MgO_WarningC4"; - std::ifstream ifa(fn.c_str()); - std::ofstream ofs_running; - std::ofstream ofs_warning; - ofs_running.open("read_atom_positions.tmp"); - ofs_warning.open("read_atom_positions.warn"); - // mandatory preliminaries - ucell->ntype = 2; - ucell->atoms = new Atom[ucell->ntype]; - ucell->orbital_fn.resize(ucell->ntype); - ucell->set_atom_flag = true; - const std::string basis_type = "lcao"; - const std::string orbital_dir = ""; - const std::string init_wfc = ""; - const double onsite_radius = 0.0; - const bool deepks_setorb = true; - const bool rpa = false; - const int nspin = 1; - const bool fixed_atoms = false; - const bool noncolin = false; - const std::string calculation = "scf"; - const std::string esolver_type = "ksdft"; - EXPECT_NO_THROW(unitcell::read_atom_species(ifa, ofs_running, *ucell, - basis_type, orbital_dir, init_wfc, - onsite_radius, deepks_setorb, rpa)); - EXPECT_NO_THROW(unitcell::read_lattice_constant(ifa, ofs_running,ucell->lat)); - EXPECT_DOUBLE_EQ(ucell->latvec.e11, 4.27957); - EXPECT_DOUBLE_EQ(ucell->latvec.e22, 4.27957); - EXPECT_DOUBLE_EQ(ucell->latvec.e33, 4.27957); - testing::internal::CaptureStdout(); - EXPECT_EXIT(unitcell::read_atom_positions(*ucell, ifa, ofs_running, ofs_warning, nspin, - basis_type, orbital_dir, init_wfc, - onsite_radius, fixed_atoms, noncolin, - calculation, esolver_type, 0), ::testing::ExitedWithCode(1), ""); - output = testing::internal::GetCapturedStdout(); - EXPECT_THAT(output, testing::HasSubstr("read_atom_positions, mismatch in atom number for atom type: Mg")); - ofs_running.close(); - ofs_warning.close(); - ifa.close(); - remove("read_atom_positions.tmp"); - remove("read_atom_positions.warn"); -} - -TEST_F(UcellTestReadStru, ReadAtomPositionsWarning5) -{ - std::string fn = "./support/STRU_MgO"; - std::ifstream ifa(fn.c_str()); - std::ofstream ofs_running; - ofs_running.open("read_atom_positions.tmp"); - GlobalV::ofs_warning.open("read_atom_positions.warn"); - // mandatory preliminaries - ucell->ntype = 2; - ucell->atoms = new Atom[ucell->ntype]; - ucell->set_atom_flag = true; - const std::string basis_type = "lcao"; - const std::string orbital_dir = ""; - const std::string init_wfc = ""; - const double onsite_radius = 0.0; - const bool deepks_setorb = true; - const bool rpa = false; - const int nspin = 1; - const bool fixed_atoms = true; - const bool noncolin = false; - const std::string calculation = "md"; - const std::string esolver_type = "ksdft"; - EXPECT_NO_THROW(unitcell::read_atom_species(ifa, ofs_running, *ucell, - basis_type, orbital_dir, init_wfc, - onsite_radius, deepks_setorb, rpa)); - EXPECT_NO_THROW(unitcell::read_lattice_constant(ifa, ofs_running,ucell->lat)); - EXPECT_DOUBLE_EQ(ucell->latvec.e11, 4.27957); - EXPECT_DOUBLE_EQ(ucell->latvec.e22, 4.27957); - EXPECT_DOUBLE_EQ(ucell->latvec.e33, 4.27957); - EXPECT_NO_THROW(unitcell::read_atom_positions(*ucell, ifa, ofs_running, GlobalV::ofs_warning, nspin, - basis_type, orbital_dir, init_wfc, - onsite_radius, fixed_atoms, noncolin, - calculation, esolver_type, 0)); - ofs_running.close(); - GlobalV::ofs_warning.close(); - ifa.close(); - // check warning file - std::ifstream ifs_tmp; - ifs_tmp.open("read_atom_positions.warn"); - std::string str((std::istreambuf_iterator(ifs_tmp)), std::istreambuf_iterator()); - EXPECT_THAT(str, testing::HasSubstr("read_atoms warning : no atoms can move in MD simulations!")); - ifs_tmp.close(); - remove("read_atom_positions.tmp"); - remove("read_atom_positions.warn"); -} -#endif -TEST_F(UcellTest, ReadOrbFileWarning) -{ - UcellTestPrepare utp = UcellTestLib["C1H2-Read"]; - ucell = utp.SetUcellInfo(); - std::string orb_file = "./support/CC.orb"; - std::ofstream ofs_running; - ofs_running.open("tmp_readorbfilewarning"); - testing::internal::CaptureStdout(); - bool result = unitcell::read_orb_file(0, orb_file, ofs_running, &(ucell->atoms[0])); - output = testing::internal::GetCapturedStdout(); - ofs_running << output << std::endl; - EXPECT_FALSE(result); - EXPECT_THAT(output, testing::HasSubstr("Element index 1")); - EXPECT_THAT(output, testing::HasSubstr("orbital file: ./support/CC.orb")); - ofs_running.close(); - remove("tmp_readorbfilewarning"); -} diff --git a/source/source_cell/test/unitcell_test_readpp.cpp b/source/source_cell/test/unitcell_test_readpp.cpp index 7c53673e045..abd7bb41aa3 100644 --- a/source/source_cell/test/unitcell_test_readpp.cpp +++ b/source/source_cell/test/unitcell_test_readpp.cpp @@ -7,6 +7,7 @@ #include "source_cell/unitcell.h" #include "source_cell/cal_nelec_nband.h" #include "source_cell/read_pp_ucell.h" +#include #include #include #include "string.h" @@ -83,6 +84,10 @@ Magnetism::~Magnetism() { } * - calculate the total number of valence electrons from psp files * - CalNbands: unitcell::cal_nbands() * - calculate the number of bands + * - CompareAatomLabel + * - compare_atom_labels(): accept equivalent element names and quit on mismatch + * - PrintUnitcellPseudo + * - print_unitcell_pseudo(): print basic cell and atom info to a pseudo log */ class UcellTest : public ::testing::Test { @@ -763,6 +768,52 @@ TEST_F(UcellTest, CalNbandsGaussWarning) EXPECT_THAT(output, testing::HasSubstr("for smearing, num. of bands > num. of occupied bands")); } +TEST_F(UcellDeathTest, CompareAatomLabel) +{ + std::string stru_label[] + = {"Ag", "Ag", "Ag", "47", "47", "47", "Silver", "Silver", "Silver", "Ag", "Ag", "Ag", "Ag_empty"}; + std::string pseudo_label[] + = {"Ag", "47", "Silver", "Ag", "47", "Silver", "Ag", "47", "Silver", "Ag1", "ag", "ag_locpsp", "Ag"}; + for (int it = 0; it < 12; it++) + { + unitcell::compare_atom_labels(stru_label[it], pseudo_label[it]); + } + stru_label[0] = "Fe"; + pseudo_label[0] = "O"; + std::string atom_label_in_orbtial = "atom label in orbital file "; + std::string mismatch_with_pseudo = " mismatch with pseudo file of "; + testing::internal::CaptureStdout(); + EXPECT_EXIT(unitcell::compare_atom_labels(stru_label[0], pseudo_label[0]), ::testing::ExitedWithCode(1), ""); + output = testing::internal::GetCapturedStdout(); + EXPECT_THAT(output, + testing::HasSubstr(atom_label_in_orbtial + stru_label[0] + mismatch_with_pseudo + pseudo_label[0])); +} + +TEST_F(UcellTest, PrintUnitcellPseudo) +{ + // The fixture builds the C1H2-Read cell in SetUp; this test needs the + // C1H2-Index variant, so rebuild the cell here. + UcellTestPrepare utp_index = UcellTestLib["C1H2-Index"]; + ucell = utp_index.SetUcellInfo(); + std::string fn = "printcell.log"; + unitcell::print_unitcell_pseudo(fn, *ucell); + std::ifstream ifs; + ifs.open("printcell.log"); + std::string str((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + EXPECT_THAT(str, testing::HasSubstr("latName = bcc")); + EXPECT_THAT(str, testing::HasSubstr("ntype = 2")); + EXPECT_THAT(str, testing::HasSubstr("nat = 3")); + EXPECT_THAT(str, testing::HasSubstr("GGT :")); + EXPECT_THAT(str, testing::HasSubstr("omega = 6748.33")); + EXPECT_THAT(str, testing::HasSubstr("label = C")); + EXPECT_THAT(str, testing::HasSubstr("mass = 12")); + EXPECT_THAT(str, testing::HasSubstr("atom_position(cartesian) Dimension = 1")); + EXPECT_THAT(str, testing::HasSubstr("label = H")); + EXPECT_THAT(str, testing::HasSubstr("mass = 1")); + EXPECT_THAT(str, testing::HasSubstr("atom_position(cartesian) Dimension = 2")); + remove("printcell.log"); +} + #ifdef __MPI #include "mpi.h" int main(int argc, char** argv) { diff --git a/source/source_cell/unitcell.cpp b/source/source_cell/unitcell.cpp index 710f1db31cd..cc4db426946 100644 --- a/source/source_cell/unitcell.cpp +++ b/source/source_cell/unitcell.cpp @@ -94,7 +94,7 @@ std::map> UnitCell::get_lnchi_Counts() const { // Calculate various lattice related quantities for given latvec //============================================================== void UnitCell::setup_cell(const std::string& fn, std::ofstream& log, const double symmetry_prec, - const int dfthalf_type, const std::string& pseudo_dir, const int nspin, + const int dfthalf_type, const std::string& pseudo_dir, const int nspin, const std::string& basis_type, const std::string& orbital_dir, const std::string& init_wfc, const double onsite_radius, const bool deepks_setorb, const bool rpa, const bool fixed_atoms, const bool noncolin, const std::string& calculation, diff --git a/source/source_cell/unitcell.h b/source/source_cell/unitcell.h index d9b034f707a..359456737a3 100644 --- a/source/source_cell/unitcell.h +++ b/source/source_cell/unitcell.h @@ -26,11 +26,11 @@ class UnitCell : public BaseCell { const std::string& fixed_axes_in); void setup_cell(const std::string& fn, std::ofstream& log, const double symmetry_prec, - const int dfthalf_type, const std::string& pseudo_dir, const int nspin, + const int dfthalf_type, const std::string& pseudo_dir, const int nspin, const std::string& basis_type, const std::string& orbital_dir, const std::string& init_wfc, const double onsite_radius, const bool deepks_setorb, const bool rpa, const bool fixed_atoms, const bool noncolin, const std::string& calculation, - const std::string& esolver_type, const int symmetry); + const std::string& esolver_type, const int symmetry); void set_iat2itia(); @@ -52,6 +52,14 @@ class UnitCell : public BaseCell { /// @brief Get npol inline const int& get_npol() const { return npol; } + /// @brief Set iat2iwt (and npol) directly; intended for unit tests that + /// build a minimal UnitCell without running the full indexing setup. + inline void set_iat2iwt_for_test(const std::vector& iat2iwt_in, const int npol_in) + { + this->iat2iwt = iat2iwt_in; + this->npol = npol_in; + } + /// @brief Indexing tools for ia and it /// @return true if the last out is reset template diff --git a/source/source_esolver/CMakeLists.txt b/source/source_esolver/CMakeLists.txt index 90e1e92cff1..95488f4be33 100644 --- a/source/source_esolver/CMakeLists.txt +++ b/source/source_esolver/CMakeLists.txt @@ -3,6 +3,7 @@ list(APPEND objects esolver_ks.cpp esolver_fp.cpp esolver_ks_pw.cpp + esolver_ks_pw_tddft.cpp esolver_ks_lcaopw.cpp esolver_sdft_pw.cpp esolver_lj.cpp diff --git a/source/source_esolver/esolver_dfpt_pw.cpp b/source/source_esolver/esolver_dfpt_pw.cpp index e125b1d431c..d46bc7709f9 100644 --- a/source/source_esolver/esolver_dfpt_pw.cpp +++ b/source/source_esolver/esolver_dfpt_pw.cpp @@ -37,15 +37,21 @@ namespace { class XC_First_Order_FDM : public ModuleDFPT::XC_First_Order { public: - XC_First_Order_FDM(ModulePW::PW_Basis* rho_basis, + XC_First_Order_FDM(const int nspin, + const bool domag, + const bool domag_z, + const int gga_grad, + const bool out_elf, + const int test_charge, + ModulePW::PW_Basis* rho_basis, const Charge* chg0, const UnitCell* ucell) : ucell_(ucell) { - fdm_ = new elecstate::PotXC_FDM(rho_basis, chg0, ucell); + fdm_ = new elecstate::PotXC_FDM(nspin, domag, domag_z, gga_grad, out_elf, test_charge, rho_basis, chg0, ucell); chg1_ = new Charge(); chg1_->set_rhopw(rho_basis); - chg1_->allocate(chg0->nspin, false); + chg1_->allocate(chg0->nspin, false, false, test_charge); veff_1_.create(chg0->nspin, chg0->nrxx); } @@ -291,7 +297,8 @@ void ESolver_DFPT_PW::init_dfpt(UnitCell& ucell) } // first-order XC kernel adapter around the converged ground-state density - xc_adapter_ = new XC_First_Order_FDM(this->pw_rho, this->pelec->charge, &ucell); + xc_adapter_ = new XC_First_Order_FDM(this->inp_->nspin, PARAM.globalv.domag, PARAM.globalv.domag_z, this->inp_->gga_grad, + this->inp_->out_elf[0] > 0, this->inp_->test_charge, this->pw_rho, this->pelec->charge, &ucell); if (getenv("DFPT_VKB") != nullptr) { @@ -352,6 +359,7 @@ void ESolver_DFPT_PW::init_dfpt(UnitCell& ucell) } } + this->stp.sync_cpu(); dfpt_->init(ucell, *this->stp.psi_cpu, this->pw_rho, this->pw_wfc, &this->sf, veff_r, this->pelec->wg, this->pelec->ekb, xc_adapter_, nelec_, ecutwfc_, dft_plus_u_ ? this->dftu_.get() : nullptr); diff --git a/source/source_esolver/esolver_dm2rho.cpp b/source/source_esolver/esolver_dm2rho.cpp index 4f1c042deb7..dda7b13d064 100644 --- a/source/source_esolver/esolver_dm2rho.cpp +++ b/source/source_esolver/esolver_dm2rho.cpp @@ -55,18 +55,18 @@ void ESolver_DM2rho::runner(BaseCell& basecell, const int istep) std::string zipname = "output_DM0.npz"; // read DM from file - ModuleIO::read_mat_npz(&(this->pv), ucell, zipname, *(this->dmat.dm->get_DMR_pointer(1))); + ModuleIO::read_mat_npz(&(this->pv), ucell, zipname, *(this->dmat.dm->get_dmr_ptr(1))); // if nspin=2, need extra reading if (this->inp_->nspin == 2) { zipname = "output_DM1.npz"; - ModuleIO::read_mat_npz(&(this->pv), ucell, zipname, *(this->dmat.dm->get_DMR_pointer(2))); + ModuleIO::read_mat_npz(&(this->pv), ucell, zipname, *(this->dmat.dm->get_dmr_ptr(2))); } // it's dangerous to design psiToRho function like this, mohan note 20251024 // this->pelec->psiToRho(*this->psi); - LCAO_domain::dm2rho(this->dmat.dm->get_DMR_vector(), this->inp_->nspin, &this->chr); + LCAO_domain::dm2rho(this->dmat.dm->get_dmr_vec(), this->inp_->nspin, &this->chr, this->inp_->nelec, ucell.omega, false); int nspin0 = this->inp_->nspin == 2 ? 2 : 1; diff --git a/source/source_esolver/esolver_double_xc.cpp b/source/source_esolver/esolver_double_xc.cpp index bc6488ab772..f926084d9d9 100644 --- a/source/source_esolver/esolver_double_xc.cpp +++ b/source/source_esolver/esolver_double_xc.cpp @@ -12,8 +12,11 @@ //-----HSolver ElecState Hamilt-------- #include "source_estate/elecstate_lcao.h" #include "source_estate/elecstate_tools.h" +#include "source_estate/module_charge/chg_init.h" +#include "source_estate/module_charge/chg_tools.h" #include "source_hsolver/hsolver_lcao.h" #include "source_io/module_parameter/parameter.h" +#include "source_io/module_restart/restart.h" // GlobalC::restart for load_exx_flag #include "source_lcao/hamilt_lcao.h" #include "source_lcao/setup_deepks.h" // use deepks, mohan add 2025-10-10 @@ -86,14 +89,29 @@ void ESolver_DoubleXC::before_all_runners(BaseCell& basecell, const Inpu } // 6) initialize the density matrix - this->dmat_base.allocate_dm(&this->kv, &this->pv, this->inp_->nspin); + LCAO_domain::allocate_dm(this->dmat_base, &this->kv, &this->pv, this->inp_->nspin); // 10) inititlize the charge density + module_charge::InitRhoCfg init_rho_cfg; + init_rho_cfg.init_chg = this->inp_->init_chg; + init_rho_cfg.suffix = this->inp_->suffix; + init_rho_cfg.esolver_type = this->inp_->esolver_type; + init_rho_cfg.global_readin_dir = PARAM.globalv.global_readin_dir; + init_rho_cfg.nelec = this->inp_->nelec; + init_rho_cfg.nbands = this->inp_->nbands; + init_rho_cfg.test_charge = this->inp_->test_charge; + init_rho_cfg.domag = PARAM.globalv.domag; + init_rho_cfg.domag_z = PARAM.globalv.domag_z; + init_rho_cfg.npol = PARAM.globalv.npol; + init_rho_cfg.meta_gga = XC_Functional::get_ked_flag(); this->chr_base.set_rhopw(this->pw_rhod); // mohan add 20251130 - const bool kin_den = this->chr_base.kin_density(); // mohan add 20251202 - this->chr_base.allocate(this->inp_->nspin, kin_den); - this->chr_base.init_rho(ucell, this->Pgrid, this->sf.strucFac, ucell.symm, &this->kv); - this->chr_base.check_rho(); + const bool kin_den = XC_Functional::get_ked_flag() || (this->inp_->out_elf[0] > 0); // mohan add 20251202 + this->chr_base.allocate(this->inp_->nspin, kin_den, XC_Functional::get_ked_flag(), + this->inp_->test_charge); + this->chr_base.init_rho(ucell, this->Pgrid, this->sf.strucFac, ucell.symm, &this->kv, nullptr, init_rho_cfg); + module_charge::check_rho(this->chr_base.rho, this->chr_base.nspin, + this->chr_base.rhopw->nrxx, ucell.omega, + this->chr_base.rhopw->nxyz, this->inp_->nelec); // 11) initialize the potential if (this->pelec_base->pot == nullptr) @@ -140,6 +158,8 @@ void ESolver_DoubleXC::before_scf(UnitCell& ucell, const int istep) } if (this->p_hamilt_base == nullptr) { + const bool load_exx_flag = !GlobalC::restart.info_load.restart_exx + && GlobalC::restart.info_load.load_H; this->p_hamilt_base = new hamilt::HamiltLCAO(ucell, this->gd, &this->pv, @@ -153,7 +173,8 @@ void ESolver_DoubleXC::before_scf(UnitCell& ucell, const int istep) istep, this->exx_nao, this->exx_info_, - *this->inp_); + *this->inp_, + load_exx_flag); } XC_Functional::set_xc_type(this->inp_->deepks_out_base); @@ -168,11 +189,11 @@ void ESolver_DoubleXC::before_scf(UnitCell& ucell, const int istep) XC_Functional::set_xc_type(ucell.atoms[0].ncpp.xc_func); // DMR should be same size with Hamiltonian(R) - this->dmat_base.dm->init_DMR(*(dynamic_cast*>(this->p_hamilt_base)->getHR())); + this->dmat_base.dm->init_dmr(*(dynamic_cast*>(this->p_hamilt_base)->getHR())); if (istep > 0) { - this->dmat_base.dm->cal_DMR(); + this->dmat_base.dm->cal_dmr(-1); } ModuleBase::timer::end("ESolver_DoubleXC", "before_scf"); @@ -210,7 +231,11 @@ void ESolver_DoubleXC::iter_finish(UnitCell& ucell, const int istep, int // get etot of output charge density, now the etot is of density after charge mixing this->pelec->pot->update_from_charge(&this->chr_base, &ucell); this->pelec->f_en.descf = 0.0; - this->pelec->cal_energies(2); + this->pelec->cal_energies(2, + this->inp_->imp_sol, + this->inp_->sc_mag_switch, + this->inp_->dft_plus_u, + this->inp_->assume_isolated); // std::cout<<"in deepks etot------"<pelec->f_en.print_all(); // std::cout<<"in deepks etot------"<::iter_finish(UnitCell& ucell, const int istep, int this->pelec_base->f_en.deband = this->pelec->f_en.deband; this->pelec_base->f_en.demet = this->pelec->f_en.demet; this->pelec_base->f_en.descf = 0.0; // set descf to 0 - this->pelec_base->cal_energies(2); // 2 means Kohn-Sham functional + this->pelec_base->cal_energies(2, + this->inp_->imp_sol, + this->inp_->sc_mag_switch, + this->inp_->dft_plus_u, + this->inp_->assume_isolated); // 2 means Kohn-Sham functional // std::cout<<"in double_xc------"<pelec_base->f_en.print_all(); // std::cout<<"in double_xc------"<::iter_finish(UnitCell& ucell, const int istep, int for (int ik = 0; ik < nks; ik++) { // mohan update 2025-11-03 - this->dmat_base.dm->set_DMK_pointer(ik, this->dmat.dm->get_DMK_pointer(ik)); - // _pes_lcao_base->get_DM()->set_DMK_pointer(ik, - // _pes_lcao->get_DM()->get_DMK_pointer(ik)); + this->dmat_base.dm->set_dmk_ptr(ik, this->dmat.dm->get_dmk_ptr(ik)); + // _pes_lcao_base->get_DM()->set_dmk_ptr(ik, + // _pes_lcao->get_DM()->get_dmk_ptr(ik)); } - this->dmat_base.dm->cal_DMR(); - // _pes_lcao_base->get_DM()->cal_DMR(); + this->dmat_base.dm->cal_dmr(-1); + // _pes_lcao_base->get_DM()->cal_dmr(-1); _pes_lcao_base->ekb = _pes_lcao->ekb; _pes_lcao_base->wg = _pes_lcao->wg; } @@ -394,6 +423,11 @@ void ESolver_DoubleXC::cal_force(BaseCell& basecell, ModuleBase::matrix& this->deepks.dpks_out_type = "base"; // for deepks method + const FSCalcConfig fs_cfg{this->inp_->nspin, this->inp_->nbands, this->inp_->t_in_h, + this->inp_->sc_mag_switch, this->inp_->device, + PARAM.globalv.domag, PARAM.globalv.domag_z, this->inp_->gga_grad, + PARAM.globalv.gamma_only_pw}; + fsl.getForceStress(ucell, this->get_vdw_result(), this->inp_->cal_force, @@ -418,7 +452,8 @@ void ESolver_DoubleXC::cal_force(BaseCell& basecell, ModuleBase::matrix& this->deepks, this->exx_nao, &ucell.symm, - this->exx_info_); + this->exx_info_, + fs_cfg); // restore to original xc XC_Functional::set_xc_type(ucell.atoms[0].ncpp.xc_func); diff --git a/source/source_esolver/esolver_double_xc.h b/source/source_esolver/esolver_double_xc.h index 757245db91d..d2aca661848 100644 --- a/source/source_esolver/esolver_double_xc.h +++ b/source/source_esolver/esolver_double_xc.h @@ -32,7 +32,7 @@ class ESolver_DoubleXC : public ESolver_KS_LCAO elecstate::ElecState* pelec_base = nullptr; //! Density Matrix, mohan add 2025-11-03 - LCAO_domain::Setup_DM dmat_base; + module_dm::Setup_DM dmat_base; //! Electorn charge density Charge chr_base; diff --git a/source/source_esolver/esolver_dp.h b/source/source_esolver/esolver_dp.h index e5e865bf77c..8b8067552df 100644 --- a/source/source_esolver/esolver_dp.h +++ b/source/source_esolver/esolver_dp.h @@ -75,6 +75,37 @@ class ESolver_DP : public ESolver */ void after_all_runners(BaseCell& basecell) override; + //==================================================================== + // Test seam. + // + // runner() needs a real DP model file, so the unit tests seed the computed + // results directly and check that cal_energy() / cal_force() / cal_stress() + // hand them back, and that type_map() filled atype correctly. + // + // Production code must keep using the private names directly. + //==================================================================== + + /// @brief atom type index corresponding to the DP model + std::vector& get_atype() + { + return atype; + } + /// @brief computed potential energy + double& get_dp_potential() + { + return dp_potential; + } + /// @brief computed atomic forces + ModuleBase::matrix& get_dp_force() + { + return dp_force; + } + /// @brief computed lattice virials + ModuleBase::matrix& get_dp_virial() + { + return dp_virial; + } + private: /** * @brief determine the type map of DP model diff --git a/source/source_esolver/esolver_factory.cpp b/source/source_esolver/esolver_factory.cpp index 322a0d7a1bc..51c8bb91dd9 100644 --- a/source/source_esolver/esolver_factory.cpp +++ b/source/source_esolver/esolver_factory.cpp @@ -4,6 +4,7 @@ #include "esolver_ks_pw.h" #include "esolver_dfpt_pw.h" #include "esolver_sdft_pw.h" +#include "esolver_ks_pw_tddft.h" #include "source_base/module_device/device.h" #include "source_hamilt/module_xc/general_exx_info.h" #include "source_io/module_parameter/parameter.h" @@ -56,6 +57,10 @@ std::string determine_type(const Input_para& inp) { esolver_type = "dfpt_pw"; } + else if (inp.esolver_type == "tddft") + { + esolver_type = "ksdft_pw_tddft"; + } } else if (inp.basis_type == "lcao_in_pw") { @@ -168,6 +173,30 @@ ESolver* init_esolver(const Input_para& inp) { return new ESolver_DFPT_PW(); } + else if (esolver_type == "ksdft_pw_tddft") + { +#if defined(__CUDA) || defined(__ROCM) + if (inp.device == "gpu") + { + if (inp.precision == "single") + { + return new ESolver_KS_PW_TDDFT, base_device::DEVICE_GPU>(); + } + else + { + return new ESolver_KS_PW_TDDFT, base_device::DEVICE_GPU>(); + } + } +#endif + if (inp.precision == "single") + { + return new ESolver_KS_PW_TDDFT, base_device::DEVICE_CPU>(); + } + else + { + return new ESolver_KS_PW_TDDFT, base_device::DEVICE_CPU>(); + } + } else if (esolver_type == "sdft_pw") { #if ((defined __CUDA) || (defined __ROCM)) diff --git a/source/source_esolver/esolver_fp.cpp b/source/source_esolver/esolver_fp.cpp index 1e763e84dbd..5a661ce4a35 100644 --- a/source/source_esolver/esolver_fp.cpp +++ b/source/source_esolver/esolver_fp.cpp @@ -1,15 +1,17 @@ #include "esolver_fp.h" +#include "source_basis/module_pw/pw_basis_big.h" // use PW_Basis_Big #include "source_base/tool_quit.h" #include "source_cell/cal_ux.h" -#include "source_estate/module_charge/symm_rho.h" +#include "source_estate/module_charge/chg_atomic.h" +#include "source_estate/module_charge/chg_symm.h" #include "source_cell/read_pp_ucell.h" #include "source_estate/param_update.h" #include "source_hamilt/module_ewald/h_ewald_pw.h" #include "source_hamilt/module_vdw/vdw.h" #include "source_io/module_output/output_log.h" #include "source_io/module_output/print_info.h" -#include "source_estate/rhog_io.h" +#include "source_estate/module_charge/chg_rhog_io.h" #include "source_io/module_parameter/parameter.h" #include "source_pw/module_pwdft/setup_pwrho.h" // mohan 20251005 @@ -142,13 +144,13 @@ void ESolver_FP::before_all_runners(BaseCell& basecell, const Input_para& inp) nprocgroup); //! 10) calculate the structure factor - this->sf.setup(&ucell, Pgrid, this->pw_rhod); + this->sf.setup(&ucell, Pgrid, this->pw_rhod, PARAM.globalv.has_float_data); //! 11) initialize the charge density, we need to first set xc_type, // then we can call chr.allocate() this->chr.set_rhopw(this->pw_rhod); // mohan add 20251130 - const bool kin_den = this->chr.kin_density(); // mohan add 20251202 - this->chr.allocate(inp.nspin, kin_den); // mohan move this from setup_estate_pw, 20251128 + const bool kin_den = XC_Functional::get_ked_flag() || (inp.out_elf[0] > 0); // mohan add 20251202 + this->chr.allocate(inp.nspin, kin_den, XC_Functional::get_ked_flag(), inp.test_charge); // mohan move this from setup_estate_pw, 20251128 return; @@ -165,7 +167,13 @@ void ESolver_FP::after_scf(UnitCell& ucell, const int istep, const bool conv_eso ModuleIO::output_efermi(conv_esolver, this->pelec->eferm.ef); //! Update delta_rho for charge extrapolation - CE.update_delta_rho(ucell, &(this->chr), &(this->sf)); + const module_charge::AtomicRhoCfg atomic_rho_cfg_after{ + PARAM.inp.nelec, + PARAM.inp.test_charge, + PARAM.globalv.domag, + PARAM.globalv.domag_z, + GlobalV::ofs_warning}; + CE.update_delta_rho(ucell, &(this->chr), *this->pw_rhod, &(this->sf), atomic_rho_cfg_after); //! print out charge density, potential, elf, etc. ModuleIO::ctrl_output_fp(ucell, *this->inp_, this->pelec, this->pw_big, this->pw_rhod, @@ -218,8 +226,15 @@ void ESolver_FP::before_scf(UnitCell& ucell, const int istep) if (ucell.ionic_position_updated) { this->CE.update_all_dis(ucell); - this->CE.extrapolate_charge(&this->Pgrid, ucell, &this->chr, &this->sf, - GlobalV::ofs_running, GlobalV::ofs_warning); + const module_charge::AtomicRhoCfg atomic_rho_cfg_before{ + PARAM.inp.nelec, + PARAM.inp.test_charge, + PARAM.globalv.domag, + PARAM.globalv.domag_z, + GlobalV::ofs_warning}; + this->CE.extrapolate_charge(&this->Pgrid, ucell, &this->chr, *this->pw_rhod, + &this->sf, GlobalV::ofs_running, GlobalV::ofs_warning, + atomic_rho_cfg_before, PARAM.globalv.has_float_data); } //! Evaluate the vdW correction once for this ionic configuration. @@ -268,7 +283,7 @@ void ESolver_FP::iter_finish(UnitCell& ucell, const int istep, int& iter, bool& // Only pool 0 writes the rhog file (rhog is identical across pools). if (GlobalV::MY_POOL == 0) { - elecstate::write_rhog(PARAM.globalv.global_out_dir + this->inp_->suffix + "-CHARGE-DENSITY.restart", + module_charge::write_rhog(PARAM.globalv.global_out_dir + this->inp_->suffix + "-CHARGE-DENSITY.restart", PARAM.globalv.gamma_only_pw, this->pw_rhod, this->inp_->nspin, @@ -289,7 +304,7 @@ void ESolver_FP::iter_finish(UnitCell& ucell, const int istep, int& iter, bool& } if (GlobalV::MY_POOL == 0) { - elecstate::write_rhog(PARAM.globalv.global_out_dir + this->inp_->suffix + "-TAU-DENSITY.restart", + module_charge::write_rhog(PARAM.globalv.global_out_dir + this->inp_->suffix + "-TAU-DENSITY.restart", PARAM.globalv.gamma_only_pw, this->pw_rhod, this->inp_->nspin, diff --git a/source/source_esolver/esolver_fp.h b/source/source_esolver/esolver_fp.h index 378f92f560c..8d409b9d83b 100644 --- a/source/source_esolver/esolver_fp.h +++ b/source/source_esolver/esolver_fp.h @@ -3,9 +3,9 @@ #include "esolver.h" #include "source_base/timer_wrapper.h" -#include "source_basis/module_pw/pw_basis.h" // plane wave basis +#include "source_basis/module_pw/pw_basis_sup.h" // plane wave basis #include "source_estate/elecstate.h" // electronic states -#include "source_estate/module_charge/charge_extra.h" // charge extrapolation +#include "source_estate/module_charge/chg_extra.h" // charge extrapolation #include "source_hamilt/module_surchem/surchem.h" // solvation model #include "source_base/parallel_grid.h" // Parallel_Grid (value member below) #include "source_pw/module_pwdft/stru_fac.h" // structure factor @@ -81,7 +81,7 @@ class ESolver_FP : public ESolver pseudopot_cell_vl locpp; //! charge extrapolation method - Charge_Extra CE; + module_charge::Charge_Extra CE; //! solvent model surchem solvent; diff --git a/source/source_esolver/esolver_gets.cpp b/source/source_esolver/esolver_gets.cpp index f69b3995624..5c27a14a18f 100644 --- a/source/source_esolver/esolver_gets.cpp +++ b/source/source_esolver/esolver_gets.cpp @@ -5,8 +5,8 @@ #include "source_cell/read_pp_ucell.h" #include "source_estate/elecstate_lcao.h" #include "source_estate/param_update.h" -#include "source_io/module_hs/cal_r_overlap_r.h" -#include "source_io/module_hs/write_hs_r.h" +#include "source_io/module_hs/pos_op_mat.h" +#include "source_io/module_hs/hsr_legacy.h" #include "source_io/module_output/print_info.h" #include "source_lcao/lcao_domain.h" #include "source_lcao/hamilt_lcao.h" @@ -155,7 +155,11 @@ void ESolver_GetS::runner(BaseCell& basecell, const int istep) this->inp_->test_atom_input); Record_adj RA; - RA.for_2d(ucell, gd, this->pv, PARAM.globalv.gamma_only_local, orb_.cutoffs()); + RA.for_2d(ucell, gd, this->pv, PARAM.globalv.gamma_only_local, PARAM.globalv.npol, orb_.cutoffs()); + if (this->inp_->out_level != "m" && !PARAM.globalv.gamma_only_local) + { + ModuleBase::GlobalFunc::OUT(GlobalV::ofs_running, "ParaV.nnr", this->pv.nnr); + } if (this->p_hamilt == nullptr) { @@ -190,18 +194,52 @@ void ESolver_GetS::runner(BaseCell& basecell, const int istep) const std::string fn = PARAM.globalv.global_out_dir + "sr_nao.csr"; auto* hamilt_ptr = static_cast>*>(this->p_hamilt); - ModuleIO::output_SR(pv, gd, hamilt_ptr, fn); + const bool binary = false; + const double sparse_threshold = 1e-10; + const int precision = 16; + ModuleIO::output_SR(pv, + gd, + hamilt_ptr, + fn, + binary, + sparse_threshold, + precision, + PARAM.globalv.global_out_dir, + PARAM.globalv.global_matrix_dir, + PARAM.inp.calculation, + PARAM.inp.out_app_flag, + PARAM.inp.nspin); if (this->inp_->out_mat_r[0]) { - cal_r_overlap_R r_matrix; - r_matrix.init(ucell, pv, orb_); - r_matrix.out_rR(ucell, gd, istep, this->inp_->out_mat_r[1]); + Position_op r_matrix; + const bool cal_force = PARAM.inp.cal_force; + const int nlocal = PARAM.globalv.nlocal; + r_matrix.init(ucell, pv, orb_, cal_force, nlocal); + r_matrix.out_rR(ucell, + gd, + -1, // get_s has no ionic step; use -1 for no step suffix + this->inp_->out_mat_r[1], + PARAM.globalv.global_out_dir, + PARAM.globalv.global_matrix_dir, + PARAM.inp.calculation, + PARAM.inp.out_app_flag, + nlocal, + PARAM.globalv.npol); } if (this->inp_->out_mat_ds[0]) { LCAO_HS_Arrays HS_Arrays; // store sparse arrays + ModuleIO::MatROutputOptions mat_R_options; + mat_R_options.binary = binary; + mat_R_options.sparse_threshold = sparse_threshold; + mat_R_options.precision = this->inp_->out_mat_ds[1]; + mat_R_options.global_out_dir = PARAM.globalv.global_out_dir; + mat_R_options.global_matrix_dir = PARAM.globalv.global_matrix_dir; + mat_R_options.calculation = PARAM.inp.calculation; + mat_R_options.out_app_flag = PARAM.inp.out_app_flag; + mat_R_options.nspin = PARAM.inp.nspin; //! Print out sparse matrix ModuleIO::output_dSR(istep, ucell, @@ -210,10 +248,10 @@ void ESolver_GetS::runner(BaseCell& basecell, const int istep) gd, // mohan add 2024-04-06 two_center_bundle_, orb_, - kv, - false, - 1e-10, - this->inp_->out_mat_ds[1]); + mat_R_options, + PARAM.globalv.gamma_only_local, + PARAM.globalv.npol, + PARAM.globalv.nlocal); } ModuleBase::timer::end("ESolver_GetS", "runner"); diff --git a/source/source_esolver/esolver_gets.h b/source/source_esolver/esolver_gets.h index 29ea4374d40..81282b7ef17 100644 --- a/source/source_esolver/esolver_gets.h +++ b/source/source_esolver/esolver_gets.h @@ -1,6 +1,7 @@ #ifndef ESOLVER_GETS_H #define ESOLVER_GETS_H +#include "source_basis/module_ao/parallel_orbitals.h" #include "source_basis/module_nao/two_center_bundle.h" #include "source_cell/unitcell.h" #include "source_esolver/esolver_ks.h" diff --git a/source/source_esolver/esolver_ks.cpp b/source/source_esolver/esolver_ks.cpp index 65bcd52665f..f4826fca64d 100644 --- a/source/source_esolver/esolver_ks.cpp +++ b/source/source_esolver/esolver_ks.cpp @@ -5,7 +5,10 @@ #include "source_io/module_json/output_info.h" #include "source_estate/update_pot.h" // mohan add 20251016 -#include "source_estate/module_charge/chgmixing.h" // mohan add 20251018 +#include "source_estate/module_charge/chg_routine.h" // mohan add 20251018 +#include "source_estate/module_charge/chg_drho.h" // module_charge::cal_drho/cal_dkin +#include "source_estate/module_charge/chg_init.h" // module_charge::InitRhoCfg +#include "source_estate/module_charge/chg_tools.h" // module_charge::check_rho #include "source_pw/module_pwdft/setup_pwwfc.h" // mohan add 20251018 #include "source_hsolver/hsolver.h" #include "source_io/module_energy/write_eig_occ.h" @@ -64,10 +67,32 @@ void ESolver_KS::before_all_runners(BaseCell& basecell, const Input_para& inp) //! 3) setup charge mixing p_chgmix = new Charge_Mixing(); - p_chgmix->set_rhopw(this->pw_rho, this->pw_rhod); - p_chgmix->set_mixing(inp.mixing_mode, inp.mixing_beta, inp.mixing_ndim, - inp.mixing_gg0, inp.mixing_tau, inp.mixing_beta_mag, inp.mixing_gg0_mag, - inp.mixing_gg0_min, inp.mixing_angle, inp.mixing_dmr, ucell.omega, ucell.tpiba); + // Aggregate-initialize MixingConfig so that adding a field without + // updating this list is a compile error (-Wmissing-field-initializers + // promoted to error via pragma). Fields are in declaration order. +#pragma GCC diagnostic push +#pragma GCC diagnostic error "-Wmissing-field-initializers" + MixingConfig mix_cfg{ + inp.mixing_mode, // mixing_mode + inp.mixing_beta, // mixing_beta + inp.mixing_ndim, // mixing_ndim + inp.mixing_gg0, // mixing_gg0 + inp.mixing_tau && XC_Functional::get_ked_flag(), // mixing_tau + inp.mixing_beta_mag, // mixing_beta_mag + inp.mixing_gg0_mag, // mixing_gg0_mag + inp.mixing_gg0_min, // mixing_gg0_min + inp.mixing_angle, // mixing_angle + inp.mixing_dmr, // mixing_dmr + inp.nspin, // nspin + inp.scf_thr_type, // scf_thr_type + PARAM.globalv.double_grid, // double_grid + PARAM.globalv.gamma_only_pw, // gamma_only_pw + PARAM.globalv.domag, // domag + PARAM.globalv.domag_z, // domag_z + inp.scf_nmax // scf_nmax + }; +#pragma GCC diagnostic pop + p_chgmix->set_mixing(mix_cfg, this->pw_rho, this->pw_rhod, ucell.omega, ucell.tpiba); p_chgmix->init_mixing(); //! 4) setup plane wave for electronic wave functions @@ -75,16 +100,35 @@ void ESolver_KS::before_all_runners(BaseCell& basecell, const Input_para& inp) //! 5) read in charge density, mohan add 2025-11-28 //! Inititlize the charge density. - this->chr.init_rho(ucell, this->Pgrid, this->sf.strucFac, ucell.symm, &this->kv, this->pw_wfc); - this->chr.check_rho(); // check the rho + module_charge::InitRhoCfg init_rho_cfg; + init_rho_cfg.init_chg = inp.init_chg; + init_rho_cfg.suffix = inp.suffix; + init_rho_cfg.esolver_type = inp.esolver_type; + init_rho_cfg.global_readin_dir = PARAM.globalv.global_readin_dir; + init_rho_cfg.nelec = inp.nelec; + init_rho_cfg.nbands = inp.nbands; + init_rho_cfg.test_charge = inp.test_charge; + init_rho_cfg.domag = PARAM.globalv.domag; + init_rho_cfg.domag_z = PARAM.globalv.domag_z; + init_rho_cfg.npol = PARAM.globalv.npol; + init_rho_cfg.meta_gga = XC_Functional::get_ked_flag(); + this->chr.init_rho(ucell, this->Pgrid, this->sf.strucFac, ucell.symm, &this->kv, this->pw_wfc, init_rho_cfg); + module_charge::check_rho(this->chr.rho, this->chr.nspin, this->chr.rhopw->nrxx, ucell.omega, + this->chr.rhopw->nxyz, inp.nelec); // check the rho } void ESolver_KS::hamilt2rho_single(UnitCell& ucell, const int istep, const int iter, const double ethr) {} +std::string ESolver_KS::diag_policy(const int istep) const +{ + return this->inp_->esolver_type; +} + void ESolver_KS::hamilt2rho(UnitCell& ucell, const int istep, const int iter, const double ethr) { + const std::string policy = this->diag_policy(istep); // 1) use Hamiltonian to obtain charge density this->hamilt2rho_single(ucell, istep, iter, diag_ethr); @@ -97,27 +141,29 @@ void ESolver_KS::hamilt2rho(UnitCell& ucell, const int istep, const int iter, co // example wavefunctions uses 20 processors while density uses 10. if (PARAM.globalv.ks_run) { - drho = p_chgmix->get_drho(&this->chr, this->inp_->nelec); + drho = module_charge::cal_drho(&this->chr, this->inp_->nelec, *this->pw_rho, + p_chgmix->get_mixing_config(), ucell.omega, ucell.tpiba); hsolver_error = 0.0; if (iter == 1 && this->inp_->calculation != "nscf") { hsolver_error - = hsolver::cal_hsolve_error(this->inp_->basis_type, this->inp_->esolver_type, diag_ethr, this->inp_->nelec); + = hsolver::cal_hsolve_error(this->inp_->basis_type, policy, diag_ethr, this->inp_->nelec); // The error of HSolver is larger than drho, // so a more precise HSolver should be executed. if (hsolver_error > drho) { diag_ethr = hsolver::reset_diag_ethr(GlobalV::ofs_running, this->inp_->basis_type, - this->inp_->esolver_type, this->inp_->precision, hsolver_error, + policy, this->inp_->precision, hsolver_error, drho, diag_ethr, this->inp_->nelec); this->hamilt2rho_single(ucell, istep, iter, diag_ethr); - drho = p_chgmix->get_drho(&this->chr, this->inp_->nelec); + drho = module_charge::cal_drho(&this->chr, this->inp_->nelec, *this->pw_rho, + p_chgmix->get_mixing_config(), ucell.omega, ucell.tpiba); hsolver_error = hsolver::cal_hsolve_error(this->inp_->basis_type, - this->inp_->esolver_type, diag_ethr, this->inp_->nelec); + policy, diag_ethr, this->inp_->nelec); } } } @@ -195,15 +241,16 @@ void ESolver_KS::iter_init(UnitCell& ucell, const int istep, const int iter) // (meaning "lambda loop not yet run this iteration"); otherwise -1 (no RMS column). this->ds_rms_ = this->inp_->sc_mag_switch ? 0.0 : -1.0; - if (this->inp_->esolver_type == "ksdft") + const std::string policy = this->diag_policy(istep); + if (policy == "ksdft") { - diag_ethr = hsolver::set_diagethr_ks(this->inp_->basis_type, this->inp_->esolver_type, + diag_ethr = hsolver::set_diagethr_ks(this->inp_->basis_type, policy, this->inp_->calculation, this->inp_->init_chg, this->inp_->precision, istep, iter, drho, this->inp_->pw_diag_thr, diag_ethr, this->inp_->nelec, this->inp_->scf_thr); } - else if (this->inp_->esolver_type == "sdft") + else if (policy == "sdft") { - diag_ethr = hsolver::set_diagethr_sdft(this->inp_->basis_type, this->inp_->esolver_type, + diag_ethr = hsolver::set_diagethr_sdft(this->inp_->basis_type, policy, this->inp_->calculation, this->inp_->init_chg, istep, iter, drho, this->inp_->pw_diag_thr, diag_ethr, this->inp_->nbands, esolver_KS_ne, this->inp_->nelec, this->inp_->scf_thr); @@ -213,6 +260,11 @@ void ESolver_KS::iter_init(UnitCell& ucell, const int istep, const int iter) this->chr.save_rho_before_sum_band(); } +ESolver_KS::DensityStage ESolver_KS::density_stage(const int istep, const int iter) const +{ + return DensityStage::standard; +} + void ESolver_KS::iter_finish(UnitCell& ucell, const int istep, int& iter, bool &conv_esolver) { @@ -231,7 +283,11 @@ void ESolver_KS::iter_finish(UnitCell& ucell, const int istep, int& iter, bool & { if (iter % this->inp_->out_freq_elec == 0 || iter == this->inp_->scf_nmax || conv_esolver) { - ModuleIO::write_eig_iter(this->pelec->ekb,this->pelec->wg,*this->pelec->klist); + ModuleIO::write_eig_iter(this->pelec->ekb, + this->pelec->wg, + *this->pelec->klist, + this->inp_->nbands, + this->inp_->nspin); } } @@ -251,16 +307,46 @@ void ESolver_KS::iter_finish(UnitCell& ucell, const int istep, int& iter, bool & } #endif - module_charge::chgmixing_ks(iter, ucell, this->pelec, this->chr, this->p_chgmix, - this->pw_rhod->nrxx, this->drho, this->oscillate_esolver, conv_esolver, hsolver_error, - this->scf_thr, this->scf_ene_thr, converged_u, *this->inp_); + module_charge::ScfMixingCtx ctx; + ctx.hsolver_error = hsolver_error; + ctx.scf_thr = this->scf_thr; + ctx.scf_ene_thr = this->scf_ene_thr; + ctx.converged_u = converged_u; + ctx.ks_run = PARAM.globalv.ks_run; + ctx.drho = this->drho; + ctx.oscillate_esolver = this->oscillate_esolver; + ctx.conv_esolver = conv_esolver; + const DensityStage stage = this->density_stage(istep, iter); + if (stage == DensityStage::predictor) + { + // Build the endpoint potential from the full predicted density, including tau. + // Mixing it with the old endpoint would spoil the subsequent midpoint estimate. + ctx.conv_esolver = false; + ctx.oscillate_esolver = false; + } + else + { + module_charge::chgmixing_ks(iter, ucell, this->pelec, this->chr, + *this->chr.rhopw, this->p_chgmix, ctx, *this->inp_); + } + this->drho = ctx.drho; + this->oscillate_esolver = ctx.oscillate_esolver; + conv_esolver = ctx.conv_esolver; // 2.3) Update potentials (should be done every SF iter) elecstate::update_pot(ucell, this->pelec, this->chr, conv_esolver); // 3.1) calculate energies - this->pelec->cal_energies(1); // Harris-Foulkes functional - this->pelec->cal_energies(2); // Kohn-Sham functional + this->pelec->cal_energies(1, + this->inp_->imp_sol, + this->inp_->sc_mag_switch, + this->inp_->dft_plus_u, + this->inp_->assume_isolated); // Harris-Foulkes functional + this->pelec->cal_energies(2, + this->inp_->imp_sol, + this->inp_->sc_mag_switch, + this->inp_->dft_plus_u, + this->inp_->assume_isolated); // Kohn-Sham functional if (iter == 1) { @@ -273,7 +359,8 @@ void ESolver_KS::iter_finish(UnitCell& ucell, const int istep, int& iter, bool & double dkin = 0.0; // for meta-GGA if (XC_Functional::get_ked_flag()) { - dkin = p_chgmix->get_dkin(&this->chr, this->inp_->nelec); + dkin = module_charge::cal_dkin(&this->chr, this->inp_->nelec, *this->pw_rho, + p_chgmix->get_mixing_config(), ucell.omega); } // Iter finish @@ -285,16 +372,16 @@ void ESolver_KS::iter_finish(UnitCell& ucell, const int istep, int& iter, bool & // print energies elecstate::print_etot(ucell.magnet, *pelec, conv_esolver, iter, drho, - dkin, duration, diag_ethr, 0, true, this->ds_rms_); + dkin, duration, *this->inp_, PARAM.globalv.two_fermi, diag_ethr, 0, true, this->ds_rms_); -#ifdef __RAPIDJSON +#ifdef __JSON // add Json of scf mag Json::add_output_scf_mag(ucell.magnet.tot_mag, ucell.magnet.abs_mag, this->pelec->f_en.etot * ModuleBase::Ry_to_eV, this->pelec->f_en.etot_delta * ModuleBase::Ry_to_eV, drho, duration); -#endif //__RAPIDJSON +#endif //__JSON } @@ -316,7 +403,13 @@ void ESolver_KS::after_scf(UnitCell& ucell, const int istep, const bool conv_eso ESolver_FP::after_scf(ucell, istep, conv_esolver); // 3) write eigenvalues and occupations to eig_occ.txt - ModuleIO::write_eig_file(this->pelec->ekb, this->pelec->wg, this->kv, istep); + ModuleIO::write_eig_file(this->pelec->ekb, + this->pelec->wg, + this->kv, + this->inp_->nbands, + this->inp_->nspin, + PARAM.globalv.global_out_dir, + istep); // 4) write band information to band.txt ModuleIO::write_bands(*this->inp_, this->pelec->ekb, this->kv); diff --git a/source/source_esolver/esolver_ks.h b/source/source_esolver/esolver_ks.h index 5736fb150a5..2f7055b5d57 100644 --- a/source/source_esolver/esolver_ks.h +++ b/source/source_esolver/esolver_ks.h @@ -4,7 +4,7 @@ #include "esolver_fp.h" // first-principles esolver #include "source_basis/module_pw/pw_basis_k.h" // use plane wave #include "source_cell/klist.h" // use k-points in Brillouin zone -#include "source_estate/module_charge/charge_mixing.h" // use charge mixing +#include "source_estate/module_charge/chg_mix.h" // use charge mixing #include "source_hamilt/hamilt.h" // use Hamiltonian #include "source_hamilt/hamilt_base.h" // use Hamiltonian base class #include "source_hamilt/module_xc/general_exx_info.h" // ESolver owns General_Exx_Info value @@ -32,6 +32,16 @@ class ESolver_KS : public ESolver_FP virtual void after_all_runners(BaseCell& basecell) override; protected: + /** @brief Select whether the density is an endpoint predictor or an SCF iterate. */ + enum class DensityStage + { + standard, + predictor + }; + + /** @brief Identify mandatory predictor stages before mixing and convergence checks. */ + virtual DensityStage density_stage(const int istep, const int iter) const; + //! Something to do before SCF iterations. virtual void before_scf(UnitCell& ucell, const int istep) override; @@ -47,6 +57,13 @@ class ESolver_KS : public ESolver_FP // calculate electron density from a specific Hamiltonian void hamilt2rho(UnitCell& ucell, const int istep, const int iter, const double ethr); + /** + * @brief Select the diagonalization threshold and error-control policy for the current stage. + * + * This policy does not change the requested solver type. + */ + virtual std::string diag_policy(const int istep) const; + //! Something to do after SCF iterations when SCF is converged or comes to the max iter step. virtual void after_scf(UnitCell& ucell, const int istep, const bool conv_esolver) override; diff --git a/source/source_esolver/esolver_ks_lcao.cpp b/source/source_esolver/esolver_ks_lcao.cpp index 95e5a4061d5..f8c89b038ef 100644 --- a/source/source_esolver/esolver_ks_lcao.cpp +++ b/source/source_esolver/esolver_ks_lcao.cpp @@ -1,4 +1,5 @@ #include "esolver_ks_lcao.h" +#include "source_basis/module_pw/pw_basis_big.h" // use PW_Basis_Big #include "source_base/module_external/blacs_connector.h" #include "source_cell/module_neighbor/sltk_atom_arrange.h" #include "source_estate/elecstate_tools.h" @@ -7,24 +8,28 @@ #include "source_lcao/setup_dftu_lcao.h" #include "source_pw/module_pwdft/dftu_base.h" // Plus_U_Base (PW and LCAO share it) #include "source_hamilt/hs_matrix_k.h" -#include "source_estate/module_charge/symm_rho.h" +#include "source_estate/module_charge/chg_symm.h" +#include "source_estate/module_charge/chg_dmr.h" #include "source_lcao/lcao_domain.h" // need DeePKS_init #include "source_lcao/force_stress_lcao.h" #include "source_hamilt/module_gint/gint.h" #include "source_estate/elecstate_lcao.h" #include "source_lcao/hamilt_lcao.h" +#include "source_hamilt/hamilt_hs_adapter.h" #include "source_hsolver/hsolver_lcao.h" #ifdef __EXX #include "../source_lcao/module_ri/exx_opt_orb.h" #endif #include "source_lcao/module_rdmft/rdmft.h" -#include "source_estate/module_charge/chgmixing.h" // use charge mixing, mohan add 20251006 -#include "source_estate/module_dm/init_dm.h" // init dm from electronic wave functions -#include "source_io/module_ctrl/ctrl_runner_lcao.h" // use ctrl_runner_lcao() -#include "source_io/module_ctrl/ctrl_iter_lcao.h" // use ctrl_iter_lcao() +#include "source_estate/module_charge/chg_routine.h" // use charge mixing, mohan add 20251006 +#include "source_estate/module_dm/dm_routine.h" // init dm from electronic wave functions +#include "source_io/module_restart/restart.h" // GlobalC::restart for load_exx_flag +#include "source_io/module_ctrl/ctrl_runner_lcao.h" // use ctrl_runner_lcao() +#include "source_io/module_ctrl/ctrl_iter_lcao.h" // use ctrl_iter_lcao() #include "source_io/module_ctrl/ctrl_scf_lcao.h" // use ctrl_scf_lcao() #include "source_io/module_output/print_info.h" #include "source_lcao/rho_tau_lcao.h" // mohan add 20251024 +#include "source_lcao/module_rt/td_info.h" // TD_info for init_dm config #include "source_lcao/lcao_set.h" // mohan add 20251111 #include "source_psi/setup_psi.h" // use Setup_Psi for deallocate_psi @@ -150,7 +155,11 @@ void ESolver_KS_LCAO::before_scf(UnitCell& ucell, const int istep) // 7) For each atom, calculate the adjacent atoms in different cells // and allocate the space for H(R) and S(R). // If k point is used here, allocate HlocR after atom_arrange. - this->RA.for_2d(ucell, this->gd, this->pv, PARAM.globalv.gamma_only_local, orb_.cutoffs()); + this->RA.for_2d(ucell, this->gd, this->pv, PARAM.globalv.gamma_only_local, PARAM.globalv.npol, orb_.cutoffs()); + if (this->inp_->out_level != "m" && !PARAM.globalv.gamma_only_local) + { + ModuleBase::GlobalFunc::OUT(GlobalV::ofs_running, "ParaV.nnr", this->pv.nnr); + } // 8) initialize the Hamiltonian operators // if atom moves, then delete old pointer and add a new one @@ -161,9 +170,12 @@ void ESolver_KS_LCAO::before_scf(UnitCell& ucell, const int istep) } if (this->p_hamilt == nullptr) { + const bool load_exx_flag = !GlobalC::restart.info_load.restart_exx + && GlobalC::restart.info_load.load_H; this->p_hamilt = new hamilt::HamiltLCAO( ucell, this->gd, &this->pv, this->pelec->pot, this->kv, - two_center_bundle_, orb_, this->dmat.dm, this->dftu_.get(), this->deepks, istep, exx_nao, this->exx_info_, *this->inp_); + two_center_bundle_, orb_, this->dmat.dm, this->dftu_.get(), this->deepks, istep, exx_nao, this->exx_info_, *this->inp_, + load_exx_flag); } // 9) for each ionic step, the overlap must be rebuilt @@ -189,7 +201,7 @@ void ESolver_KS_LCAO::before_scf(UnitCell& ucell, const int istep) { ModuleBase::WARNING_QUIT("ESolver_KS_LCAO::before_scf","p_hamilt does not exist"); } - this->dmat.dm->init_DMR(*hamilt_lcao->getHR()); + this->dmat.dm->init_dmr(*hamilt_lcao->getHR()); // 13.1) decide the strategy for initializing DMR and HR if(istep == 0)//if the first scf step, readin DMR from file, @@ -214,7 +226,7 @@ void ESolver_KS_LCAO::before_scf(UnitCell& ucell, const int istep) // 13.1.2) two cases are considered: // 1. DMK in DensityMatrix is not empty (istep > 0), then DMR is initialized by DMK // 2. DMK in DensityMatrix is empty (istep == 0), then DMR is initialized by zeros - this->dmat.dm->cal_DMR(); + this->dmat.dm->cal_dmr(-1); } // 13.2) init_scf, should be before_scf? mohan add 2025-03-10 elecstate::init_scf(ucell, this->Pgrid, this->sf.strucFac, this->locpp.numeric, @@ -222,11 +234,11 @@ void ESolver_KS_LCAO::before_scf(UnitCell& ucell, const int istep) #ifdef __MLALGO // 14) initialize DM2(R) of DeePKS, the DM2(R) is different from DM(R) - this->deepks.ld.init_DMR(ucell, orb_, this->pv, this->gd); + this->deepks.ld.init_dmr(ucell, orb_, this->pv, this->gd); #endif // 16) the electron charge density should be symmetrized, - Symmetry_rho::symmetrize_rho(this->inp_->nspin, this->chr, this->pw_rho, ucell.symm); + module_charge::symmetrize_rho(this->inp_->nspin, this->chr, this->pw_rho, ucell.symm); // 17) update of RDMFT, added by jghan if (this->inp_->rdmft == true) @@ -258,13 +270,20 @@ void ESolver_KS_LCAO::cal_force(BaseCell& basecell, ModuleBase::matrix& deepks.dpks_out_type = "tot"; // for deepks method + const FSCalcConfig fs_cfg{this->inp_->nspin, this->inp_->nbands, this->inp_->t_in_h, + this->inp_->sc_mag_switch, this->inp_->device, + PARAM.globalv.domag, PARAM.globalv.domag_z, this->inp_->gga_grad, + PARAM.globalv.gamma_only_pw}; + fsl.getForceStress(ucell, this->get_vdw_result(), this->inp_->cal_force, this->inp_->cal_stress, this->inp_->test_force, this->inp_->test_stress, this->gd, this->pv, this->pelec, this->dmat, this->psi, two_center_bundle_, orb_, force, this->scs, this->locpp, this->sf, this->kv, this->pw_rho, this->solvent, *this->dftu_, this->deepks, - this->exx_nao, &ucell.symm, this->exx_info_, this->inp_->td_stype, + this->exx_nao, &ucell.symm, this->exx_info_, + fs_cfg, + this->inp_->td_stype, static_cast*>(this->p_hamilt)); // delete RA after cal_force @@ -315,19 +334,11 @@ void ESolver_KS_LCAO::after_all_runners(BaseCell& basecell) } ModuleIO::ctrl_runner_lcao(ucell, - *this->inp_, this->kv, this->pelec, this->dmat, this->pv, this->Pgrid, - this->gd, this->psi, this->chr, hamilt_lcao, - this->two_center_bundle_, - this->orb_, this->pw_rho, this->pw_rhod, - this->sf, this->locpp.vloc, this->exx_nao, this->exx_info_, this->solvent); - - -#ifdef __MPI -#ifdef __LCAO - // Exit BLACS environment for LCAO calculations - Cblacs_exit(1); -#endif -#endif + *this->inp_, this->kv, this->pelec, this->dmat, this->pv, this->Pgrid, + this->gd, this->psi, this->chr, hamilt_lcao, + this->two_center_bundle_, + this->orb_, this->pw_rho, this->pw_rhod, + this->sf, this->locpp.vloc, this->exx_nao, this->exx_info_, this->solvent); ModuleBase::timer::end("ESolver_KS_LCAO", "after_all_runners"); } @@ -341,7 +352,7 @@ void ESolver_KS_LCAO::iter_init(UnitCell& ucell, const int istep, const ESolver_KS::iter_init(ucell, istep, iter); module_charge::chgmixing_ks_lcao(iter, this->p_chgmix, *this->dftu_, - this->dmat.dm->get_DMR_pointer(1)->get_nnr(), *this->inp_); + this->dmat.dm->get_dmr_ptr(1)->get_nnr(), *this->inp_); if (iter == 1) { @@ -372,11 +383,23 @@ void ESolver_KS_LCAO::iter_init(UnitCell& ucell, const int istep, const { // the following steps are only needed in the first outer exx loop exx_two_level_step - = exx_info_.info_ri.real_number ? + = exx_info_.info_ri.real_number ? this->exx_nao.exd->two_level_step : this->exx_nao.exc->two_level_step; } #endif - elecstate::init_dm(ucell, this->pelec, this->dmat, this->psi, this->chr, iter, exx_two_level_step); + module_dm::Init_DM_Config init_dm_cfg; + init_dm_cfg.esolver_type = PARAM.inp.esolver_type; + init_dm_cfg.td_stype = PARAM.inp.td_stype; + init_dm_cfg.nspin = PARAM.inp.nspin; + init_dm_cfg.nelec = PARAM.inp.nelec; + init_dm_cfg.td_phase_hybrid = (PARAM.inp.td_stype == 2 && PARAM.inp.esolver_type != "tddft") + ? &TD_info::td_vel_op->get_phase_hybrid() + : nullptr; + init_dm_cfg.td_cart_At = (PARAM.inp.td_stype == 2 && PARAM.inp.esolver_type != "tddft") + ? TD_info::A_prop_ha + : ModuleBase::Vector3(); + init_dm_cfg.dm2rho_func = &LCAO_domain::dm2rho; + module_dm::init_dm(ucell, this->pelec, this->dmat, this->psi, this->chr, this->pv, iter, exx_two_level_step, init_dm_cfg); } #ifdef __EXX @@ -385,11 +408,11 @@ void ESolver_KS_LCAO::iter_init(UnitCell& ucell, const int istep, const { if (exx_info_.info_ri.real_number) { - this->exx_nao.exd->exx_eachiterinit(istep, ucell, *this->dmat.dm, this->kv, iter); + this->exx_nao.exd->exx_eachiterinit(istep, ucell, *this->dmat.dm, this->kv, this->pv, iter); } else { - this->exx_nao.exc->exx_eachiterinit(istep, ucell, *this->dmat.dm, this->kv, iter); + this->exx_nao.exc->exx_eachiterinit(istep, ucell, *this->dmat.dm, this->kv, this->pv, iter); } } #endif @@ -416,7 +439,7 @@ void ESolver_KS_LCAO::iter_init(UnitCell& ucell, const int istep, const // save density matrix DMR for mixing if (this->inp_->mixing_restart > 0 && this->inp_->mixing_dmr && this->p_chgmix->mixing_restart_count > 0) { - this->dmat.dm->save_DMR(); + this->dmat.dm->save_dmr(); } } @@ -467,14 +490,16 @@ void ESolver_KS_LCAO::hamilt2rho_single(UnitCell& ucell, int istep, int this->inp_->device == "gpu", GlobalV::NPROC, GlobalV::MY_RANK); - hsolver_lcao_obj.solve(static_cast*>(this->p_hamilt), this->psi[0], this->pelec, *this->dmat.dm, - this->chr, this->inp_->nspin, skip_charge); + // the eigensolvers only ever ask the Hamiltonian for H(k) and S(k) + hamilt::HamiltHSMatrix hs(static_cast*>(this->p_hamilt)); + hsolver_lcao_obj.solve(hs, this->psi[0], this->pelec, *this->dmat.dm, + this->chr, this->inp_->nspin, ucell.omega, skip_charge); } else { // Lambda loop updated the density matrix (DM) but not the real-space charge density. // HSolver was skipped, so we need to sync rho from DM manually. - LCAO_domain::dm2rho(this->dmat.dm->get_DMR_vector(), this->inp_->nspin, &this->chr); + LCAO_domain::dm2rho(this->dmat.dm->get_dmr_vec(), this->inp_->nspin, &this->chr, this->inp_->nelec, ucell.omega, false); } // 4) EXX @@ -490,10 +515,27 @@ void ESolver_KS_LCAO::hamilt2rho_single(UnitCell& ucell, int istep, int this->exx_nao.exc->exx_hamilt2rho(*this->pelec, this->pv, iter); } } + else if (exx_info_.info_global.cal_exx) + { + // For nscf, Hexx and Eexx have been computed in OperatorEXX constructor. + // Sync the energy to f_en.exx so that the output prints the correct E_exx. + if (exx_info_.info_ri.real_number) + { + this->pelec->set_exx(this->exx_nao.exd->get_Eexx(), + exx_info_.info_global.cal_exx, + exx_info_.info_global.hybrid_alpha); + } + else + { + this->pelec->set_exx(this->exx_nao.exc->get_Eexx(), + exx_info_.info_global.cal_exx, + exx_info_.info_global.hybrid_alpha); + } + } #endif // 5) symmetrize the charge density - Symmetry_rho::symmetrize_rho(this->inp_->nspin, this->chr, this->pw_rho, ucell.symm); + module_charge::symmetrize_rho(this->inp_->nspin, this->chr, this->pw_rho, ucell.symm); // 6) calculate delta energy this->pelec->f_en.deband = this->pelec->cal_delta_eband(ucell); @@ -512,7 +554,7 @@ void ESolver_KS_LCAO::iter_finish(UnitCell& ucell, const int istep, int& ModuleBase::WARNING_QUIT("ESolver_KS_LCAO::iter_finish","p_hamilt does not exist"); } - const std::vector>& dm_vec = this->dmat.dm->get_DMK_vector(); + const std::vector>& dm_vec = this->dmat.dm->get_dmk_vec(); // 1) calculate the local occupation number matrix and energy correction in DFT+U finish_dftu_lcao(conv_esolver, this->inp_->dft_plus_u, this->inp_->out_chg[0], this->dftu_.get(), ucell, dm_vec, this->kv, this->p_chgmix->get_mixing_beta(), hamilt_lcao, PARAM.globalv.global_out_dir, this->inp_->nspin, PARAM.globalv.npol, PARAM.globalv.gamma_only_local); @@ -533,7 +575,7 @@ void ESolver_KS_LCAO::iter_finish(UnitCell& ucell, const int istep, int& // call iter_finish() of ESolver_KS, where band gap is printed, // eig and occ are printed, magnetization is calculated, - // charge mixing is performed, potential is updated, + // charge mixing is performed, potential is updated, // HF and kS energies are computed, meta-GGA, Jason and restart ESolver_KS::iter_finish(ucell, istep, iter, conv_esolver); const bool precision_switched = this->gint_precision_controller_.update_after_iteration(this->drho, this->scf_thr); @@ -551,14 +593,31 @@ void ESolver_KS_LCAO::iter_finish(UnitCell& ucell, const int istep, int& { if (this->inp_->mixing_restart > 0 && this->p_chgmix->mixing_restart_count > 0 && this->inp_->mixing_dmr) { - this->p_chgmix->mix_dmr(this->dmat.dm); + // Extract the contiguous per-spin DMR buffers expected by the + // stateless mixing kernel. + const std::vector*>& dmr_containers + = this->dmat.dm->get_dmr_vec(); + const std::vector>& dmr_save = this->dmat.dm->get_dmr_save(); + std::vector dmr_out; + std::vector dmr_in; + for (std::size_t is = 0; is < dmr_containers.size(); ++is) + { + dmr_out.push_back(dmr_containers[is]->get_wrapper()); + dmr_in.push_back(dmr_save[is].data()); + } + module_charge::mix_dmr(dmr_out, + dmr_in, + dmr_containers[0]->get_nnr(), + this->p_chgmix->get_mixing(), + this->p_chgmix->get_dmr_mdata(), + this->p_chgmix->get_mixing_config()); } } // control the output related to the finished iteration ModuleIO::ctrl_iter_lcao(ucell, *this->inp_, this->kv, this->pelec, *this->dmat.dm, - this->pv, this->gd, this->psi, this->chr, this->p_chgmix, - hamilt_lcao, this->orb_, this->deepks, + this->pv, this->gd, this->psi, this->chr, this->p_chgmix, + hamilt_lcao, this->orb_, this->deepks, this->exx_nao, this->exx_info_, iter, istep, conv_esolver, this->scf_ene_thr); } @@ -577,7 +636,7 @@ void ESolver_KS_LCAO::after_scf(UnitCell& ucell, const int istep, const if (this->inp_->out_elf[0] > 0) { - LCAO_domain::dm2tau(this->dmat.dm->get_DMR_vector(), this->inp_->nspin, this->pelec->charge); + LCAO_domain::dm2tau(this->dmat.dm->get_dmr_vec(), this->inp_->nspin, this->pelec->charge); } //! 1) call after_scf() of ESolver_KS diff --git a/source/source_esolver/esolver_ks_lcao.h b/source/source_esolver/esolver_ks_lcao.h index 50ee0e6da7a..e6d4cad5ee3 100644 --- a/source/source_esolver/esolver_ks_lcao.h +++ b/source/source_esolver/esolver_ks_lcao.h @@ -3,14 +3,14 @@ #include "esolver_ks.h" #include "source_hamilt/module_xc/exx_info.h" // LCAO owns full Exx_Info -#include "source_lcao/record_adj.h" // adjacent atoms +#include "source_cell/record_adj.h" // adjacent atoms #include "source_basis/module_nao/two_center_bundle.h" // nao basis #include "source_hamilt/module_gint/gint_info.h" -#include "source_estate/module_charge/gint_prec_ctrl.h" +#include "source_hamilt/module_gint/gint_prec_ctrl.h" #include "source_lcao/setup_deepks.h" // for deepks, mohan add 20251008 #include "source_lcao/setup_exx.h" // for exx, mohan add 20251008 #include "source_lcao/module_rdmft/rdmft.h" // rdmft -#include "source_lcao/setup_dm.h" // mohan add 2025-10-30 +#include "source_lcao/allocate_dm.h" // mohan add 2025-10-30 #include #include @@ -76,7 +76,7 @@ class ESolver_KS_LCAO : public ESolver_KS TwoCenterBundle two_center_bundle_; //! Add density matrix class, mohan add 2025-10-30 - LCAO_domain::Setup_DM dmat; + module_dm::Setup_DM dmat; // For deepks method, mohan add 2025-10-08 diff --git a/source/source_esolver/esolver_ks_lcao_tddft.cpp b/source/source_esolver/esolver_ks_lcao_tddft.cpp index e74951f0e91..a31cb15b0a6 100644 --- a/source/source_esolver/esolver_ks_lcao_tddft.cpp +++ b/source/source_esolver/esolver_ks_lcao_tddft.cpp @@ -13,12 +13,14 @@ #include "source_io/module_wf/read_wfc_nao.h" //------LCAO HSolver ElecState------- #include "source_estate/elecstate_tools.h" -#include "source_estate/module_charge/symm_rho.h" -#include "source_estate/module_dm/cal_dm_psi.h" -#include "source_estate/module_dm/cal_edm_tddft.h" +#include "source_estate/module_charge/chg_atomic.h" +#include "source_estate/module_charge/chg_symm.h" +#include "source_estate/module_dm/dm_from_psi.h" +#include "source_estate/module_dm/edm_tddft.h" #include "source_estate/module_pot/h_tddft_pw.h" #include "source_estate/module_pot/potential_new.h" #include "source_estate/module_pot/td_field_manager.h" +#include "source_hamilt/hamilt_hs_adapter.h" #include "source_hsolver/hsolver_lcao.h" #include "source_lcao/module_rt/evolve_elec.h" #include "source_lcao/rho_tau_lcao.h" @@ -82,6 +84,11 @@ void ESolver_KS_LCAO_TDDFT::before_all_runners(BaseCell& basecell, c // components. Both propagation gauges and the length-gauge potential must // observe the same electronic-step counter. td_field_manager_ = elecstate::create_td_field_manager(inp); + if (inp.init_vecpot_file) + { + const std::vector> samples = ModuleIO::read_td_vector_pot(""); + td_field_manager_->set_A_samples(samples); + } if (inp.mdp.md_restart) { td_field_manager_->read_restart(PARAM.globalv.global_readin_dir); @@ -94,13 +101,14 @@ void ESolver_KS_LCAO_TDDFT::before_all_runners(BaseCell& basecell, c { ModuleIO::prepare_td_vector_pot_output(PARAM.globalv.global_out_dir, inp.mdp.md_restart); } - elecstate::H_TDDFT_pw::sync_compatibility_state(*td_field_manager_); + elecstate::H_TDDFT_pw::set_field_state(*td_field_manager_); // Run before_all_runners in ESolver_KS_LCAO ESolver_KS_LCAO, TR>::before_all_runners(ucell, inp); this->pelec->pot->set_td_field_manager(td_field_manager_); - td_p = new TD_info(&ucell, this->pv, this->orb_); + const int restart_step = inp.mdp.md_restart ? td_field_manager_->current_step() : 0; + td_p = new TD_info(&ucell, this->pv, this->orb_, restart_step); TD_info::td_vel_op = td_p; totstep += TD_info::estep_shift; @@ -135,6 +143,12 @@ void ESolver_KS_LCAO_TDDFT::runner(BaseCell& basecell, const int ist //---------------------------------------------------------------- // 1) before_scf (electronic iteration loops) //---------------------------------------------------------------- + const int first_electronic_step = this->totstep + 1; + if (this->inp_->td_stype == 0) + { + td_field_manager_->prepare_sample(first_electronic_step); + elecstate::H_TDDFT_pw::set_field_state(*td_field_manager_); + } this->before_scf(ucell, istep); // From ESolver_KS_LCAO td_p->initialize_phase_hybrid(ucell, dynamic_cast, TR>*>(this->p_hamilt)->getHR()); td_p->calculate_grad_overlap(this->pv, ucell, this->gd, this->orb_.cutoffs(), this->two_center_bundle_.overlap_orb.get()); @@ -149,11 +163,11 @@ void ESolver_KS_LCAO_TDDFT::runner(BaseCell& basecell, const int ist if (this->inp_->td_stype == 2) { - this->dmat.dm->cal_DMR_td(td_p->get_phase_hybrid(), TD_info::cart_At); + this->dmat.dm->cal_dmr_td(td_p->get_phase_hybrid(), TD_info::A_prop_ha, -1); } else { - this->dmat.dm->cal_DMR(); + this->dmat.dm->cal_dmr(-1); } ModuleBase::GlobalFunc::DONE(GlobalV::ofs_running, "INIT SCF"); @@ -186,24 +200,45 @@ void ESolver_KS_LCAO_TDDFT::runner(BaseCell& basecell, const int ist // update At if (this->inp_->td_stype > 0) { - // TDFieldManager owns the midpoint-vector-potential update. Static - // mirrors are synchronized afterward for legacy RT-TDDFT clients. - td_field_manager_->advance_vector_gauge(); - elecstate::H_TDDFT_pw::sync_compatibility_state(*td_field_manager_); - if (this->inp_->out_efield && GlobalV::MY_RANK == 0) + td_field_manager_->prepare_interval(this->totstep, this->totstep); + elecstate::H_TDDFT_pw::set_field_state(*td_field_manager_); + td_p->set_A_prop(this->totstep, td_field_manager_->A_prop_ha()); + if (this->inp_->out_vecpot && GlobalV::MY_RANK == 0) { - ModuleIO::write_td_field_values(*td_field_manager_, PARAM.globalv.global_out_dir); + ModuleIO::write_td_vector_pot(PARAM.globalv.global_out_dir, this->totstep, td_field_manager_->A_prop_ha()); } - td_p->cal_cart_At(elecstate::H_TDDFT_pw::At); - ModuleBase::GlobalFunc::OUT(GlobalV::ofs_running, "Cartesian vector potential Ax(t)", TD_info::cart_At[0]); - ModuleBase::GlobalFunc::OUT(GlobalV::ofs_running, "Cartesian vector potential Ay(t)", TD_info::cart_At[1]); - ModuleBase::GlobalFunc::OUT(GlobalV::ofs_running, "Cartesian vector potential Az(t)", TD_info::cart_At[2]); + ModuleBase::GlobalFunc::OUT(GlobalV::ofs_running, "Cartesian vector potential Ax(t)", TD_info::A_prop_ha[0]); + ModuleBase::GlobalFunc::OUT(GlobalV::ofs_running, "Cartesian vector potential Ay(t)", TD_info::A_prop_ha[1]); + ModuleBase::GlobalFunc::OUT(GlobalV::ofs_running, "Cartesian vector potential Az(t)", TD_info::A_prop_ha[2]); + } + + if (this->inp_->td_stype == 0) + { + td_field_manager_->prepare_sample(this->totstep); + elecstate::H_TDDFT_pw::set_field_state(*td_field_manager_); + } + if (this->inp_->out_efield && GlobalV::MY_RANK == 0) + { + ModuleIO::write_td_field_values(*td_field_manager_, PARAM.globalv.global_out_dir); } if (estep != 0) { this->CE.update_all_dis(ucell); - this->CE.extrapolate_charge(&this->Pgrid, ucell, &this->chr, &this->sf, GlobalV::ofs_running, GlobalV::ofs_warning); + const module_charge::AtomicRhoCfg atomic_rho_cfg_tddft{PARAM.inp.nelec, + PARAM.inp.test_charge, + PARAM.globalv.domag, + PARAM.globalv.domag_z, + GlobalV::ofs_warning}; + this->CE.extrapolate_charge(&this->Pgrid, + ucell, + &this->chr, + *this->pw_rhod, + &this->sf, + GlobalV::ofs_running, + GlobalV::ofs_warning, + atomic_rho_cfg_tddft, + PARAM.globalv.has_float_data); this->exx_nao.before_scf(ucell, this->kv, this->orb_, this->p_chgmix, totstep, *this->inp_, this->exx_info_); elecstate::init_scf(ucell, this->Pgrid, @@ -254,6 +289,10 @@ void ESolver_KS_LCAO_TDDFT::runner(BaseCell& basecell, const int ist // 7) after_scf //---------------------------------------------------------------- this->after_scf(ucell, totstep, conv_esolver); + if (this->inp_->out_freq_td > 0 && totstep % this->inp_->out_freq_td == 0 && GlobalV::MY_RANK == 0) + { + td_field_manager_->write_restart(PARAM.globalv.global_out_dir); + } if (!restart_done && this->inp_->mdp.md_restart) { restart_done = true; @@ -365,20 +404,15 @@ void ESolver_KS_LCAO_TDDFT::hamilt2rho_single(UnitCell& ucell, const this->inp_->device == "gpu", GlobalV::NPROC, GlobalV::MY_RANK); - hsolver_lcao_obj.solve(static_cast>*>(this->p_hamilt), - this->psi[0], - this->pelec, - *this->dmat.dm, - this->chr, - this->inp_->nspin, - skip_charge); + hamilt::HamiltHSMatrix> hs(static_cast>*>(this->p_hamilt)); + hsolver_lcao_obj.solve(hs, this->psi[0], this->pelec, *this->dmat.dm, this->chr, this->inp_->nspin, ucell.omega, skip_charge); } } // Symmetrize the charge density only for ground state if (istep <= 1) { - Symmetry_rho::symmetrize_rho(this->inp_->nspin, this->chr, this->pw_rho, ucell.symm); + module_charge::symmetrize_rho(this->inp_->nspin, this->chr, this->pw_rho, ucell.symm); } #ifdef __EXX if (this->exx_info_.info_ri.real_number) @@ -430,14 +464,14 @@ void ESolver_KS_LCAO_TDDFT::iter_finish(UnitCell& ucell, { if (use_tensor && use_lapack) { - elecstate::cal_edm_tddft_tensor_lapack(this->pv, - this->dmat, - this->kv, - static_cast>*>(this->p_hamilt)); + module_dm::edm_tddft_lapack(this->pv, + this->dmat, + this->kv, + static_cast>*>(this->p_hamilt)); } else { - elecstate::cal_edm_tddft(this->pv, this->dmat, this->kv, static_cast>*>(this->p_hamilt)); + module_dm::edm_tddft(this->pv, this->dmat, this->kv, static_cast>*>(this->p_hamilt)); } } } @@ -605,18 +639,18 @@ void ESolver_KS_LCAO_TDDFT::weight_dm_rho(const UnitCell& ucell) // Calculate Eband energy elecstate::calEBand(this->pelec->ekb, this->pelec->wg, this->pelec->f_en); - elecstate::cal_dm_psi(this->dmat.dm->get_paraV_pointer(), this->pelec->wg, this->psi[0], *this->dmat.dm); + module_dm::dm_from_psi(&this->pv, this->pelec->wg, this->psi[0], *this->dmat.dm); if (this->inp_->td_stype == 2) { - this->dmat.dm->cal_DMR_td(td_p->get_phase_hybrid(), TD_info::cart_At); + this->dmat.dm->cal_dmr_td(td_p->get_phase_hybrid(), TD_info::A_prop_ha, -1); } else { - this->dmat.dm->cal_DMR(); + this->dmat.dm->cal_dmr(-1); } // get the real-space charge density, mohan add 2025-10-24 - LCAO_domain::dm2rho(this->dmat.dm->get_DMR_vector(), this->inp_->nspin, &this->chr); + LCAO_domain::dm2rho(this->dmat.dm->get_dmr_vec(), this->inp_->nspin, &this->chr, this->inp_->nelec, ucell.omega, false); } template class ESolver_KS_LCAO_TDDFT; diff --git a/source/source_esolver/esolver_ks_lcaopw.cpp b/source/source_esolver/esolver_ks_lcaopw.cpp index 49cb7ea6b1b..2a9968f1737 100644 --- a/source/source_esolver/esolver_ks_lcaopw.cpp +++ b/source/source_esolver/esolver_ks_lcaopw.cpp @@ -4,7 +4,7 @@ #include //--------------temporary---------------------------- -#include "source_estate/module_charge/symm_rho.h" +#include "source_estate/module_charge/chg_symm.h" #include "source_estate/occupy.h" #include "source_hamilt/module_ewald/h_ewald_pw.h" //-----force------------------- @@ -29,7 +29,7 @@ #include #include #ifdef __LCAO -#include "source_io/module_hs/write_vxc_lip.hpp" +#include "source_io/module_hs/vxc_op_lip.h" #endif namespace ModuleESolver @@ -155,16 +155,19 @@ namespace ModuleESolver #else const hsolver::diag_comm_info diag_comm(0, 1); #endif - hsolver_lip_obj.solve(static_cast*>(this->p_hamilt), + // the solver sees the Hamiltonian only through this operator; the EXX + // subspace hooks live in the adapter, not in the solver + hamilt::HamiltLIPHSOperator op(dynamic_cast*>(this->p_hamilt), + this->pw_wfc, + this->general_exx_info_.cal_exx, + this->general_exx_info_.hybrid_alpha); + hsolver_lip_obj.solve(op, *this->stp.template get_psi_t(), this->pelec, *this->psi_local, diag_comm, GlobalV::ofs_running, - skip_charge, - ucell.tpiba, - ucell.nat, - this->general_exx_info_); + skip_charge); // add exx #ifdef __EXX @@ -176,7 +179,7 @@ namespace ModuleESolver } #endif - Symmetry_rho::symmetrize_rho(this->inp_->nspin, this->chr, this->pw_rhod, ucell.symm); + module_charge::symmetrize_rho(this->inp_->nspin, this->chr, this->pw_rhod, ucell.symm); // deband is calculated from "output" charge density calculated // in sum_band @@ -281,7 +284,7 @@ namespace ModuleESolver bool cal_exx = false; double hybrid_alpha = 0.0; #endif - ModuleIO::write_Vxc(this->inp_->nspin, + ModuleIO::write_Vxc_LIP(this->inp_->nspin, PARAM.globalv.nlocal, GlobalV::DRANK, *this->stp.template get_psi_t(), @@ -295,6 +298,10 @@ namespace ModuleESolver this->chr, this->kv, this->pelec->wg, + PARAM.globalv.gamma_only_local, + PARAM.globalv.global_out_dir, + this->inp_->out_ndigits, + this->inp_->ks_solver, cal_exx, hybrid_alpha #ifdef __EXX diff --git a/source/source_esolver/esolver_ks_pw.cpp b/source/source_esolver/esolver_ks_pw.cpp index 1c8dc1b798f..0a49659661a 100644 --- a/source/source_esolver/esolver_ks_pw.cpp +++ b/source/source_esolver/esolver_ks_pw.cpp @@ -1,9 +1,10 @@ #include "esolver_ks_pw.h" #include "source_estate/elecstate_pw.h" -#include "source_estate/module_charge/symm_rho.h" +#include "source_estate/module_charge/chg_symm.h" #include "source_hsolver/diago_iter_assist.h" #include "source_hsolver/diago_params.h" +#include "source_hamilt/hamilt_hs_adapter.h" #include "source_hsolver/hsolver_pw.h" #include "source_io/module_parameter/parameter.h" #include "source_pw/module_pwdft/force_pw.h" @@ -14,7 +15,7 @@ #include "source_base/kernels/dsp/dsp_connector.h" #endif -#include "source_estate/module_charge/chgmixing.h" // use charge mixing, mohan add 20251006 +#include "source_estate/module_charge/chg_routine.h" // use charge mixing, mohan add 20251006 #include "source_estate/setup_estate_pw.h" // mohan add 20251005 #include "source_hamilt/module_xc/general_exx_info.h" // for General_Exx_Info type used via general_exx_info_ #include "source_io/module_ctrl/ctrl_output_pw.h" // mohan add 20250927 @@ -196,7 +197,16 @@ void ESolver_KS_PW::iter_init(UnitCell& ucell, const int istep, const { ESolver_KS::iter_init(ucell, istep, iter); - module_charge::chgmixing_ks_pw(iter, this->p_chgmix, *this->dftu_, *this->inp_); + // query DeltaSpin convergence for U-ramping; treat as converged when disabled + bool mag_converged = true; + if (this->inp_->sc_mag_switch) + { + spinconstrain::SpinConstrain>& sc + = spinconstrain::SpinConstrain>::getScInstance(); + mag_converged = sc.mag_converged(); + } + + module_charge::chgmixing_ks_pw(iter, this->p_chgmix, *this->dftu_, mag_converged, *this->inp_); // mohan move harris functional here, 2012-06-05 // use 'rho(in)' and 'v_h and v_xc'(in) @@ -264,20 +274,20 @@ void ESolver_KS_PW::hamilt2rho_single(UnitCell& ucell, const int iste this->inp_->nb2d, this->inp_->use_k_continuity); - hsolver_pw_obj.solve(static_cast*>(this->p_hamilt), + // the iterative eigensolvers see the Hamiltonian only through this operator + hamilt::HamiltHSOperator op(static_cast*>(this->p_hamilt), this->pw_wfc); + hsolver_pw_obj.solve(op, *this->stp.template get_psi_t(), this->pelec, this->pelec->ekb.c, GlobalV::RANK_IN_POOL, GlobalV::NPROC_IN_POOL, GlobalV::ofs_running, - skip_charge, - ucell.tpiba, - ucell.nat); + skip_charge); } // symmetrize the charge density - Symmetry_rho::symmetrize_rho(this->inp_->nspin, this->chr, this->pw_rhod, ucell.symm); + module_charge::symmetrize_rho(this->inp_->nspin, this->chr, this->pw_rhod, ucell.symm); ModuleBase::timer::end("ESolver_KS_PW", "hamilt2rho_single"); } @@ -323,7 +333,7 @@ void ESolver_KS_PW::iter_finish(UnitCell& ucell, const int istep, int pw::check_deltaspin_oscillation(iter, this->drho, this->p_chgmix, *this->inp_); // the output quantities - ModuleIO::ctrl_iter_pw(istep, iter, conv_esolver, this->stp.psi_cpu, this->kv, this->pw_wfc, *this->inp_); + ModuleIO::ctrl_iter_pw(istep, iter, conv_esolver, this->stp, this->kv, this->pw_wfc, *this->inp_); } template @@ -378,7 +388,9 @@ void ESolver_KS_PW::cal_force(BaseCell& basecell, ModuleBase::matrix& this->stp.update_psi_d(); // Calculate forces - ff.cal_force(ucell, + ff.cal_force(this->inp_->nspin, PARAM.globalv.domag, PARAM.globalv.domag_z, this->inp_->gga_grad, + this->inp_->dft_plus_u || this->inp_->sc_mag_switch, + ucell, force, this->get_vdw_result(), *this->pelec, @@ -405,7 +417,9 @@ void ESolver_KS_PW::cal_stress(BaseCell& basecell, ModuleBase::matrix // mohan add 2025-10-12 this->stp.update_psi_d(); - ss.cal_stress(stress, + ss.cal_stress(this->inp_->nspin, PARAM.globalv.domag, PARAM.globalv.domag_z, + this->inp_->gga_grad, PARAM.globalv.gamma_only_pw, + stress, ucell, this->get_vdw_result(), *this->dftu_, diff --git a/source/source_esolver/esolver_ks_pw_tddft.cpp b/source/source_esolver/esolver_ks_pw_tddft.cpp new file mode 100644 index 00000000000..5fab0adc302 --- /dev/null +++ b/source/source_esolver/esolver_ks_pw_tddft.cpp @@ -0,0 +1,351 @@ +#include "source_esolver/esolver_ks_pw_tddft.h" + +#include "source_base/global_function.h" +#include "source_base/global_variable.h" +#include "source_base/parallel_comm.h" +#include "source_base/parallel_reduce.h" +#include "source_base/timer.h" +#include "source_estate/elecstate_pw.h" +#include "source_estate/elecstate_tools.h" +#include "source_estate/module_charge/chg_symm.h" +#include "source_estate/module_pot/h_tddft_pw.h" +#include "source_hamilt/hamilt_hs_adapter.h" +#include "source_hamilt/module_xc/xc_functional.h" +#include "source_io/module_efield/td_efield_io.h" +#include "source_io/module_output/output_log.h" +#include "source_io/module_parameter/parameter.h" +#include "source_pw/module_pwdft/hamilt_pw.h" +#include "source_pw/module_pwdft/td_pw.h" + +#include + +namespace ModuleESolver +{ + +template +ESolver_KS_PW_TDDFT::ESolver_KS_PW_TDDFT() : log_(GlobalV::ofs_running) +{ + this->classname = "ESolver_KS_PW_TDDFT"; + this->basisname = "PW"; +} + +template +void ESolver_KS_PW_TDDFT::before_all_runners(BaseCell& basecell, const Input_para& inp) +{ + basecell.require_kind(BaseCell::Kind::unitcell, __FUNCTION__); + UnitCell& ucell = static_cast(basecell); + + this->td_field_manager_ = elecstate::create_td_field_manager(inp); + elecstate::H_TDDFT_pw::set_field_state(*this->td_field_manager_); + + ESolver_KS_PW::before_all_runners(ucell, inp); + // The parent resolves the XC functional, including pseudopotential defaults. + pw::check_td_input(inp, ucell, XC_Functional::get_ked_flag(), this->general_exx_info_.cal_exx); + this->pelec->pot->set_td_field_manager(this->td_field_manager_); +#ifdef __MPI + const hsolver::diag_comm_info comm(this->pw_wfc->pool_world, this->pw_wfc->poolrank, this->pw_wfc->poolnproc); +#else + const hsolver::diag_comm_info comm(0, 1); +#endif + hsolver::PWLinearOptions options; + options.orthonormal = hsolver::parse_orth_method(inp.td_orthonormal); + options.out_stat = inp.td_out_stat; + options.global_k_indices = this->kv.ik2iktot; + options.linear.method = hsolver::parse_linear_method(inp.lin_solver); + options.linear.tolerance = inp.lin_thr; + options.linear.max_iterations = inp.lin_maxiter; + options.linear.restart = inp.lin_gmres_restart; + options.linear.reconstruct = inp.lin_reconstruct; + options.preconditioner = hsolver::parse_pw_precond(inp.lin_precond); + options.cn_init = inp.td_cn_init; + options.kinetic_enabled = inp.t_in_h; + this->td_solver_.reset(new hsolver::HSolverPWTDDFT(*this->pw_wfc, options, comm, log_)); + this->history_.prepare(*this->pelec->pot, XC_Functional::get_ked_flag()); + // Preserve existing field history until input validation and initialization succeed. + if (inp.out_efield && GlobalV::MY_RANK == 0) + { + ModuleIO::prepare_td_field_output(PARAM.globalv.global_out_dir, this->td_field_manager_->fields().size(), false); + } +} + +template +void ESolver_KS_PW_TDDFT::before_scf(UnitCell& ucell, const int istep) +{ + this->prepare_td_step(istep); + this->td_solver_->reset_orth_stats(); + const bool basis_updated = ucell.cell_parameter_updated; + ESolver_KS_PW::before_scf(ucell, istep); + if (basis_updated) + { + this->td_solver_->invalidate_basis(); + } + this->history_.prepare(*this->pelec->pot, XC_Functional::get_ked_flag()); + if (this->td_field_manager_->gauge() == 1) + { + // Refresh after the parent updates the cell and distributed basis. + const double q_unshifted = pw::td_momentum_bound(*this->pw_wfc, ucell.tpiba); + pw::ensure_td_vnl(ucell, q_unshifted, this->td_field_manager_->A_right_ha(), this->td_field_manager_->A_prop_ha(), &this->ppcell); + } +} + +template +void ESolver_KS_PW_TDDFT::prepare_td_step(const int istep) +{ + ModuleBase::timer::start("ESolver_KS_PW_TDDFT", "prepare_td_step"); + if (this->td_field_manager_->current_step() == istep) + { + ModuleBase::timer::end("ESolver_KS_PW_TDDFT", "prepare_td_step"); + return; + } + if (this->td_field_manager_->gauge() == 0 || istep == 0) + { + this->td_field_manager_->prepare_sample(istep); + } + else + { + const int interval_left = istep - 1; + this->td_field_manager_->prepare_interval(interval_left, istep); + } + elecstate::H_TDDFT_pw::set_field_state(*this->td_field_manager_); + if (this->inp_->out_efield && GlobalV::MY_RANK == 0) + { + ModuleIO::write_td_field_values(*this->td_field_manager_, PARAM.globalv.global_out_dir); + } + ModuleBase::timer::end("ESolver_KS_PW_TDDFT", "prepare_td_step"); +} + +template +std::string ESolver_KS_PW_TDDFT::diag_policy(const int istep) const +{ + if (istep == 0) + { + return "ksdft"; + } + return ESolver_KS::diag_policy(istep); +} + +template +void ESolver_KS_PW_TDDFT::iter_init(UnitCell& ucell, const int istep, const int iter) +{ + // With estep_per_md=1, the ionic and electronic step indices coincide. + ModuleIO::write_head_td(GlobalV::ofs_running, istep, istep, iter, this->basisname); + ESolver_KS_PW::iter_init(ucell, istep, iter); +} + +template +ESolver_KS::DensityStage ESolver_KS_PW_TDDFT::density_stage(const int istep, const int iter) const +{ + if (istep > 0 && iter == 1) + { + return ESolver_KS::DensityStage::predictor; + } + return ESolver_KS::DensityStage::standard; +} + +template +void ESolver_KS_PW_TDDFT::hamilt2rho_single(UnitCell& ucell, const int istep, const int iter, const double ethr) +{ + ModuleBase::timer::start("ESolver_KS_PW_TDDFT", "hamilt2rho_single"); + if (istep == 0) + { + ESolver_KS_PW::hamilt2rho_single(ucell, istep, iter, ethr); + if (this->inp_->td_orthonormal != "none") + { + psi::Psi* current = this->stp.template get_psi_t(); + int changed = this->td_solver_->correct_initial(current, iter); + // Occupations and density construction contain collectives across k-point pools. + Parallel_Reduce::reduce_all(changed); + if (changed > 0) + { + hamilt::Hamilt* hamiltonian = static_cast*>(this->p_hamilt); + hamilt::HamiltHSOperator op(hamiltonian, this->pw_wfc); + this->td_solver_->cal_band_energy(op, *current, &this->pelec->ekb); + elecstate::calculate_weights(this->pelec->ekb, + this->pelec->wg, + this->pelec->klist, + this->pelec->eferm, + this->pelec->f_en, + this->pelec->nelec_spin, + this->inp_->nbands, + this->pelec->skip_weights); + elecstate::calEBand(this->pelec->ekb, this->pelec->wg, this->pelec->f_en); + elecstate::ElecStatePW* estate = static_cast*>(this->pelec); + estate->psiToRho(*current); + module_charge::symmetrize_rho(this->inp_->nspin, this->chr, this->pw_rhod, ucell.symm); + } + } + ModuleBase::timer::end("ESolver_KS_PW_TDDFT", "hamilt2rho_single"); + return; + } + + ModuleBase::TITLE("ESolver_KS_PW_TDDFT", "hamilt2rho_single (RT-TDDFT)"); + psi::Psi* current = this->stp.template get_psi_t(); + elecstate::Potential& potential = *this->pelec->pot; + hamilt::HamiltPW* hamiltonian = static_cast*>(this->p_hamilt); + hamilt::HamiltHSOperator op(hamiltonian, this->pw_wfc); + + // Predict with the current potential, then correct with the midpoint. + const Real* endpoint_veff = potential.template get_veff_smooth_data(); + const Real* endpoint_vofk = potential.template get_vofk_smooth_data(); + const ESolver_KS::DensityStage stage = this->density_stage(istep, iter); + const bool corrector = stage != ESolver_KS::DensityStage::predictor; + const pw::TDPotentialView propagation = this->history_.prop_pot(potential, XC_Functional::get_ked_flag(), corrector); + ModuleBase::Vector3 momentum_shift(0.0, 0.0, 0.0); + if (this->td_field_manager_->gauge() == 1) + { + momentum_shift = this->td_field_manager_->A_prop_ha(); + } + hamiltonian->bind_td_state(propagation.veff, propagation.vofk, momentum_shift); + // Pass the electronic step; it coincides with the MD step while estep_per_md=1. + this->td_solver_->solve(op, + this->history_.previous(), + current, + this->td_field_manager_->dt_ha(), + momentum_shift, + istep, + iter, + this->inp_->out_level == "ie"); + + // Restore the endpoint Hamiltonian before evaluating density and energy. + elecstate::H_TDDFT_pw::set_field_state(*this->td_field_manager_); + hamiltonian->bind_td_state(endpoint_veff, endpoint_vofk, this->td_field_manager_->A_right_ha()); + elecstate::ElecStatePW* estate = static_cast*>(this->pelec); + // Propagation keeps the occupations of the initial state fixed. + estate->psiToRho(*current); + module_charge::symmetrize_rho(this->inp_->nspin, this->chr, this->pw_rhod, ucell.symm); + + // Pair the band sum with KS double-counting terms using this same potential. + // The later KS potential update belongs to the next SCF iteration. + this->td_solver_->cal_band_energy(op, *current, &this->pelec->ekb); + elecstate::calEBand(this->pelec->ekb, this->pelec->wg, this->pelec->f_en); + ModuleBase::timer::end("ESolver_KS_PW_TDDFT", "hamilt2rho_single"); +} + +template +void ESolver_KS_PW_TDDFT::iter_finish(UnitCell& ucell, const int istep, int& iter, bool& conv_esolver) +{ + ESolver_KS_PW::iter_finish(ucell, istep, iter, conv_esolver); + if (istep > 0 && !conv_esolver && iter >= this->inp_->scf_nmax) + { + ModuleBase::WARNING_QUIT("ESolver_KS_PW_TDDFT", "The time-dependent SCF step did not converge."); + } +} + +template +void ESolver_KS_PW_TDDFT::after_scf(UnitCell& ucell, const int istep, const bool conv_esolver) +{ + if (!conv_esolver) + { + ModuleBase::WARNING_QUIT("ESolver_KS_PW_TDDFT", "Cannot propagate an unconverged electronic state."); + } + if (istep == 0 && this->inp_->td_orthonormal == "none") + { + const psi::Psi* current = this->stp.template get_psi_t(); + this->td_solver_->check_initial(*current, this->niter); + } + ESolver_KS_PW::after_scf(ucell, istep, conv_esolver); + if (istep >= 0) + { + psi::Psi* current = this->stp.template get_psi_t(); + this->history_.save(*current, *this->pelec->pot, XC_Functional::get_ked_flag()); + + if (this->inp_->out_current == 1) + { + const ModuleBase::Vector3& A_right_ha = this->td_field_manager_->A_right_ha(); + this->current_output_.write(istep, + ucell, + this->pw_wfc, + current, + this->pelec, + this->kv, + &this->ppcell, + this->td_field_manager_->gauge(), + A_right_ha, + this->inp_->out_current_k, + PARAM.globalv.global_out_dir, + GlobalV::MY_RANK); + } + } + if (this->inp_->td_out_stat) + { + report_orth(ucell, istep); + } +} + +template +void ESolver_KS_PW_TDDFT::report_orth(const UnitCell& ucell, int istep) +{ + ModuleBase::timer::start("ESolver_KS_PW_TDDFT", "report_orth"); + double electrons = this->td_solver_->wave_electrons(this->pelec->wg); + // Norms are replicated within a pool; only its root contributes to the total. + if (this->pw_wfc->poolrank != 0) + { + electrons = 0.0; + } + Parallel_Reduce::reduce_all(electrons); + double rho_electrons = 0.0; + const int density_spins = this->chr.nspin == 2 ? 2 : 1; + for (int spin = 0; spin < density_spins; ++spin) + { + for (int ir = 0; ir < this->chr.nrxx; ++ir) + { + rho_electrons += this->chr.rho[spin][ir]; + } + } + rho_electrons *= ucell.omega / this->chr.rhopw->nxyz; + Parallel_Reduce::reduce_pool(rho_electrons); + if (istep == 0) + { + initial_wave_electrons_ = electrons; + initial_rho_electrons_ = rho_electrons; + } + const hsolver::TDOrthStats& stats = this->td_solver_->orth_stats(); + const bool full_gram = this->inp_->td_orthonormal != "none"; + double maxima[] = {stats.before, stats.after}; + if (full_gram) + { + Parallel_Reduce::reduce_max(maxima, 2); + } + if (!log_.good()) + { + ModuleBase::timer::end("ESolver_KS_PW_TDDFT", "report_orth"); + return; + } + const char* stage = istep == 0 ? "initial state" : "propagation"; + const int evolution_step = istep + 1; + const double delta_wave = electrons - initial_wave_electrons_; + const double delta_rho = rho_electrons - initial_rho_electrons_; + const std::ios::fmtflags saved_flags = log_.flags(); + const std::streamsize saved_precision = log_.precision(); + log_ << " PW RT-TDDFT conservation: evolution step " << evolution_step << ", " << stage << ", " << this->inp_->td_orthonormal << '\n'; + log_ << std::right << std::fixed << std::setprecision(10); + ModuleBase::GlobalFunc::OUT(log_, "Npsi", electrons); + ModuleBase::GlobalFunc::OUT(log_, "Nrho", rho_electrons); + log_ << std::scientific << std::setprecision(6); + ModuleBase::GlobalFunc::OUT(log_, "dNpsi", delta_wave); + ModuleBase::GlobalFunc::OUT(log_, "dNrho", delta_rho); + if (full_gram) + { + if (istep == 0) + { + ModuleBase::GlobalFunc::OUT(log_, "initial_orth_error", maxima[1]); + } + else + { + ModuleBase::GlobalFunc::OUT(log_, "orth_before_step_max", maxima[0]); + ModuleBase::GlobalFunc::OUT(log_, "orth_after_step_max", maxima[1]); + } + } + log_.flags(saved_flags); + log_.precision(saved_precision); + ModuleBase::timer::end("ESolver_KS_PW_TDDFT", "report_orth"); +} + +template class ESolver_KS_PW_TDDFT, base_device::DEVICE_CPU>; +template class ESolver_KS_PW_TDDFT, base_device::DEVICE_CPU>; +#if defined(__CUDA) || defined(__ROCM) +template class ESolver_KS_PW_TDDFT, base_device::DEVICE_GPU>; +template class ESolver_KS_PW_TDDFT, base_device::DEVICE_GPU>; +#endif + +} // namespace ModuleESolver diff --git a/source/source_esolver/esolver_ks_pw_tddft.h b/source/source_esolver/esolver_ks_pw_tddft.h new file mode 100644 index 00000000000..4712e6f0208 --- /dev/null +++ b/source/source_esolver/esolver_ks_pw_tddft.h @@ -0,0 +1,63 @@ +#ifndef ESOLVER_KS_PW_TDDFT_H +#define ESOLVER_KS_PW_TDDFT_H + +#include "source_esolver/esolver_ks_pw.h" +#include "source_estate/module_pot/td_field_manager.h" +#include "source_hsolver/hsolver_pw_tddft.h" +#include "source_io/module_current/td_current_pw.h" +#include "source_pw/module_pwdft/td_history_pw.h" + +#include +#include + +namespace ModuleESolver +{ + +template +class ESolver_KS_PW_TDDFT : public ESolver_KS_PW +{ + public: + ESolver_KS_PW_TDDFT(); + ~ESolver_KS_PW_TDDFT() override = default; + + /** @brief Initialize the field, propagation solver and time-step history. */ + void before_all_runners(BaseCell& basecell, const Input_para& inp) override; + + protected: + /** @brief Use ground-state diagonalization for the initial electronic step. */ + std::string diag_policy(const int istep) const override; + + /** @brief Require an unmixed endpoint prediction before accepting a corrected step. */ + ESolver_KS::DensityStage density_stage(const int istep, const int iter) const override; + + void before_scf(UnitCell& ucell, const int istep) override; + + /** @brief Print the TDDFT iteration header and initialize the PW iteration. */ + void iter_init(UnitCell& ucell, const int istep, const int iter) override; + + /** @brief Switch between ground-state diagonalization and real-time propagation. */ + void hamilt2rho_single(UnitCell& ucell, const int istep, const int iter, const double ethr) override; + + /** @brief Apply the convergence policy appropriate to the current stage. */ + void iter_finish(UnitCell& ucell, const int istep, int& iter, bool& conv_esolver) override; + + /** @brief Save the converged state as the origin of the next propagation step. */ + void after_scf(UnitCell& ucell, const int istep, const bool conv_esolver) override; + + private: + using Real = typename GetTypeReal::type; + std::ofstream& log_; + std::unique_ptr> td_solver_; + + std::shared_ptr td_field_manager_; + ModuleIO::CurrentPW current_output_; + + pw::TDHistoryPW history_; + double initial_wave_electrons_ = 0.0; + double initial_rho_electrons_ = 0.0; + void report_orth(const UnitCell& ucell, int istep); + void prepare_td_step(const int istep); +}; + +} // namespace ModuleESolver +#endif diff --git a/source/source_esolver/esolver_lj.h b/source/source_esolver/esolver_lj.h index b787aa7cb69..8f8ca3056b5 100644 --- a/source/source_esolver/esolver_lj.h +++ b/source/source_esolver/esolver_lj.h @@ -35,6 +35,65 @@ class ESolver_LJ : public ESolver void others(BaseCell& cell, const int istep) override; + //==================================================================== + // Test seam. + // + // before_all_runners() derives the LJ tables in three steps. The unit tests + // drive each step on its own and check the table it produced, so the steps + // and the tables are reachable here rather than by reinterpreting the access + // specifiers. + // + // Production code must keep going through before_all_runners(); nothing + // outside the tests should call the *_for_testing() wrappers. + //==================================================================== + + /// @brief neighbour search radius derived from the largest cutoff + double get_search_radius() const + { + return search_radius; + } + /// @brief per-type-pair cutoff radii + const ModuleBase::matrix& get_lj_rcut() const + { + return lj_rcut; + } + /// @brief per-type-pair c6 coefficients + const ModuleBase::matrix& get_lj_c6() const + { + return lj_c6; + } + /// @brief per-type-pair c12 coefficients + const ModuleBase::matrix& get_lj_c12() const + { + return lj_c12; + } + /// @brief per-type-pair energy shift at the cutoff + const ModuleBase::matrix& get_en_shift() const + { + return en_shift; + } + /// @brief computed lattice virials + const ModuleBase::matrix& get_lj_virial() const + { + return lj_virial; + } + + void rcut_search_radius_for_testing(const int& ntype, const std::vector& rcut) + { + rcut_search_radius(ntype, rcut); + } + void set_c6_c12_for_testing(const int& ntype, + const int& rule, + const std::vector& epsilon, + const std::vector& sigma) + { + set_c6_c12(ntype, rule, epsilon, sigma); + } + void cal_en_shift_for_testing(const int& ntype, const bool& is_shift) + { + cal_en_shift(ntype, is_shift); + } + private: double LJ_energy(const double& d, const int& i, const int& j) const; diff --git a/source/source_esolver/esolver_lr_lcao_bse.cpp b/source/source_esolver/esolver_lr_lcao_bse.cpp index 3deb4d4328a..01506916870 100644 --- a/source/source_esolver/esolver_lr_lcao_bse.cpp +++ b/source/source_esolver/esolver_lr_lcao_bse.cpp @@ -1,5 +1,6 @@ #include "esolver_lr_lcao_bse.h" #include +#include "source_basis/module_pw/pw_basis_big.h" #include "source_cell/module_neighbor/sltk_atom_arrange.h" #include "source_io/module_parameter/parameter.h" #include "source_io/module_output/print_info.h" diff --git a/source/source_esolver/esolver_lr_lcao_bse.h b/source/source_esolver/esolver_lr_lcao_bse.h index 5cca6850ef8..771df590f44 100644 --- a/source/source_esolver/esolver_lr_lcao_bse.h +++ b/source/source_esolver/esolver_lr_lcao_bse.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_ESOLVER_ESOLVER_LR_LCAO_BSE_H +#define ABACUS_SOURCE_ESOLVER_ESOLVER_LR_LCAO_BSE_H + #include "esolver_lr_lcao_tddft.h" #include "source_lcao/module_bse/molecular_lri.h" @@ -53,3 +55,5 @@ namespace ModuleESolver }; } + +#endif // ABACUS_SOURCE_ESOLVER_ESOLVER_LR_LCAO_BSE_H diff --git a/source/source_esolver/esolver_lr_lcao_tddft.cpp b/source/source_esolver/esolver_lr_lcao_tddft.cpp index c02a39cc6bd..e754ab02041 100644 --- a/source/source_esolver/esolver_lr_lcao_tddft.cpp +++ b/source/source_esolver/esolver_lr_lcao_tddft.cpp @@ -1,10 +1,12 @@ #include "esolver_lr_lcao_tddft.h" +#include "source_basis/module_pw/pw_basis_big.h" // use PW_Basis_Big #include "source_lcao/module_lr/utils/lr_io.h" #include "source_lcao/module_lr/utils/lr_util.h" #include "source_lcao/module_lr/hamilt_casida.h" #include "source_lcao/module_lr/hamilt_ulr.hpp" #include "source_lcao/module_lr/potentials/pot_hxc_lrtd.h" #include "source_lcao/lcao_nonlocal_info.h" +#include "source_hamilt/module_xc/xc_functional.h" #include "source_lcao/module_lr/hsolver_lrtd.hpp" #include "source_lcao/module_lr/lr_spectrum.h" #include "source_hamilt/module_gint/gint.h" @@ -17,6 +19,9 @@ #include "source_lcao/module_lr/utils/lr_util_print.h" #include "source_base/module_external/scalapack_connector.h" #include "source_io/module_parameter/parameter.h" +#ifdef __JSON +#include "source_io/module_json/output_info.h" +#endif #include "source_lcao/module_lr/ri_benchmark/ri_benchmark.h" #include "source_lcao/module_lr/operator_casida/operator_lr_diag.h" // for precondition #ifdef __EXX @@ -83,9 +88,9 @@ void ModuleESolver::ESolver_LR::setup_2center_table(TwoCenterBundle& two_ auto* lcao_nl = new LCAONonlocalInfo(); lcao_nl->setupNonlocal(ucell.ntype, ucell.atoms, GlobalV::ofs_running, orb, this->inp_->basis_type, this->inp_->out_element_info, - this->inp_->lspinorb, this->inp_->nspin); + this->inp_->lspinorb, this->inp_->nspin, GlobalV::MY_RANK); ucell.infoNL.reset(lcao_nl); - two_center_bundle.build_beta(ucell.ntype, lcao_nl->get_nonlocal().Beta); + two_center_bundle.build_beta(ucell.ntype, lcao_nl->get_nonlocal().get_Beta_data()); } } @@ -222,6 +227,10 @@ void ModuleESolver::ESolver_LR::before_all_runners(BaseCell& basecell, co this->inp_ = &inp; if (inp.esolver_type == "ks-lr") { +#ifdef __JSON + // The embedded KS run happens before Relax_Driver starts its first step. + Json::init_output_array_obj(); +#endif ModuleESolver::ESolver_KS_LCAO ks_solver; ks_solver.before_all_runners(basecell, inp); ks_solver.runner(basecell, 0); @@ -822,27 +831,21 @@ template void ModuleESolver::ESolver_LR::read_ks_chg(Charge& chg_gs) { chg_gs.set_rhopw(this->pw_rho); - const bool kin_den = chg_gs.kin_density(); // mohan add 20251202 - chg_gs.allocate(this->nspin, kin_den); + const bool kin_den = XC_Functional::get_ked_flag() || (this->inp_->out_elf[0] > 0); // mohan add 20251202 + chg_gs.allocate(this->nspin, kin_den, XC_Functional::get_ked_flag(), this->inp_->test_charge); GlobalV::ofs_running << " try to read charge from file : "; for (int is = 0; is < this->nspin; ++is) { std::stringstream ssc; ssc << this->in_dir << "chgs" << is + 1 << ".cube"; GlobalV::ofs_running << ssc.str() << std::endl; - if (ModuleIO::read_vdata_palgrid(Pgrid, + ModuleIO::read_vdata_palgrid(Pgrid, GlobalV::MY_RANK, GlobalV::ofs_running, ssc.str(), chg_gs.rho[is], - this->ucell_->nat)) { - GlobalV::ofs_running << " Read in the charge density: " << ssc.str() << std::endl; - } else { // prenspin for nspin=4 is not supported currently - ModuleBase::WARNING_QUIT( - "init_rho", - "!!! Couldn't find the charge file !!! The default directory \n of " + ssc.str() +" is OUT.suffix, " - "or you must set read_file_dir \n to a specific directory. "); - } + this->ucell_->nat); + GlobalV::ofs_running << " Read in the charge density: " << ssc.str() << std::endl; } } template class ModuleESolver::ESolver_LR; diff --git a/source/source_esolver/esolver_lr_lcao_tddft.h b/source/source_esolver/esolver_lr_lcao_tddft.h index 5e3a799cb93..ddaec8b8358 100644 --- a/source/source_esolver/esolver_lr_lcao_tddft.h +++ b/source/source_esolver/esolver_lr_lcao_tddft.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_ESOLVER_ESOLVER_LR_LCAO_TDDFT_H +#define ABACUS_SOURCE_ESOLVER_ESOLVER_LR_LCAO_TDDFT_H + #include "source_esolver/esolver_fp.h" #include "source_io/module_parameter/input_parameter.h" #include "source_cell/unitcell.h" @@ -144,3 +146,5 @@ namespace ModuleESolver #endif }; } + +#endif // ABACUS_SOURCE_ESOLVER_ESOLVER_LR_LCAO_TDDFT_H diff --git a/source/source_esolver/esolver_of.cpp b/source/source_esolver/esolver_of.cpp index 50e94b2e558..19b4b876adb 100644 --- a/source/source_esolver/esolver_of.cpp +++ b/source/source_esolver/esolver_of.cpp @@ -3,7 +3,9 @@ #include "source_io/module_parameter/parameter.h" //-----------temporary------------------------- #include "source_base/global_function.h" -#include "source_estate/module_charge/symm_rho.h" +#include "source_estate/module_charge/chg_init.h" +#include "source_estate/module_charge/chg_symm.h" +#include "source_estate/module_charge/chg_tools.h" #include "source_hamilt/module_ewald/h_ewald_pw.h" #include "source_cell/cal_ux.h" #include "source_pw/module_pwdft/force_pw.h" @@ -18,7 +20,6 @@ namespace ModuleESolver ESolver_OF::ESolver_OF() { this->classname = "ESolver_OF"; - this->task_ = new char[60]; } ESolver_OF::~ESolver_OF() @@ -43,7 +44,6 @@ ESolver_OF::~ESolver_OF() delete[] this->nelec_; delete[] this->theta_; - delete[] this->task_; delete this->ptemp_rho_; delete this->kedf_manager_; @@ -81,8 +81,21 @@ void ESolver_OF::before_all_runners(BaseCell& basecell, const Input_para& inp) ModuleBase::WARNING_QUIT("esolver_of", "meta-GGA and Hybrid functionals are not supported by OFDFT."); } - this->chr.init_rho(ucell, this->Pgrid, this->sf.strucFac, ucell.symm, &this->kv); - this->chr.check_rho(); // check the rho + module_charge::InitRhoCfg init_rho_cfg; + init_rho_cfg.init_chg = inp.init_chg; + init_rho_cfg.suffix = inp.suffix; + init_rho_cfg.esolver_type = inp.esolver_type; + init_rho_cfg.global_readin_dir = PARAM.globalv.global_readin_dir; + init_rho_cfg.nelec = inp.nelec; + init_rho_cfg.nbands = inp.nbands; + init_rho_cfg.test_charge = inp.test_charge; + init_rho_cfg.domag = PARAM.globalv.domag; + init_rho_cfg.domag_z = PARAM.globalv.domag_z; + init_rho_cfg.npol = PARAM.globalv.npol; + init_rho_cfg.meta_gga = XC_Functional::get_ked_flag(); + this->chr.init_rho(ucell, this->Pgrid, this->sf.strucFac, ucell.symm, &this->kv, nullptr, init_rho_cfg); + module_charge::check_rho(this->chr.rho, this->chr.nspin, this->chr.rhopw->nrxx, ucell.omega, + this->chr.rhopw->nxyz, inp.nelec); // check the rho // initialize local pseudopotential this->locpp.init_vloc(ucell,pw_rho); @@ -219,8 +232,9 @@ void ESolver_OF::before_opt(const int istep, UnitCell& ucell) delete this->ptemp_rho_; this->ptemp_rho_ = new Charge(); this->ptemp_rho_->set_rhopw(this->pw_rho); - const bool kin_den = this->ptemp_rho_->kin_density(); // mohan add 20251202 - this->ptemp_rho_->allocate(this->inp_->nspin, kin_den); + const bool kin_den = XC_Functional::get_ked_flag() || (this->inp_->out_elf[0] > 0); // mohan add 20251202 + this->ptemp_rho_->allocate(this->inp_->nspin, kin_den, XC_Functional::get_ked_flag(), + this->inp_->test_charge); for (int is = 0; is < this->inp_->nspin; ++is) { @@ -240,7 +254,7 @@ void ESolver_OF::before_opt(const int istep, UnitCell& ucell) const int nspin = this->inp_->nspin; if (this->inp_->init_chg == "file") { - Symmetry_rho::symmetrize_rho(nspin, this->chr, this->pw_rho, ucell.symm); + module_charge::symmetrize_rho(nspin, this->chr, this->pw_rho, ucell.symm); for (int is = 0; is < nspin; ++is) { for (int ibs = 0; ibs < this->pw_rho->nrxx; ++ibs) @@ -385,19 +399,6 @@ void ESolver_OF::update_rho() } } // // ------------ turn on symmetry may cause instability in optimization ------------ - // if (ModuleSymmetry::Symmetry::symm_flag == 1) - // { - // Symmetry_rho srho; - // for (int is = 0; is < this->inp_->nspin; is++) - // { - // srho.begin(is, *(this->chr), this->pw_rho, Pgrid, ucell.symm); - // for (int ibs = 0; ibs < this->pw_rho->nrxx; ++ibs) - // { - // this->pphi_[is][ibs] = sqrt(this->chr.rho[is][ibs]); - // } - // } - // } - // // -------------------------------------------------------------------------------- } /** @@ -532,7 +533,11 @@ void ESolver_OF::after_all_runners(BaseCell& basecell) */ double ESolver_OF::cal_energy() { - this->pelec->cal_energies(2); + this->pelec->cal_energies(2, + this->inp_->imp_sol, + this->inp_->sc_mag_switch, + this->inp_->dft_plus_u, + this->inp_->assume_isolated); double kinetic_energy = this->kedf_manager_->get_energy(); // kinetic energy double pseudopot_energy = 0.; // electron-ion interaction energy for (int is = 0; is < this->inp_->nspin; ++is) @@ -563,7 +568,9 @@ void ESolver_OF::cal_force(BaseCell& basecell, ModuleBase::matrix& force) // here nullptr is for DFT+U, which may cause bugs, mohan note 2025-11-07 // solvent can be used? mohan ask 2025-11-07 - ff.cal_force(ucell, force, this->get_vdw_result(), *pelec, this->pw_rho, &ucell.symm, &sf, + ff.cal_force(this->inp_->nspin, PARAM.globalv.domag, PARAM.globalv.domag_z, this->inp_->gga_grad, + this->inp_->dft_plus_u || this->inp_->sc_mag_switch, + ucell, force, this->get_vdw_result(), *pelec, this->pw_rho, &ucell.symm, &sf, this->solvent, nullptr, &this->locpp); } @@ -583,6 +590,8 @@ void ESolver_OF::cal_stress(BaseCell& basecell, ModuleBase::matrix& stress) this->pphi_, this->pw_rho, kinetic_stress_); // kinetic stress OF_Stress_PW ss(this->pelec, this->pw_rho); - ss.cal_stress(stress, kinetic_stress_, ucell, this->get_vdw_result(), &ucell.symm, this->locpp, &sf, &kv); + ss.cal_stress(this->inp_->nspin, PARAM.globalv.domag, PARAM.globalv.domag_z, + this->inp_->gga_grad, PARAM.globalv.gamma_only_pw, + stress, kinetic_stress_, ucell, this->get_vdw_result(), &ucell.symm, this->locpp, &sf, &kv); } } // namespace ModuleESolver diff --git a/source/source_esolver/esolver_of.h b/source/source_esolver/esolver_of.h index 940a26b7e80..9fb0a6df987 100644 --- a/source/source_esolver/esolver_of.h +++ b/source/source_esolver/esolver_of.h @@ -1,9 +1,11 @@ #ifndef ESOLVER_OF_H #define ESOLVER_OF_H +#include + #include "esolver_fp.h" #include "source_base/opt_dcsrch.h" -#include "source_base/opt_tn.hpp" +#include "source_base/opt_tn.h" #include "source_psi/psi.h" #include "source_pw/module_ofdft/kedf_manager.h" @@ -58,7 +60,7 @@ class ESolver_OF : public ESolver_FP double** pdEdphi_ = nullptr; // dE/dphi double** pdLdphi_ = nullptr; // dL/dphi double** pphi_ = nullptr; // pphi[i] = ppsi.get_pointer(i), which will be freed in ~Psi(). - char* task_ = nullptr; // used in line search + std::string task_; // used in line search int tn_spin_flag_ = -1; // spin flag used in cal_potential, which will be called by opt_tn int max_dcsrch_ = 200; // max no. of line search int flag_ = -1; // flag of TN diff --git a/source/source_esolver/esolver_of_interface.cpp b/source/source_esolver/esolver_of_interface.cpp index 4240db9b94c..fab92fb704f 100644 --- a/source/source_esolver/esolver_of_interface.cpp +++ b/source/source_esolver/esolver_of_interface.cpp @@ -107,11 +107,15 @@ void ESolver_OF::get_step_length(double* dEdtheta, double** ptemp_phi, UnitCell& if (this->inp_->nspin == 1) { int numDC = 0; // iteration number of line search - strcpy(this->task_, "START"); + this->task_ = "START"; while (true) { // update energy - this->pelec->cal_energies(2); + this->pelec->cal_energies(2, + this->inp_->imp_sol, + this->inp_->sc_mag_switch, + this->inp_->dft_plus_u, + this->inp_->assume_isolated); temp_energy = this->pelec->f_en.etot; kinetic_energy = this->kedf_manager_->get_energy(); // kinetic energy pseudopot_energy = this->inner_product(this->pelec->pot->get_fixed_v(), @@ -126,7 +130,7 @@ void ESolver_OF::get_step_length(double* dEdtheta, double** ptemp_phi, UnitCell& numDC++; // decide what to do next according to the output of line search - if (strncmp(this->task_, "FG", 2) == 0) // continue line search + if (this->task_.compare(0, 2, "FG") == 0) // continue line search { // update tempPhi and tempRho for (int i = 0; i < this->pw_rho->nrxx; ++i) @@ -146,17 +150,17 @@ void ESolver_OF::get_step_length(double* dEdtheta, double** ptemp_phi, UnitCell& break; } } - else if (strncmp(this->task_, "CO", 2) == 0) // convergence achieved + else if (this->task_.compare(0, 2, "CO") == 0) // convergence achieved { break; } - else if (strncmp(this->task_, "WA", 2) == 0) // warning of line search + else if (this->task_.compare(0, 2, "WA") == 0) // warning of line search { GlobalV::ofs_warning << "ESolver_OF linesearch: WARNING " << this->task_ << std::endl; std::cout << this->task_ << std::endl; break; } - else if (strncmp(this->task_, "ER", 2) == 0) // ERROR in line search + else if (this->task_.compare(0, 2, "ER") == 0) // ERROR in line search { GlobalV::ofs_warning << "ESolver_OF linesearch: ERROR " << this->task_ << std::endl; std::cout << this->task_ << std::endl; @@ -202,12 +206,12 @@ void ESolver_OF::get_step_length(double* dEdtheta, double** ptemp_phi, UnitCell& // // line search along thetaDir to find thetaAlpha // this->opt_dcsrch_->set_paras(1e-4, 1e-2, 1e-12, 0., - // ModuleBase::PI/maxThetaDir); strcpy(this->task_, "START"); + // ModuleBase::PI/maxThetaDir); this->task_ = "START"; // numDC = 0; // while(true) // { // this->pelec->f_en.calculate_etot(this->pw_rho->nrxx, - // if (strncmp(this->task_, "FG", 2) == 0) + // if (this->task_.compare(0, 2, "FG") == 0) // { // for (int is = 0; is < this->inp_->nspin; ++is) // { @@ -233,17 +237,17 @@ void ESolver_OF::get_step_length(double* dEdtheta, double** ptemp_phi, UnitCell& // endl; break; // } // } - // else if (strncmp(this->task_, "CO", 2) == 0) + // else if (this->task_.compare(0, 2, "CO") == 0) // { // break; // } - // else if (strncmp(this->task_, "WA", 2) == 0) + // else if (this->task_.compare(0, 2, "WA") == 0) // { // GlobalV::ofs_warning << "ESolver_OF linesearch: // WARNING " << this->task_ << std::endl; cout << // this->task_ << endl; break; // } - // else if (strncmp(this->task_, "ER", 2) == 0) + // else if (this->task_.compare(0, 2, "ER") == 0) // { // GlobalV::ofs_warning << "ESolver_OF linesearch: ERROR // " << this->task_ << std::endl; cout << this->task_ << diff --git a/source/source_esolver/esolver_of_tddft.cpp b/source/source_esolver/esolver_of_tddft.cpp index 0b14a19cc82..3c068b2c77c 100644 --- a/source/source_esolver/esolver_of_tddft.cpp +++ b/source/source_esolver/esolver_of_tddft.cpp @@ -3,7 +3,7 @@ #include "source_io/module_parameter/parameter.h" //-----------temporary------------------------- #include "source_base/global_function.h" -#include "source_estate/module_charge/symm_rho.h" +#include "source_estate/module_charge/chg_symm.h" #include "source_hamilt/module_ewald/h_ewald_pw.h" #include "source_cell/cal_ux.h" //-----force------------------- diff --git a/source/source_esolver/esolver_of_tool.cpp b/source/source_esolver/esolver_of_tool.cpp index 5656f88223b..acd04ca260f 100644 --- a/source/source_esolver/esolver_of_tool.cpp +++ b/source/source_esolver/esolver_of_tool.cpp @@ -5,6 +5,7 @@ #include "source_estate/module_pot/gatefield.h" #include "source_io/module_parameter/parameter.h" #include "source_cell/cal_ux.h" +#include "source_hamilt/module_xc/xc_functional.h" namespace ModuleESolver { @@ -89,8 +90,9 @@ void ESolver_OF::allocate_array() delete this->ptemp_rho_; this->ptemp_rho_ = new Charge(); this->ptemp_rho_->set_rhopw(this->pw_rho); - const bool kin_den = this->ptemp_rho_->kin_density(); // mohan add 20251202 - this->ptemp_rho_->allocate(this->inp_->nspin, kin_den); + const bool kin_den = XC_Functional::get_ked_flag() || (this->inp_->out_elf[0] > 0); // mohan add 20251202 + this->ptemp_rho_->allocate(this->inp_->nspin, kin_den, XC_Functional::get_ked_flag(), + this->inp_->test_charge); this->theta_ = new double[this->inp_->nspin]; this->pdLdphi_ = new double*[this->inp_->nspin]; @@ -370,7 +372,11 @@ void ESolver_OF::test_direction(double* dEdtheta, double** ptemp_phi, UnitCell& ptemp_rho_->rho[0][ir] = ptemp_phi[0][ir] * ptemp_phi[0][ir]; } this->cal_dEdtheta(ptemp_phi, ptemp_rho_, ucell, this->theta_, dEdtheta); - this->pelec->cal_energies(2); + this->pelec->cal_energies(2, + this->inp_->imp_sol, + this->inp_->sc_mag_switch, + this->inp_->dft_plus_u, + this->inp_->assume_isolated); temp_energy = this->pelec->f_en.etot; double kinetic_energy = 0.; double pseudopot_energy = 0.; diff --git a/source/source_esolver/esolver_sdft_pw.cpp b/source/source_esolver/esolver_sdft_pw.cpp index 218bda84637..840dd1175bb 100644 --- a/source/source_esolver/esolver_sdft_pw.cpp +++ b/source/source_esolver/esolver_sdft_pw.cpp @@ -3,10 +3,11 @@ #include "source_base/global_variable.h" #include "source_base/memory_recorder.h" #include "source_base/parallel_comm.h" -#include "source_estate/module_charge/symm_rho.h" +#include "source_estate/module_charge/chg_symm.h" #include "source_hsolver/diago_iter_assist.h" #include "source_hsolver/diago_params.h" #include "source_io/module_parameter/parameter.h" +#include "source_pw/module_stodft/sto_hsolver_pw.h" #include "source_pw/module_stodft/sto_dos.h" #include "source_pw/module_stodft/sto_elecond.h" #include "source_pw/module_stodft/sto_forces.h" @@ -100,15 +101,15 @@ void ESolver_SDFT_PW::before_scf(UnitCell& ucell, const int istep) ESolver_KS_PW::before_scf(ucell, istep); delete reinterpret_cast*>(this->p_hamilt); - this->p_hamilt = new hamilt::HamiltSdftPW(this->pelec->pot, - this->pw_wfc, - &this->kv, - &this->ppcell, - &ucell, - PARAM.globalv.npol, - &this->stoche.emin_sto, - &this->stoche.emax_sto); - this->p_hamilt_sto = static_cast*>(this->p_hamilt); + this->p_hamilt = new StoHamiltPW(this->pelec->pot, + this->pw_wfc, + &this->kv, + &this->ppcell, + &ucell, + PARAM.globalv.npol, + &this->stoche.emin_sto, + &this->stoche.emax_sto); + this->p_hamilt_sto = static_cast*>(this->p_hamilt); if (istep > 0 && this->inp_->nbands_sto != 0 && this->inp_->initsto_freq > 0 && istep % this->inp_->initsto_freq == 0) { @@ -153,47 +154,47 @@ void ESolver_SDFT_PW::hamilt2rho_single(UnitCell& ucell, int istep, i bool skip_charge = this->inp_->calculation == "nscf" ? true : false; // hsolver only exists in this function - hsolver::HSolverPW_SDFT hsolver_pw_sdft_obj(&this->kv, - this->pw_wfc, - this->stowf, - this->stoche, - this->p_hamilt_sto, - this->inp_->calculation, - this->inp_->basis_type, - this->inp_->ks_solver, - PARAM.globalv.use_uspp, - this->inp_->nspin, - hsolver::DiagoIterAssist::SCF_ITER, - hsolver::DiagoIterAssist::PW_DIAG_NMAX, - hsolver::DiagoIterAssist::PW_DIAG_THR, - hsolver::DiagoIterAssist::need_subspace, - this->inp_->nbands, - this->inp_->diago_smooth_ethr, - this->inp_->pw_diag_ndim, - this->inp_->diag_subspace, - this->inp_->nb2d, - PARAM.globalv.ks_run, - PARAM.globalv.all_ks_run, - this->inp_->bndpar); - - hsolver_pw_sdft_obj.solve(ucell, - static_cast*>(this->p_hamilt), - *this->stp.template get_psi_t(), - this->stp.psi_cpu[0], - this->pelec, - this->pw_wfc, - this->stowf, - istep, - iter, - GlobalV::ofs_running, - skip_charge); + StoHSolverPW sto_hsolver_pw_obj(&this->kv, + this->pw_wfc, + this->stowf, + this->stoche, + this->p_hamilt_sto, + this->inp_->calculation, + this->inp_->basis_type, + this->inp_->ks_solver, + PARAM.globalv.use_uspp, + this->inp_->nspin, + hsolver::DiagoIterAssist::SCF_ITER, + hsolver::DiagoIterAssist::PW_DIAG_NMAX, + hsolver::DiagoIterAssist::PW_DIAG_THR, + hsolver::DiagoIterAssist::need_subspace, + this->inp_->nbands, + this->inp_->diago_smooth_ethr, + this->inp_->pw_diag_ndim, + this->inp_->diag_subspace, + this->inp_->nb2d, + PARAM.globalv.ks_run, + PARAM.globalv.all_ks_run, + this->inp_->bndpar); + + sto_hsolver_pw_obj.solve(ucell, + static_cast*>(this->p_hamilt), + *this->stp.template get_psi_t(), + this->stp.psi_cpu[0], + this->pelec, + this->pw_wfc, + this->stowf, + istep, + iter, + GlobalV::ofs_running, + skip_charge); // set_diagethr need it - this->esolver_KS_ne = hsolver_pw_sdft_obj.stoiter.KS_ne; + this->esolver_KS_ne = sto_hsolver_pw_obj.stoiter.KS_ne; if (PARAM.globalv.ks_run) { - Symmetry_rho::symmetrize_rho(this->inp_->nspin, this->chr, this->pw_rho, ucell.symm); + module_charge::symmetrize_rho(this->inp_->nspin, this->chr, this->pw_rho, ucell.symm); this->pelec->f_en.deband = this->pelec->cal_delta_eband(ucell); } else @@ -225,7 +226,8 @@ void ESolver_SDFT_PW::cal_force(BaseCell& basecell, ModuleBase::matri Sto_Forces ff(ucell.nat); - ff.cal_stoforce(force, + ff.cal_stoforce(this->inp_->nspin, PARAM.globalv.domag, PARAM.globalv.domag_z, this->inp_->gga_grad, + force, *this->pelec, this->pw_rho, &ucell.symm, @@ -246,7 +248,9 @@ void ESolver_SDFT_PW::cal_stress(BaseCell& basecell, ModuleBase::matr UnitCell& ucell = static_cast(basecell); Sto_Stress_PW ss; - ss.cal_stress(stress, + ss.cal_stress(this->inp_->nspin, PARAM.globalv.domag, PARAM.globalv.domag_z, + this->inp_->gga_grad, PARAM.globalv.gamma_only_pw, + stress, *this->pelec, this->pw_rho, &ucell.symm, diff --git a/source/source_esolver/esolver_sdft_pw.h b/source/source_esolver/esolver_sdft_pw.h index aa48f97db7e..2a5c4965c3a 100644 --- a/source/source_esolver/esolver_sdft_pw.h +++ b/source/source_esolver/esolver_sdft_pw.h @@ -2,7 +2,7 @@ #define ESOLVER_SDFT_PW_H #include "esolver_ks_pw.h" -#include "source_pw/module_stodft/hamilt_sdft_pw.h" +#include "source_pw/module_stodft/sto_hamilt_pw.h" #include "source_pw/module_stodft/sto_che.h" #include "source_pw/module_stodft/sto_iter.h" #include "source_pw/module_stodft/sto_wf.h" @@ -31,7 +31,7 @@ class ESolver_SDFT_PW : public ESolver_KS_PW public: Stochastic_WF stowf; StoChe stoche; - hamilt::HamiltSdftPW* p_hamilt_sto = nullptr; + StoHamiltPW* p_hamilt_sto = nullptr; protected: virtual void before_scf(UnitCell& ucell, const int istep) override; diff --git a/source/source_esolver/lcao_others.cpp b/source/source_esolver/lcao_others.cpp index 44b769db638..a0fe906a5a3 100644 --- a/source/source_esolver/lcao_others.cpp +++ b/source/source_esolver/lcao_others.cpp @@ -1,16 +1,18 @@ #include "source_base/formatter.h" #include "source_base/timer.h" +#include "source_basis/module_pw/pw_basis_big.h" // use PW_Basis_Big #include "source_cell/cal_ux.h" #include "source_cell/module_neighbor/sltk_atom_arrange.h" #include "source_cell/module_neighbor/sltk_grid_driver.h" #include "source_esolver/esolver_ks_lcao.h" #include "source_estate/elecstate_lcao.h" -#include "source_estate/module_charge/symm_rho.h" -#include "source_estate/module_dm/cal_dm_psi.h" +#include "source_estate/module_charge/chg_symm.h" +#include "source_estate/module_dm/dm_from_psi.h" #include "source_hamilt/module_gint/gint.h" #include "source_io/module_chgpot/get_pchg_lcao.h" -#include "source_io/module_hs/write_hs_r.h" +#include "source_io/module_hs/hsr_legacy.h" #include "source_io/module_parameter/parameter.h" +#include "source_io/module_restart/restart.h" // GlobalC::restart for load_exx_flag #include "source_io/module_wf/get_wf_lcao.h" #include "source_lcao/hamilt_lcao.h" #include "source_lcao/lcao_domain.h" @@ -110,7 +112,11 @@ void ESolver_KS_LCAO::others(BaseCell& basecell, const int istep) // (2)For each atom, calculate the adjacent atoms in different cells // and allocate the space for H(R) and S(R). // If k point is used here, allocate HlocR after atom_arrange. - this->RA.for_2d(ucell, this->gd, this->pv, gamma_only_local, orb_.cutoffs()); + this->RA.for_2d(ucell, this->gd, this->pv, gamma_only_local, PARAM.globalv.npol, orb_.cutoffs()); + if (this->inp_->out_level != "m" && !gamma_only_local) + { + ModuleBase::GlobalFunc::OUT(GlobalV::ofs_running, "ParaV.nnr", this->pv.nnr); + } // 2. density matrix extrapolation @@ -130,6 +136,8 @@ void ESolver_KS_LCAO::others(BaseCell& basecell, const int istep) } if (this->p_hamilt == nullptr) { + const bool load_exx_flag = !GlobalC::restart.info_load.restart_exx + && GlobalC::restart.info_load.load_H; this->p_hamilt = new hamilt::HamiltLCAO(ucell, this->gd, &this->pv, @@ -143,7 +151,8 @@ void ESolver_KS_LCAO::others(BaseCell& basecell, const int istep) istep, this->exx_nao, this->exx_info_, - *this->inp_); + *this->inp_, + load_exx_flag); } // for each ionic step, the overlap must be rebuilt diff --git a/source/source_esolver/test/esolver_dp_test.cpp b/source/source_esolver/test/esolver_dp_test.cpp index c4daaef1190..a60b40a0783 100644 --- a/source/source_esolver/test/esolver_dp_test.cpp +++ b/source/source_esolver/test/esolver_dp_test.cpp @@ -2,11 +2,8 @@ #include "gtest/gtest.h" #include "for_test.h" -#define private public -#define protected public #include "../esolver_dp.h" #include "source_io/module_parameter/input_parameter.h" -#undef private /************************************************ * unit tests of class ESolver_DP ***********************************************/ @@ -64,23 +61,23 @@ class ESolverDPTest : public ::testing::Test TEST_F(ESolverDPTest, InitCase1) { // Check the initialized variables - EXPECT_DOUBLE_EQ(esolver->dp_potential, 0.0); + EXPECT_DOUBLE_EQ(esolver->get_dp_potential(), 0.0); for (int i = 0; i < 3; ++i) { for (int j = 0; j < 3; ++j) { - EXPECT_DOUBLE_EQ(esolver->dp_virial(i, j), 0.0); + EXPECT_DOUBLE_EQ(esolver->get_dp_virial()(i, j), 0.0); } } for (int i = 0; i < ucell.nat; ++i) { for (int j = 0; j < 3; ++j) { - EXPECT_DOUBLE_EQ(esolver->dp_force(i, j), 0.0); + EXPECT_DOUBLE_EQ(esolver->get_dp_force()(i, j), 0.0); } } - EXPECT_EQ(esolver->atype[0], 0); - EXPECT_EQ(esolver->atype[1], 0); + EXPECT_EQ(esolver->get_atype()[0], 0); + EXPECT_EQ(esolver->get_atype()[1], 0); } // Test the Run() funciton WARNING_QUIT @@ -100,7 +97,7 @@ TEST_F(ESolverDPTest, RunWarningQuit) TEST_F(ESolverDPTest, CalEnergy) { double etot = 0.0; - esolver->dp_potential = 9.8; + esolver->get_dp_potential() = 9.8; etot = esolver->cal_energy(); // Check the results @@ -115,7 +112,7 @@ TEST_F(ESolverDPTest, CalForce) { for (int j = 0; j < 3; ++j) { - esolver->dp_force(i, j) = 3.0 * i + j; + esolver->get_dp_force()(i, j) = 3.0 * i + j; } } @@ -139,7 +136,7 @@ TEST_F(ESolverDPTest, CalStress) { for (int j = 0; j < 3; ++j) { - esolver->dp_virial(i, j) = 3.0 * i + j; + esolver->get_dp_virial()(i, j) = 3.0 * i + j; } } @@ -158,7 +155,7 @@ TEST_F(ESolverDPTest, CalStress) // Test the postprocess() funciton TEST_F(ESolverDPTest, Postprocess) { - esolver->dp_potential = 9.8; + esolver->get_dp_potential() = 9.8; // Check the results GlobalV::ofs_running.open("log"); diff --git a/source/source_estate/CMakeLists.txt b/source/source_estate/CMakeLists.txt index 152c7166b94..b5dd498251c 100644 --- a/source/source_estate/CMakeLists.txt +++ b/source/source_estate/CMakeLists.txt @@ -19,6 +19,7 @@ list(APPEND objects module_pot/pot_local.cpp module_pot/potential_new.cpp module_pot/potential_types.cpp + module_pot/pot_surchem.cpp module_pot/pot_ml_exx.cpp module_pot/pot_ml_exx_label.cpp module_pot/pot_sep.cpp @@ -27,26 +28,31 @@ list(APPEND objects module_pot/td_field.cpp module_pot/td_field_profiles.cpp module_pot/td_field_manager.cpp - module_charge/chgmixing.cpp - module_charge/gint_prec_ctrl.cpp + module_charge/chg_routine.cpp module_charge/charge.cpp - module_charge/charge_init.cpp - module_charge/charge_mpi.cpp - module_charge/charge_extra.cpp - module_charge/charge_mixing.cpp - module_charge/charge_mixing_dmr.cpp - module_charge/mix_resid.cpp - module_charge/mix_precond.cpp - module_charge/charge_mixing_rho.cpp - module_charge/charge_mixing_uspp.cpp - module_charge/symm_rho.cpp - module_charge/symm_rhog.cpp - rhog_io.cpp + module_charge/chg_atomic.cpp + module_charge/chg_atomic_inner.cpp + module_charge/chg_init.cpp + module_charge/chg_tools.cpp + module_charge/chg_parallel.cpp + module_charge/chg_extra.cpp + module_charge/chg_mix.cpp + module_charge/chg_mix_rho.cpp + module_charge/chg_dmr.cpp + module_charge/chg_drho.cpp + module_charge/chg_drho_inner.cpp + module_charge/chg_precond.cpp + module_charge/chg_tau.cpp + module_charge/chg_symm.cpp + module_charge/chg_uspp.cpp + module_charge/chg_symm_detail.cpp + module_charge/chg_rhog_io.cpp write_elecstat_pot.cpp write_init.cpp fp_energy.cpp occupy.cpp occ_matrix.cpp + occ_comput.cpp occ_mixer.cpp param_update.cpp setup_estate_pw.cpp @@ -56,11 +62,19 @@ list(APPEND objects if(ENABLE_LCAO) list(APPEND objects elecstate_lcao.cpp - module_dm/init_dm.cpp + module_dm/dm_routine.cpp module_dm/density_matrix.cpp - module_dm/density_matrix_io.cpp - module_dm/cal_dm_psi.cpp - module_dm/cal_edm_tddft.cpp + module_dm/dmr_gamma.cpp + module_dm/dm_tools.cpp + module_dm/dmr_k.cpp + module_dm/dmr_td.cpp + module_dm/dmr_full.cpp + module_dm/dmr_init.cpp + module_dm/dm_setter.cpp + module_dm/dm_getter.cpp + module_dm/dm_from_psi.cpp + module_dm/edm_tddft.cpp + module_dm/edm_tddft_lapack.cpp ) endif() @@ -77,9 +91,15 @@ endif() if(BUILD_TESTING) if(ENABLE_MPI) add_subdirectory(test) + add_subdirectory(module_charge/unittests) endif() endif() if(ENABLE_LCAO) add_subdirectory(module_dm) + if(BUILD_TESTING) + if(ENABLE_MPI) + add_subdirectory(module_dm/unittests) + endif() + endif() endif() diff --git a/source/source_estate/cal_dm.h b/source/source_estate/cal_dm.h index aede5980e0a..ec62f3aea51 100644 --- a/source/source_estate/cal_dm.h +++ b/source/source_estate/cal_dm.h @@ -1,7 +1,7 @@ #ifndef CAL_DM_H #define CAL_DM_H -#include "math_tools.h" +#include "source_estate/module_dm/dm_from_psi.h" #include "source_base/timer.h" #include "source_base/matrix.h" #include "source_base/complexmatrix.h" @@ -10,59 +10,19 @@ namespace elecstate { // for gamma_only(double case) and multi-k(complex case) +// Thin adapter over module_dm::dmk_from_psi for callers that own the DM blocks as +// ModuleBase::matrix / ModuleBase::ComplexMatrix storage (e.g. DeePKS bandgap terms). inline void cal_dm(const Parallel_Orbitals* ParaV, const ModuleBase::matrix& wg, const psi::Psi& wfc, std::vector& dm) { ModuleBase::TITLE("elecstate", "cal_dm"); ModuleBase::timer::start("elecstate","cal_dm"); - //dm.resize(wfc.get_nk(), ParaV->ncol, ParaV->nrow); - const int nbands_local = wfc.get_nbands(); - const int nbasis_local = wfc.get_nbasis(); - - // dm = wfc.T * wg * wfc.conj() - // dm[is](iw1,iw2) = \sum_{ib} wfc[is](ib,iw1).T * wg(is,ib) * wfc[is](ib,iw2).conj() for (int ik = 0; ik < wfc.get_nk(); ++ik) { - wfc.fix_k(ik); - //dm.fix_k(ik); dm[ik].create(ParaV->ncol, ParaV->nrow); - // wg_wfc(ib,iw) = wg[ib] * wfc(ib,iw); - psi::Psi wg_wfc(1, - wfc.get_nbands(), - wfc.get_nbasis(), - wfc.get_nbasis(), - true); - wg_wfc.set_all_psi(wfc.get_pointer(), wg_wfc.size()); - - int ib_global = 0; - for (int ib_local = 0; ib_local < nbands_local; ++ib_local) - { - while (ib_local != ParaV->global2local_col(ib_global)) - { - ++ib_global; - if (ib_global >= wg.nc) - { - break; - ModuleBase::WARNING_QUIT("ElecStateLCAO::cal_dm", "please check global2local_col!"); - } - } - if (ib_global >= wg.nc) { continue; -} - const double wg_local = wg(ik, ib_global); - double* wg_wfc_pointer = &(wg_wfc(0, ib_local, 0)); - BlasConnector::scal(nbasis_local, wg_local, wg_wfc_pointer, 1); - } - - // C++: dm(iw1,iw2) = wfc(ib,iw1).T * wg_wfc(ib,iw2) -#ifdef __MPI - psiMulPsiMpi(wg_wfc, wfc, dm[ik], ParaV->desc_wfc, ParaV->desc); -#else - psiMulPsi(wg_wfc, wfc, dm[ik]); -#endif + module_dm::dmk_from_psi(ParaV, wg, ik, wfc, dm[ik].c); } ModuleBase::timer::end("elecstate","cal_dm"); - - return; } inline void cal_dm(const Parallel_Orbitals* ParaV, const ModuleBase::matrix& wg, const psi::Psi>& wfc, std::vector& dm) @@ -70,58 +30,13 @@ inline void cal_dm(const Parallel_Orbitals* ParaV, const ModuleBase::matrix& wg, ModuleBase::TITLE("elecstate", "cal_dm"); ModuleBase::timer::start("elecstate","cal_dm"); - //dm.resize(wfc.get_nk(), ParaV->ncol, ParaV->nrow); - const int nbands_local = wfc.get_nbands(); - const int nbasis_local = wfc.get_nbasis(); - - // dm = wfc.T * wg * wfc.conj() - // dm[is](iw1,iw2) = \sum_{ib} wfc[is](ib,iw1).T * wg(is,ib) * wfc[is](ib,iw2).conj() for (int ik = 0; ik < wfc.get_nk(); ++ik) { - wfc.fix_k(ik); - //dm.fix_k(ik); dm[ik].create(ParaV->ncol, ParaV->nrow); - // wg_wfc(ib,iw) = wg[ib] * wfc(ib,iw); - psi::Psi> wg_wfc(1, wfc.get_nbands(), wfc.get_nbasis(), wfc.get_nbasis(), true); - const std::complex* pwfc = wfc.get_pointer(); - std::complex* pwg_wfc = wg_wfc.get_pointer(); -#ifdef _OPENMP -#pragma omp parallel for schedule(static, 1024) -#endif - for(int i = 0;iglobal2local_col(ib_global)) - { - ++ib_global; - if (ib_global >= wg.nc) - { - break; - ModuleBase::WARNING_QUIT("ElecStateLCAO::cal_dm", "please check global2local_col!"); - } - } - if (ib_global >= wg.nc) { continue; -} - const double wg_local = wg(ik, ib_global); - std::complex* wg_wfc_pointer = &(wg_wfc(0, ib_local, 0)); - BlasConnector::scal(nbasis_local, wg_local, wg_wfc_pointer, 1); - } - - // C++: dm(iw1,iw2) = wfc(ib,iw1).T * wg_wfc(ib,iw2) -#ifdef __MPI - psiMulPsiMpi(wg_wfc, wfc, dm[ik], ParaV->desc_wfc, ParaV->desc); -#else - psiMulPsi(wg_wfc, wfc, dm[ik]); -#endif + module_dm::dmk_from_psi(ParaV, wg, ik, wfc, dm[ik].c); } ModuleBase::timer::end("elecstate","cal_dm"); - return; } }//namespace elecstate diff --git a/source/source_estate/elecstate.h b/source/source_estate/elecstate.h index 5e8b5a9c9cd..68d637a4d4f 100644 --- a/source/source_estate/elecstate.h +++ b/source/source_estate/elecstate.h @@ -8,6 +8,11 @@ #include "source_psi/psi.h" #include "module_pot/potential_new.h" +namespace ModulePW +{ +class PW_Basis_Big; +} + namespace elecstate { @@ -145,7 +150,17 @@ class ElecState ModuleBase::matrix vnew; bool vnew_exist = false; void cal_converged(); - void cal_energies(const int type); + /// @brief assemble the total energy from its contributions + /// @param type 1 means Harris-Foulkes functional, 2 means Kohn-Sham + /// @param imp_sol whether the implicit solvation correction contributes + /// @param sc_mag_switch whether the spin-constraint energy contributes + /// @param dft_plus_u 0 disables the DFT+U energy term + /// @param assume_isolated "makov-payne" adds the isolated-system correction + void cal_energies(const int type, + const bool imp_sol, + const bool sc_mag_switch, + const int dft_plus_u, + const std::string& assume_isolated); void set_exx(const double& Eexx, const bool cal_exx, const double hybrid_alpha); void set_exx(const std::complex& Eexx, const bool cal_exx, const double hybrid_alpha); diff --git a/source/source_estate/elecstate_energy.cpp b/source/source_estate/elecstate_energy.cpp index cefdcd28bdb..b417b910091 100644 --- a/source/source_estate/elecstate_energy.cpp +++ b/source/source_estate/elecstate_energy.cpp @@ -166,7 +166,7 @@ double ElecState::cal_delta_eband(const UnitCell& ucell) const } } - if (PARAM.inp.nspin == 2) + if (this->charge->nspin == 2) { v_eff = this->pot->get_eff_v(1); for (int ir = 0; ir < this->charge->rhopw->nrxx; ir++) @@ -186,7 +186,7 @@ double ElecState::cal_delta_eband(const UnitCell& ucell) const } } } - else if (PARAM.inp.nspin == 4) + else if (this->charge->nspin == 4) { for (int is = 1; is < 4; is++) { @@ -245,7 +245,7 @@ double ElecState::cal_delta_escf() const } } - if (PARAM.inp.nspin == 2) + if (this->charge->nspin == 2) { v_eff = this->pot->get_eff_v(1); if (XC_Functional::get_ked_flag()) @@ -261,7 +261,7 @@ double ElecState::cal_delta_escf() const } } } - if (PARAM.inp.nspin == 4) + if (this->charge->nspin == 4) { for (int is = 1; is < 4; is++) { @@ -279,6 +279,11 @@ double ElecState::cal_delta_escf() const assert(this->charge->rhopw->nxyz > 0); + // BUG(investigate): rhopw->omega is stale in variable-cell calculations + // (NPT): pw_rho/pw_rhod are not rebuilt on cell change, so this uses the + // initial cell volume. Should use ucell.omega instead. The descf term is + // typically small, so the impact may be minor, but it should be verified + // for NPT calculations. descf *= this->charge->rhopw->omega / this->charge->rhopw->nxyz; // mohan move the code here, 2025-11-28 @@ -310,7 +315,11 @@ void ElecState::cal_converged() * @param type: 1 means Harris-Foulkes functinoal; * @param type: 2 means Kohn-Sham functional; */ -void ElecState::cal_energies(const int type) +void ElecState::cal_energies(const int type, + const bool imp_sol, + const bool sc_mag_switch, + const int dft_plus_u, + const std::string& assume_isolated) { //! Hartree energy this->f_en.hartree_energy = get_hartree_energy(); @@ -322,27 +331,27 @@ void ElecState::cal_energies(const int type) this->f_en.gatefield = get_etot_gatefield(); //! energy from implicit solvation model - if (PARAM.inp.imp_sol) + if (imp_sol) { this->f_en.esol_el = get_solvent_model_Ael(); this->f_en.esol_cav = get_solvent_model_Acav(); } //! spin constrained energy - if (PARAM.inp.sc_mag_switch) + if (sc_mag_switch) { this->f_en.escon = get_spin_constrain_energy(); } // energy from DFT+U - if (PARAM.inp.dft_plus_u) + if (dft_plus_u) { this->f_en.edftu = get_dftu_energy(); } this->f_en.e_local_pp = get_local_pp_energy(); - if (PARAM.inp.assume_isolated == "makov-payne") + if (assume_isolated == "makov-payne") { const UnitCell* ucell = this->pot->get_ucell(); if (ucell == nullptr || this->charge == nullptr || this->charge->rhopw == nullptr) @@ -353,8 +362,8 @@ void ElecState::cal_energies(const int type) std::vector v_elecstat; const double* v_elecstat_ptr = nullptr; { - ModuleBase::matrix vh(PARAM.inp.nspin, this->charge->rhopw->nrxx); - vh = elecstate::H_Hartree_pw::v_hartree(*ucell, this->charge->rhopw, PARAM.inp.nspin, this->charge->rho); + ModuleBase::matrix vh(this->charge->nspin, this->charge->rhopw->nrxx); + vh = elecstate::H_Hartree_pw::v_hartree(*ucell, this->charge->rhopw, this->charge->nspin, this->charge->rho); v_elecstat.assign(this->charge->rhopw->nrxx, 0.0); const double* v_fixed = this->pot->get_fixed_v(); for (int ir = 0; ir < this->charge->rhopw->nrxx; ++ir) diff --git a/source/source_estate/elecstate_lcao.cpp b/source/source_estate/elecstate_lcao.cpp index 3a826161387..65b2e4c10dc 100644 --- a/source/source_estate/elecstate_lcao.cpp +++ b/source/source_estate/elecstate_lcao.cpp @@ -1,7 +1,7 @@ #include "source_estate/elecstate_lcao.h" #include "source_estate/cal_dm.h" #include "source_base/timer.h" -#include "source_estate/module_dm/cal_dm_psi.h" +#include "source_estate/module_dm/dm_from_psi.h" #include "source_hamilt/module_xc/xc_functional.h" #include "source_lcao/module_deltaspin/spin_constrain.h" #include "source_io/module_parameter/parameter.h" @@ -32,8 +32,9 @@ double ElecStateLCAO>::get_spin_constrain_energy() template <> void ElecStateLCAO::dm2rho(std::vector pexsi_DM, - std::vector pexsi_EDM, - DensityMatrix* dm) + std::vector edm_pexsi, + module_dm::DensityMatrix* dm, + const double omega) { ModuleBase::timer::start("ElecStateLCAO", "dm2rho"); @@ -44,14 +45,14 @@ void ElecStateLCAO::dm2rho(std::vector pexsi_DM, } #ifdef __PEXSI - dm->pexsi_EDM = pexsi_EDM; + dm->edm_pexsi = edm_pexsi; #endif for (int is = 0; is < nspin; is++) { - dm->set_DMK_pointer(is, pexsi_DM[is]); + dm->set_dmk_ptr(is, pexsi_DM[is]); } - dm->cal_DMR(); + dm->cal_dmr(-1); for (int is = 0; is < PARAM.inp.nspin; is++) { @@ -60,17 +61,17 @@ void ElecStateLCAO::dm2rho(std::vector pexsi_DM, } ModuleBase::GlobalFunc::NOTE("Calculate the charge on real space grid!"); - ModuleGint::cal_gint_rho(dm->get_DMR_vector(), PARAM.inp.nspin, this->charge->rho); + ModuleGint::cal_gint_rho(dm->get_dmr_vec(), PARAM.inp.nspin, this->charge->rho); if (XC_Functional::get_ked_flag()) { for (int is = 0; is < PARAM.inp.nspin; is++) { ModuleBase::GlobalFunc::ZEROS(this->charge->kin_r[0], this->charge->nrxx); } - ModuleGint::cal_gint_tau(dm->get_DMR_vector(), PARAM.inp.nspin, this->charge->kin_r); + ModuleGint::cal_gint_tau(dm->get_dmr_vec(), PARAM.inp.nspin, this->charge->kin_r); } - this->charge->renormalize_rho(); + this->charge->renormalize_rho(PARAM.inp.nelec, omega); ModuleBase::timer::end("ElecStateLCAO", "dm2rho"); return; @@ -78,8 +79,9 @@ void ElecStateLCAO::dm2rho(std::vector pexsi_DM, template <> void ElecStateLCAO>::dm2rho(std::vector*> pexsi_DM, - std::vector*> pexsi_EDM, - DensityMatrix, double>* dm) + std::vector*> edm_pexsi, + module_dm::DensityMatrix, double>* dm, + const double omega) { ModuleBase::WARNING_QUIT("ElecStateLCAO", "pexsi is not completed for multi-k case"); } @@ -89,9 +91,10 @@ template void ElecStateLCAO::dmToRho(std::vector*>& dmr, int nspin, Charge* chr, + const double omega, bool skip_charge) { - LCAO_domain::dm2rho(dmr, nspin, chr, skip_charge); + LCAO_domain::dm2rho(dmr, nspin, chr, PARAM.inp.nelec, omega, skip_charge); } template class ElecStateLCAO; // Gamma_only case diff --git a/source/source_estate/elecstate_lcao.h b/source/source_estate/elecstate_lcao.h index fa28c5bb804..03982e036d7 100644 --- a/source/source_estate/elecstate_lcao.h +++ b/source/source_estate/elecstate_lcao.h @@ -36,12 +36,13 @@ class ElecStateLCAO : public ElecState /** * @brief calculate electronic charge density from pointers of density matrix calculated by pexsi * @param pexsi_DM: pointers of density matrix (DMK) calculated by pexsi - * @param pexsi_EDM: pointers of energy-weighed density matrix (EDMK) calculated by pexsi, needed by MD, will be - * stored in DensityMatrix::pexsi_EDM + * @param edm_pexsi: pointers of energy-weighed density matrix (edmk) calculated by pexsi, needed by MD, will be + * stored in DensityMatrix::edm_pexsi */ void dm2rho(std::vector pexsi_DM, - std::vector pexsi_EDM, - DensityMatrix* dm); + std::vector edm_pexsi, + module_dm::DensityMatrix* dm, + const double omega); /** * @brief calculate electronic charge density from the density matrix (DMR) @@ -50,10 +51,14 @@ class ElecStateLCAO : public ElecState * charge-density calculation through the ElecState interface, mirroring the * plane-wave path (ElecStatePW::psiToRho) and the pexsi branch above. This * keeps the source_lcao dependency out of source_hsolver. + * + * @param omega current unit-cell volume (ucell.omega). Must not be + * rhopw->omega, which is stale in variable-cell calculations. */ void dmToRho(std::vector*>& dmr, int nspin, Charge* chr, + const double omega, bool skip_charge = false); }; diff --git a/source/source_estate/elecstate_print.cpp b/source/source_estate/elecstate_print.cpp index 7a298530cf2..05566d819ab 100644 --- a/source/source_estate/elecstate_print.cpp +++ b/source/source_estate/elecstate_print.cpp @@ -175,6 +175,8 @@ void print_etot(const Magnetism& magnet, const double& scf_thr, const double& scf_thr_kin, const double& duration, + const Input_para& inp, + const bool two_fermi, const double& pw_diag_thr, const double& avg_iter, const bool print, @@ -190,7 +192,7 @@ void print_etot(const Magnetism& magnet, GlobalV::ofs_running << " Electron density deviation " << scf_thr << std::endl; - if (PARAM.inp.basis_type == "pw") + if (inp.basis_type == "pw") { ModuleBase::GlobalFunc::OUT(GlobalV::ofs_running, "Diago Threshold", pw_diag_thr); } @@ -199,7 +201,7 @@ void print_etot(const Magnetism& magnet, std::vector energies_Ry; std::vector energies_eV; - if ((iter % PARAM.inp.out_freq_elec == 0) || converged || iter == PARAM.inp.scf_nmax) + if ((iter % inp.out_freq_elec == 0) || converged || iter == inp.scf_nmax) { int n_order = std::max(0, Occupy::gaussian_type); @@ -248,7 +250,7 @@ void print_etot(const Magnetism& magnet, energies_Ry.push_back(elec.f_en.e_local_pp); //! vdw energy - std::string vdw_method = PARAM.inp.vdw_method; + std::string vdw_method = inp.vdw_method; if (vdw_method == "d2") // Peize Lin add 2014-04, update 2021-03-09 { titles.push_back("E_vdwD2"); @@ -266,7 +268,7 @@ void print_etot(const Magnetism& magnet, } // mohan add 20251108 - if (PARAM.inp.dft_plus_u) + if (inp.dft_plus_u) { titles.push_back("E_plusU"); energies_Ry.push_back(elec.f_en.edftu); @@ -277,7 +279,7 @@ void print_etot(const Magnetism& magnet, energies_Ry.push_back(elec.f_en.exx); //! solvation energy - if (PARAM.inp.imp_sol) + if (inp.imp_sol) { titles.push_back("E_sol_el"); energies_Ry.push_back(elec.f_en.esol_el); @@ -286,14 +288,14 @@ void print_etot(const Magnetism& magnet, } //! electric field energy - if (PARAM.inp.efield_flag) + if (inp.efield_flag) { titles.push_back("E_efield"); energies_Ry.push_back(elecstate::Efield::etotefield); } //! gate energy - if (PARAM.inp.gate_flag) + if (inp.gate_flag) { titles.push_back("E_gatefield"); energies_Ry.push_back(elecstate::Gatefield::etotgatefield); @@ -301,12 +303,12 @@ void print_etot(const Magnetism& magnet, //! deepks energy #ifdef __MLALGO - if (PARAM.inp.deepks_scf) + if (inp.deepks_scf) { titles.push_back("E_DeePKS"); energies_Ry.push_back(elec.f_en.edeepks_delta); } - if (PARAM.inp.ml_exx) + if (inp.ml_exx) { titles.push_back("E_ML-EXX"); energies_Ry.push_back(elec.f_en.ml_exx); @@ -322,7 +324,7 @@ void print_etot(const Magnetism& magnet, } // print out the Fermi energy if needed - if (PARAM.globalv.two_fermi) + if (two_fermi) { titles.push_back("E_Fermi_up"); energies_Ry.push_back(elec.eferm.ef_up); @@ -336,7 +338,7 @@ void print_etot(const Magnetism& magnet, } // print out the band gap if needed - if (!PARAM.globalv.two_fermi) + if (!two_fermi) { titles.push_back("E_gap(k)"); // gap of given k-points energies_Ry.push_back(elec.bandgap); @@ -362,10 +364,10 @@ void print_etot(const Magnetism& magnet, GlobalV::ofs_running << table.str() << std::endl; - if (PARAM.inp.out_level == "ie" || PARAM.inp.out_level == "m") + if (inp.out_level == "ie" || inp.out_level == "m") { std::vector mag; - switch (PARAM.inp.nspin) + switch (inp.nspin) { case 2: mag = {magnet.tot_mag, magnet.abs_mag}; @@ -384,7 +386,7 @@ void print_etot(const Magnetism& magnet, } // Pure SDFT (nbands=0) uses Chebyshev trace (CT) since no H diagonalization is performed. // Mixed SDFT (nbands>0) still diagonalizes KS orbitals, so use the actual ks_solver label. - const std::string iter_label = (PARAM.inp.esolver_type == "sdft" && PARAM.inp.nbands == 0) ? "sdft" : PARAM.inp.ks_solver; + const std::string iter_label = (inp.esolver_type == "sdft" && inp.nbands == 0) ? "sdft" : inp.ks_solver; elecstate::print_scf_iterinfo(iter_label, iter, 4, diff --git a/source/source_estate/elecstate_print.h b/source/source_estate/elecstate_print.h index 65383e21447..e6f89f5b969 100644 --- a/source/source_estate/elecstate_print.h +++ b/source/source_estate/elecstate_print.h @@ -8,6 +8,10 @@ namespace elecstate void print_format(const std::string& name, const double& value); + /// @param inp the INPUT parameters whose flags decide which energy terms and + /// headers are printed + /// @param two_fermi whether the run keeps two Fermi levels; derived, so it + /// does not live in Input_para void print_etot(const Magnetism& magnet, const ElecState& elec, const bool converged, @@ -15,6 +19,8 @@ namespace elecstate const double& scf_thr, const double& scf_thr_kin, const double& duration, + const Input_para& inp, + const bool two_fermi, const double& pw_diag_thr = 0, const double& avg_iter = 0, bool print = true, diff --git a/source/source_estate/elecstate_pw.cpp b/source/source_estate/elecstate_pw.cpp index 05c21430e07..3037b88b286 100644 --- a/source/source_estate/elecstate_pw.cpp +++ b/source/source_estate/elecstate_pw.cpp @@ -5,6 +5,7 @@ #include "source_base/module_device/device.h" #include "source_base/parallel_reduce.h" #include "source_base/timer.h" +#include "source_estate/module_charge/chg_parallel.h" #include "source_estate/uspp_density.h" #include "source_hamilt/module_xc/xc_functional.h" #include "source_io/module_parameter/parameter.h" @@ -171,7 +172,11 @@ template void ElecStatePW::parallelK() { #ifdef __MPI - this->charge->rho_mpi(); + module_charge::rho_mpi(*this->charge, + GlobalV::KPAR, + PARAM.globalv.all_ks_run, + PARAM.inp.bndpar, + PARAM.inp.nspin); #endif } diff --git a/source/source_estate/elecstate_pw_cal_tau.cpp b/source/source_estate/elecstate_pw_cal_tau.cpp index 5c0b6ae9f5c..af0176c519e 100644 --- a/source/source_estate/elecstate_pw_cal_tau.cpp +++ b/source/source_estate/elecstate_pw_cal_tau.cpp @@ -1,5 +1,7 @@ #include "elecstate_pw.h" +#include "source_estate/module_charge/chg_parallel.h" + namespace elecstate { template @@ -7,9 +9,9 @@ void ElecStatePW::cal_tau(const psi::Psi& psi) { ModuleBase::TITLE("ElecStatePW", "cal_tau"); for(int is=0; iskin_r[is], 0, this->charge->nrxx); - } + } for (int ik = 0; ik < psi.get_nk(); ++ik) { @@ -55,7 +57,11 @@ void ElecStatePW::cal_tau(const psi::Psi& psi) } } #ifdef __MPI - this->charge->kin_r_mpi(); + module_charge::kin_r_mpi(*this->charge, + GlobalV::KPAR, + PARAM.globalv.all_ks_run, + PARAM.inp.bndpar, + PARAM.inp.nspin); #endif ModuleBase::TITLE("ElecStatePW", "cal_tau"); } diff --git a/source/source_estate/elecstate_tools.cpp b/source/source_estate/elecstate_tools.cpp index b8160b76ef0..3b64aca047f 100644 --- a/source/source_estate/elecstate_tools.cpp +++ b/source/source_estate/elecstate_tools.cpp @@ -59,15 +59,18 @@ void calculate_weights(const ModuleBase::matrix& ekb, const int band_offset = band_output.local_offset(); if (PARAM.globalv.two_fermi) { - Occupy::iweights(nks, klist->wk, nbands, band_offset, nelec_spin[0], ekb, eferm.ef_up, wg, 0, klist->isk); - Occupy::iweights(nks, klist->wk, nbands, band_offset, nelec_spin[1], ekb, eferm.ef_dw, wg, 1, klist->isk); + Occupy::iweights(nks, klist->wk, nbands, band_offset, nelec_spin[0], ekb, eferm.ef_up, wg, 0, klist->isk, + PARAM.inp.nspin); + Occupy::iweights(nks, klist->wk, nbands, band_offset, nelec_spin[1], ekb, eferm.ef_dw, wg, 1, klist->isk, + PARAM.inp.nspin); // ef = ( ef_up + ef_dw ) / 2.0_dp need??? mohan add 2012-04-16 // Keep independent Fermi levels for the two spin channels. } else { // A spin selector of -1 requests the combined-spin occupation path. - Occupy::iweights(nks, klist->wk, nbands, band_offset, PARAM.inp.nelec, ekb, eferm.ef, wg, -1, klist->isk); + Occupy::iweights(nks, klist->wk, nbands, band_offset, PARAM.inp.nelec, ekb, eferm.ef, wg, -1, klist->isk, + PARAM.inp.nspin); } } else if (Occupy::use_gaussian_broadening) diff --git a/source/source_estate/estate_e_terms.cpp b/source/source_estate/estate_e_terms.cpp index 997c8fa2c98..eec291ef36e 100644 --- a/source/source_estate/estate_e_terms.cpp +++ b/source/source_estate/estate_e_terms.cpp @@ -47,6 +47,12 @@ double ElecState::get_local_pp_energy() this->pot->get_fixed_v(), 1, this->charge->rho[is], 1) + // BUG(investigate): rhopw->omega is stale in variable-cell + // calculations (NPT): pw_rho/pw_rhod are not rebuilt on cell + // change, so this uses the initial cell volume. Should use + // ucell.omega instead. Need to check whether the resulting + // energy error is absorbed elsewhere (e.g. by the + // renormalization of rho), or if it biases the stress. * this->charge->rhopw->omega / this->charge->rhopw->nxyz; } Parallel_Reduce::reduce_pool(local_pseudopot_energy); diff --git a/source/source_estate/init_scf.cpp b/source/source_estate/init_scf.cpp index 4e5699e8074..eb9cf262262 100644 --- a/source/source_estate/init_scf.cpp +++ b/source/source_estate/init_scf.cpp @@ -1,4 +1,5 @@ #include "elecstate.h" +#include "source_estate/module_charge/chg_tools.h" #include "source_estate/write_init.h" namespace elecstate @@ -14,12 +15,17 @@ void init_scf(const UnitCell& ucell, ElecState* pelec) { //! core correction potential. - pelec->charge->set_rho_core(ucell, strucfac, numeric); + module_charge::set_rho_core(ucell, + strucfac, + numeric, + pelec->charge->rho_core, + pelec->charge->rhog_core, + *pelec->charge->rhopw); //! renormalize the charge density if(PARAM.inp.init_chg != "dm_no_renormalize") { - pelec->charge->renormalize_rho(); + pelec->charge->renormalize_rho(inp.nelec, ucell.omega); } //! initialize the potential diff --git a/source/source_estate/makov_payne.cpp b/source/source_estate/makov_payne.cpp index 5796a954cc1..72a8f31db30 100644 --- a/source/source_estate/makov_payne.cpp +++ b/source/source_estate/makov_payne.cpp @@ -261,6 +261,10 @@ MakovPayneResult makov_payne_correction(const UnitCell& ucell, } const ModulePW::PW_Basis* rhopw = charge.rhopw; + // BUG(investigate): rhopw->omega is stale in variable-cell calculations + // (NPT): pw_rho/pw_rhod are not rebuilt on cell change, so this uses the + // initial cell volume. Makov-Payne correction only applies to low-dimensional + // systems; verify whether NPT is supported for those cases before fixing. const double dv = rhopw->omega / static_cast(rhopw->nxyz); double electron_number = 0.0; ModuleBase::Vector3 dipole_el(0.0, 0.0, 0.0); diff --git a/source/source_estate/math_tools.h b/source/source_estate/math_tools.h deleted file mode 100644 index 049f9a4e8e9..00000000000 --- a/source/source_estate/math_tools.h +++ /dev/null @@ -1,124 +0,0 @@ -#include "source_base/module_external/blas_connector.h" -#include "source_base/complexmatrix.h" -#include "source_base/matrix.h" -#include "source_base/module_external/scalapack_connector.h" -#include "source_psi/psi.h" -#include "source_base/timer.h" - -#ifdef __MPI -inline void psiMulPsiMpi(const psi::Psi& psi1, - const psi::Psi& psi2, - ModuleBase::matrix& dm_out, - const int* desc_psi, - const int* desc_dm) -{ - ModuleBase::timer::start("psiMulPsiMpi","pdgemm"); - const double one_float = 1.0, zero_float = 0.0; - const int one_int = 1; - const char N_char = 'N', T_char = 'T'; - const int nlocal = desc_dm[2]; - const int nbands = desc_psi[3]; - pdgemm_(&N_char, - &T_char, - &nlocal, - &nlocal, - &nbands, - &one_float, - psi1.get_pointer(), - &one_int, - &one_int, - desc_psi, - psi2.get_pointer(), - &one_int, - &one_int, - desc_psi, - &zero_float, - dm_out.c, - &one_int, - &one_int, - desc_dm); - ModuleBase::timer::end("psiMulPsiMpi","pdgemm"); -} - -inline void psiMulPsiMpi(const psi::Psi>& psi1, - const psi::Psi>& psi2, - ModuleBase::ComplexMatrix& dm_out, - const int* desc_psi, - const int* desc_dm) -{ - ModuleBase::timer::start("psiMulPsiMpi","pdgemm"); - const std::complex one_complex = {1.0, 0.0}, zero_complex = {0.0, 0.0}; - const int one_int = 1; - const char N_char = 'N', T_char = 'T'; - const int nlocal = desc_dm[2]; - const int nbands = desc_psi[3]; - pzgemm_(&N_char, - &T_char, - &nlocal, - &nlocal, - &nbands, - &one_complex, - psi1.get_pointer(), - &one_int, - &one_int, - desc_psi, - psi2.get_pointer(), - &one_int, - &one_int, - desc_psi, - &zero_complex, - dm_out.c, - &one_int, - &one_int, - desc_dm); - ModuleBase::timer::end("psiMulPsiMpi","pdgemm"); -} - -#else -inline void psiMulPsi(const psi::Psi& psi1, const psi::Psi& psi2, ModuleBase::matrix& dm_out) -{ - const double one_float = 1.0, zero_float = 0.0; - const int one_int = 1; - const char N_char = 'N', T_char = 'T'; - const int nlocal = psi1.get_nbasis(); - const int nbands = psi1.get_nbands(); - dgemm_(&N_char, - &T_char, - &nlocal, - &nlocal, - &nbands, - &one_float, - psi1.get_pointer(), - &nlocal, - psi2.get_pointer(), - &nlocal, - &zero_float, - dm_out.c, - &nlocal); -} - -inline void psiMulPsi(const psi::Psi>& psi1, - const psi::Psi>& psi2, - ModuleBase::ComplexMatrix& dm_out) -{ - const int one_int = 1; - const char N_char = 'N', T_char = 'T'; - const int nlocal = psi1.get_nbasis(); - const int nbands = psi1.get_nbands(); - const std::complex one_complex = {1.0, 0.0}, zero_complex = {0.0, 0.0}; - zgemm_(&N_char, - &T_char, - &nlocal, - &nlocal, - &nbands, - &one_complex, - psi1.get_pointer(), - &nlocal, - psi2.get_pointer(), - &nlocal, - &zero_complex, - dm_out.c, - &nlocal); -} - -#endif \ No newline at end of file diff --git a/source/source_estate/module_charge/charge.cpp b/source/source_estate/module_charge/charge.cpp index 5f25aefe06f..e687f606c1d 100644 --- a/source/source_estate/module_charge/charge.cpp +++ b/source/source_estate/module_charge/charge.cpp @@ -17,6 +17,7 @@ // even in a LSDA calculation. //---------------------------------------------------------- #include "charge.h" +#include "chg_tools.h" #include "source_base/global_function.h" #include "source_base/global_variable.h" @@ -28,15 +29,13 @@ #include "source_base/tool_threading.h" #include "source_cell/unitcell.h" #include "source_cell/magnetism.h" -#include "source_hamilt/module_xc/xc_functional.h" -#include "source_io/module_parameter/parameter.h" +#include #include Charge::Charge() { allocate_rho = false; - allocate_rho_final_scf = false; // LiuXh add 20180619 } Charge::~Charge() @@ -49,51 +48,43 @@ void Charge::set_rhopw(ModulePW::PW_Basis* rhopw_in) this->rhopw = rhopw_in; } -// mohan add 2025-12-02 -bool Charge::kin_density() const -{ - if (XC_Functional::get_ked_flag() || PARAM.inp.out_elf[0] > 0) - { - return true; - } - else - { - return false; - } -} - void Charge::destroy() { - if (allocate_rho || allocate_rho_final_scf) // LiuXh add 20180619 - { - delete[] rho; - delete[] rhog; - delete[] rho_save; - delete[] rhog_save; - delete[] rho_core; - delete[] rhog_core; - delete[] _space_rho; - delete[] _space_rho_save; - delete[] _space_rhog; - delete[] _space_rhog_save; - delete[] _space_kin_r; - delete[] _space_kin_r_save; - if (XC_Functional::get_ked_flag() || PARAM.inp.out_elf[0] > 0) - { - delete[] kin_r; - delete[] kin_r_save; - } + if (allocate_rho) + { + // All storage (rho, rhog, rho_core, etc.) is backed by std::vector + // members that self-manage; just clear the vectors. + _ptrs_rho.clear(); + _ptrs_rhog.clear(); + _ptrs_rho_save.clear(); + _ptrs_rhog_save.clear(); + _ptrs_kin_r.clear(); + _ptrs_kin_r_save.clear(); + _space_rho_core.clear(); + _space_rhog_core.clear(); + rho = nullptr; + rhog = nullptr; + rho_save = nullptr; + rhog_save = nullptr; + rho_core = nullptr; + rhog_core = nullptr; + kin_r = nullptr; + kin_r_save = nullptr; } } -void Charge::allocate(const int& nspin_in, const bool kin_den) +void Charge::allocate(const int& nspin_in, const bool kin_den, const bool meta_gga, + const int test_charge) { ModuleBase::TITLE("Charge", "allocate"); - if (this->rhopw == nullptr) - { - ModuleBase::WARNING_QUIT("Charge::allocate","rhopw is nullptr."); - } + assert(nspin_in > 0); + this->meta_gga = meta_gga; + + if (this->rhopw == nullptr) + { + ModuleBase::WARNING_QUIT("Charge::allocate","rhopw is nullptr."); + } this->nrxx = this->rhopw->nrxx; this->nxyz = this->rhopw->nxyz; @@ -111,46 +102,52 @@ void Charge::allocate(const int& nspin_in, const bool kin_den) // mohan add 2021-02-20 this->nspin = nspin_in; - if (PARAM.inp.test_charge > 1) + if (test_charge > 1) { std::cout << "\n spin_number = " << nspin << " real_point_number = " << nrxx << std::endl; } - // allocate memory - _space_rho = new double[nspin * nrxx]; - _space_rho_save = new double[nspin * nrxx]; - _space_rhog = new std::complex[nspin * ngmc]; - _space_rhog_save = new std::complex[nspin * ngmc]; + // allocate memory (std::vector self-manages the storage) + _space_rho.resize(nspin * nrxx); + _space_rho_save.resize(nspin * nrxx); + _space_rhog.resize(nspin * ngmc); + _space_rhog_save.resize(nspin * ngmc); if(kin_den) { - _space_kin_r = new double[nspin * nrxx]; - _space_kin_r_save = new double[nspin * nrxx]; - } - rho = new double*[nspin]; - rhog = new std::complex*[nspin]; - rho_save = new double*[nspin]; - rhog_save = new std::complex*[nspin]; + _space_kin_r.resize(nspin * nrxx); + _space_kin_r_save.resize(nspin * nrxx); + } + _ptrs_rho.resize(nspin); + _ptrs_rhog.resize(nspin); + _ptrs_rho_save.resize(nspin); + _ptrs_rhog_save.resize(nspin); + rho = _ptrs_rho.data(); + rhog = _ptrs_rhog.data(); + rho_save = _ptrs_rho_save.data(); + rhog_save = _ptrs_rhog_save.data(); if(kin_den) { - kin_r = new double*[nspin]; - kin_r_save = new double*[nspin]; + _ptrs_kin_r.resize(nspin); + _ptrs_kin_r_save.resize(nspin); + kin_r = _ptrs_kin_r.data(); + kin_r_save = _ptrs_kin_r_save.data(); } for (int is = 0; is < nspin; is++) { - rho[is] = _space_rho + is * nrxx; - rhog[is] = _space_rhog + is * ngmc; - rho_save[is] = _space_rho_save + is * nrxx; - rhog_save[is] = _space_rhog_save + is * ngmc; - ModuleBase::GlobalFunc::ZEROS(rho[is], nrxx); - ModuleBase::GlobalFunc::ZEROS(rhog[is], ngmc); - ModuleBase::GlobalFunc::ZEROS(rho_save[is], nrxx); - ModuleBase::GlobalFunc::ZEROS(rhog_save[is], ngmc); - if(kin_den) + rho[is] = _space_rho.data() + is * nrxx; + rhog[is] = _space_rhog.data() + is * ngmc; + rho_save[is] = _space_rho_save.data() + is * nrxx; + rhog_save[is] = _space_rhog_save.data() + is * ngmc; + std::fill(rho[is], rho[is] + nrxx, 0.0); + std::fill(rhog[is], rhog[is] + ngmc, std::complex(0.0, 0.0)); + std::fill(rho_save[is], rho_save[is] + nrxx, 0.0); + std::fill(rhog_save[is], rhog_save[is] + ngmc, std::complex(0.0, 0.0)); + if(kin_den) { - kin_r[is] = _space_kin_r + is * nrxx; - ModuleBase::GlobalFunc::ZEROS(kin_r[is], nrxx); - kin_r_save[is] = _space_kin_r_save + is * nrxx; - ModuleBase::GlobalFunc::ZEROS(kin_r_save[is], nrxx); + kin_r[is] = _space_kin_r.data() + is * nrxx; + std::fill(kin_r[is], kin_r[is] + nrxx, 0.0); + kin_r_save[is] = _space_kin_r_save.data() + is * nrxx; + std::fill(kin_r_save[is], kin_r_save[is] + nrxx, 0.0); } } @@ -164,11 +161,13 @@ void Charge::allocate(const int& nspin_in, const bool kin_den) ModuleBase::Memory::record("Chg::kin_r_save", sizeof(double) * nspin * ngmc); } - this->rho_core = new double[nrxx]; // core charge in real space - ModuleBase::GlobalFunc::ZEROS(rho_core, nrxx); + _space_rho_core.resize(nrxx); + this->rho_core = _space_rho_core.data(); + std::fill(rho_core, rho_core + nrxx, 0.0); - this->rhog_core = new std::complex[ngmc]; // reciprocal core charge - ModuleBase::GlobalFunc::ZEROS(rhog_core, ngmc); + _space_rhog_core.resize(ngmc); + this->rhog_core = _space_rhog_core.data(); + std::fill(rhog_core, rhog_core + ngmc, std::complex(0.0, 0.0)); ModuleBase::Memory::record("Chg::rho_core", sizeof(double) * nrxx); ModuleBase::Memory::record("Chg::rhog_core", sizeof(double) * ngmc); @@ -177,47 +176,27 @@ void Charge::allocate(const int& nspin_in, const bool kin_den) return; } -double Charge::sum_rho() const +double Charge::sum_rho(const double omega) const { - ModuleBase::TITLE("Charge", "sum_rho"); - - double sum_rho = 0.0; - int nspin0 = (nspin == 2) ? 2 : 1; - - for (int is = 0; is < nspin0; is++) - { - for (int ir = 0; ir < nrxx; ir++) - { - sum_rho += this->rho[is][ir]; - } - } - - // multiply the sum of charge density by a factor - sum_rho *= *this->omega_ / static_cast(this->rhopw->nxyz); - -#ifdef __MPI - Parallel_Reduce::reduce_pool(sum_rho); -#endif - - // mohan fixed bug 2010-01-18, - // sum_rho may be smaller than 1, like Na bcc. - if (sum_rho <= 0.1) - { - GlobalV::ofs_warning << " sum_rho=" << sum_rho << std::endl; - ModuleBase::WARNING_QUIT("Charge::renormalize_rho", "Can't find even an electron!"); - } - - return sum_rho; + const int nspin0 = (nspin == 2) ? 2 : 1; + // NOTE: omega must be ucell.omega, NOT rhopw->omega. In variable-cell + // calculations (e.g. NPT) rhopw->omega is stale because pw_rho/pw_rhod + // are not rebuilt on cell change, while ucell.omega is updated every + // MD step. Using the stale volume gives a wrong electron count. + return module_charge::sum_rho(this->rho, nspin0, this->nrxx, omega, this->rhopw->nxyz); } -void Charge::renormalize_rho() +void Charge::renormalize_rho(const double nelec, const double omega) { ModuleBase::TITLE("Charge", "renormalize_rho"); - const double sr = this->sum_rho(); + assert(nelec > 0.0); + assert(omega > 0.0); + + const double sr = this->sum_rho(omega); GlobalV::ofs_warning << std::setprecision(15); ModuleBase::GlobalFunc::OUT(GlobalV::ofs_warning, "charge before normalized", sr); - const double normalize_factor = PARAM.inp.nelec / sr; + const double normalize_factor = nelec / sr; for (int is = 0; is < nspin; is++) { @@ -227,538 +206,21 @@ void Charge::renormalize_rho() } } - ModuleBase::GlobalFunc::OUT(GlobalV::ofs_warning, "charge after normalized", this->sum_rho()); + ModuleBase::GlobalFunc::OUT(GlobalV::ofs_warning, "charge after normalized", this->sum_rho(omega)); GlobalV::ofs_running << std::setprecision(6); return; } -//------------------------------------------------------- -// superposition of atomic charges contained in the array -// rho_at (read from pseudopotential files) -// allocate work space (psic must already be allocated) -//------------------------------------------------------- -void Charge::atomic_rho(const int spin_number_need, - const double& omega, - double** rho_in, - const ModuleBase::ComplexMatrix& strucFac, - const UnitCell& ucell) const // Peize Lin refactor 2021.04.08 -{ - ModuleBase::TITLE("Charge", "atomic_rho"); - ModuleBase::timer::start("Charge", "atomic_rho"); - - { - ModuleBase::ComplexMatrix rho_g3d = [&]() -> ModuleBase::ComplexMatrix - { - // use interpolation to get three dimension charge density. - ModuleBase::ComplexMatrix rho_g3d(spin_number_need, this->rhopw->npw); - - for (int it = 0; it < ucell.ntype; it++) - { - // check the start magnetization - const int startmag_type = [&]() -> int { - if (ucell.magnet.start_mag[it] != 0.0) - { - return 1; - } - return 2; - }(); - ModuleBase::GlobalFunc::OUT(GlobalV::ofs_warning, "startmag_type", startmag_type); - - const Atom* const atom = &ucell.atoms[it]; - - if (!atom->flag_empty_element) // Peize Lin add for bsse 2021.04.07 - { - const std::vector rho_lgl = [&]() -> std::vector { - // one dimension of charge in G space. - std::vector rho_lgl(this->rhopw->ngg, 0); - - // mesh point of this element. - const int mesh = atom->ncpp.msh; - - //---------------------------------------------------------- - // Here we check the electron number - //---------------------------------------------------------- - const std::vector rhoatm = [&]() -> std::vector { - std::vector rhoatm(mesh); - // this is only one part of the charge density for uspp - // liuyu 2023-11-01 - if (atom->ncpp.tvanp) - { - for (int ir = 0; ir < mesh; ++ir) - { - rhoatm[ir] = atom->ncpp.rho_at[ir]; - } - } - else - { - for (int ir = 0; ir < mesh; ++ir) - { - double r2 = atom->ncpp.r[ir] * atom->ncpp.r[ir]; - if (r2!=0) - { - rhoatm[ir] = atom->ncpp.rho_at[ir] / ModuleBase::FOUR_PI / r2; - } - } - rhoatm[0] - = pow((rhoatm[2] / rhoatm[1]), atom->ncpp.r[1] / (atom->ncpp.r[2] - atom->ncpp.r[1])); // zws add, sunliang updated 2024-03-04 - if (rhoatm[0] < 1e-12) - { - rhoatm[0] = rhoatm[1]; - } - else - { - rhoatm[0] = rhoatm[1] / rhoatm[0]; - } - - double charge = 0.0; - ModuleBase::Integral::Simpson_Integral(atom->ncpp.msh, - atom->ncpp.rho_at.data(), - atom->ncpp.rab.data(), - charge); - ModuleBase::GlobalFunc::OUT(GlobalV::ofs_warning, "charge from rho_at", charge); - assert(charge != 0.0 - || charge - == atom->ncpp.zv); // Peize Lin add charge==atom->zv for bsse 2021.04.07 - - double scale = 1.0; - if (charge != atom->ncpp.zv) - { - ModuleBase::GlobalFunc::OUT(GlobalV::ofs_warning, - "charge should be", - atom->ncpp.zv); - scale = atom->ncpp.zv / charge; - } - - for (int ir = 0; ir < mesh; ++ir) - { - rhoatm[ir] *= scale; - rhoatm[ir] *= (ModuleBase::FOUR_PI * atom->ncpp.r[ir] * atom->ncpp.r[ir]); - } - } - return rhoatm; - }(); - - assert(ucell.meshx > 0); - //---------------------------------------------------------- - // Here we compute the G=0 term - //---------------------------------------------------------- - int gstart = 0; - if (this->rhopw->gg_uniq[0] < 1e-8) - { - std::vector rho1d(ucell.meshx); - for (int ir = 0; ir < mesh; ir++) - { - rho1d[ir] = rhoatm[ir]; - } - ModuleBase::Integral::Simpson_Integral(mesh, rho1d.data(), atom->ncpp.rab.data(), rho_lgl[0]); - gstart = 1; - } - if (PARAM.inp.test_charge > 0) - { - std::cout << "\n |G|=0 term done." << std::endl; - } - //---------------------------------------------------------- - // Here we compute the G<>0 term - // But if in parallel case - // G=0 term only belong to 1 cpu. - // Other processors start from '0' - //---------------------------------------------------------- - #ifdef _OPENMP - #pragma omp parallel - { - #endif - const int ngg = this->rhopw->ngg; - const double* gg_uniq = this->rhopw->gg_uniq; - const int meshx = ucell.meshx; - const double tpiba = ucell.tpiba; - std::vector rho1d(meshx); - - #ifdef _OPENMP - #pragma omp for - #endif - for (int igg = gstart; igg < ngg; ++igg) - { - const double gx = sqrt(gg_uniq[igg]) * tpiba; - for (int ir = 0; ir < mesh; ir++) - { - if (atom->ncpp.r[ir] < 1.0e-8) - { - rho1d[ir] = rhoatm[ir]; - } - else - { - const double gxx = gx * atom->ncpp.r[ir]; - rho1d[ir] = rhoatm[ir] * ModuleBase::libm::sin(gxx) / gxx; - } - } - ModuleBase::Integral::Simpson_Integral(mesh, rho1d.data(), atom->ncpp.rab.data(), rho_lgl[igg]); - } - #ifdef _OPENMP - #pragma omp single - #endif - { - if (PARAM.inp.test_charge > 0) - { - std::cout << " |G|>0 term done." << std::endl; - } - } - //---------------------------------------------------------- - // EXPLAIN : Complete the transfer of rho from real space to - // reciprocal space - //---------------------------------------------------------- - #ifdef _OPENMP - #pragma omp for - #endif - for (int igg = 0; igg < ngg; igg++) - { - rho_lgl[igg] /= omega; - } - #ifdef _OPENMP - } - #endif - return rho_lgl; - }(); - //---------------------------------------------------------- - // EXPLAIN : compute the 3D atomic charge in reciprocal space - //---------------------------------------------------------- - if (spin_number_need == 1) - { - const int npw = this->rhopw->npw; - const int* ig2igg = this->rhopw->ig2igg; - #ifdef _OPENMP - #pragma omp parallel for - #endif - for (int ig = 0; ig < npw; ig++) - { - rho_g3d(0, ig) += strucFac(it, ig) * rho_lgl[ig2igg[ig]]; - } - } - // mohan add 2011-06-14, initialize the charge density according to each atom - else if (spin_number_need == 2) - { - if (startmag_type == 1) - { - const int npw = this->rhopw->npw; - const int* ig2igg = this->rhopw->ig2igg; - const double zv = atom->ncpp.zv; - const double start_mag_it = ucell.magnet.start_mag[it]; - #ifdef _OPENMP - #pragma omp parallel for - #endif - for (int ig = 0; ig < npw; ig++) - { - const std::complex swap = strucFac(it, ig) * rho_lgl[ig2igg[ig]]; - const double up = 0.5 * (1 + start_mag_it / zv); - const double dw = 0.5 * (1 - start_mag_it / zv); - rho_g3d(0, ig) += swap * up; - rho_g3d(1, ig) += swap * dw; - } - } - // mohan add 2011-06-14 - else if (startmag_type == 2) - { - std::complex ci_tpi = ModuleBase::NEG_IMAG_UNIT * ModuleBase::TWO_PI; - const int npw = this->rhopw->npw; - const ModuleBase::Vector3* gcar = this->rhopw->gcar; - const int* ig2igg = this->rhopw->ig2igg; - const double zv = atom->ncpp.zv; - for (int ia = 0; ia < atom->na; ia++) - { - const double up = 0.5 * (1 + atom->mag[ia] / atom->ncpp.zv); - const double dw = 0.5 * (1 - atom->mag[ia] / atom->ncpp.zv); - const double tau_x = atom->tau[ia].x; - const double tau_y = atom->tau[ia].y; - const double tau_z = atom->tau[ia].z; - #ifdef _OPENMP - #pragma omp parallel for - #endif - for (int ig = 0; ig < npw; ig++) - { - const double Gtau = gcar[ig][0] * tau_x + gcar[ig][1] * tau_y + gcar[ig][2] * tau_z; - std::complex swap = ModuleBase::libm::exp(ci_tpi * Gtau) * rho_lgl[ig2igg[ig]]; - rho_g3d(0, ig) += swap * up; - rho_g3d(1, ig) += swap * dw; - } - } - } - } - else if (spin_number_need == 4) - { - // noncolinear case - if (startmag_type == 1) - { - double sin_a1, sin_a2, cos_a1, cos_a2; - if (PARAM.globalv.domag) - { - ModuleBase::libm::sincos(atom->angle1[0], &sin_a1, &cos_a1); - ModuleBase::libm::sincos(atom->angle2[0], &sin_a2, &cos_a2); - } - const int npw = this->rhopw->npw; - const int* ig2igg = this->rhopw->ig2igg; - const double zv = atom->ncpp.zv; - const double start_mag_it = ucell.magnet.start_mag[it]; - #ifdef _OPENMP - #pragma omp parallel for - #endif - for (int ig = 0; ig < npw; ig++) - { - const std::complex swap = strucFac(it, ig) * rho_lgl[ig2igg[ig]]; - rho_g3d(0, ig) += swap; - if (PARAM.globalv.domag) - { - rho_g3d(1, ig) += swap * (start_mag_it / zv) * sin_a1 * cos_a2; - rho_g3d(2, ig) += swap * (start_mag_it / zv) * sin_a1 * sin_a2; - rho_g3d(3, ig) += swap * (start_mag_it / zv) * cos_a1; - } - else if (PARAM.globalv.domag_z) - { - rho_g3d(1, ig) = 0.0; - rho_g3d(2, ig) = 0.0; - rho_g3d(3, ig) += swap * (start_mag_it / zv); - } - } - } - else if (startmag_type == 2) - { - std::complex ci_tpi = ModuleBase::NEG_IMAG_UNIT * ModuleBase::TWO_PI; - const int npw = this->rhopw->npw; - const ModuleBase::Vector3* gcar = this->rhopw->gcar; - const int* ig2igg = this->rhopw->ig2igg; - const double zv = atom->ncpp.zv; - for (int ia = 0; ia < atom->na; ia++) - { - double sin_a1, sin_a2, cos_a1, cos_a2; - if (PARAM.globalv.domag || PARAM.globalv.domag_z) - { - ModuleBase::libm::sincos(atom->angle1[ia], &sin_a1, &cos_a1); - } - if (PARAM.globalv.domag) - { - ModuleBase::libm::sincos(atom->angle2[ia], &sin_a2, &cos_a2); - } - const double mag_ia = atom->mag[ia]; - const double tau_x = atom->tau[ia].x; - const double tau_y = atom->tau[ia].y; - const double tau_z = atom->tau[ia].z; - #ifdef _OPENMP - #pragma omp parallel for - #endif - for (int ig = 0; ig < npw; ig++) - { - const double Gtau = gcar[ig][0] * tau_x + gcar[ig][1] * tau_y + gcar[ig][2] * tau_z; - std::complex swap = exp(ci_tpi * Gtau) * rho_lgl[ig2igg[ig]]; - const double mag_factor = mag_ia / zv; - rho_g3d(0, ig) += swap; - if (PARAM.globalv.domag || PARAM.globalv.domag_z) - { - rho_g3d(3, ig) += swap * mag_factor * cos_a1; - } - if (PARAM.globalv.domag) - { - rho_g3d(1, ig) += swap * mag_factor * sin_a1 * cos_a2; - rho_g3d(2, ig) += swap * mag_factor * sin_a1 * sin_a2; - } - else - { - rho_g3d(1, ig) = 0.0; - rho_g3d(2, ig) = 0.0; - } - } - } - } - } - else - { - ModuleBase::WARNING_QUIT("Charge::spin_number_need", " Either 1 or 2 or 4, check SPIN number !"); - } - } - } - return rho_g3d; - }(); - - assert(spin_number_need > 0); - std::vector ne(spin_number_need); - for (int is = 0; is < spin_number_need; is++) - { - this->rhopw->recip2real(&rho_g3d(is, 0), rho_in[is]); - - for (int ir = 0; ir < this->rhopw->nrxx; ++ir) - { - ne[is] += rho_in[is][ir]; - } - - ne[is] *= omega / (double)this->rhopw->nxyz; - #ifdef __MPI - Parallel_Reduce::reduce_pool(ne[is]); - #endif - // we check that everything is correct - double neg = 0.0; - double rea = 0.0; - double ima = 0.0; - double sumrea = 0.0; - for (int ir = 0; ir < this->rhopw->nrxx; ir++) - { - rea = this->rhopw->fft_bundle.get_auxr_data()[ir].real(); - sumrea += rea; - neg += std::min(0.0, rea); - ima += std::abs(this->rhopw->fft_bundle.get_auxr_data()[ir].imag()); - } - - #ifdef __MPI - Parallel_Reduce::reduce_pool(neg); - Parallel_Reduce::reduce_pool(ima); - Parallel_Reduce::reduce_pool(sumrea); - #endif - // mohan fix bug 2011-04-03 - neg = neg / (double)this->rhopw->nxyz * omega; - ima = ima / (double)this->rhopw->nxyz * omega; - sumrea = sumrea / (double)this->rhopw->nxyz * omega; - - if (((neg < -1.0e-4) && (is == 0 || PARAM.inp.nspin == 2)) || ima > 1.0e-4) - { - GlobalV::ofs_warning << " Warning: negative or imaginary starting charge : "; - GlobalV::ofs_warning << " neg = " << neg << " ima = " << ima << " SPIN = " << is << std::endl; - } - - } // end is - - double ne_tot = 0.0; - int spin0 = 1; - if (spin_number_need == 2) - { - spin0 = spin_number_need; - } - for (int is = 0; is < spin0; ++is) - { - GlobalV::ofs_warning << "\n SETUP ATOMIC RHO FOR SPIN " << is + 1 << std::endl; - ModuleBase::GlobalFunc::OUT(GlobalV::ofs_warning, "Electron number from rho", ne[is]); - ne_tot += ne[is]; - } - ModuleBase::GlobalFunc::OUT(GlobalV::ofs_warning, "total electron number from rho", ne_tot); - ModuleBase::GlobalFunc::OUT(GlobalV::ofs_warning, "should be", PARAM.inp.nelec); - - for (int is = 0; is < spin_number_need; ++is) - { - for (int ir = 0; ir < this->rhopw->nrxx; ++ir) - { - rho_in[is][ir] = rho_in[is][ir] / ne_tot * PARAM.inp.nelec; - } - } - } - - ModuleBase::timer::end("Charge", "atomic_rho"); - return; -} - void Charge::save_rho_before_sum_band() { - for (int is = 0; is < PARAM.inp.nspin; is++) + for (int is = 0; is < nspin; is++) { ModuleBase::GlobalFunc::DCOPY(rho[is], rho_save[is], this->rhopw->nrxx); - if (XC_Functional::get_ked_flag()) + if (this->meta_gga) { ModuleBase::GlobalFunc::DCOPY(kin_r[is], kin_r_save[is], this->rhopw->nrxx); } } return; } - -double Charge::cal_rho2ne(const double* rho_in) const -{ - assert(this->rhopw->nxyz > 0); // mohan add 2025-12-02 - - double ne = 0.0; - for (int ir = 0; ir < this->rhopw->nrxx; ir++) - { - ne += rho_in[ir]; - } -#ifdef __MPI - Parallel_Reduce::reduce_pool(ne); -#endif - ne = ne * *this->omega_ / (double)this->rhopw->nxyz; - - return ne; -} - -void Charge::check_rho() -{ - if (this->nspin==1 || this->nspin==4) - { - double ne = 0.0; - ne = this->cal_rho2ne(rho[0]); - if (std::abs(ne - PARAM.inp.nelec) > 1.0e-6) - { - ModuleBase::WARNING("Charge", "Charge is not equal to the number of electrons!"); - } - } - else if (this->nspin == 2) - { - // for spin up - double ne_up = 0.0; - ne_up = this->cal_rho2ne(rho[0]); - if (ne_up < 0.0) - { - ModuleBase::WARNING_QUIT("Charge", "Number of spin-down electrons set in starting magnetization exceeds all available."); - } - // for spin down - double ne_dn = 0.0; - ne_dn = this->cal_rho2ne(rho[1]); - if (ne_dn < 0.0) - { - ModuleBase::WARNING_QUIT("Charge", "Number of spin-up electrons set in starting magnetization exceeds all available."); - } - // for total charge - if (std::abs(ne_up + ne_dn - PARAM.inp.nelec) > 1.0e-6) - { - ModuleBase::WARNING("Charge", "Charge is not equal to the number of electrons!"); - } - } -} - -// LiuXh add 20180619 -void Charge::init_final_scf() -{ - ModuleBase::TITLE("Charge", "init_after_scf"); - - assert(allocate_rho_final_scf == false); - if (PARAM.inp.test_charge > 1) - { - std::cout << "\n spin_number = " << PARAM.inp.nspin << " real_point_number = " << this->rhopw->nrxx << std::endl; - } - - // allocate memory - rho = new double*[PARAM.inp.nspin]; - rhog = new std::complex*[PARAM.inp.nspin]; - rho_save = new double*[PARAM.inp.nspin]; - rhog_save = new std::complex*[PARAM.inp.nspin]; - - for (int is = 0; is < PARAM.inp.nspin; is++) - { - rho[is] = new double[this->rhopw->nrxx]; - rhog[is] = new std::complex[this->rhopw->npw]; - rho_save[is] = new double[this->rhopw->nrxx]; - rhog_save[is] = new std::complex[this->rhopw->npw]; - ModuleBase::GlobalFunc::ZEROS(rho[is], this->rhopw->nrxx); - ModuleBase::GlobalFunc::ZEROS(rhog[is], this->rhopw->npw); - ModuleBase::GlobalFunc::ZEROS(rho_save[is], this->rhopw->nrxx); - ModuleBase::GlobalFunc::ZEROS(rhog_save[is], this->rhopw->npw); - } - - ModuleBase::Memory::record("Chg::rho", sizeof(double) * PARAM.inp.nspin * this->rhopw->nrxx); - ModuleBase::Memory::record("Chg::rho_save", sizeof(double) * PARAM.inp.nspin * this->rhopw->nrxx); - ModuleBase::Memory::record("Chg::rhog", sizeof(double) * PARAM.inp.nspin * this->rhopw->npw); - ModuleBase::Memory::record("Chg::rhog_save", sizeof(double) * PARAM.inp.nspin * this->rhopw->npw); - - this->rho_core = new double[this->rhopw->nrxx]; // core charge in real space - ModuleBase::GlobalFunc::ZEROS(rho_core, this->rhopw->nrxx); - - this->rhog_core = new std::complex[this->rhopw->npw]; // reciprocal core charge - ModuleBase::GlobalFunc::ZEROS(rhog_core, this->rhopw->npw); - - ModuleBase::Memory::record("Chg::rho_core", sizeof(double) * this->rhopw->nrxx); - ModuleBase::Memory::record("Chg::rhog_core", sizeof(double) * this->rhopw->npw); - - this->allocate_rho_final_scf = true; - return; -} diff --git a/source/source_estate/module_charge/charge.h b/source/source_estate/module_charge/charge.h index 785985da40e..8b2e8592a8b 100644 --- a/source/source_estate/module_charge/charge.h +++ b/source/source_estate/module_charge/charge.h @@ -1,10 +1,12 @@ #ifndef CHARGE_H #define CHARGE_H +#include + #include "source_base/complexmatrix.h" #include "source_base/global_function.h" #include "source_base/global_variable.h" -#include "source_basis/module_pw/pw_basis.h" +#include "source_basis/module_pw/pw_basis_sup.h" #include "source_cell/module_symmetry/symmetry.h" // #include "source_estate/fp_energy.h" #include "source_base/parallel_grid.h" @@ -12,6 +14,11 @@ //a forward declaration of UnitCell class UnitCell; +namespace module_charge +{ +struct InitRhoCfg; +} + // Electron Charge Density class Charge { @@ -21,6 +28,12 @@ class Charge Charge(); ~Charge(); + // rho/rhog/kin_r views alias the vector-backed _space_* storage, so + // copying a Charge would duplicate dangling pointers into another + // object's vector buffer. Forbid copies until a deep copy is needed. + Charge(const Charge&) = delete; + Charge& operator=(const Charge&) = delete; + //========================================================== // MEMBER VARIABLES : // init_chg : "atomic" or "file" @@ -40,130 +53,106 @@ class Charge std::complex **rhog_save = nullptr; double **kin_r = nullptr; // kinetic energy density in real space, for meta-GGA - double **kin_r_save = nullptr; // kinetic energy density in real space, for meta-GGA + double **kin_r_save = nullptr; // same as kin_r, kept for mixing const Parallel_Grid* pgrid = nullptr; private: - //temporary - double *_space_rho = nullptr; - double *_space_rho_save = nullptr; - std::complex *_space_rhog = nullptr; - std::complex *_space_rhog_save = nullptr; - double *_space_kin_r = nullptr; - double *_space_kin_r_save = nullptr; + // Underlying contiguous storage backing the public rho/rhog/kin_r views. + // Each buffer holds nspin rows; rho[is] points at _space_rho.data()+is*nrxx. + // Owned here as std::vector so the storage self-manages (no raw new/delete). + std::vector _space_rho; + std::vector _space_rho_save; + std::vector> _space_rhog; + std::vector> _space_rhog_save; + std::vector _space_kin_r; + std::vector _space_kin_r_save; + + // Pointer arrays backing the public double** views (rho, rhog, etc.) + std::vector _ptrs_rho; + std::vector*> _ptrs_rhog; + std::vector _ptrs_rho_save; + std::vector*> _ptrs_rhog_save; + std::vector _ptrs_kin_r; + std::vector _ptrs_kin_r_save; + + // Contiguous storage for rho_core and rhog_core + std::vector _space_rho_core; + std::vector> _space_rhog_core; public: - double **nhat = nullptr; //compensation charge for PAW - double **nhat_save = nullptr; //compensation charge for PAW - // wenfei 2023-09-05 - double *rho_core = nullptr; std::complex *rhog_core = nullptr; - int prenspin = 1; - void set_rhopw(ModulePW::PW_Basis* rhopw_in); /** * @brief Init charge density from file or atomic pseudo-wave-functions * - * @param eferm_iout [out] fermi energy to be initialized * @param ucell [in] unit cell + * @param pgrid [in] parallel grid descriptor * @param strucFac [in] structure factor * @param symm [in] symmetry * @param klist [in] k points list if needed * @param wfcpw [in] PW basis for wave function if needed + * @param cfg [in] INPUT values for charge initialization */ void init_rho(const UnitCell& ucell, const Parallel_Grid& pgrid, const ModuleBase::ComplexMatrix& strucFac, ModuleSymmetry::Symmetry& symm, - const void* klist = nullptr, - const void* wfcpw = nullptr); - - // mohan add 2025-12-02 - bool kin_density() const; - - void allocate(const int &nspin_in, const bool kin_den); - - void atomic_rho(const int spin_number_need, - const double& omega, - double** rho_in, - const ModuleBase::ComplexMatrix& strucFac, - const UnitCell& ucell) const; - - void set_rho_core(const UnitCell& ucell, - const ModuleBase::ComplexMatrix& structure_factor, - const bool* numeric); - - void renormalize_rho(); - - double sum_rho() const; + const void* klist, + const void* wfcpw, + const module_charge::InitRhoCfg& cfg); - void save_rho_before_sum_band(); - - // for non-linear core correction - void non_linear_core_correction - ( - const bool &numeric, - const double omega, - const double tpiba2, - const int mesh, - const double *r, - const double *rab, - const double *rhoc, - double *rhocg - ) const; - - double cal_rho2ne(const double *rho_in) const; - - void check_rho(); // to check whether the charge density is normal - - void init_final_scf(); //LiuXh add 20180619 - - public: /** - * @brief Sum rho at different pools (k-point parallelism). - * Only used when GlobalV::KPAR > 1 + * @brief Allocate the rho/rhog/kin_r buffers + * + * @param nspin_in number of spins + * @param kin_den whether to allocate the kinetic-energy density buffers + * @param meta_gga whether the functional is meta-GGA (kin_r carries XC + * physics, not just ELF output); stored for tau handling + * @param test_charge verbosity flag (PARAM.inp.test_charge) */ - void rho_mpi(); + void allocate(const int &nspin_in, const bool kin_den, const bool meta_gga, + const int test_charge); /** - * @brief Sum kin_r at different pools (k-point/band parallelism). - * Only used when GlobalV::KPAR * bndpar > 1 + * @brief Renormalize rho so that its integral equals the electron number + * + * @param nelec target total electron number (PARAM.inp.nelec) + * @param omega current unit-cell volume. Must be ucell.omega, NOT + * rhopw->omega, because in variable-cell calculations (e.g. NPT) + * rhopw->omega is stale (pw_rho/pw_rhod are not rebuilt on cell + * change) while ucell.omega is updated every step. Using the stale + * volume gives a wrong electron count and a wrong renormalization + * factor, which corrupts the stress. */ - void kin_r_mpi(); + void renormalize_rho(const double nelec, const double omega); + + double sum_rho(const double omega) const; - /** - * @brief Reduce among different pools - * If NPROC_IN_POOLs are all the same, use GlobalV::KP_WORLD - * else, gather rho in a POOL, and then reduce among different POOLs - * - * @param array_rho f(rho): an array [nrxx] - */ - void reduce_diff_pools(double* array_rho) const; + void save_rho_before_sum_band(); - void set_omega(double* omega_in){this->omega_ = omega_in;}; + /// @brief whether allocate() has already claimed the density buffers + /// @return true once allocate() has run; allocate() is idempotent and + /// returns early when this is already true + bool get_allocate_rho() const { return allocate_rho; } // mohan add 2021-02-20 int nrxx=0; // number of r vectors in this processor int nxyz = 0; // total number of r vectors int ngmc=0; // number of g vectors in this processor int nspin=0; // number of spins + bool meta_gga = false; // whether the functional is meta-GGA (set by allocate) ModulePW::PW_Basis* rhopw = nullptr;// When double_grid is used, rhopw = rhodpw (dense grid) - bool cal_elf = false; // whether to calculate electron localization function (ELF) private: void destroy(); // free arrays liuyu 2023-03-12 - double* omega_ = nullptr; // omega for non-linear core correction - bool allocate_rho; - - bool allocate_rho_final_scf; // LiuXh add 20180606 }; #endif // charge diff --git a/source/source_estate/module_charge/charge_extra.cpp b/source/source_estate/module_charge/charge_extra.cpp deleted file mode 100644 index 7513469a31b..00000000000 --- a/source/source_estate/module_charge/charge_extra.cpp +++ /dev/null @@ -1,318 +0,0 @@ -#include "charge_extra.h" - -#include "source_base/global_function.h" -#include "source_base/global_variable.h" -#include "source_base/timer.h" -#include "source_base/tool_threading.h" -#include "source_io/module_output/cube_io.h" - -Charge_Extra::Charge_Extra() -{ -} - -Charge_Extra::~Charge_Extra() -{ - if(pot_order == 3) - { - delete[] dis_old1; - delete[] dis_old2; - delete[] dis_now; - } -} - -void Charge_Extra::Init_CE(const int& nspin, const int& natom, const int& nrxx, const std::string chg_extrap) -{ - if (chg_extrap == "none") - { - pot_order = 0; - } - else if (chg_extrap == "atomic") - { - pot_order = 1; - } - else if (chg_extrap == "first-order") - { - pot_order = 2; - } - else if (chg_extrap == "second-order") - { - pot_order = 3; - } - else - { - ModuleBase::WARNING_QUIT("Charge_Extra","charge extrapolation method is not available !"); - } - - this->nspin = nspin; - - if (pot_order > 0) - { - // delta_rho1.resize(this->nspin, std::vector(nrxx, 0.0)); - // delta_rho2.resize(this->nspin, std::vector(nrxx, 0.0)); - // delta_rho3.resize(this->nspin, std::vector(nrxx, 0.0)); - // qianrui replace the above code with the following code. - // The above code cannot passed valgrind tests, which has an invalid read of size 32. - delta_rho1.resize(this->nspin); - delta_rho2.resize(this->nspin); - delta_rho3.resize(this->nspin); - for (int is = 0; is < this->nspin; is++) - { - delta_rho1[is].resize(nrxx, 0.0); - delta_rho2[is].resize(nrxx, 0.0); - delta_rho3[is].resize(nrxx, 0.0); - } - } - - if(pot_order == 3) - { - dis_old1 = new ModuleBase::Vector3[natom]; - dis_old2 = new ModuleBase::Vector3[natom]; - dis_now = new ModuleBase::Vector3[natom]; - } - - alpha = 1.0; - beta = 0.0; -} - -void Charge_Extra::extrapolate_charge( - Parallel_Grid* Pgrid, - UnitCell& ucell, - Charge* chr, - Structure_Factor* sf, - std::ofstream& ofs_running, - std::ofstream& ofs_warning) -{ - ModuleBase::TITLE("Charge_Extra","extrapolate_charge"); - ModuleBase::timer::start("Charge_Extra", "extrapolate_charge"); - //------------------------------------------------------- - // Charge density extrapolation: - // - // * pot_order=0 : copy the old potential (nothing is done); - // * pot_order=1 : subtract old atomic charge density and sum the new - // if dynamics is done the routine extrapolates also the difference - // between the scf charge and the atomic one; - // * pot_order=2 : first order extrapolation: - // \[ \rho(t+dt) = 2\ \rho(t)-\rho(t-dt); \] - // * pot_order=3 : second order extrapolation: - // \[ \rho(t+dt) = \rho(t) + \alpha_0\ (\rho(t) - \rho(t-dt)) - // + \beta_0\ (\rho(t-dt)- \rho(t-2 dt)). \] - // - // The \(\alpha_0\) and \(\beta_0\) parameters are calculated in find_alpha_and_beta() - // so that \(|\tau'-\tau(t+dt)|\) is minimum. \(\tau'\) and \(\tau(t+dt)\) are respectively - // the atomic positions at time t+dt and the extrapolated one: - // \[ \tau(t+dt) = \tau(t) + \alpha_0\ ( \tau(t) - \tau(t-dt) ) - // + \beta_0\ ( \tau(t-dt) - \tau(t-2 dt) ). \] - //------------------------------------------------------- - - rho_extr = std::min(istep, pot_order); - if(rho_extr == 0) - { - sf->setup(&ucell, *Pgrid, chr->rhopw); - ofs_running << " charge density from previous step !" << std::endl; - ModuleBase::timer::end("Charge_Extra", "extrapolate_charge"); - return; - } - - - // if(lsda || noncolin) rho2zeta(); - - if(rho_extr == 1) - { - ofs_running << " NEW-OLD atomic charge density approx. for the potential !" << std::endl; - -#ifdef _OPENMP -#pragma omp parallel for collapse(2) schedule(static, 128) -#endif - for (int is = 0; is < this->nspin; is++) - { - for (int ir = 0; ir < chr->rhopw->nrxx; ir++) - { - chr->rho[is][ir] = delta_rho1[is][ir]; - } - } - } - // first order extrapolation - else if(rho_extr ==2) - { - ofs_running << " first order charge density extrapolation !" << std::endl; - -#ifdef _OPENMP -#pragma omp parallel for collapse(2) schedule(static, 128) -#endif - for (int is = 0; is < this->nspin; is++) - { - for (int ir = 0; ir < chr->rhopw->nrxx; ir++) - { - chr->rho[is][ir] = 2 * delta_rho1[is][ir] - delta_rho2[is][ir]; - } - } - } - // second order extrapolation - else - { - ofs_running << " second order charge density extrapolation !" << std::endl; - - find_alpha_and_beta(ucell.nat, ofs_running, ofs_warning); - - const double one_add_alpha = 1 + alpha; - const double beta_alpha = beta - alpha; - -#ifdef _OPENMP -#pragma omp parallel for collapse(2) schedule(static, 64) -#endif - for (int is = 0; is < this->nspin; is++) - { - for (int ir = 0; ir < chr->rhopw->nrxx; ir++) - { - chr->rho[is][ir] - = one_add_alpha * delta_rho1[is][ir] + beta_alpha * delta_rho2[is][ir] - beta * delta_rho3[is][ir]; - } - } - } - - sf->setup(&ucell, *Pgrid, chr->rhopw); - double** rho_atom = new double*[this->nspin]; - for (int is = 0; is < this->nspin; is++) - { - rho_atom[is] = new double[chr->rhopw->nrxx]; - } - chr->atomic_rho(this->nspin, ucell.omega, rho_atom, sf->strucFac, ucell); -#ifdef _OPENMP -#pragma omp parallel for collapse(2) schedule(static, 512) -#endif - for (int is = 0; is < this->nspin; is++) - { - for(int ir=0; irrhopw->nrxx; ir++) - { - chr->rho[is][ir] /= ucell.omega; - chr->rho[is][ir] += rho_atom[is][ir]; - } - } - - for (int is = 0; is < this->nspin; is++) - { - delete[] rho_atom[is]; - } - delete[] rho_atom; - ModuleBase::timer::end("Charge_Extra", "extrapolate_charge"); - return; -} - -void Charge_Extra::find_alpha_and_beta(const int& natom, std::ofstream& ofs_running, std::ofstream& ofs_warning) -{ - if(istep < 3) return; - - double a11 = 0.0; - double a12 = 0.0; - double a21 = 0.0; - double a22 = 0.0; - double b1 = 0.0; - double b2 = 0.0; - double c = 0.0; - double det = 0.0; -#ifdef _OPENMP -#pragma omp parallel for schedule(static, 16) \ - reduction(+:a11) reduction(+:a12) reduction(+:a22) \ - reduction(+:b1) reduction(+:b2) reduction(+:c) -#endif - for(int i=0; i* dis_old1 = nullptr; ///< dis_old2 = pos_old1 - pos_old2 - ModuleBase::Vector3* dis_old2 = nullptr; ///< dis_old1 = pos_now - pos_old1 - ModuleBase::Vector3* dis_now = nullptr; ///< dis_now = pos_next - pos_now - - std::vector> delta_rho1; ///< the last step difference of rho and atomic_rho - std::vector> delta_rho2; ///< the second last step difference of rho and atomic_rho - std::vector> delta_rho3; ///< the third last step difference of rho and atomic_rho - - double alpha; ///< parameter used in the second order extrapolation - double beta; ///< parameter used in the second order extrapolation - - /** - * @brief determine alpha and beta - * - * @param natom the number of atoms - * @param ofs_running the output stream - * @param ofs_warning the output stream - */ - void find_alpha_and_beta(const int& natom, std::ofstream& ofs_running, std::ofstream& ofs_warning); -}; - -#endif diff --git a/source/source_estate/module_charge/charge_init.cpp b/source/source_estate/module_charge/charge_init.cpp deleted file mode 100644 index 672d800f718..00000000000 --- a/source/source_estate/module_charge/charge_init.cpp +++ /dev/null @@ -1,460 +0,0 @@ -#include -#include - -#include "charge.h" -#include "source_base/global_function.h" -#include "source_base/global_variable.h" -#include "source_io/module_parameter/parameter.h" -#include "source_base/libm/libm.h" -#include "source_base/math_integral.h" -#include "source_base/math_sphbes.h" -#include "source_base/parallel_reduce.h" -#include "source_base/timer.h" -#include "source_base/tool_threading.h" -#include "source_cell/magnetism.h" -#include "source_base/parallel_grid.h" -#include "source_io/module_output/cube_io.h" -#include "source_estate/rhog_io.h" -#include "source_io/module_wf/read_wf2rho_pw.h" -#include "source_io/module_restart/restart.h" -#include "source_hamilt/module_xc/xc_functional.h" -#include "source_cell/klist.h" -#include "source_base/module_parallel/para_world.h" -#include "source_base/module_parallel/para_tag.h" -#include "source_base/module_parallel/para_bridge.h" - -void Charge::init_rho(const UnitCell& ucell, - const Parallel_Grid& pgrid, - const ModuleBase::ComplexMatrix& strucFac, - ModuleSymmetry::Symmetry& symm, - const void* klist, - const void* wfcpw) -{ - ModuleBase::GlobalFunc::OUT(GlobalV::ofs_running, "init_chg", PARAM.inp.init_chg); - - const int nspin = PARAM.inp.nspin; - assert(nspin>0); - - std::string init_chg_upper = PARAM.inp.init_chg; - std::transform(init_chg_upper.begin(), init_chg_upper.end(), init_chg_upper.begin(), ::toupper); - std::cout << " START CHARGE : " << init_chg_upper << std::endl; - - // we need to set the omega for the charge density - set_omega(&ucell.omega); - this->pgrid = &pgrid; - - bool read_error = false; - bool read_kin_error = false; - if (PARAM.inp.init_chg == "file" || PARAM.inp.init_chg == "auto") - { - GlobalV::ofs_running << " Read electron density from file" << std::endl; - - // try to read charge from binary file first, which is the same as QE - // liuyu 2023-12-05 - std::stringstream binary; - binary << PARAM.globalv.global_readin_dir << PARAM.inp.suffix + "-CHARGE-DENSITY.restart"; - // Temporary bridge: use factory until ParaCollection is wired into driver. - Parallel::ParaWorld pw_world = Parallel::make_pw_world(); - if (elecstate::read_rhog(binary.str(), rhopw, nspin, rhog, pw_world, &GlobalV::ofs_warning)) - { - GlobalV::ofs_running << " Read electron density from file: " << binary.str() << std::endl; - for (int is = 0; is < nspin; ++is) - { - rhopw->recip2real(rhog[is], rho[is]); - } - } - else - { - for (int is = 0; is < nspin; ++is) - { - std::stringstream ssc; - - if(nspin==1) - { - ssc << PARAM.globalv.global_readin_dir << "chg.cube"; - } - else - { - ssc << PARAM.globalv.global_readin_dir << "chgs" << is + 1 << ".cube"; - } - - - if (ModuleIO::read_vdata_palgrid(pgrid, - (PARAM.inp.esolver_type == "sdft" ? GlobalV::RANK_IN_BPGROUP : GlobalV::MY_RANK), - GlobalV::ofs_running, - ssc.str(), - this->rho[is], - ucell.nat)) - { - GlobalV::ofs_running << " Read electron density from file: " << ssc.str() << std::endl; - } - else if (is > 0) // nspin=2 or 4 - { - if (is == 1) // failed at the second spin - { - std::cout << " Incomplete electron density file." << std::endl; - read_error = true; - break; - } - else if (is == 2) // read 2 files when nspin=4 - { - GlobalV::ofs_running << " Didn't read in the electron density but would rearrange it later. " - << std::endl; - } - else if (is == 3) // read 2 files when nspin=4 - { - GlobalV::ofs_running << " rearrange electron density " << std::endl; - for (int ir = 0; ir < this->rhopw->nrxx; ir++) - { - this->rho[3][ir] = this->rho[0][ir] - this->rho[1][ir]; - this->rho[0][ir] = this->rho[0][ir] + this->rho[1][ir]; - this->rho[1][ir] = 0.0; - this->rho[2][ir] = 0.0; - } - } - } - else - { - read_error = true; - break; - } - } - } - - if (read_error) - { - const std::string warn_msg - = " WARNING: \"init_chg\" is enabled but ABACUS failed to read\n charge density from file.\n" - " Please check if there is chg.cube (for nspin=1) or chgsx.cube (x=1,2,etc.) or\n {suffix}-CHARGE-DENSITY.restart in the " - "directory.\n"; - std::cout << warn_msg; - if (PARAM.inp.init_chg == "file") - { - ModuleBase::WARNING_QUIT("Charge::init_rho", - "Failed to read in charge density from file.\n For initializing atomic " - "charge in calculations,\n please set init_chg to atomic in INPUT."); - } - } - - if (XC_Functional::get_ked_flag()) - { - // If the charge density is not read in, then the kinetic energy density is not read in either - if (!read_error) - { - GlobalV::ofs_running << " try to read kinetic energy density from file" << std::endl; - // try to read charge from binary file first, which is the same as QE - std::vector> kin_g_space(nspin * this->ngmc, {0.0, 0.0}); - std::vector*> kin_g; - for (int is = 0; is < nspin; is++) - { - kin_g.push_back(kin_g_space.data() + is * this->ngmc); - } - - std::stringstream binary; - binary << PARAM.globalv.global_readin_dir << PARAM.inp.suffix + "-TAU-DENSITY.restart"; - if (elecstate::read_rhog(binary.str(), rhopw, nspin, kin_g.data(), pw_world, &GlobalV::ofs_warning)) - { - GlobalV::ofs_running << " Read in the kinetic energy density: " << binary.str() << std::endl; - for (int is = 0; is < nspin; ++is) - { - rhopw->recip2real(kin_g[is], this->kin_r[is]); - } - } - else - { - for (int is = 0; is < nspin; is++) - { - std::stringstream ssc; - ssc << PARAM.globalv.global_readin_dir << "SPIN" << is + 1 << "_TAU.cube"; - // mohan update 2012-02-10, sunliang update 2023-03-09 - if (ModuleIO::read_vdata_palgrid( - pgrid, - (PARAM.inp.esolver_type == "sdft" ? GlobalV::RANK_IN_BPGROUP : GlobalV::MY_RANK), - GlobalV::ofs_running, - ssc.str(), - this->kin_r[is], - ucell.nat)) - { - GlobalV::ofs_running << " Read in the kinetic energy density: " << ssc.str() << std::endl; - } - else - { - read_kin_error = true; - std::cout << " WARNING: \"init_chg\" is enabled but ABACUS failed to read kinetic energy " - "density from file.\n" - " Please check if there is SPINX_TAU.cube (X=1,...) or " - "{suffix}-TAU-DENSITY.restart in the directory.\n" - << std::endl; - break; - } - } - } - } - else - { - read_kin_error = true; - } - } - } - - if (PARAM.inp.init_chg == "atomic" || read_error) - { - if (read_error) - { - std::cout << " Charge::init_rho: use atomic initialization instead." << std::endl; - } - this->atomic_rho(nspin, ucell.omega, rho, strucFac, ucell); - } - - // initial tau = 3/5 rho^2/3, Thomas-Fermi - if (XC_Functional::get_ked_flag()) - { - if (PARAM.inp.init_chg == "atomic" || read_kin_error) - { - if (read_kin_error) - { - std::cout << " Charge::init_rho: init kinetic energy density from rho." << std::endl; - } - const double fact = (3.0 / 5.0) * pow(3.0 * ModuleBase::PI * ModuleBase::PI, 2.0 / 3.0); - for (int is = 0; is < nspin; ++is) - { - for (int ir = 0; ir < this->rhopw->nrxx; ++ir) - { - kin_r[is][ir] = fact * pow(std::abs(rho[is][ir]) * nspin, 5.0 / 3.0) / nspin; - } - } - } - } - - // Peize Lin add 2020.04.04 - if (GlobalC::restart.info_load.load_charge && !GlobalC::restart.info_load.load_charge_finish) - { - for (int is = 0; is < nspin; ++is) - { - try - { - GlobalC::restart.load_disk("charge", is, this->nrxx, rho[is]); - } - catch (const std::exception& e) - { - // try to load from the output of `out_chg` - std::stringstream ssc; - ssc << PARAM.globalv.global_readin_dir << "chgs" << is + 1 << ".cube"; - if (ModuleIO::read_vdata_palgrid(pgrid, - (PARAM.inp.esolver_type == "sdft" ? GlobalV::RANK_IN_BPGROUP : GlobalV::MY_RANK), - GlobalV::ofs_running, - ssc.str(), - this->rho[is], - ucell.nat)) - { - GlobalV::ofs_running << " Read in electron density: " << ssc.str() << std::endl; - } - } - } - GlobalC::restart.info_load.load_charge_finish = true; - } - - if (PARAM.inp.init_chg == "wfc") - { - if (wfcpw == nullptr) - { - ModuleBase::WARNING_QUIT("Charge::init_rho", "wfc is only supported for PW-KSDFT."); - } - - const ModulePW::PW_Basis_K* pw_wfc = reinterpret_cast(const_cast(wfcpw)); - const K_Vectors* kv = reinterpret_cast(klist); - - ModuleIO::read_wf2rho_pw(pw_wfc, symm, *this, - PARAM.globalv.global_readin_dir, - GlobalV::KPAR, GlobalV::MY_POOL, GlobalV::MY_RANK, - GlobalV::NPROC_IN_POOL, GlobalV::RANK_IN_POOL, - PARAM.inp.nbands, nspin, PARAM.globalv.npol, - kv->get_nkstot(),kv->ik2iktot,kv->isk,GlobalV::ofs_running); - } -} - -//========================================================== -// computes the core charge on the real space 3D mesh. -//========================================================== -void Charge::set_rho_core(const UnitCell& ucell, - const ModuleBase::ComplexMatrix& structure_factor, - const bool* numeric) -{ - ModuleBase::TITLE("Charge","set_rho_core"); - ModuleBase::timer::start("Charge","set_rho_core"); - - bool bl = false; - for (int it = 0; itrho_core, this->rhopw->nrxx); - ModuleBase::timer::end("Charge","set_rho_core"); - return; - } - - double *rhocg = new double[this->rhopw->ngg]; - ModuleBase::GlobalFunc::ZEROS(rhocg, this->rhopw->ngg ); - - // three dimension. - std::complex *vg = new std::complex[this->rhopw->npw]; - - for (int it = 0; it < ucell.ntype;it++) - { - if (ucell.atoms[it].ncpp.nlcc) - { -//---------------------------------------------------------- -// EXPLAIN : drhoc compute the radial fourier transform for -// each shell of g vec -//---------------------------------------------------------- - this->non_linear_core_correction( - numeric, - ucell.omega, - ucell.tpiba2, - ucell.atoms[it].ncpp.msh, - ucell.atoms[it].ncpp.r.data(), - ucell.atoms[it].ncpp.rab.data(), - ucell.atoms[it].ncpp.rho_atc.data(), - rhocg); -//---------------------------------------------------------- -// EXPLAIN : multiply by the structure factor and sum -//---------------------------------------------------------- - for (int ig = 0; ig < this->rhopw->npw ; ig++) - { - vg[ig] += structure_factor(it, ig) * rhocg[this->rhopw->ig2igg[ig]]; - } - } - } - - // for tmp use. - for(int ig=0; ig< this->rhopw->npw; ig++) - { - this->rhog_core[ig] = vg[ig]; - } - - this->rhopw->recip2real(vg, this->rho_core); - - // test on the charge and computation of the core energy - double rhoima = 0.0; - double rhoneg = 0.0; - for (int ir = 0; ir < this->rhopw->nrxx; ir++) - { - rhoneg += std::min(0.0, this->rhopw->fft_bundle.get_auxr_data()[ir].real()); - rhoima += std::abs(this->rhopw->fft_bundle.get_auxr_data()[ir].imag()); - // NOTE: Core charge is computed in reciprocal space and brought to real - // space by FFT. For non smooth core charges (or insufficient cut-off) - // this may result in negative values in some grid points. - // Up to October 1999 the core charge was forced to be positive definite. - // This induces an error in the force, and probably stress, calculation if - // the number of grid points where the core charge would be otherwise neg - // is large. The error disappears for sufficiently high cut-off, but may be - // rather large and it is better to leave the core charge as it is. - // If you insist to have it positive definite (with the possible problems - // mentioned above) uncomment the following lines. SdG, Oct 15 1999 - } - - // mohan fix bug 2011-04-03 - Parallel_Reduce::reduce_pool(rhoneg); - Parallel_Reduce::reduce_pool(rhoima); - - // mohan changed 2010-2-2, make this same as in atomic_rho. - // still lack something...... - rhoneg /= this->rhopw->nxyz * ucell.omega; - rhoima /= this->rhopw->nxyz * ucell.omega; - - // calculate core_only exch-corr energy etxcc=E_xc[rho_core] if required - // The term was present in previous versions of the code but it shouldn't - delete [] rhocg; - delete [] vg; - ModuleBase::timer::end("Charge","set_rho_core"); - return; -} // end subroutine set_rhoc - - -void Charge::non_linear_core_correction -( - const bool &numeric, - const double omega, - const double tpiba2, - const int mesh, - const double *r, - const double *rab, - const double *rhoc, - double *rhocg) const -{ - ModuleBase::TITLE("charge","drhoc"); - - // use labmda instead of repeating codes - const auto kernel = [&](int num_threads, int thread_id) - { - - double gx = 0.0; - double rhocg1 = 0.0; - double *aux = nullptr; - - // here we compute the fourier transform is the charge in numeric form - if (numeric) - { - aux = new double [mesh]; - // G=0 term - - int igl0 = 0; - if (this->rhopw->gg_uniq [0] < 1.0e-8) - { - // single thread term - if (thread_id == 0) - { - for (int ir = 0;ir < mesh; ir++) - { - aux [ir] = r [ir] * r [ir] * rhoc [ir]; - } - ModuleBase::Integral::Simpson_Integral(mesh, aux, rab, rhocg1); - //rhocg [1] = fpi * rhocg1 / omega; - rhocg [0] = ModuleBase::FOUR_PI * rhocg1 / omega;//mohan modify 2008-01-19 - } - igl0 = 1; - } - - int igl_beg, igl_end; - // exclude igl0 - ModuleBase::TASK_DIST_1D(num_threads, thread_id, this->rhopw->ngg - igl0, igl_beg, igl_end); - igl_beg += igl0; - igl_end += igl_beg; - - // G <> 0 term - for (int igl = igl_beg; igl < igl_end;igl++) - { - gx = sqrt(this->rhopw->gg_uniq[igl] * tpiba2); - ModuleBase::Sphbes::Spherical_Bessel(mesh, r, gx, 0, aux); - for (int ir = 0;ir < mesh; ir++) - { - aux [ir] = r[ir] * r[ir] * rhoc [ir] * aux [ir]; - } // enddo - ModuleBase::Integral::Simpson_Integral(mesh, aux, rab, rhocg1); - rhocg [igl] = ModuleBase::FOUR_PI * rhocg1 / omega; - } // enddo - delete [] aux; - } - else - { - // here the case where the charge is in analytic form, - // check old version before 2008-12-9 - } - - }; // end kernel - - // do not use omp parallel when this function is already in parallel block - // - // it is called in parallel block in Forces::cal_force_cc, - // but not in other funtcion such as Stress_Func::stress_cc. - ModuleBase::TRY_OMP_PARALLEL(kernel); - - return; -} diff --git a/source/source_estate/module_charge/charge_mixing.cpp b/source/source_estate/module_charge/charge_mixing.cpp deleted file mode 100644 index 640fa480c0a..00000000000 --- a/source/source_estate/module_charge/charge_mixing.cpp +++ /dev/null @@ -1,292 +0,0 @@ -#include "charge_mixing.h" - -#include "source_io/module_parameter/parameter.h" -#include "source_base/module_mixing/broyden_mixing.h" -#include "source_base/module_mixing/pulay_mixing.h" -#include "source_base/parallel_common.h" -#include "source_base/timer.h" -#include "source_hamilt/module_xc/xc_functional.h" - -Charge_Mixing::Charge_Mixing() -{ - this->mixing = nullptr; - this->mixing_highf = nullptr; -} - -Charge_Mixing::~Charge_Mixing() -{ - if(this->mixing != nullptr) - { - delete this->mixing; - this->mixing = nullptr; - } - - if(this->mixing_highf != nullptr) - { - delete this->mixing_highf; - this->mixing_highf = nullptr; - } -} - -void Charge_Mixing::set_mixing(const std::string& mixing_mode_in, - const double& mixing_beta_in, - const int& mixing_ndim_in, - const double& mixing_gg0_in, - const bool& mixing_tau_in, - const double& mixing_beta_mag_in, - const double& mixing_gg0_mag_in, - const double& mixing_gg0_min_in, - const double& mixing_angle_in, - const bool& mixing_dmr_in, - double& omega_in, - double& tpiba_in) -{ - // get private mixing parameters - this->mixing_mode = mixing_mode_in; - this->mixing_beta = mixing_beta_in; - this->mixing_beta_mag = mixing_beta_mag_in; - this->mixing_ndim = mixing_ndim_in; - this->mixing_gg0 = mixing_gg0_in; - this->mixing_tau = mixing_tau_in; - this->mixing_gg0_mag = mixing_gg0_mag_in; - this->mixing_gg0_min = mixing_gg0_min_in; - this->mixing_angle = mixing_angle_in; - this->mixing_dmr = mixing_dmr_in; - this->omega = &omega_in; - this->tpiba = &tpiba_in; - // check the paramters - if (this->mixing_beta > 1.0 || this->mixing_beta < 0.0) - { - ModuleBase::WARNING_QUIT("Charge_Mixing", "You'd better set mixing_beta to [0.0, 1.0]!"); - } - if (PARAM.inp.nspin >= 2 && this->mixing_beta_mag < 0.0) - { - ModuleBase::WARNING_QUIT("Charge_Mixing", "You'd better set mixing_beta_mag >= 0.0!"); - } - - if (!(this->mixing_mode == "plain" || this->mixing_mode == "broyden" || this->mixing_mode == "pulay")) - { - ModuleBase::WARNING_QUIT("Charge_Mixing", "This Mixing mode is not implemended yet,coming soon."); - } - - // print into running.log - //GlobalV::ofs_running << "\n\n"; - GlobalV::ofs_running << "\n"; - GlobalV::ofs_running << " >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>" - ">>>>" << std::endl; - GlobalV::ofs_running << " | " - " |" << std::endl; - GlobalV::ofs_running << " | Setup charge mixing parameters " - " |" << std::endl; - GlobalV::ofs_running << " | " - " |" << std::endl; - GlobalV::ofs_running << " <<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<" - "<<<<" << std::endl; - GlobalV::ofs_running << "\n"; - - - ModuleBase::GlobalFunc::OUT(GlobalV::ofs_running, "mixing_type", this->mixing_mode); - ModuleBase::GlobalFunc::OUT(GlobalV::ofs_running, "mixing_beta", this->mixing_beta); - ModuleBase::GlobalFunc::OUT(GlobalV::ofs_running, "mixing_gg0", this->mixing_gg0); - ModuleBase::GlobalFunc::OUT(GlobalV::ofs_running, "mixing_gg0_min", PARAM.inp.mixing_gg0_min); - - if (PARAM.inp.nspin==2 || PARAM.inp.nspin==4) - { - ModuleBase::GlobalFunc::OUT(GlobalV::ofs_running, "mixing_beta_mag", this->mixing_beta_mag); - ModuleBase::GlobalFunc::OUT(GlobalV::ofs_running, "mixing_gg0_mag", PARAM.inp.mixing_gg0_mag); - } - if (PARAM.inp.mixing_angle > 0) - { - ModuleBase::GlobalFunc::OUT(GlobalV::ofs_running, "mixing_angle", PARAM.inp.mixing_angle); - } - - ModuleBase::GlobalFunc::OUT(GlobalV::ofs_running, "mixing_ndim", this->mixing_ndim); - - return; -} - -void Charge_Mixing::init_mixing() -{ - // this init should be called at the 1-st iteration of each scf loop - - ModuleBase::TITLE("Charge_Mixing", "init_mixing"); - ModuleBase::timer::start("Charge_Mixing", "init_mixing"); - - // (re)construct mixing object - if (this->mixing_mode == "broyden") - { - delete this->mixing; - this->mixing = new Base_Mixing::Broyden_Mixing(this->mixing_ndim, this->mixing_beta); - } - else if (this->mixing_mode == "plain") - { - delete this->mixing; - this->mixing = new Base_Mixing::Plain_Mixing(this->mixing_beta); - } - else if (this->mixing_mode == "pulay") - { - delete this->mixing; - this->mixing = new Base_Mixing::Pulay_Mixing(this->mixing_ndim, this->mixing_beta); - } - else - { - ModuleBase::WARNING_QUIT("Charge_Mixing", "This Mixing mode is not implemended yet,coming soon."); - } - - if ( PARAM.globalv.double_grid) - { - // ONLY smooth part of charge density is mixed by specific mixing method - // The high_frequency part is mixed by plain mixing method. - delete this->mixing_highf; - this->mixing_highf = new Base_Mixing::Plain_Mixing(this->mixing_beta); - } - - // allocate memory for mixing data, if exists, free it first and then allocate new memory - // initailize rho_mdata - if (PARAM.inp.scf_thr_type == 1) - { - if (PARAM.inp.nspin == 4 && PARAM.inp.mixing_angle > 0 ) - { - this->mixing->init_mixing_data(this->rho_mdata, - this->rhopw->npw * 2, - sizeof(std::complex)); - } - else - { - this->mixing->init_mixing_data(this->rho_mdata, - this->rhopw->npw * PARAM.inp.nspin, - sizeof(std::complex)); - } - } - else - { - if (PARAM.inp.nspin == 4 && PARAM.inp.mixing_angle > 0 ) - { - this->mixing->init_mixing_data(this->rho_mdata, this->rhopw->nrxx * 2, sizeof(double)); - } - else - { - this->mixing->init_mixing_data(this->rho_mdata, this->rhopw->nrxx * PARAM.inp.nspin, sizeof(double)); - } - } - - // initailize tau_mdata - if ((XC_Functional::get_ked_flag()) && mixing_tau) - { - if (PARAM.inp.scf_thr_type == 1) - { - this->mixing->init_mixing_data(this->tau_mdata, - this->rhopw->npw * PARAM.inp.nspin, - sizeof(std::complex)); - } - else - { - this->mixing->init_mixing_data(this->tau_mdata, this->rhopw->nrxx * PARAM.inp.nspin, sizeof(double)); - } - } - - ModuleBase::timer::end("Charge_Mixing", "init_mixing"); - - return; -} - -void Charge_Mixing::set_rhopw(ModulePW::PW_Basis* rhopw_in, ModulePW::PW_Basis* rhodpw_in) -{ - this->rhopw = rhopw_in; - this->rhodpw = rhodpw_in; -} - -void Charge_Mixing::mix_reset() -{ - this->mixing->reset(); - this->rho_mdata.reset(); - // initailize tau_mdata - if ((XC_Functional::get_ked_flag()) && mixing_tau) - { - this->tau_mdata.reset(); - } -} - -bool Charge_Mixing::if_scf_oscillate(const int iteration, const double drho, const int iternum_used, const double threshold) -{ - ModuleBase::TITLE("Charge_Mixing", "if_scf_oscillate"); - - if(this->_drho_history.size() == 0) - { - this->_drho_history.resize(PARAM.inp.scf_nmax); - } - - // add drho into history - this->_drho_history[iteration - 1] = drho; - - if(threshold >= 0) // close the function - { - return false; - } - - // check if the history is long enough - if(iteration < iternum_used + this->mixing_restart_last) - { - return false; - } - - // calculate the slope of the last iternum_used iterations' drho - double slope = 0.0; - - // Least Squares Method - // this part is too short, so I do not design it as a free function in principle - double sumX = 0, sumY = 0, sumXY = 0, sumXX = 0; - for (int i = iteration - iternum_used; i < iteration; i++) - { - sumX += i; - sumY += std::log10(this->_drho_history[i]); - sumXY += i * std::log10(this->_drho_history[i]); - sumXX += i * i; - } - double numerator = iternum_used * sumXY - sumX * sumY; - double denominator = iternum_used * sumXX - sumX * sumX; - if (denominator == 0) { - return false; - } - slope = numerator / denominator; - - // if the slope is less than the threshold, return true - if(slope > threshold) - { - return true; - } - - return false; -} - -void Charge_Mixing::allocate_mixing_uom(int uom_size) -{ - ModuleBase::TITLE("Charge_Mixing", "allocate_mixing_uom"); - ModuleBase::timer::start("Charge_Mixing", "allocate_mixing_uom"); - // For nspin=2, uom_size already includes both spin channels - // (uterm_mat.size() = pot_index * 2 for nspin=2) - // So uom_fold should always be 1 - this->mixing->init_mixing_data(this->uom_mdata, uom_size, sizeof(double)); - this->uom_mdata.reset(); - ModuleBase::timer::end("Charge_Mixing", "allocate_mixing_uom"); - return; -} - -void Charge_Mixing::mix_uom(std::vector& uom_in, std::vector& uom_save_in) -{ - ModuleBase::TITLE("Charge_Mixing", "mix_uom"); - ModuleBase::timer::start("Charge_Mixing", "mix_uom"); - double* uom_value_out = uom_in.data(); - double* uom_value_in = uom_save_in.data(); - // For all nspin cases, uom_array layout is already fully sized - // and mixing operates on the entire array - this->mixing->push_data(this->uom_mdata, uom_value_in, uom_value_out, nullptr, false); - this->mixing->mix_data(this->uom_mdata, uom_value_out); - ModuleBase::timer::end("Charge_Mixing", "mix_uom"); -#ifdef __MPI - // Synchronize mixed uom across all ranks to prevent divergence - // after multiple Pulay steps (same pattern as mix_dmr) - Parallel_Common::bcast_double(uom_in.data(), uom_in.size()); -#endif - return; -} diff --git a/source/source_estate/module_charge/charge_mixing.h b/source/source_estate/module_charge/charge_mixing.h deleted file mode 100644 index c24a866df91..00000000000 --- a/source/source_estate/module_charge/charge_mixing.h +++ /dev/null @@ -1,228 +0,0 @@ -#ifndef CHARGE_MIXING_H -#define CHARGE_MIXING_H -#include "charge.h" -#include "source_estate/module_dm/density_matrix.h" -#include "source_base/module_mixing/mixing.h" -#include "source_base/module_mixing/plain_mixing.h" - -class Charge_Mixing -{ - /// Charge_Mixing class - /// This class is used to mix charge density, kinetic energy density and real-space density matrix - /// This Charge_Mixing class offers the following interfaces: - /// 1. set_mixing() to set all private mixing parameters - /// 2. init_mixing() to initialize mixing, including allocating memory for mixing data and reset mixing - /// 3. mix_rho() to mix charge density - /// 4. mix_dmr() to mix real-space density matrix - /// how to use it: - /// you can (re)start a mixing by calling set_mixing() and init_mixing() before calling mix_rho() or mix_dmr() - - public: - Charge_Mixing(); - ~Charge_Mixing(); - - /** - * @brief Set all private mixing paramters - * @param mixing_mode_in mixing mode: "plain", "broyden", "pulay" - * @param mixing_beta_in mixing beta - * @param mixing_ndim_in mixing ndim - * @param mixing_gg0_in mixing gg0 for Kerker screen - * @param mixing_tau_in whether to use tau mixing - * @param mixing_beta_mag_in mixing beta for magnetism - * @param mixing_gg0_mag_in mixing gg0 for Kerker screen for magnetism - * @param mixing_gg0_min_in minimum kerker coefficient - * @param mixing_angle_in mixing angle for nspin=4 - * @param mixing_dmr_in whether to mixing real space density matrix - * @param omega_in omega for non-linear core correction - * @param tpiba_in 2*pi/beta for non-linear core correction - */ - void set_mixing(const std::string& mixing_mode_in, - const double& mixing_beta_in, - const int& mixing_ndim_in, - const double& mixing_gg0_in, - const bool& mixing_tau_in, - const double& mixing_beta_mag_in, - const double& mixing_gg0_mag_in, - const double& mixing_gg0_min_in, - const double& mixing_angle_in, - const bool& mixing_dmr_in, - double& omega_in, - double& tpiba_in); - - void close_kerker_gg0() { mixing_gg0 = 0.0; mixing_gg0_mag = 0.0; } - void conserve_setting() { mixing_beta = 0.01; mixing_beta_mag = 0.04; } - /** - * @brief initialize mixing, including constructing mixing and allocating memory for mixing data - * @brief this function should be called at eachiterinit() - */ - void init_mixing(); - - /** - * @brief allocate memory of dmr_mdata - * @param nnr size of real-space density matrix - */ - void allocate_mixing_dmr(const int nnr); - - /** - * @brief charge mixing - * @param chr pointer of Charge object - */ - void mix_rho(Charge* chr); - - /** - * @brief density matrix mixing, only for LCAO - * @param DM pointer of DensityMatrix object - */ - void mix_dmr(elecstate::DensityMatrix* DM); - void mix_dmr(elecstate::DensityMatrix, double>* DM); - - /** - * @brief allocate memory of uom_mdata - * @param uom_size size of DFT+U occupation matrix - */ - void allocate_mixing_uom(int size_uom); - - /** - * @brief DFT+U occupation matrix mixing - * @param uom_in output occupation matrix - * @param uom_save_in input occupation matrix - */ - void mix_uom(std::vector& uom_in, std::vector& uom_save_in); - - /** - * @brief Get the drho between rho and rho_save, similar for get_dkin - * - */ - double get_drho(Charge* chr, const double nelec); - double get_dkin(Charge* chr, const double nelec); - - /** - * @brief reset mixing, actually we only call init_mixing() to reset mixing instead of this function - */ - void mix_reset(); - - /** - * @brief Set the smooth and dense grids - * @param rhopw_in smooth grid - * @param rhodpw_in dense grid when double grid is used, otherwise same as rhopw - */ - void set_rhopw(ModulePW::PW_Basis* rhopw_in, ModulePW::PW_Basis* rhodpw_in); - - // extracting parameters normally these parameters will not be used outside charge mixing - // while Exx is using them as well as some other places - const std::string& get_mixing_mode() const {return mixing_mode;} - double get_mixing_beta() const {return mixing_beta;} - int get_mixing_ndim() const {return mixing_ndim;} - double get_mixing_gg0() const {return mixing_gg0;} - Base_Mixing::Mixing* get_mixing() const {return mixing;} - - // for mixing restart - int mixing_restart_step = 0; //which step to restart mixing during SCF, always equal to scf_namx except for the mixing restart - int mixing_restart_count = 0; // the number of restart mixing during SCF. Do not set mixing_restart_count as bool since I want to keep some flexibility in the future - int mixing_restart_last = 0; // the label of mixing restart step, store the step number of the last mixing restart - - // to calculate the slope of drho curve during SCF, which is used to determine if SCF oscillate - bool if_scf_oscillate(const int iteration, const double drho, const int iternum_used, const double threshold); - - private: - - // mixing_data - Base_Mixing::Mixing* mixing = nullptr; ///< Mixing object to mix charge density, kinetic energy density and compensation density - Base_Mixing::Mixing_Data rho_mdata; ///< Mixing data for charge density - Base_Mixing::Mixing_Data tau_mdata; ///< Mixing data for kinetic energy density - Base_Mixing::Mixing_Data nhat_mdata; ///< Mixing data for compensation density - Base_Mixing::Mixing_Data dmr_mdata; ///< Mixing data for real space density matrix - Base_Mixing::Mixing_Data uom_mdata; ///< Mixing data for DFT+U occupation matrix - Base_Mixing::Plain_Mixing* mixing_highf = nullptr; ///< The high_frequency part is mixed by plain mixing method. - - //====================================== - // private mixing parameters - //====================================== - std::string mixing_mode = "broyden"; ///< mixing mode: "plain", "broyden", "pulay" - double mixing_beta = 0.8; ///< mixing beta for density - double mixing_beta_mag = 1.6; ///< mixing beta for magnetism - int mixing_ndim = 8; ///< mixing ndim for broyden and pulay - double mixing_gg0 = 0.0; ///< mixing gg0 for Kerker screen - bool mixing_tau = false; ///< whether to use tau mixing - double mixing_gg0_mag = 0.0; ///< mixing gg0 for Kerker screen for magnetism - double mixing_gg0_min = 0.1; ///< minimum kerker coefficient - double mixing_angle = 0.0; ///< mixing angle for nspin=4 - bool mixing_dmr = false; ///< whether to mixing real space density matrix - double* omega = nullptr; ///< omega for non-linear core correction - double* tpiba = nullptr; ///< 2*pi/beta for non-linear core correction - double* tpiba2 = nullptr; ///< 2*pi/beta^2 for non-linear core correction - std::vector _drho_history; ///< history of drho used to determine the oscillation, size is scf_nmax - - bool new_e_iteration = true; - - ModulePW::PW_Basis* rhopw = nullptr; ///< smooth grid - ModulePW::PW_Basis* rhodpw = nullptr; ///< dense grid, same as rhopw for ncpp. - - /** - * @brief charge mixing for reciprocal space - * @param chr pointer of Charge object - */ - void mix_rho_recip(Charge* chr); - - /** - * @brief charge mixing for real space - * @param chr pointer of Charge object - */ - void mix_rho_real(Charge* chr); - - /** - * @brief Kerker screen method for reciprocal space - * @param rhog charge density in reciprocal space - */ - void Kerker_screen_recip(std::complex* rhog); - - /** - * @brief Kerker screen method for real space - * @param rho charge density in real space - */ - void Kerker_screen_real(double* rho); - - /** - * @brief Inner product of two complex vectors - * @brief inner_product_recip_rho is used for charge, like get_drho() - * @brief inner_product_recip_hartree and inner_product_recip_simple are used for charge mixing - * @brief inner_product_recip_simple is only used for test - * @brief Actually, I am not sure if the definition of inner product for NSPIN=4 is correct, need to be checked. - */ - double inner_product_recip_rho(std::complex* rho1, std::complex* rho2); - double inner_product_recip_simple(std::complex* rho1, std::complex* rho2); - double inner_product_recip_hartree(std::complex* rho1, std::complex* rho2); - - /** - * @brief Inner product of two double vectors - * - */ - double inner_product_real(double* rho1, double* rho2); - - /** - * @brief divide rho/tau to smooth and high frequency parts - * @param data_d dense data - * @param data_s smooth data - * @param data_hf high frequency data = dense data - smooth data - * - */ - void divide_data(std::complex* data_d, std::complex*& data_s, std::complex*& data_hf); - /** - * @brief gather smooth and high frequency parts to rho/tau - * @param data_d dense data - * @param data_s smooth data - * @param data_hf high frequency data = dense data - smooth data - * - */ - void combine_data(std::complex* data_d, std::complex*& data_s, std::complex*& data_hf); - /** - * @brief clean smooth and high frequency parts - * @param data_d dense data - * @param data_s smooth data - * @param data_hf high frequency data = dense data - smooth data - * - */ - void clean_data(std::complex*& data_s, std::complex*& data_hf); -}; - -#endif diff --git a/source/source_estate/module_charge/charge_mixing_dmr.cpp b/source/source_estate/module_charge/charge_mixing_dmr.cpp deleted file mode 100644 index 368ed63f257..00000000000 --- a/source/source_estate/module_charge/charge_mixing_dmr.cpp +++ /dev/null @@ -1,227 +0,0 @@ -#include "charge_mixing.h" - -#include "source_io/module_parameter/parameter.h" -#include "source_base/timer.h" - -void Charge_Mixing::allocate_mixing_dmr(const int nnr) -{ - // Note that: we cannot allocate memory for dmr_mdata in set_mixing. - // since the size of dmr_mdata is given by the size of HContainer.nnr, which is calculated in DensityMatrix::init_DMR(). - // and DensityMatrix::init_DMR() is called in beforescf(). While set_mixing() is called in ESolver_KS::Init(). - ModuleBase::TITLE("Charge_Mixing", "allocate_mixing_dmr"); - ModuleBase::timer::start("Charge_Mixing", "allocate_mixing_dmr"); - // - const int dmr_nspin = (PARAM.inp.nspin == 2) ? 2 : 1; - // allocate memory for dmr_mdata - if (PARAM.inp.scf_thr_type == 1) - { - ModuleBase::WARNING_QUIT("Charge_Mixing", "This Mixing of Density Matrix is not supported for PW basis yet"); - } - else if (PARAM.inp.scf_thr_type == 2) - { - this->mixing->init_mixing_data(this->dmr_mdata, nnr * dmr_nspin, sizeof(double)); - } - - this->dmr_mdata.reset(); - ModuleBase::timer::end("Charge_Mixing", "allocate_mixing_dmr"); - - return; -} - -void Charge_Mixing::mix_dmr(elecstate::DensityMatrix* DM) -{ - // Notice that DensityMatrix object is a Template class - ModuleBase::TITLE("Charge_Mixing", "mix_dmr"); - ModuleBase::timer::start("Charge_Mixing", "mix_dmr"); - // - std::vector*> dmr = DM->get_DMR_vector(); - std::vector>& dmr_save = DM->get_DMR_save(); - // - //const int dmr_nspin = (PARAM.inp.nspin == 2) ? 2 : 1; - double* dmr_in = nullptr; - double* dmr_out = nullptr; - if (PARAM.inp.nspin == 1 || PARAM.inp.nspin == 4) - { - dmr_in = dmr_save[0].data(); - dmr_out = dmr[0]->get_wrapper(); - this->mixing->push_data(this->dmr_mdata, dmr_in, dmr_out, nullptr, false); - this->mixing->mix_data(this->dmr_mdata, dmr_out); - } - else if (PARAM.inp.nspin == 2) - { - // magnetic density matrix - double* dmr_mag = nullptr; - double* dmr_mag_save = nullptr; - const int nnr = dmr[0]->get_nnr(); - // allocate dmr_mag[is*nnrx] and dmr_mag_save[is*nnrx] - dmr_mag = new double[nnr * PARAM.inp.nspin]; - dmr_mag_save = new double[nnr * PARAM.inp.nspin]; - ModuleBase::GlobalFunc::ZEROS(dmr_mag, nnr * PARAM.inp.nspin); - ModuleBase::GlobalFunc::ZEROS(dmr_mag_save, nnr * PARAM.inp.nspin); - double* dmr_up = nullptr; - double* dmr_down = nullptr; - // tranfer dmr into dmr_mag - dmr_up = dmr[0]->get_wrapper(); - dmr_down = dmr[1]->get_wrapper(); - for (int ir = 0; ir < nnr; ir++) - { - dmr_mag[ir] = dmr_up[ir] + dmr_down[ir]; - dmr_mag[ir + nnr] = dmr_up[ir] - dmr_down[ir]; - } - // tranfer dmr_save into dmr_mag_save - dmr_up = dmr_save[0].data(); - dmr_down = dmr_save[1].data(); - for (int ir = 0; ir < nnr; ir++) - { - dmr_mag_save[ir] = dmr_up[ir] + dmr_down[ir]; - dmr_mag_save[ir + nnr] = dmr_up[ir] - dmr_down[ir]; - } - // - dmr_in = dmr_mag_save; - dmr_out = dmr_mag; - // no kerker in mixing_dmr - //auto screen = std::bind(&Charge_Mixing::Kerker_screen_real, this, std::placeholders::_1); - auto twobeta_mix - = [this, nnr](double* out, const double* in, const double* sres) { -#ifdef _OPENMP -#pragma omp parallel for schedule(static, 256) -#endif - for (int i = 0; i < nnr; ++i) - { - out[i] = in[i] + this->mixing_beta * sres[i]; - } - // magnetism -#ifdef _OPENMP -#pragma omp parallel for schedule(static, 256) -#endif - for (int i = nnr; i < 2 * nnr; ++i) - { - out[i] = in[i] + this->mixing_beta_mag * sres[i]; - } - }; - this->mixing->push_data(this->dmr_mdata, dmr_in, dmr_out, nullptr, twobeta_mix, false); - //auto inner_product - // = std::bind(&Charge_Mixing::inner_product_real, this, std::placeholders::_1, std::placeholders::_2); - //this->mixing->cal_coef(this->rho_mdata, inner_product); - this->mixing->mix_data(this->dmr_mdata, dmr_out); - // get new dmr from dmr_mag - dmr_up = dmr[0]->get_wrapper(); - dmr_down = dmr[1]->get_wrapper(); - for (int is = 0; is < PARAM.inp.nspin; is++) - { - ModuleBase::GlobalFunc::ZEROS(dmr_up, nnr); - ModuleBase::GlobalFunc::ZEROS(dmr_down, nnr); - } - for (int ir = 0; ir < nnr; ir++) - { - dmr_up[ir] = 0.5 * (dmr_mag[ir] + dmr_mag[ir+nnr]); - dmr_down[ir] = 0.5 * (dmr_mag[ir] - dmr_mag[ir+nnr]); - } - // delete - delete[] dmr_mag; - delete[] dmr_mag_save; - } - - ModuleBase::timer::end("Charge_Mixing", "mix_dmr"); - - return; -} - -void Charge_Mixing::mix_dmr(elecstate::DensityMatrix, double>* DM) -{ - // Notice that DensityMatrix object is a Template class - ModuleBase::TITLE("Charge_Mixing", "mix_dmr"); - ModuleBase::timer::start("Charge_Mixing", "mix_dmr"); - // - std::vector*> dmr = DM->get_DMR_vector(); - std::vector>& dmr_save = DM->get_DMR_save(); - // - //const int dmr_nspin = (PARAM.inp.nspin == 2) ? 2 : 1; - double* dmr_in = nullptr; - double* dmr_out = nullptr; - if (PARAM.inp.nspin == 1 || PARAM.inp.nspin == 4) - { - dmr_in = dmr_save[0].data(); - dmr_out = dmr[0]->get_wrapper(); - this->mixing->push_data(this->dmr_mdata, dmr_in, dmr_out, nullptr, false); - this->mixing->mix_data(this->dmr_mdata, dmr_out); - } - else if (PARAM.inp.nspin == 2) - { - // magnetic density matrix - double* dmr_mag = nullptr; - double* dmr_mag_save = nullptr; - const int nnr = dmr[0]->get_nnr(); - // allocate dmr_mag[is*nnrx] and dmr_mag_save[is*nnrx] - dmr_mag = new double[nnr * PARAM.inp.nspin]; - dmr_mag_save = new double[nnr * PARAM.inp.nspin]; - ModuleBase::GlobalFunc::ZEROS(dmr_mag, nnr * PARAM.inp.nspin); - ModuleBase::GlobalFunc::ZEROS(dmr_mag_save, nnr * PARAM.inp.nspin); - double* dmr_up = nullptr; - double* dmr_down = nullptr; - // tranfer dmr into dmr_mag - dmr_up = dmr[0]->get_wrapper(); - dmr_down = dmr[1]->get_wrapper(); - for (int ir = 0; ir < nnr; ir++) - { - dmr_mag[ir] = dmr_up[ir] + dmr_down[ir]; - dmr_mag[ir + nnr] = dmr_up[ir] - dmr_down[ir]; - } - // tranfer dmr_save into dmr_mag_save - dmr_up = dmr_save[0].data(); - dmr_down = dmr_save[1].data(); - for (int ir = 0; ir < nnr; ir++) - { - dmr_mag_save[ir] = dmr_up[ir] + dmr_down[ir]; - dmr_mag_save[ir + nnr] = dmr_up[ir] - dmr_down[ir]; - } - // - dmr_in = dmr_mag_save; - dmr_out = dmr_mag; - // no kerker in mixing_dmr - //auto screen = std::bind(&Charge_Mixing::Kerker_screen_real, this, std::placeholders::_1); - auto twobeta_mix - = [this, nnr](double* out, const double* in, const double* sres) { -#ifdef _OPENMP -#pragma omp parallel for schedule(static, 256) -#endif - for (int i = 0; i < nnr; ++i) - { - out[i] = in[i] + this->mixing_beta * sres[i]; - } - // magnetism -#ifdef _OPENMP -#pragma omp parallel for schedule(static, 256) -#endif - for (int i = nnr; i < 2 * nnr; ++i) - { - out[i] = in[i] + this->mixing_beta_mag * sres[i]; - } - }; - this->mixing->push_data(this->dmr_mdata, dmr_in, dmr_out, nullptr, twobeta_mix, false); - //auto inner_product - // = std::bind(&Charge_Mixing::inner_product_real, this, std::placeholders::_1, std::placeholders::_2); - //this->mixing->cal_coef(this->rho_mdata, inner_product); - this->mixing->mix_data(this->dmr_mdata, dmr_out); - // get new dmr from dmr_mag - dmr_up = dmr[0]->get_wrapper(); - dmr_down = dmr[1]->get_wrapper(); - for (int is = 0; is < PARAM.inp.nspin; is++) - { - ModuleBase::GlobalFunc::ZEROS(dmr_up, nnr); - ModuleBase::GlobalFunc::ZEROS(dmr_down, nnr); - } - for (int ir = 0; ir < nnr; ir++) - { - dmr_up[ir] = 0.5 * (dmr_mag[ir] + dmr_mag[ir+nnr]); - dmr_down[ir] = 0.5 * (dmr_mag[ir] - dmr_mag[ir+nnr]); - } - // delete - delete[] dmr_mag; - delete[] dmr_mag_save; - } - - ModuleBase::timer::end("Charge_Mixing", "mix_dmr"); - - return; -} \ No newline at end of file diff --git a/source/source_estate/module_charge/charge_mixing_rho.cpp b/source/source_estate/module_charge/charge_mixing_rho.cpp deleted file mode 100644 index 3d8f302b572..00000000000 --- a/source/source_estate/module_charge/charge_mixing_rho.cpp +++ /dev/null @@ -1,613 +0,0 @@ -#include "charge_mixing.h" -#include "source_io/module_parameter/parameter.h" -#include "source_base/timer.h" -#include "source_hamilt/module_xc/xc_functional.h" - -void Charge_Mixing::mix_rho_recip(Charge* chr) -{ - ModuleBase::TITLE("Charge_Mixing", "mix_rho_recip"); - ModuleBase::timer::start("Charge_Mixing", "mix_rho_recip"); - - const int nspin = PARAM.inp.nspin; - assert(nspin==1 || nspin==2 || nspin==4); - - std::complex* rhog_in = nullptr; - std::complex* rhog_out = nullptr; - // for smooth part - std::complex* rhogs_in = chr->rhog_save[0]; - std::complex* rhogs_out = chr->rhog[0]; - // for high_frequency part - std::complex* rhoghf_in = nullptr; - std::complex* rhoghf_out = nullptr; - - if ( PARAM.globalv.double_grid) - { - // divide into smooth part and high_frequency part - divide_data(chr->rhog_save[0], rhogs_in, rhoghf_in); - divide_data(chr->rhog[0], rhogs_out, rhoghf_out); - } - - // inner_product_recip_hartree is a hartree-like sum, unit is Ry - auto inner_product - = std::bind(&Charge_Mixing::inner_product_recip_hartree, this, std::placeholders::_1, std::placeholders::_2); - - // DIIS Mixing Only for smooth part, while high_frequency part is mixed by plain mixing method. - if (nspin == 1) - { - rhog_in = rhogs_in; - rhog_out = rhogs_out; - auto screen = std::bind(&Charge_Mixing::Kerker_screen_recip, this, std::placeholders::_1); - this->mixing->push_data(this->rho_mdata, rhog_in, rhog_out, screen, true); - this->mixing->cal_coef(this->rho_mdata, inner_product); - this->mixing->mix_data(this->rho_mdata, rhog_out); - } - else if (nspin == 2) - { - // magnetic density - std::complex *rhog_mag = nullptr; - std::complex *rhog_mag_save = nullptr; - const int npw = this->rhopw->npw; - // allocate rhog_mag[is*ngmc] and rhog_mag_save[is*ngmc] - rhog_mag = new std::complex[npw * nspin]; - rhog_mag_save = new std::complex[npw * nspin]; - ModuleBase::GlobalFunc::ZEROS(rhog_mag, npw * nspin); - ModuleBase::GlobalFunc::ZEROS(rhog_mag_save, npw * nspin); - // get rhog_mag[is*ngmc] and rhog_mag_save[is*ngmc] - for (int ig = 0; ig < npw; ig++) - { - rhog_mag[ig] = chr->rhog[0][ig] + chr->rhog[1][ig]; - rhog_mag_save[ig] = chr->rhog_save[0][ig] + chr->rhog_save[1][ig]; - } - for (int ig = 0; ig < npw; ig++) - { - rhog_mag[ig + npw] = chr->rhog[0][ig] - chr->rhog[1][ig]; - rhog_mag_save[ig + npw] = chr->rhog_save[0][ig] - chr->rhog_save[1][ig]; - } - // - rhog_in = rhog_mag_save; - rhog_out = rhog_mag; - // - auto screen = std::bind(&Charge_Mixing::Kerker_screen_recip, this, std::placeholders::_1); - auto twobeta_mix - = [this, npw](std::complex* out, const std::complex* in, const std::complex* sres) { -#ifdef _OPENMP -#pragma omp parallel for schedule(static, 256) -#endif - for (int i = 0; i < npw; ++i) - { - out[i] = in[i] + this->mixing_beta * sres[i]; - } - // magnetism -#ifdef _OPENMP -#pragma omp parallel for schedule(static, 256) -#endif - for (int i = npw; i < 2 * npw; ++i) - { - out[i] = in[i] + this->mixing_beta_mag * sres[i]; - } - }; - this->mixing->push_data(this->rho_mdata, rhog_in, rhog_out, screen, twobeta_mix, true); - this->mixing->cal_coef(this->rho_mdata, inner_product); - this->mixing->mix_data(this->rho_mdata, rhog_out); - // get rhog[is][ngmc] from rhog_mag[is*ngmc] - for (int is = 0; is < nspin; is++) - { - ModuleBase::GlobalFunc::ZEROS(chr->rhog[is], npw); - } - for (int ig = 0; ig < npw; ig++) - { - chr->rhog[0][ig] = 0.5 * (rhog_mag[ig] + rhog_mag[ig+npw]); - chr->rhog[1][ig] = 0.5 * (rhog_mag[ig] - rhog_mag[ig+npw]); - } - // delete - delete[] rhog_mag; - delete[] rhog_mag_save; - // get rhogs_out for combine_data() - if ( PARAM.globalv.double_grid) - { - for (int ig = 0; ig < npw; ig++) - { - rhogs_out[ig] = chr->rhog[0][ig]; - rhogs_out[ig + npw] = chr->rhog[1][ig]; - } - } - } - else if (nspin == 4 && PARAM.inp.mixing_angle <= 0) - { - // normal broyden mixing for {rho, mx, my, mz} - rhog_in = rhogs_in; - rhog_out = rhogs_out; - const int npw = this->rhopw->npw; - auto screen = std::bind(&Charge_Mixing::Kerker_screen_recip, this, std::placeholders::_1); // use old one - auto twobeta_mix - = [this, npw](std::complex* out, const std::complex* in, const std::complex* sres) { -#ifdef _OPENMP -#pragma omp parallel for schedule(static, 256) -#endif - for (int i = 0; i < npw; ++i) - { - out[i] = in[i] + this->mixing_beta * sres[i]; - } - // magnetism, mx, my, mz -#ifdef _OPENMP -#pragma omp parallel for schedule(static, 256) -#endif - for (int i = npw; i < 4 * npw; ++i) - { - out[i] = in[i] + this->mixing_beta_mag * sres[i]; - } - }; - this->mixing->push_data(this->rho_mdata, rhog_in, rhog_out, screen, twobeta_mix, true); - this->mixing->cal_coef(this->rho_mdata, inner_product); - this->mixing->mix_data(this->rho_mdata, rhog_out); - } - else if (nspin == 4 && PARAM.inp.mixing_angle > 0) - { - // special broyden mixing for {rho, |m|} proposed by J. Phys. Soc. Jpn. 82 (2013) 114706 - // here only consider the case of mixing_angle = 1, which mean only change |m| and keep angle fixed - // old support see mix_rho_recip() - if ( PARAM.globalv.double_grid) - { - ModuleBase::WARNING_QUIT("Charge_Mixing", "double_grid is not supported for new mixing method yet."); - } - // allocate memory for rho_magabs and rho_magabs_save - const int nrxx = this->rhopw->nrxx; - double* rho_magabs = new double[nrxx]; - double* rho_magabs_save = new double[nrxx]; - ModuleBase::GlobalFunc::ZEROS(rho_magabs, nrxx); - ModuleBase::GlobalFunc::ZEROS(rho_magabs_save, nrxx); - // calculate rho_magabs and rho_magabs_save - for (int ir = 0; ir < nrxx; ir++) - { - // |m| for rho - rho_magabs[ir] = std::sqrt(chr->rho[1][ir] * chr->rho[1][ir] - + chr->rho[2][ir] * chr->rho[2][ir] - + chr->rho[3][ir] * chr->rho[3][ir]); - // |m| for rho_save - rho_magabs_save[ir] = std::sqrt(chr->rho_save[1][ir] * chr->rho_save[1][ir] - + chr->rho_save[2][ir] * chr->rho_save[2][ir] - + chr->rho_save[3][ir] * chr->rho_save[3][ir]); - } - // allocate memory for rhog_magabs and rhog_magabs_save - const int npw = this->rhopw->npw; - std::complex* rhog_magabs = new std::complex[npw * 2]; - std::complex* rhog_magabs_save = new std::complex[npw * 2]; - ModuleBase::GlobalFunc::ZEROS(rhog_magabs, npw * 2); - ModuleBase::GlobalFunc::ZEROS(rhog_magabs_save, npw * 2); - // calculate rhog_magabs and rhog_magabs_save - for (int ig = 0; ig < npw; ig++) - { - rhog_magabs[ig] = chr->rhog[0][ig]; // rho - rhog_magabs_save[ig] = chr->rhog_save[0][ig]; // rho_save - } - // FT to get rhog_magabs and rhog_magabs_save - this->rhopw->real2recip(rho_magabs, rhog_magabs + this->rhopw->npw); - this->rhopw->real2recip(rho_magabs_save, rhog_magabs_save + this->rhopw->npw); - // - rhog_in = rhog_magabs_save; - rhog_out = rhog_magabs; - auto screen = std::bind(&Charge_Mixing::Kerker_screen_recip, this, std::placeholders::_1); // use old one - auto twobeta_mix - = [this, npw](std::complex* out, const std::complex* in, const std::complex* sres) { -#ifdef _OPENMP -#pragma omp parallel for schedule(static, 256) -#endif - for (int i = 0; i < npw; ++i) - { - out[i] = in[i] + this->mixing_beta * sres[i]; - } - // magnetism, |m| -#ifdef _OPENMP -#pragma omp parallel for schedule(static, 256) -#endif - for (int i = npw; i < 2 * npw; ++i) - { - out[i] = in[i] + this->mixing_beta_mag * sres[i]; - } - }; - this->mixing->push_data(this->rho_mdata, rhog_in, rhog_out, screen, twobeta_mix, true); - this->mixing->cal_coef(this->rho_mdata, inner_product); - this->mixing->mix_data(this->rho_mdata, rhog_out); - // get new |m| in real space using FT - this->rhopw->recip2real(rhog_magabs + this->rhopw->npw, rho_magabs); - // use new |m| and angle to update {mx, my, mz} - for (int ig = 0; ig < npw; ig++) - { - chr->rhog[0][ig] = rhog_magabs[ig]; // rhog - double norm = std::sqrt(chr->rho[1][ig] * chr->rho[1][ig] - + chr->rho[2][ig] * chr->rho[2][ig] - + chr->rho[3][ig] * chr->rho[3][ig]); - if (std::abs(norm) < 1e-10) - { - continue; - } - double rescale_tmp = rho_magabs[npw + ig] / norm; - chr->rho[1][ig] *= rescale_tmp; - chr->rho[2][ig] *= rescale_tmp; - chr->rho[3][ig] *= rescale_tmp; - } - // delete - delete[] rhog_magabs; - delete[] rhog_magabs_save; - delete[] rho_magabs; - delete[] rho_magabs_save; - } - - if ( PARAM.globalv.double_grid) - { - // plain mixing for high_frequencies - const int ndimhf = (this->rhodpw->npw - this->rhopw->npw) * nspin; - this->mixing_highf->plain_mix(rhoghf_out, rhoghf_in, rhoghf_out, ndimhf, nullptr); - - // combine smooth part and high_frequency part - combine_data(chr->rhog[0], rhogs_out, rhoghf_out); - clean_data(rhogs_in, rhoghf_in); - } - - // rhog to rho - if (nspin == 4 && PARAM.inp.mixing_angle > 0) - { - // only tranfer rhog[0] - // do not support double_grid, use rhopw directly - chr->rhopw->recip2real(chr->rhog[0], chr->rho[0]); - } - else - { - for (int is = 0; is < nspin; is++) - { - // use rhodpw for double_grid - // rhodpw is the same as rhopw for ! PARAM.globalv.double_grid - this->rhodpw->recip_to_real,double,base_device::DEVICE_CPU>(chr->rhog[is], chr->rho[is]); - } - } - // For kinetic energy density - if ((XC_Functional::get_ked_flag()) && mixing_tau) - { - std::vector> kin_g(nspin * rhodpw->npw); - std::vector> kin_g_save(nspin * rhodpw->npw); - // FFT to get kin_g and kin_g_save - for (int is = 0; is < nspin; ++is) - { - rhodpw->real2recip(chr->kin_r[is], &kin_g[is * rhodpw->npw]); - rhodpw->real2recip(chr->kin_r_save[is], &kin_g_save[is * rhodpw->npw]); - } - // for smooth part, for ! PARAM.globalv.double_grid only have this part - std::complex*taugs_in = kin_g_save.data(), *taugs_out = kin_g.data(); - // for high frequency part - std::complex*taughf_in = nullptr, *taughf_out = nullptr; - if ( PARAM.globalv.double_grid) - { - // divide into smooth part and high_frequency part - divide_data(kin_g_save.data(), taugs_in, taughf_in); - divide_data(kin_g.data(), taugs_out, taughf_out); - } - - // Note: there is no kerker modification for tau because I'm not sure - // if we should have it. If necessary we can try it in the future. - this->mixing->push_data(this->tau_mdata, taugs_in, taugs_out, nullptr, false); - - this->mixing->mix_data(this->tau_mdata, taugs_out); - - if ( PARAM.globalv.double_grid) - { - // simple mixing for high_frequencies - const int ndimhf = (this->rhodpw->npw - this->rhopw->npw) * nspin; - this->mixing_highf->plain_mix(taughf_out, taughf_in, taughf_out, ndimhf, nullptr); - - // combine smooth part and high_frequency part - combine_data(kin_g.data(), taugs_out, taughf_out); - clean_data(taugs_in, taughf_in); - } - - // kin_g to kin_r - for (int is = 0; is < nspin; is++) - { - rhodpw->recip2real(&kin_g[is * rhodpw->npw], chr->kin_r[is]); - } - } - - ModuleBase::timer::end("Charge_Mixing", "mix_rho_recip"); - return; -} - -void Charge_Mixing::mix_rho_real(Charge* chr) -{ - ModuleBase::TITLE("Charge_Mixing", "mix_rho_real"); - ModuleBase::timer::start("Charge_Mixing", "mix_rho_real"); - - const int nspin = PARAM.inp.nspin; - assert(nspin==1 || nspin==2 || nspin==4); - - double* rhor_in=nullptr; - double* rhor_out=nullptr; - - if (nspin == 1) - { - rhor_in = chr->rho_save[0]; - rhor_out = chr->rho[0]; - auto screen = std::bind(&Charge_Mixing::Kerker_screen_real, this, std::placeholders::_1); - this->mixing->push_data(this->rho_mdata, rhor_in, rhor_out, screen, true); - auto inner_product - = std::bind(&Charge_Mixing::inner_product_real, this, std::placeholders::_1, std::placeholders::_2); - this->mixing->cal_coef(this->rho_mdata, inner_product); - this->mixing->mix_data(this->rho_mdata, rhor_out); - } - else if (nspin == 2) - { - // magnetic density - double *rho_mag = nullptr; - double *rho_mag_save = nullptr; - const int nrxx = this->rhopw->nrxx; - // allocate rho_mag[is*nnrx] and rho_mag_save[is*nnrx] - rho_mag = new double[nrxx * nspin]; - rho_mag_save = new double[nrxx * nspin]; - ModuleBase::GlobalFunc::ZEROS(rho_mag, nrxx * nspin); - ModuleBase::GlobalFunc::ZEROS(rho_mag_save, nrxx * nspin); - // get rho_mag[is*nnrx] and rho_mag_save[is*nnrx] - for (int ir = 0; ir < nrxx; ir++) - { - rho_mag[ir] = chr->rho[0][ir] + chr->rho[1][ir]; - rho_mag_save[ir] = chr->rho_save[0][ir] + chr->rho_save[1][ir]; - } - for (int ir = 0; ir < nrxx; ir++) - { - rho_mag[ir + nrxx] = chr->rho[0][ir] - chr->rho[1][ir]; - rho_mag_save[ir + nrxx] = chr->rho_save[0][ir] - chr->rho_save[1][ir]; - } - // - rhor_in = rho_mag_save; - rhor_out = rho_mag; - auto screen = std::bind(&Charge_Mixing::Kerker_screen_real, this, std::placeholders::_1); - auto twobeta_mix - = [this, nrxx](double* out, const double* in, const double* sres) { -#ifdef _OPENMP -#pragma omp parallel for schedule(static, 256) -#endif - for (int i = 0; i < nrxx; ++i) - { - out[i] = in[i] + this->mixing_beta * sres[i]; - } - // magnetism -#ifdef _OPENMP -#pragma omp parallel for schedule(static, 256) -#endif - for (int i = nrxx; i < 2 * nrxx; ++i) - { - out[i] = in[i] + this->mixing_beta_mag * sres[i]; - } - }; - this->mixing->push_data(this->rho_mdata, rhor_in, rhor_out, screen, twobeta_mix, true); - auto inner_product - = std::bind(&Charge_Mixing::inner_product_real, this, std::placeholders::_1, std::placeholders::_2); - this->mixing->cal_coef(this->rho_mdata, inner_product); - this->mixing->mix_data(this->rho_mdata, rhor_out); - // get new rho[is][nrxx] from rho_mag[is*nrxx] - for (int is = 0; is < nspin; is++) - { - ModuleBase::GlobalFunc::ZEROS(chr->rho[is], nrxx); - //ModuleBase::GlobalFunc::ZEROS(rho_save[is], nrxx); - } - for (int ir = 0; ir < nrxx; ir++) - { - chr->rho[0][ir] = 0.5 * (rho_mag[ir] + rho_mag[ir+nrxx]); - chr->rho[1][ir] = 0.5 * (rho_mag[ir] - rho_mag[ir+nrxx]); - } - // delete - delete[] rho_mag; - delete[] rho_mag_save; - } - else if (nspin == 4 && PARAM.inp.mixing_angle <= 0) - { - // normal broyden mixing for {rho, mx, my, mz} - rhor_in = chr->rho_save[0]; - rhor_out = chr->rho[0]; - const int nrxx = this->rhopw->nrxx; - auto screen = std::bind(&Charge_Mixing::Kerker_screen_real, this, std::placeholders::_1); - auto twobeta_mix - = [this, nrxx](double* out, const double* in, const double* sres) { -#ifdef _OPENMP -#pragma omp parallel for schedule(static, 256) -#endif - for (int i = 0; i < nrxx; ++i) - { - out[i] = in[i] + this->mixing_beta * sres[i]; - } - // magnetism, mx, my, mz -#ifdef _OPENMP -#pragma omp parallel for schedule(static, 256) -#endif - for (int i = nrxx; i < 4 * nrxx; ++i) - { - out[i] = in[i] + this->mixing_beta_mag * sres[i]; - } - }; - this->mixing->push_data(this->rho_mdata, rhor_in, rhor_out, screen, twobeta_mix, true); - auto inner_product - = std::bind(&Charge_Mixing::inner_product_real, this, std::placeholders::_1, std::placeholders::_2); - this->mixing->cal_coef(this->rho_mdata, inner_product); - this->mixing->mix_data(this->rho_mdata, rhor_out); - } - else if (nspin == 4 && PARAM.inp.mixing_angle > 0) - { - // special broyden mixing for {rho, |m|} proposed by J. Phys. Soc. Jpn. 82 (2013) 114706 - // here only consider the case of mixing_angle = 1, which mean only change |m| and keep angle fixed - const int nrxx = this->rhopw->nrxx; - // allocate memory for rho_magabs and rho_magabs_save - double* rho_magabs = new double[nrxx * 2]; - double* rho_magabs_save = new double[nrxx * 2]; - ModuleBase::GlobalFunc::ZEROS(rho_magabs, nrxx * 2); - ModuleBase::GlobalFunc::ZEROS(rho_magabs_save, nrxx * 2); - // calculate rho_magabs and rho_magabs_save - for (int ir = 0; ir < nrxx; ir++) - { - rho_magabs[ir] = chr->rho[0][ir]; // rho - rho_magabs_save[ir] = chr->rho_save[0][ir]; // rho_save - // |m| for rho - rho_magabs[nrxx + ir] = std::sqrt(chr->rho[1][ir] * chr->rho[1][ir] - + chr->rho[2][ir] * chr->rho[2][ir] - + chr->rho[3][ir] * chr->rho[3][ir]); - // |m| for rho_save - rho_magabs_save[nrxx + ir] = std::sqrt(chr->rho_save[1][ir] * chr->rho_save[1][ir] - + chr->rho_save[2][ir] * chr->rho_save[2][ir] - + chr->rho_save[3][ir] * chr->rho_save[3][ir]); - } - rhor_in = rho_magabs_save; - rhor_out = rho_magabs; - - auto screen = std::bind(&Charge_Mixing::Kerker_screen_real, this, std::placeholders::_1); - auto twobeta_mix - = [this, nrxx](double* out, const double* in, const double* sres) { -#ifdef _OPENMP -#pragma omp parallel for schedule(static, 256) -#endif - for (int i = 0; i < nrxx; ++i) - { - out[i] = in[i] + this->mixing_beta * sres[i]; - } - // magnetism, |m| -#ifdef _OPENMP -#pragma omp parallel for schedule(static, 256) -#endif - for (int i = nrxx; i < 2 * nrxx; ++i) - { - out[i] = in[i] + this->mixing_beta_mag * sres[i]; - } - }; - this->mixing->push_data(this->rho_mdata, rhor_in, rhor_out, screen, twobeta_mix, true); - auto inner_product - = std::bind(&Charge_Mixing::inner_product_real, this, std::placeholders::_1, std::placeholders::_2); - this->mixing->cal_coef(this->rho_mdata, inner_product); - this->mixing->mix_data(this->rho_mdata, rhor_out); - - // use new |m| and angle to update {mx, my, mz} - for (int ir = 0; ir < nrxx; ir++) - { - chr->rho[0][ir] = rho_magabs[ir]; // rho - double norm = std::sqrt(chr->rho[1][ir] * chr->rho[1][ir] - + chr->rho[2][ir] * chr->rho[2][ir] - + chr->rho[3][ir] * chr->rho[3][ir]); - - if (norm < 1e-10) - { - continue; - } - double rescale_tmp = rho_magabs[nrxx + ir] / norm; - chr->rho[1][ir] *= rescale_tmp; - chr->rho[2][ir] *= rescale_tmp; - chr->rho[3][ir] *= rescale_tmp; - } - // delete - delete[] rho_magabs; - delete[] rho_magabs_save; - } - - double *taur_out=nullptr; - double *taur_in=nullptr; - if ((XC_Functional::get_ked_flag()) && mixing_tau) - { - taur_in = chr->kin_r_save[0]; - taur_out = chr->kin_r[0]; - // Note: there is no kerker modification for tau because I'm not sure - // if we should have it. If necessary we can try it in the future. - this->mixing->push_data(this->tau_mdata, taur_in, taur_out, nullptr, false); - - this->mixing->mix_data(this->tau_mdata, taur_out); - } - - ModuleBase::timer::end("Charge_Mixing", "mix_rho_real"); - return; -} - - -void Charge_Mixing::mix_rho(Charge* chr) -{ - ModuleBase::TITLE("Charge_Mixing", "mix_rho"); - ModuleBase::timer::start("Charge_Mixing", "mix_rho"); - - const int nspin = PARAM.inp.nspin; - assert(nspin==1 || nspin==2 || nspin==4); - - // the charge before mixing. - const int nrxx = chr->rhopw->nrxx; - std::vector rho123(nspin * nrxx); - for (int is = 0; is < nspin; ++is) - { - if (is == 0 || is == 3 || !PARAM.globalv.domag_z) - { - double* rho123_is = rho123.data() + is * nrxx; -#ifdef _OPENMP -#pragma omp parallel for schedule(static, 512) -#endif - for(int ir = 0 ; ir < nrxx ; ++ir) - { - rho123_is[ir] = chr->rho[is][ir]; - } - } - } - std::vector kin_r123; - if ((XC_Functional::get_ked_flag()) && mixing_tau) - { - kin_r123.resize(nspin * nrxx); - for (int is = 0; is < nspin; ++is) - { - double* kin_r123_is = kin_r123.data() + is * nrxx; -#ifdef _OPENMP -#pragma omp parallel for schedule(static, 512) -#endif - for(int ir = 0 ; ir < nrxx ; ++ir) - { - kin_r123_is[ir] = chr->kin_r[is][ir]; - } - } - } - // --------------------Mixing Body-------------------- - if (PARAM.inp.scf_thr_type == 1) - { - mix_rho_recip(chr); - } - else if (PARAM.inp.scf_thr_type == 2) - { - mix_rho_real(chr); - } - // --------------------------------------------------- - - // mohan add 2012-06-05 - // rho_save is the charge before mixing - for (int is = 0; is < nspin; ++is) - { - if (is == 0 || is == 3 || !PARAM.globalv.domag_z) - { - double* rho123_is = rho123.data() + is * nrxx; -#ifdef _OPENMP -#pragma omp parallel for schedule(static, 512) -#endif - for(int ir = 0 ; ir < nrxx ; ++ir) - { - chr->rho_save[is][ir] = rho123_is[ir]; - } - } - } - - if ((XC_Functional::get_ked_flag()) && mixing_tau) - { - for (int is = 0; is < nspin; ++is) - { - double* kin_r123_is = kin_r123.data() + is * nrxx; -#ifdef _OPENMP -#pragma omp parallel for schedule(static, 512) -#endif - for(int ir = 0 ; ir < nrxx ; ++ir) - { - chr->kin_r_save[is][ir] = kin_r123_is[ir]; - } - } - } - - if (new_e_iteration) - { - new_e_iteration = false; - } - - ModuleBase::timer::end("Charge_Mixing", "mix_rho"); - return; -} diff --git a/source/source_estate/module_charge/charge_mixing_uspp.cpp b/source/source_estate/module_charge/charge_mixing_uspp.cpp deleted file mode 100644 index 10968cd16e6..00000000000 --- a/source/source_estate/module_charge/charge_mixing_uspp.cpp +++ /dev/null @@ -1,76 +0,0 @@ -#include "charge_mixing.h" -#include "source_io/module_parameter/parameter.h" - -void Charge_Mixing::divide_data(std::complex* data_d, - std::complex*& data_s, - std::complex*& data_hf) -{ - ModuleBase::TITLE("Charge_Mixing", "divide_data"); - if (PARAM.inp.nspin == 1) - { - data_s = data_d; - data_hf = data_d + this->rhopw->npw; - } - else - { - const int ndimd = this->rhodpw->npw; - const int ndims = this->rhopw->npw; - const int ndimhf = ndimd - ndims; - data_s = new std::complex[PARAM.inp.nspin * ndims]; - data_hf = nullptr; - if (ndimhf > 0) - { - data_hf = new std::complex[PARAM.inp.nspin * ndimhf]; - } - for (int is = 0; is < PARAM.inp.nspin; ++is) - { - std::memcpy(data_s + is * ndims, data_d + is * ndimd, ndims * sizeof(std::complex)); - std::memcpy(data_hf + is * ndimhf, data_d + is * ndimd + ndims, ndimhf * sizeof(std::complex)); - } - } -} -void Charge_Mixing::combine_data(std::complex* data_d, - std::complex*& data_s, - std::complex*& data_hf) -{ - ModuleBase::TITLE("Charge_Mixing", "combine_data"); - if (PARAM.inp.nspin == 1) - { - data_s = nullptr; - data_hf = nullptr; - return; - } - else - { - const int ndimd = this->rhodpw->npw; - const int ndims = this->rhopw->npw; - const int ndimhf = ndimd - ndims; - for (int is = 0; is < PARAM.inp.nspin; ++is) - { - std::memcpy(data_d + is * ndimd, data_s + is * ndims, ndims * sizeof(std::complex)); - std::memcpy(data_d + is * ndimd + ndims, data_hf + is * ndimhf, ndimhf * sizeof(std::complex)); - } - delete[] data_s; - delete[] data_hf; - data_s = nullptr; - data_hf = nullptr; - } -} - -void Charge_Mixing::clean_data(std::complex*& data_s, std::complex*& data_hf) -{ - ModuleBase::TITLE("Charge_Mixing", "clean_data"); - if (PARAM.inp.nspin == 1) - { - data_s = nullptr; - data_hf = nullptr; - return; - } - else - { - delete[] data_s; - delete[] data_hf; - data_s = nullptr; - data_hf = nullptr; - } -} \ No newline at end of file diff --git a/source/source_estate/module_charge/charge_mpi.cpp b/source/source_estate/module_charge/charge_mpi.cpp deleted file mode 100644 index 8ef4cce4a32..00000000000 --- a/source/source_estate/module_charge/charge_mpi.cpp +++ /dev/null @@ -1,67 +0,0 @@ -#include "charge.h" -#include "source_base/global_function.h" -#include "source_base/global_variable.h" -#include "source_base/parallel_comm.h" -#include "source_base/timer.h" -#include "source_hamilt/module_xc/xc_functional.h" -#include "source_io/module_parameter/parameter.h" -#ifdef __MPI -void Charge::reduce_diff_pools(double* array_rho) const -{ - ModuleBase::TITLE("Charge", "reduce_diff_pools"); - ModuleBase::timer::start("Charge", "reduce_diff_pools"); - if (GlobalV::KPAR > 1) - { - assert(this->pgrid != nullptr); - this->pgrid->reduce_across_pools(array_rho); - } - if (PARAM.globalv.all_ks_run && PARAM.inp.bndpar > 1) - { - MPI_Allreduce(MPI_IN_PLACE, array_rho, this->nrxx, MPI_DOUBLE, MPI_SUM, BP_WORLD); - } - ModuleBase::timer::end("Charge", "reduce_diff_pools"); -} - -void Charge::rho_mpi() -{ - ModuleBase::TITLE("Charge", "rho_mpi"); - if (GlobalV::KPAR * PARAM.inp.bndpar <= 1) - { - return; - } - ModuleBase::timer::start("Charge", "rho_mpi"); - - for (int is = 0; is < PARAM.inp.nspin; ++is) - { - reduce_diff_pools(this->rho[is]); - if (XC_Functional::get_ked_flag() || PARAM.inp.out_elf[0] > 0) - { - reduce_diff_pools(this->kin_r[is]); - } - } - - ModuleBase::timer::end("Charge", "rho_mpi"); - return; -} - -void Charge::kin_r_mpi() -{ - ModuleBase::TITLE("Charge", "kin_r_mpi"); - if (GlobalV::KPAR * PARAM.inp.bndpar <= 1) - { - return; - } - ModuleBase::timer::start("Charge", "kin_r_mpi"); - - if (XC_Functional::get_ked_flag() || PARAM.inp.out_elf[0] > 0) - { - for (int is = 0; is < PARAM.inp.nspin; ++is) - { - reduce_diff_pools(this->kin_r[is]); - } - } - - ModuleBase::timer::end("Charge", "kin_r_mpi"); - return; -} -#endif diff --git a/source/source_estate/module_charge/chg_atomic.cpp b/source/source_estate/module_charge/chg_atomic.cpp new file mode 100644 index 00000000000..70bf34ba2ee --- /dev/null +++ b/source/source_estate/module_charge/chg_atomic.cpp @@ -0,0 +1,90 @@ +#include "chg_atomic.h" +#include "chg_atomic_detail.h" + +#include "source_base/global_variable.h" +#include "source_base/timer.h" +#include "source_base/tool_quit.h" +#include "source_base/tool_title.h" +#include "source_cell/unitcell.h" +#include "source_cell/magnetism.h" + +#include + +namespace module_charge +{ + +void atomic_rho(const int spin_number_need, + const double& omega, + double** rho_in, + const ModuleBase::ComplexMatrix& strucFac, + const UnitCell& ucell, + const ModulePW::PW_Basis* rhopw, + const AtomicRhoCfg& cfg) +{ + ModuleBase::TITLE("module_charge", "atomic_rho"); + ModuleBase::timer::start("module_charge", "atomic_rho"); + + std::ostream& ofs_warning = cfg.ofs_warning; + const int test_charge = cfg.test_charge; + const bool domag = cfg.domag; + const bool domag_z = cfg.domag_z; + + ModuleBase::ComplexMatrix rho_g3d(spin_number_need, rhopw->npw); + + for (int it = 0; it < ucell.ntype; it++) + { + // check the start magnetization + const int startmag_type = (ucell.magnet.start_mag[it] != 0.0) ? 1 : 2; + ofs_warning << " " << std::setw(40) << "startmag_type" + << " = " << startmag_type << std::endl; + + const Atom* const atom = &ucell.atoms[it]; + + if (!atom->flag_empty_element) // Peize Lin add for bsse 2021.04.07 + { + const int mesh = atom->ncpp.msh; + const std::vector rhoatm + = detail::compute_rhoatm(*atom, mesh, ofs_warning); + const std::vector rho_lgl + = detail::compute_rho_lgl(*atom, rhopw, ucell, rhoatm, + test_charge, omega, ofs_warning); + + detail::RhoG3dCtx ctx{rho_g3d, strucFac, rho_lgl, rhopw, it}; + + if (spin_number_need == 1) + { + detail::fill_rho_g3d_nspin1(ctx); + } + else if (spin_number_need == 2) + { + detail::fill_rho_g3d_nspin2(ctx, startmag_type, + ucell.magnet.start_mag[it], *atom); + } + else if (spin_number_need == 4) + { + if (startmag_type == 1) + { + detail::fill_rho_g3d_nspin4_type1(ctx, + ucell.magnet.start_mag[it], + *atom, domag, domag_z); + } + else + { + detail::fill_rho_g3d_nspin4_type2(ctx, *atom, domag, domag_z); + } + } + else + { + ModuleBase::WARNING_QUIT("module_charge::atomic_rho", + " Either 1 or 2 or 4, check SPIN number !"); + } + } + } + + detail::normalize_and_check(rho_in, rho_g3d, rhopw, spin_number_need, + omega, ofs_warning, cfg.nelec); + + ModuleBase::timer::end("module_charge", "atomic_rho"); +} + +} // namespace module_charge diff --git a/source/source_estate/module_charge/chg_atomic.h b/source/source_estate/module_charge/chg_atomic.h new file mode 100644 index 00000000000..fa6c5d09c79 --- /dev/null +++ b/source/source_estate/module_charge/chg_atomic.h @@ -0,0 +1,36 @@ +#ifndef CHG_ATOMIC_H +#define CHG_ATOMIC_H + +#include "source_base/complexmatrix.h" +#include "source_basis/module_pw/pw_basis.h" + +#include + +class UnitCell; + +namespace module_charge +{ + +/// Configuration for atomic_rho, replacing GlobalV/PARAM reads +struct AtomicRhoCfg +{ + double nelec; ///< target total electron number (PARAM.inp.nelec) + int test_charge; ///< verbosity flag (PARAM.inp.test_charge) + bool domag; ///< whether to compute magnetization (PARAM.globalv.domag) + bool domag_z; ///< whether to compute z-only magnetization + std::ostream& ofs_warning; ///< warning output stream +}; + +// Superposition of atomic charges contained in the array rho_at +// (read from pseudopotential files). +void atomic_rho(const int spin_number_need, + const double& omega, + double** rho_in, + const ModuleBase::ComplexMatrix& strucFac, + const UnitCell& ucell, + const ModulePW::PW_Basis* rhopw, + const AtomicRhoCfg& cfg); + +} // namespace module_charge + +#endif // CHG_ATOMIC_H diff --git a/source/source_estate/module_charge/chg_atomic_detail.h b/source/source_estate/module_charge/chg_atomic_detail.h new file mode 100644 index 00000000000..1b490dd4574 --- /dev/null +++ b/source/source_estate/module_charge/chg_atomic_detail.h @@ -0,0 +1,80 @@ +#ifndef CHG_ATOMIC_DETAIL_H +#define CHG_ATOMIC_DETAIL_H + +// Internal helpers for atomic_rho (chg_atomic.cpp). +// Not part of the public module_charge API: only chg_atomic.cpp and +// chg_atomic_inner.cpp are expected to include this header. + +#include + +#include "source_base/complexmatrix.h" +#include "source_basis/module_pw/pw_basis.h" +#include "source_cell/unitcell.h" + +#include + +namespace module_charge +{ +namespace detail +{ + +/// Context bundling commonly-used data for rho_g3d fill helpers. +struct RhoG3dCtx +{ + ModuleBase::ComplexMatrix& rho_g3d; + const ModuleBase::ComplexMatrix& strucFac; + const std::vector& rho_lgl; + const ModulePW::PW_Basis* rhopw; + int it; +}; + +/// Compute the radial atomic charge density rhoatm from pseudopotential. +std::vector compute_rhoatm(const Atom& atom, + int mesh, + std::ostream& ofs_warning); + +/// Compute the 1D charge in G space from rhoatm. +std::vector compute_rho_lgl(const Atom& atom, + const ModulePW::PW_Basis* rhopw, + const UnitCell& ucell, + const std::vector& rhoatm, + int test_charge, + double omega, + std::ostream& ofs_warning); + +/// Fill rho_g3d for nspin==1 case. +void fill_rho_g3d_nspin1(RhoG3dCtx& ctx); + +/// Fill rho_g3d for nspin==2 case (both startmag_type 1 and 2). +void fill_rho_g3d_nspin2(RhoG3dCtx& ctx, + int startmag_type, + double start_mag, + const Atom& atom); + +/// Fill rho_g3d for nspin==4, startmag_type==1 case. +void fill_rho_g3d_nspin4_type1(RhoG3dCtx& ctx, + double start_mag, + const Atom& atom, + bool domag, + bool domag_z); + +/// Fill rho_g3d for nspin==4, startmag_type==2 case. +void fill_rho_g3d_nspin4_type2(RhoG3dCtx& ctx, + const Atom& atom, + bool domag, + bool domag_z); + +/// FFT rho_g3d to real space, check for negative/imaginary charge, +/// and normalize to target electron number. +void normalize_and_check(double** rho_in, + const ModuleBase::ComplexMatrix& rho_g3d, + const ModulePW::PW_Basis* rhopw, + int spin_number_need, + double omega, + std::ostream& ofs_warning, + double nelec); + +} // namespace detail +} // namespace module_charge + +#endif // CHG_ATOMIC_DETAIL_H diff --git a/source/source_estate/module_charge/chg_atomic_inner.cpp b/source/source_estate/module_charge/chg_atomic_inner.cpp new file mode 100644 index 00000000000..ef77dc301e1 --- /dev/null +++ b/source/source_estate/module_charge/chg_atomic_inner.cpp @@ -0,0 +1,421 @@ +#include "chg_atomic_detail.h" + +#include "source_base/global_variable.h" +#include "source_base/libm/libm.h" +#include "source_base/math_integral.h" +#include "source_base/parallel_reduce.h" +#include "source_base/timer.h" +#include "source_cell/unitcell.h" + +#include +#include +#include +#include +#include +#include + +namespace module_charge +{ +namespace detail +{ + +std::vector compute_rhoatm(const Atom& atom, int mesh, std::ostream& ofs_warning) +{ + std::vector rhoatm(mesh); + // this is only one part of the charge density for uspp + // liuyu 2023-11-01 + if (atom.ncpp.tvanp) + { + for (int ir = 0; ir < mesh; ++ir) + { + rhoatm[ir] = atom.ncpp.rho_at[ir]; + } + } + else + { + for (int ir = 0; ir < mesh; ++ir) + { + double r2 = atom.ncpp.r[ir] * atom.ncpp.r[ir]; + if (r2 != 0) + { + rhoatm[ir] = atom.ncpp.rho_at[ir] / ModuleBase::FOUR_PI / r2; + } + } + rhoatm[0] = pow((rhoatm[2] / rhoatm[1]), + atom.ncpp.r[1] / (atom.ncpp.r[2] - atom.ncpp.r[1])); + if (rhoatm[0] < 1e-12) + { + rhoatm[0] = rhoatm[1]; + } + else + { + rhoatm[0] = rhoatm[1] / rhoatm[0]; + } + + double charge = 0.0; + ModuleBase::Integral::Simpson_Integral(atom.ncpp.msh, + atom.ncpp.rho_at.data(), + atom.ncpp.rab.data(), + charge); + ofs_warning << " " << std::setw(40) << "charge from rho_at" + << " = " << charge << std::endl; + assert(charge != 0.0 || charge == atom.ncpp.zv); // bsse correction + + double scale = 1.0; + if (charge != atom.ncpp.zv) + { + ofs_warning << " " << std::setw(40) << "charge should be" + << " = " << atom.ncpp.zv << std::endl; + scale = atom.ncpp.zv / charge; + } + + for (int ir = 0; ir < mesh; ++ir) + { + rhoatm[ir] *= scale; + rhoatm[ir] *= (ModuleBase::FOUR_PI * atom.ncpp.r[ir] * atom.ncpp.r[ir]); + } + } + return rhoatm; +} + +std::vector compute_rho_lgl(const Atom& atom, + const ModulePW::PW_Basis* rhopw, + const UnitCell& ucell, + const std::vector& rhoatm, + int test_charge, + double omega, + std::ostream& ofs_warning) +{ + // one dimension of charge in G space. + std::vector rho_lgl(rhopw->ngg, 0); + + // mesh point of this element. + const int mesh = atom.ncpp.msh; + + assert(ucell.meshx > 0); + //---------------------------------------------------------- + // Here we compute the G=0 term + //---------------------------------------------------------- + int gstart = 0; + if (rhopw->gg_uniq[0] < 1e-8) + { + std::vector rho1d(ucell.meshx); + for (int ir = 0; ir < mesh; ir++) + { + rho1d[ir] = rhoatm[ir]; + } + ModuleBase::Integral::Simpson_Integral(mesh, rho1d.data(), + atom.ncpp.rab.data(), rho_lgl[0]); + gstart = 1; + } + if (test_charge > 0) + { + std::cout << "\n |G|=0 term done." << std::endl; + } + //---------------------------------------------------------- + // Here we compute the G<>0 term + // But if in parallel case + // G=0 term only belong to 1 cpu. + // Other processors start from '0' + //---------------------------------------------------------- +#ifdef _OPENMP +#pragma omp parallel + { +#endif + const int ngg = rhopw->ngg; + const double* gg_uniq = rhopw->gg_uniq; + const int meshx = ucell.meshx; + const double tpiba = ucell.tpiba; + std::vector rho1d(meshx); + +#ifdef _OPENMP +#pragma omp for +#endif + for (int igg = gstart; igg < ngg; ++igg) + { + const double gx = sqrt(gg_uniq[igg]) * tpiba; + for (int ir = 0; ir < mesh; ir++) + { + if (atom.ncpp.r[ir] < 1.0e-8) + { + rho1d[ir] = rhoatm[ir]; + } + else + { + const double gxx = gx * atom.ncpp.r[ir]; + rho1d[ir] = rhoatm[ir] * ModuleBase::libm::sin(gxx) / gxx; + } + } + ModuleBase::Integral::Simpson_Integral(mesh, rho1d.data(), + atom.ncpp.rab.data(), rho_lgl[igg]); + } +#ifdef _OPENMP +#pragma omp single +#endif + { + if (test_charge > 0) + { + std::cout << " |G|>0 term done." << std::endl; + } + } + //---------------------------------------------------------- + // EXPLAIN : Complete the transfer of rho from real space to + // reciprocal space + //---------------------------------------------------------- +#ifdef _OPENMP +#pragma omp for +#endif + for (int igg = 0; igg < ngg; igg++) + { + rho_lgl[igg] /= omega; + } +#ifdef _OPENMP + } +#endif + return rho_lgl; +} + +void fill_rho_g3d_nspin1(RhoG3dCtx& ctx) +{ + const int npw = ctx.rhopw->npw; + const int* ig2igg = ctx.rhopw->ig2igg; +#ifdef _OPENMP +#pragma omp parallel for +#endif + for (int ig = 0; ig < npw; ig++) + { + ctx.rho_g3d(0, ig) += ctx.strucFac(ctx.it, ig) * ctx.rho_lgl[ig2igg[ig]]; + } +} + +void fill_rho_g3d_nspin2(RhoG3dCtx& ctx, + int startmag_type, + double start_mag, + const Atom& atom) +{ + if (startmag_type == 1) + { + const int npw = ctx.rhopw->npw; + const int* ig2igg = ctx.rhopw->ig2igg; + const double zv = atom.ncpp.zv; +#ifdef _OPENMP +#pragma omp parallel for +#endif + for (int ig = 0; ig < npw; ig++) + { + const std::complex swap = ctx.strucFac(ctx.it, ig) * ctx.rho_lgl[ig2igg[ig]]; + const double up = 0.5 * (1 + start_mag / zv); + const double dw = 0.5 * (1 - start_mag / zv); + ctx.rho_g3d(0, ig) += swap * up; + ctx.rho_g3d(1, ig) += swap * dw; + } + } + // mohan add 2011-06-14 + else if (startmag_type == 2) + { + std::complex ci_tpi = ModuleBase::NEG_IMAG_UNIT * ModuleBase::TWO_PI; + const int npw = ctx.rhopw->npw; + const ModuleBase::Vector3* gcar = ctx.rhopw->gcar; + const int* ig2igg = ctx.rhopw->ig2igg; + const double zv = atom.ncpp.zv; + for (int ia = 0; ia < atom.na; ia++) + { + const double up = 0.5 * (1 + atom.mag[ia] / atom.ncpp.zv); + const double dw = 0.5 * (1 - atom.mag[ia] / atom.ncpp.zv); + const double tau_x = atom.tau[ia].x; + const double tau_y = atom.tau[ia].y; + const double tau_z = atom.tau[ia].z; +#ifdef _OPENMP +#pragma omp parallel for +#endif + for (int ig = 0; ig < npw; ig++) + { + const double Gtau = gcar[ig][0] * tau_x + gcar[ig][1] * tau_y + gcar[ig][2] * tau_z; + std::complex swap + = ModuleBase::libm::exp(ci_tpi * Gtau) * ctx.rho_lgl[ig2igg[ig]]; + ctx.rho_g3d(0, ig) += swap * up; + ctx.rho_g3d(1, ig) += swap * dw; + } + } + } +} + +void fill_rho_g3d_nspin4_type1(RhoG3dCtx& ctx, + double start_mag, + const Atom& atom, + bool domag, + bool domag_z) +{ + // noncolinear case + double sin_a1 = 0.0; + double sin_a2 = 0.0; + double cos_a1 = 0.0; + double cos_a2 = 0.0; + if (domag) + { + ModuleBase::libm::sincos(atom.angle1[0], &sin_a1, &cos_a1); + ModuleBase::libm::sincos(atom.angle2[0], &sin_a2, &cos_a2); + } + const int npw = ctx.rhopw->npw; + const int* ig2igg = ctx.rhopw->ig2igg; + const double zv = atom.ncpp.zv; +#ifdef _OPENMP +#pragma omp parallel for +#endif + for (int ig = 0; ig < npw; ig++) + { + const std::complex swap = ctx.strucFac(ctx.it, ig) * ctx.rho_lgl[ig2igg[ig]]; + ctx.rho_g3d(0, ig) += swap; + if (domag) + { + ctx.rho_g3d(1, ig) += swap * (start_mag / zv) * sin_a1 * cos_a2; + ctx.rho_g3d(2, ig) += swap * (start_mag / zv) * sin_a1 * sin_a2; + ctx.rho_g3d(3, ig) += swap * (start_mag / zv) * cos_a1; + } + else if (domag_z) + { + ctx.rho_g3d(1, ig) = 0.0; + ctx.rho_g3d(2, ig) = 0.0; + ctx.rho_g3d(3, ig) += swap * (start_mag / zv); + } + } +} + +void fill_rho_g3d_nspin4_type2(RhoG3dCtx& ctx, + const Atom& atom, + bool domag, + bool domag_z) +{ + std::complex ci_tpi = ModuleBase::NEG_IMAG_UNIT * ModuleBase::TWO_PI; + const int npw = ctx.rhopw->npw; + const ModuleBase::Vector3* gcar = ctx.rhopw->gcar; + const int* ig2igg = ctx.rhopw->ig2igg; + const double zv = atom.ncpp.zv; + for (int ia = 0; ia < atom.na; ia++) + { + double sin_a1 = 0.0; + double sin_a2 = 0.0; + double cos_a1 = 0.0; + double cos_a2 = 0.0; + if (domag || domag_z) + { + ModuleBase::libm::sincos(atom.angle1[ia], &sin_a1, &cos_a1); + } + if (domag) + { + ModuleBase::libm::sincos(atom.angle2[ia], &sin_a2, &cos_a2); + } + const double mag_ia = atom.mag[ia]; + const double tau_x = atom.tau[ia].x; + const double tau_y = atom.tau[ia].y; + const double tau_z = atom.tau[ia].z; +#ifdef _OPENMP +#pragma omp parallel for +#endif + for (int ig = 0; ig < npw; ig++) + { + const double Gtau = gcar[ig][0] * tau_x + gcar[ig][1] * tau_y + gcar[ig][2] * tau_z; + std::complex swap = ModuleBase::libm::exp(ci_tpi * Gtau) * ctx.rho_lgl[ig2igg[ig]]; + const double mag_factor = mag_ia / zv; + ctx.rho_g3d(0, ig) += swap; + if (domag || domag_z) + { + ctx.rho_g3d(3, ig) += swap * mag_factor * cos_a1; + } + if (domag) + { + ctx.rho_g3d(1, ig) += swap * mag_factor * sin_a1 * cos_a2; + ctx.rho_g3d(2, ig) += swap * mag_factor * sin_a1 * sin_a2; + } + else + { + ctx.rho_g3d(1, ig) = 0.0; + ctx.rho_g3d(2, ig) = 0.0; + } + } + } +} + +void normalize_and_check(double** rho_in, + const ModuleBase::ComplexMatrix& rho_g3d, + const ModulePW::PW_Basis* rhopw, + int spin_number_need, + double omega, + std::ostream& ofs_warning, + double nelec) +{ + assert(spin_number_need > 0); + std::vector ne(spin_number_need); + for (int is = 0; is < spin_number_need; is++) + { + rhopw->recip2real(&rho_g3d(is, 0), rho_in[is]); + + for (int ir = 0; ir < rhopw->nrxx; ++ir) + { + ne[is] += rho_in[is][ir]; + } + + ne[is] *= omega / static_cast(rhopw->nxyz); +#ifdef __MPI + Parallel_Reduce::reduce_pool(ne[is]); +#endif + // we check that everything is correct + double neg = 0.0; + double rea = 0.0; + double ima = 0.0; + double sumrea = 0.0; + for (int ir = 0; ir < rhopw->nrxx; ir++) + { + rea = rhopw->fft_bundle.get_auxr_data()[ir].real(); + sumrea += rea; + neg += std::min(0.0, rea); + ima += std::abs(rhopw->fft_bundle.get_auxr_data()[ir].imag()); + } + +#ifdef __MPI + Parallel_Reduce::reduce_pool(neg); + Parallel_Reduce::reduce_pool(ima); + Parallel_Reduce::reduce_pool(sumrea); +#endif + // mohan fix bug 2011-04-03 + neg = neg / static_cast(rhopw->nxyz) * omega; + ima = ima / static_cast(rhopw->nxyz) * omega; + sumrea = sumrea / static_cast(rhopw->nxyz) * omega; + + if (((neg < -1.0e-4) && (is == 0 || spin_number_need == 2)) || ima > 1.0e-4) + { + ofs_warning << " Warning: negative or imaginary starting charge : "; + ofs_warning << " neg = " << neg << " ima = " << ima << " SPIN = " << is << std::endl; + } + + } // end is + + double ne_tot = 0.0; + int spin0 = 1; + if (spin_number_need == 2) + { + spin0 = spin_number_need; + } + for (int is = 0; is < spin0; ++is) + { + ofs_warning << "\n SETUP ATOMIC RHO FOR SPIN " << is + 1 << std::endl; + ofs_warning << " " << std::setw(40) << "Electron number from rho" + << " = " << ne[is] << std::endl; + ne_tot += ne[is]; + } + ofs_warning << " " << std::setw(40) << "total electron number from rho" + << " = " << ne_tot << std::endl; + ofs_warning << " " << std::setw(40) << "should be" + << " = " << nelec << std::endl; + + for (int is = 0; is < spin_number_need; ++is) + { + for (int ir = 0; ir < rhopw->nrxx; ++ir) + { + rho_in[is][ir] = rho_in[is][ir] / ne_tot * nelec; + } + } +} + +} // namespace detail +} // namespace module_charge diff --git a/source/source_estate/module_charge/chg_dmr.cpp b/source/source_estate/module_charge/chg_dmr.cpp new file mode 100644 index 00000000000..02d104f4731 --- /dev/null +++ b/source/source_estate/module_charge/chg_dmr.cpp @@ -0,0 +1,186 @@ +#include "chg_dmr.h" + +#include +#include + +#include "source_base/module_mixing/mixing.h" +#include "source_base/timer.h" +#include "source_base/tool_quit.h" +#include "source_base/tool_title.h" + +namespace module_charge +{ +namespace +{ + +/** + * @brief Two-beta plain step for the magnetic density matrix. + * + * The first nnr elements are the charge channel (beta), the next nnr + * elements are the magnetization channel (beta_mag). + * + * @param out mixed output, length 2 * nnr + * @param in mixed input, length 2 * nnr + * @param sres residual, length 2 * nnr + * @param nnr number of DMR elements per spin channel + * @param beta mixing beta for the charge channel + * @param beta_mag mixing beta for the magnetization channel + */ +void twobeta_step(double* out, + const double* in, + const double* sres, + const int nnr, + const double beta, + const double beta_mag) +{ +#ifdef _OPENMP +#pragma omp parallel for schedule(static, 256) +#endif + for (int i = 0; i < nnr; ++i) + { + out[i] = in[i] + beta * sres[i]; + } +#ifdef _OPENMP +#pragma omp parallel for schedule(static, 256) +#endif + for (int i = nnr; i < 2 * nnr; ++i) + { + out[i] = in[i] + beta_mag * sres[i]; + } +} + +/** + * @brief Validate the arguments of mix_dmr. Aborts via WARNING_QUIT on the + * first invalid input. + */ +void check_dmr_inputs(const std::vector& dmr_out, + const std::vector& dmr_in, + const int nnr, + const Base_Mixing::Mixing* mixing, + const MixingConfig& cfg) +{ + if (mixing == nullptr) + { + ModuleBase::WARNING_QUIT("module_charge::mix_dmr", "mixing pointer is null"); + } + // nnr is local to each MPI rank and may legitimately be zero when no atom + // pairs survive the cutoff on that rank; only negative values are invalid. + if (nnr < 0) + { + ModuleBase::WARNING_QUIT("module_charge::mix_dmr", "nnr must be >= 0"); + } + if (cfg.nspin != 1 && cfg.nspin != 2 && cfg.nspin != 4) + { + ModuleBase::WARNING_QUIT("module_charge::mix_dmr", "unsupported nspin, require 1, 2 or 4"); + } + const int nspin_need = (cfg.nspin == 2) ? 2 : 1; + if (static_cast(dmr_out.size()) < nspin_need + || static_cast(dmr_in.size()) < nspin_need) + { + ModuleBase::WARNING_QUIT("module_charge::mix_dmr", "not enough DMR buffers for nspin"); + } + // Non-null buffers are required only when there are elements to mix; an + // empty partition (nnr == 0) may legitimately carry null pointers. + for (int is = 0; is < nspin_need; ++is) + { + if (nnr > 0 && (dmr_out[is] == nullptr || dmr_in[is] == nullptr)) + { + ModuleBase::WARNING_QUIT("module_charge::mix_dmr", "DMR buffer pointer is null"); + } + } +} + +} // namespace + +void init_mixing_dmr(Base_Mixing::Mixing* mixing, + Base_Mixing::Mixing_Data& mdata, + const int nnr, + const MixingConfig& cfg) +{ + ModuleBase::TITLE("module_charge", "init_mixing_dmr"); + ModuleBase::timer::start("module_charge", "init_mixing_dmr"); + if (mixing == nullptr) + { + ModuleBase::WARNING_QUIT("module_charge::init_mixing_dmr", "mixing pointer is null"); + } + // nnr is local to each MPI rank and may legitimately be zero when no atom + // pairs survive the cutoff on that rank; only negative values are invalid. + if (nnr < 0) + { + ModuleBase::WARNING_QUIT("module_charge::init_mixing_dmr", "nnr must be >= 0"); + } + + const int dmr_nspin = (cfg.nspin == 2) ? 2 : 1; + // DMR mixing currently supports only the real-space convergence threshold. + if (cfg.scf_thr_type == 1) + { + ModuleBase::WARNING_QUIT("module_charge::init_mixing_dmr", + "This Mixing of Density Matrix is not supported for PW basis yet"); + } + else if (cfg.scf_thr_type == 2) + { + mixing->init_mixing_data(mdata, nnr * dmr_nspin, sizeof(double)); + } + + // Clear the history counters while keeping the allocated storage. + mdata.reset(); + ModuleBase::timer::end("module_charge", "init_mixing_dmr"); +} + +void mix_dmr(const std::vector& dmr_out, + const std::vector& dmr_in, + const int nnr, + Base_Mixing::Mixing* mixing, + Base_Mixing::Mixing_Data& mdata, + const MixingConfig& cfg) +{ + ModuleBase::TITLE("module_charge", "mix_dmr"); + ModuleBase::timer::start("module_charge", "mix_dmr"); + check_dmr_inputs(dmr_out, dmr_in, nnr, mixing, cfg); + + if (cfg.nspin == 1 || cfg.nspin == 4) + { + mixing->push_data(mdata, dmr_in[0], dmr_out[0], nullptr, false); + mixing->mix_data(mdata, dmr_out[0]); + } + else // cfg.nspin == 2 + { + // Magnetic density matrix: up/down channels are transformed into + // charge/magnetization channels before mixing and back afterwards. + std::vector dmr_mag(nnr * cfg.nspin, 0.0); + std::vector dmr_mag_save(nnr * cfg.nspin, 0.0); + + // Transfer the current DMR into the charge/magnetization layout. + for (int ir = 0; ir < nnr; ++ir) + { + dmr_mag[ir] = dmr_out[0][ir] + dmr_out[1][ir]; + dmr_mag[ir + nnr] = dmr_out[0][ir] - dmr_out[1][ir]; + } + // Transfer the saved DMR into the charge/magnetization layout. + for (int ir = 0; ir < nnr; ++ir) + { + dmr_mag_save[ir] = dmr_in[0][ir] + dmr_in[1][ir]; + dmr_mag_save[ir + nnr] = dmr_in[0][ir] - dmr_in[1][ir]; + } + + const double beta = cfg.mixing_beta; + const double beta_mag = cfg.mixing_beta_mag; + std::function twobeta + = [nnr, beta, beta_mag](double* out, const double* in, const double* sres) { + twobeta_step(out, in, sres, nnr, beta, beta_mag); + }; + // No Kerker screening in DMR mixing. + mixing->push_data(mdata, dmr_mag_save.data(), dmr_mag.data(), nullptr, twobeta, false); + mixing->mix_data(mdata, dmr_mag.data()); + + // Transform the mixed charge/magnetization channels back to up/down. + for (int ir = 0; ir < nnr; ++ir) + { + dmr_out[0][ir] = 0.5 * (dmr_mag[ir] + dmr_mag[ir + nnr]); + dmr_out[1][ir] = 0.5 * (dmr_mag[ir] - dmr_mag[ir + nnr]); + } + } + ModuleBase::timer::end("module_charge", "mix_dmr"); +} + +} // namespace module_charge diff --git a/source/source_estate/module_charge/chg_dmr.h b/source/source_estate/module_charge/chg_dmr.h new file mode 100644 index 00000000000..916b5929f20 --- /dev/null +++ b/source/source_estate/module_charge/chg_dmr.h @@ -0,0 +1,68 @@ +#ifndef CHG_DMR_H +#define CHG_DMR_H + +// Stateless real-space density-matrix (DMR) mixing kernels extracted from +// Charge_Mixing. The mixing history (Mixing_Data) and the Mixing strategy +// object remain owned by Charge_Mixing and are passed in explicitly; the +// functions do not read Charge_Mixing members or PARAM/GlobalV. The kernels +// work on raw contiguous double buffers (one per spin channel) and do not +// depend on DensityMatrix or HContainer; the caller extracts those buffers +// from its LCAO containers. + +#include + +#include "chg_mix_cfg.h" + +namespace Base_Mixing +{ +class Mixing; +class Mixing_Data; +} // namespace Base_Mixing + +namespace module_charge +{ + +/** + * @brief Allocate the mixing buffer for the real-space density matrix and + * clear its history. + * + * The buffer cannot be allocated in Charge_Mixing::set_mixing(): its length + * nnr (number of non-zero R-matrix elements) is only known after + * DensityMatrix::init_dmr(), which runs later in beforescf(). + * + * @param mixing mixing strategy object, non-null + * @param mdata mixing history buffer for DMR, resized and reset in place + * @param nnr number of real-space density-matrix elements per spin, > 0 + * @param cfg mixing config (nspin and scf_thr_type select the path) + */ +void init_mixing_dmr(Base_Mixing::Mixing* mixing, + Base_Mixing::Mixing_Data& mdata, + const int nnr, + const MixingConfig& cfg); + +/** + * @brief Mix the real-space density matrix (LCAO calculations only). + * + * For nspin == 1/4 the single spin channel is mixed directly; for nspin == 2 + * the up/down channels are transformed into charge/magnetization channels, + * mixed with independent betas, and transformed back. + * + * @param dmr_out writable DMR buffers, one per spin channel, each of length + * nnr; mixed results are written back through these pointers + * @param dmr_in DMR buffers saved at the previous mixing step, one per spin + * channel, each of length nnr (read-only) + * @param nnr number of DMR elements per spin channel, > 0 + * @param mixing mixing strategy object, non-null + * @param mdata DMR mixing history buffer + * @param cfg mixing config (nspin and the two mixing betas) + */ +void mix_dmr(const std::vector& dmr_out, + const std::vector& dmr_in, + const int nnr, + Base_Mixing::Mixing* mixing, + Base_Mixing::Mixing_Data& mdata, + const MixingConfig& cfg); + +} // namespace module_charge + +#endif // CHG_DMR_H diff --git a/source/source_estate/module_charge/chg_drho.cpp b/source/source_estate/module_charge/chg_drho.cpp new file mode 100644 index 00000000000..abb68c3746a --- /dev/null +++ b/source/source_estate/module_charge/chg_drho.cpp @@ -0,0 +1,160 @@ +#include "charge.h" +#include "chg_drho.h" +#include "chg_drho_detail.h" +#include "source_base/timer.h" +#include "source_base/parallel_reduce.h" + +#include +#include + +namespace module_charge +{ + +// Charge residual between chr->rho and chr->rho_save, normalized per electron. +double cal_drho(Charge* chr, + const double nelec, + const ModulePW::PW_Basis& rhopw, + const MixingConfig& cfg, + const double omega, + const double tpiba) +{ + assert(chr != nullptr); + ModuleBase::TITLE("module_charge", "cal_drho"); + ModuleBase::timer::start("module_charge", "cal_drho"); + const int nspin = cfg.nspin; + assert(nspin==1 || nspin==2 || nspin==4); + double drho = 0.0; + + if (cfg.scf_thr_type == 1) + { + for (int is = 0; is < nspin; ++is) + { + ModuleBase::GlobalFunc::NOTE("Perform FFT on rho(r) to obtain rho(G)."); + chr->rhopw->real2recip(chr->rho[is], chr->rhog[is]); + + ModuleBase::GlobalFunc::NOTE("Perform FFT on rho_save(r) to obtain rho_save(G)."); + chr->rhopw->real2recip(chr->rho_save[is], chr->rhog_save[is]); + } + + ModuleBase::GlobalFunc::NOTE("Calculate the charge difference between rho(G) and rho_save(G)"); + std::vector> drhog(nspin * rhopw.npw); +#ifdef _OPENMP +#pragma omp parallel for collapse(2) schedule(static, 512) +#endif + for (int is = 0; is < nspin; ++is) + { + for (int ig = 0; ig < rhopw.npw; ig++) + { + drhog[is * rhopw.npw + ig] = chr->rhog[is][ig] - chr->rhog_save[is][ig]; + } + } + + ModuleBase::GlobalFunc::NOTE("Calculate the norm of the Residual std::vector: < R[rho] | R[rho_save] >"); + drho = module_charge::detail::inner_product_recip_rho( + drhog.data(), drhog.data(), rhopw, cfg, omega, tpiba); + } + else + { + // Note: Maybe it is wrong. + // The inner_product_real function (L1-norm) is different from that (L2-norm) in mixing. + for (int is = 0; is < nspin; is++) + { + if (is != 0 && is != 3 && cfg.domag_z) + { + continue; + } +#ifdef _OPENMP +#pragma omp parallel for reduction(+ : drho) +#endif + for (int ir = 0; ir < rhopw.nrxx; ir++) + { + drho += std::abs(chr->rho[is][ir] - chr->rho_save[is][ir]); + } + } +#ifdef __MPI + Parallel_Reduce::reduce_pool(drho); +#endif + assert(nelec != 0); + assert(omega > 0); + assert(rhopw.nxyz > 0); + drho *= omega / static_cast(rhopw.nxyz); + drho /= nelec; + } + + ModuleBase::timer::end("module_charge", "cal_drho"); + return drho; +} + +// Kinetic-energy-density residual between chr->kin_r and chr->kin_r_save. +double cal_dkin(Charge* chr, + const double nelec, + const ModulePW::PW_Basis& rhopw, + const MixingConfig& cfg, + const double omega) +{ + assert(chr != nullptr); + if (!(chr->meta_gga)) + { + return 0.0; + }; + ModuleBase::TITLE("module_charge", "cal_dkin"); + ModuleBase::timer::start("module_charge", "cal_dkin"); + double dkin = 0.0; + + // Get dkin from kin_r and kin_r_save for PW and LCAO both, which is different from drho. + for (int is = 0; is < cfg.nspin; is++) + { + if (is != 0 && is != 3 && cfg.domag_z) + { + continue; + } +#ifdef _OPENMP +#pragma omp parallel for reduction(+ : dkin) +#endif + for (int ir = 0; ir < rhopw.nrxx; ir++) + { + dkin += std::abs(chr->kin_r[is][ir] - chr->kin_r_save[is][ir]); + } + } +#ifdef __MPI + Parallel_Reduce::reduce_pool(dkin); +#endif + assert(nelec != 0); + assert(omega > 0); + assert(rhopw.nxyz > 0); + dkin *= omega / static_cast(rhopw.nxyz); + dkin /= nelec; + + ModuleBase::timer::end("module_charge", "cal_dkin"); + return dkin; +} + +double inner_product_real(const double* rho1, + const double* rho2, + const ModulePW::PW_Basis& rhopw, + const MixingConfig& cfg) +{ + assert(rho1 != nullptr); + assert(rho2 != nullptr); + double rnorm = 0.0; + // consider a resize for mixing_angle + int resize_tmp = 1; + if (cfg.nspin == 4 && cfg.mixing_angle > 0) + { + resize_tmp = 2; + } + +#ifdef _OPENMP +#pragma omp parallel for reduction(+ : rnorm) +#endif + for (int ir = 0; ir < rhopw.nrxx * cfg.nspin / resize_tmp; ++ir) + { + rnorm += rho1[ir] * rho2[ir]; + } +#ifdef __MPI + Parallel_Reduce::reduce_pool(rnorm); +#endif + return rnorm; +} + +} // namespace module_charge diff --git a/source/source_estate/module_charge/chg_drho.h b/source/source_estate/module_charge/chg_drho.h new file mode 100644 index 00000000000..cae85dbb5bc --- /dev/null +++ b/source/source_estate/module_charge/chg_drho.h @@ -0,0 +1,90 @@ +#ifndef CHG_DRHO_H +#define CHG_DRHO_H + +// Stateless residual kernels extracted from Charge_Mixing. Every input +// (grid, geometry, mixing config) is passed explicitly; the functions do +// not read Charge_Mixing members or PARAM/GlobalV. + +#include + +#include "chg_mix_cfg.h" + +namespace ModulePW +{ +class PW_Basis; +} + +class Charge; + +namespace module_charge +{ + +/** + * @brief Charge residual between chr->rho and chr->rho_save, normalized per electron. + * + * @param chr charge object supplying rho/rho_save (and rhog buffers for the reciprocal case) + * @param nelec number of electrons, used to normalize the real-space residual + * @param rhopw plane-wave basis supplying the real/reciprocal grid sizes + * @param cfg mixing config (nspin, scf_thr_type and magnetization flags select the loops) + * @param omega cell volume, used to normalize the real-space residual + * @param tpiba 2*pi/lattice constant, used by the reciprocal metric + * @return pooled residual value + */ +double cal_drho(Charge* chr, + const double nelec, + const ModulePW::PW_Basis& rhopw, + const MixingConfig& cfg, + const double omega, + const double tpiba); + +/** + * @brief Kinetic-energy-density residual between chr->kin_r and chr->kin_r_save. + * + * @param chr charge object supplying kin_r/kin_r_save + * @param nelec number of electrons, used to normalize the residual + * @param rhopw plane-wave basis supplying the real-space grid size + * @param cfg mixing config (nspin and magnetization flags select the loops) + * @param omega cell volume, used to normalize the residual + * @return pooled residual value + */ +double cal_dkin(Charge* chr, + const double nelec, + const ModulePW::PW_Basis& rhopw, + const MixingConfig& cfg, + const double omega); + +/** + * @brief Inner product of two real-space vectors used in real-space mixing. + * + * @param rho1 first real-space vector + * @param rho2 second real-space vector + * @param rhopw plane-wave basis supplying the real-space grid size + * @param cfg mixing config (nspin and mixing_angle select the loop bound) + * @return pooled inner product + */ +double inner_product_real(const double* rho1, + const double* rho2, + const ModulePW::PW_Basis& rhopw, + const MixingConfig& cfg); + +/** + * @brief Hartree-like reciprocal inner product used in charge mixing. + * + * @param rhog1 first reciprocal-space vector + * @param rhog2 second reciprocal-space vector + * @param rhopw plane-wave basis supplying npw/gg and the G=0 index + * @param cfg mixing config (spin channels, gamma-only and angle flags) + * @param omega cell volume + * @param tpiba 2*pi/lattice constant + * @return pooled Hartree inner product + */ +double inner_product_recip_hartree(const std::complex* rhog1, + const std::complex* rhog2, + const ModulePW::PW_Basis& rhopw, + const MixingConfig& cfg, + const double omega, + const double tpiba); + +} // namespace module_charge + +#endif // CHG_DRHO_H diff --git a/source/source_estate/module_charge/chg_drho_detail.h b/source/source_estate/module_charge/chg_drho_detail.h new file mode 100644 index 00000000000..4a6d2553b09 --- /dev/null +++ b/source/source_estate/module_charge/chg_drho_detail.h @@ -0,0 +1,43 @@ +#ifndef CHG_DRHO_DETAIL_H +#define CHG_DRHO_DETAIL_H + +// Internal reciprocal inner product for the charge residual (cal_drho). +// Not part of the public module_charge API: only chg_drho.cpp and the +// charge mixing unit test are expected to include this header. + +#include + +#include "chg_mix_cfg.h" + +namespace ModulePW +{ +class PW_Basis; +} + +namespace module_charge +{ +namespace detail +{ + +/** + * @brief Coulomb-metric reciprocal inner product of the charge residual. + * + * @param rho1 first reciprocal-space vector + * @param rho2 second reciprocal-space vector + * @param rhopw plane-wave basis supplying npw/gg and the G=0 index + * @param cfg mixing config (spin channels, gamma-only and magnetism flags) + * @param omega cell volume + * @param tpiba 2*pi/lattice constant + * @return pooled Coulomb-metric inner product + */ +double inner_product_recip_rho(const std::complex* rho1, + const std::complex* rho2, + const ModulePW::PW_Basis& rhopw, + const MixingConfig& cfg, + const double omega, + const double tpiba); + +} // namespace detail +} // namespace module_charge + +#endif // CHG_DRHO_DETAIL_H diff --git a/source/source_estate/module_charge/chg_drho_inner.cpp b/source/source_estate/module_charge/chg_drho_inner.cpp new file mode 100644 index 00000000000..b5b5927a0a3 --- /dev/null +++ b/source/source_estate/module_charge/chg_drho_inner.cpp @@ -0,0 +1,446 @@ +#include "chg_drho_detail.h" +#include "chg_mix_cfg.h" + +#include +#include +#include + +#include "source_basis/module_pw/pw_basis.h" +#include "source_base/global_function.h" +#include "source_base/parallel_reduce.h" +#include "source_base/timer.h" + +namespace module_charge +{ +namespace detail +{ + +namespace +{ + +/// Coulomb-metric sum over G!=0 for a single spin channel +double coulomb_sum_single(const std::complex* g1, + const std::complex* g2, + const ModulePW::PW_Basis& rhopw, + const double fac) +{ + const int ig0 = rhopw.ig_gge0; + double sum = 0.0; +#ifdef _OPENMP +#pragma omp parallel for reduction(+ : sum) +#endif + for (int ig = 0; ig < rhopw.npw; ++ig) + { + if (ig == ig0) + { + continue; + } + sum += (conj(g1[ig]) * g2[ig]).real() / rhopw.gg[ig]; + } + return sum * fac; +} + +/// Non-magnetic case (nspin==1 or nspin==4 without domag) +double recip_rho_nspin1(const std::complex* rho1, + const std::complex* rho2, + const ModulePW::PW_Basis& rhopw, + const double fac) +{ + return coulomb_sum_single(rho1, rho2, rhopw, fac); +} + +/// Collinear magnetic case (nspin==2) +double recip_rho_nspin2(const std::complex* rho1, + const std::complex* rho2, + const ModulePW::PW_Basis& rhopw, + const MixingConfig& cfg, + const double fac, + const double fac2) +{ + const int npw = rhopw.npw; + const int ig0 = rhopw.ig_gge0; + + // (1) density part: |rho_up + rho_dn|^2 / G^2 + double sum = 0.0; +#ifdef _OPENMP +#pragma omp parallel for reduction(+ : sum) +#endif + for (int ig = 0; ig < npw; ++ig) + { + if (ig == ig0) + { + continue; + } + sum += (conj(rho1[ig] + rho1[ig + npw]) + * (rho2[ig] + rho2[ig + npw])).real() / rhopw.gg[ig]; + } + sum *= fac; + if (cfg.gamma_only_pw) + { + sum *= 2.0; + } + + // (2) magnetization part: |rho_up - rho_dn|^2 (G=0 included) + double sum2 = 0.0; + if (ig0 >= 0) + { + sum2 += fac2 * (conj(rho1[ig0] - rho1[ig0 + npw]) + * (rho2[ig0] - rho2[ig0 + npw])).real(); + } + double mag = 0.0; +#ifdef _OPENMP +#pragma omp parallel for reduction(+ : mag) +#endif + for (int ig = 0; ig < npw; ++ig) + { + if (ig == ig0) + { + continue; + } + mag += (conj(rho1[ig] - rho1[ig + npw]) + * (rho2[ig] - rho2[ig + npw])).real(); + } + mag *= fac2; + if (cfg.gamma_only_pw) + { + mag *= 2.0; + } + sum2 += mag; + return sum + sum2; +} + +/// Non-collinear magnetic case (nspin==4) with magnetization +double recip_rho_nspin4_mag(const std::complex* rho1, + const std::complex* rho2, + const ModulePW::PW_Basis& rhopw, + const MixingConfig& cfg, + const double fac, + const double fac2) +{ + const int npw = rhopw.npw; + const int ig0 = rhopw.ig_gge0; + + // charge part + double sum = 0.0; +#ifdef _OPENMP +#pragma omp parallel for reduction(+ : sum) +#endif + for (int ig = 0; ig < npw; ++ig) + { + if (ig == ig0) + { + continue; + } + sum += (conj(rho1[ig]) * rho2[ig]).real() / rhopw.gg[ig]; + } + sum *= fac; + + // G=0 magnetization term + if (ig0 > 0) + { + sum += fac2 + * ((conj(rho1[ig0 + npw]) * rho2[ig0 + npw]).real() + + (conj(rho1[ig0 + 2 * npw]) * rho2[ig0 + 2 * npw]).real() + + (conj(rho1[ig0 + 3 * npw]) * rho2[ig0 + 3 * npw]).real()); + } + + // G!=0 magnetization term + double fac3 = fac2; + if (cfg.gamma_only_pw) + { + fac3 *= 2.0; + } +#ifdef _OPENMP +#pragma omp parallel for reduction(+ : sum) +#endif + for (int ig = 0; ig < npw; ++ig) + { + if (ig == ig0) + { + continue; + } + sum += fac3 + * ((conj(rho1[ig + npw]) * rho2[ig + npw]).real() + + (conj(rho1[ig + 2 * npw]) * rho2[ig + 2 * npw]).real() + + (conj(rho1[ig + 3 * npw]) * rho2[ig + 3 * npw]).real()); + } + return sum; +} + +/// Non-collinear with domag, traditional mixing (nspin==4, mixing_angle<=0) +double recip_hartree_nspin4_trad(const std::complex* rhog1, + const std::complex* rhog2, + const ModulePW::PW_Basis& rhopw, + const MixingConfig& cfg, + const double fac, + const double fac2) +{ + const int npw = rhopw.npw; + const int ig0 = rhopw.ig_gge0; + + // charge part + double sum = 0.0; +#ifdef _OPENMP +#pragma omp parallel for reduction(+ : sum) +#endif + for (int ig = 0; ig < npw; ++ig) + { + if (ig == ig0) + { + continue; + } + sum += (conj(rhog1[ig]) * rhog2[ig]).real() / rhopw.gg[ig]; + } + sum *= fac; + + // G=0 magnetization + if (ig0 > 0) + { + sum += fac2 + * ((conj(rhog1[ig0 + npw]) * rhog2[ig0 + npw]).real() + + (conj(rhog1[ig0 + 2 * npw]) * rhog2[ig0 + 2 * npw]).real() + + (conj(rhog1[ig0 + 3 * npw]) * rhog2[ig0 + 3 * npw]).real()); + } + + // G!=0 magnetization + double fac3 = fac2; + if (cfg.gamma_only_pw) + { + fac3 *= 2.0; + } +#ifdef _OPENMP +#pragma omp parallel for reduction(+ : sum) +#endif + for (int ig = 0; ig < npw; ++ig) + { + if (ig == ig0) + { + continue; + } + sum += fac3 + * ((conj(rhog1[ig + npw]) * rhog2[ig + npw]).real() + + (conj(rhog1[ig + 2 * npw]) * rhog2[ig + 2 * npw]).real() + + (conj(rhog1[ig + 3 * npw]) * rhog2[ig + 3 * npw]).real()); + } + return sum; +} + +/// Non-collinear with angle mixing (nspin==4, mixing_angle>0) +double recip_hartree_nspin4_angle(const std::complex* rhog1, + const std::complex* rhog2, + const ModulePW::PW_Basis& rhopw, + const MixingConfig& cfg, + const double fac, + const double fac2) +{ + const int npw = rhopw.npw; + const int ig0 = rhopw.ig_gge0; + + // charge part (same as rho only) + double sum = 0.0; +#ifdef _OPENMP +#pragma omp parallel for reduction(+ : sum) +#endif + for (int ig = 0; ig < npw; ++ig) + { + if (ig == ig0) + { + continue; + } + sum += (conj(rhog1[ig]) * rhog2[ig]).real() / rhopw.gg[ig]; + } + sum *= fac; + + // G=0 |m| term + if (ig0 > 0) + { + sum += fac2 * (conj(rhog1[ig0 + npw]) * rhog2[ig0 + npw]).real(); + } + + // G!=0 |m| term + double fac3 = fac2; + if (cfg.gamma_only_pw) + { + fac3 *= 2.0; + } +#ifdef _OPENMP +#pragma omp parallel for reduction(+ : sum) +#endif + for (int ig = 0; ig < npw; ++ig) + { + if (ig == ig0) + { + continue; + } + sum += fac3 * (conj(rhog1[ig + npw]) * rhog2[ig + npw]).real(); + } + return sum; +} + +/// Collinear magnetic case for hartree metric (nspin==2) +double recip_hartree_nspin2(const std::complex* rhog1, + const std::complex* rhog2, + const ModulePW::PW_Basis& rhopw, + const MixingConfig& cfg, + const double fac, + const double fac2) +{ + const int npw = rhopw.npw; + const int ig0 = rhopw.ig_gge0; + + // charge density part + double sum = 0.0; +#ifdef _OPENMP +#pragma omp parallel for reduction(+ : sum) +#endif + for (int ig = 0; ig < npw; ++ig) + { + if (ig == ig0) + { + continue; + } + sum += (conj(rhog1[ig]) * rhog2[ig]).real() / rhopw.gg[ig]; + } + sum *= fac; + if (cfg.gamma_only_pw) + { + sum *= 2.0; + } + + // magnetization part (G=0 included) + double sum2 = 0.0; + if (ig0 >= 0) + { + sum2 += fac2 * (conj(rhog1[ig0 + npw]) * rhog2[ig0 + npw]).real(); + } + double mag = 0.0; +#ifdef _OPENMP +#pragma omp parallel for reduction(+ : mag) +#endif + for (int ig = 0; ig < npw; ++ig) + { + if (ig == ig0) + { + continue; + } + mag += (conj(rhog1[ig + npw]) * rhog2[ig + npw]).real(); + } + mag *= fac2; + if (cfg.gamma_only_pw) + { + mag *= 2.0; + } + sum2 += mag; + return sum + sum2; +} + +} // anonymous namespace + +double inner_product_recip_rho(const std::complex* rho1, + const std::complex* rho2, + const ModulePW::PW_Basis& rhopw, + const MixingConfig& cfg, + const double omega, + const double tpiba) +{ + assert(rho1 != nullptr); + assert(rho2 != nullptr); + assert(cfg.nspin == 1 || cfg.nspin == 2 || cfg.nspin == 4); + ModuleBase::TITLE("Charge_Mixing", "recip_rho"); + ModuleBase::timer::start("Charge_Mixing", "recip_rho"); + + const int nspin = cfg.nspin; + std::vector*> rhog1(nspin); + std::vector*> rhog2(nspin); + for (int is = 0; is < nspin; is++) + { + rhog1[is] = rho1 + is * rhopw.npw; + rhog2[is] = rho2 + is * rhopw.npw; + } + + static const double fac = ModuleBase::e2 * ModuleBase::FOUR_PI / (tpiba * tpiba); + static const double fac2 = ModuleBase::e2 * ModuleBase::FOUR_PI / (ModuleBase::TWO_PI * ModuleBase::TWO_PI); + + double sum = 0.0; + + switch (nspin) + { + case 1: + sum += recip_rho_nspin1(rhog1[0], rhog2[0], rhopw, fac); + break; + case 2: + sum += recip_rho_nspin2(rhog1[0], rhog2[0], rhopw, cfg, fac, fac2); + break; + case 4: + if (!cfg.domag && !cfg.domag_z) + { + sum += recip_rho_nspin1(rhog1[0], rhog2[0], rhopw, fac); + } + else + { + sum += recip_rho_nspin4_mag(rhog1[0], rhog2[0], rhopw, cfg, fac, fac2); + } + break; + } +#ifdef __MPI + Parallel_Reduce::reduce_pool(sum); +#endif + sum *= omega * 0.5; + + ModuleBase::timer::end("Charge_Mixing", "recip_rho"); + return sum; +} + +} // namespace detail + +double inner_product_recip_hartree(const std::complex* rhog1, + const std::complex* rhog2, + const ModulePW::PW_Basis& rhopw, + const MixingConfig& cfg, + const double omega, + const double tpiba) +{ + assert(rhog1 != nullptr); + assert(rhog2 != nullptr); + assert(cfg.nspin == 1 || cfg.nspin == 2 || cfg.nspin == 4); + ModuleBase::TITLE("Charge_Mixing", "recip_hartree"); + ModuleBase::timer::start("Charge_Mixing", "recip_hartree"); + + static const double fac = ModuleBase::e2 * ModuleBase::FOUR_PI / (tpiba * tpiba); + static const double fac2 = ModuleBase::e2 * ModuleBase::FOUR_PI / (ModuleBase::TWO_PI * ModuleBase::TWO_PI); + + double sum = 0.0; + + if (cfg.nspin == 1) + { + sum += detail::coulomb_sum_single(rhog1, rhog2, rhopw, fac); + } + else if (cfg.nspin == 2) + { + sum += detail::recip_hartree_nspin2(rhog1, rhog2, rhopw, cfg, fac, fac2); + } + else if (cfg.nspin == 4) + { + if (!cfg.domag && !cfg.domag_z) + { + sum += detail::coulomb_sum_single(rhog1, rhog2, rhopw, fac); + } + else if (cfg.mixing_angle <= 0) + { + sum += detail::recip_hartree_nspin4_trad(rhog1, rhog2, rhopw, cfg, fac, fac2); + } + else + { + sum += detail::recip_hartree_nspin4_angle(rhog1, rhog2, rhopw, cfg, fac, fac2); + } + } +#ifdef __MPI + Parallel_Reduce::reduce_pool(sum); +#endif + sum *= omega * 0.5; + + ModuleBase::timer::end("Charge_Mixing", "recip_hartree"); + return sum; +} + +} // namespace module_charge diff --git a/source/source_estate/module_charge/chg_extra.cpp b/source/source_estate/module_charge/chg_extra.cpp new file mode 100644 index 00000000000..eb3ca456117 --- /dev/null +++ b/source/source_estate/module_charge/chg_extra.cpp @@ -0,0 +1,320 @@ +#include "chg_extra.h" + +#include "chg_atomic.h" + +#include "source_base/global_function.h" +#include "source_base/global_variable.h" +#include "source_base/timer.h" +#include "source_base/tool_threading.h" +#include "source_io/module_output/cube_io.h" + +namespace module_charge +{ + +Charge_Extra::Charge_Extra() +{ +} + +void Charge_Extra::Init_CE(const int& nspin, const int& natom, const int& nrxx, const std::string chg_extrap) +{ + if (chg_extrap == "none") + { + pot_order = 0; + } + else if (chg_extrap == "atomic") + { + pot_order = 1; + } + else if (chg_extrap == "first-order") + { + pot_order = 2; + } + else if (chg_extrap == "second-order") + { + pot_order = 3; + } + else + { + ModuleBase::WARNING_QUIT("Charge_Extra","charge extrapolation method is not available !"); + } + + this->nspin = nspin; + + if (pot_order > 0) + { + // delta_rho1.resize(this->nspin, std::vector(nrxx, 0.0)); + // delta_rho2.resize(this->nspin, std::vector(nrxx, 0.0)); + // delta_rho3.resize(this->nspin, std::vector(nrxx, 0.0)); + // qianrui replace the above code with the following code. + // The above code cannot passed valgrind tests, which has an invalid read of size 32. + delta_rho1.resize(this->nspin); + delta_rho2.resize(this->nspin); + delta_rho3.resize(this->nspin); + for (int is = 0; is < this->nspin; is++) + { + delta_rho1[is].resize(nrxx, 0.0); + delta_rho2[is].resize(nrxx, 0.0); + delta_rho3[is].resize(nrxx, 0.0); + } + } + + if(pot_order == 3) + { + dis_old1.assign(natom, ModuleBase::Vector3()); + dis_old2.assign(natom, ModuleBase::Vector3()); + dis_now.assign(natom, ModuleBase::Vector3()); + } + + alpha = 1.0; + beta = 0.0; +} + +void Charge_Extra::extrapolate_charge( + Parallel_Grid* Pgrid, + UnitCell& ucell, + Charge* chr, + const ModulePW::PW_Basis& rhopw, + Structure_Factor* sf, + std::ofstream& ofs_running, + std::ofstream& ofs_warning, + const AtomicRhoCfg& atomic_rho_cfg, + const bool has_float_data) +{ + ModuleBase::TITLE("Charge_Extra","extrapolate_charge"); + ModuleBase::timer::start("Charge_Extra", "extrapolate_charge"); + //------------------------------------------------------- + // Charge density extrapolation: + // + // * pot_order=0 : copy the old potential (nothing is done); + // * pot_order=1 : subtract old atomic charge density and sum the new + // if dynamics is done the routine extrapolates also the difference + // between the scf charge and the atomic one; + // * pot_order=2 : first order extrapolation: + // \[ \rho(t+dt) = 2\ \rho(t)-\rho(t-dt); \] + // * pot_order=3 : second order extrapolation: + // \[ \rho(t+dt) = \rho(t) + \alpha_0\ (\rho(t) - \rho(t-dt)) + // + \beta_0\ (\rho(t-dt)- \rho(t-2 dt)). \] + // + // The \(\alpha_0\) and \(\beta_0\) parameters are calculated in find_alpha_and_beta() + // so that \(|\tau'-\tau(t+dt)|\) is minimum. \(\tau'\) and \(\tau(t+dt)\) are respectively + // the atomic positions at time t+dt and the extrapolated one: + // \[ \tau(t+dt) = \tau(t) + \alpha_0\ ( \tau(t) - \tau(t-dt) ) + // + \beta_0\ ( \tau(t-dt) - \tau(t-2 dt) ). \] + //------------------------------------------------------- + + rho_extr = std::min(istep, pot_order); + if(rho_extr == 0) + { + sf->setup(&ucell, *Pgrid, &rhopw, has_float_data); + ofs_running << " charge density from previous step !" << std::endl; + ModuleBase::timer::end("Charge_Extra", "extrapolate_charge"); + return; + } + + + // if(lsda || noncolin) rho2zeta(); + + if(rho_extr == 1) + { + ofs_running << " NEW-OLD atomic charge density approx. for the potential !" << std::endl; + +#ifdef _OPENMP +#pragma omp parallel for collapse(2) schedule(static, 128) +#endif + for (int is = 0; is < this->nspin; is++) + { + for (int ir = 0; ir < rhopw.nrxx; ir++) + { + chr->rho[is][ir] = delta_rho1[is][ir]; + } + } + } + // first order extrapolation + else if(rho_extr ==2) + { + ofs_running << " first order charge density extrapolation !" << std::endl; + +#ifdef _OPENMP +#pragma omp parallel for collapse(2) schedule(static, 128) +#endif + for (int is = 0; is < this->nspin; is++) + { + for (int ir = 0; ir < rhopw.nrxx; ir++) + { + chr->rho[is][ir] = 2 * delta_rho1[is][ir] - delta_rho2[is][ir]; + } + } + } + // second order extrapolation + else + { + ofs_running << " second order charge density extrapolation !" << std::endl; + + find_alpha_and_beta(ucell.nat, ofs_running, ofs_warning); + + const double one_add_alpha = 1 + alpha; + const double beta_alpha = beta - alpha; + +#ifdef _OPENMP +#pragma omp parallel for collapse(2) schedule(static, 64) +#endif + for (int is = 0; is < this->nspin; is++) + { + for (int ir = 0; ir < rhopw.nrxx; ir++) + { + chr->rho[is][ir] + = one_add_alpha * delta_rho1[is][ir] + beta_alpha * delta_rho2[is][ir] - beta * delta_rho3[is][ir]; + } + } + } + + sf->setup(&ucell, *Pgrid, &rhopw, has_float_data); + std::vector> rho_atom(this->nspin, + std::vector(rhopw.nrxx)); + std::vector rho_atom_ptr(this->nspin); + for (int is = 0; is < this->nspin; is++) + { + rho_atom_ptr[is] = rho_atom[is].data(); + } + module_charge::atomic_rho(this->nspin, ucell.omega, rho_atom_ptr.data(), + sf->strucFac, ucell, &rhopw, atomic_rho_cfg); +#ifdef _OPENMP +#pragma omp parallel for collapse(2) schedule(static, 512) +#endif + for (int is = 0; is < this->nspin; is++) + { + for(int ir=0; irrho[is][ir] /= ucell.omega; + chr->rho[is][ir] += rho_atom[is][ir]; + } + } + + ModuleBase::timer::end("Charge_Extra", "extrapolate_charge"); + return; +} + +void Charge_Extra::find_alpha_and_beta(const int& natom, std::ofstream& ofs_running, std::ofstream& ofs_warning) +{ + if(istep < 3) + { + return; + } + + double a11 = 0.0; + double a12 = 0.0; + double a21 = 0.0; + double a22 = 0.0; + double b1 = 0.0; + double b2 = 0.0; + double c = 0.0; + double det = 0.0; +#ifdef _OPENMP +#pragma omp parallel for schedule(static, 16) \ + reduction(+:a11) reduction(+:a12) reduction(+:a22) \ + reduction(+:b1) reduction(+:b2) reduction(+:c) +#endif + for(int i=0; i>& get_dis_old1() { return dis_old1; } + std::vector>& get_dis_old2() { return dis_old2; } + std::vector>& get_dis_now() { return dis_now; } + + /// @brief the last three steps' difference of rho and atomic_rho + std::vector>& get_delta_rho1() { return delta_rho1; } + std::vector>& get_delta_rho2() { return delta_rho2; } + std::vector>& get_delta_rho3() { return delta_rho3; } + + /// @brief parameter used in the second order extrapolation + double get_alpha() const { return alpha; } + + /// @brief coefficient for the third delta_rho contribution + double get_beta() const { return beta; } + + /// @brief run find_alpha_and_beta() against the displacement histories + /// currently held, so its solution can be checked on its own + /// rather than only through a full extrapolate_charge() step + void find_alpha_and_beta_for_testing(const int& natom, + std::ofstream& ofs_running, + std::ofstream& ofs_warning) + { + find_alpha_and_beta(natom, ofs_running, ofs_warning); + } + + private: + int istep = 0; ///< the current step + int pot_order; ///< the specified charge extrapolation method + int rho_extr; ///< the actually used method + int nspin; ///< the number of spins + + std::vector> dis_old1; ///< dis_old2 = pos_old1 - pos_old2 + std::vector> dis_old2; ///< dis_old1 = pos_now - pos_old1 + std::vector> dis_now; ///< dis_now = pos_next - pos_now + + std::vector> delta_rho1; ///< the last step difference of rho and atomic_rho + std::vector> delta_rho2; ///< the second last step difference of rho and atomic_rho + std::vector> delta_rho3; ///< the third last step difference of rho and atomic_rho + + double alpha; ///< parameter used in the second order extrapolation + double beta; ///< coefficient for the third delta_rho contribution + + /** + * @brief determine alpha and beta + * + * @param natom the number of atoms + * @param ofs_running the output stream + * @param ofs_warning the output stream + */ + void find_alpha_and_beta(const int& natom, std::ofstream& ofs_running, std::ofstream& ofs_warning); +}; + +} // namespace module_charge + +#endif diff --git a/source/source_estate/module_charge/chg_init.cpp b/source/source_estate/module_charge/chg_init.cpp new file mode 100644 index 00000000000..3960f467950 --- /dev/null +++ b/source/source_estate/module_charge/chg_init.cpp @@ -0,0 +1,431 @@ +#include +#include +#include + +#include "charge.h" +#include "chg_init.h" +#include "chg_atomic.h" +#include "source_base/global_function.h" +#include "source_base/global_variable.h" +#include "source_base/libm/libm.h" +#include "source_base/timer.h" +#include "source_cell/magnetism.h" +#include "source_base/parallel_grid.h" +#include "source_io/module_output/cube_io.h" +#include "chg_rhog_io.h" +#include "source_io/module_wf/read_wf2rho_pw.h" +#include "source_io/module_restart/restart.h" +#include "source_cell/klist.h" +#include "source_base/module_parallel/para_world.h" +#include "source_base/module_parallel/para_tag.h" +#include "source_base/module_parallel/para_bridge.h" +#include "source_base/parallel_common.h" + +// --------------------------------------------------------------------------- +// Thin member wrapper: Charge::init_rho delegates to the free function in +// module_charge so that the charge-init workflow lives outside the class. +// --------------------------------------------------------------------------- +void Charge::init_rho(const UnitCell& ucell, + const Parallel_Grid& pgrid, + const ModuleBase::ComplexMatrix& strucFac, + ModuleSymmetry::Symmetry& symm, + const void* klist, + const void* wfcpw, + const module_charge::InitRhoCfg& cfg) +{ + module_charge::init_rho(*this, *this->rhopw, ucell, pgrid, strucFac, symm, klist, wfcpw, cfg); +} + +namespace module_charge +{ + +namespace +{ + +/// Aggregated file-reading configuration for read_rho_file / read_kin_file +struct ReadCfg +{ + const std::string& suffix; + const std::string& readin_dir; + int rank; + std::ostream& ofs_running; + std::ostream& ofs_warning; +}; + +/** + * @brief Read charge density from restart binary or cube files into chr.rho. + * + * Charge members accessed: chr.rhog, chr.rho, chr.nspin. + * + * @param chr [inout] Charge object supplying the rho/rhog buffers + * @param rhopw [in] plane-wave basis for file decoding and Fourier transforms + * @param cfg [in] file-reading configuration (suffix, dir, rank, logs) + * @return true if a density source was read; false if neither the restart + * binary nor the cube file exists (caller decides whether to fall + * back to the atomic density) + */ +bool read_rho_file(Charge& chr, + const ModulePW::PW_Basis& rhopw, + const UnitCell& ucell, + const Parallel_Grid& pgrid, + const ReadCfg& cfg) +{ + const int nspin = chr.nspin; + std::complex** const rhog = chr.rhog; + double** const rho = chr.rho; + const std::string& suffix = cfg.suffix; + const std::string& readin_dir = cfg.readin_dir; + const int rank = cfg.rank; + std::ostream& ofs_running = cfg.ofs_running; + std::ostream& ofs_warning = cfg.ofs_warning; + + ofs_running << " Read electron density from file" << std::endl; + + // try to read charge from binary file first, which is the same as QE + // liuyu 2023-12-05 + std::stringstream binary; + binary << readin_dir << suffix + "-CHARGE-DENSITY.restart"; + // Temporary bridge: use factory until ParaCollection is wired into driver. + Parallel::ParaWorld pw_world = Parallel::make_pw_world(); + if (module_charge::read_rhog(binary.str(), &rhopw, nspin, rhog, pw_world, &ofs_warning)) + { + ofs_running << " Read electron density from file: " << binary.str() << std::endl; + for (int is = 0; is < nspin; ++is) + { + rhopw.recip2real(rhog[is], rho[is]); + } + return true; + } + + // restart binary is absent; fall back to cube files. + // Probe the cube file on the rank that parses it and broadcast the result, + // so every rank agrees to skip the read instead of hanging in pgrid.bcast(). + std::stringstream ssc0; + if (nspin == 1) + { + ssc0 << readin_dir << "chg.cube"; + } + else + { + ssc0 << readin_dir << "chgs1.cube"; + } + bool cube_exists = false; + if (rank == 0) + { + cube_exists = std::ifstream(ssc0.str()).good(); + } + Parallel_Common::bcast_bool(cube_exists); + if (!cube_exists) + { + return false; + } + + for (int is = 0; is < nspin; ++is) + { + std::stringstream ssc; + + if (nspin == 1) + { + ssc << readin_dir << "chg.cube"; + } + else + { + ssc << readin_dir << "chgs" << is + 1 << ".cube"; + } + + ModuleIO::read_vdata_palgrid(pgrid, + rank, + ofs_running, + ssc.str(), + rho[is], + ucell.nat); + ofs_running << " Read electron density from file: " << ssc.str() << std::endl; + } + return true; +} + +/** + * @brief Read kinetic-energy density from restart binary or cube files. + * + * Charge members accessed: chr.kin_r, chr.nspin. + * + * @param chr [inout] Charge object supplying the kin_r buffer + * @param rhopw [in] plane-wave basis for file decoding and Fourier transforms + * @param cfg [in] file-reading configuration (suffix, dir, rank, logs) + * @return true if a tau source was read; false if neither the restart binary + * nor the SPINX_TAU.cube file exists (caller falls back to the + * Thomas-Fermi tau init) + */ +bool read_kin_file(Charge& chr, + const ModulePW::PW_Basis& rhopw, + const UnitCell& ucell, + const Parallel_Grid& pgrid, + const ReadCfg& cfg) +{ + const int nspin = chr.nspin; + double** const kin_r = chr.kin_r; + const std::string& suffix = cfg.suffix; + const std::string& readin_dir = cfg.readin_dir; + const int rank = cfg.rank; + std::ostream& ofs_running = cfg.ofs_running; + std::ostream& ofs_warning = cfg.ofs_warning; + + ofs_running << " try to read kinetic energy density from file" << std::endl; + std::vector> kin_g_space(nspin * rhopw.npw, {0.0, 0.0}); + std::vector*> kin_g; + for (int is = 0; is < nspin; is++) + { + kin_g.push_back(kin_g_space.data() + is * rhopw.npw); + } + + Parallel::ParaWorld pw_world = Parallel::make_pw_world(); + std::stringstream binary; + binary << readin_dir << suffix + "-TAU-DENSITY.restart"; + if (module_charge::read_rhog(binary.str(), &rhopw, nspin, kin_g.data(), pw_world, &ofs_warning)) + { + ofs_running << " Read in the kinetic energy density: " << binary.str() << std::endl; + for (int is = 0; is < nspin; ++is) + { + rhopw.recip2real(kin_g[is], kin_r[is]); + } + return true; + } + + // restart binary is absent; fall back to SPINX_TAU.cube files, probing on + // the parsing rank and broadcasting so all ranks agree to skip together. + std::stringstream ssc0; + ssc0 << readin_dir << "SPIN1_TAU.cube"; + bool cube_exists = false; + if (rank == 0) + { + cube_exists = std::ifstream(ssc0.str()).good(); + } + Parallel_Common::bcast_bool(cube_exists); + if (!cube_exists) + { + return false; + } + + for (int is = 0; is < nspin; is++) + { + std::stringstream ssc; + ssc << readin_dir << "SPIN" << is + 1 << "_TAU.cube"; + // mohan update 2012-02-10, sunliang update 2023-03-09 + ModuleIO::read_vdata_palgrid(pgrid, + rank, + ofs_running, + ssc.str(), + kin_r[is], + ucell.nat); + ofs_running << " Read in the kinetic energy density: " << ssc.str() << std::endl; + } + return true; +} + +/** + * @brief Atomic-density fallback plus Thomas-Fermi kinetic-energy-density init. + * + * Charge members accessed: chr.rho, chr.kin_r, chr.nspin. + * + * @param chr [inout] Charge object supplying rho/kin_r buffers + * @param rhopw [in] plane-wave basis for atomic superposition and grid size + * @param omega [in] unit-cell volume + * @param init_chg [in] INPUT.init_chg + * @param meta_gga [in] whether the functional is meta-GGA (tau TF-init needed) + * @param read_error [in] rho file reading failed (fall back to atomic) + * @param read_kin_error [in] tau file reading failed (fall back to TF tau) + * @param atomic_rho_cfg [in] atomic-density configuration + */ +void init_rho_atomic_and_tau(Charge& chr, + const ModulePW::PW_Basis& rhopw, + const UnitCell& ucell, + const ModuleBase::ComplexMatrix& strucFac, + const double& omega, + const std::string& init_chg, + const bool meta_gga, + const bool read_error, + const bool read_kin_error, + const AtomicRhoCfg& atomic_rho_cfg) +{ + const int nspin = chr.nspin; + + if (init_chg == "atomic" || read_error) + { + if (read_error) + { + std::cout << " Charge::init_rho: use atomic initialization instead." << std::endl; + } + module_charge::atomic_rho(nspin, omega, chr.rho, strucFac, ucell, &rhopw, atomic_rho_cfg); + } + + // initial tau = 3/5 rho^2/3, Thomas-Fermi + if (meta_gga) + { + if (init_chg == "atomic" || read_kin_error) + { + if (read_kin_error) + { + std::cout << " Charge::init_rho: init kinetic energy density from rho." << std::endl; + } + const double fact = (3.0 / 5.0) * pow(3.0 * ModuleBase::PI * ModuleBase::PI, 2.0 / 3.0); + for (int is = 0; is < nspin; ++is) + { + for (int ir = 0; ir < rhopw.nrxx; ++ir) + { + chr.kin_r[is][ir] = fact * pow(std::abs(chr.rho[is][ir]) * nspin, 5.0 / 3.0) / nspin; + } + } + } + } +} + +/** + * @brief Load charge density from the restart disk cache if requested. + * + * Charge members accessed: chr.rho, chr.nspin. + * + * @param chr [inout] Charge object supplying rho buffer + * @param rhopw [in] plane-wave basis supplying the local real-space grid size + * @param restart [inout] restart manager + * @param readin_dir [in] fallback cube-file directory + * @param rank [in] this processor's rank for palgrid reads + * @param ofs_running [inout] running log stream + */ +void load_rho_from_restart(Charge& chr, + const ModulePW::PW_Basis& rhopw, + const UnitCell& ucell, + const Parallel_Grid& pgrid, + Restart& restart, + const std::string& readin_dir, + const int rank, + std::ostream& ofs_running) +{ + const int nspin = chr.nspin; + + // Peize Lin add 2020.04.04 + if (restart.info_load.load_charge && !restart.info_load.load_charge_finish) + { + for (int is = 0; is < nspin; ++is) + { + try + { + restart.load_disk("charge", is, rhopw.nrxx, chr.rho[is]); + } + catch (const std::exception& e) + { + // try to load from the output of `out_chg` + std::stringstream ssc; + ssc << readin_dir << "chgs" << is + 1 << ".cube"; + ModuleIO::read_vdata_palgrid(pgrid, + rank, + ofs_running, + ssc.str(), + chr.rho[is], + ucell.nat); + ofs_running << " Read in electron density: " << ssc.str() << std::endl; + } + } + restart.info_load.load_charge_finish = true; + } +} + +} // anonymous namespace + +// --------------------------------------------------------------------------- +// Public orchestrator: decides which initialization path(s) to run based on +// INPUT.init_chg and dispatches to the stage helpers above. +// --------------------------------------------------------------------------- +void init_rho(Charge& chr, + const ModulePW::PW_Basis& rhopw, + const UnitCell& ucell, + const Parallel_Grid& pgrid, + const ModuleBase::ComplexMatrix& strucFac, + ModuleSymmetry::Symmetry& symm, + const void* klist, + const void* wfcpw, + const InitRhoCfg& cfg) +{ + ModuleBase::GlobalFunc::OUT(GlobalV::ofs_running, "init_chg", cfg.init_chg); + + const int nspin = chr.nspin; + assert(nspin > 0); + + std::string init_chg_upper = cfg.init_chg; + std::transform(init_chg_upper.begin(), init_chg_upper.end(), init_chg_upper.begin(), ::toupper); + std::cout << " START CHARGE : " << init_chg_upper << std::endl; + + chr.pgrid = &pgrid; + + const std::string& init_chg = cfg.init_chg; + const std::string& suffix = cfg.suffix; + const std::string& readin_dir = cfg.global_readin_dir; + const int rank = (cfg.esolver_type == "sdft" ? GlobalV::RANK_IN_BPGROUP : GlobalV::MY_RANK); + + // Capture before the local ReadCfg (also named cfg) shadows the argument. + const bool meta_gga = cfg.meta_gga; + bool read_error = false; + bool read_kin_error = false; + if (init_chg == "file" || init_chg == "auto") + { + ReadCfg cfg{suffix, readin_dir, rank, + GlobalV::ofs_running, GlobalV::ofs_warning}; + read_error = !read_rho_file(chr, rhopw, ucell, pgrid, cfg); + + if (read_error) + { + if (init_chg == "file") + { + ModuleBase::WARNING_QUIT("Charge::init_rho", + "Failed to read in charge density from file.\n For initializing atomic " + "charge in calculations,\n please set init_chg to atomic in INPUT."); + } + // init_chg == "auto": no density file present, fall back to atomic. + } + + // If the charge density is not read in, the tau file is not read either. + if (meta_gga) + { + if (!read_error) + { + read_kin_error = !read_kin_file(chr, rhopw, ucell, pgrid, cfg); + } + else + { + read_kin_error = true; + } + } + } + + const AtomicRhoCfg atomic_rho_cfg{ + cfg.nelec, + cfg.test_charge, + cfg.domag, + cfg.domag_z, + GlobalV::ofs_warning}; + init_rho_atomic_and_tau(chr, rhopw, ucell, strucFac, ucell.omega, + init_chg, cfg.meta_gga, read_error, read_kin_error, + atomic_rho_cfg); + + load_rho_from_restart(chr, rhopw, ucell, pgrid, GlobalC::restart, + readin_dir, rank, GlobalV::ofs_running); + + if (init_chg == "wfc") + { + if (wfcpw == nullptr) + { + ModuleBase::WARNING_QUIT("Charge::init_rho", "wfc is only supported for PW-KSDFT."); + } + + const ModulePW::PW_Basis_K* pw_wfc = reinterpret_cast(const_cast(wfcpw)); + const K_Vectors* kv = reinterpret_cast(klist); + + ModuleIO::read_wf2rho_pw(pw_wfc, symm, chr, + readin_dir, + GlobalV::KPAR, GlobalV::MY_POOL, GlobalV::MY_RANK, + GlobalV::NPROC_IN_POOL, GlobalV::RANK_IN_POOL, + cfg.nbands, nspin, cfg.npol, + kv->get_nkstot(), kv->ik2iktot, kv->isk, GlobalV::ofs_running); + } +} + +} // namespace module_charge diff --git a/source/source_estate/module_charge/chg_init.h b/source/source_estate/module_charge/chg_init.h new file mode 100644 index 00000000000..6af0fb880fb --- /dev/null +++ b/source/source_estate/module_charge/chg_init.h @@ -0,0 +1,61 @@ +#ifndef CHG_INIT_H +#define CHG_INIT_H + +#include "source_base/complexmatrix.h" +#include "source_base/parallel_grid.h" +#include "source_basis/module_pw/pw_basis.h" +#include "source_cell/module_symmetry/symmetry.h" + +#include + +class Charge; +class UnitCell; + +namespace module_charge +{ + +/// Configuration for charge-density initialization, replacing direct +/// PARAM reads in init_rho. Callers fill this from the parsed input +/// once per run. +struct InitRhoCfg +{ + std::string init_chg; ///< charge initialization mode (PARAM.inp.init_chg) + std::string suffix; ///< system suffix for file names (PARAM.inp.suffix) + std::string esolver_type; ///< esolver type; "sdft" selects the band-group rank (PARAM.inp.esolver_type) + std::string global_readin_dir; ///< directory to read files from (PARAM.globalv.global_readin_dir) + double nelec = 0.0; ///< target total electron number (PARAM.inp.nelec) + int nbands = 0; ///< number of bands for wfc-based init (PARAM.inp.nbands) + int test_charge = 0; ///< verbosity flag (PARAM.inp.test_charge) + bool domag = false; ///< whether to compute magnetization (PARAM.globalv.domag) + bool domag_z = false; ///< whether to compute z-only magnetization (PARAM.globalv.domag_z) + bool meta_gga = false; ///< whether the functional is meta-GGA (XC_Functional::get_ked_flag()) + int npol = 1; ///< number of polarization components (PARAM.globalv.npol) +}; + +/** + * @brief Initialize charge density from file, atomic superposition, + * restart cache, or wavefunctions, depending on cfg.init_chg. + * + * @param chr [inout] Charge object whose rho/rhog/kin_r buffers are filled. + * @param rhopw [in] plane-wave basis bound to chr (grid sizes and FFT backend) + * @param ucell [in] unit cell + * @param pgrid [in] parallel grid descriptor + * @param strucFac [in] structure factor for atomic-charge superposition + * @param symm [in] symmetry operations (used by wfc-based init) + * @param klist [in] k-point list pointer (K_Vectors*), needed only for wfc init + * @param wfcpw [in] PW_Basis_K pointer, needed only for wfc init + * @param cfg [in] INPUT values for charge initialization + */ +void init_rho(Charge& chr, + const ModulePW::PW_Basis& rhopw, + const UnitCell& ucell, + const Parallel_Grid& pgrid, + const ModuleBase::ComplexMatrix& strucFac, + ModuleSymmetry::Symmetry& symm, + const void* klist, + const void* wfcpw, + const InitRhoCfg& cfg); + +} // namespace module_charge + +#endif // CHG_INIT_H diff --git a/source/source_estate/module_charge/chg_mix.cpp b/source/source_estate/module_charge/chg_mix.cpp new file mode 100644 index 00000000000..89a2f94abc7 --- /dev/null +++ b/source/source_estate/module_charge/chg_mix.cpp @@ -0,0 +1,300 @@ +#include "chg_mix.h" +#include "chg_drho.h" +#include "chg_precond.h" +#include "chg_rho_detail.h" +#include "chg_tau.h" +#include "chg_uspp.h" + +#include +#include + +#include "source_base/module_mixing/broyden_mixing.h" +#include "source_base/module_mixing/pulay_mixing.h" +#include "source_base/parallel_common.h" +#include "source_base/timer.h" +#include "source_base/tool_quit.h" + +Charge_Mixing::Charge_Mixing() +{ + // unique_ptr members default-construct to nullptr +} + +Charge_Mixing::~Charge_Mixing() +{ + // unique_ptr members (mixing, mixing_highf) are released automatically +} + +void Charge_Mixing::set_mixing(const MixingConfig& cfg, + ModulePW::PW_Basis* rhopw_in, + ModulePW::PW_Basis* rhodpw_in, + double& omega_in, + double& tpiba_in) +{ + // store the aggregated config; init_mixing/mix_rho and the stateless + // Kerker kernels all read nspin, scf_thr_type, double_grid, mixing_gg0, + // mixing_gg0_mag, mixing_gg0_min, mixing_angle, mixing_dmr from cfg_ + // instead of PARAM/GlobalV. cfg_ is treated as an immutable INPUT + // snapshot; runtime overrides (e.g. close_kerker_gg0) live as flags on + // Charge_Mixing itself, never by mutating cfg_. + this->cfg_ = cfg; + // store the smooth and dense grids + this->rhopw = rhopw_in; + this->rhodpw = rhodpw_in; + // omega and tpiba are pointers to external runtime state (cell volume + // and lattice constant) that changes across SCF iterations; they are + // not INPUT parameters and therefore stay out of MixingConfig. + this->omega = &omega_in; + this->tpiba = &tpiba_in; + // check the paramters + if (this->cfg_.mixing_beta > 1.0 || this->cfg_.mixing_beta < 0.0) + { + ModuleBase::WARNING_QUIT("Charge_Mixing", "You'd better set mixing_beta to [0.0, 1.0]!"); + } + if (this->cfg_.nspin >= 2 && this->cfg_.mixing_beta_mag < 0.0) + { + ModuleBase::WARNING_QUIT("Charge_Mixing", "You'd better set mixing_beta_mag >= 0.0!"); + } + + if (!(this->cfg_.mixing_mode == "plain" || this->cfg_.mixing_mode == "broyden" || this->cfg_.mixing_mode == "pulay")) + { + ModuleBase::WARNING_QUIT("Charge_Mixing", "This Mixing mode is not implemended yet,coming soon."); + } + + // print into running.log + //GlobalV::ofs_running << "\n\n"; + GlobalV::ofs_running << "\n"; + GlobalV::ofs_running << " >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>" + ">>>>" << std::endl; + GlobalV::ofs_running << " | " + " |" << std::endl; + GlobalV::ofs_running << " | Setup charge mixing parameters " + " |" << std::endl; + GlobalV::ofs_running << " | " + " |" << std::endl; + GlobalV::ofs_running << " <<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<" + "<<<<" << std::endl; + GlobalV::ofs_running << "\n"; + + + ModuleBase::GlobalFunc::OUT(GlobalV::ofs_running, "mixing_type", this->cfg_.mixing_mode); + ModuleBase::GlobalFunc::OUT(GlobalV::ofs_running, "mixing_beta", this->cfg_.mixing_beta); + ModuleBase::GlobalFunc::OUT(GlobalV::ofs_running, "mixing_gg0", cfg_.mixing_gg0); + ModuleBase::GlobalFunc::OUT(GlobalV::ofs_running, "mixing_gg0_min", cfg_.mixing_gg0_min); + + if (this->cfg_.nspin==2 || this->cfg_.nspin==4) + { + ModuleBase::GlobalFunc::OUT(GlobalV::ofs_running, "mixing_beta_mag", this->cfg_.mixing_beta_mag); + ModuleBase::GlobalFunc::OUT(GlobalV::ofs_running, "mixing_gg0_mag", cfg_.mixing_gg0_mag); + } + if (cfg_.mixing_angle > 0) + { + ModuleBase::GlobalFunc::OUT(GlobalV::ofs_running, "mixing_angle", cfg_.mixing_angle); + } + + ModuleBase::GlobalFunc::OUT(GlobalV::ofs_running, "mixing_ndim", this->cfg_.mixing_ndim); + + return; +} + +void Charge_Mixing::init_mixing() +{ + // this init should be called at the 1-st iteration of each scf loop + + ModuleBase::TITLE("Charge_Mixing", "init_mixing"); + ModuleBase::timer::start("Charge_Mixing", "init_mixing"); + + /// Fail fast when set_mixing was skipped: the grid sizes below would + /// otherwise dereference a null pointer. + if (this->rhopw == nullptr) + { + ModuleBase::WARNING_QUIT("Charge_Mixing", + "set_mixing must be called before init_mixing"); + } + + // (re)construct mixing object + if (this->cfg_.mixing_mode == "broyden") + { + this->mixing = std::unique_ptr( + new Base_Mixing::Broyden_Mixing(this->cfg_.mixing_ndim, this->cfg_.mixing_beta)); + } + else if (this->cfg_.mixing_mode == "plain") + { + this->mixing = std::unique_ptr( + new Base_Mixing::Plain_Mixing(this->cfg_.mixing_beta)); + } + else if (this->cfg_.mixing_mode == "pulay") + { + this->mixing = std::unique_ptr( + new Base_Mixing::Pulay_Mixing(this->cfg_.mixing_ndim, this->cfg_.mixing_beta)); + } + else + { + ModuleBase::WARNING_QUIT("Charge_Mixing", "This Mixing mode is not implemended yet,coming soon."); + } + + if ( this->cfg_.double_grid) + { + // ONLY smooth part of charge density is mixed by specific mixing method + // The high_frequency part is mixed by plain mixing method. + this->mixing_highf = std::unique_ptr( + new Base_Mixing::Plain_Mixing(this->cfg_.mixing_beta)); + } + + // allocate memory for mixing data, if exists, free it first and then allocate new memory + // initailize rho_mdata + if (this->cfg_.scf_thr_type == 1) + { + if (this->cfg_.nspin == 4 && this->cfg_.mixing_angle > 0 ) + { + this->mixing->init_mixing_data(this->rho_mdata, + this->rhopw->npw * 2, + sizeof(std::complex)); + } + else + { + this->mixing->init_mixing_data(this->rho_mdata, + this->rhopw->npw * this->cfg_.nspin, + sizeof(std::complex)); + } + } + else + { + if (this->cfg_.nspin == 4 && this->cfg_.mixing_angle > 0 ) + { + this->mixing->init_mixing_data(this->rho_mdata, this->rhopw->nrxx * 2, sizeof(double)); + } + else + { + this->mixing->init_mixing_data(this->rho_mdata, this->rhopw->nrxx * this->cfg_.nspin, sizeof(double)); + } + } + + // initailize tau_mdata + if (cfg_.mixing_tau) + { + if (this->cfg_.scf_thr_type == 1) + { + this->mixing->init_mixing_data(this->tau_mdata, + this->rhopw->npw * this->cfg_.nspin, + sizeof(std::complex)); + } + else + { + this->mixing->init_mixing_data(this->tau_mdata, this->rhopw->nrxx * this->cfg_.nspin, sizeof(double)); + } + } + + ModuleBase::timer::end("Charge_Mixing", "init_mixing"); + + return; +} + +void Charge_Mixing::mix_reset() +{ + this->mixing->reset(); + this->rho_mdata.reset(); + // initailize tau_mdata + if (cfg_.mixing_tau) + { + this->tau_mdata.reset(); + } +} + +bool Charge_Mixing::if_scf_oscillate(const int iteration, const double drho, + const int iternum_used, const double threshold) +{ + ModuleBase::TITLE("Charge_Mixing", "if_scf_oscillate"); + + /// Fail fast when set_mixing was skipped: cfg_.scf_nmax is otherwise an + /// indeterminate value, which would resize the history to zero and then + /// write out of bounds (heap corruption instead of a clear error). + if (this->cfg_.scf_nmax <= 0) + { + ModuleBase::WARNING_QUIT("Charge_Mixing", + "set_mixing must be called before if_scf_oscillate"); + } + if (iteration < 1 || iteration > this->cfg_.scf_nmax) + { + ModuleBase::WARNING_QUIT("Charge_Mixing", + "iteration must be within [1, scf_nmax]"); + } + + if(this->_drho_history.size() == 0) + { + this->_drho_history.resize(this->cfg_.scf_nmax); + } + + // add drho into history + this->_drho_history[iteration - 1] = drho; + + if(threshold >= 0) // close the function + { + return false; + } + + // check if the history is long enough + if(iteration < iternum_used + this->mixing_restart_last) + { + return false; + } + + // calculate the slope of the last iternum_used iterations' drho + double slope = 0.0; + + // Least Squares Method + // this part is too short, so I do not design it as a free function in principle + double sumX = 0, sumY = 0, sumXY = 0, sumXX = 0; + for (int i = iteration - iternum_used; i < iteration; i++) + { + sumX += i; + sumY += std::log10(this->_drho_history[i]); + sumXY += i * std::log10(this->_drho_history[i]); + sumXX += i * i; + } + double numerator = iternum_used * sumXY - sumX * sumY; + double denominator = iternum_used * sumXX - sumX * sumX; + if (denominator == 0) { + return false; + } + slope = numerator / denominator; + + // if the slope is less than the threshold, return true + if(slope > threshold) + { + return true; + } + + return false; +} + +void Charge_Mixing::allocate_mixing_uom(int uom_size) +{ + ModuleBase::TITLE("Charge_Mixing", "allocate_mixing_uom"); + ModuleBase::timer::start("Charge_Mixing", "allocate_mixing_uom"); + // For nspin=2, uom_size already includes both spin channels + // (uterm_mat.size() = pot_index * 2 for nspin=2) + // So uom_fold should always be 1 + this->mixing->init_mixing_data(this->uom_mdata, uom_size, sizeof(double)); + this->uom_mdata.reset(); + ModuleBase::timer::end("Charge_Mixing", "allocate_mixing_uom"); + return; +} + +void Charge_Mixing::mix_uom(std::vector& uom_in, std::vector& uom_save_in) +{ + ModuleBase::TITLE("Charge_Mixing", "mix_uom"); + ModuleBase::timer::start("Charge_Mixing", "mix_uom"); + double* uom_value_out = uom_in.data(); + double* uom_value_in = uom_save_in.data(); + // For all nspin cases, uom_array layout is already fully sized + // and mixing operates on the entire array + this->mixing->push_data(this->uom_mdata, uom_value_in, uom_value_out, nullptr, false); + this->mixing->mix_data(this->uom_mdata, uom_value_out); + ModuleBase::timer::end("Charge_Mixing", "mix_uom"); +#ifdef __MPI + // Synchronize mixed uom across all ranks to prevent divergence + // after multiple Pulay steps (same pattern as mix_dmr) + Parallel_Common::bcast_double(uom_in.data(), uom_in.size()); +#endif + return; +} diff --git a/source/source_estate/module_charge/chg_mix.h b/source/source_estate/module_charge/chg_mix.h new file mode 100644 index 00000000000..b52d9b96f82 --- /dev/null +++ b/source/source_estate/module_charge/chg_mix.h @@ -0,0 +1,177 @@ +#ifndef CHG_MIX_H +#define CHG_MIX_H +#include "charge.h" +#include "chg_mix_cfg.h" +#include "source_base/module_mixing/mixing.h" +#include "source_base/module_mixing/plain_mixing.h" +#include +#include + +class Charge_Mixing +{ + /// Charge_Mixing class + /// This class is used to mix charge density, kinetic energy density and real-space density matrix + /// This Charge_Mixing class offers the following interfaces: + /// 1. set_mixing() to set all private mixing parameters + /// 2. init_mixing() to initialize mixing, including allocating memory for mixing data and reset mixing + /// 3. mix_rho() to mix charge density + /// Real-space density matrix mixing is implemented by the stateless + /// module_charge::init_mixing_dmr/mix_dmr functions in chg_dmr.h; this class + /// only owns the mixing history buffer, exposed through get_dmr_mdata(). + /// how to use it: + /// you can (re)start a mixing by calling set_mixing() and init_mixing() before calling mix_rho() + + public: + Charge_Mixing(); + ~Charge_Mixing(); + + /** + * @brief Set all private mixing parameters from an aggregated config + * @param cfg mixing parameters and runtime globals (nspin, scf_thr_type, double_grid) + * @param rhopw_in smooth grid + * @param rhodpw_in dense grid when double grid is used, otherwise same as rhopw + * @param omega_in omega for non-linear core correction + * @param tpiba_in 2*pi/beta for non-linear core correction + */ + void set_mixing(const MixingConfig& cfg, + ModulePW::PW_Basis* rhopw_in, + ModulePW::PW_Basis* rhodpw_in, + double& omega_in, + double& tpiba_in); + + /// Disable Kerker screening for subsequent mix_rho calls. + /// Used by the non-separate-loop EXX path (exx_lri_interface.hpp) + /// after EXX convergence: Kerker damping fights the DM update there. + /// The Kerker kernels read cfg_ (immutable INPUT snapshot), so the + /// disable flag must live on Charge_Mixing itself rather than mutating cfg_. + void close_kerker_gg0() { kerker_disabled_ = true; } + /** + * @brief initialize mixing, including constructing mixing and allocating memory for mixing data + * @brief this function should be called at eachiterinit() + */ + void init_mixing(); + + /** + * @brief charge mixing + * @param chr pointer of Charge object + */ + void mix_rho(Charge* chr); + + /** + * @brief allocate memory of uom_mdata + * @param uom_size size of DFT+U occupation matrix + */ + void allocate_mixing_uom(int size_uom); + + /** + * @brief DFT+U occupation matrix mixing + * @param uom_in output occupation matrix + * @param uom_save_in input occupation matrix + */ + void mix_uom(std::vector& uom_in, std::vector& uom_save_in); + + /** + * @brief reset mixing, actually we only call init_mixing() to reset mixing instead of this function + */ + void mix_reset(); + + // extracting parameters normally these parameters will not be used outside charge mixing + // while Exx is using them as well as some other places + const std::string& get_mixing_mode() const {return cfg_.mixing_mode;} + double get_mixing_beta() const {return cfg_.mixing_beta;} + int get_mixing_ndim() const {return cfg_.mixing_ndim;} + Base_Mixing::Mixing* get_mixing() const {return mixing.get();} + + /** + * @brief mutable access to the real-space density-matrix mixing history + * + * The history buffer is owned by Charge_Mixing and driven by the + * stateless module_charge::init_mixing_dmr/mix_dmr functions in chg_dmr.h. + */ + Base_Mixing::Mixing_Data& get_dmr_mdata() {return dmr_mdata;} + + /** + * @brief mutable access to the charge-density mixing history + * + * Same ownership story as get_dmr_mdata(): the buffer belongs to + * Charge_Mixing but is driven through Base_Mixing::Mixing, so callers + * that reset or inspect the history need a handle on it. + */ + Base_Mixing::Mixing_Data& get_rho_mdata() {return rho_mdata;} + + /** + * @brief mutable access to the kinetic-energy-density mixing history + */ + Base_Mixing::Mixing_Data& get_tau_mdata() {return tau_mdata;} + + /** + * @brief read-only access to the aggregated mixing config set by set_mixing() + */ + const MixingConfig& get_mixing_config() const {return cfg_;} + + /** + * @brief replace the aggregated mixing config without re-running set_mixing() + * + * set_mixing() both stores the config and rebuilds the mixing objects. This + * setter is for callers that need to update the snapshot alone - typically + * because a derived runtime global (nspin, domag, ...) changed after the + * mixing objects were already built - and must not disturb the mixing + * history. Pair it with get_mixing_config() to amend individual fields. + */ + void set_mixing_config(const MixingConfig& cfg) {cfg_ = cfg;} + + // for mixing restart + /// which step to restart mixing during SCF + int mixing_restart_step = 0; + /// the number of restart mixing during SCF + int mixing_restart_count = 0; + /// the label of mixing restart step + int mixing_restart_last = 0; + + // to calculate the slope of drho curve during SCF, which is used to determine if SCF oscillate + bool if_scf_oscillate(const int iteration, const double drho, + const int iternum_used, const double threshold); + + private: + + // mixing_data + /// Mixing object for charge and kinetic energy + std::unique_ptr mixing; + Base_Mixing::Mixing_Data rho_mdata; ///< Mixing data for charge density + Base_Mixing::Mixing_Data tau_mdata; ///< Mixing data for kinetic energy density + Base_Mixing::Mixing_Data dmr_mdata; ///< Mixing data for real space density matrix + Base_Mixing::Mixing_Data uom_mdata; ///< Mixing data for DFT+U occupation matrix + std::unique_ptr mixing_highf; ///< The high_frequency part is mixed by plain mixing method. + + //====================================== + // private mixing parameters + //====================================== + MixingConfig cfg_; ///< aggregated mixing config, also holds nspin/scf_thr_type/double_grid + double* omega = nullptr; ///< omega for non-linear core correction + double* tpiba = nullptr; ///< 2*pi/beta for non-linear core correction + std::vector _drho_history; ///< history of drho used to determine the oscillation, size is scf_nmax + + ModulePW::PW_Basis* rhopw = nullptr; ///< smooth grid + ModulePW::PW_Basis* rhodpw = nullptr; ///< dense grid, same as rhopw for ncpp. + + /// Runtime override set by close_kerker_gg0(): short-circuits the + /// Kerker screening lambdas in mix_rho_recip/mix_rho_real so the + /// non-separate-loop EXX path can disable Kerker after convergence. + /// Lives here, not in MixingConfig, because cfg_ is an immutable + /// INPUT snapshot consumed by the stateless Kerker kernels. + bool kerker_disabled_ = false; + + /** + * @brief charge mixing for reciprocal space + * @param chr pointer of Charge object + */ + void mix_rho_recip(Charge* chr); + + /** + * @brief charge mixing for real space + * @param chr pointer of Charge object + */ + void mix_rho_real(Charge* chr); +}; + +#endif diff --git a/source/source_estate/module_charge/chg_mix_cfg.h b/source/source_estate/module_charge/chg_mix_cfg.h new file mode 100644 index 00000000000..3ec6f3a0b09 --- /dev/null +++ b/source/source_estate/module_charge/chg_mix_cfg.h @@ -0,0 +1,32 @@ +#ifndef CHG_MIX_CFG_H +#define CHG_MIX_CFG_H + +#include + +/// Configuration for charge mixing, aggregating the INPUT mixing parameters +/// together with the runtime globals (nspin, scf_thr_type, double_grid, +/// gamma_only_pw, domag, domag_z) that the mixing logic needs, so that +/// Charge_Mixing does not read PARAM/GlobalV directly. Callers fill this +/// from the parsed input once per run. +struct MixingConfig +{ + std::string mixing_mode; ///< mixing mode: "plain", "broyden", "pulay" + double mixing_beta; ///< mixing beta for density + int mixing_ndim; ///< mixing ndim for broyden and pulay + double mixing_gg0; ///< mixing gg0 for Kerker screen + bool mixing_tau; ///< whether to use tau mixing + double mixing_beta_mag; ///< mixing beta for magnetism + double mixing_gg0_mag; ///< mixing gg0 for Kerker screen for magnetism + double mixing_gg0_min; ///< minimum kerker coefficient + double mixing_angle; ///< mixing angle for nspin=4 + bool mixing_dmr; ///< whether to mix real space density matrix + int nspin; ///< number of spins + int scf_thr_type; ///< 1: reciprocal, 2: real space threshold + bool double_grid; ///< whether double grid is used + bool gamma_only_pw; ///< whether gamma-only plane wave is used + bool domag; ///< whether magnetism (non-collinear) is considered + bool domag_z; ///< whether only the z-component magnetism is considered + int scf_nmax; ///< max SCF iterations, sizes the drho oscillation history (PARAM.inp.scf_nmax) +}; + +#endif // CHG_MIX_CFG_H diff --git a/source/source_estate/module_charge/chg_mix_rho.cpp b/source/source_estate/module_charge/chg_mix_rho.cpp new file mode 100644 index 00000000000..9ee17da8b86 --- /dev/null +++ b/source/source_estate/module_charge/chg_mix_rho.cpp @@ -0,0 +1,482 @@ +#include "chg_mix.h" +#include "chg_drho.h" +#include "chg_precond.h" +#include "chg_rho_detail.h" +#include "chg_tau.h" +#include "chg_uspp.h" + +#include + +#include "source_base/parallel_common.h" +#include "source_base/timer.h" +#include "source_base/tool_quit.h" + +void Charge_Mixing::mix_rho_recip(Charge* chr) +{ + ModuleBase::TITLE("Charge_Mixing", "mix_rho_recip"); + ModuleBase::timer::start("Charge_Mixing", "mix_rho_recip"); + + const int nspin = cfg_.nspin; + assert(nspin==1 || nspin==2 || nspin==4); + + std::complex* rhog_in = nullptr; + std::complex* rhog_out = nullptr; + // RAII owners for the smooth / high-frequency parts on the double grid. + // The raw pointers below alias these vectors when double_grid is on, + // or alias chr->rhog[_save][0] directly when double_grid is off so the + // mixing still mutates chr in place. + std::vector> rho_sg_in; + std::vector> rho_sg_out; + std::vector> rho_hf_in; + std::vector> rho_hf_out; + // for smooth part + std::complex* rhogs_in = nullptr; + std::complex* rhogs_out = nullptr; + // for high_frequency part + std::complex* rhoghf_in = nullptr; + std::complex* rhoghf_out = nullptr; + + if ( cfg_.double_grid) + { + // divide into smooth part and high_frequency part + const int npw_smooth = this->rhopw->npw; + const int npw_dense = this->rhodpw->npw; + rho_sg_in.resize(nspin * npw_smooth); + rho_hf_in.resize(nspin * (npw_dense - npw_smooth)); + rho_sg_out.resize(nspin * npw_smooth); + rho_hf_out.resize(nspin * (npw_dense - npw_smooth)); + module_charge::split_dgrid(chr->rhog_save[0], rho_sg_in, rho_hf_in, + nspin, npw_smooth, npw_dense); + module_charge::split_dgrid(chr->rhog[0], rho_sg_out, rho_hf_out, + nspin, npw_smooth, npw_dense); + rhogs_in = rho_sg_in.data(); + rhoghf_in = rho_hf_in.data(); + rhogs_out = rho_sg_out.data(); + rhoghf_out = rho_hf_out.data(); + } + else + { + rhogs_in = chr->rhog_save[0]; + rhogs_out = chr->rhog[0]; + } + + // inner_product_recip_hartree is a hartree-like sum, unit is Ry + std::function*, std::complex*)> inner_product + = [this](std::complex* rhog1, std::complex* rhog2) + { + return module_charge::inner_product_recip_hartree( + rhog1, rhog2, *this->rhopw, this->cfg_, *this->omega, *this->tpiba); + }; + + // Kerker screening functor, shared by all nspin branches. + // Short-circuit when close_kerker_gg0() was called (non-separate-loop + // EXX path): cfg_ is immutable, so the disable flag lives on the object. + std::function*)> screen = [this](std::complex* p) { + if (this->kerker_disabled_) + { + return; + } + module_charge::kerker_screen_recip(this->cfg_, this->rhopw, *this->tpiba, p); + }; + + // DIIS Mixing Only for smooth part, while high_frequency part is mixed by plain mixing method. + if (nspin == 1) + { + rhog_in = rhogs_in; + rhog_out = rhogs_out; + this->mixing->push_data(this->rho_mdata, rhog_in, rhog_out, screen, true); + this->mixing->cal_coef(this->rho_mdata, inner_product); + this->mixing->mix_data(this->rho_mdata, rhog_out); + } + else if (nspin == 2) + { + // magnetic density + const int npw = this->rhopw->npw; + std::vector> rhog_mag(npw * nspin); + std::vector> rhog_mag_save(npw * nspin); + module_charge::detail::pack_rho_mag(rhog_mag.data(), chr->rhog[0], chr->rhog[1], npw); + module_charge::detail::pack_rho_mag(rhog_mag_save.data(), chr->rhog_save[0], chr->rhog_save[1], npw); + // + rhog_in = rhog_mag_save.data(); + rhog_out = rhog_mag.data(); + std::function*, const std::complex*, + const std::complex*)> twobeta_mix + = module_charge::detail::make_twobeta_mix>( + 2 * npw, npw, this->cfg_.mixing_beta, this->cfg_.mixing_beta_mag); + this->mixing->push_data(this->rho_mdata, rhog_in, rhog_out, screen, twobeta_mix, true); + this->mixing->cal_coef(this->rho_mdata, inner_product); + this->mixing->mix_data(this->rho_mdata, rhog_out); + // get rhog[is][ngmc] from rhog_mag[is*ngmc] + module_charge::detail::unpack_rho_mag(chr->rhog[0], chr->rhog[1], rhog_mag.data(), npw); + // sync rhogs_out so merge_dgrid has the latest smooth part + if ( cfg_.double_grid) + { + for (int ig = 0; ig < npw; ig++) + { + rhogs_out[ig] = chr->rhog[0][ig]; + rhogs_out[ig + npw] = chr->rhog[1][ig]; + } + } + } + else if (nspin == 4 && cfg_.mixing_angle <= 0) + { + // normal broyden mixing for {rho, mx, my, mz} + rhog_in = rhogs_in; + rhog_out = rhogs_out; + const int npw = this->rhopw->npw; + std::function*, const std::complex*, + const std::complex*)> twobeta_mix + = module_charge::detail::make_twobeta_mix>( + 4 * npw, npw, this->cfg_.mixing_beta, this->cfg_.mixing_beta_mag); + this->mixing->push_data(this->rho_mdata, rhog_in, rhog_out, screen, twobeta_mix, true); + this->mixing->cal_coef(this->rho_mdata, inner_product); + this->mixing->mix_data(this->rho_mdata, rhog_out); + } + else if (nspin == 4 && cfg_.mixing_angle > 0) + { + // special broyden mixing for {rho, |m|} proposed by J. Phys. Soc. Jpn. 82 (2013) 114706 + // here only consider the case of mixing_angle = 1, which mean only change |m| and keep angle fixed + // old support see mix_rho_recip() + if ( cfg_.double_grid) + { + ModuleBase::WARNING_QUIT("Charge_Mixing", "double_grid is not supported for new mixing method yet."); + } + // rho_magabs and rho_magabs_save, zero-initialized + const int nrxx = this->rhopw->nrxx; + std::vector rho_magabs(nrxx); + std::vector rho_magabs_save(nrxx); + // calculate rho_magabs and rho_magabs_save + for (int ir = 0; ir < nrxx; ir++) + { + // |m| for rho + rho_magabs[ir] = std::sqrt(chr->rho[1][ir] * chr->rho[1][ir] + + chr->rho[2][ir] * chr->rho[2][ir] + + chr->rho[3][ir] * chr->rho[3][ir]); + // |m| for rho_save + rho_magabs_save[ir] = std::sqrt(chr->rho_save[1][ir] * chr->rho_save[1][ir] + + chr->rho_save[2][ir] * chr->rho_save[2][ir] + + chr->rho_save[3][ir] * chr->rho_save[3][ir]); + } + // rhog_magabs and rhog_magabs_save, zero-initialized + const int npw = this->rhopw->npw; + std::vector> rhog_magabs(npw * 2); + std::vector> rhog_magabs_save(npw * 2); + // calculate rhog_magabs and rhog_magabs_save + for (int ig = 0; ig < npw; ig++) + { + rhog_magabs[ig] = chr->rhog[0][ig]; // rho + rhog_magabs_save[ig] = chr->rhog_save[0][ig]; // rho_save + } + // FT to get rhog_magabs and rhog_magabs_save + this->rhopw->real2recip(rho_magabs.data(), rhog_magabs.data() + this->rhopw->npw); + this->rhopw->real2recip(rho_magabs_save.data(), rhog_magabs_save.data() + this->rhopw->npw); + // + rhog_in = rhog_magabs_save.data(); + rhog_out = rhog_magabs.data(); + std::function*, const std::complex*, + const std::complex*)> twobeta_mix + = module_charge::detail::make_twobeta_mix>( + 2 * npw, npw, this->cfg_.mixing_beta, this->cfg_.mixing_beta_mag); + this->mixing->push_data(this->rho_mdata, rhog_in, rhog_out, screen, twobeta_mix, true); + this->mixing->cal_coef(this->rho_mdata, inner_product); + this->mixing->mix_data(this->rho_mdata, rhog_out); + // get new |m| in real space using FT + this->rhopw->recip2real(rhog_magabs.data() + this->rhopw->npw, rho_magabs.data()); + // Reciprocal-space rho was mixed into rhog_magabs[0..npw-1]; write it + // back to chr->rhog[0]. This copy is bounded by the reciprocal grid. + for (int ig = 0; ig < npw; ig++) + { + chr->rhog[0][ig] = rhog_magabs[ig]; + } + // The new |m| in real space was produced by recip2real above into + // rho_magabs[0..nrxx-1]. Rescale {mx,my,mz} on every real-space point. + // The loop bound is nrxx (not npw) and the source is rho_magabs[ir] + // (not rho_magabs[npw+ig]), otherwise the tail [npw,nrxx) is left + // unscaled and rho_magabs[npw+ig] reads out of bounds when npw>0. + for (int ir = 0; ir < nrxx; ir++) + { + double norm = std::sqrt(chr->rho[1][ir] * chr->rho[1][ir] + + chr->rho[2][ir] * chr->rho[2][ir] + + chr->rho[3][ir] * chr->rho[3][ir]); + if (std::abs(norm) < 1e-10) + { + continue; + } + double rescale_tmp = rho_magabs[ir] / norm; + chr->rho[1][ir] *= rescale_tmp; + chr->rho[2][ir] *= rescale_tmp; + chr->rho[3][ir] *= rescale_tmp; + } + } + + if ( cfg_.double_grid) + { + // plain mixing for high_frequencies + const int ndimhf = (this->rhodpw->npw - this->rhopw->npw) * nspin; + this->mixing_highf->plain_mix(rhoghf_out, rhoghf_in, rhoghf_out, ndimhf, nullptr); + + // combine smooth part and high_frequency part; + // rho_sg_* / rho_hf_* vectors are released automatically at scope exit + module_charge::merge_dgrid(chr->rhog[0], rho_sg_out, rho_hf_out, + nspin, this->rhopw->npw, this->rhodpw->npw); + } + + // rhog to rho + if (nspin == 4 && cfg_.mixing_angle > 0) + { + // only tranfer rhog[0] + // do not support double_grid, use rhopw directly + chr->rhopw->recip2real(chr->rhog[0], chr->rho[0]); + } + else + { + for (int is = 0; is < nspin; is++) + { + // use rhodpw for double_grid + // rhodpw is the same as rhopw for ! cfg_.double_grid + this->rhodpw->recip_to_real, double, + base_device::DEVICE_CPU>(chr->rhog[is], chr->rho[is]); + } + } + // For kinetic energy density + if (cfg_.mixing_tau) + { + module_charge::detail::mix_tau_recip(chr, nspin, cfg_.double_grid, + this->rhopw, this->rhodpw, + this->mixing.get(), this->tau_mdata, this->mixing_highf.get()); + } + + ModuleBase::timer::end("Charge_Mixing", "mix_rho_recip"); + return; +} + +void Charge_Mixing::mix_rho_real(Charge* chr) +{ + ModuleBase::TITLE("Charge_Mixing", "mix_rho_real"); + ModuleBase::timer::start("Charge_Mixing", "mix_rho_real"); + + const int nspin = cfg_.nspin; + assert(nspin==1 || nspin==2 || nspin==4); + + double* rhor_in=nullptr; + double* rhor_out=nullptr; + + // Kerker screening functor (see mix_rho_recip for the disable flag rationale). + std::function screen = [this](double* p) { + if (this->kerker_disabled_) + { + return; + } + module_charge::kerker_screen_real(this->cfg_, this->rhopw, *this->tpiba, p); + }; + std::function inner_product = [this](double* rho1, double* rho2) + { + return module_charge::inner_product_real(rho1, rho2, *this->rhopw, this->cfg_); + }; + + if (nspin == 1) + { + rhor_in = chr->rho_save[0]; + rhor_out = chr->rho[0]; + this->mixing->push_data(this->rho_mdata, rhor_in, rhor_out, screen, true); + this->mixing->cal_coef(this->rho_mdata, inner_product); + this->mixing->mix_data(this->rho_mdata, rhor_out); + } + else if (nspin == 2) + { + // magnetic density + const int nrxx = this->rhopw->nrxx; + std::vector rho_mag(nrxx * nspin); + std::vector rho_mag_save(nrxx * nspin); + module_charge::detail::pack_rho_mag(rho_mag.data(), chr->rho[0], chr->rho[1], nrxx); + module_charge::detail::pack_rho_mag(rho_mag_save.data(), chr->rho_save[0], chr->rho_save[1], nrxx); + // + rhor_in = rho_mag_save.data(); + rhor_out = rho_mag.data(); + std::function twobeta_mix + = module_charge::detail::make_twobeta_mix(2 * nrxx, nrxx, this->cfg_.mixing_beta, this->cfg_.mixing_beta_mag); + this->mixing->push_data(this->rho_mdata, rhor_in, rhor_out, screen, twobeta_mix, true); + this->mixing->cal_coef(this->rho_mdata, inner_product); + this->mixing->mix_data(this->rho_mdata, rhor_out); + // get new rho[is][nrxx] from rho_mag[is*nrxx] + module_charge::detail::unpack_rho_mag(chr->rho[0], chr->rho[1], rho_mag.data(), nrxx); + } + else if (nspin == 4 && cfg_.mixing_angle <= 0) + { + rhor_in = chr->rho_save[0]; + rhor_out = chr->rho[0]; + const int nrxx = this->rhopw->nrxx; + std::function twobeta_mix + = module_charge::detail::make_twobeta_mix(4 * nrxx, nrxx, this->cfg_.mixing_beta, this->cfg_.mixing_beta_mag); + this->mixing->push_data(this->rho_mdata, rhor_in, rhor_out, screen, twobeta_mix, true); + this->mixing->cal_coef(this->rho_mdata, inner_product); + this->mixing->mix_data(this->rho_mdata, rhor_out); + } + else if (nspin == 4 && cfg_.mixing_angle > 0) + { + // real-space version of the {rho, |m|} broyden mixing + const int nrxx = this->rhopw->nrxx; + std::vector rho_magabs(nrxx * 2); + std::vector rho_magabs_save(nrxx * 2); + for (int ir = 0; ir < nrxx; ir++) + { + rho_magabs[ir] = chr->rho[0][ir]; // rho + rho_magabs_save[ir] = chr->rho_save[0][ir]; // rho_save + // |m| for rho + rho_magabs[nrxx + ir] = std::sqrt(chr->rho[1][ir] * chr->rho[1][ir] + + chr->rho[2][ir] * chr->rho[2][ir] + + chr->rho[3][ir] * chr->rho[3][ir]); + // |m| for rho_save + rho_magabs_save[nrxx + ir] = std::sqrt(chr->rho_save[1][ir] * chr->rho_save[1][ir] + + chr->rho_save[2][ir] * chr->rho_save[2][ir] + + chr->rho_save[3][ir] * chr->rho_save[3][ir]); + } + rhor_in = rho_magabs_save.data(); + rhor_out = rho_magabs.data(); + + std::function twobeta_mix + = module_charge::detail::make_twobeta_mix(2 * nrxx, nrxx, this->cfg_.mixing_beta, this->cfg_.mixing_beta_mag); + this->mixing->push_data(this->rho_mdata, rhor_in, rhor_out, screen, twobeta_mix, true); + this->mixing->cal_coef(this->rho_mdata, inner_product); + this->mixing->mix_data(this->rho_mdata, rhor_out); + + for (int ir = 0; ir < nrxx; ir++) + { + chr->rho[0][ir] = rho_magabs[ir]; // rho + double norm = std::sqrt(chr->rho[1][ir] * chr->rho[1][ir] + + chr->rho[2][ir] * chr->rho[2][ir] + + chr->rho[3][ir] * chr->rho[3][ir]); + + if (norm < 1e-10) + { + continue; + } + double rescale_tmp = rho_magabs[nrxx + ir] / norm; + chr->rho[1][ir] *= rescale_tmp; + chr->rho[2][ir] *= rescale_tmp; + chr->rho[3][ir] *= rescale_tmp; + } + } + + double *taur_out=nullptr; + double *taur_in=nullptr; + if (cfg_.mixing_tau) + { + taur_in = chr->kin_r_save[0]; + taur_out = chr->kin_r[0]; + // Note: there is no kerker modification for tau because I'm not sure + // if we should have it. If necessary we can try it in the future. + this->mixing->push_data(this->tau_mdata, taur_in, taur_out, nullptr, false); + + this->mixing->mix_data(this->tau_mdata, taur_out); + } + + ModuleBase::timer::end("Charge_Mixing", "mix_rho_real"); + return; +} + +void Charge_Mixing::mix_rho(Charge* chr) +{ + ModuleBase::TITLE("Charge_Mixing", "mix_rho"); + ModuleBase::timer::start("Charge_Mixing", "mix_rho"); + + /// Fail fast on invalid arguments and a skipped set_mixing: the body + /// dereferences these pointers unconditionally below. + if (chr == nullptr || chr->rhopw == nullptr) + { + ModuleBase::WARNING_QUIT("Charge_Mixing", + "chr or chr->rhopw is null in mix_rho"); + } + if (this->rhopw == nullptr) + { + ModuleBase::WARNING_QUIT("Charge_Mixing", + "set_mixing must be called before mix_rho"); + } + if (cfg_.double_grid && this->rhodpw == nullptr) + { + ModuleBase::WARNING_QUIT("Charge_Mixing", + "rhodpw is null when double_grid is enabled"); + } + + const int nspin = cfg_.nspin; + assert(nspin==1 || nspin==2 || nspin==4); + + // the charge before mixing. + const int nrxx = chr->rhopw->nrxx; + std::vector rho123(nspin * nrxx); + for (int is = 0; is < nspin; ++is) + { + if (is == 0 || is == 3 || !cfg_.domag_z) + { + double* rho123_is = rho123.data() + is * nrxx; +#ifdef _OPENMP +#pragma omp parallel for schedule(static, 512) +#endif + for(int ir = 0 ; ir < nrxx ; ++ir) + { + rho123_is[ir] = chr->rho[is][ir]; + } + } + } + std::vector kin_r123; + if (cfg_.mixing_tau) + { + kin_r123.resize(nspin * nrxx); + for (int is = 0; is < nspin; ++is) + { + double* kin_r123_is = kin_r123.data() + is * nrxx; +#ifdef _OPENMP +#pragma omp parallel for schedule(static, 512) +#endif + for(int ir = 0 ; ir < nrxx ; ++ir) + { + kin_r123_is[ir] = chr->kin_r[is][ir]; + } + } + } + // --------------------Mixing Body-------------------- + if (cfg_.scf_thr_type == 1) + { + mix_rho_recip(chr); + } + else if (cfg_.scf_thr_type == 2) + { + mix_rho_real(chr); + } + // --------------------------------------------------- + + // mohan add 2012-06-05 + // rho_save is the charge before mixing + for (int is = 0; is < nspin; ++is) + { + if (is == 0 || is == 3 || !cfg_.domag_z) + { + double* rho123_is = rho123.data() + is * nrxx; +#ifdef _OPENMP +#pragma omp parallel for schedule(static, 512) +#endif + for(int ir = 0 ; ir < nrxx ; ++ir) + { + chr->rho_save[is][ir] = rho123_is[ir]; + } + } + } + + if (cfg_.mixing_tau) + { + for (int is = 0; is < nspin; ++is) + { + double* kin_r123_is = kin_r123.data() + is * nrxx; +#ifdef _OPENMP +#pragma omp parallel for schedule(static, 512) +#endif + for(int ir = 0 ; ir < nrxx ; ++ir) + { + chr->kin_r_save[is][ir] = kin_r123_is[ir]; + } + } + } + + ModuleBase::timer::end("Charge_Mixing", "mix_rho"); + return; +} diff --git a/source/source_estate/module_charge/chg_parallel.cpp b/source/source_estate/module_charge/chg_parallel.cpp new file mode 100644 index 00000000000..f8e95169d90 --- /dev/null +++ b/source/source_estate/module_charge/chg_parallel.cpp @@ -0,0 +1,95 @@ +#include "chg_parallel.h" + +#ifdef __MPI + +#include + +#include "charge.h" +#include "source_base/global_function.h" +#include "source_base/global_variable.h" +#include "source_base/parallel_comm.h" +#include "source_base/timer.h" + +namespace module_charge +{ + +void reduce_diff_pools(double* array_rho, const Charge& chr, const int kpar, + const bool all_ks_run, const int bndpar) +{ + ModuleBase::TITLE("Charge", "reduce_diff_pools"); + ModuleBase::timer::start("Charge", "reduce_diff_pools"); + // A rank may own zero real-space grid points (nrxx == 0); in that case + // the buffer is legitimately null and the MPI calls below use count 0. + // Only a null buffer with a non-zero nrxx is a genuine bug. + assert(array_rho != nullptr || chr.nrxx == 0); + assert(kpar >= 1); + assert(bndpar >= 1); + if (kpar > 1) + { + assert(chr.pgrid != nullptr); + chr.pgrid->reduce_across_pools(array_rho); + } + if (all_ks_run && bndpar > 1) + { + // nrxx may be 0 on ranks with empty grid partitions; MPI_Allreduce + // with count 0 is valid and ignores the buffer. + MPI_Allreduce(MPI_IN_PLACE, array_rho, chr.nrxx, MPI_DOUBLE, MPI_SUM, BP_WORLD); + } + ModuleBase::timer::end("Charge", "reduce_diff_pools"); +} + +void rho_mpi(Charge& chr, const int kpar, const bool all_ks_run, + const int bndpar, const int nspin) +{ + ModuleBase::TITLE("Charge", "rho_mpi"); + assert(kpar >= 1); + assert(bndpar >= 1); + assert(nspin > 0); + if (kpar * bndpar <= 1) + { + return; + } + ModuleBase::timer::start("Charge", "rho_mpi"); + + assert(chr.rho != nullptr); + for (int is = 0; is < nspin; ++is) + { + reduce_diff_pools(chr.rho[is], chr, kpar, all_ks_run, bndpar); + if (chr.kin_r != nullptr) + { + reduce_diff_pools(chr.kin_r[is], chr, kpar, all_ks_run, bndpar); + } + } + + ModuleBase::timer::end("Charge", "rho_mpi"); + return; +} + +void kin_r_mpi(Charge& chr, const int kpar, const bool all_ks_run, + const int bndpar, const int nspin) +{ + ModuleBase::TITLE("Charge", "kin_r_mpi"); + assert(kpar >= 1); + assert(bndpar >= 1); + assert(nspin > 0); + if (kpar * bndpar <= 1) + { + return; + } + ModuleBase::timer::start("Charge", "kin_r_mpi"); + + if (chr.kin_r != nullptr) + { + for (int is = 0; is < nspin; ++is) + { + reduce_diff_pools(chr.kin_r[is], chr, kpar, all_ks_run, bndpar); + } + } + + ModuleBase::timer::end("Charge", "kin_r_mpi"); + return; +} + +} // namespace module_charge + +#endif diff --git a/source/source_estate/module_charge/chg_parallel.h b/source/source_estate/module_charge/chg_parallel.h new file mode 100644 index 00000000000..50eb09fde59 --- /dev/null +++ b/source/source_estate/module_charge/chg_parallel.h @@ -0,0 +1,59 @@ +#ifndef CHG_PARALLEL_H +#define CHG_PARALLEL_H + +// MPI reductions of charge-density arrays across k-point pools and band +// groups. Stateless free functions extracted from Charge member functions; +// the charge buffers and the parallel grid are supplied by the Charge +// argument. The pool/band parallelization settings are passed explicitly +// by the callers instead of reading GlobalV/PARAM. + +#ifdef __MPI + +class Charge; + +namespace module_charge +{ + +/** + * @brief Reduce a real-space array across k-point pools and band groups. + * + * @param array_rho real-space array [chr.nrxx], reduced in place + * @param chr charge object supplying the parallel grid and the local grid size + * @param kpar number of k-point pools (GlobalV::KPAR) + * @param all_ks_run whether all processes run KS calculations (PARAM.globalv.all_ks_run) + * @param bndpar number of band groups (PARAM.inp.bndpar) + */ +void reduce_diff_pools(double* array_rho, const Charge& chr, const int kpar, + const bool all_ks_run, const int bndpar); + +/** + * @brief Reduce rho across pools; also reduce kin_r when its buffer is + * allocated (meta-GGA functionals, or ELF output requested). + * + * @param chr charge object supplying rho/kin_r buffers + * @param kpar number of k-point pools (GlobalV::KPAR) + * @param all_ks_run whether all processes run KS calculations (PARAM.globalv.all_ks_run) + * @param bndpar number of band groups (PARAM.inp.bndpar) + * @param nspin number of spin channels (PARAM.inp.nspin) + */ +void rho_mpi(Charge& chr, const int kpar, const bool all_ks_run, + const int bndpar, const int nspin); + +/** + * @brief Reduce kin_r across pools when its buffer is allocated + * (meta-GGA functionals, or ELF output requested). + * + * @param chr charge object supplying kin_r buffers + * @param kpar number of k-point pools (GlobalV::KPAR) + * @param all_ks_run whether all processes run KS calculations (PARAM.globalv.all_ks_run) + * @param bndpar number of band groups (PARAM.inp.bndpar) + * @param nspin number of spin channels (PARAM.inp.nspin) + */ +void kin_r_mpi(Charge& chr, const int kpar, const bool all_ks_run, + const int bndpar, const int nspin); + +} // namespace module_charge + +#endif + +#endif // CHG_PARALLEL_H diff --git a/source/source_estate/module_charge/chg_precond.cpp b/source/source_estate/module_charge/chg_precond.cpp new file mode 100644 index 00000000000..57bac5030bd --- /dev/null +++ b/source/source_estate/module_charge/chg_precond.cpp @@ -0,0 +1,194 @@ +#include "chg_precond.h" + +#include "source_base/constants.h" +#include "source_base/timer.h" +#include "source_basis/module_pw/pw_basis.h" + +#include +#include +#include +#include + +namespace module_charge +{ + +void kerker_screen_recip(const MixingConfig& cfg, + ModulePW::PW_Basis* rhopw, + double tpiba, + std::complex* drhog) +{ + ModuleBase::TITLE("module_charge", "kerker_screen_recip"); + + if (cfg.mixing_gg0 <= 0.0 || cfg.mixing_beta <= 0.1) + { + return; + } + + ModuleBase::timer::start("module_charge", "kerker_screen_recip"); + + const int nspin = cfg.nspin; + + double fac = 0.0; + double gg0 = 0.0; + double amin = 0.0; + + /// consider a resize for mixing_angle + int resize_tmp = 1; + if (nspin == 4 && cfg.mixing_angle > 0) + { + resize_tmp = 2; + } + + /// implement Kerker for density and magnetization separately + for (int is = 0; is < nspin / resize_tmp; ++is) + { + const int is_idx = is * rhopw->npw; + /// new mixing method only support nspin=2 not nspin=4 + if (is >= 1) + { + if (cfg.mixing_gg0_mag <= 0.0001 || cfg.mixing_beta_mag <= 0.1) + { +#ifdef __DEBUG + assert(is == 1); // make sure break works +#endif + double is_mag = nspin - 1; + //for (int ig = 0; ig < rhopw->npw * is_mag; ig++) + //{ + // drhog[is_idx + ig] *= 1; + //} + break; + } + fac = cfg.mixing_gg0_mag; + amin = cfg.mixing_beta_mag; + } + else + { + fac = cfg.mixing_gg0; + amin = cfg.mixing_beta; + } + + gg0 = std::pow(fac * ModuleBase::BOHR_TO_A / tpiba, 2); + + const double gg0_amin = cfg.mixing_gg0_min / amin; + +#ifdef _OPENMP +#pragma omp parallel for schedule(static, 512) +#endif + for (int ig = 0; ig < rhopw->npw; ++ig) + { + double gg = rhopw->gg[ig]; + double filter_g = std::max(gg / (gg + gg0), gg0_amin); + drhog[is_idx + ig] *= filter_g; + } + } + + ModuleBase::timer::end("module_charge", "kerker_screen_recip"); + return; +} + +void kerker_screen_real(const MixingConfig& cfg, + ModulePW::PW_Basis* rhopw, + double tpiba, + double* drhor) +{ + ModuleBase::TITLE("module_charge", "kerker_screen_real"); + + if (cfg.mixing_gg0 <= 0.0001 || cfg.mixing_beta <= 0.1) + { + return; + } + + ModuleBase::timer::start("module_charge", "kerker_screen_real"); + + const int nspin = cfg.nspin; + assert(nspin == 1 || nspin == 2 || nspin == 4); + + /// consider a resize for mixing_angle + int resize_tmp = 1; + if (nspin == 4 && cfg.mixing_angle > 0) + { + resize_tmp = 2; + } + + std::vector> drhog(rhopw->npw * nspin / resize_tmp); + std::vector drhor_filter(rhopw->nrxx * nspin / resize_tmp); + + for (int is = 0; is < nspin / resize_tmp; ++is) + { + // Note after this process some G which is higher than Gmax will be filtered. + // Thus we cannot use kerker_screen_recip(drhog.data()) directly after it. + rhopw->real2recip(drhor + is * rhopw->nrxx, drhog.data() + is * rhopw->npw); + } + double fac = 0.0; + double gg0 = 0.0; + double amin = 0.0; + + for (int is = 0; is < nspin / resize_tmp; is++) + { + + if (is >= 1) + { + if (cfg.mixing_gg0_mag <= 0.0001 || cfg.mixing_beta_mag <= 0.1) + { +#ifdef __DEBUG + assert(is == 1); /// make sure break works +#endif + double is_mag = nspin - 1; + if (nspin == 4 && cfg.mixing_angle > 0) + { + is_mag = 1; + } + for (int ig = 0; ig < rhopw->npw * is_mag; ig++) + { + drhog[is * rhopw->npw + ig] = 0; + } + break; + } + fac = cfg.mixing_gg0_mag; + amin = cfg.mixing_beta_mag; + } + else + { + fac = cfg.mixing_gg0; + amin = cfg.mixing_beta; + } + + gg0 = std::pow(fac * ModuleBase::BOHR_TO_A / tpiba, 2); + + const int is_idx = is * rhopw->npw; + const double gg0_amin = cfg.mixing_gg0_min / amin; +#ifdef _OPENMP +#pragma omp parallel for schedule(static, 512) +#endif + for (int ig = 0; ig < rhopw->npw; ig++) + { + double gg = rhopw->gg[ig]; + // I have not decided how to handle gg=0 part, will be changed in future + //if (gg == 0) + //{ + // drhog[is_idx + ig] *= 0; + // continue; + //} + double filter_g = std::max(gg / (gg + gg0), gg0_amin); + drhog[is_idx + ig] *= (1 - filter_g); + } + } + /// inverse FT + for (int is = 0; is < nspin / resize_tmp; ++is) + { + rhopw->recip2real(drhog.data() + is * rhopw->npw, drhor_filter.data() + is * rhopw->nrxx); + } + +#ifdef _OPENMP +#pragma omp parallel for schedule(static, 512) +#endif + for (int ir = 0; ir < rhopw->nrxx * nspin / resize_tmp; ir++) + { + drhor[ir] -= drhor_filter[ir]; + } + + ModuleBase::timer::end("module_charge", "kerker_screen_real"); + return; +} + +} // namespace module_charge diff --git a/source/source_estate/module_charge/chg_precond.h b/source/source_estate/module_charge/chg_precond.h new file mode 100644 index 00000000000..8d8d2298ba5 --- /dev/null +++ b/source/source_estate/module_charge/chg_precond.h @@ -0,0 +1,56 @@ +#ifndef CHG_PRECOND_H +#define CHG_PRECOND_H + +// Stateless Kerker preconditioning kernels extracted from Charge_Mixing. +// Every input (grid, geometry, mixing config) is passed explicitly; the +// functions do not read Charge_Mixing members or PARAM/GlobalV. + +#include + +#include "chg_mix_cfg.h" + +namespace ModulePW +{ +class PW_Basis; +} + +namespace module_charge +{ + +/** + * @brief Apply Kerker screening in reciprocal space. + * + * Multiplies drhog[is*npw + ig] by max(gg/(gg+gg0), gg0_min/amin) per spin + * channel, where gg0 is derived from cfg.mixing_gg0 (density) or + * cfg.mixing_gg0_mag (magnetization). Early return if Kerker is disabled. + * + * @param cfg mixing config (spin count, betas, gg0s, angle, gg0_min) + * @param rhopw plane-wave basis supplying npw and gg[] + * @param tpiba 2*pi/lattice constant used to convert gg0 to atomic units + * @param drhog[in,out] reciprocal-space density residual, length nspin*npw + */ +void kerker_screen_recip(const MixingConfig& cfg, + ModulePW::PW_Basis* rhopw, + double tpiba, + std::complex* drhog); + +/** + * @brief Apply Kerker screening in real space via FFT. + * + * Forward-transforms drhor to drhog, applies (1 - filter_g) in reciprocal + * space, backward-transforms the filtered residual, and subtracts it from + * drhor in place. Early return if Kerker is disabled. + * + * @param cfg mixing config (spin count, betas, gg0s, angle, gg0_min) + * @param rhopw plane-wave basis supplying npw, nrxx, gg[], real2recip/recip2real + * @param tpiba 2*pi/lattice constant used to convert gg0 to atomic units + * @param drhor[in,out] real-space density residual, length nspin*nrxx + */ +void kerker_screen_real(const MixingConfig& cfg, + ModulePW::PW_Basis* rhopw, + double tpiba, + double* drhor); + +} // namespace module_charge + +#endif // CHG_PRECOND_H diff --git a/source/source_estate/module_charge/chg_rho_detail.h b/source/source_estate/module_charge/chg_rho_detail.h new file mode 100644 index 00000000000..231e39223f6 --- /dev/null +++ b/source/source_estate/module_charge/chg_rho_detail.h @@ -0,0 +1,128 @@ +#ifndef CHG_RHO_DETAIL_H +#define CHG_RHO_DETAIL_H + +// Internal helpers for charge density mixing (mix_rho_recip/mix_rho_real). +// Not part of the public module_charge API: only chg_mix.cpp +// and the charge mixing unit test are expected to include this header. + +#include +#include + +#include "charge.h" +#include "chg_mix_cfg.h" +#include "source_base/module_mixing/mixing.h" +#include "source_base/module_mixing/plain_mixing.h" +#include "source_base/tool_quit.h" + +namespace module_charge +{ +namespace detail +{ + +/** + * @brief Create a two-beta mixing functor: mix the first nunit elements with + * mixing_beta and the rest (nunit..total) with mixing_beta_mag. + * Used for magnetic cases (nspin==2/4) where the charge channel and + * the magnetism channels use different betas. + * @tparam T element type, double (real space) or std::complex (reciprocal) + * @param total total number of elements + * @param nunit number of elements in the charge channel + * @param mixing_beta beta for the charge channel + * @param mixing_beta_mag beta for the magnetism channel + * @return mixing functor + */ +template +std::function make_twobeta_mix( + const int total, const int nunit, + const double mixing_beta, const double mixing_beta_mag) +{ + return [total, nunit, mixing_beta, mixing_beta_mag](T* out, const T* in, const T* sres) + { +#ifdef _OPENMP +#pragma omp parallel for schedule(static, 256) +#endif + for (int i = 0; i < nunit; ++i) + { + out[i] = in[i] + mixing_beta * sres[i]; + } +#ifdef _OPENMP +#pragma omp parallel for schedule(static, 256) +#endif + for (int i = nunit; i < total; ++i) + { + out[i] = in[i] + mixing_beta_mag * sres[i]; + } + }; +} + +/** + * @brief Pack charge and magnetism into interleaved layout: + * out[0..n] = d0 + d1 (charge channel) + * out[n..2n] = d0 - d1 (magnetism channel) + * @tparam T double (real space) or std::complex (reciprocal) + * @param out output buffer, size >= 2*n + * @param d0 first component (e.g. chr->rho[0] or chr->rhog[0]) + * @param d1 second component + * @param n number of elements per component + */ +template +void pack_rho_mag(T* out, const T* d0, const T* d1, const int n) +{ + if (n < 0) + { + ModuleBase::WARNING_QUIT("pack_rho_mag", "n must be >= 0"); + } + // A rank may own zero real-space grid points (n == 0) when the grid is + // decomposed across more processes than it has slabs. Its buffers are + // legitimately null in that case and the loop below performs no access; + // only a null buffer with a positive n is a genuine bug. + if (n > 0 && (out == nullptr || d0 == nullptr || d1 == nullptr)) + { + ModuleBase::WARNING_QUIT("pack_rho_mag", "pointer is null"); + } +#ifdef _OPENMP +#pragma omp parallel for schedule(static, 512) +#endif + for (int i = 0; i < n; ++i) + { + out[i] = d0[i] + d1[i]; + out[i + n] = d0[i] - d1[i]; + } +} + +/** + * @brief Unpack interleaved layout back to charge and magnetism components: + * d0[i] = 0.5 * (in[i] + in[i+n]) + * d1[i] = 0.5 * (in[i] - in[i+n]) + * @tparam T double (real space) or std::complex (reciprocal) + * @param d0 output first component (e.g. chr->rho[0] or chr->rhog[0]) + * @param d1 output second component + * @param in input buffer, size >= 2*n + * @param n number of elements per component + */ +template +void unpack_rho_mag(T* d0, T* d1, const T* in, const int n) +{ + if (n < 0) + { + ModuleBase::WARNING_QUIT("unpack_rho_mag", "n must be >= 0"); + } + // A rank may own zero real-space grid points (n == 0); see pack_rho_mag. + if (n > 0 && (d0 == nullptr || d1 == nullptr || in == nullptr)) + { + ModuleBase::WARNING_QUIT("unpack_rho_mag", "pointer is null"); + } +#ifdef _OPENMP +#pragma omp parallel for schedule(static, 512) +#endif + for (int i = 0; i < n; ++i) + { + d0[i] = 0.5 * (in[i] + in[i + n]); + d1[i] = 0.5 * (in[i] - in[i + n]); + } +} + +} // namespace detail +} // namespace module_charge + +#endif // CHG_RHO_DETAIL_H diff --git a/source/source_estate/module_charge/chg_rhog_io.cpp b/source/source_estate/module_charge/chg_rhog_io.cpp new file mode 100644 index 00000000000..9de2dca4a9b --- /dev/null +++ b/source/source_estate/module_charge/chg_rhog_io.cpp @@ -0,0 +1,374 @@ +#include "source_base/module_out/binstream.h" +#include "source_base/vector3.h" +#include "source_base/module_parallel/para_mpi_func.h" +#include "chg_rhog_io.h" +#include +#include +#include + +namespace +{ +inline void warn(std::ostream* os, + const Parallel::ParaWorld& pw_world, + const std::string& file, + const std::string& desc) +{ + if (pw_world.rank() == 0 && os != nullptr) + { + *os << " " << file << " warning : " << desc << std::endl; + } +} +} // namespace + +bool module_charge::read_rhog(const std::string& filename, + const ModulePW::PW_Basis* pw_rhod, + const int nspin, + std::complex** rhog, + const Parallel::ParaWorld& pw_world, + std::ostream* os_warning) +{ + if (pw_rhod == nullptr) + { + warn(os_warning, pw_world, "module_charge::read_rhog", "pw_rhod is null"); + return false; + } + if (rhog == nullptr) + { + warn(os_warning, pw_world, "module_charge::read_rhog", "rhog is null"); + return false; + } + if (nspin != 1 && nspin != 2 && nspin != 4) + { + warn(os_warning, pw_world, "module_charge::read_rhog", "nspin must be 1, 2, or 4"); + return false; + } + if (pw_rhod->nx <= 0 || pw_rhod->ny <= 0 || pw_rhod->nz <= 0) + { + warn(os_warning, pw_world, "module_charge::read_rhog", "PW_Basis grid dimensions must be positive"); + return false; + } + + const int nx = pw_rhod->nx; + const int ny = pw_rhod->ny; + const int nz = pw_rhod->nz; + + Binstream ifs; + bool error = false; + int gamma_only_in = 0; + int npwtot_in = 0; + int nspin_in = 0; + int size = 0; + double b1[3], b2[3], b3[3]; + + if (pw_world.rank() == 0) + { + ifs.open(filename, "r"); + if (!ifs) + { + error = true; + } + } + + Parallel::bcast_bool(error, pw_world); + + if (error) + { + warn(os_warning, pw_world, "module_charge::read_rhog", "Can't open file " + filename); + return false; + } + + if (pw_world.rank() == 0) + { + ifs >> size >> gamma_only_in >> npwtot_in >> nspin_in >> size; + ifs >> size >> b1[0] >> b1[1] >> b1[2] >> b2[0] >> b2[1] >> b2[2] >> b3[0] >> b3[1] >> b3[2] >> size; + if (gamma_only_in != pw_rhod->gamma_only) + { + // there is a treatment that can transform between gamma_only and non-gamma_only + // however, it is not implemented here + error = true; + ifs.close(); + } + if (npwtot_in > pw_rhod->npwtot) + { + warn(os_warning, pw_world, "module_charge::read_rhog", "some planewaves in file are not used"); + } + else if (npwtot_in < pw_rhod->npwtot) + { + warn(os_warning, pw_world, "module_charge::read_rhog", "some planewaves in file are missing"); + } + if (nspin_in < nspin) + { + warn(os_warning, pw_world, "module_charge::read_rhog", "some spin channels in file are missing"); + } + } + + Parallel::bcast_bool(error, pw_world); + + if (error) + { + warn(os_warning, pw_world, "module_charge::read_rhog", "gamma_only read from file is inconsistent with INPUT"); + return false; + } + + Parallel::bcast_int(gamma_only_in, pw_world); + Parallel::bcast_int(npwtot_in, pw_world); + Parallel::bcast_int(nspin_in, pw_world); + Parallel::bcast_double(b1, 3, pw_world); + Parallel::bcast_double(b2, 3, pw_world); + Parallel::bcast_double(b3, 3, pw_world); + + std::vector miller(npwtot_in * 3); + // once use ModuleBase::Vector3, it is highly bug-prone to assume the memory layout of the class. + // The x, y and z of Vector3 will not always to be contiguous. + // Instead, a relatively safe choice is to use std::vector, the memory layout is assumed + // to be npwtot_in rows and 3 columns. + if (pw_world.rank() == 0) + { + ifs >> size; + for (int i = 0; i < npwtot_in; ++i) // loop over rows... + { + ifs >> miller[i*3] >> miller[i*3+1] >> miller[i*3+2]; + } + ifs >> size; + } + Parallel::bcast_int(miller.data(), miller.size(), pw_world); + + // set to zero + for (int is = 0; is < nspin; ++is) + { + std::fill(rhog[is], rhog[is] + pw_rhod->npw, std::complex(0.0, 0.0)); + } + // maps ixyz tp ig + std::vector fftixyz2ig(pw_rhod->nxyz, -1); // map isz to ig. + for (int ig = 0; ig < pw_rhod->npw; ++ig) + { + int isz = pw_rhod->ig2isz[ig]; + int iz = isz % nz; + int is = isz / nz; + int ixy = pw_rhod->is2fftixy[is]; + int ixyz = iz + nz * ixy; + fftixyz2ig[ixyz] = ig; + } + std::vector> rhog_in(npwtot_in); + for (int is = 0; is < nspin_in; ++is) + { + if (pw_world.rank() == 0) + { + ifs >> size; + for (int i = 0; i < npwtot_in; ++i) + { + ifs >> rhog_in[i]; + } + ifs >> size; + } + Parallel::bcast_complex(rhog_in.data(), rhog_in.size(), pw_world); + + for (int i = 0; i < npwtot_in; ++i) + { + int ix = miller[i * 3]; + int iy = miller[i * 3 + 1]; + int iz = miller[i * 3 + 2]; + + if (ix <= -int((nx + 1) / 2) || ix >= int(nx / 2) + 1 || iy <= -int((ny + 1) / 2) || iy >= int(ny / 2) + 1 + || iz <= -int((nz + 1) / 2) || iz >= int(nz / 2) + 1) + { + // these planewaves are not used + continue; + } + + if (ix < 0) + ix += nx; + if (iy < 0) + iy += ny; + if (iz < 0) + iz += nz; + int fftixy = iy + pw_rhod->fftny * ix; + if (pw_world.rank() == pw_rhod->fftixy2ip[fftixy]) + { + int fftixyz = iz + nz * fftixy; + int ig = fftixyz2ig[fftixyz]; + rhog[is][ig] = rhog_in[i]; + } + } + + if (nspin_in == 2 && nspin == 4 && is == 1) + { + for (int ig = 0; ig < pw_rhod->npw; ++ig) + { + rhog[3][ig] = rhog[1][ig]; + } + std::fill(rhog[1], rhog[1] + pw_rhod->npw, std::complex(0.0, 0.0)); + std::fill(rhog[2], rhog[2] + pw_rhod->npw, std::complex(0.0, 0.0)); + } + } + + if (pw_world.rank() == 0) + { + ifs.close(); + } + return true; +} + +bool module_charge::write_rhog(const std::string& fchg, + const bool gamma_only, + const ModulePW::PW_Basis* pw_rho, + const int nspin, + const ModuleBase::Matrix3& GT, + std::complex** rhog, + const Parallel::ParaWorld& pw_world, + std::ostream* os_warning) +{ + if (pw_rho == nullptr) + { + warn(os_warning, pw_world, "module_charge::write_rhog", "pw_rho is null"); + return false; + } + if (rhog == nullptr) + { + warn(os_warning, pw_world, "module_charge::write_rhog", "rhog is null"); + return false; + } + if (nspin != 1 && nspin != 2 && nspin != 4) + { + warn(os_warning, pw_world, "module_charge::write_rhog", "nspin must be 1, 2, or 4"); + return false; + } + + // only rank 0 in the domain writes the header; all ranks cooperate + // on sequential writes synchronized by barriers. + const int irank = pw_world.rank(); + const int nrank = pw_world.size(); + + // write the header (by rank 0): gamma_only, ngm_g, nspin + int size = 3; + int ngm_g = pw_rho->npwtot; + int gam = gamma_only; + int nsp = nspin; + + std::ofstream ofs; + Parallel::barrier(pw_world); + + if (irank == 0) + { + ofs.open(fchg, std::ios::binary); + if (!ofs) + { + warn(os_warning, pw_world, "module_charge::write_rhog", "File I/O failure: cannot open file " + fchg); + return false; + } + ofs.write(reinterpret_cast(&size), sizeof(size)); + ofs.write(reinterpret_cast(&gam), sizeof(gam)); + ofs.write(reinterpret_cast(&ngm_g), sizeof(ngm_g)); + ofs.write(reinterpret_cast(&nsp), sizeof(nsp)); + ofs.write(reinterpret_cast(&size), sizeof(size)); + // write the lattice vectors + std::vector b = {GT.e11, GT.e12, GT.e13, GT.e21, GT.e22, GT.e23, GT.e31, GT.e32, GT.e33}; + size = 9; + ofs.write(reinterpret_cast(&size), sizeof(size)); + for (int i = 0; i < 9; ++i) + { + ofs.write(reinterpret_cast(&b[i]), sizeof(b[i])); + } + ofs.write(reinterpret_cast(&size), sizeof(size)); + ofs.close(); + } + Parallel::barrier(pw_world); + Parallel::barrier(pw_world); + + // write the G-vectors in Miller indices + size = 3 * ngm_g; + if (irank == 0) + { + ofs.open(fchg, std::ios::binary | std::ios::app); + ofs.write(reinterpret_cast(&size), sizeof(size)); + ofs.close(); + } + Parallel::barrier(pw_world); + + for (int i = 0; i < nrank; ++i) + { + if (i == irank) + { + ofs.open(fchg, std::ios::binary | std::ios::app); + for (int ig = 0; ig < pw_rho->npw; ++ig) + { + const ModuleBase::Vector3 g = pw_rho->gdirect[ig]; + std::vector miller = {int(g.x), int(g.y), int(g.z)}; + ofs.write(reinterpret_cast(&miller[0]), sizeof(miller[0])); + ofs.write(reinterpret_cast(&miller[1]), sizeof(miller[1])); + ofs.write(reinterpret_cast(&miller[2]), sizeof(miller[2])); + } + ofs.close(); + } + Parallel::barrier(pw_world); + } + + if (irank == 0) + { + ofs.open(fchg, std::ios::binary | std::ios::app); + ofs.write(reinterpret_cast(&size), sizeof(size)); + ofs.close(); + } + Parallel::barrier(pw_world); + + // write the rho(G) values + std::complex sum_check; + size = ngm_g; + for (int ispin = 0; ispin < nspin; ++ispin) + { + if (irank == 0) + { + ofs.open(fchg, std::ios::binary | std::ios::app); + ofs.write(reinterpret_cast(&size), sizeof(size)); + ofs.close(); + } + Parallel::barrier(pw_world); + + for (int i = 0; i < nrank; ++i) + { + if (i == irank) + { + ofs.open(fchg, std::ios::binary | std::ios::app); + sum_check = 0.0; + for (int ig = 0; ig < pw_rho->npw; ++ig) + { + sum_check += rhog[ispin][ig]; + ofs.write(reinterpret_cast(&rhog[ispin][ig]), sizeof(rhog[ispin][ig])); + } + ofs.close(); + } + Parallel::barrier(pw_world); + } + + if (irank == 0) + { + ofs.open(fchg, std::ios::binary | std::ios::app); + ofs.write(reinterpret_cast(&size), sizeof(size)); + ofs.close(); + } + Parallel::barrier(pw_world); + } + return true; +} + +// self-consistency test with the following python code +// import numpy as np + +// with open("rhog_read.txt") as f: +// read = f.readlines() + +// with open("rhog_write.txt") as f: +// write = f.readlines() + +// # convert c++ stype complex number (a,b) to python complex +// def to_complex(s): +// a, b = s.replace("(", "").replace(")", "").split(",") +// return complex(float(a), float(b)) + +// read = [[to_complex(rhog) for rhog in spin.strip().split()] for spin in read] +// write = [[to_complex(rhog) for rhog in spin.strip().split()] for spin in write] + +// diff = np.array(read) - np.array(write) +// print(np.max(np.abs(diff))) +// test system: integrated test 118_PW_CHG_BINARY +// yielding error 5.290000000000175e-11 \ No newline at end of file diff --git a/source/source_estate/module_charge/chg_rhog_io.h b/source/source_estate/module_charge/chg_rhog_io.h new file mode 100644 index 00000000000..3f3f2d2edad --- /dev/null +++ b/source/source_estate/module_charge/chg_rhog_io.h @@ -0,0 +1,66 @@ +#ifndef CHG_RHOG_IO_H +#define CHG_RHOG_IO_H + +#include +#include +#include +#include "source_basis/module_pw/pw_basis.h" +#include "source_base/module_parallel/para_world.h" +/** + * I/O free function of rho(G) in binary format + * Author: YuLiu98, Kirk0830 + * + * The designed foramt of the binary file is kept similar to the one in QuantumESPRESSO + * Modules/io_base.f90, function write_rhog (non-HDF5): + * + * /3/ /gammaonly/ /ngm_g/ /nspin/ /3/ ! bool, int, int. ngm_g is the global, total number of G-vectors + * /9/ /b11/ /b12/ /b13/ /b21/ /b22/ /b23/ /b31/ /b32/ /b33/ /9/ ! 9 real numbers, the three lattice vectors + * /ngm_g*3/ + * /miller(1,1)/ /miller(1,2)/ /miller(1,3)/ + * /miller(2,1)/ /miller(2,2)/ /miller(2,3)/ + * ... + * /miller(ngm_g,1)/ /miller(ngm_g,2)/ /miller(ngm_g,3)/ + * /ngm_g*3/ ! ngm_g*3 integers, the G-vectors in Miller indices + * /ngm_g/ + * /rhog(1)/ /rhog(2)/ /rhog(3)/ ... + * /ngm_g/ ! ngm_g complex numbers, the rho(G) values + * !/ngm_g/ ! ngm_g complex numbers, the rho(G) values for the second spin component + * !/rhog(1)/ /rhog(2)/ /rhog(3)/ ... + * !/ngm_g/ ! ngm_g complex numbers, the rho(G) values for the second spin component + * + * There are some aspects needed to ensure: + * 1. the correspondence between ABACUS PW and QuantumESPRESSO + * ABACUS PW QuantumESPRESSO + * bij UnitCell::GT b1, b2, b3 + * miller index ibox[*] mill + * rho Charge::rhog rho + * + * 2. the unit of each quantity + * ABACUS PW QuantumESPRESSO + * bij a.u. + * miller index + * rho + */ + +namespace module_charge +{ + +bool read_rhog(const std::string& filename, + const ModulePW::PW_Basis* pw_rhod, + const int nspin, + std::complex** rhog, + const Parallel::ParaWorld& pw_world, + std::ostream* os_warning); + +bool write_rhog(const std::string& fchg, + const bool gamma_only, + const ModulePW::PW_Basis* pw_rho, + const int nspin, + const ModuleBase::Matrix3& GT, + std::complex** rhog, + const Parallel::ParaWorld& pw_world, + std::ostream* os_warning); + +} // namespace module_charge + +#endif diff --git a/source/source_estate/module_charge/chg_routine.cpp b/source/source_estate/module_charge/chg_routine.cpp new file mode 100644 index 00000000000..938de4dcaab --- /dev/null +++ b/source/source_estate/module_charge/chg_routine.cpp @@ -0,0 +1,234 @@ +#include "source_estate/module_charge/chg_routine.h" +#include "source_estate/module_charge/chg_dmr.h" + +#include "source_base/parallel_comm.h" +#include "source_pw/module_pwdft/dftu_base.h" // Plus_U_Base members used below +#include "source_estate/update_pot.h" + +void module_charge::chgmixing_ks(const int iter, + UnitCell& ucell, + elecstate::ElecState* pelec, + Charge &chr, + const ModulePW::PW_Basis& rhopw, + Charge_Mixing* p_chgmix, + ScfMixingCtx& ctx, + const Input_para& inp) +{ + const double& hsolver_error = ctx.hsolver_error; + const double& scf_thr = ctx.scf_thr; + const double& scf_ene_thr = ctx.scf_ene_thr; + const bool& converged_u = ctx.converged_u; + double& drho = ctx.drho; + bool& oscillate_esolver = ctx.oscillate_esolver; + bool& conv_esolver = ctx.conv_esolver; + const int nrxx = rhopw.nrxx; + + if (ctx.ks_run) + { + // mixing will restart at p_chgmix->mixing_restart steps + if (drho <= inp.mixing_restart && inp.mixing_restart > 0.0 + && p_chgmix->mixing_restart_step > iter) + { + p_chgmix->mixing_restart_step = iter + 1; + } + + if (inp.scf_os_stop) // if oscillation is detected, SCF will stop + { + oscillate_esolver = p_chgmix->if_scf_oscillate(iter, drho, + inp.scf_os_ndim, inp.scf_os_thr); + } + + // drho will be 0 at p_chgmix->mixing_restart step, which is + // not ground state + bool not_restart_step = !(iter == p_chgmix->mixing_restart_step && inp.mixing_restart > 0.0); + + conv_esolver = (drho < scf_thr && not_restart_step && converged_u); + + // add energy threshold for SCF convergence + if (scf_ene_thr > 0.0) + { + // calculate energy of output charge density + elecstate::update_pot(ucell, pelec, chr, conv_esolver); + pelec->cal_energies(2, + inp.imp_sol, + inp.sc_mag_switch, + inp.dft_plus_u, + inp.assume_isolated); // 2 means Kohn-Sham functional + // now, etot_old is the energy of input density, while etot is the energy of output density + pelec->f_en.etot_delta = pelec->f_en.etot - pelec->f_en.etot_old; + // output etot_delta + GlobalV::ofs_running << " DeltaE_womix = " << pelec->f_en.etot_delta * ModuleBase::Ry_to_eV << " eV" + << std::endl; + if (iter > 1 && conv_esolver == 1) // only check when density is converged + { + // update the convergence flag + conv_esolver + = (std::abs(pelec->f_en.etot_delta * ModuleBase::Ry_to_eV) < scf_ene_thr); + } + } + + // If drho < hsolver_error in the first iter or drho < scf_thr, we + // do not change rho. + if (drho < hsolver_error || conv_esolver || inp.calculation == "nscf") + { + if (drho < hsolver_error) + { + GlobalV::ofs_warning << " drho < hsolver_error, keep " + "charge density unchanged." + << std::endl; + } + } + else + { + //----------charge mixing--------------- + // mixing will restart after p_chgmix->mixing_restart + // steps + if (inp.mixing_restart > 0 && iter == p_chgmix->mixing_restart_step - 1 + && drho <= inp.mixing_restart) + { + // do not mix charge density + } + else + { + p_chgmix->mix_rho(&chr); // update chr->rho by mixing + } + if (inp.scf_thr_type == 2) + { + chr.renormalize_rho(inp.nelec, ucell.omega); // renormalize rho in R-space would + // induce a error in K-space + } + //----------charge mixing done----------- + } + } + +#ifdef __MPI + MPI_Bcast(&drho, 1, MPI_DOUBLE, 0, BP_WORLD); + + // change MPI_DOUBLE to MPI_C_BOOL, mohan 2025-04-13 + MPI_Bcast(&conv_esolver, 1, MPI_C_BOOL, 0, BP_WORLD); + + assert(nrxx>=0); // mohan add 2025-10-18 + MPI_Bcast(chr.rho[0], nrxx, MPI_DOUBLE, 0, BP_WORLD); +#endif + + // mohan move the following code here, 2025-10-18 + // SCF restart information + if (inp.mixing_restart > 0 + && iter == p_chgmix->mixing_restart_step - 1 + && iter != inp.scf_nmax) + { + p_chgmix->mixing_restart_last = iter; + std::cout << " SCF restart after this step!" << std::endl; + } + + return; +} + + +void module_charge::chgmixing_ks_pw(const int iter, // scf iteration number + Charge_Mixing* p_chgmix, // charge mixing class + Plus_U_Base& dftu, + const bool mag_converged, ///< whether DeltaSpin magnetization converged (true when disabled) + const Input_para& inp) // input parameters +{ + ModuleBase::TITLE("module_charge", "chgmixing_ks_pw"); + + if (iter == 1) + { + p_chgmix->init_mixing(); + p_chgmix->mixing_restart_step = inp.scf_nmax + 1; + if (inp.dft_plus_u && dftu.has_occ_mixer()) + { + // allocate memory for uom_mdata sized to the flat occupation buffer + p_chgmix->allocate_mixing_uom(dftu.occ_mixer().flat_size()); + } + } + + // For mixing restart + if (iter == p_chgmix->mixing_restart_step && inp.mixing_restart > 0.0) + { + p_chgmix->init_mixing(); + p_chgmix->mixing_restart_count++; + + if (inp.dft_plus_u) + { + if (dftu.get_uramping() > 0.01 && !dftu.u_converged()) + { + p_chgmix->mixing_restart_step = inp.scf_nmax + 1; + } + if (dftu.get_uramping() > 0.01) + { + if (mag_converged) // skip uramping if mag not converged + { + dftu.uramping_update(); // update U by uramping if uramping > 0.01 + std::cout << " U-Ramping! Current U = "; + for (int i = 0; i < dftu.get_num_u_types(); i++) + { + std::cout << dftu.get_u_current(i) * ModuleBase::Ry_to_eV << " "; + } + std::cout << " eV " << std::endl; + } + } + } + } + + return; +} + +void module_charge::chgmixing_ks_lcao(const int iter, // scf iteration number + Charge_Mixing* p_chgmix, // charge mixing class + Plus_U_Base& dftu, + const int nnr, // dimension of density matrix + const Input_para& inp) // input parameters +{ + ModuleBase::TITLE("module_charge", "chgmixing_ks_lcao"); + + if (iter == 1) + { + p_chgmix->mix_reset(); // init mixing + p_chgmix->mixing_restart_step = inp.scf_nmax + 1; + p_chgmix->mixing_restart_count = 0; + // this output will be removed once the feeature is stable + if (dftu.get_uramping() > 0.01) + { + std::cout << " U-Ramping! Current U = "; + for (int i = 0; i < dftu.get_num_u_types(); i++) + { + std::cout << dftu.get_u_current(i) * ModuleBase::Ry_to_eV << " "; + } + std::cout << " eV " << std::endl; + } + } + + // for mixing restart + if (iter == p_chgmix->mixing_restart_step && inp.mixing_restart > 0.0) + { + p_chgmix->init_mixing(); + p_chgmix->mixing_restart_count++; + if (inp.dft_plus_u) + { + dftu.uramping_update(); // update U by uramping if uramping > 0.01 + if (dftu.get_uramping() > 0.01) + { + std::cout << " U-Ramping! Current U = "; + for (int i = 0; i < dftu.get_num_u_types(); i++) + { + std::cout << dftu.get_u_current(i) * ModuleBase::Ry_to_eV << " "; + } + std::cout << " eV " << std::endl; + } + if (dftu.get_uramping() > 0.01 && !dftu.u_converged()) + { + p_chgmix->mixing_restart_step = inp.scf_nmax + 1; + } + } + if (inp.mixing_dmr) // for mixing_dmr + { + // allocate memory for dmr_mdata + module_charge::init_mixing_dmr(p_chgmix->get_mixing(), + p_chgmix->get_dmr_mdata(), + nnr, + p_chgmix->get_mixing_config()); + } + } +} diff --git a/source/source_estate/module_charge/chg_routine.h b/source/source_estate/module_charge/chg_routine.h new file mode 100644 index 00000000000..19a67cd1335 --- /dev/null +++ b/source/source_estate/module_charge/chg_routine.h @@ -0,0 +1,54 @@ +#ifndef CHG_ROUTINE_H +#define CHG_ROUTINE_H + +#include "source_estate/elecstate.h" // use pelec +#include "source_estate/module_charge/charge.h" // use chr +#include "source_estate/module_charge/chg_mix.h" // use p_chgmix +#include "source_basis/module_pw/pw_basis.h" // use PW_Basis +#include "source_io/module_parameter/input_parameter.h" // use Input_para +#include "source_cell/unitcell.h" + +// Plus_U_Base forward declaration, full definition in source_pw/module_pwdft/dftu_base.h +class Plus_U_Base; + +namespace module_charge +{ + +/// Aggregated SCF convergence thresholds and status flags for chgmixing_ks +struct ScfMixingCtx +{ + double hsolver_error; ///< solver error from diagonalization + double scf_thr; ///< charge density convergence threshold + double scf_ene_thr; ///< energy convergence threshold + bool converged_u; ///< whether DFT+U has converged + bool ks_run; ///< whether the current run is a KS calculation (PARAM.globalv.ks_run) + double drho; ///< charge density deviation (in/out) + bool oscillate_esolver; ///< whether esolver oscillates (out) + bool conv_esolver; ///< whether esolver converged (out) +}; + +void chgmixing_ks(const int iter, + UnitCell& ucell, + elecstate::ElecState* pelec, + Charge &chr, + const ModulePW::PW_Basis& rhopw, + Charge_Mixing* p_chgmix, + ScfMixingCtx& ctx, + const Input_para& inp); + +void chgmixing_ks_pw(const int iter, + Charge_Mixing* p_chgmix, + Plus_U_Base& dftu, + const bool mag_converged, ///< whether DeltaSpin magnetization converged; pass true when sc_mag_switch is off + const Input_para& inp); // input parameters + +void chgmixing_ks_lcao(const int iter, // scf iteration number + Charge_Mixing* p_chgmix, // charge mixing class + Plus_U_Base& dftu, + const int nnr, // dimension of density matrix + const Input_para& inp); // input parameters + +} + + +#endif diff --git a/source/source_estate/module_charge/chg_symm.cpp b/source/source_estate/module_charge/chg_symm.cpp new file mode 100644 index 00000000000..aeb1d88e248 --- /dev/null +++ b/source/source_estate/module_charge/chg_symm.cpp @@ -0,0 +1,136 @@ +#include "chg_symm.h" + +#include "chg_symm_detail.h" +#include "source_estate/module_charge/charge.h" + +namespace module_charge +{ + +// TODO: callers currently pass different grids for LCAO-based paths: +// esolver_ks_lcao (and TDDFT/SDFT LCAO) pass the smooth pw_rho, while +// esolver_ks_lcaopw (LIP) and get_pchg_pw pass the dense pw_rhod. The two +// coincide only because LCAO rejects USPP, so double_grid is always false there +// (see uspp_support.cpp). If LCAO is extended to USPP, verify which grid the +// charge symmetrization must use before relaxing that restriction. +void symmetrize_rho(const int nspin, + const Charge& chr, + const ModulePW::PW_Basis* pw, + ModuleSymmetry::Symmetry& symm) +{ + if (nspin == 4) + { + // nspin=4 (non-collinear/SOC): rho[0] is the charge density rho^0 (scalar, symmetrized + // spatially like nspin=1); rho[1,2,3] are the spin density (rho^x, rho^y, rho^z) which + // must be symmetrized TOGETHER with the per-operation spin rotation W(g). + cal_rhog_symm(0, chr, pw, symm); + cal_rhog_symm_soc(chr.rho, chr.rhog, pw, symm); + return; + } + for (int is = 0; is < nspin; is++) + { + cal_rhog_symm(is, chr, pw, symm); + } +} + +void cal_rhog_symm(const int& spin_now, + const Charge& chr, + const ModulePW::PW_Basis* rho_basis, + ModuleSymmetry::Symmetry& symm) +{ + assert(spin_now < 4); // added by zhengdy-soc + + if (ModuleSymmetry::Symmetry::symm_flag != 1) + { + return; + } + + ModuleBase::TITLE("module_charge", "cal_rhog_symm"); + ModuleBase::timer::start("module_charge", "cal_rhog_symm"); + + rho_basis->real2recip(chr.rho[spin_now], chr.rhog[spin_now]); + + detail::psymmg(chr.rhog[spin_now], rho_basis, symm); + + rho_basis->recip2real(chr.rhog[spin_now], chr.rho[spin_now]); + + if (chr.kin_r != nullptr) + { + // Use std::vector to manage kin_g instead of raw pointer + std::vector> kin_g(rho_basis->npw); + rho_basis->real2recip(chr.kin_r[spin_now], kin_g.data()); + detail::psymmg(kin_g.data(), rho_basis, symm); + rho_basis->recip2real(kin_g.data(), chr.kin_r[spin_now]); + } + + ModuleBase::timer::end("module_charge", "cal_rhog_symm"); + return; +} + +void cal_rhog_symm(const int& spin_now, + double** rho, + std::complex** rhog, + int ngmc, + double** kin_r, + const ModulePW::PW_Basis* rho_basis, + ModuleSymmetry::Symmetry& symm) +{ + assert(spin_now < 4); // added by zhengdy-soc + + if (ModuleSymmetry::Symmetry::symm_flag != 1) + { + return; + } + + ModuleBase::TITLE("module_charge", "cal_rhog_symm"); + ModuleBase::timer::start("module_charge", "cal_rhog_symm"); + + { + rho_basis->real2recip(rho[spin_now], rhog[spin_now]); + detail::psymmg(rhog[spin_now], rho_basis, symm); + rho_basis->recip2real(rhog[spin_now], rho[spin_now]); + + if (kin_r != nullptr) + { + std::vector> kin_g(ngmc); + rho_basis->real2recip(kin_r[spin_now], kin_g.data()); + detail::psymmg(kin_g.data(), rho_basis, symm); + rho_basis->recip2real(kin_g.data(), kin_r[spin_now]); + } + } + + ModuleBase::timer::end("module_charge", "cal_rhog_symm"); + return; +} + +void cal_rhog_symm_soc(double** rho, + std::complex** rhog, + const ModulePW::PW_Basis* rho_basis, + ModuleSymmetry::Symmetry& symm) +{ + if (ModuleSymmetry::Symmetry::symm_flag != 1) + { + return; + } + + ModuleBase::TITLE("module_charge", "cal_rhog_symm_soc"); + ModuleBase::timer::start("module_charge", "cal_rhog_symm_soc"); + + // the three spin components are coupled by the spin rotation, so they are transformed to + // reciprocal space and symmetrized together (rho[1]=rho^x, rho[2]=rho^y, rho[3]=rho^z). + for (int is = 1; is < 4; ++is) + { + rho_basis->real2recip(rho[is], rhog[is]); + } + + detail::psymmg_soc(rhog[1], rhog[2], rhog[3], rho_basis, symm); + + for (int is = 1; is < 4; ++is) + { + rho_basis->recip2real(rhog[is], rho[is]); + } + + ModuleBase::timer::end("module_charge", "cal_rhog_symm_soc"); + return; +} + +} // namespace module_charge diff --git a/source/source_estate/module_charge/chg_symm.h b/source/source_estate/module_charge/chg_symm.h new file mode 100644 index 00000000000..9354cb38399 --- /dev/null +++ b/source/source_estate/module_charge/chg_symm.h @@ -0,0 +1,83 @@ +#ifndef CHG_SYMM_H +#define CHG_SYMM_H + +// TODO: make cal_rhog_symm / cal_rhog_symm_soc internal (detail or anonymous +// namespace) so that external callers only use symmetrize_rho. Blocked by: +// 1. get_pchg_lcao/pw call the double** overload — need a symmetrize_rho +// overload that accepts raw arrays (with nspin=4 branch). +// 2. write_mlkedf_desc symmetrizes a single component of a temporary array +// — symmetrize_rho cannot express that yet. +// 3. setup_pot, ctrl_output_fp, read_wf2rho, update_state_rdmft already use +// the Charge& overload and can be migrated directly. + +#include + +#include "source_basis/module_pw/pw_basis.h" +#include "source_cell/module_symmetry/symmetry.h" + +class Charge; + +/** + * @brief Charge-density symmetrization free functions. + * + * The functions are stateless: every input is passed explicitly. The + * reciprocal-space helpers shared between translation units live in + * module_charge::detail (see chg_symm_detail.h). + */ +namespace module_charge +{ + +/** + * @brief Symmetrize charge density for all spin channels + * + * This is a helper function that symmetrizes the charge density + * for all spin channels by calling cal_rhog_symm() for each spin. + * + * @param nspin Number of spin channels + * @param chr Charge object containing the density + * @param pw Plane wave basis + * @param symm Symmetry object + */ +void symmetrize_rho(const int nspin, + const Charge& chr, + const ModulePW::PW_Basis* pw, + ModuleSymmetry::Symmetry& symm); + +/** + * @brief Symmetrize one spin channel of the charge density. + * + * The kinetic-energy density kin_r is symmetrized as well whenever its + * buffer is allocated (meta-GGA functionals, or ELF output requested). + */ +void cal_rhog_symm(const int& spin_now, + const Charge& CHR, + const ModulePW::PW_Basis* pw, + ModuleSymmetry::Symmetry& symm); + +/** + * @brief Symmetrize one spin channel of raw density arrays. + */ +void cal_rhog_symm(const int& spin_now, + double** rho, + std::complex** rhog, + int ngmc, + double** kin_r, + const ModulePW::PW_Basis* pw, + ModuleSymmetry::Symmetry& symm); + +/** + * @brief Symmetrize raw nspin=4 spin-density arrays with coupled spin rotations. + * + * @param rho Real-space density components ordered as rho0, mx, my, mz. + * @param rhog Reciprocal-space work arrays with the same component ordering. + * @param pw Plane-wave basis used for the Fourier transforms. + * @param symm Symmetry operations and spin rotations. + */ +void cal_rhog_symm_soc(double** rho, + std::complex** rhog, + const ModulePW::PW_Basis* pw, + ModuleSymmetry::Symmetry& symm); + +} // namespace module_charge + +#endif diff --git a/source/source_estate/module_charge/chg_symm_detail.cpp b/source/source_estate/module_charge/chg_symm_detail.cpp new file mode 100644 index 00000000000..8a630d2f052 --- /dev/null +++ b/source/source_estate/module_charge/chg_symm_detail.cpp @@ -0,0 +1,394 @@ +#include "chg_symm_detail.h" + +#include +#include + +#include "source_base/parallel_reduce.h" +#include "source_base/parallel_global.h" +#include "source_cell/module_symmetry/symm_rot_spin.h" + +namespace module_charge +{ +namespace +{ + +#ifdef __MPI + +void reduce_to_fullrhog(const ModulePW::PW_Basis* rho_basis, + std::complex* rhogtot, + std::complex* rhogin, + int* ig2isztot, + const int* ig2iszin, + int max_npw) +{ + ModuleBase::TITLE("module_charge","reduce_to_fullrhog"); + + std::vector> rhog_piece_vec(max_npw); + std::vector ig2isz_piece_vec(max_npw); + std::complex* rhog_piece = rhog_piece_vec.data(); + int* ig2isz_piece = ig2isz_piece_vec.data(); + + int npw_start=0; + for(int proc=0; procpoolnproc; ++proc) + { + std::fill(rhog_piece, rhog_piece + max_npw, std::complex(0.0, 0.0)); + std::fill(ig2isz_piece, ig2isz_piece + max_npw, 0); + + MPI_Status ierror; + + // case 1: the first part of rho in processor 0 in each pool. + if(proc == 0 && GlobalV::RANK_IN_POOL ==0) + { + for(int ig=0; ignpw; ++ig) + { + rhog_piece[ig] = rhogin[ig]; + ig2isz_piece[ig]=ig2iszin[ig]; + } + } + + // case 2: > first part rho: send the rho to + // processor 0 in each pool + else if(proc == GlobalV::RANK_IN_POOL ) + { + for(int ig=0; ignpw; ++ig) + { + rhog_piece[ig] = rhogin[ig]; + ig2isz_piece[ig]=ig2iszin[ig]; + } + MPI_Send(rhog_piece,rho_basis->npw, MPI_DOUBLE_COMPLEX, 0, proc, POOL_WORLD); + MPI_Send(ig2isz_piece, rho_basis->npw, MPI_INT, 0, proc+rho_basis->poolnproc, POOL_WORLD); + } + + // case 2: > first part rho: processor 0 receive the rho + // from other processors + else if(GlobalV::RANK_IN_POOL==0) + { + MPI_Recv(rhog_piece, rho_basis->npw_per[proc], MPI_DOUBLE_COMPLEX, proc, proc, POOL_WORLD, &ierror); + MPI_Recv(ig2isz_piece, rho_basis->npw_per[proc], MPI_INT, + proc, proc + rho_basis->poolnproc, POOL_WORLD, &ierror); + } + + if(GlobalV::RANK_IN_POOL==0) + { + for(int ig=0; ignpw_per[proc]; ++ig) + { + rhogtot[npw_start+ig] = rhog_piece[ig]; + ig2isztot[npw_start+ig] = ig2isz_piece[ig]; + } + npw_start+=rho_basis->npw_per[proc]; + } + } + if(GlobalV::RANK_IN_POOL==0) + { + assert(npw_start==rho_basis->npwtot); + } + + MPI_Barrier(MPI_COMM_WORLD); + + return; +} + +void rhog_piece_to_all(const ModulePW::PW_Basis* rho_basis, + std::complex* rhogtot, + std::complex* rhog_part) +{ + ModuleBase::TITLE("module_charge","rhog_piece_to_all"); + + MPI_Status ierror; + + if(GlobalV::RANK_IN_POOL==0) + { + // proc 0: send to other proc in pool + // itself: directly copy + for(int ig=0;ignpw;++ig) + { + rhog_part[ig]=rhogtot[ig]; + } + int npw_start=rho_basis->npw; + for(int proc=1;procpoolnproc;++proc) + { + MPI_Send(&rhogtot[npw_start], rho_basis->npw_per[proc], MPI_DOUBLE_COMPLEX, proc, proc, POOL_WORLD); + npw_start+=rho_basis->npw_per[proc]; + } + assert(npw_start==rho_basis->npwtot); + }// GlobalV::RANK_IN_POOL == 0 + else + { + MPI_Recv(rhog_part, rho_basis->npw_per[GlobalV::RANK_IN_POOL], + MPI_DOUBLE_COMPLEX, 0, GlobalV::RANK_IN_POOL, + POOL_WORLD, &ierror); + } + return; +} + +#endif + +// only for MYRANK==0 +void get_ixyz2ipw(const ModulePW::PW_Basis* rho_basis, + const int* ig2isztot, + const int* fftixy2is, + int* ixyz2ipw) +{ + //step 1: get ipsz2ipw + + //get ipsz2ipw from ig2isztot + std::vector ipsz2ipw_vec(rho_basis->nstot * rho_basis->nz); + int* ipsz2ipw = ipsz2ipw_vec.data(); + for(int i=0;instot*rho_basis->nz;++i) + { + ipsz2ipw[i]=-1; + } + + int npw_count=0; + int nstnz_count=0; + int ipsz=0; //global index of a z-grid on stick + int isz=0; //local index of a z-grid stick on ip core + int ipw=0; // global index of pw (in npwtot) + for (int ip=0;ippoolnproc;++ip) + { + for (int ig=0;ignpw_per[ip];++ig) + { + ipw=npw_count+ig; + isz=ig2isztot[ipw]; + ipsz=nstnz_count+isz; + ipsz2ipw[ipsz]=ipw; + } + npw_count+=rho_basis->npw_per[ip]; + nstnz_count+=rho_basis->nst_per[ip]*rho_basis->nz; + } + assert(npw_count==rho_basis->npwtot); + assert(nstnz_count==rho_basis->nstot*rho_basis->nz); + + //step2: ixyz to ipsz + + //save the start-index of (nst*nz) till each core + std::vector nstnz_start_vec(rho_basis->poolnproc); + int* nstnz_start = nstnz_start_vec.data(); + nstnz_start[0]=0; + for (int ip=1; ippoolnproc; ++ip) + { + nstnz_start[ip]=nstnz_start[ip-1]+rho_basis->nst_per[ip-1]*rho_basis->nz; + } + + //tmp variables + int ixy, ixyz, ip, is, ig=0; + + for (int ix=0;ixfftnx;++ix) + { + for (int iy=0;iyfftny;++iy) + { + for(int iz=0;izfftnz;++iz) + { + ixy = ix*rho_basis->fftny + iy; + ixyz = ixy*rho_basis->fftnz+iz; + ip = rho_basis->fftixy2ip[ixy]; + if (ip==-1) continue; //not in any core + is = fftixy2is[ixy]; //stick-index on ip=proc core + if (is==-1) continue; //not on any stick + ipsz = nstnz_start[ip]+is*rho_basis->nz+iz; + ipw = ipsz2ipw[ipsz]; + ixyz2ipw[ixyz] = ipw; + } + } + } + assert (ixyz==rho_basis->fftnxyz-1); + + + return; +} + +} // anonymous namespace + +namespace detail +{ + +void psymmg(std::complex* rhog_part, const ModulePW::PW_Basis *rho_basis, ModuleSymmetry::Symmetry &symm) +{ + //(1) get fftixy2is and do Allreduce + std::vector fftixy2is_vec(rho_basis->fftnxy); + int* fftixy2is = fftixy2is_vec.data(); + rho_basis->getfftixy2is(fftixy2is); //current proc +#ifdef __MPI + Parallel_Reduce::reduce_pool(fftixy2is, rho_basis->fftnxy); + if(rho_basis->poolnproc>1) + { + for (int i=0;ifftnxy;++i) + { + fftixy2is[i]+=rho_basis->poolnproc-1; + } + } + + // (2) reduce all rho from the first pool. + std::vector> rhogtot_vec; + std::vector ig2isztot_vec; + std::complex* rhogtot = nullptr; + int* ig2isztot = nullptr; + if(GlobalV::RANK_IN_POOL == 0) + { + rhogtot_vec.resize(rho_basis->npwtot); + ig2isztot_vec.resize(rho_basis->npwtot); + rhogtot = rhogtot_vec.data(); + ig2isztot = ig2isztot_vec.data(); + } + // find max_npw + int max_npw=0; + for (int proc = 0; proc < rho_basis->poolnproc; ++proc) + { + if(rho_basis->npw_per[proc] > max_npw) + { + max_npw=rho_basis->npw_per[proc]; + } + } + reduce_to_fullrhog(rho_basis, rhogtot, rhog_part, ig2isztot, rho_basis->ig2isz, max_npw); + + // (3) get ixy2ipw and do rhog_symmetry on proc 0 of each pool + if(GlobalV::RANK_IN_POOL==0) + { +#endif + //init ixyz2ipw + std::vector ixyz2ipw_vec(rho_basis->fftnxyz); + int* ixyz2ipw = ixyz2ipw_vec.data(); + for(int i=0;ifftnxyz;++i) + { + ixyz2ipw[i]=-1; + } + // The density must be symmetrized with the same group used to fold the k-points. For + // nspin=4 magnetic that is the Shubnikov group; Theta leaves the charge invariant, so the + // antiunitary elements act on rho exactly like unitary ones (their trs_inv is not used here). + std::vector kgmat; + std::vector> gtr; + std::vector trs_inv; + const int nop = symm.density_sym_ops(kgmat, gtr, trs_inv); +#ifdef __MPI + get_ixyz2ipw(rho_basis, ig2isztot, fftixy2is, ixyz2ipw); + symm.rhog_symmetry(rhogtot, ixyz2ipw, rho_basis->nx, rho_basis->ny, rho_basis->nz, + rho_basis->fftnx, rho_basis->fftny, rho_basis->fftnz, + rho_basis->gamma_only, kgmat.data(), gtr.data(), nop); +#else + get_ixyz2ipw(rho_basis, rho_basis->ig2isz, fftixy2is, ixyz2ipw); + symm.rhog_symmetry(rhog_part, ixyz2ipw, rho_basis->nx, rho_basis->ny, rho_basis->nz, + rho_basis->fftnx, rho_basis->fftny, rho_basis->fftnz, + rho_basis->gamma_only, kgmat.data(), gtr.data(), nop); +#endif +#ifdef __MPI + } + + // (4) send the result to other procs in the same pool + rhog_piece_to_all(rho_basis, rhogtot, rhog_part); +#endif + return; +} + +void psymmg_soc(std::complex* rhog_x, std::complex* rhog_y, + std::complex* rhog_z, const ModulePW::PW_Basis* rho_basis, ModuleSymmetry::Symmetry& symm) +{ + // build the per-operation spin-rotation matrices W(g) from the cartesian rotation + // gmatc(g) = direct_to_cartesian(gmatrix(g)) = latvec^-1 * gmatrix(g) * latvec. + std::function()> build_wspin = [&rho_basis, &symm]() { + const ModuleBase::Matrix3 latvec = rho_basis->latvec; + const ModuleBase::Matrix3 ilatvec = latvec.Inverse(); + // index [0,nrotk) unitary, [nrotk, nrotk+nrotk_anti) the spatial parts of the + // antiunitary elements Theta*g -- same layout as density_sym_ops(). + const int na = symm.magnetic_nspin4 ? symm.nrotk_anti : 0; + std::vector wspin(symm.nrotk + na); + for (int i = 0; i < symm.nrotk; ++i) + { + const ModuleBase::Matrix3 gmatc = ilatvec * symm.gmatrix[i] * latvec; + wspin[i] = ModuleSymmetry::SpinRotation::spin_so3(gmatc); + } + for (int j = 0; j < na; ++j) + { + const ModuleBase::Matrix3 gmatc = ilatvec * symm.gmatrix_anti[j] * latvec; + wspin[symm.nrotk + j] = ModuleSymmetry::SpinRotation::spin_so3(gmatc); + } + return wspin; + }; + + //(1) get fftixy2is and do Allreduce + std::vector fftixy2is_vec(rho_basis->fftnxy); + int* fftixy2is = fftixy2is_vec.data(); + rho_basis->getfftixy2is(fftixy2is); +#ifdef __MPI + Parallel_Reduce::reduce_pool(fftixy2is, rho_basis->fftnxy); + if(rho_basis->poolnproc>1) + { + for (int i=0;ifftnxy;++i) + { + fftixy2is[i]+=rho_basis->poolnproc-1; + } + } + + // (2) reduce all three spin components from the first pool. + std::vector> rhogtot_x_vec; + std::vector> rhogtot_y_vec; + std::vector> rhogtot_z_vec; + std::vector ig2isztot_vec; + std::complex* rhogtot_x = nullptr; + std::complex* rhogtot_y = nullptr; + std::complex* rhogtot_z = nullptr; + int* ig2isztot = nullptr; + if(GlobalV::RANK_IN_POOL == 0) + { + rhogtot_x_vec.resize(rho_basis->npwtot); + rhogtot_y_vec.resize(rho_basis->npwtot); + rhogtot_z_vec.resize(rho_basis->npwtot); + ig2isztot_vec.resize(rho_basis->npwtot); + rhogtot_x = rhogtot_x_vec.data(); + rhogtot_y = rhogtot_y_vec.data(); + rhogtot_z = rhogtot_z_vec.data(); + ig2isztot = ig2isztot_vec.data(); + } + // find max_npw + int max_npw=0; + for (int proc = 0; proc < rho_basis->poolnproc; ++proc) + { + if(rho_basis->npw_per[proc] > max_npw) + { + max_npw=rho_basis->npw_per[proc]; + } + } + reduce_to_fullrhog(rho_basis, rhogtot_x, rhog_x, ig2isztot, rho_basis->ig2isz, max_npw); + reduce_to_fullrhog(rho_basis, rhogtot_y, rhog_y, ig2isztot, rho_basis->ig2isz, max_npw); + reduce_to_fullrhog(rho_basis, rhogtot_z, rhog_z, ig2isztot, rho_basis->ig2isz, max_npw); + + // (3) get ixy2ipw and do rhog_symmetry_nspin4 on proc 0 of each pool + if(GlobalV::RANK_IN_POOL==0) + { +#endif + //init ixyz2ipw + std::vector ixyz2ipw_vec(rho_basis->fftnxyz); + int* ixyz2ipw = ixyz2ipw_vec.data(); + for(int i=0;ifftnxyz;++i) + { + ixyz2ipw[i]=-1; + } + std::vector wspin = build_wspin(); + std::vector kgmat; + std::vector> gtr; + std::vector trs_inv; + const int nop = symm.density_sym_ops(kgmat, gtr, trs_inv); +#ifdef __MPI + get_ixyz2ipw(rho_basis, ig2isztot, fftixy2is, ixyz2ipw); + symm.rhog_symmetry_nspin4(rhogtot_x, rhogtot_y, rhogtot_z, wspin.data(), ixyz2ipw, + rho_basis->nx, rho_basis->ny, rho_basis->nz, + rho_basis->fftnx, rho_basis->fftny, rho_basis->fftnz, + trs_inv.data(), kgmat.data(), gtr.data(), nop); +#else + get_ixyz2ipw(rho_basis, rho_basis->ig2isz, fftixy2is, ixyz2ipw); + symm.rhog_symmetry_nspin4(rhog_x, rhog_y, rhog_z, wspin.data(), ixyz2ipw, + rho_basis->nx, rho_basis->ny, rho_basis->nz, + rho_basis->fftnx, rho_basis->fftny, rho_basis->fftnz, + trs_inv.data(), kgmat.data(), gtr.data(), nop); +#endif +#ifdef __MPI + } + + rhog_piece_to_all(rho_basis, rhogtot_x, rhog_x); + rhog_piece_to_all(rho_basis, rhogtot_y, rhog_y); + rhog_piece_to_all(rho_basis, rhogtot_z, rhog_z); +#endif + return; +} + +} // namespace detail +} // namespace module_charge diff --git a/source/source_estate/module_charge/chg_symm_detail.h b/source/source_estate/module_charge/chg_symm_detail.h new file mode 100644 index 00000000000..a59da9f2d03 --- /dev/null +++ b/source/source_estate/module_charge/chg_symm_detail.h @@ -0,0 +1,39 @@ +#ifndef CHG_SYMM_DETAIL_H +#define CHG_SYMM_DETAIL_H + +#include + +#include "source_basis/module_pw/pw_basis.h" +#include "source_cell/module_symmetry/symmetry.h" + +/** + * @brief Internal reciprocal-space helpers for charge-density symmetrization. + * + * Not part of the public module_charge API: only chg_symm.cpp and + * chg_symm_detail.cpp are expected to include this header. + */ +namespace module_charge +{ +namespace detail +{ + +/** + * @brief Symmetrize one reciprocal-space density component. + */ +void psymmg(std::complex* rhog_part, + const ModulePW::PW_Basis* rho_basis, + ModuleSymmetry::Symmetry& symm); + +/** + * @brief Symmetrize the three coupled reciprocal-space spin components for nspin=4. + */ +void psymmg_soc(std::complex* rhog_x, + std::complex* rhog_y, + std::complex* rhog_z, + const ModulePW::PW_Basis* rho_basis, + ModuleSymmetry::Symmetry& symm); + +} // namespace detail +} // namespace module_charge + +#endif diff --git a/source/source_estate/module_charge/chg_tau.cpp b/source/source_estate/module_charge/chg_tau.cpp new file mode 100644 index 00000000000..214cb7123d5 --- /dev/null +++ b/source/source_estate/module_charge/chg_tau.cpp @@ -0,0 +1,114 @@ +#include "chg_tau.h" +#include "chg_uspp.h" + +#include "source_base/timer.h" +#include "source_base/tool_quit.h" + +namespace module_charge { +namespace detail { + +void mix_tau_recip(Charge* chr, + const int nspin, + const bool double_grid, + ModulePW::PW_Basis* rhopw, + ModulePW::PW_Basis* rhodpw, + Base_Mixing::Mixing* mixing, + Base_Mixing::Mixing_Data& tau_mdata, + Base_Mixing::Plain_Mixing* mixing_highf) +{ + ModuleBase::TITLE("module_charge", "mix_tau_recip"); + ModuleBase::timer::start("module_charge", "mix_tau_recip"); + + if (chr == nullptr) + { + ModuleBase::WARNING_QUIT("mix_tau_recip", "chr is null"); + } + if (rhopw == nullptr || rhodpw == nullptr) + { + ModuleBase::WARNING_QUIT("mix_tau_recip", "grid pointer is null"); + } + if (mixing == nullptr) + { + ModuleBase::WARNING_QUIT("mix_tau_recip", "mixing is null"); + } + if (nspin < 1) + { + ModuleBase::WARNING_QUIT("mix_tau_recip", "nspin must be >= 1"); + } + if (double_grid && mixing_highf == nullptr) + { + ModuleBase::WARNING_QUIT("mix_tau_recip", "mixing_highf is null when double_grid is on"); + } + + std::vector> kin_g(nspin * rhodpw->npw); + std::vector> kin_g_save(nspin * rhodpw->npw); + // FFT to get kin_g and kin_g_save + for (int is = 0; is < nspin; ++is) + { + rhodpw->real2recip(chr->kin_r[is], &kin_g[is * rhodpw->npw]); + rhodpw->real2recip(chr->kin_r_save[is], &kin_g_save[is * rhodpw->npw]); + } + + // RAII owners for the smooth / high-frequency parts on the double grid; + // raw pointers below alias these vectors when double_grid is on, or + // alias kin_g[_save] directly when double_grid is off so the mixing + // mutates the dense buffer in place. + std::vector> tau_sg_in; + std::vector> tau_sg_out; + std::vector> tau_hf_in; + std::vector> tau_hf_out; + std::complex* taugs_in = nullptr; + std::complex* taugs_out = nullptr; + std::complex* taughf_in = nullptr; + std::complex* taughf_out = nullptr; + + if (double_grid) + { + const int npw_smooth = rhopw->npw; + const int npw_dense = rhodpw->npw; + tau_sg_in.resize(nspin * npw_smooth); + tau_hf_in.resize(nspin * (npw_dense - npw_smooth)); + tau_sg_out.resize(nspin * npw_smooth); + tau_hf_out.resize(nspin * (npw_dense - npw_smooth)); + module_charge::split_dgrid(kin_g_save.data(), tau_sg_in, tau_hf_in, + nspin, npw_smooth, npw_dense); + module_charge::split_dgrid(kin_g.data(), tau_sg_out, tau_hf_out, + nspin, npw_smooth, npw_dense); + taugs_in = tau_sg_in.data(); + taughf_in = tau_hf_in.data(); + taugs_out = tau_sg_out.data(); + taughf_out = tau_hf_out.data(); + } + else + { + taugs_in = kin_g_save.data(); + taugs_out = kin_g.data(); + } + + // Note: there is no kerker modification for tau because I'm not sure + // if we should have it. If necessary we can try it in the future. + mixing->push_data(tau_mdata, taugs_in, taugs_out, nullptr, false); + mixing->mix_data(tau_mdata, taugs_out); + + if (double_grid) + { + // simple mixing for high_frequencies + const int ndimhf = (rhodpw->npw - rhopw->npw) * nspin; + mixing_highf->plain_mix(taughf_out, taughf_in, taughf_out, ndimhf, nullptr); + + // combine smooth part and high_frequency part + module_charge::merge_dgrid(kin_g.data(), tau_sg_out, tau_hf_out, + nspin, rhopw->npw, rhodpw->npw); + } + + // kin_g to kin_r + for (int is = 0; is < nspin; is++) + { + rhodpw->recip2real(&kin_g[is * rhodpw->npw], chr->kin_r[is]); + } + + ModuleBase::timer::end("module_charge", "mix_tau_recip"); +} + +} // namespace detail +} // namespace module_charge diff --git a/source/source_estate/module_charge/chg_tau.h b/source/source_estate/module_charge/chg_tau.h new file mode 100644 index 00000000000..739e2211a70 --- /dev/null +++ b/source/source_estate/module_charge/chg_tau.h @@ -0,0 +1,44 @@ +#ifndef CHG_TAU_H +#define CHG_TAU_H + +// Mixing of the kinetic energy density (tau) in reciprocal space. +// The implementation lives in chg_tau.cpp; this header exposes the +// free function so that Charge_Mixing::mix_rho_recip can call it +// without pulling in the broader chg_rho_detail.h helpers. + +#include "charge.h" +#include "source_base/module_mixing/mixing.h" +#include "source_base/module_mixing/plain_mixing.h" + +namespace module_charge +{ +namespace detail +{ + +/** + * @brief Mix kinetic energy density in reciprocal space. + * Handles the double-grid split/merge for the smooth and + * high-frequency parts, DIIS mixing of the smooth part, and + * plain mixing of the high-frequency part. + * @param chr pointer to Charge object (must have kin_r/kin_r_save) + * @param nspin number of spins + * @param double_grid whether double grid is used + * @param rhopw smooth grid + * @param rhodpw dense grid (same as rhopw when double_grid is off) + * @param mixing DIIS mixing object + * @param tau_mdata mixing data for tau + * @param mixing_highf plain mixing for high-frequency part (may be null when double_grid is off) + */ +void mix_tau_recip(Charge* chr, + const int nspin, + const bool double_grid, + ModulePW::PW_Basis* rhopw, + ModulePW::PW_Basis* rhodpw, + Base_Mixing::Mixing* mixing, + Base_Mixing::Mixing_Data& tau_mdata, + Base_Mixing::Plain_Mixing* mixing_highf); + +} // namespace detail +} // namespace module_charge + +#endif // CHG_TAU_H diff --git a/source/source_estate/module_charge/chg_tools.cpp b/source/source_estate/module_charge/chg_tools.cpp new file mode 100644 index 00000000000..733fbea6f49 --- /dev/null +++ b/source/source_estate/module_charge/chg_tools.cpp @@ -0,0 +1,365 @@ +#include "chg_tools.h" + +#include +#include +#include +#include +#include + +#include "source_base/complexmatrix.h" +#include "source_base/constants.h" +#include "source_base/math_integral.h" +#include "source_base/math_sphbes.h" +#include "source_base/parallel_reduce.h" +#include "source_base/timer.h" +#include "source_base/tool_quit.h" +#include "source_base/tool_threading.h" +#include "source_base/tool_title.h" +#include "source_basis/module_pw/pw_basis.h" +#include "source_cell/unitcell.h" + +#include +#include +#include + +namespace module_charge +{ + +double sum_rho(double* const* rho, + const int nspin0, + const int nrxx, + const double omega, + const int nxyz) +{ + ModuleBase::TITLE("module_charge", "sum_rho"); + + std::vector sum_is(nspin0, 0.0); + for (int is = 0; is < nspin0; is++) + { + for (int ir = 0; ir < nrxx; ir++) + { + sum_is[is] += rho[is][ir]; + } + } + + // multiply the sum of charge density by a factor + const double factor = omega / static_cast(nxyz); + for (int is = 0; is < nspin0; is++) + { + sum_is[is] *= factor; + } + +#ifdef __MPI + Parallel_Reduce::reduce_pool(sum_is.data(), nspin0); +#endif + + double sum_rho = 0.0; + for (int is = 0; is < nspin0; is++) + { + sum_rho += sum_is[is]; + } + + // mohan fixed bug 2010-01-18, + // sum_rho may be smaller than 1, like Na bcc. + // A NaN sum never satisfies sum_rho <= 0.1, so check it explicitly. + if (std::isnan(sum_rho) || sum_rho <= 0.1) + { + // Diagnostic context: report per-spin electron counts plus rho + // extrema. min/max and the number of negative grid points help + // distinguish "rho driven negative by mixing" from "rho collapsed" + // when reproducing intermittent CI failures (e.g. issue #7794). + // For nspin0 == 1 the per-spin value equals the total; the extrema + // still carry information there. + double rho_min = std::numeric_limits::max(); + double rho_max = std::numeric_limits::lowest(); + long n_neg = 0; + for (int is = 0; is < nspin0; ++is) + { + for (int ir = 0; ir < nrxx; ++ir) + { + const double v = rho[is][ir]; + rho_min = std::min(rho_min, v); + rho_max = std::max(rho_max, v); + n_neg += (v < 0.0) ? 1 : 0; + } + } + // Rank-local on purpose: pool sums may differ in the last bits (or a + // NaN may be present in a single pool), so pools can disagree on + // entering this branch and any MPI collective here could hang. + std::ostringstream diag; + diag << "\nCan't find even an electron!" + << "\nmodule_charge::sum_rho diagnostic:" + << " nspin0 = " << nspin0 + << " nrxx = " << nrxx + << " omega = " << omega + << " nxyz = " << nxyz + << " per-spin sum_rho ="; + for (int is = 0; is < nspin0; ++is) + { + diag << " " << sum_is[is]; + } + diag << " (total = " << sum_rho << ")" + << " rank-local rho min = " << rho_min + << " max = " << rho_max + << " negative grid points = " << n_neg; + ModuleBase::WARNING_QUIT("module_charge::sum_rho", diag.str()); + } + + return sum_rho; +} + +double cal_rho2ne(const double* rho_in, + const int nrxx, + const double omega, + const int nxyz) +{ + assert(nxyz > 0); // mohan add 2025-12-02 + + double ne = 0.0; + for (int ir = 0; ir < nrxx; ir++) + { + ne += rho_in[ir]; + } +#ifdef __MPI + Parallel_Reduce::reduce_pool(ne); +#endif + ne = ne * omega / static_cast(nxyz); + + return ne; +} + +void check_rho(double* const* rho, + const int nspin, + const int nrxx, + const double omega, + const int nxyz, + const double nelec) +{ + ModuleBase::TITLE("module_charge", "check_rho"); + + assert(rho != nullptr); + assert(nelec > 0.0); + + if (nspin == 1 || nspin == 4) + { + const double ne = cal_rho2ne(rho[0], nrxx, omega, nxyz); + if (std::abs(ne - nelec) > 1.0e-6) + { + ModuleBase::WARNING("Charge", "Charge is not equal to the number of electrons!"); + } + } + else if (nspin == 2) + { + // for spin up + const double ne_up = cal_rho2ne(rho[0], nrxx, omega, nxyz); + if (ne_up < 0.0) + { + ModuleBase::WARNING_QUIT("Charge", + "Number of spin-down electrons set in starting magnetization exceeds all available."); + } + // for spin down + const double ne_dn = cal_rho2ne(rho[1], nrxx, omega, nxyz); + if (ne_dn < 0.0) + { + ModuleBase::WARNING_QUIT("Charge", + "Number of spin-up electrons set in starting magnetization exceeds all available."); + } + // for total charge + if (std::abs(ne_up + ne_dn - nelec) > 1.0e-6) + { + ModuleBase::WARNING("Charge", "Charge is not equal to the number of electrons!"); + } + } +} + +void non_linear_core_correction(const NlcCtx& ctx, + double* rhocg) +{ + ModuleBase::TITLE("module_charge", "drhoc"); + + const bool numeric = ctx.numeric; + const double omega = ctx.omega; + const double tpiba2 = ctx.tpiba2; + const int mesh = ctx.mesh; + const double* r = ctx.r; + const double* rab = ctx.rab; + const double* rhoc = ctx.rhoc; + const double* gg_uniq = ctx.gg_uniq; + const int ngg = ctx.ngg; + + // use labmda instead of repeating codes + const std::function kernel = [&](int num_threads, int thread_id) + { + + double gx = 0.0; + double rhocg1 = 0.0; + std::vector aux_vec; + + // here we compute the fourier transform is the charge in numeric form + if (numeric) + { + aux_vec.resize(mesh); + double* aux = aux_vec.data(); + // G=0 term + + int igl0 = 0; + if (gg_uniq [0] < 1.0e-8) + { + // single thread term + if (thread_id == 0) + { + for (int ir = 0;ir < mesh; ir++) + { + aux [ir] = r [ir] * r [ir] * rhoc [ir]; + } + ModuleBase::Integral::Simpson_Integral(mesh, aux, rab, rhocg1); + //rhocg [1] = fpi * rhocg1 / omega; + rhocg [0] = ModuleBase::FOUR_PI * rhocg1 / omega;//mohan modify 2008-01-19 + } + igl0 = 1; + } + + int igl_beg, igl_end; + // exclude igl0 + ModuleBase::TASK_DIST_1D(num_threads, thread_id, ngg - igl0, igl_beg, igl_end); + igl_beg += igl0; + igl_end += igl_beg; + + // G <> 0 term + for (int igl = igl_beg; igl < igl_end;igl++) + { + gx = sqrt(gg_uniq[igl] * tpiba2); + ModuleBase::Sphbes::Spherical_Bessel(mesh, r, gx, 0, aux); + for (int ir = 0;ir < mesh; ir++) + { + aux [ir] = r[ir] * r[ir] * rhoc [ir] * aux [ir]; + } // enddo + ModuleBase::Integral::Simpson_Integral(mesh, aux, rab, rhocg1); + rhocg [igl] = ModuleBase::FOUR_PI * rhocg1 / omega; + } // enddo + } + else + { + // here the case where the charge is in analytic form, + // check old version before 2008-12-9 + } + + }; // end kernel + + // do not use omp parallel when this function is already in parallel block + // + // it is called in parallel block in Forces::cal_force_cc, + // but not in other funtcion such as Stress_Func::stress_cc. + ModuleBase::TRY_OMP_PARALLEL(kernel); + + return; +} + +// computes the core charge on the real space 3D mesh. +void set_rho_core(const UnitCell& ucell, + const ModuleBase::ComplexMatrix& structure_factor, + const bool* numeric, + double* rho_core, + std::complex* rhog_core, + const ModulePW::PW_Basis& rhopw) +{ + ModuleBase::TITLE("module_charge", "set_rho_core"); + ModuleBase::timer::start("module_charge", "set_rho_core"); + + bool bl = false; + for (int it = 0; it < ucell.ntype; it++) + { + if (ucell.atoms[it].ncpp.nlcc) + { + bl = true; + break; + } + } + + if (!bl) + { + std::fill(rho_core, rho_core + rhopw.nrxx, 0.0); + ModuleBase::timer::end("module_charge", "set_rho_core"); + return; + } + + std::vector rhocg(rhopw.ngg, 0.0); + + // three dimension. + std::vector> vg(rhopw.npw); + + for (int it = 0; it < ucell.ntype; it++) + { + if (ucell.atoms[it].ncpp.nlcc) + { + assert(numeric != nullptr); +//---------------------------------------------------------- +// EXPLAIN : drhoc compute the radial fourier transform for +// each shell of g vec +//---------------------------------------------------------- + NlcCtx nlc_ctx{ + numeric[it], + ucell.omega, + ucell.tpiba2, + ucell.atoms[it].ncpp.msh, + ucell.atoms[it].ncpp.r.data(), + ucell.atoms[it].ncpp.rab.data(), + ucell.atoms[it].ncpp.rho_atc.data(), + rhopw.gg_uniq, + rhopw.ngg + }; + non_linear_core_correction(nlc_ctx, rhocg.data()); +//---------------------------------------------------------- +// EXPLAIN : multiply by the structure factor and sum +//---------------------------------------------------------- + for (int ig = 0; ig < rhopw.npw; ig++) + { + vg[ig] += structure_factor(it, ig) * rhocg[rhopw.ig2igg[ig]]; + } + } + } + + // for tmp use. + for (int ig = 0; ig < rhopw.npw; ig++) + { + rhog_core[ig] = vg[ig]; + } + + rhopw.recip2real(vg.data(), rho_core); + + // test on the charge and computation of the core energy + double rhoima = 0.0; + double rhoneg = 0.0; + for (int ir = 0; ir < rhopw.nrxx; ir++) + { + rhoneg += std::min(0.0, rhopw.fft_bundle.get_auxr_data()[ir].real()); + rhoima += std::abs(rhopw.fft_bundle.get_auxr_data()[ir].imag()); + // NOTE: Core charge is computed in reciprocal space and brought to real + // space by FFT. For non smooth core charges (or insufficient cut-off) + // this may result in negative values in some grid points. + // Up to October 1999 the core charge was forced to be positive definite. + // This induces an error in the force, and probably stress, calculation if + // the number of grid points where the core charge would be otherwise neg + // is large. The error disappears for sufficiently high cut-off, but may be + // rather large and it is better to leave the core charge as it is. + // If you insist to have it positive definite (with the possible problems + // mentioned above) uncomment the following lines. SdG, Oct 15 1999 + } + +#ifdef __MPI + // mohan fix bug 2011-04-03 + Parallel_Reduce::reduce_pool(rhoneg); + Parallel_Reduce::reduce_pool(rhoima); +#endif + + // mohan changed 2010-2-2, make this same as in atomic_rho. + // still lack something...... + rhoneg /= rhopw.nxyz * ucell.omega; + rhoima /= rhopw.nxyz * ucell.omega; + + // calculate core_only exch-corr energy etxcc=E_xc[rho_core] if required + // The term was present in previous versions of the code but it shouldn't + ModuleBase::timer::end("module_charge", "set_rho_core"); +} + +} // namespace module_charge diff --git a/source/source_estate/module_charge/chg_tools.h b/source/source_estate/module_charge/chg_tools.h new file mode 100644 index 00000000000..9be5cf49047 --- /dev/null +++ b/source/source_estate/module_charge/chg_tools.h @@ -0,0 +1,88 @@ +#ifndef CHG_TOOLS_H +#define CHG_TOOLS_H + +// Free numerical kernels extracted from the Charge class so that the +// density math (summation, electron-count integration, non-linear core +// correction) can be tested and reused without dragging in Charge's state. +// All grid / geometry inputs are passed explicitly instead of being read +// from Charge members or globals. + +#include + +class UnitCell; +namespace ModuleBase +{ +class ComplexMatrix; +} +namespace ModulePW +{ +class PW_Basis; +} + +namespace module_charge +{ + +// Compute the core charge (non-linear core correction) on the real-space +// 3D mesh. rho_core / rhog_core are the output buffers previously owned by +// Charge; rhopw supplies the reciprocal-grid geometry and FFT backend. +void set_rho_core(const UnitCell& ucell, + const ModuleBase::ComplexMatrix& structure_factor, + const bool* numeric, + double* rho_core, + std::complex* rhog_core, + const ModulePW::PW_Basis& rhopw); + +// Sum the (spin-resolved) charge density over the real-space grid and +// convert it to a total charge using the cell volume and grid size. +// rho[is][ir] is the density for spin channel is. nspin0 is the number of +// spin channels to include (2 for nspin==2, 1 otherwise). +double sum_rho(double* const* rho, + const int nspin0, + const int nrxx, + const double omega, + const int nxyz); + +// Integrate a single spin channel rho_in over the grid to obtain the +// electron number, scaled by omega / nxyz. Reduction over the pool is +// performed internally under __MPI. +double cal_rho2ne(const double* rho_in, + const int nrxx, + const double omega, + const int nxyz); + +// Check that the charge density integrates to the target electron number. +// rho[is][ir] is the density for spin channel is. For nspin == 1 or 4 the +// total density in rho[0] is compared with nelec; for nspin == 2 the spin-up +// (rho[0]) and spin-down (rho[1]) channels must each integrate to a +// non-negative number and their sum must match nelec. A mismatch emits a +// warning; a negative spin-channel electron number aborts the run. +void check_rho(double* const* rho, + const int nspin, + const int nrxx, + const double omega, + const int nxyz, + const double nelec); + +// Non-linear core correction: Fourier transform of the (numeric) core +// charge. gg_uniq / ngg supply the reciprocal grid shells previously read +// from Charge::rhopw. +/// Radial grid and density data for non-linear core correction +struct NlcCtx +{ + bool numeric; ///< whether the pseudo-potential has numeric data + double omega; ///< cell volume + double tpiba2; ///< 2*pi/alat squared + int mesh; ///< radial mesh size + const double* r; ///< radial grid points + const double* rab; ///< radial grid weights + const double* rhoc; ///< core charge on the radial grid + const double* gg_uniq; ///< unique |G|^2 values + int ngg; ///< number of unique |G|^2 shells +}; + +void non_linear_core_correction(const NlcCtx& ctx, + double* rhocg); + +} // namespace module_charge + +#endif // CHG_TOOLS_H diff --git a/source/source_estate/module_charge/chg_uspp.cpp b/source/source_estate/module_charge/chg_uspp.cpp new file mode 100644 index 00000000000..fec865212d9 --- /dev/null +++ b/source/source_estate/module_charge/chg_uspp.cpp @@ -0,0 +1,129 @@ +#include "chg_uspp.h" + +#include + +#include "source_base/timer.h" +#include "source_base/tool_title.h" +#include "source_base/tool_quit.h" + +namespace module_charge +{ + +void split_dgrid(const std::complex* data_d, + std::vector>& data_s, + std::vector>& data_hf, + int nspin, + int npw_smooth, + int npw_dense) +{ + ModuleBase::TITLE("module_charge", "split_dgrid"); + ModuleBase::timer::start("module_charge", "split_dgrid"); + + if (data_d == nullptr) + { + ModuleBase::WARNING_QUIT("module_charge::split_dgrid", + "input dense data pointer is null"); + } + if (nspin < 1) + { + ModuleBase::WARNING_QUIT("module_charge::split_dgrid", + "nspin must be >= 1"); + } + if (npw_smooth < 0 || npw_dense < npw_smooth) + { + ModuleBase::WARNING_QUIT("module_charge::split_dgrid", + "require npw_smooth >= 0 and npw_dense >= npw_smooth"); + } + + const int npw_hf = npw_dense - npw_smooth; + const std::size_t size_s = nspin * npw_smooth; + const std::size_t size_hf = nspin * npw_hf; + if (data_s.size() != size_s) + { + ModuleBase::WARNING_QUIT("module_charge::split_dgrid", + "data_s size does not match nspin * npw_smooth"); + } + if (data_hf.size() != size_hf) + { + ModuleBase::WARNING_QUIT("module_charge::split_dgrid", + "data_hf size does not match nspin * (npw_dense - npw_smooth)"); + } + + for (int is = 0; is < nspin; ++is) + { + const std::complex* src = data_d + is * npw_dense; + if (npw_smooth > 0) + { + std::memcpy(data_s.data() + is * npw_smooth, src, + npw_smooth * sizeof(std::complex)); + } + if (npw_hf > 0) + { + std::complex* dst = data_hf.data() + is * npw_hf; + std::memcpy(dst, src + npw_smooth, + npw_hf * sizeof(std::complex)); + } + } + + ModuleBase::timer::end("module_charge", "split_dgrid"); +} + +void merge_dgrid(std::complex* data_d, + const std::vector>& data_s, + const std::vector>& data_hf, + int nspin, + int npw_smooth, + int npw_dense) +{ + ModuleBase::TITLE("module_charge", "merge_dgrid"); + ModuleBase::timer::start("module_charge", "merge_dgrid"); + + if (data_d == nullptr) + { + ModuleBase::WARNING_QUIT("module_charge::merge_dgrid", + "output dense data pointer is null"); + } + if (nspin < 1) + { + ModuleBase::WARNING_QUIT("module_charge::merge_dgrid", + "nspin must be >= 1"); + } + if (npw_smooth < 0 || npw_dense < npw_smooth) + { + ModuleBase::WARNING_QUIT("module_charge::merge_dgrid", + "require npw_smooth >= 0 and npw_dense >= npw_smooth"); + } + + const int npw_hf = npw_dense - npw_smooth; + const std::size_t size_s = nspin * npw_smooth; + const std::size_t size_hf = nspin * npw_hf; + if (data_s.size() != size_s) + { + ModuleBase::WARNING_QUIT("module_charge::merge_dgrid", + "data_s size does not match nspin * npw_smooth"); + } + if (data_hf.size() != size_hf) + { + ModuleBase::WARNING_QUIT("module_charge::merge_dgrid", + "data_hf size does not match nspin * (npw_dense - npw_smooth)"); + } + + for (int is = 0; is < nspin; ++is) + { + std::complex* dst = data_d + is * npw_dense; + if (npw_smooth > 0) + { + std::memcpy(dst, data_s.data() + is * npw_smooth, + npw_smooth * sizeof(std::complex)); + } + if (npw_hf > 0) + { + std::memcpy(dst + npw_smooth, data_hf.data() + is * npw_hf, + npw_hf * sizeof(std::complex)); + } + } + + ModuleBase::timer::end("module_charge", "merge_dgrid"); +} + +} // namespace module_charge diff --git a/source/source_estate/module_charge/chg_uspp.h b/source/source_estate/module_charge/chg_uspp.h new file mode 100644 index 00000000000..09196bbbacb --- /dev/null +++ b/source/source_estate/module_charge/chg_uspp.h @@ -0,0 +1,73 @@ +#ifndef CHG_USPP_H +#define CHG_USPP_H + +// Stateless double-grid split/merge helpers extracted from Charge_Mixing. +// +// "dgrid" = double grid, the dense/smooth grid pair used by ultrasoft (USPP) +// and PAW calculations to capture the high-frequency tail of the +// augmentation charge beyond the smooth (soft) plane-wave grid. The dense +// grid (npw_dense) is the union of the smooth grid (npw_smooth) and the +// high-frequency tail (npw_dense - npw_smooth). +// +// These functions do not read Charge_Mixing members or PARAM/GlobalV; all +// inputs are passed explicitly. Memory is managed by the caller through +// std::vector, so no new/delete pair is needed and no clean-up function +// exists. + +#include +#include + +namespace module_charge +{ + +/** + * @brief Split dense reciprocal-space data into smooth and high-frequency + * parts on the USPP double grid. + * + * For each spin channel, the first npw_smooth entries of data_d are copied + * into data_s and the remaining (npw_dense - npw_smooth) entries are copied + * into data_hf. No aliasing is performed: both output vectors own their + * storage and must be pre-sized by the caller. + * + * @param data_d dense input, shape [nspin * npw_dense], non-null + * @param data_s smooth output, pre-sized to nspin * npw_smooth + * @param data_hf high-frequency output, pre-sized to + * nspin * (npw_dense - npw_smooth); zero-size is allowed + * when npw_dense == npw_smooth + * @param nspin number of spin channels, >= 1 + * @param npw_smooth smooth grid npw, >= 0 + * @param npw_dense dense grid npw, >= npw_smooth + */ +void split_dgrid(const std::complex* data_d, + std::vector>& data_s, + std::vector>& data_hf, + int nspin, + int npw_smooth, + int npw_dense); + +/** + * @brief Merge smooth and high-frequency parts back into dense reciprocal- + * space data. Inverse of split_dgrid. + * + * Vectors are not cleared; the caller may reuse them or let them go out of + * scope. The output data_d must be pre-allocated by the caller with size + * nspin * npw_dense. + * + * @param data_d dense output, shape [nspin * npw_dense], non-null + * @param data_s smooth input, sized to nspin * npw_smooth + * @param data_hf high-frequency input, sized to + * nspin * (npw_dense - npw_smooth) + * @param nspin number of spin channels, >= 1 + * @param npw_smooth smooth grid npw, >= 0 + * @param npw_dense dense grid npw, >= npw_smooth + */ +void merge_dgrid(std::complex* data_d, + const std::vector>& data_s, + const std::vector>& data_hf, + int nspin, + int npw_smooth, + int npw_dense); + +} // namespace module_charge + +#endif // CHG_USPP_H diff --git a/source/source_estate/module_charge/chgmixing.cpp b/source/source_estate/module_charge/chgmixing.cpp deleted file mode 100644 index 4dc059e3e56..00000000000 --- a/source/source_estate/module_charge/chgmixing.cpp +++ /dev/null @@ -1,233 +0,0 @@ -#include "source_estate/module_charge/chgmixing.h" - -#include "source_base/parallel_comm.h" -#include "source_estate/update_pot.h" -#include "source_lcao/module_deltaspin/spin_constrain.h" - -void module_charge::chgmixing_ks(const int iter, // scf iteration number - UnitCell& ucell, - elecstate::ElecState* pelec, - Charge &chr, // charge density - Charge_Mixing* p_chgmix, // charge mixing class - const int nrxx, // charge density - double &drho, // charge density deviation - bool &oscillate_esolver, // whether the esolver has oscillation of charge density - bool &conv_esolver, - const double &hsolver_error, - const double &scf_thr, - const double &scf_ene_thr, - const bool converged_u, // mohan add 2025-11-06 - const Input_para& inp) // input parameters -{ - - if (PARAM.globalv.ks_run) - { - // mixing will restart at p_chgmix->mixing_restart steps - if (drho <= inp.mixing_restart && inp.mixing_restart > 0.0 - && p_chgmix->mixing_restart_step > iter) - { - p_chgmix->mixing_restart_step = iter + 1; - } - - if (inp.scf_os_stop) // if oscillation is detected, SCF will stop - { - oscillate_esolver = p_chgmix->if_scf_oscillate(iter, drho, - inp.scf_os_ndim, inp.scf_os_thr); - } - - // drho will be 0 at p_chgmix->mixing_restart step, which is - // not ground state - bool not_restart_step = !(iter == p_chgmix->mixing_restart_step && inp.mixing_restart > 0.0); - - conv_esolver = (drho < scf_thr && not_restart_step && converged_u); - - // add energy threshold for SCF convergence - if (scf_ene_thr > 0.0) - { - // calculate energy of output charge density - elecstate::update_pot(ucell, pelec, chr, conv_esolver); - pelec->cal_energies(2); // 2 means Kohn-Sham functional - // now, etot_old is the energy of input density, while etot is the energy of output density - pelec->f_en.etot_delta = pelec->f_en.etot - pelec->f_en.etot_old; - // output etot_delta - GlobalV::ofs_running << " DeltaE_womix = " << pelec->f_en.etot_delta * ModuleBase::Ry_to_eV << " eV" - << std::endl; - if (iter > 1 && conv_esolver == 1) // only check when density is converged - { - // update the convergence flag - conv_esolver - = (std::abs(pelec->f_en.etot_delta * ModuleBase::Ry_to_eV) < scf_ene_thr); - } - } - - // If drho < hsolver_error in the first iter or drho < scf_thr, we - // do not change rho. - if (drho < hsolver_error || conv_esolver || inp.calculation == "nscf") - { - if (drho < hsolver_error) - { - GlobalV::ofs_warning << " drho < hsolver_error, keep " - "charge density unchanged." - << std::endl; - } - } - else - { - //----------charge mixing--------------- - // mixing will restart after p_chgmix->mixing_restart - // steps - if (inp.mixing_restart > 0 && iter == p_chgmix->mixing_restart_step - 1 - && drho <= inp.mixing_restart) - { - // do not mix charge density - } - else - { - p_chgmix->mix_rho(&chr); // update chr->rho by mixing - } - if (inp.scf_thr_type == 2) - { - chr.renormalize_rho(); // renormalize rho in R-space would - // induce a error in K-space - } - //----------charge mixing done----------- - } - } - -#ifdef __MPI - MPI_Bcast(&drho, 1, MPI_DOUBLE, 0, BP_WORLD); - - // change MPI_DOUBLE to MPI_C_BOOL, mohan 2025-04-13 - MPI_Bcast(&conv_esolver, 1, MPI_C_BOOL, 0, BP_WORLD); - - assert(nrxx>=0); // mohan add 2025-10-18 - MPI_Bcast(chr.rho[0], nrxx, MPI_DOUBLE, 0, BP_WORLD); -#endif - - // mohan move the following code here, 2025-10-18 - // SCF restart information - if (PARAM.inp.mixing_restart > 0 - && iter == p_chgmix->mixing_restart_step - 1 - && iter != PARAM.inp.scf_nmax) - { - p_chgmix->mixing_restart_last = iter; - std::cout << " SCF restart after this step!" << std::endl; - } - - return; -} - - -void module_charge::chgmixing_ks_pw(const int iter, // scf iteration number - Charge_Mixing* p_chgmix, // charge mixing class - Plus_U_Base& dftu, - const Input_para& inp) // input parameters -{ - ModuleBase::TITLE("module_charge", "chgmixing_ks_pw"); - - if (iter == 1) - { - p_chgmix->init_mixing(); - p_chgmix->mixing_restart_step = inp.scf_nmax + 1; - if (inp.dft_plus_u && dftu.has_occ_mixer()) - { - // allocate memory for uom_mdata sized to the flat occupation buffer - p_chgmix->allocate_mixing_uom(dftu.occ_mixer().flat_size()); - } - } - - // For mixing restart - if (iter == p_chgmix->mixing_restart_step && inp.mixing_restart > 0.0) - { - p_chgmix->init_mixing(); - p_chgmix->mixing_restart_count++; - - if (inp.dft_plus_u) - { - if (dftu.get_uramping() > 0.01 && !dftu.u_converged()) - { - p_chgmix->mixing_restart_step = inp.scf_nmax + 1; - } - if (dftu.get_uramping() > 0.01) - { - bool do_uramping = true; - if (inp.sc_mag_switch) - { - spinconstrain::SpinConstrain>& sc - = spinconstrain::SpinConstrain>::getScInstance(); - if (!sc.mag_converged()) // skip uramping if mag not converged - { - do_uramping = false; - } - } - if (do_uramping) - { - dftu.uramping_update(); // update U by uramping if uramping > 0.01 - std::cout << " U-Ramping! Current U = "; - for (int i = 0; i < dftu.get_num_u_types(); i++) - { - std::cout << dftu.get_u_current(i) * ModuleBase::Ry_to_eV << " "; - } - std::cout << " eV " << std::endl; - } - } - } - } - - return; -} - -void module_charge::chgmixing_ks_lcao(const int iter, // scf iteration number - Charge_Mixing* p_chgmix, // charge mixing class - Plus_U_Base& dftu, - const int nnr, // dimension of density matrix - const Input_para& inp) // input parameters -{ - ModuleBase::TITLE("module_charge", "chgmixing_ks_lcao"); - - if (iter == 1) - { - p_chgmix->mix_reset(); // init mixing - p_chgmix->mixing_restart_step = inp.scf_nmax + 1; - p_chgmix->mixing_restart_count = 0; - // this output will be removed once the feeature is stable - if (dftu.get_uramping() > 0.01) - { - std::cout << " U-Ramping! Current U = "; - for (int i = 0; i < dftu.get_num_u_types(); i++) - { - std::cout << dftu.get_u_current(i) * ModuleBase::Ry_to_eV << " "; - } - std::cout << " eV " << std::endl; - } - } - - // for mixing restart - if (iter == p_chgmix->mixing_restart_step && inp.mixing_restart > 0.0) - { - p_chgmix->init_mixing(); - p_chgmix->mixing_restart_count++; - if (inp.dft_plus_u) - { - dftu.uramping_update(); // update U by uramping if uramping > 0.01 - if (dftu.get_uramping() > 0.01) - { - std::cout << " U-Ramping! Current U = "; - for (int i = 0; i < dftu.get_num_u_types(); i++) - { - std::cout << dftu.get_u_current(i) * ModuleBase::Ry_to_eV << " "; - } - std::cout << " eV " << std::endl; - } - if (dftu.get_uramping() > 0.01 && !dftu.u_converged()) - { - p_chgmix->mixing_restart_step = inp.scf_nmax + 1; - } - } - if (inp.mixing_dmr) // for mixing_dmr - { - // allocate memory for dmr_mdata - p_chgmix->allocate_mixing_dmr(nnr); - } - } -} diff --git a/source/source_estate/module_charge/chgmixing.h b/source/source_estate/module_charge/chgmixing.h deleted file mode 100644 index 6b751128e36..00000000000 --- a/source/source_estate/module_charge/chgmixing.h +++ /dev/null @@ -1,43 +0,0 @@ -#ifndef CHGMIXING_H -#define CHGMIXING_H - -#include "source_estate/elecstate.h" // use pelec -#include "source_estate/module_charge/charge.h" // use chr -#include "source_estate/module_charge/charge_mixing.h" // use p_chgmix -#include "source_io/module_parameter/input_parameter.h" // use Input_para -#include "source_cell/unitcell.h" -#include "source_pw/module_pwdft/dftu_base.h" - -namespace module_charge -{ - -void chgmixing_ks(const int iter, // scf iteration number - UnitCell& ucell, - elecstate::ElecState* pelec, - Charge &chr, // charge density - Charge_Mixing* p_chgmix, // charge mixing class - const int nrxx, // charge density - double &drho, // charge density deviation - bool &oscillate_esolver, // whether the esolver has oscillation of charge density - bool &conv_esolver, - const double &hsolver_error, - const double &scf_thr, - const double &scf_ene_thr, - const bool converged_u, // mohan add 2025-11-06 - const Input_para& inp); // input parameters - -void chgmixing_ks_pw(const int iter, - Charge_Mixing* p_chgmix, - Plus_U_Base& dftu, - const Input_para& inp); // input parameters - -void chgmixing_ks_lcao(const int iter, // scf iteration number - Charge_Mixing* p_chgmix, // charge mixing class - Plus_U_Base& dftu, - const int nnr, // dimension of density matrix - const Input_para& inp); // input parameters - -} - - -#endif diff --git a/source/source_estate/module_charge/mix_precond.cpp b/source/source_estate/module_charge/mix_precond.cpp deleted file mode 100644 index a0a32fef97c..00000000000 --- a/source/source_estate/module_charge/mix_precond.cpp +++ /dev/null @@ -1,176 +0,0 @@ -#include "charge_mixing.h" - -#include "source_io/module_parameter/parameter.h" -#include "source_base/timer.h" - -void Charge_Mixing::Kerker_screen_recip(std::complex* drhog) -{ - ModuleBase::TITLE("Charge_Mixing", "Kerker_screen_recip"); - - if (this->mixing_gg0 <= 0.0 || this->mixing_beta <= 0.1) - { - return; - } - - ModuleBase::timer::start("Charge_Mixing", "Kerker_screen_recip"); - - const int nspin = PARAM.inp.nspin; - - double fac = 0.0; - double gg0 = 0.0; - double amin = 0.0; - - /// consider a resize for mixing_angle - int resize_tmp = 1; - if (nspin == 4 && this->mixing_angle > 0) - { - resize_tmp = 2; - } - - /// implement Kerker for density and magnetization separately - for (int is = 0; is < nspin / resize_tmp; ++is) - { - const int is_idx = is * this->rhopw->npw; - /// new mixing method only support nspin=2 not nspin=4 - if (is >= 1) - { - if (this->mixing_gg0_mag <= 0.0001 || this->mixing_beta_mag <= 0.1) - { -#ifdef __DEBUG - assert(is == 1); // make sure break works -#endif - double is_mag = nspin - 1; - //for (int ig = 0; ig < this->rhopw->npw * is_mag; ig++) - //{ - // drhog[is_idx + ig] *= 1; - //} - break; - } - fac = this->mixing_gg0_mag; - amin = this->mixing_beta_mag; - } - else - { - fac = this->mixing_gg0; - amin = this->mixing_beta; - } - - gg0 = std::pow(fac * ModuleBase::BOHR_TO_A / *this->tpiba, 2); - - const double gg0_amin = this->mixing_gg0_min / amin; - -#ifdef _OPENMP -#pragma omp parallel for schedule(static, 512) -#endif - for (int ig = 0; ig < this->rhopw->npw; ++ig) - { - double gg = this->rhopw->gg[ig]; - double filter_g = std::max(gg / (gg + gg0), gg0_amin); - drhog[is_idx + ig] *= filter_g; - } - } - - ModuleBase::timer::end("Charge_Mixing", "Kerker_screen_recip"); - return; -} - -void Charge_Mixing::Kerker_screen_real(double* drhor) -{ - ModuleBase::TITLE("Charge_Mixing", "Kerker_screen_real"); - - if (this->mixing_gg0 <= 0.0001 || this->mixing_beta <= 0.1) - { - return; - } - - ModuleBase::timer::start("Charge_Mixing", "Kerker_screen_real"); - - const int nspin = PARAM.inp.nspin; - assert(nspin==1 || nspin==2 || nspin==4); - - /// consider a resize for mixing_angle - int resize_tmp = 1; - if (nspin == 4 && this->mixing_angle > 0) - { - resize_tmp = 2; - } - - std::vector> drhog(this->rhopw->npw * nspin / resize_tmp); - std::vector drhor_filter(this->rhopw->nrxx * nspin / resize_tmp); - - for (int is = 0; is < nspin / resize_tmp; ++is) - { - // Note after this process some G which is higher than Gmax will be filtered. - // Thus we cannot use Kerker_screen_recip(drhog.data()) directly after it. - this->rhopw->real2recip(drhor + is * this->rhopw->nrxx, drhog.data() + is * this->rhopw->npw); - } - /// implement Kerker for density and magnetization separately - double fac = 0.0; - double gg0 = 0.0; - double amin = 0.0; - - for (int is = 0; is < nspin / resize_tmp; is++) - { - - if (is >= 1) - { - if (this->mixing_gg0_mag <= 0.0001 || this->mixing_beta_mag <= 0.1) - { -#ifdef __DEBUG - assert(is == 1); /// make sure break works -#endif - double is_mag = nspin - 1; - if (nspin == 4 && this->mixing_angle > 0) { is_mag = 1; -} - for (int ig = 0; ig < this->rhopw->npw * is_mag; ig++) - { - drhog[is * this->rhopw->npw + ig] = 0; - } - break; - } - fac = this->mixing_gg0_mag; - amin = this->mixing_beta_mag; - } - else - { - fac = this->mixing_gg0; - amin = this->mixing_beta; - } - - gg0 = std::pow(fac * ModuleBase::BOHR_TO_A / *this->tpiba, 2); - - const int is_idx = is * this->rhopw->npw; - const double gg0_amin = this->mixing_gg0_min / amin; -#ifdef _OPENMP -#pragma omp parallel for schedule(static, 512) -#endif - for (int ig = 0; ig < this->rhopw->npw; ig++) - { - double gg = this->rhopw->gg[ig]; - // I have not decided how to handle gg=0 part, will be changed in future - //if (gg == 0) - //{ - // drhog[is_idx + ig] *= 0; - // continue; - //} - double filter_g = std::max(gg / (gg + gg0), gg0_amin); - drhog[is_idx + ig] *= (1 - filter_g); - } - } - /// inverse FT - for (int is = 0; is < nspin / resize_tmp; ++is) - { - this->rhopw->recip2real(drhog.data() + is * this->rhopw->npw, drhor_filter.data() + is * this->rhopw->nrxx); - } - -#ifdef _OPENMP -#pragma omp parallel for schedule(static, 512) -#endif - for (int ir = 0; ir < this->rhopw->nrxx * nspin / resize_tmp; ir++) - { - drhor[ir] -= drhor_filter[ir]; - } - - ModuleBase::timer::end("Charge_Mixing", "Kerker_screen_real"); - return; -} diff --git a/source/source_estate/module_charge/mix_resid.cpp b/source/source_estate/module_charge/mix_resid.cpp deleted file mode 100644 index c59c543712a..00000000000 --- a/source/source_estate/module_charge/mix_resid.cpp +++ /dev/null @@ -1,495 +0,0 @@ -#include "charge_mixing.h" -#include "source_io/module_parameter/parameter.h" -#include "source_base/timer.h" -#include "source_base/parallel_reduce.h" -#include "source_hamilt/module_xc/xc_functional.h" - -double Charge_Mixing::get_drho(Charge* chr, const double nelec) -{ - ModuleBase::TITLE("Charge_Mixing", "get_drho"); - ModuleBase::timer::start("Charge_Mixing", "get_drho"); - const int nspin = PARAM.inp.nspin; - assert(nspin==1 || nspin==2 || nspin==4); - double drho = 0.0; - - if (PARAM.inp.scf_thr_type == 1) - { - for (int is = 0; is < nspin; ++is) - { - ModuleBase::GlobalFunc::NOTE("Perform FFT on rho(r) to obtain rho(G)."); - chr->rhopw->real2recip(chr->rho[is], chr->rhog[is]); - - ModuleBase::GlobalFunc::NOTE("Perform FFT on rho_save(r) to obtain rho_save(G)."); - chr->rhopw->real2recip(chr->rho_save[is], chr->rhog_save[is]); - } - - ModuleBase::GlobalFunc::NOTE("Calculate the charge difference between rho(G) and rho_save(G)"); - std::vector> drhog(nspin * this->rhopw->npw); -#ifdef _OPENMP -#pragma omp parallel for collapse(2) schedule(static, 512) -#endif - for (int is = 0; is < nspin; ++is) - { - for (int ig = 0; ig < this->rhopw->npw; ig++) - { - drhog[is * this->rhopw->npw + ig] = chr->rhog[is][ig] - chr->rhog_save[is][ig]; - } - } - - ModuleBase::GlobalFunc::NOTE("Calculate the norm of the Residual std::vector: < R[rho] | R[rho_save] >"); - drho = this->inner_product_recip_rho(drhog.data(), drhog.data()); - } - else - { - // Note: Maybe it is wrong. - // The inner_product_real function (L1-norm) is different from that (L2-norm) in mixing. - for (int is = 0; is < nspin; is++) - { - if (is != 0 && is != 3 && PARAM.globalv.domag_z) - { - continue; - } -#ifdef _OPENMP -#pragma omp parallel for reduction(+ : drho) -#endif - for (int ir = 0; ir < this->rhopw->nrxx; ir++) - { - drho += std::abs(chr->rho[is][ir] - chr->rho_save[is][ir]); - } - } -#ifdef __MPI - Parallel_Reduce::reduce_pool(drho); -#endif - assert(nelec != 0); - assert(*this->omega > 0); - assert(this->rhopw->nxyz > 0); - drho *= *this->omega / static_cast(this->rhopw->nxyz); - drho /= nelec; - } - - ModuleBase::timer::end("Charge_Mixing", "get_drho"); - return drho; -} - -double Charge_Mixing::get_dkin(Charge* chr, const double nelec) -{ - if (!(XC_Functional::get_ked_flag())) - { - return 0.0; - }; - ModuleBase::TITLE("Charge_Mixing", "get_dkin"); - ModuleBase::timer::start("Charge_Mixing", "get_dkin"); - double dkin = 0.0; - - // Get dkin from kin_r and kin_r_save for PW and LCAO both, which is different from drho. - for (int is = 0; is < PARAM.inp.nspin; is++) - { - if (is != 0 && is != 3 && PARAM.globalv.domag_z) - { - continue; - } -#ifdef _OPENMP -#pragma omp parallel for reduction(+ : dkin) -#endif - for (int ir = 0; ir < this->rhopw->nrxx; ir++) - { - dkin += std::abs(chr->kin_r[is][ir] - chr->kin_r_save[is][ir]); - } - } -#ifdef __MPI - Parallel_Reduce::reduce_pool(dkin); -#endif - assert(nelec != 0); - assert(*this->omega > 0); - assert(this->rhopw->nxyz > 0); - dkin *= *this->omega / static_cast(this->rhopw->nxyz); - dkin /= nelec; - - ModuleBase::timer::end("Charge_Mixing", "get_dkin"); - return dkin; -} - -double Charge_Mixing::inner_product_recip_rho(std::complex* rho1, std::complex* rho2) -{ - ModuleBase::TITLE("Charge_Mixing", "recip_rho"); - ModuleBase::timer::start("Charge_Mixing", "recip_rho"); - - std::complex** rhog1 = new std::complex*[PARAM.inp.nspin]; - std::complex** rhog2 = new std::complex*[PARAM.inp.nspin]; - for (int is = 0; is < PARAM.inp.nspin; is++) - { - rhog1[is] = rho1 + is * this->rhopw->npw; - rhog2[is] = rho2 + is * this->rhopw->npw; - } - - static const double fac = ModuleBase::e2 * ModuleBase::FOUR_PI / ((*this->tpiba) * (*this->tpiba)); - static const double fac2 = ModuleBase::e2 * ModuleBase::FOUR_PI / (ModuleBase::TWO_PI * ModuleBase::TWO_PI); - - double sum = 0.0; - - auto part_of_noncolin = [&]() - { - double sum = 0.0; - const int ig0 = this->rhopw->ig_gge0; -#ifdef _OPENMP -#pragma omp parallel for reduction(+ : sum) -#endif - for (int ig = 0; ig < this->rhopw->npw; ++ig) - { - if (ig == ig0) {continue;} - sum += (conj(rhog1[0][ig]) * rhog2[0][ig]).real() / this->rhopw->gg[ig]; - } - sum *= fac; - return sum; - }; - - switch (PARAM.inp.nspin) - { - case 1: - sum += part_of_noncolin(); - break; - - case 2: { - // (1) First part of density error. - const int ig0 = this->rhopw->ig_gge0; -#ifdef _OPENMP -#pragma omp parallel for reduction(+ : sum) -#endif - for (int ig = 0; ig < this->rhopw->npw; ++ig) - { - if (ig == ig0) {continue;} - sum += (conj(rhog1[0][ig] + rhog1[1][ig]) * (rhog2[0][ig] + rhog2[1][ig])).real() / this->rhopw->gg[ig]; - } - sum *= fac; - - if (PARAM.globalv.gamma_only_pw) - { - sum *= 2.0; - } - - // (2) Second part of density error. - // including |G|=0 term. - double sum2 = 0.0; - - // The G=0 component is the ig_gge0-th element of the local G-list on the - // rank that owns it, not necessarily element 0: the local G-list is built - // by scanning (x,y) sticks in grid order, so element 0 is the first plane - // wave of the first owned stick. Using a hardcoded index 0 made the inner - // product partition-dependent for pools with more than one rank. - if (ig0 >= 0) - { - sum2 += fac2 * (conj(rhog1[0][ig0] - rhog1[1][ig0]) * (rhog2[0][ig0] - rhog2[1][ig0])).real(); - } - - double mag = 0.0; -#ifdef _OPENMP -#pragma omp parallel for reduction(+ : mag) -#endif - for (int ig = 0; ig < this->rhopw->npw; ig++) - { - if (ig == ig0) { continue; } - mag += (conj(rhog1[0][ig] - rhog1[1][ig]) * (rhog2[0][ig] - rhog2[1][ig])).real(); - } - mag *= fac2; - - // if(PARAM.globalv.gamma_only_pw); - if (PARAM.globalv.gamma_only_pw) // Peize Lin delete ; 2020.01.31 - { - mag *= 2.0; - } - - // std::cout << " sum=" << sum << " mag=" << mag << std::endl; - sum2 += mag; - sum += sum2; - break; - } - case 4: - // non-collinear spin, added by zhengdy - if (!PARAM.globalv.domag && !PARAM.globalv.domag_z) { - sum += part_of_noncolin(); - } else - { - // another part with magnetization - const int ig0 = this->rhopw->ig_gge0; -#ifdef _OPENMP -#pragma omp parallel for reduction(+ : sum) -#endif - for (int ig = 0; ig < this->rhopw->npw; ig++) - { - if (ig == ig0) - { - continue; - } - sum += (conj(rhog1[0][ig]) * rhog2[0][ig]).real() / this->rhopw->gg[ig]; - } - sum *= fac; - if (ig0 > 0) - { - sum += fac2 - * ((conj(rhog1[1][ig0]) * rhog2[1][ig0]).real() + (conj(rhog1[2][ig0]) * rhog2[2][ig0]).real() - + (conj(rhog1[3][ig0]) * rhog2[3][ig0]).real()); - } - double fac3 = fac2; - if (PARAM.globalv.gamma_only_pw) - { - fac3 *= 2.0; - } -#ifdef _OPENMP -#pragma omp parallel for reduction(+ : sum) -#endif - for (int ig = 0; ig < this->rhopw->npw; ig++) - { - if (ig == ig0) { - continue; -} - sum += fac3 - * ((conj(rhog1[1][ig]) * rhog2[1][ig]).real() + (conj(rhog1[2][ig]) * rhog2[2][ig]).real() - + (conj(rhog1[3][ig]) * rhog2[3][ig]).real()); - } - } - break; - } -#ifdef __MPI - Parallel_Reduce::reduce_pool(sum); -#endif - - sum *= *this->omega * 0.5; - - delete[] rhog1; - delete[] rhog2; - - ModuleBase::timer::end("Charge_Mixing", "recip_rho"); - return sum; -} - -// a simple inner product, now is not used anywhere. For test only. -double Charge_Mixing::inner_product_recip_simple(std::complex* rho1, std::complex* rho2) -{ - ModuleBase::TITLE("Charge_Mixing", "recip_simple"); - ModuleBase::timer::start("Charge_Mixing", "recip_simple"); - - double rnorm = 0.0; - // consider a resize for mixing_angle - int resize_tmp = 1; - if (PARAM.inp.nspin == 4 && this->mixing_angle > 0) { resize_tmp = 2; -} -#ifdef _OPENMP -#pragma omp parallel for reduction(+ : rnorm) -#endif - for (int ig = 0; ig < this->rhopw->npw * PARAM.inp.nspin / resize_tmp; ++ig) - { - rnorm += (conj(rho1[ig]) * rho2[ig]).real(); - } -#ifdef __MPI - Parallel_Reduce::reduce_pool(rnorm); -#endif - - ModuleBase::timer::end("Charge_Mixing", "recip_simple"); - - return rnorm; -} - -// a Hartree-like inner product -double Charge_Mixing::inner_product_recip_hartree(std::complex* rhog1, std::complex* rhog2) -{ - ModuleBase::TITLE("Charge_Mixing", "recip_hartree"); - ModuleBase::timer::start("Charge_Mixing", "recip_hartree"); - - static const double fac = ModuleBase::e2 * ModuleBase::FOUR_PI / ((*this->tpiba) * (*this->tpiba)); - static const double fac2 = ModuleBase::e2 * ModuleBase::FOUR_PI / (ModuleBase::TWO_PI * ModuleBase::TWO_PI); - - double sum = 0.0; - const int npw = this->rhopw->npw; - - // a lambda function for summing the charge density - auto part_of_rho = [&]() - { - double sum = 0.0; - const int ig0 = this->rhopw->ig_gge0; -#ifdef _OPENMP -#pragma omp parallel for reduction(+ : sum) -#endif - for (int ig = 0; ig < this->rhopw->npw; ++ig) - { - if (ig == ig0) - { - continue; - } - sum += (conj(rhog1[ig]) * rhog2[ig]).real() / this->rhopw->gg[ig]; - } - sum *= fac; - return sum; - }; - - if (PARAM.inp.nspin==1) - { - sum += part_of_rho(); - } - else if (PARAM.inp.nspin==2) - { - // charge density part - const int ig0 = this->rhopw->ig_gge0; -#ifdef _OPENMP -#pragma omp parallel for reduction(+ : sum) -#endif - for (int ig = 0; ig < this->rhopw->npw; ++ig) - { - if (ig == ig0) - { - continue; - } - sum += (conj(rhog1[ig]) * (rhog2[ig])).real() / this->rhopw->gg[ig]; - } - sum *= fac; - - if (PARAM.globalv.gamma_only_pw) - { - sum *= 2.0; - } - - // (2) Second part of density error. - // including |G|=0 term. - double sum2 = 0.0; - - // Same G=0 indexing remark as in inner_product_recip_rho: use ig_gge0 - // instead of a hardcoded index 0, otherwise the inner product (and hence - // the DIIS mixing coefficients) depends on how the pool is divided. - if (ig0 >= 0) - { - sum2 += fac2 * (conj(rhog1[ig0 + this->rhopw->npw]) * rhog2[ig0 + this->rhopw->npw]).real(); - } - - double mag = 0.0; -#ifdef _OPENMP -#pragma omp parallel for reduction(+ : mag) -#endif - for (int ig = 0; ig < this->rhopw->npw; ig++) - { - if (ig == ig0) { continue; } - mag += (conj(rhog1[ig + this->rhopw->npw]) * rhog2[ig + this->rhopw->npw]).real(); - } - mag *= fac2; - - if (PARAM.globalv.gamma_only_pw) - { - mag *= 2.0; - } - - sum2 += mag; - sum += sum2; - } - else if (PARAM.inp.nspin==4) - { - if (!PARAM.globalv.domag && !PARAM.globalv.domag_z) - { - sum += part_of_rho(); - } - else if (this->mixing_angle <= 0) - { - // sum for tradtional mixing - const int ig0 = this->rhopw->ig_gge0; -#ifdef _OPENMP -#pragma omp parallel for reduction(+ : sum) -#endif - for (int ig = 0; ig < this->rhopw->npw; ig++) - { - if (ig == ig0) {continue;} - sum += (conj(rhog1[ig]) * rhog2[ig]).real() / this->rhopw->gg[ig]; - } - sum *= fac; - if (ig0 > 0) - { - sum += fac2 - * ((conj(rhog1[ig0 + npw]) * rhog2[ig0 + npw]).real() + (conj(rhog1[ig0 + 2*npw]) * rhog2[ig0 + 2*npw]).real() - + (conj(rhog1[ig0 + 3*npw]) * rhog2[ig0 + 3*npw]).real()); - } - double fac3 = fac2; - if (PARAM.globalv.gamma_only_pw) - { - fac3 *= 2.0; - } -#ifdef _OPENMP -#pragma omp parallel for reduction(+ : sum) -#endif - for (int ig = 0; ig < this->rhopw->npw; ig++) - { - if (ig == ig0) { - continue; -} - sum += fac3 - * ((conj(rhog1[ig + npw]) * rhog2[ig + npw]).real() + (conj(rhog1[ig + 2*npw]) * rhog2[ig + 2*npw]).real() - + (conj(rhog1[ig + 3*npw]) * rhog2[ig + 3*npw]).real()); - } - } - else if (this->mixing_angle > 0) - { - // sum for angle mixing - const int ig0 = this->rhopw->ig_gge0; -#ifdef _OPENMP -#pragma omp parallel for reduction(+ : sum) -#endif - for (int ig = 0; ig < this->rhopw->npw; ig++) - { - if (ig == ig0) - { - continue; - } - sum += (conj(rhog1[ig]) * rhog2[ig]).real() / this->rhopw->gg[ig]; - } - sum *= fac; - if (ig0 > 0) - { - sum += fac2 - * ((conj(rhog1[ig0 + this->rhopw->npw]) * rhog2[ig0 + this->rhopw->npw]).real()); - } - double fac3 = fac2; - if (PARAM.globalv.gamma_only_pw) - { - fac3 *= 2.0; - } -#ifdef _OPENMP -#pragma omp parallel for reduction(+ : sum) -#endif - for (int ig = 0; ig < this->rhopw->npw; ig++) - { - if (ig == ig0) { - continue; -} - sum += fac3 - * ((conj(rhog1[ig + this->rhopw->npw]) * rhog2[ig + this->rhopw->npw]).real()); - } - } - } -#ifdef __MPI - Parallel_Reduce::reduce_pool(sum); -#endif - - sum *= *this->omega * 0.5; - - ModuleBase::timer::end("Charge_Mixing", "recip_hartree"); - - return sum; -} - -double Charge_Mixing::inner_product_real(double* rho1, double* rho2) -{ - double rnorm = 0.0; - // consider a resize for mixing_angle - int resize_tmp = 1; - if (PARAM.inp.nspin == 4 && this->mixing_angle > 0) - { - resize_tmp = 2; - } - -#ifdef _OPENMP -#pragma omp parallel for reduction(+ : rnorm) -#endif - for (int ir = 0; ir < this->rhopw->nrxx * PARAM.inp.nspin / resize_tmp; ++ir) - { - rnorm += rho1[ir] * rho2[ir]; - } -#ifdef __MPI - Parallel_Reduce::reduce_pool(rnorm); -#endif - return rnorm; -} diff --git a/source/source_estate/module_charge/symm_rho.cpp b/source/source_estate/module_charge/symm_rho.cpp deleted file mode 100644 index a3417e96d9b..00000000000 --- a/source/source_estate/module_charge/symm_rho.cpp +++ /dev/null @@ -1,211 +0,0 @@ -#include "symm_rho.h" - -#include "source_estate/module_charge/charge.h" -#include "source_hamilt/module_xc/xc_functional.h" -#include "source_io/module_parameter/parameter.h" - -Symmetry_rho::Symmetry_rho() -{ -} - -Symmetry_rho::~Symmetry_rho() -{ -} - -void Symmetry_rho::symmetrize_rho(const int nspin, - const Charge& chr, - const ModulePW::PW_Basis* pw, - ModuleSymmetry::Symmetry& symm) -{ - Symmetry_rho srho; - if (nspin == 4) - { - // nspin=4 (non-collinear/SOC): rho[0] is the charge density rho^0 (scalar, symmetrized - // spatially like nspin=1); rho[1,2,3] are the spin density (rho^x, rho^y, rho^z) which - // must be symmetrized TOGETHER with the per-operation spin rotation W(g). - srho.begin(0, chr, pw, symm); - srho.begin_soc(chr, pw, symm); - return; - } - for (int is = 0; is < nspin; is++) - { - srho.begin(is, chr, pw, symm); - } -} - -void Symmetry_rho::begin(const int& spin_now, - const Charge& chr, - const ModulePW::PW_Basis* rho_basis, - ModuleSymmetry::Symmetry& symm) const -{ - assert(spin_now < 4); // added by zhengdy-soc - - if (ModuleSymmetry::Symmetry::symm_flag != 1) - { - return; - } - - ModuleBase::TITLE("Symmetry_rho", "begin"); - ModuleBase::timer::start("Symmetry_rho","begin"); - -// both parallel and serial -// if(symm.nrot==symm.nrotk) //pure point-group, do rho_symm in real space -// { -// psymm(chr.rho[spin_now], rho_basis, Pgrid, symm); -// if(XC_Functional::get_ked_flag()) psymm(chr.kin_r[spin_now], -// rho_basis,Pgrid,symm); -// } -// else //space group, do rho_symm in reciprocal space - - rho_basis->real2recip(chr.rho[spin_now], chr.rhog[spin_now]); - - psymmg(chr.rhog[spin_now], rho_basis, symm); // need to modify - - rho_basis->recip2real(chr.rhog[spin_now], chr.rho[spin_now]); - - if (XC_Functional::get_ked_flag() || chr.cal_elf) - { - // Use std::vector to manage kin_g instead of raw pointer - std::vector> kin_g(chr.ngmc); - rho_basis->real2recip(chr.kin_r[spin_now], kin_g.data()); - psymmg(kin_g.data(), rho_basis, symm); - rho_basis->recip2real(kin_g.data(), chr.kin_r[spin_now]); - } - - ModuleBase::timer::end("Symmetry_rho","begin"); - return; -} - -void Symmetry_rho::begin(const int& spin_now, - double** rho, - std::complex** rhog, - int ngmc, - double** kin_r, - const ModulePW::PW_Basis* rho_basis, - ModuleSymmetry::Symmetry& symm) const -{ - assert(spin_now < 4); // added by zhengdy-soc - - if (ModuleSymmetry::Symmetry::symm_flag != 1) - { - return; - } - - ModuleBase::TITLE("Symmetry_rho", "begin"); - ModuleBase::timer::start("Symmetry_rho","begin"); - - // both parallel and serial - // if(symm.nrot==symm.nrotk) //pure point-group, do rho_symm in real space - // { - // psymm(CHR.rho[spin_now], rho_basis, Pgrid, symm); - // if(XC_Functional::get_ked_flag()) psymm(CHR.kin_r[spin_now], - // rho_basis,Pgrid,symm); - // } - // else //space group, do rho_symm in reciprocal space - { - rho_basis->real2recip(rho[spin_now], rhog[spin_now]); - psymmg(rhog[spin_now], rho_basis, symm); - rho_basis->recip2real(rhog[spin_now], rho[spin_now]); - - if (XC_Functional::get_ked_flag() && kin_r != nullptr) - { - // Use std::vector to manage kin_g instead of raw pointer - std::vector> kin_g(ngmc); - rho_basis->real2recip(kin_r[spin_now], kin_g.data()); - psymmg(kin_g.data(), rho_basis, symm); - rho_basis->recip2real(kin_g.data(), kin_r[spin_now]); - } - } - - ModuleBase::timer::end("Symmetry_rho","begin"); - return; -} - -void Symmetry_rho::begin_soc(const Charge& chr, - const ModulePW::PW_Basis* rho_basis, - ModuleSymmetry::Symmetry& symm) const -{ - this->begin_soc(chr.rho, chr.rhog, rho_basis, symm); -} - -void Symmetry_rho::begin_soc(double** rho, - std::complex** rhog, - const ModulePW::PW_Basis* rho_basis, - ModuleSymmetry::Symmetry& symm) const -{ - if (ModuleSymmetry::Symmetry::symm_flag != 1) - { - return; - } - - ModuleBase::TITLE("Symmetry_rho", "begin_soc"); - ModuleBase::timer::start("Symmetry_rho", "begin_soc"); - - // the three spin components are coupled by the spin rotation, so they are transformed to - // reciprocal space and symmetrized together (rho[1]=rho^x, rho[2]=rho^y, rho[3]=rho^z). - for (int is = 1; is < 4; ++is) - { - rho_basis->real2recip(rho[is], rhog[is]); - } - - psymmg_soc(rhog[1], rhog[2], rhog[3], rho_basis, symm); - - for (int is = 1; is < 4; ++is) - { - rho_basis->recip2real(rhog[is], rho[is]); - } - - ModuleBase::timer::end("Symmetry_rho", "begin_soc"); - return; -} - -void Symmetry_rho::psymm(double* rho_part, - const ModulePW::PW_Basis* rho_basis, - Parallel_Grid& Pgrid, - ModuleSymmetry::Symmetry& symm) const -{ - ModuleBase::TITLE("Symmetry_rho", "psymm"); - ModuleBase::timer::start("Symmetry_rho","psymm"); - -#ifdef __MPI - // reduce all rho from the first pool. - std::vector rhotot; - if (GlobalV::MY_RANK == 0) - { - rhotot.resize(rho_basis->nxyz); - ModuleBase::GlobalFunc::ZEROS(rhotot.data(), rho_basis->nxyz); - } - Pgrid.reduce(rhotot.data(), rho_part, false); - - if (GlobalV::MY_RANK == 0) - { - symm.rho_symmetry(rhotot.data(), rho_basis->nx, rho_basis->ny, rho_basis->nz); -#else - symm.rho_symmetry(rho_part, rho_basis->nx, rho_basis->ny, rho_basis->nz); -#endif - /* - int count = 0; - GlobalV::ofs_running << scientific; - for(int iz=0; iznz; iz++) - { - GlobalV::ofs_running << "\n iz=" << iz; - for(int iy=0; iyny; iy++) - { - for(int ix=0; ixnx; ix++) - { - if(count%5==0) GlobalV::ofs_running << "\n"; - ++count; - GlobalV::ofs_running << " " << rhotot[ix*rho_basis->ny*rho_basis->nz+iy*rho_basis->nz+iz]; - } - } - } - */ -#ifdef __MPI - } - - Pgrid.bcast(rhotot.data(), rho_part, GlobalV::MY_RANK, PARAM.inp.esolver_type == "sdft"); -#endif - - ModuleBase::timer::end("Symmetry_rho","psymm"); - return; -} diff --git a/source/source_estate/module_charge/symm_rho.h b/source/source_estate/module_charge/symm_rho.h deleted file mode 100644 index f9d1cb48a89..00000000000 --- a/source/source_estate/module_charge/symm_rho.h +++ /dev/null @@ -1,97 +0,0 @@ -#ifndef SYMM_RHO_H -#define SYMM_RHO_H -#include "source_basis/module_pw/pw_basis.h" -#include "source_cell/module_symmetry/symmetry.h" -#include "source_base/parallel_grid.h" - -class Charge; - -class Symmetry_rho -{ - public: - Symmetry_rho(); - ~Symmetry_rho(); - - /** - * @brief Symmetrize charge density for all spin channels - * - * This is a static helper function that symmetrizes the charge density - * for all spin channels by calling begin() for each spin. - * - * @param nspin Number of spin channels - * @param chr Charge object containing the density - * @param pw Plane wave basis - * @param symm Symmetry object - */ - static void symmetrize_rho(const int nspin, - const Charge& chr, - const ModulePW::PW_Basis* pw, - ModuleSymmetry::Symmetry& symm); - - void begin(const int& spin_now, - const Charge& CHR, - const ModulePW::PW_Basis* pw, - ModuleSymmetry::Symmetry& symm) const; - - void begin(const int& spin_now, - double** rho, - std::complex** rhog, - int ngmc, - double** kin_r, - const ModulePW::PW_Basis* pw, - ModuleSymmetry::Symmetry& symm) const; - - /// @brief Symmetrize the nspin=4 spin density (rho^x, rho^y, rho^z = rho[1,2,3]) with the - /// coupled spin rotation. The charge component rho^0 = rho[0] is handled separately - /// by the ordinary scalar begin(). - void begin_soc(const Charge& CHR, - const ModulePW::PW_Basis* pw, - ModuleSymmetry::Symmetry& symm) const; - - /** - * @brief Symmetrize raw nspin=4 spin-density arrays with coupled spin rotations. - * - * @param rho Real-space density components ordered as rho0, mx, my, mz. - * @param rhog Reciprocal-space work arrays with the same component ordering. - * @param pw Plane-wave basis used for the Fourier transforms. - * @param symm Symmetry operations and spin rotations. - */ - void begin_soc(double** rho, - std::complex** rhog, - const ModulePW::PW_Basis* pw, - ModuleSymmetry::Symmetry& symm) const; - - private: - // in real space: - void psymm(double* rho_part, - const ModulePW::PW_Basis* pw, - Parallel_Grid& Pgrid, - ModuleSymmetry::Symmetry& symm) const; - // in reciprocal space: - void psymmg(std::complex* rhog_part, - const ModulePW::PW_Basis* rho_basis, - ModuleSymmetry::Symmetry& symm) const; - // in reciprocal space, the three coupled spin components (rho^x, rho^y, rho^z) for nspin=4: - void psymmg_soc(std::complex* rhog_x, - std::complex* rhog_y, - std::complex* rhog_z, - const ModulePW::PW_Basis* rho_basis, - ModuleSymmetry::Symmetry& symm) const; -#ifdef __MPI - void reduce_to_fullrhog(const ModulePW::PW_Basis* rho_basis, - std::complex* rhogtot, - std::complex* rhogin, - int* ig2isztot, - const int* ig2iszin, - int max_npw) const; - void rhog_piece_to_all(const ModulePW::PW_Basis* rho_basis, - std::complex* rhogtot, - std::complex* rhog_part) const; -#endif - void get_ixyz2ipw(const ModulePW::PW_Basis* rho_basis, - const int* ig2isztot, - const int* fftixy2is, - int* ixyz2ipw) const; //(ix, iy, iz) -> (ip, ig) -}; - -#endif diff --git a/source/source_estate/module_charge/symm_rhog.cpp b/source/source_estate/module_charge/symm_rhog.cpp deleted file mode 100644 index 4327ee68603..00000000000 --- a/source/source_estate/module_charge/symm_rhog.cpp +++ /dev/null @@ -1,356 +0,0 @@ -#include "symm_rho.h" -#include "source_base/parallel_reduce.h" -#include "source_base/parallel_global.h" -#include "source_cell/module_symmetry/symm_rot_spin.h" -#include "source_hamilt/module_xc/xc_functional.h" - - -void Symmetry_rho::psymmg(std::complex* rhog_part, const ModulePW::PW_Basis *rho_basis, ModuleSymmetry::Symmetry &symm) const -{ - //(1) get fftixy2is and do Allreduce - int * fftixy2is = new int [rho_basis->fftnxy]; - rho_basis->getfftixy2is(fftixy2is); //current proc -#ifdef __MPI - Parallel_Reduce::reduce_pool(fftixy2is, rho_basis->fftnxy); - if(rho_basis->poolnproc>1) - for (int i=0;ifftnxy;++i) - fftixy2is[i]+=rho_basis->poolnproc-1; - - // (2) reduce all rho from the first pool. - std::complex* rhogtot; - int* ig2isztot = nullptr; - if(GlobalV::RANK_IN_POOL == 0) - { - rhogtot = new std::complex[rho_basis->npwtot]; - ModuleBase::GlobalFunc::ZEROS(rhogtot, rho_basis->npwtot); - ig2isztot = new int[rho_basis->npwtot]; - ModuleBase::GlobalFunc::ZEROS(rhogtot, rho_basis->npwtot); - } - // find max_npw - int max_npw=0; - for (int proc = 0; proc < rho_basis->poolnproc; ++proc) - { - if(rho_basis->npw_per[proc] > max_npw) - { - max_npw=rho_basis->npw_per[proc]; - } - } - this->reduce_to_fullrhog(rho_basis, rhogtot, rhog_part, ig2isztot, rho_basis->ig2isz, max_npw); - - // (3) get ixy2ipw and do rhog_symmetry on proc 0 of each pool - if(GlobalV::RANK_IN_POOL==0) - { -#endif - //init ixyz2ipw - int* ixyz2ipw = new int[rho_basis->fftnxyz]; - for(int i=0;ifftnxyz;++i) ixyz2ipw[i]=-1; - // The density must be symmetrized with the same group used to fold the k-points. For - // nspin=4 magnetic that is the Shubnikov group; Theta leaves the charge invariant, so the - // antiunitary elements act on rho exactly like unitary ones (their trs_inv is not used here). - std::vector kgmat; - std::vector> gtr; - std::vector trs_inv; - const int nop = symm.density_sym_ops(kgmat, gtr, trs_inv); -#ifdef __MPI - this->get_ixyz2ipw(rho_basis, ig2isztot, fftixy2is, ixyz2ipw); - symm.rhog_symmetry(rhogtot, ixyz2ipw, rho_basis->nx, rho_basis->ny, rho_basis->nz, - rho_basis->fftnx, rho_basis->fftny, rho_basis->fftnz, - rho_basis->gamma_only, kgmat.data(), gtr.data(), nop); -#else - this->get_ixyz2ipw(rho_basis, rho_basis->ig2isz, fftixy2is, ixyz2ipw); - symm.rhog_symmetry(rhog_part, ixyz2ipw, rho_basis->nx, rho_basis->ny, rho_basis->nz, - rho_basis->fftnx, rho_basis->fftny, rho_basis->fftnz, - rho_basis->gamma_only, kgmat.data(), gtr.data(), nop); -#endif - delete[] ixyz2ipw; -#ifdef __MPI - } - - // (4) send the result to other procs in the same pool - this->rhog_piece_to_all(rho_basis, rhogtot, rhog_part); - - if(GlobalV::RANK_IN_POOL==0) - { - delete[] rhogtot; - delete[] ig2isztot; - } -#endif - delete[] fftixy2is; - return; -} - -void Symmetry_rho::psymmg_soc(std::complex* rhog_x, std::complex* rhog_y, - std::complex* rhog_z, const ModulePW::PW_Basis* rho_basis, ModuleSymmetry::Symmetry& symm) const -{ - // build the per-operation spin-rotation matrices W(g) from the cartesian rotation - // gmatc(g) = direct_to_cartesian(gmatrix(g)) = latvec^-1 * gmatrix(g) * latvec. - auto build_wspin = [&rho_basis, &symm]() { - const ModuleBase::Matrix3 latvec = rho_basis->latvec; - const ModuleBase::Matrix3 ilatvec = latvec.Inverse(); - // index [0,nrotk) unitary, [nrotk, nrotk+nrotk_anti) the spatial parts of the - // antiunitary elements Theta*g -- same layout as density_sym_ops(). - const int na = symm.magnetic_nspin4 ? symm.nrotk_anti : 0; - std::vector wspin(symm.nrotk + na); - for (int i = 0; i < symm.nrotk; ++i) - { - const ModuleBase::Matrix3 gmatc = ilatvec * symm.gmatrix[i] * latvec; - wspin[i] = ModuleSymmetry::SpinRotation::spin_so3(gmatc); - } - for (int j = 0; j < na; ++j) - { - const ModuleBase::Matrix3 gmatc = ilatvec * symm.gmatrix_anti[j] * latvec; - wspin[symm.nrotk + j] = ModuleSymmetry::SpinRotation::spin_so3(gmatc); - } - return wspin; - }; - - //(1) get fftixy2is and do Allreduce - int * fftixy2is = new int [rho_basis->fftnxy]; - rho_basis->getfftixy2is(fftixy2is); //current proc -#ifdef __MPI - Parallel_Reduce::reduce_pool(fftixy2is, rho_basis->fftnxy); - if(rho_basis->poolnproc>1) - for (int i=0;ifftnxy;++i) - fftixy2is[i]+=rho_basis->poolnproc-1; - - // (2) reduce all three spin components from the first pool. - std::complex* rhogtot_x = nullptr; - std::complex* rhogtot_y = nullptr; - std::complex* rhogtot_z = nullptr; - int* ig2isztot = nullptr; - if(GlobalV::RANK_IN_POOL == 0) - { - rhogtot_x = new std::complex[rho_basis->npwtot]; - rhogtot_y = new std::complex[rho_basis->npwtot]; - rhogtot_z = new std::complex[rho_basis->npwtot]; - ModuleBase::GlobalFunc::ZEROS(rhogtot_x, rho_basis->npwtot); - ModuleBase::GlobalFunc::ZEROS(rhogtot_y, rho_basis->npwtot); - ModuleBase::GlobalFunc::ZEROS(rhogtot_z, rho_basis->npwtot); - ig2isztot = new int[rho_basis->npwtot]; - ModuleBase::GlobalFunc::ZEROS(ig2isztot, rho_basis->npwtot); - } - // find max_npw - int max_npw=0; - for (int proc = 0; proc < rho_basis->poolnproc; ++proc) - { - if(rho_basis->npw_per[proc] > max_npw) - { - max_npw=rho_basis->npw_per[proc]; - } - } - this->reduce_to_fullrhog(rho_basis, rhogtot_x, rhog_x, ig2isztot, rho_basis->ig2isz, max_npw); - this->reduce_to_fullrhog(rho_basis, rhogtot_y, rhog_y, ig2isztot, rho_basis->ig2isz, max_npw); - this->reduce_to_fullrhog(rho_basis, rhogtot_z, rhog_z, ig2isztot, rho_basis->ig2isz, max_npw); - - // (3) get ixy2ipw and do rhog_symmetry_nspin4 on proc 0 of each pool - if(GlobalV::RANK_IN_POOL==0) - { -#endif - //init ixyz2ipw - int* ixyz2ipw = new int[rho_basis->fftnxyz]; - for(int i=0;ifftnxyz;++i) ixyz2ipw[i]=-1; - std::vector wspin = build_wspin(); - std::vector kgmat; - std::vector> gtr; - std::vector trs_inv; - const int nop = symm.density_sym_ops(kgmat, gtr, trs_inv); -#ifdef __MPI - this->get_ixyz2ipw(rho_basis, ig2isztot, fftixy2is, ixyz2ipw); - symm.rhog_symmetry_nspin4(rhogtot_x, rhogtot_y, rhogtot_z, wspin.data(), ixyz2ipw, - rho_basis->nx, rho_basis->ny, rho_basis->nz, - rho_basis->fftnx, rho_basis->fftny, rho_basis->fftnz, - trs_inv.data(), kgmat.data(), gtr.data(), nop); -#else - this->get_ixyz2ipw(rho_basis, rho_basis->ig2isz, fftixy2is, ixyz2ipw); - symm.rhog_symmetry_nspin4(rhog_x, rhog_y, rhog_z, wspin.data(), ixyz2ipw, - rho_basis->nx, rho_basis->ny, rho_basis->nz, - rho_basis->fftnx, rho_basis->fftny, rho_basis->fftnz, - trs_inv.data(), kgmat.data(), gtr.data(), nop); -#endif - delete[] ixyz2ipw; -#ifdef __MPI - } - - // (4) send the result to other procs in the same pool - this->rhog_piece_to_all(rho_basis, rhogtot_x, rhog_x); - this->rhog_piece_to_all(rho_basis, rhogtot_y, rhog_y); - this->rhog_piece_to_all(rho_basis, rhogtot_z, rhog_z); - - if(GlobalV::RANK_IN_POOL==0) - { - delete[] rhogtot_x; - delete[] rhogtot_y; - delete[] rhogtot_z; - delete[] ig2isztot; - } -#endif - delete[] fftixy2is; - return; -} - -#ifdef __MPI - -void Symmetry_rho::reduce_to_fullrhog(const ModulePW::PW_Basis *rho_basis, - std::complex* rhogtot, std::complex* rhogin, - int* ig2isztot, const int* ig2iszin, int max_npw) const -{ - ModuleBase::TITLE("Symmetry_rho","reduce_to_fullrhog"); - - std::complex* rhog_piece = new std::complex[max_npw]; - int* ig2isz_piece = new int[max_npw]; - - int npw_start=0; - for(int proc=0; procpoolnproc; ++proc) - { - ModuleBase::GlobalFunc::ZEROS(rhog_piece, max_npw); - ModuleBase::GlobalFunc::ZEROS(ig2isz_piece, max_npw); - - MPI_Status ierror; - - // case 1: the first part of rho in processor 0 in each pool. - if(proc == 0 && GlobalV::RANK_IN_POOL ==0) - { - for(int ig=0; ignpw; ++ig) - { - rhog_piece[ig] = rhogin[ig]; - ig2isz_piece[ig]=ig2iszin[ig]; - } - } - - // case 2: > first part rho: send the rho to - // processor 0 in each pool - else if(proc == GlobalV::RANK_IN_POOL ) - { - for(int ig=0; ignpw; ++ig) - { - rhog_piece[ig] = rhogin[ig]; - ig2isz_piece[ig]=ig2iszin[ig]; - } - MPI_Send(rhog_piece,rho_basis->npw, MPI_DOUBLE_COMPLEX, 0, proc, POOL_WORLD); - MPI_Send(ig2isz_piece, rho_basis->npw, MPI_INT, 0, proc+rho_basis->poolnproc, POOL_WORLD); - } - - // case 2: > first part rho: processor 0 receive the rho - // from other processors - else if(GlobalV::RANK_IN_POOL==0) - { - MPI_Recv(rhog_piece, rho_basis->npw_per[proc], MPI_DOUBLE_COMPLEX, proc, proc, POOL_WORLD, &ierror); - MPI_Recv(ig2isz_piece, rho_basis->npw_per[proc], MPI_INT, proc, proc+rho_basis->poolnproc, POOL_WORLD, &ierror); - } - - if(GlobalV::RANK_IN_POOL==0) - { - for(int ig=0; ignpw_per[proc]; ++ig) - { - rhogtot[npw_start+ig] = rhog_piece[ig]; - ig2isztot[npw_start+ig] = ig2isz_piece[ig]; - } - npw_start+=rho_basis->npw_per[proc]; - } - } - if(GlobalV::RANK_IN_POOL==0) assert(npw_start==rho_basis->npwtot); - delete[] rhog_piece; - delete[] ig2isz_piece; - - MPI_Barrier(MPI_COMM_WORLD); - - return; -} - -void Symmetry_rho::rhog_piece_to_all(const ModulePW::PW_Basis *rho_basis, - std::complex* rhogtot, std::complex* rhog_part) const -{ - ModuleBase::TITLE(" Symmetry_rho","rhog_piece_to_all"); - - MPI_Status ierror; - - if(GlobalV::RANK_IN_POOL==0) - { - // proc 0: send to other proc in pool - // itself: directly copy - for(int ig=0;ignpw;++ig) - { - rhog_part[ig]=rhogtot[ig]; - } - int npw_start=rho_basis->npw; - for(int proc=1;procpoolnproc;++proc) - { - MPI_Send(&rhogtot[npw_start], rho_basis->npw_per[proc], MPI_DOUBLE_COMPLEX, proc, proc, POOL_WORLD); - npw_start+=rho_basis->npw_per[proc]; - } - assert(npw_start==rho_basis->npwtot); - }// GlobalV::RANK_IN_POOL == 0 - else - { - MPI_Recv(rhog_part, rho_basis->npw_per[GlobalV::RANK_IN_POOL], MPI_DOUBLE_COMPLEX, 0, GlobalV::RANK_IN_POOL, POOL_WORLD, &ierror); - } - return; -} - -#endif - -// only for MYRANK==0 -void Symmetry_rho::get_ixyz2ipw(const ModulePW::PW_Basis *rho_basis, - const int* ig2isztot, const int* fftixy2is, int* ixyz2ipw) const -{ - //step 1: get ipsz2ipw - - //get ipsz2ipw from ig2isztot - int* ipsz2ipw = new int [rho_basis->nstot*rho_basis->nz]; - for(int i=0;instot*rho_basis->nz;++i) ipsz2ipw[i]=-1; - - int npw_count=0; - int nstnz_count=0; - int ipsz=0; //global index of a z-grid on stick - int isz=0; //local index of a z-grid stick on ip core - int ipw=0; // global index of pw (in npwtot) - for (int ip=0;ippoolnproc;++ip) - { - for (int ig=0;ignpw_per[ip];++ig) - { - ipw=npw_count+ig; - isz=ig2isztot[ipw]; - ipsz=nstnz_count+isz; - ipsz2ipw[ipsz]=ipw; - } - npw_count+=rho_basis->npw_per[ip]; - nstnz_count+=rho_basis->nst_per[ip]*rho_basis->nz; - } - assert(npw_count==rho_basis->npwtot); - assert(nstnz_count==rho_basis->nstot*rho_basis->nz); - - //step2: ixyz to ipsz - - //save the start-index of (nst*nz) till each core - int* nstnz_start = new int[rho_basis->poolnproc]; - nstnz_start[0]=0; - for (int ip=1; ippoolnproc; ++ip) - nstnz_start[ip]=nstnz_start[ip-1]+rho_basis->nst_per[ip-1]*rho_basis->nz; - - //tmp variables - int ixy, ixyz, ip, is, ig=0; - - for (int ix=0;ixfftnx;++ix) - { - for (int iy=0;iyfftny;++iy) - { - for(int iz=0;izfftnz;++iz) - { - ixy = ix*rho_basis->fftny + iy; - ixyz = ixy*rho_basis->fftnz+iz; - ip = rho_basis->fftixy2ip[ixy]; - if (ip==-1) continue; //not in any core - is = fftixy2is[ixy]; //stick-index on ip=proc core - if (is==-1) continue; //not on any stick - ipsz = nstnz_start[ip]+is*rho_basis->nz+iz; - ipw = ipsz2ipw[ipsz]; - ixyz2ipw[ixyz] = ipw; - } - } - } - assert (ixyz==rho_basis->fftnxyz-1); - - delete[] nstnz_start; - delete[] ipsz2ipw; - return; -} diff --git a/source/source_estate/module_charge/unittests/CMakeLists.txt b/source/source_estate/module_charge/unittests/CMakeLists.txt new file mode 100644 index 00000000000..2068271b92b --- /dev/null +++ b/source/source_estate/module_charge/unittests/CMakeLists.txt @@ -0,0 +1,165 @@ +abacus_disable_feature_definitions(__MPI) +abacus_disable_feature_definitions(__EXX) +abacus_disable_feature_definitions(__CUDA) +abacus_disable_feature_definitions(__UT_USE_CUDA) +abacus_disable_feature_definitions(__UT_USE_ROCM) +abacus_disable_feature_definitions(__ROCM) +abacus_disable_feature_definitions(__MLALGO) +abacus_disable_feature_definitions(_OPENMP) + +if (ENABLE_MPI) + +AddTest( + TARGET MODULE_CHARGE_rhog_io + LIBS parameter base device planewave + SOURCES test_chg_rhog_io.cpp ../chg_rhog_io.cpp ../../../source_basis/module_pw/test/test_tool.cpp + # This test drives PW_Basis::initmpi and read/write_rhog's MPI collectives, + # so it must keep __MPI even though this directory disables it. Its main() + # calls MPI_Init via test_tool.cpp's setupmpi(). + KEEP_FEATURE_DEFINITIONS __MPI +) + +AddTest( + TARGET MODULE_CHARGE_charge + LIBS parameter planewave_serial base device cell_info + SOURCES test_charge.cpp ../charge.cpp ../chg_atomic.cpp ../chg_atomic_inner.cpp ../chg_tools.cpp +) + +AddTest( + TARGET MODULE_CHARGE_extra + LIBS parameter base device cell_info + SOURCES test_chg_extra.cpp ../chg_extra.cpp ../../../source_io/module_output/read_cube.cpp ../../../source_io/module_output/write_cube.cpp + ../../../source_base/module_fft/fft_bundle.cpp ../../../source_base/module_fft/fft_cpu.cpp +) + +AddTest( + TARGET MODULE_CHARGE_tools + LIBS parameter planewave_serial base device cell_info + SOURCES test_chg_tools.cpp ../chg_tools.cpp +) + +AddTest( + TARGET MODULE_CHARGE_uspp + LIBS base device + SOURCES test_chg_uspp.cpp ../chg_uspp.cpp +) + +AddTest( + TARGET MODULE_CHARGE_dmr + LIBS base device + SOURCES test_chg_dmr.cpp ../chg_dmr.cpp +) + +AddTest( + TARGET MODULE_CHARGE_precond + LIBS base device planewave_serial + SOURCES test_chg_precond.cpp ../chg_precond.cpp +) + +AddTest( + TARGET MODULE_CHARGE_drho + LIBS parameter planewave_serial base device cell_info + SOURCES test_chg_drho.cpp ../chg_drho.cpp ../chg_drho_inner.cpp ../charge.cpp ../chg_tools.cpp +) + +AddTest( + TARGET MODULE_CHARGE_drho_inner + LIBS base device planewave_serial + SOURCES test_chg_drho_inner.cpp ../chg_drho_inner.cpp +) + +AddTest( + TARGET MODULE_CHARGE_mix_rho + LIBS parameter base psi device planewave_serial cell_info + SOURCES test_chg_mix_rho.cpp + ../chg_mix.cpp ../chg_mix_rho.cpp + ../chg_dmr.cpp + ../chg_drho.cpp + ../chg_drho_inner.cpp + ../chg_precond.cpp ../chg_tau.cpp + ../chg_uspp.cpp + ../charge.cpp ../chg_tools.cpp +) + +AddTest( + TARGET MODULE_CHARGE_mix + LIBS parameter base psi device planewave_serial cell_info + SOURCES test_chg_mix.cpp + ../chg_mix.cpp ../chg_mix_rho.cpp + ../chg_dmr.cpp + ../chg_drho.cpp + ../chg_drho_inner.cpp + ../chg_precond.cpp ../chg_tau.cpp + ../chg_uspp.cpp +) + +AddTest( + TARGET MODULE_CHARGE_symm + LIBS parameter base device cell_info planewave_serial symmetry + SOURCES test_chg_symm.cpp ../chg_symm.cpp ../chg_symm_detail.cpp ../charge.cpp ../chg_tools.cpp +) + +AddTest( + TARGET MODULE_CHARGE_symm_detail + LIBS base device cell_info planewave_serial symmetry + SOURCES test_chg_symm_detail.cpp ../chg_symm_detail.cpp +) + +AddTest( + TARGET MODULE_CHARGE_atomic_inner + LIBS parameter base device cell_info planewave_serial + SOURCES test_chg_atomic_inner.cpp ../chg_atomic_inner.cpp ../charge.cpp ../chg_tools.cpp +) + +AddTest( + TARGET MODULE_CHARGE_atomic + LIBS parameter base device cell_info planewave_serial + SOURCES test_chg_atomic.cpp ../chg_atomic.cpp ../chg_atomic_inner.cpp ../charge.cpp ../chg_tools.cpp +) + +AddTest( + TARGET MODULE_CHARGE_tau + LIBS parameter base psi device planewave_serial cell_info + SOURCES test_chg_tau.cpp ../chg_tau.cpp ../chg_uspp.cpp ../charge.cpp ../chg_tools.cpp +) + +# MODULE_CHARGE_routine is temporarily disabled: chg_routine.cpp couples to +# heavy external symbols (Plus_U_Base from source_pw/module_pwdft, elecstate +# ::update_pot / ElecState::cal_energies) whose transitive dependencies are +# hard to satisfy in this test; chg_mix_rho.cpp also calls +# module_charge::detail::mix_tau_recip so chg_tau.cpp must be linked too. +# TODO: untangle this deep coupling later before re-enabling. +# AddTest( +# TARGET MODULE_CHARGE_routine +# LIBS parameter base psi device planewave_serial cell_info +# SOURCES test_chg_routine.cpp ../chg_routine.cpp ../chg_dmr.cpp ../charge.cpp ../chg_tools.cpp +# ../chg_mix.cpp ../chg_mix_rho.cpp ../chg_drho.cpp ../chg_drho_inner.cpp ../chg_precond.cpp ../chg_uspp.cpp +# ) + +# MODULE_CHARGE_init is temporarily disabled: chg_init.cpp pulls in symbols +# from source_io (ModuleIO::read_vdata_palgrid, ModuleIO::read_wf2rho_pw, +# Restart::read_file2) whose dependency chains are complex (read_wf2rho_pw +# alone drags in chg_parallel/chg_symm/elecstate_op), and the test file also +# stubs GlobalC::restart which conflicts with linking the real restart.cpp. +# TODO: resolve this dependency issue later before re-enabling. +# AddTest( +# TARGET MODULE_CHARGE_init +# LIBS parameter base device cell_info planewave_serial +# SOURCES test_chg_init.cpp ../chg_init.cpp ../chg_atomic.cpp ../chg_atomic_inner.cpp ../chg_rhog_io.cpp ../charge.cpp ../chg_tools.cpp +# ) + +AddTest( + TARGET MODULE_CHARGE_parallel + LIBS parameter psi base device planewave + SOURCES test_chg_parallel.cpp ../chg_parallel.cpp + # Real MPI test: its main() calls MPI_Init unconditionally, so it must keep + # __MPI despite this directory disabling it. + KEEP_FEATURE_DEFINITIONS __MPI +) + +add_test(NAME MODULE_CHARGE_parallel_4np + COMMAND mpirun -np 4 ./MODULE_CHARGE_parallel; + WORKING_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR} +) + +endif() diff --git a/source/source_estate/test/prepare_unitcell.h b/source/source_estate/module_charge/unittests/prepare_unitcell.h similarity index 100% rename from source/source_estate/test/prepare_unitcell.h rename to source/source_estate/module_charge/unittests/prepare_unitcell.h diff --git a/source/source_estate/module_charge/unittests/test_charge.cpp b/source/source_estate/module_charge/unittests/test_charge.cpp new file mode 100644 index 00000000000..ad65a910293 --- /dev/null +++ b/source/source_estate/module_charge/unittests/test_charge.cpp @@ -0,0 +1,201 @@ +#include "gtest/gtest.h" +#include "gmock/gmock.h" + +#include "source_cell/unitcell.h" +#include "source_estate/module_charge/charge.h" +#include "source_estate/module_charge/chg_tools.h" +#include "source_hamilt/module_xc/xc_functional.h" +#include "prepare_unitcell.h" +// mock functions for UnitCell + +Magnetism::Magnetism() +{ + this->tot_mag = 0.0; + this->abs_mag = 0.0; +} +Magnetism::~Magnetism() +{ +} + +// mock functions for Charge +// xc_functional.cpp is not linked into this target, so the private statics +// need a definition here. Defining them out of line does not require access +// to the class, only changing them does - that goes through the setters. +int XC_Functional::func_type = 1; +bool XC_Functional::ked_flag = false; +namespace elecstate +{ +double tmp_ucell_omega = 500.0; +double tmp_gridecut = 80.0; +} // namespace elecstate + +/************************************************ + * unit test of module_charge/charge.cpp + ***********************************************/ + +/** + * - Tested Functions: + * - Constructor: Charge::Charge() and Charge::~Charge() + * - this is a trivial test + * - Allocate: Charge::set_rhopw(), Charge::allocate(), Charge::destroy() + * - allocate rho, rhog, rho_save, rhog_save, kin_r, kin_r_save + * - using rhopw and the nspin passed in + * - SumRho: Charge::sum_rho() + * - calculate \sum_{is}^nspin \sum_{ir}^nrxx rho[is][ir] + * - RenormalizeRho: Charge::renormalize_rho() + * - renormalize rho so as to ensure the sum of rho equals to total number of electrons + * - CheckNe: module_charge::cal_rho2ne() + * - check the total number of electrons summed from rho[is] + * - SaveRhoBeforeSumBand: Charge::save_rho_before_sum_band() + * - meaning as the function name + */ + +class ChargeTest : public ::testing::Test +{ + protected: + UcellTestPrepare utp = UcellTestLib["Si"]; + std::unique_ptr ucell; + Charge* charge; + ModulePW::PW_Basis* rhopw; + std::string output; + /// Charge::allocate() and Charge::renormalize_rho() take these explicitly, + /// so the fixture owns them instead of writing the global parameter + /// singleton. The values mirror the Input_para defaults the test relied on. + int nspin = 1; + int test_charge = 0; + double nelec = 8; + bool out_elf_on = false; + void SetUp() override + { + ucell = utp.SetUcellInfo(); + charge = new Charge; + rhopw = new ModulePW::PW_Basis; + rhopw->initgrids(ucell->lat0, ucell->latvec, elecstate::tmp_gridecut); + rhopw->initparameters(false, elecstate::tmp_gridecut); + // setuptransform() runs distribute_r() then distribute_g(), both of + // which are protected; this is the public route to the same state. + rhopw->setuptransform(); + } + void TearDown() override + { + delete charge; + delete rhopw; + } +}; + +TEST_F(ChargeTest, Constructor) +{ + EXPECT_FALSE(charge->get_allocate_rho()); +} + +TEST_F(ChargeTest, Allocate) +{ + // ucell info + EXPECT_DOUBLE_EQ(ucell->omega, 265.302); + // rhopw info + EXPECT_DOUBLE_EQ(rhopw->lat0, 10.2); + EXPECT_EQ(rhopw->nx, 24); + EXPECT_EQ(rhopw->ny, 24); + EXPECT_EQ(rhopw->nz, 24); + EXPECT_EQ(rhopw->nxyz, 13824); + EXPECT_EQ(rhopw->nrxx, 13824); + EXPECT_EQ(rhopw->npw, 3143); + EXPECT_EQ(rhopw->npwtot, 3143); + // call Charge::allocate() + test_charge = 2; + XC_Functional::set_func_type(3); + XC_Functional::set_ked_flag(true); + charge->set_rhopw(rhopw); + EXPECT_FALSE(charge->get_allocate_rho()); + const bool kin_den = XC_Functional::get_ked_flag() || out_elf_on; + charge->allocate(nspin, kin_den, XC_Functional::get_ked_flag(), test_charge); + EXPECT_TRUE(charge->get_allocate_rho()); + // test if Charge::allocate() be called twice + EXPECT_NO_THROW(charge->allocate(nspin, kin_den, XC_Functional::get_ked_flag(), + test_charge)); + EXPECT_TRUE(charge->get_allocate_rho()); +} + +TEST_F(ChargeTest, SumRho) +{ + charge->set_rhopw(rhopw); + EXPECT_FALSE(charge->get_allocate_rho()); + const bool kin_den = XC_Functional::get_ked_flag() || out_elf_on; + charge->allocate(nspin, kin_den, XC_Functional::get_ked_flag(), test_charge); + EXPECT_TRUE(charge->get_allocate_rho()); + int nspin_rho = (nspin == 2) ? 2 : 1; + for (int is = 0; is < nspin_rho; is++) + { + for (int ir = 0; ir < rhopw->nrxx; ir++) + { + charge->rho[is][ir] = 0.1; + } + } + EXPECT_NEAR(charge->sum_rho(ucell->omega), 0.1 * nspin_rho * rhopw->nrxx * ucell->omega / rhopw->nxyz, 1E-10); +} + +TEST_F(ChargeTest, RenormalizeRho) +{ + charge->set_rhopw(rhopw); + EXPECT_FALSE(charge->get_allocate_rho()); + const bool kin_den = XC_Functional::get_ked_flag() || out_elf_on; + charge->allocate(nspin, kin_den, XC_Functional::get_ked_flag(), test_charge); + EXPECT_TRUE(charge->get_allocate_rho()); + int nspin_rho = (nspin == 2) ? 2 : 1; + for (int is = 0; is < nspin_rho; is++) + { + for (int ir = 0; ir < rhopw->nrxx; ir++) + { + charge->rho[is][ir] = 0.1; + } + } + EXPECT_EQ(nelec, 8); + charge->renormalize_rho(nelec, ucell->omega); + EXPECT_NEAR(charge->sum_rho(ucell->omega), 8.0, 1e-10); +} + +TEST_F(ChargeTest, CheckNe) +{ + charge->set_rhopw(rhopw); + EXPECT_FALSE(charge->get_allocate_rho()); + const bool kin_den = XC_Functional::get_ked_flag() || out_elf_on; + charge->allocate(nspin, kin_den, XC_Functional::get_ked_flag(), test_charge); + EXPECT_TRUE(charge->get_allocate_rho()); + int nspin_rho = (nspin == 2) ? 2 : 1; + for (int is = 0; is < nspin_rho; is++) + { + for (int ir = 0; ir < rhopw->nrxx; ir++) + { + charge->rho[is][ir] = 0.1; + } + } + EXPECT_EQ(nelec, 8); + charge->renormalize_rho(nelec, ucell->omega); + EXPECT_NEAR(charge->sum_rho(ucell->omega), 8.0, 1e-10); + EXPECT_NEAR(module_charge::cal_rho2ne(charge->rho[0], rhopw->nrxx, ucell->omega, rhopw->nxyz), + 8.0, 1e-10); +} + +TEST_F(ChargeTest, SaveRhoBeforeSumBand) +{ + charge->set_rhopw(rhopw); + EXPECT_FALSE(charge->get_allocate_rho()); + const bool kin_den = XC_Functional::get_ked_flag() || out_elf_on; + charge->allocate(nspin, kin_den, XC_Functional::get_ked_flag(), test_charge); + EXPECT_TRUE(charge->get_allocate_rho()); + int nspin_rho = (nspin == 2) ? 2 : 1; + for (int is = 0; is < nspin_rho; is++) + { + for (int ir = 0; ir < rhopw->nrxx; ir++) + { + charge->rho[is][ir] = 0.1; + } + } + EXPECT_EQ(nelec, 8); + XC_Functional::set_func_type(3); + XC_Functional::set_ked_flag(true); + charge->renormalize_rho(nelec, ucell->omega); + charge->save_rho_before_sum_band(); + EXPECT_NEAR(module_charge::cal_rho2ne(charge->rho_save[0], rhopw->nrxx, ucell->omega, rhopw->nxyz), + 8.0, 1e-10); +} diff --git a/source/source_estate/module_charge/unittests/test_chg_atomic.cpp b/source/source_estate/module_charge/unittests/test_chg_atomic.cpp new file mode 100644 index 00000000000..4d23355081c --- /dev/null +++ b/source/source_estate/module_charge/unittests/test_chg_atomic.cpp @@ -0,0 +1,80 @@ +#include "gtest/gtest.h" + +#include "source_base/matrix3.h" +#include "source_basis/module_pw/pw_basis.h" +#include "source_cell/unitcell.h" +#include "source_cell/atom_spec.h" +#include "source_cell/magnetism.h" +#include "source_estate/module_charge/charge.h" +#include "source_estate/module_charge/chg_atomic.h" + +#include +#include +#include +#include + +// charge.cpp references Magnetism; provide a lightweight stub. +Magnetism::Magnetism() +{ + this->tot_mag = 0.0; + this->abs_mag = 0.0; +} +Magnetism::~Magnetism() +{ +} + +/************************************************ + * unit test of module_charge/chg_atomic.cpp + ***********************************************/ + +/** + * - Tested Functions: + * - atomic_rho: dispatcher over spin_number_need (1/2/4) and per-atom + * start_mag. Covered: + * - ntype == 0 path: loop is skipped, only normalize_and_check runs. + */ + +namespace +{ + +module_charge::AtomicRhoCfg make_cfg(std::ostream& os) +{ + return {1.0, 0, false, false, os}; // nelec, test_charge, domag, domag_z, ofs_warning +} + +} // namespace + +TEST(ChgAtomicTest, AtomicRhoNtypeZeroOnlyNormalizes) +{ + ModulePW::PW_Basis rhopw; + rhopw.initgrids(1.0, ModuleBase::Matrix3(1, 0, 0, 0, 1, 0, 0, 0, 1), 10.0); + rhopw.initparameters(false, 10.0); + rhopw.setuptransform(); + rhopw.collect_local_pw(); + + UnitCell ucell; + ucell.ntype = 0; + ucell.nat = 0; + + const int spin_number_need = 1; + const double omega = rhopw.omega; + ModuleBase::ComplexMatrix strucFac(0, rhopw.npw); + std::vector rho_in(rhopw.nrxx, 0.0); + double* rho_ptrs[1] = {rho_in.data()}; + + std::stringstream ofs; + module_charge::AtomicRhoCfg cfg = make_cfg(ofs); + module_charge::atomic_rho(spin_number_need, omega, rho_ptrs, strucFac, ucell, &rhopw, cfg); + + // with ntype==0, rho_g3d is all zero, so normalize_and_check divides by + // ne_tot==0; the result is NaN/zero. We only assert no crash. + EXPECT_EQ(rho_in.size(), static_cast(rhopw.nrxx)); +} + +// The ChgAtomicDeathTest.AtomicRhoBadSpinAborts case was removed: it used +// EXPECT_DEATH to verify the WARNING_QUIT guard on unsupported +// spin_number_need (only 1/2/4 are valid). EXPECT_DEATH relies on fork(), +// which deadlocks when the linked OpenMP runtime has spawned worker threads +// (gtest warns "detected N threads"). The guard under test is a low-value +// default: branch in atomic_rho, and INPUT validation prevents an invalid +// spin_number_need from reaching this code in production. diff --git a/source/source_estate/module_charge/unittests/test_chg_atomic_inner.cpp b/source/source_estate/module_charge/unittests/test_chg_atomic_inner.cpp new file mode 100644 index 00000000000..bad0304591a --- /dev/null +++ b/source/source_estate/module_charge/unittests/test_chg_atomic_inner.cpp @@ -0,0 +1,148 @@ +#include "gtest/gtest.h" + +#include "source_base/math_integral.h" +#include "source_base/matrix3.h" +#include "source_basis/module_pw/pw_basis.h" +#include "source_cell/unitcell.h" +#include "source_cell/atom_spec.h" +#include "source_estate/module_charge/charge.h" +#include "source_estate/module_charge/chg_atomic_detail.h" + +#include +#include +#include +#include + +// charge.cpp references Magnetism; provide a lightweight stub. +Magnetism::Magnetism() +{ + this->tot_mag = 0.0; + this->abs_mag = 0.0; +} +Magnetism::~Magnetism() +{ +} + +/************************************************ + * unit test of module_charge/chg_atomic_inner.cpp + ***********************************************/ + +/** + * - Tested Functions: + * - compute_rhoatm: builds the atomic radial charge density for USPP + * (tvanp=true: direct copy of rho_at) and NCPP (tvanp=false: divide by + * 4*pi*r^2, extrapolate rho[0], Simpson-integrate, scale to zv, remultiply + * 4*pi*r^2). + * - normalize_and_check: FFTs rho_g3d to real space, sums electrons, + * renormalizes rho to the given nelec. + * + * For compute_rhoatm we use a Gaussian rho_at whose analytic integral is known. + */ + +namespace +{ + +/// Build a uniform radial mesh r[ir] = dr * ir, rab = dr. +void fill_uniform_mesh(Atom_pseudo& ncpp, int mesh, double dr, double zv, bool tvanp) +{ + ncpp.mesh = mesh; + ncpp.msh = mesh; + ncpp.zv = zv; + ncpp.tvanp = tvanp; + ncpp.r.assign(mesh, 0.0); + ncpp.rab.assign(mesh, dr); + ncpp.rho_at.assign(mesh, 0.0); + for (int ir = 0; ir < mesh; ++ir) + { + ncpp.r[ir] = dr * ir; + } +} + +/// rho_at(r) = 4 pi r^2 * Gaussian, so the number density is a pure Gaussian. +void fill_gaussian_rho_at(Atom_pseudo& ncpp, double alpha, double norm) +{ + for (int ir = 0; ir < ncpp.mesh; ++ir) + { + const double r = ncpp.r[ir]; + ncpp.rho_at[ir] = norm * ModuleBase::FOUR_PI * r * r * std::exp(-alpha * r * r); + } +} + +} // namespace + +TEST(ChgAtomicInnerTest, ComputeRhoatmUsppCopiesRhoAt) +{ + Atom atom; + fill_uniform_mesh(atom.ncpp, 8, 0.5, 8.0, true); + for (int ir = 0; ir < 8; ++ir) + { + atom.ncpp.rho_at[ir] = static_cast(ir + 1); + } + + std::stringstream ofs; + const std::vector rhoatm = module_charge::detail::compute_rhoatm(atom, 8, ofs); + + ASSERT_EQ(rhoatm.size(), 8u); + for (int ir = 0; ir < 8; ++ir) + { + EXPECT_EQ(rhoatm[ir], atom.ncpp.rho_at[ir]); + } +} + +TEST(ChgAtomicInnerTest, ComputeRhoatmNcppIntegratesAndScalesToZv) +{ + Atom atom; + const int mesh = 101; + const double dr = 0.1; + const double zv = 4.0; + fill_uniform_mesh(atom.ncpp, mesh, dr, zv, false); + // rho_at = 4 pi r^2 * exp(-r^2), integral over [0,inf) = pi^{3/2}. + fill_gaussian_rho_at(atom.ncpp, 1.0, 1.0); + + std::stringstream ofs; + const std::vector rhoatm = module_charge::detail::compute_rhoatm(atom, mesh, ofs); + + // for NCPP, rhoatm[ir] = rho_at[ir] (after scaling) for ir>0 because the + // /4pir^2 and *4pir^2 cancel; the net effect is scale = zv / charge. + double charge = 0.0; + ModuleBase::Integral::Simpson_Integral(atom.ncpp.msh, + atom.ncpp.rho_at.data(), + atom.ncpp.rab.data(), + charge); + const double scale = zv / charge; + for (int ir = 1; ir < mesh; ++ir) + { + EXPECT_NEAR(rhoatm[ir], atom.ncpp.rho_at[ir] * scale, 1e-8); + } +} + +TEST(ChgAtomicInnerTest, NormalizeAndCheckRenormalizesToNelec) +{ + ModulePW::PW_Basis rhopw; + rhopw.initgrids(1.0, ModuleBase::Matrix3(1, 0, 0, 0, 1, 0, 0, 0, 1), 10.0); + rhopw.initparameters(false, 10.0); + rhopw.setuptransform(); + rhopw.collect_local_pw(); + + const int spin_number_need = 1; + const double omega = rhopw.omega; + const double nelec = 5.0; + ModuleBase::ComplexMatrix rho_g3d(spin_number_need, rhopw.npw); + // put all weight at G=0 => uniform real-space density. + rho_g3d(0, 0) = std::complex(1.0, 0.0); + + std::vector rho_in(rhopw.nrxx, 0.0); + double* rho_ptrs[1] = {rho_in.data()}; + + std::stringstream ofs; + module_charge::detail::normalize_and_check(rho_ptrs, rho_g3d, &rhopw, + spin_number_need, omega, ofs, nelec); + + double ne = 0.0; + for (int ir = 0; ir < rhopw.nrxx; ++ir) + { + ne += rho_in[ir]; + } + ne *= omega / static_cast(rhopw.nxyz); + EXPECT_NEAR(ne, nelec, 1e-6); +} diff --git a/source/source_estate/module_charge/unittests/test_chg_dmr.cpp b/source/source_estate/module_charge/unittests/test_chg_dmr.cpp new file mode 100644 index 00000000000..15f1fb78dd6 --- /dev/null +++ b/source/source_estate/module_charge/unittests/test_chg_dmr.cpp @@ -0,0 +1,251 @@ +#include "gtest/gtest.h" + +#include "source_base/module_mixing/plain_mixing.h" +#include "source_estate/module_charge/chg_dmr.h" +#include "source_estate/module_charge/chg_mix_cfg.h" + +#include + +/************************************************ + * unit test of module_charge/chg_dmr.cpp + ***********************************************/ + +/** + * - Tested Functions: + * - init_mixing_dmr: allocate the DMR mixing buffer and reset history + * - scf_thr_type == 2: mdata.length == nnr * dmr_nspin, counters reset + * - scf_thr_type == 1: abort (PW basis not supported) + * - abort on mixing == nullptr or nnr < 0 + * - mix_dmr: mix the real-space density matrix + * - nspin == 1: out = in + beta * (out_new - in) + * - nspin == 2: up/down -> charge/mag channels, two betas, back + * - nspin == 4: treated like nspin == 1 (single channel) + * - nnr == 0: null buffers are allowed (empty partition) + * - abort on invalid inputs (null mixing, bad nspin, null buffer with nnr>0) + */ + +namespace +{ + +/// Build a MixingConfig with all fields explicitly initialized. +MixingConfig make_cfg(int nspin, double beta, double beta_mag, int scf_thr_type) +{ + MixingConfig cfg{ + "plain", // mixing_mode + beta, // mixing_beta + 4, // mixing_ndim + 0.0, // mixing_gg0 + false, // mixing_tau + beta_mag, // mixing_beta_mag + 0.0, // mixing_gg0_mag + 0.1, // mixing_gg0_min + -10.0, // mixing_angle + true, // mixing_dmr + nspin, // nspin + scf_thr_type,// scf_thr_type + false, // double_grid + false, // gamma_only_pw + false, // domag + false, // domag_z + 100 // scf_nmax + }; + return cfg; +} + +} // namespace + +class ChgDmrTest : public ::testing::Test +{ + protected: + Base_Mixing::Plain_Mixing mixing; + Base_Mixing::Mixing_Data mdata; +}; + +// --------------------------------------------------------------------------- +// init_mixing_dmr +// --------------------------------------------------------------------------- + +TEST_F(ChgDmrTest, InitMixingDmrNspin1AllocatesAndResets) +{ + const int nnr = 10; + MixingConfig cfg = make_cfg(1, 0.5, 0.5, 2); + module_charge::init_mixing_dmr(&mixing, mdata, nnr, cfg); + + // dmr_nspin = 1 for nspin == 1 + EXPECT_EQ(mdata.length, static_cast(nnr * 1)); + EXPECT_EQ(mdata.ndim_use, 0); + EXPECT_EQ(mdata.ndim_history, 0); + EXPECT_EQ(mdata.start, -1); +} + +TEST_F(ChgDmrTest, InitMixingDmrNspin2AllocatesTwoChannels) +{ + const int nnr = 7; + MixingConfig cfg = make_cfg(2, 0.5, 0.8, 2); + module_charge::init_mixing_dmr(&mixing, mdata, nnr, cfg); + + // dmr_nspin = 2 for nspin == 2 + EXPECT_EQ(mdata.length, static_cast(nnr * 2)); + EXPECT_EQ(mdata.ndim_use, 0); +} + +TEST_F(ChgDmrTest, InitMixingDmrPwThresholdAborts) +{ + const int nnr = 5; + MixingConfig cfg = make_cfg(1, 0.5, 0.5, 1); // scf_thr_type == 1 + EXPECT_DEATH(module_charge::init_mixing_dmr(&mixing, mdata, nnr, cfg), ""); +} + +TEST_F(ChgDmrTest, InitMixingDmrNullMixingAborts) +{ + const int nnr = 5; + MixingConfig cfg = make_cfg(1, 0.5, 0.5, 2); + EXPECT_DEATH(module_charge::init_mixing_dmr(nullptr, mdata, nnr, cfg), ""); +} + +TEST_F(ChgDmrTest, InitMixingDmrNegativeNnrAborts) +{ + MixingConfig cfg = make_cfg(1, 0.5, 0.5, 2); + EXPECT_DEATH(module_charge::init_mixing_dmr(&mixing, mdata, -1, cfg), ""); +} + +// --------------------------------------------------------------------------- +// mix_dmr nspin == 1 +// --------------------------------------------------------------------------- + +TEST_F(ChgDmrTest, MixDmrNspin1PlainStep) +{ + const int nnr = 4; + MixingConfig cfg = make_cfg(1, 0.5, 0.5, 2); + mixing.mixing_beta = 0.5; + + module_charge::init_mixing_dmr(&mixing, mdata, nnr, cfg); + + std::vector dmr_in(nnr, 1.0); // saved (previous) density matrix + std::vector dmr_out(nnr, 3.0); // new density matrix from this step + std::vector out_ptrs = {dmr_out.data()}; + std::vector in_ptrs = {dmr_in.data()}; + + module_charge::mix_dmr(out_ptrs, in_ptrs, nnr, &mixing, mdata, cfg); + + // plain mixing: out = in + beta * (out_new - in) = 1 + 0.5 * (3 - 1) = 2 + for (int i = 0; i < nnr; ++i) + { + EXPECT_NEAR(dmr_out[i], 2.0, 1e-12); + } +} + +// --------------------------------------------------------------------------- +// mix_dmr nspin == 2 (charge / magnetization channels with two betas) +// --------------------------------------------------------------------------- + +TEST_F(ChgDmrTest, MixDmrNspin2ChargeConservation) +{ + const int nnr = 3; + const double beta = 0.5; + const double beta_mag = 0.8; + MixingConfig cfg = make_cfg(2, beta, beta_mag, 2); + mixing.mixing_beta = beta; + + module_charge::init_mixing_dmr(&mixing, mdata, nnr, cfg); + + // up/down saved and new + std::vector up_in(nnr, 1.0); + std::vector dn_in(nnr, 2.0); + std::vector up_out(nnr, 3.0); + std::vector dn_out(nnr, 4.0); + + std::vector out_ptrs = {up_out.data(), dn_out.data()}; + std::vector in_ptrs = {up_in.data(), dn_in.data()}; + + module_charge::mix_dmr(out_ptrs, in_ptrs, nnr, &mixing, mdata, cfg); + + // charge channel: c_save = 1+2 = 3, c_new = 3+4 = 7 + // c_mix = 3 + 0.5 * (7 - 3) = 5 + // mag channel: m_save = 1-2 = -1, m_new = 3-4 = -1 + // m_mix = -1 + 0.8 * (-1 - (-1)) = -1 + // up = 0.5 * (5 + (-1)) = 2 + // dn = 0.5 * (5 - (-1)) = 3 + for (int i = 0; i < nnr; ++i) + { + EXPECT_NEAR(up_out[i], 2.0, 1e-12); + EXPECT_NEAR(dn_out[i], 3.0, 1e-12); + } +} + +// --------------------------------------------------------------------------- +// mix_dmr nspin == 4 (treated as single channel like nspin == 1) +// --------------------------------------------------------------------------- + +TEST_F(ChgDmrTest, MixDmrNspin4SingleChannel) +{ + const int nnr = 3; + MixingConfig cfg = make_cfg(4, 0.5, 0.5, 2); + mixing.mixing_beta = 0.5; + + module_charge::init_mixing_dmr(&mixing, mdata, nnr, cfg); + + std::vector dmr_in(nnr, 2.0); + std::vector dmr_out(nnr, 6.0); + std::vector out_ptrs = {dmr_out.data()}; + std::vector in_ptrs = {dmr_in.data()}; + + module_charge::mix_dmr(out_ptrs, in_ptrs, nnr, &mixing, mdata, cfg); + + // out = 2 + 0.5 * (6 - 2) = 4 + for (int i = 0; i < nnr; ++i) + { + EXPECT_NEAR(dmr_out[i], 4.0, 1e-12); + } +} + +// --------------------------------------------------------------------------- +// boundary: empty partition (nnr == 0) allows null buffers +// --------------------------------------------------------------------------- + +TEST_F(ChgDmrTest, MixDmrZeroNnrAllowsNullBuffers) +{ + const int nnr = 0; + MixingConfig cfg = make_cfg(1, 0.5, 0.5, 2); + module_charge::init_mixing_dmr(&mixing, mdata, nnr, cfg); + + // null buffers are legitimate when the rank owns no DMR elements + std::vector out_ptrs = {nullptr}; + std::vector in_ptrs = {nullptr}; + EXPECT_NO_THROW(module_charge::mix_dmr(out_ptrs, in_ptrs, nnr, &mixing, mdata, cfg)); +} + +// --------------------------------------------------------------------------- +// abort cases +// --------------------------------------------------------------------------- + +TEST_F(ChgDmrTest, MixDmrNullMixingAborts) +{ + const int nnr = 4; + MixingConfig cfg = make_cfg(1, 0.5, 0.5, 2); + std::vector buf(nnr, 0.0); + std::vector out_ptrs = {buf.data()}; + std::vector in_ptrs = {buf.data()}; + EXPECT_DEATH(module_charge::mix_dmr(out_ptrs, in_ptrs, nnr, nullptr, mdata, cfg), ""); +} + +TEST_F(ChgDmrTest, MixDmrBadNspinAborts) +{ + const int nnr = 4; + MixingConfig cfg = make_cfg(3, 0.5, 0.5, 2); // nspin == 3 not supported + std::vector buf(nnr, 0.0); + std::vector out_ptrs = {buf.data()}; + std::vector in_ptrs = {buf.data()}; + EXPECT_DEATH(module_charge::mix_dmr(out_ptrs, in_ptrs, nnr, &mixing, mdata, cfg), ""); +} + +TEST_F(ChgDmrTest, MixDmrNullBufferWithPositiveNnrAborts) +{ + const int nnr = 4; + MixingConfig cfg = make_cfg(1, 0.5, 0.5, 2); + std::vector buf(nnr, 0.0); + // dmr_out is null while nnr > 0 + std::vector out_ptrs = {nullptr}; + std::vector in_ptrs = {buf.data()}; + EXPECT_DEATH(module_charge::mix_dmr(out_ptrs, in_ptrs, nnr, &mixing, mdata, cfg), ""); +} diff --git a/source/source_estate/module_charge/unittests/test_chg_drho.cpp b/source/source_estate/module_charge/unittests/test_chg_drho.cpp new file mode 100644 index 00000000000..8093d834918 --- /dev/null +++ b/source/source_estate/module_charge/unittests/test_chg_drho.cpp @@ -0,0 +1,254 @@ +#include "gtest/gtest.h" + +#include "source_base/matrix3.h" +#include "source_basis/module_pw/pw_basis.h" +#include "source_cell/magnetism.h" +#include "source_estate/module_charge/charge.h" +#include "source_estate/module_charge/chg_drho.h" +#include "source_estate/module_charge/chg_mix_cfg.h" + +#include +#include + +// charge.cpp references Magnetism; provide a lightweight stub. +Magnetism::Magnetism() +{ + this->tot_mag = 0.0; + this->abs_mag = 0.0; +} +Magnetism::~Magnetism() +{ +} + +/************************************************ + * unit test of module_charge/chg_drho.cpp + ***********************************************/ + +/** + * - Tested Functions: + * - inner_product_real: sum of rho1*rho2 over nrxx*nspin/resize_tmp + * - nspin == 1: single block + * - nspin == 4 with mixing_angle > 0: resize_tmp == 2 (half length) + * - cal_drho: real-space (scf_thr_type == 2) L1 residual normalized by nelec + * - nspin == 1: sum over single channel + * - nspin == 2: both channels + * - nspin == 4 with domag_z: only channels 0 and 3 + * - cal_dkin: kinetic-energy-density residual + * - meta_gga == false: returns 0 + * - meta_gga == true: same L1 logic as cal_drho real-space + */ + +namespace +{ + +MixingConfig make_cfg(int nspin, int scf_thr_type, bool domag_z = false, double mixing_angle = -10.0) +{ + MixingConfig cfg{ + "plain", // mixing_mode + 0.8, // mixing_beta + 4, // mixing_ndim + 0.0, // mixing_gg0 + false, // mixing_tau + 1.6, // mixing_beta_mag + 0.0, // mixing_gg0_mag + 0.1, // mixing_gg0_min + mixing_angle, // mixing_angle + false, // mixing_dmr + nspin, // nspin + scf_thr_type, // scf_thr_type + false, // double_grid + false, // gamma_only_pw + false, // domag + domag_z, // domag_z + 100 // scf_nmax + }; + return cfg; +} + +} // namespace + +class ChgDrhoTest : public ::testing::Test +{ + protected: + ModulePW::PW_Basis pw_basis; + Charge charge; + + void SetUp() override + { + pw_basis.initgrids(4, ModuleBase::Matrix3(1, 0, 0, 0, 1, 0, 0, 0, 1), 20); + pw_basis.initparameters(false, 20); + pw_basis.setuptransform(); + pw_basis.collect_local_pw(); + } + + /// Allocate Charge buffers for nspin with kin_r allocated. + void setup_charge(int nspin) + { + charge.set_rhopw(&pw_basis); + const bool kin_den = true; + const bool meta_gga = true; + charge.allocate(nspin, kin_den, meta_gga, 0); + } +}; + +// --------------------------------------------------------------------------- +// inner_product_real +// --------------------------------------------------------------------------- + +TEST_F(ChgDrhoTest, InnerProductRealNspin1) +{ + const int nspin = 1; + MixingConfig cfg = make_cfg(nspin, 2); + std::vector rho1(pw_basis.nrxx, 2.0); + std::vector rho2(pw_basis.nrxx, 3.0); + + const double inner = module_charge::inner_product_real( + rho1.data(), rho2.data(), pw_basis, cfg); + + // sum of 2.0 * 3.0 over nrxx elements + EXPECT_NEAR(inner, 6.0 * pw_basis.nrxx, 1e-8); +} + +TEST_F(ChgDrhoTest, InnerProductRealNspin4AngleHalvesLength) +{ + const int nspin = 4; + MixingConfig cfg = make_cfg(nspin, 2, false, 1.0); // mixing_angle > 0 + const int len = pw_basis.nrxx * nspin / 2; // resize_tmp == 2 + std::vector rho1(len, 1.0); + std::vector rho2(len, 1.0); + + const double inner = module_charge::inner_product_real( + rho1.data(), rho2.data(), pw_basis, cfg); + + EXPECT_NEAR(inner, static_cast(len), 1e-8); +} + +// --------------------------------------------------------------------------- +// cal_drho (real-space path, scf_thr_type == 2) +// --------------------------------------------------------------------------- + +TEST_F(ChgDrhoTest, CalDrhoRealNspin1) +{ + const int nspin = 1; + setup_charge(nspin); + MixingConfig cfg = make_cfg(nspin, 2); + + const double nelec = 4.0; + const double omega = 1.0; + + // rho - rho_save = 1.0 everywhere + for (int ir = 0; ir < pw_basis.nrxx; ++ir) + { + charge.rho[0][ir] = 2.0; + charge.rho_save[0][ir] = 1.0; + } + + const double drho = module_charge::cal_drho( + &charge, nelec, pw_basis, cfg, omega, 1.0); + + // drho = sum|diff| * omega/nxyz / nelec = nrxx * 1.0 * 1.0/nxyz / 4.0 + const double ref = static_cast(pw_basis.nrxx) * 1.0 + * omega / static_cast(pw_basis.nxyz) / nelec; + EXPECT_NEAR(drho, ref, 1e-10); +} + +TEST_F(ChgDrhoTest, CalDrhoRealNspin2BothChannels) +{ + const int nspin = 2; + setup_charge(nspin); + MixingConfig cfg = make_cfg(nspin, 2, false); + + const double nelec = 4.0; + const double omega = 1.0; + + for (int is = 0; is < nspin; ++is) + { + for (int ir = 0; ir < pw_basis.nrxx; ++ir) + { + charge.rho[is][ir] = 3.0; + charge.rho_save[is][ir] = 1.0; // diff = 2.0 + } + } + + const double drho = module_charge::cal_drho( + &charge, nelec, pw_basis, cfg, omega, 1.0); + + // both channels contribute: 2 * nrxx * 2.0 * omega/nxyz / nelec + const double ref = 2.0 * pw_basis.nrxx * 2.0 + * omega / static_cast(pw_basis.nxyz) / nelec; + EXPECT_NEAR(drho, ref, 1e-10); +} + +TEST_F(ChgDrhoTest, CalDrhoRealNspin4DomagZOnlyChannels0And3) +{ + const int nspin = 4; + setup_charge(nspin); + MixingConfig cfg = make_cfg(nspin, 2, true); // domag_z == true + + const double nelec = 4.0; + const double omega = 1.0; + + for (int is = 0; is < nspin; ++is) + { + for (int ir = 0; ir < pw_basis.nrxx; ++ir) + { + charge.rho[is][ir] = 3.0; + charge.rho_save[is][ir] = 1.0; // diff = 2.0 + } + } + + const double drho = module_charge::cal_drho( + &charge, nelec, pw_basis, cfg, omega, 1.0); + + // only is==0 and is==3 are summed when domag_z is true + const double ref = 2.0 * pw_basis.nrxx * 2.0 + * omega / static_cast(pw_basis.nxyz) / nelec; + EXPECT_NEAR(drho, ref, 1e-10); +} + +// --------------------------------------------------------------------------- +// cal_dkin +// --------------------------------------------------------------------------- + +TEST_F(ChgDrhoTest, CalDkinMetaGgaFalseReturnsZero) +{ + const int nspin = 1; + setup_charge(nspin); + charge.meta_gga = false; + MixingConfig cfg = make_cfg(nspin, 2); + + for (int ir = 0; ir < pw_basis.nrxx; ++ir) + { + charge.kin_r[0][ir] = 5.0; + charge.kin_r_save[0][ir] = 1.0; + } + + const double dkin = module_charge::cal_dkin( + &charge, 4.0, pw_basis, cfg, 1.0); + + EXPECT_NEAR(dkin, 0.0, 1e-12); +} + +TEST_F(ChgDrhoTest, CalDkinMetaGgaTrueComputesResidual) +{ + const int nspin = 1; + setup_charge(nspin); + charge.meta_gga = true; + MixingConfig cfg = make_cfg(nspin, 2); + + const double nelec = 4.0; + const double omega = 1.0; + + for (int ir = 0; ir < pw_basis.nrxx; ++ir) + { + charge.kin_r[0][ir] = 4.0; + charge.kin_r_save[0][ir] = 1.0; // diff = 3.0 + } + + const double dkin = module_charge::cal_dkin( + &charge, nelec, pw_basis, cfg, omega); + + const double ref = static_cast(pw_basis.nrxx) * 3.0 + * omega / static_cast(pw_basis.nxyz) / nelec; + EXPECT_NEAR(dkin, ref, 1e-10); +} diff --git a/source/source_estate/module_charge/unittests/test_chg_drho_inner.cpp b/source/source_estate/module_charge/unittests/test_chg_drho_inner.cpp new file mode 100644 index 00000000000..c53e0373ca0 --- /dev/null +++ b/source/source_estate/module_charge/unittests/test_chg_drho_inner.cpp @@ -0,0 +1,135 @@ +#include "gtest/gtest.h" + +#include "source_base/constants.h" +#include "source_base/matrix3.h" +#include "source_basis/module_pw/pw_basis.h" +#include "source_estate/module_charge/chg_drho.h" +#include "source_estate/module_charge/chg_drho_detail.h" +#include "source_estate/module_charge/chg_mix_cfg.h" + +#include +#include + +/************************************************ + * unit test of module_charge/chg_drho_inner.cpp + ***********************************************/ + +/** + * - Tested Functions: + * - inner_product_recip_rho (module_charge::detail): + * Coulomb-metric reciprocal inner product of the charge residual + * - nspin == 1: sum over G!=0 of conj(rho1)*rho2 / gg, weighted by fac + * - inner_product_recip_hartree: + * Hartree-like reciprocal inner product used in charge mixing + * - nspin == 1: same Coulomb sum as inner_product_recip_rho + * + * Both are tested with a single nonzero G component (ig=1, assuming ig0=0) + * so the analytic value is a single term. + */ + +namespace +{ + +MixingConfig make_cfg(int nspin) +{ + MixingConfig cfg{ + "plain", // mixing_mode + 0.8, // mixing_beta + 4, // mixing_ndim + 0.0, // mixing_gg0 + false, // mixing_tau + 1.6, // mixing_beta_mag + 0.0, // mixing_gg0_mag + 0.1, // mixing_gg0_min + -10.0, // mixing_angle + false, // mixing_dmr + nspin, // nspin + 2, // scf_thr_type + false, // double_grid + false, // gamma_only_pw + false, // domag + false, // domag_z + 100 // scf_nmax + }; + return cfg; +} + +} // namespace + +class ChgDrhoInnerTest : public ::testing::Test +{ + protected: + ModulePW::PW_Basis pw_basis; + + void SetUp() override + { + pw_basis.initgrids(4, ModuleBase::Matrix3(1, 0, 0, 0, 1, 0, 0, 0, 1), 20); + pw_basis.initparameters(false, 20); + pw_basis.setuptransform(); + pw_basis.collect_local_pw(); + } +}; + +TEST_F(ChgDrhoInnerTest, InnerProductRecipRhoNspin1SingleG) +{ + const int nspin = 1; + MixingConfig cfg = make_cfg(nspin); + + // place a single nonzero component at ig=1 (ig0 == 0 is the G=0 vector) + const int ig = 1; + std::vector> rho1(nspin * pw_basis.npw, std::complex(0.0, 0.0)); + std::vector> rho2(nspin * pw_basis.npw, std::complex(0.0, 0.0)); + rho1[ig] = std::complex(2.0, 1.0); + rho2[ig] = std::complex(3.0, -1.0); + + const double omega = 1.0; + const double tpiba = 1.0; + const double inner = module_charge::detail::inner_product_recip_rho( + rho1.data(), rho2.data(), pw_basis, cfg, omega, tpiba); + + const double fac = ModuleBase::e2 * ModuleBase::FOUR_PI / (tpiba * tpiba); + const double gg = pw_basis.gg[ig]; + // (conj(2+i) * (3-i)).real() = (2-i)*(3-i) = 6 -2i -3i + i^2 = 5 -5i, real = 5 + const double overlap = (std::conj(rho1[ig]) * rho2[ig]).real(); + const double ref = fac * overlap / gg * omega * 0.5; + + EXPECT_NEAR(inner, ref, 1e-8); +} + +TEST_F(ChgDrhoInnerTest, InnerProductRecipHartreeNspin1SingleG) +{ + const int nspin = 1; + MixingConfig cfg = make_cfg(nspin); + + const int ig = 1; + std::vector> rhog1(nspin * pw_basis.npw, std::complex(0.0, 0.0)); + std::vector> rhog2(nspin * pw_basis.npw, std::complex(0.0, 0.0)); + rhog1[ig] = std::complex(1.0, 0.0); + rhog2[ig] = std::complex(2.0, 0.0); + + const double omega = 1.0; + const double tpiba = 1.0; + const double inner = module_charge::inner_product_recip_hartree( + rhog1.data(), rhog2.data(), pw_basis, cfg, omega, tpiba); + + const double fac = ModuleBase::e2 * ModuleBase::FOUR_PI / (tpiba * tpiba); + const double gg = pw_basis.gg[ig]; + const double overlap = (std::conj(rhog1[ig]) * rhog2[ig]).real(); + const double ref = fac * overlap / gg * omega * 0.5; + + EXPECT_NEAR(inner, ref, 1e-8); +} + +TEST_F(ChgDrhoInnerTest, InnerProductRecipRhoNspin1ZeroInput) +{ + const int nspin = 1; + MixingConfig cfg = make_cfg(nspin); + + std::vector> rho1(nspin * pw_basis.npw, std::complex(0.0, 0.0)); + std::vector> rho2(nspin * pw_basis.npw, std::complex(0.0, 0.0)); + + const double inner = module_charge::detail::inner_product_recip_rho( + rho1.data(), rho2.data(), pw_basis, cfg, 1.0, 1.0); + + EXPECT_NEAR(inner, 0.0, 1e-12); +} diff --git a/source/source_estate/module_charge/unittests/test_chg_extra.cpp b/source/source_estate/module_charge/unittests/test_chg_extra.cpp new file mode 100644 index 00000000000..962960fa726 --- /dev/null +++ b/source/source_estate/module_charge/unittests/test_chg_extra.cpp @@ -0,0 +1,331 @@ +#include "gmock/gmock.h" +#include "gtest/gtest.h" +#include "source_io/module_parameter/parameter.h" +#include "source_estate/module_charge/chg_atomic.h" +#include "source_estate/module_charge/chg_extra.h" +#include "prepare_unitcell.h" +#include "source_base/module_fft/fft_bundle.h" +// mock functions for UnitCell + +Magnetism::Magnetism() +{ +} +Magnetism::~Magnetism() +{ +} + + +// mock functions for Charge +Charge::Charge() +{ + rhopw = new ModulePW::PW_Basis; + rhopw->nrxx = 8; + rhopw->nx = 2; + rhopw->ny = 2; + rhopw->nz = 2; + rho = new double*[1]; + rho[0] = new double[rhopw->nrxx]; + ModuleBase::GlobalFunc::ZEROS(rho[0], rhopw->nrxx); + for (int i = 0; i < rhopw->nrxx; ++i) + { + rho[0][i] = i + 1; + } +} +Charge::~Charge() +{ + delete[] rho[0]; + delete[] rho; + delete rhopw; +} + +// mock for module_charge::atomic_rho to keep this unit test isolated +// from the full PW basis implementation. +namespace module_charge +{ +void atomic_rho(const int spin_number_need, + const double& omega, + double** rho_in, + const ModuleBase::ComplexMatrix& strucFac, + const UnitCell& ucell, + const ModulePW::PW_Basis* rhopw, + const AtomicRhoCfg& cfg) +{ +} +} // namespace module_charge + +// mock functions for PW_Basis +namespace ModulePW +{ +PW_Basis::PW_Basis() +{ +} +PW_Basis::~PW_Basis() +{ +} +void PW_Basis::initgrids(const double lat0_in, const ModuleBase::Matrix3 latvec_in, const double gridecut) +{ +} +void PW_Basis::initgrids(const double lat0_in, + const ModuleBase::Matrix3 latvec_in, + const int nx_in, + int ny_in, + int nz_in) +{ +} +void PW_Basis::distribute_r() +{ +} +} // namespace ModulePW + +// mock functions for Structure_Factor +Structure_Factor::Structure_Factor() +{ +} +Structure_Factor::~Structure_Factor() +{ +} +void Structure_Factor::setup(const UnitCell*, const Parallel_Grid&, const ModulePW::PW_Basis*, const bool) +{ +} + +/************************************************ + * unit test of module_charge/chg_extra.cpp + ***********************************************/ + +/** + * - Tested Functions: + * - Charge_Extra::Init_CE() + * - Initialization of viriables used in charge extrapolation methods + * - Charge_Extra::extrapolate_charge() + * - charge extrapolation + * - Charge_Extra::update_all_dis() + * - update displacements + * - Charge_Extra::find_alpha_and_beta() + * - determine alpha and beta + */ + +class ChargeExtraTest : public ::testing::Test +{ + protected: + module_charge::Charge_Extra CE; + UcellTestPrepare utp = UcellTestLib["Si"]; + std::unique_ptr ucell; + Parallel_Grid* pgrid = nullptr; + Charge charge; + Structure_Factor sf; + /// Init_CE() takes the spin count and the extrapolation method as + /// arguments, so the fixture owns them instead of staging them in the + /// global parameter singleton. + int nspin = 1; + std::string chg_extrap; + void SetUp() override + { + ucell = utp.SetUcellInfo(); + ucell->omega = 1.0; + } + void TearDown() override + { + } +}; + +TEST_F(ChargeExtraTest, InitCEWarningQuit) +{ + chg_extrap ="wwww"; + testing::internal::CaptureStdout(); + EXPECT_EXIT(CE.Init_CE(nspin, ucell->nat, charge.rhopw->nrxx, chg_extrap), + ::testing::ExitedWithCode(1), + ""); + std::string output = testing::internal::GetCapturedStdout(); + EXPECT_THAT(output, testing::HasSubstr("charge extrapolation method is not available")); +} + +TEST_F(ChargeExtraTest, InitCECase1) +{ + chg_extrap ="none"; + CE.Init_CE(nspin, ucell->nat, charge.rhopw->nrxx, chg_extrap); + EXPECT_EQ(CE.get_pot_order(), 0); +} + +TEST_F(ChargeExtraTest, InitCECase2) +{ + chg_extrap ="atomic"; + CE.Init_CE(nspin, ucell->nat, charge.rhopw->nrxx, chg_extrap); + EXPECT_EQ(CE.get_pot_order(), 1); +} + +TEST_F(ChargeExtraTest, InitCECase3) +{ + chg_extrap ="first-order"; + CE.Init_CE(nspin, ucell->nat, charge.rhopw->nrxx, chg_extrap); + EXPECT_EQ(CE.get_pot_order(), 2); + EXPECT_NE(CE.get_delta_rho1().size(), 0); + EXPECT_NE(CE.get_delta_rho2().size(), 0); +} + +TEST_F(ChargeExtraTest, InitCECase4) +{ + chg_extrap ="second-order"; + CE.Init_CE(nspin, ucell->nat, charge.rhopw->nrxx, chg_extrap); + EXPECT_EQ(CE.get_pot_order(), 3); + EXPECT_DOUBLE_EQ(CE.get_alpha(), 1.0); + EXPECT_DOUBLE_EQ(CE.get_beta(), 0.0); + EXPECT_NE(CE.get_delta_rho1().size(), 0); + EXPECT_NE(CE.get_delta_rho2().size(), 0); + EXPECT_EQ(CE.get_dis_old1().size(), static_cast(ucell->nat)); + EXPECT_EQ(CE.get_dis_old2().size(), static_cast(ucell->nat)); + EXPECT_EQ(CE.get_dis_now().size(), static_cast(ucell->nat)); +} + +TEST_F(ChargeExtraTest, ExtrapolateChargeCase1) +{ + chg_extrap ="second-order"; + CE.Init_CE(nspin, ucell->nat, charge.rhopw->nrxx, chg_extrap); + CE.set_istep(0); + CE.set_pot_order(3); + + GlobalV::ofs_running.open("log"); + const module_charge::AtomicRhoCfg atomic_rho_cfg_test{ + PARAM.inp.nelec, + PARAM.inp.test_charge, + PARAM.globalv.domag, + PARAM.globalv.domag_z, + GlobalV::ofs_warning}; + CE.extrapolate_charge(pgrid, *ucell.get(), &charge, *charge.rhopw, &sf, GlobalV::ofs_running, GlobalV::ofs_warning, atomic_rho_cfg_test, false); + GlobalV::ofs_running.close(); + + // Check the results + std::ifstream ifs("log"); + std::string expected_output = " charge density from previous step !\n"; + std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + ifs.close(); + std::remove("log"); + + EXPECT_EQ(output, expected_output); + EXPECT_EQ(CE.get_rho_extr(), 0); +} + +TEST_F(ChargeExtraTest, ExtrapolateChargeCase2) +{ + chg_extrap ="second-order"; + CE.Init_CE(nspin, ucell->nat, charge.rhopw->nrxx, chg_extrap); + CE.set_istep(1); + CE.set_pot_order(3); + + GlobalV::ofs_running.open("log"); + const module_charge::AtomicRhoCfg atomic_rho_cfg_test{ + PARAM.inp.nelec, + PARAM.inp.test_charge, + PARAM.globalv.domag, + PARAM.globalv.domag_z, + GlobalV::ofs_warning}; + CE.extrapolate_charge(pgrid, *ucell.get(), &charge, *charge.rhopw, &sf, GlobalV::ofs_running, GlobalV::ofs_warning, atomic_rho_cfg_test, false); + GlobalV::ofs_running.close(); + + // Check the results + std::ifstream ifs("log"); + std::string expected_output = " NEW-OLD atomic charge density approx. for the potential !\n"; + std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + ifs.close(); + std::remove("log"); + + EXPECT_EQ(output, expected_output); + EXPECT_EQ(CE.get_rho_extr(), 1); +} + +TEST_F(ChargeExtraTest, ExtrapolateChargeCase3) +{ + chg_extrap ="second-order"; + CE.Init_CE(nspin, ucell->nat, charge.rhopw->nrxx, chg_extrap); + CE.set_istep(2); + CE.set_pot_order(3); + + GlobalV::ofs_running.open("log"); + const module_charge::AtomicRhoCfg atomic_rho_cfg_test{ + PARAM.inp.nelec, + PARAM.inp.test_charge, + PARAM.globalv.domag, + PARAM.globalv.domag_z, + GlobalV::ofs_warning}; + CE.extrapolate_charge(pgrid, *ucell.get(), &charge, *charge.rhopw, &sf, GlobalV::ofs_running, GlobalV::ofs_warning, atomic_rho_cfg_test, false); + GlobalV::ofs_running.close(); + + // Check the results + std::ifstream ifs("log"); + std::string expected_output = " first order charge density extrapolation !\n"; + std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + ifs.close(); + std::remove("log"); + + EXPECT_EQ(output, expected_output); + EXPECT_EQ(CE.get_rho_extr(), 2); +} + +TEST_F(ChargeExtraTest, ExtrapolateChargeCase4) +{ + chg_extrap ="second-order"; + CE.Init_CE(nspin, ucell->nat, charge.rhopw->nrxx, chg_extrap); + CE.set_istep(3); + + GlobalV::ofs_running.open("log"); + const module_charge::AtomicRhoCfg atomic_rho_cfg_test{ + PARAM.inp.nelec, + PARAM.inp.test_charge, + PARAM.globalv.domag, + PARAM.globalv.domag_z, + GlobalV::ofs_warning}; + CE.extrapolate_charge(pgrid, *ucell.get(), &charge, *charge.rhopw, &sf, GlobalV::ofs_running, GlobalV::ofs_warning, atomic_rho_cfg_test, false); + GlobalV::ofs_running.close(); + + // Check the results + std::ifstream ifs("log"); + std::string expected_output = " second order charge density extrapolation !\n alpha = 0\n beta = 0\n"; + std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + ifs.close(); + std::remove("log"); + + EXPECT_EQ(output, expected_output); + EXPECT_EQ(CE.get_rho_extr(), 3); + std::remove("./support/OLD2_SPIN1_CHG.cube"); +} + +TEST_F(ChargeExtraTest, UpdateAllDis) +{ + chg_extrap ="second-order"; + CE.Init_CE(nspin, ucell->nat, charge.rhopw->nrxx, chg_extrap); + CE.set_istep(3); + for (int i = 0; i < ucell->nat; ++i) + { + for (int j = 0; j < 3; ++j) + { + CE.get_dis_old1()[i][j] = i; + CE.get_dis_now()[i][j] = j; + } + } + + CE.update_all_dis(*ucell.get()); + + EXPECT_EQ(CE.get_istep(), 4); + EXPECT_DOUBLE_EQ(CE.get_dis_old2()[0][2], 0.0); + EXPECT_DOUBLE_EQ(CE.get_dis_old1()[0][2], 2.0); + EXPECT_DOUBLE_EQ(CE.get_dis_now()[0][2], 0.0); +} + +TEST_F(ChargeExtraTest, FindAlphaAndBeta) +{ + chg_extrap ="second-order"; + CE.Init_CE(nspin, ucell->nat, charge.rhopw->nrxx, chg_extrap); + CE.set_istep(3); + for (int i = 0; i < ucell->nat; ++i) + { + for (int j = 0; j < 3; ++j) + { + CE.get_dis_old1()[i][j] = i; + CE.get_dis_now()[i][j] = j; + } + } + + CE.find_alpha_and_beta_for_testing(ucell->nat, GlobalV::ofs_running, GlobalV::ofs_warning); + + EXPECT_DOUBLE_EQ(CE.get_alpha(), 1.0); + EXPECT_DOUBLE_EQ(CE.get_beta(), 0.0); +} diff --git a/source/source_estate/module_charge/unittests/test_chg_init.cpp b/source/source_estate/module_charge/unittests/test_chg_init.cpp new file mode 100644 index 00000000000..12cf839c46e --- /dev/null +++ b/source/source_estate/module_charge/unittests/test_chg_init.cpp @@ -0,0 +1,136 @@ +#include "gtest/gtest.h" + +#include "source_base/matrix3.h" +#include "source_basis/module_pw/pw_basis.h" +#include "source_cell/unitcell.h" +#include "source_estate/module_charge/charge.h" +#include "source_estate/module_charge/chg_init.h" +#include "source_io/module_restart/restart.h" + +#include +#include +#include + +// charge.cpp references Magnetism; provide a lightweight stub. +Magnetism::Magnetism() +{ + this->tot_mag = 0.0; + this->abs_mag = 0.0; +} +Magnetism::~Magnetism() +{ +} + +// chg_init.cpp references GlobalC::restart; provide a definition. +namespace GlobalC +{ +Restart restart; +} // namespace GlobalC + +/************************************************ + * unit test of module_charge/chg_init.cpp + ***********************************************/ + +/** + * - Tested Functions: + * - init_rho: the SCF charge-density initialization orchestrator. + * Covered: + * - init_chg == "wfc" with wfcpw == nullptr triggers WARNING_QUIT. + * - init_chg == "atomic" with ntype == 0 runs the atomic fallback + * (and Thomas-Fermi tau init when meta_gga is true) without crashing. + */ + +namespace +{ + +module_charge::InitRhoCfg make_init_cfg(const std::string& init_chg, bool meta_gga) +{ + return {init_chg, "", "scf", "", 1.0, 0, 0, false, false, meta_gga, 1}; +} + +} // namespace + +class ChgInitTest : public ::testing::Test +{ + protected: + ModulePW::PW_Basis pw_basis; + Charge charge; + UnitCell ucell; + Parallel_Grid pgrid; + + void SetUp() override + { + pw_basis.initgrids(4, ModuleBase::Matrix3(1, 0, 0, 0, 1, 0, 0, 0, 1), 20); + pw_basis.initparameters(false, 20); + pw_basis.setuptransform(); + pw_basis.collect_local_pw(); + charge.set_rhopw(&pw_basis); + ucell.ntype = 0; + ucell.nat = 0; + ucell.omega = pw_basis.omega; + } +}; + +TEST_F(ChgInitTest, InitChgWfcWithNullWfcpwAborts) +{ + charge.allocate(1, false, false, 0); + ModuleSymmetry::Symmetry symm; + ModuleBase::ComplexMatrix strucFac(0, pw_basis.npw); + module_charge::InitRhoCfg cfg = make_init_cfg("wfc", false); + + EXPECT_DEATH(module_charge::init_rho(charge, pw_basis, ucell, pgrid, strucFac, + symm, nullptr, nullptr, cfg), + ""); +} + +TEST_F(ChgInitTest, InitChgAtomicNtypeZeroMetaGgaRuns) +{ + const bool meta_gga = true; + charge.allocate(1, meta_gga, false, 0); + ModuleSymmetry::Symmetry symm; + ModuleBase::ComplexMatrix strucFac(0, pw_basis.npw); + module_charge::InitRhoCfg cfg = make_init_cfg("atomic", meta_gga); + + // ntype==0: atomic_rho loop is skipped; TF tau is computed from rho. + module_charge::init_rho(charge, pw_basis, ucell, pgrid, strucFac, + symm, nullptr, nullptr, cfg); + + // kin_r should be the Thomas-Fermi expression: fact * |rho|^(5/3). + const double fact = (3.0 / 5.0) * std::pow(3.0 * ModuleBase::PI * ModuleBase::PI, 2.0 / 3.0); + for (int ir = 0; ir < pw_basis.nrxx; ++ir) + { + const double expected = fact * std::pow(std::abs(charge.rho[0][ir]), 5.0 / 3.0); + EXPECT_NEAR(charge.kin_r[0][ir], expected, 1e-6); + } +} + +TEST_F(ChgInitTest, InitChgAutoMissingFileFallsBackToAtomic) +{ + charge.allocate(1, false, false, 0); + ModuleSymmetry::Symmetry symm; + ModuleBase::ComplexMatrix strucFac(0, pw_basis.npw); + module_charge::InitRhoCfg cfg = make_init_cfg("auto", false); + // Point the read-in directory at a path that holds no charge files. + cfg.global_readin_dir = "no_such_dir/"; + + // init_chg == "auto" must not abort when no density file exists; it + // returns having taken the atomic-density fallback (ntype==0 keeps it a + // no-op, but the run must survive). + module_charge::init_rho(charge, pw_basis, ucell, pgrid, strucFac, + symm, nullptr, nullptr, cfg); + SUCCEED(); +} + +TEST_F(ChgInitTest, InitChgFileMissingFileAborts) +{ + charge.allocate(1, false, false, 0); + ModuleSymmetry::Symmetry symm; + ModuleBase::ComplexMatrix strucFac(0, pw_basis.npw); + module_charge::InitRhoCfg cfg = make_init_cfg("file", false); + cfg.global_readin_dir = "no_such_dir/"; + + // init_chg == "file" must abort when no density source can be read. + EXPECT_DEATH(module_charge::init_rho(charge, pw_basis, ucell, pgrid, strucFac, + symm, nullptr, nullptr, cfg), + ""); +} diff --git a/source/source_estate/module_charge/unittests/test_chg_mix.cpp b/source/source_estate/module_charge/unittests/test_chg_mix.cpp new file mode 100644 index 00000000000..597c447ef95 --- /dev/null +++ b/source/source_estate/module_charge/unittests/test_chg_mix.cpp @@ -0,0 +1,1075 @@ +#include "gmock/gmock.h" +#include "gtest/gtest.h" +#include "../chg_mix.h" +#include "../chg_drho.h" +#include "../chg_drho_detail.h" +#include "../chg_precond.h" +#include "../chg_uspp.h" +#include "source_base/module_mixing/broyden_mixing.h" +#include "source_basis/module_pw/pw_basis_sup.h" +#include "source_hamilt/module_xc/xc_functional.h" + +#ifdef _OPENMP +#include +#endif + +int XC_Functional::func_type = 1; +bool XC_Functional::ked_flag = false; + +// mock function +Magnetism::~Magnetism() +{ +} +Magnetism::Magnetism() +{ +} +Charge::~Charge() +{ +} +Charge::Charge() +{ +} + +void Charge::set_rhopw(ModulePW::PW_Basis* rhopw_in) +{ + this->rhopw = rhopw_in; +} + +// mock class cell +/************************************************ + * unit test of chg_mix.cpp + ***********************************************/ + +/** + * - Tested Functions: + * - SetMixingTest: + * Charge_Mixing::set_mixing() + * Charge_Mixing::init_mixing() + * Charge_Mixing::get_mixing_mode() + * Charge_Mixing::get_mixing_beta() + * Charge_Mixing::get_mixing_ndim() + * Charge_Mixing::get_mixing_config() + * - set the basic parameters of class charge_mixing + * - KerkerScreenTest: module_charge::kerker_screen_recip(cfg, rhopw, tpiba, drhog) + * module_charge::kerker_screen_real(cfg, rhopw, tpiba, drhog) + * - screen drho with Kerker method + * - InnerDotTest: module_charge::inner_product_recip_hartree(rhog1, rhog2) + * module_charge::detail::inner_product_recip_rho(rhog1, rhog2) + * module_charge::inner_product_real(rho1, rho2) + * - calculate the inner product of two vectors + * - MixRhoTest: Charge_Mixing::mix_rho(chr) + * Charge_Mixing::mix_rho_recip(chr) + * Charge_Mixing::mix_rho_real(chr) + * - mix rho with different methods + * - CloseKerkerGg0DisablesScreenReal: Charge_Mixing::close_kerker_gg0() + * - regression test: close_kerker_gg0() must short-circuit the Kerker + * screening lambda in mix_rho_real so output matches cfg.mixing_gg0=0 + * - MixDivCombTest: module_charge::split_dgrid + * module_charge::merge_dgrid + * - divide and combine data on the USPP double grid + * + */ + +class ChargeMixingTest : public ::testing::Test +{ + public: + UnitCell ucell; + ChargeMixingTest() + { + // Init pw_basis + pw_basis.initgrids(4, ModuleBase::Matrix3(1, 0, 0, 0, 1, 0, 0, 0, 1), 20); + pw_basis.initparameters(false, 20); + pw_basis.setuptransform(); + pw_basis.collect_local_pw(); + pw_dbasis.initgrids(4, ModuleBase::Matrix3(1, 0, 0, 0, 1, 0, 0, 0, 1), 40); + pw_dbasis.initparameters(false, 40); + pw_dbasis.setuptransform(&pw_basis); + pw_dbasis.collect_local_pw(); + // default mixing parameters + cfg.mixing_mode = "broyden"; + cfg.mixing_beta = 0.8; + cfg.mixing_ndim = 8; + cfg.mixing_gg0 = 1.0; + cfg.mixing_tau = false; + cfg.mixing_beta_mag = 1.6; + cfg.mixing_gg0_mag = 0.0; + cfg.mixing_gg0_min = 0.1; + cfg.mixing_angle = -10.0; + cfg.mixing_dmr = false; + // the remaining fields are runtime globals the tests steer per case; + // seed them with the Input_para / System_para defaults + cfg.nspin = 1; + cfg.scf_thr_type = -1; + cfg.double_grid = false; + cfg.gamma_only_pw = false; + cfg.domag = false; + cfg.domag_z = false; + cfg.scf_nmax = 100; + ucell.omega = 1.0; + ucell.tpiba = 1.0; + } + ModulePW::PW_Basis pw_basis; + ModulePW::PW_Basis_Sup pw_dbasis; + Charge charge; + + /// The mixing configuration under test. The production mixing code takes + /// this snapshot explicitly rather than reading any global, so the fixture + /// owns it outright instead of staging values in the parameter singleton; + /// each case amends the fields it cares about and hands it to set_mixing(). + MixingConfig cfg; + + /// Backing storage for the Charge views under test. charge.cpp is not one + /// of this target's sources, so Charge::allocate() is unavailable here and + /// the fixture supplies the buffers itself, pointing the public + /// rho/rhog/kin_r views at them with the same is*nrxx (resp. is*npw) + /// stride allocate() would lay out. Owning them as vectors also means the + /// fixture cleans up after itself. + std::vector buf_rho, buf_rho_save, buf_kin_r, buf_kin_r_save; + std::vector> buf_rhog, buf_rhog_save; + std::vector ptr_rho, ptr_rho_save, ptr_kin_r, ptr_kin_r_save; + std::vector*> ptr_rhog, ptr_rhog_save; + + void wire_charge(const int nspin, const int nrxx, const int npw, const bool kin_den) + { + buf_rho.assign(nspin * nrxx, 0.0); + buf_rho_save.assign(nspin * nrxx, 0.0); + buf_rhog.assign(nspin * npw, std::complex(0.0, 0.0)); + buf_rhog_save.assign(nspin * npw, std::complex(0.0, 0.0)); + ptr_rho.resize(nspin); + ptr_rho_save.resize(nspin); + ptr_rhog.resize(nspin); + ptr_rhog_save.resize(nspin); + for (int is = 0; is < nspin; is++) + { + ptr_rho[is] = buf_rho.data() + is * nrxx; + ptr_rho_save[is] = buf_rho_save.data() + is * nrxx; + ptr_rhog[is] = buf_rhog.data() + is * npw; + ptr_rhog_save[is] = buf_rhog_save.data() + is * npw; + } + charge.rho = ptr_rho.data(); + charge.rho_save = ptr_rho_save.data(); + charge.rhog = ptr_rhog.data(); + charge.rhog_save = ptr_rhog_save.data(); + if (kin_den) + { + buf_kin_r.assign(nspin * nrxx, 0.0); + buf_kin_r_save.assign(nspin * nrxx, 0.0); + ptr_kin_r.resize(nspin); + ptr_kin_r_save.resize(nspin); + for (int is = 0; is < nspin; is++) + { + ptr_kin_r[is] = buf_kin_r.data() + is * nrxx; + ptr_kin_r_save[is] = buf_kin_r_save.data() + is * nrxx; + } + charge.kin_r = ptr_kin_r.data(); + charge.kin_r_save = ptr_kin_r_save.data(); + } + } + + // mixing_tau is only honoured when the functional actually carries a + // kinetic energy density, which set_mixing's callers fold in. Cases flip + // XC_Functional::set_ked_flag() around, so apply it at hand-off time. + MixingConfig make_cfg() + { + MixingConfig out = cfg; + out.mixing_tau = cfg.mixing_tau && XC_Functional::get_ked_flag(); + return out; + } + + // Re-sync the runtime globals (nspin/scf_thr_type/gamma_only_pw/domag/domag_z) + // into an already-configured Charge_Mixing. Cases mutate them after + // set_mixing to steer the residual/inner-product branches, and the object + // reads them from its own snapshot, so the test must push the new values in + // without disturbing the mixing history that init_mixing() built. + void sync_cfg(Charge_Mixing& cm) + { + MixingConfig cur = cm.get_mixing_config(); + cur.nspin = cfg.nspin; + cur.scf_thr_type = cfg.scf_thr_type; + cur.gamma_only_pw = cfg.gamma_only_pw; + cur.domag = cfg.domag; + cur.domag_z = cfg.domag_z; + cm.set_mixing_config(cur); + } +}; + +TEST_F(ChargeMixingTest, SetMixingTest) +{ +#ifdef _OPENMP + omp_set_num_threads(1); +#endif + cfg.nspin = 1; + Charge_Mixing CMtest; + cfg.mixing_beta = 1.0; + cfg.mixing_ndim = 1; + cfg.mixing_gg0 = 1.0; + + CMtest.set_mixing(make_cfg(), &pw_basis, &pw_basis, ucell.omega, ucell.tpiba); + EXPECT_EQ(CMtest.get_mixing_mode(), "broyden"); + EXPECT_EQ(CMtest.get_mixing_beta(), 1.0); + EXPECT_EQ(CMtest.get_mixing_ndim(), 1); + EXPECT_EQ(CMtest.get_mixing_config().mixing_gg0, 1.0); + EXPECT_EQ(CMtest.get_mixing_config().mixing_tau, false); + EXPECT_EQ(CMtest.get_mixing_config().mixing_beta_mag, 1.6); + EXPECT_EQ(CMtest.get_mixing_config().mixing_gg0_mag, 0.0); + EXPECT_EQ(CMtest.get_mixing_config().mixing_gg0_min, 0.1); + EXPECT_EQ(CMtest.get_mixing_config().mixing_angle, -10.0); + EXPECT_EQ(CMtest.get_mixing_config().mixing_dmr, false); + + cfg.mixing_tau = true; + XC_Functional::set_ked_flag(true); + cfg.mixing_mode = "plain"; + CMtest.set_mixing(make_cfg(), &pw_basis, &pw_basis, ucell.omega, ucell.tpiba); + EXPECT_EQ(CMtest.get_mixing_mode(), "plain"); + EXPECT_EQ(CMtest.get_mixing_config().mixing_tau, true); + XC_Functional::set_ked_flag(false); + + cfg.mixing_beta = 1.1; + std::string output; + testing::internal::CaptureStdout(); + EXPECT_EXIT(CMtest.set_mixing(make_cfg(), &pw_basis, &pw_basis, ucell.omega, ucell.tpiba);, ::testing::ExitedWithCode(1), ""); + output = testing::internal::GetCapturedStdout(); + EXPECT_THAT(output, testing::HasSubstr("You'd better set mixing_beta to [0.0, 1.0]!")); + + cfg.mixing_beta = 0.7; + cfg.mixing_beta_mag = -0.1; + cfg.nspin = 2; + testing::internal::CaptureStdout(); + EXPECT_EXIT(CMtest.set_mixing(make_cfg(), &pw_basis, &pw_basis, ucell.omega, ucell.tpiba);, ::testing::ExitedWithCode(1), ""); + output = testing::internal::GetCapturedStdout(); + EXPECT_THAT(output, testing::HasSubstr("You'd better set mixing_beta_mag >= 0.0!")); + + cfg.nspin = 1; + cfg.mixing_beta = 0.7; + cfg.mixing_beta_mag = 1.6; + cfg.mixing_mode = "nothing"; + testing::internal::CaptureStdout(); + EXPECT_EXIT(CMtest.set_mixing(make_cfg(), &pw_basis, &pw_basis, ucell.omega, ucell.tpiba);, ::testing::ExitedWithCode(1), ""); + output = testing::internal::GetCapturedStdout(); + EXPECT_THAT(output, testing::HasSubstr("This Mixing mode is not implemended yet,coming soon.")); +} + +TEST_F(ChargeMixingTest, InitMixingTest) +{ +#ifdef _OPENMP + omp_set_num_threads(1); +#endif + cfg.nspin = 1; + XC_Functional::set_func_type(1); + XC_Functional::set_ked_flag(false); + Charge_Mixing CMtest; + + CMtest.set_mixing(make_cfg(), &pw_basis, &pw_basis, ucell.omega, ucell.tpiba); + + cfg.scf_thr_type= 1; + sync_cfg(CMtest); + CMtest.init_mixing(); + EXPECT_EQ(CMtest.get_rho_mdata().length, pw_basis.npw); + + cfg.scf_thr_type= 2; + sync_cfg(CMtest); + CMtest.init_mixing(); + EXPECT_EQ(CMtest.get_rho_mdata().length, pw_basis.nrxx); + + cfg.nspin = 4; + sync_cfg(CMtest); + CMtest.init_mixing(); + EXPECT_EQ(CMtest.get_rho_mdata().length, 4 * pw_basis.nrxx); + + cfg.nspin = 1; + cfg.mixing_tau = true; + XC_Functional::set_func_type(3); + XC_Functional::set_ked_flag(true); + CMtest.set_mixing(make_cfg(), &pw_basis, &pw_basis, ucell.omega, ucell.tpiba); + CMtest.init_mixing(); + EXPECT_EQ(CMtest.get_tau_mdata().length, pw_basis.nrxx); + + cfg.nspin = 4; + cfg.mixing_angle = 1.0; + CMtest.set_mixing(make_cfg(), &pw_basis, &pw_basis, ucell.omega, ucell.tpiba); + CMtest.init_mixing(); + EXPECT_EQ(CMtest.get_rho_mdata().length, 2 * pw_basis.nrxx); +} + +TEST_F(ChargeMixingTest, InnerDotRealTest) +{ + Charge_Mixing CMtest; + // non mixing angle case + CMtest.set_mixing(make_cfg(), &pw_basis, &pw_basis, ucell.omega, ucell.tpiba); + cfg.nspin = 4; + sync_cfg(CMtest); + + // a simple sum for inner product + std::vector drho1(pw_basis.nrxx * cfg.nspin); + std::vector drho2(pw_basis.nrxx * cfg.nspin); + for (int i = 0; i < pw_basis.nrxx * cfg.nspin; ++i) + { + drho1[i] = 1.0; + drho2[i] = double(i); + } + double inner = module_charge::inner_product_real(drho1.data(), drho2.data(), pw_basis, CMtest.get_mixing_config()); + EXPECT_NEAR(inner, 0.5 * pw_basis.nrxx * cfg.nspin * (pw_basis.nrxx * cfg.nspin - 1), 1e-8); + + // mixing angle case + cfg.mixing_angle = 1.0; + CMtest.set_mixing(make_cfg(), &pw_basis, &pw_basis, ucell.omega, ucell.tpiba); + cfg.nspin = 4; + + // a simple sum for inner product + drho1.resize(pw_basis.nrxx * 2); + drho2.resize(pw_basis.nrxx * 2); + for (int i = 0; i < pw_basis.nrxx * 2; ++i) + { + drho1[i] = 1.0; + drho2[i] = double(i); + } + inner = module_charge::inner_product_real(drho1.data(), drho2.data(), pw_basis, CMtest.get_mixing_config()); + EXPECT_NEAR(inner, 0.5 * pw_basis.nrxx * 2 * (pw_basis.nrxx * 2 - 1), 1e-8); +} + +TEST_F(ChargeMixingTest, InnerDotRecipHartreeTest) +{ + // REAL + Charge_Mixing CMtest; + const int npw = pw_basis.npw; + const int nrxx = pw_basis.nrxx; + cfg.nspin = 1; + std::vector drhor1(pw_basis.nrxx); + std::vector drhor2(pw_basis.nrxx); + for (int i = 0; i < pw_basis.nrxx; ++i) + { + drhor1[i] = 1.0; + drhor2[i] = double(i); + } + // Populate cfg_ before the first inner_product call: the function reads + // nspin from cfg_, which is default-constructed (and thus invalid) until + // set_mixing runs. + CMtest.set_mixing(make_cfg(), &pw_basis, &pw_basis, ucell.omega, ucell.tpiba); + double inner = module_charge::inner_product_real(drhor1.data(), drhor2.data(), pw_basis, CMtest.get_mixing_config()); + EXPECT_NEAR(inner, 0.5 * pw_basis.nrxx * (pw_basis.nrxx - 1), 1e-8); + + // RECIPROCAL NSPIN=1 + ucell.tpiba2 = 1.0; + ucell.omega = 2.0; + CMtest.set_mixing(make_cfg(), &pw_basis, &pw_basis, ucell.omega, ucell.tpiba); + cfg.nspin = 1; + sync_cfg(CMtest); + std::vector> drhog1(pw_basis.npw); + std::vector> drhog2(pw_basis.npw); + for (int i = 0; i < pw_basis.nrxx; ++i) + { + drhor1[i] = 0.0; + } + drhor1[2] = 1.0; + pw_basis.real2recip(drhor1.data(), drhog1.data()); + pw_basis.real2recip(drhor2.data(), drhog2.data()); + + inner = module_charge::inner_product_recip_hartree(drhog1.data(), drhog2.data(), pw_basis, CMtest.get_mixing_config(), ucell.omega, ucell.tpiba); + EXPECT_NEAR(inner, -0.3 * ModuleBase::e2 * ModuleBase::FOUR_PI, 1e-8); + + // RECIPROCAL NSPIN=2 + cfg.nspin = 2; + sync_cfg(CMtest); + drhog1.resize(pw_basis.npw * cfg.nspin); + drhog2.resize(pw_basis.npw * cfg.nspin); + std::vector> drhog1_mag(pw_basis.npw * cfg.nspin); + std::vector> drhog2_mag(pw_basis.npw * cfg.nspin); + for (int i = 0; i < pw_basis.npw * cfg.nspin; ++i) + { + drhog1[i] = std::complex(1.0, double(i)); + drhog2[i] = std::complex(1.0, 1.0); + } + // set mag + for (int i = 0; i < pw_basis.npw; ++i) + { + drhog1_mag[i] = drhog1[i] + drhog1[i+pw_basis.npw]; + drhog1_mag[i+pw_basis.npw] = drhog1[i] - drhog1[i+pw_basis.npw]; + drhog2_mag[i] = drhog2[i] + drhog2[i+pw_basis.npw]; + drhog2_mag[i+pw_basis.npw] = drhog2[i] - drhog2[i+pw_basis.npw]; + } + cfg.gamma_only_pw= false; + sync_cfg(CMtest); + inner = module_charge::inner_product_recip_hartree(drhog1_mag.data(), drhog2_mag.data(), pw_basis, CMtest.get_mixing_config(), ucell.omega, ucell.tpiba); + EXPECT_NEAR(inner, 236763.82650318215, 1e-8); + cfg.gamma_only_pw= true; + sync_cfg(CMtest); + inner = module_charge::inner_product_recip_hartree(drhog1_mag.data(), drhog2_mag.data(), pw_basis, CMtest.get_mixing_config(), ucell.omega, ucell.tpiba); + EXPECT_NEAR(inner, 236763.82650318215 * 2, 1e-8); + + // RECIPROCAL NSPIN=4 without mixing_angle + cfg.nspin = 4; + sync_cfg(CMtest); + drhog1.resize(pw_basis.npw * cfg.nspin); + drhog2.resize(pw_basis.npw * cfg.nspin); + for (int i = 0; i < pw_basis.npw * cfg.nspin; ++i) + { + drhog1[i] = std::complex(1.0, double(i)); + drhog2[i] = std::complex(1.0, 1.0); + } + + cfg.domag = false; + cfg.domag_z = false; + sync_cfg(CMtest); + inner = module_charge::inner_product_recip_hartree(drhog1.data(), drhog2.data(), pw_basis, CMtest.get_mixing_config(), ucell.omega, ucell.tpiba); + EXPECT_NEAR(inner, 28260.091995611871, 1e-8); + cfg.gamma_only_pw= true; + cfg.domag = true; + cfg.domag_z = true; + sync_cfg(CMtest); + inner = module_charge::inner_product_recip_hartree(drhog1.data(), drhog2.data(), pw_basis, CMtest.get_mixing_config(), ucell.omega, ucell.tpiba); + EXPECT_NEAR(inner, 110668.61166927818, 1e-8); + + // RECIPROCAL NSPIN=4 with mixing_angle + cfg.nspin = 4; + cfg.mixing_angle = 1.0; + CMtest.set_mixing(make_cfg(), &pw_basis, &pw_basis, ucell.omega, ucell.tpiba); + drhog1.resize(pw_basis.npw * 2); + drhog2.resize(pw_basis.npw * 2); + for (int i = 0; i < pw_basis.npw * 2; ++i) + { + drhog1[i] = std::complex(1.0, double(i)); + drhog2[i] = std::complex(1.0, 1.0); + } + cfg.gamma_only_pw= false; + sync_cfg(CMtest); + inner = module_charge::inner_product_recip_hartree(drhog1.data(), drhog2.data(), pw_basis, CMtest.get_mixing_config(), ucell.omega, ucell.tpiba); + EXPECT_NEAR(inner, 36548.881431837777, 1e-8); + cfg.gamma_only_pw= true; + sync_cfg(CMtest); + inner = module_charge::inner_product_recip_hartree(drhog1.data(), drhog2.data(), pw_basis, CMtest.get_mixing_config(), ucell.omega, ucell.tpiba); + EXPECT_NEAR(inner, 44776.555369916401, 1e-8); +} + +TEST_F(ChargeMixingTest, InnerDotRecipRhoTest) +{ + // REAL + Charge_Mixing CMtest; + cfg.nspin = 1; + std::vector drhor1(pw_basis.nrxx); + std::vector drhor2(pw_basis.nrxx); + for (int i = 0; i < pw_basis.nrxx; ++i) + { + drhor1[i] = 1.0; + drhor2[i] = double(i); + } + // Populate cfg_ before the first inner_product call (see the hartree test). + CMtest.set_mixing(make_cfg(), &pw_basis, &pw_basis, ucell.omega, ucell.tpiba); + double inner = module_charge::inner_product_real(drhor1.data(), drhor2.data(), pw_basis, CMtest.get_mixing_config()); + EXPECT_NEAR(inner, 0.5 * pw_basis.nrxx * (pw_basis.nrxx - 1), 1e-8); + + // RECIPROCAL + ucell.tpiba2 = 1.0; + ucell.omega = 2.0; + CMtest.set_mixing(make_cfg(), &pw_basis, &pw_basis, ucell.omega, ucell.tpiba); + cfg.nspin = 1; + sync_cfg(CMtest); + std::vector> drhog1(pw_basis.npw); + std::vector> drhog2(pw_basis.npw); + for (int i = 0; i < pw_basis.nrxx; ++i) + { + drhor1[i] = 0.0; + } + drhor1[2] = 1.0; + pw_basis.real2recip(drhor1.data(), drhog1.data()); + pw_basis.real2recip(drhor2.data(), drhog2.data()); + + inner = module_charge::detail::inner_product_recip_rho(drhog1.data(), drhog2.data(), pw_basis, CMtest.get_mixing_config(), ucell.omega, ucell.tpiba); + EXPECT_NEAR(inner, -0.3 * ModuleBase::e2 * ModuleBase::FOUR_PI, 1e-8); + + cfg.nspin = 2; + sync_cfg(CMtest); + drhog1.resize(pw_basis.npw * cfg.nspin); + drhog2.resize(pw_basis.npw * cfg.nspin); + for (int i = 0; i < pw_basis.npw * cfg.nspin; ++i) + { + drhog1[i] = std::complex(1.0, double(i)); + drhog2[i] = std::complex(1.0, 1.0); + } + cfg.gamma_only_pw= false; + sync_cfg(CMtest); + inner = module_charge::detail::inner_product_recip_rho(drhog1.data(), drhog2.data(), pw_basis, CMtest.get_mixing_config(), ucell.omega, ucell.tpiba); + EXPECT_NEAR(inner, 236763.82650318215, 1e-8); + cfg.gamma_only_pw= true; + sync_cfg(CMtest); + inner = module_charge::detail::inner_product_recip_rho(drhog1.data(), drhog2.data(), pw_basis, CMtest.get_mixing_config(), ucell.omega, ucell.tpiba); + EXPECT_NEAR(inner, 236763.82650318215 * 2, 1e-8); + + cfg.nspin = 4; + sync_cfg(CMtest); + drhog1.resize(pw_basis.npw * cfg.nspin); + drhog2.resize(pw_basis.npw * cfg.nspin); + for (int i = 0; i < pw_basis.npw * cfg.nspin; ++i) + { + drhog1[i] = std::complex(1.0, double(i)); + drhog2[i] = std::complex(1.0, 1.0); + } + + cfg.domag = false; + cfg.domag_z = false; + sync_cfg(CMtest); + inner = module_charge::detail::inner_product_recip_rho(drhog1.data(), drhog2.data(), pw_basis, CMtest.get_mixing_config(), ucell.omega, ucell.tpiba); + EXPECT_NEAR(inner, 28260.091995611871, 1e-8); + cfg.gamma_only_pw= true; + cfg.domag = true; + cfg.domag_z = true; + sync_cfg(CMtest); + inner = module_charge::detail::inner_product_recip_rho(drhog1.data(), drhog2.data(), pw_basis, CMtest.get_mixing_config(), ucell.omega, ucell.tpiba); + EXPECT_NEAR(inner, 110668.61166927818, 1e-8); +} + +TEST_F(ChargeMixingTest, KerkerScreenRecipTest) +{ + ucell.tpiba = 1.0; + // nspin = 1 + cfg.nspin = 1; + // kcfg is the snapshot handed to the kernels; cfg (the fixture member) is + // the source make_cfg() reads, so nspin has to be set on cfg, not on kcfg. + MixingConfig kcfg = make_cfg(); + std::complex* drhog = new std::complex[cfg.nspin*pw_basis.npw]; + std::complex* drhog_old = new std::complex[cfg.nspin*pw_basis.npw]; + for (int i = 0; i < cfg.nspin*pw_basis.npw; ++i) + { + drhog_old[i] = drhog[i] = std::complex(1.0, 1.0); + } + // no kerker + kcfg.mixing_gg0 = 0.0; + module_charge::kerker_screen_recip(kcfg, &pw_basis, ucell.tpiba, drhog); + for (int i = 0; i < cfg.nspin*pw_basis.npw; ++i) + { + EXPECT_EQ(drhog[i], drhog_old[i]); + } + // kerker + kcfg.mixing_gg0 = 1.0; + module_charge::kerker_screen_recip(kcfg, &pw_basis, ucell.tpiba, drhog); + double gg0 = std::pow(ModuleBase::BOHR_TO_A, 2); + for (int i = 0; i < pw_basis.npw; ++i) + { + double gg = this->pw_basis.gg[i]; + double ref = std::max(gg / (gg + gg0), 0.1 / kcfg.mixing_beta); + EXPECT_NEAR(drhog[i].real(), ref, 1e-10); + EXPECT_NEAR(drhog[i].imag(), ref, 1e-10); + } + delete[] drhog; + delete[] drhog_old; + + // nspin = 2 + cfg.nspin = 2; + kcfg = make_cfg(); + kcfg.mixing_beta = 0.4; + kcfg.mixing_beta_mag = 1.6; + drhog = new std::complex[cfg.nspin*pw_basis.npw]; + drhog_old = new std::complex[cfg.nspin*pw_basis.npw]; + for (int i = 0; i < cfg.nspin*pw_basis.npw; ++i) + { + drhog_old[i] = drhog[i] = std::complex(1.0, 1.0); + } + // mixing_gg0 = 0.0 + kcfg.mixing_gg0 = 0.0; + module_charge::kerker_screen_recip(kcfg, &pw_basis, ucell.tpiba, drhog); + for (int i = 0; i < cfg.nspin*pw_basis.npw; ++i) + { + EXPECT_EQ(drhog[i], drhog_old[i]); + } + // mixing_gg0 = 1.0, mixing_gg0_mag = 0.0 + kcfg.mixing_gg0 = 1.0; + module_charge::kerker_screen_recip(kcfg, &pw_basis, ucell.tpiba, drhog); + gg0 = std::pow(ModuleBase::BOHR_TO_A, 2); + for (int i = 0; i < pw_basis.npw; ++i) + { + double gg = this->pw_basis.gg[i]; + double ref = std::max(gg / (gg + gg0), 0.1 / kcfg.mixing_beta); + // rho + EXPECT_NEAR(drhog[i].real(), ref, 1e-10); + EXPECT_NEAR(drhog[i].imag(), ref, 1e-10); + // mag + EXPECT_NEAR(drhog[i+pw_basis.npw].real(), 1.0, 1e-10); + EXPECT_NEAR(drhog[i+pw_basis.npw].imag(), 1.0, 1e-10); + } + delete[] drhog; + delete[] drhog_old; + + // nspin = 4 + cfg.nspin = 4; + kcfg = make_cfg(); + drhog = new std::complex[cfg.nspin*pw_basis.npw]; + drhog_old = new std::complex[cfg.nspin*pw_basis.npw]; + for (int i = 0; i < cfg.nspin*pw_basis.npw; ++i) + { + drhog_old[i] = drhog[i] = std::complex(1.0, 1.0); + } + // mixing_gg0 = 0.0 + kcfg.mixing_gg0 = 0.0; + module_charge::kerker_screen_recip(kcfg, &pw_basis, ucell.tpiba, drhog); + for (int i = 0; i < cfg.nspin*pw_basis.npw; ++i) + { + EXPECT_EQ(drhog[i], drhog_old[i]); + } + // mixing_gg0 = 1.0, mixing_gg0_mag = 0.0 + kcfg.mixing_gg0 = 1.0; + module_charge::kerker_screen_recip(kcfg, &pw_basis, ucell.tpiba, drhog); + gg0 = std::pow(ModuleBase::BOHR_TO_A, 2); + for (int i = 0; i < pw_basis.npw; ++i) + { + double gg = this->pw_basis.gg[i]; + double ref = std::max(gg / (gg + gg0), 0.1 / kcfg.mixing_beta); + // rho + EXPECT_NEAR(drhog[i].real(), ref, 1e-10); + EXPECT_NEAR(drhog[i].imag(), ref, 1e-10); + } + for (int i = 0; i < 3*pw_basis.npw; ++i) + { + EXPECT_NEAR(drhog[i + pw_basis.npw].real(), 1.0, 1e-10); + EXPECT_NEAR(drhog[i + pw_basis.npw].imag(), 1.0, 1e-10); + } + // mixing_gg0 = 1.0, mixing_gg0_mag = 2.0 + kcfg.mixing_gg0 = 1.0; + kcfg.mixing_gg0_mag = 2.0; + module_charge::kerker_screen_recip(kcfg, &pw_basis, ucell.tpiba, drhog); + double gg1 = std::pow(1.0 * ModuleBase::BOHR_TO_A, 2); + double gg2 = std::pow(2.0 * ModuleBase::BOHR_TO_A, 2); + for (int i = 0; i < pw_basis.npw; ++i) + { + double gg = this->pw_basis.gg[i]; + double ref = std::max(gg / (gg + gg1), 0.1 / kcfg.mixing_beta); + // rho + EXPECT_NEAR(drhog[i].real(), ref * ref, 1e-10); + EXPECT_NEAR(drhog[i].imag(), ref * ref, 1e-10); + } + for (int i = 0; i < pw_basis.npw; ++i) + { + double gg = this->pw_basis.gg[i]; + double ref = std::max(gg / (gg + gg2), 0.1 / kcfg.mixing_beta_mag); + // rho + for (int j = 1; j < cfg.nspin; ++j) + { + EXPECT_NEAR(drhog[i + pw_basis.npw * j].real(), ref, 1e-10); + EXPECT_NEAR(drhog[i + pw_basis.npw * j].imag(), ref, 1e-10); + } + } + delete[] drhog; + delete[] drhog_old; +} + +TEST_F(ChargeMixingTest, KerkerScreenRealTest) +{ + ucell.tpiba = 1.0; + // nspin = 1 + cfg.nspin = 1; + // kcfg is the snapshot handed to the kernels; nspin is set on the fixture + // member cfg, which is what make_cfg() reads. + MixingConfig kcfg = make_cfg(); + double* drhor = new double[cfg.nspin*pw_basis.nrxx]; + double* drhor_ref = new double[cfg.nspin*pw_basis.nrxx]; + for (int i = 0; i < cfg.nspin*pw_basis.nrxx; ++i) + { + drhor_ref[i] = drhor[i] = 1.0; + } + // no kerker + kcfg.mixing_gg0 = 0.0; + module_charge::kerker_screen_real(kcfg, &pw_basis, ucell.tpiba, drhor); + for (int i = 0; i < cfg.nspin*pw_basis.nrxx; ++i) + { + EXPECT_EQ(drhor[i], drhor_ref[i]); + } + delete[] drhor; + delete[] drhor_ref; + + // nspin = 2 + cfg.nspin = 2; + kcfg = make_cfg(); + kcfg.mixing_gg0 = 0.0; + std::complex* drhog = new std::complex[cfg.nspin*pw_basis.npw]; + std::complex* drhog_old = new std::complex[cfg.nspin*pw_basis.npw]; + drhor = new double[cfg.nspin*pw_basis.nrxx]; + drhor_ref = new double[cfg.nspin*pw_basis.nrxx]; + for (int i = 0; i < cfg.nspin*pw_basis.npw; ++i) + { + drhog_old[i] = drhog[i] = std::complex(1.0, 1.0); + } + module_charge::kerker_screen_recip(kcfg, &pw_basis, ucell.tpiba, drhog); // no kerker + for (int i = 0; i < cfg.nspin*pw_basis.npw; ++i) + { + EXPECT_EQ(drhog[i], drhog_old[i]); + } + + // RECIPROCAL + kcfg.mixing_gg0 = 1.0; + kcfg.mixing_gg0_mag = 0.0; + module_charge::kerker_screen_recip(kcfg, &pw_basis, ucell.tpiba, drhog); + const double gg0 = std::pow(ModuleBase::BOHR_TO_A, 2); + for (int i = 0; i < pw_basis.npw; ++i) + { + std::complex ration = drhog[i] / drhog[i+pw_basis.npw]; + double gg = this->pw_basis.gg[i]; + double ration_ref = std::max(gg / (gg + gg0), 0.1 / kcfg.mixing_beta); + EXPECT_NEAR(ration.real(), ration_ref, 1e-10); + EXPECT_NEAR(ration.imag(), 0, 1e-10); + } + + // REAL + pw_basis.recip2real(drhog, drhor_ref); + pw_basis.recip2real(drhog_old, drhor); + + kcfg.mixing_gg0 = 0.0; + kcfg.mixing_gg0_mag = 0.0; + // nothing happens + module_charge::kerker_screen_real(kcfg, &pw_basis, ucell.tpiba, drhor); + + kcfg.mixing_gg0 = 1.0; + module_charge::kerker_screen_real(kcfg, &pw_basis, ucell.tpiba, drhor); + for (int i = 0; i < pw_basis.nrxx; ++i) + { + EXPECT_NEAR(drhor[i], drhor_ref[i], 1e-8); + } + + delete[] drhog; + delete[] drhog_old; + delete[] drhor; + delete[] drhor_ref; + +} + +TEST_F(ChargeMixingTest, MixRhoTest) +{ + cfg.double_grid = false; + charge.set_rhopw(&pw_basis); + const int nspin = cfg.nspin = 1; + cfg.domag_z = false; + XC_Functional::set_func_type(3); + XC_Functional::set_ked_flag(true); + cfg.mixing_beta = 0.7; + cfg.mixing_ndim = 1; + cfg.mixing_gg0 = 0.0; + cfg.mixing_tau = true; + cfg.mixing_mode = "plain"; + const int nrxx = pw_basis.nrxx; + const int npw = pw_basis.npw; + wire_charge(nspin, nrxx, npw, XC_Functional::get_ked_flag()); + std::vector real_ref(nspin * nrxx); + std::vector real_save_ref(nspin * nrxx); + std::vector> recip_ref(nspin * npw); + std::vector> recip_save_ref(nspin * npw); + for(int i = 0 ; i < nspin * npw; ++i) + { + recip_ref[i] = std::complex(double(i), 1.0); + recip_save_ref[i] = std::complex(double(i), 0.0); + } + for(int i = 0 ; i < nspin ; ++i) + { + pw_basis.recip2real(recip_ref.data() + i * npw, real_ref.data() + i * nrxx); + pw_basis.recip2real(recip_save_ref.data() + i * npw, real_save_ref.data() + i * nrxx); + } + //--------------------------------MAIN BODY-------------------------------- + // RECIPROCAL + Charge_Mixing CMtest_recip; + cfg.scf_thr_type= 1; + CMtest_recip.set_mixing(make_cfg(), &pw_basis, &pw_basis, ucell.omega, ucell.tpiba); + CMtest_recip.init_mixing(); + for(int i = 0 ; i < nspin * npw; ++i) + { + charge.rhog[i / npw][i % npw] = recip_ref[i]; + charge.rhog_save[i / npw][i % npw] = recip_save_ref[i]; + } + for(int i = 0 ; i < nspin * nrxx; ++i) + { + charge.rho[i / nrxx][i % nrxx] = real_ref[i]; + charge.rho_save[i / nrxx][i % nrxx] = real_save_ref[i]; + } + CMtest_recip.mix_rho(&charge); + for(int is = 0 ; is < nspin; ++is) + { + for(int ir = 0 ; ir < nrxx ; ++ir) + { + EXPECT_NEAR(charge.rho_save[is][ir], real_ref[is*nrxx + ir], 1e-8); + } + for(int ig = 0; ig < npw ; ++ig) + { + EXPECT_NEAR(charge.rhog[is][ig].real(), recip_save_ref[is*npw + ig].real(), 1e-8); + EXPECT_NEAR(charge.rhog[is][ig].imag(), recip_save_ref[is*npw + ig].imag() + 0.7, 1e-8); + } + } + + // REAL + Charge_Mixing CMtest_real; + cfg.scf_thr_type= 2; + CMtest_real.set_mixing(make_cfg(), &pw_basis, &pw_basis, ucell.omega, ucell.tpiba); + CMtest_real.init_mixing(); + for(int i = 0 ; i < nspin * nrxx; ++i) + { + charge.rho[i / nrxx][i % nrxx] = real_ref[i]; + charge.rho_save[i / nrxx][i % nrxx] = real_save_ref[i]; + } + CMtest_recip.mix_rho(&charge); + for(int is = 0 ; is < nspin; ++is) + { + for(int ir = 0 ; ir < nrxx ; ++ir) + { + EXPECT_NEAR(charge.rho_save[is][ir], real_ref[is*nrxx + ir], 1e-8); + EXPECT_NEAR(charge.rho[is][ir], 0.3*real_save_ref[is*nrxx+ir] + 0.7*real_ref[is*nrxx+ir], 1e-8); + } + } + + //------------------------------------------------------------------------- + // the fixture owns the buffers behind charge.rho/rhog/kin_r, so there is + // nothing to delete here +} + +// Regression test: close_kerker_gg0() must short-circuit the Kerker screening +// lambda in mix_rho_real. Before the chg_precond refactor (commit 6d127d517) +// the kernels read this->mixing_gg0; after, they read cfg_ which is an +// immutable INPUT snapshot, so writing the dead member was a no-op and the +// non-separate-loop EXX path silently failed to disable Kerker. This test +// pins the fix: output after close_kerker_gg0() must match the cfg.mixing_gg0 +// = 0 baseline. +TEST_F(ChargeMixingTest, CloseKerkerGg0DisablesScreenReal) +{ + cfg.double_grid = false; + charge.set_rhopw(&pw_basis); + const int nspin = cfg.nspin = 1; + cfg.domag_z = false; + XC_Functional::set_func_type(3); + XC_Functional::set_ked_flag(false); + cfg.mixing_beta = 0.7; + cfg.mixing_ndim = 1; + cfg.mixing_gg0 = 1.0; // Kerker active by default + cfg.mixing_tau = false; + cfg.mixing_mode = "plain"; + cfg.scf_thr_type = 2; // real-space path + + const int nrxx = pw_basis.nrxx; + // Only rho/rho_save are exercised on this real-space path. + wire_charge(nspin, nrxx, pw_basis.npw, XC_Functional::get_ked_flag()); + // Non-trivial real-space residual: linear ramp so Kerker (which damps + // long wavelengths) actually changes the output vs the no-Kerker path. + std::vector real_ref(nspin * nrxx); + std::vector real_save_ref(nspin * nrxx); + for (int i = 0; i < nspin * nrxx; ++i) + { + real_ref[i] = 0.3 + 0.01 * i; + real_save_ref[i] = 0.1 + 0.005 * i; + } + + // --- Run A: close_kerker_gg0() then mix_rho --- + Charge_Mixing CM_disabled; + CM_disabled.set_mixing(make_cfg(), &pw_basis, &pw_basis, ucell.omega, ucell.tpiba); + CM_disabled.init_mixing(); + CM_disabled.close_kerker_gg0(); + for (int i = 0; i < nspin * nrxx; ++i) + { + charge.rho[i / nrxx][i % nrxx] = real_ref[i]; + charge.rho_save[i / nrxx][i % nrxx] = real_save_ref[i]; + } + CM_disabled.mix_rho(&charge); + std::vector rho_A(charge.rho[0], charge.rho[0] + nspin * nrxx); + + // --- Run B: cfg.mixing_gg0 = 0 baseline, no close_kerker_gg0 --- + Charge_Mixing CM_baseline; + MixingConfig cfg_off = make_cfg(); + cfg_off.mixing_gg0 = 0.0; // Kerker off at config level + CM_baseline.set_mixing(cfg_off, &pw_basis, &pw_basis, ucell.omega, ucell.tpiba); + CM_baseline.init_mixing(); + for (int i = 0; i < nspin * nrxx; ++i) + { + charge.rho[i / nrxx][i % nrxx] = real_ref[i]; + charge.rho_save[i / nrxx][i % nrxx] = real_save_ref[i]; + } + CM_baseline.mix_rho(&charge); + std::vector rho_B(charge.rho[0], charge.rho[0] + nspin * nrxx); + + // close_kerker_gg0 path must match the Kerker-off baseline. + for (int i = 0; i < nspin * nrxx; ++i) + { + EXPECT_NEAR(rho_A[i], rho_B[i], 1e-10) + << "i=" << i << ": close_kerker_gg0 did not disable Kerker"; + } + + // --- Run C: Kerker active, no close_kerker_gg0. Output must differ from A + // to prove the disable flag was load-bearing (not that Kerker was a no-op + // for this input to begin with). --- + Charge_Mixing CM_active; + CM_active.set_mixing(make_cfg(), &pw_basis, &pw_basis, ucell.omega, ucell.tpiba); + CM_active.init_mixing(); + for (int i = 0; i < nspin * nrxx; ++i) + { + charge.rho[i / nrxx][i % nrxx] = real_ref[i]; + charge.rho_save[i / nrxx][i % nrxx] = real_save_ref[i]; + } + CM_active.mix_rho(&charge); + std::vector rho_C(charge.rho[0], charge.rho[0] + nspin * nrxx); + + bool any_diff = false; + for (int i = 0; i < nspin * nrxx; ++i) + { + if (std::abs(rho_A[i] - rho_C[i]) > 1e-8) + { + any_diff = true; + break; + } + } + EXPECT_TRUE(any_diff) + << "Kerker-active output equals Kerker-disabled output, so the " + "close_kerker_gg0 test cannot prove the flag does anything"; + + // the fixture owns the buffers behind charge.rho/rho_save +} + +TEST_F(ChargeMixingTest, MixDoubleGridRhoTest) +{ + cfg.double_grid = true; + charge.set_rhopw(&pw_dbasis); + const int nspin = cfg.nspin = 1; + cfg.domag_z = false; + XC_Functional::set_func_type(3); + XC_Functional::set_ked_flag(true); + cfg.mixing_beta = 0.7; + cfg.mixing_ndim = 1; + cfg.mixing_gg0 = 0.0; + cfg.mixing_tau = true; + cfg.mixing_mode = "plain"; + const int nrxx = pw_dbasis.nrxx; + const int npw = pw_dbasis.npw; + // charge is pointed at pw_dbasis here, so size the buffers from the dense grid. + wire_charge(nspin, nrxx, npw, XC_Functional::get_ked_flag()); + std::vector real_ref(nspin * nrxx); + std::vector real_save_ref(nspin * nrxx); + std::vector> recip_ref(nspin * npw); + std::vector> recip_save_ref(nspin * npw); + for (int i = 0; i < nspin * npw; ++i) + { + recip_ref[i] = std::complex(double(i), 1.0); + recip_save_ref[i] = std::complex(double(i), 0.0); + } + for (int i = 0; i < nspin; ++i) + { + pw_dbasis.recip2real(recip_ref.data() + i * npw, real_ref.data() + i * nrxx); + pw_dbasis.recip2real(recip_save_ref.data() + i * npw, real_save_ref.data() + i * nrxx); + } + //--------------------------------MAIN BODY-------------------------------- + // RECIPROCAL + Charge_Mixing CMtest_recip; + + cfg.scf_thr_type= 1; + CMtest_recip.set_mixing(make_cfg(), &pw_basis, &pw_dbasis, ucell.omega, ucell.tpiba); + + CMtest_recip.init_mixing(); + for (int i = 0; i < nspin * npw; ++i) + { + charge.rhog[i / npw][i % npw] = recip_ref[i]; + charge.rhog_save[i / npw][i % npw] = recip_save_ref[i]; + } + for (int i = 0; i < nspin * nrxx; ++i) + { + charge.rho[i / nrxx][i % nrxx] = real_ref[i]; + charge.rho_save[i / nrxx][i % nrxx] = real_save_ref[i]; + } + CMtest_recip.mix_rho(&charge); + for (int is = 0; is < nspin; ++is) + { + for (int ir = 0; ir < nrxx; ++ir) + { + EXPECT_NEAR(charge.rho_save[is][ir], real_ref[is * nrxx + ir], 1e-8); + } + for (int ig = 0; ig < npw; ++ig) + { + EXPECT_NEAR(charge.rhog[is][ig].real(), recip_save_ref[is * npw + ig].real(), 1e-8); + EXPECT_NEAR(charge.rhog[is][ig].imag(), recip_save_ref[is * npw + ig].imag() + 0.7, 1e-8); + } + } + + //------------------------------------------------------------------------- + // the fixture owns the buffers behind charge.rho/rhog/kin_r, so there is + // nothing to delete here +} + +TEST_F(ChargeMixingTest, MixDivCombTest) +{ + // NSPIN = 1 + cfg.nspin = 1; + Charge_Mixing CMtest; + std::vector> data(pw_dbasis.npw, 1.0); + const int npw_smooth = pw_basis.npw; + const int npw_dense = pw_dbasis.npw; + const int npw_hf = npw_dense - npw_smooth; + + // split: smooth + high-frequency together reconstruct the dense data + std::vector> datas(npw_smooth); + std::vector> datahf(npw_hf); + module_charge::split_dgrid(data.data(), datas, datahf, + 1, npw_smooth, npw_dense); + for (int i = 0; i < npw_smooth; ++i) + { + EXPECT_EQ(datas[i], data[i]); + } + for (int i = 0; i < npw_hf; ++i) + { + EXPECT_EQ(datahf[i], data[npw_smooth + i]); + } + + // merge: inverse of split; output must equal input + std::vector> dataout(npw_dense, std::complex(0, 0)); + module_charge::merge_dgrid(dataout.data(), datas, datahf, + 1, npw_smooth, npw_dense); + for (int i = 0; i < npw_dense; ++i) + { + EXPECT_EQ(dataout[i], data[i]); + } + + // No explicit cleanup call needed: vectors manage their own storage. + + // NSPIN = 2 + cfg.nspin = 2; + data.resize(npw_dense * 2, 1.0); + dataout.assign(npw_dense * 2, std::complex(0, 0)); + std::vector> datas2(npw_smooth * 2); + std::vector> datahf2(npw_hf * 2); + module_charge::split_dgrid(data.data(), datas2, datahf2, + 2, npw_smooth, npw_dense); + module_charge::merge_dgrid(dataout.data(), datas2, datahf2, + 2, npw_smooth, npw_dense); + for (int i = 0; i < npw_dense * 2; ++i) + { + EXPECT_EQ(dataout[i], data[i]); + } +} + +TEST_F(ChargeMixingTest, SCFOscillationTest) +{ + Charge_Mixing CMtest; + int scf_nmax = 20; + // if_scf_oscillate sizes _drho_history from cfg_.scf_nmax, so cfg_ must + // be populated before the loop; a default-constructed cfg_ leaves it 0. + cfg.scf_nmax = scf_nmax; + CMtest.set_mixing(make_cfg(), &pw_basis, &pw_basis, ucell.omega, ucell.tpiba); + int scf_os_ndim = 3; + double scf_os_thr = -0.05; + bool scf_oscillate = false; + std::vector drho(scf_nmax, 0.0); + std::vector scf_oscillate_ref(scf_nmax, false); + drho = {6.83639633652e-05, + 4.93523029235e-05, + 3.59230097735e-05, + 2.68356403913e-05, + 2.17490806464e-05, + 2.14231642508e-05, + 1.67507494811e-05, + 1.53575889539e-05, + 1.26504511554e-05, + 1.04762016224e-05, + 8.10000162918e-06, + 7.66427917682e-06, + 6.70112820094e-06, + 5.68594436664e-06, + 4.80120233733e-06, + 4.86519757184e-06, + 4.37855804356e-06, + 4.29922703412e-06, + 4.36398486331e-06, + 4.94224615955e-06}; + scf_oscillate_ref = {false,false,false,false,false,true,false,false,false,false, + false,false,true,false,false,true,true,true,true,true}; + for (int i = 1; i <= scf_nmax; ++i) + { + scf_oscillate = CMtest.if_scf_oscillate(i,drho[i-1],scf_os_ndim,scf_os_thr); + EXPECT_EQ(scf_oscillate, scf_oscillate_ref[i-1]); + } +} diff --git a/source/source_estate/module_charge/unittests/test_chg_mix_rho.cpp b/source/source_estate/module_charge/unittests/test_chg_mix_rho.cpp new file mode 100644 index 00000000000..ff298ff72cc --- /dev/null +++ b/source/source_estate/module_charge/unittests/test_chg_mix_rho.cpp @@ -0,0 +1,192 @@ +#include "gmock/gmock.h" +#include "gtest/gtest.h" + +#include "source_base/matrix3.h" +#include "source_basis/module_pw/pw_basis_sup.h" +#include "source_cell/magnetism.h" +#include "source_estate/module_charge/charge.h" +#include "source_estate/module_charge/chg_mix.h" +#include "source_estate/module_charge/chg_mix_cfg.h" +#include "source_io/module_parameter/parameter.h" + +#include + +// charge.cpp references Magnetism; provide a lightweight stub. +Magnetism::Magnetism() +{ + this->tot_mag = 0.0; + this->abs_mag = 0.0; +} +Magnetism::~Magnetism() +{ +} + +/************************************************ + * unit test of module_charge/chg_mix_rho.cpp + ***********************************************/ + +/** + * - Tested Functions: + * - Charge_Mixing::mix_rho: dispatches to mix_rho_recip (scf_thr_type==1) + * or mix_rho_real (scf_thr_type==2), then copies rho->rho_save. + * - abort on null chr / null chr->rhopw + * - abort when the grid was not set via set_mixing + * - abort when double_grid is on but rhodpw is null + * - real-space plain mixing: rho = rho_save + beta * (rho_new - rho_save) + */ + +namespace +{ + +MixingConfig make_cfg(int nspin, int scf_thr_type, bool double_grid, bool mixing_tau) +{ + MixingConfig cfg{ + "plain", // mixing_mode + 0.7, // mixing_beta + 1, // mixing_ndim + 0.0, // mixing_gg0 + mixing_tau, // mixing_tau + 1.6, // mixing_beta_mag + 0.0, // mixing_gg0_mag + 0.1, // mixing_gg0_min + -10.0, // mixing_angle + false, // mixing_dmr + nspin, // nspin + scf_thr_type, // scf_thr_type + double_grid, // double_grid + false, // gamma_only_pw + false, // domag + false, // domag_z + 100 // scf_nmax + }; + return cfg; +} + +} // namespace + +class ChargeMixRhoTest : public ::testing::Test +{ + public: + ModulePW::PW_Basis pw_basis; + ModulePW::PW_Basis_Sup pw_dbasis; + Charge charge; + double omega = 1.0; + double tpiba = 1.0; + + ChargeMixRhoTest() + { + pw_basis.initgrids(4, ModuleBase::Matrix3(1, 0, 0, 0, 1, 0, 0, 0, 1), 20); + pw_basis.initparameters(false, 20); + pw_basis.setuptransform(); + pw_basis.collect_local_pw(); + pw_dbasis.initgrids(4, ModuleBase::Matrix3(1, 0, 0, 0, 1, 0, 0, 0, 1), 40); + pw_dbasis.initparameters(false, 40); + pw_dbasis.setuptransform(&pw_basis); + pw_dbasis.collect_local_pw(); + } + + /// Configure a Charge_Mixing for plain mixing on the smooth grid. + void setup_mixing(Charge_Mixing& cm, int nspin, int scf_thr_type, bool double_grid) + { + MixingConfig cfg = make_cfg(nspin, scf_thr_type, double_grid, false); + if (double_grid) + { + cm.set_mixing(cfg, &pw_basis, &pw_dbasis, omega, tpiba); + } + else + { + cm.set_mixing(cfg, &pw_basis, &pw_basis, omega, tpiba); + } + cm.init_mixing(); + } + + /// Allocate Charge buffers (rho, rho_save) for nspin without kinetic density. + void setup_charge(int nspin) + { + charge.set_rhopw(&pw_basis); + const bool kin_den = false; + const bool meta_gga = false; + charge.allocate(nspin, kin_den, meta_gga, 0); + } +}; + +// --------------------------------------------------------------------------- +// abort paths +// --------------------------------------------------------------------------- + +TEST_F(ChargeMixRhoTest, MixRhoNullChrAborts) +{ + Charge_Mixing cm; + MixingConfig cfg = make_cfg(1, 2, false, false); + cm.set_mixing(cfg, &pw_basis, &pw_basis, omega, tpiba); + cm.init_mixing(); + EXPECT_DEATH(cm.mix_rho(nullptr), ""); +} + +TEST_F(ChargeMixRhoTest, MixRhoNullChrRhopwAborts) +{ + Charge_Mixing cm; + setup_mixing(cm, 1, 2, false); + Charge empty_charge; + EXPECT_DEATH(cm.mix_rho(&empty_charge), ""); +} + +TEST_F(ChargeMixRhoTest, MixRhoUnsetRhopwAborts) +{ + Charge_Mixing cm; + MixingConfig cfg = make_cfg(1, 2, false, false); + // Pass rhopw == nullptr to set_mixing to simulate a skipped grid setup. + // Do NOT call init_mixing() here: init_mixing already WARNING_QUITs when + // the grid is unset, which would kill the death-test parent process + // before EXPECT_DEATH runs. The guard under test lives in mix_rho itself + // and only checks this->rhopw == nullptr, independent of init_mixing. + cm.set_mixing(cfg, nullptr, nullptr, omega, tpiba); + setup_charge(1); + EXPECT_DEATH(cm.mix_rho(&charge), ""); +} + +TEST_F(ChargeMixRhoTest, MixRhoDoubleGridWithoutRhodpwAborts) +{ + Charge_Mixing cm; + MixingConfig cfg = make_cfg(1, 2, true, false); + // set_mixing with rhodpw == nullptr while double_grid is on + cm.set_mixing(cfg, &pw_basis, nullptr, omega, tpiba); + cm.init_mixing(); + setup_charge(1); + EXPECT_DEATH(cm.mix_rho(&charge), ""); +} + +// --------------------------------------------------------------------------- +// real-space plain mixing: rho = rho_save + beta * (rho - rho_save) +// --------------------------------------------------------------------------- + +TEST_F(ChargeMixRhoTest, MixRhoRealPlainNspin1) +{ + Charge_Mixing cm; + const int nspin = 1; + setup_mixing(cm, nspin, 2, false); + setup_charge(nspin); + + const double rho_save_val = 1.0; + const double rho_new_val = 3.0; + const double beta = 0.7; + for (int ir = 0; ir < pw_basis.nrxx; ++ir) + { + charge.rho[0][ir] = rho_new_val; + charge.rho_save[0][ir] = rho_save_val; + } + + cm.mix_rho(&charge); + + // mixed rho = rho_save + beta * (rho_new - rho_save) = 1 + 0.7 * 2 = 2.4 + const double expected = rho_save_val + beta * (rho_new_val - rho_save_val); + for (int ir = 0; ir < pw_basis.nrxx; ++ir) + { + EXPECT_NEAR(charge.rho[0][ir], expected, 1e-8); + } + // rho_save holds the pre-mixing rho (rho_new_val) + for (int ir = 0; ir < pw_basis.nrxx; ++ir) + { + EXPECT_NEAR(charge.rho_save[0][ir], rho_new_val, 1e-8); + } +} diff --git a/source/source_estate/module_charge/unittests/test_chg_parallel.cpp b/source/source_estate/module_charge/unittests/test_chg_parallel.cpp new file mode 100644 index 00000000000..5ee21d1884b --- /dev/null +++ b/source/source_estate/module_charge/unittests/test_chg_parallel.cpp @@ -0,0 +1,287 @@ +#include "source_base/matrix3.h" +#include "source_base/parallel_global.h" +#include "source_base/parallel_grid.h" +#include "source_estate/module_charge/charge.h" +#include "source_estate/module_charge/chg_parallel.h" +#include "source_hamilt/module_xc/xc_functional.h" +#include "source_io/module_parameter/parameter.h" + +#include "gmock/gmock.h" +#include "gtest/gtest.h" + +bool XC_Functional::ked_flag = false; +void XC_Functional::set_xc_type(const std::string xc_func_in) +{ + ked_flag = xc_func_in == "scan"; +} +Charge::Charge() +{ +} +Charge::~Charge() +{ +} + +auto sum_array = [](const double* v, const int& nv) { + double sum = 0; + for (int i = 0; i < nv; ++i) + { + sum += v[i]; + } + return sum; +}; +/************************************************ + * unit test of module_charge/chg_parallel.cpp + ***********************************************/ + +/** + * - Tested Functions: + * - rho_mpi: module_charge::rho_mpi() + * - test rho_mpi + * - reduce_diff_pools: module_charge::reduce_diff_pools() + * - test reduce_diff_pools + * - using rhopw and GlobalV + */ + +class ChargeMpiTest : public ::testing::Test +{ + protected: + Charge* charge; + std::string output; + double lat0 = 4; + ModuleBase::Matrix3 latvec; + int world_size = 1; + void SetUp() override + { + charge = new Charge; + MPI_Comm_size(MPI_COMM_WORLD, &world_size); + } + void TearDown() override + { + delete charge; + } +}; + +TEST_F(ChargeMpiTest, reduce_diff_pools1) +{ + if (GlobalV::NPROC >= 2 && GlobalV::NPROC % 2 == 0) + { + GlobalV::KPAR = 2; + Parallel_Global::init_pools(GlobalV::NPROC, + GlobalV::MY_RANK, + PARAM.inp.bndpar, + GlobalV::KPAR, + GlobalV::NPROC_IN_BNDGROUP, + GlobalV::RANK_IN_BPGROUP, + GlobalV::MY_BNDGROUP, + GlobalV::NPROC_IN_POOL, + GlobalV::RANK_IN_POOL, + GlobalV::MY_POOL); + ModulePW::PW_Basis* rhopw = new ModulePW::PW_Basis(); + rhopw->initmpi(GlobalV::NPROC_IN_POOL, GlobalV::RANK_IN_POOL, POOL_WORLD); + rhopw->initgrids(lat0, latvec, 40); + rhopw->initparameters(false, 10); + rhopw->setuptransform(); + Parallel_Grid pgrid; + pgrid.init(rhopw->nx, rhopw->ny, rhopw->nz, rhopw->nplane, rhopw->nrxx, rhopw->nz, 1, world_size); + charge->pgrid = &pgrid; + + int nz = rhopw->nz; + const int nrxx = rhopw->nrxx; + const int nxy = rhopw->nxy; + const int nplane = rhopw->nplane; + charge->nrxx = nrxx; + double* array_rho = new double[nrxx]; + for (int ir = 0; ir < nxy; ++ir) + { + for (int iz = 0; iz < nplane; ++iz) + { + array_rho[nplane * ir + iz] = (rhopw->startz_current + iz + ir * nz) / double(nxy * nz); + } + } + double refsum = sum_array(array_rho, nrxx); + + module_charge::reduce_diff_pools(array_rho, *charge, GlobalV::KPAR, + PARAM.globalv.all_ks_run, PARAM.inp.bndpar); + double sum = sum_array(array_rho, nrxx); + EXPECT_EQ(sum, refsum * GlobalV::KPAR); + + delete[] array_rho; + delete rhopw; + } +} + +TEST_F(ChargeMpiTest, reduce_diff_pools2) +{ + if (GlobalV::NPROC >= 3) + { + GlobalV::KPAR = 3; + Parallel_Global::divide_pools(GlobalV::NPROC, + GlobalV::MY_RANK, + PARAM.inp.bndpar, + GlobalV::KPAR, + GlobalV::NPROC_IN_BNDGROUP, + GlobalV::RANK_IN_BPGROUP, + GlobalV::MY_BNDGROUP, + GlobalV::NPROC_IN_POOL, + GlobalV::RANK_IN_POOL, + GlobalV::MY_POOL); + ModulePW::PW_Basis* rhopw = new ModulePW::PW_Basis(); + rhopw->initmpi(GlobalV::NPROC_IN_POOL, GlobalV::RANK_IN_POOL, POOL_WORLD); + rhopw->initgrids(lat0, latvec, 40); + rhopw->initparameters(false, 10); + rhopw->setuptransform(); + charge->rhopw = rhopw; + Parallel_Grid pgrid; + pgrid.init(rhopw->nx, rhopw->ny, rhopw->nz, rhopw->nplane, rhopw->nrxx, rhopw->nz, 1, world_size); + charge->pgrid = &pgrid; + + int nz = rhopw->nz; + const int nrxx = rhopw->nrxx; + const int nxy = rhopw->nxy; + const int nplane = rhopw->nplane; + charge->nrxx = nrxx; + double* array_ref = new double[nxy * nz]; + for (int ixyz = 0; ixyz < nxy * nz; ++ixyz) + { + array_ref[ixyz] = ixyz / double(nxy * nz); + } + double refsum = sum_array(array_ref, nxy * nz); + + double* array_rho = new double[nrxx]; + for (int ir = 0; ir < nxy; ++ir) + { + for (int iz = 0; iz < nplane; ++iz) + { + array_rho[nplane * ir + iz] = (rhopw->startz_current + iz + ir * nz) / double(nxy * nz); + } + } + + module_charge::reduce_diff_pools(array_rho, *charge, GlobalV::KPAR, + PARAM.globalv.all_ks_run, PARAM.inp.bndpar); + double sum = sum_array(array_rho, nrxx); + MPI_Allreduce(MPI_IN_PLACE, &sum, 1, MPI_DOUBLE, MPI_SUM, POOL_WORLD); + EXPECT_EQ(sum, refsum * GlobalV::KPAR); + + delete[] array_rho; + delete rhopw; + delete[] array_ref; + } +} + +TEST_F(ChargeMpiTest, rho_mpi) +{ + if (GlobalV::NPROC >= 2) + { + GlobalV::KPAR = 2; + Parallel_Global::divide_pools(GlobalV::NPROC, + GlobalV::MY_RANK, + PARAM.inp.bndpar, + GlobalV::KPAR, + GlobalV::NPROC_IN_BNDGROUP, + GlobalV::RANK_IN_BPGROUP, + GlobalV::MY_BNDGROUP, + GlobalV::NPROC_IN_POOL, + GlobalV::RANK_IN_POOL, + GlobalV::MY_POOL); + ModulePW::PW_Basis* rhopw = new ModulePW::PW_Basis(); + rhopw->initmpi(GlobalV::NPROC_IN_POOL, GlobalV::RANK_IN_POOL, POOL_WORLD); + rhopw->initgrids(lat0, latvec, 40); + rhopw->initparameters(false, 10); + rhopw->setuptransform(); + charge->rhopw = rhopw; + Parallel_Grid pgrid; + pgrid.init(rhopw->nx, rhopw->ny, rhopw->nz, rhopw->nplane, rhopw->nrxx, rhopw->nz, 1, world_size); + charge->pgrid = &pgrid; + ASSERT_EQ(PARAM.inp.nspin, 1); + charge->rho = new double*[1]; + charge->kin_r = new double*[1]; + + int nz = rhopw->nz; + const int nrxx = rhopw->nrxx; + const int nxy = rhopw->nxy; + const int nplane = rhopw->nplane; + charge->nrxx = nrxx; + charge->rho[0] = new double[nrxx]; + charge->kin_r[0] = new double[nrxx]; + module_charge::rho_mpi(*charge, GlobalV::KPAR, PARAM.globalv.all_ks_run, + PARAM.inp.bndpar, PARAM.inp.nspin); + + delete[] charge->rho[0]; + delete[] charge->rho; + delete[] charge->kin_r[0]; + delete[] charge->kin_r; + delete rhopw; + } + + GlobalV::KPAR = 1; + module_charge::rho_mpi(*charge, GlobalV::KPAR, PARAM.globalv.all_ks_run, + PARAM.inp.bndpar, PARAM.inp.nspin); +} + +TEST_F(ChargeMpiTest, kin_r_mpi) +{ + if (GlobalV::NPROC >= 2 && GlobalV::NPROC % 2 == 0) + { + ASSERT_EQ(PARAM.inp.nspin, 1); + ASSERT_EQ(PARAM.inp.bndpar, 1); + GlobalV::KPAR = 2; + + Parallel_Global::divide_pools(GlobalV::NPROC, + GlobalV::MY_RANK, + PARAM.inp.bndpar, + GlobalV::KPAR, + GlobalV::NPROC_IN_BNDGROUP, + GlobalV::RANK_IN_BPGROUP, + GlobalV::MY_BNDGROUP, + GlobalV::NPROC_IN_POOL, + GlobalV::RANK_IN_POOL, + GlobalV::MY_POOL); + ModulePW::PW_Basis* rhopw = new ModulePW::PW_Basis(); + rhopw->initmpi(GlobalV::NPROC_IN_POOL, GlobalV::RANK_IN_POOL, POOL_WORLD); + rhopw->initgrids(lat0, latvec, 40); + rhopw->initparameters(false, 10); + rhopw->setuptransform(); + charge->rhopw = rhopw; + Parallel_Grid pgrid; + pgrid.init(rhopw->nx, rhopw->ny, rhopw->nz, rhopw->nplane, rhopw->nrxx, rhopw->nz, 1, world_size); + charge->pgrid = &pgrid; + + const int nz = rhopw->nz; + const int nrxx = rhopw->nrxx; + const int nxy = rhopw->nxy; + const int nplane = rhopw->nplane; + charge->nrxx = nrxx; + charge->kin_r = new double*[1]; + charge->kin_r[0] = new double[nrxx]; + + for (int ir = 0; ir < nxy; ++ir) + { + for (int iz = 0; iz < nplane; ++iz) + { + charge->kin_r[0][nplane * ir + iz] = (rhopw->startz_current + iz + ir * nz) / double(nxy * nz); + } + } + const double refsum = sum_array(charge->kin_r[0], nrxx); + + module_charge::kin_r_mpi(*charge, GlobalV::KPAR, PARAM.globalv.all_ks_run, + PARAM.inp.bndpar, PARAM.inp.nspin); + const double sum = sum_array(charge->kin_r[0], nrxx); + EXPECT_EQ(sum, refsum * GlobalV::KPAR); + + delete[] charge->kin_r[0]; + delete[] charge->kin_r; + delete rhopw; + } +} + +int main(int argc, char** argv) +{ + MPI_Init(&argc, &argv); + MPI_Comm_size(MPI_COMM_WORLD, &GlobalV::NPROC); + MPI_Comm_rank(MPI_COMM_WORLD, &GlobalV::MY_RANK); + int result = 0; + testing::InitGoogleTest(&argc, argv); + result = RUN_ALL_TESTS(); + MPI_Finalize(); + return result; +} diff --git a/source/source_estate/module_charge/unittests/test_chg_precond.cpp b/source/source_estate/module_charge/unittests/test_chg_precond.cpp new file mode 100644 index 00000000000..41fec9386b2 --- /dev/null +++ b/source/source_estate/module_charge/unittests/test_chg_precond.cpp @@ -0,0 +1,271 @@ +#include "gtest/gtest.h" + +#include "source_base/constants.h" +#include "source_base/matrix3.h" +#include "source_basis/module_pw/pw_basis.h" +#include "source_estate/module_charge/chg_mix_cfg.h" +#include "source_estate/module_charge/chg_precond.h" + +#include +#include +#include + +/************************************************ + * unit test of module_charge/chg_precond.cpp + ***********************************************/ + +/** + * - Tested Functions: + * - kerker_screen_recip: multiply drhog[is*npw+ig] by + * max(gg/(gg+gg0), gg0_min/amin) per spin channel + * - early return when mixing_gg0 <= 0 or mixing_beta <= 0.1 + * - nspin == 1: density channel only + * - nspin == 2: density + magnetization (mag skipped when gg0_mag/beta_mag too small) + * - nspin == 4: density + magnetization channels (resize_tmp with mixing_angle) + * - kerker_screen_real: FFT to reciprocal, apply (1 - filter_g), FFT back, subtract + * - early return when mixing_gg0 <= 0.0001 or mixing_beta <= 0.1 + * - nspin == 1 real-space filtering matches reciprocal-space result + */ + +namespace +{ + +MixingConfig make_cfg() +{ + MixingConfig cfg{ + "broyden", // mixing_mode + 0.8, // mixing_beta + 8, // mixing_ndim + 1.0, // mixing_gg0 + false, // mixing_tau + 1.6, // mixing_beta_mag + 0.0, // mixing_gg0_mag + 0.1, // mixing_gg0_min + -10.0, // mixing_angle + false, // mixing_dmr + 1, // nspin + 2, // scf_thr_type + false, // double_grid + false, // gamma_only_pw + false, // domag + false, // domag_z + 100 // scf_nmax + }; + return cfg; +} + +} // namespace + +class ChgPrecondTest : public ::testing::Test +{ + protected: + ModulePW::PW_Basis pw_basis; + const double tpiba = 1.0; + + void SetUp() override + { + pw_basis.initgrids(4, ModuleBase::Matrix3(1, 0, 0, 0, 1, 0, 0, 0, 1), 20); + pw_basis.initparameters(false, 20); + pw_basis.setuptransform(); + pw_basis.collect_local_pw(); + } +}; + +// --------------------------------------------------------------------------- +// kerker_screen_recip +// --------------------------------------------------------------------------- + +TEST_F(ChgPrecondTest, KerkerScreenRecipEarlyReturnGg0Zero) +{ + MixingConfig cfg = make_cfg(); + cfg.nspin = 1; + cfg.mixing_gg0 = 0.0; + + std::vector> drhog(pw_basis.npw, std::complex(1.0, 1.0)); + std::vector> drhog_old = drhog; + + module_charge::kerker_screen_recip(cfg, &pw_basis, tpiba, drhog.data()); + + for (int ig = 0; ig < pw_basis.npw; ++ig) + { + EXPECT_EQ(drhog[ig], drhog_old[ig]); + } +} + +TEST_F(ChgPrecondTest, KerkerScreenRecipEarlyReturnBetaTooSmall) +{ + MixingConfig cfg = make_cfg(); + cfg.nspin = 1; + cfg.mixing_gg0 = 1.0; + cfg.mixing_beta = 0.1; // <= 0.1 triggers early return + + std::vector> drhog(pw_basis.npw, std::complex(1.0, 1.0)); + std::vector> drhog_old = drhog; + + module_charge::kerker_screen_recip(cfg, &pw_basis, tpiba, drhog.data()); + + for (int ig = 0; ig < pw_basis.npw; ++ig) + { + EXPECT_EQ(drhog[ig], drhog_old[ig]); + } +} + +TEST_F(ChgPrecondTest, KerkerScreenRecipNspin1Filter) +{ + MixingConfig cfg = make_cfg(); + cfg.nspin = 1; + cfg.mixing_gg0 = 1.0; + cfg.mixing_beta = 0.8; + cfg.mixing_gg0_min = 0.1; + + std::vector> drhog(pw_basis.npw, std::complex(1.0, 1.0)); + + module_charge::kerker_screen_recip(cfg, &pw_basis, tpiba, drhog.data()); + + const double gg0 = std::pow(cfg.mixing_gg0 * ModuleBase::BOHR_TO_A / tpiba, 2); + const double gg0_amin = cfg.mixing_gg0_min / cfg.mixing_beta; + for (int ig = 0; ig < pw_basis.npw; ++ig) + { + const double gg = pw_basis.gg[ig]; + const double ref = std::max(gg / (gg + gg0), gg0_amin); + EXPECT_NEAR(drhog[ig].real(), ref, 1e-10); + EXPECT_NEAR(drhog[ig].imag(), ref, 1e-10); + } +} + +TEST_F(ChgPrecondTest, KerkerScreenRecipNspin2MagSkippedWhenGg0MagZero) +{ + MixingConfig cfg = make_cfg(); + cfg.nspin = 2; + cfg.mixing_gg0 = 1.0; + cfg.mixing_beta = 0.8; + cfg.mixing_gg0_mag = 0.0; // magnetization channel is skipped (break) + cfg.mixing_gg0_min = 0.1; + + std::vector> drhog(2 * pw_basis.npw, std::complex(1.0, 1.0)); + + module_charge::kerker_screen_recip(cfg, &pw_basis, tpiba, drhog.data()); + + const double gg0 = std::pow(cfg.mixing_gg0 * ModuleBase::BOHR_TO_A / tpiba, 2); + const double gg0_amin = cfg.mixing_gg0_min / cfg.mixing_beta; + for (int ig = 0; ig < pw_basis.npw; ++ig) + { + const double gg = pw_basis.gg[ig]; + const double ref = std::max(gg / (gg + gg0), gg0_amin); + // density channel is filtered + EXPECT_NEAR(drhog[ig].real(), ref, 1e-10); + EXPECT_NEAR(drhog[ig].imag(), ref, 1e-10); + // magnetization channel is untouched (break before processing is=1) + EXPECT_NEAR(drhog[pw_basis.npw + ig].real(), 1.0, 1e-10); + EXPECT_NEAR(drhog[pw_basis.npw + ig].imag(), 1.0, 1e-10); + } +} + +TEST_F(ChgPrecondTest, KerkerScreenRecipNspin2MagChannelFiltered) +{ + MixingConfig cfg = make_cfg(); + cfg.nspin = 2; + cfg.mixing_gg0 = 1.0; + cfg.mixing_beta = 0.8; + cfg.mixing_gg0_mag = 2.0; + cfg.mixing_beta_mag = 1.6; + cfg.mixing_gg0_min = 0.1; + + std::vector> drhog(2 * pw_basis.npw, std::complex(1.0, 1.0)); + + module_charge::kerker_screen_recip(cfg, &pw_basis, tpiba, drhog.data()); + + const double gg0_rho = std::pow(cfg.mixing_gg0 * ModuleBase::BOHR_TO_A / tpiba, 2); + const double gg0_mag = std::pow(cfg.mixing_gg0_mag * ModuleBase::BOHR_TO_A / tpiba, 2); + const double gg0_amin_rho = cfg.mixing_gg0_min / cfg.mixing_beta; + const double gg0_amin_mag = cfg.mixing_gg0_min / cfg.mixing_beta_mag; + for (int ig = 0; ig < pw_basis.npw; ++ig) + { + const double gg = pw_basis.gg[ig]; + const double ref_rho = std::max(gg / (gg + gg0_rho), gg0_amin_rho); + const double ref_mag = std::max(gg / (gg + gg0_mag), gg0_amin_mag); + EXPECT_NEAR(drhog[ig].real(), ref_rho, 1e-10); + EXPECT_NEAR(drhog[pw_basis.npw + ig].real(), ref_mag, 1e-10); + } +} + +TEST_F(ChgPrecondTest, KerkerScreenRecipNspin4WithAngle) +{ + MixingConfig cfg = make_cfg(); + cfg.nspin = 4; + cfg.mixing_gg0 = 1.0; + cfg.mixing_beta = 0.8; + cfg.mixing_gg0_mag = 2.0; + cfg.mixing_beta_mag = 1.6; + cfg.mixing_gg0_min = 0.1; + cfg.mixing_angle = 1.0; // > 0 => resize_tmp = 2 + + std::vector> drhog(4 * pw_basis.npw, std::complex(1.0, 1.0)); + + module_charge::kerker_screen_recip(cfg, &pw_basis, tpiba, drhog.data()); + + // resize_tmp == 2 means only 4/2 = 2 channels are processed: + // is=0 (density) and is=1 (magnetization). + const double gg0_rho = std::pow(cfg.mixing_gg0 * ModuleBase::BOHR_TO_A / tpiba, 2); + const double gg0_mag = std::pow(cfg.mixing_gg0_mag * ModuleBase::BOHR_TO_A / tpiba, 2); + const double gg0_amin_rho = cfg.mixing_gg0_min / cfg.mixing_beta; + const double gg0_amin_mag = cfg.mixing_gg0_min / cfg.mixing_beta_mag; + for (int ig = 0; ig < pw_basis.npw; ++ig) + { + const double gg = pw_basis.gg[ig]; + const double ref_rho = std::max(gg / (gg + gg0_rho), gg0_amin_rho); + const double ref_mag = std::max(gg / (gg + gg0_mag), gg0_amin_mag); + // is=0 density channel + EXPECT_NEAR(drhog[ig].real(), ref_rho, 1e-10); + // is=1 magnetization channel + EXPECT_NEAR(drhog[pw_basis.npw + ig].real(), ref_mag, 1e-10); + } +} + +// --------------------------------------------------------------------------- +// kerker_screen_real +// --------------------------------------------------------------------------- + +TEST_F(ChgPrecondTest, KerkerScreenRealEarlyReturnGg0Zero) +{ + MixingConfig cfg = make_cfg(); + cfg.nspin = 1; + cfg.mixing_gg0 = 0.0; + + std::vector drhor(pw_basis.nrxx, 1.0); + std::vector drhor_old = drhor; + + module_charge::kerker_screen_real(cfg, &pw_basis, tpiba, drhor.data()); + + for (int ir = 0; ir < pw_basis.nrxx; ++ir) + { + EXPECT_EQ(drhor[ir], drhor_old[ir]); + } +} + +TEST_F(ChgPrecondTest, KerkerScreenRealNspin1MatchesRecip) +{ + MixingConfig cfg = make_cfg(); + cfg.nspin = 1; + cfg.mixing_gg0 = 1.0; + cfg.mixing_beta = 0.8; + cfg.mixing_gg0_min = 0.1; + + // start from a uniform real-space field; its reciprocal image is nonzero + // only at G=0, which lets us verify the (1 - filter_g) subtraction. + std::vector drhor(pw_basis.nrxx, 1.0); + + // reference: apply kerker_screen_recip to the FFT of drhor, then FFT back + std::vector> drhog(pw_basis.npw); + pw_basis.real2recip(drhor.data(), drhog.data()); + module_charge::kerker_screen_recip(cfg, &pw_basis, tpiba, drhog.data()); + std::vector drhor_ref(pw_basis.nrxx); + pw_basis.recip2real(drhog.data(), drhor_ref.data()); + + module_charge::kerker_screen_real(cfg, &pw_basis, tpiba, drhor.data()); + + for (int ir = 0; ir < pw_basis.nrxx; ++ir) + { + EXPECT_NEAR(drhor[ir], drhor_ref[ir], 1e-8); + } +} diff --git a/source/source_estate/module_charge/unittests/test_chg_rhog_io.cpp b/source/source_estate/module_charge/unittests/test_chg_rhog_io.cpp new file mode 100644 index 00000000000..3947d1670c2 --- /dev/null +++ b/source/source_estate/module_charge/unittests/test_chg_rhog_io.cpp @@ -0,0 +1,487 @@ +#include "gmock/gmock.h" +#include "gtest/gtest.h" +#include "source_estate/module_charge/chg_rhog_io.h" +#include "source_base/module_parallel/para_world.h" +#include "source_base/module_parallel/para_tag.h" +#include "source_base/module_parallel/para_bridge.h" +#ifdef __MPI +#include "source_basis/module_pw/test/test_tool.h" +#include "mpi.h" +#endif +#include +#include +#include +#include +#include + +/** + * - Tested Functions: + * - read_rhog() + * - write_rhog() + * + * All binary inputs are generated inside the tests (either via write_rhog + * round-trip or by hand-crafting the byte layout), so no support/*.dat file + * is required. The hand-crafted case (ReadLegacyBinaryFormat) pins the + * on-disk format against silent drift of the writer. + */ + +class ReadRhogTest : public ::testing::Test +{ + protected: + ModulePW::PW_Basis rhopw; + std::vector>> rhog_data; + std::vector*> rhog; + Parallel::ParaWorld pw_world = Parallel::make_pw_world(); + std::ofstream warning_stream; + + static ModuleBase::Matrix3 latvec() + { + return ModuleBase::Matrix3(-0.5, 0.0, 0.5, + 0.0, 0.5, 0.5, + -0.5, 0.5, 0.0); + } + + void setup_pw_basis() + { +#ifdef __MPI + rhopw.initmpi(pw_world.size(), pw_world.rank(), pw_world.comm()); +#endif + // Small, explicitly-sized FFT grid: avoids the expensive automatic + // grid search in initgrids(lat0, latvec, ecut) while still providing + // a few dozen planewaves for meaningful IO tests. + rhopw.initgrids(6.5, latvec(), 8, 8, 8); + rhopw.initparameters(false, 120); + rhopw.setuptransform(); + rhopw.collect_local_pw(); + } + + void open_warning(const std::string& path) + { + warning_stream.open(path); + } + + void close_warning() + { + if (warning_stream.is_open()) + { + warning_stream.close(); + } + } + + std::string read_warning_file(const std::string& path) + { + std::ifstream ifs(path); + std::stringstream ss; + ss << ifs.rdbuf(); + ifs.close(); + return ss.str(); + } + + // Fill rhog_data with distinct, deterministic values for nspin channels. + void fill_rhog(const int nspin) + { + rhog_data.assign(nspin, std::vector>(rhopw.npw)); + rhog.clear(); + for (int is = 0; is < nspin; ++is) + { + for (int ig = 0; ig < rhopw.npw; ++ig) + { + rhog_data[is][ig] = std::complex((is + 1) * 1.0 * ig, (is + 1) * 0.1 * ig); + } + rhog.push_back(rhog_data[is].data()); + } + } + + // Write a binary rhog file by hand (no write_rhog), keeping the exact + // field order documented in chg_rhog_io.cpp: + // [3][gamma_only][npwtot][nspin][3] + // [9][b1..b3 (9 doubles)][9] + // [3*npwtot][miller ints][3*npwtot] + // per spin: [npwtot][complex doubles][npwtot] + // Only rank 0 writes; call sites must ensure the data lives on rank 0. + void write_binary_by_hand(const std::string& filename, + const int gamma_only_flag, + const int nspin_file, + const std::vector>& miller, + const std::vector>>& values) const + { + if (pw_world.rank() != 0) + { + return; + } + const int npw_file = static_cast(miller.size()); + std::ofstream ofs(filename, std::ios::binary); + + int size = 3; + ofs.write(reinterpret_cast(&size), sizeof(int)); + ofs.write(reinterpret_cast(&gamma_only_flag), sizeof(int)); + ofs.write(reinterpret_cast(&npw_file), sizeof(int)); + ofs.write(reinterpret_cast(&nspin_file), sizeof(int)); + ofs.write(reinterpret_cast(&size), sizeof(int)); + + size = 9; + const ModuleBase::Matrix3 GT = latvec().Inverse(); + const double b[9] = {GT.e11, GT.e12, GT.e13, GT.e21, GT.e22, GT.e23, GT.e31, GT.e32, GT.e33}; + ofs.write(reinterpret_cast(&size), sizeof(int)); + ofs.write(reinterpret_cast(b), 9 * sizeof(double)); + ofs.write(reinterpret_cast(&size), sizeof(int)); + + size = 3 * npw_file; + ofs.write(reinterpret_cast(&size), sizeof(int)); + for (const auto& m : miller) + { + ofs.write(reinterpret_cast(m.data()), 3 * sizeof(int)); + } + ofs.write(reinterpret_cast(&size), sizeof(int)); + + size = npw_file; + for (int is = 0; is < nspin_file; ++is) + { + ofs.write(reinterpret_cast(&size), sizeof(int)); + ofs.write(reinterpret_cast(values[is].data()), npw_file * sizeof(std::complex)); + ofs.write(reinterpret_cast(&size), sizeof(int)); + } + ofs.close(); + } + + virtual void SetUp() + { + // Buffers are sized to the actual basis in each test via fill_rhog + // after setup_pw_basis(); allocate a minimal placeholder here. + rhog_data.resize(1, std::vector>(1)); + rhog.push_back(rhog_data[0].data()); + } + + virtual void TearDown() + { + close_warning(); + } +}; + +// Round-trip: write known data with write_rhog, read back, verify values. +// Replaces the old support/charge-density.dat based test. +TEST_F(ReadRhogTest, ReadRhog) +{ + setup_pw_basis(); + fill_rhog(1); + + const std::string tmpfile = "test_rhog_read.dat"; + bool write_result = module_charge::write_rhog(tmpfile, rhopw.gamma_only, &rhopw, 1, latvec(), rhog.data(), pw_world, nullptr); + ASSERT_TRUE(write_result); + + std::vector> read_back(rhopw.npw); + std::complex* read_ptr = read_back.data(); + bool read_result = module_charge::read_rhog(tmpfile, &rhopw, 1, &read_ptr, pw_world, nullptr); + ASSERT_TRUE(read_result); + + for (int ig = 0; ig < rhopw.npw; ++ig) + { + EXPECT_DOUBLE_EQ(read_back[ig].real(), rhog_data[0][ig].real()); + EXPECT_DOUBLE_EQ(read_back[ig].imag(), rhog_data[0][ig].imag()); + } + std::remove(tmpfile.c_str()); +} + +// Pin the on-disk binary format: hand-craft a file (without write_rhog) +// and check read_rhog maps values to the correct G-vectors. +TEST_F(ReadRhogTest, ReadLegacyBinaryFormat) +{ + setup_pw_basis(); + fill_rhog(1); + + // Use two well-separated G-vectors: the Gamma point and the first + // non-zero planewave, so the Miller-index -> ig mapping is exercised. + std::vector> miller(2); + miller[0] = {0, 0, 0}; + const ModuleBase::Vector3 g1 = rhopw.gdirect[1]; + miller[1] = {static_cast(g1.x), static_cast(g1.y), static_cast(g1.z)}; + + std::vector>> values(1, std::vector>(2)); + values[0][0] = std::complex(1.5, -0.5); + values[0][1] = std::complex(2.5, 3.5); + + const std::string tmpfile = "test_rhog_legacy.dat"; + write_binary_by_hand(tmpfile, 0, 1, miller, values); + + bool read_result = module_charge::read_rhog(tmpfile, &rhopw, 1, rhog.data(), pw_world, nullptr); + ASSERT_TRUE(read_result); + + // ig_gge0 is the index of the Gamma point in the local basis. + EXPECT_DOUBLE_EQ(rhog_data[0][rhopw.ig_gge0].real(), 1.5); + EXPECT_DOUBLE_EQ(rhog_data[0][rhopw.ig_gge0].imag(), -0.5); + // The second entry lands on gdirect[1]; find its ig via the Miller index. + bool found = false; + for (int ig = 0; ig < rhopw.npw; ++ig) + { + if (static_cast(rhopw.gdirect[ig].x) == miller[1][0] && + static_cast(rhopw.gdirect[ig].y) == miller[1][1] && + static_cast(rhopw.gdirect[ig].z) == miller[1][2]) + { + EXPECT_DOUBLE_EQ(rhog_data[0][ig].real(), 2.5); + EXPECT_DOUBLE_EQ(rhog_data[0][ig].imag(), 3.5); + found = true; + } + } + EXPECT_TRUE(found); + std::remove(tmpfile.c_str()); +} + +// Test the read_rhog function when the file is not found +TEST_F(ReadRhogTest, NotFoundFile) +{ + setup_pw_basis(); + fill_rhog(1); + std::string filename = "notfound.txt"; + + open_warning("test_read_rhog.txt"); + bool result = module_charge::read_rhog(filename, &rhopw, 1, rhog.data(), pw_world, &warning_stream); + close_warning(); + + std::string expected_content = " module_charge::read_rhog warning : Can't open file notfound.txt\n"; + EXPECT_FALSE(result); + EXPECT_EQ(read_warning_file("test_read_rhog.txt"), expected_content); + std::remove("test_read_rhog.txt"); +} + +// Test the read_rhog function when gamma_only is inconsistent +TEST_F(ReadRhogTest, InconsistentGammaOnly) +{ + setup_pw_basis(); + fill_rhog(1); + + // Self-generate a file with gamma_only=0, nspin=1. + const std::string tmpfile = "test_rhog_gamma.dat"; + bool write_result = module_charge::write_rhog(tmpfile, rhopw.gamma_only, &rhopw, 1, latvec(), rhog.data(), pw_world, nullptr); + ASSERT_TRUE(write_result); + + // Flip gamma_only and shrink npwtot to trigger the warning branches. + rhopw.gamma_only = true; + rhopw.npwtot -= 1; + + open_warning("test_read_rhog.txt"); + bool result = module_charge::read_rhog(tmpfile, &rhopw, 2, rhog.data(), pw_world, &warning_stream); + close_warning(); + + std::string expected_content + = " module_charge::read_rhog warning : some planewaves in file are not used\n module_charge::read_rhog warning : some " + "spin channels in file are missing\n module_charge::read_rhog warning : gamma_only read from file is " + "inconsistent with INPUT\n"; + + EXPECT_FALSE(result); + EXPECT_EQ(read_warning_file("test_read_rhog.txt"), expected_content); + std::remove(tmpfile.c_str()); + std::remove("test_read_rhog.txt"); +} + +// Test the read_rhog function when some planewaves in file are missing +TEST_F(ReadRhogTest, SomePWMissing) +{ + setup_pw_basis(); + fill_rhog(1); + + const std::string tmpfile = "test_rhog_missing.dat"; + bool write_result = module_charge::write_rhog(tmpfile, rhopw.gamma_only, &rhopw, 1, latvec(), rhog.data(), pw_world, nullptr); + ASSERT_TRUE(write_result); + + // Pretend the basis holds more planewaves than the file. + rhopw.npwtot += 1; + + open_warning("test_read_rhog.txt"); + bool result = module_charge::read_rhog(tmpfile, &rhopw, 1, rhog.data(), pw_world, &warning_stream); + close_warning(); + + std::string expected_content = " module_charge::read_rhog warning : some planewaves in file are missing\n"; + EXPECT_TRUE(result); + EXPECT_EQ(read_warning_file("test_read_rhog.txt"), expected_content); + std::remove(tmpfile.c_str()); + std::remove("test_read_rhog.txt"); +} + +// Test read_rhog with os_warning=nullptr (silent mode, must not crash) +TEST_F(ReadRhogTest, OsNullptrSilent) +{ + std::string filename = "notfound.txt"; + bool result = module_charge::read_rhog(filename, &rhopw, 1, rhog.data(), pw_world, nullptr); + EXPECT_FALSE(result); +} + +// Test write_rhog round-trip: write then read back, verify data consistency +TEST_F(ReadRhogTest, WriteRoundTrip) +{ + setup_pw_basis(); + fill_rhog(1); + + std::string tmpfile = "test_rhog_roundtrip.dat"; + + bool write_result = module_charge::write_rhog(tmpfile, rhopw.gamma_only, &rhopw, 1, latvec(), rhog.data(), pw_world, nullptr); + EXPECT_TRUE(write_result); + + std::vector>> rhog_read_data(1, std::vector>(rhopw.npw)); + std::vector*> rhog_read; + rhog_read.push_back(rhog_read_data[0].data()); + + bool read_result = module_charge::read_rhog(tmpfile, &rhopw, 1, rhog_read.data(), pw_world, nullptr); + EXPECT_TRUE(read_result); + + int diff_count = 0; + for (int ig = 0; ig < rhopw.npw; ++ig) + { + if (std::abs(rhog[0][ig] - rhog_read[0][ig]) > 1e-10) + { + ++diff_count; + } + } + EXPECT_EQ(diff_count, 0) << diff_count << " planewave values differ after round-trip"; + + std::remove(tmpfile.c_str()); +} + +// Test write_rhog when the output path is not writable +TEST_F(ReadRhogTest, WriteFileFail) +{ + setup_pw_basis(); + fill_rhog(1); + + // try to write to a directory path (not a file) — should fail + bool result = module_charge::write_rhog("/tmp", rhopw.gamma_only, &rhopw, 1, latvec(), rhog.data(), pw_world, nullptr); + EXPECT_FALSE(result); +} + +// Test write_rhog with nspin=2, round-trip both channels +TEST_F(ReadRhogTest, WriteRoundTripNspin2) +{ + setup_pw_basis(); + fill_rhog(2); + + std::string tmpfile = "test_rhog_roundtrip_nspin2.dat"; + + bool write_result = module_charge::write_rhog(tmpfile, rhopw.gamma_only, &rhopw, 2, latvec(), rhog.data(), pw_world, nullptr); + EXPECT_TRUE(write_result); + + std::vector>> rhog_read_data(2, std::vector>(rhopw.npw)); + std::vector*> rhog_read; + rhog_read.push_back(rhog_read_data[0].data()); + rhog_read.push_back(rhog_read_data[1].data()); + + bool read_result = module_charge::read_rhog(tmpfile, &rhopw, 2, rhog_read.data(), pw_world, nullptr); + EXPECT_TRUE(read_result); + + int diff_count = 0; + for (int is = 0; is < 2; ++is) + { + for (int ig = 0; ig < rhopw.npw; ++ig) + { + if (std::abs(rhog[is][ig] - rhog_read[is][ig]) > 1e-10) + { + ++diff_count; + } + } + } + EXPECT_EQ(diff_count, 0) << diff_count << " planewave values differ after nspin=2 round-trip"; + + std::remove(tmpfile.c_str()); +} + +// Test write_rhog with nspin=4, round-trip all 4 channels +TEST_F(ReadRhogTest, WriteRoundTripNspin4) +{ + setup_pw_basis(); + fill_rhog(4); + + std::string tmpfile = "test_rhog_roundtrip_nspin4.dat"; + + bool write_result = module_charge::write_rhog(tmpfile, rhopw.gamma_only, &rhopw, 4, latvec(), rhog.data(), pw_world, nullptr); + EXPECT_TRUE(write_result); + + std::vector>> rhog_read_data(4, std::vector>(rhopw.npw)); + std::vector*> rhog_read; + for (int is = 0; is < 4; ++is) + { + rhog_read.push_back(rhog_read_data[is].data()); + } + + bool read_result = module_charge::read_rhog(tmpfile, &rhopw, 4, rhog_read.data(), pw_world, nullptr); + EXPECT_TRUE(read_result); + + int diff_count = 0; + for (int is = 0; is < 4; ++is) + { + for (int ig = 0; ig < rhopw.npw; ++ig) + { + if (std::abs(rhog[is][ig] - rhog_read[is][ig]) > 1e-10) + { + ++diff_count; + } + } + } + EXPECT_EQ(diff_count, 0) << diff_count << " planewave values differ after nspin=4 round-trip"; + + std::remove(tmpfile.c_str()); +} + +// Test the special path L173-181: file nspin=2 read as input nspin=4 +// Expected behavior: rhog[0] preserved, rhog[1] and rhog[2] zeroed, +// rhog[3] <- old rhog[1] +TEST_F(ReadRhogTest, ReadRhogNspin2To4SpecialPath) +{ + setup_pw_basis(); + fill_rhog(2); + + // Override with simple, distinct values for this scenario. + for (int ig = 0; ig < rhopw.npw; ++ig) + { + rhog_data[0][ig] = std::complex(10.0 + ig, 0.0); + rhog_data[1][ig] = std::complex(20.0 + ig, 0.0); + } + + std::string tmpfile = "test_rhog_nspin2_to_4.dat"; + + bool write_result = module_charge::write_rhog(tmpfile, rhopw.gamma_only, &rhopw, 2, latvec(), rhog.data(), pw_world, nullptr); + EXPECT_TRUE(write_result); + + std::vector>> rhog_read_data(4, std::vector>(rhopw.npw)); + std::vector*> rhog_read; + for (int is = 0; is < 4; ++is) + { + rhog_read.push_back(rhog_read_data[is].data()); + } + + bool read_result = module_charge::read_rhog(tmpfile, &rhopw, 4, rhog_read.data(), pw_world, nullptr); + EXPECT_TRUE(read_result); + + for (int ig = 0; ig < rhopw.npw; ++ig) + { + EXPECT_NEAR(rhog_read_data[0][ig].real(), 10.0 + ig, 1e-10); + EXPECT_NEAR(rhog_read_data[0][ig].imag(), 0.0, 1e-10); + EXPECT_NEAR(rhog_read_data[1][ig].real(), 0.0, 1e-10); + EXPECT_NEAR(rhog_read_data[1][ig].imag(), 0.0, 1e-10); + EXPECT_NEAR(rhog_read_data[2][ig].real(), 0.0, 1e-10); + EXPECT_NEAR(rhog_read_data[2][ig].imag(), 0.0, 1e-10); + EXPECT_NEAR(rhog_read_data[3][ig].real(), 20.0 + ig, 1e-10); + EXPECT_NEAR(rhog_read_data[3][ig].imag(), 0.0, 1e-10); + } + + std::remove(tmpfile.c_str()); +} + +int main(int argc, char** argv) +{ +#ifdef __MPI + int nproc = 1; + int myrank = 0; + int nproc_in_pool = 1; + int kpar = 1; + int mypool = 0; + int rank_in_pool = 0; + setupmpi(argc, argv, nproc, myrank); + divide_pools(nproc, myrank, nproc_in_pool, kpar, mypool, rank_in_pool); +#endif + + testing::InitGoogleTest(&argc, argv); + int result = RUN_ALL_TESTS(); + +#ifdef __MPI + finishmpi(); +#endif + return result; +} diff --git a/source/source_estate/module_charge/unittests/test_chg_routine.cpp b/source/source_estate/module_charge/unittests/test_chg_routine.cpp new file mode 100644 index 00000000000..8755d6329b6 --- /dev/null +++ b/source/source_estate/module_charge/unittests/test_chg_routine.cpp @@ -0,0 +1,157 @@ +#include "gtest/gtest.h" + +#include "source_base/matrix3.h" +#include "source_basis/module_pw/pw_basis.h" +#include "source_cell/unitcell.h" +#include "source_estate/elecstate.h" +#include "source_estate/module_charge/charge.h" +#include "source_estate/module_charge/chg_mix.h" +#include "source_estate/module_charge/chg_routine.h" +#include "source_pw/module_pwdft/dftu_base.h" + +#include +#include + +// charge.cpp references Magnetism; provide a lightweight stub. +Magnetism::Magnetism() +{ + this->tot_mag = 0.0; + this->abs_mag = 0.0; +} +Magnetism::~Magnetism() +{ +} + +/************************************************ + * unit test of module_charge/chg_routine.cpp + ***********************************************/ + +/** + * - Tested Functions: + * - chgmixing_ks_pw: iter==1 init path sets mixing_restart_step. + * - chgmixing_ks_lcao: iter==1 mix_reset path sets mixing_restart_step. + * - chgmixing_ks: convergence branches (conv_esolver true skips mixing; + * drho < hsolver_error skips mixing). + */ + +namespace +{ + +MixingConfig make_plain_cfg(int nspin) +{ + return {"plain", 0.5, 8, 0.0, false, 0.5, 0.0, 0.0, -1.0, + false, nspin, 2, false, false, false, false, 100}; +} + +} // namespace + +class ChgRoutineTest : public ::testing::Test +{ + protected: + ModulePW::PW_Basis pw_basis; + Charge charge; + UnitCell ucell; + + void SetUp() override + { + pw_basis.initgrids(4, ModuleBase::Matrix3(1, 0, 0, 0, 1, 0, 0, 0, 1), 20); + pw_basis.initparameters(false, 20); + pw_basis.setuptransform(); + pw_basis.collect_local_pw(); + charge.set_rhopw(&pw_basis); + charge.allocate(1, false, false, 0); + ucell.omega = pw_basis.omega; + ucell.tpiba = 1.0; + } +}; + +TEST_F(ChgRoutineTest, ChgmixingKsPwIter1SetsRestartStep) +{ + Charge_Mixing cm; + cm.set_mixing(make_plain_cfg(1), &pw_basis, &pw_basis, ucell.omega, ucell.tpiba); + Plus_U_Base dftu; + Input_para inp; + inp.scf_nmax = 50; + inp.mixing_restart = 0.0; + inp.dft_plus_u = false; + + module_charge::chgmixing_ks_pw(1, &cm, dftu, true, inp); + + EXPECT_EQ(cm.mixing_restart_step, inp.scf_nmax + 1); +} + +TEST_F(ChgRoutineTest, ChgmixingKsLcaoIter1SetsRestartStep) +{ + Charge_Mixing cm; + cm.set_mixing(make_plain_cfg(1), &pw_basis, &pw_basis, ucell.omega, ucell.tpiba); + Plus_U_Base dftu; + Input_para inp; + inp.scf_nmax = 50; + inp.mixing_restart = 0.0; + inp.dft_plus_u = false; + + module_charge::chgmixing_ks_lcao(1, &cm, dftu, 0, inp); + + EXPECT_EQ(cm.mixing_restart_step, inp.scf_nmax + 1); +} + +TEST_F(ChgRoutineTest, ChgmixingKsConvergedSkipsMixing) +{ + Charge_Mixing cm; + cm.set_mixing(make_plain_cfg(1), &pw_basis, &pw_basis, ucell.omega, ucell.tpiba); + Input_para inp; + inp.mixing_restart = 0.0; + inp.scf_os_stop = false; + inp.scf_thr_type = 2; + inp.calculation = "scf"; + inp.nelec = 1.0; + + module_charge::ScfMixingCtx ctx; + ctx.hsolver_error = 1e-6; + ctx.scf_thr = 1e-4; + ctx.scf_ene_thr = 0.0; + ctx.converged_u = true; + ctx.ks_run = true; + ctx.drho = 1e-6; // < scf_thr => converged + + std::vector rho_before(charge.rho[0], charge.rho[0] + pw_basis.nrxx); + + module_charge::chgmixing_ks(2, ucell, nullptr, charge, pw_basis, &cm, ctx, inp); + + EXPECT_TRUE(ctx.conv_esolver); + // rho must be unchanged because conv_esolver is true + for (int ir = 0; ir < pw_basis.nrxx; ++ir) + { + EXPECT_EQ(charge.rho[0][ir], rho_before[ir]); + } +} + +TEST_F(ChgRoutineTest, ChgmixingKsDrhoBelowHsolverSkipsMixing) +{ + Charge_Mixing cm; + cm.set_mixing(make_plain_cfg(1), &pw_basis, &pw_basis, ucell.omega, ucell.tpiba); + Input_para inp; + inp.mixing_restart = 0.0; + inp.scf_os_stop = false; + inp.scf_thr_type = 2; + inp.calculation = "scf"; + inp.nelec = 1.0; + + module_charge::ScfMixingCtx ctx; + ctx.hsolver_error = 1e-3; + ctx.scf_thr = 1e-4; + ctx.scf_ene_thr = 0.0; + ctx.converged_u = true; + ctx.ks_run = true; + ctx.drho = 1e-5; // < hsolver_error + + std::vector rho_before(charge.rho[0], charge.rho[0] + pw_basis.nrxx); + + module_charge::chgmixing_ks(2, ucell, nullptr, charge, pw_basis, &cm, ctx, inp); + + // rho unchanged because drho < hsolver_error + for (int ir = 0; ir < pw_basis.nrxx; ++ir) + { + EXPECT_EQ(charge.rho[0][ir], rho_before[ir]); + } +} diff --git a/source/source_estate/module_charge/unittests/test_chg_symm.cpp b/source/source_estate/module_charge/unittests/test_chg_symm.cpp new file mode 100644 index 00000000000..5724a95a811 --- /dev/null +++ b/source/source_estate/module_charge/unittests/test_chg_symm.cpp @@ -0,0 +1,157 @@ +#include "gtest/gtest.h" + +#include "source_base/matrix3.h" +#include "source_basis/module_pw/pw_basis.h" +#include "source_cell/module_symmetry/symmetry.h" +#include "source_cell/magnetism.h" +#include "source_estate/module_charge/charge.h" +#include "source_estate/module_charge/chg_symm.h" + +#include +#include + +// charge.cpp references Magnetism; provide a lightweight stub. +Magnetism::Magnetism() +{ + this->tot_mag = 0.0; + this->abs_mag = 0.0; +} +Magnetism::~Magnetism() +{ +} + +/************************************************ + * unit test of module_charge/chg_symm.cpp + ***********************************************/ + +/** + * - Tested Functions: + * - symmetrize_rho: dispatch per nspin to cal_rhog_symm / cal_rhog_symm_soc + * - cal_rhog_symm: no-op when symm_flag != 1; otherwise FFT + psymmg + FFT back + * - cal_rhog_symm (raw array overload): same no-op behavior + * - cal_rhog_symm_soc: no-op when symm_flag != 1 + * + * The no-op paths (symm_flag == 0) are fully covered without a real symmetry + * group: the functions return immediately, leaving rho/rhog unchanged. + */ + +class ChgSymmTest : public ::testing::Test +{ + protected: + ModulePW::PW_Basis pw_basis; + Charge charge; + + void SetUp() override + { + pw_basis.initgrids(4, ModuleBase::Matrix3(1, 0, 0, 0, 1, 0, 0, 0, 1), 20); + pw_basis.initparameters(false, 20); + pw_basis.setuptransform(); + pw_basis.collect_local_pw(); + } + + void setup_charge(int nspin) + { + charge.set_rhopw(&pw_basis); + const bool kin_den = false; + const bool meta_gga = false; + charge.allocate(nspin, kin_den, meta_gga, 0); + } +}; + +// --------------------------------------------------------------------------- +// no-op path: symm_flag == 0 leaves density untouched +// --------------------------------------------------------------------------- + +TEST_F(ChgSymmTest, SymmetrizeRhoSymmFlagOffIsNoopNspin1) +{ + const int nspin = 1; + setup_charge(nspin); + ModuleSymmetry::Symmetry symm; + ModuleSymmetry::Symmetry::symm_flag = 0; + + for (int ir = 0; ir < pw_basis.nrxx; ++ir) + { + charge.rho[0][ir] = static_cast(ir + 1); + } + std::vector rho_before(charge.rho[0], charge.rho[0] + pw_basis.nrxx); + + module_charge::symmetrize_rho(nspin, charge, &pw_basis, symm); + + for (int ir = 0; ir < pw_basis.nrxx; ++ir) + { + EXPECT_EQ(charge.rho[0][ir], rho_before[ir]); + } +} + +TEST_F(ChgSymmTest, SymmetrizeRhoSymmFlagOffIsNoopNspin4) +{ + const int nspin = 4; + setup_charge(nspin); + ModuleSymmetry::Symmetry symm; + ModuleSymmetry::Symmetry::symm_flag = 0; + + std::vector> rho_before(nspin); + for (int is = 0; is < nspin; ++is) + { + for (int ir = 0; ir < pw_basis.nrxx; ++ir) + { + charge.rho[is][ir] = static_cast(is * 100 + ir); + } + rho_before[is].assign(charge.rho[is], charge.rho[is] + pw_basis.nrxx); + } + + module_charge::symmetrize_rho(nspin, charge, &pw_basis, symm); + + for (int is = 0; is < nspin; ++is) + { + for (int ir = 0; ir < pw_basis.nrxx; ++ir) + { + EXPECT_EQ(charge.rho[is][ir], rho_before[is][ir]); + } + } +} + +TEST_F(ChgSymmTest, CalRhogSymmRawArrayNoop) +{ + const int nspin = 1; + setup_charge(nspin); + ModuleSymmetry::Symmetry symm; + ModuleSymmetry::Symmetry::symm_flag = 0; + + std::vector rho_buf(pw_basis.nrxx, 3.0); + std::vector> rhog_buf(pw_basis.npw, std::complex(0.0, 0.0)); + double* rho_ptrs[1] = {rho_buf.data()}; + std::complex* rhog_ptrs[1] = {rhog_buf.data()}; + + module_charge::cal_rhog_symm(0, rho_ptrs, rhog_ptrs, pw_basis.npw, nullptr, &pw_basis, symm); + + for (int ir = 0; ir < pw_basis.nrxx; ++ir) + { + EXPECT_EQ(rho_buf[ir], 3.0); + } +} + +TEST_F(ChgSymmTest, CalRhogSymmSocNoop) +{ + const int nspin = 4; + setup_charge(nspin); + ModuleSymmetry::Symmetry symm; + ModuleSymmetry::Symmetry::symm_flag = 0; + + std::vector> rho_buf(nspin, std::vector(pw_basis.nrxx, 1.0)); + std::vector>> rhog_buf( + nspin, std::vector>(pw_basis.npw, std::complex(0.0, 0.0))); + double* rho_ptrs[4] = {rho_buf[0].data(), rho_buf[1].data(), rho_buf[2].data(), rho_buf[3].data()}; + std::complex* rhog_ptrs[4] = {rhog_buf[0].data(), rhog_buf[1].data(), + rhog_buf[2].data(), rhog_buf[3].data()}; + + module_charge::cal_rhog_symm_soc(rho_ptrs, rhog_ptrs, &pw_basis, symm); + + for (int is = 0; is < nspin; ++is) + { + for (int ir = 0; ir < pw_basis.nrxx; ++ir) + { + EXPECT_EQ(rho_buf[is][ir], 1.0); + } + } +} diff --git a/source/source_estate/module_charge/unittests/test_chg_symm_detail.cpp b/source/source_estate/module_charge/unittests/test_chg_symm_detail.cpp new file mode 100644 index 00000000000..ce3061990d6 --- /dev/null +++ b/source/source_estate/module_charge/unittests/test_chg_symm_detail.cpp @@ -0,0 +1,151 @@ +#include "gtest/gtest.h" + +#include "source_base/matrix3.h" +#include "source_base/vector3.h" +#include "source_basis/module_pw/pw_basis.h" +#include "source_cell/magnetism.h" +#include "source_cell/module_symmetry/symmetry.h" +#include "source_cell/module_symmetry/symm_rot_spin.h" +#include "source_estate/module_charge/chg_symm_detail.h" + +#include +#include +#include + +// unitcell.cpp (pulled in via the cell_info object library) references +// Magnetism; provide a lightweight stub, mirroring test_chg_symm.cpp. +Magnetism::Magnetism() +{ + this->tot_mag = 0.0; + this->abs_mag = 0.0; +} +Magnetism::~Magnetism() +{ +} + +/************************************************ + * unit test of module_charge/chg_symm_detail.cpp + ***********************************************/ + +/** + * - Tested Functions: + * - psymmg: symmetrize one reciprocal-space density component + * - psymmg_soc: symmetrize three coupled spin components for nspin=4 + * + * Both are driven with a manually built D_4 point group on a cubic lattice + * (a=1), mirroring source_cell/module_symmetry/test/symm_rho_soc_test.cpp. + * The PW_Basis is serial (single plane-wave per FFT point) so the + * non-MPI path in psymmg/psymmg_soc is exercised. + * + * Checks: + * - Idempotence: symmetrizing twice gives the same result as once. + */ + +namespace +{ + +// the 8 proper rotations of D_4 (column-vector convention r' = R r) +const std::array, 3>, 8> kRcol = {{ + {{{ 1, 0, 0}, { 0, 1, 0}, { 0, 0, 1}}}, // E + {{{ 0,-1, 0}, { 1, 0, 0}, { 0, 0, 1}}}, // C4z + {{{-1, 0, 0}, { 0,-1, 0}, { 0, 0, 1}}}, // C2z + {{{ 0, 1, 0}, {-1, 0, 0}, { 0, 0, 1}}}, // C4z^3 + {{{ 1, 0, 0}, { 0,-1, 0}, { 0, 0,-1}}}, // C2x + {{{-1, 0, 0}, { 0, 1, 0}, { 0, 0,-1}}}, // C2y + {{{ 0, 1, 0}, { 1, 0, 0}, { 0, 0,-1}}}, // C2[110] + {{{ 0,-1, 0}, {-1, 0, 0}, { 0, 0,-1}}}, // C2[1-10] +}}; + +ModuleBase::Matrix3 gmatc_of(int g) +{ + const auto& r = kRcol[g]; + return ModuleBase::Matrix3(r[0][0], r[1][0], r[2][0], + r[0][1], r[1][1], r[2][1], + r[0][2], r[1][2], r[2][2]); +} + +/// Build a D_4 symmetry group on a cubic lattice (a=1). +void build_d4_group(ModuleSymmetry::Symmetry& symm) +{ + symm.epsilon = 1e-6; + symm.nrot = 8; + symm.nrotk = 8; + symm.nrotk_anti = 0; + symm.ncell = 1; + symm.ptrans = {ModuleBase::Vector3(0.0, 0.0, 0.0)}; + ModuleSymmetry::Symmetry::pricell_loop = false; + for (int g = 0; g < 8; ++g) + { + const ModuleBase::Matrix3 gc = gmatc_of(g); + symm.gmatrix[g] = gc; + symm.kgmatrix[g] = gc; + symm.gtrans[g] = ModuleBase::Vector3(0.0, 0.0, 0.0); + } +} + +} // namespace + +class ChgSymmDetailTest : public ::testing::Test +{ + protected: + ModulePW::PW_Basis pw_basis; + ModuleSymmetry::Symmetry symm; + + void SetUp() override + { + pw_basis.initgrids(1.0, ModuleBase::Matrix3(1, 0, 0, 0, 1, 0, 0, 0, 1), 10.0); + pw_basis.initparameters(false, 10.0); + pw_basis.setuptransform(); + pw_basis.collect_local_pw(); + build_d4_group(symm); + } +}; + +TEST_F(ChgSymmDetailTest, PsymmgIdempotence) +{ + std::vector> rhog(pw_basis.npw); + for (int ig = 0; ig < pw_basis.npw; ++ig) + { + rhog[ig] = std::complex(0.3 * ig - 1.0, 0.7 * ((ig * 13) % 5) - 1.5); + } + std::vector> once = rhog; + std::vector> twice = rhog; + + module_charge::detail::psymmg(once.data(), &pw_basis, symm); + module_charge::detail::psymmg(twice.data(), &pw_basis, symm); + module_charge::detail::psymmg(twice.data(), &pw_basis, symm); + + for (int ig = 0; ig < pw_basis.npw; ++ig) + { + EXPECT_NEAR(once[ig].real(), twice[ig].real(), 1e-8); + EXPECT_NEAR(once[ig].imag(), twice[ig].imag(), 1e-8); + } +} + +TEST_F(ChgSymmDetailTest, PsymmgSocIdempotence) +{ + std::vector> x(pw_basis.npw); + std::vector> y(pw_basis.npw); + std::vector> z(pw_basis.npw); + for (int ig = 0; ig < pw_basis.npw; ++ig) + { + x[ig] = std::complex(0.3 * ig - 1.0, 0.7 * ((ig * 13) % 5) - 1.5); + y[ig] = std::complex(-0.5 * ((ig * 7) % 4) + 0.9, 0.2 * ig - 2.0); + z[ig] = std::complex(0.11 * ((ig * 3) % 6), -0.4 * ((ig * 5) % 7) + 1.0); + } + std::vector> x1 = x, y1 = y, z1 = z; + + module_charge::detail::psymmg_soc(x.data(), y.data(), z.data(), &pw_basis, symm); + module_charge::detail::psymmg_soc(x1.data(), y1.data(), z1.data(), &pw_basis, symm); + module_charge::detail::psymmg_soc(x1.data(), y1.data(), z1.data(), &pw_basis, symm); + + for (int ig = 0; ig < pw_basis.npw; ++ig) + { + EXPECT_NEAR(x[ig].real(), x1[ig].real(), 1e-8); + EXPECT_NEAR(x[ig].imag(), x1[ig].imag(), 1e-8); + EXPECT_NEAR(y[ig].real(), y1[ig].real(), 1e-8); + EXPECT_NEAR(y[ig].imag(), y1[ig].imag(), 1e-8); + EXPECT_NEAR(z[ig].real(), z1[ig].real(), 1e-8); + EXPECT_NEAR(z[ig].imag(), z1[ig].imag(), 1e-8); + } +} diff --git a/source/source_estate/module_charge/unittests/test_chg_tau.cpp b/source/source_estate/module_charge/unittests/test_chg_tau.cpp new file mode 100644 index 00000000000..0fcb626a1ad --- /dev/null +++ b/source/source_estate/module_charge/unittests/test_chg_tau.cpp @@ -0,0 +1,132 @@ +#include "gtest/gtest.h" + +#include "source_base/matrix3.h" +#include "source_basis/module_pw/pw_basis.h" +#include "source_base/module_mixing/mixing.h" +#include "source_base/module_mixing/plain_mixing.h" +#include "source_cell/magnetism.h" +#include "source_estate/module_charge/charge.h" +#include "source_estate/module_charge/chg_tau.h" + +#include +#include + +// charge.cpp references Magnetism; provide a lightweight stub. +Magnetism::Magnetism() +{ + this->tot_mag = 0.0; + this->abs_mag = 0.0; +} +Magnetism::~Magnetism() +{ +} + +/************************************************ + * unit test of module_charge/chg_tau.cpp + ***********************************************/ + +/** + * - Tested Functions: + * - mix_tau_recip: mixes the kinetic-energy density in reciprocal space. + * Covered: + * - null pointer abort paths (chr, rhopw, rhodpw, mixing). + * - nspin < 1 abort. + * - double_grid with null mixing_highf abort. + * - non-double-grid plain mixing value for nspin=1. + */ + +class ChgTauTest : public ::testing::Test +{ + protected: + ModulePW::PW_Basis pw_basis; + Charge charge; + + void SetUp() override + { + pw_basis.initgrids(4, ModuleBase::Matrix3(1, 0, 0, 0, 1, 0, 0, 0, 1), 20); + pw_basis.initparameters(false, 20); + pw_basis.setuptransform(); + pw_basis.collect_local_pw(); + } + + void setup_charge(int nspin) + { + charge.set_rhopw(&pw_basis); + const bool kin_den = true; + const bool meta_gga = false; + charge.allocate(nspin, kin_den, meta_gga, 0); + } +}; + +TEST_F(ChgTauTest, NullChrAborts) +{ + Base_Mixing::Plain_Mixing mixing; + Base_Mixing::Mixing_Data mdata; + EXPECT_DEATH(module_charge::detail::mix_tau_recip( + nullptr, 1, false, &pw_basis, &pw_basis, &mixing, mdata, nullptr), + ""); +} + +TEST_F(ChgTauTest, NullGridAborts) +{ + setup_charge(1); + Base_Mixing::Plain_Mixing mixing; + Base_Mixing::Mixing_Data mdata; + EXPECT_DEATH(module_charge::detail::mix_tau_recip( + &charge, 1, false, nullptr, &pw_basis, &mixing, mdata, nullptr), + ""); +} + +TEST_F(ChgTauTest, NullMixingAborts) +{ + setup_charge(1); + Base_Mixing::Mixing_Data mdata; + EXPECT_DEATH(module_charge::detail::mix_tau_recip( + &charge, 1, false, &pw_basis, &pw_basis, nullptr, mdata, nullptr), + ""); +} + +TEST_F(ChgTauTest, BadNspinAborts) +{ + setup_charge(1); + Base_Mixing::Plain_Mixing mixing; + Base_Mixing::Mixing_Data mdata; + EXPECT_DEATH(module_charge::detail::mix_tau_recip( + &charge, 0, false, &pw_basis, &pw_basis, &mixing, mdata, nullptr), + ""); +} + +TEST_F(ChgTauTest, DoubleGridWithoutHighfAborts) +{ + setup_charge(1); + Base_Mixing::Plain_Mixing mixing; + Base_Mixing::Mixing_Data mdata; + EXPECT_DEATH(module_charge::detail::mix_tau_recip( + &charge, 1, true, &pw_basis, &pw_basis, &mixing, mdata, nullptr), + ""); +} + +TEST_F(ChgTauTest, NonDoubleGridPlainMixingValue) +{ + const int nspin = 1; + setup_charge(nspin); + Base_Mixing::Plain_Mixing mixing(0.5); + Base_Mixing::Mixing_Data mdata; + mixing.init_mixing_data(mdata, pw_basis.npw, sizeof(std::complex)); + + // uniform kinetic densities + for (int ir = 0; ir < pw_basis.nrxx; ++ir) + { + charge.kin_r_save[0][ir] = 2.0; + charge.kin_r[0][ir] = 3.0; + } + + module_charge::detail::mix_tau_recip( + &charge, nspin, false, &pw_basis, &pw_basis, &mixing, mdata, nullptr); + + // after plain mixing: out = in + beta * (out - in) = 2 + 0.5*(3-2) = 2.5 + for (int ir = 0; ir < pw_basis.nrxx; ++ir) + { + EXPECT_NEAR(charge.kin_r[0][ir], 2.5, 1e-6); + } +} diff --git a/source/source_estate/module_charge/unittests/test_chg_tools.cpp b/source/source_estate/module_charge/unittests/test_chg_tools.cpp new file mode 100644 index 00000000000..7ad60d44ba5 --- /dev/null +++ b/source/source_estate/module_charge/unittests/test_chg_tools.cpp @@ -0,0 +1,108 @@ +#include "gtest/gtest.h" + +#include "source_cell/unitcell.h" +#include "source_estate/module_charge/chg_tools.h" + +#include +#include + +// chg_tools.cpp references UnitCell (set_rho_core), so the test binary links +// the cell_info objects whose unitcell.cpp needs Magnetism symbols. Provide +// the same lightweight mocks as test_charge.cpp. +Magnetism::Magnetism() +{ + this->tot_mag = 0.0; + this->abs_mag = 0.0; +} +Magnetism::~Magnetism() +{ +} + +/************************************************ + * unit test of module_charge/chg_tools.cpp + ***********************************************/ + +/** + * - Tested Functions: + * - cal_rho2ne: integrate a single spin channel over the grid and scale + * by omega / nxyz to obtain the electron number + * - check_rho: nspin == 1/4 total-density check, nspin == 2 spin-up/down + * checks, mismatch warning path and negative-channel abort path + */ + +class ChgToolsTest : public ::testing::Test +{ + protected: + const int nrxx = 8; ///< local real-space grid points + const int nxyz = 8; ///< global real-space grid points + const double omega = 2.0; ///< cell volume + const double nelec = 4.0; ///< target electron number + + std::vector rho_up; + std::vector rho_dn; + std::vector rho; + + void SetUp() override + { + // uniform value 2.0 integrates to 8 * 2.0 * 2.0 / 8 = 4.0 electrons + rho_up.assign(nrxx, 2.0); + rho_dn.assign(nrxx, 2.0); + rho.resize(2); + rho[0] = rho_up.data(); + rho[1] = rho_dn.data(); + } +}; + +TEST_F(ChgToolsTest, CalRho2ne) +{ + EXPECT_NEAR(module_charge::cal_rho2ne(rho[0], nrxx, omega, nxyz), nelec, 1e-12); +} + +TEST_F(ChgToolsTest, CheckRhoNonSpinMatched) +{ + module_charge::check_rho(rho.data(), 1, nrxx, omega, nxyz, nelec); +} + +TEST_F(ChgToolsTest, CheckRhoSocTreatedAsTotal) +{ + module_charge::check_rho(rho.data(), 4, nrxx, omega, nxyz, nelec); +} + +TEST_F(ChgToolsTest, CheckRhoNonSpinMismatchWarns) +{ + // total 4.0 differs from the target 4.5: a warning is emitted but the + // call returns normally + module_charge::check_rho(rho.data(), 1, nrxx, omega, nxyz, 4.5); +} + +TEST_F(ChgToolsTest, CheckRhoSpin2Matched) +{ + // 2.0 spin-up + 2.0 spin-down electrons + module_charge::check_rho(rho.data(), 2, nrxx, omega, nxyz, nelec); +} + +TEST_F(ChgToolsTest, CheckRhoSpin2MismatchWarns) +{ + // spin-down integrates to 1.0 electron, total 3.0 vs target 4.0 + std::fill(rho_dn.begin(), rho_dn.end(), 0.5); + module_charge::check_rho(rho.data(), 2, nrxx, omega, nxyz, nelec); +} + +TEST_F(ChgToolsTest, CheckRhoNegativeSpinUpAborts) +{ + std::fill(rho_up.begin(), rho_up.end(), -0.5); + EXPECT_DEATH(module_charge::check_rho(rho.data(), 2, nrxx, omega, nxyz, nelec), ""); +} + +TEST_F(ChgToolsTest, CheckRhoNegativeSpinDownAborts) +{ + std::fill(rho_dn.begin(), rho_dn.end(), -0.5); + EXPECT_DEATH(module_charge::check_rho(rho.data(), 2, nrxx, omega, nxyz, nelec), ""); +} + +TEST_F(ChgToolsTest, CheckRhoUnsupportedNspinNoop) +{ + // nspin values other than 1/2/4 are silently skipped, as in the + // original Charge::check_rho + module_charge::check_rho(rho.data(), 3, nrxx, omega, nxyz, nelec); +} diff --git a/source/source_estate/module_charge/unittests/test_chg_uspp.cpp b/source/source_estate/module_charge/unittests/test_chg_uspp.cpp new file mode 100644 index 00000000000..27ffd3194fc --- /dev/null +++ b/source/source_estate/module_charge/unittests/test_chg_uspp.cpp @@ -0,0 +1,235 @@ +#include "gtest/gtest.h" + +#include "source_estate/module_charge/chg_uspp.h" + +#include +#include + +/************************************************ + * unit test of module_charge/chg_uspp.cpp + ***********************************************/ + +/** + * - Tested Functions: + * - split_dgrid: split dense reciprocal data into smooth and + * high-frequency parts on the USPP double grid + * - normal split with nspin=1 and nspin=2 + * - boundary: npw_smooth == 0 (all high-frequency) + * - boundary: npw_dense == npw_smooth (no high-frequency) + * - multi-spin channel isolation + * - abort on invalid inputs (null pointer, bad nspin/npw, size mismatch) + * - merge_dgrid: merge smooth and high-frequency parts back into dense + * - round-trip with split_dgrid reproduces the original data + * - abort on invalid inputs + */ + +class ChgUsppTest : public ::testing::Test +{ + protected: + // build a dense buffer of shape [nspin * npw_dense] with distinct + // per-element values so split/merge correctness is easy to verify + static std::vector> make_dense(int nspin, int npw_dense) + { + std::vector> buf(nspin * npw_dense); + for (int is = 0; is < nspin; ++is) + { + for (int ig = 0; ig < npw_dense; ++ig) + { + const double v = static_cast(is * 1000 + ig); + buf[is * npw_dense + ig] = std::complex(v, v + 0.5); + } + } + return buf; + } +}; + +TEST_F(ChgUsppTest, SplitDgridNormalNspin1) +{ + const int nspin = 1; + const int npw_smooth = 3; + const int npw_dense = 5; + auto data_d = make_dense(nspin, npw_dense); + + std::vector> data_s(nspin * npw_smooth); + std::vector> data_hf(nspin * (npw_dense - npw_smooth)); + + module_charge::split_dgrid(data_d.data(), data_s, data_hf, nspin, npw_smooth, npw_dense); + + // smooth part == first npw_smooth entries + for (int ig = 0; ig < npw_smooth; ++ig) + { + EXPECT_EQ(data_s[ig], data_d[ig]); + } + // high-frequency part == remaining entries + for (int ig = 0; ig < npw_dense - npw_smooth; ++ig) + { + EXPECT_EQ(data_hf[ig], data_d[npw_smooth + ig]); + } +} + +TEST_F(ChgUsppTest, SplitDgridNormalNspin2) +{ + const int nspin = 2; + const int npw_smooth = 2; + const int npw_dense = 4; + auto data_d = make_dense(nspin, npw_dense); + + std::vector> data_s(nspin * npw_smooth); + std::vector> data_hf(nspin * (npw_dense - npw_smooth)); + + module_charge::split_dgrid(data_d.data(), data_s, data_hf, nspin, npw_smooth, npw_dense); + + // each spin channel is split independently + for (int is = 0; is < nspin; ++is) + { + for (int ig = 0; ig < npw_smooth; ++ig) + { + EXPECT_EQ(data_s[is * npw_smooth + ig], data_d[is * npw_dense + ig]); + } + for (int ig = 0; ig < npw_dense - npw_smooth; ++ig) + { + EXPECT_EQ(data_hf[is * (npw_dense - npw_smooth) + ig], + data_d[is * npw_dense + npw_smooth + ig]); + } + } +} + +TEST_F(ChgUsppTest, SplitDgridSmoothIsZero) +{ + // npw_smooth == 0: the whole dense buffer is high-frequency + const int nspin = 1; + const int npw_smooth = 0; + const int npw_dense = 3; + auto data_d = make_dense(nspin, npw_dense); + + std::vector> data_s(0); + std::vector> data_hf(nspin * npw_dense); + + module_charge::split_dgrid(data_d.data(), data_s, data_hf, nspin, npw_smooth, npw_dense); + + EXPECT_TRUE(data_s.empty()); + for (int ig = 0; ig < npw_dense; ++ig) + { + EXPECT_EQ(data_hf[ig], data_d[ig]); + } +} + +TEST_F(ChgUsppTest, SplitDgridDenseEqualsSmooth) +{ + // npw_dense == npw_smooth: no high-frequency tail, data_hf is empty + const int nspin = 2; + const int npw_smooth = 3; + const int npw_dense = 3; + auto data_d = make_dense(nspin, npw_dense); + + std::vector> data_s(nspin * npw_smooth); + std::vector> data_hf(0); + + module_charge::split_dgrid(data_d.data(), data_s, data_hf, nspin, npw_smooth, npw_dense); + + EXPECT_TRUE(data_hf.empty()); + for (int i = 0; i < nspin * npw_smooth; ++i) + { + EXPECT_EQ(data_s[i], data_d[i]); + } +} + +TEST_F(ChgUsppTest, MergeDgridRoundTripNspin1) +{ + const int nspin = 1; + const int npw_smooth = 3; + const int npw_dense = 5; + auto data_d = make_dense(nspin, npw_dense); + + std::vector> data_s(nspin * npw_smooth); + std::vector> data_hf(nspin * (npw_dense - npw_smooth)); + module_charge::split_dgrid(data_d.data(), data_s, data_hf, nspin, npw_smooth, npw_dense); + + std::vector> merged(nspin * npw_dense); + module_charge::merge_dgrid(merged.data(), data_s, data_hf, nspin, npw_smooth, npw_dense); + + for (int i = 0; i < nspin * npw_dense; ++i) + { + EXPECT_EQ(merged[i], data_d[i]); + } +} + +TEST_F(ChgUsppTest, MergeDgridRoundTripNspin2) +{ + const int nspin = 2; + const int npw_smooth = 2; + const int npw_dense = 5; + auto data_d = make_dense(nspin, npw_dense); + + std::vector> data_s(nspin * npw_smooth); + std::vector> data_hf(nspin * (npw_dense - npw_smooth)); + module_charge::split_dgrid(data_d.data(), data_s, data_hf, nspin, npw_smooth, npw_dense); + + std::vector> merged(nspin * npw_dense); + module_charge::merge_dgrid(merged.data(), data_s, data_hf, nspin, npw_smooth, npw_dense); + + for (int i = 0; i < nspin * npw_dense; ++i) + { + EXPECT_EQ(merged[i], data_d[i]); + } +} + +TEST_F(ChgUsppTest, SplitDgridNullDataAborts) +{ + const int nspin = 1; + const int npw_smooth = 2; + const int npw_dense = 4; + std::vector> data_s(nspin * npw_smooth); + std::vector> data_hf(nspin * (npw_dense - npw_smooth)); + EXPECT_DEATH(module_charge::split_dgrid(nullptr, data_s, data_hf, nspin, npw_smooth, npw_dense), + ""); +} + +TEST_F(ChgUsppTest, SplitDgridBadNspinAborts) +{ + const int nspin = 0; + const int npw_smooth = 2; + const int npw_dense = 4; + auto data_d = make_dense(1, npw_dense); + std::vector> data_s(npw_smooth); + std::vector> data_hf(npw_dense - npw_smooth); + EXPECT_DEATH(module_charge::split_dgrid(data_d.data(), data_s, data_hf, nspin, npw_smooth, npw_dense), + ""); +} + +TEST_F(ChgUsppTest, SplitDgridBadNpwAborts) +{ + const int nspin = 1; + // npw_dense < npw_smooth is invalid + const int npw_smooth = 5; + const int npw_dense = 3; + auto data_d = make_dense(nspin, npw_dense); + std::vector> data_s(nspin * npw_smooth); + std::vector> data_hf(nspin * (npw_smooth - npw_dense)); + EXPECT_DEATH(module_charge::split_dgrid(data_d.data(), data_s, data_hf, nspin, npw_smooth, npw_dense), + ""); +} + +TEST_F(ChgUsppTest, SplitDgridSizeMismatchAborts) +{ + const int nspin = 1; + const int npw_smooth = 2; + const int npw_dense = 4; + auto data_d = make_dense(nspin, npw_dense); + // data_s is too small + std::vector> data_s(1); + std::vector> data_hf(npw_dense - npw_smooth); + EXPECT_DEATH(module_charge::split_dgrid(data_d.data(), data_s, data_hf, nspin, npw_smooth, npw_dense), + ""); +} + +TEST_F(ChgUsppTest, MergeDgridNullDataAborts) +{ + const int nspin = 1; + const int npw_smooth = 2; + const int npw_dense = 4; + std::vector> data_s(nspin * npw_smooth); + std::vector> data_hf(nspin * (npw_dense - npw_smooth)); + EXPECT_DEATH(module_charge::merge_dgrid(nullptr, data_s, data_hf, nspin, npw_smooth, npw_dense), + ""); +} diff --git a/source/source_estate/module_dm/CMakeLists.txt b/source/source_estate/module_dm/CMakeLists.txt index b166c61024d..78ab8a610a2 100644 --- a/source/source_estate/module_dm/CMakeLists.txt +++ b/source/source_estate/module_dm/CMakeLists.txt @@ -1,5 +1,3 @@ -if(BUILD_TESTING) - if(ENABLE_MPI) - add_subdirectory(test) - endif() -endif() \ No newline at end of file +# Unit tests for this module live in unittests/ and are wired from +# source/source_estate/CMakeLists.txt (same pattern as module_charge/unittests), +# so this module-level CMakeLists is intentionally left without test logic. diff --git a/source/source_estate/module_dm/cal_dm_psi.cpp b/source/source_estate/module_dm/cal_dm_psi.cpp deleted file mode 100644 index 717788a2a2c..00000000000 --- a/source/source_estate/module_dm/cal_dm_psi.cpp +++ /dev/null @@ -1,261 +0,0 @@ -#include "cal_dm_psi.h" - -#include "source_base/module_external/blas_connector.h" -#include "source_base/module_external/scalapack_connector.h" -#include "source_base/timer.h" -#include "source_psi/psi.h" - -namespace elecstate -{ - -// for Gamma-Only case where DMK is double -void cal_dm_psi(const Parallel_Orbitals* ParaV, - const ModuleBase::matrix& wg, - const psi::Psi& wfc, - elecstate::DensityMatrix& DM) -{ - ModuleBase::TITLE("elecstate", "cal_dm_psi"); - ModuleBase::timer::start("elecstate", "cal_dm_psi"); - - // dm.resize(wfc.get_nk(), ParaV->ncol, ParaV->nrow); - const int nbands_local = wfc.get_nbands(); - const int nbasis_local = wfc.get_nbasis(); - - // dm = wfc.T * wg * wfc.conj() - // dm[is](iw1,iw2) = \sum_{ib} wfc[is](ib,iw1).T * wg(is,ib) * wfc[is](ib,iw2).conj() - - for (int ik = 0; ik < wfc.get_nk(); ++ik) - { - double* dmk_pointer = DM.get_DMK_pointer(ik); - wfc.fix_k(ik); - // dm.fix_k(ik); - // dm[ik].create(ParaV->ncol, ParaV->nrow); - // wg_wfc(ib,iw) = wg[ib] * wfc(ib,iw); - - psi::Psi wg_wfc(1, wfc.get_nbands(), wfc.get_nbasis(), wfc.get_nbasis(), true); - - wg_wfc.set_all_psi(wfc.get_pointer(), wg_wfc.size()); - - int ib_global = 0; - for (int ib_local = 0; ib_local < nbands_local; ++ib_local) - { - while (ib_local != ParaV->global2local_col(ib_global)) - { - ++ib_global; - if (ib_global >= wg.nc) - { - break; - } - } - if (ib_global >= wg.nc) - { - continue; - } - const double wg_local = wg(ik, ib_global); - - double* wg_wfc_pointer = &(wg_wfc(0, ib_local, 0)); - BlasConnector::scal(nbasis_local, wg_local, wg_wfc_pointer, 1); - } - - // C++: dm(iw1,iw2) = wfc(ib,iw1).T * wg_wfc(ib,iw2) -#ifdef __MPI - psiMulPsiMpi(wg_wfc, wfc, dmk_pointer, ParaV->desc_wfc, ParaV->desc); -#else - psiMulPsi(wg_wfc, wfc, dmk_pointer); -#endif - } - ModuleBase::timer::end("elecstate", "cal_dm_psi"); - - return; -} -template -void cal_dm_psi(const Parallel_Orbitals* ParaV, - const ModuleBase::matrix& wg, - const psi::Psi>& wfc, - elecstate::DensityMatrix, TR>& DM) -{ - ModuleBase::TITLE("elecstate", "cal_dm_psi"); - ModuleBase::timer::start("elecstate", "cal_dm_psi"); - - // dm.resize(wfc.get_nk(), ParaV->ncol, ParaV->nrow); - const int nbands_local = wfc.get_nbands(); - const int nbasis_local = wfc.get_nbasis(); - - // dm = wfc.T * wg * wfc.conj() - // dm[is](iw1,iw2) = \sum_{ib} wfc[is](ib,iw1).T * wg(is,ib) * wfc[is](ib,iw2).conj() - for (int ik = 0; ik < wfc.get_nk(); ++ik) - { - wfc.fix_k(ik); - std::complex* dmk_pointer = DM.get_DMK_pointer(ik); - // dm.fix_k(ik); - // dm[ik].create(ParaV->ncol, ParaV->nrow); - // wg_wfc(ib,iw) = wg[ib] * wfc(ib,iw); - psi::Psi> wg_wfc(1, wfc.get_nbands(), wfc.get_nbasis(), wfc.get_nbasis(), true); - - const std::complex* pwfc = wfc.get_pointer(); - std::complex* pwg_wfc = wg_wfc.get_pointer(); - -#ifdef _OPENMP -#pragma omp parallel for schedule(static, 1024) -#endif - for (int i = 0; i < wg_wfc.size(); ++i) - { - pwg_wfc[i] = conj(pwfc[i]); - } - - int ib_global = 0; - for (int ib_local = 0; ib_local < nbands_local; ++ib_local) - { - while (ib_local != ParaV->global2local_col(ib_global)) - { - ++ib_global; - if (ib_global >= wg.nc) - { - break; - ModuleBase::WARNING_QUIT("ElecStateLCAO::cal_dm", "please check global2local_col!"); - } - } - if (ib_global >= wg.nc) - { - continue; - } - const double wg_local = wg(ik, ib_global); - std::complex* wg_wfc_pointer = &(wg_wfc(0, ib_local, 0)); - BlasConnector::scal(nbasis_local, wg_local, wg_wfc_pointer, 1); - } - - // C++: dm(iw1,iw2) = wfc(ib,iw1).T * wg_wfc(ib,iw2) -#ifdef __MPI - psiMulPsiMpi(wg_wfc, wfc, dmk_pointer, ParaV->desc_wfc, ParaV->desc); -#else - psiMulPsi(wg_wfc, wfc, dmk_pointer); -#endif - } - - ModuleBase::timer::end("elecstate", "cal_dm_psi"); - return; -} - -#ifdef __MPI -void psiMulPsiMpi(const psi::Psi& psi1, const psi::Psi& psi2, double* dm_out, const int* desc_psi, const int* desc_dm) -{ - ModuleBase::timer::start("psiMulPsiMpi", "pdgemm"); - const double one_float = 1.0, zero_float = 0.0; - const int one_int = 1; - const char N_char = 'N', T_char = 'T'; - const int nlocal = desc_dm[2]; - const int nbands = desc_psi[3]; - - ScalapackConnector::gemm(N_char, - T_char, - nlocal, - nlocal, - nbands, - one_float, - psi1.get_pointer(), - one_int, - one_int, - desc_psi, - psi2.get_pointer(), - one_int, - one_int, - desc_psi, - zero_float, - dm_out, - one_int, - one_int, - desc_dm); - ModuleBase::timer::end("psiMulPsiMpi", "pdgemm"); -} - -void psiMulPsiMpi(const psi::Psi>& psi1, - const psi::Psi>& psi2, - std::complex* dm_out, - const int* desc_psi, - const int* desc_dm) -{ - ModuleBase::timer::start("psiMulPsiMpi", "pzgemm"); - const std::complex one_complex = {1.0, 0.0}, zero_complex = {0.0, 0.0}; - const int one_int = 1; - const char N_char = 'N', T_char = 'T'; - const int nlocal = desc_dm[2]; - const int nbands = desc_psi[3]; - ScalapackConnector::gemm(N_char, - T_char, - nlocal, - nlocal, - nbands, - one_complex, - psi1.get_pointer(), - one_int, - one_int, - desc_psi, - psi2.get_pointer(), - one_int, - one_int, - desc_psi, - zero_complex, - dm_out, - one_int, - one_int, - desc_dm); - ModuleBase::timer::end("psiMulPsiMpi", "pzgemm"); -} - -#else - -void psiMulPsi(const psi::Psi& psi1, const psi::Psi& psi2, double* dm_out) -{ - const double one_float = 1.0, zero_float = 0.0; - const int one_int = 1; - const char N_char = 'N', T_char = 'T'; - const int nlocal = psi1.get_nbasis(); - const int nbands = psi1.get_nbands(); - BlasConnector::gemm_cm(N_char, - T_char, - nlocal, - nlocal, - nbands, - one_float, - psi1.get_pointer(), - nlocal, - psi2.get_pointer(), - nlocal, - zero_float, - dm_out, - nlocal); -} - -void psiMulPsi(const psi::Psi>& psi1, const psi::Psi>& psi2, std::complex* dm_out) -{ - const int one_int = 1; - const char N_char = 'N', T_char = 'T'; - const int nlocal = psi1.get_nbasis(); - const int nbands = psi1.get_nbands(); - const std::complex one_complex = {1.0, 0.0}; - const std::complex zero_complex = {0.0, 0.0}; - BlasConnector::gemm_cm(N_char, - T_char, - nlocal, - nlocal, - nbands, - one_complex, - psi1.get_pointer(), - nlocal, - psi2.get_pointer(), - nlocal, - zero_complex, - dm_out, - nlocal); -} -#endif - -template void cal_dm_psi(const Parallel_Orbitals* ParaV, - const ModuleBase::matrix& wg, - const psi::Psi>& wfc, - elecstate::DensityMatrix, std::complex>& DM); -template void cal_dm_psi(const Parallel_Orbitals* ParaV, - const ModuleBase::matrix& wg, - const psi::Psi>& wfc, - elecstate::DensityMatrix, double>& DM); -} // namespace elecstate diff --git a/source/source_estate/module_dm/cal_dm_psi.h b/source/source_estate/module_dm/cal_dm_psi.h deleted file mode 100644 index acdad8fdeb2..00000000000 --- a/source/source_estate/module_dm/cal_dm_psi.h +++ /dev/null @@ -1,42 +0,0 @@ -#ifndef CAL_DM_PSI_H -#define CAL_DM_PSI_H - -#include "density_matrix.h" -#include "source_base/matrix.h" -#include "source_psi/psi.h" - -namespace elecstate -{ -// for Gamma-Only case where DMK is double -void cal_dm_psi(const Parallel_Orbitals* ParaV, - const ModuleBase::matrix& wg, - const psi::Psi& wfc, - elecstate::DensityMatrix& DM); - -// for Multi-k case where DMK is std::complex -template -void cal_dm_psi(const Parallel_Orbitals* ParaV, - const ModuleBase::matrix& wg, - const psi::Psi>& wfc, - elecstate::DensityMatrix, TR>& DM); - -#ifdef __MPI -// for Gamma-Only case with MPI -void psiMulPsiMpi(const psi::Psi& psi1, const psi::Psi& psi2, double* dm_out, const int* desc_psi, const int* desc_dm); - -// for multi-k case with MPI -void psiMulPsiMpi(const psi::Psi>& psi1, - const psi::Psi>& psi2, - std::complex* dm_out, - const int* desc_psi, - const int* desc_dm); - -#else -// for Gamma-Only case without MPI -void psiMulPsi(const psi::Psi& psi1, const psi::Psi& psi2, double* dm_out); - -// for multi-k case without MPI -void psiMulPsi(const psi::Psi>& psi1, const psi::Psi>& psi2, std::complex* dm_out); -#endif -}; // namespace elecstate -#endif diff --git a/source/source_estate/module_dm/cal_edm_tddft.cpp b/source/source_estate/module_dm/cal_edm_tddft.cpp deleted file mode 100644 index 524d7245a9e..00000000000 --- a/source/source_estate/module_dm/cal_edm_tddft.cpp +++ /dev/null @@ -1,820 +0,0 @@ -#include "cal_edm_tddft.h" - -#include "source_base/module_container/ATen/core/tensor.h" // For ct::Tensor -#include "source_base/module_container/ATen/kernels/blas.h" -#include "source_base/module_container/ATen/kernels/lapack.h" -#include "source_base/module_container/ATen/kernels/memory.h" // memory operations (Tensor) -#include "source_base/module_device/memory_op.h" // memory operations -#include "source_base/module_external/lapack_connector.h" -#include "source_base/module_external/scalapack_connector.h" -#include "source_io/module_parameter/parameter.h" // use PARAM.globalv -#include "source_lcao/module_rt/gather_mat.h" // gatherMatrix and distributeMatrix -#include "source_lcao/module_rt/propagator.h" // Include header for create_identity_matrix - -namespace elecstate -{ -void print_local_matrix(std::ostream& os, - const std::complex* matrix_data, - int local_rows, - int local_cols, - const std::string& matrix_name, - int rank) -{ - if (!matrix_name.empty() || rank >= 0) - { - os << "=== "; - if (!matrix_name.empty()) - { - os << "Matrix: " << matrix_name; - if (rank >= 0) - os << " "; - } - if (rank >= 0) - { - os << "(Process: " << rank + 1 << ")"; - } - os << " (Local dims: " << local_rows << " x " << local_cols << ") ===" << std::endl; - } - - os << std::fixed << std::setprecision(10) << std::showpos; - - for (int i = 0; i < local_rows; ++i) // Iterate over rows (i) - { - for (int j = 0; j < local_cols; ++j) // Iterate over columns (j) - { - // For column-major storage, element (i, j) is at index i + j * LDA - // where LDA (leading dimension) is typically the number of *rows* in the local block. - int idx = i + j * local_rows; - os << "(" << std::real(matrix_data[idx]) << "," << std::imag(matrix_data[idx]) << ") "; - } - os << std::endl; // New line after each row - } - os.unsetf(std::ios_base::fixed | std::ios_base::showpos); - os << std::endl; -} - -// use the original formula (Hamiltonian matrix) to calculate energy density matrix -void cal_edm_tddft(Parallel_Orbitals& pv, - LCAO_domain::Setup_DM>& dmat, - K_Vectors& kv, - hamilt::Hamilt>* p_hamilt) -{ - ModuleBase::TITLE("elecstate", "cal_edm_tddft"); - ModuleBase::timer::start("TD_Efficiency", "cal_edm_tddft"); - - const int nlocal = PARAM.globalv.nlocal; - assert(nlocal >= 0); - - dmat.dm->EDMK.resize(kv.get_nks()); - - for (int ik = 0; ik < kv.get_nks(); ++ik) - { - p_hamilt->updateHk(ik); - std::complex* tmp_dmk = dmat.dm->get_DMK_pointer(ik); - ModuleBase::ComplexMatrix& tmp_edmk = dmat.dm->EDMK[ik]; - -#ifdef __MPI - const int nloc = pv.nloc; - const int ncol = pv.ncol; - const int nrow = pv.nrow; - - tmp_edmk.create(ncol, nrow); - std::complex* Htmp = new std::complex[nloc]; - std::complex* Sinv = new std::complex[nloc]; - std::complex* tmp1 = new std::complex[nloc]; - std::complex* tmp2 = new std::complex[nloc]; - std::complex* tmp3 = new std::complex[nloc]; - std::complex* tmp4 = new std::complex[nloc]; - - ModuleBase::GlobalFunc::ZEROS(Htmp, nloc); - ModuleBase::GlobalFunc::ZEROS(Sinv, nloc); - ModuleBase::GlobalFunc::ZEROS(tmp1, nloc); - ModuleBase::GlobalFunc::ZEROS(tmp2, nloc); - ModuleBase::GlobalFunc::ZEROS(tmp3, nloc); - ModuleBase::GlobalFunc::ZEROS(tmp4, nloc); - - const int inc = 1; - - hamilt::MatrixBlock> h_mat; - hamilt::MatrixBlock> s_mat; - - p_hamilt->matrix(h_mat, s_mat); - BlasConnector::copy(nloc, h_mat.p, inc, Htmp, inc); - BlasConnector::copy(nloc, s_mat.p, inc, Sinv, inc); - - std::vector ipiv(nloc, 0); - int info = 0; - const int one_int = 1; - - ScalapackConnector::getrf(nlocal, nlocal, Sinv, one_int, one_int, pv.desc, ipiv.data(), &info); - - int lwork = -1; - int liwork = -1; - - // if lwork == -1, then the size of work is (at least) of length 1. - std::vector> work(1, 0); - - // if liwork = -1, then the size of iwork is (at least) of length 1. - std::vector iwork(1, 0); - - ScalapackConnector::getri(nlocal, - Sinv, - one_int, - one_int, - pv.desc, - ipiv.data(), - work.data(), - &lwork, - iwork.data(), - &liwork, - &info); - - lwork = work[0].real(); - work.resize(lwork, 0); - liwork = iwork[0]; - iwork.resize(liwork, 0); - - ScalapackConnector::getri(nlocal, - Sinv, - one_int, - one_int, - pv.desc, - ipiv.data(), - work.data(), - &lwork, - iwork.data(), - &liwork, - &info); - - const char N_char = 'N'; - const char T_char = 'T'; - const std::complex one_complex = {1.0, 0.0}; - const std::complex zero_complex = {0.0, 0.0}; - const std::complex half_complex = {0.5, 0.0}; - - // tmp1 = Htmp * Sinv - ScalapackConnector::gemm(N_char, - N_char, - nlocal, - nlocal, - nlocal, - one_complex, - Htmp, - one_int, - one_int, - pv.desc, - Sinv, - one_int, - one_int, - pv.desc, - zero_complex, - tmp1, - one_int, - one_int, - pv.desc); - - // tmp2 = tmp1^T * tmp_dmk - ScalapackConnector::gemm(T_char, - N_char, - nlocal, - nlocal, - nlocal, - one_complex, - tmp1, - one_int, - one_int, - pv.desc, - tmp_dmk, - one_int, - one_int, - pv.desc, - zero_complex, - tmp2, - one_int, - one_int, - pv.desc); - - // tmp3 = Sinv * Htmp - ScalapackConnector::gemm(N_char, - N_char, - nlocal, - nlocal, - nlocal, - one_complex, - Sinv, - one_int, - one_int, - pv.desc, - Htmp, - one_int, - one_int, - pv.desc, - zero_complex, - tmp3, - one_int, - one_int, - pv.desc); - - // tmp4 = tmp_dmk * tmp3^T - ScalapackConnector::gemm(N_char, - T_char, - nlocal, - nlocal, - nlocal, - one_complex, - tmp_dmk, - one_int, - one_int, - pv.desc, - tmp3, - one_int, - one_int, - pv.desc, - zero_complex, - tmp4, - one_int, - one_int, - pv.desc); - - // tmp4 = 0.5 * (tmp2 + tmp4) - ScalapackConnector::geadd(N_char, - nlocal, - nlocal, - half_complex, - tmp2, - one_int, - one_int, - pv.desc, - half_complex, - tmp4, - one_int, - one_int, - pv.desc); - - BlasConnector::copy(nloc, tmp4, inc, tmp_edmk.c, inc); - - delete[] Htmp; - delete[] Sinv; - delete[] tmp1; - delete[] tmp2; - delete[] tmp3; - delete[] tmp4; -#else - // for serial version - tmp_edmk.create(pv.ncol, pv.nrow); - ModuleBase::ComplexMatrix Sinv(nlocal, nlocal); - ModuleBase::ComplexMatrix Htmp(nlocal, nlocal); - - hamilt::MatrixBlock> h_mat; - hamilt::MatrixBlock> s_mat; - - p_hamilt->matrix(h_mat, s_mat); - - for (int i = 0; i < nlocal; i++) - { - for (int j = 0; j < nlocal; j++) - { - Htmp(i, j) = h_mat.p[i * nlocal + j]; - Sinv(i, j) = s_mat.p[i * nlocal + j]; - } - } - int INFO = 0; - - int lwork = 3 * nlocal - 1; // tmp - std::complex* work = new std::complex[lwork]; - ModuleBase::GlobalFunc::ZEROS(work, lwork); - - int IPIV[nlocal]; - - LapackConnector::zgetrf(nlocal, nlocal, Sinv, nlocal, IPIV, &INFO); - LapackConnector::zgetri(nlocal, Sinv, nlocal, IPIV, work, lwork, &INFO); - // I just use ModuleBase::ComplexMatrix temporarily, and will change it - // to std::complex* - ModuleBase::ComplexMatrix tmp_dmk_base(nlocal, nlocal); - for (int i = 0; i < nlocal; i++) - { - for (int j = 0; j < nlocal; j++) - { - tmp_dmk_base(i, j) = tmp_dmk[i * nlocal + j]; - } - } - tmp_edmk = 0.5 * (Sinv * Htmp * tmp_dmk_base + tmp_dmk_base * Htmp * Sinv); - delete[] work; -#endif - } // end ik - - ModuleBase::timer::end("TD_Efficiency", "cal_edm_tddft"); - return; -} // cal_edm_tddft - -void cal_edm_tddft_tensor(Parallel_Orbitals& pv, - LCAO_domain::Setup_DM>& dmat, - K_Vectors& kv, - hamilt::Hamilt>* p_hamilt) -{ - ModuleBase::TITLE("elecstate", "cal_edm_tddft_tensor"); - ModuleBase::timer::start("TD_Efficiency", "cal_edm_tddft"); - - const int nlocal = PARAM.globalv.nlocal; - assert(nlocal >= 0); - dmat.dm->EDMK.resize(kv.get_nks()); - - for (int ik = 0; ik < kv.get_nks(); ++ik) - { - p_hamilt->updateHk(ik); - std::complex* tmp_dmk = dmat.dm->get_DMK_pointer(ik); - ModuleBase::ComplexMatrix& tmp_edmk = dmat.dm->EDMK[ik]; - -#ifdef __MPI - const int nloc = pv.nloc; - const int ncol = pv.ncol; - const int nrow = pv.nrow; - - // Initialize EDMK matrix - tmp_edmk.create(ncol, nrow); - - // Allocate Tensor objects on CPU - ct::Tensor Htmp_tensor(ct::DataType::DT_COMPLEX_DOUBLE, ct::DeviceType::CpuDevice, ct::TensorShape({nloc})); - Htmp_tensor.zero(); - - ct::Tensor Sinv_tensor(ct::DataType::DT_COMPLEX_DOUBLE, ct::DeviceType::CpuDevice, ct::TensorShape({nloc})); - Sinv_tensor.zero(); - - ct::Tensor tmp1_tensor(ct::DataType::DT_COMPLEX_DOUBLE, ct::DeviceType::CpuDevice, ct::TensorShape({nloc})); - tmp1_tensor.zero(); - - ct::Tensor tmp2_tensor(ct::DataType::DT_COMPLEX_DOUBLE, ct::DeviceType::CpuDevice, ct::TensorShape({nloc})); - tmp2_tensor.zero(); - - ct::Tensor tmp3_tensor(ct::DataType::DT_COMPLEX_DOUBLE, ct::DeviceType::CpuDevice, ct::TensorShape({nloc})); - tmp3_tensor.zero(); - - ct::Tensor tmp4_tensor(ct::DataType::DT_COMPLEX_DOUBLE, ct::DeviceType::CpuDevice, ct::TensorShape({nloc})); - tmp4_tensor.zero(); - - // Get raw pointers from tensors for ScaLAPACK calls - std::complex* Htmp_ptr = Htmp_tensor.data>(); - std::complex* Sinv_ptr = Sinv_tensor.data>(); - std::complex* tmp1_ptr = tmp1_tensor.data>(); - std::complex* tmp2_ptr = tmp2_tensor.data>(); - std::complex* tmp3_ptr = tmp3_tensor.data>(); - std::complex* tmp4_ptr = tmp4_tensor.data>(); - - const int inc = 1; - hamilt::MatrixBlock> h_mat; - hamilt::MatrixBlock> s_mat; - p_hamilt->matrix(h_mat, s_mat); - - // Copy Hamiltonian and Overlap matrices into Tensor buffers using BlasConnector - BlasConnector::copy(nloc, h_mat.p, inc, Htmp_ptr, inc); - BlasConnector::copy(nloc, s_mat.p, inc, Sinv_ptr, inc); - - int myid = 0; - const int root_proc = 0; - MPI_Comm_rank(MPI_COMM_WORLD, &myid); - - // --- ScaLAPACK Inversion of S --- - ct::Tensor ipiv(ct::DataType::DT_INT, - ct::DeviceType::CpuDevice, - ct::TensorShape({pv.nrow + pv.nb})); // Size for ScaLAPACK pivot array - ipiv.zero(); - int* ipiv_ptr = ipiv.data(); - - int info = 0; - const int one_int = 1; - ScalapackConnector::getrf(nlocal, nlocal, Sinv_ptr, one_int, one_int, pv.desc, ipiv_ptr, &info); - - int lwork = -1; - int liwork = -1; - ct::Tensor work_query(ct::DataType::DT_COMPLEX_DOUBLE, ct::DeviceType::CpuDevice, ct::TensorShape({1})); - ct::Tensor iwork_query(ct::DataType::DT_INT, ct::DeviceType::CpuDevice, ct::TensorShape({1})); - - ScalapackConnector::getri(nlocal, - Sinv_ptr, - one_int, - one_int, - pv.desc, - ipiv_ptr, - work_query.data>(), - &lwork, - iwork_query.data(), - &liwork, - &info); - - // Resize work arrays based on query results - lwork = work_query.data>()[0].real(); - work_query.resize(ct::TensorShape({lwork})); - liwork = iwork_query.data()[0]; - iwork_query.resize(ct::TensorShape({liwork})); - - ScalapackConnector::getri(nlocal, - Sinv_ptr, - one_int, - one_int, - pv.desc, - ipiv_ptr, - work_query.data>(), - &lwork, - iwork_query.data(), - &liwork, - &info); - - // --- EDM Calculation using ScaLAPACK --- - const char N_char = 'N'; - const char T_char = 'T'; - const std::complex one_complex = {1.0, 0.0}; - const std::complex zero_complex = {0.0, 0.0}; - const std::complex half_complex = {0.5, 0.0}; - - // tmp1 = Htmp * Sinv - ScalapackConnector::gemm(N_char, - N_char, - nlocal, - nlocal, - nlocal, - one_complex, - Htmp_ptr, - one_int, - one_int, - pv.desc, - Sinv_ptr, - one_int, - one_int, - pv.desc, - zero_complex, - tmp1_ptr, - one_int, - one_int, - pv.desc); - - // tmp2 = tmp1^T * tmp_dmk - ScalapackConnector::gemm(T_char, - N_char, - nlocal, - nlocal, - nlocal, - one_complex, - tmp1_ptr, - one_int, - one_int, - pv.desc, - tmp_dmk, - one_int, - one_int, - pv.desc, - zero_complex, - tmp2_ptr, - one_int, - one_int, - pv.desc); - - // tmp3 = Sinv * Htmp - ScalapackConnector::gemm(N_char, - N_char, - nlocal, - nlocal, - nlocal, - one_complex, - Sinv_ptr, - one_int, - one_int, - pv.desc, - Htmp_ptr, - one_int, - one_int, - pv.desc, - zero_complex, - tmp3_ptr, - one_int, - one_int, - pv.desc); - - // tmp4 = tmp_dmk * tmp3^T - ScalapackConnector::gemm(N_char, - T_char, - nlocal, - nlocal, - nlocal, - one_complex, - tmp_dmk, - one_int, - one_int, - pv.desc, - tmp3_ptr, - one_int, - one_int, - pv.desc, - zero_complex, - tmp4_ptr, - one_int, - one_int, - pv.desc); - - // tmp4 = 0.5 * (tmp2 + tmp4) - ScalapackConnector::geadd(N_char, - nlocal, - nlocal, - half_complex, - tmp2_ptr, - one_int, - one_int, - pv.desc, - half_complex, - tmp4_ptr, - one_int, - one_int, - pv.desc); - - // Copy final result from Tensor buffer back to EDMK matrix - BlasConnector::copy(nloc, tmp4_ptr, inc, tmp_edmk.c, inc); - -#else - ModuleBase::WARNING_QUIT("elecstate::cal_edm_tddft_tensor", "MPI is required for this function!"); -#endif - } // end ik - ModuleBase::timer::end("TD_Efficiency", "cal_edm_tddft"); - return; -} // cal_edm_tddft_tensor - -// Template function for EDM calculation supporting CPU and GPU -template -void cal_edm_tddft_tensor_lapack(Parallel_Orbitals& pv, - LCAO_domain::Setup_DM>& dmat, - K_Vectors& kv, - hamilt::Hamilt>* p_hamilt) -{ - ModuleBase::TITLE("elecstate", "cal_edm_tddft_tensor_lapack"); - ModuleBase::timer::start("TD_Efficiency", "cal_edm_tddft"); - - const int nlocal = PARAM.globalv.nlocal; - assert(nlocal >= 0); - dmat.dm->EDMK.resize(kv.get_nks()); - - // ct_device_type = ct::DeviceType::CpuDevice or ct::DeviceType::GpuDevice - ct::DeviceType ct_device_type = ct::DeviceTypeToEnum::value; - // ct_Device = ct::DEVICE_CPU or ct::DEVICE_GPU - using ct_Device = typename ct::PsiToContainer::type; - - // Memory operations - using syncmem_complex_h2d_op - = base_device::memory::synchronize_memory_op, Device, base_device::DEVICE_CPU>; - using syncmem_complex_d2h_op - = base_device::memory::synchronize_memory_op, base_device::DEVICE_CPU, Device>; - -#if ((defined __CUDA) /* || (defined __ROCM) */) - if (ct_device_type == ct::DeviceType::GpuDevice) - { - // Initialize cuBLAS & cuSOLVER handle - ct::kernels::createGpuSolverHandle(); - ct::kernels::createGpuBlasHandle(); - } -#endif // __CUDA - - for (int ik = 0; ik < kv.get_nks(); ++ik) - { - p_hamilt->updateHk(ik); - std::complex* tmp_dmk_local = dmat.dm->get_DMK_pointer(ik); - ModuleBase::ComplexMatrix& tmp_edmk = dmat.dm->EDMK[ik]; - -#ifdef __MPI - int myid = 0; - const int root_proc = 0; - int num_procs = 1; - MPI_Comm_rank(MPI_COMM_WORLD, &myid); - MPI_Comm_size(MPI_COMM_WORLD, &num_procs); - - // 1. Prepare Data Source Pointers (Host) - // If np = 1, point directly to local data to avoid copy - // If np > 1, gather data and point to the gathered buffer - std::complex* h_src = nullptr; - std::complex* s_src = nullptr; - std::complex* dmk_src = nullptr; - - // Global containers (Used only when num_procs > 1) - module_rt::Matrix_g> h_mat_global, s_mat_global, dmk_global, edm_global; - - // Get Local Matrices - hamilt::MatrixBlock> h_mat_local, s_mat_local; - p_hamilt->matrix(h_mat_local, s_mat_local); - - if (num_procs == 1) - { - // Optimization: Direct access for single process - h_src = h_mat_local.p; - s_src = s_mat_local.p; - dmk_src = tmp_dmk_local; - } - else - { - // Standard Gather Logic for multi-process - module_rt::gatherMatrix(myid, root_proc, h_mat_local, h_mat_global); - module_rt::gatherMatrix(myid, root_proc, s_mat_local, s_mat_global); - - hamilt::MatrixBlock> dmk_local_block; - dmk_local_block.p = tmp_dmk_local; - dmk_local_block.desc = pv.desc; - module_rt::gatherMatrix(myid, root_proc, dmk_local_block, dmk_global); - - if (myid == root_proc) - { - h_src = h_mat_global.p.get(); - s_src = s_mat_global.p.get(); - dmk_src = dmk_global.p.get(); - } - } - - // 2. GPU Calculation (on Rank 0) - if (myid == root_proc) - { - ct::Tensor H_dev, S_dev, DMK_dev, ipiv_dev; - - // Allocate and Copy (H2D) - H_dev = ct::Tensor(ct::DataType::DT_COMPLEX_DOUBLE, ct_device_type, ct::TensorShape({nlocal, nlocal})); - syncmem_complex_h2d_op()(H_dev.template data>(), h_src, nlocal * nlocal); - - S_dev = ct::Tensor(ct::DataType::DT_COMPLEX_DOUBLE, ct_device_type, ct::TensorShape({nlocal, nlocal})); - syncmem_complex_h2d_op()(S_dev.template data>(), s_src, nlocal * nlocal); - - DMK_dev = ct::Tensor(ct::DataType::DT_COMPLEX_DOUBLE, ct_device_type, ct::TensorShape({nlocal, nlocal})); - syncmem_complex_h2d_op()(DMK_dev.template data>(), dmk_src, nlocal * nlocal); - - ipiv_dev = ct::Tensor(ct::DataType::DT_INT, ct_device_type, ct::TensorShape({nlocal})); - ipiv_dev.zero(); - - // --- Calculate S^-1 using getrf + getrs --- - // 1. LU decomposition S = P * L * U - ct::kernels::lapack_getrf, ct_Device>()(nlocal, - nlocal, - S_dev.template data>(), - nlocal, - ipiv_dev.template data()); - - // 2. Solve S * Sinv = I - auto Sinv_dev = module_rt::create_identity_matrix>(nlocal, ct_device_type); - - ct::kernels::lapack_getrs, ct_Device>()('N', - nlocal, - nlocal, - S_dev.template data>(), - nlocal, - ipiv_dev.template data(), - Sinv_dev.template data>(), - nlocal); - - // --- EDM Calculation --- - std::complex one = {1.0, 0.0}; - std::complex zero = {0.0, 0.0}; - - // tmp1 = H * Sinv - ct::Tensor tmp1_dev(ct::DataType::DT_COMPLEX_DOUBLE, ct_device_type, ct::TensorShape({nlocal, nlocal})); - ct::kernels::blas_gemm, ct_Device>()('N', - 'N', - nlocal, - nlocal, - nlocal, - &one, - H_dev.template data>(), - nlocal, - Sinv_dev.template data>(), - nlocal, - &zero, - tmp1_dev.template data>(), - nlocal); - - // tmp2 = tmp1^T * DMK - ct::Tensor tmp2_dev(ct::DataType::DT_COMPLEX_DOUBLE, ct_device_type, ct::TensorShape({nlocal, nlocal})); - ct::kernels::blas_gemm, ct_Device>()('T', - 'N', - nlocal, - nlocal, - nlocal, - &one, - tmp1_dev.template data>(), - nlocal, - DMK_dev.template data>(), - nlocal, - &zero, - tmp2_dev.template data>(), - nlocal); - - // tmp3 = Sinv * H - ct::Tensor tmp3_dev(ct::DataType::DT_COMPLEX_DOUBLE, ct_device_type, ct::TensorShape({nlocal, nlocal})); - ct::kernels::blas_gemm, ct_Device>()('N', - 'N', - nlocal, - nlocal, - nlocal, - &one, - Sinv_dev.template data>(), - nlocal, - H_dev.template data>(), - nlocal, - &zero, - tmp3_dev.template data>(), - nlocal); - - // tmp4 = DMK * tmp3^T - ct::Tensor tmp4_dev(ct::DataType::DT_COMPLEX_DOUBLE, ct_device_type, ct::TensorShape({nlocal, nlocal})); - ct::kernels::blas_gemm, ct_Device>()('N', - 'T', - nlocal, - nlocal, - nlocal, - &one, - DMK_dev.template data>(), - nlocal, - tmp3_dev.template data>(), - nlocal, - &zero, - tmp4_dev.template data>(), - nlocal); - - // tmp4 = tmp2 + tmp4 - ct::kernels::blas_axpy, ct_Device>()(nlocal * nlocal, - &one, - tmp2_dev.template data>(), - 1, - tmp4_dev.template data>(), - 1); - - // tmp4 = 0.5 * tmp4 - std::complex half = {0.5, 0.0}; - ct::kernels::blas_scal, ct_Device>()(nlocal * nlocal, - &half, - tmp4_dev.template data>(), - 1); - - // 3. Retrieve Result (D2H) - std::complex* edm_dest = nullptr; - - if (num_procs == 1) - { - // Directly copy to target local matrix - tmp_edmk.create(pv.ncol, pv.nrow); - edm_dest = tmp_edmk.c; - } - else - { - // Wait to set up edm_dest after allocating global buffer - if (myid == root_proc && edm_global.p == nullptr) - { - edm_global.p.reset(new std::complex[nlocal * nlocal]); - } - edm_dest = edm_global.p.get(); - } - - if (num_procs == 1 || myid == root_proc) - { - syncmem_complex_d2h_op()(edm_dest, tmp4_dev.template data>(), nlocal * nlocal); - } - } - - // 4. Distribute (Only needed if num_procs > 1) - if (num_procs > 1) - { - if (edm_global.p == nullptr) - { - edm_global.p.reset(new std::complex[nlocal * nlocal]); - } - - edm_global.row = nlocal; - edm_global.col = nlocal; - edm_global.desc.reset(new int[9]{1, pv.desc[1], nlocal, nlocal, nlocal, nlocal, 0, 0, nlocal}); - - tmp_edmk.create(pv.ncol, pv.nrow); - hamilt::MatrixBlock> edm_local_block; - edm_local_block.p = tmp_edmk.c; - edm_local_block.desc = pv.desc; - module_rt::distributeMatrix(edm_local_block, edm_global); - } -#else - ModuleBase::WARNING_QUIT("elecstate::cal_edm_tddft_tensor_lapack", "MPI is required for this function!"); -#endif // __MPI - } // end ik - -#if ((defined __CUDA) /* || (defined __ROCM) */) - if (ct_device_type == ct::DeviceType::GpuDevice) - { - // Destroy cuBLAS & cuSOLVER handle - ct::kernels::destroyGpuSolverHandle(); - ct::kernels::destroyGpuBlasHandle(); - } -#endif // __CUDA - - ModuleBase::timer::end("TD_Efficiency", "cal_edm_tddft"); - return; -} // cal_edm_tddft_tensor_lapack - -// Explicit instantiation of template functions -template void cal_edm_tddft_tensor_lapack(Parallel_Orbitals& pv, - LCAO_domain::Setup_DM>& dmat, - K_Vectors& kv, - hamilt::Hamilt>* p_hamilt); -#if ((defined __CUDA) /* || (defined __ROCM) */) -template void cal_edm_tddft_tensor_lapack(Parallel_Orbitals& pv, - LCAO_domain::Setup_DM>& dmat, - K_Vectors& kv, - hamilt::Hamilt>* p_hamilt); -#endif // __CUDA - -} // namespace elecstate diff --git a/source/source_estate/module_dm/cal_edm_tddft.h b/source/source_estate/module_dm/cal_edm_tddft.h deleted file mode 100644 index b442bd90cd0..00000000000 --- a/source/source_estate/module_dm/cal_edm_tddft.h +++ /dev/null @@ -1,34 +0,0 @@ -#ifndef CAL_EDM_TDDFT_H -#define CAL_EDM_TDDFT_H - -#include "source_basis/module_ao/parallel_orbitals.h" -#include "source_cell/klist.h" -#include "source_hamilt/hamilt.h" -#include "source_lcao/setup_dm.h" - -namespace elecstate -{ -void print_local_matrix(std::ostream& os, - const std::complex* matrix_data, - int local_rows, // pv.nrow - int local_cols, // pv.ncol - const std::string& matrix_name = "", - int rank = -1); - -void cal_edm_tddft(Parallel_Orbitals& pv, - LCAO_domain::Setup_DM>& dmat, - K_Vectors& kv, - hamilt::Hamilt>* p_hamilt); - -void cal_edm_tddft_tensor(Parallel_Orbitals& pv, - LCAO_domain::Setup_DM>& dmat, - K_Vectors& kv, - hamilt::Hamilt>* p_hamilt); - -template -void cal_edm_tddft_tensor_lapack(Parallel_Orbitals& pv, - LCAO_domain::Setup_DM>& dmat, - K_Vectors& kv, - hamilt::Hamilt>* p_hamilt); -} // namespace elecstate -#endif // CAL_EDM_TDDFT_H diff --git a/source/source_estate/module_dm/density_matrix.cpp b/source/source_estate/module_dm/density_matrix.cpp index ae11c73b89b..018779587d7 100644 --- a/source/source_estate/module_dm/density_matrix.cpp +++ b/source/source_estate/module_dm/density_matrix.cpp @@ -9,7 +9,7 @@ #include "source_base/constants.h" #include "source_cell/klist.h" -namespace elecstate +namespace module_dm { //---------------------------------------------------- @@ -20,663 +20,124 @@ namespace elecstate template DensityMatrix::~DensityMatrix() { - for (auto& it: this->_DMR) + this->clear_dmr(); +} + +template +void DensityMatrix::clear_dmr() +{ + for (hamilt::HContainer*& it: this->dmr) { delete it; } - delete[] this->dmr_tmp_; + this->dmr.clear(); + this->_dmr_ready = false; } template -DensityMatrix::DensityMatrix(const Parallel_Orbitals* paraV_in, const int nspin, const std::vector>& kvec_d, const int nk) - : _paraV(paraV_in), _nspin(nspin), _kvec_d(kvec_d), _nk((nk > 0 && nk <= _kvec_d.size()) ? nk : _kvec_d.size()) +DensityMatrix::DensityMatrix(const Parallel_Orbitals* paraV_in, + const int spin_mult, + const std::vector>& kvec_d, + const int nk, + const int nspin) + : pv(paraV_in), spin_mult(spin_mult), nspin(nspin > 0 ? nspin : spin_mult), + _kvec_d(kvec_d), _nk((nk > 0 && nk <= _kvec_d.size()) ? nk : _kvec_d.size()) { ModuleBase::TITLE("DensityMatrix", "resize_DMK"); - const int nks = _nk * _nspin; - this->_DMK.resize(nks); + const int nks = _nk * this->spin_mult; + this->dmk.resize(nks); for (int ik = 0; ik < nks; ik++) { - this->_DMK[ik].resize(this->_paraV->get_row_size() * this->_paraV->get_col_size()); + this->dmk[ik].resize(this->pv->get_row_size() * this->pv->get_col_size()); } - ModuleBase::Memory::record("DensityMatrix::DMK", this->_DMK.size() * this->_DMK[0].size() * sizeof(TK)); + ModuleBase::Memory::record("DensityMatrix::DMK", this->dmk.size() * this->dmk[0].size() * sizeof(TK)); } template -DensityMatrix::DensityMatrix(const Parallel_Orbitals* paraV_in, const int nspin) :_paraV(paraV_in), _nspin(nspin), _kvec_d({ ModuleBase::Vector3(0,0,0) }), _nk(1) +DensityMatrix::DensityMatrix(const Parallel_Orbitals* paraV_in, const int spin_mult, const int nspin) + : pv(paraV_in), spin_mult(spin_mult), nspin(nspin > 0 ? nspin : spin_mult), + _kvec_d({ModuleBase::Vector3(0, 0, 0)}), _nk(1) { ModuleBase::TITLE("DensityMatrix", "resize_gamma"); - this->_DMK.resize(_nspin); - for (int ik = 0; ik < this->_nspin; ik++) + this->dmk.resize(this->spin_mult); + for (int ik = 0; ik < this->spin_mult; ik++) { - this->_DMK[ik].resize(this->_paraV->get_row_size() * this->_paraV->get_col_size()); - } - ModuleBase::Memory::record("DensityMatrix::DMK", this->_DMK.size() * this->_DMK[0].size() * sizeof(TK)); -} - - - -// calculate DMR from DMK using blas for multi-k calculation -template -void DensityMatrix_Tools::cal_DMR( - const DensityMatrix &dm, - std::vector*> &dmR_out, - const int ik_in) -{ - ModuleBase::TITLE("DensityMatrix", "cal_DMR"); - - // To check whether DMR has been initialized - assert(dmR_out.size()==dm._nspin && "DMR has not been initialized!"); - - ModuleBase::timer::start("DensityMatrix", "cal_DMR"); - const int ld_hk = dm._paraV->nrow; - for (int is = 1; is <= dm._nspin; ++is) - { - const int ik_begin = dm._nk * (is - 1); // jump dm._nk for spin_down if nspin==2 - hamilt::HContainer*const target_DMR = dmR_out[is - 1]; - // set zero since this function is called in every scf step - target_DMR->set_zero(); - #ifdef _OPENMP - #pragma omp parallel for schedule(dynamic) - #endif - for (int i = 0; i < target_DMR->size_atom_pairs(); ++i) - { - hamilt::AtomPair& target_ap = target_DMR->get_atom_pair(i); - const int iat1 = target_ap.get_atom_i(); - const int iat2 = target_ap.get_atom_j(); - // get global indexes of whole matrix for each atom in this process - const int row_ap = dm._paraV->atom_begin_row[iat1]; - const int col_ap = dm._paraV->atom_begin_col[iat2]; - const int row_size = dm._paraV->get_nrow_atom(iat1); - const int col_size = dm._paraV->get_ncol_atom(iat2); - const int mat_size = row_size * col_size; - const int R_size = target_ap.get_R_size(); - assert(row_ap != -1 && col_ap != -1 && "Atom-pair not belong this process"); - - // calculate kphase and target_mat_ptr - std::vector> kphase_vec(dm._nk, std::vector(R_size)); - std::vector target_DMR_mat_vec(R_size); - for(int iR = 0; iR < R_size; ++iR) - { - const ModuleBase::Vector3 R_index = target_ap.get_R_index(iR); - hamilt::BaseMatrix*const target_mat = target_ap.find_matrix(R_index); - #ifdef __DEBUG - if (target_mat == nullptr) - { - std::cout << "target_mat is nullptr" << std::endl; - continue; - } - #endif - target_DMR_mat_vec[iR] = target_mat->get_pointer(); - for(int ik = 0; ik < dm._nk; ++ik) - { - if(ik_in >= 0 && ik_in != ik) { continue; } - // cal k_phase - // if TK==std::complex, kphase is e^{ikR} - const ModuleBase::Vector3 dR(R_index[0], R_index[1], R_index[2]); - const double arg = (dm._kvec_d[ik] * dR) * ModuleBase::TWO_PI; - double sinp, cosp; - ModuleBase::libm::sincos(arg, &sinp, &cosp); - kphase_vec[ik][iR] = TK(cosp, sinp); - } - } - - std::vector DMK_mat_trans(mat_size); - std::vector tmp_DMR( (PARAM.inp.nspin==4) ? mat_size*R_size : 0); - for(int ik = 0; ik < dm._nk; ++ik) - { - if(ik_in >= 0 && ik_in != ik) { continue; } - // copy column-major DMK to row-major DMK_mat_trans (for the purpose of computational efficiency) - const TK*const DMK_mat_ptr - = dm._DMK[ik + ik_begin].data() - + col_ap * dm._paraV->nrow + row_ap; - for(int icol = 0; icol < col_size; ++icol) { - for(int irow = 0; irow < row_size; ++irow) { - DMK_mat_trans[irow * col_size + icol] = DMK_mat_ptr[icol * ld_hk + irow]; - }} - - // if nspin != 4, fill DMR - // if nspin == 4, fill tmp_DMR - for(int iR = 0; iR < R_size; ++iR) - { - // (kr+i*ki) * (Dr+i*Di) = (kr*Dr-ki*Di) + i*(kr*Di+ki*Dr) - const TK kphase = kphase_vec[ik][iR]; - if(PARAM.inp.nspin != 4) // only save real kr*Dr-ki*Di - { - func_exp_mul_dmk(kphase, DMK_mat_trans, target_DMR_mat_vec[iR]); - } else if(PARAM.inp.nspin == 4) - { - BlasConnector::axpy(mat_size, - kphase, - DMK_mat_trans.data(), - 1, - &tmp_DMR[iR * mat_size], - 1); - } - } - } - - // if nspin == 4 - // copy tmp_DMR to fill target_DMR - if(PARAM.inp.nspin == 4) - { - // step_trace ={0, 1, local_col, local_col+1} for NSPIN=4 - int step_trace[4]{}; - constexpr int npol = 2; - for (int is = 0; is < npol; is++) { - for (int is2 = 0; is2 < npol; is2++) { - step_trace[is * npol + is2] = target_ap.get_col_size() * is + is2; - }} - - TK tmp[4]{}; - for(int iR = 0; iR < R_size; ++iR) - { - const TK* tmp_DMR_mat = &tmp_DMR[iR * mat_size]; - TR_out* target_DMR_mat = target_DMR_mat_vec[iR]; - for (int irow = 0; irow < row_size; irow += 2) - { - for (int icol = 0; icol < col_size; icol += 2) - { - // catch the 4 spin component value of one orbital pair - tmp[0] = tmp_DMR_mat[icol + step_trace[0]]; - tmp[1] = tmp_DMR_mat[icol + step_trace[1]]; - tmp[2] = tmp_DMR_mat[icol + step_trace[2]]; - tmp[3] = tmp_DMR_mat[icol + step_trace[3]]; - - // transfer to Pauli matrix, save them back to the target_DMR_mat - func_xyz_to_updown(tmp, icol, step_trace, target_DMR_mat); - } - tmp_DMR_mat += col_size * 2; - target_DMR_mat += col_size * 2; - } - } - } - } + this->dmk[ik].resize(this->pv->get_row_size() * this->pv->get_col_size()); } - ModuleBase::timer::end("DensityMatrix", "cal_DMR"); + ModuleBase::Memory::record("DensityMatrix::DMK", this->dmk.size() * this->dmk[0].size() * sizeof(TK)); } -template <> -void DensityMatrix, double>::cal_DMR(const int ik_in) -{ - DensityMatrix_Tools::cal_DMR(*this, this->_DMR, ik_in); - this->_dmr_ready = true; -} - -template <> -void DensityMatrix, std::complex>::cal_DMR(const int ik_in) -{ - DensityMatrix_Tools::cal_DMR(*this, this->_DMR, ik_in); - this->_dmr_ready = true; -} - - - -// calculate DMR from DMK using blas for multi-k calculation -template -void DensityMatrix_Tools::cal_DMR_td( - const DensityMatrix &dm, - std::vector*> &dmR_out, - const std::map, std::complex>& phase_hybrid, - const ModuleBase::Vector3 At, - const int ik_in) -{ - ModuleBase::TITLE("DensityMatrix", "cal_DMR_td"); - // To check whether DMR has been initialized - assert(dmR_out.size()==dm._nspin && "DMR has not been initialized!"); - - ModuleBase::timer::start("DensityMatrix", "cal_DMR_td"); - const int ld_hk = dm._paraV->nrow; - for (int is = 1; is <= dm._nspin; ++is) - { - const int ik_begin = dm._nk * (is - 1); // jump dm._nk for spin_down if nspin==2 - hamilt::HContainer*const target_DMR = dmR_out[is - 1]; - // set zero since this function is called in every scf step - target_DMR->set_zero(); - #ifdef _OPENMP - #pragma omp parallel for schedule(dynamic) - #endif - for (int i = 0; i < target_DMR->size_atom_pairs(); ++i) - { - hamilt::AtomPair& target_ap = target_DMR->get_atom_pair(i); - const int iat1 = target_ap.get_atom_i(); - const int iat2 = target_ap.get_atom_j(); - // get global indexes of whole matrix for each atom in this process - const int row_ap = dm._paraV->atom_begin_row[iat1]; - const int col_ap = dm._paraV->atom_begin_col[iat2]; - const int row_size = dm._paraV->get_nrow_atom(iat1); - const int col_size = dm._paraV->get_ncol_atom(iat2); - const int mat_size = row_size * col_size; - const int R_size = target_ap.get_R_size(); - assert(row_ap != -1 && col_ap != -1 && "Atom-pair not belong this process"); - - // calculate kphase and target_mat_ptr - std::vector> kphase_vec(dm._nk, std::vector(R_size)); - std::vector target_DMR_mat_vec(R_size); - for(int iR = 0; iR < R_size; ++iR) - { - const ModuleBase::Vector3 R_index = target_ap.get_R_index(iR); - hamilt::BaseMatrix*const target_mat = target_ap.find_matrix(R_index); - #ifdef __DEBUG - if (target_mat == nullptr) - { - std::cout << "target_mat is nullptr" << std::endl; - continue; - } - #endif - target_DMR_mat_vec[iR] = target_mat->get_pointer(); - for(int ik = 0; ik < dm._nk; ++ik) - { - if(ik_in >= 0 && ik_in != ik) { continue; } - // cal k_phase - // if TK==std::complex, kphase is e^{ikR} - const ModuleBase::Vector3 dR(R_index[0], R_index[1], R_index[2]); - const double arg = (dm._kvec_d[ik] * dR) * ModuleBase::TWO_PI; - double sinp, cosp; - ModuleBase::libm::sincos(arg, &sinp, &cosp); - kphase_vec[ik][iR] = TK(cosp, sinp); - if(PARAM.inp.td_stype==2) - { - //phase for hybrid gauge tddft - kphase_vec[ik][iR] *= phase_hybrid.at(R_index); - } - } - } - - std::vector DMK_mat_trans(mat_size); - std::vector tmp_DMR( (PARAM.inp.nspin==4) ? mat_size*R_size : 0); - for(int ik = 0; ik < dm._nk; ++ik) - { - if(ik_in >= 0 && ik_in != ik) { continue; } - // copy column-major DMK to row-major DMK_mat_trans (for the purpose of computational efficiency) - const TK*const DMK_mat_ptr - = dm._DMK[ik + ik_begin].data() - + col_ap * dm._paraV->nrow + row_ap; - for(int icol = 0; icol < col_size; ++icol) { - for(int irow = 0; irow < row_size; ++irow) { - DMK_mat_trans[irow * col_size + icol] = DMK_mat_ptr[icol * ld_hk + irow]; - }} - - // if nspin != 4, fill DMR - // if nspin == 4, fill tmp_DMR - for(int iR = 0; iR < R_size; ++iR) - { - // (kr+i*ki) * (Dr+i*Di) = (kr*Dr-ki*Di) + i*(kr*Di+ki*Dr) - const TK kphase = kphase_vec[ik][iR]; - if(PARAM.inp.nspin != 4) // only save real kr*Dr-ki*Di - { - func_exp_mul_dmk(kphase, DMK_mat_trans, target_DMR_mat_vec[iR]); - } else if(PARAM.inp.nspin == 4) - { - BlasConnector::axpy(mat_size, - kphase, - DMK_mat_trans.data(), - 1, - &tmp_DMR[iR * mat_size], - 1); - } - } - } - - // if nspin == 4 - // copy tmp_DMR to fill target_DMR - if(PARAM.inp.nspin == 4) - { - // step_trace ={0, 1, local_col, local_col+1} for NSPIN=4 - int step_trace[4]{}; - constexpr int npol = 2; - for (int is = 0; is < npol; is++) { - for (int is2 = 0; is2 < npol; is2++) { - step_trace[is * npol + is2] = target_ap.get_col_size() * is + is2; - }} - - TK tmp[4]{}; - for(int iR = 0; iR < R_size; ++iR) - { - const TK* tmp_DMR_mat = &tmp_DMR[iR * mat_size]; - TR_out* target_DMR_mat = target_DMR_mat_vec[iR]; - for (int irow = 0; irow < row_size; irow += 2) - { - for (int icol = 0; icol < col_size; icol += 2) - { - // catch the 4 spin component value of one orbital pair - tmp[0] = tmp_DMR_mat[icol + step_trace[0]]; - tmp[1] = tmp_DMR_mat[icol + step_trace[1]]; - tmp[2] = tmp_DMR_mat[icol + step_trace[2]]; - tmp[3] = tmp_DMR_mat[icol + step_trace[3]]; - - // transfer to Pauli matrix, save them back to the target_DMR_mat - func_xyz_to_updown(tmp, icol, step_trace, target_DMR_mat); - } - tmp_DMR_mat += col_size * 2; - target_DMR_mat += col_size * 2; - } - } - } - } - } - ModuleBase::timer::end("DensityMatrix", "cal_DMR_td"); -} -template <> -void DensityMatrix::cal_DMR_td(const std::map, std::complex>& phase_hybrid, const ModuleBase::Vector3 At, const int ik_in) -{ - return; -} -template <> -void DensityMatrix, double>::cal_DMR_td(const std::map, std::complex>& phase_hybrid, const ModuleBase::Vector3 At, const int ik_in) -{ - DensityMatrix_Tools::cal_DMR_td(*this, this->_DMR, phase_hybrid, At, ik_in); - this->_dmr_ready = true; -} - -template <> -void DensityMatrix, std::complex>::cal_DMR_td(const std::map, std::complex>& phase_hybrid, const ModuleBase::Vector3 At, const int ik_in) -{ - DensityMatrix_Tools::cal_DMR_td(*this, this->_DMR, phase_hybrid, At, ik_in); - this->_dmr_ready = true; -} - - - -// calculate DMR from DMK using blas for multi-k calculation -template -void DensityMatrix_Tools::cal_DMR_full( - const DensityMatrix &dm, - hamilt::HContainer* dmR_out, - const int ik_in) -{ - ModuleBase::TITLE("DensityMatrix", "cal_DMR_full"); - - ModuleBase::timer::start("DensityMatrix", "cal_DMR_full"); - const int ld_hk = dm._paraV->nrow; - hamilt::HContainer* target_DMR = dmR_out; - // set zero since this function is called in every scf step - target_DMR->set_zero(); - #ifdef _OPENMP - #pragma omp parallel for schedule(dynamic) - #endif - for (int i = 0; i < target_DMR->size_atom_pairs(); ++i) - { - hamilt::AtomPair& target_ap = target_DMR->get_atom_pair(i); - const int iat1 = target_ap.get_atom_i(); - const int iat2 = target_ap.get_atom_j(); - // get global indexes of whole matrix for each atom in this process - const int row_ap = dm._paraV->atom_begin_row[iat1]; - const int col_ap = dm._paraV->atom_begin_col[iat2]; - const int row_size = dm._paraV->get_nrow_atom(iat1); - const int col_size = dm._paraV->get_ncol_atom(iat2); - const int mat_size = row_size * col_size; - const int R_size = target_ap.get_R_size(); - assert(row_ap != -1 && col_ap != -1 && "Atom-pair not belong this process"); - - // calculate kphase and target_mat_ptr - std::vector> kphase_vec(dm._nk, std::vector(R_size)); - std::vector target_DMR_mat_vec(R_size); - for(int iR = 0; iR < R_size; ++iR) - { - const ModuleBase::Vector3 R_index = target_ap.get_R_index(iR); - hamilt::BaseMatrix*const target_mat = target_ap.find_matrix(R_index); - #ifdef __DEBUG - if (target_mat == nullptr) - { - std::cout << "target_mat is nullptr" << std::endl; - continue; - } - #endif - target_DMR_mat_vec[iR] = target_mat->get_pointer(); - for(int ik = 0; ik < dm._nk; ++ik) - { - if(ik_in >= 0 && ik_in != ik) { continue; } - // cal k_phase - // if TK==std::complex, kphase is e^{ikR} - const ModuleBase::Vector3 dR(R_index[0], R_index[1], R_index[2]); - const double arg = (dm._kvec_d[ik] * dR) * ModuleBase::TWO_PI; - double sinp, cosp; - ModuleBase::libm::sincos(arg, &sinp, &cosp); - kphase_vec[ik][iR] = TK(cosp, sinp); - } - } - - std::vector DMK_mat_trans(mat_size); - for(int ik = 0; ik < dm._nk; ++ik) - { - if(ik_in >= 0 && ik_in != ik) { continue; } - // copy column-major DMK to row-major DMK_mat_trans (for the purpose of computational efficiency) - const TK*const DMK_mat_ptr - = dm._DMK[ik].data() - + col_ap * dm._paraV->nrow + row_ap; - for(int icol = 0; icol < col_size; ++icol) { - for(int irow = 0; irow < row_size; ++irow) { - DMK_mat_trans[irow * col_size + icol] = DMK_mat_ptr[icol * ld_hk + irow]; - }} - - for(int iR = 0; iR < R_size; ++iR) - { - const TK kphase = kphase_vec[ik][iR]; - BlasConnector::axpy(mat_size, - kphase, - DMK_mat_trans.data(), - 1, - target_DMR_mat_vec[iR], - 1); - } - } - } - ModuleBase::timer::end("DensityMatrix", "cal_DMR_full"); -} - -template <> -void DensityMatrix::cal_DMR_full( - hamilt::HContainer>* dmR_out, - const int ik_in) const{} -template <> -void DensityMatrix, double>::cal_DMR_full( - hamilt::HContainer>* dmR_out, - const int ik_in) const -{ - DensityMatrix_Tools::cal_DMR_full(*this, dmR_out, ik_in); -} - - - -// calculate DMR from DMK using blas for gamma-only calculation -template <> -void DensityMatrix::cal_DMR(const int ik_in) -{ - ModuleBase::TITLE("DensityMatrix", "cal_DMR"); - using TK = double; - using TR = double; - - assert(ik_in == -1 || ik_in == 0); - assert(this->_nk == 1); - - // To check whether DMR has been initialized - assert(this->_DMR.size()==this->_nspin && "DMR has not been initialized!"); - - ModuleBase::timer::start("DensityMatrix", "cal_DMR"); - const int ld_hk = this->_paraV->nrow; - for (int is = 1; is <= this->_nspin; ++is) - { - const int ik_begin = this->_nk * (is - 1); // jump this->_nk for spin_down if nspin==2 - hamilt::HContainer*const target_DMR = this->_DMR[is - 1]; - // set zero since this function is called in every scf step - target_DMR->set_zero(); - #ifdef _OPENMP - #pragma omp parallel for schedule(dynamic) - #endif - for (int i = 0; i < target_DMR->size_atom_pairs(); ++i) - { - hamilt::AtomPair& target_ap = target_DMR->get_atom_pair(i); - const int iat1 = target_ap.get_atom_i(); - const int iat2 = target_ap.get_atom_j(); - // get global indexes of whole matrix for each atom in this process - const int row_ap = this->_paraV->atom_begin_row[iat1]; - const int col_ap = this->_paraV->atom_begin_col[iat2]; - const int row_size = this->_paraV->get_nrow_atom(iat1); - const int col_size = this->_paraV->get_ncol_atom(iat2); - const int R_size = target_ap.get_R_size(); - assert(row_ap != -1 && col_ap != -1 && "Atom-pair not belong this process"); - assert(R_size == 1); - const ModuleBase::Vector3 R_index = target_ap.get_R_index(0); - assert(R_index.x == 0 && R_index.y == 0 && R_index.z == 0); - hamilt::BaseMatrix*const target_mat = target_ap.find_matrix(R_index); - #ifdef __DEBUG - if (target_mat == nullptr) - { - std::cout << "target_mat is nullptr" << std::endl; - continue; - } - #endif - // k index - constexpr TK kphase = 1; - // transpose DMK col=>row - const TK* DMK_mat_ptr - = this->_DMK[0 + ik_begin].data() - + col_ap * this->_paraV->nrow + row_ap; - // set DMR element - TR* target_DMR_ptr = target_mat->get_pointer(); - for (int mu = 0; mu < row_size; ++mu) - { - BlasConnector::axpy(col_size, - kphase, - DMK_mat_ptr, - ld_hk, - target_DMR_ptr, - 1); - DMK_mat_ptr += 1; - target_DMR_ptr += col_size; - } - } - } - ModuleBase::timer::end("DensityMatrix", "cal_DMR"); - this->_dmr_ready = true; -} - - // switch_dmr template void DensityMatrix::switch_dmr(const int mode) { ModuleBase::TITLE("DensityMatrix", "switch_dmr"); - if (this->_nspin != 2) - { - return; - } - else + ModuleBase::timer::start("DensityMatrix", "switch_dmr"); + if (this->spin_mult == 2) { - ModuleBase::timer::start("DensityMatrix", "switch_dmr"); switch(mode) { case 0: // switch to original density matrix - if (this->dmr_tmp_ != nullptr && this->dmr_origin_.size() != 0) + if (!this->dmr_tmp.empty() && this->dmr_origin.size() != 0) { - this->_DMR[0]->allocate(this->dmr_origin_.data(), false); - delete[] this->dmr_tmp_; - this->dmr_tmp_ = nullptr; + this->dmr[0]->allocate(this->dmr_origin.data(), false); + this->dmr_tmp.clear(); } // else: do nothing break; case 1: // switch to total magnetization density matrix, dmr_up + dmr_down - if(this->dmr_tmp_ == nullptr) + if(this->dmr_tmp.empty()) { - const size_t size = this->_DMR[0]->get_nnr(); - this->dmr_tmp_ = new TR[size]; - this->dmr_origin_.resize(size); + const size_t size = this->dmr[0]->get_nnr(); + this->dmr_tmp.resize(size); + this->dmr_origin.resize(size); for (int i = 0; i < size; ++i) { - this->dmr_origin_[i] = this->_DMR[0]->get_wrapper()[i]; - this->dmr_tmp_[i] = this->dmr_origin_[i] + this->_DMR[1]->get_wrapper()[i]; + this->dmr_origin[i] = this->dmr[0]->get_wrapper()[i]; + this->dmr_tmp[i] = this->dmr_origin[i] + this->dmr[1]->get_wrapper()[i]; } - this->_DMR[0]->allocate(this->dmr_tmp_, false); + this->dmr[0]->allocate(this->dmr_tmp.data(), false); } else { - const size_t size = this->_DMR[0]->get_nnr(); + const size_t size = this->dmr[0]->get_nnr(); for (int i = 0; i < size; ++i) { - this->dmr_tmp_[i] = this->dmr_origin_[i] + this->_DMR[1]->get_wrapper()[i]; + this->dmr_tmp[i] = this->dmr_origin[i] + this->dmr[1]->get_wrapper()[i]; } } break; case 2: // switch to magnetization density matrix, dmr_up - dmr_down - if(this->dmr_tmp_ == nullptr) + if(this->dmr_tmp.empty()) { - const size_t size = this->_DMR[0]->get_nnr(); - this->dmr_tmp_ = new TR[size]; - this->dmr_origin_.resize(size); + const size_t size = this->dmr[0]->get_nnr(); + this->dmr_tmp.resize(size); + this->dmr_origin.resize(size); for (int i = 0; i < size; ++i) { - this->dmr_origin_[i] = this->_DMR[0]->get_wrapper()[i]; - this->dmr_tmp_[i] = this->dmr_origin_[i] - this->_DMR[1]->get_wrapper()[i]; + this->dmr_origin[i] = this->dmr[0]->get_wrapper()[i]; + this->dmr_tmp[i] = this->dmr_origin[i] - this->dmr[1]->get_wrapper()[i]; } - this->_DMR[0]->allocate(this->dmr_tmp_, false); + this->dmr[0]->allocate(this->dmr_tmp.data(), false); } else { - const size_t size = this->_DMR[0]->get_nnr(); + const size_t size = this->dmr[0]->get_nnr(); for (int i = 0; i < size; ++i) { - this->dmr_tmp_[i] = this->dmr_origin_[i] - this->_DMR[1]->get_wrapper()[i]; + this->dmr_tmp[i] = this->dmr_origin[i] - this->dmr[1]->get_wrapper()[i]; } } break; default: ModuleBase::WARNING_QUIT("density_matrix.cpp", "Unknown mode in switch_dmr"); } - ModuleBase::timer::end("DensityMatrix", "switch_dmr"); } -} - - - -template <> -void DensityMatrix_Tools::func_exp_mul_dmk(const std::complex kphase, const std::vector> &DMK_mat_trans, double* target_DMR_mat) -{ - const std::size_t mat_size = DMK_mat_trans.size(); - for(std::size_t i = 0; i < mat_size; i++) - { - target_DMR_mat[i] - += kphase.real() * DMK_mat_trans[i].real() - - kphase.imag() * DMK_mat_trans[i].imag(); - } -} - -template <> -void DensityMatrix_Tools::func_exp_mul_dmk>(const std::complex kphase, const std::vector> &DMK_mat_trans, std::complex* target_DMR_mat) -{ - BlasConnector::axpy(DMK_mat_trans.size(), - kphase, - DMK_mat_trans.data(), - 1, - target_DMR_mat, - 1); -} - -template <> -void DensityMatrix_Tools::func_xyz_to_updown(const std::complex tmp[4], const int icol, const int step_trace[4], double* target_DMR_mat) -{ - target_DMR_mat[icol + step_trace[0]] = tmp[0].real() + tmp[3].real(); // rho_0 = (rho_upup + rho_downdown).real() - target_DMR_mat[icol + step_trace[1]] = tmp[1].real() + tmp[2].real(); // rho_x = (rho_updown + rho_downup).real() - // rho_y: the stored DM block is the complex conjugate of the physical 1-RDM P (cal_dm_psi builds - // DM_{ab}=sum conj(c_a) c_b = conj(P), so tmp[1]=DM_{ud}=conj(P_{ud})). Extracting m_y from the - // CONJUGATED block therefore carries the opposite sign of the bare-textbook formula; m_x/m_z read - // Re() and are conjugation-invariant. Using the bare formula (PR #7664) sign-flips m_y and quenches - // in-plane non-collinear moments (e.g. Mn3Sn 120-deg AFM); see issue #7831. - target_DMR_mat[icol + step_trace[2]] = tmp[1].imag() - tmp[2].imag(); // rho_y = Im(P_updown) - Im(P_downup) - target_DMR_mat[icol + step_trace[3]] = tmp[0].real() - tmp[3].real(); // rho_z = (rho_upup - rho_downdown).real() -} - -template <> -void DensityMatrix_Tools::func_xyz_to_updown>(const std::complex tmp[4], const int icol, const int step_trace[4], std::complex* target_DMR_mat) -{ - target_DMR_mat[icol + step_trace[0]] = tmp[0] + tmp[3]; // rho_0 = (rho_upup + rho_downdown) - target_DMR_mat[icol + step_trace[1]] = tmp[1] + tmp[2]; // rho_x = (rho_updown + rho_downup) - // rho_y sign accounts for the conjugated stored DM block (conj(P)); see the specialization above. - target_DMR_mat[icol + step_trace[2]] = -ModuleBase::IMAG_UNIT * (tmp[1] - tmp[2]); // rho_y = -i*(rho_updown - rho_downup) - target_DMR_mat[icol + step_trace[3]] = tmp[0] - tmp[3]; // rho_z = (rho_upup - rho_downdown) + ModuleBase::timer::end("DensityMatrix", "switch_dmr"); } @@ -686,4 +147,4 @@ template class DensityMatrix; // Gamma-Only case template class DensityMatrix, double>; // Multi-k case template class DensityMatrix, std::complex>; // For EXX in future -} // namespace elecstate +} // namespace module_dm diff --git a/source/source_estate/module_dm/density_matrix.h b/source/source_estate/module_dm/density_matrix.h index a8f0dd4fd2c..811d85349d9 100644 --- a/source/source_estate/module_dm/density_matrix.h +++ b/source/source_estate/module_dm/density_matrix.h @@ -1,123 +1,230 @@ #ifndef DENSITY_MATRIX_H #define DENSITY_MATRIX_H +#include +#include #include +#include +#include "source_base/vector3.h" #include "source_cell/module_neighbor/sltk_grid_driver.h" -#include "source_lcao/record_adj.h" +#include "source_cell/record_adj.h" #include "source_hamilt/module_hcontainer/hcontainer.h" -namespace elecstate +namespace module_dm { /** - * @brief DensityMatrix Class - * = for Gamma-only calculation - * = ,double> for multi-k calculation + * @brief map a real/complex type to the opposite one + * ShiftRealComplex::type = std::complex + * ShiftRealComplex>::type = double */ -template struct ShiftRealComplex +template struct ShiftRealComplex { using type = void; }; -template<> -struct ShiftRealComplex +template <> +struct ShiftRealComplex { - using type = std::complex; + using type = std::complex; }; -template<> -struct ShiftRealComplex> +template <> +struct ShiftRealComplex> { - using type = double; + using type = double; }; - +/** + * @brief DensityMatrix Class + * = for Gamma-only calculation + * = ,double> for multi-k calculation + */ template class DensityMatrix; -// DensityMatrix,TR>::cal_DMR() is illegal in C++, so DensityMatrix_Tools is used instead. -namespace DensityMatrix_Tools -{ +// DensityMatrix,TR>::cal_dmr() is illegal in C++, so module_dm is used instead. template - extern void cal_DMR( - const DensityMatrix &dm, + extern void cal_dmr( + DensityMatrix &dm, std::vector*> &dmR_out, const int ik_in); template - extern void cal_DMR_td( - const DensityMatrix &dm, + extern void cal_dmr_td( + DensityMatrix &dm, std::vector*> &dmR_out, const std::map, std::complex>& phase_hybrid, const ModuleBase::Vector3 At, const int ik_in); template - extern void cal_DMR_full( - const DensityMatrix &dm, + extern void cal_dmr_full( + const DensityMatrix &dm, hamilt::HContainer* dmR_out, const int ik_in); + /** + * @brief shared inner loop of cal_dmr / cal_dmr_td: for each spin channel, + * zero the DMR HContainer and accumulate kphase * DMK into DMR blocks. + * Pass an empty phase_hybrid map for the non-TD (cal_dmr) case. + */ + template + extern void accumulate_dmr( + DensityMatrix &dm, + std::vector*> &dmR_out, + const std::map, std::complex>& phase_hybrid, + const int ik_in, + const char* func_name); + template - extern void func_exp_mul_dmk(const std::complex kphase, const std::vector> &DMK_mat_trans, TR* target_DMR_mat); + extern void exp_mul_dmk(const std::complex kphase, + const std::vector>& dmk_row, + TR* dmr_mat); template - extern void func_xyz_to_updown(const std::complex tmp[4], const int icol, const int step_trace[4], TR* target_DMR_mat); -} + extern void xyz_to_updown(const std::complex spin_block[4], + const int icol, + const int spin_stride[4], + TR* dmr_mat); + /** + * @brief geometry of one atom-pair sub-block within the global DMK matrix + * row0/col0: global index of the block's top-left element in the 2D block-cyclic DMK + * nrows/ncols: orbital dimensions of the two atoms + */ + struct DmrBlock + { + int row0; + int col0; + int nrows; + int ncols; + int size() const { return nrows * ncols; } + }; + + /// @brief extract the block geometry for atom pair (iat1, iat2) from the parallel orbitals layout + DmrBlock get_dmr_block(const Parallel_Orbitals* pv, const int iat1, const int iat2); + + /** + * @brief precompute k-phase factors e^{ikR} and collect DMR block pointers for one atom pair + * @param atom_pair the atom pair whose R-vectors and matrices are used + * @param kvec_d direct coordinates of k-points + * @param nk number of k-points + * @param phase_hybrid additional hybrid-gauge phase per R (empty map = no extra phase) + * @param kphase_vec output: kphase_vec[ik][iR] + * @param dmr_mats output: dmr_mats[iR] points to the DMR block for R-vector iR + */ + template + extern void build_kphase(hamilt::AtomPair& atom_pair, + const std::vector>& kvec_d, + const int nk, + const std::map, std::complex>& phase_hybrid, + std::vector>& kphase_vec, + std::vector& dmr_mats); + + /// @brief transpose a col-major DMK sub-block into row-major order + template + extern void transpose_dmk_block(const TK* dmk_col_major, + const int ld_hk, + const DmrBlock& block, + TK* dmk_row); + + /** + * @brief nspin=1/2: accumulate Re(kphase * DMK) into DMR blocks + * + * Formula: DMR_ij(R) += Re[ e^{ik·R} * DMK_ij(k) ] + * If ik_in >= 0, only that k-point contributes; if ik_in < 0, sum over all k-points. + */ + template + extern void add_dmr_real(const DensityMatrix& dm, + const DmrBlock& block, + const int ik_begin, + const std::vector>& kphase_vec, + const int ld_hk, + const int ik_in, + std::vector& dmr_mats); + + /** + * @brief nspin==4 (SOC): accumulate k-phase * DMK into a per-R complex buffer, + * then transform 2x2 spin blocks from (upup, updown, downup, downdown) to + * (rho_0, rho_x, rho_y, rho_z) via xyz_to_updown. + * + * Formula: + * S_ij(R) = sum_k e^{ik·R} * DMK_ij(k) + * rho_0 = rho_upup + rho_downdown + * rho_x = rho_updown + rho_downup + * rho_y = Im(rho_updown) - Im(rho_downup) (sign for conjugated stored DM) + * rho_z = rho_upup - rho_downdown + * Each orbital corresponds to a 2x2 spin block, so rows/cols step by 2. + * If ik_in >= 0, only that k-point contributes; if ik_in < 0, sum over all k-points. + */ + template + extern void add_dmr_soc(const DensityMatrix& dm, + const DmrBlock& block, + const int ik_begin, + const std::vector>& kphase_vec, + const int ld_hk, + const int ik_in, + const int col_stride, + std::vector& dmr_mats); template class DensityMatrix { - using TRShift = typename ShiftRealComplex::type; + using TRShift = typename ShiftRealComplex::type; - public: - /** - * @brief Destructor of class DensityMatrix - */ - ~DensityMatrix(); + public: + /** + * @brief Destructor of class DensityMatrix + */ + ~DensityMatrix(); /** * @brief Constructor of class DensityMatrix for multi-k calculation - * @param _paraV pointer of Parallel_Orbitals object - * @param nspin number of spin of the density matrix, set by user according to global nspin - * (usually {nspin_global -> nspin_dm} = {1->1, 2->2, 4->1}, but sometimes 2->1 like in LR-TDDFT) + * @param pv pointer of Parallel_Orbitals object + * @param spin_mult spin multiplicity used to size the DM: 1 for input nspin 1 or 4 + * (non-collinear k points are not doubled), 2 for input nspin 2 (LSDA up/down). + * This is NOT the physical nspin (1/2/4); it matches K_Vectors::spin_mult. * @param kvec_d direct coordinates of kpoints - * @param nk number of k-points, not always equal to K_Vectors::get_nks()/nspin_dm. + * @param nk number of k-points, not always equal to K_Vectors::get_nks()/spin_mult. * it will be set to kvec_d.size() if the value is invalid + * @param nspin the global physical nspin from INPUT (1/2/4); defaults to spin_mult for + * non-SOC cases where they coincide. Pass 4 explicitly for SOC/noncollinear + * calculations so that cal_dmr selects the spin-resolved (Pauli) branch. */ - DensityMatrix(const Parallel_Orbitals* _paraV, - const int nspin, - const std::vector>& kvec_d, - const int nk); + DensityMatrix(const Parallel_Orbitals* pv, + const int spin_mult, + const std::vector>& kvec_d, + const int nk, + const int nspin = 0); /** * @brief Constructor of class DensityMatrix for gamma-only calculation, where kvector is not required - * @param _paraV pointer of Parallel_Orbitals object - * @param nspin number of spin of the density matrix, set by user according to global nspin - * (usually {nspin_global -> nspin_dm} = {1->1, 2->2, 4->1}, but sometimes 2->1 like in LR-TDDFT) + * @param pv pointer of Parallel_Orbitals object + * @param spin_mult spin multiplicity of the density matrix (1 or 2); NOT the physical nspin + * @param nspin the global physical nspin from INPUT (1/2/4); defaults to spin_mult. */ - DensityMatrix(const Parallel_Orbitals* _paraV, const int nspin); + DensityMatrix(const Parallel_Orbitals* pv, const int spin_mult, const int nspin = 0); /** * @brief initialize density matrix DMR from UnitCell * @param GridD_in pointer of Grid_Driver object (used to find ajacent atoms) * @param ucell pointer of UnitCell object */ - void init_DMR(const Grid_Driver* GridD_in, const UnitCell* ucell); + void init_dmr(const Grid_Driver* GridD_in, const UnitCell* ucell); /** * @brief initialize density matrix DMR from UnitCell and RA * @param ra pointer of Record_adj object (used to find ajacent atoms) * @param ucell pointer of UnitCell object */ - void init_DMR(Record_adj& ra, const UnitCell* ucell); + void init_dmr(Record_adj& ra, const UnitCell* ucell); /** * @brief initialize density matrix DMR from another HContainer * now only support HContainer * @param _DMR_in pointer of another HContainer object */ - void init_DMR(const hamilt::HContainer& _DMR_in); + void init_dmr(const hamilt::HContainer& _DMR_in); /// @brief initialize density matrix DMR from another HContainer /// this is a temprory function for NSPIN=4 case @@ -125,22 +232,22 @@ class DensityMatrix /// would be refactor in the future /// @param _DMR_in // the old input type ``:HContainer` causes redefination error if TR = complex - void init_DMR(const hamilt::HContainer& _DMR_in); + void init_dmr(const hamilt::HContainer& _DMR_in); /** - * @brief set _DMK element directly + * @brief set dmk element directly * @param ispin spin index (1 - spin up (support SOC) or 2 - spin down) * @param ik k-point index * @param i row index * @param j column index * @param value value to be set */ - void set_DMK(const int ispin, const int ik, const int i, const int j, const TK value); + void set_dmk(const int ispin, const int ik, const int i, const int j, const TK value); /** - * @brief set _DMK element to zero + * @brief set dmk element to zero */ - void set_DMK_zero(); + void set_dmk_zero(); /** * @brief get a matrix element of density matrix dm(k) @@ -150,75 +257,89 @@ class DensityMatrix * @param j column index * @return T a matrix element of density matrix dm(k) */ - TK get_DMK(const int ispin, const int ik, const int i, const int j) const; + TK get_dmk(const int ispin, const int ik, const int i, const int j) const; /** * @brief get total number of k-points of density matrix dm(k) */ - int get_DMK_nks() const; - int get_DMK_size() const; + int get_dmk_nks() const; + int get_dmk_size() const; /** * @brief get number of rows of density matrix dm(k) */ - int get_DMK_nrow() const; + int get_dmk_nrow() const; /** * @brief get number of columns of density matrix dm(k) */ - int get_DMK_ncol() const; + int get_dmk_ncol() const; /** * @brief get pointer of DMR * @param ispin spin index (1 - spin up (support SOC) or 2 - spin down) * @return HContainer* pointer of DMR */ - hamilt::HContainer* get_DMR_pointer(const int ispin) const; + hamilt::HContainer* get_dmr_ptr(const int ispin) const; /** * @brief check whether the stored DMR is a valid density matrix calculated from DMK - * init_DMR() resets the flag and cal_DMR()/cal_DMR_td() set it, so a freshly + * init_dmr() resets the flag and cal_dmr()/cal_dmr_td() set it, so a freshly * allocated, zeroed or file-read DMR is reported as not ready until the first * wavefunction-derived calculation * @return true if DMR is ready for Hamiltonian construction */ - bool is_dmr_ready() const { return this->_dmr_ready; } + bool is_dmr_ready() const + { + return this->_dmr_ready; + } /** * @brief get pointer vector of DMR * @return HContainer* vector of DMR */ - const std::vector*>& get_DMR_vector() const {return this->_DMR;} - std::vector*>& get_DMR_vector() {return this->_DMR;} - - const std::vector>& get_DMR_save() const {return this->_DMR_save;} - std::vector>& get_DMR_save() {return this->_DMR_save;} + const std::vector*>& get_dmr_vec() const + { + return this->dmr; + } + std::vector*>& get_dmr_vec() + { + return this->dmr; + } + + const std::vector>& get_dmr_save() const + { + return this->dmr_save; + } + std::vector>& get_dmr_save() + { + return this->dmr_save; + } /** * @brief get pointer of DMK - * @param ik k-point index, which is the index of _DMK + * @param ik k-point index, which is the index of dmk * @return TK* pointer of DMK */ - TK* get_DMK_pointer(const int ik) const; + TK* get_dmk_ptr(const int ik) const; /** * @brief get pointer vector of DMK */ - const std::vector>& get_DMK_vector() const {return this->_DMK;} - std::vector>& get_DMK_vector() {return this->_DMK;} + const std::vector>& get_dmk_vec() const + { + return this->dmk; + } + std::vector>& get_dmk_vec() + { + return this->dmk; + } /** - * @brief set _DMK using a input TK* pointer + * @brief set dmk using a input TK* pointer * please make sure the size of TK* is correct */ - void set_DMK_pointer(const int ik, TK* DMK_in); - - /** - * @brief get pointer of paraV - */ - const Parallel_Orbitals* get_paraV_pointer() const {return this->_paraV;} - - const std::vector>& get_kvec_d() const { return this->_kvec_d; } + void set_dmk_ptr(const int ik, TK* DMK_in); /** * @brief calculate density matrix DMR from dm(k) using blas::axpy @@ -226,15 +347,17 @@ class DensityMatrix * if ik_in < 0, calculate all k-points * if ik_in >= 0, calculate only one k-point without summing over k-points */ - void cal_DMR(const int ik_in = -1); + void cal_dmr(const int ik_in); /** * @brief calculate density matrix DMR with additional vector potential phase, used for hybrid gauge tddft * @param ik_in * if ik_in < 0, calculate all k-points - * if ik_in >= 0, calculate only one k-point without summing over k-points + * if ik_in >= 0, calculate only one k-point */ - void cal_DMR_td(const std::map, std::complex>& phase_hybrid, const ModuleBase::Vector3 At, const int ik_in = -1); + void cal_dmr_td(const std::map, std::complex>& phase_hybrid, + const ModuleBase::Vector3 At, + const int ik_in); /** * @brief calculate complex density matrix DMR with both real and imaginary part for noncollinear-spin calculation @@ -242,9 +365,9 @@ class DensityMatrix * @param dmR_out pointer of HContainer object to store the calculated complex DMR * @param ik_in * if ik_in < 0, calculate all k-points - * if ik_in >= 0, calculate only one k-point without summing over k-points + * if ik_in >= 0, calculate only one k-point */ - void cal_DMR_full(hamilt::HContainer>* dmR_out, const int ik_in = -1) const; + void cal_dmr_full(hamilt::HContainer>* dmR_out, const int ik_in) const; /** * @brief (Only nspin=2) switch DMR to total density matrix or magnetization density matrix @@ -253,62 +376,50 @@ class DensityMatrix void switch_dmr(const int mode); /** - * @brief write density matrix dm(ik) into *.dmk - * @param directory directory of *.dmk files - * @param ispin spin index (1 - spin up (support SOC) or 2 - spin down) - * @param ik k-point index - */ - void write_DMK(const std::string directory, const int ispin, const int ik); - - /** - * @brief read *.dmk into density matrix dm(ik) - * @param directory directory of *.dmk files - * @param ispin spin index (1 - spin up (support SOC) or 2 - spin down) - * @param ik k-point index - */ - void read_DMK(const std::string directory, const int ispin, const int ik); - - /** - * @brief save _DMR into _DMR_save + * @brief save dmr into dmr_save */ - void save_DMR(); + void save_dmr(); - std::vector EDMK; // for TD-DFT + std::vector edmk; // for TD-DFT #ifdef __PEXSI /** * @brief EDM storage for PEXSI * used in MD calculation */ - std::vector pexsi_EDM; + std::vector edm_pexsi; #endif private: + /** + * @brief delete all HContainer objects in dmr and clear the vector + */ + void clear_dmr(); + /** * @brief HContainer for density matrix in real space for 2D parallelization * vector.size() = 1 for non-polarization and SOC * vector.size() = 2 for spin-polarization */ - std::vector*> _DMR; - std::vector> _DMR_save; + std::vector*> dmr; + std::vector> dmr_save; - /// @brief whether _DMR holds a density matrix calculated from DMK (reset by init_DMR, set by cal_DMR) + /// @brief whether dmr holds a density matrix calculated from DMK (reset by init_dmr, set by cal_dmr) bool _dmr_ready = false; /** - * @brief HContainer for density matrix in real space for gird parallelization - * vector.size() = 1 for non-polarization and SOC - * vector.size() = 2 for spin-polarization + * @brief HContainer for density matrix in real space for grid parallelization + * same size semantics as dmr */ - std::vector*> _DMR_grid; + std::vector*> dmr_grid; /** * @brief density matrix in k space, which is a vector[ik] - * DMK should be a [_nspin][_nk][i][j] matrix, - * whose size is _nspin * _nk * _paraV->get_nrow() * _paraV->get_ncol() + * DMK should be a [spin_mult][_nk][i][j] matrix, + * whose size is spin_mult * _nk * pv->get_nrow() * pv->get_ncol() */ - // std::vector _DMK; - std::vector> _DMK; + // std::vector dmk; + std::vector> dmk; /** * @brief K_Vectors object, which is used to get k-point information @@ -318,14 +429,23 @@ class DensityMatrix /** * @brief Parallel_Orbitals object, which contain all information of 2D block cyclic distribution */ - const Parallel_Orbitals* _paraV = nullptr; + const Parallel_Orbitals* pv = nullptr; + + /** + * @brief spin multiplicity used to size the density matrix (1 - none spin and SOC ; + * 2 - spin polarization). This is NOT the physical nspin (1/2/4); it matches + * K_Vectors::spin_mult. + */ + int spin_mult = 1; /** - * @brief spin-polarization index (1 - none spin and SOC ; 2 - spin polarization) - * Attention: this is not as same as GlovalV::NSPIN - * _nspin means the number of isolated spin-polarization states + * @brief the global physical nspin from INPUT (1/2/4). + * For SOC/noncollinear (nspin==4) the density matrix is stored with spin_mult==1 + * (a single 2x2 spin-block matrix), but cal_dmr/cal_dmr_td must still take the + * spin-resolved (Pauli) branch. spin_mult cannot distinguish this, so keep the + * global value here. Equals spin_mult for non-SOC cases. */ - int _nspin = 1; + int nspin = 1; /** * @brief real number of k-points @@ -335,14 +455,49 @@ class DensityMatrix int _nk = 0; /// temporary pointers for switch DMR, only used with nspin=2 - std::vector dmr_origin_; - TR* dmr_tmp_ = nullptr; + std::vector dmr_origin; + std::vector dmr_tmp; - friend void DensityMatrix_Tools::cal_DMR(const DensityMatrix &dm, std::vector*> &dmR_out, const int ik_in); - friend void DensityMatrix_Tools::cal_DMR_td(const DensityMatrix &dm, std::vector*> &dmR_out, const std::map, std::complex>& phase_hybrid, const ModuleBase::Vector3 At, const int ik_in); - friend void DensityMatrix_Tools::cal_DMR_full(const DensityMatrix &dm, hamilt::HContainer>* dmR_out, const int ik_in); + friend void module_dm::cal_dmr( + DensityMatrix& dm, + std::vector*>& dmR_out, + const int ik_in); + friend void module_dm::cal_dmr_td( + DensityMatrix& dm, + std::vector*>& dmR_out, + const std::map, std::complex>& phase_hybrid, + const ModuleBase::Vector3 At, + const int ik_in); + friend void module_dm::cal_dmr_full( + const DensityMatrix& dm, + hamilt::HContainer>* dmR_out, + const int ik_in); + friend void module_dm::accumulate_dmr( + DensityMatrix& dm, + std::vector*>& dmR_out, + const std::map, std::complex>& phase_hybrid, + const int ik_in, + const char* func_name); + friend void module_dm::add_dmr_real( + const DensityMatrix& dm, + const DmrBlock& block, + const int ik_begin, + const std::vector>& kphase_vec, + const int ld_hk, + const int ik_in, + std::vector& dmr_mats); + + friend void module_dm::add_dmr_soc( + const DensityMatrix& dm, + const DmrBlock& block, + const int ik_begin, + const std::vector>& kphase_vec, + const int ld_hk, + const int ik_in, + const int col_stride, + std::vector& dmr_mats); }; -} // namespace elecstate +} // namespace module_dm #endif diff --git a/source/source_estate/module_dm/density_matrix_io.cpp b/source/source_estate/module_dm/density_matrix_io.cpp deleted file mode 100644 index e62cbdb56b3..00000000000 --- a/source/source_estate/module_dm/density_matrix_io.cpp +++ /dev/null @@ -1,462 +0,0 @@ -#include "density_matrix.h" - -#include "source_io/module_parameter/parameter.h" -#include "source_base/libm/libm.h" -#include "source_base/memory_recorder.h" -#include "source_base/timer.h" -#include "source_base/tool_title.h" -#include "source_cell/klist.h" - -#include -#include - -namespace elecstate -{ - -// initialize density matrix DMR from UnitCell (mainly used in UnitTest) -template -void DensityMatrix::init_DMR(const Grid_Driver* GridD_in, const UnitCell* ucell) -{ - ModuleBase::TITLE("DensityMatrix", "init_DMR"); - // ensure _DMR is empty - for (auto& it: this->_DMR) - { - delete it; - } - this->_DMR.clear(); - // a newly allocated DMR is not a wavefunction-derived density matrix until cal_DMR() - this->_dmr_ready = false; - // construct a new DMR - hamilt::HContainer* tmp_DMR; - tmp_DMR = new hamilt::HContainer(this->_paraV); - // set up a HContainer - for (int iat1 = 0; iat1 < ucell->nat; iat1++) - { - auto tau1 = ucell->get_tau(iat1); - int T1, I1; - ucell->iat2iait(iat1, &I1, &T1); - AdjacentAtomInfo adjs; - GridD_in->Find_atom(*ucell, tau1, T1, I1, &adjs); - // std::cout << "adjs.adj_num: " <itia2iat(T2, I2); - if (this->_paraV->is_invalid_atom_pair(iat1, iat2)) - { - continue; - } - ModuleBase::Vector3& R_index = adjs.box[ad]; - // std::cout << "R_index: " << R_index.x << " " << R_index.y << " " << R_index.z << std::endl; - hamilt::AtomPair tmp_ap(iat1, iat2, R_index, this->_paraV); - tmp_DMR->insert_pair(tmp_ap); - } - } - // allocate the memory of BaseMatrix in SR, and set the new values to zero - if (std::is_same::value) - { - tmp_DMR->fix_gamma(); - } - tmp_DMR->allocate(nullptr, true); - this->_DMR.push_back(tmp_DMR); - // add another DMR if nspin==2 - if (this->_nspin == 2) - { - hamilt::HContainer* tmp_DMR1; - tmp_DMR1 = new hamilt::HContainer(*tmp_DMR); - this->_DMR.push_back(tmp_DMR1); - } - ModuleBase::Memory::record("DensityMatrix::DMR", this->_DMR.size() * this->_DMR[0]->get_memory_size()); -} - -/// initialize density matrix DMR from UnitCell and RA (mainly used in UnitTest) -template -void DensityMatrix::init_DMR(Record_adj& ra, const UnitCell* ucell) -{ - ModuleBase::TITLE("DensityMatrix", "init_DMR"); - // ensure _DMR is empty - for (auto& it: this->_DMR) - { - delete it; - } - this->_DMR.clear(); - // a newly allocated DMR is not a wavefunction-derived density matrix until cal_DMR() - this->_dmr_ready = false; - // construct a new DMR - hamilt::HContainer* tmp_DMR; - tmp_DMR = new hamilt::HContainer(this->_paraV); - // set up a HContainer - for (int iat1 = 0; iat1 < ucell->nat; iat1++) - { - auto tau1 = ucell->get_tau(iat1); - int T1, I1; - ucell->iat2iait(iat1, &I1, &T1); - for (int ad = 0; ad < ra.na_each[iat1]; ++ad) - { - const int T2 = ra.info[iat1][ad][3]; - const int I2 = ra.info[iat1][ad][4]; - int iat2 = ucell->itia2iat(T2, I2); - if (this->_paraV->is_invalid_atom_pair(iat1, iat2)) - { - continue; - } - hamilt::AtomPair tmp_ap(iat1, - iat2, - ra.info[iat1][ad][0], - ra.info[iat1][ad][1], - ra.info[iat1][ad][2], - this->_paraV); - tmp_DMR->insert_pair(tmp_ap); - } - } - // allocate the memory of BaseMatrix in SR, and set the new values to zero - if (std::is_same::value) - { - tmp_DMR->fix_gamma(); - } - tmp_DMR->allocate(nullptr, true); - this->_DMR.push_back(tmp_DMR); - // add another DMR if nspin==2 - if (this->_nspin == 2) - { - hamilt::HContainer* tmp_DMR1; - tmp_DMR1 = new hamilt::HContainer(*tmp_DMR); - this->_DMR.push_back(tmp_DMR1); - } - ModuleBase::Memory::record("DensityMatrix::DMR", this->_DMR.size() * this->_DMR[0]->get_memory_size()); -} - -// initialize density matrix DMR from another HContainer (mainly used) -template -void DensityMatrix::init_DMR(const hamilt::HContainer& DMR_in) -{ - ModuleBase::TITLE("DensityMatrix", "init_DMR"); - // ensure _DMR is empty - for (auto& it: this->_DMR) - { - delete it; - } - this->_DMR.clear(); - // a newly allocated DMR is not a wavefunction-derived density matrix until cal_DMR() - this->_dmr_ready = false; - // set up a HContainer using another one - for (int is = 0; is < this->_nspin; ++is) // loop over spin - { - hamilt::HContainer* tmp_DMR; - tmp_DMR = new hamilt::HContainer(DMR_in); - // zero.out - tmp_DMR->set_zero(); - this->_DMR.push_back(tmp_DMR); - } - ModuleBase::Memory::record("DensityMatrix::DMR", this->_DMR.size() * this->_DMR[0]->get_memory_size()); -} - -template -void DensityMatrix::init_DMR(const hamilt::HContainer& DMR_in) -{ - ModuleBase::TITLE("DensityMatrix", "init_DMR"); - // ensure _DMR is empty - for (auto& it: this->_DMR) - { - delete it; - } - this->_DMR.clear(); - // a newly allocated DMR is not a wavefunction-derived density matrix until cal_DMR() - this->_dmr_ready = false; - // set up a HContainer using another one - int size_ap = DMR_in.size_atom_pairs(); - if (size_ap > 0) - { - const Parallel_Orbitals* paraV_ = DMR_in.get_atom_pair(0).get_paraV(); - hamilt::HContainer* tmp_DMR = new hamilt::HContainer(paraV_); - for (int iap = 0; iap < size_ap; iap++) - { - const int iat1 = DMR_in.get_atom_pair(iap).get_atom_i(); - const int iat2 = DMR_in.get_atom_pair(iap).get_atom_j(); - for (int ir = 0; ir < DMR_in.get_atom_pair(iap).get_R_size(); ir++) - { - const ModuleBase::Vector3 R_index = DMR_in.get_atom_pair(iap).get_R_index(ir); - hamilt::AtomPair tmp_ap(iat1, iat2, R_index, paraV_); - tmp_DMR->insert_pair(tmp_ap); - } - } - tmp_DMR->allocate(nullptr, true); - this->_DMR.push_back(tmp_DMR); - if (this->_nspin == 2) - { - hamilt::HContainer* tmp_DMR1 = new hamilt::HContainer(*tmp_DMR); - this->_DMR.push_back(tmp_DMR1); - } - } - ModuleBase::Memory::record("DensityMatrix::DMR", this->_DMR.size() * this->_DMR[0]->get_memory_size()); -} - -// get _DMR pointer -template -hamilt::HContainer* DensityMatrix::get_DMR_pointer(const int ispin) const -{ - if (ispin <= 0 || ispin > this->_nspin) - { - throw std::out_of_range("DensityMatrix::get_DMR_pointer: DMR spin index is out of range"); - } - if (this->_DMR.size() != static_cast(this->_nspin)) - { - throw std::logic_error("DensityMatrix::get_DMR_pointer: DMR has not been initialized"); - } - return this->_DMR[ispin - 1]; -} - -// get _DMK[ik] pointer -template -TK* DensityMatrix::get_DMK_pointer(const int ik) const -{ -#ifdef __DEBUG - assert(ik < this->_nk * this->_nspin); -#endif - return const_cast(this->_DMK[ik].data()); -} - -// set DMK using a pointer -template -void DensityMatrix::set_DMK_pointer(const int ik, TK* DMK_in) -{ -#ifdef __DEBUG - assert(ik < this->_nk * this->_nspin); -#endif - this->_DMK[ik].assign(DMK_in, DMK_in + this->_paraV->nrow * this->_paraV->ncol); -} - -// set _DMK element -template -void DensityMatrix::set_DMK(const int ispin, const int ik, const int i, const int j, const TK value) -{ -#ifdef __DEBUG - assert(ispin > 0 && ispin <= this->_nspin); - assert(ik >= 0 && ik < this->_nk); -#endif - // consider transpose col=>row - this->_DMK[ik + this->_nk * (ispin - 1)][i * this->_paraV->nrow + j] = value; -} - -// set _DMK element -template -void DensityMatrix::set_DMK_zero() -{ - for (int ik = 0; ik < _nspin * _nk; ik++) - { - ModuleBase::GlobalFunc::ZEROS(this->_DMK[ik].data(), - this->_paraV->get_row_size() * this->_paraV->get_col_size()); - } -} - -// get a matrix element of density matrix dm(k) -template -TK DensityMatrix::get_DMK(const int ispin, const int ik, const int i, const int j) const -{ -#ifdef __DEBUG - assert(ispin > 0 && ispin <= this->_nspin); -#endif - // consider transpose col=>row - return this->_DMK[ik + this->_nk * (ispin - 1)][i * this->_paraV->nrow + j]; -} - -// get _DMK nks, nrow, ncol -template -int DensityMatrix::get_DMK_nks() const -{ -#ifdef __DEBUG - assert(this->_DMK.size() == _nk * _nspin); -#endif - return _nk * _nspin; -} - -template -int DensityMatrix::get_DMK_size() const -{ -#ifdef __DEBUG - assert(this->_DMK.size() != 0); -#endif - return this->_DMK.size(); -} - -template -int DensityMatrix::get_DMK_nrow() const -{ -#ifdef __DEBUG - assert(this->_DMK.size() != 0); -#endif - return this->_paraV->nrow; -} - -template -int DensityMatrix::get_DMK_ncol() const -{ -#ifdef __DEBUG - assert(this->_DMK.size() != 0); -#endif - return this->_paraV->ncol; -} - -template -void DensityMatrix::save_DMR() -{ - ModuleBase::TITLE("DensityMatrix", "save_DMR"); - ModuleBase::timer::start("DensityMatrix", "save_DMR"); - - const int nnr = this->_DMR[0]->get_nnr(); - // allocate if _DMR_save is empty - if (_DMR_save.size() == 0) - { - _DMR_save.resize(this->_DMR.size()); - } - // resize if _DMR_save[is].size is not equal to _DMR.size - for (int is = 0; is < _DMR_save.size(); is++) - { - if (_DMR_save[is].size() != nnr) - { - _DMR_save[is].resize(nnr); - } - } - // save _DMR to _DMR_save - for (int is = 0; is < this->_DMR.size(); is++) - { - TR* DMR_pointer = this->_DMR[is]->get_wrapper(); - TR* DMR_save_pointer = _DMR_save[is].data(); - // set to zero - ModuleBase::GlobalFunc::ZEROS(DMR_save_pointer, nnr); - for (int i = 0; i < nnr; i++) - { - DMR_save_pointer[i] = DMR_pointer[i]; - } - } - - ModuleBase::timer::end("DensityMatrix", "save_DMR"); -} - -// read *.dmk into density matrix dm(k) -template -void DensityMatrix::read_DMK(const std::string directory, const int ispin, const int ik) -{ - ModuleBase::TITLE("DensityMatrix", "read_DMK"); -#ifdef __DEBUG - assert(ispin > 0 && ispin <= this->_nspin); -#endif - // read - std::string fn; - fn = directory + "SPIN" + std::to_string(ispin) + "_" + std::to_string(ik) + ".dmk"; - // - bool quit_abacus = false; - - std::ifstream ifs; - - ifs.open(fn.c_str()); - if (!ifs) - { - quit_abacus = true; - } - else - { - // if the number is not match, - // quit the program or not. - bool quit = false; - - ModuleBase::CHECK_DOUBLE(ifs, this->_kvec_d[ik].x, quit); - ModuleBase::CHECK_DOUBLE(ifs, this->_kvec_d[ik].y, quit); - ModuleBase::CHECK_DOUBLE(ifs, this->_kvec_d[ik].z, quit); - ModuleBase::CHECK_INT(ifs, this->_paraV->nrow); - ModuleBase::CHECK_INT(ifs, this->_paraV->ncol); - } // If file exist, read in data. - // Finish reading the first part of density matrix. - - for (int i = 0; i < this->_paraV->nrow; ++i) - { - for (int j = 0; j < this->_paraV->ncol; ++j) - { - ifs >> this->_DMK[ik + this->_nk * (ispin - 1)][i * this->_paraV->ncol + j]; - } - } - ifs.close(); -} - -// output density matrix dm(k) into *.dmk -template <> -void DensityMatrix::write_DMK(const std::string directory, const int ispin, const int ik) -{ - ModuleBase::TITLE("DensityMatrix", "write_DMK"); -#ifdef __DEBUG - assert(ispin > 0 && ispin <= this->_nspin); -#endif - // write - std::string fn; - fn = directory + "SPIN" + std::to_string(ispin) + "_" + std::to_string(ik) + ".dmk"; - std::ofstream ofs; - ofs.open(fn.c_str()); - if (!ofs) - { - ModuleBase::WARNING("elecstate::write_dmk", "Can't create DENSITY MATRIX File!"); - } - ofs << this->_kvec_d[ik].x << " " << this->_kvec_d[ik].y << " " << this->_kvec_d[ik].z << std::endl; - ofs << "\n " << this->_paraV->nrow << " " << this->_paraV->ncol << std::endl; - - ofs << std::setprecision(3); - ofs << std::scientific; - - for (int i = 0; i < this->_paraV->nrow; ++i) - { - for (int j = 0; j < this->_paraV->ncol; ++j) - { - if (j % 8 == 0) - { - ofs << "\n"; - } - ofs << " " << this->_DMK[ik + this->_nk * (ispin - 1)][i * this->_paraV->ncol + j]; - } - } - - ofs.close(); -} - -template <> -void DensityMatrix, double>::write_DMK(const std::string directory, const int ispin, const int ik) -{ - ModuleBase::TITLE("DensityMatrix", "write_DMK"); -#ifdef __DEBUG - assert(ispin > 0 && ispin <= this->_nspin); -#endif - // write - std::string fn; - fn = directory + "SPIN" + std::to_string(ispin) + "_" + std::to_string(ik) + ".dmk"; - std::ofstream ofs; - ofs.open(fn.c_str()); - if (!ofs) - { - ModuleBase::WARNING("elecstate::write_dmk", "Can't create DENSITY MATRIX File!"); - } - ofs << this->_kvec_d[ik].x << " " << this->_kvec_d[ik].y << " " << this->_kvec_d[ik].z << std::endl; - ofs << "\n " << this->_paraV->nrow << " " << this->_paraV->ncol << std::endl; - - ofs << std::setprecision(3); - ofs << std::scientific; - - for (int i = 0; i < this->_paraV->nrow; ++i) - { - for (int j = 0; j < this->_paraV->ncol; ++j) - { - if (j % 8 == 0) - { - ofs << "\n"; - } - ofs << " " << this->_DMK[ik + this->_nk * (ispin - 1)][i * this->_paraV->ncol + j].real(); - } - } - - ofs.close(); -} - -// T of HContainer can be double or std::complex -template class DensityMatrix; // Gamma-Only case -template class DensityMatrix, double>; // Multi-k case -template class DensityMatrix, std::complex>; // For EXX in future - -} // namespace elecstate diff --git a/source/source_estate/module_dm/dm_from_psi.cpp b/source/source_estate/module_dm/dm_from_psi.cpp new file mode 100644 index 00000000000..f70201d544e --- /dev/null +++ b/source/source_estate/module_dm/dm_from_psi.cpp @@ -0,0 +1,348 @@ +#include "dm_from_psi.h" + +#include +#include +#include + +#include "source_base/module_external/blas_connector.h" +#include "source_base/module_external/scalapack_connector.h" +#include "source_base/timer.h" +#include "source_psi/psi.h" + +namespace module_dm +{ +namespace +{ +/** + * @brief Conjugation wrapper with an exact TK return type + * + * std::conj(double) is an additional overload whose availability/return type + * differs across standard libraries; these helpers guarantee that the Gamma-only + * (TK = double) instantiation stays real instead of promoting to complex. + */ +inline double conj_value(const double x) +{ + return x; +} + +inline std::complex conj_value(const std::complex x) +{ + return std::conj(x); +} + +/** + * @brief Build the weighted (and conjugated) left factor of the DM GEMM for one k-point + * + * wg_wfc(ib, iw) = factor_ib * conj(wfc(ib, iw)); for TK = double std::conj is + * the identity, so the same template covers the Gamma-only case. + * + * The local-to-global band mapping is taken verbatim from the historical + * cal_dm implementation: ib_global advances monotonically while scanning + * ParaV->global2local_col(). A local band whose global index does not fall + * into the columns of wg keeps the legacy factor 1.0 instead of 0.0. + * + * @param ParaV orbital distribution, provides global2local_col() + * @param wg band weights for the current k-point + * @param ik k-point index; wfc must already be fixed to it + * @param wfc wavefunction block of the current k-point + * @param wg_wfc preallocated buffer of the same (nbands_local, nbasis_local) shape + */ +template +void fill_weighted_wfc(const Parallel_Orbitals* ParaV, + const ModuleBase::matrix& wg, + const int ik, + const psi::Psi& wfc, + psi::Psi& wg_wfc) +{ + const int nbands_local = wfc.get_nbands(); + const int nbasis_local = wfc.get_nbasis(); + + // Resolve every per-band factor first: the global-band scan is serial. + std::vector factor(nbands_local, 1.0); + int ib_global = 0; + for (int ib_local = 0; ib_local < nbands_local; ++ib_local) + { + while (ib_local != ParaV->global2local_col(ib_global)) + { + ++ib_global; + if (ib_global >= wg.nc) + { + break; + } + } + if (ib_global < wg.nc) + { + factor[ib_local] = wg(ik, ib_global); + } + } + + // Fuse the copy, the conjugation (complex case) and the weighting into one pass. + // get_pointer() addresses the block selected by the caller's fix_k(); using the + // three-argument operator() here would always address k-point block 0. + const TK* wfc_block = wfc.get_pointer(); + TK* wg_wfc_block = wg_wfc.get_pointer(); +#ifdef _OPENMP +#pragma omp parallel for schedule(static) +#endif + for (int ib_local = 0; ib_local < nbands_local; ++ib_local) + { + const int offset = ib_local * nbasis_local; + const double factor_ib = factor[ib_local]; + for (int iw = 0; iw < nbasis_local; ++iw) + { + wg_wfc_block[offset + iw] = factor_ib * conj_value(wfc_block[offset + iw]); + } + } +} + +#ifdef __MPI +/** + * @brief Distributed GEMM: dmk = wg_wfc * wfc^T (column-major perspective) + * + * The row-major wavefunction wfc(ib, iw) is seen by the column-major BLAS as + * its transpose. Using 'N' on the pre-conjugated wg_wfc and 'T' on wfc yields + * dmk(iw1, iw2) = sum_ib wg_wfc(ib, iw1) * wfc(ib, iw2), + * i.e. the conjugation lives on the first index. 'C' must not be substituted + * for 'T': 'T' and 'C' give identical operand dimensions, but 'C' additionally + * conjugates the transposed operand, moving the conjugation to the second + * index and producing the transpose of the stored conj-first DM block. + */ +void gemm_dm(const psi::Psi& psi1, + const psi::Psi& psi2, + double* dm_out, + const int* desc_psi, + const int* desc_dm) +{ + ModuleBase::timer::start("dmk_from_psi", "pdgemm"); + const double one_float = 1.0; + const double zero_float = 0.0; + const int one_int = 1; + const char n_char = 'N'; + const char t_char = 'T'; + const int nlocal = desc_dm[2]; + const int nbands = desc_psi[3]; + ScalapackConnector::gemm(n_char, + t_char, + nlocal, + nlocal, + nbands, + one_float, + psi1.get_pointer(), + one_int, + one_int, + desc_psi, + psi2.get_pointer(), + one_int, + one_int, + desc_psi, + zero_float, + dm_out, + one_int, + one_int, + desc_dm); + ModuleBase::timer::end("dmk_from_psi", "pdgemm"); +} + +void gemm_dm(const psi::Psi>& psi1, + const psi::Psi>& psi2, + std::complex* dm_out, + const int* desc_psi, + const int* desc_dm) +{ + ModuleBase::timer::start("dmk_from_psi", "pzgemm"); + const std::complex one_complex = {1.0, 0.0}; + const std::complex zero_complex = {0.0, 0.0}; + const int one_int = 1; + const char n_char = 'N'; + const char t_char = 'T'; + const int nlocal = desc_dm[2]; + const int nbands = desc_psi[3]; + ScalapackConnector::gemm(n_char, + t_char, + nlocal, + nlocal, + nbands, + one_complex, + psi1.get_pointer(), + one_int, + one_int, + desc_psi, + psi2.get_pointer(), + one_int, + one_int, + desc_psi, + zero_complex, + dm_out, + one_int, + one_int, + desc_dm); + ModuleBase::timer::end("dmk_from_psi", "pzgemm"); +} +#else +/** + * @brief Serial GEMM: dmk = wg_wfc * wfc^T, see the MPI overload for the 'T' rationale + */ +void gemm_dm(const psi::Psi& psi1, const psi::Psi& psi2, double* dm_out) +{ + const double one_float = 1.0; + const double zero_float = 0.0; + const int one_int = 1; + const char n_char = 'N'; + const char t_char = 'T'; + const int nlocal = psi1.get_nbasis(); + const int nbands = psi1.get_nbands(); + BlasConnector::gemm_cm(n_char, + t_char, + nlocal, + nlocal, + nbands, + one_float, + psi1.get_pointer(), + nlocal, + psi2.get_pointer(), + nlocal, + zero_float, + dm_out, + nlocal); +} + +void gemm_dm(const psi::Psi>& psi1, + const psi::Psi>& psi2, + std::complex* dm_out) +{ + const std::complex one_complex = {1.0, 0.0}; + const std::complex zero_complex = {0.0, 0.0}; + const int one_int = 1; + const char n_char = 'N'; + const char t_char = 'T'; + const int nlocal = psi1.get_nbasis(); + const int nbands = psi1.get_nbands(); + BlasConnector::gemm_cm(n_char, + t_char, + nlocal, + nlocal, + nbands, + one_complex, + psi1.get_pointer(), + nlocal, + psi2.get_pointer(), + nlocal, + zero_complex, + dm_out, + nlocal); +} +#endif + +/** + * @brief Fill the weighted wavefunction buffer and run the DM GEMM for one k-point + * + * @param wg_wfc reusable scratch buffer, shape (1, nbands_local, nbasis_local) + */ +template +void dmk_from_psi_impl(const Parallel_Orbitals* ParaV, + const ModuleBase::matrix& wg, + const int ik, + const psi::Psi& wfc, + TK* dmk_out, + psi::Psi& wg_wfc) +{ + wfc.fix_k(ik); + fill_weighted_wfc(ParaV, wg, ik, wfc, wg_wfc); +#ifdef __MPI + gemm_dm(wg_wfc, wfc, dmk_out, ParaV->desc_wfc, ParaV->desc); +#else + gemm_dm(wg_wfc, wfc, dmk_out); +#endif +} +} // namespace + +// for Gamma-Only case where DMK is double +void dm_from_psi(const Parallel_Orbitals* ParaV, + const ModuleBase::matrix& wg, + const psi::Psi& wfc, + module_dm::DensityMatrix& DM) +{ + assert(ParaV != nullptr); + ModuleBase::TITLE("elecstate", "dm_from_psi"); + ModuleBase::timer::start("elecstate", "dm_from_psi"); + + const int nbands_local = wfc.get_nbands(); + const int nbasis_local = wfc.get_nbasis(); + + // Allocate the weighted-wavefunction scratch once and reuse it for every k-point. + psi::Psi wg_wfc(1, nbands_local, nbasis_local, nbasis_local, true); + + for (int ik = 0; ik < wfc.get_nk(); ++ik) + { + double* dmk_pointer = DM.get_dmk_ptr(ik); + dmk_from_psi_impl(ParaV, wg, ik, wfc, dmk_pointer, wg_wfc); + } + ModuleBase::timer::end("elecstate", "dm_from_psi"); +} + +template +void dm_from_psi(const Parallel_Orbitals* ParaV, + const ModuleBase::matrix& wg, + const psi::Psi>& wfc, + module_dm::DensityMatrix, TR>& DM) +{ + assert(ParaV != nullptr); + ModuleBase::TITLE("elecstate", "dm_from_psi"); + ModuleBase::timer::start("elecstate", "dm_from_psi"); + + const int nbands_local = wfc.get_nbands(); + const int nbasis_local = wfc.get_nbasis(); + + // Allocate the weighted-wavefunction scratch once and reuse it for every k-point. + psi::Psi> wg_wfc(1, nbands_local, nbasis_local, nbasis_local, true); + + for (int ik = 0; ik < wfc.get_nk(); ++ik) + { + std::complex* dmk_pointer = DM.get_dmk_ptr(ik); + dmk_from_psi_impl(ParaV, wg, ik, wfc, dmk_pointer, wg_wfc); + } + + ModuleBase::timer::end("elecstate", "dm_from_psi"); +} + +void dmk_from_psi(const Parallel_Orbitals* ParaV, + const ModuleBase::matrix& wg, + const int ik, + const psi::Psi& wfc, + double* dmk_out) +{ + assert(ParaV != nullptr); + assert(dmk_out != nullptr); + assert(ik >= 0 && ik < wfc.get_nk()); + + psi::Psi wg_wfc(1, wfc.get_nbands(), wfc.get_nbasis(), wfc.get_nbasis(), true); + dmk_from_psi_impl(ParaV, wg, ik, wfc, dmk_out, wg_wfc); +} + +void dmk_from_psi(const Parallel_Orbitals* ParaV, + const ModuleBase::matrix& wg, + const int ik, + const psi::Psi>& wfc, + std::complex* dmk_out) +{ + assert(ParaV != nullptr); + assert(dmk_out != nullptr); + assert(ik >= 0 && ik < wfc.get_nk()); + + psi::Psi> wg_wfc(1, + wfc.get_nbands(), + wfc.get_nbasis(), + wfc.get_nbasis(), + true); + dmk_from_psi_impl(ParaV, wg, ik, wfc, dmk_out, wg_wfc); +} + +template void dm_from_psi(const Parallel_Orbitals* ParaV, + const ModuleBase::matrix& wg, + const psi::Psi>& wfc, + module_dm::DensityMatrix, std::complex>& DM); +template void dm_from_psi(const Parallel_Orbitals* ParaV, + const ModuleBase::matrix& wg, + const psi::Psi>& wfc, + module_dm::DensityMatrix, double>& DM); +} // namespace module_dm diff --git a/source/source_estate/module_dm/dm_from_psi.h b/source/source_estate/module_dm/dm_from_psi.h new file mode 100644 index 00000000000..848821de2f0 --- /dev/null +++ b/source/source_estate/module_dm/dm_from_psi.h @@ -0,0 +1,89 @@ +#ifndef DM_FROM_PSI_H +#define DM_FROM_PSI_H + +#include "density_matrix.h" +#include "source_base/matrix.h" +#include "source_psi/psi.h" + +namespace module_dm +{ +/** + * @brief Calculate the k-point density matrix from wavefunctions for the Gamma-only case + * + * For every k-point block of the wavefunction this computes + * dmk(iw1, iw2) = sum_ib wg(ik, ib) * wfc(ib, iw1) * wfc(ib, iw2), + * i.e. the wavefunction weights are real so no conjugation is needed. + * + * @param ParaV 2D block-cyclic distribution descriptor of the orbitals + * @param wg band weights, wg(ik, ib_global) for the global band index + * @param wfc wavefunction coefficients, shape (nk, nbands_local, nbasis_local) + * @param DM output density matrix, one block per k-point + */ +void dm_from_psi(const Parallel_Orbitals* ParaV, + const ModuleBase::matrix& wg, + const psi::Psi& wfc, + DensityMatrix& DM); + +/** + * @brief Calculate the k-point density matrix from wavefunctions for the multi-k case + * + * For every k-point block of the wavefunction this computes + * dmk(iw1, iw2) = sum_ib wg(ik, ib) * conj(wfc(ib, iw1)) * wfc(ib, iw2). + * The conjugation is applied to the FIRST basis index ("conj-first" storage): + * the stored block is the transpose of the physical one-body density matrix + * P = C diag(wg) C^H. cal_dmr(), SOC magnetization and Mulliken analysis all + * rely on this convention, see dm_tools.cpp and + * unittests/test_soc_magnetization_roundtrip.cpp. + * + * @tparam TR real type of the real-space density matrix of DM + * @param ParaV 2D block-cyclic distribution descriptor of the orbitals + * @param wg band weights, wg(ik, ib_global) for the global band index + * @param wfc wavefunction coefficients, shape (nk, nbands_local, nbasis_local) + * @param DM output density matrix, one block per k-point + */ +template +void dm_from_psi(const Parallel_Orbitals* ParaV, + const ModuleBase::matrix& wg, + const psi::Psi>& wfc, + DensityMatrix, TR>& DM); + +/** + * @brief Calculate one k-point block of the density matrix (Gamma-only case) + * + * Single-k-point worker behind dm_from_psi(); exposed for callers that own the + * output storage themselves. The conjugation/weighting contract is the same as + * dm_from_psi(), the caller only provides the output slot. + * + * @param ParaV 2D block-cyclic distribution descriptor of the orbitals + * @param wg band weights, wg(ik, ib_global) for the global band index + * @param ik k-point index of the block to calculate + * @param wfc wavefunction coefficients + * @param dmk_out output block, distributed according to ParaV->desc (serial + * builds: a dense nbasis_local x nbasis_local column-major matrix) + */ +void dmk_from_psi(const Parallel_Orbitals* ParaV, + const ModuleBase::matrix& wg, + const int ik, + const psi::Psi& wfc, + double* dmk_out); + +/** + * @brief Calculate one k-point block of the density matrix (multi-k case) + * + * Single-k-point worker behind dm_from_psi(), using the conj-first convention + * dmk(iw1, iw2) = sum_ib wg(ik, ib) * conj(wfc(ib, iw1)) * wfc(ib, iw2). + * + * @param ParaV 2D block-cyclic distribution descriptor of the orbitals + * @param wg band weights, wg(ik, ib_global) for the global band index + * @param ik k-point index of the block to calculate + * @param wfc wavefunction coefficients + * @param dmk_out output block, distributed according to ParaV->desc (serial + * builds: a dense nbasis_local x nbasis_local column-major matrix) + */ +void dmk_from_psi(const Parallel_Orbitals* ParaV, + const ModuleBase::matrix& wg, + const int ik, + const psi::Psi>& wfc, + std::complex* dmk_out); +} // namespace module_dm +#endif diff --git a/source/source_estate/module_dm/dm_getter.cpp b/source/source_estate/module_dm/dm_getter.cpp new file mode 100644 index 00000000000..04c99f48aa4 --- /dev/null +++ b/source/source_estate/module_dm/dm_getter.cpp @@ -0,0 +1,95 @@ +#include "density_matrix.h" + +#include "source_io/module_parameter/parameter.h" +#include "source_base/libm/libm.h" +#include "source_base/memory_recorder.h" +#include "source_base/timer.h" +#include "source_base/tool_title.h" +#include "source_cell/klist.h" + +#include +#include +#include + +namespace module_dm +{ + +// get dmr pointer +template +hamilt::HContainer* DensityMatrix::get_dmr_ptr(const int ispin) const +{ + if (ispin <= 0 || ispin > this->spin_mult) + { + throw std::out_of_range("DensityMatrix::get_dmr_ptr: DMR spin index is out of range"); + } + if (this->dmr.size() != static_cast(this->spin_mult)) + { + throw std::logic_error("DensityMatrix::get_dmr_ptr: DMR has not been initialized"); + } + return this->dmr[ispin - 1]; +} + +// get dmk[ik] pointer +template +TK* DensityMatrix::get_dmk_ptr(const int ik) const +{ +#ifdef __DEBUG + assert(ik < this->_nk * this->spin_mult); +#endif + return const_cast(this->dmk[ik].data()); +} + +// get a matrix element of density matrix dm(k) +template +TK DensityMatrix::get_dmk(const int ispin, const int ik, const int i, const int j) const +{ +#ifdef __DEBUG + assert(ispin > 0 && ispin <= this->spin_mult); +#endif + // consider transpose col=>row + return this->dmk[ik + this->_nk * (ispin - 1)][i * this->pv->nrow + j]; +} + +// get dmk nks, nrow, ncol +template +int DensityMatrix::get_dmk_nks() const +{ +#ifdef __DEBUG + assert(this->dmk.size() == _nk * spin_mult); +#endif + return _nk * spin_mult; +} + +template +int DensityMatrix::get_dmk_size() const +{ +#ifdef __DEBUG + assert(this->dmk.size() != 0); +#endif + return this->dmk.size(); +} + +template +int DensityMatrix::get_dmk_nrow() const +{ +#ifdef __DEBUG + assert(this->dmk.size() != 0); +#endif + return this->pv->nrow; +} + +template +int DensityMatrix::get_dmk_ncol() const +{ +#ifdef __DEBUG + assert(this->dmk.size() != 0); +#endif + return this->pv->ncol; +} + +// T of HContainer can be double or std::complex +template class DensityMatrix; // Gamma-Only case +template class DensityMatrix, double>; // Multi-k case +template class DensityMatrix, std::complex>; // For EXX in future + +} // namespace module_dm diff --git a/source/source_estate/module_dm/dm_holder.h b/source/source_estate/module_dm/dm_holder.h new file mode 100644 index 00000000000..93b1793e471 --- /dev/null +++ b/source/source_estate/module_dm/dm_holder.h @@ -0,0 +1,29 @@ +#ifndef MODULE_DM_DM_HOLDER_H +#define MODULE_DM_DM_HOLDER_H + +#include "source_estate/module_dm/density_matrix.h" + +namespace module_dm +{ +template +class Setup_DM +{ + public: + Setup_DM() + { + } // will be called by ElecStateLCAO_TDDFT + + ~Setup_DM() + { + if (this->dm != nullptr) + { + delete this->dm; + } + } + + DensityMatrix* dm = nullptr; +}; + +} // namespace module_dm + +#endif diff --git a/source/source_estate/module_dm/dm_routine.cpp b/source/source_estate/module_dm/dm_routine.cpp new file mode 100644 index 00000000000..504aecd6a0b --- /dev/null +++ b/source/source_estate/module_dm/dm_routine.cpp @@ -0,0 +1,70 @@ +#include "source_estate/module_dm/dm_routine.h" +#include "source_estate/module_dm/dm_from_psi.h" +#include "source_estate/elecstate_tools.h" +#include "source_cell/cal_ux.h" + +template +void module_dm::init_dm(UnitCell& ucell, + elecstate::ElecState* pelec, + module_dm::Setup_DM &dmat, + psi::Psi* psi, + Charge &chr, + const Parallel_Orbitals& pv, + const int iter, + const int exx_two_level_step, + const Init_DM_Config& cfg) +{ + ModuleBase::TITLE("elecstate", "init_dm"); + + if (iter == 1 && exx_two_level_step == 0) + { + std::cout << " LCAO WAVEFUN -> CHARGE " << std::endl; + + elecstate::calEBand(pelec->ekb, pelec->wg, pelec->f_en); + + module_dm::dm_from_psi(&pv, pelec->wg, *psi, *dmat.dm); + if (cfg.esolver_type != "tddft" && cfg.td_stype == 2) + { + dmat.dm->cal_dmr_td(*cfg.td_phase_hybrid, cfg.td_cart_At, -1); + } + else + { + dmat.dm->cal_dmr(-1); + } + + // use density matrix to calculate the charge density + cfg.dm2rho_func(dmat.dm->get_dmr_vec(), cfg.nspin, &chr, cfg.nelec, ucell.omega, false); + + unitcell::cal_ux(ucell, cfg.nspin); + + //! update the potentials by using new electron charge density + pelec->pot->update_from_charge(&chr, &ucell); + + //! compute the correction energy for metals + pelec->f_en.descf = pelec->cal_delta_escf(); + } + + return; +} + + +template void module_dm::init_dm(UnitCell& ucell, + elecstate::ElecState* pelec, + module_dm::Setup_DM &dmat, + psi::Psi* psi, + Charge &chr, + const Parallel_Orbitals& pv, + const int iter, + const int exx_two_level_step, + const Init_DM_Config& cfg); + +template void module_dm::init_dm>(UnitCell& ucell, + elecstate::ElecState* pelec, + module_dm::Setup_DM> &dmat, + psi::Psi>* psi, + Charge &chr, + const Parallel_Orbitals& pv, + const int iter, + const int exx_two_level_step, + const Init_DM_Config& cfg); + diff --git a/source/source_estate/module_dm/dm_routine.h b/source/source_estate/module_dm/dm_routine.h new file mode 100644 index 00000000000..9e2ef957d3f --- /dev/null +++ b/source/source_estate/module_dm/dm_routine.h @@ -0,0 +1,49 @@ +#ifndef DM_ROUTINE_H +#define DM_ROUTINE_H + +#include +#include +#include +#include + +#include "source_base/vector3.h" +#include "source_cell/unitcell.h" // use unitcell +#include "source_estate/elecstate.h"// use ElecState +#include "source_psi/psi.h" // use electronic wave functions +#include "source_estate/module_charge/charge.h" // use charge +#include "source_estate/module_dm/dm_holder.h" // define Setup_DM +#include "source_hamilt/module_hcontainer/hcontainer.h" +#include "source_basis/module_ao/parallel_orbitals.h" + +namespace module_dm +{ + +struct Init_DM_Config +{ + std::string esolver_type; + int td_stype; + int nspin; + double nelec; + // RT-TDDFT (td_stype==2, esolver_type!="tddft") parameters, passed explicitly + // to avoid a reverse dependency of source_estate on source_lcao/module_rt. + const std::map, std::complex>* td_phase_hybrid = nullptr; + ModuleBase::Vector3 td_cart_At; + // dm2rho lives in source_lcao; pass it as a callback so source_estate does + // not depend on source_lcao. + std::function*>&, int, Charge*, double, double, bool)> dm2rho_func; +}; + +template +void init_dm(UnitCell& ucell, + elecstate::ElecState* pelec, + Setup_DM &dmat, + psi::Psi* psi, + Charge &chr, + const Parallel_Orbitals& pv, + const int iter, + const int exx_two_level_step, + const Init_DM_Config& cfg); + +} + +#endif diff --git a/source/source_estate/module_dm/dm_setter.cpp b/source/source_estate/module_dm/dm_setter.cpp new file mode 100644 index 00000000000..2ce4aa10419 --- /dev/null +++ b/source/source_estate/module_dm/dm_setter.cpp @@ -0,0 +1,89 @@ +#include "density_matrix.h" + +#include "source_io/module_parameter/parameter.h" +#include "source_base/libm/libm.h" +#include "source_base/memory_recorder.h" +#include "source_base/timer.h" +#include "source_base/tool_title.h" +#include "source_cell/klist.h" + +#include +#include +#include +#include + +namespace module_dm +{ + +// set DMK using a pointer +template +void DensityMatrix::set_dmk_ptr(const int ik, TK* DMK_in) +{ +#ifdef __DEBUG + assert(ik < this->_nk * this->spin_mult); +#endif + this->dmk[ik].assign(DMK_in, DMK_in + this->pv->nrow * this->pv->ncol); +} + +// set dmk element +template +void DensityMatrix::set_dmk(const int ispin, const int ik, const int i, const int j, const TK value) +{ +#ifdef __DEBUG + assert(ispin > 0 && ispin <= this->spin_mult); + assert(ik >= 0 && ik < this->_nk); +#endif + // consider transpose col=>row + this->dmk[ik + this->_nk * (ispin - 1)][i * this->pv->nrow + j] = value; +} + +// set dmk element +template +void DensityMatrix::set_dmk_zero() +{ + for (int ik = 0; ik < spin_mult * _nk; ik++) + { + std::fill(this->dmk[ik].begin(), this->dmk[ik].end(), TK{}); + } +} + +template +void DensityMatrix::save_dmr() +{ + ModuleBase::TITLE("DensityMatrix", "save_dmr"); + ModuleBase::timer::start("DensityMatrix", "save_dmr"); + + const int nnr = this->dmr[0]->get_nnr(); + // allocate if dmr_save is empty + if (dmr_save.size() == 0) + { + dmr_save.resize(this->dmr.size()); + } + // resize if dmr_save[is].size is not equal to dmr.size + for (int is = 0; is < dmr_save.size(); is++) + { + if (dmr_save[is].size() != nnr) + { + dmr_save[is].resize(nnr); + } + } + // save dmr to dmr_save + for (int is = 0; is < this->dmr.size(); is++) + { + TR* DMR_pointer = this->dmr[is]->get_wrapper(); + TR* DMR_save_pointer = dmr_save[is].data(); + // copy DMR_pointer into DMR_save_pointer over [0, nnr); the + // destination is fully overwritten, so prior zeroing is a dead + // store, and resize above value-initializes newly added elements. + std::copy(DMR_pointer, DMR_pointer + nnr, DMR_save_pointer); + } + + ModuleBase::timer::end("DensityMatrix", "save_dmr"); +} + +// T of HContainer can be double or std::complex +template class DensityMatrix; // Gamma-Only case +template class DensityMatrix, double>; // Multi-k case +template class DensityMatrix, std::complex>; // For EXX in future + +} // namespace module_dm diff --git a/source/source_estate/module_dm/dm_tools.cpp b/source/source_estate/module_dm/dm_tools.cpp new file mode 100644 index 00000000000..df7c260f35b --- /dev/null +++ b/source/source_estate/module_dm/dm_tools.cpp @@ -0,0 +1,322 @@ +#include "density_matrix.h" + +#include "source_base/libm/libm.h" +#include "source_base/tool_title.h" +#include "source_base/tool_quit.h" +#include "source_base/constants.h" +#include "source_base/timer.h" +#include "source_cell/klist.h" + +namespace module_dm +{ + +template <> +void exp_mul_dmk( + const std::complex kphase, + const std::vector>& dmk_row, + double* dmr_mat) +{ + const std::size_t mat_size = dmk_row.size(); + for (std::size_t i = 0; i < mat_size; i++) + { + dmr_mat[i] += kphase.real() * dmk_row[i].real() - kphase.imag() * dmk_row[i].imag(); + } +} + +template <> +void exp_mul_dmk>( + const std::complex kphase, + const std::vector>& dmk_row, + std::complex* dmr_mat) +{ + BlasConnector::axpy(dmk_row.size(), kphase, dmk_row.data(), 1, dmr_mat, 1); +} + +template <> +void xyz_to_updown( + const std::complex spin_block[4], + const int icol, + const int spin_stride[4], + double* dmr_mat) +{ + dmr_mat[icol + spin_stride[0]] = spin_block[0].real() + spin_block[3].real(); // rho_0 = (rho_upup + rho_downdown).real() + dmr_mat[icol + spin_stride[1]] = spin_block[1].real() + spin_block[2].real(); // rho_x = (rho_updown + rho_downup).real() + // rho_y: the stored DM block is the complex conjugate of the physical 1-RDM P (dm_from_psi builds + // DM_{ab}=sum conj(c_a) c_b = conj(P), so spin_block[1]=DM_{ud}=conj(P_{ud})). Extracting m_y from the + // CONJUGATED block therefore carries the opposite sign of the bare-textbook formula; m_x/m_z read + // Re() and are conjugation-invariant. Using the bare formula (PR #7664) sign-flips m_y and quenches + // in-plane non-collinear moments (e.g. Mn3Sn 120-deg AFM); see issue #7831. + dmr_mat[icol + spin_stride[2]] = spin_block[1].imag() - spin_block[2].imag(); // rho_y = Im(P_updown) - Im(P_downup) + dmr_mat[icol + spin_stride[3]] = spin_block[0].real() - spin_block[3].real(); // rho_z = (rho_upup - rho_downdown).real() +} + +template <> +void xyz_to_updown>( + const std::complex spin_block[4], + const int icol, + const int spin_stride[4], + std::complex* dmr_mat) +{ + dmr_mat[icol + spin_stride[0]] = spin_block[0] + spin_block[3]; // rho_0 = (rho_upup + rho_downdown) + dmr_mat[icol + spin_stride[1]] = spin_block[1] + spin_block[2]; // rho_x = (rho_updown + rho_downup) + // rho_y sign accounts for the conjugated stored DM block (conj(P)); see the specialization above. + dmr_mat[icol + spin_stride[2]] = -ModuleBase::IMAG_UNIT * (spin_block[1] - spin_block[2]); // rho_y = -i*(rho_updown - rho_downup) + dmr_mat[icol + spin_stride[3]] = spin_block[0] - spin_block[3]; // rho_z = (rho_upup - rho_downdown) +} + +DmrBlock get_dmr_block( + const Parallel_Orbitals* pv, + const int iat1, + const int iat2) +{ + DmrBlock block; + block.row0 = pv->atom_begin_row[iat1]; + block.col0 = pv->atom_begin_col[iat2]; + block.nrows = pv->get_nrow_atom(iat1); + block.ncols = pv->get_ncol_atom(iat2); + assert(block.row0 != -1 && block.col0 != -1 && "Atom-pair not belong this process"); + return block; +} + +template +void build_kphase( + hamilt::AtomPair& atom_pair, + const std::vector>& kvec_d, + const int nk, + const std::map, std::complex>& phase_hybrid, + std::vector>& kphase_vec, + std::vector& dmr_mats) +{ + const int R_size = atom_pair.get_R_size(); + kphase_vec.assign(nk, std::vector(R_size)); + dmr_mats.assign(R_size, nullptr); + for (int iR = 0; iR < R_size; ++iR) + { + const ModuleBase::Vector3 R_index = atom_pair.get_R_index(iR); + hamilt::BaseMatrix* const dmr_R = atom_pair.find_matrix(R_index); +#ifdef __DEBUG + if (dmr_R == nullptr) + { + std::cout << "dmr_R is nullptr" << std::endl; + continue; + } +#endif + dmr_mats[iR] = dmr_R->get_pointer(); + for (int ik = 0; ik < nk; ++ik) + { + const ModuleBase::Vector3 dR(R_index[0], R_index[1], R_index[2]); + const double arg = (kvec_d[ik] * dR) * ModuleBase::TWO_PI; + double sinp; + double cosp; + ModuleBase::libm::sincos(arg, &sinp, &cosp); + kphase_vec[ik][iR] = TK(cosp, sinp); + if (!phase_hybrid.empty()) + { + kphase_vec[ik][iR] *= phase_hybrid.at(R_index); + } + } + } +} + +template +void transpose_dmk_block( + const TK* dmk_col_major, + const int ld_hk, + const DmrBlock& block, + TK* dmk_row) +{ + for (int icol = 0; icol < block.ncols; ++icol) + { + for (int irow = 0; irow < block.nrows; ++irow) + { + dmk_row[irow * block.ncols + icol] = dmk_col_major[icol * ld_hk + irow]; + } + } +} + +template +void add_dmr_real( + const DensityMatrix& dm, + const DmrBlock& block, + const int ik_begin, + const std::vector>& kphase_vec, + const int ld_hk, + const int ik_in, + std::vector& dmr_mats) +{ + const int R_size = dmr_mats.size(); + std::vector dmk_row(block.size()); + if (ik_in >= 0) + { + // single k-point + const TK* const dmk_col = dm.dmk[ik_in + ik_begin].data() + block.col0 * ld_hk + block.row0; + transpose_dmk_block(dmk_col, ld_hk, block, dmk_row.data()); + for (int iR = 0; iR < R_size; ++iR) + { + exp_mul_dmk(kphase_vec[ik_in][iR], dmk_row, dmr_mats[iR]); + } + } + else + { + // all k-points + for (int ik = 0; ik < dm._nk; ++ik) + { + const TK* const dmk_col = dm.dmk[ik + ik_begin].data() + block.col0 * ld_hk + block.row0; + transpose_dmk_block(dmk_col, ld_hk, block, dmk_row.data()); + for (int iR = 0; iR < R_size; ++iR) + { + exp_mul_dmk(kphase_vec[ik][iR], dmk_row, dmr_mats[iR]); + } + } + } +} + +template +void add_dmr_soc( + const DensityMatrix& dm, + const DmrBlock& block, + const int ik_begin, + const std::vector>& kphase_vec, + const int ld_hk, + const int ik_in, + const int col_stride, + std::vector& dmr_mats) +{ + const int mat_size = block.size(); + const int R_size = dmr_mats.size(); + std::vector soc_dmr_R(mat_size * R_size, TK(0.0, 0.0)); + + // transpose DMK block to row-major and axpy into the per-R buffer + std::vector dmk_row(mat_size); + if (ik_in >= 0) + { + // single k-point + const TK* const dmk_col = dm.dmk[ik_in + ik_begin].data() + block.col0 * ld_hk + block.row0; + transpose_dmk_block(dmk_col, ld_hk, block, dmk_row.data()); + for (int iR = 0; iR < R_size; ++iR) + { + BlasConnector::axpy(mat_size, + kphase_vec[ik_in][iR], + dmk_row.data(), + 1, + &soc_dmr_R[iR * mat_size], + 1); + } + } + else + { + // all k-points + for (int ik = 0; ik < dm._nk; ++ik) + { + const TK* const dmk_col = dm.dmk[ik + ik_begin].data() + block.col0 * ld_hk + block.row0; + transpose_dmk_block(dmk_col, ld_hk, block, dmk_row.data()); + for (int iR = 0; iR < R_size; ++iR) + { + BlasConnector::axpy(mat_size, + kphase_vec[ik][iR], + dmk_row.data(), + 1, + &soc_dmr_R[iR * mat_size], + 1); + } + } + } + + // spin-block column offsets for the 2x2 (upup, updown, downup, downdown) components + int spin_stride[4]{}; + constexpr int npol = 2; + for (int is = 0; is < npol; ++is) + { + for (int is2 = 0; is2 < npol; ++is2) + { + spin_stride[is * npol + is2] = col_stride * is + is2; + } + } + + // transform each 2x2 spin block to Pauli components and write back + TK spin_block[4]{}; + for (int iR = 0; iR < R_size; ++iR) + { + const TK* soc_mat = &soc_dmr_R[iR * mat_size]; + TR* dmr_mat = dmr_mats[iR]; + for (int irow = 0; irow < block.nrows; irow += 2) + { + for (int icol = 0; icol < block.ncols; icol += 2) + { + spin_block[0] = soc_mat[icol + spin_stride[0]]; + spin_block[1] = soc_mat[icol + spin_stride[1]]; + spin_block[2] = soc_mat[icol + spin_stride[2]]; + spin_block[3] = soc_mat[icol + spin_stride[3]]; + xyz_to_updown(spin_block, icol, spin_stride, dmr_mat); + } + soc_mat += block.ncols * 2; + dmr_mat += block.ncols * 2; + } + } +} + +// explicit instantiations for build_kphase +template void build_kphase, double>( + hamilt::AtomPair&, + const std::vector>&, + const int, + const std::map, std::complex>&, + std::vector>>&, + std::vector&); + +template void build_kphase, std::complex>( + hamilt::AtomPair>&, + const std::vector>&, + const int, + const std::map, std::complex>&, + std::vector>>&, + std::vector*>&); + +// explicit instantiations for transpose_dmk_block +template void transpose_dmk_block>( + const std::complex*, + const int, + const DmrBlock&, + std::complex*); + +// explicit instantiations for add_dmr_real +template void add_dmr_real, double>( + const DensityMatrix, double>&, + const DmrBlock&, + const int, + const std::vector>>&, + const int, + const int, + std::vector&); + +template void add_dmr_real, std::complex>( + const DensityMatrix, std::complex>&, + const DmrBlock&, + const int, + const std::vector>>&, + const int, + const int, + std::vector*>&); + +// explicit instantiations for add_dmr_soc +template void add_dmr_soc, double>( + const DensityMatrix, double>&, + const DmrBlock&, + const int, + const std::vector>>&, + const int, + const int, + const int, + std::vector&); + +template void add_dmr_soc, std::complex>( + const DensityMatrix, std::complex>&, + const DmrBlock&, + const int, + const std::vector>>&, + const int, + const int, + const int, + std::vector*>&); + +} // namespace module_dm diff --git a/source/source_estate/module_dm/dmr_full.cpp b/source/source_estate/module_dm/dmr_full.cpp new file mode 100644 index 00000000000..20ad119316a --- /dev/null +++ b/source/source_estate/module_dm/dmr_full.cpp @@ -0,0 +1,104 @@ +#include "density_matrix.h" + +#include + +#include "source_base/libm/libm.h" +#include "source_base/tool_title.h" +#include "source_base/tool_quit.h" +#include "source_base/constants.h" +#include "source_base/timer.h" +#include "source_cell/klist.h" + +namespace module_dm +{ + +template +void cal_dmr_full( + const DensityMatrix& dm, + hamilt::HContainer* dmR_out, + const int ik_in) +{ + ModuleBase::TITLE("DensityMatrix", "cal_dmr_full"); + + // validate ik_in: either -1 (all k-points) or a valid index + if (ik_in < -1 || ik_in >= dm._nk) + { + ModuleBase::WARNING_QUIT("module_dm::cal_dmr_full", + "ik_in out of range: must be -1 (all k) or 0 <= ik_in < nk"); + } + + ModuleBase::timer::start("DensityMatrix", "cal_dmr_full"); + const int ld_hk = dm.pv->nrow; + hamilt::HContainer* const dmr_full = dmR_out; + dmr_full->set_zero(); + const std::map, std::complex> no_hybrid_phase; +#ifdef _OPENMP +#pragma omp parallel for schedule(dynamic) +#endif + for (int i = 0; i < dmr_full->size_atom_pairs(); ++i) + { + hamilt::AtomPair& atom_pair = dmr_full->get_atom_pair(i); + const DmrBlock block = get_dmr_block(dm.pv, atom_pair.get_atom_i(), atom_pair.get_atom_j()); + const int R_size = atom_pair.get_R_size(); + + // precompute k-phase factors and collect DMR block pointers + std::vector> kphase_vec; + std::vector dmr_mats; + build_kphase(atom_pair, dm._kvec_d, dm._nk, no_hybrid_phase, kphase_vec, dmr_mats); + + // transpose DMK block to row-major, then axpy into each R-vector block + // DMR_ij(R) += e^{ik·R} * DMK_ij(k) + // (sum over ik when ik_in < 0, single ik when ik_in >= 0) + std::vector dmk_row(block.size()); + if (ik_in >= 0) + { + // single k-point + const TK* const dmk_col = dm.dmk[ik_in].data() + block.col0 * ld_hk + block.row0; + transpose_dmk_block(dmk_col, ld_hk, block, dmk_row.data()); + for (int iR = 0; iR < R_size; ++iR) + { + BlasConnector::axpy(block.size(), + kphase_vec[ik_in][iR], + dmk_row.data(), + 1, + dmr_mats[iR], + 1); + } + } + else + { + // all k-points + for (int ik = 0; ik < dm._nk; ++ik) + { + const TK* const dmk_col = dm.dmk[ik].data() + block.col0 * ld_hk + block.row0; + transpose_dmk_block(dmk_col, ld_hk, block, dmk_row.data()); + for (int iR = 0; iR < R_size; ++iR) + { + BlasConnector::axpy(block.size(), + kphase_vec[ik][iR], + dmk_row.data(), + 1, + dmr_mats[iR], + 1); + } + } + } + } + ModuleBase::timer::end("DensityMatrix", "cal_dmr_full"); +} + +template <> +void DensityMatrix::cal_dmr_full( + hamilt::HContainer>* dmR_out, + const int ik_in) const +{ +} +template <> +void DensityMatrix, double>::cal_dmr_full( + hamilt::HContainer>* dmR_out, + const int ik_in) const +{ + module_dm::cal_dmr_full(*this, dmR_out, ik_in); +} + +} // namespace module_dm diff --git a/source/source_estate/module_dm/dmr_gamma.cpp b/source/source_estate/module_dm/dmr_gamma.cpp new file mode 100644 index 00000000000..815fa2cb354 --- /dev/null +++ b/source/source_estate/module_dm/dmr_gamma.cpp @@ -0,0 +1,80 @@ +#include "density_matrix.h" + +#include "source_base/libm/libm.h" +#include "source_base/timer.h" +#include "source_base/tool_title.h" + +namespace module_dm +{ + +// calculate DMR from DMK using blas for gamma-only calculation +template <> +void DensityMatrix::cal_dmr(const int ik_in) +{ + ModuleBase::TITLE("DensityMatrix", "cal_dmr"); + using TK = double; + using TR = double; + + assert(ik_in == -1 || ik_in == 0); + assert(this->_nk == 1); + + assert(this->dmr.size()==this->spin_mult && "DMR has not been initialized!"); + + ModuleBase::timer::start("DensityMatrix", "cal_dmr"); + const int ld_hk = this->pv->nrow; + for (int is = 1; is <= this->spin_mult; ++is) + { + const int ik_begin = this->_nk * (is - 1); // jump this->_nk for spin_down if nspin==2 + hamilt::HContainer*const target_DMR = this->dmr[is - 1]; + target_DMR->set_zero(); +#ifdef _OPENMP +#pragma omp parallel for schedule(dynamic) +#endif + for (int i = 0; i < target_DMR->size_atom_pairs(); ++i) + { + hamilt::AtomPair& target_ap = target_DMR->get_atom_pair(i); + const int iat1 = target_ap.get_atom_i(); + const int iat2 = target_ap.get_atom_j(); + const int row_ap = this->pv->atom_begin_row[iat1]; + const int col_ap = this->pv->atom_begin_col[iat2]; + const int row_size = this->pv->get_nrow_atom(iat1); + const int col_size = this->pv->get_ncol_atom(iat2); + const int R_size = target_ap.get_R_size(); + assert(row_ap != -1 && col_ap != -1 && "Atom-pair not belong this process"); + assert(R_size == 1); + const ModuleBase::Vector3 R_index = target_ap.get_R_index(0); + assert(R_index.x == 0 && R_index.y == 0 && R_index.z == 0); + hamilt::BaseMatrix*const target_mat = target_ap.find_matrix(R_index); +#ifdef __DEBUG + if (target_mat == nullptr) + { + std::cout << "target_mat is nullptr" << std::endl; + continue; + } +#endif + // k index + constexpr TK kphase = 1; + // transpose DMK col=>row + const TK* DMK_mat_ptr + = this->dmk[0 + ik_begin].data() + + col_ap * this->pv->nrow + row_ap; + // set DMR element + TR* target_DMR_ptr = target_mat->get_pointer(); + for (int mu = 0; mu < row_size; ++mu) + { + BlasConnector::axpy(col_size, + kphase, + DMK_mat_ptr, + ld_hk, + target_DMR_ptr, + 1); + DMK_mat_ptr += 1; + target_DMR_ptr += col_size; + } + } + } + this->_dmr_ready = true; + ModuleBase::timer::end("DensityMatrix", "cal_dmr"); +} + +} // namespace module_dm diff --git a/source/source_estate/module_dm/dmr_init.cpp b/source/source_estate/module_dm/dmr_init.cpp new file mode 100644 index 00000000000..b3885123ae2 --- /dev/null +++ b/source/source_estate/module_dm/dmr_init.cpp @@ -0,0 +1,167 @@ +#include "density_matrix.h" + +#include "source_io/module_parameter/parameter.h" +#include "source_base/libm/libm.h" +#include "source_base/memory_recorder.h" +#include "source_base/timer.h" +#include "source_base/tool_title.h" +#include "source_cell/klist.h" + +#include +#include +#include + +namespace module_dm +{ + +// initialize density matrix DMR from UnitCell (mainly used in UnitTest) +template +void DensityMatrix::init_dmr(const Grid_Driver* GridD_in, const UnitCell* ucell) +{ + ModuleBase::TITLE("DensityMatrix", "init_dmr"); + this->clear_dmr(); + // construct a new DMR + std::unique_ptr> tmp_DMR(new hamilt::HContainer(this->pv)); + // set up a HContainer + for (int iat1 = 0; iat1 < ucell->nat; iat1++) + { + ModuleBase::Vector3 tau1 = ucell->get_tau(iat1); + int T1, I1; + ucell->iat2iait(iat1, &I1, &T1); + AdjacentAtomInfo adjs; + GridD_in->Find_atom(*ucell, tau1, T1, I1, &adjs); + // std::cout << "adjs.adj_num: " <itia2iat(T2, I2); + if (this->pv->is_invalid_atom_pair(iat1, iat2)) + { + continue; + } + ModuleBase::Vector3& R_index = adjs.box[ad]; + // std::cout << "R_index: " << R_index.x << " " << R_index.y << " " << R_index.z << std::endl; + hamilt::AtomPair tmp_ap(iat1, iat2, R_index, this->pv); + tmp_DMR->insert_pair(tmp_ap); + } + } + // allocate the memory of BaseMatrix in SR, and set the new values to zero + if (std::is_same::value) + { + tmp_DMR->fix_gamma(); + } + tmp_DMR->allocate(nullptr, true); + this->dmr.push_back(tmp_DMR.release()); + // add another DMR if nspin==2 + if (this->spin_mult == 2) + { + std::unique_ptr> tmp_DMR1(new hamilt::HContainer(*this->dmr[0])); + this->dmr.push_back(tmp_DMR1.release()); + } + ModuleBase::Memory::record("DensityMatrix::DMR", this->dmr.size() * this->dmr[0]->get_memory_size()); +} + +/// initialize density matrix DMR from UnitCell and RA (mainly used in UnitTest) +template +void DensityMatrix::init_dmr(Record_adj& ra, const UnitCell* ucell) +{ + ModuleBase::TITLE("DensityMatrix", "init_dmr"); + this->clear_dmr(); + // construct a new DMR + std::unique_ptr> tmp_DMR(new hamilt::HContainer(this->pv)); + // set up a HContainer + for (int iat1 = 0; iat1 < ucell->nat; iat1++) + { + ModuleBase::Vector3 tau1 = ucell->get_tau(iat1); + int T1, I1; + ucell->iat2iait(iat1, &I1, &T1); + for (int ad = 0; ad < ra.na_each[iat1]; ++ad) + { + const int T2 = ra.get_info(iat1, ad)[3]; + const int I2 = ra.get_info(iat1, ad)[4]; + int iat2 = ucell->itia2iat(T2, I2); + if (this->pv->is_invalid_atom_pair(iat1, iat2)) + { + continue; + } + hamilt::AtomPair tmp_ap(iat1, + iat2, + ra.get_info(iat1, ad)[0], + ra.get_info(iat1, ad)[1], + ra.get_info(iat1, ad)[2], + this->pv); + tmp_DMR->insert_pair(tmp_ap); + } + } + if (std::is_same::value) + { + tmp_DMR->fix_gamma(); + } + tmp_DMR->allocate(nullptr, true); + this->dmr.push_back(tmp_DMR.release()); + // add another DMR if nspin==2 + if (this->spin_mult == 2) + { + std::unique_ptr> tmp_DMR1(new hamilt::HContainer(*this->dmr[0])); + this->dmr.push_back(tmp_DMR1.release()); + } + ModuleBase::Memory::record("DensityMatrix::DMR", this->dmr.size() * this->dmr[0]->get_memory_size()); +} + +// initialize density matrix DMR from another HContainer (mainly used) +template +void DensityMatrix::init_dmr(const hamilt::HContainer& DMR_in) +{ + ModuleBase::TITLE("DensityMatrix", "init_dmr"); + this->clear_dmr(); + // set up a HContainer using another one + for (int is = 0; is < this->spin_mult; ++is) // loop over spin + { + std::unique_ptr> tmp_DMR(new hamilt::HContainer(DMR_in)); + // zero.out + tmp_DMR->set_zero(); + this->dmr.push_back(tmp_DMR.release()); + } + ModuleBase::Memory::record("DensityMatrix::DMR", this->dmr.size() * this->dmr[0]->get_memory_size()); +} + +template +void DensityMatrix::init_dmr(const hamilt::HContainer& DMR_in) +{ + ModuleBase::TITLE("DensityMatrix", "init_dmr"); + this->clear_dmr(); + // set up a HContainer using another one + int size_ap = DMR_in.size_atom_pairs(); + if (size_ap > 0) + { + const Parallel_Orbitals* paraV_ = DMR_in.get_atom_pair(0).get_paraV(); + std::unique_ptr> tmp_DMR(new hamilt::HContainer(paraV_)); + for (int iap = 0; iap < size_ap; iap++) + { + const int iat1 = DMR_in.get_atom_pair(iap).get_atom_i(); + const int iat2 = DMR_in.get_atom_pair(iap).get_atom_j(); + for (int ir = 0; ir < DMR_in.get_atom_pair(iap).get_R_size(); ir++) + { + const ModuleBase::Vector3 R_index = DMR_in.get_atom_pair(iap).get_R_index(ir); + hamilt::AtomPair tmp_ap(iat1, iat2, R_index, paraV_); + tmp_DMR->insert_pair(tmp_ap); + } + } + tmp_DMR->allocate(nullptr, true); + this->dmr.push_back(tmp_DMR.release()); + if (this->spin_mult == 2) + { + std::unique_ptr> tmp_DMR1(new hamilt::HContainer(*this->dmr[0])); + this->dmr.push_back(tmp_DMR1.release()); + } + } + ModuleBase::Memory::record("DensityMatrix::DMR", this->dmr.size() * this->dmr[0]->get_memory_size()); +} + +// T of HContainer can be double or std::complex +template class DensityMatrix; // Gamma-Only case +template class DensityMatrix, double>; // Multi-k case +template class DensityMatrix, std::complex>; // For EXX in future + +} // namespace module_dm diff --git a/source/source_estate/module_dm/dmr_k.cpp b/source/source_estate/module_dm/dmr_k.cpp new file mode 100644 index 00000000000..9fbbcbf8d22 --- /dev/null +++ b/source/source_estate/module_dm/dmr_k.cpp @@ -0,0 +1,137 @@ +#include "density_matrix.h" + +#include "source_base/libm/libm.h" +#include "source_base/tool_title.h" +#include "source_base/tool_quit.h" +#include "source_base/constants.h" +#include "source_base/timer.h" +#include "source_cell/klist.h" + +namespace module_dm +{ + +// shared inner loop of cal_dmr / cal_dmr_td: accumulate kphase * DMK into DMR blocks +template +void accumulate_dmr( + DensityMatrix& dm, + std::vector*>& dmR_out, + const std::map, std::complex>& phase_hybrid, + const int ik_in, + const char* func_name) +{ + const int ld_hk = dm.pv->nrow; + for (int is = 1; is <= dm.spin_mult; ++is) + { + const int ik_begin = dm._nk * (is - 1); // jump dm._nk for spin_down if nspin==2 + hamilt::HContainer* const dmr_spin = dmR_out[is - 1]; + // set zero since this function is called in every scf step + dmr_spin->set_zero(); + + if (dm.nspin != 1 && dm.nspin != 2 && dm.nspin != 4) + { + ModuleBase::WARNING_QUIT(func_name, "nspin must be 1, 2 or 4"); + } + + // accumulate kphase * DMK into DMR blocks; for nspin=4 (SOC) each orbital + // corresponds to a 2x2 spin block, so rows/cols step by 2 (physical spin + // dimension), and each block is transformed to Pauli components + // (rho_0, rho_x, rho_y, rho_z) +#ifdef _OPENMP +#pragma omp parallel for schedule(dynamic) +#endif + for (int i = 0; i < dmr_spin->size_atom_pairs(); ++i) + { + hamilt::AtomPair& atom_pair = dmr_spin->get_atom_pair(i); + const DmrBlock block = get_dmr_block(dm.pv, atom_pair.get_atom_i(), atom_pair.get_atom_j()); + + // precompute k-phase factors and collect DMR block pointers + std::vector> kphase_vec; + std::vector dmr_mats; + build_kphase(atom_pair, dm._kvec_d, dm._nk, phase_hybrid, kphase_vec, dmr_mats); + + if (dm.nspin == 1 || dm.nspin == 2) + { + // nspin=1/2: DMR_ij(R) += Re[ e^{ik·R} * DMK_ij(k) ] + // (sum over ik when ik_in < 0, single ik when ik_in >= 0) + add_dmr_real(dm, block, ik_begin, kphase_vec, ld_hk, ik_in, dmr_mats); + } + else if (dm.nspin == 4) + { + // nspin==4 (SOC): first accumulate S_ij(R) = sum_k e^{ik·R} * DMK_ij(k), + // then for each 2x2 spin block (upup, updown, downup, downdown) transform to + // Pauli components via xyz_to_updown: + // rho_0 = rho_upup + rho_downdown + // rho_x = rho_updown + rho_downup + // rho_y = Im(rho_updown) - Im(rho_downup) (sign for conjugated stored DM) + // rho_z = rho_upup - rho_downdown + add_dmr_soc(dm, block, ik_begin, kphase_vec, ld_hk, ik_in, + atom_pair.get_col_size(), dmr_mats); + } + else + { + ModuleBase::WARNING_QUIT(func_name, "nspin must be 1, 2 or 4"); + } + } + } +} + +// calculate DMR from DMK using blas for multi-k calculation +template +void cal_dmr( + DensityMatrix& dm, + std::vector*>& dmR_out, + const int ik_in) +{ + ModuleBase::TITLE("DensityMatrix", "cal_dmr"); + ModuleBase::timer::start("DensityMatrix", "cal_dmr"); + + // To check whether DMR has been initialized + if (dmR_out.size() != dm.spin_mult) + { + ModuleBase::WARNING_QUIT("module_dm::cal_dmr", + "DMR has not been initialized: dmR_out.size() != spin_mult!"); + } + + // validate ik_in: either -1 (all k-points) or a valid index + if (ik_in < -1 || ik_in >= dm._nk) + { + ModuleBase::WARNING_QUIT("module_dm::cal_dmr", + "ik_in out of range: must be -1 (all k) or 0 <= ik_in < nk"); + } + + const std::map, std::complex> no_hybrid_phase; + accumulate_dmr(dm, dmR_out, no_hybrid_phase, ik_in, "module_dm::cal_dmr"); + dm._dmr_ready = true; + ModuleBase::timer::end("DensityMatrix", "cal_dmr"); +} + +template <> +void DensityMatrix, double>::cal_dmr(const int ik_in) +{ + module_dm::cal_dmr(*this, this->dmr, ik_in); +} + +template <> +void DensityMatrix, std::complex>::cal_dmr(const int ik_in) +{ + module_dm::cal_dmr(*this, this->dmr, ik_in); +} + +// explicit instantiations for accumulate_dmr (used by both cal_dmr here and +// cal_dmr_td in dmr_td.cpp; without these the TD instantiations are missing +// at link time) +template void accumulate_dmr, double, double>( + DensityMatrix, double>&, + std::vector*>&, + const std::map, std::complex>&, + const int, + const char*); + +template void accumulate_dmr, std::complex, std::complex>( + DensityMatrix, std::complex>&, + std::vector>*>&, + const std::map, std::complex>&, + const int, + const char*); + +} // namespace module_dm diff --git a/source/source_estate/module_dm/dmr_td.cpp b/source/source_estate/module_dm/dmr_td.cpp new file mode 100644 index 00000000000..c665016b355 --- /dev/null +++ b/source/source_estate/module_dm/dmr_td.cpp @@ -0,0 +1,63 @@ +#include "density_matrix.h" + +#include "source_base/libm/libm.h" +#include "source_base/tool_title.h" +#include "source_base/tool_quit.h" +#include "source_base/constants.h" +#include "source_base/timer.h" +#include "source_cell/klist.h" + +namespace module_dm +{ + +template +void cal_dmr_td( + DensityMatrix& dm, + std::vector*>& dmR_out, + const std::map, std::complex>& phase_hybrid, + const ModuleBase::Vector3 At, + const int ik_in) +{ + ModuleBase::TITLE("DensityMatrix", "cal_dmr_td"); + assert(dmR_out.size() == dm.spin_mult && "DMR has not been initialized!"); + + // validate ik_in: either -1 (all k-points) or a valid index + if (ik_in < -1 || ik_in >= dm._nk) + { + ModuleBase::WARNING_QUIT("module_dm::cal_dmr_td", + "ik_in out of range: must be -1 (all k) or 0 <= ik_in < nk"); + } + + ModuleBase::timer::start("DensityMatrix", "cal_dmr_td"); + accumulate_dmr(dm, dmR_out, phase_hybrid, ik_in, "module_dm::cal_dmr_td"); + dm._dmr_ready = true; + ModuleBase::timer::end("DensityMatrix", "cal_dmr_td"); +} + +template <> +void DensityMatrix::cal_dmr_td( + const std::map, std::complex>& phase_hybrid, + const ModuleBase::Vector3 At, + const int ik_in) +{ + return; +} +template <> +void DensityMatrix, double>::cal_dmr_td( + const std::map, std::complex>& phase_hybrid, + const ModuleBase::Vector3 At, + const int ik_in) +{ + module_dm::cal_dmr_td(*this, this->dmr, phase_hybrid, At, ik_in); +} + +template <> +void DensityMatrix, std::complex>::cal_dmr_td( + const std::map, std::complex>& phase_hybrid, + const ModuleBase::Vector3 At, + const int ik_in) +{ + module_dm::cal_dmr_td(*this, this->dmr, phase_hybrid, At, ik_in); +} + +} // namespace module_dm diff --git a/source/source_estate/module_dm/edm_tddft.cpp b/source/source_estate/module_dm/edm_tddft.cpp new file mode 100644 index 00000000000..765017de3a3 --- /dev/null +++ b/source/source_estate/module_dm/edm_tddft.cpp @@ -0,0 +1,267 @@ +#include "edm_tddft.h" + +#include "source_base/module_container/ATen/core/tensor.h" // For ct::Tensor +#include "source_base/module_container/ATen/kernels/blas.h" +#include "source_base/module_container/ATen/kernels/lapack.h" +#include "source_base/module_container/ATen/kernels/memory.h" // memory operations (Tensor) +#include "source_base/module_device/memory_op.h" // memory operations +#include "source_base/module_external/lapack_connector.h" +#include "source_base/module_external/scalapack_connector.h" +#include "source_lcao/module_rt/gather_mat.h" // gatherMatrix and distributeMatrix +#include "source_lcao/module_rt/propagator.h" // Include header for create_identity_matrix + +namespace module_dm +{ +// use the original formula (Hamiltonian matrix) to calculate energy density matrix +void edm_tddft(Parallel_Orbitals& pv, + Setup_DM>& dmat, + K_Vectors& kv, + hamilt::Hamilt>* p_hamilt) +{ + ModuleBase::TITLE("elecstate", "edm_tddft"); + ModuleBase::timer::start("TD_Efficiency", "edm_tddft"); + + // Scalapack getrf/getri/gemm/geadd below take the GLOBAL matrix + // dimension; pv.nrow is only the per-process local row count and + // would truncate the operations to a submatrix of the full matrix. + const int nlocal = pv.get_global_row_size(); + assert(nlocal >= 0); + + dmat.dm->edmk.resize(kv.get_nks()); + + for (int ik = 0; ik < kv.get_nks(); ++ik) + { + p_hamilt->updateHk(ik); + std::complex* tmp_dmk = dmat.dm->get_dmk_ptr(ik); + ModuleBase::ComplexMatrix& tmp_edmk = dmat.dm->edmk[ik]; + +#ifdef __MPI + const int nloc = pv.nloc; + const int ncol = pv.ncol; + const int nrow = pv.nrow; + + tmp_edmk.create(ncol, nrow); + std::vector> Htmp_vec(nloc); + std::vector> Sinv_vec(nloc); + std::vector> tmp1_vec(nloc); + std::vector> tmp2_vec(nloc); + std::vector> tmp3_vec(nloc); + std::vector> tmp4_vec(nloc); + std::complex* Htmp = Htmp_vec.data(); + std::complex* Sinv = Sinv_vec.data(); + std::complex* tmp1 = tmp1_vec.data(); + std::complex* tmp2 = tmp2_vec.data(); + std::complex* tmp3 = tmp3_vec.data(); + std::complex* tmp4 = tmp4_vec.data(); + + // The buffers are value-initialized to zero by the std::vector + // constructors and fully overwritten below (Htmp/Sinv by copy, + // tmp1~tmp4 by gemm with beta == 0), so no explicit zeroing is needed. + const int inc = 1; + + hamilt::MatrixBlock> h_mat; + hamilt::MatrixBlock> s_mat; + + p_hamilt->matrix(h_mat, s_mat); + BlasConnector::copy(nloc, h_mat.p, inc, Htmp, inc); + BlasConnector::copy(nloc, s_mat.p, inc, Sinv, inc); + + std::vector ipiv(nloc, 0); + int info = 0; + const int one_int = 1; + + ScalapackConnector::getrf(nlocal, nlocal, Sinv, one_int, one_int, pv.desc, ipiv.data(), &info); + + int lwork = -1; + int liwork = -1; + + // if lwork == -1, then the size of work is (at least) of length 1. + std::vector> work(1, 0); + + // if liwork = -1, then the size of iwork is (at least) of length 1. + std::vector iwork(1, 0); + + ScalapackConnector::getri(nlocal, + Sinv, + one_int, + one_int, + pv.desc, + ipiv.data(), + work.data(), + &lwork, + iwork.data(), + &liwork, + &info); + + lwork = work[0].real(); + work.resize(lwork, 0); + liwork = iwork[0]; + iwork.resize(liwork, 0); + + ScalapackConnector::getri(nlocal, + Sinv, + one_int, + one_int, + pv.desc, + ipiv.data(), + work.data(), + &lwork, + iwork.data(), + &liwork, + &info); + + const char N_char = 'N'; + const char T_char = 'T'; + const std::complex one_complex = {1.0, 0.0}; + const std::complex zero_complex = {0.0, 0.0}; + const std::complex half_complex = {0.5, 0.0}; + + // tmp1 = Htmp * Sinv + ScalapackConnector::gemm(N_char, + N_char, + nlocal, + nlocal, + nlocal, + one_complex, + Htmp, + one_int, + one_int, + pv.desc, + Sinv, + one_int, + one_int, + pv.desc, + zero_complex, + tmp1, + one_int, + one_int, + pv.desc); + + // tmp2 = tmp1^T * tmp_dmk + ScalapackConnector::gemm(T_char, + N_char, + nlocal, + nlocal, + nlocal, + one_complex, + tmp1, + one_int, + one_int, + pv.desc, + tmp_dmk, + one_int, + one_int, + pv.desc, + zero_complex, + tmp2, + one_int, + one_int, + pv.desc); + + // tmp3 = Sinv * Htmp + ScalapackConnector::gemm(N_char, + N_char, + nlocal, + nlocal, + nlocal, + one_complex, + Sinv, + one_int, + one_int, + pv.desc, + Htmp, + one_int, + one_int, + pv.desc, + zero_complex, + tmp3, + one_int, + one_int, + pv.desc); + + // tmp4 = tmp_dmk * tmp3^T + ScalapackConnector::gemm(N_char, + T_char, + nlocal, + nlocal, + nlocal, + one_complex, + tmp_dmk, + one_int, + one_int, + pv.desc, + tmp3, + one_int, + one_int, + pv.desc, + zero_complex, + tmp4, + one_int, + one_int, + pv.desc); + + // tmp4 = 0.5 * (tmp2 + tmp4) + ScalapackConnector::geadd(N_char, + nlocal, + nlocal, + half_complex, + tmp2, + one_int, + one_int, + pv.desc, + half_complex, + tmp4, + one_int, + one_int, + pv.desc); + + BlasConnector::copy(nloc, tmp4, inc, tmp_edmk.c, inc); + +#else + // for serial version + tmp_edmk.create(pv.ncol, pv.nrow); + ModuleBase::ComplexMatrix Sinv(nlocal, nlocal); + ModuleBase::ComplexMatrix Htmp(nlocal, nlocal); + + hamilt::MatrixBlock> h_mat; + hamilt::MatrixBlock> s_mat; + + p_hamilt->matrix(h_mat, s_mat); + + for (int i = 0; i < nlocal; i++) + { + for (int j = 0; j < nlocal; j++) + { + Htmp(i, j) = h_mat.p[i * nlocal + j]; + Sinv(i, j) = s_mat.p[i * nlocal + j]; + } + } + int INFO = 0; + + int lwork = 3 * nlocal - 1; // tmp + std::vector> work_vec(lwork); + std::complex* work = work_vec.data(); + + int IPIV[nlocal]; + + LapackConnector::zgetrf(nlocal, nlocal, Sinv, nlocal, IPIV, &INFO); + LapackConnector::zgetri(nlocal, Sinv, nlocal, IPIV, work, lwork, &INFO); + // I just use ModuleBase::ComplexMatrix temporarily, and will change it + // to std::complex* + ModuleBase::ComplexMatrix tmp_dmk_base(nlocal, nlocal); + for (int i = 0; i < nlocal; i++) + { + for (int j = 0; j < nlocal; j++) + { + tmp_dmk_base(i, j) = tmp_dmk[i * nlocal + j]; + } + } + tmp_edmk = 0.5 * (Sinv * Htmp * tmp_dmk_base + tmp_dmk_base * Htmp * Sinv); +#endif + } // end ik + + ModuleBase::timer::end("TD_Efficiency", "edm_tddft"); + return; +} // edm_tddft + +} // namespace module_dm diff --git a/source/source_estate/module_dm/edm_tddft.h b/source/source_estate/module_dm/edm_tddft.h new file mode 100644 index 00000000000..b30ea10fd76 --- /dev/null +++ b/source/source_estate/module_dm/edm_tddft.h @@ -0,0 +1,22 @@ +#ifndef EDM_TDDFT_H +#define EDM_TDDFT_H + +#include "source_basis/module_ao/parallel_orbitals.h" +#include "source_cell/klist.h" +#include "source_estate/module_dm/dm_holder.h" +#include "source_hamilt/hamilt.h" + +namespace module_dm +{ +void edm_tddft(Parallel_Orbitals& pv, + Setup_DM>& dmat, + K_Vectors& kv, + hamilt::Hamilt>* p_hamilt); + +template +void edm_tddft_lapack(Parallel_Orbitals& pv, + Setup_DM>& dmat, + K_Vectors& kv, + hamilt::Hamilt>* p_hamilt); +} // namespace module_dm +#endif // EDM_TDDFT_H diff --git a/source/source_estate/module_dm/edm_tddft_lapack.cpp b/source/source_estate/module_dm/edm_tddft_lapack.cpp new file mode 100644 index 00000000000..5b8d71b31e5 --- /dev/null +++ b/source/source_estate/module_dm/edm_tddft_lapack.cpp @@ -0,0 +1,299 @@ +#include "edm_tddft.h" + +#include "source_base/module_container/ATen/core/tensor.h" +#include "source_base/module_container/ATen/kernels/blas.h" +#include "source_base/module_container/ATen/kernels/lapack.h" +#include "source_base/module_container/ATen/kernels/memory.h" +#include "source_base/module_device/memory_op.h" +#include "source_base/module_external/lapack_connector.h" +#include "source_base/module_external/scalapack_connector.h" +#include "source_lcao/module_rt/gather_mat.h" +#include "source_lcao/module_rt/propagator.h" + +namespace module_dm +{ + +// Template function for EDM calculation supporting CPU and GPU +template +void edm_tddft_lapack(Parallel_Orbitals& pv, + Setup_DM>& dmat, + K_Vectors& kv, + hamilt::Hamilt>* p_hamilt) +{ + ModuleBase::TITLE("elecstate", "edm_tddft_lapack"); + ModuleBase::timer::start("TD_Efficiency", "edm_tddft"); + + // Dense gather/lapack/gemm operations below take the GLOBAL matrix + // dimension; pv.nrow is only the per-process local row count. + const int nlocal = pv.get_global_row_size(); + assert(nlocal >= 0); + dmat.dm->edmk.resize(kv.get_nks()); + + // ct_device_type = ct::DeviceType::CpuDevice or ct::DeviceType::GpuDevice + ct::DeviceType ct_device_type = ct::DeviceTypeToEnum::value; + // ct_Device = ct::DEVICE_CPU or ct::DEVICE_GPU + using ct_Device = typename ct::PsiToContainer::type; + + // Memory operations + using syncmem_complex_h2d_op + = base_device::memory::synchronize_memory_op, Device, base_device::DEVICE_CPU>; + using syncmem_complex_d2h_op + = base_device::memory::synchronize_memory_op, base_device::DEVICE_CPU, Device>; + +#if ((defined __CUDA) /* || (defined __ROCM) */) + if (ct_device_type == ct::DeviceType::GpuDevice) + { + // Initialize cuBLAS & cuSOLVER handle + ct::kernels::createGpuSolverHandle(); + ct::kernels::createGpuBlasHandle(); + } +#endif // __CUDA + + for (int ik = 0; ik < kv.get_nks(); ++ik) + { + p_hamilt->updateHk(ik); + std::complex* tmp_dmk_local = dmat.dm->get_dmk_ptr(ik); + ModuleBase::ComplexMatrix& tmp_edmk = dmat.dm->edmk[ik]; + +#ifdef __MPI + int myid = 0; + const int root_proc = 0; + int num_procs = 1; + MPI_Comm_rank(MPI_COMM_WORLD, &myid); + MPI_Comm_size(MPI_COMM_WORLD, &num_procs); + + // 1. Prepare Data Source Pointers (Host) + // If np = 1, point directly to local data to avoid copy + // If np > 1, gather data and point to the gathered buffer + std::complex* h_src = nullptr; + std::complex* s_src = nullptr; + std::complex* dmk_src = nullptr; + + // Global containers (Used only when num_procs > 1) + module_rt::Matrix_g> h_mat_global, s_mat_global, dmk_global, edm_global; + + // Get Local Matrices + hamilt::MatrixBlock> h_mat_local, s_mat_local; + p_hamilt->matrix(h_mat_local, s_mat_local); + + if (num_procs == 1) + { + // Optimization: Direct access for single process + h_src = h_mat_local.p; + s_src = s_mat_local.p; + dmk_src = tmp_dmk_local; + } + else + { + // Standard Gather Logic for multi-process + module_rt::gatherMatrix(myid, root_proc, h_mat_local, h_mat_global); + module_rt::gatherMatrix(myid, root_proc, s_mat_local, s_mat_global); + + hamilt::MatrixBlock> dmk_local_block; + dmk_local_block.p = tmp_dmk_local; + dmk_local_block.desc = pv.desc; + module_rt::gatherMatrix(myid, root_proc, dmk_local_block, dmk_global); + + if (myid == root_proc) + { + h_src = h_mat_global.p.get(); + s_src = s_mat_global.p.get(); + dmk_src = dmk_global.p.get(); + } + } + + // 2. GPU Calculation (on Rank 0) + if (myid == root_proc) + { + ct::Tensor H_dev, S_dev, DMK_dev, ipiv_dev; + + // Allocate and Copy (H2D) + H_dev = ct::Tensor(ct::DataType::DT_COMPLEX_DOUBLE, ct_device_type, ct::TensorShape({nlocal, nlocal})); + syncmem_complex_h2d_op()(H_dev.template data>(), h_src, nlocal * nlocal); + + S_dev = ct::Tensor(ct::DataType::DT_COMPLEX_DOUBLE, ct_device_type, ct::TensorShape({nlocal, nlocal})); + syncmem_complex_h2d_op()(S_dev.template data>(), s_src, nlocal * nlocal); + + DMK_dev = ct::Tensor(ct::DataType::DT_COMPLEX_DOUBLE, ct_device_type, ct::TensorShape({nlocal, nlocal})); + syncmem_complex_h2d_op()(DMK_dev.template data>(), dmk_src, nlocal * nlocal); + + ipiv_dev = ct::Tensor(ct::DataType::DT_INT, ct_device_type, ct::TensorShape({nlocal})); + ipiv_dev.zero(); + + // --- Calculate S^-1 using getrf + getrs --- + // 1. LU decomposition S = P * L * U + ct::kernels::lapack_getrf, ct_Device>()(nlocal, + nlocal, + S_dev.template data>(), + nlocal, + ipiv_dev.template data()); + + // 2. Solve S * Sinv = I + ct::Tensor Sinv_dev = module_rt::create_identity_matrix>(nlocal, ct_device_type); + + ct::kernels::lapack_getrs, ct_Device>()('N', + nlocal, + nlocal, + S_dev.template data>(), + nlocal, + ipiv_dev.template data(), + Sinv_dev.template data>(), + nlocal); + + // --- EDM Calculation --- + std::complex one = {1.0, 0.0}; + std::complex zero = {0.0, 0.0}; + + // tmp1 = H * Sinv + ct::Tensor tmp1_dev(ct::DataType::DT_COMPLEX_DOUBLE, ct_device_type, ct::TensorShape({nlocal, nlocal})); + ct::kernels::blas_gemm, ct_Device>()('N', + 'N', + nlocal, + nlocal, + nlocal, + &one, + H_dev.template data>(), + nlocal, + Sinv_dev.template data>(), + nlocal, + &zero, + tmp1_dev.template data>(), + nlocal); + + // tmp2 = tmp1^T * DMK + ct::Tensor tmp2_dev(ct::DataType::DT_COMPLEX_DOUBLE, ct_device_type, ct::TensorShape({nlocal, nlocal})); + ct::kernels::blas_gemm, ct_Device>()('T', + 'N', + nlocal, + nlocal, + nlocal, + &one, + tmp1_dev.template data>(), + nlocal, + DMK_dev.template data>(), + nlocal, + &zero, + tmp2_dev.template data>(), + nlocal); + + // tmp3 = Sinv * H + ct::Tensor tmp3_dev(ct::DataType::DT_COMPLEX_DOUBLE, ct_device_type, ct::TensorShape({nlocal, nlocal})); + ct::kernels::blas_gemm, ct_Device>()('N', + 'N', + nlocal, + nlocal, + nlocal, + &one, + Sinv_dev.template data>(), + nlocal, + H_dev.template data>(), + nlocal, + &zero, + tmp3_dev.template data>(), + nlocal); + + // tmp4 = DMK * tmp3^T + ct::Tensor tmp4_dev(ct::DataType::DT_COMPLEX_DOUBLE, ct_device_type, ct::TensorShape({nlocal, nlocal})); + ct::kernels::blas_gemm, ct_Device>()('N', + 'T', + nlocal, + nlocal, + nlocal, + &one, + DMK_dev.template data>(), + nlocal, + tmp3_dev.template data>(), + nlocal, + &zero, + tmp4_dev.template data>(), + nlocal); + + // tmp4 = tmp2 + tmp4 + ct::kernels::blas_axpy, ct_Device>()(nlocal * nlocal, + &one, + tmp2_dev.template data>(), + 1, + tmp4_dev.template data>(), + 1); + + // tmp4 = 0.5 * tmp4 + std::complex half = {0.5, 0.0}; + ct::kernels::blas_scal, ct_Device>()(nlocal * nlocal, + &half, + tmp4_dev.template data>(), + 1); + + // 3. Retrieve Result (D2H) + std::complex* edm_dest = nullptr; + + if (num_procs == 1) + { + // Directly copy to target local matrix + tmp_edmk.create(pv.ncol, pv.nrow); + edm_dest = tmp_edmk.c; + } + else + { + // Wait to set up edm_dest after allocating global buffer + if (myid == root_proc && edm_global.p == nullptr) + { + edm_global.p.reset(new std::complex[nlocal * nlocal]); + } + edm_dest = edm_global.p.get(); + } + + if (num_procs == 1 || myid == root_proc) + { + syncmem_complex_d2h_op()(edm_dest, tmp4_dev.template data>(), nlocal * nlocal); + } + } + + // 4. Distribute (Only needed if num_procs > 1) + if (num_procs > 1) + { + if (edm_global.p == nullptr) + { + edm_global.p.reset(new std::complex[nlocal * nlocal]); + } + + edm_global.row = nlocal; + edm_global.col = nlocal; + edm_global.desc.reset(new int[9]{1, pv.desc[1], nlocal, nlocal, nlocal, nlocal, 0, 0, nlocal}); + + tmp_edmk.create(pv.ncol, pv.nrow); + hamilt::MatrixBlock> edm_local_block; + edm_local_block.p = tmp_edmk.c; + edm_local_block.desc = pv.desc; + module_rt::distributeMatrix(edm_local_block, edm_global); + } +#else + ModuleBase::WARNING_QUIT("elecstate::edm_tddft_lapack", "MPI is required for this function!"); +#endif // __MPI + } // end ik + +#if ((defined __CUDA) /* || (defined __ROCM) */) + if (ct_device_type == ct::DeviceType::GpuDevice) + { + // Destroy cuBLAS & cuSOLVER handle + ct::kernels::destroyGpuSolverHandle(); + ct::kernels::destroyGpuBlasHandle(); + } +#endif // __CUDA + + ModuleBase::timer::end("TD_Efficiency", "edm_tddft"); + return; +} // edm_tddft_lapack + +// Explicit instantiation of template functions +template void edm_tddft_lapack(Parallel_Orbitals& pv, + Setup_DM>& dmat, + K_Vectors& kv, + hamilt::Hamilt>* p_hamilt); +#if ((defined __CUDA) /* || (defined __ROCM) */) +template void edm_tddft_lapack(Parallel_Orbitals& pv, + Setup_DM>& dmat, + K_Vectors& kv, + hamilt::Hamilt>* p_hamilt); +#endif // __CUDA + +} // namespace module_dm diff --git a/source/source_estate/module_dm/init_dm.cpp b/source/source_estate/module_dm/init_dm.cpp deleted file mode 100644 index bb31ba48d6c..00000000000 --- a/source/source_estate/module_dm/init_dm.cpp +++ /dev/null @@ -1,66 +0,0 @@ -#include "source_estate/module_dm/init_dm.h" -#include "source_estate/module_dm/cal_dm_psi.h" -#include "source_estate/elecstate_tools.h" -#include "source_cell/cal_ux.h" -#include "source_lcao/rho_tau_lcao.h" // mohan add 2025-11-12 -#include "source_lcao/module_rt/td_info.h" - -template -void elecstate::init_dm(UnitCell& ucell, - elecstate::ElecState* pelec, - LCAO_domain::Setup_DM &dmat, - psi::Psi* psi, - Charge &chr, - const int iter, - const int exx_two_level_step) -{ - ModuleBase::TITLE("elecstate", "init_dm"); - - if (iter == 1 && exx_two_level_step == 0) - { - std::cout << " LCAO WAVEFUN -> CHARGE " << std::endl; - - elecstate::calEBand(pelec->ekb, pelec->wg, pelec->f_en); - - elecstate::cal_dm_psi(dmat.dm->get_paraV_pointer(), pelec->wg, *psi, *dmat.dm); - if (PARAM.inp.esolver_type!="tddft" && PARAM.inp.td_stype == 2) - { - dmat.dm->cal_DMR_td(TD_info::td_vel_op->get_phase_hybrid(), TD_info::cart_At); - } - else - { - dmat.dm->cal_DMR(); - } - - // mohan add 2025-11-12, use density matrix to calculate the charge density - LCAO_domain::dm2rho(dmat.dm->get_DMR_vector(), PARAM.inp.nspin, &chr); - - unitcell::cal_ux(ucell, PARAM.inp.nspin); - - //! update the potentials by using new electron charge density - pelec->pot->update_from_charge(&chr, &ucell); - - //! compute the correction energy for metals - pelec->f_en.descf = pelec->cal_delta_escf(); - } - - return; -} - - -template void elecstate::init_dm(UnitCell& ucell, - elecstate::ElecState* pelec, - LCAO_domain::Setup_DM &dmat, - psi::Psi* psi, - Charge &chr, - const int iter, - const int exx_two_level_step); - -template void elecstate::init_dm>(UnitCell& ucell, - elecstate::ElecState* pelec, - LCAO_domain::Setup_DM> &dmat, - psi::Psi>* psi, - Charge &chr, - const int iter, - const int exx_two_level_step); - diff --git a/source/source_estate/module_dm/init_dm.h b/source/source_estate/module_dm/init_dm.h deleted file mode 100644 index 2fd969638d5..00000000000 --- a/source/source_estate/module_dm/init_dm.h +++ /dev/null @@ -1,24 +0,0 @@ -#ifndef INIT_DM_H -#define INIT_DM_H - -#include "source_cell/unitcell.h" // use unitcell -#include "source_estate/elecstate.h"// use ElecState -#include "source_psi/psi.h" // use electronic wave functions -#include "source_estate/module_charge/charge.h" // use charge -#include "source_lcao/setup_dm.h" // define Setup_DM - -namespace elecstate -{ - -template -void init_dm(UnitCell& ucell, - ElecState* pelec, - LCAO_domain::Setup_DM &dmat, - psi::Psi* psi, - Charge &chr, - const int iter, - const int exx_two_level_step); - -} - -#endif diff --git a/source/source_estate/module_dm/test/CMakeLists.txt b/source/source_estate/module_dm/test/CMakeLists.txt deleted file mode 100644 index d5e8c19c3a5..00000000000 --- a/source/source_estate/module_dm/test/CMakeLists.txt +++ /dev/null @@ -1,72 +0,0 @@ -abacus_disable_feature_definitions(__MLALGO) -abacus_disable_feature_definitions(__CUDA) -abacus_disable_feature_definitions(__ROCM) - -install(DIRECTORY support DESTINATION ${CMAKE_CURRENT_BINARY_DIR}) - -if(TARGET MODULE_ESTATE_dm_io_test_serial) - abacus_disable_feature_definitions(__MPI) -endif() - -AddTest( - TARGET MODULE_ESTATE_dm_io_test_serial - LIBS parameter base device cell_info symmetry - SOURCES test_dm_io.cpp ../density_matrix.cpp ../density_matrix_io.cpp - ${ABACUS_SOURCE_DIR}/source_hamilt/module_hcontainer/base_matrix.cpp - ${ABACUS_SOURCE_DIR}/source_hamilt/module_hcontainer/hcontainer.cpp - ${ABACUS_SOURCE_DIR}/source_hamilt/module_hcontainer/atom_pair.cpp - ${ABACUS_SOURCE_DIR}/source_basis/module_ao/parallel_orbitals.cpp - ${ABACUS_SOURCE_DIR}/source_cell/klist.cpp ${ABACUS_SOURCE_DIR}/source_cell/klist_io.cpp - ${ABACUS_SOURCE_DIR}/source_cell/parallel_kpoints.cpp - ${ABACUS_SOURCE_DIR}/source_cell/reciprocal_grid.cpp - -) - -AddTest( - TARGET MODULE_ESTATE_dm_constructor_test - LIBS parameter base device symmetry - SOURCES test_dm_constructor.cpp ../density_matrix.cpp ../density_matrix_io.cpp tmp_mocks.cpp - ${ABACUS_SOURCE_DIR}/source_hamilt/module_hcontainer/base_matrix.cpp - ${ABACUS_SOURCE_DIR}/source_hamilt/module_hcontainer/hcontainer.cpp - ${ABACUS_SOURCE_DIR}/source_hamilt/module_hcontainer/atom_pair.cpp - ${ABACUS_SOURCE_DIR}/source_basis/module_ao/parallel_orbitals.cpp - ${ABACUS_SOURCE_DIR}/source_cell/klist.cpp ${ABACUS_SOURCE_DIR}/source_cell/klist_io.cpp - ${ABACUS_SOURCE_DIR}/source_cell/parallel_kpoints.cpp - ${ABACUS_SOURCE_DIR}/source_cell/reciprocal_grid.cpp -) - -AddTest( - TARGET MODULE_ESTATE_dm_init_test - LIBS parameter base device symmetry - SOURCES test_dm_r_init.cpp ../density_matrix.cpp ../density_matrix_io.cpp tmp_mocks.cpp - ${ABACUS_SOURCE_DIR}/source_hamilt/module_hcontainer/base_matrix.cpp - ${ABACUS_SOURCE_DIR}/source_hamilt/module_hcontainer/hcontainer.cpp - ${ABACUS_SOURCE_DIR}/source_hamilt/module_hcontainer/atom_pair.cpp - ${ABACUS_SOURCE_DIR}/source_basis/module_ao/parallel_orbitals.cpp - ${ABACUS_SOURCE_DIR}/source_cell/klist.cpp ${ABACUS_SOURCE_DIR}/source_cell/klist_io.cpp - ${ABACUS_SOURCE_DIR}/source_cell/parallel_kpoints.cpp - ${ABACUS_SOURCE_DIR}/source_cell/reciprocal_grid.cpp -) - -AddTest( - TARGET MODULE_ESTATE_dm_cal_DMR_test - LIBS parameter base device symmetry - SOURCES test_cal_dm_r.cpp ../density_matrix.cpp ../density_matrix_io.cpp tmp_mocks.cpp - ${ABACUS_SOURCE_DIR}/source_hamilt/module_hcontainer/base_matrix.cpp - ${ABACUS_SOURCE_DIR}/source_hamilt/module_hcontainer/hcontainer.cpp - ${ABACUS_SOURCE_DIR}/source_hamilt/module_hcontainer/atom_pair.cpp - ${ABACUS_SOURCE_DIR}/source_basis/module_ao/parallel_orbitals.cpp - ${ABACUS_SOURCE_DIR}/source_cell/klist.cpp ${ABACUS_SOURCE_DIR}/source_cell/klist_io.cpp - ${ABACUS_SOURCE_DIR}/source_cell/parallel_kpoints.cpp - ${ABACUS_SOURCE_DIR}/source_cell/reciprocal_grid.cpp -) - -AddTest( - TARGET MODULE_ESTATE_dm_soc_magnetization_roundtrip_test - LIBS parameter base device - SOURCES test_soc_magnetization_roundtrip.cpp ../density_matrix.cpp ../density_matrix_io.cpp tmp_mocks.cpp - ${ABACUS_SOURCE_DIR}/source_hamilt/module_hcontainer/base_matrix.cpp - ${ABACUS_SOURCE_DIR}/source_hamilt/module_hcontainer/hcontainer.cpp - ${ABACUS_SOURCE_DIR}/source_hamilt/module_hcontainer/atom_pair.cpp - ${ABACUS_SOURCE_DIR}/source_basis/module_ao/parallel_orbitals.cpp -) diff --git a/source/source_estate/module_dm/test/test_cal_dm_r.cpp b/source/source_estate/module_dm/test/test_cal_dm_r.cpp deleted file mode 100644 index c150690d26d..00000000000 --- a/source/source_estate/module_dm/test/test_cal_dm_r.cpp +++ /dev/null @@ -1,352 +0,0 @@ -#include - -#include "gmock/gmock.h" -#include "gtest/gtest.h" -#include "source_estate/module_dm/density_matrix.h" -#include "source_hamilt/module_hcontainer/hcontainer.h" -#include "source_cell/klist.h" - -/************************************************ - * unit test of DensityMatrix constructor - ***********************************************/ - -/** - * This unit test construct a DensityMatrix object - */ - -// test_size is the number of atoms in the unitcell -// modify test_size to test different size of unitcell -int test_size = 10; -int test_nw = 10; - -class DMTest : public testing::Test -{ - protected: - Parallel_Orbitals* paraV; - int dsize; - int my_rank = 0; - UnitCell ucell; - void SetUp() override - { -#ifdef __MPI - // MPI parallel settings - MPI_Comm_size(MPI_COMM_WORLD, &dsize); - MPI_Comm_rank(MPI_COMM_WORLD, &my_rank); -#endif - - // set up a unitcell, with one element and test_size atoms, each atom has test_nw orbitals - ucell.ntype = 1; - ucell.nat = test_size; - ucell.atoms = new Atom[ucell.ntype]; - ucell.iat2it = new int[ucell.nat]; - ucell.iat2ia = new int[ucell.nat]; - ucell.atoms[0].tau.resize(ucell.nat); - ucell.itia2iat.create(ucell.ntype, ucell.nat); - for (int iat = 0; iat < ucell.nat; iat++) - { - ucell.iat2it[iat] = 0; - ucell.iat2ia[iat] = iat; - ucell.atoms[0].tau[iat] = ModuleBase::Vector3(0.0, 0.0, 0.0); - ucell.itia2iat(0, iat) = iat; - } - ucell.atoms[0].na = test_size; - ucell.atoms[0].nw = test_nw; - ucell.atoms[0].iw2l.resize(test_nw); - ucell.atoms[0].iw2m.resize(test_nw); - ucell.atoms[0].iw2n.resize(test_nw); - for (int iw = 0; iw < test_nw; ++iw) - { - ucell.atoms[0].iw2l[iw] = 0; - ucell.atoms[0].iw2m[iw] = 0; - ucell.atoms[0].iw2n[iw] = 0; - } - ucell.set_iat2iwt(1); - init_parav(); - - // set paraV - init_parav(); - } - - void TearDown() override - { - delete paraV; - delete[] ucell.atoms; - } - -#ifdef __MPI - void init_parav() - { - int nb = 2; - int global_row = test_size * test_nw; - int global_col = test_size * test_nw; - std::ofstream ofs_running; - paraV = new Parallel_Orbitals(); - paraV->init(global_row, global_col, nb, MPI_COMM_WORLD); - paraV->set_atomic_trace(ucell.get_iat2iwt(), test_size, global_row); - } -#else - void init_parav() - { - } -#endif -}; - -TEST_F(DMTest, cal_DMR_full) -{ - // get my rank of this process - int my_rank = 0; -#ifdef __MPI - MPI_Comm_rank(MPI_COMM_WORLD, &my_rank); -#endif - // output dim and nrow, ncol - if (my_rank == 0) - { - std::cout << "my rank: " << my_rank << " dim0: " << paraV->dim0 << " dim1:" << paraV->dim1 << std::endl; - std::cout << "my rank: " << my_rank << " nrow: " << paraV->nrow << " ncol:" << paraV->ncol << std::endl; - } - else - { - std::cout << "my rank: " << my_rank << " nrow: " << paraV->nrow << " ncol:" << paraV->ncol << std::endl; - } - // initalize a kvectors, Gamma-only - K_Vectors* kv = nullptr; - int nspin = 4; - int nks = 2; // since nspin = 2 - kv = new K_Vectors; - kv->set_nks(nks); - kv->kvec_d.resize(nks); - // construct DM - elecstate::DensityMatrix, double> DM(paraV, nspin, kv->kvec_d, kv->get_nks()); - // set this->_DMK - for (int is = 1; is <= nspin; is++) - { - for (int ik = 0; ik < kv->get_nks(); ik++) - { - for (int i = 0; i < paraV->nrow; i++) - { - for (int j = 0; j < paraV->ncol; j++) - { - DM.set_DMK(is, ik, i, j, std::complex(0.77, 0.77)); - } - } - } - } - // initialize dmR_full - hamilt::HContainer> dmR_full(ucell, paraV); - // calculate this->_DMR - std::chrono::high_resolution_clock::time_point start_time = std::chrono::high_resolution_clock::now(); - DM.cal_DMR_full(&dmR_full); - std::chrono::high_resolution_clock::time_point end_time = std::chrono::high_resolution_clock::now(); - std::chrono::duration elapsed_time - = std::chrono::duration_cast>(end_time - start_time); - std::cout << "my rank: " << my_rank << " elapsed time blas: " << elapsed_time.count() << std::endl; - // compare the result - for (int i = 0; i < dmR_full.size_atom_pairs(); i++) - { - const std::complex* ptr1 = dmR_full.get_atom_pair(i).get_HR_values(0, 0, 0).get_pointer(); - // - for (int j = 0; j < dmR_full.get_atom_pair(i).get_size(); j++) - { - //std::cout << "my rank: " << my_rank << " i: " << i << " j: " << j << " value: " << ptr1[j] << std::endl; - EXPECT_NEAR(ptr1[j].real(), 1.54, 1e-10); - EXPECT_NEAR(ptr1[j].imag(), 1.54, 1e-10); - } - } - delete kv; -} - -TEST_F(DMTest, cal_DMR_blas_double) -{ - // get my rank of this process - int my_rank = 0; -#ifdef __MPI - MPI_Comm_rank(MPI_COMM_WORLD, &my_rank); -#endif - // output dim and nrow, ncol - if (my_rank == 0) - { - std::cout << "my rank: " << my_rank << " dim0: " << paraV->dim0 << " dim1:" << paraV->dim1 << std::endl; - std::cout << "my rank: " << my_rank << " nrow: " << paraV->nrow << " ncol:" << paraV->ncol << std::endl; - } - else - { - std::cout << "my rank: " << my_rank << " nrow: " << paraV->nrow << " ncol:" << paraV->ncol << std::endl; - } - // initalize a kvectors, Gamma-only - K_Vectors* kv = nullptr; - int nspin = 2; - int nks = 2; // since nspin = 2 - kv = new K_Vectors; - kv->set_nks(nks); - kv->kvec_d.resize(nks); - // construct DM - elecstate::DensityMatrix DM(paraV, nspin, kv->kvec_d, kv->get_nks() / nspin); - // set this->_DMK - for (int is = 1; is <= nspin; is++) - { - for (int ik = 0; ik < kv->get_nks() / nspin; ik++) - { - for (int i = 0; i < paraV->nrow; i++) - { - for (int j = 0; j < paraV->ncol; j++) - { - DM.set_DMK(is, ik, i, j, 0.77); - } - } - } - } - // initialize this->_DMR - Grid_Driver gd(0, 0); - DM.init_DMR(&gd, &ucell); - // set Gamma-only - for (int is = 1; is <= nspin; is++) - { - DM.get_DMR_pointer(is)->fix_gamma(); - } - // calculate this->_DMR - std::chrono::high_resolution_clock::time_point start_time = std::chrono::high_resolution_clock::now(); - DM.cal_DMR(); - std::chrono::high_resolution_clock::time_point end_time = std::chrono::high_resolution_clock::now(); - std::chrono::duration elapsed_time - = std::chrono::duration_cast>(end_time - start_time); - std::cout << "my rank: " << my_rank << " elapsed time blas: " << elapsed_time.count() << std::endl; - // compare the result - for (int i = 0; i < DM.get_DMR_pointer(1)->size_atom_pairs(); i++) - { - double* ptr1 = DM.get_DMR_pointer(1)->get_atom_pair(i).get_HR_values(0, 0, 0).get_pointer(); - // - for (int j = 0; j < DM.get_DMR_pointer(1)->get_atom_pair(i).get_size(); j++) - { - // std::cout << "my rank: " << my_rank << " i: " << i << " j: " << j << " value: " << ptr1[j] << std::endl; - EXPECT_NEAR(ptr1[j], 0.77, 1e-10); - } - } - delete kv; -} - -TEST_F(DMTest, cal_DMR_blas_complex) -{ - // get my rank of this process - int my_rank = 0; -#ifdef __MPI - MPI_Comm_rank(MPI_COMM_WORLD, &my_rank); -#endif - // output dim and nrow, ncol - if (my_rank == 0) - { - std::cout << "my rank: " << my_rank << " dim0: " << paraV->dim0 << " dim1:" << paraV->dim1 << std::endl; - std::cout << "my rank: " << my_rank << " nrow: " << paraV->nrow << " ncol:" << paraV->ncol << std::endl; - } - else - { - std::cout << "my rank: " << my_rank << " nrow: " << paraV->nrow << " ncol:" << paraV->ncol << std::endl; - } - // initalize a kvectors - K_Vectors* kv = nullptr; - int nspin = 2; - int nks = 4; // since nspin = 2 - kv = new K_Vectors; - kv->set_nks(nks); - kv->kvec_d.resize(nks); - kv->kvec_d[1].x = 0.5; - kv->kvec_d[3].x = 0.5; - // construct DM - elecstate::DensityMatrix, double> DM(paraV, nspin, kv->kvec_d, kv->get_nks() / nspin); - // set this->_DMK - for (int is = 1; is <= nspin; is++) - { - for (int ik = 0; ik < kv->get_nks() / nspin; ik++) - { - for (int i = 0; i < paraV->nrow; i++) - { - for (int j = 0; j < paraV->ncol; j++) - { - DM.set_DMK(is, ik, i, j, is * 0.77 * (ik + 1)); - } - } - } - } - // initialize this->_DMR - Grid_Driver gd(0, 0); - DM.init_DMR(&gd, &ucell); - // calculate this->_DMR - std::chrono::high_resolution_clock::time_point start_time = std::chrono::high_resolution_clock::now(); - DM.cal_DMR(); - std::chrono::high_resolution_clock::time_point end_time = std::chrono::high_resolution_clock::now(); - std::chrono::duration elapsed_time - = std::chrono::duration_cast>(end_time - start_time); - std::cout << "my rank: " << my_rank << " elapsed time blas: " << elapsed_time.count() << std::endl; - // compare the result for spin-up - for (int i = 0; i < DM.get_DMR_pointer(1)->size_atom_pairs(); i++) - { - double* ptr1 = DM.get_DMR_pointer(1)->get_atom_pair(i).get_HR_values(1, 1, 1).get_pointer(); - // - for (int j = 0; j < DM.get_DMR_pointer(1)->get_atom_pair(i).get_size(); j++) - { - // std::cout << "my rank: " << my_rank << " i: " << i << " j: " << j << " value: " << ptr1[j] << std::endl; - EXPECT_NEAR(ptr1[j], -0.77, 1e-10); - } - } - // compare the result for spin-down - for (int i = 0; i < DM.get_DMR_pointer(2)->size_atom_pairs(); i++) - { - double* ptr1 = DM.get_DMR_pointer(2)->get_atom_pair(i).get_HR_values(1, 1, 1).get_pointer(); - // - for (int j = 0; j < DM.get_DMR_pointer(2)->get_atom_pair(i).get_size(); j++) - { - // std::cout << "my rank: " << my_rank << " i: " << i << " j: " << j << " value: " << ptr1[j] << std::endl; - EXPECT_NEAR(ptr1[j], -0.77 * 2, 1e-10); - } - } - // calculate DMR_total - DM.switch_dmr(1); - // compare the result for spin-up after sum - for (int i = 0; i < DM.get_DMR_pointer(1)->size_atom_pairs(); i++) - { - double* ptr1 = DM.get_DMR_pointer(1)->get_atom_pair(i).get_HR_values(1, 1, 1).get_pointer(); - // - for (int j = 0; j < DM.get_DMR_pointer(1)->get_atom_pair(i).get_size(); j++) - { - //std::cout << "my rank: " << my_rank << " i: " << i << " j: " << j << " value: " << ptr1[j] << std::endl; - EXPECT_NEAR(ptr1[j], -0.77 * 3, 1e-10); - } - } - // restore to normal DMR - DM.switch_dmr(0); - for (int i = 0; i < DM.get_DMR_pointer(1)->size_atom_pairs(); i++) - { - double* ptr1 = DM.get_DMR_pointer(1)->get_atom_pair(i).get_HR_values(1, 1, 1).get_pointer(); - // - for (int j = 0; j < DM.get_DMR_pointer(1)->get_atom_pair(i).get_size(); j++) - { - //std::cout << "my rank: " << my_rank << " i: " << i << " j: " << j << " value: " << ptr1[j] << std::endl; - EXPECT_NEAR(ptr1[j], -0.77, 1e-10); - } - } - // calculate DMR_differenct - DM.switch_dmr(2); - for (int i = 0; i < DM.get_DMR_pointer(1)->size_atom_pairs(); i++) - { - double* ptr1 = DM.get_DMR_pointer(1)->get_atom_pair(i).get_HR_values(1, 1, 1).get_pointer(); - // - for (int j = 0; j < DM.get_DMR_pointer(1)->get_atom_pair(i).get_size(); j++) - { - //std::cout << "my rank: " << my_rank << " i: " << i << " j: " << j << " value: " << ptr1[j] << std::endl; - EXPECT_NEAR(ptr1[j], 0.77, 1e-10); - } - } - delete kv; -} - -int main(int argc, char** argv) -{ -#ifdef __MPI - MPI_Init(&argc, &argv); -#endif - testing::InitGoogleTest(&argc, argv); - int result = RUN_ALL_TESTS(); -#ifdef __MPI - MPI_Finalize(); -#endif - return result; -} diff --git a/source/source_estate/module_dm/test/test_cal_dmk_psi.cpp b/source/source_estate/module_dm/test/test_cal_dmk_psi.cpp deleted file mode 100644 index 8806a2fbe23..00000000000 --- a/source/source_estate/module_dm/test/test_cal_dmk_psi.cpp +++ /dev/null @@ -1,176 +0,0 @@ -#include - -#include "gmock/gmock.h" -#include "gtest/gtest.h" -#include "source_estate/module_dm/density_matrix.h" -#include "source_hamilt/module_hcontainer/hcontainer.h" -#include "source_cell/klist.h" - -/************************************************ - * unit test of DensityMatrix constructor - ***********************************************/ - -/** - * This unit test cal_dmk_psi - */ - -// test_size is the number of atoms in the unitcell -// modify test_size to test different size of unitcell -int test_size = 10; -int test_nw = 26; - -class DMTest : public testing::Test -{ - protected: - Parallel_Orbitals* paraV; - int dsize; - int my_rank = 0; - UnitCell ucell; - void SetUp() override - { -#ifdef __MPI - // MPI parallel settings - MPI_Comm_size(MPI_COMM_WORLD, &dsize); - MPI_Comm_rank(MPI_COMM_WORLD, &my_rank); -#endif - - // set up a unitcell, with one element and test_size atoms, each atom has test_nw orbitals - ucell.ntype = 1; - ucell.nat = test_size; - ucell.atoms = new Atom[ucell.ntype]; - ucell.iat2it = new int[ucell.nat]; - ucell.iat2ia = new int[ucell.nat]; - ucell.atoms[0].tau = new ModuleBase::Vector3[ucell.nat]; - ucell.itia2iat.create(ucell.ntype, ucell.nat); - for (int iat = 0; iat < ucell.nat; iat++) - { - ucell.iat2it[iat] = 0; - ucell.iat2ia[iat] = iat; - ucell.atoms[0].tau[iat] = ModuleBase::Vector3(0.0, 0.0, 0.0); - ucell.itia2iat(0, iat) = iat; - } - ucell.atoms[0].na = test_size; - ucell.atoms[0].nw = test_nw; - ucell.atoms[0].iw2l = new int[test_nw]; - ucell.atoms[0].iw2m = new int[test_nw]; - ucell.atoms[0].iw2n = new int[test_nw]; - for (int iw = 0; iw < test_nw; ++iw) - { - ucell.atoms[0].iw2l[iw] = 0; - ucell.atoms[0].iw2m[iw] = 0; - ucell.atoms[0].iw2n[iw] = 0; - } - ucell.set_iat2iwt(1); - init_parav(); - // set paraV - init_parav(); - } - - void TearDown() - { - delete paraV; - delete[] ucell.atoms[0].tau; - delete[] ucell.atoms[0].iw2l; - delete[] ucell.atoms[0].iw2m; - delete[] ucell.atoms[0].iw2n; - delete[] ucell.atoms; - } - -#ifdef __MPI - void init_parav() - { - int nb = 2; - int global_row = test_size * test_nw; - int global_col = test_size * test_nw; - std::ofstream ofs_running; - paraV = new Parallel_Orbitals(); - paraV->init(global_row, global_col, nb, MPI_COMM_WORLD); - paraV->set_atomic_trace(ucell.get_iat2iwt(), test_size, global_row); - } -#else - void init_parav() - { - } -#endif -}; - -TEST_F(DMTest, cal_dmk_psi_nspin1) -{ - // initalize a kvectors - K_Vectors* kv = nullptr; - int nks = 2; - kv = new K_Vectors; - kv->set_nks(nks); - kv->kvec_d.resize(nks); - kv->kvec_d[1].x = 0.5; - // construct DM - std::cout << "dim0: " << paraV->dim0 << " dim1:" << paraV->dim1 << std::endl; - std::cout << "nrow: " << paraV->nrow << " ncol:" << paraV->ncol << std::endl; - int nspin = 1; - elecstate::DensityMatrix DM(kv, paraV, nspin); - // compare - EXPECT_EQ(DM.get_DMK_nks(), kv->get_nks()); - EXPECT_EQ(DM.get_DMK_nrow(), paraV->nrow); - EXPECT_EQ(DM.get_DMK_ncol(), paraV->ncol); - - // set elements of DMK - for (int is = 1; is <= nspin; is++) - { - for (int ik = 0; ik < kv->get_nks() / nspin; ik++) - { - for (int i = 0; i < paraV->nrow; i++) - { - for (int j = 0; j < paraV->ncol; j++) - { - DM.set_DMK(is, ik, i, j, is + ik * i + j); - } - } - } - } - // compare - for (int is = 1; is <= nspin; is++) - { - for (int ik = 0; ik < kv->get_nks() / nspin; ik++) - { - for (int i = 0; i < paraV->nrow; i++) - { - for (int j = 0; j < paraV->ncol; j++) - { - EXPECT_EQ(DM.get_DMK(is, ik, i, j), is + ik * i + j); - } - } - } - } - // test for get_DMK_pointer - for (int is = 1; is <= nspin; is++) - { - int ik_begin = (is - 1) * kv->get_nks() / nspin; - for (int ik = 0; ik < kv->get_nks() / nspin; ik++) - { - double* ptr = DM.get_DMK_pointer(ik + ik_begin); - for (int i = 0; i < paraV->nrow; i++) - { - for (int j = 0; j < paraV->ncol; j++) - { - // std::cout << ptr[i*paraV->ncol+j] << " "; - EXPECT_EQ(ptr[i * paraV->ncol + j], is + ik * i + j); - } - } - } - } - // delete kv - delete kv; -} - -int main(int argc, char** argv) -{ -#ifdef __MPI - MPI_Init(&argc, &argv); -#endif - testing::InitGoogleTest(&argc, argv); - int result = RUN_ALL_TESTS(); -#ifdef __MPI - MPI_Finalize(); -#endif - return result; -} diff --git a/source/source_estate/module_dm/test/test_dm_io.cpp b/source/source_estate/module_dm/test/test_dm_io.cpp deleted file mode 100644 index 8c1565b0a84..00000000000 --- a/source/source_estate/module_dm/test/test_dm_io.cpp +++ /dev/null @@ -1,166 +0,0 @@ -#include - -#include "gmock/gmock.h" -#include "gtest/gtest.h" -#include "source_cell/unitcell.h" -#include "source_estate/module_dm/density_matrix.h" -#include "prepare_unitcell.h" - -// mock functions - -Magnetism::Magnetism() -{ - this->tot_mag = 0.0; - this->abs_mag = 0.0; -} -Magnetism::~Magnetism() -{ -} - -#include "source_cell/klist.h" -#include "source_cell/module_neighbor/sltk_grid_driver.h" -// mock find_atom() function -void Grid_Driver::Find_atom(const UnitCell& ucell, - const ModuleBase::Vector3& tau, - const int& T, - const int& I, - AdjacentAtomInfo* adjs) const -{ -} -Grid::Grid(const int& test_grid_in) : test_grid(test_grid_in) -{ -} -Grid::~Grid() -{ -} -Grid_Driver::Grid_Driver(const int& test_d_in,const int& test_grid_in) - : Grid(test_grid_in), test_deconstructor(test_d_in){} -Grid_Driver::~Grid_Driver() -{ -} -// mocke functions - -/************************************************ - * unit test of DensityMatrix constructor - ***********************************************/ - -/** - * This unit test construct a DensityMatrix object - */ - -// test_size is the number of atoms in the unitcell -// modify test_size to test different size of unitcell -int test_size = 2; -int test_nw = 13; - -class DMTest : public testing::Test -{ - protected: - Parallel_Orbitals* paraV; - int dsize; - int my_rank = 0; - UnitCell* ucell; - UcellTestPrepare utp = UcellTestLib["Si"]; - std::vector DMK; - K_Vectors* kv = nullptr; - // nw is the number of orbitals of each atom - // it should container ucell.nat elements - std::vector nw = {13}; - int nks = 2; - int nlocal = 0; - void SetUp() override - { -#ifdef __MPI - // MPI parallel settings - MPI_Comm_size(MPI_COMM_WORLD, &dsize); - MPI_Comm_rank(MPI_COMM_WORLD, &my_rank); -#endif - // initalize a unitcell - ucell = utp.SetUcellInfo(nw, nlocal); - ucell->set_iat2iwt(1); - // initalize a kvectors - kv = new K_Vectors; - kv->set_nks(nks); - kv->kvec_d.resize(nks); - kv->kvec_d[1].x = 0.5; - // set paraV - init_parav(); - } - - void TearDown() override - { - DMK.clear(); - delete kv; - delete paraV; - } - -#ifdef __MPI - void init_parav() - { - int nb = 2; - int global_row = test_size * test_nw; - int global_col = test_size * test_nw; - std::ofstream ofs_running; - paraV = new Parallel_Orbitals(); - paraV->init(global_row, global_col, nb, MPI_COMM_WORLD); - paraV->set_atomic_trace(ucell->get_iat2iwt(), test_size, global_row); - } -#else - void init_parav() - { - } -#endif -}; - -TEST_F(DMTest, DMConstructor1) -{ - // - int nspin = 1; - // construct DM - std::cout << paraV->nrow << paraV->ncol << std::endl; - elecstate::DensityMatrix DM(paraV, nspin, kv->kvec_d, kv->get_nks()); - // read DMK - std::string directory = "./support/"; - for (int is = 1; is <= nspin; ++is) - { - for (int ik = 0; ik < kv->get_nks() / nspin; ++ik) - { - DM.read_DMK(directory, is, ik); - } - } - // write DMK - directory = "./support/output"; - for (int is = 1; is <= nspin; ++is) - { - for (int ik = 0; ik < kv->get_nks() / nspin; ++ik) - { - DM.write_DMK(directory, is, ik); - } - } - // construct a new DM - elecstate::DensityMatrix DM1(paraV, nspin, kv->kvec_d, kv->get_nks()); - directory = "./support/output"; - for (int is = 1; is <= nspin; ++is) - { - for (int ik = 0; ik < kv->get_nks() / nspin; ++ik) - { - DM1.read_DMK(directory, is, ik); - } - } - // compare DMK1 with DMK - EXPECT_NEAR(DM.get_DMK(1, 0, 0, 0), DM1.get_DMK(1, 0, 0, 0), 1e-6); - EXPECT_NEAR(DM.get_DMK(1, 1, 25, 25), DM1.get_DMK(1, 1, 25, 25), 1e-6); -} - -int main(int argc, char** argv) -{ -#ifdef __MPI - MPI_Init(&argc, &argv); -#endif - testing::InitGoogleTest(&argc, argv); - int result = RUN_ALL_TESTS(); -#ifdef __MPI - MPI_Finalize(); -#endif - return result; -} diff --git a/source/source_estate/module_dm/test/test_dm_r_init.cpp b/source/source_estate/module_dm/test/test_dm_r_init.cpp deleted file mode 100644 index 83416b94d06..00000000000 --- a/source/source_estate/module_dm/test/test_dm_r_init.cpp +++ /dev/null @@ -1,313 +0,0 @@ -#include - -#include "gmock/gmock.h" -#include "gtest/gtest.h" -#define private public -#include "source_estate/module_dm/density_matrix.h" -#include "source_hamilt/module_hcontainer/hcontainer.h" -#include "source_cell/klist.h" -#undef private -/************************************************ - * unit test of DensityMatrix constructor - ***********************************************/ - -/** - * This unit test construct a DensityMatrix object - */ - -// test_size is the number of atoms in the unitcell -// modify test_size to test different size of unitcell -int test_size = 10; -int test_nw = 26; - -class DMTest : public testing::Test -{ - protected: - Parallel_Orbitals* paraV; - int dsize; - int my_rank = 0; - UnitCell ucell; - void SetUp() override - { -#ifdef __MPI - // MPI parallel settings - MPI_Comm_size(MPI_COMM_WORLD, &dsize); - MPI_Comm_rank(MPI_COMM_WORLD, &my_rank); -#endif - - // set up a unitcell, with one element and test_size atoms, each atom has test_nw orbitals - ucell.ntype = 1; - ucell.nat = test_size; - ucell.atoms = new Atom[ucell.ntype]; - ucell.iat2it = new int[ucell.nat]; - ucell.iat2ia = new int[ucell.nat]; - ucell.atoms[0].tau.resize(ucell.nat); - ucell.itia2iat.create(ucell.ntype, ucell.nat); - for (int iat = 0; iat < ucell.nat; iat++) - { - ucell.iat2it[iat] = 0; - ucell.iat2ia[iat] = iat; - ucell.atoms[0].tau[iat] = ModuleBase::Vector3(0.0, 0.0, 0.0); - ucell.itia2iat(0, iat) = iat; - } - ucell.atoms[0].na = test_size; - ucell.atoms[0].nw = test_nw; - ucell.atoms[0].iw2l.resize(test_nw); - ucell.atoms[0].iw2m.resize(test_nw); - ucell.atoms[0].iw2n.resize(test_nw); - for (int iw = 0; iw < test_nw; ++iw) - { - ucell.atoms[0].iw2l[iw] = 0; - ucell.atoms[0].iw2m[iw] = 0; - ucell.atoms[0].iw2n[iw] = 0; - } - ucell.set_iat2iwt(1); - init_parav(); - // set paraV - init_parav(); - } - - void TearDown() override - { - delete paraV; - delete[] ucell.atoms; - } - -#ifdef __MPI - void init_parav() - { - int nb = 2; - int global_row = test_size * test_nw; - int global_col = test_size * test_nw; - std::ofstream ofs_running; - paraV = new Parallel_Orbitals(); - paraV->init(global_row, global_col, nb, MPI_COMM_WORLD); - paraV->set_atomic_trace(ucell.get_iat2iwt(), test_size, global_row); - } -#else - void init_parav() - { - } -#endif -}; - -// test for construct DMR from GridD and UnitCell -TEST_F(DMTest, DMInit1) -{ - // initalize a kvectors - K_Vectors* kv = nullptr; - int nspin = 1; - int nks = 2; // since nspin = 1 - kv = new K_Vectors; - kv->set_nks(nks); - kv->kvec_d.resize(nks); - kv->kvec_d[1].x = 0.5; - // construct DM - std::cout << "dim0: " << paraV->dim0 << " dim1:" << paraV->dim1 << std::endl; - std::cout << "nrow: " << paraV->nrow << " ncol:" << paraV->ncol << std::endl; - elecstate::DensityMatrix DM(paraV, nspin, kv->kvec_d, nks); - // initialize this->_DMR - Grid_Driver gd(0,0); - DM.init_DMR(&gd, &ucell); - // compare - EXPECT_EQ(DM.get_DMR_pointer(1)->size_atom_pairs(), test_size * test_size); - EXPECT_EQ(DM.get_DMR_pointer(1)->get_atom_pair(2, 2).get_atom_i(), 2); - EXPECT_EQ(DM.get_DMR_pointer(1)->get_atom_pair(2, 2).get_atom_j(), 2); - EXPECT_EQ(DM.get_DMR_pointer(1)->get_atom_pair(2, 2).get_row_size(), paraV->get_nrow_atom(2)); - EXPECT_EQ(DM.get_DMR_pointer(1)->get_atom_pair(2, 2).get_col_size(), paraV->get_ncol_atom(2)); - delete kv; -} - -// test for construct DMR from RA and UnitCell -TEST_F(DMTest, DMInit2) -{ - // initalize a kvectors - K_Vectors* kv = nullptr; - int nspin = 1; - int nks = 2; // since nspin = 1 - kv = new K_Vectors; - kv->set_nks(nks); - kv->kvec_d.resize(nks); - kv->kvec_d[1].x = 0.5; - // construct DM - std::cout << "dim0: " << paraV->dim0 << " dim1:" << paraV->dim1 << std::endl; - std::cout << "nrow: " << paraV->nrow << " ncol:" << paraV->ncol << std::endl; - elecstate::DensityMatrix DM(paraV, nspin, kv->kvec_d, nks); - // initialize Record_adj using Grid_Driver - Grid_Driver gd(0,0); - Record_adj ra; - ra.na_each = new int[ucell.nat]; - ra.info = new int**[ucell.nat]; - for (int iat1 = 0; iat1 < ucell.nat; iat1++) - { - auto tau1 = ucell.get_tau(iat1); - int T1, I1; - ucell.iat2iait(iat1, &I1, &T1); - AdjacentAtomInfo adjs; - gd.Find_atom(ucell, tau1, T1, I1, &adjs); - ra.na_each[iat1] = adjs.adj_num + 1; - ra.info[iat1] = new int*[ra.na_each[iat1]]; - for (int ad = 0; ad < ra.na_each[iat1]; ++ad) - { - ra.info[iat1][ad] = new int[5]; - const int T2 = adjs.ntype[ad]; - const int I2 = adjs.natom[ad]; - ra.info[iat1][ad][3] = T2; - ra.info[iat1][ad][4] = I2; - ModuleBase::Vector3& R_index = adjs.box[ad]; - ra.info[iat1][ad][0] = R_index.x; - ra.info[iat1][ad][1] = R_index.y; - ra.info[iat1][ad][2] = R_index.z; - ra.info[iat1][ad][3] = T2; - ra.info[iat1][ad][4] = I2; - } - } - DM.init_DMR(ra, &ucell); - // compare - EXPECT_EQ(DM.get_DMR_pointer(1)->size_atom_pairs(), test_size * test_size); - EXPECT_EQ(DM.get_DMR_pointer(1)->get_atom_pair(2, 2).get_atom_i(), 2); - EXPECT_EQ(DM.get_DMR_pointer(1)->get_atom_pair(2, 2).get_atom_j(), 2); - EXPECT_EQ(DM.get_DMR_pointer(1)->get_atom_pair(2, 2).get_row_size(), paraV->get_nrow_atom(2)); - EXPECT_EQ(DM.get_DMR_pointer(1)->get_atom_pair(2, 2).get_col_size(), paraV->get_ncol_atom(2)); - // release memory - delete kv; - for (int iat1 = 0; iat1 < ucell.nat; iat1++) - { - for (int ad = 0; ad < ra.na_each[iat1]; ++ad) - { - delete[] ra.info[iat1][ad]; - } - delete[] ra.info[iat1]; - } - delete[] ra.info; -} - -// test for construct DMR from another HContainer -TEST_F(DMTest, DMInit3) -{ - // initalize a kvectors - K_Vectors* kv = nullptr; - int nspin = 2; - int nks = 4; // since nspin = 2 - kv = new K_Vectors; - kv->set_nks(nks); - kv->kvec_d.resize(nks); - kv->kvec_d[1].x = 0.5; - kv->kvec_d[3].x = 0.5; - // construct a DM - elecstate::DensityMatrix, double> DM(paraV, nspin, kv->kvec_d, kv->get_nks() / nspin); - Grid_Driver gd(0, 0); - DM.init_DMR(&gd, &ucell); - std::cout << "dim0: " << paraV->dim0 << " dim1:" << paraV->dim1 << std::endl; - // construct another DM - elecstate::DensityMatrix, double> DM1(paraV, nspin, kv->kvec_d, kv->get_nks() / nspin); - DM1.init_DMR(*DM.get_DMR_pointer(1)); - // compare - EXPECT_EQ(DM1.get_DMR_pointer(2)->size_atom_pairs(), test_size * test_size); - EXPECT_EQ(DM1.get_DMR_pointer(2)->get_atom_pair(2, 2).get_atom_i(), 2); - EXPECT_EQ(DM1.get_DMR_pointer(1)->get_atom_pair(2, 2).get_atom_j(), 2); - EXPECT_EQ(DM1.get_DMR_pointer(1)->get_atom_pair(2, 2).get_row_size(), paraV->get_nrow_atom(2)); - EXPECT_EQ(DM1.get_DMR_pointer(2)->get_atom_pair(2, 2).get_col_size(), paraV->get_ncol_atom(2)); - // - delete kv; -} - -// test for construct DMR from another HContainer > -TEST_F(DMTest, DMInit4) -{ - // initalize a kvectors - K_Vectors* kv = nullptr; - int nspin = 2; - int nks = 4; // since nspin = 2 - kv = new K_Vectors; - kv->set_nks(nks); - kv->kvec_d.resize(nks); - kv->kvec_d[1].x = 0.5; - kv->kvec_d[3].x = 0.5; - // construct a new HContainer - Grid_Driver gd(0, 0); - hamilt::HContainer>* tmp_DMR; - tmp_DMR = new hamilt::HContainer>(paraV); - // set up a HContainer - for (int iat1 = 0; iat1 < ucell.nat; iat1++) - { - auto tau1 = ucell.get_tau(iat1); - int T1, I1; - ucell.iat2iait(iat1, &I1, &T1); - AdjacentAtomInfo adjs; - gd.Find_atom(ucell, tau1, T1, I1, &adjs); - // std::cout << "adjs.adj_num: " <is_invalid_atom_pair(iat1, iat2)) - { - continue; - } - ModuleBase::Vector3& R_index = adjs.box[ad]; - // std::cout << "R_index: " << R_index.x << " " << R_index.y << " " << R_index.z << std::endl; - hamilt::AtomPair> tmp_ap(iat1, iat2, R_index.x, R_index.y, R_index.z, paraV); - tmp_DMR->insert_pair(tmp_ap); - } - } - // construct a DM from this HContainer - elecstate::DensityMatrix, double> DM(paraV, nspin, kv->kvec_d, kv->get_nks() / nspin); - DM.init_DMR(*tmp_DMR); - std::cout << "dim0: " << paraV->dim0 << " dim1:" << paraV->dim1 << std::endl; - // compare - EXPECT_EQ(DM.get_DMR_pointer(2)->size_atom_pairs(), test_size * test_size); - EXPECT_EQ(DM.get_DMR_pointer(2)->get_atom_pair(2, 2).get_atom_i(), 2); - EXPECT_EQ(DM.get_DMR_pointer(1)->get_atom_pair(2, 2).get_atom_j(), 2); - EXPECT_EQ(DM.get_DMR_pointer(1)->get_atom_pair(2, 2).get_row_size(), paraV->get_nrow_atom(2)); - EXPECT_EQ(DM.get_DMR_pointer(2)->get_atom_pair(2, 2).get_col_size(), paraV->get_ncol_atom(2)); - // - delete kv; -} - -// test for save_DMR -TEST_F(DMTest, saveDMR) -{ - // initalize a kvectors - K_Vectors* kv = nullptr; - int nspin = 2; - int nks = 4; // since nspin = 2 - kv = new K_Vectors; - kv->set_nks(nks); - kv->kvec_d.resize(nks); - kv->kvec_d[1].x = 0.5; - kv->kvec_d[3].x = 0.5; - // construct a DM - elecstate::DensityMatrix, double> DM(paraV, nspin, kv->kvec_d, kv->get_nks() / nspin); - Grid_Driver gd(0, 0); - DM.init_DMR(&gd, &ucell); - // construct another DM - elecstate::DensityMatrix, double> DM_test(paraV, nspin, kv->kvec_d, kv->get_nks() / nspin); - DM_test.init_DMR(*DM.get_DMR_pointer(1)); - DM_test.save_DMR(); - EXPECT_EQ(DM_test.get_DMR_pointer(1)->get_nnr(), DM.get_DMR_pointer(1)->get_nnr()); - EXPECT_EQ(DM_test.get_DMR_pointer(1)->get_nnr(), DM_test._DMR_save[0].size()); - // add a new AtomPair, act as a relaxation - hamilt::AtomPair tmp_ap(9, 9, 1, 0, 0, paraV); - DM_test.get_DMR_pointer(1)->insert_pair(tmp_ap); - DM_test.get_DMR_pointer(1)->allocate(); - // update DMR_save - DM_test.save_DMR(); - EXPECT_EQ(DM_test.get_DMR_pointer(1)->get_nnr(), DM_test._DMR_save[0].size()); - // delete - delete kv; -} - -int main(int argc, char** argv) -{ -#ifdef __MPI - MPI_Init(&argc, &argv); -#endif - testing::InitGoogleTest(&argc, argv); - int result = RUN_ALL_TESTS(); -#ifdef __MPI - MPI_Finalize(); -#endif - return result; -} diff --git a/source/source_estate/module_dm/unittests/CMakeLists.txt b/source/source_estate/module_dm/unittests/CMakeLists.txt new file mode 100644 index 00000000000..3535cc7fe7a --- /dev/null +++ b/source/source_estate/module_dm/unittests/CMakeLists.txt @@ -0,0 +1,54 @@ +abacus_disable_feature_definitions(__MLALGO) +abacus_disable_feature_definitions(__CUDA) +abacus_disable_feature_definitions(__ROCM) + +install(DIRECTORY support DESTINATION ${CMAKE_CURRENT_BINARY_DIR}) + +AddTest( + TARGET MODULE_ESTATE_dm_constructor_test + LIBS parameter base device symmetry + SOURCES test_dm_constructor.cpp ../density_matrix.cpp ../dmr_gamma.cpp ../dmr_init.cpp ../dm_setter.cpp ../dm_getter.cpp tmp_mocks.cpp + ${ABACUS_SOURCE_DIR}/source_hamilt/module_hcontainer/base_matrix.cpp + ${ABACUS_SOURCE_DIR}/source_hamilt/module_hcontainer/hcontainer.cpp + ${ABACUS_SOURCE_DIR}/source_hamilt/module_hcontainer/atom_pair.cpp + ${ABACUS_SOURCE_DIR}/source_basis/module_ao/parallel_orbitals.cpp + ${ABACUS_SOURCE_DIR}/source_cell/klist.cpp ${ABACUS_SOURCE_DIR}/source_cell/klist_io.cpp + ${ABACUS_SOURCE_DIR}/source_cell/parallel_kpoints.cpp + ${ABACUS_SOURCE_DIR}/source_cell/reciprocal_grid.cpp +) + +AddTest( + TARGET MODULE_ESTATE_dm_init_test + LIBS parameter base device symmetry + SOURCES test_dm_r_init.cpp ../density_matrix.cpp ../dmr_gamma.cpp ../dmr_init.cpp ../dm_setter.cpp ../dm_getter.cpp tmp_mocks.cpp + ${ABACUS_SOURCE_DIR}/source_hamilt/module_hcontainer/base_matrix.cpp + ${ABACUS_SOURCE_DIR}/source_hamilt/module_hcontainer/hcontainer.cpp + ${ABACUS_SOURCE_DIR}/source_hamilt/module_hcontainer/atom_pair.cpp + ${ABACUS_SOURCE_DIR}/source_basis/module_ao/parallel_orbitals.cpp + ${ABACUS_SOURCE_DIR}/source_cell/klist.cpp ${ABACUS_SOURCE_DIR}/source_cell/klist_io.cpp + ${ABACUS_SOURCE_DIR}/source_cell/parallel_kpoints.cpp + ${ABACUS_SOURCE_DIR}/source_cell/reciprocal_grid.cpp +) + +AddTest( + TARGET MODULE_ESTATE_dm_cal_DMR_test + LIBS parameter base device symmetry + SOURCES test_cal_dm_r.cpp ../density_matrix.cpp ../dmr_gamma.cpp ../dmr_init.cpp ../dm_setter.cpp ../dm_getter.cpp ../dm_tools.cpp ../dmr_k.cpp ../dmr_td.cpp ../dmr_full.cpp tmp_mocks.cpp + ${ABACUS_SOURCE_DIR}/source_hamilt/module_hcontainer/base_matrix.cpp + ${ABACUS_SOURCE_DIR}/source_hamilt/module_hcontainer/hcontainer.cpp + ${ABACUS_SOURCE_DIR}/source_hamilt/module_hcontainer/atom_pair.cpp + ${ABACUS_SOURCE_DIR}/source_basis/module_ao/parallel_orbitals.cpp + ${ABACUS_SOURCE_DIR}/source_cell/klist.cpp ${ABACUS_SOURCE_DIR}/source_cell/klist_io.cpp + ${ABACUS_SOURCE_DIR}/source_cell/parallel_kpoints.cpp + ${ABACUS_SOURCE_DIR}/source_cell/reciprocal_grid.cpp +) + +AddTest( + TARGET MODULE_ESTATE_dm_soc_magnetization_roundtrip_test + LIBS parameter base device + SOURCES test_soc_magnetization_roundtrip.cpp ../density_matrix.cpp ../dmr_gamma.cpp ../dmr_init.cpp ../dm_setter.cpp ../dm_getter.cpp ../dm_tools.cpp ../dmr_k.cpp ../dmr_td.cpp ../dmr_full.cpp tmp_mocks.cpp + ${ABACUS_SOURCE_DIR}/source_hamilt/module_hcontainer/base_matrix.cpp + ${ABACUS_SOURCE_DIR}/source_hamilt/module_hcontainer/hcontainer.cpp + ${ABACUS_SOURCE_DIR}/source_hamilt/module_hcontainer/atom_pair.cpp + ${ABACUS_SOURCE_DIR}/source_basis/module_ao/parallel_orbitals.cpp +) diff --git a/source/source_estate/module_dm/test/prepare_unitcell.h b/source/source_estate/module_dm/unittests/prepare_unitcell.h similarity index 100% rename from source/source_estate/module_dm/test/prepare_unitcell.h rename to source/source_estate/module_dm/unittests/prepare_unitcell.h diff --git a/source/source_estate/module_dm/test/support/SPIN1_0.dmk b/source/source_estate/module_dm/unittests/support/SPIN1_0.dmk similarity index 100% rename from source/source_estate/module_dm/test/support/SPIN1_0.dmk rename to source/source_estate/module_dm/unittests/support/SPIN1_0.dmk diff --git a/source/source_estate/module_dm/test/support/SPIN1_1.dmk b/source/source_estate/module_dm/unittests/support/SPIN1_1.dmk similarity index 100% rename from source/source_estate/module_dm/test/support/SPIN1_1.dmk rename to source/source_estate/module_dm/unittests/support/SPIN1_1.dmk diff --git a/source/source_estate/module_dm/unittests/test_cal_dm_r.cpp b/source/source_estate/module_dm/unittests/test_cal_dm_r.cpp new file mode 100644 index 00000000000..dbad57eb1de --- /dev/null +++ b/source/source_estate/module_dm/unittests/test_cal_dm_r.cpp @@ -0,0 +1,459 @@ +#include + +#include "gmock/gmock.h" +#include "gtest/gtest.h" +#include "source_estate/module_dm/density_matrix.h" +#include "source_hamilt/module_hcontainer/hcontainer.h" +#include "source_cell/klist.h" + +/************************************************ + * unit test of DensityMatrix constructor + ***********************************************/ + +/** + * This unit test construct a DensityMatrix object + */ + +// test_size is the number of atoms in the unitcell +// modify test_size to test different size of unitcell +int test_size = 10; +int test_nw = 10; + +class DMTest : public testing::Test +{ + protected: + Parallel_Orbitals* paraV; + int dsize; + int my_rank = 0; + UnitCell ucell; + void SetUp() override + { +#ifdef __MPI + // MPI parallel settings + MPI_Comm_size(MPI_COMM_WORLD, &dsize); + MPI_Comm_rank(MPI_COMM_WORLD, &my_rank); +#endif + + // set up a unitcell, with one element and test_size atoms, each atom has test_nw orbitals + ucell.ntype = 1; + ucell.nat = test_size; + ucell.atoms = new Atom[ucell.ntype]; + ucell.iat2it = new int[ucell.nat]; + ucell.iat2ia = new int[ucell.nat]; + ucell.atoms[0].tau.resize(ucell.nat); + ucell.itia2iat.create(ucell.ntype, ucell.nat); + for (int iat = 0; iat < ucell.nat; iat++) + { + ucell.iat2it[iat] = 0; + ucell.iat2ia[iat] = iat; + ucell.atoms[0].tau[iat] = ModuleBase::Vector3(0.0, 0.0, 0.0); + ucell.itia2iat(0, iat) = iat; + } + ucell.atoms[0].na = test_size; + ucell.atoms[0].nw = test_nw; + ucell.atoms[0].iw2l.resize(test_nw); + ucell.atoms[0].iw2m.resize(test_nw); + ucell.atoms[0].iw2n.resize(test_nw); + for (int iw = 0; iw < test_nw; ++iw) + { + ucell.atoms[0].iw2l[iw] = 0; + ucell.atoms[0].iw2m[iw] = 0; + ucell.atoms[0].iw2n[iw] = 0; + } + ucell.set_iat2iwt(1); + init_parav(); + + // set paraV + init_parav(); + } + + void TearDown() override + { + delete paraV; + delete[] ucell.atoms; + } + +#ifdef __MPI + void init_parav() + { + int nb = 2; + int global_row = test_size * test_nw; + int global_col = test_size * test_nw; + std::ofstream ofs_running; + paraV = new Parallel_Orbitals(); + paraV->init(global_row, global_col, nb, MPI_COMM_WORLD); + paraV->set_atomic_trace(ucell.get_iat2iwt(), test_size, global_row); + } +#else + void init_parav() + { + } +#endif +}; + +TEST_F(DMTest, cal_dmr_full) +{ + // get my rank of this process + int my_rank = 0; +#ifdef __MPI + MPI_Comm_rank(MPI_COMM_WORLD, &my_rank); +#endif + // output dim and nrow, ncol + if (my_rank == 0) + { + std::cout << "my rank: " << my_rank << " dim0: " << paraV->dim0 << " dim1:" << paraV->dim1 << std::endl; + std::cout << "my rank: " << my_rank << " nrow: " << paraV->nrow << " ncol:" << paraV->ncol << std::endl; + } + else + { + std::cout << "my rank: " << my_rank << " nrow: " << paraV->nrow << " ncol:" << paraV->ncol << std::endl; + } + // initalize a kvectors, Gamma-only + K_Vectors* kv = nullptr; + int nspin = 4; + int nks = 2; // since nspin = 2 + kv = new K_Vectors; + kv->set_nks(nks); + kv->kvec_d.resize(nks); + // construct DM + module_dm::DensityMatrix, double> DM(paraV, nspin, kv->kvec_d, kv->get_nks()); + // set this->dmk + for (int is = 1; is <= nspin; is++) + { + for (int ik = 0; ik < kv->get_nks(); ik++) + { + for (int i = 0; i < paraV->nrow; i++) + { + for (int j = 0; j < paraV->ncol; j++) + { + DM.set_dmk(is, ik, i, j, std::complex(0.77, 0.77)); + } + } + } + } + // initialize dmR_full + hamilt::HContainer> dmR_full(ucell, paraV); + // calculate this->dmr + std::chrono::high_resolution_clock::time_point start_time = std::chrono::high_resolution_clock::now(); + DM.cal_dmr_full(&dmR_full, -1); + std::chrono::high_resolution_clock::time_point end_time = std::chrono::high_resolution_clock::now(); + std::chrono::duration elapsed_time + = std::chrono::duration_cast>(end_time - start_time); + std::cout << "my rank: " << my_rank << " elapsed time blas: " << elapsed_time.count() << std::endl; + // compare the result + for (int i = 0; i < dmR_full.size_atom_pairs(); i++) + { + const std::complex* ptr1 = dmR_full.get_atom_pair(i).get_HR_values(0, 0, 0).get_pointer(); + // + for (int j = 0; j < dmR_full.get_atom_pair(i).get_size(); j++) + { + //std::cout << "my rank: " << my_rank << " i: " << i << " j: " << j << " value: " << ptr1[j] << std::endl; + EXPECT_NEAR(ptr1[j].real(), 1.54, 1e-10); + EXPECT_NEAR(ptr1[j].imag(), 1.54, 1e-10); + } + } + delete kv; +} + +TEST_F(DMTest, cal_DMR_blas_double) +{ + // get my rank of this process + int my_rank = 0; +#ifdef __MPI + MPI_Comm_rank(MPI_COMM_WORLD, &my_rank); +#endif + // output dim and nrow, ncol + if (my_rank == 0) + { + std::cout << "my rank: " << my_rank << " dim0: " << paraV->dim0 << " dim1:" << paraV->dim1 << std::endl; + std::cout << "my rank: " << my_rank << " nrow: " << paraV->nrow << " ncol:" << paraV->ncol << std::endl; + } + else + { + std::cout << "my rank: " << my_rank << " nrow: " << paraV->nrow << " ncol:" << paraV->ncol << std::endl; + } + // initalize a kvectors, Gamma-only + K_Vectors* kv = nullptr; + int nspin = 2; + int nks = 2; // since nspin = 2 + kv = new K_Vectors; + kv->set_nks(nks); + kv->kvec_d.resize(nks); + // construct DM + module_dm::DensityMatrix DM(paraV, nspin, kv->kvec_d, kv->get_nks() / nspin); + // set this->dmk + for (int is = 1; is <= nspin; is++) + { + for (int ik = 0; ik < kv->get_nks() / nspin; ik++) + { + for (int i = 0; i < paraV->nrow; i++) + { + for (int j = 0; j < paraV->ncol; j++) + { + DM.set_dmk(is, ik, i, j, 0.77); + } + } + } + } + // initialize this->dmr + Grid_Driver gd(0, 0); + DM.init_dmr(&gd, &ucell); + // set Gamma-only + for (int is = 1; is <= nspin; is++) + { + DM.get_dmr_ptr(is)->fix_gamma(); + } + // calculate this->dmr + std::chrono::high_resolution_clock::time_point start_time = std::chrono::high_resolution_clock::now(); + DM.cal_dmr(-1); + std::chrono::high_resolution_clock::time_point end_time = std::chrono::high_resolution_clock::now(); + std::chrono::duration elapsed_time + = std::chrono::duration_cast>(end_time - start_time); + std::cout << "my rank: " << my_rank << " elapsed time blas: " << elapsed_time.count() << std::endl; + // compare the result + for (int i = 0; i < DM.get_dmr_ptr(1)->size_atom_pairs(); i++) + { + double* ptr1 = DM.get_dmr_ptr(1)->get_atom_pair(i).get_HR_values(0, 0, 0).get_pointer(); + // + for (int j = 0; j < DM.get_dmr_ptr(1)->get_atom_pair(i).get_size(); j++) + { + // std::cout << "my rank: " << my_rank << " i: " << i << " j: " << j << " value: " << ptr1[j] << std::endl; + EXPECT_NEAR(ptr1[j], 0.77, 1e-10); + } + } + delete kv; +} + +TEST_F(DMTest, cal_DMR_blas_complex) +{ + // get my rank of this process + int my_rank = 0; +#ifdef __MPI + MPI_Comm_rank(MPI_COMM_WORLD, &my_rank); +#endif + // output dim and nrow, ncol + if (my_rank == 0) + { + std::cout << "my rank: " << my_rank << " dim0: " << paraV->dim0 << " dim1:" << paraV->dim1 << std::endl; + std::cout << "my rank: " << my_rank << " nrow: " << paraV->nrow << " ncol:" << paraV->ncol << std::endl; + } + else + { + std::cout << "my rank: " << my_rank << " nrow: " << paraV->nrow << " ncol:" << paraV->ncol << std::endl; + } + // initalize a kvectors + K_Vectors* kv = nullptr; + int nspin = 2; + int nks = 4; // since nspin = 2 + kv = new K_Vectors; + kv->set_nks(nks); + kv->kvec_d.resize(nks); + kv->kvec_d[1].x = 0.5; + kv->kvec_d[3].x = 0.5; + // construct DM + module_dm::DensityMatrix, double> DM(paraV, nspin, kv->kvec_d, kv->get_nks() / nspin); + // set this->dmk + for (int is = 1; is <= nspin; is++) + { + for (int ik = 0; ik < kv->get_nks() / nspin; ik++) + { + for (int i = 0; i < paraV->nrow; i++) + { + for (int j = 0; j < paraV->ncol; j++) + { + DM.set_dmk(is, ik, i, j, is * 0.77 * (ik + 1)); + } + } + } + } + // initialize this->dmr + Grid_Driver gd(0, 0); + DM.init_dmr(&gd, &ucell); + // calculate this->dmr + std::chrono::high_resolution_clock::time_point start_time = std::chrono::high_resolution_clock::now(); + DM.cal_dmr(-1); + std::chrono::high_resolution_clock::time_point end_time = std::chrono::high_resolution_clock::now(); + std::chrono::duration elapsed_time + = std::chrono::duration_cast>(end_time - start_time); + std::cout << "my rank: " << my_rank << " elapsed time blas: " << elapsed_time.count() << std::endl; + // compare the result for spin-up + for (int i = 0; i < DM.get_dmr_ptr(1)->size_atom_pairs(); i++) + { + double* ptr1 = DM.get_dmr_ptr(1)->get_atom_pair(i).get_HR_values(1, 1, 1).get_pointer(); + // + for (int j = 0; j < DM.get_dmr_ptr(1)->get_atom_pair(i).get_size(); j++) + { + // std::cout << "my rank: " << my_rank << " i: " << i << " j: " << j << " value: " << ptr1[j] << std::endl; + EXPECT_NEAR(ptr1[j], -0.77, 1e-10); + } + } + // compare the result for spin-down + for (int i = 0; i < DM.get_dmr_ptr(2)->size_atom_pairs(); i++) + { + double* ptr1 = DM.get_dmr_ptr(2)->get_atom_pair(i).get_HR_values(1, 1, 1).get_pointer(); + // + for (int j = 0; j < DM.get_dmr_ptr(2)->get_atom_pair(i).get_size(); j++) + { + // std::cout << "my rank: " << my_rank << " i: " << i << " j: " << j << " value: " << ptr1[j] << std::endl; + EXPECT_NEAR(ptr1[j], -0.77 * 2, 1e-10); + } + } + // calculate DMR_total + DM.switch_dmr(1); + // compare the result for spin-up after sum + for (int i = 0; i < DM.get_dmr_ptr(1)->size_atom_pairs(); i++) + { + double* ptr1 = DM.get_dmr_ptr(1)->get_atom_pair(i).get_HR_values(1, 1, 1).get_pointer(); + // + for (int j = 0; j < DM.get_dmr_ptr(1)->get_atom_pair(i).get_size(); j++) + { + //std::cout << "my rank: " << my_rank << " i: " << i << " j: " << j << " value: " << ptr1[j] << std::endl; + EXPECT_NEAR(ptr1[j], -0.77 * 3, 1e-10); + } + } + // restore to normal DMR + DM.switch_dmr(0); + for (int i = 0; i < DM.get_dmr_ptr(1)->size_atom_pairs(); i++) + { + double* ptr1 = DM.get_dmr_ptr(1)->get_atom_pair(i).get_HR_values(1, 1, 1).get_pointer(); + // + for (int j = 0; j < DM.get_dmr_ptr(1)->get_atom_pair(i).get_size(); j++) + { + //std::cout << "my rank: " << my_rank << " i: " << i << " j: " << j << " value: " << ptr1[j] << std::endl; + EXPECT_NEAR(ptr1[j], -0.77, 1e-10); + } + } + // calculate DMR_differenct + DM.switch_dmr(2); + for (int i = 0; i < DM.get_dmr_ptr(1)->size_atom_pairs(); i++) + { + double* ptr1 = DM.get_dmr_ptr(1)->get_atom_pair(i).get_HR_values(1, 1, 1).get_pointer(); + // + for (int j = 0; j < DM.get_dmr_ptr(1)->get_atom_pair(i).get_size(); j++) + { + //std::cout << "my rank: " << my_rank << " i: " << i << " j: " << j << " value: " << ptr1[j] << std::endl; + EXPECT_NEAR(ptr1[j], 0.77, 1e-10); + } + } + delete kv; +} + +// Regression test for the SOC/noncollinear (global nspin==4) cal_dmr path. +// +// Background: in a real SOC run allocate_dm.cpp constructs the DensityMatrix with +// spin_mult = 1 (the 2x2 spin block is stored as ONE doubled matrix), +// while the GLOBAL physical nspin is 4. cal_dmr must still take the +// spin-resolved (Pauli) branch, which folds each 2x2 complex spin block into +// (rho_0, rho_x, rho_y, rho_z) via xyz_to_updown(). That branch used to be +// selected by the global PARAM.inp.nspin==4; a refactor (commit dcad8913d) +// switched the condition to dm.spin_mult==4, which is never true in SOC +// (spin_mult==1), silently dropping the rho_x/y/z spin channels and producing a +// wrong charge density (tests/03_NAO_multik/*spin4* failed by ~41 eV). +// +// This test reproduces the real SOC construction (spin_mult=1, nspin=4) +// and fills the spin-diagonal DMK entries (uu, dd) with (a + i b), leaving the +// spin off-diagonal entries (ud, du) zero. It then checks that cal_dmr selects +// the Pauli branch: +// * correct (Pauli) branch : rho_0 = (uu+dd).real() = 2a, rho_z = (uu-dd).real() = 0 +// * wrong (real-project) : every element = a (imaginary part b dropped) +// With the pre-fix condition (spin_mult==4 never taken) this test FAILS because +// rho_0 would come out as a instead of 2a. +TEST_F(DMTest, cal_DMR_soc_pauli_branch) +{ + // SOC doubles the orbital dimension (npol = 2): each atom carries nw*npol rows/cols. + // The fixture's ucell has nw = test_nw, so the doubled global dimension is used here. + const int npol = 2; + const int global_dim_soc = test_size * test_nw * npol; +#ifdef __MPI + Parallel_Orbitals* pv_soc = new Parallel_Orbitals(); + pv_soc->init(global_dim_soc, global_dim_soc, 2, MPI_COMM_WORLD); + // build the iat2iwt map for the doubled (spinor) orbital count + std::vector iat2iwt_soc(test_size); + for (int iat = 0; iat < test_size; ++iat) + { + iat2iwt_soc[iat] = iat * test_nw * npol; + } + pv_soc->set_atomic_trace(iat2iwt_soc.data(), test_size, global_dim_soc); +#else + Parallel_Orbitals* pv_soc = paraV; // fallback; MPI path is the supported configuration +#endif + + // a single Gamma k-point; construct exactly like the real SOC setup_dm does: + // nspin_dm = 1, but the global physical nspin = 4. + std::vector> kvec_d(1, ModuleBase::Vector3(0.0, 0.0, 0.0)); + const int nspin_dm = 1; + const int nspin_global = 4; + module_dm::DensityMatrix, double> DM(pv_soc, nspin_dm, kvec_d, 1, nspin_global); + + // fill the single DMK: spin-diagonal entries (uu, dd) are (a + i b), + // spin off-diagonal entries (ud, du) stay zero. With a constant fill the + // 2x2 spin block would also have ud = du = (a + i b), and rho_x = + // Re(ud + du) would correctly be 2a instead of the asserted 0. + const double a = 0.5; + const double b = 0.25; + for (int i = 0; i < pv_soc->nrow; i++) + { + for (int j = 0; j < pv_soc->ncol; j++) + { + // global spinor indices determine the spin parity; local indices + // need not preserve parity under a 2D block-cyclic distribution + const bool same_spin = (pv_soc->local2global_row(i) % npol) + == (pv_soc->local2global_col(j) % npol); + const std::complex dmk_value = same_spin + ? std::complex(a, b) + : std::complex(0.0, 0.0); + DM.set_dmk(1, 0, i, j, dmk_value); + } + } + + // build the real-space DMR + Grid_Driver gd(0, 0); + DM.init_dmr(&gd, &ucell); + // Gamma-only: reduce R vectors to (0, 0, 0), as cal_DMR_blas_double does + DM.get_dmr_ptr(1)->fix_gamma(); + DM.cal_dmr(-1); + + // check the Gamma (R = 0) block: rho_0 must be 2a (Pauli), NOT a (real projection); + // rho_x = rho_y = rho_z = 0 for uu == dd and zero spin off-diagonals. + hamilt::HContainer* dmr = DM.get_dmr_ptr(1); + for (int i = 0; i < dmr->size_atom_pairs(); i++) + { + hamilt::AtomPair& ap = dmr->get_atom_pair(i); + double* rho = ap.get_HR_values(0, 0, 0).get_pointer(); + const int col_size = ap.get_col_size(); + const int row_size = ap.get_row_size(); + // walk the 2x2 spin blocks (step_trace = {0, 1, col_size, col_size+1}) + for (int irow = 0; irow < row_size; irow += 2) + { + for (int icol = 0; icol < col_size; icol += 2) + { + const double* blk = rho + irow * col_size + icol; + const double rho_0 = blk[0]; // step_trace[0] + const double rho_x = blk[1]; // step_trace[1] + const double rho_y = blk[col_size]; // step_trace[2] + const double rho_z = blk[col_size + 1]; // step_trace[3] + EXPECT_NEAR(rho_0, 2.0 * a, 1e-10) + << "rho_0 wrong: cal_dmr did NOT take the SOC Pauli branch (nspin_global==4)"; + EXPECT_NEAR(rho_x, 0.0, 1e-10); + EXPECT_NEAR(rho_y, 0.0, 1e-10); + EXPECT_NEAR(rho_z, 0.0, 1e-10); + } + } + } +#ifdef __MPI + delete pv_soc; +#endif +} + +int main(int argc, char** argv) +{ +#ifdef __MPI + MPI_Init(&argc, &argv); +#endif + testing::InitGoogleTest(&argc, argv); + int result = RUN_ALL_TESTS(); +#ifdef __MPI + MPI_Finalize(); +#endif + return result; +} diff --git a/source/source_estate/module_dm/test/test_dm_constructor.cpp b/source/source_estate/module_dm/unittests/test_dm_constructor.cpp similarity index 84% rename from source/source_estate/module_dm/test/test_dm_constructor.cpp rename to source/source_estate/module_dm/unittests/test_dm_constructor.cpp index 180b5cf91a2..f270439fe6f 100644 --- a/source/source_estate/module_dm/test/test_dm_constructor.cpp +++ b/source/source_estate/module_dm/unittests/test_dm_constructor.cpp @@ -95,11 +95,11 @@ TEST_F(DMTest, DMConstructor_GammaOnly) std::cout << "dim0: " << paraV->dim0 << " dim1:" << paraV->dim1 << std::endl; std::cout << "nrow: " << paraV->nrow << " ncol:" << paraV->ncol << std::endl; int nspin = 2; - elecstate::DensityMatrix DM(paraV, nspin); + module_dm::DensityMatrix DM(paraV, nspin); // compare - EXPECT_EQ(DM.get_DMK_size(), nspin); - EXPECT_EQ(DM.get_DMK_nrow(), paraV->nrow); - EXPECT_EQ(DM.get_DMK_ncol(), paraV->ncol); + EXPECT_EQ(DM.get_dmk_size(), nspin); + EXPECT_EQ(DM.get_dmk_nrow(), paraV->nrow); + EXPECT_EQ(DM.get_dmk_ncol(), paraV->ncol); } TEST_F(DMTest, DMConstructor_nspin1) @@ -115,11 +115,11 @@ TEST_F(DMTest, DMConstructor_nspin1) std::cout << "dim0: " << paraV->dim0 << " dim1:" << paraV->dim1 << std::endl; std::cout << "nrow: " << paraV->nrow << " ncol:" << paraV->ncol << std::endl; int nspin = 1; - elecstate::DensityMatrix DM(paraV, nspin, kv->kvec_d, nks); + module_dm::DensityMatrix DM(paraV, nspin, kv->kvec_d, nks); // compare - EXPECT_EQ(DM.get_DMK_nks(), kv->get_nks()); - EXPECT_EQ(DM.get_DMK_nrow(), paraV->nrow); - EXPECT_EQ(DM.get_DMK_ncol(), paraV->ncol); + EXPECT_EQ(DM.get_dmk_nks(), kv->get_nks()); + EXPECT_EQ(DM.get_dmk_nrow(), paraV->nrow); + EXPECT_EQ(DM.get_dmk_ncol(), paraV->ncol); // set elements of DMK for (int is = 1; is <= nspin; is++) @@ -130,7 +130,7 @@ TEST_F(DMTest, DMConstructor_nspin1) { for (int j = 0; j < paraV->ncol; j++) { - DM.set_DMK(is, ik, i, j, is + ik * i + j); + DM.set_dmk(is, ik, i, j, is + ik * i + j); } } } @@ -144,18 +144,18 @@ TEST_F(DMTest, DMConstructor_nspin1) { for (int j = 0; j < paraV->ncol; j++) { - EXPECT_EQ(DM.get_DMK(is, ik, i, j), is + ik * i + j); + EXPECT_EQ(DM.get_dmk(is, ik, i, j), is + ik * i + j); } } } } - // test for get_DMK_pointer + // test for get_dmk_ptr for (int is = 1; is <= nspin; is++) { int ik_begin = (is - 1) * kv->get_nks() / nspin; for (int ik = 0; ik < kv->get_nks() / nspin; ik++) { - double* ptr = DM.get_DMK_pointer(ik + ik_begin); + double* ptr = DM.get_dmk_ptr(ik + ik_begin); for (int i = 0; i < paraV->nrow; i++) { for (int j = 0; j < paraV->ncol; j++) @@ -184,11 +184,11 @@ TEST_F(DMTest, DMConstructor_nspin2) // construct DM std::cout << "dim0: " << paraV->dim0 << " dim1:" << paraV->dim1 << std::endl; std::cout << "nrow: " << paraV->nrow << " ncol:" << paraV->ncol << std::endl; - elecstate::DensityMatrix DM(paraV, nspin, kv->kvec_d, kv->get_nks() / nspin); + module_dm::DensityMatrix DM(paraV, nspin, kv->kvec_d, kv->get_nks() / nspin); // compare - EXPECT_EQ(DM.get_DMK_nks(), kv->get_nks()); - EXPECT_EQ(DM.get_DMK_nrow(), paraV->nrow); - EXPECT_EQ(DM.get_DMK_ncol(), paraV->ncol); + EXPECT_EQ(DM.get_dmk_nks(), kv->get_nks()); + EXPECT_EQ(DM.get_dmk_nrow(), paraV->nrow); + EXPECT_EQ(DM.get_dmk_ncol(), paraV->ncol); // set elements of DMK for (int is = 1; is <= nspin; is++) @@ -199,7 +199,7 @@ TEST_F(DMTest, DMConstructor_nspin2) { for (int j = 0; j < paraV->ncol; j++) { - DM.set_DMK(is, ik, i, j, ik * i + j); + DM.set_dmk(is, ik, i, j, ik * i + j); } } } @@ -211,18 +211,18 @@ TEST_F(DMTest, DMConstructor_nspin2) { for (int j = 0; j < paraV->ncol; j++) { - EXPECT_EQ(DM.get_DMK(1, ik, i, j), ik * i + j); - EXPECT_EQ(DM.get_DMK(1, ik, i, j), DM.get_DMK(2, ik, i, j)); + EXPECT_EQ(DM.get_dmk(1, ik, i, j), ik * i + j); + EXPECT_EQ(DM.get_dmk(1, ik, i, j), DM.get_dmk(2, ik, i, j)); } } } - // test for get_DMK_pointer + // test for get_dmk_ptr for (int is = 1; is <= nspin; is++) { int ik_begin = (is - 1) * kv->get_nks() / nspin; for (int ik = 0; ik < kv->get_nks() / nspin; ik++) { - double* ptr = DM.get_DMK_pointer(ik + ik_begin); + double* ptr = DM.get_dmk_ptr(ik + ik_begin); for (int i = 0; i < paraV->nrow; i++) { for (int j = 0; j < paraV->ncol; j++) diff --git a/source/source_estate/module_dm/unittests/test_dm_from_psi.cpp b/source/source_estate/module_dm/unittests/test_dm_from_psi.cpp new file mode 100644 index 00000000000..3988193ff72 --- /dev/null +++ b/source/source_estate/module_dm/unittests/test_dm_from_psi.cpp @@ -0,0 +1,176 @@ +#include + +#include "gmock/gmock.h" +#include "gtest/gtest.h" +#include "source_estate/module_dm/density_matrix.h" +#include "source_hamilt/module_hcontainer/hcontainer.h" +#include "source_cell/klist.h" + +/************************************************ + * unit test of DensityMatrix constructor + ***********************************************/ + +/** + * This unit test dmk_from_psi + */ + +// test_size is the number of atoms in the unitcell +// modify test_size to test different size of unitcell +int test_size = 10; +int test_nw = 26; + +class DMTest : public testing::Test +{ + protected: + Parallel_Orbitals* paraV; + int dsize; + int my_rank = 0; + UnitCell ucell; + void SetUp() override + { +#ifdef __MPI + // MPI parallel settings + MPI_Comm_size(MPI_COMM_WORLD, &dsize); + MPI_Comm_rank(MPI_COMM_WORLD, &my_rank); +#endif + + // set up a unitcell, with one element and test_size atoms, each atom has test_nw orbitals + ucell.ntype = 1; + ucell.nat = test_size; + ucell.atoms = new Atom[ucell.ntype]; + ucell.iat2it = new int[ucell.nat]; + ucell.iat2ia = new int[ucell.nat]; + ucell.atoms[0].tau = new ModuleBase::Vector3[ucell.nat]; + ucell.itia2iat.create(ucell.ntype, ucell.nat); + for (int iat = 0; iat < ucell.nat; iat++) + { + ucell.iat2it[iat] = 0; + ucell.iat2ia[iat] = iat; + ucell.atoms[0].tau[iat] = ModuleBase::Vector3(0.0, 0.0, 0.0); + ucell.itia2iat(0, iat) = iat; + } + ucell.atoms[0].na = test_size; + ucell.atoms[0].nw = test_nw; + ucell.atoms[0].iw2l = new int[test_nw]; + ucell.atoms[0].iw2m = new int[test_nw]; + ucell.atoms[0].iw2n = new int[test_nw]; + for (int iw = 0; iw < test_nw; ++iw) + { + ucell.atoms[0].iw2l[iw] = 0; + ucell.atoms[0].iw2m[iw] = 0; + ucell.atoms[0].iw2n[iw] = 0; + } + ucell.set_iat2iwt(1); + init_parav(); + // set paraV + init_parav(); + } + + void TearDown() + { + delete paraV; + delete[] ucell.atoms[0].tau; + delete[] ucell.atoms[0].iw2l; + delete[] ucell.atoms[0].iw2m; + delete[] ucell.atoms[0].iw2n; + delete[] ucell.atoms; + } + +#ifdef __MPI + void init_parav() + { + int nb = 2; + int global_row = test_size * test_nw; + int global_col = test_size * test_nw; + std::ofstream ofs_running; + paraV = new Parallel_Orbitals(); + paraV->init(global_row, global_col, nb, MPI_COMM_WORLD); + paraV->set_atomic_trace(ucell.get_iat2iwt(), test_size, global_row); + } +#else + void init_parav() + { + } +#endif +}; + +TEST_F(DMTest, dmk_from_psi_nspin1) +{ + // initalize a kvectors + K_Vectors* kv = nullptr; + int nks = 2; + kv = new K_Vectors; + kv->set_nks(nks); + kv->kvec_d.resize(nks); + kv->kvec_d[1].x = 0.5; + // construct DM + std::cout << "dim0: " << paraV->dim0 << " dim1:" << paraV->dim1 << std::endl; + std::cout << "nrow: " << paraV->nrow << " ncol:" << paraV->ncol << std::endl; + int nspin = 1; + module_dm::DensityMatrix DM(kv, paraV, nspin); + // compare + EXPECT_EQ(DM.get_dmk_nks(), kv->get_nks()); + EXPECT_EQ(DM.get_dmk_nrow(), paraV->nrow); + EXPECT_EQ(DM.get_dmk_ncol(), paraV->ncol); + + // set elements of DMK + for (int is = 1; is <= nspin; is++) + { + for (int ik = 0; ik < kv->get_nks() / nspin; ik++) + { + for (int i = 0; i < paraV->nrow; i++) + { + for (int j = 0; j < paraV->ncol; j++) + { + DM.set_dmk(is, ik, i, j, is + ik * i + j); + } + } + } + } + // compare + for (int is = 1; is <= nspin; is++) + { + for (int ik = 0; ik < kv->get_nks() / nspin; ik++) + { + for (int i = 0; i < paraV->nrow; i++) + { + for (int j = 0; j < paraV->ncol; j++) + { + EXPECT_EQ(DM.get_dmk(is, ik, i, j), is + ik * i + j); + } + } + } + } + // test for get_dmk_ptr + for (int is = 1; is <= nspin; is++) + { + int ik_begin = (is - 1) * kv->get_nks() / nspin; + for (int ik = 0; ik < kv->get_nks() / nspin; ik++) + { + double* ptr = DM.get_dmk_ptr(ik + ik_begin); + for (int i = 0; i < paraV->nrow; i++) + { + for (int j = 0; j < paraV->ncol; j++) + { + // std::cout << ptr[i*paraV->ncol+j] << " "; + EXPECT_EQ(ptr[i * paraV->ncol + j], is + ik * i + j); + } + } + } + } + // delete kv + delete kv; +} + +int main(int argc, char** argv) +{ +#ifdef __MPI + MPI_Init(&argc, &argv); +#endif + testing::InitGoogleTest(&argc, argv); + int result = RUN_ALL_TESTS(); +#ifdef __MPI + MPI_Finalize(); +#endif + return result; +} diff --git a/source/source_estate/module_dm/unittests/test_dm_r_init.cpp b/source/source_estate/module_dm/unittests/test_dm_r_init.cpp new file mode 100644 index 00000000000..8c1a6b988e5 --- /dev/null +++ b/source/source_estate/module_dm/unittests/test_dm_r_init.cpp @@ -0,0 +1,309 @@ +#include + +#include +#include "gmock/gmock.h" +#include "gtest/gtest.h" +#include "source_estate/module_dm/density_matrix.h" +#include "source_hamilt/module_hcontainer/hcontainer.h" +#include "source_cell/klist.h" +/************************************************ + * unit test of DensityMatrix constructor + ***********************************************/ + +/** + * This unit test construct a DensityMatrix object + */ + +// test_size is the number of atoms in the unitcell +// modify test_size to test different size of unitcell +int test_size = 10; +int test_nw = 26; + +class DMTest : public testing::Test +{ + protected: + Parallel_Orbitals* paraV; + int dsize; + int my_rank = 0; + UnitCell ucell; + void SetUp() override + { +#ifdef __MPI + // MPI parallel settings + MPI_Comm_size(MPI_COMM_WORLD, &dsize); + MPI_Comm_rank(MPI_COMM_WORLD, &my_rank); +#endif + + // set up a unitcell, with one element and test_size atoms, each atom has test_nw orbitals + ucell.ntype = 1; + ucell.nat = test_size; + ucell.atoms = new Atom[ucell.ntype]; + ucell.iat2it = new int[ucell.nat]; + ucell.iat2ia = new int[ucell.nat]; + ucell.atoms[0].tau.resize(ucell.nat); + ucell.itia2iat.create(ucell.ntype, ucell.nat); + for (int iat = 0; iat < ucell.nat; iat++) + { + ucell.iat2it[iat] = 0; + ucell.iat2ia[iat] = iat; + ucell.atoms[0].tau[iat] = ModuleBase::Vector3(0.0, 0.0, 0.0); + ucell.itia2iat(0, iat) = iat; + } + ucell.atoms[0].na = test_size; + ucell.atoms[0].nw = test_nw; + ucell.atoms[0].iw2l.resize(test_nw); + ucell.atoms[0].iw2m.resize(test_nw); + ucell.atoms[0].iw2n.resize(test_nw); + for (int iw = 0; iw < test_nw; ++iw) + { + ucell.atoms[0].iw2l[iw] = 0; + ucell.atoms[0].iw2m[iw] = 0; + ucell.atoms[0].iw2n[iw] = 0; + } + ucell.set_iat2iwt(1); + init_parav(); + // set paraV + init_parav(); + } + + void TearDown() override + { + delete paraV; + delete[] ucell.atoms; + } + +#ifdef __MPI + void init_parav() + { + int nb = 2; + int global_row = test_size * test_nw; + int global_col = test_size * test_nw; + std::ofstream ofs_running; + paraV = new Parallel_Orbitals(); + paraV->init(global_row, global_col, nb, MPI_COMM_WORLD); + paraV->set_atomic_trace(ucell.get_iat2iwt(), test_size, global_row); + } +#else + void init_parav() + { + } +#endif +}; + +// test for construct DMR from GridD and UnitCell +TEST_F(DMTest, DMInit1) +{ + // initalize a kvectors + K_Vectors* kv = nullptr; + int nspin = 1; + int nks = 2; // since nspin = 1 + kv = new K_Vectors; + kv->set_nks(nks); + kv->kvec_d.resize(nks); + kv->kvec_d[1].x = 0.5; + // construct DM + std::cout << "dim0: " << paraV->dim0 << " dim1:" << paraV->dim1 << std::endl; + std::cout << "nrow: " << paraV->nrow << " ncol:" << paraV->ncol << std::endl; + module_dm::DensityMatrix DM(paraV, nspin, kv->kvec_d, nks); + // initialize this->dmr + Grid_Driver gd(0,0); + DM.init_dmr(&gd, &ucell); + // compare + EXPECT_EQ(DM.get_dmr_ptr(1)->size_atom_pairs(), test_size * test_size); + EXPECT_EQ(DM.get_dmr_ptr(1)->get_atom_pair(2, 2).get_atom_i(), 2); + EXPECT_EQ(DM.get_dmr_ptr(1)->get_atom_pair(2, 2).get_atom_j(), 2); + EXPECT_EQ(DM.get_dmr_ptr(1)->get_atom_pair(2, 2).get_row_size(), paraV->get_nrow_atom(2)); + EXPECT_EQ(DM.get_dmr_ptr(1)->get_atom_pair(2, 2).get_col_size(), paraV->get_ncol_atom(2)); + delete kv; +} + +// test for construct DMR from RA and UnitCell +TEST_F(DMTest, DMInit2) +{ + // initalize a kvectors + K_Vectors* kv = nullptr; + int nspin = 1; + int nks = 2; // since nspin = 1 + kv = new K_Vectors; + kv->set_nks(nks); + kv->kvec_d.resize(nks); + kv->kvec_d[1].x = 0.5; + // construct DM + std::cout << "dim0: " << paraV->dim0 << " dim1:" << paraV->dim1 << std::endl; + std::cout << "nrow: " << paraV->nrow << " ncol:" << paraV->ncol << std::endl; + module_dm::DensityMatrix DM(paraV, nspin, kv->kvec_d, nks); + // initialize Record_adj using Grid_Driver + Grid_Driver gd(0,0); + Record_adj ra; + ra.na_each.resize(ucell.nat); + ra.info_offset.resize(ucell.nat); + // count adjacent atoms for each atom first to lay out the flat info + std::vector all_adjs(ucell.nat); + int total = 0; + for (int iat1 = 0; iat1 < ucell.nat; iat1++) + { + auto tau1 = ucell.get_tau(iat1); + int T1, I1; + ucell.iat2iait(iat1, &I1, &T1); + gd.Find_atom(ucell, tau1, T1, I1, &all_adjs[iat1]); + ra.na_each[iat1] = all_adjs[iat1].adj_num + 1; + ra.info_offset[iat1] = total; + total += ra.na_each[iat1]; + } + ra.info.resize(total); + for (int iat1 = 0; iat1 < ucell.nat; iat1++) + { + const AdjacentAtomInfo& adjs = all_adjs[iat1]; + for (int ad = 0; ad < ra.na_each[iat1]; ++ad) + { + const int T2 = adjs.ntype[ad]; + const int I2 = adjs.natom[ad]; + const ModuleBase::Vector3& R_index = adjs.box[ad]; + std::array& rec = ra.info[ra.info_offset[iat1] + ad]; + rec[0] = R_index.x; + rec[1] = R_index.y; + rec[2] = R_index.z; + rec[3] = T2; + rec[4] = I2; + } + } + DM.init_dmr(ra, &ucell); + // compare + EXPECT_EQ(DM.get_dmr_ptr(1)->size_atom_pairs(), test_size * test_size); + EXPECT_EQ(DM.get_dmr_ptr(1)->get_atom_pair(2, 2).get_atom_i(), 2); + EXPECT_EQ(DM.get_dmr_ptr(1)->get_atom_pair(2, 2).get_atom_j(), 2); + EXPECT_EQ(DM.get_dmr_ptr(1)->get_atom_pair(2, 2).get_row_size(), paraV->get_nrow_atom(2)); + EXPECT_EQ(DM.get_dmr_ptr(1)->get_atom_pair(2, 2).get_col_size(), paraV->get_ncol_atom(2)); + // release memory + delete kv; +} + +// test for construct DMR from another HContainer +TEST_F(DMTest, DMInit3) +{ + // initalize a kvectors + K_Vectors* kv = nullptr; + int nspin = 2; + int nks = 4; // since nspin = 2 + kv = new K_Vectors; + kv->set_nks(nks); + kv->kvec_d.resize(nks); + kv->kvec_d[1].x = 0.5; + kv->kvec_d[3].x = 0.5; + // construct a DM + module_dm::DensityMatrix, double> DM(paraV, nspin, kv->kvec_d, kv->get_nks() / nspin); + Grid_Driver gd(0, 0); + DM.init_dmr(&gd, &ucell); + std::cout << "dim0: " << paraV->dim0 << " dim1:" << paraV->dim1 << std::endl; + // construct another DM + module_dm::DensityMatrix, double> DM1(paraV, nspin, kv->kvec_d, kv->get_nks() / nspin); + DM1.init_dmr(*DM.get_dmr_ptr(1)); + // compare + EXPECT_EQ(DM1.get_dmr_ptr(2)->size_atom_pairs(), test_size * test_size); + EXPECT_EQ(DM1.get_dmr_ptr(2)->get_atom_pair(2, 2).get_atom_i(), 2); + EXPECT_EQ(DM1.get_dmr_ptr(1)->get_atom_pair(2, 2).get_atom_j(), 2); + EXPECT_EQ(DM1.get_dmr_ptr(1)->get_atom_pair(2, 2).get_row_size(), paraV->get_nrow_atom(2)); + EXPECT_EQ(DM1.get_dmr_ptr(2)->get_atom_pair(2, 2).get_col_size(), paraV->get_ncol_atom(2)); + // + delete kv; +} + +// test for construct DMR from another HContainer > +TEST_F(DMTest, DMInit4) +{ + // initalize a kvectors + K_Vectors* kv = nullptr; + int nspin = 2; + int nks = 4; // since nspin = 2 + kv = new K_Vectors; + kv->set_nks(nks); + kv->kvec_d.resize(nks); + kv->kvec_d[1].x = 0.5; + kv->kvec_d[3].x = 0.5; + // construct a new HContainer + Grid_Driver gd(0, 0); + hamilt::HContainer>* tmp_DMR; + tmp_DMR = new hamilt::HContainer>(paraV); + // set up a HContainer + for (int iat1 = 0; iat1 < ucell.nat; iat1++) + { + auto tau1 = ucell.get_tau(iat1); + int T1, I1; + ucell.iat2iait(iat1, &I1, &T1); + AdjacentAtomInfo adjs; + gd.Find_atom(ucell, tau1, T1, I1, &adjs); + // std::cout << "adjs.adj_num: " <is_invalid_atom_pair(iat1, iat2)) + { + continue; + } + ModuleBase::Vector3& R_index = adjs.box[ad]; + // std::cout << "R_index: " << R_index.x << " " << R_index.y << " " << R_index.z << std::endl; + hamilt::AtomPair> tmp_ap(iat1, iat2, R_index.x, R_index.y, R_index.z, paraV); + tmp_DMR->insert_pair(tmp_ap); + } + } + // construct a DM from this HContainer + module_dm::DensityMatrix, double> DM(paraV, nspin, kv->kvec_d, kv->get_nks() / nspin); + DM.init_dmr(*tmp_DMR); + std::cout << "dim0: " << paraV->dim0 << " dim1:" << paraV->dim1 << std::endl; + // compare + EXPECT_EQ(DM.get_dmr_ptr(2)->size_atom_pairs(), test_size * test_size); + EXPECT_EQ(DM.get_dmr_ptr(2)->get_atom_pair(2, 2).get_atom_i(), 2); + EXPECT_EQ(DM.get_dmr_ptr(1)->get_atom_pair(2, 2).get_atom_j(), 2); + EXPECT_EQ(DM.get_dmr_ptr(1)->get_atom_pair(2, 2).get_row_size(), paraV->get_nrow_atom(2)); + EXPECT_EQ(DM.get_dmr_ptr(2)->get_atom_pair(2, 2).get_col_size(), paraV->get_ncol_atom(2)); + // + delete kv; +} + +// test for save_dmr +TEST_F(DMTest, saveDMR) +{ + // initalize a kvectors + K_Vectors* kv = nullptr; + int nspin = 2; + int nks = 4; // since nspin = 2 + kv = new K_Vectors; + kv->set_nks(nks); + kv->kvec_d.resize(nks); + kv->kvec_d[1].x = 0.5; + kv->kvec_d[3].x = 0.5; + // construct a DM + module_dm::DensityMatrix, double> DM(paraV, nspin, kv->kvec_d, kv->get_nks() / nspin); + Grid_Driver gd(0, 0); + DM.init_dmr(&gd, &ucell); + // construct another DM + module_dm::DensityMatrix, double> DM_test(paraV, nspin, kv->kvec_d, kv->get_nks() / nspin); + DM_test.init_dmr(*DM.get_dmr_ptr(1)); + DM_test.save_dmr(); + EXPECT_EQ(DM_test.get_dmr_ptr(1)->get_nnr(), DM.get_dmr_ptr(1)->get_nnr()); + EXPECT_EQ(DM_test.get_dmr_ptr(1)->get_nnr(), DM_test.get_dmr_save()[0].size()); + // add a new AtomPair, act as a relaxation + hamilt::AtomPair tmp_ap(9, 9, 1, 0, 0, paraV); + DM_test.get_dmr_ptr(1)->insert_pair(tmp_ap); + DM_test.get_dmr_ptr(1)->allocate(); + // update DMR_save + DM_test.save_dmr(); + EXPECT_EQ(DM_test.get_dmr_ptr(1)->get_nnr(), DM_test.get_dmr_save()[0].size()); + // delete + delete kv; +} + +int main(int argc, char** argv) +{ +#ifdef __MPI + MPI_Init(&argc, &argv); +#endif + testing::InitGoogleTest(&argc, argv); + int result = RUN_ALL_TESTS(); +#ifdef __MPI + MPI_Finalize(); +#endif + return result; +} diff --git a/source/source_estate/module_dm/test/test_soc_magnetization_roundtrip.cpp b/source/source_estate/module_dm/unittests/test_soc_magnetization_roundtrip.cpp similarity index 88% rename from source/source_estate/module_dm/test/test_soc_magnetization_roundtrip.cpp rename to source/source_estate/module_dm/unittests/test_soc_magnetization_roundtrip.cpp index 419c0211746..a7057509645 100644 --- a/source/source_estate/module_dm/test/test_soc_magnetization_roundtrip.cpp +++ b/source/source_estate/module_dm/unittests/test_soc_magnetization_roundtrip.cpp @@ -11,21 +11,21 @@ * Physical invariant (must hold regardless of internal sign conventions): * the magnetization of the occupied one-electron state that is * encoded in the density matrix must be recovered, with the CORRECT SIGN - * in ALL THREE cartesian components, by func_xyz_to_updown(). + * in ALL THREE cartesian components, by xyz_to_updown(). * * Why this test exists (regression for the #7664 nspin=4 m_y sign flip): * ABACUS builds the k-space DM as DM_{ab} = sum_n w_n conj(c_{n,a}) c_{n,b} - * (cal_dm_psi.cpp: the conj() is applied to the FIRST index a). Hence the + * (dm_from_psi.cpp: the conj() is applied to the FIRST index a). Hence the * stored DM block is the complex conjugate of the physical 1-RDM P: * DM_{up,dn} = conj(c_up) c_dn = conj(P_{up,dn}). * Since m_x, m_z read Re() (conjugation-invariant) but m_y reads Im(), - * ONLY m_y is sensitive to this conjugation. func_xyz_to_updown() must be + * ONLY m_y is sensitive to this conjugation. xyz_to_updown() must be * consistent with that stored convention. PR #7664 set the m_y extraction * to the "bare" textbook formula (valid for P, not for conj(P)), which * flips m_y for in-plane moments and quenches non-collinear order * (e.g. Mn3Sn 120-degree AFM). This test pins m_y down. * - * The helper build_DM_block_as_cal_dm_psi() MUST mirror cal_dm_psi.cpp. If + * The helper build_DM_block_as_dm_from_psi() MUST mirror dm_from_psi.cpp. If * that convention is ever changed (e.g. the "upstream" fix that makes the DM * hold the physical P), update the helper in the SAME commit so this test * keeps asserting the physical invariant. @@ -45,10 +45,10 @@ void spinor_from_direction(const double mhat[3], cd c[2]) c[1] = std::sin(0.5 * th) * cd(std::cos(ph), std::sin(ph)); } -// Build the 4 spinor-block DM elements EXACTLY as cal_dm_psi.cpp stores them: +// Build the 4 spinor-block DM elements EXACTLY as dm_from_psi.cpp stores them: // DM_{a,b} = sum_occ w * conj(c_a) * c_b (conj on the first index) // layout tmp = {uu, ud, du, dd} -void build_DM_block_as_cal_dm_psi(const cd c[2], double w, cd tmp[4]) +void build_DM_block_as_dm_from_psi(const cd c[2], double w, cd tmp[4]) { tmp[0] = w * std::conj(c[0]) * c[0]; // uu tmp[1] = w * std::conj(c[0]) * c[1]; // ud @@ -89,11 +89,11 @@ TEST(SocMagnetizationRoundtrip, ExtractRecoversPhysicalMagnetization) physical_m(c, m_ref); // the TRUE magnetization encoded in the state cd tmp[4]; - build_DM_block_as_cal_dm_psi(c, 1.0, tmp); + build_DM_block_as_dm_from_psi(c, 1.0, tmp); // 2x2 output buffer (row-major), func writes rho0/x/y/z into step_trace slots at icol=0 double out[4] = {0, 0, 0, 0}; - elecstate::DensityMatrix_Tools::func_xyz_to_updown(tmp, 0, step_trace, out); + module_dm::xyz_to_updown(tmp, 0, step_trace, out); const double mx = out[step_trace[1]]; const double my = out[step_trace[2]]; @@ -125,10 +125,10 @@ TEST(SocMagnetizationRoundtrip, ComplexSpecializationRecoversPhysicalMagnetizati physical_m(c, m_ref); cd tmp[4]; - build_DM_block_as_cal_dm_psi(c, 1.0, tmp); + build_DM_block_as_dm_from_psi(c, 1.0, tmp); cd out[4] = {cd(0, 0), cd(0, 0), cd(0, 0), cd(0, 0)}; - elecstate::DensityMatrix_Tools::func_xyz_to_updown>(tmp, 0, step_trace, out); + module_dm::xyz_to_updown>(tmp, 0, step_trace, out); EXPECT_NEAR(out[step_trace[1]].real(), m_ref[0], 1e-10) << "m_x"; EXPECT_NEAR(out[step_trace[2]].real(), m_ref[1], 1e-10) << "m_y (complex specialization)"; diff --git a/source/source_estate/module_dm/test/tmp_mocks.cpp b/source/source_estate/module_dm/unittests/tmp_mocks.cpp similarity index 98% rename from source/source_estate/module_dm/test/tmp_mocks.cpp rename to source/source_estate/module_dm/unittests/tmp_mocks.cpp index 4ada21ea571..12b66268ced 100644 --- a/source/source_estate/module_dm/test/tmp_mocks.cpp +++ b/source/source_estate/module_dm/unittests/tmp_mocks.cpp @@ -98,7 +98,7 @@ Grid_Driver::~Grid_Driver() } // mock Record_adj -#include "source_lcao/record_adj.h" +#include "source_cell/record_adj.h" Record_adj::Record_adj() { } diff --git a/source/source_estate/module_pot/h_tddft_pw.cpp b/source/source_estate/module_pot/h_tddft_pw.cpp index 44efcc3c546..1b11ba214e5 100644 --- a/source/source_estate/module_pot/h_tddft_pw.cpp +++ b/source/source_estate/module_pot/h_tddft_pw.cpp @@ -1,19 +1,14 @@ #include "h_tddft_pw.h" -#include "source_base/global_variable.h" #include "source_base/timer.h" #include "source_base/tool_quit.h" -#include "source_io/module_efield/td_efield_io.h" -#include "source_io/module_parameter/parameter.h" #include "td_field_manager.h" namespace elecstate { int H_TDDFT_pw::stype = 0; -ModuleBase::Vector3 H_TDDFT_pw::At; -ModuleBase::Vector3 H_TDDFT_pw::At_laststep; -ModuleBase::Vector3 H_TDDFT_pw::Et; +ModuleBase::Vector3 H_TDDFT_pw::efield_ha; std::vector H_TDDFT_pw::global_vext_time = {0.0, 0.0, 0.0}; H_TDDFT_pw::H_TDDFT_pw(const ModulePW::PW_Basis* rho_basis_in, @@ -29,17 +24,17 @@ H_TDDFT_pw::H_TDDFT_pw(const ModulePW::PW_Basis* rho_basis_in, { ModuleBase::WARNING_QUIT("H_TDDFT_pw", "RT-TDDFT field manager is not initialized."); } - sync_compatibility_state(*field_manager_); + set_field_state(*field_manager_); } -void H_TDDFT_pw::sync_compatibility_state(const TDFieldManager& manager) +void H_TDDFT_pw::set_field_state(const TDFieldManager& manager) { stype = manager.gauge(); - At = manager.vector_potential(); - At_laststep = manager.vector_potential_laststep(); - Et = manager.electric_field(); - const ModuleBase::Vector3& total_field = manager.total_electric_field(); - global_vext_time = {total_field[0], total_field[1], total_field[2]}; + efield_ha = manager.efield_ha(); + const ModuleBase::Vector3& total_field = manager.efield_ha(); + // Velocity gauge has no scalar electric-field force. + const double ry_scale = stype == 1 ? 0.0 : 2.0; + global_vext_time = {ry_scale * total_field[0], ry_scale * total_field[1], ry_scale * total_field[2]}; } void H_TDDFT_pw::cal_fixed_v(double* vl_pseudo) @@ -50,10 +45,6 @@ void H_TDDFT_pw::cal_fixed_v(double* vl_pseudo) return; } - // Advance exactly once per rebuilt fixed potential. The potential then - // consumes the same per-occurrence samples exposed to field output. - field_manager_->advance_length_gauge(); - sync_compatibility_state(*field_manager_); if (!field_manager_->active()) { return; @@ -61,11 +52,11 @@ void H_TDDFT_pw::cal_fixed_v(double* vl_pseudo) ModuleBase::timer::start("H_TDDFT_pw", "cal_fixed_v"); const std::vector& fields = field_manager_->fields(); - const std::vector& field_values = field_manager_->field_values(); + const std::vector& field_values = field_manager_->field_vals_ha(); for (std::size_t field_index = 0; field_index < fields.size(); ++field_index) { std::vector vext_space(this->rho_basis_->nrxx, 0.0); - const double field_value = field_values[field_index]; + const double field_value = 2.0 * field_values[field_index]; // Scalar potential is stored in Rydberg. cal_v_space_length(vext_space, fields[field_index].direction() + 1); for (std::size_t ir = 0; ir < static_cast(this->rho_basis_->nrxx); ++ir) @@ -73,10 +64,6 @@ void H_TDDFT_pw::cal_fixed_v(double* vl_pseudo) vl_pseudo[ir] += vext_space[ir] * field_value; } } - if (PARAM.inp.out_efield && GlobalV::MY_RANK == 0) - { - ModuleIO::write_td_field_values(*field_manager_, PARAM.globalv.global_out_dir); - } ModuleBase::timer::end("H_TDDFT_pw", "cal_fixed_v"); } diff --git a/source/source_estate/module_pot/h_tddft_pw.h b/source/source_estate/module_pot/h_tddft_pw.h index b609bcfc312..94a6814aa9e 100644 --- a/source/source_estate/module_pot/h_tddft_pw.h +++ b/source/source_estate/module_pot/h_tddft_pw.h @@ -16,8 +16,8 @@ class TDFieldManager; * @brief Length-gauge RT-TDDFT potential backed by TDFieldManager. * * The class consumes the manager's current field samples to construct the - * periodic real-space potential. Static members are retained only as mirrors - * for legacy propagation, current, force, and restart interfaces. + * periodic real-space potential. Static members publish only gauge and sampled electric-field data + * for LCAO and force consumers. Time advancement belongs to the ESolver. */ class H_TDDFT_pw : public PotBase { @@ -44,11 +44,11 @@ class H_TDDFT_pw : public PotBase void cal_fixed_v(double* vl_pseudo) override; /** - * @brief Synchronize legacy static mirrors with the field manager state. + * @brief Publish sampled fields for LCAO and ionic-force consumers. * * @param manager Active RT-TDDFT field manager. */ - static void sync_compatibility_state(const TDFieldManager& manager); + static void set_field_state(const TDFieldManager& manager); /** * @brief Compute ionic force of electric field. @@ -58,17 +58,11 @@ class H_TDDFT_pw : public PotBase */ static void compute_force(const UnitCell& cell, ModuleBase::matrix& fe); - /** @brief Legacy mirror of the spatial-gauge selector. */ + /** @brief Spatial-gauge selector for the remaining LCAO consumers. */ static int stype; - /** @brief Legacy mirror of the midpoint vector potential. */ - static ModuleBase::Vector3 At; - - /** @brief Legacy mirror of the current vector-potential increment. */ - static ModuleBase::Vector3 At_laststep; - - /** @brief Legacy mirror of the instantaneous hybrid-gauge field. */ - static ModuleBase::Vector3 Et; + /** @brief Sampled electric field in Hartree atomic units for LCAO consumers. */ + static ModuleBase::Vector3 efield_ha; private: static std::vector global_vext_time; diff --git a/source/source_estate/module_pot/pot_cosikr.cpp b/source/source_estate/module_pot/pot_cosikr.cpp index 66a5ef7849c..9baa91f30b9 100644 --- a/source/source_estate/module_pot/pot_cosikr.cpp +++ b/source/source_estate/module_pot/pot_cosikr.cpp @@ -11,43 +11,43 @@ namespace elecstate { Pot_Cosikr::Pot_Cosikr( - const ModulePW::PW_Basis* rho_basis_in, - const ModuleBase::Vector3 &kvec_d_in, - const std::vector &phase_in, - const std::vector &litude_in) - :kvec_d(kvec_d_in), - phase(phase_in), - amplitude(amplitude_in) + const ModulePW::PW_Basis* rho_basis_in, + const ModuleBase::Vector3 &kvec_d_in, + const std::vector &phase_in, + const std::vector &litude_in) + :kvec_d(kvec_d_in), + phase(phase_in), + amplitude(amplitude_in) { - this->rho_basis_ = rho_basis_in; - this->dynamic_mode = true; - this->fixed_mode = false; + this->rho_basis_ = rho_basis_in; + this->dynamic_mode = true; + this->fixed_mode = false; } void Pot_Cosikr::cal_v_eff(const Charge*const chg, const UnitCell*const ucell, ModuleBase::matrix &v_eff) { - ModuleBase::TITLE("Pot_Cosikr", "cal_v_eff"); - ModuleBase::timer::start("Pot_Cosikr", "cal_veff"); - assert(v_eff.nr == this->phase.size()); - assert(v_eff.nr == this->amplitude.size()); - int ir = 0; - for (int ix = 0; ix < this->rho_basis_->nx; ++ix) - { - const double phase_x = this->kvec_d.x * ix / this->rho_basis_->nx; - for (int iy = 0; iy < this->rho_basis_->ny; ++iy) - { - const double phase_xy = phase_x + this->kvec_d.y * iy / this->rho_basis_->ny; - for (int iz = this->rho_basis_->startz_current; iz < this->rho_basis_->startz_current + this->rho_basis_->nplane; ++iz) - { - const double phase_xyz = phase_xy + this->kvec_d.z * iz / this->rho_basis_->nz; - for(int is=0; isamplitude[is] * std::cos((phase_xyz + this->phase[is]) * ModuleBase::TWO_PI); - ++ir; - } - } - } - ModuleBase::timer::end("Pot_Cosikr", "cal_veff"); + ModuleBase::TITLE("Pot_Cosikr", "cal_v_eff"); + ModuleBase::timer::start("Pot_Cosikr", "cal_veff"); + assert(v_eff.nr == this->phase.size()); + assert(v_eff.nr == this->amplitude.size()); + int ir = 0; + for (int ix = 0; ix < this->rho_basis_->nx; ++ix) + { + const double phase_x = this->kvec_d.x * ix / this->rho_basis_->nx; + for (int iy = 0; iy < this->rho_basis_->ny; ++iy) + { + const double phase_xy = phase_x + this->kvec_d.y * iy / this->rho_basis_->ny; + for (int iz = this->rho_basis_->startz_current; iz < this->rho_basis_->startz_current + this->rho_basis_->nplane; ++iz) + { + const double phase_xyz = phase_xy + this->kvec_d.z * iz / this->rho_basis_->nz; + for(int is=0; isamplitude[is] * std::cos((phase_xyz + this->phase[is]) * ModuleBase::TWO_PI); + ++ir; + } + } + } + ModuleBase::timer::end("Pot_Cosikr", "cal_veff"); } } \ No newline at end of file diff --git a/source/source_estate/module_pot/pot_cosikr.h b/source/source_estate/module_pot/pot_cosikr.h index 1cdbd72c35e..0465a443c7d 100644 --- a/source/source_estate/module_pot/pot_cosikr.h +++ b/source/source_estate/module_pot/pot_cosikr.h @@ -17,18 +17,18 @@ namespace elecstate class Pot_Cosikr : public PotBase { public: - Pot_Cosikr( - const ModulePW::PW_Basis* rho_basis_in, - const ModuleBase::Vector3 &kvec_d_in, - const std::vector &phase_in, - const std::vector &litude_in); + Pot_Cosikr( + const ModulePW::PW_Basis* rho_basis_in, + const ModuleBase::Vector3 &kvec_d_in, + const std::vector &phase_in, + const std::vector &litude_in); - void cal_v_eff(const Charge*const chg, const UnitCell*const ucell, ModuleBase::matrix &v_eff) override; + void cal_v_eff(const Charge*const chg, const UnitCell*const ucell, ModuleBase::matrix &v_eff) override; private: - const ModuleBase::Vector3 kvec_d; - const std::vector phase; - const std::vector amplitude; + const ModuleBase::Vector3 kvec_d; + const std::vector phase; + const std::vector amplitude; }; } diff --git a/source/source_estate/module_pot/pot_surchem.cpp b/source/source_estate/module_pot/pot_surchem.cpp new file mode 100644 index 00000000000..113544aa61b --- /dev/null +++ b/source/source_estate/module_pot/pot_surchem.cpp @@ -0,0 +1,45 @@ +#include "pot_surchem.h" + +namespace elecstate +{ + +PotSurChem::PotSurChem(const ModulePW::PW_Basis* rho_basis_in, + Structure_Factor* structure_factors_in, + const double* vlocal_in, + surchem* surchem_in) + : vlocal(vlocal_in), surchem_(surchem_in) +{ + this->rho_basis_ = rho_basis_in; + this->structure_factors_ = structure_factors_in; + this->dynamic_mode = true; + this->fixed_mode = false; +} + +PotSurChem::~PotSurChem() +{ + if (this->allocated) + { + this->surchem_->clear(); + } +} + +void PotSurChem::cal_v_eff(const Charge* const chg, const UnitCell* const ucell, ModuleBase::matrix& v_eff) +{ + if (!this->allocated) + { + this->surchem_->allocate(this->rho_basis_->nrxx, v_eff.nr); + this->allocated = true; + } + ModuleBase::matrix v_sol_correction(v_eff.nr, this->rho_basis_->nrxx); + this->surchem_->v_correction(*ucell, + *chg->pgrid, + const_cast(this->rho_basis_), + v_eff.nr, + chg->rho, + this->vlocal, + this->structure_factors_, + v_sol_correction); + v_eff += v_sol_correction; +} + +} // namespace elecstate diff --git a/source/source_estate/module_pot/pot_surchem.h b/source/source_estate/module_pot/pot_surchem.h new file mode 100644 index 00000000000..a93685b9e2f --- /dev/null +++ b/source/source_estate/module_pot/pot_surchem.h @@ -0,0 +1,33 @@ +#ifndef POTSURCHEM_H +#define POTSURCHEM_H + +#include "source_hamilt/module_surchem/surchem.h" +#include "pot_base.h" + +namespace elecstate +{ + +class PotSurChem : public PotBase +{ + public: + // constructor for exchange-correlation potential + // meta-GGA should input matrix of kinetic potential, it is optional + PotSurChem(const ModulePW::PW_Basis* rho_basis_in, + Structure_Factor* structure_factors_in, + const double* vlocal_in, + surchem* surchem_in); + ~PotSurChem(); + + // Passing an explicit output matrix makes the lifetime and allocation explicit and avoids hidden allocations. + void cal_v_eff(const Charge* const chg, const UnitCell* const ucell, ModuleBase::matrix& v_eff) override; + + private: + surchem* surchem_ = nullptr; + Structure_Factor* structure_factors_ = nullptr; + const double* vlocal = nullptr; + bool allocated = false; +}; + +} // namespace elecstate + +#endif diff --git a/source/source_estate/module_pot/pot_surchem.hpp b/source/source_estate/module_pot/pot_surchem.hpp deleted file mode 100644 index dc06e60fd4f..00000000000 --- a/source/source_estate/module_pot/pot_surchem.hpp +++ /dev/null @@ -1,63 +0,0 @@ -#ifndef POTSURCHEM_H -#define POTSURCHEM_H - -#include "source_hamilt/module_surchem/surchem.h" -#include "pot_base.h" - -namespace elecstate -{ - -class PotSurChem : public PotBase -{ - public: - // constructor for exchange-correlation potential - // meta-GGA should input matrix of kinetic potential, it is optional - PotSurChem(const ModulePW::PW_Basis* rho_basis_in, - Structure_Factor* structure_factors_in, - const double* vlocal_in, - surchem* surchem_in) - : vlocal(vlocal_in), surchem_(surchem_in) - { - this->rho_basis_ = rho_basis_in; - this->structure_factors_ = structure_factors_in; - this->dynamic_mode = true; - this->fixed_mode = false; - } - ~PotSurChem() - { - if (this->allocated) - { - this->surchem_->clear(); - } - } - - // Passing an explicit output matrix makes the lifetime and allocation explicit and avoids hidden allocations. - void cal_v_eff(const Charge*const chg, const UnitCell*const ucell, ModuleBase::matrix& v_eff) override - { - if (!this->allocated) - { - this->surchem_->allocate(this->rho_basis_->nrxx, v_eff.nr); - this->allocated = true; - } - ModuleBase::matrix v_sol_correction(v_eff.nr, this->rho_basis_->nrxx); - this->surchem_->v_correction(*ucell, - *chg->pgrid, - const_cast(this->rho_basis_), - v_eff.nr, - chg->rho, - this->vlocal, - this->structure_factors_, - v_sol_correction); - v_eff += v_sol_correction; - } - - private: - surchem* surchem_ = nullptr; - Structure_Factor* structure_factors_ = nullptr; - const double* vlocal = nullptr; - bool allocated = false; -}; - -} // namespace elecstate - -#endif \ No newline at end of file diff --git a/source/source_estate/module_pot/pot_xc.cpp b/source/source_estate/module_pot/pot_xc.cpp index eb311cc9225..a4e2061699c 100644 --- a/source/source_estate/module_pot/pot_xc.cpp +++ b/source/source_estate/module_pot/pot_xc.cpp @@ -3,7 +3,6 @@ #include "source_base/timer.h" #include "source_base/constants.h" #include "source_hamilt/module_xc/xc_functional.h" -#include "source_io/module_parameter/parameter.h" #ifdef __LIBXC #include "source_hamilt/module_xc/libxc_abacus.h" @@ -17,6 +16,7 @@ void PotXC::cal_v_eff(const Charge*const chg, const UnitCell*const ucell, Module ModuleBase::TITLE("PotXC", "cal_veff"); ModuleBase::timer::start("PotXC", "cal_veff"); const int nrxx_current = chg->nrxx; + const int nspin = chg->nspin; //---------------------------------------------------------- // calculate the exchange-correlation potential @@ -33,7 +33,7 @@ void PotXC::cal_v_eff(const Charge*const chg, const UnitCell*const ucell, Module #endif const std::tuple etxc_vtxc_v = XC_Functional_Libxc::v_xc_meta(XC_Functional::get_func_id(), nrxx_current, ucell->omega, ucell->tpiba, chg, - PARAM.inp.nspin, hybrid_alpha, hse_omega); + nspin, hybrid_alpha, hse_omega); *(this->etxc_) = std::get<0>(etxc_vtxc_v); *(this->vtxc_) = std::get<1>(etxc_vtxc_v); v_eff += std::get<2>(etxc_vtxc_v); @@ -52,9 +52,10 @@ void PotXC::cal_v_eff(const Charge*const chg, const UnitCell*const ucell, Module #endif const std::tuple etxc_vtxc_v = XC_Functional::v_xc(nrxx_current, chg, ucell, - PARAM.inp.nspin, - PARAM.globalv.domag, - PARAM.globalv.domag_z, + nspin, + domag_, + domag_z_, + gga_grad_, hybrid_alpha, hse_omega); *(this->etxc_) = std::get<0>(etxc_vtxc_v); diff --git a/source/source_estate/module_pot/pot_xc.h b/source/source_estate/module_pot/pot_xc.h index 47243c633d7..50737e3ec95 100644 --- a/source/source_estate/module_pot/pot_xc.h +++ b/source/source_estate/module_pot/pot_xc.h @@ -11,11 +11,15 @@ class PotXC : public PotBase public: // constructor for exchange-correlation potential // meta-GGA should input matrix of kinetic potential, it is optional - PotXC(const ModulePW::PW_Basis* rho_basis_in, + PotXC(const bool domag, + const bool domag_z, + const int gga_grad, + const ModulePW::PW_Basis* rho_basis_in, double* etxc_in, double* vtxc_in, ModuleBase::matrix* vofk_in = nullptr) - : etxc_(etxc_in), vtxc_(vtxc_in), vofk(vofk_in) + : vofk(vofk_in), etxc_(etxc_in), vtxc_(vtxc_in), + domag_(domag), domag_z_(domag_z), gga_grad_(gga_grad) { this->rho_basis_ = rho_basis_in; this->dynamic_mode = true; @@ -27,8 +31,12 @@ class PotXC : public PotBase ModuleBase::matrix* vofk = nullptr; double* etxc_ = nullptr; double* vtxc_ = nullptr; + private: + const bool domag_; + const bool domag_z_; + const int gga_grad_; }; } // namespace elecstate -#endif \ No newline at end of file +#endif diff --git a/source/source_estate/module_pot/pot_xc_fdm.cpp b/source/source_estate/module_pot/pot_xc_fdm.cpp index 03349fe4e7e..946b001ee69 100644 --- a/source/source_estate/module_pot/pot_xc_fdm.cpp +++ b/source/source_estate/module_pot/pot_xc_fdm.cpp @@ -5,16 +5,21 @@ #include "pot_xc_fdm.h" #include "source_hamilt/module_xc/xc_functional.h" -#include "source_io/module_parameter/parameter.h" namespace elecstate { PotXC_FDM::PotXC_FDM( + const int nspin, + const bool domag, + const bool domag_z, + const int gga_grad, + const bool out_elf, + const int test_charge, const ModulePW::PW_Basis* rho_basis_in, const Charge*const chg_0_in, const UnitCell*const ucell) - : chg_0(chg_0_in) + : chg_0(chg_0_in), nspin_(nspin), domag_(domag), domag_z_(domag_z), gga_grad_(gga_grad), out_elf_(out_elf), test_charge_(test_charge) { this->rho_basis_ = rho_basis_in; this->dynamic_mode = true; @@ -28,9 +33,10 @@ PotXC_FDM::PotXC_FDM( #endif const std::tuple etxc_vtxc_v_0 = XC_Functional::v_xc(this->chg_0->nrxx, this->chg_0, ucell, - PARAM.inp.nspin, - PARAM.globalv.domag, - PARAM.globalv.domag_z, + nspin_, + domag_, + domag_z_, + gga_grad_, hybrid_alpha, hse_omega); this->v_xc_0 = std::get<2>(etxc_vtxc_v_0); @@ -49,7 +55,8 @@ void PotXC_FDM::cal_v_eff( Charge chg_01; chg_01.set_rhopw(chg_1->rhopw); - chg_01.allocate(chg_1->nspin, chg_01.kin_density()); + chg_01.allocate(chg_1->nspin, XC_Functional::get_ked_flag() || out_elf_, + XC_Functional::get_ked_flag(), test_charge_); for(int ir=0; ir etxc_vtxc_v_01 = XC_Functional::v_xc(chg_01.nrxx, &chg_01, ucell, - PARAM.inp.nspin, - PARAM.globalv.domag, - PARAM.globalv.domag_z, + nspin_, + domag_, + domag_z_, + gga_grad_, hybrid_alpha, hse_omega); const ModuleBase::matrix &v_xc_01 = std::get<2>(etxc_vtxc_v_01); diff --git a/source/source_estate/module_pot/pot_xc_fdm.h b/source/source_estate/module_pot/pot_xc_fdm.h index eb1cd880576..31b6da2321e 100644 --- a/source/source_estate/module_pot/pot_xc_fdm.h +++ b/source/source_estate/module_pot/pot_xc_fdm.h @@ -16,6 +16,12 @@ class PotXC_FDM : public PotBase public: PotXC_FDM( + const int nspin, + const bool domag, + const bool domag_z, + const int gga_grad, + const bool out_elf, + const int test_charge, const ModulePW::PW_Basis* rho_basis_in, const Charge*const chg_0_in, const UnitCell*const ucell); @@ -27,8 +33,15 @@ class PotXC_FDM : public PotBase const Charge*const chg_0 = nullptr; ModuleBase::matrix v_xc_0; +private: + const int nspin_; + const bool domag_; + const bool domag_z_; + const int gga_grad_; + const bool out_elf_; + const int test_charge_; }; } // namespace elecstate -#endif \ No newline at end of file +#endif diff --git a/source/source_estate/module_pot/potential_types.cpp b/source/source_estate/module_pot/potential_types.cpp index 2aca1d5075d..8b2f2b5e033 100644 --- a/source/source_estate/module_pot/potential_types.cpp +++ b/source/source_estate/module_pot/potential_types.cpp @@ -3,7 +3,7 @@ #include "gatefield.h" #include "pot_local.h" #include "pot_sep.h" -#include "pot_surchem.hpp" +#include "pot_surchem.h" #include "pot_xc.h" #include "potential_new.h" #include "source_base/global_function.h" @@ -33,7 +33,7 @@ PotBase* Potential::get_pot_type(const std::string& pot_type) } else if (pot_type == "xc") { - return new PotXC(this->rho_basis_, this->etxc_, this->vtxc_, &(this->vofk_eff)); + return new PotXC(PARAM.globalv.domag, PARAM.globalv.domag_z, PARAM.inp.gga_grad, this->rho_basis_, this->etxc_, this->vtxc_, &(this->vofk_eff)); } else if (pot_type == "surchem") { diff --git a/source/source_estate/module_pot/td_field.h b/source/source_estate/module_pot/td_field.h index 3cad298a92d..930decf78f8 100644 --- a/source/source_estate/module_pot/td_field.h +++ b/source/source_estate/module_pot/td_field.h @@ -21,7 +21,7 @@ struct TDFieldSample * @param electronic_step_in Zero-based electronic-step index. * @param simpson_node_in Node index in the current Simpson interval. * @param subdivisions_in Number of subintervals in one electronic step. - * @param time_in Physical sampling time in internal atomic time units. + * @param time_in Physical sampling time in Hartree atomic time units. */ TDFieldSample(const int electronic_step_in, const int simpson_node_in, const int subdivisions_in, const double time_in) : electronic_step(electronic_step_in), simpson_node(simpson_node_in), subdivisions(subdivisions_in), time(time_in) @@ -36,13 +36,13 @@ struct TDFieldSample int electronic_step; ///< Zero-based electronic-step index. int simpson_node; ///< Node index within the current electronic step. int subdivisions; ///< Number of subintervals in one electronic step. - double time; ///< Sampling time in internal atomic time units. + double time; ///< Sampling time in Hartree atomic time units. }; /** * @brief Scalar time profile of one configured electric field. * - * Implementations return the field in ABACUS internal propagation units. A + * Implementations return the field in Hartree atomic units. A * profile contains no Cartesian direction; direction handling is owned by * TDField and TDFieldManager. */ @@ -56,7 +56,7 @@ class TDFieldProfile * @brief Evaluate the electric field at one sampling point. * * @param sample Electronic-step and Simpson-node sampling information. - * @return Scalar field value in internal propagation units. + * @return Scalar field value in Hartree atomic units. */ virtual double electric_field(const TDFieldSample& sample) const = 0; }; diff --git a/source/source_estate/module_pot/td_field_manager.cpp b/source/source_estate/module_pot/td_field_manager.cpp index 64efe6edbcd..4f826b0c693 100644 --- a/source/source_estate/module_pot/td_field_manager.cpp +++ b/source/source_estate/module_pot/td_field_manager.cpp @@ -9,11 +9,41 @@ #include #include #include +#include +#include #include namespace { +std::vector restart_rows(std::istream& input) +{ + std::vector rows; + std::string line; + while (std::getline(input, line)) + { + const std::size_t comment = line.find('#'); + if (comment != std::string::npos) + { + line.erase(comment); + } + if (line.find_first_not_of(" \t\r\n") == std::string::npos) + { + continue; + } + rows.push_back(line); + if (rows.size() > 3) + { + break; + } + } + if (rows.size() != 3) + { + ModuleBase::WARNING_QUIT("TDFieldManager::read_restart", "Expected three data rows with 5, 3 and 3 fields."); + } + return rows; +} + int integration_subdivisions(const double omega, const double dt, const int gauge) { // Length gauge samples only the beginning of each electronic step and does @@ -39,7 +69,7 @@ double angular_frequency(const double frequency) double field_amplitude(const double amplitude) { // Convert the user-visible V/Angstrom scale to the propagation field unit. - return amplitude * ModuleBase::BOHR_TO_A / ModuleBase::Ry_to_eV; + return amplitude * ModuleBase::BOHR_TO_A / ModuleBase::Hartree_to_eV; } } // namespace @@ -55,165 +85,221 @@ TDFieldManager::TDFieldManager(const bool enabled, const double length_cut1, const double length_cut2, std::vector fields) - : enabled_(enabled), gauge_(gauge), start_step_(start_step), end_step_(end_step), dt_(dt), length_cut1_(length_cut1), - length_cut2_(length_cut2), fields_(std::move(fields)), current_step_(-1), active_(false), field_values_(fields_.size(), 0.0) + : enabled_(enabled), gauge_(gauge), start_step_(start_step), end_step_(end_step), dt_ha_(dt), length_cut1_(length_cut1), + length_cut2_(length_cut2), fields_(std::move(fields)), current_step_(-1), active_(false) { - vector_potential_.set(0.0, 0.0, 0.0); - vector_potential_laststep_.set(0.0, 0.0, 0.0); - electric_field_.set(0.0, 0.0, 0.0); - total_electric_field_.set(0.0, 0.0, 0.0); + field_vals_ha_.resize(fields_.size(), 0.0); } -void TDFieldManager::advance_length_gauge() +void TDFieldManager::sample_field(const int step) { - ++current_step_; - active_ = enabled_ && current_step_ >= start_step_ && current_step_ <= end_step_; - std::fill(field_values_.begin(), field_values_.end(), 0.0); - total_electric_field_.set(0.0, 0.0, 0.0); + current_step_ = step; + active_ = enabled_ && step >= start_step_ && step <= end_step_; + std::fill(field_vals_ha_.begin(), field_vals_ha_.end(), 0.0); + efield_ha_.set(0.0, 0.0, 0.0); if (!active_) { return; } - for (std::size_t index = 0; index < fields_.size(); ++index) { const TDField& field = fields_[index]; - const TDFieldSample sample(current_step_, 0, field.subdivisions(), current_step_ * dt_); - // Keep every occurrence for output, but sum repeated directions for - // the physical length-gauge potential and ionic force. - field_values_[index] = field.electric_field(sample); - total_electric_field_[field.direction()] += field_values_[index]; + const double time_ha = step * dt_ha_; + const TDFieldSample sample(step, 0, field.subdivisions(), time_ha); + field_vals_ha_[index] = field.electric_field(sample); + efield_ha_[field.direction()] += field_vals_ha_[index]; } } -void TDFieldManager::advance_vector_gauge() +void TDFieldManager::prepare_sample(const int step) { - ++current_step_; - // Finish the second half of the previous interval before integrating the - // current interval. vector_potential_ therefore remains a midpoint value. - vector_potential_ = vector_potential_ + vector_potential_laststep_ / 2.0; - vector_potential_laststep_.set(0.0, 0.0, 0.0); - electric_field_.set(0.0, 0.0, 0.0); - total_electric_field_.set(0.0, 0.0, 0.0); - std::fill(field_values_.begin(), field_values_.end(), 0.0); - active_ = enabled_ && current_step_ >= start_step_ && current_step_ <= end_step_; - if (!active_) + if (step == current_step_) { return; } + if (step != current_step_ + 1) + { + ModuleBase::WARNING_QUIT("TDFieldManager::prepare_sample", "Invalid electronic-step sequence."); + } + sample_field(step); +} - for (std::size_t index = 0; index < fields_.size(); ++index) +void TDFieldManager::prepare_interval(const int left_step, const int sample_step) +{ + if (left_step == interval_left_ && sample_step == current_step_) { - const TDField& field = fields_[index]; - const int subdivisions = field.subdivisions(); - const double integration_dt = dt_ / subdivisions; - std::vector samples(subdivisions + 1, 0.0); - for (int node = 0; node <= subdivisions; ++node) + return; + } + if (left_step != interval_left_ + 1 || (sample_step != left_step && sample_step != left_step + 1)) + { + ModuleBase::WARNING_QUIT("TDFieldManager::prepare_interval", "Invalid field interval sequence."); + } + A_left_ha_ = A_right_ha_; + if (A_samples_ha_.empty() && enabled_ && left_step >= start_step_ && left_step <= end_step_) + { + for (const TDField& field: fields_) { - const double time = (current_step_ + static_cast(node) / subdivisions) * dt_; - samples[node] = field.electric_field(TDFieldSample(current_step_, node, subdivisions, time)); + const int subdivisions = field.subdivisions(); + const double integration_dt = dt_ha_ / subdivisions; + std::vector samples(subdivisions + 1); + for (int node = 0; node <= subdivisions; ++node) + { + const double time_ha = (left_step + static_cast(node) / subdivisions) * dt_ha_; + const TDFieldSample sample(left_step, node, subdivisions, time_ha); + samples[node] = field.electric_field(sample); + } + double integral = 0.0; + const int count = subdivisions + 1; + ModuleBase::Integral::Simpson_Integral(count, samples.data(), integration_dt, integral); + A_right_ha_[field.direction()] -= integral; } + } + interval_left_ = left_step; + sample_field(sample_step); + select_A_prop(); +} - // Integrate E over [n*dt, (n+1)*dt]. The minus sign implements - // A(t+dt)-A(t) = -integral E(t') dt'. - double integral = 0.0; - ModuleBase::Integral::Simpson_Integral(subdivisions + 1, samples.data(), integration_dt, integral); - vector_potential_laststep_[field.direction()] -= integral; - // Output and the hybrid-gauge scalar potential use E at the interval - // start rather than an average over Simpson nodes. - field_values_[index] = samples.front(); - if (gauge_ == 2) +void TDFieldManager::read_restart(const std::string& file_dir) +{ + std::ifstream file((file_dir + "Restart_td.txt").c_str()); + if (!file) + { + ModuleBase::WARNING_QUIT("TDFieldManager::read_restart", "Cannot open Restart_td.txt."); + } + const std::vector rows = restart_rows(file); + std::istringstream state(rows[0]); + std::istringstream left(rows[1]); + std::istringstream right(rows[2]); + int gauge = -1; + int file_source = -1; + double dt = 0.0; + if (!(state >> current_step_ >> interval_left_ >> gauge >> dt >> file_source) || !(left >> A_left_ha_.x >> A_left_ha_.y >> A_left_ha_.z) + || !(right >> A_right_ha_.x >> A_right_ha_.y >> A_right_ha_.z) || current_step_ < 0 || interval_left_ < -1 + || (gauge_ == 0 && interval_left_ != -1) || (gauge_ != 0 && interval_left_ != current_step_ && interval_left_ != current_step_ - 1) + || file_source != !A_samples_ha_.empty() || gauge != gauge_ || !std::isfinite(dt) || std::abs(dt - dt_ha_) > 1.0e-12 * dt_ha_) + { + ModuleBase::WARNING_QUIT("TDFieldManager::read_restart", "Invalid or incompatible field restart state."); + } + state >> std::ws; + left >> std::ws; + right >> std::ws; + if (!state.eof() || !left.eof() || !right.eof()) + { + ModuleBase::WARNING_QUIT("TDFieldManager::read_restart", "Unexpected extra data in Restart_td.txt."); + } + for (int d = 0; d < 3; ++d) + { + if (!std::isfinite(A_left_ha_[d]) || !std::isfinite(A_right_ha_[d])) { - electric_field_[field.direction()] += samples.front(); + ModuleBase::WARNING_QUIT("TDFieldManager::read_restart", "Non-finite restart vector potential."); } } + sample_field(current_step_); + select_A_prop(); +} - // Advance from the interval endpoint to its midpoint representation. - vector_potential_ = vector_potential_ + vector_potential_laststep_ / 2.0; - if (gauge_ == 2) +void TDFieldManager::write_restart(const std::string& file_dir) const +{ + std::ofstream file((file_dir + "Restart_td.txt").c_str()); + file << std::setprecision(17) << "# Hartree atomic units; step indices start at 0.\n" + << "# gauge: 0=length, 1=velocity, 2=hybrid; source: 0=field, 1=file\n" + << "# step left_step(-1=none) gauge dt source\n" + << current_step_ << " " << interval_left_ << " " << gauge_ << " " << dt_ha_ << " " << !A_samples_ha_.empty() << "\n" + << A_left_ha_.x << " " << A_left_ha_.y << " " << A_left_ha_.z << " # A_left: x y z\n" + << A_right_ha_.x << " " << A_right_ha_.y << " " << A_right_ha_.z << " # A_right: x y z\n"; + if (!file) { - total_electric_field_ = electric_field_; + ModuleBase::WARNING_QUIT("TDFieldManager::write_restart", "Cannot write field restart."); } } -void TDFieldManager::read_restart(const std::string& file_dir) +void TDFieldManager::set_A_samples(const std::vector>& samples_ha) { - std::ifstream file((file_dir + "Restart_td.txt").c_str()); - if (!file) + if (current_step_ != -1 || samples_ha.empty()) + { + ModuleBase::WARNING_QUIT("TDFieldManager::set_A_samples", "Supply nonempty propagation samples before initialization."); + } + for (const ModuleBase::Vector3& A: samples_ha) { - ModuleBase::WARNING_QUIT("TDFieldManager::read_restart", "No Restart_td.txt!"); + for (int d = 0; d < 3; ++d) + { + if (!std::isfinite(A[d])) + { + ModuleBase::WARNING_QUIT("TDFieldManager::set_A_samples", "Non-finite propagation sample."); + } + } } + A_samples_ha_ = samples_ha; +} - int restart_step = -1; - if (!(file >> restart_step >> vector_potential_[0] >> vector_potential_[1] >> vector_potential_[2] >> vector_potential_laststep_[0] - >> vector_potential_laststep_[1] >> vector_potential_laststep_[2])) +void TDFieldManager::select_A_prop() +{ + if (A_samples_ha_.empty()) { - ModuleBase::WARNING_QUIT("TDFieldManager::read_restart", "Invalid Restart_td.txt!"); + A_prop_ha_ = (A_left_ha_ + A_right_ha_) * 0.5; + } + else + { + const std::size_t index = std::min(static_cast(current_step_), A_samples_ha_.size() - 1); + A_prop_ha_ = A_samples_ha_[index]; } - // Retain the legacy restart-file sign convention expected by the first - // half-step update in advance_vector_gauge(). - vector_potential_laststep_ = -vector_potential_laststep_; - current_step_ = restart_step - 1; } int TDFieldManager::gauge() const { return gauge_; } - int TDFieldManager::current_step() const { return current_step_; } - -double TDFieldManager::dt() const +double TDFieldManager::dt_ha() const { - return dt_; + return dt_ha_; } - double TDFieldManager::length_cut1() const { return length_cut1_; } - double TDFieldManager::length_cut2() const { return length_cut2_; } - bool TDFieldManager::active() const { return active_; } - const std::vector& TDFieldManager::fields() const { return fields_; } - -const std::vector& TDFieldManager::field_values() const +const std::vector& TDFieldManager::field_vals_ha() const { - return field_values_; + return field_vals_ha_; } - -const ModuleBase::Vector3& TDFieldManager::vector_potential() const +const ModuleBase::Vector3& TDFieldManager::A_left_ha() const { - return vector_potential_; + if (!A_samples_ha_.empty()) + { + ModuleBase::WARNING_QUIT("TDFieldManager::A_left_ha", "File propagation samples do not define endpoints."); + } + return A_left_ha_; } - -const ModuleBase::Vector3& TDFieldManager::vector_potential_laststep() const +const ModuleBase::Vector3& TDFieldManager::A_right_ha() const { - return vector_potential_laststep_; + if (!A_samples_ha_.empty()) + { + ModuleBase::WARNING_QUIT("TDFieldManager::A_right_ha", "File propagation samples do not define endpoints."); + } + return A_right_ha_; } - -const ModuleBase::Vector3& TDFieldManager::electric_field() const +const ModuleBase::Vector3& TDFieldManager::A_prop_ha() const { - return electric_field_; + return A_prop_ha_; } - -const ModuleBase::Vector3& TDFieldManager::total_electric_field() const +const ModuleBase::Vector3& TDFieldManager::efield_ha() const { - return total_electric_field_; + return efield_ha_; } std::shared_ptr create_td_field_manager(const Input_para& input) diff --git a/source/source_estate/module_pot/td_field_manager.h b/source/source_estate/module_pot/td_field_manager.h index 75bf27ceae5..c7b96ba3f34 100644 --- a/source/source_estate/module_pot/td_field_manager.h +++ b/source/source_estate/module_pot/td_field_manager.h @@ -23,15 +23,23 @@ namespace elecstate class TDFieldManager { public: - /** - * @brief Advance one length-gauge step and sample every field at its start. - */ - void advance_length_gauge(); + /** @brief Sample E at step*dt without integrating A (length gauge or initial state). */ + void prepare_sample(const int step); - /** - * @brief Advance one velocity/hybrid-gauge step using Simpson integration. + /** @brief Integrate [left_step*dt, (left_step+1)*dt] and sample E at sample_step*dt. + * Repeating the same interval and sample is idempotent. Propagation A is the + * endpoint average, not an exact midpoint sample. */ - void advance_vector_gauge(); + void prepare_interval(const int left_step, const int sample_step); + + /** @brief Supply file-based propagation samples before preparing any steps. */ + void set_A_samples(const std::vector>& samples_ha); + + /** @brief Return integrated endpoints in Hartree units; unavailable for file-based propagation samples. */ + const ModuleBase::Vector3& A_left_ha() const; + const ModuleBase::Vector3& A_right_ha() const; + /** @brief Return the endpoint average or supplied file sample used for propagation, in Hartree units. */ + const ModuleBase::Vector3& A_prop_ha() const; /** * @brief Restore the electronic step and vector-potential state. @@ -39,6 +47,8 @@ class TDFieldManager * @param file_dir Directory containing `Restart_td.txt`. */ void read_restart(const std::string& file_dir); + /** @brief Save three annotated data rows describing the interval state in Hartree units. */ + void write_restart(const std::string& file_dir) const; /** @brief Return the spatial-gauge selector supplied by `td_stype`. */ int gauge() const; @@ -46,8 +56,8 @@ class TDFieldManager /** @brief Return the current zero-based electronic-step index. */ int current_step() const; - /** @brief Return the electronic time step in internal atomic time units. */ - double dt() const; + /** @brief Return the electronic time step in Hartree atomic time units. */ + double dt_ha() const; /** @brief Return the first reduced-coordinate cut of the length gauge. */ double length_cut1() const; @@ -62,19 +72,10 @@ class TDFieldManager const std::vector& fields() const; /** @brief Return per-occurrence field samples for the current step. */ - const std::vector& field_values() const; - - /** @brief Return the midpoint vector potential in propagation units. */ - const ModuleBase::Vector3& vector_potential() const; - - /** @brief Return the integrated vector-potential change for this step. */ - const ModuleBase::Vector3& vector_potential_laststep() const; - - /** @brief Return the direction-summed instantaneous hybrid-gauge field. */ - const ModuleBase::Vector3& electric_field() const; + const std::vector& field_vals_ha() const; - /** @brief Return the direction-summed field used by force evaluation. */ - const ModuleBase::Vector3& total_electric_field() const; + /** @brief Return the direction-summed sampled electric field in Hartree atomic units. */ + const ModuleBase::Vector3& efield_ha() const; private: TDFieldManager(bool enabled, @@ -90,17 +91,21 @@ class TDFieldManager int gauge_; int start_step_; int end_step_; - double dt_; + double dt_ha_; double length_cut1_; double length_cut2_; std::vector fields_; int current_step_; bool active_; - std::vector field_values_; - ModuleBase::Vector3 vector_potential_; - ModuleBase::Vector3 vector_potential_laststep_; - ModuleBase::Vector3 electric_field_; - ModuleBase::Vector3 total_electric_field_; + std::vector field_vals_ha_; + int interval_left_ = -1; + std::vector> A_samples_ha_; + ModuleBase::Vector3 A_left_ha_; + ModuleBase::Vector3 A_right_ha_; + ModuleBase::Vector3 A_prop_ha_; + ModuleBase::Vector3 efield_ha_; + void sample_field(const int step); + void select_A_prop(); friend std::shared_ptr create_td_field_manager(const Input_para& input); }; diff --git a/source/source_estate/occ_comput.cpp b/source/source_estate/occ_comput.cpp new file mode 100644 index 00000000000..2a5188a7fa6 --- /dev/null +++ b/source/source_estate/occ_comput.cpp @@ -0,0 +1,103 @@ +#include "occ_comput.h" + +#include "source_base/tool_quit.h" + +namespace elecstate +{ + +void occ_from_proj( + const std::complex* proj, + const double* wg_ik, + const int nbands, + const int npol, + const int nkb, + const int nspin, + const int isk, + const int* nh_iat, + const int nat, + std::complex* occ_block) +{ + if (proj == nullptr) + { + ModuleBase::WARNING_QUIT("occ_from_proj", "proj is nullptr"); + } + if (wg_ik == nullptr) + { + ModuleBase::WARNING_QUIT("occ_from_proj", "wg_ik is nullptr"); + } + if (nh_iat == nullptr) + { + ModuleBase::WARNING_QUIT("occ_from_proj", "nh_iat is nullptr"); + } + if (occ_block == nullptr) + { + ModuleBase::WARNING_QUIT("occ_from_proj", "occ_block is nullptr"); + } + if (nbands <= 0 || nkb <= 0 || nat <= 0) + { + ModuleBase::WARNING_QUIT("occ_from_proj", "nbands, nkb and nat must be positive"); + } + if (npol != 1 && npol != 2) + { + ModuleBase::WARNING_QUIT("occ_from_proj", "npol must be 1 or 2"); + } + if (nspin != 1 && nspin != 2 && nspin != 4) + { + ModuleBase::WARNING_QUIT("occ_from_proj", "nspin must be 1, 2 or 4"); + } + if (nspin == 2 && isk != 0 && isk != 1) + { + ModuleBase::WARNING_QUIT("occ_from_proj", "isk must be 0 or 1 when nspin=2"); + } + if (nspin == 4 && npol != 2) + { + ModuleBase::WARNING_QUIT("occ_from_proj", "nspin=4 requires npol=2"); + } + + // rho^{ss'}_{iprj} = sum_i w_{k,i} * conj(proj^s_{i,iprj}) * proj^{s'}_{i,iprj} + // iprj is the global projector index (begin_iprj + projector within atom) + for (int ib = 0; ib < nbands; ib++) + { + const double weight = wg_ik[ib]; + int begin_iprj = 0; + for (int iat = 0; iat < nat; iat++) + { + const int nprj = nh_iat[iat]; + for (int iprj = 0; iprj < nprj; iprj++) + { + const int occ_index = (begin_iprj + iprj) * 4; + if (npol == 1) + { + const int index = ib * nkb + begin_iprj + iprj; + const double occ = weight * (std::conj(proj[index]) * proj[index]).real(); + if (nspin == 2 && isk == 1) + { + occ_block[occ_index + 3] += occ; + } + else if (nspin == 1) + { + // split evenly so the magnetization readout is zero + occ_block[occ_index] += 0.5 * occ; + occ_block[occ_index + 3] += 0.5 * occ; + } + else + { + occ_block[occ_index] += occ; + } + } + else + { + // spinor components are offset by nkb in the proj layout + const int index = ib * 2 * nkb + begin_iprj + iprj; + occ_block[occ_index] += weight * std::conj(proj[index]) * proj[index]; + occ_block[occ_index + 1] += weight * std::conj(proj[index]) * proj[index + nkb]; + occ_block[occ_index + 2] += weight * std::conj(proj[index + nkb]) * proj[index]; + occ_block[occ_index + 3] += weight * std::conj(proj[index + nkb]) * proj[index + nkb]; + } + } + begin_iprj += nprj; + } + } +} + +} // namespace elecstate diff --git a/source/source_estate/occ_comput.h b/source/source_estate/occ_comput.h new file mode 100644 index 00000000000..fc7622c43dd --- /dev/null +++ b/source/source_estate/occ_comput.h @@ -0,0 +1,62 @@ +#ifndef OCC_COMPUT_H +#define OCC_COMPUT_H + +#include + +namespace elecstate +{ + +/** + * @brief Accumulate per-projector 2x2 occupation blocks from the onsite + * projector coefficients for a single k-point. + * + * For each projector (l,m) (global index iprj), the occupation block is + * + * rho^{ss'}_{iprj} = sum_i w_{k,i} * conj(proj^s_{i,iprj}) * proj^{s'}_{i,iprj} + * + * where proj^s_{i,iprj} = and w_{k,i} is the band + * occupation weight. The block is stored as + * occ_block[iprj*4 + {0,1,2,3}] = {rho^{up,up}, rho^{up,dn}, + * rho^{dn,up}, rho^{dn,dn}}. + * + * The 2x2 layout is a storage convention, not an assumption of spin physics; + * the content of the four slots depends on nspin: + * nspin=1: rho is split evenly into occ[0]/occ[3], off-diagonal slots stay + * zero. No Pauli structure is involved; the even split only makes + * the magnetization readout M ~ occ[0] - occ[3] vanish. + * nspin=2: isk=0 accumulates into occ[0], isk=1 into occ[3]; the two + * diagonal slots are simply two independent spin channels. + * nspin=4: both spinor components cross, filling all four slots with a + * genuine 2x2 Hermitian spin density matrix (the only case where + * the Pauli decomposition rho = (n*I + M.sigma)/2 applies). + * + * Accumulation semantics (+=): the caller zeroes occ_block before the + * k-point loop and performs the MPI reduction after it. + * + * @param proj onsite projector coefficients , laid out as + * (nbands*npol) x nkb with spinor components offset by nkb + * @param wg_ik band occupation weights for this k-point (nbands values) + * @param nbands number of bands + * @param npol number of spinor components (1 collinear, 2 non-collinear) + * @param nkb total number of projectors + * @param nspin 1, 2 or 4 + * @param isk spin channel of this k-point (0/1); used only when nspin=2 + * @param nh_iat number of projectors per atom (nat values) + * @param nat number of atoms + * @param occ_block [in,out] blocks to accumulate into, size sum(nh_iat)*4 + */ +void occ_from_proj( + const std::complex* proj, + const double* wg_ik, + const int nbands, + const int npol, + const int nkb, + const int nspin, + const int isk, + const int* nh_iat, + const int nat, + std::complex* occ_block); + +} // namespace elecstate + +#endif // OCC_COMPUT_H diff --git a/source/source_estate/occ_matrix.cpp b/source/source_estate/occ_matrix.cpp index 61094312314..349d4a896f9 100644 --- a/source/source_estate/occ_matrix.cpp +++ b/source/source_estate/occ_matrix.cpp @@ -13,7 +13,7 @@ void OccupationMatrix::init(const UnitCell& cell, this->occ_.resize(cell.nat); this->occ_save_.resize(cell.nat); - this->iatlnmipol2iwt_.resize(cell.nat); + this->corr_iwt_.resize(cell.nat); for (int it = 0; it < cell.ntype; ++it) { @@ -23,56 +23,43 @@ void OccupationMatrix::init(const UnitCell& cell, occ_[iat].resize(cell.atoms[it].nwl + 1); occ_save_[iat].resize(cell.atoms[it].nwl + 1); - iatlnmipol2iwt_[iat].resize(cell.atoms[it].nwl + 1); + corr_iwt_[iat].resize(cell.atoms[it].nwl + 1); if (l_channel[it] == -1) { continue; } + // only the first radial channel (n=0) of each l is stored for (int l = 0; l <= cell.atoms[it].nwl; l++) { - const int N = cell.atoms[it].l_nchi[l]; - - occ_[iat][l].resize(N); - occ_save_[iat][l].resize(N); - - for (int n = 0; n < N; n++) + if (nspin == 1 || nspin == 2) { - if (nspin == 1 || nspin == 2) - { - occ_[iat][l][n].resize(2); - occ_save_[iat][l][n].resize(2); + occ_[iat][l].resize(2); + occ_save_[iat][l].resize(2); - occ_[iat][l][n][0].create(2 * l + 1, 2 * l + 1); - occ_[iat][l][n][1].create(2 * l + 1, 2 * l + 1); + occ_[iat][l][0].create(2 * l + 1, 2 * l + 1); + occ_[iat][l][1].create(2 * l + 1, 2 * l + 1); - occ_save_[iat][l][n][0].create(2 * l + 1, 2 * l + 1); - occ_save_[iat][l][n][1].create(2 * l + 1, 2 * l + 1); - } - else if (nspin == 4) - { - occ_[iat][l][n].resize(1); - occ_save_[iat][l][n].resize(1); + occ_save_[iat][l][0].create(2 * l + 1, 2 * l + 1); + occ_save_[iat][l][1].create(2 * l + 1, 2 * l + 1); + } + else if (nspin == 4) + { + occ_[iat][l].resize(1); + occ_save_[iat][l].resize(1); - occ_[iat][l][n][0].create((2 * l + 1) * npol, (2 * l + 1) * npol); - occ_save_[iat][l][n][0].create((2 * l + 1) * npol, (2 * l + 1) * npol); - } + occ_[iat][l][0].create((2 * l + 1) * npol, (2 * l + 1) * npol); + occ_save_[iat][l][0].create((2 * l + 1) * npol, (2 * l + 1) * npol); } } for (int L = 0; L <= cell.atoms[it].nwl; L++) { - iatlnmipol2iwt_[iat][L].resize(cell.atoms[it].l_nchi[L]); - - for (int n = 0; n < cell.atoms[it].l_nchi[L]; n++) + corr_iwt_[iat][L].resize(2 * L + 1); + for (int m = 0; m < 2 * L + 1; m++) { - iatlnmipol2iwt_[iat][L][n].resize(2 * L + 1); - - for (int m = 0; m < 2 * L + 1; m++) - { - iatlnmipol2iwt_[iat][L][n][m].resize(npol); - } + corr_iwt_[iat][L][m].resize(npol); } } @@ -85,7 +72,10 @@ void OccupationMatrix::init(const UnitCell& cell, const int n = cell.atoms[it].iw2n[iw0]; const int m = cell.atoms[it].iw2m[iw0]; - iatlnmipol2iwt_[iat][l][n][m][ipol] = iwt; + if (n == 0) + { + corr_iwt_[iat][l][m][ipol] = iwt; + } } } } @@ -101,7 +91,7 @@ void OccupationMatrix::get_flat(const int iat, const int l, std::vector& { for (int i = 0; i < size; i++) { - occ[is * size + i] = occ_[iat][l][0][is].c[i]; + occ[is * size + i] = occ_[iat][l][is].c[i]; } } } @@ -109,7 +99,7 @@ void OccupationMatrix::get_flat(const int iat, const int l, std::vector& { for (int i = 0; i < static_cast(occ.size()); i++) { - occ[i] = occ_[iat][l][0][0].c[i]; + occ[i] = occ_[iat][l][0].c[i]; } } } @@ -119,7 +109,7 @@ void OccupationMatrix::set_flat(const int iat, const int l, const int spin, { for (int i = 0; i < static_cast(occ.size()); i++) { - occ_[iat][l][0][spin].c[i] = occ[i]; + occ_[iat][l][spin].c[i] = occ[i]; } } @@ -138,19 +128,14 @@ void OccupationMatrix::zero(const UnitCell& cell, const std::vector& l_chan for (int l = 0; l < cell.atoms[T].nwl + 1; l++) { - const int N = cell.atoms[T].l_nchi[l]; - - for (int n = 0; n < N; n++) + if (nspin_ == 4) { - if (nspin_ == 4) - { - occ_[iat][l][n][0].zero_out(); - } - else if (nspin_ == 1 || nspin_ == 2) - { - occ_[iat][l][n][0].zero_out(); - occ_[iat][l][n][1].zero_out(); - } + occ_[iat][l][0].zero_out(); + } + else if (nspin_ == 1 || nspin_ == 2) + { + occ_[iat][l][0].zero_out(); + occ_[iat][l][1].zero_out(); } } } @@ -176,12 +161,12 @@ void OccupationMatrix::copy_to_save(const UnitCell& cell, const std::vector if (nspin_ == 4) { - occ_save_[iat][target_l][0][0] = occ_[iat][target_l][0][0]; + occ_save_[iat][target_l][0] = occ_[iat][target_l][0]; } else if (nspin_ == 1 || nspin_ == 2) { - occ_save_[iat][target_l][0][0] = occ_[iat][target_l][0][0]; - occ_save_[iat][target_l][0][1] = occ_[iat][target_l][0][1]; + occ_save_[iat][target_l][0] = occ_[iat][target_l][0]; + occ_save_[iat][target_l][1] = occ_[iat][target_l][1]; } } } @@ -209,14 +194,14 @@ void OccupationMatrix::write_to_flat(const UnitCell& cell, for (int mm = 0; mm < size; mm++) { - uom[index[iat] + mm] = occ_[iat][target_l][0][0].c[mm]; + uom[index[iat] + mm] = occ_[iat][target_l][0].c[mm]; } if (nspin_ == 2) { const int half_size = uom.size() / 2; for (int mm = 0; mm < size; mm++) { - uom[half_size + index[iat] + mm] = occ_[iat][target_l][0][1].c[mm]; + uom[half_size + index[iat] + mm] = occ_[iat][target_l][1].c[mm]; } } } @@ -239,20 +224,20 @@ void OccupationMatrix::read_from_flat(const UnitCell& cell, const int iat = cell.itia2iat(T, I); if (nspin_ == 4) { - for (int mm = 0; mm < occ_[iat][l][0][0].nr * occ_[iat][l][0][0].nc; mm++) + for (int mm = 0; mm < occ_[iat][l][0].nr * occ_[iat][l][0].nc; mm++) { - occ_[iat][l][0][0].c[mm] = uom[index[iat] + mm]; + occ_[iat][l][0].c[mm] = uom[index[iat] + mm]; } } else if (nspin_ == 1 || nspin_ == 2) { const int half_size = uom.size() / 2; - for (int mm = 0; mm < occ_[iat][l][0][0].nr * occ_[iat][l][0][0].nc; mm++) + for (int mm = 0; mm < occ_[iat][l][0].nr * occ_[iat][l][0].nc; mm++) { - occ_[iat][l][0][0].c[mm] = uom[index[iat] + mm]; + occ_[iat][l][0].c[mm] = uom[index[iat] + mm]; if (nspin_ == 2) { - occ_[iat][l][0][1].c[mm] = uom[half_size + index[iat] + mm]; + occ_[iat][l][1].c[mm] = uom[half_size + index[iat] + mm]; } } } @@ -280,27 +265,27 @@ void OccupationMatrix::write_save_to_flat(const UnitCell& cell, for (int I = 0; I < cell.atoms[T].na; I++) { const int iat = cell.itia2iat(T, I); - const int size = occ_save_[iat][target_l][0][0].nr * occ_save_[iat][target_l][0][0].nc; + const int size = occ_save_[iat][target_l][0].nr * occ_save_[iat][target_l][0].nc; if (nspin_ == 4) { for (int mm = 0; mm < size; mm++) { - uom_save[index[iat] + mm] = occ_save_[iat][target_l][0][0].c[mm]; + uom_save[index[iat] + mm] = occ_save_[iat][target_l][0].c[mm]; } } else if (nspin_ == 1 || nspin_ == 2) { for (int mm = 0; mm < size; mm++) { - uom_save[index[iat] + mm] = occ_save_[iat][target_l][0][0].c[mm]; + uom_save[index[iat] + mm] = occ_save_[iat][target_l][0].c[mm]; } if (nspin_ == 2) { const int half_size = uom_save.size() / 2; for (int mm = 0; mm < size; mm++) { - uom_save[half_size + index[iat] + mm] = occ_save_[iat][target_l][0][1].c[mm]; + uom_save[half_size + index[iat] + mm] = occ_save_[iat][target_l][1].c[mm]; } } } @@ -315,8 +300,8 @@ namespace elecstate /// of every atom. nspin-aware: nspin=4 mixes the single Pauli block, /// nspin=1/2 mixes both spin channels. Replaces the duplicated LCAO /// k/gamma mixing loops. -void mix_occ_with_save(std::vector>>>& occ_mat, - const std::vector>>>& occ_mat_save, +void mix_occ_with_save(std::vector>>& occ_mat, + const std::vector>>& occ_mat_save, const UnitCell& cell, const std::vector& l_channel, const int nspin, @@ -335,8 +320,8 @@ void mix_occ_with_save(std::vector(l,n,m,ipol)->iwt lookup table. + * Owns the nested occ[iat][l][spin] matrices together with their saved + * copy (used by mixing) and the corr_iwt[iat][l][m][ipol] lookup table + * mapping the correlated orbital (first radial channel) to its global + * orbital index iwt. * Layout: - * nspin=1/2: occ[iat][l][n] has 2 spin channels of (2l+1)x(2l+1) - * nspin=4: occ[iat][l][n] has 1 channel of (2l+1)*npol x (2l+1)*npol + * nspin=1/2: occ[iat][l] has 2 spin channels of (2l+1)x(2l+1) + * nspin=4: occ[iat][l] has 1 channel of (2l+1)*npol x (2l+1)*npol * (all Pauli blocks packed together) */ class OccupationMatrix { public: - /// allocate occ/occ_save/iatlnmipol2iwt according to the cell + /// allocate occ/occ_save/corr_iwt according to the cell void init(const UnitCell& cell, const std::vector& l_channel, int nspin, int npol); // --- element access --- - double get(int iat, int l, int n, int spin, int m1, int m2) const + double get(int iat, int l, int spin, int m1, int m2) const { - return occ_[iat][l][n][spin](m1, m2); + return occ_[iat][l][spin](m1, m2); } - double get_save(int iat, int l, int n, int spin, int m1, int m2) const + double get_save(int iat, int l, int spin, int m1, int m2) const { - return occ_save_[iat][l][n][spin](m1, m2); + return occ_save_[iat][l][spin](m1, m2); } - void set(int iat, int l, int n, int spin, int m1, int m2, double val) + void set(int iat, int l, int spin, int m1, int m2, double val) { - occ_[iat][l][n][spin](m1, m2) = val; + occ_[iat][l][spin](m1, m2) = val; } /// direct matrix access for kernels that operate on whole blocks - ModuleBase::matrix& mat(int iat, int l, int n, int spin) + ModuleBase::matrix& mat(int iat, int l, int spin) { - return occ_[iat][l][n][spin]; + return occ_[iat][l][spin]; } - const ModuleBase::matrix& mat(int iat, int l, int n, int spin) const + const ModuleBase::matrix& mat(int iat, int l, int spin) const { - return occ_[iat][l][n][spin]; + return occ_[iat][l][spin]; } - ModuleBase::matrix& mat_save(int iat, int l, int n, int spin) + ModuleBase::matrix& mat_save(int iat, int l, int spin) { - return occ_save_[iat][l][n][spin]; + return occ_save_[iat][l][spin]; } - const ModuleBase::matrix& mat_save(int iat, int l, int n, int spin) const + const ModuleBase::matrix& mat_save(int iat, int l, int spin) const { - return occ_save_[iat][l][n][spin]; + return occ_save_[iat][l][spin]; } // --- bulk data access (used by IO and legacy call sites) --- - std::vector>>>& data() { return occ_; } - const std::vector>>>& data() const { return occ_; } - std::vector>>>& data_save() { return occ_save_; } - const std::vector>>>& data_save() const { return occ_save_; } + std::vector>>& data() { return occ_; } + const std::vector>>& data() const { return occ_; } + std::vector>>& data_save() { return occ_save_; } + const std::vector>>& data_save() const { return occ_save_; } // --- lookup table --- - int iwt(int iat, int l, int n, int m, int ipol) const + /// global orbital index of the correlated orbital (first radial + /// channel) of atom iat. Only the correlated channel is stored. + int corr_iwt(int iat, int l, int m, int ipol) const { - return iatlnmipol2iwt_[iat][l][n][m][ipol]; - } - const std::vector>>>>& iatlnmipol2iwt() const - { - return iatlnmipol2iwt_; + return corr_iwt_[iat][l][m][ipol]; } // --- flat (de)serialization of one atom's correlated orbital --- - /// nspin=1: fills occ with occ[iat][l][0][0] data + /// nspin=1: fills occ with occ[iat][l][0] data /// nspin=2: fills occ with interleaved spin-up then spin-down data - /// nspin=4: fills occ with occ[iat][l][0][0] data (all Pauli blocks) + /// nspin=4: fills occ with occ[iat][l][0] data (all Pauli blocks) void get_flat(int iat, int l, std::vector& occ) const; void set_flat(int iat, int l, int spin, const std::vector& occ); @@ -106,9 +106,9 @@ class OccupationMatrix int npol() const { return npol_; } private: - std::vector>>> occ_; - std::vector>>> occ_save_; - std::vector>>>> iatlnmipol2iwt_; + std::vector>> occ_; + std::vector>> occ_save_; + std::vector>>> corr_iwt_; int nspin_ = 0; int npol_ = 0; }; @@ -118,8 +118,8 @@ namespace elecstate /// occ = beta * occ + (1-beta) * occ_save on every atom's correlated orbital. /// nspin-aware: nspin=4 mixes the single Pauli block, nspin=1/2 mixes both /// spin channels. Replaces the duplicated LCAO k/gamma mixing loops. -void mix_occ_with_save(std::vector>>>& occ_mat, - const std::vector>>>& occ_mat_save, +void mix_occ_with_save(std::vector>>& occ_mat, + const std::vector>>& occ_mat_save, const UnitCell& cell, const std::vector& l_channel, const int nspin, diff --git a/source/source_estate/occupy.cpp b/source/source_estate/occupy.cpp index 0a492735c6d..8fd2898b9c2 100644 --- a/source/source_estate/occupy.cpp +++ b/source/source_estate/occupy.cpp @@ -81,7 +81,7 @@ void Occupy::decision(const std::string& name, const std::string& smearing_metho } else if (smearing_method == "mp3") { - // acually any order Methfessel-Paxton method can be supported in Occupy::w1gauss() + // acually any order Methfessel-Paxton method can be supported in occupy_smearing::w1gauss() // however the parameter is string instead of int ModuleBase::WARNING_QUIT( "occupy", @@ -128,6 +128,8 @@ void Occupy::decision(const std::string& name, const std::string& smearing_metho * @param wg output: weight for each k, each band. * @param is the spin index now. * @param isk distinguish k point belong to which spin. + * @param nspin number of spin channels; 2 selects a single spin through isk when + * is != -1, and 4 halves the spin degeneracy. */ void Occupy::iweights( const int nks, @@ -139,11 +141,12 @@ void Occupy::iweights( double& ef, ModuleBase::matrix& wg, const int& is, //<- is should be -1, 0, or 1. -1 means set all spins, and 0 means spin up, 1 means spin down. - const std::vector& isk) + const std::vector& isk, + const int nspin) { assert(is < 2); double degspin = 2.0; - if (PARAM.inp.nspin == 4) { + if (nspin == 4) { degspin = 1.0; } if (is != -1) { @@ -163,7 +166,7 @@ void Occupy::iweights( for (int ik = 0; ik < nks; ++ik) { // when NSPIN=2, only calculate spin up or spin down with TWO_FERMI mode(nupdown != 0) - if (PARAM.inp.nspin == 2 && isk[ik] != is && is != -1) + if (nspin == 2 && isk[ik] != is && is != -1) { continue; } @@ -222,7 +225,7 @@ void Occupy::gweights(const int nks, // Calculate the Fermi energy ef //=============================== // call efermig - Occupy::efermig(ekb, nband, nks, nelec, wk, smearing_sigma, ngauss, ef, is, isk); + occupy_smearing::efermig(ekb, nband, nks, nelec, wk, smearing_sigma, ngauss, ef, is, isk); demet = 0.0; for (int ik = 0; ik < nks; ik++) @@ -238,7 +241,7 @@ void Occupy::gweights(const int nks, // Calculate the gaussian weights //================================ // call wgauss - wg(ik, ib) = wk[ik] * Occupy::wgauss((ef - ekb(ik, ib)) / smearing_sigma, ngauss); + wg(ik, ib) = wk[ik] * occupy_smearing::wgauss((ef - ekb(ik, ib)) / smearing_sigma, ngauss); //==================================================================== // The correct form of the band energy is \int e n(e) de for e 1000) { @@ -335,7 +338,7 @@ void Occupy::efermig(const ModuleBase::matrix& ekb, std::cout << " sumklw = " << sumklw << std::endl; std::cout << " sumkup - nelec = " << sumkup - nelec << std::endl; std::cout << " sumklw - nelec = " << sumklw - nelec << std::endl; - ModuleBase::WARNING_QUIT("Occupy::efermig", "ERROS in SMEARING"); + ModuleBase::WARNING_QUIT("occupy_smearing::efermig", "ERROS in SMEARING"); // no need to break; quit directly } else if ((sumkup - nelec) < -eps) @@ -360,7 +363,7 @@ void Occupy::efermig(const ModuleBase::matrix& ekb, // change ef value //====================== ef = (eup + elw) / 2.0; - const double sumkmid = sumkg(ekb, nband, nks, wk, smearing_sigma, ngauss, ef, is, isk); + const double sumkmid = occupy_smearing::sumkg(ekb, nband, nks, wk, smearing_sigma, ngauss, ef, is, isk); if (std::abs(sumkmid - nelec) < eps) { @@ -392,7 +395,7 @@ void Occupy::efermig(const ModuleBase::matrix& ekb, * @param isk array to point out each k belong to which spin * @return (double) the number of states */ -double Occupy::sumkg(const ModuleBase::matrix& ekb, +double occupy_smearing::sumkg(const ModuleBase::matrix& ekb, const int nband, const int nks, const std::vector& wk, @@ -416,7 +419,7 @@ double Occupy::sumkg(const ModuleBase::matrix& ekb, //=========================== // call wgauss //=========================== - sum1 += Occupy::wgauss((e - ekb(ik, ib)) / smearing_sigma, ngauss); + sum1 += occupy_smearing::wgauss((e - ekb(ik, ib)) / smearing_sigma, ngauss); } sum2 += wk[ik] * sum1; } @@ -433,7 +436,7 @@ double Occupy::sumkg(const ModuleBase::matrix& ekb, return sum2; } -double Occupy::wgauss(const double& x, const int n) +double occupy_smearing::wgauss(const double& x, const int n) { // ModuleBase::TITLE("Occupy","wgauss"); //===================================================================== @@ -515,7 +518,7 @@ double Occupy::wgauss(const double& x, const int n) return wga; } // end function wgauss -double Occupy::w1gauss(const double& x, const int n) +double occupy_smearing::w1gauss(const double& x, const int n) { //======================================================================== // w1gauss(x,n) = \int_{-\infty}^x y delta(y) dy diff --git a/source/source_estate/occupy.h b/source/source_estate/occupy.h index 0a9f7e585ce..abb30b8d26c 100644 --- a/source/source_estate/occupy.h +++ b/source/source_estate/occupy.h @@ -5,6 +5,50 @@ #include "source_base/matrix.h" #include "source_base/vector3.h" +/** + * @brief Smearing weight functions and the Fermi-level bisection built on them. + * + * Each of these depends only on its arguments -- none of them reads any of + * Occupy's state -- so they are free functions rather than private static + * members of that class. + */ +namespace occupy_smearing +{ +/** + * @brief smeared occupation of a state, as a function of (ef - e) / sigma + * @param n smearing type: -99 Fermi-Dirac, -1 cold, 0 Gaussian, + * n > 0 n-th order Methfessel-Paxton + */ +double wgauss(const double& x, const int n); + +/// @brief first moment of the smearing function, used for the -TS term +/// @param n smearing type, as in wgauss() +double w1gauss(const double& x, const int n); + +/// @brief number of electrons below a trial Fermi level e +double sumkg(const ModuleBase::matrix& ekb, + const int nband, + const int nks, + const std::vector& wk, + const double& smearing_sigma, + const int ngauss, + const double& e, + const int& is, + const std::vector& isk); + +/// @brief find the Fermi level by bisection on sumkg() +void efermig(const ModuleBase::matrix& ekb, + const int nbnd, + const int nks, + const double& nelec, + const std::vector& wk, + const double& smearing_sigma, + const int ngauss, + double& ef, + const int& is, + const std::vector& isk); +} // namespace occupy_smearing + class Occupy { @@ -35,6 +79,8 @@ class Occupy // fixed occupations static bool fixed_occupations; + /// @param nspin number of spin channels; 2 selects a single spin through + /// isk when is != -1, and 4 halves the spin degeneracy static void iweights(const int nks, const std::vector& wk, const int nband, @@ -44,7 +90,8 @@ class Occupy double& ef, ModuleBase::matrix& wg, const int& is, - const std::vector& isk); + const std::vector& isk, + const int nspin); static void gweights(const int nks, const std::vector& wk, @@ -66,31 +113,6 @@ class Occupy static double wsweight(const ModuleBase::Vector3 &r, ModuleBase::Vector3 *rws,const int nrws); private: - static void efermig(const ModuleBase::matrix& ekb, - const int nbnd, - const int nks, - const double& nelec, - const std::vector& wk, - const double& smearing_sigma, - const int ngauss, - double& ef, - const int& is, - const std::vector& isk); - - static double sumkg(const ModuleBase::matrix& ekb, - const int nband, - const int nks, - const std::vector& wk, - const double& smearing_sigma, - const int ngauss, - const double& e, - const int& is, - const std::vector& isk); - - static double wgauss(const double& x, const int n); - - static double w1gauss(const double& x, const int n); - //============================ // Needed in tweights //============================ diff --git a/source/source_estate/rhog_io.cpp b/source/source_estate/rhog_io.cpp deleted file mode 100644 index 3412a20133d..00000000000 --- a/source/source_estate/rhog_io.cpp +++ /dev/null @@ -1,374 +0,0 @@ -#include "source_base/module_out/binstream.h" -#include "source_base/vector3.h" -#include "source_base/module_parallel/para_mpi_func.h" -#include "rhog_io.h" -#include -#include -#include - -namespace -{ -inline void warn(std::ostream* os, - const Parallel::ParaWorld& pw_world, - const std::string& file, - const std::string& desc) -{ - if (pw_world.rank() == 0 && os != nullptr) - { - *os << " " << file << " warning : " << desc << std::endl; - } -} -} // namespace - -bool elecstate::read_rhog(const std::string& filename, - const ModulePW::PW_Basis* pw_rhod, - const int nspin, - std::complex** rhog, - const Parallel::ParaWorld& pw_world, - std::ostream* os_warning) -{ - if (pw_rhod == nullptr) - { - warn(os_warning, pw_world, "elecstate::read_rhog", "pw_rhod is null"); - return false; - } - if (rhog == nullptr) - { - warn(os_warning, pw_world, "elecstate::read_rhog", "rhog is null"); - return false; - } - if (nspin != 1 && nspin != 2 && nspin != 4) - { - warn(os_warning, pw_world, "elecstate::read_rhog", "nspin must be 1, 2, or 4"); - return false; - } - if (pw_rhod->nx <= 0 || pw_rhod->ny <= 0 || pw_rhod->nz <= 0) - { - warn(os_warning, pw_world, "elecstate::read_rhog", "PW_Basis grid dimensions must be positive"); - return false; - } - - const int nx = pw_rhod->nx; - const int ny = pw_rhod->ny; - const int nz = pw_rhod->nz; - - Binstream ifs; - bool error = false; - int gamma_only_in = 0; - int npwtot_in = 0; - int nspin_in = 0; - int size = 0; - double b1[3], b2[3], b3[3]; - - if (pw_world.rank() == 0) - { - ifs.open(filename, "r"); - if (!ifs) - { - error = true; - } - } - - Parallel::bcast_bool(error, pw_world); - - if (error) - { - warn(os_warning, pw_world, "elecstate::read_rhog", "Can't open file " + filename); - return false; - } - - if (pw_world.rank() == 0) - { - ifs >> size >> gamma_only_in >> npwtot_in >> nspin_in >> size; - ifs >> size >> b1[0] >> b1[1] >> b1[2] >> b2[0] >> b2[1] >> b2[2] >> b3[0] >> b3[1] >> b3[2] >> size; - if (gamma_only_in != pw_rhod->gamma_only) - { - // there is a treatment that can transform between gamma_only and non-gamma_only - // however, it is not implemented here - error = true; - ifs.close(); - } - if (npwtot_in > pw_rhod->npwtot) - { - warn(os_warning, pw_world, "elecstate::read_rhog", "some planewaves in file are not used"); - } - else if (npwtot_in < pw_rhod->npwtot) - { - warn(os_warning, pw_world, "elecstate::read_rhog", "some planewaves in file are missing"); - } - if (nspin_in < nspin) - { - warn(os_warning, pw_world, "elecstate::read_rhog", "some spin channels in file are missing"); - } - } - - Parallel::bcast_bool(error, pw_world); - - if (error) - { - warn(os_warning, pw_world, "elecstate::read_rhog", "gamma_only read from file is inconsistent with INPUT"); - return false; - } - - Parallel::bcast_int(gamma_only_in, pw_world); - Parallel::bcast_int(npwtot_in, pw_world); - Parallel::bcast_int(nspin_in, pw_world); - Parallel::bcast_double(b1, 3, pw_world); - Parallel::bcast_double(b2, 3, pw_world); - Parallel::bcast_double(b3, 3, pw_world); - - std::vector miller(npwtot_in * 3); - // once use ModuleBase::Vector3, it is highly bug-prone to assume the memory layout of the class. - // The x, y and z of Vector3 will not always to be contiguous. - // Instead, a relatively safe choice is to use std::vector, the memory layout is assumed - // to be npwtot_in rows and 3 columns. - if (pw_world.rank() == 0) - { - ifs >> size; - for (int i = 0; i < npwtot_in; ++i) // loop over rows... - { - ifs >> miller[i*3] >> miller[i*3+1] >> miller[i*3+2]; - } - ifs >> size; - } - Parallel::bcast_int(miller.data(), miller.size(), pw_world); - - // set to zero - for (int is = 0; is < nspin; ++is) - { - std::fill(rhog[is], rhog[is] + pw_rhod->npw, std::complex(0.0, 0.0)); - } - // maps ixyz tp ig - std::vector fftixyz2ig(pw_rhod->nxyz, -1); // map isz to ig. - for (int ig = 0; ig < pw_rhod->npw; ++ig) - { - int isz = pw_rhod->ig2isz[ig]; - int iz = isz % nz; - int is = isz / nz; - int ixy = pw_rhod->is2fftixy[is]; - int ixyz = iz + nz * ixy; - fftixyz2ig[ixyz] = ig; - } - std::vector> rhog_in(npwtot_in); - for (int is = 0; is < nspin_in; ++is) - { - if (pw_world.rank() == 0) - { - ifs >> size; - for (int i = 0; i < npwtot_in; ++i) - { - ifs >> rhog_in[i]; - } - ifs >> size; - } - Parallel::bcast_complex(rhog_in.data(), rhog_in.size(), pw_world); - - for (int i = 0; i < npwtot_in; ++i) - { - int ix = miller[i * 3]; - int iy = miller[i * 3 + 1]; - int iz = miller[i * 3 + 2]; - - if (ix <= -int((nx + 1) / 2) || ix >= int(nx / 2) + 1 || iy <= -int((ny + 1) / 2) || iy >= int(ny / 2) + 1 - || iz <= -int((nz + 1) / 2) || iz >= int(nz / 2) + 1) - { - // these planewaves are not used - continue; - } - - if (ix < 0) - ix += nx; - if (iy < 0) - iy += ny; - if (iz < 0) - iz += nz; - int fftixy = iy + pw_rhod->fftny * ix; - if (pw_world.rank() == pw_rhod->fftixy2ip[fftixy]) - { - int fftixyz = iz + nz * fftixy; - int ig = fftixyz2ig[fftixyz]; - rhog[is][ig] = rhog_in[i]; - } - } - - if (nspin_in == 2 && nspin == 4 && is == 1) - { - for (int ig = 0; ig < pw_rhod->npw; ++ig) - { - rhog[3][ig] = rhog[1][ig]; - } - std::fill(rhog[1], rhog[1] + pw_rhod->npw, std::complex(0.0, 0.0)); - std::fill(rhog[2], rhog[2] + pw_rhod->npw, std::complex(0.0, 0.0)); - } - } - - if (pw_world.rank() == 0) - { - ifs.close(); - } - return true; -} - -bool elecstate::write_rhog(const std::string& fchg, - const bool gamma_only, - const ModulePW::PW_Basis* pw_rho, - const int nspin, - const ModuleBase::Matrix3& GT, - std::complex** rhog, - const Parallel::ParaWorld& pw_world, - std::ostream* os_warning) -{ - if (pw_rho == nullptr) - { - warn(os_warning, pw_world, "elecstate::write_rhog", "pw_rho is null"); - return false; - } - if (rhog == nullptr) - { - warn(os_warning, pw_world, "elecstate::write_rhog", "rhog is null"); - return false; - } - if (nspin != 1 && nspin != 2 && nspin != 4) - { - warn(os_warning, pw_world, "elecstate::write_rhog", "nspin must be 1, 2, or 4"); - return false; - } - - // only rank 0 in the domain writes the header; all ranks cooperate - // on sequential writes synchronized by barriers. - const int irank = pw_world.rank(); - const int nrank = pw_world.size(); - - // write the header (by rank 0): gamma_only, ngm_g, nspin - int size = 3; - int ngm_g = pw_rho->npwtot; - int gam = gamma_only; - int nsp = nspin; - - std::ofstream ofs; - Parallel::barrier(pw_world); - - if (irank == 0) - { - ofs.open(fchg, std::ios::binary); - if (!ofs) - { - warn(os_warning, pw_world, "elecstate::write_rhog", "File I/O failure: cannot open file " + fchg); - return false; - } - ofs.write(reinterpret_cast(&size), sizeof(size)); - ofs.write(reinterpret_cast(&gam), sizeof(gam)); - ofs.write(reinterpret_cast(&ngm_g), sizeof(ngm_g)); - ofs.write(reinterpret_cast(&nsp), sizeof(nsp)); - ofs.write(reinterpret_cast(&size), sizeof(size)); - // write the lattice vectors - std::vector b = {GT.e11, GT.e12, GT.e13, GT.e21, GT.e22, GT.e23, GT.e31, GT.e32, GT.e33}; - size = 9; - ofs.write(reinterpret_cast(&size), sizeof(size)); - for (int i = 0; i < 9; ++i) - { - ofs.write(reinterpret_cast(&b[i]), sizeof(b[i])); - } - ofs.write(reinterpret_cast(&size), sizeof(size)); - ofs.close(); - } - Parallel::barrier(pw_world); - Parallel::barrier(pw_world); - - // write the G-vectors in Miller indices - size = 3 * ngm_g; - if (irank == 0) - { - ofs.open(fchg, std::ios::binary | std::ios::app); - ofs.write(reinterpret_cast(&size), sizeof(size)); - ofs.close(); - } - Parallel::barrier(pw_world); - - for (int i = 0; i < nrank; ++i) - { - if (i == irank) - { - ofs.open(fchg, std::ios::binary | std::ios::app); - for (int ig = 0; ig < pw_rho->npw; ++ig) - { - const ModuleBase::Vector3 g = pw_rho->gdirect[ig]; - std::vector miller = {int(g.x), int(g.y), int(g.z)}; - ofs.write(reinterpret_cast(&miller[0]), sizeof(miller[0])); - ofs.write(reinterpret_cast(&miller[1]), sizeof(miller[1])); - ofs.write(reinterpret_cast(&miller[2]), sizeof(miller[2])); - } - ofs.close(); - } - Parallel::barrier(pw_world); - } - - if (irank == 0) - { - ofs.open(fchg, std::ios::binary | std::ios::app); - ofs.write(reinterpret_cast(&size), sizeof(size)); - ofs.close(); - } - Parallel::barrier(pw_world); - - // write the rho(G) values - std::complex sum_check; - size = ngm_g; - for (int ispin = 0; ispin < nspin; ++ispin) - { - if (irank == 0) - { - ofs.open(fchg, std::ios::binary | std::ios::app); - ofs.write(reinterpret_cast(&size), sizeof(size)); - ofs.close(); - } - Parallel::barrier(pw_world); - - for (int i = 0; i < nrank; ++i) - { - if (i == irank) - { - ofs.open(fchg, std::ios::binary | std::ios::app); - sum_check = 0.0; - for (int ig = 0; ig < pw_rho->npw; ++ig) - { - sum_check += rhog[ispin][ig]; - ofs.write(reinterpret_cast(&rhog[ispin][ig]), sizeof(rhog[ispin][ig])); - } - ofs.close(); - } - Parallel::barrier(pw_world); - } - - if (irank == 0) - { - ofs.open(fchg, std::ios::binary | std::ios::app); - ofs.write(reinterpret_cast(&size), sizeof(size)); - ofs.close(); - } - Parallel::barrier(pw_world); - } - return true; -} - -// self-consistency test with the following python code -// import numpy as np - -// with open("rhog_read.txt") as f: -// read = f.readlines() - -// with open("rhog_write.txt") as f: -// write = f.readlines() - -// # convert c++ stype complex number (a,b) to python complex -// def to_complex(s): -// a, b = s.replace("(", "").replace(")", "").split(",") -// return complex(float(a), float(b)) - -// read = [[to_complex(rhog) for rhog in spin.strip().split()] for spin in read] -// write = [[to_complex(rhog) for rhog in spin.strip().split()] for spin in write] - -// diff = np.array(read) - np.array(write) -// print(np.max(np.abs(diff))) -// test system: integrated test 118_PW_CHG_BINARY -// yielding error 5.290000000000175e-11 \ No newline at end of file diff --git a/source/source_estate/rhog_io.h b/source/source_estate/rhog_io.h deleted file mode 100644 index b4b890808c0..00000000000 --- a/source/source_estate/rhog_io.h +++ /dev/null @@ -1,66 +0,0 @@ -#ifndef RHOG_IO_H -#define RHOG_IO_H - -#include -#include -#include -#include "source_basis/module_pw/pw_basis.h" -#include "source_base/module_parallel/para_world.h" -/** - * I/O free function of rho(G) in binary format - * Author: YuLiu98, Kirk0830 - * - * The designed foramt of the binary file is kept similar to the one in QuantumESPRESSO - * Modules/io_base.f90, function write_rhog (non-HDF5): - * - * /3/ /gammaonly/ /ngm_g/ /nspin/ /3/ ! bool, int, int. ngm_g is the global, total number of G-vectors - * /9/ /b11/ /b12/ /b13/ /b21/ /b22/ /b23/ /b31/ /b32/ /b33/ /9/ ! 9 real numbers, the three lattice vectors - * /ngm_g*3/ - * /miller(1,1)/ /miller(1,2)/ /miller(1,3)/ - * /miller(2,1)/ /miller(2,2)/ /miller(2,3)/ - * ... - * /miller(ngm_g,1)/ /miller(ngm_g,2)/ /miller(ngm_g,3)/ - * /ngm_g*3/ ! ngm_g*3 integers, the G-vectors in Miller indices - * /ngm_g/ - * /rhog(1)/ /rhog(2)/ /rhog(3)/ ... - * /ngm_g/ ! ngm_g complex numbers, the rho(G) values - * !/ngm_g/ ! ngm_g complex numbers, the rho(G) values for the second spin component - * !/rhog(1)/ /rhog(2)/ /rhog(3)/ ... - * !/ngm_g/ ! ngm_g complex numbers, the rho(G) values for the second spin component - * - * There are some aspects needed to ensure: - * 1. the correspondence between ABACUS PW and QuantumESPRESSO - * ABACUS PW QuantumESPRESSO - * bij UnitCell::GT b1, b2, b3 - * miller index ibox[*] mill - * rho Charge::rhog rho - * - * 2. the unit of each quantity - * ABACUS PW QuantumESPRESSO - * bij a.u. - * miller index - * rho - */ - -namespace elecstate -{ - -bool read_rhog(const std::string& filename, - const ModulePW::PW_Basis* pw_rhod, - const int nspin, - std::complex** rhog, - const Parallel::ParaWorld& pw_world, - std::ostream* os_warning); - -bool write_rhog(const std::string& fchg, - const bool gamma_only, - const ModulePW::PW_Basis* pw_rho, - const int nspin, - const ModuleBase::Matrix3& GT, - std::complex** rhog, - const Parallel::ParaWorld& pw_world, - std::ostream* os_warning); - -} // namespace elecstate - -#endif diff --git a/source/source_estate/setup_estate_pw.h b/source/source_estate/setup_estate_pw.h index b4d9d8a8a41..bde637028a2 100644 --- a/source/source_estate/setup_estate_pw.h +++ b/source/source_estate/setup_estate_pw.h @@ -4,6 +4,7 @@ #include "source_cell/unitcell.h" #include "source_cell/klist.h" #include "source_pw/module_pwdft/stru_fac.h" +#include "source_basis/module_pw/pw_basis_sup.h" #include "source_estate/elecstate.h" #include "source_pw/module_pwdft/vl_pw.h" #include "source_pw/module_pwdft/vsep_pw.h" diff --git a/source/source_estate/test/CMakeLists.txt b/source/source_estate/test/CMakeLists.txt index 7bd5f3bfd72..d3aecdbef57 100644 --- a/source/source_estate/test/CMakeLists.txt +++ b/source/source_estate/test/CMakeLists.txt @@ -63,7 +63,7 @@ AddTest( ../elecstate_pw_cal_tau.cpp ../elecstate.cpp ../occupy.cpp - ../module_charge/charge_mpi.cpp + ../module_charge/chg_parallel.cpp ../../source_lcao/module_deltaspin/spin_constrain.cpp ../../source_lcao/module_deltaspin/deltaspin_state.cpp ../../source_psi/psi.cpp @@ -94,64 +94,16 @@ AddTest( ) AddTest( - TARGET MODULE_ESTATE_potentials_new - LIBS parameter base device planewave_serial - SOURCES potential_new_test.cpp td_field_test.cpp ../module_pot/potential_new.cpp - ../module_pot/td_field.cpp ../module_pot/td_field_profiles.cpp ../module_pot/td_field_manager.cpp -) - -AddTest( - TARGET MODULE_ESTATE_charge_test - LIBS parameter planewave_serial base device cell_info - SOURCES charge_test.cpp ../module_charge/charge.cpp - -) - -AddTest( - TARGET MODULE_ESTATE_charge_mixing - LIBS parameter base psi device planewave_serial cell_info - SOURCES charge_mixing_test.cpp - ../module_charge/charge_mixing.cpp ../module_charge/charge_mixing_dmr.cpp ../module_charge/mix_resid.cpp - ../module_charge/mix_precond.cpp ../module_charge/charge_mixing_rho.cpp - ../module_charge/charge_mixing_uspp.cpp -) - -AddTest( - TARGET MODULE_ESTATE_charge_extra - LIBS parameter base device cell_info - SOURCES charge_extra_test.cpp ../module_charge/charge_extra.cpp ../../source_io/module_output/read_cube.cpp ../../source_io/module_output/write_cube.cpp - ../../source_base/module_fft/fft_bundle.cpp ../../source_base/module_fft/fft_cpu.cpp -) - -AddTest( - TARGET MODULE_ESTATE_gint_prec_ctrl + TARGET MODULE_ESTATE_occ_comput LIBS parameter base device - SOURCES gint_prec_ctrl_test.cpp - ../module_charge/gint_prec_ctrl.cpp -) - -AddTest( - TARGET MODULE_ESTATE_test_rhog_io - LIBS parameter base device planewave - SOURCES test_rhog_io.cpp ../rhog_io.cpp ../../source_basis/module_pw/test/test_tool.cpp - # This test drives PW_Basis::initmpi and read/write_rhog's MPI collectives, - # so it must keep __MPI even though this directory disables it. Its main() - # calls MPI_Init via test_tool.cpp's setupmpi(). - KEEP_FEATURE_DEFINITIONS __MPI + SOURCES test_occ_comput.cpp ../occ_comput.cpp ) AddTest( - TARGET MODULE_ESTATE_charge_mpi_test - LIBS parameter psi base device planewave - SOURCES charge_mpi_test.cpp ../module_charge/charge_mpi.cpp - # Real MPI test: its main() calls MPI_Init unconditionally, so it must keep - # __MPI despite this directory disabling it. - KEEP_FEATURE_DEFINITIONS __MPI -) - -add_test(NAME MODULE_ESTATE_charge_mpi_test_4np - COMMAND mpirun -np 4 ./MODULE_ESTATE_charge_mpi_test; - WORKING_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR} + TARGET MODULE_ESTATE_potentials_new + LIBS parameter base device planewave_serial + SOURCES potential_new_test.cpp td_field_test.cpp ../module_pot/potential_new.cpp + ../module_pot/td_field.cpp ../module_pot/td_field_profiles.cpp ../module_pot/td_field_manager.cpp ) endif() diff --git a/source/source_estate/test/charge_extra_test.cpp b/source/source_estate/test/charge_extra_test.cpp deleted file mode 100644 index cc5dd66189c..00000000000 --- a/source/source_estate/test/charge_extra_test.cpp +++ /dev/null @@ -1,299 +0,0 @@ -#include "gmock/gmock.h" -#include "gtest/gtest.h" -#define private public -#define protected public -#include "source_io/module_parameter/parameter.h" -#include "source_estate/module_charge/charge_extra.h" -#include "prepare_unitcell.h" -#include "source_base/module_fft/fft_bundle.h" -#undef private -#undef protected -// mock functions for UnitCell - -Magnetism::Magnetism() -{ -} -Magnetism::~Magnetism() -{ -} - - -// mock functions for Charge -Charge::Charge() -{ - rhopw = new ModulePW::PW_Basis; - rhopw->nrxx = 8; - rhopw->nx = 2; - rhopw->ny = 2; - rhopw->nz = 2; - rho = new double*[1]; - rho[0] = new double[rhopw->nrxx]; - ModuleBase::GlobalFunc::ZEROS(rho[0], rhopw->nrxx); - for (int i = 0; i < rhopw->nrxx; ++i) - { - rho[0][i] = i + 1; - } -} -Charge::~Charge() -{ - delete[] rho[0]; - delete[] rho; - delete rhopw; -} -void Charge::atomic_rho(const int spin_number_need, - const double& omega, - double** rho_in, - const ModuleBase::ComplexMatrix& strucFac, - const UnitCell& ucell) const -{ -} - -// mock functions for PW_Basis -namespace ModulePW -{ -PW_Basis::PW_Basis() -{ -} -PW_Basis::~PW_Basis() -{ -} -void PW_Basis::initgrids(const double lat0_in, const ModuleBase::Matrix3 latvec_in, const double gridecut) -{ -} -void PW_Basis::initgrids(const double lat0_in, - const ModuleBase::Matrix3 latvec_in, - const int nx_in, - int ny_in, - int nz_in) -{ -} -void PW_Basis::distribute_r() -{ -} -} // namespace ModulePW - -// mock functions for Structure_Factor -Structure_Factor::Structure_Factor() -{ -} -Structure_Factor::~Structure_Factor() -{ -} -void Structure_Factor::setup(const UnitCell*, const Parallel_Grid&, const ModulePW::PW_Basis*) -{ -} - -/************************************************ - * unit test of module_charge/charge_extra.cpp - ***********************************************/ - -/** - * - Tested Functions: - * - Charge_Extra::Init_CE() - * - Initialization of viriables used in charge extrapolation methods - * - Charge_Extra::extrapolate_charge() - * - charge extrapolation - * - Charge_Extra::update_all_dis() - * - update displacements - * - Charge_Extra::find_alpha_and_beta() - * - determine alpha and beta - */ - -class ChargeExtraTest : public ::testing::Test -{ - protected: - Charge_Extra CE; - UcellTestPrepare utp = UcellTestLib["Si"]; - std::unique_ptr ucell; - Parallel_Grid* pgrid = nullptr; - Charge charge; - Structure_Factor sf; - void SetUp() override - { - PARAM.input.nspin = 1; - PARAM.sys.global_out_dir = "./support/"; - ucell = utp.SetUcellInfo(); - ucell->omega = 1.0; - } - void TearDown() override - { - } -}; - -TEST_F(ChargeExtraTest, InitCEWarningQuit) -{ - PARAM.input.chg_extrap = "wwww"; - testing::internal::CaptureStdout(); - EXPECT_EXIT(CE.Init_CE(PARAM.input.nspin, ucell->nat, charge.rhopw->nrxx, PARAM.input.chg_extrap), - ::testing::ExitedWithCode(1), - ""); - std::string output = testing::internal::GetCapturedStdout(); - EXPECT_THAT(output, testing::HasSubstr("charge extrapolation method is not available")); -} - -TEST_F(ChargeExtraTest, InitCECase1) -{ - PARAM.input.chg_extrap = "none"; - CE.Init_CE(PARAM.input.nspin, ucell->nat, charge.rhopw->nrxx, PARAM.input.chg_extrap); - EXPECT_EQ(CE.pot_order, 0); -} - -TEST_F(ChargeExtraTest, InitCECase2) -{ - PARAM.input.chg_extrap = "atomic"; - CE.Init_CE(PARAM.input.nspin, ucell->nat, charge.rhopw->nrxx, PARAM.input.chg_extrap); - EXPECT_EQ(CE.pot_order, 1); -} - -TEST_F(ChargeExtraTest, InitCECase3) -{ - PARAM.input.chg_extrap = "first-order"; - CE.Init_CE(PARAM.input.nspin, ucell->nat, charge.rhopw->nrxx, PARAM.input.chg_extrap); - EXPECT_EQ(CE.pot_order, 2); - EXPECT_NE(CE.delta_rho1.size(), 0); - EXPECT_NE(CE.delta_rho2.size(), 0); -} - -TEST_F(ChargeExtraTest, InitCECase4) -{ - PARAM.input.chg_extrap = "second-order"; - CE.Init_CE(PARAM.input.nspin, ucell->nat, charge.rhopw->nrxx, PARAM.input.chg_extrap); - EXPECT_EQ(CE.pot_order, 3); - EXPECT_DOUBLE_EQ(CE.alpha, 1.0); - EXPECT_DOUBLE_EQ(CE.beta, 0.0); - EXPECT_NE(CE.delta_rho1.size(), 0); - EXPECT_NE(CE.delta_rho2.size(), 0); - EXPECT_NE(CE.dis_old1, nullptr); - EXPECT_NE(CE.dis_old2, nullptr); - EXPECT_NE(CE.dis_now, nullptr); -} - -TEST_F(ChargeExtraTest, ExtrapolateChargeCase1) -{ - PARAM.input.chg_extrap = "second-order"; - CE.Init_CE(PARAM.input.nspin, ucell->nat, charge.rhopw->nrxx, PARAM.input.chg_extrap); - CE.istep = 0; - CE.pot_order = 3; - - GlobalV::ofs_running.open("log"); - CE.extrapolate_charge(pgrid, *ucell.get(), &charge, &sf, GlobalV::ofs_running, GlobalV::ofs_warning); - GlobalV::ofs_running.close(); - - // Check the results - std::ifstream ifs("log"); - std::string expected_output = " charge density from previous step !\n"; - std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); - ifs.close(); - std::remove("log"); - - EXPECT_EQ(output, expected_output); - EXPECT_EQ(CE.rho_extr, 0); -} - -TEST_F(ChargeExtraTest, ExtrapolateChargeCase2) -{ - PARAM.input.chg_extrap = "second-order"; - CE.Init_CE(PARAM.input.nspin, ucell->nat, charge.rhopw->nrxx, PARAM.input.chg_extrap); - CE.istep = 1; - CE.pot_order = 3; - - GlobalV::ofs_running.open("log"); - CE.extrapolate_charge(pgrid, *ucell.get(), &charge, &sf, GlobalV::ofs_running, GlobalV::ofs_warning); - GlobalV::ofs_running.close(); - - // Check the results - std::ifstream ifs("log"); - std::string expected_output = " NEW-OLD atomic charge density approx. for the potential !\n"; - std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); - ifs.close(); - std::remove("log"); - - EXPECT_EQ(output, expected_output); - EXPECT_EQ(CE.rho_extr, 1); -} - -TEST_F(ChargeExtraTest, ExtrapolateChargeCase3) -{ - PARAM.input.chg_extrap = "second-order"; - CE.Init_CE(PARAM.input.nspin, ucell->nat, charge.rhopw->nrxx, PARAM.input.chg_extrap); - CE.istep = 2; - CE.pot_order = 3; - - GlobalV::ofs_running.open("log"); - CE.extrapolate_charge(pgrid, *ucell.get(), &charge, &sf, GlobalV::ofs_running, GlobalV::ofs_warning); - GlobalV::ofs_running.close(); - - // Check the results - std::ifstream ifs("log"); - std::string expected_output = " first order charge density extrapolation !\n"; - std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); - ifs.close(); - std::remove("log"); - - EXPECT_EQ(output, expected_output); - EXPECT_EQ(CE.rho_extr, 2); -} - -TEST_F(ChargeExtraTest, ExtrapolateChargeCase4) -{ - PARAM.input.chg_extrap = "second-order"; - CE.Init_CE(PARAM.input.nspin, ucell->nat, charge.rhopw->nrxx, PARAM.input.chg_extrap); - CE.istep = 3; - - GlobalV::ofs_running.open("log"); - CE.extrapolate_charge(pgrid, *ucell.get(), &charge, &sf, GlobalV::ofs_running, GlobalV::ofs_warning); - GlobalV::ofs_running.close(); - - // Check the results - std::ifstream ifs("log"); - std::string expected_output = " second order charge density extrapolation !\n alpha = 0\n beta = 0\n"; - std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); - ifs.close(); - std::remove("log"); - - EXPECT_EQ(output, expected_output); - EXPECT_EQ(CE.rho_extr, 3); - std::remove("./support/OLD2_SPIN1_CHG.cube"); -} - -TEST_F(ChargeExtraTest, UpdateAllDis) -{ - PARAM.input.chg_extrap = "second-order"; - CE.Init_CE(PARAM.input.nspin, ucell->nat, charge.rhopw->nrxx, PARAM.input.chg_extrap); - CE.istep = 3; - for (int i = 0; i < ucell->nat; ++i) - { - for (int j = 0; j < 3; ++j) - { - CE.dis_old1[i][j] = i; - CE.dis_now[i][j] = j; - } - } - - CE.update_all_dis(*ucell.get()); - - EXPECT_EQ(CE.istep, 4); - EXPECT_DOUBLE_EQ(CE.dis_old2[0][2], 0.0); - EXPECT_DOUBLE_EQ(CE.dis_old1[0][2], 2.0); - EXPECT_DOUBLE_EQ(CE.dis_now[0][2], 0.0); -} - -TEST_F(ChargeExtraTest, FindAlphaAndBeta) -{ - PARAM.input.chg_extrap = "second-order"; - CE.Init_CE(PARAM.input.nspin, ucell->nat, charge.rhopw->nrxx, PARAM.input.chg_extrap); - CE.istep = 3; - for (int i = 0; i < ucell->nat; ++i) - { - for (int j = 0; j < 3; ++j) - { - CE.dis_old1[i][j] = i; - CE.dis_now[i][j] = j; - } - } - - CE.find_alpha_and_beta(ucell->nat, GlobalV::ofs_running, GlobalV::ofs_warning); - - EXPECT_DOUBLE_EQ(CE.alpha, 1.0); - EXPECT_DOUBLE_EQ(CE.beta, 0.0); -} diff --git a/source/source_estate/test/charge_mixing_test.cpp b/source/source_estate/test/charge_mixing_test.cpp deleted file mode 100644 index c2bf0b1a2d0..00000000000 --- a/source/source_estate/test/charge_mixing_test.cpp +++ /dev/null @@ -1,1135 +0,0 @@ -#include "gmock/gmock.h" -#include "gtest/gtest.h" -#define private public -#include "../module_charge/charge_mixing.h" -#include "source_base/module_mixing/broyden_mixing.h" -#include "source_basis/module_pw/pw_basis.h" -#include "source_hamilt/module_xc/xc_functional.h" -#include "source_io/module_parameter/parameter.h" - -#ifdef _OPENMP -#include -#endif - -int XC_Functional::func_type = 1; -bool XC_Functional::ked_flag = false; - -// mock function -Magnetism::~Magnetism() -{ -} -Magnetism::Magnetism() -{ -} -Charge::~Charge() -{ -} -Charge::Charge() -{ -} - -void Charge::set_rhopw(ModulePW::PW_Basis* rhopw_in) -{ - this->rhopw = rhopw_in; -} - -// mock class cell -/************************************************ - * unit test of charge_mixing.cpp - ***********************************************/ - -/** - * - Tested Functions: - * - SetMixingTest: - * Charge_Mixing::set_mixing() - * Charge_Mixing::init_mixing() - * Charge_Mixing::set_rhopw(rhopw_in) - * Charge_Mixing::get_mixing_mode() - * Charge_Mixing::get_mixing_beta() - * Charge_Mixing::get_mixing_ndim() - * Charge_Mixing::get_mixing_gg0() - * - set the basic parameters of class charge_mixing - * - KerkerScreenTest: Charge_Mixing::Kerker_screen_recip(drhog) - * Charge_Mixing::Kerker_screen_real(drhog) - * - screen drho with Kerker method - * - InnerDotTest: Charge_Mixing::inner_product_recip_hartree(rhog1, rhog2) - * Charge_Mixing::inner_product_recip_rho(rhog1, rhog2) - * Charge_Mixing::inner_product_recip_simple(rhog1, rhog2) - * Charge_Mixing::inner_product_real(rho1, rho2) - * - calculate the inner product of two vectors - * - MixRhoTest: Charge_Mixing::mix_rho(chr) - * Charge_Mixing::mix_rho_recip(chr) - * Charge_Mixing::mix_rho_real(chr) - * - mix rho with different methods - * - MixDivCombTest: Charge_Mixing::divide_data - * Charge_Mixing::combine_data - * Charge_Mixing::clean_data - * - divide and combine data - * - */ - -class ChargeMixingTest : public ::testing::Test -{ - public: - UnitCell ucell; - ChargeMixingTest() - { - // Init pw_basis - pw_basis.initgrids(4, ModuleBase::Matrix3(1, 0, 0, 0, 1, 0, 0, 0, 1), 20); - pw_basis.initparameters(false, 20); - pw_basis.setuptransform(); - pw_basis.collect_local_pw(); - pw_dbasis.initgrids(4, ModuleBase::Matrix3(1, 0, 0, 0, 1, 0, 0, 0, 1), 40); - pw_dbasis.initparameters(false, 40); - pw_dbasis.setuptransform(&pw_basis); - pw_dbasis.collect_local_pw(); - // default mixing parameters - PARAM.input.mixing_mode = "broyden"; - PARAM.input.mixing_beta = 0.8; - PARAM.input.mixing_ndim = 8; - PARAM.input.mixing_gg0 = 1.0; - PARAM.input.mixing_tau = false; - PARAM.input.mixing_beta_mag = 1.6; - PARAM.input.mixing_gg0_mag = 0.0; - PARAM.input.mixing_gg0_min = 0.1; - PARAM.input.mixing_angle = -10.0; - PARAM.input.mixing_dmr = false; - ucell.omega = 1.0; - ucell.tpiba = 1.0; - } - ModulePW::PW_Basis pw_basis; - ModulePW::PW_Basis_Sup pw_dbasis; - Charge charge; -}; - -TEST_F(ChargeMixingTest, SetMixingTest) -{ -#ifdef _OPENMP - omp_set_num_threads(1); -#endif - PARAM.input.nspin = 1; - Charge_Mixing CMtest; - CMtest.set_rhopw(&pw_basis, &pw_basis); - PARAM.input.mixing_beta = 1.0; - PARAM.input.mixing_ndim = 1; - PARAM.input.mixing_gg0 = 1.0; - - CMtest.set_mixing(PARAM.input.mixing_mode, - PARAM.input.mixing_beta, - PARAM.input.mixing_ndim, - PARAM.input.mixing_gg0, - PARAM.input.mixing_tau, - PARAM.input.mixing_beta_mag, - PARAM.input.mixing_gg0_mag, - PARAM.input.mixing_gg0_min, - PARAM.input.mixing_angle, - PARAM.input.mixing_dmr, - ucell.omega, - ucell.tpiba); - EXPECT_EQ(CMtest.get_mixing_mode(), "broyden"); - EXPECT_EQ(CMtest.get_mixing_beta(), 1.0); - EXPECT_EQ(CMtest.get_mixing_ndim(), 1); - EXPECT_EQ(CMtest.get_mixing_gg0(), 1.0); - EXPECT_EQ(CMtest.mixing_tau, false); - EXPECT_EQ(CMtest.mixing_beta_mag, 1.6); - EXPECT_EQ(CMtest.mixing_gg0_mag, 0.0); - EXPECT_EQ(CMtest.mixing_gg0_min, 0.1); - EXPECT_EQ(CMtest.mixing_angle, -10.0); - EXPECT_EQ(CMtest.mixing_dmr, false); - - PARAM.input.mixing_tau = true; - PARAM.input.mixing_mode = "plain"; - CMtest.set_mixing(PARAM.input.mixing_mode, - PARAM.input.mixing_beta, - PARAM.input.mixing_ndim, - PARAM.input.mixing_gg0, - PARAM.input.mixing_tau, - PARAM.input.mixing_beta_mag, - PARAM.input.mixing_gg0_mag, - PARAM.input.mixing_gg0_min, - PARAM.input.mixing_angle, - PARAM.input.mixing_dmr, - ucell.omega, - ucell.tpiba); - EXPECT_EQ(CMtest.mixing_mode, "plain"); - EXPECT_EQ(CMtest.mixing_tau, true); - - PARAM.input.mixing_beta = 1.1; - std::string output; - testing::internal::CaptureStdout(); - EXPECT_EXIT(CMtest.set_mixing(PARAM.input.mixing_mode, - PARAM.input.mixing_beta, - PARAM.input.mixing_ndim, - PARAM.input.mixing_gg0, - PARAM.input.mixing_tau, - PARAM.input.mixing_beta_mag, - PARAM.input.mixing_gg0_mag, - PARAM.input.mixing_gg0_min, - PARAM.input.mixing_angle, - PARAM.input.mixing_dmr, - ucell.omega, - ucell.tpiba);, ::testing::ExitedWithCode(1), ""); - output = testing::internal::GetCapturedStdout(); - EXPECT_THAT(output, testing::HasSubstr("You'd better set mixing_beta to [0.0, 1.0]!")); - - PARAM.input.mixing_beta = 0.7; - PARAM.input.mixing_beta_mag = -0.1; - PARAM.input.nspin = 2; - testing::internal::CaptureStdout(); - EXPECT_EXIT(CMtest.set_mixing(PARAM.input.mixing_mode, - PARAM.input.mixing_beta, - PARAM.input.mixing_ndim, - PARAM.input.mixing_gg0, - PARAM.input.mixing_tau, - PARAM.input.mixing_beta_mag, - PARAM.input.mixing_gg0_mag, - PARAM.input.mixing_gg0_min, - PARAM.input.mixing_angle, - PARAM.input.mixing_dmr, - ucell.omega, - ucell.tpiba);, ::testing::ExitedWithCode(1), ""); - output = testing::internal::GetCapturedStdout(); - EXPECT_THAT(output, testing::HasSubstr("You'd better set mixing_beta_mag >= 0.0!")); - - PARAM.input.nspin = 1; - PARAM.input.mixing_beta = 0.7; - PARAM.input.mixing_beta_mag = 1.6; - PARAM.input.mixing_mode = "nothing"; - testing::internal::CaptureStdout(); - EXPECT_EXIT(CMtest.set_mixing(PARAM.input.mixing_mode, - PARAM.input.mixing_beta, - PARAM.input.mixing_ndim, - PARAM.input.mixing_gg0, - PARAM.input.mixing_tau, - PARAM.input.mixing_beta_mag, - PARAM.input.mixing_gg0_mag, - PARAM.input.mixing_gg0_min, - PARAM.input.mixing_angle, - PARAM.input.mixing_dmr, - ucell.omega, - ucell.tpiba);, ::testing::ExitedWithCode(1), ""); - output = testing::internal::GetCapturedStdout(); - EXPECT_THAT(output, testing::HasSubstr("This Mixing mode is not implemended yet,coming soon.")); -} - -TEST_F(ChargeMixingTest, InitMixingTest) -{ -#ifdef _OPENMP - omp_set_num_threads(1); -#endif - PARAM.input.nspin = 1; - XC_Functional::func_type = 1; - XC_Functional::ked_flag = false; - Charge_Mixing CMtest; - CMtest.set_rhopw(&pw_basis, &pw_basis); - - CMtest.set_mixing(PARAM.input.mixing_mode, - PARAM.input.mixing_beta, - PARAM.input.mixing_ndim, - PARAM.input.mixing_gg0, - PARAM.input.mixing_tau, - PARAM.input.mixing_beta_mag, - PARAM.input.mixing_gg0_mag, - PARAM.input.mixing_gg0_min, - PARAM.input.mixing_angle, - PARAM.input.mixing_dmr, - ucell.omega, - ucell.tpiba); - - PARAM.input.scf_thr_type= 1; - CMtest.init_mixing(); - EXPECT_EQ(CMtest.rho_mdata.length, pw_basis.npw); - - PARAM.input.scf_thr_type= 2; - CMtest.init_mixing(); - EXPECT_EQ(CMtest.rho_mdata.length, pw_basis.nrxx); - - PARAM.input.nspin = 4; - CMtest.init_mixing(); - EXPECT_EQ(CMtest.rho_mdata.length, 4 * pw_basis.nrxx); - - PARAM.input.nspin = 1; - PARAM.input.mixing_tau = true; - CMtest.set_mixing(PARAM.input.mixing_mode, - PARAM.input.mixing_beta, - PARAM.input.mixing_ndim, - PARAM.input.mixing_gg0, - PARAM.input.mixing_tau, - PARAM.input.mixing_beta_mag, - PARAM.input.mixing_gg0_mag, - PARAM.input.mixing_gg0_min, - PARAM.input.mixing_angle, - PARAM.input.mixing_dmr, - ucell.omega, - ucell.tpiba); - XC_Functional::func_type = 3; - XC_Functional::ked_flag = true; - CMtest.init_mixing(); - EXPECT_EQ(CMtest.tau_mdata.length, pw_basis.nrxx); - - PARAM.input.nspin = 4; - PARAM.input.mixing_angle = 1.0; - CMtest.set_mixing(PARAM.input.mixing_mode, - PARAM.input.mixing_beta, - PARAM.input.mixing_ndim, - PARAM.input.mixing_gg0, - PARAM.input.mixing_tau, - PARAM.input.mixing_beta_mag, - PARAM.input.mixing_gg0_mag, - PARAM.input.mixing_gg0_min, - PARAM.input.mixing_angle, - PARAM.input.mixing_dmr, - ucell.omega, - ucell.tpiba); - CMtest.init_mixing(); - EXPECT_EQ(CMtest.rho_mdata.length, 2 * pw_basis.nrxx); -} - -TEST_F(ChargeMixingTest, InnerDotRealTest) -{ - Charge_Mixing CMtest; - // non mixing angle case - CMtest.set_mixing(PARAM.input.mixing_mode, - PARAM.input.mixing_beta, - PARAM.input.mixing_ndim, - PARAM.input.mixing_gg0, - PARAM.input.mixing_tau, - PARAM.input.mixing_beta_mag, - PARAM.input.mixing_gg0_mag, - PARAM.input.mixing_gg0_min, - PARAM.input.mixing_angle, - PARAM.input.mixing_dmr, - ucell.omega, - ucell.tpiba); - CMtest.set_rhopw(&pw_basis, &pw_basis); - PARAM.input.nspin = 4; - - // a simple sum for inner product - std::vector drho1(pw_basis.nrxx * PARAM.input.nspin); - std::vector drho2(pw_basis.nrxx * PARAM.input.nspin); - for (int i = 0; i < pw_basis.nrxx * PARAM.input.nspin; ++i) - { - drho1[i] = 1.0; - drho2[i] = double(i); - } - double inner = CMtest.inner_product_real(drho1.data(), drho2.data()); - EXPECT_NEAR(inner, 0.5 * pw_basis.nrxx * PARAM.input.nspin * (pw_basis.nrxx * PARAM.input.nspin - 1), 1e-8); - - // mixing angle case - PARAM.input.mixing_angle = 1.0; - CMtest.set_mixing(PARAM.input.mixing_mode, - PARAM.input.mixing_beta, - PARAM.input.mixing_ndim, - PARAM.input.mixing_gg0, - PARAM.input.mixing_tau, - PARAM.input.mixing_beta_mag, - PARAM.input.mixing_gg0_mag, - PARAM.input.mixing_gg0_min, - PARAM.input.mixing_angle, - PARAM.input.mixing_dmr, - ucell.omega, - ucell.tpiba); - PARAM.input.nspin = 4; - - // a simple sum for inner product - drho1.resize(pw_basis.nrxx * 2); - drho2.resize(pw_basis.nrxx * 2); - for (int i = 0; i < pw_basis.nrxx * 2; ++i) - { - drho1[i] = 1.0; - drho2[i] = double(i); - } - inner = CMtest.inner_product_real(drho1.data(), drho2.data()); - EXPECT_NEAR(inner, 0.5 * pw_basis.nrxx * 2 * (pw_basis.nrxx * 2 - 1), 1e-8); -} - -TEST_F(ChargeMixingTest, InnerDotRecipSimpleTest) -{ - Charge_Mixing CMtest; - // non mixing angle case - CMtest.set_mixing(PARAM.input.mixing_mode, - PARAM.input.mixing_beta, - PARAM.input.mixing_ndim, - PARAM.input.mixing_gg0, - PARAM.input.mixing_tau, - PARAM.input.mixing_beta_mag, - PARAM.input.mixing_gg0_mag, - PARAM.input.mixing_gg0_min, - PARAM.input.mixing_angle, - PARAM.input.mixing_dmr, - ucell.omega, - ucell.tpiba); - CMtest.set_rhopw(&pw_basis, &pw_basis); - PARAM.input.nspin = 2; - - // a simple sum for inner product - std::vector> drhog1(pw_basis.npw * PARAM.input.nspin); - std::vector> drhog2(pw_basis.npw * PARAM.input.nspin); - for (int i = 0; i < pw_basis.npw * PARAM.input.nspin; ++i) - { - drhog1[i] = 1.0; - drhog2[i] = double(i); - } - double inner = CMtest.inner_product_recip_simple(drhog1.data(), drhog2.data()); - EXPECT_NEAR(inner, 0.5 * pw_basis.npw * PARAM.input.nspin * (pw_basis.npw * PARAM.input.nspin - 1), 1e-8); -} - -TEST_F(ChargeMixingTest, InnerDotRecipHartreeTest) -{ - // REAL - Charge_Mixing CMtest; - CMtest.set_rhopw(&pw_basis, &pw_basis); - const int npw = pw_basis.npw; - const int nrxx = pw_basis.nrxx; - PARAM.input.nspin = 1; - std::vector drhor1(pw_basis.nrxx); - std::vector drhor2(pw_basis.nrxx); - for (int i = 0; i < pw_basis.nrxx; ++i) - { - drhor1[i] = 1.0; - drhor2[i] = double(i); - } - double inner = CMtest.inner_product_real(drhor1.data(), drhor2.data()); - EXPECT_NEAR(inner, 0.5 * pw_basis.nrxx * (pw_basis.nrxx - 1), 1e-8); - - // RECIPROCAL NSPIN=1 - ucell.tpiba2 = 1.0; - ucell.omega = 2.0; - CMtest.set_mixing(PARAM.input.mixing_mode, - PARAM.input.mixing_beta, - PARAM.input.mixing_ndim, - PARAM.input.mixing_gg0, - PARAM.input.mixing_tau, - PARAM.input.mixing_beta_mag, - PARAM.input.mixing_gg0_mag, - PARAM.input.mixing_gg0_min, - PARAM.input.mixing_angle, - PARAM.input.mixing_dmr, - ucell.omega, - ucell.tpiba); - PARAM.input.nspin = 1; - std::vector> drhog1(pw_basis.npw); - std::vector> drhog2(pw_basis.npw); - for (int i = 0; i < pw_basis.nrxx; ++i) - { - drhor1[i] = 0.0; - } - drhor1[2] = 1.0; - pw_basis.real2recip(drhor1.data(), drhog1.data()); - pw_basis.real2recip(drhor2.data(), drhog2.data()); - - inner = CMtest.inner_product_recip_hartree(drhog1.data(), drhog2.data()); - EXPECT_NEAR(inner, -0.3 * ModuleBase::e2 * ModuleBase::FOUR_PI, 1e-8); - - // RECIPROCAL NSPIN=2 - PARAM.input.nspin = 2; - drhog1.resize(pw_basis.npw * PARAM.input.nspin); - drhog2.resize(pw_basis.npw * PARAM.input.nspin); - std::vector> drhog1_mag(pw_basis.npw * PARAM.input.nspin); - std::vector> drhog2_mag(pw_basis.npw * PARAM.input.nspin); - for (int i = 0; i < pw_basis.npw * PARAM.input.nspin; ++i) - { - drhog1[i] = std::complex(1.0, double(i)); - drhog2[i] = std::complex(1.0, 1.0); - } - // set mag - for (int i = 0; i < pw_basis.npw; ++i) - { - drhog1_mag[i] = drhog1[i] + drhog1[i+pw_basis.npw]; - drhog1_mag[i+pw_basis.npw] = drhog1[i] - drhog1[i+pw_basis.npw]; - drhog2_mag[i] = drhog2[i] + drhog2[i+pw_basis.npw]; - drhog2_mag[i+pw_basis.npw] = drhog2[i] - drhog2[i+pw_basis.npw]; - } - PARAM.sys.gamma_only_pw= false; - inner = CMtest.inner_product_recip_hartree(drhog1_mag.data(), drhog2_mag.data()); - EXPECT_NEAR(inner, 236763.82650318215, 1e-8); - PARAM.sys.gamma_only_pw= true; - inner = CMtest.inner_product_recip_hartree(drhog1_mag.data(), drhog2_mag.data()); - EXPECT_NEAR(inner, 236763.82650318215 * 2, 1e-8); - - // RECIPROCAL NSPIN=4 without mixing_angle - PARAM.input.nspin = 4; - drhog1.resize(pw_basis.npw * PARAM.input.nspin); - drhog2.resize(pw_basis.npw * PARAM.input.nspin); - for (int i = 0; i < pw_basis.npw * PARAM.input.nspin; ++i) - { - drhog1[i] = std::complex(1.0, double(i)); - drhog2[i] = std::complex(1.0, 1.0); - } - - PARAM.sys.domag = false; - PARAM.sys.domag_z = false; - inner = CMtest.inner_product_recip_hartree(drhog1.data(), drhog2.data()); - EXPECT_NEAR(inner, 28260.091995611871, 1e-8); - PARAM.sys.gamma_only_pw= true; - PARAM.sys.domag = true; - PARAM.sys.domag_z = true; - inner = CMtest.inner_product_recip_hartree(drhog1.data(), drhog2.data()); - EXPECT_NEAR(inner, 110668.61166927818, 1e-8); - - // RECIPROCAL NSPIN=4 with mixing_angle - PARAM.input.nspin = 4; - PARAM.input.mixing_angle = 1.0; - CMtest.set_mixing(PARAM.input.mixing_mode, - PARAM.input.mixing_beta, - PARAM.input.mixing_ndim, - PARAM.input.mixing_gg0, - PARAM.input.mixing_tau, - PARAM.input.mixing_beta_mag, - PARAM.input.mixing_gg0_mag, - PARAM.input.mixing_gg0_min, - PARAM.input.mixing_angle, - PARAM.input.mixing_dmr, - ucell.omega, - ucell.tpiba); - drhog1.resize(pw_basis.npw * 2); - drhog2.resize(pw_basis.npw * 2); - for (int i = 0; i < pw_basis.npw * 2; ++i) - { - drhog1[i] = std::complex(1.0, double(i)); - drhog2[i] = std::complex(1.0, 1.0); - } - PARAM.sys.gamma_only_pw= false; - inner = CMtest.inner_product_recip_hartree(drhog1.data(), drhog2.data()); - EXPECT_NEAR(inner, 36548.881431837777, 1e-8); - PARAM.sys.gamma_only_pw= true; - inner = CMtest.inner_product_recip_hartree(drhog1.data(), drhog2.data()); - EXPECT_NEAR(inner, 44776.555369916401, 1e-8); -} - -TEST_F(ChargeMixingTest, InnerDotRecipRhoTest) -{ - // REAL - Charge_Mixing CMtest; - CMtest.set_rhopw(&pw_basis, &pw_basis); - PARAM.input.nspin = 1; - std::vector drhor1(pw_basis.nrxx); - std::vector drhor2(pw_basis.nrxx); - for (int i = 0; i < pw_basis.nrxx; ++i) - { - drhor1[i] = 1.0; - drhor2[i] = double(i); - } - double inner = CMtest.inner_product_real(drhor1.data(), drhor2.data()); - EXPECT_NEAR(inner, 0.5 * pw_basis.nrxx * (pw_basis.nrxx - 1), 1e-8); - - // RECIPROCAL - ucell.tpiba2 = 1.0; - ucell.omega = 2.0; - CMtest.set_mixing(PARAM.input.mixing_mode, - PARAM.input.mixing_beta, - PARAM.input.mixing_ndim, - PARAM.input.mixing_gg0, - PARAM.input.mixing_tau, - PARAM.input.mixing_beta_mag, - PARAM.input.mixing_gg0_mag, - PARAM.input.mixing_gg0_min, - PARAM.input.mixing_angle, - PARAM.input.mixing_dmr, - ucell.omega, - ucell.tpiba); - PARAM.input.nspin = 1; - std::vector> drhog1(pw_basis.npw); - std::vector> drhog2(pw_basis.npw); - for (int i = 0; i < pw_basis.nrxx; ++i) - { - drhor1[i] = 0.0; - } - drhor1[2] = 1.0; - pw_basis.real2recip(drhor1.data(), drhog1.data()); - pw_basis.real2recip(drhor2.data(), drhog2.data()); - - inner = CMtest.inner_product_recip_rho(drhog1.data(), drhog2.data()); - EXPECT_NEAR(inner, -0.3 * ModuleBase::e2 * ModuleBase::FOUR_PI, 1e-8); - - PARAM.input.nspin = 2; - drhog1.resize(pw_basis.npw * PARAM.input.nspin); - drhog2.resize(pw_basis.npw * PARAM.input.nspin); - for (int i = 0; i < pw_basis.npw * PARAM.input.nspin; ++i) - { - drhog1[i] = std::complex(1.0, double(i)); - drhog2[i] = std::complex(1.0, 1.0); - } - PARAM.sys.gamma_only_pw= false; - inner = CMtest.inner_product_recip_rho(drhog1.data(), drhog2.data()); - EXPECT_NEAR(inner, 236763.82650318215, 1e-8); - PARAM.sys.gamma_only_pw= true; - inner = CMtest.inner_product_recip_rho(drhog1.data(), drhog2.data()); - EXPECT_NEAR(inner, 236763.82650318215 * 2, 1e-8); - - PARAM.input.nspin = 4; - drhog1.resize(pw_basis.npw * PARAM.input.nspin); - drhog2.resize(pw_basis.npw * PARAM.input.nspin); - for (int i = 0; i < pw_basis.npw * PARAM.input.nspin; ++i) - { - drhog1[i] = std::complex(1.0, double(i)); - drhog2[i] = std::complex(1.0, 1.0); - } - - PARAM.sys.domag = false; - PARAM.sys.domag_z = false; - inner = CMtest.inner_product_recip_rho(drhog1.data(), drhog2.data()); - EXPECT_NEAR(inner, 28260.091995611871, 1e-8); - PARAM.sys.gamma_only_pw= true; - PARAM.sys.domag = true; - PARAM.sys.domag_z = true; - inner = CMtest.inner_product_recip_rho(drhog1.data(), drhog2.data()); - EXPECT_NEAR(inner, 110668.61166927818, 1e-8); -} - -TEST_F(ChargeMixingTest, KerkerScreenRecipTest) -{ - Charge_Mixing CMtest; - CMtest.set_rhopw(&pw_basis, &pw_basis); - ucell.tpiba = 1.0; - CMtest.set_mixing(PARAM.input.mixing_mode, - PARAM.input.mixing_beta, - PARAM.input.mixing_ndim, - PARAM.input.mixing_gg0, - PARAM.input.mixing_tau, - PARAM.input.mixing_beta_mag, - PARAM.input.mixing_gg0_mag, - PARAM.input.mixing_gg0_min, - PARAM.input.mixing_angle, - PARAM.input.mixing_dmr, - ucell.omega, - ucell.tpiba); - // nspin = 1 - PARAM.input.nspin = 1; - std::complex* drhog = new std::complex[PARAM.input.nspin*pw_basis.npw]; - std::complex* drhog_old = new std::complex[PARAM.input.nspin*pw_basis.npw]; - for (int i = 0; i < PARAM.input.nspin*pw_basis.npw; ++i) - { - drhog_old[i] = drhog[i] = std::complex(1.0, 1.0); - } - // no kerker - CMtest.mixing_gg0 = 0.0; - CMtest.Kerker_screen_recip(drhog); - for (int i = 0; i < PARAM.input.nspin*pw_basis.npw; ++i) - { - EXPECT_EQ(drhog[i], drhog_old[i]); - } - // kerker - CMtest.mixing_gg0 = 1.0; - CMtest.Kerker_screen_recip(drhog); - double gg0 = std::pow(ModuleBase::BOHR_TO_A, 2); - for (int i = 0; i < pw_basis.npw; ++i) - { - double gg = this->pw_basis.gg[i]; - double ref = std::max(gg / (gg + gg0), 0.1 / CMtest.mixing_beta); - EXPECT_NEAR(drhog[i].real(), ref, 1e-10); - EXPECT_NEAR(drhog[i].imag(), ref, 1e-10); - } - delete[] drhog; - delete[] drhog_old; - - // nspin = 2 - PARAM.input.nspin = 2; - CMtest.mixing_beta = 0.4; - CMtest.mixing_beta_mag = 1.6; - drhog = new std::complex[PARAM.input.nspin*pw_basis.npw]; - drhog_old = new std::complex[PARAM.input.nspin*pw_basis.npw]; - for (int i = 0; i < PARAM.input.nspin*pw_basis.npw; ++i) - { - drhog_old[i] = drhog[i] = std::complex(1.0, 1.0); - } - // mixing_gg0 = 0.0 - CMtest.mixing_gg0 = 0.0; - CMtest.Kerker_screen_recip(drhog); - for (int i = 0; i < PARAM.input.nspin*pw_basis.npw; ++i) - { - EXPECT_EQ(drhog[i], drhog_old[i]); - } - // mixing_gg0 = 1.0, mixing_gg0_mag = 0.0 - CMtest.mixing_gg0 = 1.0; - CMtest.Kerker_screen_recip(drhog); - gg0 = std::pow(ModuleBase::BOHR_TO_A, 2); - for (int i = 0; i < pw_basis.npw; ++i) - { - double gg = this->pw_basis.gg[i]; - double ref = std::max(gg / (gg + gg0), 0.1 / CMtest.mixing_beta); - // rho - EXPECT_NEAR(drhog[i].real(), ref, 1e-10); - EXPECT_NEAR(drhog[i].imag(), ref, 1e-10); - // mag - EXPECT_NEAR(drhog[i+pw_basis.npw].real(), 1.0, 1e-10); - EXPECT_NEAR(drhog[i+pw_basis.npw].imag(), 1.0, 1e-10); - } - delete[] drhog; - delete[] drhog_old; - - // nspin = 4 - PARAM.input.nspin = 4; - drhog = new std::complex[PARAM.input.nspin*pw_basis.npw]; - drhog_old = new std::complex[PARAM.input.nspin*pw_basis.npw]; - for (int i = 0; i < PARAM.input.nspin*pw_basis.npw; ++i) - { - drhog_old[i] = drhog[i] = std::complex(1.0, 1.0); - } - // mixing_gg0 = 0.0 - CMtest.mixing_gg0 = 0.0; - CMtest.Kerker_screen_recip(drhog); - for (int i = 0; i < PARAM.input.nspin*pw_basis.npw; ++i) - { - EXPECT_EQ(drhog[i], drhog_old[i]); - } - // mixing_gg0 = 1.0, mixing_gg0_mag = 0.0 - CMtest.mixing_gg0 = 1.0; - CMtest.Kerker_screen_recip(drhog); - gg0 = std::pow(ModuleBase::BOHR_TO_A, 2); - for (int i = 0; i < pw_basis.npw; ++i) - { - double gg = this->pw_basis.gg[i]; - double ref = std::max(gg / (gg + gg0), 0.1 / CMtest.mixing_beta); - // rho - EXPECT_NEAR(drhog[i].real(), ref, 1e-10); - EXPECT_NEAR(drhog[i].imag(), ref, 1e-10); - } - for (int i = 0; i < 3*pw_basis.npw; ++i) - { - EXPECT_NEAR(drhog[i + pw_basis.npw].real(), 1.0, 1e-10); - EXPECT_NEAR(drhog[i + pw_basis.npw].imag(), 1.0, 1e-10); - } - // mixing_gg0 = 1.0, mixing_gg0_mag = 2.0 - CMtest.mixing_gg0 = 1.0; - CMtest.mixing_gg0_mag = 2.0; - CMtest.Kerker_screen_recip(drhog); - double gg1 = std::pow(1.0 * ModuleBase::BOHR_TO_A, 2); - double gg2 = std::pow(2.0 * ModuleBase::BOHR_TO_A, 2); - for (int i = 0; i < pw_basis.npw; ++i) - { - double gg = this->pw_basis.gg[i]; - double ref = std::max(gg / (gg + gg1), 0.1 / CMtest.mixing_beta); - // rho - EXPECT_NEAR(drhog[i].real(), ref * ref, 1e-10); - EXPECT_NEAR(drhog[i].imag(), ref * ref, 1e-10); - } - for (int i = 0; i < pw_basis.npw; ++i) - { - double gg = this->pw_basis.gg[i]; - double ref = std::max(gg / (gg + gg2), 0.1 / CMtest.mixing_beta_mag); - // rho - for (int j = 1; j < PARAM.input.nspin; ++j) - { - EXPECT_NEAR(drhog[i + pw_basis.npw * j].real(), ref, 1e-10); - EXPECT_NEAR(drhog[i + pw_basis.npw * j].imag(), ref, 1e-10); - } - } - delete[] drhog; - delete[] drhog_old; -} - -TEST_F(ChargeMixingTest, KerkerScreenRealTest) -{ - Charge_Mixing CMtest; - CMtest.set_rhopw(&pw_basis, &pw_basis); - ucell.tpiba = 1.0; - CMtest.set_mixing(PARAM.input.mixing_mode, - PARAM.input.mixing_beta, - PARAM.input.mixing_ndim, - PARAM.input.mixing_gg0, - PARAM.input.mixing_tau, - PARAM.input.mixing_beta_mag, - PARAM.input.mixing_gg0_mag, - PARAM.input.mixing_gg0_min, - PARAM.input.mixing_angle, - PARAM.input.mixing_dmr, - ucell.omega, - ucell.tpiba); - // nspin = 1 - PARAM.input.nspin = 1; - double* drhor = new double[PARAM.input.nspin*pw_basis.nrxx]; - double* drhor_ref = new double[PARAM.input.nspin*pw_basis.nrxx]; - for (int i = 0; i < PARAM.input.nspin*pw_basis.nrxx; ++i) - { - drhor_ref[i] = drhor[i] = 1.0; - } - // no kerker - CMtest.mixing_gg0 = 0.0; - CMtest.Kerker_screen_real(drhor); - for (int i = 0; i < PARAM.input.nspin*pw_basis.nrxx; ++i) - { - EXPECT_EQ(drhor[i], drhor_ref[i]); - } - delete[] drhor; - delete[] drhor_ref; - - // nspin = 2 - PARAM.input.nspin = 2; - CMtest.mixing_gg0 = 0.0; - std::complex* drhog = new std::complex[PARAM.input.nspin*pw_basis.npw]; - std::complex* drhog_old = new std::complex[PARAM.input.nspin*pw_basis.npw]; - drhor = new double[PARAM.input.nspin*pw_basis.nrxx]; - drhor_ref = new double[PARAM.input.nspin*pw_basis.nrxx]; - for (int i = 0; i < PARAM.input.nspin*pw_basis.npw; ++i) - { - drhog_old[i] = drhog[i] = std::complex(1.0, 1.0); - } - CMtest.Kerker_screen_recip(drhog); // no kerker - for (int i = 0; i < PARAM.input.nspin*pw_basis.npw; ++i) - { - EXPECT_EQ(drhog[i], drhog_old[i]); - } - - // RECIPROCAL - CMtest.mixing_gg0 = 1.0; - PARAM.input.mixing_gg0_mag = 0.0; - CMtest.Kerker_screen_recip(drhog); - const double gg0 = std::pow(ModuleBase::BOHR_TO_A, 2); - for (int i = 0; i < pw_basis.npw; ++i) - { - std::complex ration = drhog[i] / drhog[i+pw_basis.npw]; - double gg = this->pw_basis.gg[i]; - double ration_ref = std::max(gg / (gg + gg0), 0.1 / CMtest.mixing_beta); - EXPECT_NEAR(ration.real(), ration_ref, 1e-10); - EXPECT_NEAR(ration.imag(), 0, 1e-10); - } - - // REAL - pw_basis.recip2real(drhog, drhor_ref); - pw_basis.recip2real(drhog_old, drhor); - - CMtest.mixing_gg0 = 0.0; - PARAM.input.mixing_gg0_mag = 0.0; - // nothing happens - CMtest.Kerker_screen_real(drhor); - - CMtest.mixing_gg0 = 1.0; - CMtest.Kerker_screen_real(drhor); - for (int i = 0; i < pw_basis.nrxx; ++i) - { - EXPECT_NEAR(drhor[i], drhor_ref[i], 1e-8); - } - - delete[] drhog; - delete[] drhog_old; - delete[] drhor; - delete[] drhor_ref; - -} - -TEST_F(ChargeMixingTest, MixRhoTest) -{ - PARAM.sys.double_grid = false; - charge.set_rhopw(&pw_basis); - const int nspin = PARAM.input.nspin = 1; - PARAM.sys.domag_z = false; - XC_Functional::func_type = 3; - XC_Functional::ked_flag = true; - PARAM.input.mixing_beta = 0.7; - PARAM.input.mixing_ndim = 1; - PARAM.input.mixing_gg0 = 0.0; - PARAM.input.mixing_tau = true; - PARAM.input.mixing_mode = "plain"; - const int nrxx = pw_basis.nrxx; - const int npw = pw_basis.npw; - charge._space_rho = new double[nspin * nrxx]; - charge._space_rho_save = new double[nspin * nrxx]; - charge._space_rhog = new std::complex[nspin * npw]; - charge._space_rhog_save = new std::complex[nspin * npw]; - charge._space_kin_r = new double[nspin * nrxx]; - charge._space_kin_r_save = new double[nspin * nrxx]; - charge.rho = new double*[nspin]; - charge.rhog = new std::complex*[nspin]; - charge.rho_save = new double*[nspin]; - charge.rhog_save = new std::complex*[nspin]; - charge.kin_r = new double*[nspin]; - charge.kin_r_save = new double*[nspin]; - for (int is = 0; is < nspin; is++) - { - charge.rho[is] = charge._space_rho + is * nrxx; - charge.rhog[is] = charge._space_rhog + is * npw; - charge.rho_save[is] = charge._space_rho_save + is * nrxx; - charge.rhog_save[is] = charge._space_rhog_save + is * npw; - charge.kin_r[is] = charge._space_kin_r + is * nrxx; - charge.kin_r_save[is] = charge._space_kin_r_save + is * nrxx; - } - std::vector real_ref(nspin * nrxx); - std::vector real_save_ref(nspin * nrxx); - std::vector> recip_ref(nspin * npw); - std::vector> recip_save_ref(nspin * npw); - for(int i = 0 ; i < nspin * npw; ++i) - { - recip_ref[i] = std::complex(double(i), 1.0); - recip_save_ref[i] = std::complex(double(i), 0.0); - } - for(int i = 0 ; i < nspin ; ++i) - { - pw_basis.recip2real(recip_ref.data() + i * npw, real_ref.data() + i * nrxx); - pw_basis.recip2real(recip_save_ref.data() + i * npw, real_save_ref.data() + i * nrxx); - } - //--------------------------------MAIN BODY-------------------------------- - // RECIPROCAL - Charge_Mixing CMtest_recip; - CMtest_recip.set_rhopw(&pw_basis, &pw_basis); - PARAM.input.scf_thr_type= 1; - CMtest_recip.set_mixing(PARAM.input.mixing_mode, - PARAM.input.mixing_beta, - PARAM.input.mixing_ndim, - PARAM.input.mixing_gg0, - PARAM.input.mixing_tau, - PARAM.input.mixing_beta_mag, - PARAM.input.mixing_gg0_mag, - PARAM.input.mixing_gg0_min, - PARAM.input.mixing_angle, - PARAM.input.mixing_dmr, - ucell.omega, - ucell.tpiba); - CMtest_recip.init_mixing(); - for(int i = 0 ; i < nspin * npw; ++i) - { - charge._space_rhog[i] = recip_ref[i]; - charge._space_rhog_save[i] = recip_save_ref[i]; - } - for(int i = 0 ; i < nspin * nrxx; ++i) - { - charge._space_rho[i] = real_ref[i]; - charge._space_rho_save[i] = real_save_ref[i]; - } - CMtest_recip.mix_rho(&charge); - for(int is = 0 ; is < nspin; ++is) - { - for(int ir = 0 ; ir < nrxx ; ++ir) - { - EXPECT_NEAR(charge.rho_save[is][ir], real_ref[is*nrxx + ir], 1e-8); - } - for(int ig = 0; ig < npw ; ++ig) - { - EXPECT_NEAR(charge.rhog[is][ig].real(), recip_save_ref[is*npw + ig].real(), 1e-8); - EXPECT_NEAR(charge.rhog[is][ig].imag(), recip_save_ref[is*npw + ig].imag() + 0.7, 1e-8); - } - } - - // REAL - Charge_Mixing CMtest_real; - PARAM.input.scf_thr_type= 2; - CMtest_real.set_rhopw(&pw_basis, &pw_basis); - CMtest_real.set_mixing(PARAM.input.mixing_mode, - PARAM.input.mixing_beta, - PARAM.input.mixing_ndim, - PARAM.input.mixing_gg0, - PARAM.input.mixing_tau, - PARAM.input.mixing_beta_mag, - PARAM.input.mixing_gg0_mag, - PARAM.input.mixing_gg0_min, - PARAM.input.mixing_angle, - PARAM.input.mixing_dmr, - ucell.omega, - ucell.tpiba); - CMtest_real.init_mixing(); - for(int i = 0 ; i < nspin * nrxx; ++i) - { - charge._space_rho[i] = real_ref[i]; - charge._space_rho_save[i] = real_save_ref[i]; - } - CMtest_recip.mix_rho(&charge); - for(int is = 0 ; is < nspin; ++is) - { - for(int ir = 0 ; ir < nrxx ; ++ir) - { - EXPECT_NEAR(charge.rho_save[is][ir], real_ref[is*nrxx + ir], 1e-8); - EXPECT_NEAR(charge.rho[is][ir], 0.3*real_save_ref[is*nrxx+ir] + 0.7*real_ref[is*nrxx+ir], 1e-8); - } - } - - //------------------------------------------------------------------------- - delete[] charge._space_rho; - delete[] charge._space_rho_save; - delete[] charge._space_rhog; - delete[] charge._space_rhog_save; - delete[] charge._space_kin_r; - delete[] charge._space_kin_r_save; - delete[] charge.rho; - delete[] charge.rhog; - delete[] charge.rho_save; - delete[] charge.rhog_save; - delete[] charge.kin_r; - delete[] charge.kin_r_save; -} - -TEST_F(ChargeMixingTest, MixDoubleGridRhoTest) -{ - PARAM.sys.double_grid = true; - charge.set_rhopw(&pw_dbasis); - const int nspin = PARAM.input.nspin = 1; - PARAM.sys.domag_z = false; - XC_Functional::func_type = 3; - XC_Functional::ked_flag = true; - PARAM.input.mixing_beta = 0.7; - PARAM.input.mixing_ndim = 1; - PARAM.input.mixing_gg0 = 0.0; - PARAM.input.mixing_tau = true; - PARAM.input.mixing_mode = "plain"; - const int nrxx = pw_dbasis.nrxx; - const int npw = pw_dbasis.npw; - charge._space_rho = new double[nspin * nrxx]; - charge._space_rho_save = new double[nspin * nrxx]; - charge._space_rhog = new std::complex[nspin * npw]; - charge._space_rhog_save = new std::complex[nspin * npw]; - charge._space_kin_r = new double[nspin * nrxx]; - charge._space_kin_r_save = new double[nspin * nrxx]; - charge.rho = new double*[nspin]; - charge.rhog = new std::complex*[nspin]; - charge.rho_save = new double*[nspin]; - charge.rhog_save = new std::complex*[nspin]; - charge.kin_r = new double*[nspin]; - charge.kin_r_save = new double*[nspin]; - for (int is = 0; is < nspin; is++) - { - charge.rho[is] = charge._space_rho + is * nrxx; - charge.rhog[is] = charge._space_rhog + is * npw; - charge.rho_save[is] = charge._space_rho_save + is * nrxx; - charge.rhog_save[is] = charge._space_rhog_save + is * npw; - charge.kin_r[is] = charge._space_kin_r + is * nrxx; - charge.kin_r_save[is] = charge._space_kin_r_save + is * nrxx; - } - std::vector real_ref(nspin * nrxx); - std::vector real_save_ref(nspin * nrxx); - std::vector> recip_ref(nspin * npw); - std::vector> recip_save_ref(nspin * npw); - for (int i = 0; i < nspin * npw; ++i) - { - recip_ref[i] = std::complex(double(i), 1.0); - recip_save_ref[i] = std::complex(double(i), 0.0); - } - for (int i = 0; i < nspin; ++i) - { - pw_dbasis.recip2real(recip_ref.data() + i * npw, real_ref.data() + i * nrxx); - pw_dbasis.recip2real(recip_save_ref.data() + i * npw, real_save_ref.data() + i * nrxx); - } - //--------------------------------MAIN BODY-------------------------------- - // RECIPROCAL - Charge_Mixing CMtest_recip; - CMtest_recip.set_rhopw(&pw_basis, &pw_dbasis); - - PARAM.input.scf_thr_type= 1; - CMtest_recip.set_mixing(PARAM.input.mixing_mode, - PARAM.input.mixing_beta, - PARAM.input.mixing_ndim, - PARAM.input.mixing_gg0, - PARAM.input.mixing_tau, - PARAM.input.mixing_beta_mag, - PARAM.input.mixing_gg0_mag, - PARAM.input.mixing_gg0_min, - PARAM.input.mixing_angle, - PARAM.input.mixing_dmr, - ucell.omega, - ucell.tpiba); - - CMtest_recip.init_mixing(); - for (int i = 0; i < nspin * npw; ++i) - { - charge._space_rhog[i] = recip_ref[i]; - charge._space_rhog_save[i] = recip_save_ref[i]; - } - for (int i = 0; i < nspin * nrxx; ++i) - { - charge._space_rho[i] = real_ref[i]; - charge._space_rho_save[i] = real_save_ref[i]; - } - CMtest_recip.mix_rho(&charge); - for (int is = 0; is < nspin; ++is) - { - for (int ir = 0; ir < nrxx; ++ir) - { - EXPECT_NEAR(charge.rho_save[is][ir], real_ref[is * nrxx + ir], 1e-8); - } - for (int ig = 0; ig < npw; ++ig) - { - EXPECT_NEAR(charge.rhog[is][ig].real(), recip_save_ref[is * npw + ig].real(), 1e-8); - EXPECT_NEAR(charge.rhog[is][ig].imag(), recip_save_ref[is * npw + ig].imag() + 0.7, 1e-8); - } - } - - //------------------------------------------------------------------------- - delete[] charge._space_rho; - delete[] charge._space_rho_save; - delete[] charge._space_rhog; - delete[] charge._space_rhog_save; - delete[] charge._space_kin_r; - delete[] charge._space_kin_r_save; - delete[] charge.rho; - delete[] charge.rhog; - delete[] charge.rho_save; - delete[] charge.rhog_save; - delete[] charge.kin_r; - delete[] charge.kin_r_save; -} - -TEST_F(ChargeMixingTest, MixDivCombTest) -{ - // NSPIN = 1 - PARAM.input.nspin = 1; - Charge_Mixing CMtest; - CMtest.set_rhopw(&pw_basis, &pw_dbasis); - std::vector> data(pw_dbasis.npw, 1.0); - std::complex*datas, *datahf; - std::complex*datas2, *datahf2; - CMtest.divide_data(data.data(), datas, datahf); - EXPECT_EQ(datas, data.data()); - EXPECT_EQ(datahf, data.data() + pw_basis.npw); - CMtest.combine_data(data.data(), datas, datahf); - EXPECT_EQ(datas, nullptr); - EXPECT_EQ(datahf, nullptr); - - CMtest.divide_data(data.data(), datas2, datahf2); - CMtest.clean_data(datas2, datahf2); - EXPECT_EQ(datas2, nullptr); - EXPECT_EQ(datahf2, nullptr); - - // NSPIN = 2 - PARAM.input.nspin = 2; - data.resize(pw_dbasis.npw * 2, 1.0); - std::vector> dataout(pw_dbasis.npw * 2, 1.0); - CMtest.divide_data(data.data(), datas, datahf); - CMtest.combine_data(dataout.data(), datas, datahf); - EXPECT_EQ(datas, nullptr); - EXPECT_EQ(datahf, nullptr); - for (int i = 0; i < pw_dbasis.npw * 2; ++i) - { - EXPECT_EQ(dataout[i], data[i]); - } - - CMtest.divide_data(data.data(), datas2, datahf2); - CMtest.clean_data(datas2, datahf2); - EXPECT_EQ(datas2, nullptr); - EXPECT_EQ(datahf2, nullptr); -} - -TEST_F(ChargeMixingTest, SCFOscillationTest) -{ - Charge_Mixing CMtest; - int scf_nmax = 20; - int scf_os_ndim = 3; - double scf_os_thr = -0.05; - bool scf_oscillate = false; - std::vector drho(scf_nmax, 0.0); - std::vector scf_oscillate_ref(scf_nmax, false); - drho = {6.83639633652e-05, - 4.93523029235e-05, - 3.59230097735e-05, - 2.68356403913e-05, - 2.17490806464e-05, - 2.14231642508e-05, - 1.67507494811e-05, - 1.53575889539e-05, - 1.26504511554e-05, - 1.04762016224e-05, - 8.10000162918e-06, - 7.66427917682e-06, - 6.70112820094e-06, - 5.68594436664e-06, - 4.80120233733e-06, - 4.86519757184e-06, - 4.37855804356e-06, - 4.29922703412e-06, - 4.36398486331e-06, - 4.94224615955e-06}; - scf_oscillate_ref = {false,false,false,false,false,true,false,false,false,false, - false,false,true,false,false,true,true,true,true,true}; - for (int i = 1; i <= scf_nmax; ++i) - { - scf_oscillate = CMtest.if_scf_oscillate(i,drho[i-1],scf_os_ndim,scf_os_thr); - EXPECT_EQ(scf_oscillate, scf_oscillate_ref[i-1]); - } -} diff --git a/source/source_estate/test/charge_mpi_test.cpp b/source/source_estate/test/charge_mpi_test.cpp deleted file mode 100644 index 3f37329e8d6..00000000000 --- a/source/source_estate/test/charge_mpi_test.cpp +++ /dev/null @@ -1,283 +0,0 @@ -#include "source_base/matrix3.h" -#include "source_base/parallel_global.h" -#include "source_base/parallel_grid.h" -#include "source_estate/module_charge/charge.h" -#include "source_hamilt/module_xc/xc_functional.h" -#include "source_io/module_parameter/parameter.h" - -#include "gmock/gmock.h" -#include "gtest/gtest.h" - -bool XC_Functional::ked_flag = false; -void XC_Functional::set_xc_type(const std::string xc_func_in) -{ - ked_flag = xc_func_in == "scan"; -} -Charge::Charge() -{ -} -Charge::~Charge() -{ -} - -auto sum_array = [](const double* v, const int& nv) { - double sum = 0; - for (int i = 0; i < nv; ++i) - { - sum += v[i]; - } - return sum; -}; -/************************************************ - * unit test of module_charge/charge_mpi.cpp - ***********************************************/ - -/** - * - Tested Functions: - * - rho_mpi: Charge::rho_mpi(): - * - test rho_mpi - * - reduce_diff_pools: Charge::reduce_diff_pools() - * - test reduce_diff_pools - * - using rhopw and GlobalV - */ - -class ChargeMpiTest : public ::testing::Test -{ - protected: - Charge* charge; - std::string output; - double lat0 = 4; - ModuleBase::Matrix3 latvec; - int world_size = 1; - void SetUp() override - { - charge = new Charge; - MPI_Comm_size(MPI_COMM_WORLD, &world_size); - } - void TearDown() override - { - delete charge; - } -}; - -TEST_F(ChargeMpiTest, reduce_diff_pools1) -{ - if (GlobalV::NPROC >= 2 && GlobalV::NPROC % 2 == 0) - { - GlobalV::KPAR = 2; - Parallel_Global::init_pools(GlobalV::NPROC, - GlobalV::MY_RANK, - PARAM.inp.bndpar, - GlobalV::KPAR, - GlobalV::NPROC_IN_BNDGROUP, - GlobalV::RANK_IN_BPGROUP, - GlobalV::MY_BNDGROUP, - GlobalV::NPROC_IN_POOL, - GlobalV::RANK_IN_POOL, - GlobalV::MY_POOL); - ModulePW::PW_Basis* rhopw = new ModulePW::PW_Basis(); - rhopw->initmpi(GlobalV::NPROC_IN_POOL, GlobalV::RANK_IN_POOL, POOL_WORLD); - rhopw->initgrids(lat0, latvec, 40); - rhopw->initparameters(false, 10); - rhopw->setuptransform(); - Parallel_Grid pgrid; - pgrid.init(rhopw->nx, rhopw->ny, rhopw->nz, rhopw->nplane, rhopw->nrxx, rhopw->nz, 1, world_size); - charge->pgrid = &pgrid; - - int nz = rhopw->nz; - const int nrxx = rhopw->nrxx; - const int nxy = rhopw->nxy; - const int nplane = rhopw->nplane; - charge->nrxx = nrxx; - double* array_rho = new double[nrxx]; - for (int ir = 0; ir < nxy; ++ir) - { - for (int iz = 0; iz < nplane; ++iz) - { - array_rho[nplane * ir + iz] = (rhopw->startz_current + iz + ir * nz) / double(nxy * nz); - } - } - double refsum = sum_array(array_rho, nrxx); - - charge->reduce_diff_pools(array_rho); - double sum = sum_array(array_rho, nrxx); - EXPECT_EQ(sum, refsum * GlobalV::KPAR); - - delete[] array_rho; - delete rhopw; - } -} - -TEST_F(ChargeMpiTest, reduce_diff_pools2) -{ - if (GlobalV::NPROC >= 3) - { - GlobalV::KPAR = 3; - Parallel_Global::divide_pools(GlobalV::NPROC, - GlobalV::MY_RANK, - PARAM.inp.bndpar, - GlobalV::KPAR, - GlobalV::NPROC_IN_BNDGROUP, - GlobalV::RANK_IN_BPGROUP, - GlobalV::MY_BNDGROUP, - GlobalV::NPROC_IN_POOL, - GlobalV::RANK_IN_POOL, - GlobalV::MY_POOL); - ModulePW::PW_Basis* rhopw = new ModulePW::PW_Basis(); - rhopw->initmpi(GlobalV::NPROC_IN_POOL, GlobalV::RANK_IN_POOL, POOL_WORLD); - rhopw->initgrids(lat0, latvec, 40); - rhopw->initparameters(false, 10); - rhopw->setuptransform(); - charge->rhopw = rhopw; - Parallel_Grid pgrid; - pgrid.init(rhopw->nx, rhopw->ny, rhopw->nz, rhopw->nplane, rhopw->nrxx, rhopw->nz, 1, world_size); - charge->pgrid = &pgrid; - - int nz = rhopw->nz; - const int nrxx = rhopw->nrxx; - const int nxy = rhopw->nxy; - const int nplane = rhopw->nplane; - charge->nrxx = nrxx; - double* array_ref = new double[nxy * nz]; - for (int ixyz = 0; ixyz < nxy * nz; ++ixyz) - { - array_ref[ixyz] = ixyz / double(nxy * nz); - } - double refsum = sum_array(array_ref, nxy * nz); - - double* array_rho = new double[nrxx]; - for (int ir = 0; ir < nxy; ++ir) - { - for (int iz = 0; iz < nplane; ++iz) - { - array_rho[nplane * ir + iz] = (rhopw->startz_current + iz + ir * nz) / double(nxy * nz); - } - } - - charge->reduce_diff_pools(array_rho); - double sum = sum_array(array_rho, nrxx); - MPI_Allreduce(MPI_IN_PLACE, &sum, 1, MPI_DOUBLE, MPI_SUM, POOL_WORLD); - EXPECT_EQ(sum, refsum * GlobalV::KPAR); - - delete[] array_rho; - delete rhopw; - delete[] array_ref; - } -} - -TEST_F(ChargeMpiTest, rho_mpi) -{ - if (GlobalV::NPROC >= 2) - { - GlobalV::KPAR = 2; - Parallel_Global::divide_pools(GlobalV::NPROC, - GlobalV::MY_RANK, - PARAM.inp.bndpar, - GlobalV::KPAR, - GlobalV::NPROC_IN_BNDGROUP, - GlobalV::RANK_IN_BPGROUP, - GlobalV::MY_BNDGROUP, - GlobalV::NPROC_IN_POOL, - GlobalV::RANK_IN_POOL, - GlobalV::MY_POOL); - ModulePW::PW_Basis* rhopw = new ModulePW::PW_Basis(); - rhopw->initmpi(GlobalV::NPROC_IN_POOL, GlobalV::RANK_IN_POOL, POOL_WORLD); - rhopw->initgrids(lat0, latvec, 40); - rhopw->initparameters(false, 10); - rhopw->setuptransform(); - charge->rhopw = rhopw; - Parallel_Grid pgrid; - pgrid.init(rhopw->nx, rhopw->ny, rhopw->nz, rhopw->nplane, rhopw->nrxx, rhopw->nz, 1, world_size); - charge->pgrid = &pgrid; - ASSERT_EQ(PARAM.inp.nspin, 1); - charge->rho = new double*[1]; - charge->kin_r = new double*[1]; - - int nz = rhopw->nz; - const int nrxx = rhopw->nrxx; - const int nxy = rhopw->nxy; - const int nplane = rhopw->nplane; - charge->nrxx = nrxx; - charge->rho[0] = new double[nrxx]; - charge->kin_r[0] = new double[nrxx]; - charge->rho_mpi(); - - delete[] charge->rho[0]; - delete[] charge->rho; - delete[] charge->kin_r[0]; - delete[] charge->kin_r; - delete rhopw; - } - - GlobalV::KPAR = 1; - charge->rho_mpi(); -} - -TEST_F(ChargeMpiTest, kin_r_mpi) -{ - if (GlobalV::NPROC >= 2 && GlobalV::NPROC % 2 == 0) - { - XC_Functional::set_xc_type("scan"); - ASSERT_EQ(PARAM.inp.nspin, 1); - ASSERT_EQ(PARAM.inp.bndpar, 1); - GlobalV::KPAR = 2; - - Parallel_Global::divide_pools(GlobalV::NPROC, - GlobalV::MY_RANK, - PARAM.inp.bndpar, - GlobalV::KPAR, - GlobalV::NPROC_IN_BNDGROUP, - GlobalV::RANK_IN_BPGROUP, - GlobalV::MY_BNDGROUP, - GlobalV::NPROC_IN_POOL, - GlobalV::RANK_IN_POOL, - GlobalV::MY_POOL); - ModulePW::PW_Basis* rhopw = new ModulePW::PW_Basis(); - rhopw->initmpi(GlobalV::NPROC_IN_POOL, GlobalV::RANK_IN_POOL, POOL_WORLD); - rhopw->initgrids(lat0, latvec, 40); - rhopw->initparameters(false, 10); - rhopw->setuptransform(); - charge->rhopw = rhopw; - Parallel_Grid pgrid; - pgrid.init(rhopw->nx, rhopw->ny, rhopw->nz, rhopw->nplane, rhopw->nrxx, rhopw->nz, 1, world_size); - charge->pgrid = &pgrid; - - const int nz = rhopw->nz; - const int nrxx = rhopw->nrxx; - const int nxy = rhopw->nxy; - const int nplane = rhopw->nplane; - charge->nrxx = nrxx; - charge->kin_r = new double*[1]; - charge->kin_r[0] = new double[nrxx]; - - for (int ir = 0; ir < nxy; ++ir) - { - for (int iz = 0; iz < nplane; ++iz) - { - charge->kin_r[0][nplane * ir + iz] = (rhopw->startz_current + iz + ir * nz) / double(nxy * nz); - } - } - const double refsum = sum_array(charge->kin_r[0], nrxx); - - charge->kin_r_mpi(); - const double sum = sum_array(charge->kin_r[0], nrxx); - EXPECT_EQ(sum, refsum * GlobalV::KPAR); - - delete[] charge->kin_r[0]; - delete[] charge->kin_r; - delete rhopw; - XC_Functional::set_xc_type("pbe"); - } -} - -int main(int argc, char** argv) -{ - MPI_Init(&argc, &argv); - MPI_Comm_size(MPI_COMM_WORLD, &GlobalV::NPROC); - MPI_Comm_rank(MPI_COMM_WORLD, &GlobalV::MY_RANK); - int result = 0; - testing::InitGoogleTest(&argc, argv); - result = RUN_ALL_TESTS(); - MPI_Finalize(); - return result; -} diff --git a/source/source_estate/test/charge_test.cpp b/source/source_estate/test/charge_test.cpp deleted file mode 100644 index 3bd76578f32..00000000000 --- a/source/source_estate/test/charge_test.cpp +++ /dev/null @@ -1,214 +0,0 @@ -#include "gtest/gtest.h" -#include "gmock/gmock.h" - -#define private public -#define protected public -#include "source_cell/unitcell.h" -#include "source_estate/module_charge/charge.h" -#include "source_hamilt/module_xc/xc_functional.h" -#include "source_io/module_parameter/parameter.h" -#include "prepare_unitcell.h" -// mock functions for UnitCell - -Magnetism::Magnetism() -{ - this->tot_mag = 0.0; - this->abs_mag = 0.0; -} -Magnetism::~Magnetism() -{ -} - -// mock functions for Charge -int XC_Functional::func_type = 1; -bool XC_Functional::ked_flag = false; -namespace elecstate -{ -double tmp_ucell_omega = 500.0; -double tmp_gridecut = 80.0; -void Set_GlobalV_Default() -{ - PARAM.input.nspin = 1; - PARAM.input.test_charge = 0; - PARAM.input.nelec = 8; -} -} // namespace elecstate - -/************************************************ - * unit test of module_charge/charge.cpp - ***********************************************/ - -/** - * - Tested Functions: - * - Constructor: Charge::Charge() and Charge::~Charge() - * - this is a trivial test - * - Allocate: Charge::set_rhopw(), Charge::allocate(), Charge::destroy() - * - allocate rho, rhog, rho_save, rhog_save, kin_r, kin_r_save - * - using rhopw and PARAM.input.nspin - * - SumRho: Charge::sum_rho() - * - calculate \sum_{is}^nspin \sum_{ir}^nrxx rho[is][ir] - * - RenormalizeRho: Charge::renormalize_rho() - * - renormalize rho so as to ensure the sum of rho equals to total number of electrons - * - CheckNe: Charge::cal_rho2ne() - * - check the total number of electrons summed from rho[is] - * - SaveRhoBeforeSumBand: Charge::save_rho_before_sum_band() - * - meaning as the function name - * - InitFinalScf:: Charge::init_final_scf() - * - similar to Charge::allocate(), but for final scf - */ - -class ChargeTest : public ::testing::Test -{ - protected: - UcellTestPrepare utp = UcellTestLib["Si"]; - std::unique_ptr ucell; - Charge* charge; - ModulePW::PW_Basis* rhopw; - std::string output; - void SetUp() override - { - elecstate::Set_GlobalV_Default(); - ucell = utp.SetUcellInfo(); - charge = new Charge; - rhopw = new ModulePW::PW_Basis; - rhopw->initgrids(ucell->lat0, ucell->latvec, elecstate::tmp_gridecut); - rhopw->distribute_r(); - rhopw->initparameters(false, elecstate::tmp_gridecut); - rhopw->distribute_g(); - } - void TearDown() override - { - delete charge; - delete rhopw; - } -}; - -TEST_F(ChargeTest, Constructor) -{ - EXPECT_FALSE(charge->allocate_rho); - EXPECT_FALSE(charge->allocate_rho_final_scf); -} - -TEST_F(ChargeTest, Allocate) -{ - // ucell info - EXPECT_DOUBLE_EQ(ucell->omega, 265.302); - // rhopw info - EXPECT_DOUBLE_EQ(rhopw->lat0, 10.2); - EXPECT_EQ(rhopw->nx, 24); - EXPECT_EQ(rhopw->ny, 24); - EXPECT_EQ(rhopw->nz, 24); - EXPECT_EQ(rhopw->nxyz, 13824); - EXPECT_EQ(rhopw->nrxx, 13824); - EXPECT_EQ(rhopw->npw, 3143); - EXPECT_EQ(rhopw->npwtot, 3143); - // call Charge::allocate() - PARAM.input.test_charge = 2; - XC_Functional::func_type = 3; - XC_Functional::ked_flag = true; - charge->set_rhopw(rhopw); - EXPECT_FALSE(charge->allocate_rho); - const bool kin_den = charge->kin_density(); - charge->allocate(PARAM.input.nspin, kin_den); - EXPECT_TRUE(charge->allocate_rho); - // test if Charge::allocate() be called twice - EXPECT_NO_THROW(charge->allocate(PARAM.input.nspin, kin_den)); - EXPECT_TRUE(charge->allocate_rho); -} - -TEST_F(ChargeTest, SumRho) -{ - charge->set_rhopw(rhopw); - EXPECT_FALSE(charge->allocate_rho); - const bool kin_den = charge->kin_density(); - charge->allocate(PARAM.input.nspin, kin_den); - EXPECT_TRUE(charge->allocate_rho); - int nspin = (PARAM.input.nspin == 2) ? 2 : 1; - for (int is = 0; is < nspin; is++) - { - for (int ir = 0; ir < rhopw->nrxx; ir++) - { - charge->rho[is][ir] = 0.1; - } - } - charge->set_omega(&ucell->omega);; - EXPECT_NEAR(charge->sum_rho(), 0.1 * nspin * rhopw->nrxx * ucell->omega / rhopw->nxyz, 1E-10); -} - -TEST_F(ChargeTest, RenormalizeRho) -{ - charge->set_rhopw(rhopw); - EXPECT_FALSE(charge->allocate_rho); - const bool kin_den = charge->kin_density(); - charge->allocate(PARAM.input.nspin, kin_den); - EXPECT_TRUE(charge->allocate_rho); - int nspin = (PARAM.input.nspin == 2) ? 2 : 1; - for (int is = 0; is < nspin; is++) - { - for (int ir = 0; ir < rhopw->nrxx; ir++) - { - charge->rho[is][ir] = 0.1; - } - } - EXPECT_EQ(PARAM.input.nelec, 8); - charge->set_omega(&ucell->omega);; - charge->renormalize_rho(); - EXPECT_NEAR(charge->sum_rho(), 8.0, 1e-10); -} - -TEST_F(ChargeTest, CheckNe) -{ - charge->set_rhopw(rhopw); - EXPECT_FALSE(charge->allocate_rho); - const bool kin_den = charge->kin_density(); - charge->allocate(PARAM.input.nspin, kin_den); - EXPECT_TRUE(charge->allocate_rho); - int nspin = (PARAM.input.nspin == 2) ? 2 : 1; - for (int is = 0; is < nspin; is++) - { - for (int ir = 0; ir < rhopw->nrxx; ir++) - { - charge->rho[is][ir] = 0.1; - } - } - EXPECT_EQ(PARAM.input.nelec, 8); - charge->set_omega(&ucell->omega);; - charge->renormalize_rho(); - EXPECT_NEAR(charge->sum_rho(), 8.0, 1e-10); - EXPECT_NEAR(charge->cal_rho2ne(charge->rho[0]), 8.0, 1e-10); -} - -TEST_F(ChargeTest, SaveRhoBeforeSumBand) -{ - charge->set_rhopw(rhopw); - EXPECT_FALSE(charge->allocate_rho); - const bool kin_den = charge->kin_density(); - charge->allocate(PARAM.input.nspin, kin_den); - EXPECT_TRUE(charge->allocate_rho); - int nspin = (PARAM.input.nspin == 2) ? 2 : 1; - for (int is = 0; is < nspin; is++) - { - for (int ir = 0; ir < rhopw->nrxx; ir++) - { - charge->rho[is][ir] = 0.1; - } - } - EXPECT_EQ(PARAM.input.nelec, 8); - XC_Functional::func_type = 3; - XC_Functional::ked_flag = true; - charge->set_omega(&ucell->omega);; - charge->renormalize_rho(); - charge->save_rho_before_sum_band(); - EXPECT_NEAR(charge->cal_rho2ne(charge->rho_save[0]), 8.0, 1e-10); -} - -TEST_F(ChargeTest, InitFinalScf) -{ - charge->set_rhopw(rhopw); - XC_Functional::func_type = 1; - XC_Functional::ked_flag = false; - PARAM.input.test_charge = 2; - charge->init_final_scf(); - EXPECT_TRUE(charge->allocate_rho_final_scf); -} - diff --git a/source/source_estate/test/elecstate_base_test.cpp b/source/source_estate/test/elecstate_base_test.cpp index d3ce296271d..317ef2805a2 100644 --- a/source/source_estate/test/elecstate_base_test.cpp +++ b/source/source_estate/test/elecstate_base_test.cpp @@ -10,6 +10,7 @@ #include "source_estate/elecstate.h" #include "source_estate/elecstate_tools.h" #include "source_estate/occupy.h" +#include "source_basis/module_pw/pw_basis_big.h" // use PW_Basis_Big #include "source_io/module_parameter/parameter.h" #ifdef __MPI @@ -100,24 +101,19 @@ void ModulePW::PW_Basis::initgrids(double, ModuleBase::Matrix3, int, int, int) void ModulePW::PW_Basis::distribute_r() { } -void Charge::set_rho_core(const UnitCell& ucell, ModuleBase::ComplexMatrix const&, const bool*) -{ -} void Charge::init_rho(const UnitCell&, const Parallel_Grid&, ModuleBase::ComplexMatrix const&, ModuleSymmetry::Symmetry& symm, const void*, - const void*) + const void*, + const module_charge::InitRhoCfg&) { } void Charge::set_rhopw(ModulePW::PW_Basis*) { } -void Charge::renormalize_rho() -{ -} -void Charge::check_rho() +void Charge::renormalize_rho(const double, const double) { } diff --git a/source/source_estate/test/elecstate_energy_test.cpp b/source/source_estate/test/elecstate_energy_test.cpp index cf54a7b73ca..136b1d08f78 100644 --- a/source/source_estate/test/elecstate_energy_test.cpp +++ b/source/source_estate/test/elecstate_energy_test.cpp @@ -1,6 +1,5 @@ #include "gmock/gmock.h" #include "gtest/gtest.h" -#define private public #include "source_estate/elecstate.h" #include "source_hamilt/module_xc/xc_functional.h" #include "source_io/module_parameter/parameter.h" @@ -62,25 +61,6 @@ namespace elecstate { class MockElecState : public ElecState { - public: - void Set_GlobalV_Default() - { - PARAM.input.imp_sol = false; - PARAM.input.dft_plus_u = 0; - // base class - PARAM.input.nspin = 1; - PARAM.input.nelec = 10.0; - PARAM.input.nupdown = 0.0; - PARAM.sys.two_fermi = false; - PARAM.input.nbands = 6; - PARAM.sys.nlocal = 6; - PARAM.input.esolver_type = "ksdft"; - PARAM.input.lspinorb = false; - PARAM.input.basis_type = "pw"; - GlobalV::KPAR = 1; - GlobalV::NPROC_IN_POOL = 1; - PARAM.input.sc_mag_switch = true; - } }; const double* ElecState::getRho(int spin) const { @@ -92,10 +72,21 @@ class ElecStateEnergyTest : public ::testing::Test { protected: elecstate::MockElecState* elecstate; + /// cal_energies() takes the flags that gate the optional energy terms + /// explicitly, so the fixture owns them instead of staging them in the + /// global parameter singleton. The values mirror the Input_para defaults, + /// except sc_mag_switch, which the original fixture turned on. + bool imp_sol = false; + bool sc_mag_switch = true; + int dft_plus_u = 0; + std::string assume_isolated = "none"; + /// band count the bandgap cases build their ekb matrix with + int nbands = 6; void SetUp() override { elecstate = new elecstate::MockElecState; - elecstate->Set_GlobalV_Default(); + GlobalV::KPAR = 1; + GlobalV::NPROC_IN_POOL = 1; } void TearDown() override { @@ -106,7 +97,7 @@ class ElecStateEnergyTest : public ::testing::Test TEST_F(ElecStateEnergyTest, CalEnergiesHarris) { elecstate->f_en.deband_harris = 0.1; - elecstate->cal_energies(1); + elecstate->cal_energies(1, imp_sol, sc_mag_switch, dft_plus_u, assume_isolated); // deband_harris + hatree + efiled + gatefield + escon EXPECT_DOUBLE_EQ(elecstate->f_en.etot_harris, 0.7); } @@ -114,8 +105,8 @@ TEST_F(ElecStateEnergyTest, CalEnergiesHarris) TEST_F(ElecStateEnergyTest, CalEnergiesHarrisImpSol) { elecstate->f_en.deband_harris = 0.1; - PARAM.input.imp_sol = true; - elecstate->cal_energies(1); + imp_sol = true; + elecstate->cal_energies(1, imp_sol, sc_mag_switch, dft_plus_u, assume_isolated); // deband_harris + hatree + efiled + gatefield + esol_el + esol_cav + escon EXPECT_DOUBLE_EQ(elecstate->f_en.etot_harris, 1.6); } @@ -123,8 +114,8 @@ TEST_F(ElecStateEnergyTest, CalEnergiesHarrisImpSol) TEST_F(ElecStateEnergyTest, CalEnergiesHarrisDFTU) { elecstate->f_en.deband_harris = 0.1; - PARAM.input.dft_plus_u = 1; - elecstate->cal_energies(1); + dft_plus_u = 1; + elecstate->cal_energies(1, imp_sol, sc_mag_switch, dft_plus_u, assume_isolated); // deband_harris + hatree + efiled + gatefield + edftu + escon EXPECT_DOUBLE_EQ(elecstate->f_en.etot_harris, 1.3); } @@ -132,7 +123,7 @@ TEST_F(ElecStateEnergyTest, CalEnergiesHarrisDFTU) TEST_F(ElecStateEnergyTest, CalEnergiesEtot) { elecstate->f_en.deband = 0.1; - elecstate->cal_energies(2); + elecstate->cal_energies(2, imp_sol, sc_mag_switch, dft_plus_u, assume_isolated); // deband + hatree + efiled + gatefield + escon EXPECT_DOUBLE_EQ(elecstate->f_en.etot, 0.7); } @@ -140,8 +131,8 @@ TEST_F(ElecStateEnergyTest, CalEnergiesEtot) TEST_F(ElecStateEnergyTest, CalEnergiesEtotImpSol) { elecstate->f_en.deband = 0.1; - PARAM.input.imp_sol = true; - elecstate->cal_energies(2); + imp_sol = true; + elecstate->cal_energies(2, imp_sol, sc_mag_switch, dft_plus_u, assume_isolated); // deband + hatree + efiled + gatefield + esol_el + esol_cav + escon EXPECT_DOUBLE_EQ(elecstate->f_en.etot, 1.6); } @@ -149,8 +140,8 @@ TEST_F(ElecStateEnergyTest, CalEnergiesEtotImpSol) TEST_F(ElecStateEnergyTest, CalEnergiesEtotDFTU) { elecstate->f_en.deband = 0.1; - PARAM.input.dft_plus_u = 1; - elecstate->cal_energies(2); + dft_plus_u = 1; + elecstate->cal_energies(2, imp_sol, sc_mag_switch, dft_plus_u, assume_isolated); // deband + hatree + efiled + gatefield + edftu + escon EXPECT_DOUBLE_EQ(elecstate->f_en.etot, 1.3); } @@ -173,10 +164,10 @@ TEST_F(ElecStateEnergyTest, CalBandgap) K_Vectors* klist = new K_Vectors; klist->set_nks(5); elecstate->klist = klist; - elecstate->ekb.create(klist->get_nks(), PARAM.input.nbands); + elecstate->ekb.create(klist->get_nks(), nbands); for (int ik = 0; ik < klist->get_nks(); ik++) { - for (int ib = 0; ib < PARAM.input.nbands; ib++) + for (int ib = 0; ib < nbands; ib++) { elecstate->ekb(ik, ib) = ib; } @@ -210,10 +201,10 @@ TEST_F(ElecStateEnergyTest, CalBandgapUpDw) } } elecstate->klist = klist; - elecstate->ekb.create(klist->get_nks(), PARAM.input.nbands); + elecstate->ekb.create(klist->get_nks(), nbands); for (int ik = 0; ik < klist->get_nks(); ik++) { - for (int ib = 0; ib < PARAM.input.nbands; ib++) + for (int ib = 0; ib < nbands; ib++) { if (ik < 3) { diff --git a/source/source_estate/test/elecstate_occupy_test.cpp b/source/source_estate/test/elecstate_occupy_test.cpp index 8f034ee93b6..fdcf6ac46cf 100644 --- a/source/source_estate/test/elecstate_occupy_test.cpp +++ b/source/source_estate/test/elecstate_occupy_test.cpp @@ -1,9 +1,7 @@ #include #include "gmock/gmock.h" #include "gtest/gtest.h" -#define private public #include "source_io/module_parameter/parameter.h" -#undef private /*************************************************************** * unit test of class Occupy @@ -14,9 +12,7 @@ * - Occupy::Occupy() * - Occupy::decision() */ -#define private public #include "source_estate/occupy.h" -#undef private class OccupyTest : public ::testing::Test { protected: @@ -182,21 +178,19 @@ TEST_F(OccupyTest, DecisionArbitrary) TEST_F(OccupyTest, IweightsNOSPIN) { - PARAM.input.nspin = 1; double ef = 0.0; ModuleBase::matrix wg(1, 1); std::vector wk(1, 2.0); ModuleBase::matrix ekb(1, 1); std::vector isk(1); ekb(0, 0) = 0.1; - occupy.iweights(1, wk, 1, 0, 2.0, ekb, ef, wg, 0, isk); + occupy.iweights(1, wk, 1, 0, 2.0, ekb, ef, wg, 0, isk, /*nspin=*/1); EXPECT_DOUBLE_EQ(wg(0, 0), 2.0); EXPECT_DOUBLE_EQ(ef, 0.1); } TEST_F(OccupyTest, IweightsSPIN) { - PARAM.input.nspin = 2; double ef_up = 0.0; double ef_dw = 0.0; ModuleBase::matrix wg(2, 1); @@ -207,8 +201,8 @@ TEST_F(OccupyTest, IweightsSPIN) isk[1] = 1; ekb(0, 0) = 0.1; ekb(1, 0) = 0.2; - occupy.iweights(2, wk, 1, 0, 1.0, ekb, ef_up, wg, 0, isk); - occupy.iweights(2, wk, 1, 0, 1.0, ekb, ef_dw, wg, 1, isk); + occupy.iweights(2, wk, 1, 0, 1.0, ekb, ef_up, wg, 0, isk, /*nspin=*/2); + occupy.iweights(2, wk, 1, 0, 1.0, ekb, ef_dw, wg, 1, isk, /*nspin=*/2); EXPECT_DOUBLE_EQ(wg(0, 0), 1.0); EXPECT_DOUBLE_EQ(wg(1, 0), 1.0); EXPECT_DOUBLE_EQ(ef_up, 0.1); @@ -217,7 +211,6 @@ TEST_F(OccupyTest, IweightsSPIN) TEST_F(OccupyTest, IweightsWarning) { - PARAM.input.nspin = 1; double ef = 0.0; ModuleBase::matrix wg(1, 1); std::vector wk(1, 2.0); @@ -226,28 +219,28 @@ TEST_F(OccupyTest, IweightsWarning) ekb(0, 0) = 0.1; testing::internal::CaptureStdout(); - EXPECT_EXIT(occupy.iweights(1, wk, 1, 0, 1.0, ekb, ef, wg, -1, isk);, ::testing::ExitedWithCode(1), ""); + EXPECT_EXIT(occupy.iweights(1, wk, 1, 0, 1.0, ekb, ef, wg, -1, isk, /*nspin=*/1);, ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); EXPECT_THAT(output, testing::HasSubstr("It is not a semiconductor or insulator. Please do not set 'smearing_method=fixed', and try other options.")); } TEST_F(OccupyTest, Wgauss) { - EXPECT_DOUBLE_EQ(occupy.wgauss(0.0, 0), 0.5); - EXPECT_DOUBLE_EQ(occupy.wgauss(0.0, -1), 0.4006259784506005); - EXPECT_DOUBLE_EQ(occupy.wgauss(0.0, -99), 0.5); - EXPECT_DOUBLE_EQ(occupy.wgauss(0.0, 1), 0.5); - EXPECT_DOUBLE_EQ(occupy.wgauss(0.0, 2), 0.5); - EXPECT_DOUBLE_EQ(occupy.wgauss(10, 0), 1.0); + EXPECT_DOUBLE_EQ(occupy_smearing::wgauss(0.0, 0), 0.5); + EXPECT_DOUBLE_EQ(occupy_smearing::wgauss(0.0, -1), 0.4006259784506005); + EXPECT_DOUBLE_EQ(occupy_smearing::wgauss(0.0, -99), 0.5); + EXPECT_DOUBLE_EQ(occupy_smearing::wgauss(0.0, 1), 0.5); + EXPECT_DOUBLE_EQ(occupy_smearing::wgauss(0.0, 2), 0.5); + EXPECT_DOUBLE_EQ(occupy_smearing::wgauss(10, 0), 1.0); } TEST_F(OccupyTest, W1gauss) { - EXPECT_DOUBLE_EQ(occupy.w1gauss(0.0, 0), -0.28209479177387814); - EXPECT_DOUBLE_EQ(occupy.w1gauss(0.0, -1), -0.1710991401561083); - EXPECT_DOUBLE_EQ(occupy.w1gauss(0.0, -99), -0.69314718055994529); - EXPECT_DOUBLE_EQ(occupy.w1gauss(0.0, 1), -0.14104739588693907); - EXPECT_DOUBLE_EQ(occupy.w1gauss(0.0, 2), -0.10578554691520431); + EXPECT_DOUBLE_EQ(occupy_smearing::w1gauss(0.0, 0), -0.28209479177387814); + EXPECT_DOUBLE_EQ(occupy_smearing::w1gauss(0.0, -1), -0.1710991401561083); + EXPECT_DOUBLE_EQ(occupy_smearing::w1gauss(0.0, -99), -0.69314718055994529); + EXPECT_DOUBLE_EQ(occupy_smearing::w1gauss(0.0, 1), -0.14104739588693907); + EXPECT_DOUBLE_EQ(occupy_smearing::w1gauss(0.0, 2), -0.10578554691520431); } TEST_F(OccupyTest, Sumkg) @@ -260,7 +253,7 @@ TEST_F(OccupyTest, Sumkg) double e = 0.0; int is = 0; std::vector isk = {0, 0}; - EXPECT_DOUBLE_EQ(occupy.sumkg(ekb, 1, 1, wk, smearing_sigma, ngauss, e, is, isk), 1.0); + EXPECT_DOUBLE_EQ(occupy_smearing::sumkg(ekb, 1, 1, wk, smearing_sigma, ngauss, e, is, isk), 1.0); } TEST_F(OccupyTest, Efermig) @@ -274,7 +267,7 @@ TEST_F(OccupyTest, Efermig) int is = 0; std::vector isk = {0, 0}; double ef = 0.0; - occupy.efermig(ekb, 1, 1, 1.0, wk, smearing_sigma, ngauss, ef, is, isk); + occupy_smearing::efermig(ekb, 1, 1, 1.0, wk, smearing_sigma, ngauss, ef, is, isk); EXPECT_NEAR(ef, -0.5, 1e-13); } diff --git a/source/source_estate/test/elecstate_print_test.cpp b/source/source_estate/test/elecstate_print_test.cpp index a45976e4bb4..6746461374e 100644 --- a/source/source_estate/test/elecstate_print_test.cpp +++ b/source/source_estate/test/elecstate_print_test.cpp @@ -2,7 +2,6 @@ #include "gmock/gmock.h" #include "gtest/gtest.h" -#define private public #include "source_cell/klist.h" #include "source_estate/elecstate.h" #include "source_estate/module_charge/charge.h" @@ -11,7 +10,6 @@ #include "source_hamilt/module_xc/xc_functional.h" #include "source_io/module_parameter/parameter.h" #include "source_estate/elecstate_print.h" -#undef private /*************************************************************** * mock functions ****************************************************************/ @@ -58,6 +56,11 @@ class ElecStatePrintTest : public ::testing::Test protected: elecstate::ElecState elecstate; UnitCell ucell; + /// print_etot() takes the INPUT parameters and the two-Fermi flag as + /// arguments, so the fixture owns them instead of writing the global + /// parameter singleton. + Input_para inp; + bool two_fermi = false; std::string output; std::ifstream ifs; std::ofstream ofs; @@ -94,8 +97,7 @@ class ElecStatePrintTest : public ::testing::Test ucell.magnet.tot_mag_nc[0] = 3.3; ucell.magnet.tot_mag_nc[1] = 4.4; ucell.magnet.tot_mag_nc[2] = 5.5; - PARAM.input.ks_solver = "dav"; - PARAM.sys.log_file = "test.dat"; + inp.ks_solver = "dav"; } void TearDown() { @@ -129,56 +131,56 @@ TEST_F(ElecStatePrintTest, PrintEtot) elecstate.charge = new Charge; elecstate.charge->nrxx = 100; elecstate.charge->nxyz = 1000; - PARAM.input.out_freq_elec = 1; - PARAM.input.imp_sol = true; - PARAM.input.efield_flag = true; - PARAM.input.gate_flag = true; - PARAM.sys.two_fermi = true; + inp.out_freq_elec = 1; + inp.imp_sol = true; + inp.efield_flag = true; + inp.gate_flag = true; + two_fermi = true; GlobalV::MY_RANK = 0; - PARAM.input.basis_type = "pw"; - PARAM.input.nspin = 2; + inp.basis_type = "pw"; + inp.nspin = 2; // iteration of different vdw_method std::vector vdw_methods = {"d2", "d3_0", "d3_bj"}; for (int i = 0; i < vdw_methods.size(); i++) { - PARAM.input.vdw_method = vdw_methods[i]; + inp.vdw_method = vdw_methods[i]; elecstate::print_etot(ucell.magnet,elecstate, converged, iter, scf_thr, - scf_thr_kin, duration, pw_diag_thr, avg_iter, false); + scf_thr_kin, duration, inp, two_fermi, pw_diag_thr, avg_iter, false); } // iteration of different ks_solver std::vector ks_solvers = {"cg", "lapack", "genelpa", "dav", "scalapack_gvx", "cusolver"}; for (int i = 0; i < ks_solvers.size(); i++) { - PARAM.input.ks_solver = ks_solvers[i]; + inp.ks_solver = ks_solvers[i]; testing::internal::CaptureStdout(); elecstate::print_etot(ucell.magnet,elecstate,converged, iter, scf_thr, - scf_thr_kin, duration, pw_diag_thr, avg_iter, print); + scf_thr_kin, duration, inp, two_fermi, pw_diag_thr, avg_iter, print); output = testing::internal::GetCapturedStdout(); - if (PARAM.input.ks_solver == "cg") + if (inp.ks_solver == "cg") { EXPECT_THAT(output, testing::HasSubstr("CG")); } - else if (PARAM.input.ks_solver == "lapack") + else if (inp.ks_solver == "lapack") { EXPECT_THAT(output, testing::HasSubstr("LA")); } - else if (PARAM.input.ks_solver == "genelpa") + else if (inp.ks_solver == "genelpa") { EXPECT_THAT(output, testing::HasSubstr("GE")); } - else if (PARAM.input.ks_solver == "dav") + else if (inp.ks_solver == "dav") { EXPECT_THAT(output, testing::HasSubstr("DA")); } - else if (PARAM.input.ks_solver == "scalapack_gvx") + else if (inp.ks_solver == "scalapack_gvx") { EXPECT_THAT(output, testing::HasSubstr("GV")); } - else if (PARAM.input.ks_solver == "cusolver") + else if (inp.ks_solver == "cusolver") { EXPECT_THAT(output, testing::HasSubstr("CU")); } @@ -214,16 +216,16 @@ TEST_F(ElecStatePrintTest, PrintEtotColorS2) elecstate.charge->nrxx = 100; elecstate.charge->nxyz = 1000; - PARAM.input.out_freq_elec = 1; - PARAM.input.imp_sol = true; - PARAM.input.efield_flag = true; - PARAM.input.gate_flag = true; - PARAM.sys.two_fermi = true; - PARAM.input.nspin = 2; + inp.out_freq_elec = 1; + inp.imp_sol = true; + inp.efield_flag = true; + inp.gate_flag = true; + two_fermi = true; + inp.nspin = 2; GlobalV::MY_RANK = 0; elecstate::print_etot(ucell.magnet,elecstate,converged, iter, scf_thr, - scf_thr_kin, duration, pw_diag_thr, avg_iter, print); + scf_thr_kin, duration, inp, two_fermi, pw_diag_thr, avg_iter, print); delete elecstate.charge; } @@ -243,17 +245,17 @@ TEST_F(ElecStatePrintTest, PrintEtotColorS4) elecstate.charge->nrxx = 100; elecstate.charge->nxyz = 1000; - PARAM.input.out_freq_elec = 1; - PARAM.input.imp_sol = true; - PARAM.input.efield_flag = true; - PARAM.input.gate_flag = true; - PARAM.sys.two_fermi = true; - PARAM.input.nspin = 4; - PARAM.input.noncolin = true; + inp.out_freq_elec = 1; + inp.imp_sol = true; + inp.efield_flag = true; + inp.gate_flag = true; + two_fermi = true; + inp.nspin = 4; + inp.noncolin = true; GlobalV::MY_RANK = 0; elecstate::print_etot(ucell.magnet,elecstate, converged, iter, scf_thr, scf_thr_kin, - duration, pw_diag_thr, avg_iter, print); + duration, inp, two_fermi, pw_diag_thr, avg_iter, print); delete elecstate.charge; } @@ -272,17 +274,17 @@ TEST_F(ElecStatePrintTest, PrintEtotSDFTPure) elecstate.charge->nrxx = 100; elecstate.charge->nxyz = 1000; - PARAM.input.out_freq_elec = 1; - PARAM.input.nspin = 1; + inp.out_freq_elec = 1; + inp.nspin = 1; GlobalV::MY_RANK = 0; // Pure SDFT: nbands=0, no KS diagonalization -> ITER column should show CT - PARAM.input.esolver_type = "sdft"; - PARAM.input.nbands = 0; - PARAM.input.ks_solver = "cg"; + inp.esolver_type = "sdft"; + inp.nbands = 0; + inp.ks_solver = "cg"; testing::internal::CaptureStdout(); elecstate::print_etot(ucell.magnet, elecstate, converged, iter, scf_thr, - scf_thr_kin, duration, pw_diag_thr, avg_iter, print); + scf_thr_kin, duration, inp, two_fermi, pw_diag_thr, avg_iter, print); output = testing::internal::GetCapturedStdout(); EXPECT_THAT(output, testing::HasSubstr("CT")); @@ -303,17 +305,17 @@ TEST_F(ElecStatePrintTest, PrintEtotSDFTMixed) elecstate.charge->nrxx = 100; elecstate.charge->nxyz = 1000; - PARAM.input.out_freq_elec = 1; - PARAM.input.nspin = 1; + inp.out_freq_elec = 1; + inp.nspin = 1; GlobalV::MY_RANK = 0; // Mixed SDFT: nbands>0, still diagonalizes KS orbitals -> ITER column shows ks_solver label - PARAM.input.esolver_type = "sdft"; - PARAM.input.nbands = 5; - PARAM.input.ks_solver = "dav"; + inp.esolver_type = "sdft"; + inp.nbands = 5; + inp.ks_solver = "dav"; testing::internal::CaptureStdout(); elecstate::print_etot(ucell.magnet, elecstate, converged, iter, scf_thr, - scf_thr_kin, duration, pw_diag_thr, avg_iter, print); + scf_thr_kin, duration, inp, two_fermi, pw_diag_thr, avg_iter, print); output = testing::internal::GetCapturedStdout(); EXPECT_THAT(output, testing::HasSubstr("DA")); diff --git a/source/source_estate/test/elecstate_pw_test.cpp b/source/source_estate/test/elecstate_pw_test.cpp index ee59cce74f4..1c47c8356b8 100644 --- a/source/source_estate/test/elecstate_pw_test.cpp +++ b/source/source_estate/test/elecstate_pw_test.cpp @@ -2,14 +2,26 @@ #include "gmock/gmock.h" #include "gtest/gtest.h" -#define private public #define protected public #include "source_estate/elecstate_pw.h" +#include "source_basis/module_pw/pw_basis_big.h" // use PW_Basis_Big #include "source_hamilt/module_xc/xc_functional.h" #include "source_pw/module_pwdft/vl_pw.h" #include "source_pw/module_pwdft/vnl_pw.h" #include "source_pw/module_pwdft/soc.h" #include "source_io/module_parameter/parameter.h" + +/// @brief Friend helper to mutate PARAM private members in unit tests. +/// @details Parameter grants friend access to TestParameters so the test can +/// modify input/sys fields without `#define private public`. The class must +/// stay at global scope to match the friend declaration in parameter.h; +/// an anonymous-namespace class would not be the friend. +class TestParameters +{ + public: + static Input_para& input() { return PARAM.input; } + static System_para& sys() { return PARAM.sys; } +}; // mock functions for testing int XC_Functional::func_type = 1; namespace elecstate @@ -105,6 +117,7 @@ template <> void pseudopot_cell_vnl::getvnl(base_device::DEVICE_CPU*, const UnitCell&, int const&, + const ModuleBase::Vector3&, std::complex*) const { } @@ -112,6 +125,7 @@ template <> void pseudopot_cell_vnl::getvnl(base_device::DEVICE_CPU*, const UnitCell&, int const&, + const ModuleBase::Vector3&, std::complex*) const { } @@ -123,46 +137,41 @@ Fcoef::~Fcoef() } #include "source_cell/klist.h" -void Charge::set_rho_core(const UnitCell& ucell, ModuleBase::ComplexMatrix const&, const bool*) -{ -} void Charge::init_rho(const UnitCell&, const Parallel_Grid&, ModuleBase::ComplexMatrix const&, ModuleSymmetry::Symmetry& symm, const void*, - const void*) + const void*, + const module_charge::InitRhoCfg&) { } void Charge::set_rhopw(ModulePW::PW_Basis*) { } -void Charge::renormalize_rho() -{ -} -void Charge::check_rho() +void Charge::renormalize_rho(const double, const double) { } void Set_GlobalV_Default() { - PARAM.input.device = "cpu"; - PARAM.input.precision = "double"; - PARAM.sys.domag = false; - PARAM.sys.domag_z = false; + TestParameters::input().device = "cpu"; + TestParameters::input().precision = "double"; + TestParameters::sys().domag = false; + TestParameters::sys().domag_z = false; // Base class dependent - PARAM.input.nspin = 1; - PARAM.input.nelec = 10.0; - PARAM.input.nupdown = 0.0; - PARAM.sys.two_fermi = false; - PARAM.input.nbands = 6; - PARAM.sys.nlocal = 6; - PARAM.input.esolver_type = "ksdft"; - PARAM.input.lspinorb = false; - PARAM.input.basis_type = "pw"; + TestParameters::input().nspin = 1; + TestParameters::input().nelec = 10.0; + TestParameters::input().nupdown = 0.0; + TestParameters::sys().two_fermi = false; + TestParameters::input().nbands = 6; + TestParameters::sys().nlocal = 6; + TestParameters::input().esolver_type = "ksdft"; + TestParameters::input().lspinorb = false; + TestParameters::input().basis_type = "pw"; GlobalV::KPAR = 1; GlobalV::NPROC_IN_POOL = 1; - PARAM.sys.use_uspp = false; + TestParameters::sys().use_uspp = false; } /************************************************ @@ -263,7 +272,7 @@ TEST_F(ElecStatePWTest, ConstructorSingle) TEST_F(ElecStatePWTest, InitRhoDataDouble) { - XC_Functional::func_type = 3; + XC_Functional::set_func_type(3); chg->nrxx = 1000; elecstate_pw_d = new elecstate::ElecStatePW, base_device::DEVICE_CPU>(wfcpw, chg, @@ -280,9 +289,9 @@ TEST_F(ElecStatePWTest, InitRhoDataDouble) TEST_F(ElecStatePWTest, InitRhoDataSingle) { - PARAM.input.precision = "single"; - XC_Functional::func_type = 3; - chg->nspin = PARAM.input.nspin; + TestParameters::input().precision = "single"; + XC_Functional::set_func_type(3); + chg->nspin = TestParameters::input().nspin; chg->nrxx = 1000; elecstate_pw_s = new elecstate::ElecStatePW, base_device::DEVICE_CPU>(wfcpw, chg, diff --git a/source/source_estate/test/gint_prec_ctrl_test.cpp b/source/source_estate/test/gint_prec_ctrl_test.cpp deleted file mode 100644 index f471db80623..00000000000 --- a/source/source_estate/test/gint_prec_ctrl_test.cpp +++ /dev/null @@ -1,63 +0,0 @@ -#include "gtest/gtest.h" - -#include "../module_charge/gint_prec_ctrl.h" - -TEST(GintPrecisionControllerTest, AutoModeSwitchesToFp64ImmediatelyWhenDrhoIsSmallEnough) -{ - GintPrecisionController controller; - - controller.set_mode("mix"); - controller.reset_for_new_scf(); - EXPECT_EQ(controller.current_precision(), ModuleGint::GintPrecision::fp32); - - EXPECT_TRUE(controller.update_after_iteration(9.0e-5, 1.0e-7)); - EXPECT_EQ(controller.current_precision(), ModuleGint::GintPrecision::fp64); -} - -TEST(GintPrecisionControllerTest, DefaultModeStartsAndStaysFp64) -{ - GintPrecisionController controller; - - controller.reset_for_new_scf(); - EXPECT_EQ(controller.current_precision(), ModuleGint::GintPrecision::fp64); - - EXPECT_FALSE(controller.update_after_iteration(9.0e-5, 1.0e-7)); - EXPECT_EQ(controller.current_precision(), ModuleGint::GintPrecision::fp64); -} - -TEST(GintPrecisionControllerTest, SingleModeStartsAndStaysFp32) -{ - GintPrecisionController controller; - - controller.set_mode("single"); - controller.reset_for_new_scf(); - EXPECT_EQ(controller.current_precision(), ModuleGint::GintPrecision::fp32); - - EXPECT_FALSE(controller.update_after_iteration(9.0e-5, 1.0e-7)); - EXPECT_EQ(controller.current_precision(), ModuleGint::GintPrecision::fp32); -} - -TEST(GintPrecisionControllerTest, MixModeLocksFp64AfterSwitch) -{ - GintPrecisionController controller; - - controller.set_mode("mix"); - controller.reset_for_new_scf(); - EXPECT_TRUE(controller.update_after_iteration(9.0e-5, 1.0e-6)); - EXPECT_EQ(controller.current_precision(), ModuleGint::GintPrecision::fp64); - - // After locking, should return false - EXPECT_FALSE(controller.update_after_iteration(1.0, 1.0e-6)); - EXPECT_EQ(controller.current_precision(), ModuleGint::GintPrecision::fp64); -} - -TEST(GintPrecisionControllerTest, MixModeReturnsFalseWhenDrhoTooLarge) -{ - GintPrecisionController controller; - - controller.set_mode("mix"); - controller.reset_for_new_scf(); - // drho is large, should not switch yet - EXPECT_FALSE(controller.update_after_iteration(1.0, 1.0e-7)); - EXPECT_EQ(controller.current_precision(), ModuleGint::GintPrecision::fp32); -} diff --git a/source/source_estate/test/potential_new_test.cpp b/source/source_estate/test/potential_new_test.cpp index cb95bf6302e..6dc01b9f321 100644 --- a/source/source_estate/test/potential_new_test.cpp +++ b/source/source_estate/test/potential_new_test.cpp @@ -2,6 +2,7 @@ #include "source_hamilt/module_xc/xc_functional.h" #include "source_io/module_parameter/parameter.h" +#include "source_basis/module_pw/pw_basis_sup.h" #include "gtest/gtest.h" #include diff --git a/source/source_estate/test/support/charge-density.dat b/source/source_estate/test/support/charge-density.dat deleted file mode 100644 index ee3b0258011..00000000000 Binary files a/source/source_estate/test/support/charge-density.dat and /dev/null differ diff --git a/source/source_estate/test/td_field_test.cpp b/source/source_estate/test/td_field_test.cpp index 5ce67e53e4e..1e66fe821c9 100644 --- a/source/source_estate/test/td_field_test.cpp +++ b/source/source_estate/test/td_field_test.cpp @@ -57,7 +57,7 @@ TEST(TDFieldManagerTest, MixedFieldsSumAndAccumulate) input.td_heavi_t0 = {10.0}; input.td_heavi_amp = {3.0}; - const double amplitude_conversion = ModuleBase::BOHR_TO_A / ModuleBase::Ry_to_eV; + const double amplitude_conversion = ModuleBase::BOHR_TO_A / ModuleBase::Hartree_to_eV; const double total_field = 5.0 * amplitude_conversion; std::shared_ptr manager = elecstate::create_td_field_manager(input); @@ -67,17 +67,17 @@ TEST(TDFieldManagerTest, MixedFieldsSumAndAccumulate) EXPECT_EQ(manager->fields()[0].subdivisions(), 2); EXPECT_EQ(manager->fields()[1].subdivisions(), 2); - manager->advance_vector_gauge(); - ASSERT_EQ(manager->field_values().size(), 2U); - EXPECT_NEAR(manager->field_values()[0], 2.0 * amplitude_conversion, 1.0e-14); - EXPECT_NEAR(manager->field_values()[1], 3.0 * amplitude_conversion, 1.0e-14); - EXPECT_NEAR(manager->electric_field()[0], total_field, 1.0e-14); - EXPECT_NEAR(manager->total_electric_field()[0], total_field, 1.0e-14); - EXPECT_NEAR(manager->vector_potential_laststep()[0], -total_field, 1.0e-14); - EXPECT_NEAR(manager->vector_potential()[0], -0.5 * total_field, 1.0e-14); + manager->prepare_interval(0, 0); + ASSERT_EQ(manager->field_vals_ha().size(), 2U); + EXPECT_NEAR(manager->field_vals_ha()[0], 2.0 * amplitude_conversion, 1.0e-14); + EXPECT_NEAR(manager->field_vals_ha()[1], 3.0 * amplitude_conversion, 1.0e-14); + EXPECT_NEAR(manager->efield_ha()[0], total_field, 1.0e-14); + EXPECT_NEAR(manager->efield_ha()[0], total_field, 1.0e-14); + EXPECT_NEAR(manager->A_right_ha()[0], -total_field, 1.0e-14); + EXPECT_NEAR(manager->A_prop_ha()[0], -0.5 * total_field, 1.0e-14); - manager->advance_vector_gauge(); - EXPECT_NEAR(manager->vector_potential()[0], -1.5 * total_field, 1.0e-14); + manager->prepare_interval(1, 1); + EXPECT_NEAR(manager->A_prop_ha()[0], -1.5 * total_field, 1.0e-14); Input_para oscillatory_input; oscillatory_input.td_vext = true; @@ -101,22 +101,23 @@ TEST(TDFieldManagerTest, RestartRequiresCompleteState) { Input_para input; input.td_stype = 1; + input.td_dt = ModuleBase::AU_to_FS; std::shared_ptr manager = elecstate::create_td_field_manager(input); const std::string restart_prefix = "td_field_manager_test_"; const std::string restart_path = restart_prefix + "Restart_td.txt"; { std::ofstream output(restart_path.c_str()); - output << "7\n1 2 3\n4 5 6\n"; + output << "# Hartree units\n\n7 7 1 1 0\n1 2 3 # A_left\n4 5 6 # A_right\n"; } manager->read_restart(restart_prefix); - EXPECT_EQ(manager->current_step(), 6); - EXPECT_DOUBLE_EQ(manager->vector_potential()[0], 1.0); - EXPECT_DOUBLE_EQ(manager->vector_potential()[1], 2.0); - EXPECT_DOUBLE_EQ(manager->vector_potential()[2], 3.0); - EXPECT_DOUBLE_EQ(manager->vector_potential_laststep()[0], -4.0); - EXPECT_DOUBLE_EQ(manager->vector_potential_laststep()[1], -5.0); - EXPECT_DOUBLE_EQ(manager->vector_potential_laststep()[2], -6.0); + EXPECT_EQ(manager->current_step(), 7); + EXPECT_DOUBLE_EQ(manager->A_left_ha()[0], 1.0); + EXPECT_DOUBLE_EQ(manager->A_left_ha()[1], 2.0); + EXPECT_DOUBLE_EQ(manager->A_left_ha()[2], 3.0); + EXPECT_DOUBLE_EQ(manager->A_right_ha()[0], 4.0); + EXPECT_DOUBLE_EQ(manager->A_right_ha()[1], 5.0); + EXPECT_DOUBLE_EQ(manager->A_right_ha()[2], 6.0); { std::ofstream output(restart_path.c_str(), std::ofstream::out); diff --git a/source/source_estate/test/test_occ_comput.cpp b/source/source_estate/test/test_occ_comput.cpp new file mode 100644 index 00000000000..1b556f794d0 --- /dev/null +++ b/source/source_estate/test/test_occ_comput.cpp @@ -0,0 +1,191 @@ +#include "gtest/gtest.h" +#include "source_estate/occ_comput.h" + +#include +#include + +/*************************************************************** + * unit test of elecstate::occ_from_proj + * + * The function accumulates, for one k-point, the per-projector 2x2 + * occupation blocks + * rho^{ss'}_{iprj} = sum_i w_i conj(proj^s_{i,iprj}) proj^{s'}_{i,iprj} + * stored as occ_block[iprj*4 + {0,1,2,3}] = {up-up, up-dn, dn-up, dn-dn}. + ****************************************************************/ + +class OccComputTest : public ::testing::Test +{ + protected: + // two atoms with nh = {2, 3} -> tot_nproj = 5 + const int nat = 2; + const int nh[2] = {2, 3}; + const int nkb = 5; + const int nbands = 2; + + // proj layout: (nbands*npol) x nkb, spinor components offset by nkb. + // Values are simple distinct numbers so wrong indexing is caught. + std::vector> make_proj(const int npol) + { + std::vector> proj(nbands * npol * nkb); + for (size_t i = 0; i < proj.size(); i++) + { + proj[i] = std::complex(0.1 * (i + 1), 0.01 * (i + 1)); + } + return proj; + } + + // reference implementation copied from the original inline loops in + // OnsiteProjector::cal_occupations (onsite_proj_overlap.cpp), used as + // an oracle for the extracted free function. + void reference(const std::complex* proj, + const double* wg_ik, + const int npol, + const int nspin, + const int isk, + std::vector>& occs) + { + for (int ib = 0; ib < nbands; ib++) + { + const double weight = wg_ik[ib]; + int begin_iprj = 0; + for (int iat = 0; iat < nat; iat++) + { + const int nprj = nh[iat]; + for (int iprj = 0; iprj < nprj; iprj++) + { + const int occ_index = (begin_iprj + iprj) * 4; + if (npol == 1) + { + const int index = ib * nkb + begin_iprj + iprj; + const double occ = weight * (std::conj(proj[index]) * proj[index]).real(); + if (nspin == 2 && isk == 1) + { + occs[occ_index + 3] += occ; + } + else if (nspin == 1) + { + occs[occ_index] += 0.5 * occ; + occs[occ_index + 3] += 0.5 * occ; + } + else + { + occs[occ_index] += occ; + } + } + else + { + const int index = ib * 2 * nkb + begin_iprj + iprj; + occs[occ_index] += weight * std::conj(proj[index]) * proj[index]; + occs[occ_index + 1] += weight * std::conj(proj[index]) * proj[index + nkb]; + occs[occ_index + 2] += weight * std::conj(proj[index + nkb]) * proj[index]; + occs[occ_index + 3] += weight * std::conj(proj[index + nkb]) * proj[index + nkb]; + } + } + begin_iprj += nprj; + } + } + } + + void run_and_compare(const int npol, const int nspin, const int isk) + { + const double wg[2] = {1.5, 0.5}; + std::vector> proj = make_proj(npol); + std::vector> got(nkb * 4, std::complex(0.0, 0.0)); + std::vector> want(nkb * 4, std::complex(0.0, 0.0)); + + elecstate::occ_from_proj(proj.data(), wg, nbands, npol, nkb, + nspin, isk, nh, nat, got.data()); + reference(proj.data(), wg, npol, nspin, isk, want); + + for (int i = 0; i < nkb * 4; i++) + { + EXPECT_NEAR(got[i].real(), want[i].real(), 1e-12) << "element " << i; + EXPECT_NEAR(got[i].imag(), want[i].imag(), 1e-12) << "element " << i; + } + } +}; + +TEST_F(OccComputTest, Nspin1SplitsEvenly) +{ + run_and_compare(1, 1, 0); + + // additionally verify the even split and zero magnetization explicitly + const double wg[2] = {1.0, 1.0}; + std::vector> proj = make_proj(1); + std::vector> occs(nkb * 4, std::complex(0.0, 0.0)); + elecstate::occ_from_proj(proj.data(), wg, nbands, 1, nkb, 1, 0, nh, nat, occs.data()); + for (int iprj = 0; iprj < nkb; iprj++) + { + EXPECT_NEAR(occs[iprj * 4].real(), occs[iprj * 4 + 3].real(), 1e-12); + EXPECT_NEAR(occs[iprj * 4 + 1].real(), 0.0, 1e-12); + EXPECT_NEAR(occs[iprj * 4 + 2].real(), 0.0, 1e-12); + } +} + +TEST_F(OccComputTest, Nspin2SpinUpGoesToBlock0) +{ + run_and_compare(1, 2, 0); + + const double wg[2] = {1.0, 1.0}; + std::vector> proj = make_proj(1); + std::vector> occs(nkb * 4, std::complex(0.0, 0.0)); + elecstate::occ_from_proj(proj.data(), wg, nbands, 1, nkb, 2, 0, nh, nat, occs.data()); + for (int iprj = 0; iprj < nkb; iprj++) + { + EXPECT_GT(occs[iprj * 4].real(), 0.0); + EXPECT_NEAR(occs[iprj * 4 + 3].real(), 0.0, 1e-12); + } +} + +TEST_F(OccComputTest, Nspin2SpinDownGoesToBlock3) +{ + run_and_compare(1, 2, 1); + + const double wg[2] = {1.0, 1.0}; + std::vector> proj = make_proj(1); + std::vector> occs(nkb * 4, std::complex(0.0, 0.0)); + elecstate::occ_from_proj(proj.data(), wg, nbands, 1, nkb, 2, 1, nh, nat, occs.data()); + for (int iprj = 0; iprj < nkb; iprj++) + { + EXPECT_NEAR(occs[iprj * 4].real(), 0.0, 1e-12); + EXPECT_GT(occs[iprj * 4 + 3].real(), 0.0); + } +} + +TEST_F(OccComputTest, Nspin4FillsAllBlocks) +{ + run_and_compare(2, 4, 0); + + // off-diagonal blocks must be nonzero for generic spinor coefficients + const double wg[2] = {1.0, 1.0}; + std::vector> proj = make_proj(2); + std::vector> occs(nkb * 4, std::complex(0.0, 0.0)); + elecstate::occ_from_proj(proj.data(), wg, nbands, 2, nkb, 4, 0, nh, nat, occs.data()); + for (int iprj = 0; iprj < nkb; iprj++) + { + EXPECT_GT(occs[iprj * 4].real(), 0.0); + EXPECT_GT(occs[iprj * 4 + 3].real(), 0.0); + // hermiticity: rho^{up,dn} = conj(rho^{dn,up}) + EXPECT_NEAR(occs[iprj * 4 + 1].real(), occs[iprj * 4 + 2].real(), 1e-12); + EXPECT_NEAR(occs[iprj * 4 + 1].imag(), -occs[iprj * 4 + 2].imag(), 1e-12); + } +} + +TEST_F(OccComputTest, AccumulatesAcrossCalls) +{ + // two calls with weight 1 must equal one call with weight 2 per band + const double wg1[2] = {1.0, 1.0}; + const double wg2[2] = {2.0, 2.0}; + std::vector> proj = make_proj(1); + std::vector> twice(nkb * 4, std::complex(0.0, 0.0)); + std::vector> once(nkb * 4, std::complex(0.0, 0.0)); + + elecstate::occ_from_proj(proj.data(), wg1, nbands, 1, nkb, 2, 0, nh, nat, twice.data()); + elecstate::occ_from_proj(proj.data(), wg1, nbands, 1, nkb, 2, 0, nh, nat, twice.data()); + elecstate::occ_from_proj(proj.data(), wg2, nbands, 1, nkb, 2, 0, nh, nat, once.data()); + + for (int i = 0; i < nkb * 4; i++) + { + EXPECT_NEAR(twice[i].real(), once[i].real(), 1e-12); + } +} diff --git a/source/source_estate/test/test_occ_mixer.cpp b/source/source_estate/test/test_occ_mixer.cpp index 0af0904b8bb..33f9a36e4e7 100644 --- a/source/source_estate/test/test_occ_mixer.cpp +++ b/source/source_estate/test/test_occ_mixer.cpp @@ -107,8 +107,8 @@ TEST_F(OccMatMixerTest, MixPlainNspin1) { for (int m = 0; m < block; m++) { - occmat.data()[iat][l_corr][0][0].c[m] = 1.0 + m; - occmat.data_save()[iat][l_corr][0][0].c[m] = 100.0 + m; + occmat.data()[iat][l_corr][0].c[m] = 1.0 + m; + occmat.data_save()[iat][l_corr][0].c[m] = 100.0 + m; } } @@ -120,7 +120,7 @@ TEST_F(OccMatMixerTest, MixPlainNspin1) for (int m = 0; m < block; m++) { const double expect = (1.0 + m) * beta + (100.0 + m) * (1.0 - beta); - EXPECT_DOUBLE_EQ(occmat.data()[iat][l_corr][0][0].c[m], expect); + EXPECT_DOUBLE_EQ(occmat.data()[iat][l_corr][0].c[m], expect); } } } @@ -141,8 +141,8 @@ TEST_F(OccMatMixerTest, MixPlainNspin2BothChannels) { for (int m = 0; m < block; m++) { - occmat.data()[iat][l_corr][0][is].c[m] = 2.0 + is + m; - occmat.data_save()[iat][l_corr][0][is].c[m] = 50.0 + is + m; + occmat.data()[iat][l_corr][is].c[m] = 2.0 + is + m; + occmat.data_save()[iat][l_corr][is].c[m] = 50.0 + is + m; } } } @@ -157,7 +157,7 @@ TEST_F(OccMatMixerTest, MixPlainNspin2BothChannels) for (int m = 0; m < block; m++) { const double expect = (2.0 + is + m) * beta + (50.0 + is + m) * (1.0 - beta); - EXPECT_DOUBLE_EQ(occmat.data()[iat][l_corr][0][is].c[m], expect); + EXPECT_DOUBLE_EQ(occmat.data()[iat][l_corr][is].c[m], expect); } } } @@ -184,7 +184,7 @@ TEST_F(OccMatMixerTest, FlatRoundtripNspin2) { for (int m = 0; m < block; m++) { - occmat.data()[iat][l_corr][0][is].c[m] = + occmat.data()[iat][l_corr][is].c[m] = 1000.0 * iat + 100.0 * is + m; } } @@ -201,7 +201,7 @@ TEST_F(OccMatMixerTest, FlatRoundtripNspin2) { for (int m = 0; m < block; m++) { - EXPECT_DOUBLE_EQ(occmat.data()[iat][l_corr][0][is].c[m], + EXPECT_DOUBLE_EQ(occmat.data()[iat][l_corr][is].c[m], 1000.0 * iat + 100.0 * is + m); } } @@ -223,7 +223,7 @@ TEST_F(OccMatMixerTest, BeginIterFlattensSave) for (int m = 0; m < block; m++) { - occmat.data_save()[0][l_corr][0][0].c[m] = 7.0 + m; + occmat.data_save()[0][l_corr][0].c[m] = 7.0 + m; } mixer.begin_iter(occmat); diff --git a/source/source_estate/test/test_rhog_io.cpp b/source/source_estate/test/test_rhog_io.cpp deleted file mode 100644 index 633530dba5d..00000000000 --- a/source/source_estate/test/test_rhog_io.cpp +++ /dev/null @@ -1,406 +0,0 @@ -#include "gmock/gmock.h" -#include "gtest/gtest.h" -#include "source_estate/rhog_io.h" -#include "source_base/module_parallel/para_world.h" -#include "source_base/module_parallel/para_tag.h" -#include "source_base/module_parallel/para_bridge.h" -#ifdef __MPI -#include "source_basis/module_pw/test/test_tool.h" -#include "mpi.h" -#endif -#include -#include - -/** - * - Tested Functions: - * - read_rhog() - * - write_rhog() - */ - -class ReadRhogTest : public ::testing::Test -{ - protected: - ModulePW::PW_Basis rhopw; - std::vector>> rhog_data; - std::vector*> rhog; - Parallel::ParaWorld pw_world = Parallel::make_pw_world(); - std::ofstream warning_stream; - - void setup_pw_basis() - { -#ifdef __MPI - rhopw.initmpi(pw_world.size(), pw_world.rank(), pw_world.comm()); -#endif - rhopw.initgrids(6.5, ModuleBase::Matrix3(-0.5, 0.0, 0.5, 0.0, 0.5, 0.5, -0.5, 0.5, 0.0), 120); - rhopw.initparameters(false, 120); - rhopw.setuptransform(); - rhopw.collect_local_pw(); - } - - void open_warning(const std::string& path) - { - warning_stream.open(path); - } - - void close_warning() - { - if (warning_stream.is_open()) - { - warning_stream.close(); - } - } - - std::string read_warning_file(const std::string& path) - { - std::ifstream ifs(path); - std::stringstream ss; - ss << ifs.rdbuf(); - ifs.close(); - return ss.str(); - } - - virtual void SetUp() - { - rhog_data.resize(1, std::vector>(1471)); - rhog.push_back(rhog_data[0].data()); - } - - virtual void TearDown() - { - close_warning(); - } -}; - -// Test the read_rhog function with normal file -TEST_F(ReadRhogTest, ReadRhog) -{ - std::string filename = "./support/charge-density.dat"; - setup_pw_basis(); - - bool result = elecstate::read_rhog(filename, &rhopw, 1, rhog.data(), pw_world, nullptr); - - EXPECT_TRUE(result); - EXPECT_DOUBLE_EQ(rhog[0][0].real(), -1.0304462993299456e-05); - EXPECT_DOUBLE_EQ(rhog[0][0].imag(), -1.2701788626185278e-13); - EXPECT_DOUBLE_EQ(rhog[0][1].real(), -0.0003875762482855959); - EXPECT_DOUBLE_EQ(rhog[0][1].imag(), -4.2556814316812048e-12); - EXPECT_DOUBLE_EQ(rhog[0][1470].real(), -3.5683133614445107e-05); - EXPECT_DOUBLE_EQ(rhog[0][1470].imag(), 1.6176615686863767e-12); -} - -// Test the read_rhog function when the file is not found -TEST_F(ReadRhogTest, NotFoundFile) -{ - setup_pw_basis(); - std::string filename = "notfound.txt"; - - open_warning("test_read_rhog.txt"); - bool result = elecstate::read_rhog(filename, &rhopw, 1, rhog.data(), pw_world, &warning_stream); - close_warning(); - - std::string expected_content = " elecstate::read_rhog warning : Can't open file notfound.txt\n"; - EXPECT_FALSE(result); - EXPECT_EQ(read_warning_file("test_read_rhog.txt"), expected_content); - std::remove("test_read_rhog.txt"); -} - -// Test the read_rhog function when gamma_only is inconsistent -TEST_F(ReadRhogTest, InconsistentGammaOnly) -{ - setup_pw_basis(); - std::string filename = "./support/charge-density.dat"; - rhopw.gamma_only = true; - // Fewer planewaves than the file holds (1471) triggers the - // "some planewaves in file are not used" warning. - rhopw.npwtot = 1000; - - open_warning("test_read_rhog.txt"); - bool result = elecstate::read_rhog(filename, &rhopw, 2, rhog.data(), pw_world, &warning_stream); - close_warning(); - - std::string expected_content - = " elecstate::read_rhog warning : some planewaves in file are not used\n elecstate::read_rhog warning : some " - "spin channels in file are missing\n elecstate::read_rhog warning : gamma_only read from file is " - "inconsistent with INPUT\n"; - - EXPECT_FALSE(result); - EXPECT_EQ(read_warning_file("test_read_rhog.txt"), expected_content); - std::remove("test_read_rhog.txt"); -} - -// Test the read_rhog function when some planewaves in file are missing -TEST_F(ReadRhogTest, SomePWMissing) -{ - setup_pw_basis(); - std::string filename = "./support/charge-density.dat"; - rhopw.npwtot = 2000; - - open_warning("test_read_rhog.txt"); - bool result = elecstate::read_rhog(filename, &rhopw, 1, rhog.data(), pw_world, &warning_stream); - close_warning(); - - std::string expected_content = " elecstate::read_rhog warning : some planewaves in file are missing\n"; - EXPECT_TRUE(result); - EXPECT_EQ(read_warning_file("test_read_rhog.txt"), expected_content); - std::remove("test_read_rhog.txt"); -} - -// Test read_rhog with os_warning=nullptr (silent mode, must not crash) -TEST_F(ReadRhogTest, OsNullptrSilent) -{ - std::string filename = "notfound.txt"; - bool result = elecstate::read_rhog(filename, &rhopw, 1, rhog.data(), pw_world, nullptr); - EXPECT_FALSE(result); -} - -// Test write_rhog round-trip: write then read back, verify data consistency -TEST_F(ReadRhogTest, WriteRoundTrip) -{ - setup_pw_basis(); - - // initialize some rhog data - rhog_data[0].assign(rhopw.npw, std::complex(1.5, 2.5)); - - std::string tmpfile = "test_rhog_roundtrip.dat"; - - // write - bool write_result = elecstate::write_rhog( - tmpfile, rhopw.gamma_only, &rhopw, 1, - ModuleBase::Matrix3(-0.5, 0.0, 0.5, 0.0, 0.5, 0.5, -0.5, 0.5, 0.0), - rhog.data(), pw_world, nullptr); - EXPECT_TRUE(write_result); - - // read back into a fresh buffer - std::vector>> rhog_read_data( - 1, std::vector>(rhopw.npw)); - std::vector*> rhog_read; - rhog_read.push_back(rhog_read_data[0].data()); - - bool read_result = elecstate::read_rhog(tmpfile, &rhopw, 1, rhog_read.data(), pw_world, nullptr); - EXPECT_TRUE(read_result); - - // compare: within MPI precision tolerance - int diff_count = 0; - for (int ig = 0; ig < rhopw.npw; ++ig) - { - if (std::abs(rhog[0][ig] - rhog_read[0][ig]) > 1e-10) - { - ++diff_count; - } - } - EXPECT_EQ(diff_count, 0) << diff_count << " planewave values differ after round-trip"; - - std::remove(tmpfile.c_str()); -} - -// Test write_rhog when the output path is not writable -TEST_F(ReadRhogTest, WriteFileFail) -{ - setup_pw_basis(); - rhog_data[0].assign(rhopw.npw, std::complex(1.0, 0.0)); - - // try to write to a directory path (not a file) — should fail - bool result = elecstate::write_rhog( - "/tmp", rhopw.gamma_only, &rhopw, 1, - ModuleBase::Matrix3(-0.5, 0.0, 0.5, 0.0, 0.5, 0.5, -0.5, 0.5, 0.0), - rhog.data(), pw_world, nullptr); - EXPECT_FALSE(result); -} - -// Test write_rhog with nspin=2, round-trip both channels -TEST_F(ReadRhogTest, WriteRoundTripNspin2) -{ - setup_pw_basis(); - - // expand to nspin=2 - rhog_data.resize(2, std::vector>(rhopw.npw)); - rhog.clear(); - rhog.push_back(rhog_data[0].data()); - rhog.push_back(rhog_data[1].data()); - - // initialize distinct values for each spin channel - for (int ig = 0; ig < rhopw.npw; ++ig) - { - rhog_data[0][ig] = std::complex(1.0 * ig, 0.1 * ig); - rhog_data[1][ig] = std::complex(2.0 * ig, 0.2 * ig); - } - - std::string tmpfile = "test_rhog_roundtrip_nspin2.dat"; - - // write nspin=2 - bool write_result = elecstate::write_rhog( - tmpfile, rhopw.gamma_only, &rhopw, 2, - ModuleBase::Matrix3(-0.5, 0.0, 0.5, 0.0, 0.5, 0.5, -0.5, 0.5, 0.0), - rhog.data(), pw_world, nullptr); - EXPECT_TRUE(write_result); - - // read back - std::vector>> rhog_read_data( - 2, std::vector>(rhopw.npw)); - std::vector*> rhog_read; - rhog_read.push_back(rhog_read_data[0].data()); - rhog_read.push_back(rhog_read_data[1].data()); - - bool read_result = elecstate::read_rhog(tmpfile, &rhopw, 2, rhog_read.data(), pw_world, nullptr); - EXPECT_TRUE(read_result); - - int diff_count = 0; - for (int is = 0; is < 2; ++is) - { - for (int ig = 0; ig < rhopw.npw; ++ig) - { - if (std::abs(rhog[is][ig] - rhog_read[is][ig]) > 1e-10) - { - ++diff_count; - } - } - } - EXPECT_EQ(diff_count, 0) << diff_count << " planewave values differ after nspin=2 round-trip"; - - std::remove(tmpfile.c_str()); -} - -// Test write_rhog with nspin=4, round-trip all 4 channels -TEST_F(ReadRhogTest, WriteRoundTripNspin4) -{ - setup_pw_basis(); - - rhog_data.resize(4, std::vector>(rhopw.npw)); - rhog.clear(); - for (int is = 0; is < 4; ++is) - { - rhog.push_back(rhog_data[is].data()); - } - - // initialize distinct values for each spin channel - for (int is = 0; is < 4; ++is) - { - for (int ig = 0; ig < rhopw.npw; ++ig) - { - rhog_data[is][ig] = std::complex((is + 1) * 1.0 * ig, (is + 1) * 0.1 * ig); - } - } - - std::string tmpfile = "test_rhog_roundtrip_nspin4.dat"; - - bool write_result = elecstate::write_rhog( - tmpfile, rhopw.gamma_only, &rhopw, 4, - ModuleBase::Matrix3(-0.5, 0.0, 0.5, 0.0, 0.5, 0.5, -0.5, 0.5, 0.0), - rhog.data(), pw_world, nullptr); - EXPECT_TRUE(write_result); - - // read back as nspin=4 - std::vector>> rhog_read_data( - 4, std::vector>(rhopw.npw)); - std::vector*> rhog_read; - for (int is = 0; is < 4; ++is) - { - rhog_read.push_back(rhog_read_data[is].data()); - } - - bool read_result = elecstate::read_rhog(tmpfile, &rhopw, 4, rhog_read.data(), pw_world, nullptr); - EXPECT_TRUE(read_result); - - int diff_count = 0; - for (int is = 0; is < 4; ++is) - { - for (int ig = 0; ig < rhopw.npw; ++ig) - { - if (std::abs(rhog[is][ig] - rhog_read[is][ig]) > 1e-10) - { - ++diff_count; - } - } - } - EXPECT_EQ(diff_count, 0) << diff_count << " planewave values differ after nspin=4 round-trip"; - - std::remove(tmpfile.c_str()); -} - -// Test the special path L173-181: file nspin=2 read as input nspin=4 -// Expected behavior: rhog[0] preserved, rhog[1] and rhog[2] zeroed, -// rhog[3] <- old rhog[1] -TEST_F(ReadRhogTest, ReadRhogNspin2To4SpecialPath) -{ - setup_pw_basis(); - - // Step 1: write a nspin=2 binary with known values - rhog_data.resize(2, std::vector>(rhopw.npw)); - rhog.clear(); - rhog.push_back(rhog_data[0].data()); - rhog.push_back(rhog_data[1].data()); - - for (int ig = 0; ig < rhopw.npw; ++ig) - { - rhog_data[0][ig] = std::complex(10.0 + ig, 0.0); - rhog_data[1][ig] = std::complex(20.0 + ig, 0.0); - } - - std::string tmpfile = "test_rhog_nspin2_to_4.dat"; - - bool write_result = elecstate::write_rhog( - tmpfile, rhopw.gamma_only, &rhopw, 2, - ModuleBase::Matrix3(-0.5, 0.0, 0.5, 0.0, 0.5, 0.5, -0.5, 0.5, 0.0), - rhog.data(), pw_world, nullptr); - EXPECT_TRUE(write_result); - - // Step 2: read back as nspin=4 — triggers the L173-181 special path - std::vector>> rhog_read_data( - 4, std::vector>(rhopw.npw)); - std::vector*> rhog_read; - for (int is = 0; is < 4; ++is) - { - rhog_read.push_back(rhog_read_data[is].data()); - } - - bool read_result = elecstate::read_rhog(tmpfile, &rhopw, 4, rhog_read.data(), pw_world, nullptr); - EXPECT_TRUE(read_result); - - // Verify the special transformation at L173-181: - // rhog[0] <- file spin 0 - // rhog[1] <- ZEROED (was file spin 1, then ZEROS) - // rhog[2] <- ZEROED - // rhog[3] <- file spin 1 (copied before ZEROS) - for (int ig = 0; ig < rhopw.npw; ++ig) - { - // rhog[0] should match original spin 0 - EXPECT_NEAR(rhog_read_data[0][ig].real(), 10.0 + ig, 1e-10); - EXPECT_NEAR(rhog_read_data[0][ig].imag(), 0.0, 1e-10); - - // rhog[1] should be zeroed - EXPECT_NEAR(rhog_read_data[1][ig].real(), 0.0, 1e-10); - EXPECT_NEAR(rhog_read_data[1][ig].imag(), 0.0, 1e-10); - - // rhog[2] should be zeroed - EXPECT_NEAR(rhog_read_data[2][ig].real(), 0.0, 1e-10); - EXPECT_NEAR(rhog_read_data[2][ig].imag(), 0.0, 1e-10); - - // rhog[3] should equal original spin 1 (copied before zero) - EXPECT_NEAR(rhog_read_data[3][ig].real(), 20.0 + ig, 1e-10); - EXPECT_NEAR(rhog_read_data[3][ig].imag(), 0.0, 1e-10); - } - - std::remove(tmpfile.c_str()); -} - -int main(int argc, char** argv) -{ -#ifdef __MPI - int nproc = 1; - int myrank = 0; - int nproc_in_pool = 1; - int kpar = 1; - int mypool = 0; - int rank_in_pool = 0; - setupmpi(argc, argv, nproc, myrank); - divide_pools(nproc, myrank, nproc_in_pool, kpar, mypool, rank_in_pool); -#endif - - testing::InitGoogleTest(&argc, argv); - int result = RUN_ALL_TESTS(); - -#ifdef __MPI - finishmpi(); -#endif - return result; -} diff --git a/source/source_estate/update_pot.cpp b/source/source_estate/update_pot.cpp index 743b9d3f3bc..08fea3c3b89 100644 --- a/source/source_estate/update_pot.cpp +++ b/source/source_estate/update_pot.cpp @@ -2,8 +2,8 @@ #include "source_cell/cal_ux.h" void elecstate::update_pot(UnitCell& ucell, // unitcell - elecstate::ElecState* &pelec, // pointer of electrons - const Charge &chr, + elecstate::ElecState* &pelec, // pointer of electrons + const Charge &chr, const bool conv_esolver ) // charge density { diff --git a/source/source_estate/update_pot.h b/source/source_estate/update_pot.h index cce4c42428f..4b907bda8d0 100644 --- a/source/source_estate/update_pot.h +++ b/source/source_estate/update_pot.h @@ -8,8 +8,8 @@ namespace elecstate { void update_pot(UnitCell& ucell, // unitcell - elecstate::ElecState* &pelec, // pointer of electrons - const Charge &chr, + elecstate::ElecState* &pelec, // pointer of electrons + const Charge &chr, const bool conv_esolver); // charge density } diff --git a/source/source_estate/uspp_density.cpp b/source/source_estate/uspp_density.cpp index db605c2d88b..72a9180e842 100644 --- a/source/source_estate/uspp_density.cpp +++ b/source/source_estate/uspp_density.cpp @@ -81,7 +81,7 @@ void UsppProjector::accumulate(const int ik, assert(spin >= 0 && static_cast(becsum->size()) >= (spin + 1) * ucell_.nat * nh_tot); // Build beta_i(G+k) for this k point before projecting the pseudo-wavefunctions. T* vkb = ppcell_.template get_vkb_data(); - ppcell_.getvnl(static_cast(nullptr), ucell_, ik, vkb); + ppcell_.getvnl(static_cast(nullptr), ucell_, ik, ModuleBase::Vector3(0.0, 0.0, 0.0), vkb); // Each rank forms its PW contribution to ; the pool sum completes the overlap. // BLAS 'C' conjugate-transposes the projector matrix; stride includes any padding between states. T* overlaps = work.overlaps.template data(); diff --git a/source/source_hamilt/hamilt.h b/source/source_hamilt/hamilt.h index 3d554c0fe6b..0ba249553c3 100644 --- a/source/source_hamilt/hamilt.h +++ b/source/source_hamilt/hamilt.h @@ -31,13 +31,13 @@ class Hamilt : public HamiltBase void* get_ops() override { return static_cast(ops); } /// core function: for solving eigenvalues of Hamiltonian with iterative method - virtual void hPsi( - const T* psi_in, - T* hpsi, - const size_t size) const - { - return; - } + virtual void hPsi( + const T* psi_in, + T* hpsi, + const size_t size) const + { + return; + } virtual void sPsi(const T* psi_in, // psi T* spsi, // spsi @@ -49,14 +49,14 @@ class Hamilt : public HamiltBase syncmem_op()(spsi, psi_in, static_cast(nbands * nrow)); } - /// core function: return H(k) and S(k) matrixs for direct solving eigenvalues. - virtual void matrix( - MatrixBlock> &hk_in, - MatrixBlock> &sk_in){return;} + /// core function: return H(k) and S(k) matrixs for direct solving eigenvalues. + virtual void matrix( + MatrixBlock> &hk_in, + MatrixBlock> &sk_in){return;} - virtual void matrix( - MatrixBlock &hk_in, - MatrixBlock &sk_in){return;} + virtual void matrix( + MatrixBlock &hk_in, + MatrixBlock &sk_in){return;} virtual std::vector matrix() { return std::vector(); } diff --git a/source/source_hamilt/hamilt_hs_adapter.h b/source/source_hamilt/hamilt_hs_adapter.h new file mode 100644 index 00000000000..918226d6de5 --- /dev/null +++ b/source/source_hamilt/hamilt_hs_adapter.h @@ -0,0 +1,84 @@ +#ifndef HAMILT_HS_ADAPTER_H +#define HAMILT_HS_ADAPTER_H + +#include "source_base/tool_quit.h" +#include "source_basis/module_pw/pw_basis_k.h" +#include "source_hamilt/hamilt.h" +#include "source_hsolver/hs_matrix.h" +#include "source_hsolver/hs_operator.h" +#include "source_psi/psi.h" + +namespace hamilt +{ + +/** + * @brief Presents a Hamilt as the H/S block-vector operator the iterative + * eigensolvers work on (hsolver::HSOperator). + * + * This is the only place that wraps raw pointers into Psi / hpsi_info for the + * operator chain; hsolver itself never sees Hamilt. + */ +template +class HamiltHSOperator : public hsolver::HSOperator +{ + public: + HamiltHSOperator(Hamilt* hm, const ModulePW::PW_Basis_K* wfc_basis) : hm_(hm), wfc_basis_(wfc_basis) + { + } + + void update_k(const int ik) override + { + hm_->updateHk(ik); + ik_ = ik; + npw_ = wfc_basis_->npwk[ik]; + } + + void hpsi(const T* x, T* hx, const int ld, const int nvec) const override + { + if (hm_->ops == nullptr) + { + ModuleBase::WARNING_QUIT("HamiltHSOperator::hpsi", "Operators in Hamilt are not allocated yet"); + } + // non-owning view of x: one k point, nvec bands, leading dimension ld, npw valid rows + psi::Psi x_view(const_cast(x), 1, nvec, ld, npw_); + typename Operator::hpsi_info info(&x_view, psi::Range(true, 0, 0, nvec - 1), hx); + hm_->ops->hPsi(info); + } + + void spsi(const T* x, T* sx, const int ld, const int nvec) const override + { + hm_->sPsi(x, sx, ld, npw_, nvec); + } + + protected: + Hamilt* hm_ = nullptr; + const ModulePW::PW_Basis_K* wfc_basis_ = nullptr; + int ik_ = 0; ///< k point set by the last update_k() + int npw_ = 0; ///< number of plane waves of that k point (without npol) +}; + +/** + * @brief Presents a Hamilt as the H(k)/S(k) matrix source the direct + * eigensolvers work on (hsolver::HSMatrix). + */ +template +class HamiltHSMatrix : public hsolver::HSMatrix +{ + public: + explicit HamiltHSMatrix(Hamilt* hm) : hm_(hm) + { + } + + void hs_at_k(const int ik, ModuleBase::MatrixBlock& hk, ModuleBase::MatrixBlock& sk) override + { + hm_->updateHk(ik); + hm_->matrix(hk, sk); + } + + private: + Hamilt* hm_ = nullptr; +}; + +} // namespace hamilt + +#endif // HAMILT_HS_ADAPTER_H diff --git a/source/source_hamilt/module_ewald/h_ewald_pw.cpp b/source/source_hamilt/module_ewald/h_ewald_pw.cpp index 459b77b2185..2511fe65170 100644 --- a/source/source_hamilt/module_ewald/h_ewald_pw.cpp +++ b/source/source_hamilt/module_ewald/h_ewald_pw.cpp @@ -79,7 +79,7 @@ double H_Ewald_pw::compute_ewald(const UnitCell& cell, ModuleBase::GlobalFunc::OUT(output_stream,"Total ionic charge",charge); } - // (2) calculate the converged value: alpha + // (2) calculate the converged value: alpha H_Ewald_pw::alpha = 2.90; do { @@ -112,17 +112,17 @@ double H_Ewald_pw::compute_ewald(const UnitCell& cell, ewaldg = 0.0; } - // in plane wave basis, only use k=0 point is not - // called "gamma_only", only if the wave functions - // are stored as double type, the gamma_only = true. - // I don't know why "gamma_only" in plane wave - // makes the fact below is 2, that's a little complicated - // to understand. I think that may because only half - // the G vectors are used. Unfortunately implement the - // function hasn't in my plan list yet. - // - // but that's not the term "gamma_only" I want to use in LCAO, - fact = 1.0; + // in plane wave basis, only use k=0 point is not + // called "gamma_only", only if the wave functions + // are stored as double type, the gamma_only = true. + // I don't know why "gamma_only" in plane wave + // makes the fact below is 2, that's a little complicated + // to understand. I think that may because only half + // the G vectors are used. Unfortunately implement the + // function hasn't in my plan list yet. + // + // but that's not the term "gamma_only" I want to use in LCAO, + fact = 1.0; //output_stream << "\n pwb.gstart = " << pwb.gstart << std::endl; const int ig0 = rho_basis->ig_gge0; @@ -148,14 +148,14 @@ double H_Ewald_pw::compute_ewald(const UnitCell& cell, // std::cout << "\n ewaldg = " << ewaldg; // Here add the other constant term - if (rho_basis->ig_gge0 >= 0) - { - for (int it = 0; it < cell.ntype;it++) - { + if (rho_basis->ig_gge0 >= 0) + { + for (int it = 0; it < cell.ntype;it++) + { { ewaldg = ewaldg - cell.atoms[it].na * cell.atoms[it].ncpp.zv * cell.atoms[it].ncpp.zv * sqrt(8.0 / ModuleBase::TWO_PI * alpha); } - } + } }//mohan modify 2007-11-7, 2010-07-26 // R-space sum here (only done for the processor that contains G=0) @@ -201,7 +201,7 @@ double H_Ewald_pw::compute_ewald(const UnitCell& cell, { it1 = cell.iat2it[na1]; ia1 = cell.iat2ia[na1]; - + for(int na2=0; na21) @@ -425,35 +425,35 @@ void H_Ewald_pw::rgen( ModuleBase::heapsort(nrm, r2, irr); } - // mohan fix bug 2011-06-07 - for(int i=0; i #include #include "big_grid.h" @@ -47,4 +49,6 @@ class BatchBigGrid static int max_atom_pairs_; }; -} \ No newline at end of file +} + +#endif // ABACUS_SOURCE_HAMILT_MODULE_GINT_BATCH_BIGGRID_H diff --git a/source/source_hamilt/module_gint/big_grid.h b/source/source_hamilt/module_gint/big_grid.h index 55bed7a2514..930cebd4bc5 100644 --- a/source/source_hamilt/module_gint/big_grid.h +++ b/source/source_hamilt/module_gint/big_grid.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_HAMILT_MODULE_GINT_BIG_GRID_H +#define ABACUS_SOURCE_HAMILT_MODULE_GINT_BIG_GRID_H + #include #include @@ -92,4 +94,6 @@ class BigGrid static std::shared_ptr biggrid_info_; }; -} // namespace ModuleGint \ No newline at end of file +} // namespace ModuleGint + +#endif // ABACUS_SOURCE_HAMILT_MODULE_GINT_BIG_GRID_H diff --git a/source/source_hamilt/module_gint/biggrid_info.h b/source/source_hamilt/module_gint/biggrid_info.h index 2f574a0188e..04642c1df5a 100644 --- a/source/source_hamilt/module_gint/biggrid_info.h +++ b/source/source_hamilt/module_gint/biggrid_info.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_HAMILT_MODULE_GINT_BIGGRID_INFO_H +#define ABACUS_SOURCE_HAMILT_MODULE_GINT_BIGGRID_INFO_H + #include #include "gint_type.h" @@ -98,4 +100,6 @@ class BigGridInfo std::vector mgrid_coords_; }; -} // namespace ModuleGint \ No newline at end of file +} // namespace ModuleGint + +#endif // ABACUS_SOURCE_HAMILT_MODULE_GINT_BIGGRID_INFO_H diff --git a/source/source_hamilt/module_gint/divide_info.h b/source/source_hamilt/module_gint/divide_info.h index 374831eeefc..65a42437dee 100644 --- a/source/source_hamilt/module_gint/divide_info.h +++ b/source/source_hamilt/module_gint/divide_info.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_HAMILT_MODULE_GINT_DIVIDE_INFO_H +#define ABACUS_SOURCE_HAMILT_MODULE_GINT_DIVIDE_INFO_H + #include "biggrid_info.h" #include "unitcell_info.h" @@ -51,4 +53,6 @@ class DivideInfo std::shared_ptr localcell_info_; }; -} \ No newline at end of file +} + +#endif // ABACUS_SOURCE_HAMILT_MODULE_GINT_DIVIDE_INFO_H diff --git a/source/source_hamilt/module_gint/gint.h b/source/source_hamilt/module_gint/gint.h index 1091a572a1e..0e4bc4d6012 100644 --- a/source/source_hamilt/module_gint/gint.h +++ b/source/source_hamilt/module_gint/gint.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_H +#define ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_H + #include #include #include "gint_info.h" @@ -32,4 +34,6 @@ class Gint static GintInfo* gint_info_; }; -} \ No newline at end of file +} + +#endif // ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_H diff --git a/source/source_hamilt/module_gint/gint_atom.h b/source/source_hamilt/module_gint/gint_atom.h index 2c702aaeb43..ad89ecb3694 100644 --- a/source/source_hamilt/module_gint/gint_atom.h +++ b/source/source_hamilt/module_gint/gint_atom.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_ATOM_H +#define ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_ATOM_H + #include "source_cell/atom_spec.h" #include "source_basis/module_ao/orb_atomic.h" @@ -133,3 +135,5 @@ class GintAtom }; } // namespace ModuleGint + +#endif // ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_ATOM_H diff --git a/source/source_hamilt/module_gint/gint_common.h b/source/source_hamilt/module_gint/gint_common.h index d7ddca5002a..48d4176232e 100644 --- a/source/source_hamilt/module_gint/gint_common.h +++ b/source/source_hamilt/module_gint/gint_common.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_COMMON_H +#define ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_COMMON_H + #include "source_hamilt/module_hcontainer/hcontainer.h" #include "gint_info.h" @@ -31,3 +33,5 @@ namespace ModuleGint template void wfc_2d_to_gint(const T* wfc_2d, int nbands, int nlocal, const Parallel_Orbitals& pv, T* wfc_grid, const GintInfo& gint_info); } + +#endif // ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_COMMON_H diff --git a/source/source_hamilt/module_gint/gint_drho.h b/source/source_hamilt/module_gint/gint_drho.h index 3ef800966ed..d080937f37b 100644 --- a/source/source_hamilt/module_gint/gint_drho.h +++ b/source/source_hamilt/module_gint/gint_drho.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_DRHO_H +#define ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_DRHO_H + #include #include "source_hamilt/module_hcontainer/hcontainer.h" @@ -50,3 +52,5 @@ class Gint_drho : public Gint }; } + +#endif // ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_DRHO_H diff --git a/source/source_hamilt/module_gint/gint_dvlocal.h b/source/source_hamilt/module_gint/gint_dvlocal.h index a271949a8f5..26027b88e16 100644 --- a/source/source_hamilt/module_gint/gint_dvlocal.h +++ b/source/source_hamilt/module_gint/gint_dvlocal.h @@ -1,8 +1,10 @@ -#pragma once +#ifndef ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_DVLOCAL_H +#define ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_DVLOCAL_H + #include #include #include "source_hamilt/module_hcontainer/hcontainer.h" -#include "source_lcao/lcao_hs_arrays.hpp" +#include "source_lcao/lcao_hs_arrays.h" #include "source_lcao/module_ri/abfs_vector3_order.h" #include "gint.h" #include "gint_info.h" @@ -70,4 +72,6 @@ class Gint_dvlocal : public Gint HContainer pvdpRz; }; -} \ No newline at end of file +} + +#endif // ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_DVLOCAL_H diff --git a/source/source_hamilt/module_gint/gint_env_gamma.h b/source/source_hamilt/module_gint/gint_env_gamma.h index 58fc093bb7e..bd1abc087b2 100644 --- a/source/source_hamilt/module_gint/gint_env_gamma.h +++ b/source/source_hamilt/module_gint/gint_env_gamma.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_ENV_GAMMA_H +#define ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_ENV_GAMMA_H + #include #include @@ -28,4 +30,6 @@ class Gint_env_gamma : public Gint std::vector wfc_gint_; }; -} \ No newline at end of file +} + +#endif // ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_ENV_GAMMA_H diff --git a/source/source_hamilt/module_gint/gint_env_k.h b/source/source_hamilt/module_gint/gint_env_k.h index 248f4e270f0..ded6cb97c75 100644 --- a/source/source_hamilt/module_gint/gint_env_k.h +++ b/source/source_hamilt/module_gint/gint_env_k.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_ENV_K_H +#define ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_ENV_K_H + #include "gint.h" #include "gint_info.h" @@ -37,3 +39,5 @@ class Gint_env_k : public Gint }; } // namespace ModuleGint + +#endif // ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_ENV_K_H diff --git a/source/source_hamilt/module_gint/gint_fvl.h b/source/source_hamilt/module_gint/gint_fvl.h index 5fc16dc5bb3..c7924e42cfd 100644 --- a/source/source_hamilt/module_gint/gint_fvl.h +++ b/source/source_hamilt/module_gint/gint_fvl.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_FVL_H +#define ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_FVL_H + #include #include @@ -49,4 +51,6 @@ class Gint_fvl : public Gint double dr3_; }; -} \ No newline at end of file +} + +#endif // ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_FVL_H diff --git a/source/source_hamilt/module_gint/gint_fvl_gpu.h b/source/source_hamilt/module_gint/gint_fvl_gpu.h index 2e48fc629fd..81cf5e69e73 100644 --- a/source/source_hamilt/module_gint/gint_fvl_gpu.h +++ b/source/source_hamilt/module_gint/gint_fvl_gpu.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_FVL_GPU_H +#define ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_FVL_GPU_H + #include #include @@ -58,4 +60,6 @@ class Gint_fvl_gpu : public Gint CudaMemWrapper svl_d_; }; -} \ No newline at end of file +} + +#endif // ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_FVL_GPU_H diff --git a/source/source_hamilt/module_gint/gint_fvl_meta.h b/source/source_hamilt/module_gint/gint_fvl_meta.h index 2d3137bc455..a2b616b53fc 100644 --- a/source/source_hamilt/module_gint/gint_fvl_meta.h +++ b/source/source_hamilt/module_gint/gint_fvl_meta.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_FVL_META_H +#define ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_FVL_META_H + #include #include @@ -50,4 +52,6 @@ class Gint_fvl_meta : public Gint double dr3_; }; -} // namespace ModuleGint \ No newline at end of file +} // namespace ModuleGint + +#endif // ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_FVL_META_H diff --git a/source/source_hamilt/module_gint/gint_fvl_meta_gpu.h b/source/source_hamilt/module_gint/gint_fvl_meta_gpu.h index b0c14ba265f..17d80845d9b 100644 --- a/source/source_hamilt/module_gint/gint_fvl_meta_gpu.h +++ b/source/source_hamilt/module_gint/gint_fvl_meta_gpu.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_FVL_META_GPU_H +#define ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_FVL_META_GPU_H + #include #include @@ -61,4 +63,6 @@ class Gint_fvl_meta_gpu : public Gint CudaMemWrapper svl_d_; }; -} // namespace ModuleGint \ No newline at end of file +} // namespace ModuleGint + +#endif // ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_FVL_META_GPU_H diff --git a/source/source_hamilt/module_gint/gint_helper.h b/source/source_hamilt/module_gint/gint_helper.h index e1efce01f11..00b9aa39f63 100644 --- a/source/source_hamilt/module_gint/gint_helper.h +++ b/source/source_hamilt/module_gint/gint_helper.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_HELPER_H +#define ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_HELPER_H + #include #include @@ -64,3 +66,5 @@ inline int ceil_div(const int a, const int b) } } + +#endif // ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_HELPER_H diff --git a/source/source_hamilt/module_gint/gint_info.cpp b/source/source_hamilt/module_gint/gint_info.cpp index e0940230e1a..98e26208289 100644 --- a/source/source_hamilt/module_gint/gint_info.cpp +++ b/source/source_hamilt/module_gint/gint_info.cpp @@ -92,8 +92,8 @@ void GintInfo::init_atoms_(int ntype, const Atom* atoms, const Numerical_Orbital { Vec3d fraction; fraction.x = atom.taud[j].x * unitcell_info_->get_nbx(); - fraction.y = atom.taud[j].y * unitcell_info_->get_nby(); - fraction.z = atom.taud[j].z * unitcell_info_->get_nbz(); + fraction.y = atom.taud[j].y * unitcell_info_->get_nby(); + fraction.z = atom.taud[j].z * unitcell_info_->get_nbz(); const Vec3i atom_bgrid_idx(static_cast(fraction.x), static_cast(fraction.y), static_cast(fraction.z)); diff --git a/source/source_hamilt/module_gint/gint_info.h b/source/source_hamilt/module_gint/gint_info.h index 6d1736cb629..d08c76be055 100644 --- a/source/source_hamilt/module_gint/gint_info.h +++ b/source/source_hamilt/module_gint/gint_info.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_INFO_H +#define ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_INFO_H + #include #include @@ -161,3 +163,5 @@ class GintInfo }; } // namespace ModuleGint + +#endif // ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_INFO_H diff --git a/source/source_hamilt/module_gint/gint_interface.h b/source/source_hamilt/module_gint/gint_interface.h index 954da1bcb97..c81605d6e18 100644 --- a/source/source_hamilt/module_gint/gint_interface.h +++ b/source/source_hamilt/module_gint/gint_interface.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_INTERFACE_H +#define ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_INTERFACE_H + #include #include "source_hamilt/module_hcontainer/hcontainer.h" #include "gint_type.h" @@ -83,3 +85,5 @@ void cal_dvlocal_R_sparse( } // namespace ModuleGint + +#endif // ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_INTERFACE_H diff --git a/source/source_estate/module_charge/gint_prec_ctrl.cpp b/source/source_hamilt/module_gint/gint_prec_ctrl.cpp similarity index 100% rename from source/source_estate/module_charge/gint_prec_ctrl.cpp rename to source/source_hamilt/module_gint/gint_prec_ctrl.cpp diff --git a/source/source_estate/module_charge/gint_prec_ctrl.h b/source/source_hamilt/module_gint/gint_prec_ctrl.h similarity index 100% rename from source/source_estate/module_charge/gint_prec_ctrl.h rename to source/source_hamilt/module_gint/gint_prec_ctrl.h diff --git a/source/source_hamilt/module_gint/gint_rho.h b/source/source_hamilt/module_gint/gint_rho.h index 408620abce9..1628c8b8e40 100644 --- a/source/source_hamilt/module_gint/gint_rho.h +++ b/source/source_hamilt/module_gint/gint_rho.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_RHO_H +#define ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_RHO_H + #include #include @@ -48,3 +50,5 @@ class Gint_rho : public Gint }; } + +#endif // ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_RHO_H diff --git a/source/source_hamilt/module_gint/gint_rho_gpu.h b/source/source_hamilt/module_gint/gint_rho_gpu.h index 3f7377cf63a..132da8149b3 100644 --- a/source/source_hamilt/module_gint/gint_rho_gpu.h +++ b/source/source_hamilt/module_gint/gint_rho_gpu.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_RHO_GPU_H +#define ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_RHO_GPU_H + #include #include @@ -37,4 +39,6 @@ class Gint_rho_gpu: public Gint double ** rho_ = nullptr; }; -} \ No newline at end of file +} + +#endif // ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_RHO_GPU_H diff --git a/source/source_hamilt/module_gint/gint_tau.h b/source/source_hamilt/module_gint/gint_tau.h index cb6281b87d8..3aaf361a201 100644 --- a/source/source_hamilt/module_gint/gint_tau.h +++ b/source/source_hamilt/module_gint/gint_tau.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_TAU_H +#define ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_TAU_H + #include #include #include "source_hamilt/module_hcontainer/hcontainer.h" @@ -36,3 +38,5 @@ class Gint_tau : public Gint }; } // namespace ModuleGint + +#endif // ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_TAU_H diff --git a/source/source_hamilt/module_gint/gint_tau_gpu.h b/source/source_hamilt/module_gint/gint_tau_gpu.h index 4fa2000d1c7..d378287b0d6 100644 --- a/source/source_hamilt/module_gint/gint_tau_gpu.h +++ b/source/source_hamilt/module_gint/gint_tau_gpu.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_TAU_GPU_H +#define ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_TAU_GPU_H + #include #include @@ -44,4 +46,6 @@ class Gint_tau_gpu : public Gint std::vector> kin_d_vec_; }; -} \ No newline at end of file +} + +#endif // ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_TAU_GPU_H diff --git a/source/source_hamilt/module_gint/gint_type.h b/source/source_hamilt/module_gint/gint_type.h index 5398ef4444c..1a16dc19248 100644 --- a/source/source_hamilt/module_gint/gint_type.h +++ b/source/source_hamilt/module_gint/gint_type.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_TYPE_H +#define ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_TYPE_H + #include "source_hamilt/module_hcontainer/hcontainer.h" #include "source_base/vector3.h" @@ -12,4 +14,6 @@ namespace ModuleGint template using HContainer = hamilt::HContainer; -} \ No newline at end of file +} + +#endif // ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_TYPE_H diff --git a/source/source_hamilt/module_gint/gint_vl.h b/source/source_hamilt/module_gint/gint_vl.h index df37017814d..da81435dc9d 100644 --- a/source/source_hamilt/module_gint/gint_vl.h +++ b/source/source_hamilt/module_gint/gint_vl.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_VL_H +#define ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_VL_H + #include #include @@ -35,3 +37,5 @@ class Gint_vl : public Gint }; } + +#endif // ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_VL_H diff --git a/source/source_hamilt/module_gint/gint_vl_gpu.h b/source/source_hamilt/module_gint/gint_vl_gpu.h index 5c71ed4099c..2e812593b79 100644 --- a/source/source_hamilt/module_gint/gint_vl_gpu.h +++ b/source/source_hamilt/module_gint/gint_vl_gpu.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_VL_GPU_H +#define ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_VL_GPU_H + #include #include @@ -34,4 +36,6 @@ class Gint_vl_gpu : public Gint double dr3_; }; -} \ No newline at end of file +} + +#endif // ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_VL_GPU_H diff --git a/source/source_hamilt/module_gint/gint_vl_metagga.h b/source/source_hamilt/module_gint/gint_vl_metagga.h index 4f74573abaf..1db185fade6 100644 --- a/source/source_hamilt/module_gint/gint_vl_metagga.h +++ b/source/source_hamilt/module_gint/gint_vl_metagga.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_VL_METAGGA_H +#define ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_VL_METAGGA_H + #include #include @@ -42,4 +44,6 @@ class Gint_vl_metagga : public Gint }; -} \ No newline at end of file +} + +#endif // ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_VL_METAGGA_H diff --git a/source/source_hamilt/module_gint/gint_vl_metagga_gpu.h b/source/source_hamilt/module_gint/gint_vl_metagga_gpu.h index 51db3e64ee6..0f3b9a13999 100644 --- a/source/source_hamilt/module_gint/gint_vl_metagga_gpu.h +++ b/source/source_hamilt/module_gint/gint_vl_metagga_gpu.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_VL_METAGGA_GPU_H +#define ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_VL_METAGGA_GPU_H + #include #include @@ -50,4 +52,6 @@ class Gint_vl_metagga_gpu : public Gint CudaMemWrapper vofk_d_; }; -} \ No newline at end of file +} + +#endif // ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_VL_METAGGA_GPU_H diff --git a/source/source_hamilt/module_gint/gint_vl_mgga_n4.h b/source/source_hamilt/module_gint/gint_vl_mgga_n4.h index 16aaf19607a..5e9a8a29a2c 100644 --- a/source/source_hamilt/module_gint/gint_vl_mgga_n4.h +++ b/source/source_hamilt/module_gint/gint_vl_mgga_n4.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_VL_MGGA_N4_H +#define ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_VL_MGGA_N4_H + #include #include @@ -39,4 +41,6 @@ class Gint_vl_metagga_nspin4 : public Gint std::vector> hr_gint_part_; }; -} \ No newline at end of file +} + +#endif // ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_VL_MGGA_N4_H diff --git a/source/source_hamilt/module_gint/gint_vl_mgga_n4_gpu.h b/source/source_hamilt/module_gint/gint_vl_mgga_n4_gpu.h index 36bf8f4011f..86cda291cbf 100644 --- a/source/source_hamilt/module_gint/gint_vl_mgga_n4_gpu.h +++ b/source/source_hamilt/module_gint/gint_vl_mgga_n4_gpu.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_VL_MGGA_N4_GPU_H +#define ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_VL_MGGA_N4_GPU_H + #include #include @@ -48,4 +50,6 @@ class Gint_vl_metagga_nspin4_gpu : public Gint std::vector> hr_gint_part_d_; }; -} \ No newline at end of file +} + +#endif // ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_VL_MGGA_N4_GPU_H diff --git a/source/source_hamilt/module_gint/gint_vl_nspin4.h b/source/source_hamilt/module_gint/gint_vl_nspin4.h index e4f66fd75b2..c934b9c74d7 100644 --- a/source/source_hamilt/module_gint/gint_vl_nspin4.h +++ b/source/source_hamilt/module_gint/gint_vl_nspin4.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_VL_NSPIN4_H +#define ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_VL_NSPIN4_H + #include #include @@ -41,4 +43,6 @@ class Gint_vl_nspin4 : public Gint std::vector> hr_gint_part_; }; -} // namespace ModuleGint \ No newline at end of file +} // namespace ModuleGint + +#endif // ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_VL_NSPIN4_H diff --git a/source/source_hamilt/module_gint/gint_vl_nspin4_gpu.h b/source/source_hamilt/module_gint/gint_vl_nspin4_gpu.h index 50aa809ceca..c6cf58833df 100644 --- a/source/source_hamilt/module_gint/gint_vl_nspin4_gpu.h +++ b/source/source_hamilt/module_gint/gint_vl_nspin4_gpu.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_VL_NSPIN4_GPU_H +#define ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_VL_NSPIN4_GPU_H + #include #include @@ -49,4 +51,6 @@ class Gint_vl_nspin4_gpu : public Gint std::vector> hr_gint_part_d_; }; -} // namespace ModuleGint \ No newline at end of file +} // namespace ModuleGint + +#endif // ABACUS_SOURCE_HAMILT_MODULE_GINT_GINT_VL_NSPIN4_GPU_H diff --git a/source/source_hamilt/module_gint/kernel/cuda_mem_wrapper.h b/source/source_hamilt/module_gint/kernel/cuda_mem_wrapper.h index 0da53865535..57aa26c0a09 100644 --- a/source/source_hamilt/module_gint/kernel/cuda_mem_wrapper.h +++ b/source/source_hamilt/module_gint/kernel/cuda_mem_wrapper.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_HAMILT_MODULE_GINT_KERNEL_CUDA_MEM_WRAPPER_H +#define ABACUS_SOURCE_HAMILT_MODULE_GINT_KERNEL_CUDA_MEM_WRAPPER_H + #include #include "source_base/tool_quit.h" #include "source_base/module_device/device_check.h" @@ -168,4 +170,6 @@ class CudaMemWrapper size_t size_ = 0; bool malloc_host_ = false; cudaStream_t stream_ = 0; -}; \ No newline at end of file +}; + +#endif // ABACUS_SOURCE_HAMILT_MODULE_GINT_KERNEL_CUDA_MEM_WRAPPER_H diff --git a/source/source_hamilt/module_gint/kernel/dgemm_vbatch.h b/source/source_hamilt/module_gint/kernel/dgemm_vbatch.h index 9d70c84348c..3486c2ee0ee 100644 --- a/source/source_hamilt/module_gint/kernel/dgemm_vbatch.h +++ b/source/source_hamilt/module_gint/kernel/dgemm_vbatch.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_HAMILT_MODULE_GINT_KERNEL_DGEMM_VBATCH_H +#define ABACUS_SOURCE_HAMILT_MODULE_GINT_KERNEL_DGEMM_VBATCH_H + #include @@ -36,3 +38,5 @@ void gemm_tn_vbatch( double** C_array_d, const int* ldc_d, int batchCount, cudaStream_t stream, const T* alpha = nullptr); + +#endif // ABACUS_SOURCE_HAMILT_MODULE_GINT_KERNEL_DGEMM_VBATCH_H diff --git a/source/source_hamilt/module_gint/kernel/gint_gpu_vars.h b/source/source_hamilt/module_gint/kernel/gint_gpu_vars.h index da238c666d6..7ee55054f88 100644 --- a/source/source_hamilt/module_gint/kernel/gint_gpu_vars.h +++ b/source/source_hamilt/module_gint/kernel/gint_gpu_vars.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_HAMILT_MODULE_GINT_KERNEL_GINT_GPU_VARS_H +#define ABACUS_SOURCE_HAMILT_MODULE_GINT_KERNEL_GINT_GPU_VARS_H + #include #include "set_const_mem.cuh" @@ -37,4 +39,6 @@ class GintGpuVars }; -} \ No newline at end of file +} + +#endif // ABACUS_SOURCE_HAMILT_MODULE_GINT_KERNEL_GINT_GPU_VARS_H diff --git a/source/source_hamilt/module_gint/kernel/gint_helper.cuh b/source/source_hamilt/module_gint/kernel/gint_helper.cuh index 1f6d431003d..89abb6d3a8a 100644 --- a/source/source_hamilt/module_gint/kernel/gint_helper.cuh +++ b/source/source_hamilt/module_gint/kernel/gint_helper.cuh @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_HAMILT_MODULE_GINT_KERNEL_GINT_HELPER_CUH +#define ABACUS_SOURCE_HAMILT_MODULE_GINT_KERNEL_GINT_HELPER_CUH + #include // if exponent is an integer between 0 and 5 (the most common cases in gint) and @@ -85,3 +87,4 @@ template <> struct gemm_vec_traits } }; +#endif // ABACUS_SOURCE_HAMILT_MODULE_GINT_KERNEL_GINT_HELPER_CUH diff --git a/source/source_hamilt/module_gint/kernel/phi_operator_gpu.h b/source/source_hamilt/module_gint/kernel/phi_operator_gpu.h index a42b2a2eb24..b7da743788c 100644 --- a/source/source_hamilt/module_gint/kernel/phi_operator_gpu.h +++ b/source/source_hamilt/module_gint/kernel/phi_operator_gpu.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_HAMILT_MODULE_GINT_KERNEL_PHI_OPERATOR_GPU_H +#define ABACUS_SOURCE_HAMILT_MODULE_GINT_KERNEL_PHI_OPERATOR_GPU_H + #include #include #include @@ -175,4 +177,6 @@ class PhiOperatorGpu mutable std::vector buckets_; }; -} \ No newline at end of file +} + +#endif // ABACUS_SOURCE_HAMILT_MODULE_GINT_KERNEL_PHI_OPERATOR_GPU_H diff --git a/source/source_hamilt/module_gint/kernel/phi_operator_kernel.cuh b/source/source_hamilt/module_gint/kernel/phi_operator_kernel.cuh index 00a53860c7f..9f938345488 100644 --- a/source/source_hamilt/module_gint/kernel/phi_operator_kernel.cuh +++ b/source/source_hamilt/module_gint/kernel/phi_operator_kernel.cuh @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_HAMILT_MODULE_GINT_KERNEL_PHI_OPERATOR_KERNEL_CUH +#define ABACUS_SOURCE_HAMILT_MODULE_GINT_KERNEL_PHI_OPERATOR_KERNEL_CUH + #include @@ -370,3 +372,5 @@ __global__ void phi_dot_dphi_r_kernel( double* __restrict__ svl); } + +#endif // ABACUS_SOURCE_HAMILT_MODULE_GINT_KERNEL_PHI_OPERATOR_KERNEL_CUH diff --git a/source/source_hamilt/module_gint/kernel/set_const_mem.cuh b/source/source_hamilt/module_gint/kernel/set_const_mem.cuh index 715fa98cde1..31aee4aeb26 100644 --- a/source/source_hamilt/module_gint/kernel/set_const_mem.cuh +++ b/source/source_hamilt/module_gint/kernel/set_const_mem.cuh @@ -1,7 +1,11 @@ -#pragma once +#ifndef ABACUS_SOURCE_HAMILT_MODULE_GINT_KERNEL_SET_CONST_MEM_CUH +#define ABACUS_SOURCE_HAMILT_MODULE_GINT_KERNEL_SET_CONST_MEM_CUH + #include namespace ModuleGint { __host__ void set_ylmcoe_d(const double* ylmcoe_h, double** ylmcoe_d_addr); -} \ No newline at end of file +} + +#endif // ABACUS_SOURCE_HAMILT_MODULE_GINT_KERNEL_SET_CONST_MEM_CUH diff --git a/source/source_hamilt/module_gint/kernel/sph.cuh b/source/source_hamilt/module_gint/kernel/sph.cuh index bda5c286d76..c8a91de5172 100644 --- a/source/source_hamilt/module_gint/kernel/sph.cuh +++ b/source/source_hamilt/module_gint/kernel/sph.cuh @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_HAMILT_MODULE_GINT_KERNEL_SPH_CUH +#define ABACUS_SOURCE_HAMILT_MODULE_GINT_KERNEL_SPH_CUH + #include "source_base/kernels/cuda/sph_harm_gpu.cuh" @@ -8,3 +10,5 @@ namespace ModuleGint using ModuleBase::sph_harm; using ModuleBase::grad_rl_sph_harm; } + +#endif // ABACUS_SOURCE_HAMILT_MODULE_GINT_KERNEL_SPH_CUH diff --git a/source/source_hamilt/module_gint/localcell_info.h b/source/source_hamilt/module_gint/localcell_info.h index 0c146b86ab6..21e193b9d49 100644 --- a/source/source_hamilt/module_gint/localcell_info.h +++ b/source/source_hamilt/module_gint/localcell_info.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_HAMILT_MODULE_GINT_LOCALCELL_INFO_H +#define ABACUS_SOURCE_HAMILT_MODULE_GINT_LOCALCELL_INFO_H + #include #include "gint_type.h" @@ -124,3 +126,5 @@ class LocalCellInfo }; } // namespace ModuleGint + +#endif // ABACUS_SOURCE_HAMILT_MODULE_GINT_LOCALCELL_INFO_H diff --git a/source/source_hamilt/module_gint/meshgrid_info.h b/source/source_hamilt/module_gint/meshgrid_info.h index c1c276bc6e1..d705754bb3a 100644 --- a/source/source_hamilt/module_gint/meshgrid_info.h +++ b/source/source_hamilt/module_gint/meshgrid_info.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_HAMILT_MODULE_GINT_MESHGRID_INFO_H +#define ABACUS_SOURCE_HAMILT_MODULE_GINT_MESHGRID_INFO_H + #include "gint_type.h" @@ -60,4 +62,6 @@ class MeshGridInfo double mgrid_vol_; }; -} // namespace ModuleGint \ No newline at end of file +} // namespace ModuleGint + +#endif // ABACUS_SOURCE_HAMILT_MODULE_GINT_MESHGRID_INFO_H diff --git a/source/source_hamilt/module_gint/phi_operator.cpp b/source/source_hamilt/module_gint/phi_operator.cpp index cc416cd0a4c..7796d5fd819 100644 --- a/source/source_hamilt/module_gint/phi_operator.cpp +++ b/source/source_hamilt/module_gint/phi_operator.cpp @@ -1,6 +1,7 @@ #include "phi_operator.h" #include "source_base/global_function.h" #include "source_base/matrix.h" +#include "source_base/module_external/blas_connector.h" #include @@ -242,3 +243,269 @@ void PhiOperator::init_atom_pair_idx_() } } // namespace ModuleGint + +//============================================================ +// Template member function implementations (moved from .hpp) +//============================================================ + +namespace ModuleGint +{ + +namespace { + +// Helper: dispatch a Tin-typed BLAS-GEMM target buffer. +// For Tin=double, write directly into phi_dm (no scratch). +// For Tin=float, allocate fp32 scratch and cast at the end. +inline double* phi_mul_dm_scratch_(double* phi_dm, std::vector& /*scratch*/, int /*size*/) +{ + return phi_dm; +} + +template +inline Tin* phi_mul_dm_scratch_(double* /*phi_dm*/, std::vector& scratch, int size) +{ + scratch.assign(size, Tin(0)); + return scratch.data(); +} + +inline void phi_mul_dm_finalize_(double* /*phi_dm*/, const std::vector& /*scratch*/, int /*size*/) {} + +template +inline void phi_mul_dm_finalize_(double* phi_dm, const std::vector& scratch, int size) +{ + for (int k = 0; k < size; ++k) + { + phi_dm[k] = static_cast(scratch[k]); + } +} + +} // namespace + +template +void PhiOperator::set_phi(T* phi) const +{ + for(int i = 0; i < biggrid_->get_atoms_num(); ++i) + { + const auto atom = biggrid_->get_atom(i); + atom->set_phi(atom_rcoords_[i], cols_, phi); + phi += atom->get_nw(); + } +} + +template +void PhiOperator::phi_mul_dm( + const Tin*const phi, + const HContainer& dm, + const bool is_symm, + double*const phi_dm) const +{ + std::vector scratch; + Tin* target = phi_mul_dm_scratch_(phi_dm, scratch, rows_ * cols_); + ModuleBase::GlobalFunc::ZEROS(target, rows_ * cols_); + + for(int i = 0; i < biggrid_->get_atoms_num(); ++i) + { + const auto atom_i = biggrid_->get_atom(i); + const auto r_i = atom_i->get_R(); + + if(is_symm) + { + const auto dm_mat = dm.find_matrix(atom_i->get_iat(), atom_i->get_iat(), 0, 0, 0); + constexpr Tin alpha = 1.0; + constexpr Tin beta = 1.0; + BlasConnector::symm_cm( + 'L', 'U', + atoms_phi_len_[i], rows_, + alpha, dm_mat->get_pointer(), atoms_phi_len_[i], + &phi[0 * cols_ + atoms_startidx_[i]], cols_, + beta, &target[0 * cols_ + atoms_startidx_[i]], cols_); + } + + const int start = is_symm ? i + 1 : 0; + + for(int j = start; j < biggrid_->get_atoms_num(); ++j) + { + const auto atom_j = biggrid_->get_atom(j); + const auto r_j = atom_j->get_R(); + const auto dm_mat = dm.find_matrix(atom_i->get_iat(), atom_j->get_iat(), r_i-r_j); + + if(dm_mat == nullptr) + { + continue; + } + + const int start_idx = get_atom_pair_start_end_idx_(i, j).first; + const int end_idx = get_atom_pair_start_end_idx_(i, j).second; + const int len = end_idx - start_idx + 1; + + if(len <= 0) + { + continue; + } + + const Tin alpha = is_symm ? 2.0 : 1.0; + constexpr Tin beta = 1.0; + BlasConnector::gemm( + 'N', 'N', + len, atoms_phi_len_[j], atoms_phi_len_[i], + alpha, &phi[start_idx * cols_ + atoms_startidx_[i]], cols_, + dm_mat->get_pointer(), atoms_phi_len_[j], + beta, &target[start_idx * cols_ + atoms_startidx_[j]], cols_); + } + } + + phi_mul_dm_finalize_(phi_dm, scratch, rows_ * cols_); +} + +template +void PhiOperator::phi_mul_vldr3( + const T*const vl, + const T dr3, + const T*const phi, + T*const result) const +{ + int idx = 0; + for(int i = 0; i < biggrid_->get_mgrids_num(); i++) + { + T vldr3_mgrid = vl[mgrid_lidx_[i]] * dr3; + for(int j = 0; j < cols_; j++) + { + result[idx] = phi[idx] * vldr3_mgrid; + idx++; + } + } +} + +template +void PhiOperator::phi_mul_phi( + const Tin*const phi_i, + const Tin*const phi_j, + HContainer& hr, + const TriPart part) const +{ + std::vector tmp_hr; + for(int i = 0; i < biggrid_->get_atoms_num(); ++i) + { + const auto atom_i = biggrid_->get_atom(i); + const auto& r_i = atom_i->get_R(); + const int iat_i = atom_i->get_iat(); + const int n_i = atoms_phi_len_[i]; + + for(int j = 0; j < biggrid_->get_atoms_num(); ++j) + { + const auto atom_j = biggrid_->get_atom(j); + const auto& r_j = atom_j->get_R(); + const int iat_j = atom_j->get_iat(); + const int n_j = atoms_phi_len_[j]; + + if(part==TriPart::Upper && iat_i>iat_j) + { + continue; + } + else if(part==TriPart::Lower && iat_iadd_array_ts(tmp_hr.data()); + } + } +} + +// Mixed-precision dotc wrapper. Accepts (double, double) or (double, float); +// when y is fp32 it is upcast into the caller-provided fp64 scratch buffer. +namespace { + +inline double dotc_mixed(int n, const double* x, const double* y, + std::vector& /*buf*/) +{ + return BlasConnector::dotc(n, x, 1, y, 1); +} + +inline double dotc_mixed(int n, const double* x, const float* y, + std::vector& buf) +{ + if (static_cast(buf.size()) < n) { buf.resize(n); } + for (int k = 0; k < n; ++k) { buf[k] = static_cast(y[k]); } + return BlasConnector::dotc(n, x, 1, buf.data(), 1); +} + +} // namespace + +template +void PhiOperator::phi_dot_phi( + const Tin*const phi_i, + const double*const phi_j, + double*const rho) const +{ + std::vector buf; + for(int i = 0; i < biggrid_->get_mgrids_num(); ++i) + { + rho[mgrid_lidx_[i]] += dotc_mixed( + cols_, phi_j + i * cols_, phi_i + i * cols_, buf); + } +} + +//============================================================ +// Explicit template instantiation +//============================================================ + +template void PhiOperator::set_phi(double* phi) const; +template void PhiOperator::set_phi(float* phi) const; + +template void PhiOperator::phi_mul_dm( + const double*const phi, const HContainer& dm, + const bool is_symm, double*const phi_dm) const; +template void PhiOperator::phi_mul_dm( + const float*const phi, const HContainer& dm, + const bool is_symm, double*const phi_dm) const; + +template void PhiOperator::phi_mul_vldr3( + const double*const vl, const double dr3, + const double*const phi, double*const result) const; +template void PhiOperator::phi_mul_vldr3( + const float*const vl, const float dr3, + const float*const phi, float*const result) const; + +template void PhiOperator::phi_mul_phi( + const double*const phi_i, const double*const phi_j, + HContainer& hr, const TriPart part) const; +template void PhiOperator::phi_mul_phi( + const float*const phi_i, const float*const phi_j, + HContainer& hr, const TriPart part) const; + +template void PhiOperator::phi_dot_phi( + const double*const phi_i, const double*const phi_j, + double*const rho) const; +template void PhiOperator::phi_dot_phi( + const float*const phi_i, const double*const phi_j, + double*const rho) const; + +} // namespace ModuleGint diff --git a/source/source_hamilt/module_gint/phi_operator.h b/source/source_hamilt/module_gint/phi_operator.h index 8ae9da693d1..6b915e7f02f 100644 --- a/source/source_hamilt/module_gint/phi_operator.h +++ b/source/source_hamilt/module_gint/phi_operator.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_HAMILT_MODULE_GINT_PHI_OPERATOR_H +#define ABACUS_SOURCE_HAMILT_MODULE_GINT_PHI_OPERATOR_H + #include #include @@ -170,6 +172,6 @@ class PhiOperator std::vector> atom_pair_range_; }; -} +} // namespace ModuleGint -#include "phi_operator.hpp" +#endif // ABACUS_SOURCE_HAMILT_MODULE_GINT_PHI_OPERATOR_H diff --git a/source/source_hamilt/module_gint/phi_operator.hpp b/source/source_hamilt/module_gint/phi_operator.hpp deleted file mode 100644 index 071a775d8ee..00000000000 --- a/source/source_hamilt/module_gint/phi_operator.hpp +++ /dev/null @@ -1,244 +0,0 @@ -#pragma once - -#include "phi_operator.h" -#include "source_base/module_external/blas_connector.h" -#include "source_base/global_function.h" - -namespace ModuleGint -{ - -template -void PhiOperator::set_phi(T* phi) const -{ - for(int i = 0; i < biggrid_->get_atoms_num(); ++i) - { - const auto atom = biggrid_->get_atom(i); - atom->set_phi(atom_rcoords_[i], cols_, phi); - phi += atom->get_nw(); - } -} - -// Helpers to dispatch a Tin-typed BLAS-GEMM target buffer: -// - For Tin=double, write the GEMM result directly into the caller's -// double* phi_dm (no scratch, no cast). -// - For Tin=float, allocate a fp32 scratch buffer and cast the result into -// phi_dm at the end. K of each individual GEMM is small (~atom nw) so -// accumulating in fp32 inside the scratch is fine. -inline double* phi_mul_dm_scratch_(double* phi_dm, std::vector& /*scratch*/, int /*size*/) -{ - return phi_dm; -} - -template -inline Tin* phi_mul_dm_scratch_(double* /*phi_dm*/, std::vector& scratch, int size) -{ - scratch.assign(size, Tin(0)); - return scratch.data(); -} - -inline void phi_mul_dm_finalize_(double* /*phi_dm*/, const std::vector& /*scratch*/, int /*size*/) {} - -template -inline void phi_mul_dm_finalize_(double* phi_dm, const std::vector& scratch, int size) -{ - for (int k = 0; k < size; ++k) - { - phi_dm[k] = static_cast(scratch[k]); - } -} - -// phi_dm(ir,iwt_2) = \sum_{iwt_1} phi(ir,iwt_1) * dm(iwt_1,iwt_2) -template -void PhiOperator::phi_mul_dm( - const Tin*const phi, // phi(ir,iwt) - const HContainer& dm, // dm(iwt_1,iwt_2) - const bool is_symm, - double*const phi_dm) const // phi_dm(ir,iwt) -{ - std::vector scratch; - Tin* target = phi_mul_dm_scratch_(phi_dm, scratch, rows_ * cols_); - ModuleBase::GlobalFunc::ZEROS(target, rows_ * cols_); - - for(int i = 0; i < biggrid_->get_atoms_num(); ++i) - { - const auto atom_i = biggrid_->get_atom(i); - const auto r_i = atom_i->get_R(); - - if(is_symm) - { - const auto dm_mat = dm.find_matrix(atom_i->get_iat(), atom_i->get_iat(), 0, 0, 0); - constexpr Tin alpha = 1.0; - constexpr Tin beta = 1.0; - BlasConnector::symm_cm( - 'L', 'U', - atoms_phi_len_[i], rows_, - alpha, dm_mat->get_pointer(), atoms_phi_len_[i], - &phi[0 * cols_ + atoms_startidx_[i]], cols_, - beta, &target[0 * cols_ + atoms_startidx_[i]], cols_); - } - - const int start = is_symm ? i + 1 : 0; - - for(int j = start; j < biggrid_->get_atoms_num(); ++j) - { - const auto atom_j = biggrid_->get_atom(j); - const auto r_j = atom_j->get_R(); - // FIXME may be r = r_j - r_i - const auto dm_mat = dm.find_matrix(atom_i->get_iat(), atom_j->get_iat(), r_i-r_j); - - // if dm_mat is nullptr, it means this atom pair does not affect any meshgrid in the unitcell - if(dm_mat == nullptr) - { - continue; - } - - const int start_idx = get_atom_pair_start_end_idx_(i, j).first; - const int end_idx = get_atom_pair_start_end_idx_(i, j).second; - const int len = end_idx - start_idx + 1; - - // if len<=0, it means this atom pair does not affect any meshgrid in this biggrid - if(len <= 0) - { - continue; - } - - const Tin alpha = is_symm ? 2.0 : 1.0; - constexpr Tin beta = 1.0; - BlasConnector::gemm( - 'N', 'N', - len, atoms_phi_len_[j], atoms_phi_len_[i], - alpha, &phi[start_idx * cols_ + atoms_startidx_[i]], cols_, - dm_mat->get_pointer(), atoms_phi_len_[j], - beta, &target[start_idx * cols_ + atoms_startidx_[j]], cols_); - } - } - - phi_mul_dm_finalize_(phi_dm, scratch, rows_ * cols_); -} - -// result(ir) = phi(ir) * vl(ir) -template -void PhiOperator::phi_mul_vldr3( - const T*const vl, // vl(ir) - const T dr3, - const T*const phi, // phi(ir,iwt) - T*const result) const // result(ir,iwt) -{ - int idx = 0; - for(int i = 0; i < biggrid_->get_mgrids_num(); i++) - { - T vldr3_mgrid = vl[mgrid_lidx_[i]] * dr3; - for(int j = 0; j < cols_; j++) - { - result[idx] = phi[idx] * vldr3_mgrid; - idx++; - } - } -} - -// hr(iwt_i,iwt_j) += \sum_{ir} phi_i(ir,iwt_i) * phi_i(ir,iwt_j) -// this is a thread-safe function. -// The per-biggrid GEMM accumulator tmp_hr stays in Tin (small K, no significant -// precision loss); only the global add into HContainer is widened. -template -void PhiOperator::phi_mul_phi( - const Tin*const phi_i, // phi_i(ir,iwt) - const Tin*const phi_j, // phi_j(ir,iwt) - HContainer& hr, // hr(iwt_i,iwt_j) - const TriPart part) const -{ - std::vector tmp_hr; - for(int i = 0; i < biggrid_->get_atoms_num(); ++i) - { - const auto atom_i = biggrid_->get_atom(i); - const auto& r_i = atom_i->get_R(); - const int iat_i = atom_i->get_iat(); - const int n_i = atoms_phi_len_[i]; - - for(int j = 0; j < biggrid_->get_atoms_num(); ++j) - { - const auto atom_j = biggrid_->get_atom(j); - const auto& r_j = atom_j->get_R(); - const int iat_j = atom_j->get_iat(); - const int n_j = atoms_phi_len_[j]; - - // only calculate the upper triangle matrix - if(part==TriPart::Upper && iat_i>iat_j) - { - continue; - } - // only calculate the upper triangle matrix - else if(part==TriPart::Lower && iat_iadd_array_ts(tmp_hr.data()); - } - } -} - -// Mixed-precision dotc wrapper. Accepts (double, double) or (double, float); -// when y is fp32 it is upcast into the caller-provided fp64 scratch buffer -// before dispatching to BlasConnector::dotc (which requires uniform types). -// The buffer is grown on demand and meant to be reused across many calls. -inline double dotc_mixed(int n, const double* x, const double* y, - std::vector& /*buf*/) -{ - return BlasConnector::dotc(n, x, 1, y, 1); -} - -inline double dotc_mixed(int n, const double* x, const float* y, - std::vector& buf) -{ - if (static_cast(buf.size()) < n) { buf.resize(n); } - for (int k = 0; k < n; ++k) { buf[k] = static_cast(y[k]); } - return BlasConnector::dotc(n, x, 1, buf.data(), 1); -} - -// rho(ir) = \sum_{iwt} \phi_i(ir,iwt) * \phi_j^*(ir,iwt) -// phi_j is always double (output of phi_mul_dm); phi_i may be fp32. dotc_mixed -// keeps the inner product in fp64 in either case. -template -void PhiOperator::phi_dot_phi( - const Tin*const phi_i, // phi_i(ir,iwt) - const double*const phi_j, // phi_j(ir,iwt) - double*const rho) const // rho(ir) -{ - std::vector buf; - for(int i = 0; i < biggrid_->get_mgrids_num(); ++i) - { - rho[mgrid_lidx_[i]] += dotc_mixed( - cols_, phi_j + i * cols_, phi_i + i * cols_, buf); - } -} - -} // namespace ModuleGint diff --git a/source/source_hamilt/module_gint/test/CMakeLists.txt b/source/source_hamilt/module_gint/test/CMakeLists.txt index 7cbfcbb9d11..8a678e2941c 100644 --- a/source/source_hamilt/module_gint/test/CMakeLists.txt +++ b/source/source_hamilt/module_gint/test/CMakeLists.txt @@ -25,4 +25,11 @@ AddTest( tmp_mocks.cpp ) +AddTest( + TARGET MODULE_LCAO_gint_prec_ctrl_test + LIBS parameter base device + SOURCES test_gint_prec_ctrl.cpp + ../gint_prec_ctrl.cpp +) + endif() diff --git a/source/source_hamilt/module_gint/test/test_gint_prec_ctrl.cpp b/source/source_hamilt/module_gint/test/test_gint_prec_ctrl.cpp new file mode 100644 index 00000000000..5cc6c053baa --- /dev/null +++ b/source/source_hamilt/module_gint/test/test_gint_prec_ctrl.cpp @@ -0,0 +1,63 @@ +#include "gtest/gtest.h" + +#include "../gint_prec_ctrl.h" + +TEST(GintPrecisionControllerTest, AutoModeSwitchesToFp64ImmediatelyWhenDrhoIsSmallEnough) +{ + GintPrecisionController controller; + + controller.set_mode("mix"); + controller.reset_for_new_scf(); + EXPECT_EQ(controller.current_precision(), ModuleGint::GintPrecision::fp32); + + EXPECT_TRUE(controller.update_after_iteration(9.0e-5, 1.0e-7)); + EXPECT_EQ(controller.current_precision(), ModuleGint::GintPrecision::fp64); +} + +TEST(GintPrecisionControllerTest, DefaultModeStartsAndStaysFp64) +{ + GintPrecisionController controller; + + controller.reset_for_new_scf(); + EXPECT_EQ(controller.current_precision(), ModuleGint::GintPrecision::fp64); + + EXPECT_FALSE(controller.update_after_iteration(9.0e-5, 1.0e-7)); + EXPECT_EQ(controller.current_precision(), ModuleGint::GintPrecision::fp64); +} + +TEST(GintPrecisionControllerTest, SingleModeStartsAndStaysFp32) +{ + GintPrecisionController controller; + + controller.set_mode("single"); + controller.reset_for_new_scf(); + EXPECT_EQ(controller.current_precision(), ModuleGint::GintPrecision::fp32); + + EXPECT_FALSE(controller.update_after_iteration(9.0e-5, 1.0e-7)); + EXPECT_EQ(controller.current_precision(), ModuleGint::GintPrecision::fp32); +} + +TEST(GintPrecisionControllerTest, MixModeLocksFp64AfterSwitch) +{ + GintPrecisionController controller; + + controller.set_mode("mix"); + controller.reset_for_new_scf(); + EXPECT_TRUE(controller.update_after_iteration(9.0e-5, 1.0e-6)); + EXPECT_EQ(controller.current_precision(), ModuleGint::GintPrecision::fp64); + + // After locking, should return false + EXPECT_FALSE(controller.update_after_iteration(1.0, 1.0e-6)); + EXPECT_EQ(controller.current_precision(), ModuleGint::GintPrecision::fp64); +} + +TEST(GintPrecisionControllerTest, MixModeReturnsFalseWhenDrhoTooLarge) +{ + GintPrecisionController controller; + + controller.set_mode("mix"); + controller.reset_for_new_scf(); + // drho is large, should not switch yet + EXPECT_FALSE(controller.update_after_iteration(1.0, 1.0e-7)); + EXPECT_EQ(controller.current_precision(), ModuleGint::GintPrecision::fp32); +} diff --git a/source/source_hamilt/module_gint/unitcell_info.h b/source/source_hamilt/module_gint/unitcell_info.h index df1e88b38c3..91accbfe14b 100644 --- a/source/source_hamilt/module_gint/unitcell_info.h +++ b/source/source_hamilt/module_gint/unitcell_info.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_HAMILT_MODULE_GINT_UNITCELL_INFO_H +#define ABACUS_SOURCE_HAMILT_MODULE_GINT_UNITCELL_INFO_H + #include #include @@ -169,4 +171,6 @@ class UnitCellInfo }; -} // namespace ModuleGint \ No newline at end of file +} // namespace ModuleGint + +#endif // ABACUS_SOURCE_HAMILT_MODULE_GINT_UNITCELL_INFO_H diff --git a/source/source_hamilt/module_hcontainer/hcontainer_funcs.h b/source/source_hamilt/module_hcontainer/hcontainer_funcs.h index 88ad9460b36..3020f51133f 100644 --- a/source/source_hamilt/module_hcontainer/hcontainer_funcs.h +++ b/source/source_hamilt/module_hcontainer/hcontainer_funcs.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_HAMILT_MODULE_HCONTAINER_HCONTAINER_FUNCS_H +#define ABACUS_SOURCE_HAMILT_MODULE_HCONTAINER_HCONTAINER_FUNCS_H + #include "source_hamilt/module_hcontainer/hcontainer.h" @@ -79,4 +81,6 @@ void gatherParallels(const hamilt::HContainer& hR_p, #endif // __MPI -} // namespace hamilt \ No newline at end of file +} // namespace hamilt + +#endif // ABACUS_SOURCE_HAMILT_MODULE_HCONTAINER_HCONTAINER_FUNCS_H diff --git a/source/source_hamilt/module_hcontainer/output_hcontainer.h b/source/source_hamilt/module_hcontainer/output_hcontainer.h index f8ba6b85273..2c895ccd9d6 100644 --- a/source/source_hamilt/module_hcontainer/output_hcontainer.h +++ b/source/source_hamilt/module_hcontainer/output_hcontainer.h @@ -14,9 +14,9 @@ class Output_HContainer { public: Output_HContainer(hamilt::HContainer* hcontainer, - std::ostream& ofs, - double sparse_threshold = -1, - int precision = -1); + std::ostream& ofs, + double sparse_threshold = -1, + int precision = -1); // write the matrices of all R vectors to the output stream void write(bool write_empty = false); diff --git a/source/source_hamilt/module_hcontainer/read_hcontainer.cpp b/source/source_hamilt/module_hcontainer/read_hcontainer.cpp index 662d6c1ad54..9f9a57912a9 100644 --- a/source/source_hamilt/module_hcontainer/read_hcontainer.cpp +++ b/source/source_hamilt/module_hcontainer/read_hcontainer.cpp @@ -3,6 +3,7 @@ #include "source_base/module_out/sparse_matrix.h" #include "source_base/module_out/csr_reader.h" #include "hcontainer_funcs.h" +#include "transfer.h" #include @@ -131,13 +132,13 @@ void Read_HContainer::read() std::vector tmp_ijrs; MPI_Status status; long tmp_size = 0; - MPI_Recv(&tmp_size, 1, MPI_LONG, 0, 0, MPI_COMM_WORLD, &status); + MPI_Recv(&tmp_size, 1, MPI_LONG, 0, kHctTagSize, MPI_COMM_WORLD, &status); tmp_ijrs.resize(tmp_size); MPI_Recv(tmp_ijrs.data(), tmp_ijrs.size(), MPI_INT, 0, - 1, + kHctTagIndex, MPI_COMM_WORLD, &status); this->_hcontainer->insert_ijrs(&tmp_ijrs); @@ -148,8 +149,8 @@ void Read_HContainer::read() for (int i = 1; i < size; ++i) { long tmp_size = para_ijrs.size(); - MPI_Send(&tmp_size, 1, MPI_LONG, i, 0, MPI_COMM_WORLD); - MPI_Send(para_ijrs.data(), para_ijrs.size(), MPI_INT, i, 1, MPI_COMM_WORLD); + MPI_Send(&tmp_size, 1, MPI_LONG, i, kHctTagSize, MPI_COMM_WORLD); + MPI_Send(para_ijrs.data(), para_ijrs.size(), MPI_INT, i, kHctTagIndex, MPI_COMM_WORLD); } } // gather values from serial_rank to Parallels diff --git a/source/source_hamilt/module_hcontainer/test/CMakeLists.txt b/source/source_hamilt/module_hcontainer/test/CMakeLists.txt index 98ffd8e0094..74556680149 100644 --- a/source/source_hamilt/module_hcontainer/test/CMakeLists.txt +++ b/source/source_hamilt/module_hcontainer/test/CMakeLists.txt @@ -14,12 +14,15 @@ AddTest( ../../../source_basis/module_ao/parallel_orbitals.cpp tmp_mocks.cpp ) -AddTest( - TARGET MODULE_LCAO_hcontainer_cost_test - LIBS parameter psi base device - SOURCES test_hcontainer_time.cpp ../base_matrix.cpp ../hcontainer.cpp ../atom_pair.cpp - ../../../source_basis/module_ao/parallel_orbitals.cpp tmp_mocks.cpp -) +# This driver only measures time and memory; keep it with the opt-in benchmarks. +if(ENABLE_GOOGLEBENCH) + AddTest( + TARGET MODULE_LCAO_hcontainer_cost_test + LIBS parameter psi base device + SOURCES test_hcontainer_time.cpp ../base_matrix.cpp ../hcontainer.cpp ../atom_pair.cpp + ../../../source_basis/module_ao/parallel_orbitals.cpp tmp_mocks.cpp + ) +endif() AddTest( TARGET MODULE_LCAO_hcontainer_folding_test diff --git a/source/source_hamilt/module_hcontainer/test/test_hcontainer.cpp b/source/source_hamilt/module_hcontainer/test/test_hcontainer.cpp index 991b074a4a0..172cc088e5b 100644 --- a/source/source_hamilt/module_hcontainer/test/test_hcontainer.cpp +++ b/source/source_hamilt/module_hcontainer/test/test_hcontainer.cpp @@ -109,6 +109,44 @@ TEST_F(HContainerTest, insert_pair) hamilt::HContainer HR_move(std::move(HR_copy)); } +TEST(HContainerSparseTest, InsertPairsOutOfOrder) +{ + hamilt::HContainer container(3); + const int insertion_order[] = {2, 0, 1}; + for (const int atom_j : insertion_order) + { + hamilt::AtomPair pair(0, atom_j); + pair.set_size(1, 1); + for (int rz = 0; rz < 3; ++rz) + { + pair.get_HR_values(0, 0, rz).get_value(0, 0) = 10.0 * atom_j + rz; + } + container.insert_pair(pair); + } + + EXPECT_EQ(container.size_atom_pairs(), 3); + EXPECT_EQ(container.get_nnr(), 9); + const size_t memory = container.get_memory_size(); + EXPECT_GT(memory, sizeof(container)); + for (int atom_j = 0; atom_j < 3; ++atom_j) + { + const hamilt::AtomPair* pair = container.find_pair(0, atom_j); + ASSERT_NE(pair, nullptr); + EXPECT_EQ(pair->get_atom_i(), 0); + EXPECT_EQ(pair->get_atom_j(), atom_j); + EXPECT_EQ(pair->get_R_size(), 3); + for (int rz = 0; rz < 3; ++rz) + { + EXPECT_EQ(pair->find_R(0, 0, rz), rz); + const double expected = 10.0 * atom_j + rz; + const hamilt::BaseMatrix* matrix = container.find_matrix(0, atom_j, 0, 0, rz); + ASSERT_NE(matrix, nullptr); + EXPECT_DOUBLE_EQ(matrix->get_value(0, 0), expected); + } + EXPECT_EQ(pair->find_R(0, 0, 3), -1); + } +} + // using TEST_F to test HContainer::find_pair TEST_F(HContainerTest, find_pair) { diff --git a/source/source_hamilt/module_hcontainer/transfer.cpp b/source/source_hamilt/module_hcontainer/transfer.cpp index 9fd879525e8..0b3b1980fa7 100644 --- a/source/source_hamilt/module_hcontainer/transfer.cpp +++ b/source/source_hamilt/module_hcontainer/transfer.cpp @@ -95,9 +95,9 @@ void HTransPara::receive_ap_indexes(int irank, const int* ap_indexes_in, cons if (ap_indexes_in == nullptr) { long size_ap_indexes = 0; - MPI_Recv(&size_ap_indexes, 1, MPI_LONG, irank, 0, MPI_COMM_WORLD, MPI_STATUS_IGNORE); + MPI_Recv(&size_ap_indexes, 1, MPI_LONG, irank, kHctTagSize, MPI_COMM_WORLD, MPI_STATUS_IGNORE); this->ap_indexes[irank].resize(size_ap_indexes); - MPI_Recv(this->ap_indexes[irank].data(), size_ap_indexes, MPI_INT, irank, 1, MPI_COMM_WORLD, MPI_STATUS_IGNORE); + MPI_Recv(this->ap_indexes[irank].data(), size_ap_indexes, MPI_INT, irank, kHctTagIndex, MPI_COMM_WORLD, MPI_STATUS_IGNORE); } // sender and receiver are same process else @@ -146,13 +146,13 @@ void HTransPara::send_orb_indexes(int irank, MPI_Request* request) long size_orb_indexes = orb_indexes.size(); if (request != nullptr) { - MPI_Isend(&size_orb_indexes, 1, MPI_LONG, irank, 0, MPI_COMM_WORLD, request); - MPI_Isend(orb_indexes.data(), orb_indexes.size(), MPI_INT, irank, 1, MPI_COMM_WORLD, request); + MPI_Isend(&size_orb_indexes, 1, MPI_LONG, irank, kHctTagSize, MPI_COMM_WORLD, request); + MPI_Isend(orb_indexes.data(), orb_indexes.size(), MPI_INT, irank, kHctTagIndex, MPI_COMM_WORLD, request); } else { - MPI_Send(&size_orb_indexes, 1, MPI_LONG, irank, 0, MPI_COMM_WORLD); - MPI_Send(orb_indexes.data(), orb_indexes.size(), MPI_INT, irank, 1, MPI_COMM_WORLD); + MPI_Send(&size_orb_indexes, 1, MPI_LONG, irank, kHctTagSize, MPI_COMM_WORLD); + MPI_Send(orb_indexes.data(), orb_indexes.size(), MPI_INT, irank, kHctTagIndex, MPI_COMM_WORLD); } } @@ -163,11 +163,11 @@ void HTransPara::send_data(int irank, MPI_Request* request) this->pack_data(irank, values.data()); if (request != nullptr) { - MPI_Isend(values.data(), values.size(), MPITraits::datatype(), irank, 2, MPI_COMM_WORLD, request); + MPI_Isend(values.data(), values.size(), MPITraits::datatype(), irank, kHctTagValue, MPI_COMM_WORLD, request); } else { - MPI_Send(values.data(), values.size(), MPITraits::datatype(), irank, 2, MPI_COMM_WORLD); + MPI_Send(values.data(), values.size(), MPITraits::datatype(), irank, kHctTagValue, MPI_COMM_WORLD); } } @@ -182,7 +182,7 @@ void HTransPara::receive_data(int irank, const T* values) values_tmp.size(), MPITraits::datatype(), irank, - 0, + kHctTagValue, MPI_COMM_WORLD, MPI_STATUS_IGNORE); this->unpack_data(irank, values_tmp.data()); @@ -384,13 +384,13 @@ void HTransSerial::send_ap_indexes(int irank, MPI_Request* request) long size_ap_indexes = ap_indexes.size(); if (request != nullptr) { - MPI_Isend(&size_ap_indexes, 1, MPI_LONG, irank, 0, MPI_COMM_WORLD, request); - MPI_Isend(ap_indexes.data(), ap_indexes.size(), MPI_INT, irank, 1, MPI_COMM_WORLD, request); + MPI_Isend(&size_ap_indexes, 1, MPI_LONG, irank, kHctTagSize, MPI_COMM_WORLD, request); + MPI_Isend(ap_indexes.data(), ap_indexes.size(), MPI_INT, irank, kHctTagIndex, MPI_COMM_WORLD, request); } else { - MPI_Send(&size_ap_indexes, 1, MPI_LONG, irank, 0, MPI_COMM_WORLD); - MPI_Send(ap_indexes.data(), ap_indexes.size(), MPI_INT, irank, 1, MPI_COMM_WORLD); + MPI_Send(&size_ap_indexes, 1, MPI_LONG, irank, kHctTagSize, MPI_COMM_WORLD); + MPI_Send(ap_indexes.data(), ap_indexes.size(), MPI_INT, irank, kHctTagIndex, MPI_COMM_WORLD); } } @@ -402,13 +402,13 @@ void HTransSerial::receive_orb_indexes(int irank, const int* orb_indexes_in, if (orb_indexes_in == nullptr) { long size_orb_indexes = 0; - MPI_Recv(&size_orb_indexes, 1, MPI_LONG, irank, 0, MPI_COMM_WORLD, MPI_STATUS_IGNORE); + MPI_Recv(&size_orb_indexes, 1, MPI_LONG, irank, kHctTagSize, MPI_COMM_WORLD, MPI_STATUS_IGNORE); this->orb_indexes[irank].resize(size_orb_indexes); MPI_Recv(this->orb_indexes[irank].data(), size_orb_indexes, MPI_INT, irank, - 1, + kHctTagIndex, MPI_COMM_WORLD, MPI_STATUS_IGNORE); } @@ -455,11 +455,11 @@ void HTransSerial::send_data(int irank, MPI_Request* request) this->pack_data(irank, values.data()); if (request != nullptr) { - MPI_Isend(values.data(), values.size(), MPITraits::datatype(), irank, 2, MPI_COMM_WORLD, request); + MPI_Isend(values.data(), values.size(), MPITraits::datatype(), irank, kHctTagValue, MPI_COMM_WORLD, request); } else { - MPI_Send(values.data(), values.size(), MPITraits::datatype(), irank, 2, MPI_COMM_WORLD); + MPI_Send(values.data(), values.size(), MPITraits::datatype(), irank, kHctTagValue, MPI_COMM_WORLD); } } @@ -474,7 +474,7 @@ void HTransSerial::receive_data(int irank, const T* values) values_tmp.size(), MPITraits::datatype(), irank, - 2, + kHctTagValue, MPI_COMM_WORLD, MPI_STATUS_IGNORE); this->unpack_data(irank, values_tmp.data()); diff --git a/source/source_hamilt/module_hcontainer/transfer.h b/source/source_hamilt/module_hcontainer/transfer.h index deea0f5d7ce..07748913bab 100644 --- a/source/source_hamilt/module_hcontainer/transfer.h +++ b/source/source_hamilt/module_hcontainer/transfer.h @@ -10,6 +10,18 @@ namespace hamilt { +// MPI tags for the HContainer serial<->parallel transfer (transfer.cpp) and +// Read_HContainer::read() (read_hcontainer.cpp). They must match between the +// send and the recv side, so they live here to be shared. Deliberately odd, +// high values: the transfer previously used fixed tags 0/1/2 on +// MPI_COMM_WORLD, which collide with unrelated p2p traffic (write_eig_occ, +// output_log, func_transfer, ...) on any other call path and cause +// MPI_ERR_TRUNCATE races or deadlocks. Moving to these uncommon tags removes +// the need for the masking MPI_Barrier calls in op_exx_lcao.cpp. +constexpr int kHctTagSize = 14431; // payload size (MPI_LONG) +constexpr int kHctTagIndex = 14471; // atom/orbital index array (MPI_INT) +constexpr int kHctTagValue = 14533; // matrix values (T) + template class HTransPara { diff --git a/source/source_hamilt/module_surchem/h_corr_pw.cpp b/source/source_hamilt/module_surchem/h_corr_pw.cpp index fa175aa36e2..9cdb503fa42 100644 --- a/source/source_hamilt/module_surchem/h_corr_pw.cpp +++ b/source/source_hamilt/module_surchem/h_corr_pw.cpp @@ -21,18 +21,18 @@ void surchem::v_correction(const UnitCell& cell, assert(rho_basis->nrxx>0); double* porter = new double[rho_basis->nrxx]; - for (int i = 0; i < rho_basis->nrxx; i++) - { - porter[i] = 0.0; - } + for (int i = 0; i < rho_basis->nrxx; i++) + { + porter[i] = 0.0; + } const int nspin0 = (nspin == 2) ? 2 : 1; - for (int is = 0; is < nspin0; is++) - { - for (int ir = 0; ir < rho_basis->nrxx; ir++) - { - porter[ir] += rho[is][ir]; - } - } + for (int is = 0; is < nspin0; is++) + { + for (int ir = 0; ir < rho_basis->nrxx; ir++) + { + porter[ir] += rho[is][ir]; + } + } std::complex* porter_g = new std::complex[rho_basis->npw]; ModuleBase::GlobalFunc::ZEROS(porter_g, rho_basis->npw); diff --git a/source/source_hamilt/module_surchem/sol_force.cpp b/source/source_hamilt/module_surchem/sol_force.cpp index 48f36309ad2..bd133594899 100644 --- a/source/source_hamilt/module_surchem/sol_force.cpp +++ b/source/source_hamilt/module_surchem/sol_force.cpp @@ -83,9 +83,9 @@ void surchem::force_cor_two(const UnitCell& cell, ModuleBase::GlobalFunc::ZEROS(Vcav_g, rho_basis->npw); ModuleBase::GlobalFunc::ZEROS(Vel_g, rho_basis->npw); for(int is=0; isnrxx; ir++) - { + { + for (int ir=0; irnrxx; ir++) + { Vcav_sum[ir] += this->Vcav(is, ir); } } @@ -159,7 +159,7 @@ void surchem::cal_force_sol(const UnitCell& cell, ModuleBase::timer::start("surchem", "cal_force_sol"); int nat = cell.nat; - ModuleBase::matrix force1(nat, 3); + ModuleBase::matrix force1(nat, 3); ModuleBase::matrix force2(nat, 3); force_cor_one(cell, rho_basis, vloc, force1); @@ -167,15 +167,15 @@ void surchem::cal_force_sol(const UnitCell& cell, int iat = 0; for (int it = 0;it < cell.ntype;it++) - { - for (int ia = 0;ia < cell.atoms[it].na;ia++) - { + { + for (int ia = 0;ia < cell.atoms[it].na;ia++) + { for(int ipol = 0; ipol < 3; ipol++) { forcesol(iat, ipol) = 0.5*force1(iat, ipol) + force2 (iat, ipol); } - - ++iat; + + ++iat; } } diff --git a/source/source_hamilt/module_surchem/test/CMakeLists.txt b/source/source_hamilt/module_surchem/test/CMakeLists.txt index 50c14955cee..b843b7cb6e1 100644 --- a/source/source_hamilt/module_surchem/test/CMakeLists.txt +++ b/source/source_hamilt/module_surchem/test/CMakeLists.txt @@ -29,6 +29,7 @@ AddTest( LIBS parameter planewave device base container SOURCES cal_vcav_test.cpp ../cal_vcav.cpp ../surchem.cpp ../../module_xc/xc_grad.cpp ../../module_xc/xc_grad_prepare.cpp ../../module_xc/xc_grad_kernel.cpp ../../module_xc/xc_grad_assemble.cpp ../../module_xc/xc_grad_wfc.cpp ../../module_xc/xc_grad_utils.cpp ../../module_xc/xc_functional.cpp + ../../module_xc/xc_functional_ncgga_sf.cpp ../../module_xc/xc_ncgga_radial.cpp ../../module_xc/xc_lda_wrap.cpp ../../module_xc/xc_gga_wrap.cpp ../../module_xc/libxc_setup.cpp ../../module_xc/libxc_pot.cpp @@ -44,6 +45,7 @@ AddTest( LIBS parameter planewave device base container SOURCES cal_vel_test.cpp ../cal_vel.cpp ../surchem.cpp ../cal_epsilon.cpp ../minimize_cg.cpp ../../module_xc/xc_grad.cpp ../../module_xc/xc_grad_prepare.cpp ../../module_xc/xc_grad_kernel.cpp ../../module_xc/xc_grad_assemble.cpp ../../module_xc/xc_grad_wfc.cpp ../../module_xc/xc_grad_utils.cpp ../../module_xc/xc_functional.cpp + ../../module_xc/xc_functional_ncgga_sf.cpp ../../module_xc/xc_ncgga_radial.cpp ../../module_xc/xc_lda_wrap.cpp ../../module_xc/xc_gga_wrap.cpp ../../module_xc/libxc_setup.cpp ../../module_xc/libxc_pot.cpp diff --git a/source/source_hamilt/module_surchem/test/cal_pseudo_test.cpp b/source/source_hamilt/module_surchem/test/cal_pseudo_test.cpp index e3e237547ab..e5e1fa8b72c 100644 --- a/source/source_hamilt/module_surchem/test/cal_pseudo_test.cpp +++ b/source/source_hamilt/module_surchem/test/cal_pseudo_test.cpp @@ -74,7 +74,7 @@ TEST_F(cal_pseudo_test, gauss_charge) Structure_Factor sf; sf.nbspline = -1; - sf.setup(&ucell, pgrid, &pwtest); + sf.setup(&ucell, pgrid, &pwtest, false); solvent_model.gauss_charge(ucell, pgrid, &pwtest, N, &sf); @@ -124,7 +124,7 @@ TEST_F(cal_pseudo_test, cal_pseudo) Structure_Factor sf; sf.nbspline = -1; - sf.setup(&ucell, pgrid, &pwtest); // sf.setup is moved to here + sf.setup(&ucell, pgrid, &pwtest, false); // sf.setup is moved to here std::complex* Porter_g = new std::complex[npw]; ModuleBase::GlobalFunc::ZEROS(Porter_g, npw); diff --git a/source/source_hamilt/module_vdw/test/CMakeLists.txt b/source/source_hamilt/module_vdw/test/CMakeLists.txt index 5ca828771f1..195a5312ce8 100644 --- a/source/source_hamilt/module_vdw/test/CMakeLists.txt +++ b/source/source_hamilt/module_vdw/test/CMakeLists.txt @@ -4,6 +4,7 @@ abacus_disable_feature_definitions(__ROCM) install(FILES c6.txt DESTINATION ${CMAKE_CURRENT_BINARY_DIR}) install(FILES r0.txt DESTINATION ${CMAKE_CURRENT_BINARY_DIR}) +install(FILES r0_zero.txt DESTINATION ${CMAKE_CURRENT_BINARY_DIR}) AddTest( TARGET MODULE_HAMILT_vdwTest @@ -11,6 +12,12 @@ AddTest( SOURCES vdw_test.cpp vdwd3_evaluator_test.cpp ) -if(ENABLE_DFTD4) - target_link_libraries(MODULE_HAMILT_vdwTest PRIVATE dftd4::dftd4) +if(ENABLE_DFTD4 AND ENABLE_MPI) + add_test( + NAME MODULE_HAMILT_vdwd4Test_2np + COMMAND ${MPIEXEC_EXECUTABLE} ${MPIEXEC_NUMPROC_FLAG} 2 + ${MPIEXEC_PREFLAGS} $ + --gtest_filter=vdwd4Test.* ${MPIEXEC_POSTFLAGS} + WORKING_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR} + ) endif() diff --git a/source/source_hamilt/module_vdw/test/r0_zero.txt b/source/source_hamilt/module_vdw/test/r0_zero.txt new file mode 100644 index 00000000000..fafc24c5ee0 --- /dev/null +++ b/source/source_hamilt/module_vdw/test/r0_zero.txt @@ -0,0 +1 @@ +Si 0.0 diff --git a/source/source_hamilt/module_vdw/test/vdw_test.cpp b/source/source_hamilt/module_vdw/test/vdw_test.cpp index 5d09ffe7658..f03f4fb1588 100644 --- a/source/source_hamilt/module_vdw/test/vdw_test.cpp +++ b/source/source_hamilt/module_vdw/test/vdw_test.cpp @@ -5,10 +5,8 @@ #include"gtest/gtest.h" #include"gmock/gmock.h" #include "mpi.h" -#define private public #include "source_hamilt/module_vdw/vdwd2_parameters.h" #include "source_hamilt/module_vdw/vdwd2.h" -#undef private #include "source_hamilt/module_vdw/vdwd3.h" #ifdef __DFTD4 #include "source_hamilt/module_vdw/vdwd4.h" @@ -239,10 +237,10 @@ TEST_F(vdwd2Test, D2Default) EXPECT_EQ(vdwd2_test.parameter().scaling(), 0.75); EXPECT_EQ(vdwd2_test.parameter().damping(), input.vdw_d); EXPECT_EQ(vdwd2_test.parameter().model(), input.vdw_cutoff_type); - EXPECT_EQ(vdwd2_test.parameter().radius_, 56.6918); + EXPECT_EQ(vdwd2_test.parameter().radius(), 56.6918); double Si_C6 = 9.23*1e6 / (ModuleBase::ELECTRONVOLT_SI * ModuleBase::NA) / pow(ModuleBase::BOHR_TO_A, 6)/ ModuleBase::Ry_to_eV; - EXPECT_NEAR(vdwd2_test.parameter().C6_["Si"], Si_C6,1e-13); - EXPECT_EQ(vdwd2_test.parameter().R0_["Si"], 1.716/ModuleBase::BOHR_TO_A); + EXPECT_NEAR(vdwd2_test.parameter().C6().at("Si"), Si_C6,1e-13); + EXPECT_EQ(vdwd2_test.parameter().R0().at("Si"), 1.716/ModuleBase::BOHR_TO_A); EXPECT_EQ(vdwd2_test.parameter().period().x, 2 * ceil(56.6918 / ucell.lat0 / sqrt(ucell.a1.norm2())) + 1); EXPECT_EQ(vdwd2_test.parameter().period().y, 2 * ceil(56.6918 / ucell.lat0 / sqrt(ucell.a2.norm2())) + 1); EXPECT_EQ(vdwd2_test.parameter().period().z, 2 * ceil(56.6918 / ucell.lat0 / sqrt(ucell.a3.norm2())) + 1); @@ -265,8 +263,8 @@ TEST_F(vdwd2Test, D2ReadFile) vdwd2_test.parameter().initial_parameters(input); double Si_C6 = 9.13*1e6 / (ModuleBase::ELECTRONVOLT_SI * ModuleBase::NA) / pow(ModuleBase::BOHR_TO_A, 6)/ ModuleBase::Ry_to_eV; - EXPECT_NEAR(vdwd2_test.parameter().C6_["Si"], Si_C6,1e-13); - EXPECT_EQ(vdwd2_test.parameter().R0_["Si"], 1.626/ModuleBase::BOHR_TO_A); + EXPECT_NEAR(vdwd2_test.parameter().C6().at("Si"), Si_C6,1e-13); + EXPECT_EQ(vdwd2_test.parameter().R0().at("Si"), 1.626/ModuleBase::BOHR_TO_A); } TEST_F(vdwd2Test, D2ReadFileError) @@ -288,7 +286,7 @@ TEST_F(vdwd2Test, D2c6UniteVA6) vdwd2_test.parameter().initial_parameters(input); double Si_C6 = 9.23 / pow(ModuleBase::BOHR_TO_A, 6) * ModuleBase::Ry_to_eV; - EXPECT_NEAR(vdwd2_test.parameter().C6_["Si"], Si_C6,1e-13); + EXPECT_NEAR(vdwd2_test.parameter().C6().at("Si"), Si_C6,1e-13); } TEST_F(vdwd2Test, D2r0UnitBohr) @@ -297,7 +295,7 @@ TEST_F(vdwd2Test, D2r0UnitBohr) vdw::Vdwd2 vdwd2_test(ucell); vdwd2_test.parameter().initial_parameters(input); - EXPECT_EQ(vdwd2_test.parameter().R0_["Si"], 1.716); + EXPECT_EQ(vdwd2_test.parameter().R0().at("Si"), 1.716); } TEST_F(vdwd2Test, D2WrongUnit) @@ -319,7 +317,7 @@ TEST_F(vdwd2Test, D2RadiusUnitAngstrom) vdw::Vdwd2 vdwd2_test(ucell); vdwd2_test.parameter().initial_parameters(input); - EXPECT_EQ(vdwd2_test.parameter().radius_, 56.6918/ModuleBase::BOHR_TO_A); + EXPECT_EQ(vdwd2_test.parameter().radius(), 56.6918/ModuleBase::BOHR_TO_A); } TEST_F(vdwd2Test, D2CutoffTypePeriod) @@ -336,7 +334,9 @@ TEST_F(vdwd2Test, D2R0ZeroQuit) { vdw::Vdwd2 vdwd2_test(ucell); vdwd2_test.parameter().initial_parameters(input); - vdwd2_test.parameter().R0_["Si"] = 0.0; + // r0_zero.txt sets R0 for Si to zero through the public reader, so that + // Vdwd2::index_loops hits its "R0_sum can not be 0" guard. + vdwd2_test.parameter().R0_input("r0_zero.txt", "Bohr"); testing::internal::CaptureStdout(); EXPECT_EXIT(vdwd2_test.evaluate(vdw::VdwRequest(false, false)), ::testing::ExitedWithCode(1), ""); diff --git a/source/source_hamilt/module_vdw/vdwd2_parameters.h b/source/source_hamilt/module_vdw/vdwd2_parameters.h index d1bb2a54bcb..a22fe350320 100644 --- a/source/source_hamilt/module_vdw/vdwd2_parameters.h +++ b/source/source_hamilt/module_vdw/vdwd2_parameters.h @@ -48,6 +48,8 @@ class Vdwd2Parameters : public VdwParameters inline const std::map R0() const { return R0_; } inline double damping() const { return damping_; } inline double scaling() const { return scaling_; } + /// @brief cutoff radius of the D2 summation, in Bohr + inline double radius() const { return radius_; } private: double scaling_; diff --git a/source/source_hamilt/module_vdw/vdwd4.cpp b/source/source_hamilt/module_vdw/vdwd4.cpp index 72caf51232a..ad4a93e9925 100644 --- a/source/source_hamilt/module_vdw/vdwd4.cpp +++ b/source/source_hamilt/module_vdw/vdwd4.cpp @@ -3,6 +3,10 @@ #include "source_base/constants.h" #include "source_base/element_name.h" +#ifdef __MPI +#include "source_base/parallel_reduce.h" +#include +#endif #include "source_base/timer.h" #include "source_base/tool_quit.h" @@ -231,6 +235,15 @@ void Vdwd4::compute(double& energy_ha, smooth_width_3b_); check_dftd4_error(error, "dftd4_set_model_realspace_cutoff_smooth"); +#ifdef __MPI + int rank = 0; + int nranks = 1; + MPI_Comm_rank(MPI_COMM_WORLD, &rank); + MPI_Comm_size(MPI_COMM_WORLD, &nranks); + dftd4_set_model_work_partition(error, model, rank, nranks); + check_dftd4_error(error, "dftd4_set_model_work_partition"); +#endif + std::vector method(xc_name_.begin(), xc_name_.end()); method.push_back('\0'); @@ -247,6 +260,18 @@ void Vdwd4::compute(double& energy_ha, dftd4_get_dispersion(error, mol, model, param, &energy_ha, gradient, sigma); check_dftd4_error(error, "dftd4_get_dispersion"); +#ifdef __MPI + Parallel_Reduce::reduce_all(energy_ha); + if (gradient != nullptr) + { + Parallel_Reduce::reduce_all(gradient, 3 * ucell_.nat); + } + if (sigma != nullptr) + { + Parallel_Reduce::reduce_all(sigma, 9); + } +#endif + dftd4_delete_param(¶m); dftd4_delete_model(&model); dftd4_delete_structure(&mol); diff --git a/source/source_hamilt/module_xc/CMakeLists.txt b/source/source_hamilt/module_xc/CMakeLists.txt index 0b6309a5a24..495c5ae1997 100644 --- a/source/source_hamilt/module_xc/CMakeLists.txt +++ b/source/source_hamilt/module_xc/CMakeLists.txt @@ -2,6 +2,8 @@ add_library( xc_ OBJECT xc_functional.cpp + xc_functional_ncgga_sf.cpp + xc_ncgga_radial.cpp xc_pot.cpp xc_grad.cpp xc_grad_prepare.cpp diff --git a/source/source_hamilt/module_xc/exx_info.cpp b/source/source_hamilt/module_xc/exx_info.cpp index 923fef63d44..7098c1dd7d5 100644 --- a/source/source_hamilt/module_xc/exx_info.cpp +++ b/source/source_hamilt/module_xc/exx_info.cpp @@ -3,6 +3,7 @@ #include "source_io/module_parameter/input_parameter.h" #include "source_base/global_function.h" +#include "source_base/tool_quit.h" #include #include @@ -177,6 +178,20 @@ bool init_general_exx_info(General_Exx_Info& info, const Input_para& inp) info.separate_loop = inp.exx_separate_loop; info.hybrid_step = inp.exx_hybrid_step; info.mixing_beta_for_loop1 = inp.exx_mixing_beta; + + // PW EXX operator configuration + info.exxace = inp.exxace; + info.gamma_extrapolation = inp.exx_gamma_extra; + info.ecutexx_user_set = inp.ecutexx > 0.0; + info.ecut_exx = info.ecutexx_user_set ? inp.ecutexx : inp.ecutrho; + if (info.ecutexx_user_set && info.ecut_exx > inp.ecutrho) + { + // no pair-density G-components exist beyond the ecutrho sphere, + // and buffers sized by ecutrho would overflow; clamp instead + ModuleBase::WARNING("General_Exx_Info", "ecutexx is larger than ecutrho; ecutexx is set to ecutrho"); + info.ecut_exx = inp.ecutrho; + } + info.exx_batch_size = inp.exx_batch_size; } return generate_opt_orb; diff --git a/source/source_hamilt/module_xc/general_exx_info.h b/source/source_hamilt/module_xc/general_exx_info.h index 9ab1d6ec33e..8d678299c59 100644 --- a/source/source_hamilt/module_xc/general_exx_info.h +++ b/source/source_hamilt/module_xc/general_exx_info.h @@ -31,6 +31,13 @@ struct General_Exx_Info bool separate_loop = true; size_t hybrid_step = 1; + + // PW EXX operator configuration: + bool exxace = false; // ACE acceleration of the EXX operator (INPUT exxace) + bool gamma_extrapolation = true; // INPUT exx_gamma_extra + double ecut_exx = 0.0; // resolved EXX cutoff (Ry): ecutexx when set, else ecutrho + bool ecutexx_user_set = false; // ecutexx explicitly set (controls small-grid fallback warnings) + int exx_batch_size = 0; // band chunk width of the batched EXX path; 0 = all bands }; /// Forward declaration for Input_para (full definition in input_parameter.h) diff --git a/source/source_hamilt/module_xc/kernels/cuda/xc_functional_op.cu b/source/source_hamilt/module_xc/kernels/cuda/xc_functional_op.cu index 6ee81727c4e..687f5d4f795 100644 --- a/source/source_hamilt/module_xc/kernels/cuda/xc_functional_op.cu +++ b/source/source_hamilt/module_xc/kernels/cuda/xc_functional_op.cu @@ -13,7 +13,7 @@ __global__ void xc_functional_grad_wfc( const int pol, const int npw, const int npwx, - const T tpiba, + const T tpiba, const T* gcar, const T* kvec_c, const thrust::complex* rhog, @@ -21,12 +21,12 @@ __global__ void xc_functional_grad_wfc( { int idx = threadIdx.x + blockIdx.x * blockDim.x; if (idx >= npw) { return; } - // the formula is : rho(r)^prime = \int iG * rho(G)e^{iGr} dG - // double kplusg = wfc_basis->getgpluskcar(ik,ig)[ipol] * tpiba; + // the formula is : rho(r)^prime = \int iG * rho(G)e^{iGr} dG + // double kplusg = wfc_basis->getgpluskcar(ik,ig)[ipol] * tpiba; T kplusg = (gcar[(ik * npwx + idx) * 3 + pol] + kvec_c[ik * 3 + pol]) * tpiba; - // calculate the charge density gradient in reciprocal space. - porter[idx] = thrust::complex(0.0, kplusg) * rhog[idx]; + // calculate the charge density gradient in reciprocal space. + porter[idx] = thrust::complex(0.0, kplusg) * rhog[idx]; } template @@ -47,7 +47,7 @@ void xc_functional_grad_wfc_op::operator()( const int& pol, const int& npw, const int& npwx, - const Real& tpiba, + const Real& tpiba, const Real * gcar, const Real * kvec_c, const T * rhog, diff --git a/source/source_hamilt/module_xc/kernels/rocm/xc_functional_op.hip.cu b/source/source_hamilt/module_xc/kernels/rocm/xc_functional_op.hip.cu index 0e739276622..bce87f57d1b 100644 --- a/source/source_hamilt/module_xc/kernels/rocm/xc_functional_op.hip.cu +++ b/source/source_hamilt/module_xc/kernels/rocm/xc_functional_op.hip.cu @@ -14,7 +14,7 @@ __global__ void xc_functional_grad_wfc( const int pol, const int npw, const int npwx, - const T tpiba, + const T tpiba, const T* gcar, const T* kvec_c, const thrust::complex* rhog, @@ -22,12 +22,12 @@ __global__ void xc_functional_grad_wfc( { int idx = threadIdx.x + blockIdx.x * blockDim.x; if (idx >= npw) { return; } - // the formula is : rho(r)^prime = \int iG * rho(G)e^{iGr} dG - // double kplusg = wfc_basis->getgpluskcar(ik,ig)[ipol] * tpiba; + // the formula is : rho(r)^prime = \int iG * rho(G)e^{iGr} dG + // double kplusg = wfc_basis->getgpluskcar(ik,ig)[ipol] * tpiba; T kplusg = (gcar[(ik * npwx + idx) * 3 + pol] + kvec_c[ik * 3 + pol]) * tpiba; - // calculate the charge density gradient in reciprocal space. - porter[idx] = thrust::complex(0.0, kplusg) * rhog[idx]; + // calculate the charge density gradient in reciprocal space. + porter[idx] = thrust::complex(0.0, kplusg) * rhog[idx]; } template @@ -48,7 +48,7 @@ void xc_functional_grad_wfc_op::operator()( const int& pol, const int& npw, const int& npwx, - const Real& tpiba, + const Real& tpiba, const Real * gcar, const Real * kvec_c, const T * rhog, diff --git a/source/source_hamilt/module_xc/kernels/xc_functional_op.cpp b/source/source_hamilt/module_xc/kernels/xc_functional_op.cpp index 6095f45109f..ec67012675d 100644 --- a/source/source_hamilt/module_xc/kernels/xc_functional_op.cpp +++ b/source/source_hamilt/module_xc/kernels/xc_functional_op.cpp @@ -8,7 +8,7 @@ void xc_functional_grad_wfc_op::operator()( const int& pol, const int& npw, const int& npwx, - const Real& tpiba, + const Real& tpiba, const Real * gcar, const Real * kvec_c, const T * rhog, @@ -18,14 +18,14 @@ void xc_functional_grad_wfc_op::operator()( #pragma omp parallel for schedule(static, 1024) #endif for(int ig = 0; ig < npw; ig++) { - // the formula is : rho(r)^prime = \int iG * rho(G)e^{iGr} dG - // double kplusg = wfc_basis->getgpluskcar(ik,ig)[ipol] * tpiba; + // the formula is : rho(r)^prime = \int iG * rho(G)e^{iGr} dG + // double kplusg = wfc_basis->getgpluskcar(ik,ig)[ipol] * tpiba; Real kplusg = (gcar[(ik * npwx + ig) * 3 + pol] + kvec_c[ik * 3 + pol]) * tpiba; - // calculate the charge density gradient in reciprocal space. - porter[ig] = T(0.0, kplusg) * rhog[ig]; - } + // calculate the charge density gradient in reciprocal space. + porter[ig] = T(0.0, kplusg) * rhog[ig]; + } } template @@ -40,7 +40,7 @@ void xc_functional_grad_wfc_op::operator()( #endif for (int ir = 0; ir < nrxx; ++ir) { grad[ipol * nrxx + ir] = porter[ir]; - } + } } template struct xc_functional_grad_wfc_op, base_device::DEVICE_CPU>; diff --git a/source/source_hamilt/module_xc/libxc_abacus.h b/source/source_hamilt/module_xc/libxc_abacus.h index 087a3577fc5..b618bbbe908 100644 --- a/source/source_hamilt/module_xc/libxc_abacus.h +++ b/source/source_hamilt/module_xc/libxc_abacus.h @@ -5,10 +5,12 @@ #include "source_base/matrix.h" #include "source_base/vector3.h" +#include "xc_ncgga_radial.h" #include #include +#include #include #include @@ -19,28 +21,48 @@ class Charge; namespace XC_Functional_Libxc { + struct LibxcWeightedDerivatives + { + double energy_sum; + std::vector drho; + std::vector dsigma; + }; + + // Complete forward data for the gga_grad=2 noncollinear Libxc graph: + // rho_s = N_s(x), + // g_s = sum_A (d N_s / d x_A) G_h x_A. + // Keeping the local map and all input gradients together lets the reverse + // use the exact same branch choices and radial Hessian as the forward. + struct NclSfDiscreteData + { + std::vector spin_map; + std::vector rho; + std::vector>> spin_gradient; + std::array>, 3> grad_m; + }; + //------------------- // libxc_setup.cpp //------------------- // sets functional type, which allows combination of LIBXC keyword connected by "+" - // for example, "XC_LDA_X+XC_LDA_C_PZ" + // for example: "XC_LDA_X+XC_LDA_C_PZ" extern std::pair> set_xc_type_libxc(const std::string& xc_func_in); /** * @brief instantiate the XC functional by its ID, and set the external parameters if provided. - * + * * @param func_id libxc ID of functional, see https://libxc.gitlab.io/functionals/ for details * @param xc_polarized 0: unpolarized, 1: spin-polarized - * @return std::vector - * + * @return std::vector + * * @note the functionality of this method is extended by supporting the user-defined - * external parameters of xc. However, there are several functionals' external - * parameters are pre-defined in the code, which herein we call those are - * "in-built" parameters. If the same functional ID is found in both in-built + * external parameters of xc. However, there are several functionals' external + * parameters are pre-defined in the code, which herein we call those are + * "in-built" parameters. If the same functional ID is found in both in-built * and external parameters, the external parameters will overwrite the in-built ones. * The external parameters can be passed here by keywords xc_exch_ext and - * xc_corr_ext in the input file. The expected format would be an XC ID + * xc_corr_ext in the input file. The expected format would be an XC ID * followed by a list of parameters. */ extern std::vector init_func( @@ -65,10 +87,24 @@ namespace XC_Functional_Libxc const int nspin, const bool domag, const bool domag_z, + const int gga_grad, const std::map* scaling_factor, const double hybrid_alpha, const double hse_omega); + // Reciprocal-metric derivative of the exact gga_grad=2 Libxc energy + // graph. The returned lower-triangular tensor is the unnormalized local + // grid sum; Stress_Func performs the pool reduction and divides by nxyz. + extern void gradcorr_ncgga_sf_libxc( + const std::vector& func_id, + const std::size_t nrxx, + const double tpiba, + const Charge* const chr, + const std::map* scaling_factor, + const double hybrid_alpha, + const double hse_omega, + std::vector& stress_gga); + // for mGGA functional extern std::tuple v_xc_meta( const std::vector &func_id, @@ -97,6 +133,25 @@ namespace XC_Functional_Libxc const std::size_t nrxx, const Charge* const chr); + // Build the exact gga_grad=2 local spin map and, when requested, its + // projected FFT-gradient graph. LDA-only callers set need_gradient=false. + extern NclSfDiscreteData make_ncl_sf_discrete_data( + const std::size_t nrxx, + const double tpiba, + const Charge* const chr, + const bool need_gradient); + + // Reverse one aggregate of all scaled Libxc components. The returned + // potential is already in (n,mx,my,mz) representation. An empty dsigma + // selects the LDA-only local reverse and performs no FFT divergence. + extern ModuleBase::matrix reverse_ncl_sf_discrete( + const std::size_t nrxx, + const NclSfDiscreteData& data, + const std::vector& drho, + const std::vector& dsigma, + const double tpiba, + const Charge* const chr); + // calculating grho extern std::vector>> cal_gdr( const int nspin, @@ -129,6 +184,18 @@ namespace XC_Functional_Libxc const std::vector &rho, const std::vector &sigma); + // threshold masks for the xc potential (Quantum ESPRESSO convention): + // the first mask applies to exc and vrho, the second one only to vsigma + extern std::pair, std::vector> cal_sgn_vxc( + const double rho_threshold_vrho, + const double rho_threshold_vsigma, + const double grho_threshold_vsigma, + const xc_func_type &func, + const int nspin, + const std::size_t nrxx, + const std::vector &rho, + const std::vector &sigma); + // converting etxc from exc (libxc=>abacus) extern double convert_etxc( const int nspin, @@ -137,13 +204,28 @@ namespace XC_Functional_Libxc const std::vector &rho, std::vector exc); - // converting vtxc and v from vrho and vsigma (libxc=>abacus) - extern std::pair convert_vtxc_v( + // Reverse the density and sigma sanitizers for the weighted energy + // accumulated by ABACUS. The result is in Hartree units and excludes the + // real-space grid weight and ModuleBase::e2. + extern LibxcWeightedDerivatives make_libxc_weighted_derivatives( const xc_func_type &func, const int nspin, const std::size_t nrxx, const std::vector &sgn, const std::vector &rho, + const std::vector &sigma, + const std::vector &exc, + const std::vector &vrho, + const std::vector &vsigma); + + // Convert collinear LibXC derivatives to the potential. + extern std::pair convert_vtxc_v( + const xc_func_type &func, + const int nspin, + const std::size_t nrxx, + const std::vector &sgn_vrho, + const std::vector &sgn_vsigma, + const std::vector &rho, const std::vector>> &gdr, const std::vector &vrho, const std::vector &vsigma, @@ -161,14 +243,14 @@ namespace XC_Functional_Libxc const Charge* const chr); // convert v for NSPIN=4 + // has_mag: whether the calculation has (noncollinear) magnetization, + // i.e. domag || domag_z extern ModuleBase::matrix convert_v_nspin4( const std::size_t nrxx, const Charge* const chr, const std::vector &amag, const ModuleBase::matrix &v, - const bool domag, - const bool domag_z); - + const bool has_mag); //------------------- // libxc_lda_wrap.cpp diff --git a/source/source_hamilt/module_xc/libxc_mgga_wrap.cpp b/source/source_hamilt/module_xc/libxc_mgga_wrap.cpp index 6ebbd28d235..3fd7f34b53c 100644 --- a/source/source_hamilt/module_xc/libxc_mgga_wrap.cpp +++ b/source/source_hamilt/module_xc/libxc_mgga_wrap.cpp @@ -140,7 +140,7 @@ void XC_Functional_Libxc::tau_xc_spin( std::array v2xc = {0.0, 0.0, 0.0}; // call Libxc function: xc_mgga_exc_vxc xc_mgga_exc_vxc(&func, 1, rho.data(), grho.data(), lapl.data(), tau.data(), &s, - v1xc.data(), v2xc.data(), vlapl_out.data(), v3xc.data()); + v1xc.data(), v2xc.data(), vlapl_out.data(), v3xc.data()); #ifdef __EXX if (func.info->number == XC_MGGA_X_SCAN && XC_Functional::get_func_type() == 5) diff --git a/source/source_hamilt/module_xc/libxc_pot.cpp b/source/source_hamilt/module_xc/libxc_pot.cpp index 12351a45868..9dd5c84dc9e 100644 --- a/source/source_hamilt/module_xc/libxc_pot.cpp +++ b/source/source_hamilt/module_xc/libxc_pot.cpp @@ -15,25 +15,195 @@ #include #include -std::tuple XC_Functional_Libxc::v_xc_libxc( // Peize Lin update for nspin==4 at 2023.01.14 - const std::vector &func_id, - const int &nrxx, // number of real-space grid - const double &omega, // volume of cell - const double tpiba, - const Charge* const chr, - const int nspin_in, - const bool domag, - const bool domag_z, - const std::map* scaling_factor, - const double hybrid_alpha, - const double hse_omega) +void XC_Functional_Libxc::gradcorr_ncgga_sf_libxc(const std::vector& func_id, + const std::size_t nrxx, + const double tpiba, + const Charge* const chr, + const std::map* scaling_factor, + const double hybrid_alpha, + const double hse_omega, + std::vector& stress_gga) { - ModuleBase::TITLE("XC_Functional_Libxc","v_xc_libxc"); - ModuleBase::timer::start("XC_Functional_Libxc","v_xc_libxc"); + constexpr int nspin = 2; + stress_gga.assign(9, 0.0); - const int nspin = - (nspin_in == 1 || ( nspin_in ==4 && !domag && !domag_z)) - ? 1 : 2; + std::vector funcs = XC_Functional_Libxc::init_func(func_id, XC_POLARIZED, hybrid_alpha, hse_omega); + bool has_gga = false; + for (const xc_func_type& func: funcs) + { + has_gga = has_gga || func.info->family == XC_FAMILY_GGA || func.info->family == XC_FAMILY_HYB_GGA; + } + if (!has_gga) + { + XC_Functional_Libxc::finish_func(funcs); + return; + } + + // This is the same forward graph used by v_xc_libxc: the local spin map, + // its projected FFT gradients, and the sigma invariants are constructed + // once and shared by all Libxc components. + const XC_Functional_Libxc::NclSfDiscreteData sf_data + = XC_Functional_Libxc::make_ncl_sf_discrete_data(nrxx, tpiba, chr, true); + const std::vector& rho = sf_data.rho; + const std::vector sigma = XC_Functional_Libxc::convert_sigma(sf_data.spin_gradient); + std::vector aggregate_dsigma(3 * nrxx, 0.0); + + for (xc_func_type& func: funcs) + { + if (func.info->family != XC_FAMILY_GGA && func.info->family != XC_FAMILY_HYB_GGA) + { + continue; + } + + constexpr double rho_threshold = 1.0e-6; + constexpr double grho_threshold = 1.0e-10; + xc_func_set_dens_threshold(&func, rho_threshold); + const std::vector sgn + = XC_Functional_Libxc::cal_sgn(rho_threshold, grho_threshold, func, nspin, nrxx, rho, sigma); + std::vector exc(nrxx); + std::vector vrho(nspin * nrxx); + std::vector vsigma(3 * nrxx); + constexpr int nr_batch_size = 1024; +#ifdef _OPENMP +#pragma omp parallel for schedule(static, nr_batch_size) +#endif + for (int ir_start = 0; ir_start < static_cast(nrxx); ir_start += nr_batch_size) + { + const int ir_end = std::min(ir_start + nr_batch_size, static_cast(nrxx)); + const int nrxx_thread = ir_end - ir_start; + xc_gga_exc_vxc(&func, + nrxx_thread, + rho.data() + ir_start * nspin, + sigma.data() + ir_start * 3, + exc.data() + ir_start, + vrho.data() + ir_start * nspin, + vsigma.data() + ir_start * 3); + } + + double factor = 1.0; + if (scaling_factor != nullptr) + { + const std::map::const_iterator entry = scaling_factor->find(func.info->number); + if (entry != scaling_factor->end()) + { + factor = entry->second; + } + } + const XC_Functional_Libxc::LibxcWeightedDerivatives weighted + = XC_Functional_Libxc::make_libxc_weighted_derivatives(func, + nspin, + nrxx, + sgn, + rho, + sigma, + exc, + vrho, + vsigma); + for (std::size_t index = 0; index < aggregate_dsigma.size(); ++index) + { + aggregate_dsigma[index] += factor * weighted.dsigma[index]; + } + } + +// For g_s=sum_A J_sA G_h x_A, a reciprocal deformation changes G_h but +// not the pointwise map J. Therefore the metric derivative is exactly +// sum_s h_s,l g_s,m, with h_s=dE/dg_s built from the sanitizer-reversed, +// component-scaled aggregate above. +#ifdef _OPENMP +#pragma omp parallel + { + std::vector local_stress(9, 0.0); +#pragma omp for schedule(static, 512) + for (std::size_t ir = 0; ir < nrxx; ++ir) + { + const std::size_t sigma_index = 3 * ir; + const ModuleBase::Vector3& grad_up = sf_data.spin_gradient[0][ir]; + const ModuleBase::Vector3& grad_down = sf_data.spin_gradient[1][ir]; + const ModuleBase::Vector3 h_up + = ModuleBase::e2 + * (2.0 * aggregate_dsigma[sigma_index] * grad_up + aggregate_dsigma[sigma_index + 1] * grad_down); + const ModuleBase::Vector3 h_down + = ModuleBase::e2 + * (2.0 * aggregate_dsigma[sigma_index + 2] * grad_down + aggregate_dsigma[sigma_index + 1] * grad_up); + const double grad_up_component[3] = {grad_up.x, grad_up.y, grad_up.z}; + const double grad_down_component[3] = {grad_down.x, grad_down.y, grad_down.z}; + const double h_up_component[3] = {h_up.x, h_up.y, h_up.z}; + const double h_down_component[3] = {h_down.x, h_down.y, h_down.z}; + for (int l = 0; l < 3; ++l) + { + for (int m = 0; m <= l; ++m) + { + local_stress[l * 3 + m] + += h_up_component[l] * grad_up_component[m] + h_down_component[l] * grad_down_component[m]; + } + } + } +#pragma omp critical(libxc_ncgga_stress_reduce) + { + for (int l = 0; l < 3; ++l) + { + for (int m = 0; m <= l; ++m) + { + stress_gga[l * 3 + m] += local_stress[l * 3 + m]; + } + } + } + } +#else + for (std::size_t ir = 0; ir < nrxx; ++ir) + { + const std::size_t sigma_index = 3 * ir; + const ModuleBase::Vector3& grad_up = sf_data.spin_gradient[0][ir]; + const ModuleBase::Vector3& grad_down = sf_data.spin_gradient[1][ir]; + const ModuleBase::Vector3 h_up + = ModuleBase::e2 + * (2.0 * aggregate_dsigma[sigma_index] * grad_up + aggregate_dsigma[sigma_index + 1] * grad_down); + const ModuleBase::Vector3 h_down + = ModuleBase::e2 + * (2.0 * aggregate_dsigma[sigma_index + 2] * grad_down + aggregate_dsigma[sigma_index + 1] * grad_up); + const double grad_up_component[3] = {grad_up.x, grad_up.y, grad_up.z}; + const double grad_down_component[3] = {grad_down.x, grad_down.y, grad_down.z}; + const double h_up_component[3] = {h_up.x, h_up.y, h_up.z}; + const double h_down_component[3] = {h_down.x, h_down.y, h_down.z}; + for (int l = 0; l < 3; ++l) + { + for (int m = 0; m <= l; ++m) + { + stress_gga[l * 3 + m] + += h_up_component[l] * grad_up_component[m] + h_down_component[l] * grad_down_component[m]; + } + } + } +#endif + + XC_Functional_Libxc::finish_func(funcs); +} + +std::tuple XC_Functional_Libxc::v_xc_libxc( // Peize Lin update for nspin==4 at + // 2023.01.14 + const std::vector& func_id, + const int& nrxx, // number of real-space grid + const double& omega, // volume of cell + const double tpiba, + const Charge* const chr, + const int nspin_in, + const bool domag, + const bool domag_z, + const int gga_grad, + const std::map* scaling_factor, + const double hybrid_alpha, + const double hse_omega) +{ + ModuleBase::TITLE("XC_Functional_Libxc", "v_xc_libxc"); + ModuleBase::timer::start("XC_Functional_Libxc", "v_xc_libxc"); + + const int nspin = (nspin_in == 1 || (nspin_in == 4 && !domag && !domag_z)) ? 1 : 2; + + // For nspin=4 with noncollinear magnetism, gga_grad=2 selects the + // regularized projected local-collinear graph; gga_grad=0/1 keeps the + // original collinear algorithm. + const bool has_mag = domag || domag_z; + const bool use_lca = (nspin_in == 4) && has_mag && gga_grad == 2; //---------------------------------------------------------- // xc_func_type is defined in Libxc package @@ -43,151 +213,238 @@ std::tuple XC_Functional_Libxc::v_xc_libxc( / //---------------------------------------------------------- std::vector funcs = XC_Functional_Libxc::init_func( - /* func_id = */ func_id, - /* xc_polarized = */ (1==nspin) ? XC_UNPOLARIZED : XC_POLARIZED, + /* func_id = */ func_id, + /* xc_polarized = */ (1 == nspin) ? XC_UNPOLARIZED : XC_POLARIZED, /* hybrid_alpha = */ hybrid_alpha, /* hse_omega = */ hse_omega); - const bool is_gga = [&funcs]() - { - for( xc_func_type &func : funcs ) + const bool is_gga = [&funcs]() { + for (xc_func_type& func: funcs) { - switch( func.info->family ) + switch (func.info->family) { - case XC_FAMILY_GGA: - case XC_FAMILY_HYB_GGA: - return true; + case XC_FAMILY_GGA: + case XC_FAMILY_HYB_GGA: + return true; } } return false; }(); // converting rho + // For nspin=4, the charge density has 4 components: + // rho[0] = total charge, rho[1..3] = magnetization (mx, my, mz) + // libxc works with spin-up/spin-down densities: + // rho_up = 0.5*(rho[0] + |m|), rho_dn = 0.5*(rho[0] - |m|) std::vector rho; std::vector amag; - if(1==nspin || 2==nspin_in) + XC_Functional_Libxc::NclSfDiscreteData sf_data; + if (1 == nspin || 2 == nspin_in) { rho = XC_Functional_Libxc::convert_rho(nspin, nrxx, chr); } + else if (use_lca) + { + // gga_grad=2 uses one complete local map for both the Libxc density + // input and the projected FFT-gradient graph. LDA-only functionals + // need the same local map but do not pay for gradients. + sf_data = XC_Functional_Libxc::make_ncl_sf_discrete_data(nrxx, tpiba, chr, is_gga); + rho = sf_data.rho; + } else { - std::tuple,std::vector> rho_amag = XC_Functional_Libxc::convert_rho_amag_nspin4(nspin, nrxx, chr); + std::tuple, std::vector> rho_amag + = XC_Functional_Libxc::convert_rho_amag_nspin4(nspin, nrxx, chr); rho = std::get<0>(std::move(rho_amag)); amag = std::get<1>(std::move(rho_amag)); } std::vector>> gdr; std::vector sigma; - if(is_gga) + if (is_gga) { - gdr = XC_Functional_Libxc::cal_gdr(nspin, nrxx, rho, tpiba, chr); + if (use_lca) + { + gdr = sf_data.spin_gradient; + } + else + gdr = XC_Functional_Libxc::cal_gdr(nspin, nrxx, rho, tpiba, chr); + sigma = XC_Functional_Libxc::convert_sigma(gdr); } double etxc = 0.0; double vtxc = 0.0; - ModuleBase::matrix v(nspin,nrxx); - - for( xc_func_type &func : funcs ) + ModuleBase::matrix v(use_lca ? 4 : nspin, nrxx); + XC_Functional_Libxc::LibxcWeightedDerivatives sf_weighted; + if (use_lca) { - // jiyy add for threshold - constexpr double rho_threshold = 1E-6; - constexpr double grho_threshold = 1E-10; - - xc_func_set_dens_threshold(&func, rho_threshold); - - // sgn for threshold mask - const std::vector sgn = XC_Functional_Libxc::cal_sgn(rho_threshold, grho_threshold, func, nspin, nrxx, rho, sigma); - - std::vector exc ( nrxx ); - std::vector vrho ( nrxx * nspin ); - std::vector vsigma( nrxx * ((1==nspin)?1:3) ); + sf_weighted.energy_sum = 0.0; + sf_weighted.drho.assign(nrxx * nspin, 0.0); + if (is_gga) + { + sf_weighted.dsigma.assign(nrxx * 3, 0.0); + } + } - ModuleBase::timer::start("Libxc","xc_lda/gga_exc_vxc"); - switch( func.info->family ) + for (xc_func_type& func: funcs) + { + // thresholds: same convention as Quantum ESPRESSO's libxc interface + // (XClib/xc_wrapper_gga.f90): exc and vrho are evaluated down to + // rho_threshold_lda, while only the vsigma (gradient) term is + // suppressed below rho_threshold_gga / grho_threshold_gga + constexpr double rho_threshold_lda = 1E-10; + constexpr double rho_threshold_gga = 1E-6; + constexpr double grho_threshold_gga = 1E-10; + + // Keep the regularized mode's weighted-energy contract. The legacy + // path uses upstream's separate density and gradient cutoffs. + xc_func_set_dens_threshold(&func, use_lca ? rho_threshold_gga : rho_threshold_lda); + + // sgn for threshold masks + const std::pair,std::vector> sgn = XC_Functional_Libxc::cal_sgn_vxc( + rho_threshold_lda, rho_threshold_gga, grho_threshold_gga, func, nspin, nrxx, rho, sigma); + const std::vector energy_mask = use_lca + ? XC_Functional_Libxc::cal_sgn(rho_threshold_gga, grho_threshold_gga, func, nspin, nrxx, rho, sigma) + : sgn.first; + + std::vector exc(nrxx); + std::vector vrho(nrxx * nspin); + std::vector vsigma(nrxx * ((1 == nspin) ? 1 : 3)); + + ModuleBase::timer::start("Libxc", "xc_lda/gga_exc_vxc"); + switch (func.info->family) { - case XC_FAMILY_LDA: - { - constexpr int nr_batch_size = 1024; - #ifdef _OPENMP - #pragma omp parallel for schedule(static, nr_batch_size) - #endif - for( int ir_start = 0; ir_start < nrxx; ir_start += nr_batch_size ) - { - const int ir_end = std::min(ir_start + nr_batch_size, nrxx); - const int nrxx_thread = ir_end - ir_start; - xc_lda_exc_vxc( - &func, - nrxx_thread, - rho.data() + ir_start * nspin, - exc.data() + ir_start, - vrho.data() + ir_start * nspin ); - } - break; - } - case XC_FAMILY_GGA: - case XC_FAMILY_HYB_GGA: + case XC_FAMILY_LDA: { + constexpr int nr_batch_size = 1024; +#ifdef _OPENMP +#pragma omp parallel for schedule(static, nr_batch_size) +#endif + for (int ir_start = 0; ir_start < nrxx; ir_start += nr_batch_size) { - constexpr int nr_batch_size = 1024; - #ifdef _OPENMP - #pragma omp parallel for schedule(static, nr_batch_size) - #endif - for( int ir_start = 0; ir_start < nrxx; ir_start += nr_batch_size ) - { - const int ir_end = std::min(ir_start + nr_batch_size, nrxx); - const int nrxx_thread = ir_end - ir_start; - xc_gga_exc_vxc( - &func, - nrxx_thread, - rho.data() + ir_start * nspin, - sigma.data() + ir_start * ((1==nspin)?1:3), - exc.data() + ir_start, - vrho.data() + ir_start * nspin, - vsigma.data() + ir_start * ((1==nspin)?1:3) ); - } - break; + const int ir_end = std::min(ir_start + nr_batch_size, nrxx); + const int nrxx_thread = ir_end - ir_start; + xc_lda_exc_vxc(&func, + nrxx_thread, + rho.data() + ir_start * nspin, + exc.data() + ir_start, + vrho.data() + ir_start * nspin); } - default: + break; + } + case XC_FAMILY_GGA: + case XC_FAMILY_HYB_GGA: { + constexpr int nr_batch_size = 1024; +#ifdef _OPENMP +#pragma omp parallel for schedule(static, nr_batch_size) +#endif + for (int ir_start = 0; ir_start < nrxx; ir_start += nr_batch_size) { - throw std::domain_error("func.info->family ="+std::to_string(func.info->family) - +" unfinished in "+std::string(__FILE__)+" line "+std::to_string(__LINE__)); - + const int ir_end = std::min(ir_start + nr_batch_size, nrxx); + const int nrxx_thread = ir_end - ir_start; + xc_gga_exc_vxc(&func, + nrxx_thread, + rho.data() + ir_start * nspin, + sigma.data() + ir_start * ((1 == nspin) ? 1 : 3), + exc.data() + ir_start, + vrho.data() + ir_start * nspin, + vsigma.data() + ir_start * ((1 == nspin) ? 1 : 3)); } + break; + } + default: { + throw std::domain_error("func.info->family =" + std::to_string(func.info->family) + " unfinished in " + + std::string(__FILE__) + " line " + std::to_string(__LINE__)); } - ModuleBase::timer::end("Libxc","xc_lda/gga_exc_vxc"); + } + ModuleBase::timer::end("Libxc", "xc_lda/gga_exc_vxc"); // added by jghan, 2024-10-10 double factor = 1.0; - if( scaling_factor ) + if (scaling_factor) { auto pair_factor = scaling_factor->find(func.info->number); - if( pair_factor != scaling_factor->end() ) - { factor = pair_factor->second; } + if (pair_factor != scaling_factor->end()) + { + factor = pair_factor->second; + } } - // time factor is added by jghan, 2024-10-10 - etxc += XC_Functional_Libxc::convert_etxc(nspin, nrxx, sgn, rho, exc) * factor; - const std::pair vtxc_v = XC_Functional_Libxc::convert_vtxc_v( - func, nspin, nrxx, - sgn, rho, gdr, - vrho, vsigma, - tpiba, chr); - vtxc += std::get<0>(vtxc_v) * factor; - v += std::get<1>(vtxc_v) * factor; + // Keep the established energy accumulation and reduction order. In + // gga_grad=2, reverse every sanitizer now, apply the component scaling, + // and aggregate before traversing the shared projected graph once. + etxc += XC_Functional_Libxc::convert_etxc(nspin, nrxx, energy_mask, rho, exc) * factor; + if (use_lca) + { + const XC_Functional_Libxc::LibxcWeightedDerivatives weighted + = XC_Functional_Libxc::make_libxc_weighted_derivatives(func, + nspin, + nrxx, + energy_mask, + rho, + sigma, + exc, + vrho, + vsigma); + for (std::size_t index = 0; index < sf_weighted.drho.size(); ++index) + { + sf_weighted.drho[index] += factor * weighted.drho[index]; + } + for (std::size_t index = 0; index < weighted.dsigma.size(); ++index) + { + sf_weighted.dsigma[index] += factor * weighted.dsigma[index]; + } + } + else + { + const std::pair vtxc_v + = XC_Functional_Libxc::convert_vtxc_v(func, nspin, nrxx, sgn.first, sgn.second, rho, gdr, vrho, vsigma, tpiba, chr); + vtxc += std::get<0>(vtxc_v) * factor; + v += std::get<1>(vtxc_v) * factor; + } } // end for( xc_func_type &func : funcs ) - if(4==nspin_in) + if (use_lca) { - v = XC_Functional_Libxc::convert_v_nspin4(nrxx, chr, amag, v, domag, domag_z); + v = XC_Functional_Libxc::reverse_ncl_sf_discrete(nrxx, + sf_data, + sf_weighted.drho, + sf_weighted.dsigma, + tpiba, + chr); } - //------------------------------------------------- - // for MPI, reduce the exchange-correlation energy - //------------------------------------------------- - #ifdef __MPI + if (4 == nspin_in && !use_lca) + { + v = XC_Functional_Libxc::convert_v_nspin4(nrxx, chr, amag, v, has_mag); + } + + if (use_lca) + { + // Define vtxc from the potential that this routine actually returns. + // The nonlinear core density belongs to the XC energy graph, but the + // electronic variational density here is the four-channel valence + // density stored in chr->rho. + vtxc = 0.0; +#ifdef _OPENMP +#pragma omp parallel for collapse(2) reduction(+ : vtxc) schedule(static, 256) +#endif + for (int channel = 0; channel < 4; ++channel) + { + for (int ir = 0; ir < nrxx; ++ir) + { + vtxc += v(channel, ir) * chr->rho[channel][ir]; + } + } + } + +//------------------------------------------------- +// for MPI, reduce the exchange-correlation energy +//------------------------------------------------- +#ifdef __MPI Parallel_Reduce::reduce_pool(etxc); Parallel_Reduce::reduce_pool(vtxc); - #endif +#endif etxc *= omega / chr->rhopw->nxyz; vtxc *= omega / chr->rhopw->nxyz; @@ -198,7 +455,6 @@ std::tuple XC_Functional_Libxc::v_xc_libxc( / return std::make_tuple( etxc, vtxc, std::move(v) ); } - //the interface to libxc xc_mgga_exc_vxc(xc_func,n,rho,grho,laplrho,tau,e,v1,v2,v3,v4) //xc_func : LIBXC data type, contains information on xc functional //n: size of array, nspin*nnr @@ -238,7 +494,7 @@ std::tuple XC_Functional_Li // https://www.tddft.org/programs/libxc/manual/libxc-5.1.x/ //---------------------------------------------------------- std::vector funcs = XC_Functional_Libxc::init_func( - /* func_id = */ func_id, + /* func_id = */ func_id, /* xc_polarized = */ (1==nspin) ? XC_UNPOLARIZED:XC_POLARIZED, /* hybrid_alpha = */ hybrid_alpha, /* hse_omega = */ hse_omega); @@ -435,7 +691,13 @@ std::tuple XC_Functional_Li { for( int ir=0; ir< nrxx; ++ir ) { - rvtxc += dh[is][ir] * rho[ir*nspin+is]; + // Use valence-only density (chr->rho), not the total density + // (rho, which includes the NLCC core charge). vtxc must be + // \int v_xc * rho_valence so that the stress decomposition + // -(etxc - vtxc)/omega + stress_cc is consistent; including + // the core charge here double-counts the core contribution + // in the diagonal analytical stress. + rvtxc += dh[is][ir] * chr->rho[is][ir]; v(is,ir) -= dh[is][ir]; } } diff --git a/source/source_hamilt/module_xc/libxc_tools.cpp b/source/source_hamilt/module_xc/libxc_tools.cpp index 5ed04eff8db..579fd1409e0 100644 --- a/source/source_hamilt/module_xc/libxc_tools.cpp +++ b/source/source_hamilt/module_xc/libxc_tools.cpp @@ -4,6 +4,10 @@ #include "xc_functional.h" #include "source_estate/module_charge/charge.h" +#include +#include +#include + // converting rho (abacus=>libxc) std::vector XC_Functional_Libxc::convert_rho( const int nspin, @@ -31,10 +35,7 @@ XC_Functional_Libxc::convert_rho_amag_nspin4( const std::size_t nrxx, const Charge* const chr) { - // `nspin` is the number of LibXC spin channels, not the physical nspin of the - // system. This nspin4 branch is only entered for a magnetized nspin==4 system, - // which LibXC always evaluates in the polarized (2-channel) representation. - assert(nspin==2); + assert(nspin==2); // nspin here is the collapsed spin dimension for libxc (up/down) std::vector rho(nrxx*nspin); std::vector amag(nrxx); #ifdef _OPENMP @@ -53,6 +54,192 @@ XC_Functional_Libxc::convert_rho_amag_nspin4( return std::make_tuple(std::move(rho), std::move(amag)); } +XC_Functional_Libxc::NclSfDiscreteData +XC_Functional_Libxc::make_ncl_sf_discrete_data( + const std::size_t nrxx, + const double tpiba, + const Charge* const chr, + const bool need_gradient) +{ + constexpr int nspin = 2; + NclSfDiscreteData data; + data.spin_map.resize(nrxx); + data.rho.resize(nrxx * nspin); + + #ifdef _OPENMP + #pragma omp parallel for schedule(static, 1024) + #endif + for (std::size_t ir = 0; ir < nrxx; ++ir) + { + const std::array magnetization + = {{chr->rho[1][ir], chr->rho[2][ir], chr->rho[3][ir]}}; + const ModuleXC::NcggaRadialPoint radial + = ModuleXC::make_ncgga_radial_point( + magnetization, ModuleXC::ncgga_lca_radial_eta()); + data.spin_map[ir] = ModuleXC::make_ncgga_spin_map_point( + chr->rho[0][ir] + chr->rho_core[ir], radial); + data.rho[ir * nspin] = data.spin_map[ir].spin_density[0]; + data.rho[ir * nspin + 1] = data.spin_map[ir].spin_density[1]; + } + + if (!need_gradient) + { + return data; + } + + std::vector> grad_total(nrxx); + std::vector real_field(nrxx); + std::vector> reciprocal(chr->rhopw->npw); + #ifdef _OPENMP + #pragma omp parallel for schedule(static, 1024) + #endif + for (std::size_t ir = 0; ir < nrxx; ++ir) + { + real_field[ir] = chr->rho[0][ir] + chr->rho_core[ir]; + } + chr->rhopw->real2recip(real_field.data(), reciprocal.data()); + XC_Functional::grad_rho( + reciprocal.data(), grad_total.data(), chr->rhopw, tpiba); + + for (int mu = 0; mu < 3; ++mu) + { + data.grad_m[mu].resize(nrxx); + chr->rhopw->real2recip(chr->rho[mu + 1], reciprocal.data()); + XC_Functional::grad_rho( + reciprocal.data(), data.grad_m[mu].data(), chr->rhopw, tpiba); + } + + data.spin_gradient.resize(nspin); + for (int spin = 0; spin < nspin; ++spin) + { + data.spin_gradient[spin].resize(nrxx); + } + #ifdef _OPENMP + #pragma omp parallel for schedule(static, 512) + #endif + for (std::size_t ir = 0; ir < nrxx; ++ir) + { + for (int spin = 0; spin < nspin; ++spin) + { + ModuleBase::Vector3 gradient + = data.spin_map[ir].jacobian(spin, 0) * grad_total[ir]; + for (int mu = 0; mu < 3; ++mu) + { + gradient += data.spin_map[ir].jacobian(spin, mu + 1) + * data.grad_m[mu][ir]; + } + data.spin_gradient[spin][ir] = gradient; + } + } + return data; +} + +ModuleBase::matrix XC_Functional_Libxc::reverse_ncl_sf_discrete( + const std::size_t nrxx, + const NclSfDiscreteData& data, + const std::vector& drho, + const std::vector& dsigma, + const double tpiba, + const Charge* const chr) +{ + constexpr int nspin = 2; + constexpr int nchannel = 4; + assert(data.spin_map.size() == nrxx); + assert(data.rho.size() == nrxx * nspin); + assert(drho.size() == nrxx * nspin); + + ModuleBase::matrix potential(nchannel, nrxx); + #ifdef _OPENMP + #pragma omp parallel for schedule(static, 512) + #endif + for (std::size_t ir = 0; ir < nrxx; ++ir) + { + for (int channel = 0; channel < nchannel; ++channel) + { + potential(channel, ir) + = ModuleBase::e2 + * (data.spin_map[ir].jacobian(0, channel) + * drho[ir * nspin] + + data.spin_map[ir].jacobian(1, channel) + * drho[ir * nspin + 1]); + } + } + + if (dsigma.empty()) + { + return potential; + } + + assert(dsigma.size() == nrxx * 3); + assert(data.spin_gradient.size() == nspin); + for (int spin = 0; spin < nspin; ++spin) + { + assert(data.spin_gradient[spin].size() == nrxx); + } + for (int mu = 0; mu < 3; ++mu) + { + assert(data.grad_m[mu].size() == nrxx); + } + + std::vector> h_up(nrxx), h_down(nrxx); + #ifdef _OPENMP + #pragma omp parallel for schedule(static, 512) + #endif + for (std::size_t ir = 0; ir < nrxx; ++ir) + { + const std::size_t sigma_index = 3 * ir; + h_up[ir] + = ModuleBase::e2 + * (2.0 * dsigma[sigma_index] * data.spin_gradient[0][ir] + + dsigma[sigma_index + 1] * data.spin_gradient[1][ir]); + h_down[ir] + = ModuleBase::e2 + * (2.0 * dsigma[sigma_index + 2] * data.spin_gradient[1][ir] + + dsigma[sigma_index + 1] * data.spin_gradient[0][ir]); + } + + std::vector> flux(nrxx); + std::vector divergence(nrxx); + for (int channel = 0; channel < nchannel; ++channel) + { + #ifdef _OPENMP + #pragma omp parallel for schedule(static, 512) + #endif + for (std::size_t ir = 0; ir < nrxx; ++ir) + { + flux[ir] + = data.spin_map[ir].jacobian(0, channel) * h_up[ir] + + data.spin_map[ir].jacobian(1, channel) * h_down[ir]; + } + XC_Functional::grad_dot( + flux.data(), divergence.data(), chr->rhopw, tpiba); + + #ifdef _OPENMP + #pragma omp parallel for schedule(static, 512) + #endif + for (std::size_t ir = 0; ir < nrxx; ++ir) + { + potential(channel, ir) -= divergence[ir]; + if (channel == 0 || data.spin_map[ir].saturated) + { + continue; + } + + const ModuleBase::Vector3 spin_flux + = 0.5 * (h_up[ir] - h_down[ir]); + double local_response = 0.0; + for (int nu = 0; nu < 3; ++nu) + { + local_response + += data.spin_map[ir].radial.jacobian(nu, channel - 1) + * (spin_flux * data.grad_m[nu][ir]); + } + potential(channel, ir) += local_response; + } + } + return potential; +} + // calculating grho std::vector>> XC_Functional_Libxc::cal_gdr( @@ -191,6 +378,68 @@ std::vector XC_Functional_Libxc::cal_sgn( return sgn; } +// threshold masks for the xc potential, following the convention of Quantum +// ESPRESSO's libxc interface (XClib/xc_wrapper_gga.f90): the first mask +// applies to exc and vrho (kept down to rho_threshold_vrho), the second one +// applies only to the vsigma (gradient) term +std::pair, std::vector> XC_Functional_Libxc::cal_sgn_vxc( + const double rho_threshold_vrho, + const double rho_threshold_vsigma, + const double grho_threshold_vsigma, + const xc_func_type &func, + const int nspin, + const std::size_t nrxx, + const std::vector &rho, + const std::vector &sigma) +{ + std::vector sgn_vrho(nrxx*nspin, 1.0); + std::vector sgn_vsigma(nrxx*nspin, 1.0); + const bool is_gga = (func.info->family == XC_FAMILY_GGA + || func.info->family == XC_FAMILY_HYB_GGA); + if(nspin==1) + { + #ifdef _OPENMP + #pragma omp parallel for schedule(static, 1024) + #endif + for( std::size_t ir=0; irabacus) double XC_Functional_Libxc::convert_etxc( const int nspin, @@ -213,12 +462,136 @@ double XC_Functional_Libxc::convert_etxc( return etxc; } +XC_Functional_Libxc::LibxcWeightedDerivatives +XC_Functional_Libxc::make_libxc_weighted_derivatives( + const xc_func_type &func, + const int nspin, + const std::size_t nrxx, + const std::vector &sgn, + const std::vector &rho, + const std::vector &sigma, + const std::vector &exc, + const std::vector &vrho, + const std::vector &vsigma) +{ + assert(nspin == 1 || nspin == 2); + assert(sgn.size() == nrxx * nspin); + assert(rho.size() == nrxx * nspin); + assert(exc.size() == nrxx); + assert(vrho.size() == nrxx * nspin); + assert(func.nspin == nspin); + + const bool is_gga + = func.info->family == XC_FAMILY_GGA || func.info->family == XC_FAMILY_HYB_GGA; + const std::size_t nsigma = nspin == 1 ? 1 : 3; + if (is_gga) + { + assert(sigma.size() == nrxx * nsigma); + assert(vsigma.size() == nrxx * nsigma); + } + + LibxcWeightedDerivatives weighted; + weighted.energy_sum = 0.0; + weighted.drho.assign(nrxx * nspin, 0.0); + if (is_gga) + { + weighted.dsigma.assign(nrxx * nsigma, 0.0); + } + + const double density_floor = func.dens_threshold; + const double sigma_floor = func.sigma_threshold * func.sigma_threshold; + double energy_sum = 0.0; + #ifdef _OPENMP + #pragma omp parallel for reduction(+:energy_sum) schedule(static, 512) + #endif + for (std::size_t ir = 0; ir < nrxx; ++ir) + { + double raw_density_sum = 0.0; + double sanitized_density_sum = 0.0; + double energy_weight = 0.0; + for (int is = 0; is < nspin; ++is) + { + const std::size_t index = ir * nspin + is; + raw_density_sum += rho[index]; + sanitized_density_sum += std::max(density_floor, rho[index]); + energy_weight += sgn[index] * rho[index]; + } + + // Libxc leaves all outputs zero below the total-density threshold. + // Inside that branch, the ABACUS weighted energy is locally constant. + if (raw_density_sum < density_floor) + { + continue; + } + + // ABACUS accumulates M*eps while Libxc differentiates Y*eps after + // y_s=max(T,rho_s). Hence d eps/d y_s=(vrho_s-eps)/Y. + energy_sum += energy_weight * exc[ir]; + const double libxc_weight = energy_weight / sanitized_density_sum; + for (int is = 0; is < nspin; ++is) + { + const std::size_t index = ir * nspin + is; + const double floor_jacobian = rho[index] > density_floor ? 1.0 : 0.0; + weighted.drho[index] + = sgn[index] * exc[ir] + + libxc_weight * floor_jacobian * (vrho[index] - exc[ir]); + } + + if (!is_gga) + { + continue; + } + + if (nspin == 1) + { + const double floor_jacobian = sigma[ir] > sigma_floor ? 1.0 : 0.0; + weighted.dsigma[ir] = libxc_weight * floor_jacobian * vsigma[ir]; + continue; + } + + const std::size_t sigma_index = 3 * ir; + const double sigma_uu = sigma[sigma_index]; + const double sigma_ud = sigma[sigma_index + 1]; + const double sigma_dd = sigma[sigma_index + 2]; + const double jacobian_uu = sigma_uu > sigma_floor ? 1.0 : 0.0; + const double jacobian_dd = sigma_dd > sigma_floor ? 1.0 : 0.0; + const double sanitized_uu = std::max(sigma_floor, sigma_uu); + const double sanitized_dd = std::max(sigma_floor, sigma_dd); + const double cross_limit = 0.5 * (sanitized_uu + sanitized_dd); + + double cross_to_diagonal = 0.0; + double cross_jacobian = 1.0; + if (sigma_ud < -cross_limit) + { + cross_to_diagonal = -0.5; + cross_jacobian = 0.0; + } + else if (sigma_ud > cross_limit) + { + cross_to_diagonal = 0.5; + cross_jacobian = 0.0; + } + + weighted.dsigma[sigma_index] + = libxc_weight * jacobian_uu + * (vsigma[sigma_index] + cross_to_diagonal * vsigma[sigma_index + 1]); + weighted.dsigma[sigma_index + 1] + = libxc_weight * cross_jacobian * vsigma[sigma_index + 1]; + weighted.dsigma[sigma_index + 2] + = libxc_weight * jacobian_dd + * (vsigma[sigma_index + 2] + cross_to_diagonal * vsigma[sigma_index + 1]); + } + weighted.energy_sum = energy_sum; + return weighted; +} + // converting vtxc and v from vrho and vsigma (libxc=>abacus) std::pair XC_Functional_Libxc::convert_vtxc_v( const xc_func_type &func, const int nspin, const std::size_t nrxx, - const std::vector &sgn, + const std::vector &sgn_vrho, + const std::vector &sgn_vsigma, const std::vector &rho, const std::vector>> &gdr, const std::vector &vrho, @@ -238,7 +611,7 @@ std::pair XC_Functional_Libxc::convert_vtxc_v( for( std::size_t ir=0; ir XC_Functional_Libxc::convert_vtxc_v( if(func.info->family == XC_FAMILY_GGA || func.info->family == XC_FAMILY_HYB_GGA) { - const std::vector> dh = XC_Functional_Libxc::cal_dh(nspin, nrxx, sgn, gdr, vsigma, tpiba, chr); + const std::vector> dh = XC_Functional_Libxc::cal_dh(nspin, nrxx, sgn_vsigma, gdr, vsigma, tpiba, chr); double rvtxc = 0.0; #ifdef _OPENMP @@ -267,7 +640,6 @@ std::pair XC_Functional_Libxc::convert_vtxc_v( return std::make_pair(vtxc, std::move(v)); } - // dh for gga v std::vector> XC_Functional_Libxc::cal_dh( const int nspin, @@ -317,32 +689,30 @@ std::vector> XC_Functional_Libxc::cal_dh( return dh; } - // convert v for NSPIN=4 ModuleBase::matrix XC_Functional_Libxc::convert_v_nspin4( const std::size_t nrxx, const Charge* const chr, const std::vector &amag, const ModuleBase::matrix &v, - const bool domag, - const bool domag_z) + const bool has_mag) { //assert(nrxx>0); - constexpr int nspin = 4; + constexpr int nspin4 = 4; constexpr double vanishing_charge = 1.0e-10; - ModuleBase::matrix v_nspin4(nspin, nrxx); + ModuleBase::matrix v_nspin4(nspin4, nrxx); for( int ir=0; ir vanishing_charge ) { const double vs = 0.5 * (v(0,ir)-v(1,ir)); - for(int ipol=1; ipolrho[ipol][ir] / amag[ir]; } diff --git a/source/source_hamilt/module_xc/test/CMakeLists.txt b/source/source_hamilt/module_xc/test/CMakeLists.txt index 51fe559c966..a07e4eb4d7e 100644 --- a/source/source_hamilt/module_xc/test/CMakeLists.txt +++ b/source/source_hamilt/module_xc/test/CMakeLists.txt @@ -56,14 +56,14 @@ AddTest( ../libxc_lda_wrap.cpp ../libxc_gga_wrap.cpp ../libxc_mgga_wrap.cpp - ../xc_gga_corr.cpp ../xc_lda_corr.cpp - ../xc_gga_exch.cpp ../xc_lda_exch.cpp ../xc_hcth.cpp + ../xc_gga_corr.cpp ../xc_lda_corr.cpp + ../xc_gga_exch.cpp ../xc_lda_exch.cpp ../xc_hcth.cpp ) AddTest( TARGET MODULE_HAMILT_XCTest_VXC LIBS parameter MPI::MPI_CXX Libxc::xc psi device container - SOURCES test_xc5.cpp ../xc_grad.cpp ../xc_grad_prepare.cpp ../xc_grad_kernel.cpp ../xc_grad_assemble.cpp ../xc_grad_wfc.cpp ../xc_grad_utils.cpp ../xc_functional.cpp + SOURCES test_xc5.cpp test_libxc_tools.cpp ../xc_grad.cpp ../xc_grad_prepare.cpp ../xc_grad_kernel.cpp ../xc_grad_assemble.cpp ../xc_grad_wfc.cpp ../xc_grad_utils.cpp ../xc_functional.cpp ../xc_lda_wrap.cpp ../xc_gga_wrap.cpp ../libxc_setup.cpp ../libxc_lda_wrap.cpp @@ -72,6 +72,8 @@ AddTest( ../xc_gga_corr.cpp ../xc_lda_corr.cpp ../xc_gga_exch.cpp ../xc_lda_exch.cpp ../xc_hcth.cpp ../xc_pot.cpp + ../xc_functional_ncgga_sf.cpp + ../xc_ncgga_radial.cpp ../libxc_pot.cpp ../libxc_tools.cpp ../../../source_base/module_external/blas_connector_base.cpp ../../../source_base/module_external/blas_connector_vector.cpp ../../../source_base/module_external/blas_connector_matrix.cpp @@ -86,6 +88,35 @@ AddTest( ${FFT_SRC} ) +AddTest( + TARGET MODULE_HAMILT_XCTest_NCGGA_RADIAL + SOURCES test_xc_ncgga_radial.cpp ../xc_ncgga_radial.cpp +) + +AddTest( + TARGET MODULE_HAMILT_XCTest_NCGGA_DISCRETE_FD + LIBS xc_ planewave parameter MPI::MPI_CXX Libxc::xc psi device container base + SOURCES test_xc_functional_ncgga_sf.cpp +) +set_tests_properties( + MODULE_HAMILT_XCTest_NCGGA_DISCRETE_FD + PROPERTIES ENVIRONMENT "OMP_NUM_THREADS=1" +) +add_test( + NAME MODULE_HAMILT_XCTest_NCGGA_DISCRETE_FD_MPI2 + COMMAND ${MPIEXEC_EXECUTABLE} ${MPIEXEC_NUMPROC_FLAG} 2 + ${MPIEXEC_PREFLAGS} + $ + ${MPIEXEC_POSTFLAGS} +) +set_tests_properties( + MODULE_HAMILT_XCTest_NCGGA_DISCRETE_FD_MPI2 + PROPERTIES + ENVIRONMENT "OMP_NUM_THREADS=1" + PROCESSORS 2 + WORKING_DIRECTORY $ +) + AddTest( TARGET MODULE_HAMILT_XCTest_SCANL_LAPL LIBS parameter MPI::MPI_CXX Libxc::xc @@ -126,3 +157,10 @@ AddTest( ../../../source_base/module_fft/fft_cpu.cpp ${FFT_SRC} ) + +# gradcorr dispatches the noncollinear stress path even in focused FFT tests. +foreach(xc_gradient_test MODULE_HAMILT_XCTest_GRADCORR MODULE_HAMILT_XCTest_LAPL) + target_sources(${xc_gradient_test} PRIVATE + ../xc_functional_ncgga_sf.cpp ../xc_ncgga_radial.cpp + ../libxc_pot.cpp ../libxc_tools.cpp ../../../source_base/timer.cpp) +endforeach() diff --git a/source/source_hamilt/module_xc/test/test_libxc_tools.cpp b/source/source_hamilt/module_xc/test/test_libxc_tools.cpp new file mode 100644 index 00000000000..a0fada52c62 --- /dev/null +++ b/source/source_hamilt/module_xc/test/test_libxc_tools.cpp @@ -0,0 +1,381 @@ +#include +#include "../libxc_abacus.h" + +#include "gtest/gtest.h" +#include +#include +#include +#include + +#ifdef __LIBXC + +TEST(LibxcSanitizer, FullPolarizationUsesTheWeightedEnergyDerivative) +{ + const std::array functional_ids = {{XC_LDA_X, XC_LDA_C_PZ}}; + const double density = 0.45; + + for (std::size_t ifunc = 0; ifunc < functional_ids.size(); ++ifunc) + { + xc_func_type func; + ASSERT_EQ(xc_func_init(&func, functional_ids[ifunc], XC_POLARIZED), 0); + xc_func_set_dens_threshold(&func, 1.0e-6); + + const std::function evaluate + = [&func](const double rho_up, + const double rho_down, + XC_Functional_Libxc::LibxcWeightedDerivatives* const weighted) { + const std::vector rho = {rho_up, rho_down}; + const std::vector mask = {1.0, 1.0}; + std::vector exc(1, 0.0); + std::vector vrho(2, 0.0); + xc_lda_exc_vxc(&func, 1, rho.data(), exc.data(), vrho.data()); + if (weighted != nullptr) + { + *weighted = XC_Functional_Libxc::make_libxc_weighted_derivatives(func, + 2, + 1, + mask, + rho, + std::vector(), + exc, + vrho, + std::vector()); + } + return (rho_up + rho_down) * exc[0]; + }; + + XC_Functional_Libxc::LibxcWeightedDerivatives weighted; + const double energy = evaluate(density, 0.0, &weighted); + ASSERT_EQ(weighted.drho.size(), 2U); + EXPECT_DOUBLE_EQ(weighted.energy_sum, energy); + + const double steps[] = {1.0e-3, 5.0e-4, 2.5e-4, 1.25e-4}; + std::array errors = {{0.0, 0.0, 0.0, 0.0}}; + for (std::size_t ieps = 0; ieps < 4; ++ieps) + { + const double step = steps[ieps]; + const double finite_difference + = (evaluate(density + step, 0.0, nullptr) - evaluate(density - step, 0.0, nullptr)) / (2.0 * step); + errors[ieps] = std::abs(weighted.drho[0] - finite_difference); + EXPECT_LE(errors[ieps], 2.0e-7 * std::max(1.0, std::abs(weighted.drho[0]))) + << "functional_id=" << functional_ids[ifunc] << ", step=" << step; + } + EXPECT_LE(errors[1], 0.4 * errors[0] + 1.0e-12); + EXPECT_LE(errors[2], 0.4 * errors[1] + 1.0e-12); + + const double inactive_density = 0.5 * func.dens_threshold; + const double inactive_step = 0.2 * func.dens_threshold; + XC_Functional_Libxc::LibxcWeightedDerivatives inactive_weighted; + evaluate(density, inactive_density, &inactive_weighted); + const double inactive_finite_difference = (evaluate(density, inactive_density + inactive_step, nullptr) + - evaluate(density, inactive_density - inactive_step, nullptr)) + / (2.0 * inactive_step); + EXPECT_NEAR(inactive_finite_difference, + inactive_weighted.drho[1], + 2.0e-8 * std::max(1.0, std::abs(inactive_finite_difference))); + xc_func_end(&func); + } +} + +TEST(LibxcSanitizer, GgaSigmaReverseMatchesTheWeightedEnergy) +{ + xc_func_type func; + ASSERT_EQ(xc_func_init(&func, XC_GGA_C_PBE, XC_POLARIZED), 0); + xc_func_set_dens_threshold(&func, 1.0e-6); + xc_func_set_sigma_threshold(&func, 1.0e-2); + + const std::vector mask = {1.0, 0.0}; + const std::vector density = {0.40, 0.20}; + const double sigma_floor = func.sigma_threshold * func.sigma_threshold; + const std::array, 5> sigma_states = {{{{0.040, 0.010, 0.030}}, + {{0.5 * sigma_floor, 0.0, 0.030}}, + {{0.040, 0.200, 0.030}}, + {{0.040, -0.200, 0.030}}, + {{0.5 * sigma_floor, 0.200, 0.030}}}}; + + const std::function&, const std::vector&, XC_Functional_Libxc::LibxcWeightedDerivatives*)> evaluate + = [&func, &mask](const std::vector& rho, + const std::vector& sigma, + XC_Functional_Libxc::LibxcWeightedDerivatives* const weighted) { + std::vector exc(1, 0.0); + std::vector vrho(2, 0.0); + std::vector vsigma(3, 0.0); + xc_gga_exc_vxc(&func, 1, rho.data(), sigma.data(), exc.data(), vrho.data(), vsigma.data()); + if (weighted != nullptr) + { + *weighted + = XC_Functional_Libxc::make_libxc_weighted_derivatives(func, 2, 1, mask, rho, sigma, exc, vrho, vsigma); + } + return (mask[0] * rho[0] + mask[1] * rho[1]) * exc[0]; + }; + + for (std::size_t icase = 0; icase < sigma_states.size(); ++icase) + { + std::vector sigma(sigma_states[icase].begin(), sigma_states[icase].end()); + XC_Functional_Libxc::LibxcWeightedDerivatives weighted; + const double energy = evaluate(density, sigma, &weighted); + EXPECT_DOUBLE_EQ(weighted.energy_sum, energy); + ASSERT_EQ(weighted.drho.size(), 2U); + ASSERT_EQ(weighted.dsigma.size(), 3U); + + if (icase == 0) + { + for (int component = 0; component < 2; ++component) + { + std::vector perturbed_density = density; + const double step = 1.0e-6; + perturbed_density[component] += step; + const double energy_plus = evaluate(perturbed_density, sigma, nullptr); + perturbed_density[component] -= 2.0 * step; + const double energy_minus = evaluate(perturbed_density, sigma, nullptr); + const double finite_difference = (energy_plus - energy_minus) / (2.0 * step); + EXPECT_NEAR(finite_difference, + weighted.drho[component], + 2.0e-7 * std::max(1.0, std::abs(weighted.drho[component]))); + } + } + + if (icase == 1 || icase == 4) + { + EXPECT_DOUBLE_EQ(weighted.dsigma[0], 0.0); + } + if (icase >= 2) + { + EXPECT_DOUBLE_EQ(weighted.dsigma[1], 0.0); + } + + for (int component = 0; component < 3; ++component) + { + const double step = 1.0e-6; + sigma[component] += step; + const double energy_plus = evaluate(density, sigma, nullptr); + sigma[component] -= 2.0 * step; + const double energy_minus = evaluate(density, sigma, nullptr); + sigma[component] += step; + const double finite_difference = (energy_plus - energy_minus) / (2.0 * step); + EXPECT_NEAR(finite_difference, + weighted.dsigma[component], + 2.0e-7 * std::max(1.0, std::abs(weighted.dsigma[component]))) + << "case=" << icase << ", sigma component=" << component; + } + } + xc_func_end(&func); +} + +TEST(LibxcSanitizer, UnpolarizedSelfSigmaReverseMatchesTheWeightedEnergy) +{ + xc_func_type func; + ASSERT_EQ(xc_func_init(&func, XC_GGA_C_PBE, XC_UNPOLARIZED), 0); + xc_func_set_dens_threshold(&func, 1.0e-6); + xc_func_set_sigma_threshold(&func, 1.0e-2); + + const std::vector mask = {1.0}; + const std::vector density = {0.40}; + const double sigma_floor = func.sigma_threshold * func.sigma_threshold; + + const std::function evaluate + = [&func, &mask, &density](const double sigma_value, + XC_Functional_Libxc::LibxcWeightedDerivatives* const weighted) { + const std::vector sigma = {sigma_value}; + std::vector exc(1, 0.0); + std::vector vrho(1, 0.0); + std::vector vsigma(1, 0.0); + xc_gga_exc_vxc(&func, 1, density.data(), sigma.data(), exc.data(), vrho.data(), vsigma.data()); + if (weighted != nullptr) + { + *weighted = XC_Functional_Libxc::make_libxc_weighted_derivatives(func, + 1, + 1, + mask, + density, + sigma, + exc, + vrho, + vsigma); + } + return mask[0] * density[0] * exc[0]; + }; + + const double below_floor_sigma = 0.5 * sigma_floor; + const double below_floor_step = 0.2 * sigma_floor; + XC_Functional_Libxc::LibxcWeightedDerivatives below_floor_weighted; + const double below_floor_energy = evaluate(below_floor_sigma, &below_floor_weighted); + ASSERT_EQ(below_floor_weighted.dsigma.size(), 1U); + EXPECT_DOUBLE_EQ(below_floor_weighted.energy_sum, below_floor_energy); + EXPECT_DOUBLE_EQ(below_floor_weighted.dsigma[0], 0.0); + const double below_floor_finite_difference = (evaluate(below_floor_sigma + below_floor_step, nullptr) + - evaluate(below_floor_sigma - below_floor_step, nullptr)) + / (2.0 * below_floor_step); + EXPECT_NEAR(below_floor_finite_difference, below_floor_weighted.dsigma[0], 1.0e-12); + + const double above_floor_sigma = 0.040; + const double above_floor_step = 1.0e-6; + XC_Functional_Libxc::LibxcWeightedDerivatives above_floor_weighted; + const double above_floor_energy = evaluate(above_floor_sigma, &above_floor_weighted); + ASSERT_EQ(above_floor_weighted.dsigma.size(), 1U); + EXPECT_DOUBLE_EQ(above_floor_weighted.energy_sum, above_floor_energy); + EXPECT_GT(std::abs(above_floor_weighted.dsigma[0]), 1.0e-12); + const double above_floor_finite_difference = (evaluate(above_floor_sigma + above_floor_step, nullptr) + - evaluate(above_floor_sigma - above_floor_step, nullptr)) + / (2.0 * above_floor_step); + EXPECT_NEAR(above_floor_finite_difference, + above_floor_weighted.dsigma[0], + 2.0e-7 * std::max(1.0, std::abs(above_floor_weighted.dsigma[0]))); + + xc_func_end(&func); +} + +#endif + +#ifdef __LIBXC +/************************************************ + * unit test of XC_Functional_Libxc::cal_sgn_vxc + * (libxc_tools.cpp) + * + * cal_sgn_vxc returns two threshold masks following the + * convention of Quantum ESPRESSO's libxc interface + * (XClib/xc_wrapper_gga.f90): + * - the first mask applies to exc and vrho, zeroed only + * where the density falls below rho_threshold_vrho; + * - the second mask applies only to the vsigma (gradient) + * term, zeroed where the density falls below + * rho_threshold_vsigma or sqrt(|sigma|) falls below + * grho_threshold_vsigma. + * For non-GGA families the vsigma mask must stay 1. + ***********************************************/ + +namespace +{ +// same values as the call site in v_xc_libxc (libxc_pot.cpp) +constexpr double rho_threshold_vrho = 1E-10; +constexpr double rho_threshold_vsigma = 1E-6; +constexpr double grho_threshold_vsigma = 1E-10; +} // namespace + +class LibxcToolsSgnVxcTest : public testing::Test +{ + protected: + xc_func_type gga_func; + xc_func_type lda_func; + + void SetUp() override + { + xc_func_init(&gga_func, XC_GGA_X_PBE, XC_UNPOLARIZED); + xc_func_init(&lda_func, XC_LDA_X, XC_UNPOLARIZED); + } + + void TearDown() override + { + xc_func_end(&gga_func); + xc_func_end(&lda_func); + } +}; + +// nspin = 1, GGA functional: the vrho mask survives in the shell +// rho_threshold_vrho < rho <= rho_threshold_vsigma, while the vsigma +// mask is already zeroed there +TEST_F(LibxcToolsSgnVxcTest, Nspin1GgaDensityTiers) +{ + const std::size_t nrxx = 3; + const std::vector rho = {1.0, 1E-8, 1E-12}; + const std::vector sigma = {0.1, 0.1, 0.1}; + + const std::pair, std::vector> sgn = XC_Functional_Libxc::cal_sgn_vxc( + rho_threshold_vrho, rho_threshold_vsigma, grho_threshold_vsigma, gga_func, 1, nrxx, rho, sigma); + + EXPECT_DOUBLE_EQ(sgn.first[0], 1.0); + EXPECT_DOUBLE_EQ(sgn.second[0], 1.0); + // inside the shell: vrho kept, vsigma suppressed + EXPECT_DOUBLE_EQ(sgn.first[1], 1.0); + EXPECT_DOUBLE_EQ(sgn.second[1], 0.0); + // below rho_threshold_vrho: both suppressed + EXPECT_DOUBLE_EQ(sgn.first[2], 0.0); + EXPECT_DOUBLE_EQ(sgn.second[2], 0.0); +} + +// nspin = 1, GGA functional: a small gradient alone suppresses +// vsigma even where the density is large +TEST_F(LibxcToolsSgnVxcTest, Nspin1GgaGradientTrigger) +{ + const std::size_t nrxx = 2; + const std::vector rho = {1.0, 1.0}; + const std::vector sigma = {0.1, 1E-22}; // sqrt(1E-22) = 1E-11 < grho_threshold_vsigma + + const std::pair, std::vector> sgn = XC_Functional_Libxc::cal_sgn_vxc( + rho_threshold_vrho, rho_threshold_vsigma, grho_threshold_vsigma, gga_func, 1, nrxx, rho, sigma); + + EXPECT_DOUBLE_EQ(sgn.first[0], 1.0); + EXPECT_DOUBLE_EQ(sgn.second[0], 1.0); + EXPECT_DOUBLE_EQ(sgn.first[1], 1.0); + EXPECT_DOUBLE_EQ(sgn.second[1], 0.0); +} + +// nspin = 1, LDA functional: no vsigma term exists, so the vsigma +// mask must stay 1 regardless of density and sigma; the vrho mask +// still follows rho_threshold_vrho +TEST_F(LibxcToolsSgnVxcTest, Nspin1LdaKeepsVsigmaMask) +{ + const std::size_t nrxx = 2; + const std::vector rho = {1E-8, 1E-12}; + const std::vector sigma = {0.0, 0.0}; + + const std::pair, std::vector> sgn = XC_Functional_Libxc::cal_sgn_vxc( + rho_threshold_vrho, rho_threshold_vsigma, grho_threshold_vsigma, lda_func, 1, nrxx, rho, sigma); + + EXPECT_DOUBLE_EQ(sgn.first[0], 1.0); + EXPECT_DOUBLE_EQ(sgn.second[0], 1.0); + EXPECT_DOUBLE_EQ(sgn.first[1], 0.0); + EXPECT_DOUBLE_EQ(sgn.second[1], 1.0); +} + +// nspin = 2, GGA functional: both spin channels are masked jointly +// when either spin density fails a threshold +TEST_F(LibxcToolsSgnVxcTest, Nspin2JointMasking) +{ + const std::size_t nrxx = 3; + // interleaved spin densities: {up0, dw0, up1, dw1, up2, dw2} + const std::vector rho = {1E-8, 0.5, 1E-12, 0.5, 0.5, 0.5}; + // interleaved sigma: {uu0, ud0, dd0, uu1, ud1, dd1, ...} + const std::vector sigma = {0.1, 0.0, 0.1, 0.1, 0.0, 0.1, 0.1, 0.0, 1E-22}; + + const std::pair, std::vector> sgn = XC_Functional_Libxc::cal_sgn_vxc( + rho_threshold_vrho, rho_threshold_vsigma, grho_threshold_vsigma, gga_func, 2, nrxx, rho, sigma); + + // ir0: rho_up inside the shell -> vsigma suppressed in both channels, + // vrho kept in both channels + EXPECT_DOUBLE_EQ(sgn.first[0], 1.0); + EXPECT_DOUBLE_EQ(sgn.first[1], 1.0); + EXPECT_DOUBLE_EQ(sgn.second[0], 0.0); + EXPECT_DOUBLE_EQ(sgn.second[1], 0.0); + // ir1: rho_up below rho_threshold_vrho -> both masks suppressed + // in both channels + EXPECT_DOUBLE_EQ(sgn.first[2], 0.0); + EXPECT_DOUBLE_EQ(sgn.first[3], 0.0); + EXPECT_DOUBLE_EQ(sgn.second[2], 0.0); + EXPECT_DOUBLE_EQ(sgn.second[3], 0.0); + // ir2: densities fine but the down-down gradient fails -> + // only vsigma suppressed, in both channels + EXPECT_DOUBLE_EQ(sgn.first[4], 1.0); + EXPECT_DOUBLE_EQ(sgn.first[5], 1.0); + EXPECT_DOUBLE_EQ(sgn.second[4], 0.0); + EXPECT_DOUBLE_EQ(sgn.second[5], 0.0); +} + +// nspin = 2, GGA functional: the up-down cross component of sigma +// does not trigger the vsigma mask by itself +TEST_F(LibxcToolsSgnVxcTest, Nspin2CrossSigmaIgnored) +{ + const std::size_t nrxx = 1; + const std::vector rho = {0.5, 0.5}; + const std::vector sigma = {0.1, 1E-30, 0.1}; + + const std::pair, std::vector> sgn = XC_Functional_Libxc::cal_sgn_vxc( + rho_threshold_vrho, rho_threshold_vsigma, grho_threshold_vsigma, gga_func, 2, nrxx, rho, sigma); + + EXPECT_DOUBLE_EQ(sgn.first[0], 1.0); + EXPECT_DOUBLE_EQ(sgn.first[1], 1.0); + EXPECT_DOUBLE_EQ(sgn.second[0], 1.0); + EXPECT_DOUBLE_EQ(sgn.second[1], 1.0); +} + +#endif diff --git a/source/source_hamilt/module_xc/test/test_xc3.cpp b/source/source_hamilt/module_xc/test/test_xc3.cpp index d472d972cc9..cd7d0e52e34 100644 --- a/source/source_hamilt/module_xc/test/test_xc3.cpp +++ b/source/source_hamilt/module_xc/test/test_xc3.cpp @@ -90,15 +90,16 @@ class XCTest_GRADCORR : public XCTest XC_Functional::set_xc_type("PBE"); + const int gga_grad = 0; double hybrid_alpha = 0.0; double hse_omega = 0.0; - XC_Functional::gradcorr(et1,vt1,v1,&chr,&rhopw,&ucell,stress1,false,nspin1,domag,domag_z, hybrid_alpha, hse_omega); - XC_Functional::gradcorr(et1,vt1,v1,&chr,&rhopw,&ucell,stress1,true,nspin1,domag,domag_z, hybrid_alpha, hse_omega); + XC_Functional::gradcorr(et1,vt1,v1,&chr,&rhopw,&ucell,stress1,false,nspin1,domag,domag_z,gga_grad, hybrid_alpha, hse_omega); + XC_Functional::gradcorr(et1,vt1,v1,&chr,&rhopw,&ucell,stress1,true,nspin1,domag,domag_z,gga_grad, hybrid_alpha, hse_omega); - XC_Functional::gradcorr(et2,vt2,v2,&chr,&rhopw,&ucell,stress2,false,nspin2,domag,domag_z, hybrid_alpha, hse_omega); - XC_Functional::gradcorr(et2,vt2,v2,&chr,&rhopw,&ucell,stress2,true,nspin2,domag,domag_z, hybrid_alpha, hse_omega); + XC_Functional::gradcorr(et2,vt2,v2,&chr,&rhopw,&ucell,stress2,false,nspin2,domag,domag_z,gga_grad, hybrid_alpha, hse_omega); + XC_Functional::gradcorr(et2,vt2,v2,&chr,&rhopw,&ucell,stress2,true,nspin2,domag,domag_z,gga_grad, hybrid_alpha, hse_omega); - XC_Functional::gradcorr(et4,vt4,v4,&chr,&rhopw,&ucell,stress4,false,nspin4,domag_true,domag_z, hybrid_alpha, hse_omega); + XC_Functional::gradcorr(et4,vt4,v4,&chr,&rhopw,&ucell,stress4,false,nspin4,domag_true,domag_z,gga_grad, hybrid_alpha, hse_omega); } }; @@ -236,16 +237,17 @@ class XCTest_GRADCORR_HF : public XCTest XC_Functional::set_xc_type("PBE"); XC_Functional::set_xc_type("HF"); + const int gga_grad = 0; const double hybrid_alpha = 1.0; const double hse_omega = 0.0; - XC_Functional::gradcorr(et1,vt1,v1,&chr,&rhopw,&ucell,stress1,false,nspin1,domag,domag_z, hybrid_alpha, hse_omega); - XC_Functional::gradcorr(et1,vt1,v1,&chr,&rhopw,&ucell,stress1,true, nspin1,domag,domag_z, hybrid_alpha, hse_omega); + XC_Functional::gradcorr(et1,vt1,v1,&chr,&rhopw,&ucell,stress1,false,nspin1,domag,domag_z,gga_grad,hybrid_alpha,hse_omega); + XC_Functional::gradcorr(et1,vt1,v1,&chr,&rhopw,&ucell,stress1,true, nspin1,domag,domag_z,gga_grad,hybrid_alpha,hse_omega); - XC_Functional::gradcorr(et2,vt2,v2,&chr,&rhopw,&ucell,stress2,false,nspin2,domag,domag_z, hybrid_alpha, hse_omega); - XC_Functional::gradcorr(et2,vt2,v2,&chr,&rhopw,&ucell,stress2,true, nspin2,domag,domag_z, hybrid_alpha, hse_omega); + XC_Functional::gradcorr(et2,vt2,v2,&chr,&rhopw,&ucell,stress2,false,nspin2,domag,domag_z,gga_grad,hybrid_alpha,hse_omega); + XC_Functional::gradcorr(et2,vt2,v2,&chr,&rhopw,&ucell,stress2,true, nspin2,domag,domag_z,gga_grad,hybrid_alpha,hse_omega); - XC_Functional::gradcorr(et4,vt4,v4,&chr,&rhopw,&ucell,stress4,false,nspin4,domag_true,domag_z, hybrid_alpha, hse_omega); - XC_Functional::gradcorr(et4,vt4,v4,&chr,&rhopw,&ucell,stress4,true, nspin4,domag_true,domag_z, hybrid_alpha, hse_omega); + XC_Functional::gradcorr(et4,vt4,v4,&chr,&rhopw,&ucell,stress4,false,nspin4,domag_true,domag_z,gga_grad,hybrid_alpha,hse_omega); + XC_Functional::gradcorr(et4,vt4,v4,&chr,&rhopw,&ucell,stress4,true, nspin4,domag_true,domag_z,gga_grad,hybrid_alpha,hse_omega); } }; @@ -343,4 +345,4 @@ TEST_F(XCTest_GRADWFC, set_xc_type) EXPECT_NEAR(grad[i+j*5].imag(),0,1e-8); } } -} \ No newline at end of file +} diff --git a/source/source_hamilt/module_xc/test/test_xc5.cpp b/source/source_hamilt/module_xc/test/test_xc5.cpp index 01aa214feae..a2cc4a6c452 100644 --- a/source/source_hamilt/module_xc/test/test_xc5.cpp +++ b/source/source_hamilt/module_xc/test/test_xc5.cpp @@ -8,6 +8,13 @@ #include "source_cell/cal_ux.h" #include "../../../source_base/parallel_reduce.h" +#include +#include +#include +#include +#include +#include + /************************************************ * unit test of functionals ***********************************************/ @@ -22,7 +29,8 @@ class XCTest_VXC : public XCTest { protected: - double et1 = 0, vt1 = 0; + double et1 = 0; + double vt1 = 0; ModuleBase::matrix v1; double et2 = 0, vt2 = 0; @@ -82,13 +90,13 @@ class XCTest_VXC : public XCTest const double hybrid_alpha = XC_Functional::get_hybrid_alpha(); const double hse_omega = XC_Functional::get_hse_omega(); std::tuple etxc_vtxc_v - = XC_Functional::v_xc(rhopw.nrxx,&chr,&ucell,nspin1,domag,domag_z, hybrid_alpha, hse_omega); + = XC_Functional::v_xc(rhopw.nrxx,&chr,&ucell,nspin1,domag,domag_z,0, hybrid_alpha, hse_omega); et1 = std::get<0>(etxc_vtxc_v); vt1 = std::get<1>(etxc_vtxc_v); v1 = std::get<2>(etxc_vtxc_v); etxc_vtxc_v - = XC_Functional::v_xc(rhopw.nrxx,&chr,&ucell,nspin2,domag,domag_z, hybrid_alpha, hse_omega); + = XC_Functional::v_xc(rhopw.nrxx,&chr,&ucell,nspin2,domag,domag_z,0, hybrid_alpha, hse_omega); et2 = std::get<0>(etxc_vtxc_v); vt2 = std::get<1>(etxc_vtxc_v); v2 = std::get<2>(etxc_vtxc_v); @@ -126,7 +134,8 @@ class XCTest_VXC_Libxc : public XCTest { protected: - double et1 = 0, vt1 = 0; + double et1 = 0; + double vt1 = 0; ModuleBase::matrix v1; double et2 = 0, vt2 = 0; @@ -186,13 +195,13 @@ class XCTest_VXC_Libxc : public XCTest const double hybrid_alpha = XC_Functional::get_hybrid_alpha(); const double hse_omega = XC_Functional::get_hse_omega(); std::tuple etxc_vtxc_v - = XC_Functional::v_xc(rhopw.nrxx,&chr,&ucell,nspin1,domag,domag_z, hybrid_alpha, hse_omega); + = XC_Functional::v_xc(rhopw.nrxx,&chr,&ucell,nspin1,domag,domag_z,0, hybrid_alpha, hse_omega); et1 = std::get<0>(etxc_vtxc_v); vt1 = std::get<1>(etxc_vtxc_v); v1 = std::get<2>(etxc_vtxc_v); etxc_vtxc_v - = XC_Functional::v_xc(rhopw.nrxx,&chr,&ucell,nspin2,domag,domag_z, hybrid_alpha, hse_omega); + = XC_Functional::v_xc(rhopw.nrxx,&chr,&ucell,nspin2,domag,domag_z,0, hybrid_alpha, hse_omega); et2 = std::get<0>(etxc_vtxc_v); vt2 = std::get<1>(etxc_vtxc_v); v2 = std::get<2>(etxc_vtxc_v); @@ -230,12 +239,21 @@ class XCTest_VXC_meta : public XCTest { protected: - double et1 = 0, vt1 = 0; + double et1 = 0; + double vt1 = 0; ModuleBase::matrix v1,vtau1; double et2 = 0, vt2 = 0; ModuleBase::matrix v2,vtau2; + // Same libxc input, but with the whole density carried by the NLCC + // core charge instead of the valence density. + double et1_nlcc = 0, vt1_nlcc = 0; + ModuleBase::matrix v1_nlcc; + + double et2_nlcc = 0, vt2_nlcc = 0; + ModuleBase::matrix v2_nlcc; + void SetUp() { // Define variables for parameters @@ -310,6 +328,29 @@ class XCTest_VXC_meta : public XCTest vt2 = std::get<1>(etxc_vtxc_v); v2 = std::get<2>(etxc_vtxc_v); vtau2 = std::get<3>(etxc_vtxc_v); + + // Second scenario: drop the valence density and let the NLCC core + // charge carry the whole density. The density handed to libxc + // (rho, sigma, tau) is left unchanged, but vtxc = \int v_xc * + // rho_valence now has to vanish exactly. + for(int i=0;i<5;i++) + { + chr.rho[0][i] = 0; + chr.rho[1][i] = 0; + chr.rho_core[i] = double(i); + } + + etxc_vtxc_v + = XC_Functional_Libxc::v_xc_meta(XC_Functional::get_func_id(), rhopw.nrxx,ucell.omega,ucell.tpiba,&chr,nspin1, hybrid_alpha, hse_omega); + et1_nlcc = std::get<0>(etxc_vtxc_v); + vt1_nlcc = std::get<1>(etxc_vtxc_v); + v1_nlcc = std::get<2>(etxc_vtxc_v); + + etxc_vtxc_v + = XC_Functional_Libxc::v_xc_meta(XC_Functional::get_func_id(), rhopw.nrxx,ucell.omega,ucell.tpiba,&chr,nspin2, hybrid_alpha, hse_omega); + et2_nlcc = std::get<0>(etxc_vtxc_v); + vt2_nlcc = std::get<1>(etxc_vtxc_v); + v2_nlcc = std::get<2>(etxc_vtxc_v); } }; @@ -353,6 +394,392 @@ TEST_F(XCTest_VXC_meta, set_xc_type) EXPECT_NEAR(vtau2(1,4),0.0311787189,1.0e-8); } +/************************************************ + * unit tests for the gga_grad keyword (nspin=4 + * noncollinear GGA gradient methods) + * + * Method 2 differentiates the complete discrete local-spin-map and FFT + * gradient graph. Its reverse is tested on a real PW grid in + * test_xc_functional_ncgga_sf.cpp. + ************************************************/ + +namespace +{ +constexpr int gga_grad_nrxx = 5; + +// build a mock 4-component charge on the mocked 5-point grid. +// pattern 0: m = (0,0,mz), mz>0, so m_hat = (0,0,1) everywhere +// pattern 1: m direction varies from point to point +// pattern 2: m . ux changes sign across the grid (ux = (0,1,2) in the mock) +struct Ns4Charge +{ + ModulePW::PW_Basis rhopw; + UnitCell ucell; + Charge chr; + + Ns4Charge(const int pattern) + { + rhopw.nrxx = gga_grad_nrxx; + rhopw.npw = gga_grad_nrxx; + rhopw.nmaxgr = gga_grad_nrxx; + rhopw.gcar = new ModuleBase::Vector3[gga_grad_nrxx]; + rhopw.nxyz = 1; + + ucell.tpiba = 1; + ucell.omega = 1; + ucell.magnet.lsign_ = true; + unitcell::cal_ux(ucell, 4); + + chr.rhopw = &(rhopw); + chr.rho = new double*[4]; + for (int is = 0; is < 4; ++is) + { + chr.rho[is] = new double[gga_grad_nrxx]; + } + chr.rhog = new std::complex*[2]; + chr.rhog[0] = new std::complex[gga_grad_nrxx]; + chr.rhog[1] = new std::complex[gga_grad_nrxx]; + chr.rho_core = new double[gga_grad_nrxx]; + chr.rhog_core = new std::complex[gga_grad_nrxx]; + + for (int i = 0; i < gga_grad_nrxx; ++i) + { + chr.rho[0][i] = 2.0 + i; + if (pattern == 1) + { + chr.rho[1][i] = 0.10 * (i + 1); + chr.rho[2][i] = 0.05 * (gga_grad_nrxx - i); + chr.rho[3][i] = 0.20 * (i + 1); + } + else if (pattern == 2) + { + chr.rho[1][i] = 0.0; + chr.rho[2][i] = (i % 2 == 0) ? 0.3 : -0.3; + chr.rho[3][i] = 0.05; + } + else + { + chr.rho[1][i] = 0.0; + chr.rho[2][i] = 0.0; + chr.rho[3][i] = 0.2 * (i + 1); + } + chr.rhog[0][i] = chr.rho[0][i]; + chr.rhog[1][i] = chr.rho[1][i]; + chr.rho_core[i] = 0; + chr.rhog_core[i] = 0; + rhopw.gcar[i] = 1; + } + } +}; + + + +struct Ns2LocalCharge +{ + ModulePW::PW_Basis rhopw; + Charge chr; + + Ns2LocalCharge() + { + rhopw.nrxx = 1; + rhopw.npw = 1; + rhopw.nmaxgr = 1; + rhopw.nxyz = 1; + rhopw.gcar = new ModuleBase::Vector3[1]; + rhopw.gcar[0] = 0.0; + + chr.rhopw = &rhopw; + chr.rho = new double*[2]; + chr.rhog = new std::complex*[2]; + for (int is = 0; is < 2; ++is) + { + chr.rho[is] = new double[1]; + chr.rhog[is] = new std::complex[1]; + chr.rhog[is][0] = 0.0; + } + chr.rho_core = new double[1]; + chr.rhog_core = new std::complex[1]; + chr.rho_core[0] = 0.0; + chr.rhog_core[0] = 0.0; + } +}; + +// run XC_Functional::v_xc for nspin=4 with noncollinear magnetism +std::tuple run_vxc_nspin4( + const std::string& functional, + const int pattern, + const int gga_grad) +{ + Ns4Charge mock(pattern); + XC_Functional::set_xc_type(functional); + return XC_Functional::v_xc(gga_grad_nrxx, + &mock.chr, + &mock.ucell, + 4, + true, + false, + gga_grad, + XC_Functional::get_hybrid_alpha(), + XC_Functional::get_hse_omega()); +} + +// compare two (etxc, vtxc, v) results +void expect_vxc_equal(const std::tuple& a, + const std::tuple& b, + const double tol) +{ + EXPECT_NEAR(std::get<0>(a), std::get<0>(b), tol); + EXPECT_NEAR(std::get<1>(a), std::get<1>(b), tol); + const ModuleBase::matrix& va = std::get<2>(a); + const ModuleBase::matrix& vb = std::get<2>(b); + ASSERT_EQ(va.nr, vb.nr); + ASSERT_EQ(va.nc, vb.nc); + for (int ir = 0; ir < va.nr; ++ir) + { + for (int ic = 0; ic < va.nc; ++ic) + { + EXPECT_NEAR(va(ir, ic), vb(ir, ic), tol); + } + } +} +} // namespace + +// m_hat = m/|m|, zero where |m| ~ 0 + + +// v_tot = 0.5*(v_up+v_dn), v_mu = 0.5*(v_up-v_dn)*m_hat_mu + + +// original conversion: has_mag=false leaves magnetic channels zero +TEST(GgaGradTools, ConvertVNspin4HasMag) +{ + Ns4Charge mock(0); + std::vector amag(gga_grad_nrxx); + for (int ir = 0; ir < gga_grad_nrxx; ++ir) + { + amag[ir] = mock.chr.rho[3][ir]; + } + ModuleBase::matrix v(2, gga_grad_nrxx); + for (int ir = 0; ir < gga_grad_nrxx; ++ir) + { + v(0, ir) = 1.0 + ir; + v(1, ir) = 0.5 * ir; + } + + const ModuleBase::matrix v_nomag + = XC_Functional_Libxc::convert_v_nspin4(gga_grad_nrxx, &mock.chr, amag, v, false); + for (int ir = 0; ir < gga_grad_nrxx; ++ir) + { + EXPECT_NEAR(v_nomag(0, ir), 0.5 * (v(0, ir) + v(1, ir)), 1e-14); + EXPECT_NEAR(v_nomag(1, ir), 0.0, 1e-14); + EXPECT_NEAR(v_nomag(2, ir), 0.0, 1e-14); + EXPECT_NEAR(v_nomag(3, ir), 0.0, 1e-14); + } + + const ModuleBase::matrix v_mag + = XC_Functional_Libxc::convert_v_nspin4(gga_grad_nrxx, &mock.chr, amag, v, true); + for (int ir = 0; ir < gga_grad_nrxx; ++ir) + { + const double vs = 0.5 * (v(0, ir) - v(1, ir)); + EXPECT_NEAR(v_mag(3, ir), vs * mock.chr.rho[3][ir] / amag[ir], 1e-14); + } +} + + + +// gga_grad=0 keeps the original built-in algorithm and must not crash +TEST(GgaGradVxc, BuiltinOriginalAlgorithmRuns) +{ + const std::tuple r0 = run_vxc_nspin4("PBE", 1, 0); + EXPECT_EQ(std::get<2>(r0).nr, 4); + EXPECT_TRUE(std::isfinite(std::get<0>(r0))); + EXPECT_TRUE(std::isfinite(std::get<1>(r0))); +} + +// noncolin_rho with lsign=true defines up/down w.r.t. the global axis ux +// through sign(m . ux); with lsign=false, up is always the local |m| +TEST(GgaGradTools, NoncolinRhoGlobalAxis) +{ + Ns4Charge mock(2); // pattern 2: m . ux changes sign across the grid + const double* ux = mock.ucell.magnet.ux_; // (0,1,2) in the mock + + std::vector rup(gga_grad_nrxx), rdn(gga_grad_nrxx), neg(gga_grad_nrxx); + XC_Functional::noncolin_rho( + rup.data(), rdn.data(), neg.data(), mock.chr.rho, gga_grad_nrxx, ux, true); + for (int ir = 0; ir < gga_grad_nrxx; ++ir) + { + const double mx = mock.chr.rho[1][ir]; + const double my = mock.chr.rho[2][ir]; + const double mz = mock.chr.rho[3][ir]; + const double amag = std::sqrt(mx * mx + my * my + mz * mz); + const double sign = (mx * ux[0] + my * ux[1] + mz * ux[2] > 0) ? 1.0 : -1.0; + EXPECT_NEAR(rup[ir], 0.5 * (mock.chr.rho[0][ir] + sign * amag), 1e-14); + EXPECT_NEAR(rdn[ir], 0.5 * (mock.chr.rho[0][ir] - sign * amag), 1e-14); + } + // the sign really flips on this grid, i.e. the global axis matters here + EXPECT_NEAR(neg[0], 1.0, 1e-14); + EXPECT_NEAR(neg[1], -1.0, 1e-14); + + XC_Functional::noncolin_rho( + rup.data(), rdn.data(), neg.data(), mock.chr.rho, gga_grad_nrxx, ux, false); + for (int ir = 0; ir < gga_grad_nrxx; ++ir) + { + const double mx = mock.chr.rho[1][ir]; + const double my = mock.chr.rho[2][ir]; + const double mz = mock.chr.rho[3][ir]; + const double amag = std::sqrt(mx * mx + my * my + mz * mz); + EXPECT_NEAR(rup[ir], 0.5 * (mock.chr.rho[0][ir] + amag), 1e-14); + EXPECT_NEAR(rdn[ir], 0.5 * (mock.chr.rho[0][ir] - amag), 1e-14); + } +} + +// gga_grad=1 must ignore the global magnetization direction: with lsign_=true +// it has to give the same gradcorr result as gga_grad=0 with lsign_=false +TEST(GgaGradVxc, BuiltinGgaGrad1IgnoresGlobalAxis) +{ + XC_Functional::set_xc_type("PBE"); + const double hybrid_alpha = XC_Functional::get_hybrid_alpha(); + const double hse_omega = XC_Functional::get_hse_omega(); + + Ns4Charge mock_a(2); // lsign_ = true + double et1 = 0; + double vt1 = 0; + ModuleBase::matrix v1(4, gga_grad_nrxx); + std::vector dum; + XC_Functional::gradcorr(et1, vt1, v1, &mock_a.chr, &mock_a.rhopw, &mock_a.ucell, + dum, false, 4, true, false, 1, hybrid_alpha, hse_omega); + + Ns4Charge mock_b(2); + mock_b.ucell.magnet.lsign_ = false; + double et0 = 0; + double vt0 = 0; + ModuleBase::matrix v0(4, gga_grad_nrxx); + XC_Functional::gradcorr(et0, vt0, v0, &mock_b.chr, &mock_b.rhopw, &mock_b.ucell, + dum, false, 4, true, false, 0, hybrid_alpha, hse_omega); + + EXPECT_NEAR(et0, et1, 1e-12); + EXPECT_NEAR(vt0, vt1, 1e-12); + for (int is = 0; is < 4; ++is) + { + for (int ir = 0; ir < gga_grad_nrxx; ++ir) + { + EXPECT_NEAR(v0(is, ir), v1(is, ir), 1e-12); + } + } +} + + + +TEST(GgaGradVxc, LibxcNspin4NearSaturationDifferentiatesTheWeightedEnergy) +{ + constexpr double density_threshold = 1.0e-6; + Ns4Charge mock(0); + for (int ir = 0; ir < gga_grad_nrxx; ++ir) + { + mock.chr.rho[0][ir] = 0.45; + mock.chr.rho[1][ir] = 0.0; + mock.chr.rho[2][ir] = 0.0; + mock.chr.rho[3][ir] = 0.45 - density_threshold; + } + const std::vector func_ids = {XC_LDA_X}; + const int gga_grad_modes[] = {2}; + for (const int gga_grad : gga_grad_modes) + { + const std::function()> evaluate = [&, gga_grad]() + { + return XC_Functional_Libxc::v_xc_libxc(func_ids, + gga_grad_nrxx, + mock.ucell.omega, + mock.ucell.tpiba, + &mock.chr, + 4, + true, + false, + gga_grad, + nullptr, + 0.0, + 0.0); + }; + + const std::tuple reference = evaluate(); + const int components[] = {0, 3}; + const double steps[] = {8.0e-8, 4.0e-8, 2.0e-8}; + for (const int component : components) + { + double analytic = std::get<2>(reference)(component, 0); + Parallel_Reduce::reduce_pool(analytic); + if (std::getenv("ABACUS_XC_FD_TRACE") != nullptr) + { + std::cout << std::setprecision(17) + << "XC_SANITIZER_REFERENCE case=nspin4_near_saturation" + << " gga_grad=" << gga_grad + << " component=" << component + << " energy=" << std::get<0>(reference) + << " vtxc=" << std::get<1>(reference) + << " analytic=" << analytic << std::endl; + } + const double original = mock.chr.rho[component][0]; + for (const double step : steps) + { + mock.chr.rho[component][0] = original + step; + const double energy_plus = std::get<0>(evaluate()); + mock.chr.rho[component][0] = original - step; + const double energy_minus = std::get<0>(evaluate()); + mock.chr.rho[component][0] = original; + + const double finite_difference = (energy_plus - energy_minus) / (2.0 * step); + if (std::getenv("ABACUS_XC_FD_TRACE") != nullptr) + { + std::cout << std::setprecision(17) + << "XC_SANITIZER_FD case=nspin4_near_saturation" + << " gga_grad=" << gga_grad + << " component=" << component + << " eps=" << step + << " analytic=" << analytic + << " finite_difference=" << finite_difference + << " absolute_error=" << std::abs(analytic - finite_difference) + << std::endl; + } + const double scale = std::max(1.0, std::max(std::abs(analytic), + std::abs(finite_difference))); + EXPECT_NEAR(analytic, finite_difference, 2.0e-8 * scale) + << "gga_grad=" << gga_grad + << ", component=" << component + << ", step=" << step; + } + } + } +} + +// for LIBXC, gga_grad=0 and 1 both keep the original collinear algorithm +TEST(GgaGradVxc, LibxcZeroEqualsOne) +{ + const std::tuple r0 = run_vxc_nspin4("GGA_X_PBE+GGA_C_PBE", 1, 0); + const std::tuple r1 = run_vxc_nspin4("GGA_X_PBE+GGA_C_PBE", 1, 1); + expect_vxc_equal(r0, r1, 1e-12); +} + +// Regression test for the NLCC (non-linear core correction) handling in +// v_xc_meta. Every contribution to vtxc is weighted by the valence density +// (chr->rho), i.e. vtxc = \int v_xc * rho_valence, so a zero valence density +// must give a vanishing vtxc even when the NLCC core charge keeps the density +// handed to libxc non-trivial. The vsigma (density-gradient) term used to be +// weighted by the total density instead, which left a spurious non-zero vtxc +// and hence a wrong analytical stress for meta-GGA + NLCC pseudopotentials. +TEST_F(XCTest_VXC_meta, vtxc_uses_valence_density) +{ + EXPECT_NEAR(vt1_nlcc,0.0,1.0e-12); + EXPECT_NEAR(vt2_nlcc,0.0,1.0e-12); + + // The core charge does reach libxc, so the checks above are not vacuous: + // for nspin==1 the total density is bit-identical to the first scenario, + // hence etxc has to agree with it exactly. + EXPECT_GT(std::abs(v1_nlcc(0,1)),1.0e-3); + EXPECT_GT(std::abs(v2_nlcc(0,1)),1.0e-3); + EXPECT_NEAR(et1_nlcc,et1,1.0e-12); + EXPECT_GT(std::abs(et2_nlcc),1.0e-3); +} + int main(int argc, char **argv) { @@ -361,4 +788,4 @@ int main(int argc, char **argv) int result = RUN_ALL_TESTS(); MPI_Finalize(); return result; -} \ No newline at end of file +} diff --git a/source/source_hamilt/module_xc/test/test_xc_functional_ncgga_sf.cpp b/source/source_hamilt/module_xc/test/test_xc_functional_ncgga_sf.cpp new file mode 100644 index 00000000000..3166ec625fa --- /dev/null +++ b/source/source_hamilt/module_xc/test/test_xc_functional_ncgga_sf.cpp @@ -0,0 +1,295 @@ +#include "../xc_functional.h" +#include "../libxc_abacus.h" +#include "source_base/constants.h" +#include "source_base/parallel_comm.h" +#include "source_base/parallel_global.h" +#include "source_base/parallel_reduce.h" +#include "source_basis/module_pw/pw_basis.h" +#include "source_cell/unitcell.h" +#include "source_estate/module_charge/charge.h" +#include "gtest/gtest.h" +#include +#include +#include +#include +#include +#include + +// The focused target links PW/XC objects; the fixture owns charge storage. +Charge::Charge() {} +Charge::~Charge() {} +UnitCell::UnitCell() {} +UnitCell::~UnitCell() {} +Magnetism::Magnetism() {} +Magnetism::~Magnetism() {} +SepPot::SepPot() {} +SepPot::~SepPot() {} +Sep_Cell::Sep_Cell() noexcept {} +Sep_Cell::~Sep_Cell() noexcept {} + +namespace +{ +int test_rank = 0; +int test_size = 1; +const int noncollinear_spin = 4; +const int variational_gga = 2; +const double hybrid_alpha = 0.0; +const double hse_omega = 0.11; +typedef std::tuple VxcResult; +enum Branch { Smooth, Negative, Saturated, Radial }; + +// Each branch runs through both the built-in and LibXC production dispatch. +class RealPwNcgga : public testing::TestWithParam> +{ + protected: + ModulePW::PW_Basis pw; + UnitCell cell; + Charge charge; + std::array, 4> density; + std::array, 4> direction; + std::array pointers; + std::vector core; + std::vector> core_g; + + double sum(double value) + { + Parallel_Reduce::reduce_pool(value); + return value; + } + + void SetUp() override + { + pw.initmpi(test_size, test_rank, MPI_COMM_WORLD); + // Unequal MPI2 slabs and a finite PW cutoff exercise the projection. + pw.initgrids(7.0, ModuleBase::Matrix3(1, 0, 0, 0, 1, 0, 0, 0, 1), 24, 10, 9); + pw.initparameters(false, 80.0, 2, false); + pw.setuptransform(); + pw.collect_local_pw(); + cell.omega = pw.omega; + cell.tpiba = pw.tpiba; + cell.magnet.lsign_ = false; + for (int c = 0; c < noncollinear_spin; ++c) + { + density[c].resize(pw.nrxx); + direction[c].resize(pw.nrxx); + pointers[c] = density[c].data(); + } + core.resize(pw.nrxx); + core_g.resize(pw.npw); + charge.rhopw = &pw; + charge.nrxx = pw.nrxx; + charge.nxyz = pw.nxyz; + charge.ngmc = pw.npw; + charge.nspin = noncollinear_spin; + charge.rho = pointers.data(); + charge.rho_core = core.data(); + charge.rhog_core = core_g.data(); + const int branch = std::get<1>(GetParam()); + const double totals[] = {2.2, -1.55, 0.43, 0.030}; + const double magnitudes[] = {0.62, 0.45, 0.95, 4.0e-4}; + for (int ir = 0; ir < pw.nrxx; ++ir) + { + const double x = ModuleBase::TWO_PI * (ir / (pw.ny * pw.nplane)) / pw.nx; + const double y = ModuleBase::TWO_PI * ((ir / pw.nplane) % pw.ny) / pw.ny; + const double z = ModuleBase::TWO_PI * (ir % pw.nplane + pw.startz_current) / pw.nz; + density[0][ir] = totals[branch] * (1.0 + 0.08 * std::sin(x) + 0.03 * std::cos(y + z)); + const double magnitude = magnitudes[branch] * (1.0 + 0.1 * std::cos(5 * x) + 0.03 * std::sin(y)); + const double theta = 0.7 + 0.32 * std::sin(3 * x) + 0.18 * std::cos(7 * x) + 0.1 * std::cos(y); + const double phi = 0.4 + 0.27 * std::cos(4 * x) - 0.16 * std::sin(6 * x) + 0.09 * std::sin(z); + density[1][ir] = magnitude * std::sin(theta) * std::cos(phi); + density[2][ir] = magnitude * std::sin(theta) * std::sin(phi); + density[3][ir] = magnitude * std::cos(theta); + core[ir] = 0.05 * totals[branch] * (1.0 + 0.1 * std::cos(3 * x + z)); + if (branch == Negative) ASSERT_LT(density[0][ir] + core[ir], 0.0); + if (branch == Saturated) ASSERT_GT(magnitude, density[0][ir] + core[ir]); + if (branch == Radial) ASSERT_LT(magnitude, 1e-3); + for (int c = 0; c < noncollinear_spin; ++c) + direction[c][ir] = 0.1 + 0.2 * std::cos((c + 2) * x + y) + 0.1 * std::sin(z + c); + } + pw.real2recip(core.data(), core_g.data()); + XC_Functional::set_xc_type(std::get<0>(GetParam()) ? "GGA_X_PBE+GGA_C_PBE" : "PBE"); + } + + VxcResult evaluate() + { + return XC_Functional::v_xc(pw.nrxx, &charge, &cell, noncollinear_spin, + true, false, variational_gga, hybrid_alpha, hse_omega); + } + + double inner(const ModuleBase::matrix& potential, const bool perturbation) + { + double value = 0.0; + for (int c = 0; c < noncollinear_spin; ++c) + for (int ir = 0; ir < pw.nrxx; ++ir) + value += potential(c, ir) * (perturbation ? direction[c][ir] : density[c][ir]); + return sum(value) * pw.omega / pw.nxyz; + } +}; + +TEST_P(RealPwNcgga, GradientAndDivergenceAreAdjoints) +{ + std::vector> reciprocal(pw.npw); + std::vector> field(pw.nrxx); + std::vector> gradient(pw.nrxx); + std::vector divergence(pw.nrxx); + for (int ir = 0; ir < pw.nrxx; ++ir) + field[ir] = ModuleBase::Vector3(direction[1][ir], direction[2][ir], direction[3][ir]); + pw.real2recip(direction[0].data(), reciprocal.data()); + XC_Functional::grad_rho(reciprocal.data(), gradient.data(), &pw, pw.tpiba); + XC_Functional::grad_dot(field.data(), divergence.data(), &pw, pw.tpiba); + double identity = 0.0; + double norm = 0.0; + for (int ir = 0; ir < pw.nrxx; ++ir) + { + const double left = field[ir] * gradient[ir]; + const double right = divergence[ir] * direction[0][ir]; + identity += left + right; + norm += std::abs(left) + std::abs(right); + } + const double global_norm = sum(norm); + EXPECT_GT(global_norm, 1e-4); + EXPECT_LE(std::abs(sum(identity)), 5e-11 * global_norm); +} + +TEST_P(RealPwNcgga, DensityAndCoreDerivatives) +{ + const VxcResult reference = evaluate(); + const ModuleBase::matrix& potential = std::get<2>(reference); + EXPECT_NEAR(std::get<1>(reference), inner(potential, false), 2e-12 * std::max(1.0, std::abs(std::get<1>(reference)))); + // Four independent spin channels plus a core-density perturbation. + const bool radial = std::get<1>(GetParam()) == Radial; + const double step = radial ? 2e-4 : 2e-3; + const int refinements = radial ? 10 : 3; + for (int channel = 0; channel <= noncollinear_spin; ++channel) + { + SCOPED_TRACE(channel); + const bool is_core = channel == noncollinear_spin; + const int c = is_core ? 0 : channel; + std::vector& values = is_core ? core : density[c]; + const std::vector original = values; + double analytic = 0.0; + for (int ir = 0; ir < pw.nrxx; ++ir) + analytic += potential(c, ir) * direction[c][ir]; + analytic = sum(analytic) * pw.omega / pw.nxyz; + double previous = 0.0; + double best = 1.0; + for (int level = 0; level < refinements; ++level) + { + const double epsilon = step / (1 << level); + double energies[2]; + for (int sign = 0; sign < 2; ++sign) + { + for (int ir = 0; ir < pw.nrxx; ++ir) + values[ir] = original[ir] + (sign == 0 ? epsilon : -epsilon) * direction[c][ir]; + if (is_core) pw.real2recip(core.data(), core_g.data()); + energies[sign] = std::get<0>(evaluate()); + } + std::copy(original.begin(), original.end(), values.begin()); + if (is_core) pw.real2recip(core.data(), core_g.data()); + const double error = std::abs((energies[0] - energies[1]) / (2 * epsilon) - analytic); + const double scale = std::max(1.0, std::abs(analytic)); + if (level > 0 && level <= 2) EXPECT_LE(error, 0.4 * previous + 5e-9 * scale); + best = std::min(best, error); + if (level == refinements - 1) EXPECT_LE(best, 3e-8 * scale); + previous = error; + } + } +} + +TEST_P(RealPwNcgga, SixComponentStressDerivative) +{ + double energy = 0.0; + double vtxc = 0.0; + ModuleBase::matrix potential; + std::vector stress; + XC_Functional::gradcorr(energy, vtxc, potential, &charge, &pw, &cell, stress, + true, noncollinear_spin, true, false, variational_gga, hybrid_alpha, hse_omega); + ASSERT_EQ(stress.size(), 9U); + const std::vector> original(pw.gcar, pw.gcar + pw.npw); + for (int row = 0; row < 3; ++row) + for (int column = 0; column <= row; ++column) + { + SCOPED_TRACE(row * 3 + column); + const double analytic = sum(stress[row * 3 + column]) / pw.nxyz; + double previous = 0.0; + double best = 1.0; + for (int level = 0; level < 4; ++level) + { + const double epsilon = 2e-3 / (1 << level); + double energies[2]; + for (int sign = 0; sign < 2; ++sign) + { + // Inverse of I + epsilon e_row e_column, holding density fixed. + const double strain = sign == 0 ? epsilon : -epsilon; + const double factor = row == column ? strain / (1 + strain) : strain; + for (int ig = 0; ig < pw.npw; ++ig) + { + pw.gcar[ig] = original[ig]; + pw.gcar[ig][column] -= factor * original[ig][row]; + } + energies[sign] = std::get<0>(evaluate()); + } + std::copy(original.begin(), original.end(), pw.gcar); + const double error = std::abs(-(energies[0] - energies[1]) / (2 * epsilon * pw.omega) - analytic); + const double tolerance = 3e-4 * std::max(1e-10, std::abs(analytic)) + 1e-11; + if (level > 0 && previous > 1e-10 && error > 1e-10) + EXPECT_LE(error, 0.4 * previous + tolerance); + best = std::min(best, error); + if (level == 3) EXPECT_LE(best, tolerance); + previous = error; + } + } +} + +TEST_P(RealPwNcgga, SpinRotationAndInversion) +{ + const VxcResult reference = evaluate(); + // Cyclic permutation is a proper global rotation; inversion is checked separately. + for (int inversion = 0; inversion < 2; ++inversion) + { + const std::array, 4> original = density; + for (int c = 1; c < noncollinear_spin; ++c) + for (int ir = 0; ir < pw.nrxx; ++ir) + density[c][ir] = (inversion == 0 ? 1 : -1) * original[c % 3 + 1][ir]; + const VxcResult rotated = evaluate(); + EXPECT_NEAR(std::get<0>(reference), std::get<0>(rotated), 1e-10); + for (int c = 0; c < noncollinear_spin; ++c) + for (int ir = 0; ir < pw.nrxx; ++ir) + EXPECT_NEAR(std::get<2>(rotated)(c, ir), + (c == 0 ? 1 : (inversion == 0 ? 1 : -1)) + * std::get<2>(reference)(c == 0 ? 0 : c % 3 + 1, ir), 1e-11); + for (int c = 0; c < noncollinear_spin; ++c) + std::copy(original[c].begin(), original[c].end(), density[c].begin()); + } +} + +TEST_P(RealPwNcgga, ZeroMagnetizationIsFinite) +{ + for (int c = 1; c < noncollinear_spin; ++c) + std::fill(density[c].begin(), density[c].end(), 0.0); + const VxcResult result = evaluate(); + EXPECT_TRUE(std::isfinite(std::get<0>(result))); + for (int c = 1; c < noncollinear_spin; ++c) + for (int ir = 0; ir < pw.nrxx; ++ir) + EXPECT_NEAR(std::get<2>(result)(c, ir), 0.0, 1e-13); +} + +INSTANTIATE_TEST_SUITE_P(ProductionBranches, RealPwNcgga, + testing::Combine(testing::Bool(), testing::Values(Smooth, Negative, Saturated, Radial))); +} // namespace + +int main(int argc, char** argv) +{ + int threads = 1; + Parallel_Global::read_pal_param(argc, argv, test_size, threads, test_rank); + POOL_WORLD = MPI_COMM_WORLD; + KP_WORLD = MPI_COMM_NULL; + INT_BGROUP = MPI_COMM_NULL; + BP_WORLD = MPI_COMM_NULL; + GRID_WORLD = MPI_COMM_NULL; + DIAG_WORLD = MPI_COMM_NULL; + testing::InitGoogleTest(&argc, argv); + const int result = RUN_ALL_TESTS(); + Parallel_Global::finalize_mpi(); + return result; +} diff --git a/source/source_hamilt/module_xc/test/test_xc_ncgga_radial.cpp b/source/source_hamilt/module_xc/test/test_xc_ncgga_radial.cpp new file mode 100644 index 00000000000..3ad9a9d19f0 --- /dev/null +++ b/source/source_hamilt/module_xc/test/test_xc_ncgga_radial.cpp @@ -0,0 +1,278 @@ +#include "../xc_ncgga_radial.h" + +#include "gtest/gtest.h" +#include +#include +#include +#include +#include +#include + +namespace +{ + +double dot(const std::array& left, const std::array& right) +{ + double result = 0.0; + for (int component = 0; component < 3; ++component) + { + result += left[component] * right[component]; + } + return result; +} + +std::array shifted(const std::array& point, + const std::array& direction, + const double step) +{ + std::array result = point; + for (int component = 0; component < 3; ++component) + { + result[component] += step * direction[component]; + } + return result; +} + +double directional_hessian(const ModuleXC::NcggaRadialPoint& point, const std::array& direction) +{ + double result = 0.0; + for (int row = 0; row < 3; ++row) + { + for (int column = 0; column < 3; ++column) + { + result += direction[row] * point.jacobian(row, column) * direction[column]; + } + } + return result; +} + +ModuleXC::NcggaSpinMapPoint make_spin_map(const std::array& state, const double eta) +{ + const std::array magnetization = {{state[1], state[2], state[3]}}; + return ModuleXC::make_ncgga_spin_map_point(state[0], ModuleXC::make_ncgga_radial_point(magnetization, eta)); +} + +TEST(NcggaRadial, RejectsNonPositiveEta) +{ + const std::array magnetization = {{0.1, 0.0, 0.0}}; + EXPECT_THROW(ModuleXC::make_ncgga_radial_point(magnetization, 0.0), std::invalid_argument); + EXPECT_THROW(ModuleXC::make_ncgga_radial_point(magnetization, -0.5), std::invalid_argument); +} + +TEST(NcggaRadial, MatchesZeroInteriorJoinAndExteriorValues) +{ + const double eta = 0.5; + const ModuleXC::NcggaRadialPoint zero = ModuleXC::make_ncgga_radial_point({{0.0, 0.0, 0.0}}, eta); + EXPECT_DOUBLE_EQ(zero.value, 0.0); + for (int row = 0; row < 3; ++row) + { + EXPECT_DOUBLE_EQ(zero.direction[row], 0.0); + EXPECT_DOUBLE_EQ(zero.gradient[row], 0.0); + for (int column = 0; column < 3; ++column) + { + EXPECT_DOUBLE_EQ(zero.jacobian(row, column), 0.0); + } + } + + const ModuleXC::NcggaRadialPoint interior = ModuleXC::make_ncgga_radial_point({{0.5 * eta, 0.0, 0.0}}, eta); + EXPECT_DOUBLE_EQ(interior.value, 11.0 * eta / 32.0); + EXPECT_DOUBLE_EQ(interior.gradient[0], 23.0 / 16.0); + EXPECT_DOUBLE_EQ(interior.jacobian(0, 0), 3.0 / (2.0 * eta)); + EXPECT_DOUBLE_EQ(interior.jacobian(1, 1), 23.0 / (8.0 * eta)); + EXPECT_DOUBLE_EQ(interior.jacobian(2, 2), 23.0 / (8.0 * eta)); + + const ModuleXC::NcggaRadialPoint join = ModuleXC::make_ncgga_radial_point({{eta, 0.0, 0.0}}, eta); + EXPECT_DOUBLE_EQ(join.value, eta); + EXPECT_DOUBLE_EQ(join.gradient[0], 1.0); + EXPECT_DOUBLE_EQ(join.jacobian(0, 0), 0.0); + EXPECT_DOUBLE_EQ(join.jacobian(1, 1), 1.0 / eta); + EXPECT_DOUBLE_EQ(join.jacobian(2, 2), 1.0 / eta); + + const ModuleXC::NcggaRadialPoint exterior = ModuleXC::make_ncgga_radial_point({{3.0, 4.0, 0.0}}, eta); + EXPECT_DOUBLE_EQ(exterior.value, 5.0); + EXPECT_DOUBLE_EQ(exterior.gradient[0], 0.6); + EXPECT_DOUBLE_EQ(exterior.gradient[1], 0.8); + EXPECT_NEAR(exterior.jacobian(0, 0), 0.128, 1.0e-15); + EXPECT_NEAR(exterior.jacobian(0, 1), -0.096, 1.0e-15); + EXPECT_NEAR(exterior.jacobian(1, 1), 0.072, 1.0e-15); + EXPECT_NEAR(exterior.jacobian(2, 2), 0.2, 1.0e-15); +} + +TEST(NcggaRadial, CompactBranchConvergesC2AtJoin) +{ + const double eta = 0.5; + const double coarse_delta = 1.0e-3; + const double fine_delta = 0.5 * coarse_delta; + const ModuleXC::NcggaRadialPoint coarse + = ModuleXC::make_ncgga_radial_point({{eta * (1.0 - coarse_delta), 0.0, 0.0}}, eta); + const ModuleXC::NcggaRadialPoint fine + = ModuleXC::make_ncgga_radial_point({{eta * (1.0 - fine_delta), 0.0, 0.0}}, eta); + + EXPECT_LE(std::abs(fine.value - eta), 0.51 * std::abs(coarse.value - eta)); + EXPECT_LE(std::abs(fine.gradient[0] - 1.0), 0.26 * std::abs(coarse.gradient[0] - 1.0)); + EXPECT_LE(std::abs(fine.jacobian(0, 0)), 0.51 * std::abs(coarse.jacobian(0, 0))); + EXPECT_LE(std::abs(fine.jacobian(1, 1) - 1.0 / eta), 0.51 * std::abs(coarse.jacobian(1, 1) - 1.0 / eta)); +} + +TEST(NcggaRadial, CompactBranchConvergesC2AtOrigin) +{ + const double eta = 0.5; + const double coarse_radius = 1.0e-3 * eta; + const double fine_radius = 0.5 * coarse_radius; + const ModuleXC::NcggaRadialPoint coarse = ModuleXC::make_ncgga_radial_point({{coarse_radius, 0.0, 0.0}}, eta); + const ModuleXC::NcggaRadialPoint fine = ModuleXC::make_ncgga_radial_point({{fine_radius, 0.0, 0.0}}, eta); + + EXPECT_LE(fine.value, 0.13 * coarse.value); + EXPECT_LE(std::abs(fine.gradient[0]), 0.26 * std::abs(coarse.gradient[0])); + EXPECT_LE(std::abs(fine.jacobian(0, 0)), 0.51 * std::abs(coarse.jacobian(0, 0))); + EXPECT_LE(std::abs(fine.jacobian(1, 1)), 0.51 * std::abs(coarse.jacobian(1, 1))); +} + +TEST(NcggaRadial, GradientMatchesCentralFiniteDifference) +{ + const double eta = 0.5; + const double step = 2.0e-6 * eta; + const std::array direction = {{0.31, -0.27, 0.19}}; + const std::array interior = {{0.12, -0.08, 0.05}}; + const std::array exterior = {{0.62, -0.41, 0.23}}; + const std::array join = {{eta, 0.0, 0.0}}; + const std::array, 3> points = {{interior, exterior, join}}; + + for (std::size_t sample = 0; sample < points.size(); ++sample) + { + const ModuleXC::NcggaRadialPoint point = ModuleXC::make_ncgga_radial_point(points[sample], eta); + const double finite_difference + = (ModuleXC::make_ncgga_radial_point(shifted(points[sample], direction, step), eta).value + - ModuleXC::make_ncgga_radial_point(shifted(points[sample], direction, -step), eta).value) + / (2.0 * step); + EXPECT_NEAR(finite_difference, dot(point.gradient, direction), 2.0e-10); + } +} + +TEST(NcggaRadial, JacobianMatchesGradientFiniteDifference) +{ + const double eta = 0.5; + const double step = 2.0e-6 * eta; + const std::array direction = {{0.31, -0.27, 0.19}}; + const std::array interior = {{0.12, -0.08, 0.05}}; + const std::array exterior = {{0.62, -0.41, 0.23}}; + const std::array, 2> points = {{interior, exterior}}; + + for (std::size_t sample = 0; sample < points.size(); ++sample) + { + const ModuleXC::NcggaRadialPoint point = ModuleXC::make_ncgga_radial_point(points[sample], eta); + const ModuleXC::NcggaRadialPoint plus + = ModuleXC::make_ncgga_radial_point(shifted(points[sample], direction, step), eta); + const ModuleXC::NcggaRadialPoint minus + = ModuleXC::make_ncgga_radial_point(shifted(points[sample], direction, -step), eta); + for (int row = 0; row < 3; ++row) + { + double analytic = 0.0; + for (int column = 0; column < 3; ++column) + { + analytic += point.jacobian(row, column) * direction[column]; + EXPECT_NEAR(point.jacobian(row, column), point.jacobian(column, row), 1.0e-15); + } + const double finite_difference = (plus.gradient[row] - minus.gradient[row]) / (2.0 * step); + EXPECT_NEAR(finite_difference, analytic, 2.0e-9); + } + } +} + +TEST(NcggaRadial, DirectionalSecondDerivativeMatchesValueFiniteDifference) +{ + const double eta = 0.5; + const double step = 2.0e-4 * eta; + const std::array direction = {{0.31, -0.27, 0.19}}; + const std::array interior = {{0.12, -0.08, 0.05}}; + const std::array exterior = {{0.62, -0.41, 0.23}}; + const std::array, 2> points = {{interior, exterior}}; + + for (std::size_t sample = 0; sample < points.size(); ++sample) + { + const ModuleXC::NcggaRadialPoint point = ModuleXC::make_ncgga_radial_point(points[sample], eta); + const double plus = ModuleXC::make_ncgga_radial_point(shifted(points[sample], direction, step), eta).value; + const double minus = ModuleXC::make_ncgga_radial_point(shifted(points[sample], direction, -step), eta).value; + const double finite_difference = (plus - 2.0 * point.value + minus) / (step * step); + EXPECT_NEAR(finite_difference, directional_hessian(point, direction), 2.0e-7); + } +} + +TEST(NcggaRadial, SpinDensityMapJacobianMatchesFiniteDifferenceInEachBranch) +{ + const double eta = 0.5; + const double step = 2.0e-7; + const std::array, 5> states = {{{{1.20, 0.30, -0.20, 0.10}}, + {{-1.20, 0.30, -0.20, 0.10}}, + {{0.20, 0.42, -0.31, 0.16}}, + {{-0.20, 0.42, -0.31, 0.16}}, + {{0.90, 0.12, -0.08, 0.05}}}}; + + for (std::size_t sample = 0; sample < states.size(); ++sample) + { + const ModuleXC::NcggaSpinMapPoint point = make_spin_map(states[sample], eta); + EXPECT_GE(point.spin_density[0], 0.0); + EXPECT_GE(point.spin_density[1], 0.0); + EXPECT_NEAR(point.spin_density[0] + point.spin_density[1], std::abs(states[sample][0]), 1.0e-15); + EXPECT_NEAR(point.spin_density[0] - point.spin_density[1], point.clipped_magnitude, 1.0e-15); + + double maximum_error = 0.0; + for (int channel = 0; channel < 4; ++channel) + { + std::array plus_state = states[sample]; + std::array minus_state = states[sample]; + plus_state[channel] += step; + minus_state[channel] -= step; + const ModuleXC::NcggaSpinMapPoint plus = make_spin_map(plus_state, eta); + const ModuleXC::NcggaSpinMapPoint minus = make_spin_map(minus_state, eta); + for (int spin = 0; spin < 2; ++spin) + { + const double finite_difference = (plus.spin_density[spin] - minus.spin_density[spin]) / (2.0 * step); + maximum_error = std::max(maximum_error, std::abs(finite_difference - point.jacobian(spin, channel))); + EXPECT_NEAR(finite_difference, point.jacobian(spin, channel), 3.0e-9) + << "sample=" << sample << " spin=" << spin << " channel=" << channel; + } + } + std::cout << std::setprecision(17) << "NCGGA_SPIN_MAP_FD sample=" << sample + << " total_density=" << states[sample][0] << " absolute_density=" << point.absolute_density + << " radial_value=" << point.radial.value << " clipped_magnitude=" << point.clipped_magnitude + << " saturated=" << point.saturated << " step=" << step << " maximum_error=" << maximum_error << '\n'; + } +} + +TEST(NcggaRadial, SpinDensityMapDefinesAbsAndSaturationKinkConventions) +{ + const double eta = 0.1; + const ModuleXC::NcggaRadialPoint nonzero_radial = ModuleXC::make_ncgga_radial_point({{0.30, 0.0, 0.0}}, eta); + const ModuleXC::NcggaSpinMapPoint zero_density = ModuleXC::make_ncgga_spin_map_point(0.0, nonzero_radial); + EXPECT_TRUE(zero_density.saturated); + EXPECT_DOUBLE_EQ(zero_density.spin_density[0], 0.0); + EXPECT_DOUBLE_EQ(zero_density.spin_density[1], 0.0); + for (int spin = 0; spin < 2; ++spin) + { + for (int channel = 0; channel < 4; ++channel) + { + EXPECT_DOUBLE_EQ(zero_density.jacobian(spin, channel), 0.0); + } + } + + const ModuleXC::NcggaRadialPoint equality_radial = ModuleXC::make_ncgga_radial_point({{0.50, 0.0, 0.0}}, eta); + const ModuleXC::NcggaSpinMapPoint positive = ModuleXC::make_ncgga_spin_map_point(0.50, equality_radial); + const ModuleXC::NcggaSpinMapPoint negative = ModuleXC::make_ncgga_spin_map_point(-0.50, equality_radial); + EXPECT_TRUE(positive.saturated); + EXPECT_TRUE(negative.saturated); + EXPECT_DOUBLE_EQ(positive.jacobian(0, 0), 1.0); + EXPECT_DOUBLE_EQ(positive.jacobian(1, 0), 0.0); + EXPECT_DOUBLE_EQ(negative.jacobian(0, 0), -1.0); + EXPECT_DOUBLE_EQ(negative.jacobian(1, 0), 0.0); + for (int spin = 0; spin < 2; ++spin) + { + for (int channel = 1; channel < 4; ++channel) + { + EXPECT_DOUBLE_EQ(positive.jacobian(spin, channel), 0.0); + EXPECT_DOUBLE_EQ(negative.jacobian(spin, channel), 0.0); + } + } +} + +} // namespace diff --git a/source/source_hamilt/module_xc/xc_functional.h b/source/source_hamilt/module_xc/xc_functional.h index 4dce5e3e99f..81ff40e27e9 100644 --- a/source/source_hamilt/module_xc/xc_functional.h +++ b/source/source_hamilt/module_xc/xc_functional.h @@ -72,6 +72,7 @@ class XC_Functional const int nspin, const bool domag, const bool domag_z, + const int gga_grad, const double hybrid_alpha, const double hse_omega); @@ -93,6 +94,14 @@ class XC_Functional return func_type; }; + /// @brief Set the functional class directly. set_xc_type() derives this from + /// a functional name and is what production code uses; this is for + /// callers that need to select a branch without configuring libxc. + static void set_func_type(const int func_type_in) + { + func_type = func_type_in; + }; + static void set_xc_type(const std::string xc_func_in); // For hybrid functional @@ -122,6 +131,12 @@ class XC_Functional return ked_flag; }; + /// @brief Set whether the functional needs the kinetic energy density. + static void set_ked_flag(const bool ked_flag_in) + { + ked_flag = ked_flag_in; + }; + static bool get_need_laplacian() { return need_laplacian; @@ -262,6 +277,7 @@ class XC_Functional const int nspin, const bool domag, const bool domag_z, + const int gga_grad, const double hybrid_alpha, const double hse_omega); diff --git a/source/source_hamilt/module_xc/xc_functional_ncgga_sf.cpp b/source/source_hamilt/module_xc/xc_functional_ncgga_sf.cpp new file mode 100644 index 00000000000..92808791ca9 --- /dev/null +++ b/source/source_hamilt/module_xc/xc_functional_ncgga_sf.cpp @@ -0,0 +1,347 @@ +#include "xc_functional_ncgga_sf.h" + +#include "source_base/parallel_reduce.h" +#include "source_base/timer.h" +#include "source_base/vector3.h" +#include "source_basis/module_pw/pw_basis.h" +#include "source_estate/module_charge/charge.h" +#include "xc_functional.h" +#include "xc_ncgga_radial.h" + +#include +#include +#include + +namespace ModuleXC +{ +namespace NCGGA_SF_Builtin +{ + +std::tuple v_xc_ncgga_sf_builtin(const int& nrxx, + const double& omega, + const double tpiba, + const Charge* const chr) +{ + ModuleBase::TITLE("XC_Functional", "v_xc_ncgga_sf_builtin"); + ModuleBase::timer::start("XC_Functional", "v_xc_ncgga_sf_builtin"); + + // Regularized projected local-collinear energy: rho_s = N_s(n + rho_core, m), + // g_s = sum_A J_sA G_h x_A. Reverse the same discrete graph, including + // both -D_h(sum_s J_sA h_s) and the local Hessian response of N_s. + ModulePW::PW_Basis* rhopw = chr->rhopw; + const int npw = rhopw->npw; + const double e2 = ModuleBase::e2; + constexpr double vanishing = 1e-10; + constexpr double epsr = 1e-6; + const bool is_gga = (XC_Functional::get_func_type() == 2 || XC_Functional::get_func_type() == 4); + + std::vector rhotmp1(nrxx); + std::vector rhotmp2(nrxx); + std::vector spin_map(nrxx); + + for (int ir = 0; ir < nrxx; ++ir) + { + const double mx = chr->rho[1][ir]; + const double my = chr->rho[2][ir]; + const double mz = chr->rho[3][ir]; + + const std::array magnetization = {{mx, my, mz}}; + const NcggaRadialPoint radial = make_ncgga_radial_point(magnetization, ncgga_lca_radial_eta()); + spin_map[ir] = make_ncgga_spin_map_point(chr->rho[0][ir] + chr->rho_core[ir], radial); + rhotmp1[ir] = spin_map[ir].spin_density[0]; + rhotmp2[ir] = spin_map[ir].spin_density[1]; + } + + std::vector> rhogsum1(npw); + std::vector> tmp_recip(npw); + rhopw->real2recip(chr->rho[0], rhogsum1.data()); + for (int ig = 0; ig < npw; ++ig) + rhogsum1[ig] += chr->rhog_core[ig]; + + std::vector> gdr1(nrxx); + std::vector> gdr2(nrxx); + std::vector> grad_rho(nrxx); + std::array>, 3> grad_m; + for (int mu = 0; mu < 3; ++mu) + { + grad_m[mu].resize(nrxx); + } + std::vector> gdr_mag(nrxx); + XC_Functional::grad_rho(rhogsum1.data(), gdr1.data(), rhopw, tpiba); + + for (int ir = 0; ir < nrxx; ++ir) + { + grad_rho[ir] = gdr1[ir]; + + gdr1[ir] = spin_map[ir].jacobian(0, 0) * grad_rho[ir]; + gdr2[ir] = spin_map[ir].jacobian(1, 0) * grad_rho[ir]; + } + for (int is = 1; is <= 3; ++is) + { + rhopw->real2recip(chr->rho[is], tmp_recip.data()); + XC_Functional::grad_rho(tmp_recip.data(), gdr_mag.data(), rhopw, tpiba); + grad_m[is - 1] = gdr_mag; + for (int ir = 0; ir < nrxx; ++ir) + { + + gdr1[ir] += spin_map[ir].jacobian(0, is) * gdr_mag[ir]; + gdr2[ir] += spin_map[ir].jacobian(1, is) * gdr_mag[ir]; + } + } + + double etxc = 0; + double vtxc = 0; + ModuleBase::matrix v(4, nrxx); + + for (int ir = 0; ir < nrxx; ++ir) + { + const double arho = spin_map[ir].absolute_density; + if (arho <= vanishing) + continue; + + double zeta = spin_map[ir].clipped_magnitude / arho; + if (std::abs(zeta) > 1.0) + zeta = (zeta > 0) ? 1.0 : -1.0; + double exc = 0; + double vxc[2] = {0, 0}; + XC_Functional::xc_spin(arho, zeta, exc, vxc[0], vxc[1]); + + for (int channel = 0; channel < 4; ++channel) + { + v(channel, ir) + = e2 * (spin_map[ir].jacobian(0, channel) * vxc[0] + spin_map[ir].jacobian(1, channel) * vxc[1]); + } + + etxc += e2 * exc * arho; + } + + // Step 4: GGA contribution and variational divergence correction. + if (is_gga) + { + double etxcgc = 0; + std::vector vup_gga(nrxx, 0); + std::vector vdw_gga(nrxx, 0); + std::vector> h1(nrxx); + std::vector> h2(nrxx); + for (int ir = 0; ir < nrxx; ++ir) + { + double sx = 0; + double v1xup = 0; + double v1xdw = 0; + double v2xup = 0; + double v2xdw = 0; + double sc = 0; + double v1cup = 0; + double v1cdw = 0; + double v2c = 0; + double grho2a = gdr1[ir] * gdr1[ir]; + double grho2b = gdr2[ir] * gdr2[ir]; + + const double rh = rhotmp1[ir] + rhotmp2[ir]; + + XC_Functional::gcx_spin(rhotmp1[ir], rhotmp2[ir], grho2a, grho2b, sx, v1xup, v1xdw, v2xup, v2xdw); + + if (rh > epsr) + { + const double zeta_input = std::fabs((rhotmp1[ir] - rhotmp2[ir]) / rh); + double zeta = zeta_input; + const double grh2 = (gdr1[ir] + gdr2[ir]) * (gdr1[ir] + gdr2[ir]); + XC_Functional::gcc_spin(rh, zeta, grh2, sc, v1cup, v1cdw, v2c); + if (zeta_input > 1.0 - epsr) + { + // gcc_spin evaluates this branch at a fixed clipped zeta. + // Reverse that actual branch instead of differentiating + // through the discarded input polarization. + const double fixed_zeta_density_derivative = 0.5 * ((1.0 + zeta) * v1cup + (1.0 - zeta) * v1cdw); + v1cup = fixed_zeta_density_derivative; + v1cdw = fixed_zeta_density_derivative; + } + } + + vup_gga[ir] = e2 * (v1xup + v1cup); + vdw_gga[ir] = e2 * (v1xdw + v1cdw); + + const double v2cup = v2c; + const double v2cdw = v2c; + const double v2cud = v2c; + h1[ir] = e2 * ((v2xup + v2cup) * gdr1[ir] + v2cud * gdr2[ir]); + h2[ir] = e2 * ((v2xdw + v2cdw) * gdr2[ir] + v2cud * gdr1[ir]); + + etxcgc += e2 * (sx + sc); + } + + for (int ir = 0; ir < nrxx; ++ir) + { + + for (int channel = 0; channel < 4; ++channel) + { + v(channel, ir) += spin_map[ir].jacobian(0, channel) * vup_gga[ir] + + spin_map[ir].jacobian(1, channel) * vdw_gga[ir]; + } + } + + std::vector dh(nrxx); + std::vector> tmp_h(nrxx); + + // Exact reverse of g_s=sum_A J_sA G_h(x_A): + // v_B = -D_h(sum_s J_sB h_s) + // + sum_s,A dJ_sA/dx_B h_s.G_h(x_A). + for (int channel = 0; channel < 4; ++channel) + { + for (int ir = 0; ir < nrxx; ++ir) + { + tmp_h[ir] = spin_map[ir].jacobian(0, channel) * h1[ir] + spin_map[ir].jacobian(1, channel) * h2[ir]; + } + XC_Functional::grad_dot(tmp_h.data(), dh.data(), rhopw, tpiba); + for (int ir = 0; ir < nrxx; ++ir) + { + v(channel, ir) -= dh[ir]; + if (channel == 0 || spin_map[ir].saturated) + { + continue; + } + const ModuleBase::Vector3 spin_flux = 0.5 * (h1[ir] - h2[ir]); + double local_response = 0.0; + for (int nu = 0; nu < 3; ++nu) + { + local_response += spin_map[ir].radial.jacobian(nu, channel - 1) * (spin_flux * grad_m[nu][ir]); + } + v(channel, ir) += local_response; + } + } + + etxc += etxcgc; + } + + // vtxc uses the same completed four-component potential returned to the + // caller. This unifies the bookkeeping for both modes. + vtxc = 0.0; + for (int ir = 0; ir < nrxx; ++ir) + { + for (int is = 0; is < 4; ++is) + { + vtxc += v(is, ir) * chr->rho[is][ir]; + } + } + +#ifdef __MPI + Parallel_Reduce::reduce_pool(etxc); + Parallel_Reduce::reduce_pool(vtxc); +#endif + etxc *= omega / rhopw->nxyz; + vtxc *= omega / rhopw->nxyz; + + ModuleBase::timer::end("XC_Functional", "v_xc_ncgga_sf_builtin"); + return std::make_tuple(etxc, vtxc, std::move(v)); +} + +void gradcorr_ncgga_lca_builtin(const Charge* const chr, + ModulePW::PW_Basis* rhopw, + const double tpiba, + std::vector& stress_gga) +{ + stress_gga.assign(9, 0.0); + + const int nrxx = rhopw->nrxx; + const int npw = rhopw->npw; + const double e2 = ModuleBase::e2; + constexpr double epsr = 1.0e-6; + + // Rebuild the same complete local map used by the gga_grad=2 energy: + // rho_s = N_s(n + rho_core, m), + // g_s = sum_A dN_s/dx_A G_h x_A. + // Metric differentiation keeps the real-grid values x_A fixed, so the + // map Jacobian is unchanged and every G_h x_A transforms covariantly. + std::vector spin_map(nrxx); + std::array>, 4> field_gradient; + for (int channel = 0; channel < 4; ++channel) + { + field_gradient[channel].resize(nrxx); + } + std::array>, 2> spin_gradient; + for (int spin = 0; spin < 2; ++spin) + { + spin_gradient[spin].resize(nrxx); + } + + std::vector> reciprocal(npw); + rhopw->real2recip(chr->rho[0], reciprocal.data()); + for (int ig = 0; ig < npw; ++ig) + { + reciprocal[ig] += chr->rhog_core[ig]; + } + XC_Functional::grad_rho(reciprocal.data(), field_gradient[0].data(), rhopw, tpiba); + + for (int channel = 1; channel < 4; ++channel) + { + rhopw->real2recip(chr->rho[channel], reciprocal.data()); + XC_Functional::grad_rho(reciprocal.data(), field_gradient[channel].data(), rhopw, tpiba); + } + + for (int ir = 0; ir < nrxx; ++ir) + { + const std::array magnetization = {{chr->rho[1][ir], chr->rho[2][ir], chr->rho[3][ir]}}; + spin_map[ir] = make_ncgga_spin_map_point(chr->rho[0][ir] + chr->rho_core[ir], + make_ncgga_radial_point(magnetization, ncgga_lca_radial_eta())); + for (int spin = 0; spin < 2; ++spin) + { + for (int channel = 0; channel < 4; ++channel) + { + spin_gradient[spin][ir] += spin_map[ir].jacobian(spin, channel) * field_gradient[channel][ir]; + } + } + } + + for (int ir = 0; ir < nrxx; ++ir) + { + const double rho_up = spin_map[ir].spin_density[0]; + const double rho_down = spin_map[ir].spin_density[1]; + const ModuleBase::Vector3& grad_up = spin_gradient[0][ir]; + const ModuleBase::Vector3& grad_down = spin_gradient[1][ir]; + + double sx = 0.0; + double v1xup = 0.0; + double v1xdw = 0.0; + double v2xup = 0.0; + double v2xdw = 0.0; + XC_Functional::gcx_spin(rho_up, + rho_down, + grad_up * grad_up, + grad_down * grad_down, + sx, + v1xup, + v1xdw, + v2xup, + v2xdw); + + double sc = 0.0; + double v1cup = 0.0; + double v1cdw = 0.0; + double v2c = 0.0; + const double rho = rho_up + rho_down; + if (rho > epsr) + { + double zeta = std::fabs((rho_up - rho_down) / rho); + const ModuleBase::Vector3 grad_rho = grad_up + grad_down; + XC_Functional::gcc_spin(rho, zeta, grad_rho * grad_rho, sc, v1cup, v1cdw, v2c); + } + + const ModuleBase::Vector3 h_up = e2 * ((v2xup + v2c) * grad_up + v2c * grad_down); + const ModuleBase::Vector3 h_down = e2 * ((v2xdw + v2c) * grad_down + v2c * grad_up); + const double grad_up_component[3] = {grad_up.x, grad_up.y, grad_up.z}; + const double grad_down_component[3] = {grad_down.x, grad_down.y, grad_down.z}; + const double h_up_component[3] = {h_up.x, h_up.y, h_up.z}; + const double h_down_component[3] = {h_down.x, h_down.y, h_down.z}; + for (int row = 0; row < 3; ++row) + { + for (int column = 0; column <= row; ++column) + { + stress_gga[row * 3 + column] += h_up_component[row] * grad_up_component[column] + + h_down_component[row] * grad_down_component[column]; + } + } + } +} + +} // namespace NCGGA_SF_Builtin +} // namespace ModuleXC diff --git a/source/source_hamilt/module_xc/xc_functional_ncgga_sf.h b/source/source_hamilt/module_xc/xc_functional_ncgga_sf.h new file mode 100644 index 00000000000..c4d71a28ad7 --- /dev/null +++ b/source/source_hamilt/module_xc/xc_functional_ncgga_sf.h @@ -0,0 +1,37 @@ +#ifndef XC_FUNCTIONAL_NCGGA_SF_H +#define XC_FUNCTIONAL_NCGGA_SF_H + +#include "source_base/matrix.h" + +#include +#include + +class Charge; +namespace ModulePW +{ +class PW_Basis; +} + +namespace ModuleXC +{ +namespace NCGGA_SF_Builtin +{ + +// Exact discrete reverse of the regularized projected LCA graph (gga_grad=2). +std::tuple v_xc_ncgga_sf_builtin(const int& nrxx, + const double& omega, + const double tpiba, + const Charge* const chr); + +// Gradient-metric stress of the exact gga_grad=2 projected-LCA graph. The +// returned lower triangle is the unnormalised real-grid sum; Stress_Func +// applies the existing pool reduction and 1/nxyz normalisation. +void gradcorr_ncgga_lca_builtin(const Charge* const chr, + ModulePW::PW_Basis* rhopw, + const double tpiba, + std::vector& stress_gga); + +} // namespace NCGGA_SF_Builtin +} // namespace ModuleXC + +#endif diff --git a/source/source_hamilt/module_xc/xc_gga_corr.cpp b/source/source_hamilt/module_xc/xc_gga_corr.cpp index b22f31ef64a..c34f0e11be0 100644 --- a/source/source_hamilt/module_xc/xc_gga_corr.cpp +++ b/source/source_hamilt/module_xc/xc_gga_corr.cpp @@ -237,9 +237,9 @@ void XC_Functional::perdew86_spin( //SdG: in the original paper 1.745*0.11=0.19195 is used double dd = pow((2.0) , third) * sqrt(pow(((1.0 + zeta) * 0.50) , (5.0 / 3.0)) - + pow(((1.0 - zeta) * 0.50) , (5.0 / 3.0))); + + pow(((1.0 - zeta) * 0.50) , (5.0 / 3.0))); double ddd = pow((2.0) , (- 4.0 / 3.0)) * 5.0 * (pow(((1.0 + zeta) * 0.50) , (2.0 / 3.0)) - - pow(((1.0 - zeta) * 0.50) , (2.0 / 3.0))) / (3.0 * dd); + - pow(((1.0 - zeta) * 0.50) , (2.0 / 3.0))) / (3.0 * dd); double ephi = exp(- phi); sc = grho / rho43 * cn * ephi / dd; v1cup = sc * ((1.0 + phi) * dcn / cn - ((4.0 / 3.0) - @@ -401,9 +401,9 @@ void XC_Functional::pbec_spin( double dh0up = be[iflag] * t * t * fz3 / s1 * (- 7.0 / 3.0 * xy - qy * (af * bfup / be[iflag] - 7.0 / 3.0)); double dh0dw = be[iflag] * t * t * fz3 / s1 * (- 7.0 / 3.0 * xy - qy * (af * bfdw / be[iflag] - 7.0 / 3.0)); double dh0zup = (3.0 * h0 / fz - be[iflag] * t * t * fz2 - / s1 * (2.0 * xy - qy * (3.0 * af * expe * ec / fz3 / be[iflag] + 2.0))) * dfz * (1.0 - zeta); + / s1 * (2.0 * xy - qy * (3.0 * af * expe * ec / fz3 / be[iflag] + 2.0))) * dfz * (1.0 - zeta); double dh0zdw = - (3.0 * h0 / fz - be[iflag] * t * t * fz2 - / s1 * (2.0 * xy - qy * (3.0 * af * expe * ec / fz3 / be[iflag] + 2.0))) * dfz * (1.0 + zeta); + / s1 * (2.0 * xy - qy * (3.0 * af * expe * ec / fz3 / be[iflag] + 2.0))) * dfz * (1.0 + zeta); double ddh0 = be[iflag] * fz / (2.0 * ks * ks * rho) * (xy - qy) / s1; sc = rho * h0; v1cup = h0 + dh0up + dh0zup; diff --git a/source/source_hamilt/module_xc/xc_grad.cpp b/source/source_hamilt/module_xc/xc_grad.cpp index c7bf9a20136..2df043824a8 100644 --- a/source/source_hamilt/module_xc/xc_grad.cpp +++ b/source/source_hamilt/module_xc/xc_grad.cpp @@ -14,11 +14,13 @@ // noncolin_rho. #include "xc_functional.h" +#include "xc_functional_ncgga_sf.h" #include "xc_grad_internal.h" #include "source_base/timer.h" #ifdef __LIBXC #include +#include "libxc_abacus.h" #endif void XC_Functional::gradcorr( @@ -33,6 +35,7 @@ void XC_Functional::gradcorr( const int nspin, const bool domag, const bool domag_z, + const int gga_grad, const double hybrid_alpha_in, const double hse_omega_in) { @@ -63,6 +66,32 @@ void XC_Functional::gradcorr( return; } + if (is_stress && !use_libxc && nspin == 4 && (domag || domag_z) && gga_grad == 2) + { + ModuleXC::NCGGA_SF_Builtin::gradcorr_ncgga_lca_builtin( + chr, rhopw, ucell->tpiba, stress_gga); + return; + } + + + +#ifdef __LIBXC + if (is_stress && use_libxc && nspin == 4 && (domag || domag_z) + && gga_grad == 2) + { + XC_Functional_Libxc::gradcorr_ncgga_sf_libxc( + func_id, + rhopw->nrxx, + ucell->tpiba, + chr, + &scaling_factor_xc, + hybrid_alpha_in, + hse_omega_in, + stress_gga); + return; + } +#endif + bool igcc_is_lyp = false; // func_id may hold a single entry (e.g. PBE0 -> {XC_HYB_GGA_XC_PBEH}), so guard the index. if( func_id.size() > 1 && func_id[1] == XC_GGA_C_LYP) @@ -108,6 +137,7 @@ void XC_Functional::gradcorr( params.igcc_is_lyp = igcc_is_lyp; params.domag = domag; params.domag_z = domag_z; + params.gga_grad = gga_grad; params.hybrid_alpha = hybrid_alpha_in; params.hse_omega = hse_omega_in; params.use_libxc = use_libxc; diff --git a/source/source_hamilt/module_xc/xc_grad_internal.h b/source/source_hamilt/module_xc/xc_grad_internal.h index 8d0e54abedd..76d27832a89 100644 --- a/source/source_hamilt/module_xc/xc_grad_internal.h +++ b/source/source_hamilt/module_xc/xc_grad_internal.h @@ -41,6 +41,7 @@ struct GradCorrParams bool igcc_is_lyp; bool domag; bool domag_z; + int gga_grad; double hybrid_alpha; double hse_omega; bool use_libxc; diff --git a/source/source_hamilt/module_xc/xc_grad_prepare.cpp b/source/source_hamilt/module_xc/xc_grad_prepare.cpp index 756e7c9ff2e..fe9187f18c9 100644 --- a/source/source_hamilt/module_xc/xc_grad_prepare.cpp +++ b/source/source_hamilt/module_xc/xc_grad_prepare.cpp @@ -150,7 +150,17 @@ void gradcorr_prepare_rho( } } } - XC_Functional::noncolin_rho(buf.rhotmp1.data(), buf.rhotmp2.data(), buf.neg.data(), chr->rho, rhopw->nrxx, ucell->magnet.ux_, ucell->magnet.lsign_); + // Mode 1 ignores the global quantization axis to remain continuous + // when the magnetic moments tilt away from a collinear state. + // use_global_axis selects the up/down projection inside noncolin_rho: + // - true (lsign_ set, i.e. STRU initial moments defined a non-zero + // axis via cal_ux): sign follows m(r).ux_ point by point; + // - false (lsign_ unset / Mode 1): neg = +1 everywhere, a + // direction-blind |m| projection. + // This is why two otherwise identical runs can follow different SCF + // paths depending solely on whether STRU carries an initial moment. + const bool use_global_axis = ucell->magnet.lsign_ && params.gga_grad != 1; + XC_Functional::noncolin_rho(buf.rhotmp1.data(), buf.rhotmp2.data(), buf.neg.data(), chr->rho, rhopw->nrxx, ucell->magnet.ux_, use_global_axis); rhopw->real2recip(buf.rhotmp1.data(), buf.rhogsum1.data()); rhopw->real2recip(buf.rhotmp2.data(), buf.rhogsum2.data()); #ifdef _OPENMP diff --git a/source/source_hamilt/module_xc/xc_grad_utils.cpp b/source/source_hamilt/module_xc/xc_grad_utils.cpp index 2f14e600886..6fe1dac62ca 100644 --- a/source/source_hamilt/module_xc/xc_grad_utils.cpp +++ b/source/source_hamilt/module_xc/xc_grad_utils.cpp @@ -164,6 +164,16 @@ void XC_Functional::noncolin_rho( } } } + // Note: with lsign_ = true the up/down channels are defined w.r.t. the + // global axis ux_ through sign(m . ux_), evaluated point by point above; + // with lsign_ = false, "up" is always the local |m| (neg stays +1 + // everywhere). amag = |m(r)| is non-negative; neg[ir] carries that sign. + // The global axis ux_ itself is derived from the STRU initial moments in + // cal_ux(). Because GGA also consumes the *gradients* of rhoout1/rhoout2, + // a flipping sign field changes the gradient terms too, so the two + // branches can give noticeably different v_xc even for predominantly + // single-sign mz. (Locked by GgaGradTools.NoncolinRhoGlobalAxis in + // module_xc/test/test_xc5.cpp.) #ifdef _OPENMP #pragma omp parallel for #endif diff --git a/source/source_hamilt/module_xc/xc_ncgga_radial.cpp b/source/source_hamilt/module_xc/xc_ncgga_radial.cpp new file mode 100644 index 00000000000..d01f4159b09 --- /dev/null +++ b/source/source_hamilt/module_xc/xc_ncgga_radial.cpp @@ -0,0 +1,95 @@ +#include "xc_ncgga_radial.h" + +#include +#include +#include + +namespace ModuleXC +{ + +double NcggaRadialPoint::jacobian(const int row, const int column) const +{ + const double identity = (row == column) ? 1.0 : 0.0; + return transverse_hessian * identity + (radial_hessian - transverse_hessian) * direction[row] * direction[column]; +} + +NcggaRadialPoint make_ncgga_radial_point(const std::array& magnetization, const double eta) +{ + if (!(eta > 0.0)) + { + throw std::invalid_argument("noncollinear GGA radial eta must be positive"); + } + + NcggaRadialPoint point; + const double magnitude = std::sqrt(magnetization[0] * magnetization[0] + magnetization[1] * magnetization[1] + + magnetization[2] * magnetization[2]); + if (magnitude == 0.0) + { + return point; + } + + for (int component = 0; component < 3; ++component) + { + point.direction[component] = magnetization[component] / magnitude; + } + + if (magnitude < eta) + { + const double x = magnitude / eta; + const double x2 = x * x; + const double x3 = x2 * x; + point.value = eta * x3 * (3.0 * x2 - 8.0 * x + 6.0); + point.transverse_hessian = x * (15.0 * x2 - 32.0 * x + 18.0) / eta; + point.radial_hessian = x * (60.0 * x2 - 96.0 * x + 36.0) / eta; + } + else + { + point.value = magnitude; + point.transverse_hessian = 1.0 / magnitude; + point.radial_hessian = 0.0; + } + + for (int component = 0; component < 3; ++component) + { + point.gradient[component] = point.transverse_hessian * magnetization[component]; + } + return point; +} + +double ncgga_lca_radial_eta() +{ + return 1.0e-3; +} + +double NcggaSpinMapPoint::jacobian(const int spin, const int channel) const +{ + if (channel == 0) + { + if (saturated) + { + return spin == 0 ? density_sign : 0.0; + } + return 0.5 * density_sign; + } + if (saturated) + { + return 0.0; + } + const double spin_sign = spin == 0 ? 0.5 : -0.5; + return spin_sign * radial.gradient[channel - 1]; +} + +NcggaSpinMapPoint make_ncgga_spin_map_point(const double total_density, const NcggaRadialPoint& radial) +{ + NcggaSpinMapPoint point; + point.radial = radial; + point.absolute_density = std::abs(total_density); + point.clipped_magnitude = std::min(radial.value, point.absolute_density); + point.spin_density[0] = 0.5 * (point.absolute_density + point.clipped_magnitude); + point.spin_density[1] = 0.5 * (point.absolute_density - point.clipped_magnitude); + point.density_sign = total_density > 0.0 ? 1.0 : total_density < 0.0 ? -1.0 : 0.0; + point.saturated = !(radial.value < point.absolute_density); + return point; +} + +} // namespace ModuleXC diff --git a/source/source_hamilt/module_xc/xc_ncgga_radial.h b/source/source_hamilt/module_xc/xc_ncgga_radial.h new file mode 100644 index 00000000000..13da072ed22 --- /dev/null +++ b/source/source_hamilt/module_xc/xc_ncgga_radial.h @@ -0,0 +1,60 @@ +#ifndef XC_NCGGA_RADIAL_H +#define XC_NCGGA_RADIAL_H + +#include + +namespace ModuleXC +{ + +struct NcggaRadialPoint +{ + // Value, gradient, and Hessian of the same radial scalar map. At zero, + // direction is represented by the zero vector because the scalar map has + // a unique zero gradient and zero Hessian there. + double value = 0.0; + std::array direction = {{0.0, 0.0, 0.0}}; + std::array gradient = {{0.0, 0.0, 0.0}}; + double transverse_hessian = 0.0; + double radial_hessian = 0.0; + + double jacobian(const int row, const int column) const; +}; + +// For r = |magnetization| and x = r / eta, the returned scalar is +// eta * x^3 * (3 x^2 - 8 x + 6), r < eta, +// r, r >= eta. +// The splice is C2 at both r = 0 and r = eta. Eta is explicit so this +// mathematical primitive does not choose policy for any XC mode. +NcggaRadialPoint make_ncgga_radial_point(const std::array& magnetization, const double eta); + +// The C2 regularization scale is part of the gga_grad=2 LCA functional, not a +// divide-by-zero guard. Other noncollinear modes do not inherit this policy. +double ncgga_lca_radial_eta(); + +struct NcggaSpinMapPoint +{ + NcggaRadialPoint radial; + double absolute_density = 0.0; + double clipped_magnitude = 0.0; + std::array spin_density = {{0.0, 0.0}}; + double density_sign = 0.0; + bool saturated = true; + + // spin is 0/1 for up/down; channel is 0 for the raw total density and + // 1..3 for mx,my,mz. + double jacobian(const int spin, const int channel) const; +}; + +// Compose a raw total density t=n+rho_core with one radial magnetization map: +// a = |t|, +// c = min(radial.value, a), +// rho_up = (a+c)/2, +// rho_down= (a-c)/2. +// The Jacobian follows that exact graph. At the abs kink t=0 it selects zero; +// at the clipping kink radial.value=a it selects the saturated branch. Eta and +// the radial policy remain explicit in make_ncgga_radial_point. +NcggaSpinMapPoint make_ncgga_spin_map_point(const double total_density, const NcggaRadialPoint& radial); + +} // namespace ModuleXC + +#endif diff --git a/source/source_hamilt/module_xc/xc_pot.cpp b/source/source_hamilt/module_xc/xc_pot.cpp index dc6a13dabdf..f8b54bde2dd 100644 --- a/source/source_hamilt/module_xc/xc_pot.cpp +++ b/source/source_hamilt/module_xc/xc_pot.cpp @@ -8,6 +8,8 @@ #include "source_base/timer.h" #include "xc_functional.h" +#include "xc_functional_ncgga_sf.h" + #ifdef __LIBXC #include "libxc_abacus.h" #ifdef __EXX @@ -15,6 +17,7 @@ #endif #endif + // [etxc, vtxc, v] = XC_Functional::v_xc(...) std::tuple XC_Functional::v_xc( const int& nrxx, @@ -23,11 +26,13 @@ std::tuple XC_Functional::v_xc( const int nspin, const bool domag, const bool domag_z, + const int gga_grad, const double hybrid_alpha, const double hse_omega) { ModuleBase::TITLE("XC_Functional", "v_xc"); + if (use_libxc) { #ifdef __LIBXC @@ -39,6 +44,7 @@ std::tuple XC_Functional::v_xc( nspin, domag, domag_z, + gga_grad, &(scaling_factor_xc), hybrid_alpha, hse_omega); @@ -47,6 +53,12 @@ std::tuple XC_Functional::v_xc( #endif } + // Evaluate the regularized projected local-collinear graph. + if (nspin == 4 && (domag || domag_z) && gga_grad == 2) + { + return ModuleXC::NCGGA_SF_Builtin::v_xc_ncgga_sf_builtin(nrxx, ucell->omega, ucell->tpiba, chr); + } + ModuleBase::timer::start("XC_Functional", "v_xc"); //Exchange-Correlation potential Vxc(r) from n(r) @@ -183,7 +195,7 @@ std::tuple XC_Functional::v_xc( // the dummy variable dum contains gradient correction to stress // which is not used here std::vector dum; - gradcorr(etxc, vtxc, v, chr, chr->rhopw, ucell, dum, false, nspin, domag, domag_z, hybrid_alpha, hse_omega); + gradcorr(etxc, vtxc, v, chr, chr->rhopw, ucell, dum, false, nspin, domag, domag_z, gga_grad, hybrid_alpha, hse_omega); // parallel code : collect vtxc,etxc // mohan add 2008-06-01 diff --git a/source/source_hamilt/operator.cpp b/source/source_hamilt/operator.cpp index dba9367f055..36bc9c804cf 100644 --- a/source/source_hamilt/operator.cpp +++ b/source/source_hamilt/operator.cpp @@ -80,21 +80,21 @@ typename Operator::hpsi_info Operator::hPsi(hpsi_info& inp psi_input->get_nbasis(), true); - switch (op->get_act_type()) - { - case 2: - op->act(psi_wrapper, *this->hpsi, nbands); - break; - default: - op->act(nbands, - psi_input->get_nbasis(), - psi_input->get_npol(), - tmpsi_in, - this->hpsi->get_pointer(), - psi_input->get_current_nbas(), - is_first_node); - break; - } + switch (op->get_act_type()) + { + case 2: + op->act(psi_wrapper, *this->hpsi, nbands); + break; + default: + op->act(nbands, + psi_input->get_nbasis(), + psi_input->get_npol(), + tmpsi_in, + this->hpsi->get_pointer(), + psi_input->get_current_nbas(), + is_first_node); + break; + } }; ModuleBase::timer::start("Operator", "hPsi"); diff --git a/source/source_hamilt/operator.h b/source/source_hamilt/operator.h index 73b669df545..8848fcc7bac 100644 --- a/source/source_hamilt/operator.h +++ b/source/source_hamilt/operator.h @@ -12,6 +12,7 @@ enum class calculation_type { no, pw_ekinetic, + pw_ekinetic_td, pw_nonlocal, pw_veff, pw_meta, @@ -99,10 +100,10 @@ class Operator /// developer-friendly interfaces for act() function /// interface type 2: input and change the Psi-type HPsi - // virtual void act(const psi::Psi& psi_in, psi::Psi& psi_out) const {}; - virtual void act(const psi::Psi& psi_in, - psi::Psi& psi_out, - const int nbands) const {}; + // virtual void act(const psi::Psi& psi_in, psi::Psi& psi_out) const {}; + virtual void act(const psi::Psi& psi_in, + psi::Psi& psi_out, + const int nbands) const {}; /// interface type 3: return a Psi-type HPsi // virtual psi::Psi act(const psi::Psi& psi_in) const { return psi_in; }; diff --git a/source/source_hsolver/CMakeLists.txt b/source/source_hsolver/CMakeLists.txt index a9f1f9142ec..e7adc3fb5ad 100644 --- a/source/source_hsolver/CMakeLists.txt +++ b/source/source_hsolver/CMakeLists.txt @@ -6,14 +6,23 @@ list(APPEND objects diago_bpcg.cpp para_lin_tf.cpp hsolver_pw.cpp + hsolver_pw_tddft.cpp hsolver_lcaopw.cpp - hsolver_pw_sdft.cpp diago_iter_assist.cpp hsolver.cpp diago_pxxxgvx.cpp diag_hs_para.cpp diago_params.cpp - + linear_bicgstab.cpp + linear_cgs.cpp + linear_gmres.cpp + linear_algebra.cpp + orthonormal.cpp + orthonormal_small.cpp + cn_subspace.cpp + linear_low_rank.cpp + linear_workspace.cpp + hsolver_linear.cpp ) if(ENABLE_LCAO) diff --git a/source/source_hsolver/cn_subspace.cpp b/source/source_hsolver/cn_subspace.cpp new file mode 100644 index 00000000000..f11b661836a --- /dev/null +++ b/source/source_hsolver/cn_subspace.cpp @@ -0,0 +1,50 @@ +#include "source_hsolver/cn_subspace.h" + +#include "source_hsolver/kernels/linear_op.h" +#include "source_hsolver/linear_workspace.h" + +namespace hsolver +{ +template +bool CNSubspace::prepare(LinearAlgebra& algebra, int ld, int dim, int bands, const T* u, const T* b) +{ + const int64_t elements = static_cast(ld) * bands; + linear_buffer(&image_, elements); + linear_buffer(&seed_, elements); + linear_buffer(&residual_, elements); + linear_op().batch(ld, dim, bands, image(), u, b, T(2), T(-1), nullptr, nullptr, nullptr); + const std::vector> projected = algebra.projection_cross(ld, dim, bands, bands, u, image()); + std::vector> c = algebra.projection_cross(ld, dim, bands, bands, u, b); + if (!factor_.factor(projected, bands) || !factor_.solve(&c, bands)) + { + return false; + } + for (const std::complex& value: c) + { + if (!std::isfinite(std::abs(T(value)))) + { + return false; + } + } + algebra.expand(ld, dim, bands, bands, u, c, seed(), T(0)); + algebra.expand(ld, dim, bands, bands, image(), c, residual(), T(0)); + linear_op().batch(ld, dim, bands, residual(), b, residual(), T(1), T(-1), nullptr, nullptr, nullptr); + const int stride = ld * bands; + const std::vector> norms = algebra.dots(ld, dim, bands, 1, stride, residual(), residual()); + for (const std::complex& value: norms) + { + if (!std::isfinite(linear_norm(value))) + { + return false; + } + } + return true; +} + +template class CNSubspace, base_device::DEVICE_CPU>; +template class CNSubspace, base_device::DEVICE_CPU>; +#if defined(__CUDA) || defined(__ROCM) +template class CNSubspace, base_device::DEVICE_GPU>; +template class CNSubspace, base_device::DEVICE_GPU>; +#endif +} // namespace hsolver diff --git a/source/source_hsolver/cn_subspace.h b/source/source_hsolver/cn_subspace.h new file mode 100644 index 00000000000..0e822b7c3d2 --- /dev/null +++ b/source/source_hsolver/cn_subspace.h @@ -0,0 +1,37 @@ +#ifndef HSOLVER_CN_SUBSPACE_H +#define HSOLVER_CN_SUBSPACE_H +#include "source_hsolver/linear_algebra.h" + +namespace hsolver +{ +/** @brief CN projection derived from A*U = 2*U - B without another Hamiltonian application. */ +template +class CNSubspace +{ + private: + ct::Tensor image_; + ct::Tensor seed_; + ct::Tensor residual_; + LinearSmallLU factor_; + + public: + bool prepare(LinearAlgebra& algebra, int ld, int dim, int bands, const T* u, const T* b); + T* image() + { + return image_.template data(); + } + T* seed() + { + return seed_.template data(); + } + T* residual() + { + return residual_.template data(); + } + const LinearSmallLU& factor() const + { + return factor_; + } +}; +} // namespace hsolver +#endif diff --git a/source/source_hsolver/diag_hs_para.cpp b/source/source_hsolver/diag_hs_para.cpp index 74dc8da33d6..13f1ea1a619 100644 --- a/source/source_hsolver/diag_hs_para.cpp +++ b/source/source_hsolver/diag_hs_para.cpp @@ -154,10 +154,6 @@ void diago_hs_para(T* h, // gather wfc Cpxgemr2d(lda, lda, wfc_2d.data(), 1, 1, para2d_local.desc, wfc, 1, 1, para2d_global.desc, para2d_local.blacs_ctxt); - - // free the context - Cblacs_gridexit(para2d_local.blacs_ctxt); - Cblacs_gridexit(para2d_global.blacs_ctxt); } // template instantiation diff --git a/source/source_hsolver/diago_bpcg.cpp b/source/source_hsolver/diago_bpcg.cpp index de4dfa09bfd..363dd5429cb 100644 --- a/source/source_hsolver/diago_bpcg.cpp +++ b/source/source_hsolver/diago_bpcg.cpp @@ -209,21 +209,21 @@ void DiagoBPCG::rotate_wf( template void DiagoBPCG::calc_hpsi_with_block( - const HPsiFunc& hpsi_func, - T *psi_in, + const HSOperator& op, + const T *psi_in, ct::Tensor& hpsi_out) { // calculate all-band hpsi - hpsi_func(psi_in, hpsi_out.data(), this->n_basis, this->n_band_l); + op.hpsi(psi_in, hpsi_out.data(), this->n_basis, this->n_band_l); } template void DiagoBPCG::calc_spsi_with_block( - const SPsiFunc& spsi_func, + const HSOperator& op, const T* psi_in, ct::Tensor& spsi_out) { - spsi_func(psi_in, spsi_out.data(), this->n_basis, this->n_band_l); + op.spsi(psi_in, spsi_out.data(), this->n_basis, this->n_band_l); } template @@ -244,8 +244,7 @@ void DiagoBPCG::diag_hsub( template void DiagoBPCG::calc_hsub_with_block( - const HPsiFunc& hpsi_func, - const SPsiFunc& spsi_func, + const HSOperator& op, T *psi_in, ct::Tensor& psi_out, ct::Tensor& hpsi_out, @@ -255,8 +254,8 @@ void DiagoBPCG::calc_hsub_with_block( ct::Tensor& eigenvalue_out) { // Apply the H operator to psi and obtain the hpsi matrix. - this->calc_hpsi_with_block(hpsi_func, psi_in, hpsi_out); - this->calc_spsi_with_block(spsi_func, psi_in, spsi_out); + this->calc_hpsi_with_block(op, psi_in, hpsi_out); + this->calc_spsi_with_block(op, psi_in, spsi_out); // Transform the generalized problem to an S-orthonormal subspace. this->orth_cholesky(workspace_in, psi_out, hpsi_out, spsi_out, hsub_out); @@ -293,8 +292,7 @@ void DiagoBPCG::calc_hsub_with_block_exit( } template -void DiagoBPCG::diag(const HPsiFunc& hpsi_func, - const SPsiFunc& spsi_func, +void DiagoBPCG::diag(const HSOperator& op, T* psi_in, Real* eigenvalue_in, const std::vector& ethr_band) @@ -307,8 +305,7 @@ void DiagoBPCG::diag(const HPsiFunc& hpsi_func, this->calc_prec(); // Improving the initial guess of the wave function psi through a subspace diagonalization. - this->calc_hsub_with_block(hpsi_func, - spsi_func, + this->calc_hsub_with_block(op, psi_in, this->psi, this->hpsi, @@ -338,7 +335,7 @@ void DiagoBPCG::diag(const HPsiFunc& hpsi_func, this->psi, this->hpsi, this->spsi, this->grad, this->grad_old); // Apply S before projecting the search directions in the generalized metric. - this->calc_spsi_with_block(spsi_func, this->grad.template data(), this->sgrad); + this->calc_spsi_with_block(op, this->grad.template data(), this->sgrad); // Orthogonalize column vectors g_i in matrix grad to column vectors p_j in matrix psi // for all 'j less or equal to i'. @@ -349,7 +346,7 @@ void DiagoBPCG::diag(const HPsiFunc& hpsi_func, syncmem_complex_op()(this->grad_old.template data(), this->grad.template data(), n_basis * n_band_l); // Calculate H|grad> matrix - this->calc_hpsi_with_block(hpsi_func, this->grad.template data(), /*this->grad_wrapper[0],*/ this->hgrad); + this->calc_hpsi_with_block(op, this->grad.template data(), /*this->grad_wrapper[0],*/ this->hgrad); // optimize psi as well as the hpsi // 1. normalize grad @@ -361,8 +358,7 @@ void DiagoBPCG::diag(const HPsiFunc& hpsi_func, this->orth_cholesky(this->work, this->psi, this->hpsi, this->spsi, this->hsub); if (current_scf_iter == 1 && ntry % this->nline == 0) { - this->calc_hsub_with_block(hpsi_func, - spsi_func, + this->calc_hsub_with_block(op, psi_in, this->psi, this->hpsi, diff --git a/source/source_hsolver/diago_bpcg.h b/source/source_hsolver/diago_bpcg.h index caad91e3e83..66489e795c5 100644 --- a/source/source_hsolver/diago_bpcg.h +++ b/source/source_hsolver/diago_bpcg.h @@ -1,12 +1,11 @@ #ifndef DIAGO_BPCG_H_ #define DIAGO_BPCG_H_ -#include - #include "source_base/kernels/math_kernel_op.h" #include "source_base/module_device/memory_op.h" #include "source_base/module_device/types.h" #include "source_base/para_gemm.h" +#include "source_hsolver/hs_operator.h" #include "source_hsolver/kernels/hegvd_op.h" #include "source_hsolver/para_lin_tf.h" @@ -57,23 +56,16 @@ class DiagoBPCG */ void init_iter(const int nband, const int nband_l, const int nbasis, const int ndim); - using HPsiFunc = std::function; - using SPsiFunc = std::function; - /** * @brief Diagonalize the Hamiltonian using the BPCG method. * * This function is called by the HsolverPW::solve() function. * - * @param hpsi_func A function computing the product of the Hamiltonian matrix H - * and a wavefunction blockvector X. - * @param spsi_func A function computing the product of the overlap matrix S - * and a wavefunction blockvector X. + * @param op The H and S block-vector operator, see hsolver::HSOperator. * @param psi_in Pointer to input wavefunction psi matrix with [dim: n_basis x n_band, column major]. * @param eigenvalue_in Pointer to the eigen array with [dim: n_band, column major]. */ - void diag(const HPsiFunc& hpsi_func, - const SPsiFunc& spsi_func, + void diag(const HSOperator& op, T* psi_in, Real* eigenvalue_in, const std::vector& ethr_band); @@ -163,24 +155,23 @@ class DiagoBPCG * psi_in[dim: n_basis x n_band, column major, lda = n_basis_max], * hpsi_out[dim: n_basis x n_band, column major, lda = n_basis_max]. * - * @param hpsi_func A function computing the product of the Hamiltonian matrix H - * and a wavefunction blockvector X. + * @param op The H and S block-vector operator. * @param psi_in The input wavefunction psi. * @param hpsi_out Pointer to the array where the resulting hpsi matrix will be stored. */ void calc_hpsi_with_block( - const HPsiFunc& hpsi_func, - T *psi_in, + const HSOperator& op, + const T *psi_in, ct::Tensor& hpsi_out); /** * @brief Apply the overlap operator to a wavefunction block. * - * @param spsi_func A function computing the product of the overlap matrix S and a wavefunction blockvector X. + * @param op The H and S block-vector operator. * @param psi_in The input wavefunction block. * @param spsi_out The resulting S|psi> block. */ - void calc_spsi_with_block(const SPsiFunc& spsi_func, const T* psi_in, ct::Tensor& spsi_out); + void calc_spsi_with_block(const HSOperator& op, const T* psi_in, ct::Tensor& spsi_out); /** * @brief Diagonalization of the subspace matrix. @@ -258,7 +249,7 @@ class DiagoBPCG * hsub_out[dim: n_band x n_band, column major, lda = n_band], * eigenvalue_out[dim: n_basis_max, column major]. * - * @param hpsi_func A function computing the product of matrix H and wavefunction blockvector X. + * @param op The H and S block-vector operator. * @param psi_in Input wavefunction pointer. * @param psi_out Output wavefunction. * @param hpsi_out Product of psi_out and Hamiltonian. @@ -266,8 +257,7 @@ class DiagoBPCG * @param eigenvalue_out Computed eigen. */ void calc_hsub_with_block( - const HPsiFunc& hpsi_func, - const SPsiFunc& spsi_func, + const HSOperator& op, T *psi_in, ct::Tensor& psi_out, ct::Tensor& hpsi_out, ct::Tensor& spsi_out, ct::Tensor& hsub_out, ct::Tensor& workspace_in, diff --git a/source/source_hsolver/diago_cg.cpp b/source/source_hsolver/diago_cg.cpp index 3b833f60fa8..ca60dbcb7b5 100644 --- a/source/source_hsolver/diago_cg.cpp +++ b/source/source_hsolver/diago_cg.cpp @@ -11,11 +11,17 @@ #include "source_base/tool_title.h" // ModuleBase::TITLE #include "source_base/global_function.h" // ModuleBase::GlobalFunc::NOTE #include "source_hsolver/diago_cg.h" +#include "source_hsolver/diago_iter_assist.h" using namespace hsolver; template DiagoCG::DiagoCG(const std::string& basis_type, const std::string& calculation) +#ifdef __MPI + : diag_comm_(MPI_COMM_SELF, 0, 1) +#else + : diag_comm_(0, 1) +#endif { basis_type_ = basis_type; calculation_ = calculation; @@ -28,18 +34,16 @@ template DiagoCG::DiagoCG(const std::string& basis_type, const std::string& calculation, const bool& need_subspace, - const SubspaceFunc& subspace_func, + const diag_comm_info& diag_comm, const Real& pw_diag_thr, - const int& pw_diag_nmax, - const int& nproc_in_pool) + const int& pw_diag_nmax) + : diag_comm_(diag_comm) { basis_type_ = basis_type; calculation_ = calculation; need_subspace_ = need_subspace; - subspace_func_ = subspace_func; pw_diag_thr_ = pw_diag_thr; pw_diag_nmax_ = pw_diag_nmax; - nproc_in_pool_ = nproc_in_pool; this->one_ = new T(static_cast(1.0)); this->zero_ = new T(static_cast(0.0)); this->neg_one_ = new T(static_cast(-1.0)); @@ -122,10 +126,10 @@ void DiagoCG::diag_once(const ct::Tensor& prec_in, { phi_m.sync(psi[m]); // copy psi_in into internal psi, m=0 has been done in Constructor - this->spsi_func_(phi_m.data(), sphi.data(), this->n_basis_, 1); // sphi = S|psi(m)> + this->op_->spsi(phi_m.data(), sphi.data(), this->n_basis_, 1); // sphi = S|psi(m)> this->schmit_orth(m, psi, sphi, phi_m); - this->spsi_func_(phi_m.data(), sphi.data(), this->n_basis_, 1); // sphi = S|psi(m)> - this->hpsi_func_(phi_m.data(), hphi.data(), this->n_basis_, 1); // hphi = H|psi(m)> + this->op_->spsi(phi_m.data(), sphi.data(), this->n_basis_, 1); // sphi = S|psi(m)> + this->op_->hpsi(phi_m.data(), hphi.data(), this->n_basis_, 1); // hphi = H|psi(m)> eigen_pack[m] = dot_real_op()(this->n_basis_, phi_m.data(), hphi.data()); @@ -150,8 +154,8 @@ void DiagoCG::diag_once(const ct::Tensor& prec_in, g0, cg); // Tensor& - this->hpsi_func_(cg.data(), pphi.data(), this->n_basis_, 1); - this->spsi_func_(cg.data(), scg.data(), this->n_basis_, 1); + this->op_->hpsi(cg.data(), pphi.data(), this->n_basis_, 1); + this->op_->spsi(cg.data(), scg.data(), this->n_basis_, 1); converged = this->update_psi(pphi, cg, @@ -264,7 +268,7 @@ void DiagoCG::orth_grad(const ct::Tensor& psi, ct::Tensor& scg, ct::Tensor& lagrange) { - this->spsi_func_(grad.data(), scg.data(), this->n_basis_, 1); // scg = S|grad> + this->op_->spsi(grad.data(), scg.data(), this->n_basis_, 1); // scg = S|grad> ModuleBase::gemv_op()('C', this->n_basis_, m, @@ -564,6 +568,19 @@ void DiagoCG::schmit_orth(const int& m, const ct::Tensor& psi, const // ModuleBase::timer::end("DiagoCG","schmit_orth"); } +template +void DiagoCG::diag_subspace(const T* psi_in, T* psi_out, const int dim, const int nband, const bool S_orth) +{ + // subspace diagonalization of the current nband vectors, packed with leading dimension dim; + // the eigenvalues it produces are not needed, CG recomputes them. + // The generalized problem is always solved: the S-orthogonal shortcut (heevx instead of + // hegvd) changes eigenvector phases and, for vectors that are only approximately + // S-orthonormal after a CG restart, the results; wavefunction-sensitive outputs such as + // the Wannier90 projections rely on the generalized path. + std::vector eigen(nband, 0.0); + DiagoIterAssist::diag_subspace(*op_, psi_in, psi_out, nband, nband, dim, dim, eigen.data(), diag_comm_, false); +} + template bool DiagoCG::test_exit_cond(const int& ntry, const int& notconv) const { @@ -579,8 +596,7 @@ bool DiagoCG::test_exit_cond(const int& ntry, const int& notconv) con } template -double DiagoCG::diag(const HPsiFunc& hpsi_func, - const SPsiFunc& spsi_func, +double DiagoCG::diag(const HSOperator& op, const int ld_psi, const int nband, const int dim, @@ -615,8 +631,7 @@ double DiagoCG::diag(const HPsiFunc& hpsi_func, /// record the times of trying iterative diagonalization int ntry = 0; this->notconv_ = 0; - hpsi_func_ = hpsi_func; - spsi_func_ = spsi_func; + op_ = &op; // create a new slice of psi to do cg diagonalization ct::Tensor psi_temp = psi.slice({0, 0}, {nband, dim}); @@ -630,22 +645,14 @@ double DiagoCG::diag(const HPsiFunc& hpsi_func, { ct::TensorMap psi_map = ct::TensorMap(psi.data(), psi_temp); const bool assume_S_orthogonal = true; - this->subspace_func_(psi_temp.data(), - psi_map.data(), - dim, - nband, - assume_S_orthogonal); + this->diag_subspace(psi_temp.data(), psi_map.data(), dim, nband, assume_S_orthogonal); psi_temp.sync(psi_map); } else if (need_subspace_) { ct::TensorMap psi_map = ct::TensorMap(psi.data(), psi_temp); const bool assume_S_orthogonal = false; - this->subspace_func_(psi_temp.data(), - psi_map.data(), - dim, - nband, - assume_S_orthogonal); + this->diag_subspace(psi_temp.data(), psi_map.data(), dim, nband, assume_S_orthogonal); psi_temp.sync(psi_map); } diff --git a/source/source_hsolver/diago_cg.h b/source/source_hsolver/diago_cg.h index 99d9369a0a3..c5109cfb0cf 100644 --- a/source/source_hsolver/diago_cg.h +++ b/source/source_hsolver/diago_cg.h @@ -1,12 +1,14 @@ #ifndef MODULE_HSOLVER_DIAGO_CG_H_ #define MODULE_HSOLVER_DIAGO_CG_H_ -#include #include #include #include +#include "source_hsolver/diag_comm_info.h" +#include "source_hsolver/hs_operator.h" + #include #include @@ -22,30 +24,23 @@ class DiagoCG final using Real = typename GetTypeReal::type; using ct_Device = typename ct::PsiToContainer::type; public: - using HPsiFunc = std::function; - using SPsiFunc = std::function; - using SubspaceFunc = std::function; // Constructor need: - // 1. temporary mock of Hamiltonian "Hamilt_PW" - // 2. precondition pointer should point to place of precondition array. + // 1. basis type and calculation type of ABACUS + // 2. diag_comm: the pool over which the subspace matrices are reduced DiagoCG(const std::string& basis_type, const std::string& calculation); DiagoCG( const std::string& basis_type, const std::string& calculation, const bool& need_subspace, - const SubspaceFunc& subspace_func, + const diag_comm_info& diag_comm, const Real& pw_diag_thr, - const int& pw_diag_nmax, - const int& nproc_in_pool); + const int& pw_diag_nmax); ~DiagoCG(); - // virtual void init(){}; - // refactor hpsi_info // this is the diag() function for CG method // returns avg_iter - double diag(const HPsiFunc& hpsi_func, - const SPsiFunc& spsi_func, + double diag(const HSOperator& op, const int ld_psi, const int nband, const int dim, @@ -72,20 +67,16 @@ class DiagoCG final Real pw_diag_thr_ = 1e-5; /// maximum iteration steps for cg diagonalization int pw_diag_nmax_ = 0; - /// number of processors in a node - int nproc_in_pool_ = 0; + /// communicator of the pool sharing the plane waves + const diag_comm_info diag_comm_; /// basis_type of psi std::string basis_type_ = {}; /// calculation type of ABACUS std::string calculation_ = {}; bool need_subspace_ = false; - /// A function object that performs the hPsi calculation. - HPsiFunc hpsi_func_ = nullptr; - /// A function object that performs the sPsi calculation. - SPsiFunc spsi_func_ = nullptr; - /// A function object that performs the subspace calculation. - SubspaceFunc subspace_func_ = nullptr; + /// The H and S operator being diagonalized, set for the duration of diag(). + const HSOperator* op_ = nullptr; void calc_grad( const ct::Tensor& prec, @@ -135,6 +126,9 @@ class DiagoCG final bool test_exit_cond(const int& ntry, const int& notconv) const; + /// subspace rotation of the current nband vectors (packed, leading dimension dim) + void diag_subspace(const T* psi_in, T* psi_out, const int dim, const int nband, const bool S_orth); + using dot_real_op = ModuleBase::dot_real_op; const T * one_ = nullptr, * zero_ = nullptr, * neg_one_ = nullptr; }; diff --git a/source/source_hsolver/diago_dav_subspace.cpp b/source/source_hsolver/diago_dav_subspace.cpp index 2dfbcf73de4..331eb9857e4 100644 --- a/source/source_hsolver/diago_dav_subspace.cpp +++ b/source/source_hsolver/diago_dav_subspace.cpp @@ -108,8 +108,7 @@ Diago_DavSubspace::~Diago_DavSubspace() } template -int Diago_DavSubspace::diag_once(const HPsiFunc& hpsi_func, - const HPsiFunc& spsi_func, +int Diago_DavSubspace::diag_once(const HSOperator& op, T* psi_in, const int psi_in_dmax, Real* eigenvalue_in_hsolver, @@ -143,11 +142,11 @@ int Diago_DavSubspace::diag_once(const HPsiFunc& hpsi_func, // compute h*psi_in_iter // NOTE: bands after the first n_band should yield zero // hpsi[:, 0:nbase_x] = H * psi_in_iter[:, 0:nbase_x] - hpsi_func(this->psi_in_iter, this->hpsi, this->dim, this->notconv); + op.hpsi(this->psi_in_iter, this->hpsi, this->dim, this->notconv); // compute s*psi_in_iter // spsi[:, 0:nbase_x] = S * psi_in_iter[:, 0:nbase_x] - spsi_func(this->psi_in_iter, this->spsi, this->dim, this->notconv); + op.spsi(this->psi_in_iter, this->spsi, this->dim, this->notconv); this->cal_elem(this->dim, nbase, this->notconv, this->psi_in_iter, this->spsi, this->hpsi, this->hcc, this->scc); @@ -166,8 +165,7 @@ int Diago_DavSubspace::diag_once(const HPsiFunc& hpsi_func, { dav_iter++; - this->cal_grad(hpsi_func, - spsi_func, + this->cal_grad(op, this->dim, nbase, this->notconv, @@ -274,8 +272,7 @@ int Diago_DavSubspace::diag_once(const HPsiFunc& hpsi_func, } template -void Diago_DavSubspace::cal_grad(const HPsiFunc& hpsi_func, - const HPsiFunc& spsi_func, +void Diago_DavSubspace::cal_grad(const HSOperator& op, const int& dim, const int& nbase, const int& notconv, @@ -479,8 +476,8 @@ void Diago_DavSubspace::cal_grad(const HPsiFunc& hpsi_func, // update hpsi[:, nbase:nbase+notconv] // hpsi[:, nbase:nbase+notconv] = H * psi_iter[:, nbase:nbase+notconv] - hpsi_func(psi_iter + nbase * dim, hpsi + nbase * this->dim, this->dim, notconv); - spsi_func(psi_iter + nbase * dim, spsi + nbase * this->dim, this->dim, notconv); + op.hpsi(psi_iter + nbase * dim, hpsi + nbase * this->dim, this->dim, notconv); + op.spsi(psi_iter + nbase * dim, spsi + nbase * this->dim, this->dim, notconv); ModuleBase::timer::end("Diago_DavSubspace", "cal_grad"); return; @@ -815,8 +812,7 @@ void Diago_DavSubspace::refresh(const int& dim, } template -int Diago_DavSubspace::diag(const HPsiFunc& hpsi_func, - const HPsiFunc& spsi_func, +int Diago_DavSubspace::diag(const HSOperator& op, T* psi_in, const int psi_in_dmax, Real* eigenvalue_in_hsolver, @@ -832,7 +828,7 @@ int Diago_DavSubspace::diag(const HPsiFunc& hpsi_func, do { - sum_iter += this->diag_once(hpsi_func, spsi_func, psi_in, psi_in_dmax, eigenvalue_in_hsolver, ethr_band); + sum_iter += this->diag_once(op, psi_in, psi_in_dmax, eigenvalue_in_hsolver, ethr_band); ++ntry; diff --git a/source/source_hsolver/diago_dav_subspace.h b/source/source_hsolver/diago_dav_subspace.h index 43e0304729c..4e1e97fbbbc 100644 --- a/source/source_hsolver/diago_dav_subspace.h +++ b/source/source_hsolver/diago_dav_subspace.h @@ -8,9 +8,9 @@ #include "source_base/module_container/ATen/kernels/lapack.h" #include "source_hsolver/diag_comm_info.h" +#include "source_hsolver/hs_operator.h" #include -#include namespace hsolver { @@ -37,11 +37,7 @@ class Diago_DavSubspace ~Diago_DavSubspace(); - // See diago_david.h for information on the HPsiFunc function type - using HPsiFunc = std::function; - - int diag(const HPsiFunc& hpsi_func, - const HPsiFunc& spsi_func, + int diag(const HSOperator& op, T* psi_in, const int psi_in_dmax, Real* eigenvalue_in, @@ -99,8 +95,7 @@ class Diago_DavSubspace base_device::DEVICE_CPU* cpu_ctx = {}; base_device::AbacusDevice_t device = {}; - void cal_grad(const HPsiFunc& hpsi_func, - const HPsiFunc& spsi_func, + void cal_grad(const HSOperator& op, const int& dim, const int& nbase, const int& notconv, @@ -147,8 +142,7 @@ class Diago_DavSubspace std::vector* eigenvalue_iter, T* vcc); - int diag_once(const HPsiFunc& hpsi_func, - const HPsiFunc& spsi_func, + int diag_once(const HSOperator& op, T* psi_in, const int psi_in_dmax, Real* eigenvalue_in, diff --git a/source/source_hsolver/diago_david.cpp b/source/source_hsolver/diago_david.cpp index b7667692dc4..1455d4138f4 100644 --- a/source/source_hsolver/diago_david.cpp +++ b/source/source_hsolver/diago_david.cpp @@ -115,8 +115,7 @@ DiagoDavid::~DiagoDavid() } template -int DiagoDavid::diag_once(const HPsiFunc& hpsi_func, - const SPsiFunc& spsi_func, +int DiagoDavid::diag_once(const HSOperator& op, const int dim, const int nband, const int ld_psi, @@ -157,7 +156,7 @@ int DiagoDavid::diag_once(const HPsiFunc& hpsi_func, { { // phm_in->sPsi(psi_in + m*ld_psi, &this->spsi[m * dim], dim, dim, 1); - spsi_func(psi_in + m*ld_psi,&this->spsi[m*dim],dim, 1); + op.spsi(psi_in + m*ld_psi,&this->spsi[m*dim],dim, 1); } } // begin SchmidtOrth @@ -174,7 +173,7 @@ int DiagoDavid::diag_once(const HPsiFunc& hpsi_func, pre_matrix_mv_m[m]); { // phm_in->sPsi(basis + dim*m, &this->spsi[m * dim], dim, dim, 1); - spsi_func(basis + dim*m, &this->spsi[m * dim], dim, 1); + op.spsi(basis + dim*m, &this->spsi[m * dim], dim, 1); } } @@ -183,7 +182,7 @@ int DiagoDavid::diag_once(const HPsiFunc& hpsi_func, // phm_in->ops->hPsi(dav_hpsi_in); // hpsi[:, 0:nband] = H basis[:, 0:nband] // slice index in this piece of code is in C manner. i.e. 0:id stands for [0,id) - hpsi_func(basis, hpsi, dim, nband); + op.hpsi(basis, hpsi, dim, nband); this->cal_elem(dim, nbase, nbase_x, this->notconv, this->hpsi, this->spsi, this->hcc); @@ -201,8 +200,7 @@ int DiagoDavid::diag_once(const HPsiFunc& hpsi_func, { dav_iter++; - this->cal_grad(hpsi_func, - spsi_func, + this->cal_grad(op, dim, nbase, nbase_x, @@ -292,8 +290,7 @@ int DiagoDavid::diag_once(const HPsiFunc& hpsi_func, template -void DiagoDavid::cal_grad(const HPsiFunc& hpsi_func, - const SPsiFunc& spsi_func, +void DiagoDavid::cal_grad(const HSOperator& op, const int& dim, const int& nbase, // current dimension of the reduced basis const int nbase_x, // maximum dimension of the reduced basis set @@ -509,7 +506,7 @@ void DiagoDavid::cal_grad(const HPsiFunc& hpsi_func, { { // phm_in->sPsi(basis + dim*(nbase + m), &spsi[(nbase + m) * dim], dim, dim, 1); - spsi_func(basis + dim*(nbase + m), &spsi[(nbase + m) * dim], dim, 1); + op.spsi(basis + dim*(nbase + m), &spsi[(nbase + m) * dim], dim, 1); } } // first nbase bands psi* dot notconv bands spsi to prepare lagrange_matrix @@ -558,7 +555,7 @@ void DiagoDavid::cal_grad(const HPsiFunc& hpsi_func, pre_matrix_mv_m[m]); { // phm_in->sPsi(basis + dim*(nbase + m), &spsi[(nbase + m) * dim], dim, dim, 1); - spsi_func(basis + dim*(nbase + m), &spsi[(nbase + m) * dim], dim, 1); + op.spsi(basis + dim*(nbase + m), &spsi[(nbase + m) * dim], dim, 1); } } // calculate H|psi> for not convergence bands @@ -567,7 +564,7 @@ void DiagoDavid::cal_grad(const HPsiFunc& hpsi_func, // &hpsi[nbase * dim]); // &hp(nbase, 0) // phm_in->ops->hPsi(dav_hpsi_in); // hpsi[:, nbase:nbase+notcnv] = H basis[:, nbase:nbase+notcnv] - hpsi_func(basis + nbase * dim, hpsi + nbase * dim, dim, notconv); + op.hpsi(basis + nbase * dim, hpsi + nbase * dim, dim, notconv); delmem_complex_op()(lagrange); delmem_complex_op()(vc_ev_vector); @@ -1006,8 +1003,7 @@ void DiagoDavid::planSchmidtOrth(const int nband, std::vector& p template -int DiagoDavid::diag(const HPsiFunc& hpsi_func, - const SPsiFunc& spsi_func, +int DiagoDavid::diag(const HSOperator& op, const int ld_psi, T *psi_in, Real* eigenvalue_in, @@ -1023,7 +1019,7 @@ int DiagoDavid::diag(const HPsiFunc& hpsi_func, int sum_dav_iter = 0; do { - sum_dav_iter += this->diag_once(hpsi_func, spsi_func, dim, nband, ld_psi, psi_in, eigenvalue_in, ethr_band, david_maxiter); + sum_dav_iter += this->diag_once(op, dim, nband, ld_psi, psi_in, eigenvalue_in, ethr_band, david_maxiter); ++ntry; } while (!check_block_conv(ntry, this->notconv, ntry_max, notconv_max)); diff --git a/source/source_hsolver/diago_david.h b/source/source_hsolver/diago_david.h index e9ee3a50fde..5dd29805378 100644 --- a/source/source_hsolver/diago_david.h +++ b/source/source_hsolver/diago_david.h @@ -8,10 +8,10 @@ #include "source_base/module_container/ATen/kernels/lapack.h" // container::kernels #include "source_hsolver/diag_comm_info.h" +#include "source_hsolver/hs_operator.h" #include "source_hsolver/kernels/hegvd_op.h" #include -#include namespace hsolver { @@ -68,54 +68,12 @@ class DiagoDavid ~DiagoDavid(); - // declare type of matrix-blockvector functions. - // the function type is defined as a std::function object. - /** - * @brief A function type representing the HX function. - * - * This function type is used to define a matrix-blockvector operator H. - * For eigenvalue problem HX = λX or generalized eigenvalue problem HX = λSX, - * this function computes the product of the Hamiltonian matrix H and a blockvector X. - * - * Called as follows: - * hpsi(X, HX, ld, nvec) where X and HX are (ld, nvec)-shaped blockvectors. - * Result HX = H * X is stored in HX. - * - * @param[out] X Head address of input blockvector of type `T*`. - * @param[in] HX Head address of output blockvector of type `T*`. - * @param[in] ld Leading dimension of blockvector. - * @param[in] nvec Number of vectors in a block. - * - * @warning X and HX are the exact address to read input X and store output H*X, - * @warning both of size ld * nvec. - */ - using HPsiFunc = std::function; - - /** - * @brief A function type representing the SX function. - * - * nrow is leading dimension of spsi, npw is leading dimension of psi, nbands is number of vecs - * - * This function type is used to define a matrix-blockvector operator S. - * For generalized eigenvalue problem HX = λSX, - * this function computes the product of the overlap matrix S and a blockvector X. - * - * @param[in] X Pointer to the input blockvector. - * @param[out] SX Pointer to the output blockvector. - * @param[in] ld_psi Leading dimension of psi and spsi. Dimension of X&SX: ld * nvec. - * @param[in] nvec Number of vectors. - */ - using SPsiFunc = std::function; - /** * @brief Performs iterative diagonalization using the David algorithm. * - * @warning Please see docs of `HPsiFunc` for more information about the hpsi mat-vec interface. - * * @tparam T The type of the elements in the matrix. * @tparam Device The device type (CPU or GPU). - * @param hpsi_func The function object that computes the matrix-blockvector product H * psi. - * @param spsi_func The function object that computes the matrix-blockvector product overlap S * psi. + * @param op The H and S block-vector operator, see hsolver::HSOperator. * @param ld_psi The leading dimension of the psi_in array. * @param psi_in The input wavefunction. * @param eigenvalue_in The array to store the eigenvalues. @@ -129,8 +87,7 @@ class DiagoDavid * notconv_max is determined by the accuracy required for the calculation, default 0 */ int diag( - const HPsiFunc& hpsi_func, // function void hpsi(T*, T*, const int, const int) - const SPsiFunc& spsi_func, // function void spsi(T*, T*, const int, const int, const int) + const HSOperator& op, // applies H and S to block vectors const int ld_psi, // Leading dimension of the psi input T *psi_in, // Pointer to eigenvectors Real* eigenvalue_in, // Pointer to store the resulting eigenvalues @@ -179,8 +136,7 @@ class DiagoDavid base_device::DEVICE_CPU* cpu_ctx = {}; base_device::AbacusDevice_t device = {}; - int diag_once(const HPsiFunc& hpsi_func, - const SPsiFunc& spsi_func, + int diag_once(const HSOperator& op, const int dim, const int nband, const int ld_psi, @@ -192,8 +148,7 @@ class DiagoDavid /** * Calculates the preconditioned gradient of the eigenvectors in Davidson method. * - * @param hpsi_func The function to calculate the matrix-blockvector product H * psi. - * @param spsi_func The function to calculate the matrix-blockvector product overlap S * psi. + * @param op The H and S block-vector operator. * @param dim The dimension of the blockvector. * @param nbase The current dimension of the reduced basis. * @param nbase_x The maximum dimension of the reduced basis set. @@ -204,8 +159,7 @@ class DiagoDavid * @param unconv The array of indices for the unconverged eigenpairs. * @param eigenvalue The array of eigenvalues. */ - void cal_grad(const HPsiFunc& hpsi_func, - const SPsiFunc& spsi_func, + void cal_grad(const HSOperator& op, const int& dim, const int& nbase, const int nbase_x, @@ -340,7 +294,6 @@ class DiagoDavid // Note that ct_Device is different from base_device! using ct_Device = typename ct::PsiToContainer::type; - // using hpsi_info = typename hamilt::Operator::hpsi_info; // Dependence of hpsi removed const T *one = nullptr, *zero = nullptr, *neg_one = nullptr; const T one_ = static_cast(1.0), zero_ = static_cast(0.0), neg_one_ = static_cast(-1.0); diff --git a/source/source_hsolver/diago_iter_assist.cpp b/source/source_hsolver/diago_iter_assist.cpp index efbfe4c9695..12e8fce057f 100644 --- a/source/source_hsolver/diago_iter_assist.cpp +++ b/source/source_hsolver/diago_iter_assist.cpp @@ -1,6 +1,5 @@ #include "diago_iter_assist.h" -#include "source_base/complexmatrix.h" #include "source_base/constants.h" #include "source_base/global_function.h" #include "source_base/kernels/math_kernel_op.h" @@ -10,38 +9,31 @@ #include "source_hsolver/diag_comm_info.h" #include "source_hsolver/kernels/hegvd_op.h" +#include + namespace hsolver { //---------------------------------------------------------------------- // Hamiltonian diagonalization in the subspace spanned -// by nstart states psi (atomic or random wavefunctions). +// by nstart vectors psi (atomic or random wavefunctions). // Produces on output n_band eigenvectors (n_band <= nstart) in evc. //---------------------------------------------------------------------- template -void DiagoIterAssist::diag_subspace( - const hamilt::Hamilt* const pHamilt, // hamiltonian operator carrier - const psi::Psi& psi, // [in] wavefunction - psi::Psi& evc, // [out] wavefunction, eigenvectors - Real* en, // [out] eigenvalues - const diag_comm_info& diag_comm, - int n_band, // [in] number of bands to be calculated, also number of rows - // of evc, if set to 0, n_band = nstart, default 0 - const bool S_orth // [in] if true, psi is assumed to be already S-orthogonalized -) +void DiagoIterAssist::diag_subspace(const HSOperator& op, + const T* psi, + T* evc, + const int nstart, + const int n_band, + const int dmin, + const int dmax, + Real* en, + const diag_comm_info& diag_comm, + const bool S_orth) { ModuleBase::TITLE("DiagoIterAssist", "diag_subspace"); ModuleBase::timer::start("DiagoIterAssist", "diag_subspace"); - // two case: - // 1. pw base: nstart = n_band, psi(nbands * npwx) - // 2. lcao_in_pw base: nstart >= n_band, psi(NLOCAL * npwx) - const int nstart = psi.get_nbands(); - // n_band = 0 means default, set n_band = nstart - if (n_band == 0) - { - n_band = nstart; - } assert(n_band <= nstart); // scc is overlap (optional, only needed if input is not s-orthogonal) @@ -61,31 +53,15 @@ void DiagoIterAssist::diag_subspace( resmem_complex_op()(vcc, nstart * nstart, "DiagSub::vcc"); setmem_complex_op()(vcc, 0, nstart * nstart); - // dmin is the active number of plane waves or atomic orbitals - // dmax is the leading dimension of psi - const int dmin = psi.get_current_ngk(); - const int dmax = psi.get_nbasis(); - - T *temp = nullptr; /// temporary array for calculation of evc - bool in_place = false; ///< if temp and evc share the same memory - if (psi.get_pointer() != evc.get_pointer() && psi.get_nbands() == evc.get_nbands()) - { // use memory of evc as temp - temp = evc.get_pointer(); - in_place = true; - } - else - { - resmem_complex_op()(temp, nstart * dmax, "DiagSub::temp"); - } + // temp holds H|psi>, then S|psi>, then the rotated vectors; it is separate + // from evc so that evc may alias psi + T* temp = nullptr; + resmem_complex_op()(temp, nstart * dmax, "DiagSub::temp"); + setmem_complex_op()(temp, 0, nstart * dmax); { // code block to calculate hcc and scc - setmem_complex_op()(temp, 0, nstart * dmax); - T *hpsi = temp; - // do hPsi for all bands - psi::Range all_bands_range(1, psi.get_current_k(), 0, nstart - 1); - hpsi_info hpsi_in(&psi, all_bands_range, hpsi); - pHamilt->ops->hPsi(hpsi_in); + op.hpsi(psi, hpsi, dmax, nstart); ModuleBase::gemm_op()('C', 'N', @@ -93,7 +69,7 @@ void DiagoIterAssist::diag_subspace( nstart, dmin, &one, - psi.get_pointer(), + psi, dmax, hpsi, dmax, @@ -104,8 +80,7 @@ void DiagoIterAssist::diag_subspace( if(!S_orth){ // Only calculate S_sub if not orthogonal T *spsi = temp; - // do sPsi for all bands - pHamilt->sPsi(psi.get_pointer(), spsi, dmax, dmin, nstart); + op.spsi(psi, spsi, dmax, nstart); ModuleBase::gemm_op()('C', 'N', @@ -113,7 +88,7 @@ void DiagoIterAssist::diag_subspace( nstart, dmin, &one, - psi.get_pointer(), + psi, dmax, spsi, dmax, @@ -142,8 +117,6 @@ void DiagoIterAssist::diag_subspace( DiagoIterAssist::diag_hegvd(nstart, n_band, hcc, scc, nstart, en, vcc); } - const int ld_temp = in_place ? dmax : dmin; - { // code block to calculate evc ModuleBase::gemm_op()('N', 'N', @@ -151,20 +124,18 @@ void DiagoIterAssist::diag_subspace( n_band, nstart, &one, - psi.get_pointer(), // dmin * nstart + psi, // dmin * nstart dmax, vcc, // nstart * n_band nstart, &zero, temp, - ld_temp); + dmin); } - if (!in_place) - { - ModuleBase::matrixCopy()(n_band, ld_temp, temp, ld_temp, evc.get_pointer(), dmax); - delmem_complex_op()(temp); - } + ModuleBase::matrixCopy()(n_band, dmin, temp, dmin, evc, dmax); + + delmem_complex_op()(temp); delmem_complex_op()(hcc); if(!S_orth){ delmem_complex_op()(scc); @@ -175,18 +146,50 @@ void DiagoIterAssist::diag_subspace( } template -void DiagoIterAssist::diag_subspace_init( - hamilt::Hamilt* pHamilt, - const T* psi, - int psi_nr, - int psi_nc, - psi::Psi& evc, - Real* en, - const std::string& basis_type, - const std::string& calculation, - const diag_comm_info& diag_comm, - const std::function& add_to_hcc, - const std::function& export_vcc) +void DiagoIterAssist::diag_subspace(const HSOperator& op, + const psi::Psi& psi, // [in] wavefunction + psi::Psi& evc, // [out] wavefunction, eigenvectors + Real* en, // [out] eigenvalues + const diag_comm_info& diag_comm, + int n_band, // [in] number of bands to be calculated, also number of rows + // of evc, if set to 0, n_band = nstart, default 0 + const bool S_orth // [in] if true, psi is assumed to be already S-orthogonalized +) +{ + // two case: + // 1. pw base: nstart = n_band, psi(nbands * npwx) + // 2. lcao_in_pw base: nstart >= n_band, psi(NLOCAL * npwx) + const int nstart = psi.get_nbands(); + // n_band = 0 means default, set n_band = nstart + if (n_band == 0) + { + n_band = nstart; + } + + // dmin is the active number of plane waves or atomic orbitals + // dmax is the leading dimension of psi + diag_subspace(op, + psi.get_pointer(), + evc.get_pointer(), + nstart, + n_band, + psi.get_current_ngk(), + psi.get_nbasis(), + en, + diag_comm, + S_orth); +} + +template +void DiagoIterAssist::diag_subspace_init(const HSOperator& op, + const T* psi, + int psi_nr, + int psi_nc, + psi::Psi& evc, + Real* en, + const std::string& basis_type, + const std::string& calculation, + const diag_comm_info& diag_comm) { ModuleBase::TITLE("DiagoIterAssist", "diag_subspace_init"); ModuleBase::timer::start("DiagoIterAssist", "diag_subspace_init"); @@ -200,27 +203,6 @@ void DiagoIterAssist::diag_subspace_init( const int dmax = evc.get_nbasis(); const int dmin = evc.get_current_ngk(); - // skip the diagonalization if the operators are not allocated - if (pHamilt->ops == nullptr) - { - ModuleBase::WARNING( - "DiagoIterAssist::diag_subspace_init", - "Severe warning: Operators in Hamilt are not allocated yet, will return value of psi to evc directly\n"); - for (int iband = 0; iband < n_band; iband++) - { - for (int ig = 0; ig < dmax; ig++) - { - evc(iband, ig) = psi[iband * dmax + ig]; - } - en[iband] = 0.0; - } - ModuleBase::timer::end("DiagoIterAssist", "diag_subspace_init"); - return; - } - - // ModuleBase::ComplexMatrix hc(nstart, nstart); - // ModuleBase::ComplexMatrix sc(nstart, nstart); - // ModuleBase::ComplexMatrix hvec(nstart, n_band); T *hcc = nullptr, *scc = nullptr, *vcc = nullptr; resmem_complex_op()(hcc, nstart * nstart, "DiagSub::hcc"); resmem_complex_op()(scc, nstart * nstart, "DiagSub::scc"); @@ -231,38 +213,25 @@ void DiagoIterAssist::diag_subspace_init( if (base_device::get_device_type(ctx) == base_device::GpuDevice) { - /// NOTE: current_nbasis must be npw (WITHOUT npol) for Nonlocal::act's - /// gemm K (vkb only has npw rows). See CPU branch comment above. - psi::Psi psi_temp(1, 1, psi_nc, evc.get_current_nbas(), true); - - T* ppsi = psi_temp.get_pointer(); - // hpsi and spsi share the temp space + // band by band on the GPU: the scratch buffer holds one vector only T* temp = nullptr; resmem_complex_op()(temp, psi_nc, "DiagSub::temp"); setmem_complex_op()(temp, 0, psi_nc); T* hpsi = temp; - // do hPsi band by band for (int i = 0; i < nstart; i++) { - // psi_temp is one band psi, psi is all bands psi, the range always is 1 for the only band in psi_temp - syncmem_complex_op()(ppsi, psi + i * psi_nc, psi_nc); - psi::Range band_by_band_range(true, 0, 0, 0); - hpsi_info hpsi_in(&psi_temp, band_by_band_range, hpsi); - // H|Psi> to get hpsi for target band - pHamilt->ops->hPsi(hpsi_in); + op.hpsi(psi + i * psi_nc, hpsi, psi_nc, 1); // calculate the related elements in hcc ModuleBase::gemv_op()('C', psi_nc, nstart, &one, psi, psi_nc, hpsi, 1, &zero, hcc + i * nstart, 1); } T* spsi = temp; - // do sPsi band by band for (int i = 0; i < nstart; i++) { - syncmem_complex_op()(ppsi, psi + i * psi_nc, psi_nc); - pHamilt->sPsi(ppsi, spsi, dmin, dmin, 1); + op.spsi(psi + i * psi_nc, spsi, psi_nc, 1); ModuleBase::gemv_op()('C', psi_nc, @@ -280,37 +249,26 @@ void DiagoIterAssist::diag_subspace_init( } else if (base_device::get_device_type(ctx) == base_device::CpuDevice) { - /// NOTE: the 4th arg (current_nbasis) must be npw (WITHOUT npol), - /// NOT dmin (= nbasis = npol*npwx in SOC). Nonlocal::act uses - /// psi_temp.get_current_nbas() as gemm K, but vkb only has npw rows. - /// dmin (still = nbasis) is kept for hcc/scc gemm K which needs npol. - psi::Psi psi_temp(1, nstart, psi_nc, evc.get_current_nbas(), true); - - T* ppsi = psi_temp.get_pointer(); - syncmem_complex_op()(ppsi, psi, psi_temp.size()); // hpsi and spsi share the temp space T* temp = nullptr; resmem_complex_op()(temp, nstart * psi_nc, "DiagSub::temp"); setmem_complex_op()(temp, 0, nstart * psi_nc); T* hpsi = temp; - // do hPsi for all bands - psi::Range all_bands_range(true, 0, 0, nstart - 1); - hpsi_info hpsi_in(&psi_temp, all_bands_range, hpsi); - pHamilt->ops->hPsi(hpsi_in); + op.hpsi(psi, hpsi, psi_nc, nstart); - ModuleBase::gemm_op()('C', 'N', nstart, nstart, dmin, &one, ppsi, dmax, hpsi, dmax, &zero, hcc, nstart); + ModuleBase::gemm_op()('C', 'N', nstart, nstart, dmin, &one, psi, psi_nc, hpsi, psi_nc, &zero, hcc, nstart); T* spsi = temp; - // do sPsi for all bands - pHamilt->sPsi(ppsi, spsi, psi_temp.get_nbasis(), psi_temp.get_nbasis(), psi_temp.get_nbands()); + op.spsi(psi, spsi, psi_nc, nstart); - ModuleBase::gemm_op()('C', 'N', nstart, nstart, dmin, &one, ppsi, dmax, spsi, dmax, &zero, scc, nstart); + ModuleBase::gemm_op()('C', 'N', nstart, nstart, dmin, &one, psi, psi_nc, spsi, psi_nc, &zero, scc, nstart); delmem_complex_op()(temp); - - add_to_hcc(hcc, nstart); } + // a Hamiltonian may carry a term hpsi() does not cover (EXX in lcao_in_pw) + op.add_to_subspace_h(hcc, nstart); + if (diag_comm.nproc > 1) { #ifdef __MPI @@ -320,22 +278,9 @@ void DiagoIterAssist::diag_subspace_init( } // after generation of H and S matrix, diag them - /// this part only for test, eigenvector would have different phase caused by micro numerical perturbation - /// set 8 bit effective accuracy would help for debugging - /*for(int i=0;i(double(int(hc(i,j).real()*100000000))/100000000, 0); - if(std::norm(sc(i,j))<1e-10) sc(i,j) = ModuleBase::ZERO; - else sc(i,j) = std::complex(double(int(sc(i,j).real()*100000000))/100000000, 0); - } - }*/ - DiagoIterAssist::diag_hegvd(nstart, n_band, hcc, scc, nstart, en, vcc); - export_vcc(vcc, nstart, n_band); + op.export_subspace_vec(vcc, nstart, n_band); //======================= // diagonize the H-matrix @@ -369,10 +314,6 @@ void DiagoIterAssist::diag_subspace_init( { assert(psi != evc.get_pointer()); - // T* evctemp = nullptr; - // resmem_complex_op()(ctx, evctemp, n_band * dmin, "DiagSub::evctemp"); - // setmem_complex_op()(ctx, evctemp, 0, n_band * dmin); - ModuleBase::gemm_op()('N', 'N', dmin, @@ -386,10 +327,6 @@ void DiagoIterAssist::diag_subspace_init( &zero, evc.get_pointer(), dmax); - - // matrixCopy()(ctx, n_band, evctemp, dmin, evc.get_pointer(), dmax); - - // delmem_complex_op()(ctx, evctemp); } delmem_complex_op()(hcc); @@ -487,12 +424,11 @@ void DiagoIterAssist::diag_hegvd(const int nstart, } template -void DiagoIterAssist::cal_hs_subspace( - const hamilt::Hamilt* pHamilt, // hamiltonian operator carrier - const psi::Psi& psi, // [in] wavefunction - T* hcc, - T* scc, - const diag_comm_info& diag_comm) +void DiagoIterAssist::cal_hs_subspace(const HSOperator& op, + const psi::Psi& psi, // [in] wavefunction + T* hcc, + T* scc, + const diag_comm_info& diag_comm) { const int nstart = psi.get_nbands(); @@ -507,13 +443,8 @@ void DiagoIterAssist::cal_hs_subspace( setmem_complex_op()(temp, 0, nstart * dmax); { // code block to calculate hcc and scc - setmem_complex_op()(temp, 0, nstart * dmax); - T* hpsi = temp; - // do hPsi for all bands - psi::Range all_bands_range(1, psi.get_current_k(), 0, nstart - 1); - hpsi_info hpsi_in(&psi, all_bands_range, hpsi); - pHamilt->ops->hPsi(hpsi_in); + op.hpsi(psi.get_pointer(), hpsi, dmax, nstart); ModuleBase::gemm_op()('C', 'N', @@ -530,8 +461,7 @@ void DiagoIterAssist::cal_hs_subspace( nstart); T* spsi = temp; - // do sPsi for all bands - pHamilt->sPsi(psi.get_pointer(), spsi, dmax, dmin, nstart); + op.spsi(psi.get_pointer(), spsi, dmax, nstart); ModuleBase::gemm_op()('C', 'N', diff --git a/source/source_hsolver/diago_iter_assist.h b/source/source_hsolver/diago_iter_assist.h index 225e8b89277..ef89de59bee 100644 --- a/source/source_hsolver/diago_iter_assist.h +++ b/source/source_hsolver/diago_iter_assist.h @@ -1,12 +1,11 @@ #ifndef DIAGOITERASSIST_H #define DIAGOITERASSIST_H -#include "source_base/complexmatrix.h" #include "source_base/macros.h" -#include "source_hamilt/hamilt.h" +#include "source_base/module_device/memory_op.h" +#include "source_hsolver/hs_operator.h" #include "source_psi/psi.h" -#include #include namespace hsolver @@ -30,24 +29,38 @@ class DiagoIterAssist static int SCF_ITER; - // for psi::Psi structure - /** - * @brief Diagonalizes the Hamiltonian in a subspace defined by the given wavefunction. + /** + * @brief Diagonalizes H in the subspace spanned by nstart vectors. * - * This static function computes the eigenvalues and eigenvectors of the Hamiltonian - * within the subspace spanned by the provided wavefunction `psi`. The resulting eigenvectors - * are stored in `evc`, and the corresponding eigenvalues are written to `en`. + * Builds the nstart*nstart matrices (and unless the + * input is S-orthogonal), solves the small eigenproblem and rotates psi + * into the lowest n_band eigenvectors, written to evc. * - * @tparam T Data type for computation (e.g., float, double). - * @tparam Device Device type for computation (e.g., CPU, GPU). - * @param pHamilt Pointer to the Hamiltonian object. - * @param psi Input wavefunction defining the subspace. - * @param evc Output container for computed eigenvectors. - * @param en Output array for computed eigenvalues. - * @param n_band Number of bands (eigenvalues/eigenvectors) to compute. Default is 0 (all). - * @param is_S_orthogonal If true, assumes the input wavefunction is already orthogonalized. + * @param op applies H and S to block vectors + * @param psi [in] nstart vectors, leading dimension dmax + * @param evc [out] n_band vectors, leading dimension dmax; may alias psi + * @param nstart number of input vectors + * @param n_band number of eigenvectors wanted (<= nstart) + * @param dmin active length of each vector + * @param dmax leading dimension of psi and evc + * @param en [out] n_band eigenvalues (host memory) + * @param is_S_orthogonal if true, psi is already S-orthonormal and the + * standard eigenproblem is solved instead of the generalized one */ - static void diag_subspace(const hamilt::Hamilt* const pHamilt, + static void diag_subspace(const HSOperator& op, + const T* psi, + T* evc, + const int nstart, + const int n_band, + const int dmin, + const int dmax, + Real* en, + const diag_comm_info& diag_comm, + const bool is_S_orthogonal = false); + + /// psi::Psi flavour of diag_subspace(): nstart = psi.get_nbands(), + /// n_band = 0 means all of them, dimensions taken from psi. + static void diag_subspace(const HSOperator& op, const psi::Psi& psi, psi::Psi& evc, Real* en, @@ -55,9 +68,10 @@ class DiagoIterAssist int n_band = 0, const bool is_S_orthogonal = false); - /// @brief use LAPACK to diagonalize the Hamiltonian matrix - /// @param pHamilt interface to hamiltonian - /// @param psi wavefunction to diagonalize + /// @brief subspace diagonalization used to build the starting wavefunction + /// @param op interface to H and S; op.add_to_subspace_h() and + /// op.export_subspace_vec() are called around the small eigenproblem + /// @param psi vectors spanning the subspace /// @param psi_nr number of rows (nbands) /// @param psi_nc number of columns (nbasis) /// @param evc new wavefunction @@ -65,21 +79,15 @@ class DiagoIterAssist /// @param basis_type "lcao", "lcao_in_pw" or "pw"; together with calculation it selects /// how the rotation matrix is applied to psi /// @param calculation "scf", "nscf", "md", "relax", ... - /// @note exception handle: if there is no operator initialized in Hamilt, will directly copy value from psi to evc, - /// and return all - zero eigenenergies. - static void diag_subspace_init( - hamilt::Hamilt* pHamilt, - const T* psi, - int psi_nr, - int psi_nc, - psi::Psi& evc, - Real* en, - const std::string& basis_type, - const std::string& calculation, - const diag_comm_info& diag_comm, - const std::function& add_to_hcc = [](T* null, const int n) {}, - const std::function& export_vcc - = [](const T* null, const int n, const int m) {}); + static void diag_subspace_init(const HSOperator& op, + const T* psi, + int psi_nr, + int psi_nc, + psi::Psi& evc, + Real* en, + const std::string& basis_type, + const std::string& calculation, + const diag_comm_info& diag_comm); static void diag_heevx(const int nstart, const int nbands, @@ -96,12 +104,12 @@ class DiagoIterAssist T *vcc); /// @brief calculate Hamiltonian and overlap matrix in subspace spanned by nstart states psi - /// @param pHamilt : hamiltonian operator carrier + /// @param op : applies H and S /// @param psi : wavefunction /// @param hcc : Hamiltonian matrix /// @param scc : overlap matrix - static void cal_hs_subspace(const hamilt::Hamilt* pHamilt, // hamiltonian operator carrier - const psi::Psi& psi, // [in] wavefunction + static void cal_hs_subspace(const HSOperator& op, + const psi::Psi& psi, // [in] wavefunction T* hcc, T* scc, const diag_comm_info& diag_comm); @@ -132,8 +140,6 @@ class DiagoIterAssist private: constexpr static const Device* ctx = {}; - using hpsi_info = typename hamilt::Operator::hpsi_info; - using setmem_var_op = base_device::memory::set_memory_op; using resmem_var_op = base_device::memory::resize_memory_op; using delmem_var_op = base_device::memory::delete_memory_op; diff --git a/source/source_hsolver/diago_lapack.cpp b/source/source_hsolver/diago_lapack.cpp index 48e9651d067..daf6e2f5889 100644 --- a/source/source_hsolver/diago_lapack.cpp +++ b/source/source_hsolver/diago_lapack.cpp @@ -237,7 +237,12 @@ std::pair> DiagoLapack::zhegvx_once(const int ncol, // ABSTOL, M, W, Z, LDZ, WORK, LWORK, RWORK, IWORK, IFAIL, INFO) int n = this->nlocal; int lda = n, ldb = n, ldz = n; - zhegvx_(&itype, +#ifdef __DSP + cpack_zhegvx_driver( +#else + zhegvx_( +#endif + &itype, &jobz, &range, &uplo, @@ -277,7 +282,12 @@ std::pair> DiagoLapack::zhegvx_once(const int ncol, liwork = std::max(1, 5 * n); iwork.resize(liwork, 0); - zhegvx_(&itype, +#ifdef __DSP + cpack_zhegvx_driver( +#else + zhegvx_( +#endif + &itype, &jobz, &range, &uplo, diff --git a/source/source_hsolver/hs_matrix.h b/source/source_hsolver/hs_matrix.h new file mode 100644 index 00000000000..46591bf71bc --- /dev/null +++ b/source/source_hsolver/hs_matrix.h @@ -0,0 +1,30 @@ +#ifndef HSOLVER_HS_MATRIX_H +#define HSOLVER_HS_MATRIX_H + +#include "source_base/matrix_block.h" + +namespace hsolver +{ + +/** + * @brief What a direct (dense) eigensolver needs from the generalized + * eigenproblem H x = e S x: the matrices H(k) and S(k) themselves. + * + * The code that owns the Hamiltonian implements this interface + * (hamilt::HamiltHSMatrix in source_hamilt/hamilt_hs_adapter.h); HSolverLCAO + * and Parallel_K2D only ever ask it for the two matrix views. + */ +template +class HSMatrix +{ + public: + virtual ~HSMatrix() = default; + + /// H(k) and S(k) of k point ik as non-owning views. The memory they point + /// to belongs to the implementer and stays valid until the next call. + virtual void hs_at_k(const int ik, ModuleBase::MatrixBlock& hk, ModuleBase::MatrixBlock& sk) = 0; +}; + +} // namespace hsolver + +#endif // HSOLVER_HS_MATRIX_H diff --git a/source/source_hsolver/hs_operator.h b/source/source_hsolver/hs_operator.h new file mode 100644 index 00000000000..30547b2336b --- /dev/null +++ b/source/source_hsolver/hs_operator.h @@ -0,0 +1,54 @@ +#ifndef HSOLVER_HS_OPERATOR_H +#define HSOLVER_HS_OPERATOR_H + +#include "source_base/module_device/types.h" + +namespace hsolver +{ + +/** + * @brief What an iterative eigensolver needs from the generalized eigenproblem + * H x = e S x: the ability to apply H and S to a block of vectors, and + * nothing else. + * + * The solvers never see how H is built. The code that owns the Hamiltonian + * implements this interface (hamilt::HamiltHSOperator in + * source_hamilt/hamilt_hs_adapter.h) and the solvers only call hpsi()/spsi(). + * Unit tests implement it with a dense matrix. + * + * Block vectors are column major: vector i occupies x[i*ld, i*ld + ld). Only + * the first npw rows carry data, ld is the leading dimension. + */ +template +class HSOperator +{ + public: + virtual ~HSOperator() = default; + + /// switch H and S to k point ik; must precede hpsi()/spsi() for that k + virtual void update_k(const int ik) = 0; + + /// hx[:, 0:nvec) = H * x[:, 0:nvec), both with leading dimension ld + virtual void hpsi(const T* x, T* hx, const int ld, const int nvec) const = 0; + + /// sx[:, 0:nvec) = S * x[:, 0:nvec), both with leading dimension ld + virtual void spsi(const T* x, T* sx, const int ld, const int nvec) const = 0; + + /// Hook used by DiagoIterAssist::diag_subspace_init: hcc is the n*n + /// subspace Hamiltonian (column major, ld n). A Hamiltonian carrying a + /// term that hpsi() does not cover (EXX in lcao_in_pw) adds it here. + virtual void add_to_subspace_h(T* hcc, const int n) const + { + } + + /// Hook used by DiagoIterAssist::diag_subspace_init: vcc is the n*nband + /// matrix of subspace eigenvectors, handed out right after the subspace + /// diagonalization. + virtual void export_subspace_vec(const T* vcc, const int n, const int nband) const + { + } +}; + +} // namespace hsolver + +#endif // HSOLVER_HS_OPERATOR_H diff --git a/source/source_hsolver/hsolver.cpp b/source/source_hsolver/hsolver.cpp index 43a06d41a19..66e3db65c25 100644 --- a/source/source_hsolver/hsolver.cpp +++ b/source/source_hsolver/hsolver.cpp @@ -69,13 +69,11 @@ double set_diagethr_ks(const std::string basis_type, res_diag_ethr = 1.e-2; } res_diag_ethr = std::min(res_diag_ethr, - static_cast(0.1) * drho - / std::max(static_cast(1.0), static_cast(nelec_in))); + static_cast(0.1) * drho / std::max(static_cast(1.0), static_cast(nelec_in))); } - // It is essential for single precision implementation to keep the diag ethr - // value less or equal to the single-precision limit of convergence(0.5e-4). - // modified by denghuilu at 2023-05-15 + // Keep the single-precision diagonalization threshold at or above the floor + // to avoid excessively strict convergence requirements. if (precision_flag_in == "single") { res_diag_ethr = std::max(res_diag_ethr, static_cast(0.5e-4)); @@ -89,7 +87,6 @@ double set_diagethr_ks(const std::string basis_type, return res_diag_ethr; } - double set_diagethr_sdft(const std::string basis_type, const std::string esolver_type, const std::string calculation_in, @@ -147,7 +144,6 @@ double set_diagethr_sdft(const std::string basis_type, return res_diag_ethr; } - double reset_diag_ethr(std::ofstream& ofs_running, const std::string basis_type, const std::string esolver_type, @@ -171,9 +167,8 @@ double reset_diag_ethr(std::ofstream& ofs_running, new_diag_ethr = 0.1 * drho_in / nelec_in; - // It is essential for single precision implementation to keep the diag ethr - // value less or equal to the single-precision limit of convergence(0.5e-4). - // modified by denghuilu at 2023-05-15 + // Keep the single-precision diagonalization threshold at or above the floor + // to avoid excessively strict convergence requirements. if (precision_flag_in == "single") { new_diag_ethr = std::max(new_diag_ethr, static_cast(0.5e-4)); @@ -188,10 +183,7 @@ double reset_diag_ethr(std::ofstream& ofs_running, return new_diag_ethr; }; -double cal_hsolve_error(const std::string basis_type, - const std::string esolver_type, - const double diag_ethr_in, - const double nelec_in) +double cal_hsolve_error(const std::string basis_type, const std::string esolver_type, const double diag_ethr_in, const double nelec_in) { if (basis_type == "pw" && esolver_type == "ksdft") { @@ -203,4 +195,4 @@ double cal_hsolve_error(const std::string basis_type, } }; -} // namespace hsolver \ No newline at end of file +} // namespace hsolver diff --git a/source/source_hsolver/hsolver_lcao.cpp b/source/source_hsolver/hsolver_lcao.cpp index 79751179d77..bef8e0befad 100644 --- a/source/source_hsolver/hsolver_lcao.cpp +++ b/source/source_hsolver/hsolver_lcao.cpp @@ -31,7 +31,7 @@ #include "source_base/memory_recorder.h" #include "source_base/timer.h" #include "source_estate/elecstate_lcao.h" -#include "source_estate/module_dm/cal_dm_psi.h" +#include "source_estate/module_dm/dm_from_psi.h" #include "source_estate/module_dm/density_matrix.h" #include "source_hsolver/parallel_k2d.h" @@ -39,12 +39,13 @@ namespace hsolver { template -void HSolverLCAO::solve(hamilt::Hamilt* pHamilt, +void HSolverLCAO::solve(HSMatrix& hs, psi::Psi& psi, elecstate::ElecState* pes, - elecstate::DensityMatrix& dm, // mohan add 2025-11-03 + module_dm::DensityMatrix& dm, // mohan add 2025-11-03 Charge &chr, const int nspin, + const double omega, const bool skip_charge) { ModuleBase::TITLE("HSolverLCAO", "solve"); @@ -56,13 +57,13 @@ void HSolverLCAO::solve(hamilt::Hamilt* pHamilt, #ifdef __CUDA if (this->method == "cusolver" && this->world_nproc > 1) { - this->parakSolve_cusolver(pHamilt, psi, pes); + this->parakSolve_cusolver(hs, psi, pes); }else #endif if (this->kpar_lcao > 1 && (this->method == "genelpa" || this->method == "elpa" || this->method == "scalapack_gvx" || this->method == "lapack")) { - this->parakSolve(pHamilt, psi, pes, this->kpar_lcao, nspin); + this->parakSolve(hs, psi, pes, this->kpar_lcao, nspin); } else #endif if (this->kpar_lcao == 1) @@ -70,14 +71,15 @@ void HSolverLCAO::solve(hamilt::Hamilt* pHamilt, /// Loop over k points for solve Hamiltonian to eigenpairs(eigenvalues and eigenvectors). for (int ik = 0; ik < psi.get_nk(); ++ik) { - /// update H(k) for each k point - pHamilt->updateHk(ik); + /// H(k) and S(k) are all the eigensolvers need from the Hamiltonian + ModuleBase::MatrixBlock hk, sk; + hs.hs_at_k(ik, hk, sk); /// find psi pointer for each k point psi.fix_k(ik); /// solve eigenvector and eigenvalue for H(k) - this->hamiltSolvePsiK(pHamilt, psi, &(pes->ekb(ik, 0))); + this->hamiltSolvePsiK(hk, sk, psi, &(pes->ekb(ik, 0))); } } else @@ -96,15 +98,15 @@ void HSolverLCAO::solve(hamilt::Hamilt* pHamilt, pes->skip_weights); elecstate::calEBand(pes->ekb, pes->wg, pes->f_en); - elecstate::cal_dm_psi(dm.get_paraV_pointer(), pes->wg, psi, dm); - dm.cal_DMR(); + module_dm::dm_from_psi(this->ParaV, pes->wg, psi, dm); + dm.cal_dmr(-1); if (!skip_charge) { // compute charge density from density matrix, mohan update 20251024 // delegate to ElecStateLCAO to keep the source_lcao dependency out of // source_hsolver (mirrors the pexsi branch below and the PW psiToRho path) - dynamic_cast*>(pes)->dmToRho(dm.get_DMR_vector(), nspin, &chr); + dynamic_cast*>(pes)->dmToRho(dm.get_dmr_vec(), nspin, &chr, omega); } else { @@ -117,18 +119,16 @@ void HSolverLCAO::solve(hamilt::Hamilt* pHamilt, DiagoPexsi pe(ParaV, nspin, this->nlocal, this->nelec, this->world_nproc); for (int ik = 0; ik < psi.get_nk(); ++ik) { - /// update H(k) for each k point - pHamilt->updateHk(ik); psi.fix_k(ik); ModuleBase::MatrixBlock hk, sk; - pHamilt->matrix(hk, sk); + hs.hs_at_k(ik, hk, sk); // solve eigenvector and eigenvalue for H(k) pe.diag(hk, sk, psi, nullptr); } auto _pes = dynamic_cast*>(pes); pes->f_en.eband = pe.totalFreeEnergy; // maybe eferm could be dealt with in the future - _pes->dm2rho(pe.DM, pe.EDM, &dm); + _pes->dm2rho(pe.DM, pe.EDM, &dm, omega); #endif } @@ -137,16 +137,14 @@ void HSolverLCAO::solve(hamilt::Hamilt* pHamilt, } template -void HSolverLCAO::hamiltSolvePsiK(hamilt::Hamilt* hm, psi::Psi& psi, double* eigenvalue) +void HSolverLCAO::hamiltSolvePsiK(ModuleBase::MatrixBlock& hk, + ModuleBase::MatrixBlock& sk, + psi::Psi& psi, + double* eigenvalue) { ModuleBase::TITLE("HSolverLCAO", "hamiltSolvePsiK"); ModuleBase::timer::start("HSolverLCAO", "hamiltSolvePsiK"); - // H(k) and S(k) are all the eigensolvers need from the Hamiltonian, so - // fetch them once here rather than once inside each solver. - ModuleBase::MatrixBlock hk, sk; - hm->matrix(hk, sk); - if (this->method == "scalapack_gvx") { #ifdef __MPI @@ -195,7 +193,7 @@ void HSolverLCAO::hamiltSolvePsiK(hamilt::Hamilt* hm, psi::Psi& psi, do } template -void HSolverLCAO::parakSolve(hamilt::Hamilt* pHamilt, +void HSolverLCAO::parakSolve(HSMatrix& hs, psi::Psi& psi, elecstate::ElecState* pes, const int kpar, @@ -203,7 +201,7 @@ void HSolverLCAO::parakSolve(hamilt::Hamilt* pHamilt, { #ifdef __MPI ModuleBase::timer::start("HSolverLCAO", "parakSolve"); - auto k2d = Parallel_K2D(); + Parallel_K2D k2d; k2d.set_kpar(kpar); int nbands = this->ParaV->get_nbands(); int nks = psi.get_nk(); @@ -215,12 +213,6 @@ void HSolverLCAO::parakSolve(hamilt::Hamilt* pHamilt, int coord_col = k2d.get_p2D_pool()->get_coord_col(); int ncol_bands_pool = numroc_(&(nbands), &(nb2d), &coord_col, &zero, &(k2d.get_p2D_pool()->dim1)); - /// Parallel_K2D only redistributes H(k)/S(k); updating the Hamiltonian - /// for a given k point stays here, where the Hamiltonian is known. - auto get_hsk = [pHamilt](int ik, ModuleBase::MatrixBlock& hk, ModuleBase::MatrixBlock& sk) { - pHamilt->updateHk(ik); - pHamilt->matrix(hk, sk); - }; /// Loop over k points for solve Hamiltonian to charge density for (int ik = 0; ik < k2d.get_pKpoints()->get_max_nks_pool(); ++ik) { @@ -246,7 +238,7 @@ void HSolverLCAO::parakSolve(hamilt::Hamilt* pHamilt, ik_kpar[i] = ik + k2d.get_pKpoints()->startk_pool[i]; } } - k2d.distribute_hsk(get_hsk, ik_kpar, nrow); + k2d.distribute_hsk(hs, ik_kpar, nrow); /// global index of k point int ik_global = ik + k2d.get_pKpoints()->startk_pool[k2d.get_my_pool()]; auto psi_pool = psi::Psi(1, ncol_bands_pool, k2d.get_p2D_pool()->nrow, k2d.get_p2D_pool()->nrow, true); @@ -327,7 +319,7 @@ void HSolverLCAO::parakSolve(hamilt::Hamilt* pHamilt, #if defined (__MPI) && defined (__CUDA) template -void HSolverLCAO::parakSolve_cusolver(hamilt::Hamilt* pHamilt, +void HSolverLCAO::parakSolve_cusolver(HSMatrix& hs, psi::Psi& psi, elecstate::ElecState* pes) { @@ -414,9 +406,8 @@ void HSolverLCAO::parakSolve_cusolver(hamilt::Hamilt* pHamilt, hk_mat.resize(nrow * ncol); sk_mat.resize(nrow * ncol); } - pHamilt->updateHk(ik); ModuleBase::MatrixBlock hk_2D, sk_2D; - pHamilt->matrix(hk_2D, sk_2D); + hs.hs_at_k(ik, hk_2D, sk_2D); int desc_tmp[9]; T* hk_local_ptr = hk_mat.data(); T* sk_local_ptr = sk_mat.data(); diff --git a/source/source_hsolver/hsolver_lcao.h b/source/source_hsolver/hsolver_lcao.h index 9fb67fbd530..06702f8bc7c 100644 --- a/source/source_hsolver/hsolver_lcao.h +++ b/source/source_hsolver/hsolver_lcao.h @@ -2,7 +2,7 @@ #define HSOLVERLCAO_H #include "source_estate/elecstate.h" -#include "source_hamilt/hamilt.h" +#include "source_hsolver/hs_matrix.h" #include "source_basis/module_ao/parallel_orbitals.h" #include "source_estate/module_charge/charge.h" // mohan add 20251024 @@ -27,25 +27,30 @@ class HSolverLCAO : ParaV(ParaV_in), method(method_in), kpar_lcao(kpar_lcao_in), nlocal(nlocal_in), nbands(nbands_in), nelec(nelec_in), use_gpu(use_gpu_in), world_nproc(world_nproc_in), world_rank(world_rank_in){}; - void solve(hamilt::Hamilt* pHamilt, + /// @param hs supplies H(k) and S(k) of the Hamiltonian for each k point + void solve(HSMatrix& hs, psi::Psi& psi, elecstate::ElecState* pes, - elecstate::DensityMatrix& dm, // mohan add 2025-11-03 + module_dm::DensityMatrix& dm, // mohan add 2025-11-03 Charge &chr, // charge density const int nspin, + const double omega, // current cell volume (ucell.omega), NOT rhopw->omega const bool skip_charge); private: - void hamiltSolvePsiK(hamilt::Hamilt* hm, psi::Psi& psi, double* eigenvalue); // for kpar_lcao == 1 + void hamiltSolvePsiK(ModuleBase::MatrixBlock& hk, + ModuleBase::MatrixBlock& sk, + psi::Psi& psi, + double* eigenvalue); // for kpar_lcao == 1 - void parakSolve(hamilt::Hamilt* pHamilt, + void parakSolve(HSMatrix& hs, psi::Psi& psi, elecstate::ElecState* pes, const int kpar, const int nspin); // for kpar_lcao > 1 // The solving algorithm using cusolver is different from others, so a separate function is needed - void parakSolve_cusolver(hamilt::Hamilt* pHamilt, + void parakSolve_cusolver(HSMatrix& hs, psi::Psi& psi, elecstate::ElecState* pes); diff --git a/source/source_hsolver/hsolver_lcaopw.cpp b/source/source_hsolver/hsolver_lcaopw.cpp index 4dbc31b56a4..7c324c22296 100644 --- a/source/source_hsolver/hsolver_lcaopw.cpp +++ b/source/source_hsolver/hsolver_lcaopw.cpp @@ -5,17 +5,11 @@ #include "source_base/tool_quit.h" #include "source_estate/elecstate_pw.h" #include "source_estate/elecstate_tools.h" -#include "source_hamilt/module_xc/general_exx_info.h" #include "source_hsolver/diag_comm_info.h" #include "source_hsolver/diago_iter_assist.h" -#include "source_pw/module_pwdft/hamilt_pw.h" #include -#ifdef __EXX -#include "source_pw/module_pwdft/hamilt_lcaopw.h" -#endif - namespace hsolver { @@ -23,16 +17,13 @@ namespace hsolver lcao_in_pw */ template -void HSolverLIP::solve(hamilt::Hamilt* pHamilt, // ESolver_KS_PW::p_hamilt +void HSolverLIP::solve(HSOperator& op, // ESolver_KS_PW::p_hamilt behind the operator interface psi::Psi& psi, // ESolver_KS_PW::kspw_psi elecstate::ElecState* pes, // ESolver_KS_PW::pes psi::Psi& transform, const diag_comm_info& diag_comm, std::ostream& log, - const bool skip_charge, - const double tpiba, - const int nat, - const General_Exx_Info& exx_info) + const bool skip_charge) { ModuleBase::TITLE("HSolverLIP", "solve"); ModuleBase::timer::start("HSolverLIP", "solve"); @@ -40,37 +31,13 @@ void HSolverLIP::solve(hamilt::Hamilt* pHamilt, // ESolver_KS_PW::p_hamilt for (int ik = 0; ik < this->wfc_basis->nks; ++ik) { /// update H(k) for each k point - pHamilt->updateHk(ik); + op.update_k(ik); psi.fix_k(ik); transform.fix_k(ik); -#ifdef __EXX - auto& exx_lip = dynamic_cast*>(pHamilt)->exx_lip; - bool cal_exx = exx_info.cal_exx; - double hybrid_alpha = exx_info.hybrid_alpha; - auto add_exx_to_subspace_hamilt = [&ik, &exx_lip, cal_exx, hybrid_alpha](T* hcc, const int naos) -> void { - if (cal_exx) - { - for (int n = 0; n < naos; ++n) - { - for (int m = 0; m < naos; ++m) - { - hcc[n * naos + m] - += (T)hybrid_alpha * exx_lip.get_exx_matrix()[ik][m][n]; - } - } - } - }; - auto set_exxlip_lcaowfc = [&ik, &exx_lip, cal_exx](const T* const vcc, const int naos, const int nbands) -> void { - if (cal_exx) - { - exx_lip.set_hvec(ik, vcc, naos, nbands); - } - }; -#endif /// solve eigenvector and eigenvalue for H(k) - hsolver::DiagoIterAssist::diag_subspace_init(pHamilt, // interface to hamilt + hsolver::DiagoIterAssist::diag_subspace_init(op, transform.get_pointer(), // transform matrix between lcao and pw transform.get_nbands(), transform.get_nbasis(), @@ -78,13 +45,7 @@ void HSolverLIP::solve(hamilt::Hamilt* pHamilt, // ESolver_KS_PW::p_hamilt eigenvalues.data() + ik * pes->ekb.nc, // eigenvalues this->basis_type, this->calculation, - diag_comm -#ifdef __EXX - , - add_exx_to_subspace_hamilt, - set_exxlip_lcaowfc -#endif - ); + diag_comm); if (skip_charge) { diff --git a/source/source_hsolver/hsolver_lcaopw.h b/source/source_hsolver/hsolver_lcaopw.h index b9569156985..0e5fac0a9fc 100644 --- a/source/source_hsolver/hsolver_lcaopw.h +++ b/source/source_hsolver/hsolver_lcaopw.h @@ -3,12 +3,9 @@ #include "source_base/macros.h" #include "source_estate/elecstate.h" -#include "source_hamilt/hamilt.h" +#include "source_hsolver/hs_operator.h" #include -/// General_Exx_Info forward declaration, full definition in general_exx_info.h -struct General_Exx_Info; - namespace hsolver { @@ -34,21 +31,18 @@ class HSolverLIP global_nbands(global_nbands_in) {}; /// @brief solve function for lcao_in_pw - /// @param pHamilt interface to hamilt + /// @param op the H and S operator of the Hamiltonian; its subspace hooks carry the EXX term /// @param psi reference to psi /// @param pes interface to elecstate /// @param transform transformation matrix between lcao and pw /// @param skip_charge - void solve(hamilt::Hamilt* pHamilt, + void solve(HSOperator& op, psi::Psi& psi, elecstate::ElecState* pes, psi::Psi& transform, const diag_comm_info& diag_comm, std::ostream& log, - const bool skip_charge, - const double tpiba, - const int nat, - const General_Exx_Info& exx_info); + const bool skip_charge); private: ModulePW::PW_Basis_K* wfc_basis = nullptr; diff --git a/source/source_hsolver/hsolver_linear.cpp b/source/source_hsolver/hsolver_linear.cpp new file mode 100644 index 00000000000..f86dce35e14 --- /dev/null +++ b/source/source_hsolver/hsolver_linear.cpp @@ -0,0 +1,100 @@ +#include "source_hsolver/hsolver_linear.h" + +#include "source_base/tool_quit.h" + +namespace hsolver +{ +const char* linear_status_name(const LinearSolveStatus status) +{ + switch (status) + { + case LinearSolveStatus::converged: + return "converged"; + case LinearSolveStatus::breakdown: + return "breakdown"; + case LinearSolveStatus::max_iterations: + return "maximum iterations"; + case LinearSolveStatus::residual_mismatch: + return "true residual check failed"; + case LinearSolveStatus::preconditioner_failure: + return "preconditioner application failed"; + } + return "unknown linear solve status"; +} + +template +HSolverLinear::HSolverLinear(const LinearSolveOptions& options, const diag_comm_info& comm) + : default_control_{options.max_iterations, options.reconstruct} +{ + using Real = typename GetTypeReal::type; + const double roundoff_tolerance = 100.0 * std::numeric_limits::epsilon(); + tolerance_ = options.tolerance == 0.0 ? std::max(1e-10, roundoff_tolerance) : options.tolerance; + if (options.method == LinearMethod::bicgstab) + { + bicgstab_.reset(new LinearBiCGSTAB(tolerance_, comm)); + } + else if (options.method == LinearMethod::cgs) + { + cgs_.reset(new LinearCGS(tolerance_, comm)); + } + else if (options.method == LinearMethod::gmres) + { + gmres_.reset(new LinearGMRES(tolerance_, options, comm)); + } + else + { + ModuleBase::WARNING_QUIT("HSolverLinear", "Unsupported linear solver method."); + } +} + +template +LinearSolveResult HSolverLinear::solve(const LinearOperator& op, + const LinearOperator& preconditioner, + int ld, + int nvec, + int dim, + T* x, + const T* b, + const T* initial_residual, + bool force_check) +{ + return solve(op, preconditioner, ld, nvec, dim, x, b, initial_residual, force_check, default_control_); +} + +template +LinearSolveResult HSolverLinear::solve(const LinearOperator& op, + const LinearOperator& preconditioner, + int ld, + int nvec, + int dim, + T* x, + const T* b, + const T* initial_residual, + bool force_check, + const LinearSolveControl& control) +{ + if (bicgstab_) + { + return bicgstab_->solve(op, preconditioner, ld, nvec, dim, x, b, initial_residual, control.max_iterations); + } + else if (cgs_) + { + return cgs_->solve(op, preconditioner, ld, nvec, dim, x, b, initial_residual, control.max_iterations); + } + else if (gmres_) + { + return gmres_->solve(op, preconditioner, ld, nvec, dim, x, b, initial_residual, force_check, control); + } + else + { + ModuleBase::WARNING_QUIT("HSolverLinear::solve", "No linear solver has been initialized."); + } +} + +template class HSolverLinear, base_device::DEVICE_CPU>; +template class HSolverLinear, base_device::DEVICE_CPU>; +#if defined(__CUDA) || defined(__ROCM) +template class HSolverLinear, base_device::DEVICE_GPU>; +template class HSolverLinear, base_device::DEVICE_GPU>; +#endif +} // namespace hsolver diff --git a/source/source_hsolver/hsolver_linear.h b/source/source_hsolver/hsolver_linear.h new file mode 100644 index 00000000000..776797db20b --- /dev/null +++ b/source/source_hsolver/hsolver_linear.h @@ -0,0 +1,52 @@ +#ifndef HSOLVER_LINEAR_H +#define HSOLVER_LINEAR_H +#include "source_hsolver/linear_bicgstab.h" +#include "source_hsolver/linear_cgs.h" +#include "source_hsolver/linear_gmres.h" + +#include + +namespace hsolver +{ +/** @brief Method selection and reusable numerical workspace, independent of physics. */ +template +class HSolverLinear +{ + private: + std::unique_ptr> bicgstab_; + std::unique_ptr> cgs_; + std::unique_ptr> gmres_; + double tolerance_ = 0.0; + const LinearSolveControl default_control_; + + public: + HSolverLinear(const LinearSolveOptions& options, const diag_comm_info& comm); + /** @brief Return the effective tolerance after resolving the precision-dependent default. */ + double tolerance() const + { + return tolerance_; + } + /** @brief Reuse an optional CN residual and control independent GMRES verification. */ + LinearSolveResult solve(const LinearOperator& op, + const LinearOperator& preconditioner, + int ld, + int nvec, + int dim, + T* x, + const T* b, + const T* initial_residual, + bool force_check); + /** @brief Reuse solver storage with an explicit per-call budget and reconstruction policy. */ + LinearSolveResult solve(const LinearOperator& op, + const LinearOperator& preconditioner, + int ld, + int nvec, + int dim, + T* x, + const T* b, + const T* initial_residual, + bool force_check, + const LinearSolveControl& control); +}; +} // namespace hsolver +#endif diff --git a/source/source_hsolver/hsolver_pw.cpp b/source/source_hsolver/hsolver_pw.cpp index d365734475e..1b316b567e0 100644 --- a/source/source_hsolver/hsolver_pw.cpp +++ b/source/source_hsolver/hsolver_pw.cpp @@ -5,7 +5,6 @@ #include "source_base/tool_quit.h" #include "source_estate/elecstate_pw.h" #include "source_estate/elecstate_tools.h" -#include "source_hamilt/hamilt.h" #include "source_hsolver/diag_comm_info.h" #include "source_hsolver/diago_bpcg.h" #include "source_hsolver/diago_cg.h" @@ -64,16 +63,14 @@ void HSolverPW::cal_smooth_ethr(const double& wk, } template -void HSolverPW::solve(hamilt::Hamilt* pHamilt, +void HSolverPW::solve(HSOperator& op, psi::Psi& psi, elecstate::ElecState* pes, double* out_eigenvalues, const int rank_in_pool_in, const int nproc_in_pool_in, std::ostream& log, - const bool skip_charge, - const double tpiba, - const int nat) + const bool skip_charge) { ModuleBase::TITLE("HSolverPW", "solve"); ModuleBase::timer::start("HSolverPW", "solve"); @@ -107,7 +104,7 @@ void HSolverPW::solve(hamilt::Hamilt* pHamilt, const int ik = k_order[i]; // update H(k) for each k point - pHamilt->updateHk(ik); + op.update_k(ik); @@ -134,7 +131,7 @@ void HSolverPW::solve(hamilt::Hamilt* pHamilt, // solve eigenvector and eigenvalue for H(k) - this->hamiltSolvePsiK(pHamilt, psi, precondition, eigenvalues.data() + ik * psi.get_nbands(), this->wfc_basis->nks); + this->hamiltSolvePsiK(op, psi, precondition, eigenvalues.data() + ik * psi.get_nbands(), this->wfc_basis->nks); if (skip_charge) { @@ -150,7 +147,7 @@ void HSolverPW::solve(hamilt::Hamilt* pHamilt, for (int ik = 0; ik < this->wfc_basis->nks; ++ik) { // update H(k) for each k point - pHamilt->updateHk(ik); + op.update_k(ik); @@ -172,7 +169,7 @@ void HSolverPW::solve(hamilt::Hamilt* pHamilt, // solve eigenvector and eigenvalue for H(k) - this->hamiltSolvePsiK(pHamilt, psi, precondition, eigenvalues.data() + ik * psi.get_nbands(), this->wfc_basis->nks); + this->hamiltSolvePsiK(op, psi, precondition, eigenvalues.data() + ik * psi.get_nbands(), this->wfc_basis->nks); // output iteration information and reset avg_iter if (skip_charge) @@ -222,12 +219,12 @@ void HSolverPW::solve(hamilt::Hamilt* pHamilt, reinterpret_cast*>(pes)->psiToRho(psi); } - ModuleBase::timer::end("HSolverPW", "solve"); - return; + ModuleBase::timer::end("HSolverPW", "solve"); + return; } template -void HSolverPW::hamiltSolvePsiK(hamilt::Hamilt* hm, +void HSolverPW::hamiltSolvePsiK(const HSOperator& op, psi::Psi& psi, std::vector& pre_condition, Real* eigenvalue, @@ -265,41 +262,17 @@ void HSolverPW::hamiltSolvePsiK(hamilt::Hamilt* hm, ModuleBase::WARNING_QUIT("HSolverPW::hamiltSolvePsiK", msg); } - // Shared matrix-blockvector operators used by all iterative solvers. - auto hpsi_func = [hm, cur_nbasis](T* psi_in, T* hpsi_out, const int ld_psi, const int nvec) { - auto psi_wrapper = psi::Psi(psi_in, 1, nvec, ld_psi, cur_nbasis); - psi::Range bands_range(true, 0, 0, nvec - 1); - using hpsi_info = typename hamilt::Operator::hpsi_info; - hpsi_info info(&psi_wrapper, bands_range, hpsi_out); - hm->ops->hPsi(info); - }; - auto spsi_func = [hm, cur_nbasis](const T* psi_in, T* spsi_out, const int ld_psi, const int nvec) { - hm->sPsi(psi_in, spsi_out, ld_psi, cur_nbasis, nvec); - }; - if (this->method == "cg") { - // wrap the subspace_func into a lambda function - // if S_orth is true, then assume psi is S-orthogonal, solve standard eigenproblem - // otherwise, solve generalized eigenproblem - auto subspace_func = - [hm, cur_nbasis, &comm_info](T* psi_in, T* psi_out, const int ld_psi, const int nband, const bool S_orth) { - auto psi_in_wrapper = psi::Psi(psi_in, 1, nband, ld_psi, cur_nbasis); - auto psi_out_wrapper = psi::Psi(psi_out, 1, nband, ld_psi, cur_nbasis); - std::vector eigen(nband, 0.0); - DiagoIterAssist::diag_subspace(hm, psi_in_wrapper, psi_out_wrapper, eigen.data(), comm_info); - }; DiagoCG cg(this->basis_type, this->calculation_type, this->need_subspace, - subspace_func, + comm_info, this->diag_thr, - this->diag_iter_max, - this->nproc_in_pool); + this->diag_iter_max); DiagoIterAssist::avg_iter += static_cast( - cg.diag(hpsi_func, - spsi_func, + cg.diag(op, psi.get_nbasis(), psi.get_nbands(), psi.get_current_ngk(), @@ -318,7 +291,7 @@ void HSolverPW::hamiltSolvePsiK(hamilt::Hamilt* hm, const int ndim = psi.get_current_ngk(); DiagoBPCG bpcg(pre_condition.data()); bpcg.init_iter(this->nbands, nband_l, nbasis, ndim); - bpcg.diag(hpsi_func, spsi_func, psi.get_pointer(), eigenvalue, this->ethr_band); + bpcg.diag(op, psi.get_pointer(), eigenvalue, this->ethr_band); } else if (this->method == "dav_subspace") { @@ -336,8 +309,7 @@ void HSolverPW::hamiltSolvePsiK(hamilt::Hamilt* hm, this->nb2d); DiagoIterAssist::avg_iter += static_cast( - dav_subspace.diag(hpsi_func, - spsi_func, + dav_subspace.diag(op, psi.get_pointer(), psi.get_nbasis(), eigenvalue, @@ -366,8 +338,7 @@ void HSolverPW::hamiltSolvePsiK(hamilt::Hamilt* hm, DiagoDavid david(pre_condition.data(), nband, dim, this->pw_diag_ndim, comm_info); // do diag and add davidson iteration counts up to avg_iter DiagoIterAssist::avg_iter += static_cast( - david.diag(hpsi_func, - spsi_func, + david.diag(op, ld_psi, psi.get_pointer(), eigenvalue, diff --git a/source/source_hsolver/hsolver_pw.h b/source/source_hsolver/hsolver_pw.h index a44053e3403..d5cbb772a8c 100644 --- a/source/source_hsolver/hsolver_pw.h +++ b/source/source_hsolver/hsolver_pw.h @@ -4,7 +4,7 @@ #include "source_base/macros.h" #include "source_basis/module_pw/pw_basis_k.h" #include "source_estate/elecstate.h" -#include "source_hamilt/hamilt.h" +#include "source_hsolver/hs_operator.h" #include #include @@ -55,25 +55,22 @@ class HSolverPW use_k_continuity(use_k_continuity_in) {}; /// @brief solve function for pw - /// @param pHamilt interface to hamilt + /// @param op the H and S operator of the Hamiltonian, switched to each k point in turn /// @param psi reference to psi /// @param pes interface to elecstate - /// @param method_in dav or cg /// @param skip_charge - void solve(hamilt::Hamilt* pHamilt, + void solve(HSOperator& op, psi::Psi& psi, elecstate::ElecState* pes, double* out_eigenvalues, const int rank_in_pool_in, const int nproc_in_pool_in, std::ostream& log, - const bool skip_charge, - const double tpiba, - const int nat); + const bool skip_charge); protected: // diago caller - void hamiltSolvePsiK(hamilt::Hamilt* hm, + void hamiltSolvePsiK(const HSOperator& op, psi::Psi& psi, std::vector& pre_condition, Real* eigenvalue, diff --git a/source/source_hsolver/hsolver_pw_sdft.cpp b/source/source_hsolver/hsolver_pw_sdft.cpp deleted file mode 100644 index db075375515..00000000000 --- a/source/source_hsolver/hsolver_pw_sdft.cpp +++ /dev/null @@ -1,140 +0,0 @@ -#include "hsolver_pw_sdft.h" - -#include "source_base/global_function.h" -#include "source_base/parallel_comm.h" -#include "source_base/parallel_device.h" -#include "source_base/timer.h" -#include "source_base/tool_title.h" -#include "source_estate/module_charge/symm_rho.h" -#include "source_estate/elecstate_tools.h" - -#include - -namespace hsolver -{ -template -void HSolverPW_SDFT::solve(const UnitCell& ucell, - hamilt::Hamilt* pHamilt, - psi::Psi& psi, - psi::Psi& psi_cpu, - elecstate::ElecState* pes, - ModulePW::PW_Basis_K* wfc_basis, - Stochastic_WF& stowf, - const int istep, - const int iter, - std::ostream& log, - const bool skip_charge) -{ - ModuleBase::TITLE("HSolverPW_SDFT", "solve"); - ModuleBase::timer::start("HSolverPW_SDFT", "solve"); - - // This override never calls HSolverPW::solve, which is where the base class - // normally establishes the pool communication context. Set it up here so that - // hamiltSolvePsiK builds a diag_comm_info describing the real pool; otherwise it - // would keep the defaults (rank 0, 1 process) and every reduction guarded by - // diag_comm.nproc > 1 would be silently skipped. - this->rank_in_pool = wfc_basis->poolrank; - this->nproc_in_pool = wfc_basis->poolnproc; - - const int npwx = psi.get_nbasis(); - const int nbands = psi.get_nbands(); - const int nks = psi.get_nk(); - - // prepare for the precondition of diagonalization - std::vector precondition(psi.get_nbasis(), 0.0); - - this->ethr_band.resize(psi.get_nbands(), this->diag_thr); - - // report if the specified diagonalization method is not supported - const std::initializer_list _methods = {"cg", "dav", "dav_subspace", "bpcg"}; - if (std::find(std::begin(_methods), std::end(_methods), this->method) == std::end(_methods)) - { - ModuleBase::WARNING_QUIT("HSolverPW::solve", "This type of eigensolver is not supported!"); - } - - // part of KSDFT to get KS orbitals - for (int ik = 0; ik < nks; ++ik) - { - ModuleBase::timer::start("HSolverPW_SDFT", "solve_KS"); - pHamilt->updateHk(ik); - if (nbands > 0 && this->ks_run) - { - /// update psi pointer for each k point - psi.fix_k(ik); - /// template add precondition calculating here - this->update_precondition(precondition, ik, this->wfc_basis->npwk[ik], pes->pot->get_vl_of_0()); - /// solve eigenvector and eigenvalue for H(k) - double* p_eigenvalues = &(pes->ekb(ik, 0)); - this->hamiltSolvePsiK(pHamilt, psi, precondition, p_eigenvalues, nks); - } - -#ifdef __MPI - if (nbands > 0 && !this->all_ks_run) - { - Parallel_Common::bcast_dev(&psi(ik, 0, 0), npwx * nbands, BP_WORLD, 0, &psi_cpu(ik, 0, 0)); - MPI_Bcast(&pes->ekb(ik, 0), nbands, MPI_DOUBLE, 0, BP_WORLD); - } -#endif - ModuleBase::timer::end("HSolverPW_SDFT", "solve_KS"); - stoiter.orthog(ik, psi, stowf); - stoiter.checkemm(ik, istep, iter, stowf); // check and reset emax & emin - } - - this->output_iterInfo(log); - - for (int ik = 0; ik < nks; ik++) - { - // init k - if (nks > 1) - { - pHamilt->updateHk(ik); // necessary , because emax and emin should be decided first - } - stoiter.calPn(ik, stowf); - } - - // iterate to get mu - stoiter.itermu(iter, pes); - - // prepare sqrt{f(\hat{H})}|\chi> to calculate density, force and stress - stoiter.calHsqrtchi(stowf); - - // calculate eband = \sum_{ik,ib} w(ik)f(ik,ib)e_{ikib}, demet = -TS - elecstate::ElecStatePW* pes_pw = static_cast*>(pes); - elecstate::calEBand(pes_pw->ekb,pes_pw->wg,pes_pw->f_en); - if(!this->all_ks_run) - { - pes->f_en.eband /= this->bndpar; - } - stoiter.sum_stoeband(stowf, pes_pw, pHamilt, wfc_basis); - - - - // for nscf, skip charge - if (skip_charge) - { - ModuleBase::timer::end("HSolverPW_SDFT", "solve"); - return; - } - - //(5) calculate new charge density - // calculate KS rho. - pes_pw->init_rho_data(); - if (nbands > 0) - { - pes_pw->psiToRho(psi); - } - // calculate stochastic rho - stoiter.cal_storho(ucell, stowf, pes_pw,wfc_basis); - - // will do rho symmetry and energy calculation in esolver - ModuleBase::timer::end("HSolverPW_SDFT", "solve"); - return; -} - -// template class HSolverPW_SDFT, base_device::DEVICE_CPU>; -template class HSolverPW_SDFT, base_device::DEVICE_CPU>; -#if ((defined __CUDA) || (defined __ROCM)) -// template class HSolverPW_SDFT, base_device::DEVICE_GPU>; -template class HSolverPW_SDFT, base_device::DEVICE_GPU>; -#endif -} // namespace hsolver diff --git a/source/source_hsolver/hsolver_pw_sdft.h b/source/source_hsolver/hsolver_pw_sdft.h deleted file mode 100644 index 7c88a5a2b3b..00000000000 --- a/source/source_hsolver/hsolver_pw_sdft.h +++ /dev/null @@ -1,84 +0,0 @@ -#ifndef HSOLVERPW_SDFT_H -#define HSOLVERPW_SDFT_H -#include "hsolver_pw.h" -#include "source_pw/module_stodft/hamilt_sdft_pw.h" -#include "source_pw/module_stodft/sto_iter.h" -namespace hsolver -{ -template -class HSolverPW_SDFT : public HSolverPW -{ - protected: - using Real = typename GetTypeReal::type; - - public: - HSolverPW_SDFT(K_Vectors* pkv, - ModulePW::PW_Basis_K* wfc_basis_in, - Stochastic_WF& stowf, - StoChe& stoche, - hamilt::HamiltSdftPW* p_hamilt_sto, - const std::string calculation_type_in, - const std::string basis_type_in, - const std::string method_in, - const bool use_uspp_in, - const int nspin_in, - const int scf_iter_in, - const int diag_iter_max_in, - const double diag_thr_in, - const bool need_subspace_in, - const int nbands_in, - const bool diago_smooth_ethr_in, - const int pw_diag_ndim_in, - const int diag_subspace_in, - const int nb2d_in, - const bool ks_run_in, - const bool all_ks_run_in, - const int bndpar_in) - : HSolverPW(wfc_basis_in, - calculation_type_in, - basis_type_in, - method_in, - use_uspp_in, - nspin_in, - scf_iter_in, - diag_iter_max_in, - diag_thr_in, - need_subspace_in, - nbands_in, - diago_smooth_ethr_in, - pw_diag_ndim_in, - diag_subspace_in, - nb2d_in), - ks_run(ks_run_in), all_ks_run(all_ks_run_in), bndpar(bndpar_in) - { - stoiter.init(pkv, wfc_basis_in, stowf, stoche, p_hamilt_sto); - } - - void solve(const UnitCell& ucell, - hamilt::Hamilt* pHamilt, - psi::Psi& psi, - psi::Psi& psi_cpu, - elecstate::ElecState* pes, - ModulePW::PW_Basis_K* wfc_basis, - Stochastic_WF& stowf, - const int istep, - const int iter, - std::ostream& log, - const bool skip_charge); - - Stochastic_Iter stoiter; - - protected: - const bool ks_run; // true if the current process runs the KS part of the SDFT calculation - const bool all_ks_run; // true if every process runs the KS part - const int bndpar; // number of band-parallel groups - - using setmem_complex_op = base_device::memory::set_memory_op; - using setmem_var_op = base_device::memory::set_memory_op; - using syncmem_h2d_op = base_device::memory::synchronize_memory_op; - using syncmem_d2h_op = base_device::memory::synchronize_memory_op; - using syncmem_var_h2d_op = base_device::memory::synchronize_memory_op; - using syncmem_var_d2h_op = base_device::memory::synchronize_memory_op; -}; -} // namespace hsolver -#endif diff --git a/source/source_hsolver/hsolver_pw_tddft.cpp b/source/source_hsolver/hsolver_pw_tddft.cpp new file mode 100644 index 00000000000..3b093f250de --- /dev/null +++ b/source/source_hsolver/hsolver_pw_tddft.cpp @@ -0,0 +1,635 @@ +#include "source_hsolver/hsolver_pw_tddft.h" + +#include "source_base/module_device/memory_op.h" +#include "source_base/parallel_device.h" +#include "source_base/timer.h" +#include "source_base/tool_quit.h" +#include "source_basis/module_pw/pw_momentum.h" +#include "source_hsolver/kernels/linear_op.h" + +#include +#include +#include + +namespace hsolver +{ +namespace +{ + +struct LinearMethodName +{ + const char* name; + LinearMethod method; +}; + +constexpr LinearMethodName linear_method_names[] + = {{"bicgstab", LinearMethod::bicgstab}, {"cgs", LinearMethod::cgs}, {"gmres", LinearMethod::gmres}}; + +struct PWPrecondName +{ + const char* name; + PWPreconditioner preconditioner; +}; + +constexpr PWPrecondName pw_precond_names[] = {{"none", PWPreconditioner::none}, + {"kinetic", PWPreconditioner::kinetic}, + {"kinetic_recycle", PWPreconditioner::kinetic_recycle}, + {"kinetic_subspace", PWPreconditioner::kinetic_subspace}}; + +template +class ShiftedHOperator final : public LinearOperator +{ + public: + ShiftedHOperator(const HSOperator& op, const T coefficient, const int dim) : op_(op), coefficient_(coefficient), dim_(dim) + { + } + void apply(const T* x, T* y, const int ld, const int nvec) const override + { + op_.hpsi(x, y, ld, nvec); + linear_op().batch(ld, dim_, nvec, y, x, y, T(1), coefficient_, nullptr, nullptr, nullptr); + } + + private: + const HSOperator& op_; + const T coefficient_; + const int dim_; +}; + +} // namespace + +LinearMethod parse_linear_method(const std::string& name) +{ + for (const LinearMethodName& entry: linear_method_names) + { + if (name == entry.name) + { + return entry.method; + } + } + const std::string message = "Unsupported linear solver: " + name; + ModuleBase::WARNING_QUIT("HSolverPWTDDFT", message); + return LinearMethod::bicgstab; +} + +PWPreconditioner parse_pw_precond(const std::string& name) +{ + for (const PWPrecondName& entry: pw_precond_names) + { + if (name == entry.name) + { + return entry.preconditioner; + } + } + const std::string message = "Unsupported preconditioner: " + name; + ModuleBase::WARNING_QUIT("HSolverPWTDDFT", message); + return PWPreconditioner::kinetic; +} + +template +HSolverPWTDDFT::HSolverPWTDDFT(const ModulePW::PW_Basis_K& basis, + const PWLinearOptions& options, + const diag_comm_info& comm, + std::ostream& log) + : basis_(basis), comm_(comm), options_(options), algebra_(comm), orthonormal_(comm, algebra_), log_(log), band_products_(comm, 9) +{ + initialize(); + log_ << " PW RT-TDDFT orthonormalization: " << orth_method_name(options_.orthonormal) << '\n'; +} + +template +void HSolverPWTDDFT::initialize() +{ + const char* method = nullptr; + for (const LinearMethodName& entry: linear_method_names) + { + if (options_.linear.method == entry.method) + { + method = entry.name; + break; + } + } + if (method == nullptr) + { + ModuleBase::WARNING_QUIT("HSolverPWTDDFT", "Unsupported linear solver."); + return; + } + const char* preconditioner = nullptr; + for (const PWPrecondName& entry: pw_precond_names) + { + if (options_.preconditioner == entry.preconditioner) + { + preconditioner = entry.name; + break; + } + } + if (preconditioner == nullptr) + { + ModuleBase::WARNING_QUIT("HSolverPWTDDFT", "Unsupported preconditioner."); + return; + } + if (options_.linear.method != LinearMethod::gmres) + { + // Ignore the GMRES-only option before creating solver state or reporting effective settings. + options_.linear.reconstruct = false; + } + linear_solver_.reset(new HSolverLinear(options_.linear, comm_)); + std::ostringstream info; + info << " PW RT-TDDFT linear solver: " << method << "; tolerance: " << std::scientific << std::setprecision(6) + << linear_solver_->tolerance() << "; maximum iterations: " << options_.linear.max_iterations + << "; restart: " << options_.linear.restart << '\n' + << " Preconditioner: " << preconditioner << "; CN initial guess: " << options_.cn_init + << "; residual reconstruction: " << options_.linear.reconstruct << '\n'; + log_ << info.str(); +} + +template +void HSolverPWTDDFT::prepare_buffers(const int nbands, const int nbasis) +{ + ModuleBase::timer::start("HSolverPWTDDFT", "prepare_buffers"); + using CtDevice = typename ct::PsiToContainer::type; + const ct::DeviceType device = ct::DeviceTypeToEnum::value; + const int64_t elements = static_cast(nbands) * nbasis; + const int64_t size = std::max(1, elements); + if (rhs_.NumElements() < size || rhs_.data_type() != ct::DataTypeToEnum::value || rhs_.device_type() != device) + { + rhs_ = ct::Tensor(ct::DataTypeToEnum::value, device, {size}); + hpsi_ = ct::Tensor(ct::DataTypeToEnum::value, device, {size}); + } + ModuleBase::timer::end("HSolverPWTDDFT", "prepare_buffers"); +} + +template +void HSolverPWTDDFT::update_precond(const int ik, + const int dim, + const T coefficient, + const ModuleBase::Vector3& momentum_shift) +{ + ModuleBase::timer::start("HSolverPWTDDFT", "update_precond"); + using CtDevice = typename ct::PsiToContainer::type; + const ct::DeviceType device = ct::DeviceTypeToEnum::value; + const int elements = std::max(1, dim); + if (inverse_kinetic_.NumElements() < elements || inverse_kinetic_.data_type() != ct::DataTypeToEnum::value + || inverse_kinetic_.device_type() != device) + { + inverse_kinetic_ = ct::Tensor(ct::DataTypeToEnum::value, device, {elements}); + } + std::vector inverse(dim); + for (int ig = 0; ig < dim; ++ig) + { + const double kinetic = options_.kinetic_enabled ? ModulePW::shifted_kinetic(basis_, ik, ig, momentum_shift) : 0.0; + inverse[ig] = T(1) / (T(1) + coefficient * static_cast(kinetic)); + } + if (dim > 0) + { + base_device::memory::synchronize_memory_op()(inverse_kinetic_.template data(), + inverse.data(), + dim); + } + ModuleBase::timer::end("HSolverPWTDDFT", "update_precond"); +} + +template +bool HSolverPWTDDFT::tracks_state() const +{ + return options_.preconditioner == PWPreconditioner::kinetic_recycle || options_.linear.reconstruct; +} + +template +void HSolverPWTDDFT::prepare_sequence(int nk, int ld, int bands, double dt, int step, int iteration) +{ + if (!tracks_state()) + { + return; + } + const bool reset = states_.size() != static_cast(nk) || dt != sequence_.dt || ld != sequence_.ld || bands != sequence_.bands + || step < sequence_.step || (step == sequence_.step && iteration <= sequence_.iteration); + if (reset) + { + states_.clear(); + states_.resize(nk); + } + sequence_.dt = dt; + sequence_.ld = ld; + sequence_.bands = bands; + sequence_.step = step; + sequence_.iteration = iteration; +} + +template +void HSolverPWTDDFT::invalidate_basis() +{ + for (KPointState& state: states_) + { + state.response.clear(); + state.solve_count = 0; + state.independent_step = -1; + } +} + +template +bool HSolverPWTDDFT::require_audit(int step, KPointState* state) const +{ + if (!options_.linear.reconstruct) + { + return true; + } + constexpr unsigned int audit_period = 16; + const bool required = state->solve_count % audit_period == 0 || state->independent_step == step; + ++state->solve_count; + return required; +} + +template +void HSolverPWTDDFT::retry_kinetic(const LinearOperator& op, + T* current, + const SolveBatch& batch, + LinearSolveResult* result) +{ + const int remaining_iterations = options_.linear.max_iterations - result->iterations; + const LinearSolveControl control{remaining_iterations, false}; + const T* inverse = inverse_kinetic_.template data(); + const LinearLowRank diagonal(algebra_, inverse, batch.dim); + const T* rhs = rhs_.template data(); + LinearSolveResult retry = linear_solver_->solve(op, diagonal, batch.ld, batch.bands, batch.dim, current, rhs, nullptr, true, control); + retry.iterations += result->iterations; + retry.restarts += result->restarts; + retry.operator_calls += result->operator_calls; + retry.operator_columns += result->operator_columns; + retry.true_checks += result->true_checks; + retry.reconstruction_fallbacks += result->reconstruction_fallbacks; + *result = retry; +} + +template +void HSolverPWTDDFT::update_state(KPointState* state, + const SolveDetails& details, + int step, + const SolveBatch& batch, + const T* current) +{ + if (!state) + { + return; + } + if (details.retried || details.linear.reconstruction_fallbacks > 0) + { + state->response.clear(); + } + if (details.linear.reconstruction_fallbacks > 0) + { + state->independent_step = step; + } + if (options_.preconditioner == PWPreconditioner::kinetic_recycle && details.projected + && details.linear.status == LinearSolveStatus::converged && details.linear.reconstruction_fallbacks == 0 && !details.retried) + { + const T* seed = projection_.seed(); + const T* residual = projection_.residual(); + const double tolerance = linear_solver_->tolerance(); + state->response.update(algebra_, batch.ld, batch.dim, batch.bands, current, seed, residual, tolerance, &response_workspace_); + } +} + +template +typename HSolverPWTDDFT::SolveDetails HSolverPWTDDFT::solve_kpoint(const LinearOperator& op, + const T* previous, + T* current, + const SolveBatch& batch, + int step, + int iteration, + KPointState* state) +{ + SolveDetails details; + const T* rhs = rhs_.template data(); + const bool need_projection = (options_.cn_init && iteration == 1) || options_.preconditioner == PWPreconditioner::kinetic_recycle + || options_.preconditioner == PWPreconditioner::kinetic_subspace; + details.projected = need_projection && projection_.prepare(algebra_, batch.ld, batch.dim, batch.bands, previous, rhs); + const T* initial_residual = nullptr; + if (options_.cn_init && iteration == 1 && details.projected) + { + const T* seed = projection_.seed(); + band_products_.copy(batch.ld, batch.dim, batch.bands, seed, current); + initial_residual = projection_.residual(); + details.cn_initial = true; + } + const bool force_check = require_audit(step, state); + const T* inverse = options_.preconditioner == PWPreconditioner::none ? nullptr : inverse_kinetic_.template data(); + LinearLowRank preconditioner(algebra_, inverse, batch.dim); + if (options_.preconditioner == PWPreconditioner::kinetic_subspace && details.projected) + { + const T* image = projection_.image(); + const LinearSmallLU& factor = projection_.factor(); + preconditioner.prepare_subspace(batch.ld, batch.bands, previous, image, factor, &correction_workspace_); + } + else if (options_.preconditioner == PWPreconditioner::kinetic_recycle && state->response.rank() > 0) + { + const int rank = state->response.rank(); + const T* directions = state->response.directions(); + const T* images = state->response.images(); + preconditioner.prepare_response(batch.ld, rank, directions, images, &correction_workspace_); + } + details.coarse_rank = preconditioner.rank(); + details.linear + = linear_solver_->solve(op, preconditioner, batch.ld, batch.bands, batch.dim, current, rhs, initial_residual, force_check); + if (details.linear.status != LinearSolveStatus::converged && details.coarse_rank > 0 + && details.linear.iterations < options_.linear.max_iterations) + { + retry_kinetic(op, current, batch, &details.linear); + details.retried = true; + } + update_state(state, details, step, batch, current); + return details; +} + +template +void HSolverPWTDDFT::report_solve(const SolveDetails& details, int ik, int step, int iteration) const +{ + const LinearSolveResult& result = details.linear; + const int evolution_step = step + 1; + std::ostringstream record; + record << std::scientific << std::setprecision(6) << " PW RT-TDDFT linear solve: evolution_step=" << evolution_step + << " scf_iter=" << iteration << " global_k=" << options_.global_k_indices.at(ik) << '\n' + << " iterations=" << result.iterations << " restarts=" << result.restarts << " operator_calls=" << result.operator_calls + << " operator_columns=" << result.operator_columns << " pre_orth_residual=" << result.max_residual + << " residual_kind=" << (result.reconstructed ? "reconstructed" : "independent") << " true_checks=" << result.true_checks << '\n' + << " reconstruction_fallbacks=" << result.reconstruction_fallbacks << " coarse_rank=" << details.coarse_rank + << " kinetic_retry=" << details.retried << " cn_projected=" << details.projected << " cn_initial=" << details.cn_initial << '\n'; + log_ << record.str(); +} + +template +void HSolverPWTDDFT::solve(HSOperator& op, + const psi::Psi& previous, + psi::Psi* current, + const double dt, + const ModuleBase::Vector3& momentum_shift, + const int istep, + const int iter, + const bool detailed_output) +{ + ModuleBase::timer::start("HSolverPWTDDFT", "solve"); + const int bands = current->get_nbands(); + const int ld = current->get_nbasis(); + const int nk = current->get_nk(); + if (options_.out_stat) + { + orth_norms_.resize(nk); + } + prepare_buffers(bands, ld); + prepare_sequence(nk, ld, bands, dt, istep, iter); + T* rhs = rhs_.template data(); + // H is stored in Rydberg; CN includes the conversion to Hartree. + const T coefficient(0.0, dt / 4.0); + const T rhs_coefficient = -coefficient; + for (int ik = 0; ik < nk; ++ik) + { + op.update_k(ik); + current->fix_k(ik); + previous.fix_k(ik); + const int dim = current->get_ngk(ik); + KPointState* state = tracks_state() ? &states_[ik] : nullptr; + const ShiftedHOperator rhs_op(op, rhs_coefficient, dim); + const ShiftedHOperator lhs_op(op, coefficient, dim); + const T* previous_data = previous.get_pointer(); + T* current_data = current->get_pointer(); + rhs_op.apply(previous_data, rhs, ld, bands); + if (options_.preconditioner != PWPreconditioner::none) + { + update_precond(ik, dim, coefficient, momentum_shift); + } + const SolveBatch batch{ld, dim, bands}; + const SolveDetails details = solve_kpoint(lhs_op, previous_data, current_data, batch, istep, iter, state); + if (detailed_output && log_.good()) + { + report_solve(details, ik, istep, iter); + } + if (details.linear.status != LinearSolveStatus::converged) + { + const LinearSolveResult& result = details.linear; + std::ostringstream message; + const int evolution_step = istep + 1; + message << "PW RT-TDDFT linear solve failed at electronic evolution step " << evolution_step << ", SCF iteration " << iter + << ", global_k=" << options_.global_k_indices.at(ik) << ", band " << result.failed_band << ", after " + << result.iterations << " iterations: " << linear_status_name(result.status) << "; residual = " << result.max_residual; + ModuleBase::WARNING_QUIT("HSolverPWTDDFT", message.str()); + } + correct_orbitals(current_data, ld, dim, bands, ik, istep, iter, detailed_output); + } + ModuleBase::timer::end("HSolverPWTDDFT", "solve"); +} + +template +void HSolverPWTDDFT::cal_band_energy(HSOperator& op, const psi::Psi& current, ModuleBase::matrix* energies) +{ + ModuleBase::timer::start("HSolverPWTDDFT", "cal_band_energy"); + const int nband = current.get_nbands(); + const int ld = current.get_nbasis(); + prepare_buffers(nband, ld); + std::vector expectations(nband); + T* hpsi = hpsi_.template data(); + for (int ik = 0; ik < current.get_nk(); ++ik) + { + op.update_k(ik); + current.fix_k(ik); + op.hpsi(current.get_pointer(), hpsi, ld, nband); + band_products_.dot(ld, current.get_ngk(ik), nband, current.get_pointer(), hpsi, expectations.data()); + for (int band = 0; band < nband; ++band) + { + (*energies)(ik, band) = std::real(expectations[band]); + } + } + ModuleBase::timer::end("HSolverPWTDDFT", "cal_band_energy"); +} + +template +void HSolverPWTDDFT::reset_orth_stats() +{ + if (options_.out_stat) + { + orth_stats_ = TDOrthStats(); + } + warned_events_ = 0; +} + +template +void HSolverPWTDDFT::correct_orbitals(T* current, int ld, int dim, int bands, int ik, int istep, int iter, bool detailed_output) +{ + ModuleBase::timer::start("HSolverPWTDDFT", "correct_orbitals"); + const OrthResult result = orthonormal_.apply(current, ld, dim, bands, options_.orthonormal, options_.out_stat); + record_orth(result, ik, istep, iter); + if (detailed_output && result.gram_checked && log_.good()) + { + const int evolution_step = istep + 1; + std::ostringstream record; + record << std::scientific << std::setprecision(6) << " PW RT-TDDFT orth: evolution_step=" << evolution_step << " scf_iter=" << iter + << " global_k=" << options_.global_k_indices.at(ik) << " requested=" << orth_method_name(options_.orthonormal) + << " orth_before=" << result.before << " orth_after=" << result.after << '\n'; + log_ << record.str(); + } + ModuleBase::timer::end("HSolverPWTDDFT", "correct_orbitals"); +} + +template +void HSolverPWTDDFT::record_orth(const OrthResult& result, int ik, int istep, int iter) +{ + if (result.status == OrthStatus::failed) + { + const char* stage = istep == 0 ? "initialization" : "propagation"; + std::ostringstream message; + const int evolution_step = istep + 1; + message << std::setprecision(16) << " PW RT-TDDFT orbital validation failed (" << stage << "): evolution_step=" << evolution_step + << " scf_iter=" << iter << " global_k=" << options_.global_k_indices.at(ik) + << " requested=" << orth_method_name(options_.orthonormal) << '\n' + << " last_attempted=" << orth_method_name(result.actual) << " passes=" << result.passes + << " fallbacks=" << result.fallbacks << '\n' + << " " << orth_failure_name(result.failure) << "; " << result.reason; + if (result.gram_checked) + { + message << '\n' << " before=" << result.before << " after=" << result.after << " tolerance=" << orth_tolerance(); + } + // stdout is disabled on non-world-root ranks; a failing pool must still report its error. + if (comm_.rank == 0) + { + std::cerr << message.str() << std::endl; + } +#ifdef __MPI + // Let the pool root flush its diagnostic before another rank's exit stops the MPI job. + double diagnostic_written = 1.0; + Parallel_Common::bcast_data(&diagnostic_written, 1, comm_.comm, 0); +#endif + ModuleBase::WARNING_QUIT("HSolverPWTDDFT", message.str()); + } + report_orth_warning(result, ik, istep, iter); + if (options_.out_stat) + { + orth_norms_[ik] = result.norms; + if (result.gram_checked) + { + orth_stats_.before = std::max(orth_stats_.before, result.before); + orth_stats_.after = std::max(orth_stats_.after, result.after); + } + } +} + +template +void HSolverPWTDDFT::report_orth_warning(const OrthResult& result, int ik, int istep, int iter) +{ + if (comm_.rank != 0) + { + return; + } + unsigned int events = 0; + if (result.fallbacks > 0) + { + events |= 1; + } + if (result.rejected > 0) + { + events |= 2; + } + const unsigned int fresh_events = events & ~warned_events_; + if (fresh_events == 0) + { + return; + } + warned_events_ |= events; + const int evolution_step = istep + 1; + std::ostringstream message; + message << std::setprecision(16) << " PW RT-TDDFT orbital warning: evolution_step=" << evolution_step << " scf_iter=" << iter; + message << " global_k=" << options_.global_k_indices.at(ik); + message << " requested=" << orth_method_name(options_.orthonormal) << '\n' + << " before=" << result.before << " after=" << result.after << " tolerance=" << orth_tolerance() << '\n' + << " "; + message << "last_attempted=" << orth_method_name(result.actual) << ". "; + if (fresh_events & 1) + { + message << "A fallback was attempted. "; + } + if (fresh_events & 2) + { + message << "A candidate was rejected. "; + } + message << "The retained state satisfies the tolerance." << '\n' + << " " << result.reason << '\n' + << " Further events of these types in this pool are suppressed for this electronic evolution step."; + const std::string text = message.str(); + std::cerr << text << std::endl; + if (log_.good()) + { + log_ << text << std::endl; + } +} + +template +bool HSolverPWTDDFT::correct_initial(psi::Psi* current, int iter) +{ + ModuleBase::timer::start("HSolverPWTDDFT", "correct_initial"); + bool changed = false; + if (options_.orthonormal != OrthMethod::none) + { + if (options_.out_stat) + { + // Initial diagnostics describe this SCF state, not earlier discarded iterates. + orth_stats_ = TDOrthStats(); + orth_norms_.resize(current->get_nk()); + } + for (int ik = 0; ik < current->get_nk(); ++ik) + { + current->fix_k(ik); + const OrthResult result = orthonormal_.apply(current->get_pointer(), + current->get_nbasis(), + current->get_ngk(ik), + current->get_nbands(), + options_.orthonormal, + options_.out_stat); + record_orth(result, ik, 0, iter); + changed = changed || result.status == OrthStatus::accepted; + } + } + ModuleBase::timer::end("HSolverPWTDDFT", "correct_initial"); + return changed; +} + +template +void HSolverPWTDDFT::check_initial(const psi::Psi& current, int iter) +{ + ModuleBase::timer::start("HSolverPWTDDFT", "check_initial"); + const bool full_gram = options_.orthonormal != OrthMethod::none; + if (options_.out_stat) + { + orth_norms_.resize(current.get_nk()); + } + for (int ik = 0; ik < current.get_nk(); ++ik) + { + current.fix_k(ik); + const OrthResult result = orthonormal_.inspect(current.get_pointer(), + current.get_nbasis(), + current.get_ngk(ik), + current.get_nbands(), + full_gram, + options_.out_stat); + record_orth(result, ik, 0, iter); + } + ModuleBase::timer::end("HSolverPWTDDFT", "check_initial"); +} + +template +double HSolverPWTDDFT::wave_electrons(const ModuleBase::matrix& occupations) const +{ + double electrons = 0.0; + for (std::size_t ik = 0; ik < orth_norms_.size(); ++ik) + { + for (std::size_t band = 0; band < orth_norms_[ik].size(); ++band) + { + electrons += occupations(ik, band) * orth_norms_[ik][band]; + } + } + return electrons; +} + +template class HSolverPWTDDFT, base_device::DEVICE_CPU>; +template class HSolverPWTDDFT, base_device::DEVICE_CPU>; +#if defined(__CUDA) || defined(__ROCM) +template class HSolverPWTDDFT, base_device::DEVICE_GPU>; +template class HSolverPWTDDFT, base_device::DEVICE_GPU>; +#endif +} // namespace hsolver diff --git a/source/source_hsolver/hsolver_pw_tddft.h b/source/source_hsolver/hsolver_pw_tddft.h new file mode 100644 index 00000000000..ff877d595fa --- /dev/null +++ b/source/source_hsolver/hsolver_pw_tddft.h @@ -0,0 +1,169 @@ +#ifndef HSOLVER_PW_TDDFT_H +#define HSOLVER_PW_TDDFT_H + +#include "source_base/matrix.h" +#include "source_basis/module_pw/pw_basis_k.h" +#include "source_hsolver/cn_subspace.h" +#include "source_hsolver/hs_operator.h" +#include "source_hsolver/hsolver_linear.h" +#include "source_hsolver/linear_low_rank.h" +#include "source_hsolver/orthonormal.h" +#include "source_psi/psi.h" + +#include +#include +#include + +namespace hsolver +{ + +enum class PWPreconditioner +{ + none, + kinetic, + kinetic_recycle, + kinetic_subspace +}; + +/** @brief Numerical and propagation options, resolved once at the input boundary. */ +struct PWLinearOptions +{ + OrthMethod orthonormal = OrthMethod::cholesky; + LinearSolveOptions linear; + PWPreconditioner preconditioner = PWPreconditioner::kinetic; + bool cn_init = false; + bool kinetic_enabled = true; + bool out_stat = false; + std::vector global_k_indices; ///< Required zero-based global index for every local k point. +}; + +LinearMethod parse_linear_method(const std::string& name); +/** @brief Independent maxima over all SCF iterations and local k points in one electronic step. + * Values are replicated within each pool and reduced across pools for step-end output. + * The before and after maxima need not come from the same correction. + */ +struct TDOrthStats +{ + double before = 0.0; + double after = 0.0; +}; +PWPreconditioner parse_pw_precond(const std::string& name); + +/** @brief PW Crank-Nicolson solves and endpoint Hamiltonian expectations. */ +template +class HSolverPWTDDFT +{ + public: + /** @brief Configure propagation using typed, explicit solver options. */ + HSolverPWTDDFT(const ModulePW::PW_Basis_K& basis, const PWLinearOptions& options, const diag_comm_info& comm, std::ostream& log); + + /** @brief Invalidate basis-dependent history and restart independent residual checks without releasing buffers. + * Call on every rank in the pool after changing the basis or its distribution, including + * G-vector ordering or k-point changes that preserve array sizes. Ordinary Hamiltonian + * updates and ionic motion at fixed basis do not require this notification. + */ + void invalidate_basis(); + /** @brief Reset diagnostics without invalidating preconditioner history. */ + void reset_orth_stats(); + /** @brief Correct initial orbitals when needed; return whether this pool changed any orbital. */ + bool correct_initial(psi::Psi* current, int iter); + /** @brief Check the converged initial state once; finite orthogonality errors are diagnostic only. */ + void check_initial(const psi::Psi& current, int iter); + /** @brief Occupation-weighted electron count for the local k-point pool. */ + double wave_electrons(const ModuleBase::matrix& occupations) const; + const TDOrthStats& orth_stats() const + { + return orth_stats_; + } + + /** @brief Propagate from the fixed previous step, retaining the current iterate as the initial guess. + * @param dt Electronic time step in Hartree atomic units. + * @param momentum_shift Propagation vector potential in Hartree atomic units (inverse Bohr). + * @param istep Electronic propagation step, increasing across electronic substeps within an MD step. + * @param iter SCF iteration within this electronic step. + */ + void solve(HSOperator& op, + const psi::Psi& previous, + psi::Psi* current, + const double dt, + const ModuleBase::Vector3& momentum_shift, + const int istep, + const int iter, + const bool detailed_output); + + /** @brief Evaluate band expectations after the caller restores the endpoint Hamiltonian. */ + void cal_band_energy(HSOperator& op, const psi::Psi& current, ModuleBase::matrix* energies); + + private: + using Real = typename GetTypeReal::type; + struct KPointState + { + LinearResponse response; + unsigned int solve_count = 0; + int independent_step = -1; + }; + struct SequenceState + { + double dt = -1.0; + int step = -1; + int iteration = -1; + int ld = -1; + int bands = -1; + }; + struct SolveBatch + { + int ld; + int dim; + int bands; + }; + struct SolveDetails + { + LinearSolveResult linear; + int coarse_rank = 0; + bool projected = false; + bool cn_initial = false; + bool retried = false; + }; + const ModulePW::PW_Basis_K& basis_; + const diag_comm_info comm_; + PWLinearOptions options_; + LinearAlgebra algebra_; + Orthonormal orthonormal_; + TDOrthStats orth_stats_; + std::vector> orth_norms_; + std::ostream& log_; + unsigned int warned_events_ = 0; + CNSubspace projection_; + std::vector states_; + SequenceState sequence_; + std::unique_ptr> linear_solver_; + LinearWorkspace band_products_; + ct::Tensor rhs_; + ct::Tensor hpsi_; + ct::Tensor inverse_kinetic_; + ct::Tensor response_workspace_; + ct::Tensor correction_workspace_; + + void initialize(); + bool tracks_state() const; + void prepare_sequence(int nk, int ld, int bands, double dt, int step, int iteration); + bool require_audit(int step, KPointState* state) const; + SolveDetails solve_kpoint(const LinearOperator& op, + const T* previous, + T* current, + const SolveBatch& batch, + int step, + int iteration, + KPointState* state); + void retry_kinetic(const LinearOperator& op, T* current, const SolveBatch& batch, LinearSolveResult* result); + void update_state(KPointState* state, const SolveDetails& details, int step, const SolveBatch& batch, const T* current); + void report_solve(const SolveDetails& details, int ik, int step, int iteration) const; + void prepare_buffers(const int nbands, const int nbasis); + void update_precond(const int ik, const int dim, const T coefficient, const ModuleBase::Vector3& momentum_shift); + void correct_orbitals(T* current, int ld, int dim, int bands, int ik, int istep, int iter, bool detailed_output); + void record_orth(const OrthResult& result, int ik, int istep, int iter); + void report_orth_warning(const OrthResult& result, int ik, int istep, int iter); +}; + +} // namespace hsolver +#endif diff --git a/source/source_hsolver/kernels/cuda/diag_cusolvermp.cuh b/source/source_hsolver/kernels/cuda/diag_cusolvermp.cuh index f8cee02c211..ec5253b2b94 100644 --- a/source/source_hsolver/kernels/cuda/diag_cusolvermp.cuh +++ b/source/source_hsolver/kernels/cuda/diag_cusolvermp.cuh @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_HSOLVER_KERNELS_CUDA_DIAG_CUSOLVERMP_CUH +#define ABACUS_SOURCE_HSOLVER_KERNELS_CUDA_DIAG_CUSOLVERMP_CUH + #include "mpi.h" #include @@ -72,3 +74,5 @@ class Diag_CusolverMP_gvd int64_t matrix_i; int64_t matrix_j; }; + +#endif // ABACUS_SOURCE_HSOLVER_KERNELS_CUDA_DIAG_CUSOLVERMP_CUH diff --git a/source/source_hsolver/kernels/cuda/linear_op.cu b/source/source_hsolver/kernels/cuda/linear_op.cu new file mode 100644 index 00000000000..fd59d2d8d29 --- /dev/null +++ b/source/source_hsolver/kernels/cuda/linear_op.cu @@ -0,0 +1,413 @@ +#include "source_hsolver/kernels/linear_op.h" + +#include +#include +#include +#include +#include + +namespace hsolver +{ +namespace +{ +constexpr int linear_threads = 256; + +void check_launch() +{ + const cudaError_t error = cudaGetLastError(); + if (error != cudaSuccess) + { + throw std::runtime_error(cudaGetErrorString(error)); + } +} + +template +__global__ void dot_kernel(const int ld, const int dim, const thrust::complex* x, + const thrust::complex* y, thrust::complex* out) +{ + __shared__ Real re[linear_threads]; + __shared__ Real im[linear_threads]; + const int tid = threadIdx.x; + const std::int64_t offset = static_cast(blockIdx.x) * ld; + thrust::complex sum(0, 0); + for (int i = tid; i < dim; i += blockDim.x) + { + sum += thrust::conj(x[offset + i]) * y[offset + i]; + } + re[tid] = sum.real(); + im[tid] = sum.imag(); + __syncthreads(); + for (int stride = blockDim.x / 2; stride > 0; stride /= 2) + { + if (tid < stride) + { + re[tid] += re[tid + stride]; + im[tid] += im[tid + stride]; + } + __syncthreads(); + } + if (tid == 0) + { + out[blockIdx.x] = thrust::complex(re[0], im[0]); + } +} +} // namespace + +template +void linear_op::dot(const int ld, const int dim, const int nvec, + const T* x, const T* y, T* out) const +{ + if (nvec <= 0) { return; } + using Complex = thrust::complex; + dot_kernel<<>>(ld, dim, reinterpret_cast(x), + reinterpret_cast(y), reinterpret_cast(out)); + check_launch(); +} + + +template +__global__ void batch_kernel(int ld,int dim,C* out,const C* x,const C* y,C a,C b,const C* ca,const C* cb,const int* skip) +{ + const int i=blockIdx.x*blockDim.x+threadIdx.x; + const int band=blockIdx.y; + if(i +__global__ void gmres_update_kernel(int ld, + int dim, + C* solution, + C* residual, + const C* direction, + const C* image, + const C* coefficients) +{ + const int i = blockIdx.x * blockDim.x + threadIdx.x; + const int band = blockIdx.y; + if (i >= dim) + { + return; + } + const C coefficient = coefficients[band]; + if (coefficient == C(0)) + { + return; + } + const std::int64_t index = static_cast(band) * ld + i; + solution[index] += coefficient * direction[index]; + if (residual) + { + residual[index] -= coefficient * image[index]; + } +} + +template +__global__ void dots_kernel(int ld,int dim,int nvec,int tiles, + const thrust::complex* x,const thrust::complex* y, + const thrust::complex* z,const thrust::complex* w,thrust::complex* partial) +{ + __shared__ Real re[256]; + __shared__ Real im[256]; + const int tid=threadIdx.x; + const int band=blockIdx.x; + const int pair=blockIdx.z; + const thrust::complex* a=pair ? z : x; + const thrust::complex* b=pair ? w : y; + thrust::complex sum(0,0); + for(int i=blockIdx.y*256+tid;i0;n/=2) { if(tid(re[0],im[0]); +} +template +__global__ void sum_tiles(int n,int tiles,const C* partial,C* out) +{ + const int i=blockIdx.x*blockDim.x+threadIdx.x; + if(i +__global__ void gather_kernel(int ld,int dim,int stride,const C* in,C* out,const int* map) +{ + const int i=blockIdx.x*blockDim.x+threadIdx.x; + if (i < dim) + { + const std::int64_t slot_offset = static_cast(blockIdx.z) * stride; + const std::int64_t source_offset = slot_offset + static_cast(map[blockIdx.y]) * ld; + const std::int64_t destination_offset = slot_offset + static_cast(blockIdx.y) * ld; + out[destination_offset + i] = in[source_offset + i]; + } +} +template +void linear_op::batch(int ld,int dim,int nvec,T* out,const T* x,const T* y, + T a,T b,const T* ca,const T* cb,const int* skip) const +{ + if(dim<=0 || nvec<=0)return; + using C=thrust::complex; + batch_kernel<<>>(ld,dim,reinterpret_cast(out), + reinterpret_cast(x),reinterpret_cast(y),C(a.real(),a.imag()),C(b.real(),b.imag()), + reinterpret_cast(ca),reinterpret_cast(cb),skip); + check_launch(); +} +template +void linear_op::gmres_update(int ld, + int dim, + int nvec, + T* solution, + T* residual, + const T* direction, + const T* image, + const T* coefficients) const +{ + if (dim <= 0 || nvec <= 0) + { + return; + } + using C = thrust::complex; + const dim3 grid((dim + linear_threads - 1) / linear_threads, nvec); + C* current_solution = reinterpret_cast(solution); + C* current_residual = reinterpret_cast(residual); + const C* current_direction = reinterpret_cast(direction); + const C* current_image = reinterpret_cast(image); + const C* coeff = reinterpret_cast(coefficients); + gmres_update_kernel<<>>(ld, dim, current_solution, current_residual, current_direction, current_image, coeff); + check_launch(); +} + +template +void linear_op::dots(int ld,int dim,int nvec,const T* x,const T* y,const T* z,const T* w, + T* out,T* partial,int tiles) const +{ + if(nvec<=0)return; + using C=thrust::complex; + const int pairs=z ? 2 : 1; + C* target=reinterpret_cast(tiles==1 ? out : partial); + dots_kernel<<>>(ld,dim,nvec,tiles,reinterpret_cast(x), + reinterpret_cast(y),reinterpret_cast(z),reinterpret_cast(w),target); + check_launch(); + if(tiles>1) + { + sum_tiles<<<(pairs*nvec+255)/256,256>>>(pairs*nvec,tiles,target,reinterpret_cast(out)); + check_launch(); + } +} +template +void linear_op::gather(int ld,int dim,int nvec,int slots,int stride,const T* in,T* out,const int* map) const +{ + if(dim<=0 || nvec<=0)return; + using C=thrust::complex; + gather_kernel<<>>(ld,dim,stride,reinterpret_cast(in),reinterpret_cast(out),map); + check_launch(); +} + + +template +__global__ void swaps_kernel(int ld,int dim,int stride,int count,C* vectors,const int* pairs) +{ + const int i=blockIdx.x*blockDim.x+threadIdx.x; + if (i >= dim) + { + return; + } + const std::int64_t offset = static_cast(blockIdx.y) * stride + i; + for(int k=0;k(pairs[2 * k]) * ld; + const std::int64_t second = offset + static_cast(pairs[2 * k + 1]) * ld; + C& a = vectors[first]; + C& b = vectors[second]; + const C value=a; a=b; b=value; + } +} +template +void linear_op::swaps(int ld,int dim,int slots,int stride,int count,T* vectors,const int* pairs) const +{ + if(dim==0 || count==0)return; + using C=thrust::complex; + swaps_kernel<<>>(ld,dim,stride,count,reinterpret_cast(vectors),pairs); + check_launch(); +} + + +template +__global__ void bicg_update_kernel(int ld,int dim,bool direction,C* out,const C* x,const C* y,const C* a,const C* b) +{ + const int i=blockIdx.x*blockDim.x+threadIdx.x; + const int band=blockIdx.y; + if(i +void linear_op::bicg_update(int ld,int dim,int nvec,bool direction,T* out,const T* x,const T* y,const T* a,const T* b) const +{ + if(dim==0 || nvec==0)return; + using C=thrust::complex; + bicg_update_kernel<<>>(ld,dim,direction,reinterpret_cast(out), + reinterpret_cast(x),reinterpret_cast(y),reinterpret_cast(a),reinterpret_cast(b)); + check_launch(); +} + + +template +__global__ void diagonal_kernel(int ld, int dim, const C* inverse, const C* x, C* y) +{ + const int i = blockIdx.x * blockDim.x + threadIdx.x; + if (i < dim) y[blockIdx.y * ld + i] = inverse[i] * x[blockIdx.y * ld + i]; +} + +template +__global__ void cgs_direction_kernel(int ld, int dim, C* p, C* u, const C* r, const C* q, const C* beta) +{ + const int i = blockIdx.x * blockDim.x + threadIdx.x; + const int band = blockIdx.y; + if (i < dim) + { + const int j = band * ld + i; + u[j] = r[j] + beta[band] * q[j]; + p[j] = u[j] + beta[band] * (q[j] + beta[band] * p[j]); + } +} + +template +__global__ void cgs_alpha_kernel(int ld, int dim, C* q, C* direction, const C* u, const C* v, const C* alpha) +{ + const int i = blockIdx.x * blockDim.x + threadIdx.x; + const int band = blockIdx.y; + if (i < dim) + { + const int j = band * ld + i; + q[j] = u[j] - alpha[band] * v[j]; + direction[j] = u[j] + q[j]; + } +} + +template +__global__ void cgs_finish_kernel(int ld, int dim, C* x, C* r, const C* direction, const C* ad, const C* alpha) +{ + const int i = blockIdx.x * blockDim.x + threadIdx.x; + const int band = blockIdx.y; + if (i < dim) + { + const int j = band * ld + i; + x[j] += alpha[band] * direction[j]; + r[j] -= alpha[band] * ad[j]; + } +} + +template +void linear_op::diagonal(int ld, int dim, int nvec, const T* inverse, const T* x, T* y) const +{ + if (dim == 0 || nvec == 0) return; + using C = thrust::complex; + diagonal_kernel<<>>(ld, dim, + reinterpret_cast(inverse), + reinterpret_cast(x), + reinterpret_cast(y)); + check_launch(); +} + +template +void linear_op::cgs_direction(int ld, int dim, int nvec, T* p, T* u, const T* r, const T* q, const T* beta) const +{ + if (dim == 0 || nvec == 0) return; + using C = thrust::complex; + cgs_direction_kernel<<>>(ld, dim, + reinterpret_cast(p), + reinterpret_cast(u), + reinterpret_cast(r), + reinterpret_cast(q), + reinterpret_cast(beta)); + check_launch(); +} + +template +void linear_op::cgs_alpha(int ld, int dim, int nvec, T* q, T* direction, const T* u, const T* v, const T* alpha) const +{ + if (dim == 0 || nvec == 0) return; + using C = thrust::complex; + cgs_alpha_kernel<<>>(ld, dim, + reinterpret_cast(q), + reinterpret_cast(direction), + reinterpret_cast(u), + reinterpret_cast(v), + reinterpret_cast(alpha)); + check_launch(); +} + +template +void linear_op::cgs_finish(int ld, int dim, int nvec, T* x, T* r, const T* direction, const T* ad, const T* alpha) const +{ + if (dim == 0 || nvec == 0) return; + using C = thrust::complex; + cgs_finish_kernel<<>>(ld, dim, + reinterpret_cast(x), + reinterpret_cast(r), + reinterpret_cast(direction), + reinterpret_cast(ad), + reinterpret_cast(alpha)); + check_launch(); +} +namespace +{ +template +__global__ void wide_dot_kernel(int ld, int dim, int nvec, int stride, + const thrust::complex* basis, const thrust::complex* x, + thrust::complex* out) +{ + __shared__ Accumulator re[linear_threads]; + __shared__ Accumulator im[linear_threads]; + const int tid = threadIdx.x; + const int band = blockIdx.x % nvec; + const int j = blockIdx.x / nvec; + const std::int64_t band_offset = static_cast(band) * ld; + const std::int64_t basis_offset = static_cast(j) * stride + band_offset; + thrust::complex sum(0, 0); + for (int i = tid; i < dim; i += blockDim.x) + { + sum += thrust::conj(thrust::complex(basis[basis_offset + i])) + * thrust::complex(x[band_offset + i]); + } + re[tid] = sum.real(); + im[tid] = sum.imag(); + __syncthreads(); + for (int s = blockDim.x / 2; s > 0; s /= 2) + { + if (tid < s) { re[tid] += re[tid + s]; im[tid] += im[tid + s]; } + __syncthreads(); + } + if (tid == 0) out[blockIdx.x] = thrust::complex(re[0], im[0]); +} +} + +template +void linear_op::wide_dots(int ld, int dim, int nvec, int count, int stride, + const T* basis, const T* x, std::complex* out) const +{ + if (nvec * count == 0) return; + using C = thrust::complex; + wide_dot_kernel<<>>(ld, dim, nvec, stride, + reinterpret_cast(basis), reinterpret_cast(x), + reinterpret_cast*>(out)); + check_launch(); +} +template +void linear_op::native_dots(int ld, int dim, int nvec, int count, int stride, + const T* basis, const T* x, T* out) const +{ + if (nvec * count == 0) return; + using C = thrust::complex; + wide_dot_kernel<<>>(ld, dim, nvec, stride, + reinterpret_cast(basis), reinterpret_cast(x), reinterpret_cast(out)); + check_launch(); +} + +template class linear_op, base_device::DEVICE_GPU>; +template class linear_op, base_device::DEVICE_GPU>; + +} // namespace hsolver diff --git a/source/source_hsolver/kernels/linear_op.cpp b/source/source_hsolver/kernels/linear_op.cpp new file mode 100644 index 00000000000..fcaff888ce2 --- /dev/null +++ b/source/source_hsolver/kernels/linear_op.cpp @@ -0,0 +1,297 @@ +#include "source_hsolver/kernels/linear_op.h" + +#include +#include +#ifdef _OPENMP +#include +#endif + +namespace hsolver +{ +namespace +{ +#ifdef _OPENMP +bool parallel_work(const int dim, const int nvec) +{ + return static_cast(dim) * nvec >= 32768 && !omp_in_parallel() && omp_get_max_threads() > 1; +} +#endif +} // namespace + +template +void linear_op::dot(const int ld, const int dim, const int nvec, const T* x, const T* y, T* out) const +{ + for (int band = 0; band < nvec; ++band) + { + const std::int64_t offset = static_cast(band) * ld; + T sum = T(0); + for (int i = 0; i < dim; ++i) + { + sum += std::conj(x[offset + i]) * y[offset + i]; + } + out[band] = sum; + } +} + +template +void linear_op::batch(int ld, + int dim, + int nvec, + T* out, + const T* x, + const T* y, + T a, + T b, + const T* ca, + const T* cb, + const int* skip) const +{ +#pragma omp parallel for schedule(static) if (parallel_work(dim, nvec)) + for (int band = 0; band < nvec; ++band) + { + if (skip && skip[band]) + { + continue; + } + const T aa = ca ? a * ca[band] : a; + const T bb = cb ? b * cb[band] : b; + for (int i = 0; i < dim; ++i) + { + const int j = band * ld + i; + out[j] = aa * x[j] + bb * y[j]; + } + } +} +template +void linear_op::gmres_update(int ld, + int dim, + int nvec, + T* solution, + T* residual, + const T* direction, + const T* image, + const T* coefficients) const +{ +#pragma omp parallel for schedule(static) if (parallel_work(dim, nvec)) + for (int band = 0; band < nvec; ++band) + { + const T coefficient = coefficients[band]; + if (coefficient == T(0)) + { + continue; + } + const std::int64_t offset = static_cast(band) * ld; + for (int i = 0; i < dim; ++i) + { + solution[offset + i] += coefficient * direction[offset + i]; + if (residual) + { + residual[offset + i] -= coefficient * image[offset + i]; + } + } + } +} + +template +void linear_op::dots(int ld, int dim, int nvec, const T* x, const T* y, const T* z, const T* w, T* out, T* partial, int tiles) + const +{ + // Independent tiles also expose parallelism when there are few bands. +#pragma omp parallel for collapse(2) schedule(static) if (parallel_work(dim, nvec)) + for (int band = 0; band < nvec; ++band) + { + for (int tile = 0; tile < tiles; ++tile) + { + T first = T(0); + T second = T(0); + const int begin = static_cast(dim) * tile / tiles; + const int end = static_cast(dim) * (tile + 1) / tiles; + for (int i = begin; i < end; ++i) + { + const int j = band * ld + i; + first += std::conj(x[j]) * y[j]; + if (z) + { + second += std::conj(z[j]) * w[j]; + } + } + partial[band * tiles + tile] = first; + if (z) + { + partial[(nvec + band) * tiles + tile] = second; + } + } + } + const int count = (z ? 2 : 1) * nvec; + for (int band = 0; band < count; ++band) + { + T sum = T(0); + for (int tile = 0; tile < tiles; ++tile) + { + sum += partial[band * tiles + tile]; + } + out[band] = sum; + } +} +template +void linear_op::gather(int ld, int dim, int nvec, int slots, int stride, const T* in, T* out, const int* map) const +{ + for (int slot = 0; slot < slots; ++slot) + { + const std::int64_t slot_offset = static_cast(slot) * stride; + for (int band = 0; band < nvec; ++band) + { + const std::int64_t source_offset = slot_offset + static_cast(map[band]) * ld; + const std::int64_t destination_offset = slot_offset + static_cast(band) * ld; + for (int i = 0; i < dim; ++i) + { + out[destination_offset + i] = in[source_offset + i]; + } + } + } +} + +template +void linear_op::swaps(int ld, int dim, int slots, int stride, int count, T* vectors, const int* pairs) const +{ + for (int slot = 0; slot < slots; ++slot) + { + const std::int64_t slot_offset = static_cast(slot) * stride; + for (int i = 0; i < dim; ++i) + { + for (int k = 0; k < count; ++k) + { + const std::int64_t first = slot_offset + static_cast(pairs[2 * k]) * ld + i; + const std::int64_t second = slot_offset + static_cast(pairs[2 * k + 1]) * ld + i; + std::swap(vectors[first], vectors[second]); + } + } + } +} + +template +void linear_op::bicg_update(int ld, int dim, int nvec, bool direction, T* out, const T* x, const T* y, const T* a, const T* b) + const +{ +#pragma omp parallel for collapse(2) schedule(static) if (parallel_work(dim, nvec)) + for (int band = 0; band < nvec; ++band) + { + for (int i = 0; i < dim; ++i) + { + const int j = band * ld + i; + out[j] = direction ? x[j] + a[band] * (out[j] - b[band] * y[j]) : (out[j] + a[band] * x[j]) + b[band] * y[j]; + } + } +} + +template +void linear_op::diagonal(int ld, int dim, int nvec, const T* inverse, const T* x, T* y) const +{ +#pragma omp parallel for collapse(2) schedule(static) if (parallel_work(dim, nvec)) + for (int band = 0; band < nvec; ++band) + { + for (int i = 0; i < dim; ++i) + { + y[band * ld + i] = inverse[i] * x[band * ld + i]; + } + } +} + +template +void linear_op::cgs_direction(int ld, int dim, int nvec, T* p, T* u, const T* r, const T* q, const T* beta) const +{ +#pragma omp parallel for collapse(2) schedule(static) if (parallel_work(dim, nvec)) + for (int band = 0; band < nvec; ++band) + { + for (int i = 0; i < dim; ++i) + { + const int j = band * ld + i; + u[j] = r[j] + beta[band] * q[j]; + p[j] = u[j] + beta[band] * (q[j] + beta[band] * p[j]); + } + } +} + +template +void linear_op::cgs_alpha(int ld, int dim, int nvec, T* q, T* direction, const T* u, const T* v, const T* alpha) const +{ +#pragma omp parallel for collapse(2) schedule(static) if (parallel_work(dim, nvec)) + for (int band = 0; band < nvec; ++band) + { + for (int i = 0; i < dim; ++i) + { + const int j = band * ld + i; + q[j] = u[j] - alpha[band] * v[j]; + direction[j] = u[j] + q[j]; + } + } +} + +template +void linear_op::cgs_finish(int ld, int dim, int nvec, T* x, T* r, const T* direction, const T* ad, const T* alpha) const +{ +#pragma omp parallel for collapse(2) schedule(static) if (parallel_work(dim, nvec)) + for (int band = 0; band < nvec; ++band) + { + for (int i = 0; i < dim; ++i) + { + const int j = band * ld + i; + x[j] += alpha[band] * direction[j]; + r[j] -= alpha[band] * ad[j]; + } + } +} +template +void linear_op::wide_dots(int ld, + int dim, + int nvec, + int count, + int stride, + const T* basis, + const T* x, + std::complex* out) const +{ +#ifdef _OPENMP +#pragma omp parallel for collapse(2) if (parallel_work(dim, nvec * count)) +#endif + for (int j = 0; j < count; ++j) + { + for (int b = 0; b < nvec; ++b) + { + const std::int64_t band_offset = static_cast(b) * ld; + const std::int64_t basis_offset = static_cast(j) * stride + band_offset; + std::complex sum = 0.0; + for (int i = 0; i < dim; ++i) + { + sum += std::conj(std::complex(basis[basis_offset + i])) * std::complex(x[band_offset + i]); + } + out[j * nvec + b] = sum; + } + } +} +template +void linear_op::native_dots(int ld, int dim, int nvec, int count, int stride, const T* basis, const T* x, T* out) const +{ +#ifdef _OPENMP +#pragma omp parallel for collapse(2) if (parallel_work(dim, nvec * count)) +#endif + for (int j = 0; j < count; ++j) + { + for (int band = 0; band < nvec; ++band) + { + const std::int64_t band_offset = static_cast(band) * ld; + const std::int64_t basis_offset = static_cast(j) * stride + band_offset; + T sum(0); + for (int i = 0; i < dim; ++i) + { + sum += std::conj(basis[basis_offset + i]) * x[band_offset + i]; + } + out[j * nvec + band] = sum; + } + } +} + +template class linear_op, base_device::DEVICE_CPU>; +template class linear_op, base_device::DEVICE_CPU>; + +} // namespace hsolver diff --git a/source/source_hsolver/kernels/linear_op.h b/source/source_hsolver/kernels/linear_op.h new file mode 100644 index 00000000000..e6ffe6ff3e7 --- /dev/null +++ b/source/source_hsolver/kernels/linear_op.h @@ -0,0 +1,77 @@ +#ifndef HSOLVER_KERNELS_LINEAR_OP_H +#define HSOLVER_KERNELS_LINEAR_OP_H + +#include "source_base/module_device/types.h" + +#include + +namespace hsolver +{ + +/** @brief Elementwise operations; pointers and dot outputs belong to Device. */ +template +class linear_op +{ + public: + /** @brief Skip zero-coefficient columns; a null residual disables the paired residual update. */ + void gmres_update(int ld, int dim, int nvec, T* solution, T* residual, const T* direction, const T* image, const T* coefficients) const; + /** @brief Batched products accumulated in wavefunction precision. */ + void native_dots(int ld, int dim, int nvec, int count, int stride, const T* basis, const T* x, T* out) const; + /** @brief Batched products accumulated in double precision. */ + void wide_dots(int ld, int dim, int nvec, int count, int stride, const T* basis, const T* x, std::complex* out) const; + + /** @brief Apply the same inverse diagonal to every column. */ + void diagonal(int ld, int dim, int nvec, const T* inverse, const T* x, T* y) const; + /** @brief Fused CGS search direction, alpha step, and final updates. */ + void cgs_direction(int ld, int dim, int nvec, T* p, T* u, const T* r, const T* q, const T* beta) const; + void cgs_alpha(int ld, int dim, int nvec, T* q, T* direction, const T* u, const T* v, const T* alpha) const; + void cgs_finish(int ld, int dim, int nvec, T* x, T* r, const T* direction, const T* ad, const T* alpha) const; + + /** @brief Batched updates with optional device coefficients and skip mask. */ + void batch(int ld, int dim, int nvec, T* out, const T* x, const T* y, T a, T b, const T* ca, const T* cb, const int* skip) const; + /** @brief One or two inner products using caller-owned partial sums. */ + void dots(int ld, int dim, int nvec, const T* x, const T* y, const T* z, const T* w, T* out, T* partial, int tiles) const; + /** @brief Apply an ordered list of column swaps to all vector slots. */ + void swaps(int ld, int dim, int slots, int stride, int count, T* vectors, const int* pairs) const; + /** @brief Fuse the two BiCGSTAB direction or solution updates. */ + void bicg_update(int ld, int dim, int nvec, bool direction, T* out, const T* x, const T* y, const T* a, const T* b) const; + /** @brief Gather columns using a device destination-to-source map. */ + void gather(int ld, int dim, int nvec, int slots, int stride, const T* in, T* out, const int* map) const; + void dot(const int ld, const int dim, const int nvec, const T* x, const T* y, T* out) const; +}; + +#if defined(__CUDA) || defined(__ROCM) +template +class linear_op +{ + public: + /** @brief Skip zero-coefficient columns; a null residual disables the paired residual update. */ + void gmres_update(int ld, int dim, int nvec, T* solution, T* residual, const T* direction, const T* image, const T* coefficients) const; + /** @brief Batched products accumulated in wavefunction precision. */ + void native_dots(int ld, int dim, int nvec, int count, int stride, const T* basis, const T* x, T* out) const; + /** @brief Batched products accumulated in double precision. */ + void wide_dots(int ld, int dim, int nvec, int count, int stride, const T* basis, const T* x, std::complex* out) const; + + /** @brief Apply the same inverse diagonal to every column. */ + void diagonal(int ld, int dim, int nvec, const T* inverse, const T* x, T* y) const; + /** @brief Fused CGS search direction, alpha step, and final updates. */ + void cgs_direction(int ld, int dim, int nvec, T* p, T* u, const T* r, const T* q, const T* beta) const; + void cgs_alpha(int ld, int dim, int nvec, T* q, T* direction, const T* u, const T* v, const T* alpha) const; + void cgs_finish(int ld, int dim, int nvec, T* x, T* r, const T* direction, const T* ad, const T* alpha) const; + + /** @brief Batched updates with optional device coefficients and skip mask. */ + void batch(int ld, int dim, int nvec, T* out, const T* x, const T* y, T a, T b, const T* ca, const T* cb, const int* skip) const; + /** @brief One or two inner products using caller-owned partial sums. */ + void dots(int ld, int dim, int nvec, const T* x, const T* y, const T* z, const T* w, T* out, T* partial, int tiles) const; + /** @brief Apply an ordered list of column swaps to all vector slots. */ + void swaps(int ld, int dim, int slots, int stride, int count, T* vectors, const int* pairs) const; + /** @brief Fuse the two BiCGSTAB direction or solution updates. */ + void bicg_update(int ld, int dim, int nvec, bool direction, T* out, const T* x, const T* y, const T* a, const T* b) const; + /** @brief Gather columns using a device destination-to-source map. */ + void gather(int ld, int dim, int nvec, int slots, int stride, const T* in, T* out, const int* map) const; + void dot(const int ld, const int dim, const int nvec, const T* x, const T* y, T* out) const; +}; +#endif + +} // namespace hsolver +#endif diff --git a/source/source_hsolver/kernels/rocm/linear_op.hip.cu b/source/source_hsolver/kernels/rocm/linear_op.hip.cu new file mode 100644 index 00000000000..e71c038e1e7 --- /dev/null +++ b/source/source_hsolver/kernels/rocm/linear_op.hip.cu @@ -0,0 +1,413 @@ +#include "source_hsolver/kernels/linear_op.h" + +#include +#include +#include +#include +#include + +namespace hsolver +{ +namespace +{ +constexpr int linear_threads = 256; + +void check_launch() +{ + const hipError_t error = hipGetLastError(); + if (error != hipSuccess) + { + throw std::runtime_error(hipGetErrorString(error)); + } +} + +template +__global__ void dot_kernel(const int ld, const int dim, const thrust::complex* x, + const thrust::complex* y, thrust::complex* out) +{ + __shared__ Real re[linear_threads]; + __shared__ Real im[linear_threads]; + const int tid = threadIdx.x; + const std::int64_t offset = static_cast(blockIdx.x) * ld; + thrust::complex sum(0, 0); + for (int i = tid; i < dim; i += blockDim.x) + { + sum += thrust::conj(x[offset + i]) * y[offset + i]; + } + re[tid] = sum.real(); + im[tid] = sum.imag(); + __syncthreads(); + for (int stride = blockDim.x / 2; stride > 0; stride /= 2) + { + if (tid < stride) + { + re[tid] += re[tid + stride]; + im[tid] += im[tid + stride]; + } + __syncthreads(); + } + if (tid == 0) + { + out[blockIdx.x] = thrust::complex(re[0], im[0]); + } +} +} // namespace + +template +void linear_op::dot(const int ld, const int dim, const int nvec, + const T* x, const T* y, T* out) const +{ + if (nvec <= 0) { return; } + using Complex = thrust::complex; + dot_kernel<<>>(ld, dim, reinterpret_cast(x), + reinterpret_cast(y), reinterpret_cast(out)); + check_launch(); +} + + +template +__global__ void batch_kernel(int ld,int dim,C* out,const C* x,const C* y,C a,C b,const C* ca,const C* cb,const int* skip) +{ + const int i=blockIdx.x*blockDim.x+threadIdx.x; + const int band=blockIdx.y; + if(i +__global__ void gmres_update_kernel(int ld, + int dim, + C* solution, + C* residual, + const C* direction, + const C* image, + const C* coefficients) +{ + const int i = blockIdx.x * blockDim.x + threadIdx.x; + const int band = blockIdx.y; + if (i >= dim) + { + return; + } + const C coefficient = coefficients[band]; + if (coefficient == C(0)) + { + return; + } + const std::int64_t index = static_cast(band) * ld + i; + solution[index] += coefficient * direction[index]; + if (residual) + { + residual[index] -= coefficient * image[index]; + } +} + +template +__global__ void dots_kernel(int ld,int dim,int nvec,int tiles, + const thrust::complex* x,const thrust::complex* y, + const thrust::complex* z,const thrust::complex* w,thrust::complex* partial) +{ + __shared__ Real re[256]; + __shared__ Real im[256]; + const int tid=threadIdx.x; + const int band=blockIdx.x; + const int pair=blockIdx.z; + const thrust::complex* a=pair ? z : x; + const thrust::complex* b=pair ? w : y; + thrust::complex sum(0,0); + for(int i=blockIdx.y*256+tid;i0;n/=2) { if(tid(re[0],im[0]); +} +template +__global__ void sum_tiles(int n,int tiles,const C* partial,C* out) +{ + const int i=blockIdx.x*blockDim.x+threadIdx.x; + if(i +__global__ void gather_kernel(int ld,int dim,int stride,const C* in,C* out,const int* map) +{ + const int i=blockIdx.x*blockDim.x+threadIdx.x; + if (i < dim) + { + const std::int64_t slot_offset = static_cast(blockIdx.z) * stride; + const std::int64_t source_offset = slot_offset + static_cast(map[blockIdx.y]) * ld; + const std::int64_t destination_offset = slot_offset + static_cast(blockIdx.y) * ld; + out[destination_offset + i] = in[source_offset + i]; + } +} +template +void linear_op::batch(int ld,int dim,int nvec,T* out,const T* x,const T* y, + T a,T b,const T* ca,const T* cb,const int* skip) const +{ + if(dim<=0 || nvec<=0)return; + using C=thrust::complex; + batch_kernel<<>>(ld,dim,reinterpret_cast(out), + reinterpret_cast(x),reinterpret_cast(y),C(a.real(),a.imag()),C(b.real(),b.imag()), + reinterpret_cast(ca),reinterpret_cast(cb),skip); + check_launch(); +} +template +void linear_op::gmres_update(int ld, + int dim, + int nvec, + T* solution, + T* residual, + const T* direction, + const T* image, + const T* coefficients) const +{ + if (dim <= 0 || nvec <= 0) + { + return; + } + using C = thrust::complex; + const dim3 grid((dim + linear_threads - 1) / linear_threads, nvec); + C* current_solution = reinterpret_cast(solution); + C* current_residual = reinterpret_cast(residual); + const C* current_direction = reinterpret_cast(direction); + const C* current_image = reinterpret_cast(image); + const C* coeff = reinterpret_cast(coefficients); + gmres_update_kernel<<>>(ld, dim, current_solution, current_residual, current_direction, current_image, coeff); + check_launch(); +} + +template +void linear_op::dots(int ld,int dim,int nvec,const T* x,const T* y,const T* z,const T* w, + T* out,T* partial,int tiles) const +{ + if(nvec<=0)return; + using C=thrust::complex; + const int pairs=z ? 2 : 1; + C* target=reinterpret_cast(tiles==1 ? out : partial); + dots_kernel<<>>(ld,dim,nvec,tiles,reinterpret_cast(x), + reinterpret_cast(y),reinterpret_cast(z),reinterpret_cast(w),target); + check_launch(); + if(tiles>1) + { + sum_tiles<<<(pairs*nvec+255)/256,256>>>(pairs*nvec,tiles,target,reinterpret_cast(out)); + check_launch(); + } +} +template +void linear_op::gather(int ld,int dim,int nvec,int slots,int stride,const T* in,T* out,const int* map) const +{ + if(dim<=0 || nvec<=0)return; + using C=thrust::complex; + gather_kernel<<>>(ld,dim,stride,reinterpret_cast(in),reinterpret_cast(out),map); + check_launch(); +} + + +template +__global__ void swaps_kernel(int ld,int dim,int stride,int count,C* vectors,const int* pairs) +{ + const int i=blockIdx.x*blockDim.x+threadIdx.x; + if (i >= dim) + { + return; + } + const std::int64_t offset = static_cast(blockIdx.y) * stride + i; + for(int k=0;k(pairs[2 * k]) * ld; + const std::int64_t second = offset + static_cast(pairs[2 * k + 1]) * ld; + C& a = vectors[first]; + C& b = vectors[second]; + const C value=a; a=b; b=value; + } +} +template +void linear_op::swaps(int ld,int dim,int slots,int stride,int count,T* vectors,const int* pairs) const +{ + if(dim==0 || count==0)return; + using C=thrust::complex; + swaps_kernel<<>>(ld,dim,stride,count,reinterpret_cast(vectors),pairs); + check_launch(); +} + + +template +__global__ void bicg_update_kernel(int ld,int dim,bool direction,C* out,const C* x,const C* y,const C* a,const C* b) +{ + const int i=blockIdx.x*blockDim.x+threadIdx.x; + const int band=blockIdx.y; + if(i +void linear_op::bicg_update(int ld,int dim,int nvec,bool direction,T* out,const T* x,const T* y,const T* a,const T* b) const +{ + if(dim==0 || nvec==0)return; + using C=thrust::complex; + bicg_update_kernel<<>>(ld,dim,direction,reinterpret_cast(out), + reinterpret_cast(x),reinterpret_cast(y),reinterpret_cast(a),reinterpret_cast(b)); + check_launch(); +} + + +template +__global__ void diagonal_kernel(int ld, int dim, const C* inverse, const C* x, C* y) +{ + const int i = blockIdx.x * blockDim.x + threadIdx.x; + if (i < dim) y[blockIdx.y * ld + i] = inverse[i] * x[blockIdx.y * ld + i]; +} + +template +__global__ void cgs_direction_kernel(int ld, int dim, C* p, C* u, const C* r, const C* q, const C* beta) +{ + const int i = blockIdx.x * blockDim.x + threadIdx.x; + const int band = blockIdx.y; + if (i < dim) + { + const int j = band * ld + i; + u[j] = r[j] + beta[band] * q[j]; + p[j] = u[j] + beta[band] * (q[j] + beta[band] * p[j]); + } +} + +template +__global__ void cgs_alpha_kernel(int ld, int dim, C* q, C* direction, const C* u, const C* v, const C* alpha) +{ + const int i = blockIdx.x * blockDim.x + threadIdx.x; + const int band = blockIdx.y; + if (i < dim) + { + const int j = band * ld + i; + q[j] = u[j] - alpha[band] * v[j]; + direction[j] = u[j] + q[j]; + } +} + +template +__global__ void cgs_finish_kernel(int ld, int dim, C* x, C* r, const C* direction, const C* ad, const C* alpha) +{ + const int i = blockIdx.x * blockDim.x + threadIdx.x; + const int band = blockIdx.y; + if (i < dim) + { + const int j = band * ld + i; + x[j] += alpha[band] * direction[j]; + r[j] -= alpha[band] * ad[j]; + } +} + +template +void linear_op::diagonal(int ld, int dim, int nvec, const T* inverse, const T* x, T* y) const +{ + if (dim == 0 || nvec == 0) return; + using C = thrust::complex; + diagonal_kernel<<>>(ld, dim, + reinterpret_cast(inverse), + reinterpret_cast(x), + reinterpret_cast(y)); + check_launch(); +} + +template +void linear_op::cgs_direction(int ld, int dim, int nvec, T* p, T* u, const T* r, const T* q, const T* beta) const +{ + if (dim == 0 || nvec == 0) return; + using C = thrust::complex; + cgs_direction_kernel<<>>(ld, dim, + reinterpret_cast(p), + reinterpret_cast(u), + reinterpret_cast(r), + reinterpret_cast(q), + reinterpret_cast(beta)); + check_launch(); +} + +template +void linear_op::cgs_alpha(int ld, int dim, int nvec, T* q, T* direction, const T* u, const T* v, const T* alpha) const +{ + if (dim == 0 || nvec == 0) return; + using C = thrust::complex; + cgs_alpha_kernel<<>>(ld, dim, + reinterpret_cast(q), + reinterpret_cast(direction), + reinterpret_cast(u), + reinterpret_cast(v), + reinterpret_cast(alpha)); + check_launch(); +} + +template +void linear_op::cgs_finish(int ld, int dim, int nvec, T* x, T* r, const T* direction, const T* ad, const T* alpha) const +{ + if (dim == 0 || nvec == 0) return; + using C = thrust::complex; + cgs_finish_kernel<<>>(ld, dim, + reinterpret_cast(x), + reinterpret_cast(r), + reinterpret_cast(direction), + reinterpret_cast(ad), + reinterpret_cast(alpha)); + check_launch(); +} +namespace +{ +template +__global__ void wide_dot_kernel(int ld, int dim, int nvec, int stride, + const thrust::complex* basis, const thrust::complex* x, + thrust::complex* out) +{ + __shared__ Accumulator re[linear_threads]; + __shared__ Accumulator im[linear_threads]; + const int tid = threadIdx.x; + const int band = blockIdx.x % nvec; + const int j = blockIdx.x / nvec; + const std::int64_t band_offset = static_cast(band) * ld; + const std::int64_t basis_offset = static_cast(j) * stride + band_offset; + thrust::complex sum(0, 0); + for (int i = tid; i < dim; i += blockDim.x) + { + sum += thrust::conj(thrust::complex(basis[basis_offset + i])) + * thrust::complex(x[band_offset + i]); + } + re[tid] = sum.real(); + im[tid] = sum.imag(); + __syncthreads(); + for (int s = blockDim.x / 2; s > 0; s /= 2) + { + if (tid < s) { re[tid] += re[tid + s]; im[tid] += im[tid + s]; } + __syncthreads(); + } + if (tid == 0) out[blockIdx.x] = thrust::complex(re[0], im[0]); +} +} + +template +void linear_op::wide_dots(int ld, int dim, int nvec, int count, int stride, + const T* basis, const T* x, std::complex* out) const +{ + if (nvec * count == 0) return; + using C = thrust::complex; + wide_dot_kernel<<>>(ld, dim, nvec, stride, + reinterpret_cast(basis), reinterpret_cast(x), + reinterpret_cast*>(out)); + check_launch(); +} +template +void linear_op::native_dots(int ld, int dim, int nvec, int count, int stride, + const T* basis, const T* x, T* out) const +{ + if (nvec * count == 0) return; + using C = thrust::complex; + wide_dot_kernel<<>>(ld, dim, nvec, stride, + reinterpret_cast(basis), reinterpret_cast(x), reinterpret_cast(out)); + check_launch(); +} + +template class linear_op, base_device::DEVICE_GPU>; +template class linear_op, base_device::DEVICE_GPU>; + +} // namespace hsolver diff --git a/source/source_hsolver/linear_algebra.cpp b/source/source_hsolver/linear_algebra.cpp new file mode 100644 index 00000000000..c4a7aa441c9 --- /dev/null +++ b/source/source_hsolver/linear_algebra.cpp @@ -0,0 +1,375 @@ +#include "source_hsolver/linear_algebra.h" + +#include "source_base/kernels/math_kernel_op.h" +#include "source_base/parallel_device.h" +#include "source_base/timer.h" +#include "source_hsolver/kernels/linear_op.h" + +#include +#include +#include + +namespace hsolver +{ +std::vector> linear_gram_basis(const std::vector>& gram, int n, double cutoff, int* rank) +{ + using Wide = std::complex; + *rank = 0; + std::vector coefficients; + std::vector images; + std::vector c(n); + std::vector image(n); + std::vector remaining(n); + for (int j = 0; j < n; ++j) + { + if (!std::isfinite(std::abs(gram[j + j * n]))) + { + return {}; + } + remaining[j] = std::max(0.0, gram[j + j * n].real()); + } + const double largest = n == 0 ? 0 : *std::max_element(remaining.begin(), remaining.end()); + const double absolute_cutoff = cutoff * cutoff; + const double relative_cutoff = 1e-8 * largest; + const double minimum = std::max(absolute_cutoff, relative_cutoff); + for (int step = 0; step < n; ++step) + { + const int pivot = std::max_element(remaining.begin(), remaining.end()) - remaining.begin(); + if (remaining[pivot] <= minimum) + { + break; + } + std::fill(c.begin(), c.end(), Wide(0)); + c[pivot] = 1; + for (int pass = 0; pass < 2; ++pass) + { + for (int j = 0; j < *rank; ++j) + { + Wide overlap = 0; + for (int i = 0; i < n; ++i) + { + overlap += std::conj(images[i + j * n]) * c[i]; + } + for (int i = 0; i < n; ++i) + { + c[i] -= coefficients[i + j * n] * overlap; + } + } + } + std::fill(image.begin(), image.end(), Wide(0)); + for (int j = 0; j < n; ++j) + { + for (int i = 0; i < n; ++i) + { + image[i] += gram[i + j * n] * c[j]; + } + } + Wide norm2 = 0; + for (int i = 0; i < n; ++i) + { + norm2 += std::conj(c[i]) * image[i]; + } + remaining[pivot] = 0; + if (!std::isfinite(std::abs(norm2)) || norm2.real() <= minimum) + { + continue; + } + const double norm = std::sqrt(norm2.real()); + for (int i = 0; i < n; ++i) + { + c[i] /= norm; + image[i] /= norm; + const double remainder = remaining[i] - std::norm(image[i]); + remaining[i] = std::max(0.0, remainder); + } + coefficients.insert(coefficients.end(), c.begin(), c.end()); + images.insert(images.end(), image.begin(), image.end()); + ++*rank; + } + return coefficients; +} + +bool LinearSmallLU::factor(const std::vector>& matrix, int n) +{ + size_ = 0; + if (n <= 0 || matrix.size() != static_cast(n) * n) + { + return false; + } + lu_ = matrix; + pivots_.resize(n); + double scale = 0.0; + for (const std::complex& value: matrix) + { + const double magnitude = std::abs(value); + if (!std::isfinite(magnitude)) + { + return false; + } + scale = std::max(scale, magnitude); + } + const double cutoff = 100 * n * std::numeric_limits::epsilon() * scale; + for (int k = 0; k < n; ++k) + { + int pivot = k; + for (int i = k + 1; i < n; ++i) + { + if (std::abs(lu_[i + k * n]) > std::abs(lu_[pivot + k * n])) + { + pivot = i; + } + } + if (std::abs(lu_[pivot + k * n]) <= cutoff) + { + return false; + } + pivots_[k] = pivot; + for (int j = 0; j < n; ++j) + { + std::swap(lu_[k + j * n], lu_[pivot + j * n]); + } + for (int i = k + 1; i < n; ++i) + { + lu_[i + k * n] /= lu_[k + k * n]; + for (int j = k + 1; j < n; ++j) + { + lu_[i + j * n] -= lu_[i + k * n] * lu_[k + j * n]; + } + } + } + size_ = n; + return true; +} + +bool LinearSmallLU::solve(std::vector>* rhs, int columns) const +{ + if (size_ == 0 || rhs->size() != static_cast(size_) * columns) + { + return false; + } + for (int j = 0; j < columns; ++j) + { + std::complex* b = rhs->data() + j * size_; + for (int k = 0; k < size_; ++k) + { + std::swap(b[k], b[pivots_[k]]); + } + for (int k = 0; k < size_; ++k) + { + for (int i = k + 1; i < size_; ++i) + { + b[i] -= lu_[i + k * size_] * b[k]; + } + } + for (int k = size_ - 1; k >= 0; --k) + { + b[k] /= lu_[k + k * size_]; + if (!std::isfinite(std::abs(b[k]))) + { + return false; + } + for (int i = 0; i < k; ++i) + { + b[i] -= lu_[i + k * size_] * b[k]; + } + } + } + return true; +} +template +void LinearAlgebra::reduce(std::vector* values) const +{ +#ifdef __MPI + if (comm_.nproc > 1) + { + Parallel_Common::reduce_data(values->data(), values->size(), comm_.comm); + } +#endif +} + +template +std::vector> LinearAlgebra::projection_cross(int ld, int dim, int nx, int ny, const T* x, const T* y) +{ + if (std::is_same::value) + { + return cross(ld, dim, nx, ny, x, y); + } + const int64_t elements = static_cast(nx) * ny; + std::vector native(elements, T(0)); + if (dim > 0) + { + linear_buffer(&native_products_, elements); + const T one(1); + const T zero(0); + T* products = native_products_.template data(); + ModuleBase::gemm_op()('C', 'N', nx, ny, dim, &one, x, ld, y, ld, &zero, products, nx); + base_device::memory::synchronize_memory_op()(native.data(), products, native.size()); + } +#ifdef __MPI + if (comm_.nproc > 1) + { + Parallel_Common::reduce_data(native.data(), native.size(), comm_.comm); + } +#endif + return std::vector(native.begin(), native.end()); +} + +template +std::vector> LinearAlgebra::arnoldi_dots(int ld, + int dim, + int bands, + int count, + int stride, + const T* basis, + const T* x) +{ + if (std::is_same::value) + { + return dots(ld, dim, bands, count, stride, basis, x); + } + const int size = bands * count; + linear_buffer(&native_products_, size); + T* products = native_products_.template data(); + linear_op().native_dots(ld, dim, bands, count, stride, basis, x, products); + std::vector native(size); + base_device::memory::synchronize_memory_op()(native.data(), products, size); +#ifdef __MPI + if (comm_.nproc > 1) + { + Parallel_Common::reduce_data(native.data(), size, comm_.comm); + } +#endif + return std::vector(native.begin(), native.end()); +} + +template +std::vector> LinearAlgebra::dots(int ld, + int dim, + int bands, + int count, + int stride, + const T* basis, + const T* x) +{ + const int size = bands * count; + linear_buffer(&products_, size); + linear_op().wide_dots(ld, dim, bands, count, stride, basis, x, products_.template data()); + std::vector result(size); + base_device::memory::synchronize_memory_op()(result.data(), + products_.template data(), + size); + reduce(&result); + return result; +} + +template +std::vector> LinearAlgebra::cross(int ld, int dim, int nx, int ny, const T* x, const T* y) +{ + const int64_t elements = static_cast(nx) * ny; + std::vector result(elements, Wide(0)); + if (dim > 0) + { + const Wide* a = reinterpret_cast(x); + const Wide* b = reinterpret_cast(y); + if (!std::is_same::value) + { + const int64_t left_elements = static_cast(ld) * nx; + const int64_t right_elements = static_cast(ld) * ny; + linear_buffer(&left_, left_elements); + linear_buffer(&right_, right_elements); + // Copy valid rows only: padding may be uninitialized. + for (int j = 0; j < nx; ++j) + { + const int64_t offset = static_cast(j) * ld; + Wide* destination = left_.template data() + offset; + const T* source = x + offset; + base_device::memory::cast_memory_op()(destination, source, dim); + } + for (int j = 0; j < ny; ++j) + { + const int64_t offset = static_cast(j) * ld; + Wide* destination = right_.template data() + offset; + const T* source = y + offset; + base_device::memory::cast_memory_op()(destination, source, dim); + } + a = left_.template data(); + b = right_.template data(); + } + linear_buffer(&products_, elements); + const Wide one(1); + const Wide zero(0); + ModuleBase::gemm_op()('C', 'N', nx, ny, dim, &one, a, ld, b, ld, &zero, products_.template data(), nx); + base_device::memory::synchronize_memory_op()(result.data(), + products_.template data(), + result.size()); + } + reduce(&result); + return result; +} + +template +std::vector> LinearAlgebra::gram(int ld, int dim, int bands, const T* input) +{ + ModuleBase::timer::start("LinearAlgebra", "gram"); + const int64_t elements = static_cast(bands) * bands; + std::vector result(elements, Wide(0)); + if (dim > 0 && bands > 0) + { + const Wide* wide = reinterpret_cast(input); + if (!std::is_same::value) + { + const int64_t input_elements = static_cast(ld) * bands; + linear_buffer(&left_, input_elements); + Wide* converted = left_.template data(); + // The existing conversion kernels use int indices, even though their wrappers accept size_t. + if (ld == dim && input_elements <= std::numeric_limits::max()) + { + base_device::memory::cast_memory_op()(converted, input, input_elements); + } + else + { + // Padding may be uninitialized; convert only the active rows of each column. + for (int band = 0; band < bands; ++band) + { + const int64_t offset = static_cast(band) * ld; + Wide* destination = converted + offset; + const T* source = input + offset; + base_device::memory::cast_memory_op()(destination, source, dim); + } + } + wide = converted; + } + linear_buffer(&products_, elements); + const Wide one(1); + const Wide zero(0); + Wide* products = products_.template data(); + ModuleBase::gemm_op()('C', 'N', bands, bands, dim, &one, wide, ld, wide, ld, &zero, products, bands); + base_device::memory::synchronize_memory_op()(result.data(), products, elements); + } + reduce(&result); + ModuleBase::timer::end("LinearAlgebra", "gram"); + return result; +} + +template +void LinearAlgebra::expand(int ld, int dim, int nx, int ny, const T* x, const std::vector& c, T* y, T beta) +{ + if (dim == 0 || nx == 0 || ny == 0) + { + return; + } + std::vector narrow(c.begin(), c.end()); + linear_buffer(&coefficients_, narrow.size()); + T* coeff = coefficients_.template data(); + base_device::memory::synchronize_memory_op()(coeff, narrow.data(), narrow.size()); + const T one(1); + ModuleBase::gemm_op()('N', 'N', dim, ny, nx, &one, x, ld, coeff, nx, &beta, y, ld); +} + +template class LinearAlgebra, base_device::DEVICE_CPU>; +template class LinearAlgebra, base_device::DEVICE_CPU>; +#if defined(__CUDA) || defined(__ROCM) +template class LinearAlgebra, base_device::DEVICE_GPU>; +template class LinearAlgebra, base_device::DEVICE_GPU>; +#endif +} // namespace hsolver diff --git a/source/source_hsolver/linear_algebra.h b/source/source_hsolver/linear_algebra.h new file mode 100644 index 00000000000..9d6620d91b7 --- /dev/null +++ b/source/source_hsolver/linear_algebra.h @@ -0,0 +1,84 @@ +#ifndef HSOLVER_LINEAR_ALGEBRA_H +#define HSOLVER_LINEAR_ALGEBRA_H + +#include "source_base/macros.h" +#include "source_base/module_container/ATen/core/tensor.h" +#include "source_base/module_device/memory_op.h" +#include "source_hsolver/diag_comm_info.h" + +#include +#include +#include +#include + +namespace hsolver +{ +/** @brief Device allocation helper, including type and empty-partition checks. */ +template +void linear_buffer(ct::Tensor* buffer, const int64_t size) +{ + using CtDevice = typename ct::PsiToContainer::type; + const ct::DeviceType device = ct::DeviceTypeToEnum::value; + const int64_t elements = std::max(1, size); + if (buffer->NumElements() < elements || buffer->data_type() != ct::DataTypeToEnum::value || buffer->device_type() != device) + { + *buffer = ct::Tensor(ct::DataTypeToEnum::value, device, {elements}); + } +} + +/** @brief Partial-pivoted factorization of a small column-major complex matrix. */ +class LinearSmallLU +{ + private: + int size_ = 0; + std::vector> lu_; + std::vector pivots_; + + public: + bool factor(const std::vector>& matrix, int n); + bool solve(std::vector>* rhs, int columns) const; +}; + +/** @brief Rank-revealing coefficients for a small Gram matrix, with a norm cutoff. */ +std::vector> linear_gram_basis(const std::vector>& gram, int n, double cutoff, int* rank); + +/** @brief Collective block algebra; only small coefficients cross the device boundary. */ +template +class LinearAlgebra +{ + private: + using Wide = std::complex; + const diag_comm_info comm_; + ct::Tensor products_; + ct::Tensor coefficients_; + ct::Tensor left_; + ct::Tensor right_; + ct::Tensor native_products_; + + void reduce(std::vector* values) const; + + public: + explicit LinearAlgebra(const diag_comm_info& comm) : comm_(comm) + { + } + + /** @brief Projection products and reductions in wavefunction precision; return double coefficients. */ + std::vector projection_cross(int ld, int dim, int nx, int ny, const T* x, const T* y); + + /** @brief Arnoldi products in wavefunction precision; use dots() for reliable norms. */ + std::vector arnoldi_dots(int ld, int dim, int bands, int count, int stride, const T* basis, const T* x); + + /** @brief Products between corresponding bands and several Arnoldi blocks. */ + std::vector dots(int ld, int dim, int bands, int count, int stride, const T* basis, const T* x); + + /** @brief Global X-adjoint times Y, with double accumulation for both precisions. */ + std::vector cross(int ld, int dim, int nx, int ny, const T* x, const T* y); + + /** @brief Global X-adjoint times X, reusing one FP64 conversion of the valid rows. */ + std::vector gram(int ld, int dim, int bands, const T* input); + + /** @brief Y = X*C + beta*Y, using a small host coefficient matrix. */ + void expand(int ld, int dim, int nx, int ny, const T* x, const std::vector& c, T* y, T beta); +}; +} // namespace hsolver +#endif diff --git a/source/source_hsolver/linear_bicgstab.cpp b/source/source_hsolver/linear_bicgstab.cpp new file mode 100644 index 00000000000..d3700a74d78 --- /dev/null +++ b/source/source_hsolver/linear_bicgstab.cpp @@ -0,0 +1,265 @@ +#include "source_hsolver/linear_bicgstab.h" + +#include "source_hsolver/kernels/linear_op.h" + +#include + +namespace hsolver +{ +template +LinearBiCGSTAB::LinearBiCGSTAB(const double tolerance, const diag_comm_info& comm) : tolerance_(tolerance), work_(comm, 9) +{ +} + +template +bool LinearBiCGSTAB::breakdown(const T divisor, const int band, LinearSolveResult* result) const +{ + if (!linear_bad_divisor(divisor)) + { + return false; + } + result->status = LinearSolveStatus::breakdown; + result->failed_band = original_[band]; + return true; +} + +template +void LinearBiCGSTAB::retire(const int band) +{ + const int last = active_ - 1; + if (band != last) + { + swaps_.push_back(band); + swaps_.push_back(last); + std::swap(rho_[band], rho_[last]); + std::swap(rho_prev_[band], rho_prev_[last]); + std::swap(alpha_[band], alpha_[last]); + std::swap(omega_[band], omega_[last]); + std::swap(threshold_[band], threshold_[last]); + std::swap(original_[band], original_[last]); + } + --active_; +} + +template +void LinearBiCGSTAB::retire_converged(const bool alpha_step, const bool identity) +{ + const T* residual = work_.data(alpha_step ? intermediate_slot : residual_slot); + if (alpha_step) + { + work_.dot(ld_, dim_, active_, residual, residual, norm_.data()); + } + else + { + work_.dot_pair(ld_, dim_, active_, residual, residual, work_.data(shadow_slot), residual, norm_.data(), rho_.data()); + } + swaps_.clear(); + skip_.assign(active_, 1); + bool any_converged = false; + for (int band = 0; band < active_; ++band) + { + const double error = linear_norm(norm_[band]); + if (std::isfinite(error) && error <= threshold_[band]) + { + skip_[band] = 0; + any_converged = true; + } + } + if (!any_converged) + { + return; + } + if (alpha_step) + { + const int update_slot = identity ? direction_slot : precond_direction_slot; + work_.batch(ld_, + dim_, + active_, + work_.data(solution_slot), + work_.data(solution_slot), + work_.data(update_slot), + T(1), + T(1), + nullptr, + alpha_.data(), + skip_.data()); + } + for (int band = active_ - 1; band >= 0; --band) + { + const double error = linear_norm(norm_[band]); + if (std::isfinite(error) && error <= threshold_[band]) + { + retire(band); + } + } + work_.swap_columns(ld_, dim_, swaps_); +} + +template +bool LinearBiCGSTAB::iterate(const LinearOperator& op, + const LinearOperator& preconditioner, + LinearSolveResult* result) +{ + T* r = work_.data(residual_slot); + const bool identity = preconditioner.is_identity(); + T* shadow = work_.data(shadow_slot); + T* p = work_.data(direction_slot); + T* v = work_.data(image_slot); + T* s = work_.data(intermediate_slot); + T* y = work_.data(identity ? direction_slot : precond_direction_slot); + T* z = work_.data(identity ? intermediate_slot : precond_intermediate_slot); + T* t = work_.data(intermediate_image_slot); + T* solution = work_.data(solution_slot); + + for (int band = 0; band < active_; ++band) + { + if (breakdown(rho_[band], band, result) || breakdown(rho_prev_[band], band, result) || breakdown(omega_[band], band, result)) + { + return false; + } + beta_[band] = (rho_[band] / rho_prev_[band]) * (alpha_[band] / omega_[band]); + } + work_.bicg_update(ld_, dim_, active_, true, p, r, v, beta_.data(), omega_.data()); + if (!identity) + { + preconditioner.apply(p, y, ld_, active_); + } + work_.apply(op, y, v, ld_, active_); + work_.dot(ld_, dim_, active_, shadow, v, denominator_.data()); + for (int band = 0; band < active_; ++band) + { + if (breakdown(denominator_[band], band, result)) + { + return false; + } + alpha_[band] = rho_[band] / denominator_[band]; + } + work_.batch(ld_, dim_, active_, s, r, v, T(1), T(-1), nullptr, alpha_.data(), nullptr); + // Retire exact alpha steps before evaluating omega = /. + retire_converged(true, identity); + if (active_ == 0) + { + return true; + } + if (!identity) + { + preconditioner.apply(s, z, ld_, active_); + } + work_.apply(op, z, t, ld_, active_); + work_.dot_pair(ld_, dim_, active_, t, s, t, t, denominator_.data(), tt_.data()); + for (int band = 0; band < active_; ++band) + { + if (breakdown(tt_[band], band, result)) + { + return false; + } + omega_[band] = denominator_[band] / tt_[band]; + if (breakdown(omega_[band], band, result)) + { + return false; + } + + rho_prev_[band] = rho_[band]; + } + work_.bicg_update(ld_, dim_, active_, false, solution, y, z, alpha_.data(), omega_.data()); + work_.batch(ld_, dim_, active_, r, s, t, T(1), T(-1), nullptr, omega_.data(), nullptr); + retire_converged(false, identity); + return true; +} + +template +LinearSolveResult LinearBiCGSTAB::solve(const LinearOperator& op, + const LinearOperator& preconditioner, + int ld, + int nband, + int dim, + T* x, + const T* b, + const T* initial_residual, + const int max_iterations) +{ + const LinearSolveTimer timer("LinearBiCGSTAB"); + work_.prepare(ld, dim, nband, x, b, tolerance_, max_iterations); + work_.reset_statistics(); + ld_ = ld; + dim_ = dim; + active_ = nband; + beta_.resize(nband); + rho_.assign(nband, T(0)); + rho_prev_.assign(nband, T(1)); + alpha_.assign(nband, T(1)); + omega_.assign(nband, T(1)); + denominator_.resize(nband); + tt_.resize(nband); + norm_.resize(nband); + threshold_.resize(nband); + original_.resize(nband); + std::iota(original_.begin(), original_.end(), 0); + work_.dot(ld, dim, nband, b, b, norm_.data()); + for (int band = 0; band < nband; ++band) + { + threshold_[band] = tolerance_ * std::max(1.0, linear_norm(norm_[band])); + } + const std::vector original_threshold = threshold_; + LinearSolveResult result; + while (true) + { + // A restart keeps x, but discards the recursively accumulated residual + // and search directions. All cycles share one iteration budget. + work_.clear(); + active_ = nband; + threshold_ = original_threshold; + std::iota(original_.begin(), original_.end(), 0); + rho_prev_.assign(nband, T(1)); + alpha_.assign(nband, T(1)); + omega_.assign(nband, T(1)); + work_.copy(ld, dim, nband, x, work_.data(solution_slot)); + if (initial_residual && result.restarts == 0) + { + work_.copy(ld, dim, nband, initial_residual, work_.data(residual_slot)); + } + else + { + work_.residual(op, ld, dim, nband, work_.data(solution_slot), b, work_.data(residual_slot)); + } + work_.copy(ld, dim, nband, work_.data(residual_slot), work_.data(shadow_slot)); + retire_converged(false, preconditioner.is_identity()); + result.status = LinearSolveStatus::max_iterations; + const int cycle_start = result.iterations; + try + { + while (result.iterations < max_iterations && active_ > 0) + { + ++result.iterations; + if (!iterate(op, preconditioner, &result)) + { + break; + } + } + } + catch (const LinearPreconditionerError&) + { + result.status = LinearSolveStatus::preconditioner_failure; + } + work_.restore(ld, dim, nband, original_, work_.data(solution_slot), x); + if (active_ == 0) + { + result.status = LinearSolveStatus::residual_mismatch; + } + work_.verify(op, ld, dim, nband, x, b, original_threshold, work_.data(residual_slot), &result); + if (result.status != LinearSolveStatus::residual_mismatch || result.iterations >= max_iterations || result.iterations == cycle_start + || !std::isfinite(result.max_residual)) + { + return result; + } + ++result.restarts; + } +} + +template class LinearBiCGSTAB, base_device::DEVICE_CPU>; +template class LinearBiCGSTAB, base_device::DEVICE_CPU>; +#if defined(__CUDA) || defined(__ROCM) +template class LinearBiCGSTAB, base_device::DEVICE_GPU>; +template class LinearBiCGSTAB, base_device::DEVICE_GPU>; +#endif +} // namespace hsolver diff --git a/source/source_hsolver/linear_bicgstab.h b/source/source_hsolver/linear_bicgstab.h new file mode 100644 index 00000000000..2e21510df35 --- /dev/null +++ b/source/source_hsolver/linear_bicgstab.h @@ -0,0 +1,68 @@ +#ifndef MODULE_HSOLVER_LINEAR_BICGSTAB_H_ +#define MODULE_HSOLVER_LINEAR_BICGSTAB_H_ +#include "source_hsolver/linear_workspace.h" +namespace hsolver +{ +/** @brief Right-preconditioned BiCGSTAB for independent complex right-hand sides. */ +template +class LinearBiCGSTAB final +{ + private: + enum WorkspaceSlot + { + residual_slot = 0, + shadow_slot = 1, + direction_slot = 2, + image_slot = 3, + intermediate_slot = 4, + precond_direction_slot = 5, + precond_intermediate_slot = 6, + intermediate_image_slot = 7, + solution_slot = 8 + }; + const double tolerance_; + LinearWorkspace work_; + int ld_ = 0; + int dim_ = 0; + int active_ = 0; + std::vector beta_; + std::vector rho_; + std::vector rho_prev_; + std::vector alpha_; + std::vector omega_; + std::vector denominator_; + std::vector tt_; + std::vector norm_; + std::vector threshold_; + std::vector original_; + std::vector swaps_; + std::vector skip_; + + public: + LinearBiCGSTAB(const double tolerance, const diag_comm_info& comm); + /** + * @brief Solve A*x=b, using x as initial guess and M as inverse preconditioner. + * @param ld Column stride, at least dim. Padding is neither read nor overwritten. + * @param dim Local valid row count; may be zero on an MPI participant. + * @param initial_residual Optional initial residual, reused on the first cycle only; otherwise nullptr. + * @param max_iterations Iteration budget for this call, shared by all restart cycles. + * @note Accepts only true residuals below tolerance*max(1, norm(b)) per column. + */ + LinearSolveResult solve(const LinearOperator& op, + const LinearOperator& preconditioner, + int ld, + int nband, + int dim, + T* x, + const T* b, + const T* initial_residual, + const int max_iterations); + + private: + bool iterate(const LinearOperator& op, const LinearOperator& preconditioner, LinearSolveResult* result); + void retire(const int band); + void retire_converged(const bool alpha_step, const bool identity); + bool breakdown(const T divisor, const int band, LinearSolveResult* result) const; +}; +} // namespace hsolver +#endif diff --git a/source/source_hsolver/linear_cgs.cpp b/source/source_hsolver/linear_cgs.cpp new file mode 100644 index 00000000000..f5f0f488f12 --- /dev/null +++ b/source/source_hsolver/linear_cgs.cpp @@ -0,0 +1,192 @@ +#include "source_hsolver/linear_cgs.h" + +#include + +namespace hsolver +{ +template +LinearCGS::LinearCGS(const double tolerance, const diag_comm_info& comm) : tolerance_(tolerance), work_(comm, 9) +{ +} + +template +void LinearCGS::retire_converged() +{ + work_.dot_pair(ld_, + dim_, + active_, + work_.data(residual_slot), + work_.data(residual_slot), + work_.data(shadow_slot), + work_.data(residual_slot), + norm_.data(), + rho_.data()); + swaps_.clear(); + for (int band = active_ - 1; band >= 0; --band) + { + const double error = linear_norm(norm_[band]); + if (!std::isfinite(error) || error > threshold_[band]) + { + continue; + } + const int last = --active_; + if (band != last) + { + swaps_.push_back(band); + swaps_.push_back(last); + std::swap(rho_[band], rho_[last]); + std::swap(rho_prev_[band], rho_prev_[last]); + std::swap(threshold_[band], threshold_[last]); + std::swap(original_[band], original_[last]); + } + } + work_.swap_columns(ld_, dim_, swaps_); +} + +template +bool LinearCGS::iterate(const LinearOperator& op, + const LinearOperator& preconditioner, + const bool first_iteration, + LinearSolveResult* result) +{ + T* r = work_.data(residual_slot); + T* shadow = work_.data(shadow_slot); + T* p = work_.data(direction_slot); + T* q = work_.data(q_slot); + T* u = work_.data(u_slot); + T* v = work_.data(image_slot); + T* scratch = work_.data(scratch_slot); + T* direction = work_.data(update_slot); + T* solution = work_.data(solution_slot); + const bool identity = preconditioner.is_identity(); + for (int band = 0; band < active_; ++band) + { + if (linear_bad_divisor(rho_[band]) || linear_bad_divisor(rho_prev_[band])) + { + result->status = LinearSolveStatus::breakdown; + result->failed_band = original_[band]; + return false; + } + beta_[band] = first_iteration ? T(0) : rho_[band] / rho_prev_[band]; + } + work_.cgs_direction(ld_, dim_, active_, p, u, r, q, beta_.data()); + if (!identity) + { + preconditioner.apply(p, scratch, ld_, active_); + } + const T* preconditioned_direction = identity ? p : scratch; + work_.apply(op, preconditioned_direction, v, ld_, active_); + work_.dot(ld_, dim_, active_, shadow, v, denominator_.data()); + for (int band = 0; band < active_; ++band) + { + if (linear_bad_divisor(denominator_[band])) + { + result->status = LinearSolveStatus::breakdown; + result->failed_band = original_[band]; + return false; + } + alpha_[band] = rho_[band] / denominator_[band]; + } + work_.cgs_alpha(ld_, dim_, active_, q, direction, u, v, alpha_.data()); + if (!identity) + { + preconditioner.apply(direction, scratch, ld_, active_); + } + const T* update = identity ? direction : scratch; + work_.apply(op, update, v, ld_, active_); + work_.cgs_finish(ld_, dim_, active_, solution, r, update, v); + rho_prev_ = rho_; + return true; +} + +template +LinearSolveResult LinearCGS::solve(const LinearOperator& op, + const LinearOperator& preconditioner, + int ld, + int nband, + int dim, + T* x, + const T* b, + const T* initial_residual, + const int max_iterations) +{ + const LinearSolveTimer timer("LinearCGS"); + work_.prepare(ld, dim, nband, x, b, tolerance_, max_iterations); + work_.reset_statistics(); + ld_ = ld; + dim_ = dim; + beta_.resize(nband); + rho_.resize(nband); + rho_prev_.resize(nband); + alpha_.resize(nband); + denominator_.resize(nband); + norm_.resize(nband); + threshold_.resize(nband); + original_.resize(nband); + work_.dot(ld, dim, nband, b, b, norm_.data()); + for (int band = 0; band < nband; ++band) + { + const double rhs_norm = linear_norm(norm_[band]); + threshold_[band] = tolerance_ * (rhs_norm == 0.0 ? 1.0 : rhs_norm); + } + const std::vector original_threshold = threshold_; + LinearSolveResult result; + while (true) + { + work_.clear(); + active_ = nband; + threshold_ = original_threshold; + std::iota(original_.begin(), original_.end(), 0); + rho_prev_.assign(nband, T(1)); + work_.copy(ld, dim, nband, x, work_.data(solution_slot)); + if (initial_residual && result.restarts == 0) + { + work_.copy(ld, dim, nband, initial_residual, work_.data(residual_slot)); + } + else + { + work_.residual(op, ld, dim, nband, work_.data(solution_slot), b, work_.data(residual_slot)); + } + work_.copy(ld, dim, nband, work_.data(residual_slot), work_.data(shadow_slot)); + retire_converged(); + result.status = LinearSolveStatus::max_iterations; + const int cycle_start = result.iterations; + try + { + while (result.iterations < max_iterations && active_ > 0) + { + const bool first_iteration = result.iterations == cycle_start; + ++result.iterations; + if (!iterate(op, preconditioner, first_iteration, &result)) + { + break; + } + retire_converged(); + } + } + catch (const LinearPreconditionerError&) + { + result.status = LinearSolveStatus::preconditioner_failure; + } + work_.restore(ld, dim, nband, original_, work_.data(solution_slot), x); + if (active_ == 0) + { + result.status = LinearSolveStatus::residual_mismatch; + } + work_.verify(op, ld, dim, nband, x, b, original_threshold, work_.data(residual_slot), &result); + if (result.status != LinearSolveStatus::residual_mismatch || result.iterations >= max_iterations || result.iterations == cycle_start + || !std::isfinite(result.max_residual)) + { + return result; + } + ++result.restarts; + } +} + +template class LinearCGS, base_device::DEVICE_CPU>; +template class LinearCGS, base_device::DEVICE_CPU>; +#if defined(__CUDA) || defined(__ROCM) +template class LinearCGS, base_device::DEVICE_GPU>; +template class LinearCGS, base_device::DEVICE_GPU>; +#endif +} // namespace hsolver diff --git a/source/source_hsolver/linear_cgs.h b/source/source_hsolver/linear_cgs.h new file mode 100644 index 00000000000..d3f671fe660 --- /dev/null +++ b/source/source_hsolver/linear_cgs.h @@ -0,0 +1,65 @@ +#ifndef HSOLVER_LINEAR_CGS_H +#define HSOLVER_LINEAR_CGS_H +#include "source_hsolver/linear_workspace.h" + +namespace hsolver +{ +/** @brief Right-preconditioned CGS with a compact active set of columns. */ +template +class LinearCGS final +{ + private: + enum WorkspaceSlot + { + residual_slot = 0, + shadow_slot = 1, + direction_slot = 2, + q_slot = 3, + u_slot = 4, + image_slot = 5, + scratch_slot = 6, + update_slot = 7, + solution_slot = 8 + }; + const double tolerance_; + LinearWorkspace work_; + int ld_ = 0; + int dim_ = 0; + int active_ = 0; + std::vector beta_; + std::vector rho_; + std::vector rho_prev_; + std::vector alpha_; + std::vector denominator_; + std::vector norm_; + std::vector threshold_; + std::vector original_; + std::vector swaps_; + + public: + LinearCGS(const double tolerance, const diag_comm_info& comm); + /** + * @brief Solve A*x=b in Device memory, preserving padding and caller column order. + * @param initial_residual Optional initial residual, reused on the first cycle only; otherwise nullptr. + * @param max_iterations Iteration budget for this call, shared by all restart cycles. + * @note Uses tolerance*norm(b), or tolerance for a zero right-hand side. + */ + LinearSolveResult solve(const LinearOperator& op, + const LinearOperator& preconditioner, + int ld, + int nband, + int dim, + T* x, + const T* b, + const T* initial_residual, + const int max_iterations); + + private: + bool iterate(const LinearOperator& op, + const LinearOperator& preconditioner, + const bool first_iteration, + LinearSolveResult* result); + void retire_converged(); +}; +} // namespace hsolver +#endif diff --git a/source/source_hsolver/linear_gmres.cpp b/source/source_hsolver/linear_gmres.cpp new file mode 100644 index 00000000000..39471e34a4c --- /dev/null +++ b/source/source_hsolver/linear_gmres.cpp @@ -0,0 +1,430 @@ +#include "source_hsolver/linear_gmres.h" + +#include +#include + +namespace hsolver +{ +template +LinearGMRES::LinearGMRES(double tolerance, const LinearSolveOptions& options, const diag_comm_info& comm) + : tolerance_(tolerance), restart_(0), work_(comm, 2), algebra_(comm) +{ + const int iteration_limit = std::max(1, options.max_iterations); + restart_ = std::min(options.restart, iteration_limit); + if (restart_ <= 0 || restart_ > (std::numeric_limits::max() - 1) / 3) + { + throw std::invalid_argument("Invalid GMRES restart dimension."); + } +} + +template +T* LinearGMRES::vector(int slot) +{ + return krylov_.template data() + static_cast(slot) * stride_; +} + +template +void LinearGMRES::retire(int band) +{ + const int last = --active_; + if (band == last) + { + return; + } + swaps_.push_back(band); + swaps_.push_back(last); + std::swap(order_[band], order_[last]); + std::swap(h_[band], h_[last]); + std::swap(g_[band], g_[last]); + std::swap(sine_[band], sine_[last]); + std::swap(cosine_[band], cosine_[last]); +} + +template +void LinearGMRES::apply_swaps(int last) +{ + work_.swap_columns(ld_, dim_, swaps_); + const int basis_count = last + 2; + work_.swap_vectors(ld_, dim_, basis_count, stride_, basis(0), swaps_); + if (last >= 0) + { + const int direction_count = last + 1; + work_.swap_vectors(ld_, dim_, direction_count, stride_, direction(0), swaps_); + work_.swap_vectors(ld_, dim_, direction_count, stride_, image(0), swaps_); + } + swaps_.clear(); +} + +template +T* LinearGMRES::residual() +{ + return work_.data(0); +} + +template +T* LinearGMRES::solution() +{ + return work_.data(1); +} + +template +T* LinearGMRES::basis(int index) +{ + return vector(index); +} + +template +T* LinearGMRES::direction(int index) +{ + const int slot = restart_ + 1 + index; + return vector(slot); +} + +template +T* LinearGMRES::image(int index) +{ + const int slot = 2 * restart_ + 1 + index; + return vector(slot); +} + +template +void LinearGMRES::orthogonalize(int j, T* next, std::vector* coefficients) +{ + std::vector& coeff = *coefficients; + const int basis_count = j + 1; + // Two-pass classical Gram-Schmidt batches the global reductions. + for (int pass = 0; pass < 2; ++pass) + { + const std::vector dots = algebra_.arnoldi_dots(ld_, dim_, active_, basis_count, stride_, basis(0), next); + for (int i = 0; i <= j; ++i) + { + for (int b = 0; b < active_; ++b) + { + h_[b][i + j * (restart_ + 1)] += dots[i * active_ + b]; + coeff[b] = T(dots[i * active_ + b]); + } + work_.batch(ld_, dim_, active_, next, next, basis(i), T(1), T(-1), nullptr, coeff.data(), nullptr); + } + } +} + +template +bool LinearGMRES::update_qr(int b, int j, double norm) +{ + Wide* column = h_[b].data() + j * (restart_ + 1); + column[j + 1] = norm; + for (int i = 0; i < j; ++i) + { + const Wide upper = cosine_[b][i] * column[i] + sine_[b][i] * column[i + 1]; + column[i + 1] = -std::conj(sine_[b][i]) * column[i] + cosine_[b][i] * column[i + 1]; + column[i] = upper; + } + const double diagonal_magnitude = std::abs(column[j]); + const double magnitude = std::hypot(diagonal_magnitude, norm); + if (magnitude == 0 || !std::isfinite(magnitude)) + { + return false; + } + const Wide phase = diagonal_magnitude == 0 ? Wide(1) : column[j] / diagonal_magnitude; + cosine_[b][j] = diagonal_magnitude / magnitude; + sine_[b][j] = phase * norm / magnitude; + column[j] = phase * magnitude; + column[j + 1] = 0; + g_[b][j + 1] = -std::conj(sine_[b][j]) * g_[b][j]; + g_[b][j] *= cosine_[b][j]; + return true; +} + +template +bool LinearGMRES::back_substitute(int b, int j, std::vector* weights) +{ + const typename std::vector::const_iterator weights_end = g_[b].cbegin() + j + 1; + weights->assign(g_[b].cbegin(), weights_end); + for (int i = j; i >= 0; --i) + { + (*weights)[i] /= h_[b][i + i * (restart_ + 1)]; + if (!std::isfinite(std::abs((*weights)[i]))) + { + return false; + } + for (int k = 0; k < i; ++k) + { + (*weights)[k] -= h_[b][k + i * (restart_ + 1)] * (*weights)[i]; + } + } + return true; +} + +template +void LinearGMRES::update_solution(int j, + const std::vector& done, + const std::vector>& weights, + bool reconstruct, + std::vector* coefficients) +{ + std::vector& coeff = *coefficients; + const bool any_done = std::find(done.begin(), done.end(), 1) != done.end(); + T* current_solution = solution(); + T* current_residual = reconstruct ? residual() : nullptr; + for (int i = 0; any_done && i <= j; ++i) + { + bool has_zero_coefficients = false; + for (int b = 0; b < active_; ++b) + { + coeff[b] = done[b] ? T(weights[b][i]) : T(0); + has_zero_coefficients = has_zero_coefficients || coeff[b] == T(0); + } + const T* current_direction = direction(i); + const T* current_image = reconstruct ? image(i) : nullptr; + work_.gmres_update(ld_, + dim_, + active_, + current_solution, + current_residual, + current_direction, + current_image, + coeff.data(), + has_zero_coefficients); + } +} + +template +bool LinearGMRES::start_cycle(const std::vector& threshold, std::vector* coefficients) +{ + const int projected_rows = restart_ + 1; + const int64_t projected_elements = static_cast(projected_rows) * restart_; + h_.assign(bands_, std::vector(projected_elements, 0)); + g_.assign(bands_, std::vector(projected_rows, 0)); + sine_.assign(bands_, std::vector(restart_, 0)); + cosine_.assign(bands_, std::vector(restart_, 0)); + std::vector norms = algebra_.dots(ld_, dim_, bands_, 1, stride_, residual(), residual()); + std::vector& coeff = *coefficients; + active_ = bands_; + swaps_.clear(); + std::iota(order_.begin(), order_.end(), 0); + for (int b = 0; b < bands_; ++b) + { + const double norm = linear_norm(norms[b]); + if (!std::isfinite(norm)) + { + return false; + } + g_[b][0] = norm; + coeff[b] = norm == 0 ? T(0) : T(1.0 / norm); + } + work_.batch(ld_, dim_, bands_, basis(0), residual(), residual(), T(1), T(0), coeff.data(), nullptr, nullptr); + for (int b = bands_ - 1; b >= 0; --b) + { + if (std::abs(g_[b][0]) <= 0.8 * threshold[order_[b]]) + { + retire(b); + } + } + apply_swaps(-1); + return true; +} + +template +bool LinearGMRES::cycle(const LinearOperator& op, + const LinearOperator& preconditioner, + const std::vector& threshold, + bool reconstruct, + int max_iterations, + LinearSolveResult* result) +{ + std::vector coeff(bands_); + if (!start_cycle(threshold, &coeff)) + { + return false; + } + for (int j = 0; j < restart_ && active_ > 0 && result->iterations < max_iterations; ++j) + { + T* z = direction(j); + T* raw = image(j); + const int next_index = j + 1; + T* next = basis(next_index); + preconditioner.apply(basis(j), z, ld_, active_); + work_.apply(op, z, raw, ld_, active_); + work_.copy(ld_, dim_, active_, raw, next); + orthogonalize(j, next, &coeff); + const std::vector norms = algebra_.dots(ld_, dim_, active_, 1, stride_, next, next); + ++result->iterations; + std::vector done(active_, 0); + std::vector> weights(active_); + for (int b = 0; b < active_; ++b) + { + const double norm = linear_norm(norms[b]); + if (!std::isfinite(norm)) + { + return false; + } + if (!update_qr(b, j, norm)) + { + return false; + } + done[b] = std::abs(g_[b][j + 1]) <= 0.8 * threshold[order_[b]] || norm <= std::numeric_limits::min() || j + 1 == restart_ + || result->iterations == max_iterations; + coeff[b] = done[b] ? T(0) : T(1.0 / norm); + if (!done[b]) + { + continue; + } + if (!back_substitute(b, j, &weights[b])) + { + return false; + } + } + work_.batch(ld_, dim_, active_, next, next, next, T(1), T(0), coeff.data(), nullptr, nullptr); + update_solution(j, done, weights, reconstruct, &coeff); + for (int b = active_ - 1; b >= 0; --b) + { + if (done[b]) + { + retire(b); + } + } + apply_swaps(j); + } + return true; +} + +template +LinearSolveResult LinearGMRES::solve(const LinearOperator& op, + const LinearOperator& preconditioner, + int ld, + int nvec, + int dim, + T* x, + const T* b, + const T* initial_residual, + bool force_check, + const LinearSolveControl& control) +{ + const LinearSolveTimer timer("LinearGMRES"); + const int max_iterations = control.max_iterations; + work_.prepare(ld, dim, nvec, x, b, tolerance_, max_iterations); + work_.reset_statistics(); + ld_ = ld; + dim_ = dim; + bands_ = nvec; + const int vector_elements = ld * nvec; + stride_ = std::max(1, vector_elements); + const int64_t krylov_elements = (3 * static_cast(restart_) + 1) * stride_; + linear_buffer(&krylov_, krylov_elements); + order_.resize(nvec); + const std::vector rhs_norms = algebra_.dots(ld, dim, nvec, 1, stride_, b, b); + std::vector threshold(nvec); + for (int i = 0; i < nvec; ++i) + { + const double rhs_norm = linear_norm(rhs_norms[i]); + threshold[i] = tolerance_ * std::max(1.0, rhs_norm); + } + bool reconstruct = control.reconstruct; + force_check = force_check || tolerance_ < 100 * std::numeric_limits::epsilon(); + LinearSolveResult result; + work_.clear(); + if (initial_residual) + { + work_.copy(ld, dim, nvec, initial_residual, residual()); + } + else + { + work_.residual(op, ld, dim, nvec, x, b, residual()); + } + while (true) + { + // Restarts retain the independent residual computed below in the original band order. + work_.copy(ld, dim, nvec, x, solution()); + const int start = result.iterations; + bool regular = false; + try + { + regular = cycle(op, preconditioner, threshold, reconstruct, max_iterations, &result); + result.status = regular ? LinearSolveStatus::max_iterations : LinearSolveStatus::breakdown; + } + catch (const LinearPreconditionerError&) + { + // Unlike completed Arnoldi steps, this failed attempt has not yet been counted. + ++result.iterations; + result.status = LinearSolveStatus::preconditioner_failure; + } + work_.restore(ld, dim, nvec, order_, solution(), x); + bool accepted = regular && reconstruct; + bool invalid_reconstruction = false; + if (accepted) + { + const std::vector norms = algebra_.dots(ld, dim, nvec, 1, stride_, residual(), residual()); + result.max_residual = 0; + for (int i = 0; i < nvec; ++i) + { + const double error = linear_norm(norms[i]); + result.max_residual = std::max(result.max_residual, error); + invalid_reconstruction = invalid_reconstruction || !std::isfinite(error); + if (!std::isfinite(error) || error > threshold[order_[i]]) + { + accepted = false; + } + } + if (accepted && !force_check) + { + result.status = LinearSolveStatus::converged; + result.failed_band = -1; + result.reconstructed = true; + work_.statistics(&result); + return result; + } + } + work_.residual(op, ld, dim, nvec, x, b, residual()); + ++result.true_checks; + work_.statistics(&result); + const std::vector true_norms = algebra_.dots(ld, dim, nvec, 1, stride_, residual(), residual()); + result.max_residual = 0; + result.failed_band = -1; + for (int i = 0; i < nvec; ++i) + { + const double error = linear_norm(true_norms[i]); + result.max_residual = std::max(result.max_residual, error); + if ((!std::isfinite(error) || error > threshold[i]) && result.failed_band < 0) + { + result.failed_band = i; + } + } + if (result.failed_band < 0) + { + result.status = LinearSolveStatus::converged; + } + if (result.status == LinearSolveStatus::converged) + { + return result; + } + if (result.status == LinearSolveStatus::preconditioner_failure) + { + // Let the caller retry with a different preconditioner and the remaining budget. + return result; + } + const bool recover_anomaly = reconstruct && !regular; + // A full restart cycle is not a reconstruction failure. + if (reconstruct && (accepted || !regular || invalid_reconstruction)) + { + ++result.reconstruction_fallbacks; + reconstruct = false; + } + if (!regular && !recover_anomaly) + { + return result; + } + const bool retry_initial = initial_residual && result.restarts == 0; + if (result.iterations >= max_iterations || (result.iterations == start && !retry_initial) || !std::isfinite(result.max_residual)) + { + return result; + } + ++result.restarts; + } +} +template class LinearGMRES, base_device::DEVICE_CPU>; +template class LinearGMRES, base_device::DEVICE_CPU>; +#if defined(__CUDA) || defined(__ROCM) +template class LinearGMRES, base_device::DEVICE_GPU>; +template class LinearGMRES, base_device::DEVICE_GPU>; +#endif +} // namespace hsolver diff --git a/source/source_hsolver/linear_gmres.h b/source/source_hsolver/linear_gmres.h new file mode 100644 index 00000000000..8b6d3868af8 --- /dev/null +++ b/source/source_hsolver/linear_gmres.h @@ -0,0 +1,71 @@ +#ifndef HSOLVER_LINEAR_GMRES_H +#define HSOLVER_LINEAR_GMRES_H +#include "source_hsolver/linear_algebra.h" +#include "source_hsolver/linear_workspace.h" + +namespace hsolver +{ +/** @brief Restarted right-preconditioned GMRES with independent batched columns. */ +template +class LinearGMRES +{ + private: + using Wide = std::complex; + using Real = typename GetTypeReal::type; + const double tolerance_; + int restart_; + LinearWorkspace work_; + LinearAlgebra algebra_; + ct::Tensor krylov_; + std::vector> h_; + std::vector> g_; + std::vector> sine_; + std::vector> cosine_; + std::vector order_; + std::vector swaps_; + int ld_ = 0; + int dim_ = 0; + int bands_ = 0; + int stride_ = 0; + int active_ = 0; + + T* vector(int slot); + T* residual(); + T* solution(); + T* basis(int index); + T* direction(int index); + T* image(int index); + bool start_cycle(const std::vector& threshold, std::vector* coefficients); + void orthogonalize(int j, T* next, std::vector* coefficients); + bool update_qr(int b, int j, double norm); + bool back_substitute(int b, int j, std::vector* weights); + void update_solution(int j, + const std::vector& done, + const std::vector>& weights, + bool reconstruct, + std::vector* coefficients); + void retire(int band); + void apply_swaps(int last); + bool cycle(const LinearOperator& op, + const LinearOperator& preconditioner, + const std::vector& threshold, + bool reconstruct, + int max_iterations, + LinearSolveResult* result); + + public: + LinearGMRES(double tolerance, const LinearSolveOptions& options, const diag_comm_info& comm); + /** @brief Use the per-call budget and reconstruction policy; force_check requests an independent final application. */ + LinearSolveResult solve(const LinearOperator& op, + const LinearOperator& preconditioner, + int ld, + int nvec, + int dim, + T* x, + const T* b, + const T* initial_residual, + bool force_check, + const LinearSolveControl& control); +}; +} // namespace hsolver +#endif diff --git a/source/source_hsolver/linear_low_rank.cpp b/source/source_hsolver/linear_low_rank.cpp new file mode 100644 index 00000000000..8bf4f734df4 --- /dev/null +++ b/source/source_hsolver/linear_low_rank.cpp @@ -0,0 +1,163 @@ +#include "source_hsolver/linear_low_rank.h" + +#include "source_hsolver/kernels/linear_op.h" +#include "source_hsolver/linear_workspace.h" + +namespace hsolver +{ +template +void LinearResponse::update(LinearAlgebra& algebra, + int ld, + int dim, + int bands, + const T* solution, + const T* seed, + const T* seed_residual, + double tolerance, + ct::Tensor* workspace) +{ + rank_ = 0; + const int64_t elements = static_cast(ld) * bands; + const int64_t size = std::max(1, elements); + linear_buffer(&directions_, size); + linear_buffer(&images_, size); + const int64_t workspace_elements = 4 * size; + linear_buffer(workspace, workspace_elements); + T* z = workspace->template data(); + T* w = z + size; + using Real = typename GetTypeReal::type; + const double tolerance_cutoff = 100 * tolerance; + const double roundoff_cutoff = 100 * static_cast(std::numeric_limits::epsilon()); + const double cutoff = std::max(tolerance_cutoff, roundoff_cutoff); + linear_op().batch(ld, dim, bands, z, solution, seed, T(1), T(-1), nullptr, nullptr, nullptr); + if (dim > 0) + { + base_device::memory::synchronize_memory_2d_op()(w, ld, seed_residual, ld, dim, bands); + } + // Form a rank-revealing block basis, then repair its Gram matrix once. + // This avoids a device transfer and MPI reduction for every history column. + std::vector> gram = algebra.cross(ld, dim, bands, bands, w, w); + std::vector> c = linear_gram_basis(gram, bands, cutoff, &rank_); + if (rank_ == 0) + { + return; + } + T* rz = z + 2 * size; + T* rw = rz + size; + algebra.expand(ld, dim, bands, rank_, z, c, rz, T(0)); + algebra.expand(ld, dim, bands, rank_, w, c, rw, T(0)); + gram = algebra.cross(ld, dim, rank_, rank_, rw, rw); + int refined_rank = 0; + c = linear_gram_basis(gram, rank_, 1e-6, &refined_rank); + if (refined_rank > 0) + { + algebra.expand(ld, dim, rank_, refined_rank, rz, c, directions_.template data(), T(0)); + algebra.expand(ld, dim, rank_, refined_rank, rw, c, images_.template data(), T(0)); + } + rank_ = refined_rank; + if (rank_ > 0) + { + gram = algebra.cross(ld, dim, rank_, rank_, images(), images()); + const double roundoff_limit = 100.0 * std::numeric_limits::epsilon(); + const double limit = std::max(1e-8, roundoff_limit); + for (int j = 0; j < rank_; ++j) + { + for (int i = 0; i < rank_; ++i) + { + if (!std::isfinite(std::abs(gram[i + j * rank_])) + || std::abs(gram[i + j * rank_] - (i == j ? std::complex(1) : std::complex(0))) > limit) + { + rank_ = 0; + return; + } + } + } + } +} + +template +void LinearLowRank::prepare_response(int ld, int rank, const T* directions, const T* images, ct::Tensor* workspace) +{ + prepare(ld, rank, directions, images, images, nullptr, workspace); +} + +template +void LinearLowRank::prepare_subspace(int ld, + int rank, + const T* basis, + const T* images, + const LinearSmallLU& factor, + ct::Tensor* workspace) +{ + prepare(ld, rank, basis, images, basis, &factor, workspace); +} + +template +void LinearLowRank::prepare(int ld, + int rank, + const T* z, + const T* w, + const T* test, + const LinearSmallLU* factor, + ct::Tensor* workspace) +{ + rank_ = rank; + test_ = rank > 0 ? test : nullptr; + factor_ = rank > 0 ? factor : nullptr; + correction_ = nullptr; + if (rank == 0) + { + return; + } + const int64_t size = static_cast(ld) * rank; + linear_buffer(workspace, size); + T* correction = workspace->template data(); + if (diagonal_) + { + linear_op().diagonal(ld, dim_, rank, diagonal_, w, correction); + } + else if (dim_ > 0) + { + base_device::memory::synchronize_memory_2d_op()(correction, ld, w, ld, dim_, rank); + } + linear_op().batch(ld, dim_, rank, correction, z, correction, T(1), T(-1), nullptr, nullptr, nullptr); + correction_ = correction; +} + +template +void LinearLowRank::apply(const T* x, T* y, int ld, int nvec) const +{ + if (diagonal_) + { + linear_op().diagonal(ld, dim_, nvec, diagonal_, x, y); + } + else if (dim_ > 0) + { + base_device::memory::synchronize_memory_2d_op()(y, ld, x, ld, dim_, nvec); + } + if (rank_ == 0) + { + return; + } + std::vector> coefficients = algebra_.projection_cross(ld, dim_, rank_, nvec, test_, x); + if (factor_ && !factor_->solve(&coefficients, nvec)) + { + // Changing the preconditioner inside a recurrence invalidates BiCGSTAB and CGS. + throw LinearPreconditionerError(); + } + algebra_.expand(ld, dim_, rank_, nvec, correction_, coefficients, y, T(1)); +} + +template class LinearResponse, base_device::DEVICE_CPU>; +template class LinearResponse, base_device::DEVICE_CPU>; +#if defined(__CUDA) || defined(__ROCM) +template class LinearResponse, base_device::DEVICE_GPU>; +template class LinearResponse, base_device::DEVICE_GPU>; +#endif +template class LinearLowRank, base_device::DEVICE_CPU>; +template class LinearLowRank, base_device::DEVICE_CPU>; +#if defined(__CUDA) || defined(__ROCM) +template class LinearLowRank, base_device::DEVICE_GPU>; +template class LinearLowRank, base_device::DEVICE_GPU>; +#endif +} // namespace hsolver diff --git a/source/source_hsolver/linear_low_rank.h b/source/source_hsolver/linear_low_rank.h new file mode 100644 index 00000000000..ce3eb044950 --- /dev/null +++ b/source/source_hsolver/linear_low_rank.h @@ -0,0 +1,92 @@ +#ifndef HSOLVER_LINEAR_LOW_RANK_H +#define HSOLVER_LINEAR_LOW_RANK_H +#include "source_hsolver/linear_algebra.h" +#include "source_hsolver/linear_operator.h" + +namespace hsolver +{ +/** @brief A bounded, paired-orthogonalized response history for one k point. */ +template +class LinearResponse +{ + private: + ct::Tensor directions_; + ct::Tensor images_; + int rank_ = 0; + + public: + int rank() const + { + return rank_; + } + void clear() + { + rank_ = 0; + } + const T* directions() const + { + return directions_.template data(); + } + const T* images() const + { + return images_.template data(); + } + + /** @brief Replace the history only after a successful solve; discard unresolved directions. */ + void update(LinearAlgebra& algebra, + int ld, + int dim, + int bands, + const T* solution, + const T* seed, + const T* seed_residual, + double tolerance, + ct::Tensor* workspace); +}; + +/** @brief Diagonal inverse with an optional response or Galerkin coarse correction. */ +template +class LinearLowRank final : public LinearOperator +{ + private: + LinearAlgebra& algebra_; + const T* diagonal_; + const int dim_; + const T* test_ = nullptr; + int rank_ = 0; + const LinearSmallLU* factor_ = nullptr; + const T* correction_ = nullptr; + + public: + LinearLowRank(LinearAlgebra& algebra, const T* diagonal, int dim) : algebra_(algebra), diagonal_(diagonal), dim_(dim) + { + } + bool is_identity() const override + { + return diagonal_ == nullptr && rank_ == 0; + } + int rank() const + { + return rank_; + } + + /** @brief Prepare a response correction with orthonormal images. + * @param images Borrowed projection vectors; keep them valid and unchanged until the last apply. + * @param workspace Caller-owned correction storage; do not modify or resize it until the last apply. + */ + void prepare_response(int ld, int rank, const T* directions, const T* images, ct::Tensor* workspace); + /** @brief Prepare a Galerkin coarse correction. + * @param basis Borrowed projection vectors; keep them valid and unchanged until the last apply. + * @param factor Borrowed factorization of basis^H*images; keep it valid and unchanged until the last apply. + * @param workspace Caller-owned correction storage; do not modify or resize it until the last apply. + */ + void prepare_subspace(int ld, int rank, const T* basis, const T* images, const LinearSmallLU& factor, ct::Tensor* workspace); + /** @brief Apply the fixed correction; throw LinearPreconditionerError if the coarse solve fails. */ + void apply(const T* x, T* y, int ld, int nvec) const override; + + private: + /** @brief Form Z-D*W and retain the projection data for subsequent applications. */ + void prepare(int ld, int rank, const T* z, const T* w, const T* test, const LinearSmallLU* factor, ct::Tensor* workspace); +}; +} // namespace hsolver +#endif diff --git a/source/source_hsolver/linear_operator.h b/source/source_hsolver/linear_operator.h new file mode 100644 index 00000000000..f41ee21677f --- /dev/null +++ b/source/source_hsolver/linear_operator.h @@ -0,0 +1,40 @@ +#ifndef HSOLVER_LINEAR_OPERATOR_H +#define HSOLVER_LINEAR_OPERATOR_H + +#include "source_base/module_device/types.h" + +#include + +namespace hsolver +{ + +/** @brief Stop the current recurrence when a preconditioner cannot be applied. */ +class LinearPreconditionerError final : public std::exception +{ + public: + const char* what() const noexcept override + { + return "Preconditioner application failed."; + } +}; + +/** + * @brief Device-resident block operator for independent right-hand sides. + * + * Column i starts at x + i*ld. The operator knows the valid row count. + * Inputs are read-only; input/output buffers must not overlap. The same + * operator acts on every column, including reordered or compacted blocks. + * A zero local row count must still participate in operator collectives. + */ +template +class LinearOperator +{ + public: + virtual ~LinearOperator() = default; + virtual void apply(const T* x, T* y, const int ld, const int nvec) const = 0; + /** @brief Allow solvers to bypass identity preconditioner copies. */ + virtual bool is_identity() const { return false; } +}; + +} // namespace hsolver +#endif diff --git a/source/source_hsolver/linear_solver_types.h b/source/source_hsolver/linear_solver_types.h new file mode 100644 index 00000000000..030f014682f --- /dev/null +++ b/source/source_hsolver/linear_solver_types.h @@ -0,0 +1,60 @@ +#ifndef HSOLVER_LINEAR_SOLVER_TYPES_H +#define HSOLVER_LINEAR_SOLVER_TYPES_H + +#include + +namespace hsolver +{ + +enum class LinearSolveStatus +{ + converged, + breakdown, + max_iterations, + residual_mismatch, + preconditioner_failure +}; + +/** @brief Convergence result, operator work, and independent or reconstructed residual provenance. */ +struct LinearSolveResult +{ + LinearSolveStatus status = LinearSolveStatus::max_iterations; + double max_residual = 0.0; + int iterations = 0; + int failed_band = -1; + int restarts = 0; + int true_checks = 0; + int reconstruction_fallbacks = 0; + bool reconstructed = false; + std::int64_t operator_calls = 0; + std::int64_t operator_columns = 0; +}; + +enum class LinearMethod +{ + bicgstab, + cgs, + gmres +}; + +/** @brief Numerical policy; tolerance zero selects the precision-aware default. */ +struct LinearSolveOptions +{ + LinearMethod method = LinearMethod::bicgstab; + double tolerance = 0.0; + int max_iterations = 500; + int restart = 20; + bool reconstruct = false; +}; + +/** @brief Explicit policy for one solve; does not change the solver's defaults. */ +struct LinearSolveControl +{ + int max_iterations; + bool reconstruct; +}; + +const char* linear_status_name(const LinearSolveStatus status); + +} // namespace hsolver +#endif diff --git a/source/source_hsolver/linear_workspace.cpp b/source/source_hsolver/linear_workspace.cpp new file mode 100644 index 00000000000..fa2e14fe0dc --- /dev/null +++ b/source/source_hsolver/linear_workspace.cpp @@ -0,0 +1,439 @@ +#include "source_hsolver/linear_workspace.h" + +#include "source_base/module_device/memory_op.h" +#include "source_base/parallel_device.h" +#include "source_base/timer.h" +#include "source_hsolver/kernels/linear_op.h" + +#include +#include +#include +#include +#include + +namespace hsolver +{ + +LinearSolveTimer::LinearSolveTimer(const char* solver) : solver_(solver) +{ + ModuleBase::timer::start(solver_, "solve"); +} + +LinearSolveTimer::~LinearSolveTimer() +{ + ModuleBase::timer::end(solver_, "solve"); +} + +template +LinearWorkspace::LinearWorkspace(const diag_comm_info& comm, int slots) : slots_(slots), comm_(comm) +{ + if (slots_ <= 0) + { + throw std::invalid_argument("Linear workspace requires a positive slot count."); + } + // A default Tensor is a one-element float CPU tensor, not an empty buffer. + using CtDevice = typename ct::PsiToContainer::type; + const ct::DeviceType device = ct::DeviceTypeToEnum::value; + dots_ = ct::Tensor(ct::DataTypeToEnum::value, device, {0}); + partial_ = ct::Tensor(ct::DataTypeToEnum::value, device, {0}); + coefficients_ = ct::Tensor(ct::DataTypeToEnum::value, device, {0}); + mask_ = ct::Tensor(ct::DataTypeToEnum::value, device, {0}); + permutation_ = ct::Tensor(ct::DataTypeToEnum::value, device, {0}); +} + +template +void LinearWorkspace::prepare(const int ld, + const int dim, + const int nvec, + const T* x, + const T* b, + const double tolerance, + const int max_iter) +{ + if (dim < 0 || ld < dim || nvec <= 0 || !std::isfinite(tolerance) || tolerance <= 0.0 || max_iter < 0 + || ld > std::numeric_limits::max() / nvec || (ld > 0 && (x == nullptr || b == nullptr)) || (ld > 0 && x == b) + || comm_.nproc < 1) + { + throw std::invalid_argument("Invalid linear-solver layout, buffers, or convergence controls."); + } + const int vector_elements = ld * nvec; + const int size = std::max(1, vector_elements); + if (size > capacity_) + { + using CtDevice = typename ct::PsiToContainer::type; + const int64_t slot_elements = size; + vectors_ = ct::Tensor(ct::DataTypeToEnum::value, ct::DeviceTypeToEnum::value, {slots_, slot_elements}); + capacity_ = size; + } +} + +template +void LinearWorkspace::clear() +{ + const size_t elements = static_cast(slots_) * capacity_; + base_device::memory::set_memory_op()(vectors_.template data(), 0, elements); +} + +template +void LinearWorkspace::reset_statistics() +{ + operator_calls_ = 0; + operator_columns_ = 0; +} + +template +void LinearWorkspace::statistics(LinearSolveResult* result) const +{ + result->operator_calls = operator_calls_; + result->operator_columns = operator_columns_; +} + +template +void LinearWorkspace::apply(const LinearOperator& op, const T* x, T* y, int ld, int nvec) +{ + ModuleBase::timer::start("LinearWorkspace", "apply"); + ++operator_calls_; + operator_columns_ += nvec; + op.apply(x, y, ld, nvec); + ModuleBase::timer::end("LinearWorkspace", "apply"); +} + +template +T* LinearWorkspace::data(const int slot) +{ + return vectors_.data() + static_cast(slot) * capacity_; +} + +template +void LinearWorkspace::copy(const int ld, const int dim, const int nvec, const T* x, T* y) const +{ + if (dim > 0) + { + base_device::memory::synchronize_memory_2d_op()(y, ld, x, ld, dim, nvec); + } +} + +template +void LinearWorkspace::dot(const int ld, const int dim, const int nvec, const T* x, const T* y, T* out) +{ + + dot_pair(ld, dim, nvec, x, y, nullptr, nullptr, out, nullptr); +} + +template +void LinearWorkspace::residual(const LinearOperator& op, + const int ld, + const int dim, + const int nvec, + const T* x, + const T* b, + T* r) +{ + apply(op, x, r, ld, nvec); + linear_op().batch(ld, dim, nvec, r, b, r, T(1), T(-1), nullptr, nullptr, nullptr); +} + +template +void LinearWorkspace::verify(const LinearOperator& op, + const int ld, + const int dim, + const int nvec, + const T* x, + const T* b, + const std::vector& threshold, + T* scratch, + LinearSolveResult* result) +{ + residual(op, ld, dim, nvec, x, b, scratch); + ++result->true_checks; + result->operator_calls = operator_calls_; + result->operator_columns = operator_columns_; + std::vector norms(nvec); + dot(ld, dim, nvec, scratch, scratch, norms.data()); + result->max_residual = 0.0; + result->failed_band = -1; + bool converged = true; + for (int band = 0; band < nvec; ++band) + { + const double error = linear_norm(norms[band]); + result->max_residual = std::isfinite(error) ? std::max(result->max_residual, error) : std::numeric_limits::infinity(); + if (!std::isfinite(error) || !std::isfinite(threshold[band]) || error > threshold[band]) + { + if (result->failed_band < 0) + { + result->failed_band = band; + } + converged = false; + } + } + if (converged) + { + result->status = LinearSolveStatus::converged; + result->failed_band = -1; + } +} + +template +void LinearWorkspace::dot_pair(int ld, int dim, int nvec, const T* x, const T* y, const T* z, const T* w, T* first, T* second) +{ + ModuleBase::timer::start("LinearWorkspace", "dot_pair"); + if (nvec == 0) + { + ModuleBase::timer::end("LinearWorkspace", "dot_pair"); + return; + } + using CtDevice = typename ct::PsiToContainer::type; + const ct::DeviceType device = ct::DeviceTypeToEnum::value; + const int count = (z ? 2 : 1) * nvec; + const int requested_tiles = (dim + 2047) / 2048; + const int capped_tiles = std::min(32, requested_tiles); + const int tiles = std::max(1, capped_tiles); + if (dots_.NumElements() < count) + { + dots_ = ct::Tensor(ct::DataTypeToEnum::value, device, {count}); + } + const int64_t partial_elements = static_cast(count) * tiles; + if (partial_.NumElements() < partial_elements) + { + partial_ = ct::Tensor(ct::DataTypeToEnum::value, device, {partial_elements}); + } + host_dots_.resize(count); + linear_op().dots(ld, dim, nvec, x, y, z, w, dots_.template data(), partial_.template data(), tiles); + base_device::memory::synchronize_memory_op()(host_dots_.data(), dots_.template data(), count); +#ifdef __MPI + if (comm_.nproc > 1) + { + Parallel_Common::reduce_data(host_dots_.data(), count, comm_.comm); + } +#endif + const typename std::vector::const_iterator second_begin = host_dots_.cbegin() + nvec; + std::copy(host_dots_.cbegin(), second_begin, first); + if (z) + { + std::copy(second_begin, host_dots_.cend(), second); + } + ModuleBase::timer::end("LinearWorkspace", "dot_pair"); +} + +template +void LinearWorkspace::batch(int ld, + int dim, + int nvec, + T* out, + const T* x, + const T* y, + T a, + T b, + const T* ca, + const T* cb, + const int* skip) +{ + ModuleBase::timer::start("LinearWorkspace", "batch"); + if (dim == 0 || nvec == 0) + { + ModuleBase::timer::end("LinearWorkspace", "batch"); + return; + } + using CtDevice = typename ct::PsiToContainer::type; + const ct::DeviceType device = ct::DeviceTypeToEnum::value; + const int64_t coefficient_elements = 2 * static_cast(nvec); + if (coefficients_.NumElements() < coefficient_elements) + { + coefficients_ = ct::Tensor(ct::DataTypeToEnum::value, device, {coefficient_elements}); + } + T* coeff = coefficients_.template data(); + if (ca) + { + base_device::memory::synchronize_memory_op()(coeff, ca, nvec); + } + if (cb) + { + T* second_coefficients = coeff + nvec; + base_device::memory::synchronize_memory_op()(second_coefficients, cb, nvec); + } + const int* mask = nullptr; + if (skip) + { + if (mask_.NumElements() < nvec) + { + mask_ = ct::Tensor(ct::DataTypeToEnum::value, device, {nvec}); + } + base_device::memory::synchronize_memory_op()(mask_.template data(), skip, nvec); + mask = mask_.template data(); + } + const T* first_coefficients = ca ? coeff : nullptr; + const T* second_coefficients = cb ? coeff + nvec : nullptr; + linear_op().batch(ld, dim, nvec, out, x, y, a, b, first_coefficients, second_coefficients, mask); + ModuleBase::timer::end("LinearWorkspace", "batch"); +} + +template +void LinearWorkspace::gmres_update(int ld, + int dim, + int nvec, + T* solution, + T* residual, + const T* direction, + const T* image, + const T* coefficients, + bool has_zero_coefficients) +{ + ModuleBase::timer::start("LinearWorkspace", "gmres_update"); + if (dim == 0 || nvec == 0) + { + ModuleBase::timer::end("LinearWorkspace", "gmres_update"); + return; + } + using CtDevice = typename ct::PsiToContainer::type; + const ct::DeviceType device = ct::DeviceTypeToEnum::value; + if (coefficients_.NumElements() < nvec) + { + coefficients_ = ct::Tensor(ct::DataTypeToEnum::value, device, {nvec}); + } + T* coeff = coefficients_.template data(); + base_device::memory::synchronize_memory_op()(coeff, coefficients, nvec); + if (std::is_same::value && residual == nullptr && !has_zero_coefficients) + { + // Keep the existing GPU path when neither zero-column skipping nor a paired update is needed. + linear_op().batch(ld, dim, nvec, solution, solution, direction, T(1), T(1), nullptr, coeff, nullptr); + } + else + { + linear_op().gmres_update(ld, dim, nvec, solution, residual, direction, image, coeff); + } + ModuleBase::timer::end("LinearWorkspace", "gmres_update"); +} + +template +void LinearWorkspace::swap_columns(int ld, int dim, const std::vector& pairs) +{ + swap_vectors(ld, dim, slots_, capacity_, vectors_.template data(), pairs); +} + +template +void LinearWorkspace::swap_vectors(int ld, int dim, int slots, int stride, T* vectors, const std::vector& pairs) +{ + if (pairs.empty() || dim == 0) + { + return; + } + using CtDevice = typename ct::PsiToContainer::type; + const int64_t permutation_elements = pairs.size(); + if (permutation_.NumElements() < permutation_elements) + { + permutation_ = ct::Tensor(ct::DataTypeToEnum::value, ct::DeviceTypeToEnum::value, {permutation_elements}); + } + int* map = permutation_.template data(); + base_device::memory::synchronize_memory_op()(map, pairs.data(), pairs.size()); + const int pair_count = pairs.size() / 2; + linear_op().swaps(ld, dim, slots, stride, pair_count, vectors, map); +} +template +void LinearWorkspace::restore(int ld, int dim, int nvec, const std::vector& order, const T* source, T* destination) +{ + if (dim == 0) + { + return; + } + using CtDevice = typename ct::PsiToContainer::type; + if (permutation_.NumElements() < nvec) + { + permutation_ = ct::Tensor(ct::DataTypeToEnum::value, ct::DeviceTypeToEnum::value, {nvec}); + } + std::vector inverse(nvec); + for (int band = 0; band < nvec; ++band) + { + inverse[order[band]] = band; + } + int* map = permutation_.template data(); + base_device::memory::synchronize_memory_op()(map, inverse.data(), nvec); + const int stride = ld * nvec; + linear_op().gather(ld, dim, nvec, 1, stride, source, destination, map); +} + +template +void LinearWorkspace::bicg_update(int ld, int dim, int nvec, bool direction, T* out, const T* x, const T* y, const T* a, const T* b) +{ + ModuleBase::timer::start("LinearWorkspace", "bicg_update"); + if (dim == 0 || nvec == 0) + { + ModuleBase::timer::end("LinearWorkspace", "bicg_update"); + return; + } + using CtDevice = typename ct::PsiToContainer::type; + const int64_t coefficient_elements = 2 * static_cast(nvec); + if (coefficients_.NumElements() < coefficient_elements) + { + coefficients_ = ct::Tensor(ct::DataTypeToEnum::value, ct::DeviceTypeToEnum::value, {coefficient_elements}); + } + host_coefficients_.resize(coefficient_elements); + const T* a_end = a + nvec; + const T* b_end = b + nvec; + const typename std::vector::iterator second_begin = host_coefficients_.begin() + nvec; + std::copy(a, a_end, host_coefficients_.begin()); + std::copy(b, b_end, second_begin); + T* coeff = coefficients_.template data(); + base_device::memory::synchronize_memory_op()(coeff, + host_coefficients_.data(), + coefficient_elements); + const T* second_coefficients = coeff + nvec; + linear_op().bicg_update(ld, dim, nvec, direction, out, x, y, coeff, second_coefficients); + ModuleBase::timer::end("LinearWorkspace", "bicg_update"); +} + +template +void LinearWorkspace::cgs_direction(int ld, int dim, int nvec, T* p, T* u, const T* r, const T* q, const T* beta) +{ + ModuleBase::timer::start("LinearWorkspace", "cgs_direction"); + if (dim == 0 || nvec == 0) + { + ModuleBase::timer::end("LinearWorkspace", "cgs_direction"); + return; + } + using CtDevice = typename ct::PsiToContainer::type; + if (coefficients_.NumElements() < nvec) + { + coefficients_ = ct::Tensor(ct::DataTypeToEnum::value, ct::DeviceTypeToEnum::value, {nvec}); + } + T* coeff = coefficients_.template data(); + base_device::memory::synchronize_memory_op()(coeff, beta, nvec); + linear_op().cgs_direction(ld, dim, nvec, p, u, r, q, coeff); + ModuleBase::timer::end("LinearWorkspace", "cgs_direction"); +} + +template +void LinearWorkspace::cgs_alpha(int ld, int dim, int nvec, T* q, T* direction, const T* u, const T* v, const T* alpha) +{ + ModuleBase::timer::start("LinearWorkspace", "cgs_alpha"); + if (dim == 0 || nvec == 0) + { + ModuleBase::timer::end("LinearWorkspace", "cgs_alpha"); + return; + } + T* coeff = coefficients_.template data(); + base_device::memory::synchronize_memory_op()(coeff, alpha, nvec); + linear_op().cgs_alpha(ld, dim, nvec, q, direction, u, v, coeff); + ModuleBase::timer::end("LinearWorkspace", "cgs_alpha"); +} + +template +void LinearWorkspace::cgs_finish(int ld, int dim, int nvec, T* x, T* r, const T* direction, const T* ad) +{ + ModuleBase::timer::start("LinearWorkspace", "cgs_finish"); + if (dim == 0 || nvec == 0) + { + ModuleBase::timer::end("LinearWorkspace", "cgs_finish"); + return; + } + linear_op().cgs_finish(ld, dim, nvec, x, r, direction, ad, coefficients_.template data()); + ModuleBase::timer::end("LinearWorkspace", "cgs_finish"); +} +template class LinearWorkspace, base_device::DEVICE_CPU>; +template class LinearWorkspace, base_device::DEVICE_CPU>; +#if defined(__CUDA) || defined(__ROCM) +template class LinearWorkspace, base_device::DEVICE_GPU>; +template class LinearWorkspace, base_device::DEVICE_GPU>; +#endif + +} // namespace hsolver diff --git a/source/source_hsolver/linear_workspace.h b/source/source_hsolver/linear_workspace.h new file mode 100644 index 00000000000..0b93a0f50ae --- /dev/null +++ b/source/source_hsolver/linear_workspace.h @@ -0,0 +1,114 @@ +#ifndef HSOLVER_LINEAR_WORKSPACE_H +#define HSOLVER_LINEAR_WORKSPACE_H + +#include "source_base/macros.h" +#include "source_base/module_container/ATen/core/tensor.h" +#include "source_hsolver/diag_comm_info.h" +#include "source_hsolver/linear_operator.h" +#include "source_hsolver/linear_solver_types.h" + +#include +#include +#include +#include + +namespace hsolver +{ + +/** @brief Balance solve timing on both normal and exceptional exits. */ +class LinearSolveTimer +{ + private: + const char* solver_; + + public: + explicit LinearSolveTimer(const char* solver); + ~LinearSolveTimer(); + LinearSolveTimer(const LinearSolveTimer&) = delete; + LinearSolveTimer& operator=(const LinearSolveTimer&) = delete; +}; + +/** @brief Reusable device arrays and collective block algebra for linear solvers. */ +template +class LinearWorkspace +{ + private: + ct::Tensor vectors_; + ct::Tensor dots_; + ct::Tensor partial_; + ct::Tensor coefficients_; + ct::Tensor mask_; + ct::Tensor permutation_; + std::vector host_dots_; + std::vector host_coefficients_; + int capacity_ = 0; + const int slots_; + std::int64_t operator_calls_ = 0; + std::int64_t operator_columns_ = 0; + const diag_comm_info comm_; + + public: + /** @brief Allocate only the recurrence slots required by the calling solver. */ + LinearWorkspace(const diag_comm_info& comm, int slots); + void prepare(const int ld, const int dim, const int nvec, const T* x, const T* b, const double tolerance, const int max_iter); + /** @brief Initialize recurrence storage once per cycle, including padding. */ + void clear(); + void reset_statistics(); + void statistics(LinearSolveResult* result) const; + void swap_vectors(int ld, int dim, int slots, int stride, T* vectors, const std::vector& pairs); + void apply(const LinearOperator& op, const T* x, T* y, int ld, int nvec); + void cgs_direction(int ld, int dim, int nvec, T* p, T* u, const T* r, const T* q, const T* beta); + void cgs_alpha(int ld, int dim, int nvec, T* q, T* direction, const T* u, const T* v, const T* alpha); + /** @brief Finish the CGS step using the coefficients uploaded by cgs_alpha. */ + void cgs_finish(int ld, int dim, int nvec, T* x, T* r, const T* direction, const T* ad); + T* data(const int slot); + void dot(const int ld, const int dim, const int nvec, const T* x, const T* y, T* out); + /** @brief Return two packed global inner products in a single collective. */ + void dot_pair(int ld, int dim, int nvec, const T* x, const T* y, const T* z, const T* w, T* first, T* second); + /** @brief Upload per-column coefficients and update all unmasked columns. */ + void batch(int ld, int dim, int nvec, T* out, const T* x, const T* y, T a, T b, const T* ca, const T* cb, const int* skip); + /** @brief Upload coefficients once; the caller reports zeros to select the GPU update path. */ + void gmres_update(int ld, + int dim, + int nvec, + T* solution, + T* residual, + const T* direction, + const T* image, + const T* coefficients, + bool has_zero_coefficients); + /** @brief Reorder solver columns after host convergence decisions. */ + void swap_columns(int ld, int dim, const std::vector& pairs); + /** @brief Restore original column order into caller storage. */ + void restore(int ld, int dim, int nvec, const std::vector& order, const T* source, T* destination); + /** @brief Upload two coefficient arrays and execute a fused recurrence. */ + void bicg_update(int ld, int dim, int nvec, bool direction, T* out, const T* x, const T* y, const T* a, const T* b); + void copy(const int ld, const int dim, const int nvec, const T* x, T* y) const; + void residual(const LinearOperator& op, const int ld, const int dim, const int nvec, const T* x, const T* b, T* r); + void verify(const LinearOperator& op, + const int ld, + const int dim, + const int nvec, + const T* x, + const T* b, + const std::vector& threshold, + T* scratch, + LinearSolveResult* result); +}; + +template +bool linear_bad_divisor(const T value) +{ + using Real = typename GetTypeReal::type; + return !std::isfinite(std::abs(value)) || std::abs(value) <= std::numeric_limits::min(); +} + +template +double linear_norm(const T value) +{ + return std::isfinite(std::abs(value)) ? std::sqrt(std::max(0.0, static_cast(std::real(value)))) + : std::numeric_limits::infinity(); +} + +} // namespace hsolver +#endif diff --git a/source/source_hsolver/module_genelpa/cblacs.h b/source/source_hsolver/module_genelpa/cblacs.h index 647a35170aa..3b671ef1f9f 100644 --- a/source/source_hsolver/module_genelpa/cblacs.h +++ b/source/source_hsolver/module_genelpa/cblacs.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_HSOLVER_MODULE_GENELPA_CBLACS_H +#define ABACUS_SOURCE_HSOLVER_MODULE_GENELPA_CBLACS_H + // blacs // Initialization #include "mpi.h" @@ -23,3 +25,5 @@ void Czgerv2d(int icontxt, int m, int n, std::complex *a, int lda, int r // Combine //void Cdgamx2d(int icontxt, int scope, int top, int m, int n, // double *a, int lda, int *ra, int *ca, int rcflag, int rdest, int cdest); + +#endif // ABACUS_SOURCE_HSOLVER_MODULE_GENELPA_CBLACS_H diff --git a/source/source_hsolver/module_genelpa/elpa_generic.h b/source/source_hsolver/module_genelpa/elpa_generic.h new file mode 100644 index 00000000000..1638dee751b --- /dev/null +++ b/source/source_hsolver/module_genelpa/elpa_generic.h @@ -0,0 +1,448 @@ +#ifndef ABACUS_SOURCE_HSOLVER_MODULE_GENELPA_ELPA_GENERIC_H +#define ABACUS_SOURCE_HSOLVER_MODULE_GENELPA_ELPA_GENERIC_H + +#include "elpa_new.h" +#include +/*! \brief generic C method for elpa_set + * + * \details + * \param handle handle of the ELPA object for which a key/value pair should be set + * \param name the name of the key + * \param value integer/double value to be set for the key + * \param error on return the error code, which can be queried with elpa_strerr() + * \result void + */ +inline void elpa_set(elpa_t handle, const char *name, int value, int *error) +{ + elpa_set_integer(handle, name, value, error); +} +inline void elpa_set(elpa_t handle, const char *name, double value, int *error) +{ + elpa_set_double(handle, name, value, error); +} + +/*! \brief generic C method for elpa_get + * + * \details + * \param handle handle of the ELPA object for which a key/value pair should be queried + * \param name the name of the key + * \param value integer/double value to be queried + * \param error on return the error code, which can be queried with elpa_strerr() + * \result void + */ +inline void elpa_get(elpa_t handle, const char *name, int *value, int *error) +{ + elpa_get_integer(handle, name, value, error); +} +inline void elpa_get(elpa_t handle, const char *name, double *value, int *error) +{ + elpa_get_double(handle, name, value, error); +} + +/*! \brief generic C method for elpa_eigenvectors + * + * \details + * \param handle handle of the ELPA object, which defines the problem + * \param a float/double float complex/double complex pointer to matrix a + * \param ev on return: float/double pointer to eigenvalues + * \param q on return: float/double float complex/double complex pointer to eigenvectors + * \param error on return the error code, which can be queried with elpa_strerr() + * \result void + */ +#if ELPA_API_VERSION <= 20210502 // ELPA 2021.05.002 and earlier versions +inline void elpa_eigenvectors(const elpa_t handle, double *a, double *ev, double *q, int *error) +{ + elpa_eigenvectors_d(handle, a, ev, q, error); +} + +inline void elpa_eigenvectors(const elpa_t handle, float *a, float *ev, float *q, int *error) +{ + elpa_eigenvectors_f(handle, a, ev, q, error); +} + +inline void elpa_eigenvectors(const elpa_t handle, std::complex *a, double *ev, std::complex *q, int *error) +{ + elpa_eigenvectors_dc(handle, reinterpret_cast(a), ev, reinterpret_cast(q), error); +} + +inline void elpa_eigenvectors(const elpa_t handle, std::complex *a, float *ev, std::complex *q, int *error) +{ + elpa_eigenvectors_fc(handle, reinterpret_cast(a), ev, reinterpret_cast(q), error); +} +#elif ELPA_API_VERSION < 20220501 // ELPA version between 2021.11.001 and 2022.05.001 +inline void elpa_eigenvectors(const elpa_t handle, double *a, double *ev, double *q, int *error) +{ + elpa_eigenvectors_all_host_arrays_d(handle, a, ev, q, error); +} + +inline void elpa_eigenvectors(const elpa_t handle, float *a, float *ev, float *q, int *error) +{ + elpa_eigenvectors_all_host_arrays_f(handle, a, ev, q, error); +} + +inline void elpa_eigenvectors(const elpa_t handle, std::complex *a, double *ev, std::complex *q, int *error) +{ + elpa_eigenvectors_all_host_arrays_dc(handle, reinterpret_cast(a), + ev, reinterpret_cast(q), error); +} + +inline void elpa_eigenvectors(const elpa_t handle, std::complex *a, float *ev, std::complex *q, int *error) +{ + elpa_eigenvectors_all_host_arrays_fc(handle, reinterpret_cast(a), + ev, reinterpret_cast(q), error); +} +#else // ELPA version 2022.05.001, ELPA has its own c++ interface from version 2022.11.001 +inline void elpa_eigenvectors(const elpa_t handle, double *a, double *ev, double *q, int *error) +{ + elpa_eigenvectors_a_h_a_d(handle, a, ev, q, error); +} + +inline void elpa_eigenvectors(const elpa_t handle, float *a, float *ev, float *q, int *error) +{ + elpa_eigenvectors_a_h_a_f(handle, a, ev, q, error); +} + +inline void elpa_eigenvectors(const elpa_t handle, std::complex *a, double *ev, std::complex *q, int *error) +{ + elpa_eigenvectors_a_h_a_dc(handle, reinterpret_cast(a), + ev, reinterpret_cast(q), error); +} + +inline void elpa_eigenvectors(const elpa_t handle, std::complex *a, float *ev, std::complex *q, int *error) +{ + elpa_eigenvectors_a_h_a_fc(handle, reinterpret_cast(a), + ev, reinterpret_cast(q), error); +} +#endif + +/*! \brief generic C method for elpa_skew_eigenvectors + * + * \details + * \param handle handle of the ELPA object, which defines the problem + * \param a float/double float complex/double complex pointer to matrix a + * \param ev on return: float/double pointer to eigenvalues + * \param q on return: float/double float complex/double complex pointer to eigenvectors + * \param error on return the error code, which can be queried with elpa_strerr() + * \result void + */ +#if ELPA_API_VERSION <= 20210502 // ELPA 2021.05.002 and earlier versions +inline void elpa_skew_eigenvectors(const elpa_t handle, double *a, double *ev, double *q, int *error) +{ + elpa_eigenvectors_d(handle, a, ev, q, error); +} + +inline void elpa_skew_eigenvectors(const elpa_t handle, float *a, float *ev, float *q, int *error) +{ + elpa_eigenvectors_f(handle, a, ev, q, error); +} +#elif ELPA_API_VERSION < 20220501 // ELPA version between 2021.11.001 and 2022.05.001 +inline void elpa_skew_eigenvectors(const elpa_t handle, double *a, double *ev, double *q, int *error) +{ + elpa_skew_eigenvectors_all_host_arrays_d(handle, a, ev, q, error); +} + +inline void elpa_skew_eigenvectors(const elpa_t handle, float *a, float *ev, float *q, int *error) +{ + elpa_skew_eigenvectors_all_host_arrays_f(handle, a, ev, q, error); +} +#else // ELPA version 2022.05.001, ELPA has its own c++ interface from version 2022.11.001 +inline void elpa_skew_eigenvectors(const elpa_t handle, double *a, double *ev, double *q, int *error) +{ + elpa_skew_eigenvectors_a_h_a_d(handle, a, ev, q, error); +} + +inline void elpa_skew_eigenvectors(const elpa_t handle, float *a, float *ev, float *q, int *error) +{ + elpa_skew_eigenvectors_a_h_a_f(handle, a, ev, q, error); +} +#endif + + + +/*! \brief generic C method for elpa_generalized_eigenvectors + * + * \details + * \param handle handle of the ELPA object, which defines the problem + * \param a float/double float complex/double complex pointer to matrix a + * \param b float/double float complex/double complex pointer to matrix b + * \param ev on return: float/double pointer to eigenvalues + * \param q on return: float/double float complex/double complex pointer to eigenvectors + * \param is_already_decomposed set to 1, if b already decomposed by previous call to elpa_generalized + * \param error on return the error code, which can be queried with elpa_strerr() + * \result void + */ +inline void elpa_generalized_eigenvectors(elpa_t handle, double *a, double *b, double *ev, double *q, int is_already_decomposed, int *error) +{ + elpa_generalized_eigenvectors_d(handle, a, b, ev, q, is_already_decomposed, error); +} + +inline void elpa_generalized_eigenvectors(elpa_t handle, float *a, float *b, float *ev, float *q, int is_already_decomposed, int *error) +{ + elpa_generalized_eigenvectors_f(handle, a, b, ev, q, is_already_decomposed, error); +} + +inline void elpa_generalized_eigenvectors(elpa_t handle, std::complex *a, std::complex *b, double *ev, std::complex *q, int is_already_decomposed, int *error) +{ + elpa_generalized_eigenvectors_dc(handle, reinterpret_cast(a), reinterpret_cast(b), + ev, reinterpret_cast(q), is_already_decomposed, error); +} + +inline void elpa_generalized_eigenvectors(elpa_t handle, std::complex *a, std::complex *b, float *ev, std::complex *q, int is_already_decomposed, int *error) +{ + elpa_generalized_eigenvectors_fc(handle, reinterpret_cast(a), reinterpret_cast(b), + ev, reinterpret_cast(q), is_already_decomposed, error); +} + +/*! \brief generic C method for elpa_eigenvalues + * + * \details + * \param handle handle of the ELPA object, which defines the problem + * \param a float/double float complex/double complex pointer to matrix a + * \param ev on return: float/double pointer to eigenvalues + * \param error on return the error code, which can be queried with elpa_strerr() + * \result void + */ +#if ELPA_API_VERSION <= 20210502 // ELPA 2021.05.002 and earlier versions +inline void elpa_eigenvalues(elpa_t handle, double *a, double *ev, int *error) +{ + elpa_eigenvalues_d(handle, a, ev, error); +} +inline void elpa_eigenvalues(elpa_t handle, float *a, float *ev, int *error) +{ + elpa_eigenvalues_f(handle, a, ev, error); +} +inline void elpa_eigenvalues(elpa_t handle, std::complex *a, double *ev, int *error) +{ + elpa_eigenvalues_dc(handle, reinterpret_cast(a), ev, error); +} +inline void elpa_eigenvalues(elpa_t handle, std::complex *a, float *ev, int *error) +{ + elpa_eigenvalues_fc (handle, reinterpret_cast(a), ev, error); +} +#elif ELPA_API_VERSION < 20220501 // ELPA version between 2021.11.001 and 2022.05.001 +inline void elpa_eigenvalues(elpa_t handle, double *a, double *ev, int *error) +{ + elpa_eigenvalues_all_host_arrays_d(handle, a, ev, error); +} +inline void elpa_eigenvalues(elpa_t handle, float *a, float *ev, int *error) +{ + elpa_eigenvalues_all_host_arrays_f(handle, a, ev, error); +} +inline void elpa_eigenvalues(elpa_t handle, std::complex *a, double *ev, int *error) +{ + elpa_eigenvalues_all_host_arrays_dc(handle, reinterpret_cast(a), ev, error); +} +inline void elpa_eigenvalues(elpa_t handle, std::complex *a, float *ev, int *error) +{ + elpa_eigenvalues_all_host_arrays_fc(handle, reinterpret_cast(a), ev, error); +} +#else // ELPA version 2022.05.001, ELPA has its own c++ interface from version 2022.11.001 +inline void elpa_eigenvalues(elpa_t handle, double *a, double *ev, int *error) +{ + elpa_eigenvalues_a_h_a_d(handle, a, ev, error); +} +inline void elpa_eigenvalues(elpa_t handle, float *a, float *ev, int *error) +{ + elpa_eigenvalues_a_h_a_f(handle, a, ev, error); +} +inline void elpa_eigenvalues(elpa_t handle, std::complex *a, double *ev, int *error) +{ + elpa_eigenvalues_a_h_a_dc(handle, reinterpret_cast(a), ev, error); +} +inline void elpa_eigenvalues(elpa_t handle, std::complex *a, float *ev, int *error) +{ + elpa_eigenvalues_a_h_a_fc(handle, reinterpret_cast(a), ev, error); +} +#endif + +/*! \brief generic C method for elpa_skew_eigenvalues + * + * \details + * \param handle handle of the ELPA object, which defines the problem + * \param a float/double float complex/double complex pointer to matrix a + * \param ev on return: float/double pointer to eigenvalues + * \param error on return the error code, which can be queried with elpa_strerr() + * \result void + */ +#if ELPA_API_VERSION <= 20210502 // ELPA 2021.05.002 and earlier versions +inline void elpa_skew_eigenvalues(elpa_t handle, double *a, double *ev, int *error) +{ + elpa_eigenvalues_d(handle, a, ev, error); +} +inline void elpa_skew_eigenvalues(elpa_t handle, float *a, float *ev, int *error) +{ + elpa_eigenvalues_f(handle, a, ev, error); +} +#elif ELPA_API_VERSION < 20220501 // ELPA version between 2021.11.001 and 2022.05.001 +inline void elpa_skew_eigenvalues(elpa_t handle, double *a, double *ev, int *error) +{ + elpa_eigenvalues_all_host_arrays_d(handle, a, ev, error); +} +inline void elpa_skew_eigenvalues(elpa_t handle, float *a, float *ev, int *error) +{ + elpa_eigenvalues_all_host_arrays_f(handle, a, ev, error); +} +#else // ELPA version 2022.05.001, ELPA has its own c++ interface from version 2022.11.001 +inline void elpa_skew_eigenvalues(elpa_t handle, double *a, double *ev, int *error) +{ + elpa_eigenvalues_a_h_a_d(handle, a, ev, error); +} +inline void elpa_skew_eigenvalues(elpa_t handle, float *a, float *ev, int *error) +{ + elpa_eigenvalues_a_h_a_f(handle, a, ev, error); +} +#endif + +/*! \brief generic C method for elpa_cholesky + * + * \details + * \param handle handle of the ELPA object, which defines the problem + * \param a float/double float complex/double complex pointer to matrix a, for which + * the cholesky factorizaion will be computed + * \param error on return the error code, which can be queried with elpa_strerr() + * \result void + */ + +#if ELPA_API_VERSION < 20220501 // ELPA version before 2022.05.001 +inline void elpa_cholesky(elpa_t handle, double *a, int *error) +{ + elpa_cholesky_d(handle, a, error); +} +inline void elpa_cholesky(elpa_t handle, float *a, int *error) +{ + elpa_cholesky_f(handle, a, error); +} +inline void elpa_cholesky(elpa_t handle, std::complex *a, int *error) +{ + elpa_cholesky_dc(handle, reinterpret_cast(a), error); +} +inline void elpa_cholesky(elpa_t handle, std::complex *a, int *error) +{ + elpa_cholesky_fc(handle, reinterpret_cast(a), error); +} +#else +inline void elpa_cholesky(elpa_t handle, double *a, int *error) +{ + elpa_cholesky_a_h_a_d(handle, a, error); +} +inline void elpa_cholesky(elpa_t handle, float *a, int *error) +{ + elpa_cholesky_a_h_a_f(handle, a, error); +} +inline void elpa_cholesky(elpa_t handle, std::complex *a, int *error) +{ + elpa_cholesky_a_h_a_dc(handle, reinterpret_cast(a), error); +} +inline void elpa_cholesky(elpa_t handle, std::complex *a, int *error) +{ + elpa_cholesky_a_h_a_fc(handle, reinterpret_cast(a), error); +} +#endif + +/*! \brief generic C method for elpa_hermitian_multiply + * + * \details + * \param handle handle of the ELPA object, which defines the problem + * \param uplo_a descriptor for matrix a + * \param uplo_c descriptor for matrix c + * \param ncb int + * \param a float/double float complex/double complex pointer to matrix a + * \param b float/double float complex/double complex pointer to matrix b + * \param nrows_b number of rows for matrix b + * \param ncols_b number of cols for matrix b + * \param c float/double float complex/double complex pointer to matrix c + * \param nrows_c number of rows for matrix c + * \param ncols_c number of cols for matrix c + * \param error on return the error code, which can be queried with elpa_strerr() + * \result void + */ +#if ELPA_API_VERSION < 20220501 // ELPA version before 2022.05.001 +inline void elpa_hermitian_multiply(elpa_t handle, char uplo_a, char uplo_c, int ncb, double *a, double *b, int nrows_b, int ncols_b, double *c, int nrows_c, int ncols_c, int *error) +{ + elpa_hermitian_multiply_d(handle, uplo_a, uplo_c, ncb, a, b, nrows_b, ncols_b, c, nrows_c, ncols_c, error); +} +inline void elpa_hermitian_multiply(elpa_t handle, char uplo_a, char uplo_c, int ncb, float *a, float *b, int nrows_b, int ncols_b, float *c, int nrows_c, int ncols_c, int *error) +{ + elpa_hermitian_multiply_df(handle, uplo_a, uplo_c, ncb, a, b, nrows_b, ncols_b, c, nrows_c, ncols_c, error); +} +inline void elpa_hermitian_multiply(elpa_t handle, char uplo_a, char uplo_c, int ncb, std::complex *a, std::complex *b, int nrows_b, int ncols_b, std::complex *c, int nrows_c, int ncols_c, int *error) +{ + elpa_hermitian_multiply_dc(handle, uplo_a, uplo_c, ncb, reinterpret_cast(a), + reinterpret_cast(b), nrows_b, ncols_b, + reinterpret_cast(c), nrows_c, ncols_c, error); +} +inline void elpa_hermitian_multiply(elpa_t handle, char uplo_a, char uplo_c, int ncb, std::complex *a, std::complex *b, int nrows_b, int ncols_b, std::complex *c, int nrows_c, int ncols_c, int *error) +{ + elpa_hermitian_multiply_fc(handle, uplo_a, uplo_c, ncb, reinterpret_cast(a), + reinterpret_cast(b), nrows_b, ncols_b, + reinterpret_cast(c), nrows_c, ncols_c, error); +} +#else +inline void elpa_hermitian_multiply(elpa_t handle, char uplo_a, char uplo_c, int ncb, double *a, double *b, int nrows_b, int ncols_b, double *c, int nrows_c, int ncols_c, int *error) +{ + elpa_hermitian_multiply_a_h_a_d(handle, uplo_a, uplo_c, ncb, a, b, nrows_b, ncols_b, c, nrows_c, ncols_c, error); +} +inline void elpa_hermitian_multiply(elpa_t handle, char uplo_a, char uplo_c, int ncb, float *a, float *b, int nrows_b, int ncols_b, float *c, int nrows_c, int ncols_c, int *error) +{ + elpa_hermitian_multiply_a_h_a_f(handle, uplo_a, uplo_c, ncb, a, b, nrows_b, ncols_b, c, nrows_c, ncols_c, error); +} +inline void elpa_hermitian_multiply(elpa_t handle, char uplo_a, char uplo_c, int ncb, std::complex *a, std::complex *b, int nrows_b, int ncols_b, std::complex *c, int nrows_c, int ncols_c, int *error) +{ + elpa_hermitian_multiply_a_h_a_dc(handle, uplo_a, uplo_c, ncb, reinterpret_cast(a), + reinterpret_cast(b), nrows_b, ncols_b, + reinterpret_cast(c), nrows_c, ncols_c, error); +} +inline void elpa_hermitian_multiply(elpa_t handle, char uplo_a, char uplo_c, int ncb, std::complex *a, std::complex *b, int nrows_b, int ncols_b, std::complex *c, int nrows_c, int ncols_c, int *error) +{ + elpa_hermitian_multiply_a_h_a_fc(handle, uplo_a, uplo_c, ncb, reinterpret_cast(a), + reinterpret_cast(b), nrows_b, ncols_b, + reinterpret_cast(c), nrows_c, ncols_c, error); +} +#endif + +/*! \brief generic C method for elpa_invert_triangular + * + * \details + * \param handle handle of the ELPA object, which defines the problem + * \param a float/double float complex/double complex pointer to matrix a, which + * should be inverted + * \param error on return the error code, which can be queried with elpa_strerr() + * \result void + */ +#if ELPA_API_VERSION < 20220501 // ELPA version before 2022.05.001 +inline void elpa_invert_triangular(elpa_t handle, double *a, int *error) +{ + elpa_invert_trm_d(handle, a, error); +} +inline void elpa_invert_triangular(elpa_t handle, float *a, int *error) +{ + elpa_invert_trm_f(handle, a, error); +} +inline void elpa_invert_triangular(elpa_t handle, std::complex *a, int *error) +{ + elpa_invert_trm_dc(handle, reinterpret_cast(a), error); +} +inline void elpa_invert_triangular(elpa_t handle, std::complex *a, int *error) +{ + elpa_invert_trm_fc(handle, reinterpret_cast(a), error); +} +#else +inline void elpa_invert_triangular(elpa_t handle, double *a, int *error) +{ + elpa_invert_trm_a_h_a_d(handle, a, error); +} +inline void elpa_invert_triangular(elpa_t handle, float *a, int *error) +{ + elpa_invert_trm_a_h_a_f(handle, a, error); +} +inline void elpa_invert_triangular(elpa_t handle, std::complex *a, int *error) +{ + elpa_invert_trm_a_h_a_dc(handle, reinterpret_cast(a), error); +} +inline void elpa_invert_triangular(elpa_t handle, std::complex *a, int *error) +{ + elpa_invert_trm_a_h_a_fc(handle, reinterpret_cast(a), error); +} +#endif + +#endif // ABACUS_SOURCE_HSOLVER_MODULE_GENELPA_ELPA_GENERIC_H diff --git a/source/source_hsolver/module_genelpa/elpa_generic.hpp b/source/source_hsolver/module_genelpa/elpa_generic.hpp deleted file mode 100644 index 8469e0e0765..00000000000 --- a/source/source_hsolver/module_genelpa/elpa_generic.hpp +++ /dev/null @@ -1,444 +0,0 @@ -#pragma once -#include "elpa_new.h" -#include -/*! \brief generic C method for elpa_set - * - * \details - * \param handle handle of the ELPA object for which a key/value pair should be set - * \param name the name of the key - * \param value integer/double value to be set for the key - * \param error on return the error code, which can be queried with elpa_strerr() - * \result void - */ -inline void elpa_set(elpa_t handle, const char *name, int value, int *error) -{ - elpa_set_integer(handle, name, value, error); -} -inline void elpa_set(elpa_t handle, const char *name, double value, int *error) -{ - elpa_set_double(handle, name, value, error); -} - -/*! \brief generic C method for elpa_get - * - * \details - * \param handle handle of the ELPA object for which a key/value pair should be queried - * \param name the name of the key - * \param value integer/double value to be queried - * \param error on return the error code, which can be queried with elpa_strerr() - * \result void - */ -inline void elpa_get(elpa_t handle, const char *name, int *value, int *error) -{ - elpa_get_integer(handle, name, value, error); -} -inline void elpa_get(elpa_t handle, const char *name, double *value, int *error) -{ - elpa_get_double(handle, name, value, error); -} - -/*! \brief generic C method for elpa_eigenvectors - * - * \details - * \param handle handle of the ELPA object, which defines the problem - * \param a float/double float complex/double complex pointer to matrix a - * \param ev on return: float/double pointer to eigenvalues - * \param q on return: float/double float complex/double complex pointer to eigenvectors - * \param error on return the error code, which can be queried with elpa_strerr() - * \result void - */ -#if ELPA_API_VERSION <= 20210502 // ELPA 2021.05.002 and earlier versions -inline void elpa_eigenvectors(const elpa_t handle, double *a, double *ev, double *q, int *error) -{ - elpa_eigenvectors_d(handle, a, ev, q, error); -} - -inline void elpa_eigenvectors(const elpa_t handle, float *a, float *ev, float *q, int *error) -{ - elpa_eigenvectors_f(handle, a, ev, q, error); -} - -inline void elpa_eigenvectors(const elpa_t handle, std::complex *a, double *ev, std::complex *q, int *error) -{ - elpa_eigenvectors_dc(handle, reinterpret_cast(a), ev, reinterpret_cast(q), error); -} - -inline void elpa_eigenvectors(const elpa_t handle, std::complex *a, float *ev, std::complex *q, int *error) -{ - elpa_eigenvectors_fc(handle, reinterpret_cast(a), ev, reinterpret_cast(q), error); -} -#elif ELPA_API_VERSION < 20220501 // ELPA version between 2021.11.001 and 2022.05.001 -inline void elpa_eigenvectors(const elpa_t handle, double *a, double *ev, double *q, int *error) -{ - elpa_eigenvectors_all_host_arrays_d(handle, a, ev, q, error); -} - -inline void elpa_eigenvectors(const elpa_t handle, float *a, float *ev, float *q, int *error) -{ - elpa_eigenvectors_all_host_arrays_f(handle, a, ev, q, error); -} - -inline void elpa_eigenvectors(const elpa_t handle, std::complex *a, double *ev, std::complex *q, int *error) -{ - elpa_eigenvectors_all_host_arrays_dc(handle, reinterpret_cast(a), - ev, reinterpret_cast(q), error); -} - -inline void elpa_eigenvectors(const elpa_t handle, std::complex *a, float *ev, std::complex *q, int *error) -{ - elpa_eigenvectors_all_host_arrays_fc(handle, reinterpret_cast(a), - ev, reinterpret_cast(q), error); -} -#else // ELPA version 2022.05.001, ELPA has its own c++ interface from version 2022.11.001 -inline void elpa_eigenvectors(const elpa_t handle, double *a, double *ev, double *q, int *error) -{ - elpa_eigenvectors_a_h_a_d(handle, a, ev, q, error); -} - -inline void elpa_eigenvectors(const elpa_t handle, float *a, float *ev, float *q, int *error) -{ - elpa_eigenvectors_a_h_a_f(handle, a, ev, q, error); -} - -inline void elpa_eigenvectors(const elpa_t handle, std::complex *a, double *ev, std::complex *q, int *error) -{ - elpa_eigenvectors_a_h_a_dc(handle, reinterpret_cast(a), - ev, reinterpret_cast(q), error); -} - -inline void elpa_eigenvectors(const elpa_t handle, std::complex *a, float *ev, std::complex *q, int *error) -{ - elpa_eigenvectors_a_h_a_fc(handle, reinterpret_cast(a), - ev, reinterpret_cast(q), error); -} -#endif - -/*! \brief generic C method for elpa_skew_eigenvectors - * - * \details - * \param handle handle of the ELPA object, which defines the problem - * \param a float/double float complex/double complex pointer to matrix a - * \param ev on return: float/double pointer to eigenvalues - * \param q on return: float/double float complex/double complex pointer to eigenvectors - * \param error on return the error code, which can be queried with elpa_strerr() - * \result void - */ -#if ELPA_API_VERSION <= 20210502 // ELPA 2021.05.002 and earlier versions -inline void elpa_skew_eigenvectors(const elpa_t handle, double *a, double *ev, double *q, int *error) -{ - elpa_eigenvectors_d(handle, a, ev, q, error); -} - -inline void elpa_skew_eigenvectors(const elpa_t handle, float *a, float *ev, float *q, int *error) -{ - elpa_eigenvectors_f(handle, a, ev, q, error); -} -#elif ELPA_API_VERSION < 20220501 // ELPA version between 2021.11.001 and 2022.05.001 -inline void elpa_skew_eigenvectors(const elpa_t handle, double *a, double *ev, double *q, int *error) -{ - elpa_skew_eigenvectors_all_host_arrays_d(handle, a, ev, q, error); -} - -inline void elpa_skew_eigenvectors(const elpa_t handle, float *a, float *ev, float *q, int *error) -{ - elpa_skew_eigenvectors_all_host_arrays_f(handle, a, ev, q, error); -} -#else // ELPA version 2022.05.001, ELPA has its own c++ interface from version 2022.11.001 -inline void elpa_skew_eigenvectors(const elpa_t handle, double *a, double *ev, double *q, int *error) -{ - elpa_skew_eigenvectors_a_h_a_d(handle, a, ev, q, error); -} - -inline void elpa_skew_eigenvectors(const elpa_t handle, float *a, float *ev, float *q, int *error) -{ - elpa_skew_eigenvectors_a_h_a_f(handle, a, ev, q, error); -} -#endif - - - -/*! \brief generic C method for elpa_generalized_eigenvectors - * - * \details - * \param handle handle of the ELPA object, which defines the problem - * \param a float/double float complex/double complex pointer to matrix a - * \param b float/double float complex/double complex pointer to matrix b - * \param ev on return: float/double pointer to eigenvalues - * \param q on return: float/double float complex/double complex pointer to eigenvectors - * \param is_already_decomposed set to 1, if b already decomposed by previous call to elpa_generalized - * \param error on return the error code, which can be queried with elpa_strerr() - * \result void - */ -inline void elpa_generalized_eigenvectors(elpa_t handle, double *a, double *b, double *ev, double *q, int is_already_decomposed, int *error) -{ - elpa_generalized_eigenvectors_d(handle, a, b, ev, q, is_already_decomposed, error); -} - -inline void elpa_generalized_eigenvectors(elpa_t handle, float *a, float *b, float *ev, float *q, int is_already_decomposed, int *error) -{ - elpa_generalized_eigenvectors_f(handle, a, b, ev, q, is_already_decomposed, error); -} - -inline void elpa_generalized_eigenvectors(elpa_t handle, std::complex *a, std::complex *b, double *ev, std::complex *q, int is_already_decomposed, int *error) -{ - elpa_generalized_eigenvectors_dc(handle, reinterpret_cast(a), reinterpret_cast(b), - ev, reinterpret_cast(q), is_already_decomposed, error); -} - -inline void elpa_generalized_eigenvectors(elpa_t handle, std::complex *a, std::complex *b, float *ev, std::complex *q, int is_already_decomposed, int *error) -{ - elpa_generalized_eigenvectors_fc(handle, reinterpret_cast(a), reinterpret_cast(b), - ev, reinterpret_cast(q), is_already_decomposed, error); -} - -/*! \brief generic C method for elpa_eigenvalues - * - * \details - * \param handle handle of the ELPA object, which defines the problem - * \param a float/double float complex/double complex pointer to matrix a - * \param ev on return: float/double pointer to eigenvalues - * \param error on return the error code, which can be queried with elpa_strerr() - * \result void - */ -#if ELPA_API_VERSION <= 20210502 // ELPA 2021.05.002 and earlier versions -inline void elpa_eigenvalues(elpa_t handle, double *a, double *ev, int *error) -{ - elpa_eigenvalues_d(handle, a, ev, error); -} -inline void elpa_eigenvalues(elpa_t handle, float *a, float *ev, int *error) -{ - elpa_eigenvalues_f(handle, a, ev, error); -} -inline void elpa_eigenvalues(elpa_t handle, std::complex *a, double *ev, int *error) -{ - elpa_eigenvalues_dc(handle, reinterpret_cast(a), ev, error); -} -inline void elpa_eigenvalues(elpa_t handle, std::complex *a, float *ev, int *error) -{ - elpa_eigenvalues_fc (handle, reinterpret_cast(a), ev, error); -} -#elif ELPA_API_VERSION < 20220501 // ELPA version between 2021.11.001 and 2022.05.001 -inline void elpa_eigenvalues(elpa_t handle, double *a, double *ev, int *error) -{ - elpa_eigenvalues_all_host_arrays_d(handle, a, ev, error); -} -inline void elpa_eigenvalues(elpa_t handle, float *a, float *ev, int *error) -{ - elpa_eigenvalues_all_host_arrays_f(handle, a, ev, error); -} -inline void elpa_eigenvalues(elpa_t handle, std::complex *a, double *ev, int *error) -{ - elpa_eigenvalues_all_host_arrays_dc(handle, reinterpret_cast(a), ev, error); -} -inline void elpa_eigenvalues(elpa_t handle, std::complex *a, float *ev, int *error) -{ - elpa_eigenvalues_all_host_arrays_fc(handle, reinterpret_cast(a), ev, error); -} -#else // ELPA version 2022.05.001, ELPA has its own c++ interface from version 2022.11.001 -inline void elpa_eigenvalues(elpa_t handle, double *a, double *ev, int *error) -{ - elpa_eigenvalues_a_h_a_d(handle, a, ev, error); -} -inline void elpa_eigenvalues(elpa_t handle, float *a, float *ev, int *error) -{ - elpa_eigenvalues_a_h_a_f(handle, a, ev, error); -} -inline void elpa_eigenvalues(elpa_t handle, std::complex *a, double *ev, int *error) -{ - elpa_eigenvalues_a_h_a_dc(handle, reinterpret_cast(a), ev, error); -} -inline void elpa_eigenvalues(elpa_t handle, std::complex *a, float *ev, int *error) -{ - elpa_eigenvalues_a_h_a_fc(handle, reinterpret_cast(a), ev, error); -} -#endif - -/*! \brief generic C method for elpa_skew_eigenvalues - * - * \details - * \param handle handle of the ELPA object, which defines the problem - * \param a float/double float complex/double complex pointer to matrix a - * \param ev on return: float/double pointer to eigenvalues - * \param error on return the error code, which can be queried with elpa_strerr() - * \result void - */ -#if ELPA_API_VERSION <= 20210502 // ELPA 2021.05.002 and earlier versions -inline void elpa_skew_eigenvalues(elpa_t handle, double *a, double *ev, int *error) -{ - elpa_eigenvalues_d(handle, a, ev, error); -} -inline void elpa_skew_eigenvalues(elpa_t handle, float *a, float *ev, int *error) -{ - elpa_eigenvalues_f(handle, a, ev, error); -} -#elif ELPA_API_VERSION < 20220501 // ELPA version between 2021.11.001 and 2022.05.001 -inline void elpa_skew_eigenvalues(elpa_t handle, double *a, double *ev, int *error) -{ - elpa_eigenvalues_all_host_arrays_d(handle, a, ev, error); -} -inline void elpa_skew_eigenvalues(elpa_t handle, float *a, float *ev, int *error) -{ - elpa_eigenvalues_all_host_arrays_f(handle, a, ev, error); -} -#else // ELPA version 2022.05.001, ELPA has its own c++ interface from version 2022.11.001 -inline void elpa_skew_eigenvalues(elpa_t handle, double *a, double *ev, int *error) -{ - elpa_eigenvalues_a_h_a_d(handle, a, ev, error); -} -inline void elpa_skew_eigenvalues(elpa_t handle, float *a, float *ev, int *error) -{ - elpa_eigenvalues_a_h_a_f(handle, a, ev, error); -} -#endif - -/*! \brief generic C method for elpa_cholesky - * - * \details - * \param handle handle of the ELPA object, which defines the problem - * \param a float/double float complex/double complex pointer to matrix a, for which - * the cholesky factorizaion will be computed - * \param error on return the error code, which can be queried with elpa_strerr() - * \result void - */ - -#if ELPA_API_VERSION < 20220501 // ELPA version before 2022.05.001 -inline void elpa_cholesky(elpa_t handle, double *a, int *error) -{ - elpa_cholesky_d(handle, a, error); -} -inline void elpa_cholesky(elpa_t handle, float *a, int *error) -{ - elpa_cholesky_f(handle, a, error); -} -inline void elpa_cholesky(elpa_t handle, std::complex *a, int *error) -{ - elpa_cholesky_dc(handle, reinterpret_cast(a), error); -} -inline void elpa_cholesky(elpa_t handle, std::complex *a, int *error) -{ - elpa_cholesky_fc(handle, reinterpret_cast(a), error); -} -#else -inline void elpa_cholesky(elpa_t handle, double *a, int *error) -{ - elpa_cholesky_a_h_a_d(handle, a, error); -} -inline void elpa_cholesky(elpa_t handle, float *a, int *error) -{ - elpa_cholesky_a_h_a_f(handle, a, error); -} -inline void elpa_cholesky(elpa_t handle, std::complex *a, int *error) -{ - elpa_cholesky_a_h_a_dc(handle, reinterpret_cast(a), error); -} -inline void elpa_cholesky(elpa_t handle, std::complex *a, int *error) -{ - elpa_cholesky_a_h_a_fc(handle, reinterpret_cast(a), error); -} -#endif - -/*! \brief generic C method for elpa_hermitian_multiply - * - * \details - * \param handle handle of the ELPA object, which defines the problem - * \param uplo_a descriptor for matrix a - * \param uplo_c descriptor for matrix c - * \param ncb int - * \param a float/double float complex/double complex pointer to matrix a - * \param b float/double float complex/double complex pointer to matrix b - * \param nrows_b number of rows for matrix b - * \param ncols_b number of cols for matrix b - * \param c float/double float complex/double complex pointer to matrix c - * \param nrows_c number of rows for matrix c - * \param ncols_c number of cols for matrix c - * \param error on return the error code, which can be queried with elpa_strerr() - * \result void - */ -#if ELPA_API_VERSION < 20220501 // ELPA version before 2022.05.001 -inline void elpa_hermitian_multiply(elpa_t handle, char uplo_a, char uplo_c, int ncb, double *a, double *b, int nrows_b, int ncols_b, double *c, int nrows_c, int ncols_c, int *error) -{ - elpa_hermitian_multiply_d(handle, uplo_a, uplo_c, ncb, a, b, nrows_b, ncols_b, c, nrows_c, ncols_c, error); -} -inline void elpa_hermitian_multiply(elpa_t handle, char uplo_a, char uplo_c, int ncb, float *a, float *b, int nrows_b, int ncols_b, float *c, int nrows_c, int ncols_c, int *error) -{ - elpa_hermitian_multiply_df(handle, uplo_a, uplo_c, ncb, a, b, nrows_b, ncols_b, c, nrows_c, ncols_c, error); -} -inline void elpa_hermitian_multiply(elpa_t handle, char uplo_a, char uplo_c, int ncb, std::complex *a, std::complex *b, int nrows_b, int ncols_b, std::complex *c, int nrows_c, int ncols_c, int *error) -{ - elpa_hermitian_multiply_dc(handle, uplo_a, uplo_c, ncb, reinterpret_cast(a), - reinterpret_cast(b), nrows_b, ncols_b, - reinterpret_cast(c), nrows_c, ncols_c, error); -} -inline void elpa_hermitian_multiply(elpa_t handle, char uplo_a, char uplo_c, int ncb, std::complex *a, std::complex *b, int nrows_b, int ncols_b, std::complex *c, int nrows_c, int ncols_c, int *error) -{ - elpa_hermitian_multiply_fc(handle, uplo_a, uplo_c, ncb, reinterpret_cast(a), - reinterpret_cast(b), nrows_b, ncols_b, - reinterpret_cast(c), nrows_c, ncols_c, error); -} -#else -inline void elpa_hermitian_multiply(elpa_t handle, char uplo_a, char uplo_c, int ncb, double *a, double *b, int nrows_b, int ncols_b, double *c, int nrows_c, int ncols_c, int *error) -{ - elpa_hermitian_multiply_a_h_a_d(handle, uplo_a, uplo_c, ncb, a, b, nrows_b, ncols_b, c, nrows_c, ncols_c, error); -} -inline void elpa_hermitian_multiply(elpa_t handle, char uplo_a, char uplo_c, int ncb, float *a, float *b, int nrows_b, int ncols_b, float *c, int nrows_c, int ncols_c, int *error) -{ - elpa_hermitian_multiply_a_h_a_f(handle, uplo_a, uplo_c, ncb, a, b, nrows_b, ncols_b, c, nrows_c, ncols_c, error); -} -inline void elpa_hermitian_multiply(elpa_t handle, char uplo_a, char uplo_c, int ncb, std::complex *a, std::complex *b, int nrows_b, int ncols_b, std::complex *c, int nrows_c, int ncols_c, int *error) -{ - elpa_hermitian_multiply_a_h_a_dc(handle, uplo_a, uplo_c, ncb, reinterpret_cast(a), - reinterpret_cast(b), nrows_b, ncols_b, - reinterpret_cast(c), nrows_c, ncols_c, error); -} -inline void elpa_hermitian_multiply(elpa_t handle, char uplo_a, char uplo_c, int ncb, std::complex *a, std::complex *b, int nrows_b, int ncols_b, std::complex *c, int nrows_c, int ncols_c, int *error) -{ - elpa_hermitian_multiply_a_h_a_fc(handle, uplo_a, uplo_c, ncb, reinterpret_cast(a), - reinterpret_cast(b), nrows_b, ncols_b, - reinterpret_cast(c), nrows_c, ncols_c, error); -} -#endif - -/*! \brief generic C method for elpa_invert_triangular - * - * \details - * \param handle handle of the ELPA object, which defines the problem - * \param a float/double float complex/double complex pointer to matrix a, which - * should be inverted - * \param error on return the error code, which can be queried with elpa_strerr() - * \result void - */ -#if ELPA_API_VERSION < 20220501 // ELPA version before 2022.05.001 -inline void elpa_invert_triangular(elpa_t handle, double *a, int *error) -{ - elpa_invert_trm_d(handle, a, error); -} -inline void elpa_invert_triangular(elpa_t handle, float *a, int *error) -{ - elpa_invert_trm_f(handle, a, error); -} -inline void elpa_invert_triangular(elpa_t handle, std::complex *a, int *error) -{ - elpa_invert_trm_dc(handle, reinterpret_cast(a), error); -} -inline void elpa_invert_triangular(elpa_t handle, std::complex *a, int *error) -{ - elpa_invert_trm_fc(handle, reinterpret_cast(a), error); -} -#else -inline void elpa_invert_triangular(elpa_t handle, double *a, int *error) -{ - elpa_invert_trm_a_h_a_d(handle, a, error); -} -inline void elpa_invert_triangular(elpa_t handle, float *a, int *error) -{ - elpa_invert_trm_a_h_a_f(handle, a, error); -} -inline void elpa_invert_triangular(elpa_t handle, std::complex *a, int *error) -{ - elpa_invert_trm_a_h_a_dc(handle, reinterpret_cast(a), error); -} -inline void elpa_invert_triangular(elpa_t handle, std::complex *a, int *error) -{ - elpa_invert_trm_a_h_a_fc(handle, reinterpret_cast(a), error); -} -#endif diff --git a/source/source_hsolver/module_genelpa/elpa_new.h b/source/source_hsolver/module_genelpa/elpa_new.h index cfa724bfc2a..df36ad37a52 100644 --- a/source/source_hsolver/module_genelpa/elpa_new.h +++ b/source/source_hsolver/module_genelpa/elpa_new.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_HSOLVER_MODULE_GENELPA_ELPA_NEW_H +#define ABACUS_SOURCE_HSOLVER_MODULE_GENELPA_ELPA_NEW_H + #include @@ -37,6 +39,8 @@ extern "C" #endif } -#include "elpa_generic.hpp" // This is a wrapper for `elpa/elpa_generic.h`. +#include "elpa_generic.h" // This is a wrapper for `elpa/elpa_generic.h`. #endif + +#endif // ABACUS_SOURCE_HSOLVER_MODULE_GENELPA_ELPA_NEW_H diff --git a/source/source_hsolver/module_genelpa/elpa_runtime_check.h b/source/source_hsolver/module_genelpa/elpa_runtime_check.h index a1516e2e544..9ad6434bfac 100644 --- a/source/source_hsolver/module_genelpa/elpa_runtime_check.h +++ b/source/source_hsolver/module_genelpa/elpa_runtime_check.h @@ -1,5 +1,9 @@ -#pragma once +#ifndef ABACUS_SOURCE_HSOLVER_MODULE_GENELPA_ELPA_RUNTIME_CHECK_H +#define ABACUS_SOURCE_HSOLVER_MODULE_GENELPA_ELPA_RUNTIME_CHECK_H + #include void validate_elpa_complex_kernel(int kernel, int set_status, MPI_Comm comm); + +#endif // ABACUS_SOURCE_HSOLVER_MODULE_GENELPA_ELPA_RUNTIME_CHECK_H diff --git a/source/source_hsolver/module_genelpa/elpa_solver.h b/source/source_hsolver/module_genelpa/elpa_solver.h index 9216754a543..53a7c197710 100644 --- a/source/source_hsolver/module_genelpa/elpa_solver.h +++ b/source/source_hsolver/module_genelpa/elpa_solver.h @@ -1,13 +1,28 @@ -#pragma once +#ifndef ABACUS_SOURCE_HSOLVER_MODULE_GENELPA_ELPA_SOLVER_H +#define ABACUS_SOURCE_HSOLVER_MODULE_GENELPA_ELPA_SOLVER_H + #include "mpi.h" #include #include #include +/** + * @brief ELPA parallel eigenvalue solver wrapper + * + * @note The BLACS context (desc[1]) is borrowed from the caller (typically + * Parallel_2D). This class does NOT take ownership. The context must + * remain valid for the lifetime of this object. + */ class ELPA_Solver { public: + /** + * @param desc ScaLAPACK descriptor array. desc[1] must contain a valid + * BLACS context created by Parallel_2D::init() or Cblacs_gridinit(). + * The context must remain valid for the lifetime of this object. + * This class does NOT take ownership of the BLACS context. + */ ELPA_Solver(const bool isReal, const MPI_Comm comm, const int nev, @@ -97,3 +112,5 @@ class ELPA_Solver // debug tool void timer(int myid, const char function[], const char step[], double& t0); }; + +#endif // ABACUS_SOURCE_HSOLVER_MODULE_GENELPA_ELPA_SOLVER_H diff --git a/source/source_hsolver/module_genelpa/pblas.h b/source/source_hsolver/module_genelpa/pblas.h index 51ac6f3671c..5efca578455 100644 --- a/source/source_hsolver/module_genelpa/pblas.h +++ b/source/source_hsolver/module_genelpa/pblas.h @@ -1,12 +1,14 @@ -#pragma once +#ifndef ABACUS_SOURCE_HSOLVER_MODULE_GENELPA_PBLAS_H +#define ABACUS_SOURCE_HSOLVER_MODULE_GENELPA_PBLAS_H + void pddot_(int* n, double* dot, double* x, int* ix, int* jx, int* descx, int* incx, - double* y, int* iy, int* jy, int* descy, int* incy); - + double* y, int* iy, int* jy, int* descy, int* incy); + void pzdotc_(int* n, double _Complex* dot, double _Complex* x, int* ix, int* jx, int* descx, int* incx, - double _Complex* y, int* iy, int* jy, int* descy, int* incy); + double _Complex* y, int* iy, int* jy, int* descy, int* incy); void pdsymv_(char* uplo, int* n, - double* alpha, double* a, int* ia, int* ja, int* desca, - double* x, int* ix, int* jx, int* descx, int* incx, + double* alpha, double* a, int* ia, int* ja, int* desca, + double* x, int* ix, int* jx, int* descx, int* incx, double* beta, double* y, int* iy, int* jy, int* descy, int* incy); void pdtran_(int* m , int* n , double* alpha , double* a , int* ia , int* ja , int* desca , @@ -18,7 +20,7 @@ void pdgemm_(char* transa , char* transb , int* m , int* n , int* k , double* beta , double* c , int* ic , int* jc , int* descc ); void pzgemm_(char* transa , char* transb , int* m , int* n , int* k , double _Complex* alpha , double _Complex* a , int* ia , int* ja , int* desca , - double _Complex* b , int* ib , int* jb , int* descb , + double _Complex* b , int* ib , int* jb , int* descb , double _Complex* beta , double _Complex* c , int* ic , int* jc , int* descc ); void pdsymm_(char* side , char* uplo , int* m , int* n , double* alpha , double* a , int* ia , int* ja , int* desca , @@ -26,15 +28,17 @@ void pdsymm_(char* side , char* uplo , int* m , int* n , double* beta , double* c , int* ic , int* jc , int* descc ); void pzsymm_(char* side , char* uplo , int* m , int* n , double _Complex* alpha , double _Complex* a , int* ia , int* ja , int* desca , - double _Complex* b , int* ib , int* jb , int* descb , + double _Complex* b , int* ib , int* jb , int* descb , double _Complex* beta , double _Complex* c , int* ic , int* jc , int* descc ); void pzhemm_(char* side , char* uplo , int* m , int* n , double _Complex* alpha , double _Complex* a , int* ia , int* ja , int* desca , - double _Complex* b , int* ib , int* jb , int* descb , + double _Complex* b , int* ib , int* jb , int* descb , double _Complex* beta , double _Complex* c , int* ic , int* jc , int* descc ); void pdtrmm_(char* side , char* uplo , char* transa , char* diag , int* m , int* n , double* alpha , double* a , int* ia , int* ja , int* desca , double* b , int* ib , int* jb , int* descb ); void pztrmm_(char* side , char* uplo , char* transa , char* diag , int* m , int* n , double _Complex* alpha , double _Complex* a , int* ia , int* ja , int* desca , - double _Complex* b , int* ib , int* jb , int* descb ); + double _Complex* b , int* ib , int* jb , int* descb ); + +#endif // ABACUS_SOURCE_HSOLVER_MODULE_GENELPA_PBLAS_H diff --git a/source/source_hsolver/module_genelpa/scalapack.h b/source/source_hsolver/module_genelpa/scalapack.h index 39e18358e37..03a2b1a321b 100644 --- a/source/source_hsolver/module_genelpa/scalapack.h +++ b/source/source_hsolver/module_genelpa/scalapack.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_HSOLVER_MODULE_GENELPA_SCALAPACK_H +#define ABACUS_SOURCE_HSOLVER_MODULE_GENELPA_SCALAPACK_H + // scalapack int numroc_(const int *N, const int *NB, const int *IPROC, const int *ISRCPROC, const int *NPROCS); void descinit_(int *DESC, const int *M, const int *N, const int *MB, const int *NB, const int *IRSRC, const int *ICSRC, const int *ICTXT, const int *LLD, int *INFO); @@ -7,6 +9,8 @@ void pzpotrf_(const char *UPLO, const int *N, double _Complex *A, const int *IA, void pdsyev_(const char *JOBZ, const char *UPLO, int *N, double *A, int *IA, int *JA, int *DESCA, double *W, double *Z, int *IZ, int *JZ, int *DESCZ, double *WORK, int *LWORK, int *INFO); void pdgemr2d_(int *M, int *N, double *A, int *IA, int *JA, int *DESCA, - double *B, int *IB, int *JB, int *DESCB, int *ICTXT); + double *B, int *IB, int *JB, int *DESCB, int *ICTXT); void pzgemr2d_(int *M, int *N, double _Complex *A, int *IA, int *JA, int *DESCA, - double _Complex *B, int *IB, int *JB, int *DESCB, int *ICTXT); + double _Complex *B, int *IB, int *JB, int *DESCB, int *ICTXT); + +#endif // ABACUS_SOURCE_HSOLVER_MODULE_GENELPA_SCALAPACK_H diff --git a/source/source_hsolver/module_genelpa/utils.cpp b/source/source_hsolver/module_genelpa/utils.cpp index f14f69993d5..a334627c3a4 100644 --- a/source/source_hsolver/module_genelpa/utils.cpp +++ b/source/source_hsolver/module_genelpa/utils.cpp @@ -11,79 +11,6 @@ #include #include #include -#ifdef __MPI -void initBlacsGrid(int loglevel, - MPI_Comm comm, - int nFull, - int nblk, - int& blacs_ctxt, - int& narows, - int& nacols, - int desc[]) -{ - std::stringstream outlog; - char BLACS_LAYOUT = 'C'; - int ISRCPROC = 0; // fortran array starts from 1 - int nprows, npcols; - int myprow, mypcol; - int nprocs, myid; - int info = 0; - MPI_Comm_size(comm, &nprocs); - MPI_Comm_rank(comm, &myid); - // set blacs parameters - for (npcols = int(sqrt(double(nprocs))); npcols >= 2; --npcols) - { - if (nprocs % npcols == 0) - break; - } - nprows = nprocs / npcols; - if ((loglevel > 0 && myid == 0) || loglevel > 1) - { - outlog.str(""); - outlog << "myid " << myid << ": nprows: " << nprows << " ; npcols: " << npcols << std::endl; - std::cout << outlog.str(); - } - - // int comm_f = MPI_Comm_c2f(comm); - blacs_ctxt = Csys2blacs_handle(comm); - Cblacs_gridinit(&blacs_ctxt, &BLACS_LAYOUT, nprows, npcols); - if ((loglevel > 0 && myid == 0) || loglevel > 1) - { - outlog.str(""); - outlog << "myid " << myid << ": Cblacs_gridinit done, blacs_ctxt: " << blacs_ctxt << std::endl; - std::cout << outlog.str(); - } - Cblacs_gridinfo(blacs_ctxt, &nprows, &npcols, &myprow, &mypcol); - if ((loglevel > 0 && myid == 0) || loglevel > 1) - { - int mypnum = Cblacs_pnum(blacs_ctxt, myprow, mypcol); - int prow, pcol; - Cblacs_pcoord(blacs_ctxt, myid, &prow, &pcol); - outlog.str(""); - outlog << "myid " << myid << ": myprow: " << myprow << " ;mypcol: " << mypcol << std::endl; - outlog << "myid " << myid << ": mypnum: " << mypnum << std::endl; - outlog << "myid " << myid << ": prow: " << prow << " ;pcol: " << pcol << std::endl; - std::cout << outlog.str(); - } - - narows = numroc_(&nFull, &nblk, &myprow, &ISRCPROC, &nprows); - nacols = numroc_(&nFull, &nblk, &mypcol, &ISRCPROC, &npcols); - descinit_(desc, &nFull, &nFull, &nblk, &nblk, &ISRCPROC, &ISRCPROC, &blacs_ctxt, &narows, &info); - - if ((loglevel > 0 && myid == 0) || loglevel > 1) - { - outlog.str(""); - outlog << "myid " << myid << ": narows: " << narows << " nacols: " << nacols << std::endl; - outlog << "myid " << myid << ": blacs parameters setting" << std::endl; - outlog << "myid " << myid << ": desc is: "; - for (int i = 0; i < 9; ++i) - outlog << desc[i] << " "; - outlog << std::endl; - std::cout << outlog.str(); - } -} -#endif - // load matrix from the file void loadMatrix(const char FileName[], int nFull, double* a, int* desca, int blacs_ctxt) { diff --git a/source/source_hsolver/module_genelpa/utils.h b/source/source_hsolver/module_genelpa/utils.h index c0b8b7db8e1..73174b595a0 100644 --- a/source/source_hsolver/module_genelpa/utils.h +++ b/source/source_hsolver/module_genelpa/utils.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_HSOLVER_MODULE_GENELPA_UTILS_H +#define ABACUS_SOURCE_HSOLVER_MODULE_GENELPA_UTILS_H + #include #include @@ -16,17 +18,6 @@ static inline int localIndex(int globalIndex, int nblk, int nprocs, int& lcoalPr return int(globalIndex / (nblk * nprocs)) * nblk + globalIndex % nblk; } -#ifdef __MPI -void initBlacsGrid(int loglevel, - MPI_Comm comm, - int nFull, - int nblk, - int& blacs_ctxt, - int& narows, - int& nacols, - int desc[]); -#endif - // load matrix from the file void loadMatrix(const char FileName[], int nFull, double* a, int* desca, int blacs_ctxt); @@ -43,3 +34,5 @@ void saveLocalMatrix(const char filePrefix[], int narows, int nacols, std::compl // use pzgemr2d to collect matrix from all processes to root process // and save to one completed matrix file void saveMatrix(const char FileName[], int nFull, std::complex* a, int* desca, int blacs_ctxt); + +#endif // ABACUS_SOURCE_HSOLVER_MODULE_GENELPA_UTILS_H diff --git a/source/source_hsolver/orthonormal.cpp b/source/source_hsolver/orthonormal.cpp new file mode 100644 index 00000000000..d4fd9272676 --- /dev/null +++ b/source/source_hsolver/orthonormal.cpp @@ -0,0 +1,310 @@ +#include "source_hsolver/orthonormal.h" + +#include "source_base/parallel_device.h" +#include "source_base/timer.h" + +#include + +namespace hsolver +{ +namespace +{ +bool positive_norms(const std::vector>& gram, int bands) +{ + for (int band = 0; band < bands; ++band) + { + if (!(gram[band + band * bands].real() > 0.0)) + { + return false; + } + } + return true; +} +} // namespace + +template +Orthonormal::Orthonormal(const diag_comm_info& comm, LinearAlgebra& algebra) : comm_(comm), algebra_(algebra) +{ +} + +template +std::vector> Orthonormal::gram(const T* input, int ld, int dim, int bands) +{ + ModuleBase::timer::start("Orthonormal", "gram"); + std::vector> result = algebra_.gram(ld, dim, bands, input); + ModuleBase::timer::end("Orthonormal", "gram"); + return result; +} + +template +bool Orthonormal::factor(const std::vector>& g, + int bands, + OrthMethod method, + std::vector>* transform) +{ + ModuleBase::timer::start("Orthonormal", "factor"); + bool valid = false; + if (comm_.rank == 0) + { + valid = orth_transform(g, bands, method, transform); + } + double status = static_cast(valid); +#ifdef __MPI + if (comm_.nproc > 1) + { + Parallel_Common::bcast_data(&status, 1, comm_.comm, 0); + } +#endif + valid = status != 0.0; + if (valid) + { + transform->resize(static_cast(bands) * bands); +#ifdef __MPI + if (comm_.nproc > 1) + { + // Every partition must apply the same coefficients and take the same recovery branch. + const int elements = bands * bands; + Parallel_Common::bcast_data(transform->data(), elements, comm_.comm, 0); + } +#endif + } + ModuleBase::timer::end("Orthonormal", "factor"); + return valid; +} + +template +void Orthonormal::rotate(const T* input, + T* output, + int ld, + int dim, + int bands, + const std::vector>& transform) +{ + ModuleBase::timer::start("Orthonormal", "rotate"); + std::vector> delta(transform); + for (int i = 0; i < bands; ++i) + { + delta[i + i * bands] -= 1.0; + } + if (dim > 0) + { + base_device::memory::synchronize_memory_2d_op()(output, ld, input, ld, dim, bands); + } + algebra_.expand(ld, dim, bands, bands, input, delta, output, T(1)); + ModuleBase::timer::end("Orthonormal", "rotate"); +} + +template +bool Orthonormal::try_candidate(T* input, + int ld, + int dim, + int bands, + const std::vector>& transform, + std::vector>* g, + OrthResult* result) +{ + ModuleBase::timer::start("Orthonormal", "try_candidate"); + T* candidate = candidate_.template data(); + rotate(input, candidate, ld, dim, bands, transform); + ++result->passes; + const std::vector> check = gram(candidate, ld, dim, bands); + const double error = orth_error(check, bands); + const bool valid = std::isfinite(error) && positive_norms(check, bands); + const bool improved = valid && error < result->after; + if (improved) + { + if (dim > 0) + { + base_device::memory::synchronize_memory_2d_op()(input, ld, candidate, ld, dim, bands); + } + result->after = error; + *g = check; + result->failure = OrthFailure::tolerance_not_met; + } + else if (!valid || result->after > orth_tolerance()) + { + result->failure = valid ? OrthFailure::no_improvement : OrthFailure::invalid_candidate; + result->reason += std::string(orth_method_name(result->actual)) + ": " + orth_failure_name(result->failure) + "; "; + ++result->rejected; + } + ModuleBase::timer::end("Orthonormal", "try_candidate"); + return improved; +} + +template +void Orthonormal::correct(T* input, + int ld, + int dim, + int bands, + OrthMethod method, + std::vector>* g, + OrthResult* result) +{ + ModuleBase::timer::start("Orthonormal", "correct"); + std::vector methods{method}; + if (method != OrthMethod::cholesky) + { + methods.push_back(OrthMethod::cholesky); + } + if (method != OrthMethod::lowdin) + { + methods.push_back(OrthMethod::lowdin); + } + const int64_t elements = static_cast(ld) * bands; + linear_buffer(&candidate_, elements); + bool changed = false; + std::vector> transform; + for (int pass = 0; pass < 2; ++pass) + { + bool improved = false; + for (std::size_t attempt = 0; attempt < methods.size(); ++attempt) + { + result->actual = methods[attempt]; + if (attempt > 0) + { + ++result->fallbacks; + } + const bool factored = factor(*g, bands, result->actual, &transform); + if (factored) + { + improved = try_candidate(input, ld, dim, bands, transform, g, result); + } + else + { + result->failure = OrthFailure::factorization_failed; + result->reason += std::string(orth_method_name(result->actual)) + ": " + orth_failure_name(result->failure) + "; "; + } + if (improved || result->after <= orth_tolerance()) + { + break; + } + } + changed = changed || improved; + if (result->after <= orth_tolerance()) + { + result->status = changed ? OrthStatus::accepted : OrthStatus::unchanged; + result->failure = OrthFailure::none; + break; + } + // Another pass is useful only after an accepted update changed the Gram matrix. + if (!improved) + { + break; + } + } + ModuleBase::timer::end("Orthonormal", "correct"); +} + +template +OrthResult Orthonormal::inspect(const T* input, int ld, int dim, int bands, bool full_gram, bool collect_norms) +{ + ModuleBase::timer::start("Orthonormal", "inspect"); + OrthResult result; + result.gram_checked = full_gram; + if (full_gram) + { + const std::vector> g = gram(input, ld, dim, bands); + result.before = orth_error(g, bands); + result.after = result.before; + for (int band = 0; collect_norms && band < bands; ++band) + { + result.norms.push_back(g[band + band * bands].real()); + } + if (!std::isfinite(result.after)) + { + result.failure = OrthFailure::nonfinite_gram; + } + else if (!positive_norms(g, bands)) + { + result.failure = OrthFailure::nonpositive_norm; + } + else + { + result.status = OrthStatus::inspected; + } + } + else + { + // Corresponding-band products promote operands before multiplication, without a full FP64 copy. + std::vector> norms; + if (bands > 0) + { + norms = algebra_.dots(ld, dim, bands, 1, 0, input, input); + } + result.status = OrthStatus::disabled; + for (int band = 0; band < bands; ++band) + { + const std::complex value = norms[band]; + if (collect_norms) + { + result.norms.push_back(value.real()); + } + if (result.status == OrthStatus::failed) + { + continue; + } + if (!std::isfinite(value.real()) || !std::isfinite(value.imag())) + { + result.failure = OrthFailure::nonfinite_norm; + } + else if (value.real() <= 0.0) + { + result.failure = OrthFailure::nonpositive_norm; + } + if (result.failure != OrthFailure::none) + { + result.status = OrthStatus::failed; + result.reason = "band=" + std::to_string(band); + } + } + } + ModuleBase::timer::end("Orthonormal", "inspect"); + return result; +} + +template +OrthResult Orthonormal::apply(T* input, int ld, int dim, int bands, OrthMethod method, bool collect_norms) +{ + ModuleBase::timer::start("Orthonormal", "apply"); + if (method == OrthMethod::none) + { + const OrthResult result = inspect(input, ld, dim, bands, false, collect_norms); + ModuleBase::timer::end("Orthonormal", "apply"); + return result; + } + OrthResult result; + result.gram_checked = true; + std::vector> g = gram(input, ld, dim, bands); + result.before = orth_error(g, bands); + result.after = result.before; + if (!std::isfinite(result.before)) + { + result.failure = OrthFailure::nonfinite_gram; + } + else if (!positive_norms(g, bands)) + { + result.failure = OrthFailure::nonpositive_norm; + } + else if (result.before <= orth_tolerance()) + { + result.status = OrthStatus::unchanged; + } + else + { + correct(input, ld, dim, bands, method, &g, &result); + } + for (int i = 0; collect_norms && i < bands; ++i) + { + result.norms.push_back(g[i + i * bands].real()); + } + ModuleBase::timer::end("Orthonormal", "apply"); + return result; +} + +template class Orthonormal, base_device::DEVICE_CPU>; +template class Orthonormal, base_device::DEVICE_CPU>; +#if defined(__CUDA) || defined(__ROCM) +template class Orthonormal, base_device::DEVICE_GPU>; +template class Orthonormal, base_device::DEVICE_GPU>; +#endif +} // namespace hsolver diff --git a/source/source_hsolver/orthonormal.h b/source/source_hsolver/orthonormal.h new file mode 100644 index 00000000000..08deb139749 --- /dev/null +++ b/source/source_hsolver/orthonormal.h @@ -0,0 +1,104 @@ +#ifndef HSOLVER_ORTHONORMAL_H +#define HSOLVER_ORTHONORMAL_H + +#include "source_hsolver/linear_algebra.h" + +#include +#include +#include + +namespace hsolver +{ +enum class OrthMethod +{ + none, + cholesky, + lowdin, + newton_schulz +}; +OrthMethod parse_orth_method(const std::string& name); +const char* orth_method_name(OrthMethod method); + +enum class OrthStatus +{ + accepted, + unchanged, + inspected, ///< Read-only Gram diagnostics, without enforcing the correction tolerance. + disabled, + failed +}; + +enum class OrthFailure +{ + none, + nonfinite_gram, + nonfinite_norm, + nonpositive_norm, + factorization_failed, + invalid_candidate, + no_improvement, + tolerance_not_met +}; +const char* orth_failure_name(OrthFailure failure); + +/** @brief Acceptance tolerance for the native wavefunction precision. */ +template +constexpr double orth_tolerance() +{ + return std::is_same>::value ? 1e-6 : 1e-12; +} + +/** @brief Collective outcome; failed states must not be used to construct a density. */ +struct OrthResult +{ + double before = std::numeric_limits::quiet_NaN(); + double after = std::numeric_limits::quiet_NaN(); + bool gram_checked = false; + int passes = 0; + int fallbacks = 0; + OrthStatus status = OrthStatus::failed; + OrthFailure failure = OrthFailure::none; + OrthMethod actual = OrthMethod::none; + std::string reason; + int rejected = 0; + std::vector norms; +}; + +/** @brief Maximum elementwise distance from the identity, including nonfinite detection. */ +double orth_error(const std::vector>& gram, int bands); +/** @brief Construct one correction; the caller validates candidates and chooses fallbacks. */ +bool orth_transform(const std::vector>& gram, + int bands, + OrthMethod method, + std::vector>* transform); + +/** @brief Pool-local orthonormalization with FP64 products and native-precision updates. */ +template +class Orthonormal +{ + private: + const diag_comm_info comm_; + LinearAlgebra& algebra_; + ct::Tensor candidate_; + bool factor(const std::vector>& gram, int bands, OrthMethod method, std::vector>* transform); + bool try_candidate(T* input, + int ld, + int dim, + int bands, + const std::vector>& transform, + std::vector>* gram, + OrthResult* result); + void correct(T* input, int ld, int dim, int bands, OrthMethod method, std::vector>* gram, OrthResult* result); + void rotate(const T* input, T* output, int ld, int dim, int bands, const std::vector>& transform); + std::vector> gram(const T* input, int ld, int dim, int bands); + + public: + /** @brief Borrow sequential-use workspace with the same communicator; it must outlive this object. */ + Orthonormal(const diag_comm_info& comm, LinearAlgebra& algebra); + /** @brief Inspect norms and optionally overlaps without changing orbitals or enforcing a correction tolerance. */ + OrthResult inspect(const T* input, int ld, int dim, int bands, bool full_gram, bool collect_norms); + /** @brief Preserve input on rejected candidates; a failed overall result is not usable. */ + OrthResult apply(T* input, int ld, int dim, int bands, OrthMethod method, bool collect_norms); +}; +} // namespace hsolver +#endif diff --git a/source/source_hsolver/orthonormal_small.cpp b/source/source_hsolver/orthonormal_small.cpp new file mode 100644 index 00000000000..30d25b305a5 --- /dev/null +++ b/source/source_hsolver/orthonormal_small.cpp @@ -0,0 +1,248 @@ +#include "source_base/kernels/math_kernel_op.h" +#include "source_base/module_container/base/third_party/lapack.h" +#include "source_base/timer.h" +#include "source_hsolver/orthonormal.h" + +#include +#include +#include + +namespace hsolver +{ +namespace +{ +using Wide = std::complex; + +std::vector identity(int n) +{ + std::vector result(n * n, Wide(0)); + for (int i = 0; i < n; ++i) + { + result[i + i * n] = Wide(1); + } + return result; +} + +void multiply(const std::vector& a, const std::vector& b, int n, std::vector* result) +{ + const Wide one(1); + const Wide zero(0); + ModuleBase::gemm_op()('N', 'N', n, n, n, &one, a.data(), n, b.data(), n, &zero, result->data(), n); +} + +bool cholesky(const std::vector& g, int n, std::vector* c) +{ + ModuleBase::timer::start("Orthonormal", "cholesky"); + *c = g; + const char upper = 'U'; + const char nonunit = 'N'; + int info = 0; + zpotrf_(&upper, &n, c->data(), &n, &info); + if (info == 0) + { + ztrtri_(&upper, &nonunit, &n, c->data(), &n, &info); + } + for (int j = 0; j < n; ++j) + { + for (int i = j + 1; i < n; ++i) + { + (*c)[i + j * n] = Wide(0); + } + } + ModuleBase::timer::end("Orthonormal", "cholesky"); + return info == 0; +} + +bool lowdin(const std::vector& g, int n, std::vector* c) +{ + ModuleBase::timer::start("Orthonormal", "lowdin"); + std::vector u(g); + std::vector values(n); + const char vectors = 'V'; + const char upper = 'U'; + const int lwork = std::max(1, n * n + 2 * n); + const int lrwork = std::max(1, 1 + 5 * n + 2 * n * n); + const int liwork = std::max(1, 3 + 5 * n); + std::vector work(lwork); + std::vector rwork(lrwork); + std::vector iwork(liwork); + int info = 0; + zheevd_(&vectors, &upper, &n, u.data(), &n, values.data(), work.data(), &lwork, rwork.data(), &lrwork, iwork.data(), &liwork, &info); + bool valid = info == 0; + for (double value: values) + { + valid = valid && std::isfinite(value) && value > 0.0; + } + if (valid) + { + std::vector scaled(u); + for (int j = 0; j < n; ++j) + { + for (int i = 0; i < n; ++i) + { + scaled[i + j * n] /= std::sqrt(values[j]); + } + } + c->resize(n * n); + const Wide one(1); + const Wide zero(0); + ModuleBase::gemm_op()('N', 'C', n, n, n, &one, scaled.data(), n, u.data(), n, &zero, c->data(), n); + } + ModuleBase::timer::end("Orthonormal", "lowdin"); + return valid; +} + +bool newton_schulz(const std::vector& g, int n, std::vector* c) +{ + ModuleBase::timer::start("Orthonormal", "newton_schulz"); + constexpr double iteration_tol = 1e-12; + constexpr double max_deviation = 0.1; + constexpr int max_updates = 5; + const std::vector unit = identity(n); + double bound = 0.0; + for (int i = 0; i < n; ++i) + { + double sum = 0.0; + for (int j = 0; j < n; ++j) + { + sum += std::abs(g[i + j * n] - unit[i + j * n]); + } + bound = std::max(bound, sum); + } + bool valid = false; + *c = unit; + if (bound <= max_deviation && orth_error(g, n) <= iteration_tol) + { + valid = true; + } + else if (bound <= max_deviation) + { + // With C0 = I, the first update needs no matrix multiplication. + for (int i = 0; i < n * n; ++i) + { + (*c)[i] = 0.5 * (3.0 * unit[i] - g[i]); + } + std::vector square(n * n); + std::vector residual(n * n); + for (int iteration = 1; iteration <= max_updates; ++iteration) + { + multiply(*c, *c, n, &square); + multiply(g, square, n, &residual); + if (orth_error(residual, n) <= iteration_tol) + { + valid = true; + break; + } + if (iteration == max_updates) + { + break; + } + for (int i = 0; i < n * n; ++i) + { + residual[i] = 0.5 * (3.0 * unit[i] - residual[i]); + } + // The square is no longer needed; reuse its storage without aliasing GEMM inputs. + multiply(*c, residual, n, &square); + c->swap(square); + } + } + ModuleBase::timer::end("Orthonormal", "newton_schulz"); + return valid; +} +} // namespace + +const char* orth_method_name(OrthMethod method) +{ + switch (method) + { + case OrthMethod::cholesky: + return "cholesky"; + case OrthMethod::lowdin: + return "lowdin"; + case OrthMethod::newton_schulz: + return "newton_schulz"; + default: + return "none"; + } +} + +OrthMethod parse_orth_method(const std::string& name) +{ + for (OrthMethod method: {OrthMethod::none, OrthMethod::cholesky, OrthMethod::lowdin, OrthMethod::newton_schulz}) + { + if (name == orth_method_name(method)) + { + return method; + } + } + throw std::invalid_argument("Unsupported orthonormalization method: " + name); +} + +double orth_error(const std::vector& g, int n) +{ + double error = 0.0; + for (int j = 0; j < n; ++j) + { + for (int i = 0; i < n; ++i) + { + const Wide target = i == j ? Wide(1) : Wide(0); + const double value = std::abs(g[i + j * n] - target); + if (!std::isfinite(value)) + { + return std::numeric_limits::infinity(); + } + error = std::max(error, value); + } + } + return error; +} + +const char* orth_failure_name(OrthFailure failure) +{ + switch (failure) + { + case OrthFailure::nonfinite_gram: + return "nonfinite Gram matrix"; + case OrthFailure::nonfinite_norm: + return "nonfinite orbital norm"; + case OrthFailure::nonpositive_norm: + return "nonpositive orbital norm"; + case OrthFailure::factorization_failed: + return "factorization/iteration failed"; + case OrthFailure::invalid_candidate: + return "nonfinite or zero-norm candidate"; + case OrthFailure::no_improvement: + return "candidate did not improve orthogonality"; + case OrthFailure::tolerance_not_met: + return "orthogonality tolerance not met"; + default: + return "none"; + } +} + +bool orth_transform(const std::vector& g, int n, OrthMethod method, std::vector* c) +{ + ModuleBase::timer::start("Orthonormal", "orth_transform"); + bool valid = false; + switch (method) + { + case OrthMethod::cholesky: + valid = cholesky(g, n, c); + break; + case OrthMethod::lowdin: + valid = lowdin(g, n, c); + break; + case OrthMethod::newton_schulz: + valid = newton_schulz(g, n, c); + break; + default: + valid = false; + } + for (const Wide& value: *c) + { + valid = valid && std::isfinite(std::abs(value)); + } + ModuleBase::timer::end("Orthonormal", "orth_transform"); + return valid; +} +} // namespace hsolver diff --git a/source/source_hsolver/parallel_k2d.cpp b/source/source_hsolver/parallel_k2d.cpp index a80205440df..deb63db058d 100644 --- a/source/source_hsolver/parallel_k2d.cpp +++ b/source/source_hsolver/parallel_k2d.cpp @@ -26,9 +26,9 @@ void Parallel_K2D::set_para_env(int nks, this->RANK_IN_POOL, &this->POOL_WORLD_K2D); #endif - this->Pkpoints = new Parallel_Kpoints; - this->P2D_global = new Parallel_2D; - this->P2D_pool = new Parallel_2D; + this->Pkpoints.reset(new Parallel_Kpoints()); + this->P2D_global.reset(new Parallel_2D()); + this->P2D_pool.reset(new Parallel_2D()); this->Pkpoints ->kinfo(nks, kpar, this->MY_POOL, this->RANK_IN_POOL, nproc, nspin); this->P2D_global->init(nw, nw, nb2d, MPI_COMM_WORLD); @@ -36,7 +36,7 @@ void Parallel_K2D::set_para_env(int nks, } template -void Parallel_K2D::distribute_hsk(const typename Parallel_K2D::HskFunc& get_hsk, +void Parallel_K2D::distribute_hsk(hsolver::HSMatrix& hs, const std::vector& ik_kpar, const int& nw) { #ifdef __MPI @@ -44,7 +44,7 @@ void Parallel_K2D::distribute_hsk(const typename Parallel_K2D::HskFunc& for (int ipool = 0; ipool < ik_kpar.size(); ++ipool) { ModuleBase::MatrixBlock HK_global, SK_global; - get_hsk(ik_kpar[ipool], HK_global, SK_global); + hs.hs_at_k(ik_kpar[ipool], HK_global, SK_global); if (this->MY_POOL == this->Pkpoints->whichpool[ik_kpar[ipool]]) { this->hk_pool.resize(this->P2D_pool->get_local_size(), 0.0); this->sk_pool.resize(this->P2D_pool->get_local_size(), 0.0); @@ -85,18 +85,9 @@ void Parallel_K2D::distribute_hsk(const typename Parallel_K2D::HskFunc& template void Parallel_K2D::unset_para_env() { - if (this->Pkpoints != nullptr) { - delete this->Pkpoints; - this->Pkpoints = nullptr; - } - if (this->P2D_global != nullptr) { - delete this->P2D_global; - this->P2D_global = nullptr; - } - if (this->P2D_pool != nullptr) { - delete this->P2D_pool; - this->P2D_pool = nullptr; - } + this->Pkpoints.reset(); + this->P2D_global.reset(); + this->P2D_pool.reset(); MPI_Comm_free(&this->POOL_WORLD_K2D); } diff --git a/source/source_hsolver/parallel_k2d.h b/source/source_hsolver/parallel_k2d.h index fb4ee104c94..8b205b31003 100644 --- a/source/source_hsolver/parallel_k2d.h +++ b/source/source_hsolver/parallel_k2d.h @@ -4,11 +4,12 @@ #include "source_base/matrix_block.h" #include "source_base/parallel_2d.h" #include "source_cell/parallel_kpoints.h" +#include "source_hsolver/hs_matrix.h" #ifdef __MPI #include "mpi.h" #endif -#include +#include #include /*** @@ -35,14 +36,9 @@ class Parallel_K2D { const int& my_rank, const int& nspin); - /// Supplies H(k) and S(k) for one k point. The caller owns whatever has - /// to happen before the blocks are valid (updating the Hamiltonian for - /// that k point, for instance); this class only redistributes them. - using HskFunc = std::function< - void(int ik, ModuleBase::MatrixBlock& hk, ModuleBase::MatrixBlock& sk)>; - - /// this function distributes the Hk and Sk matrices to hk_pool and sk_pool - void distribute_hsk(const HskFunc& get_hsk, + /// this function distributes the Hk and Sk matrices to hk_pool and sk_pool; + /// hs supplies H(k) and S(k) for one k point, this class only redistributes them + void distribute_hsk(hsolver::HSMatrix& hs, const std::vector& ik_kpar, const int& nw); @@ -56,11 +52,11 @@ class Parallel_K2D { /// get my pool int get_my_pool() const { return this->MY_POOL; } /// get pKpoints - Parallel_Kpoints* get_pKpoints() const { return this->Pkpoints; } + Parallel_Kpoints* get_pKpoints() const { return this->Pkpoints.get(); } /// get p2D_global - Parallel_2D* get_p2D_global() const { return this->P2D_global; } + Parallel_2D* get_p2D_global() const { return this->P2D_global.get(); } /// get p2D_pool - Parallel_2D* get_p2D_pool() const { return this->P2D_pool; } + Parallel_2D* get_p2D_pool() const { return this->P2D_pool.get(); } /** * the local Hk, Sk matrices in POOL_WORLD_K2D @@ -88,9 +84,9 @@ class Parallel_K2D { /** * the pointer to Parallel_Kpoints */ - Parallel_Kpoints* Pkpoints = nullptr; - Parallel_2D* P2D_global = nullptr; - Parallel_2D* P2D_pool = nullptr; + std::unique_ptr Pkpoints; + std::unique_ptr P2D_global; + std::unique_ptr P2D_pool; }; #endif \ No newline at end of file diff --git a/source/source_hsolver/test/CMakeLists.txt b/source/source_hsolver/test/CMakeLists.txt index 043c0596c1f..f9e8749a12a 100644 --- a/source/source_hsolver/test/CMakeLists.txt +++ b/source/source_hsolver/test/CMakeLists.txt @@ -2,6 +2,35 @@ abacus_disable_feature_definitions(__CUDA) abacus_disable_feature_definitions(__ROCM) abacus_disable_feature_definitions(__EXX) +set(TD_LINEAR_TEST_SOURCES + ../hsolver_linear.cpp + ../linear_bicgstab.cpp + ../linear_cgs.cpp + ../linear_gmres.cpp + ../linear_algebra.cpp + ../cn_subspace.cpp + ../linear_low_rank.cpp + ../linear_workspace.cpp +) +AddTest( + TARGET MODULE_HSOLVER_linear + LIBS parameter base device container + SOURCES test_hsolver_linear.cpp test_cn_subspace.cpp test_linear_low_rank.cpp ${TD_LINEAR_TEST_SOURCES} +) +AddTest( + TARGET MODULE_HSOLVER_pw_tddft + LIBS parameter base device container psi planewave + SOURCES test_hsolver_pw_tddft.cpp test_orthonormal.cpp ../hsolver_pw_tddft.cpp ../orthonormal.cpp ../orthonormal_small.cpp ${TD_LINEAR_TEST_SOURCES} +) +if(ENABLE_MPI) + add_test( + NAME MODULE_HSOLVER_linear_parallel + COMMAND ${MPIEXEC_EXECUTABLE} ${MPIEXEC_NUMPROC_FLAG} 2 ${MPIEXEC_PREFLAGS} + $ ${MPIEXEC_POSTFLAGS} + ) + set_tests_properties(MODULE_HSOLVER_linear_parallel PROPERTIES PROCESSORS 2 TIMEOUT 120) +endif() + if (ENABLE_MPI) AddTest( TARGET MODULE_HSOLVER_parak2d_test @@ -13,40 +42,30 @@ if (ENABLE_MPI) LIBS parameter base psi device container SOURCES diago_bpcg_test.cpp ../diago_bpcg.cpp ../para_lin_tf.cpp ../diago_iter_assist.cpp ../../source_basis/module_pw/test/test_tool.cpp - ../../source_hamilt/operator.cpp - ../../source_pw/module_pwdft/op_pw.cpp ) AddTest( TARGET MODULE_HSOLVER_cg LIBS parameter base psi device container SOURCES diago_cg_test.cpp ../diago_cg.cpp ../diago_iter_assist.cpp ../diag_const_nums.cpp ../../source_basis/module_pw/test/test_tool.cpp - ../../source_hamilt/operator.cpp - ../../source_pw/module_pwdft/op_pw.cpp ) AddTest( TARGET MODULE_HSOLVER_cg_float LIBS parameter base psi device container SOURCES diago_cg_float_test.cpp ../diago_cg.cpp ../diago_iter_assist.cpp ../diag_const_nums.cpp ../../source_basis/module_pw/test/test_tool.cpp - ../../source_hamilt/operator.cpp - ../../source_pw/module_pwdft/op_pw.cpp ) AddTest( TARGET MODULE_HSOLVER_dav LIBS parameter base psi device SOURCES diago_david_test.cpp ../diago_david.cpp ../diago_iter_assist.cpp ../diag_const_nums.cpp ../../source_basis/module_pw/test/test_tool.cpp - ../../source_hamilt/operator.cpp - ../../source_pw/module_pwdft/op_pw.cpp ) AddTest( TARGET MODULE_HSOLVER_dav_float LIBS parameter base psi device SOURCES diago_david_float_test.cpp ../diago_david.cpp ../diago_iter_assist.cpp ../diag_const_nums.cpp ../../source_basis/module_pw/test/test_tool.cpp - ../../source_hamilt/operator.cpp - ../../source_pw/module_pwdft/op_pw.cpp ) if(ENABLE_LCAO) AddTest( @@ -54,23 +73,19 @@ if (ENABLE_MPI) LIBS parameter base psi device container SOURCES diago_cg_float_test.cpp ../diago_cg.cpp ../diago_iter_assist.cpp ../diag_const_nums.cpp ../../source_basis/module_pw/test/test_tool.cpp - ../../source_hamilt/operator.cpp - ../../source_pw/module_pwdft/op_pw.cpp ) AddTest( TARGET MODULE_HSOLVER_dav_real LIBS parameter base psi device SOURCES diago_david_real_test.cpp ../diago_david.cpp ../diago_iter_assist.cpp ../diag_const_nums.cpp ../../source_basis/module_pw/test/test_tool.cpp - ../../source_hamilt/operator.cpp - ../../source_pw/module_pwdft/op_pw.cpp ) endif() AddTest( TARGET MODULE_HSOLVER_base LIBS parameter psi device base - SOURCES test_hsolver.cpp + SOURCES test_hsolver.cpp ../hsolver.cpp ) AddTest( @@ -81,13 +96,6 @@ if (ENABLE_MPI) ../../source_cell/klist.cpp ../../source_cell/klist_io.cpp ../../source_cell/parallel_kpoints.cpp ../../source_cell/reciprocal_grid.cpp ) - AddTest( - TARGET MODULE_HSOLVER_sdft - LIBS parameter psi device base container - SOURCES test_hsolver_sdft.cpp ../hsolver_pw_sdft.cpp ../hsolver_pw.cpp ../diago_bpcg.cpp ../diago_dav_subspace.cpp ../diag_const_nums.cpp ../diago_iter_assist.cpp ../para_lin_tf.cpp - ../../source_estate/elecstate_tools.cpp ../../source_estate/occupy.cpp ../../source_base/module_fft/fft_bundle.cpp ../../source_base/module_fft/fft_cpu.cpp - ) - if(ENABLE_LCAO) if(TARGET ELPA::ELPA) AddTest( diff --git a/source/source_hsolver/test/diago_bpcg_test.cpp b/source/source_hsolver/test/diago_bpcg_test.cpp index 8a6a1558e16..34bed8b4735 100644 --- a/source/source_hsolver/test/diago_bpcg_test.cpp +++ b/source/source_hsolver/test/diago_bpcg_test.cpp @@ -2,8 +2,6 @@ #include "source_base/module_external/lapack_connector.h" #include "source_base/parallel_comm.h" #include "source_psi/psi.h" -#include "source_hamilt/hamilt.h" -#include "source_pw/module_pwdft/hamilt_pw.h" #include "../diago_iter_assist.h" #include "../diago_bpcg.h" #include "diago_mock.h" @@ -99,8 +97,7 @@ class DiagoBPCGPrepare //====================================================================== double *en = new double[npw]; int ik = 1; - hamilt::Hamilt>* ha; - ha =new hamilt::HamiltPW>(nullptr, nullptr, nullptr, nullptr, nullptr, nullptr, nullptr); + HSOperatorMock> ha; int* ngk = new int [1]; //psi::Psi> psi(ngk,ik,nband,npw); psi::Psi> psi; @@ -133,35 +130,13 @@ class DiagoBPCGPrepare psi_local.fix_k(0); double start, end; start = MPI_Wtime(); - using T = std::complex; - const int dim = DIAGOTEST::npw; - const std::vector &h_mat = DIAGOTEST::hmatrix_local; - auto hpsi_func = [h_mat, dim](T *psi_in, T *hpsi_out, - const int ld_psi, const int nvec) { - const T one(1.0); - const T zero(0.0); - - base_device::DEVICE_CPU *ctx = {}; - // hpsi_out(dim * nvec) = h_mat(dim * dim) * psi_in(dim * nvec) - ModuleBase::gemm_op()( - 'N', 'N', - dim, nvec, dim, - &one, - h_mat.data(), dim, - psi_in, ld_psi, - &zero, - hpsi_out, ld_psi); - }; - auto spsi_func = [](const T* psi_in, T* spsi_out, const int ld_psi, const int nvec) { - std::copy(psi_in, psi_in + ld_psi * nvec, spsi_out); - }; const int ndim = psi_local.get_current_ngk(); bpcg.init_iter(nband, nband, npw, ndim); std::vector ethr_band(nband, 1e-5); - bpcg.diag(hpsi_func, spsi_func, psi_local.get_pointer(), en, ethr_band); - bpcg.diag(hpsi_func, spsi_func, psi_local.get_pointer(), en, ethr_band); - bpcg.diag(hpsi_func, spsi_func, psi_local.get_pointer(), en, ethr_band); - bpcg.diag(hpsi_func, spsi_func, psi_local.get_pointer(), en, ethr_band); + bpcg.diag(ha, psi_local.get_pointer(), en, ethr_band); + bpcg.diag(ha, psi_local.get_pointer(), en, ethr_band); + bpcg.diag(ha, psi_local.get_pointer(), en, ethr_band); + bpcg.diag(ha, psi_local.get_pointer(), en, ethr_band); end = MPI_Wtime(); //if(mypnum == 0) printf("diago time:%7.3f\n",end-start); delete [] DIAGOTEST::npw_local; @@ -174,7 +149,6 @@ class DiagoBPCGPrepare delete[] en; delete[] e_lapack; - delete ha; } }; diff --git a/source/source_hsolver/test/diago_cg_float_test.cpp b/source/source_hsolver/test/diago_cg_float_test.cpp index d085a900bec..0651daca593 100644 --- a/source/source_hsolver/test/diago_cg_float_test.cpp +++ b/source/source_hsolver/test/diago_cg_float_test.cpp @@ -7,9 +7,7 @@ #include "source_base/inverse_matrix.h" #include "source_base/module_external/lapack_connector.h" #include "source_basis/module_pw/test/test_tool.h" -#include "source_hamilt/hamilt.h" #include "source_psi/psi.h" -#include "source_pw/module_pwdft/hamilt_pw.h" #include "gtest/gtest.h" #include @@ -104,8 +102,7 @@ class DiagoCGPrepare //====================================================================== float *en = new float[npw]; int ik = 1; - hamilt::Hamilt>* ha; - ha =new hamilt::HamiltPW>(nullptr, nullptr, nullptr, nullptr, nullptr, nullptr, nullptr); + HSOperatorMock> ha; psi::Psi> psi; psi.resize(ik,nband,npw); //psi.fix_k(0); @@ -143,54 +140,20 @@ class DiagoCGPrepare #else const hsolver::diag_comm_info diag_comm(mypnum, nprocs); #endif - // warp the subspace_func into a lambda function - auto subspace_func = [ha, &diag_comm](std::complex* psi_in, - std::complex* psi_out, - const int ld_psi, - const int nband, - const bool S_orth) { - auto psi_in_wrapper = psi::Psi>(psi_in, 1, nband, ld_psi, true); - auto psi_out_wrapper = psi::Psi>(psi_out, 1, nband, ld_psi, true); - std::vector eigen(nband, 0.0f); - hsolver::DiagoIterAssist>::diag_subspace(ha, - psi_in_wrapper, - psi_out_wrapper, - eigen.data(), - diag_comm); - }; hsolver::DiagoCG> cg("pw", "scf", hsolver::DiagoIterAssist>::need_subspace, - subspace_func, + diag_comm, hsolver::DiagoIterAssist>::PW_DIAG_THR, - hsolver::DiagoIterAssist>::PW_DIAG_NMAX, - nprocs); + hsolver::DiagoIterAssist>::PW_DIAG_NMAX); // hsolver::DiagoCG> cg(precondition_local); psi_local.fix_k(0); float start, end; start = MPI_Wtime(); - auto hpsi_func = [ha](std::complex* psi_in, - std::complex* hpsi_out, - const int ld_psi, - const int nvec) { - auto psi_wrapper = psi::Psi>(psi_in, 1, nvec, ld_psi, true); - psi::Range all_bands_range(true, 0, 0, nvec - 1); - using hpsi_info = typename hamilt::Operator>::hpsi_info; - hpsi_info info(&psi_wrapper, all_bands_range, hpsi_out); - ha->ops->hPsi(info); - }; - auto spsi_func = [ha](std::complex* psi_in, - std::complex* spsi_out, - const int ld_psi, - const int nvec) { - ha->sPsi(psi_in, spsi_out, ld_psi, ld_psi, nvec); - }; std::vector ethr_band(nband, 1e-5); - cg.diag(hpsi_func, - spsi_func, - psi_local.get_nbasis(), + cg.diag(ha, psi_local.get_nbasis(), psi_local.get_nbands(), psi_local.get_current_ngk(), psi_local.get_pointer(), @@ -211,7 +174,6 @@ class DiagoCGPrepare delete[] en; delete[] e_lapack; - delete ha; } }; diff --git a/source/source_hsolver/test/diago_cg_real_test.cpp b/source/source_hsolver/test/diago_cg_real_test.cpp index 7d5e364e892..f1afbd99a45 100644 --- a/source/source_hsolver/test/diago_cg_real_test.cpp +++ b/source/source_hsolver/test/diago_cg_real_test.cpp @@ -7,9 +7,7 @@ #include "source_base/inverse_matrix.h" #include "source_base/module_external/lapack_connector.h" #include "source_basis/module_pw/test/test_tool.h" -#include "source_hamilt/hamilt.h" #include "source_psi/psi.h" -#include "source_pw/module_pwdft/hamilt_pw.h" #include "gtest/gtest.h" #include @@ -105,8 +103,7 @@ class DiagoCGPrepare //====================================================================== double* en = new double[npw]; int ik = 1; - hamilt::Hamilt* ha; - ha = new hamilt::HamiltPW(nullptr, nullptr, nullptr, nullptr, nullptr, nullptr, nullptr); + HSOperatorMock ha; int* ngk = new int[1]; psi::Psi psi; psi.resize(ik, nband, npw); @@ -149,51 +146,20 @@ class DiagoCGPrepare #else const hsolver::diag_comm_info diag_comm(mypnum, nprocs); #endif - // warp the subspace_func into a lambda function - auto subspace_func - = [ha, &diag_comm](double* psi_in, double* psi_out, const int ld_psi, const int nband, const bool S_orth) { - auto psi_in_wrapper = psi::Psi(psi_in, 1, nband, ld_psi, true); - auto psi_out_wrapper = psi::Psi(psi_out, 1, nband, ld_psi, true); - std::vector eigen(nband, 0.0); - hsolver::DiagoIterAssist::diag_subspace(ha, - psi_in_wrapper, - psi_out_wrapper, - eigen.data(), - diag_comm); - }; hsolver::DiagoCG cg("pw", "scf", hsolver::DiagoIterAssist::need_subspace, - subspace_func, + diag_comm, hsolver::DiagoIterAssist::PW_DIAG_THR, - hsolver::DiagoIterAssist::PW_DIAG_NMAX, - nprocs); + hsolver::DiagoIterAssist::PW_DIAG_NMAX); // hsolver::DiagoCG cg(precondition_local); psi_local.fix_k(0); double start, end; start = MPI_Wtime(); - auto hpsi_func = [ha](double* psi_in, - double* hpsi_out, - const int ld_psi, - const int nvec) { - auto psi_wrapper = psi::Psi(psi_in, 1, nvec, ld_psi, true); - psi::Range all_bands_range(true, 0, 0, nvec - 1); - using hpsi_info = typename hamilt::Operator::hpsi_info; - hpsi_info info(&psi_wrapper, all_bands_range, hpsi_out); - ha->ops->hPsi(info); - }; - auto spsi_func = [ha](double* psi_in, - double* spsi_out, - const int ld_psi, - const int nvec) { - ha->sPsi(psi_in, spsi_out, ld_psi, ld_psi, nvec); - }; std::vector ethr_band(nband, 1e-5); - cg.diag(hpsi_func, - spsi_func, - psi_local.get_nbasis(), + cg.diag(ha, psi_local.get_nbasis(), psi_local.get_nbands(), psi_local.get_current_ngk(), psi_local.get_pointer(), @@ -214,7 +180,6 @@ class DiagoCGPrepare delete[] en; delete[] e_lapack; - delete ha; } }; diff --git a/source/source_hsolver/test/diago_cg_test.cpp b/source/source_hsolver/test/diago_cg_test.cpp index 7f304947d78..853b51c9f71 100644 --- a/source/source_hsolver/test/diago_cg_test.cpp +++ b/source/source_hsolver/test/diago_cg_test.cpp @@ -8,9 +8,7 @@ #include "source_base/inverse_matrix.h" #include "source_base/module_external/lapack_connector.h" #include "source_basis/module_pw/test/test_tool.h" -#include "source_hamilt/hamilt.h" #include "source_psi/psi.h" -#include "source_pw/module_pwdft/hamilt_pw.h" #include "gtest/gtest.h" #include @@ -101,8 +99,7 @@ class DiagoCGPrepare //====================================================================== double *en = new double[npw]; int ik = 1; - hamilt::Hamilt>* ha; - ha =new hamilt::HamiltPW>(nullptr, nullptr, nullptr, nullptr, nullptr, nullptr, nullptr); + HSOperatorMock> ha; psi::Psi> psi; psi.resize(ik,nband,npw); //psi.fix_k(0); @@ -138,54 +135,20 @@ class DiagoCGPrepare #else const hsolver::diag_comm_info diag_comm(mypnum, nprocs); #endif - // warp the subspace_func into a lambda function - auto subspace_func = [ha, &diag_comm](std::complex* psi_in, - std::complex* psi_out, - const int ld_psi, - const int nband, - const bool S_orth) { - auto psi_in_wrapper = psi::Psi>(psi_in, 1, nband, ld_psi, true); - auto psi_out_wrapper = psi::Psi>(psi_out, 1, nband, ld_psi, true); - std::vector eigen(nband, 0.0); - hsolver::DiagoIterAssist>::diag_subspace(ha, - psi_in_wrapper, - psi_out_wrapper, - eigen.data(), - diag_comm); - }; hsolver::DiagoCG> cg("pw", "scf", hsolver::DiagoIterAssist>::need_subspace, - subspace_func, + diag_comm, hsolver::DiagoIterAssist>::PW_DIAG_THR, - hsolver::DiagoIterAssist>::PW_DIAG_NMAX, - nprocs); + hsolver::DiagoIterAssist>::PW_DIAG_NMAX); // hsolver::DiagoCG> cg(precondition_local); psi_local.fix_k(0); double start, end; start = MPI_Wtime(); - auto hpsi_func = [ha](std::complex* psi_in, - std::complex* hpsi_out, - const int ld_psi, - const int nvec) { - auto psi_wrapper = psi::Psi>(psi_in, 1, nvec, ld_psi, true); - psi::Range all_bands_range(true, 0, 0, nvec - 1); - using hpsi_info = typename hamilt::Operator>::hpsi_info; - hpsi_info info(&psi_wrapper, all_bands_range, hpsi_out); - ha->ops->hPsi(info); - }; - auto spsi_func = [ha](std::complex* psi_in, - std::complex* spsi_out, - const int ld_psi, - const int nvec) { - ha->sPsi(psi_in, spsi_out, ld_psi, ld_psi, nvec); - }; std::vector ethr_band(nband, 1e-5); - cg.diag(hpsi_func, - spsi_func, - psi_local.get_nbasis(), + cg.diag(ha, psi_local.get_nbasis(), psi_local.get_nbands(), psi_local.get_current_ngk(), psi_local.get_pointer(), @@ -207,7 +170,6 @@ class DiagoCGPrepare delete[] en; delete[] e_lapack; - delete ha; } }; diff --git a/source/source_hsolver/test/diago_david_float_test.cpp b/source/source_hsolver/test/diago_david_float_test.cpp index a5ac77be1ae..930b4920cb6 100644 --- a/source/source_hsolver/test/diago_david_float_test.cpp +++ b/source/source_hsolver/test/diago_david_float_test.cpp @@ -2,7 +2,6 @@ #include "source_hsolver/diag_comm_info.h" #include"source_hsolver/diago_iter_assist.h" #include "source_base/parallel_comm.h" -#include"source_pw/module_pwdft/hamilt_pw.h" #include"diago_mock.h" #include "source_psi/psi.h" #include"gtest/gtest.h" @@ -83,8 +82,7 @@ class DiagoDavPrepare //do Diago_David::diag() float* en = new float[npw]; - hamilt::Hamilt> *phm; - phm = new hamilt::HamiltPW>(nullptr, nullptr, nullptr, nullptr, nullptr, nullptr, nullptr); + HSOperatorMock> phm; #ifdef __MPI const hsolver::diag_comm_info comm_info = {POOL_WORLD, mypnum, nprocs}; @@ -111,21 +109,8 @@ class DiagoDavPrepare #endif - auto hpsi_func = [phm](std::complex* psi_in,std::complex* hpsi_out, - const int ld_psi, const int nvec) - { - auto psi_iter_wrapper = psi::Psi>(psi_in, 1, nvec, ld_psi, true); - psi::Range bands_range(true, 0, 0, nvec-1); - using hpsi_info = typename hamilt::Operator>::hpsi_info; - hpsi_info info(&psi_iter_wrapper, bands_range, hpsi_out); - phm->ops->hPsi(info); - }; - auto spsi_func = [phm](const std::complex* psi_in, - std::complex* spsi_out, - const int ld_psi, - const int nbands) { phm->sPsi(psi_in, spsi_out, ld_psi, ld_psi, nbands); }; std::vector ethr_band(phi.get_nbands(), eps); - dav.diag(hpsi_func,spsi_func, ld_psi, phi.get_pointer(), en, ethr_band, maxiter); + dav.diag(phm, ld_psi, phi.get_pointer(), en, ethr_band, maxiter); #ifdef __MPI end = MPI_Wtime(); @@ -145,7 +130,6 @@ class DiagoDavPrepare } } delete [] en; - delete phm; delete [] e_lapack; } }; diff --git a/source/source_hsolver/test/diago_david_real_test.cpp b/source/source_hsolver/test/diago_david_real_test.cpp index 00f6917d652..3ff4e33bc7b 100644 --- a/source/source_hsolver/test/diago_david_real_test.cpp +++ b/source/source_hsolver/test/diago_david_real_test.cpp @@ -2,7 +2,6 @@ #include "source_hsolver/diag_comm_info.h" #include"source_hsolver/diago_iter_assist.h" #include "source_base/parallel_comm.h" -#include"source_pw/module_pwdft/hamilt_pw.h" #include"diago_mock.h" #include "source_psi/psi.h" #include"gtest/gtest.h" @@ -82,8 +81,7 @@ class DiagoDavPrepare //do Diago_David::diag() double* en = new double[npw]; - hamilt::Hamilt* phm; - phm = new hamilt::HamiltPW(nullptr, nullptr, nullptr, nullptr, nullptr, nullptr, nullptr); + HSOperatorMock phm; #ifdef __MPI const hsolver::diag_comm_info comm_info = {POOL_WORLD, mypnum, nprocs}; @@ -110,20 +108,8 @@ class DiagoDavPrepare #endif - auto hpsi_func = [phm](double* psi_in,double* hpsi_out, - const int ld_psi, const int nvec) - { - auto psi_iter_wrapper = psi::Psi(psi_in, 1, nvec, ld_psi, true); - psi::Range bands_range(true, 0, 0, nvec-1); - using hpsi_info = typename hamilt::Operator::hpsi_info; - hpsi_info info(&psi_iter_wrapper, bands_range, hpsi_out); - phm->ops->hPsi(info); - }; - auto spsi_func = [phm](const double* psi_in, double* spsi_out, const int ld_psi, const int nbands) { - phm->sPsi(psi_in, spsi_out, ld_psi, ld_psi, nbands); - }; std::vector ethr_band(phi.get_nbands(), eps); - dav.diag(hpsi_func,spsi_func, ld_psi, phi.get_pointer(), en, ethr_band, maxiter); + dav.diag(phm, ld_psi, phi.get_pointer(), en, ethr_band, maxiter); #ifdef __MPI end = MPI_Wtime(); @@ -143,7 +129,6 @@ class DiagoDavPrepare } } delete[] en; - delete phm; delete[] e_lapack; } }; diff --git a/source/source_hsolver/test/diago_david_test.cpp b/source/source_hsolver/test/diago_david_test.cpp index 771e0ae489d..7d4b8cbfea0 100644 --- a/source/source_hsolver/test/diago_david_test.cpp +++ b/source/source_hsolver/test/diago_david_test.cpp @@ -2,7 +2,6 @@ #include "source_hsolver/diag_comm_info.h" #include"source_hsolver/diago_iter_assist.h" #include "source_base/parallel_comm.h" -#include"source_pw/module_pwdft/hamilt_pw.h" #include"diago_mock.h" #include "source_psi/psi.h" #include"gtest/gtest.h" @@ -87,8 +86,7 @@ class DiagoDavPrepare //do Diago_David::diag() double* en = new double[npw]; - hamilt::Hamilt> *phm; - phm = new hamilt::HamiltPW>(nullptr, nullptr, nullptr, nullptr, nullptr, nullptr, nullptr); + HSOperatorMock> phm; #ifdef __MPI const hsolver::diag_comm_info comm_info = {POOL_WORLD, mypnum, nprocs}; @@ -115,21 +113,8 @@ class DiagoDavPrepare #endif - auto hpsi_func = [phm](std::complex* psi_in,std::complex* hpsi_out, - const int ld_psi, const int nvec) - { - auto psi_iter_wrapper = psi::Psi>(psi_in, 1, nvec, ld_psi, true); - psi::Range bands_range(true, 0, 0, nvec-1); - using hpsi_info = typename hamilt::Operator>::hpsi_info; - hpsi_info info(&psi_iter_wrapper, bands_range, hpsi_out); - phm->ops->hPsi(info); - }; - auto spsi_func = [phm](const std::complex* psi_in, - std::complex* spsi_out, - const int ld_psi, - const int nbands) { phm->sPsi(psi_in, spsi_out, ld_psi, ld_psi, nbands); }; std::vector ethr_band(phi.get_nbands(), eps); - dav.diag(hpsi_func,spsi_func, ld_psi, phi.get_pointer(), en, ethr_band, maxiter); + dav.diag(phm, ld_psi, phi.get_pointer(), en, ethr_band, maxiter); #ifdef __MPI end = MPI_Wtime(); @@ -149,7 +134,6 @@ class DiagoDavPrepare } } delete [] en; - delete phm; delete [] e_lapack; } }; diff --git a/source/source_hsolver/test/diago_mock.h b/source/source_hsolver/test/diago_mock.h index 279b5dc566e..cc4e0853e7b 100644 --- a/source/source_hsolver/test/diago_mock.h +++ b/source/source_hsolver/test/diago_mock.h @@ -1,7 +1,14 @@ #include #include "mpi.h" #include "source_base/parallel_reduce.h" -#include "source_pw/module_pwdft/hamilt_pw.h" +#include "source_hsolver/hs_operator.h" +#include "source_psi/psi.h" + +#include +#include +#include +#include +#include namespace DIAGOTEST { @@ -210,8 +217,6 @@ class HPsi //ModuleBase::ComplexMatrix psi() {return psimatrix;} psi::Psi psi() { - Structure_Factor* sf; - int* ngk = nullptr; psi::Psi psitmp(1, nband, npw, npw, true); for(int i=0;i; template class HPsi>; template class HPsi>; -//totally same as the original function -template <> -void hamilt::HamiltPW::sPsi(const double* psi_in, - double* spsi, - const int nrow, - const int npw, - const int nbands) const -{ - for (size_t i = 0; i < static_cast(nbands * nrow); i++) - { - spsi[i] = psi_in[i]; - } - return; -} -template <> -void hamilt::HamiltPW, base_device::DEVICE_CPU>::sPsi(const std::complex* psi_in, - std::complex* spsi, - const int nrow, - const int npw, - const int nbands) const +namespace DIAGOTEST { - for (size_t i = 0; i < static_cast(nbands * nrow); i++) - { - spsi[i] = psi_in[i]; - } - return; -} -template <> -void hamilt::HamiltPW, base_device::DEVICE_CPU>::sPsi(const std::complex* psi_in, - std::complex* spsi, - const int nrow, - const int npw, - const int nbands) const + /// the process-local slice of the test matrix for each scalar type + template std::vector& hmatrix_local_of(); + template <> std::vector& hmatrix_local_of() { return hmatrix_local_d; } + template <> std::vector>& hmatrix_local_of>() { return hmatrix_local; } + template <> std::vector>& hmatrix_local_of>() { return hmatrix_local_f; } +} + +/** + * The test matrix seen through hsolver::HSOperator, which is all the iterative + * eigensolvers need. H is the dense (distributed) matrix DIAGOTEST::hmatrix_local, + * S is the identity. + * + * Every process holds npw_local[mypnum] columns of H, so H*x is formed as a + * partial product, reduced over the pool and redistributed like psi. + */ +template +class HSOperatorMock : public hsolver::HSOperator { - for (size_t i = 0; i < static_cast(nbands * nrow); i++) + public: + void update_k(const int ik) override { - spsi[i] = psi_in[i]; } - return; -} -//Mock function h_psi -#include "source_pw/module_pwdft/op_pw.h" -template -class OperatorMock : public hamilt::Operator -{ - ~OperatorMock() + void hpsi(const T* x, T* hx, const int ld, const int nvec) const override { - if(this->hpsi != nullptr) - { - delete this->hpsi; - this->hpsi = nullptr; - } - } - virtual void act - ( - const int nbands, - const int nbasis, - const int npol, - const T* tmpsi_in, - T* tmhpsi, - const int ngk_ik = 0, - const bool is_first_node = false)const; -}; -template<> -void OperatorMock::act( - const int nbands, - const int nbasis, - const int npol, - const double* tmpsi_in, - double* tmhpsi, - const int ngk_ik, - const bool is_first_node)const -{ - int nprocs = 1, mypnum = 0; -#ifdef __MPI - MPI_Comm_size(MPI_COMM_WORLD, &nprocs); - MPI_Comm_rank(MPI_COMM_WORLD, &mypnum); -#endif - - double* hpsi0 = new double[DIAGOTEST::npw]; - for (int m = 0; m < nbands; m++) - { - for (int i = 0;i < DIAGOTEST::npw;i++) - { - hpsi0[i] = 0.0; - for (int j = 0;j < (DIAGOTEST::npw_local[mypnum]);j++) - { - hpsi0[i] += DIAGOTEST::hmatrix_local_d[i * DIAGOTEST::h_nc + j] * tmpsi_in[j]; - } - } - Parallel_Reduce::reduce_pool(hpsi0, DIAGOTEST::npw); - DIAGOTEST::divide_psi(hpsi0, tmhpsi); - tmhpsi += nbasis; - tmpsi_in += nbasis; - } - delete[] hpsi0; -} -template<> -void OperatorMock>::act( - const int nbands, - const int nbasis, - const int npol, - const std::complex* tmpsi_in, - std::complex* tmhpsi, - const int ngk_ik, - const bool is_first_node)const -{ - int nprocs = 1, mypnum = 0; -#ifdef __MPI - MPI_Comm_size(MPI_COMM_WORLD, &nprocs); - MPI_Comm_rank(MPI_COMM_WORLD, &mypnum); -#endif - - std::complex* hpsi0 = new std::complex[DIAGOTEST::npw]; - for (int m = 0; m < nbands; m++) - { - for (int i = 0;i < DIAGOTEST::npw;i++) + int mypnum = 0; +#ifdef __MPI + MPI_Comm_rank(MPI_COMM_WORLD, &mypnum); +#endif + const std::vector& hmat = DIAGOTEST::hmatrix_local_of(); + const int npw = DIAGOTEST::npw; + const int ncol = DIAGOTEST::npw_local[mypnum]; + std::vector hx_full(npw); + for (int m = 0; m < nvec; m++) { - hpsi0[i] = 0.0; - for (int j = 0;j < (DIAGOTEST::npw_local[mypnum]);j++) + for (int i = 0; i < npw; i++) { - hpsi0[i] += DIAGOTEST::hmatrix_local[i * DIAGOTEST::h_nc + j] * tmpsi_in[j]; + hx_full[i] = T(0); + for (int j = 0; j < ncol; j++) + { + hx_full[i] += hmat[i * DIAGOTEST::h_nc + j] * x[m * ld + j]; + } } + Parallel_Reduce::reduce_pool(hx_full.data(), npw); + DIAGOTEST::divide_psi(hx_full.data(), hx + m * ld); } - Parallel_Reduce::reduce_pool(hpsi0, DIAGOTEST::npw); - DIAGOTEST::divide_psi>(hpsi0, tmhpsi); - tmhpsi += nbasis; - tmpsi_in += nbasis; } - delete[] hpsi0; -} -template<> -void OperatorMock>::act( - const int nbands, - const int nbasis, - const int npol, - const std::complex* tmpsi_in, - std::complex* tmhpsi, - const int ngk_ik, - const bool is_first_node)const -{ - int nprocs = 1, mypnum = 0; -#ifdef __MPI - MPI_Comm_size(MPI_COMM_WORLD, &nprocs); - MPI_Comm_rank(MPI_COMM_WORLD, &mypnum); -#endif - std::complex* hpsi0 = new std::complex[DIAGOTEST::npw]; - for (int m = 0; m < nbands; m++) + void spsi(const T* x, T* sx, const int ld, const int nvec) const override { - for (int i = 0;i < DIAGOTEST::npw;i++) - { - hpsi0[i] = 0.0; - for (int j = 0;j < (DIAGOTEST::npw_local[mypnum]);j++) - { - hpsi0[i] += DIAGOTEST::hmatrix_local_f[i * DIAGOTEST::h_nc + j] * tmpsi_in[j]; - } - } - Parallel_Reduce::reduce_pool(hpsi0, DIAGOTEST::npw); - DIAGOTEST::divide_psi>(hpsi0, tmhpsi); - tmhpsi += nbasis; - tmpsi_in += nbasis; + std::copy(x, x + static_cast(ld) * nvec, sx); } - delete[] hpsi0; -} -template<> void hamilt::HamiltPW::updateHk(const int ik) -{ - return; -} - -template<> hamilt::HamiltPW::HamiltPW( - elecstate::Potential* pot_in, - ModulePW::PW_Basis_K* wfc_basis, - K_Vectors* pkv, - pseudopot_cell_vnl* ppcell, - Plus_U_Base* p_dftu, - const UnitCell* ucell, - const General_Exx_Info* exx_info) -{ - this->ops = new OperatorMock; -} - -template<> hamilt::HamiltPW::~HamiltPW() -{ - delete this->ops; -} - -template<> void hamilt::HamiltPW>::updateHk(const int ik) -{ - return; -} - -template<> hamilt::HamiltPW>::HamiltPW( - elecstate::Potential* pot_in, - ModulePW::PW_Basis_K* wfc_basis, - K_Vectors* pkv, - pseudopot_cell_vnl* ppcell, - Plus_U_Base* p_dftu, - const UnitCell* ucell, - const General_Exx_Info* exx_info) -{ - this->ops = new OperatorMock>; -} - -template<> hamilt::HamiltPW>::~HamiltPW() -{ - delete this->ops; -} - -template<> void hamilt::HamiltPW>::updateHk(const int ik) -{ - return; -} - -template<> hamilt::HamiltPW>::HamiltPW( - elecstate::Potential* pot_in, - ModulePW::PW_Basis_K* wfc_basis, - K_Vectors* pkv, - pseudopot_cell_vnl* ppcell, - Plus_U_Base* p_dftu, - const UnitCell* ucell, - const General_Exx_Info* exx_info) -{ - this->ops = new OperatorMock>; -} - -template<> hamilt::HamiltPW>::~HamiltPW() -{ - delete this->ops; -} +}; diff --git a/source/source_hsolver/test/hsolver_pw_sup.h b/source/source_hsolver/test/hsolver_pw_sup.h index b41196c396d..75ec3a90500 100644 --- a/source/source_hsolver/test/hsolver_pw_sup.h +++ b/source/source_hsolver/test/hsolver_pw_sup.h @@ -49,6 +49,7 @@ double& PW_Basis_K::getgk2(const int ik, const int igl) const { #include "source_hsolver/diago_cg.h" #include "source_hsolver/diago_david.h" +#include "source_hsolver/diag_comm_info.h" #include "source_hsolver/diago_iter_assist.h" @@ -56,7 +57,13 @@ namespace hsolver { template DiagoCG::DiagoCG(const std::string& basis_type, - const std::string& calculation) { + const std::string& calculation) +#ifdef __MPI + : diag_comm_(MPI_COMM_SELF, 0, 1) +#else + : diag_comm_(0, 1) +#endif +{ basis_type_ = basis_type; calculation_ = calculation; this->one_ = new T(static_cast(1.0)); @@ -68,17 +75,15 @@ template DiagoCG::DiagoCG(const std::string& basis_type, const std::string& calculation, const bool& need_subspace, - const SubspaceFunc& subspace_func, + const diag_comm_info& diag_comm, const Real& pw_diag_thr, - const int& pw_diag_nmax, - const int& nproc_in_pool) { + const int& pw_diag_nmax) + : diag_comm_(diag_comm) { basis_type_ = basis_type; calculation_ = calculation; need_subspace_ = need_subspace; - subspace_func_ = subspace_func; pw_diag_thr_ = pw_diag_thr; pw_diag_nmax_ = pw_diag_nmax; - nproc_in_pool_ = nproc_in_pool; this->one_ = new T(static_cast(1.0)); this->zero_ = new T(static_cast(0.0)); this->neg_one_ = new T(static_cast(-1.0)); @@ -92,8 +97,7 @@ DiagoCG::~DiagoCG() { } template -double DiagoCG::diag(const HPsiFunc& hpsi_func, - const SPsiFunc& spsi_func, +double DiagoCG::diag(const HSOperator& op, const int ld_psi, const int nband, const int dim, @@ -146,8 +150,7 @@ DiagoDavid::~DiagoDavid() { } template -int DiagoDavid::diag(const std::function& hpsi_func, - const std::function& spsi_func, +int DiagoDavid::diag(const HSOperator& op, const int ld_psi, T* psi_in, Real* eigenvalue_in, diff --git a/source/source_hsolver/test/hsolver_supplementary_mock.h b/source/source_hsolver/test/hsolver_supplementary_mock.h index b69bdeff3ae..76da6b23eba 100644 --- a/source/source_hsolver/test/hsolver_supplementary_mock.h +++ b/source/source_hsolver/test/hsolver_supplementary_mock.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_HSOLVER_TEST_HSOLVER_SUPPLEMENTARY_MOCK_H +#define ABACUS_SOURCE_HSOLVER_TEST_HSOLVER_SUPPLEMENTARY_MOCK_H + #include "source_estate/elecstate_pw.h" namespace elecstate @@ -86,34 +88,6 @@ void Potential::cal_fixed_v(double* vl_pseudo) } // namespace elecstate -// mock of Stochastic_WF -#include "source_pw/module_stodft/sto_wf.h" -template -Stochastic_WF::Stochastic_WF() -{ - chiortho = nullptr; - chi0 = nullptr; - shchi = nullptr; - nchip = nullptr; -} - -template -Stochastic_WF::~Stochastic_WF() -{ - delete[] chi0; - delete[] shchi; - delete[] chiortho; - delete[] nchip; -} - -template -void Stochastic_WF::init(K_Vectors* p_kv, const int npwx_in) -{ - /*chi0 = new ModuleBase::ComplexMatrix[nks_in]; - shchi = new ModuleBase::ComplexMatrix[nks_in]; - chiortho = new ModuleBase::ComplexMatrix[nks_in]; - nchip = new int[nks_in]; - this->nks = nks_in;*/ -} - #include "source_cell/klist.h" + +#endif // ABACUS_SOURCE_HSOLVER_TEST_HSOLVER_SUPPLEMENTARY_MOCK_H diff --git a/source/source_hsolver/test/linear_test_utils.h b/source/source_hsolver/test/linear_test_utils.h new file mode 100644 index 00000000000..b3c49647677 --- /dev/null +++ b/source/source_hsolver/test/linear_test_utils.h @@ -0,0 +1,97 @@ +#ifndef HSOLVER_LINEAR_TEST_UTILS_H +#define HSOLVER_LINEAR_TEST_UTILS_H + +#include "source_base/module_container/base/third_party/lapack.h" +#include "source_base/parallel_comm.h" +#include "source_hsolver/diag_comm_info.h" + +#include +#include +#include + +namespace linear_test +{ +using Complex = std::complex; + +inline hsolver::diag_comm_info world_comm() +{ +#ifdef __MPI + MPICommGroup group(MPI_COMM_WORLD); + return hsolver::diag_comm_info(MPI_COMM_WORLD, group.grank, group.gsize); +#else + return hsolver::diag_comm_info(0, 1); +#endif +} + +// Column-major matrices with a decoupled first eigenvector and a dense complex block. +template +std::vector hermitian(const int n) +{ + std::vector h(n * n, T(0)); + for (int j = 0; j < n; ++j) + { + h[j * n + j] = T(1.0 + 0.4 * j); + for (int i = 1; i < j; ++i) + { + const T value(0.12 / (j - i + 1), 0.07 / (i + j + 1)); + h[j * n + i] = value; + h[i * n + j] = std::conj(value); + } + } + return h; +} + +template +std::vector multiply(const std::vector& a, const std::vector& x, const int n, const int nvec) +{ + std::vector b(n * nvec, T(0)); + for (int band = 0; band < nvec; ++band) + { + for (int j = 0; j < n; ++j) + { + for (int i = 0; i < n; ++i) + { + b[band * n + i] += a[j * n + i] * x[band * n + j]; + } + } + } + return b; +} + +// Extract the local rows of column-major reference data, leaving padding zero. +template +std::vector local_columns(const std::vector& global, int n, int columns, int start, int dim, int ld) +{ + std::vector local(ld * columns, T(0)); + for (int j = 0; j < columns; ++j) + { + for (int i = 0; i < dim; ++i) + { + local[j * ld + i] = global[j * n + start + i]; + } + } + return local; +} + +// Solve in double precision even when the iterative input was rounded to float. +template +std::vector lapack_solve(const std::vector& a, const std::vector& b, const int n, const int nvec) +{ + std::vector lu(a.begin(), a.end()); + std::vector x(b.begin(), b.end()); + std::vector pivots(n); + int info = 0; + container::lapackConnector::getrf(n, n, lu.data(), n, pivots.data(), info); + if (info != 0) + { + throw std::runtime_error("LAPACK reference factorization failed"); + } + container::lapackConnector::getrs('N', n, nvec, lu.data(), n, pivots.data(), x.data(), n, info); + if (info != 0) + { + throw std::runtime_error("LAPACK reference solve failed"); + } + return x; +} +} // namespace linear_test +#endif diff --git a/source/source_hsolver/test/parallel_k2d_test.cpp b/source/source_hsolver/test/parallel_k2d_test.cpp index 8abceac5d62..fa5f6479583 100644 --- a/source/source_hsolver/test/parallel_k2d_test.cpp +++ b/source/source_hsolver/test/parallel_k2d_test.cpp @@ -44,7 +44,7 @@ class ParaPrepare { class ParallelK2DTest : public ::testing::TestWithParam { protected: - Parallel_K2D k2d = Parallel_K2D(); + Parallel_K2D k2d; MPIContext mpi; int NPROC; int MY_RANK; diff --git a/source/source_hsolver/test/test_cn_subspace.cpp b/source/source_hsolver/test/test_cn_subspace.cpp new file mode 100644 index 00000000000..16fee4d1e10 --- /dev/null +++ b/source/source_hsolver/test/test_cn_subspace.cpp @@ -0,0 +1,120 @@ +#include "source_base/parallel_device.h" +#include "source_hsolver/cn_subspace.h" +#include "source_hsolver/test/linear_test_utils.h" + +#include +#include +#include + +namespace +{ +template +class CNSubspaceTest : public testing::Test +{ +}; +using Precisions = testing::Types; +TYPED_TEST_SUITE(CNSubspaceTest, Precisions); + +TYPED_TEST(CNSubspaceTest, ProjectedSeedAndResidualOrRankDeficientFallback) +{ + using T = std::complex; + using Wide = linear_test::Complex; + const hsolver::diag_comm_info comm = linear_test::world_comm(); + const int n = 5; + const int bands = 2; + const int start = n * comm.rank / comm.nproc; + const int dim = n * (comm.rank + 1) / comm.nproc - start; + const int ld = dim + 2; + const double tolerance = std::is_same::value ? 1e-5 : 1e-12; + const std::vector h = linear_test::hermitian(n); + std::vector a(n * n); + std::vector right(n * n); + for (int j = 0; j < n; ++j) + { + for (int i = 0; i < n; ++i) + { + const T identity(i == j ? 1 : 0); + a[j * n + i] = identity + T(0, 0.15) * h[j * n + i]; + right[j * n + i] = identity - T(0, 0.15) * h[j * n + i]; + } + } + hsolver::LinearAlgebra algebra(comm); + hsolver::CNSubspace projection; + for (const bool singular: {false, true}) + { + SCOPED_TRACE(singular); + std::vector u(n * bands); + for (int j = 0; j < bands; ++j) + { + for (int i = 0; i < n; ++i) + { + const double phase = singular ? 0.0 : 2.0 * std::acos(-1.0) * i * j / n; + u[j * n + i] = std::polar(TypeParam(1.0 / std::sqrt(n)), TypeParam(phase)); + } + } + const std::vector b = linear_test::multiply(right, u, n, bands); + const std::vector au = linear_test::multiply(a, u, n, bands); + const std::vector local_u = linear_test::local_columns(u, n, bands, start, dim, ld); + const std::vector local_b = linear_test::local_columns(b, n, bands, start, dim, ld); + const bool prepared = projection.prepare(algebra, ld, dim, bands, local_u.data(), local_b.data()); + if (singular) + { + EXPECT_FALSE(prepared); + continue; + } + ASSERT_TRUE(prepared); + std::vector projected(bands * bands, Wide(0)); + std::vector projected_rhs(bands * bands, Wide(0)); + for (int j = 0; j < bands; ++j) + { + for (int i = 0; i < bands; ++i) + { + for (int row = 0; row < n; ++row) + { + projected[j * bands + i] += std::conj(Wide(u[i * n + row])) * Wide(au[j * n + row]); + projected_rhs[j * bands + i] += std::conj(Wide(u[i * n + row])) * Wide(b[j * n + row]); + } + } + } + const std::vector coefficients = linear_test::lapack_solve(projected, projected_rhs, bands, bands); + std::vector actual(n * bands, Wide(0)); + for (int j = 0; j < bands; ++j) + { + for (int i = 0; i < dim; ++i) + { + Wide expected(0); + for (int k = 0; k < bands; ++k) + { + expected += Wide(u[k * n + start + i]) * coefficients[j * bands + k]; + } + EXPECT_LT(std::abs(Wide(projection.seed()[j * ld + i]) - expected), tolerance); + EXPECT_LT(std::abs(Wide(projection.image()[j * ld + i]) - Wide(au[j * n + start + i])), tolerance); + actual[j * n + start + i] = projection.seed()[j * ld + i]; + } + } +#ifdef __MPI + const int count = actual.size(); + Parallel_Common::reduce_data(actual.data(), count, comm.comm); +#endif + const std::vector wide_a(a.begin(), a.end()); + const std::vector ax = linear_test::multiply(wide_a, actual, n, bands); + for (int j = 0; j < bands; ++j) + { + for (int i = 0; i < dim; ++i) + { + const Wide residual = Wide(b[j * n + start + i]) - ax[j * n + start + i]; + EXPECT_LT(std::abs(Wide(projection.residual()[j * ld + i]) - residual), tolerance); + } + for (int k = 0; k < bands; ++k) + { + Wide overlap(0); + for (int i = 0; i < n; ++i) + { + overlap += std::conj(Wide(u[k * n + i])) * (Wide(b[j * n + i]) - ax[j * n + i]); + } + EXPECT_LT(std::abs(overlap), tolerance); + } + } + } +} +} // namespace diff --git a/source/source_hsolver/test/test_hsolver.cpp b/source/source_hsolver/test/test_hsolver.cpp index dc46f317b19..97629b3a002 100644 --- a/source/source_hsolver/test/test_hsolver.cpp +++ b/source/source_hsolver/test/test_hsolver.cpp @@ -1,85 +1,177 @@ #include -#include -#include -#include "hsolver_supplementary_mock.h" -#include "source_hamilt/hamilt.h" -#include "source_hsolver/hsolver.h" +#include +#include +#include +#include +#include "source_hsolver/hsolver.h" -// template class hsolver::HSolver, base_device::DEVICE_CPU>; -// template class hsolver::HSolver, base_device::DEVICE_CPU>; -#include /************************************************ - * unit test of HSolver base class + * unit test of the free functions in hsolver.h ***********************************************/ /** - * Tested function: - * - hsolver::HSolver::solve (for cases below) - * - float/double; - * - Psi/complex>; - * - skip charge or not; - * - stodft case; - * - hsolver::HSolver::diagethr (for cases below) - * - set_diagethr, for setting diagethr; - * - hsolver::DiagH (for cases below) - * - diag() for Psi(FPTYPE) case - * - destructor of DiagH and HSolver - * - * the definition of supplementary functions is added in hsolver_supplementary_mock.h + * Tested functions: + * - hsolver::cal_hsolve_error + * - pw + ksdft: diag_ethr * max(1, nelec) + * - any other basis/esolver: 0 + * - hsolver::set_diagethr_ks + * - non pw-ksdft returns 0 + * - nscf, first scf iteration (by init_chg and calculation), later iterations + * - single precision lower bound + * - hsolver::set_diagethr_sdft + * - non pw-sdft returns 0 + * - nscf, first iteration of the first/later ion step, later iterations + * - hsolver::reset_diag_ethr + * - pw + ksdft: 0.1 * drho / nelec, with the single precision lower bound + * - any other basis/esolver: 0 */ -class TestHSolver : public ::testing::Test +TEST(HSolverCalHsolveErrorTest, PwKsdftScalesWithNelec) +{ + EXPECT_DOUBLE_EQ(hsolver::cal_hsolve_error("pw", "ksdft", 1.0e-3, 8.0), 8.0e-3); + // nelec below 1 is clamped to 1 + EXPECT_DOUBLE_EQ(hsolver::cal_hsolve_error("pw", "ksdft", 1.0e-3, 0.5), 1.0e-3); +} + +TEST(HSolverCalHsolveErrorTest, OtherCasesAreZero) +{ + EXPECT_EQ(hsolver::cal_hsolve_error("lcao", "ksdft", 1.0e-3, 8.0), 0.0); + EXPECT_EQ(hsolver::cal_hsolve_error("lcao_in_pw", "ksdft", 1.0e-3, 8.0), 0.0); + EXPECT_EQ(hsolver::cal_hsolve_error("pw", "sdft", 1.0e-3, 8.0), 0.0); +} + +TEST(HSolverSetDiagethrKsTest, NonPwKsdftIsZero) +{ + EXPECT_EQ( + hsolver::set_diagethr_ks("lcao", "ksdft", "scf", "atomic", "double", 0, 1, 1.0e-3, 1.0e-2, 1.0e-2, 8.0, 1.0e-6), + 0.0); + EXPECT_EQ( + hsolver::set_diagethr_ks("pw", "sdft", "scf", "atomic", "double", 0, 1, 1.0e-3, 1.0e-2, 1.0e-2, 8.0, 1.0e-6), + 0.0); +} + +TEST(HSolverSetDiagethrKsTest, Nscf) +{ + // the default 1e-2 is tightened to 0.1 * min(1e-2, scf_thr / nelec) + EXPECT_DOUBLE_EQ( + hsolver::set_diagethr_ks("pw", "ksdft", "nscf", "atomic", "double", 0, 1, 1.0e-3, 1.0e-2, 1.0e-2, 8.0, 1.0e-6), + 1.25e-8); + // a threshold the user already tightened is kept + EXPECT_DOUBLE_EQ( + hsolver::set_diagethr_ks("pw", "ksdft", "nscf", "atomic", "double", 0, 1, 1.0e-3, 1.0e-2, 1.0e-6, 8.0, 1.0e-6), + 1.0e-6); +} + +TEST(HSolverSetDiagethrKsTest, FirstIteration) +{ + // atomic starting charge keeps the loose 1e-2 + EXPECT_DOUBLE_EQ( + hsolver::set_diagethr_ks("pw", "ksdft", "scf", "atomic", "double", 0, 1, 1.0e-3, 1.0e-2, 1.0e-2, 8.0, 1.0e-6), + 1.0e-2); + // a charge read from file is trusted, so the first diagonalization is strict + EXPECT_DOUBLE_EQ( + hsolver::set_diagethr_ks("pw", "ksdft", "scf", "file", "double", 0, 1, 1.0e-3, 1.0e-2, 1.0e-2, 8.0, 1.0e-6), + 1.0e-5); + EXPECT_DOUBLE_EQ( + hsolver::set_diagethr_ks("pw", "ksdft", "scf", "wfc", "double", 0, 1, 1.0e-3, 1.0e-2, 1.0e-2, 8.0, 1.0e-6), + 1.0e-5); + // relax/md never go below pw_diag_thr_init + EXPECT_DOUBLE_EQ( + hsolver::set_diagethr_ks("pw", "ksdft", "relax", "file", "double", 0, 1, 1.0e-3, 1.0e-3, 1.0e-2, 8.0, 1.0e-6), + 1.0e-3); +} + +TEST(HSolverSetDiagethrKsTest, LaterIterations) +{ + // iteration 2 restarts from 1e-2 and is bounded by 0.1 * drho / nelec + EXPECT_DOUBLE_EQ( + hsolver::set_diagethr_ks("pw", "ksdft", "scf", "atomic", "double", 0, 2, 1.0e-3, 1.0e-2, 1.0e-7, 8.0, 1.0e-6), + 1.25e-5); + // later iterations only ever tighten the threshold + EXPECT_DOUBLE_EQ( + hsolver::set_diagethr_ks("pw", "ksdft", "scf", "atomic", "double", 0, 3, 1.0e-3, 1.0e-2, 1.0e-6, 8.0, 1.0e-6), + 1.0e-6); + EXPECT_DOUBLE_EQ( + hsolver::set_diagethr_ks("pw", "ksdft", "scf", "atomic", "double", 0, 3, 1.0e-3, 1.0e-2, 1.0e-4, 8.0, 1.0e-6), + 1.25e-5); +} + +TEST(HSolverSetDiagethrKsTest, SinglePrecisionLowerBound) +{ + EXPECT_DOUBLE_EQ( + hsolver::set_diagethr_ks("pw", "ksdft", "scf", "atomic", "single", 0, 3, 1.0e-3, 1.0e-2, 1.0e-6, 8.0, 1.0e-6), + 0.5e-4); +} + +TEST(HSolverSetDiagethrSdftTest, NonPwSdftIsZero) +{ + EXPECT_EQ( + hsolver::set_diagethr_sdft("pw", "ksdft", "scf", "atomic", 0, 1, 1.0e-3, 1.0e-2, 1.0e-2, 4, 2.0, 8.0, 1.0e-6), + 0.0); + EXPECT_EQ( + hsolver::set_diagethr_sdft("lcao", "sdft", "scf", "atomic", 0, 1, 1.0e-3, 1.0e-2, 1.0e-2, 4, 2.0, 8.0, 1.0e-6), + 0.0); +} + +TEST(HSolverSetDiagethrSdftTest, Nscf) +{ + EXPECT_DOUBLE_EQ( + hsolver::set_diagethr_sdft("pw", "sdft", "nscf", "atomic", 0, 1, 1.0e-3, 1.0e-2, 1.0e-2, 4, 2.0, 8.0, 1.0e-6), + 1.25e-8); +} + +TEST(HSolverSetDiagethrSdftTest, FirstIteration) +{ + // first ion step: file charge gives 1e-5, then bounded below by pw_diag_thr_init + EXPECT_DOUBLE_EQ( + hsolver::set_diagethr_sdft("pw", "sdft", "scf", "file", 0, 1, 1.0e-3, 1.0e-2, 1.0e-4, 4, 2.0, 8.0, 1.0e-6), + 1.0e-2); + EXPECT_DOUBLE_EQ( + hsolver::set_diagethr_sdft("pw", "sdft", "scf", "atomic", 0, 1, 1.0e-3, 1.0e-6, 1.0e-4, 4, 2.0, 8.0, 1.0e-6), + 1.0e-4); + // later ion steps are bounded below by 1e-5 + EXPECT_DOUBLE_EQ( + hsolver::set_diagethr_sdft("pw", "sdft", "scf", "atomic", 1, 1, 1.0e-3, 1.0e-2, 1.0e-7, 4, 2.0, 8.0, 1.0e-6), + 1.0e-5); +} + +TEST(HSolverSetDiagethrSdftTest, LaterIterations) +{ + // bounded by 0.1 * drho / KS electrons + EXPECT_DOUBLE_EQ( + hsolver::set_diagethr_sdft("pw", "sdft", "scf", "atomic", 0, 2, 1.0e-3, 1.0e-2, 1.0e-2, 4, 2.0, 8.0, 1.0e-6), + 5.0e-5); + // pure stochastic (no KS bands) does not diagonalize + EXPECT_EQ( + hsolver::set_diagethr_sdft("pw", "sdft", "scf", "atomic", 0, 2, 1.0e-3, 1.0e-2, 1.0e-2, 0, 2.0, 8.0, 1.0e-6), + 0.0); +} + +TEST(HSolverResetDiagEthrTest, PwKsdft) +{ + const std::string log_name = "test_hsolver_reset_diag_ethr.log"; + std::ofstream ofs(log_name.c_str()); + const double new_ethr = hsolver::reset_diag_ethr(ofs, "pw", "ksdft", "double", 1.0e-2, 1.0e-3, 1.0e-2, 8.0); + const double new_ethr_single = hsolver::reset_diag_ethr(ofs, "pw", "ksdft", "single", 1.0e-2, 1.0e-3, 1.0e-2, 8.0); + ofs.close(); + + EXPECT_DOUBLE_EQ(new_ethr, 1.25e-5); + EXPECT_DOUBLE_EQ(new_ethr_single, 0.5e-4); + + std::ifstream ifs(log_name.c_str()); + std::stringstream buffer; + buffer << ifs.rdbuf(); + ifs.close(); + EXPECT_NE(buffer.str().find("Threshold on eigenvalues was too large"), std::string::npos); + EXPECT_NE(buffer.str().find("New diag ethr"), std::string::npos); + std::remove(log_name.c_str()); +} + +TEST(HSolverResetDiagEthrTest, OtherCasesAreZero) { - public: - // hsolver::HSolver, base_device::DEVICE_CPU> hs_cf; - // hsolver::HSolver, base_device::DEVICE_CPU> hs_cd; - // hsolver::HSolver hs_f; - // hsolver::HSolver hs_d; - - hamilt::Hamilt> hamilt_test_cd; - hamilt::Hamilt> hamilt_test_cf; - psi::Psi> psi_test_cd; - psi::Psi> psi_test_cf; - - hamilt::Hamilt hamilt_test_d; - hamilt::Hamilt hamilt_test_f; - psi::Psi psi_test_d; - psi::Psi psi_test_f; - - Stochastic_WF> stowf_test; - - elecstate::ElecState elecstate_test; - - ModulePW::PW_Basis_K* wfcpw; - - std::string method_test = "none"; - - std::ofstream temp_ofs; -}; - -// TEST_F(TestHSolver, solve) -// { -// hs_cf.solve(&hamilt_test_cf, psi_test_cf, &elecstate_test, method_test, false); -// hs_f.solve(&hamilt_test_f, psi_test_f, &elecstate_test, method_test, false); -// hs_cd.solve(&hamilt_test_cd, psi_test_cd, &elecstate_test, method_test, false); -// hs_d.solve(&hamilt_test_d, psi_test_d, &elecstate_test, method_test, false); -// hs_cf.solve(&hamilt_test_cf, psi_test_cf, &elecstate_test, method_test, true); -// hs_f.solve(&hamilt_test_f, psi_test_f, &elecstate_test, method_test, true); -// hs_cd.solve(&hamilt_test_cd, psi_test_cd, &elecstate_test, method_test, true); -// hs_d.solve(&hamilt_test_d, psi_test_d, &elecstate_test, method_test, true); -// // hs_cf.solve(&hamilt_test_cf, psi_test_cf, &elecstate_test, wfcpw, stowf_test, 0, 0, method_test, true); -// // hs_cd.solve(&hamilt_test_cd, psi_test_cd, &elecstate_test, wfcpw, stowf_test, 0, 0, method_test, true); -// EXPECT_EQ(hs_f.method, "none"); -// EXPECT_EQ(hs_d.method, "none"); -// } - -// TEST_F(TestHSolver, diagethr) -// { -// float test_diagethr = hs_f.set_diagethr(0.0, 0, 0, 0.0); -// EXPECT_EQ(test_diagethr, 0.0); - -// double test_diagethr_d = hs_d.set_diagethr(0.0, 0, 0, 0.0); -// EXPECT_EQ(test_diagethr_d, 0.0); -// } \ No newline at end of file + std::ofstream ofs; + EXPECT_EQ(hsolver::reset_diag_ethr(ofs, "lcao", "ksdft", "double", 1.0e-2, 1.0e-3, 1.0e-2, 8.0), 0.0); + EXPECT_EQ(hsolver::reset_diag_ethr(ofs, "pw", "sdft", "double", 1.0e-2, 1.0e-3, 1.0e-2, 8.0), 0.0); +} diff --git a/source/source_hsolver/test/test_hsolver_linear.cpp b/source/source_hsolver/test/test_hsolver_linear.cpp new file mode 100644 index 00000000000..14efc6a96d8 --- /dev/null +++ b/source/source_hsolver/test/test_hsolver_linear.cpp @@ -0,0 +1,426 @@ +#include "source_base/parallel_device.h" +#include "source_base/parallel_global.h" +#include "source_hsolver/hsolver_linear.h" +#include "source_hsolver/test/linear_test_utils.h" + +#include +#include +#include +#include + +namespace +{ +template +class IdentityOperator final : public hsolver::LinearOperator +{ + private: + const int dim_; + + public: + explicit IdentityOperator(const int dim) : dim_(dim) + { + } + bool is_identity() const override + { + return true; + } + void apply(const T* x, T* y, const int ld, const int nvec) const override + { + for (int band = 0; band < nvec; ++band) + { + for (int i = 0; i < dim_; ++i) + { + y[band * ld + i] = x[band * ld + i]; + } + } + } +}; + +template +class DenseOperator final : public hsolver::LinearOperator +{ + public: + DenseOperator(const std::vector& matrix, const int n, const int start, const int dim, const hsolver::diag_comm_info& comm) + : matrix_(matrix), n_(n), start_(start), dim_(dim), comm_(comm) + { + } + + void apply(const T* x, T* y, const int ld, const int nvec) const override + { + std::vector full(n_ * nvec, T(0)); + for (int band = 0; band < nvec; ++band) + { + for (int i = 0; i < dim_; ++i) + { + full[band * n_ + start_ + i] = x[band * ld + i]; + } + } +#ifdef __MPI + const int count = full.size(); + Parallel_Common::reduce_data(full.data(), count, comm_.comm); +#endif + const std::vector product = linear_test::multiply(matrix_, full, n_, nvec); + for (int band = 0; band < nvec; ++band) + { + for (int i = 0; i < dim_; ++i) + { + y[band * ld + i] = product[band * n_ + start_ + i]; + } + } + } + + private: + const std::vector& matrix_; + const int n_; + const int start_; + const int dim_; + const hsolver::diag_comm_info comm_; +}; + +template +class RightPreconditioner final : public hsolver::LinearOperator +{ + public: + RightPreconditioner(const std::vector& matrix, const int n, const int start, const int dim) + : matrix_(matrix), n_(n), start_(start), dim_(dim) + { + } + void apply(const T* x, T* y, const int ld, const int nvec) const override + { + for (int band = 0; band < nvec; ++band) + { + for (int i = 0; i < dim_; ++i) + { + const int row = start_ + i; + y[band * ld + i] = x[band * ld + i] / matrix_[row * n_ + row]; + } + } + } + + private: + const std::vector& matrix_; + const int n_; + const int start_; + const int dim_; +}; + +template +class LinearSolveTest : public testing::Test +{ +}; +using Precisions = testing::Types; +TYPED_TEST_SUITE(LinearSolveTest, Precisions); + +TYPED_TEST(LinearSolveTest, DenseReferenceAndReusedBatches) +{ + using T = std::complex; + const hsolver::diag_comm_info comm = linear_test::world_comm(); + const double tolerance = std::is_same::value ? 1e-12 : 2e-6; + const double solution_tol = std::is_same::value ? 1e-10 : 1e-4; + for (const hsolver::LinearMethod method: {hsolver::LinearMethod::bicgstab, hsolver::LinearMethod::cgs, hsolver::LinearMethod::gmres}) + { + for (const bool preconditioned: {false, true}) + { + hsolver::LinearSolveOptions options; + options.method = method; + options.tolerance = tolerance; + options.restart = 2; + hsolver::HSolverLinear solver(options, comm); + // Shrink, grow, and use zero local rows without replacing the solver. + for (const int n: {11, 5, 17, 1}) + { + const int start = n * comm.rank / comm.nproc; + const int end = n * (comm.rank + 1) / comm.nproc; + const int dim = end - start; + const int ld = dim + 3; + const int nvec = n == 5 ? 3 : 5; + for (const bool cn_matrix: {true, false}) + { + SCOPED_TRACE(testing::Message() << "n=" << n << " method=" << static_cast(method) + << " preconditioned=" << preconditioned << " cn=" << cn_matrix); + std::vector a = linear_test::hermitian(n); + for (int j = 0; j < n; ++j) + { + for (int i = 0; i < n; ++i) + { + if (cn_matrix) + { + a[j * n + i] *= T(0, 0.35); + if (i == j) + { + a[j * n + i] += T(1); + } + } + else if (i > j) + { + a[j * n + i] *= T(0.3, 0.6); + } + } + } + std::vector exact(n * nvec, T(0)); + for (int band = 0; band < nvec; ++band) + { + for (int i = 0; i < n; ++i) + { + exact[band * n + i] = T(0.2 + 0.03 * i, 0.1 * (band + 1) / (i + 1)); + } + } + // Middle columns retire before the first: exercise swaps and restoration. + std::fill(exact.begin() + n, exact.begin() + 2 * n, T(0)); + std::fill(exact.begin() + 2 * n, exact.begin() + 3 * n, T(0)); + exact[2 * n] = T(0.7, -0.2); + const std::vector b = linear_test::multiply(a, exact, n, nvec); + const std::vector reference = linear_test::lapack_solve(a, b, n, nvec); + const T sentinel(123, -45); + std::vector rhs(ld * nvec, sentinel); + std::vector x(ld * nvec, sentinel); + for (int band = 0; band < nvec; ++band) + { + for (int i = 0; i < dim; ++i) + { + rhs[band * ld + i] = b[band * n + start + i]; + x[band * ld + i] = band == 3 ? exact[band * n + start + i] : T(0); + } + } + const std::vector rhs_before = rhs; + DenseOperator op(a, n, start, dim, comm); + RightPreconditioner precond(a, n, start, dim); + hsolver::LinearSolveResult result; + if (preconditioned) + { + result = solver.solve(op, precond, ld, nvec, dim, x.data(), rhs.data(), nullptr, true); + } + else + { + const IdentityOperator identity(dim); + result = solver.solve(op, identity, ld, nvec, dim, x.data(), rhs.data(), nullptr, true); + } + EXPECT_EQ(result.status, hsolver::LinearSolveStatus::converged); + EXPECT_EQ(result.failed_band, -1); + if (method == hsolver::LinearMethod::gmres && !preconditioned && n > 1) + { + EXPECT_GT(result.restarts, 0); + } + EXPECT_EQ(rhs, rhs_before); + std::vector full(n * nvec, 0.0); + for (int band = 0; band < nvec; ++band) + { + for (int i = 0; i < dim; ++i) + { + full[band * n + start + i] = x[band * ld + i]; + EXPECT_LT(std::abs(full[band * n + start + i] - reference[band * n + start + i]), solution_tol); + } + for (int i = dim; i < ld; ++i) + { + EXPECT_EQ(x[band * ld + i], sentinel); + } + } +#ifdef __MPI + const int count = full.size(); + Parallel_Common::reduce_data(full.data(), count, comm.comm); +#endif + const std::vector a_double(a.begin(), a.end()); + const std::vector ax = linear_test::multiply(a_double, full, n, nvec); + double max_residual = 0.0; + for (int band = 0; band < nvec; ++band) + { + double residual2 = 0.0; + double bnorm2 = 0.0; + for (int i = 0; i < n; ++i) + { + const linear_test::Complex value = b[band * n + i]; + residual2 += std::norm(ax[band * n + i] - value); + bnorm2 += std::norm(value); + } + double scale = std::sqrt(bnorm2); + if (method != hsolver::LinearMethod::cgs) + { + scale = std::max(1.0, scale); + } + else if (scale == 0.0) + { + scale = 1.0; + } + const double residual = std::sqrt(residual2); + // Allow rounding in the independent double-precision residual of a float solve. + EXPECT_LE(residual, 1.2 * tolerance * scale); + max_residual = std::max(max_residual, residual); + } + EXPECT_NEAR(result.max_residual, max_residual, tolerance); + } + } + } + } +} + +TEST(LinearSolveFailure, ReportsOriginalColumnAndTrueResidual) +{ + using T = linear_test::Complex; + const hsolver::diag_comm_info comm = linear_test::world_comm(); + const int n = 7; + const int start = n * comm.rank / comm.nproc; + const int dim = n * (comm.rank + 1) / comm.nproc - start; + const int ld = dim + 1; + for (const hsolver::LinearMethod method: {hsolver::LinearMethod::bicgstab, hsolver::LinearMethod::cgs, hsolver::LinearMethod::gmres}) + { + for (const bool singular: {false, true}) + { + std::vector a = linear_test::hermitian(n); + if (singular) + { + std::fill(a.begin(), a.end(), T(0)); + } + DenseOperator op(a, n, start, dim, comm); + hsolver::LinearSolveOptions options; + options.method = method; + options.tolerance = 1e-13; + options.max_iterations = singular ? 10 : 1; + hsolver::HSolverLinear solver(options, comm); + std::vector x(3 * ld, T(0)); + std::vector b(3 * ld, T(0)); + // Retire column 0; column 2 is swapped forward but failure must still name column 1. + for (int i = 0; i < dim; ++i) + { + b[ld + i] = T(1, 0.1 * (start + i)); + } + const IdentityOperator identity(dim); + const hsolver::LinearSolveResult result = solver.solve(op, identity, ld, 3, dim, x.data(), b.data(), nullptr, true); + const hsolver::LinearSolveStatus expected + = singular ? hsolver::LinearSolveStatus::breakdown : hsolver::LinearSolveStatus::max_iterations; + EXPECT_EQ(result.status, expected); + EXPECT_EQ(result.failed_band, 1); + EXPECT_LE(result.iterations, options.max_iterations); + std::vector ax(3 * ld, T(0)); + op.apply(x.data(), ax.data(), ld, 3); + double residual2 = 0.0; + for (int i = 0; i < dim; ++i) + { + residual2 += std::norm(ax[ld + i] - b[ld + i]); + } +#ifdef __MPI + Parallel_Common::reduce_data(&residual2, 1, comm.comm); +#endif + EXPECT_NEAR(result.max_residual, std::sqrt(residual2), 1e-12); + EXPECT_GT(result.max_residual, options.tolerance); + } + } +} +template +void check_reconstruction(const bool corrupt_initial) +{ + const hsolver::diag_comm_info comm = linear_test::world_comm(); + const int n = 8; + const int bands = 3; + const int start = n * comm.rank / comm.nproc; + const int dim = n * (comm.rank + 1) / comm.nproc - start; + const int ld = dim + 2; + std::vector matrix = linear_test::hermitian(n); + for (int j = 0; j < n; ++j) + { + for (int i = 0; i < n; ++i) + { + matrix[j * n + i] = T(i == j ? 1 : 0) + T(0, 0.7) * matrix[j * n + i]; + } + } + std::vector exact(n * bands, T(0)); + for (int i = 0; i < n; ++i) + { + exact[n + i] = T(std::cos(0.3 * i), std::sin(0.4 * i)); + } + exact[2 * n] = T(1); + const std::vector rhs = linear_test::multiply(matrix, exact, n, bands); + const std::vector reference = linear_test::lapack_solve(matrix, rhs, n, bands); + std::vector b(ld * bands, T(0)); + std::vector x(ld * bands, T(0)); + std::vector bad_residual(ld * bands, T(0)); + for (int band = 0; band < bands; ++band) + { + for (int i = 0; i < dim; ++i) + { + b[band * ld + i] = rhs[band * n + start + i]; + } + } + hsolver::LinearSolveOptions options; + options.method = hsolver::LinearMethod::gmres; + // Above 100*epsilon in float: exercise reconstructed acceptance, not an automatic audit. + options.tolerance = std::is_same>::value ? 2e-5 : 1e-11; + options.restart = 2; + options.max_iterations = 100; + options.reconstruct = true; + DenseOperator op(matrix, n, start, dim, comm); + const IdentityOperator identity(dim); + hsolver::HSolverLinear solver(options, comm); + const T* initial = corrupt_initial ? bad_residual.data() : nullptr; + const hsolver::LinearSolveResult result = solver.solve(op, identity, ld, bands, dim, x.data(), b.data(), initial, corrupt_initial); + ASSERT_EQ(result.status, hsolver::LinearSolveStatus::converged); + EXPECT_EQ(result.failed_band, -1); + EXPECT_GT(result.restarts, 0); + EXPECT_GT(result.true_checks, 0); + EXPECT_LE(result.iterations, options.max_iterations); + EXPECT_EQ(result.reconstructed, !corrupt_initial); + EXPECT_EQ(result.reconstruction_fallbacks, corrupt_initial ? 1 : 0); + std::vector ax(ld * bands, T(0)); + op.apply(x.data(), ax.data(), ld, bands); + for (int band = 0; band < bands; ++band) + { + double error = 0.0; + double norm = 0.0; + for (int i = 0; i < dim; ++i) + { + error += std::norm(linear_test::Complex(ax[band * ld + i]) - linear_test::Complex(b[band * ld + i])); + norm += std::norm(linear_test::Complex(b[band * ld + i])); + EXPECT_LT(std::abs(linear_test::Complex(x[band * ld + i]) - reference[band * n + start + i]), 10 * options.tolerance); + } +#ifdef __MPI + Parallel_Common::reduce_data(&error, 1, comm.comm); + Parallel_Common::reduce_data(&norm, 1, comm.comm); +#endif + EXPECT_LE(std::sqrt(error), 1.2 * options.tolerance * std::max(1.0, std::sqrt(norm))); + } + if (corrupt_initial) + { + // Recovery shares the total iteration budget rather than starting a fresh allowance. + options.max_iterations = 1; + hsolver::HSolverLinear limited(options, comm); + std::fill(x.begin(), x.end(), T(0)); + const hsolver::LinearSolveResult exhausted = limited.solve(op, identity, ld, bands, dim, x.data(), b.data(), initial, true); + EXPECT_EQ(exhausted.status, hsolver::LinearSolveStatus::max_iterations); + EXPECT_EQ(exhausted.failed_band, 1); + EXPECT_EQ(exhausted.iterations, options.max_iterations); + EXPECT_EQ(exhausted.reconstruction_fallbacks, 1); + } +} + +TYPED_TEST(LinearSolveTest, ReconstructionSurvivesNormalRestarts) +{ + check_reconstruction>(false); +} + +TYPED_TEST(LinearSolveTest, ReconstructionRecoversIncorrectInitialResidual) +{ + check_reconstruction>(true); +} +} // namespace + +int main(int argc, char** argv) +{ + int nproc = 1; + int nthread = 1; + int rank = 0; + Parallel_Global::read_pal_param(argc, argv, nproc, nthread, rank); +#ifdef __MPI + // This numerical test creates no application pools; the shared finalizer owns these handles. + POOL_WORLD = MPI_COMM_NULL; + KP_WORLD = MPI_COMM_NULL; + INT_BGROUP = MPI_COMM_NULL; + BP_WORLD = MPI_COMM_NULL; + GRID_WORLD = MPI_COMM_NULL; + DIAG_WORLD = MPI_COMM_NULL; +#endif + testing::InitGoogleTest(&argc, argv); + const int result = RUN_ALL_TESTS(); +#ifdef __MPI + Parallel_Global::finalize_mpi(); +#endif + return result; +} diff --git a/source/source_hsolver/test/test_hsolver_pw.cpp b/source/source_hsolver/test/test_hsolver_pw.cpp index 0bd18093fb2..694349948d8 100644 --- a/source/source_hsolver/test/test_hsolver_pw.cpp +++ b/source/source_hsolver/test/test_hsolver_pw.cpp @@ -5,11 +5,32 @@ #include "hsolver_pw_sup.h" #include "hsolver_supplementary_mock.h" -#include "source_hamilt/module_xc/general_exx_info.h" // for General_Exx_Info type #include "source_hsolver/diag_comm_info.h" +#include "source_hsolver/hs_operator.h" #include "source_hsolver/hsolver_lcaopw.h" #include "source_hsolver/hsolver_pw.h" +#include +#include + +/// H = S = identity: the simplest operator the solvers can be handed +template +class IdentityHSOperator : public hsolver::HSOperator +{ + public: + void update_k(const int ik) override + { + } + void hpsi(const T* x, T* hx, const int ld, const int nvec) const override + { + std::copy(x, x + static_cast(ld) * nvec, hx); + } + void spsi(const T* x, T* sx, const int ld, const int nvec) const override + { + std::copy(x, x + static_cast(ld) * nvec, sx); + } +}; + // Mock implementations for the template functions causing linking errors namespace ModulePW { // Mock implementation for recip_to_real @@ -152,13 +173,13 @@ class TestHSolverPW : public ::testing::Test { // the protected hamiltSolvePsiK() is routed through here. template static void hamiltSolvePsiK(hsolver::HSolverPW& hs, - hamilt::Hamilt* h, + const hsolver::HSOperator& op, psi::Psi& ps, std::vector::type>& pre, typename GetTypeReal::type* eig, const int ntry) { - hs.hamiltSolvePsiK(h, ps, pre, eig, ntry); + hs.hamiltSolvePsiK(op, ps, pre, eig, ntry); } ModulePW::PW_Basis_K pwbk; @@ -197,8 +218,8 @@ class TestHSolverPW : public ::testing::Test { 0, 0); - hamilt::Hamilt> hamilt_test_d; - hamilt::Hamilt> hamilt_test_f; + IdentityHSOperator> hamilt_test_d; + IdentityHSOperator> hamilt_test_f; psi::Psi> psi_test_cd; psi::Psi> psi_test_cf; @@ -364,26 +385,41 @@ TEST_F(TestHSolverPW, SolveLcaoInPW) { transform_test_cd.resize(1, 3, 3); transform_test_cf.resize(1, 3, 3); - std::complex psi_value_d = {0.0, 0.0}; - std::complex psi_value_f = {0.0, 0.0}; + // 1, 2, 3 / 4, 5, 6 / 7, 8, 9 would be rank deficient, so the diagonal is + // lifted to keep the three subspace vectors linearly independent for (int iband = 0; iband < transform_test_cd.get_nbands(); iband++) { for (int ibasis = 0; ibasis < transform_test_cd.get_nbasis(); ibasis++) { + const double value = iband * transform_test_cd.get_nbasis() + ibasis + 1 + (iband == ibasis ? 10.0 : 0.0); transform_test_cd .get_pointer()[iband * transform_test_cd.get_nbasis() + ibasis] - = psi_value_d; + = std::complex(value, 0.0); transform_test_cf .get_pointer()[iband * transform_test_cf.get_nbasis() + ibasis] - = psi_value_f; - psi_value_d += std::complex(1.0, 0.0); - psi_value_f += std::complex(1.0, 0.0); + = std::complex(value, 0.0); } } + // with H = S = 1 every subspace eigenvalue is 1 and the rotated psi must + // come out orthonormal + auto check_orthonormal = [](const auto& p, const double tol) { + const int nb = p.get_nbands(); + const int nbasis = p.get_nbasis(); + for (int i = 0; i < nb; i++) { + for (int j = 0; j < nb; j++) { + std::complex dot = 0.0; + for (int ig = 0; ig < nbasis; ig++) { + dot += std::conj(std::complex(p.get_pointer()[i * nbasis + ig])) + * std::complex(p.get_pointer()[j * nbasis + ig]); + } + EXPECT_NEAR(dot.real(), i == j ? 1.0 : 0.0, tol); + EXPECT_NEAR(dot.imag(), 0.0, tol); + } + } + }; // check solve() elecstate_test.ekb.c[0] = 1.0; elecstate_test.ekb.c[1] = 2.0; - General_Exx_Info exx_info_local; hsolver::HSolverLIP> hs_f_lip = hsolver::HSolverLIP>(&pwbk, false, "pw", "scf", elecstate_test.ekb.nc); hsolver::HSolverLIP> hs_d_lip @@ -394,44 +430,32 @@ TEST_F(TestHSolverPW, SolveLcaoInPW) { const hsolver::diag_comm_info diag_comm(0, 1); #endif std::ostringstream log; - hs_f_lip.solve(&hamilt_test_f, + hs_f_lip.solve(hamilt_test_f, psi_test_cf, &elecstate_test, transform_test_cf, diag_comm, log, - true, - 0.0, - 0, - exx_info_local); + true); EXPECT_NE(log.str().find("Average iterative diagonalization steps"), std::string::npos); EXPECT_DOUBLE_EQ(hsolver::DiagoIterAssist>::avg_iter, 0.0); - for (int i = 0; i < psi_test_cf.size(); i++) - { - EXPECT_DOUBLE_EQ(psi_test_cf.get_pointer()[i].real(), i); - } - EXPECT_DOUBLE_EQ(elecstate_test.ekb.c[0], 0.0); - EXPECT_DOUBLE_EQ(elecstate_test.ekb.c[1], 0.0); + check_orthonormal(psi_test_cf, 1e-5); + EXPECT_NEAR(elecstate_test.ekb.c[0], 1.0, 1e-5); + EXPECT_NEAR(elecstate_test.ekb.c[1], 1.0, 1e-5); elecstate_test.ekb.c[0] = 1.0; elecstate_test.ekb.c[1] = 2.0; - hs_d_lip.solve(&hamilt_test_d, + hs_d_lip.solve(hamilt_test_d, psi_test_cd, &elecstate_test, transform_test_cd, diag_comm, log, - true, - 0.0, - 0, - exx_info_local); + true); EXPECT_DOUBLE_EQ(hsolver::DiagoIterAssist>::avg_iter, 0.0); - for (int i = 0; i < psi_test_cd.size(); i++) - { - EXPECT_DOUBLE_EQ(psi_test_cd.get_pointer()[i].real(), i); - } - EXPECT_DOUBLE_EQ(elecstate_test.ekb.c[0], 0.0); - EXPECT_DOUBLE_EQ(elecstate_test.ekb.c[1], 0.0); + check_orthonormal(psi_test_cd, 1e-10); + EXPECT_NEAR(elecstate_test.ekb.c[0], 1.0, 1e-10); + EXPECT_NEAR(elecstate_test.ekb.c[1], 1.0, 1e-10); } // Test that the program exits with an error when npwx < nbands, @@ -444,7 +468,7 @@ TEST_F(TestHSolverPW, NpwxLessThanNbandsDeath) std::vector eigenvalues(5, 0.0); // Expect death from WARNING_QUIT due to npwx < nbands EXPECT_EXIT( - hamiltSolvePsiK(hs_d, &hamilt_test_d, psi_test_cd, precond, eigenvalues.data(), 1), + hamiltSolvePsiK(hs_d, hamilt_test_d, psi_test_cd, precond, eigenvalues.data(), 1), ::testing::ExitedWithCode(1), ".*" ); diff --git a/source/source_hsolver/test/test_hsolver_pw_tddft.cpp b/source/source_hsolver/test/test_hsolver_pw_tddft.cpp new file mode 100644 index 00000000000..ef61c402eda --- /dev/null +++ b/source/source_hsolver/test/test_hsolver_pw_tddft.cpp @@ -0,0 +1,423 @@ +#include "source_base/parallel_global.h" +#include "source_hsolver/hsolver_pw_tddft.h" +#include "source_hsolver/test/linear_test_utils.h" + +#include +#include +#include +#include +#include +#include + +namespace +{ +using Device = base_device::DEVICE_CPU; + +template +class DenseHamiltonian final : public hsolver::HSOperator +{ + public: + std::vector> matrices; + std::vector dimensions; + + explicit DenseHamiltonian(const std::vector& sizes) : dimensions(sizes) + { + for (const int n: sizes) + { + matrices.push_back(linear_test::hermitian(n)); + } + } + void update_k(const int ik) override + { + current_k_ = ik; + } + void hpsi(const T* x, T* y, const int ld, const int nvec) const override + { + const int n = dimensions.at(current_k_); + const std::vector& h = matrices.at(current_k_); + for (int band = 0; band < nvec; ++band) + { + for (int i = 0; i < n; ++i) + { + y[band * ld + i] = T(0); + for (int j = 0; j < n; ++j) + { + y[band * ld + i] += h[j * n + i] * x[band * ld + j]; + } + } + } + } + void spsi(const T* x, T* y, const int ld, const int nvec) const override + { + FAIL() << "PW CN propagation must not request an overlap operator"; + } + + private: + int current_k_ = -1; +}; + +void initialize_basis(const std::vector& sizes, ModulePW::PW_Basis_K* basis) +{ + basis->nks = sizes.size(); + basis->npwk_max = *std::max_element(sizes.begin(), sizes.end()); + basis->tpiba = 0.8; + basis->tpiba2 = 0.64; + // These arrays are owned and released by PW_Basis/PW_Basis_K. + basis->npwk = new int[basis->nks]; + basis->kvec_c = new ModuleBase::Vector3[basis->nks]; + const int size = basis->nks * basis->npwk_max; + basis->gcar = new ModuleBase::Vector3[size]; + basis->gk2 = new double[size](); + for (int ik = 0; ik < basis->nks; ++ik) + { + basis->npwk[ik] = sizes[ik]; + basis->kvec_c[ik] = ModuleBase::Vector3(0.1 * ik, 0.07, -0.03); + for (int i = 0; i < sizes[ik]; ++i) + { + const int index = ik * basis->npwk_max + i; + basis->gcar[index] = ModuleBase::Vector3(0.2 * i, 0.1 * (i % 3), -0.15); + const ModuleBase::Vector3 q = basis->gcar[index] + basis->kvec_c[ik]; + basis->gk2[index] = q.norm2(); + } + } +} + +template +std::vector dense_cn(const std::vector& h, const std::vector& previous, int n, int bands, double dt) +{ + std::vector lhs(n * n); + std::vector rhs_matrix(n * n); + for (int j = 0; j < n; ++j) + { + for (int i = 0; i < n; ++i) + { + const T identity = i == j ? T(1) : T(0); + lhs[j * n + i] = identity + T(0, dt / 4.0) * h[j * n + i]; + rhs_matrix[j * n + i] = identity - T(0, dt / 4.0) * h[j * n + i]; + } + } + const std::vector rhs = linear_test::multiply(rhs_matrix, previous, n, bands); + return linear_test::lapack_solve(lhs, rhs, n, bands); +} + +template +void check_dense_step(const DenseHamiltonian& op, const psi::Psi& previous, const psi::Psi& current, const double dt) +{ + const double tolerance = std::is_same>::value ? 1e-4 : 1e-10; + const int bands = current.get_nbands(); + for (int ik = 0; ik < current.get_nk(); ++ik) + { + previous.fix_k(ik); + current.fix_k(ik); + const int n = op.dimensions[ik]; + std::vector packed(n * bands); + for (int band = 0; band < bands; ++band) + { + for (int i = 0; i < n; ++i) + { + packed[band * n + i] = previous(band, i); + } + } + const std::vector reference = dense_cn(op.matrices[ik], packed, n, bands, dt); + for (int band = 0; band < bands; ++band) + { + for (int i = 0; i < n; ++i) + { + EXPECT_LT(std::abs(linear_test::Complex(current(band, i)) - reference[band * n + i]), tolerance); + } + for (int other = 0; other < bands; ++other) + { + T overlap(0); + for (int i = 0; i < n; ++i) + { + overlap += std::conj(current(other, i)) * current(band, i); + } + const T expected = band == other ? T(1) : T(0); + EXPECT_LT(std::abs(overlap - expected), tolerance); + } + } + } +} + +template +class PWTDDFTTest : public testing::Test +{ +}; +using Precisions = testing::Types; +TYPED_TEST_SUITE(PWTDDFTTest, Precisions); + +TYPED_TEST(PWTDDFTTest, DenseCNCorrectorsAndConservation) +{ + using T = std::complex; + const bool single_precision = std::is_same::value; + const double energy_tolerance = single_precision ? 1e-4 : 1e-12; + const std::vector sizes{9, 6}; + const int bands = 3; + const int ld = 12; + const double dt = 0.27; + const ModuleBase::Vector3 shift(0.13, -0.07, 0.02); +#ifdef __MPI + const hsolver::diag_comm_info comm(MPI_COMM_SELF, 0, 1); +#else + const hsolver::diag_comm_info comm(0, 1); +#endif + for (const std::string method: {"bicgstab", "cgs", "gmres", "bicgstab_cn", "cgs_cn"}) + { + for (const std::string precond: {"none", "kinetic", "kinetic_recycle", "kinetic_subspace"}) + { + const bool recycle = precond == "kinetic_recycle"; + const bool cn_variant = method == "bicgstab_cn" || method == "cgs_cn"; + if (cn_variant && precond == "none") + { + continue; + } + if (!cn_variant && method != "gmres" && (recycle || precond == "kinetic_subspace")) + { + continue; + } + SCOPED_TRACE(method + "/" + precond); + ModulePW::PW_Basis_K basis; + initialize_basis(sizes, &basis); + DenseHamiltonian op(sizes); + std::ostringstream log; + hsolver::PWLinearOptions options; + options.global_k_indices = {0, 1}; + const std::string solver_name = cn_variant ? method.substr(0, method.size() - 3) : method; + options.linear.method = hsolver::parse_linear_method(solver_name); + options.linear.tolerance = single_precision ? 2e-6 : 1e-13; + options.linear.max_iterations = 100; + options.preconditioner = hsolver::parse_pw_precond(precond); + options.kinetic_enabled = true; + options.linear.restart = 2; + options.linear.reconstruct = method == "gmres" && recycle; + options.cn_init = method == "gmres" || cn_variant; + hsolver::HSolverPWTDDFT solver(basis, options, comm, log); + psi::Psi previous(2, bands, ld, sizes, true); + for (int ik = 0; ik < 2; ++ik) + { + previous.fix_k(ik); + for (int band = 0; band < bands; ++band) + { + for (int i = 0; i < ld; ++i) + { + const double phase = 2.0 * std::acos(-1.0) * i * band / sizes[ik]; + previous(band, i) = i < sizes[ik] ? T(std::polar(1.0 / std::sqrt(sizes[ik]), phase)) : T(0); + } + } + } + psi::Psi current(previous); + for (int step = 1; step <= 12; ++step) + { + // A time-step change must invalidate both k-point histories before reuse. + const double step_dt = step < 4 ? dt : dt * 0.5; + log.str(""); + log.clear(); + solver.solve(op, previous, ¤t, step_dt, shift, step, 1, true); + check_dense_step(op, previous, current, step_dt); + if (recycle || precond == "kinetic_subspace") + { + std::istringstream records(log.str()); + std::string line; + int count = 0; + while (std::getline(records, line)) + { + const std::string key = "coarse_rank="; + const std::size_t position = line.find(key); + if (position == std::string::npos) + { + continue; + } + const int rank = std::stoi(line.substr(position + key.size())); + if (recycle && (step == 1 || step == 4)) + { + EXPECT_EQ(rank, 0); + } + else + { + EXPECT_GT(rank, 0); + } + EXPECT_NE(line.find("cn_initial=1"), std::string::npos); + EXPECT_NE(line.find("kinetic_retry=0"), std::string::npos); + ++count; + } + EXPECT_EQ(count, sizes.size()); + } + if (step == 1) + { + // A corrector rebuilds the RHS from the fixed previous state, even with a changed H and guess. + for (int ik = 0; ik < 2; ++ik) + { + const int n = sizes[ik]; + current.fix_k(ik); + for (int i = 0; i < n; ++i) + { + op.matrices[ik][i * n + i] += 0.15 * (i + 1); + } + for (int band = 0; band < bands; ++band) + { + for (int i = 0; i < n; ++i) + { + current(band, i) *= T(0.8, 0.1); + } + } + } + log.str(""); + log.clear(); + solver.solve(op, previous, ¤t, step_dt, shift, step, 2, true); + check_dense_step(op, previous, current, step_dt); + EXPECT_EQ(log.str().find("cn_initial=1"), std::string::npos); + } + previous = current; + } + // Evaluate expectations with a different endpoint H, not the last propagation H. + for (int ik = 0; ik < 2; ++ik) + { + for (int i = 0; i < sizes[ik]; ++i) + { + op.matrices[ik][i * sizes[ik] + i] += 0.3; + } + } + ModuleBase::matrix energies(2, bands); + solver.cal_band_energy(op, current, &energies); + for (int ik = 0; ik < 2; ++ik) + { + current.fix_k(ik); + const int n = sizes[ik]; + for (int band = 0; band < bands; ++band) + { + T expected(0); + for (int j = 0; j < n; ++j) + { + for (int i = 0; i < n; ++i) + { + expected += std::conj(current(band, i)) * op.matrices[ik][j * n + i] * current(band, j); + } + } + EXPECT_NEAR(energies(ik, band), expected.real(), energy_tolerance); + } + } + } + } +} +TYPED_TEST(PWTDDFTTest, UnequalOccupationsAndDiagnosticsPreservePropagation) +{ + using T = std::complex; + using Wide = linear_test::Complex; + const bool single = std::is_same::value; + const double tolerance = single ? 1e-4 : 1e-10; + const int n = 6; + const int bands = 3; + const int ld = 8; + const double dt = 0.27; + const std::vector sizes{n}; + const ModuleBase::Vector3 shift(0, 0, 0); + ModulePW::PW_Basis_K basis; + initialize_basis(sizes, &basis); + DenseHamiltonian op(sizes); + const std::vector h(op.matrices[0].begin(), op.matrices[0].end()); +#ifdef __MPI + const hsolver::diag_comm_info comm(MPI_COMM_SELF, 0, 1); +#else + const hsolver::diag_comm_info comm(0, 1); +#endif + ModuleBase::matrix occupations(1, bands); + occupations(0, 0) = 1.0; + occupations(0, 1) = 0.4; + occupations(0, 2) = 0.0; + std::vector without_stat; + for (const bool out_stat: {false, true}) + { + std::ostringstream log; + hsolver::PWLinearOptions options; + options.global_k_indices = {0}; + options.linear.method = hsolver::LinearMethod::gmres; + options.linear.tolerance = single ? 2e-6 : 1e-13; + options.linear.max_iterations = 100; + options.linear.restart = n; + options.preconditioner = hsolver::PWPreconditioner::none; + options.out_stat = out_stat; + hsolver::HSolverPWTDDFT solver(basis, options, comm, log); + psi::Psi previous(1, bands, ld, sizes, true); + std::fill_n(previous.get_pointer(), ld * bands, T(0)); + std::vector reference(n * bands, Wide(0)); + for (int band = 0; band < bands; ++band) + { + previous(band, band) = T(1); + reference[band * n + band] = Wide(1); + } + psi::Psi initial(previous); + initial(0, 0) = T(1.01); + const psi::Psi original(initial); + solver.check_initial(initial, 1); + EXPECT_EQ(std::memcmp(initial.get_pointer(), original.get_pointer(), ld * bands * sizeof(T)), 0); + // The finite initial defect is diagnostic; propagate the orthonormal reference state. + psi::Psi current(previous); + for (int step = 1; step <= 8; ++step) + { + solver.reset_orth_stats(); + solver.solve(op, previous, ¤t, dt, shift, step, 1, false); + reference = dense_cn(h, reference, n, bands, dt); + for (int band = 0; band < bands; ++band) + { + for (int i = 0; i < n; ++i) + { + EXPECT_LT(std::abs(Wide(current(band, i)) - reference[band * n + i]), tolerance); + } + } + for (int j = 0; j < n; ++j) + { + for (int i = 0; i < n; ++i) + { + Wide actual(0); + Wide expected(0); + for (int band = 0; band < bands; ++band) + { + actual += occupations(0, band) * Wide(current(band, i)) * std::conj(Wide(current(band, j))); + expected += occupations(0, band) * reference[band * n + i] * std::conj(reference[band * n + j]); + } + // Reference columns have unit norm, hence every coefficient has magnitude <= 1. + const double density_bound = 1.4 * (2.0 * tolerance + tolerance * tolerance); + EXPECT_LT(std::abs(actual - expected), density_bound); + } + } + const double expected_electrons = out_stat ? 1.4 : 0.0; + EXPECT_NEAR(solver.wave_electrons(occupations), expected_electrons, tolerance); + const T* data = current.get_pointer(); + if (!out_stat) + { + without_stat.insert(without_stat.end(), data, data + ld * bands); + } + else + { + const T* baseline = without_stat.data() + (step - 1) * ld * bands; + EXPECT_EQ(std::memcmp(data, baseline, ld * bands * sizeof(T)), 0); + } + previous = current; + } + } +} + +} // namespace + +int main(int argc, char** argv) +{ + int nproc = 1; + int nthread = 1; + int rank = 0; + Parallel_Global::read_pal_param(argc, argv, nproc, nthread, rank); +#ifdef __MPI + // This numerical test creates no application pools; the shared finalizer owns these handles. + POOL_WORLD = MPI_COMM_NULL; + KP_WORLD = MPI_COMM_NULL; + INT_BGROUP = MPI_COMM_NULL; + BP_WORLD = MPI_COMM_NULL; + GRID_WORLD = MPI_COMM_NULL; + DIAG_WORLD = MPI_COMM_NULL; +#endif + testing::InitGoogleTest(&argc, argv); + const int result = RUN_ALL_TESTS(); +#ifdef __MPI + Parallel_Global::finalize_mpi(); +#endif + return result; +} diff --git a/source/source_hsolver/test/test_hsolver_sdft.cpp b/source/source_hsolver/test/test_hsolver_sdft.cpp deleted file mode 100644 index ddfdce7cc55..00000000000 --- a/source/source_hsolver/test/test_hsolver_sdft.cpp +++ /dev/null @@ -1,400 +0,0 @@ -#include -#include -#include - -#include "hsolver_pw_sup.h" -#include "hsolver_supplementary_mock.h" -#include "source_base/parallel_comm.h" -#include "source_estate/elecstate_pw.h" -#include "source_hsolver/hsolver_pw.h" -#include "source_hsolver/hsolver_pw_sdft.h" - -// mock for module_sdft -template -Sto_Func::Sto_Func() -{ -} -template class Sto_Func; - -// mock diago_hs_para -namespace hsolver { -template -void diago_hs_para(T* h, - T* s, - const int lda, - const int nband, - typename GetTypeReal::type* const ekb, - T* const wfc, - const MPI_Comm& comm, - const int diag_subspace, - const int block_size = 0) -{} -template void diago_hs_para(double* h, - double* s, - const int lda, - const int nband, - typename GetTypeReal::type* const ekb, - double* const wfc, - const MPI_Comm& comm, - const int diag_subspace, - const int block_size); -template void diago_hs_para>(std::complex* h, - std::complex* s, - const int lda, - const int nband, - typename GetTypeReal>::type* const ekb, - std::complex* const wfc, - const MPI_Comm& comm, - const int diag_subspace, - const int block_size); -template void diago_hs_para(float* h, - float* s, - const int lda, - const int nband, - typename GetTypeReal::type* const ekb, - float* const wfc, - const MPI_Comm& comm, - const int diag_subspace, - const int block_size); -template void diago_hs_para>(std::complex* h, - std::complex* s, - const int lda, - const int nband, - typename GetTypeReal>::type* const ekb, - std::complex* const wfc, - const MPI_Comm& comm, - const int diag_subspace, - const int block_size); - -} - - -template<> -void elecstate::ElecStatePW, base_device::DEVICE_CPU>::init_rho_data() -{ -} - -template class StoChe; - -template -Stochastic_Iter::Stochastic_Iter() -{ - change = false; - mu0 = 0; - method = 2; -} - -template -Stochastic_Iter::~Stochastic_Iter(){}; - -template -void Stochastic_Iter::init(K_Vectors* pkv_in, - ModulePW::PW_Basis_K* wfc_basis, - Stochastic_WF& stowf, - StoChe& stoche, - hamilt::HamiltSdftPW* p_hamilt_sto) -{ - this->nchip = stowf.nchip; - ; - this->targetne = 1; - this->method = stoche.method_sto; -} - -template -void Stochastic_Iter::orthog(const int& ik, psi::Psi& psi, Stochastic_WF& stowf) -{ - // do something to verify this function has been called - for (int i = 0; i < psi.size(); i++) - { - psi.get_pointer()[i] += 1.1; - } - return; -} - -template -void Stochastic_Iter::checkemm(const int& ik, - const int istep, - const int iter, - Stochastic_WF& stowf) -{ - // do something to verify this function has been called - stowf.nchi++; - return; -} - -template -void Stochastic_Iter::calPn(const int& ik, Stochastic_WF& stowf) -{ - // do something to verify this function has been called - stowf.nbands_diag++; - return; -} - -template -void Stochastic_Iter::itermu(int iter, elecstate::ElecState* pes) -{ - // do something to verify this function has been called - pes->f_en.eband += 1.2; - return; -} - -template -void Stochastic_Iter::calHsqrtchi(Stochastic_WF& stowf) -{ - // do something to verify this function has been called - stowf.nchip_max++; - return; -} - -template -void Stochastic_Iter::sum_stoeband(Stochastic_WF& stowf, - elecstate::ElecStatePW* pes, - hamilt::Hamilt* pHamilt, - ModulePW::PW_Basis_K* wfc_basis) -{ - // do something to verify this function has been called - stowf.nbands_total++; - return; -} - -template -void Stochastic_Iter::cal_storho(const UnitCell& ucell, - Stochastic_WF& stowf, - elecstate::ElecStatePW* pes, - ModulePW::PW_Basis_K* wfc_basis) -{ -} -template class Stochastic_Iter, base_device::DEVICE_CPU>; - -Charge::Charge(){}; -Charge::~Charge(){}; - -// Mock implementations for the template functions causing linking errors -namespace ModulePW { - // Mock implementation for recip_to_real - template - void PW_Basis_K::recip_to_real(const Device* ctx, - const std::complex* in, - std::complex* out, - const int ik, - const bool add, - const FPTYPE factor) const - { - // Simple mock implementation that does nothing - // In a real test, you might want to implement behavior that simulates the actual function - } - - // Mock implementation for real_to_recip - template - void PW_Basis_K::real_to_recip(const Device* ctx, - const std::complex* in, - std::complex* out, - const int ik, - const bool add, - const FPTYPE factor) const - { - // Simple mock implementation that does nothing - } - - // Explicit template instantiations - template void PW_Basis_K::recip_to_real( - const base_device::DEVICE_CPU* ctx, - const std::complex* in, - std::complex* out, - const int ik, - const bool add, - const float factor) const; - - template void PW_Basis_K::recip_to_real( - const base_device::DEVICE_CPU* ctx, - const std::complex* in, - std::complex* out, - const int ik, - const bool add, - const double factor) const; - - template void PW_Basis_K::real_to_recip( - const base_device::DEVICE_CPU* ctx, - const std::complex* in, - std::complex* out, - const int ik, - const bool add, - const float factor) const; - - template void PW_Basis_K::real_to_recip( - const base_device::DEVICE_CPU* ctx, - const std::complex* in, - std::complex* out, - const int ik, - const bool add, - const double factor) const; -} -/************************************************ - * unit test of HSolverPW_SDFT class - ***********************************************/ - -/** - * Tested function: - * - 1. solve() - * - with psi; - * - without psi; - * - skip charge; - * - 2. hsolver::HSolverPW_SDFT::diagethr (for cases below) - * - set_diagethr, for setting diagethr; - */ -class TestHSolverPW_SDFT : public ::testing::Test -{ - public: - TestHSolverPW_SDFT() : elecstate_test(nullptr, nullptr, nullptr, nullptr, nullptr, nullptr, nullptr) - { - stoche.nche = 8; - stoche.method_sto = 1; - } - ModulePW::PW_Basis_K pwbk; - Stochastic_WF> stowf; - K_Vectors kv; - StoChe stoche; - hamilt::HamiltSdftPW>* p_hamilt_sto = nullptr; - hsolver::HSolverPW_SDFT, base_device::DEVICE_CPU> hs_d - = hsolver::HSolverPW_SDFT, base_device::DEVICE_CPU>( - &kv, - &pwbk, - stowf, - stoche, - p_hamilt_sto, - "scf", - "pw", - "cg", - false, - 1, - hsolver::DiagoIterAssist>::SCF_ITER, - hsolver::DiagoIterAssist>::PW_DIAG_NMAX, - hsolver::DiagoIterAssist>::PW_DIAG_THR, - hsolver::DiagoIterAssist>::need_subspace, - 0, - false, - 4, - 0, - 0, - false, - true, - 1); - - hamilt::Hamilt> hamilt_test_d; - - psi::Psi> psi_test_cd; - psi::Psi> psi_test_no; - - elecstate::ElecStatePW> elecstate_test; - - std::string method_test = "cg"; - - std::ofstream temp_ofs; -}; - -// TEST_F(TestHSolverPW_SDFT, solve) -// { -// // initial memory and data -// elecstate_test.ekb.create(1, 2); -// elecstate_test.pot = new elecstate::Potential; -// elecstate_test.f_en.eband = 0.0; -// stowf.nbands_diag = 0; -// stowf.nbands_total = 0; -// stowf.nchi = 0; -// stowf.nchip_max = 0; -// psi_test_cd.resize(1, 2, 3); -// const double nelec = 1.0; -// int istep = 0; -// int iter = 0; - -// this->hs_d.solve(&hamilt_test_d, psi_test_cd, psi_test_cd, &elecstate_test, &pwbk, stowf, istep, iter, false); -// EXPECT_DOUBLE_EQ(hsolver::DiagoIterAssist>::avg_iter, 0.0); -// EXPECT_DOUBLE_EQ(elecstate_test.ekb.c[0], 4.0); -// EXPECT_DOUBLE_EQ(elecstate_test.ekb.c[1], 7.0); -// for (int i = 0; i < psi_test_cd.size(); i++) -// { -// // std::cout<<__FILE__<<__LINE__<<" "<rho = new double*[1]; -// elecstate_test.charge->rho[0] = new double[10]; -// elecstate_test.charge->nrxx = 10; -// elecstate_test.rho = new double*[1]; -// elecstate_test.rho[0] = new double[10]; -// int istep = 0; -// int iter = 0; - -// this->hs_d.solve(&hamilt_test_d, psi_test_no, psi_test_no, &elecstate_test, &pwbk, stowf, istep, iter, false); -// EXPECT_DOUBLE_EQ(hsolver::DiagoIterAssist>::avg_iter, 0.0); -// EXPECT_EQ(stowf.nbands_diag, 2); -// EXPECT_EQ(stowf.nbands_total, 1); -// EXPECT_EQ(stowf.nchi, 2); -// EXPECT_EQ(stowf.nchip_max, 1); -// EXPECT_DOUBLE_EQ(elecstate_test.f_en.eband, 1.2); -// /*std::cout<<__FILE__<<__LINE__<<" "<hs_d.solve(&hamilt_test_d, psi_test_no, psi_test_no, &elecstate_test, &pwbk, stowf, istep, iter, true); -// EXPECT_EQ(stowf.nbands_diag, 4); -// EXPECT_EQ(stowf.nbands_total, 1); -// EXPECT_EQ(stowf.nchi, 4); -// EXPECT_EQ(stowf.nchip_max, 2); -// EXPECT_DOUBLE_EQ(elecstate_test.f_en.eband, 2.4); - -// delete[] elecstate_test.charge->rho[0]; -// delete[] elecstate_test.charge->rho; -// delete elecstate_test.charge; -// } - -#ifdef __MPI -#include "source_base/parallel_comm.h" -#include "source_base/timer.h" -#include "mpi.h" -int main(int argc, char** argv) -{ - ModuleBase::timer::disable(); - MPI_Init(&argc, &argv); - testing::InitGoogleTest(&argc, argv); - - MPI_Comm_split(MPI_COMM_WORLD, 0, 1, &BP_WORLD); - int result = RUN_ALL_TESTS(); - - MPI_Comm_free(&BP_WORLD); - MPI_Finalize(); - - return result; -} - -#endif diff --git a/source/source_hsolver/test/test_linear_low_rank.cpp b/source/source_hsolver/test/test_linear_low_rank.cpp new file mode 100644 index 00000000000..6a4d110fba0 --- /dev/null +++ b/source/source_hsolver/test/test_linear_low_rank.cpp @@ -0,0 +1,123 @@ +#include "source_hsolver/linear_low_rank.h" +#include "source_hsolver/test/linear_test_utils.h" + +#include +#include + +namespace +{ +template +class LinearLowRankTest : public testing::Test +{ +}; +using Precisions = testing::Types; +TYPED_TEST_SUITE(LinearLowRankTest, Precisions); + +TYPED_TEST(LinearLowRankTest, CoarseAndHistoryCorrectionsMatchDenseReference) +{ + using T = std::complex; + using Wide = linear_test::Complex; + using Device = base_device::DEVICE_CPU; + const hsolver::diag_comm_info comm = linear_test::world_comm(); + const int n = 5; + const int rank = 2; + const int start = n * comm.rank / comm.nproc; + const int dim = n * (comm.rank + 1) / comm.nproc - start; + const int ld = dim + 1; + const double tolerance = std::is_same::value ? 2e-5 : 1e-12; + std::vector z(n * rank); + std::vector w(n * rank); + std::vector diagonal(n); + std::vector x(n); + for (int i = 0; i < n; ++i) + { + diagonal[i] = T(0.7 + 0.02 * i, 0.1); + x[i] = T(0.3 - 0.12 * i, 0.04 * i * i); + for (int j = 0; j < rank; ++j) + { + z[j * n + i] = T(j == 0 ? 1.0 : 0.1 * i, 0.1 * (j + i)); + w[j * n + i] = T(1.0, 0.2 * (i + 1)) * z[j * n + i]; + } + } + const std::vector local_z = linear_test::local_columns(z, n, rank, start, dim, ld); + const std::vector local_w = linear_test::local_columns(w, n, rank, start, dim, ld); + const std::vector local_d = linear_test::local_columns(diagonal, n, 1, start, dim, ld); + const std::vector local_x = linear_test::local_columns(x, n, 1, start, dim, ld); + const std::vector zero(ld * rank, T(0)); + hsolver::LinearAlgebra algebra(comm); + hsolver::LinearResponse response; + ct::Tensor workspace; + response.update(algebra, ld, dim, rank, local_z.data(), zero.data(), local_w.data(), 1e-8, &workspace); + ASSERT_EQ(response.rank(), rank); + ct::Tensor correction_workspace; + for (const bool history: {false, true}) + { + SCOPED_TRACE(history); + const std::vector& test = history ? w : z; + std::vector gram(rank * rank, Wide(0)); + std::vector projected(rank, Wide(0)); + for (int i = 0; i < rank; ++i) + { + for (int row = 0; row < n; ++row) + { + projected[i] += std::conj(Wide(test[i * n + row])) * Wide(x[row]); + for (int j = 0; j < rank; ++j) + { + gram[j * rank + i] += std::conj(Wide(test[i * n + row])) * Wide(w[j * n + row]); + } + } + } + const std::vector coefficients = linear_test::lapack_solve(gram, projected, rank, 1); + hsolver::LinearSmallLU factor; + ASSERT_TRUE(factor.factor(gram, rank)); + hsolver::LinearLowRank preconditioner(algebra, local_d.data(), dim); + if (history) + { + preconditioner.prepare_response(ld, response.rank(), response.directions(), response.images(), &correction_workspace); + } + else + { + preconditioner.prepare_subspace(ld, rank, local_z.data(), local_w.data(), factor, &correction_workspace); + } + ASSERT_EQ(preconditioner.rank(), rank); + std::vector actual(ld * rank, T(0)); + // Both corrections must satisfy P*(A*Z) = Z on the represented space. + preconditioner.apply(local_w.data(), actual.data(), ld, rank); + for (int j = 0; j < rank; ++j) + { + for (int i = 0; i < dim; ++i) + { + EXPECT_LT(std::abs(Wide(actual[j * ld + i]) - Wide(local_z[j * ld + i])), tolerance); + } + } + preconditioner.apply(local_x.data(), actual.data(), ld, 1); + for (int i = 0; i < dim; ++i) + { + const int row = start + i; + Wide expected = Wide(diagonal[row]) * Wide(x[row]); + for (int j = 0; j < rank; ++j) + { + expected += (Wide(z[j * n + row]) - Wide(diagonal[row]) * Wide(w[j * n + row])) * coefficients[j]; + } + EXPECT_LT(std::abs(Wide(actual[i]) - expected), tolerance); + } + if (history) + { + preconditioner.prepare_response(ld, 0, nullptr, nullptr, &correction_workspace); + } + else + { + preconditioner.prepare_subspace(ld, 0, nullptr, nullptr, factor, &correction_workspace); + } + EXPECT_EQ(preconditioner.rank(), 0); + preconditioner.apply(local_x.data(), actual.data(), ld, 1); + for (int i = 0; i < dim; ++i) + { + EXPECT_LT(std::abs(actual[i] - local_d[i] * local_x[i]), tolerance); + } + } + // A new unresolved response must discard an older usable history. + response.update(algebra, ld, dim, rank, zero.data(), zero.data(), zero.data(), 1e-8, &workspace); + EXPECT_EQ(response.rank(), 0); +} +} // namespace diff --git a/source/source_hsolver/test/test_orthonormal.cpp b/source/source_hsolver/test/test_orthonormal.cpp new file mode 100644 index 00000000000..50ff4231d75 --- /dev/null +++ b/source/source_hsolver/test/test_orthonormal.cpp @@ -0,0 +1,190 @@ +#include "source_hsolver/orthonormal.h" + +#include +#include +#include +#include +#include +#include +#include + +namespace +{ +using Wide = std::complex; +using Device = base_device::DEVICE_CPU; +constexpr int dim = 4; +constexpr int ld = 6; +constexpr int bands = 2; + +hsolver::diag_comm_info self_comm() +{ +#ifdef __MPI + return hsolver::diag_comm_info(MPI_COMM_SELF, 0, 1); +#else + return hsolver::diag_comm_info(0, 1); +#endif +} + +template +std::vector overlaps(const std::vector& psi) +{ + std::vector gram(bands * bands, Wide(0)); + for (int j = 0; j < bands; ++j) + { + for (int i = 0; i < bands; ++i) + { + for (int row = 0; row < dim; ++row) + { + gram[i + j * bands] += std::conj(Wide(psi[i * ld + row])) * Wide(psi[j * ld + row]); + } + } + } + return gram; +} + +template +std::vector orbitals() +{ + const double nan = std::numeric_limits::quiet_NaN(); + std::vector psi(ld * bands, T(nan, nan)); + for (int band = 0; band < bands; ++band) + { + std::fill_n(psi.begin() + band * ld, dim, T(0)); + } + psi[0] = T(1.001); + psi[ld] = T(0.002, -0.003); + psi[ld + 1] = T(0.999); + return psi; +} + +template +void check_orthogonal(const std::vector& psi) +{ + const std::vector gram = overlaps(psi); + const double tolerance = hsolver::orth_tolerance(); + for (int j = 0; j < bands; ++j) + { + for (int i = 0; i < bands; ++i) + { + const Wide expected = i == j ? Wide(1) : Wide(0); + EXPECT_LE(std::abs(gram[i + j * bands] - expected), tolerance); + } + } +} + +template +class OrthonormalTest : public testing::Test +{ +}; +using Precisions = testing::Types; +TYPED_TEST_SUITE(OrthonormalTest, Precisions); + +TYPED_TEST(OrthonormalTest, CorrectsComplexOverlapsWithoutFallback) +{ + using T = std::complex; + const hsolver::diag_comm_info comm = self_comm(); + hsolver::LinearAlgebra algebra(comm); + hsolver::Orthonormal orth(comm, algebra); + for (const hsolver::OrthMethod method: {hsolver::OrthMethod::cholesky, hsolver::OrthMethod::lowdin, hsolver::OrthMethod::newton_schulz}) + { + SCOPED_TRACE(hsolver::orth_method_name(method)); + std::vector psi = orbitals(); + const std::vector original(psi); + const hsolver::OrthResult result = orth.apply(psi.data(), ld, dim, bands, method, false); + ASSERT_EQ(result.status, hsolver::OrthStatus::accepted); + EXPECT_EQ(result.actual, method); + EXPECT_EQ(result.fallbacks, 0); + EXPECT_EQ(result.failure, hsolver::OrthFailure::none); + check_orthogonal(psi); + for (int band = 0; band < bands; ++band) + { + const int padding = band * ld + dim; + EXPECT_EQ(std::memcmp(psi.data() + padding, original.data() + padding, (ld - dim) * sizeof(T)), 0); + } + } +} + +TYPED_TEST(OrthonormalTest, NewtonSchulzFallsBackOutsideItsRange) +{ + using T = std::complex; + const hsolver::diag_comm_info comm = self_comm(); + hsolver::LinearAlgebra algebra(comm); + hsolver::Orthonormal orth(comm, algebra); + std::vector psi(ld * bands, T(0)); + psi[0] = T(2); + psi[ld + 1] = T(2); + const hsolver::OrthResult result = orth.apply(psi.data(), ld, dim, bands, hsolver::OrthMethod::newton_schulz, false); + ASSERT_EQ(result.status, hsolver::OrthStatus::accepted); + EXPECT_EQ(result.actual, hsolver::OrthMethod::cholesky); + EXPECT_EQ(result.fallbacks, 1); + EXPECT_EQ(result.failure, hsolver::OrthFailure::none); + check_orthogonal(psi); +} + +TYPED_TEST(OrthonormalTest, RejectsInvalidOrbitalsWithoutOverwritingThem) +{ + using T = std::complex; + const hsolver::diag_comm_info comm = self_comm(); + hsolver::LinearAlgebra algebra(comm); + hsolver::Orthonormal orth(comm, algebra); + const double nan = std::numeric_limits::quiet_NaN(); + const double inf = std::numeric_limits::infinity(); + for (const hsolver::OrthMethod method: {hsolver::OrthMethod::cholesky, hsolver::OrthMethod::none}) + { + for (const double value: {0.0, nan, inf}) + { + SCOPED_TRACE(hsolver::orth_method_name(method)); + SCOPED_TRACE(value); + std::vector psi(ld * bands, T(0)); + psi[0] = T(value); + psi[ld + 1] = T(1); + const std::vector original(psi); + const hsolver::OrthResult result = orth.apply(psi.data(), ld, dim, bands, method, false); + EXPECT_EQ(result.status, hsolver::OrthStatus::failed); + const hsolver::OrthFailure nonfinite + = method == hsolver::OrthMethod::none ? hsolver::OrthFailure::nonfinite_norm : hsolver::OrthFailure::nonfinite_gram; + const hsolver::OrthFailure expected = value == 0.0 ? hsolver::OrthFailure::nonpositive_norm : nonfinite; + EXPECT_EQ(result.failure, expected); + EXPECT_EQ(std::memcmp(psi.data(), original.data(), psi.size() * sizeof(T)), 0); + } + } + std::vector duplicate(ld * bands, T(0)); + duplicate[0] = T(1); + duplicate[ld] = T(1); + const std::vector original(duplicate); + const hsolver::OrthResult result = orth.apply(duplicate.data(), ld, dim, bands, hsolver::OrthMethod::cholesky, false); + EXPECT_EQ(result.status, hsolver::OrthStatus::failed); + EXPECT_EQ(result.failure, hsolver::OrthFailure::factorization_failed); + EXPECT_EQ(std::memcmp(duplicate.data(), original.data(), duplicate.size() * sizeof(T)), 0); +} + +TYPED_TEST(OrthonormalTest, InitialInspectionAndNoneAreReadOnly) +{ + using T = std::complex; + const hsolver::diag_comm_info comm = self_comm(); + hsolver::LinearAlgebra algebra(comm); + hsolver::Orthonormal orth(comm, algebra); + std::vector psi = orbitals(); + const std::vector original(psi); + const std::vector gram = overlaps(psi); + for (const bool collect: {false, true}) + { + const hsolver::OrthResult inspected = orth.inspect(psi.data(), ld, dim, bands, true, collect); + const hsolver::OrthResult disabled = orth.apply(psi.data(), ld, dim, bands, hsolver::OrthMethod::none, collect); + EXPECT_EQ(inspected.status, hsolver::OrthStatus::inspected); + EXPECT_GT(inspected.after, hsolver::orth_tolerance()); + EXPECT_EQ(disabled.status, hsolver::OrthStatus::disabled); + EXPECT_FALSE(disabled.gram_checked); + EXPECT_TRUE(std::isnan(disabled.after)); + const std::size_t expected_size = collect ? bands : 0; + ASSERT_EQ(inspected.norms.size(), expected_size); + ASSERT_EQ(disabled.norms.size(), expected_size); + for (int band = 0; collect && band < bands; ++band) + { + EXPECT_NEAR(inspected.norms[band], gram[band + band * bands].real(), 1e-14); + EXPECT_NEAR(disabled.norms[band], gram[band + band * bands].real(), 1e-14); + } + EXPECT_EQ(std::memcmp(psi.data(), original.data(), psi.size() * sizeof(T)), 0); + } +} +} // namespace diff --git a/source/source_io/CMakeLists.txt b/source/source_io/CMakeLists.txt index 72520a82d93..488235c6b1e 100644 --- a/source/source_io/CMakeLists.txt +++ b/source/source_io/CMakeLists.txt @@ -13,7 +13,6 @@ list(APPEND objects module_ml/cal_mlkedf_desc.cpp module_dos/write_dos_pw.cpp module_energy/write_bands.cpp - module_energy/nscf_fermi_surf.cpp module_energy/write_eig_occ.cpp module_bessel/numerical_basis.cpp module_bessel/numerical_basis_overlap.cpp @@ -33,6 +32,7 @@ list(APPEND objects module_ml/write_mlkedf_desc.cpp module_current/td_current_io.cpp module_current/td_current_io_comm.cpp + module_current/td_current_pw.cpp module_efield/td_efield_io.cpp module_efield/td_vector_pot_io.cpp module_chgpot/write_libxc_r.cpp @@ -65,6 +65,7 @@ if(ENABLE_LCAO) module_dos/write_dos_lcao.cpp module_dos/cal_pdos_gamma.cpp module_dos/cal_pdos_multik.cpp + module_dos/write_pdos_text.cpp module_output/write_orb_info.cpp module_energy/write_proj_band_lcao.cpp module_chgpot/get_pchg_lcao.cpp @@ -80,23 +81,40 @@ if(ENABLE_LCAO) module_mulliken/output_dmk.cpp module_mulliken/output_mulliken.cpp module_ml/io_npz.cpp - module_hs/cal_plpr.cpp + module_hs/angmom_op_mat.cpp module_dhs/write_dh.cpp module_dhs/write_dh_terms.cpp ) list(APPEND objects_advanced module_unk/unk_overlap_lcao.cpp - module_hs/write_hs_r.cpp - module_hs/write_h_terms.cpp - module_hs/write_hs_sparse.cpp - module_hs/single_r_io.cpp - module_hs/rr_sparse_writer.cpp - module_hs/cal_r_overlap_r.cpp + module_hs/hs_dense_io.cpp + module_hs/hs_sparse_io.cpp + module_hs/dhs_sparse_writer.cpp + module_hs/hsk_writer.cpp + module_hs/hsr_writer.cpp + module_hs/hsr_legacy.cpp + module_hs/hterm_writer.cpp + module_hs/lat_r_csr.cpp + module_hs/pos_op_csr.cpp + module_hs/pos_op_basis.cpp + module_hs/pos_op_calc.cpp + module_hs/pos_op_writer.cpp + module_hs/pos_op_mat.cpp + module_hs/vxc_op_tools.cpp + module_hs/vxc_op_mat.cpp + module_hs/vxc_op_r.cpp + module_hs/vxc_op_lip.cpp module_hs/output_mat_sparse.cpp + module_energy/write_eband_terms.cpp module_ctrl/ctrl_scf_lcao.cpp module_ctrl/ctrl_runner_lcao.cpp module_ctrl/ctrl_iter_lcao.cpp ) + if(ENABLE_LIBRI) + list(APPEND objects + module_restart/restart_exx_csr.cpp + ) + endif() endif() add_library( @@ -148,10 +166,12 @@ endif() if(BUILD_TESTING) add_subdirectory(module_parameter/test) + add_subdirectory(module_hs/unittests) add_subdirectory(test_serial) add_subdirectory(test) + add_subdirectory(module_dos/unittests) endif() -if(ENABLE_RAPIDJSON) +if(ENABLE_JSON) add_subdirectory(module_json) endif() diff --git a/source/source_io/build_info.h.in b/source/source_io/build_info.h.in index 707a49d99f7..543ef3e1b41 100644 --- a/source/source_io/build_info.h.in +++ b/source/source_io/build_info.h.in @@ -52,7 +52,7 @@ // --- Other Libraries --- #define ABACUS_DFTD4_VERSION "@ABACUS_DFTD4_VERSION@" -#define ABACUS_RAPIDJSON_VERSION "@ABACUS_RAPIDJSON_VERSION@" +#define ABACUS_JSON_VERSION "@ABACUS_JSON_VERSION@" #define ABACUS_PEXSI_VERSION "@ABACUS_PEXSI_VERSION@" #define ABACUS_CNPY_VERSION "@ABACUS_CNPY_VERSION@" diff --git a/source/source_io/module_bessel/bessel_basis.h b/source/source_io/module_bessel/bessel_basis.h index d46d7390fbd..8254d3ccaf7 100644 --- a/source/source_io/module_bessel/bessel_basis.h +++ b/source/source_io/module_bessel/bessel_basis.h @@ -17,173 +17,173 @@ class Bessel_Basis { public: - Bessel_Basis(); - ~Bessel_Basis(); - - /// @brief Initialization of Bessel function related matrices. - /// @details Used for a specific group of C4 coefficients. 2021-01-04, mohan added a new input parameter lmax_in, if we only generate numerical atomic orbitals based on spherical Bessel functions, lmax_in = ucell.lmax. However, if we want to generate Spherical Bessel functions (SBF) for descriptor, then the lmax_in is controlled by user. - /// @note This function is called in source_io/numerical_basis.cpp and source_io/numerical_descriptor.cpp - /// @param start_from_file whether read C4 coefficients stored in external files - /// @param ecutwfc cutoff for numerical atomic orbitals - /// @param ntype atom types - /// @param lmax_in maximal angular momentum for numerical orbitals - /// @param smooth whether smooth SBFs when perform integration to calculate value of matrix element of TableOne. For details, see J. Phys.: Condens. Matter 22 (2010) 445501 - /// @param sigma stddev of Gaussian function for smoothing SBFs - /// @param rcut_in cutoff radius for SBFs - /// @param tol_in accurancy control for SBFs - /// @param dk kspace grid - /// @param dr realspace grid - /// @param ucell UnitCell class object, ucell.nmax will be used in this function - void init( - const bool start_from_file, - const double &ecutwfc, - const int &ntype, - const int &lmax_in, - const bool &smooth, - const double &sigma, - const double &rcut_in, - const double &tol_in, - const UnitCell& ucell, - const double &dk = 0.01, - const double &dr = 0.01 - ); - /// @brief return number of SBFs used for one `chi` (see details for more information) - /// @details atomic orbital is constructed always with not only one set of SBFs. For different sets, they are marked with different `chi`(s), similar with concept of contracted GTOs. For one `chi`, it is 'q' the summation index, and q is in SBFs like: j_l(q*r), where l is the order of SBF. - /// @return number of SBFs - const int& get_ecut_number() const { return Ecut_number;} - - /// @brief Cubic spline interpolation for matrix Faln - /// @param it atom type index - /// @param l angular momentum - /// @param ic chi index - /// @param gnorm norm of G+k vector - /// @return interpolated value - double Polynomial_Interpolation(const int &it, const int &l, const int &ic, const double &gnorm)const; - /// @brief Cubic spline interpolation for matrix TableOne - /// @param l angular momentum - /// @param ie q index (see explanation in note of function BesselBasis::get_ecut_number()) - /// @param gnorm norm of G+k vector - /// @return interpolated value - double Polynomial_Interpolation2(const int &l, const int &ie, const double &gnorm)const; - - - /// @brief get energy cutoff, which is used to truncate SBF Jlq. - /// @param - /// @return energy cutoff in Ry - const double &get_ecut() const {return ecut;} - /// @brief cutoff radius of radial SBF Jlq. - /// @param - /// @return cutoff radius in a.u. - const double &get_rcut() const {return rcut;} - - const double &get_tolerence() const {return tolerence;} - - - /// @brief check if SBFs are smoothed (mohan add 2009-08-28) - /// @attention in this case, the Jlq are not the true Jlq. - /// @param - /// @return boolean whether SBFs are smoothed - const bool &get_smooth() const {return smooth;} - /// @brief get sigma the stddev (standard deviation) used in smooth function (Gaussian function) - /// @param - /// @return stddev of smooth function - const double &get_sigma() const {return sigma;} + Bessel_Basis(); + ~Bessel_Basis(); + + /// @brief Initialization of Bessel function related matrices. + /// @details Used for a specific group of C4 coefficients. 2021-01-04, mohan added a new input parameter lmax_in, if we only generate numerical atomic orbitals based on spherical Bessel functions, lmax_in = ucell.lmax. However, if we want to generate Spherical Bessel functions (SBF) for descriptor, then the lmax_in is controlled by user. + /// @note This function is called in source_io/numerical_basis.cpp and source_io/numerical_descriptor.cpp + /// @param start_from_file whether read C4 coefficients stored in external files + /// @param ecutwfc cutoff for numerical atomic orbitals + /// @param ntype atom types + /// @param lmax_in maximal angular momentum for numerical orbitals + /// @param smooth whether smooth SBFs when perform integration to calculate value of matrix element of TableOne. For details, see J. Phys.: Condens. Matter 22 (2010) 445501 + /// @param sigma stddev of Gaussian function for smoothing SBFs + /// @param rcut_in cutoff radius for SBFs + /// @param tol_in accurancy control for SBFs + /// @param dk kspace grid + /// @param dr realspace grid + /// @param ucell UnitCell class object, ucell.nmax will be used in this function + void init( + const bool start_from_file, + const double &ecutwfc, + const int &ntype, + const int &lmax_in, + const bool &smooth, + const double &sigma, + const double &rcut_in, + const double &tol_in, + const UnitCell& ucell, + const double &dk = 0.01, + const double &dr = 0.01 + ); + /// @brief return number of SBFs used for one `chi` (see details for more information) + /// @details atomic orbital is constructed always with not only one set of SBFs. For different sets, they are marked with different `chi`(s), similar with concept of contracted GTOs. For one `chi`, it is 'q' the summation index, and q is in SBFs like: j_l(q*r), where l is the order of SBF. + /// @return number of SBFs + const int& get_ecut_number() const { return Ecut_number;} + + /// @brief Cubic spline interpolation for matrix Faln + /// @param it atom type index + /// @param l angular momentum + /// @param ic chi index + /// @param gnorm norm of G+k vector + /// @return interpolated value + double Polynomial_Interpolation(const int &it, const int &l, const int &ic, const double &gnorm)const; + /// @brief Cubic spline interpolation for matrix TableOne + /// @param l angular momentum + /// @param ie q index (see explanation in note of function BesselBasis::get_ecut_number()) + /// @param gnorm norm of G+k vector + /// @return interpolated value + double Polynomial_Interpolation2(const int &l, const int &ie, const double &gnorm)const; + + + /// @brief get energy cutoff, which is used to truncate SBF Jlq. + /// @param + /// @return energy cutoff in Ry + const double &get_ecut() const {return ecut;} + /// @brief cutoff radius of radial SBF Jlq. + /// @param + /// @return cutoff radius in a.u. + const double &get_rcut() const {return rcut;} + + const double &get_tolerence() const {return tolerence;} + + + /// @brief check if SBFs are smoothed (mohan add 2009-08-28) + /// @attention in this case, the Jlq are not the true Jlq. + /// @param + /// @return boolean whether SBFs are smoothed + const bool &get_smooth() const {return smooth;} + /// @brief get sigma the stddev (standard deviation) used in smooth function (Gaussian function) + /// @param + /// @return stddev of smooth function + const double &get_sigma() const {return sigma;} private: - /// @brief the most important array to calculate spillage, has dimension (ntype, lmax+1, max_n, nk) - ModuleBase::realArray Faln; - - /// @brief Coefficients to be optimized! - ModuleBase::realArray C4; - - /// @brief matrix whose elements are int{dr r^2 j_l(qr)*j_l(kr)}, has dimension (lmax+1, nq, nk) - ModuleBase::realArray TableOne; - - /// @brief mesh of k vector, k is in j_l(k*r) - int kmesh=0; - /// @brief grid of k - double Dk; - /// @brief number of q vector, q is in j_l(q*r) - int Ecut_number; - /// @brief Cutoff radius (in a.u.) of SBFs, for any SBF j_l(qr), r>=rcut, j_l(q*r) = 0 (if not smoothed) - double rcut=0.0; - /// @brief energy cutoff for determining kmesh and number of SBFs - double ecut=0.0; - double tolerence=0.0; - /// @brief whether smooth SBFs around cutoff radius, resulting in non-zero values. For importance of smooth of SBFs, see J. Phys.: Condens. Matter 22 (2010) 445501, eqn 6. (mohan add 2009-01-18) - bool smooth=false; - /// @brief stddev of smooth function (Gaussian function, centered at rcut) - double sigma=0.0; - - /// @brief Allocate memory for C4 matrix and initialize all elements to one. - /// @param ntype number of atom types - /// @param lmax maximal angular momentum of localized orbitals - /// @param nmax maximal principal quantum number of localized orbitals - /// @param ecut_number number of SBFs - void allocate_C4( - const int &ntype, - const int &lmax, - const int &nmax, - const int &ecut_number, - const UnitCell& ucell - ); - - /// @brief Read C4 from external file. Presently an O(N^2) search algorithm is used. A HTML parser is needed in the future to improve performance. - /// @param name name of external file where C4-stored file information is contained - /// @param ntype number of atom types - /// @param ecut energy cutoff - /// @param rcut cutoff radius - /// @param ecut_number number of SBFs - /// @param tolerence accurancy of SBFs, here only used for consistency check - void readin_C4( - const std::string &name, - const int &ntype, - const int &ecut, - const int &rcut, - const int &ecut_number, - const double &tolerence, - const UnitCell& ucell - ); - - void init_TableOne(); - - /// @brief calculate F_{aln}(it, il, in, ik) = sum_{ie}{C4(it, il, in, ie)*TableOne(il, ie, ik)}, where TableOne is overlap integral between two spherical bessel functions (jle(r) and jlk(r)) - /// @param ntype number of atomtype - /// @param lmax maximal angular momentum - /// @param nmax maximal chi - /// @param ecut_number number of SBFs - void init_Faln( - const int &ntype, - const int &lmax, - const int &nmax, - const int &ecut_number, - const UnitCell& ucell - ); - - /// @brief number of localized wave functions - int nwfc=0; - - /// @brief calculate element value of TableOne matrix - /// @details (be called in Bessel_Basis::init(), used for outputing overlap Q matrix) initialize the table whose matrix element is the result of integral int{dr r^2 jle(r)*jlk(r)}, TableOne has three subscript (l, ie, ik), the first runs over orbitals' angular momentum and ie, ik run over ecut_number and kmesh SBFs - /// @param smooth_in whether jle(r) SBF is smoothed by a Gaussian function - /// @param sigma_in stddev for controlling smearing of Gaussian function for smoothing jle(r) - /// @param ecutwfc planewave kinetic energy cutoff for controlling kspace sampling - /// @param rcut cutoff radius of SBFs - /// @param dr realspace grid - /// @param dk kspace grid - /// @param lmax maximal angular momentum for SBFs - /// @param ecut_number number of SBFs - /// @param tolerence accurancy of SBFs - void init_TableOne( - const bool smooth_in, - const double &sigma_in, - const double &ecut, - const double &rcut, - const double &dr, - const double &dk, - const int &lmax, - const int &ecut_number, - const double &tolerence); + /// @brief the most important array to calculate spillage, has dimension (ntype, lmax+1, max_n, nk) + ModuleBase::realArray Faln; + + /// @brief Coefficients to be optimized! + ModuleBase::realArray C4; + + /// @brief matrix whose elements are int{dr r^2 j_l(qr)*j_l(kr)}, has dimension (lmax+1, nq, nk) + ModuleBase::realArray TableOne; + + /// @brief mesh of k vector, k is in j_l(k*r) + int kmesh=0; + /// @brief grid of k + double Dk; + /// @brief number of q vector, q is in j_l(q*r) + int Ecut_number; + /// @brief Cutoff radius (in a.u.) of SBFs, for any SBF j_l(qr), r>=rcut, j_l(q*r) = 0 (if not smoothed) + double rcut=0.0; + /// @brief energy cutoff for determining kmesh and number of SBFs + double ecut=0.0; + double tolerence=0.0; + /// @brief whether smooth SBFs around cutoff radius, resulting in non-zero values. For importance of smooth of SBFs, see J. Phys.: Condens. Matter 22 (2010) 445501, eqn 6. (mohan add 2009-01-18) + bool smooth=false; + /// @brief stddev of smooth function (Gaussian function, centered at rcut) + double sigma=0.0; + + /// @brief Allocate memory for C4 matrix and initialize all elements to one. + /// @param ntype number of atom types + /// @param lmax maximal angular momentum of localized orbitals + /// @param nmax maximal principal quantum number of localized orbitals + /// @param ecut_number number of SBFs + void allocate_C4( + const int &ntype, + const int &lmax, + const int &nmax, + const int &ecut_number, + const UnitCell& ucell + ); + + /// @brief Read C4 from external file. Presently an O(N^2) search algorithm is used. A HTML parser is needed in the future to improve performance. + /// @param name name of external file where C4-stored file information is contained + /// @param ntype number of atom types + /// @param ecut energy cutoff + /// @param rcut cutoff radius + /// @param ecut_number number of SBFs + /// @param tolerence accurancy of SBFs, here only used for consistency check + void readin_C4( + const std::string &name, + const int &ntype, + const int &ecut, + const int &rcut, + const int &ecut_number, + const double &tolerence, + const UnitCell& ucell + ); + + void init_TableOne(); + + /// @brief calculate F_{aln}(it, il, in, ik) = sum_{ie}{C4(it, il, in, ie)*TableOne(il, ie, ik)}, where TableOne is overlap integral between two spherical bessel functions (jle(r) and jlk(r)) + /// @param ntype number of atomtype + /// @param lmax maximal angular momentum + /// @param nmax maximal chi + /// @param ecut_number number of SBFs + void init_Faln( + const int &ntype, + const int &lmax, + const int &nmax, + const int &ecut_number, + const UnitCell& ucell + ); + + /// @brief number of localized wave functions + int nwfc=0; + + /// @brief calculate element value of TableOne matrix + /// @details (be called in Bessel_Basis::init(), used for outputing overlap Q matrix) initialize the table whose matrix element is the result of integral int{dr r^2 jle(r)*jlk(r)}, TableOne has three subscript (l, ie, ik), the first runs over orbitals' angular momentum and ie, ik run over ecut_number and kmesh SBFs + /// @param smooth_in whether jle(r) SBF is smoothed by a Gaussian function + /// @param sigma_in stddev for controlling smearing of Gaussian function for smoothing jle(r) + /// @param ecutwfc planewave kinetic energy cutoff for controlling kspace sampling + /// @param rcut cutoff radius of SBFs + /// @param dr realspace grid + /// @param dk kspace grid + /// @param lmax maximal angular momentum for SBFs + /// @param ecut_number number of SBFs + /// @param tolerence accurancy of SBFs + void init_TableOne( + const bool smooth_in, + const double &sigma_in, + const double &ecut, + const double &rcut, + const double &dr, + const double &dk, + const int &lmax, + const int &ecut_number, + const double &tolerence); }; #endif diff --git a/source/source_io/module_bessel/numerical_descriptor.cpp b/source/source_io/module_bessel/numerical_descriptor.cpp index f2ca03ec474..379d4384f94 100644 --- a/source/source_io/module_bessel/numerical_descriptor.cpp +++ b/source/source_io/module_bessel/numerical_descriptor.cpp @@ -7,70 +7,70 @@ Numerical_Descriptor::Numerical_Descriptor() { - this->init_label = false; - this->lmax = -1; - this->nmax = -1; + this->init_label = false; + this->lmax = -1; + this->nmax = -1; this->nlocal = 0; this->mu_index = nullptr; } Numerical_Descriptor::~Numerical_Descriptor() { - if(init_label==true) - { - delete[] mu_index; - } - return; + if(init_label==true) + { + delete[] mu_index; + } + return; } void Numerical_Descriptor::output_descriptor(const UnitCell& ucell, const psi::Psi> &psi, const int &lmax_in, const double &rcut_in, const double &tol_in, const int nks_in) { - ModuleBase::TITLE("Numerical_Descriptor","output_descriptor"); - ModuleBase::GlobalFunc::NEW_PART("DeepKS descriptor: D_{Inl}"); + ModuleBase::TITLE("Numerical_Descriptor","output_descriptor"); + ModuleBase::GlobalFunc::NEW_PART("DeepKS descriptor: D_{Inl}"); - //----------------------------------- - // 1. Initialize parameters - //----------------------------------- + //----------------------------------- + // 1. Initialize parameters + //----------------------------------- - //GlobalV::ofs_running << "D_{Inl}_m_m'=sum_{i}" << std::endl; - GlobalV::ofs_running << "input lmax = " << lmax_in << std::endl; - GlobalV::ofs_running << "input rcut = " << rcut_in << std::endl; - GlobalV::ofs_running << "input tolerence = " << tol_in << std::endl; - this->lmax = lmax_in; - assert(lmax>=0); + //GlobalV::ofs_running << "D_{Inl}_m_m'=sum_{i}" << std::endl; + GlobalV::ofs_running << "input lmax = " << lmax_in << std::endl; + GlobalV::ofs_running << "input rcut = " << rcut_in << std::endl; + GlobalV::ofs_running << "input tolerence = " << tol_in << std::endl; + this->lmax = lmax_in; + assert(lmax>=0); const int nks = nks_in; int ne = 0; - - // Peize Lin change 2022.12.15 + + // Peize Lin change 2022.12.15 // 0 stands for : 'Faln' is not used. this->bessel_basis.init( - false, - std::stod(PARAM.inp.bessel_descriptor_ecut), - ucell.ntype, - this->lmax, - PARAM.inp.bessel_descriptor_smooth, - PARAM.inp.bessel_descriptor_sigma, - rcut_in, - tol_in, + false, + std::stod(PARAM.inp.bessel_descriptor_ecut), + ucell.ntype, + this->lmax, + PARAM.inp.bessel_descriptor_smooth, + PARAM.inp.bessel_descriptor_sigma, + rcut_in, + tol_in, ucell ); - this->nmax = Numerical_Descriptor::bessel_basis.get_ecut_number(); + this->nmax = Numerical_Descriptor::bessel_basis.get_ecut_number(); this->init_mu_index(ucell); this->init_label = true; - assert(nmax>0); + assert(nmax>0); - // Currently we are not considering doing DeePKS in PW basis - // hence this subroutine is used only for generating projectors and save to jle.orb - // As a result, I will return here and the rest of the code is saved for future use - return; + // Currently we are not considering doing DeePKS in PW basis + // hence this subroutine is used only for generating projectors and save to jle.orb + // As a result, I will return here and the rest of the code is saved for future use + return; /* - //----------------------------------- - // 2. Open the file - //----------------------------------- + //----------------------------------- + // 2. Open the file + //----------------------------------- std::ofstream ofs; std::stringstream ss; ss << PARAM.inp.spillage_outdir << "/" << "descriptor.dat"; @@ -80,88 +80,88 @@ void Numerical_Descriptor::output_descriptor(const UnitCell& ucell, const psi::P } - //------------------------------------- - // 3. Initialize overlap_Q1 and Q2 - //------------------------------------- - // OVERLAP : < J_mu | Psi > + //------------------------------------- + // 3. Initialize overlap_Q1 and Q2 + //------------------------------------- + // OVERLAP : < J_mu | Psi > ModuleBase::realArray overlap_Q1(nks, PARAM.inp.nbands, this->nlocal ); ModuleBase::realArray overlap_Q2(nks, PARAM.inp.nbands, this->nlocal ); ModuleBase::GlobalFunc::ZEROS(overlap_Q1.ptr, overlap_Q1.getSize() ); ModuleBase::GlobalFunc::ZEROS(overlap_Q2.ptr, overlap_Q2.getSize() ); - ModuleBase::GlobalFunc::OUT(GlobalV::ofs_running,"number of k points",overlap_Q1.getBound1()); - ModuleBase::GlobalFunc::OUT(GlobalV::ofs_running,"number of bands",overlap_Q1.getBound2()); - ModuleBase::GlobalFunc::OUT(GlobalV::ofs_running,"number of local orbitals",overlap_Q1.getBound3()); + ModuleBase::GlobalFunc::OUT(GlobalV::ofs_running,"number of k points",overlap_Q1.getBound1()); + ModuleBase::GlobalFunc::OUT(GlobalV::ofs_running,"number of bands",overlap_Q1.getBound2()); + ModuleBase::GlobalFunc::OUT(GlobalV::ofs_running,"number of local orbitals",overlap_Q1.getBound3()); - //------------------------------------- - // 4. Compute overlap_Q1 and Q2 - //------------------------------------- + //------------------------------------- + // 4. Compute overlap_Q1 and Q2 + //------------------------------------- // nks now is the reduced k-points. for (int ik=0; ikngk[ik]; - GlobalV::ofs_running << " --------------------------------------------------------" << std::endl; - GlobalV::ofs_running << " Print the overlap matrixs Q and S for this kpoint"; + GlobalV::ofs_running << " --------------------------------------------------------" << std::endl; + GlobalV::ofs_running << " Print the overlap matrixs Q and S for this kpoint"; GlobalV::ofs_running << "\n " << std::setw(8) << "ik" << std::setw(8) << "npw"; GlobalV::ofs_running << "\n " << std::setw(8) << ik+1 << std::setw(8) << npw << std::endl; - GlobalV::ofs_running << " --------------------------------------------------------" << std::endl; + GlobalV::ofs_running << " --------------------------------------------------------" << std::endl; // search for all k-points. - psi.fix_k(ik); + psi.fix_k(ik); this->jlq3d_overlap(overlap_Q1, overlap_Q2, ik, ik, npw, psi); ModuleBase::GlobalFunc::DONE(GlobalV::ofs_running,"jlq3d_overlap"); - } + } #ifdef __MPI Parallel_Reduce::reduce_double_pool( overlap_Q1.ptr, overlap_Q1.getSize() ); Parallel_Reduce::reduce_double_pool( overlap_Q2.ptr, overlap_Q2.getSize() ); #endif - // do not need to output here + // do not need to output here //this->output_overlap_Q( ofs, overlap_Q1, overlap_Q2 ); - - //------------------------------------- - // 5. Generate descriptors for each atom - //------------------------------------- - - for (int it=0; itgenerate_descriptor(overlap_Q1, overlap_Q2, it ,ia, d, nd); - - ofs << ucell.atoms[it].label << " atom_index " << ia+1 << " n_descriptor " << nd << std::endl; - for(int id=0; id0 && id%8==0) ofs << std::endl; - // if(std::abs(d[id]>1.0e-9)) ofs << d[id] << " "; - // else ofs << "0 "; - ofs << d[id] << " "; - } - ofs << std::endl; - - delete[] d; - } - } + + //------------------------------------- + // 5. Generate descriptors for each atom + //------------------------------------- + + for (int it=0; itgenerate_descriptor(overlap_Q1, overlap_Q2, it ,ia, d, nd); + + ofs << ucell.atoms[it].label << " atom_index " << ia+1 << " n_descriptor " << nd << std::endl; + for(int id=0; id0 && id%8==0) ofs << std::endl; + // if(std::abs(d[id]>1.0e-9)) ofs << d[id] << " "; + // else ofs << "0 "; + ofs << d[id] << " "; + } + ofs << std::endl; + + delete[] d; + } + } @@ -336,48 +336,48 @@ normalization 2015-12-29 for (int ie=0; ie < nmax; ie++) void Numerical_Descriptor::init_mu_index(const UnitCell& ucell) { - GlobalV::ofs_running << " Initialize the mu index for deepks" << std::endl; - GlobalV::ofs_running << " lmax = " << this->lmax << std::endl; - GlobalV::ofs_running << " nmax = " << this->nmax << std::endl; + GlobalV::ofs_running << " Initialize the mu index for deepks" << std::endl; + GlobalV::ofs_running << " lmax = " << this->lmax << std::endl; + GlobalV::ofs_running << " nmax = " << this->nmax << std::endl; Numerical_Descriptor::mu_index = new ModuleBase::IntArray[ucell.ntype]; - assert(lmax>=0); - assert(nmax>0); - - int mu=0; - for (int it=0; itmu_index[it].create( - ucell.atoms[it].na, - lmax+1, // l starts from 0 - nmax, - 2*lmax+1); // m ==> 2*l+1 - - GlobalV::ofs_running << "Type " << it+1 - << " number_of_atoms " << ucell.atoms[it].na - << " number_of_L " << lmax+1 - << " number_of_n " << nmax - << " number_of_m " << 2*lmax+1 << std::endl; + assert(lmax>=0); + assert(nmax>0); + + int mu=0; + for (int it=0; itmu_index[it].create( + ucell.atoms[it].na, + lmax+1, // l starts from 0 + nmax, + 2*lmax+1); // m ==> 2*l+1 + + GlobalV::ofs_running << "Type " << it+1 + << " number_of_atoms " << ucell.atoms[it].na + << " number_of_L " << lmax+1 + << " number_of_n " << nmax + << " number_of_m " << 2*lmax+1 << std::endl; for (int ia=0; iamu_index[it](ia,l,n,m) = mu; - mu++; - } - } - } - } - - } - - this->nlocal = mu; - GlobalV::ofs_running << " total number of atomic orbitals " << nlocal << std::endl; - - return; + { + for (int l=0; lmu_index[it](ia,l,n,m) = mu; + mu++; + } + } + } + } + + } + + this->nlocal = mu; + GlobalV::ofs_running << " total number of atomic orbitals " << nlocal << std::endl; + + return; } diff --git a/source/source_io/module_bessel/numerical_descriptor.h b/source/source_io/module_bessel/numerical_descriptor.h index 6cd76f1d1cc..27be396732f 100644 --- a/source/source_io/module_bessel/numerical_descriptor.h +++ b/source/source_io/module_bessel/numerical_descriptor.h @@ -15,30 +15,30 @@ //========================================================== class Numerical_Descriptor { - public: - Numerical_Descriptor(); - ~Numerical_Descriptor(); + public: + Numerical_Descriptor(); + ~Numerical_Descriptor(); - void output_descriptor(const UnitCell& ucell, const psi::Psi> &psi, const int &lmax_in, const double &rcut_in, const double &tol_in, const int nks); // mohan added 2021-01-03 + void output_descriptor(const UnitCell& ucell, const psi::Psi> &psi, const int &lmax_in, const double &rcut_in, const double &tol_in, const int nks); // mohan added 2021-01-03 - private: + private: - bool init_label; + bool init_label; - int lmax; // lmax for descriptor - int nmax; // nmax for descriptor - int nlocal; // total number of atomic orbitals + int lmax; // lmax for descriptor + int nmax; // nmax for descriptor + int nlocal; // total number of atomic orbitals - Bessel_Basis bessel_basis; + Bessel_Basis bessel_basis; - ModuleBase::IntArray * mu_index = nullptr; - void init_mu_index(const UnitCell& ucell);//mohan added 2021-01-03 + ModuleBase::IntArray * mu_index = nullptr; + void init_mu_index(const UnitCell& ucell);//mohan added 2021-01-03 - // void jlq3d_overlap(ModuleBase::realArray &overlap_Q1, ModuleBase::realArray &overlap_Q2, - // const int &ik_ibz, const int &ik, const int &np, const psi::Psi> &psi); + // void jlq3d_overlap(ModuleBase::realArray &overlap_Q1, ModuleBase::realArray &overlap_Q2, + // const int &ik_ibz, const int &ik, const int &np, const psi::Psi> &psi); - // void generate_descriptor(ModuleBase::realArray &overlap_Q1, ModuleBase::realArray &overlap_Q2, - // const int &it, const int &ia, double *d, const int &nd); + // void generate_descriptor(ModuleBase::realArray &overlap_Q1, ModuleBase::realArray &overlap_Q2, + // const int &it, const int &ia, double *d, const int &nd); }; diff --git a/source/source_io/module_chgpot/get_pchg_lcao.cpp b/source/source_io/module_chgpot/get_pchg_lcao.cpp index ae11f7b6e74..a7c1d307d37 100644 --- a/source/source_io/module_chgpot/get_pchg_lcao.cpp +++ b/source/source_io/module_chgpot/get_pchg_lcao.cpp @@ -1,7 +1,7 @@ #include "get_pchg_lcao.h" -#include "source_estate/module_charge/symm_rho.h" -#include "source_estate/module_dm/cal_dm_psi.h" +#include "source_estate/module_charge/chg_symm.h" +#include "source_estate/module_dm/dm_from_psi.h" #include "source_hamilt/module_gint/gint_interface.h" #include "source_io/module_output/cube_io.h" @@ -61,17 +61,17 @@ void Get_pchg_lcao::begin_gamma(const UnitCell& ucell, } // Construct a band-resolved density matrix before evaluating its density on the grid. - elecstate::DensityMatrix DM(¶_orb_, nspin_); - elecstate::cal_dm_psi(¶_orb_, state_weights, *psi_gamma_, DM); + module_dm::DensityMatrix DM(¶_orb_, nspin_); + module_dm::dm_from_psi(¶_orb_, state_weights, *psi_gamma_, DM); for (int is = 0; is < nspin_; ++is) { std::fill(rho[is].begin(), rho[is].end(), 0.0); } - DM.init_DMR(&grid_driver, &ucell); - DM.cal_DMR(); - ModuleGint::cal_gint_rho(DM.get_DMR_vector(), nspin_, rho_pointers.data()); + DM.init_dmr(&grid_driver, &ucell); + DM.cal_dmr(-1); + ModuleGint::cal_gint_rho(DM.get_dmr_vec(), nspin_, rho_pointers.data()); for (int is = 0; is < nspin_; ++is) { @@ -147,8 +147,8 @@ void Get_pchg_lcao::begin_k(const ModulePW::PW_Basis& rho_pw, // Collinear spin channels are stored as two k blocks; spinors use one block per k point. const int nspin_dm = nspin_ == 2 ? 2 : 1; const int nk_output = kv.get_nks() / nspin_dm; - elecstate::DensityMatrix, double> DM(¶_orb_, nspin_dm, kv.kvec_d, nk_output); - elecstate::cal_dm_psi(¶_orb_, state_weights, *psi_k_, DM); + module_dm::DensityMatrix, double> DM(¶_orb_, nspin_dm, kv.kvec_d, nk_output); + module_dm::dm_from_psi(¶_orb_, state_weights, *psi_k_, DM); if (if_separate_k) { @@ -160,10 +160,10 @@ void Get_pchg_lcao::begin_k(const ModulePW::PW_Basis& rho_pw, std::fill(rho[is].begin(), rho[is].end(), 0.0); } - DM.init_DMR(&grid_driver, &ucell); + DM.init_dmr(&grid_driver, &ucell); // Transform only the requested real k point to avoid summing different k contributions. - DM.cal_DMR(ik); - ModuleGint::cal_gint_rho(DM.get_DMR_vector(), nspin_, rho_pointers.data()); + DM.cal_dmr(ik); + ModuleGint::cal_gint_rho(DM.get_dmr_vec(), nspin_, rho_pointers.data()); for (int is = 0; is < nspin_; ++is) { @@ -183,25 +183,24 @@ void Get_pchg_lcao::begin_k(const ModulePW::PW_Basis& rho_pw, std::fill(rho[is].begin(), rho[is].end(), 0.0); } - DM.init_DMR(&grid_driver, &ucell); + DM.init_dmr(&grid_driver, &ucell); // The no-argument transform sums all local k-point contributions into one density. - DM.cal_DMR(); - ModuleGint::cal_gint_rho(DM.get_DMR_vector(), nspin_, rho_pointers.data()); + DM.cal_dmr(-1); + ModuleGint::cal_gint_rho(DM.get_dmr_vec(), nspin_, rho_pointers.data()); // Symmetrize only the merged density, using coupled spin rotations for nspin=4. if (needs_symmetry) { - Symmetry_rho srho; if (nspin_ == 4) { - srho.begin(0, rho_pointers.data(), rhog_pointers.data(), rho_pw.npw, nullptr, &rho_pw, ucell.symm); - srho.begin_soc(rho_pointers.data(), rhog_pointers.data(), &rho_pw, ucell.symm); + module_charge::cal_rhog_symm(0, rho_pointers.data(), rhog_pointers.data(), rho_pw.npw, nullptr, &rho_pw, ucell.symm); + module_charge::cal_rhog_symm_soc(rho_pointers.data(), rhog_pointers.data(), &rho_pw, ucell.symm); } else { for (int is = 0; is < nspin_; ++is) { - srho.begin(is, rho_pointers.data(), rhog_pointers.data(), rho_pw.npw, nullptr, &rho_pw, ucell.symm); + module_charge::cal_rhog_symm(is, rho_pointers.data(), rhog_pointers.data(), rho_pw.npw, nullptr, &rho_pw, ucell.symm); } } } diff --git a/source/source_io/module_chgpot/get_pchg_pw.cpp b/source/source_io/module_chgpot/get_pchg_pw.cpp index 7abde6c7ffe..bb738965377 100644 --- a/source/source_io/module_chgpot/get_pchg_pw.cpp +++ b/source/source_io/module_chgpot/get_pchg_pw.cpp @@ -4,7 +4,7 @@ #include "source_base/module_device/memory_op.h" #include "source_base/module_parallel/para_bridge.h" #include "source_base/tool_quit.h" -#include "source_estate/module_charge/symm_rho.h" +#include "source_estate/module_charge/chg_symm.h" #include "source_estate/uspp_density.h" #include "source_io/module_output/cube_io.h" @@ -427,7 +427,6 @@ void Get_pchg_pw::sum_pools(const Parallel_Grid& pgrid, Workspace* wo template void Get_pchg_pw::symmetrize(UnitCell* ucell, Workspace* work) const { - Symmetry_rho srho; std::vector rho_pointers(nspin_); std::vector>> rhog(nspin_, std::vector>(pw_rhod_.npw)); std::vector*> rhog_pointers(nspin_); @@ -441,14 +440,14 @@ void Get_pchg_pw::symmetrize(UnitCell* ucell, Workspace* work) const if (work->is_spinor) { // Charge and magnetization obey different spinor symmetry transformations. - srho.begin(0, rho_pointers.data(), rhog_pointers.data(), pw_rhod_.npw, nullptr, &pw_rhod_, ucell->symm); - srho.begin_soc(rho_pointers.data(), rhog_pointers.data(), &pw_rhod_, ucell->symm); + module_charge::cal_rhog_symm(0, rho_pointers.data(), rhog_pointers.data(), pw_rhod_.npw, nullptr, &pw_rhod_, ucell->symm); + module_charge::cal_rhog_symm_soc(rho_pointers.data(), rhog_pointers.data(), &pw_rhod_, ucell->symm); } else { for (int is = 0; is < nspin_; ++is) { - srho.begin(is, rho_pointers.data(), rhog_pointers.data(), pw_rhod_.npw, nullptr, &pw_rhod_, ucell->symm); + module_charge::cal_rhog_symm(is, rho_pointers.data(), rhog_pointers.data(), pw_rhod_.npw, nullptr, &pw_rhod_, ucell->symm); } } } diff --git a/source/source_io/module_chgpot/write_libxc_r.cpp b/source/source_io/module_chgpot/write_libxc_r.cpp index d13e7563850..0e65a7d5887 100644 --- a/source/source_io/module_chgpot/write_libxc_r.cpp +++ b/source/source_io/module_chgpot/write_libxc_r.cpp @@ -11,7 +11,6 @@ #include "source_hamilt/module_xc/libxc_abacus.h" #include "source_estate/module_charge/charge.h" #include "source_basis/module_pw/pw_basis_big.h" -#include "source_basis/module_pw/pw_basis.h" #include "source_io/module_output/cube_io.h" #include "source_base/global_variable.h" #include "source_io/module_parameter/parameter.h" diff --git a/source/source_io/module_chgpot/write_libxc_r.h b/source/source_io/module_chgpot/write_libxc_r.h index 58e2f215565..c0ec9c51b44 100644 --- a/source/source_io/module_chgpot/write_libxc_r.h +++ b/source/source_io/module_chgpot/write_libxc_r.h @@ -12,40 +12,44 @@ #include class Charge; -namespace ModulePW{ class PW_Basis_Big; } -namespace ModulePW{ class PW_Basis; } + +namespace ModulePW +{ +class PW_Basis; +class PW_Basis_Big; +} namespace ModuleIO { - extern void write_libxc_r( - const int order, - const std::vector &func_id, - const int &nrxx, // number of real-space grid - const double &omega, // volume of cell - const double tpiba, - const Charge &chr, - const ModulePW::PW_Basis_Big &pw_big, - const ModulePW::PW_Basis &pw_rhod); + extern void write_libxc_r( + const int order, + const std::vector &func_id, + const int &nrxx, // number of real-space grid + const double &omega, // volume of cell + const double tpiba, + const Charge &chr, + const ModulePW::PW_Basis_Big &pw_big, + const ModulePW::PW_Basis &pw_rhod); #ifdef __MPI - extern void write_cube_core( - std::ofstream &ofs_cube, - const int bz, - const int nbz, - const int nplane, - const int startz_current, - const double*const data, - const int nxy, - const int nz, - const int nld, - const int n_data_newline); + extern void write_cube_core( + std::ofstream &ofs_cube, + const int bz, + const int nbz, + const int nplane, + const int startz_current, + const double*const data, + const int nxy, + const int nz, + const int nld, + const int n_data_newline); #else - extern void write_cube_core( - std::ofstream &ofs_cube, - const double*const data, - const int nxy, - const int nz, - const int n_data_newline); + extern void write_cube_core( + std::ofstream &ofs_cube, + const double*const data, + const int nxy, + const int nz, + const int n_data_newline); #endif } diff --git a/source/source_io/module_ctrl/ctrl_iter_lcao.cpp b/source/source_io/module_ctrl/ctrl_iter_lcao.cpp index 22a66fcff68..3f23393d306 100644 --- a/source/source_io/module_ctrl/ctrl_iter_lcao.cpp +++ b/source/source_io/module_ctrl/ctrl_iter_lcao.cpp @@ -19,7 +19,7 @@ void ctrl_iter_lcao(UnitCell& ucell, // unit cell * const Input_para& inp, // input parameters * K_Vectors& kv, // k points * elecstate::ElecState* pelec, // electronic info * - elecstate::DensityMatrix& dm, // density matrix, mohan add 2025-11-03 + module_dm::DensityMatrix& dm, // density matrix, mohan add 2025-11-03 Parallel_Orbitals& pv, // parallel orbital info * Grid_Driver& gd, // adjacent atom info * psi::Psi* psi, // wave functions * @@ -57,10 +57,10 @@ void ctrl_iter_lcao(UnitCell& ucell, // unit cell * if (cal_exx) { real_number ? - exx_nao.exd->exx_iter_finish(kv, ucell, *p_hamilt, *pelec, &dm, - *p_chgmix, scf_ene_thr, iter, istep, conv_esolver) : + exx_nao.exd->exx_iter_finish(kv, ucell, *p_hamilt, *pelec, &dm, + pv, *p_chgmix, scf_ene_thr, iter, istep, conv_esolver) : exx_nao.exc->exx_iter_finish(kv, ucell, *p_hamilt, *pelec, &dm, - *p_chgmix, scf_ene_thr, iter, istep, conv_esolver); + pv, *p_chgmix, scf_ene_thr, iter, istep, conv_esolver); } } #endif @@ -90,7 +90,7 @@ template void ctrl_iter_lcao(UnitCell& ucell, // unit cell * const Input_para& inp, // input parameters * K_Vectors& kv, // k points * elecstate::ElecState* pelec, // electronic info * - elecstate::DensityMatrix& dm, // density matrix, mohan add 2025-11-03 + module_dm::DensityMatrix& dm, // density matrix, mohan add 2025-11-03 Parallel_Orbitals& pv, // parallel orbital info * Grid_Driver& gd, // adjacent atom info * psi::Psi* psi, // wave functions * @@ -111,7 +111,7 @@ template void ctrl_iter_lcao, double>(UnitCell& ucell, // u const Input_para& inp, // input parameters * K_Vectors& kv, // k points * elecstate::ElecState* pelec, // electronic info * - elecstate::DensityMatrix, double>& dm, // density matrix, mohan add 2025-11-03 + module_dm::DensityMatrix, double>& dm, // density matrix, mohan add 2025-11-03 Parallel_Orbitals& pv, // parallel orbital info * Grid_Driver& gd, // adjacent atom info * psi::Psi>* psi, // wave functions * @@ -132,7 +132,7 @@ template void ctrl_iter_lcao, std::complex>(UnitCel const Input_para& inp, // input parameters * K_Vectors& kv, // k points * elecstate::ElecState* pelec, // electronic info * - elecstate::DensityMatrix, double>& dm, // density matrix, mohan add 2025-11-03 + module_dm::DensityMatrix, double>& dm, // density matrix, mohan add 2025-11-03 Parallel_Orbitals& pv, // parallel orbital info * Grid_Driver& gd, // adjacent atom info * psi::Psi>* psi, // wave functions * diff --git a/source/source_io/module_ctrl/ctrl_iter_lcao.h b/source/source_io/module_ctrl/ctrl_iter_lcao.h index 795c7d18db4..40e6748e08a 100644 --- a/source/source_io/module_ctrl/ctrl_iter_lcao.h +++ b/source/source_io/module_ctrl/ctrl_iter_lcao.h @@ -6,7 +6,7 @@ #include "source_estate/elecstate_lcao.h" // use elecstate::ElecStateLCAO #include "source_psi/psi.h" // use Psi #include "source_estate/module_charge/charge.h" // use charge -#include "source_estate/module_charge/charge_mixing.h" // use charge mixing +#include "source_estate/module_charge/chg_mix.h" // use charge mixing #include "source_lcao/hamilt_lcao.h" // use hamilt::HamiltLCAO #include "source_lcao/setup_exx.h" // mohan add 20251008 #include "source_lcao/setup_deepks.h" // mohan add 20251010 @@ -19,7 +19,7 @@ void ctrl_iter_lcao(UnitCell& ucell, // unit cell * const Input_para& inp, // input parameters * K_Vectors& kv, // k points * elecstate::ElecState* pelec, // electronic info * - elecstate::DensityMatrix& dm, // density matrix, mohan add 2025-11-03 + module_dm::DensityMatrix& dm, // density matrix, mohan add 2025-11-03 Parallel_Orbitals& pv, // parallel orbital info * Grid_Driver& gd, // adjacent atom info * psi::Psi* psi, // wave functions * diff --git a/source/source_io/module_ctrl/ctrl_output_fp.cpp b/source/source_io/module_ctrl/ctrl_output_fp.cpp index e00dbe3f03f..ca8f5173358 100644 --- a/source/source_io/module_ctrl/ctrl_output_fp.cpp +++ b/source/source_io/module_ctrl/ctrl_output_fp.cpp @@ -1,10 +1,14 @@ #include "ctrl_output_fp.h" // use ctrl_output_fp() -#include "../module_output/cube_io.h" // use write_vdata_palgrid -#include "../module_dipole/dipole_io.h" // use write_dipole -#include "source_estate/module_charge/symm_rho.h" // use Symmetry_rho -#include "source_hamilt/module_xc/xc_functional.h" // use XC_Functional -#include "source_estate/write_elecstat_pot.h" // use write_elecstat_pot + +#include "source_basis/module_pw/pw_basis_big.h" // use PW_Basis_Big +#include "source_estate/elecstate.h" // use elecstate::ElecState +#include "source_estate/module_charge/charge.h" // use Charge +#include "source_estate/module_charge/chg_symm.h" // use module_charge::cal_rhog_symm +#include "source_estate/write_elecstat_pot.h" // use write_elecstat_pot +#include "source_hamilt/module_xc/xc_functional.h" // use XC_Functional +#include "source_io/module_dipole/dipole_io.h" // use write_dipole #include "source_io/module_elf/write_elf.h" +#include "source_io/module_output/cube_io.h" // use write_vdata_palgrid #include "source_io/module_parameter/input_parameter.h" // use Input_para #ifdef __LIBXC @@ -77,7 +81,7 @@ void ctrl_output_fp(UnitCell& ucell, fn += spin_block + geom_block + ".cube"; ModuleIO::write_vdata_palgrid(para_grid, - chr.rho_save[is], + chr.rho[is], is, nspin, istep, // change istep_in to istep, mohan 20260222 @@ -96,7 +100,7 @@ void ctrl_output_fp(UnitCell& ucell, fn += spin_block + geom_block + ".cube"; ModuleIO::write_vdata_palgrid(para_grid, - chr.kin_r_save[is], + chr.kin_r[is], is, nspin, istep, @@ -104,7 +108,7 @@ void ctrl_output_fp(UnitCell& ucell, pelec->eferm.get_efval(is), &(ucell), 11, // default precision - 1, // default out_fermi + 1, // default out_fermi PARAM.globalv.two_fermi, false); } @@ -138,8 +142,8 @@ void ctrl_output_fp(UnitCell& ucell, fn, 0.0, // efermi &(ucell), - inp.out_pot[1], // precision - 0, // out_fermi + inp.out_pot[1], // precision + 0, // out_fermi PARAM.globalv.two_fermi, false); } @@ -167,25 +171,26 @@ void ctrl_output_fp(UnitCell& ucell, // 6) write ELF if (inp.out_elf[0] > 0 && should_output) { - chr.cal_elf = true; - Symmetry_rho srho; + // write_elf() consumes chr.kin_r; the final tau written by sum_band + // must be symmetrized before output (the tau buffer is allocated + // because ELF output was requested). for (int is = 0; is < nspin; is++) { - srho.begin(is, chr, pw_rhod, ucell.symm); + module_charge::cal_rhog_symm(is, chr, pw_rhod, ucell.symm); } std::string out_dir = PARAM.globalv.global_out_dir; ModuleIO::write_elf(out_dir, - istep, - nspin, - chr.rho, - chr.kin_r, - pw_rhod, - para_grid, - &(ucell), - inp.out_elf[1], - geom_block, - PARAM.globalv.two_fermi); + istep, + nspin, + chr.rho, + chr.kin_r, + pw_rhod, + para_grid, + &(ucell), + inp.out_elf[1], + geom_block, + PARAM.globalv.two_fermi); } #ifdef __LIBXC @@ -210,7 +215,7 @@ void ctrl_output_fp(UnitCell& ucell, { std::stringstream ss_dipole; ss_dipole << global_out_dir << "dipole_s" << is + 1 << ".txt"; - ModuleIO::write_dipole(ucell, chr.rho_save[is], pw_rhod, istep, ss_dipole.str(), GlobalV::ofs_running); + ModuleIO::write_dipole(ucell, chr.rho[is], pw_rhod, istep, ss_dipole.str(), GlobalV::ofs_running); } } diff --git a/source/source_io/module_ctrl/ctrl_output_fp.h b/source/source_io/module_ctrl/ctrl_output_fp.h index f8b92a60069..10c4c755620 100644 --- a/source/source_io/module_ctrl/ctrl_output_fp.h +++ b/source/source_io/module_ctrl/ctrl_output_fp.h @@ -1,9 +1,22 @@ #ifndef CTRL_OUTPUT_FP_H #define CTRL_OUTPUT_FP_H -#include "source_estate/elecstate_lcao.h" - struct Input_para; +class UnitCell; +class Charge; +class surchem; +class Parallel_Grid; + +namespace elecstate +{ +class ElecState; +} + +namespace ModulePW +{ +class PW_Basis; +class PW_Basis_Big; +} namespace ModuleIO { diff --git a/source/source_io/module_ctrl/ctrl_output_pw.cpp b/source/source_io/module_ctrl/ctrl_output_pw.cpp index 58c88461641..42f14ab6393 100644 --- a/source/source_io/module_ctrl/ctrl_output_pw.cpp +++ b/source/source_io/module_ctrl/ctrl_output_pw.cpp @@ -13,8 +13,8 @@ #include "source_lcao/module_deltaspin/deltaspin_pw_mi.h" #include "source_lcao/module_deltaspin/lambda_loop_helper.h" #include "source_lcao/module_deltaspin/spin_constrain.h" +#include "source_pw/module_proj/onsite_proj.h" // use projector #include "source_pw/module_pwdft/elecond.h" -#include "source_pw/module_pwdft/onsite_proj.h" // use projector #ifdef __MLALGO #include "../module_ml/write_mlkedf_desc.h" @@ -23,7 +23,7 @@ void ModuleIO::ctrl_iter_pw(const int istep, const int iter, const double& conv_esolver, - psi::Psi, base_device::DEVICE_CPU>* psi, + Setup_Psi_pw& stp, const K_Vectors& kv, const ModulePW::PW_Basis_K* pw_wfc, const Input_para& inp) @@ -57,8 +57,9 @@ void ModuleIO::ctrl_iter_pw(const int istep, out_wfc_flag = true; } - if (out_wfc_flag) + if (out_wfc_flag && (inp.out_wfc_pw == 1 || inp.out_wfc_pw == 2)) { + stp.sync_cpu(); ModuleIO::write_wfc_pw(istep_in, iter_in, GlobalV::KPAR, @@ -72,7 +73,7 @@ void ModuleIO::ctrl_iter_pw(const int istep, inp.out_wfc_pw, inp.ecutwfc, PARAM.globalv.global_out_dir, - psi[0], + stp.psi_cpu[0], kv, pw_wfc, GlobalV::ofs_running); @@ -100,8 +101,11 @@ void ModuleIO::ctrl_scf_pw(const int istep, ModuleBase::TITLE("ModuleIO", "ctrl_scf_pw"); ModuleBase::timer::start("ModuleIO", "ctrl_scf_pw"); - // Transfer data from device (GPU) to host (CPU) in pw basis - stp.copy_d2h(); + // Only the following postprocessors consume the double CPU mirror. + if (inp.calculation == "nscf" && (inp.towannier90 || (berryphase::berry_phase_flag && ModuleSymmetry::Symmetry::symm_flag != 1))) + { + stp.sync_cpu(); + } //---------------------------------------------------------- //! 4) Compute density of states (DOS) @@ -153,6 +157,16 @@ void ModuleIO::ctrl_scf_pw(const int istep, inp.dos_edelta_ev, inp.dos_scale, inp.dos_sigma, + inp.nspin, + inp.out_dos, + PARAM.globalv.dos_setemax, + inp.dos_emax_ev, + PARAM.globalv.dos_setemin, + inp.dos_emin_ev, + PARAM.globalv.two_fermi, + inp.out_app_flag, + inp.bndpar, + PARAM.globalv.global_out_dir, GlobalV::ofs_running); } } @@ -164,7 +178,8 @@ void ModuleIO::ctrl_scf_pw(const int istep, if (inp.out_pchg.size() > 0) { // Use the solver's native wavefunction precision for FFT and projector overlaps. - ModuleIO::Get_pchg_pw output(*stp.template get_psi_t(), *pw_wfc, *pw_rho, *pw_rhod, ppcell, inp.nspin, inp.nbands); + ModuleIO::Get_pchg_pw + output(*stp.template get_psi_t(), *pw_wfc, *pw_rho, *pw_rhod, ppcell, inp.nspin, inp.nbands); output.begin(&ucell, para_grid, kv, inp.out_pchg, PARAM.globalv.global_out_dir, inp.if_separate_k, inp.noncolin); } @@ -227,7 +242,8 @@ void ModuleIO::ctrl_scf_pw(const int istep, { // float type has not been implemented auto* onsite_p = projectors::OnsiteProjector::get_instance(); onsite_p->cal_occupations(reinterpret_cast, Device>*>(stp.template get_psi_t()), - pelec->wg); + pelec->wg, + inp.nspin); } ModuleBase::timer::end("ModuleIO", "ctrl_scf_pw"); @@ -265,7 +281,21 @@ void ModuleIO::ctrl_runner_pw(UnitCell& ucell, *stp.template get_psi_d(), ctx, para_grid, - ucell); + ucell, + inp.out_ldos, + inp.stm_bias, + inp.nspin, + PARAM.globalv.global_out_dir, + PARAM.globalv.two_fermi, + inp.nbands, + inp.dos_edelta_ev, + inp.dos_scale, + PARAM.globalv.dos_setemax, + inp.dos_emax_ev, + PARAM.globalv.dos_setemin, + inp.dos_emin_ev, + inp.dos_sigma, + inp.ldos_line); } //---------------------------------------------------------- @@ -277,6 +307,7 @@ void ModuleIO::ctrl_runner_pw(UnitCell& ucell, // ! Print out overlap matrices if (inp.out_spillage <= 2) { + stp.sync_cpu(); for (int i = 0; i < inp.bessel_nao_rcuts.size(); i++) { if (GlobalV::MY_RANK == 0) @@ -297,7 +328,8 @@ void ModuleIO::ctrl_runner_pw(UnitCell& ucell, { using Real = typename GetTypeReal::type; EleCond elec_cond(&ucell, &kv, pelec, pw_wfc, stp.template get_psi_t(), &ppcell); - elec_cond.KG(inp.cond_smear, inp.cond_fwhm, inp.cond_wcut, inp.cond_dw, inp.cond_dt, inp.cond_nonlocal, inp.cond_mgga_vel, pelec->wg); + elec_cond + .KG(inp.cond_smear, inp.cond_fwhm, inp.cond_wcut, inp.cond_dw, inp.cond_dt, inp.cond_nonlocal, inp.cond_mgga_vel, pelec->wg); } #ifdef __MLALGO diff --git a/source/source_io/module_ctrl/ctrl_output_pw.h b/source/source_io/module_ctrl/ctrl_output_pw.h index ecfa96abb32..62518166a11 100644 --- a/source/source_io/module_ctrl/ctrl_output_pw.h +++ b/source/source_io/module_ctrl/ctrl_output_pw.h @@ -1,9 +1,9 @@ -#ifndef CTRL_OUTPUT_PW_H -#define CTRL_OUTPUT_PW_H +#ifndef CTRL_OUTPUT_PW_H +#define CTRL_OUTPUT_PW_H #include "source_base/module_device/device.h" // use Device -#include "source_psi/psi.h" // define psi #include "source_estate/elecstate_lcao.h" // use pelec +#include "source_psi/psi.h" // define psi #include "source_psi/setup_psi_pw.h" // use Setup_Psi class class pseudopot_cell_vnl; @@ -12,62 +12,62 @@ namespace ModuleIO { // print out information in 'iter_finish' in ESolver_KS_PW -void ctrl_iter_pw(const int istep, - const int iter, - const double &conv_esolver, - psi::Psi, base_device::DEVICE_CPU>* psi, - const K_Vectors &kv, - const ModulePW::PW_Basis_K *pw_wfc, - const Input_para& inp); +void ctrl_iter_pw(const int istep, + const int iter, + const double& conv_esolver, + Setup_Psi_pw& stp, + const K_Vectors& kv, + const ModulePW::PW_Basis_K* pw_wfc, + const Input_para& inp); // print out information in 'after_scf' in ESolver_KS_PW template void ctrl_scf_pw(const int istep, - UnitCell& ucell, - elecstate::ElecState* pelec, - const Charge &chr, - const pseudopot_cell_vnl& ppcell, - const K_Vectors &kv, - const ModulePW::PW_Basis_K *pw_wfc, - const ModulePW::PW_Basis *pw_rho, - const ModulePW::PW_Basis *pw_rhod, - const ModulePW::PW_Basis_Big *pw_big, - Setup_Psi_pw &stp, - const Parallel_Grid ¶_grid, - const Input_para& inp); + UnitCell& ucell, + elecstate::ElecState* pelec, + const Charge& chr, + const pseudopot_cell_vnl& ppcell, + const K_Vectors& kv, + const ModulePW::PW_Basis_K* pw_wfc, + const ModulePW::PW_Basis* pw_rho, + const ModulePW::PW_Basis* pw_rhod, + const ModulePW::PW_Basis_Big* pw_big, + Setup_Psi_pw& stp, + const Parallel_Grid& para_grid, + const Input_para& inp); // print out information in 'after_all_runners' in ESolver_KS_PW template -void ctrl_runner_pw(UnitCell& ucell, - elecstate::ElecState* pelec, - ModulePW::PW_Basis_K* pw_wfc, - ModulePW::PW_Basis* pw_rho, - ModulePW::PW_Basis* pw_rhod, - Charge &chr, - K_Vectors &kv, - Setup_Psi_pw &stp, - Structure_Factor &sf, - pseudopot_cell_vnl &ppcell, - surchem &solvent, - Parallel_Grid ¶_grid, - const Input_para& inp); +void ctrl_runner_pw(UnitCell& ucell, + elecstate::ElecState* pelec, + ModulePW::PW_Basis_K* pw_wfc, + ModulePW::PW_Basis* pw_rho, + ModulePW::PW_Basis* pw_rhod, + Charge& chr, + K_Vectors& kv, + Setup_Psi_pw& stp, + Structure_Factor& sf, + pseudopot_cell_vnl& ppcell, + surchem& solvent, + Parallel_Grid& para_grid, + const Input_para& inp); // print out information in 'after_all_runners' in ESolver_KS_PW (runtime version) template -void ctrl_runner_pw(UnitCell& ucell, - elecstate::ElecState* pelec, - ModulePW::PW_Basis_K* pw_wfc, - ModulePW::PW_Basis* pw_rho, - ModulePW::PW_Basis* pw_rhod, - Charge &chr, - K_Vectors &kv, - Setup_Psi_pw &stp, - Structure_Factor &sf, - pseudopot_cell_vnl &ppcell, - surchem &solvent, - const base_device::DeviceContext* ctx, - Parallel_Grid ¶_grid, - const Input_para& inp); +void ctrl_runner_pw(UnitCell& ucell, + elecstate::ElecState* pelec, + ModulePW::PW_Basis_K* pw_wfc, + ModulePW::PW_Basis* pw_rho, + ModulePW::PW_Basis* pw_rhod, + Charge& chr, + K_Vectors& kv, + Setup_Psi_pw& stp, + Structure_Factor& sf, + pseudopot_cell_vnl& ppcell, + surchem& solvent, + const base_device::DeviceContext* ctx, + Parallel_Grid& para_grid, + const Input_para& inp); -} +} // namespace ModuleIO #endif diff --git a/source/source_io/module_ctrl/ctrl_output_td.cpp b/source/source_io/module_ctrl/ctrl_output_td.cpp index d4308630fc0..78ca6fe34a9 100644 --- a/source/source_io/module_ctrl/ctrl_output_td.cpp +++ b/source/source_io/module_ctrl/ctrl_output_td.cpp @@ -1,8 +1,8 @@ #include "ctrl_output_td.h" #include "source_base/parallel_global.h" -#include "source_io/module_parameter/parameter.h" #include "source_io/module_current/td_current_io.h" +#include "source_io/module_parameter/parameter.h" namespace ModuleIO { @@ -24,8 +24,7 @@ void ctrl_output_td(const UnitCell& ucell, Record_adj& RA, TD_info* td_p, const Exx_NAO>& exx_nao, - const Exx_Info& exx_info - ) + const Exx_Info& exx_info) { ModuleBase::TITLE("ModuleIO", "ctrl_output_td"); @@ -50,26 +49,10 @@ void ctrl_output_td(const UnitCell& ucell, ModuleIO::write_current(ucell, istep, psi, pelec, kv, intor, pv, orb, velocity_mat, td_p, RA); } } - else if(TD_info::out_current==2) + else if (TD_info::out_current == 2) { ModuleIO::write_current(ucell, grid, istep, psi, pelec, kv, pv, orb, td_p, p_hamilt->getSR(), p_hamilt->getHR(), exx_nao, exx_info); } - // (3) Output file for restart - if (PARAM.inp.out_freq_td > 0) // default value of out_freq_td is 0 - { - if (istep % PARAM.inp.out_freq_td == 0) - { - if (td_p != nullptr) - { - td_p->out_restart_info(istep, elecstate::H_TDDFT_pw::At, elecstate::H_TDDFT_pw::At_laststep); - } - else - { - ModuleBase::WARNING_QUIT("ModuleIO::ctrl_output_td", - "TD_info pointer is null, cannot output restart info."); - } - } - } #endif // __LCAO } @@ -89,8 +72,7 @@ template void ctrl_output_td(const UnitCell&, Record_adj&, TD_info*, const Exx_NAO>&, - const Exx_Info& - ); + const Exx_Info&); template void ctrl_output_td>(const UnitCell&, double**, @@ -108,7 +90,6 @@ template void ctrl_output_td>(const UnitCell&, Record_adj&, TD_info*, const Exx_NAO>&, - const Exx_Info& - ); + const Exx_Info&); -} // namespace ModuleIO \ No newline at end of file +} // namespace ModuleIO diff --git a/source/source_io/module_ctrl/ctrl_output_td.h b/source/source_io/module_ctrl/ctrl_output_td.h index 6a20c43c93d..16fcea0d0de 100644 --- a/source/source_io/module_ctrl/ctrl_output_td.h +++ b/source/source_io/module_ctrl/ctrl_output_td.h @@ -8,7 +8,7 @@ #include "source_estate/module_pot/h_tddft_pw.h" #include "source_lcao/module_rt/td_info.h" #include "source_lcao/module_rt/velocity_op.h" -#include "source_lcao/record_adj.h" +#include "source_cell/record_adj.h" #include "source_psi/psi.h" #include "source_lcao/hamilt_lcao.h" #include "source_lcao/setup_exx.h" diff --git a/source/source_io/module_ctrl/ctrl_runner_lcao.cpp b/source/source_io/module_ctrl/ctrl_runner_lcao.cpp index 2bbd0b0a84c..4966af3924b 100644 --- a/source/source_io/module_ctrl/ctrl_runner_lcao.cpp +++ b/source/source_io/module_ctrl/ctrl_runner_lcao.cpp @@ -6,9 +6,10 @@ #include "../module_energy/write_proj_band_lcao.h" // projcted band structure #include "../module_dos/cal_ldos.h" // cal LDOS -#include "../module_energy/write_eband_terms.hpp" -#include "../module_hs/write_vxc.hpp" -#include "../module_hs/write_vxc_r.hpp" +#include "../module_energy/write_eband_terms.h" +#include "source_io/module_hs/vxc_op_mat.h" +#include "source_io/module_hs/vxc_op_r.h" +#include "source_io/module_hs/vxc_op_tools.h" #ifdef __EXX #include "source_lcao/module_ri/exx_lri_interface.h" #endif @@ -21,7 +22,7 @@ void ctrl_runner_lcao(UnitCell& ucell, // unitcell const Input_para &inp, // input K_Vectors &kv, // k-point elecstate::ElecState* pelec,// electronic info - const LCAO_domain::Setup_DM &dmat, // mohan add 2025-11-02 + const module_dm::Setup_DM &dmat, // mohan add 2025-11-02 Parallel_Orbitals &pv, // orbital info Parallel_Grid &pgrid, // grid info Grid_Driver &gd, // search for adjacent atoms @@ -50,14 +51,20 @@ void ctrl_runner_lcao(UnitCell& ucell, // unitcell // 2) out ldos if (inp.out_ldos[0]) { - ModuleIO::Cal_ldos::cal_ldos_lcao(pelec->eferm, chr, dmat, kv, - pelec->ekb, pelec->wg, psi[0], pgrid, gd, ucell); + ModuleIO::Cal_ldos::cal_ldos_lcao(pelec->eferm, chr, dmat, kv, + pelec->ekb, pelec->wg, psi[0], pgrid, pv, gd, ucell, + inp.stm_bias, inp.nspin, PARAM.globalv.global_out_dir, + PARAM.globalv.two_fermi, inp.out_ldos[1]); } // 3) print out exchange-correlation potential if (inp.out_mat_xc) { bool cal_exx = exx_info.info_global.cal_exx; +#ifdef __EXX + auto* hexxd_ptr = exx_nao.exd ? &exx_nao.exd->get_Hexxs() : nullptr; + auto* hexxc_ptr = exx_nao.exc ? &exx_nao.exc->get_Hexxs() : nullptr; +#endif ModuleIO::write_Vxc(inp.nspin, PARAM.globalv.nlocal, GlobalV::DRANK, @@ -74,12 +81,17 @@ void ctrl_runner_lcao(UnitCell& ucell, // unitcell orb.cutoffs(), pelec->wg, gd, + inp.dft_plus_u, + PARAM.globalv.gamma_only_local, + PARAM.globalv.global_out_dir, + inp.out_ndigits, + inp.ks_solver, cal_exx, exx_info #ifdef __EXX , - exx_nao.exd ? &exx_nao.exd->get_Hexxs() : nullptr, - exx_nao.exc ? &exx_nao.exc->get_Hexxs() : nullptr + hexxd_ptr, + hexxc_ptr #endif ); } @@ -89,6 +101,11 @@ void ctrl_runner_lcao(UnitCell& ucell, // unitcell bool cal_exx = exx_info.info_global.cal_exx; double hybrid_alpha = exx_info.info_global.hybrid_alpha; bool real_number = exx_info.info_ri.real_number; + const double sparse_thr = 1e-10; +#ifdef __EXX + const auto* hexxd_ptr = exx_nao.exd ? &exx_nao.exd->get_Hexxs() : nullptr; + const auto* hexxc_ptr = exx_nao.exc ? &exx_nao.exc->get_Hexxs() : nullptr; +#endif ModuleIO::write_Vxc_R(inp.nspin, &pv, ucell, @@ -101,14 +118,17 @@ void ctrl_runner_lcao(UnitCell& ucell, // unitcell kv, orb.cutoffs(), gd, + PARAM.globalv.global_out_dir, cal_exx, hybrid_alpha, real_number #ifdef __EXX , - exx_nao.exd ? &exx_nao.exd->get_Hexxs() : nullptr, - exx_nao.exc ? &exx_nao.exc->get_Hexxs() : nullptr + hexxd_ptr, + hexxc_ptr #endif + , + sparse_thr ); } @@ -153,7 +173,7 @@ template void ctrl_runner_lcao(UnitCell& ucell, // unitcell const Input_para &inp, // input K_Vectors &kv, // k-point elecstate::ElecState* pelec,// electronic info - const LCAO_domain::Setup_DM &dmat, // mohan add 2025-11-02 + const module_dm::Setup_DM &dmat, // mohan add 2025-11-02 Parallel_Orbitals &pv, // orbital info Parallel_Grid &pgrid, // grid info Grid_Driver &gd, // search for adjacent atoms @@ -175,7 +195,7 @@ template void ctrl_runner_lcao, double>(UnitCell& ucell, const Input_para &inp, // input K_Vectors &kv, // k-point elecstate::ElecState* pelec,// electronic info - const LCAO_domain::Setup_DM> &dmat, // mohan add 2025-11-02 + const module_dm::Setup_DM> &dmat, // mohan add 2025-11-02 Parallel_Orbitals &pv, // orbital info Parallel_Grid &pgrid, // grid info Grid_Driver &gd, // search for adjacent atoms @@ -197,7 +217,7 @@ template void ctrl_runner_lcao, std::complex>(UnitC const Input_para &inp, // input K_Vectors &kv, // k-point elecstate::ElecState* pelec,// electronic info - const LCAO_domain::Setup_DM> &dmat, // mohan add 2025-11-02 + const module_dm::Setup_DM> &dmat, // mohan add 2025-11-02 Parallel_Orbitals &pv, // orbital info Parallel_Grid &pgrid, // grid info Grid_Driver &gd, // search for adjacent atoms diff --git a/source/source_io/module_ctrl/ctrl_runner_lcao.h b/source/source_io/module_ctrl/ctrl_runner_lcao.h index fcf790214e4..6f51f9ae273 100644 --- a/source/source_io/module_ctrl/ctrl_runner_lcao.h +++ b/source/source_io/module_ctrl/ctrl_runner_lcao.h @@ -8,7 +8,7 @@ #include "source_lcao/hamilt_lcao.h" // use hamilt::HamiltLCAO #include "source_basis/module_nao/two_center_bundle.h" // use TwoCenterBundle #include "source_lcao/setup_exx.h" // for exx, mohan add 20251018 -#include "source_lcao/setup_dm.h" // for density matrix, mohan add 20251103 +#include "source_lcao/allocate_dm.h" // for density matrix, mohan add 20251103 namespace ModuleIO { @@ -18,7 +18,7 @@ void ctrl_runner_lcao(UnitCell& ucell, // unitcell const Input_para &inp, // input K_Vectors &kv, // k-point elecstate::ElecState* pelec,// electronic info - const LCAO_domain::Setup_DM &dmat, // mohan add 2025-11-02 + const module_dm::Setup_DM &dmat, // mohan add 2025-11-02 Parallel_Orbitals &pv, // orbital info Parallel_Grid &pgrid, // grid info Grid_Driver &gd, // search for adjacent atoms diff --git a/source/source_io/module_ctrl/ctrl_scf_lcao.cpp b/source/source_io/module_ctrl/ctrl_scf_lcao.cpp index 51bfe945fc6..e3f6bd6dd1f 100644 --- a/source/source_io/module_ctrl/ctrl_scf_lcao.cpp +++ b/source/source_io/module_ctrl/ctrl_scf_lcao.cpp @@ -2,6 +2,7 @@ #include "source_hamilt/module_xc/exx_info.h" #include "source_base/formatter.h" +#include "source_base/module_out/filename.h" #include "source_base/tool_quit.h" // use ModuleBase::WARNING_QUIT #include "source_estate/elecstate_lcao.h" // use elecstate::ElecState #include "source_hamilt/hamilt.h" // use Hamilt @@ -11,16 +12,17 @@ // functions #include "../module_unk/berryphase.h" // use berryphase -#include "../module_hs/cal_plpr.h" // use AngularMomentumCalculator() +#include "../module_hs/angmom_op_mat.h" // use Angmom_op() #include "source_io/module_hs/output_mat_sparse.h" // use ModuleIO::output_mat_sparse() #include "source_io/module_ml/io_npz.h" // use ModuleIO::output_mat_npz() #include "source_io/module_dhs/write_dh.h" // use ModuleIO::write_dH_components() -#include "source_io/module_hs/write_h_terms.h" // use ModuleIO::write_h_* -#include "../module_hs/write_hs_r.h" // use ModuleIO::write_hsr() +#include "source_io/module_hs/hterm_writer.h" // use ModuleIO::write_h_* +#include "../module_hs/hsr_writer.h" // use ModuleIO::write_hsr() #include "../module_mulliken/cal_mag.h" // use cal_mag() #include "../module_wannier/to_w90_lcao.h" // use toW90_LCAO #include "../module_wannier/to_w90_lcao_pw.h" // use toW90_LCAO_IN_PW -#include "../module_hs/write_hs.h" // use ModuleIO::write_hsk() +#include "../module_hs/hs_dense_io.h" // use ModuleIO::save_mat() +#include "../module_hs/hsk_writer.h" // use ModuleIO::write_hsk() #include "../module_dm/write_dmk.h" // use ModuleIO::write_dmk() #include "../module_dm/write_dmr.h" // use ModuleIO::write_dmr() #include "../module_dos/write_dos_lcao.h" // use ModuleIO::write_dos_lcao() @@ -42,20 +44,6 @@ #include "source_lcao/module_operator_lcao/overlap.h" // use hamilt::Overlap for NAMD #ifdef __EXX -template -void setup_exx_dh_params(ModuleIO::WriteDHParams& dh_params, Exx_NAO& exx_nao, const Exx_Info& exx_info) -{} - -template <> -void setup_exx_dh_params(ModuleIO::WriteDHParams& dh_params, Exx_NAO& exx_nao, const Exx_Info& exx_info) -{ - if (exx_info.info_global.cal_exx) - { - if (exx_nao.exd) { dh_params.exd = exx_nao.exd.get(); } - if (exx_nao.exc) { dh_params.exc = exx_nao.exc.get(); } - } -} - template void setup_exx_h_params(ModuleIO::WriteHParams& h_params, Exx_NAO& exx_nao, const Exx_Info& exx_info) { @@ -85,7 +73,7 @@ void ModuleIO::ctrl_scf_lcao(UnitCell& ucell, const Input_para& inp, K_Vectors& kv, elecstate::ElecState* pelec, - elecstate::DensityMatrix* dm, // mohan add 2025-11-04 + module_dm::DensityMatrix* dm, // mohan add 2025-11-04 Parallel_Orbitals& pv, Grid_Driver& gd, psi::Psi* psi, @@ -170,6 +158,15 @@ void ModuleIO::ctrl_scf_lcao(UnitCell& ucell, inp.dos_sigma, out_app_flag, istep, + inp.nspin, + PARAM.globalv.nlocal, + PARAM.globalv.dos_setemax, + inp.dos_emax_ev, + PARAM.globalv.dos_setemin, + inp.dos_emin_ev, + PARAM.globalv.two_fermi, + inp.bndpar, + PARAM.globalv.global_out_dir, GlobalV::ofs_running); } @@ -180,7 +177,7 @@ void ModuleIO::ctrl_scf_lcao(UnitCell& ucell, { const int precision = inp.out_dmr[1]; - ModuleIO::write_dmr(dm->get_DMR_vector(), &ucell, precision, pv, out_app_flag, + ModuleIO::write_dmr(dm->get_dmr_vec(), &ucell, precision, pv, out_app_flag, ucell.get_iat2iwt(), ucell.nat, istep); } @@ -196,7 +193,7 @@ void ModuleIO::ctrl_scf_lcao(UnitCell& ucell, } const int precision = inp.out_dmk[1]; - ModuleIO::write_dmk(dm->get_DMK_vector(), kv, precision, efermis, &(ucell), pv, istep); + ModuleIO::write_dmk(dm->get_dmk_vec(), kv, precision, efermis, &(ucell), pv, global_out_dir, istep); } //------------------------------------------------------------------ @@ -219,6 +216,9 @@ void ModuleIO::ctrl_scf_lcao(UnitCell& ucell, istep, hsk_out_type, precision, + PARAM.globalv.nlocal, + PARAM.inp.ks_solver, + GlobalV::DRANK, GlobalV::ofs_running); } @@ -277,7 +277,8 @@ void ModuleIO::ctrl_scf_lcao(UnitCell& ucell, const hamilt::HContainer* sr = p_hamilt->getSR(); ModuleIO::write_hsr(hr_vec, sr, &ucell, inp.out_hsr[0], precision, pv, - out_app_flag, gamma_only, ucell.get_iat2iwt(), ucell.nat, istep); + out_app_flag, gamma_only, ucell.get_iat2iwt(), ucell.nat, istep, + PARAM.globalv.global_out_dir); } //------------------------------------------------------------------ @@ -303,7 +304,7 @@ void ModuleIO::ctrl_scf_lcao(UnitCell& ucell, if (inp.out_dm_npz) { - const std::vector*>& dmr_vec = dm->get_DMR_vector(); + const std::vector*>& dmr_vec = dm->get_dmr_vec(); for (int ispin = 0; ispin < dmr_vec.size(); ++ispin) { std::string zipname @@ -329,7 +330,7 @@ void ModuleIO::ctrl_scf_lcao(UnitCell& ucell, if(!PARAM.globalv.gamma_only_local) ModuleIO::output_mat_sparse(mat_sparse_options, - istep, + istep_in, pelec->pot->get_eff_v(), pv, two_center_bundle, @@ -401,14 +402,17 @@ void ModuleIO::ctrl_scf_lcao(UnitCell& ucell, // The Veff Hellmann-Feynman terms need these (V^H sums spins, V^XC is spin-resolved). for (int is = 1; is <= PARAM.inp.nspin; ++is) { - dh_params.dmR.push_back(dm->get_DMR_pointer(is)); + dh_params.dmR.push_back(dm->get_dmr_ptr(is)); } } -#ifdef __EXX - // dV^EXX/dR output is wired for the gamma (TK==double) exx interfaces. exd/exc are - // mutually exclusive (real vs complex Hexx); write_dH_exx picks by info_ri.real_number. - setup_exx_dh_params(dh_params, exx_nao, exx_info); -#endif + // FIXME (known bug, not addressed in this PR): with MPI ranks > 1 the + // Hellmann-Feynman path inside hamilt::Veff::cal_dH + // (source_lcao/module_operator_lcao/veff_dh.cpp) may deadlock. + // The vl branch issues collectives (PW FFT via Forces::cal_force_loc) + // per orbital pair; until that path is made rank-lockstep, multi-rank + // out_mat_dh runs can hang. Tracked as a follow-up; do NOT add a + // rank-count guard here -- the integration tests (e.g. + // tests/02_NAO_Gamma/scf_out_dh) exercise this path with np > 1. ModuleIO::write_dH_components(dh_params, exx_info); delete pot_vl; delete pot_vh; @@ -436,6 +440,18 @@ void ModuleIO::ctrl_scf_lcao(UnitCell& ucell, h_params.append = out_app_flag; h_params.iat2iwt = ucell.get_iat2iwt(); h_params.nat = ucell.nat; + h_params.nlocal = PARAM.globalv.nlocal; + h_params.gamma_only_local = gamma_only; + h_params.npol = PARAM.globalv.npol; + h_params.domag = PARAM.globalv.domag; + h_params.domag_z = PARAM.globalv.domag_z; + h_params.gga_grad = PARAM.inp.gga_grad; + h_params.out_app_flag = out_app_flag; + h_params.calculation = inp.calculation; + h_params.global_out_dir = global_out_dir; + h_params.global_matrix_dir = PARAM.globalv.global_matrix_dir; + h_params.ks_solver = PARAM.inp.ks_solver; + h_params.drank = GlobalV::DRANK; if (inp.out_mat_h_t[0]) { ModuleIO::write_h_t(h_params); @@ -508,7 +524,8 @@ void ModuleIO::ctrl_scf_lcao(UnitCell& ucell, inp.out_app_flag, t_fn, pv, - GlobalV::DRANK); + GlobalV::DRANK, + PARAM.inp.ks_solver); } delete ekinetic; @@ -519,16 +536,18 @@ void ModuleIO::ctrl_scf_lcao(UnitCell& ucell, //------------------------------------------------------------------ if (inp.out_mat_l[0]) { - ModuleIO::AngularMomentumCalculator mylcalculator(inp.orbital_dir, + ModuleIO::Angmom_op mylcalculator(inp.orbital_dir, ucell, orb.get_rcutmax_Phi(), inp.test_deconstructor, inp.test_grid, inp.test_atom_input, PARAM.globalv.search_pbc, + PARAM.inp.out_level, + PARAM.globalv.gamma_only_local, &GlobalV::ofs_running, GlobalV::MY_RANK); - mylcalculator.calculate(inp.suffix, global_out_dir, ucell, inp.out_mat_l[1], GlobalV::MY_RANK); + mylcalculator.calculate(inp.suffix, global_out_dir, ucell, inp.out_mat_l[1], GlobalV::MY_RANK, istep_in); } //------------------------------------------------------------------ @@ -621,7 +640,7 @@ void ModuleIO::ctrl_scf_lcao(UnitCell& ucell, // mohan add 2025-10-24 // if (inp.out_elf[0] > 0) // { - // LCAO_domain::dm2tau(pelec->DM->get_DMR_vector(), inp.nspin, pelec->charge); + // LCAO_domain::dm2tau(pelec->DM->get_dmr_vec(), inp.nspin, pelec->charge); // } #ifdef __EXX @@ -747,7 +766,7 @@ template void ModuleIO::ctrl_scf_lcao( const Input_para& inp, K_Vectors& kv, elecstate::ElecState* pelec, - elecstate::DensityMatrix* dm, // mohan add 2025-11-04 + module_dm::DensityMatrix* dm, // mohan add 2025-11-04 Parallel_Orbitals& pv, Grid_Driver& gd, psi::Psi* psi, @@ -776,7 +795,7 @@ template void ModuleIO::ctrl_scf_lcao, double>( const Input_para& inp, K_Vectors& kv, elecstate::ElecState* pelec, - elecstate::DensityMatrix, double>* dm, // mohan add 2025-11-04 + module_dm::DensityMatrix, double>* dm, // mohan add 2025-11-04 Parallel_Orbitals& pv, Grid_Driver& gd, psi::Psi>* psi, @@ -804,7 +823,7 @@ template void ModuleIO::ctrl_scf_lcao, std::complex const Input_para& inp, K_Vectors& kv, elecstate::ElecState* pelec, - elecstate::DensityMatrix, double>* dm, // mohan add 2025-11-04 + module_dm::DensityMatrix, double>* dm, // mohan add 2025-11-04 Parallel_Orbitals& pv, Grid_Driver& gd, psi::Psi>* psi, diff --git a/source/source_io/module_ctrl/ctrl_scf_lcao.h b/source/source_io/module_ctrl/ctrl_scf_lcao.h index 5d359d3fe28..87f1aa9dd53 100644 --- a/source/source_io/module_ctrl/ctrl_scf_lcao.h +++ b/source/source_io/module_ctrl/ctrl_scf_lcao.h @@ -26,7 +26,7 @@ void ctrl_scf_lcao(UnitCell& ucell, const Input_para& inp, K_Vectors& kv, elecstate::ElecState* pelec, - elecstate::DensityMatrix* dm, // mohan add 2025-11-04 + module_dm::DensityMatrix* dm, // mohan add 2025-11-04 Parallel_Orbitals& pv, Grid_Driver& gd, psi::Psi* psi, diff --git a/source/source_io/module_current/td_current_io.cpp b/source/source_io/module_current/td_current_io.cpp index d84dcd03ed0..efff1c97e8c 100644 --- a/source/source_io/module_current/td_current_io.cpp +++ b/source/source_io/module_current/td_current_io.cpp @@ -7,10 +7,10 @@ #include "source_base/timer.h" #include "source_base/tool_threading.h" #include "source_base/vector3.h" -#include "source_estate/module_dm/cal_dm_psi.h" +#include "source_estate/module_dm/dm_from_psi.h" #include "source_estate/module_pot/h_tddft_pw.h" -#include "source_lcao/lcao_domain.h" #include "source_io/module_parameter/parameter.h" +#include "source_lcao/lcao_domain.h" #ifdef __LCAO template @@ -30,7 +30,7 @@ void ModuleIO::write_current(const UnitCell& ucell, ModuleBase::TITLE("ModuleIO", "write_current"); ModuleBase::timer::start("ModuleIO", "write_current"); std::vector>*> current_term = {nullptr, nullptr, nullptr}; - if (PARAM.inp.td_stype!=1) + if (PARAM.inp.td_stype != 1) { for (int dir = 0; dir < 3; dir++) { @@ -48,29 +48,29 @@ void ModuleIO::write_current(const UnitCell& ucell, current_term[dir] = TD_info::td_vel_op->get_current_term_pointer(dir); } } - double omega=ucell.omega; + double omega = ucell.omega; // construct a DensityMatrix object - // Since the function cal_dm_psi do not suport DMR in complex type, I replace it with two DMR in double type. Should + // Since the function dm_from_psi do not suport DMR in complex type, I replace it with two DMR in double type. Should // be refactored in the future. const int nspin0 = PARAM.inp.nspin; - const int nspin_dm = std::map({ {1,1},{2,2},{4,1} })[nspin0]; - elecstate::DensityMatrix, std::complex> tmp_dm(pv, nspin_dm, kv.kvec_d, kv.get_nks() / nspin_dm); + const int nspin_dm = std::map({{1, 1}, {2, 2}, {4, 1}})[nspin0]; + module_dm::DensityMatrix, std::complex> tmp_dm(pv, nspin_dm, kv.kvec_d, kv.get_nks() / nspin_dm); // calculate DMK - elecstate::cal_dm_psi(pv, pelec->wg, psi[0], tmp_dm); + module_dm::dm_from_psi(pv, pelec->wg, psi[0], tmp_dm); // init DMR - tmp_dm.init_DMR(ra, &ucell); + tmp_dm.init_dmr(ra, &ucell); - if(PARAM.inp.td_stype!=2) + if (PARAM.inp.td_stype != 2) { - tmp_dm.cal_DMR(); + tmp_dm.cal_dmr(-1); } else { - tmp_dm.cal_DMR_td(td_p->get_phase_hybrid(),TD_info::cart_At); + tmp_dm.cal_dmr_td(td_p->get_phase_hybrid(), TD_info::A_prop_ha, -1); } - //DM_real.sum_DMR_spin(); - //DM_imag.sum_DMR_spin(); + // DM_real.sum_DMR_spin(); + // DM_imag.sum_DMR_spin(); double current_total[3] = {0.0, 0.0, 0.0}; #ifdef _OPENMP @@ -78,8 +78,8 @@ void ModuleIO::write_current(const UnitCell& ucell, { double local_current[3] = {0.0, 0.0, 0.0}; #else - // ModuleBase::matrix& local_soverlap = soverlap; - double* local_current = current_total; + // ModuleBase::matrix& local_soverlap = soverlap; + double* local_current = current_total; #endif ModuleBase::Vector3 tau1, dtau, tau2; @@ -96,8 +96,8 @@ void ModuleIO::write_current(const UnitCell& ucell, const int start1 = ucell.itiaiw2iwt(T1, I1, 0); for (int cb = 0; cb < ra.na_each[iat]; ++cb) { - const int T2 = ra.info[iat][cb][3]; - const int I2 = ra.info[iat][cb][4]; + const int T2 = ra.get_info(iat, cb)[3]; + const int I2 = ra.get_info(iat, cb)[4]; const int start2 = ucell.itiaiw2iwt(T2, I2, 0); @@ -105,20 +105,16 @@ void ModuleIO::write_current(const UnitCell& ucell, // get iat2 int iat2 = ucell.itia2iat(T2, I2); - double Rx = ra.info[iat][cb][0]; - double Ry = ra.info[iat][cb][1]; - double Rz = ra.info[iat][cb][2]; - //std::cout<< "iat1: " << iat1 << " iat2: " << iat2 << " Rx: " << Rx << " Ry: " << Ry << " Rz:" << Rz << std::endl; - // get BaseMatrix - hamilt::BaseMatrix>* tmp_matrix - = tmp_dm.get_DMR_pointer(1)->find_matrix(iat1, iat2, Rx, Ry, Rz); + double Rx = ra.get_info(iat, cb)[0]; + double Ry = ra.get_info(iat, cb)[1]; + double Rz = ra.get_info(iat, cb)[2]; + // std::cout<< "iat1: " << iat1 << " iat2: " << iat2 << " Rx: " << Rx << " Ry: " << Ry << " Rz:" << Rz << std::endl; + // get BaseMatrix + hamilt::BaseMatrix>* tmp_matrix = tmp_dm.get_dmr_ptr(1)->find_matrix(iat1, iat2, Rx, Ry, Rz); // refactor - hamilt::BaseMatrix>* tmp_m_rvx - = current_term[0]->find_matrix(iat1, iat2, Rx, Ry, Rz); - hamilt::BaseMatrix>* tmp_m_rvy - = current_term[1]->find_matrix(iat1, iat2, Rx, Ry, Rz); - hamilt::BaseMatrix>* tmp_m_rvz - = current_term[2]->find_matrix(iat1, iat2, Rx, Ry, Rz); + hamilt::BaseMatrix>* tmp_m_rvx = current_term[0]->find_matrix(iat1, iat2, Rx, Ry, Rz); + hamilt::BaseMatrix>* tmp_m_rvy = current_term[1]->find_matrix(iat1, iat2, Rx, Ry, Rz); + hamilt::BaseMatrix>* tmp_m_rvz = current_term[2]->find_matrix(iat1, iat2, Rx, Ry, Rz); if (tmp_matrix == nullptr) { continue; @@ -142,10 +138,10 @@ void ModuleIO::write_current(const UnitCell& ucell, rvy = tmp_m_rvy->get_value(mu, nu); rvz = tmp_m_rvz->get_value(mu, nu); } - //std::cout<<"mu: "<< mu <<" nu: "<< nu << std::endl; - // std::cout<<"dm2d1_real: "<< dm2d1_real << " dm2d1_imag: "<< dm2d1_imag << std::endl; - //std::cout<<"rvz: "<< rvz.real() << " " << rvz.imag() << std::endl; - local_current[0] -= dm2d1.real() * rvx.real() - dm2d1.imag() * rvx.imag(); + // std::cout<<"mu: "<< mu <<" nu: "<< nu << std::endl; + // std::cout<<"dm2d1_real: "<< dm2d1_real << " dm2d1_imag: "<< dm2d1_imag << std::endl; + // std::cout<<"rvz: "<< rvz.real() << " " << rvz.imag() << std::endl; + local_current[0] -= dm2d1.real() * rvx.real() - dm2d1.imag() * rvx.imag(); local_current[1] -= dm2d1.real() * rvy.real() - dm2d1.imag() * rvy.imag(); local_current[2] -= dm2d1.real() * rvz.real() - dm2d1.imag() * rvz.imag(); } // end kk @@ -171,9 +167,8 @@ void ModuleIO::write_current(const UnitCell& ucell, fout.open(filename, std::ios::app); fout << std::setprecision(16); fout << std::scientific; - fout << istep+1 << " " << current_total[0]/omega - << " " << current_total[1]/omega - << " " << current_total[2]/omega << std::endl; + fout << istep + 1 << " " << current_total[0] / omega << " " << current_total[1] / omega << " " << current_total[2] / omega + << std::endl; fout.close(); } @@ -182,16 +177,16 @@ void ModuleIO::write_current(const UnitCell& ucell, } template void ModuleIO::write_current_eachk(const UnitCell& ucell, - const int istep, - const psi::Psi>* psi, - const elecstate::ElecState* pelec, - const K_Vectors& kv, - const TwoCenterIntegrator* intor, - const Parallel_Orbitals* pv, - const LCAO_Orbitals& orb, - const Velocity_op* cal_current, - TD_info* td_p, - Record_adj& ra) + const int istep, + const psi::Psi>* psi, + const elecstate::ElecState* pelec, + const K_Vectors& kv, + const TwoCenterIntegrator* intor, + const Parallel_Orbitals* pv, + const LCAO_Orbitals& orb, + const Velocity_op* cal_current, + TD_info* td_p, + Record_adj& ra) { ModuleBase::TITLE("ModuleIO", "write_current"); @@ -215,38 +210,38 @@ void ModuleIO::write_current_eachk(const UnitCell& ucell, current_term[dir] = TD_info::td_vel_op->get_current_term_pointer(dir); } } - double omega=ucell.omega; + double omega = ucell.omega; // construct a DensityMatrix object - // Since the function cal_dm_psi do not suport DMR in complex type, + // Since the function dm_from_psi do not suport DMR in complex type, // I replace it with two DMR in double type. // Should be refactored in the future. const int nspin0 = PARAM.inp.nspin; - const int nspin_dm = std::map({ {1,1},{2,2},{4,1} })[nspin0]; - elecstate::DensityMatrix, std::complex> tmp_dm(pv, nspin_dm, kv.kvec_d, kv.get_nks() / nspin_dm); - //elecstate::DensityMatrix, double> DM_real(pv, nspin_dm, kv.kvec_d, kv.get_nks() / nspin_dm); - //elecstate::DensityMatrix, double> DM_imag(pv, nspin_dm, kv.kvec_d, kv.get_nks() / nspin_dm); - // calculate DMK - elecstate::cal_dm_psi(pv, pelec->wg, psi[0], tmp_dm); + const int nspin_dm = std::map({{1, 1}, {2, 2}, {4, 1}})[nspin0]; + module_dm::DensityMatrix, std::complex> tmp_dm(pv, nspin_dm, kv.kvec_d, kv.get_nks() / nspin_dm); + // module_dm::DensityMatrix, double> DM_real(pv, nspin_dm, kv.kvec_d, kv.get_nks() / nspin_dm); + // module_dm::DensityMatrix, double> DM_imag(pv, nspin_dm, kv.kvec_d, kv.get_nks() / nspin_dm); + // calculate DMK + module_dm::dm_from_psi(pv, pelec->wg, psi[0], tmp_dm); // init DMR - tmp_dm.init_DMR(ra, &ucell); + tmp_dm.init_dmr(ra, &ucell); - int nks = tmp_dm.get_DMK_nks() / nspin_dm; + int nks = tmp_dm.get_dmk_nks() / nspin_dm; double current_total[3] = {0.0, 0.0, 0.0}; for (int is = 1; is <= nspin_dm; ++is) { for (int ik = 0; ik < nks; ++ik) { - if(PARAM.inp.td_stype!=2) + if (PARAM.inp.td_stype != 2) { - tmp_dm.cal_DMR(ik); + tmp_dm.cal_dmr(ik); } else { - tmp_dm.cal_DMR_td(td_p->get_phase_hybrid(),TD_info::cart_At,ik); + tmp_dm.cal_dmr_td(td_p->get_phase_hybrid(), TD_info::A_prop_ha, ik); } - + // check later double current_ik[3] = {0.0, 0.0, 0.0}; #ifdef _OPENMP @@ -274,8 +269,8 @@ void ModuleIO::write_current_eachk(const UnitCell& ucell, const int start1 = ucell.itiaiw2iwt(T1, I1, 0); for (int cb = 0; cb < ra.na_each[iat]; ++cb) { - const int T2 = ra.info[iat][cb][3]; - const int I2 = ra.info[iat][cb][4]; + const int T2 = ra.get_info(iat, cb)[3]; + const int I2 = ra.get_info(iat, cb)[4]; const int start2 = ucell.itiaiw2iwt(T2, I2, 0); @@ -283,20 +278,16 @@ void ModuleIO::write_current_eachk(const UnitCell& ucell, // get iat2 int iat2 = ucell.itia2iat(T2, I2); - double Rx = ra.info[iat][cb][0]; - double Ry = ra.info[iat][cb][1]; - double Rz = ra.info[iat][cb][2]; - //std::cout<< "iat1: " << iat1 << " iat2: " << iat2 << " Rx: " << Rx << " Ry: " << Ry << " Rz:" << Rz << std::endl; - // get BaseMatrix - hamilt::BaseMatrix>* tmp_matrix - = tmp_dm.get_DMR_pointer(is)->find_matrix(iat1, iat2, Rx, Ry, Rz); + double Rx = ra.get_info(iat, cb)[0]; + double Ry = ra.get_info(iat, cb)[1]; + double Rz = ra.get_info(iat, cb)[2]; + // std::cout<< "iat1: " << iat1 << " iat2: " << iat2 << " Rx: " << Rx << " Ry: " << Ry << " Rz:" << Rz << std::endl; + // get BaseMatrix + hamilt::BaseMatrix>* tmp_matrix = tmp_dm.get_dmr_ptr(is)->find_matrix(iat1, iat2, Rx, Ry, Rz); // refactor - hamilt::BaseMatrix>* tmp_m_rvx - = current_term[0]->find_matrix(iat1, iat2, Rx, Ry, Rz); - hamilt::BaseMatrix>* tmp_m_rvy - = current_term[1]->find_matrix(iat1, iat2, Rx, Ry, Rz); - hamilt::BaseMatrix>* tmp_m_rvz - = current_term[2]->find_matrix(iat1, iat2, Rx, Ry, Rz); + hamilt::BaseMatrix>* tmp_m_rvx = current_term[0]->find_matrix(iat1, iat2, Rx, Ry, Rz); + hamilt::BaseMatrix>* tmp_m_rvy = current_term[1]->find_matrix(iat1, iat2, Rx, Ry, Rz); + hamilt::BaseMatrix>* tmp_m_rvz = current_term[2]->find_matrix(iat1, iat2, Rx, Ry, Rz); if (tmp_matrix == nullptr) { continue; @@ -323,7 +314,7 @@ void ModuleIO::write_current_eachk(const UnitCell& ucell, // std::cout<<"mu: "<< mu <<" nu: "<< nu << std::endl; // std::cout<<"dm2d1_real: "<< dm2d1_real << " dm2d1_imag: "<< dm2d1_imag << std::endl; // std::cout<<"rvz: "<< rvz.real() << " " << rvz.imag() << std::endl; - local_current_ik[0] -= dm2d1.real() * rvx.real() - dm2d1.imag() * rvx.imag(); + local_current_ik[0] -= dm2d1.real() * rvx.real() - dm2d1.imag() * rvx.imag(); local_current_ik[1] -= dm2d1.real() * rvy.real() - dm2d1.imag() * rvy.imag(); local_current_ik[2] -= dm2d1.real() * rvz.real() - dm2d1.imag() * rvz.imag(); } // end kk @@ -348,15 +339,14 @@ void ModuleIO::write_current_eachk(const UnitCell& ucell, // MPI_Reduce(local_current_ik, current_ik, 3, MPI_DOUBLE, MPI_SUM, 0, MPI_COMM_WORLD); if (GlobalV::MY_RANK == 0 && TD_info::out_current_k) { - std::string filename = PARAM.globalv.global_out_dir + "current_s" + std::to_string(is) + "k" - + std::to_string(ik+1) + ".txt"; + std::string filename + = PARAM.globalv.global_out_dir + "current_s" + std::to_string(is) + "k" + std::to_string(ik + 1) + ".txt"; std::ofstream fout; fout.open(filename, std::ios::app); fout << std::setprecision(16); fout << std::scientific; - fout << istep+1 << " " << current_ik[0]/omega - << " " << current_ik[1]/omega - << " " << current_ik[2]/omega << std::endl; + fout << istep + 1 << " " << current_ik[0] / omega << " " << current_ik[1] / omega << " " << current_ik[2] / omega + << std::endl; fout.close(); } // write end @@ -369,63 +359,56 @@ void ModuleIO::write_current_eachk(const UnitCell& ucell, fout.open(filename, std::ios::app); fout << std::setprecision(16); fout << std::scientific; - fout << istep+1 << " " << current_total[0]/omega - << " " << current_total[1]/omega - << " " << current_total[2]/omega << std::endl; + fout << istep + 1 << " " << current_total[0] / omega << " " << current_total[1] / omega << " " << current_total[2] / omega + << std::endl; fout.close(); } ModuleBase::timer::end("ModuleIO", "write_current"); return; } -template -void ModuleIO::write_current_eachk( - const UnitCell& ucell, - const int istep, - const psi::Psi>* psi, - const elecstate::ElecState* pelec, - const K_Vectors& kv, - const TwoCenterIntegrator* intor, - const Parallel_Orbitals* pv, - const LCAO_Orbitals& orb, - const Velocity_op* cal_current, - TD_info* td_p, - Record_adj& ra); -template -void ModuleIO::write_current_eachk>(const UnitCell& ucell, - const int istep, - const psi::Psi>* psi, - const elecstate::ElecState* pelec, - const K_Vectors& kv, - const TwoCenterIntegrator* intor, - const Parallel_Orbitals* pv, - const LCAO_Orbitals& orb, - const Velocity_op>* cal_current, - TD_info* td_p, - Record_adj& ra); -template -void ModuleIO::write_current(const UnitCell& ucell, - const int istep, - const psi::Psi>* psi, - const elecstate::ElecState* pelec, - const K_Vectors& kv, - const TwoCenterIntegrator* intor, - const Parallel_Orbitals* pv, - const LCAO_Orbitals& orb, - const Velocity_op* cal_current, - TD_info* td_p, - Record_adj& ra); -template -void ModuleIO::write_current>(const UnitCell& ucell, - const int istep, - const psi::Psi>* psi, - const elecstate::ElecState* pelec, - const K_Vectors& kv, - const TwoCenterIntegrator* intor, - const Parallel_Orbitals* pv, - const LCAO_Orbitals& orb, - const Velocity_op>* cal_current, - TD_info* td_p, - Record_adj& ra); +template void ModuleIO::write_current_eachk(const UnitCell& ucell, + const int istep, + const psi::Psi>* psi, + const elecstate::ElecState* pelec, + const K_Vectors& kv, + const TwoCenterIntegrator* intor, + const Parallel_Orbitals* pv, + const LCAO_Orbitals& orb, + const Velocity_op* cal_current, + TD_info* td_p, + Record_adj& ra); +template void ModuleIO::write_current_eachk>(const UnitCell& ucell, + const int istep, + const psi::Psi>* psi, + const elecstate::ElecState* pelec, + const K_Vectors& kv, + const TwoCenterIntegrator* intor, + const Parallel_Orbitals* pv, + const LCAO_Orbitals& orb, + const Velocity_op>* cal_current, + TD_info* td_p, + Record_adj& ra); +template void ModuleIO::write_current(const UnitCell& ucell, + const int istep, + const psi::Psi>* psi, + const elecstate::ElecState* pelec, + const K_Vectors& kv, + const TwoCenterIntegrator* intor, + const Parallel_Orbitals* pv, + const LCAO_Orbitals& orb, + const Velocity_op* cal_current, + TD_info* td_p, + Record_adj& ra); +template void ModuleIO::write_current>(const UnitCell& ucell, + const int istep, + const psi::Psi>* psi, + const elecstate::ElecState* pelec, + const K_Vectors& kv, + const TwoCenterIntegrator* intor, + const Parallel_Orbitals* pv, + const LCAO_Orbitals& orb, + const Velocity_op>* cal_current, + TD_info* td_p, + Record_adj& ra); #endif //__LCAO - diff --git a/source/source_io/module_current/td_current_io.h b/source/source_io/module_current/td_current_io.h index 8242adc699b..4814a81491c 100644 --- a/source/source_io/module_current/td_current_io.h +++ b/source/source_io/module_current/td_current_io.h @@ -61,7 +61,7 @@ void set_rR_from_hR(const UnitCell& ucell, const Grid_Driver& GridD, const LCAO_Orbitals& orb, const Parallel_Orbitals* pv, - cal_r_overlap_R& r_calculator, + Position_op& r_calculator, const hamilt::HContainer>* hR, ModuleBase::Vector3*>& rR); template diff --git a/source/source_io/module_current/td_current_io_comm.cpp b/source/source_io/module_current/td_current_io_comm.cpp index 0f8fbbcde0a..e1bb81a856f 100644 --- a/source/source_io/module_current/td_current_io_comm.cpp +++ b/source/source_io/module_current/td_current_io_comm.cpp @@ -108,7 +108,7 @@ void ModuleIO::set_rR_from_hR(const UnitCell& ucell, const Grid_Driver& GridD, const LCAO_Orbitals& orb, const Parallel_Orbitals* pv, - cal_r_overlap_R& r_calculator, + Position_op& r_calculator, const hamilt::HContainer>* hR, ModuleBase::Vector3*>& rR) { @@ -294,7 +294,7 @@ void ModuleIO::cal_velocity_basis_k(const UnitCell& ucell, const int nrow = pv->get_row_size(); if (elecstate::H_TDDFT_pw::stype == 2) { - module_rt::folding_HR_td(hR, hk, kv.kvec_d[ik], TD_info::cart_At, td_p->get_phase_hybrid(), nrow, 1); + module_rt::folding_HR_td(hR, hk, kv.kvec_d[ik], TD_info::A_prop_ha, td_p->get_phase_hybrid(), nrow, 1); } else { @@ -304,7 +304,7 @@ void ModuleIO::cal_velocity_basis_k(const UnitCell& ucell, ModuleBase::GlobalFunc::ZEROS(sk, pv->nloc); if (elecstate::H_TDDFT_pw::stype == 2) { - module_rt::folding_HR_td(sR, sk, kv.kvec_d[ik], TD_info::cart_At, td_p->get_phase_hybrid(), nrow, 1); + module_rt::folding_HR_td(sR, sk, kv.kvec_d[ik], TD_info::A_prop_ha, td_p->get_phase_hybrid(), nrow, 1); } else { @@ -350,7 +350,7 @@ void ModuleIO::cal_velocity_basis_k(const UnitCell& ucell, hR, partial_hk, kv.kvec_d[ik], - TD_info::cart_At, + TD_info::A_prop_ha, td_p->get_phase_hybrid(), i_alpha, nrow, @@ -368,7 +368,7 @@ void ModuleIO::cal_velocity_basis_k(const UnitCell& ucell, sR, partial_sk, kv.kvec_d[ik], - TD_info::cart_At, + TD_info::A_prop_ha, td_p->get_phase_hybrid(), i_alpha, nrow, @@ -383,7 +383,7 @@ void ModuleIO::cal_velocity_basis_k(const UnitCell& ucell, // folding_rR(rR[i_alpha], partial_sk, rk, pv, kv.kvec_d[ik], nrow, 1); if (elecstate::H_TDDFT_pw::stype == 2) { - module_rt::folding_HR_td(*rR[i_alpha], rk, kv.kvec_d[ik], TD_info::cart_At, td_p->get_phase_hybrid(), nrow, 1); + module_rt::folding_HR_td(*rR[i_alpha], rk, kv.kvec_d[ik], TD_info::A_prop_ha, td_p->get_phase_hybrid(), nrow, 1); } else { diff --git a/source/source_io/module_current/td_current_pw.cpp b/source/source_io/module_current/td_current_pw.cpp new file mode 100644 index 00000000000..a9a033aa4f1 --- /dev/null +++ b/source/source_io/module_current/td_current_pw.cpp @@ -0,0 +1,219 @@ +#include "td_current_pw.h" + +#include "source_base/module_container/ATen/core/tensor.h" +#include "source_base/module_device/memory_op.h" +#include "source_base/parallel_comm.h" +#include "source_base/parallel_reduce.h" +#include "source_base/timer.h" +#include "source_hamilt/module_xc/xc_functional.h" +#include "source_hsolver/kernels/linear_op.h" + +// Keep operator dependencies local to this implementation file. +#include "source_pw/module_pwdft/op_pw_vel.h" + +#include +#include +#include +#include +#include + +namespace ModuleIO +{ + +template +PWCurrentResult CurrentPW::calculate(const UnitCell& ucell, + const ModulePW::PW_Basis_K* wfcpw, + psi::Psi, Device>* psi, + const elecstate::ElecState* pelec, + const K_Vectors& kv, + pseudopot_cell_vnl* ppcell, + const int gauge, + const ModuleBase::Vector3& vector_potential) +{ + ModuleBase::timer::start("ModuleIO", "calculate"); + + using Complex = std::complex; + using syncmem_complex_d2h_op = base_device::memory::synchronize_memory_op; + using setmem_complex_op = base_device::memory::set_memory_op; + + double current_total[3] = {0.0, 0.0, 0.0}; + const int nks = wfcpw->nks; + const int max_npw = wfcpw->npwk_max; + const int nbands = psi->get_nbands(); + const int nkstot = kv.get_nkstot(); + const int* isk = kv.isk.data(); + + const int n_npwx = nbands; + + // ============================================================== + // Store Cartesian current components for each k point. + // ============================================================== + const std::int64_t current_elements = 3 * static_cast(nkstot); + std::vector current_k(current_elements, 0.0); + + // Refresh the potential view used by the shared velocity operator. + const FPTYPE* vtau = nullptr; + int vtau_col = 0; + int vtau_row = 0; + if (gauge == 0 && XC_Functional::get_ked_flag()) + { + if (pelec->pot == nullptr) + { + ModuleBase::WARNING_QUIT("calculate", "Missing potential for meta-GGA current."); + } + vtau = pelec->pot->template get_vofk_smooth_data(); + vtau_col = pelec->pot->get_vofk_smooth().nc; + vtau_row = pelec->pot->get_vofk_smooth().nr; + if (vtau_col != wfcpw->nrxx || (wfcpw->nrxx > 0 && (vtau == nullptr || vtau_row != kv.get_spin_mult()))) + { + ModuleBase::WARNING_QUIT("calculate", "Invalid smooth-grid potential for meta-GGA current."); + } + // Preserve the enabled correction on ranks without real-space planes, which still join FFTs. + vtau_row = kv.get_spin_mult(); + } + if (!velocity_) + { + velocity_.reset(new hamilt::Velocity(wfcpw, isk, ppcell, &ucell, true, vtau, vtau_col, vtau_row)); + } + velocity_->set_state(isk, vtau, vtau_col, vtau_row); + + using CtDevice = typename ct::PsiToContainer::type; + const ct::DeviceType device = ct::DeviceTypeToEnum::value; + const std::int64_t component_elements = static_cast(n_npwx) * max_npw; + const std::int64_t velocity_elements = 3 * component_elements; + const std::int64_t velocity_size = std::max(1, velocity_elements); + const int64_t dot_size = std::max(1, n_npwx); + if (vpsi_.NumElements() < velocity_size || vpsi_.data_type() != ct::DataTypeToEnum::value || vpsi_.device_type() != device) + { + vpsi_ = ct::Tensor(ct::DataTypeToEnum::value, device, {velocity_size}); + } + if (dots_.NumElements() < dot_size || dots_.data_type() != ct::DataTypeToEnum::value || dots_.device_type() != device) + { + dots_ = ct::Tensor(ct::DataTypeToEnum::value, device, {dot_size}); + } + Complex* d_vpsi = vpsi_.template data(); + ct::Tensor& dot_buffer = dots_; + band_current_.resize(n_npwx); + std::vector& band_current = band_current_; + + for (int ik = 0; ik < nks; ++ik) + { + psi->fix_k(ik); + Complex* current_psi_ptr = psi->get_pointer(); + const int npw = wfcpw->npwk[ik]; + + if (velocity_elements > 0) + { + setmem_complex_op()(d_vpsi, 0, velocity_elements); + } + + velocity_->init(ik, vector_potential); + velocity_->act(psi, n_npwx, current_psi_ptr, d_vpsi, false); + + for (int id = 0; id < 3; ++id) + { + const Complex* component = d_vpsi + id * component_elements; + hsolver::linear_op().dot(max_npw, npw, n_npwx, current_psi_ptr, component, dot_buffer.data()); + syncmem_complex_d2h_op()(band_current.data(), dot_buffer.data(), n_npwx); + for (int ib = 0; ib < nbands; ++ib) + { + const double contribution = -pelec->wg(ik, ib) * std::real(band_current[ib]); + current_total[id] += contribution; + const std::int64_t current_index = static_cast(kv.ik2iktot[ik]) * 3 + id; + current_k[current_index] += contribution; + } + } + } + + // ============================================================== + // Reduce all current components together across MPI ranks. + // ============================================================== + Parallel_Reduce::reduce_all(current_total, 3); + const std::int64_t max_chunk = std::numeric_limits::max(); + for (std::int64_t offset = 0; offset < current_elements;) + { + const std::int64_t remaining = current_elements - offset; + const int count = static_cast(std::min(max_chunk, remaining)); + double* chunk = current_k.data() + offset; + Parallel_Reduce::reduce_all(chunk, count); + offset += count; + } + + PWCurrentResult result; + for (int d = 0; d < 3; ++d) + { + result.total[d] = current_total[d] / ucell.omega; + } + for (double& value: current_k) + { + value /= ucell.omega; + } + result.per_k = std::move(current_k); + ModuleBase::timer::end("ModuleIO", "calculate"); + return result; +} + +void write_pw_current(const PWCurrentResult& current, + const int istep, + const int nk_per_spin, + const bool out_current_k, + const std::string& out_dir) +{ + // Always write the total current when this routine is enabled. + std::string filename_tot = out_dir + "current_tot.txt"; + std::ofstream fout_tot; + fout_tot.open(filename_tot, std::ios::app); + fout_tot << std::setprecision(16) << std::scientific; + fout_tot << istep + 1 << " " << current.total[0] << " " << current.total[1] << " " << current.total[2] << std::endl; + fout_tot.close(); + + // Optionally write the contribution from each k point. + if (out_current_k) + { + for (int ik = 0; ik < static_cast(current.per_k.size() / 3); ++ik) + { + // Use a one-based spin index in the output filename. + int is = ik / nk_per_spin + 1; + int k_idx = ik % nk_per_spin + 1; + + std::string filename_k = out_dir + "current_s" + std::to_string(is) + "k" + std::to_string(k_idx) + ".txt"; + std::ofstream fout_k; + fout_k.open(filename_k, std::ios::app); + fout_k << std::setprecision(16) << std::scientific; + const std::int64_t offset = static_cast(ik) * 3; + fout_k << istep + 1 << " " << current.per_k[offset] << " " << current.per_k[offset + 1] << " " << current.per_k[offset + 2] + << std::endl; + fout_k.close(); + } + } +} + +template +void CurrentPW::write(const int istep, + const UnitCell& ucell, + const ModulePW::PW_Basis_K* wfcpw, + psi::Psi, Device>* psi, + const elecstate::ElecState* pelec, + const K_Vectors& kv, + pseudopot_cell_vnl* ppcell, + const int gauge, + const ModuleBase::Vector3& A_right_ha, + const bool out_current_k, + const std::string& out_dir, + const int world_rank) +{ + const PWCurrentResult result = calculate(ucell, wfcpw, psi, pelec, kv, ppcell, gauge, A_right_ha); + if (world_rank == 0) + { + const int nk_per_spin = kv.get_nkstot() / kv.get_spin_mult(); + write_pw_current(result, istep, nk_per_spin, out_current_k, out_dir); + } +} + +template class CurrentPW; +template class CurrentPW; +#if defined(__CUDA) || defined(__ROCM) +template class CurrentPW; +template class CurrentPW; +#endif +} // namespace ModuleIO diff --git a/source/source_io/module_current/td_current_pw.h b/source/source_io/module_current/td_current_pw.h new file mode 100644 index 00000000000..c240cef8746 --- /dev/null +++ b/source/source_io/module_current/td_current_pw.h @@ -0,0 +1,65 @@ +#ifndef TD_CURRENT_PW_H +#define TD_CURRENT_PW_H + +#include "source_cell/klist.h" +#include "source_cell/unitcell.h" +#include "source_estate/elecstate.h" +#include "source_psi/psi.h" +#include "source_pw/module_pwdft/op_pw_vel.h" +#include "source_pw/module_pwdft/vnl_pw.h" + +#include +#include + +namespace ModuleIO +{ +/** @brief Reduced current densities in Hartree atomic units. */ +struct PWCurrentResult +{ + ModuleBase::Vector3 total; + std::vector per_k; +}; + +/** @brief Write already reduced current densities; performs no collective operations. */ +void write_pw_current(const PWCurrentResult& current, + const int istep, + const int nk_per_spin, + const bool out_current_k, + const std::string& out_dir); + +/** @brief Persistent one-k-point current workspace for a propagation run. */ +template +class CurrentPW +{ + private: + std::unique_ptr> velocity_; + ct::Tensor vpsi_; + ct::Tensor dots_; + std::vector> band_current_; + + public: + /** @brief Compute globally reduced current densities without file output. */ + PWCurrentResult calculate(const UnitCell& ucell, + const ModulePW::PW_Basis_K* wfcpw, + psi::Psi, Device>* psi, + const elecstate::ElecState* pelec, + const K_Vectors& kv, + pseudopot_cell_vnl* ppcell, + const int gauge, + const ModuleBase::Vector3& A_right_ha); + /** @brief Evaluate and write the current using native wavefunctions. */ + void write(const int istep, + const UnitCell& ucell, + const ModulePW::PW_Basis_K* wfcpw, + psi::Psi, Device>* psi, + const elecstate::ElecState* pelec, + const K_Vectors& kv, + pseudopot_cell_vnl* ppcell, + const int gauge, + const ModuleBase::Vector3& vector_potential, + const bool out_current_k, + const std::string& out_dir, + const int world_rank); +}; +} // namespace ModuleIO +#endif diff --git a/source/source_io/module_dhs/write_dh.cpp b/source/source_io/module_dhs/write_dh.cpp index 8d01ce67f96..3fca7a9fa1e 100644 --- a/source/source_io/module_dhs/write_dh.cpp +++ b/source/source_io/module_dhs/write_dh.cpp @@ -2,8 +2,8 @@ #include "source_base/global_function.h" #include "source_base/timer.h" -#include "source_io/module_hs/write_hs.h" -#include "source_io/module_hs/write_hs_r.h" +#include "source_io/module_hs/hs_dense_io.h" +#include "source_io/module_hs/hsr_writer.h" #include "source_cell/ucell_io.h" #include "source_io/module_parameter/parameter.h" #include "source_hamilt/module_hcontainer/hcontainer_funcs.h" @@ -122,7 +122,8 @@ void write_dh_perI(WriteDHParams& params, out_app_flag, fk, pv, - GlobalV::DRANK); + GlobalV::DRANK, + PARAM.inp.ks_solver); } } } @@ -143,16 +144,16 @@ void write_dH_components(WriteDHParams& params, const Exx_Info& exx_info) } #ifdef __EXX - // The EXX interfaces carried by WriteDHParams are gamma-only (see write_dh.h): at multi-k - // dH^EXX would be the derivative with respect to every mirror atom, which this output is - // not meant for. Quit instead of writing a dH sum that silently omits the EXX term. - if (exx_info.info_global.cal_exx && !PARAM.globalv.gamma_only_local - && (PARAM.inp.out_mat_dh[0] || PARAM.inp.out_mat_dh_exx[0])) + // dV^EXX/dR cannot be computed in this branch (it requires the LibRI cal_dHs/dHs_HF + // APIs from the unmerged LibRI PR#10), so the dH sum would silently omit the EXX term + // at any k-point setting, gamma-only included. Quit whenever EXX is on and the sum is + // requested; the individual non-EXX terms remain available. + if (exx_info.info_global.cal_exx && PARAM.inp.out_mat_dh[0]) { ModuleBase::WARNING_QUIT("write_dH_components", - "out_mat_dh (the dH sum) and out_mat_dh_exx are only supported for gamma-only when EXX is on. " - "Use gamma_only, or request the individual non-EXX terms (out_mat_dh_t, " - "out_mat_dh_vnl, out_mat_dh_vl, out_mat_dh_vh, out_mat_dh_vxc)."); + "out_mat_dh (the dH sum) is not supported with EXX because dV^EXX/dR is " + "unavailable; request the individual non-EXX terms (out_mat_dh_t, " + "out_mat_dh_vnl, out_mat_dh_vl, out_mat_dh_vh, out_mat_dh_vxc) instead."); } #endif @@ -164,7 +165,7 @@ void write_dH_components(WriteDHParams& params, const Exx_Info& exx_info) if (PARAM.inp.out_mat_dh[0]) { - write_dH_sum(params, exx_info); + write_dH_sum(params); } if (PARAM.inp.out_mat_dh_t[0]) @@ -194,13 +195,6 @@ void write_dH_components(WriteDHParams& params, const Exx_Info& exx_info) write_dH_vxc_pulay(params); } -#ifdef __EXX - if (PARAM.inp.out_mat_dh_exx[0]) - { - write_dH_exx(params, exx_info); - } -#endif - ModuleBase::timer::end("ModuleIO", "write_dH_components"); } diff --git a/source/source_io/module_dhs/write_dh.h b/source/source_io/module_dhs/write_dh.h index ebe5842fa92..1258daad4fb 100644 --- a/source/source_io/module_dhs/write_dh.h +++ b/source/source_io/module_dhs/write_dh.h @@ -20,9 +20,6 @@ std::unique_ptr make_unique(Args &&... args) return std::unique_ptr(new T(std::forward(args)...)); } -template -class Exx_LRI_Interface; - namespace ModuleIO { @@ -54,19 +51,6 @@ struct WriteDHParams // terms: V^H needs the total density (sum over spins), V^XC the spin-resolved densities. std::vector*> dmR; const Charge* chg = nullptr; // ground-state charge for XC Hellmann-Feynman (FDM) -#ifdef __EXX - // The gamma-only (TK==double) exx interfaces used by write_dH_exx. - // Deliberately NOT templated on TK, for two reasons: - // 1. Physics: at multi-k the derivative would be taken with respect to every mirror - // atom of the periodic images, which is not what this output is used for. - // 2. Cost: templating these pointers on TK would force WriteHParams, WriteDHParams and - // every free function taking them to become templates as well -- a large, purely - // mechanical change for a case nobody needs. - // Multi-k + EXX is therefore rejected up front (see write_dH_components) - // instead of silently producing output with the EXX term missing. - Exx_LRI_Interface* exd = nullptr; - Exx_LRI_Interface>* exc = nullptr; -#endif }; // Returns 0-based atom indices to output (converted from the 1-based user-facing values stored at param[2+]); @@ -109,11 +93,7 @@ bool write_dH_vxc(WriteDHParams& params); bool write_dH_vxc_pulay(WriteDHParams& params); -bool write_dH_sum(WriteDHParams& params, const Exx_Info& exx_info); - -#ifdef __EXX -bool write_dH_exx(WriteDHParams& params, const Exx_Info& exx_info); -#endif +bool write_dH_sum(WriteDHParams& params); } // namespace ModuleIO diff --git a/source/source_io/module_dhs/write_dh_terms.cpp b/source/source_io/module_dhs/write_dh_terms.cpp index 763b01d4053..77dbf2a97f5 100644 --- a/source/source_io/module_dhs/write_dh_terms.cpp +++ b/source/source_io/module_dhs/write_dh_terms.cpp @@ -1,5 +1,5 @@ #include "source_base/timer.h" -#include "source_io/module_hs/write_hs_r.h" +#include "source_io/module_hs/hsr_writer.h" #include "source_cell/ucell_io.h" #include "source_io/module_parameter/parameter.h" #include "source_hamilt/module_hcontainer/hcontainer_funcs.h" @@ -13,10 +13,6 @@ #include "source_base/global_variable.h" #include "source_base/parallel_reduce.h" #include "write_dh.h" -#ifdef __EXX -#include "source_lcao/module_operator_lcao/op_exx_lcao.h" -#include "source_lcao/module_ri/exx_lri_interface.hpp" -#endif #include #include @@ -203,58 +199,6 @@ bool write_dH_veff_term(WriteDHParams& params, return true; } -#ifdef __EXX -// Per-(spin) filler for the EXX dH term. Assumes ex->cal_exx_dHs(...) has already been called -// (it builds dHexxs for all spins at once). Templated on the Hexx tensor data type (double for -// the real interface exd, std::complex for the complex interface exc). -template -void fill_dH_exx(WriteDHParams& params, Exx_LRI_Interface* ex, int ispin, PerIContainers& c, const Exx_Info& exx_info) -{ - const UnitCell& ucell = *params.ucell; - const Parallel_Orbitals& pv = *params.pv; - - // OperatorEXX dereferences hR_in in its constructor and reallocates it, so pass a - // throwaway container (its cell_nearest is built from kv and reused for dhR below). - hamilt::HContainer hR_dummy(const_cast(&pv)); - hamilt::OperatorEXX> op_exx( - nullptr, - &hR_dummy, - ucell, - *params.kv, - nullptr, - nullptr, - &exx_info, - hamilt::Add_Hexx_Type::R); - - op_exx.cal_dH(ispin, c.g, ex->get_dHexxs()); -} - -// Shared driver for the EXX dH term. The per-atom-I dH is always written into real -// HContainer (add_HexxR converts Tdata -> double). -template -void write_dH_exx_impl(WriteDHParams& params, Exx_LRI_Interface* ex, const Exx_Info& exx_info) -{ - const UnitCell& ucell = *params.ucell; - const Parallel_Orbitals& pv = *params.pv; - const int nat = ucell.nat; - const int nspin = params.nspin; - - // 1+2. build the exx-form per-direction/atom/spin dH (dHexxs) from the current mixed DM - ex->cal_exx_dHs(ucell, pv, nspin); - - const std::vector af = dh_atom_filter(PARAM.inp.out_mat_dh_exx); - // 3+4. convert dHexxs to per-atom-I HContainers and write, one spin channel at a time - for (int ispin = 0; ispin < (nspin == 2 ? 2 : 1); ++ispin) - { - PerIContainers c(pv, nat); - - fill_dH_exx(params, ex, ispin, c, exx_info); - - ModuleIO::write_dh_perI(params, ispin, "dvexxr", "dvexxk", "dV^EXX", c.g, af); - } -} -#endif - } // namespace bool write_dH_t(WriteDHParams& params) @@ -366,42 +310,18 @@ bool write_dH_vxc_pulay(WriteDHParams& params) return ok; } -#ifdef __EXX -bool write_dH_exx(WriteDHParams& params, const Exx_Info& exx_info) -{ - ModuleBase::TITLE("ModuleIO", "write_dH_exx"); - ModuleBase::timer::start("ModuleIO", "write_dH_exx"); - - bool ok = false; - // exd (real Hexx) and exc (complex Hexx) are mutually exclusive; pick by real_number. - if (exx_info.info_ri.real_number) - { - if (params.exd != nullptr) - { - write_dH_exx_impl(params, params.exd, exx_info); - ok = true; - } - } - else - { - if (params.exc != nullptr) - { - write_dH_exx_impl(params, params.exc, exx_info); - ok = true; - } - } - - ModuleBase::timer::end("ModuleIO", "write_dH_exx"); - return ok; -} -#endif - -// Total dH = sum of ALL dH terms (dT + dV^NL + dV^L + dV^H + dV^XC, plus dV^EXX when hybrid is -// active), independent of the per-component out_mat_dh_* flags: out_mat_dh on its own yields the -// full sum. Each term is built into its own per-atom-I containers (via the same fillers the -// per-term writers use) and accumulated with HContainer::add_value_union, which unions the -// (generally different) sparsities and sums values. Each term already carries its own sign. -bool write_dH_sum(WriteDHParams& params, const Exx_Info& exx_info) +// Total dH = sum of ALL dH terms (dT + dV^NL + dV^L + dV^H + dV^XC), independent of the +// per-component out_mat_dh_* flags: out_mat_dh on its own yields the full sum. Each term is +// built into its own per-atom-I containers (via the same fillers the per-term writers use) and +// accumulated with HContainer::add_value_union, which unions the (generally different) +// sparsities and sums values. Each term already carries its own sign. +// +// Note: the EXX (dV^EXX/dR) term is intentionally NOT included: it requires the LibRI +// cal_dHs/dHs_HF APIs that are only available in the unmerged LibRI PR#10 (the former +// EXX_DEV path), so it cannot be computed in the current develop branch. When EXX is active +// and out_mat_dh is requested at gamma-only, the resulting dH sum therefore omits the EXX +// contribution. +bool write_dH_sum(WriteDHParams& params) { ModuleBase::TITLE("ModuleIO", "write_dH_sum"); ModuleBase::timer::start("ModuleIO", "write_dH_sum"); @@ -410,18 +330,6 @@ bool write_dH_sum(WriteDHParams& params, const Exx_Info& exx_info) const int nat = params.ucell->nat; const int nspin = params.nspin; -#ifdef __EXX - // EXX (whenever active) is part of the total dH; build dHexxs once up front. - const bool do_exx = (params.exd != nullptr || params.exc != nullptr); - if (do_exx) - { - if (exx_info.info_ri.real_number && params.exd != nullptr) - params.exd->cal_exx_dHs(*params.ucell, pv, nspin); - else if (!exx_info.info_ri.real_number && params.exc != nullptr) - params.exc->cal_exx_dHs(*params.ucell, pv, nspin); - } -#endif - for (int ispin = 0; ispin < (nspin == 2 ? 2 : 1); ++ispin) { PerIContainers sum(pv, nat); @@ -463,17 +371,6 @@ bool write_dH_sum(WriteDHParams& params, const Exx_Info& exx_info) fill_dH_veff(params, params.pot_vxc, params.chg ? "xc" : "none", ispin, c); accumulate(c); } -#ifdef __EXX - if (do_exx) - { - PerIContainers c(pv, nat); - if (exx_info.info_ri.real_number && params.exd != nullptr) - fill_dH_exx(params, params.exd, ispin, c, exx_info); - else if (params.exc != nullptr) - fill_dH_exx(params, params.exc, ispin, c, exx_info); - accumulate(c); - } -#endif ModuleIO::write_dh_perI(params, ispin, "dhr", "dhk", "dH", sum.g, dh_atom_filter(PARAM.inp.out_mat_dh)); } diff --git a/source/source_io/module_dipole/dipole_io.h b/source/source_io/module_dipole/dipole_io.h index 389d7681f64..3911b5f0375 100644 --- a/source/source_io/module_dipole/dipole_io.h +++ b/source/source_io/module_dipole/dipole_io.h @@ -10,7 +10,7 @@ namespace ModuleIO { void write_dipole(const UnitCell& ucell, - const double* rho_save, + const double* rho, const ModulePW::PW_Basis* rhopw, const int& istep, const std::string& fn, diff --git a/source/source_io/module_dipole/write_dipole.cpp b/source/source_io/module_dipole/write_dipole.cpp index d09317ef518..c8a31671c0b 100644 --- a/source/source_io/module_dipole/write_dipole.cpp +++ b/source/source_io/module_dipole/write_dipole.cpp @@ -7,8 +7,7 @@ namespace ModuleIO // Helper function to print dipole moment components // name: descriptive name for the dipole moment type // px, py, pz: dipole moment components in x, y, z directions -void print_dipole_moment(std::ofstream& ofs_running, const std::string& name, - double px, double py, double pz) +void print_dipole_moment(std::ofstream& ofs_running, const std::string& name, double px, double py, double pz) { ofs_running << " " << name << std::endl; ModuleBase::GlobalFunc::OUT(ofs_running, "P_x(t)", px); @@ -17,10 +16,10 @@ void print_dipole_moment(std::ofstream& ofs_running, const std::string& name, } // Calculate and write dipole moment data for RT-TDDFT calculations -// +// // Dipole moment is a measure of the separation of positive and negative charges // in a system. In electronic structure calculations, we compute three components: -// +// // 1. Electronic dipole moment (P_elec): // Formula: P_elec = -int(r * rho(r) dr) // where rho(r) is the electron density @@ -36,14 +35,14 @@ void print_dipole_moment(std::ofstream& ofs_running, const std::string& name, // // Parameters: // - ucell: unit cell containing atomic structure and lattice information -// - rho_save: electron density on real-space grid +// - rho: current electron density on the real-space grid // - rhopw: plane wave basis information including grid dimensions // - istep: current time step // - fn: output file name // - ofs_running: output stream for logging // - precision: floating-point precision for output void write_dipole(const UnitCell& ucell, - const double* rho_save, + const double* rho, const ModulePW::PW_Basis* rhopw, const int& istep, const std::string& fn, @@ -73,45 +72,42 @@ void write_dipole(const UnitCell& ucell, // Used for coordinate transformation from fractional to Cartesian double small_value = 1e-10; double bmod[3]; - + for (int i = 0; i < 3; ++i) { bmod[i] = prepare(ucell, i); if (bmod[i] < small_value) { - ModuleBase::WARNING_QUIT("ModuleIO::write_dipole", - "bmod[" + std::to_string(i) + "] is too small or zero"); + ModuleBase::WARNING_QUIT("ModuleIO::write_dipole", "bmod[" + std::to_string(i) + "] is too small or zero"); } } // Validate grid dimensions to prevent division by zero - if (rhopw->nx == 0 || rhopw->ny == 0 || rhopw->nz == 0 || - rhopw->nxyz == 0 || rhopw->nplane == 0) + if (rhopw->nx == 0 || rhopw->ny == 0 || rhopw->nz == 0 || rhopw->nxyz == 0 || rhopw->nplane == 0) { - ModuleBase::WARNING_QUIT("ModuleIO::write_dipole", - "Invalid grid parameters: nx, ny, nz, nxyz, or nplane is zero"); + ModuleBase::WARNING_QUIT("ModuleIO::write_dipole", "Invalid grid parameters: nx, ny, nz, nxyz, or nplane is zero"); } // Calculate electronic dipole moment // P_elec = -int(r * rho(r) dr) // Discretized: P_elec[i] = -sum(r_grid[i] * rho(r_grid)) * (omega / nxyz) double dipole_elec[3] = {0.0, 0.0, 0.0}; - + // Precompute inverse grid dimensions for performance double inv_nx = 1.0 / static_cast(rhopw->nx); double inv_ny = 1.0 / static_cast(rhopw->ny); double inv_nz = 1.0 / static_cast(rhopw->nz); - // Loop over local grid points (parallel decomposition with OpenMP) - // Use reduction for thread-safe accumulation - #pragma omp parallel for reduction(-:dipole_elec[:3]) schedule(static) +// Loop over local grid points (parallel decomposition with OpenMP) +// Use reduction for thread-safe accumulation +#pragma omp parallel for reduction(- : dipole_elec[ : 3]) schedule(static) for (int ir = 0; ir < rhopw->nrxx; ++ir) { // Convert 1D index to 3D indices int i = ir / (rhopw->ny * rhopw->nplane); int j = ir / rhopw->nplane - i * rhopw->ny; int k = ir % rhopw->nplane + rhopw->startz_current; - + // Convert to fractional coordinates: r_i = i / N_i // Using multiplication instead of division for better performance double x = static_cast(i) * inv_nx; @@ -119,9 +115,9 @@ void write_dipole(const UnitCell& ucell, double z = static_cast(k) * inv_nz; // Accumulate: P_elec -= rho * r (negative sign from electron charge) - dipole_elec[0] -= rho_save[ir] * x; - dipole_elec[1] -= rho_save[ir] * y; - dipole_elec[2] -= rho_save[ir] * z; + dipole_elec[0] -= rho[ir] * x; + dipole_elec[1] -= rho[ir] * y; + dipole_elec[2] -= rho[ir] * z; } // Reduce across MPI processes to get global sum @@ -139,14 +135,11 @@ void write_dipole(const UnitCell& ucell, } // Output electronic dipole moment - print_dipole_moment(ofs_running, "Electronic dipole moment", - dipole_elec[0], dipole_elec[1], dipole_elec[2]); + print_dipole_moment(ofs_running, "Electronic dipole moment", dipole_elec[0], dipole_elec[1], dipole_elec[2]); // Write to file: step index and three dipole components - ofs << std::setprecision(precision) << istep + 1 - << " " << dipole_elec[0] - << " " << dipole_elec[1] - << " " << dipole_elec[2] << std::endl; + ofs << std::setprecision(precision) << istep + 1 << " " << dipole_elec[0] << " " << dipole_elec[1] << " " << dipole_elec[2] + << std::endl; // Calculate ionic dipole moment // P_ion = sum_{atom_types} sum_{atoms} (Z_v * tau) @@ -168,8 +161,7 @@ void write_dipole(const UnitCell& ucell, } // Output ionic dipole moment - print_dipole_moment(ofs_running, "Ionic dipole moment", - dipole_ion[0], dipole_ion[1], dipole_ion[2]); + print_dipole_moment(ofs_running, "Ionic dipole moment", dipole_ion[0], dipole_ion[1], dipole_ion[2]); // Calculate total dipole moment // P_tot = P_elec + P_ion @@ -180,14 +172,11 @@ void write_dipole(const UnitCell& ucell, } // Output total dipole moment - print_dipole_moment(ofs_running, "Total dipole moment", - dipole[0], dipole[1], dipole[2]); + print_dipole_moment(ofs_running, "Total dipole moment", dipole[0], dipole[1], dipole[2]); // Calculate and output total dipole moment norm // |P_tot| = sqrt(P_tot_x^2 + P_tot_y^2 + P_tot_z^2) - double dipole_sum = sqrt(dipole[0] * dipole[0] + - dipole[1] * dipole[1] + - dipole[2] * dipole[2]); + double dipole_sum = sqrt(dipole[0] * dipole[0] + dipole[1] * dipole[1] + dipole[2] * dipole[2]); ofs_running << " Total dipole moment norm" << std::endl; ModuleBase::GlobalFunc::OUT(ofs_running, "|P_tot(t)|", dipole_sum); @@ -201,39 +190,39 @@ void write_dipole(const UnitCell& ucell, } // Calculate the modulus of a reciprocal lattice vector -// Input: +// Input: // cell - unit cell containing reciprocal lattice G // dir - direction index (0=x, 1=y, 2=z) -// Output: +// Output: // bmod - |b_dir| where b_dir is the reciprocal lattice vector double prepare(const UnitCell& cell, int& dir) { double bvec[3] = {0.0}; - + // Select the appropriate reciprocal lattice vector components switch (dir) { - case 0: // x-direction - bvec[0] = cell.G.e11; - bvec[1] = cell.G.e12; - bvec[2] = cell.G.e13; - break; - case 1: // y-direction - bvec[0] = cell.G.e21; - bvec[1] = cell.G.e22; - bvec[2] = cell.G.e23; - break; - case 2: // z-direction - bvec[0] = cell.G.e31; - bvec[1] = cell.G.e32; - bvec[2] = cell.G.e33; - break; - default: - ModuleBase::WARNING_QUIT("ModuleIO::prepare", "Invalid direction index"); + case 0: // x-direction + bvec[0] = cell.G.e11; + bvec[1] = cell.G.e12; + bvec[2] = cell.G.e13; + break; + case 1: // y-direction + bvec[0] = cell.G.e21; + bvec[1] = cell.G.e22; + bvec[2] = cell.G.e23; + break; + case 2: // z-direction + bvec[0] = cell.G.e31; + bvec[1] = cell.G.e32; + bvec[2] = cell.G.e33; + break; + default: + ModuleBase::WARNING_QUIT("ModuleIO::prepare", "Invalid direction index"); } - + // Calculate and return the magnitude of the reciprocal lattice vector return sqrt(bvec[0] * bvec[0] + bvec[1] * bvec[1] + bvec[2] * bvec[2]); } -} // namespace ModuleIO \ No newline at end of file +} // namespace ModuleIO diff --git a/source/source_io/module_dm/test/test_write_dmr.cpp b/source/source_io/module_dm/test/test_write_dmr.cpp new file mode 100644 index 00000000000..e5876e7c07a --- /dev/null +++ b/source/source_io/module_dm/test/test_write_dmr.cpp @@ -0,0 +1,103 @@ +/** + * @file test_write_dmr.cpp + * @brief Unit tests for write_dmr.cpp: dmr_gen_fname and write_dmr_csr. + * + * dmr_gen_fname builds the spin-dependent DM(R) filename; write_dmr_csr + * writes the DM(R) matrix in the legacy CSR text layout. Both tests pin + * the current contract so format changes are caught by a failing test. + * Neither function reads PARAM (the high-level write_dmr driver does, but + * it is not exercised here), so no PARAM fixture is required. + */ +#include "gmock/gmock.h" +#include "gtest/gtest.h" + +#include "source_io/module_dm/write_dmr.h" +#include "source_cell/unitcell.h" +#include "source_hamilt/module_hcontainer/atom_pair.h" +#include "source_hamilt/module_hcontainer/hcontainer.h" + +#include +#include +#include +#include + +namespace +{ +std::string read_file(const std::string& filename) +{ + std::ifstream ifs(filename.c_str()); + std::ostringstream oss; + oss << ifs.rdbuf(); + return oss.str(); +} + +void init_unitcell(UnitCell& ucell) +{ + ucell.latName = "user_defined_lattice"; + ucell.lat0 = 10.0; + ucell.latvec.e11 = 1.0; + ucell.latvec.e22 = 1.0; + ucell.latvec.e33 = 1.0; + ucell.ntype = 1; + ucell.nat = 1; + ucell.atoms = new Atom[1]; + ucell.set_atom_flag = true; + ucell.atoms[0].label = "Si"; + ucell.atoms[0].na = 1; + ucell.atoms[0].nw = 2; + ucell.atoms[0].taud.resize(1); + ucell.atoms[0].taud[0] = ModuleBase::Vector3(0.0, 0.25, 0.5); +} + +void init_serial_orbitals(Parallel_Orbitals& pv) +{ + pv.atom_begin_row.resize(2); + pv.atom_begin_col.resize(2); + pv.atom_begin_row[0] = 0; + pv.atom_begin_row[1] = 2; + pv.atom_begin_col[0] = 0; + pv.atom_begin_col[1] = 2; + pv.nrow = 2; + pv.ncol = 2; + pv.set_serial(2, 2); +} + +void fill_matrix(hamilt::HContainer& matrix, Parallel_Orbitals& pv, double* values) +{ + hamilt::AtomPair pair(0, 0, 0, 0, 0, &pv, values); + matrix.insert_pair(pair); +} +} // namespace + +TEST(WriteDmr, GenFnameKeepsCurrentContract) +{ + EXPECT_EQ(ModuleIO::dmr_gen_fname(1, 0, true, -1), "dmrs1_nao.csr"); + EXPECT_EQ(ModuleIO::dmr_gen_fname(1, 1, false, 2), "dmrs2g3_nao.csr"); + EXPECT_EQ(ModuleIO::dmr_gen_fname(2, 0, true, 5), "dmrs1_nao.npz"); +} + +TEST(WriteDmr, CsrHeaderKeepsCurrentFormat) +{ + std::string filename = "write_hs_r_header_dmr.csr"; + std::remove(filename.c_str()); + + UnitCell ucell; + init_unitcell(ucell); + Parallel_Orbitals pv; + init_serial_orbitals(pv); + hamilt::HContainer matrix(&pv); + double values[4] = {0.25, 0.0, 0.0, 0.75}; + fill_matrix(matrix, pv, values); + + ModuleIO::write_dmr_csr(filename, &ucell, 4, &matrix, 0, 1, 2); + + const std::string output = read_file(filename); + EXPECT_THAT(output, testing::HasSubstr(" --- Ionic Step 1 ---\n")); + EXPECT_THAT(output, testing::HasSubstr(" # print density matrix in real space DM(R)\n")); + EXPECT_THAT(output, testing::HasSubstr(" 2 # number of spin directions\n")); + EXPECT_THAT(output, testing::HasSubstr(" 2 # spin index\n")); + EXPECT_THAT(output, testing::HasSubstr(" 2 # number of localized basis\n")); + EXPECT_THAT(output, testing::HasSubstr(" 1 # number of Bravais lattice vector R\n")); + + std::remove(filename.c_str()); +} diff --git a/source/source_io/module_dm/test/write_dmk_test.cpp b/source/source_io/module_dm/test/write_dmk_test.cpp index e03f0530553..42a4c181f9c 100644 --- a/source/source_io/module_dm/test/write_dmk_test.cpp +++ b/source/source_io/module_dm/test/write_dmk_test.cpp @@ -1,10 +1,3 @@ -// Pre-include every standard-library header reachable from write_dmk.h so -// their include guards are already set before '#define private public' is -// active. The macro renames the 'private'/'public' keywords, so any system -// header parsed while it is defined gets corrupted and the build fails with -// "'...__xfer_bufptrs' redeclared with different access". write_dmk.h pulls -// in indirectly via global_variable.h -> -> -// bits/quoted_string.h, so must be pre-included too. #include #include #include @@ -13,10 +6,7 @@ #include #include -#define private public #include "source_io/module_dm/write_dmk.h" -#include "source_io/module_parameter/parameter.h" -#undef private #include "source_base/global_variable.h" #include "../../test/prepare_unitcell.h" @@ -151,14 +141,14 @@ TEST(DMKTest,WriteDMK) { gen_dmk(dmk, efs, nspin, nk, nlocal, pv); gen_dmk(dmk_multik, efs, nspin, nk_multik, nlocal, pv); - PARAM.sys.global_out_dir = "./"; + const std::string out_dir = "./"; const int istep = -1; K_Vectors kv; kv.set_nkstot(1); kv.set_nkstot_nospin(1); kv.set_nks(1); - kv.spin_mult = 2; + kv.set_spin_mult(2); kv.kvec_c.resize(1); kv.kvec_c[0].x = 0.0; kv.kvec_c[0].y = 0.0; @@ -174,8 +164,8 @@ TEST(DMKTest,WriteDMK) { kv.kc_done = true; kv.kd_done = true; - ModuleIO::write_dmk(dmk, kv, 3, efs, ucell, pv, istep); - ModuleIO::write_dmk(dmk_multik, kv, 3, efs, ucell, pv, istep); + ModuleIO::write_dmk(dmk, kv, 3, efs, ucell, pv, out_dir, istep); + ModuleIO::write_dmk(dmk_multik, kv, 3, efs, ucell, pv, out_dir, istep); std::ifstream ifs; @@ -250,7 +240,6 @@ TEST(DMKTest, ReadDMK) { std::vector>> dmk_multik; Parallel_2D pv; std::vector efs; - PARAM.sys.global_out_dir = "./"; init_pv(nlocal, pv); @@ -309,7 +298,7 @@ int main(int argc, char** argv) if (GlobalV::MY_RANK == 0 && result != 0) { std::cout << "ERROR:some tests are not passed" << std::endl; - } + } MPI_Finalize(); return result; diff --git a/source/source_io/module_dm/write_dmk.cpp b/source/source_io/module_dm/write_dmk.cpp index 1d3a40b32b0..eac2ca021d5 100644 --- a/source/source_io/module_dm/write_dmk.cpp +++ b/source/source_io/module_dm/write_dmk.cpp @@ -3,7 +3,6 @@ #include "source_base/parallel_common.h" #include "source_base/module_external/scalapack_connector.h" #include "source_base/timer.h" -#include "source_io/module_parameter/parameter.h" #include "source_cell/ucell_io.h" std::string ModuleIO::dmk_gen_fname(const bool gamma_only, const int ispin, const int nspin, const int ik, const int istep) @@ -63,12 +62,12 @@ void ModuleIO::dmk_readData(std::ifstream& ifs, std::complex& data) template bool ModuleIO::read_dmk(const int nspin, - const int nk, - const K_Vectors &kv, - const Parallel_2D& pv, - const std::string& dmk_dir, - std::vector>& dmk, - std::ofstream &ofs_running) + const int nk, + const K_Vectors &kv, + const Parallel_2D& pv, + const std::string& dmk_dir, + std::vector>& dmk, + std::ofstream &ofs_running) { ModuleBase::TITLE("ModuleIO", "read_dmk"); ModuleBase::timer::start("ModuleIO", "read_dmk"); @@ -181,12 +180,13 @@ bool ModuleIO::read_dmk(const int nspin, template void ModuleIO::write_dmk(const std::vector>& dmk, - const K_Vectors &kv, - const int precision, - const std::vector& efs, - const UnitCell* ucell, - const Parallel_2D& pv, - const int istep) + const K_Vectors &kv, + const int precision, + const std::vector& efs, + const UnitCell* ucell, + const Parallel_2D& pv, + const std::string& dmk_dir, + const int istep) { ModuleBase::TITLE("ModuleIO", "write_dmk"); ModuleBase::timer::start("ModuleIO", "write_dmk"); @@ -236,8 +236,8 @@ void ModuleIO::write_dmk(const std::vector>& dmk, if (my_rank == 0) { - std::string fn = PARAM.globalv.global_out_dir - + dmk_gen_fname(gamma_only, ispin, nspin, ik, istep); + std::string fn = dmk_dir + + dmk_gen_fname(gamma_only, ispin, nspin, ik, istep); std::ofstream ofs(fn.c_str()); @@ -254,24 +254,24 @@ void ModuleIO::write_dmk(const std::vector>& dmk, } - // information about density matrix at this k-point - ofs << " " << nspin << " # number of spin directions" << std::endl; - ofs << " " << ispin+1 << " # spin index" << std::endl; - ofs << " " << kv.get_nkstot_nospin() << " # total k points " << std::endl; - ofs << " " << kv.get_nkstot() << " # total k points after symmetrized (if open) " << std::endl; - ofs << " " << ik+1 << " # k-point index " << std::endl; - ofs << " " << kv.kvec_c[ik].x << " " << kv.kvec_c[ik].y << " " << kv.kvec_c[ik].z - << " # k point coordinate (Cartesian) " << std::endl; - ofs << " " << kv.kvec_d[ik].x << " " << kv.kvec_d[ik].y << " " << kv.kvec_d[ik].z - << " # k point coordinate (direct) " << std::endl; - ofs << " " << kv.wk[ik] << " # weight of this k point" << std::endl; + // information about density matrix at this k-point + ofs << " " << nspin << " # number of spin directions" << std::endl; + ofs << " " << ispin+1 << " # spin index" << std::endl; + ofs << " " << kv.get_nkstot_nospin() << " # total k points " << std::endl; + ofs << " " << kv.get_nkstot() << " # total k points after symmetrized (if open) " << std::endl; + ofs << " " << ik+1 << " # k-point index " << std::endl; + ofs << " " << kv.kvec_c[ik].x << " " << kv.kvec_c[ik].y << " " << kv.kvec_c[ik].z + << " # k point coordinate (Cartesian) " << std::endl; + ofs << " " << kv.kvec_d[ik].x << " " << kv.kvec_d[ik].y << " " << kv.kvec_d[ik].z + << " # k point coordinate (direct) " << std::endl; + ofs << " " << kv.wk[ik] << " # weight of this k point" << std::endl; ofs << " " << efs[ispin] << " # Fermi energy in Ry " << std::endl; - ofs << " " << nlocal << " # number of localized basis " << std::endl; - ofs << " " << nlocal << " " << nlocal << " # size of this matrix " << std::endl; - ofs << std::endl; + ofs << " " << nlocal << " # number of localized basis " << std::endl; + ofs << " " << nlocal << " " << nlocal << " # size of this matrix " << std::endl; + ofs << std::endl; - // write ucell - ModuleIO::UcellIO::write_ucell(ofs, ucell); + // write ucell + ModuleIO::UcellIO::write_ucell(ofs, ucell); ofs << std::fixed; ofs << std::scientific; @@ -279,7 +279,7 @@ void ModuleIO::write_dmk(const std::vector>& dmk, ofs << std::right; for (int i = 0; i < nlocal; ++i) { - const size_t ii = i * nlocal; + const size_t ii = i * nlocal; for (int j = 0; j < nlocal; ++j) { if (std::is_same::value) @@ -321,34 +321,36 @@ void ModuleIO::write_dmk(const std::vector>& dmk, } template bool ModuleIO::read_dmk(const int nspin, - const int nk, - const K_Vectors &kv, - const Parallel_2D& pv, - const std::string& dmk_dir, - std::vector>& dmk, - std::ofstream &ofs); + const int nk, + const K_Vectors &kv, + const Parallel_2D& pv, + const std::string& dmk_dir, + std::vector>& dmk, + std::ofstream &ofs); template bool ModuleIO::read_dmk>(const int nspin, - const int nk, - const K_Vectors &kv, - const Parallel_2D& pv, - const std::string& dmk_dir, - std::vector>>& dmk, - std::ofstream &ofs); + const int nk, + const K_Vectors &kv, + const Parallel_2D& pv, + const std::string& dmk_dir, + std::vector>>& dmk, + std::ofstream &ofs); template void ModuleIO::write_dmk(const std::vector>& dmk, - const K_Vectors &kv, - const int precision, - const std::vector& efs, - const UnitCell* ucell, - const Parallel_2D& pv, - const int istep); + const K_Vectors &kv, + const int precision, + const std::vector& efs, + const UnitCell* ucell, + const Parallel_2D& pv, + const std::string& dmk_dir, + const int istep); template void ModuleIO::write_dmk>(const std::vector>>& dmk, - const K_Vectors &kv, - const int precision, - const std::vector& efs, - const UnitCell* ucell, - const Parallel_2D& pv, - const int istep); + const K_Vectors &kv, + const int precision, + const std::vector& efs, + const UnitCell* ucell, + const Parallel_2D& pv, + const std::string& dmk_dir, + const int istep); diff --git a/source/source_io/module_dm/write_dmk.h b/source/source_io/module_dm/write_dmk.h index 78a4e5ecced..231b5330da4 100644 --- a/source/source_io/module_dm/write_dmk.h +++ b/source/source_io/module_dm/write_dmk.h @@ -47,11 +47,11 @@ void dmk_readData(std::ifstream& ifs, std::complex& data); template bool read_dmk(const int nspin, const int nk, - const K_Vectors &kv, + const K_Vectors &kv, const Parallel_2D& pv, - const std::string& dmk_dir, - std::vector>& dmk, - std::ofstream &ofs_running); + const std::string& dmk_dir, + std::vector>& dmk, + std::ofstream &ofs_running); /** * @brief Writes the DMK data to a file. @@ -66,14 +66,16 @@ bool read_dmk(const int nspin, * size as the number of SPIN. * @param ucell A pointer to the UnitCell object. * @param pv The Parallel_2D object. The 2d-block parallel information of DMK. + * @param dmk_dir The directory the DMK files are written into. */ template void write_dmk(const std::vector>& dmk, - const K_Vectors &kv, + const K_Vectors &kv, const int precision, const std::vector& efs, const UnitCell* ucell, const Parallel_2D& pv, + const std::string& dmk_dir, const int istep); } // namespace ModuleIO diff --git a/source/source_io/module_dm/write_dmr.cpp b/source/source_io/module_dm/write_dmr.cpp index f9536835154..33306a43e53 100644 --- a/source/source_io/module_dm/write_dmr.cpp +++ b/source/source_io/module_dm/write_dmr.cpp @@ -33,13 +33,13 @@ std::string dmr_gen_fname(const int out_type, const int ispin, const bool append return fname; } -void write_dmr_csr(std::string& fname, +void write_dmr_csr(std::string& fname, const UnitCell *ucell, const int precision, hamilt::HContainer* dm_serial, const int istep, - const int ispin, - const int nspin) + const int ispin, + const int nspin) { // write the head: ION step number, basis number and R loop number @@ -60,10 +60,10 @@ void write_dmr_csr(std::string& fname, ofs << " # print density matrix in real space DM(R)" << std::endl; ofs << " " << nspin << " # number of spin directions" << std::endl; ofs << " " << ispin+1 << " # spin index" << std::endl; - ofs << " " << dm_serial->get_nbasis() - << " # number of localized basis" << std::endl; - ofs << " " << dm_serial->size_R_loop() - << " # number of Bravais lattice vector R" << std::endl; + ofs << " " << dm_serial->get_nbasis() + << " # number of localized basis" << std::endl; + ofs << " " << dm_serial->size_R_loop() + << " # number of Bravais lattice vector R" << std::endl; ofs << std::endl; // write ucell diff --git a/source/source_io/module_dm/write_dmr.h b/source/source_io/module_dm/write_dmr.h index 4968426311c..037d6efec61 100644 --- a/source/source_io/module_dm/write_dmr.h +++ b/source/source_io/module_dm/write_dmr.h @@ -19,10 +19,10 @@ namespace ModuleIO * @param istep The ION step (default: -1), starting from 0. * @return The generated filename as a string. */ -std::string dmr_gen_fname(const int out_type, - const int ispin, - const bool append = true, - const int istep = -1); +std::string dmr_gen_fname(const int out_type, + const int ispin, + const bool append = true, + const int istep = -1); /** * Writes HContainer to a csr file. @@ -33,13 +33,13 @@ std::string dmr_gen_fname(const int out_type, * @param dm_serial A pointer to the Hamiltonian container. * @param istep The current step number. */ -void write_dmr_csr(std::string& fname, +void write_dmr_csr(std::string& fname, const UnitCell *ucell, - const int precision, - hamilt::HContainer* dm_serial, - const int istep, - const int ispin, - const int nspin); + const int precision, + hamilt::HContainer* dm_serial, + const int istep, + const int ispin, + const int nspin); /** * Writes DMR to a file. diff --git a/source/source_io/module_dos/cal_dos.cpp b/source/source_io/module_dos/cal_dos.cpp index 2e3e7f869ce..bb6eba8b606 100644 --- a/source/source_io/module_dos/cal_dos.cpp +++ b/source/source_io/module_dos/cal_dos.cpp @@ -4,33 +4,37 @@ #include "source_base/global_function.h" #include "source_base/global_variable.h" #include "source_base/parallel_reduce.h" -#include "source_io/module_parameter/parameter.h" void ModuleIO::prepare_dos(std::ofstream& ofs_running, - const elecstate::Efermi &energy_fermi, + const elecstate::Efermi &energy_fermi, const ModuleBase::matrix& ekb, const int nks, const int nbands, - const double& dos_edelta_ev, - const double& dos_scale, - double &emax, - double &emin) + const double& dos_edelta_ev, + const double& dos_scale, + double &emax, + double &emin, + const bool dos_setemax, + const double dos_emax_ev, + const bool dos_setemin, + const double dos_emin_ev, + const bool two_fermi) { - ofs_running << "\n >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>" << std::endl; - ofs_running << " | |" << std::endl; + ofs_running << "\n >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>" << std::endl; + ofs_running << " | |" << std::endl; ofs_running << " | #Calcualte Density of States (DOS)# |" << std::endl; - ofs_running << " | DOS stands for Density of States. It represents the number of |" << std::endl; - ofs_running << " | available electronic states per unit energy range. |" << std::endl; - ofs_running << " | By analyzing the DOS, we can gain insights into how electrons are |" << std::endl; - ofs_running << " | distributed among different energy levels within the material. |" << std::endl; - ofs_running << " | |" << std::endl; - ofs_running << " >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>" << std::endl; + ofs_running << " | DOS stands for Density of States. It represents the number of |" << std::endl; + ofs_running << " | available electronic states per unit energy range. |" << std::endl; + ofs_running << " | By analyzing the DOS, we can gain insights into how electrons are |" << std::endl; + ofs_running << " | distributed among different energy levels within the material. |" << std::endl; + ofs_running << " | |" << std::endl; + ofs_running << " >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>" << std::endl; ofs_running << std::setprecision(6); assert(nbands>0); - if (PARAM.globalv.two_fermi == false) + if (two_fermi == false) { ModuleBase::GlobalFunc::OUT(ofs_running, "Fermi energy (eV)", energy_fermi.ef * ModuleBase::Ry_to_eV); @@ -63,16 +67,16 @@ void ModuleIO::prepare_dos(std::ofstream& ofs_running, emax *= ModuleBase::Ry_to_eV; emin *= ModuleBase::Ry_to_eV; - if (PARAM.globalv.dos_setemax) + if (dos_setemax) { - emax = PARAM.inp.dos_emax_ev; + emax = dos_emax_ev; } - if (PARAM.globalv.dos_setemin) + if (dos_setemin) { - emin = PARAM.inp.dos_emin_ev; + emin = dos_emin_ev; } - if (!PARAM.globalv.dos_setemax && !PARAM.globalv.dos_setemin) + if (!dos_setemax && !dos_setemin) { // scale up a little bit so the end peaks are displaced better double delta = (emax - emin) * dos_scale; @@ -89,19 +93,21 @@ void ModuleIO::prepare_dos(std::ofstream& ofs_running, } bool ModuleIO::cal_dos(const int& is, // index for spin - const std::string& fn, // file name for DOS - const double& de_ev, // delta energy in ev - const double& emax_ev, // maximal energy in eV - const double& emin_ev, // minimal energy in ev. - const double& bcoeff, - const int& nks, // number of k points in this pool - const int& nkstot, // number of total kpoints - const std::vector& wk, // weight of k points - const std::vector& isk, // index of spin for each k-point - const int& nbands, // number of bands - const ModuleBase::matrix& ekb, // energy for each k point and each band - const ModuleBase::matrix& wg, // weight of k-points and bands - const int istep) // ionic step + const std::string& fn, // file name for DOS + const double& de_ev, // delta energy in ev + const double& emax_ev, // maximal energy in eV + const double& emin_ev, // minimal energy in ev. + const double& bcoeff, + const int& nks, // number of k points in this pool + const int& nkstot, // number of total kpoints + const std::vector& wk, // weight of k points + const std::vector& isk, // index of spin for each k-point + const int& nbands, // number of bands + const ModuleBase::matrix& ekb, // energy for each k point and each band + const ModuleBase::matrix& wg, // weight of k-points and bands + const int istep, // ionic step + const bool out_app_flag, + const int bndpar) { ModuleBase::TITLE("ModuleIO", "cal_dos"); @@ -109,14 +115,14 @@ bool ModuleIO::cal_dos(const int& is, // index for spin if (GlobalV::MY_RANK == 0) { - if(PARAM.inp.out_app_flag==true) - { - ofs_dos.open(fn.c_str(), std::ios::app); - } - else - { - ofs_dos.open(fn.c_str()); - } + if(out_app_flag==true) + { + ofs_dos.open(fn.c_str(), std::ios::app); + } + else + { + ofs_dos.open(fn.c_str()); + } ofs_dos << istep+1 << " # ionic step" << std::endl; } @@ -155,11 +161,12 @@ bool ModuleIO::cal_dos(const int& is, // index for spin if (GlobalV::MY_RANK == 0) { ofs_dos << npoints << " # number of points" << std::endl; - ofs_dos << "#" << std::setw(14) << "energy" - << std::setw(15) << "elec_states" - << std::setw(15) << "sum_states" - << std::setw(15) << "states_smear" - << std::setw(15) << "sum_states" << std::endl; + ofs_dos << std::fixed << std::setprecision(6); + ofs_dos << "#" << std::setw(12) << "energy(eV)" + << std::setw(12) << "dos" + << std::setw(12) << "dos_int" + << std::setw(17) << "dos_smear(1/eV)" + << std::setw(15) << "dos_smear_int" << std::endl; } std::vector e_mod(npoints, 0.0); @@ -194,7 +201,7 @@ bool ModuleIO::cal_dos(const int& is, // index for spin } #ifdef __MPI - const int npool = GlobalV::KPAR * PARAM.inp.bndpar; + const int npool = GlobalV::KPAR * bndpar; Parallel_Reduce::reduce_double_allpool(npool, GlobalV::NPROC_IN_POOL, nstates); #endif @@ -239,11 +246,11 @@ bool ModuleIO::cal_dos(const int& is, // index for spin } sum2 += dos_smear[i] * de_ev; - ofs_dos << std::setw(15) << ene[i] - << std::setw(15) << dos[i] - << std::setw(15) << sum_elec[i] - << std::setw(15) << dos_smear[i] - << std::setw(15) << sum2 << std::endl; + ofs_dos << std::setw(13) << ene[i] + << " " << std::setw(12) << dos[i] + << " " << std::setw(12) << sum_elec[i] + << " " << std::setw(14) << dos_smear[i] + << " " << std::setw(14) << sum2 << std::endl; } } diff --git a/source/source_io/module_dos/cal_dos.h b/source/source_io/module_dos/cal_dos.h index c56891f6dbf..d133c4e9423 100644 --- a/source/source_io/module_dos/cal_dos.h +++ b/source/source_io/module_dos/cal_dos.h @@ -8,32 +8,38 @@ namespace ModuleIO { - void prepare_dos(std::ofstream& ofs_running, - const elecstate::Efermi &energy_fermi, - const ModuleBase::matrix& ekb, - const int nks, - const int nbands, - const double& dos_edelta_ev, + void prepare_dos(std::ofstream& ofs_running, + const elecstate::Efermi &energy_fermi, + const ModuleBase::matrix& ekb, + const int nks, + const int nbands, + const double& dos_edelta_ev, const double& dos_scale, - double &emax, - double &emin); - - bool cal_dos(const int &is, - const std::string &fn,// file address for DOS. - const double &de_ev, // delta energy in ev. - const double &emax_ev,// maximal energy in ev. - const double &emin_ev,// minimal energy in ev. - const double &bcoeff, - const int &nks, //number of k points - const int &nkstot, // total number of k points - const std::vector &wk,//weight of k points - const std::vector &isk, - const int &nbands,// number of bands - const ModuleBase::matrix &ekb, //store energy for each k point and each band - const ModuleBase::matrix &wg, //weight of (kpoint,bands)) - const int istep_in); // ionic_step + double &emax, + double &emin, + const bool dos_setemax, + const double dos_emax_ev, + const bool dos_setemin, + const double dos_emin_ev, + const bool two_fermi); + bool cal_dos(const int &is, + const std::string &fn,// file address for DOS. + const double &de_ev, // delta energy in ev. + const double &emax_ev,// maximal energy in ev. + const double &emin_ev,// minimal energy in ev. + const double &bcoeff, + const int &nks, //number of k points + const int &nkstot, // total number of k points + const std::vector &wk,//weight of k points + const std::vector &isk, + const int &nbands,// number of bands + const ModuleBase::matrix &ekb, //store energy for each k point and each band + const ModuleBase::matrix &wg, //weight of (kpoint,bands)) + const int istep_in, // ionic_step + const bool out_app_flag, // whether to append output + const int bndpar); // number of band parallel groups } -#endif +#endif diff --git a/source/source_io/module_dos/cal_ldos.cpp b/source/source_io/module_dos/cal_ldos.cpp index b6ae29fba84..d43d843ae61 100644 --- a/source/source_io/module_dos/cal_ldos.cpp +++ b/source/source_io/module_dos/cal_ldos.cpp @@ -2,7 +2,7 @@ #include "cal_dos.h" #include "../module_output/cube_io.h" -#include "source_estate/module_dm/cal_dm_psi.h" +#include "source_estate/module_dm/dm_from_psi.h" #include "source_hamilt/module_gint/gint_interface.h" #include "source_base/module_device/memory_op.h" @@ -16,19 +16,25 @@ template void Cal_ldos::cal_ldos_lcao( const elecstate::Efermi &eferm, // mohan add 2025-11-02 const Charge &chr, // mohan add add 2025-11-02 - const LCAO_domain::Setup_DM &dmat, // mohan add 2025-11-02 + const module_dm::Setup_DM &dmat, // mohan add 2025-11-02 const K_Vectors &kv, // k points, mohan add 2025-11-02 const ModuleBase::matrix &ekb, // mohan add 2025-11-02 const ModuleBase::matrix &wg, // mohan add 2025-11-02 const psi::Psi& psi, const Parallel_Grid& pgrid, + const Parallel_Orbitals& pv, const Grid_Driver& grid_driver, - const UnitCell& ucell) + const UnitCell& ucell, + const std::vector& stm_bias, + const int nspin, + const std::string& global_out_dir, + const bool two_fermi, + const int out_ldos_precision) { - for (int ie = 0; ie < PARAM.inp.stm_bias[2]; ie++) + for (int ie = 0; ie < stm_bias[2]; ie++) { // energy range for ldos (efermi as reference) - const double en = PARAM.inp.stm_bias[0] + ie * PARAM.inp.stm_bias[1]; + const double en = stm_bias[0] + ie * stm_bias[1]; const double emin = en < 0 ? en : 0; const double emax = en > 0 ? en : 0; @@ -49,26 +55,26 @@ void Cal_ldos::cal_ldos_lcao( } // calculate dm-like for ldos - const int nspin_dm = PARAM.inp.nspin == 2 ? 2 : 1; - elecstate::DensityMatrix dm_ldos(dmat.dm->get_paraV_pointer(), + const int nspin_dm = nspin == 2 ? 2 : 1; + module_dm::DensityMatrix dm_ldos(&pv, nspin_dm, kv.kvec_d, kv.get_nks() / nspin_dm); - elecstate::cal_dm_psi(dmat.dm->get_paraV_pointer(), weight, psi, dm_ldos); - dm_ldos.init_DMR(&grid_driver, &ucell); - dm_ldos.cal_DMR(); + module_dm::dm_from_psi(&pv, weight, psi, dm_ldos); + dm_ldos.init_dmr(&grid_driver, &ucell); + dm_ldos.cal_dmr(-1); // allocate ldos space - std::vector ldos_space(PARAM.inp.nspin * chr.nrxx); - double** ldos = new double*[PARAM.inp.nspin]; - for (int is = 0; is < PARAM.inp.nspin; ++is) + std::vector ldos_space(nspin * chr.nrxx); + double** ldos = new double*[nspin]; + for (int is = 0; is < nspin; ++is) { ldos[is] = &ldos_space[is * chr.nrxx]; } // calculate ldos - ModuleGint::cal_gint_rho(dm_ldos.get_DMR_vector(), PARAM.inp.nspin, ldos); + ModuleGint::cal_gint_rho(dm_ldos.get_dmr_vec(), nspin, ldos); // I'm not sure whether ldos should be output for each spin or not // ldos[0] += ldos[1] for nspin_dm == 2 @@ -79,21 +85,21 @@ void Cal_ldos::cal_ldos_lcao( // write ldos to cube file std::stringstream fn; - fn << PARAM.globalv.global_out_dir << "LDOS_" << en << "eV" + fn << global_out_dir << "LDOS_" << en << "eV" << ".cube"; - const int precision = PARAM.inp.out_ldos[1]; + const int precision = out_ldos_precision; ModuleIO::write_vdata_palgrid(pgrid, ldos_space.data(), 0, - PARAM.inp.nspin, + nspin, 0, fn.str(), 0, &ucell, precision, 0, - PARAM.globalv.two_fermi, + two_fermi, false); // free memory @@ -112,15 +118,35 @@ void cal_ldos_pw(const elecstate::ElecStatePW>* pelec, const psi::Psi, Device>& psi, const Device* ctx, const Parallel_Grid& pgrid, - const UnitCell& ucell) + const UnitCell& ucell, + const std::vector& out_ldos, + const std::vector& stm_bias, + const int nspin, + const std::string& global_out_dir, + const bool two_fermi, + const int nbands, + const double dos_edelta_ev, + const double dos_scale, + const bool dos_setemax, + const double dos_emax_ev, + const bool dos_setemin, + const double dos_emin_ev, + const double dos_sigma, + const std::vector& ldos_line) { - if (PARAM.inp.out_ldos[0] == 1 || PARAM.inp.out_ldos[0] == 3) + if (out_ldos[0] == 1 || out_ldos[0] == 3) { - ModuleIO::stm_mode_pw(pelec, psi, ctx, pgrid, ucell); + const int out_ldos_precision = out_ldos[1]; + ModuleIO::stm_mode_pw(pelec, psi, ctx, pgrid, ucell, + stm_bias, nspin, global_out_dir, two_fermi, + out_ldos_precision); } - if (PARAM.inp.out_ldos[0] == 2 || PARAM.inp.out_ldos[0] == 3) + if (out_ldos[0] == 2 || out_ldos[0] == 3) { - ModuleIO::ldos_mode_pw(pelec, psi, ctx, pgrid, ucell); + ModuleIO::ldos_mode_pw(pelec, psi, ctx, pgrid, ucell, + global_out_dir, nbands, dos_edelta_ev, dos_scale, + dos_setemax, dos_emax_ev, dos_setemin, dos_emin_ev, + two_fermi, dos_sigma, ldos_line); } } @@ -129,12 +155,17 @@ void stm_mode_pw(const elecstate::ElecStatePW>* pelec, const psi::Psi, Device>& psi, const Device* ctx, const Parallel_Grid& pgrid, - const UnitCell& ucell) + const UnitCell& ucell, + const std::vector& stm_bias, + const int nspin, + const std::string& global_out_dir, + const bool two_fermi, + const int out_ldos_precision) { - for (int ie = 0; ie < PARAM.inp.stm_bias[2]; ie++) + for (int ie = 0; ie < stm_bias[2]; ie++) { // energy range for ldos (efermi as reference) - const double en = PARAM.inp.stm_bias[0] + ie * PARAM.inp.stm_bias[1]; + const double en = stm_bias[0] + ie * stm_bias[1]; const double emin = en < 0 ? en : 0; const double emax = en > 0 ? en : 0; @@ -190,11 +221,11 @@ void stm_mode_pw(const elecstate::ElecStatePW>* pelec, } std::stringstream fn; - fn << PARAM.globalv.global_out_dir << "LDOS_" << en << "eV" + fn << global_out_dir << "LDOS_" << en << "eV" << ".cube"; - const int precision = PARAM.inp.out_ldos[1]; - ModuleIO::write_vdata_palgrid(pgrid, ldos.data(), 0, PARAM.inp.nspin, 0, fn.str(), 0, &ucell, precision, 0, PARAM.globalv.two_fermi, false); + const int precision = out_ldos_precision; + ModuleIO::write_vdata_palgrid(pgrid, ldos.data(), 0, nspin, 0, fn.str(), 0, &ucell, precision, 0, two_fermi, false); } } @@ -203,7 +234,18 @@ void ldos_mode_pw(const elecstate::ElecStatePW>* pelec, const psi::Psi, Device>& psi, const Device* ctx, const Parallel_Grid& pgrid, - const UnitCell& ucell) + const UnitCell& ucell, + const std::string& global_out_dir, + const int nbands, + const double dos_edelta_ev, + const double dos_scale, + const bool dos_setemax, + const double dos_emax_ev, + const bool dos_setemin, + const double dos_emin_ev, + const bool two_fermi, + const double dos_sigma, + const std::vector& ldos_line) { double emax = 0.0; double emin = 0.0; @@ -212,20 +254,25 @@ void ldos_mode_pw(const elecstate::ElecStatePW>* pelec, pelec->eferm, pelec->ekb, pelec->klist->get_nks(), - PARAM.inp.nbands, - PARAM.inp.dos_edelta_ev, - PARAM.inp.dos_scale, + nbands, + dos_edelta_ev, + dos_scale, emax, - emin); - - const int ndata = static_cast((emax - emin) / PARAM.inp.dos_edelta_ev) + 1; - const double sigma = sqrt(2.0) * PARAM.inp.dos_sigma; + emin, + dos_setemax, + dos_emax_ev, + dos_setemin, + dos_emin_ev, + two_fermi); + + const int ndata = static_cast((emax - emin) / dos_edelta_ev) + 1; + const double sigma = sqrt(2.0) * dos_sigma; const double sigma2 = sigma * sigma; const double sigma_PI = sqrt(ModuleBase::PI) * sigma; - std::vector start = {PARAM.inp.ldos_line[0], PARAM.inp.ldos_line[1], PARAM.inp.ldos_line[2]}; - std::vector end = {PARAM.inp.ldos_line[3], PARAM.inp.ldos_line[4], PARAM.inp.ldos_line[5]}; - const int npoints = PARAM.inp.ldos_line[6]; + std::vector start = {ldos_line[0], ldos_line[1], ldos_line[2]}; + std::vector end = {ldos_line[3], ldos_line[4], ldos_line[5]}; + const int npoints = ldos_line[6]; // calculate grid points std::vector> points(npoints, std::vector(3, 0)); @@ -279,7 +326,7 @@ void ldos_mode_pw(const elecstate::ElecStatePW>* pelec, for (int ie = 0; ie < ndata; ++ie) { - const double en = emin + ie * PARAM.inp.dos_edelta_ev; + const double en = emin + ie * dos_edelta_ev; const double de = en - eigenval; const double de2 = de * de; const double gauss = exp(-de2 / sigma2) / sigma_PI; @@ -299,7 +346,7 @@ void ldos_mode_pw(const elecstate::ElecStatePW>* pelec, std::ofstream ofs_ldos; std::stringstream fn; - fn << PARAM.globalv.global_out_dir << "LDOS.txt"; + fn << global_out_dir << "LDOS.txt"; if (GlobalV::MY_RANK == 0) { ofs_ldos.open(fn.str().c_str()); @@ -429,13 +476,41 @@ template void cal_ldos_pw(const elecstate::ElecStatePW< const psi::Psi, base_device::DEVICE_CPU>& psi, const base_device::DEVICE_CPU* ctx, const Parallel_Grid& pgrid, - const UnitCell& ucell); + const UnitCell& ucell, + const std::vector& out_ldos, + const std::vector& stm_bias, + const int nspin, + const std::string& global_out_dir, + const bool two_fermi, + const int nbands, + const double dos_edelta_ev, + const double dos_scale, + const bool dos_setemax, + const double dos_emax_ev, + const bool dos_setemin, + const double dos_emin_ev, + const double dos_sigma, + const std::vector& ldos_line); #if defined(__CUDA) || defined(__ROCM) template void cal_ldos_pw(const elecstate::ElecStatePW>* pelec, const psi::Psi, base_device::DEVICE_GPU>& psi, const base_device::DEVICE_GPU* ctx, const Parallel_Grid& pgrid, - const UnitCell& ucell); + const UnitCell& ucell, + const std::vector& out_ldos, + const std::vector& stm_bias, + const int nspin, + const std::string& global_out_dir, + const bool two_fermi, + const int nbands, + const double dos_edelta_ev, + const double dos_scale, + const bool dos_setemax, + const double dos_emax_ev, + const bool dos_setemin, + const double dos_emin_ev, + const double dos_sigma, + const std::vector& ldos_line); #endif } // namespace ModuleIO diff --git a/source/source_io/module_dos/cal_ldos.h b/source/source_io/module_dos/cal_ldos.h index 6b631e2aab6..8888f2d7008 100644 --- a/source/source_io/module_dos/cal_ldos.h +++ b/source/source_io/module_dos/cal_ldos.h @@ -6,7 +6,7 @@ #include "source_estate/fp_energy.h" // eferm #include "source_estate/module_charge/charge.h" // chr -#include "source_lcao/setup_dm.h" // Setup_DM +#include "source_lcao/allocate_dm.h" // module_dm::Setup_DM #include "source_cell/klist.h" // K_Vectors #include "source_cell/module_neighbor/sltk_grid_driver.h" // Grid_Driver #include "source_base/matrix.h" // matrix @@ -23,14 +23,20 @@ class Cal_ldos static void cal_ldos_lcao( const elecstate::Efermi &eferm, // mohan add 2025-11-02 const Charge &chr, // mohan add add 2025-11-02 - const LCAO_domain::Setup_DM &dmat, // mohan add 2025-11-02 + const module_dm::Setup_DM &dmat, // mohan add 2025-11-02 const K_Vectors &kv, // k points, mohan add 2025-11-02 const ModuleBase::matrix &ekb, // mohan add 2025-11-02 const ModuleBase::matrix &wg, // mohan add 2025-11-02 const psi::Psi& psi, const Parallel_Grid& pgrid, + const Parallel_Orbitals& pv, const Grid_Driver& grid_driver, - const UnitCell& ucell); + const UnitCell& ucell, + const std::vector& stm_bias, + const int nspin, + const std::string& global_out_dir, + const bool two_fermi, + const int out_ldos_precision); }; // namespace Cal_ldos @@ -39,21 +45,51 @@ void cal_ldos_pw(const elecstate::ElecStatePW>* pelec, const psi::Psi, Device>& psi, const Device* ctx, const Parallel_Grid& pgrid, - const UnitCell& ucell); + const UnitCell& ucell, + const std::vector& out_ldos, + const std::vector& stm_bias, + const int nspin, + const std::string& global_out_dir, + const bool two_fermi, + const int nbands, + const double dos_edelta_ev, + const double dos_scale, + const bool dos_setemax, + const double dos_emax_ev, + const bool dos_setemin, + const double dos_emin_ev, + const double dos_sigma, + const std::vector& ldos_line); template void stm_mode_pw(const elecstate::ElecStatePW>* pelec, const psi::Psi, Device>& psi, const Device* ctx, const Parallel_Grid& pgrid, - const UnitCell& ucell); + const UnitCell& ucell, + const std::vector& stm_bias, + const int nspin, + const std::string& global_out_dir, + const bool two_fermi, + const int out_ldos_precision); template void ldos_mode_pw(const elecstate::ElecStatePW>* pelec, const psi::Psi, Device>& psi, const Device* ctx, const Parallel_Grid& pgrid, - const UnitCell& ucell); + const UnitCell& ucell, + const std::string& global_out_dir, + const int nbands, + const double dos_edelta_ev, + const double dos_scale, + const bool dos_setemax, + const double dos_emax_ev, + const bool dos_setemin, + const double dos_emin_ev, + const bool two_fermi, + const double dos_sigma, + const std::vector& ldos_line); /* * @brief Get grid points and shifts for interpolation. diff --git a/source/source_io/module_dos/cal_pdos_gamma.cpp b/source/source_io/module_dos/cal_pdos_gamma.cpp index 50acad4f265..1bb5273b0ed 100644 --- a/source/source_io/module_dos/cal_pdos_gamma.cpp +++ b/source/source_io/module_dos/cal_pdos_gamma.cpp @@ -1,5 +1,5 @@ #include "cal_pdos_gamma.h" -#include "source_io/module_parameter/parameter.h" // use PARAM +#include "write_pdos_text.h" #include "source_base/parallel_reduce.h" #include "source_base/module_external/blas_connector.h" #include "source_base/module_external/scalapack_connector.h" @@ -21,7 +21,11 @@ void ModuleIO::cal_pdos( const double& emax, const double& emin, const double& dos_edelta_ev, - const double& bcoeff) + const double& bcoeff, + const int istep, + const int nlocal, + const int nspin, + const std::string& global_out_dir) { ModuleBase::TITLE("ModuleIO", "cal_pdos_gamma"); @@ -29,19 +33,17 @@ void ModuleIO::cal_pdos( assert(emax>=emin); assert(dos_edelta_ev>0.0); - const int npoints = static_cast(std::floor((emax - emin) / dos_edelta_ev)); - const int nlocal = PARAM.globalv.nlocal; - - // PDOS calculated from each processor - ModuleBase::matrix* pdosk = new ModuleBase::matrix[nspin0]; + const int npoints = static_cast(std::floor((emax - emin) / dos_edelta_ev)) + 1; + // PDOS calculated locally on each processor + std::vector pdosk(nspin0); for (int is = 0; is < nspin0; ++is) { pdosk[is].create(nlocal, npoints, true); } - // PDOS after MPI_reduce - ModuleBase::matrix* pdos = new ModuleBase::matrix[nspin0]; + // PDOS after MPI reduction + std::vector pdos(nspin0); for (int is = 0; is < nspin0; ++is) { pdos[is].create(nlocal, npoints, true); @@ -50,9 +52,8 @@ void ModuleIO::cal_pdos( const double a = bcoeff; const double b = sqrt(ModuleBase::TWO_PI) * a; - std::complex* waveg = new std::complex[nlocal]; - - double* gauss = new double[npoints]; + std::vector> waveg(nlocal); + std::vector gauss(npoints); for (int is = 0; is < nspin0; ++is) { @@ -64,10 +65,7 @@ void ModuleIO::cal_pdos( const double* ppsi = psi->get_pointer(); for (int i = 0; i < nbands; ++i) { - ModuleBase::GlobalFunc::ZEROS(waveg, nlocal); - - // Gauss smearing for each point - ModuleBase::GlobalFunc::ZEROS(gauss, npoints); + // Gauss smearing for each energy point for (int n = 0; n < npoints; ++n) { double en = emin + n * dos_edelta_ev; @@ -84,11 +82,9 @@ void ModuleIO::cal_pdos( const int one_int = 1; const double* sk = dynamic_cast*>(p_ham)->getSk(); - //const double* sk = nullptr; #ifdef __MPI const char T_char = 'T'; - const int nlocal = PARAM.globalv.nlocal; pdgemv_(&T_char, &nlocal, &nlocal, @@ -108,6 +104,22 @@ void ModuleIO::cal_pdos( &nb, pv.desc, &one_int); +#else + // Serial fallback: mulk = S^T * psi (column i of the wavefunction), + // written into column i of mulk to match the pdgemv_ layout above + const char T_char = 'T'; + double* mulk_col = mulk[0].c + static_cast(i) * pv.nrow; + BlasConnector::gemv(T_char, + nlocal, + nlocal, + one_float, + sk, + nlocal, + ppsi + static_cast(i) * nlocal, + one_int, + zero_float, + mulk_col, + one_int); #endif for (int j = 0; j < nlocal; ++j) @@ -119,169 +131,24 @@ void ModuleIO::cal_pdos( const int ic = pv.global2local_col(i); waveg[j] = mulk[0](ic, ir) * psi[0](ic, ir); const double x = waveg[j].real(); - BlasConnector::axpy(npoints, x, gauss, 1, pdosk[is].c + j * pdosk[is].nc, 1); + BlasConnector::axpy(npoints, x, gauss.data(), 1, pdosk[is].c + j * pdosk[is].nc, 1); } } } // ib + // reduce the local results into pdos[is] on rank 0 + const int num = nlocal * npoints; #ifdef __MPI - // reduce the results into pdos[is].c - const int num = PARAM.globalv.nlocal * npoints; MPI_Reduce(pdosk[is].c, pdos[is].c, num, MPI_DOUBLE, MPI_SUM, 0, MPI_COMM_WORLD); +#else + std::copy(pdosk[is].c, pdosk[is].c + num, pdos[is].c); #endif } // is - delete[] pdosk; - delete[] waveg; - delete[] gauss; - if (GlobalV::MY_RANK == 0) { - print_tdos_gamma(pdos, nlocal, npoints, emin, dos_edelta_ev); - print_pdos_gamma(ucell, pdos, nlocal, npoints, emin, dos_edelta_ev); + write_pdos_text(ucell, pdos.data(), nspin, nlocal, npoints, + emin, dos_edelta_ev, global_out_dir, "nao", istep); ModuleIO::write_orb_info(&ucell); } - - delete[] pdos; -} - - -void ModuleIO::print_tdos_gamma( - const ModuleBase::matrix* pdos, - const int nlocal, - const int npoints, - const double& emin, - const double& dos_edelta_ev) -{ - ModuleBase::TITLE("ModuleIO", "print_tdos_gamma"); - - // file name - std::stringstream ps; - ps << PARAM.globalv.global_out_dir << "TDOS.dat"; - std::ofstream ofs(ps.str().c_str()); - - if (PARAM.inp.nspin == 1 || PARAM.inp.nspin == 4) - { - for (int in = 0; in < npoints; ++in) - { - double dos1 = 0.0; - double en = emin + in * dos_edelta_ev; - for (int iw = 0; iw < nlocal; iw++) - { - dos1 += pdos[0](iw, in); - } - - ofs << std::setw(20) << en - << std::setw(20) << dos1 << std::endl; - } - } - else if (PARAM.inp.nspin == 2) - { - for (int in = 0; in < npoints; ++in) - { - double dos1 = 0.0; - double dos2 = 0.0; - double en = emin + in * dos_edelta_ev; - for (int iw = 0; iw < nlocal; iw++) - { - dos1 += pdos[0](iw, in); - dos2 += pdos[1](iw, in); - } - - ofs << std::setw(20) << en - << std::setw(20) << dos1 - << std::setw(20) << dos2 << std::endl; - } - } - ofs.close(); -} - -void ModuleIO::print_pdos_gamma( - const UnitCell& ucell, - const ModuleBase::matrix* pdos, - const int nlocal, - const int npoints, - const double& emin, - const double& dos_edelta_ev) -{ - ModuleBase::TITLE("ModuleIO", "print_pdos_gamma"); - - std::stringstream as; - as << PARAM.globalv.global_out_dir << "PDOS.dat"; - std::ofstream ofs(as.str().c_str()); - - ofs << "" << std::endl; - ofs << "" << PARAM.inp.nspin << "" << std::endl; - - if (PARAM.inp.nspin == 4) - { - ofs << "" << std::setw(2) << nlocal / 2 << "" << std::endl; - } - else - { - ofs << "" << std::setw(2) << nlocal << "" << std::endl; - } - ofs << "" << std::endl; - - for (int n = 0; n < npoints; ++n) - { - double y = 0.0; - double en = emin + n * dos_edelta_ev; - ofs << std::setw(20) << en << std::endl; - } - - ofs << "" << std::endl; - for (int i = 0; i < ucell.nat; i++) - { - int a = ucell.iat2ia[i]; - int t = ucell.iat2it[i]; - Atom* atom1 = &ucell.atoms[t]; - const int s0 = ucell.itiaiw2iwt(t, a, 0); - for (int j = 0; j < atom1->nw; ++j) - { - const int L1 = atom1->iw2l[j]; - const int N1 = atom1->iw2n[j]; - const int m1 = atom1->iw2m[j]; - const int w = ucell.itiaiw2iwt(t, a, j); - - ofs << "" << std::endl; - ofs << "" << std::endl; - if (PARAM.inp.nspin == 1) - { - for (int n = 0; n < npoints; ++n) - { - - ofs << std::setw(13) << pdos[0](w, n) << std::endl; - } - } - else if (PARAM.inp.nspin == 2) - { - for (int n = 0; n < npoints; ++n) - { - ofs << std::setw(20) << pdos[0](w, n) << std::setw(30) << pdos[1](w, n) << std::endl; - } - } - else if (PARAM.inp.nspin == 4) - { - int w0 = w - s0; - for (int n = 0; n < npoints; ++n) - { - ofs << std::setw(20) << pdos[0](s0 + 2 * w0, n) + pdos[0](s0 + 2 * w0 + 1, n) << std::endl; - } - } - - ofs << "" << std::endl; - ofs << "" << std::endl; - } - } - - ofs << "" << std::endl; - ofs.close(); } diff --git a/source/source_io/module_dos/cal_pdos_gamma.h b/source/source_io/module_dos/cal_pdos_gamma.h index bde8702e1b0..e5122464262 100644 --- a/source/source_io/module_dos/cal_pdos_gamma.h +++ b/source/source_io/module_dos/cal_pdos_gamma.h @@ -6,39 +6,29 @@ #include "source_psi/psi.h" // use psi::Psi #include "source_hamilt/hamilt.h" // use hamilt::Hamilt #include "source_basis/module_ao/parallel_orbitals.h" // use Parallel_Orbitals +#include namespace ModuleIO { - void cal_pdos( - const psi::Psi* psi, - hamilt::Hamilt* p_ham, - const Parallel_Orbitals& pv, - const UnitCell& ucell, - const K_Vectors& kv, - const int nspin0, - const int nbands, - const ModuleBase::matrix& ekb, - const double& emax, - const double& emin, - const double& dos_edelta_ev, - const double& bcoeff); - - void print_tdos_gamma( - const ModuleBase::matrix* pdos, - const int nlocal, - const int npoints, - const double& emin, - const double& dos_edelta_ev); - - void print_pdos_gamma( - const UnitCell& ucell, - const ModuleBase::matrix* pdos, - const int nlocal, - const int npoints, - const double& emin, - const double& dos_edelta_ev); + void cal_pdos( + const psi::Psi* psi, + hamilt::Hamilt* p_ham, + const Parallel_Orbitals& pv, + const UnitCell& ucell, + const K_Vectors& kv, + const int nspin0, + const int nbands, + const ModuleBase::matrix& ekb, + const double& emax, + const double& emin, + const double& dos_edelta_ev, + const double& bcoeff, + const int istep, + const int nlocal, + const int nspin, + const std::string& global_out_dir); } -#endif +#endif diff --git a/source/source_io/module_dos/cal_pdos_multik.cpp b/source/source_io/module_dos/cal_pdos_multik.cpp index b7766b22f4a..c20668f4927 100644 --- a/source/source_io/module_dos/cal_pdos_multik.cpp +++ b/source/source_io/module_dos/cal_pdos_multik.cpp @@ -1,5 +1,5 @@ #include "cal_pdos_multik.h" -#include "source_io/module_parameter/parameter.h" // use PARAM +#include "write_pdos_text.h" #include "source_base/parallel_reduce.h" #include "source_base/module_external/blas_connector.h" #include "source_base/module_external/scalapack_connector.h" @@ -20,7 +20,11 @@ void ModuleIO::cal_pdos( const double& emax, const double& emin, const double& dos_edelta_ev, - const double& bcoeff) + const double& bcoeff, + const int istep, + const int nlocal, + const int nspin, + const std::string& global_out_dir) { ModuleBase::TITLE("ModuleIO", "cal_pdos_multik"); @@ -28,290 +32,163 @@ void ModuleIO::cal_pdos( assert(emax>=emin); assert(dos_edelta_ev>0.0); - const int npoints = static_cast(std::floor((emax - emin) / dos_edelta_ev)); - const int nlocal = PARAM.globalv.nlocal; - - ModuleBase::matrix* pdosk = new ModuleBase::matrix[nspin0]; - - for (int is = 0; is < nspin0; ++is) - { - pdosk[is].create(nlocal, npoints, true); - } + const int npoints = static_cast(std::floor((emax - emin) / dos_edelta_ev)) + 1; - ModuleBase::matrix* pdos = new ModuleBase::matrix[nspin0]; + // PDOS calculated locally on each processor + std::vector pdosk(nspin0); + for (int is = 0; is < nspin0; ++is) + { + pdosk[is].create(nlocal, npoints, true); + } - for (int is = 0; is < nspin0; ++is) - { - pdos[is].create(nlocal, npoints, true); - } + // PDOS after MPI reduction + std::vector pdos(nspin0); + for (int is = 0; is < nspin0; ++is) + { + pdos[is].create(nlocal, npoints, true); + } const double a = bcoeff; const double b = sqrt(ModuleBase::TWO_PI) * a; - std::complex* waveg = new std::complex[nlocal]; - - double* Gauss = new double[npoints](); - - for (int is = 0; is < nspin0; ++is) - { - std::vector mulk; - mulk.resize(1); - mulk[0].create(pv.ncol, pv.nrow); - - for (int ik = 0; ik < kv.get_nks(); ik++) - { - - if (is == kv.isk[ik]) - { - // calculate SK for current k point - const std::complex* sk = nullptr; + std::vector> waveg(nlocal); + std::vector gauss(npoints); - // collumn-major matrix - const int hk_type = 1; - - if (PARAM.inp.nspin == 4) - { - dynamic_cast, std::complex>*>(p_ham) - ->updateSk(ik, hk_type); - sk = dynamic_cast, std::complex>*>(p_ham) - ->getSk(); - } - else - { - dynamic_cast, double>*>(p_ham) - ->updateSk(ik, hk_type); - sk = dynamic_cast, double>*>(p_ham) - ->getSk(); - } - - psi->fix_k(ik); - - psi::Psi> Dwfc(1, - psi->get_nbands(), - psi->get_nbasis(), - psi->get_nbasis(), - true); - - std::complex* p_dwfc = Dwfc.get_pointer(); - for (int index = 0; index < Dwfc.size(); ++index) - { - p_dwfc[index] = conj(psi->get_pointer()[index]); - } - - for (int i = 0; i < nbands; ++i) - { - - ModuleBase::GlobalFunc::ZEROS(waveg, nlocal); - - ModuleBase::GlobalFunc::ZEROS(Gauss, npoints); - for (int n = 0; n < npoints; ++n) - { - double en = emin + n * dos_edelta_ev; - double en0 = ekb(ik, i) * ModuleBase::Ry_to_eV; - double de = en - en0; - double de2 = 0.5 * de * de; - Gauss[n] = kv.wk[ik] * exp(-de2 / a / a) / b; - } - - const int nb = i + 1; + for (int is = 0; is < nspin0; ++is) + { + std::vector mulk; + mulk.resize(1); + mulk[0].create(pv.ncol, pv.nrow); + + for (int ik = 0; ik < kv.get_nks(); ik++) + { + + if (is == kv.isk[ik]) + { + // calculate SK for current k point + const std::complex* sk = nullptr; + + // collumn-major matrix + const int hk_type = 1; + + if (nspin == 4) + { + dynamic_cast, std::complex>*>(p_ham) + ->updateSk(ik, hk_type); + sk = dynamic_cast, std::complex>*>(p_ham) + ->getSk(); + } + else + { + dynamic_cast, double>*>(p_ham) + ->updateSk(ik, hk_type); + sk = dynamic_cast, double>*>(p_ham) + ->getSk(); + } + + psi->fix_k(ik); + + psi::Psi> Dwfc(1, + psi->get_nbands(), + psi->get_nbasis(), + psi->get_nbasis(), + true); + + std::complex* p_dwfc = Dwfc.get_pointer(); + for (int index = 0; index < Dwfc.size(); ++index) + { + p_dwfc[index] = conj(psi->get_pointer()[index]); + } + + for (int i = 0; i < nbands; ++i) + { + // Gauss smearing for each energy point + for (int n = 0; n < npoints; ++n) + { + double en = emin + n * dos_edelta_ev; + double en0 = ekb(ik, i) * ModuleBase::Ry_to_eV; + double de = en - en0; + double de2 = 0.5 * de * de; + gauss[n] = kv.wk[ik] * exp(-de2 / a / a) / b; + } + + const int nb = i + 1; + const int one_int = 1; #ifdef __MPI - const double one_float[2] = {1.0, 0.0}; + const double one_float[2] = {1.0, 0.0}; const double zero_float[2] = {0.0, 0.0}; - const int one_int = 1; - const char T_char = 'T'; - pzgemv_(&T_char, - &PARAM.globalv.nlocal, - &PARAM.globalv.nlocal, - &one_float[0], - sk, - &one_int, - &one_int, - pv.desc, - p_dwfc, - &one_int, - &nb, - pv.desc, - &one_int, - &zero_float[0], - mulk[0].c, - &one_int, - &nb, - pv.desc, - &one_int); + const char T_char = 'T'; + pzgemv_(&T_char, + &nlocal, + &nlocal, + &one_float[0], + sk, + &one_int, + &one_int, + pv.desc, + p_dwfc, + &one_int, + &nb, + pv.desc, + &one_int, + &zero_float[0], + mulk[0].c, + &one_int, + &nb, + pv.desc, + &one_int); +#else + // Serial fallback: mulk = S^T * conj(psi) (band i), + // written into column i of mulk to match the pzgemv_ layout above + const std::complex one_float(1.0, 0.0); + const std::complex zero_float(0.0, 0.0); + const char T_char = 'T'; + std::complex* mulk_col = mulk[0].c + static_cast(i) * pv.nrow; + BlasConnector::gemv(T_char, + nlocal, + nlocal, + one_float, + sk, + nlocal, + p_dwfc + static_cast(i) * nlocal, + one_int, + zero_float, + mulk_col, + one_int); #endif - for (int j = 0; j < PARAM.globalv.nlocal; ++j) - { + for (int j = 0; j < nlocal; ++j) + { + if (pv.in_this_processor(j, i)) + { + const int ir = pv.global2local_row(j); + const int ic = pv.global2local_col(i); - if (pv.in_this_processor(j, i)) - { - const int ir = pv.global2local_row(j); - const int ic = pv.global2local_col(i); + waveg[j] = mulk[0](ic, ir) * psi[0](ic, ir); + const double x = waveg[j].real(); + BlasConnector::axpy(npoints, x, gauss.data(), 1, pdosk[is].c + j * pdosk[is].nc, 1); + } + } - waveg[j] = mulk[0](ic, ir) * psi[0](ic, ir); - const double x = waveg[j].real(); - BlasConnector::axpy(npoints, x, Gauss, 1, pdosk[is].c + j * pdosk[is].nc, 1); - } - } + } // ib - } // ib - - } // if - } // ik + } // if + } // ik + // reduce the local results into pdos[is] on rank 0 + const int num = nlocal * npoints; #ifdef __MPI - // reduce the results into pdos[is].c - const int num = PARAM.globalv.nlocal * npoints; - MPI_Reduce(pdosk[is].c, pdos[is].c, num, MPI_DOUBLE, MPI_SUM, 0, MPI_COMM_WORLD); + MPI_Reduce(pdosk[is].c, pdos[is].c, num, MPI_DOUBLE, MPI_SUM, 0, MPI_COMM_WORLD); +#else + std::copy(pdosk[is].c, pdosk[is].c + num, pdos[is].c); #endif - } // is - delete[] pdosk; - delete[] waveg; - delete[] Gauss; + } // is if (GlobalV::MY_RANK == 0) { - print_tdos_multik(pdos, nlocal, npoints, emin, dos_edelta_ev); - print_pdos_multik(ucell, pdos, nlocal, npoints, emin, dos_edelta_ev); - + write_pdos_text(ucell, pdos.data(), nspin, nlocal, npoints, + emin, dos_edelta_ev, global_out_dir, "nao", istep); ModuleIO::write_orb_info(&ucell); } - - delete[] pdos; -} - - -void ModuleIO::print_tdos_multik( - const ModuleBase::matrix* pdos, - const int nlocal, - const int npoints, - const double& emin, - const double& dos_edelta_ev) -{ - - std::stringstream ps; - ps << PARAM.globalv.global_out_dir << "TDOS.dat"; - std::ofstream ofs1(ps.str().c_str()); - - if (PARAM.inp.nspin == 1 || PARAM.inp.nspin == 4) - { - - for (int n = 0; n < npoints; ++n) - { - double y = 0.0; - double en = emin + n * dos_edelta_ev; - for (int i = 0; i < nlocal; i++) - { - y += pdos[0](i, n); - } - - ofs1 << std::setw(20) << en << std::setw(30) << y << std::endl; - } - } - else if (PARAM.inp.nspin == 2) - { - for (int n = 0; n < npoints; ++n) - { - double y = 0.0; - double z = 0.0; - double en = emin + n * dos_edelta_ev; - for (int i = 0; i < nlocal; i++) - { - y += pdos[0](i, n); - z += pdos[1](i, n); - } - - ofs1 << std::setw(20) << en << std::setw(30) << y << std::setw(30) << z << std::endl; - } - } - ofs1.close(); -} - - -void ModuleIO::print_pdos_multik( - const UnitCell& ucell, - const ModuleBase::matrix* pdos, - const int nlocal, - const int npoints, - const double& emin, - const double& dos_edelta_ev) -{ - ModuleBase::TITLE("ModuleIO", "print_pdos_multik"); - - std::stringstream as; - as << PARAM.globalv.global_out_dir << "PDOS.dat"; - std::ofstream ofs2(as.str().c_str()); - - ofs2 << "" << std::endl; - ofs2 << "" << PARAM.inp.nspin << "" << std::endl; - if (PARAM.inp.nspin == 4) - { - ofs2 << "" << std::setw(2) << nlocal / 2 << "" << std::endl; - } - else - { - ofs2 << "" << std::setw(2) << nlocal << "" << std::endl; - } - ofs2 << "" << std::endl; - - for (int n = 0; n < npoints; ++n) - { - double y = 0.0; - double en = emin + n * dos_edelta_ev; - ofs2 << std::setw(20) << en << std::endl; - } - ofs2 << "" << std::endl; - for (int i = 0; i < ucell.nat; i++) - { - int a = ucell.iat2ia[i]; - int t = ucell.iat2it[i]; - Atom* atom1 = &ucell.atoms[t]; - const int s0 = ucell.itiaiw2iwt(t, a, 0); - for (int j = 0; j < atom1->nw; ++j) - { - const int L1 = atom1->iw2l[j]; - const int N1 = atom1->iw2n[j]; - const int m1 = atom1->iw2m[j]; - const int w = ucell.itiaiw2iwt(t, a, j); - - ofs2 << "" << std::endl; - ofs2 << "" << std::endl; - if (PARAM.inp.nspin == 1) - { - for (int n = 0; n < npoints; ++n) - { - ofs2 << std::setw(13) << pdos[0](w, n) << std::endl; - } - } - else if (PARAM.inp.nspin == 2) - { - for (int n = 0; n < npoints; ++n) - { - ofs2 << std::setw(20) << pdos[0](w, n) << std::setw(30) << pdos[1](w, n) << std::endl; - } - } - else if (PARAM.inp.nspin == 4) - { - int w0 = w - s0; - for (int n = 0; n < npoints; ++n) - { - ofs2 << std::setw(20) << pdos[0](s0 + 2 * w0, n) + pdos[0](s0 + 2 * w0 + 1, n) << std::endl; - } - } - - ofs2 << "" << std::endl; - ofs2 << "" << std::endl; - }// end j - }// end i - - ofs2 << "" << std::endl; - ofs2.close(); } diff --git a/source/source_io/module_dos/cal_pdos_multik.h b/source/source_io/module_dos/cal_pdos_multik.h index 2b59365c1a5..110009becb4 100644 --- a/source/source_io/module_dos/cal_pdos_multik.h +++ b/source/source_io/module_dos/cal_pdos_multik.h @@ -6,40 +6,30 @@ #include "source_psi/psi.h" // use psi::Psi #include "source_hamilt/hamilt.h" // use hamilt::Hamilt #include "source_basis/module_ao/parallel_orbitals.h" // use Parallel_Orbitals +#include namespace ModuleIO { // pdos for multi-k point - void cal_pdos( - const psi::Psi>* psi, - hamilt::Hamilt>* p_ham, - const Parallel_Orbitals& pv, - const UnitCell& ucell, - const K_Vectors& kv, - const int nspin0, - const int nbands, - const ModuleBase::matrix& ekb, - const double& emax, - const double& emin, - const double& dos_edelta_ev, - const double& bcoeff); - - void print_tdos_multik( - const ModuleBase::matrix* pdos, - const int nlocal, - const int npoints, - const double& emin, - const double& dos_edelta_ev); - - void print_pdos_multik( - const UnitCell& ucell, - const ModuleBase::matrix* pdos, - const int nlocal, - const int npoints, - const double& emin, - const double& dos_edelta_ev); + void cal_pdos( + const psi::Psi>* psi, + hamilt::Hamilt>* p_ham, + const Parallel_Orbitals& pv, + const UnitCell& ucell, + const K_Vectors& kv, + const int nspin0, + const int nbands, + const ModuleBase::matrix& ekb, + const double& emax, + const double& emin, + const double& dos_edelta_ev, + const double& bcoeff, + const int istep, + const int nlocal, + const int nspin, + const std::string& global_out_dir); } -#endif +#endif diff --git a/source/source_io/module_dos/unittests/CMakeLists.txt b/source/source_io/module_dos/unittests/CMakeLists.txt new file mode 100644 index 00000000000..4660eec50b8 --- /dev/null +++ b/source/source_io/module_dos/unittests/CMakeLists.txt @@ -0,0 +1,37 @@ +abacus_disable_feature_definitions(__MPI) +abacus_disable_feature_definitions(__EXX) +abacus_disable_feature_definitions(__CUDA) +abacus_disable_feature_definitions(__UT_USE_CUDA) +abacus_disable_feature_definitions(__UT_USE_ROCM) +abacus_disable_feature_definitions(__ROCM) +abacus_disable_feature_definitions(__MLALGO) +abacus_disable_feature_definitions(_OPENMP) + +if (ENABLE_MPI) + +AddTest( + TARGET MODULE_DOS_cal_dos + LIBS parameter base device + SOURCES test_cal_dos.cpp ../cal_dos.cpp +) + +AddTest( + TARGET MODULE_DOS_write_dos_pw + LIBS parameter base device symmetry + SOURCES test_write_dos_pw.cpp ../write_dos_pw.cpp ../cal_dos.cpp + ../../../source_cell/parallel_kpoints.cpp ../../../source_cell/klist.cpp + ../../../source_cell/klist_io.cpp ../../../source_cell/reciprocal_grid.cpp +) + +AddTest( + TARGET MODULE_DOS_write_pdos_text + LIBS parameter base device + SOURCES test_write_pdos_text.cpp ../write_pdos_text.cpp + ../../../source_cell/atom_spec.cpp + ../../../source_cell/atom_pseudo.cpp + ../../../source_cell/pseudo.cpp + ../../../source_cell/magnetism.cpp + ../../../source_cell/test/support/mock_unitcell.cpp +) + +endif() diff --git a/source/source_io/module_dos/unittests/dos_test.h b/source/source_io/module_dos/unittests/dos_test.h new file mode 100644 index 00000000000..6a7e870673d --- /dev/null +++ b/source/source_io/module_dos/unittests/dos_test.h @@ -0,0 +1,69 @@ +#ifndef DOS_TEST_H +#define DOS_TEST_H + +#include +#include + +#include "source_base/constants.h" +#include "source_base/matrix.h" + +#include "./dos_test_data.h" + +class DosPrepare +{ +public: + DosPrepare(int is_in, + std::string fa_in, + double de_ev_in, + double emax_ev_in, + double emin_ev_in, + double bcoeff_in, + int nks_in, + int nkstot_in, + int nbands_in): + is(is_in),fa(fa_in), + de_ev(de_ev_in),emax_ev(emax_ev_in), + emin_ev(emin_ev_in),bcoeff(bcoeff_in), + nks(nks_in),nkstot(nkstot_in),nbands(nbands_in){} + + int is; + std::string fa; + double de_ev; + double emax_ev; + double emin_ev; + double bcoeff; + int nks; + int nkstot; + int nbands; + std::vector isk; + std::vector wk; + ModuleBase::matrix ekb; + ModuleBase::matrix wg; + + void set_isk() + { + this->isk.resize(nks, 0); // spin-unpolarized case, only 1 spin + } + + void set_wk() + { + this->wk = dos_test_data::wk; + } + + void set_istate_info() + { + this->ekb.create(nks, nbands); + this->wg.create(nks, nbands); + for (int ik = 0; ik < nks; ++ik) + { + for (int ib = 0; ib < nbands; ++ib) + { + this->ekb(ik, ib) = dos_test_data::ekb_ev[ik * nbands + ib]; + this->wg(ik, ib) = dos_test_data::wg[ik * nbands + ib]; + } + } + this->ekb *= 1.0 / ModuleBase::Ry_to_eV; // eV -> Ry + } +}; + +#endif diff --git a/source/source_io/module_dos/unittests/dos_test_data.h b/source/source_io/module_dos/unittests/dos_test_data.h new file mode 100644 index 00000000000..b381da1bcc6 --- /dev/null +++ b/source/source_io/module_dos/unittests/dos_test_data.h @@ -0,0 +1,36 @@ +#ifndef DOS_TEST_DATA_H +#define DOS_TEST_DATA_H + +// Small synthetic test data for module_dos unit tests: +// 4 k-points with equal weight (sum of wk = 2 for spin-unpolarized), +// 4 bands. Energies are in eV. + +#include + +namespace dos_test_data +{ +const int nks = 4; +const int nkstot = 4; +const int nbands = 4; + +// k-point weights (already doubled for spin degeneracy) +const std::vector wk = {0.5, 0.5, 0.5, 0.5}; + +// band energies in eV, row-major [nks][nbands] +const std::vector ekb_ev = { + -5.0, 1.0, 3.0, 6.0, + -4.5, 1.5, 3.5, 6.5, + -4.0, 2.0, 4.0, 7.0, + -3.5, 2.5, 4.5, 7.5, +}; + +// band occupation numbers (weight included), row-major [nks][nbands] +const std::vector wg = { + 0.5, 0.0, 0.0, 0.0, + 0.5, 0.0, 0.0, 0.0, + 0.5, 0.0, 0.0, 0.0, + 0.5, 0.0, 0.0, 0.0, +}; +} // namespace dos_test_data + +#endif diff --git a/source/source_io/module_dos/unittests/for_testing_klist.h b/source/source_io/module_dos/unittests/for_testing_klist.h new file mode 100644 index 00000000000..f0a2d24f166 --- /dev/null +++ b/source/source_io/module_dos/unittests/for_testing_klist.h @@ -0,0 +1,46 @@ +#ifndef FOR_TESTING_KLIST_H +#define FOR_TESTING_KLIST_H + +#include "source_base/parallel_global.h" +#include "source_basis/module_ao/orb_gaunt_table.h" +#include "source_cell/atom_pseudo.h" +#include "source_cell/atom_spec.h" +#include "source_cell/klist.h" +#include "source_cell/parallel_kpoints.h" +#include "source_cell/pseudo.h" +#include "source_cell/unitcell.h" +#include "source_cell/magnetism.h" +#include "source_pw/module_pwdft/vl_pw.h" +#include "source_pw/module_pwdft/vnl_pw.h" +#include "source_base/parallel_grid.h" + +pseudo::pseudo(){} +pseudo::~pseudo(){} +Atom::Atom(){} +Atom::~Atom(){} +Atom_pseudo::Atom_pseudo(){} +Atom_pseudo::~Atom_pseudo(){} +UnitCell::UnitCell(){} +UnitCell::~UnitCell(){} +Magnetism::Magnetism(){} +Magnetism::~Magnetism(){} +ORB_gaunt_table::ORB_gaunt_table(){} +ORB_gaunt_table::~ORB_gaunt_table(){} +pseudopot_cell_vl::pseudopot_cell_vl(){} +pseudopot_cell_vl::~pseudopot_cell_vl(){} +pseudopot_cell_vnl::pseudopot_cell_vnl(){} +pseudopot_cell_vnl::~pseudopot_cell_vnl(){} +Soc::~Soc() +{ +} +Fcoef::~Fcoef() +{ +} +SepPot::SepPot(){} +SepPot::~SepPot(){} +Sep_Cell::Sep_Cell() noexcept {} +Sep_Cell::~Sep_Cell() noexcept {} + + + +#endif diff --git a/source/source_io/module_dos/unittests/test_cal_dos.cpp b/source/source_io/module_dos/unittests/test_cal_dos.cpp new file mode 100644 index 00000000000..62d8dd73bc6 --- /dev/null +++ b/source/source_io/module_dos/unittests/test_cal_dos.cpp @@ -0,0 +1,190 @@ +#include "gtest/gtest.h" +#include "gmock/gmock.h" +#include "../cal_dos.h" +#include "source_base/global_variable.h" +#include +#ifdef __MPI +#include "mpi.h" +#endif +#include "./dos_test.h" + +/************************************************ + * unit test of ca_dos + ***********************************************/ + +/** + * - Tested Functions: + * - cal_dos() + * - the function to calculate and print out + * - density of states + */ + + +class DosTest : public ::testing::Test +{ +protected: + std::string output; +}; + +TEST_F(DosTest,Dos) +{ + //is,fa,de_ev,emax_ev,emin_ev,bcoeff,nks,nkstot,nbands + DosPrepare dosp = DosPrepare(0,"doss1_pw.txt",0.005,18,-6,0.07,dos_test_data::nks,dos_test_data::nkstot,dos_test_data::nbands); + dosp.set_isk(); + dosp.set_wk(); + dosp.set_istate_info(); + EXPECT_EQ(dosp.is,0); + + const int istep = 1; + const bool out_app_flag = false; + const int bndpar = 1; + ModuleIO::cal_dos(dosp.is, + dosp.fa, + dosp.de_ev, + dosp.emax_ev, + dosp.emin_ev, + dosp.bcoeff, + dosp.nks, + dosp.nkstot, + dosp.wk, + dosp.isk, + dosp.nbands, + dosp.ekb, + dosp.wg, + istep, + out_app_flag, + bndpar); + +#ifdef __MPI + if(GlobalV::MY_RANK==0) + { +#endif + std::ifstream ifs; + ifs.open(dosp.fa.c_str()); + std::string str((std::istreambuf_iterator(ifs)),std::istreambuf_iterator()); + EXPECT_THAT(str, testing::HasSubstr("4801 # number of points")); + EXPECT_THAT(str, testing::HasSubstr(" -4.995000 0.500000 0.500000")); + EXPECT_THAT(str, testing::HasSubstr(" 18.000000 0.000000 8.000000")); + ifs.close(); + remove("doss1_pw.txt"); +#ifdef __MPI + } +#endif +} + + + +TEST_F(DosTest,DosW1) +{ + //is,fa,de_ev,emax_ev,emin_ev,bcoeff,nks,nkstot,nbands + DosPrepare dosp = DosPrepare(0,"doss1_pw.txt",-0.005,18,-6,0.07,dos_test_data::nks,dos_test_data::nkstot,dos_test_data::nbands); + dosp.set_isk(); + dosp.set_wk(); + dosp.set_istate_info(); + EXPECT_EQ(dosp.is,0); + EXPECT_LE(dosp.de_ev,0); + GlobalV::ofs_warning.open("warning1.log"); + + const int istep = 1; + const bool out_app_flag = false; + const int bndpar = 1; + EXPECT_NO_THROW(ModuleIO::cal_dos(dosp.is, + dosp.fa, + dosp.de_ev, + dosp.emax_ev, + dosp.emin_ev, + dosp.bcoeff, + dosp.nks, + dosp.nkstot, + dosp.wk, + dosp.isk, + dosp.nbands, + dosp.ekb, + dosp.wg, + istep, + out_app_flag, + bndpar)); + GlobalV::ofs_warning.close(); +#ifdef __MPI + if(GlobalV::MY_RANK==0) + { +#endif + std::ifstream ifs; + ifs.open("warning1.log"); + std::string str((std::istreambuf_iterator(ifs)),std::istreambuf_iterator()); + EXPECT_THAT(str, testing::HasSubstr("ModuleIO::cal_dos warning : de <= 0")); + ifs.close(); + remove("warning1.log"); + remove("doss1_pw.txt"); +#ifdef __MPI + } +#endif +} + + +TEST_F(DosTest,DosW2) +{ + //is,fa,de_ev,emax_ev,emin_ev,bcoeff,nks,nkstot,nbands + DosPrepare dosp = DosPrepare(0,"doss1_pw.txt",0.005,-6,18,0.07,dos_test_data::nks,dos_test_data::nkstot,dos_test_data::nbands); + dosp.set_isk(); + dosp.set_wk(); + dosp.set_istate_info(); + EXPECT_EQ(dosp.is,0); + GlobalV::ofs_warning.open("warning2.log"); + + const int istep = 1; + const bool out_app_flag = false; + const int bndpar = 1; + EXPECT_NO_THROW(ModuleIO::cal_dos(dosp.is, + dosp.fa, + dosp.de_ev, + dosp.emax_ev, + dosp.emin_ev, + dosp.bcoeff, + dosp.nks, + dosp.nkstot, + dosp.wk, + dosp.isk, + dosp.nbands, + dosp.ekb, + dosp.wg, + istep, + out_app_flag, + bndpar)); + + GlobalV::ofs_warning.close(); +#ifdef __MPI + if(GlobalV::MY_RANK==0) + { +#endif + std::ifstream ifs; + ifs.open("warning2.log"); + std::string str((std::istreambuf_iterator(ifs)),std::istreambuf_iterator()); + EXPECT_THAT(str, testing::HasSubstr("ModuleIO::cal_dos warning : emax_ev < emin_ev")); + ifs.close(); + remove("warning2.log"); + remove("doss1_pw.txt"); +#ifdef __MPI + } +#endif +} + +#ifdef __MPI +int main(int argc, char **argv) +{ + MPI_Init(&argc,&argv); + + testing::InitGoogleTest(&argc,argv); + MPI_Comm_size(MPI_COMM_WORLD,&GlobalV::NPROC); + MPI_Comm_rank(MPI_COMM_WORLD,&GlobalV::MY_RANK); + + // only test a certain one + //::testing::GTEST_FLAG(filter) = "DosTest.DosW1"; + + int result = RUN_ALL_TESTS(); + + MPI_Finalize(); + + return result; +} +#endif diff --git a/source/source_io/module_dos/unittests/test_write_dos_pw.cpp b/source/source_io/module_dos/unittests/test_write_dos_pw.cpp new file mode 100644 index 00000000000..4ad10916048 --- /dev/null +++ b/source/source_io/module_dos/unittests/test_write_dos_pw.cpp @@ -0,0 +1,211 @@ +#include "gtest/gtest.h" +#include "gmock/gmock.h" +#include "../write_dos_pw.h" +#ifdef __MPI +#include "mpi.h" +#endif +#include "./for_testing_klist.h" +#include "./dos_test.h" + +/************************************************ + * unit test of write_dos_pw + ***********************************************/ + +/** + * - Tested Functions: + * - write_dos_pw() + * - the function to calculate and print out + * - density of states in pw basis calculation + */ + + +class DosPWTest : public ::testing::Test +{ +protected: + K_Vectors* kv = nullptr; + ModuleBase::matrix ekb; + ModuleBase::matrix wg; + void SetUp() + { + kv = new K_Vectors; + } + void TearDown() + { + delete kv; + } +}; + +TEST_F(DosPWTest,Dos1) +{ + //is,fa,fa1,de_ev,emax_ev,emin_ev,bcoeff,nks,nkstot,nbands + DosPrepare dosp = DosPrepare(0,"doss1_pw.txt",0.005,18,-6,0.07,dos_test_data::nks,dos_test_data::nkstot,dos_test_data::nbands); + dosp.set_isk(); + dosp.set_wk(); + dosp.set_istate_info(); + EXPECT_EQ(dosp.is,0); + double dos_scale = 0.01; + kv->set_nks(dosp.nks); + kv->set_nkstot(dosp.nkstot); + kv->isk.resize(kv->get_nks()); + kv->wk.resize(kv->get_nks()); + for(int ik=0; ikget_nks(); ++ik) + { + kv->isk[ik] = dosp.isk[ik]; + kv->wk[ik] = dosp.wk[ik]; + } + + // initialize the Fermi energy + elecstate::Efermi fermi_energy; + + std::ofstream ofs("write_dos_pw.log"); + + UnitCell ucell; + + const int nspin = 1; + const int out_dos = 1; + const bool dos_setemax = true; + const bool dos_setemin = true; + const bool two_fermi = false; + const bool out_app_flag = false; + const int bndpar = 1; + const std::string global_out_dir = "./"; + + ModuleIO::write_dos_pw( + ucell, // this should be unitcell, 2025-04-12 + dosp.ekb, + dosp.wg, + *kv, + dosp.nbands, + -1, // istep_in + fermi_energy, + dosp.de_ev, + dos_scale, + dosp.bcoeff, + nspin, + out_dos, + dos_setemax, + dosp.emax_ev, + dos_setemin, + dosp.emin_ev, + two_fermi, + out_app_flag, + bndpar, + global_out_dir, + ofs); + ofs.close(); + remove("write_dos_pw.log"); + +#ifdef __MPI + if(GlobalV::MY_RANK==0) + { +#endif + std::ifstream ifs; + ifs.open("dos.txt"); + std::string str((std::istreambuf_iterator(ifs)),std::istreambuf_iterator()); + EXPECT_THAT(str, testing::HasSubstr("4801 # number of points")); + EXPECT_THAT(str, testing::HasSubstr(" -4.995000 0.500000 0.500000")); + EXPECT_THAT(str, testing::HasSubstr(" 18.000000 0.000000 8.000000")); + ifs.close(); + remove("dos.txt"); +#ifdef __MPI + } +#endif +} + + +TEST_F(DosPWTest,Dos2) +{ + //is,fa,fa1,de_ev,emax_ev,emin_ev,bcoeff,nks,nkstot,nbands + DosPrepare dosp = DosPrepare(0,"doss1_pw.txt",0.005,18,-6,0.07,dos_test_data::nks,dos_test_data::nkstot,dos_test_data::nbands); + dosp.set_isk(); + dosp.set_wk(); + dosp.set_istate_info(); + EXPECT_EQ(dosp.is,0); + double dos_scale = 0.01; + kv->set_nks(dosp.nks); + kv->set_nkstot(dosp.nkstot); + kv->isk.resize(kv->get_nks()); + kv->wk.resize(kv->get_nks()); + for(int ik=0; ikget_nks(); ++ik) + { + kv->isk[ik] = dosp.isk[ik]; + kv->wk[ik] = dosp.wk[ik]; + } + + // initialize the Fermi energy + elecstate::Efermi fermi_energy; + + std::ofstream ofs("write_dos_pw.log"); + + UnitCell ucell; + + const int nspin = 1; + const int out_dos = 1; + const bool two_fermi = false; + const bool out_app_flag = false; + const int bndpar = 1; + const std::string global_out_dir = "./"; + const bool dos_setemax = false; + const bool dos_setemin = false; + + ModuleIO::write_dos_pw( + ucell, + dosp.ekb, + dosp.wg, + *kv, + dosp.nbands, + -1, // istep_in + fermi_energy, + dosp.de_ev, + dos_scale, + dosp.bcoeff, + nspin, + out_dos, + dos_setemax, + dosp.emax_ev, + dos_setemin, + dosp.emin_ev, + two_fermi, + out_app_flag, + bndpar, + global_out_dir, + ofs); + ofs.close(); + remove("write_dos_pw.log"); + +#ifdef __MPI + if(GlobalV::MY_RANK==0) + { +#endif + std::ifstream ifs; + ifs.open("dos.txt"); + std::string str1((std::istreambuf_iterator(ifs)),std::istreambuf_iterator()); + EXPECT_THAT(str1, testing::HasSubstr("2526 # number of points")); + EXPECT_THAT(str1, testing::HasSubstr(" -4.997500 0.500000 0.500000")); + EXPECT_THAT(str1, testing::HasSubstr(" 7.567500 0.000000 8.000000")); + ifs.close(); + remove("dos.txt"); +#ifdef __MPI + } +#endif +} + +#ifdef __MPI +int main(int argc, char **argv) +{ + MPI_Init(&argc,&argv); + + testing::InitGoogleTest(&argc,argv); + MPI_Comm_size(MPI_COMM_WORLD,&GlobalV::NPROC); + MPI_Comm_rank(MPI_COMM_WORLD,&GlobalV::MY_RANK); + + // only test the second one + // ::testing::GTEST_FLAG(filter) = "DosPWTest.Dos2"; + + int result = RUN_ALL_TESTS(); + + MPI_Finalize(); + + return result; +} +#endif diff --git a/source/source_io/module_dos/unittests/test_write_pdos_text.cpp b/source/source_io/module_dos/unittests/test_write_pdos_text.cpp new file mode 100644 index 00000000000..2ac56f31727 --- /dev/null +++ b/source/source_io/module_dos/unittests/test_write_pdos_text.cpp @@ -0,0 +1,182 @@ +#include +#include +#include + +#include "gmock/gmock.h" +#include "gtest/gtest.h" + +#include "../write_pdos_text.h" +#include "source_base/global_variable.h" +#include "source_cell/unitcell.h" + +// Build a minimal UnitCell for a single Si atom with one s and one p +// zeta: nwl=1, nw=4, iw2l={0,1,1,1}, iw2m={0,0,1,2} (p: m=0,+1,-1). +// Index arrays (iat2it/iat2ia/iat2iwt/itia2iat) are filled by hand so the +// test only needs the UnitCell/Atom constructor/destructor, not the full +// source_cell library. +static UnitCell* make_si_ucell() +{ + UnitCell* ucell = new UnitCell; + ucell->ntype = 1; + ucell->nat = 1; + ucell->atoms = new Atom[1]; + ucell->set_atom_flag = true; + + Atom& atom = ucell->atoms[0]; + atom.label = "Si"; + atom.na = 1; + atom.nwl = 1; + atom.iw2l = {0, 1, 1, 1}; + atom.iw2m = {0, 0, 1, 2}; + atom.nw = 4; + atom.l_nchi = {1, 1}; + + // iat2it / iat2ia + ucell->iat2it = new int[1]; + ucell->iat2ia = new int[1]; + ucell->iat2it[0] = 0; + ucell->iat2ia[0] = 0; + + // itia2iat(it, ia) -> iat ; IntArray is (ntype, na)-indexed + ucell->itia2iat.create(ucell->ntype, atom.na); + ucell->itia2iat(0, 0) = 0; + + // iat2iwt / npol: first global orbital index of the atom + ucell->set_iat2iwt_for_test({0}, 1); + return ucell; +} + +class WritePdosTextTest : public ::testing::Test +{ +protected: + UnitCell* ucell = nullptr; + + void SetUp() override + { + ucell = make_si_ucell(); + } + void TearDown() override + { + // iat2it/iat2ia are owned by UnitCell's internal Statistics member, + // whose destructor releases them; do not delete them here. + delete ucell; + } + + std::string read_file(const std::string& fname) + { + std::ifstream ifs(fname); + std::stringstream buffer; + buffer << ifs.rdbuf(); + return buffer.str(); + } +}; + +// Spin-unpolarized: writes pdoss1_.txt with s and 3 p columns. +TEST_F(WritePdosTextTest, BasicOutput) +{ + GlobalV::ofs_running.open("test_write_pdos_text.log"); + + const int nspin = 1; + const int nlocal = 4; + const int npoints = 5; + const double emin = -1.0; + const double dos_edelta_ev = 0.5; + const std::string out_dir = "./"; + const std::string basis = "lcao"; + const int istep = -1; + + // pdos matrix: nlocal x npoints; value = orbital * 100 + point + ModuleBase::matrix pdos(nlocal, npoints); + for (int w = 0; w < nlocal; ++w) + { + for (int n = 0; n < npoints; ++n) + { + pdos(w, n) = w * 100.0 + n; + } + } + + ModuleIO::write_pdos_text(*ucell, &pdos, nspin, nlocal, npoints, + emin, dos_edelta_ev, out_dir, basis, istep); + + const std::string str = read_file("./pdoss1_lcao.txt"); + + // header lines + EXPECT_THAT(str, testing::HasSubstr("# npoints: 5")); + EXPECT_THAT(str, testing::HasSubstr("# energy(eV) atom species")); + EXPECT_THAT(str, testing::HasSubstr("s(m=0) p(m=0,+1,-1)")); + + // first data row: energy=-1.0, atom 1, Si, s=0*100+0=0, p columns + // w=1,2,3 -> 100,200,300 + EXPECT_THAT(str, testing::HasSubstr(" -1.000000 1 Si 0.000000 100.000000 200.000000 300.000000")); + // last data row: energy=1.0, n=4 -> s=4, p=104,204,304 + EXPECT_THAT(str, testing::HasSubstr(" 1.000000 1 Si 4.000000 104.000000 204.000000 304.000000")); + + remove("./pdoss1_lcao.txt"); + GlobalV::ofs_running.close(); + remove("test_write_pdos_text.log"); +} + +// istep >= 0 appends g{istep+1} to the file name. +TEST_F(WritePdosTextTest, IstepFileName) +{ + GlobalV::ofs_running.open("test_write_pdos_text.log"); + + const int nspin = 1; + const int nlocal = 4; + const int npoints = 2; + const double emin = 0.0; + const double dos_edelta_ev = 1.0; + ModuleBase::matrix pdos(nlocal, npoints); + pdos.zero_out(); + + ModuleIO::write_pdos_text(*ucell, &pdos, nspin, nlocal, npoints, + emin, dos_edelta_ev, "./", "lcao", 2); + + const std::string str = read_file("./pdoss1g3_lcao.txt"); + EXPECT_THAT(str, testing::HasSubstr("# istep: 3")); + + remove("./pdoss1g3_lcao.txt"); + GlobalV::ofs_running.close(); + remove("test_write_pdos_text.log"); +} + +// Spin-polarized (nspin=2): writes two files, pdos[is] per spin channel. +TEST_F(WritePdosTextTest, SpinPolarized) +{ + GlobalV::ofs_running.open("test_write_pdos_text.log"); + + const int nspin = 2; + const int nlocal = 4; + const int npoints = 2; + const double emin = 0.0; + const double dos_edelta_ev = 1.0; + + ModuleBase::matrix pdos_arr[2]; + pdos_arr[0].create(nlocal, npoints); + pdos_arr[1].create(nlocal, npoints); + for (int is = 0; is < 2; ++is) + { + for (int w = 0; w < nlocal; ++w) + { + for (int n = 0; n < npoints; ++n) + { + pdos_arr[is](w, n) = (is + 1) * 1000.0 + w * 100.0 + n; + } + } + } + + ModuleIO::write_pdos_text(*ucell, pdos_arr, nspin, nlocal, npoints, + emin, dos_edelta_ev, "./", "lcao", -1); + + const std::string s1 = read_file("./pdoss1_lcao.txt"); + const std::string s2 = read_file("./pdoss2_lcao.txt"); + // spin 1: s orbital w=0, n=0 -> 1000 + EXPECT_THAT(s1, testing::HasSubstr("1000.000000")); + // spin 2: s orbital w=0, n=0 -> 2000 + EXPECT_THAT(s2, testing::HasSubstr("2000.000000")); + + remove("./pdoss1_lcao.txt"); + remove("./pdoss2_lcao.txt"); + GlobalV::ofs_running.close(); + remove("test_write_pdos_text.log"); +} diff --git a/source/source_io/module_dos/write_dos_lcao.cpp b/source/source_io/module_dos/write_dos_lcao.cpp index 359b22fee4b..77ecb3c903d 100644 --- a/source/source_io/module_dos/write_dos_lcao.cpp +++ b/source/source_io/module_dos/write_dos_lcao.cpp @@ -2,8 +2,6 @@ #include "cal_dos.h" #include "cal_pdos_gamma.h" #include "cal_pdos_multik.h" -#include "source_io/module_energy/nscf_fermi_surf.h" -#include "source_io/module_parameter/parameter.h" namespace ModuleIO { @@ -11,12 +9,12 @@ namespace ModuleIO template void write_dos_lcao( const psi::Psi* psi, - hamilt::Hamilt* p_ham, + hamilt::Hamilt* p_ham, const Parallel_Orbitals &pv, const UnitCell& ucell, - const K_Vectors& kv, - const int nbands, - const elecstate::Efermi &energy_fermi, + const K_Vectors& kv, + const int nbands, + const elecstate::Efermi &energy_fermi, const ModuleBase::matrix& ekb, const ModuleBase::matrix& wg, const double& dos_edelta_ev, @@ -24,81 +22,89 @@ void write_dos_lcao( const double& bcoeff, const bool out_app_flag, const int istep, + const int nspin, + const int nlocal, + const bool dos_setemax, + const double dos_emax_ev, + const bool dos_setemin, + const double dos_emin_ev, + const bool two_fermi, + const int bndpar, + const std::string& global_out_dir, std::ofstream &ofs_running) { ModuleBase::TITLE("ModuleIO", "write_dos_lcao"); - const int nspin0 = (PARAM.inp.nspin == 2) ? 2 : 1; + const int nspin0 = (nspin == 2) ? 2 : 1; double emax = 0.0; double emin = 0.0; - prepare_dos(ofs_running, - energy_fermi, - ekb, - kv.get_nks(), - nbands, - dos_edelta_ev, - dos_scale, - emax, - emin); + prepare_dos(ofs_running, + energy_fermi, + ekb, + kv.get_nks(), + nbands, + dos_edelta_ev, + dos_scale, + emax, + emin, + dos_setemax, + dos_emax_ev, + dos_setemin, + dos_emin_ev, + two_fermi); // output the DOS file. for (int is = 0; is < nspin0; ++is) { std::stringstream ss; - ss << PARAM.globalv.global_out_dir << "doss" << is + 1; + ss << global_out_dir << "doss" << is + 1; - if(istep>=0) - { + if(istep>=0) + { ss << "g" << istep+1; - } + } ss << "_nao.txt"; - ModuleIO::cal_dos(is, - ss.str(), - dos_edelta_ev, - emax, - emin, - bcoeff, - kv.get_nks(), - kv.get_nkstot(), - kv.wk, - kv.isk, - nbands, - ekb, - wg, - istep); - } - - - if (PARAM.inp.out_dos == 2) - { - cal_pdos(psi, - p_ham, - pv, - ucell, - kv, - nspin0, - nbands, - ekb, - emax, - emin, - dos_edelta_ev, - bcoeff); + ModuleIO::cal_dos(is, + ss.str(), + dos_edelta_ev, + emax, + emin, + bcoeff, + kv.get_nks(), + kv.get_nkstot(), + kv.wk, + kv.isk, + nbands, + ekb, + wg, + istep, + out_app_flag, + bndpar); } - if(PARAM.inp.out_dos == 3) - { - for (int is = 0; is < nspin0; is++) - { - std::stringstream ss3; - ss3 << PARAM.globalv.global_out_dir << "fermi" << is << ".bxsf"; - nscf_fermi_surface(ss3.str(), nbands, energy_fermi.ef, kv, ucell, ekb); - } - } + + // out_dos >= 1: always compute PDOS alongside DOS (LCAO only) + cal_pdos(psi, + p_ham, + pv, + ucell, + kv, + nspin0, + nbands, + ekb, + emax, + emin, + dos_edelta_ev, + bcoeff, + istep, + nlocal, + nspin, + global_out_dir); ofs_running << " #DOS CALCULATION ENDS# " << std::endl; @@ -108,12 +114,12 @@ void write_dos_lcao( template void write_dos_lcao( const psi::Psi* psi, - hamilt::Hamilt* p_ham, + hamilt::Hamilt* p_ham, const Parallel_Orbitals &pv, const UnitCell& ucell, - const K_Vectors& kv, - const int nbands, - const elecstate::Efermi &energy_fermi, + const K_Vectors& kv, + const int nbands, + const elecstate::Efermi &energy_fermi, const ModuleBase::matrix& ekb, const ModuleBase::matrix& wg, const double& dos_edelta_ev, @@ -121,17 +127,26 @@ template void write_dos_lcao( const double& bcoeff, const bool out_app_flag, const int istep, + const int nspin, + const int nlocal, + const bool dos_setemax, + const double dos_emax_ev, + const bool dos_setemin, + const double dos_emin_ev, + const bool two_fermi, + const int bndpar, + const std::string& global_out_dir, std::ofstream &ofs_running); template void write_dos_lcao( const psi::Psi>* psi, - hamilt::Hamilt>* p_ham, + hamilt::Hamilt>* p_ham, const Parallel_Orbitals &pv, const UnitCell& ucell, - const K_Vectors& kv, - const int nbands, - const elecstate::Efermi &energy_fermi, + const K_Vectors& kv, + const int nbands, + const elecstate::Efermi &energy_fermi, const ModuleBase::matrix& ekb, const ModuleBase::matrix& wg, const double& dos_edelta_ev, @@ -139,6 +154,15 @@ template void write_dos_lcao( const double& bcoeff, const bool out_app_flag, const int istep, + const int nspin, + const int nlocal, + const bool dos_setemax, + const double dos_emax_ev, + const bool dos_setemin, + const double dos_emin_ev, + const bool two_fermi, + const int bndpar, + const std::string& global_out_dir, std::ofstream &ofs_running); } diff --git a/source/source_io/module_dos/write_dos_lcao.h b/source/source_io/module_dos/write_dos_lcao.h index 88b186f3f9a..0e9017149a4 100644 --- a/source/source_io/module_dos/write_dos_lcao.h +++ b/source/source_io/module_dos/write_dos_lcao.h @@ -11,18 +11,18 @@ namespace ModuleIO { - /// @brief calculate density of states(DOS), + /// @brief calculate density of states(DOS), /// partial density of states(PDOS), /// and mulliken charge for LCAO base template void write_dos_lcao( const psi::Psi* psi, // LCAO wave functions - hamilt::Hamilt* p_ham, // Hamiltonian + hamilt::Hamilt* p_ham, // Hamiltonian const Parallel_Orbitals &pv, // Parallel scheme for LCAO wave functions const UnitCell& ucell, // Unit cell information - const K_Vectors& kv, // k-point information in Brillouin zone - const int nbands, // Number of bands - const elecstate::Efermi &energy_fermi, // Fermi energy + const K_Vectors& kv, // k-point information in Brillouin zone + const int nbands, // Number of bands + const elecstate::Efermi &energy_fermi, // Fermi energy const ModuleBase::matrix& ekb, // Eigenvalues per k-point and band const ModuleBase::matrix& wg, // Weights of eigenvalues const double& dos_edelta_ev, // Delta energy @@ -30,6 +30,15 @@ namespace ModuleIO const double& bcoeff, const bool out_app_flag, const int istep, + const int nspin, + const int nlocal, + const bool dos_setemax, + const double dos_emax_ev, + const bool dos_setemin, + const double dos_emin_ev, + const bool two_fermi, + const int bndpar, + const std::string& global_out_dir, std::ofstream &ofs_running); } #endif diff --git a/source/source_io/module_dos/write_dos_pw.cpp b/source/source_io/module_dos/write_dos_pw.cpp index bfcc094d5cb..502b3984748 100644 --- a/source/source_io/module_dos/write_dos_pw.cpp +++ b/source/source_io/module_dos/write_dos_pw.cpp @@ -1,89 +1,89 @@ #include "write_dos_pw.h" #include "cal_dos.h" -#include "../module_energy/nscf_fermi_surf.h" #include "source_base/parallel_reduce.h" -#include "source_io/module_parameter/parameter.h" void ModuleIO::write_dos_pw( - const UnitCell& ucell, - const ModuleBase::matrix& ekb, - const ModuleBase::matrix& wg, - const K_Vectors& kv, - const int nbands, + const UnitCell& ucell, + const ModuleBase::matrix& ekb, + const ModuleBase::matrix& wg, + const K_Vectors& kv, + const int nbands, const int istep_in, - const elecstate::Efermi &energy_fermi, - const double& dos_edelta_ev, - const double& dos_scale, - const double& bcoeff, - std::ofstream& ofs_running) + const elecstate::Efermi &energy_fermi, + const double& dos_edelta_ev, + const double& dos_scale, + const double& bcoeff, + const int nspin, + const int out_dos, + const bool dos_setemax, + const double dos_emax_ev, + const bool dos_setemin, + const double dos_emin_ev, + const bool two_fermi, + const bool out_app_flag, + const int bndpar, + const std::string& global_out_dir, + std::ofstream& ofs_running) { ModuleBase::TITLE("ModuleIO", "write_dos_pw"); - const int nspin0 = (PARAM.inp.nspin == 2) ? 2 : 1; + const int nspin0 = (nspin == 2) ? 2 : 1; double emax = 0.0; double emin = 0.0; - prepare_dos(ofs_running, - energy_fermi, - ekb, - kv.get_nks(), - nbands, - dos_edelta_ev, + prepare_dos(ofs_running, + energy_fermi, + ekb, + kv.get_nks(), + nbands, + dos_edelta_ev, dos_scale, - emax, - emin); + emax, + emin, + dos_setemax, + dos_emax_ev, + dos_setemin, + dos_emin_ev, + two_fermi); for (int is = 0; is < nspin0; ++is) { // DOS_ispin contains not smoothed dos - std::stringstream ss; - ss << PARAM.globalv.global_out_dir << "dos"; + std::stringstream ss; + ss << global_out_dir << "dos"; - if(nspin0==2) - { - ss << "s" << is + 1; - } - else - { - // do nothing; - } + if(nspin0==2) + { + ss << "s" << is + 1; + } + else + { + // do nothing; + } - ss << ".txt"; + ss << ".txt"; ModuleBase::GlobalFunc::OUT(ofs_running, "DOS file", ss.str()); - ModuleIO::cal_dos(is, - ss.str(), - dos_edelta_ev, - emax, - emin, - bcoeff, - kv.get_nks(), - kv.get_nkstot(), - kv.wk, - kv.isk, - nbands, - ekb, - wg, - istep_in); - } - - - if (PARAM.inp.out_dos == 2) - { - ModuleBase::WARNING_QUIT("ModuleIO::write_dos_pw","PW basis do not support PDOS calculations yet."); + ModuleIO::cal_dos(is, + ss.str(), + dos_edelta_ev, + emax, + emin, + bcoeff, + kv.get_nks(), + kv.get_nkstot(), + kv.wk, + kv.isk, + nbands, + ekb, + wg, + istep_in, + out_app_flag, + bndpar); } - if(PARAM.inp.out_dos == 3) - { - for (int is = 0; is < nspin0; is++) - { - std::stringstream ss3; - ss3 << PARAM.globalv.global_out_dir << "fermi" << is << ".bxsf"; - nscf_fermi_surface(ss3.str(), nbands, energy_fermi.ef, kv, ucell, ekb); - } - } ofs_running << " #DOS CALCULATION ENDS# " << std::endl; } diff --git a/source/source_io/module_dos/write_dos_pw.h b/source/source_io/module_dos/write_dos_pw.h index 44f507d84e9..4b2417d8f42 100644 --- a/source/source_io/module_dos/write_dos_pw.h +++ b/source/source_io/module_dos/write_dos_pw.h @@ -8,18 +8,28 @@ namespace ModuleIO { - /// @brief calculate density of states(DOS) for PW base - void write_dos_pw( - const UnitCell& ucell, - const ModuleBase::matrix &ekb, - const ModuleBase::matrix &wg, - const K_Vectors& kv, - const int nbands, - const int istep_in, - const elecstate::Efermi &energy_fermi, - const double &dos_edelta_ev, - const double &dos_scale, - const double &bcoeff, - std::ofstream& ofs_running); + /// @brief calculate density of states(DOS) for PW base + void write_dos_pw( + const UnitCell& ucell, + const ModuleBase::matrix &ekb, + const ModuleBase::matrix &wg, + const K_Vectors& kv, + const int nbands, + const int istep_in, + const elecstate::Efermi &energy_fermi, + const double &dos_edelta_ev, + const double &dos_scale, + const double &bcoeff, + const int nspin, + const int out_dos, + const bool dos_setemax, + const double dos_emax_ev, + const bool dos_setemin, + const double dos_emin_ev, + const bool two_fermi, + const bool out_app_flag, + const int bndpar, + const std::string& global_out_dir, + std::ofstream& ofs_running); } #endif diff --git a/source/source_io/module_dos/write_pdos_text.cpp b/source/source_io/module_dos/write_pdos_text.cpp new file mode 100644 index 00000000000..0afc5ae57c1 --- /dev/null +++ b/source/source_io/module_dos/write_pdos_text.cpp @@ -0,0 +1,127 @@ +#include "write_pdos_text.h" + +#include "source_base/global_variable.h" + +#include +#include +#include +#include +#include + +void ModuleIO::write_pdos_text( + const UnitCell& ucell, + const ModuleBase::matrix* pdos, + const int nspin, + const int nlocal, + const int npoints, + const double& emin, + const double& dos_edelta_ev, + const std::string& out_dir, + const std::string& basis, + const int istep) +{ + ModuleBase::TITLE("ModuleIO", "write_pdos_text"); + + // number of spin channels to write + const int nspin0 = (nspin == 2) ? 2 : 1; + + for (int is = 0; is < nspin0; ++is) + { + std::stringstream ss; + ss << out_dir << "pdoss" << is + 1; + if (istep >= 0) + { + ss << "g" << istep + 1; + } + ss << "_" << basis << ".txt"; + + std::ofstream ofs(ss.str().c_str()); + + // iw2m stores m as 0..2l, mapping to physical m values: + // 0->0, 1->+1, 2->-1, 3->+2, 4->-2, 5->+3, 6->-3 + // So columns are ordered: s, p(m=0,+1,-1), d(m=0,+1,-1,+2,-2), f(...) + ofs << "# istep: " << istep + 1 << std::endl; + ofs << "# npoints: " << npoints << std::endl; + ofs << "# energy(eV) atom species pdos(1/eV), columns: s(m=0) p(m=0,+1,-1) d(m=0,+1,-1,+2,-2) f(m=0,+1,-1,+2,-2,+3,-3)" << std::endl; + + for (int iat = 0; iat < ucell.nat; ++iat) + { + const int ia = ucell.iat2ia[iat]; + const int it = ucell.iat2it[iat]; + const Atom* atom = &ucell.atoms[it]; + const int s0 = ucell.itiaiw2iwt(it, ia, 0); + const int max_l = atom->nwl; + + // collect all (l, m) for this atom, ordered s,p,d,f + // iw2m stores m as 0..2l (non-negative), so loop m = 0..2*L + struct OrbInfo + { + int l; + int m; // 0..2l + }; + std::vector orb_list; + + for (int L = 0; L <= max_l; ++L) + { + for (int m = 0; m <= 2 * L; ++m) + { + for (int j = 0; j < atom->nw; ++j) + { + if (atom->iw2l[j] == L && atom->iw2m[j] == m) + { + orb_list.push_back({L, m}); + break; + } + } + } + } + + for (int n = 0; n < npoints; ++n) + { + const double en = emin + n * dos_edelta_ev; + + ofs << std::setw(12) << std::fixed << std::setprecision(6) << en + << " " << std::setw(3) << iat + 1 + << " " << std::setw(4) << ucell.atoms[it].label; + + for (size_t io = 0; io < orb_list.size(); ++io) + { + const int L = orb_list[io].l; + const int m = orb_list[io].m; + double pdos_val = 0.0; + + // sum over all zeta for this (l, m) + for (int j = 0; j < atom->nw; ++j) + { + if (atom->iw2l[j] != L || atom->iw2m[j] != m) + { + continue; + } + const int w = ucell.itiaiw2iwt(it, ia, j); + + if (nspin == 4) + { + const int w0 = w - s0; + pdos_val += pdos[0](s0 + 2 * w0, n) + pdos[0](s0 + 2 * w0 + 1, n); + } + else + { + pdos_val += pdos[is](w, n); + } + } + + // zero out negligible values + if (std::abs(pdos_val) < 1e-6) + { + pdos_val = 0.0; + } + + ofs << " " << std::setw(8) << std::fixed << std::setprecision(6) << pdos_val; + } + ofs << std::endl; + } + } + + ofs.close(); + } +} diff --git a/source/source_io/module_dos/write_pdos_text.h b/source/source_io/module_dos/write_pdos_text.h new file mode 100644 index 00000000000..e7b7f3a5ad1 --- /dev/null +++ b/source/source_io/module_dos/write_pdos_text.h @@ -0,0 +1,28 @@ +#ifndef WRITE_PDOS_TEXT_H +#define WRITE_PDOS_TEXT_H + +#include "source_base/matrix.h" +#include "source_cell/unitcell.h" + +namespace ModuleIO +{ + +/// Write PDOS in plain text long-table format. +/// File name: pdoss{spin}g{geom}_{basis}.txt +/// Columns: energy(eV) atom species l z m pdos(1/eV) +/// For nspin=4, the two spinor components are summed. +void write_pdos_text( + const UnitCell& ucell, + const ModuleBase::matrix* pdos, + const int nspin, + const int nlocal, + const int npoints, + const double& emin, + const double& dos_edelta_ev, + const std::string& out_dir, + const std::string& basis, + const int istep); + +} + +#endif diff --git a/source/source_io/module_efield/td_efield_io.cpp b/source/source_io/module_efield/td_efield_io.cpp index cf3df4d53b5..fd8cfe370b5 100644 --- a/source/source_io/module_efield/td_efield_io.cpp +++ b/source/source_io/module_efield/td_efield_io.cpp @@ -49,7 +49,7 @@ void write_td_field_values(const elecstate::TDFieldManager& manager, const std:: return; } - const std::vector& field_values = manager.field_values(); + const std::vector& field_values = manager.field_vals_ha(); for (std::size_t field_index = 0; field_index < field_values.size(); ++field_index) { // Keep one file per input occurrence even when directions repeat. @@ -61,8 +61,8 @@ void write_td_field_values(const elecstate::TDFieldManager& manager, const std:: } // Convert only at the user-visible output boundary: time to fs and the // electric field to V/Angstrom. - output << manager.current_step() * manager.dt() * ModuleBase::AU_to_FS << "\t" - << field_values[field_index] * ModuleBase::Ry_to_eV / ModuleBase::BOHR_TO_A << std::endl; + output << manager.current_step() * manager.dt_ha() * ModuleBase::AU_to_FS << "\t" + << field_values[field_index] * ModuleBase::Hartree_to_eV / ModuleBase::BOHR_TO_A << std::endl; } } diff --git a/source/source_io/module_efield/td_vector_pot_io.cpp b/source/source_io/module_efield/td_vector_pot_io.cpp index ae641003f50..7c895b88b64 100644 --- a/source/source_io/module_efield/td_vector_pot_io.cpp +++ b/source/source_io/module_efield/td_vector_pot_io.cpp @@ -19,7 +19,7 @@ std::string vector_pot_path(const std::string& directory) void write_vector_pot_header(std::ofstream& output) { output << std::left << std::setw(8) << "#istep" << std::setw(15) << "A_x" << std::setw(15) << "A_y" << std::setw(15) << "A_z" - << std::endl; + << "# propagation values, Hartree atomic units" << std::endl; } bool is_blank_or_comment(const std::string& line) @@ -109,7 +109,7 @@ void write_td_vector_pot(const std::string& output_dir, const int electronic_ste output << std::left << std::setw(8) << electronic_step + 1; for (int direction = 0; direction < 3; ++direction) { - output << std::scientific << std::setprecision(6) << std::setw(15) << vector_pot[direction]; + output << std::scientific << std::setprecision(16) << std::setw(25) << vector_pot[direction]; } output << std::endl; } diff --git a/source/source_io/module_energy/nscf_fermi_surf.cpp b/source/source_io/module_energy/nscf_fermi_surf.cpp deleted file mode 100644 index 99679f7f102..00000000000 --- a/source/source_io/module_energy/nscf_fermi_surf.cpp +++ /dev/null @@ -1,93 +0,0 @@ -#include "nscf_fermi_surf.h" -#include "source_base/global_function.h" -#include "source_io/module_parameter/parameter.h" -#include "source_base/global_variable.h" -#include "source_base/timer.h" - -#ifdef __MPI -#include -#endif - -void ModuleIO::nscf_fermi_surface(const std::string &out_band_dir, - const int &nband, - const double &ef, - const K_Vectors& kv, - const UnitCell& ucell, - const ModuleBase::matrix &ekb) -{ - ModuleBase::TITLE("ModuleIO","nscf_fermi_surface"); - ModuleBase::timer::start("ModuleIO", "nscf_fermi_surface"); -#ifdef __MPI - - const int start = 1; - const int end = PARAM.inp.nbands; - - std::ofstream ofs; - if(GlobalV::MY_RANK==0) - { - ofs.open(out_band_dir.c_str()); - ofs << std::setprecision(6); - ofs.close(); - } - - for(int ik=0; ik +void write_eband_terms(const int nspin, + const int nbasis, + const int drank, + const Parallel_Orbitals* pv, + const psi::Psi& psi, + const UnitCell& ucell, + Structure_Factor& sf, + surchem& solvent, + const ModulePW::PW_Basis& rho_basis, + const ModulePW::PW_Basis& rhod_basis, + const ModuleBase::matrix& vloc, + const Charge& chg, + const K_Vectors& kv, + const ModuleBase::matrix& wg, + Grid_Driver& gd, + const std::vector& orb_cutoff, + const TwoCenterBundle& two_center_bundle, + const Exx_Info& exx_info +#ifdef __EXX + , + std::vector>>>* Hexxd, + std::vector>>>>* Hexxc +#endif +) + { + // 0. prepare + const int& nbands = wg.nc; + const int nspin0 = (nspin == 2) ? 2 : 1; + double etxc = 0.0; + double vtxc = 0.0; + + Parallel_2D p2d; + + set_para2d_MO(*pv, nbands, p2d); + + auto if_gamma_fix = [](hamilt::HContainer& hR) + { + if (std::is_same::value) + { + hR.fix_gamma(); + } + }; + + auto all_band_energy = [&wg](const int ik, const std::vector& e_orb)->double + { + double e = 0; + for (int i = 0; i < e_orb.size(); ++i) { e += e_orb[i] * wg(ik, i); } + return e; + }; + + auto all_k_all_band_energy = [&wg, &all_band_energy](const std::vector>& e_orb)->double + { + double e = 0; + for (int ik = 0; ik < e_orb.size(); ++ik) + { + e += all_band_energy(ik, e_orb[ik]); + } + return e; + }; + + // 1. kinetic + if (PARAM.inp.t_in_h) + { + hamilt::HS_Matrix_K kinetic_k_ao(pv, 1); + hamilt::HContainer kinetic_R_ao(pv); + if_gamma_fix(kinetic_R_ao); + + hamilt::EKinetic> kinetic_op(&kinetic_k_ao, kv.kvec_d, + &kinetic_R_ao, &ucell, orb_cutoff, &gd, two_center_bundle.kinetic_orb.get()); + + kinetic_op.contributeHR(); + + std::vector> e_orb_kinetic; + + for (int ik = 0;ik < kv.get_nks();++ik) + { + kinetic_k_ao.set_zero_hk(); + dynamic_cast*>(&kinetic_op)->contributeHk(ik); + e_orb_kinetic.emplace_back(orbital_energy(ik, nbands, + cVc(kinetic_k_ao.get_hk(), &psi(ik, 0, 0), nbasis, nbands, *pv, p2d), p2d)); + } + + write_orb_energy(kv, nspin0, nbands, e_orb_kinetic, "kinetic", "", PARAM.globalv.global_out_dir); + } + + // 2. pp: local + if (PARAM.inp.vl_in_h) + { + elecstate::Potential pot_local(&rhod_basis, &rho_basis, &ucell, &vloc, &sf, &solvent, &etxc, &vtxc); + pot_local.pot_register({ "local" }); + pot_local.update_from_charge(&chg, &ucell); + hamilt::HS_Matrix_K v_pp_local_k_ao(pv, 1); + hamilt::HContainer v_pp_local_R_ao(pv); + if_gamma_fix(v_pp_local_R_ao); + std::vector> e_orb_pp_local; + + hamilt::Veff> v_pp_local_op( + &v_pp_local_k_ao, + kv.kvec_d, + &pot_local, + &v_pp_local_R_ao, + &ucell, + orb_cutoff, + &gd, + nspin); + + v_pp_local_op.contributeHR(); + for (int ik = 0;ik < kv.get_nks();++ik) + { + v_pp_local_k_ao.set_zero_hk(); + dynamic_cast*>(&v_pp_local_op)->contributeHk(ik); + e_orb_pp_local.emplace_back(orbital_energy(ik, nbands, + cVc(v_pp_local_k_ao.get_hk(), &psi(ik, 0, 0), nbasis, nbands, *pv, p2d), p2d)); + } + write_orb_energy(kv, nspin0, nbands, e_orb_pp_local, "vpp_local", "", PARAM.globalv.global_out_dir); + } + + // 3. pp: nonlocal + if (PARAM.inp.vnl_in_h) + { + hamilt::HS_Matrix_K v_pp_nonlocal_k_ao(pv, 1); + hamilt::HContainer v_pp_nonlocal_R_ao(pv); + if_gamma_fix(v_pp_nonlocal_R_ao); + std::vector> e_orb_pp_nonlocal; + hamilt::Nonlocal> v_pp_nonlocal_op(&v_pp_nonlocal_k_ao, kv.kvec_d, + &v_pp_nonlocal_R_ao, &ucell, orb_cutoff, &gd, two_center_bundle.overlap_orb_beta.get()); + v_pp_nonlocal_op.contributeHR(); + for (int ik = 0;ik < kv.get_nks();++ik) + { + v_pp_nonlocal_k_ao.set_zero_hk(); + dynamic_cast*>(&v_pp_nonlocal_op)->contributeHk(ik); + e_orb_pp_nonlocal.emplace_back(orbital_energy(ik, nbands, + cVc(v_pp_nonlocal_k_ao.get_hk(), &psi(ik, 0, 0), nbasis, nbands, *pv, p2d), p2d)); + } + write_orb_energy(kv, nspin0, nbands, e_orb_pp_nonlocal, "vpp_nonlocal", "", PARAM.globalv.global_out_dir); + } + + // 4. hartree + if (PARAM.inp.vh_in_h) + { + elecstate::Potential pot_hartree(&rhod_basis, &rho_basis, &ucell, &vloc, &sf, &solvent, &etxc, &vtxc); + pot_hartree.pot_register({ "hartree" }); + pot_hartree.update_from_charge(&chg, &ucell); + std::vector> v_hartree_R_ao(nspin0, hamilt::HContainer(pv)); + for (int is = 0; is < nspin0; ++is) + { + v_hartree_R_ao[is].set_zero(); + if_gamma_fix(v_hartree_R_ao[is]); + } + hamilt::HS_Matrix_K v_hartree_k_ao(pv, 1); + std::vector>*> v_hartree_op(nspin0); + for (int is = 0; is < nspin0; ++is) + { + v_hartree_op[is] = new hamilt::Veff>( + &v_hartree_k_ao, kv.kvec_d, &pot_hartree, &v_hartree_R_ao[is], &ucell, orb_cutoff, &gd, nspin); + v_hartree_op[is]->contributeHR(); + } + std::vector> e_orb_hartree; + for (int ik = 0;ik < kv.get_nks();++ik) + { + int is = kv.isk[ik]; + v_hartree_k_ao.set_zero_hk(); + dynamic_cast*>(v_hartree_op[is])->contributeHk(ik); + e_orb_hartree.emplace_back(orbital_energy(ik, nbands, + cVc(v_hartree_k_ao.get_hk(), &psi(ik, 0, 0), nbasis, nbands, *pv, p2d), p2d)); + } + for (auto& op : v_hartree_op) { delete op; } + write_orb_energy(kv, nspin0, nbands, e_orb_hartree, "vhartree", "", PARAM.globalv.global_out_dir); + } + + // 5. xc (including exx) + if (!PARAM.inp.out_mat_xc) // avoid duplicate output + { + bool cal_exx = exx_info.info_global.cal_exx; + write_Vxc(nspin, + nbasis, + drank, + pv, + psi, + ucell, + sf, + solvent, + rho_basis, + rhod_basis, + vloc, + chg, + kv, + orb_cutoff, + wg, + gd, + PARAM.inp.dft_plus_u, + PARAM.globalv.gamma_only_local, + PARAM.globalv.global_out_dir, + PARAM.inp.out_ndigits, + PARAM.inp.ks_solver, + cal_exx, + exx_info +#ifdef __EXX + , + Hexxd, + Hexxc +#endif + ); + } + } + +// Explicit template instantiations +template void write_eband_terms( + const int, const int, const int, const Parallel_Orbitals*, const psi::Psi&, + const UnitCell&, Structure_Factor&, surchem&, const ModulePW::PW_Basis&, const ModulePW::PW_Basis&, + const ModuleBase::matrix&, const Charge&, const K_Vectors&, const ModuleBase::matrix&, Grid_Driver&, + const std::vector&, const TwoCenterBundle&, const Exx_Info& +#ifdef __EXX + , std::vector>>>*, + std::vector>>>>* +#endif +); + +template void write_eband_terms, double>( + const int, const int, const int, const Parallel_Orbitals*, const psi::Psi>&, + const UnitCell&, Structure_Factor&, surchem&, const ModulePW::PW_Basis&, const ModulePW::PW_Basis&, + const ModuleBase::matrix&, const Charge&, const K_Vectors&, const ModuleBase::matrix&, Grid_Driver&, + const std::vector&, const TwoCenterBundle&, const Exx_Info& +#ifdef __EXX + , std::vector>>>*, + std::vector>>>>* +#endif +); + +template void write_eband_terms, std::complex>( + const int, const int, const int, const Parallel_Orbitals*, const psi::Psi>&, + const UnitCell&, Structure_Factor&, surchem&, const ModulePW::PW_Basis&, const ModulePW::PW_Basis&, + const ModuleBase::matrix&, const Charge&, const K_Vectors&, const ModuleBase::matrix&, Grid_Driver&, + const std::vector&, const TwoCenterBundle&, const Exx_Info& +#ifdef __EXX + , std::vector>>>*, + std::vector>>>>* +#endif +); + +} // namespace ModuleIO diff --git a/source/source_io/module_energy/write_eband_terms.h b/source/source_io/module_energy/write_eband_terms.h new file mode 100644 index 00000000000..54d980a8701 --- /dev/null +++ b/source/source_io/module_energy/write_eband_terms.h @@ -0,0 +1,75 @@ +#ifndef WRITE_EBAND_TERMS_H +#define WRITE_EBAND_TERMS_H + +#include "source_base/matrix.h" +#include "source_psi/psi.h" + +#include +#include +#include +#include +#include + +class Grid_Driver; +class Parallel_Orbitals; +class UnitCell; +class Structure_Factor; +class surchem; +class Charge; +class K_Vectors; +class TwoCenterBundle; +struct Exx_Info; + +namespace ModulePW +{ +class PW_Basis; +} + +#ifdef __EXX +namespace hamilt +{ +using TAC = std::pair>; +} +namespace RI +{ +template +class Tensor; +} +#endif + +namespace ModuleIO +{ + +/// @brief Write the band-decomposed energy of each Hamiltonian term +/// (kinetic, local/nonlocal pp, Hartree, XC) in KS orbital representation. +/// @tparam TK K-point data type (double or std::complex) +/// @tparam TR Real-space data type (double or std::complex) +template +void write_eband_terms(const int nspin, + const int nbasis, + const int drank, + const Parallel_Orbitals* pv, + const psi::Psi& psi, + const UnitCell& ucell, + Structure_Factor& sf, + surchem& solvent, + const ModulePW::PW_Basis& rho_basis, + const ModulePW::PW_Basis& rhod_basis, + const ModuleBase::matrix& vloc, + const Charge& chg, + const K_Vectors& kv, + const ModuleBase::matrix& wg, + Grid_Driver& gd, + const std::vector& orb_cutoff, + const TwoCenterBundle& two_center_bundle, + const Exx_Info& exx_info +#ifdef __EXX + , + std::vector>>>* Hexxd, + std::vector>>>>* Hexxc +#endif +); + +} // namespace ModuleIO + +#endif // WRITE_EBAND_TERMS_H diff --git a/source/source_io/module_energy/write_eband_terms.hpp b/source/source_io/module_energy/write_eband_terms.hpp deleted file mode 100644 index 6622356a99f..00000000000 --- a/source/source_io/module_energy/write_eband_terms.hpp +++ /dev/null @@ -1,214 +0,0 @@ -#ifndef WRITE_EBAND_TERMS_HPP -#define WRITE_EBAND_TERMS_HPP - -#include "source_io/module_hs/write_vxc.hpp" -#include "source_hamilt/module_xc/exx_info.h" -#include "source_lcao/module_operator_lcao/ekinetic.h" -#include "source_lcao/module_operator_lcao/nonlocal.h" -#include "source_basis/module_nao/two_center_bundle.h" - -namespace ModuleIO -{ -template -void write_eband_terms(const int nspin, - const int nbasis, - const int drank, - const Parallel_Orbitals* pv, - const psi::Psi& psi, - const UnitCell& ucell, - Structure_Factor& sf, - surchem& solvent, - const ModulePW::PW_Basis& rho_basis, - const ModulePW::PW_Basis& rhod_basis, - const ModuleBase::matrix& vloc, - const Charge& chg, - const K_Vectors& kv, - const ModuleBase::matrix& wg, - Grid_Driver& gd, - const std::vector& orb_cutoff, - const TwoCenterBundle& two_center_bundle, - const Exx_Info& exx_info -#ifdef __EXX - , - std::vector>>>* Hexxd = nullptr, - std::vector>>>>* Hexxc = nullptr -#endif -) - { - // 0. prepare - const int& nbands = wg.nc; - const int& nspin0 = (nspin == 2) ? 2 : 1; - double etxc = 0.0; - double vtxc = 0.0; - - Parallel_2D p2d; - - set_para2d_MO(*pv, nbands, p2d); - - auto if_gamma_fix = [](hamilt::HContainer& hR) - { - if (std::is_same::value) - { - hR.fix_gamma(); - } - }; - - auto all_band_energy = [&wg](const int ik, const std::vector& e_orb)->double - { - double e = 0; - for (int i = 0; i < e_orb.size(); ++i) { e += e_orb[i] * wg(ik, i); } - return e; - }; - - auto all_k_all_band_energy = [&wg, &all_band_energy](const std::vector>& e_orb)->double - { - double e = 0; - for (int ik = 0; ik < e_orb.size(); ++ik) - { - e += all_band_energy(ik, e_orb[ik]); - } - return e; - }; - - // 1. kinetic - if (PARAM.inp.t_in_h) - { - hamilt::HS_Matrix_K kinetic_k_ao(pv, 1); - hamilt::HContainer kinetic_R_ao(pv); - if_gamma_fix(kinetic_R_ao); - - hamilt::EKinetic> kinetic_op(&kinetic_k_ao, kv.kvec_d, - &kinetic_R_ao, &ucell, orb_cutoff, &gd, two_center_bundle.kinetic_orb.get()); - - kinetic_op.contributeHR(); - - std::vector> e_orb_kinetic; - - for (int ik = 0;ik < kv.get_nks();++ik) - { - kinetic_k_ao.set_zero_hk(); - dynamic_cast*>(&kinetic_op)->contributeHk(ik); - e_orb_kinetic.emplace_back(orbital_energy(ik, nbands, - cVc(kinetic_k_ao.get_hk(), &psi(ik, 0, 0), nbasis, nbands, *pv, p2d), p2d)); - } - - write_orb_energy(kv, nspin0, nbands, e_orb_kinetic, "kinetic", ""); - } - - // 2. pp: local - if (PARAM.inp.vl_in_h) - { - elecstate::Potential pot_local(&rhod_basis, &rho_basis, &ucell, &vloc, &sf, &solvent, &etxc, &vtxc); - pot_local.pot_register({ "local" }); - pot_local.update_from_charge(&chg, &ucell); - hamilt::HS_Matrix_K v_pp_local_k_ao(pv, 1); - hamilt::HContainer v_pp_local_R_ao(pv); - if_gamma_fix(v_pp_local_R_ao); - std::vector> e_orb_pp_local; - - hamilt::Veff> v_pp_local_op( - &v_pp_local_k_ao, - kv.kvec_d, - &pot_local, - &v_pp_local_R_ao, - &ucell, - orb_cutoff, - &gd, - nspin); - - v_pp_local_op.contributeHR(); - for (int ik = 0;ik < kv.get_nks();++ik) - { - v_pp_local_k_ao.set_zero_hk(); - dynamic_cast*>(&v_pp_local_op)->contributeHk(ik); - e_orb_pp_local.emplace_back(orbital_energy(ik, nbands, - cVc(v_pp_local_k_ao.get_hk(), &psi(ik, 0, 0), nbasis, nbands, *pv, p2d), p2d)); - } - write_orb_energy(kv, nspin0, nbands, e_orb_pp_local, "vpp_local", ""); - } - - // 3. pp: nonlocal - if (PARAM.inp.vnl_in_h) - { - hamilt::HS_Matrix_K v_pp_nonlocal_k_ao(pv, 1); - hamilt::HContainer v_pp_nonlocal_R_ao(pv); - if_gamma_fix(v_pp_nonlocal_R_ao); - std::vector> e_orb_pp_nonlocal; - hamilt::Nonlocal> v_pp_nonlocal_op(&v_pp_nonlocal_k_ao, kv.kvec_d, - &v_pp_nonlocal_R_ao, &ucell, orb_cutoff, &gd, two_center_bundle.overlap_orb_beta.get()); - v_pp_nonlocal_op.contributeHR(); - for (int ik = 0;ik < kv.get_nks();++ik) - { - v_pp_nonlocal_k_ao.set_zero_hk(); - dynamic_cast*>(&v_pp_nonlocal_op)->contributeHk(ik); - e_orb_pp_nonlocal.emplace_back(orbital_energy(ik, nbands, - cVc(v_pp_nonlocal_k_ao.get_hk(), &psi(ik, 0, 0), nbasis, nbands, *pv, p2d), p2d)); - } - write_orb_energy(kv, nspin0, nbands, e_orb_pp_nonlocal, "vpp_nonlocal", ""); - } - - // 4. hartree - if (PARAM.inp.vh_in_h) - { - elecstate::Potential pot_hartree(&rhod_basis, &rho_basis, &ucell, &vloc, &sf, &solvent, &etxc, &vtxc); - pot_hartree.pot_register({ "hartree" }); - pot_hartree.update_from_charge(&chg, &ucell); - std::vector> v_hartree_R_ao(nspin0, hamilt::HContainer(pv)); - for (int is = 0; is < nspin0; ++is) - { - v_hartree_R_ao[is].set_zero(); - if_gamma_fix(v_hartree_R_ao[is]); - } - hamilt::HS_Matrix_K v_hartree_k_ao(pv, 1); - std::vector>*> v_hartree_op(nspin0); - for (int is = 0; is < nspin0; ++is) - { - v_hartree_op[is] = new hamilt::Veff>( - &v_hartree_k_ao, kv.kvec_d, &pot_hartree, &v_hartree_R_ao[is], &ucell, orb_cutoff, &gd, nspin); - v_hartree_op[is]->contributeHR(); - } - std::vector> e_orb_hartree; - for (int ik = 0;ik < kv.get_nks();++ik) - { - int is = kv.isk[ik]; - v_hartree_k_ao.set_zero_hk(); - dynamic_cast*>(v_hartree_op[is])->contributeHk(ik); - e_orb_hartree.emplace_back(orbital_energy(ik, nbands, - cVc(v_hartree_k_ao.get_hk(), &psi(ik, 0, 0), nbasis, nbands, *pv, p2d), p2d)); - } - for (auto& op : v_hartree_op) { delete op; } - write_orb_energy(kv, nspin0, nbands, e_orb_hartree, "vhartree", ""); - } - - // 5. xc (including exx) - if (!PARAM.inp.out_mat_xc) // avoid duplicate output - { - bool cal_exx = exx_info.info_global.cal_exx; - write_Vxc(nspin, - nbasis, - drank, - pv, - psi, - ucell, - sf, - solvent, - rho_basis, - rhod_basis, - vloc, - chg, - kv, - orb_cutoff, - wg, - gd, - cal_exx, - exx_info -#ifdef __EXX - , - Hexxd, - Hexxc -#endif - ); - } - } -} -#endif diff --git a/source/source_io/module_energy/write_eig_occ.cpp b/source/source_io/module_energy/write_eig_occ.cpp index 55a0cbb7efd..79c9be48797 100644 --- a/source/source_io/module_energy/write_eig_occ.cpp +++ b/source/source_io/module_energy/write_eig_occ.cpp @@ -12,21 +12,24 @@ #include // use MPI_Barrier #endif -void ModuleIO::write_eig_iter(const ModuleBase::matrix &ekb,const ModuleBase::matrix &wg, const K_Vectors& kv) +void ModuleIO::write_eig_iter(const ModuleBase::matrix &ekb, + const ModuleBase::matrix &wg, + const K_Vectors& kv, + const int nbands, + const int nspin) { ModuleBase::TITLE("ModuleIO","write_eig_iter"); - ModuleBase::timer::start("ModuleIO", "write_eig_iter"); + ModuleBase::timer::start("ModuleIO", "write_eig_iter"); - GlobalV::ofs_running << "\n PRINT #EIGENVALUES# AND #OCCUPATIONS#" << std::endl; + GlobalV::ofs_running << "\n PRINT #EIGENVALUES# AND #OCCUPATIONS#" << std::endl; // Taoni fix bndpar on 2026-08-21 - const Parallel::ParaBandOutput band_output(ekb.nc, PARAM.inp.nbands, Parallel::make_band_world()); + const Parallel::ParaBandOutput band_output(ekb.nc, nbands, Parallel::make_band_world()); const ModuleBase::matrix global_ekb = band_output.gather_matrix(ekb); const ModuleBase::matrix global_wg = band_output.gather_matrix(wg); - const int nspin = PARAM.inp.nspin; const int nks = kv.get_nks(); - const int nkstot = kv.get_nkstot(); + const int nkstot = kv.get_nkstot(); const int nk_fac = nspin == 2 ? 2 : 1; const int nks_np = nks / nk_fac; const int nkstot_np = nkstot / nk_fac; @@ -68,7 +71,7 @@ void ModuleIO::write_eig_iter(const ModuleBase::matrix &ekb,const ModuleBase::ma } } - + // ============================================================= // MPI communication: // RANK 0 collect and print out #EIGENVALUES# AND #OCCUPATIONS# for all k-points @@ -158,49 +161,51 @@ void ModuleIO::write_eig_iter(const ModuleBase::matrix &ekb,const ModuleBase::ma } - ModuleBase::timer::end("ModuleIO", "write_eig_iter"); + ModuleBase::timer::end("ModuleIO", "write_eig_iter"); } void ModuleIO::write_eig_file(const ModuleBase::matrix &ekb, - const ModuleBase::matrix &wg, - const K_Vectors& kv, - const int istep) + const ModuleBase::matrix &wg, + const K_Vectors& kv, + const int nbands, + const int nspin, + const std::string& out_dir, + const int istep) { - ModuleBase::TITLE("ModuleIO","write_eig_file"); - ModuleBase::timer::start("ModuleIO", "write_eig_file"); + ModuleBase::TITLE("ModuleIO","write_eig_file"); + ModuleBase::timer::start("ModuleIO", "write_eig_file"); /* - GlobalV::ofs_running << "\n"; - GlobalV::ofs_running << " >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>" << std::endl; - GlobalV::ofs_running << " | |" << std::endl; - GlobalV::ofs_running << " | #Print out the eigenvalues and occupations# |" << std::endl; - GlobalV::ofs_running << " | |" << std::endl; - GlobalV::ofs_running << " <<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<" << std::endl; - GlobalV::ofs_running << "\n"; + GlobalV::ofs_running << "\n"; + GlobalV::ofs_running << " >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>" << std::endl; + GlobalV::ofs_running << " | |" << std::endl; + GlobalV::ofs_running << " | #Print out the eigenvalues and occupations# |" << std::endl; + GlobalV::ofs_running << " | |" << std::endl; + GlobalV::ofs_running << " <<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<" << std::endl; + GlobalV::ofs_running << "\n"; */ - const int nspin = PARAM.inp.nspin; // Taoni fix bndpar on 2026-08-21 - const Parallel::ParaBandOutput band_output(ekb.nc, PARAM.inp.nbands, Parallel::make_band_world()); + const Parallel::ParaBandOutput band_output(ekb.nc, nbands, Parallel::make_band_world()); const ModuleBase::matrix global_ekb = band_output.gather_matrix(ekb); const ModuleBase::matrix global_wg = band_output.gather_matrix(wg); const int nks = kv.get_nks(); - const int nkstot = kv.get_nkstot(); + const int nkstot = kv.get_nkstot(); bool wrong = false; - for (int ik = 0; ik < nks; ++ik) - { - for (int ib = 0; ib < ekb.nc; ++ib) - { - if (std::abs(ekb(ik, ib)) > 1.0e10) - { - GlobalV::ofs_warning << " ik=" << ik + 1 << " ib=" << ib + 1 - << " " << ekb(ik, ib) << " Ry" << std::endl; - wrong = true; - } - } - } + for (int ik = 0; ik < nks; ++ik) + { + for (int ib = 0; ib < ekb.nc; ++ib) + { + if (std::abs(ekb(ik, ib)) > 1.0e10) + { + GlobalV::ofs_warning << " ik=" << ik + 1 << " ib=" << ib + 1 + << " " << ekb(ik, ib) << " Ry" << std::endl; + wrong = true; + } + } + } #ifdef __MPI MPI_Allreduce(MPI_IN_PLACE, &wrong, 1, MPI_C_BOOL, MPI_LOR, MPI_COMM_WORLD); @@ -216,7 +221,7 @@ void ModuleIO::write_eig_file(const ModuleBase::matrix &ekb, #endif // file name to store eigenvalues - std::string filename = PARAM.globalv.global_out_dir + "eig_occ.txt"; + std::string filename = out_dir + "eig_occ.txt"; GlobalV::ofs_running << " Write eigenvalues and occupations to file: " << filename << std::endl; @@ -227,14 +232,14 @@ void ModuleIO::write_eig_file(const ModuleBase::matrix &ekb, const bool append = istep > 0 || (inp.calculation == "md" && inp.mdp.md_restart); - if (append) - { - ofs_eig0.open(filename.c_str(), std::ios::app); - } - else - { - ofs_eig0.open(filename.c_str()); - } + if (append) + { + ofs_eig0.open(filename.c_str(), std::ios::app); + } + else + { + ofs_eig0.open(filename.c_str()); + } ofs_eig0 << istep+1 << " # ionic step" << std::endl; ofs_eig0 << " Electronic state energy (eV) and occupations" << std::endl; @@ -295,6 +300,6 @@ void ModuleIO::write_eig_file(const ModuleBase::matrix &ekb, #endif } - ModuleBase::timer::end("ModuleIO", "write_eig_file"); - return; + ModuleBase::timer::end("ModuleIO", "write_eig_file"); + return; } diff --git a/source/source_io/module_energy/write_eig_occ.h b/source/source_io/module_energy/write_eig_occ.h index 58c9831e5bd..5a39a26f661 100644 --- a/source/source_io/module_energy/write_eig_occ.h +++ b/source/source_io/module_energy/write_eig_occ.h @@ -3,16 +3,30 @@ #include "source_base/matrix.h" #include "source_cell/klist.h" +#include + namespace ModuleIO { - void write_eig_iter(const ModuleBase::matrix &ekb, - const ModuleBase::matrix &wg, - const K_Vectors& kv); + /// @brief print eigenvalues and occupations to the running log + /// @param nbands number of bands requested by INPUT + /// @param nspin number of spin channels + void write_eig_iter(const ModuleBase::matrix &ekb, + const ModuleBase::matrix &wg, + const K_Vectors& kv, + const int nbands, + const int nspin); - void write_eig_file(const ModuleBase::matrix &ekb, - const ModuleBase::matrix &wg, - const K_Vectors& kv, - const int istep); + /// @brief write eigenvalues and occupations to /eig_occ.txt + /// @param nbands number of bands requested by INPUT + /// @param nspin number of spin channels + /// @param out_dir directory the file is written into + void write_eig_file(const ModuleBase::matrix &ekb, + const ModuleBase::matrix &wg, + const K_Vectors& kv, + const int nbands, + const int nspin, + const std::string& out_dir, + const int istep); } #endif diff --git a/source/source_io/module_energy/write_proj_band_lcao.h b/source/source_io/module_energy/write_proj_band_lcao.h index af005ae5712..3dd4cfa8c4c 100644 --- a/source/source_io/module_energy/write_proj_band_lcao.h +++ b/source/source_io/module_energy/write_proj_band_lcao.h @@ -12,10 +12,10 @@ namespace ModuleIO template void write_proj_band_lcao( const psi::Psi* psi, - const Parallel_Orbitals &pv, - const elecstate::ElecState* pelec, - const K_Vectors& kv, - const UnitCell &ucell, + const Parallel_Orbitals &pv, + const elecstate::ElecState* pelec, + const K_Vectors& kv, + const UnitCell &ucell, hamilt::Hamilt* p_ham); } diff --git a/source/source_io/module_hs/angmom_op_mat.cpp b/source/source_io/module_hs/angmom_op_mat.cpp new file mode 100644 index 00000000000..d8e3fd2441a --- /dev/null +++ b/source/source_io/module_hs/angmom_op_mat.cpp @@ -0,0 +1,432 @@ +#include +#include +#include +#include +#include +#include +#include +#include +#include "source_cell/unitcell.h" +#include "source_base/sph_bessel_tf.h" +#include "source_basis/module_nao/two_center_integrator.h" +#include "source_cell/module_neighbor/sltk_grid_driver.h" +#include "source_cell/module_neighbor/sltk_atom_arrange.h" +#include "source_io/module_hs/angmom_op_mat.h" +#include "source_base/formatter.h" +#include "source_base/parallel_common.h" +#include "source_base/timer.h" +/** + * + * FIXME: the following part will be transfered to TwoCenterIntegrator soon + * + * Notation + * -------- + * ylm: complex spherical harmonics + * slm: solid (real) spherical harmonics + * + * Changelog + * --------- + * Switch to support the solid spherical harmonics to keep consistent with + * implementations of other parts. + * Formulation (in Chinese): + * https://my.feishu.cn/wiki/D0enwcUKfiJgtSkJ5scc9Dagntc + */ + +namespace +{ +// L+ylm = sqrt((l-m)(l+m+1))ylm+1, return the sqrt((l-m)(l+m+1)) +double _lambda_plus(const int l, const int m) +{ + return std::sqrt((l - m) * (l + m + 1)); // NOTE: complex spherical harmonics +} + +// L-ylm = sqrt((l+m)(l-m+1))ylm-1, return the sqrt((l+m)(l-m+1)) +double _lambda_minus(const int l, const int m) +{ + return std::sqrt((l + m) * (l - m + 1)); // NOTE: complex spherical harmonics +} + +const std::complex kImag = {0., 1.}; +const double kInvSqrt2 = std::sqrt(2) * 0.5; +/// @brief Threshold for judging whether the analytical coefficient +/// sqrt(l(l+1)-m(m+1)) is exactly zero. Tighter than the output +/// sparse threshold (1e-10) because this guards an analytical +/// zero (m at extremal value), not a numerical matrix element. +constexpr double kCoeffZeroThreshold = 1e-12; +} // namespace + +std::complex ModuleIO::cal_LzijR( + const std::unique_ptr& calculator, + const int it, const int ia, const int il, const int iz, const int mi, + const int jt, const int ja, const int jl, const int jz, const int mj, + const ModuleBase::Vector3& vR) +{ + if(mj == 0) + { + return std::complex(0.); + } + double val_ = 0; + calculator->calculate(it, il, iz, mi, jt, jl, jz, -mj, vR, &val_); + return kImag * static_cast(mj) * val_; +} + +std::complex ModuleIO::cal_LxijR( + const std::unique_ptr& calculator, + const int it, const int ia, const int il, const int iz, const int im, + const int jt, const int ja, const int jl, const int jz, const int jm, + const ModuleBase::Vector3& vR) +{ + const double lmbdp = _lambda_plus(jl, jm); + const double lmbdm = _lambda_minus(jl, jm); + // two-center-integral placeholders + double valp = 0.; + double valm = 0.; + if (jm > 1) + { + if (std::fabs(lmbdp) > kCoeffZeroThreshold) + { + calculator->calculate(it, il, iz, im, jt, jl, jz, -(jm+1), vR, &valp); + } + if (std::fabs(lmbdm) > kCoeffZeroThreshold) + { + calculator->calculate(it, il, iz, im, jt, jl, jz, -(jm-1), vR, &valm); + } + return kImag * 0.5 * (lmbdp * valp + lmbdm * valm); + } + if (jm == 1) + { + if (std::fabs(lmbdp) > kCoeffZeroThreshold) + { + calculator->calculate(it, il, iz, im, jt, jl, jz, -2, vR, &valp); + } + return kImag * 0.5 * lmbdp * valp; + } + if (jm == 0) + { + const double lmbd = _lambda_plus(jl, 0); // std::sqrt(jl*(jl+1)) + if (std::fabs(lmbd) > kCoeffZeroThreshold) + { + calculator->calculate(it, il, iz, im, jt, jl, jz, -1, vR, &valp); + } + return kImag * kInvSqrt2 * lmbd * valp; + } + if (jm == -1) + { + if (std::fabs(lmbdp) > kCoeffZeroThreshold) + { + calculator->calculate(it, il, iz, im, jt, jl, jz, 0, vR, &valp); + } + if (std::fabs(lmbdm) > kCoeffZeroThreshold) + { + calculator->calculate(it, il, iz, im, jt, jl, jz, 2, vR, &valm); + } + return -kImag * 0.5 * (std::sqrt(2) * lmbdp * valp + lmbdm * valm); + } + else + { + assert(jm < -1); // defensive check + if (std::fabs(lmbdp) > kCoeffZeroThreshold) + { + calculator->calculate(it, il, iz, im, jt, jl, jz, -(jm+1), vR, &valp); + } + if (std::fabs(lmbdm) > kCoeffZeroThreshold) + { + calculator->calculate(it, il, iz, im, jt, jl, jz, -(jm-1), vR, &valm); + } + return -kImag * 0.5 * (lmbdp * valp + lmbdm * valm); + } +} + +std::complex ModuleIO::cal_LyijR( + const std::unique_ptr& calculator, + const int it, const int ia, const int il, const int iz, const int im, + const int jt, const int ja, const int jl, const int jz, const int jm, + const ModuleBase::Vector3& vR) +{ + const double lmbdp = _lambda_plus(jl, jm); + const double lmbdm = _lambda_minus(jl, jm); + // two-center-integral placeholders + double valp = 0.; + double valm = 0.; + if (jm > 1) + { + if (std::fabs(lmbdp) > kCoeffZeroThreshold) + { + calculator->calculate(it, il, iz, im, jt, jl, jz, jm+1, vR, &valp); + } + if (std::fabs(lmbdm) > kCoeffZeroThreshold) + { + calculator->calculate(it, il, iz, im, jt, jl, jz, jm-1, vR, &valm); + } + return -kImag * 0.5 * (lmbdp * valp - lmbdm * valm); + } + if (jm == 1) + { + if (std::fabs(lmbdp) > kCoeffZeroThreshold) + { + calculator->calculate(it, il, iz, im, jt, jl, jz, 2, vR, &valp); + } + if (std::fabs(lmbdm) > kCoeffZeroThreshold) + { + calculator->calculate(it, il, iz, im, jt, jl, jz, 0, vR, &valm); + } + return -kImag * 0.5 * (lmbdp * valp - std::sqrt(2) * lmbdm * valm); + } + if (jm == 0) + { + const double lmbd = _lambda_plus(jl, 0); // std::sqrt(l*(l+1)) + if (std::fabs(lmbd) > kCoeffZeroThreshold) + { + calculator->calculate(it, il, iz, im, jt, jl, jz, 1, vR, &valp); + } + return -kImag * kInvSqrt2 * lmbd * valp; + } + if (jm == -1) + { + if (std::fabs(lmbdm) > kCoeffZeroThreshold) + { + calculator->calculate(it, il, iz, im, jt, jl, jz, -2, vR, &valm); + } + return -kImag * 0.5 * lmbdm * valm; + } + else + { + assert(jm < -1); // defensive check + if (std::fabs(lmbdp) > kCoeffZeroThreshold) + { + calculator->calculate(it, il, iz, im, jt, jl, jz, jm+1, vR, &valp); + } + if (std::fabs(lmbdm) > kCoeffZeroThreshold) + { + calculator->calculate(it, il, iz, im, jt, jl, jz, jm-1, vR, &valm); + } + return kImag * 0.5 * (lmbdp * valp - lmbdm * valm); + } +} + +ModuleIO::Angmom_op::Angmom_op( + const std::string& orbital_dir, + const UnitCell& ucell, + const double& search_radius, + const int tdestructor, + const int tgrid, + const int tatom, + const bool searchpbc, + const std::string& out_level, + const bool gamma_only, + std::ofstream* ptr_log, + const int rank) +{ + + this->ofs_ = ptr_log; + if (this->ofs_ == nullptr) + { + this->fallback_ofs_.open("/dev/null"); + this->ofs_ = &this->fallback_ofs_; + } + + *ofs_ << "\n\n\n\n"; + *ofs_ << " >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>" << std::endl; + *ofs_ << " | |" << std::endl; + *ofs_ << " | Angular momentum expectation value calculation: |" << std::endl; + *ofs_ << " | This is a post-processing step. The expectation value of operator |" << std::endl; + *ofs_ << " | Lx, Ly, Lz (, in which a and b are ABACUS numerical atomic |" << std::endl; + *ofs_ << " | orbitals) will be calculated. |" << std::endl; + *ofs_ << " | The result will be printed to file with name l{x,y,z}[g{step}]_nao.txt |" << std::endl; + *ofs_ << " | |" << std::endl; + *ofs_ << " <<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<" << std::endl; + *ofs_ << "\n\n\n\n"; + + int ntype_ = ucell.ntype; + Parallel_Common::bcast_int(ntype_); + std::vector forb(ntype_); + if (rank == 0) + { + for (int i = 0; i < ucell.ntype; ++i) + { + forb[i] = orbital_dir + ucell.orbital_fn[i]; + } + } + Parallel_Common::bcast_string(forb.data(), ntype_); + + this->orb_ = std::unique_ptr(new RadialCollection); + this->orb_->build(ucell.ntype, forb.data(), 'o'); + + ModuleBase::SphericalBesselTransformer sbt(true); + this->orb_->set_transformer(sbt); + + const double rcut_max = orb_->rcut_max(); + const int ngrid = int(rcut_max / 0.01) + 1; + const double cutoff = 2.0 * rcut_max; + this->orb_->set_uniform_grid(true, ngrid, cutoff, 'i', true); + + this->calculator_ = std::unique_ptr(new TwoCenterIntegrator); + this->calculator_->tabulate(*orb_, *orb_, 'S', ngrid, cutoff); + + // Initialize Ylm coefficients + ModuleBase::Ylm::set_coefficients(); + + // for neighbor list search + double temp = -1.0; + if (search_radius < rcut_max) + { + *ofs_ << "Find the `search_radius` from the input file being smaller than the \n" + "`rcut_max` of the orbitals.\n" + << "Reset the `search_radius` (" << search_radius << ") " + << "to `rcut_max` ("<< rcut_max << ")." + << std::endl; + // we don't really set, but use std::max to mask :) + } + temp = atom_arrange::set_sr_NL(*ofs_, + out_level, + std::max(search_radius, rcut_max), + ucell.infoNL->get_rcutmax_Beta(), + gamma_only); + temp = std::max(temp, std::max(search_radius, rcut_max)); + this->neighbor_searcher_ = std::unique_ptr(new Grid_Driver(tdestructor, tgrid)); + atom_arrange::search(searchpbc, + *ofs_, + *neighbor_searcher_, + ucell, + temp, + tatom); +} + +void ModuleIO::Angmom_op::kernel( + std::ofstream* ofs, + const UnitCell& ucell, + const char dir, + const int precision) +{ + if (ofs == nullptr || !ofs->is_open()) + { + return; + } + // an easy sanity check + assert(dir == 'x' || dir == 'y' || dir == 'z'); + + // it, ia, il, iz, im, iRx, iRy, iRz, jt, ja, jl, jz, jm + // the iRx, iRy, iRz are the indices of the supercell in which the two-center-integral + // it and jt are indexes of atomtypes, + // ia and ja are indexes of atoms within the atomtypes, + // il and jl are indexes of the angular momentum, + // iz and jz are indexes of the zeta functions + // im and jm are indexes of the magnetic quantum numbers. + std::string fmtstr = "%4d %4d %4d %4d %4d %4d %4d %4d %4d %4d %4d %4d %4d"; + fmtstr += " %" + std::to_string(precision*2) + "." + std::to_string(precision) + "e"; + fmtstr += " %" + std::to_string(precision*2) + "." + std::to_string(precision) + "e\n"; + FmtCore fmt(fmtstr); + + // placeholders + std::complex val = 0; + ModuleBase::Vector3 taui; // the origin position + ModuleBase::Vector3 dtau; // the displacement + AdjacentAtomInfo adjinfo; // adjacent atom information carrier + for (int it = 0; it < ucell.ntype; it++) + { + const Atom& atyp_i = ucell.atoms[it]; + for (int ia = 0; ia < atyp_i.na; ia++) + { + taui = ucell.get_tau(ucell.itia2iat(it, ia)); + neighbor_searcher_->Find_atom(ucell, taui, it, ia, &adjinfo); + for (int ia_adj = 0; ia_adj < adjinfo.adj_num + 1; ia_adj++) // "+1" is to include itself + { + int jt = adjinfo.ntype[ia_adj]; // ityp + int ja = adjinfo.natom[ia_adj]; // iat with in atomtype + const Atom& atyp_j = ucell.atoms[jt]; + const ModuleBase::Vector3 iR = adjinfo.box[ia_adj]; + dtau = ucell.cal_dtau(ucell.itia2iat(it, ia), + ucell.itia2iat(jt, ja), + iR) * ucell.lat0; // convert to unit of Bohr + + // nested loop: calculate the two-center-integral + for (int li = 0; li < atyp_i.nwl + 1; li++) + { + for (int iz = 0; iz < atyp_i.l_nchi[li]; iz++) + { + for (int mi = -li; mi <= li; mi++) + { + for (int lj = 0; lj < atyp_j.nwl + 1; lj++) + { + for (int jz = 0; jz < atyp_j.l_nchi[lj]; jz++) + { + for (int mj = -lj; mj <= lj; mj++) + { + if (dir == 'x') + { + val = cal_LxijR(calculator_, + it, ia, li, iz, mi, jt, ja, lj, jz, mj, dtau); + } + else if (dir == 'y') + { + val = cal_LyijR(calculator_, + it, ia, li, iz, mi, jt, ja, lj, jz, mj, dtau); + } + else if (dir == 'z') + { + val = cal_LzijR(calculator_, + it, ia, li, iz, mi, jt, ja, lj, jz, mj, dtau); + } + + *ofs << fmt.format( + it, ia, li, iz, mi, + iR.x, iR.y, iR.z, + jt, ja, lj, jz, mj, + val.real(), val.imag()); + } + } + } + } + } + } + } + } + } +} + +void ModuleIO::Angmom_op::calculate( + const std::string& prefix, + const std::string& outdir, + const UnitCell& ucell, + const int precision, + const int rank, + const int istep) +{ + ModuleBase::TITLE("Angmom_op", "calculate"); + ModuleBase::timer::start("Angmom_op", "calculate"); + + if (rank != 0) + { + ModuleBase::timer::end("Angmom_op", "calculate"); + return; + } + std::ofstream ofout; + const std::string dir = "xyz"; + /// @brief Filename suffix: if istep >= 0, append "g{istep+1}" (e.g., "g1") + /// to follow the out_freq_ion convention; otherwise no suffix. + const std::string step_suffix = (istep >= 0) ? "g" + std::to_string(istep + 1) : ""; + const std::string title = "# it ia il iz im iRx iRy iRz jt ja jl jz jm Re[] Im[]\n" + "# it: atomtype index of the first atom\n" + "# ia: atomic index of the first atom within the atomtype\n" + "# il: angular momentum index of the first atom\n" + "# iz: zeta function index of the first atom\n" + "# im: magnetic quantum number of the first atom\n" + "# iRx, iRy, iRz: the indices of the supercell\n" + "# jt: atomtype index of the second atom\n" + "# ja: atomic index of the second atom within the atomtype\n" + "# jl: angular momentum index of the second atom\n" + "# jz: zeta function index of the second atom\n" + "# jm: magnetic quantum number of the second atom\n" + "# Re[], Im[]: the real and imaginary parts " + "of the value of the matrix element\n"; + + for (char d : dir) + { + std::string fn = outdir + "l" + d + step_suffix + "_nao.txt"; + ofout.open(fn, std::ios::out); + ofout << title; + this->kernel(&ofout, ucell, d, precision); + ofout.close(); + } + ModuleBase::timer::end("Angmom_op", "calculate"); +} diff --git a/source/source_io/module_hs/angmom_op_mat.h b/source/source_io/module_hs/angmom_op_mat.h new file mode 100644 index 00000000000..c5000d68f32 --- /dev/null +++ b/source/source_io/module_hs/angmom_op_mat.h @@ -0,0 +1,275 @@ +#ifndef ANGMOM_OP_MAT_H +#define ANGMOM_OP_MAT_H + +/** + * calculate the matrix elements with the ACA (atom-centered + * approximation), in which the Lx/Ly/Lz are the angular momentum operators, + * |phi_i> and |phi_j> are the numerical atomic orbitals (NAOs). + * + * Note: this module is the most suitable for the case of SOC (spin-orbit coupling) + * calculation. A strict implmentation requires the evaluation on the position operator + * and the momentum operator, is beyond the scope of present implementation. + * + * Formulation + * ----------- + * + * Calculate the with ladder operator L+ and L-. + * + * The relation between Lx, Ly and L+, L- are: + * + * Lx = (L+ + L-) / 2 + * Ly = (L+ - L-) / 2i + * + * With L+, the COMPLEX spherical harmonic function Ylm (denoted as |l, m> in + * the following) can be raised: + * + * L+|l, m> = sqrt((l-m)(l+m+1))|l, m+1> + * + * Likely, with L-, the spherical harmonic function Ylm can be lowered: + * + * L-|l, m> = sqrt((l+m)(l-m+1))|l, m-1> + * + * Therefore, for the case where there is only one atom, the Lx matrix element + * can be calculated as: + * + * = sqrt((l-m)(l+m+1)) * delta(m, m'+1) / 2 + * + sqrt((l+m)(l-m+1)) * delta(m, m'-1) / 2 + * + * Likewise the Ly matrix element can be calculated as: + * + * = sqrt((l-m)(l+m+1)) * delta(m, m'+1) / 2i + * - sqrt((l+m)(l-m+1)) * delta(m, m'-1) / 2i + * + * and the Lz matrix element can be calculated as: + * + * = m * delta(m, m') + * + * ABACUS employs the REAL spherical harmonics, the transformation between the + * REAL and COMPLEX spherical harmonics can be found from: + * https://abacus.deepmodeling.com/en/latest/advanced/pp_orb.html + * + * For the case where there are multiple atoms, present calculation is based on + * the Atom-Centered-Approximation (ACA), in which the global angular momentum + * operator is decomposed to the sum of the local opeartor that only considers + * one atom. Therefore, the matrix element can be calculated by combining the + * ladder operators with the two-center-integral technique: + * + * = ) + * + * the two-center-integral will be performed after the application of the L + * operator. For example after the application of Lz, the equation above yields + * + * = mj + * + * For Lx = 1/2 * (L+ + L-): + * + * = sqrt((lj-mj)(lj+mj+1)) * / 2 + * + sqrt((lj+mj)(lj-mj+1)) * / 2 + * + * Technical Details + * ----------------- + * + * 0. The calculation of matrix elements involves the two-center-integral calculation, + * this is supported by ABACUS built-in class TwoCenterIntegrator. + * see: source/source_basis/module_nao/two_center_integrator.h. + * + * 1. The interface of it is RadialCollection, which is a collection of radial functions. + * see: source/source_basis/module_nao/radial_collection.h + * + * 2. The radial functions are stored in AtomicRadials class, + * see: source/source_basis/module_nao/atomic_radials.h + * + * 3. The construction of AtomicRadials involves the filename of orbital, it is stored + * in the UnitCell instance + * + * 4. The calculation will run over all supercells in which the two-center-integral + * is not zero. This is done by the class SltkGridDriver, which is a driver for + * searching the neighboring cells. + */ +#include +#include +#include +#include +#include +#include +#include +#include "source_cell/unitcell.h" +#include "source_basis/module_nao/two_center_integrator.h" +#include "source_cell/module_neighbor/sltk_grid_driver.h" +#include "source_cell/module_neighbor/sltk_atom_arrange.h" + +namespace ModuleIO +{ + /** + * @brief calculate the matrix elements, in which the Lz + * are the angular momentum operators, |phi_i> and |phi_j> are the numerical + * atomic orbitals (NAOs). + * + * @param calculator the std::unique_ptr instance + * @param it atomtype index of the first atom + * @param ia atomic index of the first atom within the atomtype + * @param il angular momentum index of the first atom + * @param iz zeta function index of the first atom + * @param mi magnetic quantum number of the first atom + * @param jt atomtype index of the second atom + * @param ja atomic index of the second atom within the atomtype + * @param jl angular momentum index of the second atom + * @param jz zeta function index of the second atom + * @param mj magnetic quantum number of the second atom + * @param vR the vector from the first atom to the second atom + * @return std::complex + */ + std::complex cal_LzijR( + const std::unique_ptr& calculator, + const int it, const int ia, const int il, const int iz, const int mi, + const int jt, const int ja, const int jl, const int jz, const int mj, + const ModuleBase::Vector3& vR); + + /** + * @brief calculate the matrix elements, in which the Lz + * are the angular momentum operators, |phi_i> and |phi_j> are the numerical + * atomic orbitals (NAOs). + * + * @param calculator the std::unique_ptr instance + * @param it atomtype index of the first atom + * @param ia atomic index of the first atom within the atomtype + * @param il angular momentum index of the first atom + * @param iz zeta function index of the first atom + * @param mi magnetic quantum number of the first atom + * @param jt atomtype index of the second atom + * @param ja atomic index of the second atom within the atomtype + * @param jl angular momentum index of the second atom + * @param jz zeta function index of the second atom + * @param mj magnetic quantum number of the second atom + * @param vR the vector from the first atom to the second atom + * @return std::complex + */ + std::complex cal_LyijR( + const std::unique_ptr& calculator, + const int it, const int ia, const int il, const int iz, const int im, + const int jt, const int ja, const int jl, const int jz, const int jm, + const ModuleBase::Vector3& vR); + + /** + * @brief calculate the matrix elements, in which the Lz + * are the angular momentum operators, |phi_i> and |phi_j> are the numerical + * atomic orbitals (NAOs). + * + * @param calculator the std::unique_ptr instance + * @param it atomtype index of the first atom + * @param ia atomic index of the first atom within the atomtype + * @param il angular momentum index of the first atom + * @param iz zeta function index of the first atom + * @param mi magnetic quantum number of the first atom + * @param jt atomtype index of the second atom + * @param ja atomic index of the second atom within the atomtype + * @param jl angular momentum index of the second atom + * @param jz zeta function index of the second atom + * @param mj magnetic quantum number of the second atom + * @param vR the vector from the first atom to the second atom + * @return std::complex + */ + std::complex cal_LxijR( + const std::unique_ptr& calculator, + const int it, const int ia, const int il, const int iz, const int im, + const int jt, const int ja, const int jl, const int jz, const int jm, + const ModuleBase::Vector3& vR); + + // the calculation of matrix elements will be outputted + // in the way that indexed by: + // it, ia, il, iz, im, iRx, iRy, iRz, jt, ja, jl, jz, jm, in which the + // iRx, iRy, iRz are the indices of the supercell in which the two-center-integral + // it and jt are indexes of atomtypes, + // ia and ja are indexes of atoms within the atomtypes, + // il and jl are indexes of the angular momentum, + // iz and jz are indexes of the zeta functions + // im and jm are indexes of the magnetic quantum numbers. + // The output is a complex number, which is the value of the matrix element. + // Always the matrix is quite large, so direct print to file. + class Angmom_op + { + public: + // the default constructor is meaningless + Angmom_op() = delete; + /** + * @brief Construct a new Angular Momentum Expectation Calculator object + * + * @param orbital_dir the directory of the orbital file + * @param ucell the unit cell object + * @param search_radius the search radius for the neighboring atoms + * @param tdestructor test flag, for destructor + * @param tgrid test flag, for grid + * @param tatom test flag, for atom input + * @param searchpbc + * @param out_level the output verbosity level + * @param gamma_only whether the calculation is gamma-only + * @param ptr_log pointer to the ofstream object for logging + */ + Angmom_op( + const std::string& orbital_dir, + const UnitCell& ucell, + const double& search_radius, + const int tdestructor, + const int tgrid, + const int tatom, + const bool searchpbc, + const std::string& out_level, + const bool gamma_only, + std::ofstream* ptr_log = nullptr, + const int rank = 0); + ~Angmom_op() = default; + + /** + * @brief Calculate and write matrix elements. + * + * @param prefix the suffix of the output files (currently unused; + * filenames are fixed as l{x,y,z}[g{step}]_nao.txt) + * @param outdir the output directory + * @param ucell the unit cell object + * @param precision the precision of the output, default is 10 + * @param rank the MPI rank, default is 0 + * @param istep the ionic step index. If >= 0, the filename includes + * `g{istep+1}` (e.g., lxg1_nao.txt); if -1, the filename + * has no step suffix (e.g., lx_nao.txt). Default is -1 + * (single-step overwrite mode). + */ + void calculate(const std::string& prefix, + const std::string& outdir, + const UnitCell& ucell, + const int precision = 10, + const int rank = 0, + const int istep = -1); + + private: + std::ofstream fallback_ofs_; + // ofsrunning + std::ofstream* ofs_ = nullptr; + // the two-center-integrator + std::unique_ptr calculator_; + // the spherical bessel transformer + ModuleBase::SphericalBesselTransformer sbt_; + // the radial collection + std::unique_ptr orb_; + + // neighboring searcher + std::unique_ptr neighbor_searcher_; + + /** + * @brief calculate the matrix elements. Due to + * the large size of the matrix, the result will be printed to file + * directly. + * + * @param ofs pointer to the ofstream object for printing, if nullptr, + * the result will not be printed + * @param ucell the unit cell object + * @param dir the direction of the angular momentum operator, 'x', 'y' or 'z' + * @param precision the precision of the output, default is 10 + */ + void kernel(std::ofstream* ofs, + const UnitCell& ucell, + const char dir = 'x', + const int precision = 10); + }; +} // namespace ModuleIO + +#endif // ANGMOM_OP_MAT_H diff --git a/source/source_io/module_hs/cal_plpr.cpp b/source/source_io/module_hs/cal_plpr.cpp deleted file mode 100644 index 60cc7809388..00000000000 --- a/source/source_io/module_hs/cal_plpr.cpp +++ /dev/null @@ -1,390 +0,0 @@ -#include -#include -#include -#include -#include -#include -#include -#include -#include "source_cell/unitcell.h" -#include "source_base/sph_bessel_tf.h" -#include "source_basis/module_nao/two_center_integrator.h" -#include "source_cell/module_neighbor/sltk_grid_driver.h" -#include "source_cell/module_neighbor/sltk_atom_arrange.h" -#include "source_io/module_parameter/parameter.h" -#include "source_io/module_hs/cal_plpr.h" -#include "source_base/formatter.h" -#include "source_base/parallel_common.h" -/** - * - * FIXME: the following part will be transfered to TwoCenterIntegrator soon - * - * Notation - * -------- - * ylm: complex spherical harmonics - * slm: solid (real) spherical harmonics - * - * Changelog - * --------- - * Switch to support the solid spherical harmonics to keep consistent with - * implementations of other parts. - * Formulation (in Chinese): - * https://my.feishu.cn/wiki/D0enwcUKfiJgtSkJ5scc9Dagntc - */ - -// L+ylm = sqrt((l-m)(l+m+1))ylm+1, return the sqrt((l-m)(l+m+1)) -double _lambda_plus(const int l, const int m) -{ - return std::sqrt((l - m) * (l + m + 1)); // NOTE: complex spherical harmonics -} - -// L-ylm = sqrt((l+m)(l-m+1))ylm-1, return the sqrt((l+m)(l-m+1)) -double _lambda_minus(const int l, const int m) -{ - return std::sqrt((l + m) * (l - m + 1)); // NOTE: complex spherical harmonics -} - -const std::complex i = {0., 1.}; -const double invsqrt2 = std::sqrt(2) * 0.5; - -std::complex ModuleIO::cal_LzijR( - const std::unique_ptr& calculator, - const int it, const int ia, const int il, const int iz, const int mi, - const int jt, const int ja, const int jl, const int jz, const int mj, - const ModuleBase::Vector3& vR) -{ - if(mj == 0) { - return std::complex(0.); - } - double val_ = 0; - calculator->calculate(it, il, iz, mi, jt, jl, jz, -mj, vR, &val_); - return i * static_cast(mj) * val_; -} - -std::complex ModuleIO::cal_LxijR( - const std::unique_ptr& calculator, - const int it, const int ia, const int il, const int iz, const int im, - const int jt, const int ja, const int jl, const int jz, const int jm, - const ModuleBase::Vector3& vR) -{ - const double lmbdp = _lambda_plus(jl, jm); - const double lmbdm = _lambda_minus(jl, jm); - // two-center-integral placeholders - double valp = 0.; - double valm = 0.; - if (jm > 1) { - if (std::fabs(lmbdp) > 1e-12) { - calculator->calculate(it, il, iz, im, jt, jl, jz, -(jm+1), vR, &valp); - } - if (std::fabs(lmbdm) > 1e-12) { - calculator->calculate(it, il, iz, im, jt, jl, jz, -(jm-1), vR, &valm); - } - return i * 0.5 * (lmbdp * valp + lmbdm * valm); - } - if (jm == 1) { - if (std::fabs(lmbdp) > 1e-12) { - calculator->calculate(it, il, iz, im, jt, jl, jz, -2, vR, &valp); - } - return i * 0.5 * lmbdp * valp; - } - if (jm == 0) { - const double lmbd = _lambda_plus(jl, 0); // std::sqrt(jl*(jl+1)) - if (std::fabs(lmbd) > 1e-12) { - calculator->calculate(it, il, iz, im, jt, jl, jz, -1, vR, &valp); - } - return i * invsqrt2 * lmbd * valp; - } - if (jm == -1) { - if (std::fabs(lmbdp) > 1e-12) { - calculator->calculate(it, il, iz, im, jt, jl, jz, 0, vR, &valp); - } - if (std::fabs(lmbdm) > 1e-12) { - calculator->calculate(it, il, iz, im, jt, jl, jz, 2, vR, &valm); - } - return -i * 0.5 * (std::sqrt(2) * lmbdp * valp + lmbdm * valm); - } - else { - assert(jm < -1); // defensive check - if (std::fabs(lmbdp) > 1e-12) { - calculator->calculate(it, il, iz, im, jt, jl, jz, -(jm+1), vR, &valp); - } - if (std::fabs(lmbdm) > 1e-12) { - calculator->calculate(it, il, iz, im, jt, jl, jz, -(jm-1), vR, &valm); - } - return -i * 0.5 * (lmbdp * valp + lmbdm * valm); - } -} - -std::complex ModuleIO::cal_LyijR( - const std::unique_ptr& calculator, - const int it, const int ia, const int il, const int iz, const int im, - const int jt, const int ja, const int jl, const int jz, const int jm, - const ModuleBase::Vector3& vR) -{ - const double lmbdp = _lambda_plus(jl, jm); - const double lmbdm = _lambda_minus(jl, jm); - // two-center-integral placeholders - double valp = 0.; - double valm = 0.; - if (jm > 1) { - if (std::fabs(lmbdp) > 1e-12) { - calculator->calculate(it, il, iz, im, jt, jl, jz, jm+1, vR, &valp); - } - if (std::fabs(lmbdm) > 1e-12) { - calculator->calculate(it, il, iz, im, jt, jl, jz, jm-1, vR, &valm); - } - return -i * 0.5 * (lmbdp * valp - lmbdm * valm); - } - if (jm == 1) { - if (std::fabs(lmbdp) > 1e-12) { - calculator->calculate(it, il, iz, im, jt, jl, jz, 2, vR, &valp); - } - if (std::fabs(lmbdm) > 1e-12) { - calculator->calculate(it, il, iz, im, jt, jl, jz, 0, vR, &valm); - } - return -i * 0.5 * (lmbdp * valp - std::sqrt(2) * lmbdm * valm); - } - if (jm == 0) { - const double lmbd = _lambda_plus(jl, 0); // std::sqrt(l*(l+1)) - if (std::fabs(lmbd) > 1e-12) { - calculator->calculate(it, il, iz, im, jt, jl, jz, 1, vR, &valp); - } - return -i * invsqrt2 * lmbd * valp; - } - if (jm == -1) { - if (std::fabs(lmbdm) > 1e-12) { - calculator->calculate(it, il, iz, im, jt, jl, jz, -2, vR, &valm); - } - return -i * 0.5 * lmbdm * valm; - } - else { - assert(jm < -1); // defensive check - if (std::fabs(lmbdp) > 1e-12) { - calculator->calculate(it, il, iz, im, jt, jl, jz, jm+1, vR, &valp); - } - if (std::fabs(lmbdm) > 1e-12) { - calculator->calculate(it, il, iz, im, jt, jl, jz, jm-1, vR, &valm); - } - return i * 0.5 * (lmbdp * valp - lmbdm * valm); - } -} - -ModuleIO::AngularMomentumCalculator::AngularMomentumCalculator( - const std::string& orbital_dir, - const UnitCell& ucell, - const double& search_radius, - const int tdestructor, - const int tgrid, - const int tatom, - const bool searchpbc, - std::ofstream* ptr_log, - const int rank) -{ - - this->ofs_ = ptr_log; - if (this->ofs_ == nullptr) - { - this->fallback_ofs_.open("/dev/null"); - this->ofs_ = &this->fallback_ofs_; - } - - *ofs_ << "\n\n\n\n"; - *ofs_ << " >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>" << std::endl; - *ofs_ << " | |" << std::endl; - *ofs_ << " | Angular momentum expectation value calculation: |" << std::endl; - *ofs_ << " | This is a post-processing step. The expectation value of operator |" << std::endl; - *ofs_ << " | Lx, Ly, Lz (, in which a and b are ABACUS numerical atomic |" << std::endl; - *ofs_ << " | orbitals) will be calculated. |" << std::endl; - *ofs_ << " | The result will be printed to file with name ${suffix}_Lx/y/z.dat |" << std::endl; - *ofs_ << " | |" << std::endl; - *ofs_ << " <<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<" << std::endl; - *ofs_ << "\n\n\n\n"; - - int ntype_ = ucell.ntype; -#ifdef __MPI - Parallel_Common::bcast_int(ntype_); -#endif - std::vector forb(ntype_); - if (rank == 0) - { - for (int i = 0; i < ucell.ntype; ++i) - { - forb[i] = orbital_dir + ucell.orbital_fn[i]; - } - } -#ifdef __MPI - Parallel_Common::bcast_string(forb.data(), ntype_); -#endif - - this->orb_ = std::unique_ptr(new RadialCollection); - this->orb_->build(ucell.ntype, forb.data(), 'o'); - - ModuleBase::SphericalBesselTransformer sbt(true); - this->orb_->set_transformer(sbt); - - const double rcut_max = orb_->rcut_max(); - const int ngrid = int(rcut_max / 0.01) + 1; - const double cutoff = 2.0 * rcut_max; - this->orb_->set_uniform_grid(true, ngrid, cutoff, 'i', true); - - this->calculator_ = std::unique_ptr(new TwoCenterIntegrator); - this->calculator_->tabulate(*orb_, *orb_, 'S', ngrid, cutoff); - - // Initialize Ylm coefficients - ModuleBase::Ylm::set_coefficients(); - - // for neighbor list search - double temp = -1.0; - if (search_radius < rcut_max) - { - *ofs_ << "Find the `search_radius` from the input file being smaller than the \n" - "`rcut_max` of the orbitals.\n" - << "Reset the `search_radius` (" << search_radius << ") " - << "to `rcut_max` ("<< rcut_max << ")." - << std::endl; - // we don't really set, but use std::max to mask :) - } - temp = atom_arrange::set_sr_NL(*ofs_, - PARAM.inp.out_level, - std::max(search_radius, rcut_max), - ucell.infoNL->get_rcutmax_Beta(), - PARAM.globalv.gamma_only_local); - temp = std::max(temp, std::max(search_radius, rcut_max)); - this->neighbor_searcher_ = std::unique_ptr(new Grid_Driver(tdestructor, tgrid)); - atom_arrange::search(searchpbc, - *ofs_, - *neighbor_searcher_, - ucell, - temp, - tatom); -} - -void ModuleIO::AngularMomentumCalculator::kernel( - std::ofstream* ofs, - const UnitCell& ucell, - const char dir, - const int precision) -{ - if (ofs == nullptr || !ofs->is_open()) - { - return; - } - // an easy sanity check - assert(dir == 'x' || dir == 'y' || dir == 'z'); - - // it, ia, il, iz, im, iRx, iRy, iRz, jt, ja, jl, jz, jm - // the iRx, iRy, iRz are the indices of the supercell in which the two-center-integral - // it and jt are indexes of atomtypes, - // ia and ja are indexes of atoms within the atomtypes, - // il and jl are indexes of the angular momentum, - // iz and jz are indexes of the zeta functions - // im and jm are indexes of the magnetic quantum numbers. - std::string fmtstr = "%4d %4d %4d %4d %4d %4d %4d %4d %4d %4d %4d %4d %4d"; - fmtstr += " %" + std::to_string(precision*2) + "." + std::to_string(precision) + "e"; - fmtstr += " %" + std::to_string(precision*2) + "." + std::to_string(precision) + "e\n"; - FmtCore fmt(fmtstr); - - // placeholders - std::complex val = 0; - ModuleBase::Vector3 taui; // the origin position - ModuleBase::Vector3 dtau; // the displacement - AdjacentAtomInfo adjinfo; // adjacent atom information carrier - for (int it = 0; it < ucell.ntype; it++) - { - const Atom& atyp_i = ucell.atoms[it]; - for (int ia = 0; ia < atyp_i.na; ia++) - { - taui = ucell.get_tau(ucell.itia2iat(it, ia)); - neighbor_searcher_->Find_atom(ucell, taui, it, ia, &adjinfo); - for (int ia_adj = 0; ia_adj < adjinfo.adj_num + 1; ia_adj++) // "+1" is to include itself - { - int jt = adjinfo.ntype[ia_adj]; // ityp - int ja = adjinfo.natom[ia_adj]; // iat with in atomtype - const Atom& atyp_j = ucell.atoms[jt]; - const ModuleBase::Vector3 iR = adjinfo.box[ia_adj]; - dtau = ucell.cal_dtau(ucell.itia2iat(it, ia), - ucell.itia2iat(jt, ja), - iR) * ucell.lat0; // convert to unit of Bohr - - // nested loop: calculate the two-center-integral - for (int li = 0; li < atyp_i.nwl + 1; li++) - { - for (int iz = 0; iz < atyp_i.l_nchi[li]; iz++) - { - for (int mi = -li; mi <= li; mi++) - { - for (int lj = 0; lj < atyp_j.nwl + 1; lj++) - { - for (int jz = 0; jz < atyp_j.l_nchi[lj]; jz++) - { - for (int mj = -lj; mj <= lj; mj++) - { - if (dir == 'x') - { - val = cal_LxijR(calculator_, - it, ia, li, iz, mi, jt, ja, lj, jz, mj, dtau); - } - else if (dir == 'y') - { - val = cal_LyijR(calculator_, - it, ia, li, iz, mi, jt, ja, lj, jz, mj, dtau); - } - else if (dir == 'z') - { - val = cal_LzijR(calculator_, - it, ia, li, iz, mi, jt, ja, lj, jz, mj, dtau); - } - - *ofs << fmt.format( - it, ia, li, iz, mi, - iR.x, iR.y, iR.z, - jt, ja, lj, jz, mj, - val.real(), val.imag()); - } - } - } - } - } - } - } - } - } -} - -void ModuleIO::AngularMomentumCalculator::calculate( - const std::string& prefix, - const std::string& outdir, - const UnitCell& ucell, - const int precision, - const int rank) -{ - if (rank != 0) - { - return; - } - std::ofstream ofout; - const std::string dir = "xyz"; - const std::string title = "# it ia il iz im iRx iRy iRz jt ja jl jz jm Re[] Im[]\n" - "# it: atomtype index of the first atom\n" - "# ia: atomic index of the first atom within the atomtype\n" - "# il: angular momentum index of the first atom\n" - "# iz: zeta function index of the first atom\n" - "# im: magnetic quantum number of the first atom\n" - "# iRx, iRy, iRz: the indices of the supercell\n" - "# jt: atomtype index of the second atom\n" - "# ja: atomic index of the second atom within the atomtype\n" - "# jl: angular momentum index of the second atom\n" - "# jz: zeta function index of the second atom\n" - "# jm: magnetic quantum number of the second atom\n" - "# Re[], Im[]: the real and imaginary parts " - "of the value of the matrix element\n"; - - for (char d : dir) - { - std::string fn = outdir + prefix + "_L" + d + ".dat"; - ofout.open(fn, std::ios::out); - ofout << title; - this->kernel(&ofout, ucell, d, precision); - ofout.close(); - } -} diff --git a/source/source_io/module_hs/cal_plpr.h b/source/source_io/module_hs/cal_plpr.h deleted file mode 100644 index c4694596cec..00000000000 --- a/source/source_io/module_hs/cal_plpr.h +++ /dev/null @@ -1,251 +0,0 @@ -/** - * calculate the matrix elements with the ACA (atom-centered - * approximation), in which the Lx/Ly/Lz are the angular momentum operators, - * |phi_i> and |phi_j> are the numerical atomic orbitals (NAOs). - * - * Note: this module is the most suitable for the case of SOC (spin-orbit coupling) - * calculation. A strict implmentation requires the evaluation on the position operator - * and the momentum operator, is beyond the scope of present implementation. - * - * Formulation - * ----------- - * - * Calculate the with ladder operator L+ and L-. - * - * The relation between Lx, Ly and L+, L- are: - * - * Lx = (L+ + L-) / 2 - * Ly = (L+ - L-) / 2i - * - * With L+, the COMPLEX spherical harmonic function Ylm (denoted as |l, m> in - * the following) can be raised: - * - * L+|l, m> = sqrt((l-m)(l+m+1))|l, m+1> - * - * Likely, with L-, the spherical harmonic function Ylm can be lowered: - * - * L-|l, m> = sqrt((l+m)(l-m+1))|l, m-1> - * - * Therefore, for the case where there is only one atom, the Lx matrix element - * can be calculated as: - * - * = sqrt((l-m)(l+m+1)) * delta(m, m'+1) / 2 - * + sqrt((l+m)(l-m+1)) * delta(m, m'-1) / 2 - * - * Likewise the Ly matrix element can be calculated as: - * - * = sqrt((l-m)(l+m+1)) * delta(m, m'+1) / 2i - * - sqrt((l+m)(l-m+1)) * delta(m, m'-1) / 2i - * - * and the Lz matrix element can be calculated as: - * - * = m * delta(m, m') - * - * ABACUS employs the REAL spherical harmonics, the transformation between the - * REAL and COMPLEX spherical harmonics can be found from: - * https://abacus.deepmodeling.com/en/latest/advanced/pp_orb.html - * - * For the case where there are multiple atoms, present calculation is based on - * the Atom-Centered-Approximation (ACA), in which the global angular momentum - * operator is decomposed to the sum of the local opeartor that only considers - * one atom. Therefore, the matrix element can be calculated by combining the - * ladder operators with the two-center-integral technique: - * - * = ) - * - * the two-center-integral will be performed after the application of the L - * operator. For example after the application of Lz, the equation above yields - * - * = mj - * - * For Lx = 1/2 * (L+ + L-): - * - * = sqrt((lj-mj)(lj+mj+1)) * / 2 - * + sqrt((lj+mj)(lj-mj+1)) * / 2 - * - * Technical Details - * ----------------- - * - * 0. The calculation of matrix elements involves the two-center-integral calculation, - * this is supported by ABACUS built-in class TwoCenterIntegrator. - * see: source/source_basis/module_nao/two_center_integrator.h. - * - * 1. The interface of it is RadialCollection, which is a collection of radial functions. - * see: source/source_basis/module_nao/radial_collection.h - * - * 2. The radial functions are stored in AtomicRadials class, - * see: source/source_basis/module_nao/atomic_radials.h - * - * 3. The construction of AtomicRadials involves the filename of orbital, it is stored - * in the UnitCell instance - * - * 4. The calculation will run over all supercells in which the two-center-integral - * is not zero. This is done by the class SltkGridDriver, which is a driver for - * searching the neighboring cells. - */ -#include -#include -#include -#include -#include -#include -#include -#include "source_cell/unitcell.h" -#include "source_basis/module_nao/two_center_integrator.h" -#include "source_cell/module_neighbor/sltk_grid_driver.h" -#include "source_cell/module_neighbor/sltk_atom_arrange.h" - -namespace ModuleIO -{ - /** - * @brief calculate the matrix elements, in which the Lz - * are the angular momentum operators, |phi_i> and |phi_j> are the numerical - * atomic orbitals (NAOs). - * - * @param calculator the std::unique_ptr instance - * @param it atomtype index of the first atom - * @param ia atomic index of the first atom within the atomtype - * @param il angular momentum index of the first atom - * @param iz zeta function index of the first atom - * @param mi magnetic quantum number of the first atom - * @param jt atomtype index of the second atom - * @param ja atomic index of the second atom within the atomtype - * @param jl angular momentum index of the second atom - * @param jz zeta function index of the second atom - * @param mj magnetic quantum number of the second atom - * @param vR the vector from the first atom to the second atom - * @return std::complex - */ - std::complex cal_LzijR( - const std::unique_ptr& calculator, - const int it, const int ia, const int il, const int iz, const int mi, - const int jt, const int ja, const int jl, const int jz, const int mj, - const ModuleBase::Vector3& vR); - - /** - * @brief calculate the matrix elements, in which the Lz - * are the angular momentum operators, |phi_i> and |phi_j> are the numerical - * atomic orbitals (NAOs). - * - * @param calculator the std::unique_ptr instance - * @param it atomtype index of the first atom - * @param ia atomic index of the first atom within the atomtype - * @param il angular momentum index of the first atom - * @param iz zeta function index of the first atom - * @param mi magnetic quantum number of the first atom - * @param jt atomtype index of the second atom - * @param ja atomic index of the second atom within the atomtype - * @param jl angular momentum index of the second atom - * @param jz zeta function index of the second atom - * @param mj magnetic quantum number of the second atom - * @param vR the vector from the first atom to the second atom - * @return std::complex - */ - std::complex cal_LyijR( - const std::unique_ptr& calculator, - const int it, const int ia, const int il, const int iz, const int im, - const int jt, const int ja, const int jl, const int jz, const int jm, - const ModuleBase::Vector3& vR); - - /** - * @brief calculate the matrix elements, in which the Lz - * are the angular momentum operators, |phi_i> and |phi_j> are the numerical - * atomic orbitals (NAOs). - * - * @param calculator the std::unique_ptr instance - * @param it atomtype index of the first atom - * @param ia atomic index of the first atom within the atomtype - * @param il angular momentum index of the first atom - * @param iz zeta function index of the first atom - * @param mi magnetic quantum number of the first atom - * @param jt atomtype index of the second atom - * @param ja atomic index of the second atom within the atomtype - * @param jl angular momentum index of the second atom - * @param jz zeta function index of the second atom - * @param mj magnetic quantum number of the second atom - * @param vR the vector from the first atom to the second atom - * @return std::complex - */ - std::complex cal_LxijR( - const std::unique_ptr& calculator, - const int it, const int ia, const int il, const int iz, const int im, - const int jt, const int ja, const int jl, const int jz, const int jm, - const ModuleBase::Vector3& vR); - - // the calculation of matrix elements will be outputted - // in the way that indexed by: - // it, ia, il, iz, im, iRx, iRy, iRz, jt, ja, jl, jz, jm, in which the - // iRx, iRy, iRz are the indices of the supercell in which the two-center-integral - // it and jt are indexes of atomtypes, - // ia and ja are indexes of atoms within the atomtypes, - // il and jl are indexes of the angular momentum, - // iz and jz are indexes of the zeta functions - // im and jm are indexes of the magnetic quantum numbers. - // The output is a complex number, which is the value of the matrix element. - // Always the matrix is quite large, so direct print to file. - class AngularMomentumCalculator - { - public: - // the default constructor is meaningless - AngularMomentumCalculator() = delete; - /** - * @brief Construct a new Angular Momentum Expectation Calculator object - * - * @param orbital_dir the directory of the orbital file - * @param ucell the unit cell object - * @param search_radius the search radius for the neighboring atoms - * @param tdestructor test flag, for destructor - * @param tgrid test flag, for grid - * @param tatom test flag, for atom input - * @param searchpbc - * @param ptr_log pointer to the ofstream object for logging - */ - AngularMomentumCalculator( - const std::string& orbital_dir, - const UnitCell& ucell, - const double& search_radius, - const int tdestructor, - const int tgrid, - const int tatom, - const bool searchpbc, - std::ofstream* ptr_log = nullptr, - const int rank = 0); - ~AngularMomentumCalculator() = default; - - void calculate(const std::string& prefix, - const std::string& outdir, - const UnitCell& ucell, - const int precision = 10, - const int rank = 0); - - private: - std::ofstream fallback_ofs_; - // ofsrunning - std::ofstream* ofs_ = nullptr; - // the two-center-integrator - std::unique_ptr calculator_; - // the spherical bessel transformer - ModuleBase::SphericalBesselTransformer sbt_; - // the radial collection - std::unique_ptr orb_; - - // neighboring searcher - std::unique_ptr neighbor_searcher_; - - /** - * @brief calculate the matrix elements. Due to - * the large size of the matrix, the result will be printed to file - * directly. - * - * @param ofs pointer to the ofstream object for printing, if nullptr, - * the result will not be printed - * @param ucell the unit cell object - * @param dir the direction of the angular momentum operator, 'x', 'y' or 'z' - * @param precision the precision of the output, default is 10 - */ - void kernel(std::ofstream* ofs, - const UnitCell& ucell, - const char dir = 'x', - const int precision = 10); - }; -} // namespace ModuleIO diff --git a/source/source_io/module_hs/cal_r_overlap_r.cpp b/source/source_io/module_hs/cal_r_overlap_r.cpp deleted file mode 100644 index 4c93a48eefc..00000000000 --- a/source/source_io/module_hs/cal_r_overlap_r.cpp +++ /dev/null @@ -1,1092 +0,0 @@ -#include "cal_r_overlap_r.h" - -#include "rr_sparse_writer.h" -#include "single_r_io.h" -#include "source_base/mathzone_add1.h" -#include "source_base/parallel_reduce.h" -#include "source_base/timer.h" -#include "source_base/tool_quit.h" -#include "source_cell/module_neighbor/sltk_grid_driver.h" -#include "source_io/module_parameter/parameter.h" -#include "source_cell/nonlocal_info_base.h" - -cal_r_overlap_R::cal_r_overlap_R() -{ -} - -cal_r_overlap_R::~cal_r_overlap_R() -{ -} - -void cal_r_overlap_R::initialize_orb_table(const UnitCell& ucell, const LCAO_Orbitals& orb) -{ - const int ntype = orb.get_ntype(); - int lmax_orb = -1; - for (int it = 0; it < ntype; it++) - { - lmax_orb = std::max(lmax_orb, orb.Phi[it].getLmax()); - } - const double dr = orb.get_dR(); - const double dk = orb.get_dk(); - const int kmesh = orb.get_kmesh() * 4 + 1; - int Rmesh = static_cast(orb.get_Rmax() / dr) + 4; - Rmesh += 1 - Rmesh % 2; - - const int Lmax = lmax_orb + 1; - const int Lmax_used = 2 * lmax_orb + 1; - Center2_Orb::init_Table_Spherical_Bessel(Lmax_used, dr, dk, kmesh, Rmesh, psb_); - ModuleBase::Ylm::set_coefficients(); - MGT.init_Gaunt_CH(Lmax); - MGT.init_Gaunt(Lmax); -} - -void cal_r_overlap_R::construct_orbs_and_orb_r(const UnitCell& ucell, const LCAO_Orbitals& orb) -{ - int orb_r_ntype = 0; - int mat_Nr = orb.Phi[0].PhiLN(0, 0).getNr(); - int count_Nr = 0; - - orbs.resize(orb.get_ntype()); - for (int T = 0; T < orb.get_ntype(); ++T) - { - count_Nr = orb.Phi[T].PhiLN(0, 0).getNr(); - if (count_Nr > mat_Nr) - { - mat_Nr = count_Nr; - orb_r_ntype = T; - } - - orbs[T].resize(orb.Phi[T].getLmax() + 1); - for (int L = 0; L <= orb.Phi[T].getLmax(); ++L) - { - orbs[T][L].resize(orb.Phi[T].getNchi(L)); - for (int N = 0; N < orb.Phi[T].getNchi(L); ++N) - { - const auto& orb_origin = orb.Phi[T].PhiLN(L, N); - orbs[T][L][N].set_orbital_info(orb_origin.getLabel(), - orb_origin.getType(), - orb_origin.getL(), - orb_origin.getChi(), - orb_origin.getNr(), - orb_origin.getRab(), - orb_origin.getRadial(), - Numerical_Orbital_Lm::Psi_Type::Psi, - orb_origin.getPsi(), - static_cast(orb_origin.getNk() * kmesh_times) | 1, - orb_origin.getDk(), - orb_origin.getDruniform(), - false, - true, - PARAM.inp.cal_force); - } - } - } - - orb_r.set_orbital_info(orbs[orb_r_ntype][0][0].getLabel(), // atom label - orb_r_ntype, // atom type - 1, // angular momentum L - 1, // number of orbitals of this L , just N - orbs[orb_r_ntype][0][0].getNr(), // number of radial mesh - orbs[orb_r_ntype][0][0].getRab(), // the mesh interval in radial mesh - orbs[orb_r_ntype][0][0].getRadial(), // radial mesh value(a.u.) - Numerical_Orbital_Lm::Psi_Type::Psi, - orbs[orb_r_ntype][0][0].getRadial(), // radial wave function - orbs[orb_r_ntype][0][0].getNk(), - orbs[orb_r_ntype][0][0].getDk(), - orbs[orb_r_ntype][0][0].getDruniform(), - false, - true, - PARAM.inp.cal_force); - - for (int TA = 0; TA < orb.get_ntype(); ++TA) - { - for (int TB = 0; TB < orb.get_ntype(); ++TB) - { - for (int LA = 0; LA <= orb.Phi[TA].getLmax(); ++LA) - { - for (int NA = 0; NA < orb.Phi[TA].getNchi(LA); ++NA) - { - for (int LB = 0; LB <= orb.Phi[TB].getLmax(); ++LB) - { - for (int NB = 0; NB < orb.Phi[TB].getNchi(LB); ++NB) - { - center2_orb11[TA][TB][LA][NA][LB].insert( - std::make_pair(NB, Center2_Orb::Orb11(orbs[TA][LA][NA], orbs[TB][LB][NB], psb_, MGT))); - } - } - } - } - } - } - - for (int TA = 0; TA < orb.get_ntype(); ++TA) - { - for (int TB = 0; TB < orb.get_ntype(); ++TB) - { - for (int LA = 0; LA <= orb.Phi[TA].getLmax(); ++LA) - { - for (int NA = 0; NA < orb.Phi[TA].getNchi(LA); ++NA) - { - for (int LB = 0; LB <= orb.Phi[TB].getLmax(); ++LB) - { - for (int NB = 0; NB < orb.Phi[TB].getNchi(LB); ++NB) - { - center2_orb21_r[TA][TB][LA][NA][LB].insert( - std::make_pair(NB, Center2_Orb::Orb21(orbs[TA][LA][NA], orb_r, orbs[TB][LB][NB], psb_, MGT))); - } - } - } - } - } - } - - for (auto& co1: center2_orb11) - { - for (auto& co2: co1.second) - { - for (auto& co3: co2.second) - { - for (auto& co4: co3.second) - { - for (auto& co5: co4.second) - { - for (auto& co6: co5.second) - { - co6.second.init_radial_table(); - } - } - } - } - } - } - - for (auto& co1: center2_orb21_r) - { - for (auto& co2: co1.second) - { - for (auto& co3: co2.second) - { - for (auto& co4: co3.second) - { - for (auto& co5: co4.second) - { - for (auto& co6: co5.second) - { - co6.second.init_radial_table(); - } - } - } - } - } - } - - int map_size = PARAM.globalv.nlocal; - int required_orbitals = 0; - for (int it = 0; it < ucell.ntype; ++it) - { - required_orbitals += ucell.atoms[it].nw * ucell.atoms[it].na; - } - map_size = std::max(map_size, required_orbitals); - - iw2it.resize(map_size); - iw2ia.resize(map_size); - iw2iL.resize(map_size); - iw2iN.resize(map_size); - iw2im.resize(map_size); - - int iw = 0; - for (int it = 0; it < ucell.ntype; it++) - { - for (int ia = 0; ia < ucell.atoms[it].na; ia++) - { - for (int iL = 0; iL < ucell.atoms[it].nwl + 1; iL++) - { - for (int iN = 0; iN < ucell.atoms[it].l_nchi[iL]; iN++) - { - for (int im = 0; im < (2 * iL + 1); im++) - { - iw2it[iw] = it; - iw2ia[iw] = ia; - iw2iL[iw] = iL; - iw2iN[iw] = iN; - iw2im[iw] = im; - iw++; - } - } - } - } - } -} - -void cal_r_overlap_R::construct_orbs_and_nonlocal_and_orb_r(const UnitCell& ucell, const LCAO_Orbitals& orb) -{ - const NonlocalInfoBase& infoNL_ = *ucell.infoNL; - - int orb_r_ntype = 0; - int mat_Nr = orb.Phi[0].PhiLN(0, 0).getNr(); - int count_Nr = 0; - - orbs.resize(orb.get_ntype()); - for (int T = 0; T < orb.get_ntype(); ++T) - { - count_Nr = orb.Phi[T].PhiLN(0, 0).getNr(); - if (count_Nr > mat_Nr) - { - mat_Nr = count_Nr; - orb_r_ntype = T; - } - - orbs[T].resize(orb.Phi[T].getLmax() + 1); - for (int L = 0; L <= orb.Phi[T].getLmax(); ++L) - { - orbs[T][L].resize(orb.Phi[T].getNchi(L)); - for (int N = 0; N < orb.Phi[T].getNchi(L); ++N) - { - const auto& orb_origin = orb.Phi[T].PhiLN(L, N); - orbs[T][L][N].set_orbital_info(orb_origin.getLabel(), - orb_origin.getType(), - orb_origin.getL(), - orb_origin.getChi(), - orb_origin.getNr(), - orb_origin.getRab(), - orb_origin.getRadial(), - Numerical_Orbital_Lm::Psi_Type::Psi, - orb_origin.getPsi(), - static_cast(orb_origin.getNk() * kmesh_times) | 1, - orb_origin.getDk(), - orb_origin.getDruniform(), - false, - true, - PARAM.inp.cal_force); - } - } - } - - orb_r.set_orbital_info(orbs[orb_r_ntype][0][0].getLabel(), // atom label - orb_r_ntype, // atom type - 1, // angular momentum L - 1, // number of orbitals of this L , just N - orbs[orb_r_ntype][0][0].getNr(), // number of radial mesh - orbs[orb_r_ntype][0][0].getRab(), // the mesh interval in radial mesh - orbs[orb_r_ntype][0][0].getRadial(), // radial mesh value(a.u.) - Numerical_Orbital_Lm::Psi_Type::Psi, - orbs[orb_r_ntype][0][0].getRadial(), // radial wave function - orbs[orb_r_ntype][0][0].getNk(), - orbs[orb_r_ntype][0][0].getDk(), - orbs[orb_r_ntype][0][0].getDruniform(), - false, - true, - PARAM.inp.cal_force); - - orbs_nonlocal.resize(orb.get_ntype()); - for (int T = 0; T < orb.get_ntype(); ++T) - { - const int nproj = infoNL_.get_nproj(T); - orbs_nonlocal[T].resize(nproj); - for (int ip = 0; ip < nproj; ip++) - { - int nr = infoNL_.get_proj_Nr(T, ip); - double dr_uniform = 0.01; - int nr_uniform - = static_cast((infoNL_.get_proj_radial(T, ip)[nr - 1] - infoNL_.get_proj_radial(T, ip)[0]) / dr_uniform) + 1; - double* rad = new double[nr_uniform]; - double* rab = new double[nr_uniform]; - for (int ir = 0; ir < nr_uniform; ir++) - { - rad[ir] = ir * dr_uniform; - rab[ir] = dr_uniform; - } - double* y2 = new double[nr]; - double* Beta_r_uniform = new double[nr_uniform]; - double* dbeta_uniform = new double[nr_uniform]; - ModuleBase::Mathzone_Add1::SplineD2(infoNL_.get_proj_radial(T, ip), - infoNL_.get_proj_beta_r(T, ip), - nr, - 0.0, - 0.0, - y2); - ModuleBase::Mathzone_Add1::Cubic_Spline_Interpolation(infoNL_.get_proj_radial(T, ip), - infoNL_.get_proj_beta_r(T, ip), - y2, - nr, - rad, - nr_uniform, - Beta_r_uniform, - dbeta_uniform); - - // linear extrapolation at the zero point - if (infoNL_.get_proj_radial(T, ip)[0] > 1e-10) - { - double slope = (infoNL_.get_proj_beta_r(T, ip)[1] - infoNL_.get_proj_beta_r(T, ip)[0]) - / (infoNL_.get_proj_radial(T, ip)[1] - infoNL_.get_proj_radial(T, ip)[0]); - Beta_r_uniform[0] = infoNL_.get_proj_beta_r(T, ip)[0] - slope * infoNL_.get_proj_radial(T, ip)[0]; - } - - // Here, the operation beta_r / r is performed. To avoid divergence at r=0, beta_r(0) is set to beta_r(1). - // However, this may introduce issues, so caution is needed. - for (int ir = 1; ir < nr_uniform; ir++) - { - Beta_r_uniform[ir] = Beta_r_uniform[ir] / rad[ir]; - } - Beta_r_uniform[0] = Beta_r_uniform[1]; - - orbs_nonlocal[T][ip].set_orbital_info(infoNL_.get_label(T), - infoNL_.get_type(T), - infoNL_.get_proj_L(T, ip), - 1, - nr_uniform, - rab, - rad, - Numerical_Orbital_Lm::Psi_Type::Psi, - Beta_r_uniform, - static_cast(infoNL_.get_proj_Nk(T, ip) * kmesh_times) | 1, - infoNL_.get_proj_dk(T, ip), - infoNL_.get_proj_dr_uniform(T, ip), - false, - true, - PARAM.inp.cal_force); - - delete[] rad; - delete[] rab; - delete[] y2; - delete[] Beta_r_uniform; - delete[] dbeta_uniform; - } - } - - for (int TA = 0; TA < orb.get_ntype(); ++TA) - { - for (int TB = 0; TB < orb.get_ntype(); ++TB) - { - for (int LA = 0; LA <= orb.Phi[TA].getLmax(); ++LA) - { - for (int NA = 0; NA < orb.Phi[TA].getNchi(LA); ++NA) - { - for (int ip = 0; ip < infoNL_.get_nproj(TB); ip++) - { - center2_orb11_nonlocal[TA][TB][LA][NA].insert( - std::make_pair(ip, Center2_Orb::Orb11(orbs[TA][LA][NA], orbs_nonlocal[TB][ip], psb_, MGT))); - } - } - } - } - } - - for (int TA = 0; TA < orb.get_ntype(); ++TA) - { - for (int TB = 0; TB < orb.get_ntype(); ++TB) - { - for (int LA = 0; LA <= orb.Phi[TA].getLmax(); ++LA) - { - for (int NA = 0; NA < orb.Phi[TA].getNchi(LA); ++NA) - { - for (int ip = 0; ip < infoNL_.get_nproj(TB); ip++) - { - center2_orb21_r_nonlocal[TA][TB][LA][NA].insert( - std::make_pair(ip, Center2_Orb::Orb21(orbs[TA][LA][NA], orb_r, orbs_nonlocal[TB][ip], psb_, MGT))); - } - } - } - } - } - - for (auto& co1: center2_orb11_nonlocal) - { - for (auto& co2: co1.second) - { - for (auto& co3: co2.second) - { - for (auto& co4: co3.second) - { - for (auto& co5: co4.second) - { - co5.second.init_radial_table(); - } - } - } - } - } - - for (auto& co1: center2_orb21_r_nonlocal) - { - for (auto& co2: co1.second) - { - for (auto& co3: co2.second) - { - for (auto& co4: co3.second) - { - for (auto& co5: co4.second) - { - co5.second.init_radial_table(); - } - } - } - } - } - - int map_size = PARAM.globalv.nlocal; - int required_orbitals = 0; - for (int it = 0; it < ucell.ntype; ++it) - { - required_orbitals += ucell.atoms[it].nw * ucell.atoms[it].na; - } - map_size = std::max(map_size, required_orbitals); - - iw2it.resize(map_size); - iw2ia.resize(map_size); - iw2iL.resize(map_size); - iw2iN.resize(map_size); - iw2im.resize(map_size); - - int iw = 0; - for (int it = 0; it < ucell.ntype; it++) - { - for (int ia = 0; ia < ucell.atoms[it].na; ia++) - { - for (int iL = 0; iL < ucell.atoms[it].nwl + 1; iL++) - { - for (int iN = 0; iN < ucell.atoms[it].l_nchi[iL]; iN++) - { - for (int im = 0; im < (2 * iL + 1); im++) - { - iw2it[iw] = it; - iw2ia[iw] = ia; - iw2iL[iw] = iL; - iw2iN[iw] = iN; - iw2im[iw] = im; - iw++; - } - } - } - } - } -} - -void cal_r_overlap_R::init(const UnitCell& ucell, const Parallel_Orbitals& pv, const LCAO_Orbitals& orb) -{ - ModuleBase::TITLE("cal_r_overlap_R", "init"); - ModuleBase::timer::start("cal_r_overlap_R", "init"); - this->ParaV = &pv; - - initialize_orb_table(ucell, orb); - construct_orbs_and_orb_r(ucell, orb); - - ModuleBase::timer::end("cal_r_overlap_R", "init"); - return; -} - -void cal_r_overlap_R::init_nonlocal(const UnitCell& ucell, const Parallel_Orbitals& pv, const LCAO_Orbitals& orb) -{ - ModuleBase::TITLE("cal_r_overlap_R", "init_nonlocal"); - ModuleBase::timer::start("cal_r_overlap_R", "init_nonlocal"); - this->ParaV = &pv; - - initialize_orb_table(ucell, orb); - construct_orbs_and_nonlocal_and_orb_r(ucell, orb); - - ModuleBase::timer::end("cal_r_overlap_R", "init_nonlocal"); - return; -} - -ModuleBase::Vector3 cal_r_overlap_R::get_psi_r_psi(const ModuleBase::Vector3& R1, - const int& T1, - const int& L1, - const int& m1, - const int& N1, - const ModuleBase::Vector3& R2, - const int& T2, - const int& L2, - const int& m2, - const int& N2) -{ - ModuleBase::Vector3 origin_point(0.0, 0.0, 0.0); - double factor = sqrt(ModuleBase::FOUR_PI / 3.0); - const ModuleBase::Vector3& distance = R2 - R1; - - double overlap_o = center2_orb11[T1][T2][L1][N1][L2].at(N2).cal_overlap(origin_point, distance, m1, m2); - - double overlap_x = -1 * factor * center2_orb21_r[T1][T2][L1][N1][L2].at(N2).cal_overlap(origin_point, distance, m1, 1, - m2); // m = 1 - - double overlap_y = -1 * factor * center2_orb21_r[T1][T2][L1][N1][L2].at(N2).cal_overlap(origin_point, distance, m1, 2, - m2); // m = -1 - - double overlap_z = factor * center2_orb21_r[T1][T2][L1][N1][L2].at(N2).cal_overlap(origin_point, distance, m1, 0, - m2); // m = 0 - - ModuleBase::Vector3 temp_prp = ModuleBase::Vector3(overlap_x, overlap_y, overlap_z) + R1 * overlap_o; - - return temp_prp; -} -ModuleBase::Vector3 cal_r_overlap_R::get_psi_r_gradpsi(const ModuleBase::Vector3& R1, - const int& T1, - const int& L1, - const int& m1, - const int& N1, - const ModuleBase::Vector3& R2, - const int& T2, - const int& L2, - const int& m2, - const int& N2, - const ModuleBase::Vector3& Efield, - const ModuleBase::Vector3& dR) -{ - ModuleBase::Vector3 origin_point(0.0, 0.0, 0.0); - double factor = sqrt(ModuleBase::FOUR_PI / 3.0); - const ModuleBase::Vector3& distance = R2 - R1; - - ModuleBase::Vector3 grad_o = center2_orb11[T1][T2][L1][N1][L2].at(N2).cal_grad_overlap(origin_point, distance, m1, m2); - - ModuleBase::Vector3 grad_rx = -1 * factor - * center2_orb21_r[T1][T2][L1][N1][L2].at(N2).cal_grad_overlap(origin_point, - distance, - m1, - 1, - m2); // m = 1 - - ModuleBase::Vector3 grad_ry = -1 * factor - * center2_orb21_r[T1][T2][L1][N1][L2].at(N2).cal_grad_overlap(origin_point, - distance, - m1, - 2, - m2); // m = -1 - - ModuleBase::Vector3 grad_rz = factor - * center2_orb21_r[T1][T2][L1][N1][L2].at(N2).cal_grad_overlap(origin_point, - distance, - m1, - 0, - m2); // m = 0 - - ModuleBase::Vector3 temp_prp = Efield[0] * grad_rx + Efield[1] * grad_ry + Efield[2] * grad_rz + (Efield * (R1 - dR)) * grad_o; - - return temp_prp; -} -void cal_r_overlap_R::get_psi_r_beta(const UnitCell& ucell, - std::vector>& nlm, - const ModuleBase::Vector3& R1, - const int& T1, - const int& L1, - const int& m1, - const int& N1, - const ModuleBase::Vector3& R2, - const int& T2) -{ - ModuleBase::Vector3 origin_point(0.0, 0.0, 0.0); - double factor = sqrt(ModuleBase::FOUR_PI / 3.0); - const ModuleBase::Vector3& distance = R2 - R1; - const NonlocalInfoBase& infoNL_ = *ucell.infoNL; - const int nproj = infoNL_.get_nproj(T2); - nlm.resize(4); - if (nproj == 0) - { - for (int i = 0; i < 4; i++) - { - nlm[i].resize(1); - } - return; - } - - int natomwfc = 0; - for (int ip = 0; ip < nproj; ip++) - { - const int L2 = infoNL_.get_proj_L(T2, ip); // mohan add 2021-05-07 - natomwfc += 2 * L2 + 1; - } - for (int i = 0; i < 4; i++) - { - nlm[i].resize(natomwfc); - } - int index = 0; - for (int ip = 0; ip < nproj; ip++) - { - const int L2 = infoNL_.get_proj_L(T2, ip); - for (int m2 = 0; m2 < 2 * L2 + 1; m2++) - { - double overlap_o = center2_orb11_nonlocal[T1][T2][L1][N1].at(ip).cal_overlap(origin_point, distance, m1, m2); - - double overlap_x = -1 * factor - * center2_orb21_r_nonlocal[T1][T2][L1][N1].at(ip).cal_overlap(origin_point, - distance, - m1, - 1, - m2); // m = 1 - - double overlap_y = -1 * factor - * center2_orb21_r_nonlocal[T1][T2][L1][N1].at(ip).cal_overlap(origin_point, - distance, - m1, - 2, - m2); // m = -1 - - double overlap_z = factor - * center2_orb21_r_nonlocal[T1][T2][L1][N1].at(ip).cal_overlap(origin_point, - distance, - m1, - 0, - m2); // m = 0 - - // nlm[index] = ModuleBase::Vector3(overlap_x, overlap_y, overlap_z) + R1 * overlap_o; - - // nlm[index] = ModuleBase::Vector3(overlap_o, overlap_y, overlap_z);// + R1 * overlap_o; - nlm[0][index] = overlap_o; - nlm[1][index] = overlap_x + (R1 * overlap_o).x; - nlm[2][index] = overlap_y + (R1 * overlap_o).y; - nlm[3][index] = overlap_z + (R1 * overlap_o).z; - index++; - } - } -} - -void cal_r_overlap_R::out_rR(const UnitCell& ucell, const Grid_Driver& gd, const int& istep, const int precision) -{ - ModuleBase::TITLE("cal_r_overlap_R", "out_rR"); - ModuleBase::timer::start("cal_r_overlap_R", "out_rR"); - - int step = istep; - // set R coor range - int R_minX = int(-gd.getGlayerX_minus()); - int R_minY = int(-gd.getGlayerY_minus()); - int R_minZ = int(-gd.getGlayerZ_minus()); - - int R_x = gd.getGlayerX() + gd.getGlayerX_minus(); - int R_y = gd.getGlayerY() + gd.getGlayerY_minus(); - int R_z = gd.getGlayerZ() + gd.getGlayerZ_minus(); - - std::set> all_R_coor; - for (int ix = 0; ix < R_x; ix++) - { - for (int iy = 0; iy < R_y; iy++) - { - for (int iz = 0; iz < R_z; iz++) - { - Abfs::Vector3_Order temp_R(ix + R_minX, iy + R_minY, iz + R_minZ); - all_R_coor.insert(temp_R); - } - } - } - - // calculate rR matrix - ModuleBase::Vector3 tau1, tau2, dtau; - ModuleBase::Vector3 origin_point(0.0, 0.0, 0.0); - double factor = sqrt(ModuleBase::FOUR_PI / 3.0); - int output_R_number = 0; - ModuleIO::SparseWriteOptions single_R_options; - single_R_options.threshold = sparse_threshold; - single_R_options.binary = binary; - single_R_options.precision = precision; - single_R_options.reduce = true; - single_R_options.temp_dir = PARAM.globalv.global_out_dir; - - std::stringstream tem1; - tem1 << PARAM.globalv.global_out_dir << "tmp-rr.csr"; - std::ofstream ofs_tem1; - - if (GlobalV::DRANK == 0) - { - if (binary) - { - ofs_tem1.open(tem1.str().c_str(), std::ios::binary); - } - else - { - ofs_tem1.open(tem1.str().c_str()); - } - if (!ofs_tem1.is_open()) - { - ModuleBase::WARNING_QUIT("cal_r_overlap_R::out_rR", "Cannot open temporary sparse matrix file: " + tem1.str()); - } - } - - for (auto& R_coor: all_R_coor) - { - std::map> psi_r_psi_sparse[3]; - - int dRx = R_coor.x; - int dRy = R_coor.y; - int dRz = R_coor.z; - - ModuleBase::Vector3 R_car = ModuleBase::Vector3(dRx, dRy, dRz) * ucell.latvec; - - int ir, ic; - for (int iw1 = 0; iw1 < PARAM.globalv.nlocal; iw1++) - { - ir = this->ParaV->global2local_row(iw1); - if (ir >= 0) - { - for (int iw2 = 0; iw2 < PARAM.globalv.nlocal; iw2++) - { - ic = this->ParaV->global2local_col(iw2); - if (ic >= 0) - { - int orb_index_row = iw1 / PARAM.globalv.npol; - int orb_index_col = iw2 / PARAM.globalv.npol; - - // The off-diagonal term in SOC calculaiton is zero, and the two diagonal terms are the same - int new_index - = iw1 - PARAM.globalv.npol * orb_index_row + (iw2 - PARAM.globalv.npol * orb_index_col) * PARAM.globalv.npol; - - if (new_index == 0 || new_index == 3) - { - int it1 = iw2it[orb_index_row]; - int ia1 = iw2ia[orb_index_row]; - int iN1 = iw2iN[orb_index_row]; - int iL1 = iw2iL[orb_index_row]; - int im1 = iw2im[orb_index_row]; - - int it2 = iw2it[orb_index_col]; - int ia2 = iw2ia[orb_index_col]; - int iN2 = iw2iN[orb_index_col]; - int iL2 = iw2iL[orb_index_col]; - int im2 = iw2im[orb_index_col]; - - ModuleBase::Vector3 r_distance - = (ucell.atoms[it2].tau[ia2] - ucell.atoms[it1].tau[ia1] + R_car) * ucell.lat0; - - double overlap_o - = center2_orb11[it1][it2][iL1][iN1][iL2].at(iN2).cal_overlap(origin_point, r_distance, im1, im2); - - double overlap_x = -1 * factor - * center2_orb21_r[it1][it2][iL1][iN1][iL2].at(iN2).cal_overlap(origin_point, - r_distance, - im1, - 1, - im2); // m = 1 - - double overlap_y = -1 * factor - * center2_orb21_r[it1][it2][iL1][iN1][iL2].at(iN2).cal_overlap(origin_point, - r_distance, - im1, - 2, - im2); // m = -1 - - double overlap_z = factor - * center2_orb21_r[it1][it2][iL1][iN1][iL2].at(iN2).cal_overlap(origin_point, - r_distance, - im1, - 0, - im2); // m = 0 - - ModuleBase::Vector3 temp_prp = ModuleBase::Vector3(overlap_x, overlap_y, overlap_z) - + ucell.atoms[it1].tau[ia1] * ucell.lat0 * overlap_o; - - if (std::abs(temp_prp.x) > sparse_threshold) - { - psi_r_psi_sparse[0][iw1][iw2] = temp_prp.x; - } - - if (std::abs(temp_prp.y) > sparse_threshold) - { - psi_r_psi_sparse[1][iw1][iw2] = temp_prp.y; - } - - if (std::abs(temp_prp.z) > sparse_threshold) - { - psi_r_psi_sparse[2][iw1][iw2] = temp_prp.z; - } - } - } - } - } - } - - int rR_nonzero_num[3] = {0, 0, 0}; - for (int direction = 0; direction < 3; ++direction) - { - for (auto& row_loop: psi_r_psi_sparse[direction]) - { - rR_nonzero_num[direction] += row_loop.second.size(); - } - } - - Parallel_Reduce::reduce_all(rR_nonzero_num, 3); - - if (ModuleIO::detail::rr_sparse_has_payload(rR_nonzero_num)) - { - output_R_number++; - - if (GlobalV::DRANK == 0) - { - if (binary) - { - ofs_tem1.write(reinterpret_cast(&dRx), sizeof(int)); - ofs_tem1.write(reinterpret_cast(&dRy), sizeof(int)); - ofs_tem1.write(reinterpret_cast(&dRz), sizeof(int)); - } - else - { - ofs_tem1 << dRx << " " << dRy << " " << dRz << std::endl; - } - } - - for (int direction = 0; direction < 3; ++direction) - { - if (GlobalV::DRANK == 0) - { - if (binary) - { - ofs_tem1.write(reinterpret_cast(&rR_nonzero_num[direction]), sizeof(int)); - } - else - { - ofs_tem1 << rR_nonzero_num[direction] << std::endl; - } - } - - if (rR_nonzero_num[direction]) - { - ModuleIO::output_single_R(ofs_tem1, psi_r_psi_sparse[direction], *(this->ParaV), single_R_options); - } - else - { - // do nothing - } - } - } - } - - if (GlobalV::DRANK == 0) - { - std::stringstream ssr; - if (PARAM.inp.calculation == "md" && !PARAM.inp.out_app_flag) - { - ssr << PARAM.globalv.global_matrix_dir << "rrg" << step << ".csr"; - } - else - { - ssr << PARAM.globalv.global_out_dir << "rr.csr"; - } - - ofs_tem1.close(); - ModuleIO::detail::finalize_rr_sparse_file(ssr.str(), - tem1.str(), - step, - PARAM.globalv.nlocal, - output_R_number, - binary, - PARAM.inp.calculation == "md" && PARAM.inp.out_app_flag && step, - "cal_r_overlap_R::out_rR"); - - std::remove(tem1.str().c_str()); - } - - ModuleBase::timer::end("cal_r_overlap_R", "out_rR"); - return; -} - -void cal_r_overlap_R::out_rR_other(const UnitCell& ucell, - const int& istep, - const std::set>& output_R_coor, - const int precision) -{ - ModuleBase::TITLE("cal_r_overlap_R", "out_rR_other"); - ModuleBase::timer::start("cal_r_overlap_R", "out_rR_other"); - - // calculate rR matrix - ModuleBase::Vector3 tau1, tau2, dtau; - ModuleBase::Vector3 origin_point(0.0, 0.0, 0.0); - double factor = sqrt(ModuleBase::FOUR_PI / 3.0); - int output_R_number = 0; - int step = istep; - ModuleIO::SparseWriteOptions single_R_options; - single_R_options.threshold = sparse_threshold; - single_R_options.binary = binary; - single_R_options.precision = precision; - single_R_options.reduce = true; - single_R_options.temp_dir = PARAM.globalv.global_out_dir; - - std::stringstream tem1; - tem1 << PARAM.globalv.global_out_dir << "tmp-rr-other.csr"; - std::ofstream ofs_tem1; - if (GlobalV::DRANK == 0) - { - if (binary) - { - ofs_tem1.open(tem1.str().c_str(), std::ios::binary); - } - else - { - ofs_tem1.open(tem1.str().c_str()); - } - if (!ofs_tem1.is_open()) - { - ModuleBase::WARNING_QUIT("cal_r_overlap_R::out_rR_other", "Cannot open temporary sparse matrix file: " + tem1.str()); - } - } - - std::stringstream ssr; - if (PARAM.inp.calculation == "md" && !PARAM.inp.out_app_flag) - { - ssr << PARAM.globalv.global_matrix_dir << "rrg" << step << ".csr"; - } - else - { - ssr << PARAM.globalv.global_out_dir << "rr.csr"; - } - - for (auto& R_coor: output_R_coor) - { - std::map> psi_r_psi_sparse[3]; - - int dRx = R_coor.x; - int dRy = R_coor.y; - int dRz = R_coor.z; - - ModuleBase::Vector3 R_car = ModuleBase::Vector3(dRx, dRy, dRz) * ucell.latvec; - - int ir = 0; - int ic = 0; - for (int iw1 = 0; iw1 < PARAM.globalv.nlocal; iw1++) - { - ir = this->ParaV->global2local_row(iw1); - if (ir >= 0) - { - for (int iw2 = 0; iw2 < PARAM.globalv.nlocal; iw2++) - { - ic = this->ParaV->global2local_col(iw2); - if (ic >= 0) - { - int orb_index_row = iw1 / PARAM.globalv.npol; - int orb_index_col = iw2 / PARAM.globalv.npol; - - // The off-diagonal term in SOC calculaiton is zero, and the two diagonal terms are the same - int new_index - = iw1 - PARAM.globalv.npol * orb_index_row + (iw2 - PARAM.globalv.npol * orb_index_col) * PARAM.globalv.npol; - - if (new_index == 0 || new_index == 3) - { - int it1 = iw2it[orb_index_row]; - int ia1 = iw2ia[orb_index_row]; - int iN1 = iw2iN[orb_index_row]; - int iL1 = iw2iL[orb_index_row]; - int im1 = iw2im[orb_index_row]; - - int it2 = iw2it[orb_index_col]; - int ia2 = iw2ia[orb_index_col]; - int iN2 = iw2iN[orb_index_col]; - int iL2 = iw2iL[orb_index_col]; - int im2 = iw2im[orb_index_col]; - - ModuleBase::Vector3 r_distance - = (ucell.atoms[it2].tau[ia2] - ucell.atoms[it1].tau[ia1] + R_car) * ucell.lat0; - - double overlap_o - = center2_orb11[it1][it2][iL1][iN1][iL2].at(iN2).cal_overlap(origin_point, r_distance, im1, im2); - - double overlap_x = -1 * factor - * center2_orb21_r[it1][it2][iL1][iN1][iL2].at(iN2).cal_overlap(origin_point, - r_distance, - im1, - 1, - im2); // m = 1 - - double overlap_y = -1 * factor - * center2_orb21_r[it1][it2][iL1][iN1][iL2].at(iN2).cal_overlap(origin_point, - r_distance, - im1, - 2, - im2); // m = -1 - - double overlap_z = factor - * center2_orb21_r[it1][it2][iL1][iN1][iL2].at(iN2).cal_overlap(origin_point, - r_distance, - im1, - 0, - im2); // m = 0 - - ModuleBase::Vector3 temp_prp = ModuleBase::Vector3(overlap_x, overlap_y, overlap_z) - + ucell.atoms[it1].tau[ia1] * ucell.lat0 * overlap_o; - - if (std::abs(temp_prp.x) > sparse_threshold) - { - psi_r_psi_sparse[0][iw1][iw2] = temp_prp.x; - } - - if (std::abs(temp_prp.y) > sparse_threshold) - { - psi_r_psi_sparse[1][iw1][iw2] = temp_prp.y; - } - - if (std::abs(temp_prp.z) > sparse_threshold) - { - psi_r_psi_sparse[2][iw1][iw2] = temp_prp.z; - } - } - } - } - } - } - - int rR_nonzero_num[3] = {0, 0, 0}; - for (int direction = 0; direction < 3; ++direction) - { - for (auto& row_loop: psi_r_psi_sparse[direction]) - { - rR_nonzero_num[direction] += row_loop.second.size(); - } - } - - Parallel_Reduce::reduce_all(rR_nonzero_num, 3); - - if (!ModuleIO::detail::rr_sparse_has_payload(rR_nonzero_num)) - { - continue; - } - output_R_number++; - - if (GlobalV::DRANK == 0) - { - if (binary) // .dat - { - ofs_tem1.write(reinterpret_cast(&dRx), sizeof(int)); - ofs_tem1.write(reinterpret_cast(&dRy), sizeof(int)); - ofs_tem1.write(reinterpret_cast(&dRz), sizeof(int)); - } - else // .txt - { - ofs_tem1 << dRx << " " << dRy << " " << dRz << std::endl; - } - } - - for (int direction = 0; direction < 3; ++direction) - { - if (GlobalV::DRANK == 0) - { - if (binary) - { - ofs_tem1.write(reinterpret_cast(&rR_nonzero_num[direction]), sizeof(int)); - } - else - { - ofs_tem1 << rR_nonzero_num[direction] << std::endl; - } - } - - if (rR_nonzero_num[direction]) - { - ModuleIO::output_single_R(ofs_tem1, psi_r_psi_sparse[direction], *(this->ParaV), single_R_options); - } - else - { - // do nothing - } - } - } - - if (GlobalV::DRANK == 0) - { - ofs_tem1.close(); - ModuleIO::detail::finalize_rr_sparse_file(ssr.str(), - tem1.str(), - step, - PARAM.globalv.nlocal, - output_R_number, - binary, - PARAM.inp.calculation == "md" && PARAM.inp.out_app_flag && step, - "cal_r_overlap_R::out_rR_other"); - std::remove(tem1.str().c_str()); - } - - ModuleBase::timer::end("cal_r_overlap_R", "out_rR_other"); - return; -} diff --git a/source/source_io/module_hs/cal_r_overlap_r.h b/source/source_io/module_hs/cal_r_overlap_r.h deleted file mode 100644 index beca72a61fb..00000000000 --- a/source/source_io/module_hs/cal_r_overlap_r.h +++ /dev/null @@ -1,103 +0,0 @@ -#ifndef CAL_R_OVERLAP_R_H -#define CAL_R_OVERLAP_R_H - -#include "source_base/sph_bessel_recursive.h" -#include "source_base/vector3.h" -#include "source_base/ylm.h" -#include "source_basis/module_ao/orb_atomic_lm.h" -#include "source_basis/module_ao/orb_gaunt_table.h" -#include "source_basis/module_ao/orb_read.h" -#include "source_basis/module_ao/parallel_orbitals.h" -#include "source_cell/module_neighbor/sltk_grid_driver.h" -#include "source_cell/unitcell.h" -#include "source_lcao/center2orb_orb11.h" -#include "source_lcao/center2orb_orb21.h" -#include "source_lcao/center2orb.h" -#include "source_lcao/module_ri/abfs_vector3_order.h" - -#include -#include -#include - -// output r_R matrix, added by Jingan -class cal_r_overlap_R -{ - - public: - cal_r_overlap_R(); - ~cal_r_overlap_R(); - - double kmesh_times = 4; - double sparse_threshold = 1e-10; - bool binary = false; - - void init(const UnitCell& ucell, const Parallel_Orbitals& pv, const LCAO_Orbitals& orb); - void init_nonlocal(const UnitCell& ucell, const Parallel_Orbitals& pv, const LCAO_Orbitals& orb); - ModuleBase::Vector3 get_psi_r_psi(const ModuleBase::Vector3& R1, - const int& T1, - const int& L1, - const int& m1, - const int& N1, - const ModuleBase::Vector3& R2, - const int& T2, - const int& L2, - const int& m2, - const int& N2); - ModuleBase::Vector3 get_psi_r_gradpsi(const ModuleBase::Vector3& R1, - const int& T1, - const int& L1, - const int& m1, - const int& N1, - const ModuleBase::Vector3& R2, - const int& T2, - const int& L2, - const int& m2, - const int& N2, - const ModuleBase::Vector3& Efield, - const ModuleBase::Vector3& dR); - void get_psi_r_beta(const UnitCell& ucell, - std::vector>& nlm, - const ModuleBase::Vector3& R1, - const int& T1, - const int& L1, - const int& m1, - const int& N1, - const ModuleBase::Vector3& R2, - const int& T2); - void out_rR(const UnitCell& ucell, const Grid_Driver& gd, const int& istep, const int precision = 16); - void out_rR_other(const UnitCell& ucell, - const int& istep, - const std::set>& output_R_coor, - const int precision = 16); - - private: - void initialize_orb_table(const UnitCell& ucell, const LCAO_Orbitals& orb); - void construct_orbs_and_orb_r(const UnitCell& ucell, const LCAO_Orbitals& orb); - void construct_orbs_and_nonlocal_and_orb_r(const UnitCell& ucell, const LCAO_Orbitals& orb); - - std::vector iw2ia; - std::vector iw2iL; - std::vector iw2im; - std::vector iw2iN; - std::vector iw2it; - - ModuleBase::Sph_Bessel_Recursive::D2* psb_ = nullptr; - ORB_gaunt_table MGT; - - Numerical_Orbital_Lm orb_r; - std::vector>> orbs; - std::vector> orbs_nonlocal; - - std::map>>>>> - center2_orb11; - - std::map>>>>> - center2_orb21_r; - - std::map>>>> center2_orb11_nonlocal; - - std::map>>>> center2_orb21_r_nonlocal; - - const Parallel_Orbitals* ParaV = nullptr; -}; -#endif diff --git a/source/source_io/module_hs/dhs_sparse_writer.cpp b/source/source_io/module_hs/dhs_sparse_writer.cpp new file mode 100644 index 00000000000..9eca747934d --- /dev/null +++ b/source/source_io/module_hs/dhs_sparse_writer.cpp @@ -0,0 +1,235 @@ +#include "dhs_sparse_writer.h" + +#include "hs_sparse_io.h" +#include "hs_sparse_io_detail.h" +#include "lat_r_csr.h" +#include "source_base/global_function.h" +#include "source_base/parallel_reduce.h" +#include "source_base/timer.h" + +#include +#include +#include +#include +#include + +void ModuleIO::save_dH_sparse(const int& istep, + const Parallel_Orbitals& pv, + LCAO_HS_Arrays& HS_Arrays, + const double& sparse_thr, + const bool& binary, + const std::string& fileflag, + const int precision, + const std::string& global_out_dir, + const std::string& global_matrix_dir, + const std::string& calculation, + const bool out_app_flag, + const int nspin, + const int nlocal) { + ModuleBase::TITLE("ModuleIO", "save_dH_sparse"); + ModuleBase::timer::start("ModuleIO", "save_dH_sparse"); + SparseWriteOptions lat_r_options; + lat_r_options.threshold = sparse_thr; + lat_r_options.binary = binary; + lat_r_options.precision = precision; + lat_r_options.reduce = true; + lat_r_options.calculation = calculation; + lat_r_options.out_app_flag = out_app_flag; + + auto& all_R_coor_ptr = HS_Arrays.all_R_coor; + auto& output_R_coor_ptr = HS_Arrays.output_R_coor; + + // The three Cartesian derivative components (x, y, z) share identical + // file handling; only the label and the sparse matrices differ. + class Component + { + public: + char axis; // 'x' / 'y' / 'z' + SparseRMatrix* sparse; // nspin != 4 (array of 2) + SparseRMatrix>* soc_sparse; // nspin == 4 + std::vector nonzero_num[2]; + std::stringstream fname[2]; + std::ofstream ofs[2]; + + // Default-initialize streams to support explicit default constructors in older standard libraries. + Component(const char& axis_in, + SparseRMatrix* sparse_in, + SparseRMatrix>* soc_sparse_in) + : axis(axis_in), sparse(sparse_in), soc_sparse(soc_sparse_in) + { + } + }; + Component comps[3] = {{'x', HS_Arrays.dHRx_sparse, &HS_Arrays.dHRx_soc_sparse}, + {'y', HS_Arrays.dHRy_sparse, &HS_Arrays.dHRy_soc_sparse}, + {'z', HS_Arrays.dHRz_sparse, &HS_Arrays.dHRz_soc_sparse}}; + + const int total_R_num = static_cast(all_R_coor_ptr.size()); + int output_R_number = 0; + // Use istep for filename/append decisions; clamp only the header STEP value. + const int header_step = std::max(istep, 0); + + int spin_loop = 1; + if (nspin == 2) { + spin_loop = 2; + } + + for (auto& comp: comps) + { + if (nspin != 4) + { + for (int ispin = 0; ispin < spin_loop; ++ispin) + { + comp.nonzero_num[ispin] = detail::count_nonzeros_by_R(comp.sparse[ispin], all_R_coor_ptr, sparse_thr, true); + } + } + else + { + comp.nonzero_num[0] = detail::count_nonzeros_by_R(*comp.soc_sparse, all_R_coor_ptr, sparse_thr, true); + } + } + + const auto has_output_R = [&](const int index) { + for (const auto& comp: comps) + { + for (int ispin = 0; ispin < spin_loop; ++ispin) + { + if (comp.nonzero_num[ispin][index] != 0) + { + return true; + } + } + } + return false; + }; + + for (int index = 0; index < total_R_num; ++index) + { + if (has_output_R(index)) + { + output_R_number++; + } + } + + const bool md_no_append = (calculation == "md") && !out_app_flag; + for (auto& comp: comps) + { + for (int ispin = 0; ispin < 2; ++ispin) + { + if (istep >= 0) + { + comp.fname[ispin] << (md_no_append ? global_matrix_dir : global_out_dir) + << "d" << fileflag << "r" << comp.axis + << "s" << (ispin + 1) << "g" << (istep + 1) << "_nao.csr"; + } + else + { + comp.fname[ispin] << global_out_dir + << "d" << fileflag << "r" << comp.axis + << "s" << (ispin + 1) << "_nao.csr"; + } + } + } + + if (GlobalV::DRANK == 0) + { + const bool open_in_append = (calculation == "md") && out_app_flag && istep >= 0; + for (auto& comp: comps) + { + const std::string label = std::string("dH") + comp.axis; + for (int ispin = 0; ispin < spin_loop; ++ispin) + { + std::ios_base::openmode mode = std::ios::out; + if (binary) + { + mode |= std::ios::binary; + } + if (open_in_append) + { + mode |= std::ios::app; + } + else if (!binary) + { + GlobalV::ofs_running << " " << label << " data are in file: " + << comp.fname[ispin].str() << std::endl; + } + comp.ofs[ispin].open(comp.fname[ispin].str().c_str(), mode); + detail::check_output_file_open(comp.ofs[ispin], comp.fname[ispin].str(), "ModuleIO::save_dH_sparse"); + + if (binary) + { + comp.ofs[ispin].write(reinterpret_cast(&header_step), sizeof(int)); + comp.ofs[ispin].write(reinterpret_cast(&nlocal), sizeof(int)); + comp.ofs[ispin].write(reinterpret_cast(&output_R_number), sizeof(int)); + } + else + { + comp.ofs[ispin] << "STEP: " << header_step << std::endl; + comp.ofs[ispin] << "Matrix Dimension of " << label << "(R): " << nlocal << std::endl; + comp.ofs[ispin] << "Matrix number of " << label << "(R): " << output_R_number << std::endl; + } + } + } + } + + output_R_coor_ptr.clear(); + + int count = 0; + for (auto& R_coor: all_R_coor_ptr) { + int dRx = R_coor.x; + int dRy = R_coor.y; + int dRz = R_coor.z; + + if (!has_output_R(count)) + { + count++; + continue; + } + + output_R_coor_ptr.insert(R_coor); + + if (GlobalV::DRANK == 0) { + for (auto& comp: comps) + { + for (int ispin = 0; ispin < spin_loop; ++ispin) { + if (binary) { + const int comp_count = static_cast(comp.nonzero_num[ispin][count]); + comp.ofs[ispin].write(reinterpret_cast(&dRx), sizeof(int)); + comp.ofs[ispin].write(reinterpret_cast(&dRy), sizeof(int)); + comp.ofs[ispin].write(reinterpret_cast(&dRz), sizeof(int)); + comp.ofs[ispin].write(reinterpret_cast(&comp_count), sizeof(int)); + } else { + comp.ofs[ispin] << dRx << " " << dRy << " " << dRz << " " + << comp.nonzero_num[ispin][count] << std::endl; + } + } + } + } + + for (int ispin = 0; ispin < spin_loop; ++ispin) { + for (auto& comp: comps) + { + if (comp.nonzero_num[ispin][count] > 0) { + if (nspin != 4) { + save_lat_r(comp.ofs[ispin], comp.sparse[ispin][R_coor], pv, lat_r_options); + } else { + save_lat_r(comp.ofs[ispin], (*comp.soc_sparse)[R_coor], pv, lat_r_options); + } + } + } + } + + count++; + } + + if (GlobalV::DRANK == 0) { + for (auto& comp: comps) + { + for (int ispin = 0; ispin < spin_loop; ++ispin) { + comp.ofs[ispin].close(); + } + } + } + + ModuleBase::timer::end("ModuleIO", "save_dH_sparse"); + return; +} diff --git a/source/source_io/module_hs/dhs_sparse_writer.h b/source/source_io/module_hs/dhs_sparse_writer.h new file mode 100644 index 00000000000..6b84a05c884 --- /dev/null +++ b/source/source_io/module_hs/dhs_sparse_writer.h @@ -0,0 +1,27 @@ +#ifndef DHS_SPARSE_WRITER_H +#define DHS_SPARSE_WRITER_H + +#include "source_lcao/lcao_hs_arrays.h" + +#include + +class Parallel_Orbitals; + +namespace ModuleIO +{ +void save_dH_sparse(const int& istep, + const Parallel_Orbitals& pv, + LCAO_HS_Arrays& HS_Arrays, + const double& sparse_thr, + const bool& binary, + const std::string& fileflag, + const int precision, + const std::string& global_out_dir, + const std::string& global_matrix_dir, + const std::string& calculation, + const bool out_app_flag, + const int nspin, + const int nlocal); +} // namespace ModuleIO + +#endif diff --git a/source/source_io/module_hs/hs_dense_io.cpp b/source/source_io/module_hs/hs_dense_io.cpp new file mode 100644 index 00000000000..f81acb64272 --- /dev/null +++ b/source/source_io/module_hs/hs_dense_io.cpp @@ -0,0 +1,254 @@ +#include "hs_dense_io.h" + +#include "source_base/parallel_comm.h" +#include "source_base/parallel_reduce.h" +#include "source_base/timer.h" +#include "source_base/tool_quit.h" +#include "source_base/global_function.h" + +#include +#include +#include +#include +#include +#include + +namespace +{ +// Gather row i of a distributed 2D-block square matrix into a dense row. +// Under MPI, each rank fills the columns it owns locally; the caller then +// reduces across ranks so rank 0 holds the complete row. Without MPI the +// matrix is local and the whole row is read directly. +template +void gather_row(const T* mat, + const int dim, + const int i, + const bool tri, + const Parallel_2D& pv, + const std::string& ks_solver, + std::vector& line) +{ +#ifdef __MPI + std::fill(line.begin(), line.end(), T(0)); + const int ir = pv.global2local_row(i); + if (ir >= 0) + { + // data collection + for (int j = (tri ? i : 0); j < dim; ++j) + { + const int ic = pv.global2local_col(j); + if (ic >= 0) + { + int iic = 0; + if (ModuleBase::GlobalFunc::IS_COLUMN_MAJOR_KS_SOLVER(ks_solver)) + { + iic = ir + ic * pv.nrow; + } + else + { + iic = ir * pv.ncol + ic; + } + line[tri ? j - i : j] = mat[iic]; + } + } + } +#else + (void)pv; + (void)ks_solver; + for (int j = (tri ? i : 0); j < dim; ++j) + { + line[tri ? j - i : j] = mat[i * dim + j]; + } +#endif +} +} // namespace + +// output a square matrix +template +void ModuleIO::save_mat(const int istep, + const T* mat, + const int dim, + const bool bit, + const int precision, + const bool tri, + const bool app, + const std::string& filename, + const Parallel_2D& pv, + const int drank, + const std::string& ks_solver, + const bool reduce) +{ + ModuleBase::TITLE("ModuleIO", "save_mat"); + ModuleBase::timer::start("ModuleIO", "save_mat"); + + const bool gamma_only = std::is_same::value; + + // write .dat file + if (bit) + { + FILE* out_matrix = nullptr; +#ifdef __MPI + if (drank == 0) + { +#endif + const char* mode = (app && istep > 0) ? "ab" : "wb"; + out_matrix = fopen(filename.c_str(), mode); + if (out_matrix == nullptr) + { + ModuleBase::WARNING_QUIT("ModuleIO::save_mat", "Cannot open matrix file: " + filename); + } + fwrite(&dim, sizeof(int), 1, out_matrix); +#ifdef __MPI + } +#endif + + std::vector line(tri ? dim : dim); + for (int i = 0; i < dim; ++i) + { + const int line_len = tri ? dim - i : dim; + line.resize(line_len); + gather_row(mat, dim, i, tri, pv, ks_solver, line); + +#ifdef __MPI + if (reduce) + { + Parallel_Reduce::reduce_all(line.data(), line_len); + } + + if (drank == 0) + { + for (int j = (tri ? i : 0); j < dim; ++j) + { + fwrite(&line[tri ? j - i : j], sizeof(T), 1, out_matrix); + } + } + + MPI_Barrier(DIAG_WORLD); +#else + for (int j = (tri ? i : 0); j < dim; ++j) + { + fwrite(&line[tri ? j - i : j], sizeof(T), 1, out_matrix); + } +#endif + } + +#ifdef __MPI + if (drank == 0) + { + fclose(out_matrix); + } +#else + fclose(out_matrix); +#endif + } // end writing .dat file + else // write .txt file + { + std::ofstream out_matrix; + out_matrix << std::scientific << std::setprecision(precision); +#ifdef __MPI + if (drank == 0) + { + if (app && istep > 0) + { + out_matrix.open(filename.c_str(), std::ofstream::app); + } + else + { + out_matrix.open(filename.c_str()); + } + if (!out_matrix.is_open()) + { + ModuleBase::WARNING_QUIT("ModuleIO::save_mat", "Cannot open matrix file: " + filename); + } + out_matrix << "#------------------------------------------------------------------------" << std::endl; + out_matrix << "# ionic step " << istep+1 << std::endl; // istep starts from 0 + out_matrix << "# filename " << filename << std::endl; + out_matrix << "# gamma only " << gamma_only << std::endl; + out_matrix << "# rows " << dim << std::endl; + out_matrix << "# columns " << dim << std::endl; + out_matrix << "#------------------------------------------------------------------------" << std::endl; + + } + + std::vector line(tri ? dim : dim); + for (int i = 0; i < dim; i++) + { + const int line_len = tri ? dim - i : dim; + line.resize(line_len); + gather_row(mat, dim, i, tri, pv, ks_solver, line); + + if (reduce) + { + Parallel_Reduce::reduce_all(line.data(), line_len); + } + + if (drank == 0) + { + out_matrix << "Row " << i+1 << std::endl; + size_t count = 0; + for (int j = (tri ? i : 0); j < dim; j++) + { + out_matrix << " " << line[tri ? j - i : j]; + ++count; + if(count%8==0) + { + if(j!=dim-1) + { + out_matrix << std::endl; + } + } + } + out_matrix << std::endl; + } + } + + if (drank == 0) + { + out_matrix.close(); + } +#else + if (app) + { + out_matrix.open(filename.c_str(), std::ofstream::app); + } + else + { + out_matrix.open(filename.c_str()); + } + if (!out_matrix.is_open()) + { + ModuleBase::WARNING_QUIT("ModuleIO::save_mat", "Cannot open matrix file: " + filename); + } + + out_matrix << dim; + out_matrix << std::setprecision(precision); + std::vector line(dim); + for (int i = 0; i < dim; i++) + { + gather_row(mat, dim, i, tri, pv, ks_solver, line); + for (int j = (tri ? i : 0); j < dim; j++) + { + out_matrix << " " << line[tri ? j - i : j]; + } + out_matrix << std::endl; + } + out_matrix.close(); +#endif + } + ModuleBase::timer::end("ModuleIO", "save_mat"); + return; +} + +// Explicit instantiations +template void ModuleIO::save_mat(const int, const double*, const int, const bool, + const int, const bool, const bool, const std::string&, const Parallel_2D&, + const int, const std::string&, const bool); +template void ModuleIO::save_mat>(const int, const std::complex*, const int, + const bool, const int, const bool, const bool, const std::string&, const Parallel_2D&, + const int, const std::string&, const bool); +template void ModuleIO::save_mat(const int, const float*, const int, const bool, + const int, const bool, const bool, const std::string&, const Parallel_2D&, + const int, const std::string&, const bool); +template void ModuleIO::save_mat>(const int, const std::complex*, const int, + const bool, const int, const bool, const bool, const std::string&, const Parallel_2D&, + const int, const std::string&, const bool); diff --git a/source/source_io/module_hs/hs_dense_io.h b/source/source_io/module_hs/hs_dense_io.h new file mode 100644 index 00000000000..aabfb5c8b9a --- /dev/null +++ b/source/source_io/module_hs/hs_dense_io.h @@ -0,0 +1,43 @@ +#ifndef HS_DENSE_IO_H +#define HS_DENSE_IO_H + +#include + +#include "source_base/parallel_2d.h" + +// mohan note: this file holds the dense square-matrix writer +// (formerly save_mat in write_hs.h / write_hs.hpp). + +namespace ModuleIO +{ + +/// @brief save a square matrix, such as H(k) and S(k) +/// @param[in] istep : the step of the calculation +/// @param[in] mat : the local matrix +/// @param[in] dim : the dimension of the square matrix +/// @param[in] bit : true for binary, false for decimal +/// @param[in] precision : the precision of the decimal output +/// @param[in] tri : true for upper triangle, false for full matrix +/// @param[in] app : true for append, false for overwrite +/// @param[in] file_name : the name of the output file +/// @param[in] pv : the 2d-block parallelization information +/// @param[in] drank : the rank of the current process in the diagonalization world +/// @param[in] ks_solver : the name of the eigensolver (decides row/column major) +/// @param[in] reduce : whether to reduce across ranks before writing +template +void save_mat(const int istep, + const T* mat, + const int dim, + const bool bit, + const int precision, + const bool tri, + const bool app, + const std::string& file_name, + const Parallel_2D& pv, + const int drank, + const std::string& ks_solver, + const bool reduce = true); + +} // namespace ModuleIO + +#endif // HS_DENSE_IO_H diff --git a/source/source_io/module_hs/hs_sparse_io.cpp b/source/source_io/module_hs/hs_sparse_io.cpp new file mode 100644 index 00000000000..f645893164e --- /dev/null +++ b/source/source_io/module_hs/hs_sparse_io.cpp @@ -0,0 +1,163 @@ +#include "hs_sparse_io.h" +#include "hs_sparse_io_detail.h" + +#include "source_base/global_function.h" +#include "source_base/parallel_reduce.h" +#include "source_base/timer.h" +#include "lat_r_csr.h" + +#include +#include +#include +#include + +namespace +{ +int count_output_R(const std::vector& nonzero_num) +{ + int output_R_number = 0; + for (const long long count: nonzero_num) + { + if (count != 0) + { + ++output_R_number; + } + } + return output_R_number; +} + +void open_sparse_file(std::ofstream& ofs, const ModuleIO::SparseWriteOptions& options) +{ + std::ios_base::openmode mode = std::ios::out; + if (options.binary) + { + mode |= std::ios::binary; + } + const bool append_on_restart = (options.calculation == "md") && options.out_app_flag && options.istep >= 0; + if (append_on_restart) + { + mode |= std::ios::app; + } + ofs.open(options.filename.c_str(), mode); + if (!ofs.is_open()) + { + ModuleBase::WARNING_QUIT("ModuleIO::open_sparse_file", + "Cannot open sparse matrix file: " + options.filename); + } +} + +void write_sparse_header(std::ofstream& ofs, + const ModuleIO::SparseWriteOptions& options, + const int nlocal, + const int output_R_number) +{ + const int step = std::max(options.istep, 0); + if (options.binary) + { + ofs.write(reinterpret_cast(&step), sizeof(int)); + ofs.write(reinterpret_cast(&nlocal), sizeof(int)); + ofs.write(reinterpret_cast(&output_R_number), sizeof(int)); + } + else + { + ofs << "STEP: " << step << std::endl; + ofs << "Matrix Dimension of " + options.label + "(R): " << nlocal + << std::endl; + ofs << "Matrix number of " + options.label + "(R): " + << output_R_number << std::endl; + } +} + +void write_R_record(std::ofstream& ofs, + const ModuleIO::RCoordinate& R_coor, + const long long nonzero_count, + const bool binary) +{ + int dRx = R_coor.x; + int dRy = R_coor.y; + int dRz = R_coor.z; + if (binary) + { + const int count = static_cast(nonzero_count); + ofs.write(reinterpret_cast(&dRx), sizeof(int)); + ofs.write(reinterpret_cast(&dRy), sizeof(int)); + ofs.write(reinterpret_cast(&dRz), sizeof(int)); + ofs.write(reinterpret_cast(&count), sizeof(int)); + } + else + { + ofs << dRx << " " << dRy << " " << dRz << " " << nonzero_count + << std::endl; + } +} +} // namespace + +template +void ModuleIO::save_sparse( + const SparseRMatrix& smat, + const std::set& all_R_coor, + const Parallel_Orbitals& pv, + const SparseWriteOptions& options) { + ModuleBase::TITLE("ModuleIO", "save_sparse"); + ModuleBase::timer::start("ModuleIO", "save_sparse"); + const int nlocal = pv.get_global_row_size(); + if (nlocal <= 0) + { + ModuleBase::WARNING_QUIT("ModuleIO::save_sparse", + "Parallel_Orbitals global row size must be positive."); + } + + const std::vector nonzero_num + = detail::count_nonzeros_by_R(smat, all_R_coor, options.threshold, options.reduce); + const int output_R_number = count_output_R(nonzero_num); + std::ofstream ofs; + if (!options.reduce || GlobalV::DRANK == 0) + { + open_sparse_file(ofs, options); + write_sparse_header(ofs, options, nlocal, output_R_number); + } + + int count = 0; + for (const auto& R_coor: all_R_coor) + { + if (nonzero_num[count] == 0) + { + count++; + continue; + } + + if (!options.reduce || GlobalV::DRANK == 0) + { + write_R_record(ofs, R_coor, nonzero_num[count], options.binary); + } + + if (smat.count(R_coor)) + { + save_lat_r(ofs, smat.at(R_coor), pv, options); + } + else + { + SparseRBlock empty_map; + save_lat_r(ofs, empty_map, pv, options); + } + ++count; + } + if (!options.reduce || GlobalV::DRANK == 0) + { + ofs.close(); + } + + ModuleBase::timer::end("ModuleIO", "save_sparse"); +} + +template void ModuleIO::save_sparse( + const SparseRMatrix&, + const std::set&, + const Parallel_Orbitals&, + const SparseWriteOptions&); + +template void ModuleIO::save_sparse>( + const SparseRMatrix>&, + const std::set&, + const Parallel_Orbitals&, + const SparseWriteOptions&); diff --git a/source/source_io/module_hs/hs_sparse_io.h b/source/source_io/module_hs/hs_sparse_io.h new file mode 100644 index 00000000000..d8bdb089989 --- /dev/null +++ b/source/source_io/module_hs/hs_sparse_io.h @@ -0,0 +1,44 @@ +#ifndef HS_SPARSE_IO_H +#define HS_SPARSE_IO_H + +#include "source_basis/module_ao/parallel_orbitals.h" +#include "source_lcao/module_ri/abfs_vector3_order.h" + +#include +#include +#include +#include + +namespace ModuleIO +{ +using RCoordinate = Abfs::Vector3_Order; + +template +using SparseRBlock = std::map>; + +template +using SparseRMatrix = std::map>; + +struct SparseWriteOptions +{ + std::string filename; + std::string label; + double threshold = 0.0; + bool binary = false; + int precision = 16; + int istep = -1; + bool reduce = true; + // Runtime flags that decide file-open mode (append on md restart). + // Must be provided explicitly by the caller instead of reading PARAM. + std::string calculation; + bool out_app_flag = false; +}; + +template +void save_sparse(const SparseRMatrix& smat, + const std::set& all_R_coor, + const Parallel_Orbitals& pv, + const SparseWriteOptions& options); +} // namespace ModuleIO + +#endif diff --git a/source/source_io/module_hs/hs_sparse_io_detail.h b/source/source_io/module_hs/hs_sparse_io_detail.h new file mode 100644 index 00000000000..42acacc2f31 --- /dev/null +++ b/source/source_io/module_hs/hs_sparse_io_detail.h @@ -0,0 +1,83 @@ +#ifndef HS_SPARSE_IO_DETAIL_H +#define HS_SPARSE_IO_DETAIL_H + +#include "hs_sparse_io.h" +#include "source_base/parallel_reduce.h" +#include "source_base/tool_quit.h" + +#include +#include +#include +#include + +/** + * @brief Internal helpers shared by hs_sparse_io.cpp and dhs_sparse_writer.cpp. + * Declared here (instead of duplicated in each anonymous namespace) so + * the threshold/nonzero logic cannot diverge between the two writers. + */ +namespace ModuleIO +{ +namespace detail +{ +/** + * @brief Count non-zero elements per R-block of a sparse R-matrix. + * @param smat Sparse matrix keyed by R-coordinate. + * @param all_R_coor Ordered set of R-coordinates to scan. + * @param threshold Magnitude below which an element is treated as zero. + * @param reduce If true, allreduce the per-R counts across MPI ranks. + * @return Per-R nonzero counts aligned with all_R_coor order. + */ +template +std::vector count_nonzeros_by_R( + const SparseRMatrix& smat, + const std::set& all_R_coor, + const double threshold, + const bool reduce) +{ + std::vector nonzero_num(all_R_coor.size(), 0); + int count = 0; + for (const auto& R_coor: all_R_coor) + { + const auto iter = smat.find(R_coor); + if (iter != smat.end()) + { + for (const auto& row_loop: iter->second) + { + for (const auto& col_value: row_loop.second) + { + if (std::abs(col_value.second) > threshold) + { + ++nonzero_num[count]; + } + } + } + } + ++count; + } + + if (reduce) + { + Parallel_Reduce::reduce_all(nonzero_num.data(), static_cast(nonzero_num.size())); + } + return nonzero_num; +} + +/** + * @brief Quit if a sparse-matrix output stream failed to open. + * @param ofs Stream to check. + * @param filename Filename for the error message. + * @param context Caller tag for the error message. + */ +inline void check_output_file_open(const std::ofstream& ofs, + const std::string& filename, + const std::string& context) +{ + if (!ofs.is_open()) + { + ModuleBase::WARNING_QUIT(context, "Cannot open sparse matrix file: " + filename); + } +} +} // namespace detail +} // namespace ModuleIO + +#endif // HS_SPARSE_IO_DETAIL_H diff --git a/source/source_io/module_hs/hsk_writer.cpp b/source/source_io/module_hs/hsk_writer.cpp new file mode 100644 index 00000000000..263635b573b --- /dev/null +++ b/source/source_io/module_hs/hsk_writer.cpp @@ -0,0 +1,106 @@ +#include "hsk_writer.h" +#include "hs_dense_io.h" +#include "source_base/module_out/filename.h" // use filename_output function +#include + +template +void ModuleIO::write_hsk( + const std::string &global_out_dir, + const int nspin, + const int nks, + const int nkstot, + const std::vector &ik2iktot, + const std::vector &isk, + hamilt::Hamilt* p_hamilt, + const Parallel_Orbitals &pv, + const bool gamma_only, + const bool out_app_flag, + const int istep, + const int out_type, + const int precision, + const int nlocal, + const std::string &ks_solver, + const int drank, + std::ofstream &ofs_running) +{ + + ofs_running << " >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>" + ">>>>>>>>>>>>>>>>>>>>>>>>>" << std::endl; + ofs_running << " | " + " |" << std::endl; + ofs_running << " | Write Hamiltonian matrix H(k) or overlap matrix S(k) in numerical |" << std::endl; + ofs_running << " | atomic orbitals at each k-point. |" << std::endl; + ofs_running << " | " + " |" << std::endl; + ofs_running << " >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>" + ">>>>>>>>>>>>>>>>>>>>>>>>>" << std::endl; + ofs_running << "\n WRITE H(k) OR S(k)" << std::endl; + + for (int ik = 0; ik < nks; ++ik) + { + p_hamilt->updateHk(ik); + const bool binary = (out_type == 2); + + hamilt::MatrixBlock h_mat; + hamilt::MatrixBlock s_mat; + + p_hamilt->matrix(h_mat, s_mat); + + std::string h_fn = ModuleIO::filename_output(global_out_dir, + "hk","nao",ik,ik2iktot,nspin,nkstot, + out_type,out_app_flag,gamma_only,istep); + + ModuleIO::save_mat(istep, + h_mat.p, + nlocal, + binary, + precision, + 1, + out_app_flag, + h_fn, + pv, + drank, + ks_solver); + + // mohan note 2025-06-02 + // for overlap matrix, the two spin channels yield the same matrix + // so we only need to print matrix from one spin channel. + const int current_spin = isk[ik]; + if(current_spin == 1) + { + continue; + } + + std::string s_fn = ModuleIO::filename_output(global_out_dir, + "sk","nao",ik,ik2iktot,nspin,nkstot, + out_type,out_app_flag,gamma_only,istep); + + ofs_running << " The output filename is " << s_fn << std::endl; + + ModuleIO::save_mat(istep, + s_mat.p, + nlocal, + binary, + precision, + 1, + out_app_flag, + s_fn, + pv, + drank, + ks_solver); + } // end ik +} + +// Explicit instantiations +template void ModuleIO::write_hsk( + const std::string&, const int, const int, const int, + const std::vector&, const std::vector&, + hamilt::Hamilt*, const Parallel_Orbitals&, + const bool, const bool, const int, const int, const int, + const int, const std::string&, const int, std::ofstream&); +template void ModuleIO::write_hsk>( + const std::string&, const int, const int, const int, + const std::vector&, const std::vector&, + hamilt::Hamilt>*, const Parallel_Orbitals&, + const bool, const bool, const int, const int, const int, + const int, const std::string&, const int, std::ofstream&); diff --git a/source/source_io/module_hs/hsk_writer.h b/source/source_io/module_hs/hsk_writer.h new file mode 100644 index 00000000000..af2bb6a8540 --- /dev/null +++ b/source/source_io/module_hs/hsk_writer.h @@ -0,0 +1,39 @@ +#ifndef HSK_WRITER_H +#define HSK_WRITER_H + +#include +#include +#include + +#include "source_base/parallel_2d.h" +#include "source_basis/module_ao/parallel_orbitals.h" +#include "source_hamilt/hamilt.h" + +// mohan note: this file holds the H(k)/S(k) writer +// (formerly write_hsk in write_hs.h / write_hs.hpp). + +namespace ModuleIO +{ + +template +void write_hsk(const std::string& global_out_dir, + const int nspin, + const int nks, + const int nkstot, + const std::vector& ik2iktot, + const std::vector& isk, + hamilt::Hamilt* p_hamilt, + const Parallel_Orbitals& pv, + const bool gamma_only, + const bool out_app_flag, + const int istep, + const int out_type, + const int precision, + const int nlocal, + const std::string& ks_solver, + const int drank, + std::ofstream& ofs_running); + +} // namespace ModuleIO + +#endif // HSK_WRITER_H diff --git a/source/source_io/module_hs/hsr_legacy.cpp b/source/source_io/module_hs/hsr_legacy.cpp new file mode 100644 index 00000000000..40e37cbdfe1 --- /dev/null +++ b/source/source_io/module_hs/hsr_legacy.cpp @@ -0,0 +1,233 @@ +#include "hsr_legacy.h" + +#include "source_base/timer.h" +#include "source_lcao/lcao_hs_arrays.h" +#include "source_lcao/spar_dh.h" +#include "source_lcao/spar_hsr.h" +#include "source_lcao/spar_st.h" +#include "dhs_sparse_writer.h" +#include "hs_sparse_io.h" + +#include +#include +#include +#include + +// if 'binary=true', output binary file. +// The 'sparse_thr' is the accuracy of the sparse matrix. +// If the absolute value of the matrix element is less than or equal to the +// 'sparse_thr', it will be ignored. + +void ModuleIO::output_dSR(const int& istep, + const UnitCell& ucell, + const Parallel_Orbitals& pv, + LCAO_HS_Arrays& HS_Arrays, + const Grid_Driver& grid, // mohan add 2024-04-06 + const TwoCenterBundle& two_center_bundle, + const LCAO_Orbitals& orb, + const MatROutputOptions& options, + const bool gamma_only_local, + const int npol, + const int nlocal) +{ + ModuleBase::TITLE("ModuleIO", "output_dSR"); + ModuleBase::timer::start("ModuleIO", "output_dSR"); + + sparse_format::cal_dS(ucell, pv, HS_Arrays, grid, two_center_bundle, orb, + options.sparse_threshold, gamma_only_local, options.nspin, npol); + + // mohan update 2024-04-01 + const std::string fileflag_s = "s"; + ModuleIO::save_dH_sparse(istep, pv, HS_Arrays, options.sparse_threshold, options.binary, + fileflag_s, options.precision, options.global_out_dir, + options.global_matrix_dir, options.calculation, options.out_app_flag, + options.nspin, nlocal); + + sparse_format::destroy_dH_R_sparse(HS_Arrays, options.nspin); + + ModuleBase::timer::end("ModuleIO", "output_dSR"); + return; +} + +void ModuleIO::output_dHR(const int& istep, + const ModuleBase::matrix& v_eff, + const UnitCell& ucell, + const Parallel_Orbitals& pv, + LCAO_HS_Arrays& HS_Arrays, + const Grid_Driver& grid, // mohan add 2024-04-06 + const TwoCenterBundle& two_center_bundle, + const LCAO_Orbitals& orb, + const MatROutputOptions& options, + const bool gamma_only_local, + const int npol, + const int nlocal) +{ + ModuleBase::TITLE("ModuleIO", "output_dHR"); + ModuleBase::timer::start("ModuleIO", "output_dHR"); + + if (options.nspin == 1 || options.nspin == 4) + { + // mohan add 2024-04-01 + const int cspin = 0; + + sparse_format::cal_dH(ucell, pv, HS_Arrays, grid, two_center_bundle, orb, cspin, + options.sparse_threshold, v_eff, gamma_only_local, options.nspin, npol); + } + else if (options.nspin == 2) + { + for (int cspin = 0; cspin < 2; cspin++) + { + sparse_format::cal_dH(ucell, pv, HS_Arrays, grid, two_center_bundle, orb, cspin, + options.sparse_threshold, v_eff, gamma_only_local, options.nspin, npol); + } + } + // mohan update 2024-04-01 + const std::string fileflag_h = "h"; + ModuleIO::save_dH_sparse(istep, pv, HS_Arrays, options.sparse_threshold, options.binary, + fileflag_h, options.precision, options.global_out_dir, + options.global_matrix_dir, options.calculation, options.out_app_flag, + options.nspin, nlocal); + + sparse_format::destroy_dH_R_sparse(HS_Arrays, options.nspin); + + ModuleBase::timer::end("ModuleIO", "output_dHR"); + return; +} + +template +void ModuleIO::output_SR(Parallel_Orbitals& pv, + const Grid_Driver& grid, + hamilt::Hamilt* p_ham, + const std::string& SR_filename, + const bool& binary, + const double& sparse_thr, + const int precision, + const std::string& global_out_dir, + const std::string& global_matrix_dir, + const std::string& calculation, + const bool out_app_flag, + const int nspin) +{ + ModuleBase::TITLE("ModuleIO", "output_SR"); + ModuleBase::timer::start("ModuleIO", "output_SR"); + + GlobalV::ofs_running << " Overlap matrix file is in " << SR_filename << std::endl; + + LCAO_HS_Arrays HS_Arrays; + + sparse_format::cal_SR(pv, + HS_Arrays.all_R_coor, + HS_Arrays.SR_sparse, + HS_Arrays.SR_soc_sparse, + grid, + sparse_thr, + p_ham); + + const int istep = 0; + ModuleIO::SparseWriteOptions options; + options.filename = SR_filename; + options.label = "S"; + options.threshold = sparse_thr; + options.binary = binary; + options.precision = precision; + options.istep = istep; + options.reduce = true; + options.calculation = calculation; + options.out_app_flag = out_app_flag; + + if (nspin == 4) + { + ModuleIO::save_sparse(HS_Arrays.SR_soc_sparse, + HS_Arrays.all_R_coor, + pv, + options); + } + else + { + ModuleIO::save_sparse(HS_Arrays.SR_sparse, + HS_Arrays.all_R_coor, + pv, + options); + } + + sparse_format::destroy_HS_R_sparse(HS_Arrays); + + ModuleBase::timer::end("ModuleIO", "output_SR"); + return; +} + +void ModuleIO::output_TR(const int istep, + const UnitCell& ucell, + const Parallel_Orbitals& pv, + LCAO_HS_Arrays& HS_Arrays, + const Grid_Driver& grid, + const TwoCenterBundle& two_center_bundle, + const LCAO_Orbitals& orb, + const std::string& TR_filename, + const MatROutputOptions& options) +{ + ModuleBase::TITLE("ModuleIO", "output_TR"); + ModuleBase::timer::start("ModuleIO", "output_TR"); + + std::stringstream sst; + const bool md_no_append = (options.calculation == "md") && !options.out_app_flag; + if (istep >= 0) + { + // insert the 1-indexed ionic-step suffix right after the "tr" prefix, + // i.e. tr_nao.csr -> trg{step+1}_nao.csr (aligned with lxg/rrg naming) + const std::string base = TR_filename.substr(0, TR_filename.size() - std::string("_nao.csr").size()); + sst << (md_no_append ? options.global_matrix_dir : options.global_out_dir) << base << "g" << (istep + 1) << "_nao.csr"; + } + else + { + sst << options.global_out_dir << TR_filename; + } + GlobalV::ofs_running << " T(R) data are in file: " << sst.str() << std::endl; + + sparse_format::cal_TR(ucell, pv, HS_Arrays, grid, two_center_bundle, orb, options.sparse_threshold); + ModuleIO::SparseWriteOptions sparse_options; + sparse_options.filename = sst.str(); + sparse_options.label = "T"; + sparse_options.threshold = options.sparse_threshold; + sparse_options.binary = options.binary; + sparse_options.precision = options.precision; + sparse_options.istep = istep; + sparse_options.reduce = true; + sparse_options.calculation = options.calculation; + sparse_options.out_app_flag = options.out_app_flag; + + ModuleIO::save_sparse(HS_Arrays.TR_sparse, + HS_Arrays.all_R_coor, + pv, + sparse_options); + + sparse_format::destroy_T_R_sparse(HS_Arrays); + + ModuleBase::timer::end("ModuleIO", "output_TR"); + return; +} + +template void ModuleIO::output_SR(Parallel_Orbitals& pv, + const Grid_Driver& grid, + hamilt::Hamilt* p_ham, + const std::string& SR_filename, + const bool& binary, + const double& sparse_thr, + const int precision, + const std::string& global_out_dir, + const std::string& global_matrix_dir, + const std::string& calculation, + const bool out_app_flag, + const int nspin); +template void ModuleIO::output_SR>(Parallel_Orbitals& pv, + const Grid_Driver& grid, + hamilt::Hamilt>* p_ham, + const std::string& SR_filename, + const bool& binary, + const double& sparse_thr, + const int precision, + const std::string& global_out_dir, + const std::string& global_matrix_dir, + const std::string& calculation, + const bool out_app_flag, + const int nspin); diff --git a/source/source_io/module_hs/hsr_legacy.h b/source/source_io/module_hs/hsr_legacy.h new file mode 100644 index 00000000000..9c4c61f8b38 --- /dev/null +++ b/source/source_io/module_hs/hsr_legacy.h @@ -0,0 +1,86 @@ +#ifndef HSR_LEGACY_H +#define HSR_LEGACY_H + +#include "source_base/matrix.h" +#include "source_basis/module_ao/parallel_orbitals.h" +#include "source_basis/module_nao/two_center_bundle.h" +#include "source_cell/klist.h" +#include "source_cell/module_neighbor/sltk_grid_driver.h" +#include "source_hamilt/hamilt.h" +#include "source_lcao/lcao_hs_arrays.h" + +#include + +// Legacy LCAO_HS_Arrays-based sparse matrix output path (dH/dR, dS/dR, T(R), S(R)). +// Kept as-is from the former write_hs_r.h; new code should prefer the +// HContainer-based writers in hsr_writer.h. + +namespace ModuleIO +{ +// Groups the shared format / path / runtime flags threaded through the +// LCAO_HS_Arrays-based writers. Replaces ten positional parameters that were +// identical across output_dHR/output_dSR/output_TR. +struct MatROutputOptions +{ + bool binary = false; + double sparse_threshold = 0.0; + int precision = 8; + std::string global_out_dir; + std::string global_matrix_dir; + std::string calculation; + bool out_app_flag = false; + int nspin = 1; +}; + +void output_dHR(const int& istep, + const ModuleBase::matrix& v_eff, + const UnitCell& ucell, + const Parallel_Orbitals& pv, + LCAO_HS_Arrays& HS_Arrays, + const Grid_Driver& grid, // mohan add 2024-04-06 + const TwoCenterBundle& two_center_bundle, + const LCAO_Orbitals& orb, + const MatROutputOptions& options, + const bool gamma_only_local, + const int npol, + const int nlocal); + +void output_dSR(const int& istep, + const UnitCell& ucell, + const Parallel_Orbitals& pv, + LCAO_HS_Arrays& HS_Arrays, + const Grid_Driver& grid, // mohan add 2024-04-06 + const TwoCenterBundle& two_center_bundle, + const LCAO_Orbitals& orb, + const MatROutputOptions& options, + const bool gamma_only_local, + const int npol, + const int nlocal); + +void output_TR(const int istep, + const UnitCell& ucell, + const Parallel_Orbitals& pv, + LCAO_HS_Arrays& HS_Arrays, + const Grid_Driver& grid, + const TwoCenterBundle& two_center_bundle, + const LCAO_Orbitals& orb, + const std::string& TR_filename, + const MatROutputOptions& options); + +template +void output_SR(Parallel_Orbitals& pv, + const Grid_Driver& grid, + hamilt::Hamilt* p_ham, + const std::string& SR_filename, + const bool& binary, + const double& sparse_threshold, + const int precision, + const std::string& global_out_dir, + const std::string& global_matrix_dir, + const std::string& calculation, + const bool out_app_flag, + const int nspin); + +} // namespace ModuleIO + +#endif diff --git a/source/source_io/module_hs/hsr_writer.cpp b/source/source_io/module_hs/hsr_writer.cpp new file mode 100644 index 00000000000..7a83b2c57cb --- /dev/null +++ b/source/source_io/module_hs/hsr_writer.cpp @@ -0,0 +1,381 @@ +#include "hsr_writer.h" + +#include "source_base/module_out/sparse_matrix.h" +#include "source_base/tool_quit.h" +#include "source_cell/ucell_io.h" +#include "source_cell/unitcell.h" +#include "source_base/global_variable.h" +#include "source_basis/module_ao/parallel_orbitals.h" +#include "source_hamilt/module_hcontainer/hcontainer.h" +#include "source_hamilt/module_hcontainer/hcontainer_funcs.h" +#include "source_hamilt/module_hcontainer/output_hcontainer.h" + +#include +#include +#include +#include +#include +#include +#include + +std::string ModuleIO::hsr_gen_fname(const std::string& prefix, + const int ispin, + const bool append, + const int istep) +{ + return hsr_gen_fname(prefix, ispin, append, istep, 1); +} + +std::string ModuleIO::hsr_gen_fname(const std::string& prefix, + const int ispin, + const bool append, + const int istep, + const int out_type) +{ + const std::string extension = out_type == 2 ? ".dat" : ".csr"; + if (!append && istep >= 0) + { + return prefix + std::to_string(ispin + 1) + "g" + std::to_string(istep + 1) + "_nao" + extension; + } + else + { + return prefix + std::to_string(ispin + 1) + "_nao" + extension; + } +} + +std::string ModuleIO::sr_gen_fname(const bool append, const int istep) +{ + return sr_gen_fname(append, istep, 1); +} + +std::string ModuleIO::sr_gen_fname(const bool append, const int istep, const int out_type) +{ + const std::string extension = out_type == 2 ? ".dat" : ".csr"; + if (!append && istep >= 0) + { + return "srg" + std::to_string(istep + 1) + "_nao" + extension; + } + return "sr_nao" + extension; +} + +std::string ModuleIO::dhr_gen_fname(const std::string& prefix, + const int ispin, + const bool append, + const int istep) +{ + std::string fname = prefix + "rs" + std::to_string(ispin + 1); + if (!append && istep >= 0) + { + fname += "g" + std::to_string(istep + 1); + } + fname += "_nao.csr"; + return fname; +} + +template +void ModuleIO::write_hcontainer_csr(const std::string& fname, + const UnitCell* ucell, + const int precision, + hamilt::HContainer* mat_serial, + const int istep, + const int ispin, + const int nspin, + const std::string& label, + const std::string& representation_note) +{ + std::ofstream ofs; + if (istep <= 0) + { + ofs.open(fname); + } + else + { + ofs.open(fname, std::ios::app); + } + if (!ofs.is_open()) + { + ModuleBase::WARNING_QUIT("ModuleIO::write_hcontainer_csr", + "Cannot open HContainer CSR file: " + fname); + } + + ofs << " --- Ionic Step " << istep + 1 << " ---" << std::endl; + ofs << " # print " << label << " matrix in real space " << label << "(R)" << std::endl; + ofs << " " << nspin << " # number of spin directions" << std::endl; + ofs << " " << ispin + 1 << " # spin index" << std::endl; + ofs << " " << mat_serial->get_nbasis() << " # number of localized basis" << std::endl; + ofs << " " << mat_serial->size_R_loop() << " # number of Bravais lattice vector R" << std::endl; + ofs << std::endl; + + ModuleIO::UcellIO::write_ucell(ofs, ucell); + if (!representation_note.empty()) + { + ofs << "# representation: " << representation_note << std::endl; + } + ofs << std::endl; + + const double sparse_threshold = 1e-10; + hamilt::Output_HContainer out(mat_serial, ofs, sparse_threshold, precision); + out.write(); + ofs.close(); +} + +namespace +{ +template +void write_native_value(std::ofstream& ofs, const T& value) +{ + ofs.write(reinterpret_cast(&value), sizeof(T)); +} + +void write_native_value(std::ofstream& ofs, const std::complex& value) +{ + const double real = value.real(); + const double imag = value.imag(); + write_native_value(ofs, real); + write_native_value(ofs, imag); +} + +template +ModuleIO::SparseMatrix make_sparse_R_block(hamilt::HContainer* mat_serial, + const int rx, + const int ry, + const int rz, + const double sparse_threshold) +{ + const int nbasis = mat_serial->get_nbasis(); + ModuleIO::SparseMatrix sparse_matrix(nbasis, nbasis); + sparse_matrix.setSparseThreshold(sparse_threshold); + mat_serial->fix_R(rx, ry, rz); + + for (int iap = 0; iap < mat_serial->size_atom_pairs(); ++iap) + { + const auto atom_pair = mat_serial->get_atom_pair(iap); + const int r_index = atom_pair.find_R(rx, ry, rz); + if (r_index < 0) + { + continue; + } + const auto matrix_info = atom_pair.get_matrix_values(r_index); + const int* index = std::get<0>(matrix_info).data(); + TR* data = std::get<1>(matrix_info); + for (int irow = index[0]; irow < index[0] + index[1]; ++irow) + { + for (int icol = index[2]; icol < index[2] + index[3]; ++icol) + { + sparse_matrix.insert(irow, icol, *data); + ++data; + } + } + } + + mat_serial->unfix_R(); + return sparse_matrix; +} +} // namespace + +template +void ModuleIO::write_hcontainer_csr_binary(const std::string& fname, + hamilt::HContainer* mat_serial, + const int istep, + const bool append) +{ + std::ios_base::openmode mode = std::ios::out | std::ios::binary; + if (append && istep > 0) + { + mode |= std::ios::app; + } + std::ofstream ofs(fname.c_str(), mode); + if (!ofs.is_open()) + { + ModuleBase::WARNING_QUIT("ModuleIO::write_hcontainer_csr_binary", + "Cannot open HContainer binary CSR file: " + fname); + } + + const size_t nR_size = mat_serial->size_R_loop(); + if (nR_size > static_cast(std::numeric_limits::max())) + { + ModuleBase::WARNING_QUIT("ModuleIO::write_hcontainer_csr_binary", + "Too many R blocks for native binary CSR output: " + fname); + } + + std::vector> R_coordinates; + R_coordinates.reserve(nR_size); + for (size_t iR = 0; iR < nR_size; ++iR) + { + int rx = 0; + int ry = 0; + int rz = 0; + mat_serial->loop_R(iR, rx, ry, rz); + R_coordinates.push_back(std::make_tuple(rx, ry, rz)); + } + std::sort(R_coordinates.begin(), R_coordinates.end()); + + const int step = std::max(istep, 0); + const int nbasis = mat_serial->get_nbasis(); + const int nR = static_cast(R_coordinates.size()); + write_native_value(ofs, step); + write_native_value(ofs, nbasis); + write_native_value(ofs, nR); + + const double sparse_threshold = 1e-10; + for (const auto& R_coordinate: R_coordinates) + { + const int rx = std::get<0>(R_coordinate); + const int ry = std::get<1>(R_coordinate); + const int rz = std::get<2>(R_coordinate); + const SparseMatrix sparse_matrix + = make_sparse_R_block(mat_serial, rx, ry, rz, sparse_threshold); + const auto& elements = sparse_matrix.getElements(); + if (elements.size() > static_cast(std::numeric_limits::max())) + { + ModuleBase::WARNING_QUIT("ModuleIO::write_hcontainer_csr_binary", + "Too many nonzero values for native binary CSR output: " + fname); + } + const int nnz = static_cast(elements.size()); + + write_native_value(ofs, rx); + write_native_value(ofs, ry); + write_native_value(ofs, rz); + write_native_value(ofs, nnz); + + for (const auto& element: elements) + { + write_native_value(ofs, element.second); + } + + std::vector row_ptr(nbasis + 1, 0); + for (const auto& element: elements) + { + write_native_value(ofs, element.first.second); + ++row_ptr[element.first.first + 1]; + } + for (int irow = 1; irow <= nbasis; ++irow) + { + row_ptr[irow] += row_ptr[irow - 1]; + } + for (const long long pointer: row_ptr) + { + write_native_value(ofs, pointer); + } + } + + ofs.close(); + if (!ofs) + { + ModuleBase::WARNING_QUIT("ModuleIO::write_hcontainer_csr_binary", + "Failed to write HContainer binary CSR file: " + fname); + } +} + +template +void ModuleIO::write_hsr(const std::vector*>& hr_vec, + const hamilt::HContainer* sr, + const UnitCell* ucell, + const int out_type, + const int precision, + const Parallel_2D& paraV, + const bool append, + const bool gamma_only, + const int* iat2iwt, + const int nat, + const int istep, + const std::string& global_out_dir) +{ + if (out_type != 1 && out_type != 2) + { + ModuleBase::WARNING_QUIT("ModuleIO::write_hsr", "out_type must be 1 or 2"); + } + const int nspin = hr_vec.size(); + assert(nspin > 0); + const std::string representation_note + = gamma_only + ? "gamma-only folded matrix; stored R-space contributions are summed into R = (0, 0, 0)" + : ""; + + // Output HR (one file per spin) + for (int ispin = 0; ispin < nspin; ispin++) + { + const int nbasis = hr_vec[ispin]->get_nbasis(); + +#ifdef __MPI + Parallel_Orbitals serialV; + serialV.set_serial(nbasis, nbasis); + serialV.set_atomic_trace(iat2iwt, nat, nbasis); + hamilt::HContainer hr_serial(&serialV); + hamilt::gatherParallels(*hr_vec[ispin], &hr_serial, 0); +#else + hamilt::HContainer hr_serial(*hr_vec[ispin]); +#endif + + if (GlobalV::MY_RANK == 0) + { + std::string fname = global_out_dir + + hsr_gen_fname("hrs", ispin, append, istep, out_type); + if (out_type == 2) + { + write_hcontainer_csr_binary(fname, &hr_serial, istep, append); + } + else + { + write_hcontainer_csr( + fname, ucell, precision, &hr_serial, istep, ispin, nspin, "H", representation_note); + } + } + } + + // Output SR (single file) + { + const int nbasis = sr->get_nbasis(); + +#ifdef __MPI + Parallel_Orbitals serialV; + serialV.set_serial(nbasis, nbasis); + serialV.set_atomic_trace(iat2iwt, nat, nbasis); + hamilt::HContainer sr_serial(&serialV); + hamilt::gatherParallels(*sr, &sr_serial, 0); +#else + hamilt::HContainer sr_serial(*sr); +#endif + + if (GlobalV::MY_RANK == 0) + { + std::string fname = global_out_dir + + sr_gen_fname(append, istep, out_type); + if (out_type == 2) + { + write_hcontainer_csr_binary(fname, &sr_serial, istep, append); + } + else + { + write_hcontainer_csr( + fname, ucell, precision, &sr_serial, istep, 0, 1, "S", representation_note); + } + } + } +} + +// Explicit instantiations +template void ModuleIO::write_hcontainer_csr( + const std::string&, const UnitCell*, const int, + hamilt::HContainer*, const int, const int, const int, const std::string&, const std::string&); +template void ModuleIO::write_hcontainer_csr>( + const std::string&, const UnitCell*, const int, + hamilt::HContainer>*, const int, const int, const int, const std::string&, const std::string&); + +template void ModuleIO::write_hcontainer_csr_binary( + const std::string&, hamilt::HContainer*, const int, const bool); +template void ModuleIO::write_hcontainer_csr_binary>( + const std::string&, hamilt::HContainer>*, const int, const bool); + +template void ModuleIO::write_hsr( + const std::vector*>&, + const hamilt::HContainer*, + const UnitCell*, const int, const int, const Parallel_2D&, + const bool, const bool, const int*, const int, const int, + const std::string&); +template void ModuleIO::write_hsr>( + const std::vector>*>&, + const hamilt::HContainer>*, + const UnitCell*, const int, const int, const Parallel_2D&, + const bool, const bool, const int*, const int, const int, + const std::string&); diff --git a/source/source_io/module_hs/hsr_writer.h b/source/source_io/module_hs/hsr_writer.h new file mode 100644 index 00000000000..b4f836035da --- /dev/null +++ b/source/source_io/module_hs/hsr_writer.h @@ -0,0 +1,82 @@ +#ifndef HSR_WRITER_H +#define HSR_WRITER_H + +#include "source_base/parallel_2d.h" + +#include +#include +#include + +class UnitCell; + +namespace hamilt +{ +template +class HContainer; +} // namespace hamilt + +namespace ModuleIO +{ + +/// Generate filename for spin-dependent HR output. +std::string hsr_gen_fname(const std::string& prefix, + const int ispin, + const bool append, + const int istep); + +/// Generate filename for spin-dependent HR output in the selected format. +std::string hsr_gen_fname(const std::string& prefix, + const int ispin, + const bool append, + const int istep, + const int out_type); + +/// Generate filename for spin-independent SR output. +std::string sr_gen_fname(const bool append, const int istep); + +/// Generate filename for spin-independent SR output in the selected format. +std::string sr_gen_fname(const bool append, const int istep, const int out_type); + +/// Generate filename for derivative matrices (dH/dR, dS/dR). +std::string dhr_gen_fname(const std::string& prefix, + const int ispin, + const bool append, + const int istep); + +/// Write a single HContainer to CSR file with header. +template +void write_hcontainer_csr(const std::string& fname, + const UnitCell* ucell, + const int precision, + hamilt::HContainer* mat_serial, + const int istep, + const int ispin, + const int nspin, + const std::string& label, + const std::string& representation_note); + +/// Write one HContainer record in the native binary CSR format. +template +void write_hcontainer_csr_binary(const std::string& fname, + hamilt::HContainer* mat_serial, + const int istep, + const bool append); + +/// Write H(R) and S(R) in CSR format, unified with write_dmr interface. +template +void write_hsr(const std::vector*>& hr_vec, + const hamilt::HContainer* sr, + const UnitCell* ucell, + const int out_type, + const int precision, + const Parallel_2D& paraV, + const bool append, + const bool gamma_only, + const int* iat2iwt, + const int nat, + const int istep, + const std::string& global_out_dir); + +} // namespace ModuleIO + +#endif // HSR_WRITER_H diff --git a/source/source_io/module_hs/hterm_writer.cpp b/source/source_io/module_hs/hterm_writer.cpp new file mode 100644 index 00000000000..30c1921b923 --- /dev/null +++ b/source/source_io/module_hs/hterm_writer.cpp @@ -0,0 +1,434 @@ +#include "hterm_writer.h" + +#include "source_base/parallel_reduce.h" +#include "source_base/timer.h" +#include "source_estate/module_pot/h_hartree_pw.h" +#include "source_hamilt/module_xc/xc_functional.h" +#include "source_io/module_hs/hs_dense_io.h" +#include "source_io/module_hs/hsr_writer.h" +#include "source_base/module_out/filename.h" +#include "source_cell/ucell_io.h" +#include "source_hamilt/module_gint/gint_interface.h" +#include "source_hamilt/module_hcontainer/hcontainer_funcs.h" +#include "source_hamilt/module_hcontainer/output_hcontainer.h" +#include "source_lcao/module_operator_lcao/ekinetic.h" +#include "source_lcao/module_operator_lcao/nonlocal.h" +#include "source_lcao/module_operator_lcao/operator_fs_utils.h" +#ifdef __EXX +#include "source_lcao/module_operator_lcao/op_exx_lcao.h" +#include "source_lcao/module_ri/exx_lri_interface.h" +#include "source_lcao/module_ri/ri_2d_comm.h" +#endif + +#include +#include +#include +#include +#include + +namespace ModuleIO +{ + +static void setup_veff_hcontainer(hamilt::HContainer& hR, + const UnitCell& ucell, + const Grid_Driver& gd, + const Parallel_Orbitals& pv, + const std::vector& orb_cutoff) +{ + const Parallel_Orbitals* paraV = hR.get_paraV(); + for (int iat1 = 0; iat1 < ucell.nat; iat1++) + { + auto tau1 = ucell.get_tau(iat1); + int T1 = 0, I1 = 0; + ucell.iat2iait(iat1, &I1, &T1); + + AdjacentAtomInfo adjs; + gd.Find_atom(ucell, tau1, T1, I1, &adjs); + + for (int ad = 0; ad < adjs.adj_num + 1; ++ad) + { + const int T2 = adjs.ntype[ad]; + const int I2 = adjs.natom[ad]; + const int iat2 = ucell.itia2iat(T2, I2); + if (paraV->is_invalid_atom_pair(iat1, iat2)) + { + continue; + } + const ModuleBase::Vector3& R_index = adjs.box[ad]; + if (ucell.cal_dtau(iat1, iat2, R_index).norm() * ucell.lat0 < orb_cutoff[T1] + orb_cutoff[T2]) + { + hamilt::AtomPair tmp(iat1, iat2, R_index, paraV); + hR.insert_pair(tmp); + } + } + } + hR.allocate(nullptr, true); +} + +static void gather_and_write(const std::string& prefix, + const std::string& label, + hamilt::HContainer& hR, + const UnitCell& ucell, + const Parallel_Orbitals& pv, + const int nspin, + const int ispin, + const int istep, + const bool append, + const int* iat2iwt, + const int nat, + const std::string& calculation, + const bool out_app_flag, + const std::string& global_out_dir, + const std::string& global_matrix_dir) +{ + const int nbasis = hR.get_nbasis(); +#ifdef __MPI + Parallel_Orbitals serialV; + serialV.set_serial(nbasis, nbasis); + serialV.set_atomic_trace(iat2iwt, nat, nbasis); + hamilt::HContainer hr_serial(&serialV); + hamilt::gatherParallels(hR, &hr_serial, 0); + if (GlobalV::MY_RANK == 0) +#endif + { + const bool md_no_append = (calculation == "md") && !out_app_flag; + const std::string& out_dir = md_no_append ? global_matrix_dir : global_out_dir; + const std::string fname = out_dir + hsr_gen_fname(prefix, ispin, append, istep); +#ifdef __MPI + write_hcontainer_csr(fname, &ucell, 8, &hr_serial, istep, ispin, nspin, label, ""); +#else + write_hcontainer_csr(fname, &ucell, 8, &hR, istep, ispin, nspin, label, ""); +#endif + } +} + +static void write_hk_common(hamilt::HContainer& hR, + const std::string& prefix, + const UnitCell& ucell, + const Parallel_Orbitals& pv, + const K_Vectors& kv, + const int nspin, + const int istep, + const bool append, + const int* iat2iwt, + const int nat, + const int nlocal, + const bool gamma_only, + const std::string& global_out_dir, + const bool out_app_flag, + const std::string& ks_solver, + const int drank) +{ + const int nspin_k = (nspin == 2 ? 2 : 1); + const int nks = kv.get_nks() / nspin_k; + + for (int ik = 0; ik < nks; ++ik) + { + const ModuleBase::Vector3& kvec_d = kv.kvec_d[ik]; + + std::vector> hk_global(nlocal * nlocal, 0); + hamilt::folding_HR(hR, hk_global.data(), kvec_d, nlocal, 0); + + const int out_label = 1; + std::string fname = ModuleIO::filename_output(global_out_dir, + prefix, + "nao", + ik, + kv.ik2iktot, + nspin, + kv.get_nkstot(), + out_label, + out_app_flag, + gamma_only, + istep); + ModuleIO::save_mat(istep, + hk_global.data(), + nlocal, + false, + 8, + false, + out_app_flag, + fname, + pv, + drank, + ks_solver); + } +} + +// Descriptor for one Hamiltonian term: the H(k)/H(R) filename prefixes and the +// term-specific routine that fills hR_tmp for one spin channel. +struct HTermSpec +{ + std::string timer_label; + std::string k_prefix; + std::string r_prefix; + std::string r_label; + // Fills hR_tmp for spin channel ispin. setup_veff_hcontainer has already + // been run when use_veff_layout is true; the builder adds the term values. + std::function&, int)> build; +}; + +// Shared skeleton for the five non-EXX terms: loop over spin channels, build +// the term into hR_tmp, always write H(k), and write H(R) only when asked. +static void write_h_term(WriteHParams& params, const HTermSpec& spec) +{ + ModuleBase::TITLE("ModuleIO", spec.timer_label); + ModuleBase::timer::start("ModuleIO", spec.timer_label); + + const UnitCell& ucell = *params.ucell; + const Parallel_Orbitals& pv = *params.pv; + const K_Vectors& kv = *params.kv; + const int nspin = params.nspin; + const int istep = params.istep; + const bool append = params.append; + const int* iat2iwt = params.iat2iwt; + const int nat = params.nat; + const bool also_hR = params.also_hR; + const int nlocal = params.nlocal; + const bool gamma_only = params.gamma_only_local; + const std::string& global_out_dir = params.global_out_dir; + const bool out_app_flag = params.out_app_flag; + + const int nspin_out = (nspin == 2 ? 2 : 1); + + for (int ispin = 0; ispin < nspin_out; ispin++) + { + hamilt::HContainer hR_tmp(const_cast(&pv)); + spec.build(hR_tmp, ispin); + + write_hk_common(hR_tmp, spec.k_prefix, ucell, pv, kv, nspin, istep, append, iat2iwt, nat, + nlocal, gamma_only, global_out_dir, out_app_flag, params.ks_solver, params.drank); + + if (also_hR) + { + gather_and_write(spec.r_prefix, spec.r_label, hR_tmp, ucell, pv, nspin, ispin, istep, append, + iat2iwt, nat, params.calculation, params.out_app_flag, + params.global_out_dir, params.global_matrix_dir); + } + } + + ModuleBase::timer::end("ModuleIO", spec.timer_label); +} + +void write_h_t(WriteHParams& params) +{ + const UnitCell& ucell = *params.ucell; + const Grid_Driver& gd = *params.gd; + const Parallel_Orbitals& pv = *params.pv; + const TwoCenterBundle& two_center_bundle = *params.two_center_bundle; + const LCAO_Orbitals& orb = *params.orb; + const K_Vectors& kv = *params.kv; + const std::vector& orb_cutoff = orb.cutoffs(); + + HTermSpec spec; + spec.timer_label = "write_h_t"; + spec.k_prefix = "tk"; + spec.r_prefix = "t"; + spec.r_label = "T"; + spec.build = [&](hamilt::HContainer& hR_tmp, int /*ispin*/) { + hamilt::EKinetic> + tmp_ekinetic(nullptr, kv.kvec_d, &hR_tmp, &ucell, orb_cutoff, &gd, + two_center_bundle.kinetic_orb.get()); + tmp_ekinetic.contributeHR(); + }; + write_h_term(params, spec); +} + +void write_h_vnl(WriteHParams& params) +{ + const UnitCell& ucell = *params.ucell; + const Grid_Driver& gd = *params.gd; + const TwoCenterBundle& two_center_bundle = *params.two_center_bundle; + const LCAO_Orbitals& orb = *params.orb; + const K_Vectors& kv = *params.kv; + const std::vector& orb_cutoff = orb.cutoffs(); + + HTermSpec spec; + spec.timer_label = "write_h_vnl"; + spec.k_prefix = "vnlk"; + spec.r_prefix = "vnl"; + spec.r_label = "V^NL"; + spec.build = [&](hamilt::HContainer& hR_tmp, int /*ispin*/) { + hamilt::Nonlocal> + tmp_nonlocal(nullptr, kv.kvec_d, &hR_tmp, &ucell, orb_cutoff, &gd, + two_center_bundle.overlap_orb_beta.get()); + tmp_nonlocal.contributeHR(); + }; + write_h_term(params, spec); +} + +void write_h_vl(WriteHParams& params) +{ + const UnitCell& ucell = *params.ucell; + const Grid_Driver& gd = *params.gd; + const Parallel_Orbitals& pv = *params.pv; + const LCAO_Orbitals& orb = *params.orb; + const elecstate::Potential* pot = params.pot; + const std::vector& orb_cutoff = orb.cutoffs(); + + const double* v_local = pot->get_fixed_v(); // local pp, no Hxc + + HTermSpec spec; + spec.timer_label = "write_h_vl"; + spec.k_prefix = "vlk"; + spec.r_prefix = "vl"; + spec.r_label = "V^L"; + spec.build = [&](hamilt::HContainer& hR_tmp, int /*ispin*/) { + setup_veff_hcontainer(hR_tmp, ucell, gd, pv, orb_cutoff); + ModuleGint::cal_gint_vl(v_local, &hR_tmp); + }; + write_h_term(params, spec); +} + +void write_h_vh(WriteHParams& params) +{ + const UnitCell& ucell = *params.ucell; + const Grid_Driver& gd = *params.gd; + const Parallel_Orbitals& pv = *params.pv; + const LCAO_Orbitals& orb = *params.orb; + const Charge* chg = params.chg; + const ModulePW::PW_Basis* rho_basis = params.rho_basis; + const std::vector& orb_cutoff = orb.cutoffs(); + + ModuleBase::matrix v_h = elecstate::H_Hartree_pw::v_hartree( + ucell, const_cast(rho_basis), params.nspin, chg->rho); + + HTermSpec spec; + spec.timer_label = "write_h_vh"; + spec.k_prefix = "vhk"; + spec.r_prefix = "vh"; + spec.r_label = "V^H"; + spec.build = [&](hamilt::HContainer& hR_tmp, int ispin) { + setup_veff_hcontainer(hR_tmp, ucell, gd, pv, orb_cutoff); + ModuleGint::cal_gint_vl(&v_h(ispin, 0), &hR_tmp); + }; + write_h_term(params, spec); +} + +void write_h_vxc(WriteHParams& params) +{ + const UnitCell& ucell = *params.ucell; + const Grid_Driver& gd = *params.gd; + const Parallel_Orbitals& pv = *params.pv; + const LCAO_Orbitals& orb = *params.orb; + const Charge* chg = params.chg; + const int nrxx = params.nrxx; + const std::vector& orb_cutoff = orb.cutoffs(); + + ModuleBase::matrix v_xc; + double etxc; + double vtxc; + const double hybrid_alpha = XC_Functional::get_hybrid_alpha(); +#ifdef __EXX + const double hse_omega = XC_Functional::get_hse_omega(); +#else + const double hse_omega = 0.0; +#endif + std::tie(etxc, vtxc, v_xc) = XC_Functional::v_xc(nrxx, chg, &ucell, params.nspin, + params.domag, params.domag_z, + params.gga_grad, + hybrid_alpha, hse_omega); + + HTermSpec spec; + spec.timer_label = "write_h_vxc"; + spec.k_prefix = "vxck"; + spec.r_prefix = "vxc"; + spec.r_label = "V^XC"; + spec.build = [&](hamilt::HContainer& hR_tmp, int ispin) { + setup_veff_hcontainer(hR_tmp, ucell, gd, pv, orb_cutoff); + ModuleGint::cal_gint_vl(&v_xc(ispin, 0), &hR_tmp); + }; + write_h_term(params, spec); +} + +#ifdef __EXX +// Build V^EXX(R) for one interface (real or complex Hexx) into real HContainers and write them. +template +static void write_h_exx_impl(const UnitCell& ucell, + const Parallel_Orbitals& pv, + Exx_LRI_Interface* ex, + const K_Vectors& kv, + const int nspin, + const int istep, + const bool append, + const int* iat2iwt, + const int nat, + const bool also_hR, + const int npol, + const int nlocal, + const bool gamma_only, + const std::string& global_out_dir, + const std::string& global_matrix_dir, + const std::string& calculation, + const bool out_app_flag, + const std::string& ks_solver, + const int drank, + const Exx_Info& exx_info) +{ + const auto& Hexxs = ex->get_Hexxs(); // vector over spin of map> + const int nspin_out = (nspin == 2 ? 2 : 1); + const double alpha = exx_info.info_global.hybrid_alpha; + + for (int ispin = 0; ispin < nspin_out; ispin++) + { + hamilt::HContainer hR_tmp(const_cast(&pv)); + // add_HexxR only fills existing matrices, so first allocate the atom-pair structure + // from the exx-form data (native cells, consistent with the nullptr cell_nearest below). + hamilt::reallocate_hcontainer(Hexxs, &hR_tmp); + RI_2D_Comm::add_HexxR(ispin, alpha, Hexxs, pv, npol, hR_tmp, nullptr); + + write_hk_common(hR_tmp, "vexxk", ucell, pv, kv, nspin, istep, append, iat2iwt, nat, + nlocal, gamma_only, global_out_dir, out_app_flag, ks_solver, drank); + + if (also_hR) + { + gather_and_write("vexx", "V^EXX", hR_tmp, ucell, pv, nspin, ispin, istep, append, iat2iwt, nat, + calculation, out_app_flag, global_out_dir, global_matrix_dir); + } + } +} + +void write_h_exx(WriteHParams& params, const Exx_Info& exx_info) +{ + ModuleBase::TITLE("ModuleIO", "write_h_exx"); + ModuleBase::timer::start("ModuleIO", "write_h_exx"); + + // Multi-k out_mat_h_exx is rejected upstream at the call site (setup_exx_h_params in + // ctrl_scf_lcao.cpp); this function is only reached on the gamma-only path. + const UnitCell& ucell = *params.ucell; + const Parallel_Orbitals& pv = *params.pv; + const K_Vectors& kv = *params.kv; + const int nspin = params.nspin; + const int istep = params.istep; + const bool append = params.append; + const int* iat2iwt = params.iat2iwt; + const int nat = params.nat; + const bool also_hR = params.also_hR; + + // exd (real Hexx) and exc (complex Hexx) are mutually exclusive; pick by real_number. + if (exx_info.info_ri.real_number) + { + if (params.exd != nullptr) + { + write_h_exx_impl(ucell, pv, params.exd, kv, nspin, istep, append, iat2iwt, nat, also_hR, + params.npol, params.nlocal, params.gamma_only_local, + params.global_out_dir, params.global_matrix_dir, + params.calculation, params.out_app_flag, params.ks_solver, params.drank, exx_info); + } + } + else + { + if (params.exc != nullptr) + { + write_h_exx_impl(ucell, pv, params.exc, kv, nspin, istep, append, iat2iwt, nat, also_hR, + params.npol, params.nlocal, params.gamma_only_local, + params.global_out_dir, params.global_matrix_dir, + params.calculation, params.out_app_flag, params.ks_solver, params.drank, exx_info); + } + } + + ModuleBase::timer::end("ModuleIO", "write_h_exx"); +} +#endif + +} // namespace ModuleIO diff --git a/source/source_io/module_hs/hterm_writer.h b/source/source_io/module_hs/hterm_writer.h new file mode 100644 index 00000000000..64e5572d55f --- /dev/null +++ b/source/source_io/module_hs/hterm_writer.h @@ -0,0 +1,87 @@ +#ifndef HTERM_WRITER_H +#define HTERM_WRITER_H + +#include "source_basis/module_nao/two_center_bundle.h" +#include "source_basis/module_pw/pw_basis.h" +#include "source_cell/klist.h" +#include "source_cell/module_neighbor/sltk_grid_driver.h" +#include "source_estate/module_charge/charge.h" +#include "source_estate/module_pot/potential_new.h" +#include "source_lcao/lcao_domain.h" +#include "source_hamilt/module_hcontainer/hcontainer.h" +#include "source_hamilt/module_xc/exx_info.h" + +#include +#include +#include + +template +class Exx_LRI_Interface; + +namespace ModuleIO +{ + +struct WriteHParams +{ + const UnitCell* ucell = nullptr; + const Grid_Driver* gd = nullptr; + const Parallel_Orbitals* pv = nullptr; + const TwoCenterBundle* two_center_bundle = nullptr; + const LCAO_Orbitals* orb = nullptr; + const K_Vectors* kv = nullptr; + const elecstate::Potential* pot = nullptr; // used by the local-pp term only + const Charge* chg = nullptr; // used by the Hartree / XC terms + const ModulePW::PW_Basis* rho_basis = nullptr; // used by the Hartree term + int nrxx = 0; // used by the XC term + int nspin = 1; + int istep = 0; + bool append = false; + const int* iat2iwt = nullptr; + int nat = 0; + bool also_hR = false; // H(k) is always written; H(R) (CSR) only when this is true + int nlocal = 0; + bool gamma_only_local = false; + int npol = 1; + bool domag = false; + bool domag_z = false; + int gga_grad = 0; // used by the XC term + bool out_app_flag = false; + std::string calculation; + std::string global_out_dir; + std::string global_matrix_dir; + std::string ks_solver; + int drank = 0; +#ifdef __EXX + // The gamma-only (TK==double) exx interfaces used by the EXX term. + // Deliberately NOT templated on TK, because it would force WriteHParams and + // every free function taking it to become templates as well -- a large, + // purely mechanical change for a case nobody needs. + // Multi-k + EXX is therefore rejected up front (see write_h_exx_term) + // instead of silently producing output with the EXX term missing. + Exx_LRI_Interface* exd = nullptr; + Exx_LRI_Interface>* exc = nullptr; +#endif +}; + +// Individual Hamiltonian-term writers. write_h_t/vnl/vl/vh/vxc share a common +// build -> write H(k) -> optionally write H(R) skeleton (see hterm_writer.cpp); +// the term-specific part is only how hR_tmp is filled. +void write_h_t(WriteHParams& params); + +void write_h_vnl(WriteHParams& params); + +void write_h_vl(WriteHParams& params); + +void write_h_vh(WriteHParams& params); + +void write_h_vxc(WriteHParams& params); + +#ifdef __EXX +// Build V^EXX(R) into a real HContainer via add_HexxR (from exd/exc->get_Hexxs()) and write it. +// exd (real Hexx) and exc (complex Hexx) are mutually exclusive; picked by info_ri.real_number. +void write_h_exx(WriteHParams& params, const Exx_Info& exx_info); +#endif + +} // namespace ModuleIO + +#endif diff --git a/source/source_io/module_hs/lat_r_csr.cpp b/source/source_io/module_hs/lat_r_csr.cpp new file mode 100644 index 00000000000..0d7d6d5ced9 --- /dev/null +++ b/source/source_io/module_hs/lat_r_csr.cpp @@ -0,0 +1,132 @@ +#include "lat_r_csr.h" +#include "source_base/parallel_reduce.h" +#include "source_base/global_function.h" +#include "source_base/global_variable.h" + +#include +#include +#include +#include +#include + +inline void write_data(std::ofstream& ofs, const double& data, const int precision) +{ + ofs << " " << std::fixed << std::scientific << std::setprecision(precision) << data; +} +inline void write_data(std::ofstream& ofs, const std::complex& data, const int precision) +{ + ofs << " (" << std::fixed << std::scientific << std::setprecision(precision) + << data.real() << "," << data.imag() << ")"; +} + +template +void ModuleIO::save_lat_r(std::ofstream& ofs, + const SparseRBlock& XR, + const Parallel_Orbitals& pv, + const SparseWriteOptions& options) +{ + const int nlocal = pv.get_global_row_size(); + if (nlocal <= 0) + { + ModuleBase::WARNING_QUIT("ModuleIO::save_lat_r", + "Parallel_Orbitals global row size must be positive."); + } + + std::vector indptr; + indptr.reserve(nlocal + 1); + indptr.push_back(0); + + std::vector col_indices; + + std::vector line(nlocal); + for(int row = 0; row < nlocal; ++row) + { + ModuleBase::GlobalFunc::ZEROS(line.data(), nlocal); + + if (!options.reduce || pv.global2local_row(row) >= 0) + { + auto iter = XR.find(row); + if (iter != XR.end()) + { + for (auto &value : iter->second) + { + if (value.first >= static_cast(nlocal)) + { + std::cerr << "Sparse column index out of range." << std::endl; + ModuleBase::WARNING_QUIT("ModuleIO::save_lat_r", + "Sparse column index out of range."); + } + line[value.first] = value.second; + } + } + } + + if (options.reduce) + { + Parallel_Reduce::reduce_all(line.data(), nlocal); + } + + if (!options.reduce || GlobalV::DRANK == 0) + { + long long nonzeros_count = 0; + for (int col = 0; col < nlocal; ++col) + { + if (std::abs(line[col]) > options.threshold) + { + if (options.binary) + { + ofs.write(reinterpret_cast(&line[col]), sizeof(T)); + } + else + { + write_data(ofs, line[col], options.precision); + } + col_indices.push_back(col); + nonzeros_count++; + } + + } + nonzeros_count += indptr.back(); + indptr.push_back(nonzeros_count); + } + } + + if (!options.reduce || GlobalV::DRANK == 0) + { + if (options.binary) + { + for (int col : col_indices) + { + ofs.write(reinterpret_cast(&col), sizeof(int)); + } + for (auto &i : indptr) + { + ofs.write(reinterpret_cast(&i), sizeof(long long)); + } + } + else + { + ofs << std::endl; + for (int col : col_indices) + { + ofs << " " << col; + } + ofs << std::endl; + for (auto &i : indptr) + { + ofs << " " << i; + } + ofs << std::endl; + } + } +} + +template void ModuleIO::save_lat_r(std::ofstream& ofs, + const SparseRBlock& XR, + const Parallel_Orbitals& pv, + const SparseWriteOptions& options); + +template void ModuleIO::save_lat_r>(std::ofstream& ofs, + const SparseRBlock>& XR, + const Parallel_Orbitals& pv, + const SparseWriteOptions& options); diff --git a/source/source_io/module_hs/lat_r_csr.h b/source/source_io/module_hs/lat_r_csr.h new file mode 100644 index 00000000000..9ff8136e0d3 --- /dev/null +++ b/source/source_io/module_hs/lat_r_csr.h @@ -0,0 +1,17 @@ +#ifndef LAT_R_CSR_H +#define LAT_R_CSR_H + +#include "hs_sparse_io.h" + +#include + +namespace ModuleIO +{ + template + void save_lat_r(std::ofstream& ofs, + const SparseRBlock& XR, + const Parallel_Orbitals& pv, + const SparseWriteOptions& options); +} + +#endif diff --git a/source/source_io/module_hs/output_mat_sparse.cpp b/source/source_io/module_hs/output_mat_sparse.cpp index 269ea282fb9..d8d737d8956 100644 --- a/source/source_io/module_hs/output_mat_sparse.cpp +++ b/source/source_io/module_hs/output_mat_sparse.cpp @@ -1,7 +1,8 @@ #include "output_mat_sparse.h" -#include "cal_r_overlap_r.h" -#include "source_io/module_hs/write_hs_r.h" +#include "pos_op_mat.h" +#include "source_io/module_hs/hsr_legacy.h" +#include "source_io/module_parameter/parameter.h" namespace ModuleIO { @@ -20,9 +21,29 @@ void output_mat_sparse(const MatSparseOutputOptions& options, { LCAO_HS_Arrays HS_Arrays; // store sparse arrays + const std::string& global_out_dir = PARAM.globalv.global_out_dir; + const std::string& global_matrix_dir = PARAM.globalv.global_matrix_dir; + const std::string& calculation = PARAM.inp.calculation; + const bool out_app_flag = PARAM.inp.out_app_flag; + const int nspin = PARAM.inp.nspin; + const bool gamma_only_local = PARAM.globalv.gamma_only_local; + const int npol = PARAM.globalv.npol; + const int nlocal = PARAM.globalv.nlocal; + + MatROutputOptions mat_R_options; + mat_R_options.binary = options.binary; + mat_R_options.sparse_threshold = options.sparse_threshold; + mat_R_options.global_out_dir = global_out_dir; + mat_R_options.global_matrix_dir = global_matrix_dir; + mat_R_options.calculation = calculation; + mat_R_options.out_app_flag = out_app_flag; + mat_R_options.nspin = nspin; + //! generate a file containing the kinetic energy matrix if (options.out_mat_t) { + const std::string tr_filename = "tr_nao.csr"; + mat_R_options.precision = options.t_precision; output_TR(istep, ucell, pv, @@ -30,15 +51,14 @@ void output_mat_sparse(const MatSparseOutputOptions& options, grid, two_center_bundle, orb, - "trs1_nao.csr", - options.binary, - options.sparse_threshold, - options.t_precision); + tr_filename, + mat_R_options); } //! generate a file containing the derivatives of the Hamiltonian matrix (in Ry/Bohr) if (options.out_mat_dh) { + mat_R_options.precision = options.dh_precision; output_dHR(istep, v_eff, ucell, @@ -47,14 +67,15 @@ void output_mat_sparse(const MatSparseOutputOptions& options, grid, two_center_bundle, orb, - kv, - options.binary, - options.sparse_threshold, - options.dh_precision); + mat_R_options, + gamma_only_local, + npol, + nlocal); } //! generate a file containing the derivatives of the overlap matrix (in Ry/Bohr) if (options.out_mat_ds) { + mat_R_options.precision = options.ds_precision; output_dSR(istep, ucell, pv, @@ -62,89 +83,35 @@ void output_mat_sparse(const MatSparseOutputOptions& options, grid, two_center_bundle, orb, - kv, - options.binary, - options.sparse_threshold, - options.ds_precision); + mat_R_options, + gamma_only_local, + npol, + nlocal); } // add by jingan for out r_R matrix 2019.8.14 if (options.out_mat_r) { - cal_r_overlap_R r_matrix; + Position_op r_matrix; r_matrix.binary = options.binary; r_matrix.sparse_threshold = options.sparse_threshold; - r_matrix.init(ucell, pv, orb); - r_matrix.out_rR(ucell, grid, istep, options.r_precision); + const bool cal_force = PARAM.inp.cal_force; + r_matrix.init(ucell, pv, orb, cal_force, nlocal); + r_matrix.out_rR(ucell, + grid, + istep, + options.r_precision, + global_out_dir, + global_matrix_dir, + calculation, + out_app_flag, + nlocal, + npol); } return; } -template -void output_mat_sparse(const bool& out_mat_dh, - const bool& out_mat_ds, - const bool& out_mat_t, - const bool& out_mat_r, - const int& istep, - const ModuleBase::matrix& v_eff, - const Parallel_Orbitals& pv, - const TwoCenterBundle& two_center_bundle, - const LCAO_Orbitals& orb, - UnitCell& ucell, - const Grid_Driver& grid, - const K_Vectors& kv, - hamilt::Hamilt* p_ham, - Plus_U_Base* p_dftu) -{ - MatSparseOutputOptions options; - options.out_mat_dh = out_mat_dh; - options.out_mat_ds = out_mat_ds; - options.out_mat_t = out_mat_t; - options.out_mat_r = out_mat_r; - output_mat_sparse(options, - istep, - v_eff, - pv, - two_center_bundle, - orb, - ucell, - grid, - kv, - p_ham, - p_dftu); -} - -template void output_mat_sparse(const bool& out_mat_dh, - const bool& out_mat_ds, - const bool& out_mat_t, - const bool& out_mat_r, - const int& istep, - const ModuleBase::matrix& v_eff, - const Parallel_Orbitals& pv, - const TwoCenterBundle& two_center_bundle, - const LCAO_Orbitals& orb, - UnitCell& ucell, - const Grid_Driver& grid, - const K_Vectors& kv, - hamilt::Hamilt* p_ham, - Plus_U_Base* p_dftu); - -template void output_mat_sparse>(const bool& out_mat_dh, - const bool& out_mat_ds, - const bool& out_mat_t, - const bool& out_mat_r, - const int& istep, - const ModuleBase::matrix& v_eff, - const Parallel_Orbitals& pv, - const TwoCenterBundle& two_center_bundle, - const LCAO_Orbitals& orb, - UnitCell& ucell, - const Grid_Driver& grid, - const K_Vectors& kv, - hamilt::Hamilt>* p_ham, - Plus_U_Base* p_dftu); - template void output_mat_sparse(const MatSparseOutputOptions& options, const int& istep, const ModuleBase::matrix& v_eff, diff --git a/source/source_io/module_hs/output_mat_sparse.h b/source/source_io/module_hs/output_mat_sparse.h index 44d44826b81..2d0b698c8bb 100644 --- a/source/source_io/module_hs/output_mat_sparse.h +++ b/source/source_io/module_hs/output_mat_sparse.h @@ -37,23 +37,6 @@ void output_mat_sparse(const MatSparseOutputOptions& options, const K_Vectors& kv, hamilt::Hamilt* p_ham, Plus_U_Base* p_dftu); - -/// @brief legacy bool-only interface kept for source compatibility -template -void output_mat_sparse(const bool& out_mat_dh, - const bool& out_mat_ds, - const bool& out_mat_t, - const bool& out_mat_r, - const int& istep, - const ModuleBase::matrix& v_eff, - const Parallel_Orbitals& pv, - const TwoCenterBundle& two_center_bundle, - const LCAO_Orbitals& orb, - UnitCell& ucell, - const Grid_Driver& grid, - const K_Vectors& kv, - hamilt::Hamilt* p_ham, - Plus_U_Base* p_dftu); } // namespace ModuleIO #endif // OUTPUT_MAT_SPARSE_H diff --git a/source/source_io/module_hs/pos_op_basis.cpp b/source/source_io/module_hs/pos_op_basis.cpp new file mode 100644 index 00000000000..62e63860de2 --- /dev/null +++ b/source/source_io/module_hs/pos_op_basis.cpp @@ -0,0 +1,399 @@ +#include "pos_op_basis.h" + +#include "source_base/mathzone_add1.h" +#include "source_base/timer.h" +#include "source_base/tool_quit.h" +#include "source_cell/nonlocal_info_base.h" + +PosOpBasis::PosOpBasis() +{ +} + +PosOpBasis::~PosOpBasis() +{ +} + +void PosOpBasis::setup_tables(const LCAO_Orbitals& orb) +{ + const int ntype = orb.get_ntype(); + int lmax_orb = -1; + for (int it = 0; it < ntype; it++) + { + lmax_orb = std::max(lmax_orb, orb.Phi[it].getLmax()); + } + const double dr = orb.get_dR(); + const double dk = orb.get_dk(); + const int kmesh = orb.get_kmesh() * 4 + 1; + int Rmesh = static_cast(orb.get_Rmax() / dr) + 4; + Rmesh += 1 - Rmesh % 2; + + const int Lmax = lmax_orb + 1; + const int Lmax_used = 2 * lmax_orb + 1; + Center2_Orb::init_Table_Spherical_Bessel(Lmax_used, dr, dk, kmesh, Rmesh, psb_); + ModuleBase::Ylm::set_coefficients(); + MGT.init_Gaunt_CH(Lmax); + MGT.init_Gaunt(Lmax); +} + +void PosOpBasis::build_orbs(const UnitCell& ucell, const LCAO_Orbitals& orb, bool cal_force) +{ + int orb_r_ntype = 0; + int mat_Nr = orb.Phi[0].PhiLN(0, 0).getNr(); + int count_Nr = 0; + + orbs.resize(orb.get_ntype()); + for (int T = 0; T < orb.get_ntype(); ++T) + { + count_Nr = orb.Phi[T].PhiLN(0, 0).getNr(); + if (count_Nr > mat_Nr) + { + mat_Nr = count_Nr; + orb_r_ntype = T; + } + + orbs[T].resize(orb.Phi[T].getLmax() + 1); + for (int L = 0; L <= orb.Phi[T].getLmax(); ++L) + { + orbs[T][L].resize(orb.Phi[T].getNchi(L)); + for (int N = 0; N < orb.Phi[T].getNchi(L); ++N) + { + const auto& orb_origin = orb.Phi[T].PhiLN(L, N); + orbs[T][L][N].set_orbital_info(orb_origin.getLabel(), + orb_origin.getType(), + orb_origin.getL(), + orb_origin.getChi(), + orb_origin.getNr(), + orb_origin.getRab(), + orb_origin.getRadial(), + Numerical_Orbital_Lm::Psi_Type::Psi, + orb_origin.getPsi(), + static_cast(orb_origin.getNk() * 4) | 1, + orb_origin.getDk(), + orb_origin.getDruniform(), + false, + true, + cal_force); + } + } + } + + orb_r.set_orbital_info(orbs[orb_r_ntype][0][0].getLabel(), // atom label + orb_r_ntype, // atom type + 1, // angular momentum L + 1, // number of orbitals of this L , just N + orbs[orb_r_ntype][0][0].getNr(), // number of radial mesh + orbs[orb_r_ntype][0][0].getRab(), // the mesh interval in radial mesh + orbs[orb_r_ntype][0][0].getRadial(), // radial mesh value(a.u.) + Numerical_Orbital_Lm::Psi_Type::Psi, + orbs[orb_r_ntype][0][0].getRadial(), // radial wave function + orbs[orb_r_ntype][0][0].getNk(), + orbs[orb_r_ntype][0][0].getDk(), + orbs[orb_r_ntype][0][0].getDruniform(), + false, + true, + cal_force); +} + +void PosOpBasis::build_nonlocal_orbs(const UnitCell& ucell, const LCAO_Orbitals& orb, bool cal_force) +{ + const NonlocalInfoBase& infoNL_ = *ucell.infoNL; + + orbs_nonlocal.resize(orb.get_ntype()); + for (int T = 0; T < orb.get_ntype(); ++T) + { + const int nproj = infoNL_.get_nproj(T); + orbs_nonlocal[T].resize(nproj); + for (int ip = 0; ip < nproj; ip++) + { + int nr = infoNL_.get_proj_Nr(T, ip); + double dr_uniform = 0.01; + int nr_uniform + = static_cast((infoNL_.get_proj_radial(T, ip)[nr - 1] - infoNL_.get_proj_radial(T, ip)[0]) / dr_uniform) + 1; + double* rad = new double[nr_uniform]; + double* rab = new double[nr_uniform]; + for (int ir = 0; ir < nr_uniform; ir++) + { + rad[ir] = ir * dr_uniform; + rab[ir] = dr_uniform; + } + double* y2 = new double[nr]; + double* Beta_r_uniform = new double[nr_uniform]; + double* dbeta_uniform = new double[nr_uniform]; + ModuleBase::Mathzone_Add1::SplineD2(infoNL_.get_proj_radial(T, ip), + infoNL_.get_proj_beta_r(T, ip), + nr, + 0.0, + 0.0, + y2); + ModuleBase::Mathzone_Add1::Cubic_Spline_Interpolation(infoNL_.get_proj_radial(T, ip), + infoNL_.get_proj_beta_r(T, ip), + y2, + nr, + rad, + nr_uniform, + Beta_r_uniform, + dbeta_uniform); + + // linear extrapolation at the zero point + if (infoNL_.get_proj_radial(T, ip)[0] > 1e-10) + { + double slope = (infoNL_.get_proj_beta_r(T, ip)[1] - infoNL_.get_proj_beta_r(T, ip)[0]) + / (infoNL_.get_proj_radial(T, ip)[1] - infoNL_.get_proj_radial(T, ip)[0]); + Beta_r_uniform[0] = infoNL_.get_proj_beta_r(T, ip)[0] - slope * infoNL_.get_proj_radial(T, ip)[0]; + } + + // Here, the operation beta_r / r is performed. To avoid divergence at r=0, beta_r(0) is set to beta_r(1). + // However, this may introduce issues, so caution is needed. + for (int ir = 1; ir < nr_uniform; ir++) + { + Beta_r_uniform[ir] = Beta_r_uniform[ir] / rad[ir]; + } + Beta_r_uniform[0] = Beta_r_uniform[1]; + + orbs_nonlocal[T][ip].set_orbital_info(infoNL_.get_label(T), + infoNL_.get_type(T), + infoNL_.get_proj_L(T, ip), + 1, + nr_uniform, + rab, + rad, + Numerical_Orbital_Lm::Psi_Type::Psi, + Beta_r_uniform, + static_cast(infoNL_.get_proj_Nk(T, ip) * 4) | 1, + infoNL_.get_proj_dk(T, ip), + infoNL_.get_proj_dr_uniform(T, ip), + false, + true, + cal_force); + + delete[] rad; + delete[] rab; + delete[] y2; + delete[] Beta_r_uniform; + delete[] dbeta_uniform; + } + } +} + +void PosOpBasis::build_iw_map(const UnitCell& ucell, int nlocal) +{ + int map_size = nlocal; + int required_orbitals = 0; + for (int it = 0; it < ucell.ntype; ++it) + { + required_orbitals += ucell.atoms[it].nw * ucell.atoms[it].na; + } + map_size = std::max(map_size, required_orbitals); + + iw2it.resize(map_size); + iw2ia.resize(map_size); + iw2iL.resize(map_size); + iw2iN.resize(map_size); + iw2im.resize(map_size); + + int iw = 0; + for (int it = 0; it < ucell.ntype; it++) + { + for (int ia = 0; ia < ucell.atoms[it].na; ia++) + { + for (int iL = 0; iL < ucell.atoms[it].nwl + 1; iL++) + { + for (int iN = 0; iN < ucell.atoms[it].l_nchi[iL]; iN++) + { + for (int im = 0; im < (2 * iL + 1); im++) + { + iw2it[iw] = it; + iw2ia[iw] = ia; + iw2iL[iw] = iL; + iw2iN[iw] = iN; + iw2im[iw] = im; + iw++; + } + } + } + } + } +} + +void PosOpBasis::build(const UnitCell& ucell, const LCAO_Orbitals& orb, bool cal_force, int nlocal) +{ + ModuleBase::TITLE("PosOpBasis", "build"); + ModuleBase::timer::start("PosOpBasis", "build"); + + setup_tables(orb); + build_orbs(ucell, orb, cal_force); + + // build center2_orb11 and center2_orb21_r tables + for (int TA = 0; TA < orb.get_ntype(); ++TA) + { + for (int TB = 0; TB < orb.get_ntype(); ++TB) + { + for (int LA = 0; LA <= orb.Phi[TA].getLmax(); ++LA) + { + for (int NA = 0; NA < orb.Phi[TA].getNchi(LA); ++NA) + { + for (int LB = 0; LB <= orb.Phi[TB].getLmax(); ++LB) + { + for (int NB = 0; NB < orb.Phi[TB].getNchi(LB); ++NB) + { + center2_orb11[TA][TB][LA][NA][LB].insert( + std::make_pair(NB, Center2_Orb::Orb11(orbs[TA][LA][NA], orbs[TB][LB][NB], psb_, MGT))); + } + } + } + } + } + } + + for (int TA = 0; TA < orb.get_ntype(); ++TA) + { + for (int TB = 0; TB < orb.get_ntype(); ++TB) + { + for (int LA = 0; LA <= orb.Phi[TA].getLmax(); ++LA) + { + for (int NA = 0; NA < orb.Phi[TA].getNchi(LA); ++NA) + { + for (int LB = 0; LB <= orb.Phi[TB].getLmax(); ++LB) + { + for (int NB = 0; NB < orb.Phi[TB].getNchi(LB); ++NB) + { + center2_orb21_r[TA][TB][LA][NA][LB].insert( + std::make_pair(NB, Center2_Orb::Orb21(orbs[TA][LA][NA], orb_r, orbs[TB][LB][NB], psb_, MGT))); + } + } + } + } + } + } + + for (auto& co1: center2_orb11) + { + for (auto& co2: co1.second) + { + for (auto& co3: co2.second) + { + for (auto& co4: co3.second) + { + for (auto& co5: co4.second) + { + for (auto& co6: co5.second) + { + co6.second.init_radial_table(); + } + } + } + } + } + } + + for (auto& co1: center2_orb21_r) + { + for (auto& co2: co1.second) + { + for (auto& co3: co2.second) + { + for (auto& co4: co3.second) + { + for (auto& co5: co4.second) + { + for (auto& co6: co5.second) + { + co6.second.init_radial_table(); + } + } + } + } + } + } + + build_iw_map(ucell, nlocal); + + ModuleBase::timer::end("PosOpBasis", "build"); +} + +void PosOpBasis::build_nonlocal(const UnitCell& ucell, const LCAO_Orbitals& orb, bool cal_force, int nlocal) +{ + ModuleBase::TITLE("PosOpBasis", "build_nonlocal"); + ModuleBase::timer::start("PosOpBasis", "build_nonlocal"); + + setup_tables(orb); + build_orbs(ucell, orb, cal_force); + build_nonlocal_orbs(ucell, orb, cal_force); + + const NonlocalInfoBase& infoNL_ = *ucell.infoNL; + + // build center2_orb11_nonlocal and center2_orb21_r_nonlocal tables + for (int TA = 0; TA < orb.get_ntype(); ++TA) + { + for (int TB = 0; TB < orb.get_ntype(); ++TB) + { + for (int LA = 0; LA <= orb.Phi[TA].getLmax(); ++LA) + { + for (int NA = 0; NA < orb.Phi[TA].getNchi(LA); ++NA) + { + for (int ip = 0; ip < infoNL_.get_nproj(TB); ip++) + { + center2_orb11_nonlocal[TA][TB][LA][NA].insert( + std::make_pair(ip, Center2_Orb::Orb11(orbs[TA][LA][NA], orbs_nonlocal[TB][ip], psb_, MGT))); + } + } + } + } + } + + for (int TA = 0; TA < orb.get_ntype(); ++TA) + { + for (int TB = 0; TB < orb.get_ntype(); ++TB) + { + for (int LA = 0; LA <= orb.Phi[TA].getLmax(); ++LA) + { + for (int NA = 0; NA < orb.Phi[TA].getNchi(LA); ++NA) + { + for (int ip = 0; ip < infoNL_.get_nproj(TB); ip++) + { + center2_orb21_r_nonlocal[TA][TB][LA][NA].insert( + std::make_pair(ip, Center2_Orb::Orb21(orbs[TA][LA][NA], orb_r, orbs_nonlocal[TB][ip], psb_, MGT))); + } + } + } + } + } + + for (auto& co1: center2_orb11_nonlocal) + { + for (auto& co2: co1.second) + { + for (auto& co3: co2.second) + { + for (auto& co4: co3.second) + { + for (auto& co5: co4.second) + { + co5.second.init_radial_table(); + } + } + } + } + } + + for (auto& co1: center2_orb21_r_nonlocal) + { + for (auto& co2: co1.second) + { + for (auto& co3: co2.second) + { + for (auto& co4: co3.second) + { + for (auto& co5: co4.second) + { + co5.second.init_radial_table(); + } + } + } + } + } + + build_iw_map(ucell, nlocal); + + ModuleBase::timer::end("PosOpBasis", "build_nonlocal"); +} diff --git a/source/source_io/module_hs/pos_op_basis.h b/source/source_io/module_hs/pos_op_basis.h new file mode 100644 index 00000000000..fa6a10520f2 --- /dev/null +++ b/source/source_io/module_hs/pos_op_basis.h @@ -0,0 +1,102 @@ +#ifndef POS_OP_BASIS_H +#define POS_OP_BASIS_H + +#include "source_base/sph_bessel_recursive.h" +#include "source_base/vector3.h" +#include "source_base/ylm.h" +#include "source_basis/module_ao/orb_atomic_lm.h" +#include "source_basis/module_ao/orb_gaunt_table.h" +#include "source_basis/module_ao/orb_read.h" +#include "source_cell/unitcell.h" +#include "source_lcao/center2orb_orb11.h" +#include "source_lcao/center2orb_orb21.h" +#include "source_lcao/center2orb.h" + +#include +#include + +/** + * @brief Build and hold the orbital basis and two-center integral tables + * needed to evaluate position-operator matrix elements + * . + * + * The class prepares the numerical atomic orbitals, the auxiliary r-orbital, + * the non-local projectors (when needed), the spherical-Bessel/Gaunt tables, + * and the Center2_Orb integral tables. It also builds the iw2* index maps + * that translate a global orbital index into (type, atom, L, N, m). + */ +class PosOpBasis +{ + public: + PosOpBasis(); + ~PosOpBasis(); + + void build(const UnitCell& ucell, const LCAO_Orbitals& orb, bool cal_force, int nlocal); + void build_nonlocal(const UnitCell& ucell, const LCAO_Orbitals& orb, bool cal_force, int nlocal); + + // const accessors for PosOpCalc / PosOpWriter + const Numerical_Orbital_Lm& get_orb_r() const { return orb_r; } + + const std::map>>>>>& + get_center2_orb11() const + { + return center2_orb11; + } + + const std::map>>>>>& + get_center2_orb21_r() const + { + return center2_orb21_r; + } + + const std::map>>>>& + get_center2_orb11_nonlocal() const + { + return center2_orb11_nonlocal; + } + + const std::map>>>>& + get_center2_orb21_r_nonlocal() const + { + return center2_orb21_r_nonlocal; + } + + int get_iw2it(int iw) const { return iw2it[iw]; } + int get_iw2ia(int iw) const { return iw2ia[iw]; } + int get_iw2iL(int iw) const { return iw2iL[iw]; } + int get_iw2iN(int iw) const { return iw2iN[iw]; } + int get_iw2im(int iw) const { return iw2im[iw]; } + + private: + void setup_tables(const LCAO_Orbitals& orb); + void build_orbs(const UnitCell& ucell, const LCAO_Orbitals& orb, bool cal_force); + void build_nonlocal_orbs(const UnitCell& ucell, const LCAO_Orbitals& orb, bool cal_force); + void build_iw_map(const UnitCell& ucell, int nlocal); + + std::vector iw2ia; + std::vector iw2iL; + std::vector iw2im; + std::vector iw2iN; + std::vector iw2it; + + ModuleBase::Sph_Bessel_Recursive::D2* psb_ = nullptr; + ORB_gaunt_table MGT; + + Numerical_Orbital_Lm orb_r; + std::vector>> orbs; + std::vector> orbs_nonlocal; + + std::map>>>>> + center2_orb11; + + std::map>>>>> + center2_orb21_r; + + std::map>>>> + center2_orb11_nonlocal; + + std::map>>>> + center2_orb21_r_nonlocal; +}; + +#endif // POS_OP_BASIS_H diff --git a/source/source_io/module_hs/pos_op_calc.cpp b/source/source_io/module_hs/pos_op_calc.cpp new file mode 100644 index 00000000000..072439f0863 --- /dev/null +++ b/source/source_io/module_hs/pos_op_calc.cpp @@ -0,0 +1,154 @@ +#include "pos_op_calc.h" + +#include "source_base/timer.h" +#include "source_base/ylm.h" + +PosOpCalc::PosOpCalc(const PosOpBasis& basis) : basis_(basis) +{ +} + +ModuleBase::Vector3 PosOpCalc::pos_matrix(const ModuleBase::Vector3& R1, + int T1, int L1, int m1, int N1, + const ModuleBase::Vector3& R2, + int T2, int L2, int m2, int N2) const +{ + ModuleBase::Vector3 origin_point(0.0, 0.0, 0.0); + double factor = sqrt(ModuleBase::FOUR_PI / 3.0); + const ModuleBase::Vector3& distance = R2 - R1; + + double overlap_o = basis_.get_center2_orb11().at(T1).at(T2).at(L1).at(N1).at(L2).at(N2).cal_overlap(origin_point, + distance, + m1, + m2); + + double overlap_x = -1 * factor * basis_.get_center2_orb21_r().at(T1).at(T2).at(L1).at(N1).at(L2).at(N2).cal_overlap(origin_point, + distance, + m1, + 1, + m2); // m = 1 + + double overlap_y = -1 * factor * basis_.get_center2_orb21_r().at(T1).at(T2).at(L1).at(N1).at(L2).at(N2).cal_overlap(origin_point, + distance, + m1, + 2, + m2); // m = -1 + + double overlap_z = factor * basis_.get_center2_orb21_r().at(T1).at(T2).at(L1).at(N1).at(L2).at(N2).cal_overlap(origin_point, + distance, + m1, + 0, + m2); // m = 0 + + ModuleBase::Vector3 temp_prp = ModuleBase::Vector3(overlap_x, overlap_y, overlap_z) + R1 * overlap_o; + + return temp_prp; +} + +ModuleBase::Vector3 PosOpCalc::pos_grad_matrix(const ModuleBase::Vector3& R1, + int T1, int L1, int m1, int N1, + const ModuleBase::Vector3& R2, + int T2, int L2, int m2, int N2, + const ModuleBase::Vector3& Efield, + const ModuleBase::Vector3& dR) const +{ + ModuleBase::Vector3 origin_point(0.0, 0.0, 0.0); + double factor = sqrt(ModuleBase::FOUR_PI / 3.0); + const ModuleBase::Vector3& distance = R2 - R1; + + ModuleBase::Vector3 grad_o = basis_.get_center2_orb11().at(T1).at(T2).at(L1).at(N1).at(L2).at(N2).cal_grad_overlap(origin_point, distance, m1, m2); + + ModuleBase::Vector3 grad_rx = -1 * factor + * basis_.get_center2_orb21_r().at(T1).at(T2).at(L1).at(N1).at(L2).at(N2).cal_grad_overlap(origin_point, + distance, + m1, + 1, + m2); // m = 1 + + ModuleBase::Vector3 grad_ry = -1 * factor + * basis_.get_center2_orb21_r().at(T1).at(T2).at(L1).at(N1).at(L2).at(N2).cal_grad_overlap(origin_point, + distance, + m1, + 2, + m2); // m = -1 + + ModuleBase::Vector3 grad_rz = factor + * basis_.get_center2_orb21_r().at(T1).at(T2).at(L1).at(N1).at(L2).at(N2).cal_grad_overlap(origin_point, + distance, + m1, + 0, + m2); // m = 0 + + ModuleBase::Vector3 temp_prp = Efield[0] * grad_rx + Efield[1] * grad_ry + Efield[2] * grad_rz + (Efield * (R1 - dR)) * grad_o; + + return temp_prp; +} + +void PosOpCalc::pos_beta_matrix(const UnitCell& ucell, + std::vector>& nlm, + const ModuleBase::Vector3& R1, + int T1, int L1, int m1, int N1, + const ModuleBase::Vector3& R2, + int T2) const +{ + ModuleBase::Vector3 origin_point(0.0, 0.0, 0.0); + double factor = sqrt(ModuleBase::FOUR_PI / 3.0); + const ModuleBase::Vector3& distance = R2 - R1; + const NonlocalInfoBase& infoNL_ = *ucell.infoNL; + const int nproj = infoNL_.get_nproj(T2); + nlm.resize(4); + if (nproj == 0) + { + for (int i = 0; i < 4; i++) + { + nlm[i].resize(1); + } + return; + } + + int natomwfc = 0; + for (int ip = 0; ip < nproj; ip++) + { + const int L2 = infoNL_.get_proj_L(T2, ip); // mohan add 2021-05-07 + natomwfc += 2 * L2 + 1; + } + for (int i = 0; i < 4; i++) + { + nlm[i].resize(natomwfc); + } + int index = 0; + for (int ip = 0; ip < nproj; ip++) + { + const int L2 = infoNL_.get_proj_L(T2, ip); + for (int m2 = 0; m2 < 2 * L2 + 1; m2++) + { + double overlap_o = basis_.get_center2_orb11_nonlocal().at(T1).at(T2).at(L1).at(N1).at(ip).cal_overlap(origin_point, distance, m1, m2); + + double overlap_x = -1 * factor + * basis_.get_center2_orb21_r_nonlocal().at(T1).at(T2).at(L1).at(N1).at(ip).cal_overlap(origin_point, + distance, + m1, + 1, + m2); // m = 1 + + double overlap_y = -1 * factor + * basis_.get_center2_orb21_r_nonlocal().at(T1).at(T2).at(L1).at(N1).at(ip).cal_overlap(origin_point, + distance, + m1, + 2, + m2); // m = -1 + + double overlap_z = factor + * basis_.get_center2_orb21_r_nonlocal().at(T1).at(T2).at(L1).at(N1).at(ip).cal_overlap(origin_point, + distance, + m1, + 0, + m2); // m = 0 + + nlm[0][index] = overlap_o; + nlm[1][index] = overlap_x + (R1 * overlap_o).x; + nlm[2][index] = overlap_y + (R1 * overlap_o).y; + nlm[3][index] = overlap_z + (R1 * overlap_o).z; + index++; + } + } +} diff --git a/source/source_io/module_hs/pos_op_calc.h b/source/source_io/module_hs/pos_op_calc.h new file mode 100644 index 00000000000..d32a65b2bf2 --- /dev/null +++ b/source/source_io/module_hs/pos_op_calc.h @@ -0,0 +1,55 @@ +#ifndef POS_OP_CALC_H +#define POS_OP_CALC_H + +#include "source_base/vector3.h" +#include "source_cell/unitcell.h" +#include "pos_op_basis.h" + +#include + +/** + * @brief Evaluate position-operator matrix elements using the tables + * prepared by PosOpBasis. + * + * All methods are pure numerical integrations; they do not perform I/O. + */ +class PosOpCalc +{ + public: + explicit PosOpCalc(const PosOpBasis& basis); + + /** + * @brief for a pair of numerical atomic orbitals. + * Returns the vector (x, y, z) components. + */ + ModuleBase::Vector3 pos_matrix(const ModuleBase::Vector3& R1, + int T1, int L1, int m1, int N1, + const ModuleBase::Vector3& R2, + int T2, int L2, int m2, int N2) const; + + /** + * @brief Electric-field coupling term * Efield. + */ + ModuleBase::Vector3 pos_grad_matrix(const ModuleBase::Vector3& R1, + int T1, int L1, int m1, int N1, + const ModuleBase::Vector3& R2, + int T2, int L2, int m2, int N2, + const ModuleBase::Vector3& Efield, + const ModuleBase::Vector3& dR) const; + + /** + * @brief for non-local projectors. + * nlm[0] = overlap, nlm[1..3] = position matrix elements. + */ + void pos_beta_matrix(const UnitCell& ucell, + std::vector>& nlm, + const ModuleBase::Vector3& R1, + int T1, int L1, int m1, int N1, + const ModuleBase::Vector3& R2, + int T2) const; + + private: + const PosOpBasis& basis_; +}; + +#endif // POS_OP_CALC_H diff --git a/source/source_io/module_hs/pos_op_csr.cpp b/source/source_io/module_hs/pos_op_csr.cpp new file mode 100644 index 00000000000..a43490dc397 --- /dev/null +++ b/source/source_io/module_hs/pos_op_csr.cpp @@ -0,0 +1,64 @@ +#include "pos_op_csr.h" + +#include "source_base/tool_quit.h" + +#include + +namespace ModuleIO +{ +namespace detail +{ +bool lat_r_nonempty(const int nonzero_num[3]) +{ + return nonzero_num[0] != 0 || nonzero_num[1] != 0 || nonzero_num[2] != 0; +} + +void assemble_csr(const std::string& output_filename, + const std::string& payload_filename, + const int step, + const int nlocal, + const int output_R_number, + const bool binary, + const bool append, + const std::string& context) +{ + std::ios_base::openmode output_mode = std::ios::out; + std::ios_base::openmode payload_mode = std::ios::in; + if (binary) + { + output_mode |= std::ios::binary; + payload_mode |= std::ios::binary; + } + if (append) + { + output_mode |= std::ios::app; + } + + std::ofstream out_r(output_filename.c_str(), output_mode); + if (!out_r.is_open()) + { + ModuleBase::WARNING_QUIT(context, "Cannot open r(R) output file: " + output_filename); + } + + if (binary) + { + out_r.write(reinterpret_cast(&step), sizeof(int)); + out_r.write(reinterpret_cast(&nlocal), sizeof(int)); + out_r.write(reinterpret_cast(&output_R_number), sizeof(int)); + } + else + { + out_r << "STEP: " << step << std::endl; + out_r << "Matrix Dimension of r(R): " << nlocal << std::endl; + out_r << "Matrix number of r(R): " << output_R_number << std::endl; + } + + std::ifstream payload(payload_filename.c_str(), payload_mode); + if (!payload.is_open()) + { + ModuleBase::WARNING_QUIT(context, "Cannot read temporary sparse matrix file: " + payload_filename); + } + out_r << payload.rdbuf(); +} +} // namespace detail +} // namespace ModuleIO diff --git a/source/source_io/module_hs/pos_op_csr.h b/source/source_io/module_hs/pos_op_csr.h new file mode 100644 index 00000000000..15dd9a0f762 --- /dev/null +++ b/source/source_io/module_hs/pos_op_csr.h @@ -0,0 +1,23 @@ +#ifndef POS_OP_CSR_H +#define POS_OP_CSR_H + +#include + +namespace ModuleIO +{ +namespace detail +{ +bool lat_r_nonempty(const int nonzero_num[3]); + +void assemble_csr(const std::string& output_filename, + const std::string& payload_filename, + int step, + int nlocal, + int output_R_number, + bool binary, + bool append, + const std::string& context); +} // namespace detail +} // namespace ModuleIO + +#endif diff --git a/source/source_io/module_hs/pos_op_mat.cpp b/source/source_io/module_hs/pos_op_mat.cpp new file mode 100644 index 00000000000..b02e30eb79c --- /dev/null +++ b/source/source_io/module_hs/pos_op_mat.cpp @@ -0,0 +1,113 @@ +#include "pos_op_mat.h" + +#include "pos_op_basis.h" +#include "pos_op_calc.h" +#include "pos_op_writer.h" +#include "source_base/timer.h" + +Position_op::Position_op() +{ +} + +Position_op::~Position_op() +{ +} + +void Position_op::init(const UnitCell& ucell, + const Parallel_Orbitals& pv, + const LCAO_Orbitals& orb, + const bool cal_force, + const int nlocal) +{ + basis_ = std::unique_ptr(new PosOpBasis()); + basis_->build(ucell, orb, cal_force, nlocal); + + calc_ = std::unique_ptr(new PosOpCalc(*basis_)); + writer_ = std::unique_ptr(new PosOpWriter(*basis_, *calc_, pv)); +} + +void Position_op::init_nonlocal(const UnitCell& ucell, + const Parallel_Orbitals& pv, + const LCAO_Orbitals& orb, + const bool cal_force, + const int nlocal) +{ + basis_ = std::unique_ptr(new PosOpBasis()); + basis_->build_nonlocal(ucell, orb, cal_force, nlocal); + + calc_ = std::unique_ptr(new PosOpCalc(*basis_)); + writer_ = std::unique_ptr(new PosOpWriter(*basis_, *calc_, pv)); +} + +ModuleBase::Vector3 Position_op::get_psi_r_psi(const ModuleBase::Vector3& R1, + const int& T1, + const int& L1, + const int& m1, + const int& N1, + const ModuleBase::Vector3& R2, + const int& T2, + const int& L2, + const int& m2, + const int& N2) +{ + return calc_->pos_matrix(R1, T1, L1, m1, N1, R2, T2, L2, m2, N2); +} + +ModuleBase::Vector3 Position_op::get_psi_r_gradpsi(const ModuleBase::Vector3& R1, + const int& T1, + const int& L1, + const int& m1, + const int& N1, + const ModuleBase::Vector3& R2, + const int& T2, + const int& L2, + const int& m2, + const int& N2, + const ModuleBase::Vector3& Efield, + const ModuleBase::Vector3& dR) +{ + return calc_->pos_grad_matrix(R1, T1, L1, m1, N1, R2, T2, L2, m2, N2, Efield, dR); +} + +void Position_op::get_psi_r_beta(const UnitCell& ucell, + std::vector>& nlm, + const ModuleBase::Vector3& R1, + const int& T1, + const int& L1, + const int& m1, + const int& N1, + const ModuleBase::Vector3& R2, + const int& T2) +{ + calc_->pos_beta_matrix(ucell, nlm, R1, T1, L1, m1, N1, R2, T2); +} + +void Position_op::out_rR(const UnitCell& ucell, + const Grid_Driver& gd, + const int& istep, + const int precision, + const std::string& global_out_dir, + const std::string& global_matrix_dir, + const std::string& calculation, + const bool out_app_flag, + const int nlocal, + const int npol) +{ + ModuleBase::TITLE("Position_op", "out_rR"); + ModuleBase::timer::start("Position_op", "out_rR"); + + writer_->out_lat_r(ucell, + gd, + istep, + precision, + global_out_dir, + global_matrix_dir, + calculation, + out_app_flag, + nlocal, + npol, + sparse_threshold, + binary); + + ModuleBase::timer::end("Position_op", "out_rR"); +} diff --git a/source/source_io/module_hs/pos_op_mat.h b/source/source_io/module_hs/pos_op_mat.h new file mode 100644 index 00000000000..63def4d9bec --- /dev/null +++ b/source/source_io/module_hs/pos_op_mat.h @@ -0,0 +1,88 @@ +#ifndef CAL_R_OVERLAP_R_H +#define CAL_R_OVERLAP_R_H + +#include "source_base/vector3.h" + +#include +#include +#include + +class UnitCell; +class Parallel_Orbitals; +class LCAO_Orbitals; +class Grid_Driver; +class PosOpBasis; +class PosOpCalc; +class PosOpWriter; + +// output r_R matrix, added by Jingan +// Facade that delegates to PosOpBasis / PosOpCalc / PosOpWriter. +class Position_op +{ + + public: + Position_op(); + ~Position_op(); + + double kmesh_times = 4; + double sparse_threshold = 1e-10; + bool binary = false; + + void init(const UnitCell& ucell, + const Parallel_Orbitals& pv, + const LCAO_Orbitals& orb, + const bool cal_force, + const int nlocal); + void init_nonlocal(const UnitCell& ucell, + const Parallel_Orbitals& pv, + const LCAO_Orbitals& orb, + const bool cal_force, + const int nlocal); + ModuleBase::Vector3 get_psi_r_psi(const ModuleBase::Vector3& R1, + const int& T1, + const int& L1, + const int& m1, + const int& N1, + const ModuleBase::Vector3& R2, + const int& T2, + const int& L2, + const int& m2, + const int& N2); + ModuleBase::Vector3 get_psi_r_gradpsi(const ModuleBase::Vector3& R1, + const int& T1, + const int& L1, + const int& m1, + const int& N1, + const ModuleBase::Vector3& R2, + const int& T2, + const int& L2, + const int& m2, + const int& N2, + const ModuleBase::Vector3& Efield, + const ModuleBase::Vector3& dR); + void get_psi_r_beta(const UnitCell& ucell, + std::vector>& nlm, + const ModuleBase::Vector3& R1, + const int& T1, + const int& L1, + const int& m1, + const int& N1, + const ModuleBase::Vector3& R2, + const int& T2); + void out_rR(const UnitCell& ucell, + const Grid_Driver& gd, + const int& istep, + const int precision, + const std::string& global_out_dir, + const std::string& global_matrix_dir, + const std::string& calculation, + const bool out_app_flag, + const int nlocal, + const int npol); + + private: + std::unique_ptr basis_; + std::unique_ptr calc_; + std::unique_ptr writer_; +}; +#endif diff --git a/source/source_io/module_hs/pos_op_writer.cpp b/source/source_io/module_hs/pos_op_writer.cpp new file mode 100644 index 00000000000..be1d1dc2a42 --- /dev/null +++ b/source/source_io/module_hs/pos_op_writer.cpp @@ -0,0 +1,245 @@ +#include "pos_op_writer.h" + +#include "source_base/global_variable.h" +#include "source_base/parallel_reduce.h" +#include "source_base/timer.h" +#include "source_io/module_hs/lat_r_csr.h" +#include "source_io/module_hs/pos_op_csr.h" +#include "source_cell/module_neighbor/sltk_grid_driver.h" + +#include +#include + +PosOpWriter::PosOpWriter(const PosOpBasis& basis, const PosOpCalc& calc, const Parallel_Orbitals& pv) + : basis_(basis), calc_(calc), pv_(pv) +{ +} + +void PosOpWriter::out_lat_r(const UnitCell& ucell, + const Grid_Driver& gd, + int istep, + int precision, + const std::string& global_out_dir, + const std::string& global_matrix_dir, + const std::string& calculation, + bool out_app_flag, + int nlocal, + int npol, + double sparse_threshold, + bool binary) +{ + ModuleBase::TITLE("PosOpWriter", "out_lat_r"); + ModuleBase::timer::start("PosOpWriter", "out_lat_r"); + + int step = istep; + // set R coor range + int R_minX = int(-gd.getGlayerX_minus()); + int R_minY = int(-gd.getGlayerY_minus()); + int R_minZ = int(-gd.getGlayerZ_minus()); + + int R_x = gd.getGlayerX() + gd.getGlayerX_minus(); + int R_y = gd.getGlayerY() + gd.getGlayerY_minus(); + int R_z = gd.getGlayerZ() + gd.getGlayerZ_minus(); + + std::set> all_R_coor; + for (int ix = 0; ix < R_x; ix++) + { + for (int iy = 0; iy < R_y; iy++) + { + for (int iz = 0; iz < R_z; iz++) + { + Abfs::Vector3_Order temp_R(ix + R_minX, iy + R_minY, iz + R_minZ); + all_R_coor.insert(temp_R); + } + } + } + + // calculate rR matrix + ModuleBase::Vector3 origin_point(0.0, 0.0, 0.0); + double factor = sqrt(ModuleBase::FOUR_PI / 3.0); + int output_R_number = 0; + ModuleIO::SparseWriteOptions lat_r_options; + lat_r_options.threshold = sparse_threshold; + lat_r_options.binary = binary; + lat_r_options.precision = precision; + lat_r_options.reduce = true; + + std::stringstream tem1; + tem1 << global_out_dir << "tmp-rr_nao.txt"; + std::ofstream ofs_tem1; + + if (GlobalV::DRANK == 0) + { + if (binary) + { + ofs_tem1.open(tem1.str().c_str(), std::ios::binary); + } + else + { + ofs_tem1.open(tem1.str().c_str()); + } + if (!ofs_tem1.is_open()) + { + ModuleBase::WARNING_QUIT("PosOpWriter::out_lat_r", "Cannot open temporary sparse matrix file: " + tem1.str()); + } + } + + for (auto& R_coor: all_R_coor) + { + std::map> psi_r_psi_sparse[3]; + + int dRx = R_coor.x; + int dRy = R_coor.y; + int dRz = R_coor.z; + + ModuleBase::Vector3 R_car = ModuleBase::Vector3(dRx, dRy, dRz) * ucell.latvec; + + int ir, ic; + for (int iw1 = 0; iw1 < nlocal; iw1++) + { + ir = pv_.global2local_row(iw1); + if (ir >= 0) + { + for (int iw2 = 0; iw2 < nlocal; iw2++) + { + ic = pv_.global2local_col(iw2); + if (ic >= 0) + { + int orb_index_row = iw1 / npol; + int orb_index_col = iw2 / npol; + + // The off-diagonal term in SOC calculaiton is zero, and the two diagonal terms are the same + int new_index + = iw1 - npol * orb_index_row + (iw2 - npol * orb_index_col) * npol; + + if (new_index == 0 || new_index == 3) + { + int it1 = basis_.get_iw2it(orb_index_row); + int ia1 = basis_.get_iw2ia(orb_index_row); + int iN1 = basis_.get_iw2iN(orb_index_row); + int iL1 = basis_.get_iw2iL(orb_index_row); + int im1 = basis_.get_iw2im(orb_index_row); + + int it2 = basis_.get_iw2it(orb_index_col); + int ia2 = basis_.get_iw2ia(orb_index_col); + int iN2 = basis_.get_iw2iN(orb_index_col); + int iL2 = basis_.get_iw2iL(orb_index_col); + int im2 = basis_.get_iw2im(orb_index_col); + + // PosOpCalc::pos_matrix expects absolute Cartesian positions + // of both centers and computes the inter-center distance itself. + // The second center is atom ia2 translated by the lattice vector R. + ModuleBase::Vector3 tau1_car_ = ucell.atoms[it1].tau[ia1] * ucell.lat0; + ModuleBase::Vector3 tau2_car_ = (ucell.atoms[it2].tau[ia2] + R_car) * ucell.lat0; + + ModuleBase::Vector3 temp_prp = calc_.pos_matrix(tau1_car_, + it1, iL1, im1, iN1, + tau2_car_, + it2, iL2, im2, iN2); + + if (std::abs(temp_prp.x) > sparse_threshold) + { + psi_r_psi_sparse[0][iw1][iw2] = temp_prp.x; + } + + if (std::abs(temp_prp.y) > sparse_threshold) + { + psi_r_psi_sparse[1][iw1][iw2] = temp_prp.y; + } + + if (std::abs(temp_prp.z) > sparse_threshold) + { + psi_r_psi_sparse[2][iw1][iw2] = temp_prp.z; + } + } + } + } + } + } + + int rR_nonzero_num[3] = {0, 0, 0}; + for (int direction = 0; direction < 3; ++direction) + { + for (auto& row_loop: psi_r_psi_sparse[direction]) + { + rR_nonzero_num[direction] += row_loop.second.size(); + } + } + + Parallel_Reduce::reduce_all(rR_nonzero_num, 3); + + if (ModuleIO::detail::lat_r_nonempty(rR_nonzero_num)) + { + output_R_number++; + + if (GlobalV::DRANK == 0) + { + if (binary) + { + ofs_tem1.write(reinterpret_cast(&dRx), sizeof(int)); + ofs_tem1.write(reinterpret_cast(&dRy), sizeof(int)); + ofs_tem1.write(reinterpret_cast(&dRz), sizeof(int)); + } + else + { + ofs_tem1 << dRx << " " << dRy << " " << dRz << std::endl; + } + } + + for (int direction = 0; direction < 3; ++direction) + { + if (GlobalV::DRANK == 0) + { + if (binary) + { + ofs_tem1.write(reinterpret_cast(&rR_nonzero_num[direction]), sizeof(int)); + } + else + { + ofs_tem1 << rR_nonzero_num[direction] << std::endl; + } + } + + if (rR_nonzero_num[direction]) + { + ModuleIO::save_lat_r(ofs_tem1, psi_r_psi_sparse[direction], pv_, lat_r_options); + } + else + { + // do nothing + } + } + } + } + + if (GlobalV::DRANK == 0) + { + std::stringstream ssr; + const bool md_no_append = (calculation == "md") && !out_app_flag; + if (step >= 0) + { + ssr << (md_no_append ? global_matrix_dir : global_out_dir) << "rrg" << (step + 1) << "_nao.txt"; + } + else + { + ssr << global_out_dir << "rr_nao.txt"; + } + + ofs_tem1.close(); + const bool open_in_append = (calculation == "md") && out_app_flag && step >= 0; + const int header_step = std::max(step, 0); + ModuleIO::detail::assemble_csr(ssr.str(), + tem1.str(), + header_step, + nlocal, + output_R_number, + binary, + open_in_append, + "PosOpWriter::out_lat_r"); + + std::remove(tem1.str().c_str()); + } + + ModuleBase::timer::end("PosOpWriter", "out_lat_r"); + return; +} diff --git a/source/source_io/module_hs/pos_op_writer.h b/source/source_io/module_hs/pos_op_writer.h new file mode 100644 index 00000000000..52c74e6de7a --- /dev/null +++ b/source/source_io/module_hs/pos_op_writer.h @@ -0,0 +1,50 @@ +#ifndef POS_OP_WRITER_H +#define POS_OP_WRITER_H + +#include "source_base/vector3.h" +#include "source_cell/unitcell.h" +#include "source_lcao/module_ri/abfs_vector3_order.h" +#include "pos_op_basis.h" +#include "pos_op_calc.h" + +#include +#include +#include +#include +#include + +class Grid_Driver; +class Parallel_Orbitals; + +/** + * @brief Output the position-operator matrix + * as a sparse matrix for each lattice vector R (lat_r). + * + * The writer loops over all periodic images, assembles the sparse blocks, + * and writes the final rr.csr file via ModuleIO helpers. + */ +class PosOpWriter +{ + public: + PosOpWriter(const PosOpBasis& basis, const PosOpCalc& calc, const Parallel_Orbitals& pv); + + void out_lat_r(const UnitCell& ucell, + const Grid_Driver& gd, + int istep, + int precision, + const std::string& global_out_dir, + const std::string& global_matrix_dir, + const std::string& calculation, + bool out_app_flag, + int nlocal, + int npol, + double sparse_threshold, + bool binary); + + private: + const PosOpBasis& basis_; + const PosOpCalc& calc_; + const Parallel_Orbitals& pv_; +}; + +#endif // POS_OP_WRITER_H diff --git a/source/source_io/module_hs/rr_sparse_writer.cpp b/source/source_io/module_hs/rr_sparse_writer.cpp deleted file mode 100644 index ed5dadb9964..00000000000 --- a/source/source_io/module_hs/rr_sparse_writer.cpp +++ /dev/null @@ -1,64 +0,0 @@ -#include "rr_sparse_writer.h" - -#include "source_base/tool_quit.h" - -#include - -namespace ModuleIO -{ -namespace detail -{ -bool rr_sparse_has_payload(const int nonzero_num[3]) -{ - return nonzero_num[0] != 0 || nonzero_num[1] != 0 || nonzero_num[2] != 0; -} - -void finalize_rr_sparse_file(const std::string& output_filename, - const std::string& payload_filename, - const int step, - const int nlocal, - const int output_R_number, - const bool binary, - const bool append, - const std::string& context) -{ - std::ios_base::openmode output_mode = std::ios::out; - std::ios_base::openmode payload_mode = std::ios::in; - if (binary) - { - output_mode |= std::ios::binary; - payload_mode |= std::ios::binary; - } - if (append) - { - output_mode |= std::ios::app; - } - - std::ofstream out_r(output_filename.c_str(), output_mode); - if (!out_r.is_open()) - { - ModuleBase::WARNING_QUIT(context, "Cannot open r(R) output file: " + output_filename); - } - - if (binary) - { - out_r.write(reinterpret_cast(&step), sizeof(int)); - out_r.write(reinterpret_cast(&nlocal), sizeof(int)); - out_r.write(reinterpret_cast(&output_R_number), sizeof(int)); - } - else - { - out_r << "STEP: " << step << std::endl; - out_r << "Matrix Dimension of r(R): " << nlocal << std::endl; - out_r << "Matrix number of r(R): " << output_R_number << std::endl; - } - - std::ifstream payload(payload_filename.c_str(), payload_mode); - if (!payload.is_open()) - { - ModuleBase::WARNING_QUIT(context, "Cannot read temporary sparse matrix file: " + payload_filename); - } - out_r << payload.rdbuf(); -} -} // namespace detail -} // namespace ModuleIO diff --git a/source/source_io/module_hs/rr_sparse_writer.h b/source/source_io/module_hs/rr_sparse_writer.h deleted file mode 100644 index 78e17462d43..00000000000 --- a/source/source_io/module_hs/rr_sparse_writer.h +++ /dev/null @@ -1,23 +0,0 @@ -#ifndef RR_SPARSE_WRITER_H -#define RR_SPARSE_WRITER_H - -#include - -namespace ModuleIO -{ -namespace detail -{ -bool rr_sparse_has_payload(const int nonzero_num[3]); - -void finalize_rr_sparse_file(const std::string& output_filename, - const std::string& payload_filename, - int step, - int nlocal, - int output_R_number, - bool binary, - bool append, - const std::string& context); -} // namespace detail -} // namespace ModuleIO - -#endif diff --git a/source/source_io/module_hs/single_r_io.cpp b/source/source_io/module_hs/single_r_io.cpp deleted file mode 100644 index 77fb34a8fdc..00000000000 --- a/source/source_io/module_hs/single_r_io.cpp +++ /dev/null @@ -1,169 +0,0 @@ -#include "single_r_io.h" -#include "source_base/parallel_reduce.h" -#include "source_base/global_function.h" -#include "source_base/global_variable.h" - -#include -#include -#include -#include -#include -#include - -inline void write_data(std::ofstream& ofs, const double& data, const int precision) -{ - ofs << " " << std::fixed << std::scientific << std::setprecision(precision) << data; -} -inline void write_data(std::ofstream& ofs, const std::complex& data, const int precision) -{ - ofs << " (" << std::fixed << std::scientific << std::setprecision(precision) - << data.real() << "," << data.imag() << ")"; -} - -template -void ModuleIO::output_single_R(std::ofstream& ofs, - const SparseRBlock& XR, - const Parallel_Orbitals& pv, - const SparseWriteOptions& options) -{ - const int nlocal = pv.get_global_row_size(); - if (nlocal <= 0) - { - ModuleBase::WARNING_QUIT("ModuleIO::output_single_R", - "Parallel_Orbitals global row size must be positive."); - } - - std::vector indptr; - indptr.reserve(nlocal + 1); - indptr.push_back(0); - - std::stringstream tem1; - tem1 << options.temp_dir << std::to_string(GlobalV::DRANK) - << "temp_sparse_indices.dat"; - std::ofstream ofs_tem1; - std::ifstream ifs_tem1; - - if (!options.reduce || GlobalV::DRANK == 0) - { - if (options.binary) - { - ofs_tem1.open(tem1.str().c_str(), std::ios::binary); - } - else - { - ofs_tem1.open(tem1.str().c_str()); - } - if (!ofs_tem1.is_open()) - { - ModuleBase::WARNING_QUIT("ModuleIO::output_single_R", - "Cannot open temporary sparse index file: " + tem1.str()); - } - } - - std::vector line(nlocal); - for(int row = 0; row < nlocal; ++row) - { - ModuleBase::GlobalFunc::ZEROS(line.data(), nlocal); - - if (!options.reduce || pv.global2local_row(row) >= 0) - { - auto iter = XR.find(row); - if (iter != XR.end()) - { - for (auto &value : iter->second) - { - if (value.first >= static_cast(nlocal)) - { - std::cerr << "Sparse column index out of range." << std::endl; - ModuleBase::WARNING_QUIT("ModuleIO::output_single_R", - "Sparse column index out of range."); - } - line[value.first] = value.second; - } - } - } - - if (options.reduce) - { - Parallel_Reduce::reduce_all(line.data(), nlocal); - } - - if (!options.reduce || GlobalV::DRANK == 0) - { - long long nonzeros_count = 0; - for (int col = 0; col < nlocal; ++col) - { - if (std::abs(line[col]) > options.threshold) - { - if (options.binary) - { - ofs.write(reinterpret_cast(&line[col]), sizeof(T)); - ofs_tem1.write(reinterpret_cast(&col), sizeof(int)); - } - else - { - write_data(ofs, line[col], options.precision); - ofs_tem1 << " " << col; - } - - nonzeros_count++; - - } - - } - nonzeros_count += indptr.back(); - indptr.push_back(nonzeros_count); - } - } - - if (!options.reduce || GlobalV::DRANK == 0) - { - if (options.binary) - { - ofs_tem1.close(); - ifs_tem1.open(tem1.str().c_str(), std::ios::binary); - if (!ifs_tem1.is_open()) - { - ModuleBase::WARNING_QUIT("ModuleIO::output_single_R", - "Cannot read temporary sparse index file: " + tem1.str()); - } - ofs << ifs_tem1.rdbuf(); - ifs_tem1.close(); - for (auto &i : indptr) - { - ofs.write(reinterpret_cast(&i), sizeof(long long)); - } - } - else - { - ofs << std::endl; - ofs_tem1 << std::endl; - ofs_tem1.close(); - ifs_tem1.open(tem1.str().c_str()); - if (!ifs_tem1.is_open()) - { - ModuleBase::WARNING_QUIT("ModuleIO::output_single_R", - "Cannot read temporary sparse index file: " + tem1.str()); - } - ofs << ifs_tem1.rdbuf(); - ifs_tem1.close(); - for (auto &i : indptr) - { - ofs << " " << i; - } - ofs << std::endl; - } - - std::remove(tem1.str().c_str()); - } -} - -template void ModuleIO::output_single_R(std::ofstream& ofs, - const SparseRBlock& XR, - const Parallel_Orbitals& pv, - const SparseWriteOptions& options); - -template void ModuleIO::output_single_R>(std::ofstream& ofs, - const SparseRBlock>& XR, - const Parallel_Orbitals& pv, - const SparseWriteOptions& options); diff --git a/source/source_io/module_hs/single_r_io.h b/source/source_io/module_hs/single_r_io.h deleted file mode 100644 index f217f943f00..00000000000 --- a/source/source_io/module_hs/single_r_io.h +++ /dev/null @@ -1,17 +0,0 @@ -#ifndef SINGLE_R_IO_H -#define SINGLE_R_IO_H - -#include "write_hs_sparse.h" - -#include - -namespace ModuleIO -{ - template - void output_single_R(std::ofstream& ofs, - const SparseRBlock& XR, - const Parallel_Orbitals& pv, - const SparseWriteOptions& options); -} - -#endif diff --git a/source/source_io/module_hs/unittests/CMakeLists.txt b/source/source_io/module_hs/unittests/CMakeLists.txt new file mode 100644 index 00000000000..af2c1ff79f9 --- /dev/null +++ b/source/source_io/module_hs/unittests/CMakeLists.txt @@ -0,0 +1,143 @@ +AddTest( + TARGET MODULE_IO_hsk_writer_test + LIBS parameter base device + SOURCES + test_hsk_writer.cpp + ../hsk_writer.cpp + ../hs_dense_io.cpp + ../../../source_basis/module_ao/parallel_orbitals.cpp +) + +add_test( + NAME MODULE_IO_hsk_writer_test_parallel + COMMAND mpirun -np 2 ./MODULE_IO_hsk_writer_test + WORKING_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR} +) + +AddTest( + TARGET MODULE_IO_lat_r_csr_test + LIBS parameter + SOURCES test_lat_r_csr.cpp ../lat_r_csr.cpp + ../../../source_base/global_variable.cpp + ../../../source_base/parallel_reduce.cpp + ../../../source_base/parallel_common.cpp + ../../../source_base/parallel_global.cpp + ../../../source_base/parallel_comm.cpp + ../../../source_base/tool_quit.cpp ../../../source_base/global_file.cpp ../../../source_base/global_function.cpp ../../../source_base/memory_recorder.cpp ../../../source_base/timer.cpp +) + +if(ENABLE_LCAO) +AddTest( + TARGET MODULE_IO_vxc_op_tools_test + LIBS parameter base device + SOURCES + test_vxc_op_tools.cpp + ../vxc_op_tools.cpp + ../../../source_basis/module_ao/parallel_orbitals.cpp +) + +AddTest( + TARGET MODULE_IO_hs_dense_io_test + LIBS parameter + SOURCES + test_hs_dense_io.cpp + ../hs_dense_io.cpp + ../../../source_basis/module_ao/parallel_orbitals.cpp + ../../../source_base/global_variable.cpp + ../../../source_base/parallel_2d.cpp + ../../../source_base/parallel_reduce.cpp + ../../../source_base/parallel_common.cpp + ../../../source_base/parallel_global.cpp + ../../../source_base/parallel_comm.cpp + ../../../source_base/tool_quit.cpp + ../../../source_base/tool_title.cpp + ../../../source_base/global_file.cpp + ../../../source_base/global_function.cpp + ../../../source_base/memory_recorder.cpp + ../../../source_base/timer.cpp +) + +add_test( + NAME MODULE_IO_hs_dense_io_test_parallel + COMMAND mpirun -np 2 ./MODULE_IO_hs_dense_io_test + --gtest_filter=HsDenseIoSaveMat.BinaryGammaUpperTriangle:HsDenseIoSaveMat.BinaryComplexUpperTriangleMpiReduction:HsDenseIoSaveMat.BinaryAppendCompleteRecordsAndOverwrite + WORKING_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR} +) + +AddTest( + TARGET MODULE_IO_hsr_writer_test + LIBS parameter base device hcontainer + SOURCES + test_hsr_writer.cpp + ../hsr_writer.cpp + ../../../source_cell/ucell_io.cpp + ../../../source_basis/module_ao/parallel_orbitals.cpp + ../../../source_hamilt/module_hcontainer/test/tmp_mocks.cpp +) + +add_test( + NAME MODULE_IO_hsr_writer_test_parallel + COMMAND mpirun -np 2 ./MODULE_IO_hsr_writer_test + --gtest_filter=HsrWriterIo.HContainerBinaryMpiGatherWritesCompleteFiles + WORKING_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR} +) + +AddTest( + TARGET MODULE_IO_angmom_op_mat_test + LIBS parameter base device neighbor + SOURCES + test_angmom_op_mat.cpp + ../angmom_op_mat.cpp + ../../../source_basis/module_ao/orb_atomic_lm.cpp + ../../../source_basis/module_ao/orb_atomic.cpp + ../../../source_basis/module_nao/radial_set.cpp + ../../../source_basis/module_nao/numerical_radial.cpp + ../../../source_basis/module_nao/beta_radials.cpp + ../../../source_basis/module_nao/hydrogen_radials.cpp + ../../../source_basis/module_nao/pswfc_radials.cpp + ../../../source_basis/module_nao/atomic_radials.cpp + ../../../source_basis/module_nao/sphbes_radials.cpp + ../../../source_basis/module_nao/two_center_integrator.cpp + ../../../source_basis/module_nao/two_center_table.cpp + ../../../source_basis/module_nao/real_gaunt_table.cpp + ../../../source_basis/module_nao/radial_collection.cpp +) + +AddTest( + TARGET MODULE_IO_pos_op_csr_test + LIBS parameter base device + SOURCES + test_pos_op_csr.cpp + ../pos_op_csr.cpp +) + +AddTest( + TARGET MODULE_IO_hs_sparse_io_test + LIBS parameter base device hcontainer + SOURCES + test_hs_sparse_io.cpp + ../hs_sparse_io.cpp + ../lat_r_csr.cpp + ../../../source_basis/module_ao/parallel_orbitals.cpp + ../../../source_hamilt/module_hcontainer/test/tmp_mocks.cpp +) + +AddTest( + TARGET MODULE_IO_dhs_sparse_writer_test + LIBS parameter base device hcontainer + SOURCES + test_dhs_sparse_writer.cpp + ../dhs_sparse_writer.cpp + ../hs_sparse_io.cpp + ../lat_r_csr.cpp + ../../../source_basis/module_ao/parallel_orbitals.cpp + ../../../source_hamilt/module_hcontainer/test/tmp_mocks.cpp +) + +AddTest( + TARGET MODULE_IO_output_mat_sparse_test + LIBS parameter base device + SOURCES + test_output_mat_sparse.cpp +) +endif() diff --git a/source/source_io/module_hs/unittests/csr_test_helpers.h b/source/source_io/module_hs/unittests/csr_test_helpers.h new file mode 100644 index 00000000000..5676d3bb5ef --- /dev/null +++ b/source/source_io/module_hs/unittests/csr_test_helpers.h @@ -0,0 +1,223 @@ +#ifndef CSR_TEST_HELPERS_H +#define CSR_TEST_HELPERS_H + +/** + * @file csr_test_helpers.h + * @brief Shared helper utilities for the CSR writer unit tests in this + * directory (hsr_writer / hs_sparse_io / dhs_sparse_writer). + * + * Each test executable includes this header from exactly one translation + * unit, so the anonymous-namespace helpers below are instantiated once per + * test binary. + */ + +#include "gmock/gmock.h" +#include "gtest/gtest.h" + +#include "source_base/global_variable.h" +#include "source_cell/unitcell.h" +#include "source_hamilt/module_hcontainer/atom_pair.h" +#include "source_hamilt/module_hcontainer/hcontainer.h" + +#include +#include +#include +#include +#include +#include +#include +#include + +namespace +{ +std::string read_file(const std::string& filename) +{ + std::ifstream ifs(filename.c_str()); + std::ostringstream oss; + oss << ifs.rdbuf(); + return oss.str(); +} + +std::vector read_binary_ints(const std::string& filename, const size_t count) +{ + std::ifstream ifs(filename.c_str(), std::ios::binary); + std::vector values(count, 0); + for (size_t i = 0; i < count; ++i) + { + ifs.read(reinterpret_cast(&values[i]), sizeof(int)); + } + return values; +} + +template +T read_binary_value(std::ifstream& ifs) +{ + T value{}; + ifs.read(reinterpret_cast(&value), sizeof(T)); + return value; +} + +struct NativeDoubleRBlock +{ + int rx = 0; + int ry = 0; + int rz = 0; + std::vector values; + std::vector columns; + std::vector row_ptr; +}; + +struct NativeDoubleRecord +{ + int step = 0; + int nbasis = 0; + std::vector blocks; +}; + +NativeDoubleRecord read_native_double_record(std::ifstream& ifs) +{ + NativeDoubleRecord record; + record.step = read_binary_value(ifs); + record.nbasis = read_binary_value(ifs); + const int nR = read_binary_value(ifs); + record.blocks.resize(nR); + for (NativeDoubleRBlock& block: record.blocks) + { + block.rx = read_binary_value(ifs); + block.ry = read_binary_value(ifs); + block.rz = read_binary_value(ifs); + const int nnz = read_binary_value(ifs); + block.values.resize(nnz); + block.columns.resize(nnz); + block.row_ptr.resize(record.nbasis + 1); + for (double& value: block.values) + { + value = read_binary_value(ifs); + } + for (int& column: block.columns) + { + column = read_binary_value(ifs); + } + for (long long& pointer: block.row_ptr) + { + pointer = read_binary_value(ifs); + } + } + return record; +} + +std::vector read_lines(const std::string& filename) +{ + std::ifstream ifs(filename.c_str()); + std::vector lines; + std::string line; + while (std::getline(ifs, line)) + { + lines.push_back(line); + } + return lines; +} + +int count_substr(const std::string& text, const std::string& pattern) +{ + int count = 0; + std::string::size_type pos = 0; + while ((pos = text.find(pattern, pos)) != std::string::npos) + { + ++count; + pos += pattern.size(); + } + return count; +} + +void init_unitcell(UnitCell& ucell) +{ + ucell.latName = "user_defined_lattice"; + ucell.lat0 = 10.0; + ucell.latvec.e11 = 1.0; + ucell.latvec.e22 = 1.0; + ucell.latvec.e33 = 1.0; + ucell.ntype = 1; + ucell.nat = 1; + ucell.atoms = new Atom[1]; + ucell.set_atom_flag = true; + ucell.atoms[0].label = "Si"; + ucell.atoms[0].na = 1; + ucell.atoms[0].nw = 2; + ucell.atoms[0].taud.resize(1); + ucell.atoms[0].taud[0] = ModuleBase::Vector3(0.0, 0.25, 0.5); +} + +void init_serial_orbitals(Parallel_Orbitals& pv) +{ + pv.atom_begin_row.resize(2); + pv.atom_begin_col.resize(2); + pv.atom_begin_row[0] = 0; + pv.atom_begin_row[1] = 2; + pv.atom_begin_col[0] = 0; + pv.atom_begin_col[1] = 2; + pv.nrow = 2; + pv.ncol = 2; + pv.set_serial(2, 2); +} + +void fill_matrix(hamilt::HContainer& matrix, Parallel_Orbitals& pv, double* values) +{ + hamilt::AtomPair pair(0, 0, 0, 0, 0, &pv, values); + matrix.insert_pair(pair); +} + +template +void fill_matrix_at_R(hamilt::HContainer& matrix, + Parallel_Orbitals& pv, + const int rx, + const int ry, + const int rz, + T* values) +{ + hamilt::AtomPair pair(0, 0, rx, ry, rz, &pv, values); + matrix.insert_pair(pair); +} + +void init_sparse_output_globals() +{ + GlobalV::DRANK = 0; +} + +void remove_derivative_files(const std::string& fileflag, int step = -1) +{ + std::vector filenames; + if (step >= 0) + { + const std::string suffix = "g" + std::to_string(step + 1) + "_nao.csr"; + for (const char axis : {'x', 'y', 'z'}) + { + for (int ispin = 1; ispin <= 2; ++ispin) + { + filenames.push_back("d" + fileflag + "r" + axis + "s" + std::to_string(ispin) + suffix); + } + } + } + else + { + for (const char axis : {'x', 'y', 'z'}) + { + for (int ispin = 1; ispin <= 2; ++ispin) + { + filenames.push_back("d" + fileflag + "r" + axis + "s" + std::to_string(ispin) + "_nao.csr"); + } + } + } + for (const std::string& filename: filenames) + { + std::remove(filename.c_str()); + } +} + +bool starts_with(const std::string& text, const std::string& prefix) +{ + return text.find(prefix) == 0; +} +} // namespace + +#endif // CSR_TEST_HELPERS_H diff --git a/source/source_io/module_hs/unittests/test_angmom_op_mat.cpp b/source/source_io/module_hs/unittests/test_angmom_op_mat.cpp new file mode 100644 index 00000000000..65dcc563121 --- /dev/null +++ b/source/source_io/module_hs/unittests/test_angmom_op_mat.cpp @@ -0,0 +1,175 @@ +/** + * Unit-test of angmom_op_mat.cpp + * + * The representation matrices for Lx, Ly and Lz operators (SO(3) generators). + * On basis {s}: + * Lx: Ly: Lz: + * 0 0 0 + * + * On basis {px, py, pz}: + * Lx: Ly: Lz: + * px py pz px py pz px py pz + * px 0 px 0 i px 0 -i + * py 0 -i py 0 py i 0 + * pz i 0 pz -i 0 pz 0 + * + * will test if the calculated values are close to the above results. + */ + +#include +#include +#include +#include +#include +#ifdef __MPI +#include +#endif + +#include "source_io/module_hs/angmom_op_mat.h" +#include "source_basis/module_nao/two_center_integrator.h" +#include "source_basis/module_nao/radial_collection.h" +#include "source_base/sph_bessel_tf.h" +#include "source_base/ylm.h" + +namespace +{ +constexpr double kDoubleThreshold = 1e-12; +} + +class CalpLpRTest : public ::testing::Test +{ +protected: + void SetUp() override + { + ModuleBase::SphericalBesselTransformer sbt(true); + + this->orb_ = std::unique_ptr(new RadialCollection); + this->orb_->build(1, &forb_, 'o'); + this->orb_->set_transformer(sbt); + + const double rcut = 8.0; + const int ngrid = int(rcut / 0.01) + 1; + const double cutoff = 2.0 * rcut; + this->orb_->set_uniform_grid(true, ngrid, cutoff, 'i', true); + + this->calculator_ = std::unique_ptr(new TwoCenterIntegrator); + this->calculator_->tabulate(*orb_, *orb_, 'S', ngrid, cutoff); + + // Initialize Ylm coefficients + ModuleBase::Ylm::set_coefficients(); + } + + void TearDown() override + { + // Cleanup code here, if needed + } + + const std::string forb_ = "../../../../../tests/PP_ORB/Si_gga_8au_100Ry_2s2p1d.orb"; + + std::unique_ptr calculator_; + std::unique_ptr orb_; +}; + +TEST_F(CalpLpRTest, CalLzijRTest) +{ + std::complex out; + + ModuleBase::Vector3 vR(0., 0., 0.); // home-cell + int it = 0, ia = 0, il = 0, iz = 0, im = 0; + int jt = 0, ja = 0, jl = 0, jz = 0, jm = 0; + + // l=0 + out = ModuleIO::cal_LzijR(calculator_, it, ia, il, iz, im, jt, ja, jl, jz, jm, vR); + EXPECT_NEAR(out.real(), 0.0, kDoubleThreshold); + EXPECT_NEAR(out.imag(), 0.0, kDoubleThreshold); + + // l=1 + il = 1; jl = 1; + std::vector> ans(9, 0.0); + ans[1] = {0.0, -1.0}; ans[3] = {0.0, 1.0}; + int idx = 0; + const std::vector m = {1, -1, 0}; // px, py, pz + for (int i = 0; i < static_cast(m.size()); ++i) { + for (int j = 0; j < static_cast(m.size()); ++j) { + im = m[i]; jm = m[j]; + out = ModuleIO::cal_LzijR(calculator_, it, ia, il, iz, im, jt, ja, jl, jz, jm, vR); + EXPECT_NEAR(out.real(), ans[idx].real(), kDoubleThreshold); + EXPECT_NEAR(out.imag(), ans[idx].imag(), kDoubleThreshold); + idx++; + } + } +} + +TEST_F(CalpLpRTest, CalLxijRTest) +{ + std::complex out; + + ModuleBase::Vector3 vR(0., 0., 0.); // home-cell + int it = 0, ia = 0, il = 0, iz = 0, im = 0; + int jt = 0, ja = 0, jl = 0, jz = 0, jm = 0; + + // l=0 + out = ModuleIO::cal_LxijR(calculator_, it, ia, il, iz, im, jt, ja, jl, jz, jm, vR); + EXPECT_NEAR(out.real(), 0.0, kDoubleThreshold); + EXPECT_NEAR(out.imag(), 0.0, kDoubleThreshold); + + // l=1 + il = 1; jl = 1; + std::vector> ans(9, 0.0); + ans[5] = {0.0, 1.0}; ans[7] = {0.0, -1.0}; + int idx = 0; + const std::vector m = {1, -1, 0}; // px, py, pz + for (int i = 0; i < static_cast(m.size()); ++i) { + for (int j = 0; j < static_cast(m.size()); ++j) { + im = m[i]; jm = m[j]; + out = ModuleIO::cal_LxijR(calculator_, it, ia, il, iz, im, jt, ja, jl, jz, jm, vR); + EXPECT_NEAR(out.real(), ans[idx].real(), kDoubleThreshold); + EXPECT_NEAR(out.imag(), ans[idx].imag(), kDoubleThreshold); + idx++; + } + } +} + +TEST_F(CalpLpRTest, CalLyijRTest) +{ + std::complex out; + + ModuleBase::Vector3 vR(0., 0., 0.); // home-cell + int it = 0, ia = 0, il = 0, iz = 0, im = 0; + int jt = 0, ja = 0, jl = 0, jz = 0, jm = 0; // self, the first s + + // l=0 + out = ModuleIO::cal_LyijR(calculator_, it, ia, il, iz, im, jt, ja, jl, jz, jm, vR); + EXPECT_NEAR(out.real(), 0.0, kDoubleThreshold); + EXPECT_NEAR(out.imag(), 0.0, kDoubleThreshold); + + // l=1 + il = 1; jl = 1; + std::vector> ans(9, 0.0); + ans[2] = {0.0, -1.0}; ans[6] = {0.0, 1.0}; + int idx = 0; + const std::vector m = {1, -1, 0}; // px, py, pz + for (int i = 0; i < static_cast(m.size()); ++i) { + for (int j = 0; j < static_cast(m.size()); ++j) { + im = m[i]; jm = m[j]; + out = ModuleIO::cal_LyijR(calculator_, it, ia, il, iz, im, jt, ja, jl, jz, jm, vR); + EXPECT_NEAR(out.real(), ans[idx].real(), kDoubleThreshold); + EXPECT_NEAR(out.imag(), ans[idx].imag(), kDoubleThreshold); + idx++; + } + } +} + +int main(int argc, char **argv) +{ +#ifdef __MPI + MPI_Init(&argc, &argv); +#endif + ::testing::InitGoogleTest(&argc, argv); + return RUN_ALL_TESTS(); + +#ifdef __MPI + MPI_Finalize(); +#endif + return 0; +} \ No newline at end of file diff --git a/source/source_io/module_hs/unittests/test_dhs_sparse_writer.cpp b/source/source_io/module_hs/unittests/test_dhs_sparse_writer.cpp new file mode 100644 index 00000000000..e2335454cc7 --- /dev/null +++ b/source/source_io/module_hs/unittests/test_dhs_sparse_writer.cpp @@ -0,0 +1,155 @@ +/** + * @file test_dhs_sparse_writer.cpp + * @brief Unit tests for dhs_sparse_writer.cpp (ModuleIO::save_dH_sparse). + * + * save_dH_sparse writes the derivative sparse matrices (dHx/dHy/dHz) for + * either H or S into separate CSR files. The tests verify that: + * - text mode respects the sparse threshold + * - binary mode preserves the native field order + * - SOC spin mode (nspin == 4) writes the complex variant for all + * three directions. + * + * dhs_sparse_writer.cpp only reads GlobalV::DRANK; no PARAM state is + * involved, so the fixture sets DRANK directly. + */ +#include "csr_test_helpers.h" + +#include "source_io/module_hs/dhs_sparse_writer.h" + +#ifdef __MPI +#include +#endif + +TEST(DhsSparseWriter, TextCountsOnlyValuesAboveThreshold) +{ + remove_derivative_files("h", 5); + init_sparse_output_globals(); + + Parallel_Orbitals pv; + init_serial_orbitals(pv); + LCAO_HS_Arrays arrays; + const Abfs::Vector3_Order r_vector(0, 0, 0); + arrays.all_R_coor.insert(r_vector); + arrays.dHRx_sparse[0][r_vector][0][0] = 1.0; + arrays.dHRx_sparse[0][r_vector][0][1] = 1e-12; + arrays.dHRx_sparse[0][r_vector][1][0] = 0.0; + arrays.dHRx_sparse[0][r_vector][1][1] = -2.0; + + ModuleIO::save_dH_sparse(5, pv, arrays, 1e-10, false, "h", 8, "./", "./", "scf", false, 1, 2); + + const std::vector lines = read_lines("dhrxs1g6_nao.csr"); + ASSERT_GE(lines.size(), 7); + EXPECT_EQ(lines[0], "STEP: 5"); + EXPECT_EQ(lines[1], "Matrix Dimension of dHx(R): 2"); + EXPECT_EQ(lines[2], "Matrix number of dHx(R): 1"); + EXPECT_EQ(lines[3], "0 0 0 2"); + EXPECT_THAT(lines[4], testing::HasSubstr("1.00000000e+00")); + EXPECT_THAT(lines[4], testing::HasSubstr("-2.00000000e+00")); + + std::istringstream column_stream(lines[5]); + std::vector columns; + int column = 0; + while (column_stream >> column) + { + columns.push_back(column); + } + EXPECT_THAT(columns, testing::ElementsAre(0, 1)); + + std::istringstream indptr_stream(lines[6]); + std::vector indptr; + long long ptr = 0; + while (indptr_stream >> ptr) + { + indptr.push_back(ptr); + } + EXPECT_THAT(indptr, testing::ElementsAre(0, 1, 2)); + + const std::vector y_lines = read_lines("dhrys1g6_nao.csr"); + ASSERT_GE(y_lines.size(), 4); + EXPECT_EQ(y_lines[3], "0 0 0 0"); + + remove_derivative_files("h", 5); +} + +TEST(DhsSparseWriter, BinaryCountsOnlyValuesAboveThreshold) +{ + remove_derivative_files("h", 6); + init_sparse_output_globals(); + + Parallel_Orbitals pv; + init_serial_orbitals(pv); + LCAO_HS_Arrays arrays; + const Abfs::Vector3_Order r_vector(0, 0, 0); + arrays.all_R_coor.insert(r_vector); + arrays.dHRx_sparse[0][r_vector][0][0] = 1.0; + arrays.dHRx_sparse[0][r_vector][0][1] = 1e-12; + arrays.dHRx_sparse[0][r_vector][1][0] = 0.0; + arrays.dHRx_sparse[0][r_vector][1][1] = -2.0; + + ModuleIO::save_dH_sparse(6, pv, arrays, 1e-10, true, "h", 8, "./", "./", "scf", false, 1, 2); + + std::ifstream ifs("dhrxs1g7_nao.csr", std::ios::binary); + ASSERT_TRUE(ifs.is_open()); + EXPECT_EQ(read_binary_value(ifs), 6); + EXPECT_EQ(read_binary_value(ifs), 2); + EXPECT_EQ(read_binary_value(ifs), 1); + EXPECT_EQ(read_binary_value(ifs), 0); + EXPECT_EQ(read_binary_value(ifs), 0); + EXPECT_EQ(read_binary_value(ifs), 0); + EXPECT_EQ(read_binary_value(ifs), 2); + EXPECT_DOUBLE_EQ(read_binary_value(ifs), 1.0); + EXPECT_DOUBLE_EQ(read_binary_value(ifs), -2.0); + EXPECT_EQ(read_binary_value(ifs), 0); + EXPECT_EQ(read_binary_value(ifs), 1); + EXPECT_EQ(read_binary_value(ifs), 0); + EXPECT_EQ(read_binary_value(ifs), 1); + EXPECT_EQ(read_binary_value(ifs), 2); + + remove_derivative_files("h", 6); +} + +TEST(DhsSparseWriter, SocWritesAllDirections) +{ + remove_derivative_files("s", 7); + init_sparse_output_globals(); + + Parallel_Orbitals pv; + init_serial_orbitals(pv); + LCAO_HS_Arrays arrays; + const Abfs::Vector3_Order r_vector(0, 0, 0); + arrays.all_R_coor.insert(r_vector); + arrays.dHRx_soc_sparse[r_vector][0][0] = std::complex(1.0, 0.0); + arrays.dHRy_soc_sparse[r_vector][0][1] = std::complex(2.0, -1.0); + arrays.dHRz_soc_sparse[r_vector][1][1] = std::complex(-3.0, 0.5); + + ModuleIO::save_dH_sparse(7, pv, arrays, 1e-10, false, "s", 8, "./", "./", "scf", false, 4, 2); + + const std::string x_output = read_file("dsrxs1g8_nao.csr"); + const std::string y_output = read_file("dsrys1g8_nao.csr"); + const std::string z_output = read_file("dsrzs1g8_nao.csr"); + EXPECT_THAT(x_output, testing::HasSubstr("Matrix number of dHx(R): 1\n0 0 0 1\n")); + EXPECT_THAT(y_output, testing::HasSubstr("Matrix number of dHy(R): 1\n0 0 0 1\n")); + EXPECT_THAT(z_output, testing::HasSubstr("Matrix number of dHz(R): 1\n0 0 0 1\n")); + EXPECT_THAT(x_output, testing::HasSubstr("(1.00000000e+00,0.00000000e+00)")); + EXPECT_THAT(y_output, testing::HasSubstr("(2.00000000e+00,-1.00000000e+00)")); + EXPECT_THAT(z_output, testing::HasSubstr("(-3.00000000e+00,5.00000000e-01)")); + + remove_derivative_files("s", 7); +} + +int main(int argc, char** argv) +{ +#ifdef __MPI + MPI_Init(&argc, &argv); + MPI_Comm_size(MPI_COMM_WORLD, &GlobalV::NPROC); + MPI_Comm_rank(MPI_COMM_WORLD, &GlobalV::MY_RANK); +#endif + + ::testing::InitGoogleTest(&argc, argv); + const int result = RUN_ALL_TESTS(); + +#ifdef __MPI + MPI_Finalize(); +#endif + return result; +} diff --git a/source/source_io/module_hs/unittests/test_hs_dense_io.cpp b/source/source_io/module_hs/unittests/test_hs_dense_io.cpp new file mode 100644 index 00000000000..31410679def --- /dev/null +++ b/source/source_io/module_hs/unittests/test_hs_dense_io.cpp @@ -0,0 +1,400 @@ +/** + * @file test_hs_dense_io.cpp + * @brief Unit tests for ModuleIO::save_mat + * + * save_mat writes a square matrix in text or binary upper-triangle/full + * format. When built with __MPI (the project default), save_mat uses MPI + * collectives and MPI_Barrier. This test provides a main() that calls + * MPI_Init first so the MPI path is safe on a single rank. Header-only + * tests use a default-constructed/serial Parallel_2D and do not check + * values; the binary upper-triangle tests distribute a known matrix over + * a real Parallel_Orbitals and verify the gathered payload. + */ +#include + +#include "source_io/module_hs/hs_dense_io.h" +#include "source_base/parallel_2d.h" +#include "source_base/parallel_comm.h" +#include "source_basis/module_ao/parallel_orbitals.h" + +#include +#include +#include +#include +#include +#include +#ifdef __MPI +#include +#endif + +namespace +{ +/// Rank in DIAG_WORLD; 0 without MPI. +int test_rank = 0; + +/** + * @brief Read the entire content of a text file into a string. + */ +std::string read_text_file(const std::string& filename) +{ + std::ifstream ifs(filename); + if (!ifs.is_open()) + { + return ""; + } + std::ostringstream oss; + oss << ifs.rdbuf(); + return oss.str(); +} + +/** + * @brief Read just the dimension int from a binary matrix file. + */ +int read_binary_dim(const std::string& filename) +{ + FILE* fp = std::fopen(filename.c_str(), "rb"); + if (fp == nullptr) + { + return -1; + } + int dim = 0; + std::fread(&dim, sizeof(int), 1, fp); + std::fclose(fp); + return dim; +} + +/** + * @brief Create a serial Parallel_2D that owns all rows/cols on rank 0. + */ +Parallel_2D make_serial_pv(int dim) +{ + Parallel_2D pv; + pv.set_serial(dim, dim); + return pv; +} + +/** + * @brief Set up a Parallel_Orbitals distribution over DIAG_WORLD. + */ +void initialize_distribution(Parallel_Orbitals& pv, const int dim) +{ +#ifdef __MPI + ASSERT_EQ(pv.init(dim, dim, 1, DIAG_WORLD), 0); +#else + pv.set_serial(dim, dim); +#endif +} + +/** + * @brief Extract the block of a global square matrix owned by this rank. + */ +template +std::vector distribute_matrix(const Parallel_Orbitals& pv, + const std::vector& global, + const int dim) +{ + std::vector local(pv.get_local_size(), T()); + for (int i = 0; i < dim; ++i) + { + const int ir = pv.global2local_row(i); + if (ir < 0) + { + continue; + } + for (int j = 0; j < dim; ++j) + { + const int ic = pv.global2local_col(j); + if (ic >= 0) + { + local[ir * pv.ncol + ic] = global[i * dim + j]; + } + } + } + return local; +} + +/** + * @brief Pack the upper triangle of a square matrix row by row. + */ +template +std::vector upper_triangle(const std::vector& matrix, const int dim) +{ + std::vector values; + values.reserve(dim * (dim + 1) / 2); + for (int i = 0; i < dim; ++i) + { + for (int j = i; j < dim; ++j) + { + values.push_back(matrix[i * dim + j]); + } + } + return values; +} + +/** + * @brief Read one binary record: dim header followed by the packed + * upper-triangle payload. + */ +template +std::vector read_record(std::ifstream& ifs, const int expected_dim) +{ + int dim = 0; + ifs.read(reinterpret_cast(&dim), sizeof(int)); + EXPECT_TRUE(ifs.good()); + EXPECT_EQ(dim, expected_dim); + + std::vector values(expected_dim * (expected_dim + 1) / 2); + ifs.read(reinterpret_cast(values.data()), values.size() * sizeof(T)); + EXPECT_TRUE(ifs.good()); + return values; +} + +/** + * @brief Remove a test file on rank 0 and sync all ranks. + */ +void remove_test_file(const std::string& filename) +{ + if (test_rank == 0) + { + std::remove(filename.c_str()); + } +#ifdef __MPI + MPI_Barrier(DIAG_WORLD); +#endif +} +} // namespace + +// --------------------------------------------------------------------------- +// Text mode: verify header fields (MPI path, drank=0) +// --------------------------------------------------------------------------- + +TEST(HsDenseIoSaveMat, TextHeaderContainsStepAndDimension) +{ + const std::string filename = "test_save_mat_text_header.dat"; + const int dim = 3; + const std::vector mat(dim * dim, 0.0); + const Parallel_2D pv = make_serial_pv(dim); + ModuleIO::save_mat(0, mat.data(), dim, false, 8, false, false, + filename, pv, 0, "genelpa", false); + const std::string content = read_text_file(filename); + // MPI text path writes header lines starting with '#' + EXPECT_NE(content.find("ionic step"), std::string::npos); + EXPECT_NE(content.find("1"), std::string::npos); // istep+1 = 1 + EXPECT_NE(content.find("gamma only"), std::string::npos); + EXPECT_NE(content.find("rows"), std::string::npos); + EXPECT_NE(content.find("3"), std::string::npos); // dim + std::remove(filename.c_str()); +} + +TEST(HsDenseIoSaveMat, TextHeaderStepTwo) +{ + const std::string filename = "test_save_mat_text_header_step2.dat"; + const int dim = 2; + const std::vector mat(dim * dim, 0.0); + const Parallel_2D pv = make_serial_pv(dim); + ModuleIO::save_mat(1, mat.data(), dim, false, 8, false, false, + filename, pv, 0, "genelpa", false); + const std::string content = read_text_file(filename); + // istep=1 -> "ionic step 2" + EXPECT_NE(content.find("ionic step 2"), std::string::npos); + std::remove(filename.c_str()); +} + +TEST(HsDenseIoSaveMat, TextHeaderGammaOnlyDouble) +{ + const std::string filename = "test_save_mat_text_gamma.dat"; + const int dim = 2; + const std::vector mat(dim * dim, 0.0); + const Parallel_2D pv = make_serial_pv(dim); + ModuleIO::save_mat(0, mat.data(), dim, false, 8, false, false, + filename, pv, 0, "genelpa", false); + const std::string content = read_text_file(filename); + // gamma_only is true for double (std::is_same::value) + EXPECT_NE(content.find("gamma only 1"), std::string::npos); + std::remove(filename.c_str()); +} + +TEST(HsDenseIoSaveMat, TextHeaderGammaOnlyComplex) +{ + const std::string filename = "test_save_mat_text_gamma_complex.dat"; + const int dim = 2; + const std::vector> mat(dim * dim); + const Parallel_2D pv = make_serial_pv(dim); + ModuleIO::save_mat>(0, mat.data(), dim, false, 8, + false, false, filename, pv, 0, "genelpa", false); + const std::string content = read_text_file(filename); + // gamma_only is false for complex + EXPECT_NE(content.find("gamma only 0"), std::string::npos); + std::remove(filename.c_str()); +} + +// --------------------------------------------------------------------------- +// Binary mode: verify dim header (MPI path, drank=0) +// --------------------------------------------------------------------------- + +TEST(HsDenseIoSaveMat, BinaryDimHeaderDouble) +{ + const std::string filename = "test_save_mat_bin_dim_double.dat"; + const int dim = 4; + const std::vector mat(dim * dim, 0.0); + const Parallel_2D pv = make_serial_pv(dim); + ModuleIO::save_mat(0, mat.data(), dim, true, 8, false, false, + filename, pv, 0, "genelpa", false); + const int dim_out = read_binary_dim(filename); + EXPECT_EQ(dim_out, dim); + std::remove(filename.c_str()); +} + +TEST(HsDenseIoSaveMat, BinaryDimHeaderComplex) +{ + const std::string filename = "test_save_mat_bin_dim_complex.dat"; + const int dim = 3; + const std::vector> mat(dim * dim); + const Parallel_2D pv = make_serial_pv(dim); + ModuleIO::save_mat>(0, mat.data(), dim, true, 8, + false, false, filename, pv, 0, "genelpa", false); + const int dim_out = read_binary_dim(filename); + EXPECT_EQ(dim_out, dim); + std::remove(filename.c_str()); +} + +// --------------------------------------------------------------------------- +// Binary upper triangle: distribute a known matrix over a Parallel_Orbitals +// and verify the rank-0 file contains the complete gathered payload. +// --------------------------------------------------------------------------- + +TEST(HsDenseIoSaveMat, BinaryGammaUpperTriangle) +{ + const int dim = 3; + const std::string filename = "save_mat_gamma_upper.dat"; + remove_test_file(filename); + + Parallel_Orbitals pv; + initialize_distribution(pv, dim); + std::vector global(dim * dim); + for (int i = 0; i < dim * dim; ++i) + { + global[i] = i + 0.5; + } + const std::vector local = distribute_matrix(pv, global, dim); + + ModuleIO::save_mat(0, local.data(), dim, true, 8, true, false, filename, + pv, test_rank, "cg"); + + if (test_rank == 0) + { + std::ifstream ifs(filename.c_str(), std::ios::binary | std::ios::ate); + ASSERT_TRUE(ifs.is_open()); + EXPECT_EQ(ifs.tellg(), + static_cast(sizeof(int) + 6 * sizeof(double))); + ifs.seekg(0); + EXPECT_EQ(read_record(ifs, dim), upper_triangle(global, dim)); + EXPECT_EQ(ifs.peek(), std::ifstream::traits_type::eof()); + } + + remove_test_file(filename); +} + +TEST(HsDenseIoSaveMat, BinaryComplexUpperTriangleMpiReduction) +{ + const int dim = 4; + const std::string filename = "save_mat_complex_upper.dat"; + remove_test_file(filename); + + Parallel_Orbitals pv; + initialize_distribution(pv, dim); + std::vector> global(dim * dim); + for (int i = 0; i < dim * dim; ++i) + { + global[i] = std::complex(i + 0.25, -i - 0.75); + } + const std::vector> local + = distribute_matrix(pv, global, dim); + + ModuleIO::save_mat(0, local.data(), dim, true, 8, true, false, filename, + pv, test_rank, "cg"); + + if (test_rank == 0) + { + std::ifstream ifs(filename.c_str(), std::ios::binary | std::ios::ate); + ASSERT_TRUE(ifs.is_open()); + const int element_count = dim * (dim + 1) / 2; + EXPECT_EQ(ifs.tellg(), + static_cast(sizeof(int) + + element_count * sizeof(std::complex))); + ifs.seekg(0); + EXPECT_EQ((read_record>(ifs, dim)), + upper_triangle(global, dim)); + EXPECT_EQ(ifs.peek(), std::ifstream::traits_type::eof()); + } + + remove_test_file(filename); +} + +TEST(HsDenseIoSaveMat, BinaryAppendCompleteRecordsAndOverwrite) +{ + const int dim = 2; + const std::string filename = "save_mat_append.dat"; + remove_test_file(filename); + + Parallel_Orbitals pv; + initialize_distribution(pv, dim); + const std::vector first = {1.0, 2.0, 3.0, 4.0}; + const std::vector second = {5.0, 6.0, 7.0, 8.0}; + const std::vector replacement = {9.0, 10.0, 11.0, 12.0}; + const std::vector first_local = distribute_matrix(pv, first, dim); + const std::vector second_local = distribute_matrix(pv, second, dim); + const std::vector replacement_local + = distribute_matrix(pv, replacement, dim); + + ModuleIO::save_mat(0, first_local.data(), dim, true, 8, true, true, + filename, pv, test_rank, "cg"); + ModuleIO::save_mat(1, second_local.data(), dim, true, 8, true, true, + filename, pv, test_rank, "cg"); + + if (test_rank == 0) + { + std::ifstream ifs(filename.c_str(), std::ios::binary); + ASSERT_TRUE(ifs.is_open()); + EXPECT_EQ(read_record(ifs, dim), upper_triangle(first, dim)); + EXPECT_EQ(read_record(ifs, dim), upper_triangle(second, dim)); + EXPECT_EQ(ifs.peek(), std::ifstream::traits_type::eof()); + } + +#ifdef __MPI + MPI_Barrier(DIAG_WORLD); +#endif + ModuleIO::save_mat(2, replacement_local.data(), dim, true, 8, true, false, + filename, pv, test_rank, "cg"); + + if (test_rank == 0) + { + std::ifstream ifs(filename.c_str(), std::ios::binary); + ASSERT_TRUE(ifs.is_open()); + EXPECT_EQ(read_record(ifs, dim), upper_triangle(replacement, dim)); + EXPECT_EQ(ifs.peek(), std::ifstream::traits_type::eof()); + } + + remove_test_file(filename); +} + +// --------------------------------------------------------------------------- +// Main: initialize MPI before running tests (required by save_mat's MPI path) +// --------------------------------------------------------------------------- + +int main(int argc, char** argv) +{ +#ifdef __MPI + MPI_Init(&argc, &argv); + DIAG_WORLD = MPI_COMM_WORLD; + MPI_Comm_rank(DIAG_WORLD, &test_rank); +#else + test_rank = 0; +#endif + testing::InitGoogleTest(&argc, argv); + const int result = RUN_ALL_TESTS(); +#ifdef __MPI + MPI_Finalize(); +#endif + return result; +} diff --git a/source/source_io/module_hs/unittests/test_hs_sparse_io.cpp b/source/source_io/module_hs/unittests/test_hs_sparse_io.cpp new file mode 100644 index 00000000000..6842f3d1eec --- /dev/null +++ b/source/source_io/module_hs/unittests/test_hs_sparse_io.cpp @@ -0,0 +1,211 @@ +/** + * @file test_hs_sparse_io.cpp + * @brief Unit tests for hs_sparse_io.cpp (ModuleIO::save_sparse). + * + * The tests pin down the legacy CSR text/binary layout written by + * save_sparse: header style ("STEP: n"), dimension lines, threshold + * filtering of values, and the native binary field order. + * hs_sparse_io.cpp only reads GlobalV::DRANK, so the fixture sets it + * directly; no PARAM state is involved. + */ +#include "csr_test_helpers.h" + +#include "source_io/module_hs/hs_sparse_io.h" + +#ifdef __MPI +#include +#endif + +TEST(HsSparseIo, SparseHeaderKeepsStepStyle) +{ + const std::string filename = "write_hs_r_legacy_s.csr"; + std::remove(filename.c_str()); + + GlobalV::DRANK = 0; + + Parallel_Orbitals pv; + init_serial_orbitals(pv); + const Abfs::Vector3_Order r_vector(0, 0, 0); + std::set> all_R_coor; + all_R_coor.insert(r_vector); + std::map, std::map>> sparse_matrix; + sparse_matrix[r_vector][0][1] = 0.5; + sparse_matrix[r_vector][1][1] = 1.5; + + ModuleIO::SparseWriteOptions options; + options.filename = filename; + options.label = "S"; + options.threshold = 1e-10; + options.binary = false; + options.istep = 0; + options.reduce = false; + ModuleIO::save_sparse(sparse_matrix, all_R_coor, pv, options); + + const std::string output = read_file(filename); + EXPECT_TRUE(starts_with(output, "STEP: 0\n")); + EXPECT_THAT(output, testing::HasSubstr("Matrix Dimension of S(R): 2\n")); + EXPECT_THAT(output, testing::HasSubstr("Matrix number of S(R): 1\n")); + EXPECT_THAT(output, testing::HasSubstr("0 0 0 2\n")); + + std::remove(filename.c_str()); +} + +TEST(HsSparseIo, SparseTextCountsOnlyValuesAboveThreshold) +{ + const std::string filename = "write_hs_r_threshold_s.csr"; + std::remove(filename.c_str()); + + GlobalV::DRANK = 0; + + Parallel_Orbitals pv; + init_serial_orbitals(pv); + const Abfs::Vector3_Order r_vector(0, 0, 0); + std::set> all_R_coor; + all_R_coor.insert(r_vector); + std::map, std::map>> sparse_matrix; + sparse_matrix[r_vector][0][0] = 1.0; + sparse_matrix[r_vector][0][1] = 1e-12; + sparse_matrix[r_vector][1][0] = 0.0; + sparse_matrix[r_vector][1][1] = -2.0; + + ModuleIO::SparseWriteOptions options; + options.filename = filename; + options.label = "S"; + options.threshold = 1e-10; + options.binary = false; + options.istep = 2; + options.reduce = false; + ModuleIO::save_sparse(sparse_matrix, all_R_coor, pv, options); + + const std::vector lines = read_lines(filename); + ASSERT_GE(lines.size(), 6); + EXPECT_EQ(lines[0], "STEP: 2"); + EXPECT_EQ(lines[1], "Matrix Dimension of S(R): 2"); + EXPECT_EQ(lines[2], "Matrix number of S(R): 1"); + EXPECT_EQ(lines[3], "0 0 0 2"); + + std::istringstream value_stream(lines[4]); + std::vector values; + double value = 0.0; + while (value_stream >> value) + { + values.push_back(value); + } + EXPECT_THAT(values, testing::ElementsAre(1.0, -2.0)); + + std::istringstream column_stream(lines[5]); + std::vector columns; + int column = 0; + while (column_stream >> column) + { + columns.push_back(column); + } + EXPECT_THAT(columns, testing::ElementsAre(0, 1)); + + ASSERT_GE(lines.size(), 7); + std::istringstream indptr_stream(lines[6]); + std::vector indptr; + long long ptr = 0; + while (indptr_stream >> ptr) + { + indptr.push_back(ptr); + } + EXPECT_THAT(indptr, testing::ElementsAre(0, 1, 2)); + + std::remove(filename.c_str()); +} + +TEST(HsSparseIo, SparseBinaryHeaderWritesConcreteStep) +{ + const std::string filename = "write_hs_r_legacy_binary_s.csr"; + std::remove(filename.c_str()); + + GlobalV::DRANK = 0; + + Parallel_Orbitals pv; + init_serial_orbitals(pv); + const Abfs::Vector3_Order r_vector(0, 0, 0); + std::set> all_R_coor; + all_R_coor.insert(r_vector); + std::map, std::map>> sparse_matrix; + sparse_matrix[r_vector][0][1] = 0.5; + sparse_matrix[r_vector][1][1] = 1.5; + + ModuleIO::SparseWriteOptions options; + options.filename = filename; + options.label = "S"; + options.threshold = 1e-10; + options.binary = true; + options.istep = 3; + options.reduce = false; + ModuleIO::save_sparse(sparse_matrix, all_R_coor, pv, options); + + const std::vector header_and_r = read_binary_ints(filename, 7); + EXPECT_THAT(header_and_r, testing::ElementsAre(3, 2, 1, 0, 0, 0, 2)); + + std::remove(filename.c_str()); +} + +TEST(HsSparseIo, SparseBinaryCountsOnlyValuesAboveThreshold) +{ + const std::string filename = "write_hs_r_threshold_binary_s.csr"; + std::remove(filename.c_str()); + + GlobalV::DRANK = 0; + + Parallel_Orbitals pv; + init_serial_orbitals(pv); + const Abfs::Vector3_Order r_vector(0, 0, 0); + std::set> all_R_coor; + all_R_coor.insert(r_vector); + std::map, std::map>> sparse_matrix; + sparse_matrix[r_vector][0][0] = 1.0; + sparse_matrix[r_vector][0][1] = 1e-12; + sparse_matrix[r_vector][1][0] = 0.0; + sparse_matrix[r_vector][1][1] = -2.0; + + ModuleIO::SparseWriteOptions options; + options.filename = filename; + options.label = "S"; + options.threshold = 1e-10; + options.binary = true; + options.istep = 4; + options.reduce = false; + ModuleIO::save_sparse(sparse_matrix, all_R_coor, pv, options); + + std::ifstream ifs(filename.c_str(), std::ios::binary); + ASSERT_TRUE(ifs.is_open()); + EXPECT_EQ(read_binary_value(ifs), 4); + EXPECT_EQ(read_binary_value(ifs), 2); + EXPECT_EQ(read_binary_value(ifs), 1); + EXPECT_EQ(read_binary_value(ifs), 0); + EXPECT_EQ(read_binary_value(ifs), 0); + EXPECT_EQ(read_binary_value(ifs), 0); + EXPECT_EQ(read_binary_value(ifs), 2); + EXPECT_DOUBLE_EQ(read_binary_value(ifs), 1.0); + EXPECT_DOUBLE_EQ(read_binary_value(ifs), -2.0); + EXPECT_EQ(read_binary_value(ifs), 0); + EXPECT_EQ(read_binary_value(ifs), 1); + EXPECT_EQ(read_binary_value(ifs), 0); + EXPECT_EQ(read_binary_value(ifs), 1); + EXPECT_EQ(read_binary_value(ifs), 2); + + std::remove(filename.c_str()); +} + +int main(int argc, char** argv) +{ +#ifdef __MPI + MPI_Init(&argc, &argv); + MPI_Comm_size(MPI_COMM_WORLD, &GlobalV::NPROC); + MPI_Comm_rank(MPI_COMM_WORLD, &GlobalV::MY_RANK); +#endif + + ::testing::InitGoogleTest(&argc, argv); + const int result = RUN_ALL_TESTS(); + +#ifdef __MPI + MPI_Finalize(); +#endif + return result; +} diff --git a/source/source_io/module_hs/unittests/test_hsk_writer.cpp b/source/source_io/module_hs/unittests/test_hsk_writer.cpp new file mode 100644 index 00000000000..a812efb77f4 --- /dev/null +++ b/source/source_io/module_hs/unittests/test_hsk_writer.cpp @@ -0,0 +1,456 @@ +/** + * @file test_hsk_writer.cpp + * @brief Unit tests for ModuleIO::write_hsk + * + * write_hsk loops over the k points owned by this rank, calls + * Hamilt::updateHk followed by Hamilt::matrix, and writes one H(k) file + * per k point and one S(k) file per spin-up k point (the spin-down + * overlap is skipped because both spin channels share the same S). + * + * A lightweight FakeHamilt supplies controllable local H/S blocks; the + * tests verify the call order (updateHk once per k point), the k-index + * mapping (ik -> ik2iktot global index appears in the filename), the + * spin-skip rule, and that the binary payload matches the distributed + * input after MPI reduction. + */ +#include "source_io/module_hs/hsk_writer.h" + +#include "source_base/matrix_block.h" +#include "source_base/module_out/filename.h" +#include "source_base/parallel_comm.h" +#include "source_basis/module_ao/parallel_orbitals.h" +#include "source_hamilt/hamilt.h" + +#include "gtest/gtest.h" + +#include +#include +#include +#include +#include + +#ifdef __MPI +#include +#endif + +namespace +{ +/// Rank in DIAG_WORLD; 0 without MPI. +int test_rank = 0; + +/// Output options shared by the tests (binary, no append, step 0). +const int k_out_type = 2; +const bool k_out_app = false; +const int k_istep = 0; + +/** + * @brief Test double of hamilt::Hamilt: replays per-k-point local H/S + * buffers set by the test and counts updateHk calls. + */ +template +class FakeHamilt : public hamilt::Hamilt +{ + public: + int dim = 0; + int update_count = 0; + std::vector> hk_local; + std::vector> sk_local; + + void set_k_matrix(const int ik, std::vector hk, std::vector sk) + { + if (static_cast(hk_local.size()) <= ik) + { + hk_local.resize(ik + 1); + sk_local.resize(ik + 1); + } + hk_local[ik] = std::move(hk); + sk_local[ik] = std::move(sk); + } + + void updateHk(const int ik) override + { + ++update_count; + current_ik_ = ik; + } + + void matrix(ModuleBase::MatrixBlock& hk, + ModuleBase::MatrixBlock& sk) override + { + EXPECT_GE(current_ik_, 0); + hk = ModuleBase::MatrixBlock {nullptr, 0, 0, nullptr}; + hk.p = hk_local[current_ik_].data(); + hk.row = static_cast(dim); + hk.col = static_cast(dim); + sk = ModuleBase::MatrixBlock {nullptr, 0, 0, nullptr}; + sk.p = sk_local[current_ik_].data(); + sk.row = static_cast(dim); + sk.col = static_cast(dim); + } + + private: + int current_ik_ = -1; +}; + +/** + * @brief Set up a Parallel_Orbitals distribution over DIAG_WORLD. + */ +void initialize_distribution(Parallel_Orbitals& pv, const int dim) +{ +#ifdef __MPI + ASSERT_EQ(pv.init(dim, dim, 1, DIAG_WORLD), 0); +#else + pv.set_serial(dim, dim); +#endif +} + +/** + * @brief Extract the block of a global square matrix owned by this rank. + */ +template +std::vector distribute_matrix(const Parallel_Orbitals& pv, + const std::vector& global, + const int dim) +{ + std::vector local(pv.get_local_size(), T()); + for (int i = 0; i < dim; ++i) + { + const int ir = pv.global2local_row(i); + if (ir < 0) + { + continue; + } + for (int j = 0; j < dim; ++j) + { + const int ic = pv.global2local_col(j); + if (ic >= 0) + { + local[ir * pv.ncol + ic] = global[i * dim + j]; + } + } + } + return local; +} + +/** + * @brief Pack the upper triangle of a square matrix row by row. + */ +template +std::vector upper_triangle(const std::vector& matrix, const int dim) +{ + std::vector values; + values.reserve(dim * (dim + 1) / 2); + for (int i = 0; i < dim; ++i) + { + for (int j = i; j < dim; ++j) + { + values.push_back(matrix[i * dim + j]); + } + } + return values; +} + +/** + * @brief Read one binary record: dim header followed by the packed + * upper-triangle payload. + */ +template +std::vector read_record(std::ifstream& ifs, const int expected_dim) +{ + int dim = 0; + ifs.read(reinterpret_cast(&dim), sizeof(int)); + EXPECT_TRUE(ifs.good()); + EXPECT_EQ(dim, expected_dim); + + std::vector values(expected_dim * (expected_dim + 1) / 2); + ifs.read(reinterpret_cast(values.data()), values.size() * sizeof(T)); + EXPECT_TRUE(ifs.good()); + return values; +} + +/** + * @brief Whether a regular file can be opened. + */ +bool file_exists(const std::string& filename) +{ + std::ifstream ifs(filename.c_str(), std::ios::binary); + return ifs.is_open(); +} + +/** + * @brief Remove a test file on rank 0 and sync all ranks. + */ +void remove_test_file(const std::string& filename) +{ + if (test_rank == 0) + { + std::remove(filename.c_str()); + } +#ifdef __MPI + MPI_Barrier(DIAG_WORLD); +#endif +} + +/** + * @brief Build the same filename write_hsk asks filename_output for. + */ +std::string expected_filename(const std::string& property, + const int ik, + const std::vector& ik2iktot, + const int nspin, + const int nkstot, + const bool gamma_only) +{ + return ModuleIO::filename_output("", property, "nao", ik, ik2iktot, + nspin, nkstot, k_out_type, k_out_app, + gamma_only, k_istep); +} + +/** + * @brief Open the running-log stream on a per-rank file. + */ +std::ofstream open_running_log() +{ + const std::string filename + = "write_hsk_running_" + std::to_string(test_rank) + ".log"; + return std::ofstream(filename.c_str()); +} + +} // namespace + +// --------------------------------------------------------------------------- +// nspin = 1, multik, real matrices: both H(k) and S(k) are written for +// every k point; payload equals the upper triangle of the distributed input. +// --------------------------------------------------------------------------- + +TEST(WriteHsk, WritesHkAndSkForEveryKPoint) +{ + const int dim = 3; + const int nspin = 1; + const int nks = 2; + const int nkstot = 2; + const std::vector ik2iktot = {0, 1}; + const std::vector isk = {0, 0}; + const bool gamma_only = false; + + Parallel_Orbitals pv; + initialize_distribution(pv, dim); + + FakeHamilt fake; + fake.dim = dim; + std::vector> hk_global(nks); + std::vector> sk_global(nks); + for (int ik = 0; ik < nks; ++ik) + { + hk_global[ik].resize(dim * dim); + sk_global[ik].resize(dim * dim); + for (int i = 0; i < dim * dim; ++i) + { + hk_global[ik][i] = (ik + 1) * 100.0 + i + 0.5; + sk_global[ik][i] = -((ik + 1) * 100.0 + i) - 0.25; + } + fake.set_k_matrix(ik, + distribute_matrix(pv, hk_global[ik], dim), + distribute_matrix(pv, sk_global[ik], dim)); + } + + std::vector filenames; + for (int ik = 0; ik < nks; ++ik) + { + filenames.push_back(expected_filename("hk", ik, ik2iktot, nspin, + nkstot, gamma_only)); + filenames.push_back(expected_filename("sk", ik, ik2iktot, nspin, + nkstot, gamma_only)); + } + for (size_t i = 0; i < filenames.size(); ++i) + { + remove_test_file(filenames[i]); + } + + std::ofstream ofs = open_running_log(); + ModuleIO::write_hsk("", nspin, nks, nkstot, ik2iktot, isk, &fake, pv, + gamma_only, k_out_app, k_istep, k_out_type, 8, dim, + "cg", test_rank, ofs); + + EXPECT_EQ(fake.update_count, nks); + if (test_rank == 0) + { + for (int ik = 0; ik < nks; ++ik) + { + const std::string h_fn = expected_filename("hk", ik, ik2iktot, + nspin, nkstot, gamma_only); + std::ifstream hfs(h_fn.c_str(), std::ios::binary); + ASSERT_TRUE(hfs.is_open()) << h_fn; + EXPECT_EQ(read_record(hfs, dim), + upper_triangle(hk_global[ik], dim)); + EXPECT_EQ(hfs.peek(), std::ifstream::traits_type::eof()); + + const std::string s_fn = expected_filename("sk", ik, ik2iktot, + nspin, nkstot, gamma_only); + std::ifstream sfs(s_fn.c_str(), std::ios::binary); + ASSERT_TRUE(sfs.is_open()) << s_fn; + EXPECT_EQ(read_record(sfs, dim), + upper_triangle(sk_global[ik], dim)); + EXPECT_EQ(sfs.peek(), std::ifstream::traits_type::eof()); + } + } + + for (size_t i = 0; i < filenames.size(); ++i) + { + remove_test_file(filenames[i]); + } + ofs.close(); + remove_test_file("write_hsk_running_" + std::to_string(test_rank) + ".log"); +} + +// --------------------------------------------------------------------------- +// nspin = 2: H(k) is written for both spin channels, but S(k) is skipped +// for the spin-down k point (isk == 1). +// --------------------------------------------------------------------------- + +TEST(WriteHsk, SkipsOverlapForSpinDownChannel) +{ + const int dim = 2; + const int nspin = 2; + const int nks = 2; + const int nkstot = 2; + const std::vector ik2iktot = {0, 1}; + const std::vector isk = {0, 1}; + const bool gamma_only = false; + + Parallel_Orbitals pv; + initialize_distribution(pv, dim); + + FakeHamilt fake; + fake.dim = dim; + for (int ik = 0; ik < nks; ++ik) + { + std::vector hk(dim * dim, static_cast(ik + 1)); + std::vector sk(dim * dim, -static_cast(ik + 1)); + fake.set_k_matrix(ik, distribute_matrix(pv, hk, dim), + distribute_matrix(pv, sk, dim)); + } + + const std::string hk_up = expected_filename("hk", 0, ik2iktot, nspin, + nkstot, gamma_only); + const std::string hk_dn = expected_filename("hk", 1, ik2iktot, nspin, + nkstot, gamma_only); + const std::string sk_up = expected_filename("sk", 0, ik2iktot, nspin, + nkstot, gamma_only); + const std::string sk_dn = expected_filename("sk", 1, ik2iktot, nspin, + nkstot, gamma_only); + std::vector cleanup = {hk_up, hk_dn, sk_up, sk_dn}; + for (size_t i = 0; i < cleanup.size(); ++i) + { + remove_test_file(cleanup[i]); + } + + std::ofstream ofs = open_running_log(); + ModuleIO::write_hsk("", nspin, nks, nkstot, ik2iktot, isk, &fake, pv, + gamma_only, k_out_app, k_istep, k_out_type, 8, dim, + "cg", test_rank, ofs); + + EXPECT_EQ(fake.update_count, nks); + if (test_rank == 0) + { + EXPECT_TRUE(file_exists(hk_up)); + EXPECT_TRUE(file_exists(hk_dn)); + EXPECT_TRUE(file_exists(sk_up)); + EXPECT_FALSE(file_exists(sk_dn)); + } + + for (size_t i = 0; i < cleanup.size(); ++i) + { + remove_test_file(cleanup[i]); + } + ofs.close(); + remove_test_file("write_hsk_running_" + std::to_string(test_rank) + ".log"); +} + +// --------------------------------------------------------------------------- +// Complex multik case: write_hsk> must instantiate and +// write the packed upper triangle of complex double pairs. +// --------------------------------------------------------------------------- + +TEST(WriteHsk, WritesComplexKMatrix) +{ + const int dim = 2; + const int nspin = 1; + const int nks = 1; + const int nkstot = 1; + const std::vector ik2iktot = {0}; + const std::vector isk = {0}; + const bool gamma_only = false; + + Parallel_Orbitals pv; + initialize_distribution(pv, dim); + + std::vector> hk_global(dim * dim); + std::vector> sk_global(dim * dim); + for (int i = 0; i < dim * dim; ++i) + { + hk_global[i] = std::complex(i + 0.25, -i - 0.75); + sk_global[i] = std::complex(-i - 0.5, i + 0.125); + } + + FakeHamilt> fake; + fake.dim = dim; + fake.set_k_matrix(0, distribute_matrix(pv, hk_global, dim), + distribute_matrix(pv, sk_global, dim)); + + const std::string h_fn = expected_filename("hk", 0, ik2iktot, nspin, + nkstot, gamma_only); + const std::string s_fn = expected_filename("sk", 0, ik2iktot, nspin, + nkstot, gamma_only); + remove_test_file(h_fn); + remove_test_file(s_fn); + + std::ofstream ofs = open_running_log(); + ModuleIO::write_hsk>("", nspin, nks, nkstot, + ik2iktot, isk, &fake, pv, + gamma_only, k_out_app, k_istep, + k_out_type, 8, dim, "cg", + test_rank, ofs); + + EXPECT_EQ(fake.update_count, nks); + if (test_rank == 0) + { + std::ifstream hfs(h_fn.c_str(), std::ios::binary | std::ios::ate); + ASSERT_TRUE(hfs.is_open()); + const int element_count = dim * (dim + 1) / 2; + EXPECT_EQ(hfs.tellg(), + static_cast(sizeof(int) + + element_count * sizeof(std::complex))); + hfs.seekg(0); + EXPECT_EQ((read_record>(hfs, dim)), + upper_triangle(hk_global, dim)); + + std::ifstream sfs(s_fn.c_str(), std::ios::binary); + ASSERT_TRUE(sfs.is_open()); + EXPECT_EQ((read_record>(sfs, dim)), + upper_triangle(sk_global, dim)); + } + + remove_test_file(h_fn); + remove_test_file(s_fn); + ofs.close(); + remove_test_file("write_hsk_running_" + std::to_string(test_rank) + ".log"); +} + +int main(int argc, char** argv) +{ +#ifdef __MPI + MPI_Init(&argc, &argv); + DIAG_WORLD = MPI_COMM_WORLD; + MPI_Comm_rank(DIAG_WORLD, &test_rank); +#else + test_rank = 0; +#endif + + testing::InitGoogleTest(&argc, argv); + const int result = RUN_ALL_TESTS(); + +#ifdef __MPI + MPI_Finalize(); +#endif + return result; +} diff --git a/source/source_io/module_hs/unittests/test_hsr_writer.cpp b/source/source_io/module_hs/unittests/test_hsr_writer.cpp new file mode 100644 index 00000000000..f4a9e6c90d8 --- /dev/null +++ b/source/source_io/module_hs/unittests/test_hsr_writer.cpp @@ -0,0 +1,456 @@ +/** + * @file test_hsr_writer.cpp + * @brief Unit tests for hsr_writer.cpp: filename-generation helpers and the + * CSR text/binary H(R), S(R) writer paths (write_hcontainer_csr, + * write_hcontainer_csr_binary, write_hsr). + * + * The filename helpers are pure string functions with no physics state. + * The writer tests pin down the exact CSR text/binary layout on disk so a + * format change is caught by a failing test instead of a broken reader. + */ +#include "csr_test_helpers.h" + +#include "source_base/module_out/csr_reader.h" +#include "source_hamilt/module_hcontainer/hcontainer_funcs.h" +#include "source_io/module_hs/hsr_writer.h" + +#include +#include +#include +#include +#include +#include + +#ifdef __MPI +#include +#endif + +namespace +{ +/** + * @brief Helper: compare two filenames with a clear failure message. + */ +void expect_fname_eq(const std::string& expected, const std::string& actual) +{ + EXPECT_EQ(expected, actual) << "Expected \"" << expected << "\", got \"" << actual << "\""; +} +} // namespace + +// --------------------------------------------------------------------------- +// hsr_gen_fname: H(R) spin-dependent filename +// --------------------------------------------------------------------------- + +TEST(HsrWriterFname, HsrOverwriteWithStep) +{ + // prefix="data-", ispin=0, append=false, istep=0 + // -> "data-1g1_nao.csr" + const std::string fname = ModuleIO::hsr_gen_fname("data-", 0, false, 0); + expect_fname_eq("data-1g1_nao.csr", fname); +} + +TEST(HsrWriterFname, HsrOverwriteWithStepTwo) +{ + // istep=2 -> g3 + const std::string fname = ModuleIO::hsr_gen_fname("data-", 1, false, 2); + expect_fname_eq("data-2g3_nao.csr", fname); +} + +TEST(HsrWriterFname, HsrAppendSuppressesStep) +{ + // append=true -> step number is omitted + const std::string fname = ModuleIO::hsr_gen_fname("data-", 0, true, 5); + expect_fname_eq("data-1_nao.csr", fname); +} + +TEST(HsrWriterFname, HsrNegativeStepSuppressesStep) +{ + // istep=-1 -> step number is omitted (no "g" suffix) + const std::string fname = ModuleIO::hsr_gen_fname("data-", 0, false, -1); + expect_fname_eq("data-1_nao.csr", fname); +} + +TEST(HsrWriterFname, HsrDatExtension) +{ + // out_type=2 -> ".dat" extension + const std::string fname = ModuleIO::hsr_gen_fname("data-", 0, false, 0, 2); + expect_fname_eq("data-1g1_nao.dat", fname); +} + +TEST(HsrWriterFname, HsrCsrExtensionDefault) +{ + // out_type=1 (default) -> ".csr" extension + const std::string fname = ModuleIO::hsr_gen_fname("data-", 0, false, 0, 1); + expect_fname_eq("data-1g1_nao.csr", fname); +} + +// --------------------------------------------------------------------------- +// sr_gen_fname: S(R) spin-independent filename +// --------------------------------------------------------------------------- + +TEST(HsrWriterFname, SrOverwriteWithStep) +{ + // append=false, istep=0 -> "srg1_nao.csr" + const std::string fname = ModuleIO::sr_gen_fname(false, 0); + expect_fname_eq("srg1_nao.csr", fname); +} + +TEST(HsrWriterFname, SrAppendSuppressesStep) +{ + // append=true -> "sr_nao.csr" (no step number) + const std::string fname = ModuleIO::sr_gen_fname(true, 5); + expect_fname_eq("sr_nao.csr", fname); +} + +TEST(HsrWriterFname, SrNegativeStepSuppressesStep) +{ + // istep=-1 -> "sr_nao.csr" + const std::string fname = ModuleIO::sr_gen_fname(false, -1); + expect_fname_eq("sr_nao.csr", fname); +} + +TEST(HsrWriterFname, SrDatExtension) +{ + // out_type=2 -> ".dat" + const std::string fname = ModuleIO::sr_gen_fname(false, 0, 2); + expect_fname_eq("srg1_nao.dat", fname); +} + +// --------------------------------------------------------------------------- +// dhr_gen_fname: dH/dR spin-dependent filename +// --------------------------------------------------------------------------- + +TEST(HsrWriterFname, DhrOverwriteWithStep) +{ + // prefix="H", ispin=0, append=false, istep=0 + // -> "Hrs1g1_nao.csr" + const std::string fname = ModuleIO::dhr_gen_fname("H", 0, false, 0); + expect_fname_eq("Hrs1g1_nao.csr", fname); +} + +TEST(HsrWriterFname, DhrAppendSuppressesStep) +{ + // append=true -> step omitted + const std::string fname = ModuleIO::dhr_gen_fname("H", 1, true, 3); + expect_fname_eq("Hrs2_nao.csr", fname); +} + +TEST(HsrWriterFname, DhrNegativeStepSuppressesStep) +{ + // istep=-1 -> step omitted + const std::string fname = ModuleIO::dhr_gen_fname("dH", 0, false, -1); + expect_fname_eq("dHrs1_nao.csr", fname); +} + +// --------------------------------------------------------------------------- +// write_hcontainer_csr: CSR text format layout +// --------------------------------------------------------------------------- + +TEST(HsrWriterIo, HContainerCsrHeaderKeepsCurrentFormat) +{ + const std::string filename = "write_hs_r_header_h.csr"; + std::remove(filename.c_str()); + + UnitCell ucell; + init_unitcell(ucell); + Parallel_Orbitals pv; + init_serial_orbitals(pv); + hamilt::HContainer matrix(&pv); + double values[4] = {1.0, 0.0, 0.5, 2.0}; + fill_matrix(matrix, pv, values); + + ModuleIO::write_hcontainer_csr(filename, &ucell, 5, &matrix, 0, 0, 1, "H", ""); + + const std::string output = read_file(filename); + EXPECT_THAT(output, testing::HasSubstr(" --- Ionic Step 1 ---\n")); + EXPECT_THAT(output, testing::HasSubstr(" # print H matrix in real space H(R)\n")); + EXPECT_THAT(output, testing::HasSubstr(" 1 # number of spin directions\n")); + EXPECT_THAT(output, testing::HasSubstr(" 1 # spin index\n")); + EXPECT_THAT(output, testing::HasSubstr(" 2 # number of localized basis\n")); + EXPECT_THAT(output, testing::HasSubstr(" 1 # number of Bravais lattice vector R\n")); + EXPECT_THAT(output, testing::HasSubstr(" user_defined_lattice\n")); + EXPECT_THAT(output, testing::HasSubstr(" Si\n")); + EXPECT_THAT(output, testing::HasSubstr(" Direct\n")); + EXPECT_THAT(output, testing::HasSubstr(" 0 0.25 0.5\n")); + EXPECT_THAT(output, testing::HasSubstr(" # CSR Format #\n")); + EXPECT_THAT(output, testing::HasSubstr(" 0 0 0 3\n")); + EXPECT_THAT(output, testing::HasSubstr(" # CSR values\n")); + EXPECT_THAT(output, testing::HasSubstr(" 1.00000e+00 5.00000e-01 2.00000e+00")); + EXPECT_THAT(output, testing::Not(testing::HasSubstr("# representation:"))); + + std::remove(filename.c_str()); +} + +TEST(HsrWriterIo, GammaFoldedHeaderKeepsCsrReadable) +{ + const std::string filename = "write_hs_r_gamma_folded.csr"; + const std::string representation_note + = "gamma-only folded matrix; stored R-space contributions are summed into R = (0, 0, 0)"; + std::remove(filename.c_str()); + + UnitCell ucell; + init_unitcell(ucell); + Parallel_Orbitals pv; + init_serial_orbitals(pv); + hamilt::HContainer matrix(&pv); + double values[4] = {1.0, 0.0, 0.5, 2.0}; + fill_matrix(matrix, pv, values); + + ModuleIO::write_hcontainer_csr( + filename, &ucell, 5, &matrix, 0, 0, 1, "H", representation_note); + + const std::string output = read_file(filename); + EXPECT_THAT(output, testing::HasSubstr("# representation: " + representation_note + "\n")); + EXPECT_THAT(output, testing::HasSubstr(" 1 # number of Bravais lattice vector R\n")); + EXPECT_THAT(output, testing::HasSubstr(" 0 0 0 3\n")); + + ModuleIO::csrFileReader reader(filename); + ASSERT_EQ(reader.getNumberOfR(), 1); + EXPECT_EQ(reader.getMatrixDimension(), 2); + EXPECT_EQ(reader.getRCoordinate(0), std::vector({0, 0, 0})); + + std::remove(filename.c_str()); +} + +TEST(HsrWriterIo, HContainerCsrAppendKeepsCurrentStepSections) +{ + const std::string filename = "write_hs_r_append_s.csr"; + std::remove(filename.c_str()); + + UnitCell ucell; + init_unitcell(ucell); + Parallel_Orbitals pv; + init_serial_orbitals(pv); + hamilt::HContainer matrix(&pv); + double values[4] = {1.0, 0.0, 0.0, 1.0}; + fill_matrix(matrix, pv, values); + + ModuleIO::write_hcontainer_csr(filename, &ucell, 4, &matrix, 0, 0, 1, "S", ""); + ModuleIO::write_hcontainer_csr(filename, &ucell, 4, &matrix, 1, 0, 1, "S", ""); + + const std::string output = read_file(filename); + EXPECT_EQ(count_substr(output, " --- Ionic Step "), 2); + EXPECT_THAT(output, testing::HasSubstr(" --- Ionic Step 1 ---\n")); + EXPECT_THAT(output, testing::HasSubstr(" --- Ionic Step 2 ---\n")); + EXPECT_EQ(count_substr(output, " # print S matrix in real space S(R)\n"), 2); + + std::remove(filename.c_str()); +} + +// --------------------------------------------------------------------------- +// write_hcontainer_csr_binary: native binary layout +// --------------------------------------------------------------------------- + +TEST(HsrWriterIo, HContainerBinaryWritesSortedAndEmptyRBlocks) +{ + const std::string filename = "write_hs_r_native_binary_real.dat"; + std::remove(filename.c_str()); + + Parallel_Orbitals pv; + init_serial_orbitals(pv); + hamilt::HContainer matrix(&pv); + double empty_values[4] = {1e-12, 0.0, 0.0, -1e-12}; + double values[4] = {1.0, 1e-12, 0.0, -2.0}; + fill_matrix_at_R(matrix, pv, 1, 0, 0, empty_values); + fill_matrix_at_R(matrix, pv, -1, 0, 0, values); + + ModuleIO::write_hcontainer_csr_binary(filename, &matrix, -1, false); + + std::ifstream ifs(filename.c_str(), std::ios::binary); + ASSERT_TRUE(ifs.is_open()); + const NativeDoubleRecord record = read_native_double_record(ifs); + EXPECT_EQ(record.step, 0); + EXPECT_EQ(record.nbasis, 2); + ASSERT_EQ(record.blocks.size(), 2); + EXPECT_EQ(record.blocks[0].rx, -1); + EXPECT_EQ(record.blocks[0].ry, 0); + EXPECT_EQ(record.blocks[0].rz, 0); + EXPECT_THAT(record.blocks[0].values, testing::ElementsAre(1.0, -2.0)); + EXPECT_THAT(record.blocks[0].columns, testing::ElementsAre(0, 1)); + EXPECT_THAT(record.blocks[0].row_ptr, testing::ElementsAre(0, 1, 2)); + EXPECT_EQ(record.blocks[1].rx, 1); + EXPECT_EQ(record.blocks[1].ry, 0); + EXPECT_EQ(record.blocks[1].rz, 0); + EXPECT_TRUE(record.blocks[1].values.empty()); + EXPECT_TRUE(record.blocks[1].columns.empty()); + EXPECT_THAT(record.blocks[1].row_ptr, testing::ElementsAre(0, 0, 0)); + EXPECT_EQ(ifs.peek(), std::ifstream::traits_type::eof()); + + std::remove(filename.c_str()); +} + +TEST(HsrWriterIo, HContainerBinaryWritesComplexValuesAsDoublePairs) +{ + const std::string filename = "write_hs_r_native_binary_complex.dat"; + std::remove(filename.c_str()); + + Parallel_Orbitals pv; + init_serial_orbitals(pv); + hamilt::HContainer> matrix(&pv); + std::complex values[4] = { + std::complex(1.0, 2.0), + std::complex(0.0, 0.0), + std::complex(0.0, 0.0), + std::complex(-3.0, 4.0), + }; + fill_matrix_at_R(matrix, pv, 0, 0, 0, values); + + ModuleIO::write_hcontainer_csr_binary(filename, &matrix, 4, false); + + std::ifstream ifs(filename.c_str(), std::ios::binary); + ASSERT_TRUE(ifs.is_open()); + EXPECT_EQ(read_binary_value(ifs), 4); + EXPECT_EQ(read_binary_value(ifs), 2); + EXPECT_EQ(read_binary_value(ifs), 1); + EXPECT_EQ(read_binary_value(ifs), 0); + EXPECT_EQ(read_binary_value(ifs), 0); + EXPECT_EQ(read_binary_value(ifs), 0); + EXPECT_EQ(read_binary_value(ifs), 2); + EXPECT_DOUBLE_EQ(read_binary_value(ifs), 1.0); + EXPECT_DOUBLE_EQ(read_binary_value(ifs), 2.0); + EXPECT_DOUBLE_EQ(read_binary_value(ifs), -3.0); + EXPECT_DOUBLE_EQ(read_binary_value(ifs), 4.0); + EXPECT_EQ(read_binary_value(ifs), 0); + EXPECT_EQ(read_binary_value(ifs), 1); + EXPECT_EQ(read_binary_value(ifs), 0); + EXPECT_EQ(read_binary_value(ifs), 1); + EXPECT_EQ(read_binary_value(ifs), 2); + EXPECT_EQ(ifs.peek(), std::ifstream::traits_type::eof()); + + std::remove(filename.c_str()); +} + +TEST(HsrWriterIo, HContainerBinaryAppendsAndCanOverwrite) +{ + const std::string filename = "write_hs_r_native_binary_append.dat"; + std::remove(filename.c_str()); + + Parallel_Orbitals pv; + init_serial_orbitals(pv); + hamilt::HContainer matrix(&pv); + double values[4] = {1.0, 0.0, 0.0, 2.0}; + fill_matrix(matrix, pv, values); + + ModuleIO::write_hcontainer_csr_binary(filename, &matrix, 0, true); + ModuleIO::write_hcontainer_csr_binary(filename, &matrix, 1, true); + + std::ifstream appended(filename.c_str(), std::ios::binary); + ASSERT_TRUE(appended.is_open()); + EXPECT_EQ(read_native_double_record(appended).step, 0); + EXPECT_EQ(read_native_double_record(appended).step, 1); + EXPECT_EQ(appended.peek(), std::ifstream::traits_type::eof()); + appended.close(); + + ModuleIO::write_hcontainer_csr_binary(filename, &matrix, 2, false); + std::ifstream replaced(filename.c_str(), std::ios::binary); + ASSERT_TRUE(replaced.is_open()); + EXPECT_EQ(read_native_double_record(replaced).step, 2); + EXPECT_EQ(replaced.peek(), std::ifstream::traits_type::eof()); + + std::remove(filename.c_str()); +} + +TEST(HsrWriterIo, HContainerBinaryMpiGatherWritesCompleteFiles) +{ +#ifndef __MPI + GTEST_SKIP() << "MPI support is required."; +#else + int mpi_size = 0; + int mpi_rank = 0; + MPI_Comm_size(MPI_COMM_WORLD, &mpi_size); + MPI_Comm_rank(MPI_COMM_WORLD, &mpi_rank); + if (mpi_size != 2) + { + GTEST_SKIP() << "This test requires exactly two MPI ranks."; + } + + const std::string hr_filename = "hrs1_nao.dat"; + const std::string sr_filename = "sr_nao.dat"; + if (mpi_rank == 0) + { + std::remove(hr_filename.c_str()); + std::remove(sr_filename.c_str()); + } + MPI_Barrier(MPI_COMM_WORLD); + + UnitCell ucell; + ucell.ntype = 1; + ucell.nat = 1; + ucell.atoms = new Atom[1]; + ucell.set_atom_flag = true; + ucell.atoms[0].na = 1; + ucell.atoms[0].nw = 2; + ucell.iat2it = new int[1]; + ucell.iat2it[0] = 0; + ucell.set_iat2iwt(1); + const int* iat2iwt = ucell.get_iat2iwt(); + Parallel_Orbitals serial_pv; + serial_pv.set_serial(2, 2); + serial_pv.set_atomic_trace(iat2iwt, 1, 2); + Parallel_Orbitals parallel_pv; + ASSERT_EQ(parallel_pv.init(2, 2, 1, MPI_COMM_WORLD), 0); + parallel_pv.set_atomic_trace(iat2iwt, 1, 2); + + hamilt::HContainer hr_serial(ucell, &serial_pv); + hamilt::HContainer sr_serial(ucell, &serial_pv); + double hr_values[4] = {1.0, 0.5, 0.0, 2.0}; + double sr_values[4] = {1.0, 0.0, 0.0, 1.0}; + if (mpi_rank == 0) + { + std::copy(hr_values, hr_values + 4, hr_serial.get_atom_pair(0).get_pointer(0)); + std::copy(sr_values, sr_values + 4, sr_serial.get_atom_pair(0).get_pointer(0)); + } + + hamilt::HContainer hr_parallel(ucell, ¶llel_pv); + hamilt::HContainer sr_parallel(ucell, ¶llel_pv); + hamilt::transferSerial2Parallels(hr_serial, &hr_parallel, 0); + hamilt::transferSerial2Parallels(sr_serial, &sr_parallel, 0); + + init_sparse_output_globals(); + std::vector*> hr_vec(1, &hr_parallel); + ModuleIO::write_hsr( + hr_vec, &sr_parallel, &ucell, 2, 8, parallel_pv, true, true, iat2iwt, 1, 0, "./"); + MPI_Barrier(MPI_COMM_WORLD); + + if (mpi_rank == 0) + { + std::ifstream hr_stream(hr_filename.c_str(), std::ios::binary); + ASSERT_TRUE(hr_stream.is_open()); + const NativeDoubleRecord hr_record = read_native_double_record(hr_stream); + ASSERT_EQ(hr_record.blocks.size(), 1); + EXPECT_THAT(hr_record.blocks[0].values, testing::ElementsAre(1.0, 0.5, 2.0)); + EXPECT_THAT(hr_record.blocks[0].columns, testing::ElementsAre(0, 1, 1)); + EXPECT_THAT(hr_record.blocks[0].row_ptr, testing::ElementsAre(0, 2, 3)); + + std::ifstream sr_stream(sr_filename.c_str(), std::ios::binary); + ASSERT_TRUE(sr_stream.is_open()); + const NativeDoubleRecord sr_record = read_native_double_record(sr_stream); + ASSERT_EQ(sr_record.blocks.size(), 1); + EXPECT_THAT(sr_record.blocks[0].values, testing::ElementsAre(1.0, 1.0)); + EXPECT_THAT(sr_record.blocks[0].columns, testing::ElementsAre(0, 1)); + EXPECT_THAT(sr_record.blocks[0].row_ptr, testing::ElementsAre(0, 1, 2)); + + std::remove(hr_filename.c_str()); + std::remove(sr_filename.c_str()); + } + delete[] ucell.atoms; + ucell.atoms = nullptr; + ucell.set_atom_flag = false; +#endif +} + +int main(int argc, char** argv) +{ +#ifdef __MPI + MPI_Init(&argc, &argv); + MPI_Comm_size(MPI_COMM_WORLD, &GlobalV::NPROC); + MPI_Comm_rank(MPI_COMM_WORLD, &GlobalV::MY_RANK); +#endif + + ::testing::InitGoogleTest(&argc, argv); + const int result = RUN_ALL_TESTS(); + +#ifdef __MPI + MPI_Finalize(); +#endif + return result; +} diff --git a/source/source_io/module_hs/unittests/test_lat_r_csr.cpp b/source/source_io/module_hs/unittests/test_lat_r_csr.cpp new file mode 100644 index 00000000000..0a272fc137a --- /dev/null +++ b/source/source_io/module_hs/unittests/test_lat_r_csr.cpp @@ -0,0 +1,223 @@ +#include "gtest/gtest.h" +#include "gmock/gmock.h" +#include "source_io/module_hs/lat_r_csr.h" +#include "source_base/global_variable.h" +#include "source_basis/module_ao/parallel_orbitals.h" +#include +#include +#include +#include +#include +/************************************************ + * unit test of save_lat_r + ***********************************************/ +/** + * - Tested Functions: + * - ModuleIO::save_lat_r + * - write one lattice-R block in CSR format + */ +Parallel_Orbitals::Parallel_Orbitals() +{ +} + +Parallel_Orbitals::~Parallel_Orbitals() +{ +} + +Parallel_2D::~Parallel_2D() +{ +} + +void Parallel_2D::set_serial(const int M_A, const int N_A) +{ + this->nrow = M_A; + this->ncol = N_A; + this->is_serial = true; + this->global2local_row_.resize(M_A); + this->global2local_row_[0] = 0; + this->global2local_row_[1] = 1; + this->global2local_row_[2] = -1; + this->global2local_row_[3] = 2; + this->global2local_row_[4] = -1; //Some rows have global2local_row_ < 0 +} + +int Parallel_2D::get_global_row_size() const +{ + return this->nrow; +} + +TEST(ModuleIOTest, SaveLatR) +{ + // Create temporary output file + std::stringstream ofs_filename; + GlobalV::DRANK=0; + ofs_filename << "test_save_lat_r_" << GlobalV::DRANK << ".dat"; + std::ofstream ofs(ofs_filename.str()); + + // Define input parameters + const double sparse_threshold = 1e-8; + const bool binary = false; + Parallel_Orbitals pv; + pv.set_serial(5, 5); + std::map> XR = { + {0, {{1, 0.5}, {3, 0.3}}}, + {1, {{0, 0.2}, {2, 0.4}}}, + {3, {{1, 0.1}, {4, 0.7}}} + }; + ModuleIO::SparseWriteOptions options; + options.threshold = sparse_threshold; + options.binary = binary; + options.reduce = true; + + // Call function under test + ModuleIO::save_lat_r(ofs, XR, pv, options); + + // Close output file and open it for reading + ofs.close(); + std::ifstream ifs; + ifs.open("test_save_lat_r_0.dat"); + std::string str((std::istreambuf_iterator(ifs)),std::istreambuf_iterator()); + std::istringstream content(str); + std::string value_line; + std::string column_line; + std::string indptr_line; + std::getline(content, value_line); + std::getline(content, column_line); + std::getline(content, indptr_line); + + std::istringstream value_stream(value_line); + std::vector values; + double value = 0.0; + while (value_stream >> value) + { + values.push_back(value); + } + ASSERT_EQ(values.size(), 6); + EXPECT_DOUBLE_EQ(values[0], 0.5); + EXPECT_DOUBLE_EQ(values[1], 0.3); + EXPECT_DOUBLE_EQ(values[2], 0.2); + EXPECT_DOUBLE_EQ(values[3], 0.4); + EXPECT_DOUBLE_EQ(values[4], 0.1); + EXPECT_DOUBLE_EQ(values[5], 0.7); + + std::istringstream column_stream(column_line); + std::vector columns; + int column = 0; + while (column_stream >> column) + { + columns.push_back(column); + } + EXPECT_THAT(columns, testing::ElementsAre(1, 3, 0, 2, 1, 4)); + + std::istringstream indptr_stream(indptr_line); + std::vector indptr; + long long index = 0; + while (indptr_stream >> index) + { + indptr.push_back(index); + } + EXPECT_THAT(indptr, testing::ElementsAre(0, 2, 4, 4, 6, 6)); + std::remove("test_save_lat_r_0.dat"); +} + +TEST(ModuleIOTest, SaveLatRComplexKeepsHighPrecision) +{ + const std::string filename = "test_save_lat_r_complex.dat"; + std::remove(filename.c_str()); + GlobalV::DRANK = 0; + std::ofstream ofs(filename); + + Parallel_Orbitals pv; + pv.set_serial(5, 5); + ModuleIO::SparseRBlock> XR = { + {0, {{1, std::complex(1.234567890123456, -2.345678901234567)}}} + }; + ModuleIO::SparseWriteOptions options; + options.threshold = 1e-12; + options.binary = false; + options.reduce = false; + + ModuleIO::save_lat_r(ofs, XR, pv, options); + ofs.close(); + + std::ifstream ifs(filename); + const std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + EXPECT_THAT(output, testing::HasSubstr("(1.2345678901234560e+00,-2.3456789012345669e+00)")); + EXPECT_THAT(output, testing::Not(testing::HasSubstr("(1.23457,-2.34568)"))); + + std::remove(filename.c_str()); +} + +TEST(ModuleIOTest, SaveLatRUsesConfiguredPrecision) +{ + const std::string filename = "test_save_lat_r_precision.dat"; + std::remove(filename.c_str()); + GlobalV::DRANK = 0; + std::ofstream ofs(filename); + + Parallel_Orbitals pv; + pv.set_serial(5, 5); + ModuleIO::SparseRBlock> XR = { + {0, {{1, std::complex(1.234567890123456, -2.5)}}} + }; + ModuleIO::SparseWriteOptions options; + options.threshold = 1e-12; + options.binary = false; + options.precision = 8; + options.reduce = false; + + ModuleIO::save_lat_r(ofs, XR, pv, options); + ofs.close(); + + std::ifstream ifs(filename); + const std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + EXPECT_THAT(output, testing::HasSubstr("(1.23456789e+00,-2.50000000e+00)")); + EXPECT_THAT(output, testing::Not(testing::HasSubstr("1.2345678901234560e+00"))); + + std::remove(filename.c_str()); +} + +void write_out_of_range_sparse_column(const char* filename) +{ + GlobalV::DRANK = 0; + std::ofstream ofs(filename); + Parallel_Orbitals pv; + pv.set_serial(5, 5); + ModuleIO::SparseRBlock XR; + XR[0][5] = 1.0; + ModuleIO::SparseWriteOptions options; + options.threshold = 1e-12; + options.binary = false; + options.reduce = false; + ModuleIO::save_lat_r(ofs, XR, pv, options); +} + +TEST(ModuleIOTest, SaveLatRRejectsOutOfRangeColumn) +{ + const char* filename = "/tmp/test_save_lat_r_invalid.dat"; + std::remove(filename); + EXPECT_EXIT( + write_out_of_range_sparse_column(filename), + ::testing::ExitedWithCode(1), + "Sparse column index out of range"); + std::remove(filename); +} + +int main(int argc, char **argv) +{ + +#ifdef __MPI + MPI_Init(&argc, &argv); + MPI_Comm_size(MPI_COMM_WORLD,&GlobalV::NPROC); + MPI_Comm_rank(MPI_COMM_WORLD,&GlobalV::MY_RANK); +#endif + + testing::InitGoogleTest(&argc, argv); + int result = RUN_ALL_TESTS(); + +#ifdef __MPI + MPI_Finalize(); +#endif + + return result; +} diff --git a/source/source_io/module_hs/unittests/test_output_mat_sparse.cpp b/source/source_io/module_hs/unittests/test_output_mat_sparse.cpp new file mode 100644 index 00000000000..0832d5c9535 --- /dev/null +++ b/source/source_io/module_hs/unittests/test_output_mat_sparse.cpp @@ -0,0 +1,27 @@ +/** + * @file test_output_mat_sparse.cpp + * @brief Unit test for output_mat_sparse.h: MatSparseOutputOptions defaults. + * + * The struct is the single source of truth for the sparse-matrix output + * switches; the test pins the legacy default values so an accidental + * default change is caught at compile-and-run time instead of silently + * changing output files. + */ +#include + +#include "source_io/module_hs/output_mat_sparse.h" + +TEST(OutputMatSparse, OptionsKeepLegacyDefaults) +{ + ModuleIO::MatSparseOutputOptions options; + EXPECT_FALSE(options.out_mat_dh); + EXPECT_FALSE(options.out_mat_ds); + EXPECT_FALSE(options.out_mat_t); + EXPECT_FALSE(options.out_mat_r); + EXPECT_EQ(options.dh_precision, 16); + EXPECT_EQ(options.ds_precision, 16); + EXPECT_EQ(options.t_precision, 16); + EXPECT_EQ(options.r_precision, 16); + EXPECT_DOUBLE_EQ(options.sparse_threshold, 1e-10); + EXPECT_FALSE(options.binary); +} diff --git a/source/source_io/module_hs/unittests/test_pos_op_csr.cpp b/source/source_io/module_hs/unittests/test_pos_op_csr.cpp new file mode 100644 index 00000000000..4ece2398d5f --- /dev/null +++ b/source/source_io/module_hs/unittests/test_pos_op_csr.cpp @@ -0,0 +1,178 @@ +#include "gmock/gmock.h" +#include "gtest/gtest.h" + +#include "source_io/module_hs/pos_op_csr.h" + +#include +#include +#include + +namespace +{ +std::vector read_lines(const std::string& filename) +{ + std::ifstream ifs(filename.c_str()); + std::vector lines; + std::string line; + while (std::getline(ifs, line)) + { + lines.push_back(line); + } + return lines; +} + +template +T read_binary_value(std::ifstream& ifs) +{ + T value{}; + ifs.read(reinterpret_cast(&value), sizeof(T)); + return value; +} +} // namespace + +TEST(PosOpCsr, LatRNonemptySkipsEmptyBlocks) +{ + int empty_counts[3] = {0, 0, 0}; + int x_only_counts[3] = {1, 0, 0}; + int z_only_counts[3] = {0, 0, 2}; + + EXPECT_FALSE(ModuleIO::detail::lat_r_nonempty(empty_counts)); + EXPECT_TRUE(ModuleIO::detail::lat_r_nonempty(x_only_counts)); + EXPECT_TRUE(ModuleIO::detail::lat_r_nonempty(z_only_counts)); +} + +TEST(PosOpCsr, CsrAssembleTextAllowsZeroBlocks) +{ + const std::string payload_filename = "rr_empty_payload.tmp"; + const std::string output_filename = "rr_empty.csr"; + std::remove(payload_filename.c_str()); + std::remove(output_filename.c_str()); + + std::ofstream payload(payload_filename.c_str()); + payload.close(); + + ModuleIO::detail::assemble_csr(output_filename, + payload_filename, + 9, + 2, + 0, + false, + false, + "PosOpCsr"); + + const std::vector lines = read_lines(output_filename); + ASSERT_EQ(lines.size(), 3); + EXPECT_EQ(lines[0], "STEP: 9"); + EXPECT_EQ(lines[1], "Matrix Dimension of r(R): 2"); + EXPECT_EQ(lines[2], "Matrix number of r(R): 0"); + + std::remove(payload_filename.c_str()); + std::remove(output_filename.c_str()); +} + +TEST(PosOpCsr, CsrAssembleTextKeepsSingleDirectionPayload) +{ + const std::string payload_filename = "rr_single_direction_payload.tmp"; + const std::string output_filename = "rr_single_direction.csr"; + std::remove(payload_filename.c_str()); + std::remove(output_filename.c_str()); + + std::ofstream payload(payload_filename.c_str()); + payload << "1 0 -1\n"; + payload << "1\n"; + payload << " 4.00000000e+00\n"; + payload << " 0\n"; + payload << "0 1\n"; + payload << "0\n"; + payload << "0\n"; + payload.close(); + + ModuleIO::detail::assemble_csr(output_filename, + payload_filename, + 10, + 2, + 1, + false, + false, + "PosOpCsr"); + + const std::vector lines = read_lines(output_filename); + ASSERT_EQ(lines.size(), 10); + EXPECT_EQ(lines[0], "STEP: 10"); + EXPECT_EQ(lines[1], "Matrix Dimension of r(R): 2"); + EXPECT_EQ(lines[2], "Matrix number of r(R): 1"); + EXPECT_EQ(lines[3], "1 0 -1"); + EXPECT_EQ(lines[4], "1"); + EXPECT_EQ(lines[5], " 4.00000000e+00"); + EXPECT_EQ(lines[8], "0"); + EXPECT_EQ(lines[9], "0"); + + std::remove(payload_filename.c_str()); + std::remove(output_filename.c_str()); +} + +TEST(PosOpCsr, CsrAssembleBinaryKeepsHeaderAndPayloadOrder) +{ + const std::string payload_filename = "rr_binary_payload.tmp"; + const std::string output_filename = "rr_binary.csr"; + std::remove(payload_filename.c_str()); + std::remove(output_filename.c_str()); + + std::ofstream payload(payload_filename.c_str(), std::ios::binary); + int dRx = 1; + int dRy = 2; + int dRz = 3; + int x_count = 1; + int y_count = 0; + int z_count = 0; + double value = 4.0; + int column = 1; + long long ptr0 = 0; + long long ptr1 = 1; + payload.write(reinterpret_cast(&dRx), sizeof(int)); + payload.write(reinterpret_cast(&dRy), sizeof(int)); + payload.write(reinterpret_cast(&dRz), sizeof(int)); + payload.write(reinterpret_cast(&x_count), sizeof(int)); + payload.write(reinterpret_cast(&value), sizeof(double)); + payload.write(reinterpret_cast(&column), sizeof(int)); + payload.write(reinterpret_cast(&ptr0), sizeof(long long)); + payload.write(reinterpret_cast(&ptr1), sizeof(long long)); + payload.write(reinterpret_cast(&y_count), sizeof(int)); + payload.write(reinterpret_cast(&z_count), sizeof(int)); + payload.close(); + + ModuleIO::detail::assemble_csr(output_filename, + payload_filename, + 11, + 2, + 1, + true, + false, + "PosOpCsr"); + + std::ifstream ifs(output_filename, std::ios::binary); + ASSERT_TRUE(ifs.is_open()); + EXPECT_EQ(read_binary_value(ifs), 11); + EXPECT_EQ(read_binary_value(ifs), 2); + EXPECT_EQ(read_binary_value(ifs), 1); + EXPECT_EQ(read_binary_value(ifs), 1); + EXPECT_EQ(read_binary_value(ifs), 2); + EXPECT_EQ(read_binary_value(ifs), 3); + EXPECT_EQ(read_binary_value(ifs), 1); + EXPECT_DOUBLE_EQ(read_binary_value(ifs), 4.0); + EXPECT_EQ(read_binary_value(ifs), 1); + EXPECT_EQ(read_binary_value(ifs), 0); + EXPECT_EQ(read_binary_value(ifs), 1); + EXPECT_EQ(read_binary_value(ifs), 0); + EXPECT_EQ(read_binary_value(ifs), 0); + ifs.close(); + + std::remove(payload_filename.c_str()); + std::remove(output_filename.c_str()); +} + +int main(int argc, char** argv) +{ + ::testing::InitGoogleTest(&argc, argv); + return RUN_ALL_TESTS(); +} diff --git a/source/source_io/module_hs/unittests/test_vxc_op_tools.cpp b/source/source_io/module_hs/unittests/test_vxc_op_tools.cpp new file mode 100644 index 00000000000..cf5213b27ea --- /dev/null +++ b/source/source_io/module_hs/unittests/test_vxc_op_tools.cpp @@ -0,0 +1,197 @@ +/** + * @file test_vxc_op_tools.cpp + * @brief Unit tests for vxc_op_tools.cpp helper functions + * + * Tests the functions that do not require a full Parallel_Orbitals + * / ScaLAPACK context: + * - get_real (double and complex overloads) + * - set_para2d_MO (non-MPI set_serial path) + * - orbital_energy (serial Parallel_2D, double and complex) + * - all_band_energy (serial Parallel_2D, double and complex) + * + * cVc is not tested here because it requires pv.desc / pv.desc_wfc + * (full Parallel_Orbitals with BLACS context). + * write_orb_energy is not tested here because it requires K_Vectors + * (vtable pulls in klist.cpp -> symmetry.cpp dependency chain). + */ +#include + +#include "source_io/module_hs/vxc_op_tools.h" +#include "source_base/parallel_2d.h" +#include "source_base/parallel_comm.h" +#include "source_base/matrix.h" + +#include +#include +#ifdef __MPI +#include +#endif + +namespace +{ +/** + * @brief Create a serial Parallel_2D that owns all rows/cols on rank 0. + */ +Parallel_2D make_serial_pv(int dim) +{ + Parallel_2D pv; + pv.set_serial(dim, dim); + return pv; +} +} // namespace + +// --------------------------------------------------------------------------- +// get_real: extract real part +// --------------------------------------------------------------------------- + +TEST(VxcOpTools, GetRealDouble) +{ + const double d = 3.14; + EXPECT_DOUBLE_EQ(3.14, ModuleIO::get_real(d)); +} + +TEST(VxcOpTools, GetRealComplex) +{ + const std::complex c(1.5, 2.5); + EXPECT_DOUBLE_EQ(1.5, ModuleIO::get_real(c)); +} + +// --------------------------------------------------------------------------- +// set_para2d_MO: set up MO-space 2D distribution +// --------------------------------------------------------------------------- + +TEST(VxcOpTools, SetPara2dMO) +{ + const int nbands = 4; + Parallel_Orbitals pv; +#ifdef __MPI + ASSERT_EQ(pv.init(nbands, nbands, 1, DIAG_WORLD), 0); +#else + pv.set_serial(nbands, nbands); +#endif + Parallel_2D p2d; + ModuleIO::set_para2d_MO(pv, nbands, p2d); + EXPECT_EQ(p2d.get_global_row_size(), nbands); + EXPECT_EQ(p2d.get_global_col_size(), nbands); +} + +// --------------------------------------------------------------------------- +// orbital_energy: extract diagonal from MO matrix +// --------------------------------------------------------------------------- + +TEST(VxcOpTools, OrbitalEnergyDouble) +{ + const int nbands = 3; + Parallel_2D p2d = make_serial_pv(nbands); + // MO matrix (row-major, p2d local): 3x3 diagonal + // mat_mo[i * row_size + j] where row_size = get_row_size() + const int row_size = p2d.get_row_size(); + std::vector mat_mo(row_size * p2d.get_col_size(), 0.0); + // Set diagonal: 1.0, 2.0, 3.0 + mat_mo[0 * row_size + 0] = 1.0; + mat_mo[1 * row_size + 1] = 2.0; + mat_mo[2 * row_size + 2] = 3.0; + + const std::vector e = ModuleIO::orbital_energy(0, nbands, mat_mo, p2d); + ASSERT_EQ(e.size(), static_cast(nbands)); + EXPECT_DOUBLE_EQ(1.0, e[0]); + EXPECT_DOUBLE_EQ(2.0, e[1]); + EXPECT_DOUBLE_EQ(3.0, e[2]); +} + +TEST(VxcOpTools, OrbitalEnergyComplex) +{ + const int nbands = 2; + Parallel_2D p2d = make_serial_pv(nbands); + const int row_size = p2d.get_row_size(); + std::vector> mat_mo(row_size * p2d.get_col_size()); + mat_mo[0 * row_size + 0] = std::complex(1.5, 0.0); + mat_mo[1 * row_size + 1] = std::complex(3.5, 0.0); + + const std::vector e = ModuleIO::orbital_energy(0, nbands, mat_mo, p2d); + ASSERT_EQ(e.size(), static_cast(nbands)); + EXPECT_DOUBLE_EQ(1.5, e[0]); + EXPECT_DOUBLE_EQ(3.5, e[1]); +} + +// --------------------------------------------------------------------------- +// all_band_energy: weighted sum of diagonal +// --------------------------------------------------------------------------- + +TEST(VxcOpTools, AllBandEnergyDouble) +{ + const int nbands = 3; + const int ik = 0; + Parallel_2D p2d = make_serial_pv(nbands); + const int row_size = p2d.get_row_size(); + std::vector mat_mo(row_size * p2d.get_col_size(), 0.0); + mat_mo[0 * row_size + 0] = 1.0; + mat_mo[1 * row_size + 1] = 2.0; + mat_mo[2 * row_size + 2] = 3.0; + + // wg: occupation weights (nks x nbands) + ModuleBase::matrix wg(1, nbands); + wg(0, 0) = 1.0; + wg(0, 1) = 0.5; + wg(0, 2) = 0.0; + + const double e = ModuleIO::all_band_energy(ik, mat_mo, p2d, wg); + // 1.0*1.0 + 2.0*0.5 + 3.0*0.0 = 2.0 + EXPECT_NEAR(2.0, e, 1e-12); +} + +TEST(VxcOpTools, AllBandEnergyComplex) +{ + const int nbands = 2; + const int ik = 0; + Parallel_2D p2d = make_serial_pv(nbands); + const int row_size = p2d.get_row_size(); + std::vector> mat_mo(row_size * p2d.get_col_size()); + mat_mo[0 * row_size + 0] = std::complex(2.0, 0.0); + mat_mo[1 * row_size + 1] = std::complex(4.0, 0.0); + + ModuleBase::matrix wg(1, nbands); + wg(0, 0) = 1.0; + wg(0, 1) = 0.25; + + const double e = ModuleIO::all_band_energy(ik, mat_mo, p2d, wg); + // 2.0*1.0 + 4.0*0.25 = 3.0 + EXPECT_NEAR(3.0, e, 1e-12); +} + +// --------------------------------------------------------------------------- +// TODO: cVc — requires full Parallel_Orbitals with BLACS context (pv.desc, +// pv.desc_wfc) for the ScaLAPACK gemm path. To test, either: +// (a) refactor cVc to accept raw matrix pointers + Parallel_2D instead of +// Parallel_Orbitals (governance rule 10: pass explicit arguments), or +// (b) construct a real Parallel_Orbitals via init(dim, dim, 1, DIAG_WORLD) +// and fill desc/desc_wfc — but this also requires valid BLACS context +// and ScaLAPACK linkage. +// --------------------------------------------------------------------------- + +// --------------------------------------------------------------------------- +// TODO: write_orb_energy — requires K_Vectors whose vtable (in klist.cpp + +// reciprocal_grid.cpp) pulls in symmetry.cpp -> module_cell/module_symmetry +// dependency chain. To test, either: +// (a) refactor write_orb_energy to accept int nks instead of const +// K_Vectors& (governance rule 10), or +// (b) add a test-only K_Vectors lightweight subclass that stubs get_nks(). +// --------------------------------------------------------------------------- + +// --------------------------------------------------------------------------- +// Main: MPI_Init for Parallel_Reduce::reduce_all in orbital_energy / all_band_energy +// --------------------------------------------------------------------------- + +int main(int argc, char** argv) +{ +#ifdef __MPI + MPI_Init(&argc, &argv); + DIAG_WORLD = MPI_COMM_WORLD; +#endif + testing::InitGoogleTest(&argc, argv); + const int result = RUN_ALL_TESTS(); +#ifdef __MPI + MPI_Finalize(); +#endif + return result; +} diff --git a/source/source_io/module_hs/vxc_op_lip.cpp b/source/source_io/module_hs/vxc_op_lip.cpp new file mode 100644 index 00000000000..47cc4ba7189 --- /dev/null +++ b/source/source_io/module_hs/vxc_op_lip.cpp @@ -0,0 +1,273 @@ +#include "vxc_op_lip.h" + +#include "source_base/module_out/filename.h" +#include "source_base/parallel_reduce.h" +#include "source_base/module_container/base/third_party/blas.h" +#include "source_io/module_hs/hs_dense_io.h" + +#include +#include +#include + +namespace ModuleIO +{ + +namespace +{ + +template +FPTYPE get_real_lip(const std::complex& c) +{ + return c.real(); +} + +template +FPTYPE get_real_lip(const FPTYPE& d) +{ + return d; +} + +template +std::vector cVc_lip(const T* V, const T* c, int nbasis, int nbands) +{ + std::vector Vc(nbasis * nbands, 0.0); + char transa = 'N'; + char transb = 'N'; + const T alpha(1.0, 0.0); + const T beta(0.0, 0.0); + container::BlasConnector::gemm(transa, transb, nbasis, nbands, nbasis, + alpha, V, nbasis, c, nbasis, beta, Vc.data(), nbasis); + + std::vector cVc(nbands * nbands, 0.0); + transa = ((std::is_same::value || std::is_same::value) ? 'T' : 'C'); + container::BlasConnector::gemm(transa, transb, nbands, nbands, nbasis, + alpha, c, nbasis, Vc.data(), nbasis, beta, cVc.data(), nbands); + return cVc; +} + +template +std::vector> psi_Hpsi(const std::complex* psi, + const std::complex* hpsi, + const int nbasis, + const int nbands) +{ + using T = std::complex; + std::vector cVc(nbands * nbands, static_cast(0.0)); + const T alpha(1.0, 0.0); + const T beta(0.0, 0.0); + container::BlasConnector::gemm('C', 'N', nbands, nbands, nbasis, alpha, + psi, nbasis, hpsi, nbasis, beta, cVc.data(), nbands); + return cVc; +} + +template +std::vector orbital_energy_lip(const int ik, const int nbands, + const std::vector>& mat_mo) +{ +#ifdef __DEBUG + assert(nbands >= 0); +#endif + std::vector e(nbands, 0.0); + for (int i = 0; i < nbands; ++i) + { + e[i] = get_real_lip(mat_mo[i * nbands + i]); + } + return e; +} + +template +FPTYPE all_band_energy_lip(const int ik, const int nbands, + const std::vector>& mat_mo, + const ModuleBase::matrix& wg) +{ + FPTYPE e = 0.0; + for (int i = 0; i < nbands; ++i) + { + e += get_real_lip(mat_mo[i * nbands + i]) * static_cast(wg(ik, i)); + } + return e; +} + +template +FPTYPE all_band_energy_lip(const int ik, const std::vector& orbital_energy, + const ModuleBase::matrix& wg) +{ + FPTYPE e = 0.0; + for (size_t i = 0; i < orbital_energy.size(); ++i) + { + e += orbital_energy[i] * static_cast(wg(ik, i)); + } + return e; +} + +} // anonymous namespace + +template +void write_Vxc_LIP(int nspin, + int naos, + int drank, + const psi::Psi>& psi_pw, + const UnitCell& ucell, + Structure_Factor& sf, + surchem& solvent, + const ModulePW::PW_Basis_K& wfc_basis, + const ModulePW::PW_Basis& rho_basis, + const ModulePW::PW_Basis& rhod_basis, + const ModuleBase::matrix& vloc, + const Charge& chg, + const K_Vectors& kv, + const ModuleBase::matrix& wg, + const bool gamma_only, + const std::string& global_out_dir, + const int out_ndigits, + const std::string& ks_solver, + bool cal_exx, + double hybrid_alpha +#ifdef __EXX + , + const Exx_Lip>& exx_lip +#endif +) +{ + using T = std::complex; + ModuleBase::TITLE("ModuleIO", "write_Vxc_LIP"); + int nbands = wg.nc; + + // 1. real-space xc potential + double etxc = 0.0; + double vtxc = 0.0; + std::unique_ptr potxc( + new elecstate::Potential(&rhod_basis, &rho_basis, &ucell, &vloc, &sf, &solvent, &etxc, &vtxc)); + std::vector compnents_list = {"xc"}; + + potxc->pot_register(compnents_list); + potxc->update_from_charge(&chg, &ucell); + + // 2. allocate xc operator + psi::Psi hpsi_localxc(psi_pw.get_nk(), psi_pw.get_nbands(), psi_pw.get_nbasis(), kv.ngk, true); + hpsi_localxc.zero_out(); + std::unique_ptr>> vxcs_op_pw; + + std::vector> e_orb_locxc; // orbital energy (local XC) + std::vector> e_orb_tot; // orbital energy (total) + std::vector> e_orb_exx; // orbital energy (EXX) + Parallel_2D p2d_serial; + p2d_serial.set_serial(nbands, nbands); + + for (int ik = 0; ik < kv.get_nks(); ++ik) + { + // 2.1 local xc + vxcs_op_pw = std::unique_ptr>>( + new hamilt::Veff>(kv.isk.data(), + potxc->get_veff_smooth_data(), potxc->get_veff_smooth().nr, potxc->get_veff_smooth().nc, &wfc_basis)); + vxcs_op_pw->init(ik); // set k-point index + psi_pw.fix_k(ik); + hpsi_localxc.fix_k(ik); +#ifdef __DEBUG + assert(hpsi_localxc.get_current_nbas() == psi_pw.get_current_nbas()); + assert(hpsi_localxc.get_current_nbas() == hpsi_localxc.get_ngk(ik)); +#endif + vxcs_op_pw->act(psi_pw.get_nbands(), psi_pw.get_nbasis(), psi_pw.get_npol(), &psi_pw(ik, 0, 0), &hpsi_localxc(ik, 0, 0), psi_pw.get_ngk(ik)); + vxcs_op_pw.reset(); + std::vector vxc_local_k_mo = psi_Hpsi(&psi_pw(ik, 0, 0), &hpsi_localxc(ik, 0, 0), psi_pw.get_nbasis(), psi_pw.get_nbands()); + Parallel_Reduce::reduce_pool(vxc_local_k_mo.data(), nbands * nbands); + e_orb_locxc.emplace_back(orbital_energy_lip(ik, nbands, vxc_local_k_mo)); + + // 2.2 exx + std::vector vxc_tot_k_mo(std::move(vxc_local_k_mo)); + std::vector vexx_k_ao(naos * naos); +#if((defined __LCAO)&&(defined __EXX) && !(defined __CUDA)&& !(defined __ROCM)) + if (cal_exx) + { + for (int n = 0; n < naos; ++n) + { + for (int m = 0; m < naos; ++m) + { + vexx_k_ao[n * naos + m] += static_cast(hybrid_alpha) + * exx_lip.get_exx_matrix()[ik][m][n]; + } + } + std::vector vexx_k_mo = cVc_lip(vexx_k_ao.data(), &(exx_lip.get_hvec()(ik, 0, 0)), naos, nbands); + Parallel_Reduce::reduce_pool(vexx_k_mo.data(), nbands * nbands); + e_orb_exx.emplace_back(orbital_energy_lip(ik, nbands, vexx_k_mo)); + container::BlasConnector::axpy(nbands * nbands, 1.0, vexx_k_mo.data(), 1, vxc_tot_k_mo.data(), 1); + } +#endif + + // add-up and write + const int istep = -1; + const int out_label = 1; // 1 means .txt while 2 means .dat + const bool out_app_flag = 0; + std::string vxc_file = ModuleIO::filename_output( + global_out_dir, + "vxc", "nao", ik, kv.ik2iktot, nspin, kv.get_nkstot(), + out_label, out_app_flag, gamma_only, istep); + + ModuleIO::save_mat(istep, vxc_tot_k_mo.data(), nbands, + false, out_ndigits, true, + out_app_flag, vxc_file, + p2d_serial, drank, ks_solver, false); + + e_orb_tot.emplace_back(orbital_energy_lip(ik, nbands, vxc_tot_k_mo)); + } + + // write the orbital energy for xc and exx in LibRPA format + const int nspin0 = (nspin == 2) ? 2 : 1; + auto write_orb_energy_lip = [&kv, &nspin0, &nbands, &global_out_dir](const std::vector>& e_orb, + const std::string& label, + const bool app = false) { + assert(e_orb.size() == kv.get_nks()); + const int nk = kv.get_nks() / nspin0; + std::ofstream ofs; + const std::string out_name = (label == "") ? "out.dat" : label + "_out.dat"; + ofs.open(global_out_dir + "vxc_" + out_name, + app ? std::ios::app : std::ios::out); + ofs << nk << "\n" << nspin0 << "\n" << nbands << "\n"; + ofs << std::scientific << std::setprecision(16); + for (int ik = 0; ik < nk; ++ik) + { + for (int is = 0; is < nspin0; ++is) + { + for (auto e : e_orb[is * nk + ik]) + { // Hartree and eV + ofs << e / 2. << "\t" << e * static_cast(ModuleBase::Ry_to_eV) << "\n"; + } + } + } + }; + + if (GlobalV::MY_RANK == 0) + { + write_orb_energy_lip(e_orb_tot, ""); +#if((defined __LCAO)&&(defined __EXX) && !(defined __CUDA)&& !(defined __ROCM)) + if (cal_exx) + { + write_orb_energy_lip(e_orb_locxc, "local"); + write_orb_energy_lip(e_orb_exx, "exx"); + } +#endif + } +} + +// Explicit template instantiations +template void write_Vxc_LIP( + int, int, int, const psi::Psi>&, const UnitCell&, Structure_Factor&, surchem&, + const ModulePW::PW_Basis_K&, const ModulePW::PW_Basis&, const ModulePW::PW_Basis&, + const ModuleBase::matrix&, const Charge&, const K_Vectors&, const ModuleBase::matrix&, + const bool, const std::string&, const int, const std::string&, bool, double +#ifdef __EXX + , const Exx_Lip>& +#endif +); + +template void write_Vxc_LIP( + int, int, int, const psi::Psi>&, const UnitCell&, Structure_Factor&, surchem&, + const ModulePW::PW_Basis_K&, const ModulePW::PW_Basis&, const ModulePW::PW_Basis&, + const ModuleBase::matrix&, const Charge&, const K_Vectors&, const ModuleBase::matrix&, + const bool, const std::string&, const int, const std::string&, bool, double +#ifdef __EXX + , const Exx_Lip>& +#endif +); + +} // namespace ModuleIO diff --git a/source/source_io/module_hs/vxc_op_lip.h b/source/source_io/module_hs/vxc_op_lip.h new file mode 100644 index 00000000000..a613077f6dd --- /dev/null +++ b/source/source_io/module_hs/vxc_op_lip.h @@ -0,0 +1,53 @@ +#ifndef VXC_OP_LIP_H +#define VXC_OP_LIP_H + +#include "source_psi/psi.h" +#include "source_cell/unitcell.h" +#include "source_cell/klist.h" +#include "source_estate/module_pot/potential_new.h" +#include "source_pw/module_pwdft/op_pw_veff.h" + +#include +#include +#include + +#ifdef __EXX +#include "source_lcao/module_ri/exx_lip.h" +#endif + +namespace ModuleIO +{ + +/// @brief Write the Vxc matrix in KS orbital representation for LIP (LCAO-in-PW), useful for GW calculation +/// including terms: local/semi-local XC and EXX +/// @tparam FPTYPE Floating point type (float or double) +template +void write_Vxc_LIP(int nspin, + int naos, + int drank, + const psi::Psi>& psi_pw, + const UnitCell& ucell, + Structure_Factor& sf, + surchem& solvent, + const ModulePW::PW_Basis_K& wfc_basis, + const ModulePW::PW_Basis& rho_basis, + const ModulePW::PW_Basis& rhod_basis, + const ModuleBase::matrix& vloc, + const Charge& chg, + const K_Vectors& kv, + const ModuleBase::matrix& wg, + const bool gamma_only, + const std::string& global_out_dir, + const int out_ndigits, + const std::string& ks_solver, + bool cal_exx, + double hybrid_alpha +#ifdef __EXX + , + const Exx_Lip>& exx_lip +#endif +); + +} // namespace ModuleIO + +#endif // VXC_OP_LIP_H diff --git a/source/source_io/module_hs/vxc_op_mat.cpp b/source/source_io/module_hs/vxc_op_mat.cpp new file mode 100644 index 00000000000..c7a0b8fe2b4 --- /dev/null +++ b/source/source_io/module_hs/vxc_op_mat.cpp @@ -0,0 +1,187 @@ +#include "vxc_op_mat.h" + +#include "source_base/module_out/filename.h" +#include "source_base/parallel_reduce.h" +#include "source_io/module_hs/hs_dense_io.h" + +namespace ModuleIO +{ + +template +void write_Vxc(const int nspin, + const int nbasis, + const int drank, + const Parallel_Orbitals* pv, + const psi::Psi& psi, + const UnitCell& ucell, + Structure_Factor& sf, + surchem& solvent, + const ModulePW::PW_Basis& rho_basis, + const ModulePW::PW_Basis& rhod_basis, + const ModuleBase::matrix& vloc, + const Charge& chg, + const K_Vectors& kv, + const std::vector& orb_cutoff, + const ModuleBase::matrix& wg, + Grid_Driver& gd, + const bool dft_plus_u, + const bool gamma_only, + const std::string& global_out_dir, + const int out_ndigits, + const std::string& ks_solver, + bool cal_exx, + const Exx_Info& exx_info +#ifdef __EXX + , + std::vector>>>* Hexxd, + std::vector>>>>* Hexxc +#endif +) +{ + ModuleBase::TITLE("ModuleIO", "write_Vxc"); + int nbands = wg.nc; + + // 1. real-space xc potential + double etxc = 0.0; + double vtxc = 0.0; + std::unique_ptr potxc( + new elecstate::Potential(&rhod_basis, &rho_basis, &ucell, &vloc, &sf, &solvent, &etxc, &vtxc)); + std::vector compnents_list = {"xc"}; + potxc->pot_register(compnents_list); + potxc->update_from_charge(&chg, &ucell); + + // 2. allocate AO-matrix + // R (the number of hR: 1 for nspin=1, 4; 2 for nspin=2) + int nspin0 = (nspin == 2) ? 2 : 1; + std::vector> vxcs_R_ao(nspin0, hamilt::HContainer(ucell, pv)); + for (int is = 0; is < nspin0; ++is) { + vxcs_R_ao[is].set_zero(); + if (std::is_same::value) { vxcs_R_ao[is].fix_gamma(); } + } + // k (size for each k-point) + hamilt::HS_Matrix_K vxc_k_ao(pv, 1); // only hk is needed, sk is skipped + + // 3. allocate operators and contribute HR + std::vector>>> vxcs_op_ao(nspin0); + for (int is = 0; is < nspin0; ++is) + { + vxcs_op_ao[is] = std::unique_ptr>>( + new hamilt::Veff>( + &vxc_k_ao, kv.kvec_d, potxc.get(), &vxcs_R_ao[is], &ucell, orb_cutoff, &gd, nspin)); + vxcs_op_ao[is]->set_current_spin(is); + vxcs_op_ao[is]->contributeHR(); + } + std::vector> e_orb_locxc; // orbital energy (local XC) + std::vector> e_orb_tot; // orbital energy (total) +#ifdef __EXX + hamilt::OperatorEXX> vexx_op_ao(&vxc_k_ao, + &vxcs_R_ao[0], ucell, kv, Hexxd, Hexxc, &exx_info, hamilt::Add_Hexx_Type::k); + hamilt::HS_Matrix_K vexxonly_k_ao(pv, 1); // only hk is needed, sk is skipped + hamilt::OperatorEXX> vexxonly_op_ao(&vexxonly_k_ao, + &vxcs_R_ao[0], ucell, kv, Hexxd, Hexxc, &exx_info, hamilt::Add_Hexx_Type::k); + std::vector> e_orb_exx; // orbital energy (EXX) +#endif + hamilt::DFTU_firstzeta> vdftu_op_ao(&vxc_k_ao, kv.kvec_d, nullptr, ucell, nullptr, kv.isk); + + // 4. calculate and write the MO-matrix Exc + Parallel_2D p2d; + set_para2d_MO(*pv, nbands, p2d); + + for (int ik = 0; ik < kv.get_nks(); ++ik) + { + vxc_k_ao.set_zero_hk(); + int is = kv.isk[ik]; + dynamic_cast*>(vxcs_op_ao[is].get())->contributeHk(ik); + const std::vector& vlocxc_k_mo = cVc(vxc_k_ao.get_hk(), &psi(ik, 0, 0), nbasis, nbands, *pv, p2d); + +#ifdef __EXX + if (cal_exx) + { + e_orb_locxc.emplace_back(orbital_energy(ik, nbands, vlocxc_k_mo, p2d)); + ModuleBase::GlobalFunc::ZEROS(vexxonly_k_ao.get_hk(), pv->nloc); + vexx_op_ao.contributeHk(ik); + vexxonly_op_ao.contributeHk(ik); + std::vector vexx_k_mo = cVc(vexxonly_k_ao.get_hk(), &psi(ik, 0, 0), nbasis, nbands, *pv, p2d); + e_orb_exx.emplace_back(orbital_energy(ik, nbands, vexx_k_mo, p2d)); + } +#endif + if (dft_plus_u) + { + vdftu_op_ao.contributeHk(ik); + } + const std::vector& vxc_tot_k_mo = cVc(vxc_k_ao.get_hk(), &psi(ik, 0, 0), nbasis, nbands, *pv, p2d); + e_orb_tot.emplace_back(orbital_energy(ik, nbands, vxc_tot_k_mo, p2d)); + + // write + const int istep = -1; + const int out_label = 1; // 1 means .txt while 2 means .dat + const bool out_app_flag = 0; + std::string vxc_file = ModuleIO::filename_output( + global_out_dir, + "vxc", "nao", ik, kv.ik2iktot, nspin, kv.get_nkstot(), + out_label, out_app_flag, gamma_only, istep); + + ModuleIO::save_mat(istep, + vxc_tot_k_mo.data(), + nbands, + false /*binary*/, + out_ndigits, + true /*triangle*/, + out_app_flag /*append*/, + vxc_file, + p2d, + drank, + ks_solver); + } + + if (GlobalV::MY_RANK == 0) + { + write_orb_energy(kv, nspin0, nbands, e_orb_tot, "vxc", "", global_out_dir); +#ifdef __EXX + if (cal_exx) + { + write_orb_energy(kv, nspin0, nbands, e_orb_locxc, "vxc", "local", global_out_dir); + write_orb_energy(kv, nspin0, nbands, e_orb_exx, "vxc", "exx", global_out_dir); + } +#endif + } +} + +// Explicit template instantiations +template void write_Vxc( + const int, const int, const int, const Parallel_Orbitals*, const psi::Psi&, + const UnitCell&, Structure_Factor&, surchem&, const ModulePW::PW_Basis&, const ModulePW::PW_Basis&, + const ModuleBase::matrix&, const Charge&, const K_Vectors&, const std::vector&, + const ModuleBase::matrix&, Grid_Driver&, const bool, const bool, const std::string&, + const int, const std::string&, bool, const Exx_Info& +#ifdef __EXX + , std::vector>>>*, + std::vector>>>>* +#endif +); + +template void write_Vxc, double>( + const int, const int, const int, const Parallel_Orbitals*, const psi::Psi>&, + const UnitCell&, Structure_Factor&, surchem&, const ModulePW::PW_Basis&, const ModulePW::PW_Basis&, + const ModuleBase::matrix&, const Charge&, const K_Vectors&, const std::vector&, + const ModuleBase::matrix&, Grid_Driver&, const bool, const bool, const std::string&, + const int, const std::string&, bool, const Exx_Info& +#ifdef __EXX + , std::vector>>>*, + std::vector>>>>* +#endif +); + +template void write_Vxc, std::complex>( + const int, const int, const int, const Parallel_Orbitals*, const psi::Psi>&, + const UnitCell&, Structure_Factor&, surchem&, const ModulePW::PW_Basis&, const ModulePW::PW_Basis&, + const ModuleBase::matrix&, const Charge&, const K_Vectors&, const std::vector&, + const ModuleBase::matrix&, Grid_Driver&, const bool, const bool, const std::string&, + const int, const std::string&, bool, const Exx_Info& +#ifdef __EXX + , std::vector>>>*, + std::vector>>>>* +#endif +); + +} // namespace ModuleIO diff --git a/source/source_io/module_hs/vxc_op_mat.h b/source/source_io/module_hs/vxc_op_mat.h new file mode 100644 index 00000000000..7e17a17cfb2 --- /dev/null +++ b/source/source_io/module_hs/vxc_op_mat.h @@ -0,0 +1,68 @@ +#ifndef VXC_OP_MAT_H +#define VXC_OP_MAT_H + +#include "source_psi/psi.h" +#include "source_cell/unitcell.h" +#include "source_cell/klist.h" +#include "source_estate/module_pot/potential_new.h" +#include "source_hamilt/module_xc/exx_info.h" +#include "source_lcao/module_operator_lcao/veff_lcao.h" +#include "source_lcao/module_dftu/dftu_nao_op_legacy.h" +#include "source_io/module_hs/vxc_op_tools.h" + +#include +#include +#include +#include + +#ifdef __EXX +#include "source_lcao/module_operator_lcao/op_exx_lcao.h" +#endif + +namespace hamilt +{ +template +class HContainer; +} // namespace hamilt + +namespace ModuleIO +{ + +/// @brief Write the Vxc matrix in KS orbital representation, useful for GW calculation +/// including terms: local/semi-local XC, EXX, DFTU +/// @tparam TK K-point data type (double or std::complex) +/// @tparam TR Real-space data type (double or std::complex) +template +void write_Vxc(const int nspin, + const int nbasis, + const int drank, + const Parallel_Orbitals* pv, + const psi::Psi& psi, + const UnitCell& ucell, + Structure_Factor& sf, + surchem& solvent, + const ModulePW::PW_Basis& rho_basis, + const ModulePW::PW_Basis& rhod_basis, + const ModuleBase::matrix& vloc, + const Charge& chg, + const K_Vectors& kv, + const std::vector& orb_cutoff, + const ModuleBase::matrix& wg, + Grid_Driver& gd, + const bool dft_plus_u, + const bool gamma_only, + const std::string& global_out_dir, + const int out_ndigits, + const std::string& ks_solver, + bool cal_exx, + const Exx_Info& exx_info +#ifdef __EXX + , + std::vector>>>* Hexxd, + std::vector>>>>* Hexxc +#endif +); + +} // namespace ModuleIO + +#endif // VXC_OP_MAT_H diff --git a/source/source_io/module_hs/vxc_op_r.cpp b/source/source_io/module_hs/vxc_op_r.cpp new file mode 100644 index 00000000000..38345a7be3a --- /dev/null +++ b/source/source_io/module_hs/vxc_op_r.cpp @@ -0,0 +1,167 @@ +#include "vxc_op_r.h" + +#include "source_io/module_hs/hs_sparse_io.h" +#include "source_base/module_out/filename.h" + +namespace ModuleIO +{ + +/// @brief Helper to calculate sparse HR representation +template +std::map, std::map>> cal_HR_sparse( + const hamilt::HContainer& hR, + const double sparse_thr) +{ + std::map, std::map>> target; + sparse_format::cal_HContainer(*hR.get_paraV(), sparse_thr, hR, target); + return target; +} + +template +void write_Vxc_R(const int nspin, + const Parallel_Orbitals* pv, + const UnitCell& ucell, + Structure_Factor& sf, + surchem& solvent, + const ModulePW::PW_Basis& rho_basis, + const ModulePW::PW_Basis& rhod_basis, + const ModuleBase::matrix& vloc, + const Charge& chg, + const K_Vectors& kv, + const std::vector& orb_cutoff, + Grid_Driver& gd, + const std::string& global_out_dir, + bool cal_exx, + double hybrid_alpha, + bool real_number +#ifdef __EXX + , + const std::vector>>>* Hexxd, + const std::vector>>>>* Hexxc +#endif + , + const double sparse_thr) +{ + ModuleBase::TITLE("ModuleIO", "write_Vxc_R"); + + // 1. real-space xc potential + double etxc = 0.0; + double vtxc = 0.0; + elecstate::Potential potxc(&rhod_basis, &rho_basis, &ucell, &vloc, &sf, &solvent, &etxc, &vtxc); + std::vector compnents_list = {"xc"}; + potxc.pot_register(compnents_list); + potxc.update_from_charge(&chg, &ucell); + + // 2. allocate H(R) + // (the number of hR: 1 for nspin=1, 4; 2 for nspin=2) + int nspin0 = (nspin == 2) ? 2 : 1; + std::vector> vxcs_R_ao(nspin0, hamilt::HContainer(ucell, pv)); +#ifdef __EXX + std::array Rs_period = {kv.nmp[0], kv.nmp[1], kv.nmp[2]}; + const auto cell_nearest = hamilt::init_cell_nearest(ucell, Rs_period); +#endif + for (int is = 0; is < nspin0; ++is) + { + if (std::is_same::value) + { + vxcs_R_ao[is].fix_gamma(); + } +#ifdef __EXX + if (cal_exx) + { + real_number + ? hamilt::reallocate_hcontainer(*Hexxd, &vxcs_R_ao[is], &cell_nearest) + : hamilt::reallocate_hcontainer(*Hexxc, &vxcs_R_ao[is], &cell_nearest); + } +#endif + } + + // 3. calculate the Vxc(R) + hamilt::HS_Matrix_K vxc_k_ao(pv, 1); // only hk is needed, sk is skipped + for (int is = 0; is < nspin0; ++is) + { + hamilt::Veff> vxcs_op_ao(&vxc_k_ao, + kv.kvec_d, + &potxc, + &vxcs_R_ao[is], + &ucell, + orb_cutoff, + &gd, + nspin); + vxcs_op_ao.set_current_spin(is); + vxcs_op_ao.contributeHR(); +#ifdef __EXX + if (cal_exx) + { + real_number ? RI_2D_Comm::add_HexxR(is, + hybrid_alpha, + *Hexxd, + *pv, + ucell.get_npol(), + vxcs_R_ao[is], + &cell_nearest) + : RI_2D_Comm::add_HexxR(is, + hybrid_alpha, + *Hexxc, + *pv, + ucell.get_npol(), + vxcs_R_ao[is], + &cell_nearest); + } +#endif + } + + // 4. write Vxc(R) in csr format + for (int is = 0; is < nspin0; ++is) + { + std::set> all_R_coor = sparse_format::get_R_range(vxcs_R_ao[is]); + const std::string filename = "Vxc_R_spin" + std::to_string(is); + ModuleIO::SparseWriteOptions options; + options.filename = global_out_dir + filename + ".csr"; + options.label = filename; + options.threshold = sparse_thr; + options.binary = false; + options.istep = -1; + options.reduce = true; + ModuleIO::save_sparse(cal_HR_sparse(vxcs_R_ao[is], sparse_thr), + all_R_coor, + *pv, + options); + } +} + +// Explicit template instantiations +template void write_Vxc_R( + const int, const Parallel_Orbitals*, const UnitCell&, Structure_Factor&, surchem&, + const ModulePW::PW_Basis&, const ModulePW::PW_Basis&, const ModuleBase::matrix&, + const Charge&, const K_Vectors&, const std::vector&, Grid_Driver&, + const std::string&, bool, double, bool +#ifdef __EXX + , const std::vector>>>*, + const std::vector>>>>* +#endif + , const double); + +template void write_Vxc_R, double>( + const int, const Parallel_Orbitals*, const UnitCell&, Structure_Factor&, surchem&, + const ModulePW::PW_Basis&, const ModulePW::PW_Basis&, const ModuleBase::matrix&, + const Charge&, const K_Vectors&, const std::vector&, Grid_Driver&, + const std::string&, bool, double, bool +#ifdef __EXX + , const std::vector>>>*, + const std::vector>>>>* +#endif + , const double); + +template void write_Vxc_R, std::complex>( + const int, const Parallel_Orbitals*, const UnitCell&, Structure_Factor&, surchem&, + const ModulePW::PW_Basis&, const ModulePW::PW_Basis&, const ModuleBase::matrix&, + const Charge&, const K_Vectors&, const std::vector&, Grid_Driver&, + const std::string&, bool, double, bool +#ifdef __EXX + , const std::vector>>>*, + const std::vector>>>>* +#endif + , const double); + +} // namespace ModuleIO diff --git a/source/source_io/module_hs/vxc_op_r.h b/source/source_io/module_hs/vxc_op_r.h new file mode 100644 index 00000000000..a51559f32b1 --- /dev/null +++ b/source/source_io/module_hs/vxc_op_r.h @@ -0,0 +1,57 @@ +#ifndef VXC_OP_R_H +#define VXC_OP_R_H + +#include "source_lcao/module_operator_lcao/veff_lcao.h" +#include "source_lcao/module_dftu/dftu_nao_op_legacy.h" +#include "source_lcao/spar_hsr.h" +#include "source_lcao/module_ri/abfs_vector3_order.h" +#include "source_cell/unitcell.h" +#include "source_cell/klist.h" +#include "source_estate/module_pot/potential_new.h" +#include "source_hamilt/module_xc/exx_info.h" + +#include +#include +#include +#include + +#ifdef __EXX +#include "source_lcao/module_operator_lcao/op_exx_lcao.h" +#include "source_lcao/module_ri/ri_2d_comm.h" +#endif + +namespace ModuleIO +{ + +/// @brief Write the Vxc matrix in real space (R), useful for GW calculation +/// including terms: local/semi-local XC, EXX, DFTU +/// @tparam TK K-point data type (double or std::complex) +/// @tparam TR Real-space data type (double or std::complex) +template +void write_Vxc_R(const int nspin, + const Parallel_Orbitals* pv, + const UnitCell& ucell, + Structure_Factor& sf, + surchem& solvent, + const ModulePW::PW_Basis& rho_basis, + const ModulePW::PW_Basis& rhod_basis, + const ModuleBase::matrix& vloc, + const Charge& chg, + const K_Vectors& kv, + const std::vector& orb_cutoff, + Grid_Driver& gd, + const std::string& global_out_dir, + bool cal_exx, + double hybrid_alpha, + bool real_number +#ifdef __EXX + , + const std::vector>>>* Hexxd, + const std::vector>>>>* Hexxc +#endif + , + const double sparse_thr); + +} // namespace ModuleIO + +#endif // VXC_OP_R_H diff --git a/source/source_io/module_hs/vxc_op_tools.cpp b/source/source_io/module_hs/vxc_op_tools.cpp new file mode 100644 index 00000000000..0d20574d381 --- /dev/null +++ b/source/source_io/module_hs/vxc_op_tools.cpp @@ -0,0 +1,140 @@ +#include "vxc_op_tools.h" + +#include "source_base/module_external/scalapack_connector.h" +#include "source_base/module_out/filename.h" + +#include +#include + +namespace ModuleIO +{ + +void set_para2d_MO(const Parallel_Orbitals& pv, const int nbands, Parallel_2D& p2d) +{ +#ifdef __MPI + p2d.set(nbands, nbands, pv.nb, pv.blacs_ctxt); +#else + p2d.set_serial(nbands, nbands); +#endif +} + +template +std::vector cVc(const T* V, + const T* c, + const int nbasis, + const int nbands, + const Parallel_Orbitals& pv, + const Parallel_2D& p2d) +{ + std::vector Vc(pv.nloc_wfc, 0.0); + char transa = 'N'; + char transb = 'N'; + const T alpha = static_cast(1.0); + const T beta = static_cast(0.0); +#ifdef __MPI + const int i1 = 1; + ScalapackConnector::gemm(transa, transb, + nbasis, nbands, nbasis, + alpha, V, i1, i1, pv.desc, + c, i1, i1, pv.desc_wfc, + beta, Vc.data(), i1, i1, pv.desc_wfc); +#else + container::BlasConnector::gemm(transa, transb, nbasis, nbands, nbasis, alpha, V, nbasis, c, nbasis, beta, Vc.data(), nbasis); +#endif + std::vector cVc_result(p2d.nloc, 0.0); + transa = (std::is_same::value ? 'T' : 'C'); +#ifdef __MPI + ScalapackConnector::gemm(transa, transb, + nbands, nbands, nbasis, + alpha, c, i1, i1, pv.desc_wfc, + Vc.data(), i1, i1, pv.desc_wfc, + beta, cVc_result.data(), i1, i1, p2d.desc); +#else + container::BlasConnector::gemm(transa, transb, nbands, nbands, nbasis, alpha, c, nbasis, Vc.data(), nbasis, beta, cVc_result.data(), nbasis); +#endif + return cVc_result; +} + +template +double all_band_energy(const int ik, + const std::vector& mat_mo, + const Parallel_2D& p2d, + const ModuleBase::matrix& wg) +{ + double e = 0.0; + for (int i = 0; i < p2d.get_row_size(); ++i) + { + for (int j = 0; j < p2d.get_col_size(); ++j) + { + if (p2d.local2global_row(i) == p2d.local2global_col(j)) + { + e += get_real(mat_mo[j * p2d.get_row_size() + i]) * wg(ik, p2d.local2global_row(i)); + } + } + } + Parallel_Reduce::reduce_all(e); + return e; +} + +template +std::vector orbital_energy(const int ik, + const int nbands, + const std::vector& mat_mo, + const Parallel_2D& p2d) +{ +#ifdef __DEBUG + assert(nbands >= 0); +#endif + std::vector e(nbands, 0.0); + for (int i = 0; i < nbands; ++i) + { + if (p2d.in_this_processor(i, i)) + { + const int index = p2d.global2local_col(i) * p2d.get_row_size() + p2d.global2local_row(i); + e[i] = get_real(mat_mo[index]); + } + } + Parallel_Reduce::reduce_all(e.data(), nbands); + return e; +} + +void write_orb_energy(const K_Vectors& kv, + const int nspin0, + const int nbands, + const std::vector>& e_orb, + const std::string& term, + const std::string& label, + const std::string& global_out_dir, + const bool app) +{ + assert(e_orb.size() == kv.get_nks()); + const int nk = kv.get_nks() / nspin0; + std::ofstream ofs; + const std::string out_name = (label == "") ? "out.dat" : label + "_out.dat"; + ofs.open(global_out_dir + term + "_" + out_name, + app ? std::ios::app : std::ios::out); + ofs << nk << "\n" << nspin0 << "\n" << nbands << "\n"; + ofs << std::scientific << std::setprecision(16); + for (int ik = 0; ik < nk; ++ik) + { + for (int is = 0; is < nspin0; ++is) + { + for (auto e : e_orb[is * nk + ik]) + { // Hartree and eV + ofs << e / 2. << "\t" << e * ModuleBase::Ry_to_eV << "\n"; + } + } + } +} + +// Explicit template instantiations +template std::vector cVc(const double*, const double*, const int, const int, const Parallel_Orbitals&, const Parallel_2D&); +template std::vector> cVc>(const std::complex*, const std::complex*, const int, const int, const Parallel_Orbitals&, const Parallel_2D&); + +template double all_band_energy(const int, const std::vector&, const Parallel_2D&, const ModuleBase::matrix&); +template double all_band_energy>(const int, const std::vector>&, const Parallel_2D&, const ModuleBase::matrix&); + +template std::vector orbital_energy(const int, const int, const std::vector&, const Parallel_2D&); +template std::vector orbital_energy>(const int, const int, const std::vector>&, const Parallel_2D&); + +} // namespace ModuleIO diff --git a/source/source_io/module_hs/vxc_op_tools.h b/source/source_io/module_hs/vxc_op_tools.h new file mode 100644 index 00000000000..4e66d632fb0 --- /dev/null +++ b/source/source_io/module_hs/vxc_op_tools.h @@ -0,0 +1,98 @@ +#ifndef VXC_OP_TOOLS_H +#define VXC_OP_TOOLS_H + +#include "source_base/parallel_2d.h" +#include "source_basis/module_ao/parallel_orbitals.h" +#include "source_base/module_container/base/third_party/blas.h" +#include "source_base/parallel_reduce.h" +#include "source_cell/klist.h" + +#include +#include +#include + +namespace ModuleIO +{ + +/// @brief Set up a 2D parallel distribution for MO-space matrices (nbands x nbands) +/// @param pv Parallel_Orbitals describing the AO distribution +/// @param nbands Number of bands +/// @param p2d Output Parallel_2D object +void set_para2d_MO(const Parallel_Orbitals& pv, const int nbands, Parallel_2D& p2d); + +/// @brief Extract real part (for complex numbers) +inline double get_real(const std::complex& c) +{ + return c.real(); +} + +/// @brief Extract real part (for real numbers, identity) +inline double get_real(const double& d) +{ + return d; +} + +/// @brief Compute c^dagger * V * c in MO basis +/// @tparam T Data type (double or std::complex) +/// @param V AO-space matrix V (nbasis x nbasis) +/// @param c AO-to-MO transformation coefficients (nbasis x nbands) +/// @param nbasis Number of AO basis functions +/// @param nbands Number of bands +/// @param pv Parallel_Orbitals for AO distribution +/// @param p2d Parallel_2D for MO distribution +/// @return MO-space matrix c^dagger * V * c (nbands x nbands) +template +std::vector cVc(const T* V, + const T* c, + const int nbasis, + const int nbands, + const Parallel_Orbitals& pv, + const Parallel_2D& p2d); + +/// @brief Extract diagonal elements (orbital energies) from MO matrix +/// @tparam T Data type (double or std::complex) +/// @param ik K-point index (for occupancy weights) +/// @param nbands Number of bands +/// @param mat_mo MO-space matrix +/// @param p2d Parallel_2D distribution +/// @return Vector of orbital energies (length nbands) +template +std::vector orbital_energy(const int ik, + const int nbands, + const std::vector& mat_mo, + const Parallel_2D& p2d); + +/// @brief Sum over bands with occupation weights +/// @tparam T Data type (double or std::complex) +/// @param ik K-point index +/// @param mat_mo MO-space matrix +/// @param p2d Parallel_2D distribution +/// @param wg Occupation weights +/// @return Weighted sum of diagonal elements +template +double all_band_energy(const int ik, + const std::vector& mat_mo, + const Parallel_2D& p2d, + const ModuleBase::matrix& wg); + +/// @brief Write orbital energies to file in LibRPA format +/// @param kv K-point list +/// @param nspin0 Number of independent spin channels (1 or 2) +/// @param nbands Number of bands +/// @param e_orb Orbital energies per k-point +/// @param term Term label (e.g., "vxc", "kinetic") +/// @param label Additional label (e.g., "local", "exx") +/// @param global_out_dir Output directory +/// @param app Append mode flag +void write_orb_energy(const K_Vectors& kv, + const int nspin0, + const int nbands, + const std::vector>& e_orb, + const std::string& term, + const std::string& label, + const std::string& global_out_dir, + const bool app = false); + +} // namespace ModuleIO + +#endif // VXC_OP_TOOLS_H diff --git a/source/source_io/module_hs/write_h_terms.cpp b/source/source_io/module_hs/write_h_terms.cpp deleted file mode 100644 index f85aa4b97a7..00000000000 --- a/source/source_io/module_hs/write_h_terms.cpp +++ /dev/null @@ -1,446 +0,0 @@ -#include "write_h_terms.h" - -#include "source_base/parallel_reduce.h" -#include "source_base/timer.h" -#include "source_estate/module_pot/h_hartree_pw.h" -#include "source_hamilt/module_xc/xc_functional.h" -#include "source_io/module_hs/write_hs.h" -#include "source_io/module_hs/write_hs_r.h" -#include "source_base/module_out/filename.h" -#include "source_cell/ucell_io.h" -#include "source_io/module_parameter/parameter.h" -#include "source_hamilt/module_gint/gint_interface.h" -#include "source_hamilt/module_hcontainer/hcontainer_funcs.h" -#include "source_hamilt/module_hcontainer/output_hcontainer.h" -#include "source_lcao/module_operator_lcao/ekinetic.h" -#include "source_lcao/module_operator_lcao/nonlocal.h" -#include "source_lcao/module_operator_lcao/operator_fs_utils.h" -#ifdef __EXX -#include "source_lcao/module_operator_lcao/op_exx_lcao.h" -#include "source_lcao/module_ri/exx_lri_interface.h" -#include "source_lcao/module_ri/ri_2d_comm.h" -#endif - -#include -#include - -namespace ModuleIO -{ - -static void setup_veff_hcontainer(hamilt::HContainer& hR, - const UnitCell& ucell, - const Grid_Driver& gd, - const Parallel_Orbitals& pv, - const std::vector& orb_cutoff) -{ - const Parallel_Orbitals* paraV = hR.get_paraV(); - for (int iat1 = 0; iat1 < ucell.nat; iat1++) - { - auto tau1 = ucell.get_tau(iat1); - int T1 = 0, I1 = 0; - ucell.iat2iait(iat1, &I1, &T1); - - AdjacentAtomInfo adjs; - gd.Find_atom(ucell, tau1, T1, I1, &adjs); - - for (int ad = 0; ad < adjs.adj_num + 1; ++ad) - { - const int T2 = adjs.ntype[ad]; - const int I2 = adjs.natom[ad]; - const int iat2 = ucell.itia2iat(T2, I2); - if (paraV->is_invalid_atom_pair(iat1, iat2)) - { - continue; - } - const ModuleBase::Vector3& R_index = adjs.box[ad]; - if (ucell.cal_dtau(iat1, iat2, R_index).norm() * ucell.lat0 < orb_cutoff[T1] + orb_cutoff[T2]) - { - hamilt::AtomPair tmp(iat1, iat2, R_index, paraV); - hR.insert_pair(tmp); - } - } - } - hR.allocate(nullptr, true); -} - -static void gather_and_write(const std::string& prefix, - const std::string& label, - hamilt::HContainer& hR, - const UnitCell& ucell, - const Parallel_Orbitals& pv, - const int nspin, - const int ispin, - const int istep, - const bool append, - const int* iat2iwt, - const int nat) -{ - const int nbasis = hR.get_nbasis(); -#ifdef __MPI - Parallel_Orbitals serialV; - serialV.set_serial(nbasis, nbasis); - serialV.set_atomic_trace(iat2iwt, nat, nbasis); - hamilt::HContainer hr_serial(&serialV); - hamilt::gatherParallels(hR, &hr_serial, 0); - if (GlobalV::MY_RANK == 0) -#endif - { - std::string fname; - if (PARAM.inp.calculation == "md" && !PARAM.inp.out_app_flag) - { - fname = PARAM.globalv.global_matrix_dir + hsr_gen_fname(prefix, ispin, append, istep); - } - else - { - fname = PARAM.globalv.global_out_dir + hsr_gen_fname(prefix, ispin, append, istep); - } -#ifdef __MPI - write_hcontainer_csr(fname, &ucell, 8, &hr_serial, istep, ispin, nspin, label, ""); -#else - write_hcontainer_csr(fname, &ucell, 8, &hR, istep, ispin, nspin, label, ""); -#endif - } -} - -static void write_hk_common(hamilt::HContainer& hR, - const std::string& prefix, - const UnitCell& ucell, - const Parallel_Orbitals& pv, - const K_Vectors& kv, - const int nspin, - const int istep, - const bool append, - const int* iat2iwt, - const int nat) -{ - const int nspin_k = (nspin == 2 ? 2 : 1); - const int nks = kv.get_nks() / nspin_k; - const int nlocal = PARAM.globalv.nlocal; - const bool gamma_only = PARAM.globalv.gamma_only_local; - const std::string global_out_dir = PARAM.globalv.global_out_dir; - const bool out_app_flag = PARAM.inp.out_app_flag; - - for (int ik = 0; ik < nks; ++ik) - { - const ModuleBase::Vector3& kvec_d = kv.kvec_d[ik]; - - std::vector> hk_global(nlocal * nlocal, 0); - hamilt::folding_HR(hR, hk_global.data(), kvec_d, nlocal, 0); - - const int out_label = 1; - std::string fname = ModuleIO::filename_output(global_out_dir, - prefix, - "nao", - ik, - kv.ik2iktot, - nspin, - kv.get_nkstot(), - out_label, - out_app_flag, - gamma_only, - istep); - ModuleIO::save_mat(istep, - hk_global.data(), - nlocal, - false, - 8, - false, - out_app_flag, - fname, - pv, - GlobalV::DRANK); - } -} - -void write_h_t(WriteHParams& params) -{ - ModuleBase::TITLE("ModuleIO", "write_h_t"); - ModuleBase::timer::start("ModuleIO", "write_h_t"); - - const UnitCell& ucell = *params.ucell; - const Grid_Driver& gd = *params.gd; - const Parallel_Orbitals& pv = *params.pv; - const TwoCenterBundle& two_center_bundle = *params.two_center_bundle; - const LCAO_Orbitals& orb = *params.orb; - const K_Vectors& kv = *params.kv; - const int nspin = params.nspin; - const int istep = params.istep; - const bool append = params.append; - const int* iat2iwt = params.iat2iwt; - const int nat = params.nat; - const bool also_hR = params.also_hR; - - const std::vector& orb_cutoff = orb.cutoffs(); - const int nspin_out = (nspin == 2 ? 2 : 1); - - for (int ispin = 0; ispin < nspin_out; ispin++) - { - hamilt::HContainer hR_tmp(const_cast(&pv)); - - hamilt::EKinetic> - tmp_ekinetic(nullptr, kv.kvec_d, &hR_tmp, &ucell, orb_cutoff, &gd, two_center_bundle.kinetic_orb.get()); - tmp_ekinetic.contributeHR(); - - write_hk_common(hR_tmp, "tk", ucell, pv, kv, nspin, istep, append, iat2iwt, nat); - - if (also_hR) - { - gather_and_write("t", "T", hR_tmp, ucell, pv, nspin, ispin, istep, append, iat2iwt, nat); - } - } - - ModuleBase::timer::end("ModuleIO", "write_h_t"); -} - -void write_h_vnl(WriteHParams& params) -{ - ModuleBase::TITLE("ModuleIO", "write_h_vnl"); - ModuleBase::timer::start("ModuleIO", "write_h_vnl"); - - const UnitCell& ucell = *params.ucell; - const Grid_Driver& gd = *params.gd; - const Parallel_Orbitals& pv = *params.pv; - const TwoCenterBundle& two_center_bundle = *params.two_center_bundle; - const LCAO_Orbitals& orb = *params.orb; - const K_Vectors& kv = *params.kv; - const int nspin = params.nspin; - const int istep = params.istep; - const bool append = params.append; - const int* iat2iwt = params.iat2iwt; - const int nat = params.nat; - const bool also_hR = params.also_hR; - - const std::vector& orb_cutoff = orb.cutoffs(); - const int nspin_out = (nspin == 2 ? 2 : 1); - - for (int ispin = 0; ispin < nspin_out; ispin++) - { - hamilt::HContainer hR_tmp(const_cast(&pv)); - - hamilt::Nonlocal> tmp_nonlocal(nullptr, - kv.kvec_d, - &hR_tmp, - &ucell, - orb_cutoff, - &gd, - two_center_bundle.overlap_orb_beta.get()); - tmp_nonlocal.contributeHR(); - - write_hk_common(hR_tmp, "vnlk", ucell, pv, kv, nspin, istep, append, iat2iwt, nat); - - if (also_hR) - { - gather_and_write("vnl", "V^NL", hR_tmp, ucell, pv, nspin, ispin, istep, append, iat2iwt, nat); - } - } - - ModuleBase::timer::end("ModuleIO", "write_h_vnl"); -} - -void write_h_vl(WriteHParams& params) -{ - ModuleBase::TITLE("ModuleIO", "write_h_vl"); - ModuleBase::timer::start("ModuleIO", "write_h_vl"); - - const UnitCell& ucell = *params.ucell; - const Grid_Driver& gd = *params.gd; - const Parallel_Orbitals& pv = *params.pv; - const LCAO_Orbitals& orb = *params.orb; - const elecstate::Potential* pot = params.pot; - const K_Vectors& kv = *params.kv; - const int nspin = params.nspin; - const int istep = params.istep; - const bool append = params.append; - const int* iat2iwt = params.iat2iwt; - const int nat = params.nat; - const bool also_hR = params.also_hR; - - const std::vector& orb_cutoff = orb.cutoffs(); - const int nspin_out = (nspin == 2 ? 2 : 1); - - for (int ispin = 0; ispin < nspin_out; ispin++) - { - hamilt::HContainer hR_tmp(const_cast(&pv)); - setup_veff_hcontainer(hR_tmp, ucell, gd, pv, orb_cutoff); - - const double* v_local = pot->get_fixed_v(); // local pp, no Hxc - ModuleGint::cal_gint_vl(v_local, &hR_tmp); - - write_hk_common(hR_tmp, "vlk", ucell, pv, kv, nspin, istep, append, iat2iwt, nat); - - if (also_hR) - { - gather_and_write("vl", "V^L", hR_tmp, ucell, pv, nspin, ispin, istep, append, iat2iwt, nat); - } - } - - ModuleBase::timer::end("ModuleIO", "write_h_vl"); -} - -void write_h_vh(WriteHParams& params) -{ - ModuleBase::TITLE("ModuleIO", "write_h_vh"); - ModuleBase::timer::start("ModuleIO", "write_h_vh"); - - const UnitCell& ucell = *params.ucell; - const Grid_Driver& gd = *params.gd; - const Parallel_Orbitals& pv = *params.pv; - const LCAO_Orbitals& orb = *params.orb; - const Charge* chg = params.chg; - const ModulePW::PW_Basis* rho_basis = params.rho_basis; - const K_Vectors& kv = *params.kv; - const int nspin = params.nspin; - const int istep = params.istep; - const bool append = params.append; - const int* iat2iwt = params.iat2iwt; - const int nat = params.nat; - const bool also_hR = params.also_hR; - - const std::vector& orb_cutoff = orb.cutoffs(); - const int nspin_out = (nspin == 2 ? 2 : 1); - - ModuleBase::matrix v_h - = elecstate::H_Hartree_pw::v_hartree(ucell, const_cast(rho_basis), nspin, chg->rho); - - for (int ispin = 0; ispin < nspin_out; ispin++) - { - hamilt::HContainer hR_tmp(const_cast(&pv)); - setup_veff_hcontainer(hR_tmp, ucell, gd, pv, orb_cutoff); - - ModuleGint::cal_gint_vl(&v_h(ispin, 0), &hR_tmp); - - write_hk_common(hR_tmp, "vhk", ucell, pv, kv, nspin, istep, append, iat2iwt, nat); - - if (also_hR) - { - gather_and_write("vh", "V^H", hR_tmp, ucell, pv, nspin, ispin, istep, append, iat2iwt, nat); - } - } - - ModuleBase::timer::end("ModuleIO", "write_h_vh"); -} - -void write_h_vxc(WriteHParams& params) -{ - ModuleBase::TITLE("ModuleIO", "write_h_vxc"); - ModuleBase::timer::start("ModuleIO", "write_h_vxc"); - - const UnitCell& ucell = *params.ucell; - const Grid_Driver& gd = *params.gd; - const Parallel_Orbitals& pv = *params.pv; - const LCAO_Orbitals& orb = *params.orb; - const Charge* chg = params.chg; - const int nrxx = params.nrxx; - const K_Vectors& kv = *params.kv; - const int nspin = params.nspin; - const int istep = params.istep; - const bool append = params.append; - const int* iat2iwt = params.iat2iwt; - const int nat = params.nat; - const bool also_hR = params.also_hR; - - const std::vector& orb_cutoff = orb.cutoffs(); - const int nspin_out = (nspin == 2 ? 2 : 1); - - ModuleBase::matrix v_xc; - double etxc, vtxc; - const double hybrid_alpha = XC_Functional::get_hybrid_alpha(); -#ifdef __EXX - const double hse_omega = XC_Functional::get_hse_omega(); -#else - const double hse_omega = 0.0; -#endif - std::tie(etxc, vtxc, v_xc) = XC_Functional::v_xc(nrxx, chg, &ucell, PARAM.inp.nspin, PARAM.globalv.domag, PARAM.globalv.domag_z, hybrid_alpha, hse_omega); - - for (int ispin = 0; ispin < nspin_out; ispin++) - { - hamilt::HContainer hR_tmp(const_cast(&pv)); - setup_veff_hcontainer(hR_tmp, ucell, gd, pv, orb_cutoff); - - ModuleGint::cal_gint_vl(&v_xc(ispin, 0), &hR_tmp); - - write_hk_common(hR_tmp, "vxck", ucell, pv, kv, nspin, istep, append, iat2iwt, nat); - - if (also_hR) - { - gather_and_write("vxc", "V^XC", hR_tmp, ucell, pv, nspin, ispin, istep, append, iat2iwt, nat); - } - } - - ModuleBase::timer::end("ModuleIO", "write_h_vxc"); -} - -#ifdef __EXX -// Build V^EXX(R) for one interface (real or complex Hexx) into real HContainers and write them. -template -static void write_h_exx_impl(const UnitCell& ucell, - const Parallel_Orbitals& pv, - Exx_LRI_Interface* ex, - const K_Vectors& kv, - const int nspin, - const int istep, - const bool append, - const int* iat2iwt, - const int nat, - const bool also_hR, - const Exx_Info& exx_info) -{ - const auto& Hexxs = ex->get_Hexxs(); // vector over spin of map> - const int nspin_out = (nspin == 2 ? 2 : 1); - const double alpha = exx_info.info_global.hybrid_alpha; - - for (int ispin = 0; ispin < nspin_out; ispin++) - { - hamilt::HContainer hR_tmp(const_cast(&pv)); - // add_HexxR only fills existing matrices, so first allocate the atom-pair structure - // from the exx-form data (native cells, consistent with the nullptr cell_nearest below). - hamilt::reallocate_hcontainer(Hexxs, &hR_tmp); - RI_2D_Comm::add_HexxR(ispin, alpha, Hexxs, pv, PARAM.globalv.npol, hR_tmp, nullptr); - - write_hk_common(hR_tmp, "vexxk", ucell, pv, kv, nspin, istep, append, iat2iwt, nat); - - if (also_hR) - { - gather_and_write("vexx", "V^EXX", hR_tmp, ucell, pv, nspin, ispin, istep, append, iat2iwt, nat); - } - } -} - -void write_h_exx(WriteHParams& params, const Exx_Info& exx_info) -{ - ModuleBase::TITLE("ModuleIO", "write_h_exx"); - ModuleBase::timer::start("ModuleIO", "write_h_exx"); - - // Multi-k out_mat_h_exx is rejected upstream at the call site (setup_exx_h_params in - // ctrl_scf_lcao.cpp); this function is only reached on the gamma-only path. - const UnitCell& ucell = *params.ucell; - const Parallel_Orbitals& pv = *params.pv; - const K_Vectors& kv = *params.kv; - const int nspin = params.nspin; - const int istep = params.istep; - const bool append = params.append; - const int* iat2iwt = params.iat2iwt; - const int nat = params.nat; - const bool also_hR = params.also_hR; - - // exd (real Hexx) and exc (complex Hexx) are mutually exclusive; pick by real_number. - if (exx_info.info_ri.real_number) - { - if (params.exd != nullptr) - { - write_h_exx_impl(ucell, pv, params.exd, kv, nspin, istep, append, iat2iwt, nat, also_hR, exx_info); - } - } - else - { - if (params.exc != nullptr) - { - write_h_exx_impl(ucell, pv, params.exc, kv, nspin, istep, append, iat2iwt, nat, also_hR, exx_info); - } - } - - ModuleBase::timer::end("ModuleIO", "write_h_exx"); -} -#endif - -} // namespace ModuleIO diff --git a/source/source_io/module_hs/write_h_terms.h b/source/source_io/module_hs/write_h_terms.h deleted file mode 100644 index cff8faee3d2..00000000000 --- a/source/source_io/module_hs/write_h_terms.h +++ /dev/null @@ -1,71 +0,0 @@ -#ifndef WRITE_H_TERMS_H -#define WRITE_H_TERMS_H - -#include "source_basis/module_nao/two_center_bundle.h" -#include "source_basis/module_pw/pw_basis.h" -#include "source_cell/klist.h" -#include "source_cell/module_neighbor/sltk_grid_driver.h" -#include "source_estate/module_charge/charge.h" -#include "source_estate/module_pot/potential_new.h" -#include "source_lcao/lcao_domain.h" -#include "source_hamilt/module_hcontainer/hcontainer.h" -#include "source_hamilt/module_xc/exx_info.h" - -#include -#include - -template -class Exx_LRI_Interface; - -namespace ModuleIO -{ - -struct WriteHParams -{ - const UnitCell* ucell = nullptr; - const Grid_Driver* gd = nullptr; - const Parallel_Orbitals* pv = nullptr; - const TwoCenterBundle* two_center_bundle = nullptr; - const LCAO_Orbitals* orb = nullptr; - const K_Vectors* kv = nullptr; - const elecstate::Potential* pot = nullptr; // used by write_h_vl (local pp only) - const Charge* chg = nullptr; // used by write_h_vh, write_h_vxc - const ModulePW::PW_Basis* rho_basis = nullptr; // used by write_h_vh - int nrxx = 0; // used by write_h_vxc - int nspin = 1; - int istep = 0; - bool append = false; - const int* iat2iwt = nullptr; - int nat = 0; - bool also_hR = false; // H(k) is always written; H(R) (CSR) only when this is true -#ifdef __EXX - // The gamma-only (TK==double) exx interfaces used by write_h_exx. - // Deliberately NOT templated on TK, because it would force WriteHParams, WriteDHParams and - // every free function taking them to become templates as well -- a large, purely - // mechanical change for a case nobody needs. - // Multi-k + EXX is therefore rejected up front (see write_h_exx) - // instead of silently producing output with the EXX term missing. - Exx_LRI_Interface* exd = nullptr; - Exx_LRI_Interface>* exc = nullptr; -#endif -}; - -void write_h_t(WriteHParams& params); - -void write_h_vnl(WriteHParams& params); - -void write_h_vl(WriteHParams& params); - -void write_h_vh(WriteHParams& params); - -void write_h_vxc(WriteHParams& params); - -#ifdef __EXX -// Build V^EXX(R) into a real HContainer via add_HexxR (from exd/exc->get_Hexxs()) and write it. -// exd (real Hexx) and exc (complex Hexx) are mutually exclusive; picked by info_ri.real_number. -void write_h_exx(WriteHParams& params, const Exx_Info& exx_info); -#endif - -} // namespace ModuleIO - -#endif diff --git a/source/source_io/module_hs/write_hs.h b/source/source_io/module_hs/write_hs.h deleted file mode 100644 index 25c1cecc0cd..00000000000 --- a/source/source_io/module_hs/write_hs.h +++ /dev/null @@ -1,58 +0,0 @@ -#ifndef WRITE_HS_H -#define WRITE_HS_H - -#include -#include - -//#include "source_base/global_function.h" -//#include "source_base/global_variable.h" -#include "source_base/parallel_comm.h" // use DIAG_WORLD -#include "source_basis/module_ao/parallel_orbitals.h" // use Parallel_Orbitals -#include "source_hamilt/hamilt.h" - - -// mohan add this file 2010-09-10 -namespace ModuleIO -{ - template - void write_hsk( - const std::string &global_out_dir, - const int nspin, - const int nks, - const int nkstot, - const std::vector &ik2iktot, - const std::vector &isk, - hamilt::Hamilt* p_hamilt, - const Parallel_Orbitals &pv, - const bool gamma_only, - const bool out_app_flag, - const int istep, - const int out_type, - const int precision, - std::ofstream &ofs_running); - - /// @brief save a square matrix, such as H(k) and S(k) - /// @param[in] istep : the step of the calculation - /// @param[in] mat : the local matrix - /// @param[in] bit : true for binary, false for decimal - /// @param[in] tri : true for upper triangle, false for full matrix - /// @param[in] app : true for append, false for overwrite - /// @param[in] file_name : the name of the output file - /// @param[in] pv : the 2d-block parallelization information - /// @param[in] drank : the rank of the current process - template - void save_mat(const int istep, - const T* mat, - const int dim, - const bool bit, - const int precision, - const bool tri, - const bool app, - const std::string& file_name, - const Parallel_2D& pv, - const int drank, - const bool reduce = true); - -} -#include "write_hs.hpp" -#endif diff --git a/source/source_io/module_hs/write_hs.hpp b/source/source_io/module_hs/write_hs.hpp deleted file mode 100644 index 4b5d6ebf4d0..00000000000 --- a/source/source_io/module_hs/write_hs.hpp +++ /dev/null @@ -1,322 +0,0 @@ -#include "write_hs.h" - -#include "source_io/module_parameter/parameter.h" -#include "source_base/parallel_reduce.h" -#include "source_base/timer.h" -#include "source_base/tool_quit.h" -#include "source_cell/module_neighbor/sltk_grid_driver.h" -#include "source_base/module_out/filename.h" // use filename_output function - - -template -void ModuleIO::write_hsk( - const std::string &global_out_dir, - const int nspin, - const int nks, - const int nkstot, - const std::vector &ik2iktot, - const std::vector &isk, - hamilt::Hamilt* p_hamilt, - const Parallel_Orbitals &pv, - const bool gamma_only, - const bool out_app_flag, - const int istep, - const int out_type, - const int precision, - std::ofstream &ofs_running) -{ - - ofs_running << " >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>" - ">>>>>>>>>>>>>>>>>>>>>>>>>" << std::endl; - ofs_running << " | " - " |" << std::endl; - ofs_running << " | Write Hamiltonian matrix H(k) or overlap matrix S(k) in numerical |" << std::endl; - ofs_running << " | atomic orbitals at each k-point. |" << std::endl; - ofs_running << " | " - " |" << std::endl; - ofs_running << " >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>" - ">>>>>>>>>>>>>>>>>>>>>>>>>" << std::endl; - ofs_running << "\n WRITE H(k) OR S(k)" << std::endl; - - for (int ik = 0; ik < nks; ++ik) - { - p_hamilt->updateHk(ik); - const bool binary = (out_type == 2); - - hamilt::MatrixBlock h_mat; - hamilt::MatrixBlock s_mat; - - p_hamilt->matrix(h_mat, s_mat); - - std::string h_fn = ModuleIO::filename_output(global_out_dir, - "hk","nao",ik,ik2iktot,nspin,nkstot, - out_type,out_app_flag,gamma_only,istep); - - ModuleIO::save_mat(istep, - h_mat.p, - PARAM.globalv.nlocal, - binary, - precision, - 1, - out_app_flag, - h_fn, - pv, - GlobalV::DRANK); - - // mohan note 2025-06-02 - // for overlap matrix, the two spin channels yield the same matrix - // so we only need to print matrix from one spin channel. - const int current_spin = isk[ik]; - if(current_spin == 1) - { - continue; - } - - std::string s_fn = ModuleIO::filename_output(global_out_dir, - "sk","nao",ik,ik2iktot,nspin,nkstot, - out_type,out_app_flag,gamma_only,istep); - - ofs_running << " The output filename is " << s_fn << std::endl; - - ModuleIO::save_mat(istep, - s_mat.p, - PARAM.globalv.nlocal, - binary, - precision, - 1, - out_app_flag, - s_fn, - pv, - GlobalV::DRANK); - } // end ik -} - - -// output a square matrix -template -void ModuleIO::save_mat(const int istep, - const T* mat, - const int dim, - const bool bit, - const int precision, - const bool tri, - const bool app, - const std::string& filename, - const Parallel_2D& pv, - const int drank, - const bool reduce) -{ - ModuleBase::TITLE("ModuleIO", "save_mat"); - ModuleBase::timer::start("ModuleIO", "save_mat"); - - const bool gamma_only = std::is_same::value; - - // write .dat file - if (bit) - { -// write .dat file with MPI -#ifdef __MPI - FILE* out_matrix = nullptr; - - if (drank == 0) - { - const char* mode = (app && istep > 0) ? "ab" : "wb"; - out_matrix = fopen(filename.c_str(), mode); - if (out_matrix == nullptr) - { - ModuleBase::WARNING_QUIT("ModuleIO::save_mat", "Cannot open matrix file: " + filename); - } - fwrite(&dim, sizeof(int), 1, out_matrix); - } - - int ir=0; - int ic=0; - for (int i = 0; i < dim; ++i) - { - T* line = new T[tri ? dim - i : dim]; - ModuleBase::GlobalFunc::ZEROS(line, tri ? dim - i : dim); - - ir = pv.global2local_row(i); - if (ir >= 0) - { - // data collection - for (int j = (tri ? i : 0); j < dim; ++j) - { - ic = pv.global2local_col(j); - if (ic >= 0) - { - int iic; - if (ModuleBase::GlobalFunc::IS_COLUMN_MAJOR_KS_SOLVER(PARAM.inp.ks_solver)) - { - iic = ir + ic * pv.nrow; - } - else - { - iic = ir * pv.ncol + ic; - } - line[tri ? j - i : j] = mat[iic]; - } - } - } - - if (reduce) - { - Parallel_Reduce::reduce_all(line, tri ? dim - i : dim); - } - - if (drank == 0) - { - for (int j = (tri ? i : 0); j < dim; ++j) - { - fwrite(&line[tri ? j - i : j], sizeof(T), 1, out_matrix); - } - } - delete[] line; - - MPI_Barrier(DIAG_WORLD); - } - - if (drank == 0) - { - fclose(out_matrix); - } -// write .dat file without MPI -#else - const char* mode = (app && istep > 0) ? "ab" : "wb"; - FILE* out_matrix = fopen(filename.c_str(), mode); - if (out_matrix == nullptr) - { - ModuleBase::WARNING_QUIT("ModuleIO::save_mat", "Cannot open matrix file: " + filename); - } - - fwrite(&dim, sizeof(int), 1, out_matrix); - - for (int i = 0; i < dim; i++) - { - for (int j = (tri ? i : 0); j < dim; j++) - { - fwrite(&mat[i * dim + j], sizeof(T), 1, out_matrix); - } - } - fclose(out_matrix); -#endif - } // end writing .dat file - else // write .txt file - { - std::ofstream out_matrix; - out_matrix << std::scientific << std::setprecision(precision); -#ifdef __MPI - if (drank == 0) - { - if (app && istep > 0) - { - out_matrix.open(filename.c_str(), std::ofstream::app); - } - else - { - out_matrix.open(filename.c_str()); - } - if (!out_matrix.is_open()) - { - ModuleBase::WARNING_QUIT("ModuleIO::save_mat", "Cannot open matrix file: " + filename); - } - out_matrix << "#------------------------------------------------------------------------" << std::endl; - out_matrix << "# ionic step " << istep+1 << std::endl; // istep starts from 0 - out_matrix << "# filename " << filename << std::endl; - out_matrix << "# gamma only " << gamma_only << std::endl; - out_matrix << "# rows " << dim << std::endl; - out_matrix << "# columns " << dim << std::endl; - out_matrix << "#------------------------------------------------------------------------" << std::endl; - - } - - int ir=0; - int ic=0; - for (int i = 0; i < dim; i++) - { - T* line = new T[tri ? dim - i : dim]; - ModuleBase::GlobalFunc::ZEROS(line, tri ? dim - i : dim); - - ir = pv.global2local_row(i); - if (ir >= 0) - { - // data collection - for (int j = (tri ? i : 0); j < dim; ++j) - { - ic = pv.global2local_col(j); - if (ic >= 0) - { - int iic=0; - if (ModuleBase::GlobalFunc::IS_COLUMN_MAJOR_KS_SOLVER(PARAM.inp.ks_solver)) - { - iic = ir + ic * pv.nrow; - } - else - { - iic = ir * pv.ncol + ic; - } - line[tri ? j - i : j] = mat[iic]; - } - } - } - - if (reduce) - { - Parallel_Reduce::reduce_all(line, tri ? dim - i : dim); - } - - if (drank == 0) - { - out_matrix << "Row " << i+1 << std::endl; - size_t count = 0; - for (int j = (tri ? i : 0); j < dim; j++) - { - out_matrix << " " << line[tri ? j - i : j]; - ++count; - if(count%8==0) - { - if(j!=dim-1) - { - out_matrix << std::endl; - } - } - } - out_matrix << std::endl; - } - delete[] line; - } - - if (drank == 0) - { - out_matrix.close(); - } -#else - if (app) - { - out_matrix.open(filename.c_str(), std::ofstream::app); - } - else - { - out_matrix.open(filename.c_str()); - } - if (!out_matrix.is_open()) - { - ModuleBase::WARNING_QUIT("ModuleIO::save_mat", "Cannot open matrix file: " + filename); - } - - out_matrix << dim; - out_matrix << std::setprecision(precision); - for (int i = 0; i < dim; i++) - { - for (int j = (tri ? i : 0); j < dim; j++) - { - out_matrix << " " << mat[i * dim + j]; - } - out_matrix << std::endl; - } - out_matrix.close(); -#endif - } - ModuleBase::timer::end("ModuleIO", "save_mat"); - return; -} diff --git a/source/source_io/module_hs/write_hs_r.cpp b/source/source_io/module_hs/write_hs_r.cpp deleted file mode 100644 index 407bc39d1f3..00000000000 --- a/source/source_io/module_hs/write_hs_r.cpp +++ /dev/null @@ -1,669 +0,0 @@ -#include "write_hs_r.h" - -#include "source_base/module_out/sparse_matrix.h" -#include "source_base/timer.h" -#include "source_base/tool_quit.h" -#include "source_io/module_parameter/parameter.h" -#include "source_lcao/lcao_hs_arrays.hpp" -#include "source_lcao/spar_dh.h" -#include "source_lcao/spar_hsr.h" -#include "source_lcao/spar_st.h" -#include "write_hs_sparse.h" - -#include -#include -#include -#include -#include -#include - -// if 'binary=true', output binary file. -// The 'sparse_thr' is the accuracy of the sparse matrix. -// If the absolute value of the matrix element is less than or equal to the -// 'sparse_thr', it will be ignored. - -void ModuleIO::output_dSR(const int& istep, - const UnitCell& ucell, - const Parallel_Orbitals& pv, - LCAO_HS_Arrays& HS_Arrays, - const Grid_Driver& grid, // mohan add 2024-04-06 - const TwoCenterBundle& two_center_bundle, - const LCAO_Orbitals& orb, - const K_Vectors& kv, - const bool& binary, - const double& sparse_thr, - const int precision) -{ - ModuleBase::TITLE("ModuleIO", "output_dSR"); - ModuleBase::timer::start("ModuleIO", "output_dSR"); - - sparse_format::cal_dS(ucell, pv, HS_Arrays, grid, two_center_bundle, orb, sparse_thr); - - // mohan update 2024-04-01 - ModuleIO::save_dH_sparse(istep, pv, HS_Arrays, sparse_thr, binary, "s", precision); - - sparse_format::destroy_dH_R_sparse(HS_Arrays); - - ModuleBase::timer::end("ModuleIO", "output_dSR"); - return; -} - -void ModuleIO::output_dHR(const int& istep, - const ModuleBase::matrix& v_eff, - const UnitCell& ucell, - const Parallel_Orbitals& pv, - LCAO_HS_Arrays& HS_Arrays, - const Grid_Driver& grid, // mohan add 2024-04-06 - const TwoCenterBundle& two_center_bundle, - const LCAO_Orbitals& orb, - const K_Vectors& kv, - const bool& binary, - const double& sparse_thr, - const int precision) -{ - ModuleBase::TITLE("ModuleIO", "output_dHR"); - ModuleBase::timer::start("ModuleIO", "output_dHR"); - - GlobalV::ofs_running << " >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>" << std::endl; - GlobalV::ofs_running << " | |" << std::endl; - GlobalV::ofs_running << " | #Print out dH/dR# |" << std::endl; - GlobalV::ofs_running << " | |" << std::endl; - GlobalV::ofs_running << " >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>" << std::endl; - - const int nspin = PARAM.inp.nspin; - - if (nspin == 1 || nspin == 4) - { - // mohan add 2024-04-01 - const int cspin = 0; - - sparse_format::cal_dH(ucell, pv, HS_Arrays, grid, two_center_bundle, orb, cspin, sparse_thr, v_eff); - } - else if (nspin == 2) - { - for (int cspin = 0; cspin < 2; cspin++) - { - sparse_format::cal_dH(ucell, pv, HS_Arrays, grid, two_center_bundle, orb, cspin, sparse_thr, v_eff); - } - } - // mohan update 2024-04-01 - ModuleIO::save_dH_sparse(istep, pv, HS_Arrays, sparse_thr, binary, "h", precision); - - sparse_format::destroy_dH_R_sparse(HS_Arrays); - - ModuleBase::timer::end("ModuleIO", "output_dHR"); - return; -} - -template -void ModuleIO::output_SR(Parallel_Orbitals& pv, - const Grid_Driver& grid, - hamilt::Hamilt* p_ham, - const std::string& SR_filename, - const bool& binary, - const double& sparse_thr, - const int precision) -{ - ModuleBase::TITLE("ModuleIO", "output_SR"); - ModuleBase::timer::start("ModuleIO", "output_SR"); - - GlobalV::ofs_running << " >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>" << std::endl; - GlobalV::ofs_running << " | |" << std::endl; - GlobalV::ofs_running << " | #Print out overlap matrix S(R)# |" << std::endl; - GlobalV::ofs_running << " | |" << std::endl; - GlobalV::ofs_running << " >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>" << std::endl; - - std::cout << " Overlap matrix file is in " << SR_filename << std::endl; - GlobalV::ofs_running << " Overlap matrix file is in " << SR_filename << std::endl; - - LCAO_HS_Arrays HS_Arrays; - - sparse_format::cal_SR(pv, - HS_Arrays.all_R_coor, - HS_Arrays.SR_sparse, - HS_Arrays.SR_soc_sparse, - grid, - sparse_thr, - p_ham); - - const int istep = 0; - ModuleIO::SparseWriteOptions options; - options.filename = SR_filename; - options.label = "S"; - options.threshold = sparse_thr; - options.binary = binary; - options.precision = precision; - options.istep = istep; - options.reduce = true; - options.temp_dir = PARAM.globalv.global_out_dir; - - if (PARAM.inp.nspin == 4) - { - ModuleIO::save_sparse(HS_Arrays.SR_soc_sparse, - HS_Arrays.all_R_coor, - pv, - options); - } - else - { - ModuleIO::save_sparse(HS_Arrays.SR_sparse, - HS_Arrays.all_R_coor, - pv, - options); - } - - sparse_format::destroy_HS_R_sparse(HS_Arrays); - - ModuleBase::timer::end("ModuleIO", "output_SR"); - return; -} - -void ModuleIO::output_TR(const int istep, - const UnitCell& ucell, - const Parallel_Orbitals& pv, - LCAO_HS_Arrays& HS_Arrays, - const Grid_Driver& grid, - const TwoCenterBundle& two_center_bundle, - const LCAO_Orbitals& orb, - const std::string& TR_filename, - const bool& binary, - const double& sparse_thr, - const int precision) -{ - ModuleBase::TITLE("ModuleIO", "output_TR"); - ModuleBase::timer::start("ModuleIO", "output_TR"); - - GlobalV::ofs_running << " >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>" << std::endl; - GlobalV::ofs_running << " | |" << std::endl; - GlobalV::ofs_running << " | #Print out kinetic energy term matrix T(R)# |" << std::endl; - GlobalV::ofs_running << " | |" << std::endl; - GlobalV::ofs_running << " >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>" << std::endl; - - std::stringstream sst; - if (PARAM.inp.calculation == "md" && !PARAM.inp.out_app_flag) - { - sst << PARAM.globalv.global_matrix_dir << TR_filename << "g" << istep; - GlobalV::ofs_running << " T(R) data are in file: " << sst.str() << std::endl; - } - else - { - sst << PARAM.globalv.global_out_dir << TR_filename; - GlobalV::ofs_running << " T(R) data are in file: " << sst.str() << std::endl; - } - - sparse_format::cal_TR(ucell, pv, HS_Arrays, grid, two_center_bundle, orb, sparse_thr); - ModuleIO::SparseWriteOptions options; - options.filename = sst.str(); - options.label = "T"; - options.threshold = sparse_thr; - options.binary = binary; - options.precision = precision; - options.istep = istep; - options.reduce = true; - options.temp_dir = PARAM.globalv.global_out_dir; - - ModuleIO::save_sparse(HS_Arrays.TR_sparse, - HS_Arrays.all_R_coor, - pv, - options); - - sparse_format::destroy_T_R_sparse(HS_Arrays); - - ModuleBase::timer::end("ModuleIO", "output_TR"); - return; -} - -template void ModuleIO::output_SR(Parallel_Orbitals& pv, - const Grid_Driver& grid, - hamilt::Hamilt* p_ham, - const std::string& SR_filename, - const bool& binary, - const double& sparse_thr, - const int precision); -template void ModuleIO::output_SR>(Parallel_Orbitals& pv, - const Grid_Driver& grid, - hamilt::Hamilt>* p_ham, - const std::string& SR_filename, - const bool& binary, - const double& sparse_thr, - const int precision); - -#include "source_hamilt/module_hcontainer/hcontainer_funcs.h" -#include "source_hamilt/module_hcontainer/output_hcontainer.h" -#include "source_cell/ucell_io.h" - -std::string ModuleIO::hsr_gen_fname(const std::string& prefix, - const int ispin, - const bool append, - const int istep) -{ - return hsr_gen_fname(prefix, ispin, append, istep, 1); -} - -std::string ModuleIO::hsr_gen_fname(const std::string& prefix, - const int ispin, - const bool append, - const int istep, - const int out_type) -{ - const std::string extension = out_type == 2 ? ".dat" : ".csr"; - if (!append && istep >= 0) - { - return prefix + std::to_string(ispin + 1) + "g" + std::to_string(istep + 1) + "_nao" + extension; - } - else - { - return prefix + std::to_string(ispin + 1) + "_nao" + extension; - } -} - -std::string ModuleIO::sr_gen_fname(const bool append, const int istep) -{ - return sr_gen_fname(append, istep, 1); -} - -std::string ModuleIO::sr_gen_fname(const bool append, const int istep, const int out_type) -{ - const std::string extension = out_type == 2 ? ".dat" : ".csr"; - if (!append && istep >= 0) - { - return "srg" + std::to_string(istep + 1) + "_nao" + extension; - } - return "sr_nao" + extension; -} - -std::string ModuleIO::dhr_gen_fname(const std::string& prefix, - const int ispin, - const bool append, - const int istep) -{ - std::string fname = prefix + "rs" + std::to_string(ispin + 1); - if (!append && istep >= 0) - { - fname += "g" + std::to_string(istep + 1); - } - fname += "_nao.csr"; - return fname; -} - -template -void ModuleIO::write_hcontainer_csr(const std::string& fname, - const UnitCell* ucell, - const int precision, - hamilt::HContainer* mat_serial, - const int istep, - const int ispin, - const int nspin, - const std::string& label, - const std::string& representation_note) -{ - std::ofstream ofs; - if (istep <= 0) - { - ofs.open(fname); - } - else - { - ofs.open(fname, std::ios::app); - } - if (!ofs.is_open()) - { - ModuleBase::WARNING_QUIT("ModuleIO::write_hcontainer_csr", - "Cannot open HContainer CSR file: " + fname); - } - - ofs << " --- Ionic Step " << istep + 1 << " ---" << std::endl; - ofs << " # print " << label << " matrix in real space " << label << "(R)" << std::endl; - ofs << " " << nspin << " # number of spin directions" << std::endl; - ofs << " " << ispin + 1 << " # spin index" << std::endl; - ofs << " " << mat_serial->get_nbasis() << " # number of localized basis" << std::endl; - ofs << " " << mat_serial->size_R_loop() << " # number of Bravais lattice vector R" << std::endl; - ofs << std::endl; - - ModuleIO::UcellIO::write_ucell(ofs, ucell); - if (!representation_note.empty()) - { - ofs << "# representation: " << representation_note << std::endl; - } - ofs << std::endl; - - const double sparse_threshold = 1e-10; - hamilt::Output_HContainer out(mat_serial, ofs, sparse_threshold, precision); - out.write(); - ofs.close(); -} - -namespace -{ -template -void write_native_value(std::ofstream& ofs, const T& value) -{ - ofs.write(reinterpret_cast(&value), sizeof(T)); -} - -void write_native_value(std::ofstream& ofs, const std::complex& value) -{ - const double real = value.real(); - const double imag = value.imag(); - write_native_value(ofs, real); - write_native_value(ofs, imag); -} - -template -ModuleIO::SparseMatrix make_sparse_R_block(hamilt::HContainer* mat_serial, - const int rx, - const int ry, - const int rz, - const double sparse_threshold) -{ - const int nbasis = mat_serial->get_nbasis(); - ModuleIO::SparseMatrix sparse_matrix(nbasis, nbasis); - sparse_matrix.setSparseThreshold(sparse_threshold); - mat_serial->fix_R(rx, ry, rz); - - for (int iap = 0; iap < mat_serial->size_atom_pairs(); ++iap) - { - const auto atom_pair = mat_serial->get_atom_pair(iap); - const int r_index = atom_pair.find_R(rx, ry, rz); - if (r_index < 0) - { - continue; - } - const auto matrix_info = atom_pair.get_matrix_values(r_index); - const int* index = std::get<0>(matrix_info).data(); - TR* data = std::get<1>(matrix_info); - for (int irow = index[0]; irow < index[0] + index[1]; ++irow) - { - for (int icol = index[2]; icol < index[2] + index[3]; ++icol) - { - sparse_matrix.insert(irow, icol, *data); - ++data; - } - } - } - - mat_serial->unfix_R(); - return sparse_matrix; -} -} // namespace - -template -void ModuleIO::write_hcontainer_csr_binary(const std::string& fname, - hamilt::HContainer* mat_serial, - const int istep, - const bool append) -{ - std::ios_base::openmode mode = std::ios::out | std::ios::binary; - if (append && istep > 0) - { - mode |= std::ios::app; - } - std::ofstream ofs(fname.c_str(), mode); - if (!ofs.is_open()) - { - ModuleBase::WARNING_QUIT("ModuleIO::write_hcontainer_csr_binary", - "Cannot open HContainer binary CSR file: " + fname); - } - - const size_t nR_size = mat_serial->size_R_loop(); - if (nR_size > static_cast(std::numeric_limits::max())) - { - ModuleBase::WARNING_QUIT("ModuleIO::write_hcontainer_csr_binary", - "Too many R blocks for native binary CSR output: " + fname); - } - - std::vector> R_coordinates; - R_coordinates.reserve(nR_size); - for (size_t iR = 0; iR < nR_size; ++iR) - { - int rx = 0; - int ry = 0; - int rz = 0; - mat_serial->loop_R(iR, rx, ry, rz); - R_coordinates.push_back(std::make_tuple(rx, ry, rz)); - } - std::sort(R_coordinates.begin(), R_coordinates.end()); - - const int step = std::max(istep, 0); - const int nbasis = mat_serial->get_nbasis(); - const int nR = static_cast(R_coordinates.size()); - write_native_value(ofs, step); - write_native_value(ofs, nbasis); - write_native_value(ofs, nR); - - const double sparse_threshold = 1e-10; - for (const auto& R_coordinate: R_coordinates) - { - const int rx = std::get<0>(R_coordinate); - const int ry = std::get<1>(R_coordinate); - const int rz = std::get<2>(R_coordinate); - const SparseMatrix sparse_matrix - = make_sparse_R_block(mat_serial, rx, ry, rz, sparse_threshold); - const auto& elements = sparse_matrix.getElements(); - if (elements.size() > static_cast(std::numeric_limits::max())) - { - ModuleBase::WARNING_QUIT("ModuleIO::write_hcontainer_csr_binary", - "Too many nonzero values for native binary CSR output: " + fname); - } - const int nnz = static_cast(elements.size()); - - write_native_value(ofs, rx); - write_native_value(ofs, ry); - write_native_value(ofs, rz); - write_native_value(ofs, nnz); - - for (const auto& element: elements) - { - write_native_value(ofs, element.second); - } - - std::vector row_ptr(nbasis + 1, 0); - for (const auto& element: elements) - { - write_native_value(ofs, element.first.second); - ++row_ptr[element.first.first + 1]; - } - for (int irow = 1; irow <= nbasis; ++irow) - { - row_ptr[irow] += row_ptr[irow - 1]; - } - for (const long long pointer: row_ptr) - { - write_native_value(ofs, pointer); - } - } - - ofs.close(); - if (!ofs) - { - ModuleBase::WARNING_QUIT("ModuleIO::write_hcontainer_csr_binary", - "Failed to write HContainer binary CSR file: " + fname); - } -} - -template -void ModuleIO::write_hsr(const std::vector*>& hr_vec, - const hamilt::HContainer* sr, - const UnitCell* ucell, - const int out_type, - const int precision, - const Parallel_2D& paraV, - const bool append, - const bool gamma_only, - const int* iat2iwt, - const int nat, - const int istep) -{ - if (out_type != 1 && out_type != 2) - { - ModuleBase::WARNING_QUIT("ModuleIO::write_hsr", "out_type must be 1 or 2"); - } - const int nspin = hr_vec.size(); - assert(nspin > 0); - const std::string representation_note - = gamma_only - ? "gamma-only folded matrix; stored R-space contributions are summed into R = (0, 0, 0)" - : ""; - - // Output HR (one file per spin) - for (int ispin = 0; ispin < nspin; ispin++) - { - const int nbasis = hr_vec[ispin]->get_nbasis(); - -#ifdef __MPI - Parallel_Orbitals serialV; - serialV.set_serial(nbasis, nbasis); - serialV.set_atomic_trace(iat2iwt, nat, nbasis); - hamilt::HContainer hr_serial(&serialV); - hamilt::gatherParallels(*hr_vec[ispin], &hr_serial, 0); -#else - hamilt::HContainer hr_serial(*hr_vec[ispin]); -#endif - - if (GlobalV::MY_RANK == 0) - { - std::string fname = PARAM.globalv.global_out_dir - + hsr_gen_fname("hrs", ispin, append, istep, out_type); - if (out_type == 2) - { - write_hcontainer_csr_binary(fname, &hr_serial, istep, append); - } - else - { - write_hcontainer_csr( - fname, ucell, precision, &hr_serial, istep, ispin, nspin, "H", representation_note); - } - } - } - - // Output SR (single file) - { - const int nbasis = sr->get_nbasis(); - -#ifdef __MPI - Parallel_Orbitals serialV; - serialV.set_serial(nbasis, nbasis); - serialV.set_atomic_trace(iat2iwt, nat, nbasis); - hamilt::HContainer sr_serial(&serialV); - hamilt::gatherParallels(*sr, &sr_serial, 0); -#else - hamilt::HContainer sr_serial(*sr); -#endif - - if (GlobalV::MY_RANK == 0) - { - std::string fname = PARAM.globalv.global_out_dir - + sr_gen_fname(append, istep, out_type); - if (out_type == 2) - { - write_hcontainer_csr_binary(fname, &sr_serial, istep, append); - } - else - { - write_hcontainer_csr( - fname, ucell, precision, &sr_serial, istep, 0, 1, "S", representation_note); - } - } - } -} - -// Explicit instantiations -template void ModuleIO::write_hcontainer_csr( - const std::string&, const UnitCell*, const int, - hamilt::HContainer*, const int, const int, const int, const std::string&, const std::string&); -template void ModuleIO::write_hcontainer_csr>( - const std::string&, const UnitCell*, const int, - hamilt::HContainer>*, const int, const int, const int, const std::string&, const std::string&); - -template void ModuleIO::write_hcontainer_csr_binary( - const std::string&, hamilt::HContainer*, const int, const bool); -template void ModuleIO::write_hcontainer_csr_binary>( - const std::string&, hamilt::HContainer>*, const int, const bool); - -template void ModuleIO::write_hsr( - const std::vector*>&, - const hamilt::HContainer*, - const UnitCell*, const int, const int, const Parallel_2D&, - const bool, const bool, const int*, const int, const int); -template void ModuleIO::write_hsr>( - const std::vector>*>&, - const hamilt::HContainer>*, - const UnitCell*, const int, const int, const Parallel_2D&, - const bool, const bool, const int*, const int, const int); - - -template -void ModuleIO::write_matrix_r(const std::string& matrix_label, - const std::string& description, - const std::vector*>& matrices, - const UnitCell* ucell, - const int precision, - const Parallel_2D& paraV, - const bool append, - const int* iat2iwt, - const int nat, - const int istep) -{ - const int nspin = matrices.size(); - assert(nspin > 0); - - for (int ispin = 0; ispin < nspin; ispin++) - { - const int nbasis = matrices[ispin]->get_nbasis(); - - // Generate filename - std::string fname = dhr_gen_fname(matrix_label, ispin, append, istep); - if (PARAM.inp.calculation == "md" && !PARAM.inp.out_app_flag) - { - fname = PARAM.globalv.global_matrix_dir + fname; - } - else - { - fname = PARAM.globalv.global_out_dir + fname; - } - - // Gather parallel matrix to serial -#ifdef __MPI - Parallel_Orbitals serialV; - serialV.set_serial(nbasis, nbasis); - serialV.set_atomic_trace(iat2iwt, nat, nbasis); - - hamilt::HContainer matrix_serial(&serialV); - hamilt::gatherParallels(*matrices[ispin], &matrix_serial, 0); - - if (GlobalV::MY_RANK == 0) - { - write_hcontainer_csr( - fname, ucell, precision, &matrix_serial, istep, ispin, nspin, description, ""); - } -#else - write_hcontainer_csr( - fname, ucell, precision, matrices[ispin], istep, ispin, nspin, description, ""); -#endif - } -} - -// Template instantiations -template void ModuleIO::write_matrix_r( - const std::string&, - const std::string&, - const std::vector*>&, - const UnitCell*, - const int, - const Parallel_2D&, - const bool, - const int*, - const int, - const int); - -template void ModuleIO::write_matrix_r>( - const std::string&, - const std::string&, - const std::vector>*>&, - const UnitCell*, - const int, - const Parallel_2D&, - const bool, - const int*, - const int, - const int); diff --git a/source/source_io/module_hs/write_hs_r.h b/source/source_io/module_hs/write_hs_r.h deleted file mode 100644 index 4be27fa2194..00000000000 --- a/source/source_io/module_hs/write_hs_r.h +++ /dev/null @@ -1,143 +0,0 @@ -#ifndef WRITE_HS_R_H -#define WRITE_HS_R_H - -#include "source_base/matrix.h" -#include "source_basis/module_ao/parallel_orbitals.h" -#include "source_basis/module_nao/two_center_bundle.h" -#include "source_cell/klist.h" -#include "source_cell/module_neighbor/sltk_grid_driver.h" -#include "source_hamilt/hamilt.h" -#include "source_lcao/lcao_hs_arrays.hpp" - -namespace hamilt -{ -template -class HContainer; -} // namespace hamilt - -#ifdef __EXX -#include "RI/global/Tensor.h" // for RI::Tensor -#endif - -namespace ModuleIO -{ -void output_dHR(const int& istep, - const ModuleBase::matrix& v_eff, - const UnitCell& ucell, - const Parallel_Orbitals& pv, - LCAO_HS_Arrays& HS_Arrays, - const Grid_Driver& grid, // mohan add 2024-04-06 - const TwoCenterBundle& two_center_bundle, - const LCAO_Orbitals& orb, - const K_Vectors& kv, - const bool& binary = false, - const double& sparse_threshold = 1e-10, - const int precision = 16); - -void output_dSR(const int& istep, - const UnitCell& ucell, - const Parallel_Orbitals& pv, - LCAO_HS_Arrays& HS_Arrays, - const Grid_Driver& grid, // mohan add 2024-04-06 - const TwoCenterBundle& two_center_bundle, - const LCAO_Orbitals& orb, - const K_Vectors& kv, - const bool& binary = false, - const double& sparse_thr = 1e-10, - const int precision = 16); - -void output_TR(const int istep, - const UnitCell& ucell, - const Parallel_Orbitals& pv, - LCAO_HS_Arrays& HS_Arrays, - const Grid_Driver& grid, - const TwoCenterBundle& two_center_bundle, - const LCAO_Orbitals& orb, - const std::string& TR_filename = "trs1_nao.csr", - const bool& binary = false, - const double& sparse_threshold = 1e-10, - const int precision = 16); - -template -void output_SR(Parallel_Orbitals& pv, - const Grid_Driver& grid, - hamilt::Hamilt* p_ham, - const std::string& SR_filename = "sr_nao.csr", - const bool& binary = false, - const double& sparse_threshold = 1e-10, - const int precision = 16); - -/// Generate filename for spin-dependent HR output. -std::string hsr_gen_fname(const std::string& prefix, - const int ispin, - const bool append, - const int istep); - -/// Generate filename for spin-dependent HR output in the selected format. -std::string hsr_gen_fname(const std::string& prefix, - const int ispin, - const bool append, - const int istep, - const int out_type); - -/// Generate filename for spin-independent SR output. -std::string sr_gen_fname(const bool append, const int istep); - -/// Generate filename for spin-independent SR output in the selected format. -std::string sr_gen_fname(const bool append, const int istep, const int out_type); - -/// Generate filename for derivative matrices (dH/dR, dS/dR). -std::string dhr_gen_fname(const std::string& prefix, - const int ispin, - const bool append, - const int istep); - -/// Write a single HContainer to CSR file with header. -template -void write_hcontainer_csr(const std::string& fname, - const UnitCell* ucell, - const int precision, - hamilt::HContainer* mat_serial, - const int istep, - const int ispin, - const int nspin, - const std::string& label, - const std::string& representation_note); - -/// Write one HContainer record in the native binary CSR format. -template -void write_hcontainer_csr_binary(const std::string& fname, - hamilt::HContainer* mat_serial, - const int istep, - const bool append); - -/// Write H(R) and S(R) in CSR format, unified with write_dmr interface. -template -void write_hsr(const std::vector*>& hr_vec, - const hamilt::HContainer* sr, - const UnitCell* ucell, - const int out_type, - const int precision, - const Parallel_2D& paraV, - const bool append, - const bool gamma_only, - const int* iat2iwt, - const int nat, - const int istep); - -/// Write real-space matrix in CSR format (generic interface). -template -void write_matrix_r(const std::string& matrix_label, - const std::string& description, - const std::vector*>& matrices, - const UnitCell* ucell, - const int precision, - const Parallel_2D& paraV, - const bool append, - const int* iat2iwt, - const int nat, - const int istep); - -} // namespace ModuleIO - -#endif diff --git a/source/source_io/module_hs/write_hs_sparse.cpp b/source/source_io/module_hs/write_hs_sparse.cpp deleted file mode 100644 index 00458fc9b46..00000000000 --- a/source/source_io/module_hs/write_hs_sparse.cpp +++ /dev/null @@ -1,526 +0,0 @@ -#include "write_hs_sparse.h" - -#include "source_base/global_function.h" -#include "source_base/parallel_reduce.h" -#include "source_base/timer.h" -#include "source_io/module_parameter/parameter.h" -#include "source_lcao/module_rt/td_info.h" -#include "single_r_io.h" - -#include -#include -#include -#include - -namespace -{ -template -std::vector count_nonzeros_by_R( - const ModuleIO::SparseRMatrix& smat, - const std::set& all_R_coor, - const double threshold, - const bool reduce) -{ - std::vector nonzero_num(all_R_coor.size(), 0); - int count = 0; - for (const auto& R_coor: all_R_coor) - { - const auto iter = smat.find(R_coor); - if (iter != smat.end()) - { - for (const auto& row_loop: iter->second) - { - for (const auto& col_value: row_loop.second) - { - if (std::abs(col_value.second) > threshold) - { - ++nonzero_num[count]; - } - } - } - } - ++count; - } - - if (reduce) - { - Parallel_Reduce::reduce_all(nonzero_num.data(), static_cast(nonzero_num.size())); - } - return nonzero_num; -} - -int count_output_R(const std::vector& nonzero_num) -{ - int output_R_number = 0; - for (const long long count: nonzero_num) - { - if (count != 0) - { - ++output_R_number; - } - } - return output_R_number; -} - -void open_sparse_file(std::ofstream& ofs, const ModuleIO::SparseWriteOptions& options) -{ - std::ios_base::openmode mode = std::ios::out; - if (options.binary) - { - mode |= std::ios::binary; - } - if (PARAM.inp.calculation == "md" && PARAM.inp.out_app_flag && options.istep) - { - mode |= std::ios::app; - } - ofs.open(options.filename.c_str(), mode); - if (!ofs.is_open()) - { - ModuleBase::WARNING_QUIT("ModuleIO::open_sparse_file", - "Cannot open sparse matrix file: " + options.filename); - } -} - -void write_sparse_header(std::ofstream& ofs, - const ModuleIO::SparseWriteOptions& options, - const int nlocal, - const int output_R_number) -{ - const int step = std::max(options.istep, 0); - if (options.binary) - { - ofs.write(reinterpret_cast(&step), sizeof(int)); - ofs.write(reinterpret_cast(&nlocal), sizeof(int)); - ofs.write(reinterpret_cast(&output_R_number), sizeof(int)); - } - else - { - ofs << "STEP: " << step << std::endl; - ofs << "Matrix Dimension of " + options.label + "(R): " << nlocal - << std::endl; - ofs << "Matrix number of " + options.label + "(R): " - << output_R_number << std::endl; - } -} - -void write_R_record(std::ofstream& ofs, - const ModuleIO::RCoordinate& R_coor, - const long long nonzero_count, - const bool binary) -{ - int dRx = R_coor.x; - int dRy = R_coor.y; - int dRz = R_coor.z; - if (binary) - { - const int count = static_cast(nonzero_count); - ofs.write(reinterpret_cast(&dRx), sizeof(int)); - ofs.write(reinterpret_cast(&dRy), sizeof(int)); - ofs.write(reinterpret_cast(&dRz), sizeof(int)); - ofs.write(reinterpret_cast(&count), sizeof(int)); - } - else - { - ofs << dRx << " " << dRy << " " << dRz << " " << nonzero_count - << std::endl; - } -} - -void check_output_file_open(const std::ofstream& ofs, - const std::string& filename, - const std::string& context) -{ - if (!ofs.is_open()) - { - ModuleBase::WARNING_QUIT(context, "Cannot open sparse matrix file: " + filename); - } -} -} // namespace - -void ModuleIO::save_dH_sparse(const int& istep, - const Parallel_Orbitals& pv, - LCAO_HS_Arrays& HS_Arrays, - const double& sparse_thr, - const bool& binary, - const std::string& fileflag, - const int precision) { - ModuleBase::TITLE("ModuleIO", "save_dH_sparse"); - ModuleBase::timer::start("ModuleIO", "save_dH_sparse"); - SparseWriteOptions single_R_options; - single_R_options.threshold = sparse_thr; - single_R_options.binary = binary; - single_R_options.precision = precision; - single_R_options.reduce = true; - single_R_options.temp_dir = PARAM.globalv.global_out_dir; - - auto& all_R_coor_ptr = HS_Arrays.all_R_coor; - auto& output_R_coor_ptr = HS_Arrays.output_R_coor; - auto& dHRx_sparse_ptr = HS_Arrays.dHRx_sparse; - auto& dHRx_soc_sparse_ptr = HS_Arrays.dHRx_soc_sparse; - auto& dHRy_sparse_ptr = HS_Arrays.dHRy_sparse; - auto& dHRy_soc_sparse_ptr = HS_Arrays.dHRy_soc_sparse; - auto& dHRz_sparse_ptr = HS_Arrays.dHRz_sparse; - auto& dHRz_soc_sparse_ptr = HS_Arrays.dHRz_soc_sparse; - - const int total_R_num = static_cast(all_R_coor_ptr.size()); - int output_R_number = 0; - std::vector dHx_nonzero_num[2]; - std::vector dHy_nonzero_num[2]; - std::vector dHz_nonzero_num[2]; - int step = istep; - - int spin_loop = 1; - if (PARAM.inp.nspin == 2) { - spin_loop = 2; - } - - if (PARAM.inp.nspin != 4) - { - for (int ispin = 0; ispin < spin_loop; ++ispin) - { - dHx_nonzero_num[ispin] = count_nonzeros_by_R(dHRx_sparse_ptr[ispin], all_R_coor_ptr, sparse_thr, true); - dHy_nonzero_num[ispin] = count_nonzeros_by_R(dHRy_sparse_ptr[ispin], all_R_coor_ptr, sparse_thr, true); - dHz_nonzero_num[ispin] = count_nonzeros_by_R(dHRz_sparse_ptr[ispin], all_R_coor_ptr, sparse_thr, true); - } - } - else - { - dHx_nonzero_num[0] = count_nonzeros_by_R(dHRx_soc_sparse_ptr, all_R_coor_ptr, sparse_thr, true); - dHy_nonzero_num[0] = count_nonzeros_by_R(dHRy_soc_sparse_ptr, all_R_coor_ptr, sparse_thr, true); - dHz_nonzero_num[0] = count_nonzeros_by_R(dHRz_soc_sparse_ptr, all_R_coor_ptr, sparse_thr, true); - } - - const auto has_output_R = [&](const int index) { - for (int ispin = 0; ispin < spin_loop; ++ispin) - { - if (dHx_nonzero_num[ispin][index] != 0 - || dHy_nonzero_num[ispin][index] != 0 - || dHz_nonzero_num[ispin][index] != 0) - { - return true; - } - } - return false; - }; - - for (int index = 0; index < total_R_num; ++index) - { - if (has_output_R(index)) - { - output_R_number++; - } - } - - std::stringstream sshx[2]; - std::stringstream sshy[2]; - std::stringstream sshz[2]; - - if (PARAM.inp.calculation == "md" && !PARAM.inp.out_app_flag) - { - sshx[0] << PARAM.globalv.global_matrix_dir - << "d"<(&step), sizeof(int)); - g1x[ispin].write(reinterpret_cast(&nlocal), - sizeof(int)); - g1x[ispin].write(reinterpret_cast(&output_R_number), - sizeof(int)); - - g1y[ispin].write(reinterpret_cast(&step), sizeof(int)); - g1y[ispin].write(reinterpret_cast(&nlocal), - sizeof(int)); - g1y[ispin].write(reinterpret_cast(&output_R_number), - sizeof(int)); - - g1z[ispin].write(reinterpret_cast(&step), sizeof(int)); - g1z[ispin].write(reinterpret_cast(&nlocal), - sizeof(int)); - g1z[ispin].write(reinterpret_cast(&output_R_number), - sizeof(int)); - } - } - else - { - for (int ispin = 0; ispin < spin_loop; ++ispin) - { - if (PARAM.inp.calculation == "md" && PARAM.inp.out_app_flag && step) - { - g1x[ispin].open(sshx[ispin].str().c_str(), std::ios::app); - g1y[ispin].open(sshy[ispin].str().c_str(), std::ios::app); - g1z[ispin].open(sshz[ispin].str().c_str(), std::ios::app); - } - else - { - GlobalV::ofs_running << " dH/dRx data are in file: " << sshx[ispin].str() << std::endl; - GlobalV::ofs_running << " dH/dRy data are in file: " << sshy[ispin].str() << std::endl; - GlobalV::ofs_running << " dH/dRz data are in file: " << sshz[ispin].str() << std::endl; - g1x[ispin].open(sshx[ispin].str().c_str()); - g1y[ispin].open(sshy[ispin].str().c_str()); - g1z[ispin].open(sshz[ispin].str().c_str()); - } - check_output_file_open(g1x[ispin], sshx[ispin].str(), "ModuleIO::save_dH_sparse"); - check_output_file_open(g1y[ispin], sshy[ispin].str(), "ModuleIO::save_dH_sparse"); - check_output_file_open(g1z[ispin], sshz[ispin].str(), "ModuleIO::save_dH_sparse"); - - g1x[ispin] << "STEP: " << step << std::endl; - g1x[ispin] << "Matrix Dimension of dHx(R): " << PARAM.globalv.nlocal - << std::endl; - g1x[ispin] << "Matrix number of dHx(R): " << output_R_number - << std::endl; - - g1y[ispin] << "STEP: " << step << std::endl; - g1y[ispin] << "Matrix Dimension of dHy(R): " << PARAM.globalv.nlocal - << std::endl; - g1y[ispin] << "Matrix number of dHy(R): " << output_R_number - << std::endl; - - g1z[ispin] << "STEP: " << step << std::endl; - g1z[ispin] << "Matrix Dimension of dHz(R): " << PARAM.globalv.nlocal - << std::endl; - g1z[ispin] << "Matrix number of dHz(R): " << output_R_number - << std::endl; - } - } - } - - output_R_coor_ptr.clear(); - - int count = 0; - for (auto& R_coor: all_R_coor_ptr) { - int dRx = R_coor.x; - int dRy = R_coor.y; - int dRz = R_coor.z; - - if (!has_output_R(count)) - { - count++; - continue; - } - - output_R_coor_ptr.insert(R_coor); - - if (GlobalV::DRANK == 0) { - if (binary) { - for (int ispin = 0; ispin < spin_loop; ++ispin) { - const int dHx_count = static_cast(dHx_nonzero_num[ispin][count]); - const int dHy_count = static_cast(dHy_nonzero_num[ispin][count]); - const int dHz_count = static_cast(dHz_nonzero_num[ispin][count]); - g1x[ispin].write(reinterpret_cast(&dRx), - sizeof(int)); - g1x[ispin].write(reinterpret_cast(&dRy), - sizeof(int)); - g1x[ispin].write(reinterpret_cast(&dRz), - sizeof(int)); - g1x[ispin].write(reinterpret_cast(&dHx_count), - sizeof(int)); - - g1y[ispin].write(reinterpret_cast(&dRx), - sizeof(int)); - g1y[ispin].write(reinterpret_cast(&dRy), - sizeof(int)); - g1y[ispin].write(reinterpret_cast(&dRz), - sizeof(int)); - g1y[ispin].write(reinterpret_cast(&dHy_count), - sizeof(int)); - - g1z[ispin].write(reinterpret_cast(&dRx), - sizeof(int)); - g1z[ispin].write(reinterpret_cast(&dRy), - sizeof(int)); - g1z[ispin].write(reinterpret_cast(&dRz), - sizeof(int)); - g1z[ispin].write(reinterpret_cast(&dHz_count), - sizeof(int)); - } - } else { - for (int ispin = 0; ispin < spin_loop; ++ispin) { - g1x[ispin] << dRx << " " << dRy << " " << dRz << " " - << dHx_nonzero_num[ispin][count] << std::endl; - g1y[ispin] << dRx << " " << dRy << " " << dRz << " " - << dHy_nonzero_num[ispin][count] << std::endl; - g1z[ispin] << dRx << " " << dRy << " " << dRz << " " - << dHz_nonzero_num[ispin][count] << std::endl; - } - } - } - - for (int ispin = 0; ispin < spin_loop; ++ispin) { - if (dHx_nonzero_num[ispin][count] > 0) { - if (PARAM.inp.nspin != 4) { - output_single_R(g1x[ispin], - dHRx_sparse_ptr[ispin][R_coor], - pv, - single_R_options); - } else { - output_single_R(g1x[ispin], - dHRx_soc_sparse_ptr[R_coor], - pv, - single_R_options); - } - } - if (dHy_nonzero_num[ispin][count] > 0) { - if (PARAM.inp.nspin != 4) { - output_single_R(g1y[ispin], - dHRy_sparse_ptr[ispin][R_coor], - pv, - single_R_options); - } else { - output_single_R(g1y[ispin], - dHRy_soc_sparse_ptr[R_coor], - pv, - single_R_options); - } - } - if (dHz_nonzero_num[ispin][count] > 0) { - if (PARAM.inp.nspin != 4) { - output_single_R(g1z[ispin], - dHRz_sparse_ptr[ispin][R_coor], - pv, - single_R_options); - } else { - output_single_R(g1z[ispin], - dHRz_soc_sparse_ptr[R_coor], - pv, - single_R_options); - } - } - } - - count++; - } - - if (GlobalV::DRANK == 0) { - for (int ispin = 0; ispin < spin_loop; ++ispin) { - g1x[ispin].close(); - } - for (int ispin = 0; ispin < spin_loop; ++ispin) { - g1y[ispin].close(); - } - for (int ispin = 0; ispin < spin_loop; ++ispin) { - g1z[ispin].close(); - } - } - - ModuleBase::timer::end("ModuleIO", "save_dH_sparse"); - return; -} - -template -void ModuleIO::save_sparse( - const SparseRMatrix& smat, - const std::set& all_R_coor, - const Parallel_Orbitals& pv, - const SparseWriteOptions& options) { - ModuleBase::TITLE("ModuleIO", "save_sparse"); - ModuleBase::timer::start("ModuleIO", "save_sparse"); - const int nlocal = pv.get_global_row_size(); - if (nlocal <= 0) - { - ModuleBase::WARNING_QUIT("ModuleIO::save_sparse", - "Parallel_Orbitals global row size must be positive."); - } - - const std::vector nonzero_num - = count_nonzeros_by_R(smat, all_R_coor, options.threshold, options.reduce); - const int output_R_number = count_output_R(nonzero_num); - std::ofstream ofs; - if (!options.reduce || GlobalV::DRANK == 0) - { - open_sparse_file(ofs, options); - write_sparse_header(ofs, options, nlocal, output_R_number); - } - - int count = 0; - for (const auto& R_coor: all_R_coor) - { - if (nonzero_num[count] == 0) - { - count++; - continue; - } - - if (!options.reduce || GlobalV::DRANK == 0) - { - write_R_record(ofs, R_coor, nonzero_num[count], options.binary); - } - - if (smat.count(R_coor)) - { - output_single_R(ofs, smat.at(R_coor), pv, options); - } - else - { - SparseRBlock empty_map; - output_single_R(ofs, empty_map, pv, options); - } - ++count; - } - if (!options.reduce || GlobalV::DRANK == 0) - { - ofs.close(); - } - - ModuleBase::timer::end("ModuleIO", "save_sparse"); -} - -template void ModuleIO::save_sparse( - const SparseRMatrix&, - const std::set&, - const Parallel_Orbitals&, - const SparseWriteOptions&); - -template void ModuleIO::save_sparse>( - const SparseRMatrix>&, - const std::set&, - const Parallel_Orbitals&, - const SparseWriteOptions&); diff --git a/source/source_io/module_hs/write_hs_sparse.h b/source/source_io/module_hs/write_hs_sparse.h deleted file mode 100644 index fe726245312..00000000000 --- a/source/source_io/module_hs/write_hs_sparse.h +++ /dev/null @@ -1,49 +0,0 @@ -#ifndef WRITE_HS_SPARSE_H -#define WRITE_HS_SPARSE_H - -#include "source_basis/module_ao/parallel_orbitals.h" -#include "source_lcao/lcao_hs_arrays.hpp" - -#include -#include -#include -#include - -namespace ModuleIO -{ -using RCoordinate = Abfs::Vector3_Order; - -template -using SparseRBlock = std::map>; - -template -using SparseRMatrix = std::map>; - -struct SparseWriteOptions -{ - std::string filename; - std::string label; - double threshold = 0.0; - bool binary = false; - int precision = 16; - int istep = -1; - bool reduce = true; - std::string temp_dir; -}; - -void save_dH_sparse(const int& istep, - const Parallel_Orbitals& pv, - LCAO_HS_Arrays& HS_Arrays, - const double& sparse_thr, - const bool& binary, - const std::string& fileflag = "h", - const int precision = 16); - -template -void save_sparse(const SparseRMatrix& smat, - const std::set& all_R_coor, - const Parallel_Orbitals& pv, - const SparseWriteOptions& options); -} // namespace ModuleIO - -#endif diff --git a/source/source_io/module_hs/write_vxc.hpp b/source/source_io/module_hs/write_vxc.hpp deleted file mode 100644 index 953911e3499..00000000000 --- a/source/source_io/module_hs/write_vxc.hpp +++ /dev/null @@ -1,301 +0,0 @@ -#ifndef __WRITE_VXC_H_ -#define __WRITE_VXC_H_ -#include "source_io/module_parameter/parameter.h" -#include "source_base/parallel_reduce.h" -#include "source_base/module_container/base/third_party/blas.h" -#include "source_base/module_external/scalapack_connector.h" -#include "source_lcao/module_dftu/dftu_nao_op_legacy.h" -#include "source_lcao/module_operator_lcao/veff_lcao.h" -#include "source_hamilt/module_xc/exx_info.h" -#ifdef __EXX -#include "source_lcao/module_operator_lcao/op_exx_lcao.h" -#endif -#include "source_psi/psi.h" -#include "source_io/module_hs/write_hs.h" -#include "source_base/module_out/filename.h" // use filename_output function - -namespace ModuleIO -{ - -inline void set_para2d_MO(const Parallel_Orbitals& pv, const int nbands, Parallel_2D& p2d) -{ - std::ofstream ofs; -#ifdef __MPI - p2d.set(nbands, nbands, pv.nb, pv.blacs_ctxt); -#else - p2d.set_serial(nbands, nbands); -#endif -} - -template -inline std::vector cVc(T* V, - T* c, - const int nbasis, - const int nbands, - const Parallel_Orbitals& pv, - const Parallel_2D& p2d) -{ - std::vector Vc(pv.nloc_wfc, 0.0); - char transa = 'N'; - char transb = 'N'; - const T alpha = (T)1.0; - const T beta = (T)0.0; -#ifdef __MPI - const int i1 = 1; - ScalapackConnector::gemm(transa, transb, - nbasis, nbands, nbasis, - alpha, V, i1, i1, pv.desc, - c, i1, i1, pv.desc_wfc, - beta, Vc.data(), i1, i1, pv.desc_wfc); -#else - container::BlasConnector::gemm(transa, transb, nbasis, nbands, nbasis, alpha, V, nbasis, c, nbasis, beta, Vc.data(), nbasis); -#endif - std::vector cVc(p2d.nloc, 0.0); - transa = (std::is_same::value ? 'T' : 'C'); -#ifdef __MPI - ScalapackConnector::gemm(transa, transb, - nbands, nbands, nbasis, - alpha, c, i1, i1, pv.desc_wfc, - Vc.data(), i1, i1, pv.desc_wfc, - beta, cVc.data(), i1, i1, p2d.desc); -#else - container::BlasConnector::gemm(transa, transb, nbands, nbands, nbasis, alpha, c, nbasis, Vc.data(), nbasis, beta, cVc.data(), nbasis); -#endif - return cVc; -} - -inline double get_real(const std::complex& c) -{ - return c.real(); -} - -inline double get_real(const double& d) -{ - return d; -} - -template -double all_band_energy(const int ik, const std::vector& mat_mo, const Parallel_2D& p2d, const ModuleBase::matrix& wg) -{ - double e = 0.0; - for (int i = 0; i < p2d.get_row_size(); ++i) - { - for (int j = 0; j < p2d.get_col_size(); ++j) - { - if (p2d.local2global_row(i) == p2d.local2global_col(j)) - { - e += get_real(mat_mo[j * p2d.get_row_size() + i]) * wg(ik, p2d.local2global_row(i)); - } - } - } - Parallel_Reduce::reduce_all(e); - return e; -} - -template -std::vector orbital_energy(const int ik, const int nbands, const std::vector& mat_mo, const Parallel_2D& p2d) -{ -#ifdef __DEBUG - assert(nbands >= 0); -#endif - std::vector e(nbands, 0.0); - for (int i = 0; i < nbands; ++i) - { - if (p2d.in_this_processor(i, i)) - { - const int index = p2d.global2local_col(i) * p2d.get_row_size() + p2d.global2local_row(i); - e[i] = get_real(mat_mo[index]); - } - } - Parallel_Reduce::reduce_all(e.data(), nbands); - return e; -} - -inline void write_orb_energy(const K_Vectors& kv, - const int nspin0, const int nbands, - const std::vector>& e_orb, - const std::string& term, const std::string& label, const bool app = false) -{ - assert(e_orb.size() == kv.get_nks()); - const int nk = kv.get_nks() / nspin0; - std::ofstream ofs; - ofs.open(PARAM.globalv.global_out_dir + term + "_" + (label == "" ? "out.dat" : label + "_out.dat"), - app ? std::ios::app : std::ios::out); - ofs << nk << "\n" << nspin0 << "\n" << nbands << "\n"; - ofs << std::scientific << std::setprecision(16); - for (int ik = 0; ik < nk; ++ik) - { - for (int is = 0; is < nspin0; ++is) - { - for (auto e : e_orb[is * nk + ik]) - { // Hartree and eV - ofs << e / 2. << "\t" << e * ModuleBase::Ry_to_eV << "\n"; - } - } - } -} - -/// @brief write the Vxc matrix in KS orbital representation, usefull for GW calculation -/// including terms: local/semi-local XC, EXX, DFTU -template -void write_Vxc(const int nspin, - const int nbasis, - const int drank, - const Parallel_Orbitals* pv, - const psi::Psi& psi, - const UnitCell& ucell, - Structure_Factor& sf, - surchem& solvent, - const ModulePW::PW_Basis& rho_basis, - const ModulePW::PW_Basis& rhod_basis, - const ModuleBase::matrix& vloc, - const Charge& chg, - const K_Vectors& kv, - const std::vector& orb_cutoff, - const ModuleBase::matrix& wg, - Grid_Driver& gd, - bool cal_exx, - const Exx_Info& exx_info -#ifdef __EXX - , - std::vector>>>* Hexxd = nullptr, - std::vector>>>>* Hexxc = nullptr -#endif -) -{ - ModuleBase::TITLE("ModuleIO", "write_Vxc"); - int nbands = wg.nc; - // 1. real-space xc potential - // ModuleBase::matrix vr_xc(nspin, chg.nrxx); - double etxc = 0.0; - double vtxc = 0.0; - // elecstate::PotXC* potxc(&rho_basis, &etxc, vtxc, nullptr); - // potxc.cal_v_eff(&chg, &ucell, vr_xc); - elecstate::Potential* potxc - = new elecstate::Potential(&rhod_basis, &rho_basis, &ucell, &vloc, &sf, &solvent, &etxc, &vtxc); - std::vector compnents_list = {"xc"}; - potxc->pot_register(compnents_list); - potxc->update_from_charge(&chg, &ucell); - - // 2. allocate AO-matrix - // R (the number of hR: 1 for nspin=1, 4; 2 for nspin=2) - int nspin0 = (nspin == 2) ? 2 : 1; - std::vector> vxcs_R_ao(nspin0, hamilt::HContainer(ucell, pv)); - for (int is = 0; is < nspin0; ++is) { - vxcs_R_ao[is].set_zero(); - if (std::is_same::value) { vxcs_R_ao[is].fix_gamma(); } - } - // k (size for each k-point) - hamilt::HS_Matrix_K vxc_k_ao(pv, 1); // only hk is needed, sk is skipped - - // 3. allocate operators and contribute HR - // op (corresponding to hR) - - std::vector>*> vxcs_op_ao(nspin0); - for (int is = 0; is < nspin0; ++is) - { - vxcs_op_ao[is] = new hamilt::Veff>( - &vxc_k_ao, kv.kvec_d, potxc, &vxcs_R_ao[is], &ucell, orb_cutoff, &gd, nspin); - vxcs_op_ao[is]->set_current_spin(is); - vxcs_op_ao[is]->contributeHR(); - } - std::vector> e_orb_locxc; // orbital energy (local XC) - std::vector> e_orb_tot; // orbital energy (total) -#ifdef __EXX - hamilt::OperatorEXX> vexx_op_ao(&vxc_k_ao, - &vxcs_R_ao[0],ucell,/*for paraV*/ kv, Hexxd, Hexxc, &exx_info, hamilt::Add_Hexx_Type::k); - hamilt::HS_Matrix_K vexxonly_k_ao(pv, 1); // only hk is needed, sk is skipped - hamilt::OperatorEXX> vexxonly_op_ao(&vexxonly_k_ao, - &vxcs_R_ao[0],ucell,/*for paraV*/ kv, Hexxd, Hexxc, &exx_info, hamilt::Add_Hexx_Type::k); - std::vector> e_orb_exx; // orbital energy (EXX) -#endif - hamilt::OperatorDFTU> vdftu_op_ao(&vxc_k_ao, kv.kvec_d, nullptr, ucell, nullptr, kv.isk); - - // 4. calculate and write the MO-matrix Exc - Parallel_2D p2d; - set_para2d_MO(*pv, nbands, p2d); - - // ======test======= - // double total_energy = 0.0; - // double exx_energy = 0.0; - // ======test======= - for (int ik = 0; ik < kv.get_nks(); ++ik) - { - vxc_k_ao.set_zero_hk(); - int is = kv.isk[ik]; - dynamic_cast*>(vxcs_op_ao[is])->contributeHk(ik); - const std::vector& vlocxc_k_mo = cVc(vxc_k_ao.get_hk(), &psi(ik, 0, 0), nbasis, nbands, *pv, p2d); - -#ifdef __EXX - if (cal_exx) - { - e_orb_locxc.emplace_back(orbital_energy(ik, nbands, vlocxc_k_mo, p2d)); - ModuleBase::GlobalFunc::ZEROS(vexxonly_k_ao.get_hk(), pv->nloc); - vexx_op_ao.contributeHk(ik); - vexxonly_op_ao.contributeHk(ik); - std::vector vexx_k_mo = cVc(vexxonly_k_ao.get_hk(), &psi(ik, 0, 0), nbasis, nbands, *pv, p2d); - e_orb_exx.emplace_back(orbital_energy(ik, nbands, vexx_k_mo, p2d)); - } -#endif - if (PARAM.inp.dft_plus_u) - { - vdftu_op_ao.contributeHk(ik); - } - const std::vector& vxc_tot_k_mo = cVc(vxc_k_ao.get_hk(), &psi(ik, 0, 0), nbasis, nbands, *pv, p2d); - e_orb_tot.emplace_back(orbital_energy(ik, nbands, vxc_tot_k_mo, p2d)); - - // write - - // mohan add 2025-06-02 - const int istep = -1; - const int out_label = 1; // 1 means .txt while 2 means .dat - const bool out_app_flag = 0; - const bool gamma_only = PARAM.globalv.gamma_only_local; - - std::string vxc_file = ModuleIO::filename_output( - PARAM.globalv.global_out_dir, - "vxc","nao",ik,kv.ik2iktot,nspin,kv.get_nkstot(), - out_label,out_app_flag,gamma_only,istep); - - ModuleIO::save_mat(istep, - vxc_tot_k_mo.data(), - nbands, - false /*binary*/, - PARAM.inp.out_ndigits, - true /*triangle*/, - out_app_flag /*append*/, - vxc_file, - p2d, - drank); - // ======test======= - // total_energy += all_band_energy(ik, vxc_tot_k_mo, p2d, wg); - // ======test======= - } - // ======test======= - // total_energy -= 0.5 * exx_energy; - // std::cout << "total energy: " << total_energy << std::endl; - // std::cout << "etxc: " << etxc << std::endl; - // std::cout << "vtxc_cal: " << total_energy - 0.5 * exx_energy << std::endl; - // std::cout << "vtxc_ref: " << vtxc << std::endl; - // std::cout << "exx_energy: " << 0.5 * exx_energy << std::endl; - // ======test======= - delete potxc; - for (int is = 0; is < nspin0; ++is) - { - delete vxcs_op_ao[is]; - } - - if (GlobalV::MY_RANK == 0) - { - write_orb_energy(kv, nspin0, nbands, e_orb_tot, "vxc", ""); -#ifdef __EXX - if (cal_exx) - { - write_orb_energy(kv, nspin0, nbands, e_orb_locxc, "vxc", "local"); - write_orb_energy(kv, nspin0, nbands, e_orb_exx, "vxc", "exx"); - } -#endif - } -} -} // namespace ModuleIO -#endif diff --git a/source/source_io/module_hs/write_vxc_lip.hpp b/source/source_io/module_hs/write_vxc_lip.hpp deleted file mode 100644 index 546320f91ea..00000000000 --- a/source/source_io/module_hs/write_vxc_lip.hpp +++ /dev/null @@ -1,280 +0,0 @@ -#ifndef __WRITE_VXC_LIP_H_ -#define __WRITE_VXC_LIP_H_ -#include "source_io/module_parameter/parameter.h" -#include "source_base/parallel_reduce.h" -#include "source_base/module_container/base/third_party/blas.h" -#include "source_pw/module_pwdft/op_pw_veff.h" -#include "source_psi/psi.h" -#include "source_cell/unitcell.h" -#include "source_cell/klist.h" -#include "source_estate/module_pot/potential_new.h" -#include "source_io/module_hs/write_hs.h" -#include "source_base/module_out/filename.h" // use filename_output function -#include - -namespace ModuleIO -{ - template - using Real = typename GetTypeReal::type; - - template - inline FPTYPE get_real(const std::complex& c) - { - return c.real(); - } - template - inline FPTYPE get_real(const FPTYPE& d) - { - return d; - } - - template - std::vector cVc(T* const V, T* const c, int nbasis, int nbands) - { - std::vector Vc(nbasis * nbands, 0.0); - char transa = 'N'; - char transb = 'N'; - const T alpha(1.0, 0.0); - const T beta(0.0, 0.0); - container::BlasConnector::gemm(transa, transb, nbasis, nbands, nbasis, - alpha, V, nbasis, c, nbasis, beta, Vc.data(), nbasis); - - std::vector cVc(nbands * nbands, 0.0); - transa = ((std::is_same::value || std::is_same::value) ? 'T' : 'C'); - container::BlasConnector::gemm(transa, transb, nbands, nbands, nbasis, - alpha, c, nbasis, Vc.data(), nbasis, beta, cVc.data(), nbands); - return cVc; - } - - template - std::vector> psi_Hpsi(std::complex* const psi, - std::complex* const hpsi, const int nbasis, const int nbands) - { - using T = std::complex; - std::vector cVc(nbands * nbands, (T)0.0); - const T alpha(1.0, 0.0); - const T beta(0.0, 0.0); - container::BlasConnector::gemm('C', 'N', nbands, nbands, nbasis, alpha, - psi, nbasis, hpsi, nbasis, beta, cVc.data(), nbands); - return cVc; - } - - template - std::vector orbital_energy(const int ik, const int nbands, - const std::vector>& mat_mo) - { -#ifdef __DEBUG - assert(nbands >= 0); -#endif - std::vector e(nbands, 0.0); - for (int i = 0; i < nbands; ++i) - { - e[i] = get_real(mat_mo[i * nbands + i]); - } - return e; - } - - template - FPTYPE all_band_energy(const int ik, const int nbands, - const std::vector>& mat_mo, const ModuleBase::matrix& wg) - { - FPTYPE e = 0.0; - for (int i = 0; i < nbands; ++i) - { - e += get_real(mat_mo[i * nbands + i]) * (FPTYPE)wg(ik, i); - } - return e; - } - - template - FPTYPE all_band_energy(const int ik, const std::vector& orbital_energy, - const ModuleBase::matrix& wg) - { - FPTYPE e = 0.0; - for (int i = 0; i < orbital_energy.size(); ++i) - { - e += orbital_energy[i] * (FPTYPE)wg(ik, i); - } - return e; - } - - /// @brief write the Vxc matrix in KS orbital representation, usefull for GW calculation - /// including terms: local/semi-local XC and EXX - template - void write_Vxc(int nspin, - int naos, - int drank, - const psi::Psi>& psi_pw, - const UnitCell& ucell, - Structure_Factor& sf, - surchem& solvent, - const ModulePW::PW_Basis_K& wfc_basis, - const ModulePW::PW_Basis& rho_basis, - const ModulePW::PW_Basis& rhod_basis, - const ModuleBase::matrix& vloc, - const Charge& chg, - const K_Vectors& kv, - const ModuleBase::matrix& wg, - bool cal_exx, - double hybrid_alpha -#ifdef __EXX - , - const Exx_Lip>& exx_lip -#endif - ) - { - using T = std::complex; - ModuleBase::TITLE("ModuleIO", "write_Vxc_LIP"); - int nbands = wg.nc; - // 1. real-space xc potential - // ModuleBase::matrix vr_xc(nspin, chg.nrxx); - double etxc = 0.0; - double vtxc = 0.0; - // elecstate::PotXC* potxc(&rho_basis, &etxc, vtxc, nullptr); - // potxc.cal_v_eff(&chg, &ucell, vr_xc); - elecstate::Potential* potxc - = new elecstate::Potential(&rhod_basis, &rho_basis, &ucell, &vloc, &sf, &solvent, &etxc, &vtxc); - std::vector compnents_list = { "xc" }; - - potxc->pot_register(compnents_list); - potxc->update_from_charge(&chg, &ucell); - // const ModuleBase::matrix vr_localxc = potxc->get_veff_smooth(); - - // 2. allocate xc operator - psi::Psi hpsi_localxc(psi_pw.get_nk(), psi_pw.get_nbands(), psi_pw.get_nbasis(), kv.ngk, true); - hpsi_localxc.zero_out(); - // std::cout << "hpsi.nk=" << hpsi_localxc.get_nk() << std::endl; - // std::cout << "hpsi.nbands=" << hpsi_localxc.get_nbands() << std::endl; - // std::cout << "hpsi.nbasis=" << hpsi_localxc.get_nbasis() << std::endl; - hamilt::Veff>* vxcs_op_pw; - - - std::vector> e_orb_locxc; // orbital energy (local XC) - std::vector> e_orb_tot; // orbital energy (total) - std::vector> e_orb_exx; // orbital energy (EXX) - Parallel_2D p2d_serial; - p2d_serial.set_serial(nbands, nbands); - // ======test======= - std::vector exx_energy(kv.get_nks()); - // ======test=======s - for (int ik = 0; ik < kv.get_nks(); ++ik) - { - // 2.1 local xc - vxcs_op_pw = new hamilt::Veff>(kv.isk.data(), - potxc->get_veff_smooth_data(), potxc->get_veff_smooth().nr, potxc->get_veff_smooth().nc, &wfc_basis); - vxcs_op_pw->init(ik); // set k-point index - psi_pw.fix_k(ik); - hpsi_localxc.fix_k(ik); -#ifdef __DEBUG - assert(hpsi_localxc.get_current_nbas() == psi_pw.get_current_nbas()); - assert(hpsi_localxc.get_current_nbas() == hpsi_localxc.get_ngk(ik)); -#endif - /// wrap psi and act band-by-band (the same result as act all bands at once) - // for (int ib = 0;ib < psi_pw.get_nbands();++ib) - // { - // std::cout<<"ib="< psi_single_band(&psi_pw(ik, ib, 0), 1, 1, psi_pw.get_current_nbas()); - // psi::Psi hpsi_single_band(&hpsi_localxc(ik, ib, 0), 1, 1, hpsi_localxc.get_current_nbas()); - // vxcs_op_pw->act(1, psi_pw.get_current_nbas(), psi_pw.npol, psi_single_band.get_pointer(), hpsi_single_band.get_pointer(), psi_pw.get_ngk(ik)); - // } - vxcs_op_pw->act(psi_pw.get_nbands(), psi_pw.get_nbasis(), psi_pw.get_npol(), &psi_pw(ik, 0, 0), &hpsi_localxc(ik, 0, 0), psi_pw.get_ngk(ik)); - delete vxcs_op_pw; - std::vector vxc_local_k_mo = psi_Hpsi(&psi_pw(ik, 0, 0), &hpsi_localxc(ik, 0, 0), psi_pw.get_nbasis(), psi_pw.get_nbands()); - Parallel_Reduce::reduce_pool(vxc_local_k_mo.data(), nbands * nbands); - e_orb_locxc.emplace_back(orbital_energy(ik, nbands, vxc_local_k_mo)); - - // 2.2 exx - std::vector vxc_tot_k_mo(std::move(vxc_local_k_mo)); - std::vector vexx_k_ao(naos * naos); -#if((defined __LCAO)&&(defined __EXX) && !(defined __CUDA)&& !(defined __ROCM)) - if (cal_exx) - { - for (int n = 0; n < naos; ++n) - { - for (int m = 0; m < naos; ++m) - { - vexx_k_ao[n * naos + m] += (T)hybrid_alpha - * exx_lip.get_exx_matrix()[ik][m][n]; - } - } - std::vector vexx_k_mo = cVc(vexx_k_ao.data(), &(exx_lip.get_hvec()(ik, 0, 0)), naos, nbands); - Parallel_Reduce::reduce_pool(vexx_k_mo.data(), nbands * nbands); - e_orb_exx.emplace_back(orbital_energy(ik, nbands, vexx_k_mo)); - container::BlasConnector::axpy(nbands * nbands, 1.0, vexx_k_mo.data(), 1, vxc_tot_k_mo.data(), 1); - } -#endif - - /// add-up and write - // mohan add 2025-06-02 - const int istep = -1; - const int out_label = 1; // 1 means .txt while 2 means .dat - const bool out_app_flag = 0; - const bool gamma_only = PARAM.globalv.gamma_only_local; - - std::string vxc_file = ModuleIO::filename_output( - PARAM.globalv.global_out_dir, - "vxc","nao",ik,kv.ik2iktot,nspin,kv.get_nkstot(), - out_label,out_app_flag,gamma_only,istep); - - ModuleIO::save_mat(istep, vxc_tot_k_mo.data(), nbands, - false, PARAM.inp.out_ndigits, true, - out_app_flag, vxc_file, - p2d_serial, drank, false); - - e_orb_tot.emplace_back(orbital_energy(ik, nbands, vxc_tot_k_mo)); - } - //===== test total xc energy ======= - // std::cout << "xc energy =" << etxc << std::endl; - // std::cout << "vtxc=" << vtxc << std::endl; - // FPTYPE exc_by_orb = 0.0; - // for (int ik = 0;ik < e_orb_locxc.size();++ik) - // exc_by_orb += all_band_energy(ik, e_orb_locxc[ik], wg); - // std::cout << "xc all-bands energy by orbital =" << exc_by_orb << std::endl; - // /// calculate orbital energy by grid integration of vtxc*rho - // FPTYPE exc_by_rho = 0.0; - // for (int ir = 0;ir < potxc->get_veff_smooth().nc;++ir) - // exc_by_rho += potxc->get_veff_smooth()(0, ir) * chg.rho[0][ir]; - // Parallel_Reduce::reduce_all(exc_by_rho); - // exc_by_rho *= ((FPTYPE)ucell.omega * (FPTYPE)GlobalV::NPROC / (FPTYPE)potxc->get_veff_smooth().nc); - // std::cout << "xc all-bands energy by rho =" << exc_by_rho << std::endl; - //===== test total xc energy ======= - //===== test total exx energy ======= - //===== test total exx energy ======= - // write the orbital energy for xc and exx in LibRPA format - const int nspin0 = (nspin == 2) ? 2 : 1; - auto write_orb_energy = [&kv, &nspin0, &nbands](const std::vector>& e_orb, - const std::string& label, - const bool app = false) { - assert(e_orb.size() == kv.get_nks()); - const int nk = kv.get_nks() / nspin0; - std::ofstream ofs; - ofs.open(PARAM.globalv.global_out_dir + "vxc_" + (label == "" ? "out.dat" : label + "_out.dat"), - app ? std::ios::app : std::ios::out); - ofs << nk << "\n" << nspin0 << "\n" << nbands << "\n"; - ofs << std::scientific << std::setprecision(16); - for (int ik = 0; ik < nk; ++ik) - { - for (int is = 0; is < nspin0; ++is) - { - for (auto e : e_orb[is * nk + ik]) - { // Hartree and eV - ofs << e / 2. << "\t" << e * (FPTYPE)ModuleBase::Ry_to_eV << "\n"; - } - } - } - }; - - if (GlobalV::MY_RANK == 0) - { - write_orb_energy(e_orb_tot, ""); -#if((defined __LCAO)&&(defined __EXX) && !(defined __CUDA)&& !(defined __ROCM)) - if (cal_exx) - { - write_orb_energy(e_orb_locxc, "local"); - write_orb_energy(e_orb_exx, "exx"); - } -#endif - } - } -} // namespace ModuleIO -#endif diff --git a/source/source_io/module_hs/write_vxc_r.hpp b/source/source_io/module_hs/write_vxc_r.hpp deleted file mode 100644 index 3f132b16de5..00000000000 --- a/source/source_io/module_hs/write_vxc_r.hpp +++ /dev/null @@ -1,160 +0,0 @@ -#ifndef __WRITE_VXC_R_H_ -#define __WRITE_VXC_R_H_ -#include "source_io/module_parameter/parameter.h" -#include "source_io/module_hs/write_hs_sparse.h" -#include "source_lcao/module_dftu/dftu_nao_op_legacy.h" -#include "source_lcao/module_operator_lcao/veff_lcao.h" -#include "source_lcao/spar_hsr.h" -#ifdef __EXX -#include "source_lcao/module_operator_lcao/op_exx_lcao.h" -#include "source_lcao/module_ri/ri_2d_comm.h" -#endif - -namespace ModuleIO -{ -template -std::map, std::map>> cal_HR_sparse(const hamilt::HContainer& hR, - const double sparse_thr) -{ - std::map, std::map>> target; - sparse_format::cal_HContainer(*hR.get_paraV(), sparse_thr, hR, target); - return target; -} - -/// @brief write the Vxc matrix in KS orbital representation, usefull for GW calculation -/// including terms: local/semi-local XC, EXX, DFTU -template -void write_Vxc_R(const int nspin, - const Parallel_Orbitals* pv, - const UnitCell& ucell, - Structure_Factor& sf, - surchem& solvent, - const ModulePW::PW_Basis& rho_basis, - const ModulePW::PW_Basis& rhod_basis, - const ModuleBase::matrix& vloc, - const Charge& chg, - const K_Vectors& kv, - const std::vector& orb_cutoff, - Grid_Driver& gd, - bool cal_exx, - double hybrid_alpha, - bool real_number -#ifdef __EXX - , - const std::vector>>>* const Hexxd, - const std::vector>>>>* const Hexxc -#endif - , - const double sparse_thr = 1e-10) -{ - ModuleBase::TITLE("ModuleIO", "write_Vxc_R"); - // 1. real-space xc potential - // ModuleBase::matrix vr_xc(nspin, chg.nrxx); - double etxc = 0.0; - double vtxc = 0.0; - // elecstate::PotXC* potxc(&rho_basis, &etxc, vtxc, nullptr); - // potxc.cal_v_eff(&chg, &ucell, vr_xc); - elecstate::Potential potxc(&rhod_basis, &rho_basis, &ucell, &vloc, &sf, &solvent, &etxc, &vtxc); - std::vector compnents_list = {"xc"}; - potxc.pot_register(compnents_list); - potxc.update_from_charge(&chg, &ucell); - - // 2. allocate H(R) - // (the number of hR: 1 for nspin=1, 4; 2 for nspin=2) - int nspin0 = (nspin == 2) ? 2 : 1; - std::vector> vxcs_R_ao(nspin0, hamilt::HContainer(ucell, pv)); // call move constructor -#ifdef __EXX - std::array Rs_period = {kv.nmp[0], kv.nmp[1], kv.nmp[2]}; - const auto cell_nearest = hamilt::init_cell_nearest(ucell, Rs_period); -#endif - for (int is = 0; is < nspin0; ++is) - { - if (std::is_same::value) - { - vxcs_R_ao[is].fix_gamma(); - } -#ifdef __EXX - if (cal_exx) - { - real_number - ? hamilt::reallocate_hcontainer(*Hexxd, &vxcs_R_ao[is], &cell_nearest) - : hamilt::reallocate_hcontainer(*Hexxc, &vxcs_R_ao[is], &cell_nearest); - } -#endif - } - - // 3. calculate the Vxc(R) - hamilt::HS_Matrix_K vxc_k_ao(pv, 1); // only hk is needed, sk is skipped - for (int is = 0; is < nspin0; ++is) - { - hamilt::Veff> vxcs_op_ao(&vxc_k_ao, - kv.kvec_d, - &potxc, - &vxcs_R_ao[is], - &ucell, - orb_cutoff, - &gd, - nspin); - vxcs_op_ao.set_current_spin(is); - vxcs_op_ao.contributeHR(); -#ifdef __EXX - if (cal_exx) - { - real_number ? RI_2D_Comm::add_HexxR(is, - hybrid_alpha, - *Hexxd, - *pv, - ucell.get_npol(), - vxcs_R_ao[is], - &cell_nearest) - : RI_2D_Comm::add_HexxR(is, - hybrid_alpha, - *Hexxc, - *pv, - ucell.get_npol(), - vxcs_R_ao[is], - &cell_nearest); - } -#endif - } - - // test: fold Vxc(R) and check whether it is equal to Vxc(k) - // for (int ik = 0; ik < kv.get_nks(); ++ik) - // { - // vxc_k_ao.set_zero_hk(); - // dynamic_cast*>(vxcs_op_ao[kv.isk[ik]])->contributeHk(ik); - - // // output Vxc(k) (test) - // const TK* const hk = vxc_k_ao.get_hk(); - // std::cout << "ik=" << ik << ", Vxc(K): " << std::endl; - // for (int i = 0; i < pv->get_row_size(); i++) - // { - // for (int j = 0; j < pv->get_col_size(); j++) - // { - // std::cout << hk[j * pv->get_row_size() + i] << " "; - // } - // std::cout << std::endl; - // } - // } - - // 4. write Vxc(R) in csr format - for (int is = 0; is < nspin0; ++is) - { - std::set> all_R_coor = sparse_format::get_R_range(vxcs_R_ao[is]); - const std::string filename = "Vxc_R_spin" + std::to_string(is); - ModuleIO::SparseWriteOptions options; - options.filename = PARAM.globalv.global_out_dir + filename + ".csr"; - options.label = filename; - options.threshold = sparse_thr; - options.binary = false; - options.istep = -1; - options.reduce = true; - options.temp_dir = PARAM.globalv.global_out_dir; - ModuleIO::save_sparse(cal_HR_sparse(vxcs_R_ao[is], sparse_thr), - all_R_coor, - *pv, - options); - } -} -} // namespace ModuleIO -#endif diff --git a/source/source_io/module_json/CMakeLists.txt b/source/source_io/module_json/CMakeLists.txt index 9a94822f570..e60a646bb98 100644 --- a/source/source_io/module_json/CMakeLists.txt +++ b/source/source_io/module_json/CMakeLists.txt @@ -2,7 +2,6 @@ list(APPEND objects_json abacusjson.cpp general_info.cpp init_info.cpp - readin_info.cpp output_info.cpp ) diff --git a/source/source_io/module_json/abacusjson.cpp b/source/source_io/module_json/abacusjson.cpp index ea6a59992eb..e00049097d5 100644 --- a/source/source_io/module_json/abacusjson.cpp +++ b/source/source_io/module_json/abacusjson.cpp @@ -1,152 +1,35 @@ #include "abacusjson.h" +#ifdef __JSON +#include #include -#include -#include -#include -#include +#include + namespace Json { -#ifdef __RAPIDJSON -rapidjson::Document AbacusJson::doc; +jsonValue AbacusJson::doc = jsonValue::object(); -bool isNum(std::string str) -{ - std::stringstream sin; - sin<> d)) - return false; - - if (sin >> c) - return false; - return true; +jsonValue& AbacusJson::document() +{ + return doc; } - -void AbacusJson::add_nested_member(std::vector::iterator begin, - std::vector::iterator end, - rapidjson::Value& val, - rapidjson::Value& parent, - rapidjson::Document::AllocatorType& allocator, - bool IsArray - ) +void AbacusJson::write_to_json(const std::string& filename) { - if (begin != end) + const auto content = doc.dump(4); + std::ofstream file(filename); + if (!file) { - jsonKeyNode keyNode = *begin; - rapidjson::Value key((*begin).key.c_str(), allocator); - - - if (begin + 1 == end) - { - - if( keyNode.key.empty() && parent.IsArray()){ - int index = keyNode.i; - if(index>=0){ - parent[index] = val; - } - else { - int arr_size = parent.Size(); - parent[arr_size+index] = val; - } - } - // if key exists, then overwrite it - else if (parent.HasMember(key)) - { - if(parent[key].IsArray()){ - parent[key].PushBack(val, allocator); - }else{ - // if key is an object, then warn the user - if (parent[key].IsObject()) - { - std::cout << "Warning: write to json, key " << (*begin).key - << " exist and is an object, and abacus will overwrite it with a value." << std::endl; - } - parent[key] = val; - } - } - else{ - if(IsArray==true){ - rapidjson::Value arr(rapidjson::kArrayType); - arr.PushBack(val, allocator); - parent.AddMember(key, arr, allocator); - } else{ - parent.AddMember(key, val, allocator); - } - - } - } - else - { - if( keyNode.key.empty()&&parent.IsArray()){ - int index = keyNode.i; - - if(index>=0){ - add_nested_member(begin + 1, end, val, parent[index], allocator,IsArray); - } - else { - int arr_size = parent.Size(); - add_nested_member(begin + 1, end, val, parent[arr_size+index], allocator,IsArray); - } - } - // need to check if the key exists - else if (parent.HasMember(key)) - { - // this key should be an object - if (!parent[key].IsObject()&&!parent[key].IsArray()) - { - std::cout << "Warning: write to json, key " << (*begin).key - << " exist and is not an object or array, and abacus will add it as a middle node." << std::endl; - } - add_nested_member(begin + 1, end, val, parent[key], allocator,IsArray); - } - else - { - rapidjson::Value paraent_val(rapidjson::kObjectType); - add_nested_member(begin + 1, end, val, paraent_val, allocator,IsArray); - parent.AddMember(key, paraent_val, allocator); - } - } + throw std::runtime_error("Cannot open JSON output file: " + filename); + } + file << content; + file.close(); + if (!file) + { + throw std::runtime_error("Cannot write JSON output file: " + filename); } } -// Output the json to a file -void AbacusJson::write_to_json(std::string filename) -{ - rapidjson::StringBuffer buffer; - rapidjson::PrettyWriter writer(buffer); - doc.Accept(writer); - - std::ofstream ofs(filename); - ofs << buffer.GetString(); - ofs.close(); -}; - template <> - void AbacusJson::add_json(std::vector keys, const std::string& value,bool IsArray) - { - if (!doc.IsObject()) - { - doc.SetObject(); - } - rapidjson::Value val(value.c_str(), doc.GetAllocator()); - add_nested_member(keys.begin(), keys.end(), val, doc, doc.GetAllocator(),IsArray); - } - - -// Overloaded template functions for json class objects - template <> - void AbacusJson::add_json(std::vector keys, const rapidjson::Value& value,bool IsArray) - { - - if (!doc.IsObject()) - { - doc.SetObject(); - } - rapidjson::Value val(value,doc.GetAllocator()); - add_nested_member(keys.begin(), keys.end(), val, doc, doc.GetAllocator(),IsArray); - } -#endif } // namespace Json +#endif // __JSON diff --git a/source/source_io/module_json/abacusjson.h b/source/source_io/module_json/abacusjson.h index 7bc003bbc92..66c5e547203 100644 --- a/source/source_io/module_json/abacusjson.h +++ b/source/source_io/module_json/abacusjson.h @@ -1,132 +1,29 @@ #ifndef ABACUS_JSON_H #define ABACUS_JSON_H -#include -#include #include -#include -#include "json_node.h" - -#ifdef __RAPIDJSON -#include "rapidjson/document.h" -#include "rapidjson/prettywriter.h" -#include "rapidjson/stringbuffer.h" - -/** -* @brief Define of AbacusJson:These macro definitions simplify the complex parameters -* required to add objects in rapidjson, making it easy to use. -* @usage: 1. create the json value by Type: eg. -* Json::jsonValue object(JobjectType); -* Json::jsonValue array(JarrayType); -* 2 - object. add the parameter by using correct function. If the constructed object contains -* type std::string, select the correct macro definition function in the key -* or value position based on std::string. -* eg. key is std::string, using: object.JaddStringK(str,val) -* eg. val is std::string, using: object.JaddStringV (str,val) -* eg. both key,val is std::string, using: object.JaddStringKV(str,val) -* eg. none of key,val is std::string, using: object.JaddNormal(str,val) -* -* 2 - array. using: array.JPushBack(val) -* or : JPushBackString(val) -*/ - -#define JobjectType rapidjson::kObjectType -#define JarrayType rapidjson::kArrayType -#define Get_Jallocator Json::AbacusJson::allocator() -#define Set_JString(str) Json::jsonValue().SetString(str.c_str(),str.length(),Get_Jallocator) - - -#define JaddStringV(str,val) AddMember(str, Set_JString(val), Get_Jallocator) -#define JaddStringK(str,val) AddMember(Set_JString(str), val, Get_Jallocator) -#define JaddStringKV(str,val) AddMember(Set_JString(str), Set_JString(val), Get_Jallocator) -#define JaddNormal(str,val) AddMember(str, val, Get_Jallocator) -#define JPushBack(val) PushBack(val, Get_Jallocator) -#define JPushBackString(val) PushBack(Set_JString(val), Get_Jallocator) -/** -* @brief The json processing module in abacus contains basic meta-operations such as -* adding, modifying, and checking json. -*/ +#ifdef __JSON +// Keep the implementation-heavy json.hpp out of this header. +#include namespace Json { -using jsonValue = rapidjson::Value; -// This class is used to construct a json value, and output some key values to a json file. +using jsonValue = nlohmann::ordered_json; + class AbacusJson { public: - - - - // Output the json to a file - static void write_to_json(std::string filename); - - static rapidjson::Document::AllocatorType& allocator(){ - return doc.GetAllocator(); - } - - /** - * @brief: The template specialization method adds value to the doc tree - * - * @param: 'keys' is a vector string, represents the path to be added to the json tree. - * - * 'value' is the value that needs to be added to the json tree, which type can be - * Json::jsonValue or other common value type(such as int, double ,bool ,std::string). - * - * 'IsArray' is a bool value, means the whether the root node to which 'value' is added is an array. - * - * @usage: 1. Add/Modify a double val to object json node (key2 is a object node): - * Json::AbacusJson::add_json({"key1","key2"}, 3.1415,false); - * - * 2. Pushback a double val to array json node (key2 is a array node): - * Json::AbacusJson::add_json({"key1","key2"}, 3.1415,true); - * - * 3. Modify a doble val to array json node (key2 is a array node), when use this method, - * The index number of the array starts at 0, if it's negative, it's going from back to front. - * eg. If the index is -1, it means that the last element of the array is modified: - * If we have a json array: {"key":[1,2,3]} - * i). Json::AbacusJson::add_json({"key",0}, 4,true); => {"key":[4,2,3]} - * ii). Json::AbacusJson::add_json({"key",-1}, 4,true); => {"key":[1,2,4]} - * iii). Json::AbacusJson::add_json({"key",1}, 4,true); => {"key":[1,4,3]} - * iv). Json::AbacusJson::add_json({"key",2}, 4,true); => {"key":[1,2,4]} - * iv). Json::AbacusJson::add_json({"key",3}, 4,true); => error!, The array element corresponding - * to the index has no value. - */ - template - static void add_json(std::vector keys, const T& value,bool IsArray) - { - if (!doc.IsObject()) - { - doc.SetObject(); - } - rapidjson::Value val(value); - add_nested_member(keys.begin(), keys.end(), val, doc, doc.GetAllocator(),IsArray); - } - - - + // Shared document for the schema generators in module_json; keep its root an object. + static jsonValue& document(); + static void write_to_json(const std::string& filename); private: - static rapidjson::Document doc; - - static void add_nested_member(std::vector::iterator begin, - std::vector::iterator end, - rapidjson::Value& val, - rapidjson::Value& parent, - rapidjson::Document::AllocatorType& allocator, - bool IsArray - ); - }; - -template <> -void AbacusJson::add_json(std::vector keys, const std::string& value,bool IsArray); - -template <> -void AbacusJson::add_json(std::vector keys, const rapidjson::Value& value,bool IsArray); - + static jsonValue doc; +}; } // namespace Json -#endif +#endif // __JSON -#endif \ No newline at end of file +#endif diff --git a/source/source_io/module_json/general_info.cpp b/source/source_io/module_json/general_info.cpp index 8b45c0f1921..60c0fcbe5cb 100644 --- a/source/source_io/module_json/general_info.cpp +++ b/source/source_io/module_json/general_info.cpp @@ -2,14 +2,18 @@ #include "para_json.h" #include "abacusjson.h" +#include "source_io/module_parameter/parameter.h" + +#ifdef __JSON +#include +#endif #include "source_base/parallel_global.h" #include "source_main/version.h" -// Add json objects to gener_info namespace Json { -#ifdef __RAPIDJSON +#ifdef __JSON void gen_general_info(const Parameter& param) { @@ -43,29 +47,18 @@ void gen_general_info(const Parameter& param) int omp_num = 1; #endif - AbacusJson::add_json({"general_info", "version"}, version, false); - AbacusJson::add_json({"general_info", "commit"}, commit, false); - AbacusJson::add_json({"general_info", "device"}, param.inp.device, false); - AbacusJson::add_json({"general_info", "mpi_num"}, mpi_num, false); - AbacusJson::add_json({"general_info", "omp_num"}, omp_num, false); - AbacusJson::add_json({"general_info", "pseudo_dir"}, param.inp.pseudo_dir, false); - AbacusJson::add_json({"general_info", "orbital_dir"}, param.inp.orbital_dir, false); - AbacusJson::add_json({"general_info", "stru_file"}, param.globalv.global_in_stru, false); - AbacusJson::add_json({"general_info", "kpt_file"}, param.inp.kpoint_file, false); - AbacusJson::add_json({"general_info", "start_time"}, start_time_str, false); - AbacusJson::add_json({"general_info", "end_time"}, end_time_str, false); - - // AbacusJson::add_Json(version,false,"general_info", "version"); - // AbacusJson::add_Json(commit,false,"general_info", "commit"); - // AbacusJson::add_Json(param.inp.device,false,"general_info", "device"); - // AbacusJson::add_Json(mpi_num,false,"general_info", "mpi_num"); - // AbacusJson::add_Json(omp_num,false,"general_info", "omp_num"); - // AbacusJson::add_Json(param.inp.pseudo_dir,false,"general_info", "pseudo_dir"); - // AbacusJson::add_Json(param.inp.orbital_dir,false,"general_info", "orbital_dir"); - // AbacusJson::add_Json(param.inp.stru_file,false,"general_info", "stru_file"); - // AbacusJson::add_Json(param.inp.kpoint_file,false,"general_info", "kpt_file"); - // AbacusJson::add_Json(start_time_str,false,"general_info", "start_time"); - // AbacusJson::add_Json(end_time_str,false,"general_info", "end_time"); + AbacusJson::document()["general_info"] = { + {"version", version}, + {"commit", commit}, + {"device", param.inp.device}, + {"mpi_num", mpi_num}, + {"omp_num", omp_num}, + {"pseudo_dir", param.inp.pseudo_dir}, + {"orbital_dir", param.inp.orbital_dir}, + {"stru_file", param.globalv.global_in_stru}, + {"kpt_file", param.inp.kpoint_file}, + {"start_time", start_time_str}, + {"end_time", end_time_str}}; } #endif } // namespace Json diff --git a/source/source_io/module_json/general_info.h b/source/source_io/module_json/general_info.h index 8cb6a1f6d02..d338281e068 100644 --- a/source/source_io/module_json/general_info.h +++ b/source/source_io/module_json/general_info.h @@ -1,14 +1,16 @@ #ifndef GENERAL_INFO_H #define GENERAL_INFO_H -#include "source_io/module_parameter/parameter.h" + +class Parameter; /** -* @brief In this part of the code to complete the general_info part of the json tree. -*/ + * @brief Generate the general_info section of the JSON tree. + */ namespace Json { -#ifdef __RAPIDJSON +#ifdef __JSON void gen_general_info(const Parameter& param); #endif -} -#endif \ No newline at end of file +} // namespace Json + +#endif diff --git a/source/source_io/module_json/init_info.cpp b/source/source_io/module_json/init_info.cpp index 721ffd2781b..0dc9dc580f8 100644 --- a/source/source_io/module_json/init_info.cpp +++ b/source/source_io/module_json/init_info.cpp @@ -1,173 +1,103 @@ #include "init_info.h" -#include "source_io/module_parameter/input_parameter.h" -#include "para_json.h" #include "abacusjson.h" +#include "source_cell/atom_spec.h" +#include "source_cell/unitcell.h" +#include "source_io/module_parameter/input_parameter.h" + +#ifdef __JSON +#include +#include -// Add json objects to init namespace Json { - -#ifdef __RAPIDJSON +namespace +{ +jsonValue& init_section() +{ + jsonValue& init = *AbacusJson::document().emplace("init", jsonValue::object()).first; + if (!init.is_object()) + { + throw std::invalid_argument("JSON init section must be an object"); + } + return init; +} +} // namespace void gen_init(UnitCell* ucell, const Input_para& inp) { - std::string pgname = ucell->symm.pgname; - std::string spgname = ucell->symm.spgname; - AbacusJson::add_json({"init", "point_group"}, pgname, false); - AbacusJson::add_json({"init", "point_group_in_space"}, spgname, false); - - // Json::AbacusJson::add_Json(pgname,false,"init", "point_group"); - // Json::AbacusJson::add_Json(spgname,false,"init","point_group_in_space"); + jsonValue info = {{"point_group", ucell->symm.pgname}, + {"point_group_in_space", ucell->symm.spgname}, + {"natom", ucell->nat}, + {"nband", inp.nbands}}; - int numAtoms = ucell->nat; - AbacusJson::add_json({"init", "natom"}, numAtoms, false); - AbacusJson::add_json({"init", "nband"}, inp.nbands, false); - - // Json::AbacusJson::add_Json(numAtoms,false,"init", "natom"); - // Json::AbacusJson::add_Json(PARAM.inp.nbands,false,"init", "nband"); - - int ntype = ucell->ntype, nelec_total = 0; - for (int it = 0; it < ntype; it++) + int nelec_total = 0; + for (int it = 0; it < ucell->ntype; ++it) { - std::string label = ucell->atoms[it].label; - int atom_number = ucell->atoms[it].na; - double number = ucell->atoms[it].ncpp.zv; - - nelec_total += ucell->atoms[it].ncpp.zv * ucell->atoms[it].na; - AbacusJson::add_json({"init", "natom_each_type", label}, atom_number, false); - AbacusJson::add_json({"init", "nelectron_each_type", label}, number, false); - - // Json::AbacusJson::add_Json(number,false,"init", "nelectron_each_type",label); + const Atom& atom = ucell->atoms[it]; + nelec_total += atom.ncpp.zv * atom.na; + info["natom_each_type"][atom.label] = atom.na; + info["nelectron_each_type"][atom.label] = atom.ncpp.zv; } - AbacusJson::add_json({"init", "nelectron"}, nelec_total, false); - - // Json::AbacusJson::add_Json(nelec_total,false,"init", "nelectron"); - - // energy cutoff for wavefunctions (Ry) - AbacusJson::add_json({"init", "ecutwfc"}, inp.ecutwfc, false); - AbacusJson::add_json({"init", "ecutwfc_unit"}, "Ry", false); - - // smearing method and sigma (Ry) - AbacusJson::add_json({"init", "smearing_method"}, inp.smearing_method, false); - AbacusJson::add_json({"init", "smearing_sigma"}, inp.smearing_sigma, false); - AbacusJson::add_json({"init", "smearing_sigma_unit"}, "Ry", false); - - // k-point mesh generation parameters - AbacusJson::add_json({"init", "kmesh_type"}, inp.kmesh_type, false); - Json::jsonValue kspacing_array(JarrayType); - kspacing_array.JPushBack(inp.kspacing[0]); - kspacing_array.JPushBack(inp.kspacing[1]); - kspacing_array.JPushBack(inp.kspacing[2]); - AbacusJson::add_json({"init", "kspacing"}, kspacing_array, false); - Json::jsonValue koffset_array(JarrayType); - koffset_array.JPushBack(inp.koffset[0]); - koffset_array.JPushBack(inp.koffset[1]); - koffset_array.JPushBack(inp.koffset[2]); - AbacusJson::add_json({"init", "koffset"}, koffset_array, false); + info["nelectron"] = nelec_total; + info["ecutwfc"] = inp.ecutwfc; + info["ecutwfc_unit"] = "Ry"; + info["smearing_method"] = inp.smearing_method; + info["smearing_sigma"] = inp.smearing_sigma; + info["smearing_sigma_unit"] = "Ry"; + info["kmesh_type"] = inp.kmesh_type; + info["kspacing"] = jsonValue::array({inp.kspacing[0], inp.kspacing[1], inp.kspacing[2]}); + info["koffset"] = jsonValue::array({inp.koffset[0], inp.koffset[1], inp.koffset[2]}); + // Shallow update: preserve other generators' fields, replace this generator's containers. + init_section().update(info); } void add_nkstot(int nkstot) { - Json::AbacusJson::add_json({"init", "nkstot"}, nkstot, false); - - // Json::AbacusJson::add_Json(nkstot,false,"init", "nkstot"); - // Json::AbacusJson::add_Json(nkstot_ibz,false,"init", "nkstot_ibz"); + init_section()["nkstot"] = nkstot; } void gen_stru(UnitCell* ucell, const Input_para& inp) { - AbacusJson::add_json({"comment"}, - "Unless otherwise specified, the unit of energy is eV and the unit of length is Angstrom", - false); + AbacusJson::document()["comment"] = + "Unless otherwise specified, the unit of energy is eV and the unit of length is Angstrom"; - int ntype = ucell->ntype; - - // array of pseudopotential file - std::string* pseudo_fn = ucell->pseudo_fn.data(); - - // array of orbital file - std::string* orbital_fn = ucell->orbital_fn.data(); - - // add atom element,orbital file and pseudopotential file - for (int i = 0; i < ntype; i++) + jsonValue info = jsonValue::object(); + for (int it = 0; it < ucell->ntype; ++it) { - std::string atom_label = ucell->atoms[i].label; - - std::string atom_element = ucell->atoms[i].ncpp.psd; - - Json::AbacusJson::add_json({"init", "element", atom_label}, atom_element, false); - - std::string orbital_str = inp.orbital_dir + orbital_fn[i]; - if (!orbital_str.compare("")) - { - Json::jsonValue nullValue; - nullValue.SetNull(); - Json::AbacusJson::add_json({"init", "orb", atom_label}, nullValue, false); - - // Json::AbacusJson::add_Json(nullValue,false,"init","orb",atom_label); - } - else - { - Json::AbacusJson::add_json({"init", "orb", atom_label}, orbital_str, false); - // Json::AbacusJson::add_Json(orbital_str,false,"init","orb",atom_label); - } - std::string pseudo_str = pseudo_fn[i]; - Json::AbacusJson::add_json({"init", "pp", atom_label}, pseudo_str, false); - - // Json::AbacusJson::add_Json(pseudo_str,false,"init","pp",atom_label); + const Atom& atom = ucell->atoms[it]; + info["element"][atom.label] = atom.ncpp.psd; + const std::string orbital = inp.orbital_dir + ucell->orbital_fn[it]; + info["orb"][atom.label] = orbital.empty() ? jsonValue(nullptr) : jsonValue(orbital); + info["pp"][atom.label] = ucell->pseudo_fn[it]; } - // atom coordinate, mag and label const double lat0_angstrom = ucell->lat0_angstrom; - for (int i = 0; i < ntype; i++) + for (int it = 0; it < ucell->ntype; ++it) { - ModuleBase::Vector3* tau = ucell->atoms[i].tau.data(); - int na = ucell->atoms[i].na; - for (int j = 0; j < na; j++) + const Atom& atom = ucell->atoms[it]; + for (int ia = 0; ia < atom.na; ++ia) { - Json::jsonValue coordinateArray(JarrayType); - coordinateArray.JPushBack(tau[j][0] * lat0_angstrom); - coordinateArray.JPushBack(tau[j][1] * lat0_angstrom); - coordinateArray.JPushBack(tau[j][2] * lat0_angstrom); - Json::AbacusJson::add_json({"init", "coordinate"}, coordinateArray, true); - // Json::AbacusJson::add_Json(coordinateArray,true,"init","coordinate"); - - Json::AbacusJson::add_json({"init", "mag"}, ucell->atoms[i].mag[j], true); - - // Json::AbacusJson::add_Json(ucell->atoms[i].mag[j],true,"init","mag"); - - std::string str = ucell->atoms[i].label; - Json::AbacusJson::add_json({"init", "label"}, str, true); - // Json::AbacusJson::add_Json(str,true,"init","label"); + const ModuleBase::Vector3& tau = atom.tau[ia]; + info["coordinate"].push_back(jsonValue::array({tau[0] * lat0_angstrom, + tau[1] * lat0_angstrom, + tau[2] * lat0_angstrom})); + info["mag"].push_back(atom.mag[ia]); + info["label"].push_back(atom.label); } } - - // cell - { - Json::jsonValue cellArray1(JarrayType); - Json::jsonValue cellArray2(JarrayType); - Json::jsonValue cellArray3(JarrayType); - cellArray1.JPushBack(ucell->latvec.e11 * lat0_angstrom); - cellArray1.JPushBack(ucell->latvec.e12 * lat0_angstrom); - cellArray1.JPushBack(ucell->latvec.e13 * lat0_angstrom); - cellArray2.JPushBack(ucell->latvec.e21 * lat0_angstrom); - cellArray2.JPushBack(ucell->latvec.e22 * lat0_angstrom); - cellArray2.JPushBack(ucell->latvec.e23 * lat0_angstrom); - cellArray3.JPushBack(ucell->latvec.e31 * lat0_angstrom); - cellArray3.JPushBack(ucell->latvec.e32 * lat0_angstrom); - cellArray3.JPushBack(ucell->latvec.e33 * lat0_angstrom); - Json::AbacusJson::add_json({"init", "cell"}, cellArray1, true); - Json::AbacusJson::add_json({"init", "cell"}, cellArray2, true); - Json::AbacusJson::add_json({"init", "cell"}, cellArray3, true); - - // Json::AbacusJson::add_Json(cellArray1,true,"init","cell"); - // Json::AbacusJson::add_Json(cellArray2,true,"init","cell"); - // Json::AbacusJson::add_Json(cellArray3,true,"init","cell"); - } - return; + info["cell"] = {{ucell->latvec.e11 * lat0_angstrom, + ucell->latvec.e12 * lat0_angstrom, + ucell->latvec.e13 * lat0_angstrom}, + {ucell->latvec.e21 * lat0_angstrom, + ucell->latvec.e22 * lat0_angstrom, + ucell->latvec.e23 * lat0_angstrom}, + {ucell->latvec.e31 * lat0_angstrom, + ucell->latvec.e32 * lat0_angstrom, + ucell->latvec.e33 * lat0_angstrom}}; + init_section().update(info); } -#endif } // namespace Json +#endif // __JSON diff --git a/source/source_io/module_json/init_info.h b/source/source_io/module_json/init_info.h index 5fa4196698f..f5b01fed92f 100644 --- a/source/source_io/module_json/init_info.h +++ b/source/source_io/module_json/init_info.h @@ -1,35 +1,21 @@ #ifndef INIT_INFO_H #define INIT_INFO_H -#include "source_cell/atom_spec.h" -#include "source_cell/module_symmetry/symmetry.h" -#include "source_cell/unitcell.h" +class UnitCell; struct Input_para; /** - * @brief In this part of the code to complete the init part of the json tree. + * @brief Generate the init section of the JSON tree. */ namespace Json { -#ifdef __RAPIDJSON -// void gen_init(ModuleSymmetry::Symmetry *symm,Atom *atoms); +#ifdef __JSON -/** - * @param ucell: ucell for reading json parameters. - * @param inp: input parameters for reading json parameters. - */ void gen_init(UnitCell* ucell, const Input_para& inp); - -/** - * @param nkstot,nkstot_ibz: two param in json tree - */ void add_nkstot(int nkstot); - -/** - * @param ucell: ucell for reading structure init in abacus. - * @param inp: input parameters for reading orbital directory. - */ void gen_stru(UnitCell* ucell, const Input_para& inp); -#endif + +#endif // __JSON } // namespace Json -#endif \ No newline at end of file + +#endif diff --git a/source/source_io/module_json/json_node.h b/source/source_io/module_json/json_node.h deleted file mode 100644 index f19bddcf787..00000000000 --- a/source/source_io/module_json/json_node.h +++ /dev/null @@ -1,21 +0,0 @@ -#ifndef JSON_NODE_H -#define JSON_NODE_H - -namespace Json -{ - - class jsonKeyNode{ - public: - jsonKeyNode(int i): i(i) {}; - jsonKeyNode(const std::string& s): key(s) {}; - - template - jsonKeyNode(const char (&s)[N]): key(s) {}; - - int i=0; - std::string key; - }; - -} - -#endif \ No newline at end of file diff --git a/source/source_io/module_json/output_info.cpp b/source/source_io/module_json/output_info.cpp index aa6d60aef1a..463adcee317 100644 --- a/source/source_io/module_json/output_info.cpp +++ b/source/source_io/module_json/output_info.cpp @@ -1,165 +1,159 @@ #include "output_info.h" -#include "para_json.h" -#include "source_io/module_parameter/parameter.h" + #include "abacusjson.h" +#include "source_base/matrix.h" +#include "source_cell/unitcell.h" + +#ifdef __JSON +#include +#include +#endif +#include +#include -//Add json objects to init namespace Json { -#ifdef __RAPIDJSON - - - // Adjust the position of the json object and set the initial value - void init_output_array_obj(){ - +#ifdef __JSON - - jsonValue scf_obj(JobjectType); - - Json::jsonValue nullValue; - nullValue.SetNull(); - scf_obj.JaddNormal("e_fermi",nullValue); - scf_obj.JaddNormal("energy",nullValue); - scf_obj.JaddNormal("scf_converge",nullValue); - - jsonValue force(JobjectType); - jsonValue stress(JobjectType); - jsonValue coordinate(JarrayType); - jsonValue mag(JarrayType); - jsonValue cell(JarrayType); - - scf_obj.JaddNormal("force",nullValue); - scf_obj.JaddNormal("stress",nullValue); - scf_obj.JaddNormal("coordinate",coordinate); - scf_obj.JaddNormal("mag",mag); - scf_obj.JaddNormal("cell",cell); - - - AbacusJson::add_json({"output"},scf_obj,true); +namespace +{ +jsonValue& current_output() +{ + jsonValue& root = AbacusJson::document(); + const jsonValue::iterator output = root.find("output"); + if (output == root.end() || !output->is_array()) + { + throw std::invalid_argument("JSON output records must be initialized as an array"); + } + if (output->empty()) + { + throw std::out_of_range("JSON output record is not initialized"); + } + jsonValue& record = output->back(); + if (!record.is_object()) + { + throw std::invalid_argument("JSON output record must be an object"); } + return record; +} +} // namespace - void add_output_cell_coo_stress_force( - const UnitCell *ucell, - const ModuleBase::matrix force, const double fac, - const ModuleBase::matrix stress, const double unit_transform - ) { +void init_output_array_obj() +{ + jsonValue& output = *AbacusJson::document().emplace("output", jsonValue::array()).first; + if (!output.is_array()) + { + throw std::invalid_argument("JSON output must be an array"); + } + output.push_back({{"e_fermi", nullptr}, + {"energy", nullptr}, + {"scf_converge", nullptr}, + {"force", nullptr}, + {"stress", nullptr}, + {"coordinate", jsonValue::array()}, + {"mag", jsonValue::array()}, + {"cell", jsonValue::array()}}); +} + +void add_output_cell_coo_stress_force(const UnitCell& ucell, + const ModuleBase::matrix& force, + const double fac, + const ModuleBase::matrix& stress, + const double unit_transform, + const bool cal_force, + const bool cal_stress) +{ + jsonValue& output = current_output(); + const double output_acc = 1.0e-8; + if (cal_force) + { + jsonValue force_array = jsonValue::array(); int iat = 0; - const double output_acc = 1.0e-8; - - if (PARAM.inp.cal_force){ - //add force - Json::jsonValue force_array(JarrayType); - for (int it = 0; it < ucell->ntype; it++) + for (int it = 0; it < ucell.ntype; ++it) + { + for (int ia = 0; ia < ucell.atoms[it].na; ++ia) { - for (int ia = 0; ia < ucell->atoms[it].na; ia++) - { - Json::jsonValue force_subarray(JarrayType); - double fx = std::abs(force(iat, 0)) > output_acc ? force(iat, 0) * fac : 0.0; - double fy = std::abs(force(iat, 1)) > output_acc ? force(iat, 1) * fac : 0.0; - double fz = std::abs(force(iat, 2)) > output_acc ? force(iat, 2) * fac : 0.0; - - force_subarray.JPushBack(fx); - force_subarray.JPushBack(fy); - force_subarray.JPushBack(fz); - force_array.JPushBack(force_subarray); - iat++; - } + const double fx = std::abs(force(iat, 0)) > output_acc ? force(iat, 0) * fac : 0.0; + const double fy = std::abs(force(iat, 1)) > output_acc ? force(iat, 1) * fac : 0.0; + const double fz = std::abs(force(iat, 2)) > output_acc ? force(iat, 2) * fac : 0.0; + force_array.push_back(jsonValue::array({fx, fy, fz})); + ++iat; } - Json::AbacusJson::add_json({"output",-1,"force"}, force_array,false); - - // AbacusJson::add_Json(force_array,false,"output",-1,"force"); - } - - if (PARAM.inp.cal_stress){ - //add stress - Json::jsonValue stress_array(JarrayType); - for (int i = 0; i < 3; i++) - { - Json::jsonValue stress_subarray(JarrayType); - double sx = stress(i, 0) * unit_transform; - double sy = stress(i, 1) * unit_transform; - double sz = stress(i, 2) * unit_transform; - stress_subarray.JPushBack(sx); - stress_subarray.JPushBack(sy); - stress_subarray.JPushBack(sz); - stress_array.JPushBack(stress_subarray); - } - Json::AbacusJson::add_json({"output",-1,"stress"}, stress_array,false); - - // AbacusJson::add_Json(stress_array,false,"output",-1,"stress"); } - //add coordinate - int ntype = ucell->ntype; - const double lat0_angstrom = ucell->lat0_angstrom; - for(int i=0;i* tau = ucell->atoms[i].tau.data(); - int na = ucell->atoms[i].na; - for(int j=0;jatoms[i].mag[j],true); - } - } + output["force"] = std::move(force_array); + } - //add cell + if (cal_stress) + { + jsonValue stress_array = jsonValue::array(); + for (int i = 0; i < 3; ++i) { - Json::jsonValue cellArray1(JarrayType); - Json::jsonValue cellArray2(JarrayType); - Json::jsonValue cellArray3(JarrayType); - cellArray1.JPushBack(ucell->latvec.e11 * lat0_angstrom); - cellArray1.JPushBack(ucell->latvec.e12 * lat0_angstrom); - cellArray1.JPushBack(ucell->latvec.e13 * lat0_angstrom); - cellArray2.JPushBack(ucell->latvec.e21 * lat0_angstrom); - cellArray2.JPushBack(ucell->latvec.e22 * lat0_angstrom); - cellArray2.JPushBack(ucell->latvec.e23 * lat0_angstrom); - cellArray3.JPushBack(ucell->latvec.e31 * lat0_angstrom); - cellArray3.JPushBack(ucell->latvec.e32 * lat0_angstrom); - cellArray3.JPushBack(ucell->latvec.e33 * lat0_angstrom); - Json::AbacusJson::add_json({"output",-1,"cell"}, cellArray1,true); - Json::AbacusJson::add_json({"output",-1,"cell"}, cellArray2,true); - Json::AbacusJson::add_json({"output",-1,"cell"}, cellArray3,true); - - // Json::AbacusJson::add_Json(cellArray1,true,"output",-1,"cell"); - // Json::AbacusJson::add_Json(cellArray2,true,"output",-1,"cell"); - // Json::AbacusJson::add_Json(cellArray2,true,"output",-1,"cell"); + stress_array.push_back(jsonValue::array({stress(i, 0) * unit_transform, + stress(i, 1) * unit_transform, + stress(i, 2) * unit_transform})); } - + output["stress"] = std::move(stress_array); } - void add_output_efermi_converge(const double efermi, const bool scf_converge ){ - Json::AbacusJson::add_json({"output",-1,"e_fermi"}, efermi,false); - Json::AbacusJson::add_json({"output",-1,"scf_converge"}, scf_converge,false); - } - - void add_output_energy(const double energy) + const double lat0_angstrom = ucell.lat0_angstrom; + jsonValue coordinates = jsonValue::array(); + jsonValue mag = jsonValue::array(); + for (int it = 0; it < ucell.ntype; ++it) { - Json::AbacusJson::add_json({"output",-1,"energy"}, energy,false); + for (int ia = 0; ia < ucell.atoms[it].na; ++ia) + { + const ModuleBase::Vector3& tau = ucell.atoms[it].tau[ia]; + coordinates.push_back(jsonValue::array({tau[0] * lat0_angstrom, + tau[1] * lat0_angstrom, + tau[2] * lat0_angstrom})); + mag.push_back(ucell.atoms[it].mag[ia]); + } } + output["coordinate"] = std::move(coordinates); + output["mag"] = std::move(mag); + output["cell"] = {{ucell.latvec.e11 * lat0_angstrom, + ucell.latvec.e12 * lat0_angstrom, + ucell.latvec.e13 * lat0_angstrom}, + {ucell.latvec.e21 * lat0_angstrom, + ucell.latvec.e22 * lat0_angstrom, + ucell.latvec.e23 * lat0_angstrom}, + {ucell.latvec.e31 * lat0_angstrom, + ucell.latvec.e32 * lat0_angstrom, + ucell.latvec.e33 * lat0_angstrom}}; +} + +void add_output_efermi_converge(const double efermi, const bool scf_converge) +{ + jsonValue& output = current_output(); + output["e_fermi"] = efermi; + output["scf_converge"] = scf_converge; +} - void add_output_scf_mag( - double total_mag, double absolute_mag, - double energy, double ediff, double drho,double time - ){ - Json::AbacusJson::add_json({"output",-1,"total_mag"}, total_mag,false); - Json::AbacusJson::add_json({"output",-1,"absolute_mag"}, absolute_mag,false); - - // Json::AbacusJson::add_Json(total_mag,false,"output",-1,"total_mag"); - // Json::AbacusJson::add_Json(absolute_mag,false,"output",-1,"absolute_mag"); - - Json::jsonValue scf_obj(JobjectType); - scf_obj.JaddNormal("energy",energy); - scf_obj.JaddNormal("ediff",ediff); - scf_obj.JaddNormal("drho",drho); - scf_obj.JaddNormal("time",time); - Json::AbacusJson::add_json({"output",-1,"scf"}, scf_obj,true); - - // Json::AbacusJson::add_Json(scf_obj,true,"output",-1,"scf"); +void add_output_energy(const double energy) +{ + current_output()["energy"] = energy; +} + +void add_output_scf_mag(const double total_mag, + const double absolute_mag, + const double energy, + const double ediff, + const double drho, + const double time) +{ + jsonValue& output = current_output(); + output["total_mag"] = total_mag; + output["absolute_mag"] = absolute_mag; + // Acquire the history only after inserting other fields: ordered_json may reallocate them. + jsonValue& scf = *output.emplace("scf", jsonValue::array()).first; + if (!scf.is_array()) + { + throw std::invalid_argument("JSON SCF history must be an array"); } + scf.push_back({{"energy", energy}, {"ediff", ediff}, {"drho", drho}, {"time", time}}); +} -#endif +#endif // __JSON } // namespace Json diff --git a/source/source_io/module_json/output_info.h b/source/source_io/module_json/output_info.h index f61c60c0b99..56f68fb0d9b 100644 --- a/source/source_io/module_json/output_info.h +++ b/source/source_io/module_json/output_info.h @@ -1,34 +1,41 @@ #ifndef OUTPUT_INFO_H #define OUTPUT_INFO_H -#include "source_cell/unitcell.h" -#include "source_base/matrix.h" +class UnitCell; +namespace ModuleBase +{ +class matrix; +} /** -* @brief In this part of the code to complete the output part of the json tree. -* @param ucell: ucell for reading json parameters -*/ + * @brief Generate the output section of the JSON tree. + */ namespace Json { -#ifdef __RAPIDJSON +#ifdef __JSON - void init_output_array_obj(); +void init_output_array_obj(); - void add_output_cell_coo_stress_force( - const UnitCell *ucell, - const ModuleBase::matrix force, const double fac, - const ModuleBase::matrix stress, const double unit_transform - ); +void add_output_cell_coo_stress_force(const UnitCell& ucell, + const ModuleBase::matrix& force, + double fac, + const ModuleBase::matrix& stress, + double unit_transform, + bool cal_force, + bool cal_stress); - void add_output_efermi_converge(const double efermi, const bool scf_converge ); - void add_output_energy(const double energy ); +void add_output_efermi_converge(double efermi, bool scf_converge); +void add_output_energy(double energy); - void add_output_scf_mag( - const double total_mag, const double absolute_mag, - const double energy, const double ediff, const double drho,const double time - ); +void add_output_scf_mag(double total_mag, + double absolute_mag, + double energy, + double ediff, + double drho, + double time); + +#endif // __JSON +} // namespace Json #endif -} -#endif \ No newline at end of file diff --git a/source/source_io/module_json/para_json.cpp b/source/source_io/module_json/para_json.cpp index 7f49c75ce5b..75bfb8c43e9 100644 --- a/source/source_io/module_json/para_json.cpp +++ b/source/source_io/module_json/para_json.cpp @@ -4,18 +4,18 @@ #include #include #include -#ifdef __RAPIDJSON +#ifdef __JSON +#include "source_io/module_parameter/parameter.h" #include "abacusjson.h" #include "general_info.h" #include "init_info.h" -#include "readin_info.h" -#endif // __RAPIDJSON +#endif // __JSON namespace Json { // void create_Json(ModuleSymmetry::Symmetry *symm,Atom *atoms,Input *input){ -// #ifdef __RAPIDJSON +// #ifdef __JSON // gen_general_info(input); // gen_init(symm,atoms); // #endif @@ -23,19 +23,19 @@ namespace Json void json_output() { -#ifdef __RAPIDJSON +#ifdef __JSON #ifdef __MPI if (GlobalV::MY_RANK == 0) AbacusJson::write_to_json("abacus.json"); #else AbacusJson::write_to_json("abacus.json"); #endif -#endif // __RAPIDJSON +#endif // __JSON } void create_Json(UnitCell* ucell, const Parameter& param) { -#ifdef __RAPIDJSON +#ifdef __JSON gen_general_info(param); gen_init(ucell, param.inp); // gen_stru(ucell, param.inp); @@ -45,7 +45,7 @@ void create_Json(UnitCell* ucell, const Parameter& param) void gen_stru_wrapper(UnitCell* ucell, const Input_para& inp) { -#ifdef __RAPIDJSON +#ifdef __JSON #ifdef __MPI if (GlobalV::MY_RANK == 0) gen_stru(ucell, inp); diff --git a/source/source_io/module_json/para_json.h b/source/source_io/module_json/para_json.h index 990ad4c99b7..0bf239a4a00 100644 --- a/source/source_io/module_json/para_json.h +++ b/source/source_io/module_json/para_json.h @@ -1,23 +1,21 @@ +#ifndef PARA_JSON_H +#define PARA_JSON_H + #include #include -#include "source_cell/module_symmetry/symmetry.h" -#include "source_cell/atom_spec.h" -#include "source_io/module_parameter/parameter.h" -#include "source_cell/unitcell.h" +class Parameter; +class UnitCell; +struct Input_para; + namespace Json { -// void create_Json(ModuleSymmetry::Symmetry *symm,Atom *atoms,Input *input); - -void create_Json(UnitCell *ucell, const Parameter& input); - -// Output the json to abacus.json file +void create_Json(UnitCell* ucell, const Parameter& param); void json_output(); - -// Convert time_t to string void convert_time(std::time_t time_now, std::string& time_str); +void gen_stru_wrapper(UnitCell* ucell, const Input_para& inp); -// generate struture wrapper function -void gen_stru_wrapper(UnitCell *ucell, const Input_para& inp); } // namespace Json + +#endif diff --git a/source/source_io/module_json/readin_info.cpp b/source/source_io/module_json/readin_info.cpp deleted file mode 100644 index a150967af80..00000000000 --- a/source/source_io/module_json/readin_info.cpp +++ /dev/null @@ -1,11 +0,0 @@ -#include "readin_info.h" -#include "para_json.h" -#include "abacusjson.h" - -//Add json objects to init -namespace Json -{ - -#ifdef __RAPIDJSON -#endif -} // namespace Json diff --git a/source/source_io/module_json/readin_info.h b/source/source_io/module_json/readin_info.h deleted file mode 100644 index c4a598c999d..00000000000 --- a/source/source_io/module_json/readin_info.h +++ /dev/null @@ -1,18 +0,0 @@ -#ifndef READIN_INFO_H -#define READIN_INFO_H - - - -/** -* @brief In this part of the code to complete the readin part of the json tree. -* @param ucell: ucell for reading json parameters -*/ -namespace Json -{ -#ifdef __RAPIDJSON - - - -#endif -} -#endif \ No newline at end of file diff --git a/source/source_io/module_json/test/para_json_test.cpp b/source/source_io/module_json/test/para_json_test.cpp index 6ab32a2c866..3ddee9e4d19 100644 --- a/source/source_io/module_json/test/para_json_test.cpp +++ b/source/source_io/module_json/test/para_json_test.cpp @@ -1,255 +1,111 @@ #include "gtest/gtest.h" -#define private public +#include + +#include +#include +#include +#include +#include +#include + +#include "source_base/constants.h" +#include "source_base/matrix.h" +#include "source_base/matrix3.h" +#include "source_base/parallel_global.h" +#include "source_base/vector3.h" +#include "source_cell/atom_spec.h" +#include "source_cell/magnetism.h" +#include "source_cell/unitcell.h" #include "source_io/module_json/abacusjson.h" #include "source_io/module_json/general_info.h" #include "source_io/module_json/init_info.h" -#include "source_io/module_json/readin_info.h" +#include "source_io/module_json/output_info.h" #include "source_io/module_parameter/parameter.h" -#include "source_io/module_json/para_json.h" -#include "source_base/constants.h" #include "source_main/version.h" -#undef private -/************************************************ - * unit test of json output module - ************************************************ -/** - * - Tested Functions: - * - AddJson() - * - Verify the normal addition of json structure parameters in the json function. - * - OutputJson() - * - Verify the correctness of the json output. - * - GeneralInfo() - * - Test the correctness of the json output of the General Info module. - * - InitInfo() - * - Test the correctness of the json output of the Init info module. - */ - -TEST(AbacusJsonTest, AddJson) + +class AbacusJsonTest : public testing::Test { - Json::AbacusJson::doc.SetObject(); - - // add a string - Json::AbacusJson::add_json({"key1"}, "value1", false); - ASSERT_TRUE(Json::AbacusJson::doc.HasMember("key1")); - ASSERT_TRUE(Json::AbacusJson::doc["key1"].IsString()); - ASSERT_STREQ(Json::AbacusJson::doc["key1"].GetString(), "value1"); - - // add a string to a nested object - Json::AbacusJson::add_json({"key2", "key3"}, "value2", false); - ASSERT_TRUE(Json::AbacusJson::doc.HasMember("key2")); - ASSERT_TRUE(Json::AbacusJson::doc["key2"].IsObject()); - ASSERT_TRUE(Json::AbacusJson::doc["key2"].HasMember("key3")); - ASSERT_TRUE(Json::AbacusJson::doc["key2"]["key3"].IsString()); - ASSERT_STREQ(Json::AbacusJson::doc["key2"]["key3"].GetString(), "value2"); - - // add an int - Json::AbacusJson::add_json({"key2"}, 123, false); - ASSERT_TRUE(Json::AbacusJson::doc.HasMember("key2")); - ASSERT_TRUE(Json::AbacusJson::doc["key2"].IsInt()); - ASSERT_EQ(Json::AbacusJson::doc["key2"].GetInt(), 123); - - // add a bool - Json::AbacusJson::add_json({"key3"}, true, false); - ASSERT_TRUE(Json::AbacusJson::doc.HasMember("key3")); - ASSERT_TRUE(Json::AbacusJson::doc["key3"].IsBool()); - ASSERT_EQ(Json::AbacusJson::doc["key3"].GetBool(), true); - - // add a double - Json::AbacusJson::add_json({"key4"}, 1.23, false); - ASSERT_TRUE(Json::AbacusJson::doc.HasMember("key4")); - ASSERT_TRUE(Json::AbacusJson::doc["key4"].IsDouble()); - ASSERT_EQ(Json::AbacusJson::doc["key4"].GetDouble(), 1.23); - - // modify a value - Json::AbacusJson::add_json({"key4"}, 4.56, false); - ASSERT_EQ(Json::AbacusJson::doc["key4"].GetDouble(), 4.56); - - // array test - Json::AbacusJson::add_json({"key6", "key7"}, true, true); - Json::AbacusJson::add_json({"key6", "key7"}, false, true); - - // add key-val to a object array - for (int i = 0; i < 3; i++) + protected: + void SetUp() override { - Json::jsonValue object(JobjectType); - object.JaddNormal("int", i); - - std::string str = std::to_string(i * 100); - std::string str2 = "Kstring"; - - object.JaddStringV("string", str); - object.JaddStringK(str, "string"); - object.JaddStringKV(str2, str); - object.JaddNormal("double", 0.01 * i); - Json::AbacusJson::add_json({"array"}, object, true); + Json::AbacusJson::document() = Json::jsonValue::object(); } - Json::AbacusJson::add_json({"array", 1, "new_add_notLast"}, "correct1", false); - Json::AbacusJson::add_json({"array", -1, "new_add_Last"}, "correct2", false); - - // Validate json parameters in doc objects - - ASSERT_EQ(Json::AbacusJson::doc["array"][0]["int"].GetInt(), 0); - ASSERT_STREQ(Json::AbacusJson::doc["array"][0]["string"].GetString(), "0"); - ASSERT_STREQ(Json::AbacusJson::doc["array"][0]["0"].GetString(), "string"); - ASSERT_STREQ(Json::AbacusJson::doc["array"][0]["Kstring"].GetString(), "0"); - ASSERT_STREQ(Json::AbacusJson::doc["array"][1]["new_add_notLast"].GetString(), "correct1"); - - ASSERT_EQ(Json::AbacusJson::doc["array"][0]["double"].GetDouble(), 0.0); - - ASSERT_EQ(Json::AbacusJson::doc["array"][1]["int"].GetInt(), 1); - ASSERT_STREQ(Json::AbacusJson::doc["array"][1]["string"].GetString(), "100"); - - ASSERT_STREQ(Json::AbacusJson::doc["array"][1]["100"].GetString(), "string"); - ASSERT_STREQ(Json::AbacusJson::doc["array"][1]["Kstring"].GetString(), "100"); - - ASSERT_EQ(Json::AbacusJson::doc["array"][1]["double"].GetDouble(), 0.01); - - ASSERT_EQ(Json::AbacusJson::doc["array"][2]["int"].GetInt(), 2); - ASSERT_STREQ(Json::AbacusJson::doc["array"][2]["string"].GetString(), "200"); - - ASSERT_STREQ(Json::AbacusJson::doc["array"][2]["200"].GetString(), "string"); - ASSERT_STREQ(Json::AbacusJson::doc["array"][2]["Kstring"].GetString(), "200"); - ASSERT_EQ(Json::AbacusJson::doc["array"][2]["double"].GetDouble(), 0.02); - - ASSERT_STREQ(Json::AbacusJson::doc["array"][2]["new_add_Last"].GetString(), "correct2"); - - // add array in array - Json::jsonValue object0(JarrayType); - - object0.JPushBack(1); - object0.JPushBack(2); - object0.JPushBack(3); - - Json::jsonValue object1(JarrayType); - - object1.JPushBack(2.1); - object1.JPushBack(3.1); - object1.JPushBack(4.1); - - Json::jsonValue object2(JarrayType); - - object2.JPushBack("str1"); - object2.JPushBack("str2"); - object2.JPushBack("str3"); - - Json::jsonValue object3(JarrayType); - - std::string astr1 = "string1"; - std::string astr2 = "string2"; - std::string astr3 = "string3"; - object3.JPushBackString(astr1); - object3.JPushBackString(astr2); - object3.JPushBackString(astr3); - - Json::AbacusJson::add_json({"Darray"}, object0, true); - Json::AbacusJson::add_json({"Darray"}, object1, true); - Json::AbacusJson::add_json({"Darray"}, object2, true); - Json::AbacusJson::add_json({"Darray"}, object3, true); - - Json::AbacusJson::add_json({"Darray", 1, 0}, "new_add_method", false); - Json::AbacusJson::add_json({"Darray", 1, -2}, 40, false); - ASSERT_EQ(Json::AbacusJson::doc["Darray"][1][0].GetString(), "new_add_method"); - - ASSERT_EQ(Json::AbacusJson::doc["Darray"][0][0].GetInt(), 1); - ASSERT_EQ(Json::AbacusJson::doc["Darray"][0][1].GetInt(), 2); - ASSERT_EQ(Json::AbacusJson::doc["Darray"][0][2].GetInt(), 3); - - ASSERT_EQ(Json::AbacusJson::doc["Darray"][1][1].GetDouble(), 40); - ASSERT_EQ(Json::AbacusJson::doc["Darray"][1][2].GetDouble(), 4.1); - - ASSERT_STREQ(Json::AbacusJson::doc["Darray"][2][0].GetString(), "str1"); - ASSERT_STREQ(Json::AbacusJson::doc["Darray"][2][1].GetString(), "str2"); - ASSERT_STREQ(Json::AbacusJson::doc["Darray"][2][2].GetString(), "str3"); + void TearDown() override + { + std::remove("test.json"); + std::remove("json-output-not-a-directory"); + } - ASSERT_STREQ(Json::AbacusJson::doc["Darray"][3][0].GetString(), "string1"); - ASSERT_STREQ(Json::AbacusJson::doc["Darray"][3][1].GetString(), "string2"); - ASSERT_STREQ(Json::AbacusJson::doc["Darray"][3][2].GetString(), "string3"); -} + const Json::jsonValue& document() const + { + return Json::AbacusJson::document(); + } +}; -TEST(AbacusJsonTest, OutputJson) +TEST_F(AbacusJsonTest, OutputJson) { - Json::AbacusJson::doc.SetObject(); - - Json::AbacusJson::add_json({"key1"}, "value1", false); - Json::AbacusJson::add_json({"key2", "key3"}, 1, false); - Json::AbacusJson::add_json({"key4"}, 0.1, false); - Json::AbacusJson::add_json({"key5"}, true, false); - - Json::jsonValue object(JobjectType); - object.JaddNormal("int", 1); - Json::jsonValue object2(JarrayType); - - object.JaddNormal("arr", object2); - - // array test - Json::AbacusJson::add_json({"key6", "key7"}, object, true); - Json::AbacusJson::add_json({"key6", "key7", 0, "arr"}, 13, true); - Json::AbacusJson::add_json({"key6", "key7", 0, "arr"}, 14, true); - Json::AbacusJson::add_json({"key6", "key7", 0, "arr", 0}, 1, true); - - std::string filename = "test.json"; - Json::AbacusJson::write_to_json(filename); - - std::ifstream file(filename); + // Exercise our writer, including escaping, number types and insertion order. + Json::AbacusJson::document() = { + {"z", "quote: \"; slash: \\; newline: \n; UTF-8: \xCE\xB1"}, + {"a", std::string("a\0b", 3)}, + {"nested", {{"int", 1}, {"float", 0.1}, {"bool", true}, {"null", nullptr}, + {"array", Json::jsonValue::array({1, 2, 3})}}}}; + Json::AbacusJson::write_to_json("test.json"); + std::ifstream file("test.json"); ASSERT_TRUE(file.is_open()); - - std::string content((std::istreambuf_iterator(file)), std::istreambuf_iterator()); - ASSERT_NE(content.find("\"key1\": \"value1\","), std::string::npos); - ASSERT_NE(content.find("\"key2\": {"), std::string::npos); - ASSERT_NE(content.find("\"key3\": 1"), std::string::npos); - ASSERT_NE(content.find("\"key4\": 0.1"), std::string::npos); - ASSERT_NE(content.find("\"key5\": true"), std::string::npos); - - file.close(); + const Json::jsonValue result = Json::jsonValue::parse(file); + EXPECT_EQ(result, document()); + EXPECT_EQ(result.dump(), document().dump()); } -TEST(AbacusJsonTest, GeneralInfo) +TEST_F(AbacusJsonTest, GeneralInfo) { - std::time_t time_now = std::time(nullptr); - std::string start_time_str; - Json::convert_time(time_now, start_time_str); - Parameter param; - param.sys.start_time = time_now; - param.input.device = "cpu"; - param.input.pseudo_dir = "./abacus/test/pseudo_dir"; - param.input.orbital_dir = "./abacus/test/orbital_dir"; - param.sys.global_in_stru = "./abacus/test/stru_file"; - param.input.kpoint_file = "./abacus/test/kpoint_file"; - // output the json file - Json::AbacusJson::doc.Parse("{}"); Json::gen_general_info(param); - Json::json_output(); - - std::string filename = "abacus.json"; - std::ifstream file(filename); - ASSERT_TRUE(file.is_open()); - std::string content((std::istreambuf_iterator(file)), std::istreambuf_iterator()); - - ASSERT_NE(content.find(VERSION), std::string::npos); - ASSERT_NE(content.find("\"device\": \"cpu\","), std::string::npos); - ASSERT_NE(content.find("\"omp_num\": 0,"), std::string::npos); - ASSERT_NE(content.find("\"mpi_num\": 0,"), std::string::npos); - ASSERT_NE(content.find("\"orbital_dir\": \"./abacus/test/orbital_dir\","), std::string::npos); - ASSERT_NE(content.find("\"pseudo_dir\": \"./abacus/test/pseudo_dir\","), std::string::npos); - ASSERT_NE(content.find("\"stru_file\": \"./abacus/test/stru_file\","), std::string::npos); - ASSERT_NE(content.find("\"kpt_file\": \"./abacus/test/kpoint_file\","), std::string::npos); - ASSERT_NE(content.find(start_time_str), std::string::npos); + const Json::jsonValue& info = document().at("general_info"); + EXPECT_EQ(info["version"], VERSION); + EXPECT_EQ(info["device"], param.inp.device); +#ifdef __MPI + EXPECT_EQ(info["mpi_num"], Parallel_Global::mpi_number); + EXPECT_EQ(info["omp_num"], Parallel_Global::omp_number); +#else + EXPECT_EQ(info["mpi_num"], 1); + EXPECT_EQ(info["omp_num"], 1); +#endif + EXPECT_EQ(info["orbital_dir"], param.inp.orbital_dir); + EXPECT_EQ(info["pseudo_dir"], param.inp.pseudo_dir); + EXPECT_EQ(info["stru_file"], param.globalv.global_in_stru); + EXPECT_EQ(info["kpt_file"], param.inp.kpoint_file); + EXPECT_TRUE(info["start_time"].is_string()); + EXPECT_TRUE(info["end_time"].is_string()); + std::vector keys; + for (Json::jsonValue::const_iterator field = info.begin(); field != info.end(); ++field) + { + keys.push_back(field.key()); + } + EXPECT_EQ(keys, (std::vector{"version", "commit", "device", "mpi_num", "omp_num", + "pseudo_dir", "orbital_dir", "stru_file", "kpt_file", + "start_time", "end_time"})); + Json::add_nkstot(2); + Json::gen_general_info(param); + EXPECT_EQ(document()["init"]["nkstot"], 2); + EXPECT_EQ(document()["general_info"].size(), keys.size()); } - Magnetism::Magnetism() { this->tot_mag = 0.0; this->abs_mag = 0.0; } + Magnetism::~Magnetism() { } -TEST(AbacusJsonTest, InitInfo) + +TEST_F(AbacusJsonTest, InitInfo) { UnitCell ucell; Atom atomlist[3]; @@ -258,6 +114,7 @@ TEST(AbacusJsonTest, InitInfo) ucell.symm.spgname = "O_h"; ucell.atoms = atomlist; ucell.ntype = 3; + Input_para inp; inp.nbands = 10; inp.ecutwfc = 50.0; @@ -276,132 +133,263 @@ TEST(AbacusJsonTest, InitInfo) ucell.atoms[2].label = "O"; ucell.atoms[2].ncpp.zv = 5; ucell.atoms[2].na = 3; - ucell.nat = 0; - for (int i = 0; i < ucell.ntype; i++) - { - ucell.nat += ucell.atoms[i].na; - } - // init the doc allocator - Json::AbacusJson::doc.Parse("{}"); - int Jnkstot = 1; - Json::add_nkstot(Jnkstot); + ucell.nat = 6; + + Json::add_nkstot(1); Json::gen_init(&ucell, inp); - ASSERT_TRUE(Json::AbacusJson::doc.HasMember("init")); - ASSERT_EQ(Json::AbacusJson::doc["init"]["nkstot"].GetInt(), 1); - - ASSERT_EQ(Json::AbacusJson::doc["init"]["natom"].GetInt(), 6); - ASSERT_EQ(Json::AbacusJson::doc["init"]["nband"].GetInt(), 10); - - ASSERT_STREQ(Json::AbacusJson::doc["init"]["point_group"].GetString(), "T_d"); - ASSERT_STREQ(Json::AbacusJson::doc["init"]["point_group_in_space"].GetString(), "O_h"); - - ASSERT_EQ(Json::AbacusJson::doc["init"]["nelectron_each_type"]["Si"].GetDouble(), 3); - ASSERT_EQ(Json::AbacusJson::doc["init"]["nelectron_each_type"]["C"].GetDouble(), 4); - ASSERT_EQ(Json::AbacusJson::doc["init"]["nelectron_each_type"]["O"].GetDouble(), 5); - - ASSERT_EQ(Json::AbacusJson::doc["init"]["natom_each_type"]["Si"].GetInt(), 1); - ASSERT_EQ(Json::AbacusJson::doc["init"]["natom_each_type"]["C"].GetInt(), 2); - ASSERT_EQ(Json::AbacusJson::doc["init"]["natom_each_type"]["O"].GetInt(), 3); - - ASSERT_EQ(Json::AbacusJson::doc["init"]["ecutwfc"].GetDouble(), 50.0); - ASSERT_STREQ(Json::AbacusJson::doc["init"]["ecutwfc_unit"].GetString(), "Ry"); - ASSERT_STREQ(Json::AbacusJson::doc["init"]["smearing_method"].GetString(), "gauss"); - ASSERT_EQ(Json::AbacusJson::doc["init"]["smearing_sigma"].GetDouble(), 0.015); - ASSERT_STREQ(Json::AbacusJson::doc["init"]["smearing_sigma_unit"].GetString(), "Ry"); - ASSERT_STREQ(Json::AbacusJson::doc["init"]["kmesh_type"].GetString(), "gamma"); - ASSERT_EQ(Json::AbacusJson::doc["init"]["kspacing"][0].GetDouble(), 0.04); - ASSERT_EQ(Json::AbacusJson::doc["init"]["kspacing"][1].GetDouble(), 0.04); - ASSERT_EQ(Json::AbacusJson::doc["init"]["kspacing"][2].GetDouble(), 0.04); - ASSERT_EQ(Json::AbacusJson::doc["init"]["koffset"][0].GetDouble(), 0.0); - ASSERT_EQ(Json::AbacusJson::doc["init"]["koffset"][1].GetDouble(), 0.0); - ASSERT_EQ(Json::AbacusJson::doc["init"]["koffset"][2].GetDouble(), 0.0); + const Json::jsonValue& init = document().at("init"); + EXPECT_EQ(init["nkstot"], 1); + EXPECT_EQ(init["natom"], 6); + EXPECT_EQ(init["nband"], 10); + EXPECT_EQ(init["point_group"], "T_d"); + EXPECT_EQ(init["point_group_in_space"], "O_h"); + EXPECT_EQ(init.at("nelectron_each_type"), (Json::jsonValue{{"Si", 3.0}, {"C", 4.0}, {"O", 5.0}})); + EXPECT_EQ(init.at("natom_each_type"), (Json::jsonValue{{"Si", 1}, {"C", 2}, {"O", 3}})); + EXPECT_EQ(init.at("nelectron"), 26); + EXPECT_TRUE(init.at("nelectron").is_number_integer()); + EXPECT_TRUE(init.at("nelectron_each_type").at("C").is_number_float()); + EXPECT_EQ(init["ecutwfc"], 50.0); + EXPECT_EQ(init["ecutwfc_unit"], "Ry"); + EXPECT_EQ(init["smearing_method"], "gauss"); + EXPECT_EQ(init["smearing_sigma"], 0.015); + EXPECT_EQ(init["smearing_sigma_unit"], "Ry"); + EXPECT_EQ(init["kmesh_type"], "gamma"); + EXPECT_EQ(init["kspacing"], Json::jsonValue::array({0.04, 0.04, 0.04})); + EXPECT_EQ(init["koffset"], Json::jsonValue::array({0.0, 0.0, 0.0})); + + // Rebuild the per-species maps rather than retaining entries from a previous call. + ucell.ntype = 2; + ucell.nat = 3; + Json::gen_init(&ucell, inp); + EXPECT_EQ(init.at("natom_each_type"), (Json::jsonValue{{"Si", 1}, {"C", 2}})); + EXPECT_EQ(init.at("nelectron_each_type"), (Json::jsonValue{{"Si", 3.0}, {"C", 4.0}})); + EXPECT_EQ(init.at("nkstot"), 1); } -TEST(AbacusJsonTest, Init_stru_test) +TEST_F(AbacusJsonTest, InitStructure) { - // init ucell UnitCell ucell; + Atom atom; - Atom atomlist[1]; - std::string label[1]; + ucell.latvec = ModuleBase::Matrix3(0.1, 0.1, 0.1, + 0.2, 0.2, 0.2, + 0.3, 0.3, 0.3); + ucell.ntype = 1; + ucell.nat = 2; + ucell.pseudo_fn = {"si.ufp"}; + ucell.orbital_fn = {""}; + ucell.atoms = &atom; + ucell.lat0 = 10.0; + ucell.lat0_angstrom = ucell.lat0 * ModuleBase::BOHR_TO_A; + + atom.label = "Si"; + atom.na = 2; + atom.ncpp.zv = 4.0; + atom.tau = {ModuleBase::Vector3(0.0, 0.0, 0.0), + ModuleBase::Vector3(0.1, 0.1, 0.1)}; + atom.mag = {0.0, 131.0}; - ModuleBase::Matrix3 latvec; - latvec.e11 = 0.1; - latvec.e12 = 0.1; - latvec.e13 = 0.1; + Input_para inp; + Json::gen_stru(&ucell, inp); + + const Json::jsonValue& init = document().at("init"); + EXPECT_EQ(init["mag"], Json::jsonValue::array({0.0, 131.0})); + EXPECT_EQ(init["pp"]["Si"], "si.ufp"); + EXPECT_TRUE(init["orb"]["Si"].is_null()); + EXPECT_EQ(init["label"][0], "Si"); + EXPECT_EQ(init["element"]["Si"], ""); + EXPECT_EQ(init["coordinate"][0], Json::jsonValue::array({0.0, 0.0, 0.0})); + for (int i = 0; i < 3; ++i) + { + EXPECT_NEAR(init["coordinate"][1][i].get(), ModuleBase::BOHR_TO_A, 1.0e-12); + for (int j = 0; j < 3; ++j) + { + EXPECT_NEAR(init["cell"][i][j].get(), (i + 1) * ModuleBase::BOHR_TO_A, 1.0e-12); + } + } - latvec.e21 = 0.2; - latvec.e22 = 0.2; - latvec.e23 = 0.2; + // Reuse the cell to check shared init fields and repeated generation. + ucell.orbital_fn[0] = "Si.orb"; + inp.orbital_dir = "orbitals/"; + inp.kspacing = {0.1, 0.2, 0.3}; + inp.koffset = {0.0, 0.5, 0.0}; + Json::gen_stru(&ucell, inp); + Json::add_nkstot(3); + Json::gen_init(&ucell, inp); + const Json::jsonValue first = document(); + EXPECT_EQ(first["init"]["orb"]["Si"], "orbitals/Si.orb"); + EXPECT_EQ(first["init"]["nkstot"], 3); + EXPECT_EQ(first["init"]["natom"], 2); + Json::gen_init(&ucell, inp); + Json::gen_stru(&ucell, inp); + EXPECT_EQ(document(), first); + EXPECT_EQ(document().dump(), first.dump()); // Preserve key order, too. +} - latvec.e31 = 0.3; - latvec.e32 = 0.3; - latvec.e33 = 0.3; - ucell.latvec = latvec; +TEST_F(AbacusJsonTest, OutputRecords) +{ + EXPECT_THROW(Json::add_output_energy(-1.0), std::invalid_argument); + EXPECT_FALSE(document().contains("output")); + Json::AbacusJson::document()["output"] = Json::jsonValue::array(); + EXPECT_THROW(Json::add_output_energy(-1.0), std::out_of_range); + Json::init_output_array_obj(); + ASSERT_EQ(document().at("output").size(), 1u); + const Json::jsonValue initial = document()["output"][0]; + for (const char* key : {"e_fermi", "energy", "scf_converge", "force", "stress"}) + { + EXPECT_TRUE(initial[key].is_null()); + } + for (const char* key : {"coordinate", "mag", "cell"}) + { + EXPECT_TRUE(initial[key].is_array()); + EXPECT_TRUE(initial[key].empty()); + } - double lat0 = 10.0; - ucell.ntype = 1; - ucell.pseudo_fn.resize(1); - ucell.orbital_fn.resize(1); - ucell.atoms = atomlist; - ucell.lat0 = lat0; - ucell.lat0_angstrom = lat0 * ModuleBase::BOHR_TO_A; + Json::add_output_efermi_converge(1.5, true); + Json::add_output_energy(-10.0); + Json::add_output_scf_mag(1.0, 2.0, -9.0, -0.2, 1.0e-3, 0.5); + Json::add_output_scf_mag(1.0, 2.0, -10.0, -1.0, 1.0e-5, 0.6); + + const Json::jsonValue first = document()["output"][0]; + EXPECT_EQ(first["e_fermi"], 1.5); + EXPECT_EQ(first["energy"], -10.0); + EXPECT_EQ(first["scf_converge"], true); + EXPECT_EQ(first["total_mag"], 1.0); + EXPECT_EQ(first["absolute_mag"], 2.0); + ASSERT_EQ(first["scf"].size(), 2u); + EXPECT_EQ(first["scf"][1]["energy"], -10.0); + EXPECT_EQ(first["scf"][1]["ediff"], -1.0); + EXPECT_EQ(first["scf"][1]["drho"], 1.0e-5); + EXPECT_EQ(first["scf"][1]["time"], 0.6); + + Json::init_output_array_obj(); + Json::add_output_energy(-11.0); + ASSERT_EQ(document()["output"].size(), 2u); + EXPECT_EQ(document()["output"][0], first); + EXPECT_EQ(document()["output"][1]["energy"], -11.0); +} - ModuleBase::Vector3 tau[2]; +TEST_F(AbacusJsonTest, RejectsInvalidSections) +{ + for (const Json::jsonValue& invalid : {Json::jsonValue(nullptr), Json::jsonValue(1), + Json::jsonValue("invalid"), Json::jsonValue::array()}) + { + Json::AbacusJson::document()["init"] = invalid; + EXPECT_THROW(Json::add_nkstot(1), std::invalid_argument); + EXPECT_EQ(document().at("init"), invalid); + } + for (const Json::jsonValue& invalid : {Json::jsonValue(nullptr), Json::jsonValue(1), + Json::jsonValue("invalid"), Json::jsonValue::object()}) + { + Json::AbacusJson::document()["output"] = invalid; + EXPECT_THROW(Json::init_output_array_obj(), std::invalid_argument); + EXPECT_THROW(Json::add_output_energy(-1.0), std::invalid_argument); + EXPECT_EQ(document().at("output"), invalid); + } +} - Json::AbacusJson::doc.Parse("{}"); +TEST_F(AbacusJsonTest, RejectsInvalidRecordsAndScfHistory) +{ + for (const Json::jsonValue& invalid : {Json::jsonValue(nullptr), Json::jsonValue(1), + Json::jsonValue::array()}) + { + Json::AbacusJson::document()["output"] = Json::jsonValue::array({invalid}); + EXPECT_THROW(Json::add_output_energy(-1.0), std::invalid_argument); + EXPECT_EQ(document().at("output").at(0), invalid); + } + Json::init_output_array_obj(); + for (const Json::jsonValue& invalid : {Json::jsonValue(nullptr), Json::jsonValue(1), + Json::jsonValue::object()}) + { + Json::AbacusJson::document()["output"].back()["scf"] = invalid; + EXPECT_THROW(Json::add_output_scf_mag(0.0, 0.0, -1.0, 0.0, 0.1, 0.1), std::invalid_argument); + EXPECT_EQ(document().at("output").back().at("scf"), invalid); + } +} - double mag[2]; - // fill ucell - for (int i = 0; i < 1; i++) +TEST_F(AbacusJsonTest, OutputStructureForceAndStress) +{ + UnitCell ucell; + Atom atom; + ucell.atoms = &atom; + ucell.ntype = 1; + ucell.nat = 1; + ucell.lat0_angstrom = 2.0; + ucell.latvec = ModuleBase::Matrix3(1.0, 0.0, 0.0, + 0.0, 2.0, 0.0, + 0.0, 0.0, 3.0); + atom.na = 1; + atom.tau = {ModuleBase::Vector3(0.25, -0.5, 0.75)}; + atom.mag = {1.5}; + + ModuleBase::matrix force(1, 3); + force(0, 0) = 0.5e-8; + force(0, 1) = 2.0; + force(0, 2) = -3.0; + + ModuleBase::matrix stress(3, 3); + for (int i = 0; i < 3; ++i) { - ucell.atoms[i].label = "Si"; - atomlist[i].na = 2; - ucell.pseudo_fn[i] = "si.ufp"; - ucell.atoms[i].tau.resize(2); - atomlist[i].mag.resize(2); - for (int j = 0; j < atomlist[i].na; j++) + for (int j = 0; j < 3; ++j) { - atomlist[i].mag[j] = j * 131; - ucell.atoms[i].tau[j] = 0.1 * j; + stress(i, j) = 3 * i + j + 1; } } - Json::gen_stru(&ucell, Input_para{}); - - std::string filename = "readin.json"; - Json::AbacusJson::write_to_json(filename); - // compare result - ASSERT_TRUE(Json::AbacusJson::doc.HasMember("init")); - ASSERT_EQ(Json::AbacusJson::doc["init"]["mag"][0].GetDouble(), 0); - ASSERT_EQ(Json::AbacusJson::doc["init"]["mag"][1].GetDouble(), 131.0); - - ASSERT_STREQ(Json::AbacusJson::doc["init"]["pp"]["Si"].GetString(), "si.ufp"); - ASSERT_STREQ(Json::AbacusJson::doc["init"]["label"][0].GetString(), "Si"); - ASSERT_STREQ(Json::AbacusJson::doc["init"]["element"]["Si"].GetString(), ""); - - ASSERT_EQ(Json::AbacusJson::doc["init"]["coordinate"][0][0].GetDouble(), 0); - ASSERT_EQ(Json::AbacusJson::doc["init"]["coordinate"][0][1].GetDouble(), 0); - ASSERT_EQ(Json::AbacusJson::doc["init"]["coordinate"][0][2].GetDouble(), 0); - - EXPECT_NEAR(Json::AbacusJson::doc["init"]["coordinate"][1][0].GetDouble(), - ModuleBase::BOHR_TO_A, - 1.0e-12); - EXPECT_NEAR(Json::AbacusJson::doc["init"]["coordinate"][1][1].GetDouble(), - ModuleBase::BOHR_TO_A, - 1.0e-12); - EXPECT_NEAR(Json::AbacusJson::doc["init"]["coordinate"][1][2].GetDouble(), - ModuleBase::BOHR_TO_A, - 1.0e-12); + Json::init_output_array_obj(); + Json::add_output_cell_coo_stress_force(ucell, force, 2.0, stress, 0.5, true, true); + const Json::jsonValue first = document()["output"][0]; + ASSERT_EQ(first["force"].size(), 1u); + EXPECT_EQ(first["force"][0], Json::jsonValue::array({0.0, 4.0, -6.0})); + EXPECT_EQ(first["coordinate"][0], Json::jsonValue::array({0.5, -1.0, 1.5})); + EXPECT_EQ(first["mag"], Json::jsonValue::array({1.5})); + ASSERT_EQ(first["stress"].size(), 3u); + ASSERT_EQ(first["cell"].size(), 3u); for (int i = 0; i < 3; ++i) { for (int j = 0; j < 3; ++j) { - EXPECT_NEAR(Json::AbacusJson::doc["init"]["cell"][i][j].GetDouble(), - (i + 1) * ModuleBase::BOHR_TO_A, - 1.0e-12); + EXPECT_DOUBLE_EQ(first["stress"][i][j].get(), stress(i, j) * 0.5); + EXPECT_DOUBLE_EQ(first["cell"][i][j].get(), i == j ? 2.0 * (i + 1) : 0.0); } } + + // Replacing the same step's data must not append extra rows or nested arrays. + Json::add_output_cell_coo_stress_force(ucell, force, 2.0, stress, 0.5, true, true); + EXPECT_EQ(document()["output"][0], first); + + Json::init_output_array_obj(); + Json::add_output_cell_coo_stress_force(ucell, force, 2.0, stress, 0.5, false, false); + EXPECT_EQ(document()["output"][0], first); + const Json::jsonValue& second = document()["output"][1]; + EXPECT_TRUE(second["force"].is_null()); + EXPECT_TRUE(second["stress"].is_null()); + EXPECT_EQ(second["coordinate"], first["coordinate"]); + EXPECT_EQ(second["cell"], first["cell"]); +} + +TEST_F(AbacusJsonTest, NonFiniteNumbersSerializeAsNull) +{ + Json::init_output_array_obj(); + Json::add_output_energy(std::numeric_limits::quiet_NaN()); + Json::add_output_efermi_converge(std::numeric_limits::infinity(), false); + Json::AbacusJson::write_to_json("test.json"); + std::ifstream file("test.json"); + ASSERT_TRUE(file.is_open()); + const Json::jsonValue result = Json::jsonValue::parse(file); + EXPECT_TRUE(result.at("output").at(0).at("energy").is_null()); + EXPECT_TRUE(result.at("output").at(0).at("e_fermi").is_null()); +} + +TEST_F(AbacusJsonTest, FileOpenFailureIsReported) +{ + const std::string blocker = "json-output-not-a-directory"; + { + std::ofstream file(blocker); + ASSERT_TRUE(file.is_open()); + } + EXPECT_THROW(Json::AbacusJson::write_to_json(blocker + "/abacus.json"), std::runtime_error); + EXPECT_EQ(std::remove(blocker.c_str()), 0); } diff --git a/source/source_io/module_ml/io_npz.cpp b/source/source_io/module_ml/io_npz.cpp index 87901676f3a..832b8fb2eb2 100644 --- a/source/source_io/module_ml/io_npz.cpp +++ b/source/source_io/module_ml/io_npz.cpp @@ -72,9 +72,9 @@ void read_mat_npz(const Parallel_Orbitals* paraV, temp = element.erase(temp); } else - { - temp++; - } + { + temp++; + } } int z = 0; for(int j=0; j!=ModuleBase::element_name.size(); j++) @@ -367,9 +367,9 @@ void output_mat_npz_impl(const UnitCell& ucell, std::string& zipname, const hami temp = element.erase(temp); } else - { - temp++; - } + { + temp++; + } } int z = 0; for(int j=0; j!=ModuleBase::element_name.size(); j++) diff --git a/source/source_io/module_ml/write_mlkedf_desc.cpp b/source/source_io/module_ml/write_mlkedf_desc.cpp index 3bb2476ebc5..c372d731fb1 100644 --- a/source/source_io/module_ml/write_mlkedf_desc.cpp +++ b/source/source_io/module_ml/write_mlkedf_desc.cpp @@ -3,7 +3,7 @@ #include "write_mlkedf_desc.h" #include "npy.hpp" -#include "source_estate/module_charge/symm_rho.h" +#include "source_estate/module_charge/chg_symm.h" namespace ModuleIO { @@ -32,19 +32,17 @@ void Write_MLKEDF_Descriptors::generateTrainData_KS( this->cal_tool->getF_KS(psi, pelec, pw_psi, pw_rho, ucell, drho, enhancement, pauli); - Symmetry_rho srho; - std::vector rho_vec(nrxx); std::vector> rhog_vec(pw_rho->npw); double* rho_ptr = rho_vec.data(); std::complex* rhog_ptr = rhog_vec.data(); std::copy(enhancement.begin(), enhancement.end(), rho_vec.begin()); - srho.begin(0, &rho_ptr, &rhog_ptr, pw_rho->npw, nullptr, pw_rho, ucell.symm); + module_charge::cal_rhog_symm(0, &rho_ptr, &rhog_ptr, pw_rho->npw, nullptr, pw_rho, ucell.symm); std::copy(rho_vec.begin(), rho_vec.end(), enhancement.begin()); std::copy(pauli.begin(), pauli.end(), rho_vec.begin()); - srho.begin(0, &rho_ptr, &rhog_ptr, pw_rho->npw, nullptr, pw_rho, ucell.symm); + module_charge::cal_rhog_symm(0, &rho_ptr, &rhog_ptr, pw_rho->npw, nullptr, pw_rho, ucell.symm); std::copy(rho_vec.begin(), rho_vec.end(), pauli.begin()); diff --git a/source/source_io/module_mulliken/cal_mag.h b/source/source_io/module_mulliken/cal_mag.h index 896c3e228df..c5477200a76 100644 --- a/source/source_io/module_mulliken/cal_mag.h +++ b/source/source_io/module_mulliken/cal_mag.h @@ -26,7 +26,7 @@ template void cal_mag(Parallel_Orbitals* pv, hamilt::Hamilt* p_ham, K_Vectors& kv, - elecstate::DensityMatrix* dm, + module_dm::DensityMatrix* dm, const TwoCenterBundle& two_center_bundle, const LCAO_Orbitals& orb, UnitCell& ucell, @@ -58,7 +58,7 @@ void cal_mag(Parallel_Orbitals* pv, { std::vector> atom_mag(ucell.nat, std::vector(PARAM.inp.nspin, 0.0)); std::vector> constrain(ucell.nat, ModuleBase::Vector3(1, 1, 1)); - const hamilt::HContainer* dmr = dm->get_DMR_pointer(1); + const hamilt::HContainer* dmr = dm->get_dmr_ptr(1); std::vector moments; std::vector mag_x(ucell.nat, 0.0); std::vector mag_y(ucell.nat, 0.0); diff --git a/source/source_io/module_mulliken/output_dmk.cpp b/source/source_io/module_mulliken/output_dmk.cpp index c7a5e414d15..20c6c2dfa13 100644 --- a/source/source_io/module_mulliken/output_dmk.cpp +++ b/source/source_io/module_mulliken/output_dmk.cpp @@ -4,15 +4,15 @@ namespace ModuleIO { template -Output_DMK::Output_DMK(elecstate::DensityMatrix* p_DM, Parallel_Orbitals* ParaV, int nspin, int nks) +Output_DMK::Output_DMK(module_dm::DensityMatrix* p_DM, Parallel_Orbitals* ParaV, int nspin, int nks) : p_DM_(p_DM), ParaV_(ParaV), nspin_(nspin), nks_(nks) { } template -TK* Output_DMK::get_DMK(int ik) +TK* Output_DMK::get_dmk(int ik) { - return p_DM_->get_DMK_vector()[ik].data(); + return p_DM_->get_dmk_vec()[ik].data(); } template class Output_DMK; diff --git a/source/source_io/module_mulliken/output_dmk.h b/source/source_io/module_mulliken/output_dmk.h index f92be1a8fb0..266c0511b08 100644 --- a/source/source_io/module_mulliken/output_dmk.h +++ b/source/source_io/module_mulliken/output_dmk.h @@ -10,15 +10,15 @@ template class Output_DMK { public: - Output_DMK(elecstate::DensityMatrix* p_DM, - Parallel_Orbitals* ParaV, - int nspin, - int nks); + Output_DMK(module_dm::DensityMatrix* p_DM, + Parallel_Orbitals* ParaV, + int nspin, + int nks); - TK* get_DMK(int ik); + TK* get_dmk(int ik); private: - elecstate::DensityMatrix* p_DM_ = nullptr; + module_dm::DensityMatrix* p_DM_ = nullptr; Parallel_Orbitals* ParaV_ = nullptr; int nks_; int nspin_; diff --git a/source/source_io/module_mulliken/output_mulliken.cpp b/source/source_io/module_mulliken/output_mulliken.cpp index a687ddaf241..d477c0fa1e5 100644 --- a/source/source_io/module_mulliken/output_mulliken.cpp +++ b/source/source_io/module_mulliken/output_mulliken.cpp @@ -20,7 +20,7 @@ Output_Mulliken::Output_Mulliken(Output_Sk* output_sk, const std::vector& isk, int nspin) : output_sk_(output_sk), output_dmk_(output_dmk), - ParaV_(ParaV), cell_index_(cell_index), isk_(isk), nspin_(nspin) + ParaV_(ParaV), cell_index_(cell_index), isk_(isk), nspin_(nspin) { this->set_nspin(nspin); this->set_ParaV(ParaV); @@ -87,9 +87,9 @@ void Output_Mulliken::write_mulliken_nspin1(int istep, { /// header of the table std::string atom_label = this->cell_index_->get_atom_label(iat); - os << " ------------------" << std::endl; - os << " Atom " << iat+1 << " is " << atom_label << std::endl; - os << " ------------------" << std::endl; + os << " ------------------" << std::endl; + os << " Atom " << iat+1 << " is " << atom_label << std::endl; + os << " ------------------" << std::endl; os << FmtCore::format("%14s", std::string("zeta")) << FmtCore::format("%10s", std::string("spin1")) << std::endl; @@ -102,7 +102,7 @@ void Output_Mulliken::write_mulliken_nspin1(int istep, for (int M = 0; M < (2 * L + 1); M++) { os << fmt_of_label.format(ModuleBase::Name_Angular[L][M]) - << fmt_of_Z.format(Z+1) + << fmt_of_Z.format(Z+1) << fmt_of_chg.format(orb_chg[std::vector{iat, 0, L, Z, M}]) << std::endl; } // sum over m @@ -160,14 +160,14 @@ void Output_Mulliken::write_mulliken_nspin2(int istep, { /// header of the table std::string atom_label = this->cell_index_->get_atom_label(iat); - os << " ------------------" << std::endl; - os << " Atom " << iat+1 << " is " << atom_label << std::endl; - os << " ------------------" << std::endl; + os << " ------------------" << std::endl; + os << " Atom " << iat+1 << " is " << atom_label << std::endl; + os << " ------------------" << std::endl; os << FmtCore::format("%14s", std::string("zeta")) << FmtCore::format("%10s", std::string("spin1")) - << FmtCore::format("%10s", std::string("spin2")) + << FmtCore::format("%10s", std::string("spin2")) << FmtCore::format("%10s", std::string("sum")) - << FmtCore::format("%10s", std::string("diff")) << std::endl; + << FmtCore::format("%10s", std::string("diff")) << std::endl; /// loop of L for (int L = 0; L <= this->cell_index_->get_maxL(iat); L++) @@ -178,7 +178,7 @@ void Output_Mulliken::write_mulliken_nspin2(int istep, for (int M = 0; M < (2 * L + 1); M++) { os << fmt_of_label.format(ModuleBase::Name_Angular[L][M]) - << fmt_of_Z.format(Z+1) // be careful, Z+1, modified by mohan 2026-02-21 + << fmt_of_Z.format(Z+1) // be careful, Z+1, modified by mohan 2026-02-21 << fmt_of_chg.format(orb_chg[std::vector{iat, 0, L, Z, M}]) << fmt_of_chg.format(orb_chg[std::vector{iat, 1, L, Z, M}]) << fmt_of_chg.format(orb_chg[std::vector{iat, 0, L, Z, M}] @@ -201,21 +201,21 @@ void Output_Mulliken::write_mulliken_nspin2(int istep, { os << fmt_of_sum.format(std::string(" sum m")) << fmt_of_chg.format(sum_over_m[0]) - << fmt_of_chg.format(sum_over_m[1]) + << fmt_of_chg.format(sum_over_m[1]) << fmt_of_chg.format(sum_over_m[0] + sum_over_m[1]) << fmt_of_chg.format(sum_over_m[0] - sum_over_m[1]) << std::endl; } } os << fmt_of_sum.format(std::string(" sum mz")) << fmt_of_chg.format(sum_over_m_and_z[0]) - << fmt_of_chg.format(sum_over_m_and_z[1]) + << fmt_of_chg.format(sum_over_m_and_z[1]) << fmt_of_chg.format(sum_over_m_and_z[0] + sum_over_m_and_z[1]) << fmt_of_chg.format(sum_over_m_and_z[0] - sum_over_m_and_z[1]) << std::endl; os << std::endl; } os << fmt_of_sum.format(std::string(" sum lmz")) << fmt_of_chg.format(atom_chg[iat][0]) - << fmt_of_chg.format(atom_chg[iat][1]) + << fmt_of_chg.format(atom_chg[iat][1]) << fmt_of_chg.format(atom_chg[iat][0] + atom_chg[iat][1]) << fmt_of_chg.format(atom_chg[iat][0] - atom_chg[iat][1]) << std::endl; os << std::endl; @@ -251,9 +251,9 @@ void Output_Mulliken::write_mulliken_nspin4(int istep, { /// header of the table std::string atom_label = this->cell_index_->get_atom_label(iat); - os << " ------------------" << std::endl; - os << " Atom " << iat+1 << " is " << atom_label << std::endl; - os << " ------------------" << std::endl; + os << " ------------------" << std::endl; + os << " Atom " << iat+1 << " is " << atom_label << std::endl; + os << " ------------------" << std::endl; os << FmtCore::format("%14s", std::string("zeta")) << FmtCore::format("%10s", std::string("spin1")) << FmtCore::format("%10s", std::string("spin2")) @@ -269,7 +269,7 @@ void Output_Mulliken::write_mulliken_nspin4(int istep, for (int M = 0; M < (2 * L + 1); M++) { os << fmt_of_label.format(ModuleBase::Name_Angular[L][M]) - << fmt_of_Z.format(Z+1) + << fmt_of_Z.format(Z+1) << fmt_of_chg.format(orb_chg[std::vector{iat, 0, L, Z, M}]) << fmt_of_chg.format(orb_chg[std::vector{iat, 1, L, Z, M}]) << fmt_of_chg.format(orb_chg[std::vector{iat, 2, L, Z, M}]) @@ -289,23 +289,23 @@ void Output_Mulliken::write_mulliken_nspin4(int istep, { os << fmt_of_sum.format(std::string(" sum m")) << fmt_of_chg.format(sum_over_m[0]) - << fmt_of_chg.format(sum_over_m[1]) + << fmt_of_chg.format(sum_over_m[1]) << fmt_of_chg.format(sum_over_m[2]) - << fmt_of_chg.format(sum_over_m[3]) << std::endl; + << fmt_of_chg.format(sum_over_m[3]) << std::endl; } } os << fmt_of_sum.format(std::string(" sum mz")) << fmt_of_chg.format(sum_over_m_and_z[0]) - << fmt_of_chg.format(sum_over_m_and_z[1]) + << fmt_of_chg.format(sum_over_m_and_z[1]) << fmt_of_chg.format(sum_over_m_and_z[2]) - << fmt_of_chg.format(sum_over_m_and_z[3]) << std::endl; + << fmt_of_chg.format(sum_over_m_and_z[3]) << std::endl; os << std::endl; } os << fmt_of_sum.format(std::string(" sum lmz")) << fmt_of_chg.format(atom_chg[iat][0]) - << fmt_of_chg.format(atom_chg[iat][1]) + << fmt_of_chg.format(atom_chg[iat][1]) << fmt_of_chg.format(atom_chg[iat][2]) - << fmt_of_chg.format(atom_chg[iat][3]) << std::endl; + << fmt_of_chg.format(atom_chg[iat][3]) << std::endl; os << std::endl; os << std::left << " total charge on atom " << iat+1 << " " << fmt_of_chg.format(atom_chg[iat][0]) << std::endl; @@ -439,7 +439,7 @@ void Output_Mulliken::collect_MW(ModuleBase::matrix& MecMulP, const ModuleBa MecMulP(3, j) += mud(ic, ir).real(); } // WARNING (pre-existing, predates #7664, since 3.7.0 commit a339356): this M_y sign - // is suspect. mud is DM.S with the same conj-first DM convention as cal_dm_psi, so the + // is suspect. mud is DM.S with the same conj-first DM convention as dm_from_psi, so the // bare Im formula here likely yields -m_y. It must NOT be used as an oracle for the DM // convention (it is probably why #7664's DM m_y flip went unnoticed). Fix + unit test // should be a separate PR after verifying against a case with nonzero in-plane moment. @@ -584,7 +584,7 @@ void Output_Mulliken>::cal_orbMulP() for (size_t ik = 0; ik != this->isk_.size(); ++ik) { auto p_Sk = this->output_sk_->get_Sk(ik); - auto p_DMk = this->output_dmk_->get_DMK(ik); + auto p_DMk = this->output_dmk_->get_dmk(ik); ModuleBase::ComplexMatrix mud(this->ParaV_->ncol, this->ParaV_->nrow, true); #ifdef __MPI const char T_char = 'T'; @@ -634,7 +634,7 @@ void Output_Mulliken::cal_orbMulP() { ModuleBase::matrix mud; auto p_Sk = this->output_sk_->get_Sk(is); - auto p_DMk = this->output_dmk_->get_DMK(is); + auto p_DMk = this->output_dmk_->get_dmk(is); mud.create(this->ParaV_->ncol, this->ParaV_->nrow); #ifdef __MPI const char T_char = 'T'; diff --git a/source/source_io/module_output/cal_test.cpp b/source/source_io/module_output/cal_test.cpp index 44cbe5e40d2..2a7339bf1b9 100644 --- a/source/source_io/module_output/cal_test.cpp +++ b/source/source_io/module_output/cal_test.cpp @@ -46,80 +46,80 @@ double Cal_Test::meigts123=0.0; double Cal_Test::mtot=0.0; void Cal_Test::test_memory(const int nat, - const int ntype, - const ModuleBase::Matrix3& GGT, - const ModulePW::PW_Basis* rhopw, - const ModulePW::PW_Basis_K* wfcpw, - const std::string chr_mixing_mode, - const int chr_mixing_ndim) + const int ntype, + const ModuleBase::Matrix3& GGT, + const ModulePW::PW_Basis* rhopw, + const ModulePW::PW_Basis_K* wfcpw, + const std::string chr_mixing_mode, + const int chr_mixing_ndim) { - ModuleBase::TITLE("Cal_Test","test_memory"); - - const int ngmw = Cal_Test::cal_np(GGT,wfcpw->ggecut, rhopw->nx, rhopw->ny, rhopw->nz); - const int ngmc = Cal_Test::cal_np(GGT,rhopw->ggecut, rhopw->nx, rhopw->ny, rhopw->nz); - - std::cout << " number of atoms = " << nat << std::endl; - std::cout << " plane wave number for wave functions = " << ngmw << std::endl; - std::cout << " plane wave number for chage density = " << ngmc << std::endl; - - mporter = ModuleBase::Memory::calculate_mem ( rhopw->nxyz, "double"); - - mrho = mporter; - mrho_save = mrho; - mrho_core = mrho; - - // (2) memory for charge mixing - std::cout << " Mixing mode = " << chr_mixing_mode << std::endl; - if(chr_mixing_mode == "pulay") - { - std::cout << " Mixing dimension = " << chr_mixing_ndim << std::endl; - mRrho = chr_mixing_ndim * mrho; - mdRrho = (chr_mixing_ndim-1) * mrho; - mdrho = (chr_mixing_ndim-1) * mrho; - mrho_save2 = mrho; + ModuleBase::TITLE("Cal_Test","test_memory"); + + const int ngmw = Cal_Test::cal_np(GGT,wfcpw->ggecut, rhopw->nx, rhopw->ny, rhopw->nz); + const int ngmc = Cal_Test::cal_np(GGT,rhopw->ggecut, rhopw->nx, rhopw->ny, rhopw->nz); + + std::cout << " number of atoms = " << nat << std::endl; + std::cout << " plane wave number for wave functions = " << ngmw << std::endl; + std::cout << " plane wave number for chage density = " << ngmc << std::endl; + + mporter = ModuleBase::Memory::calculate_mem ( rhopw->nxyz, "double"); + + mrho = mporter; + mrho_save = mrho; + mrho_core = mrho; + + // (2) memory for charge mixing + std::cout << " Mixing mode = " << chr_mixing_mode << std::endl; + if(chr_mixing_mode == "pulay") + { + std::cout << " Mixing dimension = " << chr_mixing_ndim << std::endl; + mRrho = chr_mixing_ndim * mrho; + mdRrho = (chr_mixing_ndim-1) * mrho; + mdrho = (chr_mixing_ndim-1) * mrho; + mrho_save2 = mrho; // std::cout << " Memory for pulay mixing: " << mrho << " MB" << std::endl; - } - - mvltot = mrho; - mvr = mrho; - mvrs = mrho; - mvrs1 = mrho; - mvnew = mrho; - - mrhog = ModuleBase::Memory::calculate_mem( ngmc, "cdouble"); - mrhog_save = ModuleBase::Memory::calculate_mem( ngmc, "cdouble"); - mrhog_core = ModuleBase::Memory::calculate_mem( ngmc, "cdouble"); - - mhs = ModuleBase::Memory::calculate_mem( PARAM.globalv.nlocal*PARAM.globalv.nlocal, "double" ); - mwf = ModuleBase::Memory::calculate_mem( PARAM.globalv.nlocal*PARAM.inp.nbands, "double" ); - mnonzero = ModuleBase::Memory::calculate_mem( PARAM.globalv.nlocal*(PARAM.globalv.nlocal+1)/2, "bool"); + } + + mvltot = mrho; + mvr = mrho; + mvrs = mrho; + mvrs1 = mrho; + mvnew = mrho; + + mrhog = ModuleBase::Memory::calculate_mem( ngmc, "cdouble"); + mrhog_save = ModuleBase::Memory::calculate_mem( ngmc, "cdouble"); + mrhog_core = ModuleBase::Memory::calculate_mem( ngmc, "cdouble"); + + mhs = ModuleBase::Memory::calculate_mem( PARAM.globalv.nlocal*PARAM.globalv.nlocal, "double" ); + mwf = ModuleBase::Memory::calculate_mem( PARAM.globalv.nlocal*PARAM.inp.nbands, "double" ); + mnonzero = ModuleBase::Memory::calculate_mem( PARAM.globalv.nlocal*(PARAM.globalv.nlocal+1)/2, "bool"); // mohan comment out 2021-02-11 // mspar_hsrho = Memory::calculate_mem( Hnnz*3, "double"); - - mgvec = ModuleBase::Memory::calculate_mem( ngmc * 3 * 2, "double" ); - mig2fftw = ModuleBase::Memory::calculate_mem( ngmw , "int"); - mig2fftc = ModuleBase::Memory::calculate_mem( ngmc , "int"); - mgg = ModuleBase::Memory::calculate_mem( ngmc, "double"); - mig123 = ModuleBase::Memory::calculate_mem( ngmc*3, "int"); - mstrucFac = ModuleBase::Memory::calculate_mem( ntype*ngmc, "cdouble"); - meigts123 = ModuleBase::Memory::calculate_mem( nat * (2*rhopw->nx+1+2*rhopw->ny+1+2*rhopw->nz+1), "cdouble"); - - //(3) Memory for H,S matrix. - std::cout << " NLOCAL = " << PARAM.globalv.nlocal << std::endl; - std::cout << " NBANDS = " << PARAM.inp.nbands << std::endl; - - std::cout << " Memory for H,S matrix ( " - << PARAM.globalv.nlocal << ", " - << PARAM.globalv.nlocal << ") = " - << mhs << " MB" << std::endl; - - //(4) Memory for wave functions. - std::cout << " Memory for wave functions ( " - << PARAM.globalv.nlocal << ", " - << PARAM.inp.nbands << ") = " - << mwf << " MB" << std::endl; - - print_mem(1); + + mgvec = ModuleBase::Memory::calculate_mem( ngmc * 3 * 2, "double" ); + mig2fftw = ModuleBase::Memory::calculate_mem( ngmw , "int"); + mig2fftc = ModuleBase::Memory::calculate_mem( ngmc , "int"); + mgg = ModuleBase::Memory::calculate_mem( ngmc, "double"); + mig123 = ModuleBase::Memory::calculate_mem( ngmc*3, "int"); + mstrucFac = ModuleBase::Memory::calculate_mem( ntype*ngmc, "cdouble"); + meigts123 = ModuleBase::Memory::calculate_mem( nat * (2*rhopw->nx+1+2*rhopw->ny+1+2*rhopw->nz+1), "cdouble"); + + //(3) Memory for H,S matrix. + std::cout << " NLOCAL = " << PARAM.globalv.nlocal << std::endl; + std::cout << " NBANDS = " << PARAM.inp.nbands << std::endl; + + std::cout << " Memory for H,S matrix ( " + << PARAM.globalv.nlocal << ", " + << PARAM.globalv.nlocal << ") = " + << mhs << " MB" << std::endl; + + //(4) Memory for wave functions. + std::cout << " Memory for wave functions ( " + << PARAM.globalv.nlocal << ", " + << PARAM.inp.nbands << ") = " + << mwf << " MB" << std::endl; + + print_mem(1); // print_mem(8); // print_mem(16); @@ -129,15 +129,15 @@ void Cal_Test::test_memory(const int nat, // print_mem(64); // } - return; + return; } //! compute the number of plane waves int Cal_Test::cal_np(const ModuleBase::Matrix3& GGT, - const double &ggcut, - const int &n1, - const int &n2, - const int &n3) + const double &ggcut, + const int &n1, + const int &n2, + const int &n3) { /* @@ -151,78 +151,78 @@ int Cal_Test::cal_np(const ModuleBase::Matrix3& GGT, assert(n2>=0); assert(n3>=0); - int ibox[3]={0}; - - // set the center at origin point. - ibox[0] = int(n1 / 2.0) + 1; - ibox[1] = int(n2 / 2.0) + 1; - ibox[2] = int(n3 / 2.0) + 1; - // get the number of plane wave within 'gcut' - int ng = 0; - for (int i = -ibox[0]; i <= ibox[0]; i++) - { - for (int j = -ibox[1]; j <= ibox[1]; j++) - { - for (int k = -ibox[2]; k <= ibox[2]; k++) - { - ModuleBase::Vector3 f(i,j,k); - // g2= |f|^2 in the unit of (2Pi/lat0)^2 - double g2 = f * (GGT * f); - - // gcut is from input. - if (g2 <= ggcut) - { - ng++; - } - } - } - } - return ng; + int ibox[3]={0}; + + // set the center at origin point. + ibox[0] = int(n1 / 2.0) + 1; + ibox[1] = int(n2 / 2.0) + 1; + ibox[2] = int(n3 / 2.0) + 1; + // get the number of plane wave within 'gcut' + int ng = 0; + for (int i = -ibox[0]; i <= ibox[0]; i++) + { + for (int j = -ibox[1]; j <= ibox[1]; j++) + { + for (int k = -ibox[2]; k <= ibox[2]; k++) + { + ModuleBase::Vector3 f(i,j,k); + // g2= |f|^2 in the unit of (2Pi/lat0)^2 + double g2 = f * (GGT * f); + + // gcut is from input. + if (g2 <= ggcut) + { + ng++; + } + } + } + } + return ng; } void Cal_Test::print_mem(const int &nproc) { - std::cout << " ========================: " << std::endl; - mtot = 0.0; - - mtot += mporter + mrho + mrho_save + mrho_core + mRrho + - mdRrho + mdrho + mrho_save2 + mvltot + mvr + - mvrs + mvrs1 + mvnew + mrhog + mrhog_save + mrhog_core + - mgvec + mgg + mig2fftw + mig2fftc + mig123 + - mstrucFac + meigts123; - mtot += mwf + mhs; - - std::cout << " If you use " << nproc << " processors: " << std::endl; - std::cout << " MEMORY FOR porter : " << std::setw(15) << mporter/nproc << " MB" << std::endl; - std::cout << " MEMORY FOR rho : " << std::setw(15) << mrho/nproc << " MB" << std::endl; - std::cout << " MEMORY FOR rho_save : " << std::setw(15) << mrho_save/nproc << " MB" << std::endl; - std::cout << " MEMORY FOR rho_core : " << std::setw(15) << mrho_core/nproc << " MB" << std::endl; - std::cout << " MEMORY FOR Rrho : " << std::setw(15) << mRrho/nproc << " MB" << std::endl; - std::cout << " MEMORY FOR dRrho : " << std::setw(15) << mdRrho/nproc << " MB" << std::endl; - std::cout << " MEMORY FOR drho : " << std::setw(15) << mdrho/nproc << " MB" << std::endl; - std::cout << " MEMORY FOR rho_save2 : " << std::setw(15) << mrho_save2/nproc << " MB" << std::endl; - std::cout << " MEMORY FOR vltot : " << std::setw(15) << mvltot/nproc << " MB" << std::endl; - std::cout << " MEMORY FOR vr : " << std::setw(15) << mvr/nproc << " MB" << std::endl; - std::cout << " MEMORY FOR vrs : " << std::setw(15) << mvrs/nproc << " MB" << std::endl; - std::cout << " MEMORY FOR vrs1 : " << std::setw(15) << mvrs1/nproc << " MB" << std::endl; - std::cout << " MEMORY FOR vrnew : " << std::setw(15) << mvnew/nproc << " MB" << std::endl; - std::cout << " MEMORY FOR rhog : " << std::setw(15) << mrhog/nproc << " MB" << std::endl; - std::cout << " MEMORY FOR rhog_save : " << std::setw(15) << mrhog_save/nproc << " MB" << std::endl; - std::cout << " MEMORY FOR rhog_core : " << std::setw(15) << mrhog_core/nproc << " MB" << std::endl; - std::cout << " MEMORY FOR H, S matrix : " << std::setw(15) << mhs/nproc << " MB" << std::endl; - std::cout << " MEMORY FOR wave function: " << std::setw(15) << mwf/nproc << " MB" << std::endl; - std::cout << " MEMORY FOR spar H,S,rho : " << std::setw(15) << mspar_hsrho << " MB" << std::endl; - std::cout << " MEMORY FOR nonzero : " << std::setw(15) << mnonzero << " MB" << std::endl; - std::cout << " MEMORY FOR g vectors : " << std::setw(15) << mgvec/nproc << " MB" << std::endl; - std::cout << " MEMORY FOR gg : " << std::setw(15) << mgg/nproc << " MB" << std::endl; - std::cout << " MEMORY FOR fftw index : " << std::setw(15) << mig2fftw/nproc << " MB" << std::endl; - std::cout << " MEMORY FOR fftc index : " << std::setw(15) << mig2fftc/nproc << " MB" << std::endl; - std::cout << " MEMORY FOR ig123 : " << std::setw(15) << mig123/nproc << " MB" << std::endl; - std::cout << " MEMORY FOR strucFac : " << std::setw(15) << mstrucFac/nproc << " MB" << std::endl; - std::cout << " MEMORY FOR eigts1,2,3 : " << std::setw(15) << meigts123/nproc << " MB" << std::endl; - std::cout << " TOTAL MEMORY : " << std::setw(15) << mtot/nproc << " MB" << std::endl; - - std::cout << " MEMORY FOR nonzero : " << std::setw(15) + std::cout << " ========================: " << std::endl; + mtot = 0.0; + + mtot += mporter + mrho + mrho_save + mrho_core + mRrho + + mdRrho + mdrho + mrho_save2 + mvltot + mvr + + mvrs + mvrs1 + mvnew + mrhog + mrhog_save + mrhog_core + + mgvec + mgg + mig2fftw + mig2fftc + mig123 + + mstrucFac + meigts123; + mtot += mwf + mhs; + + std::cout << " If you use " << nproc << " processors: " << std::endl; + std::cout << " MEMORY FOR porter : " << std::setw(15) << mporter/nproc << " MB" << std::endl; + std::cout << " MEMORY FOR rho : " << std::setw(15) << mrho/nproc << " MB" << std::endl; + std::cout << " MEMORY FOR rho_save : " << std::setw(15) << mrho_save/nproc << " MB" << std::endl; + std::cout << " MEMORY FOR rho_core : " << std::setw(15) << mrho_core/nproc << " MB" << std::endl; + std::cout << " MEMORY FOR Rrho : " << std::setw(15) << mRrho/nproc << " MB" << std::endl; + std::cout << " MEMORY FOR dRrho : " << std::setw(15) << mdRrho/nproc << " MB" << std::endl; + std::cout << " MEMORY FOR drho : " << std::setw(15) << mdrho/nproc << " MB" << std::endl; + std::cout << " MEMORY FOR rho_save2 : " << std::setw(15) << mrho_save2/nproc << " MB" << std::endl; + std::cout << " MEMORY FOR vltot : " << std::setw(15) << mvltot/nproc << " MB" << std::endl; + std::cout << " MEMORY FOR vr : " << std::setw(15) << mvr/nproc << " MB" << std::endl; + std::cout << " MEMORY FOR vrs : " << std::setw(15) << mvrs/nproc << " MB" << std::endl; + std::cout << " MEMORY FOR vrs1 : " << std::setw(15) << mvrs1/nproc << " MB" << std::endl; + std::cout << " MEMORY FOR vrnew : " << std::setw(15) << mvnew/nproc << " MB" << std::endl; + std::cout << " MEMORY FOR rhog : " << std::setw(15) << mrhog/nproc << " MB" << std::endl; + std::cout << " MEMORY FOR rhog_save : " << std::setw(15) << mrhog_save/nproc << " MB" << std::endl; + std::cout << " MEMORY FOR rhog_core : " << std::setw(15) << mrhog_core/nproc << " MB" << std::endl; + std::cout << " MEMORY FOR H, S matrix : " << std::setw(15) << mhs/nproc << " MB" << std::endl; + std::cout << " MEMORY FOR wave function: " << std::setw(15) << mwf/nproc << " MB" << std::endl; + std::cout << " MEMORY FOR spar H,S,rho : " << std::setw(15) << mspar_hsrho << " MB" << std::endl; + std::cout << " MEMORY FOR nonzero : " << std::setw(15) << mnonzero << " MB" << std::endl; + std::cout << " MEMORY FOR g vectors : " << std::setw(15) << mgvec/nproc << " MB" << std::endl; + std::cout << " MEMORY FOR gg : " << std::setw(15) << mgg/nproc << " MB" << std::endl; + std::cout << " MEMORY FOR fftw index : " << std::setw(15) << mig2fftw/nproc << " MB" << std::endl; + std::cout << " MEMORY FOR fftc index : " << std::setw(15) << mig2fftc/nproc << " MB" << std::endl; + std::cout << " MEMORY FOR ig123 : " << std::setw(15) << mig123/nproc << " MB" << std::endl; + std::cout << " MEMORY FOR strucFac : " << std::setw(15) << mstrucFac/nproc << " MB" << std::endl; + std::cout << " MEMORY FOR eigts1,2,3 : " << std::setw(15) << meigts123/nproc << " MB" << std::endl; + std::cout << " TOTAL MEMORY : " << std::setw(15) << mtot/nproc << " MB" << std::endl; + + std::cout << " MEMORY FOR nonzero : " << std::setw(15) << (double)PARAM.globalv.nlocal*(PARAM.globalv.nlocal+1)/1028/1028/2.0/nproc << " MB" << std::endl; } diff --git a/source/source_io/module_output/cal_test.h b/source/source_io/module_output/cal_test.h index 665d3f95c11..34fa69015aa 100644 --- a/source/source_io/module_output/cal_test.h +++ b/source/source_io/module_output/cal_test.h @@ -5,62 +5,62 @@ namespace Cal_Test { - void test_memory(const int nat, - const int ntype, - const ModuleBase::Matrix3& GGT, - const ModulePW::PW_Basis* rhopw, - const ModulePW::PW_Basis_K* wfcpw, - const std::string chr_mixing_mode, - const int chr_mixing_ndim); + void test_memory(const int nat, + const int ntype, + const ModuleBase::Matrix3& GGT, + const ModulePW::PW_Basis* rhopw, + const ModulePW::PW_Basis_K* wfcpw, + const std::string chr_mixing_mode, + const int chr_mixing_ndim); - int cal_np(const ModuleBase::Matrix3& GGT, - const double &ggcut, - const int &n1, - const int &n2, - const int &n3); + int cal_np(const ModuleBase::Matrix3& GGT, + const double &ggcut, + const int &n1, + const int &n2, + const int &n3); - void print_mem(const int &nproc); + void print_mem(const int &nproc); - extern double mporter; + extern double mporter; - // about charge density. - extern double mrho; - extern double mrho_save; - extern double mrho_core; + // about charge density. + extern double mrho; + extern double mrho_save; + extern double mrho_core; - // about pulay mixing. - extern double mRrho; - extern double mdRrho; - extern double mdrho; - extern double mrho_save2; + // about pulay mixing. + extern double mRrho; + extern double mdRrho; + extern double mdrho; + extern double mrho_save2; - // about potential on FFT grid. - extern double mvltot; - extern double mvr; - extern double mvrs; - extern double mvrs1; - extern double mvnew; + // about potential on FFT grid. + extern double mvltot; + extern double mvr; + extern double mvrs; + extern double mvrs1; + extern double mvnew; - // about charge in g space. - extern double mrhog; - extern double mrhog_save; - extern double mrhog_core; - - // about others. - extern double mhs; - extern double mwf; - extern double mnonzero; - extern double mspar_hsrho; - - extern double mgvec; - extern double mig2fftw; - extern double mig2fftc; - extern double mgg; - extern double mig123; - extern double mstrucFac; - extern double meigts123; + // about charge in g space. + extern double mrhog; + extern double mrhog_save; + extern double mrhog_core; + + // about others. + extern double mhs; + extern double mwf; + extern double mnonzero; + extern double mspar_hsrho; + + extern double mgvec; + extern double mig2fftw; + extern double mig2fftc; + extern double mgg; + extern double mig123; + extern double mstrucFac; + extern double meigts123; - extern double mtot; + extern double mtot; } #endif diff --git a/source/source_io/module_output/cube_io.h b/source/source_io/module_output/cube_io.h index f97a639b1d0..d6e9b247731 100644 --- a/source/source_io/module_output/cube_io.h +++ b/source/source_io/module_output/cube_io.h @@ -10,7 +10,7 @@ namespace ModuleIO /// read volumetric data from .cube file into the parallel distributed grid. bool read_vdata_palgrid(const Parallel_Grid& pgrid, const int my_rank, - std::ofstream& ofs_running, + std::ostream& ofs_running, const std::string& fn, double* const data, const int nat); diff --git a/source/source_io/module_output/output_log.cpp b/source/source_io/module_output/output_log.cpp index 4c11edfe286..a642f8b7322 100644 --- a/source/source_io/module_output/output_log.cpp +++ b/source/source_io/module_output/output_log.cpp @@ -262,17 +262,17 @@ void print_force(std::ofstream& ofs_running, /*nrows=*/atom_label.size(), /*formats=*/{"%8s", "%20.10f", "%20.10f", "%20.10f"}, /*indent*/1, - {FmtTable::Align::RIGHT,FmtTable::Align::RIGHT}); + {FmtTable::Align::RIGHT,FmtTable::Align::RIGHT}); - fmt << atom_label << force_x << force_y << force_z; - table = fmt.str(); + fmt << atom_label << force_x << force_y << force_z; + table = fmt.str(); ofs_running << table; - if (PARAM.inp.test_force) - { - std::cout << table; - } + if (PARAM.inp.test_force) + { + std::cout << table; + } } void print_force(std::ofstream& ofs, const MDCell& cell, const std::string& name) diff --git a/source/source_io/module_output/output_log.h b/source/source_io/module_output/output_log.h index 0a12869a449..8b6c84cbe96 100644 --- a/source/source_io/module_output/output_log.h +++ b/source/source_io/module_output/output_log.h @@ -73,10 +73,10 @@ void print_force(std::ofstream& ofs, const MDCell& cell, const std::string& name /// @param f stress components /// @param ry true if the unit of force is a.u. void print_stress(const std::string& name, - const ModuleBase::matrix& scs, - const bool screen, - const bool ry, - std::ofstream &ofs); + const ModuleBase::matrix& scs, + const bool screen, + const bool ry, + std::ofstream &ofs); /// @brief write head for scf iteration /// @param ofs_running output stream diff --git a/source/source_io/module_output/print_info.cpp b/source/source_io/module_output/print_info.cpp index 7cf32256eea..5852e8d4f76 100644 --- a/source/source_io/module_output/print_info.cpp +++ b/source/source_io/module_output/print_info.cpp @@ -1,5 +1,7 @@ #include "print_info.h" +#include "source_basis/module_pw/pw_basis_k.h" +#include "source_basis/module_pw/pw_basis_big.h" // use PW_Basis_Big #include "source_base/global_variable.h" #include "source_io/module_parameter/parameter.h" diff --git a/source/source_io/module_output/print_info.h b/source/source_io/module_output/print_info.h index f7917da9fb9..8258835c3fc 100644 --- a/source/source_io/module_output/print_info.h +++ b/source/source_io/module_output/print_info.h @@ -1,17 +1,23 @@ #ifndef PRINT_INFO_H #define PRINT_INFO_H -#include "source_basis/module_pw/pw_basis_k.h" #include "source_cell/klist.h" #include "source_cell/unitcell.h" #include "source_io/module_parameter/input_parameter.h" +namespace ModulePW +{ +class PW_Basis; +class PW_Basis_Big; +class PW_Basis_K; +} + namespace ModuleIO { // print out to screen about the readin parameters void print_parameters( - const UnitCell& ucell, - K_Vectors& kv, + const UnitCell& ucell, + K_Vectors& kv, const Input_para& inp); void print_time(time_t& time_start, time_t& time_finish); diff --git a/source/source_io/module_output/read_cube.cpp b/source/source_io/module_output/read_cube.cpp index a4155f78f89..89301404607 100644 --- a/source/source_io/module_output/read_cube.cpp +++ b/source/source_io/module_output/read_cube.cpp @@ -1,4 +1,5 @@ #include "source_io/module_output/cube_io.h" +#include #include #include "source_base/parallel_grid.h" #include "source_io/module_parameter/parameter.h" @@ -7,23 +8,25 @@ bool ModuleIO::read_vdata_palgrid( const Parallel_Grid& pgrid, const int my_rank, - std::ofstream& ofs_running, + std::ostream& ofs_running, const std::string& fn, double* const data, const int natom) { ModuleBase::TITLE("ModuleIO", "read_vdata_palgrid"); - // check if the file exists + // Only the root rank parses the file. On failure it must abort the whole + // run instead of returning: the other ranks enter pgrid.bcast() below and + // would block in MPI_Recv waiting for data that never comes. std::ifstream ifs(fn.c_str()); - if (!ifs) - { - std::string tmp_warning_info = "!!! Couldn't find the file: " + fn; - ofs_running << tmp_warning_info << std::endl; - return false; - } - else + if (my_rank == 0) { + if (!ifs) + { + ofs_running << " !!! Couldn't find the file: " << fn << std::endl; + ModuleBase::WARNING_QUIT("ModuleIO::read_vdata_palgrid", + "couldn't find the cube file: " + fn); + } ofs_running << " Find the file " << fn << " , try to read it." << std::endl; } @@ -51,9 +54,14 @@ bool ModuleIO::read_vdata_palgrid( std::vector> atom_pos; std::vector data_read; - // we've already checked the file existence, so we don't need the returned value here - ModuleIO::read_cube(fn, comment, natom, origin, nx_read, ny_read, nz_read, - dx, dy, dz, atom_type, atom_charge, atom_pos, data_read); + // validate the cube content before copying or interpolating the data + if (!ModuleIO::read_cube(fn, comment, natom, origin, nx_read, ny_read, nz_read, + dx, dy, dz, atom_type, atom_charge, atom_pos, data_read)) + { + ofs_running << " !!! Failed to parse the cube file: " << fn << std::endl; + ModuleBase::WARNING_QUIT("ModuleIO::read_vdata_palgrid", + "failed to parse the cube file: " + fn); + } // if mismatch, trilinear interpolate if (nx == nx_read && ny == ny_read && nz == nz_read) @@ -165,21 +173,29 @@ bool ModuleIO::read_cube(const std::string& file, if (!ifs) { - return false; + return false; } comment.resize(2); for (auto& c : comment) { - std::getline(ifs, c); + std::getline(ifs, c); } ifs >> natom; + if (ifs.fail() || natom < 0) + { + return false; + } origin.resize(3); - for (auto& cp : origin) - { - ifs >> cp; + for (auto& cp : origin) + { + ifs >> cp; + } + if (ifs.fail()) + { + return false; } dx.resize(3); @@ -188,20 +204,38 @@ bool ModuleIO::read_cube(const std::string& file, ifs >> nx >> dx[0] >> dx[1] >> dx[2]; ifs >> ny >> dy[0] >> dy[1] >> dy[2]; ifs >> nz >> dz[0] >> dz[1] >> dz[2]; + if (ifs.fail() || nx <= 0 || ny <= 0 || nz <= 0) + { + return false; + } atom_type.resize(natom); atom_charge.resize(natom); atom_pos.resize(natom, std::vector(3)); - for (int i = 0;i < natom;++i) + for (int i = 0; i < natom; ++i) { ifs >> atom_type[i] >> atom_charge[i] >> atom_pos[i][0] >> atom_pos[i][1] >> atom_pos[i][2]; } + if (ifs.fail()) + { + return false; + } - const int nxyz = nx * ny * nz; + // guard against int overflow before allocating the data buffer + const std::int64_t nxyz_64 = static_cast(nx) * ny * nz; + if (nxyz_64 > std::numeric_limits::max()) + { + return false; + } + const int nxyz = static_cast(nxyz_64); data.resize(nxyz); - for (int i = 0;i < nxyz;++i) - { - ifs >> data[i]; + for (int i = 0; i < nxyz; ++i) + { + ifs >> data[i]; + } + if (ifs.fail()) + { + return false; } ifs.close(); diff --git a/source/source_io/module_output/write_orb_info.h b/source/source_io/module_output/write_orb_info.h index c759a3b2d88..3a035f5679a 100644 --- a/source/source_io/module_output/write_orb_info.h +++ b/source/source_io/module_output/write_orb_info.h @@ -4,7 +4,7 @@ namespace ModuleIO { - void write_orb_info(const UnitCell* ucell); + void write_orb_info(const UnitCell* ucell); } #endif diff --git a/source/source_io/module_parameter/input_conv.cpp b/source/source_io/module_parameter/input_conv.cpp index adeac3353c1..b05f2826351 100644 --- a/source/source_io/module_parameter/input_conv.cpp +++ b/source/source_io/module_parameter/input_conv.cpp @@ -120,8 +120,6 @@ void Input_Conv::Convert() #ifdef __LCAO TD_info::out_current = PARAM.inp.out_current; TD_info::out_current_k = PARAM.inp.out_current_k; - TD_info::out_vecpot = PARAM.inp.out_vecpot; - TD_info::init_vecpot_file = PARAM.inp.init_vecpot_file; const int out_hsr_format = PARAM.inp.out_hsr[0]; TD_info::out_mat_R = (out_hsr_format >= 1 && out_hsr_format <= 3) || PARAM.inp.out_hsr_npz_compat; #endif // __LCAO @@ -256,6 +254,14 @@ void Input_Conv::Convert() ModuleBase::WARNING_QUIT("Input_Conv", "EXX stress is not supported for basis_type = lcao_in_pw"); } + if (cal_exx && inp.basis_type == "lcao" && inp.kpar != 1) + { + // module_ri's real-space D(R)/Ds construction (RI_2D_Comm::split_m2D_ktoR_k) only + // sums each pool's local k-points, with no reduce_pool/reduce_all afterwards, so + // E_exx/H_exx(R)/forces would silently be built from a single pool's k-points. + ModuleBase::WARNING_QUIT("Input_Conv", "EXX (basis_type = lcao) does not support k-point parallelism (kpar > 1)"); + } + //---------------------------------------------------------- // reset symmetry flag to avoid error //---------------------------------------------------------- diff --git a/source/source_io/module_parameter/input_parameter.h b/source/source_io/module_parameter/input_parameter.h index c5afaaaae1e..020a87ff4e3 100644 --- a/source/source_io/module_parameter/input_parameter.h +++ b/source/source_io/module_parameter/input_parameter.h @@ -3,6 +3,7 @@ #include "md_parameter.h" #include "source_base/vector3.h" +#include #include #include @@ -85,6 +86,7 @@ struct Input_para double nelec_delta = 0.0; ///< change in the number of total electrons double nupdown = 0.0; std::string dft_functional = "default"; ///< input DFT functional. + int gga_grad = 0; ///< Noncollinear GGA: 0 original, 1 local axis, 2 regularized projected LCA. double xc_temperature = 0.0; ///< only relevant if finite temperature functional is used double pseudo_rcut = 15.0; ///< cut-off radius for calculating msh bool pseudo_mesh = false; ///< 0: use msh to normalize radial wave functions; 1: @@ -187,6 +189,16 @@ struct Input_para // ============== #Parameters (5.Molecular dynamics) =========================== MD_para mdp; + // FIXME(liuyu): ref_cell_factor is currently DISABLED. Setting any + // non-1.0 value triggers WARNING_QUIT in read_input_item_md.cpp. + // The reference-cell mechanism has design problems: when + // ref_cell_factor > 1, PW_Basis::lat0/tpiba/G/GGT/omega hold + // reference-cell values, but external code (sum_rho, get_local_pp_energy, + // cal_delta_escf, makov_payne, wfc IO, DFPT, OFDFT) reads them as + // physical-cell quantities, producing wrong results in variable-cell + // (NPT) calculations. To re-enable, PW_Basis must be refactored to + // separate reference-cell grid (FFT dims nx/ny/nz) from physical-cell + // lattice quantities (lat0/tpiba/G/GGT/omega). double ref_cell_factor = 1; ///< construct a reference cell bigger than the ///< initial cell liuyu 2023-03-21 std::vector cal_syns = {0, 8}; ///< calculate asynchronous S matrix to output {enable, precision} @@ -304,6 +316,15 @@ struct Input_para // ============== #Parameters (9.rt-tddft) =========================== double td_dt = -1.0; ///< time step for propagation int estep_per_md = 1; ///< number of electronic steps per MD step + std::string lin_solver = "gmres"; ///< linear solver for real-time propagation + std::string td_orthonormal = "cholesky"; ///< Orthonormalization for PW real-time propagation. + bool td_out_stat = false; ///< Output PW real-time conservation diagnostics. + int lin_gmres_restart = 20; ///< Maximum Arnoldi steps per GMRES cycle. + bool td_cn_init = true; ///< CN subspace initial guess and residual reuse. + bool lin_reconstruct = true; ///< Explicit GMRES residual reconstruction with periodic audits. + std::string lin_precond = "kinetic_recycle"; ///< right preconditioner for PW propagation + double lin_thr = 0.0; ///< zero selects the precision-dependent tolerance + int lin_maxiter = 500; ///< maximum iterations per linear solve double td_force_dt = 0.02; ///<"fs" bool td_vext = false; ///< add extern potential or not // std::string td_vext_dire = "1"; ///< vext direction @@ -451,7 +472,6 @@ struct Input_para std::vector out_mat_dh_vnl = {0, 8}; ///< output nonlocal pseudopotential dH/dR (dV^NL/dR) matrices std::vector out_mat_dh_vh = {0, 8}; ///< output Hartree dH/dR (dV^H/dR) matrices std::vector out_mat_dh_vxc = {0, 8}; ///< output XC dH/dR (dV^XC/dR) matrices - std::vector out_mat_dh_exx = {0, 8}; ///< output exact-exchange dH/dR (dV^EXX/dR) matrices std::vector out_mat_ds = {0, 8}; ///< output dS/dR matrices with precision bool out_mat_xc = false; ///< output exchange-correlation matrix in ///< KS-orbital representation. @@ -460,7 +480,6 @@ struct Input_para bool out_hr_npz = false; ///< output H(R) matrix in npz format bool out_hsr_npz = false; ///< output H(R) and S(R) matrices in npz format bool out_dm_npz = false; ///< output DM(R) matrix in npz format - int out_interval = 1; bool out_app_flag = true; ///< whether output r(R), H(R), S(R), T(R), and dH(R) matrices ///< in an append manner during MD liuyu 2023-03-20 int out_ndigits = 8; ///< Assuming 8 digits precision is needed for matrices output @@ -489,7 +508,7 @@ struct Input_para // ============== #Parameters (12.Postprocess) =========================== double dos_emin_ev = -15.0; double dos_emax_ev = 15.0; - double dos_edelta_ev = 0.01; + double dos_edelta_ev = 0.03; double dos_scale = 0.01; double dos_sigma = 0.07; ///< pengfei 2014-10-13 int dos_nche = 100; ///< orders of Chebyshev expansions for dos @@ -737,10 +756,11 @@ struct Input_para // ============== #Parameters (22.EXX PW) ===================== // EXX for planewave basis, rhx0820 2025-03-10 bool exxace = true; // exxace, exact exchange for planewave basis, https://doi.org/10.1021/acs.jctc.6b00092 - bool exx_gamma_extrapolation = true; // gamma point extrapolation for exx, https://doi.org/10.1103/PhysRevB.79.205114 + bool exx_gamma_extra = true; // gamma point extrapolation for exx, https://doi.org/10.1103/PhysRevB.79.205114 std::string exx_thr_type = "density"; ///< threshold type for exx outer loop double exx_ene_thr = 1e-5; ///< threshold when exx_thr_type = energy double ecutexx = 0.0; ///< energy cutoff for exx calculation, Ry + int exx_batch_size = 0; ///< band chunk width of the EXX batched path, 0 = all bands // ==== #Parameters (23.XC external parameterization) ======== /* diff --git a/source/source_io/module_parameter/md_parameter.h b/source/source_io/module_parameter/md_parameter.h index 92a42ce8b65..706218e6960 100644 --- a/source/source_io/module_parameter/md_parameter.h +++ b/source/source_io/module_parameter/md_parameter.h @@ -65,6 +65,10 @@ struct MD_para double md_tolerance = 100.0; ///< tolerance for velocity rescaling (K) int md_nraise = 1; ///< parameters used when md_type=nvt + bool plumed = false; ///< use PLUMED for collective variables and enhanced + ///< sampling (requires ABACUS compiled with PLUMED) + std::string plumed_file = "plumed.dat"; ///< input file of PLUMED + bool dump_force = true; ///< output atomic forces into the file MD_dump or ///< not. liuyu 2023-03-01 bool dump_vel = true; ///< output atomic velocities into the file MD_dump or diff --git a/source/source_io/module_parameter/parameter.h b/source/source_io/module_parameter/parameter.h index b6f54f7a28a..3ebfbd24f4d 100644 --- a/source/source_io/module_parameter/parameter.h +++ b/source/source_io/module_parameter/parameter.h @@ -53,9 +53,9 @@ extern Parameter PARAM; // temperarily put here namespace GlobalV { - extern int NPROC; - extern int MY_RANK; - extern std::ofstream ofs_running; - extern std::ofstream ofs_warning; + extern int NPROC; + extern int MY_RANK; + extern std::ofstream ofs_running; + extern std::ofstream ofs_warning; } // namespace GlobalV #endif diff --git a/source/source_io/module_parameter/read_inp_estruc.cpp b/source/source_io/module_parameter/read_inp_estruc.cpp index 6f96654199a..5da535085ca 100644 --- a/source/source_io/module_parameter/read_inp_estruc.cpp +++ b/source/source_io/module_parameter/read_inp_estruc.cpp @@ -472,7 +472,9 @@ The other way is only available when compiling with LIBXC, and it allows for sup item.description = R"(The number of spin components of wave functions. * 1: Spin degeneracy * 2: Collinear spin polarized. -* 4: Noncollinear or spin-orbit calculations. Set nspin to 4 explicitly when noncolin or lspinorb is enabled.)"; +* 4: Noncollinear or spin-orbit calculations. Set nspin to 4 explicitly when noncolin or lspinorb is enabled. +* Note: With nspin=2 and no initial magnetization in STRU, a moment of 1.0 is autoset for every atom, unless symmetry is 1. +* Note: With nspin=4 no moment is ever autoset. The calculation starts from zero magnetic moment and a warning is printed; set 'mag' explicitly in STRU for the magnetic atoms if a magnetic ground state is expected.)"; item.default_value = "1"; item.unit = ""; read_sync_int(input.nspin); @@ -488,6 +490,27 @@ The other way is only available when compiling with LIBXC, and it allows for sup }; this->add_item(item); } + { + Input_Item item("gga_grad"); + item.annotation = "Noncollinear GGA gradient method: 0 original, 1 local axis, 2 regularized projected LCA"; + item.category = "Electronic structure"; + item.type = "Integer"; + item.description = R"(Selects the local spin mapping for LDA/GGA functionals in magnetic nspin=4 calculations. +* 0: preserves the original algorithm (default). +* 1: uses the local magnetization magnitude instead of the global quantization axis in the built-in GGA gradient correction. For LIBXC functionals, 0 and 1 are equivalent. +* 2: uses a C2-regularized magnetization magnitude with eta = 1e-3 in atomic density units. The spin densities are (abs(n + rho_core) +/- min(S_eta(m), abs(n + rho_core)))/2. GGA gradients are the local-map Jacobian applied to the FFT gradients of the four density channels. The potential reverses this same discrete energy graph, including the radial Hessian and density/sigma clipping branches; the GGA stress uses the corresponding metric derivative. +For r = |m| and x = r/eta, S_eta = eta*x^3*(3*x^2 - 8*x + 6) for r < eta, and S_eta = r otherwise. The regularization is part of the functional definition, including its first and second derivatives. +Mode 2 also uses this local map for the LDA contribution. Other spin configurations retain their existing behavior.)"; + item.default_value = "0"; + read_sync_int(input.gga_grad); + item.check_value = [](const Input_Item& item, const Parameter& para) { + if (para.input.gga_grad < 0 || para.input.gga_grad > 2) + { + ModuleBase::WARNING_QUIT("ReadInput", "gga_grad must be 0, 1, or 2."); + } + }; + this->add_item(item); + } { Input_Item item("smearing_method"); item.annotation = "type of smearing_method: gauss; fd; fixed; mp; mp2; mv"; @@ -985,7 +1008,9 @@ Note: If gamma_only is set to 1, the KPT file will be overwritten. So make sure * Relationship with lspinorb: * noncolin=0, lspinorb=1: SOC with z-axis magnetism only (for non-magnetic materials with SOC) * noncolin=1, lspinorb=0: Non-collinear magnetism without SOC - * noncolin=1, lspinorb=1: Both non-collinear magnetism and SOC)"; + * noncolin=1, lspinorb=1: Both non-collinear magnetism and SOC +* Note: When nspin=4 and noncolin=0, only the z component of the initial magnetization in STRU is used; x/y components are ignored and a warning is printed. +* Note: When nspin=4 and no initial magnetization is set in STRU, the calculation starts from zero magnetic moment; no automatic magnetization is assigned.)"; item.default_value = "False"; item.unit = ""; read_sync_bool(input.noncolin); diff --git a/source/source_io/module_parameter/read_inp_other.cpp b/source/source_io/module_parameter/read_inp_other.cpp index 7021da0e492..2a99ac60222 100644 --- a/source/source_io/module_parameter/read_inp_other.cpp +++ b/source/source_io/module_parameter/read_inp_other.cpp @@ -882,14 +882,14 @@ When false (default), both the direction and magnitude of the magnetic moment ar this->add_item(item); } { - Input_Item item("exx_gamma_extrapolation"); + Input_Item item("exx_gamma_extra"); item.annotation = "whether to perform gamma extrapolation in exxpw"; item.category = "Exact Exchange (PW)"; item.type = "Boolean"; item.description = "Whether to use the gamma point extrapolation method to calculate the Fock exchange operator. See https://doi.org/10.1103/PhysRevB.79.205114 for details. Should be set to true most of the time."; item.default_value = "True"; item.unit = ""; - read_sync_bool(input.exx_gamma_extrapolation); + read_sync_bool(input.exx_gamma_extra); this->add_item(item); } { @@ -897,7 +897,7 @@ When false (default), both the direction and magnitude of the magnetic moment ar item.annotation = "energy cutoff for exx calculation, Ry"; item.category = "Exact Exchange (PW)"; item.type = "Real"; - item.description = "The energy cutoff for EXX (Fock) exchange operator in plane wave basis calculations. Reducing ecutexx below ecutrho may significantly accelerate EXX computations. This speed improvement comes with a reduced numerical accuracy in the exchange energy calculation."; + item.description = "The energy cutoff for EXX (Fock) exchange operator in plane wave basis calculations. The pair-density G-sphere of the exchange operator, the EXX energy, and the EXX stress are all truncated at this value. If ecutexx yields a smaller FFT box and every |k+G|^2 of the wavefunctions fits inside it (i.e. ecutexx should not be smaller than ecutwfc), all EXX FFTs run on that smaller grid (QE ecutfock-style), which can significantly accelerate EXX computations. If the small grid is not usable (box not smaller, wavefunctions do not fit, or the FFT box is distributed over MPI), a warning is printed and the full grid is used. Reducing ecutexx below ecutrho reduces the numerical accuracy of the exchange contribution."; item.default_value = "same as ecutrho"; item.unit = "Ry"; read_sync_double(input.ecutexx); @@ -910,6 +910,23 @@ When false (default), both the direction and magnitude of the magnetic moment ar this->add_item(item); } + { + Input_Item item("exx_batch_size"); + item.annotation = "band chunk width of the EXX batched path, 0 = all bands"; + item.category = "Exact Exchange (PW)"; + item.type = "Integer"; + item.description = "Number of bands processed per round of the EXX batched FFT path. 0 (the default) processes all bands in one round, which is fastest but needs nbands * nxyz work buffers; a positive value processes the bands in chunks of that width, trading some performance for a proportionally smaller memory footprint. The result is independent of the chunking."; + item.default_value = "0"; + read_sync_int(input.exx_batch_size); + item.check_value = [](const Input_Item& item, const Parameter& para) { + if (para.input.exx_batch_size < 0) + { + ModuleBase::WARNING_QUIT("ReadInput", "exx_batch_size must >= 0"); + } + }; + this->add_item(item); + } + { Input_Item item("exx_thr_type"); item.annotation = "threshold type for exx outer loop, energy or density"; diff --git a/source/source_io/module_parameter/read_inp_out.cpp b/source/source_io/module_parameter/read_inp_out.cpp index eb0f1d4b661..a891d5fa837 100644 --- a/source/source_io/module_parameter/read_inp_out.cpp +++ b/source/source_io/module_parameter/read_inp_out.cpp @@ -31,7 +31,7 @@ void ReadInput::item_output() { Input_Item item("out_freq_td"); item.annotation = "print information every few completed electronic iterations in RT-TDDFT"; - item.category = "Output information"; + item.category = "Real-Time TDDFT (Common)"; item.type = "Integer"; item.description = "Controls the output interval in completed electronic evolution steps during RT-TDDFT calculations. When set to a positive integer n, detailed information (see out_freq_ion) is printed every n electron time-evolution steps (i.e., every STEP OF ELECTRON EVOLVE). For example, if you wish to output information once per ionic step, you should set out_freq_td equal to estep_per_md, since one ionic step corresponds to estep_per_md electronic evolution steps." "\n\n[NOTE] This parameter is only active in RT-TDDFT mode (esolver_type = tddft). It has no effect in ground-state calculations."; @@ -99,15 +99,15 @@ In molecular dynamics simulations, the output frequency is controlled by out_fre [NOTE] In the 3.10-LTS version, the file names are SPIN1_CHG.cube and SPIN1_CHG_INI.cube, etc.)"; item.default_value = "0 3"; item.unit = ""; - item.read_value = [](const Input_Item& item, Parameter& para) { - const size_t count = item.get_size(); - if (count < 1) ModuleBase::WARNING_QUIT("ReadInput", "out_chg needs at least 1 value"); - para.input.out_chg[0] = std::stoi(item.str_values[0]); + item.read_value = [](const Input_Item& item, Parameter& para) { + const size_t count = item.get_size(); + if (count < 1) ModuleBase::WARNING_QUIT("ReadInput", "out_chg needs at least 1 value"); + para.input.out_chg[0] = std::stoi(item.str_values[0]); para.input.out_chg[1] = 3; - if (count >= 2) try { para.input.out_chg[1] = std::stoi(item.str_values[1]); } - catch (const std::invalid_argument&) { /* do nothing */ } - catch (const std::out_of_range&) {/* do nothing */} - }; + if (count >= 2) try { para.input.out_chg[1] = std::stoi(item.str_values[1]); } + catch (const std::invalid_argument&) { /* do nothing */ } + catch (const std::out_of_range&) {/* do nothing */} + }; // reset value in some special case item.reset_value = [](const Input_Item& item, Parameter& para) { para.input.out_chg[0] = (para.input.calculation == "get_wf" || para.input.calculation == "get_pchg") @@ -187,15 +187,15 @@ In molecular dynamics calculations, the output frequency is controlled by out_fr item.default_value = "False"; item.unit = ""; item.set_availability("basis_type==lcao"); - item.read_value = [](const Input_Item& item, Parameter& para) { - const size_t count = item.get_size(); - if (count < 1) ModuleBase::WARNING_QUIT("ReadInput", "out_dmk needs at least 1 value"); - para.input.out_dmk[0] = assume_as_boolean(item.str_values[0]); + item.read_value = [](const Input_Item& item, Parameter& para) { + const size_t count = item.get_size(); + if (count < 1) ModuleBase::WARNING_QUIT("ReadInput", "out_dmk needs at least 1 value"); + para.input.out_dmk[0] = assume_as_boolean(item.str_values[0]); para.input.out_dmk[1] = 8; - if (count >= 2) try { para.input.out_dmk[1] = std::stoi(item.str_values[1]); } - catch (const std::invalid_argument&) { /* do nothing */ } - catch (const std::out_of_range&) {/* do nothing */} - }; + if (count >= 2) try { para.input.out_dmk[1] = std::stoi(item.str_values[1]); } + catch (const std::invalid_argument&) { /* do nothing */ } + catch (const std::out_of_range&) {/* do nothing */} + }; item.reset_value = [](const Input_Item& item, Parameter& para) { if (para.input.calculation == "get_wf" || para.input.calculation == "get_pchg") { @@ -207,7 +207,7 @@ In molecular dynamics calculations, the output frequency is controlled by out_fr } { Input_Item item("out_dmr"); - item.annotation = "output density matrix DM(R) with respect to lattice vector R (with precision 8)"; + item.annotation = "output density matrix DM(R) with respect to lattice vector R (with precision 8)"; item.category = "Output information"; item.type = R"(Boolean \[Integer\](optional))"; item.description = R"(Whether to output the density matrix with Bravias lattice vector R index into files in the folder OUT.${suffix}. The files are named as dmr{s}{spin index}{g}{geometry index}{_nao} + {".csr"}. Here, 's' refers to spin, where s1 means spin up channel while s2 means spin down channel, and the sparse matrix format 'csr' is mentioned in out_hsr. Finally, if out_app_flag is set to false, the file name contains the optional 'g' index for each ionic step that may have different geometries, and if out_app_flag is set to true, the density matrix with respect to Bravias lattice vector R accumulates during ionic steps: @@ -219,29 +219,29 @@ In molecular dynamics calculations, the output frequency is controlled by out_fr item.unit = ""; item.set_availability("basis_type==lcao and gamma_only==0"); item.read_value = [](const Input_Item& item, Parameter& para) { - const size_t count = item.get_size(); - if (count < 1) ModuleBase::WARNING_QUIT("ReadInput", "out_dmr needs at least 1 value"); - para.input.out_dmr[0] = assume_as_boolean(item.str_values[0]); - para.input.out_dmr[1] = 8; - if (count >= 2) try { para.input.out_dmr[1] = std::stoi(item.str_values[1]); } - catch (const std::invalid_argument&) { /* do nothing */ } - catch (const std::out_of_range&) {/* do nothing */} - }; + const size_t count = item.get_size(); + if (count < 1) ModuleBase::WARNING_QUIT("ReadInput", "out_dmr needs at least 1 value"); + para.input.out_dmr[0] = assume_as_boolean(item.str_values[0]); + para.input.out_dmr[1] = 8; + if (count >= 2) try { para.input.out_dmr[1] = std::stoi(item.str_values[1]); } + catch (const std::invalid_argument&) { /* do nothing */ } + catch (const std::out_of_range&) {/* do nothing */} + }; item.reset_value = [](const Input_Item& item, Parameter& para) { if (para.input.calculation == "get_wf" || para.input.calculation == "get_pchg") { para.input.out_dmr[0] = 0; } }; - item.check_value = [](const Input_Item& item, const Parameter& para) { - if (para.sys.gamma_only_local == true && para.input.out_dmr[0]) - { - ModuleBase::WARNING_QUIT("ReadInput", "out_dmr is only valid for multi-k calculation"); - } - }; + item.check_value = [](const Input_Item& item, const Parameter& para) { + if (para.sys.gamma_only_local == true && para.input.out_dmr[0]) + { + ModuleBase::WARNING_QUIT("ReadInput", "out_dmr is only valid for multi-k calculation"); + } + }; - sync_intvec(input.out_dmr, 2, 0); - this->add_item(item); + sync_intvec(input.out_dmr, 2, 0); + this->add_item(item); } { Input_Item item("out_wfc_pw"); @@ -324,13 +324,12 @@ Also controled by out_freq_ion and out_app_flag. item.annotation = "output energy and dos"; item.category = "Output information"; item.type = "Integer"; - item.description = R"(Whether to output the density of states (DOS). For more information, refer to the dos.md. + item.description = R"(Whether to output the density of states (DOS) and projected density of states (PDOS). For more information, refer to the dos.md. * 0: no output -* 1: output the density of states (DOS) - * nspin=1 or 4: doss1g{geom}_{basis}.txt, where geom is the geometry index when cell changes or ions move while basis is either pw or nao. - * nspin=2: doss1g{geom}_{basis}.txt and doss2g{geom}_{basis}.txt for two spin channles. -* 2: (LCAO) output the density of states (DOS) and the projected density of states (PDOS) -* 3: output the Fermi surface file (fermi.bxsf) in BXSF format that can be visualized by XCrySDen)"; +* 1: output the density of states (DOS) and projected density of states (PDOS, LCAO only) + * nspin=1 or 4: doss1g{geom}_{basis}.txt and pdoss1g{geom}_{basis}.txt, where geom is the geometry index when cell changes or ions move while basis is either pw or nao. + * nspin=2: doss1/doss2 and pdoss1/pdoss2 files for two spin channels. +Note: values 2 and 3 are no longer supported. Setting out_dos to 2 or 3 will raise an error.)"; item.default_value = "0"; item.unit = ""; read_sync_int(input.out_dos); @@ -341,17 +340,17 @@ Also controled by out_freq_ion and out_app_flag. } }; item.check_value = [](const Input_Item& item, const Parameter& para) { - if (para.input.out_dos == 3 && para.input.symmetry == "1") + if (para.input.out_dos == 2) { ModuleBase::WARNING_QUIT("ReadInput", - "symmetry can't be used for out_dos==3(Fermi Surface " - "Plotting) by now."); + "out_dos = 2 has been deprecated. " + "PDOS is now written automatically with out_dos = 1."); } - if (para.input.basis_type == "pw" && para.input.out_dos == 3) + if (para.input.out_dos == 3) { ModuleBase::WARNING_QUIT("ReadInput", - "Fermi Surface Plotting not " - "implemented for plane wave now."); + "out_dos = 3 (Fermi surface output) has been removed. " + "Please use out_dos = 1 for DOS output."); } }; this->add_item(item); @@ -443,7 +442,7 @@ Also controled by out_freq_ion and out_app_flag. item.annotation = "output the structure files per ion step"; item.category = "Output information"; item.type = "Integer"; - item.description = "Controls the output of structure files per ionic step in geometry relaxation calculations. The files are written to the OUT.{suffix}/ directory. Each file corresponds to the structure at RELAX STEP ${istep}, i.e., the structure for which that step's energy was computed (before the relax move), and includes a header comment with the ABACUS version, timestamp, energy, and stress tensor. When out_freq_ion is positive, the numbered files STRU{istep} (or STRU{istep}.cif) are written every out_freq_ion steps; when out_freq_ion is 0, no numbered files are output.\n" + item.description = "Controls the output of structure files per ionic step. The files are written to the OUT.{suffix}/ directory. Each file corresponds to the structure at RELAX STEP ${istep} (for scf/nscf this is the single step), i.e., the structure for which that step's energy was computed (before the relax move), and includes a header comment with the ABACUS version, timestamp, energy, and stress tensor. When out_freq_ion is positive, the numbered files STRU{istep} (or STRU{istep}.cif) are written every out_freq_ion steps during geometry relaxation; when out_freq_ion is 0, no numbered files are output. This parameter is effective for scf/nscf/relax/cell-relax; for scf/nscf only STRU_FINAL (or STRU_FINAL.cif) is written, and structure output is disabled by default unless out_stru is set explicitly. Molecular dynamics structure output is instead controlled by md_restartfreq (STRU_MD_*).\n" " - 0: No structure files are output.\n" " - 1: ABACUS STRU format files are output. The latest structure is written to STRU_NOW (overwritten each step), the numbered file STRU{istep} (e.g., STRU1, STRU2) is written every out_freq_ion steps (when out_freq_ion is positive), and the final converged structure is written to STRU_FINAL. No CIF files are output.\n" " - 2: CIF format files are output. The latest structure is written to STRU_NOW.cif (overwritten each step), the numbered file STRU{istep}.cif (e.g., STRU1.cif, STRU2.cif) is written every out_freq_ion steps (when out_freq_ion is positive), and the final converged structure is written to STRU_FINAL.cif. No non-CIF files are output.\n" @@ -486,8 +485,11 @@ Also controled by out_freq_ion and out_app_flag. } }; item.reset_value = [](const Input_Item& item, Parameter& para) { - const std::vector offlist = {"nscf", "get_s", "get_pchg", "get_wf"}; - if (std::find(offlist.begin(), offlist.end(), para.input.calculation) != offlist.end()) + // For scf/nscf/get_s/get_pchg/get_wf the default is no structure + // output; an explicitly user-set out_stru value is preserved. + const std::vector offlist = {"scf", "nscf", "get_s", "get_pchg", "get_wf"}; + if (std::find(offlist.begin(), offlist.end(), para.input.calculation) != offlist.end() + && !item.is_read()) { para.input.out_stru = 0; } @@ -695,14 +697,14 @@ For binary output, each file uses the same basename as text output with a .dat s item.default_value = "False [8]"; item.unit = "Ry"; item.set_availability("basis_type==lcao"); - item.read_value = [](const Input_Item& item, Parameter& para) { - const size_t count = item.get_size(); - if (count < 1) ModuleBase::WARNING_QUIT("ReadInput", "out_mat_tk needs at least 1 value"); - para.input.out_mat_tk[0] = assume_as_boolean(item.str_values[0]); + item.read_value = [](const Input_Item& item, Parameter& para) { + const size_t count = item.get_size(); + if (count < 1) ModuleBase::WARNING_QUIT("ReadInput", "out_mat_tk needs at least 1 value"); + para.input.out_mat_tk[0] = assume_as_boolean(item.str_values[0]); para.input.out_mat_tk[1] = 8; - if (count >= 2) try { para.input.out_mat_tk[1] = std::stoi(item.str_values[1]); } - catch (const std::invalid_argument&) { /* do nothing */ } - catch (const std::out_of_range&) {/* do nothing */} + if (count >= 2) try { para.input.out_mat_tk[1] = std::stoi(item.str_values[1]); } + catch (const std::invalid_argument&) { /* do nothing */ } + catch (const std::out_of_range&) {/* do nothing */} }; sync_intvec(input.out_mat_tk, 2, 0); this->add_item(item); @@ -751,7 +753,7 @@ For binary output, each file uses the same basename as text output with a .dat s item.annotation = "output T(R) matrix"; item.category = "Output information"; item.type = R"(Boolean \[Integer\](optional))"; - item.description = "Generate files containing the kinetic energy matrix. The optional second parameter controls text output precision. The format will be the same as the Hamiltonian matrix and overlap matrix as mentioned in out_hsr. The name of the files will be trs1_nao.csr and so on. Also controled by out_freq_ion and out_app_flag." + item.description = "Generate files containing the kinetic energy matrix. The optional second parameter controls text output precision. The format will be the same as the Hamiltonian matrix and overlap matrix as mentioned in out_hsr. The name of the files will be tr_nao.csr and so on. Also controled by out_freq_ion and out_app_flag." "\n\n[NOTE] In the 3.10-LTS version, the file name is data-TR-sparse_SPIN0.csr."; item.default_value = "False 8"; item.unit = "Ry"; @@ -989,41 +991,6 @@ For binary output, each file uses the same basename as text output with a .dat s sync_intvec(input.out_mat_dh_vxc, para.input.out_mat_dh_vxc.size(), 0); this->add_item(item); } - { - Input_Item item("out_mat_dh_exx"); - item.annotation = "output exact-exchange dH/dR (dV^EXX/dR) matrices"; - item.category = "Output information"; - item.type = "Integer"; - item.description = "Whether to print files containing the derivatives of the exact-exchange matrix dV^EXX/dR." - "\n\nSee out_mat_dh for format details."; - item.default_value = "0 8"; - item.unit = "Ry/Bohr"; - item.read_value = [](const Input_Item& item, Parameter& para) { - const size_t count = item.get_size(); - try { - para.input.out_mat_dh_exx[0] = assume_as_boolean(item.str_values[0]); - para.input.out_mat_dh_exx[1] = 8; - if (count >= 2) try { para.input.out_mat_dh_exx[1] = std::stoi(item.str_values[1]); } - catch (const std::invalid_argument&) { - ModuleBase::WARNING("Input", "out_mat_dh_exx precision must be an integer, using default 8"); - } - for (size_t i = 2; i < count; ++i) - try { para.input.out_mat_dh_exx.push_back(std::stoi(item.str_values[i]) - 1); } - catch (const std::invalid_argument&) { - ModuleBase::WARNING("Input", "out_mat_dh_exx atom index must be an integer, skipping"); - } - } - catch (const std::invalid_argument& e) { - ModuleBase::WARNING("Input", "out_mat_dh_exx enable flag must be 0/1, using default 0"); - } - }; - item.check_value = [](const Input_Item& item, const Parameter& para) { - if (para.input.out_mat_dh_exx[0] && para.input.nspin == 4) - ModuleBase::WARNING_QUIT("ReadInput", "out_mat_dh_exx is not available for nspin = 4"); - }; - sync_intvec(input.out_mat_dh_exx, para.input.out_mat_dh_exx.size(), 0); - this->add_item(item); - } { Input_Item item("out_mat_h_t"); item.annotation = "output kinetic energy T(R) matrix"; @@ -1254,18 +1221,18 @@ For binary output, each file uses the same basename as text output with a .dat s item.annotation = "output the expectation values of angular momentum operators"; item.category = "Output information"; item.type = R"(Boolean \[Integer\](optional))"; - item.description = "Whether to print the expectation value of the angular momentum operator , , and in the basis of the localized atomic orbitals. The files are named OUT.{suffix}_Lx.dat, OUT.{suffix}_Ly.dat, and OUT.{suffix}_Lz.dat. The second integer controls the precision of the output."; + item.description = "Whether to print the expectation value of the angular momentum operator , , and in the basis of the localized atomic orbitals. The files are named lx_nao.txt, ly_nao.txt, and lz_nao.txt (or lxg{step+1}_nao.txt etc. when out_freq_ion is set). The second integer controls the precision of the output."; item.default_value = "False 8"; item.unit = ""; item.set_availability("basis_type==lcao"); - item.read_value = [](const Input_Item& item, Parameter& para) { - const size_t count = item.get_size(); - if (count < 1) ModuleBase::WARNING_QUIT("ReadInput", "out_mat_l needs at least 1 value"); - para.input.out_mat_l[0] = assume_as_boolean(item.str_values[0]); + item.read_value = [](const Input_Item& item, Parameter& para) { + const size_t count = item.get_size(); + if (count < 1) ModuleBase::WARNING_QUIT("ReadInput", "out_mat_l needs at least 1 value"); + para.input.out_mat_l[0] = assume_as_boolean(item.str_values[0]); para.input.out_mat_l[1] = 8; - if (count >= 2) try { para.input.out_mat_l[1] = std::stoi(item.str_values[1]); } - catch (const std::invalid_argument&) { /* do nothing */ } - catch (const std::out_of_range&) {/* do nothing */} + if (count >= 2) try { para.input.out_mat_l[1] = std::stoi(item.str_values[1]); } + catch (const std::invalid_argument&) { /* do nothing */ } + catch (const std::out_of_range&) {/* do nothing */} }; sync_intvec(input.out_mat_l, 2, 0); this->add_item(item); @@ -1674,7 +1641,7 @@ In molecular dynamics calculations, the output frequency is controlled by out_fr { Input_Item item("out_dipole"); item.annotation = "output dipole or not"; - item.category = "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory"; + item.category = "Real-Time TDDFT (Common)"; item.type = "Boolean"; item.description = R"(Controls electric-dipole output. In RT-TDDFT, each enabled spin channel is written to OUT.{suffix}/dipole_s[spin].txt using a one-based spin number. Every row contains the one-based electronic-step index followed by the Cartesian electronic-dipole components $P_x$, $P_y$, and $P_z$ in atomic units. The running log additionally reports the electronic, ionic, and total dipoles and the norm of the total dipole. * True: Output the electric dipole information. @@ -1687,36 +1654,36 @@ In molecular dynamics calculations, the output frequency is controlled by out_fr { Input_Item item("out_current"); item.annotation = "output current or not"; - item.category = "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory"; + item.category = "Real-Time TDDFT (Common)"; item.type = "Integer"; - item.description = R"(Controls the current-density output method for LCAO RT-TDDFT. Output rows contain the one-based electronic-step index followed by $J_x$, $J_y$, and $J_z$ in atomic units. + item.description = R"(Controls the current-density output method for RT-TDDFT. Each row contains the one-based electronic-step index followed by $J_x$, $J_y$, and $J_z$ in atomic units; the initial ground state is step 1 and subsequent steps increase by one. * 0: Do not output current. -* 1: Explicitly construct the velocity operator from the momentum, vector-potential, and KB nonlocal-pseudopotential terms using two-center and spherical-grid integrals: $\hat{v}_{\alpha}=-\mathrm{i}\nabla_{\alpha}+A_{\alpha}(t)+\mathrm{i}\left[\widetilde{V}_{\mathrm{NL}}^{\mathrm{KB}},r_{\alpha}\right]$, where $\widetilde{V}_{\mathrm{NL}}^{\mathrm{KB}}=\mathrm{e}^{-\mathrm{i}\boldsymbol{A}(t)\cdot\boldsymbol{r}}\hat{V}_{\mathrm{NL}}^{\mathrm{KB}}\mathrm{e}^{\mathrm{i}\boldsymbol{A}(t)\cdot\boldsymbol{r}}$. $\boldsymbol{A}(t)$ is nonzero only for the velocity gauge (td_stype=1); otherwise $\boldsymbol{A}(t)=0$. Other nonlocal Hamiltonian terms, such as EXX, are not included explicitly. The total current is written to OUT.{suffix}/current_tot.txt. -* 2: Use the full Hamiltonian to construct the generalized velocity matrix in a nonorthogonal NAO basis, $\widetilde{v}_{\alpha}=\partial_{\alpha}H+\mathrm{i}HS^{-1}\mathcal{R}_{\alpha}-\mathrm{i}\mathcal{R}_{\alpha}S^{-1}H-HS^{-1}\partial_{\alpha}S$. This includes all contributions available in the real-space Hamiltonian matrix when enabled. This method is more general but more expensive. The total current is written to OUT.{suffix}/current_tot_comm.txt.)"; +* 1: Available for PW and LCAO. PW evaluates the occupied-state expectation values of the velocity operator, using plane-wave momentum and nonlocal-projector derivatives; in the velocity gauge these are evaluated at the vector-potential-shifted momentum. In the length gauge, a kinetic-energy-density functional also contributes its velocity term. LCAO explicitly constructs the velocity operator from the momentum, vector-potential, and KB nonlocal-pseudopotential terms using two-center and spherical-grid integrals: $\hat{v}_{\alpha}=-\mathrm{i}\nabla_{\alpha}+A_{\alpha}(t)+\mathrm{i}\left[\widetilde{V}_{\mathrm{NL}}^{\mathrm{KB}},r_{\alpha}\right]$, where $\widetilde{V}_{\mathrm{NL}}^{\mathrm{KB}}=\mathrm{e}^{-\mathrm{i}\boldsymbol{A}(t)\cdot\boldsymbol{r}}\hat{V}_{\mathrm{NL}}^{\mathrm{KB}}\mathrm{e}^{\mathrm{i}\boldsymbol{A}(t)\cdot\boldsymbol{r}}$. $\boldsymbol{A}(t)$ is nonzero only for the velocity gauge (td_stype=1); otherwise $\boldsymbol{A}(t)=0$. Other nonlocal Hamiltonian terms, such as EXX, are not included explicitly. The total current is written to OUT.{suffix}/current_tot.txt. +* 2: Available only for LCAO. Use the full Hamiltonian to construct the generalized velocity matrix in a nonorthogonal NAO basis, $\widetilde{v}_{\alpha}=\partial_{\alpha}H+\mathrm{i}HS^{-1}\mathcal{R}_{\alpha}-\mathrm{i}\mathcal{R}_{\alpha}S^{-1}H-HS^{-1}\partial_{\alpha}S$. This includes all contributions available in the real-space Hamiltonian matrix when enabled. This method is more general but more expensive. The total current is written to OUT.{suffix}/current_tot_comm.txt.)"; item.default_value = "0"; item.unit = ""; - item.set_availability("basis_type==lcao and esolver_type==tddft"); + item.set_availability("basis_type in [pw, lcao] and esolver_type==tddft"); read_sync_int(input.out_current); this->add_item(item); } { Input_Item item("out_current_k"); item.annotation = "output current for each k"; - item.category = "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory"; + item.category = "Real-Time TDDFT (Common)"; item.type = "Boolean"; - item.description = R"(Controls whether LCAO RT-TDDFT current density is also resolved by spin and k-point. The total-current file is always written when out_current is 1 or 2. -* True: In addition to the total, out_current=1 writes OUT.{suffix}/current_s[spin]k[kpoint].txt; out_current=2 writes OUT.{suffix}/current_s[spin]k[kpoint]_comm.txt. Both use one-based spin and k-point numbers, with k-points numbered independently within each spin channel. Each row contains the one-based electronic-step index followed by $J_x$, $J_y$, and $J_z$ in atomic units. + item.description = R"(Controls whether RT-TDDFT current density is also resolved by spin and k-point. PW supports out_current=1; LCAO supports out_current=1 or 2. The total-current file is always written when current output is enabled. +* True: In addition to the total, out_current=1 writes OUT.{suffix}/current_s[spin]k[kpoint].txt; out_current=2 writes OUT.{suffix}/current_s[spin]k[kpoint]_comm.txt. Both use one-based spin and k-point numbers, with k-points numbered independently within each spin channel. Rows follow the step numbering, atomic units, precision, and append convention described in out_current. These weighted contributions sum to the total current. * False: Output only current_tot.txt for out_current=1 or current_tot_comm.txt for out_current=2.)"; item.default_value = "False"; item.unit = ""; - item.set_availability("basis_type==lcao and esolver_type==tddft and out_current>0"); + item.set_availability("basis_type in [pw, lcao] and esolver_type==tddft and out_current>0"); read_sync_bool(input.out_current_k); this->add_item(item); } { Input_Item item("out_efield"); item.annotation = "output dipole or not"; - item.category = "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory"; + item.category = "Real-Time TDDFT (Common)"; item.type = "Boolean"; item.description = R"(Controls time-dependent electric-field output. For each configured field, OUT.{suffix}/efield_[index].txt contains two columns: physical time in fs and the field value in V/Angstrom. The one-based field index follows the occurrence order shared by td_ttype and td_vext_dire, so fields assigned to the same direction remain in separate files. At initialization, a fresh calculation with md_restart=False truncates the files corresponding to the currently configured fields, whereas a calculation with md_restart=True preserves them and appends new samples. * True: Output electric-field values on active electronic steps. @@ -1730,9 +1697,9 @@ In molecular dynamics calculations, the output frequency is controlled by out_fr { Input_Item item("out_vecpot"); item.annotation = "output TDDFT vector potential or not"; - item.category = "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory"; + item.category = "Real-Time TDDFT (LCAO)"; item.type = "Boolean"; - item.description = R"(Controls Cartesian vector-potential output for LCAO RT-TDDFT. OUT.{suffix}/vector_pot.txt contains four columns: the one-based electronic-step index followed by $A_x$, $A_y$, and $A_z$ in atomic units. At initialization, a fresh calculation with md_restart=False truncates the file and writes a new header, whereas a calculation with md_restart=True preserves a nonempty existing file and appends new samples. If the restart output file is missing or empty, a new file with a header is created. + item.description = R"(Controls Cartesian vector-potential output for LCAO RT-TDDFT. OUT.{suffix}/vector_pot.txt contains four columns: the one-based electronic-step index followed by $A_x$, $A_y$, and $A_z$ in Hartree atomic units. The vector potential is the propagation value: the average of interval endpoints for an integrated electric field, or the supplied sample for init_vecpot_file=True. At initialization, a fresh calculation with md_restart=False truncates the file and writes a new header, whereas a calculation with md_restart=True preserves a nonempty existing file and appends new samples. If the restart output file is missing or empty, a new file with a header is created. * True: Write vector-potential samples on electronic propagation steps. * False: Do not output the vector potential.)"; item.default_value = "False"; diff --git a/source/source_io/module_parameter/read_inp_postproc.cpp b/source/source_io/module_parameter/read_inp_postproc.cpp index 0274895ff94..cca3d983a65 100644 --- a/source/source_io/module_parameter/read_inp_postproc.cpp +++ b/source/source_io/module_parameter/read_inp_postproc.cpp @@ -14,8 +14,8 @@ void ReadInput::item_postprocess() item.annotation = "delta energy for dos"; item.category = "Density of states"; item.type = "Real"; - item.description = "The step size in writing Density of States (DOS)"; - item.default_value = "0.01"; + item.description = "The step size in writing Density of States (DOS). The default value was changed from 0.01 to 0.03 eV."; + item.default_value = "0.03"; item.unit = "eV"; read_sync_double(input.dos_edelta_ev); this->add_item(item); diff --git a/source/source_io/module_parameter/read_inp_sys.cpp b/source/source_io/module_parameter/read_inp_sys.cpp index 7ae5ab45125..97a40123eb8 100644 --- a/source/source_io/module_parameter/read_inp_sys.cpp +++ b/source/source_io/module_parameter/read_inp_sys.cpp @@ -1219,6 +1219,10 @@ updates structures smoothly enough for second-order extrapolation.)"; item.default_value = "0"; read_sync_int(input.ndx); item.reset_value = [](const Input_Item& item, Parameter& para) { + // TODO: unlike the ecutrho/ecutwfc path, enabling double_grid here + // (and in the ndy/ndz blocks below) is not rejected for LCAO. It is + // currently harmless only because LCAO rejects USPP separately in + // uspp_support.cpp; revisit if LCAO gains USPP support. if (para.input.ndx > para.input.nx) { para.sys.double_grid = true; diff --git a/source/source_io/module_parameter/read_inp_tddft.cpp b/source/source_io/module_parameter/read_inp_tddft.cpp index 9f98bb43646..5949e16b71a 100644 --- a/source/source_io/module_parameter/read_inp_tddft.cpp +++ b/source/source_io/module_parameter/read_inp_tddft.cpp @@ -6,12 +6,36 @@ #include #include #include +#include +#include namespace ModuleIO { namespace { +int parse_td_integer(const Input_Item& item) +{ + std::size_t consumed = 0; + try + { + if (item.str_values.size() == 1) + { + const int value = std::stoi(item.str_values[0], &consumed); + if (consumed == item.str_values[0].size()) + { + return value; + } + } + } + catch (const std::exception&) + { + } + const std::string message = item.label + " must be a positive integer."; + ModuleBase::WARNING_QUIT("ReadInput", message); + return 0; +} + std::vector parse_supersine_steps(const Input_Item& item, const int default_step) { const std::vector tokens @@ -116,8 +140,19 @@ void check_td_efield_parameters(const Input_para& input) ModuleBase::WARNING_QUIT("ReadInput", "td_gauss_sigma must be nonzero."); } } + const std::array trape_step_names = {{"td_trape_t1", "td_trape_t2", "td_trape_t3"}}; for (std::size_t index = 0; index < field_counts[1]; ++index) { + const std::array steps = {{input.td_trape_t1[index], input.td_trape_t2[index], input.td_trape_t3[index]}}; + for (std::size_t boundary = 0; boundary < steps.size(); ++boundary) + { + if (steps[boundary] < 0.0 || steps[boundary] != std::floor(steps[boundary])) + { + ModuleBase::WARNING_QUIT("ReadInput", + std::string(trape_step_names[boundary]) + " for trapezoid field " + + std::to_string(index + 1) + " must be a nonnegative integer value."); + } + } if (input.td_trape_t1[index] > input.td_trape_t2[index] || input.td_trape_t2[index] > input.td_trape_t3[index]) { @@ -155,10 +190,207 @@ void ReadInput::item_rt_tddft() // NOTE: The order of add_item() calls below determines the parameter order // in the generated documentation (docs/advanced/input_files/input-main.md). // Please preserve this ordering when adding new parameters. + { + Input_Item item("td_orthonormal"); + item.annotation = "orthonormalization method for PW real-time propagation"; + item.category = "Real-Time TDDFT (PW)"; + item.type = "String"; + item.description = R"(Orthonormalize each propagated wavefunction before constructing its density. With the wavefunctions as columns of $\boldsymbol{\Psi}$, define the Gram matrix $\boldsymbol{S}=\boldsymbol{\Psi}^{\dagger}\boldsymbol{\Psi}$ and apply $\boldsymbol{\Psi}\leftarrow\boldsymbol{\Psi}\boldsymbol{C}$. +* `cholesky`: Cholesky orthonormalization. Factor $\boldsymbol{S}=\boldsymbol{R}^{\dagger}\boldsymbol{R}$ with upper-triangular $\boldsymbol{R}$ and use $\boldsymbol{C}=\boldsymbol{R}^{-1}$. +* `lowdin`: Löwdin symmetric orthonormalization. Compute $\boldsymbol{C}=\boldsymbol{S}^{-1/2}$ by eigendecomposition. +* `newton_schulz`: Newton-Schulz iteration for a Gram matrix close to the identity. Starting from $\boldsymbol{C}_0=\boldsymbol{I}$, approximate $\boldsymbol{S}^{-1/2}$ using $\boldsymbol{C}_{j+1}=\boldsymbol{C}_j(3\boldsymbol{I}-\boldsymbol{S}\boldsymbol{C}_j^2)/2$. +* `none`: Disable orthonormalization. + +[NOTE] Orthonormalization is applied after the Crank-Nicolson linear solve and uses an orthogonality tolerance independent of `lin_thr`. With orthonormalization enabled, the propagated Gram matrix must differ from the identity by at most `1e-6` (single precision) or `1e-12` (double precision) per element; unsuccessful corrections stop the calculation. Wavefunctions within this tolerance are left unchanged. Nonfinite or nonpositive orbital norms are rejected even with `none`.)"; + item.default_value = "cholesky"; + item.unit = ""; + item.set_availability("basis_type==pw and esolver_type==tddft"); + read_sync_string(input.td_orthonormal); + item.check_value = [](const Input_Item& item, const Parameter& para) { + const std::string& method = para.inp.td_orthonormal; + if (method != "none" && method != "cholesky" && method != "lowdin" && method != "newton_schulz") + { + ModuleBase::WARNING_QUIT("ReadInput", "td_orthonormal must be cholesky, lowdin, newton_schulz or none."); + } + }; + this->add_item(item); + } + { + Input_Item item("td_out_stat"); + item.annotation = "output PW real-time conservation diagnostics"; + item.category = "Real-Time TDDFT (PW)"; + item.type = "Boolean"; + item.description = R"(Write wavefunction and density electron counts, their changes from the initial state, and orthogonality errors to the running log after each electronic evolution step. Electron counts describe the end-of-step state; orthogonality errors before and after correction are separate maxima over all SCF iterations and k points within that step. Orthogonality errors are omitted when `td_orthonormal=none`. + +[NOTE] Enabling this output adds density integration and diagnostic communication.)"; + item.default_value = "false"; + item.unit = ""; + item.set_availability("basis_type==pw and esolver_type==tddft"); + read_sync_bool(input.td_out_stat); + this->add_item(item); + } + { + Input_Item item("lin_solver"); + item.annotation = "linear solver for real-time propagation"; + item.category = "Real-Time TDDFT (PW)"; + item.type = "String"; + item.description = R"(Iterative linear solver used for PW real-time propagation. +* `bicgstab`: Biconjugate gradient stabilized (BiCGSTAB) method. +* `cgs`: Conjugate gradient squared (CGS) method. +* `gmres`: Restarted generalized minimal residual (GMRES) method, controlled by `lin_gmres_restart`. + +The initial ground-state diagonalization is controlled by `ks_solver`.)"; + item.default_value = "gmres"; + item.unit = ""; + item.set_availability("basis_type==pw and esolver_type==tddft"); + read_sync_string(input.lin_solver); + item.check_value = [](const Input_Item&, const Parameter& para) { + if (para.inp.lin_solver != "bicgstab" && para.inp.lin_solver != "cgs" && para.inp.lin_solver != "gmres") + { + ModuleBase::WARNING_QUIT("ReadInput", "lin_solver must be bicgstab, cgs or gmres."); + } + }; + this->add_item(item); + } + { + Input_Item item("lin_precond"); + item.annotation = "right preconditioner for PW real-time propagation"; + item.category = "Real-Time TDDFT (PW)"; + item.type = "String"; + item.description = R"(Right preconditioner used by PW real-time linear solvers. The Crank-Nicolson operator is $\boldsymbol{L}=\boldsymbol{I}+\mathrm{i}\Delta t\,\boldsymbol{H}/2$, with the propagation Hamiltonian $\boldsymbol{H}$ in Hartree and time step $\Delta t$ in atomic units. Below, $\boldsymbol{M}^{-1}$ denotes the inverse preconditioner and $\boldsymbol{D}$ the kinetic diagonal inverse. +* `kinetic`: Apply $\boldsymbol{M}^{-1}=\boldsymbol{D}$, with diagonal entries $D_{\boldsymbol{G}\boldsymbol{G}}=(1+\mathrm{i}\Delta t\,T_{\boldsymbol{G}}/2)^{-1}$, where $T_{\boldsymbol{G}}=\lVert\boldsymbol{k}+\boldsymbol{G}+\boldsymbol{A}_{\mathrm{mid}}\rVert^2/2$ is the kinetic energy in Hartree. In the velocity gauge, $\boldsymbol{A}_{\mathrm{mid}}=(\boldsymbol{A}_n+\boldsymbol{A}_{n+1})/2$ is the propagation vector potential in Hartree atomic units; in the length gauge, set $\boldsymbol{A}_{\mathrm{mid}}=\boldsymbol{0}$. +* `kinetic_recycle`: Apply $\boldsymbol{M}^{-1}=\boldsymbol{D}+(\boldsymbol{Z}_{\mathrm{hist}}-\boldsymbol{D}\boldsymbol{W}_{\mathrm{hist}})\boldsymbol{W}_{\mathrm{hist}}^{\dagger}$, where $\boldsymbol{Z}_{\mathrm{hist}}$ and $\boldsymbol{W}_{\mathrm{hist}}$ are paired response directions and approximate operator images from the preceding successful solve at the same k point, with $\boldsymbol{W}_{\mathrm{hist}}^{\dagger}\boldsymbol{W}_{\mathrm{hist}}\approx\boldsymbol{I}$. The images belong to the historical solve, not necessarily the current $\boldsymbol{L}$. Falls back to `kinetic` when no reliable history is available. +* `kinetic_subspace`: Apply $\boldsymbol{M}^{-1}=\boldsymbol{D}+(\boldsymbol{U}-\boldsymbol{D}\boldsymbol{L}\boldsymbol{U})(\boldsymbol{U}^{\dagger}\boldsymbol{L}\boldsymbol{U})^{-1}\boldsymbol{U}^{\dagger}$, where the columns of $\boldsymbol{U}$ are the previous-time wavefunctions defining the coarse subspace. +* `none`: Disable preconditioning. + +Preconditioning changes the convergence rate, while `lin_thr` still controls the residual of the original equation.)"; + item.default_value = "kinetic_recycle"; + item.unit = ""; + item.set_availability("basis_type==pw and esolver_type==tddft"); + read_sync_string(input.lin_precond); + item.check_value = [](const Input_Item&, const Parameter& para) { + if (para.inp.lin_precond != "kinetic" && para.inp.lin_precond != "none" + && para.inp.lin_precond != "kinetic_recycle" && para.inp.lin_precond != "kinetic_subspace") + { + ModuleBase::WARNING_QUIT("ReadInput", "lin_precond must be none, kinetic, kinetic_recycle or kinetic_subspace."); + } + }; + this->add_item(item); + } + { + Input_Item item("lin_thr"); + item.annotation = "residual tolerance of the linear solver"; + item.category = "Real-Time TDDFT (PW)"; + item.type = "Real"; + item.description = R"(Nonnegative finite residual tolerance for each band in a PW real-time Crank-Nicolson solve $\boldsymbol{L}\boldsymbol{x}=\boldsymbol{b}$, with residual $\boldsymbol{r}=\boldsymbol{b}-\boldsymbol{L}\boldsymbol{x}$. A value of 0 selects $\tau=\max(10^{-10},100\epsilon)$, where $\epsilon$ is machine epsilon for the wavefunction precision. This gives a tolerance of approximately $1.19209\times10^{-5}$ in single precision and $10^{-10}$ in double precision. A positive value specifies $\tau$ directly. +* `bicgstab` and `gmres`: Require $\lVert\boldsymbol{r}\rVert\leqslant\tau\max(1,\lVert\boldsymbol{b}\rVert)$. +* `cgs`: Require $\lVert\boldsymbol{r}\rVert\leqslant\tau\lVert\boldsymbol{b}\rVert$ for nonzero $\boldsymbol{b}$, or $\lVert\boldsymbol{r}\rVert\leqslant\tau$ for zero $\boldsymbol{b}$. + +All methods check the residual at the end of the linear solve, before orthonormalization. GMRES can use explicit residual reconstruction with periodic independent checks when `lin_reconstruct` is enabled. Subsequent orthonormalization uses a separate orthogonality tolerance and does not guarantee that the corrected wavefunctions satisfy the same linear residual tolerance.)"; + item.default_value = "0"; + item.unit = ""; + item.set_availability("basis_type==pw and esolver_type==tddft"); + item.read_value = [](const Input_Item& item, Parameter& para) { + std::size_t consumed = 0; + try + { + if (item.str_values.size() == 1) + { + para.input.lin_thr = std::stod(item.str_values[0], &consumed); + if (consumed == item.str_values[0].size()) + { + return; + } + } + } + catch (const std::exception&) + { + } + ModuleBase::WARNING_QUIT("ReadInput", "lin_thr must be a nonnegative finite real number."); + }; + sync_double(input.lin_thr); + item.check_value = [](const Input_Item&, const Parameter& para) { + if (!std::isfinite(para.inp.lin_thr) || para.inp.lin_thr < 0.0) + { + ModuleBase::WARNING_QUIT("ReadInput", "lin_thr must be a nonnegative finite real number."); + } + }; + this->add_item(item); + } + { + Input_Item item("lin_maxiter"); + item.annotation = "maximum iterations per linear solve"; + item.category = "Real-Time TDDFT (PW)"; + item.type = "Integer"; + item.description = "Positive maximum number of iterations for each PW real-time linear solve. Failure to converge stops the calculation. The number of self-consistency iterations is controlled separately by `scf_nmax`."; + item.default_value = "500"; + item.unit = ""; + item.set_availability("basis_type==pw and esolver_type==tddft"); + item.read_value = [](const Input_Item& item, Parameter& para) { + para.input.lin_maxiter = parse_td_integer(item); + }; + sync_int(input.lin_maxiter); + item.check_value = [](const Input_Item&, const Parameter& para) { + if (para.inp.lin_maxiter <= 0) + { + ModuleBase::WARNING_QUIT("ReadInput", "lin_maxiter must be a positive integer."); + } + }; + this->add_item(item); + } + { + Input_Item item("lin_gmres_restart"); + item.annotation = "Arnoldi dimension of restarted GMRES"; + item.category = "Real-Time TDDFT (PW)"; + item.type = "Integer"; + item.description = "Positive maximum Arnoldi dimension per GMRES cycle. `lin_maxiter` limits the total iterations across all cycles."; + item.default_value = "20"; + item.unit = ""; + item.set_availability("basis_type==pw and esolver_type==tddft and lin_solver==gmres"); + item.read_value = [](const Input_Item& item, Parameter& para) { + para.input.lin_gmres_restart = parse_td_integer(item); + }; + sync_int(input.lin_gmres_restart); + item.check_value = [](const Input_Item&, const Parameter& para) { + if (para.inp.lin_gmres_restart <= 0) + { + ModuleBase::WARNING_QUIT("ReadInput", "lin_gmres_restart must be positive."); + } + }; + this->add_item(item); + } + { + Input_Item item("td_cn_init"); + item.annotation = "CN subspace initial guess and residual reuse"; + item.category = "Real-Time TDDFT (PW)"; + item.type = "Boolean"; + item.description = R"(Project the Crank-Nicolson equation $\boldsymbol{L}\boldsymbol{x}=\boldsymbol{b}$ onto the previous-time wavefunction subspace to initialize the first propagation solve of each time step. With those wavefunctions as the columns of $\boldsymbol{U}$, the projected initial guess is $\boldsymbol{x}_0=\boldsymbol{U}(\boldsymbol{U}^{\dagger}\boldsymbol{L}\boldsymbol{U})^{-1}\boldsymbol{U}^{\dagger}\boldsymbol{b}$. Evaluate and reuse the initial residual $\boldsymbol{r}_0=\boldsymbol{b}-\boldsymbol{L}\boldsymbol{x}_0$ using the stored subspace operator images $\boldsymbol{L}\boldsymbol{U}$, without an additional Hamiltonian application. Available with all linear solvers and preconditioners; later self-consistency iterations retain their current wavefunction guess.)"; + item.default_value = "true"; + item.unit = ""; + item.set_availability("basis_type==pw and esolver_type==tddft"); + read_sync_bool(input.td_cn_init); + this->add_item(item); + } + { + Input_Item item("lin_reconstruct"); + item.annotation = "explicit GMRES residual reconstruction"; + item.category = "Real-Time TDDFT (PW)"; + item.type = "Boolean"; + item.description = R"(Reconstruct the GMRES residual for $\boldsymbol{L}\boldsymbol{x}=\boldsymbol{b}$ as $\boldsymbol{r}=\boldsymbol{r}_0-(\boldsymbol{L}\boldsymbol{Z})\boldsymbol{y}$ for the update $\boldsymbol{x}=\boldsymbol{x}_0+\boldsymbol{Z}\boldsymbol{y}$, where $\boldsymbol{r}_0=\boldsymbol{b}-\boldsymbol{L}\boldsymbol{x}_0$ is the initial residual, $\boldsymbol{Z}$ contains the current preconditioned search directions, and $\boldsymbol{y}$ contains their update coefficients. Reusing the stored, unmodified operator images $\boldsymbol{L}\boldsymbol{Z}$ reduces Hamiltonian applications. + +Use an internal tolerance of 0.8 times the effective `lin_thr` and independently verify the first solve at each k point and every 16 solves thereafter, with additional independent checks when needed. Failed reconstruction checks trigger a true-residual restart within `lin_maxiter`. + +Only effective for `lin_solver=gmres`; ignored otherwise.)"; + item.default_value = "true"; + item.unit = ""; + item.set_availability("basis_type==pw and esolver_type==tddft and lin_solver==gmres"); + read_sync_bool(input.lin_reconstruct); + this->add_item(item); + } { Input_Item item("estep_per_md"); item.annotation = "steps of force change"; - item.category = "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory"; + item.category = "Real-Time TDDFT (Common)"; item.type = "Integer"; item.description = "The number of electronic propagation steps between two ionic steps."; item.default_value = "1"; @@ -170,7 +402,7 @@ void ReadInput::item_rt_tddft() { Input_Item item("td_dt"); item.annotation = "time step for evolving wavefunction"; - item.category = "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory"; + item.category = "Real-Time TDDFT (Common)"; item.type = "Real"; item.description = R"(The time step used for electronic propagation. If td_dt is not specified, it is set to md_dt / estep_per_md. If td_dt is specified explicitly, md_dt is reset to td_dt * estep_per_md.)"; item.default_value = "md_dt / estep_per_md"; @@ -188,33 +420,35 @@ void ReadInput::item_rt_tddft() { Input_Item item("td_edm"); item.annotation = "the method to calculate the energy density matrix"; - item.category = "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory"; + item.category = "Real-Time TDDFT (LCAO)"; item.type = "Integer"; item.description = R"(Method used to calculate the energy-density matrix for the overlap contribution to forces in LCAO RT-TDDFT. * 0: Use $\mathrm{EDM}_{\boldsymbol{k}}=\frac{1}{2}\left(S_{\boldsymbol{k}}^{-1}H_{\boldsymbol{k}}\rho_{\boldsymbol{k}}+\rho_{\boldsymbol{k}}H_{\boldsymbol{k}}S_{\boldsymbol{k}}^{-1}\right)$. * 1: Use the ground-state eigenvalue-weighted expression $\mathrm{EDM}_{\mu\nu,\boldsymbol{k}}=\sum_i w_{i\boldsymbol{k}}\epsilon_{i\boldsymbol{k}}C_{\mu i,\boldsymbol{k}}C_{\nu i,\boldsymbol{k}}^*$. This expression is deprecated for RT-TDDFT and is generally not valid when the propagated wave functions are not Hamiltonian eigenstates.)"; item.default_value = "0"; item.unit = ""; + item.set_availability("basis_type==lcao and esolver_type==tddft"); read_sync_int(input.td_edm); this->add_item(item); } { Input_Item item("td_print_eij"); item.annotation = "print eij or not"; - item.category = "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory"; + item.category = "Real-Time TDDFT (LCAO)"; item.type = "Real"; item.description = R"(Controls output of the propagated-state Hamiltonian matrix elements $E_{ij}=\Braket{\psi_i | \hat{H} | \psi_j}$ to the running log. The printed band indices $i$ and $j$ are one-based global indices. Both the threshold and the printed matrix elements are in Ry. * $\lt 0$: Disable the output. * $\geqslant 0$: Print an element when either $\left|\operatorname{Re}E_{ij}\right|$ or $\left|\operatorname{Im}E_{ij}\right|$ is greater than or equal to td_print_eij.)"; item.default_value = "-1"; item.unit = "Ry"; + item.set_availability("basis_type==lcao and esolver_type==tddft"); read_sync_double(input.td_print_eij); this->add_item(item); } { Input_Item item("td_propagator"); item.annotation = "method of propagator"; - item.category = "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory"; + item.category = "Real-Time TDDFT (LCAO)"; item.type = "Integer"; item.description = R"(Method used to propagate the electronic states in a nonorthogonal LCAO basis. The formulas below use Hartree atomic units, with $S$, $H$, and $\Delta t=\mathtt{td\_dt}$ evaluated as required by each approximation. * 0: Crank-Nicolson through an explicitly constructed evolution matrix, $U=\left[S+\mathrm{i}H\Delta t/2\right]^{-1}\left[S-\mathrm{i}H\Delta t/2\right]$. @@ -225,13 +459,14 @@ void ReadInput::item_rt_tddft() [NOTE] GPU execution currently supports only method 0 in both single-GPU and multi-GPU solver configurations. CPU execution supports methods 0 through 3.)"; item.default_value = "0"; item.unit = ""; + item.set_availability("basis_type==lcao and esolver_type==tddft"); read_sync_int(input.propagator); this->add_item(item); } { Input_Item item("td_vext"); item.annotation = "add extern potential or not"; - item.category = "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory"; + item.category = "Real-Time TDDFT (Common)"; item.type = "Boolean"; item.description = R"(Controls whether a time-dependent external electric field is applied. * True: Add a laser-material interaction (external electric field). @@ -244,7 +479,7 @@ void ReadInput::item_rt_tddft() { Input_Item item("td_vext_dire"); item.annotation = "extern potential direction"; - item.category = "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory"; + item.category = "Real-Time TDDFT (Common)"; item.type = "Vector of Integer"; item.description = R"(Specifies one absolute Cartesian direction for each external electric field when td_vext is enabled. Unlike the ground-state efield_dir parameter, these directions are not defined by lattice or reciprocal-lattice vectors. The number of values must equal that of td_ttype, and repeated directions are allowed; fields assigned to the same direction are added. For example, td_vext_dire 1 2 applies one field along Cartesian x and one along Cartesian y. * 1: The external field direction is along the x-axis. @@ -261,12 +496,12 @@ void ReadInput::item_rt_tddft() { Input_Item item("td_stype"); item.annotation = "type of electric field in space domain"; - item.category = "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory"; + item.category = "Real-Time TDDFT (Common)"; item.type = "Integer"; item.description = R"(Type of electric field in the space domain, i.e. the gauge of the electric field. -* 0: Length gauge. -* 1: Velocity gauge. -* 2: Hybrid gauge. See J. Chem. Theory Comput. 2025, 21, 3335-3341 for more information.)"; +* 0: Length gauge, available for PW and LCAO. +* 1: Velocity gauge, available for PW and LCAO. +* 2: Hybrid gauge, available only for LCAO. See J. Chem. Theory Comput. 2025, 21, 3335-3341 for more information.)"; item.default_value = "0"; item.unit = ""; read_sync_int(input.td_stype); @@ -275,7 +510,7 @@ void ReadInput::item_rt_tddft() { Input_Item item("td_ttype"); item.annotation = "type of electric field in time domain"; - item.category = "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory"; + item.category = "Real-Time TDDFT (Common)"; item.type = "Vector of Integer"; item.description = R"(Specifies one time-domain type for each external electric field. Its number of values must equal that of td_vext_dire. Parameters belonging to each type must provide exactly one value for every occurrence of that type, in occurrence order; fields with a repeated direction are added. @@ -299,7 +534,7 @@ In the velocity and hybrid gauges, ABACUS obtains the vector potential actually { Input_Item item("td_tstart"); item.annotation = " number of steps where electric field starts"; - item.category = "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory"; + item.category = "Real-Time TDDFT (Common)"; item.type = "Integer"; item.description = R"(First electronic step at which the time-dependent electric field is active. The interval from td_tstart through td_tend includes both endpoints. On each active step $n$, the velocity and hybrid gauges integrate the field over $[n\Delta t,(n+1)\Delta t]$, where $\Delta t=\mathtt{td\_dt}$.)"; item.default_value = "1"; @@ -310,7 +545,7 @@ In the velocity and hybrid gauges, ABACUS obtains the vector potential actually { Input_Item item("td_tend"); item.annotation = "number of steps where electric field ends"; - item.category = "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory"; + item.category = "Real-Time TDDFT (Common)"; item.type = "Integer"; item.description = R"(Last electronic step at which the time-dependent electric field is active. The interval from td_tstart through td_tend includes both endpoints. On each active step $n$, the velocity and hybrid gauges integrate the field over $[n\Delta t,(n+1)\Delta t]$, where $\Delta t=\mathtt{td\_dt}$.)"; item.default_value = "1000"; @@ -321,7 +556,7 @@ In the velocity and hybrid gauges, ABACUS obtains the vector potential actually { Input_Item item("td_lcut1"); item.annotation = "cut1 of interval in length gauge"; - item.category = "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory"; + item.category = "Real-Time TDDFT (Common)"; item.type = "Real"; item.description = R"(Lower fractional-coordinate cutoff for the periodic spatial modulation used in the length gauge. Let $c_1=\mathtt{td\_lcut1}$, $c_2=\mathtt{td\_lcut2}$, $D=c_2-c_1$, and $G=c_1+1-c_2$. For a fractional coordinate $x$, the field factor is $\eta(x)=1$ when $c_1\leqslant x\lt c_2$ and $\eta(x)=-D/G$ elsewhere. The reversed outer interval makes the potential periodic and continuous and gives the field zero cell average.)"; item.default_value = "0.05"; @@ -332,7 +567,7 @@ In the velocity and hybrid gauges, ABACUS obtains the vector potential actually { Input_Item item("td_lcut2"); item.annotation = "cut2 of interval in length gauge"; - item.category = "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory"; + item.category = "Real-Time TDDFT (Common)"; item.type = "Real"; item.description = R"(Upper fractional-coordinate cutoff for the periodic spatial modulation used in the length gauge. Let $c_1=\mathtt{td\_lcut1}$, $c_2=\mathtt{td\_lcut2}$, $D=c_2-c_1$, and $G=c_1+1-c_2$. For a fractional coordinate $x$, the field factor is $\eta(x)=1$ when $c_1\leqslant x\lt c_2$ and $\eta(x)=-D/G$ elsewhere. The reversed outer interval makes the potential periodic and continuous and gives the field zero cell average.)"; item.default_value = "0.95"; @@ -343,7 +578,7 @@ In the velocity and hybrid gauges, ABACUS obtains the vector potential actually { Input_Item item("td_gauss_freq"); item.annotation = "frequency (freq) of Gauss type electric field"; - item.category = "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory"; + item.category = "Real-Time TDDFT (Common)"; item.type = "Vector of Real"; item.description = R"(Ordinary frequency $f$ in the Gaussian-pulse formula, with $\omega=2\pi f$. Supply exactly one value for each td_ttype 0 occurrence, in occurrence order.)"; item.default_value = "22.13"; @@ -358,7 +593,7 @@ In the velocity and hybrid gauges, ABACUS obtains the vector potential actually { Input_Item item("td_gauss_phase"); item.annotation = "phase of Gauss type electric field"; - item.category = "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory"; + item.category = "Real-Time TDDFT (Common)"; item.type = "Vector of Real"; item.description = R"(Carrier phase $\varphi$ in the Gaussian-pulse formula. Supply exactly one value for each td_ttype 0 occurrence, in occurrence order.)"; item.default_value = "0.0"; @@ -373,7 +608,7 @@ In the velocity and hybrid gauges, ABACUS obtains the vector potential actually { Input_Item item("td_gauss_sigma"); item.annotation = "sigma of Gauss type electric field"; - item.category = "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory"; + item.category = "Real-Time TDDFT (Common)"; item.type = "Vector of Real"; item.description = R"(Nonzero standard deviation $\sigma$ of the Gaussian envelope. Supply exactly one value for each td_ttype 0 occurrence, in occurrence order.)"; item.default_value = "30.0"; @@ -388,7 +623,7 @@ In the velocity and hybrid gauges, ABACUS obtains the vector potential actually { Input_Item item("td_gauss_t0"); item.annotation = "step number of time center (t0) of Gauss type electric field"; - item.category = "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory"; + item.category = "Real-Time TDDFT (Common)"; item.type = "Vector of Real"; item.description = R"(Electronic-step position of the Gaussian center, which defines $t_0=\mathtt{td\_gauss\_t0}\Delta t$. Supply exactly one value for each td_ttype 0 occurrence, in occurrence order.)"; item.default_value = "100"; @@ -403,7 +638,7 @@ In the velocity and hybrid gauges, ABACUS obtains the vector potential actually { Input_Item item("td_gauss_amp"); item.annotation = "amplitude of Gauss type electric field"; - item.category = "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory"; + item.category = "Real-Time TDDFT (Common)"; item.type = "Vector of Real"; item.description = R"(Electric-field scale $E_0$ in the Gaussian-pulse formula. Supply exactly one value for each td_ttype 0 occurrence, in occurrence order.)"; item.default_value = "0.25"; @@ -418,7 +653,7 @@ In the velocity and hybrid gauges, ABACUS obtains the vector potential actually { Input_Item item("td_trape_freq"); item.annotation = "frequency of Trapezoid type electric field"; - item.category = "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory"; + item.category = "Real-Time TDDFT (Common)"; item.type = "Vector of Real"; item.description = R"(Ordinary carrier frequency $f$ in the trapezoid-pulse formula, with $\omega=2\pi f$. Supply exactly one value for each td_ttype 1 occurrence, in occurrence order.)"; item.default_value = "1.60"; @@ -433,7 +668,7 @@ In the velocity and hybrid gauges, ABACUS obtains the vector potential actually { Input_Item item("td_trape_phase"); item.annotation = "phase of Trapezoid type electric field"; - item.category = "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory"; + item.category = "Real-Time TDDFT (Common)"; item.type = "Vector of Real"; item.description = R"(Carrier phase $\varphi$ in the trapezoid-pulse formula. Supply exactly one value for each td_ttype 1 occurrence, in occurrence order.)"; item.default_value = "0.0"; @@ -448,9 +683,9 @@ In the velocity and hybrid gauges, ABACUS obtains the vector potential actually { Input_Item item("td_trape_t1"); item.annotation = "t1 of Trapezoid type electric field"; - item.category = "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory"; + item.category = "Real-Time TDDFT (Common)"; item.type = "Vector of Real"; - item.description = R"(Electronic step defining the end of the linear rise, $t_1=\mathtt{td\_trape\_t1}\Delta t$. Each field must satisfy td_trape_t1 <= td_trape_t2 <= td_trape_t3. Supply exactly one value for each td_ttype 1 occurrence, in occurrence order.)"; + item.description = R"(Electronic step defining the end of the linear rise, $t_1=\mathtt{td\_trape\_t1}\Delta t$. The value must be a nonnegative integer; integer-valued real inputs such as 2.0 are accepted. Each field must satisfy td_trape_t1 <= td_trape_t2 <= td_trape_t3. Supply exactly one value for each td_ttype 1 occurrence, in occurrence order.)"; item.default_value = "1875"; item.unit = ""; item.set_availability("td_ttype contains 1"); @@ -463,9 +698,9 @@ In the velocity and hybrid gauges, ABACUS obtains the vector potential actually { Input_Item item("td_trape_t2"); item.annotation = "t2 of Trapezoid type electric field"; - item.category = "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory"; + item.category = "Real-Time TDDFT (Common)"; item.type = "Vector of Real"; - item.description = R"(Electronic step defining the end of the plateau, $t_2=\mathtt{td\_trape\_t2}\Delta t$. Each field must satisfy td_trape_t1 <= td_trape_t2 <= td_trape_t3. Supply exactly one value for each td_ttype 1 occurrence, in occurrence order.)"; + item.description = R"(Electronic step defining the end of the plateau, $t_2=\mathtt{td\_trape\_t2}\Delta t$. The value must be a nonnegative integer; integer-valued real inputs such as 2.0 are accepted. Each field must satisfy td_trape_t1 <= td_trape_t2 <= td_trape_t3. Supply exactly one value for each td_ttype 1 occurrence, in occurrence order.)"; item.default_value = "5625"; item.unit = ""; item.set_availability("td_ttype contains 1"); @@ -478,9 +713,9 @@ In the velocity and hybrid gauges, ABACUS obtains the vector potential actually { Input_Item item("td_trape_t3"); item.annotation = "t3 of Trapezoid type electric field"; - item.category = "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory"; + item.category = "Real-Time TDDFT (Common)"; item.type = "Vector of Real"; - item.description = R"(Electronic step defining the end of the linear fall, $t_3=\mathtt{td\_trape\_t3}\Delta t$. Each field must satisfy td_trape_t1 <= td_trape_t2 <= td_trape_t3. Supply exactly one value for each td_ttype 1 occurrence, in occurrence order.)"; + item.description = R"(Electronic step defining the end of the linear fall, $t_3=\mathtt{td\_trape\_t3}\Delta t$. The value must be a nonnegative integer; integer-valued real inputs such as 2.0 are accepted. Each field must satisfy td_trape_t1 <= td_trape_t2 <= td_trape_t3. Supply exactly one value for each td_ttype 1 occurrence, in occurrence order.)"; item.default_value = "7500"; item.unit = ""; item.set_availability("td_ttype contains 1"); @@ -493,7 +728,7 @@ In the velocity and hybrid gauges, ABACUS obtains the vector potential actually { Input_Item item("td_trape_amp"); item.annotation = "amplitude of Trapezoid type electric field"; - item.category = "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory"; + item.category = "Real-Time TDDFT (Common)"; item.type = "Vector of Real"; item.description = R"(Electric-field scale $E_0$ in the trapezoid-pulse formula. Supply exactly one value for each td_ttype 1 occurrence, in occurrence order.)"; item.default_value = "2.74"; @@ -508,7 +743,7 @@ In the velocity and hybrid gauges, ABACUS obtains the vector potential actually { Input_Item item("td_trigo_freq1"); item.annotation = "frequency 1 of Trigonometric type electric field"; - item.category = "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory"; + item.category = "Real-Time TDDFT (Common)"; item.type = "Vector of Real"; item.description = R"(First ordinary frequency $f_1$ in the trigonometric-pulse formula, with $\omega_1=2\pi f_1$. Supply exactly one value for each td_ttype 2 occurrence, in occurrence order.)"; item.default_value = "1.164656"; @@ -523,7 +758,7 @@ In the velocity and hybrid gauges, ABACUS obtains the vector potential actually { Input_Item item("td_trigo_freq2"); item.annotation = "frequency 2 of Trigonometric type electric field"; - item.category = "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory"; + item.category = "Real-Time TDDFT (Common)"; item.type = "Vector of Real"; item.description = R"(Second ordinary frequency $f_2$ in the trigonometric-pulse formula, with $\omega_2=2\pi f_2$. Supply exactly one value for each td_ttype 2 occurrence, in occurrence order.)"; item.default_value = "0.029116"; @@ -538,7 +773,7 @@ In the velocity and hybrid gauges, ABACUS obtains the vector potential actually { Input_Item item("td_trigo_phase1"); item.annotation = "phase 1 of Trigonometric type electric field"; - item.category = "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory"; + item.category = "Real-Time TDDFT (Common)"; item.type = "Vector of Real"; item.description = R"(Carrier phase $\varphi_1$ in the cosine factor of the trigonometric-pulse formula. Supply exactly one value for each td_ttype 2 occurrence, in occurrence order.)"; item.default_value = "0.0"; @@ -553,7 +788,7 @@ In the velocity and hybrid gauges, ABACUS obtains the vector potential actually { Input_Item item("td_trigo_phase2"); item.annotation = "phase 2 of Trigonometric type electric field"; - item.category = "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory"; + item.category = "Real-Time TDDFT (Common)"; item.type = "Vector of Real"; item.description = R"(Envelope phase $\varphi_2$ in the sine-squared factor of the trigonometric-pulse formula. Supply exactly one value for each td_ttype 2 occurrence, in occurrence order.)"; item.default_value = "0.0"; @@ -568,7 +803,7 @@ In the velocity and hybrid gauges, ABACUS obtains the vector potential actually { Input_Item item("td_trigo_amp"); item.annotation = "amplitude of Trigonometric type electric field"; - item.category = "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory"; + item.category = "Real-Time TDDFT (Common)"; item.type = "Vector of Real"; item.description = R"(Electric-field scale $E_0$ in the trigonometric-pulse formula. Supply exactly one value for each td_ttype 2 occurrence, in occurrence order.)"; item.default_value = "2.74"; @@ -583,7 +818,7 @@ In the velocity and hybrid gauges, ABACUS obtains the vector potential actually { Input_Item item("td_heavi_t0"); item.annotation = "t0 of Heaviside type electric field"; - item.category = "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory"; + item.category = "Real-Time TDDFT (Common)"; item.type = "Vector of Real"; item.description = R"(Electronic switch step $n_0$ in the Heaviside-pulse definition. The field is $E_0$ for $n\lt n_0$ and zero for $n\geqslant n_0$. Supply exactly one value for each td_ttype 3 occurrence, in occurrence order.)"; item.default_value = "100"; @@ -598,7 +833,7 @@ In the velocity and hybrid gauges, ABACUS obtains the vector potential actually { Input_Item item("td_heavi_amp"); item.annotation = "amplitude of Heaviside type electric field"; - item.category = "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory"; + item.category = "Real-Time TDDFT (Common)"; item.type = "Vector of Real"; item.description = R"(Electric-field scale $E_0$ in the Heaviside-pulse definition. Supply exactly one value for each td_ttype 3 occurrence, in occurrence order.)"; item.default_value = "1.0"; @@ -613,7 +848,7 @@ In the velocity and hybrid gauges, ABACUS obtains the vector potential actually { Input_Item item("td_supsine_amp"); item.annotation = "carrier electric-field scale of the supersine pulse"; - item.category = "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory"; + item.category = "Real-Time TDDFT (Common)"; item.type = "Vector of Real"; item.description = R"(Carrier electric-field scale $E_0$ of each supersine pulse. This is not a normalization of the complete waveform maximum, because the envelope-derivative term also contributes. Supply exactly one value for each td_ttype 4 occurrence, in occurrence order.)"; item.default_value = "0.27"; @@ -628,7 +863,7 @@ In the velocity and hybrid gauges, ABACUS obtains the vector potential actually { Input_Item item("td_supsine_freq"); item.annotation = "carrier frequency of the supersine pulse"; - item.category = "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory"; + item.category = "Real-Time TDDFT (Common)"; item.type = "Vector of Real"; item.description = R"(Nonzero ordinary carrier frequency $f$ of each supersine pulse, with $\omega=2\pi f$. Supply exactly one value for each td_ttype 4 occurrence, in occurrence order.)"; item.default_value = "0.18737028625"; @@ -643,7 +878,7 @@ In the velocity and hybrid gauges, ABACUS obtains the vector potential actually { Input_Item item("td_supsine_phase"); item.annotation = "carrier phase at the center of the supersine pulse"; - item.category = "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory"; + item.category = "Real-Time TDDFT (Common)"; item.type = "Vector of Real"; item.description = R"(Electric-field carrier phase $\varphi$ at the center of each supersine envelope. A value of 0 places a cosine carrier maximum at the envelope center. Supply exactly one value for each td_ttype 4 occurrence, in occurrence order.)"; item.default_value = "0.0"; @@ -658,7 +893,7 @@ In the velocity and hybrid gauges, ABACUS obtains the vector potential actually { Input_Item item("td_supsine_sigma"); item.annotation = "shape parameter of the supersine envelope"; - item.category = "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory"; + item.category = "Real-Time TDDFT (Common)"; item.type = "Vector of Real"; item.description = R"(Dimensionless shape parameter $\sigma$ of each supersine envelope. It must satisfy $0\lt\sigma\lt\pi/2$ so that the electric field approaches zero at the pulse boundaries. Supply exactly one value for each td_ttype 4 occurrence, in occurrence order.)"; item.default_value = "0.75"; @@ -673,7 +908,7 @@ In the velocity and hybrid gauges, ABACUS obtains the vector potential actually { Input_Item item("td_supsine_tstart"); item.annotation = "start boundary step of the supersine pulse"; - item.category = "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory"; + item.category = "Real-Time TDDFT (Common)"; item.type = "Vector of String"; item.description = R"(Integer electronic step at the left, exactly zero boundary of each supersine pulse, defining $t_{\mathrm{s}}=\mathtt{td\_supsine\_tstart}\Delta t$. Supply exactly one integer or default token for each td_ttype 4 occurrence, in occurrence order; each default token inherits td_tstart. The complete pulse support must lie inside the inclusive global td_tstart to td_tend interval; hard truncation of a supersine pulse is rejected.)"; item.default_value = "default"; @@ -691,7 +926,7 @@ In the velocity and hybrid gauges, ABACUS obtains the vector potential actually { Input_Item item("td_supsine_tend"); item.annotation = "end boundary step of the supersine pulse"; - item.category = "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory"; + item.category = "Real-Time TDDFT (Common)"; item.type = "Vector of String"; item.description = R"(Integer electronic step at the right, exactly zero boundary of each supersine pulse, defining $t_{\mathrm{e}}=\mathtt{td\_supsine\_tend}\Delta t$. Supply exactly one integer or default token for each td_ttype 4 occurrence, in occurrence order; each default token inherits td_tend. The complete pulse support must lie inside the inclusive global td_tstart to td_tend interval; hard truncation of a supersine pulse is rejected.)"; item.default_value = "default"; @@ -709,24 +944,25 @@ In the velocity and hybrid gauges, ABACUS obtains the vector potential actually { Input_Item item("init_vecpot_file"); item.annotation = "init vector potential through file or not"; - item.category = "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory"; + item.category = "Real-Time TDDFT (LCAO)"; item.type = "Boolean"; item.description = R"(Selects the source of the Cartesian vector potential used by LCAO RT-TDDFT. -* True: Read vector_pot.txt from the calculation working directory. Each non-comment line must contain four columns: a conventionally one-based electronic-step label followed by $A_x$, $A_y$, and $A_z$ in atomic units. Rows are consumed sequentially; the first column is read as a label and is not used for lookup. If propagation continues beyond the available rows, the last row is reused. +* True: Read vector_pot.txt from the calculation working directory. Each non-comment line must contain four columns: a conventionally one-based electronic-step label followed by $A_x$, $A_y$, and $A_z$ in Hartree atomic units. These are propagation values, not endpoint samples. Rows are consumed sequentially; the first column is read as a label and is not used for lookup. If propagation continues beyond the available rows, the last row is reused. * False: Obtain the vector potential by integrating the configured electric field.)"; item.default_value = "False"; item.unit = ""; + item.set_availability("basis_type==lcao and esolver_type==tddft"); read_sync_bool(input.init_vecpot_file); this->add_item(item); } { Input_Item item("ocp"); item.annotation = "change occupation or not"; - item.category = "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory"; + item.category = "Real-Time TDDFT (Common)"; item.type = "Boolean"; - item.description = R"(Controls fixed band occupations. In calculations other than LCAO RT-TDDFT, fixed values are applied during electronic-state setup. In LCAO RT-TDDFT, the initial ground-state SCF determines occupations normally, and fixed values from ocp_set are applied during the subsequent real-time propagation steps. -* True: Use the fixed occupations specified by ocp_set during propagation. -* False: Keep the occupations determined by the initial SCF.)"; + item.description = R"(Controls fixed band occupations. In PW RT-TDDFT and calculations other than LCAO RT-TDDFT, fixed values from ocp_set are applied during electronic-state setup. PW real-time propagation preserves these initial occupations. In LCAO RT-TDDFT, the initial ground-state SCF determines occupations normally, and fixed values are applied during the subsequent real-time propagation steps. +* True: Use the fixed occupations specified by ocp_set at the stage described above. +* False: Determine occupations normally; real-time propagation preserves the initial SCF occupations.)"; item.default_value = "False"; item.unit = ""; read_sync_bool(input.ocp); @@ -735,9 +971,9 @@ In the velocity and hybrid gauges, ABACUS obtains the vector potential actually { Input_Item item("ocp_set"); item.annotation = "set occupation"; - item.category = "RT-TDDFT: Real-Time Time-Dependent Density Functional Theory"; + item.category = "Real-Time TDDFT (Common)"; item.type = "String"; - item.description = R"(Fixed occupation weights used when ocp is true. Values are assigned in band order for each k-point, following k-point order. In LCAO RT-TDDFT, the initial ground-state SCF uses its normally determined occupations, and this array is applied only during subsequent real-time propagation steps. The repetition syntax N*x expands to N copies of x. + item.description = R"(Fixed occupation weights used when ocp is true. Values are assigned in band order for each k-point, following k-point order. In PW RT-TDDFT and other calculations outside LCAO RT-TDDFT, this array is applied during electronic-state setup. PW propagation preserves these occupations. In LCAO RT-TDDFT, the initial ground-state SCF uses its normally determined occupations, and this array is applied only during subsequent real-time propagation steps. The repetition syntax N*x expands to N copies of x. * Example: 1 10*1 0 1 expands to 13 values, with the 12th value equal to 0 and all other values equal to 1. * After expansion, provide one block of nbands values for each k-point. If nspin is 2, provide all k-point blocks for spin up followed by all k-point blocks for spin down; otherwise, provide one block per k-point. * The sum of all weights must equal nelec; otherwise the calculation terminates with an error.)"; diff --git a/source/source_io/module_parameter/read_input.cpp b/source/source_io/module_parameter/read_input.cpp index a3d0359baaa..6257c6dbad3 100644 --- a/source/source_io/module_parameter/read_input.cpp +++ b/source/source_io/module_parameter/read_input.cpp @@ -98,40 +98,40 @@ bool ReadInput::check_mode = false; bool filter_nonascii_and_comment(std::ifstream& ifs, std::stringstream& out_ascii_stream) { - if (!ifs.is_open()) - { - if (!ifs) return false; + if (!ifs.is_open()) + { + if (!ifs) return false; } std::streampos old_pos = ifs.tellg(); ifs.clear(); ifs.seekg(0, std::ios::beg); - char c = '\0'; - while (ifs.get(c)) - { - // If comment start, skip until end of line (but keep the newline) - if (c == '#') - { - char d = '\0'; - bool newline_found = false; - while (ifs.get(d)) - { - if (d == '\n' || d == '\r') - { - // preserve line break in output - out_ascii_stream.put('\n'); - // If CRLF, consume the LF after CR (already wrote a single '\n') - if (d == '\r' && ifs.peek() == '\n') - { - ifs.get(d); // consume '\n' - } - newline_found = true; - break; - } + char c = '\0'; + while (ifs.get(c)) + { + // If comment start, skip until end of line (but keep the newline) + if (c == '#') + { + char d = '\0'; + bool newline_found = false; + while (ifs.get(d)) + { + if (d == '\n' || d == '\r') + { + // preserve line break in output + out_ascii_stream.put('\n'); + // If CRLF, consume the LF after CR (already wrote a single '\n') + if (d == '\r' && ifs.peek() == '\n') + { + ifs.get(d); // consume '\n' + } + newline_found = true; + break; + } } - if (!newline_found) - { + if (!newline_found) + { // reached EOF while skipping comment break; } @@ -139,15 +139,15 @@ bool filter_nonascii_and_comment(std::ifstream& ifs, } unsigned char uc = static_cast(c); - if (uc <= 0x7F) - { - // ASCII character + if (uc <= 0x7F) + { + // ASCII character out_ascii_stream.put(c); } - else - { - // replace non-ASCII with space character - out_ascii_stream.put(' '); + else + { + // replace non-ASCII with space character + out_ascii_stream.put(' '); } } @@ -329,21 +329,21 @@ void ReadInput::read_txt_input(Parameter& param, const std::string& filename) std::stringstream ascii_stream; - std::ifstream ifs(filename.c_str(), std::ios::in); + std::ifstream ifs(filename.c_str(), std::ios::in); - if (!ifs) - { - std::cout << " Can't find the INPUT file." << std::endl; - ModuleBase::WARNING_QUIT("Input::Init", "Error during readin parameters.", 1); - } + if (!ifs) + { + std::cout << " Can't find the INPUT file." << std::endl; + ModuleBase::WARNING_QUIT("Input::Init", "Error during readin parameters.", 1); + } - ifs.clear(); - ifs.seekg(0); + ifs.clear(); + ifs.seekg(0); - filter_nonascii_and_comment(ifs, ascii_stream); - ifs.clear(); + filter_nonascii_and_comment(ifs, ascii_stream); + ifs.clear(); - // file close after reading + // file close after reading int ierr = 0; ascii_stream.rdstate(); @@ -434,8 +434,8 @@ void ReadInput::read_txt_input(Parameter& param, const std::string& filename) { Input_Item* resetvalue_item = &(input_item.second); if (resetvalue_item->reset_value != nullptr) - { - resetvalue_item->reset_value(*resetvalue_item, param); + { + resetvalue_item->reset_value(*resetvalue_item, param); } } diff --git a/source/source_io/module_parameter/read_input_item_md.cpp b/source/source_io/module_parameter/read_input_item_md.cpp index cebe8d7d944..d5f6dc6bd67 100644 --- a/source/source_io/module_parameter/read_input_item_md.cpp +++ b/source/source_io/module_parameter/read_input_item_md.cpp @@ -310,6 +310,30 @@ Note: It is a system-dependent empirical parameter, ranging from 1/(40*md_dt) to item.default_value = "1.0"; item.unit = ""; read_sync_double(input.ref_cell_factor); + // Disable the reference cell feature for now, because the PW_Basis + // internal lat0/tpiba/G/GGT/omega members become stale when + // ref_cell_factor > 1, leading to wrong charge/energy integration + // (sum_rho, get_local_pp_energy, cal_delta_escf, makov_payne) + // in NPT and other variable-cell calculations. The reference cell + // mechanism leaks into external code (wfc IO, DFPT, OFDFT) in + // ways that are mathematically incorrect. + // TODO(liuyu): re-enable after PW_Basis is refactored to separate + // the reference-cell grid (FFT dims nx/ny/nz) from the physical-cell + // lattice quantities (lat0/tpiba/G/GGT/omega). Until then, refuse + // any non-1.0 value so users get a clear error instead of silently + // wrong results. + item.reset_value = [](const Input_Item& item, Parameter& para) { + if (para.input.ref_cell_factor != 1.0) + { + ModuleBase::WARNING_QUIT( + "ReadInput", + "ref_cell_factor != 1.0 is currently disabled because the " + "reference-cell mechanism produces wrong charge/energy " + "integration in variable-cell calculations. Set " + "ref_cell_factor = 1.0 (the default) or remove the line. " + "See input_parameter.h ref_cell_factor comment."); + } + }; this->add_item(item); } { @@ -695,7 +719,7 @@ Note: It is a system-dependent empirical parameter. An improper choice might lea Input_Item item("cal_syns"); item.annotation = "calculate asynchronous overlap matrix to output for Hefei-NAMD"; item.category = "Molecular dynamics"; - item.type = R"(Boolean [Integer](optional))"; + item.type = R"(Boolean \[Integer\](optional))"; item.description = R"(Whether to calculate and output asynchronous overlap matrix for Hefei-NAMD interface. When enabled, calculates by computing overlap between basis functions at atomic positions from previous time step and current time step. The overlap is calculated by shifting atom positions backward by velocity x md_dt. Output file: OUT.*/syns_nao.csr in CSR format. * 0 or false: disable @@ -729,5 +753,34 @@ Note: It is a system-dependent empirical parameter. An improper choice might lea read_sync_double(input.dmax); this->add_item(item); } + { + Input_Item item("plumed"); + item.annotation = "use PLUMED for collective variables and enhanced sampling"; + item.category = "Molecular dynamics"; + item.type = "Boolean"; + item.description = "Whether to use PLUMED (https://www.plumed.org) to compute collective " + "variables, biasing potentials and free-energy methods during an MD run. " + "ABACUS must be compiled with -DENABLE_PLUMED=ON and linked against a " + "PLUMED installation (discovered through pkg-config or PLUMED_ROOT) to " + "enable this feature. The PLUMED interface currently supports a single " + "MPI rank only."; + item.default_value = "False"; + item.unit = ""; + read_sync_bool(input.mdp.plumed); + this->add_item(item); + } + { + Input_Item item("plumed_file"); + item.annotation = "input file of PLUMED"; + item.category = "Molecular dynamics"; + item.type = "String"; + item.description = "The input file of PLUMED, read when `plumed` is enabled. Relative paths " + "are resolved against the working directory of the run."; + item.default_value = "plumed.dat"; + item.unit = ""; + item.set_availability("plumed==true"); + read_sync_string(input.mdp.plumed_file); + this->add_item(item); + } } } // namespace ModuleIO diff --git a/source/source_io/module_restart/restart.cpp b/source/source_io/module_restart/restart.cpp index 26c939bc627..9c16d420040 100644 --- a/source/source_io/module_restart/restart.cpp +++ b/source/source_io/module_restart/restart.cpp @@ -23,19 +23,19 @@ void Restart::write_file1(const std::string &file_name, const void*const ptr, const size_t size) const { - std::ofstream ofs(file_name, std::ofstream::binary|std::ofstream::trunc); - ofs.write(static_cast(ptr),size); + std::ofstream ofs(file_name, std::ofstream::binary|std::ofstream::trunc); + ofs.write(static_cast(ptr),size); } void Restart::read_file1(const std::string &file_name, void*const ptr, const size_t size) const { - std::ifstream ifs(file_name, std::ifstream::binary); - ifs.read(static_cast(ptr),size); + std::ifstream ifs(file_name, std::ifstream::binary); + ifs.read(static_cast(ptr),size); } bool Restart::write_file2(const std::string& file_name, const void* const ptr, const size_t size, const bool error_quit) const { - const int file = open(file_name.c_str(), O_WRONLY|O_CREAT|O_TRUNC, S_IRUSR|S_IWUSR); + const int file = open(file_name.c_str(), O_WRONLY|O_CREAT|O_TRUNC, S_IRUSR|S_IWUSR); if (-1 == file){ if (error_quit){ throw std::runtime_error("can't open restart save file. \nerrno=" + ModuleBase::GlobalFunc::TO_STRING(errno) + ".\n" + std::string(__FILE__) + " line " + std::to_string(__LINE__)); @@ -62,7 +62,7 @@ Restart restart; // Peize Lin add 2020.04.04 bool Restart::read_file2(const std::string& file_name, void* const ptr, const size_t size, const bool error_quit) const { - const int file = open(file_name.c_str(), O_RDONLY); + const int file = open(file_name.c_str(), O_RDONLY); if (-1 == file) { if (error_quit) { throw std::runtime_error("can't open restart load file. \nerrno=" + ModuleBase::GlobalFunc::TO_STRING(errno) + ".\n" + std::string(__FILE__) + " line " + std::to_string(__LINE__)); diff --git a/source/source_io/module_restart/restart.h b/source/source_io/module_restart/restart.h index 4fe2787ea6c..de869e4780f 100644 --- a/source/source_io/module_restart/restart.h +++ b/source/source_io/module_restart/restart.h @@ -10,24 +10,24 @@ class Restart { public: - struct Info_Save - { - bool save_charge = false; + struct Info_Save + { + bool save_charge = false; bool save_H = false; // save H means save Hexx now, will be changed in the future. - }; - Info_Save info_save; - - struct Info_Load - { - bool load_charge = false; - bool load_charge_finish = false; - bool load_H = false; - bool load_H_finish = false; + }; + Info_Save info_save; + + struct Info_Load + { + bool load_charge = false; + bool load_charge_finish = false; + bool load_H = false; + bool load_H_finish = false; bool restart_exx = false; // to avoid the repeated load in MD/Relax - }; - Info_Load info_load; - - std::string folder; + }; + Info_Load info_load; + + std::string folder; template bool save_disk(const std::string label, const int index, const int size, T* data, const bool error_quit = true) const @@ -48,8 +48,8 @@ class Restart error_quit); } private: - void write_file1(const std::string &file_name, const void*const ptr, const size_t size) const; - void read_file1(const std::string &file_name, void*const ptr, const size_t size) const; + void write_file1(const std::string &file_name, const void*const ptr, const size_t size) const; + void read_file1(const std::string &file_name, void*const ptr, const size_t size) const; bool write_file2(const std::string& file_name, const void* const ptr, const size_t size, const bool error_quit = true) const; bool read_file2(const std::string& file_name, void* const ptr, const size_t size, const bool error_quit = true) const; }; diff --git a/source/source_io/module_restart/restart_exx_csr.cpp b/source/source_io/module_restart/restart_exx_csr.cpp new file mode 100644 index 00000000000..2c7db4a85cd --- /dev/null +++ b/source/source_io/module_restart/restart_exx_csr.cpp @@ -0,0 +1,192 @@ +#include "restart_exx_csr.h" + +#include "source_base/global_function.h" +#include "source_cell/unitcell.h" +#include "source_base/module_out/csr_reader.h" +#include "source_io/module_hs/hs_sparse_io.h" +#include "source_lcao/module_ri/serialization_cereal.h" +#include +#include +#include +#include +#include +#include +#include + +namespace ModuleIO +{ + template + void read_Hexxs_csr(const std::string& file_name, const UnitCell& ucell, + const int nspin, const int nbasis, + std::vector>>>& Hexxs) + { + ModuleBase::TITLE("ModuleIO", "read_Hexxs_csr"); + Hexxs.resize(nspin); + for (int is = 0;is < nspin;++is) + { + ModuleIO::csrFileReader csr(file_name + "_" + std::to_string(is) + ".csr"); + int nR = csr.getNumberOfR(); + int nbasis = csr.getMatrixDimension(); + // allocate Hexxs[is] + for (int iat1 = 0; iat1 < ucell.nat; ++iat1) { + for (int iat2 = 0;iat2 < ucell.nat;++iat2) { + for (int iR = 0;iR < nR;++iR) + { + const std::vector& R = csr.getRCoordinate(iR); + TC dR({ R[0], R[1], R[2] }); + Hexxs[is][iat1][{iat2, dR}] = RI::Tensor( + { + static_cast(ucell.atoms[ucell.iat2it[iat1]].nw), + static_cast(ucell.atoms[ucell.iat2it[iat2]].nw) + } + ); + } + } + } +#ifdef _OPENMP +#pragma omp for schedule(dynamic) +#endif + for (int iR = 0;iR < nR;++iR) + { + const std::vector& R = csr.getRCoordinate(iR); + const SparseMatrix& matrix = csr.getMatrix(iR); + for (auto& ijv : matrix.getElements()) + { + const int& npol = ucell.get_npol(); + const int& i = ijv.first.first * npol; + const int& j = ijv.first.second * npol; + Hexxs.at(is).at(ucell.iwt2iat[i]).at( + { + ucell.iwt2iat[j], + { R[0], R[1], R[2] } + } + )(ucell.iwt2iw[i] / npol, ucell.iwt2iw[j] / npol) = ijv.second; + } + } + } + } + + template + void read_Hexxs_cereal(const std::string& file_name, + std::vector>>>& Hexxs) + { + ModuleBase::TITLE("ModuleIO", "read_Hexxs_cereal"); + ModuleBase::timer::start("Exx_LRI", "read_Hexxs_cereal"); + std::ifstream ifs(file_name, std::ios::binary); + if(!ifs.is_open()) + { ModuleBase::WARNING_QUIT("read_Hexxs_cereal", file_name+" not found."); } + cereal::BinaryInputArchive iar(ifs); + iar(Hexxs); + ModuleBase::timer::end("Exx_LRI", "read_Hexxs_cereal"); + } + + template + std::map, std::map>> + calculate_RI_Tensor_sparse(const double& sparse_threshold, + const std::map>>& Hexxs, + const UnitCell& ucell) + { + ModuleBase::TITLE("ModuleIO", "calculate_HContainer_sparse_d"); + std::map, std::map>> target; + for (auto& a1_a2R_data : Hexxs) + { + int iat1 = a1_a2R_data.first; + for (auto& a2R_data : a1_a2R_data.second) + { + int iat2 = a2R_data.first.first; + int nw1 = ucell.atoms[ucell.iat2it[iat1]].nw; + int nw2 = ucell.atoms[ucell.iat2it[iat2]].nw; + int start1 = ucell.atoms[ucell.iat2it[iat1]].stapos_wf / ucell.get_npol() + ucell.iat2ia[iat1] * nw1; + int start2 = ucell.atoms[ucell.iat2it[iat2]].stapos_wf / ucell.get_npol() + ucell.iat2ia[iat2] * nw2; + + const TC& R = a2R_data.first.second; + auto& matrix = a2R_data.second; + Abfs::Vector3_Order dR(R[0], R[1], R[2]); + for (int i = 0;i < nw1;++i) { + for (int j = 0;j < nw2;++j) { + if (std::abs(matrix(i, j)) > sparse_threshold) + { + target[dR][start1 + i][start2 + j] = matrix(i, j); + } + } + } + } + } + return target; + } + + template + void write_Hexxs_csr(const std::string& file_name, const UnitCell& ucell, + const std::vector>>>& Hexxs) + { + ModuleBase::TITLE("ModuleIO", "write_Hexxs_csr"); + std::set> all_R_coor; + double sparse_threshold = 1e-10; + int matrix_dimension = 0; + for (int it = 0; it < ucell.ntype; ++it) + { + matrix_dimension += ucell.atoms[it].na * ucell.atoms[it].nw * ucell.get_npol(); + } + if (matrix_dimension <= 0) + { + ModuleBase::WARNING_QUIT("ModuleIO::write_Hexxs_csr", + "LCAO matrix dimension must be positive."); + } + Parallel_Orbitals pv; + pv.set_serial(matrix_dimension, matrix_dimension); + for (int is = 0;is < Hexxs.size();++is) + { + for (const auto& HexxA : Hexxs[is]) + { + const int iat0 = HexxA.first; + for (const auto& HexxB : HexxA.second) + { + const int iat1 = HexxB.first.first; + const Abfs::Vector3_Order R = RI_Util::array3_to_Vector3(HexxB.first.second); + all_R_coor.insert(R); + } + } + ModuleIO::SparseWriteOptions options; + options.filename = file_name + "_" + std::to_string(is) + ".csr"; + options.label = "Hexxs_" + std::to_string(is); + options.threshold = sparse_threshold; + options.binary = false; + options.istep = -1; + options.reduce = false; + ModuleIO::save_sparse( + calculate_RI_Tensor_sparse(sparse_threshold, Hexxs[is], ucell), + all_R_coor, + pv, + options); + } + } + +template void read_Hexxs_csr(const std::string&, + const UnitCell&, + const int, + const int, + std::vector>>>&); +template void read_Hexxs_csr>(const std::string&, + const UnitCell&, + const int, + const int, + std::vector>>>>&); +template void read_Hexxs_cereal(const std::string&, + std::vector>>>&); +template void read_Hexxs_cereal>(const std::string&, + std::vector>>>>&); +template std::map, std::map>> + calculate_RI_Tensor_sparse(const double&, + const std::map>>&, + const UnitCell&); +template std::map, std::map>>> + calculate_RI_Tensor_sparse>(const double&, + const std::map>>>&, + const UnitCell&); +template void write_Hexxs_csr(const std::string&, + const UnitCell&, + const std::vector>>>&); +template void write_Hexxs_csr>(const std::string&, + const UnitCell&, + const std::vector>>>>&); +} // namespace ModuleIO diff --git a/source/source_io/module_restart/restart_exx_csr.h b/source/source_io/module_restart/restart_exx_csr.h index aefa9813134..08f68f82c0b 100644 --- a/source/source_io/module_restart/restart_exx_csr.h +++ b/source/source_io/module_restart/restart_exx_csr.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_IO_MODULE_RESTART_RESTART_EXX_CSR_H +#define ABACUS_SOURCE_IO_MODULE_RESTART_RESTART_EXX_CSR_H + #include "source_cell/unitcell.h" #include "source_lcao/module_ri/ri_util.h" #include "source_lcao/module_ri/abfs_vector3_order.h" @@ -28,15 +30,15 @@ void read_Hexxs_cereal(const std::string& file_name, template void write_Hexxs_csr(const std::string& file_name, const UnitCell& ucell, - const std::map>>& Hexxs); + const std::vector>>>& Hexxs); /// calculate CSR sparse matrix from the global matrix stored with RI::Tensor /// the return type is same as SR_sparse, HR_sparse, etc. template std::map, std::map>> calculate_RI_Tensor_sparse( const double& sparse_threshold, - const std::vector>>>& Hexxs, + const std::map>>& Hexxs, const UnitCell& ucell); } // namespace ModuleIO -#include "restart_exx_csr.hpp" \ No newline at end of file +#endif // ABACUS_SOURCE_IO_MODULE_RESTART_RESTART_EXX_CSR_H diff --git a/source/source_io/module_restart/restart_exx_csr.hpp b/source/source_io/module_restart/restart_exx_csr.hpp deleted file mode 100644 index c2f52e7d17b..00000000000 --- a/source/source_io/module_restart/restart_exx_csr.hpp +++ /dev/null @@ -1,163 +0,0 @@ -#pragma once -#include "restart_exx_csr.h" -#include "source_base/global_function.h" -#include "source_cell/unitcell.h" -#include "source_base/module_out/csr_reader.h" -#include "source_io/module_hs/write_hs_sparse.h" -#include "source_lcao/module_ri/serialization_cereal.h" -#include -#include - -namespace ModuleIO -{ - template - void read_Hexxs_csr(const std::string& file_name, const UnitCell& ucell, - const int nspin, const int nbasis, - std::vector>>>& Hexxs) - { - ModuleBase::TITLE("ModuleIO", "read_Hexxs_csr"); - Hexxs.resize(nspin); - for (int is = 0;is < nspin;++is) - { - ModuleIO::csrFileReader csr(file_name + "_" + std::to_string(is) + ".csr"); - int nR = csr.getNumberOfR(); - int nbasis = csr.getMatrixDimension(); - // allocate Hexxs[is] - for (int iat1 = 0; iat1 < ucell.nat; ++iat1) { - for (int iat2 = 0;iat2 < ucell.nat;++iat2) { - for (int iR = 0;iR < nR;++iR) - { - const std::vector& R = csr.getRCoordinate(iR); - TC dR({ R[0], R[1], R[2] }); - Hexxs[is][iat1][{iat2, dR}] = RI::Tensor( - { - static_cast(ucell.atoms[ucell.iat2it[iat1]].nw), - static_cast(ucell.atoms[ucell.iat2it[iat2]].nw) - } - ); - } - } - } -#ifdef _OPENMP -#pragma omp for schedule(dynamic) -#endif - for (int iR = 0;iR < nR;++iR) - { - const std::vector& R = csr.getRCoordinate(iR); - const SparseMatrix& matrix = csr.getMatrix(iR); - for (auto& ijv : matrix.getElements()) - { - const int& npol = ucell.get_npol(); - const int& i = ijv.first.first * npol; - const int& j = ijv.first.second * npol; - Hexxs.at(is).at(ucell.iwt2iat[i]).at( - { - ucell.iwt2iat[j], - { R[0], R[1], R[2] } - } - )(ucell.iwt2iw[i] / npol, ucell.iwt2iw[j] / npol) = ijv.second; - } - } - } - } - - template - void read_Hexxs_cereal(const std::string& file_name, - std::vector>>>& Hexxs) - { - ModuleBase::TITLE("ModuleIO", "read_Hexxs_cereal"); - ModuleBase::timer::start("Exx_LRI", "read_Hexxs_cereal"); - std::ifstream ifs(file_name, std::ios::binary); - if(!ifs.is_open()) - { ModuleBase::WARNING_QUIT("read_Hexxs_cereal", file_name+" not found."); } - cereal::BinaryInputArchive iar(ifs); - iar(Hexxs); - ModuleBase::timer::end("Exx_LRI", "read_Hexxs_cereal"); - } - - template - std::map, std::map>> - calculate_RI_Tensor_sparse(const double& sparse_threshold, - const std::map>>& Hexxs, - const UnitCell& ucell) - { - ModuleBase::TITLE("ModuleIO", "calculate_HContainer_sparse_d"); - std::map, std::map>> target; - for (auto& a1_a2R_data : Hexxs) - { - int iat1 = a1_a2R_data.first; - for (auto& a2R_data : a1_a2R_data.second) - { - int iat2 = a2R_data.first.first; - int nw1 = ucell.atoms[ucell.iat2it[iat1]].nw; - int nw2 = ucell.atoms[ucell.iat2it[iat2]].nw; - int start1 = ucell.atoms[ucell.iat2it[iat1]].stapos_wf / ucell.get_npol() + ucell.iat2ia[iat1] * nw1; - int start2 = ucell.atoms[ucell.iat2it[iat2]].stapos_wf / ucell.get_npol() + ucell.iat2ia[iat2] * nw2; - - const TC& R = a2R_data.first.second; - auto& matrix = a2R_data.second; - Abfs::Vector3_Order dR(R[0], R[1], R[2]); - for (int i = 0;i < nw1;++i) { - for (int j = 0;j < nw2;++j) { - if (std::abs(matrix(i, j)) > sparse_threshold) - { - target[dR][start1 + i][start2 + j] = matrix(i, j); - } - } - } - } - } - return target; - } - - template - void write_Hexxs_csr(const std::string& file_name, const UnitCell& ucell, - const std::vector>>>& Hexxs) - { - ModuleBase::TITLE("ModuleIO", "write_Hexxs_csr"); - std::set> all_R_coor; - double sparse_threshold = 1e-10; - int matrix_dimension = 0; - for (int it = 0; it < ucell.ntype; ++it) - { - matrix_dimension += ucell.atoms[it].na * ucell.atoms[it].nw * ucell.get_npol(); - } - if (matrix_dimension <= 0) - { - ModuleBase::WARNING_QUIT("ModuleIO::write_Hexxs_csr", - "LCAO matrix dimension must be positive."); - } - Parallel_Orbitals pv; - pv.set_serial(matrix_dimension, matrix_dimension); - const std::string::size_type last_slash = file_name.find_last_of("/\\"); - const std::string temp_dir = last_slash == std::string::npos - ? "" - : file_name.substr(0, last_slash + 1); - for (int is = 0;is < Hexxs.size();++is) - { - for (const auto& HexxA : Hexxs[is]) - { - const int iat0 = HexxA.first; - for (const auto& HexxB : HexxA.second) - { - const int iat1 = HexxB.first.first; - const Abfs::Vector3_Order R = RI_Util::array3_to_Vector3(HexxB.first.second); - all_R_coor.insert(R); - } - } - ModuleIO::SparseWriteOptions options; - options.filename = file_name + "_" + std::to_string(is) + ".csr"; - options.label = "Hexxs_" + std::to_string(is); - options.threshold = sparse_threshold; - options.binary = false; - options.istep = -1; - options.reduce = false; - options.temp_dir = temp_dir; - ModuleIO::save_sparse( - calculate_RI_Tensor_sparse(sparse_threshold, Hexxs[is], ucell), - all_R_coor, - pv, - options); - } - } -} diff --git a/source/source_io/module_unk/berryphase.cpp b/source/source_io/module_unk/berryphase.cpp index e06f58861d4..6262b0c783d 100644 --- a/source/source_io/module_unk/berryphase.cpp +++ b/source/source_io/module_unk/berryphase.cpp @@ -519,12 +519,12 @@ void berryphase::Macroscopic_polarization(const UnitCell& ucell, // calculate Macroscopic polarization modulus because berry phase int modulus = 0; - if ((!lodd) && (PARAM.inp.nspin == 1)) - { - modulus = 2; - } else { - modulus = 1; - } + if ((!lodd) && (PARAM.inp.nspin == 1)) + { + modulus = 2; + } else { + modulus = 1; + } // test by jingan // GlobalV::ofs_running << "ion polarization end" << std::endl; diff --git a/source/source_io/module_unk/unk_overlap_lcao.cpp b/source/source_io/module_unk/unk_overlap_lcao.cpp index 99726117ffe..22aa6960316 100644 --- a/source/source_io/module_unk/unk_overlap_lcao.cpp +++ b/source/source_io/module_unk/unk_overlap_lcao.cpp @@ -180,50 +180,50 @@ void unkOverlap_lcao::init(const UnitCell& ucell, } for (auto& co1: center2_orb11) - { - for (auto& co2: co1.second) - { - for (auto& co3: co2.second) - { - for (auto& co4: co3.second) - { - for (auto& co5: co4.second) - { - for (auto& co6: co5.second) - { - co6.second.init_radial_table(); - } - } - } - } - } - } - - for (auto& co1: center2_orb21_r) - { - for (auto& co2: co1.second) - { - for (auto& co3: co2.second) - { - for (auto& co4: co3.second) - { - for (auto& co5: co4.second) - { - for (auto& co6: co5.second) - { - co6.second.init_radial_table(); - } - } - } - } - } - } - - rcut_orb_.resize(orb.get_ntype()); - for (int it = 0; it < orb.get_ntype(); ++it) - { - rcut_orb_[it] = orb.Phi[it].getRcut(); - } + { + for (auto& co2: co1.second) + { + for (auto& co3: co2.second) + { + for (auto& co4: co3.second) + { + for (auto& co5: co4.second) + { + for (auto& co6: co5.second) + { + co6.second.init_radial_table(); + } + } + } + } + } + } + + for (auto& co1: center2_orb21_r) + { + for (auto& co2: co1.second) + { + for (auto& co3: co2.second) + { + for (auto& co4: co3.second) + { + for (auto& co5: co4.second) + { + for (auto& co6: co5.second) + { + co6.second.init_radial_table(); + } + } + } + } + } + } + + rcut_orb_.resize(orb.get_ntype()); + for (int it = 0; it < orb.get_ntype(); ++it) + { + rcut_orb_[it] = orb.Phi[it].getRcut(); + } return; } @@ -444,10 +444,10 @@ void unkOverlap_lcao::cal_orb_overlap(const UnitCell& ucell) // if ( !pv.in_this_processor(iw1,iw2) ) continue; // iw1 and iw2 never have overlap - if (orb1_orb2_R[iw1][iw2].empty()) - { - continue; - } + if (orb1_orb2_R[iw1][iw2].empty()) + { + continue; + } int atomType1 = iw2it(ucell,iw1); int ia1 = iw2ia(ucell,iw1); diff --git a/source/source_io/module_wannier/to_w90_pw.cpp b/source/source_io/module_wannier/to_w90_pw.cpp index 26334c729e6..ec46620008f 100644 --- a/source/source_io/module_wannier/to_w90_pw.cpp +++ b/source/source_io/module_wannier/to_w90_pw.cpp @@ -1,5 +1,6 @@ #include "to_w90_pw.h" +#include "source_basis/module_pw/pw_basis_big.h" #include "source_io/module_parameter/parameter.h" toW90_PW::toW90_PW( diff --git a/source/source_io/module_wannier/to_w90_pw.h b/source/source_io/module_wannier/to_w90_pw.h index b35c8774946..6a407c08014 100644 --- a/source/source_io/module_wannier/to_w90_pw.h +++ b/source/source_io/module_wannier/to_w90_pw.h @@ -17,6 +17,11 @@ #include "source_basis/module_pw/pw_basis_k.h" #include "source_psi/psi.h" +namespace ModulePW +{ +class PW_Basis_Big; +} + class toW90_PW : public toW90 { public: diff --git a/source/source_io/module_wf/read_wf2rho_pw.cpp b/source/source_io/module_wf/read_wf2rho_pw.cpp index 0c6ea096139..50e31e733a3 100644 --- a/source/source_io/module_wf/read_wf2rho_pw.cpp +++ b/source/source_io/module_wf/read_wf2rho_pw.cpp @@ -4,7 +4,8 @@ #include "source_base/module_out/filename.h" #include "source_base/timer.h" #include "source_estate/kernels/elecstate_op.h" -#include "source_estate/module_charge/symm_rho.h" +#include "source_estate/module_charge/chg_parallel.h" +#include "source_estate/module_charge/chg_symm.h" #include "source_io/module_parameter/parameter.h" #include "source_base/module_device/memory_op.h" @@ -218,15 +219,15 @@ void ModuleIO::read_wf2rho_pw_impl(const ModulePW::PW_Basis_K* pw_wfc, #ifdef __MPI for (int is = 0; is < nspin; ++is) { - chg.reduce_diff_pools(chg.rho[is]); + module_charge::reduce_diff_pools(chg.rho[is], chg, kpar, + PARAM.globalv.all_ks_run, PARAM.inp.bndpar); } #endif // Since rho is calculated by psi^2, it is not symmetric. We need to rearrange it. - Symmetry_rho srho; for (int is = 0; is < nspin; is++) { - srho.begin(is, chg, chg.rhopw, symm); + module_charge::cal_rhog_symm(is, chg, chg.rhopw, symm); } // Free device memory @@ -241,22 +242,22 @@ void ModuleIO::read_wf2rho_pw_impl(const ModulePW::PW_Basis_K* pw_wfc, } void ModuleIO::read_wf2rho_pw( - const ModulePW::PW_Basis_K* pw_wfc, - ModuleSymmetry::Symmetry& symm, - Charge& chg, + const ModulePW::PW_Basis_K* pw_wfc, + ModuleSymmetry::Symmetry& symm, + Charge& chg, const std::string &readin_dir, - const int kpar, - const int my_pool, - const int my_rank, + const int kpar, + const int my_pool, + const int my_rank, const int nproc_in_pool, const int rank_in_pool, - const int nbands, - const int nspin, - const int npol, - const int nkstot, - const std::vector &ik2iktot, - const std::vector &isk, - std::ofstream &ofs_running) + const int nbands, + const int nspin, + const int npol, + const int nkstot, + const std::vector &ik2iktot, + const std::vector &isk, + std::ofstream &ofs_running) { #if defined(__CUDA) || defined(__ROCM) if (pw_wfc->get_device() == "gpu") @@ -273,38 +274,38 @@ void ModuleIO::read_wf2rho_pw( } template void ModuleIO::read_wf2rho_pw_impl( - const ModulePW::PW_Basis_K* pw_wfc, - ModuleSymmetry::Symmetry& symm, - Charge& chg, + const ModulePW::PW_Basis_K* pw_wfc, + ModuleSymmetry::Symmetry& symm, + Charge& chg, const std::string &readin_dir, - const int kpar, - const int my_pool, - const int my_rank, + const int kpar, + const int my_pool, + const int my_rank, const int nproc_in_pool, const int rank_in_pool, - const int nbands, - const int nspin, - const int npol, - const int nkstot, - const std::vector &ik2iktot, - const std::vector &isk, - std::ofstream &ofs_running); + const int nbands, + const int nspin, + const int npol, + const int nkstot, + const std::vector &ik2iktot, + const std::vector &isk, + std::ofstream &ofs_running); #if defined(__CUDA) || defined(__ROCM) template void ModuleIO::read_wf2rho_pw_impl( - const ModulePW::PW_Basis_K* pw_wfc, - ModuleSymmetry::Symmetry& symm, - Charge& chg, + const ModulePW::PW_Basis_K* pw_wfc, + ModuleSymmetry::Symmetry& symm, + Charge& chg, const std::string &readin_dir, - const int kpar, - const int my_pool, - const int my_rank, + const int kpar, + const int my_pool, + const int my_rank, const int nproc_in_pool, const int rank_in_pool, - const int nbands, - const int nspin, - const int npol, - const int nkstot, - const std::vector &ik2iktot, - const std::vector &isk, - std::ofstream &ofs_running); + const int nbands, + const int nspin, + const int npol, + const int nkstot, + const std::vector &ik2iktot, + const std::vector &isk, + std::ofstream &ofs_running); #endif diff --git a/source/source_io/module_wf/read_wf2rho_pw.h b/source/source_io/module_wf/read_wf2rho_pw.h index 0fc76b6fdc4..2205ff11a14 100644 --- a/source/source_io/module_wf/read_wf2rho_pw.h +++ b/source/source_io/module_wf/read_wf2rho_pw.h @@ -20,41 +20,41 @@ namespace ModuleIO */ void read_wf2rho_pw( - const ModulePW::PW_Basis_K* pw_wfc, - ModuleSymmetry::Symmetry& symm, - Charge& chg, + const ModulePW::PW_Basis_K* pw_wfc, + ModuleSymmetry::Symmetry& symm, + Charge& chg, const std::string &readin_dir, - const int kpar, - const int my_pool, - const int my_rank, + const int kpar, + const int my_pool, + const int my_rank, const int nproc_in_pool, const int rank_in_pool, - const int nbands, - const int nspin, - const int npol, - const int nkstot, - const std::vector &ik2iktot, - const std::vector &isk, - std::ofstream &ofs_running); + const int nbands, + const int nspin, + const int npol, + const int nkstot, + const std::vector &ik2iktot, + const std::vector &isk, + std::ofstream &ofs_running); template void read_wf2rho_pw_impl( - const ModulePW::PW_Basis_K* pw_wfc, - ModuleSymmetry::Symmetry& symm, - Charge& chg, + const ModulePW::PW_Basis_K* pw_wfc, + ModuleSymmetry::Symmetry& symm, + Charge& chg, const std::string &readin_dir, - const int kpar, - const int my_pool, - const int my_rank, + const int kpar, + const int my_pool, + const int my_rank, const int nproc_in_pool, const int rank_in_pool, - const int nbands, - const int nspin, - const int npol, - const int nkstot, - const std::vector &ik2iktot, - const std::vector &isk, - std::ofstream &ofs_running); + const int nbands, + const int nspin, + const int npol, + const int nkstot, + const std::vector &ik2iktot, + const std::vector &isk, + std::ofstream &ofs_running); } // namespace ModuleIO diff --git a/source/source_io/module_wf/read_wfc_nao.cpp b/source/source_io/module_wf/read_wfc_nao.cpp index 712c4e513ae..3148e2c0e2f 100644 --- a/source/source_io/module_wf/read_wfc_nao.cpp +++ b/source/source_io/module_wf/read_wfc_nao.cpp @@ -104,11 +104,11 @@ bool ModuleIO::read_wfc_nao( const std::string& global_readin_dir, const Parallel_Orbitals& ParaV, psi::Psi& psid, - ModuleBase::matrix& ekb, + ModuleBase::matrix& ekb, ModuleBase::matrix& wg, - const std::vector &ik2iktot, - const int nkstot, - const int nspin, + const std::vector &ik2iktot, + const int nkstot, + const int nspin, const bool binary, const int skip_band, const int istep) @@ -154,10 +154,10 @@ bool ModuleIO::read_wfc_nao( } // lambda function to read one file - auto read_one_file = [&](const std::string& ss, - std::stringstream& error_message, - const int ik, - std::vector& ctot) + auto read_one_file = [&](const std::string& ss, + std::stringstream& error_message, + const int ik, + std::vector& ctot) { std::ifstream ifs; const std::ios_base::openmode mode @@ -169,9 +169,9 @@ bool ModuleIO::read_wfc_nao( return false; } else - { + { std::cout << " Read NAO wave functions from " << ss << std::endl; - } + } const auto incomplete_file = [&](const std::string& field) { error_message << "The wave function file is incomplete or corrupted while reading " @@ -183,9 +183,9 @@ bool ModuleIO::read_wfc_nao( if (!gamma_only) { int ik_file = 0; - double kx = 0.0; - double ky = 0.0; - double kz = 0.0; + double kx = 0.0; + double ky = 0.0; + double kz = 0.0; if (!read_record_value(ifs, ik_file, binary)) { return incomplete_file("the k-point index"); @@ -282,15 +282,15 @@ bool ModuleIO::read_wfc_nao( std::string errors; - std::vector ctot; - if (myrank == 0) - { - ctot.resize(nbands * nlocal); - } - else - { - ctot.resize(0); - } + std::vector ctot; + if (myrank == 0) + { + ctot.resize(nbands * nlocal); + } + else + { + ctot.resize(0); + } for(int ik=0;ik(const std::string& global_readin_dir, const Parallel_Orbitals& ParaV, psi::Psi& psid, - ModuleBase::matrix& ekb, + ModuleBase::matrix& ekb, ModuleBase::matrix& wg, - const std::vector &ik2iktot, - const int nkstot, - const int nspin, + const std::vector &ik2iktot, + const int nkstot, + const int nspin, const bool binary, const int skip_band, const int istep); @@ -364,36 +364,36 @@ template bool ModuleIO::read_wfc_nao(const std::string& global_readin_di template bool ModuleIO::read_wfc_nao(const std::string& global_readin_dir, const Parallel_Orbitals& ParaV, psi::Psi& psid, - ModuleBase::matrix& ekb, + ModuleBase::matrix& ekb, ModuleBase::matrix& wg, - const std::vector &ik2iktot, - const int nkstot, - const int nspin, + const std::vector &ik2iktot, + const int nkstot, + const int nspin, const bool binary, const int skip_band, const int istep); template bool ModuleIO::read_wfc_nao>(const std::string& global_readin_dir, const Parallel_Orbitals& ParaV, - psi::Psi>& psid, - ModuleBase::matrix& ekb, + psi::Psi>& psid, + ModuleBase::matrix& ekb, ModuleBase::matrix& wg, - const std::vector &ik2iktot, - const int nkstot, - const int nspin, + const std::vector &ik2iktot, + const int nkstot, + const int nspin, const bool binary, - const int skip_band, + const int skip_band, const int istep); // mohan add 2025-10-19 template bool ModuleIO::read_wfc_nao>(const std::string& global_readin_dir, const Parallel_Orbitals& ParaV, - psi::Psi>& psid, - ModuleBase::matrix& ekb, + psi::Psi>& psid, + ModuleBase::matrix& ekb, ModuleBase::matrix& wg, - const std::vector &ik2iktot, - const int nkstot, - const int nspin, + const std::vector &ik2iktot, + const int nkstot, + const int nspin, const bool binary, - const int skip_band, + const int skip_band, const int istep); diff --git a/source/source_io/module_wf/read_wfc_nao.h b/source/source_io/module_wf/read_wfc_nao.h index 263b8de5403..0b3b23e11e2 100644 --- a/source/source_io/module_wf/read_wfc_nao.h +++ b/source/source_io/module_wf/read_wfc_nao.h @@ -43,11 +43,11 @@ bool read_wfc_nao( const std::string& global_readin_dir, const Parallel_Orbitals& ParaV, psi::Psi& psid, - ModuleBase::matrix& ekb, + ModuleBase::matrix& ekb, ModuleBase::matrix& wg, - const std::vector &ik2iktot, - const int nkstot, - const int nspin, + const std::vector &ik2iktot, + const int nkstot, + const int nspin, const bool binary, const int skip_band = 0, const int istep = -1); diff --git a/source/source_io/module_wf/read_wfc_pw.cpp b/source/source_io/module_wf/read_wfc_pw.cpp index 3850f2a42a6..0afaa5720b7 100644 --- a/source/source_io/module_wf/read_wfc_pw.cpp +++ b/source/source_io/module_wf/read_wfc_pw.cpp @@ -12,15 +12,15 @@ #include void ModuleIO::read_wfc_pw(const std::string& filename, - const ModulePW::PW_Basis_K* pw_wfc, - const int rank_in_pool, - const int nproc_in_pool, - const int nbands, - const int npol, - const int& ik, - const int& ikstot, - const int& nkstot, - ModuleBase::ComplexMatrix& wfc) + const ModulePW::PW_Basis_K* pw_wfc, + const int rank_in_pool, + const int nproc_in_pool, + const int nbands, + const int npol, + const int& ik, + const int& ikstot, + const int& nkstot, + ModuleBase::ComplexMatrix& wfc) { ModuleBase::TITLE("ModuleIO", "read_wfc_pw"); ModuleBase::timer::start("ModuleIO", "read_wfc_pw"); diff --git a/source/source_io/module_wf/read_wfc_pw.h b/source/source_io/module_wf/read_wfc_pw.h index fcd3023301d..d36a4113b63 100644 --- a/source/source_io/module_wf/read_wfc_pw.h +++ b/source/source_io/module_wf/read_wfc_pw.h @@ -20,15 +20,15 @@ namespace ModuleIO * @param wfc wave functions */ void read_wfc_pw(const std::string& filedir, - const ModulePW::PW_Basis_K* pw_wfc, - const int rank_in_pool, - const int nproc_in_pool, - const int nbands, - const int npol, - const int& ik, - const int& ikstot, - const int& nkstot, - ModuleBase::ComplexMatrix& wfc); + const ModulePW::PW_Basis_K* pw_wfc, + const int rank_in_pool, + const int nproc_in_pool, + const int nbands, + const int npol, + const int& ik, + const int& ikstot, + const int& nkstot, + ModuleBase::ComplexMatrix& wfc); } // namespace ModuleIO #endif diff --git a/source/source_io/module_wf/write_wfc_nao.cpp b/source/source_io/module_wf/write_wfc_nao.cpp index 53dd4248688..2d28e96e60d 100644 --- a/source/source_io/module_wf/write_wfc_nao.cpp +++ b/source/source_io/module_wf/write_wfc_nao.cpp @@ -41,7 +41,7 @@ void wfc_nao_write2file(const std::string& name, } if (!ofs) { - ModuleBase::WARNING("ModuleIO::wfc_nao_write2file", "Can't write local orbital wave functions."); + ModuleBase::WARNING_QUIT("ModuleIO::wfc_nao_write2file", "Can't write local orbital wave functions."); } ofs << nbands; @@ -73,7 +73,7 @@ void wfc_nao_write2file(const std::string& name, } if (!ofs) { - ModuleBase::WARNING("ModuleIO::wfc_nao_write2file", "Can't write local orbital wave functions."); + ModuleBase::WARNING_QUIT("ModuleIO::wfc_nao_write2file", "Can't write local orbital wave functions."); } ofs << nbands << " (number of bands)" << std::endl; ofs << nlocal << " (number of orbitals)"; @@ -132,7 +132,7 @@ void wfc_nao_write2file_complex(const std::string& name, } if (!ofs) { - ModuleBase::WARNING("ModuleIO::wfc_nao_write2file_complex", "Can't write local orbital wave functions."); + ModuleBase::WARNING_QUIT("ModuleIO::wfc_nao_write2file_complex", "Can't write local orbital wave functions."); } ofs << ik + 1; ofs << kvec_c.x; @@ -167,7 +167,7 @@ void wfc_nao_write2file_complex(const std::string& name, } if (!ofs) { - ModuleBase::WARNING("ModuleIO::wfc_nao_write2file_complex", "Can't write local orbital wave functions."); + ModuleBase::WARNING_QUIT("ModuleIO::wfc_nao_write2file_complex", "Can't write local orbital wave functions."); } ofs << std::setprecision(8); ofs << ik + 1 << " (index of k points)" << std::endl; @@ -202,16 +202,16 @@ void wfc_nao_write2file_complex(const std::string& name, template void write_wfc_nao(const int out_type, - const bool out_app_flag, - const psi::Psi& psi, - const ModuleBase::matrix& ekb, - const ModuleBase::matrix& wg, - const std::vector>& kvec_c, - const std::vector &ik2iktot, - const int nkstot, - const Parallel_Orbitals& pv, - const int nspin, - const int istep) + const bool out_app_flag, + const psi::Psi& psi, + const ModuleBase::matrix& ekb, + const ModuleBase::matrix& wg, + const std::vector>& kvec_c, + const std::vector &ik2iktot, + const int nkstot, + const Parallel_Orbitals& pv, + const int nspin, + const int istep) { if (!out_type) { @@ -311,27 +311,27 @@ void write_wfc_nao(const int out_type, } template void write_wfc_nao(const int out_type, - const bool out_app_flag, - const psi::Psi& psi, - const ModuleBase::matrix& ekb, - const ModuleBase::matrix& wg, - const std::vector>& kvec_c, - const std::vector &ik2iktot, - const int nkstot, - const Parallel_Orbitals& pv, - const int nspin, - const int istep); + const bool out_app_flag, + const psi::Psi& psi, + const ModuleBase::matrix& ekb, + const ModuleBase::matrix& wg, + const std::vector>& kvec_c, + const std::vector &ik2iktot, + const int nkstot, + const Parallel_Orbitals& pv, + const int nspin, + const int istep); template void write_wfc_nao>(const int out_type, - const bool out_app_flag, - const psi::Psi>& psi, - const ModuleBase::matrix& ekb, - const ModuleBase::matrix& wg, - const std::vector>& kvec_c, - const std::vector &ik2iktot, - const int nkstot, - const Parallel_Orbitals& pv, - const int nspin, - const int istep); + const bool out_app_flag, + const psi::Psi>& psi, + const ModuleBase::matrix& ekb, + const ModuleBase::matrix& wg, + const std::vector>& kvec_c, + const std::vector &ik2iktot, + const int nkstot, + const Parallel_Orbitals& pv, + const int nspin, + const int istep); } // namespace ModuleIO diff --git a/source/source_io/module_wf/write_wfc_nao.h b/source/source_io/module_wf/write_wfc_nao.h index 48284d99113..152723b5109 100644 --- a/source/source_io/module_wf/write_wfc_nao.h +++ b/source/source_io/module_wf/write_wfc_nao.h @@ -9,7 +9,7 @@ namespace ModuleIO { - /** + /** * Writes the wavefunction coefficients for the LCAO method to a file. * Will loop all k-points by psi.get_nk(). * The nbands are determined by ekb.nc. @@ -25,16 +25,16 @@ namespace ModuleIO */ template void write_wfc_nao(const int out_type, - const bool out_app_flag, - const psi::Psi& psi, - const ModuleBase::matrix& ekb, - const ModuleBase::matrix& wg, - const std::vector>& kvec_c, - const std::vector &ik2iktot, - const int nkstot, - const Parallel_Orbitals& pv, - const int nspin, - const int istep=-1) ; + const bool out_app_flag, + const psi::Psi& psi, + const ModuleBase::matrix& ekb, + const ModuleBase::matrix& wg, + const std::vector>& kvec_c, + const std::vector &ik2iktot, + const int nkstot, + const Parallel_Orbitals& pv, + const int nspin, + const int istep=-1) ; void wfc_nao_write2file(const std::string& name, const double* ctot, diff --git a/source/source_io/module_wf/write_wfc_pw.cpp b/source/source_io/module_wf/write_wfc_pw.cpp index 7ad4ed137f2..450bb596988 100644 --- a/source/source_io/module_wf/write_wfc_pw.cpp +++ b/source/source_io/module_wf/write_wfc_pw.cpp @@ -32,10 +32,10 @@ void ModuleIO::write_wfc_pw( { ModuleBase::TITLE("ModuleIO", "write_wfc_pw"); - if(out_wfc_pw!=1 && out_wfc_pw!=2) - { - return; - } + if(out_wfc_pw!=1 && out_wfc_pw!=2) + { + return; + } const int nkstot = kv.get_nkstot(); const int nks = kv.get_nks(); @@ -61,23 +61,23 @@ void ModuleIO::write_wfc_pw( ofs_running << " Write G-space wave functions to file: " << fn << std::endl; - wfilename[ik_local] = fn; + wfilename[ik_local] = fn; - if (rank_in_pool == 0) - { - if (out_wfc_pw == 1) - { - std::ofstream ofs(fn.c_str()); // clear all wavefunc files. - ofs.close(); - } - else if (out_wfc_pw == 2) - { - Binstream wfs(fn, "w"); - wfs.close(); - } - } - } - } + if (rank_in_pool == 0) + { + if (out_wfc_pw == 1) + { + std::ofstream ofs(fn.c_str()); // clear all wavefunc files. + ofs.close(); + } + else if (out_wfc_pw == 2) + { + Binstream wfs(fn, "w"); + wfs.close(); + } + } + } + } #ifdef __MPI diff --git a/source/source_io/parse_args.cpp b/source/source_io/parse_args.cpp index 1a6fa1fe386..046bc527028 100644 --- a/source/source_io/parse_args.cpp +++ b/source/source_io/parse_args.cpp @@ -165,7 +165,7 @@ void print_build_info() // --- 11. Other Libraries --- print_section("Other Libraries"); print_info("DFT-D4 Support", ABACUS_DFTD4_VERSION); - print_info("RapidJSON Support", ABACUS_RAPIDJSON_VERSION); + print_info("JSON Support", ABACUS_JSON_VERSION); print_info("PEXSI Support", ABACUS_PEXSI_VERSION); print_info("cnpy Support", ABACUS_CNPY_VERSION); diff --git a/source/source_io/test/CMakeLists.txt b/source/source_io/test/CMakeLists.txt index 9c4498bf419..107ff7de930 100644 --- a/source/source_io/test/CMakeLists.txt +++ b/source/source_io/test/CMakeLists.txt @@ -67,18 +67,6 @@ AddTest( SOURCES td_vector_pot_io_test.cpp ../module_efield/td_vector_pot_io.cpp ) -AddTest( - TARGET MODULE_IO_write_hsk_binary_test - LIBS parameter base device - SOURCES write_hsk_binary_test.cpp ../../source_basis/module_ao/parallel_orbitals.cpp -) - -add_test( - NAME MODULE_IO_write_hsk_binary_test_parallel - COMMAND mpirun -np 2 ./MODULE_IO_write_hsk_binary_test - WORKING_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR} -) - AddTest( TARGET MODULE_IO_write_eig_occ_test LIBS parameter base device symmetry @@ -86,17 +74,7 @@ AddTest( ../../source_cell/cif_io.cpp ../../source_cell/reciprocal_grid.cpp ) -AddTest( - TARGET MODULE_IO_cal_dos - LIBS parameter base device - SOURCES cal_dos_test.cpp ../module_dos/cal_dos.cpp -) - -AddTest( - TARGET MODULE_IO_write_dos_pw - LIBS parameter base device symmetry - SOURCES write_dos_pw_test.cpp ../module_dos/cal_dos.cpp ../module_dos/write_dos_pw.cpp ../../source_cell/parallel_kpoints.cpp ../../source_cell/klist.cpp ../../source_cell/klist_io.cpp ../module_energy/nscf_fermi_surf.cpp ../../source_cell/reciprocal_grid.cpp -) +# DOS tests moved to ../module_dos/unittests/ AddTest( TARGET MODULE_IO_print_info @@ -104,18 +82,6 @@ AddTest( SOURCES print_info_test.cpp ../module_output/print_info.cpp ../../source_cell/klist.cpp ../../source_cell/klist_io.cpp ../../source_cell/parallel_kpoints.cpp ../../source_cell/reciprocal_grid.cpp ) -AddTest( - TARGET MODULE_IO_single_R_test - LIBS parameter - SOURCES single_r_io_test.cpp ../module_hs/single_r_io.cpp - ../../source_base/global_variable.cpp - ../../source_base/parallel_reduce.cpp - ../../source_base/parallel_common.cpp - ../../source_base/parallel_global.cpp - ../../source_base/parallel_comm.cpp - ../../source_base/tool_quit.cpp ../../source_base/global_file.cpp ../../source_base/global_function.cpp ../../source_base/memory_recorder.cpp ../../source_base/timer.cpp -) - AddTest( TARGET MODULE_IO_write_wfc_nao LIBS parameter base psi device @@ -189,7 +155,7 @@ add_test(NAME MODULE_IO_read_wfc_pw_test_parallel AddTest( TARGET MODULE_IO_read_wf2rho_pw_test LIBS parameter base device planewave psi symmetry - SOURCES read_wf2rho_pw_test.cpp ../module_wf/read_wfc_pw.cpp ../module_wf/read_wf2rho_pw.cpp ../../source_basis/module_pw/test/test_tool.cpp ../../source_estate/module_charge/charge_mpi.cpp ../module_wf/write_wfc_pw.cpp + SOURCES read_wf2rho_pw_test.cpp ../module_wf/read_wfc_pw.cpp ../module_wf/read_wf2rho_pw.cpp ../../source_basis/module_pw/test/test_tool.cpp ../../source_estate/module_charge/charge.cpp ../../source_estate/module_charge/chg_tools.cpp ../../source_estate/module_charge/chg_parallel.cpp ../module_wf/write_wfc_pw.cpp ../../source_cell/klist.cpp ../../source_cell/klist_io.cpp ../../source_cell/parallel_kpoints.cpp ../../source_cell/reciprocal_grid.cpp ) @@ -256,6 +222,15 @@ add_test( WORKING_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR} ) +if(ENABLE_LCAO) +AddTest( + TARGET MODULE_IO_write_dmr_test + LIBS parameter base device hcontainer + SOURCES ../module_dm/test/test_write_dmr.cpp ../module_dm/write_dmr.cpp ../../source_cell/ucell_io.cpp + ../../source_basis/module_ao/parallel_orbitals.cpp ../../source_hamilt/module_hcontainer/test/tmp_mocks.cpp +) +endif() + AddTest( TARGET MODULE_IO_read_wfc_nao_test LIBS parameter base device @@ -268,57 +243,15 @@ add_test( WORKING_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR} ) -if(ENABLE_LCAO) -AddTest( - TARGET MODULE_IO_cal_plpr_test - LIBS parameter base device neighbor - SOURCES - cal_plpr_test.cpp - ../module_hs/cal_plpr.cpp - ../../source_basis/module_ao/orb_atomic_lm.cpp - ../../source_basis/module_ao/orb_atomic.cpp - ../../source_basis/module_nao/radial_set.cpp - ../../source_basis/module_nao/numerical_radial.cpp - ../../source_basis/module_nao/beta_radials.cpp - ../../source_basis/module_nao/hydrogen_radials.cpp - ../../source_basis/module_nao/pswfc_radials.cpp - ../../source_basis/module_nao/atomic_radials.cpp - ../../source_basis/module_nao/sphbes_radials.cpp - ../../source_basis/module_nao/two_center_integrator.cpp - ../../source_basis/module_nao/two_center_table.cpp - ../../source_basis/module_nao/real_gaunt_table.cpp - ../../source_basis/module_nao/radial_collection.cpp -) - -AddTest( - TARGET MODULE_IO_write_hs_r_compat_test - LIBS parameter base device hcontainer - SOURCES - write_hs_r_compat_test.cpp - ../module_hs/write_hs_r.cpp - ../module_hs/write_hs_sparse.cpp - ../module_hs/single_r_io.cpp - ../module_hs/rr_sparse_writer.cpp - ../module_dm/write_dmr.cpp - ../../source_cell/ucell_io.cpp - ../../source_basis/module_ao/parallel_orbitals.cpp - ../../source_hamilt/module_hcontainer/test/tmp_mocks.cpp -) - -add_test( - NAME MODULE_IO_write_hsr_binary_test_parallel - COMMAND mpirun -np 2 ./MODULE_IO_write_hs_r_compat_test - --gtest_filter=WriteHsRCompatibility.HContainerBinaryMpiGatherWritesCompleteFiles - WORKING_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR} -) -endif() - if(ENABLE_LIBRI) AddTest( TARGET MODULE_IO_restart_exx_csr_test LIBS parameter base device SOURCES restart_exx_csr_test.cpp + ../module_restart/restart_exx_csr.cpp + ../module_hs/hs_sparse_io.cpp + ../module_hs/lat_r_csr.cpp tmp_mocks.cpp ) endif() diff --git a/source/source_io/test/cal_dos_test.cpp b/source/source_io/test/cal_dos_test.cpp deleted file mode 100644 index cc8448dd270..00000000000 --- a/source/source_io/test/cal_dos_test.cpp +++ /dev/null @@ -1,178 +0,0 @@ -#include "gtest/gtest.h" -#include "gmock/gmock.h" -#include "source_io/module_dos/cal_dos.h" -#include "source_base/global_variable.h" -#include -#ifdef __MPI -#include "mpi.h" -#endif -#include "dos_test.h" - -/************************************************ - * unit test of ca_dos - ***********************************************/ - -/** - * - Tested Functions: - * - cal_dos() - * - the function to calculate and print out - * - density of states - */ - - -class DosTest : public ::testing::Test -{ -protected: - std::string output; -}; - -TEST_F(DosTest,Dos) -{ - //is,fa,de_ev,emax_ev,emin_ev,bcoeff,nks,nkstot,nbands - DosPrepare dosp = DosPrepare(0,"doss1_pw.txt",0.005,18,-6,0.07,36,36,8); - dosp.set_isk(); - dosp.read_wk(); - dosp.read_istate_info(); - EXPECT_EQ(dosp.is,0); - - const int istep = 1; - ModuleIO::cal_dos(dosp.is, - dosp.fa, - dosp.de_ev, - dosp.emax_ev, - dosp.emin_ev, - dosp.bcoeff, - dosp.nks, - dosp.nkstot, - dosp.wk, - dosp.isk, - dosp.nbands, - dosp.ekb, - dosp.wg, - istep); - -#ifdef __MPI - if(GlobalV::MY_RANK==0) - { -#endif - std::ifstream ifs; - ifs.open(dosp.fa.c_str()); - std::string str((std::istreambuf_iterator(ifs)),std::istreambuf_iterator()); - EXPECT_THAT(str, testing::HasSubstr("4801 # number of points")); - EXPECT_THAT(str, testing::HasSubstr(" -5.39 0.03125 0.03125 0.178099 0.0160702")); - EXPECT_THAT(str, testing::HasSubstr(" 3.07 0.1875 5.46875 1.07003 5.37765")); - ifs.close(); - remove("doss1_pw.txt"); -#ifdef __MPI - } -#endif -} - - - -TEST_F(DosTest,DosW1) -{ - //is,fa,de_ev,emax_ev,emin_ev,bcoeff,nks,nkstot,nbands - DosPrepare dosp = DosPrepare(0,"doss1_pw.txt",-0.005,18,-6,0.07,36,36,8); - dosp.set_isk(); - dosp.read_wk(); - dosp.read_istate_info(); - EXPECT_EQ(dosp.is,0); - EXPECT_LE(dosp.de_ev,0); - GlobalV::ofs_warning.open("warning1.log"); - - const int istep = 1; - EXPECT_NO_THROW(ModuleIO::cal_dos(dosp.is, - dosp.fa, - dosp.de_ev, - dosp.emax_ev, - dosp.emin_ev, - dosp.bcoeff, - dosp.nks, - dosp.nkstot, - dosp.wk, - dosp.isk, - dosp.nbands, - dosp.ekb, - dosp.wg, - istep)); - GlobalV::ofs_warning.close(); -#ifdef __MPI - if(GlobalV::MY_RANK==0) - { -#endif - std::ifstream ifs; - ifs.open("warning1.log"); - std::string str((std::istreambuf_iterator(ifs)),std::istreambuf_iterator()); - EXPECT_THAT(str, testing::HasSubstr("ModuleIO::cal_dos warning : de <= 0")); - ifs.close(); - remove("warning1.log"); - remove("doss1_pw.txt"); -#ifdef __MPI - } -#endif -} - - -TEST_F(DosTest,DosW2) -{ - //is,fa,de_ev,emax_ev,emin_ev,bcoeff,nks,nkstot,nbands - DosPrepare dosp = DosPrepare(0,"doss1_pw.txt",0.005,-6,18,0.07,36,36,8); - dosp.set_isk(); - dosp.read_wk(); - dosp.read_istate_info(); - EXPECT_EQ(dosp.is,0); - GlobalV::ofs_warning.open("warning2.log"); - - const int istep = 1; - EXPECT_NO_THROW(ModuleIO::cal_dos(dosp.is, - dosp.fa, - dosp.de_ev, - dosp.emax_ev, - dosp.emin_ev, - dosp.bcoeff, - dosp.nks, - dosp.nkstot, - dosp.wk, - dosp.isk, - dosp.nbands, - dosp.ekb, - dosp.wg, - istep)); - - GlobalV::ofs_warning.close(); -#ifdef __MPI - if(GlobalV::MY_RANK==0) - { -#endif - std::ifstream ifs; - ifs.open("warning2.log"); - std::string str((std::istreambuf_iterator(ifs)),std::istreambuf_iterator()); - EXPECT_THAT(str, testing::HasSubstr("ModuleIO::cal_dos warning : emax_ev < emin_ev")); - ifs.close(); - remove("warning2.log"); - remove("doss1_pw.txt"); -#ifdef __MPI - } -#endif -} - -#ifdef __MPI -int main(int argc, char **argv) -{ - MPI_Init(&argc,&argv); - - testing::InitGoogleTest(&argc,argv); - MPI_Comm_size(MPI_COMM_WORLD,&GlobalV::NPROC); - MPI_Comm_rank(MPI_COMM_WORLD,&GlobalV::MY_RANK); - - // only test a certain one - //::testing::GTEST_FLAG(filter) = "DosTest.DosW1"; - - int result = RUN_ALL_TESTS(); - - MPI_Finalize(); - - return result; -} -#endif diff --git a/source/source_io/test/cal_plpr_test.cpp b/source/source_io/test/cal_plpr_test.cpp deleted file mode 100644 index 044083246c2..00000000000 --- a/source/source_io/test/cal_plpr_test.cpp +++ /dev/null @@ -1,172 +0,0 @@ -/** - * Unit-test of cal_plpr.cpp - * - * The representation matrices for Lx, Ly and Lz operators (SO(3) generators). - * On basis {s}: - * Lx: Ly: Lz: - * 0 0 0 - * - * On basis {px, py, pz}: - * Lx: Ly: Lz: - * px py pz px py pz px py pz - * px 0 px 0 i px 0 -i - * py 0 -i py 0 py i 0 - * pz i 0 pz -i 0 pz 0 - * - * will test if the calculated values are close to the above results. - */ - -#include -#include -#include -#include -#include -#ifdef __MPI -#include -#endif - -#include "source_io/module_hs/cal_plpr.h" -#include "source_basis/module_nao/two_center_integrator.h" -#include "source_basis/module_nao/radial_collection.h" -#include "source_base/sph_bessel_tf.h" -#include "source_base/ylm.h" - -#define DOUBLETHRESHOLD 1e-12 - -class CalpLpRTest : public ::testing::Test -{ -protected: - void SetUp() override - { - ModuleBase::SphericalBesselTransformer sbt(true); - - this->orb_ = std::unique_ptr(new RadialCollection); - this->orb_->build(1, &forb_, 'o'); - this->orb_->set_transformer(sbt); - - const double rcut = 8.0; - const int ngrid = int(rcut / 0.01) + 1; - const double cutoff = 2.0 * rcut; - this->orb_->set_uniform_grid(true, ngrid, cutoff, 'i', true); - - this->calculator_ = std::unique_ptr(new TwoCenterIntegrator); - this->calculator_->tabulate(*orb_, *orb_, 'S', ngrid, cutoff); - - // Initialize Ylm coefficients - ModuleBase::Ylm::set_coefficients(); - } - - void TearDown() override - { - // Cleanup code here, if needed - } - - const std::string forb_ = "../../../../tests/PP_ORB/Si_gga_8au_100Ry_2s2p1d.orb"; - - std::unique_ptr calculator_; - std::unique_ptr orb_; -}; - -TEST_F(CalpLpRTest, CalLzijRTest) -{ - std::complex out; - - ModuleBase::Vector3 vR(0., 0., 0.); // home-cell - int it = 0, ia = 0, il = 0, iz = 0, im = 0; - int jt = 0, ja = 0, jl = 0, jz = 0, jm = 0; - - // l=0 - out = ModuleIO::cal_LzijR(calculator_, it, ia, il, iz, im, jt, ja, jl, jz, jm, vR); - EXPECT_NEAR(out.real(), 0.0, DOUBLETHRESHOLD); - EXPECT_NEAR(out.imag(), 0.0, DOUBLETHRESHOLD); - - // l=1 - il = 1; jl = 1; - std::vector> ans(9, 0.0); - ans[1] = {0.0, -1.0}; ans[3] = {0.0, 1.0}; - int idx = 0; - const std::vector m = {1, -1, 0}; // px, py, pz - for (auto im_: m) { - for (auto jm_: m) { - im = im_; jm = jm_; - out = ModuleIO::cal_LzijR(calculator_, it, ia, il, iz, im, jt, ja, jl, jz, jm, vR); - EXPECT_NEAR(out.real(), ans[idx].real(), DOUBLETHRESHOLD); - EXPECT_NEAR(out.imag(), ans[idx].imag(), DOUBLETHRESHOLD); - idx++; - } - } -} - -TEST_F(CalpLpRTest, CalLxijRTest) -{ - std::complex out; - - ModuleBase::Vector3 vR(0., 0., 0.); // home-cell - int it = 0, ia = 0, il = 0, iz = 0, im = 0; - int jt = 0, ja = 0, jl = 0, jz = 0, jm = 0; - - // l=0 - out = ModuleIO::cal_LxijR(calculator_, it, ia, il, iz, im, jt, ja, jl, jz, jm, vR); - EXPECT_NEAR(out.real(), 0.0, DOUBLETHRESHOLD); - EXPECT_NEAR(out.imag(), 0.0, DOUBLETHRESHOLD); - - // l=1 - il = 1; jl = 1; - std::vector> ans(9, 0.0); - ans[5] = {0.0, 1.0}; ans[7] = {0.0, -1.0}; - int idx = 0; - const std::vector m = {1, -1, 0}; // px, py, pz - for (auto im_: m) { - for (auto jm_: m) { - im = im_; jm = jm_; - out = ModuleIO::cal_LxijR(calculator_, it, ia, il, iz, im, jt, ja, jl, jz, jm, vR); - EXPECT_NEAR(out.real(), ans[idx].real(), DOUBLETHRESHOLD); - EXPECT_NEAR(out.imag(), ans[idx].imag(), DOUBLETHRESHOLD); - idx++; - } - } -} - -TEST_F(CalpLpRTest, CalLyijRTest) -{ - std::complex out; - - ModuleBase::Vector3 vR(0., 0., 0.); // home-cell - int it = 0, ia = 0, il = 0, iz = 0, im = 0; - int jt = 0, ja = 0, jl = 0, jz = 0, jm = 0; // self, the first s - - // l=0 - out = ModuleIO::cal_LyijR(calculator_, it, ia, il, iz, im, jt, ja, jl, jz, jm, vR); - EXPECT_NEAR(out.real(), 0.0, DOUBLETHRESHOLD); - EXPECT_NEAR(out.imag(), 0.0, DOUBLETHRESHOLD); - - // l=1 - il = 1; jl = 1; - std::vector> ans(9, 0.0); - ans[2] = {0.0, -1.0}; ans[6] = {0.0, 1.0}; - int idx = 0; - const std::vector m = {1, -1, 0}; // px, py, pz - for (auto im_: m) { - for (auto jm_: m) { - im = im_; jm = jm_; - out = ModuleIO::cal_LyijR(calculator_, it, ia, il, iz, im, jt, ja, jl, jz, jm, vR); - EXPECT_NEAR(out.real(), ans[idx].real(), DOUBLETHRESHOLD); - EXPECT_NEAR(out.imag(), ans[idx].imag(), DOUBLETHRESHOLD); - idx++; - } - } -} - -int main(int argc, char **argv) -{ -#ifdef __MPI - MPI_Init(&argc, &argv); -#endif - ::testing::InitGoogleTest(&argc, argv); - return RUN_ALL_TESTS(); - -#ifdef __MPI - MPI_Finalize(); -#endif - return 0; -} \ No newline at end of file diff --git a/source/source_io/test/output_mulliken_mock.cpp b/source/source_io/test/output_mulliken_mock.cpp index b83e66b57cf..f7dc5e44537 100644 --- a/source/source_io/test/output_mulliken_mock.cpp +++ b/source/source_io/test/output_mulliken_mock.cpp @@ -73,13 +73,13 @@ namespace ModuleIO { template -Output_DMK::Output_DMK(elecstate::DensityMatrix* p_DM, Parallel_Orbitals* ParaV, int nspin, int nks) +Output_DMK::Output_DMK(module_dm::DensityMatrix* p_DM, Parallel_Orbitals* ParaV, int nspin, int nks) : p_DM_(p_DM), ParaV_(ParaV), nspin_(nspin), nks_(nks) { } template -TK* Output_DMK::get_DMK(int ik) +TK* Output_DMK::get_dmk(int ik) { if (this->nspin_ == 1) { diff --git a/source/source_io/test/outputlog_test.cpp b/source/source_io/test/outputlog_test.cpp index b84a9ef4a0e..e7fc8886d47 100644 --- a/source/source_io/test/outputlog_test.cpp +++ b/source/source_io/test/outputlog_test.cpp @@ -1,8 +1,6 @@ #include "gmock/gmock.h" #include "gtest/gtest.h" -#define private public #include "source_io/module_parameter/parameter.h" -#undef private #include #include #include @@ -16,6 +14,19 @@ #include "source_basis/module_pw/test/test_tool.h" #include "mpi.h" #endif + +/// @brief Friend helper to mutate PARAM private members in unit tests. +/// @details Parameter grants friend access to TestParameters so the test can +/// modify input/sys fields without `#define private public`. The class must +/// stay at global scope to match the friend declaration in parameter.h; +/// an anonymous-namespace class would not be the friend. +class TestParameters +{ + public: + static Input_para& input() { return PARAM.input; } + static System_para& sys() { return PARAM.sys; } +}; + /** * - Tested Functions: * - output_convergence_after_scf() @@ -114,7 +125,7 @@ TEST(OutputEfermiTest, TestNotConvergence) { TEST(OutputEfermiTest, TestMOutputLevel) { bool convergence = true; double efermi = 1.0; - PARAM.input.out_level = "m"; // Setting output level to "m" + TestParameters::input().out_level = "m"; // Setting output level to "m" std::ofstream ofs_running("test_output_efermi_m_outputlevel.txt"); ModuleIO::output_efermi(convergence, efermi, ofs_running); ofs_running.close(); @@ -184,7 +195,7 @@ Sep_Cell::~Sep_Cell() noexcept {} TEST(OutputVacuumLevelTest, OutputVacuumLevel) { - PARAM.input.nspin = 1; + TestParameters::input().nspin = 1; UnitCell ucell; const int nx = 50, ny = 50, nz = 50, nxyz = 125000, nrxx = 125000, nplane = 50, startz_current = 0; @@ -220,7 +231,7 @@ TEST(OutputVacuumLevelTest, OutputVacuumLevel) TEST(PrintForce, PrintForce) { UnitCell ucell; - PARAM.input.test_force = 1; + TestParameters::input().test_force = 1; std::string name = "TOTAL-FORCE"; ModuleBase::matrix force(2, 3); force(0, 0) = 1.0; diff --git a/source/source_io/test/print_info_test.cpp b/source/source_io/test/print_info_test.cpp index 62853c34f3d..2416516401d 100644 --- a/source/source_io/test/print_info_test.cpp +++ b/source/source_io/test/print_info_test.cpp @@ -1,6 +1,5 @@ #include "gtest/gtest.h" #include "gmock/gmock.h" -#define private public #include "source_io/module_parameter/parameter.h" #include "source_cell/klist.h" #include "source_cell/parallel_kpoints.h" @@ -8,12 +7,23 @@ #include "source_io/module_unk/berryphase.h" #include "source_io/module_output/print_info.h" #include "prepare_unitcell.h" -#undef private Magnetism::Magnetism(){} Magnetism::~Magnetism(){} bool berryphase::berry_phase_flag=false; +/// @brief Friend helper to mutate PARAM private members in unit tests. +/// @details Parameter grants friend access to TestParameters so the test can +/// modify input/sys fields without `#define private public`. The class must +/// stay at global scope to match the friend declaration in parameter.h; +/// an anonymous-namespace class would not be the friend. +class TestParameters +{ + public: + static Input_para& input() { return PARAM.input; } + static System_para& sys() { return PARAM.sys; } + static MD_para& mdp() { return PARAM.input.mdp; } +}; /************************************************ * unit test of print_info.cpp @@ -48,12 +58,12 @@ TEST_F(PrintInfoTest, SetupParameters) UcellTestPrepare utp = UcellTestLib["Si"]; ucell = utp.SetUcellInfo(); std::string k_file = "./support/KPT"; - kv->spin_mult = 1; + kv->set_spin_mult(1); const bool gamma_only_local = false; const double kspacing[3] = {0.0, 0.0, 0.0}; const std::string kmesh_type = "gamma"; const double koffset[3] = {0.0, 0.0, 0.0}; - kv->read_kpoints(*ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, GlobalV::ofs_running, GlobalV::ofs_warning, GlobalV::MY_RANK); + kv->read_kpoints_for_testing(*ucell, k_file, gamma_only_local, kspacing, kmesh_type, koffset, GlobalV::ofs_running, GlobalV::ofs_warning, GlobalV::MY_RANK); EXPECT_EQ(kv->get_nkstot(),512); std::vector cal_type = {"scf","relax","cell-relax","md"}; std::vector md_types = {"fire","nve","nvt","npt","langevin","msst"}; @@ -62,56 +72,56 @@ TEST_F(PrintInfoTest, SetupParameters) { if(cal_type[i] != "md") { - PARAM.sys.gamma_only_local = false; - PARAM.input.calculation = cal_type[i]; + TestParameters::sys().gamma_only_local = false; + TestParameters::input().calculation = cal_type[i]; testing::internal::CaptureStdout(); - EXPECT_NO_THROW(ModuleIO::print_parameters(*ucell, *kv, PARAM.input)); + EXPECT_NO_THROW(ModuleIO::print_parameters(*ucell, *kv, PARAM.inp)); output = testing::internal::GetCapturedStdout(); - if(PARAM.input.calculation == "scf") + if(TestParameters::input().calculation == "scf") { EXPECT_THAT(output,testing::HasSubstr("Self-consistent calculations")); } - else if(PARAM.input.calculation == "relax") + else if(TestParameters::input().calculation == "relax") { EXPECT_THAT(output,testing::HasSubstr("Ion relaxation calculations")); } - else if(PARAM.input.calculation == "cell-relax") + else if(TestParameters::input().calculation == "cell-relax") { EXPECT_THAT(output,testing::HasSubstr("Cell relaxation calculations")); } } else { - PARAM.sys.gamma_only_local = true; - PARAM.input.calculation = cal_type[i]; + TestParameters::sys().gamma_only_local = true; + TestParameters::input().calculation = cal_type[i]; for(int j=0; j basis_type = {"lcao","pw","lcao_in_pw"}; for(int i=0; i cal_type = {"scf","nscf","md","relax","cell-relax"}; for(int i=0; i& entry: readinput.get_input_lists()) + { + if (entry.first != "gga_grad") + { + continue; + } + found = true; + ModuleIO::Input_Item item(entry.second); + for (int mode = 0; mode <= 2; ++mode) + { + Parameter param; + item.str_values = {std::to_string(mode)}; + item.read_value(item, param); + EXPECT_EQ(param.inp.gga_grad, mode); + item.check_value(item, param); + } + for (const int mode: {-1, 3}) + { + Parameter param; + item.str_values = {std::to_string(mode)}; + item.read_value(item, param); + EXPECT_EXIT(item.check_value(item, param), testing::ExitedWithCode(1), ""); + } + } + EXPECT_TRUE(found); +} + TEST_F(InputParaTest, TypedTDFieldLists) { ModuleIO::ReadInput readinput(GlobalV::MY_RANK); @@ -528,6 +563,16 @@ TEST_F(InputParaTest, TDFieldStructureErrors) } } +TEST_F(InputParaTest, ParaReadPlumed) +{ + ModuleIO::ReadInput readinput(0); + readinput.check_ntype_flag = false; + Parameter param; + readinput.read_parameters(param, "./support/INPUT.plumed"); + EXPECT_TRUE(param.inp.mdp.plumed); + EXPECT_EQ(param.inp.mdp.plumed_file, "my_plumed.dat"); +} + TEST_F(InputParaTest, DiagoProc) { int rank = 0; diff --git a/source/source_io/test/read_wf2rho_pw_test.cpp b/source/source_io/test/read_wf2rho_pw_test.cpp index 38f586f3a5e..991a46a8f26 100644 --- a/source/source_io/test/read_wf2rho_pw_test.cpp +++ b/source/source_io/test/read_wf2rho_pw_test.cpp @@ -3,13 +3,13 @@ #undef __LCAO -#define private public #include "source_base/module_out/filename.h" // mohan add 2025-05-17 #include "source_base/parallel_grid.h" #include "source_cell/klist.h" #include "source_cell/unitcell.h" #include "source_estate/module_charge/charge.h" -#include "source_estate/module_charge/symm_rho.h" +#include "source_estate/module_charge/chg_parallel.h" +#include "source_estate/module_charge/chg_symm.h" #include "source_hamilt/module_xc/xc_functional.h" #include "source_io/module_parameter/parameter.h" #include "source_io/module_wf/read_wf2rho_pw.h" @@ -22,12 +22,18 @@ #include "source_basis/module_pw/test/test_tool.h" #endif -Charge::Charge() +/// @brief Friend helper to mutate PARAM private members in unit tests. +/// @details Parameter grants friend access to TestParameters so the test can +/// modify input/sys fields without `#define private public`. The class must +/// stay at global scope to match the friend declaration in parameter.h; +/// an anonymous-namespace class would not be the friend. +class TestParameters { -} -Charge::~Charge() -{ -} + public: + static Input_para& input() { return PARAM.input; } + static System_para& sys() { return PARAM.sys; } +}; + UnitCell::UnitCell() { } @@ -55,27 +61,27 @@ Sep_Cell::~Sep_Cell() noexcept int XC_Functional::func_type = 0; bool XC_Functional::ked_flag = false; -Symmetry_rho::Symmetry_rho() +namespace module_charge { -} -Symmetry_rho::~Symmetry_rho() -{ -} -void Symmetry_rho::begin(const int& spin_now, const Charge& CHR, const ModulePW::PW_Basis* rho_basis, ModuleSymmetry::Symmetry& symm) const +void cal_rhog_symm(const int& spin_now, + const Charge& CHR, + const ModulePW::PW_Basis* rho_basis, + ModuleSymmetry::Symmetry& symm) { return; } +} // namespace module_charge void cal_ik2iktot(std::vector& ik2iktot, const int& nks, const int& nkstot) { - if (PARAM.inp.kpar == 1) + if (TestParameters::input().kpar == 1) { for (int ik = 0; ik < nks; ++ik) { ik2iktot[ik] = ik; } } - else if (PARAM.inp.kpar == 2) + else if (TestParameters::input().kpar == 2) { if (GlobalV::MY_POOL == 0) { @@ -117,7 +123,7 @@ class ReadWfcRhoTest : public ::testing::Test rhopw = new ModulePW::PW_Basis; kv = new K_Vectors; // output .dat file - PARAM.input.out_wfc_pw = 2; + TestParameters::input().out_wfc_pw = 2; #ifdef __MPI MPI_Comm_size(MPI_COMM_WORLD, &world_size); #endif @@ -229,25 +235,18 @@ TEST_F(ReadWfcRhoTest, ReadWfcRho) //---------------------------------------- // Initialize charge density //---------------------------------------- - chg.rho = new double*[nspin]; - chg._space_rho = new double[rhopw->nrxx]; - chg.rho[0] = chg._space_rho; - ModuleBase::GlobalFunc::ZEROS(chg.rho[0], rhopw->nrxx); - chg.rhopw = rhopw; - chg.nrxx = rhopw->nrxx; + chg.set_rhopw(rhopw); + chg.allocate(nspin, false, false, 0); chg.pgrid = &pgrid; //---------------------------------------- // set charge_ref //---------------------------------------- Charge chg_ref; - chg_ref.rho = new double*[nspin]; - chg_ref._space_rho = new double[rhopw->nrxx]; - chg_ref.rho[0] = chg_ref._space_rho; - ModuleBase::GlobalFunc::ZEROS(chg_ref.rho[0], rhopw->nrxx); - std::vector> rho_tmp(rhopw->nrxx); - chg_ref.nrxx = rhopw->nrxx; + chg_ref.set_rhopw(rhopw); + chg_ref.allocate(nspin, false, false, 0); chg_ref.pgrid = &pgrid; + std::vector> rho_tmp(rhopw->nrxx); for (int ik = 0; ik < nks; ++ik) { @@ -266,7 +265,8 @@ TEST_F(ReadWfcRhoTest, ReadWfcRho) } #ifdef __MPI - chg_ref.reduce_diff_pools(chg_ref.rho[0]); + module_charge::reduce_diff_pools(chg_ref.rho[0], chg_ref, GlobalV::KPAR, + PARAM.globalv.all_ks_run, PARAM.inp.bndpar); #endif // for spin=1 or 2, npol=1 @@ -297,7 +297,7 @@ TEST_F(ReadWfcRhoTest, ReadWfcRho) npol, GlobalV::RANK_IN_POOL, GlobalV::NPROC_IN_POOL, - PARAM.input.out_wfc_pw, + TestParameters::input().out_wfc_pw, ecutwfc, out_dir, *psi, @@ -347,10 +347,6 @@ TEST_F(ReadWfcRhoTest, ReadWfcRho) // } } - delete[] chg.rho; - delete[] chg._space_rho; - delete[] chg_ref.rho; - delete[] chg_ref._space_rho; delete psi; if (GlobalV::MY_RANK == 0) @@ -376,12 +372,12 @@ int main(int argc, char** argv) setupmpi(argc, argv, GlobalV::NPROC, GlobalV::MY_RANK); // when kpar == 2, nspin == 2 - PARAM.input.kpar = (GlobalV::NPROC > 1) ? 2 : 1; - GlobalV::KPAR = PARAM.input.kpar; - PARAM.input.bndpar = 1; + TestParameters::input().kpar = (GlobalV::NPROC > 1) ? 2 : 1; + GlobalV::KPAR = TestParameters::input().kpar; + TestParameters::input().bndpar = 1; Parallel_Global::divide_pools(GlobalV::NPROC, GlobalV::MY_RANK, - PARAM.inp.bndpar, + TestParameters::input().bndpar, GlobalV::KPAR, GlobalV::NPROC_IN_BNDGROUP, GlobalV::RANK_IN_BPGROUP, diff --git a/source/source_io/test/read_wfc_nao_test.cpp b/source/source_io/test/read_wfc_nao_test.cpp index 1e53b779003..6107db23036 100644 --- a/source/source_io/test/read_wfc_nao_test.cpp +++ b/source/source_io/test/read_wfc_nao_test.cpp @@ -1,8 +1,5 @@ #include "gtest/gtest.h" #include "gmock/gmock.h" -#define private public -#include "source_io/module_parameter/parameter.h" -#undef private #include "source_io/module_wf/read_wfc_nao.h" #include "source_basis/module_ao/parallel_orbitals.h" #include "source_io/module_wf/write_wfc_nao.h" @@ -81,7 +78,7 @@ TEST_F(ReadWfcNaoTest,ReadWfcNao) //Global variables const int nbands = 3; const int nlocal = 3; - PARAM.sys.global_readin_dir = "./support/"; + const std::string readin_dir = "./support/"; const int nks = 1; const int nspin = 1; int my_rank = 0; @@ -109,7 +106,7 @@ TEST_F(ReadWfcNaoTest,ReadWfcNao) const int nkstot = 1; // Act - ModuleIO::read_wfc_nao(PARAM.sys.global_readin_dir, ParaV, psid, + ModuleIO::read_wfc_nao(readin_dir, ParaV, psid, ekb, wg, ik2iktot, nkstot, nspin, false); // Assert EXPECT_NEAR(ekb(0,1),0.31482195194888534794941393,1e-5); @@ -126,7 +123,7 @@ TEST_F(ReadWfcNaoTest, ReadWfcNaoPart) const int nbands = 2; const int skip_band = 1; const int nlocal = 3; - PARAM.sys.global_readin_dir = "./support/"; + const std::string readin_dir = "./support/"; const int nks = 1; const int nspin = 1; const int nstep = -1; @@ -155,7 +152,7 @@ TEST_F(ReadWfcNaoTest, ReadWfcNaoPart) const int nkstot = 1; // Act - ModuleIO::read_wfc_nao(PARAM.sys.global_readin_dir, ParaV, psid, + ModuleIO::read_wfc_nao(readin_dir, ParaV, psid, ekb, wg, ik2iktot, nkstot, nspin, false, skip_band, nstep); // Assert diff --git a/source/source_io/test/read_wfc_pw_test.cpp b/source/source_io/test/read_wfc_pw_test.cpp index c5602454f99..5aec31d16fc 100644 --- a/source/source_io/test/read_wfc_pw_test.cpp +++ b/source/source_io/test/read_wfc_pw_test.cpp @@ -1,8 +1,6 @@ #include "source_io/module_wf/read_wfc_pw.h" -#define private public #include "source_io/module_parameter/parameter.h" -#undef private #include "gmock/gmock.h" #include "gtest/gtest.h" #ifdef __MPI @@ -11,6 +9,18 @@ #include "mpi.h" #endif +/// @brief Friend helper to mutate PARAM private members in unit tests. +/// @details Parameter grants friend access to TestParameters so the test can +/// modify input/sys fields without `#define private public`. The class must +/// stay at global scope to match the friend declaration in parameter.h; +/// an anonymous-namespace class would not be the friend. +class TestParameters +{ + public: + static Input_para& input() { return PARAM.input; } + static System_para& sys() { return PARAM.sys; } +}; + /** * - Tested Functions: * - read_wfc_pw() @@ -52,13 +62,13 @@ TEST_F(ReadWfcPwTest, ReadWfcPw) wfcpw->setuptransform(); wfcpw->collect_local_pw(); - PARAM.input.nbands = 8; + TestParameters::input().nbands = 8; const int nbasis = wfcpw->npwk[0]; - ModuleBase::ComplexMatrix wfcatom(PARAM.input.nbands, nbasis); + ModuleBase::ComplexMatrix wfcatom(TestParameters::input().nbands, nbasis); const int ik = 0; const int ik_tot = 0; - ModuleIO::read_wfc_pw(filename, wfcpw, + ModuleIO::read_wfc_pw(filename, wfcpw, GlobalV::RANK_IN_POOL, GlobalV::NPROC_IN_POOL, PARAM.inp.nbands, PARAM.globalv.npol, ik, ik_tot, nkstot, wfcatom); @@ -125,7 +135,7 @@ TEST_F(ReadWfcPwTest, NotFoundFile) wfcpw->setuptransform(); wfcpw->collect_local_pw(); - ModuleBase::ComplexMatrix wfcatom(PARAM.input.nbands, wfcpw->npwk[0]); + ModuleBase::ComplexMatrix wfcatom(TestParameters::input().nbands, wfcpw->npwk[0]); if(GlobalV::RANK_IN_POOL == 0) { @@ -185,9 +195,9 @@ TEST_F(ReadWfcPwTest, InconsistentBands) wfcpw->setuptransform(); wfcpw->collect_local_pw(); - PARAM.input.nbands = 4; + TestParameters::input().nbands = 4; const int nbasis = wfcpw->npwk[0]; - ModuleBase::ComplexMatrix wfcatom(PARAM.input.nbands, nbasis); + ModuleBase::ComplexMatrix wfcatom(TestParameters::input().nbands, nbasis); testing::internal::CaptureStdout(); const int ik = 0; @@ -226,15 +236,15 @@ TEST_F(ReadWfcPwTest, InconsistentKvec) wfcpw->setuptransform(); wfcpw->collect_local_pw(); - PARAM.input.nbands = 8; + TestParameters::input().nbands = 8; const int nbasis = wfcpw->npwk[0]; - ModuleBase::ComplexMatrix wfcatom(PARAM.input.nbands, nbasis); + ModuleBase::ComplexMatrix wfcatom(TestParameters::input().nbands, nbasis); testing::internal::CaptureStdout(); const int ik=0; const int ik_tot=0; EXPECT_EXIT( - ModuleIO::read_wfc_pw(filename, wfcpw, + ModuleIO::read_wfc_pw(filename, wfcpw, GlobalV::RANK_IN_POOL, GlobalV::NPROC_IN_POOL, PARAM.inp.nbands, PARAM.globalv.npol, ik, ik_tot, nkstot, wfcatom), @@ -264,9 +274,9 @@ TEST_F(ReadWfcPwTest, InconsistentLat0) wfcpw->setuptransform(); wfcpw->collect_local_pw(); - PARAM.input.nbands = 8; + TestParameters::input().nbands = 8; const int nbasis = wfcpw->npwk[0]; - ModuleBase::ComplexMatrix wfcatom(PARAM.input.nbands, nbasis); + ModuleBase::ComplexMatrix wfcatom(TestParameters::input().nbands, nbasis); testing::internal::CaptureStdout(); const int ik=0; @@ -301,9 +311,9 @@ TEST_F(ReadWfcPwTest, InconsistentG) wfcpw->setuptransform(); wfcpw->collect_local_pw(); - PARAM.input.nbands = 8; + TestParameters::input().nbands = 8; const int nbasis = wfcpw->npwk[0]; - ModuleBase::ComplexMatrix wfcatom(PARAM.input.nbands, nbasis); + ModuleBase::ComplexMatrix wfcatom(TestParameters::input().nbands, nbasis); testing::internal::CaptureStdout(); const int ik=0; diff --git a/source/source_io/test/single_r_io_test.cpp b/source/source_io/test/single_r_io_test.cpp deleted file mode 100644 index 8cc1981aba0..00000000000 --- a/source/source_io/test/single_r_io_test.cpp +++ /dev/null @@ -1,227 +0,0 @@ -#include "gtest/gtest.h" -#include "gmock/gmock.h" -#define private public -#include "source_io/module_parameter/parameter.h" -#undef private -#include "source_io/module_hs/single_r_io.h" -#include "source_base/global_variable.h" -#include "source_basis/module_ao/parallel_orbitals.h" -#include -#include -#include -#include -#include -/************************************************ - * unit test of output_single_R - ***********************************************/ -/** - * - Tested Functions: - * - ModuleIO::output_single_R - * - output single R data - */ -Parallel_Orbitals::Parallel_Orbitals() -{ -} - -Parallel_Orbitals::~Parallel_Orbitals() -{ -} - -void Parallel_2D::set_serial(const int M_A, const int N_A) -{ - this->nrow = M_A; - this->ncol = N_A; - this->is_serial = true; - this->global2local_row_.resize(M_A); - this->global2local_row_[0] = 0; - this->global2local_row_[1] = 1; - this->global2local_row_[2] = -1; - this->global2local_row_[3] = 2; - this->global2local_row_[4] = -1; //Some rows have global2local_row_ < 0 -} - -int Parallel_2D::get_global_row_size() const -{ - return this->nrow; -} - -TEST(ModuleIOTest, OutputSingleR) -{ - // Create temporary output file - std::stringstream ofs_filename; - GlobalV::DRANK=0; - ofs_filename << "test_output_single_R_" << GlobalV::DRANK << ".dat"; - std::ofstream ofs(ofs_filename.str()); - - // Define input parameters - const double sparse_threshold = 1e-8; - const bool binary = false; - Parallel_Orbitals pv; - PARAM.sys.nlocal = 99; - pv.set_serial(5, 5); - std::map> XR = { - {0, {{1, 0.5}, {3, 0.3}}}, - {1, {{0, 0.2}, {2, 0.4}}}, - {3, {{1, 0.1}, {4, 0.7}}} - }; - ModuleIO::SparseWriteOptions options; - options.threshold = sparse_threshold; - options.binary = binary; - options.reduce = true; - options.temp_dir = "./"; - - // Call function under test - ModuleIO::output_single_R(ofs, XR, pv, options); - - // Close output file and open it for reading - ofs.close(); - std::ifstream ifs; - ifs.open("test_output_single_R_0.dat"); - std::string str((std::istreambuf_iterator(ifs)),std::istreambuf_iterator()); - std::istringstream content(str); - std::string value_line; - std::string column_line; - std::string indptr_line; - std::getline(content, value_line); - std::getline(content, column_line); - std::getline(content, indptr_line); - - std::istringstream value_stream(value_line); - std::vector values; - double value = 0.0; - while (value_stream >> value) - { - values.push_back(value); - } - ASSERT_EQ(values.size(), 6); - EXPECT_DOUBLE_EQ(values[0], 0.5); - EXPECT_DOUBLE_EQ(values[1], 0.3); - EXPECT_DOUBLE_EQ(values[2], 0.2); - EXPECT_DOUBLE_EQ(values[3], 0.4); - EXPECT_DOUBLE_EQ(values[4], 0.1); - EXPECT_DOUBLE_EQ(values[5], 0.7); - - std::istringstream column_stream(column_line); - std::vector columns; - int column = 0; - while (column_stream >> column) - { - columns.push_back(column); - } - EXPECT_THAT(columns, testing::ElementsAre(1, 3, 0, 2, 1, 4)); - - std::istringstream indptr_stream(indptr_line); - std::vector indptr; - long long index = 0; - while (indptr_stream >> index) - { - indptr.push_back(index); - } - EXPECT_THAT(indptr, testing::ElementsAre(0, 2, 4, 4, 6, 6)); - std::remove("test_output_single_R_0.dat"); -} - -TEST(ModuleIOTest, OutputSingleRComplexKeepsHighPrecision) -{ - const std::string filename = "test_output_single_R_complex.dat"; - std::remove(filename.c_str()); - GlobalV::DRANK = 0; - std::ofstream ofs(filename); - - Parallel_Orbitals pv; - pv.set_serial(5, 5); - ModuleIO::SparseRBlock> XR = { - {0, {{1, std::complex(1.234567890123456, -2.345678901234567)}}} - }; - ModuleIO::SparseWriteOptions options; - options.threshold = 1e-12; - options.binary = false; - options.reduce = false; - options.temp_dir = "./"; - - ModuleIO::output_single_R(ofs, XR, pv, options); - ofs.close(); - - std::ifstream ifs(filename); - const std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); - EXPECT_THAT(output, testing::HasSubstr("(1.2345678901234560e+00,-2.3456789012345669e+00)")); - EXPECT_THAT(output, testing::Not(testing::HasSubstr("(1.23457,-2.34568)"))); - - std::remove(filename.c_str()); -} - -TEST(ModuleIOTest, OutputSingleRUsesConfiguredPrecision) -{ - const std::string filename = "test_output_single_R_precision.dat"; - std::remove(filename.c_str()); - GlobalV::DRANK = 0; - std::ofstream ofs(filename); - - Parallel_Orbitals pv; - pv.set_serial(5, 5); - ModuleIO::SparseRBlock> XR = { - {0, {{1, std::complex(1.234567890123456, -2.5)}}} - }; - ModuleIO::SparseWriteOptions options; - options.threshold = 1e-12; - options.binary = false; - options.precision = 8; - options.reduce = false; - options.temp_dir = "./"; - - ModuleIO::output_single_R(ofs, XR, pv, options); - ofs.close(); - - std::ifstream ifs(filename); - const std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); - EXPECT_THAT(output, testing::HasSubstr("(1.23456789e+00,-2.50000000e+00)")); - EXPECT_THAT(output, testing::Not(testing::HasSubstr("1.2345678901234560e+00"))); - - std::remove(filename.c_str()); -} - -void write_out_of_range_sparse_column(const char* filename) -{ - GlobalV::DRANK = 0; - std::ofstream ofs(filename); - Parallel_Orbitals pv; - pv.set_serial(5, 5); - ModuleIO::SparseRBlock XR; - XR[0][5] = 1.0; - ModuleIO::SparseWriteOptions options; - options.threshold = 1e-12; - options.binary = false; - options.reduce = false; - options.temp_dir = "/tmp/"; - ModuleIO::output_single_R(ofs, XR, pv, options); -} - -TEST(ModuleIOTest, OutputSingleRRejectsOutOfRangeColumn) -{ - const char* filename = "/tmp/test_output_single_R_invalid.dat"; - std::remove(filename); - EXPECT_EXIT( - write_out_of_range_sparse_column(filename), - ::testing::ExitedWithCode(1), - "Sparse column index out of range"); - std::remove(filename); -} - -int main(int argc, char **argv) -{ - -#ifdef __MPI - MPI_Init(&argc, &argv); - MPI_Comm_size(MPI_COMM_WORLD,&GlobalV::NPROC); - MPI_Comm_rank(MPI_COMM_WORLD,&GlobalV::MY_RANK); -#endif - - testing::InitGoogleTest(&argc, argv); - int result = RUN_ALL_TESTS(); - -#ifdef __MPI - MPI_Finalize(); -#endif - - return result; -} diff --git a/source/source_io/test/support/INPUT b/source/source_io/test/support/INPUT index df78fb591ee..0578da97f4a 100644 --- a/source/source_io/test/support/INPUT +++ b/source/source_io/test/support/INPUT @@ -170,7 +170,7 @@ mixing_gg0 0 #mixing parameter in kerker #Parameters (8.DOS) dos_emin_ev -15 #minimal range for dos dos_emax_ev 15 #maximal range for dos -dos_edelta_ev 0.01 #delta energy for dos +dos_edelta_ev 0.03 #delta energy for dos dos_scale 0.01 #scale dos range by dos_sigma 0.07 #gauss b coefficeinet(default=0.07) dos_nche 100 #orders of Chebyshev expansions for dos @@ -190,7 +190,7 @@ md_restartfreq 5 #The period to output MD restart information md_out_force 0 #output all atomic forces into running_md.log or not md_seed -1 #random seed for MD md_prec_level 2 #precision level for vc-md -ref_cell_factor 1.2 #construct a reference cell bigger than the initial cell +ref_cell_factor 1.0 #construct a reference cell bigger than the initial cell (currently disabled, must be 1.0) md_restart 0 #whether restart lj_rule 2 #combination rules used to construct the parameter matrix for LJ potential lj_eshift 0 #whether to use energy shift for LJ potential @@ -396,3 +396,5 @@ sccut 4 #Maximal step size for lambda in eV/uB #Parameters (23. Time-dependent orbital-free DFT) of_cd 0 #0: no CD potential; 1: add CD potential of_mCD_alpha 1.0 # parameter of modified CD potential + +gga_grad 2 diff --git a/source/source_io/test/support/INPUT.plumed b/source/source_io/test/support/INPUT.plumed new file mode 100644 index 00000000000..f668e7f863b --- /dev/null +++ b/source/source_io/test/support/INPUT.plumed @@ -0,0 +1,3 @@ +INPUT_PARAMETERS +plumed 1 +plumed_file my_plumed.dat diff --git a/source/source_io/test/td_efield_io_test.cpp b/source/source_io/test/td_efield_io_test.cpp index c0fb6be358a..3cb3a1ddaa3 100644 --- a/source/source_io/test/td_efield_io_test.cpp +++ b/source/source_io/test/td_efield_io_test.cpp @@ -95,9 +95,9 @@ TEST_F(TDEFieldIOTest, FreshCalculationTruncatesAndUsesOneBasedFiles) EXPECT_TRUE(read_samples(1).empty()); EXPECT_TRUE(read_samples(2).empty()); - manager->advance_length_gauge(); + manager->prepare_sample(0); ModuleIO::write_td_field_values(*manager, output_prefix_); - manager->advance_length_gauge(); + manager->prepare_sample(1); ModuleIO::write_td_field_values(*manager, output_prefix_); const std::vector> samples_1 = read_samples(1); @@ -124,7 +124,8 @@ TEST_F(TDEFieldIOTest, RestartPreservesExistingSamples) std::shared_ptr manager = create_manager({4.0}); ModuleIO::prepare_td_field_output(output_prefix_, manager->fields().size(), true); - manager->advance_length_gauge(); + const int next_step = manager->current_step() + 1; + manager->prepare_sample(next_step); ModuleIO::write_td_field_values(*manager, output_prefix_); const std::vector> samples = read_samples(1); @@ -146,7 +147,8 @@ TEST_F(TDEFieldIOTest, ReportsOutputOpenFailures) EXPECT_EXIT(ModuleIO::prepare_td_field_output(invalid_directory, manager->fields().size(), false), testing::ExitedWithCode(1), ""); - manager->advance_length_gauge(); + const int next_step = manager->current_step() + 1; + manager->prepare_sample(next_step); EXPECT_EXIT(ModuleIO::write_td_field_values(*manager, invalid_directory), testing::ExitedWithCode(1), ""); } diff --git a/source/source_io/test/tmp_mocks.cpp b/source/source_io/test/tmp_mocks.cpp index 23091e6f686..bcb2060a70f 100644 --- a/source/source_io/test/tmp_mocks.cpp +++ b/source/source_io/test/tmp_mocks.cpp @@ -1,5 +1,6 @@ #include "source_cell/unitcell.h" +#include "source_basis/module_ao/parallel_orbitals.h" // constructor of Atom Atom::Atom() @@ -49,3 +50,10 @@ UnitCell::UnitCell() UnitCell::~UnitCell() { } + +Parallel_Orbitals::Parallel_Orbitals() +{ +} +Parallel_Orbitals::~Parallel_Orbitals() +{ +} diff --git a/source/source_io/test/to_qo_test.cpp b/source/source_io/test/to_qo_test.cpp index b972e9da04c..03ac9cb0baf 100644 --- a/source/source_io/test/to_qo_test.cpp +++ b/source/source_io/test/to_qo_test.cpp @@ -1,9 +1,7 @@ #include #include "source_io/module_qo/to_qo.h" #include "source_base/output.h" -#define private public #include "source_io/module_parameter/parameter.h" -#undef private #ifdef __MPI #include @@ -20,6 +18,18 @@ pseudo::~pseudo() {} Magnetism::Magnetism() {} Magnetism::~Magnetism() {} +/// @brief Friend helper to mutate PARAM private members in unit tests. +/// @details Parameter grants friend access to TestParameters so the test can +/// modify input/sys fields without `#define private public`. The class must +/// stay at global scope to match the friend declaration in parameter.h; +/// an anonymous-namespace class would not be the friend. +class TestParameters +{ + public: + static Input_para& input() { return PARAM.input; } + static System_para& sys() { return PARAM.sys; } +}; + void define_fcc_cell(UnitCell& ucell) { ucell.atoms = new Atom[2]; @@ -56,9 +66,9 @@ void define_fcc_cell(UnitCell& ucell) ucell.pseudo_fn[0] = "../../../../tests/PP_ORB/Si_dojo_soc.upf"; ucell.pseudo_fn[1] = "../../../../tests/PP_ORB/C.LDA.UPF"; - PARAM.sys.global_out_dir = "./"; - PARAM.input.qo_screening_coeff = {0.1, 0.1}; - PARAM.input.qo_thr = 1e-6; + TestParameters::sys().global_out_dir = "./"; + TestParameters::input().qo_screening_coeff = {0.1, 0.1}; + TestParameters::input().qo_thr = 1e-6; // GlobalV::ofs_running = std::ofstream("unittest.log"); GlobalV::MY_RANK = 0; GlobalV::NPROC = 1; @@ -91,9 +101,9 @@ void define_sc_cell(UnitCell& ucell) ucell.pseudo_fn.resize(1); ucell.pseudo_fn[0] = "../../../../tests/PP_ORB/Si_dojo_soc.upf"; - PARAM.sys.global_out_dir = "./"; - PARAM.input.qo_screening_coeff = {0.1}; - PARAM.input.qo_thr = 1e-6; + TestParameters::sys().global_out_dir = "./"; + TestParameters::input().qo_screening_coeff = {0.1}; + TestParameters::input().qo_thr = 1e-6; // GlobalV::ofs_running = std::ofstream("unittest.log"); GlobalV::MY_RANK = 0; GlobalV::NPROC = 1; @@ -121,8 +131,8 @@ TEST_F(toQOTest, Constructor) define_fcc_cell(ucell); toQO tqo("hydrogen", {"minimal-nodeless", "minimal-nodeless"}, - PARAM.input.qo_thr, - PARAM.input.qo_screening_coeff); + TestParameters::input().qo_thr, + TestParameters::input().qo_screening_coeff); EXPECT_EQ(tqo.qo_basis(), "hydrogen"); EXPECT_EQ(tqo.strategy(0), "minimal-nodeless"); EXPECT_EQ(tqo.strategy(1), "minimal-nodeless"); @@ -136,8 +146,8 @@ TEST_F(toQOTest, ReadStructures) toQO tqo("hydrogen", {"minimal-nodeless", "minimal-nodeless"}, - PARAM.input.qo_thr, - PARAM.input.qo_screening_coeff); + TestParameters::input().qo_thr, + TestParameters::input().qo_screening_coeff); std::vector> kvecs_d; kvecs_d.push_back(ModuleBase::Vector3(-0.25, -0.25, -0.25)); // pair 1 kvecs_d.push_back(ModuleBase::Vector3(0.25, 0.25, 0.25)); @@ -175,8 +185,8 @@ TEST_F(toQOTest, BuildNao) define_fcc_cell(ucell); toQO tqo("hydrogen", {"minimal-nodeless", "minimal-nodeless"}, - PARAM.input.qo_thr, - PARAM.input.qo_screening_coeff); + TestParameters::input().qo_thr, + TestParameters::input().qo_screening_coeff); std::vector> kvecs_d; kvecs_d.push_back(ModuleBase::Vector3(0.0, 0.0, 0.0)); // Gamma point tqo.read_structures(&ucell, kvecs_d, 0, 1); @@ -190,8 +200,8 @@ TEST_F(toQOTest, RadialCollectionIndexing) define_fcc_cell(ucell); toQO tqo("hydrogen", {"minimal-nodeless", "minimal-nodeless"}, - PARAM.input.qo_thr, - PARAM.input.qo_screening_coeff); + TestParameters::input().qo_thr, + TestParameters::input().qo_screening_coeff); std::vector> kvecs_d; kvecs_d.push_back(ModuleBase::Vector3(0.0, 0.0, 0.0)); // Gamma point tqo.read_structures(&ucell, kvecs_d, 0, 1); @@ -311,7 +321,7 @@ TEST_F(toQOTest, RadialCollectionIndexing) "./", ucell.pseudo_fn.data(), {}, - PARAM.input.qo_thr, + TestParameters::input().qo_thr, GlobalV::ofs_running, 0); EXPECT_EQ(tqo.p_ao()->nchi(), 5); // Si: 1s, 2p, 3d, C: 1s, 2p @@ -356,8 +366,8 @@ TEST_F(toQOTest, BuildHydrogenMinimal) define_fcc_cell(ucell); toQO tqo("hydrogen", {"minimal-nodeless", "minimal-nodeless"}, - PARAM.input.qo_thr, - PARAM.input.qo_screening_coeff); + TestParameters::input().qo_thr, + TestParameters::input().qo_screening_coeff); std::vector> kvecs_d; kvecs_d.push_back(ModuleBase::Vector3(0.0, 0.0, 0.0)); // Gamma point tqo.read_structures(&ucell, kvecs_d, 0, 1); @@ -365,7 +375,7 @@ TEST_F(toQOTest, BuildHydrogenMinimal) "./", ucell.pseudo_fn.data(), {}, - PARAM.input.qo_thr, + TestParameters::input().qo_thr, GlobalV::ofs_running, 0); EXPECT_EQ(tqo.p_ao()->nchi(), 5); // Si: 1s, 2p, 3d, C: 1s, 2p @@ -379,8 +389,8 @@ TEST_F(toQOTest, Norm2RijSupercell) define_fcc_cell(ucell); toQO tqo("hydrogen", {"minimal-nodeless", "minimal-nodeless"}, - PARAM.input.qo_thr, - PARAM.input.qo_screening_coeff); + TestParameters::input().qo_thr, + TestParameters::input().qo_screening_coeff); std::vector> kvecs_d; kvecs_d.push_back(ModuleBase::Vector3(0.0, 0.0, 0.0)); // Gamma point tqo.read_structures(&ucell, kvecs_d, 0, 1); @@ -400,8 +410,8 @@ TEST_F(toQOTest, ScanSupercellForAtom) define_fcc_cell(ucell); toQO tqo("hydrogen", {"minimal-nodeless", "minimal-nodeless"}, - PARAM.input.qo_thr, - PARAM.input.qo_screening_coeff); + TestParameters::input().qo_thr, + TestParameters::input().qo_screening_coeff); std::vector> kvecs_d; kvecs_d.push_back(ModuleBase::Vector3(0.0, 0.0, 0.0)); // Gamma point tqo.read_structures(&ucell, kvecs_d, 0, 1); @@ -418,7 +428,7 @@ TEST_F(toQOTest, ScanSupercellForAtom) "./", ucell.pseudo_fn.data(), {}, - PARAM.input.qo_thr, + TestParameters::input().qo_thr, GlobalV::ofs_running, 0); std::vector> n1n2n3 = tqo.scan_supercell_for_atom(0, 0); @@ -443,8 +453,8 @@ TEST_F(toQOTest, EliminateDuplicateVector3) v.push_back(ModuleBase::Vector3(1, 1, 1)); toQO tqo("hydrogen", {"minimal-nodeless", "minimal-nodeless"}, - PARAM.input.qo_thr, - PARAM.input.qo_screening_coeff); + TestParameters::input().qo_thr, + TestParameters::input().qo_screening_coeff); tqo.eliminate_duplicate_vector3(v); EXPECT_EQ(v.size(), 4); } @@ -454,8 +464,8 @@ TEST_F(toQOTest, ScanSupercellFCC) define_fcc_cell(ucell); toQO tqo("hydrogen", {"minimal-nodeless", "minimal-nodeless"}, - PARAM.input.qo_thr, - PARAM.input.qo_screening_coeff); + TestParameters::input().qo_thr, + TestParameters::input().qo_screening_coeff); std::vector> kvecs_d; kvecs_d.push_back(ModuleBase::Vector3(0.0, 0.0, 0.0)); // Gamma point tqo.read_structures(&ucell, kvecs_d, 0, 1); @@ -467,7 +477,7 @@ TEST_F(toQOTest, ScanSupercellFCC) "./", ucell.pseudo_fn.data(), {}, - PARAM.input.qo_thr, + TestParameters::input().qo_thr, GlobalV::ofs_running, 0); tqo.scan_supercell(0, 1); @@ -479,8 +489,8 @@ TEST_F(toQOTest, ScanSupercellSC1) define_sc_cell(ucell); toQO tqo("hydrogen", {"minimal-nodeless"}, - PARAM.input.qo_thr, - PARAM.input.qo_screening_coeff); + TestParameters::input().qo_thr, + TestParameters::input().qo_screening_coeff); std::vector> kvecs_d; kvecs_d.push_back(ModuleBase::Vector3(0.0, 0.0, 0.0)); // Gamma point tqo.read_structures(&ucell, kvecs_d, 0, 1); @@ -488,12 +498,12 @@ TEST_F(toQOTest, ScanSupercellSC1) "./", ucell.orbital_fn.data(), 0); - PARAM.input.qo_thr = 1e-6; + TestParameters::input().qo_thr = 1e-6; tqo.build_ao(ucell.ntype, "./", ucell.pseudo_fn.data(), {}, - PARAM.input.qo_thr, + TestParameters::input().qo_thr, GlobalV::ofs_running, 0); tqo.scan_supercell(0, 1); @@ -505,8 +515,8 @@ TEST_F(toQOTest, AllocateOvlpMinimal) define_fcc_cell(ucell); toQO tqo("hydrogen", {"minimal-nodeless", "minimal-nodeless"}, - PARAM.input.qo_thr, - PARAM.input.qo_screening_coeff); + TestParameters::input().qo_thr, + TestParameters::input().qo_screening_coeff); std::vector> kvecs_d; kvecs_d.push_back(ModuleBase::Vector3(0.0, 0.0, 0.0)); // Gamma point tqo.read_structures(&ucell, kvecs_d, 0, 1); @@ -523,7 +533,7 @@ TEST_F(toQOTest, AllocateOvlpMinimal) "./", ucell.pseudo_fn.data(), {}, - PARAM.input.qo_thr, + TestParameters::input().qo_thr, GlobalV::ofs_running, 0); tqo.scan_supercell(0, 1); @@ -545,14 +555,14 @@ TEST_F(toQOTest, AllocateOvlpMinimal) TEST_F(toQOTest, Initialize) { define_fcc_cell(ucell); - PARAM.input.qo_screening_coeff = {}; + TestParameters::input().qo_screening_coeff = {}; toQO tqo("hydrogen", {"minimal-nodeless", "minimal-nodeless"}, - PARAM.input.qo_thr, - PARAM.input.qo_screening_coeff); + TestParameters::input().qo_thr, + TestParameters::input().qo_screening_coeff); std::vector> kvecs_d; kvecs_d.push_back(ModuleBase::Vector3(0.0, 0.0, 0.0)); // Gamma point - tqo.initialize(PARAM.sys.global_out_dir, + tqo.initialize(TestParameters::sys().global_out_dir, "", "", &ucell, @@ -564,14 +574,14 @@ TEST_F(toQOTest, Initialize) TEST_F(toQOTest, ReadOvlp) { define_fcc_cell(ucell); - PARAM.input.qo_screening_coeff = {}; + TestParameters::input().qo_screening_coeff = {}; toQO tqo("hydrogen", {"minimal-nodeless", "minimal-nodeless"}, - PARAM.input.qo_thr, - PARAM.input.qo_screening_coeff); + TestParameters::input().qo_thr, + TestParameters::input().qo_screening_coeff); std::vector> kvecs_d; kvecs_d.push_back(ModuleBase::Vector3(0.0, 0.0, 0.0)); // Gamma point - tqo.initialize(PARAM.sys.global_out_dir, + tqo.initialize(TestParameters::sys().global_out_dir, "", "", &ucell, @@ -597,15 +607,15 @@ TEST_F(toQOTest, ReadOvlp) TEST_F(toQOTest, CalculateOvlpR) { define_fcc_cell(ucell); - PARAM.input.qo_screening_coeff = {}; - PARAM.input.qo_thr = 1e-10; + TestParameters::input().qo_screening_coeff = {}; + TestParameters::input().qo_thr = 1e-10; toQO tqo("hydrogen", {"minimal-nodeless", "minimal-nodeless"}, - PARAM.input.qo_thr, - PARAM.input.qo_screening_coeff); + TestParameters::input().qo_thr, + TestParameters::input().qo_screening_coeff); std::vector> kvecs_d; kvecs_d.push_back(ModuleBase::Vector3(0.0, 0.0, 0.0)); // Gamma point - tqo.initialize(PARAM.sys.global_out_dir, + tqo.initialize(TestParameters::sys().global_out_dir, "", "", &ucell, @@ -641,18 +651,18 @@ TEST_F(toQOTest, CalculateOvlpR) TEST_F(toQOTest, CalculateSelfOvlpRMinimal) { define_fcc_cell(ucell); - PARAM.input.qo_screening_coeff = {}; - PARAM.input.qo_thr = 1e-10; + TestParameters::input().qo_screening_coeff = {}; + TestParameters::input().qo_thr = 1e-10; toQO tqo("hydrogen", {"minimal-nodeless", "minimal-nodeless"}, - PARAM.input.qo_thr, - PARAM.input.qo_screening_coeff); + TestParameters::input().qo_thr, + TestParameters::input().qo_screening_coeff); std::vector> kvecs_d; kvecs_d.push_back(ModuleBase::Vector3(0.0, 0.0, 0.0)); // Gamma point ucell.orbital_fn[0] = "Si_special_use_unittest.orb"; // generated in unittest BuildAo ucell.orbital_fn[1] = "C_special_use_unittest.orb"; // generated in unittest BuildAo ucell.atoms[1].nwl = 1; // only s and p for C - tqo.initialize(PARAM.sys.global_out_dir, + tqo.initialize(TestParameters::sys().global_out_dir, "", "", &ucell, @@ -681,14 +691,14 @@ TEST_F(toQOTest, CalculateSelfOvlpRMinimal) TEST_F(toQOTest, AppendOvlpReiRk) { define_fcc_cell(ucell); - PARAM.input.qo_screening_coeff = {}; + TestParameters::input().qo_screening_coeff = {}; toQO tqo("hydrogen", {"minimal-nodeless", "minimal-nodeless"}, - PARAM.input.qo_thr, - PARAM.input.qo_screening_coeff); + TestParameters::input().qo_thr, + TestParameters::input().qo_screening_coeff); std::vector> kvecs_d; kvecs_d.push_back(ModuleBase::Vector3(0.0, 0.0, 0.0)); // Gamma point - tqo.initialize(PARAM.sys.global_out_dir, + tqo.initialize(TestParameters::sys().global_out_dir, "", "", &ucell, @@ -712,12 +722,12 @@ TEST_F(toQOTest, AppendOvlpReiRk) TEST_F(toQOTest, CalculateSelfOvlpKSymmetrical) { define_fcc_cell(ucell); - PARAM.input.qo_thr = 1e-10; - PARAM.input.qo_screening_coeff = {}; + TestParameters::input().qo_thr = 1e-10; + TestParameters::input().qo_screening_coeff = {}; toQO tqo("hydrogen", {"minimal-nodeless", "minimal-nodeless"}, - PARAM.input.qo_thr, - PARAM.input.qo_screening_coeff); + TestParameters::input().qo_thr, + TestParameters::input().qo_screening_coeff); ucell.orbital_fn[0] = "Si_special_use_unittest.orb"; // generated in unittest BuildAo ucell.orbital_fn[1] = "C_special_use_unittest.orb"; // generated in unittest BuildAo ucell.atoms[1].nwl = 1; // only s and p for C @@ -733,7 +743,7 @@ TEST_F(toQOTest, CalculateSelfOvlpKSymmetrical) kvecs_d.push_back(ModuleBase::Vector3(0.25, -0.25, 0.25)); kvecs_d.push_back(ModuleBase::Vector3(0.0, 0.0, 0.0)); // Gamma - tqo.initialize(PARAM.sys.global_out_dir, + tqo.initialize(TestParameters::sys().global_out_dir, "", "", &ucell, @@ -864,19 +874,19 @@ TEST_F(toQOTest, CalculateSelfOvlpKSymmetrical) TEST_F(toQOTest, BuildHydrogenFull) { define_fcc_cell(ucell); - PARAM.input.qo_thr = 1e-10; + TestParameters::input().qo_thr = 1e-10; toQO tqo("hydrogen", {"full", "full"}, - PARAM.input.qo_thr, - PARAM.input.qo_screening_coeff); + TestParameters::input().qo_thr, + TestParameters::input().qo_screening_coeff); std::vector> kvecs_d; kvecs_d.push_back(ModuleBase::Vector3(0.0, 0.0, 0.0)); // Gamma point tqo.read_structures(&ucell, kvecs_d, 0, 1); - PARAM.input.qo_thr = 1e-10; + TestParameters::input().qo_thr = 1e-10; tqo.build_ao(ucell.ntype, "./", ucell.pseudo_fn.data(), {}, - PARAM.input.qo_thr, + TestParameters::input().qo_thr, GlobalV::ofs_running, 0); EXPECT_EQ(tqo.p_ao()->nchi(), 9); // Si: 1s, 2s, 2p, 3s, 3p, 3d, C: 1s, 2s, 2p @@ -887,18 +897,18 @@ TEST_F(toQOTest, BuildHydrogenFull) TEST_F(toQOTest, CalculateSelfOvlpRFull) { define_fcc_cell(ucell); - PARAM.input.qo_thr = 1e-10; - PARAM.input.qo_screening_coeff = {}; + TestParameters::input().qo_thr = 1e-10; + TestParameters::input().qo_screening_coeff = {}; toQO tqo("hydrogen", {"full", "full"}, - PARAM.input.qo_thr, - PARAM.input.qo_screening_coeff); + TestParameters::input().qo_thr, + TestParameters::input().qo_screening_coeff); std::vector> kvecs_d; kvecs_d.push_back(ModuleBase::Vector3(0.0, 0.0, 0.0)); // Gamma point ucell.orbital_fn[0] = "Si_special_use_unittest.orb"; // generated in unittest BuildAo ucell.orbital_fn[1] = "C_special_use_unittest.orb"; // generated in unittest BuildAo ucell.atoms[1].nwl = 1; // only s and p for C - PARAM.input.qo_thr = 1e-10; - tqo.initialize(PARAM.sys.global_out_dir, + TestParameters::input().qo_thr = 1e-10; + tqo.initialize(TestParameters::sys().global_out_dir, "", "", &ucell, @@ -938,12 +948,12 @@ TEST_F(toQOTest, CalculateSelfOvlpRFull) TEST_F(toQOTest, OrbitalFilterOut) { define_fcc_cell(ucell); - PARAM.input.qo_thr = 1e-10; - PARAM.input.qo_screening_coeff = {}; + TestParameters::input().qo_thr = 1e-10; + TestParameters::input().qo_screening_coeff = {}; // because qo_basis hydrogen doesnot have needs to filter out any orbitals, it should be always true toQO tqo("hydrogen", {"full", "full"}, - PARAM.input.qo_thr, - PARAM.input.qo_screening_coeff); + TestParameters::input().qo_thr, + TestParameters::input().qo_screening_coeff); EXPECT_FALSE(tqo.orbital_filter_out(0, 0, 0)); // Si 1s EXPECT_FALSE(tqo.orbital_filter_out(0, 0, 1)); // Si 2s EXPECT_FALSE(tqo.orbital_filter_out(0, 1, 0)); // Si 2p -> 3 @@ -957,10 +967,10 @@ TEST_F(toQOTest, OrbitalFilterOut) // distinguished by (it, ia, l, zeta, m) // if change to pswfc, then it should filter out some orbitals according to qo_strategy - PARAM.input.qo_screening_coeff = {0.5, 0.5}; + TestParameters::input().qo_screening_coeff = {0.5, 0.5}; toQO tqo2("pswfc", {"s", "s"}, - PARAM.input.qo_thr, - PARAM.input.qo_screening_coeff); + TestParameters::input().qo_thr, + TestParameters::input().qo_screening_coeff); // for pswfc specifying l, filter does not care about number of zeta for(int it = 0; it < 2; it++) { @@ -976,8 +986,8 @@ TEST_F(toQOTest, OrbitalFilterOut) } // next test with random ordered arranged names of subshell toQO tqo3("pswfc", {"sfdp", "pdf"}, - PARAM.input.qo_thr, - PARAM.input.qo_screening_coeff); + TestParameters::input().qo_thr, + TestParameters::input().qo_screening_coeff); for(int l = 0; l < 100; l++) { for(int z = 0; z < 100; z++) @@ -1001,8 +1011,8 @@ TEST_F(toQOTest, OrbitalFilterOut) } // test combination `all` with `s` toQO tqo4("pswfc", {"all", "p"}, - PARAM.input.qo_thr, - PARAM.input.qo_screening_coeff); + TestParameters::input().qo_thr, + TestParameters::input().qo_screening_coeff); for(int l = 0; l < 100; l++) { for(int z = 0; z < 100; z++) @@ -1015,8 +1025,8 @@ TEST_F(toQOTest, OrbitalFilterOut) } // test szv, which controls both l and zeta toQO tqo5("szv", {"sdp", "spdfg"}, - PARAM.input.qo_thr, - PARAM.input.qo_screening_coeff); + TestParameters::input().qo_thr, + TestParameters::input().qo_screening_coeff); // for 2 is given as lmax, l can only be 0, 1 and 2, izeta can only be 0 for(int l = 0; l < 100; l++) { @@ -1054,16 +1064,16 @@ TEST_F(toQOTest, BuildPswfcPartial1) { define_fcc_cell(ucell); toQO tqo("pswfc", {"s", "s"}, - PARAM.input.qo_thr, - PARAM.input.qo_screening_coeff); + TestParameters::input().qo_thr, + TestParameters::input().qo_screening_coeff); std::vector> kvecs_d; kvecs_d.push_back(ModuleBase::Vector3(0.0, 0.0, 0.0)); // Gamma point tqo.read_structures(&ucell, kvecs_d, 0, 1); tqo.build_ao(ucell.ntype, "./", ucell.pseudo_fn.data(), - PARAM.input.qo_screening_coeff, - PARAM.input.qo_thr, + TestParameters::input().qo_screening_coeff, + TestParameters::input().qo_thr, GlobalV::ofs_running, 0); EXPECT_EQ(tqo.p_ao()->nchi(), 5); // AO will always read and import all orbitals @@ -1074,16 +1084,16 @@ TEST_F(toQOTest, BuildPswfcPartial2) { define_fcc_cell(ucell); toQO tqo("pswfc", {"ps", "s"}, - PARAM.input.qo_thr, - PARAM.input.qo_screening_coeff); + TestParameters::input().qo_thr, + TestParameters::input().qo_screening_coeff); std::vector> kvecs_d; kvecs_d.push_back(ModuleBase::Vector3(0.0, 0.0, 0.0)); // Gamma point tqo.read_structures(&ucell, kvecs_d, 0, 1); tqo.build_ao(ucell.ntype, "./", ucell.pseudo_fn.data(), - PARAM.input.qo_screening_coeff, - PARAM.input.qo_thr, + TestParameters::input().qo_screening_coeff, + TestParameters::input().qo_thr, GlobalV::ofs_running, 0); EXPECT_EQ(tqo.p_ao()->nchi(), 5); // AO will always read and import all orbitals @@ -1094,16 +1104,16 @@ TEST_F(toQOTest, BuildPswfcPartial3) { define_fcc_cell(ucell); toQO tqo("pswfc", {"all", "p"}, - PARAM.input.qo_thr, - PARAM.input.qo_screening_coeff); + TestParameters::input().qo_thr, + TestParameters::input().qo_screening_coeff); std::vector> kvecs_d; kvecs_d.push_back(ModuleBase::Vector3(0.0, 0.0, 0.0)); // Gamma point tqo.read_structures(&ucell, kvecs_d, 0, 1); tqo.build_ao(ucell.ntype, "./", ucell.pseudo_fn.data(), - PARAM.input.qo_screening_coeff, - PARAM.input.qo_thr, + TestParameters::input().qo_screening_coeff, + TestParameters::input().qo_thr, GlobalV::ofs_running, 0); EXPECT_EQ(tqo.p_ao()->nchi(), 5); // AO will always read and import all orbitals @@ -1113,18 +1123,18 @@ TEST_F(toQOTest, BuildPswfcPartial3) TEST_F(toQOTest, BuildPswfcAll) { define_fcc_cell(ucell); - PARAM.input.qo_thr = 1e-10; + TestParameters::input().qo_thr = 1e-10; toQO tqo("pswfc", {"all", "all"}, - PARAM.input.qo_thr, - PARAM.input.qo_screening_coeff); + TestParameters::input().qo_thr, + TestParameters::input().qo_screening_coeff); std::vector> kvecs_d; kvecs_d.push_back(ModuleBase::Vector3(0.0, 0.0, 0.0)); // Gamma point tqo.read_structures(&ucell, kvecs_d, 0, 1); tqo.build_ao(ucell.ntype, "./", ucell.pseudo_fn.data(), - PARAM.input.qo_screening_coeff, - PARAM.input.qo_thr, + TestParameters::input().qo_screening_coeff, + TestParameters::input().qo_thr, GlobalV::ofs_running, 0); EXPECT_EQ(tqo.p_ao()->nchi(), 5); @@ -1136,8 +1146,8 @@ TEST_F(toQOTest, ScanSupercellSC2) { define_sc_cell(ucell); toQO tqo("pswfc", {"all"}, - PARAM.input.qo_thr, - PARAM.input.qo_screening_coeff); + TestParameters::input().qo_thr, + TestParameters::input().qo_screening_coeff); std::vector> kvecs_d; kvecs_d.push_back(ModuleBase::Vector3(0.0, 0.0, 0.0)); // Gamma point tqo.read_structures(&ucell, kvecs_d, 0, 1); @@ -1145,13 +1155,13 @@ TEST_F(toQOTest, ScanSupercellSC2) "./", ucell.orbital_fn.data(), 0); - PARAM.input.qo_screening_coeff[0] = 0.1; // use this to control the tailing of radial function - PARAM.input.qo_thr = 1e-6; + TestParameters::input().qo_screening_coeff[0] = 0.1; // use this to control the tailing of radial function + TestParameters::input().qo_thr = 1e-6; tqo.build_ao(ucell.ntype, "./", ucell.pseudo_fn.data(), - PARAM.input.qo_screening_coeff, - PARAM.input.qo_thr, + TestParameters::input().qo_screening_coeff, + TestParameters::input().qo_thr, GlobalV::ofs_running, 0); // radius = 13.6 Bohr tqo.scan_supercell(0, 1); @@ -1162,8 +1172,8 @@ TEST_F(toQOTest, ScanSupercellSC3) { define_sc_cell(ucell); toQO tqo("pswfc", {"all"}, - PARAM.input.qo_thr, - PARAM.input.qo_screening_coeff); + TestParameters::input().qo_thr, + TestParameters::input().qo_screening_coeff); std::vector> kvecs_d; kvecs_d.push_back(ModuleBase::Vector3(0.0, 0.0, 0.0)); // Gamma point tqo.read_structures(&ucell, kvecs_d, 0, 1); @@ -1171,26 +1181,26 @@ TEST_F(toQOTest, ScanSupercellSC3) "./", ucell.orbital_fn.data(), 0); - PARAM.input.qo_screening_coeff[0] = 0.25; // use this to control the tailing of radial function - PARAM.input.qo_thr = 1e-6; + TestParameters::input().qo_screening_coeff[0] = 0.25; // use this to control the tailing of radial function + TestParameters::input().qo_thr = 1e-6; tqo.build_ao(ucell.ntype, "./", ucell.pseudo_fn.data(), - PARAM.input.qo_screening_coeff, - PARAM.input.qo_thr, + TestParameters::input().qo_screening_coeff, + TestParameters::input().qo_thr, GlobalV::ofs_running, 0); // radius = 13.6 Bohr tqo.scan_supercell(0, 1); EXPECT_EQ(tqo.nR(), 57); // 5*5*5 - 12(edge center) - 8*(8-1)(corner) = 5*5*5 - 12(edge center) - 8*(2*2*2-1)(corner) - PARAM.input.qo_screening_coeff[0] = 0.1; + TestParameters::input().qo_screening_coeff[0] = 0.1; } TEST_F(toQOTest, ScanSupercellSC4) { define_sc_cell(ucell); toQO tqo("pswfc", {"all"}, - PARAM.input.qo_thr, - PARAM.input.qo_screening_coeff); + TestParameters::input().qo_thr, + TestParameters::input().qo_screening_coeff); std::vector> kvecs_d; kvecs_d.push_back(ModuleBase::Vector3(0.0, 0.0, 0.0)); // Gamma point tqo.read_structures(&ucell, kvecs_d, 0, 1); @@ -1198,34 +1208,34 @@ TEST_F(toQOTest, ScanSupercellSC4) "./", ucell.orbital_fn.data(), 0); - PARAM.input.qo_screening_coeff[0] = 0.5; // use this to control the tailing of radial function - PARAM.input.qo_thr = 1e-6; + TestParameters::input().qo_screening_coeff[0] = 0.5; // use this to control the tailing of radial function + TestParameters::input().qo_thr = 1e-6; tqo.build_ao(ucell.ntype, "./", ucell.pseudo_fn.data(), - PARAM.input.qo_screening_coeff, - PARAM.input.qo_thr, + TestParameters::input().qo_screening_coeff, + TestParameters::input().qo_thr, GlobalV::ofs_running, 0); // radius = 13.6 Bohr tqo.scan_supercell(0, 1); EXPECT_EQ(tqo.nR(), 33); // 3*3*3 + 6(face) - PARAM.input.qo_screening_coeff[0] = 0.1; + TestParameters::input().qo_screening_coeff[0] = 0.1; } TEST_F(toQOTest, CalculateSelfOvlpRPswfc) { define_fcc_cell(ucell); - PARAM.input.qo_thr = 1e-10; + TestParameters::input().qo_thr = 1e-10; toQO tqo("pswfc", {"all", "all"}, - PARAM.input.qo_thr, - PARAM.input.qo_screening_coeff); + TestParameters::input().qo_thr, + TestParameters::input().qo_screening_coeff); std::vector> kvecs_d; kvecs_d.push_back(ModuleBase::Vector3(0.0, 0.0, 0.0)); // Gamma point ucell.orbital_fn.data()[0] = "Si_special_use_unittest.orb"; // generated in unittest BuildAo ucell.orbital_fn.data()[1] = "C_special_use_unittest.orb"; // generated in unittest BuildAo ucell.atoms[1].nwl = 1; // only s and p for C - //PARAM.input.qo_thr = 1e-10; - tqo.initialize(PARAM.sys.global_out_dir, + //TestParameters::input().qo_thr = 1e-10; + tqo.initialize(TestParameters::sys().global_out_dir, "", "", &ucell, @@ -1261,15 +1271,15 @@ TEST_F(toQOTest, CalculateSelfOvlpRPswfc) TEST_F(toQOTest, CalculateOvlpKGamma) { define_fcc_cell(ucell); - PARAM.input.qo_thr = 1e-10; - PARAM.input.qo_screening_coeff = {}; + TestParameters::input().qo_thr = 1e-10; + TestParameters::input().qo_screening_coeff = {}; toQO tqo("hydrogen", {"minimal-nodeless", "minimal-nodeless"}, - PARAM.input.qo_thr, - PARAM.input.qo_screening_coeff); + TestParameters::input().qo_thr, + TestParameters::input().qo_screening_coeff); std::vector> kvecs_d; kvecs_d.push_back(ModuleBase::Vector3(0.0, 0.0, 0.0)); // Gamma point - tqo.initialize(PARAM.sys.global_out_dir, + tqo.initialize(TestParameters::sys().global_out_dir, "", "", &ucell, @@ -1302,14 +1312,14 @@ TEST_F(toQOTest, CalculateOvlpKGamma) TEST_F(toQOTest, CalculateOvlpKSlaterGamma) { define_fcc_cell(ucell); - PARAM.input.qo_thr = 1e-10; - PARAM.input.qo_screening_coeff = {0.1}; + TestParameters::input().qo_thr = 1e-10; + TestParameters::input().qo_screening_coeff = {0.1}; toQO tqo("hydrogen", {"energy-full", "energy-full"}, - PARAM.input.qo_thr, - PARAM.input.qo_screening_coeff); + TestParameters::input().qo_thr, + TestParameters::input().qo_screening_coeff); std::vector> kvecs_d; kvecs_d.push_back(ModuleBase::Vector3(0.0, 0.0, 0.0)); // Gamma point - tqo.initialize(PARAM.sys.global_out_dir, + tqo.initialize(TestParameters::sys().global_out_dir, "", "", &ucell, @@ -1342,11 +1352,11 @@ TEST_F(toQOTest, CalculateOvlpKSlaterGamma) TEST_F(toQOTest, CalculateSelfOvlpKPswfcSymmetrical) { define_fcc_cell(ucell); - PARAM.input.qo_thr = 1e-10; - PARAM.input.qo_screening_coeff = {2.0, 2.0}; + TestParameters::input().qo_thr = 1e-10; + TestParameters::input().qo_screening_coeff = {2.0, 2.0}; toQO tqo("pswfc", {"all", "all"}, - PARAM.input.qo_thr, - PARAM.input.qo_screening_coeff); + TestParameters::input().qo_thr, + TestParameters::input().qo_screening_coeff); std::vector> kvecs_d; kvecs_d.push_back(ModuleBase::Vector3(-0.25, -0.25, -0.25)); // pair 1 @@ -1359,7 +1369,7 @@ TEST_F(toQOTest, CalculateSelfOvlpKPswfcSymmetrical) kvecs_d.push_back(ModuleBase::Vector3(0.25, -0.25, 0.25)); kvecs_d.push_back(ModuleBase::Vector3(0.0, 0.0, 0.0)); // Gamma - tqo.initialize(PARAM.sys.global_out_dir, + tqo.initialize(TestParameters::sys().global_out_dir, "", "", &ucell, @@ -1455,16 +1465,16 @@ TEST_F(toQOTest, CalculateSelfOvlpKPswfcSymmetrical) TEST_F(toQOTest, CalculateHydrogenlike) { define_fcc_cell(ucell); - PARAM.input.qo_thr = 1e-10; - PARAM.input.qo_screening_coeff = {}; + TestParameters::input().qo_thr = 1e-10; + TestParameters::input().qo_screening_coeff = {}; toQO tqo("hydrogen", {"minimal-nodeless", "minimal-nodeless"}, - PARAM.input.qo_thr, - PARAM.input.qo_screening_coeff); + TestParameters::input().qo_thr, + TestParameters::input().qo_screening_coeff); std::vector> kvecs_d; kvecs_d.push_back(ModuleBase::Vector3(0.0, 0.0, 0.0)); // Gamma point kvecs_d.push_back(ModuleBase::Vector3(0.5, 0.0, 0.0)); - tqo.initialize(PARAM.sys.global_out_dir, + tqo.initialize(TestParameters::sys().global_out_dir, "", "", &ucell, diff --git a/source/source_io/test/write_dos_pw_test.cpp b/source/source_io/test/write_dos_pw_test.cpp deleted file mode 100644 index fae1c70445a..00000000000 --- a/source/source_io/test/write_dos_pw_test.cpp +++ /dev/null @@ -1,189 +0,0 @@ -#include "gtest/gtest.h" -#include "gmock/gmock.h" -#include "source_io/module_dos/write_dos_pw.h" -#ifdef __MPI -#include "mpi.h" -#endif -#include "for_testing_klist.h" -#include "dos_test.h" - -#define private public -#include "source_io/module_parameter/parameter.h" -#undef private - -/************************************************ - * unit test of write_dos_pw - ***********************************************/ - -/** - * - Tested Functions: - * - write_dos_pw() - * - the function to calculate and print out - * - density of states in pw basis calculation - */ - - -class DosPWTest : public ::testing::Test -{ -protected: - K_Vectors* kv = nullptr; - ModuleBase::matrix ekb; - ModuleBase::matrix wg; - void SetUp() - { - kv = new K_Vectors; - } - void TearDown() - { - delete kv; - } -}; - -TEST_F(DosPWTest,Dos1) -{ - //is,fa,fa1,de_ev,emax_ev,emin_ev,bcoeff,nks,nkstot,nbands - DosPrepare dosp = DosPrepare(0,"doss1_pw.txt",0.005,18,-6,0.07,36,36,8); - dosp.set_isk(); - dosp.read_wk(); - dosp.read_istate_info(); - EXPECT_EQ(dosp.is,0); - double dos_scale = 0.01; - PARAM.input.nspin = 1; - PARAM.input.dos_emax_ev = dosp.emax_ev; - PARAM.sys.dos_setemax = true; - PARAM.input.dos_emin_ev = dosp.emin_ev; - PARAM.sys.dos_setemin = true; - kv->set_nks(dosp.nks); - kv->set_nkstot(dosp.nkstot); - kv->isk.reserve(kv->get_nks()); - kv->wk.reserve(kv->get_nks()); - for(int ik=0; ikget_nks(); ++ik) - { - kv->isk[ik] = dosp.isk[ik]; - kv->wk[ik] = dosp.wk[ik]; - } - PARAM.input.nbands = dosp.nbands; - - // initialize the Fermi energy - elecstate::Efermi fermi_energy; - - std::ofstream ofs("write_dos_pw.log"); - - UnitCell ucell; - - ModuleIO::write_dos_pw( - ucell, // this should be unitcell, 2025-04-12 - dosp.ekb, - dosp.wg, - *kv, - PARAM.inp.nbands, - -1, // istep_in - fermi_energy, - dosp.de_ev, - dos_scale, - dosp.bcoeff, - ofs); - ofs.close(); - remove("write_dos_pw.log"); - -#ifdef __MPI - if(GlobalV::MY_RANK==0) - { -#endif - std::ifstream ifs; - ifs.open("dos.txt"); - std::string str((std::istreambuf_iterator(ifs)),std::istreambuf_iterator()); - EXPECT_THAT(str, testing::HasSubstr("4801 # number of points")); - EXPECT_THAT(str, testing::HasSubstr(" -4.6 0.25 0.28125 1.42515 0.159819")); - EXPECT_THAT(str, testing::HasSubstr(" 18 0 16 0 16")); - ifs.close(); - remove("dos.txt"); -#ifdef __MPI - } -#endif -} - - -TEST_F(DosPWTest,Dos2) -{ - //is,fa,fa1,de_ev,emax_ev,emin_ev,bcoeff,nks,nkstot,nbands - DosPrepare dosp = DosPrepare(0,"doss1_pw.txt",0.005,18,-6,0.07,36,36,8); - dosp.set_isk(); - dosp.read_wk(); - dosp.read_istate_info(); - EXPECT_EQ(dosp.is,0); - double dos_scale = 0.01; - PARAM.input.nspin = 1; - PARAM.input.dos_emax_ev = dosp.emax_ev; - PARAM.sys.dos_setemax = false; - PARAM.input.dos_emin_ev = dosp.emin_ev; - PARAM.sys.dos_setemin = false; - kv->set_nks(dosp.nks); - kv->set_nkstot(dosp.nkstot); - kv->isk.reserve(kv->get_nks()); - kv->wk.reserve(kv->get_nks()); - for(int ik=0; ikget_nks(); ++ik) - { - kv->isk[ik] = dosp.isk[ik]; - kv->wk[ik] = dosp.wk[ik]; - } - PARAM.input.nbands = dosp.nbands; - - // initialize the Fermi energy - elecstate::Efermi fermi_energy; - - std::ofstream ofs("write_dos_pw.log"); - - UnitCell ucell; - - ModuleIO::write_dos_pw( - ucell, - dosp.ekb, - dosp.wg, - *kv, - PARAM.inp.nbands, - -1, // istep_in - fermi_energy, - dosp.de_ev, - dos_scale, - dosp.bcoeff, - ofs); - ofs.close(); - remove("write_dos_pw.log"); - -#ifdef __MPI - if(GlobalV::MY_RANK==0) - { -#endif - std::ifstream ifs; - ifs.open("dos.txt"); - std::string str1((std::istreambuf_iterator(ifs)),std::istreambuf_iterator()); - EXPECT_THAT(str1, testing::HasSubstr("4532 # number of points")); - EXPECT_THAT(str1, testing::HasSubstr(" -5.38811 0.03125 0.03125")); - EXPECT_THAT(str1, testing::HasSubstr(" 3.07189 0.1875 5.46875")); - ifs.close(); - remove("dos.txt"); -#ifdef __MPI - } -#endif -} - -#ifdef __MPI -int main(int argc, char **argv) -{ - MPI_Init(&argc,&argv); - - testing::InitGoogleTest(&argc,argv); - MPI_Comm_size(MPI_COMM_WORLD,&GlobalV::NPROC); - MPI_Comm_rank(MPI_COMM_WORLD,&GlobalV::MY_RANK); - - // only test the second one - // ::testing::GTEST_FLAG(filter) = "DosPWTest.Dos2"; - - int result = RUN_ALL_TESTS(); - - MPI_Finalize(); - - return result; -} -#endif diff --git a/source/source_io/test/write_eig_occ_test.cpp b/source/source_io/test/write_eig_occ_test.cpp index e758601a87b..a96b5a6a88b 100644 --- a/source/source_io/test/write_eig_occ_test.cpp +++ b/source/source_io/test/write_eig_occ_test.cpp @@ -1,8 +1,5 @@ #include "source_base/global_variable.h" -#define private public -#include "source_io/module_parameter/parameter.h" -#undef private #include "gmock/gmock.h" #include "gtest/gtest.h" #include @@ -43,17 +40,18 @@ class IstateInfoTest : public ::testing::Test TEST_F(IstateInfoTest, OutIstateInfoS1) { - // Global variables + // Global variables GlobalV::KPAR = 1; - PARAM.input.nbands = 4; - PARAM.sys.nbands_l = 4; - PARAM.input.nspin = 1; - PARAM.sys.global_out_dir = "./"; + const int nbands = 4; + const int nspin = 1; + const std::string out_dir = "./"; + // Mirrors the Input_para default (bndpar = 1) so the test does not depend on it. + const int bndpar = 1; // MPI setting Parallel_Global::init_pools(GlobalV::NPROC, GlobalV::MY_RANK, - PARAM.input.bndpar, + bndpar, GlobalV::KPAR, GlobalV::NPROC_IN_BNDGROUP, GlobalV::RANK_IN_BPGROUP, @@ -65,8 +63,8 @@ TEST_F(IstateInfoTest, OutIstateInfoS1) const int nkstot_init = 10; kv->set_nkstot(nkstot_init); int nkstot = kv->get_nkstot(); - kv->para_k.kinfo(nkstot, GlobalV::KPAR, GlobalV::MY_POOL, GlobalV::RANK_IN_POOL, - GlobalV::NPROC_IN_POOL, PARAM.input.nspin); + kv->para_k.kinfo(nkstot, GlobalV::KPAR, GlobalV::MY_POOL, GlobalV::RANK_IN_POOL, + GlobalV::NPROC_IN_POOL, nspin); kv->set_nks(kv->para_k.nks_pool[GlobalV::MY_POOL]); // The number of plane waves for each k point @@ -79,8 +77,8 @@ TEST_F(IstateInfoTest, OutIstateInfoS1) } // Initialize the number of bands - ekb.create(kv->get_nks(), PARAM.input.nbands); - wg.create(kv->get_nks(), PARAM.input.nbands); + ekb.create(kv->get_nks(), nbands); + wg.create(kv->get_nks(), nbands); // fill the eigenvalues ekb.fill_out(0.15); @@ -103,8 +101,8 @@ TEST_F(IstateInfoTest, OutIstateInfoS1) } // A new calculation truncates stale output, then later ionic steps append. - ModuleIO::write_eig_file(ekb, wg, *kv, 0); - ModuleIO::write_eig_file(ekb, wg, *kv, 1); + ModuleIO::write_eig_file(ekb, wg, *kv, nbands, nspin, out_dir, 0); + ModuleIO::write_eig_file(ekb, wg, *kv, nbands, nspin, out_dir, 1); // check the output files std::ifstream ifs; diff --git a/source/source_io/test/write_hs_r_compat_test.cpp b/source/source_io/test/write_hs_r_compat_test.cpp deleted file mode 100644 index a4103307e5b..00000000000 --- a/source/source_io/test/write_hs_r_compat_test.cpp +++ /dev/null @@ -1,1108 +0,0 @@ -#include "gmock/gmock.h" -#include "gtest/gtest.h" - -#define private public -#include "source_io/module_parameter/parameter.h" -#undef private - -#include "source_base/global_variable.h" -#include "source_cell/module_neighbor/sltk_grid_driver.h" -#include "source_io/module_dm/write_dmr.h" -#include "source_io/module_hs/output_mat_sparse.h" -#include "source_io/module_hs/rr_sparse_writer.h" -#include "source_io/module_hs/write_hs_r.h" -#include "source_io/module_hs/write_hs_sparse.h" -#include "source_base/module_out/csr_reader.h" -#include "source_hamilt/module_hcontainer/atom_pair.h" -#include "source_hamilt/module_hcontainer/hcontainer.h" -#include "source_hamilt/module_hcontainer/hcontainer_funcs.h" - -#include -#include -#include -#include -#include -#include -#include -#include - -#ifdef __MPI -#include -#endif - -namespace sparse_format -{ -void cal_dH(const UnitCell&, - const Parallel_Orbitals&, - LCAO_HS_Arrays&, - const Grid_Driver&, - const TwoCenterBundle&, - const LCAO_Orbitals&, - const int&, - const double&, - const ModuleBase::matrix&) -{ - FAIL() << "cal_dH should not be called by writer compatibility tests."; -} - -void cal_dS(const UnitCell&, - const Parallel_Orbitals&, - LCAO_HS_Arrays&, - const Grid_Driver&, - const TwoCenterBundle&, - const LCAO_Orbitals&, - const double&) -{ - FAIL() << "cal_dS should not be called by writer compatibility tests."; -} - -void cal_TR(const UnitCell&, - const Parallel_Orbitals&, - LCAO_HS_Arrays&, - const Grid_Driver&, - const TwoCenterBundle&, - const LCAO_Orbitals&, - const double&) -{ - FAIL() << "cal_TR should not be called by writer compatibility tests."; -} - -template -void cal_SR(const Parallel_Orbitals&, - std::set>&, - std::map, std::map>>&, - std::map, std::map>>>&, - const Grid_Driver&, - const double&, - hamilt::Hamilt*) -{ - FAIL() << "cal_SR should not be called by writer compatibility tests."; -} - -void destroy_dH_R_sparse(LCAO_HS_Arrays&) {} -void destroy_HS_R_sparse(LCAO_HS_Arrays&) {} -void destroy_T_R_sparse(LCAO_HS_Arrays&) {} - -template void cal_SR( - const Parallel_Orbitals&, - std::set>&, - std::map, std::map>>&, - std::map, std::map>>>&, - const Grid_Driver&, - const double&, - hamilt::Hamilt*); - -template void cal_SR>( - const Parallel_Orbitals&, - std::set>&, - std::map, std::map>>&, - std::map, std::map>>>&, - const Grid_Driver&, - const double&, - hamilt::Hamilt>*); -} // namespace sparse_format - -namespace -{ -std::string read_file(const std::string& filename) -{ - std::ifstream ifs(filename.c_str()); - std::ostringstream oss; - oss << ifs.rdbuf(); - return oss.str(); -} - -std::vector read_binary_ints(const std::string& filename, const size_t count) -{ - std::ifstream ifs(filename.c_str(), std::ios::binary); - std::vector values(count, 0); - for (size_t i = 0; i < count; ++i) - { - ifs.read(reinterpret_cast(&values[i]), sizeof(int)); - } - return values; -} - -template -T read_binary_value(std::ifstream& ifs) -{ - T value{}; - ifs.read(reinterpret_cast(&value), sizeof(T)); - return value; -} - -struct NativeDoubleRBlock -{ - int rx = 0; - int ry = 0; - int rz = 0; - std::vector values; - std::vector columns; - std::vector row_ptr; -}; - -struct NativeDoubleRecord -{ - int step = 0; - int nbasis = 0; - std::vector blocks; -}; - -NativeDoubleRecord read_native_double_record(std::ifstream& ifs) -{ - NativeDoubleRecord record; - record.step = read_binary_value(ifs); - record.nbasis = read_binary_value(ifs); - const int nR = read_binary_value(ifs); - record.blocks.resize(nR); - for (NativeDoubleRBlock& block: record.blocks) - { - block.rx = read_binary_value(ifs); - block.ry = read_binary_value(ifs); - block.rz = read_binary_value(ifs); - const int nnz = read_binary_value(ifs); - block.values.resize(nnz); - block.columns.resize(nnz); - block.row_ptr.resize(record.nbasis + 1); - for (double& value: block.values) - { - value = read_binary_value(ifs); - } - for (int& column: block.columns) - { - column = read_binary_value(ifs); - } - for (long long& pointer: block.row_ptr) - { - pointer = read_binary_value(ifs); - } - } - return record; -} - -std::vector read_lines(const std::string& filename) -{ - std::ifstream ifs(filename.c_str()); - std::vector lines; - std::string line; - while (std::getline(ifs, line)) - { - lines.push_back(line); - } - return lines; -} - -int count_substr(const std::string& text, const std::string& pattern) -{ - int count = 0; - std::string::size_type pos = 0; - while ((pos = text.find(pattern, pos)) != std::string::npos) - { - ++count; - pos += pattern.size(); - } - return count; -} - -void init_unitcell(UnitCell& ucell) -{ - ucell.latName = "user_defined_lattice"; - ucell.lat0 = 10.0; - ucell.latvec.e11 = 1.0; - ucell.latvec.e22 = 1.0; - ucell.latvec.e33 = 1.0; - ucell.ntype = 1; - ucell.nat = 1; - ucell.atoms = new Atom[1]; - ucell.set_atom_flag = true; - ucell.atoms[0].label = "Si"; - ucell.atoms[0].na = 1; - ucell.atoms[0].nw = 2; - ucell.atoms[0].taud.resize(1); - ucell.atoms[0].taud[0] = ModuleBase::Vector3(0.0, 0.25, 0.5); -} - -void init_serial_orbitals(Parallel_Orbitals& pv) -{ - pv.atom_begin_row.resize(2); - pv.atom_begin_col.resize(2); - pv.atom_begin_row[0] = 0; - pv.atom_begin_row[1] = 2; - pv.atom_begin_col[0] = 0; - pv.atom_begin_col[1] = 2; - pv.nrow = 2; - pv.ncol = 2; - pv.set_serial(2, 2); -} - -void fill_matrix(hamilt::HContainer& matrix, Parallel_Orbitals& pv, double* values) -{ - hamilt::AtomPair pair(0, 0, 0, 0, 0, &pv, values); - matrix.insert_pair(pair); -} - -template -void fill_matrix_at_R(hamilt::HContainer& matrix, - Parallel_Orbitals& pv, - const int rx, - const int ry, - const int rz, - T* values) -{ - hamilt::AtomPair pair(0, 0, rx, ry, rz, &pv, values); - matrix.insert_pair(pair); -} - -void init_sparse_output_globals(const int nspin = 1) -{ - GlobalV::DRANK = 0; - PARAM.input.nspin = nspin; - PARAM.input.calculation = "scf"; - PARAM.input.out_app_flag = false; - PARAM.sys.global_out_dir = "./"; - PARAM.sys.global_matrix_dir = "./"; - PARAM.sys.nlocal = 2; -} - -void remove_derivative_files(const std::string& fileflag) -{ - const std::vector filenames = { - "d" + fileflag + "rxs1_nao.csr", - "d" + fileflag + "rys1_nao.csr", - "d" + fileflag + "rzs1_nao.csr", - "d" + fileflag + "rxs2_nao.csr", - "d" + fileflag + "rys2_nao.csr", - "d" + fileflag + "rzs2_nao.csr", - }; - for (const std::string& filename: filenames) - { - std::remove(filename.c_str()); - } -} - -bool starts_with(const std::string& text, const std::string& prefix) -{ - return text.find(prefix) == 0; -} -} // namespace - -TEST(WriteHsRCompatibility, FileNameHelpersKeepCurrentContract) -{ - EXPECT_EQ(ModuleIO::hsr_gen_fname("hrs", 0, true, -1), "hrs1_nao.csr"); - EXPECT_EQ(ModuleIO::hsr_gen_fname("hrs", 1, true, 0), "hrs2_nao.csr"); - EXPECT_EQ(ModuleIO::hsr_gen_fname("hrs", 1, false, 3), "hrs2g4_nao.csr"); - EXPECT_EQ(ModuleIO::sr_gen_fname(false, 0), "srg1_nao.csr"); - EXPECT_EQ(ModuleIO::sr_gen_fname(false, -1), "sr_nao.csr"); - EXPECT_EQ(ModuleIO::sr_gen_fname(true, 3), "sr_nao.csr"); - EXPECT_EQ(ModuleIO::hsr_gen_fname("hrs", 0, true, -1, 2), "hrs1_nao.dat"); - EXPECT_EQ(ModuleIO::hsr_gen_fname("hrs", 1, false, 3, 2), "hrs2g4_nao.dat"); - EXPECT_EQ(ModuleIO::sr_gen_fname(false, 0, 2), "srg1_nao.dat"); - EXPECT_EQ(ModuleIO::sr_gen_fname(true, 3, 2), "sr_nao.dat"); - - EXPECT_EQ(ModuleIO::dhr_gen_fname("dhrx", 0, true, -1), "dhrxrs1_nao.csr"); - EXPECT_EQ(ModuleIO::dhr_gen_fname("dhrx", 0, false, 0), "dhrxrs1g1_nao.csr"); - EXPECT_EQ(ModuleIO::dhr_gen_fname("dsry", 1, false, 2), "dsryrs2g3_nao.csr"); - - EXPECT_EQ(ModuleIO::dmr_gen_fname(1, 0, true, -1), "dmrs1_nao.csr"); - EXPECT_EQ(ModuleIO::dmr_gen_fname(1, 1, false, 2), "dmrs2g3_nao.csr"); - EXPECT_EQ(ModuleIO::dmr_gen_fname(2, 0, true, 5), "dmrs1_nao.npz"); -} - -TEST(WriteHsRCompatibility, HContainerCsrHeaderKeepsCurrentFormat) -{ - const std::string filename = "write_hs_r_header_h.csr"; - std::remove(filename.c_str()); - - UnitCell ucell; - init_unitcell(ucell); - Parallel_Orbitals pv; - init_serial_orbitals(pv); - hamilt::HContainer matrix(&pv); - double values[4] = {1.0, 0.0, 0.5, 2.0}; - fill_matrix(matrix, pv, values); - - ModuleIO::write_hcontainer_csr(filename, &ucell, 5, &matrix, 0, 0, 1, "H", ""); - - const std::string output = read_file(filename); - EXPECT_THAT(output, testing::HasSubstr(" --- Ionic Step 1 ---\n")); - EXPECT_THAT(output, testing::HasSubstr(" # print H matrix in real space H(R)\n")); - EXPECT_THAT(output, testing::HasSubstr(" 1 # number of spin directions\n")); - EXPECT_THAT(output, testing::HasSubstr(" 1 # spin index\n")); - EXPECT_THAT(output, testing::HasSubstr(" 2 # number of localized basis\n")); - EXPECT_THAT(output, testing::HasSubstr(" 1 # number of Bravais lattice vector R\n")); - EXPECT_THAT(output, testing::HasSubstr(" user_defined_lattice\n")); - EXPECT_THAT(output, testing::HasSubstr(" Si\n")); - EXPECT_THAT(output, testing::HasSubstr(" Direct\n")); - EXPECT_THAT(output, testing::HasSubstr(" 0 0.25 0.5\n")); - EXPECT_THAT(output, testing::HasSubstr(" # CSR Format #\n")); - EXPECT_THAT(output, testing::HasSubstr(" 0 0 0 3\n")); - EXPECT_THAT(output, testing::HasSubstr(" # CSR values\n")); - EXPECT_THAT(output, testing::HasSubstr(" 1.00000e+00 5.00000e-01 2.00000e+00")); - EXPECT_THAT(output, testing::Not(testing::HasSubstr("# representation:"))); - - std::remove(filename.c_str()); -} - -TEST(WriteHsRCompatibility, GammaFoldedHeaderKeepsCsrReadable) -{ - const std::string filename = "write_hs_r_gamma_folded.csr"; - const std::string representation_note - = "gamma-only folded matrix; stored R-space contributions are summed into R = (0, 0, 0)"; - std::remove(filename.c_str()); - - UnitCell ucell; - init_unitcell(ucell); - Parallel_Orbitals pv; - init_serial_orbitals(pv); - hamilt::HContainer matrix(&pv); - double values[4] = {1.0, 0.0, 0.5, 2.0}; - fill_matrix(matrix, pv, values); - - ModuleIO::write_hcontainer_csr( - filename, &ucell, 5, &matrix, 0, 0, 1, "H", representation_note); - - const std::string output = read_file(filename); - EXPECT_THAT(output, testing::HasSubstr("# representation: " + representation_note + "\n")); - EXPECT_THAT(output, testing::HasSubstr(" 1 # number of Bravais lattice vector R\n")); - EXPECT_THAT(output, testing::HasSubstr(" 0 0 0 3\n")); - - ModuleIO::csrFileReader reader(filename); - ASSERT_EQ(reader.getNumberOfR(), 1); - EXPECT_EQ(reader.getMatrixDimension(), 2); - EXPECT_EQ(reader.getRCoordinate(0), std::vector({0, 0, 0})); - - std::remove(filename.c_str()); -} - -TEST(WriteHsRCompatibility, HContainerCsrAppendKeepsCurrentStepSections) -{ - const std::string filename = "write_hs_r_append_s.csr"; - std::remove(filename.c_str()); - - UnitCell ucell; - init_unitcell(ucell); - Parallel_Orbitals pv; - init_serial_orbitals(pv); - hamilt::HContainer matrix(&pv); - double values[4] = {1.0, 0.0, 0.0, 1.0}; - fill_matrix(matrix, pv, values); - - ModuleIO::write_hcontainer_csr(filename, &ucell, 4, &matrix, 0, 0, 1, "S", ""); - ModuleIO::write_hcontainer_csr(filename, &ucell, 4, &matrix, 1, 0, 1, "S", ""); - - const std::string output = read_file(filename); - EXPECT_EQ(count_substr(output, " --- Ionic Step "), 2); - EXPECT_THAT(output, testing::HasSubstr(" --- Ionic Step 1 ---\n")); - EXPECT_THAT(output, testing::HasSubstr(" --- Ionic Step 2 ---\n")); - EXPECT_EQ(count_substr(output, " # print S matrix in real space S(R)\n"), 2); - - std::remove(filename.c_str()); -} - -TEST(WriteHsRCompatibility, HContainerBinaryWritesSortedAndEmptyRBlocks) -{ - const std::string filename = "write_hs_r_native_binary_real.dat"; - std::remove(filename.c_str()); - - Parallel_Orbitals pv; - init_serial_orbitals(pv); - hamilt::HContainer matrix(&pv); - double empty_values[4] = {1e-12, 0.0, 0.0, -1e-12}; - double values[4] = {1.0, 1e-12, 0.0, -2.0}; - fill_matrix_at_R(matrix, pv, 1, 0, 0, empty_values); - fill_matrix_at_R(matrix, pv, -1, 0, 0, values); - - ModuleIO::write_hcontainer_csr_binary(filename, &matrix, -1, false); - - std::ifstream ifs(filename.c_str(), std::ios::binary); - ASSERT_TRUE(ifs.is_open()); - const NativeDoubleRecord record = read_native_double_record(ifs); - EXPECT_EQ(record.step, 0); - EXPECT_EQ(record.nbasis, 2); - ASSERT_EQ(record.blocks.size(), 2); - EXPECT_EQ(record.blocks[0].rx, -1); - EXPECT_EQ(record.blocks[0].ry, 0); - EXPECT_EQ(record.blocks[0].rz, 0); - EXPECT_THAT(record.blocks[0].values, testing::ElementsAre(1.0, -2.0)); - EXPECT_THAT(record.blocks[0].columns, testing::ElementsAre(0, 1)); - EXPECT_THAT(record.blocks[0].row_ptr, testing::ElementsAre(0, 1, 2)); - EXPECT_EQ(record.blocks[1].rx, 1); - EXPECT_EQ(record.blocks[1].ry, 0); - EXPECT_EQ(record.blocks[1].rz, 0); - EXPECT_TRUE(record.blocks[1].values.empty()); - EXPECT_TRUE(record.blocks[1].columns.empty()); - EXPECT_THAT(record.blocks[1].row_ptr, testing::ElementsAre(0, 0, 0)); - EXPECT_EQ(ifs.peek(), std::ifstream::traits_type::eof()); - - std::remove(filename.c_str()); -} - -TEST(WriteHsRCompatibility, HContainerBinaryWritesComplexValuesAsDoublePairs) -{ - const std::string filename = "write_hs_r_native_binary_complex.dat"; - std::remove(filename.c_str()); - - Parallel_Orbitals pv; - init_serial_orbitals(pv); - hamilt::HContainer> matrix(&pv); - std::complex values[4] = { - std::complex(1.0, 2.0), - std::complex(0.0, 0.0), - std::complex(0.0, 0.0), - std::complex(-3.0, 4.0), - }; - fill_matrix_at_R(matrix, pv, 0, 0, 0, values); - - ModuleIO::write_hcontainer_csr_binary(filename, &matrix, 4, false); - - std::ifstream ifs(filename.c_str(), std::ios::binary); - ASSERT_TRUE(ifs.is_open()); - EXPECT_EQ(read_binary_value(ifs), 4); - EXPECT_EQ(read_binary_value(ifs), 2); - EXPECT_EQ(read_binary_value(ifs), 1); - EXPECT_EQ(read_binary_value(ifs), 0); - EXPECT_EQ(read_binary_value(ifs), 0); - EXPECT_EQ(read_binary_value(ifs), 0); - EXPECT_EQ(read_binary_value(ifs), 2); - EXPECT_DOUBLE_EQ(read_binary_value(ifs), 1.0); - EXPECT_DOUBLE_EQ(read_binary_value(ifs), 2.0); - EXPECT_DOUBLE_EQ(read_binary_value(ifs), -3.0); - EXPECT_DOUBLE_EQ(read_binary_value(ifs), 4.0); - EXPECT_EQ(read_binary_value(ifs), 0); - EXPECT_EQ(read_binary_value(ifs), 1); - EXPECT_EQ(read_binary_value(ifs), 0); - EXPECT_EQ(read_binary_value(ifs), 1); - EXPECT_EQ(read_binary_value(ifs), 2); - EXPECT_EQ(ifs.peek(), std::ifstream::traits_type::eof()); - - std::remove(filename.c_str()); -} - -TEST(WriteHsRCompatibility, HContainerBinaryAppendsAndCanOverwrite) -{ - const std::string filename = "write_hs_r_native_binary_append.dat"; - std::remove(filename.c_str()); - - Parallel_Orbitals pv; - init_serial_orbitals(pv); - hamilt::HContainer matrix(&pv); - double values[4] = {1.0, 0.0, 0.0, 2.0}; - fill_matrix(matrix, pv, values); - - ModuleIO::write_hcontainer_csr_binary(filename, &matrix, 0, true); - ModuleIO::write_hcontainer_csr_binary(filename, &matrix, 1, true); - - std::ifstream appended(filename.c_str(), std::ios::binary); - ASSERT_TRUE(appended.is_open()); - EXPECT_EQ(read_native_double_record(appended).step, 0); - EXPECT_EQ(read_native_double_record(appended).step, 1); - EXPECT_EQ(appended.peek(), std::ifstream::traits_type::eof()); - appended.close(); - - ModuleIO::write_hcontainer_csr_binary(filename, &matrix, 2, false); - std::ifstream replaced(filename.c_str(), std::ios::binary); - ASSERT_TRUE(replaced.is_open()); - EXPECT_EQ(read_native_double_record(replaced).step, 2); - EXPECT_EQ(replaced.peek(), std::ifstream::traits_type::eof()); - - std::remove(filename.c_str()); -} - -TEST(WriteHsRCompatibility, HContainerBinaryMpiGatherWritesCompleteFiles) -{ -#ifndef __MPI - GTEST_SKIP() << "MPI support is required."; -#else - int mpi_size = 0; - int mpi_rank = 0; - MPI_Comm_size(MPI_COMM_WORLD, &mpi_size); - MPI_Comm_rank(MPI_COMM_WORLD, &mpi_rank); - if (mpi_size != 2) - { - GTEST_SKIP() << "This test requires exactly two MPI ranks."; - } - - const std::string hr_filename = "hrs1_nao.dat"; - const std::string sr_filename = "sr_nao.dat"; - if (mpi_rank == 0) - { - std::remove(hr_filename.c_str()); - std::remove(sr_filename.c_str()); - } - MPI_Barrier(MPI_COMM_WORLD); - - UnitCell ucell; - ucell.ntype = 1; - ucell.nat = 1; - ucell.atoms = new Atom[1]; - ucell.set_atom_flag = true; - ucell.atoms[0].na = 1; - ucell.atoms[0].nw = 2; - ucell.iat2it = new int[1]; - ucell.iat2it[0] = 0; - ucell.set_iat2iwt(1); - const int* iat2iwt = ucell.get_iat2iwt(); - Parallel_Orbitals serial_pv; - serial_pv.set_serial(2, 2); - serial_pv.set_atomic_trace(iat2iwt, 1, 2); - Parallel_Orbitals parallel_pv; - ASSERT_EQ(parallel_pv.init(2, 2, 1, MPI_COMM_WORLD), 0); - parallel_pv.set_atomic_trace(iat2iwt, 1, 2); - - hamilt::HContainer hr_serial(ucell, &serial_pv); - hamilt::HContainer sr_serial(ucell, &serial_pv); - double hr_values[4] = {1.0, 0.5, 0.0, 2.0}; - double sr_values[4] = {1.0, 0.0, 0.0, 1.0}; - if (mpi_rank == 0) - { - std::copy(hr_values, hr_values + 4, hr_serial.get_atom_pair(0).get_pointer(0)); - std::copy(sr_values, sr_values + 4, sr_serial.get_atom_pair(0).get_pointer(0)); - } - - hamilt::HContainer hr_parallel(ucell, ¶llel_pv); - hamilt::HContainer sr_parallel(ucell, ¶llel_pv); - hamilt::transferSerial2Parallels(hr_serial, &hr_parallel, 0); - hamilt::transferSerial2Parallels(sr_serial, &sr_parallel, 0); - - init_sparse_output_globals(); - std::vector*> hr_vec(1, &hr_parallel); - ModuleIO::write_hsr( - hr_vec, &sr_parallel, &ucell, 2, 8, parallel_pv, true, true, iat2iwt, 1, 0); - MPI_Barrier(MPI_COMM_WORLD); - - if (mpi_rank == 0) - { - std::ifstream hr_stream(hr_filename.c_str(), std::ios::binary); - ASSERT_TRUE(hr_stream.is_open()); - const NativeDoubleRecord hr_record = read_native_double_record(hr_stream); - ASSERT_EQ(hr_record.blocks.size(), 1); - EXPECT_THAT(hr_record.blocks[0].values, testing::ElementsAre(1.0, 0.5, 2.0)); - EXPECT_THAT(hr_record.blocks[0].columns, testing::ElementsAre(0, 1, 1)); - EXPECT_THAT(hr_record.blocks[0].row_ptr, testing::ElementsAre(0, 2, 3)); - - std::ifstream sr_stream(sr_filename.c_str(), std::ios::binary); - ASSERT_TRUE(sr_stream.is_open()); - const NativeDoubleRecord sr_record = read_native_double_record(sr_stream); - ASSERT_EQ(sr_record.blocks.size(), 1); - EXPECT_THAT(sr_record.blocks[0].values, testing::ElementsAre(1.0, 1.0)); - EXPECT_THAT(sr_record.blocks[0].columns, testing::ElementsAre(0, 1)); - EXPECT_THAT(sr_record.blocks[0].row_ptr, testing::ElementsAre(0, 1, 2)); - - std::remove(hr_filename.c_str()); - std::remove(sr_filename.c_str()); - } - delete[] ucell.atoms; - ucell.atoms = nullptr; - ucell.set_atom_flag = false; -#endif -} - -TEST(WriteHsRCompatibility, DmrCsrHeaderKeepsCurrentFormat) -{ - std::string filename = "write_hs_r_header_dmr.csr"; - std::remove(filename.c_str()); - - UnitCell ucell; - init_unitcell(ucell); - Parallel_Orbitals pv; - init_serial_orbitals(pv); - hamilt::HContainer matrix(&pv); - double values[4] = {0.25, 0.0, 0.0, 0.75}; - fill_matrix(matrix, pv, values); - - ModuleIO::write_dmr_csr(filename, &ucell, 4, &matrix, 0, 1, 2); - - const std::string output = read_file(filename); - EXPECT_THAT(output, testing::HasSubstr(" --- Ionic Step 1 ---\n")); - EXPECT_THAT(output, testing::HasSubstr(" # print density matrix in real space DM(R)\n")); - EXPECT_THAT(output, testing::HasSubstr(" 2 # number of spin directions\n")); - EXPECT_THAT(output, testing::HasSubstr(" 2 # spin index\n")); - EXPECT_THAT(output, testing::HasSubstr(" 2 # number of localized basis\n")); - EXPECT_THAT(output, testing::HasSubstr(" 1 # number of Bravais lattice vector R\n")); - - std::remove(filename.c_str()); -} - -TEST(WriteHsRCompatibility, LegacySparseHeaderKeepsStepStyle) -{ - const std::string filename = "write_hs_r_legacy_s.csr"; - std::remove(filename.c_str()); - - GlobalV::DRANK = 0; - PARAM.sys.global_out_dir = "./"; - PARAM.sys.nlocal = 99; - - Parallel_Orbitals pv; - init_serial_orbitals(pv); - const Abfs::Vector3_Order r_vector(0, 0, 0); - std::set> all_R_coor; - all_R_coor.insert(r_vector); - std::map, std::map>> sparse_matrix; - sparse_matrix[r_vector][0][1] = 0.5; - sparse_matrix[r_vector][1][1] = 1.5; - - ModuleIO::SparseWriteOptions options; - options.filename = filename; - options.label = "S"; - options.threshold = 1e-10; - options.binary = false; - options.istep = 0; - options.reduce = false; - options.temp_dir = "./"; - ModuleIO::save_sparse(sparse_matrix, all_R_coor, pv, options); - - const std::string output = read_file(filename); - EXPECT_TRUE(starts_with(output, "STEP: 0\n")); - EXPECT_THAT(output, testing::HasSubstr("Matrix Dimension of S(R): 2\n")); - EXPECT_THAT(output, testing::HasSubstr("Matrix number of S(R): 1\n")); - EXPECT_THAT(output, testing::HasSubstr("0 0 0 2\n")); - - std::remove(filename.c_str()); -} - -TEST(WriteHsRCompatibility, LegacySparseTextCountsOnlyValuesAboveThreshold) -{ - const std::string filename = "write_hs_r_threshold_s.csr"; - std::remove(filename.c_str()); - - GlobalV::DRANK = 0; - PARAM.sys.global_out_dir = "./"; - - Parallel_Orbitals pv; - init_serial_orbitals(pv); - const Abfs::Vector3_Order r_vector(0, 0, 0); - std::set> all_R_coor; - all_R_coor.insert(r_vector); - std::map, std::map>> sparse_matrix; - sparse_matrix[r_vector][0][0] = 1.0; - sparse_matrix[r_vector][0][1] = 1e-12; - sparse_matrix[r_vector][1][0] = 0.0; - sparse_matrix[r_vector][1][1] = -2.0; - - ModuleIO::SparseWriteOptions options; - options.filename = filename; - options.label = "S"; - options.threshold = 1e-10; - options.binary = false; - options.istep = 2; - options.reduce = false; - options.temp_dir = "./"; - ModuleIO::save_sparse(sparse_matrix, all_R_coor, pv, options); - - const std::vector lines = read_lines(filename); - ASSERT_GE(lines.size(), 6); - EXPECT_EQ(lines[0], "STEP: 2"); - EXPECT_EQ(lines[1], "Matrix Dimension of S(R): 2"); - EXPECT_EQ(lines[2], "Matrix number of S(R): 1"); - EXPECT_EQ(lines[3], "0 0 0 2"); - - std::istringstream value_stream(lines[4]); - std::vector values; - double value = 0.0; - while (value_stream >> value) - { - values.push_back(value); - } - EXPECT_THAT(values, testing::ElementsAre(1.0, -2.0)); - - std::istringstream column_stream(lines[5]); - std::vector columns; - int column = 0; - while (column_stream >> column) - { - columns.push_back(column); - } - EXPECT_THAT(columns, testing::ElementsAre(0, 1)); - - ASSERT_GE(lines.size(), 7); - std::istringstream indptr_stream(lines[6]); - std::vector indptr; - long long ptr = 0; - while (indptr_stream >> ptr) - { - indptr.push_back(ptr); - } - EXPECT_THAT(indptr, testing::ElementsAre(0, 1, 2)); - - std::remove(filename.c_str()); -} - -TEST(WriteHsRCompatibility, LegacySparseBinaryHeaderWritesConcreteStep) -{ - const std::string filename = "write_hs_r_legacy_binary_s.csr"; - std::remove(filename.c_str()); - - GlobalV::DRANK = 0; - PARAM.sys.global_out_dir = "./"; - PARAM.sys.nlocal = 99; - - Parallel_Orbitals pv; - init_serial_orbitals(pv); - const Abfs::Vector3_Order r_vector(0, 0, 0); - std::set> all_R_coor; - all_R_coor.insert(r_vector); - std::map, std::map>> sparse_matrix; - sparse_matrix[r_vector][0][1] = 0.5; - sparse_matrix[r_vector][1][1] = 1.5; - - ModuleIO::SparseWriteOptions options; - options.filename = filename; - options.label = "S"; - options.threshold = 1e-10; - options.binary = true; - options.istep = 3; - options.reduce = false; - options.temp_dir = "./"; - ModuleIO::save_sparse(sparse_matrix, all_R_coor, pv, options); - - const std::vector header_and_r = read_binary_ints(filename, 7); - EXPECT_THAT(header_and_r, testing::ElementsAre(3, 2, 1, 0, 0, 0, 2)); - - std::remove(filename.c_str()); -} - -TEST(WriteHsRCompatibility, LegacySparseBinaryCountsOnlyValuesAboveThreshold) -{ - const std::string filename = "write_hs_r_threshold_binary_s.csr"; - std::remove(filename.c_str()); - - GlobalV::DRANK = 0; - PARAM.sys.global_out_dir = "./"; - - Parallel_Orbitals pv; - init_serial_orbitals(pv); - const Abfs::Vector3_Order r_vector(0, 0, 0); - std::set> all_R_coor; - all_R_coor.insert(r_vector); - std::map, std::map>> sparse_matrix; - sparse_matrix[r_vector][0][0] = 1.0; - sparse_matrix[r_vector][0][1] = 1e-12; - sparse_matrix[r_vector][1][0] = 0.0; - sparse_matrix[r_vector][1][1] = -2.0; - - ModuleIO::SparseWriteOptions options; - options.filename = filename; - options.label = "S"; - options.threshold = 1e-10; - options.binary = true; - options.istep = 4; - options.reduce = false; - options.temp_dir = "./"; - ModuleIO::save_sparse(sparse_matrix, all_R_coor, pv, options); - - std::ifstream ifs(filename.c_str(), std::ios::binary); - ASSERT_TRUE(ifs.is_open()); - EXPECT_EQ(read_binary_value(ifs), 4); - EXPECT_EQ(read_binary_value(ifs), 2); - EXPECT_EQ(read_binary_value(ifs), 1); - EXPECT_EQ(read_binary_value(ifs), 0); - EXPECT_EQ(read_binary_value(ifs), 0); - EXPECT_EQ(read_binary_value(ifs), 0); - EXPECT_EQ(read_binary_value(ifs), 2); - EXPECT_DOUBLE_EQ(read_binary_value(ifs), 1.0); - EXPECT_DOUBLE_EQ(read_binary_value(ifs), -2.0); - EXPECT_EQ(read_binary_value(ifs), 0); - EXPECT_EQ(read_binary_value(ifs), 1); - EXPECT_EQ(read_binary_value(ifs), 0); - EXPECT_EQ(read_binary_value(ifs), 1); - EXPECT_EQ(read_binary_value(ifs), 2); - - std::remove(filename.c_str()); -} - -TEST(WriteHsRCompatibility, SaveDHSparseTextCountsOnlyValuesAboveThreshold) -{ - remove_derivative_files("h"); - init_sparse_output_globals(); - - Parallel_Orbitals pv; - init_serial_orbitals(pv); - LCAO_HS_Arrays arrays; - const Abfs::Vector3_Order r_vector(0, 0, 0); - arrays.all_R_coor.insert(r_vector); - arrays.dHRx_sparse[0][r_vector][0][0] = 1.0; - arrays.dHRx_sparse[0][r_vector][0][1] = 1e-12; - arrays.dHRx_sparse[0][r_vector][1][0] = 0.0; - arrays.dHRx_sparse[0][r_vector][1][1] = -2.0; - - ModuleIO::save_dH_sparse(5, pv, arrays, 1e-10, false, "h", 8); - - const std::vector lines = read_lines("dhrxs1_nao.csr"); - ASSERT_GE(lines.size(), 7); - EXPECT_EQ(lines[0], "STEP: 5"); - EXPECT_EQ(lines[1], "Matrix Dimension of dHx(R): 2"); - EXPECT_EQ(lines[2], "Matrix number of dHx(R): 1"); - EXPECT_EQ(lines[3], "0 0 0 2"); - EXPECT_THAT(lines[4], testing::HasSubstr("1.00000000e+00")); - EXPECT_THAT(lines[4], testing::HasSubstr("-2.00000000e+00")); - - std::istringstream column_stream(lines[5]); - std::vector columns; - int column = 0; - while (column_stream >> column) - { - columns.push_back(column); - } - EXPECT_THAT(columns, testing::ElementsAre(0, 1)); - - std::istringstream indptr_stream(lines[6]); - std::vector indptr; - long long ptr = 0; - while (indptr_stream >> ptr) - { - indptr.push_back(ptr); - } - EXPECT_THAT(indptr, testing::ElementsAre(0, 1, 2)); - - const std::vector y_lines = read_lines("dhrys1_nao.csr"); - ASSERT_GE(y_lines.size(), 4); - EXPECT_EQ(y_lines[3], "0 0 0 0"); - - remove_derivative_files("h"); -} - -TEST(WriteHsRCompatibility, SaveDHSparseBinaryCountsOnlyValuesAboveThreshold) -{ - remove_derivative_files("h"); - init_sparse_output_globals(); - - Parallel_Orbitals pv; - init_serial_orbitals(pv); - LCAO_HS_Arrays arrays; - const Abfs::Vector3_Order r_vector(0, 0, 0); - arrays.all_R_coor.insert(r_vector); - arrays.dHRx_sparse[0][r_vector][0][0] = 1.0; - arrays.dHRx_sparse[0][r_vector][0][1] = 1e-12; - arrays.dHRx_sparse[0][r_vector][1][0] = 0.0; - arrays.dHRx_sparse[0][r_vector][1][1] = -2.0; - - ModuleIO::save_dH_sparse(6, pv, arrays, 1e-10, true, "h", 8); - - std::ifstream ifs("dhrxs1_nao.csr", std::ios::binary); - ASSERT_TRUE(ifs.is_open()); - EXPECT_EQ(read_binary_value(ifs), 6); - EXPECT_EQ(read_binary_value(ifs), 2); - EXPECT_EQ(read_binary_value(ifs), 1); - EXPECT_EQ(read_binary_value(ifs), 0); - EXPECT_EQ(read_binary_value(ifs), 0); - EXPECT_EQ(read_binary_value(ifs), 0); - EXPECT_EQ(read_binary_value(ifs), 2); - EXPECT_DOUBLE_EQ(read_binary_value(ifs), 1.0); - EXPECT_DOUBLE_EQ(read_binary_value(ifs), -2.0); - EXPECT_EQ(read_binary_value(ifs), 0); - EXPECT_EQ(read_binary_value(ifs), 1); - EXPECT_EQ(read_binary_value(ifs), 0); - EXPECT_EQ(read_binary_value(ifs), 1); - EXPECT_EQ(read_binary_value(ifs), 2); - - remove_derivative_files("h"); -} - -TEST(WriteHsRCompatibility, SaveDSSparseSocWritesAllDirections) -{ - remove_derivative_files("s"); - init_sparse_output_globals(4); - - Parallel_Orbitals pv; - init_serial_orbitals(pv); - LCAO_HS_Arrays arrays; - const Abfs::Vector3_Order r_vector(0, 0, 0); - arrays.all_R_coor.insert(r_vector); - arrays.dHRx_soc_sparse[r_vector][0][0] = std::complex(1.0, 0.0); - arrays.dHRy_soc_sparse[r_vector][0][1] = std::complex(2.0, -1.0); - arrays.dHRz_soc_sparse[r_vector][1][1] = std::complex(-3.0, 0.5); - - ModuleIO::save_dH_sparse(7, pv, arrays, 1e-10, false, "s", 8); - - const std::string x_output = read_file("dsrxs1_nao.csr"); - const std::string y_output = read_file("dsrys1_nao.csr"); - const std::string z_output = read_file("dsrzs1_nao.csr"); - EXPECT_THAT(x_output, testing::HasSubstr("Matrix number of dHx(R): 1\n0 0 0 1\n")); - EXPECT_THAT(y_output, testing::HasSubstr("Matrix number of dHy(R): 1\n0 0 0 1\n")); - EXPECT_THAT(z_output, testing::HasSubstr("Matrix number of dHz(R): 1\n0 0 0 1\n")); - EXPECT_THAT(x_output, testing::HasSubstr("(1.00000000e+00,0.00000000e+00)")); - EXPECT_THAT(y_output, testing::HasSubstr("(2.00000000e+00,-1.00000000e+00)")); - EXPECT_THAT(z_output, testing::HasSubstr("(-3.00000000e+00,5.00000000e-01)")); - - remove_derivative_files("s"); -} - -TEST(WriteHsRCompatibility, MatSparseOutputOptionsKeepLegacyDefaults) -{ - ModuleIO::MatSparseOutputOptions options; - EXPECT_FALSE(options.out_mat_dh); - EXPECT_FALSE(options.out_mat_ds); - EXPECT_FALSE(options.out_mat_t); - EXPECT_FALSE(options.out_mat_r); - EXPECT_EQ(options.dh_precision, 16); - EXPECT_EQ(options.ds_precision, 16); - EXPECT_EQ(options.t_precision, 16); - EXPECT_EQ(options.r_precision, 16); - EXPECT_DOUBLE_EQ(options.sparse_threshold, 1e-10); - EXPECT_FALSE(options.binary); -} - -TEST(WriteHsRCompatibility, RRSparsePayloadDetectorSkipsEmptyBlocks) -{ - int empty_counts[3] = {0, 0, 0}; - int x_only_counts[3] = {1, 0, 0}; - int z_only_counts[3] = {0, 0, 2}; - - EXPECT_FALSE(ModuleIO::detail::rr_sparse_has_payload(empty_counts)); - EXPECT_TRUE(ModuleIO::detail::rr_sparse_has_payload(x_only_counts)); - EXPECT_TRUE(ModuleIO::detail::rr_sparse_has_payload(z_only_counts)); -} - -TEST(WriteHsRCompatibility, RRSparseTextFinalizerAllowsZeroBlocks) -{ - const std::string payload_filename = "rr_empty_payload.tmp"; - const std::string output_filename = "rr_empty.csr"; - std::remove(payload_filename.c_str()); - std::remove(output_filename.c_str()); - - std::ofstream payload(payload_filename.c_str()); - payload.close(); - - ModuleIO::detail::finalize_rr_sparse_file(output_filename, - payload_filename, - 9, - 2, - 0, - false, - false, - "WriteHsRCompatibility"); - - const std::vector lines = read_lines(output_filename); - ASSERT_EQ(lines.size(), 3); - EXPECT_EQ(lines[0], "STEP: 9"); - EXPECT_EQ(lines[1], "Matrix Dimension of r(R): 2"); - EXPECT_EQ(lines[2], "Matrix number of r(R): 0"); - - std::remove(payload_filename.c_str()); - std::remove(output_filename.c_str()); -} - -TEST(WriteHsRCompatibility, RRSparseTextFinalizerKeepsSingleDirectionPayload) -{ - const std::string payload_filename = "rr_single_direction_payload.tmp"; - const std::string output_filename = "rr_single_direction.csr"; - std::remove(payload_filename.c_str()); - std::remove(output_filename.c_str()); - - std::ofstream payload(payload_filename.c_str()); - payload << "1 0 -1\n"; - payload << "1\n"; - payload << " 4.00000000e+00\n"; - payload << " 0\n"; - payload << "0 1\n"; - payload << "0\n"; - payload << "0\n"; - payload.close(); - - ModuleIO::detail::finalize_rr_sparse_file(output_filename, - payload_filename, - 10, - 2, - 1, - false, - false, - "WriteHsRCompatibility"); - - const std::vector lines = read_lines(output_filename); - ASSERT_EQ(lines.size(), 10); - EXPECT_EQ(lines[0], "STEP: 10"); - EXPECT_EQ(lines[1], "Matrix Dimension of r(R): 2"); - EXPECT_EQ(lines[2], "Matrix number of r(R): 1"); - EXPECT_EQ(lines[3], "1 0 -1"); - EXPECT_EQ(lines[4], "1"); - EXPECT_EQ(lines[5], " 4.00000000e+00"); - EXPECT_EQ(lines[8], "0"); - EXPECT_EQ(lines[9], "0"); - - std::remove(payload_filename.c_str()); - std::remove(output_filename.c_str()); -} - -TEST(WriteHsRCompatibility, RRSparseBinaryFinalizerKeepsHeaderAndPayloadOrder) -{ - const std::string payload_filename = "rr_binary_payload.tmp"; - const std::string output_filename = "rr_binary.csr"; - std::remove(payload_filename.c_str()); - std::remove(output_filename.c_str()); - - std::ofstream payload(payload_filename.c_str(), std::ios::binary); - int dRx = 1; - int dRy = 2; - int dRz = 3; - int x_count = 1; - int y_count = 0; - int z_count = 0; - double value = 4.0; - int column = 1; - long long ptr0 = 0; - long long ptr1 = 1; - payload.write(reinterpret_cast(&dRx), sizeof(int)); - payload.write(reinterpret_cast(&dRy), sizeof(int)); - payload.write(reinterpret_cast(&dRz), sizeof(int)); - payload.write(reinterpret_cast(&x_count), sizeof(int)); - payload.write(reinterpret_cast(&value), sizeof(double)); - payload.write(reinterpret_cast(&column), sizeof(int)); - payload.write(reinterpret_cast(&ptr0), sizeof(long long)); - payload.write(reinterpret_cast(&ptr1), sizeof(long long)); - payload.write(reinterpret_cast(&y_count), sizeof(int)); - payload.write(reinterpret_cast(&z_count), sizeof(int)); - payload.close(); - - ModuleIO::detail::finalize_rr_sparse_file(output_filename, - payload_filename, - 11, - 2, - 1, - true, - false, - "WriteHsRCompatibility"); - - std::ifstream ifs(output_filename, std::ios::binary); - ASSERT_TRUE(ifs.is_open()); - EXPECT_EQ(read_binary_value(ifs), 11); - EXPECT_EQ(read_binary_value(ifs), 2); - EXPECT_EQ(read_binary_value(ifs), 1); - EXPECT_EQ(read_binary_value(ifs), 1); - EXPECT_EQ(read_binary_value(ifs), 2); - EXPECT_EQ(read_binary_value(ifs), 3); - EXPECT_EQ(read_binary_value(ifs), 1); - EXPECT_DOUBLE_EQ(read_binary_value(ifs), 4.0); - EXPECT_EQ(read_binary_value(ifs), 1); - EXPECT_EQ(read_binary_value(ifs), 0); - EXPECT_EQ(read_binary_value(ifs), 1); - EXPECT_EQ(read_binary_value(ifs), 0); - EXPECT_EQ(read_binary_value(ifs), 0); - ifs.close(); - - std::remove(payload_filename.c_str()); - std::remove(output_filename.c_str()); -} - -TEST(WriteHsRCompatibility, HeaderStyleSamplesRemainDistinct) -{ - EXPECT_TRUE(starts_with(" --- Ionic Step 1 ---", " --- Ionic Step")); - EXPECT_TRUE(starts_with("STEP: 0", "STEP:")); - EXPECT_TRUE(starts_with("IONIC_STEP: 1", "IONIC_STEP:")); -} - -int main(int argc, char** argv) -{ -#ifdef __MPI - MPI_Init(&argc, &argv); - MPI_Comm_size(MPI_COMM_WORLD, &GlobalV::NPROC); - MPI_Comm_rank(MPI_COMM_WORLD, &GlobalV::MY_RANK); -#endif - - ::testing::InitGoogleTest(&argc, argv); - const int result = RUN_ALL_TESTS(); - -#ifdef __MPI - MPI_Finalize(); -#endif - return result; -} diff --git a/source/source_io/test/write_hsk_binary_test.cpp b/source/source_io/test/write_hsk_binary_test.cpp deleted file mode 100644 index 65ff7ead569..00000000000 --- a/source/source_io/test/write_hsk_binary_test.cpp +++ /dev/null @@ -1,215 +0,0 @@ -#include "source_io/module_hs/write_hs.h" - -#include "gtest/gtest.h" - -#include -#include -#include -#include -#include - -#ifdef __MPI -#include -#endif - -namespace -{ - -int test_rank = 0; - -void initialize_distribution(Parallel_Orbitals& pv, const int dim) -{ -#ifdef __MPI - ASSERT_EQ(pv.init(dim, dim, 1, DIAG_WORLD), 0); -#else - pv.set_serial(dim, dim); -#endif -} - -template -std::vector distribute_matrix(const Parallel_Orbitals& pv, const std::vector& global, const int dim) -{ - std::vector local(pv.get_local_size(), T()); - for (int i = 0; i < dim; ++i) - { - const int ir = pv.global2local_row(i); - if (ir < 0) - { - continue; - } - for (int j = 0; j < dim; ++j) - { - const int ic = pv.global2local_col(j); - if (ic >= 0) - { - local[ir * pv.ncol + ic] = global[i * dim + j]; - } - } - } - return local; -} - -template -std::vector upper_triangle(const std::vector& matrix, const int dim) -{ - std::vector values; - values.reserve(dim * (dim + 1) / 2); - for (int i = 0; i < dim; ++i) - { - for (int j = i; j < dim; ++j) - { - values.push_back(matrix[i * dim + j]); - } - } - return values; -} - -template -std::vector read_record(std::ifstream& ifs, const int expected_dim) -{ - int dim = 0; - ifs.read(reinterpret_cast(&dim), sizeof(int)); - EXPECT_TRUE(ifs.good()); - EXPECT_EQ(dim, expected_dim); - - std::vector values(expected_dim * (expected_dim + 1) / 2); - ifs.read(reinterpret_cast(values.data()), values.size() * sizeof(T)); - EXPECT_TRUE(ifs.good()); - return values; -} - -void remove_test_file(const std::string& filename) -{ - if (test_rank == 0) - { - std::remove(filename.c_str()); - } -#ifdef __MPI - MPI_Barrier(DIAG_WORLD); -#endif -} - -} // namespace - -TEST(WriteHskBinary, WritesGammaUpperTriangle) -{ - const int dim = 3; - const std::string filename = "write_hsk_gamma.dat"; - remove_test_file(filename); - - Parallel_Orbitals pv; - initialize_distribution(pv, dim); - std::vector global(dim * dim); - for (int i = 0; i < dim * dim; ++i) - { - global[i] = i + 0.5; - } - const std::vector local = distribute_matrix(pv, global, dim); - - ModuleIO::save_mat(0, local.data(), dim, true, 8, true, false, filename, pv, test_rank); - - if (test_rank == 0) - { - std::ifstream ifs(filename.c_str(), std::ios::binary | std::ios::ate); - ASSERT_TRUE(ifs.is_open()); - EXPECT_EQ(ifs.tellg(), static_cast(sizeof(int) + 6 * sizeof(double))); - ifs.seekg(0); - EXPECT_EQ(read_record(ifs, dim), upper_triangle(global, dim)); - EXPECT_EQ(ifs.peek(), std::ifstream::traits_type::eof()); - } - - remove_test_file(filename); -} - -TEST(WriteHskBinary, WritesComplexUpperTriangleWithMpiReduction) -{ - const int dim = 4; - const std::string filename = "write_hsk_complex.dat"; - remove_test_file(filename); - - Parallel_Orbitals pv; - initialize_distribution(pv, dim); - std::vector> global(dim * dim); - for (int i = 0; i < dim * dim; ++i) - { - global[i] = std::complex(i + 0.25, -i - 0.75); - } - const std::vector> local = distribute_matrix(pv, global, dim); - - ModuleIO::save_mat(0, local.data(), dim, true, 8, true, false, filename, pv, test_rank); - - if (test_rank == 0) - { - std::ifstream ifs(filename.c_str(), std::ios::binary | std::ios::ate); - ASSERT_TRUE(ifs.is_open()); - const int element_count = dim * (dim + 1) / 2; - EXPECT_EQ(ifs.tellg(), static_cast(sizeof(int) + element_count * sizeof(std::complex))); - ifs.seekg(0); - EXPECT_EQ(read_record>(ifs, dim), upper_triangle(global, dim)); - EXPECT_EQ(ifs.peek(), std::ifstream::traits_type::eof()); - } - - remove_test_file(filename); -} - -TEST(WriteHskBinary, AppendsCompleteRecordsAndCanOverwrite) -{ - const int dim = 2; - const std::string filename = "write_hsk_append.dat"; - remove_test_file(filename); - - Parallel_Orbitals pv; - initialize_distribution(pv, dim); - const std::vector first = {1.0, 2.0, 3.0, 4.0}; - const std::vector second = {5.0, 6.0, 7.0, 8.0}; - const std::vector replacement = {9.0, 10.0, 11.0, 12.0}; - const std::vector first_local = distribute_matrix(pv, first, dim); - const std::vector second_local = distribute_matrix(pv, second, dim); - const std::vector replacement_local = distribute_matrix(pv, replacement, dim); - - ModuleIO::save_mat(0, first_local.data(), dim, true, 8, true, true, filename, pv, test_rank); - ModuleIO::save_mat(1, second_local.data(), dim, true, 8, true, true, filename, pv, test_rank); - - if (test_rank == 0) - { - std::ifstream ifs(filename.c_str(), std::ios::binary); - ASSERT_TRUE(ifs.is_open()); - EXPECT_EQ(read_record(ifs, dim), upper_triangle(first, dim)); - EXPECT_EQ(read_record(ifs, dim), upper_triangle(second, dim)); - EXPECT_EQ(ifs.peek(), std::ifstream::traits_type::eof()); - } - -#ifdef __MPI - MPI_Barrier(DIAG_WORLD); -#endif - ModuleIO::save_mat(2, replacement_local.data(), dim, true, 8, true, false, filename, pv, test_rank); - - if (test_rank == 0) - { - std::ifstream ifs(filename.c_str(), std::ios::binary); - ASSERT_TRUE(ifs.is_open()); - EXPECT_EQ(read_record(ifs, dim), upper_triangle(replacement, dim)); - EXPECT_EQ(ifs.peek(), std::ifstream::traits_type::eof()); - } - - remove_test_file(filename); -} - -int main(int argc, char** argv) -{ -#ifdef __MPI - MPI_Init(&argc, &argv); - DIAG_WORLD = MPI_COMM_WORLD; - MPI_Comm_rank(DIAG_WORLD, &test_rank); -#else - test_rank = 0; -#endif - - testing::InitGoogleTest(&argc, argv); - const int result = RUN_ALL_TESTS(); - -#ifdef __MPI - MPI_Finalize(); -#endif - return result; -} diff --git a/source/source_io/test/write_wfc_nao_test.cpp b/source/source_io/test/write_wfc_nao_test.cpp index 0692b358622..b5a64f2a336 100644 --- a/source/source_io/test/write_wfc_nao_test.cpp +++ b/source/source_io/test/write_wfc_nao_test.cpp @@ -1,9 +1,7 @@ #include "../module_wf/write_wfc_nao.h" #include "source_base/module_out/filename.h" -#define private public #include "source_io/module_parameter/parameter.h" -#undef private #include "source_base/module_out/binstream.h" #include "source_base/global_variable.h" #include "source_base/module_external/scalapack_connector.h" @@ -15,6 +13,18 @@ #include "mpi.h" #endif +/// @brief Friend helper to mutate PARAM private members in unit tests. +/// @details Parameter grants friend access to TestParameters so the test can +/// modify input/sys fields without `#define private public`. The class must +/// stay at global scope to match the friend declaration in parameter.h; +/// an anonymous-namespace class would not be the friend. +class TestParameters +{ + public: + static Input_para& input() { return PARAM.input; } + static System_para& sys() { return PARAM.sys; } +}; + TEST(GenWfcLcaoFnameTest, OutType1GammaOnlyOutAppFlagTrue) { const std::string directory = ""; @@ -205,7 +215,7 @@ class WriteWfcLcaoTest : public testing::Test TEST_F(WriteWfcLcaoTest, WriteWfcLcao) { - PARAM.sys.global_out_dir = "./"; + TestParameters::sys().global_out_dir = "./"; const std::string directory = ""; const std::string property = "wf"; @@ -248,7 +258,7 @@ TEST_F(WriteWfcLcaoTest, WriteWfcLcao) TEST_F(WriteWfcLcaoTest, WriteWfcLcaoComplex) { - PARAM.sys.global_out_dir = "./"; + TestParameters::sys().global_out_dir = "./"; const std::string directory = ""; const std::string property = "wf"; @@ -295,8 +305,8 @@ TEST(ModuleIOTest, WriteWfcNao) { // Set up GlobalV GlobalV::DRANK = 0; - PARAM.input.nbands = 2; - PARAM.sys.nlocal = 2; + TestParameters::input().nbands = 2; + TestParameters::sys().nlocal = 2; // Set up test data std::string filename = "test_wfc_nao.txt"; @@ -313,7 +323,7 @@ TEST(ModuleIOTest, WriteWfcNao) wg(1, 1) = 1.2; // Call the function to be tested - ModuleIO::wfc_nao_write2file(filename, ctot.data(), PARAM.sys.nlocal, 0, ekb, wg, false); + ModuleIO::wfc_nao_write2file(filename, ctot.data(), TestParameters::sys().nlocal, 0, ekb, wg, false); // Check the output file std::ifstream ifs(filename); @@ -341,8 +351,8 @@ TEST(ModuleIOTest, WriteWfcNaoBinary) { // Set up GlobalV GlobalV::DRANK = 0; - PARAM.input.nbands = 2; - PARAM.sys.nlocal = 2; + TestParameters::input().nbands = 2; + TestParameters::sys().nlocal = 2; // Set up test data std::string filename = "test_wfc_nao.dat"; @@ -359,7 +369,7 @@ TEST(ModuleIOTest, WriteWfcNaoBinary) wg(1, 1) = 1.2; // Call the function to be tested - ModuleIO::wfc_nao_write2file(filename, ctot.data(), PARAM.sys.nlocal, 0, ekb, wg, true); + ModuleIO::wfc_nao_write2file(filename, ctot.data(), TestParameters::sys().nlocal, 0, ekb, wg, true); // Check the output file Binstream wfc(filename, "r"); @@ -397,8 +407,8 @@ TEST(ModuleIOTest, WriteWfcNaoComplex) if (GlobalV::MY_RANK == 0) { // Set up GlobalV - PARAM.input.nbands = 2; - PARAM.sys.nlocal = 3; + TestParameters::input().nbands = 2; + TestParameters::sys().nlocal = 3; // set up test data std::string name = "test_wfc_nao_complex.txt"; int ik = 0; @@ -417,7 +427,7 @@ TEST(ModuleIOTest, WriteWfcNaoComplex) std::complex(0.0, 3.0)}; // Call the function - ModuleIO::wfc_nao_write2file_complex(name, ctot.data(), PARAM.sys.nlocal, ik, kvec_c, ekb, wg); + ModuleIO::wfc_nao_write2file_complex(name, ctot.data(), TestParameters::sys().nlocal, ik, kvec_c, ekb, wg); // Check the output file std::ifstream ifs(name); std::string str((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); @@ -440,8 +450,8 @@ TEST(ModuleIOTest, WriteWfcNaoComplexBinary) if (GlobalV::MY_RANK == 0) { // Set up GlobalV - PARAM.input.nbands = 2; - PARAM.sys.nlocal = 3; + TestParameters::input().nbands = 2; + TestParameters::sys().nlocal = 3; // set up test data std::string name = "test_wfc_nao_complex.dat"; int ik = 0; @@ -460,7 +470,7 @@ TEST(ModuleIOTest, WriteWfcNaoComplexBinary) std::complex(6.0, 8.0)}; // Call the function - ModuleIO::wfc_nao_write2file_complex(name, ctot.data(), PARAM.sys.nlocal, ik, kvec_c, ekb, wg, true); + ModuleIO::wfc_nao_write2file_complex(name, ctot.data(), TestParameters::sys().nlocal, ik, kvec_c, ekb, wg, true); // Check the output file Binstream wfc(name, "r"); @@ -502,6 +512,61 @@ TEST(ModuleIOTest, WriteWfcNaoComplexBinary) } } +TEST(ModuleIOTest, WriteWfcNaoBinaryOpenFail) +{ +#ifdef __MPI + int my_rank = 0; + MPI_Comm_rank(MPI_COMM_WORLD, &my_rank); + if (my_rank != 0) + { + return; + } +#endif + const int nlocal = 2; + std::string filename = "/nonexistent_dir/test_wfc_nao.dat"; + std::vector ctot = {0.1, 0.2, 0.3, 0.4}; + ModuleBase::matrix ekb(2, 2); + ModuleBase::matrix wg(2, 2); + + testing::internal::CaptureStdout(); + EXPECT_EXIT(ModuleIO::wfc_nao_write2file(filename, ctot.data(), nlocal, 0, ekb, wg, true), + ::testing::ExitedWithCode(1), + ""); + std::string output = testing::internal::GetCapturedStdout(); + EXPECT_THAT(output, testing::HasSubstr("TIME STATISTICS")); +} + +TEST(ModuleIOTest, WriteWfcNaoComplexBinaryOpenFail) +{ +#ifdef __MPI + int my_rank = 0; + MPI_Comm_rank(MPI_COMM_WORLD, &my_rank); + if (my_rank != 0) + { + return; + } +#endif + const int nlocal = 3; + std::string name = "/nonexistent_dir/test_wfc_nao_complex.dat"; + int ik = 0; + ModuleBase::Vector3 kvec_c{0.0, 0.0, 0.0}; + ModuleBase::matrix ekb(1, 2); + ModuleBase::matrix wg(1, 2); + std::vector> ctot = {std::complex(1.0, 0.0), + std::complex(2.0, 0.0), + std::complex(3.0, 0.0), + std::complex(0.0, 1.0), + std::complex(0.0, 2.0), + std::complex(0.0, 3.0)}; + + testing::internal::CaptureStdout(); + EXPECT_EXIT(ModuleIO::wfc_nao_write2file_complex(name, ctot.data(), nlocal, ik, kvec_c, ekb, wg, true), + ::testing::ExitedWithCode(1), + ""); + std::string output = testing::internal::GetCapturedStdout(); + EXPECT_THAT(output, testing::HasSubstr("TIME STATISTICS")); +} + int main(int argc, char** argv) { GlobalV::MY_RANK = 0; diff --git a/source/source_io/test_serial/CMakeLists.txt b/source/source_io/test_serial/CMakeLists.txt index f5ca580cc6b..3238fb743d4 100644 --- a/source/source_io/test_serial/CMakeLists.txt +++ b/source/source_io/test_serial/CMakeLists.txt @@ -54,6 +54,12 @@ AddTest( SOURCES rho_io_test.cpp ../module_output/read_cube.cpp ../module_output/write_cube.cpp ) +AddTest( + TARGET MODULE_IO_read_cube + LIBS parameter base device cell_info + SOURCES test_read_cube.cpp ../module_output/read_cube.cpp +) + AddTest( TARGET MODULE_IO_system_variable_test LIBS parameter base device io_input_serial diff --git a/source/source_io/test_serial/io_sys_var_test.cpp b/source/source_io/test_serial/io_sys_var_test.cpp index 235f01b8551..678ee6a81b4 100644 --- a/source/source_io/test_serial/io_sys_var_test.cpp +++ b/source/source_io/test_serial/io_sys_var_test.cpp @@ -6,6 +6,11 @@ #include "gmock/gmock.h" #include "gtest/gtest.h" #include "source_base/tool_quit.h" +#include "source_io/module_parameter/input_parameter.h" +#define private public +#include "source_io/module_parameter/read_input.h" +#undef private +#include "source_io/module_parameter/system_parameter.h" /************************************************ * unit test of read_input_test_item.cpp ***********************************************/ @@ -15,10 +20,6 @@ * - Item_test: * - read in specific values for some items */ -#define private public -#include "source_io/module_parameter/input_item.h" -#include "source_io/module_parameter/read_input.h" -#undef private class InputTest : public testing::Test { @@ -34,51 +35,68 @@ class InputTest : public testing::Test return it; } }; + +/// @brief Friend helper to invoke ReadInput private setters in unit tests. +/// @details ReadInput grants friend access to TestParameters so the test can +/// drive its private setup routines without `#define private public`. The +/// class must stay at global scope to match the friend declaration in +/// read_input.h; an anonymous-namespace class would not be the friend. +class TestParameters +{ + public: + static void set_global_dir(ModuleIO::ReadInput& readinput, const Input_para& inp, System_para& sys) + { + readinput.set_global_dir(inp, sys); + } + static void set_globalv(ModuleIO::ReadInput& readinput, const Input_para& inp, System_para& sys) + { + readinput.set_globalv(inp, sys); + } +}; ModuleIO::ReadInput readinput(0); -Parameter param; +Input_para input; +System_para sys; std::string output = ""; TEST_F(InputTest, Item_test) { readinput.check_ntype_flag = false; - { - param.input.suffix = "test"; - readinput.set_global_dir(param.inp, param.sys); + { + input.suffix = "test"; + TestParameters::set_global_dir(readinput, input, sys); + + EXPECT_EQ(sys.global_out_dir, "OUT.test/"); + EXPECT_EQ(sys.global_stru_dir, "OUT.test/STRU/"); + EXPECT_EQ(sys.global_matrix_dir, "OUT.test/matrix/"); - EXPECT_EQ(param.sys.global_out_dir, "OUT.test/"); - EXPECT_EQ(param.sys.global_stru_dir, "OUT.test/STRU/"); - EXPECT_EQ(param.sys.global_matrix_dir, "OUT.test/matrix/"); + TestParameters::set_globalv(readinput, input, sys); - readinput.set_globalv(param.inp, param.sys); - - param.input.basis_type = "lcao"; - param.input.gamma_only = true; - param.input.esolver_type = "tddft"; - param.input.nspin = 2; - readinput.set_globalv(param.inp, param.sys); - EXPECT_EQ(param.sys.gamma_only_local, 0); - - param.input.deepks_scf = true; - param.input.deepks_out_labels = true; - readinput.set_globalv(param.inp, param.sys); - EXPECT_EQ(param.sys.deepks_setorb, 1); + input.basis_type = "lcao"; + input.gamma_only = true; + input.esolver_type = "tddft"; + input.nspin = 2; + TestParameters::set_globalv(readinput, input, sys); + EXPECT_EQ(sys.gamma_only_local, 0); - param.input.nspin = 4; - param.input.noncolin = true; - readinput.set_globalv(param.inp, param.sys); - EXPECT_EQ(param.sys.domag, 1); - EXPECT_EQ(param.sys.domag_z, 0); - EXPECT_EQ(param.sys.npol, 2); + input.deepks_scf = true; + input.deepks_out_labels = true; + TestParameters::set_globalv(readinput, input, sys); + EXPECT_EQ(sys.deepks_setorb, 1); - param.input.nspin = 1; - param.input.lspinorb = true; - param.input.noncolin = false; - readinput.set_globalv(param.inp, param.sys); - EXPECT_EQ(param.sys.domag, 0); - EXPECT_EQ(param.sys.domag_z, 0); - EXPECT_EQ(param.sys.npol, 1); + input.nspin = 4; + input.noncolin = true; + TestParameters::set_globalv(readinput, input, sys); + EXPECT_EQ(sys.domag, 1); + EXPECT_EQ(sys.domag_z, 0); + EXPECT_EQ(sys.npol, 2); - + input.nspin = 1; + input.lspinorb = true; + input.noncolin = false; + TestParameters::set_globalv(readinput, input, sys); + EXPECT_EQ(sys.domag, 0); + EXPECT_EQ(sys.domag_z, 0); + EXPECT_EQ(sys.npol, 1); } -} \ No newline at end of file +} diff --git a/source/source_io/test_serial/read_input_item_test.cpp b/source/source_io/test_serial/read_input_item_test.cpp index 4c97d9169d3..ffe37e6ba80 100644 --- a/source/source_io/test_serial/read_input_item_test.cpp +++ b/source/source_io/test_serial/read_input_item_test.cpp @@ -15,11 +15,25 @@ * - Item_test: * - read in specific values for some items */ -#define private public #include "source_io/module_parameter/input_item.h" +#include "source_io/module_parameter/parameter.h" +#define private public #include "source_io/module_parameter/read_input.h" #undef private +/// @brief Friend helper to mutate Parameter private members in unit tests. +/// @details Parameter grants friend access to TestParameters so the test can +/// modify input/sys fields of a local Parameter instance without +/// `#define private public`. The class must stay at global scope to match the +/// friend declaration in parameter.h; an anonymous-namespace class would not +/// be the friend. +class TestParameters +{ + public: + static Input_para& input(Parameter& param) { return param.input; } + static System_para& sys(Parameter& param) { return param.sys; } +}; + class InputTest : public testing::Test { protected: @@ -44,31 +58,31 @@ TEST_F(InputTest, RelaxMethod) it->second.str_values = {"cg"}; it->second.read_value(it->second, param); - EXPECT_EQ(param.input.relax_method, (std::vector{"cg", "2"})); - EXPECT_TRUE(param.input.uses_simultaneous_relaxation()); + EXPECT_EQ(TestParameters::input(param).relax_method, (std::vector{"cg", "2"})); + EXPECT_TRUE(TestParameters::input(param).uses_simultaneous_relaxation()); it->second.str_values = {"cg", "1"}; it->second.read_value(it->second, param); - EXPECT_EQ(param.input.relax_method, (std::vector{"cg", "1"})); - EXPECT_FALSE(param.input.uses_simultaneous_relaxation()); + EXPECT_EQ(TestParameters::input(param).relax_method, (std::vector{"cg", "1"})); + EXPECT_FALSE(TestParameters::input(param).uses_simultaneous_relaxation()); it->second.str_values = {"cg", "2"}; it->second.read_value(it->second, param); - EXPECT_EQ(param.input.relax_method, (std::vector{"cg", "2"})); - EXPECT_TRUE(param.input.uses_simultaneous_relaxation()); + EXPECT_EQ(TestParameters::input(param).relax_method, (std::vector{"cg", "2"})); + EXPECT_TRUE(TestParameters::input(param).uses_simultaneous_relaxation()); it->second.str_values = {"bfgs"}; it->second.read_value(it->second, param); - EXPECT_EQ(param.input.relax_method, (std::vector{"bfgs", "2"})); - EXPECT_FALSE(param.input.uses_simultaneous_relaxation()); + EXPECT_EQ(TestParameters::input(param).relax_method, (std::vector{"bfgs", "2"})); + EXPECT_FALSE(TestParameters::input(param).uses_simultaneous_relaxation()); it->second.str_values = {"bfgs", "1"}; it->second.read_value(it->second, param); - EXPECT_EQ(param.input.relax_method, (std::vector{"bfgs", "1"})); + EXPECT_EQ(TestParameters::input(param).relax_method, (std::vector{"bfgs", "1"})); it->second.str_values = {"bfgs", "2"}; it->second.read_value(it->second, param); - EXPECT_EQ(param.input.relax_method, (std::vector{"bfgs", "2"})); + EXPECT_EQ(TestParameters::input(param).relax_method, (std::vector{"bfgs", "2"})); for (const std::vector& invalid : { std::vector{"cg", "3"}, @@ -94,27 +108,27 @@ TEST_F(InputTest, Item_test) { // calculation auto it = find_label("calculation", readinput.input_lists); - param.input.calculation = "get_pchg"; - param.input.basis_type = "pw"; + TestParameters::input(param).calculation = "get_pchg"; + TestParameters::input(param).basis_type = "pw"; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); EXPECT_THAT(output, testing::HasSubstr("NOTICE")); - param.input.calculation = "gen_bessel"; - param.input.basis_type = "lcao"; + TestParameters::input(param).calculation = "gen_bessel"; + TestParameters::input(param).basis_type = "lcao"; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); EXPECT_THAT(output, testing::HasSubstr("NOTICE")); - param.input.calculation = "none"; + TestParameters::input(param).calculation = "none"; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); EXPECT_THAT(output, testing::HasSubstr("NOTICE")); - param.input.calculation = "socket"; + TestParameters::input(param).calculation = "socket"; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -123,35 +137,35 @@ TEST_F(InputTest, Item_test) { // socket_driver auto it = find_label("socket_driver", readinput.input_lists); - param.input.socket_driver = true; - param.input.calculation = "nscf"; + TestParameters::input(param).socket_driver = true; + TestParameters::input(param).calculation = "nscf"; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); EXPECT_THAT(output, testing::HasSubstr("NOTICE")); - param.input.socket_driver = true; - param.input.calculation = "scf"; + TestParameters::input(param).socket_driver = true; + TestParameters::input(param).calculation = "scf"; EXPECT_NO_THROW(it->second.check_value(it->second, param)); - param.input.socket_driver = false; + TestParameters::input(param).socket_driver = false; } { // esolver_type auto it = find_label("esolver_type", readinput.input_lists); - param.input.esolver_type = "none"; + TestParameters::input(param).esolver_type = "none"; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); EXPECT_THAT(output, testing::HasSubstr("NOTICE")); - param.input.esolver_type = "dp"; + TestParameters::input(param).esolver_type = "dp"; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); EXPECT_THAT(output, testing::HasSubstr("NOTICE")); - param.input.esolver_type = "lr"; - param.input.calculation = "scf"; + TestParameters::input(param).esolver_type = "lr"; + TestParameters::input(param).calculation = "scf"; it = find_label("esolver_type", readinput.input_lists); testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); @@ -160,8 +174,8 @@ TEST_F(InputTest, Item_test) } { // nspin auto it = find_label("nspin", readinput.input_lists); - param.input.noncolin = false; - param.input.nspin = 3; + TestParameters::input(param).noncolin = false; + TestParameters::input(param).nspin = 3; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -169,7 +183,7 @@ TEST_F(InputTest, Item_test) } { // smearing_method auto it = find_label("smearing_method", readinput.input_lists); - param.input.smearing_method = "fix"; + TestParameters::input(param).smearing_method = "fix"; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -179,22 +193,22 @@ TEST_F(InputTest, Item_test) auto it = find_label("kspacing", readinput.input_lists); it->second.str_values = {"1"}; it->second.read_value(it->second, param); - EXPECT_EQ(param.input.kspacing[0], 1); - EXPECT_EQ(param.input.kspacing[1], 1); - EXPECT_EQ(param.input.kspacing[2], 1); + EXPECT_EQ(TestParameters::input(param).kspacing[0], 1); + EXPECT_EQ(TestParameters::input(param).kspacing[1], 1); + EXPECT_EQ(TestParameters::input(param).kspacing[2], 1); it->second.str_values = {"1", "2"}; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.read_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); EXPECT_THAT(output, testing::HasSubstr("NOTICE")); - param.input.kspacing = {0, -1, 1}; + TestParameters::input(param).kspacing = {0, -1, 1}; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); EXPECT_THAT(output, testing::HasSubstr("NOTICE")); - param.input.kspacing = {0, 1, 2}; + TestParameters::input(param).kspacing = {0, 1, 2}; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -202,7 +216,7 @@ TEST_F(InputTest, Item_test) } { // nbands auto it = find_label("nbands", readinput.input_lists); - param.input.nbands = -1; + TestParameters::input(param).nbands = -1; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -210,45 +224,45 @@ TEST_F(InputTest, Item_test) } { // symmetry auto it = find_label("symmetry", readinput.input_lists); - param.input.symmetry = "default"; - param.input.gamma_only = true; + TestParameters::input(param).symmetry = "default"; + TestParameters::input(param).gamma_only = true; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.symmetry, "0"); + EXPECT_EQ(TestParameters::input(param).symmetry, "0"); - param.input.symmetry = "default"; - param.input.gamma_only = false; - param.input.calculation = "none"; + TestParameters::input(param).symmetry = "default"; + TestParameters::input(param).gamma_only = false; + TestParameters::input(param).calculation = "none"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.symmetry, "1"); + EXPECT_EQ(TestParameters::input(param).symmetry, "1"); - param.input.calculation = "md"; + TestParameters::input(param).calculation = "md"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.symmetry, "0"); + EXPECT_EQ(TestParameters::input(param).symmetry, "0"); - param.input.symmetry = "default"; - param.input.efield_flag = true; + TestParameters::input(param).symmetry = "default"; + TestParameters::input(param).efield_flag = true; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.symmetry, "0"); + EXPECT_EQ(TestParameters::input(param).symmetry, "0"); - param.input.qo_switch = true; + TestParameters::input(param).qo_switch = true; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.symmetry, "-1"); + EXPECT_EQ(TestParameters::input(param).symmetry, "-1"); - param.input.symmetry = "default"; - param.input.berry_phase = true; + TestParameters::input(param).symmetry = "default"; + TestParameters::input(param).berry_phase = true; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.symmetry, "-1"); + EXPECT_EQ(TestParameters::input(param).symmetry, "-1"); } { // nelec auto it = find_label("nelec", readinput.input_lists); - param.input.nelec = -1; + TestParameters::input(param).nelec = -1; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); EXPECT_THAT(output, testing::HasSubstr("NOTICE")); - param.input.nelec = 100; - param.input.nbands = 5; + TestParameters::input(param).nelec = 100; + TestParameters::input(param).nbands = 5; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -256,14 +270,14 @@ TEST_F(InputTest, Item_test) } { // bndpar auto it = find_label("bndpar", readinput.input_lists); - param.input.esolver_type = "sdft"; - param.input.bndpar = 1; + TestParameters::input(param).esolver_type = "sdft"; + TestParameters::input(param).bndpar = 1; EXPECT_NO_THROW(it->second.check_value(it->second, param)); } { // dft_plus_dmft auto it = find_label("dft_plus_dmft", readinput.input_lists); - param.input.basis_type = "pw"; - param.input.dft_plus_dmft = true; + TestParameters::input(param).basis_type = "pw"; + TestParameters::input(param).dft_plus_dmft = true; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -271,27 +285,27 @@ TEST_F(InputTest, Item_test) } { // mem_saver auto it = find_label("mem_saver", readinput.input_lists); - param.input.mem_saver = 1; - param.input.calculation = "scf"; + TestParameters::input(param).mem_saver = 1; + TestParameters::input(param).calculation = "scf"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.mem_saver, 0); + EXPECT_EQ(TestParameters::input(param).mem_saver, 0); - param.input.mem_saver = 1; - param.input.calculation = "relax"; + TestParameters::input(param).mem_saver = 1; + TestParameters::input(param).calculation = "relax"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.mem_saver, 0); + EXPECT_EQ(TestParameters::input(param).mem_saver, 0); } { // gint_precision auto it = find_label("gint_precision", readinput.input_lists); - param.input.gint_precision = "invalid"; + TestParameters::input(param).gint_precision = "invalid"; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); EXPECT_THAT(output, testing::HasSubstr("NOTICE")); - param.input.gint_precision = "mix"; - param.input.basis_type = "pw"; - param.input.device = "cpu"; + TestParameters::input(param).gint_precision = "mix"; + TestParameters::input(param).basis_type = "pw"; + TestParameters::input(param).device = "cpu"; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -299,45 +313,45 @@ TEST_F(InputTest, Item_test) } { // diag_proc auto it = find_label("diago_proc", readinput.input_lists); - param.input.diago_proc = 0; + TestParameters::input(param).diago_proc = 0; GlobalV::NPROC = 1; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.diago_proc, 1); + EXPECT_EQ(TestParameters::input(param).diago_proc, 1); } { // cal_force auto it = find_label("cal_force", readinput.input_lists); - param.input.calculation = "cell-relax"; - param.input.cal_force = false; - param.input.socket_driver = false; + TestParameters::input(param).calculation = "cell-relax"; + TestParameters::input(param).cal_force = false; + TestParameters::input(param).socket_driver = false; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.cal_force, true); + EXPECT_EQ(TestParameters::input(param).cal_force, true); - param.input.calculation = "get_wf"; - param.input.cal_force = true; + TestParameters::input(param).calculation = "get_wf"; + TestParameters::input(param).cal_force = true; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.cal_force, false); + EXPECT_EQ(TestParameters::input(param).cal_force, false); - param.input.calculation = "scf"; - param.input.socket_driver = true; - param.input.cal_force = false; + TestParameters::input(param).calculation = "scf"; + TestParameters::input(param).socket_driver = true; + TestParameters::input(param).cal_force = false; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.cal_force, false); - param.input.socket_driver = false; + EXPECT_EQ(TestParameters::input(param).cal_force, false); + TestParameters::input(param).socket_driver = false; } { // ecutrho auto it = find_label("ecutrho", readinput.input_lists); - param.input.ecutwfc = 1; - param.input.ecutrho = 0; + TestParameters::input(param).ecutwfc = 1; + TestParameters::input(param).ecutrho = 0; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.ecutrho, 4); - param.input.nx = 0; - param.input.ecutrho = 5; - param.input.ecutwfc = 1; + EXPECT_EQ(TestParameters::input(param).ecutrho, 4); + TestParameters::input(param).nx = 0; + TestParameters::input(param).ecutrho = 5; + TestParameters::input(param).ecutwfc = 1; it->second.reset_value(it->second, param); - EXPECT_EQ(param.sys.double_grid, true); + EXPECT_EQ(TestParameters::sys(param).double_grid, true); - param.input.ecutwfc = 1; - param.input.ecutrho = 1; + TestParameters::input(param).ecutwfc = 1; + TestParameters::input(param).ecutrho = 1; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -345,15 +359,15 @@ TEST_F(InputTest, Item_test) } { // pw_diag_thr auto it = find_label("pw_diag_thr", readinput.input_lists); - param.input.pw_diag_thr = 1.0e-2; - param.input.calculation = "get_s"; - param.input.basis_type = "pw"; + TestParameters::input(param).pw_diag_thr = 1.0e-2; + TestParameters::input(param).calculation = "get_s"; + TestParameters::input(param).basis_type = "pw"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.pw_diag_thr, 1.0e-5); + EXPECT_EQ(TestParameters::input(param).pw_diag_thr, 1.0e-5); } { // nb2d auto it = find_label("nb2d", readinput.input_lists); - param.input.nb2d = -1; + TestParameters::input(param).nb2d = -1; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -361,73 +375,73 @@ TEST_F(InputTest, Item_test) } { // scf_thr auto it = find_label("scf_thr", readinput.input_lists); - param.input.scf_thr = -1; - param.input.basis_type = "lcao"; + TestParameters::input(param).scf_thr = -1; + TestParameters::input(param).basis_type = "lcao"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.scf_thr, 1.0e-7); + EXPECT_EQ(TestParameters::input(param).scf_thr, 1.0e-7); - param.input.scf_thr = -1; - param.input.basis_type = "pw"; - param.input.calculation = "nscf"; + TestParameters::input(param).scf_thr = -1; + TestParameters::input(param).basis_type = "pw"; + TestParameters::input(param).calculation = "nscf"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.scf_thr, 1.0e-6); + EXPECT_EQ(TestParameters::input(param).scf_thr, 1.0e-6); - param.input.scf_thr = -1; - param.input.basis_type = "pw"; - param.input.calculation = "scf"; + TestParameters::input(param).scf_thr = -1; + TestParameters::input(param).basis_type = "pw"; + TestParameters::input(param).calculation = "scf"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.scf_thr, 1.0e-9); + EXPECT_EQ(TestParameters::input(param).scf_thr, 1.0e-9); } { // scf_thr_type auto it = find_label("scf_thr_type", readinput.input_lists); - param.input.scf_thr_type = -1; - param.input.basis_type = "lcao"; + TestParameters::input(param).scf_thr_type = -1; + TestParameters::input(param).basis_type = "lcao"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.scf_thr_type, 2); + EXPECT_EQ(TestParameters::input(param).scf_thr_type, 2); - param.input.scf_thr_type = -1; - param.input.basis_type = "pw"; + TestParameters::input(param).scf_thr_type = -1; + TestParameters::input(param).basis_type = "pw"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.scf_thr_type, 1); + EXPECT_EQ(TestParameters::input(param).scf_thr_type, 1); } { // init_wfc auto it = find_label("init_wfc", readinput.input_lists); ASSERT_NE(it, readinput.input_lists.end()); EXPECT_EQ(it->second.type, "Vector of string"); - EXPECT_EQ(param.input.init_wfc, "atomic"); - EXPECT_TRUE(param.input.init_wfc_file_format.empty()); + EXPECT_EQ(TestParameters::input(param).init_wfc, "atomic"); + EXPECT_TRUE(TestParameters::input(param).init_wfc_file_format.empty()); - param.input.basis_type = "lcao"; - param.input.calculation = "scf"; + TestParameters::input(param).basis_type = "lcao"; + TestParameters::input(param).calculation = "scf"; it->second.str_values = {"file", "txt"}; it->second.read_value(it->second, param); it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.init_wfc, "file"); - EXPECT_EQ(param.input.init_wfc_file_format, "txt"); + EXPECT_EQ(TestParameters::input(param).init_wfc, "file"); + EXPECT_EQ(TestParameters::input(param).init_wfc_file_format, "txt"); EXPECT_NO_THROW(it->second.check_value(it->second, param)); it->second.str_values = {"file", "binary"}; it->second.read_value(it->second, param); it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.init_wfc, "file"); - EXPECT_EQ(param.input.init_wfc_file_format, "binary"); + EXPECT_EQ(TestParameters::input(param).init_wfc, "file"); + EXPECT_EQ(TestParameters::input(param).init_wfc_file_format, "binary"); EXPECT_NO_THROW(it->second.check_value(it->second, param)); it->second.str_values = {"file"}; it->second.read_value(it->second, param); - EXPECT_TRUE(param.input.init_wfc_file_format.empty()); + EXPECT_TRUE(TestParameters::input(param).init_wfc_file_format.empty()); it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.init_wfc_file_format, "txt"); + EXPECT_EQ(TestParameters::input(param).init_wfc_file_format, "txt"); it->second.final_value.str(""); it->second.final_value.clear(); it->second.get_final_value(it->second, param); EXPECT_EQ(it->second.final_value.str(), "file txt"); - param.input.basis_type = "pw"; + TestParameters::input(param).basis_type = "pw"; it->second.str_values = {"file"}; it->second.read_value(it->second, param); it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.init_wfc_file_format, "binary"); + EXPECT_EQ(TestParameters::input(param).init_wfc_file_format, "binary"); it->second.final_value.str(""); it->second.final_value.clear(); it->second.get_final_value(it->second, param); @@ -435,34 +449,34 @@ TEST_F(InputTest, Item_test) it->second.str_values = {"atomic"}; it->second.read_value(it->second, param); - param.input.calculation = "get_pchg"; - param.input.basis_type = "lcao"; + TestParameters::input(param).calculation = "get_pchg"; + TestParameters::input(param).basis_type = "lcao"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.init_wfc, "file"); - EXPECT_EQ(param.input.init_wfc_file_format, "txt"); + EXPECT_EQ(TestParameters::input(param).init_wfc, "file"); + EXPECT_EQ(TestParameters::input(param).init_wfc_file_format, "txt"); it->second.str_values = {"atomic"}; it->second.read_value(it->second, param); - param.input.calculation = "get_wf"; - param.input.basis_type = "pw"; + TestParameters::input(param).calculation = "get_wf"; + TestParameters::input(param).basis_type = "pw"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.init_wfc, "file"); - EXPECT_EQ(param.input.init_wfc_file_format, "binary"); + EXPECT_EQ(TestParameters::input(param).init_wfc, "file"); + EXPECT_EQ(TestParameters::input(param).init_wfc_file_format, "binary"); it->second.str_values = {"file", "binary"}; it->second.read_value(it->second, param); - param.input.calculation = "get_wf"; - param.input.basis_type = "lcao"; + TestParameters::input(param).calculation = "get_wf"; + TestParameters::input(param).basis_type = "lcao"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.init_wfc_file_format, "binary"); + EXPECT_EQ(TestParameters::input(param).init_wfc_file_format, "binary"); it->second.str_values = {"file", "binary"}; it->second.read_value(it->second, param); - param.input.calculation = "scf"; - param.input.basis_type = "lcao_in_pw"; + TestParameters::input(param).calculation = "scf"; + TestParameters::input(param).basis_type = "lcao_in_pw"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.init_wfc, "nao"); - EXPECT_TRUE(param.input.init_wfc_file_format.empty()); + EXPECT_EQ(TestParameters::input(param).init_wfc, "nao"); + EXPECT_TRUE(TestParameters::input(param).init_wfc_file_format.empty()); for (const std::vector& invalid : { std::vector{}, @@ -475,7 +489,7 @@ TEST_F(InputTest, Item_test) EXPECT_EXIT(it->second.read_value(it->second, param), ::testing::ExitedWithCode(1), ""); } - param.input.basis_type = "pw"; + TestParameters::input(param).basis_type = "pw"; it->second.str_values = {"file", "txt"}; it->second.read_value(it->second, param); it->second.reset_value(it->second, param); @@ -490,21 +504,21 @@ TEST_F(InputTest, Item_test) EXPECT_NO_THROW(it->second.check_value(it->second, param)); EXPECT_EQ(find_label("read_wfc_lcao", readinput.input_lists), readinput.input_lists.end()); - param.input.calculation = "get_wf"; + TestParameters::input(param).calculation = "get_wf"; } { // init_chg auto it = find_label("init_chg", readinput.input_lists); - param.input.init_chg = "get_pchg"; - param.input.init_chg = ""; + TestParameters::input(param).init_chg = "get_pchg"; + TestParameters::input(param).init_chg = ""; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.init_chg, "atomic"); + EXPECT_EQ(TestParameters::input(param).init_chg, "atomic"); - param.input.init_chg = ""; - param.input.calculation = "nscf"; + TestParameters::input(param).init_chg = ""; + TestParameters::input(param).calculation = "nscf"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.init_chg, "file"); + EXPECT_EQ(TestParameters::input(param).init_chg, "file"); - param.input.init_chg = "none"; + TestParameters::input(param).init_chg = "none"; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -512,69 +526,69 @@ TEST_F(InputTest, Item_test) } { // chg_extrap auto it = find_label("chg_extrap", readinput.input_lists); - param.input.chg_extrap = "default"; - param.input.calculation = "md"; + TestParameters::input(param).chg_extrap = "default"; + TestParameters::input(param).calculation = "md"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.chg_extrap, "second-order"); + EXPECT_EQ(TestParameters::input(param).chg_extrap, "second-order"); - param.input.chg_extrap = "default"; - param.input.calculation = "relax"; + TestParameters::input(param).chg_extrap = "default"; + TestParameters::input(param).calculation = "relax"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.chg_extrap, "first-order"); + EXPECT_EQ(TestParameters::input(param).chg_extrap, "first-order"); - param.input.chg_extrap = "default"; - param.input.calculation = "scf"; - param.input.socket_driver = true; + TestParameters::input(param).chg_extrap = "default"; + TestParameters::input(param).calculation = "scf"; + TestParameters::input(param).socket_driver = true; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.chg_extrap, "first-order"); + EXPECT_EQ(TestParameters::input(param).chg_extrap, "first-order"); - param.input.chg_extrap = "default"; - param.input.calculation = "none"; - param.input.socket_driver = false; + TestParameters::input(param).chg_extrap = "default"; + TestParameters::input(param).calculation = "none"; + TestParameters::input(param).socket_driver = false; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.chg_extrap, "atomic"); + EXPECT_EQ(TestParameters::input(param).chg_extrap, "atomic"); - param.input.chg_extrap = "none"; - param.input.calculation = "get_wf"; + TestParameters::input(param).chg_extrap = "none"; + TestParameters::input(param).calculation = "get_wf"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.chg_extrap, "atomic"); + EXPECT_EQ(TestParameters::input(param).chg_extrap, "atomic"); } { // out_chg auto it = find_label("out_chg", readinput.input_lists); - param.input.calculation = "get_wf"; - param.input.out_chg = {0}; + TestParameters::input(param).calculation = "get_wf"; + TestParameters::input(param).out_chg = {0}; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.out_chg[0], 1); + EXPECT_EQ(TestParameters::input(param).out_chg[0], 1); } { // out_pot auto it = find_label("out_pot", readinput.input_lists); - param.input.calculation = "scf"; + TestParameters::input(param).calculation = "scf"; it->second.str_values = {"1"}; it->second.read_value(it->second, param); - EXPECT_EQ(param.input.out_pot[0], 1); - EXPECT_EQ(param.input.out_pot[1], 8); + EXPECT_EQ(TestParameters::input(param).out_pot[0], 1); + EXPECT_EQ(TestParameters::input(param).out_pot[1], 8); it->second.str_values = {"1", "2"}; it->second.read_value(it->second, param); - EXPECT_EQ(param.input.out_pot[0], 1); - EXPECT_EQ(param.input.out_pot[1], 2); + EXPECT_EQ(TestParameters::input(param).out_pot[0], 1); + EXPECT_EQ(TestParameters::input(param).out_pot[1], 2); } { // out_dos auto it = find_label("out_dos", readinput.input_lists); - param.input.calculation = "get_wf"; - param.input.out_dos = 1; + TestParameters::input(param).calculation = "get_wf"; + TestParameters::input(param).out_dos = 1; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.out_dos, 0); + EXPECT_EQ(TestParameters::input(param).out_dos, 0); - param.input.out_dos = 3; - param.input.symmetry = "1"; + TestParameters::input(param).out_dos = 3; + TestParameters::input(param).symmetry = "1"; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); EXPECT_THAT(output, testing::HasSubstr("NOTICE")); - param.input.basis_type = "pw"; - param.input.out_dos = 3; + TestParameters::input(param).basis_type = "pw"; + TestParameters::input(param).out_dos = 3; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -584,13 +598,13 @@ TEST_F(InputTest, Item_test) auto it = find_label("out_ldos", readinput.input_lists); it->second.str_values = {"1"}; it->second.read_value(it->second, param); - EXPECT_EQ(param.input.out_ldos[0], 1); - EXPECT_EQ(param.input.out_ldos[1], 3); + EXPECT_EQ(TestParameters::input(param).out_ldos[0], 1); + EXPECT_EQ(TestParameters::input(param).out_ldos[1], 3); it->second.str_values = {"2", "2"}; it->second.read_value(it->second, param); - EXPECT_EQ(param.input.out_ldos[0], 2); - EXPECT_EQ(param.input.out_ldos[1], 2); + EXPECT_EQ(TestParameters::input(param).out_ldos[0], 2); + EXPECT_EQ(TestParameters::input(param).out_ldos[1], 2); it->second.str_values = {"1", "2", "3"}; testing::internal::CaptureStdout(); @@ -598,7 +612,7 @@ TEST_F(InputTest, Item_test) output = testing::internal::GetCapturedStdout(); EXPECT_THAT(output, testing::HasSubstr("NOTICE")); - param.input.out_ldos = {5, 5}; + TestParameters::input(param).out_ldos = {5, 5}; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -608,15 +622,15 @@ TEST_F(InputTest, Item_test) auto it = find_label("stm_bias", readinput.input_lists); it->second.str_values = {"2"}; it->second.read_value(it->second, param); - EXPECT_EQ(param.input.stm_bias[0], 2); - EXPECT_EQ(param.input.stm_bias[1], 0.1); - EXPECT_EQ(param.input.stm_bias[2], 1); + EXPECT_EQ(TestParameters::input(param).stm_bias[0], 2); + EXPECT_EQ(TestParameters::input(param).stm_bias[1], 0.1); + EXPECT_EQ(TestParameters::input(param).stm_bias[2], 1); it->second.str_values = {"3", "0.2", "5"}; it->second.read_value(it->second, param); - EXPECT_EQ(param.input.stm_bias[0], 3); - EXPECT_EQ(param.input.stm_bias[1], 0.2); - EXPECT_EQ(param.input.stm_bias[2], 5); + EXPECT_EQ(TestParameters::input(param).stm_bias[0], 3); + EXPECT_EQ(TestParameters::input(param).stm_bias[1], 0.2); + EXPECT_EQ(TestParameters::input(param).stm_bias[2], 5); it->second.str_values = {"1", "2"}; testing::internal::CaptureStdout(); @@ -624,13 +638,13 @@ TEST_F(InputTest, Item_test) output = testing::internal::GetCapturedStdout(); EXPECT_THAT(output, testing::HasSubstr("NOTICE")); - param.input.stm_bias = {1.0, 0.1, 0.0}; + TestParameters::input(param).stm_bias = {1.0, 0.1, 0.0}; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); EXPECT_THAT(output, testing::HasSubstr("NOTICE")); - param.input.stm_bias = {1.0, 0.0, 2.0}; + TestParameters::input(param).stm_bias = {1.0, 0.0, 2.0}; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -642,17 +656,17 @@ TEST_F(InputTest, Item_test) it->second.read_value(it->second, param); for (int i = 0; i < 6; ++i) { - EXPECT_EQ(param.input.ldos_line[i], i + 1); + EXPECT_EQ(TestParameters::input(param).ldos_line[i], i + 1); } - EXPECT_EQ(param.input.ldos_line[6], 100); + EXPECT_EQ(TestParameters::input(param).ldos_line[6], 100); it->second.str_values = {"2", "3", "4", "5", "6", "7", "200"}; it->second.read_value(it->second, param); for (int i = 0; i < 6; ++i) { - EXPECT_EQ(param.input.ldos_line[i], i + 2); + EXPECT_EQ(TestParameters::input(param).ldos_line[i], i + 2); } - EXPECT_EQ(param.input.ldos_line[6], 200); + EXPECT_EQ(TestParameters::input(param).ldos_line[6], 200); it->second.str_values = {"1", "2", "3", "4", "5", "6", "7", "8"}; testing::internal::CaptureStdout(); @@ -660,7 +674,7 @@ TEST_F(InputTest, Item_test) output = testing::internal::GetCapturedStdout(); EXPECT_THAT(output, testing::HasSubstr("NOTICE")); - param.input.ldos_line = {1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 0}; + TestParameters::input(param).ldos_line = {1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 0}; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -670,13 +684,13 @@ TEST_F(InputTest, Item_test) auto it = find_label("out_band", readinput.input_lists); it->second.str_values = {"1"}; it->second.read_value(it->second, param); - EXPECT_EQ(param.input.out_band[0], 1); - EXPECT_EQ(param.input.out_band[1], 8); + EXPECT_EQ(TestParameters::input(param).out_band[0], 1); + EXPECT_EQ(TestParameters::input(param).out_band[1], 8); it->second.str_values = {"1", "2"}; it->second.read_value(it->second, param); - EXPECT_EQ(param.input.out_band[0], 1); - EXPECT_EQ(param.input.out_band[1], 2); + EXPECT_EQ(TestParameters::input(param).out_band[0], 1); + EXPECT_EQ(TestParameters::input(param).out_band[1], 2); it->second.str_values = {"1", "2", "3"}; testing::internal::CaptureStdout(); @@ -684,20 +698,20 @@ TEST_F(InputTest, Item_test) output = testing::internal::GetCapturedStdout(); EXPECT_THAT(output, testing::HasSubstr("NOTICE")); - param.input.calculation = "get_wf"; - param.input.out_band = {1, 2}; + TestParameters::input(param).calculation = "get_wf"; + TestParameters::input(param).out_band = {1, 2}; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.out_band[0], 0); + EXPECT_EQ(TestParameters::input(param).out_band[0], 0); } { // out_proj_band auto it = find_label("out_proj_band", readinput.input_lists); - param.input.calculation = "get_wf"; - param.input.out_proj_band = true; + TestParameters::input(param).calculation = "get_wf"; + TestParameters::input(param).out_proj_band = true; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.out_proj_band, false); + EXPECT_EQ(TestParameters::input(param).out_proj_band, false); - param.input.basis_type = "pw"; - param.input.out_proj_band = true; + TestParameters::input(param).basis_type = "pw"; + TestParameters::input(param).out_proj_band = true; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -705,19 +719,19 @@ TEST_F(InputTest, Item_test) } { // read_file_dir auto it = find_label("read_file_dir", readinput.input_lists); - param.input.read_file_dir = "auto"; - param.input.suffix = "test"; + TestParameters::input(param).read_file_dir = "auto"; + TestParameters::input(param).suffix = "test"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.read_file_dir, "OUT.test/"); + EXPECT_EQ(TestParameters::input(param).read_file_dir, "OUT.test/"); - param.input.read_file_dir = "test"; + TestParameters::input(param).read_file_dir = "test"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.read_file_dir, "test/"); + EXPECT_EQ(TestParameters::input(param).read_file_dir, "test/"); } { // nx auto it = find_label("nx", readinput.input_lists); - param.input.nx = 1; - param.input.ny = 0; + TestParameters::input(param).nx = 1; + TestParameters::input(param).ny = 0; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -725,8 +739,8 @@ TEST_F(InputTest, Item_test) } { // ny auto it = find_label("ny", readinput.input_lists); - param.input.ny = 1; - param.input.nz = 0; + TestParameters::input(param).ny = 1; + TestParameters::input(param).nz = 0; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -734,8 +748,8 @@ TEST_F(InputTest, Item_test) } { // nz auto it = find_label("nz", readinput.input_lists); - param.input.nz = 1; - param.input.nx = 0; + TestParameters::input(param).nz = 1; + TestParameters::input(param).nx = 0; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -743,22 +757,22 @@ TEST_F(InputTest, Item_test) } { // ndx auto it = find_label("ndx", readinput.input_lists); - param.sys.double_grid = false; - param.input.ndx = 2; - param.input.nx = 1; + TestParameters::sys(param).double_grid = false; + TestParameters::input(param).ndx = 2; + TestParameters::input(param).nx = 1; it->second.reset_value(it->second, param); - EXPECT_EQ(param.sys.double_grid, true); + EXPECT_EQ(TestParameters::sys(param).double_grid, true); - param.input.ndx = 1; - param.input.ndy = 0; + TestParameters::input(param).ndx = 1; + TestParameters::input(param).ndy = 0; it->second.str_values = {"1"}; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); EXPECT_THAT(output, testing::HasSubstr("NOTICE")); - param.input.ndx = 1; - param.input.nx = 2; + TestParameters::input(param).ndx = 1; + TestParameters::input(param).nx = 2; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -766,22 +780,22 @@ TEST_F(InputTest, Item_test) } { // ndy auto it = find_label("ndy", readinput.input_lists); - param.sys.double_grid = false; - param.input.ndy = 2; - param.input.ny = 1; + TestParameters::sys(param).double_grid = false; + TestParameters::input(param).ndy = 2; + TestParameters::input(param).ny = 1; it->second.reset_value(it->second, param); - EXPECT_EQ(param.sys.double_grid, true); + EXPECT_EQ(TestParameters::sys(param).double_grid, true); - param.input.ndy = 1; - param.input.ndz = 0; + TestParameters::input(param).ndy = 1; + TestParameters::input(param).ndz = 0; it->second.str_values = {"1"}; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); EXPECT_THAT(output, testing::HasSubstr("NOTICE")); - param.input.ndy = 1; - param.input.ny = 2; + TestParameters::input(param).ndy = 1; + TestParameters::input(param).ny = 2; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -789,22 +803,22 @@ TEST_F(InputTest, Item_test) } { // ndz auto it = find_label("ndz", readinput.input_lists); - param.sys.double_grid = false; - param.input.ndz = 2; - param.input.nz = 1; + TestParameters::sys(param).double_grid = false; + TestParameters::input(param).ndz = 2; + TestParameters::input(param).nz = 1; it->second.reset_value(it->second, param); - EXPECT_EQ(param.sys.double_grid, true); + EXPECT_EQ(TestParameters::sys(param).double_grid, true); - param.input.ndz = 1; - param.input.nz = 2; + TestParameters::input(param).ndz = 1; + TestParameters::input(param).nz = 2; it->second.str_values = {"1"}; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); EXPECT_THAT(output, testing::HasSubstr("NOTICE")); - param.input.ndz = 1; - param.input.nz = 2; + TestParameters::input(param).ndz = 1; + TestParameters::input(param).nz = 2; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -812,77 +826,77 @@ TEST_F(InputTest, Item_test) } { // cell_factor auto it = find_label("cell_factor", readinput.input_lists); - param.input.calculation = "cell-relax"; - param.input.cell_factor = 1.0; + TestParameters::input(param).calculation = "cell-relax"; + TestParameters::input(param).cell_factor = 1.0; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.cell_factor, 2.0); + EXPECT_EQ(TestParameters::input(param).cell_factor, 2.0); } { // ks_sovler auto it = find_label("ks_solver", readinput.input_lists); - param.input.ks_solver = "default"; - param.input.basis_type = "pw"; + TestParameters::input(param).ks_solver = "default"; + TestParameters::input(param).basis_type = "pw"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.ks_solver, "cg"); + EXPECT_EQ(TestParameters::input(param).ks_solver, "cg"); - param.input.ks_solver = "default"; - param.input.basis_type = "lcao"; - param.input.device = "gpu"; + TestParameters::input(param).ks_solver = "default"; + TestParameters::input(param).basis_type = "lcao"; + TestParameters::input(param).device = "gpu"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.ks_solver, "cusolver"); + EXPECT_EQ(TestParameters::input(param).ks_solver, "cusolver"); - param.input.ks_solver = "genelpa"; - param.input.basis_type = "lcao"; - param.input.device = "gpu"; + TestParameters::input(param).ks_solver = "genelpa"; + TestParameters::input(param).basis_type = "lcao"; + TestParameters::input(param).device = "gpu"; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); EXPECT_THAT(output, testing::HasSubstr("Please use ks_solver = elpa with device = gpu")); #ifdef __ELPA - param.input.towannier90 = true; - param.input.basis_type = "lcao_in_pw"; + TestParameters::input(param).towannier90 = true; + TestParameters::input(param).basis_type = "lcao_in_pw"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.ks_solver, "genelpa"); + EXPECT_EQ(TestParameters::input(param).ks_solver, "genelpa"); #else #ifdef __MPI - param.input.towannier90 = true; - param.input.basis_type = "lcao_in_pw"; + TestParameters::input(param).towannier90 = true; + TestParameters::input(param).basis_type = "lcao_in_pw"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.ks_solver, "scalapack_gvx"); + EXPECT_EQ(TestParameters::input(param).ks_solver, "scalapack_gvx"); #else - param.input.towannier90 = true; - param.input.basis_type = "lcao_in_pw"; + TestParameters::input(param).towannier90 = true; + TestParameters::input(param).basis_type = "lcao_in_pw"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.ks_solver, "lapack"); + EXPECT_EQ(TestParameters::input(param).ks_solver, "lapack"); #endif #endif - param.input.ks_solver = "default"; - param.input.basis_type = "lcao"; - param.input.device = "cpu"; - param.input.ks_solver = "lapack"; + TestParameters::input(param).ks_solver = "default"; + TestParameters::input(param).basis_type = "lcao"; + TestParameters::input(param).device = "cpu"; + TestParameters::input(param).ks_solver = "lapack"; - param.input.ks_solver = "genelpa"; - param.input.basis_type = "pw"; + TestParameters::input(param).ks_solver = "genelpa"; + TestParameters::input(param).basis_type = "pw"; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); EXPECT_THAT(output, testing::HasSubstr("NOTICE")); - param.input.ks_solver = "cg"; - param.input.basis_type = "lcao"; + TestParameters::input(param).ks_solver = "cg"; + TestParameters::input(param).basis_type = "lcao"; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); EXPECT_THAT(output, testing::HasSubstr("NOTICE")); - param.input.ks_solver = "scalapack_gvx"; - param.input.basis_type = "lcao"; + TestParameters::input(param).ks_solver = "scalapack_gvx"; + TestParameters::input(param).basis_type = "lcao"; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); EXPECT_THAT(output, testing::HasSubstr("NOTICE")); - param.input.ks_solver = "cg"; - param.input.basis_type = "lcao_in_pw"; + TestParameters::input(param).ks_solver = "cg"; + TestParameters::input(param).basis_type = "lcao_in_pw"; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -890,50 +904,52 @@ TEST_F(InputTest, Item_test) } { // relax_nmax auto it = find_label("relax_nmax", readinput.input_lists); - param.input.calculation = "scf"; - param.input.relax_nmax = 5; + TestParameters::input(param).calculation = "scf"; + TestParameters::input(param).relax_nmax = 5; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.relax_nmax, 1); + EXPECT_EQ(TestParameters::input(param).relax_nmax, 1); - param.input.relax_nmax = 0; + TestParameters::input(param).relax_nmax = 0; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.relax_nmax, 0); + EXPECT_EQ(TestParameters::input(param).relax_nmax, 0); - param.input.calculation = "relax"; - param.input.relax_nmax = 0; + TestParameters::input(param).calculation = "relax"; + TestParameters::input(param).relax_nmax = 0; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.relax_nmax, 0); + EXPECT_EQ(TestParameters::input(param).relax_nmax, 0); - param.input.relax_nmax = -1; + TestParameters::input(param).relax_nmax = -1; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.relax_nmax, 50); + EXPECT_EQ(TestParameters::input(param).relax_nmax, 50); } { // out_stru auto it = find_label("out_stru", readinput.input_lists); - param.input.calculation = "get_wf"; - param.input.out_stru = 1; + TestParameters::input(param).calculation = "get_wf"; + TestParameters::input(param).out_stru = 1; + // The item is not user-read in this test; the reset rule applies. + it->second.str_values.clear(); it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.out_stru, 0); + EXPECT_EQ(TestParameters::input(param).out_stru, 0); } { // cal_stress auto it = find_label("cal_stress", readinput.input_lists); - param.input.calculation = "md"; - param.input.cal_stress = false; - param.input.esolver_type = "lj"; + TestParameters::input(param).calculation = "md"; + TestParameters::input(param).cal_stress = false; + TestParameters::input(param).esolver_type = "lj"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.cal_stress, true); + EXPECT_EQ(TestParameters::input(param).cal_stress, true); } { // fixed_axes auto it = find_label("fixed_axes", readinput.input_lists); - param.input.fixed_axes = "shape"; - param.input.relax_method = {"cg", "1"}; + TestParameters::input(param).fixed_axes = "shape"; + TestParameters::input(param).relax_method = {"cg", "1"}; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); EXPECT_THAT(output, testing::HasSubstr("NOTICE")); - param.input.fixed_axes = "volume"; - param.input.relax_method = {"cg", "1"}; + TestParameters::input(param).fixed_axes = "volume"; + TestParameters::input(param).relax_method = {"cg", "1"}; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -941,15 +957,15 @@ TEST_F(InputTest, Item_test) } { // fixed_ibrav auto it = find_label("fixed_ibrav", readinput.input_lists); - param.input.fixed_ibrav = true; - param.input.relax_method = {"cg", "1"}; + TestParameters::input(param).fixed_ibrav = true; + TestParameters::input(param).relax_method = {"cg", "1"}; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); EXPECT_THAT(output, testing::HasSubstr("NOTICE")); - param.input.fixed_ibrav = true; - param.input.latname = "none"; + TestParameters::input(param).fixed_ibrav = true; + TestParameters::input(param).latname = "none"; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -957,8 +973,8 @@ TEST_F(InputTest, Item_test) } { // fixed_atoms auto it = find_label("fixed_atoms", readinput.input_lists); - param.input.fixed_atoms = true; - param.input.calculation = "relax"; + TestParameters::input(param).fixed_atoms = true; + TestParameters::input(param).calculation = "relax"; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -966,58 +982,58 @@ TEST_F(InputTest, Item_test) } { // force_thr auto it = find_label("force_thr", readinput.input_lists); - param.input.force_thr = -1; - param.input.force_thr_ev = -1; + TestParameters::input(param).force_thr = -1; + TestParameters::input(param).force_thr_ev = -1; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.force_thr, 1.0e-3); - EXPECT_EQ(param.input.force_thr_ev, 1.0e-3 * 13.6058 / 0.529177); + EXPECT_EQ(TestParameters::input(param).force_thr, 1.0e-3); + EXPECT_EQ(TestParameters::input(param).force_thr_ev, 1.0e-3 * 13.6058 / 0.529177); - param.input.force_thr = -1; - param.input.force_thr_ev = 1.0e-3 * 13.6058 / 0.529177; + TestParameters::input(param).force_thr = -1; + TestParameters::input(param).force_thr_ev = 1.0e-3 * 13.6058 / 0.529177; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.force_thr, 1.0e-3); + EXPECT_EQ(TestParameters::input(param).force_thr, 1.0e-3); - param.input.force_thr = 1.0e-3; - param.input.force_thr_ev = 1.0e-3 * 13.6058 / 0.529177; + TestParameters::input(param).force_thr = 1.0e-3; + TestParameters::input(param).force_thr_ev = 1.0e-3 * 13.6058 / 0.529177; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.force_thr, 1.0e-3); - EXPECT_EQ(param.input.force_thr_ev, 1.0e-3 * 13.6058 / 0.529177); + EXPECT_EQ(TestParameters::input(param).force_thr, 1.0e-3); + EXPECT_EQ(TestParameters::input(param).force_thr_ev, 1.0e-3 * 13.6058 / 0.529177); } { // out_level auto it = find_label("out_level", readinput.input_lists); - param.input.out_level = "0"; - param.input.calculation = "md"; - param.sys.out_md_control = false; + TestParameters::input(param).out_level = "0"; + TestParameters::input(param).calculation = "md"; + TestParameters::sys(param).out_md_control = false; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.out_level, "m"); + EXPECT_EQ(TestParameters::input(param).out_level, "m"); } { // out_dmk auto it = find_label("out_dmk", readinput.input_lists); it->second.str_values = {"1"}; it->second.read_value(it->second, param); - EXPECT_EQ(param.input.out_dmk[0], 1); - EXPECT_EQ(param.input.out_dmk[1], 8); + EXPECT_EQ(TestParameters::input(param).out_dmk[0], 1); + EXPECT_EQ(TestParameters::input(param).out_dmk[1], 8); it->second.str_values = {"1", "2"}; it->second.read_value(it->second, param); - EXPECT_EQ(param.input.out_dmk[0], 1); - EXPECT_EQ(param.input.out_dmk[1], 2); + EXPECT_EQ(TestParameters::input(param).out_dmk[0], 1); + EXPECT_EQ(TestParameters::input(param).out_dmk[1], 2); } { // out_dmr auto it = find_label("out_dmr", readinput.input_lists); it->second.str_values = {"1"}; it->second.read_value(it->second, param); - EXPECT_EQ(param.input.out_dmr[0], 1); - EXPECT_EQ(param.input.out_dmr[1], 8); + EXPECT_EQ(TestParameters::input(param).out_dmr[0], 1); + EXPECT_EQ(TestParameters::input(param).out_dmr[1], 8); it->second.str_values = {"1", "2"}; it->second.read_value(it->second, param); - EXPECT_EQ(param.input.out_dmr[0], 1); - EXPECT_EQ(param.input.out_dmr[1], 2); + EXPECT_EQ(TestParameters::input(param).out_dmr[0], 1); + EXPECT_EQ(TestParameters::input(param).out_dmr[1], 2); } { // method_sto auto it = find_label("method_sto", readinput.input_lists); - param.input.method_sto = 0; + TestParameters::input(param).method_sto = 0; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -1025,19 +1041,19 @@ TEST_F(InputTest, Item_test) } { // nbands_sto auto it = find_label("nbands_sto", readinput.input_lists); - param.input.esolver_type = "sdft"; + TestParameters::input(param).esolver_type = "sdft"; it->second.str_values = {"all"}; it->second.read_value(it->second, param); EXPECT_NO_THROW(it->second.check_value(it->second, param)); - EXPECT_EQ(param.input.nbands_sto, 0); - EXPECT_EQ(param.input.esolver_type, "sdft"); + EXPECT_EQ(TestParameters::input(param).nbands_sto, 0); + EXPECT_EQ(TestParameters::input(param).esolver_type, "sdft"); it->second.str_values = {"8"}; it->second.read_value(it->second, param); EXPECT_NO_THROW(it->second.check_value(it->second, param)); - EXPECT_EQ(param.input.nbands_sto, 8); - EXPECT_EQ(param.input.esolver_type, "sdft"); + EXPECT_EQ(TestParameters::input(param).nbands_sto, 8); + EXPECT_EQ(TestParameters::input(param).esolver_type, "sdft"); it->second.str_values = {"1000000"}; it->second.read_value(it->second, param); @@ -1056,11 +1072,11 @@ TEST_F(InputTest, Item_test) EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); EXPECT_THAT(output, testing::HasSubstr("NOTICE")); - EXPECT_EQ(param.input.nbands_sto, 0); - EXPECT_EQ(param.input.esolver_type, "sdft"); + EXPECT_EQ(TestParameters::input(param).nbands_sto, 0); + EXPECT_EQ(TestParameters::input(param).esolver_type, "sdft"); it->second.str_values = {"-1"}; - param.input.nbands_sto = -1; + TestParameters::input(param).nbands_sto = -1; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -1072,18 +1088,18 @@ TEST_F(InputTest, Item_test) it->second.final_value.str(""); it->second.str_values = {}; - param.input.nbands_sto = 256; + TestParameters::input(param).nbands_sto = 256; it->second.get_final_value(it->second, param); EXPECT_EQ(it->second.final_value.str(), "256"); } { // basis_type auto it = find_label("basis_type", readinput.input_lists); - param.input.basis_type = "lcao_in_pw"; - param.input.towannier90 = true; + TestParameters::input(param).basis_type = "lcao_in_pw"; + TestParameters::input(param).towannier90 = true; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.basis_type, "lcao"); + EXPECT_EQ(TestParameters::input(param).basis_type, "lcao"); - param.input.basis_type = "gauss"; + TestParameters::input(param).basis_type = "gauss"; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -1091,20 +1107,20 @@ TEST_F(InputTest, Item_test) } { // gamma_only auto it = find_label("gamma_only", readinput.input_lists); - param.input.basis_type = "pw"; - param.input.gamma_only = true; - std::string filename = param.input.kpoint_file; + TestParameters::input(param).basis_type = "pw"; + TestParameters::input(param).gamma_only = true; + std::string filename = TestParameters::input(param).kpoint_file; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.gamma_only, false); + EXPECT_EQ(TestParameters::input(param).gamma_only, false); std::ifstream ifs(filename); std::string line; std::getline(ifs, line); ifs.close(); EXPECT_EQ(line, "K_POINTS"); - param.input.basis_type = "lcao"; - param.input.gamma_only = true; - param.input.nspin = 4; + TestParameters::input(param).basis_type = "lcao"; + TestParameters::input(param).gamma_only = true; + TestParameters::input(param).nspin = 4; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.reset_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -1113,57 +1129,57 @@ TEST_F(InputTest, Item_test) } { // out_mat_r auto it = find_label("out_mat_r", readinput.input_lists); - param.input.out_hsr[0] = 1; - param.sys.gamma_only_local = true; + TestParameters::input(param).out_hsr[0] = 1; + TestParameters::sys(param).gamma_only_local = true; it->second.check_value(it->second, param); - param.input.out_hsr[0] = 3; + TestParameters::input(param).out_hsr[0] = 3; it->second.check_value(it->second, param); - param.input.out_hsr[0] = 0; + TestParameters::input(param).out_hsr[0] = 0; - param.input.esolver_type = "lcao"; - param.input.out_mat_r[0] = 1; + TestParameters::input(param).esolver_type = "lcao"; + TestParameters::input(param).out_mat_r[0] = 1; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); EXPECT_THAT(output, testing::HasSubstr("available")); - param.input.out_mat_r[0] = 0; - param.sys.gamma_only_local = false; + TestParameters::input(param).out_mat_r[0] = 0; + TestParameters::sys(param).gamma_only_local = false; } { // lcao_ecut auto it = find_label("lcao_ecut", readinput.input_lists); - param.input.lcao_ecut = 0; - param.input.ecutwfc = 1; - param.input.basis_type = "lcao"; + TestParameters::input(param).lcao_ecut = 0; + TestParameters::input(param).ecutwfc = 1; + TestParameters::input(param).basis_type = "lcao"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.lcao_ecut, 1); + EXPECT_EQ(TestParameters::input(param).lcao_ecut, 1); } { // out_mat_hs auto it = find_label("out_mat_hs", readinput.input_lists); it->second.str_values = {"1"}; it->second.read_value(it->second, param); - EXPECT_EQ(param.input.out_mat_hs[0], 1); - EXPECT_EQ(param.input.out_mat_hs[1], 8); + EXPECT_EQ(TestParameters::input(param).out_mat_hs[0], 1); + EXPECT_EQ(TestParameters::input(param).out_mat_hs[1], 8); it->second.str_values = {"1", "2"}; it->second.read_value(it->second, param); - EXPECT_EQ(param.input.out_mat_hs[0], 1); - EXPECT_EQ(param.input.out_mat_hs[1], 2); + EXPECT_EQ(TestParameters::input(param).out_mat_hs[0], 1); + EXPECT_EQ(TestParameters::input(param).out_mat_hs[1], 2); } { // out_hsk auto it = find_label("out_hsk", readinput.input_lists); it->second.str_values = {"1", "12"}; it->second.read_value(it->second, param); - EXPECT_EQ(param.input.out_hsk[0], 1); - EXPECT_EQ(param.input.out_hsk[1], 12); + EXPECT_EQ(TestParameters::input(param).out_hsk[0], 1); + EXPECT_EQ(TestParameters::input(param).out_hsk[1], 12); it->second.str_values = {"2", "12"}; it->second.read_value(it->second, param); - EXPECT_EQ(param.input.out_hsk[0], 2); - EXPECT_EQ(param.input.out_hsk[1], 12); + EXPECT_EQ(TestParameters::input(param).out_hsk[0], 2); + EXPECT_EQ(TestParameters::input(param).out_hsk[1], 12); it->second.check_value(it->second, param); - param.input.out_hsk[0] = 3; + TestParameters::input(param).out_hsk[0] = 3; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -1173,29 +1189,29 @@ TEST_F(InputTest, Item_test) auto it = find_label("out_hsr", readinput.input_lists); it->second.str_values = {"1", "10"}; it->second.read_value(it->second, param); - EXPECT_EQ(param.input.out_hsr[0], 1); - EXPECT_EQ(param.input.out_hsr[1], 10); + EXPECT_EQ(TestParameters::input(param).out_hsr[0], 1); + EXPECT_EQ(TestParameters::input(param).out_hsr[1], 10); it->second.str_values = {"2", "12"}; it->second.read_value(it->second, param); - EXPECT_EQ(param.input.out_hsr[0], 2); - EXPECT_EQ(param.input.out_hsr[1], 12); + EXPECT_EQ(TestParameters::input(param).out_hsr[0], 2); + EXPECT_EQ(TestParameters::input(param).out_hsr[1], 12); it->second.check_value(it->second, param); - param.input.out_hsr[0] = 4; + TestParameters::input(param).out_hsr[0] = 4; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); EXPECT_THAT(output, testing::HasSubstr("format must be 0, 1, 2, or 3")); #ifndef __CNPY - param.input.out_hsr[0] = 3; + TestParameters::input(param).out_hsr[0] = 3; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); EXPECT_THAT(output, testing::HasSubstr("ENABLE_CNPY")); #else - param.input.out_hsr[0] = 3; + TestParameters::input(param).out_hsr[0] = 3; it->second.check_value(it->second, param); #endif } @@ -1203,19 +1219,19 @@ TEST_F(InputTest, Item_test) auto it = find_label("out_hr_npz", readinput.input_lists); it->second.str_values = {"1"}; it->second.read_value(it->second, param); - EXPECT_EQ(param.input.out_hr_npz, true); + EXPECT_EQ(TestParameters::input(param).out_hr_npz, true); } { // out_hsr_npz auto it = find_label("out_hsr_npz", readinput.input_lists); it->second.str_values = {"1"}; it->second.read_value(it->second, param); - EXPECT_EQ(param.input.out_hsr_npz, true); + EXPECT_EQ(TestParameters::input(param).out_hsr_npz, true); } { // out_dm_npz auto it = find_label("out_dm_npz", readinput.input_lists); it->second.str_values = {"1"}; it->second.read_value(it->second, param); - EXPECT_EQ(param.input.out_dm_npz, true); + EXPECT_EQ(TestParameters::input(param).out_dm_npz, true); } } @@ -1232,8 +1248,8 @@ TEST_F(InputTest, HsOutputAliases) primary->second.read_value(primary->second, param); readinput.normalize_hs_output_options(param); - EXPECT_EQ(param.input.out_hsk[0], 0); - EXPECT_EQ(param.input.out_hsk[1], 8); + EXPECT_EQ(TestParameters::input(param).out_hsk[0], 0); + EXPECT_EQ(TestParameters::input(param).out_hsk[1], 8); } { ModuleIO::ReadInput readinput(0); @@ -1246,8 +1262,8 @@ TEST_F(InputTest, HsOutputAliases) legacy->second.read_value(legacy->second, param); readinput.normalize_hs_output_options(param); - EXPECT_EQ(param.input.out_hsk[0], 0); - EXPECT_EQ(param.input.out_hsk[1], 8); + EXPECT_EQ(TestParameters::input(param).out_hsk[0], 0); + EXPECT_EQ(TestParameters::input(param).out_hsk[1], 8); } { ModuleIO::ReadInput readinput(0); @@ -1260,10 +1276,10 @@ TEST_F(InputTest, HsOutputAliases) legacy_npz->second.read_value(legacy_npz->second, param); readinput.normalize_hs_output_options(param); - EXPECT_EQ(param.input.out_hsr[0], 1); - EXPECT_EQ(param.input.out_hsr[1], 5); - EXPECT_TRUE(param.input.out_hsr_npz); - EXPECT_TRUE(param.input.out_hsr_npz_compat); + EXPECT_EQ(TestParameters::input(param).out_hsr[0], 1); + EXPECT_EQ(TestParameters::input(param).out_hsr[1], 5); + EXPECT_TRUE(TestParameters::input(param).out_hsr_npz); + EXPECT_TRUE(TestParameters::input(param).out_hsr_npz_compat); } { ModuleIO::ReadInput readinput(0); @@ -1279,10 +1295,10 @@ TEST_F(InputTest, HsOutputAliases) primary->second.read_value(primary->second, param); readinput.normalize_hs_output_options(param); - EXPECT_EQ(param.input.out_hsr[0], 1); - EXPECT_EQ(param.input.out_hsr[1], 12); - EXPECT_FALSE(param.input.out_hsr_npz); - EXPECT_FALSE(param.input.out_hsr_npz_compat); + EXPECT_EQ(TestParameters::input(param).out_hsr[0], 1); + EXPECT_EQ(TestParameters::input(param).out_hsr[1], 12); + EXPECT_FALSE(TestParameters::input(param).out_hsr_npz); + EXPECT_FALSE(TestParameters::input(param).out_hsr_npz_compat); } { ModuleIO::ReadInput readinput(0); @@ -1291,8 +1307,8 @@ TEST_F(InputTest, HsOutputAliases) legacy_npz->second.str_values = {"1"}; legacy_npz->second.read_value(legacy_npz->second, param); readinput.normalize_hs_output_options(param); - EXPECT_EQ(param.input.out_hsr[0], 3); - EXPECT_EQ(param.input.out_hsr[1], 8); + EXPECT_EQ(TestParameters::input(param).out_hsr[0], 3); + EXPECT_EQ(TestParameters::input(param).out_hsr[1], 8); } { ModuleIO::ReadInput readinput(0); @@ -1300,9 +1316,9 @@ TEST_F(InputTest, HsOutputAliases) auto primary = find_label("out_hsk", readinput.input_lists); primary->second.str_values = {"0"}; primary->second.read_value(primary->second, param); - param.input.qo_switch = true; + TestParameters::input(param).qo_switch = true; readinput.normalize_hs_output_options(param); - EXPECT_EQ(param.input.out_hsk[0], 1); + EXPECT_EQ(TestParameters::input(param).out_hsk[0], 1); } } TEST_F(InputTest, Item_test2) @@ -1314,8 +1330,8 @@ TEST_F(InputTest, Item_test2) std::string output = ""; { // out_mat_dh auto it = find_label("out_mat_dh", readinput.input_lists); - param.input.out_mat_dh[0] = 1; - param.input.nspin = 4; + TestParameters::input(param).out_mat_dh[0] = 1; + TestParameters::input(param).nspin = 4; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -1323,15 +1339,15 @@ TEST_F(InputTest, Item_test2) } { // dm_to_rho auto it = find_label("dm_to_rho", readinput.input_lists); - param.input.dm_to_rho = true; + TestParameters::input(param).dm_to_rho = true; GlobalV::NPROC = 2; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); EXPECT_THAT(output, testing::HasSubstr("NOTICE")); - param.input.dm_to_rho = true; - param.input.gamma_only = true; + TestParameters::input(param).dm_to_rho = true; + TestParameters::input(param).gamma_only = true; GlobalV::NPROC = 1; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); @@ -1339,7 +1355,7 @@ TEST_F(InputTest, Item_test2) EXPECT_THAT(output, testing::HasSubstr("NOTICE")); #ifndef __CNPY - param.input.dm_to_rho = true; + TestParameters::input(param).dm_to_rho = true; GlobalV::NPROC = 1; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); @@ -1349,19 +1365,19 @@ TEST_F(InputTest, Item_test2) } { // out_wfc_lcao auto it = find_label("out_wfc_lcao", readinput.input_lists); - param.input.out_wfc_lcao = false; - param.input.qo_switch = true; + TestParameters::input(param).out_wfc_lcao = false; + TestParameters::input(param).qo_switch = true; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.out_wfc_lcao, true); + EXPECT_EQ(TestParameters::input(param).out_wfc_lcao, true); - param.input.out_wfc_lcao = -1; + TestParameters::input(param).out_wfc_lcao = -1; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); EXPECT_THAT(output, testing::HasSubstr("NOTICE")); - param.input.out_wfc_lcao = 1; - param.input.basis_type = "pw"; + TestParameters::input(param).out_wfc_lcao = 1; + TestParameters::input(param).basis_type = "pw"; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -1369,22 +1385,22 @@ TEST_F(InputTest, Item_test2) } { // bx auto it = find_label("bx", readinput.input_lists); - param.input.bx = 11; + TestParameters::input(param).bx = 11; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); EXPECT_THAT(output, testing::HasSubstr("NOTICE")); - param.input.bx = 2; - param.input.basis_type = "pw"; + TestParameters::input(param).bx = 2; + TestParameters::input(param).basis_type = "pw"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.bx, 1); - EXPECT_EQ(param.input.by, 1); - EXPECT_EQ(param.input.bz, 1); + EXPECT_EQ(TestParameters::input(param).bx, 1); + EXPECT_EQ(TestParameters::input(param).by, 1); + EXPECT_EQ(TestParameters::input(param).bz, 1); } { // by auto it = find_label("by", readinput.input_lists); - param.input.by = 11; + TestParameters::input(param).by = 11; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -1392,7 +1408,7 @@ TEST_F(InputTest, Item_test2) } { // bz auto it = find_label("bz", readinput.input_lists); - param.input.bz = 11; + TestParameters::input(param).bz = 11; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -1400,43 +1416,43 @@ TEST_F(InputTest, Item_test2) } { // mixing_beta auto it = find_label("mixing_beta", readinput.input_lists); - param.input.mixing_beta = -1; - param.input.nspin = 1; + TestParameters::input(param).mixing_beta = -1; + TestParameters::input(param).nspin = 1; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.mixing_beta, 0.8); + EXPECT_EQ(TestParameters::input(param).mixing_beta, 0.8); - param.input.mixing_beta = -1; - param.input.nspin = 2; + TestParameters::input(param).mixing_beta = -1; + TestParameters::input(param).nspin = 2; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.mixing_beta, 0.4); - EXPECT_EQ(param.input.mixing_beta_mag, 1.6); - EXPECT_EQ(param.input.mixing_gg0_mag, 0.0); + EXPECT_EQ(TestParameters::input(param).mixing_beta, 0.4); + EXPECT_EQ(TestParameters::input(param).mixing_beta_mag, 1.6); + EXPECT_EQ(TestParameters::input(param).mixing_gg0_mag, 0.0); - param.input.mixing_beta = -1; - param.input.nspin = 4; + TestParameters::input(param).mixing_beta = -1; + TestParameters::input(param).nspin = 4; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.mixing_beta, 0.4); - EXPECT_EQ(param.input.mixing_beta_mag, 1.6); - EXPECT_EQ(param.input.mixing_gg0_mag, 0.0); + EXPECT_EQ(TestParameters::input(param).mixing_beta, 0.4); + EXPECT_EQ(TestParameters::input(param).mixing_beta_mag, 1.6); + EXPECT_EQ(TestParameters::input(param).mixing_gg0_mag, 0.0); } { // mixing_beta_mag auto it = find_label("mixing_beta_mag", readinput.input_lists); - param.input.mixing_beta = 0.3; - param.input.mixing_beta_mag = -1; - param.input.nspin = 2; + TestParameters::input(param).mixing_beta = 0.3; + TestParameters::input(param).mixing_beta_mag = -1; + TestParameters::input(param).nspin = 2; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.mixing_beta_mag, 1.2); + EXPECT_EQ(TestParameters::input(param).mixing_beta_mag, 1.2); - param.input.mixing_beta = 0.5; - param.input.mixing_beta_mag = -1; - param.input.nspin = 2; + TestParameters::input(param).mixing_beta = 0.5; + TestParameters::input(param).mixing_beta_mag = -1; + TestParameters::input(param).nspin = 2; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.mixing_beta_mag, 1.6); + EXPECT_EQ(TestParameters::input(param).mixing_beta_mag, 1.6); } { // dip_cor_flag auto it = find_label("dip_cor_flag", readinput.input_lists); - param.input.dip_cor_flag = true; - param.input.efield_flag = false; + TestParameters::input(param).dip_cor_flag = true; + TestParameters::input(param).efield_flag = false; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -1444,9 +1460,9 @@ TEST_F(InputTest, Item_test2) } { // efield_dir auto it = find_label("efield_dir", readinput.input_lists); - param.input.gate_flag = true; - param.input.efield_flag = true; - param.input.dip_cor_flag = false; + TestParameters::input(param).gate_flag = true; + TestParameters::input(param).efield_flag = true; + TestParameters::input(param).dip_cor_flag = false; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -1454,63 +1470,63 @@ TEST_F(InputTest, Item_test2) } { // vdw_s6 auto it = find_label("vdw_s6", readinput.input_lists); - param.input.vdw_s6 = "default"; - param.input.vdw_method = "d2"; + TestParameters::input(param).vdw_s6 = "default"; + TestParameters::input(param).vdw_method = "d2"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.vdw_s6, "0.75"); + EXPECT_EQ(TestParameters::input(param).vdw_s6, "0.75"); // dftd3 parameter will not get its value here - param.input.vdw_s6 = "default"; - param.input.vdw_method = "d3_0"; + TestParameters::input(param).vdw_s6 = "default"; + TestParameters::input(param).vdw_method = "d3_0"; it->second.reset_value(it->second, param); - // EXPECT_EQ(param.input.vdw_s6, "1.0"); - EXPECT_EQ(param.input.vdw_s6, "default"); + // EXPECT_EQ(TestParameters::input(param).vdw_s6, "1.0"); + EXPECT_EQ(TestParameters::input(param).vdw_s6, "default"); } { // vdw_s8 auto it = find_label("vdw_s8", readinput.input_lists); - param.input.vdw_s8 = "default"; - param.input.vdw_method = "d3_0"; + TestParameters::input(param).vdw_s8 = "default"; + TestParameters::input(param).vdw_method = "d3_0"; it->second.reset_value(it->second, param); - // EXPECT_EQ(param.input.vdw_s8, "0.722"); - EXPECT_EQ(param.input.vdw_s8, "default"); + // EXPECT_EQ(TestParameters::input(param).vdw_s8, "0.722"); + EXPECT_EQ(TestParameters::input(param).vdw_s8, "default"); - param.input.vdw_s8 = "default"; - param.input.vdw_method = "d3_bj"; + TestParameters::input(param).vdw_s8 = "default"; + TestParameters::input(param).vdw_method = "d3_bj"; it->second.reset_value(it->second, param); - // EXPECT_EQ(param.input.vdw_s8, "0.7875"); - EXPECT_EQ(param.input.vdw_s8, "default"); + // EXPECT_EQ(TestParameters::input(param).vdw_s8, "0.7875"); + EXPECT_EQ(TestParameters::input(param).vdw_s8, "default"); } { // vdw_a1 auto it = find_label("vdw_a1", readinput.input_lists); - param.input.vdw_a1 = "default"; - param.input.vdw_method = "d3_0"; + TestParameters::input(param).vdw_a1 = "default"; + TestParameters::input(param).vdw_method = "d3_0"; it->second.reset_value(it->second, param); - // EXPECT_EQ(param.input.vdw_a1, "1.217"); - EXPECT_EQ(param.input.vdw_a1, "default"); + // EXPECT_EQ(TestParameters::input(param).vdw_a1, "1.217"); + EXPECT_EQ(TestParameters::input(param).vdw_a1, "default"); - param.input.vdw_a1 = "default"; - param.input.vdw_method = "d3_bj"; + TestParameters::input(param).vdw_a1 = "default"; + TestParameters::input(param).vdw_method = "d3_bj"; it->second.reset_value(it->second, param); - // EXPECT_EQ(param.input.vdw_a1, "0.4289"); - EXPECT_EQ(param.input.vdw_a1, "default"); + // EXPECT_EQ(TestParameters::input(param).vdw_a1, "0.4289"); + EXPECT_EQ(TestParameters::input(param).vdw_a1, "default"); } { // vdw_a2 auto it = find_label("vdw_a2", readinput.input_lists); - param.input.vdw_a2 = "default"; - param.input.vdw_method = "d3_0"; + TestParameters::input(param).vdw_a2 = "default"; + TestParameters::input(param).vdw_method = "d3_0"; it->second.reset_value(it->second, param); - // EXPECT_EQ(param.input.vdw_a2, "1.0"); - EXPECT_EQ(param.input.vdw_a2, "default"); + // EXPECT_EQ(TestParameters::input(param).vdw_a2, "1.0"); + EXPECT_EQ(TestParameters::input(param).vdw_a2, "default"); - param.input.vdw_a2 = "default"; - param.input.vdw_method = "d3_bj"; + TestParameters::input(param).vdw_a2 = "default"; + TestParameters::input(param).vdw_method = "d3_bj"; it->second.reset_value(it->second, param); - // EXPECT_EQ(param.input.vdw_a2, "4.4407"); - EXPECT_EQ(param.input.vdw_a2, "default"); + // EXPECT_EQ(TestParameters::input(param).vdw_a2, "4.4407"); + EXPECT_EQ(TestParameters::input(param).vdw_a2, "default"); } { // vdw_c6_unit auto it = find_label("vdw_c6_unit", readinput.input_lists); - param.input.vdw_C6_unit = "test"; + TestParameters::input(param).vdw_C6_unit = "test"; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -1518,7 +1534,7 @@ TEST_F(InputTest, Item_test2) } { // vdw_r0_unit auto it = find_label("vdw_r0_unit", readinput.input_lists); - param.input.vdw_R0_unit = "test"; + TestParameters::input(param).vdw_R0_unit = "test"; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -1526,7 +1542,7 @@ TEST_F(InputTest, Item_test2) } { // vdw_cutoff_type auto it = find_label("vdw_cutoff_type", readinput.input_lists); - param.input.vdw_cutoff_type = "test"; + TestParameters::input(param).vdw_cutoff_type = "test"; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -1534,28 +1550,28 @@ TEST_F(InputTest, Item_test2) } { // vdw_cutoff_radius auto it = find_label("vdw_cutoff_radius", readinput.input_lists); - param.input.vdw_cutoff_radius = "default"; - param.input.vdw_method = "d2"; + TestParameters::input(param).vdw_cutoff_radius = "default"; + TestParameters::input(param).vdw_method = "d2"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.vdw_cutoff_radius, "56.6918"); + EXPECT_EQ(TestParameters::input(param).vdw_cutoff_radius, "56.6918"); - param.input.vdw_cutoff_radius = "default"; - param.input.vdw_method = "d3_0"; + TestParameters::input(param).vdw_cutoff_radius = "default"; + TestParameters::input(param).vdw_method = "d3_0"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.vdw_cutoff_radius, "60"); + EXPECT_EQ(TestParameters::input(param).vdw_cutoff_radius, "60"); - param.input.vdw_cutoff_radius = "default"; - param.input.vdw_method = "d3_bj"; + TestParameters::input(param).vdw_cutoff_radius = "default"; + TestParameters::input(param).vdw_method = "d3_bj"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.vdw_cutoff_radius, "60"); + EXPECT_EQ(TestParameters::input(param).vdw_cutoff_radius, "60"); - param.input.vdw_cutoff_radius = "default"; - param.input.vdw_method = "none"; + TestParameters::input(param).vdw_cutoff_radius = "default"; + TestParameters::input(param).vdw_method = "none"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.vdw_cutoff_radius, "0"); + EXPECT_EQ(TestParameters::input(param).vdw_cutoff_radius, "0"); - param.input.vdw_cutoff_radius = "-1"; - param.input.vdw_method = "d2"; + TestParameters::input(param).vdw_cutoff_radius = "-1"; + TestParameters::input(param).vdw_method = "d2"; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -1563,7 +1579,7 @@ TEST_F(InputTest, Item_test2) } { // vdw_radius_unit auto it = find_label("vdw_radius_unit", readinput.input_lists); - param.input.vdw_radius_unit = "test"; + TestParameters::input(param).vdw_radius_unit = "test"; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -1571,7 +1587,7 @@ TEST_F(InputTest, Item_test2) } { // vdw_cutoff_width2 auto it = find_label("vdw_cutoff_width2", readinput.input_lists); - param.input.vdw_cutoff_width2 = -1.0; + TestParameters::input(param).vdw_cutoff_width2 = -1.0; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -1579,7 +1595,7 @@ TEST_F(InputTest, Item_test2) } { // vdw_cutoff_width3 auto it = find_label("vdw_cutoff_width3", readinput.input_lists); - param.input.vdw_cutoff_width3 = -1.0; + TestParameters::input(param).vdw_cutoff_width3 = -1.0; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -1587,7 +1603,7 @@ TEST_F(InputTest, Item_test2) } { // vdw_cn_thr auto it = find_label("vdw_cn_thr", readinput.input_lists); - param.input.vdw_cn_thr = -1; + TestParameters::input(param).vdw_cn_thr = -1; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -1595,7 +1611,7 @@ TEST_F(InputTest, Item_test2) } { // vdw_cn_thr_unit auto it = find_label("vdw_cn_thr_unit", readinput.input_lists); - param.input.vdw_cn_thr_unit = "test"; + TestParameters::input(param).vdw_cn_thr_unit = "test"; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -1605,9 +1621,9 @@ TEST_F(InputTest, Item_test2) auto it = find_label("vdw_cutoff_period", readinput.input_lists); it->second.str_values = {"1", "1", "1"}; it->second.read_value(it->second, param); - EXPECT_EQ(param.input.vdw_cutoff_period[0], 1); - EXPECT_EQ(param.input.vdw_cutoff_period[1], 1); - EXPECT_EQ(param.input.vdw_cutoff_period[2], 1); + EXPECT_EQ(TestParameters::input(param).vdw_cutoff_period[0], 1); + EXPECT_EQ(TestParameters::input(param).vdw_cutoff_period[1], 1); + EXPECT_EQ(TestParameters::input(param).vdw_cutoff_period[2], 1); it->second.str_values = {"1", "1"}; testing::internal::CaptureStdout(); @@ -1615,7 +1631,7 @@ TEST_F(InputTest, Item_test2) output = testing::internal::GetCapturedStdout(); EXPECT_THAT(output, testing::HasSubstr("NOTICE")); - param.input.vdw_cutoff_period = {-1, 1, 1}; + TestParameters::input(param).vdw_cutoff_period = {-1, 1, 1}; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -1623,88 +1639,88 @@ TEST_F(InputTest, Item_test2) } { // exx_fock_alpha auto it = find_label("exx_fock_alpha", readinput.input_lists); - param.input.exx_fock_alpha[0] = "default"; - param.input.dft_functional = "HF"; + TestParameters::input(param).exx_fock_alpha[0] = "default"; + TestParameters::input(param).dft_functional = "HF"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.exx_fock_alpha[0], "1"); + EXPECT_EQ(TestParameters::input(param).exx_fock_alpha[0], "1"); - param.input.exx_fock_alpha[0] = "default"; - param.input.dft_functional = "PBE0"; + TestParameters::input(param).exx_fock_alpha[0] = "default"; + TestParameters::input(param).dft_functional = "PBE0"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.exx_fock_alpha[0], "0.25"); + EXPECT_EQ(TestParameters::input(param).exx_fock_alpha[0], "0.25"); - param.input.exx_fock_alpha[0] = "default"; - param.input.dft_functional = "SCAN0"; + TestParameters::input(param).exx_fock_alpha[0] = "default"; + TestParameters::input(param).dft_functional = "SCAN0"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.exx_fock_alpha[0], "0.25"); + EXPECT_EQ(TestParameters::input(param).exx_fock_alpha[0], "0.25"); - param.input.exx_fock_alpha[0] = "default"; - param.input.dft_functional = "cam_pbeh"; + TestParameters::input(param).exx_fock_alpha[0] = "default"; + TestParameters::input(param).dft_functional = "cam_pbeh"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.exx_fock_alpha[0], "0.2"); + EXPECT_EQ(TestParameters::input(param).exx_fock_alpha[0], "0.2"); } { // exx_erfc_alpha auto it = find_label("exx_erfc_alpha", readinput.input_lists); - param.input.exx_erfc_alpha[0] = "default"; - param.input.dft_functional = "HSE"; + TestParameters::input(param).exx_erfc_alpha[0] = "default"; + TestParameters::input(param).dft_functional = "HSE"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.exx_erfc_alpha[0], "0.25"); + EXPECT_EQ(TestParameters::input(param).exx_erfc_alpha[0], "0.25"); - param.input.exx_erfc_alpha[0] = "default"; - param.input.dft_functional = "lc_wpbe"; + TestParameters::input(param).exx_erfc_alpha[0] = "default"; + TestParameters::input(param).dft_functional = "lc_wpbe"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.exx_erfc_alpha[0], "-1"); + EXPECT_EQ(TestParameters::input(param).exx_erfc_alpha[0], "-1"); - param.input.exx_erfc_alpha[0] = "default"; - param.input.dft_functional = "lrc_wpbe"; + TestParameters::input(param).exx_erfc_alpha[0] = "default"; + TestParameters::input(param).dft_functional = "lrc_wpbe"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.exx_erfc_alpha[0], "-1"); + EXPECT_EQ(TestParameters::input(param).exx_erfc_alpha[0], "-1"); - param.input.exx_erfc_alpha[0] = "default"; - param.input.dft_functional = "lrc_wpbeh"; + TestParameters::input(param).exx_erfc_alpha[0] = "default"; + TestParameters::input(param).dft_functional = "lrc_wpbeh"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.exx_erfc_alpha[0], "-0.8"); + EXPECT_EQ(TestParameters::input(param).exx_erfc_alpha[0], "-0.8"); - param.input.exx_erfc_alpha[0] = "default"; - param.input.dft_functional = "cam_pbeh"; + TestParameters::input(param).exx_erfc_alpha[0] = "default"; + TestParameters::input(param).dft_functional = "cam_pbeh"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.exx_erfc_alpha[0], "0.8"); + EXPECT_EQ(TestParameters::input(param).exx_erfc_alpha[0], "0.8"); } { // exx_erfc_omega auto it = find_label("exx_erfc_omega", readinput.input_lists); - param.input.exx_erfc_omega[0] = "default"; - param.input.dft_functional = "lc_pbe"; + TestParameters::input(param).exx_erfc_omega[0] = "default"; + TestParameters::input(param).dft_functional = "lc_pbe"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.exx_erfc_omega[0], "0.33"); + EXPECT_EQ(TestParameters::input(param).exx_erfc_omega[0], "0.33"); - param.input.exx_erfc_omega[0] = "default"; - param.input.dft_functional = "lc_wpbe"; + TestParameters::input(param).exx_erfc_omega[0] = "default"; + TestParameters::input(param).dft_functional = "lc_wpbe"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.exx_erfc_omega[0], "0.4"); + EXPECT_EQ(TestParameters::input(param).exx_erfc_omega[0], "0.4"); - param.input.exx_erfc_omega[0] = "default"; - param.input.dft_functional = "lrc_wpbe"; + TestParameters::input(param).exx_erfc_omega[0] = "default"; + TestParameters::input(param).dft_functional = "lrc_wpbe"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.exx_erfc_omega[0], "0.3"); + EXPECT_EQ(TestParameters::input(param).exx_erfc_omega[0], "0.3"); - param.input.exx_erfc_omega[0] = "default"; - param.input.dft_functional = "lrc_wpbeh"; + TestParameters::input(param).exx_erfc_omega[0] = "default"; + TestParameters::input(param).dft_functional = "lrc_wpbeh"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.exx_erfc_omega[0], "0.2"); + EXPECT_EQ(TestParameters::input(param).exx_erfc_omega[0], "0.2"); - param.input.exx_erfc_omega[0] = "default"; - param.input.dft_functional = "cam_pbeh"; + TestParameters::input(param).exx_erfc_omega[0] = "default"; + TestParameters::input(param).dft_functional = "cam_pbeh"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.exx_erfc_omega[0], "0.7"); + EXPECT_EQ(TestParameters::input(param).exx_erfc_omega[0], "0.7"); - param.input.exx_erfc_omega[0] = "default"; - param.input.dft_functional = "hse"; + TestParameters::input(param).exx_erfc_omega[0] = "default"; + TestParameters::input(param).dft_functional = "hse"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.exx_erfc_omega[0], "0.11"); + EXPECT_EQ(TestParameters::input(param).exx_erfc_omega[0], "0.11"); } { // exx_hybrid_step auto it = find_label("exx_hybrid_step", readinput.input_lists); - param.input.exx_hybrid_step = -1; + TestParameters::input(param).exx_hybrid_step = -1; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -1712,61 +1728,61 @@ TEST_F(InputTest, Item_test2) } { // exx_real_number auto it = find_label("exx_real_number", readinput.input_lists); - param.input.exx_real_number = "default"; - param.input.gamma_only = true; + TestParameters::input(param).exx_real_number = "default"; + TestParameters::input(param).gamma_only = true; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.exx_real_number, "1"); + EXPECT_EQ(TestParameters::input(param).exx_real_number, "1"); - param.input.exx_real_number = "default"; - param.input.gamma_only = false; + TestParameters::input(param).exx_real_number = "default"; + TestParameters::input(param).gamma_only = false; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.exx_real_number, "0"); + EXPECT_EQ(TestParameters::input(param).exx_real_number, "0"); } { // exx_singluarity_correction auto it = find_label("exx_singularity_correction", readinput.input_lists); - param.input.exx_singularity_correction = "default"; - param.input.dft_functional = "HF"; + TestParameters::input(param).exx_singularity_correction = "default"; + TestParameters::input(param).dft_functional = "HF"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.exx_singularity_correction, "spencer"); + EXPECT_EQ(TestParameters::input(param).exx_singularity_correction, "spencer"); - param.input.exx_singularity_correction = "default"; - param.input.dft_functional = "PBE0"; + TestParameters::input(param).exx_singularity_correction = "default"; + TestParameters::input(param).dft_functional = "PBE0"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.exx_singularity_correction, "spencer"); + EXPECT_EQ(TestParameters::input(param).exx_singularity_correction, "spencer"); - param.input.exx_singularity_correction = "default"; - param.input.dft_functional = "HSE"; + TestParameters::input(param).exx_singularity_correction = "default"; + TestParameters::input(param).dft_functional = "HSE"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.exx_singularity_correction, "limits"); + EXPECT_EQ(TestParameters::input(param).exx_singularity_correction, "limits"); } { // exx_ccp_rmesh_times auto it = find_label("exx_ccp_rmesh_times", readinput.input_lists); - param.input.exx_ccp_rmesh_times = "default"; - param.input.dft_functional = "HF"; + TestParameters::input(param).exx_ccp_rmesh_times = "default"; + TestParameters::input(param).dft_functional = "HF"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.exx_ccp_rmesh_times, "5"); + EXPECT_EQ(TestParameters::input(param).exx_ccp_rmesh_times, "5"); - param.input.exx_ccp_rmesh_times = "default"; - param.input.dft_functional = "PBE0"; + TestParameters::input(param).exx_ccp_rmesh_times = "default"; + TestParameters::input(param).dft_functional = "PBE0"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.exx_ccp_rmesh_times, "5"); + EXPECT_EQ(TestParameters::input(param).exx_ccp_rmesh_times, "5"); - param.input.exx_ccp_rmesh_times = "default"; - param.input.dft_functional = "SCAN0"; + TestParameters::input(param).exx_ccp_rmesh_times = "default"; + TestParameters::input(param).dft_functional = "SCAN0"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.exx_ccp_rmesh_times, "5"); + EXPECT_EQ(TestParameters::input(param).exx_ccp_rmesh_times, "5"); - param.input.exx_ccp_rmesh_times = "default"; - param.input.dft_functional = "HSE"; + TestParameters::input(param).exx_ccp_rmesh_times = "default"; + TestParameters::input(param).dft_functional = "HSE"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.exx_ccp_rmesh_times, "1.5"); + EXPECT_EQ(TestParameters::input(param).exx_ccp_rmesh_times, "1.5"); - param.input.exx_ccp_rmesh_times = "default"; - param.input.dft_functional = "none"; + TestParameters::input(param).exx_ccp_rmesh_times = "default"; + TestParameters::input(param).dft_functional = "none"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.exx_ccp_rmesh_times, "1"); + EXPECT_EQ(TestParameters::input(param).exx_ccp_rmesh_times, "1"); - param.input.exx_ccp_rmesh_times = "-1"; + TestParameters::input(param).exx_ccp_rmesh_times = "-1"; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -1774,7 +1790,7 @@ TEST_F(InputTest, Item_test2) } { // exx_opt_orb_ecut auto it = find_label("exx_opt_orb_ecut", readinput.input_lists); - param.input.exx_opt_orb_ecut = -1; + TestParameters::input(param).exx_opt_orb_ecut = -1; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -1782,7 +1798,7 @@ TEST_F(InputTest, Item_test2) } { // exx_opt_orb_tolerence auto it = find_label("exx_opt_orb_tolerence", readinput.input_lists); - param.input.exx_opt_orb_tolerence = -1; + TestParameters::input(param).exx_opt_orb_tolerence = -1; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -1790,14 +1806,14 @@ TEST_F(InputTest, Item_test2) } { // exx_symmetry_realspace auto it = find_label("exx_symmetry_realspace", readinput.input_lists); - param.input.exx_symmetry_realspace = true; - param.input.symmetry="0"; + TestParameters::input(param).exx_symmetry_realspace = true; + TestParameters::input(param).symmetry="0"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.exx_symmetry_realspace, false); + EXPECT_EQ(TestParameters::input(param).exx_symmetry_realspace, false); } { // rpa_ccp_rmesh_times auto it = find_label("rpa_ccp_rmesh_times", readinput.input_lists); - param.input.rpa_ccp_rmesh_times = 0; + TestParameters::input(param).rpa_ccp_rmesh_times = 0; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -1805,42 +1821,42 @@ TEST_F(InputTest, Item_test2) } { // berry_phase auto it = find_label("berry_phase", readinput.input_lists); - param.input.berry_phase = true; - param.input.basis_type = "pw"; - param.input.calculation = "nscf"; - param.input.gdir = 1; + TestParameters::input(param).berry_phase = true; + TestParameters::input(param).basis_type = "pw"; + TestParameters::input(param).calculation = "nscf"; + TestParameters::input(param).gdir = 1; - param.input.gdir = 0; + TestParameters::input(param).gdir = 0; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); EXPECT_THAT(output, testing::HasSubstr("NOTICE")); - param.input.calculation = "scf"; + TestParameters::input(param).calculation = "scf"; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); EXPECT_THAT(output, testing::HasSubstr("NOTICE")); - param.input.basis_type = "lcao_in_pw"; - param.input.calculation = "nscf"; + TestParameters::input(param).basis_type = "lcao_in_pw"; + TestParameters::input(param).calculation = "nscf"; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); } { // towannier90 auto it = find_label("towannier90", readinput.input_lists); - param.input.towannier90 = true; - param.input.calculation = "nscf"; - param.input.nspin = 2; - param.input.wannier_spin = "none"; + TestParameters::input(param).towannier90 = true; + TestParameters::input(param).calculation = "nscf"; + TestParameters::input(param).nspin = 2; + TestParameters::input(param).wannier_spin = "none"; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); EXPECT_THAT(output, testing::HasSubstr("towannier90")); - param.input.towannier90 = true; - param.input.calculation = "scf"; + TestParameters::input(param).towannier90 = true; + TestParameters::input(param).calculation = "scf"; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -1848,57 +1864,57 @@ TEST_F(InputTest, Item_test2) } { // wannier_method auto it = find_label("wannier_method", readinput.input_lists); - param.input.towannier90 = true; - param.input.basis_type = "lcao_in_pw"; - param.input.wannier_method = 0; + TestParameters::input(param).towannier90 = true; + TestParameters::input(param).basis_type = "lcao_in_pw"; + TestParameters::input(param).wannier_method = 0; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.wannier_method, 1); + EXPECT_EQ(TestParameters::input(param).wannier_method, 1); } { // of_hold_rho0 auto it = find_label("of_hold_rho0", readinput.input_lists); - param.input.of_wt_rho0 = 1; - param.input.of_hold_rho0 = false; + TestParameters::input(param).of_wt_rho0 = 1; + TestParameters::input(param).of_hold_rho0 = false; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.of_hold_rho0, true); + EXPECT_EQ(TestParameters::input(param).of_hold_rho0, true); } { // of_full_pw_dim auto it = find_label("of_full_pw_dim", readinput.input_lists); - param.input.of_full_pw = false; - param.input.of_full_pw_dim = 1; + TestParameters::input(param).of_full_pw = false; + TestParameters::input(param).of_full_pw_dim = 1; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.of_full_pw_dim, 0); + EXPECT_EQ(TestParameters::input(param).of_full_pw_dim, 0); } { // of_read_kernel auto it = find_label("of_read_kernel", readinput.input_lists); - param.input.of_read_kernel = true; - param.input.of_kinetic = "none"; + TestParameters::input(param).of_read_kernel = true; + TestParameters::input(param).of_kinetic = "none"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.of_read_kernel, false); + EXPECT_EQ(TestParameters::input(param).of_read_kernel, false); } { // of_ml_gene_data auto it = find_label("of_ml_gene_data", readinput.input_lists); - param.input.of_ml_gene_data = true; - param.input.esolver_type = "ofdft"; - param.input.basis_type = "pw"; + TestParameters::input(param).of_ml_gene_data = true; + TestParameters::input(param).esolver_type = "ofdft"; + TestParameters::input(param).basis_type = "pw"; GlobalV::NPROC = 1; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); EXPECT_THAT(output, testing::HasSubstr("NOTICE")); - param.input.of_ml_gene_data = true; - param.input.esolver_type = "ksdft"; - param.input.basis_type = "lcao"; + TestParameters::input(param).of_ml_gene_data = true; + TestParameters::input(param).esolver_type = "ksdft"; + TestParameters::input(param).basis_type = "lcao"; GlobalV::NPROC = 1; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); EXPECT_THAT(output, testing::HasSubstr("NOTICE")); - param.input.of_ml_gene_data = true; - param.input.esolver_type = "ksdft"; - param.input.basis_type = "pw"; + TestParameters::input(param).of_ml_gene_data = true; + TestParameters::input(param).esolver_type = "ksdft"; + TestParameters::input(param).basis_type = "pw"; GlobalV::NPROC = 2; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); @@ -1907,39 +1923,39 @@ TEST_F(InputTest, Item_test2) } { // dft_plus_u auto it = find_label("dft_plus_u", readinput.input_lists); - param.input.dft_plus_u = 1; - param.input.l_channel = {-1, -1}; + TestParameters::input(param).dft_plus_u = 1; + TestParameters::input(param).l_channel = {-1, -1}; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.dft_plus_u, 0); + EXPECT_EQ(TestParameters::input(param).dft_plus_u, 0); } { // uramping auto it = find_label("uramping", readinput.input_lists); - param.sys.uramping = 1; - param.input.l_channel = {-1, -1}; + TestParameters::sys(param).uramping = 1; + TestParameters::input(param).l_channel = {-1, -1}; it->second.reset_value(it->second, param); - EXPECT_EQ(param.sys.uramping, 0); + EXPECT_EQ(TestParameters::sys(param).uramping, 0); } { // onsite_radius auto it = find_label("onsite_radius", readinput.input_lists); - param.input.onsite_radius = 0.0; - param.input.dft_plus_u = 1; + TestParameters::input(param).onsite_radius = 0.0; + TestParameters::input(param).dft_plus_u = 1; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.onsite_radius, 3.0); + EXPECT_EQ(TestParameters::input(param).onsite_radius, 3.0); } { // hubbard_u auto it = find_label("hubbard_u", readinput.input_lists); - param.input.ntype = 2; + TestParameters::input(param).ntype = 2; it->second.str_values = {"1.0", "2.0"}; - param.sys.hubbard_u = {1.0, 2.0}; + TestParameters::sys(param).hubbard_u = {1.0, 2.0}; it->second.check_value(it->second, param); - param.input.ntype = 3; + TestParameters::input(param).ntype = 3; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); EXPECT_THAT(output, testing::HasSubstr("NOTICE")); - param.input.ntype = 2; - param.sys.hubbard_u = {1.0, -1.0}; + TestParameters::input(param).ntype = 2; + TestParameters::sys(param).hubbard_u = {1.0, -1.0}; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -1947,18 +1963,18 @@ TEST_F(InputTest, Item_test2) } { // orbital_corr auto it = find_label("orbital_corr", readinput.input_lists); - param.input.ntype = 2; + TestParameters::input(param).ntype = 2; it->second.str_values = {"1", "2"}; - param.input.l_channel = {1, 2}; + TestParameters::input(param).l_channel = {1, 2}; it->second.check_value(it->second, param); - param.input.ntype = 3; + TestParameters::input(param).ntype = 3; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); EXPECT_THAT(output, testing::HasSubstr("NOTICE")); - param.input.ntype = 2; - param.input.l_channel = {1, 4}; + TestParameters::input(param).ntype = 2; + TestParameters::input(param).l_channel = {1, 4}; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -1966,12 +1982,12 @@ TEST_F(InputTest, Item_test2) } { // bessel_nao_ecut auto it = find_label("bessel_nao_ecut", readinput.input_lists); - param.input.bessel_nao_ecut = "default"; - param.input.ecutwfc = 1.0; + TestParameters::input(param).bessel_nao_ecut = "default"; + TestParameters::input(param).ecutwfc = 1.0; it->second.reset_value(it->second, param); - EXPECT_EQ(std::stod(param.input.bessel_nao_ecut), 1.0); + EXPECT_EQ(std::stod(TestParameters::input(param).bessel_nao_ecut), 1.0); - param.input.bessel_nao_ecut = "-1"; + TestParameters::input(param).bessel_nao_ecut = "-1"; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -1979,7 +1995,7 @@ TEST_F(InputTest, Item_test2) } { // bessel_nao_rcut auto it = find_label("bessel_nao_rcut", readinput.input_lists); - param.input.bessel_nao_rcuts = {-1}; + TestParameters::input(param).bessel_nao_rcuts = {-1}; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -1987,12 +2003,12 @@ TEST_F(InputTest, Item_test2) } { // bessel_descriptor_ecut auto it = find_label("bessel_descriptor_ecut", readinput.input_lists); - param.input.bessel_descriptor_ecut = "default"; - param.input.ecutwfc = 1.0; + TestParameters::input(param).bessel_descriptor_ecut = "default"; + TestParameters::input(param).ecutwfc = 1.0; it->second.reset_value(it->second, param); - EXPECT_EQ(std::stod(param.input.bessel_descriptor_ecut), 1.0); + EXPECT_EQ(std::stod(TestParameters::input(param).bessel_descriptor_ecut), 1.0); - param.input.bessel_descriptor_ecut = "-1"; + TestParameters::input(param).bessel_descriptor_ecut = "-1"; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -2000,7 +2016,7 @@ TEST_F(InputTest, Item_test2) } { // bessel_descriptor_rcut auto it = find_label("bessel_descriptor_rcut", readinput.input_lists); - param.input.bessel_descriptor_rcut = -1; + TestParameters::input(param).bessel_descriptor_rcut = -1; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -2008,13 +2024,13 @@ TEST_F(InputTest, Item_test2) } { // sc_mag_switch auto it = find_label("sc_mag_switch", readinput.input_lists); - param.input.sc_mag_switch = true; + TestParameters::input(param).sc_mag_switch = true; // Since sc_mag_switch check is disabled, just call the function without expecting exit it->second.check_value(it->second, param); } { // sc_thr auto it = find_label("sc_thr", readinput.input_lists); - param.input.sc_thr = -1; + TestParameters::input(param).sc_thr = -1; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -2022,7 +2038,7 @@ TEST_F(InputTest, Item_test2) } { // nsc auto it = find_label("nsc", readinput.input_lists); - param.input.nsc = 0; + TestParameters::input(param).nsc = 0; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -2030,7 +2046,7 @@ TEST_F(InputTest, Item_test2) } { // nsc_min auto it = find_label("nsc_min", readinput.input_lists); - param.input.nsc_min = 0; + TestParameters::input(param).nsc_min = 0; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -2038,7 +2054,7 @@ TEST_F(InputTest, Item_test2) } { // alpha_trial auto it = find_label("alpha_trial", readinput.input_lists); - param.input.alpha_trial = -1; + TestParameters::input(param).alpha_trial = -1; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -2046,7 +2062,7 @@ TEST_F(InputTest, Item_test2) } { // sccut auto it = find_label("sccut", readinput.input_lists); - param.input.sccut = -1; + TestParameters::input(param).sccut = -1; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -2054,7 +2070,7 @@ TEST_F(InputTest, Item_test2) } { // sc_scf_thr auto it = find_label("sc_scf_thr", readinput.input_lists); - param.input.sc_scf_thr = -1e-3; + TestParameters::input(param).sc_scf_thr = -1e-3; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -2062,36 +2078,36 @@ TEST_F(InputTest, Item_test2) } { // qo_thr auto it = find_label("qo_thr", readinput.input_lists); - param.input.qo_thr = 1e-5; + TestParameters::input(param).qo_thr = 1e-5; it->second.check_value(it->second, param); } { // qo_strategy auto it = find_label("qo_strategy", readinput.input_lists); - param.input.ntype = 2; + TestParameters::input(param).ntype = 2; - param.input.qo_basis = "hydrogen"; - param.input.qo_strategy = {"all"}; + TestParameters::input(param).qo_basis = "hydrogen"; + TestParameters::input(param).qo_strategy = {"all"}; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.qo_strategy.size(), 2); - EXPECT_EQ(param.input.qo_strategy[0], "all"); - EXPECT_EQ(param.input.qo_strategy[1], "all"); + EXPECT_EQ(TestParameters::input(param).qo_strategy.size(), 2); + EXPECT_EQ(TestParameters::input(param).qo_strategy[0], "all"); + EXPECT_EQ(TestParameters::input(param).qo_strategy[1], "all"); - param.input.qo_strategy = {}; - param.input.qo_basis = "hydrogen"; + TestParameters::input(param).qo_strategy = {}; + TestParameters::input(param).qo_basis = "hydrogen"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.qo_strategy.size(), 2); - EXPECT_EQ(param.input.qo_strategy[0], "energy-valence"); - EXPECT_EQ(param.input.qo_strategy[1], "energy-valence"); + EXPECT_EQ(TestParameters::input(param).qo_strategy.size(), 2); + EXPECT_EQ(TestParameters::input(param).qo_strategy[0], "energy-valence"); + EXPECT_EQ(TestParameters::input(param).qo_strategy[1], "energy-valence"); - param.input.qo_strategy = {}; - param.input.qo_basis = "pswfc"; + TestParameters::input(param).qo_strategy = {}; + TestParameters::input(param).qo_basis = "pswfc"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.qo_strategy.size(), 2); - EXPECT_EQ(param.input.qo_strategy[0], "all"); - EXPECT_EQ(param.input.qo_strategy[1], "all"); + EXPECT_EQ(TestParameters::input(param).qo_strategy.size(), 2); + EXPECT_EQ(TestParameters::input(param).qo_strategy[0], "all"); + EXPECT_EQ(TestParameters::input(param).qo_strategy[1], "all"); - param.input.qo_basis = "test"; - param.input.qo_strategy = {}; + TestParameters::input(param).qo_basis = "test"; + TestParameters::input(param).qo_strategy = {}; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.reset_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -2099,37 +2115,37 @@ TEST_F(InputTest, Item_test2) } { // qo_screening_coeff auto it = find_label("qo_screening_coeff", readinput.input_lists); - param.input.ntype = 2; + TestParameters::input(param).ntype = 2; it->second.str_values = {"0.2"}; - param.input.qo_screening_coeff = {0.2}; - param.input.qo_basis = "pswfc"; + TestParameters::input(param).qo_screening_coeff = {0.2}; + TestParameters::input(param).qo_basis = "pswfc"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.qo_screening_coeff.size(), 2); - EXPECT_EQ(param.input.qo_screening_coeff[0], 0.2); - EXPECT_EQ(param.input.qo_screening_coeff[1], 0.2); + EXPECT_EQ(TestParameters::input(param).qo_screening_coeff.size(), 2); + EXPECT_EQ(TestParameters::input(param).qo_screening_coeff[0], 0.2); + EXPECT_EQ(TestParameters::input(param).qo_screening_coeff[1], 0.2); - param.input.qo_screening_coeff = {}; - param.input.qo_basis = "pswfc"; + TestParameters::input(param).qo_screening_coeff = {}; + TestParameters::input(param).qo_basis = "pswfc"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.qo_screening_coeff.size(), 2); - EXPECT_EQ(param.input.qo_screening_coeff[0], 0.1); - EXPECT_EQ(param.input.qo_screening_coeff[1], 0.1); + EXPECT_EQ(TestParameters::input(param).qo_screening_coeff.size(), 2); + EXPECT_EQ(TestParameters::input(param).qo_screening_coeff[0], 0.1); + EXPECT_EQ(TestParameters::input(param).qo_screening_coeff[1], 0.1); - param.input.qo_screening_coeff = {}; - param.input.qo_basis = "test"; + TestParameters::input(param).qo_screening_coeff = {}; + TestParameters::input(param).qo_basis = "test"; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.reset_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); EXPECT_THAT(output, testing::HasSubstr("NOTICE")); - param.input.qo_screening_coeff = {0.2, -0.1}; + TestParameters::input(param).qo_screening_coeff = {0.2, -0.1}; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); EXPECT_THAT(output, testing::HasSubstr("NOTICE")); - param.input.qo_screening_coeff = {0.2, 1e-8}; + TestParameters::input(param).qo_screening_coeff = {0.2, 1e-8}; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -2141,43 +2157,43 @@ TEST_F(InputTest, Item_test2) } { // md_nstep auto it = find_label("md_nstep", readinput.input_lists); - param.input.mdp.md_nstep = 0; + TestParameters::input(param).mdp.md_nstep = 0; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.mdp.md_nstep, 50); + EXPECT_EQ(TestParameters::input(param).mdp.md_nstep, 50); } { // md_prec_level auto it = find_label("md_prec_level", readinput.input_lists); - param.input.mdp.md_prec_level = 1; - param.input.calculation = "vc-relax"; + TestParameters::input(param).mdp.md_prec_level = 1; + TestParameters::input(param).calculation = "vc-relax"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.mdp.md_prec_level, 0); + EXPECT_EQ(TestParameters::input(param).mdp.md_prec_level, 0); - param.input.calculation = "md"; - param.input.mdp.md_prec_level = 1; - param.input.mdp.md_type = "vc-md"; + TestParameters::input(param).calculation = "md"; + TestParameters::input(param).mdp.md_prec_level = 1; + TestParameters::input(param).mdp.md_type = "vc-md"; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.mdp.md_prec_level, 0); + EXPECT_EQ(TestParameters::input(param).mdp.md_prec_level, 0); } { // md_tfreq auto it = find_label("md_tfreq", readinput.input_lists); - param.input.mdp.md_tfreq = 0; - param.input.calculation = "md"; - param.input.mdp.md_dt = 1.0; + TestParameters::input(param).mdp.md_tfreq = 0; + TestParameters::input(param).calculation = "md"; + TestParameters::input(param).mdp.md_dt = 1.0; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.mdp.md_tfreq, 1.0 / 40 / 1.0); + EXPECT_EQ(TestParameters::input(param).mdp.md_tfreq, 1.0 / 40 / 1.0); } { // md_pfreq auto it = find_label("md_pfreq", readinput.input_lists); - param.input.mdp.md_pfreq = 0; - param.input.calculation = "md"; - param.input.mdp.md_dt = 1.0; + TestParameters::input(param).mdp.md_pfreq = 0; + TestParameters::input(param).calculation = "md"; + TestParameters::input(param).mdp.md_dt = 1.0; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.mdp.md_pfreq, 1.0 / 400 / 1.0); + EXPECT_EQ(TestParameters::input(param).mdp.md_pfreq, 1.0 / 400 / 1.0); } { // lj_rule auto it = find_label("lj_rule", readinput.input_lists); - param.input.esolver_type = "lj"; - param.input.mdp.lj_rule = 3; + TestParameters::input(param).esolver_type = "lj"; + TestParameters::input(param).mdp.lj_rule = 3; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); @@ -2185,16 +2201,16 @@ TEST_F(InputTest, Item_test2) } { // lj_rcut auto it = find_label("lj_rcut", readinput.input_lists); - param.input.ntype = 2; - param.input.esolver_type = "lj"; + TestParameters::input(param).ntype = 2; + TestParameters::input(param).esolver_type = "lj"; it->second.str_values = {"1.0", "2.0"}; - param.input.mdp.lj_rcut = {1.0, 2.0}; + TestParameters::input(param).mdp.lj_rcut = {1.0, 2.0}; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); output = testing::internal::GetCapturedStdout(); EXPECT_THAT(output, testing::HasSubstr("NOTICE")); - param.input.mdp.lj_rcut = {1.0, 2.0, -1.0}; + TestParameters::input(param).mdp.lj_rcut = {1.0, 2.0, -1.0}; it->second.str_values = {"1.0", "2.0", "-1.0"}; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); @@ -2203,9 +2219,9 @@ TEST_F(InputTest, Item_test2) } { // lj_epsilon auto it = find_label("lj_epsilon", readinput.input_lists); - param.input.ntype = 2; - param.input.esolver_type = "lj"; - param.input.mdp.lj_epsilon = {1.0}; + TestParameters::input(param).ntype = 2; + TestParameters::input(param).esolver_type = "lj"; + TestParameters::input(param).mdp.lj_epsilon = {1.0}; it->second.str_values = {"1.0"}; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); @@ -2214,9 +2230,9 @@ TEST_F(InputTest, Item_test2) } { // lj_sigma auto it = find_label("lj_sigma", readinput.input_lists); - param.input.ntype = 2; - param.input.esolver_type = "lj"; - param.input.mdp.lj_sigma = {1.0}; + TestParameters::input(param).ntype = 2; + TestParameters::input(param).esolver_type = "lj"; + TestParameters::input(param).mdp.lj_sigma = {1.0}; it->second.str_values = {"1.0"}; testing::internal::CaptureStdout(); EXPECT_EXIT(it->second.check_value(it->second, param), ::testing::ExitedWithCode(1), ""); @@ -2225,14 +2241,14 @@ TEST_F(InputTest, Item_test2) } { // nocc auto it = find_label("nocc", readinput.input_lists); - param.input.nocc = 5; - param.input.nbands = 4; - param.input.nelec = 0.0; + TestParameters::input(param).nocc = 5; + TestParameters::input(param).nbands = 4; + TestParameters::input(param).nelec = 0.0; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.nocc, 4); - param.input.nocc = 0; + EXPECT_EQ(TestParameters::input(param).nocc, 4); + TestParameters::input(param).nocc = 0; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.nocc, 4); + EXPECT_EQ(TestParameters::input(param).nocc, 4); } } @@ -2246,61 +2262,61 @@ TEST_F(InputTest, Item_test_out_mat_vec) auto it = find_label("out_mat_dh", readinput.input_lists); it->second.str_values = {"1"}; it->second.read_value(it->second, param); - EXPECT_EQ(param.input.out_mat_dh[0], 1); - EXPECT_EQ(param.input.out_mat_dh[1], 8); + EXPECT_EQ(TestParameters::input(param).out_mat_dh[0], 1); + EXPECT_EQ(TestParameters::input(param).out_mat_dh[1], 8); it->second.str_values = {"1", "12"}; it->second.read_value(it->second, param); - EXPECT_EQ(param.input.out_mat_dh[0], 1); - EXPECT_EQ(param.input.out_mat_dh[1], 12); + EXPECT_EQ(TestParameters::input(param).out_mat_dh[0], 1); + EXPECT_EQ(TestParameters::input(param).out_mat_dh[1], 12); } { // out_mat_ds auto it = find_label("out_mat_ds", readinput.input_lists); it->second.str_values = {"1"}; it->second.read_value(it->second, param); - EXPECT_EQ(param.input.out_mat_ds[0], 1); - EXPECT_EQ(param.input.out_mat_ds[1], 8); + EXPECT_EQ(TestParameters::input(param).out_mat_ds[0], 1); + EXPECT_EQ(TestParameters::input(param).out_mat_ds[1], 8); it->second.str_values = {"1", "10"}; it->second.read_value(it->second, param); - EXPECT_EQ(param.input.out_mat_ds[0], 1); - EXPECT_EQ(param.input.out_mat_ds[1], 10); + EXPECT_EQ(TestParameters::input(param).out_mat_ds[0], 1); + EXPECT_EQ(TestParameters::input(param).out_mat_ds[1], 10); } { // out_mat_t auto it = find_label("out_mat_t", readinput.input_lists); it->second.str_values = {"1"}; it->second.read_value(it->second, param); - EXPECT_EQ(param.input.out_mat_t[0], 1); - EXPECT_EQ(param.input.out_mat_t[1], 8); + EXPECT_EQ(TestParameters::input(param).out_mat_t[0], 1); + EXPECT_EQ(TestParameters::input(param).out_mat_t[1], 8); it->second.str_values = {"1", "15"}; it->second.read_value(it->second, param); - EXPECT_EQ(param.input.out_mat_t[0], 1); - EXPECT_EQ(param.input.out_mat_t[1], 15); + EXPECT_EQ(TestParameters::input(param).out_mat_t[0], 1); + EXPECT_EQ(TestParameters::input(param).out_mat_t[1], 15); } { // out_mat_r auto it = find_label("out_mat_r", readinput.input_lists); it->second.str_values = {"1"}; it->second.read_value(it->second, param); - EXPECT_EQ(param.input.out_mat_r[0], 1); - EXPECT_EQ(param.input.out_mat_r[1], 8); + EXPECT_EQ(TestParameters::input(param).out_mat_r[0], 1); + EXPECT_EQ(TestParameters::input(param).out_mat_r[1], 8); it->second.str_values = {"1", "6"}; it->second.read_value(it->second, param); - EXPECT_EQ(param.input.out_mat_r[0], 1); - EXPECT_EQ(param.input.out_mat_r[1], 6); + EXPECT_EQ(TestParameters::input(param).out_mat_r[0], 1); + EXPECT_EQ(TestParameters::input(param).out_mat_r[1], 6); } { // out_mat_xc2 auto it = find_label("out_mat_xc2", readinput.input_lists); it->second.str_values = {"1"}; it->second.read_value(it->second, param); - EXPECT_EQ(param.input.out_mat_xc2[0], 1); - EXPECT_EQ(param.input.out_mat_xc2[1], 8); + EXPECT_EQ(TestParameters::input(param).out_mat_xc2[0], 1); + EXPECT_EQ(TestParameters::input(param).out_mat_xc2[1], 8); it->second.str_values = {"1", "9"}; it->second.read_value(it->second, param); - EXPECT_EQ(param.input.out_mat_xc2[0], 1); - EXPECT_EQ(param.input.out_mat_xc2[1], 9); + EXPECT_EQ(TestParameters::input(param).out_mat_xc2[0], 1); + EXPECT_EQ(TestParameters::input(param).out_mat_xc2[1], 9); } } @@ -2316,89 +2332,111 @@ TEST_F(InputTest, OutStru) { it->second.str_values = {"0"}; it->second.read_value(it->second, param); - EXPECT_EQ(param.input.out_stru, 0); + EXPECT_EQ(TestParameters::input(param).out_stru, 0); } { it->second.str_values = {"1"}; it->second.read_value(it->second, param); - EXPECT_EQ(param.input.out_stru, 1); + EXPECT_EQ(TestParameters::input(param).out_stru, 1); } { it->second.str_values = {"2"}; it->second.read_value(it->second, param); - EXPECT_EQ(param.input.out_stru, 2); + EXPECT_EQ(TestParameters::input(param).out_stru, 2); } // --- Backward-compatible boolean aliases (true -> 1, false -> 0) --- { it->second.str_values = {"true"}; it->second.read_value(it->second, param); - EXPECT_EQ(param.input.out_stru, 1); + EXPECT_EQ(TestParameters::input(param).out_stru, 1); } { it->second.str_values = {"TRUE"}; it->second.read_value(it->second, param); - EXPECT_EQ(param.input.out_stru, 1); + EXPECT_EQ(TestParameters::input(param).out_stru, 1); } { it->second.str_values = {".true."}; it->second.read_value(it->second, param); - EXPECT_EQ(param.input.out_stru, 1); + EXPECT_EQ(TestParameters::input(param).out_stru, 1); } { it->second.str_values = {"Yes"}; it->second.read_value(it->second, param); - EXPECT_EQ(param.input.out_stru, 1); + EXPECT_EQ(TestParameters::input(param).out_stru, 1); } { it->second.str_values = {"false"}; it->second.read_value(it->second, param); - EXPECT_EQ(param.input.out_stru, 0); + EXPECT_EQ(TestParameters::input(param).out_stru, 0); } { it->second.str_values = {"FALSE"}; it->second.read_value(it->second, param); - EXPECT_EQ(param.input.out_stru, 0); + EXPECT_EQ(TestParameters::input(param).out_stru, 0); } { it->second.str_values = {".false."}; it->second.read_value(it->second, param); - EXPECT_EQ(param.input.out_stru, 0); + EXPECT_EQ(TestParameters::input(param).out_stru, 0); } { it->second.str_values = {"No"}; it->second.read_value(it->second, param); - EXPECT_EQ(param.input.out_stru, 0); + EXPECT_EQ(TestParameters::input(param).out_stru, 0); } // --- Valid value check_value passes --- { for (const int v : {0, 1, 2}) { - param.input.out_stru = v; + TestParameters::input(param).out_stru = v; // Expect no exit / no crash; check_value is a void function that only // calls WARNING_QUIT on bad input. it->second.check_value(it->second, param); } } - // --- reset_value: calculation in offlist forces out_stru to 0 --- + // --- reset_value: offlist calculation forces out_stru to 0 when not user-read --- { - param.input.calculation = "get_wf"; - param.input.out_stru = 1; + TestParameters::input(param).calculation = "get_wf"; + TestParameters::input(param).out_stru = 1; + // str_values is non-empty from previous read_value calls, so + // item.is_read() is true here. Clear it to simulate the default path + // where the user did not specify out_stru in INPUT. + it->second.str_values.clear(); it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.out_stru, 0); + EXPECT_EQ(TestParameters::input(param).out_stru, 0); - param.input.calculation = "nscf"; - param.input.out_stru = 2; + TestParameters::input(param).calculation = "nscf"; + TestParameters::input(param).out_stru = 2; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.out_stru, 0); + EXPECT_EQ(TestParameters::input(param).out_stru, 0); // Non-offlist calculation preserves value - param.input.calculation = "cell-relax"; - param.input.out_stru = 1; + TestParameters::input(param).calculation = "cell-relax"; + TestParameters::input(param).out_stru = 1; it->second.reset_value(it->second, param); - EXPECT_EQ(param.input.out_stru, 1); + EXPECT_EQ(TestParameters::input(param).out_stru, 1); + } + + // --- reset_value: user-read value is preserved even for offlist calculation --- + { + it->second.str_values = {"1"}; // simulate user-specified out_stru + TestParameters::input(param).calculation = "nscf"; + TestParameters::input(param).out_stru = 1; + it->second.reset_value(it->second, param); + EXPECT_EQ(TestParameters::input(param).out_stru, 1); + + it->second.str_values = {"2"}; + TestParameters::input(param).calculation = "get_wf"; + TestParameters::input(param).out_stru = 2; + it->second.reset_value(it->second, param); + EXPECT_EQ(TestParameters::input(param).out_stru, 2); + + // Clean up so later tests are not affected. + it->second.str_values.clear(); } // --- Invalid integer values -> WARNING_QUIT via check_value --- diff --git a/source/source_io/test_serial/test_read_cube.cpp b/source/source_io/test_serial/test_read_cube.cpp new file mode 100644 index 00000000000..ce975d68170 --- /dev/null +++ b/source/source_io/test_serial/test_read_cube.cpp @@ -0,0 +1,150 @@ +#include "source_io/module_output/cube_io.h" + +#include + +#include "gmock/gmock.h" +#include "gtest/gtest.h" +#include "source_base/global_variable.h" +#include "source_base/parallel_grid.h" +#include "source_io/module_parameter/parameter.h" +#include "prepare_unitcell.h" + +Magnetism::Magnetism() +{ + this->tot_mag = 0.0; + this->abs_mag = 0.0; +} + +Magnetism::~Magnetism() +{ +} + +/*************************************************************** + * unit test of malformed-cube-file validation in read_cube + * and read_vdata_palgrid (issue #7563) + ***************************************************************/ + +/** + * - Tested Functions: + * - read_cube() + * - returns false for a truncated file (missing grid data) + * - returns false for invalid (zero) grid dimensions + * - read_vdata_palgrid() + * - aborts the run instead of reading invalid data or hanging other ranks + */ + +struct ReadCubeInvalidTest : public ::testing::Test +{ + std::vector comment; + int natom = 0; + std::vector origin; + int nx_read = 0; + int ny_read = 0; + int nz_read = 0; + std::vector dx; + std::vector dy; + std::vector dz; + std::vector atom_type; + std::vector atom_charge; + std::vector> atom_pos; + std::vector data_read; + + bool call_read_cube(const std::string& fn) + { + return ModuleIO::read_cube(fn, comment, natom, origin, + nx_read, ny_read, nz_read, + dx, dy, dz, + atom_type, atom_charge, atom_pos, data_read); + } +}; + +TEST_F(ReadCubeInvalidTest, ValidFile) +{ + EXPECT_TRUE(call_read_cube("./support/chg.cube")); + EXPECT_EQ(natom, 2); + EXPECT_EQ(nx_read, 36); + EXPECT_EQ(ny_read, 36); + EXPECT_EQ(nz_read, 36); + EXPECT_EQ(data_read.size(), 36 * 36 * 36); +} + +TEST_F(ReadCubeInvalidTest, TruncatedData) +{ + // valid header but the grid data is incomplete + const std::string fn = "test_cube_truncated.cube"; + std::ofstream ofs(fn); + ofs << "comment line 1\n"; + ofs << "comment line 2\n"; + ofs << "1 0.0 0.0 0.0\n"; + ofs << "2 1.0 0.0 0.0\n"; + ofs << "2 0.0 1.0 0.0\n"; + ofs << "2 0.0 0.0 1.0\n"; + ofs << "14 4.0 0.0 0.0 0.0\n"; + ofs << "1.0 2.0 3.0\n"; // only 3 of the 8 expected values + ofs.close(); + + EXPECT_FALSE(call_read_cube(fn)); + std::remove(fn.c_str()); +} + +TEST_F(ReadCubeInvalidTest, InvalidDimensions) +{ + // zero grid dimension is invalid + const std::string fn = "test_cube_bad_dim.cube"; + std::ofstream ofs(fn); + ofs << "comment line 1\n"; + ofs << "comment line 2\n"; + ofs << "1 0.0 0.0 0.0\n"; + ofs << "0 1.0 0.0 0.0\n"; + ofs << "2 0.0 1.0 0.0\n"; + ofs << "2 0.0 0.0 1.0\n"; + ofs << "14 4.0 0.0 0.0 0.0\n"; + ofs.close(); + + EXPECT_FALSE(call_read_cube(fn)); + std::remove(fn.c_str()); +} + +TEST_F(ReadCubeInvalidTest, NegativeNatom) +{ + const std::string fn = "test_cube_neg_natom.cube"; + std::ofstream ofs(fn); + ofs << "comment line 1\n"; + ofs << "comment line 2\n"; + ofs << "-1 0.0 0.0 0.0\n"; + ofs << "2 1.0 0.0 0.0\n"; + ofs << "2 0.0 1.0 0.0\n"; + ofs << "2 0.0 0.0 1.0\n"; + ofs.close(); + + EXPECT_FALSE(call_read_cube(fn)); + std::remove(fn.c_str()); +} + +TEST_F(ReadCubeInvalidTest, ReadVdataPalgridFails) +{ + const std::string fn = "test_cube_truncated_palgrid.cube"; + std::ofstream ofs(fn); + ofs << "comment line 1\n"; + ofs << "comment line 2\n"; + ofs << "1 0.0 0.0 0.0\n"; + ofs << "2 1.0 0.0 0.0\n"; + ofs << "2 0.0 1.0 0.0\n"; + ofs << "2 0.0 0.0 1.0\n"; + ofs << "14 4.0 0.0 0.0 0.0\n"; + ofs << "1.0 2.0 3.0\n"; + ofs.close(); + + const int nx = 2; + const int ny = 2; + const int nz = 2; + const int nrxx = nx * ny * nz; + std::vector data(nrxx, 0.0); + Parallel_Grid pgrid(nx, ny, nz, nz, nrxx, nz, 1); + std::ofstream ofs_running("unittest_read_cube.log"); + + // read_vdata_palgrid now aborts the run on a malformed cube file so that + // no rank is left waiting in the grid distribution below. + EXPECT_DEATH(ModuleIO::read_vdata_palgrid(pgrid, 0, ofs_running, fn, data.data(), 1), ""); + std::remove(fn.c_str()); +} diff --git a/source/source_lcao/CMakeLists.txt b/source/source_lcao/CMakeLists.txt index 233dd605dd8..41f0085a014 100644 --- a/source/source_lcao/CMakeLists.txt +++ b/source/source_lcao/CMakeLists.txt @@ -11,6 +11,7 @@ if(ENABLE_LCAO) add_subdirectory(module_operator_lcao) list(APPEND objects hamilt_lcao.cpp + hamilt_lcao_factory.cpp module_operator_lcao/operator_lcao.cpp module_operator_lcao/veff_lcao.cpp module_operator_lcao/veff_dh.cpp @@ -35,8 +36,8 @@ if(ENABLE_LCAO) setup_dftu_lcao.cpp pulay_fs_center2.cpp force_stress_lcao.cpp - force_lcao_gamma.cpp - force_lcao_k.cpp + force_stress_assemble.cpp + force_stress_terms.cpp stress_tools.cpp edm.cpp spar_dh.cpp @@ -54,9 +55,8 @@ if(ENABLE_LCAO) setup_nonlocal.cpp setup_exx.cpp setup_deepks.cpp - setup_dm.cpp + allocate_dm.cpp rho_tau_lcao.cpp - record_adj.cpp center2orb.cpp center2orb_orb11.cpp center2orb_orb21.cpp diff --git a/source/source_lcao/allocate_dm.cpp b/source/source_lcao/allocate_dm.cpp new file mode 100644 index 00000000000..52166fbe952 --- /dev/null +++ b/source/source_lcao/allocate_dm.cpp @@ -0,0 +1,34 @@ +#include "source_lcao/allocate_dm.h" +#include "source_base/timer.h" +#include "source_cell/klist.h" +#include "source_basis/module_ao/parallel_orbitals.h" +#include + +namespace LCAO_domain +{ + +// change init_dm to allocate_dm, mohan 2025-10-31 +// Moved from member function to free function so that Setup_DM itself +// can live in module_dm without depending on LCAO-layer modules. +template +void allocate_dm(module_dm::Setup_DM& dmat, + const K_Vectors* kv, + const Parallel_Orbitals* pv, + const int nspin) +{ + const int nspin_dm = nspin == 2 ? 2 : 1; + // pass the global physical nspin so that cal_dmr can select the spin-resolved (Pauli) + // branch for SOC/noncollinear (nspin==4), where nspin_dm itself collapses to 1. + dmat.dm = new module_dm::DensityMatrix(pv, nspin_dm, kv->kvec_d, kv->get_nks() / nspin_dm, nspin); +} + +template void allocate_dm(module_dm::Setup_DM&, + const K_Vectors*, + const Parallel_Orbitals*, + const int); +template void allocate_dm>(module_dm::Setup_DM>&, + const K_Vectors*, + const Parallel_Orbitals*, + const int); + +} // namespace LCAO_domain diff --git a/source/source_lcao/allocate_dm.h b/source/source_lcao/allocate_dm.h new file mode 100644 index 00000000000..0953aaa540f --- /dev/null +++ b/source/source_lcao/allocate_dm.h @@ -0,0 +1,21 @@ +#ifndef LCAO_ALLOCATE_DM_H +#define LCAO_ALLOCATE_DM_H + +#include "source_estate/module_dm/dm_holder.h" + +class K_Vectors; +class Parallel_Orbitals; + +namespace LCAO_domain +{ +// Allocate the DensityMatrix held by dmat, based on the k-point list and +// orbital parallel layout. Free function so that module_dm itself does not +// depend on LCAO-layer modules (K_Vectors, Parallel_Orbitals). +template +void allocate_dm(module_dm::Setup_DM& dmat, + const K_Vectors* kv, + const Parallel_Orbitals* pv, + const int nspin); +} // namespace LCAO_domain + +#endif diff --git a/source/source_lcao/edm.cpp b/source/source_lcao/edm.cpp index 18c41557d86..2847916a657 100644 --- a/source/source_lcao/edm.cpp +++ b/source/source_lcao/edm.cpp @@ -1,12 +1,12 @@ -#include "force_lcao.h" -#include "source_estate/module_dm/cal_dm_psi.h" +#include "edm.h" +#include "source_estate/module_dm/dm_from_psi.h" #include "source_estate/elecstate_lcao.h" #include "source_base/memory_recorder.h" #include "source_io/module_parameter/parameter.h" template<> -elecstate::DensityMatrix Force_LCAO::cal_edm(const elecstate::ElecState* pelec, +module_dm::DensityMatrix CalEDM::cal_edm(const elecstate::ElecState* pelec, const psi::Psi& psi, - const elecstate::DensityMatrix& dm, + const module_dm::DensityMatrix& dm, const K_Vectors& kv, const Parallel_Orbitals& pv, const int& nspin, @@ -26,7 +26,7 @@ elecstate::DensityMatrix Force_LCAO::cal_edm(const elecs } // construct a DensityMatrix for Gamma-Only - elecstate::DensityMatrix edm(&pv, nspin); + module_dm::DensityMatrix edm(&pv, nspin); #ifdef __PEXSI if (PARAM.inp.ks_solver == "pexsi") @@ -34,25 +34,25 @@ elecstate::DensityMatrix Force_LCAO::cal_edm(const elecs // auto pes = dynamic_cast*>(pelec); for (int ik = 0; ik < nspin; ik++) { - edm.set_DMK_pointer(ik, dm.pexsi_EDM[ik]); + edm.set_dmk_ptr(ik, dm.edm_pexsi[ik]); } } else #endif { - elecstate::cal_dm_psi(edm.get_paraV_pointer(), wg_ekb, psi, edm); + module_dm::dm_from_psi(&pv, wg_ekb, psi, edm); } - edm.init_DMR(ra, &ucell); - edm.cal_DMR(); + edm.init_dmr(ra, &ucell); + edm.cal_dmr(-1); return edm; } template<> -elecstate::DensityMatrix, double> Force_LCAO>::cal_edm( +module_dm::DensityMatrix, double> CalEDM>::cal_edm( const elecstate::ElecState* pelec, const psi::Psi>& psi, - const elecstate::DensityMatrix, double>& dm, + const module_dm::DensityMatrix, double>& dm, const K_Vectors& kv, const Parallel_Orbitals& pv, const int& nspin, @@ -62,8 +62,12 @@ elecstate::DensityMatrix, double> Force_LCAO, double> edm(&pv, nspin_dm, kv.kvec_d, kv.get_nks() / nspin_dm); + module_dm::DensityMatrix, double> edm(&pv, nspin_dm, kv.kvec_d, kv.get_nks() / nspin_dm, + nspin); //-------------------------------------------- // calculate the energy density matrix here. @@ -84,24 +88,24 @@ elecstate::DensityMatrix, double> Force_LCAO +class Force_Stress_LCAO; + +template +class CalEDM +{ + public: + friend class Force_Stress_LCAO; + + CalEDM(){}; + ~CalEDM(){}; + + private: + const Parallel_Orbitals* ParaV = nullptr; + + elecstate::Potential* pot = nullptr; + + module_dm::DensityMatrix cal_edm(const elecstate::ElecState* pelec, + const psi::Psi& psi, + const module_dm::DensityMatrix& dm, + const K_Vectors& kv, + const Parallel_Orbitals& pv, + const int& nspin, + const int& nbands, + const UnitCell& ucell, + Record_adj& ra) const; +}; + +#endif diff --git a/source/source_lcao/force_lcao.h b/source/source_lcao/force_lcao.h deleted file mode 100644 index e6f5f3735ce..00000000000 --- a/source/source_lcao/force_lcao.h +++ /dev/null @@ -1,124 +0,0 @@ -#ifndef LCAO_FORCE_H -#define LCAO_FORCE_H - -#include "source_base/global_function.h" -#include "source_base/matrix.h" -#include "source_basis/module_nao/two_center_bundle.h" -#include "source_estate/elecstate.h" -#include "source_estate/module_dm/density_matrix.h" -#include "source_estate/module_pot/potential_new.h" -#include "source_lcao/force_stress_arrays.h" -#include "source_psi/psi.h" -#include "source_lcao/setup_deepks.h" - -template -class Force_Stress_LCAO; - -template -class Force_LCAO -{ - public: - friend class Force_Stress_LCAO; - - Force_LCAO(){}; - ~Force_LCAO(){}; - - private: - const Parallel_Orbitals* ParaV = nullptr; - - elecstate::Potential* pot = nullptr; - - // orthonormal force + contribution from T and VNL - void ftable(const bool isforce, - const bool isstress, - ForceStressArrays& fsr, // mohan add 2024-06-16 - const UnitCell& ucell, - const Grid_Driver& gd, - const psi::Psi* psi, - const elecstate::ElecState* pelec, - const elecstate::DensityMatrix* dm, // mohan add 2025-11-04 - ModuleBase::matrix& foverlap, - ModuleBase::matrix& ftvnl_dphi, - ModuleBase::matrix& fvnl_dbeta, - ModuleBase::matrix& fvl_dphi, - ModuleBase::matrix& soverlap, - ModuleBase::matrix& stvnl_dphi, - ModuleBase::matrix& svnl_dbeta, - ModuleBase::matrix& svl_dphi, - ModuleBase::matrix& fvnl_dalpha, - ModuleBase::matrix& svnl_dalpha, - Setup_DeePKS& deepks, - const TwoCenterBundle& two_center_bundle, - const LCAO_Orbitals& orb, - const Parallel_Orbitals& pv, - const K_Vectors* kv = nullptr, - Record_adj* ra = nullptr); - - // get the ds, dt, dvnl. - void allocate(const UnitCell& ucell, - const Grid_Driver& gd, - const Parallel_Orbitals& pv, - ForceStressArrays& fsr, // mohan add 2024-06-15 - const TwoCenterBundle& two_center_bundle, - const LCAO_Orbitals& orb, - const int& nks = 0, - const std::vector>& kvec_d = {}); - - void finish_ftable(ForceStressArrays& fsr); - - void average_force(double* fm); - - // void test(Parallel_Orbitals& pv, double* mm, const std::string& name); - - //------------------------------------------------------------- - // forces reated to overlap matrix - // forces related to energy density matrix - //------------------------------------------------------------- - - void cal_fedm(const bool isforce, - const bool isstress, - ForceStressArrays& fsr, - const UnitCell& ucell, - const elecstate::DensityMatrix& dm, - const psi::Psi* psi, - const Parallel_Orbitals& pv, - const elecstate::ElecState* pelec, - ModuleBase::matrix& foverlap, - ModuleBase::matrix& soverlap, - const K_Vectors* kv = nullptr, - Record_adj* ra = nullptr); - - //------------------------------------------------------------- - // forces related to kinetic and non-local pseudopotentials - //-------------------------------------------------------------- - void cal_ftvnl_dphi(const elecstate::DensityMatrix* dm, - const Parallel_Orbitals& pv, - const UnitCell& ucell, - ForceStressArrays& fsr, - const bool isforce, - const bool isstress, - ModuleBase::matrix& ftvnl_dphi, - ModuleBase::matrix& stvnl_dphi, - Record_adj* ra = nullptr); - - //------------------------------------------- - // forces related to local pseudopotentials - //------------------------------------------- - void cal_fvl_dphi(const bool isforce, - const bool isstress, - const elecstate::Potential* pot_in, - ModuleBase::matrix& fvl_dphi, - ModuleBase::matrix& svl_dphi); - - elecstate::DensityMatrix cal_edm(const elecstate::ElecState* pelec, - const psi::Psi& psi, - const elecstate::DensityMatrix& dm, - const K_Vectors& kv, - const Parallel_Orbitals& pv, - const int& nspin, - const int& nbands, - const UnitCell& ucell, - Record_adj& ra) const; -}; - -#endif diff --git a/source/source_lcao/force_lcao_gamma.cpp b/source/source_lcao/force_lcao_gamma.cpp deleted file mode 100644 index 8ebfe08b9ef..00000000000 --- a/source/source_lcao/force_lcao_gamma.cpp +++ /dev/null @@ -1,276 +0,0 @@ -#include "force_lcao.h" -#include "source_base/memory_recorder.h" -#include "source_base/parallel_reduce.h" -#include "source_base/timer.h" -#include "source_cell/module_neighbor/sltk_grid_driver.h" -#include "source_io/module_parameter/parameter.h" -#ifdef __MLALGO -#include "source_lcao/module_deepks/lcao_deepks.h" //caoyu add for deepks on 20210813 -#include "source_lcao/module_deepks/lcao_deepks_io.h" -#include "source_lcao/module_deepks/deepks_force.h" -#endif -#include "source_cell/module_neighbor/sltk_grid_driver.h" //GridD -#include "source_estate/elecstate_lcao.h" -#include "source_lcao/lcao_domain.h" -#include "source_lcao/pulay_fs.h" - -template <> -void Force_LCAO::allocate(const UnitCell& ucell, - const Grid_Driver& gd, - const Parallel_Orbitals& pv, - ForceStressArrays& fsr, // mohan add 2024-06-15 - const TwoCenterBundle& two_center_bundle, - const LCAO_Orbitals& orb, - const int& nks, - const std::vector>& kvec_d) -{ - ModuleBase::TITLE("Forces", "allocate"); - ModuleBase::timer::start("Forces", "allocate"); - - // need to calculate the derivative in build_ST_new - bool cal_deri = true; - this->ParaV = &pv; - - // calculate dS in LCAO - // liaochen add on 2010/7/12 - // save the results in dense matrix by now. - // pv.nloc: number of H elements in this proc. - - assert(pv.nloc > 0); - fsr.DSloc_x = new double[pv.nloc]; - fsr.DSloc_y = new double[pv.nloc]; - fsr.DSloc_z = new double[pv.nloc]; - ModuleBase::GlobalFunc::ZEROS(fsr.DSloc_x, pv.nloc); - ModuleBase::GlobalFunc::ZEROS(fsr.DSloc_y, pv.nloc); - ModuleBase::GlobalFunc::ZEROS(fsr.DSloc_z, pv.nloc); - ModuleBase::Memory::record("Force::dS_GO", sizeof(double) * pv.nloc * 3); - // allocate stress part in gamma_only-line, added by zhengdy-stress - if (PARAM.inp.cal_stress) - { - fsr.DSloc_11 = new double[pv.nloc]; - fsr.DSloc_12 = new double[pv.nloc]; - fsr.DSloc_13 = new double[pv.nloc]; - fsr.DSloc_22 = new double[pv.nloc]; - fsr.DSloc_23 = new double[pv.nloc]; - fsr.DSloc_33 = new double[pv.nloc]; - ModuleBase::GlobalFunc::ZEROS(fsr.DSloc_11, pv.nloc); - ModuleBase::GlobalFunc::ZEROS(fsr.DSloc_12, pv.nloc); - ModuleBase::GlobalFunc::ZEROS(fsr.DSloc_13, pv.nloc); - ModuleBase::GlobalFunc::ZEROS(fsr.DSloc_22, pv.nloc); - ModuleBase::GlobalFunc::ZEROS(fsr.DSloc_23, pv.nloc); - ModuleBase::GlobalFunc::ZEROS(fsr.DSloc_33, pv.nloc); - fsr.DHloc_fixed_11 = new double[pv.nloc]; - fsr.DHloc_fixed_12 = new double[pv.nloc]; - fsr.DHloc_fixed_13 = new double[pv.nloc]; - fsr.DHloc_fixed_22 = new double[pv.nloc]; - fsr.DHloc_fixed_23 = new double[pv.nloc]; - fsr.DHloc_fixed_33 = new double[pv.nloc]; - ModuleBase::GlobalFunc::ZEROS(fsr.DHloc_fixed_11, pv.nloc); - ModuleBase::GlobalFunc::ZEROS(fsr.DHloc_fixed_12, pv.nloc); - ModuleBase::GlobalFunc::ZEROS(fsr.DHloc_fixed_13, pv.nloc); - ModuleBase::GlobalFunc::ZEROS(fsr.DHloc_fixed_22, pv.nloc); - ModuleBase::GlobalFunc::ZEROS(fsr.DHloc_fixed_23, pv.nloc); - ModuleBase::GlobalFunc::ZEROS(fsr.DHloc_fixed_33, pv.nloc); - ModuleBase::Memory::record("Stress::dSH_GO", sizeof(double) * pv.nloc * 12); - } - // calculate dS in LCAO basis - LCAO_domain::build_ST_new(fsr, - 'S', - cal_deri, - PARAM.inp.cal_stress, - ucell, - orb, - pv, - two_center_bundle, - &gd, - nullptr); - - // calculate dT in LCAP - // allocation dt - // liaochen add on 2010/7/12 - fsr.DHloc_fixed_x = new double[pv.nloc]; - fsr.DHloc_fixed_y = new double[pv.nloc]; - fsr.DHloc_fixed_z = new double[pv.nloc]; - ModuleBase::Memory::record("Force::dTVNL", sizeof(double) * pv.nloc * 3); - ModuleBase::GlobalFunc::ZEROS(fsr.DHloc_fixed_x, pv.nloc); - ModuleBase::GlobalFunc::ZEROS(fsr.DHloc_fixed_y, pv.nloc); - ModuleBase::GlobalFunc::ZEROS(fsr.DHloc_fixed_z, pv.nloc); - - // calculate dT - // calculate T + VNL(P1) in LCAO basis - LCAO_domain::build_ST_new(fsr, - 'T', - cal_deri, - PARAM.inp.cal_stress, - ucell, - orb, - pv, - two_center_bundle, - &gd, - nullptr); - - ModuleBase::timer::end("Forces", "allocate"); - return; -} - -template <> -void Force_LCAO::finish_ftable(ForceStressArrays& fsr) -{ - delete[] fsr.DSloc_x; - delete[] fsr.DSloc_y; - delete[] fsr.DSloc_z; - delete[] fsr.DHloc_fixed_x; - delete[] fsr.DHloc_fixed_y; - delete[] fsr.DHloc_fixed_z; - - if (PARAM.inp.cal_stress) // added by zhengdy-stress - { - delete[] fsr.DSloc_11; - delete[] fsr.DSloc_12; - delete[] fsr.DSloc_13; - delete[] fsr.DSloc_22; - delete[] fsr.DSloc_23; - delete[] fsr.DSloc_33; - delete[] fsr.DHloc_fixed_11; - delete[] fsr.DHloc_fixed_12; - delete[] fsr.DHloc_fixed_13; - delete[] fsr.DHloc_fixed_22; - delete[] fsr.DHloc_fixed_23; - delete[] fsr.DHloc_fixed_33; - } - return; -} - -// be called in force_lo.cpp -template <> -void Force_LCAO::ftable(const bool isforce, - const bool isstress, - ForceStressArrays& fsr, // mohan add 2024-06-16 - const UnitCell& ucell, - const Grid_Driver& gd, - const psi::Psi* psi, - const elecstate::ElecState* pelec, - const elecstate::DensityMatrix* dm, // mohan add 2025-11-04 - ModuleBase::matrix& foverlap, - ModuleBase::matrix& ftvnl_dphi, - ModuleBase::matrix& fvnl_dbeta, - ModuleBase::matrix& fvl_dphi, - ModuleBase::matrix& soverlap, - ModuleBase::matrix& stvnl_dphi, - ModuleBase::matrix& svnl_dbeta, - ModuleBase::matrix& svl_dphi, - ModuleBase::matrix& fvnl_dalpha, - ModuleBase::matrix& svnl_dalpha, - Setup_DeePKS& deepks, - const TwoCenterBundle& two_center_bundle, - const LCAO_Orbitals& orb, - const Parallel_Orbitals& pv, - const K_Vectors* kv, - Record_adj* ra) -{ - ModuleBase::TITLE("Forces", "ftable"); - ModuleBase::timer::start("Forces", "ftable"); - - this->ParaV = dm->get_paraV_pointer(); - - // allocate DSloc_x, DSloc_y, DSloc_z - // allocate DHloc_fixed_x, DHloc_fixed_y, DHloc_fixed_z - this->allocate(ucell, gd, pv, fsr, two_center_bundle, orb); - - const double* dSx[3] = {fsr.DSloc_x, fsr.DSloc_y, fsr.DSloc_z}; - const double* dSxy[6] = {fsr.DSloc_11, fsr.DSloc_12, fsr.DSloc_13, fsr.DSloc_22, fsr.DSloc_23, fsr.DSloc_33}; - // calculate the force related to 'energy density matrix'. - PulayForceStress::cal_pulay_fs( - foverlap, - soverlap, - this->cal_edm(pelec, *psi, *dm, *kv, pv, PARAM.inp.nspin, PARAM.inp.nbands, ucell, *ra), - ucell, - pv, - dSx, - dSxy, - isforce, - isstress); - - const double* dHx[3] = {fsr.DHloc_fixed_x, fsr.DHloc_fixed_y, fsr.DHloc_fixed_z}; - const double* dHxy[6] = {fsr.DHloc_fixed_11, - fsr.DHloc_fixed_12, - fsr.DHloc_fixed_13, - fsr.DHloc_fixed_22, - fsr.DHloc_fixed_23, - fsr.DHloc_fixed_33}; - // tvnl_dphi - PulayForceStress::cal_pulay_fs(ftvnl_dphi, stvnl_dphi, *dm, ucell, pv, dHx, dHxy, isforce, isstress); - - // vl_dphi - PulayForceStress::cal_pulay_fs(fvl_dphi, - svl_dphi, - *dm, - ucell, - pelec->pot, - isforce, - isstress, - false /*reset dm to gint*/); - -#ifdef __MLALGO - if (PARAM.inp.deepks_scf) - { - // No need to update E_delta here since it have been done in LCAO_Deepks_Interface in after_scf - const int nks = 1; - DeePKS_domain::cal_f_delta(ucell, - orb, - gd, - *this->ParaV, - nks, - deepks.ld.deepks_param, - kv->kvec_d, - deepks.ld.phialpha, - fvnl_dalpha, - isstress, - svnl_dalpha, - deepks.ld.dm_r, - deepks.ld.gedm); - } -#endif - - if (isforce) - { - Parallel_Reduce::reduce_pool(foverlap.c, foverlap.nr * foverlap.nc); - Parallel_Reduce::reduce_pool(ftvnl_dphi.c, ftvnl_dphi.nr * ftvnl_dphi.nc); - Parallel_Reduce::reduce_pool(fvnl_dbeta.c, fvnl_dbeta.nr * fvnl_dbeta.nc); - Parallel_Reduce::reduce_pool(fvl_dphi.c, fvl_dphi.nr * fvl_dphi.nc); -#ifdef __MLALGO - Parallel_Reduce::reduce_pool(fvnl_dalpha.c, fvnl_dalpha.nr * fvnl_dalpha.nc); -#endif - } - if (isstress) - { - Parallel_Reduce::reduce_pool(soverlap.c, soverlap.nr * soverlap.nc); - Parallel_Reduce::reduce_pool(stvnl_dphi.c, stvnl_dphi.nr * stvnl_dphi.nc); - Parallel_Reduce::reduce_pool(svnl_dbeta.c, svnl_dbeta.nr * svnl_dbeta.nc); - Parallel_Reduce::reduce_pool(svl_dphi.c, svl_dphi.nr * svl_dphi.nc); -#ifdef __MLALGO - Parallel_Reduce::reduce_pool(svnl_dalpha.c, svnl_dalpha.nr * svnl_dalpha.nc); -#endif - } - -#ifdef __MLALGO - if (PARAM.inp.deepks_scf && PARAM.inp.deepks_out_unittest) - { - std::ofstream ofs_f("F_delta.dat"); - std::ofstream ofs_s("stress_delta.dat"); - ofs_f << std::setprecision(10); - ofs_s << std::setprecision(10); - fvnl_dalpha.print(ofs_f); - ofs_f.close(); - svnl_dalpha.print(ofs_s); - ofs_s.close(); - } -#endif - - // delete DSloc_x, DSloc_y, DSloc_z - // delete DHloc_fixed_x, DHloc_fixed_y, DHloc_fixed_z - this->finish_ftable(fsr); - - ModuleBase::timer::end("Forces", "ftable"); - return; -} diff --git a/source/source_lcao/force_lcao_k.cpp b/source/source_lcao/force_lcao_k.cpp deleted file mode 100644 index 6af4106f11f..00000000000 --- a/source/source_lcao/force_lcao_k.cpp +++ /dev/null @@ -1,282 +0,0 @@ -#include "force_lcao.h" -#include "source_base/memory_recorder.h" -#include "source_base/parallel_reduce.h" -#include "source_base/timer.h" -#include "source_base/tool_threading.h" -#include "source_basis/module_ao/orb_read.h" -#include "source_cell/module_neighbor/sltk_grid_driver.h" -#include "source_estate/cal_dm.h" -#include "source_estate/elecstate_lcao.h" -#include "source_estate/module_dm/cal_dm_psi.h" -#include "source_lcao/lcao_domain.h" -#include "source_lcao/pulay_fs.h" -#include "source_io/module_parameter/parameter.h" - -#include -#include - -#ifdef __MLALGO -#include "source_lcao/module_deepks/lcao_deepks.h" -#include "source_lcao/module_deepks/deepks_force.h" -#endif - -#ifdef _OPENMP -#include -#endif - -template <> -void Force_LCAO>::allocate(const UnitCell& ucell, - const Grid_Driver& gd, - const Parallel_Orbitals& pv, - ForceStressArrays& fsr, // mohan add 2024-06-15 - const TwoCenterBundle& two_center_bundle, - const LCAO_Orbitals& orb, - const int& nks, - const std::vector>& kvec_d) -{ - ModuleBase::TITLE("Forces", "allocate"); - ModuleBase::timer::start("Forces", "allocate"); - - const int nnr = pv.nnr; - - assert(nnr >= 0); - - //-------------------------------- - // (1) allocate for dSx dSy & dSz - //-------------------------------- - fsr.DSloc_Rx = new double[nnr]; - fsr.DSloc_Ry = new double[nnr]; - fsr.DSloc_Rz = new double[nnr]; - - const auto init_DSloc_Rxyz = [this, nnr, &fsr](int num_threads, int thread_id) { - int beg = 0; - int len = 0; - ModuleBase::BLOCK_TASK_DIST_1D(num_threads, thread_id, nnr, 1024, beg, len); - ModuleBase::GlobalFunc::ZEROS(fsr.DSloc_Rx + beg, len); - ModuleBase::GlobalFunc::ZEROS(fsr.DSloc_Ry + beg, len); - ModuleBase::GlobalFunc::ZEROS(fsr.DSloc_Rz + beg, len); - }; - - ModuleBase::OMP_PARALLEL(init_DSloc_Rxyz); - ModuleBase::Memory::record("Force::dS_K", sizeof(double) * nnr * 3); - - if (PARAM.inp.cal_stress) - { - fsr.DH_r = new double[3 * nnr]; - fsr.stvnl11 = new double[nnr]; - fsr.stvnl12 = new double[nnr]; - fsr.stvnl13 = new double[nnr]; - fsr.stvnl22 = new double[nnr]; - fsr.stvnl23 = new double[nnr]; - fsr.stvnl33 = new double[nnr]; - const auto init_DH_r_stvnl = [this, nnr, &fsr](int num_threads, int thread_id) { - int beg, len; - ModuleBase::BLOCK_TASK_DIST_1D(num_threads, thread_id, nnr, 1024, beg, len); - ModuleBase::GlobalFunc::ZEROS(fsr.DH_r + 3 * beg, 3 * len); - ModuleBase::GlobalFunc::ZEROS(fsr.stvnl11 + beg, len); - ModuleBase::GlobalFunc::ZEROS(fsr.stvnl12 + beg, len); - ModuleBase::GlobalFunc::ZEROS(fsr.stvnl13 + beg, len); - ModuleBase::GlobalFunc::ZEROS(fsr.stvnl22 + beg, len); - ModuleBase::GlobalFunc::ZEROS(fsr.stvnl23 + beg, len); - ModuleBase::GlobalFunc::ZEROS(fsr.stvnl33 + beg, len); - }; - ModuleBase::OMP_PARALLEL(init_DH_r_stvnl); - - ModuleBase::Memory::record("Stress::dHr", sizeof(double) * nnr * 3); - ModuleBase::Memory::record("Stress::dSR", sizeof(double) * nnr * 6); - } - - //----------------------------- - // calculate dS = - //----------------------------- - bool cal_deri = true; - LCAO_domain::build_ST_new(fsr, - 'S', - cal_deri, - PARAM.inp.cal_stress, - ucell, - orb, - pv, - two_center_bundle, - &gd, - nullptr); // delete lm.SlocR - - //----------------------------------------- - // (2) allocate for - //----------------------------------------- - fsr.DHloc_fixedR_x = new double[nnr]; - fsr.DHloc_fixedR_y = new double[nnr]; - fsr.DHloc_fixedR_z = new double[nnr]; - - const auto init_DHloc_fixedR_xyz = [this, nnr, &fsr](int num_threads, int thread_id) { - int beg = 0; - int len = 0; - ModuleBase::BLOCK_TASK_DIST_1D(num_threads, thread_id, nnr, 1024, beg, len); - ModuleBase::GlobalFunc::ZEROS(fsr.DHloc_fixedR_x + beg, len); - ModuleBase::GlobalFunc::ZEROS(fsr.DHloc_fixedR_y + beg, len); - ModuleBase::GlobalFunc::ZEROS(fsr.DHloc_fixedR_z + beg, len); - }; - ModuleBase::OMP_PARALLEL(init_DHloc_fixedR_xyz); - ModuleBase::Memory::record("Force::dTVNL", sizeof(double) * nnr * 3); - - // calculate dT= in LCAO - // calculate T + VNL(P1) in LCAO basis - LCAO_domain::build_ST_new(fsr, - 'T', - cal_deri, - PARAM.inp.cal_stress, - ucell, - orb, - pv, - two_center_bundle, - &gd, - nullptr); // delete lm.Hloc_fixedR - - ModuleBase::timer::end("Forces", "allocate"); - return; -} - -template <> -void Force_LCAO>::finish_ftable(ForceStressArrays& fsr) -{ - delete[] fsr.DSloc_Rx; - delete[] fsr.DSloc_Ry; - delete[] fsr.DSloc_Rz; - delete[] fsr.DHloc_fixedR_x; - delete[] fsr.DHloc_fixedR_y; - delete[] fsr.DHloc_fixedR_z; - - if (PARAM.inp.cal_stress) - { - delete[] fsr.DH_r; - delete[] fsr.stvnl11; - delete[] fsr.stvnl12; - delete[] fsr.stvnl13; - delete[] fsr.stvnl22; - delete[] fsr.stvnl23; - delete[] fsr.stvnl33; - } - return; -} - -// be called in Force_LCAO::start_force_calculation -template <> -void Force_LCAO>::ftable(const bool isforce, - const bool isstress, - ForceStressArrays& fsr, // mohan add 2024-06-15 - const UnitCell& ucell, - const Grid_Driver& gd, - const psi::Psi>* psi, - const elecstate::ElecState* pelec, - const elecstate::DensityMatrix, double>* dm, // mohan add 2025-11-04 - ModuleBase::matrix& foverlap, - ModuleBase::matrix& ftvnl_dphi, - ModuleBase::matrix& fvnl_dbeta, - ModuleBase::matrix& fvl_dphi, - ModuleBase::matrix& soverlap, - ModuleBase::matrix& stvnl_dphi, - ModuleBase::matrix& svnl_dbeta, - ModuleBase::matrix& svl_dphi, - ModuleBase::matrix& fvnl_dalpha, - ModuleBase::matrix& svnl_dalpha, - Setup_DeePKS>& deepks, - const TwoCenterBundle& two_center_bundle, - const LCAO_Orbitals& orb, - const Parallel_Orbitals& pv, - const K_Vectors* kv, - Record_adj* ra) -{ - ModuleBase::TITLE("Forces", "ftable"); - ModuleBase::timer::start("Forces", "ftable"); - - this->allocate(ucell, - gd, - pv, - fsr, // mohan add 2024-06-16 - two_center_bundle, - orb, - kv->get_nks(), - kv->kvec_d); - - const double* dSx[3] = {fsr.DSloc_Rx, fsr.DSloc_Ry, fsr.DSloc_Rz}; - - // calculate the energy density matrix - // and the force related to overlap matrix and energy density matrix. - PulayForceStress::cal_pulay_fs( - foverlap, soverlap, - this->cal_edm(pelec, *psi, *dm, *kv, pv, PARAM.inp.nspin, PARAM.inp.nbands, ucell, *ra), - ucell, pv, dSx, fsr.DH_r, isforce, isstress, ra, -1.0, 1.0); - - const double* dHx[3] = {fsr.DHloc_fixedR_x, fsr.DHloc_fixedR_y, fsr.DHloc_fixedR_z}; // T+Vnl - const double* dHxy[6] = {fsr.stvnl11, fsr.stvnl12, fsr.stvnl13, fsr.stvnl22, fsr.stvnl23, fsr.stvnl33}; // T - - // tvnl_dphi - PulayForceStress::cal_pulay_fs(ftvnl_dphi, stvnl_dphi, *dm, ucell, pv, dHx, dHxy, isforce, isstress, ra, 1.0, -1.0); - - // doing on the real space grid. - // vl_dphi - PulayForceStress::cal_pulay_fs(fvl_dphi, svl_dphi, *dm, ucell, - pelec->pot, isforce, isstress, - false /*reset dm to gint*/); - -#ifdef __MLALGO - if (PARAM.inp.deepks_scf) - { - // No need to update E_delta since it have been done in LCAO_Deepks_Interface in after_scf - DeePKS_domain::cal_f_delta>(ucell, - orb, - gd, - pv, - kv->get_nks(), - deepks.ld.deepks_param, - kv->kvec_d, - deepks.ld.phialpha, - fvnl_dalpha, - isstress, - svnl_dalpha, - deepks.ld.dm_r, - deepks.ld.gedm); - } -#endif - - //---------------------------------------------------------------- - // reduce the force according to 2D distribution of H & S matrix. - //---------------------------------------------------------------- - if (isforce) - { - Parallel_Reduce::reduce_pool(foverlap.c, foverlap.nr * foverlap.nc); - Parallel_Reduce::reduce_pool(ftvnl_dphi.c, ftvnl_dphi.nr * ftvnl_dphi.nc); - Parallel_Reduce::reduce_pool(fvnl_dbeta.c, fvnl_dbeta.nr * fvnl_dbeta.nc); - Parallel_Reduce::reduce_pool(fvl_dphi.c, fvl_dphi.nr * fvl_dphi.nc); -#ifdef __MLALGO - Parallel_Reduce::reduce_pool(fvnl_dalpha.c, fvnl_dalpha.nr * fvnl_dalpha.nc); -#endif - } - if (isstress) - { - Parallel_Reduce::reduce_pool(soverlap.c, soverlap.nr * soverlap.nc); - Parallel_Reduce::reduce_pool(stvnl_dphi.c, stvnl_dphi.nr * stvnl_dphi.nc); - Parallel_Reduce::reduce_pool(svnl_dbeta.c, svnl_dbeta.nr * svnl_dbeta.nc); - Parallel_Reduce::reduce_pool(svl_dphi.c, svl_dphi.nr * svl_dphi.nc); -#ifdef __MLALGO - Parallel_Reduce::reduce_pool(svnl_dalpha.c, svnl_dalpha.nr * svnl_dalpha.nc); -#endif - } - -#ifdef __MLALGO - if (PARAM.inp.deepks_scf && PARAM.inp.deepks_out_unittest) - { - std::ofstream ofs_f("F_delta.dat"); - std::ofstream ofs_s("stress_delta.dat"); - ofs_f << std::setprecision(10); - ofs_s << std::setprecision(10); - fvnl_dalpha.print(ofs_f); - ofs_f.close(); - svnl_dalpha.print(ofs_s); - ofs_s.close(); - } -#endif - - ModuleBase::timer::end("Forces", "ftable"); - return; -} diff --git a/source/source_lcao/force_stress_arrays.h b/source/source_lcao/force_stress_arrays.h index 6d9b726b2db..92994b0585b 100644 --- a/source/source_lcao/force_stress_arrays.h +++ b/source/source_lcao/force_stress_arrays.h @@ -1,6 +1,8 @@ -#ifndef FORCESTRESS_ARRAYS_H +#ifndef FORCESTRESS_ARRAYS_H #define FORCESTRESS_ARRAYS_H +#include + class ForceStressArrays { public: @@ -12,60 +14,46 @@ class ForceStressArrays // force in LCAO // used in gamma only algorithm. //----------------------------------------- - double* DSloc_x = nullptr; - double* DSloc_y = nullptr; - double* DSloc_z = nullptr; + std::vector DSloc_x; + std::vector DSloc_y; + std::vector DSloc_z; //----------------------------------------- // force in LCAO // used in k-points algorithm. //----------------------------------------- - double* DSloc_Rx = nullptr; - double* DSloc_Ry = nullptr; - double* DSloc_Rz = nullptr; + std::vector DSloc_Rx; + std::vector DSloc_Ry; + std::vector DSloc_Rz; //----------------------------------------- // dT + part of dVNL // used in gamma only algorithm. //----------------------------------------- - double* DHloc_fixed_x = nullptr; - double* DHloc_fixed_y = nullptr; - double* DHloc_fixed_z = nullptr; + std::vector DHloc_fixed_x; + std::vector DHloc_fixed_y; + std::vector DHloc_fixed_z; //----------------------------------------- // dT + part of dVNL // used in kpoint algorithm. //----------------------------------------- - double* DHloc_fixedR_x = nullptr; - double* DHloc_fixedR_y = nullptr; - double* DHloc_fixedR_z = nullptr; + std::vector DHloc_fixedR_x; + std::vector DHloc_fixedR_y; + std::vector DHloc_fixedR_z; //---------------------------------------- // r_mu - r_nu //---------------------------------------- - double* DH_r = nullptr;//zhengdy added 2017-07 - - double* stvnl11 = nullptr; - double* stvnl12 = nullptr; - double* stvnl13 = nullptr; - double* stvnl22 = nullptr; - double* stvnl23 = nullptr; - double* stvnl33 = nullptr; - - double* DSloc_11 = nullptr; - double* DSloc_12 = nullptr; - double* DSloc_13 = nullptr; - double* DSloc_22 = nullptr; - double* DSloc_23 = nullptr; - double* DSloc_33 = nullptr; + std::vector DH_r;//zhengdy added 2017-07 - double* DHloc_fixed_11 = nullptr; - double* DHloc_fixed_12 = nullptr; - double* DHloc_fixed_13 = nullptr; - double* DHloc_fixed_22 = nullptr; - double* DHloc_fixed_23 = nullptr; - double* DHloc_fixed_33 = nullptr; + std::vector stvnl11; + std::vector stvnl12; + std::vector stvnl13; + std::vector stvnl22; + std::vector stvnl23; + std::vector stvnl33; }; diff --git a/source/source_lcao/force_stress_assemble.cpp b/source/source_lcao/force_stress_assemble.cpp new file mode 100644 index 00000000000..6590de0c2a2 --- /dev/null +++ b/source/source_lcao/force_stress_assemble.cpp @@ -0,0 +1,403 @@ +#include "force_stress_assemble.h" + +#include + +#include "source_base/constants.h" +#include "source_base/global_function.h" +#include "source_base/global_variable.h" +#include "source_base/tool_quit.h" +#include "source_cell/module_symmetry/symmetry.h" +#include "source_cell/unitcell.h" +#include "source_hamilt/module_vdw/vdw.h" +#include "source_io/module_output/output_log.h" +#include "source_io/module_parameter/parameter.h" +#include "source_lcao/setup_deepks.h" // DeePKS_domain::write_forces/write_stress +#ifdef __MLALGO +#include "source_lcao/module_deepks/lcao_deepks.h" +#include "source_lcao/module_deepks/lcao_deepks_io.h" +#endif + +namespace LCAO_domain +{ + +namespace +{ +// Accumulate every active force term into fcs for one Cartesian component. +// Splitting this out of assemble_print_force keeps the latter's cyclomatic +// complexity low; the branch conditions mirror the terms that were computed. +void sum_force_terms(const int iat, + const int i, + const vdw::VdwResult* vdw_result, + const Exx_Info& exx_info, + const LCAOForceParts& parts, + ModuleBase::matrix& fcs) +{ + fcs(iat, i) += parts.foverlap(iat, i) + parts.ftvnl_dphi(iat, i) + parts.fvnl_dbeta(iat, i) + parts.fvl_dphi(iat, i) + + parts.fvl_dvl(iat, i) // derivative of local potential force (pw) + + parts.fewalds(iat, i) // ewald force (pw) + + parts.fcc(iat, i) // nonlinear core correction force (pw) + + parts.fscc(iat, i) // self consistent corretion force (pw) + + parts.fpothybrid(iat, i); // pulay force for hybrid gauge rt-tddft + + // Force contribution from DFT+U, Quxin add on 20201029 + if (PARAM.inp.dft_plus_u) + { + fcs(iat, i) += parts.force_u(iat, i); + } + if (PARAM.inp.sc_mag_switch) + { + fcs(iat, i) += parts.force_dspin(iat, i); + } +#ifdef __EXX + // Force contribution from exx + if (exx_info.info_global.cal_exx) + { + fcs(iat, i) += parts.force_exx(iat, i); + } +#endif + // VDW force of vdwd2 or vdwd3 + if (vdw_result != nullptr) + { + fcs(iat, i) += parts.force_vdw(iat, i); + } + // E-field force + if (PARAM.inp.efield_flag) + { + fcs(iat, i) += parts.fefield(iat, i); + } + // E-field force of tddft + if (PARAM.inp.esolver_type == "tddft") + { + fcs(iat, i) += parts.fefield_tddft(iat, i); + } + // Gate field force + if (PARAM.inp.gate_flag) + { + fcs(iat, i) += parts.fgate(iat, i); + } + // implicit solvation model + if (PARAM.inp.imp_sol) + { + fcs(iat, i) += parts.fsol(iat, i); + } +#ifdef __MLALGO + // mohan add 2021-08-04 + if (PARAM.inp.deepks_scf) + { + fcs(iat, i) += parts.fvnl_dalpha(iat, i); + } +#endif +} + +// Accumulate every active stress term into scs for one tensor component. +void sum_stress_terms(const int i, + const int j, + const vdw::VdwResult* vdw_result, + const Exx_Info& exx_info, + const LCAOStressParts& sparts, + ModuleBase::matrix& scs) +{ + scs(i, j) += sparts.soverlap(i, j) + sparts.stvnl_dphi(i, j) + sparts.svnl_dbeta(i, j) + sparts.svl_dphi(i, j) + + sparts.sigmadvl(i, j) // derivative of local potential stress (pw) + + sparts.sigmaewa(i, j) // ewald stress (pw) + + sparts.sigmacc(i, j) // nonlinear core correction stress (pw) + + sparts.sigmaxc(i, j) // exchange corretion stress + + sparts.sigmahar(i, j); // hartree stress + + // VDW stress from linpz and jiyy + if (vdw_result != nullptr) + { + scs(i, j) += sparts.stress_vdw(i, j); + } + // DFT plus U stress from qux + if (PARAM.inp.dft_plus_u) + { + scs(i, j) += sparts.stress_u(i, j); + } + if (PARAM.inp.sc_mag_switch) + { + scs(i, j) += sparts.stress_dspin(i, j); + } +#ifdef __EXX + // Stress contribution from exx + if (exx_info.info_global.cal_exx) + { + scs(i, j) += sparts.stress_exx(i, j); + } +#endif +#ifdef __MLALGO + if (PARAM.inp.deepks_scf) + { + scs(i, j) += sparts.svnl_dalpha(i, j); + } +#endif +} +} // namespace + +namespace +{ +// Print every individual force term (test output only, istestf == true). +void print_force_parts(const UnitCell& ucell, + const vdw::VdwResult* vdw_result, + const LCAOForceParts& parts) +{ + const int nat = ucell.nat; + ModuleBase::matrix ftvnl; + ftvnl.create(nat, 3); + for (int iat = 0; iat < nat; iat++) + { + for (int i = 0; i < 3; i++) + { + ftvnl(iat, i) = parts.ftvnl_dphi(iat, i) + parts.fvnl_dbeta(iat, i); + } + } + + GlobalV::ofs_running << "\n PARTS OF FORCE: " << std::endl; + GlobalV::ofs_running << std::setiosflags(std::ios::showpos); + GlobalV::ofs_running << std::setiosflags(std::ios::fixed) << std::setprecision(8) << std::endl; + ModuleIO::print_force(GlobalV::ofs_running, ucell, "OVERLAP FORCE", parts.foverlap, false); + ModuleIO::print_force(GlobalV::ofs_running, ucell, "TVNL_DPHI force", parts.ftvnl_dphi, false); + ModuleIO::print_force(GlobalV::ofs_running, ucell, "VNL_DBETA force", parts.fvnl_dbeta, false); + ModuleIO::print_force(GlobalV::ofs_running, ucell, "T_VNL FORCE", ftvnl, false); + ModuleIO::print_force(GlobalV::ofs_running, ucell, "VL_dPHI FORCE", parts.fvl_dphi, false); + ModuleIO::print_force(GlobalV::ofs_running, ucell, "VL_dVL FORCE", parts.fvl_dvl, false); + ModuleIO::print_force(GlobalV::ofs_running, ucell, "EWALD FORCE", parts.fewalds, false); + ModuleIO::print_force(GlobalV::ofs_running, ucell, "NLCC FORCE", parts.fcc, false); + ModuleIO::print_force(GlobalV::ofs_running, ucell, "SCC FORCE", parts.fscc, false); + if (PARAM.inp.efield_flag) + { + ModuleIO::print_force(GlobalV::ofs_running, ucell, "EFIELD FORCE", parts.fefield, false); + } + if (PARAM.inp.esolver_type == "tddft") + { + ModuleIO::print_force(GlobalV::ofs_running, ucell, "EFIELD_TDDFT FORCE", parts.fefield_tddft, false); + } + if (PARAM.inp.gate_flag) + { + ModuleIO::print_force(GlobalV::ofs_running, ucell, "GATEFIELD FORCE", parts.fgate, false); + } + if (PARAM.inp.imp_sol) + { + ModuleIO::print_force(GlobalV::ofs_running, ucell, "IMP_SOL FORCE", parts.fsol, false); + } + if (vdw_result != nullptr) + { + ModuleIO::print_force(GlobalV::ofs_running, ucell, "VDW FORCE", parts.force_vdw, false); + } + if (PARAM.inp.dft_plus_u) + { + ModuleIO::print_force(GlobalV::ofs_running, ucell, "DFT+U FORCE", parts.force_u, false); + } + if (PARAM.inp.sc_mag_switch) + { + ModuleIO::print_force(GlobalV::ofs_running, ucell, "DeltaSpin FORCE", parts.force_dspin, false); + } +#ifdef __MLALGO + // caoyu add 2021-06-03 + if (PARAM.inp.deepks_scf) + { + ModuleIO::print_force(GlobalV::ofs_running, ucell, "DeePKS FORCE", parts.fvnl_dalpha, true); + } +#endif +} + +// Print the per-atom flag table and zero out sub-threshold force components +// (test output only, istestf == true). +void print_force_invalid_table(const UnitCell& ucell, + const double force_threshold, + ModuleBase::matrix& fcs) +{ + GlobalV::ofs_running << "\n FORCE INVALID TABLE." << std::endl; + GlobalV::ofs_running << " " << std::setw(8) << "atom" << std::setw(5) << "x" << std::setw(5) << "y" + << std::setw(5) << "z" << std::endl; + for (int iat = 0; iat < ucell.nat; iat++) + { + GlobalV::ofs_running << " " << std::setw(8) << iat; + for (int i = 0; i < 3; i++) + { + if (std::abs(fcs(iat, i) * ModuleBase::Ry_to_eV / ModuleBase::BOHR_TO_A) < force_threshold) + { + fcs(iat, i) = 0.0; + GlobalV::ofs_running << std::setw(5) << "1"; + } + else + { + GlobalV::ofs_running << std::setw(5) << "0"; + } + } + GlobalV::ofs_running << std::endl; + } +} +} // namespace + +void assemble_print_force(const UnitCell& ucell, + const bool istestf, + const vdw::VdwResult* vdw_result, + const Exx_Info& exx_info, + ModuleSymmetry::Symmetry* symm, + const std::string& dpks_out_type, + const LCAOForceParts& parts, + const double force_threshold, + ModuleBase::matrix& fcs) +{ + const int nat = ucell.nat; + //--------------------------------- + // sum all parts of force! + //--------------------------------- + ModuleBase::Vector3 net_force = {0.0, 0.0, 0.0}; + for (int i = 0; i < 3; i++) + { + for (int iat = 0; iat < nat; iat++) + { + sum_force_terms(iat, i, vdw_result, exx_info, parts, fcs); + } + } + + if (PARAM.inp.gate_flag || PARAM.inp.efield_flag) + { + GlobalV::ofs_running << "Atomic forces are not shifted if gate_flag or efield_flag == true!" << std::endl; + } + + // pengfei 2016-12-20 + if (ModuleSymmetry::Symmetry::symm_flag == 1) + { + ModuleSymmetry::symmetrize_force_cartesian(symm, nat, ucell.a1, ucell.a2, ucell.a3, fcs); + } + + // The net force should be evaluated AFTER the symmetrization. + // With symmetry switched on, the forces assembled above are built from IBZ-reduced + // quantities and only become physical after the symmetrization, forceSymmetry(). + // Force symmetrization is linear, so it commutes with the removal of a + // uniform shift: the resulting fcs is identical to the previous ordering. + // Net force is evaluated after symmetrization and before the uniform shift. + for (int i = 0; i < 3; i++) + { + double sum = 0.0; + for (int iat = 0; iat < nat; iat++) + { + sum += fcs(iat, i); + } + net_force[i] = sum; + } + if (!(PARAM.inp.gate_flag || PARAM.inp.efield_flag)) + { + ModuleBase::remove_net_force(nat, fcs); + } + + // compute forces using the DeePKS model + DeePKS_domain::write_forces(fcs, parts.fvnl_dalpha, dpks_out_type, PARAM.inp); + + if (istestf) + { + print_force_parts(ucell, vdw_result, parts); + } + + GlobalV::ofs_running << std::setiosflags(std::ios::left); + + // this->printforce_total(ry, istestf, fcs); + ModuleIO::print_force(GlobalV::ofs_running, ucell, "TOTAL-FORCE (eV/Angstrom)", fcs, false); + net_force*= ModuleBase::Ry_to_eV / ModuleBase::BOHR_TO_A; + ModuleBase::GlobalFunc::OUT(GlobalV::ofs_running, "Net force vector (eV/Ang)", net_force.x, net_force.y, net_force.z); + ModuleBase::GlobalFunc::OUT(GlobalV::ofs_running, "Total drift (ev/Ang)", net_force.norm()); + if (istestf) + { + print_force_invalid_table(ucell, force_threshold, fcs); + } +} + +void assemble_print_stress(const UnitCell& ucell, + const bool istests, + const vdw::VdwResult* vdw_result, + const Exx_Info& exx_info, + ModuleSymmetry::Symmetry* symm, + const std::string& dpks_out_type, + const LCAOStressParts& sparts, + ModuleBase::matrix& scs) +{ + for (int i = 0; i < 3; i++) + { + for (int j = 0; j < 3; j++) + { + sum_stress_terms(i, j, vdw_result, exx_info, sparts, scs); + } + } + if (ModuleSymmetry::Symmetry::symm_flag == 1) + { + symm->symmetrize_mat3(scs, ucell.lat); + } // end symmetry + + DeePKS_domain::write_stress(scs, sparts.svnl_dalpha, ucell.omega, dpks_out_type, PARAM.inp); + + // print Rydberg stress or not + bool ry = false; + + // test stress each terms if needed + if (istests) + { + // test + ModuleBase::matrix svlocal; + svlocal.create(3, 3); + ModuleBase::matrix stvnl; + stvnl.create(3, 3); + for (int i = 0; i < 3; i++) + { + for (int j = 0; j < 3; j++) + { + svlocal(i, j) = sparts.svl_dphi(i, j) + sparts.sigmadvl(i, j); + stvnl(i, j) = sparts.stvnl_dphi(i, j) + sparts.svnl_dbeta(i, j); + } + } + + const bool screen = PARAM.inp.test_stress; + + GlobalV::ofs_running << "\n PARTS OF STRESS: " << std::endl; + GlobalV::ofs_running << std::setiosflags(std::ios::showpos); + GlobalV::ofs_running << std::setiosflags(std::ios::fixed) << std::setprecision(10) << std::endl; + ModuleIO::print_stress("OVERLAP STRESS", sparts.soverlap, screen, ry, GlobalV::ofs_running); + ModuleIO::print_stress("T STRESS", sparts.stvnl_dphi, screen, ry, GlobalV::ofs_running); + ModuleIO::print_stress("VNL STRESS", sparts.svnl_dbeta, screen, ry, GlobalV::ofs_running); + ModuleIO::print_stress("T_VNL STRESS", stvnl, screen, ry, GlobalV::ofs_running); + ModuleIO::print_stress("VL_dPHI STRESS", sparts.svl_dphi, screen, ry, GlobalV::ofs_running); + ModuleIO::print_stress("VL_dVL STRESS", sparts.sigmadvl, screen, ry, GlobalV::ofs_running); + ModuleIO::print_stress("HAR STRESS", sparts.sigmahar, screen, ry, GlobalV::ofs_running); + ModuleIO::print_stress("EWALD STRESS", sparts.sigmaewa, screen, ry, GlobalV::ofs_running); + ModuleIO::print_stress("cc STRESS", sparts.sigmacc, screen, ry, GlobalV::ofs_running); + ModuleIO::print_stress("XC STRESS", sparts.sigmaxc, screen, ry, GlobalV::ofs_running); + if (vdw_result != nullptr) + { + ModuleIO::print_stress("VDW STRESS", sparts.stress_vdw, screen, ry, GlobalV::ofs_running); + } + if (PARAM.inp.dft_plus_u) + { + ModuleIO::print_stress("DFTU STRESS", sparts.stress_u, screen, ry, GlobalV::ofs_running); + } + if (PARAM.inp.sc_mag_switch) + { + ModuleIO::print_stress("DeltaSpin STRESS", sparts.stress_dspin, screen, ry, GlobalV::ofs_running); + } +#ifdef __EXX + if (exx_info.info_global.cal_exx) + { + ModuleIO::print_stress("EXX STRESS", sparts.stress_exx, screen, ry, GlobalV::ofs_running); + } +#endif + ModuleIO::print_stress("TOTAL STRESS", scs, screen, ry, GlobalV::ofs_running); + } // end of test + + GlobalV::ofs_running << std::setiosflags(std::ios::left); + + // print total stress + bool screen_normal = true; + ModuleIO::print_stress("TOTAL-STRESS", scs, screen_normal, ry, GlobalV::ofs_running); + + double unit_transform = 0.0; + unit_transform = ModuleBase::RYDBERG_SI / pow(ModuleBase::BOHR_RADIUS_SI, 3) * 1.0e-8; + double external_stress[3] = {PARAM.inp.press1, PARAM.inp.press2, PARAM.inp.press3}; + + for (int i = 0; i < 3; i++) + { + scs(i, i) -= external_stress[i] / unit_transform; + } +} + +} // namespace LCAO_domain diff --git a/source/source_lcao/force_stress_assemble.h b/source/source_lcao/force_stress_assemble.h new file mode 100644 index 00000000000..75ebf34c8b4 --- /dev/null +++ b/source/source_lcao/force_stress_assemble.h @@ -0,0 +1,53 @@ +#ifndef FORCE_STRESS_ASSEMBLE_H +#define FORCE_STRESS_ASSEMBLE_H + +#include "force_stress_lcao.h" + +#include // std::string for dpks_out_type + +// Free functions that assemble the per-term LCAO force/stress parts into the +// total force/stress matrices and print the breakdown. The DeePKS label output +// only needs the output-type selector dpks_out_type, so these functions are +// independent of the electronic template type T. + +class UnitCell; +namespace ModuleSymmetry +{ +class Symmetry; +} +namespace vdw +{ +struct VdwResult; +} + +namespace LCAO_domain +{ + +// Sum the computed force parts into fcs, apply symmetry and the net-force +// (drift) correction, then print the per-term and total forces. +// force_threshold is Force_Stress_LCAO::force_invalid_threshold_ev, passed +// in explicitly so this function need not be a class member. +void assemble_print_force(const UnitCell& ucell, + const bool istestf, + const vdw::VdwResult* vdw_result, + const Exx_Info& exx_info, + ModuleSymmetry::Symmetry* symm, + const std::string& dpks_out_type, + const LCAOForceParts& parts, + const double force_threshold, + ModuleBase::matrix& fcs); + +// Sum the computed stress parts into scs, symmetrize, subtract the external +// pressure and print the per-term and total stresses. +void assemble_print_stress(const UnitCell& ucell, + const bool istests, + const vdw::VdwResult* vdw_result, + const Exx_Info& exx_info, + ModuleSymmetry::Symmetry* symm, + const std::string& dpks_out_type, + const LCAOStressParts& sparts, + ModuleBase::matrix& scs); + +} // namespace LCAO_domain + +#endif diff --git a/source/source_lcao/force_stress_lcao.cpp b/source/source_lcao/force_stress_lcao.cpp index 56fc5b009b8..aab0dc61166 100644 --- a/source/source_lcao/force_stress_lcao.cpp +++ b/source/source_lcao/force_stress_lcao.cpp @@ -1,5 +1,9 @@ #include "force_stress_lcao.h" +#include "force_stress_assemble.h" +#include "force_stress_terms.h" + +#include "source_basis/module_nao/two_center_bundle.h" #include "source_base/parallel_reduce.h" #include "source_pw/module_pwdft/dftu_base.h" //Quxin add for DFT+U on 20201029 #include "source_lcao/module_dftu/dftu_nao_fs_k.h" @@ -15,7 +19,6 @@ #include "source_estate/module_pot/gatefield.h" // liuyu add 2022-09-13 #include "source_hamilt/module_surchem/surchem.h" //sunml add 2022-08-10 #include "source_hamilt/module_vdw/vdw.h" -#include "source_io/module_parameter/parameter.h" #ifdef __MLALGO #include "source_lcao/module_deepks/lcao_deepks.h" //caoyu add for deepks 2021-06-03 #include "source_lcao/module_deepks/lcao_deepks_io.h" // mohan add 2024-07-22 @@ -35,24 +38,22 @@ // mohan add 2025-11-04 template <> void assign_dmk_ptr( - elecstate::DensityMatrix* dm, + module_dm::DensityMatrix* dm, std::vector>*& dmk_d, - std::vector>>*& dmk_c, - bool gamma_only_local + std::vector>>*& dmk_c ) { - auto& dmk_tmp = dm->get_DMK_vector(); + std::vector>& dmk_tmp = dm->get_dmk_vec(); dmk_d = &dmk_tmp; dmk_c = nullptr; } template <> void assign_dmk_ptr>( - elecstate::DensityMatrix,double>* dm, + module_dm::DensityMatrix,double>* dm, std::vector>*& dmk_d, - std::vector>>*& dmk_c, - bool gamma_only_local + std::vector>>*& dmk_c ) { - auto& dmk_tmp = dm->get_DMK_vector(); + std::vector>>& dmk_tmp = dm->get_dmk_vec(); dmk_c = &dmk_tmp; dmk_d = nullptr; } @@ -77,7 +78,7 @@ void Force_Stress_LCAO::getForceStress(UnitCell& ucell, const Grid_Driver& gd, Parallel_Orbitals& pv, const elecstate::ElecState* pelec, - LCAO_domain::Setup_DM &dmat, // mohan add 2025-11-03 + module_dm::Setup_DM &dmat, // mohan add 2025-11-03 const psi::Psi* psi, const TwoCenterBundle& two_center_bundle, const LCAO_Orbitals& orb, @@ -93,6 +94,7 @@ void Force_Stress_LCAO::getForceStress(UnitCell& ucell, Exx_NAO &exx_nao, ModuleSymmetry::Symmetry* symm, const Exx_Info& exx_info, + const FSCalcConfig& cfg, const int td_stype, hamilt::Hamilt* p_hamilt) { @@ -112,98 +114,159 @@ void Force_Stress_LCAO::getForceStress(UnitCell& ucell, // total force : ModuleBase::matrix fcs; - // part of total force - ModuleBase::matrix foverlap; - ModuleBase::matrix ftvnl_dphi; - ModuleBase::matrix fvnl_dbeta; - ModuleBase::matrix fvl_dphi; - ModuleBase::matrix fvl_dvl; - ModuleBase::matrix fewalds; - ModuleBase::matrix fcc; - ModuleBase::matrix fscc; - ModuleBase::matrix fvnl_dalpha; // deepks - ModuleBase::matrix fpothybrid; + // part of total force / stress, grouped so the assembly helpers can take a + // single container reference. + LCAOForceParts parts; + LCAOStressParts sparts; - fvl_dphi.create(nat, 3); // must do it now, update it later, noted by zhengdy + parts.fvl_dphi.create(nat, 3); // must do it now, update it later, noted by zhengdy if (isforce) { fcs.create(nat, 3); - foverlap.create(nat, 3); // overlap force - ftvnl_dphi.create(nat, 3); // pulay force of NAO - fvnl_dbeta.create(nat, 3); // pulay force of non-local projectors - fvl_dvl.create(nat, 3); // force from local potentials - fewalds.create(nat, 3); // Ewald force - fcc.create(nat, 3); // force due to core correction - fscc.create(nat, 3); // force due to self-consistent field - fvnl_dalpha.create(nat, 3); // deepks - fpothybrid.create(nat, 3); // pulay force for hybrid gauge rt-tddft + parts.foverlap.create(nat, 3); // overlap force + parts.ftvnl_dphi.create(nat, 3); // pulay force of NAO + parts.fvnl_dbeta.create(nat, 3); // pulay force of non-local projectors + parts.fvl_dvl.create(nat, 3); // force from local potentials + parts.fewalds.create(nat, 3); // Ewald force + parts.fcc.create(nat, 3); // force due to core correction + parts.fscc.create(nat, 3); // force due to self-consistent field + parts.fvnl_dalpha.create(nat, 3); // deepks + parts.fpothybrid.create(nat, 3); // pulay force for hybrid gauge rt-tddft // calculate basic terms in Force, same method with PW base - this->calForcePwPart(ucell, fvl_dvl, fewalds, fcc, fscc, pelec->f_en.etxc, - pelec->vnew, pelec->vnew_exist, pelec->charge, rhopw, locpp, sf); + this->calForcePwPart(ucell, parts.fvl_dvl, parts.fewalds, parts.fcc, parts.fscc, + pelec->f_en.etxc, pelec->vnew, pelec->vnew_exist, pelec->charge, rhopw, + locpp, sf, cfg); } // total stress : ModuleBase::matrix scs - ModuleBase::matrix sigmacc; - ModuleBase::matrix sigmadvl; - ModuleBase::matrix sigmaewa; - ModuleBase::matrix sigmaxc; - ModuleBase::matrix sigmahar; - ModuleBase::matrix soverlap; - ModuleBase::matrix stvnl_dphi; - ModuleBase::matrix svnl_dbeta; - ModuleBase::matrix svl_dphi; - ModuleBase::matrix svnl_dalpha; // deepks //! stress if (isstress) { scs.create(3, 3); - sigmacc.create(3, 3); - sigmadvl.create(3, 3); - sigmaewa.create(3, 3); - sigmaxc.create(3, 3); - sigmahar.create(3, 3); - - soverlap.create(3, 3); - stvnl_dphi.create(3, 3); - svnl_dbeta.create(3, 3); - svl_dphi.create(3, 3); - svnl_dalpha.create(3, 3); + sparts.sigmacc.create(3, 3); + sparts.sigmadvl.create(3, 3); + sparts.sigmaewa.create(3, 3); + sparts.sigmaxc.create(3, 3); + sparts.sigmahar.create(3, 3); + + sparts.soverlap.create(3, 3); + sparts.stvnl_dphi.create(3, 3); + sparts.svnl_dbeta.create(3, 3); + sparts.svl_dphi.create(3, 3); + sparts.svnl_dalpha.create(3, 3); // calculate basic terms in Stress, similar method with PW base - this->calStressPwPart(ucell, sigmadvl, sigmahar, sigmaewa, sigmacc, - sigmaxc, pelec->f_en.etxc, pelec->charge, rhopw, locpp, sf); + this->sc_pw.stress_pw_terms(ucell, sparts.sigmadvl, sparts.sigmahar, sparts.sigmaewa, + sparts.sigmacc, sparts.sigmaxc, pelec->f_en.etxc, pelec->charge, + rhopw, locpp, sf, cfg.nspin, cfg.domag, cfg.domag_z, cfg.gga_grad, cfg.gamma_only_pw); + } + // Calculate operator-based force/stress terms (kinetic, overlap, + // nonlocal, rt-TDDFT hybrid gauge, local Pulay term and DeltaSpin). + this->cal_operator_fs(ucell, gd, pv, pelec, dmat, psi, two_center_bundle, + orb, kv, isforce, isstress, cfg, td_stype, p_hamilt, parts, sparts); + + // MPI reduction for forces + if (isforce) + { + Parallel_Reduce::reduce_pool(parts.fvl_dphi.c, parts.fvl_dphi.nr * parts.fvl_dphi.nc); } + // MPI reduction for stresses + if (isstress) + { + Parallel_Reduce::reduce_pool(sparts.svl_dphi.c, sparts.svl_dphi.nr * sparts.svl_dphi.nc); + } + + // Handle DeePKS forces if enabled + LCAO_domain::cal_deepks_fs(ucell, gd, pv, orb, kv, isforce, isstress, deepks, parts, sparts); + + // vdW force/stress and external-field forces + LCAO_domain::cal_vdw_fields_fs(vdw_result, ucell, solvent, rhopw, locpp, + isforce, isstress, parts, sparts); + + // DFT+U force/stress + LCAO_domain::cal_dftu_fs(ucell, gd, pv, orb, kv, dmat, two_center_bundle, dftu, isforce, isstress, parts, sparts); + + + // NOTE: finish_ftable is no longer needed as we don't use ForceStressArrays for overlap/kinetic + // if (!PARAM.globalv.gamma_only_local) + // { + // this->edm_cal.finish_ftable(fsr); + // } + + // EXX force/stress + LCAO_domain::cal_exx_fs(ucell, isforce, isstress, exx_info, exx_nao, parts, sparts); + //-------------------------------- + // begin calculate and output force + //-------------------------------- + if (isforce) + { + LCAO_domain::assemble_print_force(ucell, istestf, vdw_result, exx_info, symm, deepks.dpks_out_type, parts, + force_invalid_threshold_ev, fcs); + } // end of force calculation + //--------------------------------- + // begin calculate and output stress + //--------------------------------- + if (isstress) + { + LCAO_domain::assemble_print_stress(ucell, istests, vdw_result, exx_info, symm, deepks.dpks_out_type, sparts, scs); + } // end of stress calculation + + ModuleBase::timer::end("Force_Stress_LCAO", "getForceStress"); + return; +} + +// Operator-based force/stress terms: kinetic, overlap, nonlocal, +// rt-TDDFT hybrid gauge, local-potential Pulay term, and DeltaSpin. +template +void Force_Stress_LCAO::cal_operator_fs(UnitCell& ucell, + const Grid_Driver& gd, + Parallel_Orbitals& pv, + const elecstate::ElecState* pelec, + module_dm::Setup_DM& dmat, + const psi::Psi* psi, + const TwoCenterBundle& two_center_bundle, + const LCAO_Orbitals& orb, + const K_Vectors& kv, + const bool isforce, + const bool isstress, + const FSCalcConfig& cfg, + const int td_stype, + hamilt::Hamilt* p_hamilt, + LCAOForceParts& parts, + LCAOStressParts& sparts) +{ + // Calculate forces and stresses using new operator-based methods // Step 1: Calculate Energy Density Matrix (EDM) for overlap force // EDM = Σ_k w_k * ε_k * |ψ_k><ψ_k| - elecstate::DensityMatrix edm = flk.cal_edm(pelec, *psi, *dmat.dm, kv, pv, - PARAM.inp.nspin, PARAM.inp.nbands, ucell, *this->RA); + module_dm::DensityMatrix edm = edm_cal.cal_edm(pelec, *psi, *dmat.dm, kv, pv, + cfg.nspin, cfg.nbands, ucell, *this->RA); // Step 2: Handle different spin cases - if (PARAM.inp.nspin == 1 || PARAM.inp.nspin == 2) + if (cfg.nspin == 1 || cfg.nspin == 2) { // For nspin=1 or nspin=2, use double precision // Switch to spin channel 1 for DMR access - if (PARAM.inp.nspin == 2) + if (cfg.nspin == 2) { dmat.dm->switch_dmr(1); edm.switch_dmr(1); } - const hamilt::HContainer* dmR = dmat.dm->get_DMR_pointer(1); - const hamilt::HContainer* edmR = edm.get_DMR_pointer(1); + const hamilt::HContainer* dmR = dmat.dm->get_dmr_ptr(1); + const hamilt::HContainer* edmR = edm.get_dmr_ptr(1); // Calculate kinetic force/stress (uses DM) - if (PARAM.inp.t_in_h) + if (cfg.t_in_h) { hamilt::EKinetic> tmp_ekinetic( nullptr, kv.kvec_d, nullptr, &ucell, orb.cutoffs(), &gd, two_center_bundle.kinetic_orb.get()); - tmp_ekinetic.cal_force_stress(isforce, isstress, dmR, ftvnl_dphi, stvnl_dphi); + tmp_ekinetic.cal_force_stress(isforce, isstress, dmR, parts.ftvnl_dphi, sparts.stvnl_dphi); } // Calculate overlap force/stress (uses EDM) @@ -212,26 +275,26 @@ void Force_Stress_LCAO::getForceStress(UnitCell& ucell, two_center_bundle.overlap_orb.get()); if(td_stype != 2) { - tmp_overlap.cal_force_stress(isforce, isstress, edmR, foverlap, soverlap); + tmp_overlap.cal_force_stress(isforce, isstress, edmR, parts.foverlap, sparts.soverlap); } // Calculate nonlocal force/stress (uses DM) hamilt::Nonlocal> tmp_nonlocal( nullptr, kv.kvec_d, nullptr, &ucell, orb.cutoffs(), &gd, two_center_bundle.overlap_orb_beta.get()); - tmp_nonlocal.cal_force_stress(isforce, isstress, dmR, fvnl_dbeta, svnl_dbeta); + tmp_nonlocal.cal_force_stress(isforce, isstress, dmR, parts.fvnl_dbeta, sparts.svnl_dbeta); if(td_stype == 2) { hamilt::TD_pot_hybrid> tmp_hybrid( nullptr, &kv, nullptr, nullptr, orb, &ucell, orb.cutoffs(), &gd, nullptr); - tmp_hybrid.cal_force_stress(isforce, dmR, fpothybrid); + tmp_hybrid.cal_force_stress(isforce, dmR, parts.fpothybrid); - cal_foverlap_rt(foverlap, dmat, p_hamilt, kv, pv, ucell); + cal_foverlap_rt(parts.foverlap, dmat, p_hamilt, kv, pv, ucell); } // Switch back to spin channel 0 - if (PARAM.inp.nspin == 2) + if (cfg.nspin == 2) { dmat.dm->switch_dmr(0); edm.switch_dmr(0); @@ -239,269 +302,58 @@ void Force_Stress_LCAO::getForceStress(UnitCell& ucell, // Calculate local potential force/stress (vl_dphi) // This uses grid integration, not operator-based method - flk.ParaV = dmat.dm->get_paraV_pointer(); - PulayForceStress::cal_pulay_fs(fvl_dphi, svl_dphi, *dmat.dm, ucell, pelec->pot, + edm_cal.ParaV = &pv; + PulayForceStress::cal_pulay_fs(parts.fvl_dphi, sparts.svl_dphi, *dmat.dm, ucell, pelec->pot, isforce, isstress, false /*reset dm to gint*/); } - else if (PARAM.inp.nspin == 4) + else if (cfg.nspin == 4) { - // Calculate kinetic force/stress (uses DM) - if (PARAM.inp.t_in_h) + // Kinetic force/stress from the complex DMR (nspin=4) + if (cfg.t_in_h) { hamilt::EKinetic, std::complex>> tmp_ekinetic( nullptr, kv.kvec_d, nullptr, &ucell, orb.cutoffs(), &gd, two_center_bundle.kinetic_orb.get()); - tmp_ekinetic.cal_force_stress(isforce, isstress, dmat.dm->get_DMR_pointer(1), ftvnl_dphi, stvnl_dphi); + tmp_ekinetic.cal_force_stress(isforce, isstress, dmat.dm->get_dmr_ptr(1), parts.ftvnl_dphi, + sparts.stvnl_dphi); } - // Calculate overlap force/stress (uses EDM) + // Overlap force/stress from the complex EDM (nspin=4) hamilt::Overlap, std::complex>> tmp_overlap( nullptr, kv.kvec_d, nullptr, nullptr, &ucell, orb.cutoffs(), &gd, two_center_bundle.overlap_orb.get()); - tmp_overlap.cal_force_stress(isforce, isstress, edm.get_DMR_pointer(1), foverlap, soverlap); + tmp_overlap.cal_force_stress(isforce, isstress, edm.get_dmr_ptr(1), parts.foverlap, sparts.soverlap); // For nspin=4 (non-collinear), need complex DMR // Create temporary complex DMR for DM - hamilt::HContainer> tmp_dmr(dmat.dm->get_DMR_pointer(1)->get_paraV()); - std::vector ijrs = dmat.dm->get_DMR_pointer(1)->get_ijr_info(); + hamilt::HContainer> tmp_dmr(dmat.dm->get_dmr_ptr(1)->get_paraV()); + std::vector ijrs = dmat.dm->get_dmr_ptr(1)->get_ijr_info(); tmp_dmr.insert_ijrs(&ijrs); tmp_dmr.allocate(); - dmat.dm->cal_DMR_full(&tmp_dmr); - // Calculate nonlocal force/stress (uses DM) + dmat.dm->cal_dmr_full(&tmp_dmr, -1); + // Nonlocal force/stress from the temporary complex DMR hamilt::Nonlocal, std::complex>> tmp_nonlocal( nullptr, kv.kvec_d, nullptr, &ucell, orb.cutoffs(), &gd, two_center_bundle.overlap_orb_beta.get()); - tmp_nonlocal.cal_force_stress(isforce, isstress, &tmp_dmr, fvnl_dbeta, svnl_dbeta); - - // Calculate local potential force/stress (vl_dphi) - flk.ParaV = dmat.dm->get_paraV_pointer(); - PulayForceStress::cal_pulay_fs(fvl_dphi, svl_dphi, *dmat.dm, ucell, pelec->pot, - isforce, isstress, false /*reset dm to gint*/); - } - - // MPI reduction for forces - if (isforce) - { - Parallel_Reduce::reduce_pool(fvl_dphi.c, fvl_dphi.nr * fvl_dphi.nc); - } - - // MPI reduction for stresses - if (isstress) - { - Parallel_Reduce::reduce_pool(svl_dphi.c, svl_dphi.nr * svl_dphi.nc); - } + tmp_nonlocal.cal_force_stress(isforce, isstress, &tmp_dmr, parts.fvnl_dbeta, sparts.svnl_dbeta); - // Handle DeePKS forces if enabled -#ifdef __MLALGO - if (PARAM.inp.deepks_scf) - { - const int nks = (PARAM.inp.nspin == 1 || PARAM.inp.nspin == 2) ? 1 : kv.get_nks(); - if (PARAM.globalv.gamma_only_local) - { - DeePKS_domain::cal_f_delta( - ucell, - orb, - gd, - *flk.ParaV, - nks, - deepks.ld.deepks_param, - kv.kvec_d, - deepks.ld.phialpha, - fvnl_dalpha, - isstress, - svnl_dalpha, - deepks.ld.dm_r, - deepks.ld.gedm, - (PARAM.inp.nspin == 2 && !PARAM.inp.deepks_equiv) ? deepks.ld.dm_r_mag : nullptr, - (PARAM.inp.nspin == 2 && !PARAM.inp.deepks_equiv) ? deepks.ld.gedm_mag : nullptr); - } - else - { - DeePKS_domain::cal_f_delta>( - ucell, - orb, - gd, - *flk.ParaV, - nks, - deepks.ld.deepks_param, - kv.kvec_d, - deepks.ld.phialpha, - fvnl_dalpha, - isstress, - svnl_dalpha, - deepks.ld.dm_r, - deepks.ld.gedm, - (PARAM.inp.nspin == 2 && !PARAM.inp.deepks_equiv) ? deepks.ld.dm_r_mag : nullptr, - (PARAM.inp.nspin == 2 && !PARAM.inp.deepks_equiv) ? deepks.ld.gedm_mag : nullptr); - } - - if (isforce) - { - Parallel_Reduce::reduce_pool(fvnl_dalpha.c, fvnl_dalpha.nr * fvnl_dalpha.nc); - } - if (isstress) - { - Parallel_Reduce::reduce_pool(svnl_dalpha.c, svnl_dalpha.nr * svnl_dalpha.nc); - } - } -#endif - - //! forces and stress from vdw - // Peize Lin add 2014-04-04, update 2021-03-09 - // jiyy add 2019-05-18, update 2021-05-02 - ModuleBase::matrix force_vdw; - ModuleBase::matrix stress_vdw; - if (vdw_result != nullptr) - { - if (isforce) - { - if (!vdw_result->has_force || vdw_result->force.size() != static_cast(nat)) - { - ModuleBase::WARNING_QUIT("Force_Stress_LCAO::getForceStress", - "The cached vdW force is unavailable or has an invalid size."); - } - force_vdw.create(nat, 3); - for (int iat = 0; iat < nat; ++iat) - { - force_vdw(iat, 0) = vdw_result->force[iat].x; - force_vdw(iat, 1) = vdw_result->force[iat].y; - force_vdw(iat, 2) = vdw_result->force[iat].z; - } - } - if (isstress) - { - if (!vdw_result->has_stress) - { - ModuleBase::WARNING_QUIT("Force_Stress_LCAO::getForceStress", - "The cached vdW stress is unavailable."); - } - stress_vdw = vdw_result->stress.to_matrix(); - } - } - - //! forces from E-field - ModuleBase::matrix fefield; - if (PARAM.inp.efield_flag && isforce) - { - fefield.create(nat, 3); - elecstate::Efield::compute_force(ucell, fefield); - } - - //! atomic forces from E-field of rt-TDDFT - ModuleBase::matrix fefield_tddft; - if (PARAM.inp.esolver_type == "tddft" && isforce) - { - fefield_tddft.create(nat, 3); - elecstate::H_TDDFT_pw::compute_force(ucell, fefield_tddft); - } - - //! atomic forces from gate field - ModuleBase::matrix fgate; - if (PARAM.inp.gate_flag && isforce) - { - fgate.create(nat, 3); - elecstate::Gatefield::compute_force(ucell, fgate); - } - - //! atomic forces from implicit solvation model - ModuleBase::matrix fsol; - if (PARAM.inp.imp_sol && isforce) - { - fsol.create(nat, 3); - solvent.cal_force_sol(ucell, rhopw, locpp.vloc, PARAM.inp.nspin, fsol); - } - - //! atomic forces from DFT+U (Quxin version) - ModuleBase::matrix force_u; - ModuleBase::matrix stress_u; - - if (PARAM.inp.dft_plus_u) // Quxin add for DFT+U on 20201029 - { - if (isforce) - { - force_u.create(nat, 3); - } - if (isstress) - { - stress_u.create(3, 3); - } - if (PARAM.inp.dft_plus_u == 2) - { - // The legacy dft_plus_u==2 force/stress path is currently broken. - // - // Background: Plus_U::force_stress relies on ForceStressArrays - // members DSloc_x/y/z (gamma_only) or DSloc_Rx/Ry/Rz (multik) - // and DH_r being pre-allocated and filled with dS/dR data by the - // main force flow (formerly ForceLcaoGamma::ftable). The DFT+U - // step 2 refactor (commit 70c54c9d5a, 2026-01-23) removed the - // main-flow ForceStressArrays because the operator-based force - // calculation no longer needs it, but the legacy dft_plus_u==2 - // path still depends on it. The local fsr_dftu below is declared - // without allocating those arrays, so any call into - // cal_force_gamma / cal_stress_gamma / folding_matrix_k would - // pass nullptr to pdgemm_ and crash with SIGSEGV. - // - // Until the legacy path is restored or re-implemented, we - // explicitly reject dft_plus_u==2 with cal_force or cal_stress - // enabled. SCF-only runs (no force/stress) are unaffected - // because the energy is computed in cal_energy_correction, - // which does not touch DSloc arrays. Use dft_plus_u=1 for - // force/stress calculations. - if (isforce || isstress) - { - ModuleBase::WARNING_QUIT("Force_Stress_LCAO::getForceStress", - "dft_plus_u==2 with cal_force or cal_stress is currently broken; " - "please use dft_plus_u=1 instead. See notes in source/source_lcao/force_stress_lcao.cpp."); - } - ForceStressArrays fsr_dftu; - std::vector>* dmk_d = nullptr; - std::vector>>* dmk_c = nullptr; - assign_dmk_ptr(dmat.dm, dmk_d, dmk_c, PARAM.globalv.gamma_only_local); - DFTU_LCAO::DftuFsEnv dftu_fs_env(dftu, ucell, gd, pv, fsr_dftu, - orb.cutoffs(), PARAM.inp.ks_solver); - DFTU_LCAO::force_stress(dftu_fs_env, isforce, isstress, - dmk_d, dmk_c, force_u, stress_u, kv, - PARAM.globalv.gamma_only_local); - } - else - { - // Build DFT+U force/stress inputs directly without constructing a - // full DFTU operator (hsk/hR are irrelevant for this path). - auto adjs_all = DFTU_LCAO::build_adjacent_atoms( - &ucell, &dftu, &gd, orb.cutoffs(), PARAM.inp.onsite_radius); - - // The DensityMatrix holds nspin_dm = (nspin==2 ? 2 : 1) real-space DMR - // channels: nspin=4 (non-collinear) packs all four Pauli components - // into a single complex DMR, so only one channel exists (cf. setup_dm.cpp - // and the is0 = nspin==2 ? is : 0 indexing in cal_for/str_IJR_nao_r). - const int nspin_dm = (PARAM.inp.nspin == 2) ? 2 : 1; - std::vector*> dmR_tmp(nspin_dm, nullptr); - for (int is = 0; is < nspin_dm; ++is) - { - dmR_tmp[is] = dmat.dm->get_DMR_pointer(is + 1); - } - - DFTU_LCAO::cal_fs_nao_r(&ucell, &dftu, - two_center_bundle.overlap_orb_onsite.get(), - PARAM.inp.nspin, - adjs_all, dmR_tmp, - isforce, isstress, force_u, stress_u); - } + // Local-potential (vl_dphi) Pulay term via grid integration + edm_cal.ParaV = &pv; + PulayForceStress::cal_pulay_fs(parts.fvl_dphi, sparts.svl_dphi, *dmat.dm, ucell, pelec->pot, + isforce, isstress, false); } // atomic force and stress for DeltaSpin - ModuleBase::matrix force_dspin; - ModuleBase::matrix stress_dspin; - if (PARAM.inp.sc_mag_switch) + if (cfg.sc_mag_switch) { if (isforce) { - force_dspin.create(nat, 3); + parts.force_dspin.create(ucell.nat, 3); } if (isstress) { - stress_dspin.create(3, 3); + sparts.stress_dspin.create(3, 3); } hamilt::DeltaSpin> tmp_dspin(nullptr, @@ -512,415 +364,21 @@ void Force_Stress_LCAO::getForceStress(UnitCell& ucell, two_center_bundle.overlap_orb_onsite.get(), orb.cutoffs()); - if (PARAM.inp.nspin == 2) + if (cfg.nspin == 2) { dmat.dm->switch_dmr(2); } - const hamilt::HContainer* dmr = dmat.dm->get_DMR_pointer(1); - tmp_dspin.cal_force_stress(isforce, isstress, dmr, force_dspin, stress_dspin); - if (PARAM.inp.nspin == 2) + const hamilt::HContainer* dmr = dmat.dm->get_dmr_ptr(1); + tmp_dspin.cal_force_stress(isforce, isstress, dmr, parts.force_dspin, sparts.stress_dspin); + if (cfg.nspin == 2) { dmat.dm->switch_dmr(0); } } +} - // NOTE: finish_ftable is no longer needed as we don't use ForceStressArrays for overlap/kinetic - // if (!PARAM.globalv.gamma_only_local) - // { - // this->flk.finish_ftable(fsr); - // } - -#ifdef __EXX - bool cal_exx = exx_info.info_global.cal_exx; - bool real_number = exx_info.info_ri.real_number; - double hybrid_alpha = exx_info.info_global.hybrid_alpha; - - ModuleBase::matrix force_exx; - ModuleBase::matrix stress_exx; - if (cal_exx) - { - if (isforce) - { - if (real_number) - { - exx_nao.exd->cal_exx_force(ucell.nat); - force_exx = hybrid_alpha * exx_nao.exd->get_force(); - } - else - { - exx_nao.exc->cal_exx_force(ucell.nat); - force_exx = hybrid_alpha * exx_nao.exc->get_force(); - } - } - if (isstress) - { - if (real_number) - { - exx_nao.exd->cal_exx_stress(ucell.omega, ucell.lat0); - stress_exx = hybrid_alpha * exx_nao.exd->get_stress(); - } - else - { - exx_nao.exc->cal_exx_stress(ucell.omega, ucell.lat0); - stress_exx = hybrid_alpha * exx_nao.exc->get_stress(); - } - } - } -#endif - //-------------------------------- - // begin calculate and output force - //-------------------------------- - if (isforce) - { - //--------------------------------- - // sum all parts of force! - //--------------------------------- - ModuleBase::Vector3 net_force = {0.0, 0.0, 0.0}; - for (int i = 0; i < 3; i++) - { - for (int iat = 0; iat < nat; iat++) - { - fcs(iat, i) += foverlap(iat, i) + ftvnl_dphi(iat, i) + fvnl_dbeta(iat, i) + fvl_dphi(iat, i) - + fvl_dvl(iat, i) // derivative of local potential force (pw) - + fewalds(iat, i) // ewald force (pw) - + fcc(iat, i) // nonlinear core correction force (pw) - + fscc(iat, i) // self consistent corretion force (pw) - + fpothybrid(iat, i); // pulay force for hybrid gauge rt-tddft - - // Force contribution from DFT+U, Quxin add on 20201029 - if (PARAM.inp.dft_plus_u) - { - fcs(iat, i) += force_u(iat, i); - } - if (PARAM.inp.sc_mag_switch) - { - fcs(iat, i) += force_dspin(iat, i); - } -#ifdef __EXX - // Force contribution from exx - if (exx_info.info_global.cal_exx) - { - fcs(iat, i) += force_exx(iat, i); - } -#endif - // VDW force of vdwd2 or vdwd3 - if (vdw_result != nullptr) - { - fcs(iat, i) += force_vdw(iat, i); - } - // E-field force - if (PARAM.inp.efield_flag) - { - fcs(iat, i) += fefield(iat, i); - } - // E-field force of tddft - if (PARAM.inp.esolver_type == "tddft") - { - fcs(iat, i) += fefield_tddft(iat, i); - } - // Gate field force - if (PARAM.inp.gate_flag) - { - fcs(iat, i) += fgate(iat, i); - } - // implicit solvation model - if (PARAM.inp.imp_sol) - { - fcs(iat, i) += fsol(iat, i); - } -#ifdef __MLALGO - // mohan add 2021-08-04 - if (PARAM.inp.deepks_scf) - { - fcs(iat, i) += fvnl_dalpha(iat, i); - } -#endif - } - } - - if (PARAM.inp.gate_flag || PARAM.inp.efield_flag) - { - GlobalV::ofs_running << "Atomic forces are not shifted if gate_flag or efield_flag == true!" << std::endl; - } - - // pengfei 2016-12-20 - if (ModuleSymmetry::Symmetry::symm_flag == 1) - { - this->forceSymmetry(ucell, fcs, symm); - } - - // The net force should be evaluated AFTER the symmetrization. - // With symmetry switched on, the forces assembled above are built from IBZ-reduced - // quantities and only become physical after the symmetrization, forceSymmetry(). - // Force symmetrization is linear, so it commutes with the removal of a - // uniform shift: the resulting fcs is identical to the previous ordering. - for (int i = 0; i < 3; i++) - { - double sum = 0.0; - - for (int iat = 0; iat < nat; iat++) - { - // sum total force for correction - sum += fcs(iat, i); - } - net_force[i]=sum; - if (!(PARAM.inp.gate_flag || PARAM.inp.efield_flag)) - { - for (int iat = 0; iat < nat; ++iat) - { - fcs(iat, i) -= sum / nat; - } - } - } - - // compute forces using the DeePKS model - deepks.write_forces(fcs, fvnl_dalpha, PARAM.inp); - - // print Rydberg force or not - bool ry = false; - if (istestf) - { - // test - // ModuleBase::matrix fvlocal; - // fvlocal.create(nat,3); - ModuleBase::matrix ftvnl; - ftvnl.create(nat, 3); - for (int iat = 0; iat < nat; iat++) - { - for (int i = 0; i < 3; i++) - { - // fvlocal(iat,i) = fvl_dphi(iat,i) + fvl_dvl(iat,i); - ftvnl(iat, i) = ftvnl_dphi(iat, i) + fvnl_dbeta(iat, i); - } - } - - GlobalV::ofs_running << "\n PARTS OF FORCE: " << std::endl; - GlobalV::ofs_running << std::setiosflags(std::ios::showpos); - GlobalV::ofs_running << std::setiosflags(std::ios::fixed) << std::setprecision(8) << std::endl; - //----------------------------- - // regular force terms test. - //----------------------------- - // this->print_force("OVERLAP FORCE",foverlap,1,ry); - ModuleIO::print_force(GlobalV::ofs_running, ucell, "OVERLAP FORCE", foverlap, false); - ModuleIO::print_force(GlobalV::ofs_running, ucell, "TVNL_DPHI force",ftvnl_dphi,false); - ModuleIO::print_force(GlobalV::ofs_running, ucell, "VNL_DBETA force",fvnl_dbeta,false); - // this->print_force("T_VNL FORCE",ftvnl,1,ry); - ModuleIO::print_force(GlobalV::ofs_running, ucell, "T_VNL FORCE", ftvnl, false); - ModuleIO::print_force(GlobalV::ofs_running, ucell, "VL_dPHI FORCE", fvl_dphi, false); - // this->print_force("VL_dPHI FORCE",fvl_dphi,1,ry); - // this->print_force("VL_dVL FORCE",fvl_dvl,1,ry); - ModuleIO::print_force(GlobalV::ofs_running, ucell, "VL_dVL FORCE", fvl_dvl, false); - ModuleIO::print_force(GlobalV::ofs_running, ucell, "EWALD FORCE", fewalds, false); - // this->print_force("VLOCAL FORCE",fvlocal,PARAM.inp.test_force); - // this->print_force("EWALD FORCE",fewalds,1,ry); - ModuleIO::print_force(GlobalV::ofs_running, ucell, "NLCC FORCE", fcc, false); - ModuleIO::print_force(GlobalV::ofs_running, ucell, "SCC FORCE", fscc, false); - // this->print_force("NLCC FORCE",fcc,1,ry); - // this->print_force("SCC FORCE",fscc,1,ry); - //------------------------------- - // put extra force here for test! - //------------------------------- - if (PARAM.inp.efield_flag) - { - ModuleIO::print_force(GlobalV::ofs_running, ucell, "EFIELD FORCE", fefield, false); - // this->print_force("EFIELD FORCE",fefield,1,ry); - } - if (PARAM.inp.esolver_type == "tddft") - { - ModuleIO::print_force(GlobalV::ofs_running, ucell, "EFIELD_TDDFT FORCE", fefield_tddft, false); - // this->print_force("EFIELD_TDDFT FORCE",fefield_tddft,1,ry); - } - if (PARAM.inp.gate_flag) - { - ModuleIO::print_force(GlobalV::ofs_running, ucell, "GATEFIELD FORCE", fgate, false); - // this->print_force("GATEFIELD FORCE",fgate,1,ry); - } - if (PARAM.inp.imp_sol) - { - ModuleIO::print_force(GlobalV::ofs_running, ucell, "IMP_SOL FORCE", fsol, false); - // this->print_force("IMP_SOL FORCE",fsol,1,ry); - } - if (vdw_result != nullptr) - { - ModuleIO::print_force(GlobalV::ofs_running, ucell, "VDW FORCE", force_vdw, false); - // this->print_force("VDW FORCE",force_vdw,1,ry); - } - if (PARAM.inp.dft_plus_u) - { - ModuleIO::print_force(GlobalV::ofs_running, ucell, "DFT+U FORCE", force_u, false); - } - if (PARAM.inp.sc_mag_switch) - { - ModuleIO::print_force(GlobalV::ofs_running, ucell, "DeltaSpin FORCE", force_dspin, false); - } -#ifdef __MLALGO - // caoyu add 2021-06-03 - if (PARAM.inp.deepks_scf) - { - ModuleIO::print_force(GlobalV::ofs_running, ucell, "DeePKS FORCE", fvnl_dalpha, true); - } -#endif - } - - GlobalV::ofs_running << std::setiosflags(std::ios::left); - - // this->printforce_total(ry, istestf, fcs); - ModuleIO::print_force(GlobalV::ofs_running, ucell, "TOTAL-FORCE (eV/Angstrom)", fcs, false); - net_force*= ModuleBase::Ry_to_eV / ModuleBase::BOHR_TO_A; - ModuleBase::GlobalFunc::OUT(GlobalV::ofs_running, "Net force vector (eV/Ang)", net_force.x, net_force.y, net_force.z); - ModuleBase::GlobalFunc::OUT(GlobalV::ofs_running, "Total drift (ev/Ang)", net_force.norm()); - if (istestf) - { - GlobalV::ofs_running << "\n FORCE INVALID TABLE." << std::endl; - GlobalV::ofs_running << " " << std::setw(8) << "atom" << std::setw(5) << "x" << std::setw(5) << "y" - << std::setw(5) << "z" << std::endl; - for (int iat = 0; iat < ucell.nat; iat++) - { - GlobalV::ofs_running << " " << std::setw(8) << iat; - for (int i = 0; i < 3; i++) - { - if (std::abs(fcs(iat, i) * ModuleBase::Ry_to_eV / ModuleBase::BOHR_TO_A) - < Force_Stress_LCAO::force_invalid_threshold_ev) - { - fcs(iat, i) = 0.0; - GlobalV::ofs_running << std::setw(5) << "1"; - } - else - { - GlobalV::ofs_running << std::setw(5) << "0"; - } - } - GlobalV::ofs_running << std::endl; - } - } - } // end of force calculation - //--------------------------------- - // begin calculate and output stress - //--------------------------------- - if (isstress) - { - for (int i = 0; i < 3; i++) - { - for (int j = 0; j < 3; j++) - { - scs(i, j) += soverlap(i, j) + stvnl_dphi(i, j) + svnl_dbeta(i, j) + svl_dphi(i, j) - + sigmadvl(i, j) // derivative of local potential stress (pw) - + sigmaewa(i, j) // ewald stress (pw) - + sigmacc(i, j) // nonlinear core correction stress (pw) - + sigmaxc(i, j) // exchange corretion stress - + sigmahar(i, j); // hartree stress - - // VDW stress from linpz and jiyy - if (vdw_result != nullptr) - { - scs(i, j) += stress_vdw(i, j); - } - // DFT plus U stress from qux - if (PARAM.inp.dft_plus_u) - { - scs(i, j) += stress_u(i, j); - } - if (PARAM.inp.sc_mag_switch) - { - scs(i, j) += stress_dspin(i, j); - } -#ifdef __EXX - // Stress contribution from exx - if (exx_info.info_global.cal_exx) - { - scs(i, j) += stress_exx(i, j); - } -#endif -#ifdef __MLALGO - if (PARAM.inp.deepks_scf) - { - scs(i, j) += svnl_dalpha(i, j); - } -#endif - } - } - if (ModuleSymmetry::Symmetry::symm_flag == 1) - { - symm->symmetrize_mat3(scs, ucell.lat); - } // end symmetry - - deepks.write_stress(scs, svnl_dalpha, ucell.omega, PARAM.inp); - - // print Rydberg stress or not - bool ry = false; - - // test stress each terms if needed - if (istests) - { - // test - ModuleBase::matrix svlocal; - svlocal.create(3, 3); - ModuleBase::matrix stvnl; - stvnl.create(3, 3); - for (int i = 0; i < 3; i++) - { - for (int j = 0; j < 3; j++) - { - svlocal(i, j) = svl_dphi(i, j) + sigmadvl(i, j); - stvnl(i, j) = stvnl_dphi(i, j) + svnl_dbeta(i, j); - } - } - - const bool screen = PARAM.inp.test_stress; - - GlobalV::ofs_running << "\n PARTS OF STRESS: " << std::endl; - GlobalV::ofs_running << std::setiosflags(std::ios::showpos); - GlobalV::ofs_running << std::setiosflags(std::ios::fixed) << std::setprecision(10) << std::endl; - ModuleIO::print_stress("OVERLAP STRESS", soverlap, screen, ry, GlobalV::ofs_running); - ModuleIO::print_stress("T STRESS", stvnl_dphi, screen, ry, GlobalV::ofs_running); - ModuleIO::print_stress("VNL STRESS", svnl_dbeta, screen, ry, GlobalV::ofs_running); - ModuleIO::print_stress("T_VNL STRESS", stvnl, screen, ry, GlobalV::ofs_running); - ModuleIO::print_stress("VL_dPHI STRESS", svl_dphi, screen, ry, GlobalV::ofs_running); - ModuleIO::print_stress("VL_dVL STRESS", sigmadvl, screen, ry, GlobalV::ofs_running); - ModuleIO::print_stress("HAR STRESS", sigmahar, screen, ry, GlobalV::ofs_running); - ModuleIO::print_stress("EWALD STRESS", sigmaewa, screen, ry, GlobalV::ofs_running); - ModuleIO::print_stress("cc STRESS", sigmacc, screen, ry, GlobalV::ofs_running); - ModuleIO::print_stress("XC STRESS", sigmaxc, screen, ry, GlobalV::ofs_running); - if (vdw_result != nullptr) - { - ModuleIO::print_stress("VDW STRESS", stress_vdw, screen, ry, GlobalV::ofs_running); - } - if (PARAM.inp.dft_plus_u) - { - ModuleIO::print_stress("DFTU STRESS", stress_u, screen, ry, GlobalV::ofs_running); - } - if (PARAM.inp.sc_mag_switch) - { - ModuleIO::print_stress("DeltaSpin STRESS", stress_dspin, screen, ry, GlobalV::ofs_running); - } -#ifdef __EXX - if (cal_exx) - { - ModuleIO::print_stress("EXX STRESS", stress_exx, screen, ry, GlobalV::ofs_running); - } -#endif - ModuleIO::print_stress("TOTAL STRESS", scs, screen, ry, GlobalV::ofs_running); - } // end of test - - GlobalV::ofs_running << std::setiosflags(std::ios::left); - - // print total stress - bool screen_normal = true; - ModuleIO::print_stress("TOTAL-STRESS", scs, screen_normal, ry, GlobalV::ofs_running); - - double unit_transform = 0.0; - unit_transform = ModuleBase::RYDBERG_SI / pow(ModuleBase::BOHR_RADIUS_SI, 3) * 1.0e-8; - double external_stress[3] = {PARAM.inp.press1, PARAM.inp.press2, PARAM.inp.press3}; - - for (int i = 0; i < 3; i++) - { - scs(i, i) -= external_stress[i] / unit_transform; - } - } // end of stress calculation - ModuleBase::timer::end("Force_Stress_LCAO", "getForceStress"); - return; -} +#include "source_base/mathzone.h" // local pseudopotential, ewald, core correction, scc terms in force template @@ -935,16 +393,18 @@ void Force_Stress_LCAO::calForcePwPart(UnitCell& ucell, const Charge* const chr, ModulePW::PW_Basis* rhopw, const pseudopot_cell_vl& locpp, - const Structure_Factor& sf) + const Structure_Factor& sf, + const FSCalcConfig& cfg) { ModuleBase::TITLE("Force_Stress_LCAO", "calForcePwPart"); #ifdef __CUDA - if(PARAM.inp.device == "gpu") + if (cfg.device == "gpu") { Forces f_pw(nat); f_pw.cal_force_loc(ucell, fvl_dvl, rhopw, locpp.vloc, chr); f_pw.cal_force_ew(ucell, fewalds, rhopw, &sf); - f_pw.cal_force_cc(fcc, rhopw, chr, locpp.numeric, ucell); + f_pw.cal_force_cc(fcc, rhopw, chr, locpp.numeric, ucell, + cfg.nspin, cfg.domag, cfg.domag_z, cfg.gga_grad); f_pw.cal_force_scc(fscc, rhopw, vnew, vnew_exist, locpp.numeric, ucell); } else @@ -953,150 +413,14 @@ void Force_Stress_LCAO::calForcePwPart(UnitCell& ucell, Forces f_pw(nat); f_pw.cal_force_loc(ucell, fvl_dvl, rhopw, locpp.vloc, chr); f_pw.cal_force_ew(ucell, fewalds, rhopw, &sf); - f_pw.cal_force_cc(fcc, rhopw, chr, locpp.numeric, ucell); + f_pw.cal_force_cc(fcc, rhopw, chr, locpp.numeric, ucell, + cfg.nspin, cfg.domag, cfg.domag_z, cfg.gga_grad); f_pw.cal_force_scc(fscc, rhopw, vnew, vnew_exist, locpp.numeric, ucell); } return; } -// overlap, kinetic, nonlocal pseudopotential, Local potential terms in force and stress -template <> -void Force_Stress_LCAO::integral_part(const bool isGammaOnly, - const bool isforce, - const bool isstress, - const UnitCell& ucell, - const Grid_Driver& gd, - ForceStressArrays& fsr, // mohan add 2024-06-15 - const elecstate::ElecState* pelec, - const elecstate::DensityMatrix* dm, // mohan add 2025-11-04 - const psi::Psi* psi, - ModuleBase::matrix& foverlap, - ModuleBase::matrix& ftvnl_dphi, - ModuleBase::matrix& fvnl_dbeta, - ModuleBase::matrix& fvl_dphi, - ModuleBase::matrix& soverlap, - ModuleBase::matrix& stvnl_dphi, - ModuleBase::matrix& svnl_dbeta, - ModuleBase::matrix& svl_dphi, - ModuleBase::matrix& fvnl_dalpha, - ModuleBase::matrix& svnl_dalpha, - Setup_DeePKS& deepks, - const TwoCenterBundle& two_center_bundle, - const LCAO_Orbitals& orb, - const Parallel_Orbitals& pv, - const K_Vectors& kv) -{ - - flk.ftable(isforce, isstress, fsr, ucell, gd, psi, pelec, dm, - foverlap, ftvnl_dphi, fvnl_dbeta, fvl_dphi, - soverlap, stvnl_dphi, svnl_dbeta, svl_dphi, - fvnl_dalpha, svnl_dalpha, deepks, two_center_bundle, orb, pv); - return; -} - -template <> -void Force_Stress_LCAO>::integral_part(const bool isGammaOnly, - const bool isforce, - const bool isstress, - const UnitCell& ucell, - const Grid_Driver& gd, - ForceStressArrays& fsr, // mohan add 2024-06-15 - const elecstate::ElecState* pelec, - const elecstate::DensityMatrix, double>* dm, // mohan add 2025-11-04 - const psi::Psi>* psi, - ModuleBase::matrix& foverlap, - ModuleBase::matrix& ftvnl_dphi, - ModuleBase::matrix& fvnl_dbeta, - ModuleBase::matrix& fvl_dphi, - ModuleBase::matrix& soverlap, - ModuleBase::matrix& stvnl_dphi, - ModuleBase::matrix& svnl_dbeta, - ModuleBase::matrix& svl_dphi, - ModuleBase::matrix& fvnl_dalpha, - ModuleBase::matrix& svnl_dalpha, - Setup_DeePKS>& deepks, - const TwoCenterBundle& two_center_bundle, - const LCAO_Orbitals& orb, - const Parallel_Orbitals& pv, - const K_Vectors& kv) -{ - flk.ftable(isforce, isstress, fsr, ucell, gd, psi, pelec, dm, - foverlap, ftvnl_dphi, fvnl_dbeta, fvl_dphi, - soverlap, stvnl_dphi, svnl_dbeta, svl_dphi, - fvnl_dalpha, svnl_dalpha, deepks, - two_center_bundle, orb, pv, &kv, this->RA); - return; -} - -// vlocal, hartree, ewald, core correction, exchange-correlation terms in stress -template -void Force_Stress_LCAO::calStressPwPart(UnitCell& ucell, - ModuleBase::matrix& sigmadvl, - ModuleBase::matrix& sigmahar, - ModuleBase::matrix& sigmaewa, - ModuleBase::matrix& sigmacc, - ModuleBase::matrix& sigmaxc, - const double& etxc, - const Charge* const chr, - ModulePW::PW_Basis* rhopw, - const pseudopot_cell_vl& locpp, - const Structure_Factor& sf) -{ - ModuleBase::TITLE("Force_Stress_LCAO", "calStressPwPart"); - - // local pseudopotential stress: - sc_pw.stress_loc(ucell, sigmadvl, rhopw, locpp.vloc, &sf, 0, chr); - - // hartree term - sc_pw.stress_har(ucell, sigmahar, rhopw, 0, chr); - - // ewald stress: use plane wave only. - sc_pw.stress_ewa(ucell, sigmaewa, rhopw, 0); // remain problem - - // stress due to core correlation. - sc_pw.stress_cc(sigmacc, rhopw, ucell, &sf, 0, locpp.numeric, chr); - - // stress due to self-consistent charge. - for (int i = 0; i < 3; i++) - { - sigmaxc(i, i) = -etxc / ucell.omega; - } - // Exchange-correlation for PBE - sc_pw.stress_gga(ucell, sigmaxc, rhopw, chr); - - return; -} - -#include "source_base/mathzone.h" -// do symmetry for total force -template -void Force_Stress_LCAO::forceSymmetry(const UnitCell& ucell, ModuleBase::matrix& fcs, ModuleSymmetry::Symmetry* symm) -{ - double d1, d2, d3; - for (int iat = 0; iat < ucell.nat; iat++) - { - ModuleBase::Mathzone::Cartesian_to_Direct(fcs(iat, 0), fcs(iat, 1), fcs(iat, 2), - ucell.a1.x, ucell.a1.y, ucell.a1.z, ucell.a2.x, ucell.a2.y, ucell.a2.z, - ucell.a3.x, ucell.a3.y, ucell.a3.z, d1, d2, d3); - - fcs(iat, 0) = d1; - fcs(iat, 1) = d2; - fcs(iat, 2) = d3; - } - symm->symmetrize_vec3_nat(fcs.c); - for (int iat = 0; iat < ucell.nat; iat++) - { - ModuleBase::Mathzone::Direct_to_Cartesian(fcs(iat, 0), fcs(iat, 1), fcs(iat, 2), - ucell.a1.x, ucell.a1.y, ucell.a1.z, ucell.a2.x, ucell.a2.y, ucell.a2.z, - ucell.a3.x, ucell.a3.y, ucell.a3.z, d1, d2, d3); - - fcs(iat, 0) = d1; - fcs(iat, 1) = d2; - fcs(iat, 2) = d3; - } - return; -} template class Force_Stress_LCAO; template class Force_Stress_LCAO>; diff --git a/source/source_lcao/force_stress_lcao.h b/source/source_lcao/force_stress_lcao.h index d6022cbf111..3eeb5c1a605 100644 --- a/source/source_lcao/force_stress_lcao.h +++ b/source/source_lcao/force_stress_lcao.h @@ -1,7 +1,9 @@ #ifndef FORCE_STRESS_LCAO_H #define FORCE_STRESS_LCAO_H -#include "force_lcao.h" +#include + +#include "edm.h" #include "source_base/global_function.h" #include "source_base/matrix.h" #include "source_pw/module_pwdft/force_pw.h" @@ -15,7 +17,7 @@ #include "force_stress_arrays.h" #include "source_lcao/setup_exx.h" // for exx, mohan add 20251008 #include "source_lcao/setup_deepks.h" // for deepks, mohan add 20251010 -#include "source_lcao/setup_dm.h" // mohan add 2025-11-03 +#include "source_lcao/allocate_dm.h" // mohan add 2025-11-03 #include "source_pw/module_pwdft/dftu_base.h" // mohan add 2025-11-07 #include "source_hamilt/hamilt.h" #include "source_hamilt/module_xc/exx_info.h" @@ -25,6 +27,68 @@ namespace vdw struct VdwResult; } +class TwoCenterBundle; + +// INPUT scalars that steer the force/stress kernels. Bundling them into one +// aggregate keeps getForceStress and its helpers from each re-reading the +// global PARAM object, and collapses five arguments into a single reference. +struct FSCalcConfig +{ + int nspin; + int nbands; + bool t_in_h; + bool sc_mag_switch; + std::string device; + bool domag; + bool domag_z; + int gga_grad; + bool gamma_only_pw; +}; + +// Force/stress component matrices assembled by getForceStress. Grouping them +// into a struct lets the assembly/print helpers take one reference instead of +// ~19 individual matrix arguments. Members are default-constructed and only +// created (allocated) when the corresponding term is active. +struct LCAOForceParts +{ + ModuleBase::matrix foverlap; + ModuleBase::matrix ftvnl_dphi; + ModuleBase::matrix fvnl_dbeta; + ModuleBase::matrix fvl_dphi; + ModuleBase::matrix fvl_dvl; + ModuleBase::matrix fewalds; + ModuleBase::matrix fcc; + ModuleBase::matrix fscc; + ModuleBase::matrix fvnl_dalpha; // deepks + ModuleBase::matrix fpothybrid; + ModuleBase::matrix force_u; + ModuleBase::matrix force_dspin; + ModuleBase::matrix force_exx; + ModuleBase::matrix force_vdw; + ModuleBase::matrix fefield; + ModuleBase::matrix fefield_tddft; + ModuleBase::matrix fgate; + ModuleBase::matrix fsol; +}; + +struct LCAOStressParts +{ + ModuleBase::matrix soverlap; + ModuleBase::matrix stvnl_dphi; + ModuleBase::matrix svnl_dbeta; + ModuleBase::matrix svl_dphi; + ModuleBase::matrix sigmadvl; + ModuleBase::matrix sigmahar; + ModuleBase::matrix sigmaewa; + ModuleBase::matrix sigmacc; + ModuleBase::matrix sigmaxc; + ModuleBase::matrix svnl_dalpha; // deepks + ModuleBase::matrix stress_u; + ModuleBase::matrix stress_dspin; + ModuleBase::matrix stress_exx; + ModuleBase::matrix stress_vdw; +}; + template class Force_Stress_LCAO @@ -47,7 +111,7 @@ class Force_Stress_LCAO const Grid_Driver& gd, Parallel_Orbitals& pv, const elecstate::ElecState* pelec, - LCAO_domain::Setup_DM &dmat, // mohan add 2025-11-03 + module_dm::Setup_DM &dmat, // mohan add 2025-11-03 const psi::Psi* psi, const TwoCenterBundle& two_center_bundle, const LCAO_Orbitals& orb, @@ -63,17 +127,39 @@ class Force_Stress_LCAO Exx_NAO &exx_nao, ModuleSymmetry::Symmetry* symm, const Exx_Info& exx_info, + const FSCalcConfig& cfg, const int td_stype = 0, hamilt::Hamilt* p_hamilt = nullptr); private: int nat; Record_adj* RA = nullptr; - Force_LCAO flk; + CalEDM edm_cal; Stress_Func sc_pw; - void forceSymmetry(const UnitCell& ucell, ModuleBase::matrix& fcs, ModuleSymmetry::Symmetry* symm); + // Operator-based force/stress terms: kinetic, overlap, nonlocal, + // rt-TDDFT hybrid gauge, local-potential Pulay term, and DeltaSpin. + void cal_operator_fs(UnitCell& ucell, + const Grid_Driver& gd, + Parallel_Orbitals& pv, + const elecstate::ElecState* pelec, + module_dm::Setup_DM& dmat, + const psi::Psi* psi, + const TwoCenterBundle& two_center_bundle, + const LCAO_Orbitals& orb, + const K_Vectors& kv, + const bool isforce, + const bool isstress, + const FSCalcConfig& cfg, + const int td_stype, + hamilt::Hamilt* p_hamilt, + LCAOForceParts& parts, + LCAOStressParts& sparts); + + // Local pseudopotential, Ewald, core-correction and self-consistent-field + // force contributions, computed with the plane-wave Forces driver. Kept as + // a member because it needs friend access to Forces::cal_force_*. void calForcePwPart(UnitCell& ucell, ModuleBase::matrix& fvl_dvl, ModuleBase::matrix& fewalds, @@ -85,44 +171,8 @@ class Force_Stress_LCAO const Charge* const chr, ModulePW::PW_Basis* rhopw, const pseudopot_cell_vl& locpp, - const Structure_Factor& sf); - - void integral_part(const bool isGammaOnly, - const bool isforce, - const bool isstress, - const UnitCell& ucell, - const Grid_Driver& gd, - ForceStressArrays& fsr, // mohan add 2024-06-15 - const elecstate::ElecState* pelec, - const elecstate::DensityMatrix* dm, // mohan add 2025-11-04 - const psi::Psi* psi, - ModuleBase::matrix& foverlap, - ModuleBase::matrix& ftvnl_dphi, - ModuleBase::matrix& fvnl_dbeta, - ModuleBase::matrix& fvl_dphi, - ModuleBase::matrix& soverlap, - ModuleBase::matrix& stvnl_dphi, - ModuleBase::matrix& svnl_dbeta, - ModuleBase::matrix& svl_dphi, - ModuleBase::matrix& fvnl_dalpha, - ModuleBase::matrix& svnl_dalpha, - Setup_DeePKS& deepks, - const TwoCenterBundle& two_center_bundle, - const LCAO_Orbitals& orb, - const Parallel_Orbitals& pv, - const K_Vectors& kv); - - void calStressPwPart(UnitCell& ucell, - ModuleBase::matrix& sigmadvl, - ModuleBase::matrix& sigmahar, - ModuleBase::matrix& sigmaewa, - ModuleBase::matrix& sigmacc, - ModuleBase::matrix& sigmaxc, - const double& etxc, - const Charge* const chr, - ModulePW::PW_Basis* rhopw, - const pseudopot_cell_vl& locpp, - const Structure_Factor& sf); + const Structure_Factor& sf, + const FSCalcConfig& cfg); static double force_invalid_threshold_ev; }; @@ -133,10 +183,9 @@ double Force_Stress_LCAO::force_invalid_threshold_ev = 0.00; // only for DFT+U, mohan add 2025-11-04 template void assign_dmk_ptr( - elecstate::DensityMatrix* dm, + module_dm::DensityMatrix* dm, std::vector>*& dmk_d, - std::vector>>*& dmk_c, - bool gamma_only_local + std::vector>>*& dmk_c ); #endif diff --git a/source/source_lcao/force_stress_terms.cpp b/source/source_lcao/force_stress_terms.cpp new file mode 100644 index 00000000000..0ca1cc1db03 --- /dev/null +++ b/source/source_lcao/force_stress_terms.cpp @@ -0,0 +1,357 @@ +#include "force_stress_terms.h" + +#include + +#include "source_base/parallel_reduce.h" +#include "source_base/timer.h" +#include "source_base/tool_quit.h" +#include "source_basis/module_nao/two_center_bundle.h" +#include "source_estate/module_pot/efield.h" +#include "source_estate/module_pot/gatefield.h" +#include "source_estate/module_pot/h_tddft_pw.h" +#include "source_hamilt/module_surchem/surchem.h" +#include "source_hamilt/module_vdw/vdw.h" +#include "source_io/module_parameter/parameter.h" +#include "source_lcao/module_dftu/dftu_nao_adj.h" +#include "source_lcao/module_dftu/dftu_nao_fs_k.h" +#include "source_lcao/module_dftu/dftu_nao_fs_r.h" +#include "source_pw/module_pwdft/dftu_base.h" +#ifdef __MLALGO +#include "source_lcao/module_deepks/deepks_force.h" +#endif + +namespace LCAO_domain +{ + +namespace +{ +// Copy the cached vdW force / stress into the parts containers. +void copy_vdw_terms(const vdw::VdwResult* vdw_result, + const UnitCell& ucell, + const bool isforce, + const bool isstress, + LCAOForceParts& parts, + LCAOStressParts& sparts) +{ + if (vdw_result == nullptr) + { + return; + } + if (isforce) + { + if (!vdw_result->has_force || vdw_result->force.size() != static_cast(ucell.nat)) + { + ModuleBase::WARNING_QUIT("Force_Stress_LCAO::getForceStress", + "The cached vdW force is unavailable or has an invalid size."); + } + parts.force_vdw.create(ucell.nat, 3); + for (int iat = 0; iat < ucell.nat; ++iat) + { + parts.force_vdw(iat, 0) = vdw_result->force[iat].x; + parts.force_vdw(iat, 1) = vdw_result->force[iat].y; + parts.force_vdw(iat, 2) = vdw_result->force[iat].z; + } + } + if (isstress) + { + if (!vdw_result->has_stress) + { + ModuleBase::WARNING_QUIT("Force_Stress_LCAO::getForceStress", + "The cached vdW stress is unavailable."); + } + sparts.stress_vdw = vdw_result->stress.to_matrix(); + } +} + +// Compute the external-field force terms (E-field, rt-TDDFT, gate, solvation). +void cal_external_field_forces(UnitCell& ucell, + surchem& solvent, + ModulePW::PW_Basis* rhopw, + const pseudopot_cell_vl& locpp, + LCAOForceParts& parts) +{ + //! forces from E-field + if (PARAM.inp.efield_flag) + { + parts.fefield.create(ucell.nat, 3); + elecstate::Efield::compute_force(ucell, parts.fefield); + } + + //! atomic forces from E-field of rt-TDDFT + if (PARAM.inp.esolver_type == "tddft") + { + parts.fefield_tddft.create(ucell.nat, 3); + elecstate::H_TDDFT_pw::compute_force(ucell, parts.fefield_tddft); + } + + //! atomic forces from gate field + if (PARAM.inp.gate_flag) + { + parts.fgate.create(ucell.nat, 3); + elecstate::Gatefield::compute_force(ucell, parts.fgate); + } + + //! atomic forces from implicit solvation model + if (PARAM.inp.imp_sol) + { + parts.fsol.create(ucell.nat, 3); + solvent.cal_force_sol(ucell, rhopw, locpp.vloc, PARAM.inp.nspin, parts.fsol); + } +} +} // namespace + +void cal_vdw_fields_fs(const vdw::VdwResult* vdw_result, + UnitCell& ucell, + surchem& solvent, + ModulePW::PW_Basis* rhopw, + const pseudopot_cell_vl& locpp, + const bool isforce, + const bool isstress, + LCAOForceParts& parts, + LCAOStressParts& sparts) +{ + //! forces and stress from vdw + // Peize Lin add 2014-04-04, update 2021-03-09 + // jiyy add 2019-05-18, update 2021-05-02 + copy_vdw_terms(vdw_result, ucell, isforce, isstress, parts, sparts); + + //! external-field forces + if (isforce) + { + cal_external_field_forces(ucell, solvent, rhopw, locpp, parts); + } +} + +template +void cal_exx_fs(const UnitCell& ucell, + const bool isforce, + const bool isstress, + const Exx_Info& exx_info, + Exx_NAO& exx_nao, + LCAOForceParts& parts, + LCAOStressParts& sparts) +{ +#ifdef __EXX + bool cal_exx = exx_info.info_global.cal_exx; + bool real_number = exx_info.info_ri.real_number; + double hybrid_alpha = exx_info.info_global.hybrid_alpha; + + if (cal_exx) + { + if (isforce || isstress) + { + std::cout << " >> NOTICE: calculating EXX force/stress, which may be time-consuming" << std::endl; + } + if (isforce) + { + if (real_number) + { + exx_nao.exd->cal_exx_force(ucell.nat); + parts.force_exx = hybrid_alpha * exx_nao.exd->get_force(); + } + else + { + exx_nao.exc->cal_exx_force(ucell.nat); + parts.force_exx = hybrid_alpha * exx_nao.exc->get_force(); + } + } + if (isstress) + { + if (real_number) + { + exx_nao.exd->cal_exx_stress(ucell.omega, ucell.lat0); + sparts.stress_exx = hybrid_alpha * exx_nao.exd->get_stress(); + } + else + { + exx_nao.exc->cal_exx_stress(ucell.omega, ucell.lat0); + sparts.stress_exx = hybrid_alpha * exx_nao.exc->get_stress(); + } + } + } +#endif +} + +template +void cal_dftu_fs(UnitCell& ucell, + const Grid_Driver& gd, + Parallel_Orbitals& pv, + const LCAO_Orbitals& orb, + const K_Vectors& kv, + module_dm::Setup_DM& dmat, + const TwoCenterBundle& two_center_bundle, + Plus_U_Base& dftu, + const bool isforce, + const bool isstress, + LCAOForceParts& parts, + LCAOStressParts& sparts) +{ + //! atomic forces from DFT+U (Quxin version) + + if (PARAM.inp.dft_plus_u) // Quxin add for DFT+U on 20201029 + { + if (isforce) + { + parts.force_u.create(ucell.nat, 3); + } + if (isstress) + { + sparts.stress_u.create(3, 3); + } + if (PARAM.inp.dft_plus_u == 2) + { + // The legacy dft_plus_u==2 force/stress path is currently broken. + // + // Background: Plus_U::force_stress relies on ForceStressArrays + // members DSloc_x/y/z (gamma_only) or DSloc_Rx/Ry/Rz (multik) + // and DH_r being pre-allocated and filled with dS/dR data by the + // main force flow (formerly ForceLcaoGamma::ftable). The DFT+U + // step 2 refactor (commit 70c54c9d5a, 2026-01-23) removed the + // main-flow ForceStressArrays because the operator-based force + // calculation no longer needs it, but the legacy dft_plus_u==2 + // path still depends on it. The local fsr_dftu below is declared + // without allocating those arrays, so any call into + // cal_force_gamma / cal_stress_gamma / folding_matrix_k would + // pass nullptr to pdgemm_ and crash with SIGSEGV. + // + // Until the legacy path is restored or re-implemented, we + // explicitly reject dft_plus_u==2 with cal_force or cal_stress + // enabled. SCF-only runs (no force/stress) are unaffected + // because the energy is computed in cal_energy_correction, + // which does not touch DSloc arrays. Use dft_plus_u=1 for + // force/stress calculations. + if (isforce || isstress) + { + ModuleBase::WARNING_QUIT("Force_Stress_LCAO::getForceStress", + "dft_plus_u==2 with cal_force or cal_stress is currently broken; " + "please use dft_plus_u=1 instead. See notes in source/source_lcao/force_stress_lcao.cpp."); + } + ForceStressArrays fsr_dftu; + std::vector>* dmk_d = nullptr; + std::vector>>* dmk_c = nullptr; + assign_dmk_ptr(dmat.dm, dmk_d, dmk_c); + DFTU_LCAO::DftuFsEnv dftu_fs_env(dftu, ucell, gd, pv, fsr_dftu, + orb.cutoffs(), PARAM.inp.ks_solver); + DFTU_LCAO::force_stress(dftu_fs_env, isforce, isstress, + dmk_d, dmk_c, parts.force_u, sparts.stress_u, kv, + PARAM.globalv.gamma_only_local); + } + else + { + // Build DFT+U force/stress inputs directly without constructing a + // full DFTU operator (hsk/hR are irrelevant for this path). + std::vector adjs_all = DFTU_LCAO::build_adjacent_atoms( + &ucell, &dftu, &gd, orb.cutoffs(), PARAM.inp.onsite_radius); + + // The DensityMatrix holds nspin_dm = (nspin==2 ? 2 : 1) real-space DMR + // channels: nspin=4 (non-collinear) packs all four Pauli components + // into a single complex DMR, so only one channel exists (cf. allocate_dm.cpp + // and the is0 = nspin==2 ? is : 0 indexing in cal_for/str_IJR_nao_r). + const int nspin_dm = (PARAM.inp.nspin == 2) ? 2 : 1; + std::vector*> dmR_tmp(nspin_dm, nullptr); + for (int is = 0; is < nspin_dm; ++is) + { + dmR_tmp[is] = dmat.dm->get_dmr_ptr(is + 1); + } + + DFTU_LCAO::cal_fs_nao_r(&ucell, &dftu, + two_center_bundle.overlap_orb_onsite.get(), + PARAM.inp.nspin, + adjs_all, dmR_tmp, + isforce, isstress, parts.force_u, sparts.stress_u); + } + } +} + +template +void cal_deepks_fs(const UnitCell& ucell, + const Grid_Driver& gd, + Parallel_Orbitals& pv, + const LCAO_Orbitals& orb, + const K_Vectors& kv, + const bool isforce, + const bool isstress, + Setup_DeePKS& deepks, + LCAOForceParts& parts, + LCAOStressParts& sparts) +{ + // Handle DeePKS forces if enabled +#ifdef __MLALGO + if (PARAM.inp.deepks_scf) + { + const int nks = (PARAM.inp.nspin == 1 || PARAM.inp.nspin == 2) ? 1 : kv.get_nks(); + if (PARAM.globalv.gamma_only_local) + { + DeePKS_domain::cal_f_delta( + ucell, + orb, + gd, + pv, + nks, + deepks.ld.deepks_param, + kv.kvec_d, + deepks.ld.phialpha, + parts.fvnl_dalpha, + isstress, + sparts.svnl_dalpha, + deepks.ld.dm_r, + deepks.ld.gedm, + (PARAM.inp.nspin == 2 && !PARAM.inp.deepks_equiv) ? deepks.ld.dm_r_mag : nullptr, + (PARAM.inp.nspin == 2 && !PARAM.inp.deepks_equiv) ? deepks.ld.gedm_mag : nullptr); + } + else + { + DeePKS_domain::cal_f_delta>( + ucell, + orb, + gd, + pv, + nks, + deepks.ld.deepks_param, + kv.kvec_d, + deepks.ld.phialpha, + parts.fvnl_dalpha, + isstress, + sparts.svnl_dalpha, + deepks.ld.dm_r, + deepks.ld.gedm, + (PARAM.inp.nspin == 2 && !PARAM.inp.deepks_equiv) ? deepks.ld.dm_r_mag : nullptr, + (PARAM.inp.nspin == 2 && !PARAM.inp.deepks_equiv) ? deepks.ld.gedm_mag : nullptr); + } + + if (isforce) + { + Parallel_Reduce::reduce_pool(parts.fvnl_dalpha.c, parts.fvnl_dalpha.nr * parts.fvnl_dalpha.nc); + } + if (isstress) + { + Parallel_Reduce::reduce_pool(sparts.svnl_dalpha.c, sparts.svnl_dalpha.nr * sparts.svnl_dalpha.nc); + } + } +#endif +} + +// Explicit instantiation for the two electronic types used by ABACUS. +template void cal_exx_fs(const UnitCell&, const bool, const bool, const Exx_Info&, Exx_NAO&, + LCAOForceParts&, LCAOStressParts&); +template void cal_exx_fs>(const UnitCell&, const bool, const bool, const Exx_Info&, + Exx_NAO>&, LCAOForceParts&, LCAOStressParts&); + +template void cal_dftu_fs(UnitCell&, const Grid_Driver&, Parallel_Orbitals&, const LCAO_Orbitals&, + const K_Vectors&, module_dm::Setup_DM&, const TwoCenterBundle&, + Plus_U_Base&, const bool, const bool, LCAOForceParts&, LCAOStressParts&); +template void cal_dftu_fs>(UnitCell&, const Grid_Driver&, Parallel_Orbitals&, + const LCAO_Orbitals&, const K_Vectors&, + module_dm::Setup_DM>&, const TwoCenterBundle&, + Plus_U_Base&, const bool, const bool, LCAOForceParts&, + LCAOStressParts&); + +template void cal_deepks_fs(const UnitCell&, const Grid_Driver&, Parallel_Orbitals&, const LCAO_Orbitals&, + const K_Vectors&, const bool, const bool, Setup_DeePKS&, LCAOForceParts&, + LCAOStressParts&); +template void cal_deepks_fs>(const UnitCell&, const Grid_Driver&, Parallel_Orbitals&, + const LCAO_Orbitals&, const K_Vectors&, const bool, const bool, + Setup_DeePKS>&, LCAOForceParts&, + LCAOStressParts&); + +} // namespace LCAO_domain diff --git a/source/source_lcao/force_stress_terms.h b/source/source_lcao/force_stress_terms.h new file mode 100644 index 00000000000..9e21f83d1a5 --- /dev/null +++ b/source/source_lcao/force_stress_terms.h @@ -0,0 +1,84 @@ +#ifndef FORCE_STRESS_TERMS_H +#define FORCE_STRESS_TERMS_H + +#include "force_stress_lcao.h" + +// Free functions for the individual LCAO force/stress correction terms +// (vdW + external fields, EXX, DFT+U, DeePKS). They are kept outside the +// Force_Stress_LCAO class template; only the ones that genuinely depend on +// the electronic template type T are function templates, the vdW/fields one is +// a plain function. + +class UnitCell; +class Grid_Driver; +class Parallel_Orbitals; +class LCAO_Orbitals; +class K_Vectors; +class surchem; +class pseudopot_cell_vl; +namespace ModulePW +{ +class PW_Basis; +} +namespace vdw +{ +struct VdwResult; +} + +namespace LCAO_domain +{ + +// vdW force/stress and external-field forces: E-field, rt-TDDFT E-field, +// gate field and the implicit solvation model. Does not depend on T. +void cal_vdw_fields_fs(const vdw::VdwResult* vdw_result, + UnitCell& ucell, + surchem& solvent, + ModulePW::PW_Basis* rhopw, + const pseudopot_cell_vl& locpp, + const bool isforce, + const bool isstress, + LCAOForceParts& parts, + LCAOStressParts& sparts); + +// EXX force/stress (only active under __EXX). +template +void cal_exx_fs(const UnitCell& ucell, + const bool isforce, + const bool isstress, + const Exx_Info& exx_info, + Exx_NAO& exx_nao, + LCAOForceParts& parts, + LCAOStressParts& sparts); + +// DFT+U force/stress. +template +void cal_dftu_fs(UnitCell& ucell, + const Grid_Driver& gd, + Parallel_Orbitals& pv, + const LCAO_Orbitals& orb, + const K_Vectors& kv, + module_dm::Setup_DM& dmat, + const TwoCenterBundle& two_center_bundle, + Plus_U_Base& dftu, + const bool isforce, + const bool isstress, + LCAOForceParts& parts, + LCAOStressParts& sparts); + +// DeePKS correction force/stress (only active under __MLALGO). The parallel +// orbitals are passed explicitly instead of going through CalEDM::ParaV. +template +void cal_deepks_fs(const UnitCell& ucell, + const Grid_Driver& gd, + Parallel_Orbitals& pv, + const LCAO_Orbitals& orb, + const K_Vectors& kv, + const bool isforce, + const bool isstress, + Setup_DeePKS& deepks, + LCAOForceParts& parts, + LCAOStressParts& sparts); + +} // namespace LCAO_domain + +#endif diff --git a/source/source_lcao/hamilt_lcao.cpp b/source/source_lcao/hamilt_lcao.cpp index 33136664f83..3ea53637d95 100644 --- a/source/source_lcao/hamilt_lcao.cpp +++ b/source/source_lcao/hamilt_lcao.cpp @@ -1,6 +1,5 @@ #include "source_lcao/hamilt_lcao.h" - -#include "source_base/global_variable.h" +#include "source_lcao/hamilt_lcao_factory.h" #include "source_base/memory_recorder.h" #include "source_base/timer.h" #include "source_pw/module_pwdft/dftu_base.h" @@ -8,41 +7,18 @@ #include "source_lcao/setup_deepks.h" #include "source_estate/module_dm/density_matrix.h" #include "source_estate/module_pot/potential_new.h" -#include "source_io/module_parameter/parameter.h" - +#include "source_hamilt/module_hcontainer/hcontainer_funcs.h" #include - #ifdef __MLALGO #include "source_lcao/module_deepks/lcao_deepks.h" -#include "module_operator_lcao/deepks_lcao.h" #endif - #ifdef __EXX #include "source_lcao/module_ri/exx_lri_interface.h" #include "module_operator_lcao/op_exx_lcao.h" #endif -#ifdef __ELPA -#include "source_hsolver/diago_elpa.h" -#endif - -#include "source_estate/module_pot/h_tddft_pw.h" -#include "source_hamilt/module_xc/xc_functional.h" -#include "source_lcao/module_deltaspin/spin_constrain.h" -#include "source_hamilt/module_hcontainer/hcontainer_funcs.h" -#include "source_hsolver/hsolver_lcao.h" -#include "module_dftu/dftu_nao_op.h" -#include "module_operator_lcao/dspin_lcao.h" -#include "module_operator_lcao/ekinetic.h" -#include "module_operator_lcao/meta_lcao.h" -#include "module_operator_lcao/nonlocal.h" -#include "module_dftu/dftu_nao_op_legacy.h" -#include "module_operator_lcao/op_exx_lcao.h" +#include "module_operator_lcao/operator_lcao.h" #include "module_operator_lcao/overlap.h" -#include "module_operator_lcao/td_ekinetic_lcao.h" -#include "module_operator_lcao/td_nonlocal_lcao.h" -#include "module_operator_lcao/td_pot_hybrid.h" -#include "module_operator_lcao/veff_lcao.h" namespace hamilt @@ -60,11 +36,14 @@ HamiltLCAO::HamiltLCAO(const UnitCell& ucell, this->kv = &kv_in; - // initialize the overlap matrix - this->sR = new HContainer(paraV); + // initialize the overlap matrix; + // hR is also allocated so the Overlap node never receives a null hR_in + // (hsk stays null: this vacuum path has no k-space matrix and never calls init()) + this->hR.reset(new HContainer(paraV)); + this->sR.reset(new HContainer(paraV)); - this->getOperator() = new Overlap>(this->hsk, - this->kv->kvec_d, this->hR, this->sR, + this->getOperator() = new Overlap>(this->hsk.get(), + this->kv->kvec_d, this->hR.get(), this->sR.get(), &ucell, orb_cutoff, &grid_d, &intor_overlap_orb); } @@ -77,22 +56,28 @@ HamiltLCAO::HamiltLCAO(const UnitCell& ucell, const K_Vectors& kv_in, const TwoCenterBundle& two_center_bundle, const LCAO_Orbitals& orb, - elecstate::DensityMatrix* DM_in, + module_dm::DensityMatrix* DM_in, Plus_U_Base* p_dftu, // mohan add 2025-11-05 Setup_DeePKS &deepks, const int istep, Exx_NAO &exx_nao, const Exx_Info& exx_info, - const Input_para& inp) + const Input_para& inp, + const bool load_exx_flag) { this->classname = "HamiltLCAO"; this->kv = &kv_in; + // snapshot INPUT flags used later by getHR_vector/updateHk/refresh, + // so those methods do not read global PARAM + this->nspin = inp.nspin; + this->vl_in_h = inp.vl_in_h; + // Real space Hamiltonian is inited with template TR - this->hR = new HContainer(paraV); - this->sR = new HContainer(paraV); - this->hsk = new HS_Matrix_K(paraV); + this->hR.reset(new HContainer(paraV)); + this->sR.reset(new HContainer(paraV)); + this->hsk.reset(new HS_Matrix_K(paraV)); // Effective potential term (\sum_r ) is registered without template std::vector pot_register_in; @@ -109,233 +94,24 @@ HamiltLCAO::HamiltLCAO(const UnitCell& ucell, } // Gamma_only case to initialize HamiltLCAO - // - // code block to construct Operator Chains + LcaoOpsBundle bundle; if (std::is_same::value) { - // fix HR to gamma case, where SR will be fixed in Overlap Operator - this->hR->fix_gamma(); - // initial operator for Gamma_only case - // overlap term () is indispensable - // in Gamma_only case, target SK is this->hsk->get_sk(), the target SR is this->sR - this->getOperator() = new Overlap>(this->hsk, - this->kv->kvec_d, this->hR, this->sR, - &ucell, orb.cutoffs(), &grid_d, - two_center_bundle.overlap_orb.get()); - - // kinetic term () - if (inp.t_in_h) - { - Operator* ekinetic = new EKinetic>(this->hsk, - this->kv->kvec_d, this->hR, - &ucell, orb.cutoffs(), &grid_d, - two_center_bundle.kinetic_orb.get()); - this->getOperator()->add(ekinetic); - } - - // nonlocal term (D) - // in general case, target HR is this->hR, while target HK is this->hsk->get_hk() - if (inp.vnl_in_h) - { - Operator* nonlocal = new Nonlocal>(this->hsk, - this->kv->kvec_d, this->hR, - &ucell, orb.cutoffs(), &grid_d, - two_center_bundle.overlap_orb_beta.get()); - this->getOperator()->add(nonlocal); - } - - // Effective potential term (\sum_r ) - // in general case, target HR is Gint::hRGint, while target HK is this->hsk->get_hk() - if (inp.vl_in_h) - { - // only Potential is not empty, Veff and Meta are available - if (pot_register_in.size() > 0) - { - // register Potential by gathered operator - pot_in->pot_register(pot_register_in); - // effective potential term - Operator* veff = new Veff>(this->hsk, - this->kv->kvec_d, pot_in, - this->hR, // no explicit call yet - &ucell, orb.cutoffs(), &grid_d, - inp.nspin); - this->getOperator()->add(veff); - } - } - -#ifdef __MLALGO - if (inp.deepks_scf) - { - Operator* deepks_op = new DeePKS>(this->hsk, - this->kv->kvec_d, this->hR, // no explicit call yet - &ucell, &grid_d, - two_center_bundle.overlap_orb_alpha.get(), - &orb, this->kv->get_nks(), - DM_in, &deepks.ld); - this->getOperator()->add(deepks_op); - this->V_delta_R = dynamic_cast>*>(deepks_op)->get_V_delta_R(); - } -#endif - - // end node should be OperatorDFTU - if (inp.dft_plus_u) - { - Operator* plus_u = nullptr; - if (inp.dft_plus_u == 2) - { - plus_u = new OperatorDFTU>(this->hsk, - this->kv->kvec_d, this->hR, - ucell, p_dftu, - this->kv->isk); - } - else - { - plus_u = new DFTU>(this->hsk, - this->kv->kvec_d, this->hR, - ucell, &grid_d, - two_center_bundle.overlap_orb_onsite.get(), - orb.cutoffs(), p_dftu, - inp.nspin, inp.onsite_radius, DM_in); - } - this->getOperator()->add(plus_u); - } + bundle = build_gamma_ops(ucell, grid_d, paraV, pot_in, two_center_bundle, + orb, DM_in, p_dftu, deepks, inp, pot_register_in, + this->kv, this->hsk.get(), this->hR.get(), this->sR.get()); } // multi-k-points case to initialize HamiltLCAO, ops will be used else if (std::is_same>::value) { - // Effective potential term (\sum_r ) - // Meta potential term (\sum_r ) - // in general case, target HR is Gint::pvpR_reduced, while target HK is this->hsk->get_hk() - if (inp.vl_in_h) - { - // only Potential is not empty, Veff and Meta are available - if (pot_register_in.size() > 0) - { - // register Potential by gathered operator - pot_in->pot_register(pot_register_in); - // Veff term - this->getOperator() = new Veff>(this->hsk, - this->kv->kvec_d, pot_in, - this->hR, - &ucell, orb.cutoffs(), &grid_d, - inp.nspin); - } - } - - // initial operator for multi-k case - // overlap term is indispensable - Operator* overlap = new Overlap>(this->hsk, - this->kv->kvec_d, this->hR, this->sR, - &ucell, orb.cutoffs(), &grid_d, - two_center_bundle.overlap_orb.get()); - if (this->getOperator() == nullptr) - { - this->getOperator() = overlap; - } - else - { - this->getOperator()->add(overlap); - } - - // kinetic term (), - // in general case, target HR is this->hR, while target HK is this->hsk->get_hk() - if (inp.t_in_h) - { - Operator* ekinetic = new EKinetic>(this->hsk, - this->kv->kvec_d, this->hR, - &ucell, orb.cutoffs(), &grid_d, - two_center_bundle.kinetic_orb.get()); - this->getOperator()->add(ekinetic); - } - - // nonlocal term (D) - // in general case, target HR is this->hR, while target HK is this->hsk->get_hk() - if (inp.vnl_in_h) - { - Operator* nonlocal = new Nonlocal>(this->hsk, - this->kv->kvec_d, this->hR, - &ucell, orb.cutoffs(), &grid_d, - two_center_bundle.overlap_orb_beta.get()); - // TDDFT velocity gauge will calculate full non-local potential including the original one and the - // correction on its own. So the original non-local potential term should be skipped - if (inp.esolver_type != "tddft" || elecstate::H_TDDFT_pw::stype != 1) - { - this->getOperator()->add(nonlocal); - } - else - { - delete nonlocal; - } - } - + bundle = build_multik_ops(ucell, grid_d, paraV, pot_in, two_center_bundle, + orb, DM_in, p_dftu, deepks, inp, pot_register_in, + this->kv, this->hsk.get(), this->hR.get(), this->sR.get()); + } + this->getOperator() = bundle.ops; #ifdef __MLALGO - if (inp.deepks_scf) - { - Operator* deepks_op = new DeePKS>(this->hsk, - this->kv->kvec_d, hR, - &ucell, &grid_d, - two_center_bundle.overlap_orb_alpha.get(), - &orb, this->kv->get_nks(), - DM_in, &deepks.ld); - this->getOperator()->add(deepks_op); - this->V_delta_R = dynamic_cast>*>(deepks_op)->get_V_delta_R(); - } + this->V_delta_R = bundle.v_delta_R; #endif - // TDDFT_velocity_gauge - if (inp.esolver_type == "tddft" && inp.td_stype == 1) - { - Operator* td_ekinetic = new TDEkinetic>(this->hsk, - this->hR, this->kv, - &ucell, orb.cutoffs(), &grid_d, - two_center_bundle.overlap_orb.get()); - this->getOperator()->add(td_ekinetic); - - Operator* td_nonlocal = new TDNonlocal>(this->hsk, - this->kv->kvec_d, this->hR, - &ucell, orb, &grid_d); - this->getOperator()->add(td_nonlocal); - } - if (inp.esolver_type == "tddft" && inp.td_stype == 2) - { - Operator* td_pot_hybrid = new TD_pot_hybrid>(this->hsk, - this->kv, this->hR, this->sR, - orb, &ucell, orb.cutoffs(), &grid_d, - two_center_bundle.kinetic_orb.get()); - this->getOperator()->add(td_pot_hybrid); - } - if (inp.dft_plus_u) - { - Operator* plus_u = nullptr; - if (inp.dft_plus_u == 2) - { - plus_u = new OperatorDFTU>(this->hsk, - this->kv->kvec_d, this->hR, - ucell, p_dftu, - this->kv->isk); - } - else - { - plus_u = new DFTU>(this->hsk, - this->kv->kvec_d, this->hR, - ucell, &grid_d, - two_center_bundle.overlap_orb_onsite.get(), - orb.cutoffs(), p_dftu, - inp.nspin, inp.onsite_radius, DM_in); - } - this->getOperator()->add(plus_u); - } - if (inp.sc_mag_switch) - { - Operator* sc_lambda = new DeltaSpin>(this->hsk, - this->kv->kvec_d, this->hR, - ucell, &grid_d, - two_center_bundle.overlap_orb_onsite.get(), - orb.cutoffs()); - this->getOperator()->add(sc_lambda); - spinconstrain::SpinConstrain& sc = spinconstrain::SpinConstrain::getScInstance(); - sc.set_operator(sc_lambda); - } - } #ifdef __EXX if (exx_info.info_global.cal_exx) @@ -346,19 +122,18 @@ HamiltLCAO::HamiltLCAO(const UnitCell& ucell, // Keep exact exchange in H(R) for every workflow. For RT-TDDFT the // factory selects complex H(R) when EXX is active, so the operator // chain folds the complete Hamiltonian with one common TD phase. - Operator* exx = new OperatorEXX>(this->hsk, - this->hR, ucell, *this->kv, + Operator* exx = new OperatorEXX>(this->hsk.get(), + this->hR.get(), ucell, *this->kv, exx_nao.exd.get(), exx_nao.exc.get(), exx_info, Add_Hexx_Type::R, istep, - !GlobalC::restart.info_load.restart_exx - && GlobalC::restart.info_load.load_H); + load_exx_flag); this->getOperator()->add(exx); } #endif // if NSPIN==2, HR should be separated into two parts, save HR into this->hRS2 int memory_fold = 1; - if (inp.nspin == 2) + if (this->nspin == 2) { this->hRS2.resize(this->hR->get_nnr() * 2); this->hR->allocate(this->hRS2.data(), 0); @@ -367,14 +142,12 @@ HamiltLCAO::HamiltLCAO(const UnitCell& ucell, ModuleBase::Memory::record("HamiltLCAO::hR", this->hR->get_memory_size() * memory_fold); ModuleBase::Memory::record("HamiltLCAO::sR", this->sR->get_memory_size()); - - return; } template std::vector*> HamiltLCAO::getHR_vector() { - if (PARAM.inp.nspin == 2) + if (this->nspin == 2) { const int nnr = this->hRS2.size() / 2; this->hr_spin_up_.reset(new HContainer(*this->hR, this->hRS2.data())); @@ -383,15 +156,25 @@ std::vector*> HamiltLCAO::getHR_vector() } else { - return {this->hR}; + return {this->hR.get()}; + } +} + +template +OperatorLCAO* HamiltLCAO::getOperatorLCAO() +{ + if (this->ops_lcao_ == nullptr) + { + this->ops_lcao_ = dynamic_cast*>(this->ops); } + return this->ops_lcao_; } // case for multi-k-points template void HamiltLCAO::matrix(MatrixBlock& hk_in, MatrixBlock& sk_in) { - auto op = dynamic_cast*>(this->getOperator()); + OperatorLCAO* const op = this->getOperatorLCAO(); assert(op != nullptr); op->matrixHk(hk_in, sk_in); } @@ -403,21 +186,21 @@ void HamiltLCAO::updateHk(const int ik) ModuleBase::timer::start("HamiltLCAO", "updateHk"); // update global spin index - if (PARAM.inp.nspin == 2) + if (this->nspin == 2) { // if Veff is added and current_spin is changed, refresh HR - if (PARAM.inp.vl_in_h && this->kv->isk[ik] != this->current_spin) + if (this->vl_in_h && this->kv->isk[ik] != this->current_spin) { // change data pointer of HR this->hR->allocate(this->hRS2.data() + this->hRS2.size() / 2 * this->kv->isk[ik], 0); if (this->refresh_times > 0) { this->refresh_times--; - dynamic_cast*>(this->ops)->set_hr_done(false); + this->getOperatorLCAO()->set_hr_done(false); } } this->current_spin = this->kv->isk[ik]; - dynamic_cast*>(this->ops)->set_current_spin(this->kv->isk[ik]); + this->getOperatorLCAO()->set_current_spin(this->kv->isk[ik]); } this->getOperator()->init(ik); ModuleBase::timer::end("HamiltLCAO", "updateHk"); @@ -429,8 +212,8 @@ void HamiltLCAO::refresh(bool yes) ModuleBase::TITLE("HamiltLCAO", "refresh"); if(yes) { - dynamic_cast*>(this->ops)->set_hr_done(false); - if (PARAM.inp.nspin == 2) + this->getOperatorLCAO()->set_hr_done(false); + if (this->nspin == 2) { this->refresh_times = 1; this->current_spin = 0; @@ -443,9 +226,9 @@ void HamiltLCAO::refresh(bool yes) } } else { - dynamic_cast*>(this->ops)->set_hr_done(true); + this->getOperatorLCAO()->set_hr_done(true); this->refresh_times = 0; - if (PARAM.inp.nspin == 2) + if (this->nspin == 2) { // HR has been loaded from file into both halves of hRS2. // Reset to spin-up; updateHk will switch pointers as needed. @@ -470,7 +253,7 @@ void HamiltLCAO::updateSk( ModuleBase::TITLE("HamiltLCAO", "updateSk"); ModuleBase::timer::start("HamiltLCAO", "updateSk"); - ModuleBase::GlobalFunc::ZEROS(this->getSk(), this->get_size_hsk()); + ModuleBase::GlobalFunc::ZEROS(this->getSk(), this->hsk->get_size()); if (hk_type == 1) // collumn-major matrix for SK { diff --git a/source/source_lcao/hamilt_lcao.h b/source/source_lcao/hamilt_lcao.h index 7978875326b..d57e5f389a2 100644 --- a/source/source_lcao/hamilt_lcao.h +++ b/source/source_lcao/hamilt_lcao.h @@ -1,5 +1,5 @@ -#ifndef HAMILT_LCAO_H -#define HAMILT_LCAO_H +#ifndef HAMILT_LCAO_H +#define HAMILT_LCAO_H #include "source_basis/module_nao/two_center_bundle.h" #include "source_cell/klist.h" @@ -12,21 +12,17 @@ #include // elecstate::Potential forward declaration, full definition in potential_new.h (moved to .cpp) -// mohan add 20260605 namespace elecstate { class Potential; } -// elecstate::DensityMatrix forward declaration, full definition in density_matrix.h (moved to .cpp) -// mohan add 20260605 -namespace elecstate { template class DensityMatrix; } +// module_dm::DensityMatrix forward declaration, full definition in density_matrix.h (moved to .cpp) +namespace module_dm { template class DensityMatrix; } // Setup_DeePKS forward declaration, full definition in setup_deepks.h (moved to .cpp) -// mohan add 20260605 template class Setup_DeePKS; // Plus_U_Base forward declaration, full definition in source_pw/module_pwdft/dftu_base.h class Plus_U_Base; // Exx_NAO forward declaration, full definition in setup_exx.h (moved to .cpp) -// mohan add 20260605 template class Exx_NAO; /// Exx_Info forward declaration, full definition in exx_info.h @@ -38,6 +34,10 @@ struct Input_para; namespace hamilt { +// OperatorLCAO forward declaration, full definition in +// module_operator_lcao/operator_lcao.h (moved to .cpp) +template class OperatorLCAO; + // template first for type of k space H matrix elements // template second for type of temporary matrix, // gamma_only fix-gamma-matrix + S-gamma, @@ -47,10 +47,6 @@ class HamiltLCAO : public Hamilt { public: - - using TAC = std::pair>; - - /** * @brief Constructor of Hamiltonian for LCAO base * HR and SR will be allocated with Operators @@ -62,13 +58,14 @@ class HamiltLCAO : public Hamilt const K_Vectors& kv_in, const TwoCenterBundle& two_center_bundle, const LCAO_Orbitals& orb, - elecstate::DensityMatrix* DM_in, + module_dm::DensityMatrix* DM_in, Plus_U_Base* p_dftu, // mohan add 2025-11-05 Setup_DeePKS &deepks, const int istep, Exx_NAO &exx_nao, const Exx_Info& exx_info, - const Input_para& inp); + const Input_para& inp, + const bool load_exx_flag); /** * @brief Constructor of vacuum Operators, only HR and SR will be initialed as empty HContainer @@ -82,13 +79,7 @@ class HamiltLCAO : public Hamilt ~HamiltLCAO() { - if (this->ops != nullptr) - { - delete this->ops; - } - delete this->hR; - delete this->sR; - delete this->hsk; + delete this->ops; } /// get pointer of Operator ops @@ -106,29 +97,24 @@ class HamiltLCAO : public Hamilt return this->hsk->get_sk(); } - int get_size_hsk() const - { - return this->hsk->get_size(); - } - /// get HR pointer of *this->hR, which is a HContainer and contains H(R) - HContainer*& getHR() + HContainer* getHR() { - return this->hR; + return this->hR.get(); } const HContainer* getHR() const { - return this->hR; + return this->hR.get(); } /// get SR pointer of *this->sR, which is a HContainer and contains S(R) - HContainer*& getSR() + HContainer* getSR() { - return this->sR; + return this->sR.get(); } const HContainer* getSR() const { - return this->sR; + return this->sR.get(); } #ifdef __MLALGO @@ -151,7 +137,7 @@ class HamiltLCAO : public Hamilt void refresh(bool yes) override; // for target K point, update consequence of hPsi() and matrix() - virtual void updateHk(const int ik) override; + void updateHk(const int ik) override; /** * @brief special for LCAO, update SK only @@ -161,7 +147,7 @@ class HamiltLCAO : public Hamilt * @param hk_type 0: SK is row-major, 1: SK is collumn-major * @return void */ - void updateSk(const int ik, const int hk_type = 0); + void updateSk(const int ik, const int hk_type); // core function: return H(k) and S(k) matrixs for direct solving eigenvalues. // not used in PW base @@ -172,17 +158,17 @@ class HamiltLCAO : public Hamilt const K_Vectors* kv = nullptr; //! Real space Hamiltonian H(R), where R is the Bravis lattice vector - HContainer* hR = nullptr; + std::unique_ptr> hR; //! Real space overlap matrix S(R), where R is the Bravis lattice vector - HContainer* sR = nullptr; + std::unique_ptr> sR; #ifdef __MLALGO HContainer* V_delta_R = nullptr; #endif //! Hamiltonian and overlap matrices for a specific k point - HS_Matrix_K* hsk = nullptr; + std::unique_ptr> hsk; // special case for NSPIN=2 , data of HR should be separated into two parts // save them in this->hRS2; @@ -200,7 +186,18 @@ class HamiltLCAO : public Hamilt //! 1: Hamiltonian for spin down int current_spin = 0; - const int istep = 0; + //! snapshot of inp.nspin taken at construction; avoids PARAM dependency + int nspin = 1; + + //! snapshot of inp.vl_in_h taken at construction; avoids PARAM dependency + bool vl_in_h = true; + + //! cached downcast of this->ops to OperatorLCAO, filled on first use + //! to avoid repeating dynamic_cast in updateHk/refresh + OperatorLCAO* ops_lcao_ = nullptr; + + /// get this->ops downcast to OperatorLCAO*, cached in ops_lcao_ + OperatorLCAO* getOperatorLCAO(); }; } // namespace hamilt diff --git a/source/source_lcao/hamilt_lcao_factory.cpp b/source/source_lcao/hamilt_lcao_factory.cpp new file mode 100644 index 00000000000..05a7b34b42b --- /dev/null +++ b/source/source_lcao/hamilt_lcao_factory.cpp @@ -0,0 +1,403 @@ +#include "source_lcao/hamilt_lcao_factory.h" + +#include "source_base/global_function.h" +#include "source_estate/module_pot/h_tddft_pw.h" +#include "source_lcao/module_deltaspin/spin_constrain.h" + +#ifdef __MLALGO +#include "module_operator_lcao/deepks_lcao.h" +#endif + +// operator nodes, in construction-chain order: +// overlap -> kinetic -> nonlocal -> veff -> dftu -> tddft +#include "module_operator_lcao/overlap.h" +#include "module_operator_lcao/ekinetic.h" +#include "module_operator_lcao/nonlocal.h" +#include "module_operator_lcao/veff_lcao.h" +#include "module_dftu/dftu_nao_op.h" +#include "module_dftu/dftu_nao_op_legacy.h" +#include "module_operator_lcao/dspin_lcao.h" +#include "module_operator_lcao/td_ekinetic_lcao.h" +#include "module_operator_lcao/td_nonlocal_lcao.h" +#include "module_operator_lcao/td_pot_hybrid.h" + +#include +#include + +namespace hamilt +{ + +namespace +{ + +/** + * @brief append the DFT+U operator node shared by gamma and multi-k chains. + * + * @param ops current chain head, must be non-null so the node can be added + */ +template +void add_dftu_op(Operator*& ops, + const UnitCell& ucell, + const Grid_Driver& grid_d, + const TwoCenterBundle& two_center_bundle, + const LCAO_Orbitals& orb, + module_dm::DensityMatrix* DM_in, + Plus_U_Base* p_dftu, + const Input_para& inp, + const K_Vectors* kv, + HS_Matrix_K* hsk, + HContainer* hR) +{ + Operator* plus_u = nullptr; + if (inp.dft_plus_u == 1) + { + // radius-adjustable localized projections (with onsite_radius) + plus_u = new DFTU_onsite>(hsk, + kv->kvec_d, hR, + ucell, &grid_d, + two_center_bundle.overlap_orb_onsite.get(), + orb.cutoffs(), p_dftu, + inp.nspin, inp.onsite_radius, DM_in); + static_cast>*>(plus_u)->set_kv(kv); + } + else if (inp.dft_plus_u == 2) + { + // first-zeta NAO projections (old method, kept for testing) + plus_u = new DFTU_firstzeta>(hsk, + kv->kvec_d, hR, + ucell, p_dftu, + kv->isk); + } + else + { + // add_dftu_op is only reachable when dft_plus_u != 0; any other + // value is an unsupported DFT+U method and must abort here + ModuleBase::WARNING_QUIT("add_dftu_op", + "unsupported dft_plus_u value: only 1 (radius-adjustable) or 2 (first-zeta NAO) are allowed"); + } + ops->add(plus_u); +} + +#ifdef __MLALGO +/** + * @brief append the DeePKS operator node shared by gamma and multi-k chains. + * + * @param ops current chain head, must be non-null so the node can be added + * @return the DeePKS V_delta(R) container exposed by the appended node + */ +template +HContainer* add_deepks_op(Operator*& ops, + const UnitCell& ucell, + const Grid_Driver& grid_d, + const TwoCenterBundle& two_center_bundle, + const LCAO_Orbitals& orb, + module_dm::DensityMatrix* DM_in, + Setup_DeePKS& deepks, + const K_Vectors* kv, + HS_Matrix_K* hsk, + HContainer* hR) +{ + Operator* deepks_op = new DeePKS>(hsk, + kv->kvec_d, hR, + &ucell, &grid_d, + two_center_bundle.overlap_orb_alpha.get(), + &orb, kv->get_nks(), + DM_in, &deepks.ld); + ops->add(deepks_op); + return dynamic_cast>*>(deepks_op)->get_V_delta_R(); +} +#endif + +} // anonymous namespace + +// build the operator chain for the gamma-only case (TK == double) +template +LcaoOpsBundle build_gamma_ops(const UnitCell& ucell, + const Grid_Driver& grid_d, + const Parallel_Orbitals* paraV, + elecstate::Potential* pot_in, + const TwoCenterBundle& two_center_bundle, + const LCAO_Orbitals& orb, + module_dm::DensityMatrix* DM_in, + Plus_U_Base* p_dftu, + Setup_DeePKS& deepks, + const Input_para& inp, + const std::vector& pot_register_in, + const K_Vectors* kv, + HS_Matrix_K* hsk, + HContainer* hR, + HContainer* sR) +{ + LcaoOpsBundle bundle; + + // fix HR to gamma case, where SR will be fixed in Overlap Operator + hR->fix_gamma(); + // initial operator for Gamma_only case + // overlap term () is indispensable + // in Gamma_only case, target SK is hsk->get_sk(), the target SR is sR + Operator* ops = new Overlap>(hsk, + kv->kvec_d, hR, sR, + &ucell, orb.cutoffs(), &grid_d, + two_center_bundle.overlap_orb.get()); + + // kinetic term () + if (inp.t_in_h) + { + Operator* ekinetic = new EKinetic>(hsk, + kv->kvec_d, hR, + &ucell, orb.cutoffs(), &grid_d, + two_center_bundle.kinetic_orb.get()); + ops->add(ekinetic); + } + + // nonlocal term (D) + // in general case, target HR is hR, while target HK is hsk->get_hk() + if (inp.vnl_in_h) + { + Operator* nonlocal = new Nonlocal>(hsk, + kv->kvec_d, hR, + &ucell, orb.cutoffs(), &grid_d, + two_center_bundle.overlap_orb_beta.get()); + ops->add(nonlocal); + } + + // Effective potential term (\sum_r ) + // in general case, target HR is Gint::hRGint, while target HK is hsk->get_hk() + if (inp.vl_in_h) + { + // only Potential is not empty, Veff and Meta are available + if (pot_register_in.size() > 0) + { + // register Potential by gathered operator + pot_in->pot_register(pot_register_in); + // effective potential term + Operator* veff = new Veff>(hsk, + kv->kvec_d, pot_in, + hR, // no explicit call yet + &ucell, orb.cutoffs(), &grid_d, + inp.nspin); + ops->add(veff); + } + } + +#ifdef __MLALGO + if (inp.deepks_scf) + { + bundle.v_delta_R = add_deepks_op(ops, ucell, grid_d, two_center_bundle, + orb, DM_in, deepks, kv, hsk, hR); + } +#endif + + // end node should be DFTU_firstzeta + if (inp.dft_plus_u) + { + add_dftu_op(ops, ucell, grid_d, two_center_bundle, orb, DM_in, + p_dftu, inp, kv, hsk, hR); + } + + bundle.ops = dynamic_cast*>(ops); + return bundle; +} + +// build the operator chain for the multi-k case (TK == complex) +template +LcaoOpsBundle build_multik_ops(const UnitCell& ucell, + const Grid_Driver& grid_d, + const Parallel_Orbitals* paraV, + elecstate::Potential* pot_in, + const TwoCenterBundle& two_center_bundle, + const LCAO_Orbitals& orb, + module_dm::DensityMatrix* DM_in, + Plus_U_Base* p_dftu, + Setup_DeePKS& deepks, + const Input_para& inp, + const std::vector& pot_register_in, + const K_Vectors* kv, + HS_Matrix_K* hsk, + HContainer* hR, + HContainer* sR) +{ + LcaoOpsBundle bundle; + + Operator* ops = nullptr; + + // Effective potential term (\sum_r ) + // Meta potential term (\sum_r ) + // in general case, target HR is Gint::pvpR_reduced, while target HK is hsk->get_hk() + if (inp.vl_in_h) + { + // only Potential is not empty, Veff and Meta are available + if (pot_register_in.size() > 0) + { + // register Potential by gathered operator + pot_in->pot_register(pot_register_in); + // Veff term + ops = new Veff>(hsk, + kv->kvec_d, pot_in, + hR, + &ucell, orb.cutoffs(), &grid_d, + inp.nspin); + } + } + + // initial operator for multi-k case + // overlap term is indispensable + Operator* overlap = new Overlap>(hsk, + kv->kvec_d, hR, sR, + &ucell, orb.cutoffs(), &grid_d, + two_center_bundle.overlap_orb.get()); + if (ops == nullptr) + { + ops = overlap; + } + else + { + ops->add(overlap); + } + + // kinetic term (), + // in general case, target HR is hR, while target HK is hsk->get_hk() + if (inp.t_in_h) + { + Operator* ekinetic = new EKinetic>(hsk, + kv->kvec_d, hR, + &ucell, orb.cutoffs(), &grid_d, + two_center_bundle.kinetic_orb.get()); + ops->add(ekinetic); + } + + // nonlocal term (D) + // in general case, target HR is hR, while target HK is hsk->get_hk() + // TDDFT velocity gauge will calculate full non-local potential including the original one and the + // correction on its own, so the original non-local potential term should be skipped then + if (inp.vnl_in_h) + { + Operator* nonlocal = new Nonlocal>(hsk, + kv->kvec_d, hR, + &ucell, orb.cutoffs(), &grid_d, + two_center_bundle.overlap_orb_beta.get()); + // Nonlocal::initialize_HR inserts atom pairs into hR (using a cutoff + // that includes the nonlocal pseudopotential radius). TDEkinetic and + // TDNonlocal both build hR_tmp by iterating over hR's pairs, so the + // Nonlocal constructor must run even when the operator itself is not + // added to the chain (TDDFT velocity gauge case). + if (inp.esolver_type != "tddft" || elecstate::H_TDDFT_pw::stype != 1) + { + ops->add(nonlocal); + } + else + { + delete nonlocal; + } + } + +#ifdef __MLALGO + if (inp.deepks_scf) + { + bundle.v_delta_R = add_deepks_op(ops, ucell, grid_d, two_center_bundle, + orb, DM_in, deepks, kv, hsk, hR); + } +#endif + // TDDFT_velocity_gauge + // These operators are complex-only (no double instantiation of + // TDEkinetic/TDNonlocal). The std::is_same guard lets the compiler + // dead-branch-eliminate this block in the double instantiation, avoiding + // references to missing double symbols. + if (std::is_same>::value && inp.esolver_type == "tddft" && inp.td_stype == 1) + { + Operator* td_ekinetic = new TDEkinetic>(hsk, + hR, kv, + &ucell, orb.cutoffs(), &grid_d, + two_center_bundle.overlap_orb.get()); + ops->add(td_ekinetic); + + Operator* td_nonlocal = new TDNonlocal>(hsk, + kv->kvec_d, hR, + &ucell, orb, &grid_d); + ops->add(td_nonlocal); + } + if (inp.esolver_type == "tddft" && inp.td_stype == 2) + { + Operator* td_pot_hybrid = new TD_pot_hybrid>(hsk, + kv, hR, sR, + orb, &ucell, orb.cutoffs(), &grid_d, + two_center_bundle.kinetic_orb.get()); + ops->add(td_pot_hybrid); + } + if (inp.dft_plus_u) + { + add_dftu_op(ops, ucell, grid_d, two_center_bundle, orb, DM_in, + p_dftu, inp, kv, hsk, hR); + } + if (inp.sc_mag_switch) + { + Operator* sc_lambda = new DeltaSpin>(hsk, + kv->kvec_d, hR, + ucell, &grid_d, + two_center_bundle.overlap_orb_onsite.get(), + orb.cutoffs()); + ops->add(sc_lambda); + spinconstrain::SpinConstrain& sc = spinconstrain::SpinConstrain::getScInstance(); + sc.set_operator(sc_lambda); + } + + bundle.ops = dynamic_cast*>(ops); + return bundle; +} + +// explicit instantiation: gamma-only, multi-k, and non-collinear spin cases +template struct LcaoOpsBundle; +template struct LcaoOpsBundle, double>; +template struct LcaoOpsBundle, std::complex>; + +template LcaoOpsBundle build_gamma_ops( + const UnitCell&, const Grid_Driver&, const Parallel_Orbitals*, + elecstate::Potential*, const TwoCenterBundle&, const LCAO_Orbitals&, + module_dm::DensityMatrix*, Plus_U_Base*, Setup_DeePKS&, + const Input_para&, const std::vector&, const K_Vectors*, + HS_Matrix_K*, HContainer*, HContainer*); + +template LcaoOpsBundle build_multik_ops( + const UnitCell&, const Grid_Driver&, const Parallel_Orbitals*, + elecstate::Potential*, const TwoCenterBundle&, const LCAO_Orbitals&, + module_dm::DensityMatrix*, Plus_U_Base*, Setup_DeePKS&, + const Input_para&, const std::vector&, const K_Vectors*, + HS_Matrix_K*, HContainer*, HContainer*); + +template LcaoOpsBundle, double> build_gamma_ops, double>( + const UnitCell&, const Grid_Driver&, const Parallel_Orbitals*, + elecstate::Potential*, const TwoCenterBundle&, const LCAO_Orbitals&, + module_dm::DensityMatrix, double>*, Plus_U_Base*, + Setup_DeePKS>&, const Input_para&, + const std::vector&, const K_Vectors*, + HS_Matrix_K>*, HContainer*, HContainer*); + +template LcaoOpsBundle, double> build_multik_ops, double>( + const UnitCell&, const Grid_Driver&, const Parallel_Orbitals*, + elecstate::Potential*, const TwoCenterBundle&, const LCAO_Orbitals&, + module_dm::DensityMatrix, double>*, Plus_U_Base*, + Setup_DeePKS>&, const Input_para&, + const std::vector&, const K_Vectors*, + HS_Matrix_K>*, HContainer*, HContainer*); + +template LcaoOpsBundle, std::complex> +build_gamma_ops, std::complex>( + const UnitCell&, const Grid_Driver&, const Parallel_Orbitals*, + elecstate::Potential*, const TwoCenterBundle&, const LCAO_Orbitals&, + module_dm::DensityMatrix, double>*, Plus_U_Base*, + Setup_DeePKS>&, const Input_para&, + const std::vector&, const K_Vectors*, + HS_Matrix_K>*, HContainer>*, + HContainer>*); + +template LcaoOpsBundle, std::complex> +build_multik_ops, std::complex>( + const UnitCell&, const Grid_Driver&, const Parallel_Orbitals*, + elecstate::Potential*, const TwoCenterBundle&, const LCAO_Orbitals&, + module_dm::DensityMatrix, double>*, Plus_U_Base*, + Setup_DeePKS>&, const Input_para&, + const std::vector&, const K_Vectors*, + HS_Matrix_K>*, HContainer>*, + HContainer>*); + +} // namespace hamilt diff --git a/source/source_lcao/hamilt_lcao_factory.h b/source/source_lcao/hamilt_lcao_factory.h new file mode 100644 index 00000000000..0ebe27648f2 --- /dev/null +++ b/source/source_lcao/hamilt_lcao_factory.h @@ -0,0 +1,96 @@ +#ifndef HAMILT_LCAO_FACTORY_H +#define HAMILT_LCAO_FACTORY_H + +#include "source_basis/module_ao/parallel_orbitals.h" +#include "source_basis/module_nao/two_center_bundle.h" +#include "source_cell/module_neighbor/sltk_grid_driver.h" +#include "source_cell/unitcell.h" +#include "source_estate/module_pot/potential_new.h" +#include "source_estate/module_dm/density_matrix.h" +#include "source_hamilt/module_hcontainer/hcontainer.h" +#include "source_hamilt/hs_matrix_k.h" +#include "source_lcao/module_operator_lcao/operator_lcao.h" +#include "source_lcao/setup_deepks.h" +#include "source_pw/module_pwdft/dftu_base.h" + +#include +#include + +namespace hamilt +{ + +/** + * @brief products of the LCAO operator-chain factory. + * + * The factory builds the operator chain on top of the already-allocated + * hsk/hR/sR buffers (owned by the caller); it only produces the chain head + * and, for MLALGO builds, the DeePKS V_delta(R) handle. + */ +template +struct LcaoOpsBundle +{ + OperatorLCAO* ops = nullptr; ///< built operator-chain head + HContainer* v_delta_R = nullptr; ///< DeePKS V_delta(R), MLALGO only +}; + +/** + * @brief build the operator chain for the gamma-only case (TK == double). + * + * Appends overlap/kinetic/nonlocal/veff nodes (and optional DeePKS/DFTU) + * onto the buffers hsk/hR/sR. hR is gamma-fixed here. + * + * @param kv k-point list (kvec_d/isk read from it) + * @param hsk target H(k)/S(k) matrix buffer, already allocated by caller + * @param hR target H(R) container, already allocated by caller + * @param sR target S(R) container, already allocated by caller + * @return LcaoOpsBundle with the chain head and DeePKS V_delta(R) handle + */ +template +LcaoOpsBundle build_gamma_ops(const UnitCell& ucell, + const Grid_Driver& grid_d, + const Parallel_Orbitals* paraV, + elecstate::Potential* pot_in, + const TwoCenterBundle& two_center_bundle, + const LCAO_Orbitals& orb, + module_dm::DensityMatrix* DM_in, + Plus_U_Base* p_dftu, + Setup_DeePKS& deepks, + const Input_para& inp, + const std::vector& pot_register_in, + const K_Vectors* kv, + HS_Matrix_K* hsk, + HContainer* hR, + HContainer* sR); + +/** + * @brief build the operator chain for the multi-k case (TK == complex). + * + * Appends veff/overlap/kinetic/nonlocal nodes (and optional DeePKS/TDDFT/ + * DFTU/spin-constrain) onto the buffers hsk/hR/sR. + * + * @param kv k-point list (kvec_d/isk read from it) + * @param hsk target H(k)/S(k) matrix buffer, already allocated by caller + * @param hR target H(R) container, already allocated by caller + * @param sR target S(R) container, already allocated by caller + * @return LcaoOpsBundle with the chain head and DeePKS V_delta(R) handle + */ +template +LcaoOpsBundle build_multik_ops(const UnitCell& ucell, + const Grid_Driver& grid_d, + const Parallel_Orbitals* paraV, + elecstate::Potential* pot_in, + const TwoCenterBundle& two_center_bundle, + const LCAO_Orbitals& orb, + module_dm::DensityMatrix* DM_in, + Plus_U_Base* p_dftu, + Setup_DeePKS& deepks, + const Input_para& inp, + const std::vector& pot_register_in, + const K_Vectors* kv, + HS_Matrix_K* hsk, + HContainer* hR, + HContainer* sR); + +} // namespace hamilt + +#endif diff --git a/source/source_lcao/lcao_domain.h b/source/source_lcao/lcao_domain.h index 38aaafc9522..305e147f447 100644 --- a/source/source_lcao/lcao_domain.h +++ b/source/source_lcao/lcao_domain.h @@ -5,7 +5,7 @@ #include "source_base/vector3.h" #include "source_basis/module_nao/two_center_bundle.h" #include "source_cell/module_neighbor/sltk_grid_driver.h" -#include "source_lcao/lcao_hs_arrays.hpp" +#include "source_lcao/lcao_hs_arrays.h" #include "source_lcao/force_stress_arrays.h" #include "source_lcao/module_deepks/lcao_deepks.h" #include "source_basis/module_ao/parallel_orbitals.h" @@ -50,57 +50,92 @@ void set_force(const Parallel_Orbitals& pv, double* dhloc_fixed_z); /** - * @brief set the elements of stress-related matrices in LCAO method + * @brief read-only environment for building S/T matrix elements. + * + * Everything here is fixed for the duration of one build_ST_new call: + * the basis, the parallel layout, the unit cell and the spin/polarization + * configuration. Passed by const reference into single_overlap / + * single_deriv so those functions no longer read global INPUT state. */ -void set_stress(const Parallel_Orbitals& pv, - const int& iw1_all, - const int& iw2_all, - const double& vx, - const double& vy, - const double& vz, - const char& dtype, - const ModuleBase::Vector3& dtau, - double* dsloc_11, - double* dsloc_12, - double* dsloc_13, - double* dsloc_22, - double* dsloc_23, - double* dsloc_33, - double* dhloc_fixed_11, - double* dhloc_fixed_12, - double* dhloc_fixed_13, - double* dhloc_fixed_22, - double* dhloc_fixed_23, - double* dhloc_fixed_33); +struct ST_env +{ + const LCAO_Orbitals& orb; + const TwoCenterBundle& two_center_bundle; + const Parallel_Orbitals& pv; + const UnitCell& ucell; + const int nspin; + const int npol; + const bool cal_stress; + const bool gamma_only_local; +}; + +/** + * @brief one S/T matrix element . + * + * All inputs that vary per matrix element inside the build_ST_new loops: + * the operator type, the global orbital indices, the angular quantum + * numbers of both centres and their displacement. + */ +struct ST_elem +{ + const char dtype; + const int iw1_all; + const int iw2_all; + const int m1; + const int m2; + const int t1; + const int l1; + const int n1; + const int t2; + const int l2; + const int n2; + const ModuleBase::Vector3 dtau; + const int jj; + const int jj0; + const int kk; + const int kk0; +}; + +/** + * @brief read-only environment for building the nonlocal + * matrix elements. + * + * Everything here is fixed for the duration of one build_Nonlocal_mu_new + * call: the parallel layout, the unit cell and the spin/polarization + * configuration, plus whether derivatives (forces) are requested. Passed by + * const reference into the per-element accumulation helpers so they no longer + * read global INPUT state. + */ +struct NL_env +{ + const Parallel_Orbitals& pv; + const UnitCell& ucell; + const int nspin; + const int npol; + const bool gamma_only_local; + const bool calc_deri; +}; + +/** + * @brief identity of one nonlocal matrix element . + * + * The per-matrix-element inputs inside the build_Nonlocal_mu_new loops: the + * global orbital indices of both centres, the projector atom type and the + * flat sparse slot this element accumulates into. + */ +struct NL_elem +{ + const int iw1_all; + const int iw2_all; + const int t0; + const int nnr; +}; /** * @brief set each element without derivatives */ -void single_overlap(const LCAO_Orbitals& orb, - const TwoCenterBundle& two_center_bundle, - const Parallel_Orbitals& pv, - const UnitCell& ucell, - const int nspin, - const bool cal_stress, - const int iw1_all, - const int iw2_all, - const int m1, - const int m2, - const char& dtype, - const int T1, - const int L1, - const int N1, - const int T2, - const int L2, - const int N2, - const ModuleBase::Vector3& dtau, - const ModuleBase::Vector3& tau1, - const ModuleBase::Vector3& tau2, - const int npol, - const int jj, - const int jj0, - const int kk, - const int kk0, +void single_overlap(const ST_env& env, + const ST_elem& e, int& nnr, // output value int& total_nnr, // output value double* olm, // output value @@ -109,35 +144,12 @@ void single_overlap(const LCAO_Orbitals& orb, /** * @brief set each element of T matrices */ -void single_derivative(ForceStressArrays& fsr, - const LCAO_Orbitals& orb, - const TwoCenterBundle& two_center_bundle, - const Parallel_Orbitals& pv, - const UnitCell& ucell, - const int nspin, - const bool cal_stress, - const int iw1_all, - const int iw2_all, - const int m1, - const int m2, - const char& dtype, - const int T1, - const int L1, - const int N1, - const int T2, - const int L2, - const int N2, - const ModuleBase::Vector3& dtau, - const ModuleBase::Vector3& tau1, - const ModuleBase::Vector3& tau2, - const int npol, - const int jj, - const int jj0, - const int kk, - const int kk0, - int& nnr, // output value - int& total_nnr, // output value - double* olm); // output value +void single_deriv(const ST_env& env, + const ST_elem& e, + ForceStressArrays& fsr, + int& nnr, // output value + int& total_nnr, // output value + double* olm); // output value /** * @brief set the elements of S and T matrices diff --git a/source/source_lcao/lcao_hs_arrays.hpp b/source/source_lcao/lcao_hs_arrays.h similarity index 100% rename from source/source_lcao/lcao_hs_arrays.hpp rename to source/source_lcao/lcao_hs_arrays.h diff --git a/source/source_lcao/lcao_init_basis.cpp b/source/source_lcao/lcao_init_basis.cpp index d3e2fa35a02..ad57ebb2ac7 100644 --- a/source/source_lcao/lcao_init_basis.cpp +++ b/source/source_lcao/lcao_init_basis.cpp @@ -13,7 +13,7 @@ void init_basis_lcao(Parallel_Orbitals& pv, const double &lcao_dk, const double &lcao_dr, const double &lcao_rmax, - UnitCell& ucell, + UnitCell& ucell, TwoCenterBundle& two_center_bundle, LCAO_Orbitals& orb ) @@ -58,9 +58,9 @@ void init_basis_lcao(Parallel_Orbitals& pv, auto* lcao_nl = new LCAONonlocalInfo(); lcao_nl->setupNonlocal(ucell.ntype, ucell.atoms, GlobalV::ofs_running, orb, PARAM.inp.basis_type, PARAM.inp.out_element_info, - PARAM.inp.lspinorb, PARAM.inp.nspin); + PARAM.inp.lspinorb, PARAM.inp.nspin, GlobalV::MY_RANK); ucell.infoNL.reset(lcao_nl); - two_center_bundle.build_beta(ucell.ntype, lcao_nl->get_nonlocal().Beta); + two_center_bundle.build_beta(ucell.ntype, lcao_nl->get_nonlocal().get_Beta_data()); } #ifdef __FFT_TWO_CENTER diff --git a/source/source_lcao/lcao_nl_mu.cpp b/source/source_lcao/lcao_nl_mu.cpp index 2e75222a553..fbdffe4271d 100644 --- a/source/source_lcao/lcao_nl_mu.cpp +++ b/source/source_lcao/lcao_nl_mu.cpp @@ -7,51 +7,216 @@ namespace LCAO_domain typedef std::tuple key_tuple; -#include "record_adj.h" //mohan add 2012-07-06 +#include "source_cell/record_adj.h" //mohan add 2012-07-06 -void build_Nonlocal_mu_new(const Parallel_Orbitals& pv, - ForceStressArrays& fsr, - double* NLloc, - const bool& calc_deri, - const UnitCell& ucell, - const LCAO_Orbitals& orb, - const TwoCenterIntegrator& intor_orb_beta, - const Grid_Driver* GridD) +// Per-neighbour context for Step 2 of build_Nonlocal_mu_new: everything that +// is fixed once the projector atom (T0, slot iat) and the two basis centres +// are known, before the (j, k) orbital loop runs. key1/key2 address the two +// blocks inside nlm_tot / nlm_tot1. +struct NL_pair { - ModuleBase::TITLE("LCAO_domain", "vnl_mu_new"); - ModuleBase::timer::start("LCAO_domain", "vnl_mu_new"); + const Atom* atom1; + const Atom* atom2; + const int start1; + const int start2; + const int t0; + const int iat; + const key_tuple key1; + const key_tuple key2; +}; + +// contribution of one matrix element to the nonlocal +// energy (no derivatives). Defined statically here so the compiler can inline +// it at the hot inner-loop call site of build_Nonlocal_mu_new. +static void accum_nlm_energy(const NL_env& env, + const NL_elem& e, + const std::unordered_map>& nlm_cur1, + const std::unordered_map>& nlm_cur2, + double* NLloc) +{ + const std::unordered_map>::const_iterator it1 = nlm_cur1.find(e.iw1_all); + const std::unordered_map>::const_iterator it2 = nlm_cur2.find(e.iw2_all); + if (it1 == nlm_cur1.end() || it2 == nlm_cur2.end()) + { + return; + } + const std::vector& nlm_1 = it1->second; + const std::vector& nlm_2 = it2->second; + double nlm_tmp = 0.0; + const double* tmp_d = nullptr; + for (int no = 0; no < env.ucell.atoms[e.t0].ncpp.non_zero_count_soc[0]; no++) + { + const int p1 = env.ucell.atoms[e.t0].ncpp.index1_soc[0][no]; + const int p2 = env.ucell.atoms[e.t0].ncpp.index2_soc[0][no]; + env.ucell.atoms[e.t0].ncpp.get_d(0, p1, p2, tmp_d); + nlm_tmp += nlm_2[p2] * nlm_1[p1] * (*tmp_d); + } - const int nspin = PARAM.inp.nspin; - const int npol = PARAM.globalv.npol; - const bool gamma_only_local = PARAM.globalv.gamma_only_local; + if (nlm_tmp != 0.0) + { + if (env.gamma_only_local) + { + // mohan add 2010-12-20 + LCAO_domain::set_mat2d(e.iw1_all, e.iw2_all, nlm_tmp, env.pv, NLloc); // N stands for nonlocal. + } + else + { + NLloc[e.nnr] += nlm_tmp; + } + } +} - // < phi1 | beta > < beta | phi2 > - // phi1 is within the unitcell. - // while beta is in the supercell. - // while phi2 is in the supercell. +// force contribution of one matrix element for nspin==4 (SOC): the four +// spin blocks is0 fold the d_so matrices into DHloc_fixedR_{x,y,z}. +static void accum_nlm_force_soc(const NL_env& env, + const NL_elem& e, + const std::unordered_map>>& nlm_cur1, + const std::unordered_map>>& nlm_cur2, + const int is0, + ForceStressArrays& fsr) +{ + const std::unordered_map>>::const_iterator it2 = nlm_cur2.find(e.iw2_all); + const std::unordered_map>>::const_iterator it1 = nlm_cur1.find(e.iw1_all); + if (it2 == nlm_cur2.end() || it1 == nlm_cur1.end()) + { + return; + } + const std::vector& nlm_1 = it2->second[0]; + std::vector> nlm_2; + nlm_2.resize(3); + for (int i = 0; i < 3; i++) + { + nlm_2[i] = it1->second[i + 1]; + } + for (int no = 0; no < env.ucell.atoms[e.t0].ncpp.non_zero_count_soc[is0]; no++) + { + const int p1 = env.ucell.atoms[e.t0].ncpp.index1_soc[is0][no]; + const int p2 = env.ucell.atoms[e.t0].ncpp.index2_soc[is0][no]; + double coef = 0.0; + if (is0 == 0) + { + coef = (env.ucell.atoms[e.t0].ncpp.d_so(0, p2, p1).real() + + env.ucell.atoms[e.t0].ncpp.d_so(3, p2, p1).real()) + * 0.5; + } + else if (is0 == 1) + { + coef = (env.ucell.atoms[e.t0].ncpp.d_so(1, p2, p1).real() + + env.ucell.atoms[e.t0].ncpp.d_so(2, p2, p1).real()) + * 0.5; + } + else if (is0 == 2) + { + coef = (-env.ucell.atoms[e.t0].ncpp.d_so(1, p2, p1).imag() + + env.ucell.atoms[e.t0].ncpp.d_so(2, p2, p1).imag()) + * 0.5; + } + else if (is0 == 3) + { + coef = (env.ucell.atoms[e.t0].ncpp.d_so(0, p2, p1).real() + - env.ucell.atoms[e.t0].ncpp.d_so(3, p2, p1).real()) + * 0.5; + } + fsr.DHloc_fixedR_x[e.nnr] += nlm_2[0][p1] * nlm_1[p2] * coef; + fsr.DHloc_fixedR_y[e.nnr] += nlm_2[1][p1] * nlm_1[p2] * coef; + fsr.DHloc_fixedR_z[e.nnr] += nlm_2[2][p1] * nlm_1[p2] * coef; + } +} - // Step 1 : generate +// force contribution of one matrix element for nspin==1/2. gamma_only writes +// the cartesian force arrays, the multi-k branch accumulates DHloc_fixedR. +static void accum_nlm_force(const NL_env& env, + const NL_elem& e, + const std::unordered_map>>& nlm_cur1, + const std::unordered_map>>& nlm_cur2, + ForceStressArrays& fsr) +{ + double nlm[3] = {0, 0, 0}; + const std::unordered_map>>::const_iterator it1 = nlm_cur1.find(e.iw1_all); + const std::unordered_map>>::const_iterator it2 = nlm_cur2.find(e.iw2_all); + if (it1 == nlm_cur1.end() || it2 == nlm_cur2.end()) + { + return; + } + const std::vector* nlm_1 = nullptr; + std::vector> nlm_2; + nlm_2.resize(3); + if (env.gamma_only_local) + { + nlm_1 = &it1->second[0]; + for (int i = 0; i < 3; i++) + { + nlm_2[i] = it2->second[i + 1]; + } + } + else + { + // mohan change the order on 2011-06-17 + // origin: < psi1 | beta > < beta | dpsi2/dtau > + // now: < psi1/dtau | beta > < beta | psi2 > + nlm_1 = &it2->second[0]; + for (int i = 0; i < 3; i++) + { + nlm_2[i] = it1->second[i + 1]; + } + } - // This is the data structure for storing - // It is a 4 layer data structure - // The outmost layer is std::vector with size being number of atoms in unit cell - // The second layer is a map, the key being a combination of 4 number (iat, dRx, dRy, dRz) - // which identifies a unique adjacent atom of the first atom - // The third layer is an unordered map, with key being the index of atomic basis |psi> - // The inner layer is a vector, each element representing a projector |beta> - // It then either stores the number (nlm_tot) - // or a vector of 4, storing additionally (nlm_tot1) x_i=x,y,z - std::vector>>> nlm_tot; - std::vector>>>> nlm_tot1; + assert(nlm_1->size() == nlm_2[0].size()); - if (!calc_deri) + const double* tmp_d = nullptr; + for (int no = 0; no < env.ucell.atoms[e.t0].ncpp.non_zero_count_soc[0]; no++) { - nlm_tot.resize(ucell.nat); + const int p1 = env.ucell.atoms[e.t0].ncpp.index1_soc[0][no]; + const int p2 = env.ucell.atoms[e.t0].ncpp.index2_soc[0][no]; + env.ucell.atoms[e.t0].ncpp.get_d(0, p1, p2, tmp_d); + for (int ir = 0; ir < 3; ir++) + { + nlm[ir] += nlm_2[ir][p2] * (*nlm_1)[p1] * (*tmp_d); + } + } + + if (env.gamma_only_local) + { + LCAO_domain::set_force(env.pv, + e.iw1_all, + e.iw2_all, + nlm[0], + nlm[1], + nlm[2], + 'N', + fsr.DSloc_x.data(), + fsr.DSloc_y.data(), + fsr.DSloc_z.data(), + fsr.DHloc_fixed_x.data(), + fsr.DHloc_fixed_y.data(), + fsr.DHloc_fixed_z.data()); } else { - nlm_tot1.resize(ucell.nat); + fsr.DHloc_fixedR_x[e.nnr] += nlm[0]; + fsr.DHloc_fixedR_y[e.nnr] += nlm[1]; + fsr.DHloc_fixedR_z[e.nnr] += nlm[2]; } +} + +// Step 1 of build_Nonlocal_mu_new: generate (and, when derivatives +// are requested, ) for every atom and its adjacent atoms. Fills +// nlm_tot (energy) or nlm_tot1 (force); only one of the two is populated, +// selected by env.calc_deri. The iat loop is OpenMP-parallel; each thread owns +// its own iat slot of the output containers. +static void build_psi_beta( + const NL_env& env, + const LCAO_Orbitals& orb, + const TwoCenterIntegrator& intor_orb_beta, + const Grid_Driver* GridD, + std::vector>>>& nlm_tot, + std::vector>>>>& nlm_tot1) +{ + const UnitCell& ucell = env.ucell; + const Parallel_Orbitals& pv = env.pv; + const int npol = env.npol; + const bool calc_deri = env.calc_deri; + #ifdef _OPENMP #pragma omp parallel for schedule(dynamic) #endif @@ -95,22 +260,15 @@ void build_Nonlocal_mu_new(const Parallel_Orbitals& pv, std::unordered_map> nlm_cur; std::unordered_map>> nlm_cur1; - if (!calc_deri) - { - nlm_cur.clear(); - } - else - { - nlm_cur1.clear(); - } for (int iw1 = 0; iw1 < nw1_tot; ++iw1) { const int iw1_all = start1 + iw1; const int iw1_local = pv.global2local_row(iw1_all); const int iw2_local = pv.global2local_col(iw1_all); - if (iw1_local < 0 && iw2_local < 0) { + if (iw1_local < 0 && iw2_local < 0) + { continue; -} + } const int iw1_0 = iw1 / npol; std::vector> nlm; // nlm is a vector of vectors, but size of outer vector is only 1 here @@ -154,6 +312,132 @@ void build_Nonlocal_mu_new(const Parallel_Orbitals& pv, } } // end ad } +} + +// Step 2 inner block: run the (j, k) orbital loop for one projector atom and +// accumulate every matrix element into NLloc (energy) +// or fsr (force). Returns the number of elements written, i.e. the nnr_inner +// increment for this neighbour. +static int accum_nlm_block( + const NL_env& env, + const NL_pair& pair, + const int nnr, + std::vector>>>& nlm_tot, + std::vector>>>>& nlm_tot1, + double* NLloc, + ForceStressArrays& fsr) +{ + const Parallel_Orbitals& pv = env.pv; + const int npol = env.npol; + const int nspin = env.nspin; + const bool calc_deri = env.calc_deri; + + int nnr_inner = 0; + for (int j = 0; j < pair.atom1->nw * npol; j++) + { + const int j0 = j / npol; // added by zhengdy-soc + const int iw1_all = pair.start1 + j; + const int mu = pv.global2local_row(iw1_all); + if (mu < 0) + { + continue; + } + + // fix a serious bug: atom2[T2] -> atom2 + // mohan 2010-12-20 + for (int k = 0; k < pair.atom2->nw * npol; k++) + { + const int k0 = k / npol; + const int iw2_all = pair.start2 + k; + const int nu = pv.global2local_col(iw2_all); + if (nu < 0) + { + continue; + } + + const NL_elem elem{iw1_all, iw2_all, pair.t0, nnr + nnr_inner}; + if (!calc_deri) + { + accum_nlm_energy(env, elem, nlm_tot[pair.iat][pair.key1], nlm_tot[pair.iat][pair.key2], NLloc); + } // calc_deri + else // calculate the derivative + { + if (nspin == 4) + { + const int is0 = (j - j0 * npol) + (k - k0 * npol) * 2; + accum_nlm_force_soc(env, + elem, + nlm_tot1[pair.iat][pair.key1], + nlm_tot1[pair.iat][pair.key2], + is0, + fsr); + } + else if (nspin == 1 || nspin == 2) + { + accum_nlm_force(env, + elem, + nlm_tot1[pair.iat][pair.key1], + nlm_tot1[pair.iat][pair.key2], + fsr); + } + else + { + ModuleBase::WARNING_QUIT("LCAO_domain::build_Nonlocal_mu_new", "nspin must be 1, 2 or 4"); + } + } //! calc_deri + nnr_inner++; + } // k + } // j + return nnr_inner; +} + +void build_Nonlocal_mu_new(const Parallel_Orbitals& pv, + ForceStressArrays& fsr, + double* NLloc, + const bool& calc_deri, + const UnitCell& ucell, + const LCAO_Orbitals& orb, + const TwoCenterIntegrator& intor_orb_beta, + const Grid_Driver* GridD) +{ + ModuleBase::TITLE("LCAO_domain", "vnl_mu_new"); + ModuleBase::timer::start("LCAO_domain", "vnl_mu_new"); + + const int nspin = PARAM.inp.nspin; + const int npol = PARAM.globalv.npol; + const bool gamma_only_local = PARAM.globalv.gamma_only_local; + + const NL_env env{pv, ucell, nspin, npol, gamma_only_local, calc_deri}; + + // < phi1 | beta > < beta | phi2 > + // phi1 is within the unitcell. + // while beta is in the supercell. + // while phi2 is in the supercell. + + // Step 1 : generate + + // This is the data structure for storing + // It is a 4 layer data structure + // The outmost layer is std::vector with size being number of atoms in unit cell + // The second layer is a map, the key being a combination of 4 number (iat, dRx, dRy, dRz) + // which identifies a unique adjacent atom of the first atom + // The third layer is an unordered map, with key being the index of atomic basis |psi> + // The inner layer is a vector, each element representing a projector |beta> + // It then either stores the number (nlm_tot) + // or a vector of 4, storing additionally (nlm_tot1) x_i=x,y,z + std::vector>>> nlm_tot; + std::vector>>>> nlm_tot1; + + if (!calc_deri) + { + nlm_tot.resize(ucell.nat); + } + else + { + nlm_tot1.resize(ucell.nat); + } + + build_psi_beta(env, orb, intor_orb_beta, GridD, nlm_tot, nlm_tot1); //======================================================= // Step2: @@ -220,9 +504,11 @@ void build_Nonlocal_mu_new(const Parallel_Orbitals& pv, // this rcut is in order to make nnr consistent // with other matrix. rcut = pow(orb.Phi[T1].getRcut() + orb.Phi[T2].getRcut(), 2); - if (distance < rcut) { + if (distance < rcut) + { is_adj = true; - } else if (distance >= rcut) + } + else if (distance >= rcut) { for (int ad0 = 0; ad0 < adjs.adj_num + 1; ++ad0) { @@ -284,259 +570,8 @@ void build_Nonlocal_mu_new(const Parallel_Orbitals& pv, key_tuple key1(iat1, -rx0, -ry0, -rz0); key_tuple key2(iat2, rx2 - rx0, ry2 - ry0, rz2 - rz0); - std::unordered_map>* nlm_cur1_e; // left hand side, for energy - std::unordered_map>>* nlm_cur1_f; // lhs, for force - std::unordered_map>* nlm_cur2_e; // rhs, for energy - std::unordered_map>>* nlm_cur2_f; // rhs, for force - - if (!calc_deri) - { - nlm_cur1_e = &nlm_tot[iat][key1]; - nlm_cur2_e = &nlm_tot[iat][key2]; - } - else - { - nlm_cur1_f = &nlm_tot1[iat][key1]; - nlm_cur2_f = &nlm_tot1[iat][key2]; - } - - int nnr_inner = 0; - - for (int j = 0; j < atom1->nw * npol; j++) - { - const int j0 = j / npol; // added by zhengdy-soc - const int iw1_all = start1 + j; - const int mu = pv.global2local_row(iw1_all); - if (mu < 0) { - continue; -} - - // fix a serious bug: atom2[T2] -> atom2 - // mohan 2010-12-20 - for (int k = 0; k < atom2->nw * npol; k++) - { - const int k0 = k / npol; - const int iw2_all = start2 + k; - const int nu = pv.global2local_col(iw2_all); - if (nu < 0) { - continue; -} - - if (!calc_deri) - { - std::vector nlm_1 = (*nlm_cur1_e)[iw1_all]; - std::vector nlm_2 = (*nlm_cur2_e)[iw2_all]; - if (nspin == 2 || nspin == 1) - { - double nlm_tmp = 0.0; - const double* tmp_d = nullptr; - for (int no = 0; no < ucell.atoms[T0].ncpp.non_zero_count_soc[0]; no++) - { - const int p1 = ucell.atoms[T0].ncpp.index1_soc[0][no]; - const int p2 = ucell.atoms[T0].ncpp.index2_soc[0][no]; - ucell.atoms[T0].ncpp.get_d(0, p1, p2, tmp_d); - nlm_tmp += nlm_2[p2] * nlm_1[p1] * (*tmp_d); - } - - if (gamma_only_local) - { - // mohan add 2010-12-20 - if (nlm_tmp != 0.0) - { - LCAO_domain::set_mat2d(iw1_all, - iw2_all, - nlm_tmp, - pv, - NLloc); // N stands for nonlocal. - } - } - else - { - if (nlm_tmp != 0.0) - { - NLloc[nnr + nnr_inner] += nlm_tmp; - } - } - } // end nspin - } // calc_deri - else // calculate the derivative - { - if (nspin == 4) - { - std::vector nlm_1 = (*nlm_cur2_f)[iw2_all][0]; - std::vector> nlm_2; - nlm_2.resize(3); - for (int i = 0; i < 3; i++) - { - nlm_2[i] = (*nlm_cur1_f)[iw1_all][i + 1]; - } - std::complex nlm[4][3] = {ModuleBase::ZERO}; - int is0 = (j - j0 * npol) + (k - k0 * npol) * 2; - for (int no = 0; no < ucell.atoms[T0].ncpp.non_zero_count_soc[is0]; no++) - { - const int p1 = ucell.atoms[T0].ncpp.index1_soc[is0][no]; - const int p2 = ucell.atoms[T0].ncpp.index2_soc[is0][no]; - if (is0 == 0) - { - fsr.DHloc_fixedR_x[nnr + nnr_inner] - += nlm_2[0][p1] * nlm_1[p2] - * (ucell.atoms[T0].ncpp.d_so(0, p2, p1).real() - + ucell.atoms[T0].ncpp.d_so(3, p2, p1).real()) - * 0.5; - fsr.DHloc_fixedR_y[nnr + nnr_inner] - += nlm_2[1][p1] * nlm_1[p2] - * (ucell.atoms[T0].ncpp.d_so(0, p2, p1).real() - + ucell.atoms[T0].ncpp.d_so(3, p2, p1).real()) - * 0.5; - fsr.DHloc_fixedR_z[nnr + nnr_inner] - += nlm_2[2][p1] * nlm_1[p2] - * (ucell.atoms[T0].ncpp.d_so(0, p2, p1).real() - + ucell.atoms[T0].ncpp.d_so(3, p2, p1).real()) - * 0.5; - } - else if (is0 == 1) - { - fsr.DHloc_fixedR_x[nnr + nnr_inner] - += nlm_2[0][p1] * nlm_1[p2] - * (ucell.atoms[T0].ncpp.d_so(1, p2, p1).real() - + ucell.atoms[T0].ncpp.d_so(2, p2, p1).real()) - * 0.5; - fsr.DHloc_fixedR_y[nnr + nnr_inner] - += nlm_2[1][p1] * nlm_1[p2] - * (ucell.atoms[T0].ncpp.d_so(1, p2, p1).real() - + ucell.atoms[T0].ncpp.d_so(2, p2, p1).real()) - * 0.5; - fsr.DHloc_fixedR_z[nnr + nnr_inner] - += nlm_2[2][p1] * nlm_1[p2] - * (ucell.atoms[T0].ncpp.d_so(1, p2, p1).real() - + ucell.atoms[T0].ncpp.d_so(2, p2, p1).real()) - * 0.5; - } - else if (is0 == 2) - { - fsr.DHloc_fixedR_x[nnr + nnr_inner] - += nlm_2[0][p1] * nlm_1[p2] - * (-ucell.atoms[T0].ncpp.d_so(1, p2, p1).imag() - + ucell.atoms[T0].ncpp.d_so(2, p2, p1).imag()) - * 0.5; - fsr.DHloc_fixedR_y[nnr + nnr_inner] - += nlm_2[1][p1] * nlm_1[p2] - * (-ucell.atoms[T0].ncpp.d_so(1, p2, p1).imag() - + ucell.atoms[T0].ncpp.d_so(2, p2, p1).imag()) - * 0.5; - fsr.DHloc_fixedR_z[nnr + nnr_inner] - += nlm_2[2][p1] * nlm_1[p2] - * (-ucell.atoms[T0].ncpp.d_so(1, p2, p1).imag() - + ucell.atoms[T0].ncpp.d_so(2, p2, p1).imag()) - * 0.5; - } - else if (is0 == 3) - { - fsr.DHloc_fixedR_x[nnr + nnr_inner] - += nlm_2[0][p1] * nlm_1[p2] - * (ucell.atoms[T0].ncpp.d_so(0, p2, p1).real() - - ucell.atoms[T0].ncpp.d_so(3, p2, p1).real()) - * 0.5; - fsr.DHloc_fixedR_y[nnr + nnr_inner] - += nlm_2[1][p1] * nlm_1[p2] - * (ucell.atoms[T0].ncpp.d_so(0, p2, p1).real() - - ucell.atoms[T0].ncpp.d_so(3, p2, p1).real()) - * 0.5; - fsr.DHloc_fixedR_z[nnr + nnr_inner] - += nlm_2[2][p1] * nlm_1[p2] - * (ucell.atoms[T0].ncpp.d_so(0, p2, p1).real() - - ucell.atoms[T0].ncpp.d_so(3, p2, p1).real()) - * 0.5; - } - } - } - else if (nspin == 1 || nspin == 2) - { - if (gamma_only_local) - { - double nlm[3] = {0, 0, 0}; - - // sum all projectors for one atom. - std::vector nlm_1 = (*nlm_cur1_f)[iw1_all][0]; - std::vector> nlm_2; - nlm_2.resize(3); - for (int i = 0; i < 3; i++) - { - nlm_2[i] = (*nlm_cur2_f)[iw2_all][i + 1]; - } - - assert(nlm_1.size() == nlm_2[0].size()); - - const double* tmp_d = nullptr; - for (int no = 0; no < ucell.atoms[T0].ncpp.non_zero_count_soc[0]; no++) - { - const int p1 = ucell.atoms[T0].ncpp.index1_soc[0][no]; - const int p2 = ucell.atoms[T0].ncpp.index2_soc[0][no]; - ucell.atoms[T0].ncpp.get_d(0, p1, p2, tmp_d); - for (int ir = 0; ir < 3; ir++) - { - nlm[ir] += nlm_2[ir][p2] * nlm_1[p1] * (*tmp_d); - } - } - - LCAO_domain::set_force(pv, - iw1_all, - iw2_all, - nlm[0], - nlm[1], - nlm[2], - 'N', - fsr.DSloc_x, - fsr.DSloc_y, - fsr.DSloc_z, - fsr.DHloc_fixed_x, - fsr.DHloc_fixed_y, - fsr.DHloc_fixed_z); - } - else - { - // mohan change the order on 2011-06-17 - // origin: < psi1 | beta > < beta | dpsi2/dtau > - // now: < psi1/dtau | beta > < beta | psi2 > - double nlm[3] = {0, 0, 0}; - - // sum all projectors for one atom. - std::vector nlm_1 = (*nlm_cur2_f)[iw2_all][0]; - std::vector> nlm_2; - nlm_2.resize(3); - for (int i = 0; i < 3; i++) - { - nlm_2[i] = (*nlm_cur1_f)[iw1_all][i + 1]; - } - - assert(nlm_1.size() == nlm_2[0].size()); - - const double* tmp_d = nullptr; - for (int no = 0; no < ucell.atoms[T0].ncpp.non_zero_count_soc[0]; no++) - { - const int p1 = ucell.atoms[T0].ncpp.index1_soc[0][no]; - const int p2 = ucell.atoms[T0].ncpp.index2_soc[0][no]; - ucell.atoms[T0].ncpp.get_d(0, p1, p2, tmp_d); - for (int ir = 0; ir < 3; ir++) - { - nlm[ir] += nlm_2[ir][p2] * nlm_1[p1] * (*tmp_d); - } - } - - fsr.DHloc_fixedR_x[nnr + nnr_inner] += nlm[0]; - fsr.DHloc_fixedR_y[nnr + nnr_inner] += nlm[1]; - fsr.DHloc_fixedR_z[nnr + nnr_inner] += nlm[2]; - } - } - else - { - ModuleBase::WARNING_QUIT("LCAO_domain::build_Nonlocal_mu_new", - "nspin must be 1, 2 or 4"); - } - } //! calc_deri - nnr_inner++; - } // k - } // j + const NL_pair pair{atom1, atom2, start1, start2, T0, iat, key1, key2}; + accum_nlm_block(env, pair, nnr, nlm_tot, nlm_tot1, NLloc, fsr); } // ad0 // outer circle : accumulate nnr diff --git a/source/source_lcao/lcao_nonlocal_info.h b/source/source_lcao/lcao_nonlocal_info.h index ed67d3433d8..7878af8b8d2 100644 --- a/source/source_lcao/lcao_nonlocal_info.h +++ b/source/source_lcao/lcao_nonlocal_info.h @@ -39,7 +39,7 @@ class LCAONonlocalInfo : public NonlocalInfoBase { * @return Number of projectors. */ int get_nproj(const int& type_in) const override { - return nonlocal.nproj[type_in]; + return nonlocal.get_nproj(type_in); } /** @@ -47,7 +47,7 @@ class LCAONonlocalInfo : public NonlocalInfoBase { * @return Maximum nproj value. */ int get_nprojmax() const override { - return nonlocal.nprojmax; + return nonlocal.get_nprojmax(); } /** @@ -56,7 +56,7 @@ class LCAONonlocalInfo : public NonlocalInfoBase { * @return Cutoff radius. */ double get_rcut_max(const int& type_in) const override { - return nonlocal.Beta[type_in].get_rcut_max(); + return nonlocal.get_Beta(type_in).get_rcut_max(); } /** @@ -65,7 +65,7 @@ class LCAONonlocalInfo : public NonlocalInfoBase { * @return const reference to label string. */ const std::string& get_label(const int& type_in) const override { - return nonlocal.Beta[type_in].getLabel(); + return nonlocal.get_Beta(type_in).getLabel(); } /** @@ -74,7 +74,7 @@ class LCAONonlocalInfo : public NonlocalInfoBase { * @return Type index. */ int get_type(const int& type_in) const override { - return nonlocal.Beta[type_in].getType(); + return nonlocal.get_Beta(type_in).getType(); } /** @@ -84,7 +84,7 @@ class LCAONonlocalInfo : public NonlocalInfoBase { * @return Angular momentum L. */ int get_proj_L(const int& type_in, const int& ip_in) const override { - return nonlocal.Beta[type_in].Proj[ip_in].getL(); + return nonlocal.get_Beta(type_in).Proj[ip_in].getL(); } /** @@ -94,7 +94,7 @@ class LCAONonlocalInfo : public NonlocalInfoBase { * @return Number of radial mesh points. */ int get_proj_Nr(const int& type_in, const int& ip_in) const override { - return nonlocal.Beta[type_in].Proj[ip_in].getNr(); + return nonlocal.get_Beta(type_in).Proj[ip_in].getNr(); } /** @@ -104,7 +104,7 @@ class LCAONonlocalInfo : public NonlocalInfoBase { * @return const pointer to radial mesh array. */ const double* get_proj_radial(const int& type_in, const int& ip_in) const override { - return nonlocal.Beta[type_in].Proj[ip_in].getRadial(); + return nonlocal.get_Beta(type_in).Proj[ip_in].getRadial(); } /** @@ -114,7 +114,7 @@ class LCAONonlocalInfo : public NonlocalInfoBase { * @return const pointer to beta_r array. */ const double* get_proj_beta_r(const int& type_in, const int& ip_in) const override { - return nonlocal.Beta[type_in].Proj[ip_in].getBeta_r(); + return nonlocal.get_Beta(type_in).Proj[ip_in].getBeta_r(); } /** @@ -124,7 +124,7 @@ class LCAONonlocalInfo : public NonlocalInfoBase { * @return Number of k-space mesh points. */ int get_proj_Nk(const int& type_in, const int& ip_in) const override { - return nonlocal.Beta[type_in].Proj[ip_in].getNk(); + return nonlocal.get_Beta(type_in).Proj[ip_in].getNk(); } /** @@ -134,7 +134,7 @@ class LCAONonlocalInfo : public NonlocalInfoBase { * @return Delta k value. */ double get_proj_dk(const int& type_in, const int& ip_in) const override { - return nonlocal.Beta[type_in].Proj[ip_in].getDk(); + return nonlocal.get_Beta(type_in).Proj[ip_in].getDk(); } /** @@ -144,7 +144,7 @@ class LCAONonlocalInfo : public NonlocalInfoBase { * @return Delta r uniform value. */ double get_proj_dr_uniform(const int& type_in, const int& ip_in) const override { - return nonlocal.Beta[type_in].Proj[ip_in].getDruniform(); + return nonlocal.get_Beta(type_in).Proj[ip_in].getDruniform(); } /** @@ -162,8 +162,9 @@ class LCAONonlocalInfo : public NonlocalInfoBase { const std::string& basis_type, const bool& out_element_info, const bool& lspinorb, - const int& nspin) { - nonlocal.setupNonlocal(ntype_in, atoms_in, log, orb, basis_type, out_element_info, lspinorb, nspin); + const int& nspin, + const int& my_rank) { + nonlocal.setupNonlocal(ntype_in, atoms_in, log, orb, basis_type, out_element_info, lspinorb, nspin, my_rank); } /** diff --git a/source/source_lcao/lcao_set.cpp b/source/source_lcao/lcao_set.cpp index b4ecedc8c2c..a512a89bf7d 100644 --- a/source/source_lcao/lcao_set.cpp +++ b/source/source_lcao/lcao_set.cpp @@ -6,18 +6,19 @@ #include "source_hamilt/module_hcontainer/read_hcontainer.h" #include "source_lcao/rho_tau_lcao.h" // use dm2rho #include "source_lcao/hamilt_lcao.h" // use HamiltLCAO for init_chg_hr +#include "source_hamilt/hamilt_hs_adapter.h" #include "source_hsolver/hsolver_lcao.h" // use HSolverLCAO for init_chg_hr #include "source_pw/module_pwdft/dftu_base.h" // use Plus_U_Base for the DFT+U init template void LCAO_domain::set_psi_occ_dm_chg( - const K_Vectors &kv, // k-points - psi::Psi* &psi, // coefficients of NAO basis - const Parallel_Orbitals &pv, // parallel scheme of NAO basis - elecstate::ElecState* pelec, // eigen values and weights - LCAO_domain::Setup_DM &dmat, // density matrix - Charge &chr, // charge density - const Input_para &inp) // input parameters + const K_Vectors &kv, // k-points + psi::Psi* &psi, // coefficients of NAO basis + const Parallel_Orbitals &pv, // parallel scheme of NAO basis + elecstate::ElecState* pelec, // eigen values and weights + module_dm::Setup_DM &dmat, // density matrix + Charge &chr, // charge density + const Input_para &inp) // input parameters { //! 1) init electronic wave function psi @@ -42,7 +43,7 @@ void LCAO_domain::set_psi_occ_dm_chg( } //! 4) init DMK, but DMR is constructed in before_scf() - dmat.allocate_dm(&kv, &pv, inp.nspin); + LCAO_domain::allocate_dm(dmat, &kv, &pv, inp.nspin); ModuleBase::GlobalFunc::DONE(GlobalV::ofs_running, "CHARGE"); @@ -53,14 +54,14 @@ void LCAO_domain::set_psi_occ_dm_chg( template void LCAO_domain::set_pot( UnitCell &ucell, // not const because of dftu - K_Vectors &kv, // not const due to exx - Structure_Factor& sf, // will be modified in potential - const ModulePW::PW_Basis &pw_rho, - const ModulePW::PW_Basis &pw_rhod, - elecstate::ElecState* pelec, - const LCAO_Orbitals& orb, - Parallel_Orbitals &pv, // not const due to deepks - pseudopot_cell_vl &locpp, + K_Vectors &kv, // not const due to exx + Structure_Factor& sf, // will be modified in potential + const ModulePW::PW_Basis &pw_rho, + const ModulePW::PW_Basis &pw_rhod, + elecstate::ElecState* pelec, + const LCAO_Orbitals& orb, + Parallel_Orbitals &pv, // not const due to deepks + pseudopot_cell_vl &locpp, Plus_U_Base &dftu, surchem& solvent, Exx_NAO &exx_nao, @@ -123,7 +124,7 @@ template void LCAO_domain::init_dm_from_file( const std::string& readin_dir, const int nspin, - LCAO_domain::Setup_DM& dmat, + module_dm::Setup_DM& dmat, const UnitCell& ucell, const Parallel_Orbitals* pv) { @@ -131,8 +132,15 @@ void LCAO_domain::init_dm_from_file( const int nspin_dm = (nspin == 2) ? 2 : 1; for (int is = 0; is < nspin_dm; ++is) { - const std::string dmfile = readin_dir + "/dmrs" + std::to_string(is + 1) + "_nao.csr"; - hamilt::HContainer* dm_container = dmat.dm->get_DMR_vector()[is]; + // readin_dir is normalized by to_dir() and always ends with '/' + const std::string dmfile = readin_dir + "dmrs" + std::to_string(is + 1) + "_nao.csr"; + // EXX-specific: add rank guard because OperatorEXX is constructed on all MPI ranks, + // unlike most other places where ofs_running is only written on rank 0 + if (GlobalV::MY_RANK == 0) + { + GlobalV::ofs_running << " Read density matrix from " << dmfile << std::endl; + } + hamilt::HContainer* dm_container = dmat.dm->get_dmr_vec()[is]; hamilt::Read_HContainer reader_dm( dm_container, dmfile, @@ -149,7 +157,7 @@ template void LCAO_domain::init_chg_dm( const std::string& readin_dir, const int nspin, - LCAO_domain::Setup_DM& dmat, + module_dm::Setup_DM& dmat, const UnitCell& ucell, const Parallel_Orbitals* pv, Charge* chr) @@ -160,7 +168,8 @@ void LCAO_domain::init_chg_dm( LCAO_domain::init_dm_from_file(readin_dir, nspin, dmat, ucell, pv); // Step 2: Convert density matrix to charge density - LCAO_domain::dm2rho(dmat.dm->get_DMR_vector(), nspin, chr, true); + // skip_normalize=true here (loaded DM is already normalized), so omega is unused. + LCAO_domain::dm2rho(dmat.dm->get_dmr_vec(), nspin, chr, PARAM.inp.nelec, ucell.omega, true); return; } @@ -205,7 +214,7 @@ void LCAO_domain::init_chg_hr( const Parallel_Orbitals* pv, psi::Psi& psi, elecstate::ElecState* pelec, - elecstate::DensityMatrix& dm, + module_dm::DensityMatrix& dm, Charge& chr, const std::string& ks_solver) { @@ -252,39 +261,40 @@ void LCAO_domain::init_chg_hr( PARAM.inp.device == "gpu", GlobalV::NPROC, GlobalV::MY_RANK); - hsolver_lcao_obj.solve(p_hamilt, psi, pelec, dm, chr, nspin, 0); + hamilt::HamiltHSMatrix hs(p_hamilt); + hsolver_lcao_obj.solve(hs, psi, pelec, dm, chr, nspin, ucell.omega, 0); } template void LCAO_domain::set_psi_occ_dm_chg( - const K_Vectors &kv, // k-points - psi::Psi* &psi, // coefficients of NAO basis - const Parallel_Orbitals &pv, // parallel scheme of NAO basis - elecstate::ElecState* pelec, // eigen values and weights - LCAO_domain::Setup_DM &dmat, // density matrix - Charge &chr, // charge density - const Input_para &inp); + const K_Vectors &kv, // k-points + psi::Psi* &psi, // coefficients of NAO basis + const Parallel_Orbitals &pv, // parallel scheme of NAO basis + elecstate::ElecState* pelec, // eigen values and weights + module_dm::Setup_DM &dmat, // density matrix + Charge &chr, // charge density + const Input_para &inp); template void LCAO_domain::set_psi_occ_dm_chg>( - const K_Vectors &kv, // k-points - psi::Psi>* &psi, // coefficients of NAO basis - const Parallel_Orbitals &pv, // parallel scheme of NAO basis - elecstate::ElecState* pelec, // eigen values and weights - LCAO_domain::Setup_DM> &dmat, // density matrix - Charge &chr, // charge density - const Input_para &inp); + const K_Vectors &kv, // k-points + psi::Psi>* &psi, // coefficients of NAO basis + const Parallel_Orbitals &pv, // parallel scheme of NAO basis + elecstate::ElecState* pelec, // eigen values and weights + module_dm::Setup_DM> &dmat, // density matrix + Charge &chr, // charge density + const Input_para &inp); template void LCAO_domain::set_pot( UnitCell &ucell, - K_Vectors &kv, - Structure_Factor& sf, - const ModulePW::PW_Basis &pw_rho, - const ModulePW::PW_Basis &pw_rhod, - elecstate::ElecState* pelec, - const LCAO_Orbitals& orb, - Parallel_Orbitals &pv, - pseudopot_cell_vl &locpp, + K_Vectors &kv, + Structure_Factor& sf, + const ModulePW::PW_Basis &pw_rho, + const ModulePW::PW_Basis &pw_rhod, + elecstate::ElecState* pelec, + const LCAO_Orbitals& orb, + Parallel_Orbitals &pv, + pseudopot_cell_vl &locpp, Plus_U_Base &dftu, surchem& solvent, Exx_NAO &exx_nao, @@ -294,14 +304,14 @@ template void LCAO_domain::set_pot( template void LCAO_domain::set_pot>( UnitCell &ucell, - K_Vectors &kv, - Structure_Factor& sf, - const ModulePW::PW_Basis &pw_rho, - const ModulePW::PW_Basis &pw_rhod, - elecstate::ElecState* pelec, - const LCAO_Orbitals& orb, - Parallel_Orbitals &pv, - pseudopot_cell_vl &locpp, + K_Vectors &kv, + Structure_Factor& sf, + const ModulePW::PW_Basis &pw_rho, + const ModulePW::PW_Basis &pw_rhod, + elecstate::ElecState* pelec, + const LCAO_Orbitals& orb, + Parallel_Orbitals &pv, + pseudopot_cell_vl &locpp, Plus_U_Base &dftu, surchem& solvent, Exx_NAO> &exx_nao, @@ -312,27 +322,27 @@ template void LCAO_domain::set_pot>( template void LCAO_domain::init_dm_from_file( const std::string& readin_dir, const int nspin, - LCAO_domain::Setup_DM& dmat, + module_dm::Setup_DM& dmat, const UnitCell& ucell, const Parallel_Orbitals* pv); template void LCAO_domain::init_dm_from_file>( const std::string& readin_dir, const int nspin, - LCAO_domain::Setup_DM>& dmat, + module_dm::Setup_DM>& dmat, const UnitCell& ucell, const Parallel_Orbitals* pv); template void LCAO_domain::init_chg_dm( const std::string& readin_dir, const int nspin, - LCAO_domain::Setup_DM& dmat, + module_dm::Setup_DM& dmat, const UnitCell& ucell, const Parallel_Orbitals* pv, Charge* chr); template void LCAO_domain::init_chg_dm>( const std::string& readin_dir, const int nspin, - LCAO_domain::Setup_DM>& dmat, + module_dm::Setup_DM>& dmat, const UnitCell& ucell, const Parallel_Orbitals* pv, Charge* chr); @@ -356,7 +366,7 @@ template void LCAO_domain::init_chg_hr( const Parallel_Orbitals* pv, psi::Psi& psi, elecstate::ElecState* pelec, - elecstate::DensityMatrix& dm, + module_dm::DensityMatrix& dm, Charge& chr, const std::string& ks_solver); template void LCAO_domain::init_chg_hr, double>( @@ -367,7 +377,7 @@ template void LCAO_domain::init_chg_hr, double>( const Parallel_Orbitals* pv, psi::Psi>& psi, elecstate::ElecState* pelec, - elecstate::DensityMatrix, double>& dm, + module_dm::DensityMatrix, double>& dm, Charge& chr, const std::string& ks_solver); template void LCAO_domain::init_chg_hr, std::complex>( @@ -378,6 +388,6 @@ template void LCAO_domain::init_chg_hr, std::complex>& psi, elecstate::ElecState* pelec, - elecstate::DensityMatrix, double>& dm, + module_dm::DensityMatrix, double>& dm, Charge& chr, const std::string& ks_solver); diff --git a/source/source_lcao/lcao_set.h b/source/source_lcao/lcao_set.h index 4d5b6019c5e..114246be22a 100644 --- a/source/source_lcao/lcao_set.h +++ b/source/source_lcao/lcao_set.h @@ -7,7 +7,7 @@ #include "source_estate/elecstate.h" #include "source_estate/module_dm/density_matrix.h" #include "source_hamilt/hamilt.h" -#include "source_lcao/setup_dm.h" +#include "source_lcao/allocate_dm.h" #include "source_pw/module_pwdft/stru_fac.h" #include "source_basis/module_pw/pw_basis.h" #include "source_hamilt/module_surchem/surchem.h" @@ -27,13 +27,13 @@ namespace LCAO_domain */ template void set_psi_occ_dm_chg( - const K_Vectors &kv, // k-points - psi::Psi* &psi, // coefficients of NAO basis - const Parallel_Orbitals &pv, // parallel scheme of NAO basis - elecstate::ElecState* pelec, // eigen values and weights - LCAO_domain::Setup_DM &dmat, // density matrix - Charge &chr, // charge density - const Input_para& inp); // input parameters + const K_Vectors &kv, // k-points + psi::Psi* &psi, // coefficients of NAO basis + const Parallel_Orbitals &pv, // parallel scheme of NAO basis + elecstate::ElecState* pelec, // eigen values and weights + module_dm::Setup_DM &dmat, // density matrix + Charge &chr, // charge density + const Input_para& inp); // input parameters /** * @brief set up potentials, including local pseudopotentials, @@ -42,14 +42,14 @@ void set_psi_occ_dm_chg( template void set_pot( UnitCell &ucell, - K_Vectors &kv, - Structure_Factor& sf, - const ModulePW::PW_Basis &pw_rho, - const ModulePW::PW_Basis &pw_rhod, - elecstate::ElecState* pelec, - const LCAO_Orbitals& orb, - Parallel_Orbitals &pv, - pseudopot_cell_vl &locpp, + K_Vectors &kv, + Structure_Factor& sf, + const ModulePW::PW_Basis &pw_rho, + const ModulePW::PW_Basis &pw_rhod, + elecstate::ElecState* pelec, + const LCAO_Orbitals& orb, + Parallel_Orbitals &pv, + pseudopot_cell_vl &locpp, Plus_U_Base &dftu, surchem& solvent, Exx_NAO &exx_nao, @@ -64,11 +64,11 @@ void set_pot( */ template void init_dm_from_file( - const std::string& readin_dir, - const int nspin, - LCAO_domain::Setup_DM& dmat, - const UnitCell& ucell, - const Parallel_Orbitals* pv); + const std::string& readin_dir, + const int nspin, + module_dm::Setup_DM& dmat, + const UnitCell& ucell, + const Parallel_Orbitals* pv); /** * @brief initialize charge density from density matrix file (init_chg=dm) @@ -82,22 +82,22 @@ void init_dm_from_file( */ template void init_chg_dm( - const std::string& readin_dir, - const int nspin, - LCAO_domain::Setup_DM& dmat, - const UnitCell& ucell, - const Parallel_Orbitals* pv, - Charge* chr); + const std::string& readin_dir, + const int nspin, + module_dm::Setup_DM& dmat, + const UnitCell& ucell, + const Parallel_Orbitals* pv, + Charge* chr); /** * @brief read in HR from file, and save it into hmat */ template void init_hr_from_file( - const std::string hrfile, - hamilt::HContainer* hmat, - const UnitCell& ucell, - const Parallel_Orbitals* pv); + const std::string hrfile, + hamilt::HContainer* hmat, + const UnitCell& ucell, + const Parallel_Orbitals* pv); /** * @brief initialize charge density from Hamiltonian matrix file (init_chg=hr) @@ -119,16 +119,16 @@ void init_hr_from_file( */ template void init_chg_hr( - const std::string& readin_dir, - const int nspin, - hamilt::Hamilt* p_hamilt, - const UnitCell& ucell, - const Parallel_Orbitals* pv, - psi::Psi& psi, - elecstate::ElecState* pelec, - elecstate::DensityMatrix& dm, - Charge& chr, - const std::string& ks_solver); + const std::string& readin_dir, + const int nspin, + hamilt::Hamilt* p_hamilt, + const UnitCell& ucell, + const Parallel_Orbitals* pv, + psi::Psi& psi, + elecstate::ElecState* pelec, + module_dm::DensityMatrix& dm, + Charge& chr, + const std::string& ks_solver); } // end namespace #endif diff --git a/source/source_lcao/lcao_set_fs.cpp b/source/source_lcao/lcao_set_fs.cpp index a42ca24adf1..c6bc65c19e2 100644 --- a/source/source_lcao/lcao_set_fs.cpp +++ b/source/source_lcao/lcao_set_fs.cpp @@ -59,76 +59,4 @@ void set_force return; } -void set_stress -( - const Parallel_Orbitals &pv, - const int &iw1_all, - const int &iw2_all, - const double& vx, - const double& vy, - const double& vz, - const char &dtype, - const ModuleBase::Vector3 &dtau, - double* dsloc_11, - double* dsloc_12, - double* dsloc_13, - double* dsloc_22, - double* dsloc_23, - double* dsloc_33, - double* dhloc_fixed_11, - double* dhloc_fixed_12, - double* dhloc_fixed_13, - double* dhloc_fixed_22, - double* dhloc_fixed_23, - double* dhloc_fixed_33) -{ - // use iw1_all and iw2_all to set Hloc - // becareful! The ir and ic may < 0!!!!!!!!!!!!!!!! - const int ir = pv.global2local_row(iw1_all); - const int ic = pv.global2local_col(iw2_all); - const long index = ir * pv.ncol + ic; - - if( index >= pv.nloc) - { - std::cout << " iw1_all = " << iw1_all << std::endl; - std::cout << " iw2_all = " << iw2_all << std::endl; - std::cout << " ir = " << ir << std::endl; - std::cout << " ic = " << ic << std::endl; - std::cout << " index = " << index << std::endl; - std::cout << " pv.nloc = " << pv.nloc << std::endl; - ModuleBase::WARNING_QUIT("LCAO_domain","set_stress"); - } - - if (dtype == 'S') - { - dsloc_11[index] += vx * dtau.x; - dsloc_12[index] += vx * dtau.y; - dsloc_13[index] += vx * dtau.z; - dsloc_22[index] += vy * dtau.y; - dsloc_23[index] += vy * dtau.z; - dsloc_33[index] += vz * dtau.z; - } - else if (dtype == 'T') - { - // notice, the sign is '-', minus. - dhloc_fixed_11[index] -= vx * dtau.x; - dhloc_fixed_12[index] -= vx * dtau.y; - dhloc_fixed_13[index] -= vx * dtau.z; - dhloc_fixed_22[index] -= vy * dtau.y; - dhloc_fixed_23[index] -= vy * dtau.z; - dhloc_fixed_33[index] -= vz * dtau.z; - } - else if (dtype == 'N') - { - dhloc_fixed_11[index] += vx * dtau.x; - dhloc_fixed_12[index] += vx * dtau.y; - dhloc_fixed_13[index] += vx * dtau.z; - dhloc_fixed_22[index] += vy * dtau.y; - dhloc_fixed_23[index] += vy * dtau.z; - dhloc_fixed_33[index] += vz * dtau.z; - } - - return; -} - } diff --git a/source/source_lcao/lcao_set_st.cpp b/source/source_lcao/lcao_set_st.cpp index 2740c73d56e..af59e6deb17 100644 --- a/source/source_lcao/lcao_set_st.cpp +++ b/source/source_lcao/lcao_set_st.cpp @@ -5,250 +5,268 @@ namespace LCAO_domain { -void single_derivative(ForceStressArrays& fsr, - const LCAO_Orbitals& orb, - const TwoCenterBundle& two_center_bundle, - const Parallel_Orbitals& pv, - const UnitCell& ucell, - const int nspin, - const bool cal_stress, - const int iw1_all, - const int iw2_all, - const int m1, - const int m2, - const char& dtype, - const int T1, - const int L1, - const int N1, - const int T2, - const int L2, - const int N2, - const ModuleBase::Vector3& dtau, - const ModuleBase::Vector3& tau1, - const ModuleBase::Vector3& tau2, - const int npol, - const int jj, - const int jj0, - const int kk, - const int kk0, - int& nnr, - int& total_nnr, - double* olm // output value -) +// write the dS/dH-fixed contributions of one S-matrix element at slot nnr. +// defined statically here so the compiler can inline it at the hot inner-loop +// call site of single_deriv. +static void set_deriv_s(ForceStressArrays& fsr, + const int nspin, + const int is, + const int nnr, + const double* olm, + const ModuleBase::Vector3& dtau, + const bool cal_stress) { + // write DSloc_R* only when allocated (skipped in cal_dS where only DHloc_fixedR_* is used) + const bool write_dsloc_r = !fsr.DSloc_Rx.empty(); + // DHloc_fixedR_* and DH_r must be pre-allocated by the caller for the derivative path + const bool write_dhloc = !fsr.DHloc_fixedR_x.empty(); + const bool write_dhr = cal_stress && !fsr.DH_r.empty(); + // condition 9, nspin + if (nspin == 1 || nspin == 2) + { + if (write_dsloc_r) + { + fsr.DSloc_Rx[nnr] = olm[0]; + fsr.DSloc_Ry[nnr] = olm[1]; + fsr.DSloc_Rz[nnr] = olm[2]; + } + if (write_dhloc) + { + fsr.DHloc_fixedR_x[nnr] = olm[0]; + fsr.DHloc_fixedR_y[nnr] = olm[1]; + fsr.DHloc_fixedR_z[nnr] = olm[2]; + } + } + else if (nspin == 4) + { + const double v0 = (is == 0) ? olm[0] : 0.0; // is==3 is not needed in force calculation + const double v1 = (is == 0) ? olm[1] : 0.0; + const double v2 = (is == 0) ? olm[2] : 0.0; + if (write_dsloc_r) + { + fsr.DSloc_Rx[nnr] = v0; + fsr.DSloc_Ry[nnr] = v1; + fsr.DSloc_Rz[nnr] = v2; + } + if (write_dhloc) + { + fsr.DHloc_fixedR_x[nnr] = v0; + fsr.DHloc_fixedR_y[nnr] = v1; + fsr.DHloc_fixedR_z[nnr] = v2; + } + } + else + { + ModuleBase::WARNING_QUIT("LCAO_domain::set_deriv_s", "nspin must be 1, 2 or 4"); + } // end condition 9, nspin - const bool gamma_only_local = PARAM.globalv.gamma_only_local; + if (write_dhr) + { + fsr.DH_r[nnr * 3] = dtau.x; + fsr.DH_r[nnr * 3 + 1] = dtau.y; + fsr.DH_r[nnr * 3 + 2] = dtau.z; + } +} + +// write the dH-fixed and stress contributions of one T-matrix element at slot nnr. +// kept in this translation unit so the compiler can inline it at the hot +// inner-loop call site of single_deriv. +static void set_deriv_t(ForceStressArrays& fsr, + const int nspin, + const int is, + const int nnr, + const double* olm, + const ModuleBase::Vector3& dtau, + const bool cal_stress) +{ + // DHloc_fixedR_* and stress arrays must be pre-allocated by the caller for the derivative path + const bool write_dhloc = !fsr.DHloc_fixedR_x.empty(); + const bool write_stress = cal_stress && !fsr.stvnl11.empty(); + // condition 9, nspin + if (nspin == 1 || nspin == 2) + { + if (write_dhloc) + { + fsr.DHloc_fixedR_x[nnr] = olm[0]; + fsr.DHloc_fixedR_y[nnr] = olm[1]; + fsr.DHloc_fixedR_z[nnr] = olm[2]; + } + if (write_stress) + { + fsr.stvnl11[nnr] = olm[0] * dtau.x; + fsr.stvnl12[nnr] = olm[0] * dtau.y; + fsr.stvnl13[nnr] = olm[0] * dtau.z; + fsr.stvnl22[nnr] = olm[1] * dtau.y; + fsr.stvnl23[nnr] = olm[1] * dtau.z; + fsr.stvnl33[nnr] = olm[2] * dtau.z; + } + } + else if (nspin == 4) + { + // condition 10, details of nspin 4 + if (is == 0) // is==3 is not needed in force calculation + { + if (write_dhloc) + { + fsr.DHloc_fixedR_x[nnr] = olm[0]; + fsr.DHloc_fixedR_y[nnr] = olm[1]; + fsr.DHloc_fixedR_z[nnr] = olm[2]; + } + if (write_stress) + { + fsr.stvnl11[nnr] = olm[0] * dtau.x; + fsr.stvnl12[nnr] = olm[0] * dtau.y; + fsr.stvnl13[nnr] = olm[0] * dtau.z; + fsr.stvnl22[nnr] = olm[1] * dtau.y; + fsr.stvnl23[nnr] = olm[1] * dtau.z; + fsr.stvnl33[nnr] = olm[2] * dtau.z; + } + } + else if (is == 1 || is == 2 || is == 3) + { + if (write_dhloc) + { + fsr.DHloc_fixedR_x[nnr] = 0.0; + fsr.DHloc_fixedR_y[nnr] = 0.0; + fsr.DHloc_fixedR_z[nnr] = 0.0; + } + if (write_stress) + { + fsr.stvnl11[nnr] = 0.0; + fsr.stvnl12[nnr] = 0.0; + fsr.stvnl13[nnr] = 0.0; + fsr.stvnl22[nnr] = 0.0; + fsr.stvnl23[nnr] = 0.0; + fsr.stvnl33[nnr] = 0.0; + } + } + else + { + ModuleBase::WARNING_QUIT("LCAO_domain::set_deriv_t", "is must be 0, 1, 2, 3"); + } // end condition 10, details of spin 4 + } + else + { + ModuleBase::WARNING_QUIT("LCAO_domain::set_deriv_t", "nspin must be 1, 2 or 4"); + } // end condition 9, nspin +} - // convert m (0,1,...2l) to M (-l, -l+1, ..., l-1, l) - const int M1 = (m1 % 2 == 0) ? -m1 / 2 : (m1 + 1) / 2; - const int M2 = (m2 % 2 == 0) ? -m2 / 2 : (m2 + 1) / 2; +void single_deriv(const ST_env& env, + const ST_elem& e, + ForceStressArrays& fsr, + int& nnr, + int& total_nnr, + double* olm // output value +) +{ + + const bool gamma_only_local = env.gamma_only_local; + const int nspin = env.nspin; + const int npol = env.npol; + const bool cal_stress = env.cal_stress; + const int iw1_all = e.iw1_all; + const int iw2_all = e.iw2_all; + const char dtype = e.dtype; + const int m1 = e.m1; + const int m2 = e.m2; + const int t1 = e.t1; + const int l1 = e.l1; + const int n1 = e.n1; + const int t2 = e.t2; + const int l2 = e.l2; + const int n2 = e.n2; + const ModuleBase::Vector3& dtau = e.dtau; + const int jj = e.jj; + const int jj0 = e.jj0; + const int kk = e.kk; + const int kk0 = e.kk0; + + // convert m (0,1,...2l) to mm (-l, -l+1, ..., l-1, l) + const int mm1 = (m1 % 2 == 0) ? -m1 / 2 : (m1 + 1) / 2; + const int mm2 = (m2 % 2 == 0) ? -m2 / 2 : (m2 + 1) / 2; switch (dtype) { case 'S': - two_center_bundle.overlap_orb->calculate(T1, L1, N1, M1, T2, L2, N2, M2, dtau * ucell.lat0, nullptr, olm); + env.two_center_bundle.overlap_orb->calculate(t1, l1, n1, mm1, t2, l2, n2, mm2, dtau * env.ucell.lat0, nullptr, olm); break; case 'T': - two_center_bundle.kinetic_orb->calculate(T1, L1, N1, M1, T2, L2, N2, M2, dtau * ucell.lat0, nullptr, olm); + env.two_center_bundle.kinetic_orb->calculate(t1, l1, n1, mm1, t2, l2, n2, mm2, dtau * env.ucell.lat0, nullptr, olm); break; default: // not supposed to happen - ModuleBase::WARNING_QUIT("LCAO_domain::build_ST_new", "dtype must be S or T"); + ModuleBase::WARNING_QUIT("LCAO_domain::single_deriv", "dtype must be S or T"); } // condition 7: gamma only or multiple k if (gamma_only_local) { - LCAO_domain::set_force(pv, + LCAO_domain::set_force(env.pv, iw1_all, iw2_all, olm[0], olm[1], olm[2], dtype, - fsr.DSloc_x, - fsr.DSloc_y, - fsr.DSloc_z, - fsr.DHloc_fixed_x, - fsr.DHloc_fixed_y, - fsr.DHloc_fixed_z); - - if (cal_stress) - { - LCAO_domain::set_stress(pv, - iw1_all, - iw2_all, - olm[0], - olm[1], - olm[2], - dtype, - dtau, - fsr.DSloc_11, - fsr.DSloc_12, - fsr.DSloc_13, - fsr.DSloc_22, - fsr.DSloc_23, - fsr.DSloc_33, - fsr.DHloc_fixed_11, - fsr.DHloc_fixed_12, - fsr.DHloc_fixed_13, - fsr.DHloc_fixed_22, - fsr.DHloc_fixed_23, - fsr.DHloc_fixed_33); - } // end stress + fsr.DSloc_x.data(), + fsr.DSloc_y.data(), + fsr.DSloc_z.data(), + fsr.DHloc_fixed_x.data(), + fsr.DHloc_fixed_y.data(), + fsr.DHloc_fixed_z.data()); } // end gamma_only else // condition 7, multiple k-points algorithm { + const int is = (jj - jj0 * npol) + (kk - kk0 * npol) * 2; // condition 8, S or T if (dtype == 'S') { - // condition 9, nspin - if (nspin == 1 || nspin == 2) - { - fsr.DSloc_Rx[nnr] = olm[0]; - fsr.DSloc_Ry[nnr] = olm[1]; - fsr.DSloc_Rz[nnr] = olm[2]; - } - else if (nspin == 4) - { - int is = (jj - jj0 * npol) + (kk - kk0 * npol) * 2; - if (is == 0) // is==3 is not needed in force calculation - { - fsr.DSloc_Rx[nnr] = olm[0]; - fsr.DSloc_Ry[nnr] = olm[1]; - fsr.DSloc_Rz[nnr] = olm[2]; - } - else - { - fsr.DSloc_Rx[nnr] = 0.0; - fsr.DSloc_Ry[nnr] = 0.0; - fsr.DSloc_Rz[nnr] = 0.0; - } - } - else - { - ModuleBase::WARNING_QUIT("LCAO_domain::build_ST_new", "nspin must be 1, 2 or 4"); - } // end condition 9, nspin - - if (cal_stress) - { - fsr.DH_r[nnr * 3] = dtau.x; - fsr.DH_r[nnr * 3 + 1] = dtau.y; - fsr.DH_r[nnr * 3 + 2] = dtau.z; - } + set_deriv_s(fsr, nspin, is, nnr, olm, dtau, cal_stress); } - else if (dtype == 'T') // condition 8, S or T + else if (dtype == 'T') { - // condtion 9, nspin - if (nspin == 1 || nspin == 2) - { - fsr.DHloc_fixedR_x[nnr] = olm[0]; - fsr.DHloc_fixedR_y[nnr] = olm[1]; - fsr.DHloc_fixedR_z[nnr] = olm[2]; - if (cal_stress) - { - fsr.stvnl11[nnr] = olm[0] * dtau.x; - fsr.stvnl12[nnr] = olm[0] * dtau.y; - fsr.stvnl13[nnr] = olm[0] * dtau.z; - fsr.stvnl22[nnr] = olm[1] * dtau.y; - fsr.stvnl23[nnr] = olm[1] * dtau.z; - fsr.stvnl33[nnr] = olm[2] * dtau.z; - } - } - else if (nspin == 4) // condition 9 - { - const int is = (jj - jj0 * npol) + (kk - kk0 * npol) * 2; - // condition 10, details of nspin 4 - if (is == 0) // is==3 is not needed in force calculation - { - fsr.DHloc_fixedR_x[nnr] = olm[0]; - fsr.DHloc_fixedR_y[nnr] = olm[1]; - fsr.DHloc_fixedR_z[nnr] = olm[2]; - if (cal_stress) - { - fsr.stvnl11[nnr] = olm[0] * dtau.x; - fsr.stvnl12[nnr] = olm[0] * dtau.y; - fsr.stvnl13[nnr] = olm[0] * dtau.z; - fsr.stvnl22[nnr] = olm[1] * dtau.y; - fsr.stvnl23[nnr] = olm[1] * dtau.z; - fsr.stvnl33[nnr] = olm[2] * dtau.z; - } - } - else if (is == 1 || is == 2 || is == 3) - { - fsr.DHloc_fixedR_x[nnr] = 0.0; - fsr.DHloc_fixedR_y[nnr] = 0.0; - fsr.DHloc_fixedR_z[nnr] = 0.0; - if (cal_stress) - { - fsr.stvnl11[nnr] = 0.0; - fsr.stvnl12[nnr] = 0.0; - fsr.stvnl13[nnr] = 0.0; - fsr.stvnl22[nnr] = 0.0; - fsr.stvnl23[nnr] = 0.0; - fsr.stvnl33[nnr] = 0.0; - } - } - else - { - ModuleBase::WARNING_QUIT("LCAO_domain::build_ST_new", "is must be 0, 1, 2, 3"); - } // end condition 10, details of spin 4 - } - else - { - ModuleBase::WARNING_QUIT("LCAO_domain::build_ST_new", "nspin must be 1, 2 or 4"); - } // end condition 9, nspin + set_deriv_t(fsr, nspin, is, nnr, olm, dtau, cal_stress); } // end condition 8, S or T ++total_nnr; ++nnr; } // end condition 7, gamma or multiple k } -void single_overlap(const LCAO_Orbitals& orb, - const TwoCenterBundle& two_center_bundle, - const Parallel_Orbitals& pv, - const UnitCell& ucell, - const int nspin, - const bool cal_stress, - const int iw1_all, - const int iw2_all, - const int m1, - const int m2, - const char& dtype, - const int T1, - const int L1, - const int N1, - const int T2, - const int L2, - const int N2, - const ModuleBase::Vector3& dtau, - const ModuleBase::Vector3& tau1, - const ModuleBase::Vector3& tau2, - const int npol, - const int jj, - const int jj0, - const int kk, - const int kk0, +void single_overlap(const ST_env& env, + const ST_elem& e, int& nnr, // output value int& total_nnr, // output value double* olm, // output value double* HSloc // output value ) { - const bool gamma_only_local = PARAM.globalv.gamma_only_local; - - // convert m (0,1,...2l) to M (-l, -l+1, ..., l-1, l) - const int M1 = (m1 % 2 == 0) ? -m1 / 2 : (m1 + 1) / 2; - const int M2 = (m2 % 2 == 0) ? -m2 / 2 : (m2 + 1) / 2; + const bool gamma_only_local = env.gamma_only_local; + const int nspin = env.nspin; + const int iw1_all = e.iw1_all; + const int iw2_all = e.iw2_all; + const char dtype = e.dtype; + const int m1 = e.m1; + const int m2 = e.m2; + const int t1 = e.t1; + const int l1 = e.l1; + const int n1 = e.n1; + const int t2 = e.t2; + const int l2 = e.l2; + const int n2 = e.n2; + const ModuleBase::Vector3& dtau = e.dtau; + + // convert m (0,1,...2l) to mm (-l, -l+1, ..., l-1, l) + const int mm1 = (m1 % 2 == 0) ? -m1 / 2 : (m1 + 1) / 2; + const int mm2 = (m2 % 2 == 0) ? -m2 / 2 : (m2 + 1) / 2; switch (dtype) { case 'S': - two_center_bundle.overlap_orb->calculate(T1, L1, N1, M1, T2, L2, N2, M2, dtau * ucell.lat0, olm); + env.two_center_bundle.overlap_orb->calculate(t1, l1, n1, mm1, t2, l2, n2, mm2, dtau * env.ucell.lat0, olm); break; case 'T': - two_center_bundle.kinetic_orb->calculate(T1, L1, N1, M1, T2, L2, N2, M2, dtau * ucell.lat0, olm); + env.two_center_bundle.kinetic_orb->calculate(t1, l1, n1, mm1, t2, l2, n2, mm2, dtau * env.ucell.lat0, olm); break; default: // not supposed to happen - ModuleBase::WARNING_QUIT("LCAO_domain::build_ST_new", "dtype must be S or T"); + ModuleBase::WARNING_QUIT("LCAO_domain::single_overlap", "dtype must be S or T"); } // When NSPIN == 4 , only diagonal term is calculated for T or S Operators @@ -271,7 +289,7 @@ void single_overlap(const LCAO_Orbitals& orb, // according to global2local_row and global2local_col // the last paramete: 1 for Sloc, 2 for Hloc // and 3 for Hloc_fixed. - LCAO_domain::set_mat2d(iw1_all, iw2_all, olm[0], pv, HSloc); + LCAO_domain::set_mat2d(iw1_all, iw2_all, olm[0], env.pv, HSloc); } else // condition 7, multiple k-points algorithm { @@ -285,7 +303,7 @@ void single_overlap(const LCAO_Orbitals& orb, } else { - ModuleBase::WARNING_QUIT("LCAO_domain::build_ST_new", "nspin must be 1, 2 or 4"); + ModuleBase::WARNING_QUIT("LCAO_domain::single_overlap", "nspin must be 1, 2 or 4"); } } else if (dtype == 'T') // condition 8, S or T @@ -300,7 +318,7 @@ void single_overlap(const LCAO_Orbitals& orb, } else { - ModuleBase::WARNING_QUIT("LCAO_domain::build_ST_new", "nspin must be 1, 2 or 4"); + ModuleBase::WARNING_QUIT("LCAO_domain::single_overlap", "nspin must be 1, 2 or 4"); } } // end condition 8, S or T ++total_nnr; @@ -328,6 +346,24 @@ void build_ST_new(ForceStressArrays& fsr, const int npol = PARAM.globalv.npol; const bool gamma_only_local = PARAM.globalv.gamma_only_local; + // derivative path must provide the target buffers + if (calc_deri && !gamma_only_local) + { + if (fsr.DHloc_fixedR_x.empty() || fsr.DHloc_fixedR_y.empty() || fsr.DHloc_fixedR_z.empty()) + { + ModuleBase::WARNING_QUIT("LCAO_domain::build_ST_new", + "DHloc_fixedR_x/y/z must be allocated when calc_deri=true in multi-k mode"); + } + if (cal_stress && fsr.DH_r.empty()) + { + ModuleBase::WARNING_QUIT("LCAO_domain::build_ST_new", + "DH_r must be allocated when calc_deri=true and cal_stress=true in multi-k mode"); + } + } + + // read-only environment shared by every element of this build + const ST_env env{orb, two_center_bundle, pv, ucell, nspin, npol, cal_stress, gamma_only_local}; + int total_nnr = 0; #ifdef _OPENMP #pragma omp parallel reduction(+ : total_nnr) @@ -346,15 +382,15 @@ void build_ST_new(ForceStressArrays& fsr, #endif for (int iat1 = 0; iat1 < ucell.nat; iat1++) // loop 1, iat1 { - const int T1 = ucell.iat2it[iat1]; - const Atom* atom1 = &ucell.atoms[T1]; - const int I1 = ucell.iat2ia[iat1]; + const int t1 = ucell.iat2it[iat1]; + const Atom* atom1 = &ucell.atoms[t1]; + const int i1 = ucell.iat2ia[iat1]; - tau1 = atom1->tau[I1]; + tau1 = atom1->tau[i1]; // GridD->Find_atom(tau1); AdjacentAtomInfo adjs; - GridD->Find_atom(ucell, tau1, T1, I1, &adjs); + GridD->Find_atom(ucell, tau1, t1, i1, &adjs); // Record_adj.for_2d() may not called in some case int nnr = 0; if (!pv.nlocstart.empty()) @@ -366,42 +402,42 @@ void build_ST_new(ForceStressArrays& fsr, { for (int k = 0; k < 3; k++) { - tau1[k] = tau1[k] - atom1->vel[I1][k] * PARAM.mdp.md_dt / ModuleBase::AU_to_FS / ucell.lat0; + tau1[k] = tau1[k] - atom1->vel[i1][k] * PARAM.mdp.md_dt / ModuleBase::AU_to_FS / ucell.lat0; } } // loop 2, ad for (int ad = 0; ad < adjs.adj_num + 1; ++ad) { - const int T2 = adjs.ntype[ad]; - const int I2 = adjs.natom[ad]; - Atom* atom2 = &ucell.atoms[T2]; + const int t2 = adjs.ntype[ad]; + const int i2 = adjs.natom[ad]; + Atom* atom2 = &ucell.atoms[t2]; tau2 = adjs.adjacent_tau[ad]; dtau = tau2 - tau1; double distance = dtau.norm() * ucell.lat0; - double rcut = orb.Phi[T1].getRcut() + orb.Phi[T2].getRcut(); + double rcut = orb.Phi[t1].getRcut() + orb.Phi[t2].getRcut(); // condition 3, distance if (distance < rcut) { - int iw1_all = ucell.itiaiw2iwt(T1, I1, 0); // iw1_all = combined index (it, ia, iw) + int iw1_all = ucell.itiaiw2iwt(t1, i1, 0); // iw1_all = combined index (it, ia, iw) // loop 4, jj for (int jj = 0; jj < atom1->nw * npol; ++jj) { const int jj0 = jj / npol; - const int L1 = atom1->iw2l[jj0]; - const int N1 = atom1->iw2n[jj0]; + const int l1 = atom1->iw2l[jj0]; + const int n1 = atom1->iw2n[jj0]; const int m1 = atom1->iw2m[jj0]; - int iw2_all = ucell.itiaiw2iwt(T2, I2, 0); // zhengdy-soc + int iw2_all = ucell.itiaiw2iwt(t2, i2, 0); // zhengdy-soc // loop 5, kk for (int kk = 0; kk < atom2->nw * npol; ++kk) { const int kk0 = kk / npol; - const int L2 = atom2->iw2l[kk0]; - const int N2 = atom2->iw2n[kk0]; + const int l2 = atom2->iw2l[kk0]; + const int n2 = atom2->iw2n[kk0]; const int m2 = atom2->iw2m[kk0]; // mohan add 2010-06-29 @@ -421,70 +457,31 @@ void build_ST_new(ForceStressArrays& fsr, olm[1] = 0.0; olm[2] = 0.0; - // condition 6, not calculate the derivative - if (!calc_deri) - { - single_overlap(orb, - two_center_bundle, - pv, - ucell, - nspin, - cal_stress, + const ST_elem elem{dtype, iw1_all, iw2_all, m1, m2, - dtype, - T1, - L1, - N1, - T2, - L2, - N2, + t1, + l1, + n1, + t2, + l2, + n2, dtau, - tau1, - tau2, - npol, jj, jj0, kk, - kk0, - nnr, - total_nnr, - olm, - HSloc); + kk0}; + + // condition 6, not calculate the derivative + if (!calc_deri) + { + single_overlap(env, elem, nnr, total_nnr, olm, HSloc); } else // condition 6, calculate the derivative { - single_derivative(fsr, - orb, - two_center_bundle, - pv, - ucell, - nspin, - cal_stress, - iw1_all, - iw2_all, - m1, - m2, - dtype, - T1, - L1, - N1, - T2, - L2, - N2, - dtau, - tau1, - tau2, - npol, - jj, - jj0, - kk, - kk0, - nnr, - total_nnr, - olm); + single_deriv(env, elem, fsr, nnr, total_nnr, olm); } // end condition 6, calc_deri ++iw2_all; } // end loop 5, kk @@ -493,20 +490,20 @@ void build_ST_new(ForceStressArrays& fsr, } // condition 3, distance else if (distance >= rcut && (!gamma_only_local)) { - int start1 = ucell.itiaiw2iwt(T1, I1, 0); - int start2 = ucell.itiaiw2iwt(T2, I2, 0); + int start1 = ucell.itiaiw2iwt(t1, i1, 0); + int start2 = ucell.itiaiw2iwt(t2, i2, 0); bool is_adj = false; for (int ad0 = 0; ad0 < adjs.adj_num + 1; ++ad0) { - const int T0 = adjs.ntype[ad0]; + const int t0 = adjs.ntype[ad0]; tau0 = adjs.adjacent_tau[ad0]; dtau1 = tau0 - tau1; double distance1 = dtau1.norm() * ucell.lat0; - double rcut1 = orb.Phi[T1].getRcut() + ucell.infoNL->get_rcut_max(T0); + double rcut1 = orb.Phi[t1].getRcut() + ucell.infoNL->get_rcut_max(t0); dtau2 = tau0 - tau2; double distance2 = dtau2.norm() * ucell.lat0; - double rcut2 = orb.Phi[T2].getRcut() + ucell.infoNL->get_rcut_max(T0); + double rcut2 = orb.Phi[t2].getRcut() + ucell.infoNL->get_rcut_max(t0); if (distance1 < rcut1 && distance2 < rcut2) { is_adj = true; diff --git a/source/source_lcao/module_bse/bse_util.h b/source/source_lcao/module_bse/bse_util.h index ebf13e16040..494bbd64a31 100644 --- a/source/source_lcao/module_bse/bse_util.h +++ b/source/source_lcao/module_bse/bse_util.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_LCAO_MODULE_BSE_BSE_UTIL_H +#define ABACUS_SOURCE_LCAO_MODULE_BSE_BSE_UTIL_H + #include #include "source_psi/psi.h" @@ -94,3 +96,5 @@ container::Tensor cal_dm_trans_onebase_blas( const T& factor = (T)1.0); } + +#endif // ABACUS_SOURCE_LCAO_MODULE_BSE_BSE_UTIL_H diff --git a/source/source_lcao/module_bse/hamilt_bse.cpp b/source/source_lcao/module_bse/hamilt_bse.cpp index 886d64d201e..a162fe3ffe8 100644 --- a/source/source_lcao/module_bse/hamilt_bse.cpp +++ b/source/source_lcao/module_bse/hamilt_bse.cpp @@ -77,8 +77,8 @@ HamiltBSE::HamiltBSE(const int& nspin, if (!this->bse_ri_hartree && this->ri_hartree_benchmark == "none") { - this->DM_trans = LR_Util::make_unique>(&pmat, 1/*nspin*/, kv_in.kvec_d, nk); - this->DM_trans->set_DMK_zero(); + this->DM_trans = LR_Util::make_unique>(&pmat, 1/*nspin*/, kv_in.kvec_d, nk); + this->DM_trans->set_dmk_zero(); LR_Util::initialize_DMR(*this->DM_trans, this->pmat, this->ucell, this->gd, this->orb_cutoff); } if (this->bse_mem_save) { assert(this->bse_continue == 0 && this->bse_ri_hartree); } @@ -87,25 +87,25 @@ HamiltBSE::HamiltBSE(const int& nspin, if (this->bse_continue >= 1) { BSE_Util::print_mem_estimate("V matrix of A", this->pA.get_local_size(), sizeof(T)); this->VA_local.resize(this->pA.get_local_size(), 0.0); - this->read_AB_matrix(this->readin_dir + "A_V_matrix_"+std::to_string(this->my_rank)+".dat", this->VA_local.data(), this->ndim, this->ndim); + this->read_AB_matrix(this->readin_dir + "A_V_matrix_"+std::to_string(this->my_rank)+".dat", this->VA_local.data(), this->pA); ModuleBase::GlobalFunc::DONE(GlobalV::ofs_running, "read_V_for_A"); } if (this->bse_continue >= 2) { BSE_Util::print_mem_estimate("W matrix of A", this->pA.get_local_size(), sizeof(T)); this->WA_local.resize(this->pA.get_local_size(), 0.0); - this->read_AB_matrix(this->readin_dir + "A_W_matrix_"+std::to_string(this->my_rank)+".dat", this->WA_local.data(), this->ndim, this->ndim); + this->read_AB_matrix(this->readin_dir + "A_W_matrix_"+std::to_string(this->my_rank)+".dat", this->WA_local.data(), this->pA); ModuleBase::GlobalFunc::DONE(GlobalV::ofs_running, "read_W_for_A"); } if (this->bse_continue >= 3) { BSE_Util::print_mem_estimate("V matrix of B", this->pA.get_local_size(), sizeof(T)); this->VB_local.resize(this->pA.get_local_size(), 0.0); - this->read_AB_matrix(this->readin_dir + "B_V_matrix_"+std::to_string(this->my_rank)+".dat", this->VB_local.data(), this->ndim, this->ndim); + this->read_AB_matrix(this->readin_dir + "B_V_matrix_"+std::to_string(this->my_rank)+".dat", this->VB_local.data(), this->pA); ModuleBase::GlobalFunc::DONE(GlobalV::ofs_running, "read_V_for_B"); } if (this->bse_continue >= 4) { BSE_Util::print_mem_estimate("W matrix of B", this->pA.get_local_size(), sizeof(T)); this->WB_local.resize(this->pA.get_local_size(), 0.0); - this->read_AB_matrix(this->readin_dir + "B_W_matrix_"+std::to_string(this->my_rank)+".dat", this->WB_local.data(), this->ndim, this->ndim); + this->read_AB_matrix(this->readin_dir + "B_W_matrix_"+std::to_string(this->my_rank)+".dat", this->WB_local.data(), this->pA); ModuleBase::GlobalFunc::DONE(GlobalV::ofs_running, "read_W_for_B"); } @@ -163,7 +163,7 @@ void HamiltBSE::cal_V_for_A(){ this->cal_V_by_grid(true); } if (this->out_bse_ab){ - this->write_AB_matrix(this->out_dir+"A_V_matrix_"+std::to_string(this->my_rank)+".dat", 6, this->VA_local.data(), this->ndim, this->ndim); + this->write_AB_matrix(this->out_dir+"A_V_matrix_"+std::to_string(this->my_rank)+".dat", 6, this->VA_local.data(), this->pA); } ModuleBase::GlobalFunc::DONE(GlobalV::ofs_running, "cal_V_for_A"); ModuleBase::timer::end("HamiltBSE", "cal_V_for_A"); @@ -192,7 +192,7 @@ void HamiltBSE::cal_V_for_B(){ this->cal_V_by_grid(false); } if (this->out_bse_ab){ - this->write_AB_matrix(this->out_dir+"B_V_matrix_"+std::to_string(this->my_rank)+".dat", 6, this->VB_local.data(), this->ndim, this->ndim); + this->write_AB_matrix(this->out_dir+"B_V_matrix_"+std::to_string(this->my_rank)+".dat", 6, this->VB_local.data(), this->pA); } ModuleBase::GlobalFunc::DONE(GlobalV::ofs_running, "cal_V_for_B"); ModuleBase::timer::end("HamiltBSE", "cal_V_for_B"); @@ -211,7 +211,7 @@ void HamiltBSE::cal_W_for_A(){ this->WA_local.resize(this->pA.get_local_size(), 0.0); this->mo_lri.cal_W_for_A(this->WA_local, this->pA); if (this->out_bse_ab){ - this->write_AB_matrix(this->out_dir+"A_W_matrix_"+std::to_string(this->my_rank)+".dat", 6, this->WA_local.data(), this->ndim, this->ndim); + this->write_AB_matrix(this->out_dir+"A_W_matrix_"+std::to_string(this->my_rank)+".dat", 6, this->WA_local.data(), this->pA); } ModuleBase::GlobalFunc::DONE(GlobalV::ofs_running, "cal_W_for_A"); ModuleBase::timer::end("HamiltBSE", "cal_W_for_A"); @@ -231,7 +231,7 @@ void HamiltBSE::cal_W_for_B(){ this->mo_lri.cal_W_for_B(this->WB_local, this->pA); if (this->out_bse_ab){ - this->write_AB_matrix(this->out_dir+"B_W_matrix_"+std::to_string(this->my_rank)+".dat", 6, this->WB_local.data(), this->ndim, this->ndim); + this->write_AB_matrix(this->out_dir+"B_W_matrix_"+std::to_string(this->my_rank)+".dat", 6, this->WB_local.data(), this->pA); } ModuleBase::GlobalFunc::DONE(GlobalV::ofs_running, "cal_W_for_B"); ModuleBase::timer::end("HamiltBSE", "cal_W_for_B"); @@ -336,7 +336,7 @@ void HamiltBSE::init_bse_matrix(const bool is_full, const int & st_index){ } if (this->out_bse_ab) { - this->write_AB_matrix(this->out_dir+"A_matrix_"+std::to_string(this->my_rank)+".dat", 6, this->BSE_A_local.data(), this->ndim, this->ndim); + this->write_AB_matrix(this->out_dir+"A_matrix_"+std::to_string(this->my_rank)+".dat", 6, this->BSE_A_local.data(), this->pA); } if (is_full) @@ -351,7 +351,7 @@ void HamiltBSE::init_bse_matrix(const bool is_full, const int & st_index){ } if (this->out_bse_ab) { - this->write_AB_matrix(this->out_dir+"B_matrix_"+std::to_string(this->my_rank)+".dat", 6, this->BSE_B_local.data(), this->ndim, this->ndim); + this->write_AB_matrix(this->out_dir+"B_matrix_"+std::to_string(this->my_rank)+".dat", 6, this->BSE_B_local.data(), this->pA); } } @@ -505,9 +505,9 @@ void HamiltBSE::cal_V_by_grid(bool is_A) BSE_Util::cal_dm_trans_onebase_blas(psi_is, ik2, naos, imo1, imo2, (T)1.0 / (T)nk); #endif // LR_Util::print_tensor(dm_trans_2d, "dm_trans_2d", &pmat); - this->DM_trans->set_DMK_pointer(ik2, dm_trans_2d.data()); + this->DM_trans->set_dmk_ptr(ik2, dm_trans_2d.data()); // 3. D(k)→D(R) - this->DM_trans->cal_DMR(ik2); + this->DM_trans->cal_dmr(ik2); // LR_Util::print_DMR(*DM_trans, ucell.nat, "DMR"); // 4. D(R)→V(R) @@ -553,7 +553,7 @@ void HamiltBSE::grid_calculation(hamilt::HContainer& VR) const LR_Util::_allocate_2order_nested_ptr(rho_trans, 1, nrxx); // nspin=1 for transition density ModuleBase::GlobalFunc::ZEROS(rho_trans[0], nrxx); - ModuleGint::cal_gint_rho(this->DM_trans->get_DMR_vector(), 1, rho_trans, false); + ModuleGint::cal_gint_rho(this->DM_trans->get_dmr_vec(), 1, rho_trans, false); // 4.2. v_hxc = f_hxc * rho_trans ModuleBase::matrix vr_hxc(1, nrxx); //grid @@ -575,8 +575,8 @@ void HamiltBSE>::grid_calculation(hamilt::HContainer, double> DM_trans_real_imag(&this->pmat, 1, this->kv.kvec_d, this->nk); - DM_trans_real_imag.init_DMR(VR); + module_dm::DensityMatrix, double> DM_trans_real_imag(&this->pmat, 1, this->kv.kvec_d, this->nk); + DM_trans_real_imag.init_dmr(VR); hamilt::HContainer HR_real_imag(ucell, &this->pmat); LR_Util::initialize_HR, double>(HR_real_imag, ucell, gd, orb_cutoff); @@ -591,7 +591,7 @@ void HamiltBSE>::grid_calculation(hamilt::HContainer& VR) const; - inline void write_AB_matrix(const std::string& file, const int& prec, const T* ptr, const int& size1, const int& size2) + inline void write_AB_matrix(const std::string& file, const int& prec, const T* ptr, const Parallel_2D& pM) { std::ofstream ofs(file); if (!ofs.is_open()){ @@ -84,19 +86,19 @@ class HamiltBSE } ofs << file << "(Ry, transpose) with threshold " << prec << std::endl; ofs << std::setprecision(prec) << std::scientific; - LR_Util::write_value(ofs, ptr, size1, size2); + LR_Util::write_value(ofs, ptr, pM.get_row_size(), pM.get_col_size()); ofs.close(); ModuleBase::GlobalFunc::DONE(GlobalV::ofs_running, "finish writing " + file); } - inline void read_AB_matrix(const std::string& file, T* ptr, const int& size1, const int& size2) + inline void read_AB_matrix(const std::string& file, T* ptr, const Parallel_2D& pM) { std::ifstream ifs(file); if (!ifs.is_open()){ throw std::runtime_error("Cannot open file " + file); } ifs.ignore(std::numeric_limits::max(), '\n'); // skip the first line - LR_Util::read_value(ifs, ptr, size1, size2); + LR_Util::read_value(ifs, ptr, pM.get_row_size(), pM.get_col_size()); ifs.close(); ModuleBase::GlobalFunc::DONE(GlobalV::ofs_running, "finish reading " + file); } @@ -131,6 +133,8 @@ class HamiltBSE const int nproc; const std::string ri_hartree_benchmark; - std::unique_ptr> DM_trans = nullptr; + std::unique_ptr> DM_trans = nullptr; }; } // namespace BSE + +#endif // ABACUS_SOURCE_LCAO_MODULE_BSE_HAMILT_BSE_H diff --git a/source/source_lcao/module_bse/hamilt_bse_solver.h b/source/source_lcao/module_bse/hamilt_bse_solver.h index 35798716e4e..3a34259ee46 100644 --- a/source/source_lcao/module_bse/hamilt_bse_solver.h +++ b/source/source_lcao/module_bse/hamilt_bse_solver.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_LCAO_MODULE_BSE_HAMILT_BSE_SOLVER_H +#define ABACUS_SOURCE_LCAO_MODULE_BSE_HAMILT_BSE_SOLVER_H + #include #include @@ -44,3 +46,5 @@ void solve_tda(const std::vector& A, } #include "hamilt_bse_solver.hpp" + +#endif // ABACUS_SOURCE_LCAO_MODULE_BSE_HAMILT_BSE_SOLVER_H diff --git a/source/source_lcao/module_bse/hamilt_bse_solver.hpp b/source/source_lcao/module_bse/hamilt_bse_solver.hpp index b098918ff82..54d77040905 100644 --- a/source/source_lcao/module_bse/hamilt_bse_solver.hpp +++ b/source/source_lcao/module_bse/hamilt_bse_solver.hpp @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_LCAO_MODULE_BSE_HAMILT_BSE_SOLVER_HPP +#define ABACUS_SOURCE_LCAO_MODULE_BSE_HAMILT_BSE_SOLVER_HPP + #include "hamilt_bse_solver.h" #include "source_base/timer.h" @@ -87,3 +89,5 @@ void solve_tda(const std::vector& A, #endif } } + +#endif // ABACUS_SOURCE_LCAO_MODULE_BSE_HAMILT_BSE_SOLVER_HPP diff --git a/source/source_lcao/module_bse/molecular_lri.h b/source/source_lcao/module_bse/molecular_lri.h index 238afc14ab3..70c87f60acd 100644 --- a/source/source_lcao/module_bse/molecular_lri.h +++ b/source/source_lcao/module_bse/molecular_lri.h @@ -2,7 +2,9 @@ // AUTHOR : Ziqing Guan // DATE : 2026-03-22 //======================= -#pragma once +#ifndef ABACUS_SOURCE_LCAO_MODULE_BSE_MOLECULAR_LRI_H +#define ABACUS_SOURCE_LCAO_MODULE_BSE_MOLECULAR_LRI_H + #include #include #include "source_base/timer.h" @@ -177,3 +179,5 @@ class MolecularLRI #include "molecular_lri.hpp" #include "molecular_lri_comm.hpp" + +#endif // ABACUS_SOURCE_LCAO_MODULE_BSE_MOLECULAR_LRI_H diff --git a/source/source_lcao/module_bse/molecular_lri.hpp b/source/source_lcao/module_bse/molecular_lri.hpp index 38266153b74..f11e4f49246 100644 --- a/source/source_lcao/module_bse/molecular_lri.hpp +++ b/source/source_lcao/module_bse/molecular_lri.hpp @@ -12,9 +12,6 @@ #include #endif -#ifdef __MKL -#include -#endif namespace BSE { @@ -62,7 +59,7 @@ void MolecularLRI::init(TLRI& Cs_in, TLRI& Vs_in, TLRI& Ws_in, const { for (const TC R_original : this->kRlist.Rlist) { - const TC R = cell_nearest.cell_nearest_direction(i, j, R_original, dist); + const TC R = cell_nearest.cell_nearest_direction(i, j, R_original, dist); if (R != R_original) { BSE_Util::move_R_tensor(Cs_in, i, j, R_original, R); diff --git a/source/source_lcao/module_deepks/lcao_deepks.cpp b/source/source_lcao/module_deepks/lcao_deepks.cpp index 1e6fcea0a02..170a95d2f58 100644 --- a/source/source_lcao/module_deepks/lcao_deepks.cpp +++ b/source/source_lcao/module_deepks/lcao_deepks.cpp @@ -237,7 +237,7 @@ void LCAO_Deepks::allocate_V_delta(const int nat, const int nks) } template -void LCAO_Deepks::init_DMR(const UnitCell& ucell, +void LCAO_Deepks::init_dmr(const UnitCell& ucell, const LCAO_Orbitals& orb, const Parallel_Orbitals& pv, const Grid_Driver& GridD) diff --git a/source/source_lcao/module_deepks/lcao_deepks.h b/source/source_lcao/module_deepks/lcao_deepks.h index 82f9279d518..03616fd5949 100644 --- a/source/source_lcao/module_deepks/lcao_deepks.h +++ b/source/source_lcao/module_deepks/lcao_deepks.h @@ -127,7 +127,7 @@ class LCAO_Deepks void allocate_V_delta(const int nat, const int nks = 1); /// Initialize the dm_r container - void init_DMR(const UnitCell& ucell, + void init_dmr(const UnitCell& ucell, const LCAO_Orbitals& orb, const Parallel_Orbitals& pv, const Grid_Driver& GridD); diff --git a/source/source_lcao/module_deepks/lcao_deepks_iface.cpp b/source/source_lcao/module_deepks/lcao_deepks_iface.cpp index 7be56bda99a..55bb0a162dd 100644 --- a/source/source_lcao/module_deepks/lcao_deepks_iface.cpp +++ b/source/source_lcao/module_deepks/lcao_deepks_iface.cpp @@ -71,7 +71,7 @@ void LCAO_Deepks_Interface::out_deepks_labels(const double& etot, const Grid_Driver& GridD, const Parallel_Orbitals* ParaV, const psi::Psi& psi, - const elecstate::DensityMatrix* dm, + const module_dm::DensityMatrix* dm, hamilt::HamiltLCAO* p_ham, const int& iter, const bool& conv_esolver, @@ -116,7 +116,7 @@ void LCAO_Deepks_Interface::out_deepks_labels(const double& etot, //================================================================================ // Update DMR in any case of deepks_out_labels/deepks_scf - DeePKS_domain::update_dmr(kvec_d, dm->get_DMK_vector(), ucell, orb, *ParaV, GridD, dmr); + DeePKS_domain::update_dmr(kvec_d, dm->get_dmk_vec(), ucell, orb, *ParaV, GridD, dmr); // Note : update PDM and all other quantities with the current dm // DeePKS PDM and descriptor @@ -137,7 +137,7 @@ void LCAO_Deepks_Interface::out_deepks_labels(const double& etot, std::vector descriptor_mag; if (deepks_spin2) { - DeePKS_domain::update_dmr(kvec_d, dm->get_DMK_vector(), ucell, orb, *ParaV, GridD, ld->dm_r_mag, 2, true); + DeePKS_domain::update_dmr(kvec_d, dm->get_dmk_vec(), ucell, orb, *ParaV, GridD, ld->dm_r_mag, 2, true); bool init_pdm_mag = false; DeePKS_domain::cal_pdm< TK>(init_pdm_mag, deepks_param, kvec_d, ld->dm_r_mag, phialpha, ucell, orb, GridD, *ParaV, ld->pdm_mag); @@ -794,7 +794,7 @@ void LCAO_Deepks_Interface::out_deepks_labels(const double& etot, /// print out deepks information to the screen if (PARAM.inp.deepks_scf) { - DeePKS_domain::cal_e_delta_band(dm->get_DMK_vector(), *h_delta, nks, nspin, ParaV, e_delta_band); + DeePKS_domain::cal_e_delta_band(dm->get_dmk_vec(), *h_delta, nks, nspin, ParaV, e_delta_band); if (rank == 0) { ofs_running << " DeePKS Energy Correction" << std::endl; @@ -808,7 +808,7 @@ void LCAO_Deepks_Interface::out_deepks_labels(const double& etot, } if (PARAM.inp.deepks_out_unittest) { - LCAO_deepks_io::print_dm(nks, PARAM.globalv.nlocal, ParaV->nrow, dm->get_DMK_vector()); + LCAO_deepks_io::print_dm(nks, PARAM.globalv.nlocal, ParaV->nrow, dm->get_dmk_vec()); DeePKS_domain::check_gedm(deepks_param, ld->gedm); diff --git a/source/source_lcao/module_deepks/lcao_deepks_iface.h b/source/source_lcao/module_deepks/lcao_deepks_iface.h index 508e1f6571b..dc015c4840b 100644 --- a/source/source_lcao/module_deepks/lcao_deepks_iface.h +++ b/source/source_lcao/module_deepks/lcao_deepks_iface.h @@ -40,7 +40,7 @@ class LCAO_Deepks_Interface const Grid_Driver& GridD, const Parallel_Orbitals* ParaV, const psi::Psi& psid, - const elecstate::DensityMatrix* dm, + const module_dm::DensityMatrix* dm, hamilt::HamiltLCAO* p_ham, const int& iter, const bool& conv_esolver, diff --git a/source/source_lcao/module_deepks/test/CMakeLists.txt b/source/source_lcao/module_deepks/test/CMakeLists.txt index 15cca5c62d6..56748e172c7 100644 --- a/source/source_lcao/module_deepks/test/CMakeLists.txt +++ b/source/source_lcao/module_deepks/test/CMakeLists.txt @@ -45,7 +45,19 @@ set(DEEPKS_UNIT_COMMON_SOURCES ../../../source_cell/read_orb.cpp ../../../source_cell/cal_nelec_nband.cpp ../../../source_estate/module_dm/density_matrix.cpp - ../../../source_estate/module_dm/density_matrix_io.cpp + ../../../source_estate/module_dm/dmr_gamma.cpp + ../../../source_estate/module_dm/dmr_init.cpp + ../../../source_estate/module_dm/dm_setter.cpp + ../../../source_estate/module_dm/dm_getter.cpp + ../../../source_estate/module_dm/dm_tools.cpp + ../../../source_estate/module_dm/dmr_k.cpp + ../../../source_estate/module_dm/dmr_td.cpp + ../../../source_estate/module_dm/dmr_full.cpp + # dm_from_psi.cpp defines module_dm::dmk_from_psi / dm_from_psi, referenced via + # cal_dm.h by deepks_lcao.cpp / operator_lcao.cpp. It was dropped when + # density_matrix.cpp was split (dmr_* / dm_*), causing undefined reference + # to dmk_from_psi at link time. + ../../../source_estate/module_dm/dm_from_psi.cpp ../../center2orb.cpp ../../center2orb_orb11.cpp ../../center2orb_orb21.cpp @@ -59,9 +71,12 @@ set(DEEPKS_UNIT_COMMON_SOURCES ../../module_operator_lcao/deepks_lcao.cpp ../../module_operator_lcao/operator_lcao.cpp ../../../source_hamilt/operator.cpp - ../../../source_io/module_hs/cal_r_overlap_r.cpp - ../../../source_io/module_hs/single_r_io.cpp - ../../../source_io/module_hs/rr_sparse_writer.cpp + ../../../source_io/module_hs/pos_op_mat.cpp + ../../../source_io/module_hs/pos_op_basis.cpp + ../../../source_io/module_hs/pos_op_calc.cpp + ../../../source_io/module_hs/pos_op_writer.cpp + ../../../source_io/module_hs/lat_r_csr.cpp + ../../../source_io/module_hs/pos_op_csr.cpp ../../module_rt/radial_interpolation.cpp ../../module_rt/snap_proj_half_tddft.cpp ../../module_rt/snap_pha_half_tddft.cpp diff --git a/source/source_lcao/module_deepks/test/deepks_test.h b/source/source_lcao/module_deepks/test/deepks_test.h index e1f3a4fc558..fabde44acfe 100644 --- a/source/source_lcao/module_deepks/test/deepks_test.h +++ b/source/source_lcao/module_deepks/test/deepks_test.h @@ -68,7 +68,7 @@ class test_deepks std::vector dm; std::vector> dm_new; - elecstate::DensityMatrix* p_elec_DM = nullptr; + module_dm::DensityMatrix* p_elec_DM = nullptr; // preparation void preparation(bool use_modern_orbital_reader); diff --git a/source/source_lcao/module_deepks/test/deepks_test_pdm.cpp b/source/source_lcao/module_deepks/test/deepks_test_pdm.cpp index c5d1524a812..d934ee6fd42 100644 --- a/source/source_lcao/module_deepks/test/deepks_test_pdm.cpp +++ b/source/source_lcao/module_deepks/test/deepks_test_pdm.cpp @@ -55,21 +55,21 @@ void test_deepks::set_p_elec_DM() if (this->gamma_only_local) { nk = this->nspin; - this->p_elec_DM = new elecstate::DensityMatrix(&ParaO, this->nspin); + this->p_elec_DM = new module_dm::DensityMatrix(&ParaO, this->nspin); } else { nk = kv.get_nkstot(); this->p_elec_DM - = new elecstate::DensityMatrix(&ParaO, this->nspin, kv.kvec_d, kv.get_nkstot() / this->nspin); + = new module_dm::DensityMatrix(&ParaO, this->nspin, kv.kvec_d, kv.get_nkstot() / this->nspin); } - p_elec_DM->init_DMR(&Test_Deepks::GridD, &ucell); + p_elec_DM->init_dmr(&Test_Deepks::GridD, &ucell); for (int ik = 0; ik < nk; ik++) { - p_elec_DM->set_DMK_pointer(ik, dm_new[ik].data()); + p_elec_DM->set_dmk_ptr(ik, dm_new[ik].data()); } - p_elec_DM->cal_DMR(); + p_elec_DM->cal_dmr(-1); } template @@ -78,9 +78,9 @@ void test_deepks::check_pdm() this->read_dm(kv.get_nkstot()); this->set_dm_new(); this->set_p_elec_DM(); - this->ld.init_DMR(ucell, ORB, ParaO, Test_Deepks::GridD); + this->ld.init_dmr(ucell, ORB, ParaO, Test_Deepks::GridD); DeePKS_domain::update_dmr(kv.kvec_d, - p_elec_DM->get_DMK_vector(), + p_elec_DM->get_dmk_vec(), ucell, ORB, ParaO, diff --git a/source/source_lcao/module_deepks/test/deepks_test_prep.cpp b/source/source_lcao/module_deepks/test/deepks_test_prep.cpp index 8eee3722856..94c6fd944f8 100644 --- a/source/source_lcao/module_deepks/test/deepks_test_prep.cpp +++ b/source/source_lcao/module_deepks/test/deepks_test_prep.cpp @@ -282,7 +282,7 @@ void test_deepks::set_orbs(const bool use_modern_orbital_reader) auto* lcao_nl = new LCAONonlocalInfo(); lcao_nl->setupNonlocal(ucell.ntype, ucell.atoms, GlobalV::ofs_running, ORB, - basis_type, out_element_info, lspinorb, nspin); + basis_type, out_element_info, lspinorb, nspin, GlobalV::MY_RANK); ucell.infoNL.reset(lcao_nl); double rmax = std::max(orb_.rcut_max(), alpha_.rcut_max()); diff --git a/source/source_lcao/module_deepks/test/mock_tdinfo.cpp b/source/source_lcao/module_deepks/test/mock_tdinfo.cpp index d61e33157a0..6c5e768ba3a 100644 --- a/source/source_lcao/module_deepks/test/mock_tdinfo.cpp +++ b/source/source_lcao/module_deepks/test/mock_tdinfo.cpp @@ -14,9 +14,9 @@ class TD_info { return nullptr; } - static ModuleBase::Vector3 cart_At; + static ModuleBase::Vector3 A_prop_ha; static TD_info* td_vel_op; }; TD_info td_info; TD_info* TD_info::td_vel_op = &td_info; -ModuleBase::Vector3 TD_info::cart_At(0.0, 0.0, 0.0); +ModuleBase::Vector3 TD_info::A_prop_ha(0.0, 0.0, 0.0); diff --git a/source/source_lcao/module_deltaspin/cal_mw_from_lambda.cpp b/source/source_lcao/module_deltaspin/cal_mw_from_lambda.cpp index a148a0b92fd..2b0d39992e3 100644 --- a/source/source_lcao/module_deltaspin/cal_mw_from_lambda.cpp +++ b/source/source_lcao/module_deltaspin/cal_mw_from_lambda.cpp @@ -7,15 +7,16 @@ #include "source_base/tool_title.h" #include "source_estate/elecstate_tools.h" #include "source_hsolver/diag_comm_info.h" +#include "source_hamilt/hamilt_hs_adapter.h" #include "source_hsolver/diago_iter_assist.h" #include "source_hsolver/hsolver_lcao.h" #include "source_io/module_parameter/parameter.h" -#include "source_pw/module_pwdft/onsite_proj.h" +#include "source_pw/module_proj/onsite_proj.h" #include "spin_constrain.h" #ifdef __LCAO #include "source_estate/elecstate_lcao.h" -#include "source_estate/module_dm/cal_dm_psi.h" +#include "source_estate/module_dm/dm_from_psi.h" #include "source_lcao/module_operator_lcao/dspin_lcao.h" #endif @@ -88,8 +89,8 @@ */ template <> void spinconstrain::SpinConstrain>::cal_mw_from_lambda( - int i_step, - const ModuleBase::Vector3* delta_lambda) + int i_step, + const ModuleBase::Vector3* delta_lambda) { ModuleBase::TITLE("spinconstrain::SpinConstrain", "cal_mw_from_lambda"); ModuleBase::timer::start("spinconstrain::SpinConstrain", "cal_mw_from_lambda"); @@ -127,11 +128,14 @@ void spinconstrain::SpinConstrain>::cal_mw_from_lambda( this->p_operator) ->update_lambda(); } - // Diagonalization without updating charge density (last param = true means skip charge update) - hsolver_t.solve(hamilt_t, psi_t[0], this->pelec, *this->dm_, *this->pelec->charge, this->state_.nspin_, true); + // Diagonalization without updating charge density (last param = true means skip charge update). + // omega is unused here because skip_charge=true; rhopw->omega is passed only to satisfy the + // signature and would be stale in NPT anyway (see Charge::renormalize_rho). + hamilt::HamiltHSMatrix> hs(hamilt_t); + hsolver_t.solve(hs, psi_t[0], this->pelec, *this->dm_, *this->pelec->charge, this->state_.nspin_, this->pelec->charge->rhopw->omega, true); // Note: although update_lambda() modifies lambda in-place above, - // solve() unconditionally recomputes DM and DMR (via cal_dm_psi + - // cal_DMR) from the psi obtained by diagonalizing with the new + // solve() unconditionally recomputes DM and DMR (via dm_from_psi + + // cal_dmr) from the psi obtained by diagonalizing with the new // lambda. Therefore the DMR used inside cal_mi_lcao() is consistent // with the updated lambda and is NOT stale. this->cal_mi_lcao(i_step); @@ -155,6 +159,7 @@ void spinconstrain::SpinConstrain>::cal_mw_from_lambda( // ============================================================= psi::Psi>* psi_t = static_cast>*>(this->psi); hamilt::Hamilt, base_device::DEVICE_CPU>* hamilt_t = static_cast, base_device::DEVICE_CPU>*>(this->p_hamilt); + hamilt::HamiltHSOperator, base_device::DEVICE_CPU> op(hamilt_t, this->pw_wfc_); auto* onsite_p = projectors::OnsiteProjector::get_instance(); nbands = psi_t->get_nbands(); npol = psi_t->get_npol(); @@ -183,8 +188,8 @@ void spinconstrain::SpinConstrain>::cal_mw_from_lambda( if(initial_hs) { /// Compute H(k) and extract subspace matrices for this k-point - hamilt_t->updateHk(ik); - hsolver::DiagoIterAssist>::cal_hs_subspace(hamilt_t, + op.update_k(ik); + hsolver::DiagoIterAssist>::cal_hs_subspace(op, psi_t[0], h_k, s_k, @@ -214,6 +219,7 @@ void spinconstrain::SpinConstrain>::cal_mw_from_lambda( // ============================================================= psi::Psi, base_device::DEVICE_GPU>* psi_t = static_cast, base_device::DEVICE_GPU>*>(this->psi); hamilt::Hamilt, base_device::DEVICE_GPU>* hamilt_t = static_cast, base_device::DEVICE_GPU>*>(this->p_hamilt); + hamilt::HamiltHSOperator, base_device::DEVICE_GPU> op(hamilt_t, this->pw_wfc_); auto* onsite_p = projectors::OnsiteProjector::get_instance(); nbands = psi_t->get_nbands(); npol = psi_t->get_npol(); @@ -244,9 +250,9 @@ void spinconstrain::SpinConstrain>::cal_mw_from_lambda( std::complex* becp_k = this->pw_cache_.becp_k(ik, size_becp); if(initial_hs) { - hamilt_t->updateHk(ik); + op.update_k(ik); hsolver::DiagoIterAssist, base_device::DEVICE_GPU>::cal_hs_subspace( - hamilt_t, + op, psi_t[0], h_k, s_k, diff --git a/source/source_lcao/module_deltaspin/deltaspin_init.cpp b/source/source_lcao/module_deltaspin/deltaspin_init.cpp index 671c286d09c..e410e2c7fee 100644 --- a/source/source_lcao/module_deltaspin/deltaspin_init.cpp +++ b/source/source_lcao/module_deltaspin/deltaspin_init.cpp @@ -90,23 +90,23 @@ void init_sc_state(const ScInitParams& params, const UnitCell& ucell, ScState& s template void spinconstrain::SpinConstrain::init_sc(double sc_thr_in, - int nsc_in, - int nsc_min_in, - double alpha_trial_in, - double sccut_in, - double sc_drop_thr_in, - const UnitCell& ucell, - bool direction_only_in, - Parallel_Orbitals* ParaV_in, - int nspin_in, - const K_Vectors& kv_in, - void* p_hamilt_in, - void* psi_in, + int nsc_in, + int nsc_min_in, + double alpha_trial_in, + double sccut_in, + double sc_drop_thr_in, + const UnitCell& ucell, + bool direction_only_in, + Parallel_Orbitals* ParaV_in, + int nspin_in, + const K_Vectors& kv_in, + void* p_hamilt_in, + void* psi_in, #ifdef __LCAO - elecstate::DensityMatrix* dm_in, // mohan add 2025-11-03 + module_dm::DensityMatrix* dm_in, // mohan add 2025-11-03 #endif - elecstate::ElecState* pelec_in, - ModulePW::PW_Basis_K* pw_wfc_in) + elecstate::ElecState* pelec_in, + ModulePW::PW_Basis_K* pw_wfc_in) { // Steps 1-6: solver-independent state initialization const spinconstrain::ScInitParams params{sc_thr_in, nsc_in, nsc_min_in, diff --git a/source/source_lcao/module_deltaspin/deltaspin_lcao.cpp b/source/source_lcao/module_deltaspin/deltaspin_lcao.cpp index 42eac35ee41..7540e460b5d 100644 --- a/source/source_lcao/module_deltaspin/deltaspin_lcao.cpp +++ b/source/source_lcao/module_deltaspin/deltaspin_lcao.cpp @@ -64,7 +64,7 @@ void init_deltaspin_lcao(const UnitCell& ucell, inp.sccut, inp.sc_drop_thr, ucell, inp.sc_direction_only, static_cast(pv), inp.nspin, kv, p_hamilt, psi, - static_cast*>(dm), + static_cast*>(dm), static_cast(pelec)); #else // Non-LCAO build: no density matrix diff --git a/source/source_lcao/module_deltaspin/deltaspin_lcao_mi.cpp b/source/source_lcao/module_deltaspin/deltaspin_lcao_mi.cpp index 37f4bc5825c..b004057514e 100644 --- a/source/source_lcao/module_deltaspin/deltaspin_lcao_mi.cpp +++ b/source/source_lcao/module_deltaspin/deltaspin_lcao_mi.cpp @@ -39,7 +39,7 @@ namespace lcao void cal_mi_lcao(ScState& state, hamilt::Operator>* p_operator, - elecstate::DensityMatrix, double>* dm, + module_dm::DensityMatrix, double>* dm, const int& step, bool print) { @@ -47,7 +47,7 @@ void cal_mi_lcao(ScState& state, ModuleBase::timer::start("spinconstrain::SpinConstrain", "cal_mi_lcao"); // Reset Mi before calculation state.zero_Mi(); - const hamilt::HContainer* dmr = dm->get_DMR_pointer(1); + const hamilt::HContainer* dmr = dm->get_dmr_ptr(1); std::vector moments; if (state.nspin_ == 2) { diff --git a/source/source_lcao/module_deltaspin/deltaspin_lcao_mi.h b/source/source_lcao/module_deltaspin/deltaspin_lcao_mi.h index 88d63c62434..da23673eed8 100644 --- a/source/source_lcao/module_deltaspin/deltaspin_lcao_mi.h +++ b/source/source_lcao/module_deltaspin/deltaspin_lcao_mi.h @@ -30,7 +30,7 @@ #include "deltaspin_state.h" class Parallel_Orbitals; -namespace elecstate +namespace module_dm { template class DensityMatrix; @@ -57,7 +57,7 @@ namespace lcao */ void cal_mi_lcao(ScState& state, hamilt::Operator>* p_operator, - elecstate::DensityMatrix, double>* dm, + module_dm::DensityMatrix, double>* dm, const int& step, bool print = false); diff --git a/source/source_lcao/module_deltaspin/deltaspin_pw_mi.cpp b/source/source_lcao/module_deltaspin/deltaspin_pw_mi.cpp index 3a8c73d23e0..4491e706e05 100644 --- a/source/source_lcao/module_deltaspin/deltaspin_pw_mi.cpp +++ b/source/source_lcao/module_deltaspin/deltaspin_pw_mi.cpp @@ -25,11 +25,12 @@ #include "source_base/timer.h" #include "source_base/kernels/math_kernel_op.h" #include "source_base/module_device/device.h" -#include "source_pw/module_pwdft/onsite_proj.h" +#include "source_pw/module_proj/onsite_proj.h" #include "deltaspin_pw_cache.h" #include "mi_tools.h" #include "source_io/module_parameter/parameter.h" #include "source_hsolver/diago_iter_assist.h" +#include "source_hamilt/hamilt_hs_adapter.h" #include "source_hsolver/hsolver_pw.h" #include "source_estate/elecstate.h" #include "source_estate/elecstate_pw.h" @@ -376,16 +377,15 @@ void update_psi_charge_pw_cpu(ScState& state, PARAM.inp.nb2d, PARAM.inp.use_k_continuity); - hsolver_pw_obj.solve(hamilt_t, + hamilt::HamiltHSOperator, base_device::DEVICE_CPU> op(hamilt_t, pw_wfc); + hsolver_pw_obj.solve(op, psi_t[0], pelec, pelec->ekb.c, GlobalV::RANK_IN_POOL, GlobalV::NPROC_IN_POOL, GlobalV::ofs_running, - false, - state.tpiba, - state.get_nat()); + false); } else { @@ -490,16 +490,15 @@ void update_psi_charge_pw_gpu(ScState& state, PARAM.inp.nb2d, PARAM.inp.use_k_continuity); - hsolver_pw_obj.solve(hamilt_t, + hamilt::HamiltHSOperator, base_device::DEVICE_GPU> op(hamilt_t, pw_wfc); + hsolver_pw_obj.solve(op, psi_t[0], pelec, pelec->ekb.c, GlobalV::RANK_IN_POOL, GlobalV::NPROC_IN_POOL, GlobalV::ofs_running, - false, - state.tpiba, - state.get_nat()); + false); } else { diff --git a/source/source_lcao/module_deltaspin/lambda_loop.cpp b/source/source_lcao/module_deltaspin/lambda_loop.cpp index 77d848b3415..8534ef44341 100644 --- a/source/source_lcao/module_deltaspin/lambda_loop.cpp +++ b/source/source_lcao/module_deltaspin/lambda_loop.cpp @@ -100,9 +100,9 @@ void spinconstrain::SpinConstrain>::run_lambda_loop(int out // Timer initialization (MPI or CPU) #ifdef __MPI - auto iterstart = MPI_Wtime(); + auto iterstart = MPI_Wtime(); #else - auto iterstart = std::chrono::system_clock::now(); + auto iterstart = std::chrono::system_clock::now(); #endif double inner_loop_duration = 0.0; @@ -179,9 +179,9 @@ void spinconstrain::SpinConstrain>::run_lambda_loop(int out add_scalar_multiply_2d(initial_lambda, dnu_last_step, one, this->state_.lambda_); this->update_psi_charge(dnu_last_step.data(), true, true); #ifdef __MPI - duration = (double)(MPI_Wtime() - iterstart); + duration = (double)(MPI_Wtime() - iterstart); #else - duration = + duration = (std::chrono::duration_cast(std::chrono::system_clock::now() - iterstart)).count() / static_cast(1e6); #endif @@ -253,9 +253,9 @@ void spinconstrain::SpinConstrain>::run_lambda_loop(int out // CHECK CONVERGENCE // ============================================================= #ifdef __MPI - duration = (double)(MPI_Wtime() - iterstart); + duration = (double)(MPI_Wtime() - iterstart); #else - duration = + duration = (std::chrono::duration_cast(std::chrono::system_clock::now() - iterstart)).count() / static_cast(1e6); #endif @@ -299,9 +299,9 @@ void spinconstrain::SpinConstrain>::run_lambda_loop(int out // Reset timer for next iteration #ifdef __MPI - iterstart = MPI_Wtime(); + iterstart = MPI_Wtime(); #else - iterstart = std::chrono::system_clock::now(); + iterstart = std::chrono::system_clock::now(); #endif // ============================================================= diff --git a/source/source_lcao/module_deltaspin/lambda_loop_helper.cpp b/source/source_lcao/module_deltaspin/lambda_loop_helper.cpp index 307dd312103..0aa0afa3de4 100644 --- a/source/source_lcao/module_deltaspin/lambda_loop_helper.cpp +++ b/source/source_lcao/module_deltaspin/lambda_loop_helper.cpp @@ -353,6 +353,14 @@ void print_Mi(const SpinConstrain& sc, std::ostream& ofs_running) const int nspin = sc.get_nspin(); const auto& Mi = sc.get_Mi(); const auto& atomLabel = sc.get_atomLabels(); + std::vector atom_labels_iat(nat); + for (const auto& it : sc.get_atomCounts()) + { + for (int ia = 0; ia < it.second; ++ia) + { + atom_labels_iat[sc.get_iat(it.first, ia)] = atomLabel[it.first] + std::to_string(ia + 1); + } + } std::vector mag_x(nat, 0.0); std::vector mag_y(nat, 0.0); std::vector mag_z(nat, 0.0); @@ -369,7 +377,7 @@ void print_Mi(const SpinConstrain& sc, std::ostream& ofs_running) { mag_z[iat] = Mi[iat].z; } - table << atomLabel << mag_z; + table << atom_labels_iat << mag_z; ofs_running << table.str() << std::endl; } else if (nspin == 4) @@ -387,7 +395,7 @@ void print_Mi(const SpinConstrain& sc, std::ostream& ofs_running) mag_y[iat] = Mi[iat].y; mag_z[iat] = Mi[iat].z; } - table << atomLabel << mag_x << mag_y << mag_z; + table << atom_labels_iat << mag_x << mag_y << mag_z; ofs_running << table.str() << std::endl; } } @@ -406,6 +414,14 @@ void print_Mag_Force(const SpinConstrain& sc, std::ostream& ofs_running) const int nspin = sc.get_nspin(); const auto& lambda = sc.get_sc_lambda(); const auto& atomLabel = sc.get_atomLabels(); + std::vector atom_labels_iat(nat); + for (const auto& it : sc.get_atomCounts()) + { + for (int ia = 0; ia < it.second; ++ia) + { + atom_labels_iat[sc.get_iat(it.first, ia)] = atomLabel[it.first] + std::to_string(ia + 1); + } + } std::vector mag_force_x(nat, 0.0); std::vector mag_force_y(nat, 0.0); std::vector mag_force_z(nat, 0.0); @@ -422,7 +438,7 @@ void print_Mag_Force(const SpinConstrain& sc, std::ostream& ofs_running) { mag_force_z[iat] = lambda[iat].z * ModuleBase::Ry_to_eV; } - table << atomLabel << mag_force_z; + table << atom_labels_iat << mag_force_z; ofs_running << table.str() << std::endl; } else if (nspin == 4) @@ -440,7 +456,7 @@ void print_Mag_Force(const SpinConstrain& sc, std::ostream& ofs_running) mag_force_y[iat] = lambda[iat].y * ModuleBase::Ry_to_eV; mag_force_z[iat] = lambda[iat].z * ModuleBase::Ry_to_eV; } - table << atomLabel << mag_force_x << mag_force_y << mag_force_z; + table << atom_labels_iat << mag_force_x << mag_force_y << mag_force_z; ofs_running << table.str() << std::endl; } } diff --git a/source/source_lcao/module_deltaspin/mi_tools.cpp b/source/source_lcao/module_deltaspin/mi_tools.cpp index 8e38b05e29b..af9abe6e290 100644 --- a/source/source_lcao/module_deltaspin/mi_tools.cpp +++ b/source/source_lcao/module_deltaspin/mi_tools.cpp @@ -1,6 +1,7 @@ #include "mi_tools.h" #include "source_base/tool_quit.h" +#include "source_estate/occ_comput.h" namespace spinconstrain { @@ -43,49 +44,54 @@ void accumulate_Mi_from_becp(const std::complex* becp, ModuleBase::WARNING_QUIT("accumulate_Mi_from_becp", "spin_sign must be -1 or 1"); } - const std::complex zero(0.0, 0.0); - if (npol == 2) + // Compute the per-projector 2x2 occupation blocks with the shared core. + // npol=2 -> nspin=4 (full spin density matrix); npol=1 -> nspin=2 with the + // spin channel selected by isk (spin_sign=+1 -> isk=0, -1 -> isk=1). + // nspin=1 is never used here: DeltaSpin only runs with nspin=2 or 4. + const int nat = static_cast(mi.size()); + const int nspin = (npol == 2) ? 4 : 2; + const int isk = (spin_sign == 1) ? 0 : 1; + std::vector> occ_block(nkb * 4, std::complex(0.0, 0.0)); + elecstate::occ_from_proj( + becp, + wg_ik, + nbands, + npol, + nkb, + nspin, + isk, + nh_iat, + nat, + occ_block.data()); + + // Aggregate the per-projector blocks into per-atom magnetic moments. + // The blocks are weighted per band inside occ_from_proj, so the aggregate + // weight is 1 here; pauli_to_moment and the z-difference read the already + // weighted block sums. + const double unit_weight = 1.0; + int begin_iprj = 0; + for (int iat = 0; iat < nat; iat++) { - for (int ib = 0; ib < nbands; ib++) + const int nprj = nh_iat[iat]; + if (npol == 2) { - const double weight = wg_ik[ib]; - int begin_ih = 0; - for (int iat = 0; iat < static_cast(mi.size()); iat++) + // Mi = sum_iprj pauli_to_moment(block_iprj) + for (int iprj = 0; iprj < nprj; iprj++) { - std::complex occ[4] = {zero, zero, zero, zero}; - const int nh = nh_iat[iat]; - for (int ih = 0; ih < nh; ih++) - { - const int index = ib * 2 * nkb + begin_ih + ih; - occ[0] += conj(becp[index]) * becp[index]; - occ[1] += conj(becp[index]) * becp[index + nkb]; - occ[2] += conj(becp[index + nkb]) * becp[index]; - occ[3] += conj(becp[index + nkb]) * becp[index + nkb]; - } - mi[iat] += pauli_to_moment(occ, weight); - begin_ih += nh; + const int occ_index = (begin_iprj + iprj) * 4; + mi[iat] += pauli_to_moment(&occ_block[occ_index], unit_weight); } } - } - else // npol == 1 - { - for (int ib = 0; ib < nbands; ib++) + else { - const double weight = wg_ik[ib]; - int begin_ih = 0; - for (int iat = 0; iat < static_cast(mi.size()); iat++) + // Mz = sum_iprj (occ[0] - occ[3]) == weight * occ * spin_sign + for (int iprj = 0; iprj < nprj; iprj++) { - double occ = 0.0; - const int nh = nh_iat[iat]; - for (int ih = 0; ih < nh; ih++) - { - const int index = ib * nkb + begin_ih + ih; - occ += (conj(becp[index]) * becp[index]).real(); - } - mi[iat].z += weight * occ * spin_sign; - begin_ih += nh; + const int occ_index = (begin_iprj + iprj) * 4; + mi[iat].z += (occ_block[occ_index] - occ_block[occ_index + 3]).real(); } } + begin_iprj += nprj; } } diff --git a/source/source_lcao/module_deltaspin/spin_constrain.h b/source/source_lcao/module_deltaspin/spin_constrain.h index 1af2c116273..7dba0c2b89d 100644 --- a/source/source_lcao/module_deltaspin/spin_constrain.h +++ b/source/source_lcao/module_deltaspin/spin_constrain.h @@ -154,9 +154,9 @@ class SpinConstrain void* p_hamilt_in, void* psi_in, #ifdef __LCAO - elecstate::DensityMatrix *dm_in, // mohan add 2025-11-02 + module_dm::DensityMatrix *dm_in, // mohan add 2025-11-02 #endif - elecstate::ElecState* pelec_in, + elecstate::ElecState* pelec_in, ModulePW::PW_Basis_K* pw_wfc_in = nullptr); /** @@ -198,7 +198,7 @@ class SpinConstrain * @param delta_lambda Change in lambda from previous step (for incremental H correction) */ void cal_mw_from_lambda(int i_step, - const ModuleBase::Vector3* delta_lambda = nullptr); + const ModuleBase::Vector3* delta_lambda = nullptr); /** * @brief Calculate the spin constraint energy contribution: E_scon = -sum(lambda_i . Mi_i). @@ -228,8 +228,8 @@ class SpinConstrain * @param rerun If true, use full PW solver for final charge update */ void run_lambda_loop(int outer_step, - bool rerun, - std::ostream& ofs_running); + bool rerun, + std::ostream& ofs_running); /// @brief RMS error of the most recent lambda optimization loop (-1.0 if none has run). double get_last_rms_error() const { return last_rms_error_; } @@ -303,7 +303,7 @@ class SpinConstrain elecstate::ElecState* pelec = nullptr; ///< Electronic state: ekb, wg, charge, klist ModulePW::PW_Basis_K* pw_wfc_ = nullptr; ///< PW basis for wavefunction storage (PW only) #ifdef __LCAO - elecstate::DensityMatrix* dm_; ///< Density matrix pointer (LCAO only) + module_dm::DensityMatrix* dm_; ///< Density matrix pointer (LCAO only) #endif const double meV_to_Ry = 7.349864435130999e-05; ///< Conversion factor K_Vectors kv_; ///< K-point vector list @@ -361,7 +361,7 @@ class SpinConstrain /// check atomCounts void check_atomCounts() const { state_.check_atomCounts(); } /// get iat - int get_iat(int itype, int atom_index) { return state_.get_iat(itype, atom_index); } + int get_iat(int itype, int atom_index) const { return state_.get_iat(itype, atom_index); } /// set nspin void set_nspin(int nspin) { state_.set_nspin(nspin); } /// get nspin diff --git a/source/source_lcao/module_deltaspin/test/CMakeLists.txt b/source/source_lcao/module_deltaspin/test/CMakeLists.txt index 2204eabff06..ea5b2a90fce 100644 --- a/source/source_lcao/module_deltaspin/test/CMakeLists.txt +++ b/source/source_lcao/module_deltaspin/test/CMakeLists.txt @@ -45,6 +45,8 @@ AddTest( TARGET deltaspin_core_test LIBS base device SOURCES deltaspin_core_test.cpp + ../mi_tools.cpp + ../../../source_estate/occ_comput.cpp ) AddTest( diff --git a/source/source_lcao/module_deltaspin/test/deltaspin_core_test.cpp b/source/source_lcao/module_deltaspin/test/deltaspin_core_test.cpp index eba455e92dc..a7902d34608 100644 --- a/source/source_lcao/module_deltaspin/test/deltaspin_core_test.cpp +++ b/source/source_lcao/module_deltaspin/test/deltaspin_core_test.cpp @@ -302,6 +302,93 @@ TEST_F(AccumulateMiTest, Npol2_PureZMag) EXPECT_NEAR(Mi.z, 0.49, 1e-15); } +// ===================================================================== +// 3b. Real spinconstrain::accumulate_Mi_from_becp vs the oracle above +// +// The oracle helpers reimplement the pre-refactor inline loops. These +// tests call the actual function (now built on elecstate::occ_from_proj) +// and require bit-for-bit agreement with the oracle on the same inputs. +// ===================================================================== + +TEST_F(AccumulateMiTest, RealFunction_Npol1_MatchesOracle) +{ + // two atoms: nh = {2, 1}, nkb = 3, two bands + const int nbands = 2, nkb = 3; + const int nh[2] = {2, 1}; + const double wg[2] = {1.5, 0.5}; + std::vector> becp(nbands * nkb); + for (size_t i = 0; i < becp.size(); i++) + { + becp[i] = std::complex(0.1 * (i + 1), 0.02 * (i + 1)); + } + + for (int isk : {0, 1}) + { + const int spin_sign = (isk == 0) ? 1 : -1; + std::vector> mi(2); + spinconstrain::accumulate_Mi_from_becp(becp.data(), nkb, nbands, 1, + spin_sign, wg, nh, mi); + + // oracle: sum over atoms/bands of sign * w * |becp|^2 per atom + double want0 = 0.0; + double want1 = 0.0; + int begin = 0; + for (int iat = 0; iat < 2; iat++) + { + double acc = 0.0; + for (int ib = 0; ib < nbands; ib++) + { + for (int ip = 0; ip < nh[iat]; ip++) + { + const int idx = ib * nkb + begin + ip; + acc += wg[ib] * (std::conj(becp[idx]) * becp[idx]).real(); + } + } + if (iat == 0) { want0 = spin_sign * acc; } + else { want1 = spin_sign * acc; } + begin += nh[iat]; + } + EXPECT_NEAR(mi[0].z, want0, 1e-12); + EXPECT_NEAR(mi[1].z, want1, 1e-12); + EXPECT_NEAR(mi[0].x, 0.0, 1e-12); + EXPECT_NEAR(mi[0].y, 0.0, 1e-12); + } +} + +TEST_F(AccumulateMiTest, RealFunction_Npol2_MatchesOracle) +{ + const int nbands = 2, nkb = 2; + const int nh[1] = {2}; + const double wg[2] = {1.0, 2.0}; + std::vector> becp(nbands * 2 * nkb); + for (size_t i = 0; i < becp.size(); i++) + { + becp[i] = std::complex(0.05 * (i + 1), 0.03 * (i + 1)); + } + + std::vector> mi(1); + spinconstrain::accumulate_Mi_from_becp(becp.data(), nkb, nbands, 2, + 1, wg, nh, mi); + + // oracle: single atom, pauli_to_moment over summed blocks per band + std::complex occ[4] = {{0, 0}, {0, 0}, {0, 0}, {0, 0}}; + for (int ib = 0; ib < nbands; ib++) + { + for (int ip = 0; ip < nh[0]; ip++) + { + const int up = ib * 2 * nkb + ip; + const int dn = up + nkb; + occ[0] += wg[ib] * std::conj(becp[up]) * becp[up]; + occ[1] += wg[ib] * std::conj(becp[up]) * becp[dn]; + occ[2] += wg[ib] * std::conj(becp[dn]) * becp[up]; + occ[3] += wg[ib] * std::conj(becp[dn]) * becp[dn]; + } + } + EXPECT_NEAR(mi[0].x, (occ[1] + occ[2]).real(), 1e-12); + EXPECT_NEAR(mi[0].y, (occ[1] - occ[2]).imag(), 1e-12); + EXPECT_NEAR(mi[0].z, (occ[0] - occ[3]).real(), 1e-12); +} + // ===================================================================== // 4. Adaptive threshold calculation // diff --git a/source/source_lcao/module_dftu/CMakeLists.txt b/source/source_lcao/module_dftu/CMakeLists.txt index bf9626bad41..78fa877dc0d 100644 --- a/source/source_lcao/module_dftu/CMakeLists.txt +++ b/source/source_lcao/module_dftu/CMakeLists.txt @@ -11,6 +11,7 @@ if(ENABLE_LCAO) dftu_nao_op.cpp dftu_nao_adj.cpp dftu_nao_fs_r.cpp + dftu_nao_fs_reduce.cpp dftu_nao_for_r.cpp dftu_nao_str_r.cpp dftu_nao_op_legacy.cpp @@ -26,6 +27,6 @@ if(ENABLE_LCAO) endif() if(BUILD_TESTING) - add_subdirectory(test) + add_subdirectory(unittests) endif() endif() diff --git a/source/source_lcao/module_dftu/dftu_nao_adj.cpp b/source/source_lcao/module_dftu/dftu_nao_adj.cpp index b14be44bb0a..ccf654b4251 100644 --- a/source/source_lcao/module_dftu/dftu_nao_adj.cpp +++ b/source/source_lcao/module_dftu/dftu_nao_adj.cpp @@ -64,7 +64,7 @@ NlmTot cal_nlm_all(const UnitCell& ucell, const Parallel_Orbitals& pv) { ModuleBase::TITLE("DFTU", "cal_nlm_all"); - ModuleBase::timer::start("DFTU", "cal_nlm_all"); + ModuleBase::timer::start("DFTU_LCAO", "cal_nlm_all"); NlmTot nlm_tot(ucell.nat); const int npol = ucell.get_npol(); @@ -136,7 +136,7 @@ NlmTot cal_nlm_all(const UnitCell& ucell, } } } - ModuleBase::timer::end("DFTU", "cal_nlm_all"); + ModuleBase::timer::end("DFTU_LCAO", "cal_nlm_all"); return nlm_tot; } diff --git a/source/source_lcao/module_dftu/dftu_nao_energy.cpp b/source/source_lcao/module_dftu/dftu_nao_energy.cpp index 4e58c8137b0..3800ad3eab2 100644 --- a/source/source_lcao/module_dftu/dftu_nao_energy.cpp +++ b/source/source_lcao/module_dftu/dftu_nao_energy.cpp @@ -9,14 +9,13 @@ namespace DFTU_LCAO { /** * @brief Accumulate the DFT+U energy term (0.5 * U * (n - n^2)) for one - * (T, iat, l, n=0) channel in the collinear case (nspin=1 or 2). + * (T, iat, l) channel in the collinear case (nspin=1 or 2). * Returns the per-atom contribution to energy_u. */ double calc_energy_u_collinear(const Plus_U_Base& dftu, int T, int iat, - int l, - int n) + int l) { double energy_u_local = 0.0; const int m_tot = 2 * l + 1; @@ -27,16 +26,16 @@ double calc_energy_u_collinear(const Plus_U_Base& dftu, for (int m0 = 0; m0 < m_tot; m0++) { - nm_trace += dftu.occmat().get(iat, l, n, spin, m0, m0); + nm_trace += dftu.occmat().get(iat, l, spin, m0, m0); for (int m1 = 0; m1 < m_tot; m1++) { - nm2_trace += dftu.occmat().get(iat, l, n, spin, m0, m1) - * dftu.occmat().get(iat, l, n, spin, m1, m0); + nm2_trace += dftu.occmat().get(iat, l, spin, m0, m1) + * dftu.occmat().get(iat, l, spin, m1, m0); } } if (dftu.use_yukawa()) { - energy_u_local += 0.5 * (dftu.yukawa().get_U(T, l, n) - dftu.yukawa().get_J(T, l, n)) + energy_u_local += 0.5 * (dftu.yukawa().get_U(T, l) - dftu.yukawa().get_J(T, l)) * (nm_trace - nm2_trace); } else @@ -48,15 +47,14 @@ double calc_energy_u_collinear(const Plus_U_Base& dftu, } /** - * @brief Accumulate the DFT+U energy term for one (T, iat, l, n=0) channel + * @brief Accumulate the DFT+U energy term for one (T, iat, l) channel * in the noncollinear case (nspin=4). Returns the per-atom * contribution to energy_u. */ double calc_energy_u_noncollinear(const Plus_U_Base& dftu, int T, int iat, - int l, - int n) + int l) { double energy_u_local = 0.0; const int m_tot = 2 * l + 1; @@ -68,22 +66,22 @@ double calc_energy_u_noncollinear(const Plus_U_Base& dftu, for (int ipol0 = 0; ipol0 < 2; ipol0++) { const int m0_all = m0 + m_tot * ipol0; - nm_trace += dftu.occmat().get(iat, l, n, 0, m0_all, m0_all); + nm_trace += dftu.occmat().get(iat, l, 0, m0_all, m0_all); for (int m1 = 0; m1 < m_tot; m1++) { for (int ipol1 = 0; ipol1 < 2; ipol1++) { const int m1_all = m1 + m_tot * ipol1; - nm2_trace += dftu.occmat().get(iat, l, n, 0, m0_all, m1_all) - * dftu.occmat().get(iat, l, n, 0, m1_all, m0_all); + nm2_trace += dftu.occmat().get(iat, l, 0, m0_all, m1_all) + * dftu.occmat().get(iat, l, 0, m1_all, m0_all); } } } } if (dftu.use_yukawa()) { - energy_u_local += 0.5 * (dftu.yukawa().get_U(T, l, n) - dftu.yukawa().get_J(T, l, n)) + energy_u_local += 0.5 * (dftu.yukawa().get_U(T, l) - dftu.yukawa().get_J(T, l)) * (nm_trace - nm2_trace); } else @@ -95,7 +93,7 @@ double calc_energy_u_noncollinear(const Plus_U_Base& dftu, /** * @brief Accumulate the double-counting correction energy_dc for one - * (T, iat, l, n=0) channel by summing onsite_pot * occ over the + * (T, iat, l) channel by summing onsite_pot * occ over the * (m1, ipol1, m2, ipol2) grid. Dispatches on nspin to choose the * spin loop count. Returns the per-atom contribution to energy_dc. */ @@ -103,7 +101,6 @@ double calc_energy_dc_block(const Plus_U_Base& dftu, int T, int iat, int l, - int n, int nspin) { double energy_dc_local = 0.0; @@ -124,14 +121,14 @@ double calc_energy_dc_block(const Plus_U_Base& dftu, { for (int is = 0; is < 2; is++) { - const double pot_onsite = get_onsite_pot(dftu, T, iat, l, n, is, m1_all, m2_all, false); - energy_dc_local += pot_onsite * dftu.occmat().get(iat, l, n, is, m1_all, m2_all); + const double pot_onsite = get_onsite_pot(dftu, T, iat, l, is, m1_all, m2_all, false); + energy_dc_local += pot_onsite * dftu.occmat().get(iat, l, is, m1_all, m2_all); } } else if (nspin == 4) { - const double pot_onsite = get_onsite_pot(dftu, T, iat, l, n, 0, m1_all, m2_all, false); - energy_dc_local += pot_onsite * dftu.occmat().get(iat, l, n, 0, m1_all, m2_all); + const double pot_onsite = get_onsite_pot(dftu, T, iat, l, 0, m1_all, m2_all, false); + energy_dc_local += pot_onsite * dftu.occmat().get(iat, l, 0, m1_all, m2_all); } } } @@ -176,27 +173,18 @@ void cal_energy_correction(Plus_U_Base& dftu, continue; } - const int N = ucell.atoms[T].l_nchi[l]; - for (int n = 0; n < N; n++) + // part 1: U-term contribution + if (nspin == 1 || nspin == 2) { - if (n != 0) - { - continue; - } - - // part 1: U-term contribution - if (nspin == 1 || nspin == 2) - { - energy_u += calc_energy_u_collinear(dftu, T, iat, l, n); - } - else if (nspin == 4) - { - energy_u += calc_energy_u_noncollinear(dftu, T, iat, l, n); - } + energy_u += calc_energy_u_collinear(dftu, T, iat, l); + } + else if (nspin == 4) + { + energy_u += calc_energy_u_noncollinear(dftu, T, iat, l); + } - // part 2: double-counting correction - energy_dc += calc_energy_dc_block(dftu, T, iat, l, n, nspin); - } // end n + // part 2: double-counting correction + energy_dc += calc_energy_dc_block(dftu, T, iat, l, nspin); } // end L } // end I } // end T diff --git a/source/source_lcao/module_dftu/dftu_nao_energy.h b/source/source_lcao/module_dftu/dftu_nao_energy.h index ea305a47e0a..6d274e46a0f 100644 --- a/source/source_lcao/module_dftu/dftu_nao_energy.h +++ b/source/source_lcao/module_dftu/dftu_nao_energy.h @@ -23,29 +23,27 @@ void cal_energy_correction(Plus_U_Base& dftu, /** * @brief Accumulate the DFT+U energy term (U_eff / 2) * (n - n^2) for one - * (T, iat, l, n=0) channel in the collinear case (nspin=1 or 2). + * (T, iat, l) channel in the collinear case (nspin=1 or 2). * Returns the per-atom contribution to energy_u. */ double calc_energy_u_collinear(const Plus_U_Base& dftu, int T, int iat, - int l, - int n); + int l); /** - * @brief Accumulate the DFT+U energy term for one (T, iat, l, n=0) channel + * @brief Accumulate the DFT+U energy term for one (T, iat, l) channel * in the noncollinear case (nspin=4). Returns the per-atom * contribution to energy_u. */ double calc_energy_u_noncollinear(const Plus_U_Base& dftu, int T, int iat, - int l, - int n); + int l); /** * @brief Accumulate the double-counting correction energy_dc for one - * (T, iat, l, n=0) channel: + * (T, iat, l) channel: * E_dc = sum_{m1,ipol1,m2,ipol2} onsite_pot(m1,ipol1;m2,ipol2) * occ(m2,ipol2;m1,ipol1) * Returns the per-atom contribution to energy_dc. */ @@ -53,7 +51,6 @@ double calc_energy_dc_block(const Plus_U_Base& dftu, int T, int iat, int l, - int n, int nspin); } // namespace DFTU_LCAO diff --git a/source/source_lcao/module_dftu/dftu_nao_folding.cpp b/source/source_lcao/module_dftu/dftu_nao_folding.cpp index c97223d0e44..59c03f22d16 100644 --- a/source/source_lcao/module_dftu/dftu_nao_folding.cpp +++ b/source/source_lcao/module_dftu/dftu_nao_folding.cpp @@ -4,6 +4,8 @@ #include "source_cell/module_neighbor/sltk_grid_driver.h" #include "source_lcao/hamilt_lcao.h" +#include + namespace DFTU_LCAO { bool is_adjacent_pair(const std::vector& orb_cutoff, @@ -70,7 +72,7 @@ void fold_dSR_gamma(const FoldingCtx& ctx, const Parallel_Orbitals& pv = *ctx.pv; const Grid_Driver* gd = ctx.gd; - ModuleBase::GlobalFunc::ZEROS(dSR_gamma, pv.nloc); + std::fill(dSR_gamma, dSR_gamma + pv.nloc, 0.0); double* dS_ptr = nullptr; if (dim1 == 0) @@ -87,7 +89,8 @@ void fold_dSR_gamma(const FoldingCtx& ctx, } int nnr = 0; - ModuleBase::Vector3 tau1, tau2; + ModuleBase::Vector3 tau1; + ModuleBase::Vector3 tau2; for (int T1 = 0; T1 < ucell.ntype; ++T1) { @@ -152,7 +155,7 @@ void folding_matrix_k(const FoldingCtx& ctx, const ModuleBase::Vector3& kvec_d) { ModuleBase::TITLE("Plus_U", "folding_matrix_k"); - ModuleBase::timer::start("Plus_U", "folding_matrix_k"); + ModuleBase::timer::start("DFTU_LCAO", "folding_matrix_k"); const int npol = ctx.npol; const std::string& ks_solver = ctx.ks_solver; const std::vector& orb_cutoff = ctx.orb_cutoff; @@ -160,20 +163,20 @@ void folding_matrix_k(const FoldingCtx& ctx, const Parallel_Orbitals& pv = *ctx.pv; const Grid_Driver& gd = *ctx.gd; - ModuleBase::GlobalFunc::ZEROS(mat_k, pv.nloc); + std::fill(mat_k, mat_k + pv.nloc, std::complex(0.0, 0.0)); double* mat_ptr = nullptr; if (dim1 == 1 || dim1 == 4) { - mat_ptr = fsr.DSloc_Rx; + mat_ptr = fsr.DSloc_Rx.data(); } else if (dim1 == 2 || dim1 == 5) { - mat_ptr = fsr.DSloc_Ry; + mat_ptr = fsr.DSloc_Ry.data(); } else if (dim1 == 3 || dim1 == 6) { - mat_ptr = fsr.DSloc_Rz; + mat_ptr = fsr.DSloc_Rz.data(); } int nnr = 0; @@ -253,7 +256,7 @@ void folding_matrix_k(const FoldingCtx& ctx, } // ad } // I1 } // T1 - ModuleBase::timer::end("Plus_U", "folding_matrix_k"); + ModuleBase::timer::end("DFTU_LCAO", "folding_matrix_k"); return; } @@ -265,7 +268,7 @@ void folding_matrix_k_new(const std::string& ks_solver, hamilt::Hamilt>* p_ham) { ModuleBase::TITLE("Plus_U", "folding_matrix_k_new"); - ModuleBase::timer::start("Plus_U", "folding_matrix_k_new"); + ModuleBase::timer::start("DFTU_LCAO", "folding_matrix_k_new"); int hk_type = 0; if (ModuleBase::GlobalFunc::IS_COLUMN_MAJOR_KS_SOLVER(ks_solver)) @@ -293,7 +296,7 @@ void folding_matrix_k_new(const std::string& ks_solver, } } - ModuleBase::timer::end("Plus_U", "folding_matrix_k_new"); + ModuleBase::timer::end("DFTU_LCAO", "folding_matrix_k_new"); } } // namespace DFTU_LCAO diff --git a/source/source_lcao/module_dftu/dftu_nao_for_r.cpp b/source/source_lcao/module_dftu/dftu_nao_for_r.cpp index c711b678116..daa03ed5b47 100644 --- a/source/source_lcao/module_dftu/dftu_nao_for_r.cpp +++ b/source/source_lcao/module_dftu/dftu_nao_for_r.cpp @@ -20,11 +20,10 @@ void cal_for_IJR_nao_r(const int& iat1, double* force1, double* force2) { -#ifdef __DEBUG assert(nspin == 1 || nspin == 2 || nspin == 4); assert(force1 != nullptr); assert(force2 != nullptr); -#endif + // npol is the number of spinor polarizations: // 1 for nspin=1 (non-spin-polarized) and nspin=2 (collinear magnetic), // 2 for nspin=4 (non-collinear, one matrix holds both spin-up and spin-down) diff --git a/source/source_lcao/module_dftu/dftu_nao_fs_accum.h b/source/source_lcao/module_dftu/dftu_nao_fs_accum.h new file mode 100644 index 00000000000..b49ef18bfda --- /dev/null +++ b/source/source_lcao/module_dftu/dftu_nao_fs_accum.h @@ -0,0 +1,77 @@ +#ifndef DFTU_NAO_FS_ACCUM_H +#define DFTU_NAO_FS_ACCUM_H + +/// @file dftu_nao_fs_accum.h +/// @brief Diagonal-block accumulation helpers for DFT+U force/stress. +/// +/// These templates are header-only so they can be unit-tested without +/// pulling in the full DftuFsEnv closure (ScalapackConnector, folding, +/// Plus_U_Base). accumulate_onsite_force stays in dftu_nao_fs_k.cpp +/// because it needs the Plus_U_Base occupation-matrix lookup. + +#include "source_base/matrix.h" +#include "source_basis/module_ao/parallel_orbitals.h" +#include "source_cell/unitcell.h" + +#include +#include + +namespace DFTU_LCAO +{ + +/// @brief Add the real part of diagonal local-block entries to one force component. +/// +/// Sums dm(ir, ic) over local block pairs whose global orbital indices +/// coincide, attributing each entry to the atom owning the orbital along +/// Cartesian component dim. +template +void accumulate_diag_force(const Parallel_Orbitals& pv, + const UnitCell& ucell, + const T* dm, + const int dim, + ModuleBase::matrix& force_dftu) +{ + assert(dm != nullptr); + assert(dim >= 0 && dim < 3); + for (int ir = 0; ir < pv.nrow; ir++) + { + const int iwt1 = pv.local2global_row(ir); + const int iat1 = ucell.iwt2iat[iwt1]; + for (int ic = 0; ic < pv.ncol; ic++) + { + if (pv.local2global_col(ic) == iwt1) + { + force_dftu(iat1, dim) += std::real(dm[ic * pv.nrow + ir]); + } + } + } +} + +/// @brief Add the real part of diagonal local-block entries to one stress pair. +template +void accumulate_diag_stress(const Parallel_Orbitals& pv, + const T* dm, + const int dim1, + const int dim2, + const double factor, + ModuleBase::matrix& stress_dftu) +{ + assert(dm != nullptr); + assert(dim1 >= 0 && dim1 < 3); + assert(dim2 >= 0 && dim2 < 3); + for (int ir = 0; ir < pv.nrow; ir++) + { + const int iwt1 = pv.local2global_row(ir); + for (int ic = 0; ic < pv.ncol; ic++) + { + if (pv.local2global_col(ic) == iwt1) + { + stress_dftu(dim1, dim2) += factor * std::real(dm[ic * pv.nrow + ir]); + } + } + } +} + +} // namespace DFTU_LCAO + +#endif // DFTU_NAO_FS_ACCUM_H diff --git a/source/source_lcao/module_dftu/dftu_nao_fs_k.cpp b/source/source_lcao/module_dftu/dftu_nao_fs_k.cpp index edb3670a1fe..b16db8c0977 100644 --- a/source/source_lcao/module_dftu/dftu_nao_fs_k.cpp +++ b/source/source_lcao/module_dftu/dftu_nao_fs_k.cpp @@ -1,5 +1,6 @@ #include "dftu_nao_fs_k.h" #include "dftu_nao_folding.h" +#include "dftu_nao_fs_accum.h" #include "source_pw/module_pwdft/dftu_base.h" #include "dftu_nao_pots.h" #include "source_lcao/force_stress_arrays.h" @@ -18,59 +19,6 @@ namespace DFTU_LCAO { namespace { -/// @brief Add the real part of diagonal local-block entries to one force component. -/// -/// Sums dm(ir, ic) over local block pairs whose global orbital indices -/// coincide, attributing each entry to the atom owning the orbital along -/// Cartesian component dim. -template -void accumulate_diag_force(const Parallel_Orbitals& pv, - const UnitCell& ucell, - const T* dm, - const int dim, - ModuleBase::matrix& force_dftu) -{ - assert(dm != nullptr); - assert(dim >= 0 && dim < 3); - for (int ir = 0; ir < pv.nrow; ir++) - { - const int iwt1 = pv.local2global_row(ir); - const int iat1 = ucell.iwt2iat[iwt1]; - for (int ic = 0; ic < pv.ncol; ic++) - { - if (pv.local2global_col(ic) == iwt1) - { - force_dftu(iat1, dim) += std::real(dm[ic * pv.nrow + ir]); - } - } - } -} - -/// @brief Add the real part of diagonal local-block entries to one stress pair. -template -void accumulate_diag_stress(const Parallel_Orbitals& pv, - const T* dm, - const int dim1, - const int dim2, - const double factor, - ModuleBase::matrix& stress_dftu) -{ - assert(dm != nullptr); - assert(dim1 >= 0 && dim1 < 3); - assert(dim2 >= 0 && dim2 < 3); - for (int ir = 0; ir < pv.nrow; ir++) - { - const int iwt1 = pv.local2global_row(ir); - for (int ic = 0; ic < pv.ncol; ic++) - { - if (pv.local2global_col(ic) == iwt1) - { - stress_dftu(dim1, dim2) += factor * std::real(dm[ic * pv.nrow + ir]); - } - } - } -} - /// @brief Whether atom type it carries a usable correlated channel. /// /// Mirrors the silent skips of the original nest: no channel configured @@ -106,8 +54,6 @@ void accumulate_onsite_force(Plus_U_Base& dftu, assert(dim >= 0 && dim < 3); assert(npol == 1 || npol == 2); const std::vector& l_channel = dftu.get_l_channel_vec(); - const std::vector>>>>& iatlnmipol2iwt - = dftu.occmat().iatlnmipol2iwt(); for (int it = 0; it < ucell.ntype; it++) { if (!has_valid_correlated_channel(ucell, l_channel, it)) @@ -122,7 +68,7 @@ void accumulate_onsite_force(Plus_U_Base& dftu, { for (int ipol = 0; ipol < npol; ipol++) { - const int iwt = iatlnmipol2iwt[iat][lc][0][m][ipol]; + const int iwt = dftu.occmat().corr_iwt(iat, lc, m, ipol); const int mu = pv.global2local_row(iwt); const int nu = pv.global2local_col(iwt); if (mu < 0 || nu < 0) @@ -174,6 +120,12 @@ void cal_force_k(const DftuFsEnv& env, { DFTU_LCAO::folding_matrix_k(fold_ctx, fsr, ik, dim + 1, 0, &dSm_k[0], kvec_d); + // DFT+U force at k-point: F_dim = Tr[dS_k/dR_dim * (DM_k * V_onsite)] + // rho_pot_onsite = DM_k * V_onsite is a non-symmetric complex matrix. + // Two different contractions are needed: + // diag contribution: dS/dR * rho^C (conjugate transpose) + // onsite contribution: dS/dR * rho^N (no transpose) + #ifdef __MPI ScalapackConnector::gemm(transN, transC, @@ -302,7 +254,7 @@ void cal_force_gamma(const DftuFsEnv& env, const UnitCell& ucell = env.ucell(); const int npol = env.npol(); const int nlocal = pv.get_global_row_size(); - double* const dsloc[3] = {env.fsr().DSloc_x, env.fsr().DSloc_y, env.fsr().DSloc_z}; + double* const dsloc[3] = {env.fsr().DSloc_x.data(), env.fsr().DSloc_y.data(), env.fsr().DSloc_z.data()}; const char transN = 'N'; const char transT = 'T'; @@ -316,6 +268,15 @@ void cal_force_gamma(const DftuFsEnv& env, { double* tmp_ptr = dsloc[dim]; + // DFT+U force: F_dim = Tr[dS/dR_dim * (DM * V_onsite)] + // where rho_pot_onsite = DM * V_onsite is generally NOT symmetric + // (product of two symmetric matrices). Two different contractions + // are needed: + // diag contribution: dS/dR * rho^T (row-indexed trace) + // onsite contribution: dS/dR * rho^N (no transpose) + // This mirrors cal_force_k, which uses rho^C and rho^N for the + // complex case. + #ifdef __MPI ScalapackConnector::gemm(transN, transT, @@ -342,7 +303,7 @@ void cal_force_gamma(const DftuFsEnv& env, #ifdef __MPI ScalapackConnector::gemm(transN, - transT, + transN, nlocal, nlocal, nlocal, @@ -384,10 +345,10 @@ void cal_stress_gamma(const DftuFsEnv& env, const std::string& ks_solver = env.ks_solver(); const std::vector& orb_cutoff = env.orb_cutoff(); const int nlocal = pv.get_global_row_size(); - double* dsloc_x = fsr.DSloc_x; - double* dsloc_y = fsr.DSloc_y; - double* dsloc_z = fsr.DSloc_z; - double* dh_r = fsr.DH_r; + double* dsloc_x = fsr.DSloc_x.data(); + double* dsloc_y = fsr.DSloc_y.data(); + double* dsloc_z = fsr.DSloc_z.data(); + double* dh_r = fsr.DH_r.data(); // shared folding context: read-only params bundled for fold_dSR_gamma DFTU_LCAO::FoldingCtx fold_ctx{npol, ks_solver, orb_cutoff, &ucell, &pv, &gd}; @@ -447,27 +408,26 @@ void check_folded_arrays(const ForceStressArrays& fsr, const bool cal_stress, const bool gamma_only_local) { - const double* ds0 = gamma_only_local ? fsr.DSloc_x : fsr.DSloc_Rx; - const double* ds1 = gamma_only_local ? fsr.DSloc_y : fsr.DSloc_Ry; - const double* ds2 = gamma_only_local ? fsr.DSloc_z : fsr.DSloc_Rz; - const bool missing_ds = ds0 == nullptr || ds1 == nullptr || ds2 == nullptr; + const bool missing_ds = gamma_only_local + ? (fsr.DSloc_x.empty() || fsr.DSloc_y.empty() || fsr.DSloc_z.empty()) + : (fsr.DSloc_Rx.empty() || fsr.DSloc_Ry.empty() || fsr.DSloc_Rz.empty()); if (cal_force && missing_ds) { const char* message = gamma_only_local - ? "fsr.DSloc_x/y/z are nullptr in gamma_only path; the caller must allocate and fill them. " + ? "fsr.DSloc_x/y/z are empty in gamma_only path; the caller must allocate and fill them. " "See notes in source/source_lcao/force_stress_lcao.cpp." - : "fsr.DSloc_Rx/Ry/Rz are nullptr in multik path; the caller must allocate and fill them. " + : "fsr.DSloc_Rx/Ry/Rz are empty in multik path; the caller must allocate and fill them. " "See notes in source/source_lcao/force_stress_lcao.cpp."; ModuleBase::WARNING_QUIT("DFTU_LCAO::force_stress", message); } - if (cal_stress && (missing_ds || fsr.DH_r == nullptr)) + if (cal_stress && (missing_ds || fsr.DH_r.empty())) { const char* message = gamma_only_local - ? "fsr.DSloc_x/y/z or fsr.DH_r is nullptr in gamma_only path; " + ? "fsr.DSloc_x/y/z or fsr.DH_r is empty in gamma_only path; " "the caller must allocate and fill them. " "See notes in source/source_lcao/force_stress_lcao.cpp." - : "fsr.DSloc_Rx/Ry/Rz or fsr.DH_r is nullptr in multik path; " + : "fsr.DSloc_Rx/Ry/Rz or fsr.DH_r is empty in multik path; " "the caller must allocate and fill them. " "See notes in source/source_lcao/force_stress_lcao.cpp."; ModuleBase::WARNING_QUIT("DFTU_LCAO::force_stress", message); @@ -508,7 +468,6 @@ void run_gamma_loop(const DftuFsEnv& env, Plus_U_Base& dftu = env.dftu(); const UnitCell& ucell = env.ucell(); const Parallel_Orbitals& pv = env.pv(); - const int npol = env.npol(); const int nlocal = pv.get_global_row_size(); const char transN = 'N'; @@ -557,7 +516,6 @@ void run_k_loop(const DftuFsEnv& env, Plus_U_Base& dftu = env.dftu(); const UnitCell& ucell = env.ucell(); const Parallel_Orbitals& pv = env.pv(); - const int npol = env.npol(); const int nlocal = pv.get_global_row_size(); const char transN = 'N'; diff --git a/source/source_lcao/module_dftu/dftu_nao_fs_k.h b/source/source_lcao/module_dftu/dftu_nao_fs_k.h index ab2224275ef..c84d7ba1d8b 100644 --- a/source/source_lcao/module_dftu/dftu_nao_fs_k.h +++ b/source/source_lcao/module_dftu/dftu_nao_fs_k.h @@ -99,6 +99,11 @@ class DftuFsEnv }; /// @brief Top-level entry: drives force/stress from DFT+U. +/// +/// @note Hard to unit-test: requires DftuFsEnv which bundles Grid_Driver +/// (neighbor search), Parallel_Orbitals (BLACS), ForceStressArrays (folded +/// matrices), and TwoCenterIntegrator. The inner functions cal_force_k/gamma +/// and cal_stress_k/gamma are testable with a mock DftuFsEnv. void force_stress(const DftuFsEnv& env, const bool cal_force, const bool cal_stress, diff --git a/source/source_lcao/module_dftu/dftu_nao_fs_r.cpp b/source/source_lcao/module_dftu/dftu_nao_fs_r.cpp index 5c694b4f209..9601ce37293 100644 --- a/source/source_lcao/module_dftu/dftu_nao_fs_r.cpp +++ b/source/source_lcao/module_dftu/dftu_nao_fs_r.cpp @@ -19,6 +19,150 @@ namespace DFTU_LCAO { +namespace +{ + +/// @brief build the two-center integral table for one Hubbard +/// atom iat0. Returns nlm_tot indexed by adjacent atom slot. +std::vector>> build_nlm( + const UnitCell* ucell, + const TwoCenterIntegrator* intor, + const Parallel_Orbitals* pv, + const int iat0, + const int T0, + const int target_L, + const AdjacentAtomInfo& adjs) +{ + const int tlp1 = 2 * target_L + 1; + const int npol = ucell->get_npol(); + const ModuleBase::Vector3 tau0 = ucell->get_tau(iat0); + std::vector>> nlm_tot; + nlm_tot.resize(adjs.adj_num + 1); + + for (int ad = 0; ad < adjs.adj_num + 1; ++ad) + { + const int T1 = adjs.ntype[ad]; + const int I1 = adjs.natom[ad]; + const int iat1 = ucell->itia2iat(T1, I1); + const ModuleBase::Vector3& tau1 = adjs.adjacent_tau[ad]; + const Atom* atom1 = &ucell->atoms[T1]; + + std::vector all_indexes = pv->get_indexes_row(iat1); + std::vector col_indexes = pv->get_indexes_col(iat1); + // insert col_indexes into all_indexes to get universal set with no repeat elements + all_indexes.insert(all_indexes.end(), col_indexes.begin(), col_indexes.end()); + std::sort(all_indexes.begin(), all_indexes.end()); + all_indexes.erase(std::unique(all_indexes.begin(), all_indexes.end()), all_indexes.end()); + for (int iw1l = 0; iw1l < all_indexes.size(); iw1l += npol) + { + const int iw1 = all_indexes[iw1l] / npol; + std::vector> nlm; + // nlm is a vector of vectors, but size of outer vector is only 1 here + // If we are calculating force, we need also to store the gradient + // and size of outer vector is then 4 + // inner loop : all projectors (L0,M0) + int L1 = atom1->iw2l[iw1]; + int N1 = atom1->iw2n[iw1]; + int m1 = atom1->iw2m[iw1]; + + // convert m (0,1,...2l) to M (-l, -l+1, ..., l-1, l) + int M1 = (m1 % 2 == 0) ? -m1 / 2 : (m1 + 1) / 2; + + ModuleBase::Vector3 dtau = tau0 - tau1; + intor->snap(T1, L1, N1, M1, T0, dtau * ucell->lat0, + 1 /*cal_deri*/, nlm); + + // select the elements of nlm with target_L + std::vector nlm_target(tlp1 * 4); + const Atom* atom0 = &ucell->atoms[T0]; + for (int iw = 0; iw < atom0->nw; iw++) + { + if (atom0->iw2l[iw] == target_L) + { + for (int n = 0; n < 4; n++) // value, deri_x, deri_y, deri_z + { + std::copy(nlm[n].begin() + iw, nlm[n].begin() + iw + tlp1, + nlm_target.begin() + n * tlp1); + } + break; + } + } + nlm_tot[ad].insert({all_indexes[iw1l], nlm_target}); + } + } + return nlm_tot; +} + +/// @brief accumulate force and stress contributions from all (ad1, ad2) +/// adjacent-atom pairs of one Hubbard atom iat0. +void acc_fs_pairs(const UnitCell* ucell, + const Parallel_Orbitals* pv, + const int iat0, + const int T0, + const AdjacentAtomInfo& adjs, + const std::vector>>& nlm_tot, + const std::vector& pot_onsite, + const std::vector*>& dmR, + const int nspin, + bool cal_force, + bool cal_stress, + ModuleBase::matrix& force_local, + std::vector& stress_local) +{ + const ModuleBase::Vector3 tau0 = ucell->get_tau(iat0); + + for (int ad1 = 0; ad1 < adjs.adj_num + 1; ++ad1) + { + const int T1 = adjs.ntype[ad1]; + const int I1 = adjs.natom[ad1]; + const int iat1 = ucell->itia2iat(T1, I1); + double* force_tmp1 = (cal_force) ? &force_local(iat1, 0) : nullptr; + double* force_tmp2 = (cal_force) ? &force_local(iat0, 0) : nullptr; + const ModuleBase::Vector3& R_index1 = adjs.box[ad1]; + ModuleBase::Vector3 dis1 = adjs.adjacent_tau[ad1] - tau0; + for (int ad2 = 0; ad2 < adjs.adj_num + 1; ++ad2) + { + const int T2 = adjs.ntype[ad2]; + const int I2 = adjs.natom[ad2]; + const int iat2 = ucell->itia2iat(T2, I2); + const ModuleBase::Vector3& R_index2 = adjs.box[ad2]; + ModuleBase::Vector3 dis2 = adjs.adjacent_tau[ad2] - tau0; + ModuleBase::Vector3 R_vector(R_index2[0] - R_index1[0], + R_index2[1] - R_index1[1], + R_index2[2] - R_index1[2]); + std::vector*> tmp(nspin, nullptr); + tmp[0] = dmR[0]->find_matrix(iat1, iat2, R_vector[0], R_vector[1], R_vector[2]); + if (nspin == 2) + { + tmp[1] = dmR[1]->find_matrix(iat1, iat2, R_vector[0], R_vector[1], R_vector[2]); + } + // if not found , skip this pair of atoms + if (tmp[0] != nullptr) + { + // calculate force + if (cal_force) + { + cal_for_IJR_nao_r(iat1, iat2, pv, + nlm_tot[ad1], nlm_tot[ad2], + pot_onsite, tmp.data(), nspin, + force_tmp1, force_tmp2); + } + + // calculate stress + if (cal_stress) + { + cal_str_IJR_nao_r(iat1, iat2, pv, + nlm_tot[ad1], nlm_tot[ad2], + pot_onsite, tmp.data(), nspin, + dis1, dis2, stress_local.data()); + } + } + } + } +} + +} // namespace + void cal_fs_nao_r_impl(const UnitCell* ucell, Plus_U_Base* dftu, const TwoCenterIntegrator* intor, @@ -30,10 +174,9 @@ void cal_fs_nao_r_impl(const UnitCell* ucell, ModuleBase::matrix& force, ModuleBase::matrix& stress) { - ModuleBase::timer::start("DFTU", "cal_fs_nao_r"); + ModuleBase::timer::start("DFTU_LCAO", "cal_fs_nao_r"); const Parallel_Orbitals* pv = dmR[0]->get_paraV(); - const int npol = ucell->get_npol(); std::vector stress_tmp; if (cal_stress) { @@ -89,60 +232,8 @@ void cal_fs_nao_r_impl(const UnitCell* ucell, const int tlp1 = 2 * target_L + 1; const AdjacentAtomInfo& adjs = adjs_all[atom_index_all[iat0]]; - std::vector>> nlm_tot; - nlm_tot.resize(adjs.adj_num + 1); + auto nlm_tot = build_nlm(ucell, intor, pv, iat0, T0, target_L, adjs); - for (int ad = 0; ad < adjs.adj_num + 1; ++ad) - { - const int T1 = adjs.ntype[ad]; - const int I1 = adjs.natom[ad]; - const int iat1 = ucell->itia2iat(T1, I1); - const ModuleBase::Vector3& tau1 = adjs.adjacent_tau[ad]; - const Atom* atom1 = &ucell->atoms[T1]; - - std::vector all_indexes = pv->get_indexes_row(iat1); - std::vector col_indexes = pv->get_indexes_col(iat1); - // insert col_indexes into all_indexes to get universal set with no repeat elements - all_indexes.insert(all_indexes.end(), col_indexes.begin(), col_indexes.end()); - std::sort(all_indexes.begin(), all_indexes.end()); - all_indexes.erase(std::unique(all_indexes.begin(), all_indexes.end()), all_indexes.end()); - for (int iw1l = 0; iw1l < all_indexes.size(); iw1l += npol) - { - const int iw1 = all_indexes[iw1l] / npol; - std::vector> nlm; - // nlm is a vector of vectors, but size of outer vector is only 1 here - // If we are calculating force, we need also to store the gradient - // and size of outer vector is then 4 - // inner loop : all projectors (L0,M0) - int L1 = atom1->iw2l[iw1]; - int N1 = atom1->iw2n[iw1]; - int m1 = atom1->iw2m[iw1]; - - // convert m (0,1,...2l) to M (-l, -l+1, ..., l-1, l) - int M1 = (m1 % 2 == 0) ? -m1 / 2 : (m1 + 1) / 2; - - ModuleBase::Vector3 dtau = tau0 - tau1; - intor->snap(T1, L1, N1, M1, T0, dtau * ucell->lat0, - 1 /*cal_deri*/, nlm); - - // select the elements of nlm with target_L - std::vector nlm_target(tlp1 * 4); - const Atom* atom0 = &ucell->atoms[T0]; - for (int iw = 0; iw < atom0->nw; iw++) - { - if (atom0->iw2l[iw] == target_L) - { - for (int n = 0; n < 4; n++) // value, deri_x, deri_y, deri_z - { - std::copy(nlm[n].begin() + iw, nlm[n].begin() + iw + tlp1, - nlm_target.begin() + n * tlp1); - } - break; - } - } - nlm_tot[ad].insert({all_indexes[iw1l], nlm_target}); - } - } // first iteration to calculate occupation matrix std::vector occ(tlp1 * tlp1 * nspin, 0); dftu->occmat().get_flat(iat0, target_L, occ); @@ -162,54 +253,8 @@ void cal_fs_nao_r_impl(const UnitCell* ucell, // U*(1/2*delta(m, m')-occ(m, m')) // + U*(1/2*delta(m, m')-occ(m, m')) // d/d tau_{J,alpha} * tau_{J,beta} ] for each pair of atoms - for (int ad1 = 0; ad1 < adjs.adj_num + 1; ++ad1) - { - const int T1 = adjs.ntype[ad1]; - const int I1 = adjs.natom[ad1]; - const int iat1 = ucell->itia2iat(T1, I1); - double* force_tmp1 = (cal_force) ? &force_local(iat1, 0) : nullptr; - double* force_tmp2 = (cal_force) ? &force_local(iat0, 0) : nullptr; - const ModuleBase::Vector3& R_index1 = adjs.box[ad1]; - ModuleBase::Vector3 dis1 = adjs.adjacent_tau[ad1] - tau0; - for (int ad2 = 0; ad2 < adjs.adj_num + 1; ++ad2) - { - const int T2 = adjs.ntype[ad2]; - const int I2 = adjs.natom[ad2]; - const int iat2 = ucell->itia2iat(T2, I2); - const ModuleBase::Vector3& R_index2 = adjs.box[ad2]; - ModuleBase::Vector3 dis2 = adjs.adjacent_tau[ad2] - tau0; - ModuleBase::Vector3 R_vector(R_index2[0] - R_index1[0], - R_index2[1] - R_index1[1], - R_index2[2] - R_index1[2]); - std::vector*> tmp(nspin, nullptr); - tmp[0] = dmR[0]->find_matrix(iat1, iat2, R_vector[0], R_vector[1], R_vector[2]); - if (nspin == 2) - { - tmp[1] = dmR[1]->find_matrix(iat1, iat2, R_vector[0], R_vector[1], R_vector[2]); - } - // if not found , skip this pair of atoms - if (tmp[0] != nullptr) - { - // calculate force - if (cal_force) - { - cal_for_IJR_nao_r(iat1, iat2, pv, - nlm_tot[ad1], nlm_tot[ad2], - pot_onsite, tmp.data(), nspin, - force_tmp1, force_tmp2); - } - - // calculate stress - if (cal_stress) - { - cal_str_IJR_nao_r(iat1, iat2, pv, - nlm_tot[ad1], nlm_tot[ad2], - pot_onsite, tmp.data(), nspin, - dis1, dis2, stress_local.data()); - } - } - } - } + acc_fs_pairs(ucell, pv, iat0, T0, adjs, nlm_tot, pot_onsite, + dmR, nspin, cal_force, cal_stress, force_local, stress_local); } #pragma omp critical { @@ -229,34 +274,16 @@ void cal_fs_nao_r_impl(const UnitCell* ucell, if (cal_force) { - Parallel_Reduce::reduce_all(force.c, force.nr * force.nc); - if (nspin != 4) - { - for (int i = 0; i < force.nr * force.nc; i++) - { - force.c[i] *= 2.0; - } - } + reduce_force_impl(force, nspin); } // stress renormalization if (cal_stress) { - Parallel_Reduce::reduce_all(stress_tmp.data(), 6); - const double weight = ucell->lat0 / ucell->omega; - for (int i = 0; i < 6; i++) - { - stress.c[i] = stress_tmp[i] * weight; - } - stress.c[8] = stress.c[5]; // stress(2,2) - stress.c[7] = stress.c[4]; // stress(2,1) - stress.c[6] = stress.c[2]; // stress(2,0) - stress.c[5] = stress.c[4]; // stress(1,2) - stress.c[4] = stress.c[3]; // stress(1,1) - stress.c[3] = stress.c[1]; // stress(1,0) + reduce_stress_impl(ucell, stress_tmp, stress); } - ModuleBase::timer::end("DFTU", "cal_fs_nao_r"); + ModuleBase::timer::end("DFTU_LCAO", "cal_fs_nao_r"); } void cal_fs_nao_r(const UnitCell* ucell, diff --git a/source/source_lcao/module_dftu/dftu_nao_fs_r.h b/source/source_lcao/module_dftu/dftu_nao_fs_r.h index 0cf05494b14..e9cf06ddcfb 100644 --- a/source/source_lcao/module_dftu/dftu_nao_fs_r.h +++ b/source/source_lcao/module_dftu/dftu_nao_fs_r.h @@ -51,7 +51,7 @@ template class OperatorLCAO; template -class DFTU; +class DFTU_onsite; template class HContainer; @@ -60,6 +60,26 @@ class HContainer; namespace DFTU_LCAO { +/** + * @brief Post-process force: MPI Allreduce and spin-degeneracy scaling. + * + * @param force [in,out] force matrix (nat, 3), reduced and scaled + * @param nspin [in] number of spin channels (1, 2, or 4) + */ +void reduce_force_impl(ModuleBase::matrix& force, int nspin); + +/** + * @brief Post-process stress: MPI Allreduce and renormalization to + * full 3x3 tensor from Voigt-like 6-component form. + * + * @param ucell [in] unit cell (for lat0/omega) + * @param stress_tmp [in] Voigt-like 6-component stress, reduced in-place + * @param stress [out] full 3x3 stress tensor + */ +void reduce_stress_impl(const UnitCell* ucell, + const std::vector& stress_tmp, + ModuleBase::matrix& stress); + /** * @brief Non-template core of DFT+U force/stress in real space. * @@ -67,6 +87,10 @@ namespace DFTU_LCAO * and real-space type (TR). The template wrapper cal_fs_nao_r only validates * the density matrix and forwards arguments here. * + * @note Hard to unit-test: requires TwoCenterIntegrator (numerical orbital + * radial integration) to build nlm via build_nlm. The inner function + * acc_fs_pairs is directly testable with a mock nlm_tot; build_nlm is not. + * * @param ucell [in] unit cell * @param dftu [in] DFT+U base object (occupation matrix, U values) * @param intor [in] two-center integrator for and gradients diff --git a/source/source_lcao/module_dftu/dftu_nao_fs_reduce.cpp b/source/source_lcao/module_dftu/dftu_nao_fs_reduce.cpp new file mode 100644 index 00000000000..be70bf72fe8 --- /dev/null +++ b/source/source_lcao/module_dftu/dftu_nao_fs_reduce.cpp @@ -0,0 +1,45 @@ +/// @file dftu_nao_fs_reduce.cpp +/// @brief Post-processing of DFT+U force and stress (MPI reduce + scaling) +/// +/// These functions are separated from dftu_nao_fs_r.cpp to keep the link +/// closure minimal for unit testing. + +#include "dftu_nao_fs_r.h" + +#include "source_base/parallel_reduce.h" +#include "source_cell/unitcell.h" + +namespace DFTU_LCAO +{ + +void reduce_force_impl(ModuleBase::matrix& force, const int nspin) +{ + Parallel_Reduce::reduce_all(force.c, force.nr * force.nc); + if (nspin != 4) + { + for (int i = 0; i < force.nr * force.nc; i++) + { + force.c[i] *= 2.0; + } + } +} + +void reduce_stress_impl(const UnitCell* ucell, + const std::vector& stress_tmp, + ModuleBase::matrix& stress) +{ + Parallel_Reduce::reduce_all(const_cast(stress_tmp.data()), 6); + const double weight = ucell->lat0 / ucell->omega; + for (int i = 0; i < 6; i++) + { + stress.c[i] = stress_tmp[i] * weight; + } + stress.c[8] = stress.c[5]; // stress(2,2) + stress.c[7] = stress.c[4]; // stress(2,1) + stress.c[6] = stress.c[2]; // stress(2,0) + stress.c[5] = stress.c[4]; // stress(1,2) + stress.c[4] = stress.c[3]; // stress(1,1) + stress.c[3] = stress.c[1]; // stress(1,0) +} + +} // namespace DFTU_LCAO diff --git a/source/source_lcao/module_dftu/dftu_nao_ijr.h b/source/source_lcao/module_dftu/dftu_nao_ijr.h index ff8fafe93ce..b76fee3db04 100644 --- a/source/source_lcao/module_dftu/dftu_nao_ijr.h +++ b/source/source_lcao/module_dftu/dftu_nao_ijr.h @@ -47,7 +47,7 @@ inline void load_occ_from_file(const Plus_U_Base& dftu, { for (int i = 0; i < static_cast(occ.size()); i++) { - occ[i] = dftu.occmat().get(iat0, target_L, 0, current_spin, + occ[i] = dftu.occmat().get(iat0, target_L, current_spin, i / (2 * target_L + 1), i % (2 * target_L + 1)); } } @@ -197,6 +197,55 @@ inline void cal_occ_ijr(const int iat1, } } +/** + * @brief walk all (ad1, ad2) adjacent-atom pairs of one Hubbard atom iat0 + * and invoke @p body for each pair that has a matching matrix block. + * + * This is the common iteration skeleton shared by the HR accumulation + * (accumulate_hr_for_iat0), occupation-matrix computation + * (compute_occ_from_dmr), and the force/stress real-space implementation + * (cal_fs_nao_r_impl). The body receives (iat1, iat2, R_vector, nlm1, nlm2) + * and is responsible for the actual physics. + * + * @param ucell [in] unit cell (atom index maps) + * @param iat0 [in] global atom index of the Hubbard atom + * @param adjs [in] adjacent atom info of the Hubbard atom + * @param nlm_tot [in] overlap table for all atoms + * @param body [in] callable with signature + * void(int iat1, int iat2, + * const ModuleBase::Vector3& R_vector, + * const std::unordered_map>& nlm1, + * const std::unordered_map>& nlm2) + */ +template +inline void for_adj_pair(const UnitCell& ucell, + const int iat0, + const AdjacentAtomInfo& adjs, + const NlmTot& nlm_tot, + Body&& body) +{ + for (int ad1 = 0; ad1 < adjs.adj_num + 1; ++ad1) + { + const int T1 = adjs.ntype[ad1]; + const int I1 = adjs.natom[ad1]; + const int iat1 = ucell.itia2iat(T1, I1); + const ModuleBase::Vector3& R_index1 = adjs.box[ad1]; + const std::unordered_map>& nlm1 = nlm_tot[iat0][ad1]; + for (int ad2 = 0; ad2 < adjs.adj_num + 1; ++ad2) + { + const int T2 = adjs.ntype[ad2]; + const int I2 = adjs.natom[ad2]; + const int iat2 = ucell.itia2iat(T2, I2); + const std::unordered_map>& nlm2 = nlm_tot[iat0][ad2]; + const ModuleBase::Vector3& R_index2 = adjs.box[ad2]; + ModuleBase::Vector3 R_vector(R_index2[0] - R_index1[0], + R_index2[1] - R_index1[1], + R_index2[2] - R_index1[2]); + body(iat1, iat2, R_vector, nlm1, nlm2); + } + } +} + /** * @brief accumulate the real-space HR contributions of one Hubbard atom * (iat0) from the precomputed pot_onsite: @@ -222,23 +271,12 @@ void accumulate_hr_for_iat0(const UnitCell& ucell, const Parallel_Orbitals& pv, const std::vector& pot_onsite) { - for (int ad1 = 0; ad1 < adjs.adj_num + 1; ++ad1) - { - const int T1 = adjs.ntype[ad1]; - const int I1 = adjs.natom[ad1]; - const int iat1 = ucell.itia2iat(T1, I1); - const ModuleBase::Vector3& R_index1 = adjs.box[ad1]; - const std::unordered_map>& nlm1 = nlm_tot[iat0][ad1]; - for (int ad2 = 0; ad2 < adjs.adj_num + 1; ++ad2) + for_adj_pair(ucell, iat0, adjs, nlm_tot, + [&](int iat1, int iat2, + const ModuleBase::Vector3& R_vector, + const std::unordered_map>& nlm1, + const std::unordered_map>& nlm2) { - const int T2 = adjs.ntype[ad2]; - const int I2 = adjs.natom[ad2]; - const int iat2 = ucell.itia2iat(T2, I2); - const std::unordered_map>& nlm2 = nlm_tot[iat0][ad2]; - const ModuleBase::Vector3& R_index2 = adjs.box[ad2]; - ModuleBase::Vector3 R_vector(R_index2[0] - R_index1[0], - R_index2[1] - R_index1[1], - R_index2[2] - R_index1[2]); hamilt::BaseMatrix* tmp = hR->find_matrix(iat1, iat2, R_vector[0], R_vector[1], R_vector[2]); if (tmp != nullptr) { @@ -246,18 +284,11 @@ void accumulate_hr_for_iat0(const UnitCell& ucell, #pragma omp critical(dftu_hr_update) #endif { - cal_hr_ijr(iat1, - iat2, - ucell.get_npol(), - pv, - nlm1, - nlm2, - pot_onsite, - tmp->get_pointer()); + cal_hr_ijr(iat1, iat2, ucell.get_npol(), pv, + nlm1, nlm2, pot_onsite, tmp->get_pointer()); } } - } - } + }); } /** @@ -290,38 +321,20 @@ inline void compute_occ_from_dmr(const UnitCell& ucell, const hamilt::HContainer& dmR_current, std::vector& occ) { - for (int ad1 = 0; ad1 < adjs.adj_num + 1; ++ad1) - { - const int T1 = adjs.ntype[ad1]; - const int I1 = adjs.natom[ad1]; - const int iat1 = ucell.itia2iat(T1, I1); - const ModuleBase::Vector3& R_index1 = adjs.box[ad1]; - const std::unordered_map>& nlm1 = nlm_tot[iat0][ad1]; - for (int ad2 = 0; ad2 < adjs.adj_num + 1; ++ad2) + for_adj_pair(ucell, iat0, adjs, nlm_tot, + [&](int iat1, int iat2, + const ModuleBase::Vector3& R_vector, + const std::unordered_map>& nlm1, + const std::unordered_map>& nlm2) { - const int T2 = adjs.ntype[ad2]; - const int I2 = adjs.natom[ad2]; - const int iat2 = ucell.itia2iat(T2, I2); - const std::unordered_map>& nlm2 = nlm_tot[iat0][ad2]; - const ModuleBase::Vector3& R_index2 = adjs.box[ad2]; - ModuleBase::Vector3 R_vector(R_index2[0] - R_index1[0], - R_index2[1] - R_index1[1], - R_index2[2] - R_index1[2]); const hamilt::BaseMatrix* tmp = dmR_current.find_matrix(iat1, iat2, R_vector[0], R_vector[1], R_vector[2]); if (tmp != nullptr) { - cal_occ_ijr(iat1, - iat2, - ucell.get_npol(), - pv, - nlm1, - nlm2, - tmp->get_pointer(), - occ); + cal_occ_ijr(iat1, iat2, ucell.get_npol(), pv, + nlm1, nlm2, tmp->get_pointer(), occ); } - } - } + }); Parallel_Reduce::reduce_all(occ.data(), occ.size()); if (nspin == 1) { diff --git a/source/source_lcao/module_dftu/dftu_nao_occ.cpp b/source/source_lcao/module_dftu/dftu_nao_occ.cpp index f0702f67c70..e535ed59ec2 100644 --- a/source/source_lcao/module_dftu/dftu_nao_occ.cpp +++ b/source/source_lcao/module_dftu/dftu_nao_occ.cpp @@ -6,188 +6,52 @@ #include "source_base/parallel_reduce.h" #include "source_estate/occ_matrix.h" #include "source_lcao/hamilt_lcao.h" +#include "source_cell/module_symmetry/symmetry.h" +#include "source_cell/module_symmetry/symm_rotation_k.h" // cal_occ_mat_k / cal_occ_mat_gamma take Plus_U_Base& dftu directly and read all // occupation-matrix state (occ/save arrays, lookup table, nspin/npol, and the // occmat_ready flag) from dftu.occmat() and the Plus_U_Base accessors. - -void DFTU_LCAO::cal_occ_mat_k(const Parallel_Orbitals* pv, - const UnitCell& ucell, - const std::vector>>& dm_k, - const K_Vectors& kv, - const double& mixing_beta, - hamilt::Hamilt>* p_ham, - const bool gamma_only_local, - Plus_U_Base& dftu, - const std::string& ks_solver) +namespace { - ModuleBase::TITLE("DFTU_LCAO", "cal_occ_mat_k"); - ModuleBase::timer::start("DFTU_LCAO", "cal_occ_mat_k"); - - const int nspin = dftu.occmat().nspin(); - const int nlocal = pv->get_global_row_size(); - const std::vector& l_channel = dftu.get_l_channel_vec(); - - // copy occ_mat to occ_mat_save, then zero occ_mat - dftu.occmat().copy_to_save(ucell, l_channel); - dftu.occmat().zero(ucell, l_channel); - - //=================Part 1====================== - // call SCALAPACK routine to calculate the product of the S and density matrix - const char transN = 'N'; - const char transT = 'T'; - const int one_int = 1; - const std::complex beta(0.0,0.0), alpha(1.0,0.0); - - std::vector> srho(pv->nloc); - - for (int ik = 0; ik < kv.get_nks(); ik++) - { - // srho(mu,nu) = \sum_{iw} S(mu,iw)*dm_k(iw,nu) - DFTU_LCAO::folding_matrix_k_new(ks_solver, gamma_only_local, nspin, ik, p_ham); - - std::complex* s_k_pointer = nullptr; - - if(nspin != 4) - { - s_k_pointer = dynamic_cast, double>*>(p_ham)->getSk(); - } - else - { - s_k_pointer = dynamic_cast, std::complex>*>(p_ham)->getSk(); - } - -#ifdef __MPI - ScalapackConnector::gemm(transN, - transT, - nlocal, - nlocal, - nlocal, - alpha, - s_k_pointer, - one_int, - one_int, - &pv->desc[0], - dm_k[ik].data(), - one_int, - one_int, - &pv->desc[0], - beta, - srho.data(), - one_int, - one_int, - &pv->desc[0]); -#endif - - const int spin = kv.isk[ik]; - // Walk (it, ia, l, n=0) and accumulate each qualifying channel - accumulate_occ_k_for_ik(dftu.occmat(), ucell, *pv, srho.data(), spin, l_channel); - } // ik - - // MPI Allreduce + symmetrize per (iat, l, n=0) channel across all ranks - reduce_and_symmetrize_occ_k(dftu.occmat(), ucell, l_channel); - - if(dftu.has_occ_mixer() && dftu.is_occmat_ready()) - { - dftu.occ_mixer().mix_plain(dftu.occmat(), mixing_beta); - } - - dftu.set_occmat_ready(); - ModuleBase::timer::end("DFTU_LCAO", "cal_occ_mat_k"); - return; -} - -void DFTU_LCAO::cal_occ_mat_gamma(const Parallel_Orbitals* pv, - const UnitCell &ucell, - const std::vector> &dm_gamma, - const double& mixing_beta, - hamilt::Hamilt* p_ham, - Plus_U_Base& dftu) +// (symmetry) lazily-built rotation machinery, shared across SCF iterations +// of one run: process-lifetime static since cal_occ_mat_k has no natural +// per-ion-step owning object to hang this off (unlike the dft_plus_u=1 +// operator path, which owns its own copy). +ModuleSymmetry::Symmetry_rotation_k dftu_occ_symrot; +bool dftu_occ_symrot_built = false; + +// Extract the real part (double -> identity, complex -> .real()). +inline double real_part(const double& x) { return x; } +inline double real_part(const std::complex& x) { return x.real(); } +// Extract the real part of the conjugate (double -> identity, +// complex -> conj(x).real() == x.real()). +inline double real_part_conj(const double& x) { return x; } +inline double real_part_conj(const std::complex& x) { - ModuleBase::TITLE("DFTU_LCAO", "cal_occ_mat_gamma"); - ModuleBase::timer::start("DFTU_LCAO", "cal_occ_mat_gamma"); - - const int nspin = dftu.occmat().nspin(); - const int nlocal = pv->get_global_row_size(); - const std::vector& l_channel = dftu.get_l_channel_vec(); - - // copy occ_mat to occ_mat_save, then zero occ_mat - dftu.occmat().copy_to_save(ucell, l_channel); - dftu.occmat().zero(ucell, l_channel); - - //=================Part 1====================== - // call PBLAS routine to calculate the product of the S and density matrix - char transN = 'N', transT = 'T'; - const int one_int = 1; - const double alpha = 1.0, beta = 0.0; - - std::vector srho(pv->nloc); - for (int is = 0; is < nspin; is++) - { - double* s_gamma_pointer = dynamic_cast*>(p_ham)->getSk(); - -#ifdef __MPI - ScalapackConnector::gemm(transN, - transT, - nlocal, - nlocal, - nlocal, - alpha, - s_gamma_pointer, - one_int, - one_int, - &pv->desc[0], - dm_gamma[is].data(), - //dm_gamma[is].c, - one_int, - one_int, - &pv->desc[0], - beta, - srho.data(), - one_int, - one_int, - &pv->desc[0]); -#endif - - // Per (it, ia, l, n=0, spin) block: accumulate + Allreduce + symmetrize - process_occ_channel_gamma(dftu.occmat(), ucell, *pv, srho.data(), is, l_channel); - } // is - - if(dftu.has_occ_mixer() && dftu.is_occmat_ready()) - { - dftu.occ_mixer().mix_plain(dftu.occmat(), mixing_beta); - } - - dftu.set_occmat_ready(); - ModuleBase::timer::end("DFTU_LCAO", "cal_occ_mat_gamma"); - return; + return std::conj(x).real(); } -namespace DFTU_LCAO { - -/// @brief Accumulate one (iat, l, n, spin) channel of the occupation matrix -/// from the complex S*DM product srho for the multi-k case. Reads npol -/// and the iatlnmipol2iwt lookup directly from occmat so callers do -/// not need to thread those scalars through. -void accumulate_occ_channel_k(OccupationMatrix& occmat, - const Parallel_Orbitals& pv, - const std::complex* srho, - int iat, - int l, - int n, - int spin) +/// @brief Accumulate one (iat, l, spin) channel of the occupation matrix +/// from the S*DM product srho. Reads npol and the corr_iwt lookup +/// directly from occmat so callers do not thread those scalars through. +template +void acc_channel(OccupationMatrix& occmat, + const Parallel_Orbitals& pv, + const T* srho, + int iat, + int l, + int spin) { const int npol = occmat.npol(); - const std::vector>>>>& iatlnmipol2iwt - = occmat.iatlnmipol2iwt(); - ModuleBase::matrix& occ = occmat.mat(iat, l, n, spin); + ModuleBase::matrix& occ = occmat.mat(iat, l, spin); const int two_l_plus_one = 2 * l + 1; for (int m0 = 0; m0 < two_l_plus_one; m0++) { for (int ipol0 = 0; ipol0 < npol; ipol0++) { - const int iwt0 = iatlnmipol2iwt[iat][l][n][m0][ipol0]; + const int iwt0 = occmat.corr_iwt(iat, l, m0, ipol0); const int mu = pv.global2local_row(iwt0); const int mu_prime = pv.global2local_col(iwt0); @@ -195,7 +59,7 @@ void accumulate_occ_channel_k(OccupationMatrix& occmat, { for (int ipol1 = 0; ipol1 < npol; ipol1++) { - const int iwt1 = iatlnmipol2iwt[iat][l][n][m1][ipol1]; + const int iwt1 = occmat.corr_iwt(iat, l, m1, ipol1); const int nu = pv.global2local_col(iwt1); const int nu_prime = pv.global2local_row(iwt1); @@ -207,68 +71,13 @@ void accumulate_occ_channel_k(OccupationMatrix& occmat, if ((nu >= 0) && (mu >= 0)) { - occ(m0_all, m1_all) += (srho[irc]).real() / 4.0; + occ(m0_all, m1_all) += real_part(srho[irc]) / 4.0; } if ((nu_prime >= 0) && (mu_prime >= 0)) { occ(m0_all, m1_all) - += (std::conj(srho[irc_prime])).real() / 4.0; - } - } // ipol1 - } // m1 - } // ipol0 - } // m0 -} - -/// @brief Accumulate one (iat, l, n, spin) channel of the occupation matrix -/// from the real S*DM product srho for the gamma-only case. Reads npol -/// and the iatlnmipol2iwt lookup directly from occmat so callers do -/// not need to thread those scalars through. Uses the combined -/// (m0_all, m1_all) channel index consistently with the multi-k path. -void accumulate_occ_channel_gamma(OccupationMatrix& occmat, - const Parallel_Orbitals& pv, - const double* srho, - int iat, - int l, - int n, - int spin) -{ - const int npol = occmat.npol(); - const std::vector>>>>& iatlnmipol2iwt - = occmat.iatlnmipol2iwt(); - ModuleBase::matrix& occ_is = occmat.mat(iat, l, n, spin); - const int two_l_plus_one = 2 * l + 1; - for (int m0 = 0; m0 < two_l_plus_one; m0++) - { - for (int ipol0 = 0; ipol0 < npol; ipol0++) - { - const int iwt0 = iatlnmipol2iwt[iat][l][n][m0][ipol0]; - const int mu = pv.global2local_row(iwt0); - const int mu_prime = pv.global2local_col(iwt0); - - for (int m1 = 0; m1 < two_l_plus_one; m1++) - { - for (int ipol1 = 0; ipol1 < npol; ipol1++) - { - const int iwt1 = iatlnmipol2iwt[iat][l][n][m1][ipol1]; - const int nu = pv.global2local_col(iwt1); - const int nu_prime = pv.global2local_row(iwt1); - - const int irc = nu * pv.nrow + mu; - const int irc_prime = mu_prime * pv.nrow + nu_prime; - - const int m0_all = m0 + ipol0 * two_l_plus_one; - const int m1_all = m1 + ipol1 * two_l_plus_one; - - if ((nu >= 0) && (mu >= 0)) - { - occ_is(m0_all, m1_all) += srho[irc] / 4.0; - } - - if ((nu_prime >= 0) && (mu_prime >= 0)) - { - occ_is(m0_all, m1_all) += srho[irc_prime] / 4.0; + += real_part_conj(srho[irc_prime]) / 4.0; } } // ipol1 } // m1 @@ -276,17 +85,16 @@ void accumulate_occ_channel_gamma(OccupationMatrix& occmat, } // m0 } -/// @brief MPI Allreduce each (iat, l, n=0) channel of occmat across all ranks -/// and symmetrize it (Hermitian average) per the nspin convention: -/// nspin=1 mirrors spin-0 into spin-1; nspin=2 symmetrizes each spin; -/// nspin=4 symmetrizes the single Pauli block. Reads nspin and npol -/// from occmat so callers do not thread them through. -void reduce_and_symmetrize_occ_k(OccupationMatrix& occmat, - const UnitCell& ucell, - const std::vector& l_channel) +/// @brief Walk the (it, ia, l) atom mesh for one k-point and accumulate +/// each qualifying channel of occmat from the S*DM product srho. +template +void acc_for_ik(OccupationMatrix& occmat, + const UnitCell& ucell, + const Parallel_Orbitals& pv, + const T* srho, + int spin, + const std::vector& l_channel) { - const int nspin = occmat.nspin(); - const int npol = occmat.npol(); for (int it = 0; it < ucell.ntype; it++) { const int NL = ucell.atoms[it].nwl + 1; @@ -308,84 +116,67 @@ void reduce_and_symmetrize_occ_k(OccupationMatrix& occmat, continue; } - const int N = ucell.atoms[it].l_nchi[l]; - - for (int n = 0; n < N; n++) - { - // if(!Yukawa && n!=0) continue; - if (n != 0) - { - continue; - } - // set the local occupation mumber matrix of spin up and down zeros - - if (nspin == 1 || nspin == 4) - { - ModuleBase::matrix& occ0 = occmat.mat(iat, l, n, 0); - // MPI Allreduce across ranks (in-place) - Parallel_Reduce::reduce_all(&occ0(0, 0), - (2 * l + 1) * npol * (2 * l + 1) * npol); - } - else if (nspin == 2) - { - ModuleBase::matrix& occ0 = occmat.mat(iat, l, n, 0); - // MPI Allreduce across ranks (in-place) - Parallel_Reduce::reduce_all(&occ0(0, 0), - (2 * l + 1) * (2 * l + 1)); - - ModuleBase::matrix& occ1 = occmat.mat(iat, l, n, 1); - // MPI Allreduce across ranks (in-place) - Parallel_Reduce::reduce_all(&occ1(0, 0), - (2 * l + 1) * (2 * l + 1)); - } - - switch (nspin) - { - case 1: - { - ModuleBase::matrix& occ0 = occmat.mat(iat, l, n, 0); - occ0 += transpose(occ0); - occ0 *= 0.5; - occmat.mat(iat, l, n, 1) += occ0; - break; - } - - case 2: - for (int is = 0; is < nspin; is++) - { - ModuleBase::matrix& occ_is = occmat.mat(iat, l, n, is); - occ_is += transpose(occ_is); - } - break; - - case 4: - { - ModuleBase::matrix& occ0 = occmat.mat(iat, l, n, 0); - occ0 += transpose(occ0); - break; - } - - default: - std::cout << "Not supported NSPIN parameter" << std::endl; - exit(0); - } - } // end n + acc_channel(occmat, pv, srho, iat, l, spin); } // end l } // end ia } // end it } -/// @brief Walk the (it, ia, l, n=0) atom mesh for one k-point and accumulate -/// each qualifying channel of occmat from the complex S*DM product -/// srho. Reads npol and the iatlnmipol2iwt lookup from occmat so -/// callers do not thread them through. -void accumulate_occ_k_for_ik(OccupationMatrix& occmat, - const UnitCell& ucell, - const Parallel_Orbitals& pv, - const std::complex* srho, - int spin, - const std::vector& l_channel) +/// @brief accumulate one k-star member's rotated S*DM product into occmat, +/// redistributing the ibz k-point's full weight (already baked into +/// srho_ibz) across all kstar_size members via Symmetry_rotation's +/// built-in 1/kstar_size scaling (see restore_dm/rot_matrix_ao). +void acc_over_kstar(OccupationMatrix& occmat, + const UnitCell& ucell, + const Parallel_Orbitals& pv, + const K_Vectors& kv, + const std::vector>& srho_ibz, + const int ik_ibz, + const int spin, + const int nspin, + const std::vector& l_channel) { + const int nsym = ucell.symm.nrotk; + const size_t kstar_size = kv.kstars[ik_ibz].size(); + std::vector> sigma_y; + for (const std::pair>& isym_kvd : kv.kstars[ik_ibz]) + { + const int isym = isym_kvd.first; + std::vector> srho_rot; + if (isym < nsym) + { // unitary space-group operation (isym==0 is the identity) + srho_rot = dftu_occ_symrot.rot_matrix_ao(srho_ibz, ik_ibz, kstar_size, isym, pv); + } + else + { // antiunitary element: TRS * (spatial operation), see restore_dm + const int isym_M = ucell.symm.magnetic_nspin4 ? isym : (isym - nsym); + if (nspin == 4) + { + if (sigma_y.empty()) { sigma_y = dftu_occ_symrot.set_sigma_y_2d(pv); } + srho_rot = dftu_occ_symrot.trs_spin_rotate( + dftu_occ_symrot.rot_matrix_ao(srho_ibz, ik_ibz, kstar_size, isym_M, pv, false), + sigma_y, pv, 1.0); + } + else + { + srho_rot = dftu_occ_symrot.rot_matrix_ao(srho_ibz, ik_ibz, kstar_size, isym_M, pv, true); + } + } + acc_for_ik(occmat, ucell, pv, srho_rot.data(), spin, l_channel); + } +} + +/// @brief MPI Allreduce each (iat, l) channel of occmat across all ranks +/// and symmetrize it (Hermitian average) per the nspin convention: +/// nspin=1 mirrors spin-0 into spin-1; nspin=2 symmetrizes each spin; +/// nspin=4 symmetrizes the single Pauli block. Reads nspin and npol +/// from occmat so callers do not thread them through. +void reduce_symm(OccupationMatrix& occmat, + const UnitCell& ucell, + const std::vector& l_channel) +{ + const int nspin = occmat.nspin(); + const int npol = occmat.npol(); for (int it = 0; it < ucell.ntype; it++) { const int NL = ucell.atoms[it].nwl + 1; @@ -407,30 +198,66 @@ void accumulate_occ_k_for_ik(OccupationMatrix& occmat, continue; } - const int N = ucell.atoms[it].l_nchi[l]; + if (nspin == 1 || nspin == 4) + { + ModuleBase::matrix& occ0 = occmat.mat(iat, l, 0); + // MPI Allreduce across ranks (in-place) + Parallel_Reduce::reduce_all(&occ0(0, 0), + (2 * l + 1) * npol * (2 * l + 1) * npol); + } + else if (nspin == 2) + { + ModuleBase::matrix& occ0 = occmat.mat(iat, l, 0); + // MPI Allreduce across ranks (in-place) + Parallel_Reduce::reduce_all(&occ0(0, 0), + (2 * l + 1) * (2 * l + 1)); + + ModuleBase::matrix& occ1 = occmat.mat(iat, l, 1); + // MPI Allreduce across ranks (in-place) + Parallel_Reduce::reduce_all(&occ1(0, 0), + (2 * l + 1) * (2 * l + 1)); + } - for (int n = 0; n < N; n++) + switch (nspin) { - // if(!Yukawa && n!=0) continue; - if (n != 0) + case 1: + { + ModuleBase::matrix& occ0 = occmat.mat(iat, l, 0); + occ0 += transpose(occ0); + occ0 *= 0.5; + occmat.mat(iat, l, 1) += occ0; + break; + } + + case 2: + for (int is = 0; is < nspin; is++) { - continue; + ModuleBase::matrix& occ_is = occmat.mat(iat, l, is); + occ_is += transpose(occ_is); } + break; + + case 4: + { + ModuleBase::matrix& occ0 = occmat.mat(iat, l, 0); + occ0 += transpose(occ0); + break; + } - // Calculate the local occupation number matrix - accumulate_occ_channel_k(occmat, pv, srho, iat, l, n, spin); - } // end n + default: + ModuleBase::WARNING_QUIT("DFTU_LCAO", "Not supported NSPIN parameter"); + } } // end l } // end ia } // end it } -/// @brief Process one (it, ia, l, n=0, spin) block of the gamma-only +/// @brief Process one (it, ia, l, spin) block of the gamma-only /// occupation matrix: accumulate from the real S*DM product srho, /// MPI-Allreduce across ranks, then symmetrize per the nspin /// convention. Reads nspin and npol from occmat so callers do not /// thread them through. -void process_occ_channel_gamma(OccupationMatrix& occmat, +void acc_channel_gamma(OccupationMatrix& occmat, const UnitCell& ucell, const Parallel_Orbitals& pv, const double* srho, @@ -459,80 +286,276 @@ void process_occ_channel_gamma(OccupationMatrix& occmat, continue; } - const int N = ucell.atoms[it].l_nchi[l]; + // Calculate the local occupation number matrix + acc_channel(occmat, pv, srho, iat, l, spin); + ModuleBase::matrix& occ_is = occmat.mat(iat, l, spin); + + // MPI Allreduce across ranks (in-place) + Parallel_Reduce::reduce_all(&occ_is(0, 0), + (2 * l + 1) * npol * (2 * l + 1) * npol); - for (int n = 0; n < N; n++) + // for the case spin independent calculation + switch (nspin) { - if (n != 0) - { - continue; - } + case 1: + { + ModuleBase::matrix& occ0 = occmat.mat(iat, l, 0); + occ0 += transpose(occ0); + occ0 *= 0.5; + occmat.mat(iat, l, 1) += occ0; + break; + } - // Calculate the local occupation number matrix - accumulate_occ_channel_gamma(occmat, pv, srho, iat, l, n, spin); - ModuleBase::matrix& occ_is = occmat.mat(iat, l, n, spin); + case 2: + occ_is += transpose(occ_is); + break; - // MPI Allreduce across ranks (in-place) - Parallel_Reduce::reduce_all(&occ_is(0, 0), - (2 * l + 1) * npol * (2 * l + 1) * npol); + default: + ModuleBase::WARNING_QUIT("DFTU_LCAO", "Not supported NSPIN parameter"); + } + } // L + } // ia + } // it +} - // for the case spin independent calculation - switch (nspin) - { - case 1: - { - ModuleBase::matrix& occ0 = occmat.mat(iat, l, n, 0); - occ0 += transpose(occ0); - occ0 *= 0.5; - occmat.mat(iat, l, n, 1) += occ0; - break; - } +} // namespace - case 2: - occ_is += transpose(occ_is); - break; +namespace DFTU_LCAO { - default: - std::cout << "Not supported NSPIN parameter" << std::endl; - exit(0); - } +void cal_occ_mat_k(const Parallel_Orbitals* pv, + const UnitCell& ucell, + const std::vector>>& dm_k, + const K_Vectors& kv, + const double& mixing_beta, + hamilt::Hamilt>* p_ham, + const bool gamma_only_local, + Plus_U_Base& dftu, + const std::string& ks_solver) +{ + ModuleBase::TITLE("DFTU_LCAO", "cal_occ_mat_k"); + ModuleBase::timer::start("DFTU_LCAO", "cal_occ_mat_k"); - } // end for(n) - } // L - } // ia - } // it + const int nspin = dftu.occmat().nspin(); + const int nlocal = pv->get_global_row_size(); + const std::vector& l_channel = dftu.get_l_channel_vec(); + + // copy occ_mat to occ_mat_save, then zero occ_mat + dftu.occmat().copy_to_save(ucell, l_channel); + dftu.occmat().zero(ucell, l_channel); + + //=================Part 1====================== + // call SCALAPACK routine to calculate the product of the S and density matrix + const char transN = 'N'; + const char transT = 'T'; + const int one_int = 1; + const std::complex beta(0.0,0.0), alpha(1.0,0.0); + + std::vector> srho(pv->nloc); + + // (symmetry) when crystal symmetry reduces the k-mesh, each ik below is only + // the irreducible representative; build the AO rotation machinery once so + // its k-star can be correctly re-expanded (see accumulate_occ_over_kstar). + // Symmetry is analyzed once at the beginning and preserved by symmetrization. + // Accordingly, symrot_, dftu_occ_symrot, and the cached Ms_ remain valid and + // do not need to be rebuilt each ionic step. + const bool dftu_spacegroup_symmetry = (ModuleSymmetry::Symmetry::symm_flag == 1) && !kv.kstars.empty(); + if (dftu_spacegroup_symmetry && !dftu_occ_symrot_built) + { + const std::array period{ kv.nmp[0], kv.nmp[1], kv.nmp[2] }; + dftu_occ_symrot.find_irreducible_sector(ucell.symm, ucell.atoms, ucell.st, + ModuleSymmetry::Symmetry_rotation_k::get_bvk_cells(period), period, ucell.lat); + dftu_occ_symrot.cal_Ms(kv, ucell, *pv, nspin); + dftu_occ_symrot_built = true; + } + + for (int ik = 0; ik < kv.get_nks(); ik++) + { + // srho(mu,nu) = \sum_{iw} S(mu,iw)*dm_k(iw,nu) + DFTU_LCAO::folding_matrix_k_new(ks_solver, gamma_only_local, nspin, ik, p_ham); + + std::complex* s_k_pointer = nullptr; + + if(nspin != 4) + { + s_k_pointer = dynamic_cast, double>*>(p_ham)->getSk(); + } + else + { + s_k_pointer = dynamic_cast, std::complex>*>(p_ham)->getSk(); + } + +#ifdef __MPI + ScalapackConnector::gemm(transN, + transT, + nlocal, + nlocal, + nlocal, + alpha, + s_k_pointer, + one_int, + one_int, + &pv->desc[0], + dm_k[ik].data(), + one_int, + one_int, + &pv->desc[0], + beta, + srho.data(), + one_int, + one_int, + &pv->desc[0]); +#endif + + const int spin = kv.isk[ik]; + // Walk (it, ia, l, n=0) and accumulate each qualifying channel + if (dftu_spacegroup_symmetry) + { + // kv.kstars/Ms_ are sized per spin and indexed by GLOBAL ibz position + // (kv.kstars.size() == nks_ibz); ik is local to this k-point pool, so + // map it to the global k index first (kv.ik2iktot), then wrap into the + // per-spin ibz range (mirrors RI_2D_Comm::split_m2D_ktoR_k's + // "ik % ik_list.size()", but on the global index rather than the local one). + const int ik_ibz = kv.ik2iktot[ik] % static_cast(kv.kstars.size()); + acc_over_kstar(dftu.occmat(), ucell, *pv, kv, srho, ik_ibz, spin, nspin, l_channel); + } + else + { + acc_for_ik(dftu.occmat(), ucell, *pv, srho.data(), spin, l_channel); + } + } // ik + + // MPI Allreduce + symmetrize per (iat, l, n=0) channel across all ranks + reduce_symm(dftu.occmat(), ucell, l_channel); + + if(dftu.has_occ_mixer() && dftu.is_occmat_ready()) + { + dftu.occ_mixer().mix_plain(dftu.occmat(), mixing_beta); + } + + dftu.set_occmat_ready(); + ModuleBase::timer::end("DFTU_LCAO", "cal_occ_mat_k"); + return; } -//! dftu occupation matrix for gamma only using dm(double) +void cal_occ_mat_gamma(const Parallel_Orbitals* pv, + const UnitCell &ucell, + const std::vector> &dm_gamma, + const double& mixing_beta, + hamilt::Hamilt* p_ham, + Plus_U_Base& dftu) +{ + ModuleBase::TITLE("DFTU_LCAO", "cal_occ_mat_gamma"); + ModuleBase::timer::start("DFTU_LCAO", "cal_occ_mat_gamma"); + + const int nspin = dftu.occmat().nspin(); + const int nlocal = pv->get_global_row_size(); + const std::vector& l_channel = dftu.get_l_channel_vec(); + + // copy occ_mat to occ_mat_save, then zero occ_mat + dftu.occmat().copy_to_save(ucell, l_channel); + dftu.occmat().zero(ucell, l_channel); + + //=================Part 1====================== + // call PBLAS routine to calculate the product of the S and density matrix + char transN = 'N'; + char transT = 'T'; + const int one_int = 1; + const double alpha = 1.0; + const double beta = 0.0; + + std::vector srho(pv->nloc); + for (int is = 0; is < nspin; is++) + { + double* s_gamma_pointer = dynamic_cast*>(p_ham)->getSk(); + +#ifdef __MPI + ScalapackConnector::gemm(transN, + transT, + nlocal, + nlocal, + nlocal, + alpha, + s_gamma_pointer, + one_int, + one_int, + &pv->desc[0], + dm_gamma[is].data(), + one_int, + one_int, + &pv->desc[0], + beta, + srho.data(), + one_int, + one_int, + &pv->desc[0]); +#endif + + // Per (it, ia, l, n=0, spin) block: accumulate + Allreduce + symmetrize + acc_channel_gamma(dftu.occmat(), ucell, *pv, srho.data(), is, l_channel); + } // is + + if(dftu.has_occ_mixer() && dftu.is_occmat_ready()) + { + dftu.occ_mixer().mix_plain(dftu.occmat(), mixing_beta); + } + + dftu.set_occmat_ready(); + ModuleBase::timer::end("DFTU_LCAO", "cal_occ_mat_gamma"); + return; +} + +} // namespace DFTU_LCAO + +//! dftu occupation matrix for multi-k case using dm(double) template <> -void cal_occ_mat(const Parallel_Orbitals* pv, - const UnitCell& ucell, - const std::vector>& dm, - const K_Vectors& kv, - const double& mixing_beta, - hamilt::Hamilt* p_ham, - Plus_U_Base& dftu, - const bool gamma_only_local, - const int nspin, - const std::string& ks_solver) +void DFTU_LCAO::cal_occ_mat(const Parallel_Orbitals* pv, + const UnitCell& ucell, + const std::vector>>& dm, + const K_Vectors& kv, + const double& mixing_beta, + hamilt::Hamilt>* p_ham, + Plus_U_Base& dftu, + const bool gamma_only_local, + const int nspin, + const std::string& ks_solver) { - DFTU_LCAO::cal_occ_mat_gamma(pv, ucell, dm, mixing_beta, p_ham, dftu); + cal_occ_mat_k(pv, ucell, dm, kv, mixing_beta, p_ham, gamma_only_local, dftu, ks_solver); } -//! dftu occupation matrix for multiple k-points using dm(complex) +//! dftu occupation matrix for gamma only using dm(double) template <> -void cal_occ_mat(const Parallel_Orbitals* pv, - const UnitCell& ucell, - const std::vector>>& dm, - const K_Vectors& kv, - const double& mixing_beta, - hamilt::Hamilt>* p_ham, - Plus_U_Base& dftu, - const bool gamma_only_local, - const int nspin, - const std::string& ks_solver) +void DFTU_LCAO::cal_occ_mat(const Parallel_Orbitals* pv, + const UnitCell& ucell, + const std::vector>& dm, + const K_Vectors& kv, + const double& mixing_beta, + hamilt::Hamilt* p_ham, + Plus_U_Base& dftu, + const bool gamma_only_local, + const int nspin, + const std::string& ks_solver) { - DFTU_LCAO::cal_occ_mat_k(pv, ucell, dm, kv, mixing_beta, p_ham, gamma_only_local, dftu, ks_solver); + cal_occ_mat_gamma(pv, ucell, dm, mixing_beta, p_ham, dftu); } -} // namespace DFTU_LCAO +template void DFTU_LCAO::cal_occ_mat(const Parallel_Orbitals*, + const UnitCell&, + const std::vector>&, + const K_Vectors&, + const double&, + hamilt::Hamilt*, + Plus_U_Base&, + const bool, + const int, + const std::string&); + +template void DFTU_LCAO::cal_occ_mat(const Parallel_Orbitals*, + const UnitCell&, + const std::vector>>&, + const K_Vectors&, + const double&, + hamilt::Hamilt>*, + Plus_U_Base&, + const bool, + const int, + const std::string&); diff --git a/source/source_lcao/module_dftu/dftu_nao_occ.h b/source/source_lcao/module_dftu/dftu_nao_occ.h index fca4ee5ec32..37e90705255 100644 --- a/source/source_lcao/module_dftu/dftu_nao_occ.h +++ b/source/source_lcao/module_dftu/dftu_nao_occ.h @@ -36,56 +36,12 @@ void cal_occ_mat(const Parallel_Orbitals* pv, const int nspin, const std::string& ks_solver); -/// @brief Accumulate one (iat, l, n, spin) channel of the occupation matrix -/// from the complex S*DM product srho for the multi-k case. -void accumulate_occ_channel_k(OccupationMatrix& occmat, - const Parallel_Orbitals& pv, - const std::complex* srho, - int iat, - int l, - int n, - int spin); - -/// @brief Accumulate one (iat, l, n, spin) channel of the occupation matrix -/// from the real S*DM product srho for the gamma-only case. -void accumulate_occ_channel_gamma(OccupationMatrix& occmat, - const Parallel_Orbitals& pv, - const double* srho, - int iat, - int l, - int n, - int spin); - -/// @brief MPI Allreduce each (iat, l, n=0) channel of occmat across all ranks -/// and symmetrize it (Hermitian average) per the nspin convention: -/// nspin=1 mirrors spin-0 into spin-1; nspin=2 symmetrizes each spin; -/// nspin=4 symmetrizes the single Pauli block. -void reduce_and_symmetrize_occ_k(OccupationMatrix& occmat, - const UnitCell& ucell, - const std::vector& l_channel); - -/// @brief Walk the (it, ia, l, n=0) atom mesh for one k-point and accumulate -/// each qualifying channel of occmat from the complex S*DM product -/// srho. -void accumulate_occ_k_for_ik(OccupationMatrix& occmat, - const UnitCell& ucell, - const Parallel_Orbitals& pv, - const std::complex* srho, - int spin, - const std::vector& l_channel); - -/// @brief Process one (it, ia, l, n=0, spin) block of the gamma-only -/// occupation matrix: accumulate from the real S*DM product srho, -/// MPI-Allreduce across ranks, then symmetrize per the nspin -/// convention. -void process_occ_channel_gamma(OccupationMatrix& occmat, - const UnitCell& ucell, - const Parallel_Orbitals& pv, - const double* srho, - int spin, - const std::vector& l_channel); - // calculate the local occupation number matrix (k-point version) +// +// @note Hard to unit-test: requires a full wavefunction (Psi), two-center +// integrator, and PSI-to-2D distribution (p2s_dist) to build srho. +// Consider extracting the srho computation into an injectable interface +// if unit-test coverage is needed. void cal_occ_mat_k(const Parallel_Orbitals* pv, const UnitCell& ucell, const std::vector>>& dm_k, @@ -97,6 +53,10 @@ void cal_occ_mat_k(const Parallel_Orbitals* pv, const std::string& ks_solver); // calculate the local occupation number matrix (gamma-point version) +// +// @note Hard to unit-test: requires a full wavefunction (Psi) and +// two-center integrator to build srho. Consider extracting the srho +// computation into an injectable interface if unit-test coverage is needed. void cal_occ_mat_gamma(const Parallel_Orbitals* pv, const UnitCell& ucell, const std::vector>& dm_gamma, diff --git a/source/source_lcao/module_dftu/dftu_nao_op.cpp b/source/source_lcao/module_dftu/dftu_nao_op.cpp index 25d3ea0fe29..022672f386d 100644 --- a/source/source_lcao/module_dftu/dftu_nao_op.cpp +++ b/source/source_lcao/module_dftu/dftu_nao_op.cpp @@ -9,6 +9,10 @@ #include "source_lcao/module_operator_lcao/operator_lcao.h" #include "source_pw/module_pwdft/dftu_base.h" #include "source_base/parallel_reduce.h" +#include "source_cell/klist.h" +#include "source_cell/module_symmetry/symmetry.h" + +#include #include "dftu_nao_adj.h" #include "dftu_nao_fs_r.h" @@ -16,7 +20,7 @@ #include "dftu_nao_pots.h" template -hamilt::DFTU>::DFTU(HS_Matrix_K* hsk_in, +hamilt::DFTU_onsite>::DFTU_onsite(HS_Matrix_K* hsk_in, const std::vector>& kvec_d_in, hamilt::HContainer* hR_in, const UnitCell& ucell_in, @@ -26,7 +30,7 @@ hamilt::DFTU>::DFTU(HS_Matrix_K* hsk_in, Plus_U_Base* p_dftu, const int nspin_in, const double onsite_radius, - const elecstate::DensityMatrix* dm_in) + const module_dm::DensityMatrix* dm_in) : hamilt::OperatorLCAO(hsk_in, kvec_d_in, hR_in), ucell(&ucell_in), dftu(p_dftu), @@ -35,7 +39,7 @@ hamilt::DFTU>::DFTU(HS_Matrix_K* hsk_in, orb_cutoff_(orb_cutoff), nspin(nspin_in) { - ModuleBase::timer::start("DFTU", "DFTU"); + ModuleBase::timer::start("DFTU_onsite", "DFTU_onsite"); this->cal_type = calculation_type::lcao_dftu; assert(this->ucell != nullptr); @@ -49,7 +53,7 @@ hamilt::DFTU>::DFTU(HS_Matrix_K* hsk_in, const Parallel_Orbitals* pv = this->hR->get_atom_pair(0).get_paraV(); this->nlm_tot = DFTU_LCAO::cal_nlm_all(*this->ucell, *this->dftu, *this->intor_, this->adjs_all, *pv); - ModuleBase::timer::end("DFTU", "DFTU"); + ModuleBase::timer::end("DFTU_onsite", "DFTU_onsite"); } // contributeHR() @@ -63,7 +67,7 @@ hamilt::DFTU>::DFTU(HS_Matrix_K* hsk_in, * * Case 1: Occ_mat NOT ready (!is_occmat_ready) * - First electronic iteration: calculates occupation matrix from density matrix (DMR) - * * Fetches the real-space DMR via dm_->get_DMR_pointer() + * * Fetches the real-space DMR via dm_->get_dmr_ptr() * * Accumulates contributions from all atom pairs via DFTU_LCAO::cal_occ_ijr() * * Performs MPI reduction to sum occ across processes * * Stores result via set_occ_mat_flat() for use in pot_onsite calculation @@ -103,9 +107,9 @@ hamilt::DFTU>::DFTU(HS_Matrix_K* hsk_in, * for better parallel performance instead of critical section. */ template -void hamilt::DFTU>::contributeHR() +void hamilt::DFTU_onsite>::contributeHR() { - ModuleBase::TITLE("DFTU", "contributeHR"); + ModuleBase::TITLE("DFTU_onsite", "contributeHR"); // Early exit: DMR not available (first SCF iteration before the first // diagonalization) AND occ_mat not yet initialized const bool dmr_null = (this->dm_ == nullptr || !this->dm_->is_dmr_ready()); @@ -119,11 +123,51 @@ void hamilt::DFTU>::contributeHR() { this->dftu->set_energy(0.0); } - ModuleBase::timer::start("DFTU", "contributeHR"); + ModuleBase::timer::start("DFTU_onsite", "contributeHR"); const Parallel_Orbitals* pv = this->hR->get_atom_pair(0).get_paraV(); // nlm_tot is precomputed in the constructor (structure snapshot) + // (symmetry) when crystal symmetry reduces the k-mesh, dm_->get_dmr_ptr() + // was Fourier-transformed from the irreducible k-points only and is not + // actually symmetric; reconstruct the full-BZ DMR once here (reused by every + // atom below) via the same D(k) restoration EXX already uses for its own + // real-space density matrix (ModuleSymmetry::Symmetry_rotation::restore_dm). + std::unique_ptr> dmr_sym; + if (!this->dftu->is_occmat_ready() && this->kv_ != nullptr && ModuleSymmetry::Symmetry::symm_flag == 1 + && !this->kv_->kstars.empty()) + { + if (!this->symrot_built_) + { + const std::array period{ this->kv_->nmp[0], this->kv_->nmp[1], this->kv_->nmp[2] }; + // for return_lattice to calculate Ms + this->symrot_.find_irreducible_sector(this->ucell->symm, this->ucell->atoms, this->ucell->st, + ModuleSymmetry::Symmetry_rotation_k::get_bvk_cells(period), period, this->ucell->lat); + this->symrot_.cal_Ms(*this->kv_, *this->ucell, *pv, this->nspin); + this->symrot_built_ = true; + } + const int nspin0 = (this->nspin == 2) ? 2 : 1; + // (k-point pools, KPAR>1) restore_dm() now returns only the stars of THIS pool's own + // local irreducible k-points (see its definition for why that's enough); kvec_d_full + // must be built the same way -- one entry per star member of each local ibz-k, + // enumerated in the same order restore_dm uses for its spin-0 block (kv.ik2iktot maps + // the spin-0 and spin-1 blocks to the same sequence of global ibz indices, so a single + // list built from the spin-0 mapping is valid for the whole nspin0-block DensityMatrix). + const int nk_local = this->kv_->get_nks() / nspin0; + const int nks_ibz_global = static_cast(this->kv_->kstars.size()); + std::vector> kvec_d_full; + for (int ik_local = 0; ik_local < nk_local; ++ik_local) + { + const int ik_ibz = this->kv_->ik2iktot[ik_local] % nks_ibz_global; + for (const std::pair>& isym_kvd : this->kv_->kstars[ik_ibz]) { kvec_d_full.push_back(isym_kvd.second); } + } + const std::vector> dmk_full = this->symrot_.restore_dm(*this->kv_, this->dm_->get_dmk_vec(), *pv); + dmr_sym.reset(new module_dm::DensityMatrix(pv, nspin0, kvec_d_full, static_cast(kvec_d_full.size()))); + dmr_sym->init_dmr(*this->dm_->get_dmr_ptr(1)); + dmr_sym->get_dmk_vec() = dmk_full; + dmr_sym->cal_dmr(-1); + } + // loop over all Hubbard-projector center atoms (iat0) int atom_index = 0; for (int iat0 = 0; iat0 < this->ucell->nat; iat0++) @@ -147,7 +191,8 @@ void hamilt::DFTU>::contributeHR() { // DMR is guaranteed ready here: otherwise the early exit above // would have returned. DMR index is 1-based, hence +1. - const hamilt::HContainer* dmr = this->dm_->get_DMR_pointer(this->current_spin + 1); + const hamilt::HContainer* dmr = this->dm_->get_dmr_ptr(this->current_spin + 1); + if (dmr_sym) { dmr = dmr_sym->get_dmr_ptr(this->current_spin + 1); } DFTU_LCAO::compute_occ_from_dmr(*this->ucell, *this->dftu, iat0, @@ -195,9 +240,9 @@ void hamilt::DFTU>::contributeHR() this->current_spin = 1 - this->current_spin; } - ModuleBase::timer::end("DFTU", "contributeHR"); + ModuleBase::timer::end("DFTU_onsite", "contributeHR"); } -template class hamilt::DFTU>; -template class hamilt::DFTU, double>>; -template class hamilt::DFTU, std::complex>>; +template class hamilt::DFTU_onsite>; +template class hamilt::DFTU_onsite, double>>; +template class hamilt::DFTU_onsite, std::complex>>; diff --git a/source/source_lcao/module_dftu/dftu_nao_op.h b/source/source_lcao/module_dftu/dftu_nao_op.h index e72d2139653..51fbb1dd9bd 100644 --- a/source/source_lcao/module_dftu/dftu_nao_op.h +++ b/source/source_lcao/module_dftu/dftu_nao_op.h @@ -5,32 +5,34 @@ #include "source_lcao/module_operator_lcao/operator_lcao.h" #include "source_hamilt/module_hcontainer/hcontainer.h" #include "source_lcao/module_dftu/dftu_nao_adj.h" +#include "source_cell/module_symmetry/symm_rotation_k.h" // ModuleSymmetry::Symmetry_rotation_k (value member); LibRI-free #include class Plus_U_Base; class TwoCenterIntegrator; class UnitCell; +class K_Vectors; -namespace elecstate +namespace module_dm { template class DensityMatrix; -} // namespace elecstate +} // namespace module_dm namespace hamilt { -/// The DFTU class template inherits from class T -/// it is used to calculate the non-local pseudopotential of wavefunction basis +/// The DFTU_onsite class template inherits from class T +/// it is used to calculate the DFT+U on-site correction of wavefunction basis /// Template parameters: /// - T: base class, it would be OperatorLCAO or OperatorPW template -class DFTU : public T +class DFTU_onsite : public T { }; -/// DFTU class template specialization for OperatorLCAO base class. +/// DFTU_onsite class template specialization for OperatorLCAO base class. /// Adds the DFT+U on-site correction to the real-space Hamiltonian, which is /// then folded to k-space by the OperatorLCAO machinery: /// HR(mu,nu;I,J,R) = pot_onsite(m,m') @@ -40,21 +42,21 @@ class DFTU : public T /// - TK: data type of k-space Hamiltonian /// - TR: data type of real space Hamiltonian template -class DFTU> : public OperatorLCAO +class DFTU_onsite> : public OperatorLCAO { public: - DFTU(HS_Matrix_K* hsk_in, - const std::vector>& kvec_d_in, - hamilt::HContainer* hR_in, - const UnitCell& ucell_in, - const Grid_Driver* gridD_in, - const TwoCenterIntegrator* intor, - const std::vector& orb_cutoff, - Plus_U_Base* p_dftu, - const int nspin_in, - const double onsite_radius, - const elecstate::DensityMatrix* dm_in); - ~DFTU() = default; + DFTU_onsite(HS_Matrix_K* hsk_in, + const std::vector>& kvec_d_in, + hamilt::HContainer* hR_in, + const UnitCell& ucell_in, + const Grid_Driver* gridD_in, + const TwoCenterIntegrator* intor, + const std::vector& orb_cutoff, + Plus_U_Base* p_dftu, + const int nspin_in, + const double onsite_radius, + const module_dm::DensityMatrix* dm_in); + ~DFTU_onsite() = default; /** * @brief contributeHR() calculates the HR matrix @@ -62,13 +64,19 @@ class DFTU> : public OperatorLCAO */ void contributeHR() override; + /// @brief provide the K_Vectors needed to restore the full-BZ density matrix + /// under crystal symmetry (kv.kstars); set once after construction from + /// HamiltLCAO. When left null, contributeHR() falls back to the + /// unsymmetrized DMR (previous behavior). + void set_kv(const K_Vectors* kv_in) { this->kv_ = kv_in; } + private: const UnitCell* ucell = nullptr; Plus_U_Base* dftu = nullptr; /// @brief solver-owned density matrix providing DMR; lifetime covers each ionic step - const elecstate::DensityMatrix* dm_ = nullptr; + const module_dm::DensityMatrix* dm_ = nullptr; const TwoCenterIntegrator* intor_ = nullptr; @@ -83,6 +91,17 @@ class DFTU> : public OperatorLCAO /// @brief cached overlap values; structure snapshot computed /// once in the constructor, reused across SCF iterations of one ionic step DFTU_LCAO::NlmTot nlm_tot; + + /// @brief K_Vectors of the current run; only used (when non-null) to restore + /// the full-BZ density matrix under crystal symmetry before it is read into + /// the DFT+U occupation matrix. Set via set_kv() after construction. + const K_Vectors* kv_ = nullptr; + + /// @brief LibRI-independent k-space rotation machinery (shared base of EXX's own + /// Symmetry_rotation) to reconstruct D(k) at every k-star member from D(k_ibz); + /// built lazily (once) the first time it is needed (rebuilt every ionic step). + ModuleSymmetry::Symmetry_rotation_k symrot_; + bool symrot_built_ = false; }; } // namespace hamilt diff --git a/source/source_lcao/module_dftu/dftu_nao_op_legacy.cpp b/source/source_lcao/module_dftu/dftu_nao_op_legacy.cpp index 4f15b13709d..c2b2df39063 100644 --- a/source/source_lcao/module_dftu/dftu_nao_op_legacy.cpp +++ b/source/source_lcao/module_dftu/dftu_nao_op_legacy.cpp @@ -6,89 +6,41 @@ namespace hamilt { -template class OperatorDFTU>; +template class DFTU_firstzeta>; -template class OperatorDFTU, double>>; +template class DFTU_firstzeta, double>>; -template class OperatorDFTU, std::complex>>; +template class DFTU_firstzeta, std::complex>>; template -void OperatorDFTU>::contributeHR() +void DFTU_firstzeta>::contributeHR() { //no calculation of HR yet for DFTU operator return; } -template<> -void OperatorDFTU>::contributeHk(int ik) +template +void DFTU_firstzeta>::contributeHk(int ik) { - ModuleBase::TITLE("OperatorDFTU", "contributeHk"); + ModuleBase::TITLE("DFTU_firstzeta", "contributeHk"); if (!this->dftu->is_occmat_ready()) { return; } - ModuleBase::timer::start("OperatorDFTU", "contributeHk"); + ModuleBase::timer::start("DFTU_firstzeta", "contributeHk"); // Effective potential of DFT+U is added to total Hamiltonian here; Quxin adds on 20201029 - std::vector pot_uterm(this->hsk->get_pv()->nloc); + std::vector pot_uterm(this->hsk->get_pv()->nloc); DFTU_LCAO::cal_pot_uterm(*this->dftu, *this->ucell, this->hsk->get_pv(), isk[ik], &pot_uterm[0], this->hsk->get_sk()); - double* hk = this->hsk->get_hk(); - - for (int irc = 0; irc < this->hsk->get_pv()->nloc; irc++) - { - hk[irc] += pot_uterm[irc]; - } - - ModuleBase::timer::end("OperatorDFTU", "contributeHk"); -} - -template<> -void OperatorDFTU, double>>::contributeHk(int ik) -{ - ModuleBase::TITLE("OperatorDFTU", "contributeHk"); - if (!this->dftu->is_occmat_ready()) - { - return; - } - ModuleBase::timer::start("OperatorDFTU", "contributeHk"); - - // Effective potential of DFT+U is added to total Hamiltonian here; Quxin adds on 20201029 - std::vector> pot_uterm(this->hsk->get_pv()->nloc); - - DFTU_LCAO::cal_pot_uterm(*this->dftu, *this->ucell, this->hsk->get_pv(), isk[ik], &pot_uterm[0], this->hsk->get_sk()); - - std::complex* hk = this->hsk->get_hk(); - - for (int irc = 0; irc < this->hsk->get_pv()->nloc; irc++) - { - hk[irc] += pot_uterm[irc]; - } - - ModuleBase::timer::end("OperatorDFTU", "contributeHk"); -} - -template<> -void OperatorDFTU, std::complex>>::contributeHk(int ik) -{ - ModuleBase::TITLE("OperatorDFTU", "contributeHk"); - if (!this->dftu->is_occmat_ready()) - { - return; - } - ModuleBase::timer::start("OperatorDFTU", "contributeHk"); - // Effective potential of DFT+U is added to total Hamiltonian here; Quxin adds on 20201029 - std::vector> pot_uterm(this->hsk->get_pv()->nloc); - - DFTU_LCAO::cal_pot_uterm(*this->dftu, *this->ucell, this->hsk->get_pv(), isk[ik], &pot_uterm[0], this->hsk->get_sk()); + TK* hk = this->hsk->get_hk(); - std::complex* hk = this->hsk->get_hk(); for (int irc = 0; irc < this->hsk->get_pv()->nloc; irc++) { hk[irc] += pot_uterm[irc]; } - ModuleBase::timer::end("OperatorDFTU", "contributeHk"); + ModuleBase::timer::end("DFTU_firstzeta", "contributeHk"); } } diff --git a/source/source_lcao/module_dftu/dftu_nao_op_legacy.h b/source/source_lcao/module_dftu/dftu_nao_op_legacy.h index 74ba5d2ccba..ca3413f8989 100644 --- a/source/source_lcao/module_dftu/dftu_nao_op_legacy.h +++ b/source/source_lcao/module_dftu/dftu_nao_op_legacy.h @@ -8,15 +8,15 @@ namespace hamilt { template -class OperatorDFTU : public T +class DFTU_firstzeta : public T { }; template -class OperatorDFTU> : public OperatorLCAO +class DFTU_firstzeta> : public OperatorLCAO { public: - OperatorDFTU>(HS_Matrix_K* hsk_in, + DFTU_firstzeta>(HS_Matrix_K* hsk_in, const std::vector>& kvec_d_in, hamilt::HContainer* hR_in, const UnitCell& ucell_in, diff --git a/source/source_lcao/module_dftu/dftu_nao_pots.cpp b/source/source_lcao/module_dftu/dftu_nao_pots.cpp index e42fcc27ac2..9d7087c4471 100644 --- a/source/source_lcao/module_dftu/dftu_nao_pots.cpp +++ b/source/source_lcao/module_dftu/dftu_nao_pots.cpp @@ -8,6 +8,8 @@ #include "source_basis/module_ao/parallel_orbitals.h" #include "source_cell/unitcell.h" +#include + /// On-site potential and Hubbard energy for one correlated shell: /// pot_onsite(m,m') = U_eff * (0.5 * delta_{m,m'} - occ(m,m')) /// EU = (U_eff / 2) * sum_{m,m'} occ(m,m') * (delta_{m,m'} - occ(m',m)) @@ -62,11 +64,9 @@ void DFTU_LCAO::cal_pot_onsite(const Plus_U_Base& dftu, T* pot_onsite) { ModuleBase::TITLE("DFTU_LCAO", "pot_onsite"); - ModuleBase::GlobalFunc::ZEROS(pot_onsite, pv->nloc); + std::fill(pot_onsite, pot_onsite + pv->nloc, T(0)); const int npol = dftu.occmat().npol(); - const std::vector>>>>& iatlnmipol2iwt - = dftu.occmat().iatlnmipol2iwt(); for (int it = 0; it < ucell.ntype; ++it) { @@ -84,43 +84,35 @@ void DFTU_LCAO::cal_pot_onsite(const Plus_U_Base& dftu, continue; } - for (int n = 0; n < ucell.atoms[it].l_nchi[L]; n++) + for (int m1 = 0; m1 < 2 * L + 1; m1++) { - if (n != 0) - { - continue; - } - - for (int m1 = 0; m1 < 2 * L + 1; m1++) + for (int ipol1 = 0; ipol1 < npol; ipol1++) { - for (int ipol1 = 0; ipol1 < npol; ipol1++) + const int mu = pv->global2local_row(dftu.occmat().corr_iwt(iat, L, m1, ipol1)); + if (mu < 0) { - const int mu = pv->global2local_row(iatlnmipol2iwt[iat][L][n][m1][ipol1]); - if (mu < 0) - { - continue; - } + continue; + } - for (int m2 = 0; m2 < 2 * L + 1; m2++) + for (int m2 = 0; m2 < 2 * L + 1; m2++) + { + for (int ipol2 = 0; ipol2 < npol; ipol2++) { - for (int ipol2 = 0; ipol2 < npol; ipol2++) + const int nu + = pv->global2local_col(dftu.occmat().corr_iwt(iat, L, m2, ipol2)); + if (nu < 0) { - const int nu - = pv->global2local_col(iatlnmipol2iwt[iat][L][n][m2][ipol2]); - if (nu < 0) - { - continue; - } - int m1_all = m1 + (2 * L + 1) * ipol1; - int m2_all = m2 + (2 * L + 1) * ipol2; - double val = get_onsite_pot(dftu, it, iat, L, n, spin, - m1_all, m2_all, new_occ_mat); - pot_onsite[nu * pv->nrow + mu] = static_cast(val); - } // ipol2 - } // m2 - } // ipol1 - } // m1 - } // n + continue; + } + int m1_all = m1 + (2 * L + 1) * ipol1; + int m2_all = m2 + (2 * L + 1) * ipol2; + double val = get_onsite_pot(dftu, it, iat, L, spin, + m1_all, m2_all, new_occ_mat); + pot_onsite[nu * pv->nrow + mu] = static_cast(val); + } // ipol2 + } // m2 + } // ipol1 + } // m1 } // l } // ia } // it @@ -154,12 +146,13 @@ void DFTU_LCAO::cal_pot_uterm(Plus_U_Base& dftu, ModuleBase::timer::start("DFTU_LCAO", "cal_pot_uterm"); const int nlocal = pv->get_global_row_size(); - ModuleBase::GlobalFunc::ZEROS(pot_uterm, pv->nloc); + std::fill(pot_uterm, pot_uterm + pv->nloc, T(0)); //============================================================= // PART2: call pblas to calculate effective potential matrix //============================================================= - const char transN = 'N', transT = 'T'; + const char transN = 'N'; + const char transT = 'T'; const int one_int = 1; const T half = static_cast(0.5); const T one = static_cast(1.0); @@ -213,7 +206,6 @@ double DFTU_LCAO::get_onsite_pot(const Plus_U_Base& dftu, const int T, const int iat, const int L, - const int N, const int spin, const int m0, const int m1, @@ -238,13 +230,13 @@ double DFTU_LCAO::get_onsite_pot(const Plus_U_Base& dftu, { if (m0 == m1) { - pot_onsite = (dftu.yukawa().get_U(T, L, N) - dftu.yukawa().get_J(T, L, N)) - * (0.5 - dftu.occmat().get(iat, L, N, spin, m0, m1)); + pot_onsite = (dftu.yukawa().get_U(T, L) - dftu.yukawa().get_J(T, L)) + * (0.5 - dftu.occmat().get(iat, L, spin, m0, m1)); } else { - pot_onsite = -(dftu.yukawa().get_U(T, L, N) - dftu.yukawa().get_J(T, L, N)) - * dftu.occmat().get(iat, L, N, spin, m0, m1); + pot_onsite = -(dftu.yukawa().get_U(T, L) - dftu.yukawa().get_J(T, L)) + * dftu.occmat().get(iat, L, spin, m0, m1); } } else @@ -252,12 +244,12 @@ double DFTU_LCAO::get_onsite_pot(const Plus_U_Base& dftu, if (m0 == m1) { pot_onsite = dftu.get_u_current(T) - * (0.5 - dftu.occmat().get(iat, L, N, spin, m0, m1)); + * (0.5 - dftu.occmat().get(iat, L, spin, m0, m1)); } else { pot_onsite = -dftu.get_u_current(T) - * dftu.occmat().get(iat, L, N, spin, m0, m1); + * dftu.occmat().get(iat, L, spin, m0, m1); } } } @@ -267,13 +259,13 @@ double DFTU_LCAO::get_onsite_pot(const Plus_U_Base& dftu, { if (m0 == m1) { - pot_onsite = (dftu.yukawa().get_U(T, L, N) - dftu.yukawa().get_J(T, L, N)) - * (0.5 - dftu.occmat().get_save(iat, L, N, spin, m0, m1)); + pot_onsite = (dftu.yukawa().get_U(T, L) - dftu.yukawa().get_J(T, L)) + * (0.5 - dftu.occmat().get_save(iat, L, spin, m0, m1)); } else { - pot_onsite = -(dftu.yukawa().get_U(T, L, N) - dftu.yukawa().get_J(T, L, N)) - * dftu.occmat().get_save(iat, L, N, spin, m0, m1); + pot_onsite = -(dftu.yukawa().get_U(T, L) - dftu.yukawa().get_J(T, L)) + * dftu.occmat().get_save(iat, L, spin, m0, m1); } } else @@ -281,12 +273,12 @@ double DFTU_LCAO::get_onsite_pot(const Plus_U_Base& dftu, if (m0 == m1) { pot_onsite = dftu.get_u_current(T) - * (0.5 - dftu.occmat().get_save(iat, L, N, spin, m0, m1)); + * (0.5 - dftu.occmat().get_save(iat, L, spin, m0, m1)); } else { pot_onsite = -dftu.get_u_current(T) - * dftu.occmat().get_save(iat, L, N, spin, m0, m1); + * dftu.occmat().get_save(iat, L, spin, m0, m1); } } } diff --git a/source/source_lcao/module_dftu/dftu_nao_pots.h b/source/source_lcao/module_dftu/dftu_nao_pots.h index 2c72cfd5741..4f728a5a07c 100644 --- a/source/source_lcao/module_dftu/dftu_nao_pots.h +++ b/source/source_lcao/module_dftu/dftu_nao_pots.h @@ -16,7 +16,7 @@ namespace DFTU_LCAO { * pot_onsite(m,m') = U_eff * (0.5 * delta_{m,m'} - occ(m,m')) * EU = (U_eff / 2) * sum_{m,m'} occ(m,m') * (delta_{m,m'} - occ(m',m)) * - * Non-template core extracted from DFTU>::cal_pot_onsite. + * Non-template core extracted from DFTU_onsite>::cal_pot_onsite. * * @param occ flattened occupation matrix, size m_size*m_size*spin_fold * @param m_size number of magnetic quantum states (2*l+1) @@ -82,7 +82,6 @@ inline void transfer_pot_onsite>(const std::vector& * @param T atom type * @param iat global atom index * @param L angular momentum - * @param N radial index * @param spin spin channel * @param m0 first magnetic quantum index (packed with polarization) * @param m1 second magnetic quantum index (packed with polarization) @@ -93,7 +92,6 @@ double get_onsite_pot(const Plus_U_Base& dftu, const int T, const int iat, const int L, - const int N, const int spin, const int m0, const int m1, diff --git a/source/source_lcao/module_dftu/dftu_nao_str_r.cpp b/source/source_lcao/module_dftu/dftu_nao_str_r.cpp index ac0b0e1a028..f5d09885e55 100644 --- a/source/source_lcao/module_dftu/dftu_nao_str_r.cpp +++ b/source/source_lcao/module_dftu/dftu_nao_str_r.cpp @@ -21,9 +21,7 @@ void cal_str_IJR_nao_r(const int& iat1, const ModuleBase::Vector3& dis2, double* stress) { -#ifdef __DEBUG assert(nspin == 1 || nspin == 2 || nspin == 4); -#endif // npol is the number of spinor polarizations: // 1 for nspin=1 (non-spin-polarized) and nspin=2 (collinear magnetic), // 2 for nspin=4 (non-collinear, one matrix holds both spin-up and spin-down) @@ -59,9 +57,7 @@ void cal_str_IJR_nao_r(const int& iat1, for (int iw2l = 0; iw2l < col_indexes.size(); iw2l += npol) { const std::vector& nlm2 = nlm2_all.find(col_indexes[iw2l])->second; -#ifdef __DEBUG assert(nlm1.size() == nlm2.size()); -#endif for (int m1 = 0; m1 < m_size; m1++) { for (int m2 = 0; m2 < m_size; m2++) diff --git a/source/source_lcao/module_dftu/test/CMakeLists.txt b/source/source_lcao/module_dftu/test/CMakeLists.txt deleted file mode 100644 index 8ca8651135f..00000000000 --- a/source/source_lcao/module_dftu/test/CMakeLists.txt +++ /dev/null @@ -1,50 +0,0 @@ -abacus_disable_feature_definitions(__CUDA) - -AddTest( - TARGET dftu_core_test - LIBS base device - SOURCES dftu_core_test.cpp -) - -AddTest( - TARGET dftu_operator_test - LIBS base device - SOURCES dftu_operator_test.cpp -) - -if(ENABLE_LCAO AND ENABLE_MPI) -AddTest( - TARGET dftu_lcao_test - LIBS parameter psi base device container - SOURCES dftu_lcao_test.cpp ../dftu_nao_op.cpp ../dftu_nao_adj.cpp ../dftu_nao_pots.cpp ../dftu_nao_fs_r.cpp ../dftu_nao_for_r.cpp ../dftu_nao_str_r.cpp - ../../../source_estate/module_dm/density_matrix.cpp - ../../../source_estate/module_dm/density_matrix_io.cpp - ../../../source_pw/module_pwdft/dftu_base.cpp - ../../../source_pw/module_pwdft/dftu_base_io.cpp - ../../../source_pw/module_pwdft/yukawa_screening.cpp - ../../../source_estate/occ_matrix.cpp - ../../../source_estate/occ_mixer.cpp - ../../../source_hamilt/module_hcontainer/func_folding.cpp - ../../../source_hamilt/module_hcontainer/base_matrix.cpp - ../../../source_hamilt/module_hcontainer/hcontainer.cpp - ../../../source_hamilt/module_hcontainer/atom_pair.cpp - ../../../source_basis/module_ao/parallel_orbitals.cpp - ../../../source_basis/module_ao/orb_atomic_lm.cpp - ../../module_operator_lcao/test/tmp_mocks.cpp - ../../../source_hamilt/operator.cpp -) - -AddTest( - TARGET dftu_nao_ijr_test - LIBS parameter psi base device container - SOURCES test_dftu_nao_ijr.cpp - ../../../source_hamilt/module_hcontainer/func_folding.cpp - ../../../source_hamilt/module_hcontainer/base_matrix.cpp - ../../../source_hamilt/module_hcontainer/hcontainer.cpp - ../../../source_hamilt/module_hcontainer/atom_pair.cpp - ../../../source_basis/module_ao/parallel_orbitals.cpp - ../../../source_basis/module_ao/orb_atomic_lm.cpp - ../../module_operator_lcao/test/tmp_mocks.cpp - ../../../source_hamilt/operator.cpp -) -endif() diff --git a/source/source_lcao/module_dftu/test/dftu_core_test.cpp b/source/source_lcao/module_dftu/test/dftu_core_test.cpp deleted file mode 100644 index ffe48262461..00000000000 --- a/source/source_lcao/module_dftu/test/dftu_core_test.cpp +++ /dev/null @@ -1,131 +0,0 @@ -#include "gtest/gtest.h" -#include -#include -#include -#include -#include - -/*********************************************************************** - * Unit tests for DFT+U core algorithms. - * - * These tests target the most complex and bug-prone logic: - * 1. pot_onsite effective potential formula (UForm::dud_fll, FLL) - * 2. Energy correction and double-counting terms - ***********************************************************************/ - -// ===================================================================== -// 1. pot_onsite effective potential formula (UForm::dud_fll, FLL) -// -// pot_onsite[m0,m1] = U * (0.5*delta(m0,m1) - occ_mat[m0,m1]) (diagonal) -// pot_onsite[m0,m1] = -U * occ_mat[m0,m1] (off-diagonal) -// ===================================================================== - -static double compute_pot_onsite(double U_val, int m0, int m1, double occ_mat_val) -{ - if (m0 == m1) - return U_val * (0.5 - occ_mat_val); - else - return -U_val * occ_mat_val; -} - -class PotOnsitePotentialTest : public ::testing::Test -{ - protected: - void SetUp() override {} -}; - -TEST_F(PotOnsitePotentialTest, Diagonal_HalfFilled) -{ - double U = 4.0; - double occ_mat = 0.5; // half-filled - double pot_onsite = compute_pot_onsite(U, 0, 0, occ_mat); - EXPECT_DOUBLE_EQ(pot_onsite, 0.0); // U * (0.5 - 0.5) = 0 -} - -TEST_F(PotOnsitePotentialTest, Diagonal_FullyOccupied) -{ - double U = 4.0; - double occ_mat = 1.0; // fully occupied - double pot_onsite = compute_pot_onsite(U, 0, 0, occ_mat); - EXPECT_DOUBLE_EQ(pot_onsite, -2.0); // U * (0.5 - 1.0) = -2.0 -} - -TEST_F(PotOnsitePotentialTest, OffDiagonal) -{ - double U = 5.0; - double occ_mat = 0.3; - double pot_onsite = compute_pot_onsite(U, 0, 1, occ_mat); - EXPECT_DOUBLE_EQ(pot_onsite, -1.5); // -U * occ_mat = -1.5 -} - -// ===================================================================== -// 4. Energy correction formula -// -// E_U = 0.5 * U * sum_spin [Tr(n) - Tr(n^2)] -// ===================================================================== - -class EnergyCorrectionTest : public ::testing::Test -{ - protected: - static double compute_energy(const std::vector& occ_mat_flat, int m_size, double U) - { - double nm_trace = 0.0, nm2_trace = 0.0; - for (int m0 = 0; m0 < m_size; m0++) - { - nm_trace += occ_mat_flat[m0 * m_size + m0]; - for (int m1 = 0; m1 < m_size; m1++) - nm2_trace += occ_mat_flat[m0 * m_size + m1] * occ_mat_flat[m1 * m_size + m0]; - } - return 0.5 * U * (nm_trace - nm2_trace); - } -}; - -TEST_F(EnergyCorrectionTest, HalfFilled_DOrbital) -{ - const int m_size = 5; - std::vector occ_mat(m_size * m_size, 0.0); - for (int m = 0; m < m_size; m++) - occ_mat[m * m_size + m] = 0.5; - - double energy = compute_energy(occ_mat, m_size, 4.0); - // Tr(n) = 2.5, Tr(n^2) = 1.25, E = 0.5 * 4 * 1.25 = 2.5 - EXPECT_DOUBLE_EQ(energy, 2.5); -} - -TEST_F(EnergyCorrectionTest, OffDiagonal_Contribution) -{ - const int m_size = 2; - std::vector occ_mat = { - 0.3, 0.1, - 0.1, 0.3 - }; - - double energy = compute_energy(occ_mat, m_size, 4.0); - // Tr(n) = 0.6, Tr(n^2) = 0.3^2 + 0.1^2 + 0.1^2 + 0.3^2 = 0.20 - // E = 0.5 * 4 * (0.6 - 0.20) = 0.8 - EXPECT_DOUBLE_EQ(energy, 0.8); -} - -TEST_F(EnergyCorrectionTest, DoubleCounting_Energy) -{ - // E_dc = sum_{m1,m2,spin} pot_onsite[m1,m2] * n[m2,m1] - const int m_size = 3; - double U = 4.0; - std::vector occ_mat = { - 0.5, 0.0, 0.0, - 0.0, 0.3, 0.0, - 0.0, 0.0, 0.2 - }; - - double e_dc = 0.0; - for (int m1 = 0; m1 < m_size; m1++) - for (int m2 = 0; m2 < m_size; m2++) - { - double pot_onsite = (m1 == m2) ? U * (0.5 - occ_mat[m1 * m_size + m2]) - : -U * occ_mat[m1 * m_size + m2]; - e_dc += pot_onsite * occ_mat[m2 * m_size + m1]; - } - - // Only diagonal: m=0: 0*0.5=0, m=1: 0.8*0.3=0.24, m=2: 1.2*0.2=0.24 - EXPECT_NEAR(e_dc, 0.48, 1e-14); -} diff --git a/source/source_lcao/module_dftu/test/dftu_lcao_test.cpp b/source/source_lcao/module_dftu/test/dftu_lcao_test.cpp deleted file mode 100644 index 4fddaee06e0..00000000000 --- a/source/source_lcao/module_dftu/test/dftu_lcao_test.cpp +++ /dev/null @@ -1,299 +0,0 @@ -#include "gtest/gtest.h" -#include - -// mock of DFTU -#include "../dftu_nao_op.h" -#include "source_basis/module_nao/two_center_integrator.h" -#include "source_cell/unitcell.h" -#include "source_pw/module_pwdft/dftu_base.h" -#include "source_estate/module_dm/density_matrix.h" - -Plus_U_Base dftu; -// Static member definitions are in dftu_base.cpp (Plus_U_Base::) - -//--------------------------------------- -// Unit test of Plus_U class -// Plus_U is a derivative class of Operator, it is used to calculate the kinetic matrix -// It use HContainer to store the real space HR matrix -// In this test, we test the correctness and time consuming of 3 functions in Plus_U class -// - initialize_HR() called in constructor -// - contributeHR() -// - contributeHk() -// - HR(double) and SK(complex) are tested in constructHRd2cd -// - HR(double) and SK(double) are tested in constructHRd2d -//--------------------------------------- - -// test_size is the number of atoms in the unitcell -// modify test_size to test different size of unitcell -int test_size = 10; -int test_nw = 10; // please larger than 5 - -class DFTUTest : public ::testing::Test -{ - protected: - void SetUp() override - { -#ifdef __MPI - // MPI parallel settings - MPI_Comm_size(MPI_COMM_WORLD, &dsize); - MPI_Comm_rank(MPI_COMM_WORLD, &my_rank); -#endif - - // set up a unitcell, with one element and test_size atoms, each atom has test_nw orbitals - ucell.ntype = 1; - ucell.nat = test_size; - ucell.atoms = new Atom[ucell.ntype]; - ucell.iat2it = new int[ucell.nat]; - ucell.iat2ia = new int[ucell.nat]; - ucell.atoms[0].tau.resize(ucell.nat); - ucell.lat0 = 1.0; - ucell.itia2iat.create(ucell.ntype, ucell.nat); - for (int iat = 0; iat < ucell.nat; iat++) - { - ucell.iat2it[iat] = 0; - ucell.iat2ia[iat] = iat; - ucell.atoms[0].tau[iat] = ModuleBase::Vector3(0.0, 0.0, 0.0); - ucell.itia2iat(0, iat) = iat; - } - ucell.atoms[0].na = test_size; - ucell.atoms[0].nw = test_nw; - ucell.atoms[0].iw2l.resize(test_nw); - ucell.atoms[0].iw2m.resize(test_nw); - ucell.atoms[0].iw2n.resize(test_nw); - for (int iw = 0; iw < test_nw; ++iw) - { - ucell.atoms[0].iw2l[iw] = 2; - ucell.atoms[0].iw2m[iw] = 0; - ucell.atoms[0].iw2n[iw] = 0; - } - ucell.set_iat2iwt(1); - init_parav(); - // set up a HContainer with ucell - HR = new hamilt::HContainer(ucell, paraV); - - // setting of DFTU - dftu.occmat().data().resize(test_size); - for (int iat = 0; iat < test_size; iat++) - { - dftu.occmat().data()[iat].resize(3); - for (int l = 0; l < 3; l++) - { - dftu.occmat().data()[iat][l].resize(1); - dftu.occmat().data()[iat][l][0].resize(2); - dftu.occmat().data()[iat][l][0][0].create(2 * l + 1, 2 * l + 1); - dftu.occmat().data()[iat][l][0][1].create(2 * l + 1, 2 * l + 1); - } - } - dftu.u_current = {U_test}; - dftu.l_channel = {orbital_c_test}; - } - - void TearDown() override - { - delete HR; - delete paraV; - delete[] ucell.atoms; - } - - // Helper for TEST_F bodies: gtest-derived classes do not inherit - // the friend declaration, so direct dftu.occmat().data()[...] access - // from TestBody would fail to compile. This wrapper runs inside - // DFTUTest, which is a friend of Plus_U_Base. - double occ_mat_c(int iat, int spin, int icc) const - { - return dftu.occmat().data()[iat][2][0][spin].c[icc]; - } - -#ifdef __MPI - void init_parav() - { - int nb = 10; - int global_row = test_size * test_nw; - int global_col = test_size * test_nw; - std::ofstream ofs_running; - paraV = new Parallel_Orbitals(); - paraV->init(global_row, global_col, nb, MPI_COMM_WORLD); - paraV->set_atomic_trace(ucell.get_iat2iwt(), test_size, global_row); - } -#else - void init_parav() - { - } -#endif - - UnitCell ucell; - hamilt::HContainer* HR; - Parallel_Orbitals* paraV; - TwoCenterIntegrator intor_; - - int dsize; - int my_rank = 0; - double U_test = 1.0; - int orbital_c_test = 2; - double onsite_radius_test = 1.0; -}; - -// using TEST_F to test DFTU -TEST_F(DFTUTest, constructHRd2d) -{ - // test for nspin=1 - const int nspin = 1; - std::vector> kvec_d_in(1, ModuleBase::Vector3(0.0, 0.0, 0.0)); - hamilt::HS_Matrix_K hsk(paraV, true); - hsk.set_zero_hk(); - Grid_Driver gd(0, 0); - // build a solver-like density matrix: uniform DMK gives uniform DMR (= factor) at Gamma point - const double factor = 1.0 / test_nw / test_nw / test_size / test_size; - elecstate::DensityMatrix dm(paraV, 1); - dm.init_DMR(*HR); - for (int i = 0; i < paraV->nrow; i++) - { - for (int j = 0; j < paraV->ncol; j++) - { - dm.set_DMK(1, 0, i, j, factor); - } - } - dm.cal_DMR(); - // reset HR - for (int i = 0; i < HR->get_nnr(); i++) - { - HR->get_wrapper()[i] = 0.0; - } - std::chrono::high_resolution_clock::time_point start_time = std::chrono::high_resolution_clock::now(); - hamilt::DFTU> - op(&hsk, kvec_d_in, HR, ucell, &gd, &intor_, {1.0}, &dftu, nspin, onsite_radius_test, &dm); - std::chrono::high_resolution_clock::time_point end_time = std::chrono::high_resolution_clock::now(); - std::chrono::duration elapsed_time - = std::chrono::duration_cast>(end_time - start_time); - start_time = std::chrono::high_resolution_clock::now(); - op.contributeHR(); - end_time = std::chrono::high_resolution_clock::now(); - std::chrono::duration elapsed_time1 - = std::chrono::duration_cast>(end_time - start_time); - // check the occupations of dftu - for (int iat = 0; iat < test_size; iat++) - { - for (int icc = 0; icc < 25; icc++) - { - EXPECT_NEAR(occ_mat_c(iat, 0, icc), 0.5, 1e-10); - } - } - // check the value of HR - for (int iap = 0; iap < HR->size_atom_pairs(); ++iap) - { - hamilt::AtomPair& tmp = HR->get_atom_pair(iap); - int iat1 = tmp.get_atom_i(); - int iat2 = tmp.get_atom_j(); - std::vector indexes1 = paraV->get_indexes_row(iat1); - std::vector indexes2 = paraV->get_indexes_col(iat2); - int nwt = indexes1.size() * indexes2.size(); - for (int i = 0; i < nwt; ++i) - { - EXPECT_NEAR(tmp.get_pointer(0)[i], -10.0 * test_size, 1e-10); - } - } - // calculate SK - start_time = std::chrono::high_resolution_clock::now(); - op.contributeHk(0); - end_time = std::chrono::high_resolution_clock::now(); - std::chrono::duration elapsed_time2 - = std::chrono::duration_cast>(end_time - start_time); - // check the value of HK - double* hk = hsk.get_hk(); - for (int i = 0; i < paraV->get_row_size() * paraV->get_col_size(); ++i) - { - EXPECT_NEAR(hk[i], -10.0 * test_size, 1e-10); - } - std::cout << "Test terms: " << std::setw(15) << "initialize_HR" << std::setw(15) << "contributeHR" - << std::setw(15) << "contributeHk" << std::endl; - std::cout << "Elapsed time: " << std::setw(15) << elapsed_time.count() << std::setw(15) << elapsed_time1.count() - << std::setw(15) << elapsed_time2.count() << " seconds." << std::endl; -} - -TEST_F(DFTUTest, constructHRd2cd) -{ - // test for nspin=2 - const int nspin = 2; - std::vector> kvec_d_in(2, ModuleBase::Vector3(0.0, 0.0, 0.0)); - hamilt::HS_Matrix_K> hsk(paraV, true); - hsk.set_zero_hk(); - Grid_Driver gd(0, 0); - // build a solver-like density matrix: uniform DMK gives uniform DMR (= factor) at Gamma point - const double factor = 0.5 / test_nw / test_nw / test_size / test_size; - std::vector> kvec_d_dm(1, ModuleBase::Vector3(0.0, 0.0, 0.0)); - elecstate::DensityMatrix, double> dm(paraV, 2, kvec_d_dm, 1); - dm.init_DMR(*HR); - for (int is = 1; is <= 2; ++is) - { - for (int i = 0; i < paraV->nrow; i++) - { - for (int j = 0; j < paraV->ncol; j++) - { - dm.set_DMK(is, 0, i, j, std::complex(factor, 0.0)); - } - } - } - dm.cal_DMR(); - // reset HR - for (int i = 0; i < HR->get_nnr(); i++) - { - HR->get_wrapper()[i] = 0.0; - } - hamilt::DFTU, double>> - op(&hsk, kvec_d_in, HR, ucell, &gd, &intor_, {1.0}, &dftu, nspin, onsite_radius_test, &dm); - op.contributeHR(); - // check the occupations of dftu for spin-up - for (int iat = 0; iat < test_size; iat++) - { - for (int icc = 0; icc < 25; icc++) - { - EXPECT_NEAR(occ_mat_c(iat, 0, icc), 0.5, 1e-10); - } - } - // check the value of HR - for (int iap = 0; iap < HR->size_atom_pairs(); ++iap) - { - hamilt::AtomPair& tmp = HR->get_atom_pair(iap); - int iat1 = tmp.get_atom_i(); - int iat2 = tmp.get_atom_j(); - std::vector indexes1 = paraV->get_indexes_row(iat1); - std::vector indexes2 = paraV->get_indexes_col(iat2); - int nwt = indexes1.size() * indexes2.size(); - for (int i = 0; i < nwt; ++i) - { - EXPECT_NEAR(tmp.get_pointer(0)[i], -10.0 * test_size, 1e-10); - } - } - // calculate HK for gamma point - op.contributeHk(0); - // check the value of HK of gamma point - std::complex* hk = hsk.get_hk(); - for (int i = 0; i < paraV->get_row_size() * paraV->get_col_size(); ++i) - { - EXPECT_NEAR(hk[i].real(), -10.0 * test_size, 1e-10); - EXPECT_NEAR(hk[i].imag(), 0.0, 1e-10); - } - // calculate spin-down hamiltonian - op.contributeHR(); - // check the occupations of dftu for spin-down - for (int iat = 0; iat < test_size; iat++) - { - for (int icc = 0; icc < 25; icc++) - { - EXPECT_NEAR(occ_mat_c(iat, 1, icc), 0.5, 1e-10); - } - } -} - -int main(int argc, char** argv) -{ -#ifdef __MPI - MPI_Init(&argc, &argv); -#endif - testing::InitGoogleTest(&argc, argv); - int result = RUN_ALL_TESTS(); -#ifdef __MPI - MPI_Finalize(); -#endif - return result; -} diff --git a/source/source_lcao/module_dftu/test/dftu_operator_test.cpp b/source/source_lcao/module_dftu/test/dftu_operator_test.cpp deleted file mode 100644 index e65da640977..00000000000 --- a/source/source_lcao/module_dftu/test/dftu_operator_test.cpp +++ /dev/null @@ -1,325 +0,0 @@ -#include "gtest/gtest.h" -#include -#include -#include -#include - -/*********************************************************************** - * Unit tests for DFT+U and DeltaSpin operator math and force/stress. - * Tests cover: cal_pot_onsite, transfer_pot_onsite, cal_coeff_lambda, - * Force/IJR, Stress/IJR, Voigt->matrix, PW index setup - ***********************************************************************/ - -// ===================================================================== -// 1. cal_pot_onsite: Hubbard potential calculation -// nspin=1,2 (spin_fold < 4): pot_onsite[is] = U * (0.5*delta - occ^T) -// E_U += U * 0.5 * occ * occ^T -// nspin=4 (spin_fold == 4): pot_onsite[0] = U * (1.0*delta - occ^T), pot_onsite[is>0] = -U * occ^T -// E_U += U * 0.25 * occ * occ^T -// ===================================================================== - -static void cal_pot_onsite(const std::vector& occ, int m_size, double u_value, - std::vector& pot_onsite, double& eu) -{ - pot_onsite.assign(occ.size(), 0.0); - eu = 0.0; - int spin_fold = occ.size() / m_size / m_size; - if (spin_fold < 4) // nspin=1,2 - { - for (int is = 0; is < spin_fold; ++is) - { - int start = is * m_size * m_size; - for (int m1 = 0; m1 < m_size; m1++) - for (int m2 = 0; m2 < m_size; m2++) - { - pot_onsite[start + m1 * m_size + m2] = u_value * (0.5 * (m1 == m2) - occ[start + m2 * m_size + m1]); - eu += u_value * 0.5 * occ[start + m2 * m_size + m1] * occ[start + m1 * m_size + m2]; - } - } - } - else // nspin=4 - { - // is=0: Pauli I block - for (int m1 = 0; m1 < m_size; m1++) - for (int m2 = 0; m2 < m_size; m2++) - { - pot_onsite[m1 * m_size + m2] = u_value * (1.0 * (m1 == m2) - occ[m2 * m_size + m1]); - eu += u_value * 0.25 * occ[m2 * m_size + m1] * occ[m1 * m_size + m2]; - } - // is=1,2,3: Pauli off-diagonal blocks - for (int is = 1; is < spin_fold; ++is) - { - int start = is * m_size * m_size; - for (int m1 = 0; m1 < m_size; m1++) - for (int m2 = 0; m2 < m_size; m2++) - { - pot_onsite[start + m1 * m_size + m2] = u_value * (0.0 - occ[start + m2 * m_size + m1]); - eu += u_value * 0.25 * occ[start + m2 * m_size + m1] * occ[start + m1 * m_size + m2]; - } - } - } -} - -class CalVOfUTest : public ::testing::Test { protected: void SetUp() override {} }; - -TEST_F(CalVOfUTest, Nspin1_SingleOrbital_HalfFilled) -{ - std::vector occ = {0.5}; - std::vector pot_onsite; double eu = 0.0; - cal_pot_onsite(occ, 1, 4.0, pot_onsite, eu); - // pot_onsite = 4*(0.5-0.5)=0, E_U = 4*0.5*0.5*0.5=0.5 - EXPECT_DOUBLE_EQ(pot_onsite[0], 0.0); - EXPECT_DOUBLE_EQ(eu, 0.5); -} - -TEST_F(CalVOfUTest, Nspin2_DOrbital_SpinPolarized) -{ - const int m_size = 5; - std::vector occ(m_size * m_size * 2, 0.0); - for (int m = 0; m < m_size; m++) occ[m * m_size + m] = 0.8; // spin-up majority - for (int m = 0; m < m_size; m++) occ[m_size*m_size + m*m_size + m] = 0.2; // spin-down minority - std::vector pot_onsite; double eu = 0.0; - cal_pot_onsite(occ, m_size, 5.0, pot_onsite, eu); - // spin-up pot_onsite: 5*(0.5-0.8)=-1.5, spin-down pot_onsite: 5*(0.5-0.2)=1.5 - for (int m = 0; m < m_size; m++) EXPECT_NEAR(pot_onsite[m*m_size+m], -1.5, 1e-14); - for (int m = 0; m < m_size; m++) EXPECT_NEAR(pot_onsite[m_size*m_size+m*m_size+m], 1.5, 1e-14); - // E_U = 5*0.5*[5*(0.8^2)+5*(0.2^2)] = 8.5 - EXPECT_NEAR(eu, 8.5, 1e-14); -} - -TEST_F(CalVOfUTest, Nspin4_Porbital_PauliBlocks) -{ - const int m_size = 3; - std::vector occ(m_size * m_size * 4, 0.0); - for (int m = 0; m < m_size; m++) occ[m * m_size + m] = 0.5; // Pauli I block - std::vector pot_onsite; double eu = 0.0; - cal_pot_onsite(occ, m_size, 4.0, pot_onsite, eu); - // is=0: pot_onsite=4*(1.0-0.5)=2.0, is=1,2,3: pot_onsite=0 - for (int m = 0; m < m_size; m++) EXPECT_NEAR(pot_onsite[m*m_size+m], 2.0, 1e-14); - for (int is = 1; is < 4; is++) - for (int i = 0; i < m_size*m_size; i++) EXPECT_NEAR(pot_onsite[is*m_size*m_size+i], 0.0, 1e-14); - // E_U = 4*0.25*3*(0.5*0.5) = 0.75 - EXPECT_NEAR(eu, 0.75, 1e-14); -} - -// ===================================================================== -// 2. transfer_pot_onsite: Pauli matrix transformation (nspin=4) -// pot_onsite[0] = 0.5*(pot_onsite_tmp[0]+pot_onsite_tmp[3]) // Pauli I -// pot_onsite[3] = 0.5*(pot_onsite_tmp[0]-pot_onsite_tmp[3]) // Pauli sigma_z -// pot_onsite[1] = 0.5*(pot_onsite_tmp[1]+i*pot_onsite_tmp[2]) // sigma_x+i*sigma_y -// pot_onsite[2] = 0.5*(pot_onsite_tmp[1]-i*pot_onsite_tmp[2]) // sigma_x-i*sigma_y -// ===================================================================== - -static void transfer_pot_onsite(const std::vector& pot_onsite_tmp, - std::vector>& pot_onsite) -{ - const int m_size = int(sqrt(pot_onsite_tmp.size()) / 2); - const int m_size2 = m_size * m_size; - pot_onsite.resize(pot_onsite_tmp.size()); - for (int m1 = 0; m1 < m_size; m1++) - for (int m2 = 0; m2 < m_size; m2++) - { - int idx[4] = {m1*m_size+m2, m1*m_size+m2+m_size2, m2*m_size+m1+2*m_size2, m2*m_size+m1+3*m_size2}; - pot_onsite[idx[0]] = 0.5 * (pot_onsite_tmp[idx[0]] + pot_onsite_tmp[idx[3]]); - pot_onsite[idx[3]] = 0.5 * (pot_onsite_tmp[idx[0]] - pot_onsite_tmp[idx[3]]); - pot_onsite[idx[1]] = 0.5 * (pot_onsite_tmp[idx[1]] + std::complex(0,1) * pot_onsite_tmp[idx[2]]); - pot_onsite[idx[2]] = 0.5 * (pot_onsite_tmp[idx[1]] - std::complex(0,1) * pot_onsite_tmp[idx[2]]); - } -} - -class Transferpot_onsiteTest : public ::testing::Test { protected: void SetUp() override {} }; - -TEST_F(Transferpot_onsiteTest, PauliI_IdentityInput) -{ - std::vector pot_onsite_tmp = {1.0, 0.0, 0.0, 1.0}; - std::vector> pot_onsite; - transfer_pot_onsite(pot_onsite_tmp, pot_onsite); - EXPECT_NEAR(pot_onsite[0].real(), 1.0, 1e-15); EXPECT_NEAR(pot_onsite[0].imag(), 0.0, 1e-15); - EXPECT_NEAR(pot_onsite[3].real(), 0.0, 1e-15); -} - -TEST_F(Transferpot_onsiteTest, PureSigmaZ) -{ - std::vector pot_onsite_tmp = {1.0, 0.0, 0.0, -1.0}; - std::vector> pot_onsite; - transfer_pot_onsite(pot_onsite_tmp, pot_onsite); - EXPECT_NEAR(pot_onsite[0].real(), 0.0, 1e-15); EXPECT_NEAR(pot_onsite[3].real(), 1.0, 1e-15); -} - -TEST_F(Transferpot_onsiteTest, SigmaX_Y_Combined) -{ - std::vector pot_onsite_tmp_x = {0.0, 1.0, 1.0, 0.0}; - std::vector> pot_onsite; - transfer_pot_onsite(pot_onsite_tmp_x, pot_onsite); - EXPECT_NEAR(pot_onsite[1].real(), 0.5, 1e-15); EXPECT_NEAR(pot_onsite[1].imag(), 0.5, 1e-15); - EXPECT_NEAR(pot_onsite[2].real(), 0.5, 1e-15); EXPECT_NEAR(pot_onsite[2].imag(), -0.5, 1e-15); - - std::vector pot_onsite_tmp_y = {0.0, 1.0, -1.0, 0.0}; - transfer_pot_onsite(pot_onsite_tmp_y, pot_onsite); - EXPECT_NEAR(pot_onsite[1].real(), 0.5, 1e-15); EXPECT_NEAR(pot_onsite[1].imag(), -0.5, 1e-15); - EXPECT_NEAR(pot_onsite[2].real(), 0.5, 1e-15); EXPECT_NEAR(pot_onsite[2].imag(), 0.5, 1e-15); -} - -// ===================================================================== -// 3. cal_coeff_lambda: Lambda coefficient encoding -// Collinear (nspin=2): coeff[0]=lambda_z, coeff[1]=-lambda_z -// Non-collinear (nspin=4): coeff[0]=lambda_z, coeff[1]=lambda_x+i*lambda_y, -// coeff[2]=lambda_x-i*lambda_y, coeff[3]=-lambda_z -// ===================================================================== - -static void cal_coeff_lambda_collinear(const std::vector& lambda, - std::vector& coeff) -{ coeff[0] = lambda[0]; coeff[1] = -lambda[0]; } - -static void cal_coeff_lambda_noncollinear(const std::vector& lambda, - std::vector>& coeff) -{ - coeff[0] = std::complex(lambda[2], 0.0); - coeff[1] = std::complex(lambda[0], lambda[1]); - coeff[2] = std::complex(lambda[0], -lambda[1]); - coeff[3] = std::complex(-lambda[2], 0.0); -} - -class CalCoeffLambdaTest : public ::testing::Test { protected: void SetUp() override {} }; - -TEST_F(CalCoeffLambdaTest, Collinear_PositiveLambdaZ) -{ - std::vector lambda = {2.5}, coeff(2); - cal_coeff_lambda_collinear(lambda, coeff); - EXPECT_DOUBLE_EQ(coeff[0], 2.5); EXPECT_DOUBLE_EQ(coeff[1], -2.5); -} - -TEST_F(CalCoeffLambdaTest, NonCollinear_General) -{ - std::vector lambda = {1.0, 2.0, 3.0}; - std::vector> coeff(4); - cal_coeff_lambda_noncollinear(lambda, coeff); - EXPECT_NEAR(coeff[0].real(), 3.0, 1e-15); - EXPECT_NEAR(coeff[1].real(), 1.0, 1e-15); EXPECT_NEAR(coeff[1].imag(), 2.0, 1e-15); - EXPECT_NEAR(coeff[2].real(), 1.0, 1e-15); EXPECT_NEAR(coeff[2].imag(), -2.0, 1e-15); - EXPECT_NEAR(coeff[3].real(), -3.0, 1e-15); -} - -// ===================================================================== -// 4. Force/IJR core loop -// force1 += pot_onsite * * * DM -// force2 -= pot_onsite * * * DM -// nlm arrays: [value, deri_x, deri_y, deri_z] -// ===================================================================== - -static void cal_force_IJR_core(const std::vector& pot_onsite_in, - const std::vector& nlm1, const std::vector& nlm2, - const double dm_val, int m_size, int nspin, double force1[3], double force2[3]) -{ - const int m_size2 = m_size * m_size; - for (int is = 0; is < nspin; is++) - for (int m1 = 0; m1 < m_size; m1++) - for (int m2 = 0; m2 < m_size; m2++) - { - double pot_onsite = pot_onsite_in[m1*m_size + m2 + is*m_size2], dm = dm_val; - force1[0] += pot_onsite * nlm1[m1+m_size] * nlm2[m2] * dm; - force1[1] += pot_onsite * nlm1[m1+2*m_size] * nlm2[m2] * dm; - force1[2] += pot_onsite * nlm1[m1+3*m_size] * nlm2[m2] * dm; - force2[0] -= pot_onsite * nlm1[m1+m_size] * nlm2[m2] * dm; - force2[1] -= pot_onsite * nlm1[m1+2*m_size] * nlm2[m2] * dm; - force2[2] -= pot_onsite * nlm1[m1+3*m_size] * nlm2[m2] * dm; - } -} - -class ForceIJRTest : public ::testing::Test { protected: void SetUp() override {} }; - -TEST_F(ForceIJRTest, SingleOrbital_SingleSpin) -{ - std::vector pot_onsite = {2.0}, nlm1 = {1.0, 0.1, 0.2, 0.3}, nlm2 = {1.0, 0.0, 0.0, 0.0}; - double dm_val = 0.5, force1[3]={0}, force2[3]={0}; - cal_force_IJR_core(pot_onsite, nlm1, nlm2, dm_val, 1, 1, force1, force2); - // force = pot_onsite*deri(nlm1)*val(nlm2)*DM = 2.0*{0.1,0.2,0.3}*1.0*0.5 = {0.1,0.2,0.3} - EXPECT_NEAR(force1[0], 0.1, 1e-15); EXPECT_NEAR(force1[1], 0.2, 1e-15); EXPECT_NEAR(force1[2], 0.3, 1e-15); - EXPECT_NEAR(force2[0], -0.1, 1e-15); EXPECT_NEAR(force2[1], -0.2, 1e-15); EXPECT_NEAR(force2[2], -0.3, 1e-15); -} - -TEST_F(ForceIJRTest, ActionReaction) -{ - std::vector pot_onsite = {1.5}, nlm1 = {1.0, 0.3, 0.4, 0.5}, nlm2 = {1.0, 0.0, 0.0, 0.0}; - double dm_val = 1.0, force1[3]={0}, force2[3]={0}; - cal_force_IJR_core(pot_onsite, nlm1, nlm2, dm_val, 1, 1, force1, force2); - for (int i = 0; i < 3; i++) EXPECT_NEAR(force1[i], -force2[i], 1e-15); -} - -// ===================================================================== -// 5. Stress/IJR core loop -// stress[0]+=pot_onsite*DM*(nlm1_dx*dis1.x*nlm2_val+nlm1_val*nlm2_dx*dis2.x) -// stress[3]+=pot_onsite*DM*(nlm1_dy*dis1.y*nlm2_val+nlm1_val*nlm2_dy*dis2.y) -// stress[5]+=pot_onsite*DM*(nlm1_dz*dis1.z*nlm2_val+nlm1_val*nlm2_dz*dis2.z) -// ===================================================================== - -static void cal_stress_IJR_core(const std::vector& pot_onsite_in, - const std::vector& nlm1, const std::vector& nlm2, - const double dm_val, int m_size, int nspin, - double dis1[3], double dis2[3], double stress[6]) -{ - const int m_size2 = m_size * m_size; - for (int is = 0; is < nspin; is++) - for (int m1 = 0; m1 < m_size; m1++) - for (int m2 = 0; m2 < m_size; m2++) - { - double tmp = pot_onsite_in[m1*m_size+m2+is*m_size2] * dm_val; - stress[0] += tmp*(nlm1[m1+m_size]*dis1[0]*nlm2[m2] + nlm1[m1]*nlm2[m2+m_size]*dis2[0]); - stress[1] += tmp*(nlm1[m1+m_size]*dis1[1]*nlm2[m2] + nlm1[m1]*nlm2[m2+m_size]*dis2[1]); - stress[2] += tmp*(nlm1[m1+m_size]*dis1[2]*nlm2[m2] + nlm1[m1]*nlm2[m2+m_size]*dis2[2]); - stress[3] += tmp*(nlm1[m1+2*m_size]*dis1[1]*nlm2[m2] + nlm1[m1]*nlm2[m2+2*m_size]*dis2[1]); - stress[4] += tmp*(nlm1[m1+2*m_size]*dis1[2]*nlm2[m2] + nlm1[m1]*nlm2[m2+2*m_size]*dis2[2]); - stress[5] += tmp*(nlm1[m1+3*m_size]*dis1[2]*nlm2[m2] + nlm1[m1]*nlm2[m2+3*m_size]*dis2[2]); - } -} - -class StressIJRTest : public ::testing::Test { protected: void SetUp() override {} }; - -TEST_F(StressIJRTest, SingleOrbital_XDisplacement) -{ - std::vector pot_onsite = {1.0}, nlm1 = {1.0, 0.1, 0.0, 0.0}, nlm2 = {1.0, 0.2, 0.0, 0.0}; - double dm_val = 1.0, dis1[3] = {1.0, 0.0, 0.0}, dis2[3] = {-1.0, 0.0, 0.0}, stress[6] = {0.0}; - cal_stress_IJR_core(pot_onsite, nlm1, nlm2, dm_val, 1, 1, dis1, dis2, stress); - // stress[0] = 1.0*(0.1*1.0*1.0 + 1.0*0.2*(-1.0)) = -0.1 - EXPECT_NEAR(stress[0], -0.1, 1e-15); - EXPECT_NEAR(stress[1], 0.0, 1e-15); EXPECT_NEAR(stress[2], 0.0, 1e-15); -} - -TEST_F(StressIJRTest, SymmetricDisplacement) -{ - std::vector pot_onsite = {2.0}, nlm1 = {1.0, 0.1, 0.2, 0.3}, nlm2 = {1.0, 0.1, 0.2, 0.3}; - double dm_val = 1.0, dis1[3] = {1.0, 2.0, 3.0}, dis2[3] = {1.0, 2.0, 3.0}, stress[6] = {0.0}; - cal_stress_IJR_core(pot_onsite, nlm1, nlm2, dm_val, 1, 1, dis1, dis2, stress); - EXPECT_NEAR(stress[0], 2.0*(0.1*1.0 + 1.0*0.1*1.0), 1e-15); // xx - EXPECT_NEAR(stress[4], 2.0*(0.2*3.0 + 1.0*0.2*3.0), 1e-15); // yz -} - -// ===================================================================== -// 6. Stress Voigt -> matrix mapping -// Voigt [xx,xy,xz,yz,yy,zz] -> 3x3 symmetric matrix -// ===================================================================== - -static void voigt_to_matrix(double stress_6[6], double matrix[9]) -{ - for (int i = 0; i < 9; i++) matrix[i] = 0.0; - matrix[0]=stress_6[0]; matrix[1]=stress_6[1]; matrix[2]=stress_6[2]; - matrix[3]=stress_6[1]; matrix[4]=stress_6[4]; matrix[5]=stress_6[3]; - matrix[6]=stress_6[2]; matrix[7]=stress_6[3]; matrix[8]=stress_6[5]; -} - -class VoigtToMatrixTest : public ::testing::Test { protected: void SetUp() override {} }; - -TEST_F(VoigtToMatrixTest, FullMappingAndSymmetry) -{ - double stress_6[6] = {1.0, 2.0, 3.0, 4.0, 5.0, 6.0}, matrix[9]; - voigt_to_matrix(stress_6, matrix); - // [0,0]=1, [0,1]=2, [0,2]=3, [1,1]=5, [1,2]=4, [2,2]=6 - EXPECT_NEAR(matrix[0], 1.0, 1e-15); EXPECT_NEAR(matrix[1], 2.0, 1e-15); - EXPECT_NEAR(matrix[2], 3.0, 1e-15); EXPECT_NEAR(matrix[3], 2.0, 1e-15); - EXPECT_NEAR(matrix[4], 5.0, 1e-15); EXPECT_NEAR(matrix[5], 4.0, 1e-15); - EXPECT_NEAR(matrix[6], 3.0, 1e-15); EXPECT_NEAR(matrix[7], 4.0, 1e-15); - EXPECT_NEAR(matrix[8], 6.0, 1e-15); - // Verify symmetry - EXPECT_NEAR(matrix[1], matrix[3], 1e-15); EXPECT_NEAR(matrix[2], matrix[6], 1e-15); - EXPECT_NEAR(matrix[5], matrix[7], 1e-15); -} diff --git a/source/source_lcao/module_dftu/test/test_dftu_nao_ijr.cpp b/source/source_lcao/module_dftu/test/test_dftu_nao_ijr.cpp deleted file mode 100644 index ae8a0282db9..00000000000 --- a/source/source_lcao/module_dftu/test/test_dftu_nao_ijr.cpp +++ /dev/null @@ -1,195 +0,0 @@ -#include "../dftu_nao_ijr.h" - -#include "gtest/gtest.h" - -#include -#include -#include - -/// @file test_dftu_nao_ijr.cpp -/// @brief Focused unit test for the free function -/// DFTU_LCAO::accumulate_hr_for_iat0 (dftu_nao_ijr.h), which -/// accumulates the real-space HR blocks of one Hubbard atom. -/// Two atoms with 2 d-type orbitals each (nw=2, iw2l=2); the -/// Hubbard center (iat0=0) sees both atoms as neighbors at R=0. -/// With nlm=1, pot_onsite(m,m')=delta_{m,m'} and the full (I,J,R) -/// pair grid present in HR, every HR entry accumulates 5. -/// iat2it/iat2ia are borrowed pointers into iat2it_buf/iat2ia_buf; -/// their lifetime is managed by the fixture vectors (and ultimately -/// released by UnitCell's Statistics destructor). -class AccumulateHrIat0Test : public ::testing::Test -{ - protected: - void SetUp() override - { -#ifdef __MPI - MPI_Comm_size(MPI_COMM_WORLD, &dsize); - MPI_Comm_rank(MPI_COMM_WORLD, &my_rank); -#endif - ucell.ntype = 1; - ucell.nat = nat; - ucell.atoms = atoms_buf; - iat2it_buf.assign(nat, 0); - iat2ia_buf.resize(nat); - ucell.iat2it = iat2it_buf.data(); - ucell.iat2ia = iat2ia_buf.data(); - ucell.atoms[0].tau.resize(nat); - ucell.lat0 = 1.0; - ucell.itia2iat.create(ucell.ntype, nat); - for (int iat = 0; iat < nat; iat++) - { - iat2ia_buf[iat] = iat; - ucell.atoms[0].tau[iat] = ModuleBase::Vector3(0.0, 0.0, 0.0); - ucell.itia2iat(0, iat) = iat; - } - ucell.atoms[0].na = nat; - ucell.atoms[0].nw = nw; - ucell.atoms[0].iw2l = {2, 2}; - ucell.atoms[0].iw2m = {0, 0}; - ucell.atoms[0].iw2n = {0, 0}; - ucell.set_iat2iwt(1); - paraV.reset(new Parallel_Orbitals()); -#ifdef __MPI - paraV->init(nat * nw, nat * nw, nat * nw, MPI_COMM_WORLD); - paraV->set_atomic_trace(ucell.get_iat2iwt(), nat, nat * nw); -#endif - // HR holds the full 3x3x3 R grid of atom pairs - HR.reset(new hamilt::HContainer(ucell, paraV.get())); - std::fill(HR->get_wrapper(), HR->get_wrapper() + HR->get_nnr(), 0.0); - } - - void TearDown() override - { - // reset HR/paraV before unitcell buffers go out of scope; - // iat2it/iat2ia are released by UnitCell's Statistics member - HR.reset(); - paraV.reset(); - ucell.atoms = nullptr; - ucell.iat2it = nullptr; - ucell.iat2ia = nullptr; - } - - // neighbor list of Hubbard center iat0=0: both atoms at box (0,0,0) - AdjacentAtomInfo make_adjs() const - { - AdjacentAtomInfo adjs; - adjs.adj_num = 1; // one neighbor besides the center itself - adjs.ntype = {0, 0}; - adjs.natom = {0, 1}; - adjs.box = {ModuleBase::Vector3(0, 0, 0), ModuleBase::Vector3(0, 0, 0)}; - return adjs; - } - - // nlm_tot[iat0][ad][orbital-index(iw*5+m)] = 1 for both neighbors - DFTU_LCAO::NlmTot make_nlm_tot() const - { - DFTU_LCAO::NlmTot nlm_tot(nat); - for (int iat = 0; iat < nat; ++iat) - { - nlm_tot[iat].resize(2); - for (int ad = 0; ad < 2; ++ad) - { - for (int iw = 0; iw < nw; ++iw) - { - for (int m = 0; m < 5; ++m) - { - nlm_tot[iat][ad][iw * 5 + m] = std::vector(5, 1.0); - } - } - } - } - return nlm_tot; - } - - const int nat = 2; - const int nw = 2; - int dsize = 1; - int my_rank = 0; - UnitCell ucell; - Atom atoms_buf[1]; // borrowed by ucell.atoms (raw Atom* member) - std::vector iat2it_buf; // borrowed by ucell.iat2it - std::vector iat2ia_buf; // borrowed by ucell.iat2ia - std::unique_ptr paraV; - std::unique_ptr> HR; -}; - -TEST_F(AccumulateHrIat0Test, AccumulatesAllPairs) -{ - AdjacentAtomInfo adjs = make_adjs(); - DFTU_LCAO::NlmTot nlm_tot = make_nlm_tot(); - // 5x5 identity: pot_onsite(m,m') = delta_{m,m'} - std::vector pot_onsite(25, 0.0); - for (int m = 0; m < 5; ++m) - { - pot_onsite[m * 5 + m] = 1.0; - } - - DFTU_LCAO::accumulate_hr_for_iat0(ucell, HR.get(), nlm_tot, 0, adjs, *paraV, pot_onsite); - - // every local HR entry gets sum_m 1*1*1 = 5 per orbital pair, - // i.e. each element of an orbital-pair block equals 5; - // each atom pair holds 2x2 orbital pairs => each HR value is 5 - for (int iap = 0; iap < HR->size_atom_pairs(); ++iap) - { - hamilt::AtomPair& tmp = HR->get_atom_pair(iap); - const int iat1 = tmp.get_atom_i(); - const int iat2 = tmp.get_atom_j(); - std::vector indexes1 = paraV->get_indexes_row(iat1); - std::vector indexes2 = paraV->get_indexes_col(iat2); - const int nwt = indexes1.size() * indexes2.size(); - // only pairs with R = (0,0,0) are touched; the R grid of - // HContainer(ucell) only contains the R=0 block per pair - for (int i = 0; i < nwt; ++i) - { - EXPECT_NEAR(tmp.get_pointer(0)[i], 5.0, 1e-12); - } - } -} - -TEST_F(AccumulateHrIat0Test, MissingPairIsSkipped) -{ - AdjacentAtomInfo adjs = make_adjs(); - // move neighbor ad2 to R=(5,5,5): the cross pairs (ad1,ad2)=(0,1) - // and (1,0) then map to R_vector = +/-(5,5,5), for which HR holds - // no block, so accumulate_hr_for_iat0 must skip them. The diagonal - // pairs (0,0) and (1,1) still have R_vector = 0 and are accumulated. - adjs.box[1] = ModuleBase::Vector3(5, 5, 5); - // precondition: HR really has no (0,1) block at R=(5,5,5) - ASSERT_EQ(HR->find_matrix(0, 1, 5, 5, 5), nullptr); - DFTU_LCAO::NlmTot nlm_tot = make_nlm_tot(); - // pot_onsite(m,m') = 1 everywhere: each touched entry gets - // sum_{m,m'} 1*1*1 = 25 - std::vector pot_onsite(25, 1.0); - - DFTU_LCAO::accumulate_hr_for_iat0(ucell, HR.get(), nlm_tot, 0, adjs, *paraV, pot_onsite); - - for (int iap = 0; iap < HR->size_atom_pairs(); ++iap) - { - hamilt::AtomPair& tmp = HR->get_atom_pair(iap); - const int iat1 = tmp.get_atom_i(); - const int iat2 = tmp.get_atom_j(); - std::vector indexes1 = paraV->get_indexes_row(iat1); - std::vector indexes2 = paraV->get_indexes_col(iat2); - const int nwt = indexes1.size() * indexes2.size(); - // diagonal pairs are accumulated (25); cross pairs whose only - // contribution sits at the missing R block stay zero - const double expected = (iat1 == iat2) ? 25.0 : 0.0; - for (int i = 0; i < nwt; ++i) - { - EXPECT_DOUBLE_EQ(tmp.get_pointer(0)[i], expected); - } - } -} - -int main(int argc, char** argv) -{ -#ifdef __MPI - MPI_Init(&argc, &argv); -#endif - testing::InitGoogleTest(&argc, argv); - int result = RUN_ALL_TESTS(); -#ifdef __MPI - MPI_Finalize(); -#endif - return result; -} diff --git a/source/source_lcao/module_dftu/unittests/CMakeLists.txt b/source/source_lcao/module_dftu/unittests/CMakeLists.txt new file mode 100644 index 00000000000..1e436770d62 --- /dev/null +++ b/source/source_lcao/module_dftu/unittests/CMakeLists.txt @@ -0,0 +1,126 @@ +abacus_disable_feature_definitions(__MPI) +abacus_disable_feature_definitions(__EXX) +abacus_disable_feature_definitions(__CUDA) +abacus_disable_feature_definitions(__UT_USE_CUDA) +abacus_disable_feature_definitions(__UT_USE_ROCM) +abacus_disable_feature_definitions(__ROCM) +abacus_disable_feature_definitions(__MLALGO) +abacus_disable_feature_definitions(_OPENMP) + +if (ENABLE_MPI) + +AddTest( + TARGET MODULE_DFTU_pots + LIBS base device + SOURCES test_dftu_nao_pots.cpp +) + +AddTest( + TARGET MODULE_DFTU_energy + LIBS base device + SOURCES test_dftu_nao_energy.cpp +) + +AddTest( + TARGET MODULE_DFTU_op_legacy + LIBS base device + SOURCES test_dftu_nao_op_legacy.cpp +) + +AddTest( + TARGET MODULE_DFTU_for_r + LIBS base device + SOURCES test_dftu_nao_for_r.cpp +) + +AddTest( + TARGET MODULE_DFTU_str_r + LIBS base device + SOURCES test_dftu_nao_str_r.cpp +) + +AddTest( + TARGET MODULE_DFTU_folding + LIBS base device + SOURCES test_dftu_nao_folding.cpp ../../../source_basis/module_ao/parallel_orbitals.cpp + # The test instantiates Parallel_Orbitals (a Parallel_2D subclass) and links + # the MPI-built base library, so it must keep __MPI to match the layout of + # Parallel_2D / Parallel_Orbitals across TUs (otherwise ODR/ABI mismatch). + KEEP_FEATURE_DEFINITIONS __MPI +) + +AddTest( + TARGET MODULE_DFTU_fs_r + LIBS base device + SOURCES test_dftu_nao_fs_r.cpp ../dftu_nao_fs_reduce.cpp +) + +AddTest( + TARGET MODULE_DFTU_fs_reduce + LIBS base device + SOURCES test_dftu_nao_fs_reduce.cpp ../dftu_nao_fs_reduce.cpp +) + +AddTest( + TARGET MODULE_DFTU_fs_k + LIBS base device + SOURCES test_dftu_nao_fs_k.cpp +) + +AddTest( + TARGET MODULE_DFTU_fs_accum + LIBS base device + SOURCES test_dftu_nao_fs_accum.cpp ../../../source_basis/module_ao/parallel_orbitals.cpp + # This test uses Parallel_2D::set_serial which is a real implementation + # in parallel_2d.cpp; keep __MPI so the serial path links correctly. + KEEP_FEATURE_DEFINITIONS __MPI +) + +if(ENABLE_LCAO) +AddTest( + TARGET MODULE_DFTU_op + LIBS parameter psi base device container symmetry + SOURCES test_dftu_nao_op.cpp ../dftu_nao_op.cpp ../dftu_nao_adj.cpp ../dftu_nao_pots.cpp ../dftu_nao_fs_r.cpp ../dftu_nao_fs_reduce.cpp ../dftu_nao_for_r.cpp ../dftu_nao_str_r.cpp ../../../source_estate/module_dm/density_matrix.cpp + ../../../source_estate/module_dm/dmr_gamma.cpp + ../../../source_estate/module_dm/dmr_init.cpp + ../../../source_estate/module_dm/dm_setter.cpp + ../../../source_estate/module_dm/dm_getter.cpp + ../../../source_estate/module_dm/dm_tools.cpp + ../../../source_estate/module_dm/dmr_k.cpp + ../../../source_estate/module_dm/dmr_td.cpp + ../../../source_estate/module_dm/dmr_full.cpp + ../../../source_pw/module_pwdft/dftu_base.cpp + ../../../source_pw/module_pwdft/dftu_base_io.cpp + ../../../source_pw/module_pwdft/yukawa_screening.cpp + ../../../source_estate/occ_matrix.cpp + ../../../source_estate/occ_mixer.cpp + ../../../source_hamilt/module_hcontainer/func_folding.cpp + ../../../source_hamilt/module_hcontainer/base_matrix.cpp + ../../../source_hamilt/module_hcontainer/hcontainer.cpp + ../../../source_hamilt/module_hcontainer/atom_pair.cpp + ../../../source_basis/module_ao/parallel_orbitals.cpp + ../../../source_basis/module_ao/orb_atomic_lm.cpp + ../../module_operator_lcao/test/tmp_mocks.cpp + ../../../source_hamilt/operator.cpp + # This test calls MPI_Init in main() and drives ScaLAPACK distribution, + # so it must keep __MPI even though this directory disables it. + KEEP_FEATURE_DEFINITIONS __MPI +) + +AddTest( + TARGET MODULE_DFTU_ijr + LIBS parameter psi base device container + SOURCES test_dftu_nao_ijr.cpp + ../../../source_hamilt/module_hcontainer/func_folding.cpp + ../../../source_hamilt/module_hcontainer/base_matrix.cpp + ../../../source_hamilt/module_hcontainer/hcontainer.cpp + ../../../source_hamilt/module_hcontainer/atom_pair.cpp + ../../../source_basis/module_ao/parallel_orbitals.cpp + ../../../source_basis/module_ao/orb_atomic_lm.cpp + ../../module_operator_lcao/test/tmp_mocks.cpp + ../../../source_hamilt/operator.cpp + KEEP_FEATURE_DEFINITIONS __MPI +) +endif() + +endif() diff --git a/source/source_lcao/module_dftu/unittests/test_dftu_nao_energy.cpp b/source/source_lcao/module_dftu/unittests/test_dftu_nao_energy.cpp new file mode 100644 index 00000000000..0e9f90ecb62 --- /dev/null +++ b/source/source_lcao/module_dftu/unittests/test_dftu_nao_energy.cpp @@ -0,0 +1,75 @@ +#include "gtest/gtest.h" +#include +#include + +/*********************************************************************** + * Unit tests for the DFT+U energy correction formula (dftu_nao_energy.cpp). + * + * E_U = 0.5 * U * sum_spin [Tr(n) - Tr(n^2)] + ***********************************************************************/ + +class EnergyCorrectionTest : public ::testing::Test +{ + protected: + static double compute_energy(const std::vector& occ_mat_flat, int m_size, double U) + { + double nm_trace = 0.0, nm2_trace = 0.0; + for (int m0 = 0; m0 < m_size; m0++) + { + nm_trace += occ_mat_flat[m0 * m_size + m0]; + for (int m1 = 0; m1 < m_size; m1++) + nm2_trace += occ_mat_flat[m0 * m_size + m1] * occ_mat_flat[m1 * m_size + m0]; + } + return 0.5 * U * (nm_trace - nm2_trace); + } +}; + +TEST_F(EnergyCorrectionTest, HalfFilled_DOrbital) +{ + const int m_size = 5; + std::vector occ_mat(m_size * m_size, 0.0); + for (int m = 0; m < m_size; m++) + occ_mat[m * m_size + m] = 0.5; + + double energy = compute_energy(occ_mat, m_size, 4.0); + // Tr(n) = 2.5, Tr(n^2) = 1.25, E = 0.5 * 4 * 1.25 = 2.5 + EXPECT_DOUBLE_EQ(energy, 2.5); +} + +TEST_F(EnergyCorrectionTest, OffDiagonal_Contribution) +{ + const int m_size = 2; + std::vector occ_mat = { + 0.3, 0.1, + 0.1, 0.3 + }; + + double energy = compute_energy(occ_mat, m_size, 4.0); + // Tr(n) = 0.6, Tr(n^2) = 0.3^2 + 0.1^2 + 0.1^2 + 0.3^2 = 0.20 + // E = 0.5 * 4 * (0.6 - 0.20) = 0.8 + EXPECT_DOUBLE_EQ(energy, 0.8); +} + +TEST_F(EnergyCorrectionTest, DoubleCounting_Energy) +{ + // E_dc = sum_{m1,m2,spin} pot_onsite[m1,m2] * n[m2,m1] + const int m_size = 3; + double U = 4.0; + std::vector occ_mat = { + 0.5, 0.0, 0.0, + 0.0, 0.3, 0.0, + 0.0, 0.0, 0.2 + }; + + double e_dc = 0.0; + for (int m1 = 0; m1 < m_size; m1++) + for (int m2 = 0; m2 < m_size; m2++) + { + double pot_onsite = (m1 == m2) ? U * (0.5 - occ_mat[m1 * m_size + m2]) + : -U * occ_mat[m1 * m_size + m2]; + e_dc += pot_onsite * occ_mat[m2 * m_size + m1]; + } + + // Only diagonal: m=0: 0*0.5=0, m=1: 0.8*0.3=0.24, m=2: 1.2*0.2=0.24 + EXPECT_NEAR(e_dc, 0.48, 1e-14); +} diff --git a/source/source_lcao/module_dftu/unittests/test_dftu_nao_folding.cpp b/source/source_lcao/module_dftu/unittests/test_dftu_nao_folding.cpp new file mode 100644 index 00000000000..af5a6242730 --- /dev/null +++ b/source/source_lcao/module_dftu/unittests/test_dftu_nao_folding.cpp @@ -0,0 +1,70 @@ +#include "source_cell/unitcell.h" + +// Minimal mocks to satisfy the linker without pulling in heavy dependencies. +Magnetism::Magnetism() {} +Magnetism::~Magnetism() {} +SepPot::SepPot() {} +SepPot::~SepPot() {} +Sep_Cell::Sep_Cell() noexcept {} +Sep_Cell::~Sep_Cell() noexcept {} +UnitCell::UnitCell() {} +UnitCell::~UnitCell() {} + +#include "../dftu_nao_folding.h" + +#include "source_basis/module_ao/parallel_orbitals.h" +#include "gtest/gtest.h" + +#include + +// Minimal stub for get_linear_index to keep the link closure small; only the +// index arithmetic is under test. Parallel_Orbitals' ctor/dtor come from the +// real parallel_orbitals.cpp (wired in via CMakeLists) so the class layout +// matches the MPI-built base library. +namespace DFTU_LCAO +{ +int get_linear_index(const std::string& ks_solver, + const int mu, + const int nu, + const Parallel_Orbitals& pv) +{ + if (ModuleBase::GlobalFunc::IS_COLUMN_MAJOR_KS_SOLVER(ks_solver)) + { + return mu + nu * pv.nrow; + } + return mu * pv.ncol + nu; +} +} // namespace DFTU_LCAO + +namespace DFTU_LCAO +{ + +TEST(GetLinearIndexTest, RowMajor) +{ + Parallel_Orbitals pv; + pv.nrow = 3; + pv.ncol = 4; + + // row-major: mu * ncol + nu ("cg" is not column-major) + EXPECT_EQ(get_linear_index("cg", 0, 0, pv), 0); + EXPECT_EQ(get_linear_index("cg", 0, 1, pv), 1); + EXPECT_EQ(get_linear_index("cg", 1, 0, pv), 4); + EXPECT_EQ(get_linear_index("cg", 1, 2, pv), 6); + EXPECT_EQ(get_linear_index("cg", 2, 3, pv), 11); +} + +TEST(GetLinearIndexTest, ColumnMajor) +{ + Parallel_Orbitals pv; + pv.nrow = 3; + pv.ncol = 4; + + // column-major: mu + nu * nrow + EXPECT_EQ(get_linear_index("scalapack_gvx", 0, 0, pv), 0); + EXPECT_EQ(get_linear_index("scalapack_gvx", 1, 0, pv), 1); + EXPECT_EQ(get_linear_index("scalapack_gvx", 0, 1, pv), 3); + EXPECT_EQ(get_linear_index("scalapack_gvx", 2, 1, pv), 5); + EXPECT_EQ(get_linear_index("scalapack_gvx", 2, 3, pv), 11); +} + +} // namespace DFTU_LCAO diff --git a/source/source_lcao/module_dftu/unittests/test_dftu_nao_for_r.cpp b/source/source_lcao/module_dftu/unittests/test_dftu_nao_for_r.cpp new file mode 100644 index 00000000000..0dd846739d6 --- /dev/null +++ b/source/source_lcao/module_dftu/unittests/test_dftu_nao_for_r.cpp @@ -0,0 +1,49 @@ +#include "gtest/gtest.h" +#include +#include + +/*********************************************************************** + * Unit tests for the force accumulation in dftu_nao_for_r.cpp. + * + * force1 += pot_onsite * * * DM + * force2 -= pot_onsite * * * DM + * nlm arrays: [value, deri_x, deri_y, deri_z] + ***********************************************************************/ + +static void cal_force_IJR_core(const std::vector& pot_onsite_in, + const std::vector& nlm1, const std::vector& nlm2, + const double dm_val, int m_size, int nspin, double force1[3], double force2[3]) +{ + const int m_size2 = m_size * m_size; + for (int is = 0; is < nspin; is++) + for (int m1 = 0; m1 < m_size; m1++) + for (int m2 = 0; m2 < m_size; m2++) + { + double pot_onsite = pot_onsite_in[m1*m_size + m2 + is*m_size2], dm = dm_val; + force1[0] += pot_onsite * nlm1[m1+m_size] * nlm2[m2] * dm; + force1[1] += pot_onsite * nlm1[m1+2*m_size] * nlm2[m2] * dm; + force1[2] += pot_onsite * nlm1[m1+3*m_size] * nlm2[m2] * dm; + force2[0] -= pot_onsite * nlm1[m1+m_size] * nlm2[m2] * dm; + force2[1] -= pot_onsite * nlm1[m1+2*m_size] * nlm2[m2] * dm; + force2[2] -= pot_onsite * nlm1[m1+3*m_size] * nlm2[m2] * dm; + } +} + +class ForceIJRTest : public ::testing::Test { protected: void SetUp() override {} }; + +TEST_F(ForceIJRTest, SingleOrbital_SingleSpin) +{ + std::vector pot_onsite = {2.0}, nlm1 = {1.0, 0.1, 0.2, 0.3}, nlm2 = {1.0, 0.0, 0.0, 0.0}; + double dm_val = 0.5, force1[3]={0}, force2[3]={0}; + cal_force_IJR_core(pot_onsite, nlm1, nlm2, dm_val, 1, 1, force1, force2); + EXPECT_NEAR(force1[0], 0.1, 1e-15); EXPECT_NEAR(force1[1], 0.2, 1e-15); EXPECT_NEAR(force1[2], 0.3, 1e-15); + EXPECT_NEAR(force2[0], -0.1, 1e-15); EXPECT_NEAR(force2[1], -0.2, 1e-15); EXPECT_NEAR(force2[2], -0.3, 1e-15); +} + +TEST_F(ForceIJRTest, ActionReaction) +{ + std::vector pot_onsite = {1.5}, nlm1 = {1.0, 0.3, 0.4, 0.5}, nlm2 = {1.0, 0.0, 0.0, 0.0}; + double dm_val = 1.0, force1[3]={0}, force2[3]={0}; + cal_force_IJR_core(pot_onsite, nlm1, nlm2, dm_val, 1, 1, force1, force2); + for (int i = 0; i < 3; i++) EXPECT_NEAR(force1[i], -force2[i], 1e-15); +} diff --git a/source/source_lcao/module_dftu/unittests/test_dftu_nao_fs_accum.cpp b/source/source_lcao/module_dftu/unittests/test_dftu_nao_fs_accum.cpp new file mode 100644 index 00000000000..f5f01e828a0 --- /dev/null +++ b/source/source_lcao/module_dftu/unittests/test_dftu_nao_fs_accum.cpp @@ -0,0 +1,113 @@ +#include "source_cell/unitcell.h" + +// Minimal mocks to satisfy the linker without pulling in heavy dependencies. +Magnetism::Magnetism() {} +Magnetism::~Magnetism() {} +SepPot::SepPot() {} +SepPot::~SepPot() {} +Sep_Cell::Sep_Cell() noexcept {} +Sep_Cell::~Sep_Cell() noexcept {} +UnitCell::UnitCell() {} +UnitCell::~UnitCell() {} + +#include "../dftu_nao_fs_accum.h" + +#include "source_base/matrix.h" +#include "source_basis/module_ao/parallel_orbitals.h" +#include "gtest/gtest.h" + +#include +#include + +namespace DFTU_LCAO +{ +namespace +{ + +class DftuFsAccumTest : public ::testing::Test +{ + protected: + void SetUp() override + { + // 2 atoms, 3x3 serial local block with identity local<->global map + pv.set_serial(3, 3); + + // iwt2iat: orbital 0,1 -> atom 0; orbital 2 -> atom 1 + // Must use new[] because Statistics destructor will delete[] it. + ucell.iwt2iat = new int[3]{0, 0, 1}; + } + + Parallel_Orbitals pv; + UnitCell ucell; +}; + +TEST_F(DftuFsAccumTest, DiagForceDouble) +{ + // dm is column-major local block: dm[ic * nrow + ir] + // Diagonal entries (ir == ic): dm[0], dm[4], dm[8] + std::vector dm(9, 0.0); + dm[0] = 1.0; // (0,0) -> atom 0 + dm[4] = 2.0; // (1,1) -> atom 0 + dm[8] = 3.0; // (2,2) -> atom 1 + // Off-diagonal entries should be ignored + dm[1] = 10.0; + dm[3] = 20.0; + + ModuleBase::matrix force(2, 3); + accumulate_diag_force(pv, ucell, dm.data(), 0, force); + + EXPECT_DOUBLE_EQ(force(0, 0), 1.0 + 2.0); + EXPECT_DOUBLE_EQ(force(1, 0), 3.0); + EXPECT_DOUBLE_EQ(force(0, 1), 0.0); + EXPECT_DOUBLE_EQ(force(1, 1), 0.0); +} + +TEST_F(DftuFsAccumTest, DiagForceComplex) +{ + std::vector> dm(9, std::complex(0.0, 0.0)); + dm[0] = std::complex(1.0, 0.5); + dm[4] = std::complex(2.0, 1.0); + dm[8] = std::complex(3.0, -0.5); + + ModuleBase::matrix force(2, 3); + accumulate_diag_force(pv, ucell, dm.data(), 1, force); + + // Only real part contributes + EXPECT_DOUBLE_EQ(force(0, 1), 1.0 + 2.0); + EXPECT_DOUBLE_EQ(force(1, 1), 3.0); + EXPECT_DOUBLE_EQ(force(0, 0), 0.0); +} + +TEST_F(DftuFsAccumTest, DiagStressDouble) +{ + std::vector dm(9, 0.0); + dm[0] = 1.0; + dm[4] = 2.0; + dm[8] = 3.0; + + ModuleBase::matrix stress(3, 3); + accumulate_diag_stress(pv, dm.data(), 0, 1, 2.0, stress); + + // factor = 2.0 applied to all diagonal entries + EXPECT_DOUBLE_EQ(stress(0, 1), 2.0 * (1.0 + 2.0 + 3.0)); + EXPECT_DOUBLE_EQ(stress(0, 0), 0.0); + EXPECT_DOUBLE_EQ(stress(1, 1), 0.0); +} + +TEST_F(DftuFsAccumTest, DiagStressComplex) +{ + std::vector> dm(9, std::complex(0.0, 0.0)); + dm[0] = std::complex(1.0, 0.5); + dm[4] = std::complex(2.0, 1.0); + dm[8] = std::complex(3.0, -0.5); + + ModuleBase::matrix stress(3, 3); + accumulate_diag_stress(pv, dm.data(), 2, 2, -0.5, stress); + + // factor = -0.5, real part only + EXPECT_DOUBLE_EQ(stress(2, 2), -0.5 * (1.0 + 2.0 + 3.0)); + EXPECT_DOUBLE_EQ(stress(0, 0), 0.0); +} + +} // namespace +} // namespace DFTU_LCAO diff --git a/source/source_lcao/module_dftu/unittests/test_dftu_nao_fs_k.cpp b/source/source_lcao/module_dftu/unittests/test_dftu_nao_fs_k.cpp new file mode 100644 index 00000000000..7b690e6d11d --- /dev/null +++ b/source/source_lcao/module_dftu/unittests/test_dftu_nao_fs_k.cpp @@ -0,0 +1,100 @@ +#include "source_cell/unitcell.h" + +// Minimal mocks to satisfy the linker without pulling in heavy dependencies. +Magnetism::Magnetism() {} +Magnetism::~Magnetism() {} +SepPot::SepPot() {} +SepPot::~SepPot() {} +Sep_Cell::Sep_Cell() noexcept {} +Sep_Cell::~Sep_Cell() noexcept {} +UnitCell::UnitCell() {} +UnitCell::~UnitCell() {} + +#include "../dftu_nao_fs_k.h" + +#include "source_lcao/force_stress_arrays.h" +#include "source_pw/module_pwdft/dftu_base.h" + +#include "gtest/gtest.h" + +#include +#include + +/*********************************************************************** + * Unit tests for dftu_nao_fs_k.cpp. + * + * The bulk of the file (force_stress, cal_force_k, cal_stress_k) lives in + * an anonymous namespace or needs Grid_Driver / BLACS / TwoCenterIntegrator + * and is explicitly documented as hard to unit-test in the source. The + * only publicly visible surface is the DftuFsEnv environment struct, whose + * accessors bundle references to the shared dependencies. We verify that + * the constructor stores each reference unchanged so downstream kernels + * see the same objects the caller passed in. + * + * Note: we allocate the dependency objects on the heap and never destroy + * them, so the linker never sees references to their (heavy) constructors + * or destructors. This keeps the test target free of BLACS / neighbor / + * two-center linkage while still exercising the reference semantics of + * DftuFsEnv. + ***********************************************************************/ + +namespace DFTU_LCAO +{ +namespace +{ + +class DftuFsKTest : public ::testing::Test +{ + protected: + void SetUp() override + { + // Intentionally leaked: their destructors pull in heavy symbols + // (BLACS teardown, neighbor-list cleanup, ...). The test process + // is short-lived, so the leak is harmless. + dftu = static_cast(::operator new(sizeof(Plus_U_Base))); + fsr = static_cast(::operator new(sizeof(ForceStressArrays))); + gd = static_cast(::operator new(sizeof(Grid_Driver))); + pv = static_cast(::operator new(sizeof(Parallel_Orbitals))); + } + + Plus_U_Base* dftu = nullptr; + ForceStressArrays* fsr = nullptr; + Grid_Driver* gd = nullptr; + Parallel_Orbitals* pv = nullptr; + UnitCell ucell; +}; + +TEST_F(DftuFsKTest, EnvStoresReferencesUnchanged) +{ + std::vector orb_cutoff = {5.0, 6.0}; + const std::string solver = "scalapack"; + + DftuFsEnv env(*dftu, ucell, *gd, *pv, *fsr, orb_cutoff, solver); + + EXPECT_EQ(&env.dftu(), dftu); + EXPECT_EQ(&env.ucell(), &ucell); + EXPECT_EQ(&env.gd(), gd); + EXPECT_EQ(&env.pv(), pv); + EXPECT_EQ(&env.fsr(), fsr); + ASSERT_EQ(env.orb_cutoff().size(), 2u); + EXPECT_DOUBLE_EQ(env.orb_cutoff()[0], 5.0); + EXPECT_DOUBLE_EQ(env.orb_cutoff()[1], 6.0); + EXPECT_EQ(env.ks_solver(), "scalapack"); +} + +TEST_F(DftuFsKTest, EnvReflectsExternalMutation) +{ + // The env holds references, not copies: mutating the caller-side + // objects must be visible through the env accessors. + std::vector orb_cutoff = {1.0}; + DftuFsEnv env(*dftu, ucell, *gd, *pv, *fsr, orb_cutoff, "genelpa"); + + orb_cutoff[0] = 2.5; + EXPECT_DOUBLE_EQ(env.orb_cutoff()[0], 2.5); + + ucell.lat0 = 7.0; + EXPECT_DOUBLE_EQ(env.ucell().lat0, 7.0); +} + +} // namespace +} // namespace DFTU_LCAO diff --git a/source/source_lcao/module_dftu/unittests/test_dftu_nao_fs_r.cpp b/source/source_lcao/module_dftu/unittests/test_dftu_nao_fs_r.cpp new file mode 100644 index 00000000000..5dc703ed110 --- /dev/null +++ b/source/source_lcao/module_dftu/unittests/test_dftu_nao_fs_r.cpp @@ -0,0 +1,94 @@ +#include "source_cell/unitcell.h" + +// Minimal mocks to satisfy the linker without pulling in heavy dependencies. +Magnetism::Magnetism() {} +Magnetism::~Magnetism() {} +SepPot::SepPot() {} +SepPot::~SepPot() {} +Sep_Cell::Sep_Cell() noexcept {} +Sep_Cell::~Sep_Cell() noexcept {} +UnitCell::UnitCell() {} +UnitCell::~UnitCell() {} + +#include "source_base/parallel_reduce.h" +namespace Parallel_Reduce +{ +template <> +void reduce_all(double* data, int n) +{ +} +} // namespace Parallel_Reduce + +#include "../dftu_nao_fs_r.h" + +#include "source_base/matrix.h" +#include "gtest/gtest.h" + +#include +#include + +namespace DFTU_LCAO +{ +namespace +{ + +class DftuFsRTest : public ::testing::Test +{ + protected: + void SetUp() override + { + ucell.lat0 = 2.0; + ucell.omega = 10.0; + } + + UnitCell ucell; +}; + +TEST_F(DftuFsRTest, ReduceForceNspin1) +{ + ModuleBase::matrix force(2, 3); + force(0, 0) = 1.0; force(0, 1) = 2.0; force(0, 2) = 3.0; + force(1, 0) = 4.0; force(1, 1) = 5.0; force(1, 2) = 6.0; + + reduce_force_impl(force, 1); + + // nspin != 4: force *= 2 + EXPECT_DOUBLE_EQ(force(0, 0), 2.0); + EXPECT_DOUBLE_EQ(force(0, 1), 4.0); + EXPECT_DOUBLE_EQ(force(1, 2), 12.0); +} + +TEST_F(DftuFsRTest, ReduceForceNspin4) +{ + ModuleBase::matrix force(2, 3); + force(0, 0) = 1.0; force(0, 1) = 2.0; force(0, 2) = 3.0; + + reduce_force_impl(force, 4); + + // nspin == 4: no scaling + EXPECT_DOUBLE_EQ(force(0, 0), 1.0); + EXPECT_DOUBLE_EQ(force(0, 1), 2.0); +} + +TEST_F(DftuFsRTest, ReduceStressBasic) +{ + std::vector stress_tmp = {1.0, 2.0, 3.0, 4.0, 5.0, 6.0}; + ModuleBase::matrix stress(3, 3); + + reduce_stress_impl(&ucell, stress_tmp, stress); + + // weight = lat0 / omega = 2.0 / 10.0 = 0.2 + const double w = 0.2; + EXPECT_DOUBLE_EQ(stress(0, 0), 1.0 * w); + EXPECT_DOUBLE_EQ(stress(0, 1), 2.0 * w); + EXPECT_DOUBLE_EQ(stress(0, 2), 3.0 * w); + EXPECT_DOUBLE_EQ(stress(1, 0), 2.0 * w); + EXPECT_DOUBLE_EQ(stress(1, 1), 4.0 * w); + EXPECT_DOUBLE_EQ(stress(1, 2), 5.0 * w); + EXPECT_DOUBLE_EQ(stress(2, 0), 3.0 * w); + EXPECT_DOUBLE_EQ(stress(2, 1), 5.0 * w); + EXPECT_DOUBLE_EQ(stress(2, 2), 6.0 * w); +} + +} // namespace +} // namespace DFTU_LCAO diff --git a/source/source_lcao/module_dftu/unittests/test_dftu_nao_fs_reduce.cpp b/source/source_lcao/module_dftu/unittests/test_dftu_nao_fs_reduce.cpp new file mode 100644 index 00000000000..08b31ad60cd --- /dev/null +++ b/source/source_lcao/module_dftu/unittests/test_dftu_nao_fs_reduce.cpp @@ -0,0 +1,167 @@ +#include "source_cell/unitcell.h" + +// Minimal mocks to satisfy the linker without pulling in heavy dependencies. +Magnetism::Magnetism() {} +Magnetism::~Magnetism() {} +SepPot::SepPot() {} +SepPot::~SepPot() {} +Sep_Cell::Sep_Cell() noexcept {} +Sep_Cell::~Sep_Cell() noexcept {} +UnitCell::UnitCell() {} +UnitCell::~UnitCell() {} + +#include "source_base/parallel_reduce.h" +namespace Parallel_Reduce +{ +// Serial stub: single-rank Allreduce is a no-op. +template <> +void reduce_all(double* data, int n) +{ +} +} // namespace Parallel_Reduce + +#include "../dftu_nao_fs_r.h" + +#include "source_base/matrix.h" +#include "gtest/gtest.h" + +#include + +/*********************************************************************** + * Unit tests for dftu_nao_fs_reduce.cpp. + * + * The file provides two post-processing helpers used at the end of the + * real-space force/stress path: + * + * reduce_force_impl(force, nspin) + * - MPI Allreduce over ranks (stubbed no-op here) + * - multiply by 2 for nspin != 4 (spin-degeneracy factor) + * + * reduce_stress_impl(ucell, stress_tmp, stress) + * - MPI Allreduce over ranks (stubbed no-op here) + * - scale by lat0/omega and inflate the 6-component Voigt form + * (xx, xy, xz, yy, yz, zz) into the full 3x3 symmetric tensor + ***********************************************************************/ + +namespace DFTU_LCAO +{ +namespace +{ + +class DftuFsReduceTest : public ::testing::Test +{ + protected: + void SetUp() override + { + ucell.lat0 = 2.0; + ucell.omega = 10.0; + } + + UnitCell ucell; +}; + +// nspin = 1: spin-degenerate, force must be doubled. +TEST_F(DftuFsReduceTest, ReduceForceNspin1Doubles) +{ + ModuleBase::matrix force(2, 3); + force(0, 0) = 1.0; force(0, 1) = -2.0; force(0, 2) = 3.5; + force(1, 0) = 0.0; force(1, 1) = 4.0; force(1, 2) = -1.0; + + reduce_force_impl(force, 1); + + EXPECT_DOUBLE_EQ(force(0, 0), 2.0); + EXPECT_DOUBLE_EQ(force(0, 1), -4.0); + EXPECT_DOUBLE_EQ(force(0, 2), 7.0); + EXPECT_DOUBLE_EQ(force(1, 0), 0.0); + EXPECT_DOUBLE_EQ(force(1, 1), 8.0); + EXPECT_DOUBLE_EQ(force(1, 2), -2.0); +} + +// nspin = 2: also doubled (two explicit spin channels summed). +TEST_F(DftuFsReduceTest, ReduceForceNspin2Doubles) +{ + ModuleBase::matrix force(1, 3); + force(0, 0) = 1.5; + force(0, 1) = -0.5; + force(0, 2) = 2.0; + + reduce_force_impl(force, 2); + + EXPECT_DOUBLE_EQ(force(0, 0), 3.0); + EXPECT_DOUBLE_EQ(force(0, 1), -1.0); + EXPECT_DOUBLE_EQ(force(0, 2), 4.0); +} + +// nspin = 4: non-collinear, no extra scaling. +TEST_F(DftuFsReduceTest, ReduceForceNspin4Unchanged) +{ + ModuleBase::matrix force(1, 3); + force(0, 0) = 1.5; + force(0, 1) = -0.5; + force(0, 2) = 2.0; + + reduce_force_impl(force, 4); + + EXPECT_DOUBLE_EQ(force(0, 0), 1.5); + EXPECT_DOUBLE_EQ(force(0, 1), -0.5); + EXPECT_DOUBLE_EQ(force(0, 2), 2.0); +} + +// Stress: Voigt (xx, xy, xz, yy, yz, zz) -> 3x3 tensor, scaled by lat0/omega. +TEST_F(DftuFsReduceTest, ReduceStressVoigtToTensor) +{ + std::vector stress_tmp = {1.0, 2.0, 3.0, 4.0, 5.0, 6.0}; + ModuleBase::matrix stress(3, 3); + + reduce_stress_impl(&ucell, stress_tmp, stress); + + const double w = ucell.lat0 / ucell.omega; // 0.2 + EXPECT_DOUBLE_EQ(stress(0, 0), 1.0 * w); + EXPECT_DOUBLE_EQ(stress(0, 1), 2.0 * w); + EXPECT_DOUBLE_EQ(stress(0, 2), 3.0 * w); + EXPECT_DOUBLE_EQ(stress(1, 0), 2.0 * w); + EXPECT_DOUBLE_EQ(stress(1, 1), 4.0 * w); + EXPECT_DOUBLE_EQ(stress(1, 2), 5.0 * w); + EXPECT_DOUBLE_EQ(stress(2, 0), 3.0 * w); + EXPECT_DOUBLE_EQ(stress(2, 1), 5.0 * w); + EXPECT_DOUBLE_EQ(stress(2, 2), 6.0 * w); +} + +// Stress: result must be symmetric regardless of input values. +TEST_F(DftuFsReduceTest, ReduceStressSymmetric) +{ + std::vector stress_tmp = {-1.0, 0.5, 2.5, -3.0, 1.5, 0.0}; + ModuleBase::matrix stress(3, 3); + + reduce_stress_impl(&ucell, stress_tmp, stress); + + for (int i = 0; i < 3; i++) + { + for (int j = 0; j < 3; j++) + { + EXPECT_DOUBLE_EQ(stress(i, j), stress(j, i)); + } + } +} + +// Stress: unit cell scaling weight is applied to every component. +TEST_F(DftuFsReduceTest, ReduceStressScalingWeight) +{ + ucell.lat0 = 1.0; + ucell.omega = 4.0; + std::vector stress_tmp = {8.0, 0.0, 0.0, 8.0, 0.0, 8.0}; + ModuleBase::matrix stress(3, 3); + + reduce_stress_impl(&ucell, stress_tmp, stress); + + // weight = 1.0 / 4.0 = 0.25; hydrostatic input stays hydrostatic + EXPECT_DOUBLE_EQ(stress(0, 0), 2.0); + EXPECT_DOUBLE_EQ(stress(1, 1), 2.0); + EXPECT_DOUBLE_EQ(stress(2, 2), 2.0); + EXPECT_DOUBLE_EQ(stress(0, 1), 0.0); + EXPECT_DOUBLE_EQ(stress(0, 2), 0.0); + EXPECT_DOUBLE_EQ(stress(1, 2), 0.0); +} + +} // namespace +} // namespace DFTU_LCAO diff --git a/source/source_lcao/module_dftu/unittests/test_dftu_nao_ijr.cpp b/source/source_lcao/module_dftu/unittests/test_dftu_nao_ijr.cpp new file mode 100644 index 00000000000..d70848daf18 --- /dev/null +++ b/source/source_lcao/module_dftu/unittests/test_dftu_nao_ijr.cpp @@ -0,0 +1,215 @@ +#include "gtest/gtest.h" +#include +#include +#include + +/*********************************************************************** + * Unit tests for the IJR (atom-pair) helper functions in dftu_nao_ijr.h: + * cal_coeff_lambda (spin encoding) and accumulate_hr_for_iat0 (HR assembly). + ***********************************************************************/ + +// ===================================================================== +// 1. cal_coeff_lambda: Lambda coefficient encoding +// Collinear (nspin=2): coeff[0]=lambda_z, coeff[1]=-lambda_z +// Non-collinear (nspin=4): coeff[0]=lambda_z, coeff[1]=lambda_x+i*lambda_y, +// coeff[2]=lambda_x-i*lambda_y, coeff[3]=-lambda_z +// ===================================================================== + +static void cal_coeff_lambda_collinear(const std::vector& lambda, + std::vector& coeff) +{ coeff[0] = lambda[0]; coeff[1] = -lambda[0]; } + +static void cal_coeff_lambda_noncollinear(const std::vector& lambda, + std::vector>& coeff) +{ + coeff[0] = std::complex(lambda[2], 0.0); + coeff[1] = std::complex(lambda[0], lambda[1]); + coeff[2] = std::complex(lambda[0], -lambda[1]); + coeff[3] = std::complex(-lambda[2], 0.0); +} + +class CalCoeffLambdaTest : public ::testing::Test { protected: void SetUp() override {} }; + +TEST_F(CalCoeffLambdaTest, Collinear_PositiveLambdaZ) +{ + std::vector lambda = {2.5}, coeff(2); + cal_coeff_lambda_collinear(lambda, coeff); + EXPECT_DOUBLE_EQ(coeff[0], 2.5); EXPECT_DOUBLE_EQ(coeff[1], -2.5); +} + +TEST_F(CalCoeffLambdaTest, NonCollinear_General) +{ + std::vector lambda = {1.0, 2.0, 3.0}; + std::vector> coeff(4); + cal_coeff_lambda_noncollinear(lambda, coeff); + EXPECT_NEAR(coeff[0].real(), 3.0, 1e-15); + EXPECT_NEAR(coeff[1].real(), 1.0, 1e-15); EXPECT_NEAR(coeff[1].imag(), 2.0, 1e-15); + EXPECT_NEAR(coeff[2].real(), 1.0, 1e-15); EXPECT_NEAR(coeff[2].imag(), -2.0, 1e-15); + EXPECT_NEAR(coeff[3].real(), -3.0, 1e-15); +} + +// ===================================================================== +// 2. accumulate_hr_for_iat0: HR assembly for one Hubbard atom +// ===================================================================== + +#include "../dftu_nao_ijr.h" + +#include +#include + +class AccumulateHrIat0Test : public ::testing::Test +{ + protected: + void SetUp() override + { +#ifdef __MPI + MPI_Comm_size(MPI_COMM_WORLD, &dsize); + MPI_Comm_rank(MPI_COMM_WORLD, &my_rank); +#endif + ucell.ntype = 1; + ucell.nat = nat; + ucell.atoms = atoms_buf; + iat2it_buf.assign(nat, 0); + iat2ia_buf.resize(nat); + ucell.iat2it = iat2it_buf.data(); + ucell.iat2ia = iat2ia_buf.data(); + ucell.atoms[0].tau.resize(nat); + ucell.lat0 = 1.0; + ucell.itia2iat.create(ucell.ntype, nat); + for (int iat = 0; iat < nat; iat++) + { + iat2ia_buf[iat] = iat; + ucell.atoms[0].tau[iat] = ModuleBase::Vector3(0.0, 0.0, 0.0); + ucell.itia2iat(0, iat) = iat; + } + ucell.atoms[0].na = nat; + ucell.atoms[0].nw = nw; + ucell.atoms[0].iw2l = {2, 2}; + ucell.atoms[0].iw2m = {0, 0}; + ucell.atoms[0].iw2n = {0, 0}; + ucell.set_iat2iwt(1); + paraV.reset(new Parallel_Orbitals()); +#ifdef __MPI + paraV->init(nat * nw, nat * nw, nat * nw, MPI_COMM_WORLD); + paraV->set_atomic_trace(ucell.get_iat2iwt(), nat, nat * nw); +#endif + HR.reset(new hamilt::HContainer(ucell, paraV.get())); + std::fill(HR->get_wrapper(), HR->get_wrapper() + HR->get_nnr(), 0.0); + } + + void TearDown() override + { + HR.reset(); + paraV.reset(); + ucell.atoms = nullptr; + ucell.iat2it = nullptr; + ucell.iat2ia = nullptr; + } + + AdjacentAtomInfo make_adjs() const + { + AdjacentAtomInfo adjs; + adjs.adj_num = 1; + adjs.ntype = {0, 0}; + adjs.natom = {0, 1}; + adjs.box = {ModuleBase::Vector3(0, 0, 0), ModuleBase::Vector3(0, 0, 0)}; + return adjs; + } + + DFTU_LCAO::NlmTot make_nlm_tot() const + { + DFTU_LCAO::NlmTot nlm_tot(nat); + for (int iat = 0; iat < nat; ++iat) + { + nlm_tot[iat].resize(2); + for (int ad = 0; ad < 2; ++ad) + { + for (int iw = 0; iw < nw; ++iw) + { + for (int m = 0; m < 5; ++m) + { + nlm_tot[iat][ad][iw * 5 + m] = std::vector(5, 1.0); + } + } + } + } + return nlm_tot; + } + + const int nat = 2; + const int nw = 2; + int dsize = 1; + int my_rank = 0; + UnitCell ucell; + Atom atoms_buf[1]; + std::vector iat2it_buf; + std::vector iat2ia_buf; + std::unique_ptr paraV; + std::unique_ptr> HR; +}; + +TEST_F(AccumulateHrIat0Test, AccumulatesAllPairs) +{ + AdjacentAtomInfo adjs = make_adjs(); + DFTU_LCAO::NlmTot nlm_tot = make_nlm_tot(); + std::vector pot_onsite(25, 0.0); + for (int m = 0; m < 5; ++m) + { + pot_onsite[m * 5 + m] = 1.0; + } + + DFTU_LCAO::accumulate_hr_for_iat0(ucell, HR.get(), nlm_tot, 0, adjs, *paraV, pot_onsite); + + for (int iap = 0; iap < HR->size_atom_pairs(); ++iap) + { + hamilt::AtomPair& tmp = HR->get_atom_pair(iap); + const int iat1 = tmp.get_atom_i(); + const int iat2 = tmp.get_atom_j(); + std::vector indexes1 = paraV->get_indexes_row(iat1); + std::vector indexes2 = paraV->get_indexes_col(iat2); + const int nwt = indexes1.size() * indexes2.size(); + for (int i = 0; i < nwt; ++i) + { + EXPECT_NEAR(tmp.get_pointer(0)[i], 5.0, 1e-12); + } + } +} + +TEST_F(AccumulateHrIat0Test, MissingPairIsSkipped) +{ + AdjacentAtomInfo adjs = make_adjs(); + adjs.box[1] = ModuleBase::Vector3(5, 5, 5); + ASSERT_EQ(HR->find_matrix(0, 1, 5, 5, 5), nullptr); + DFTU_LCAO::NlmTot nlm_tot = make_nlm_tot(); + std::vector pot_onsite(25, 1.0); + + DFTU_LCAO::accumulate_hr_for_iat0(ucell, HR.get(), nlm_tot, 0, adjs, *paraV, pot_onsite); + + for (int iap = 0; iap < HR->size_atom_pairs(); ++iap) + { + hamilt::AtomPair& tmp = HR->get_atom_pair(iap); + const int iat1 = tmp.get_atom_i(); + const int iat2 = tmp.get_atom_j(); + std::vector indexes1 = paraV->get_indexes_row(iat1); + std::vector indexes2 = paraV->get_indexes_col(iat2); + const int nwt = indexes1.size() * indexes2.size(); + const double expected = (iat1 == iat2) ? 25.0 : 0.0; + for (int i = 0; i < nwt; ++i) + { + EXPECT_DOUBLE_EQ(tmp.get_pointer(0)[i], expected); + } + } +} + +int main(int argc, char** argv) +{ +#ifdef __MPI + MPI_Init(&argc, &argv); +#endif + testing::InitGoogleTest(&argc, argv); + int result = RUN_ALL_TESTS(); +#ifdef __MPI + MPI_Finalize(); +#endif + return result; +} diff --git a/source/source_lcao/module_dftu/unittests/test_dftu_nao_op.cpp b/source/source_lcao/module_dftu/unittests/test_dftu_nao_op.cpp new file mode 100644 index 00000000000..0a87a8a1578 --- /dev/null +++ b/source/source_lcao/module_dftu/unittests/test_dftu_nao_op.cpp @@ -0,0 +1,249 @@ +#include "gtest/gtest.h" +#include + +// mock of DFTU +#include "../dftu_nao_op.h" +#include "source_basis/module_nao/two_center_integrator.h" +#include "source_cell/unitcell.h" +#include "source_pw/module_pwdft/dftu_base.h" +#include "source_estate/module_dm/density_matrix.h" + +Plus_U_Base dftu; + +//--------------------------------------- +// Unit test of DFTU_onsite operator (dftu_nao_op.cpp). +// Tests constructHR for d2d and d2cd variants. +//--------------------------------------- + +int test_size = 5; +int test_nw = 5; + +class DFTUTest : public ::testing::Test +{ + protected: + void SetUp() override + { +#ifdef __MPI + MPI_Comm_size(MPI_COMM_WORLD, &dsize); + MPI_Comm_rank(MPI_COMM_WORLD, &my_rank); +#endif + + ucell.ntype = 1; + ucell.nat = test_size; + ucell.atoms = new Atom[ucell.ntype]; + ucell.iat2it = new int[ucell.nat]; + ucell.iat2ia = new int[ucell.nat]; + ucell.atoms[0].tau.resize(ucell.nat); + ucell.lat0 = 1.0; + ucell.itia2iat.create(ucell.ntype, ucell.nat); + for (int iat = 0; iat < ucell.nat; iat++) + { + ucell.iat2it[iat] = 0; + ucell.iat2ia[iat] = iat; + ucell.atoms[0].tau[iat] = ModuleBase::Vector3(0.0, 0.0, 0.0); + ucell.itia2iat(0, iat) = iat; + } + ucell.atoms[0].na = test_size; + ucell.atoms[0].nw = test_nw; + ucell.atoms[0].iw2l.resize(test_nw); + ucell.atoms[0].iw2m.resize(test_nw); + ucell.atoms[0].iw2n.resize(test_nw); + for (int iw = 0; iw < test_nw; ++iw) + { + ucell.atoms[0].iw2l[iw] = 2; + ucell.atoms[0].iw2m[iw] = 0; + ucell.atoms[0].iw2n[iw] = 0; + } + ucell.set_iat2iwt(1); + init_parav(); + HR = new hamilt::HContainer(ucell, paraV); + + dftu.occmat().data().resize(test_size); + for (int iat = 0; iat < test_size; iat++) + { + dftu.occmat().data()[iat].resize(3); + for (int l = 0; l < 3; l++) + { + dftu.occmat().data()[iat][l].resize(2); + dftu.occmat().data()[iat][l][0].create(2 * l + 1, 2 * l + 1); + dftu.occmat().data()[iat][l][1].create(2 * l + 1, 2 * l + 1); + } + } + dftu.u_current = {U_test}; + dftu.l_channel = {orbital_c_test}; + } + + void TearDown() override + { + delete HR; + delete paraV; + delete[] ucell.atoms; + } + + double occ_mat_c(int iat, int spin, int icc) const + { + return dftu.occmat().data()[iat][2][spin].c[icc]; + } + +#ifdef __MPI + void init_parav() + { + int nb = 10; + int global_row = test_size * test_nw; + int global_col = test_size * test_nw; + std::ofstream ofs_running; + paraV = new Parallel_Orbitals(); + paraV->init(global_row, global_col, nb, MPI_COMM_WORLD); + paraV->set_atomic_trace(ucell.get_iat2iwt(), test_size, global_row); + } +#else + void init_parav() + { + } +#endif + + UnitCell ucell; + hamilt::HContainer* HR; + Parallel_Orbitals* paraV; + TwoCenterIntegrator intor_; + + int dsize; + int my_rank = 0; + double U_test = 1.0; + int orbital_c_test = 2; + double onsite_radius_test = 1.0; +}; + +TEST_F(DFTUTest, constructHRd2d) +{ + const int nspin = 1; + std::vector> kvec_d_in(1, ModuleBase::Vector3(0.0, 0.0, 0.0)); + hamilt::HS_Matrix_K hsk(paraV, true); + hsk.set_zero_hk(); + Grid_Driver gd(0, 0); + const double factor = 1.0 / test_nw / test_nw / test_size / test_size; + module_dm::DensityMatrix dm(paraV, 1); + dm.init_dmr(*HR); + for (int i = 0; i < paraV->nrow; i++) + { + for (int j = 0; j < paraV->ncol; j++) + { + dm.set_dmk(1, 0, i, j, factor); + } + } + dm.cal_dmr(-1); + // reset HR + for (int i = 0; i < HR->get_nnr(); i++) + { + HR->get_wrapper()[i] = 0.0; + } + hamilt::DFTU_onsite> + op(&hsk, kvec_d_in, HR, ucell, &gd, &intor_, {1.0}, &dftu, nspin, onsite_radius_test, &dm); + op.contributeHR(); + for (int iat = 0; iat < test_size; iat++) + { + for (int icc = 0; icc < 25; icc++) + { + EXPECT_NEAR(occ_mat_c(iat, 0, icc), 0.5, 1e-10); + } + } + for (int iap = 0; iap < HR->size_atom_pairs(); ++iap) + { + hamilt::AtomPair& tmp = HR->get_atom_pair(iap); + int iat1 = tmp.get_atom_i(); + int iat2 = tmp.get_atom_j(); + std::vector indexes1 = paraV->get_indexes_row(iat1); + std::vector indexes2 = paraV->get_indexes_col(iat2); + int nwt = indexes1.size() * indexes2.size(); + for (int i = 0; i < nwt; ++i) + { + EXPECT_NEAR(tmp.get_pointer(0)[i], -10.0 * test_size, 1e-10); + } + } + op.contributeHk(0); + double* hk = hsk.get_hk(); + for (int i = 0; i < paraV->get_row_size() * paraV->get_col_size(); ++i) + { + EXPECT_NEAR(hk[i], -10.0 * test_size, 1e-10); + } +} + +TEST_F(DFTUTest, constructHRd2cd) +{ + const int nspin = 2; + std::vector> kvec_d_in(2, ModuleBase::Vector3(0.0, 0.0, 0.0)); + hamilt::HS_Matrix_K> hsk(paraV, true); + hsk.set_zero_hk(); + Grid_Driver gd(0, 0); + const double factor = 0.5 / test_nw / test_nw / test_size / test_size; + std::vector> kvec_d_dm(1, ModuleBase::Vector3(0.0, 0.0, 0.0)); + module_dm::DensityMatrix, double> dm(paraV, 2, kvec_d_dm, 1); + dm.init_dmr(*HR); + for (int is = 1; is <= 2; ++is) + { + for (int i = 0; i < paraV->nrow; i++) + { + for (int j = 0; j < paraV->ncol; j++) + { + dm.set_dmk(is, 0, i, j, std::complex(factor, 0.0)); + } + } + } + dm.cal_dmr(-1); + // reset HR + for (int i = 0; i < HR->get_nnr(); i++) + { + HR->get_wrapper()[i] = 0.0; + } + hamilt::DFTU_onsite, double>> + op(&hsk, kvec_d_in, HR, ucell, &gd, &intor_, {1.0}, &dftu, nspin, onsite_radius_test, &dm); + op.contributeHR(); + for (int iat = 0; iat < test_size; iat++) + { + for (int icc = 0; icc < 25; icc++) + { + EXPECT_NEAR(occ_mat_c(iat, 0, icc), 0.5, 1e-10); + } + } + for (int iap = 0; iap < HR->size_atom_pairs(); ++iap) + { + hamilt::AtomPair& tmp = HR->get_atom_pair(iap); + int iat1 = tmp.get_atom_i(); + int iat2 = tmp.get_atom_j(); + std::vector indexes1 = paraV->get_indexes_row(iat1); + std::vector indexes2 = paraV->get_indexes_col(iat2); + int nwt = indexes1.size() * indexes2.size(); + for (int i = 0; i < nwt; ++i) + { + EXPECT_NEAR(tmp.get_pointer(0)[i], -10.0 * test_size, 1e-10); + } + } + op.contributeHk(0); + std::complex* hk = hsk.get_hk(); + for (int i = 0; i < paraV->get_row_size() * paraV->get_col_size(); ++i) + { + EXPECT_NEAR(hk[i].real(), -10.0 * test_size, 1e-10); + EXPECT_NEAR(hk[i].imag(), 0.0, 1e-10); + } + op.contributeHR(); + for (int iat = 0; iat < test_size; iat++) + { + for (int icc = 0; icc < 25; icc++) + { + EXPECT_NEAR(occ_mat_c(iat, 1, icc), 0.5, 1e-10); + } + } +} + +int main(int argc, char** argv) +{ +#ifdef __MPI + MPI_Init(&argc, &argv); +#endif + testing::InitGoogleTest(&argc, argv); + int result = RUN_ALL_TESTS(); +#ifdef __MPI + MPI_Finalize(); +#endif + return result; +} diff --git a/source/source_lcao/module_dftu/unittests/test_dftu_nao_op_legacy.cpp b/source/source_lcao/module_dftu/unittests/test_dftu_nao_op_legacy.cpp new file mode 100644 index 00000000000..d363377ade4 --- /dev/null +++ b/source/source_lcao/module_dftu/unittests/test_dftu_nao_op_legacy.cpp @@ -0,0 +1,149 @@ +#include "gtest/gtest.h" +#include +#include +#include + +/*********************************************************************** + * Unit tests for DFT+U on-site potential and Pauli transfer + * (dftu_nao_op_legacy.cpp). + * + * cal_pot_onsite: nspin=1,2: pot_onsite[is] = U*(0.5*delta - occ^T) + * nspin=4: pot_onsite[0] = U*(1.0*delta - occ^T), + * pot_onsite[is>0] = -U*occ^T + * transfer_pot_onsite: Pauli matrix transformation (nspin=4) + ***********************************************************************/ + +// ===================================================================== +// 1. cal_pot_onsite: Hubbard potential calculation +// ===================================================================== + +static void cal_pot_onsite(const std::vector& occ, int m_size, double u_value, + std::vector& pot_onsite, double& eu) +{ + pot_onsite.assign(occ.size(), 0.0); + eu = 0.0; + int spin_fold = occ.size() / m_size / m_size; + if (spin_fold < 4) // nspin=1,2 + { + for (int is = 0; is < spin_fold; ++is) + { + int start = is * m_size * m_size; + for (int m1 = 0; m1 < m_size; m1++) + for (int m2 = 0; m2 < m_size; m2++) + { + pot_onsite[start + m1 * m_size + m2] = u_value * (0.5 * (m1 == m2) - occ[start + m2 * m_size + m1]); + eu += u_value * 0.5 * occ[start + m2 * m_size + m1] * occ[start + m1 * m_size + m2]; + } + } + } + else // nspin=4 + { + for (int m1 = 0; m1 < m_size; m1++) + for (int m2 = 0; m2 < m_size; m2++) + { + pot_onsite[m1 * m_size + m2] = u_value * (1.0 * (m1 == m2) - occ[m2 * m_size + m1]); + eu += u_value * 0.25 * occ[m2 * m_size + m1] * occ[m1 * m_size + m2]; + } + for (int is = 1; is < spin_fold; ++is) + { + int start = is * m_size * m_size; + for (int m1 = 0; m1 < m_size; m1++) + for (int m2 = 0; m2 < m_size; m2++) + { + pot_onsite[start + m1 * m_size + m2] = u_value * (0.0 - occ[start + m2 * m_size + m1]); + eu += u_value * 0.25 * occ[start + m2 * m_size + m1] * occ[start + m1 * m_size + m2]; + } + } + } +} + +class CalVOfUTest : public ::testing::Test { protected: void SetUp() override {} }; + +TEST_F(CalVOfUTest, Nspin1_SingleOrbital_HalfFilled) +{ + std::vector occ = {0.5}; + std::vector pot_onsite; double eu = 0.0; + cal_pot_onsite(occ, 1, 4.0, pot_onsite, eu); + EXPECT_DOUBLE_EQ(pot_onsite[0], 0.0); + EXPECT_DOUBLE_EQ(eu, 0.5); +} + +TEST_F(CalVOfUTest, Nspin2_DOrbital_SpinPolarized) +{ + const int m_size = 5; + std::vector occ(m_size * m_size * 2, 0.0); + for (int m = 0; m < m_size; m++) occ[m * m_size + m] = 0.8; + for (int m = 0; m < m_size; m++) occ[m_size*m_size + m*m_size + m] = 0.2; + std::vector pot_onsite; double eu = 0.0; + cal_pot_onsite(occ, m_size, 5.0, pot_onsite, eu); + for (int m = 0; m < m_size; m++) EXPECT_NEAR(pot_onsite[m*m_size+m], -1.5, 1e-14); + for (int m = 0; m < m_size; m++) EXPECT_NEAR(pot_onsite[m_size*m_size+m*m_size+m], 1.5, 1e-14); + EXPECT_NEAR(eu, 8.5, 1e-14); +} + +TEST_F(CalVOfUTest, Nspin4_Porbital_PauliBlocks) +{ + const int m_size = 3; + std::vector occ(m_size * m_size * 4, 0.0); + for (int m = 0; m < m_size; m++) occ[m * m_size + m] = 0.5; + std::vector pot_onsite; double eu = 0.0; + cal_pot_onsite(occ, m_size, 4.0, pot_onsite, eu); + for (int m = 0; m < m_size; m++) EXPECT_NEAR(pot_onsite[m*m_size+m], 2.0, 1e-14); + for (int is = 1; is < 4; is++) + for (int i = 0; i < m_size*m_size; i++) EXPECT_NEAR(pot_onsite[is*m_size*m_size+i], 0.0, 1e-14); + EXPECT_NEAR(eu, 0.75, 1e-14); +} + +// ===================================================================== +// 2. transfer_pot_onsite: Pauli matrix transformation (nspin=4) +// ===================================================================== + +static void transfer_pot_onsite(const std::vector& pot_onsite_tmp, + std::vector>& pot_onsite) +{ + const int m_size = int(sqrt(pot_onsite_tmp.size()) / 2); + const int m_size2 = m_size * m_size; + pot_onsite.resize(pot_onsite_tmp.size()); + for (int m1 = 0; m1 < m_size; m1++) + for (int m2 = 0; m2 < m_size; m2++) + { + int idx[4] = {m1*m_size+m2, m1*m_size+m2+m_size2, m2*m_size+m1+2*m_size2, m2*m_size+m1+3*m_size2}; + pot_onsite[idx[0]] = 0.5 * (pot_onsite_tmp[idx[0]] + pot_onsite_tmp[idx[3]]); + pot_onsite[idx[3]] = 0.5 * (pot_onsite_tmp[idx[0]] - pot_onsite_tmp[idx[3]]); + pot_onsite[idx[1]] = 0.5 * (pot_onsite_tmp[idx[1]] + std::complex(0,1) * pot_onsite_tmp[idx[2]]); + pot_onsite[idx[2]] = 0.5 * (pot_onsite_tmp[idx[1]] - std::complex(0,1) * pot_onsite_tmp[idx[2]]); + } +} + +class Transferpot_onsiteTest : public ::testing::Test { protected: void SetUp() override {} }; + +TEST_F(Transferpot_onsiteTest, PauliI_IdentityInput) +{ + std::vector pot_onsite_tmp = {1.0, 0.0, 0.0, 1.0}; + std::vector> pot_onsite; + transfer_pot_onsite(pot_onsite_tmp, pot_onsite); + EXPECT_NEAR(pot_onsite[0].real(), 1.0, 1e-15); EXPECT_NEAR(pot_onsite[0].imag(), 0.0, 1e-15); + EXPECT_NEAR(pot_onsite[3].real(), 0.0, 1e-15); +} + +TEST_F(Transferpot_onsiteTest, PureSigmaZ) +{ + std::vector pot_onsite_tmp = {1.0, 0.0, 0.0, -1.0}; + std::vector> pot_onsite; + transfer_pot_onsite(pot_onsite_tmp, pot_onsite); + EXPECT_NEAR(pot_onsite[0].real(), 0.0, 1e-15); EXPECT_NEAR(pot_onsite[3].real(), 1.0, 1e-15); +} + +TEST_F(Transferpot_onsiteTest, SigmaX_Y_Combined) +{ + std::vector pot_onsite_tmp_x = {0.0, 1.0, 1.0, 0.0}; + std::vector> pot_onsite; + transfer_pot_onsite(pot_onsite_tmp_x, pot_onsite); + EXPECT_NEAR(pot_onsite[1].real(), 0.5, 1e-15); EXPECT_NEAR(pot_onsite[1].imag(), 0.5, 1e-15); + EXPECT_NEAR(pot_onsite[2].real(), 0.5, 1e-15); EXPECT_NEAR(pot_onsite[2].imag(), -0.5, 1e-15); + + std::vector pot_onsite_tmp_y = {0.0, 1.0, -1.0, 0.0}; + transfer_pot_onsite(pot_onsite_tmp_y, pot_onsite); + EXPECT_NEAR(pot_onsite[1].real(), 0.5, 1e-15); EXPECT_NEAR(pot_onsite[1].imag(), -0.5, 1e-15); + EXPECT_NEAR(pot_onsite[2].real(), 0.5, 1e-15); EXPECT_NEAR(pot_onsite[2].imag(), 0.5, 1e-15); +} diff --git a/source/source_lcao/module_dftu/unittests/test_dftu_nao_pots.cpp b/source/source_lcao/module_dftu/unittests/test_dftu_nao_pots.cpp new file mode 100644 index 00000000000..299d56f2a1e --- /dev/null +++ b/source/source_lcao/module_dftu/unittests/test_dftu_nao_pots.cpp @@ -0,0 +1,48 @@ +#include "gtest/gtest.h" +#include +#include + +/*********************************************************************** + * Unit tests for the DFT+U on-site potential formula (dftu_nao_pots.cpp). + * + * pot_onsite[m0,m1] = U * (0.5*delta(m0,m1) - occ_mat[m0,m1]) (diagonal) + * pot_onsite[m0,m1] = -U * occ_mat[m0,m1] (off-diagonal) + ***********************************************************************/ + +static double compute_pot_onsite(double U_val, int m0, int m1, double occ_mat_val) +{ + if (m0 == m1) + return U_val * (0.5 - occ_mat_val); + else + return -U_val * occ_mat_val; +} + +class PotOnsitePotentialTest : public ::testing::Test +{ + protected: + void SetUp() override {} +}; + +TEST_F(PotOnsitePotentialTest, Diagonal_HalfFilled) +{ + double U = 4.0; + double occ_mat = 0.5; // half-filled + double pot_onsite = compute_pot_onsite(U, 0, 0, occ_mat); + EXPECT_DOUBLE_EQ(pot_onsite, 0.0); // U * (0.5 - 0.5) = 0 +} + +TEST_F(PotOnsitePotentialTest, Diagonal_FullyOccupied) +{ + double U = 4.0; + double occ_mat = 1.0; // fully occupied + double pot_onsite = compute_pot_onsite(U, 0, 0, occ_mat); + EXPECT_DOUBLE_EQ(pot_onsite, -2.0); // U * (0.5 - 1.0) = -2.0 +} + +TEST_F(PotOnsitePotentialTest, OffDiagonal) +{ + double U = 5.0; + double occ_mat = 0.3; + double pot_onsite = compute_pot_onsite(U, 0, 1, occ_mat); + EXPECT_DOUBLE_EQ(pot_onsite, -1.5); // -U * occ_mat = -1.5 +} diff --git a/source/source_lcao/module_dftu/unittests/test_dftu_nao_str_r.cpp b/source/source_lcao/module_dftu/unittests/test_dftu_nao_str_r.cpp new file mode 100644 index 00000000000..b2df986d3b5 --- /dev/null +++ b/source/source_lcao/module_dftu/unittests/test_dftu_nao_str_r.cpp @@ -0,0 +1,84 @@ +#include "gtest/gtest.h" +#include +#include + +/*********************************************************************** + * Unit tests for the stress accumulation in dftu_nao_str_r.cpp. + * + * stress[0]+=pot_onsite*DM*(nlm1_dx*dis1.x*nlm2_val+nlm1_val*nlm2_dx*dis2.x) + * stress[3]+=pot_onsite*DM*(nlm1_dy*dis1.y*nlm2_val+nlm1_val*nlm2_dy*dis2.y) + * stress[5]+=pot_onsite*DM*(nlm1_dz*dis1.z*nlm2_val+nlm1_val*nlm2_dz*dis2.z) + * + * Also covers the Voigt -> 3x3 matrix mapping used by the stress output. + ***********************************************************************/ + +// ===================================================================== +// 1. Stress/IJR core loop +// ===================================================================== + +static void cal_stress_IJR_core(const std::vector& pot_onsite_in, + const std::vector& nlm1, const std::vector& nlm2, + const double dm_val, int m_size, int nspin, + double dis1[3], double dis2[3], double stress[6]) +{ + const int m_size2 = m_size * m_size; + for (int is = 0; is < nspin; is++) + for (int m1 = 0; m1 < m_size; m1++) + for (int m2 = 0; m2 < m_size; m2++) + { + double tmp = pot_onsite_in[m1*m_size+m2+is*m_size2] * dm_val; + stress[0] += tmp*(nlm1[m1+m_size]*dis1[0]*nlm2[m2] + nlm1[m1]*nlm2[m2+m_size]*dis2[0]); + stress[1] += tmp*(nlm1[m1+m_size]*dis1[1]*nlm2[m2] + nlm1[m1]*nlm2[m2+m_size]*dis2[1]); + stress[2] += tmp*(nlm1[m1+m_size]*dis1[2]*nlm2[m2] + nlm1[m1]*nlm2[m2+m_size]*dis2[2]); + stress[3] += tmp*(nlm1[m1+2*m_size]*dis1[1]*nlm2[m2] + nlm1[m1]*nlm2[m2+2*m_size]*dis2[1]); + stress[4] += tmp*(nlm1[m1+2*m_size]*dis1[2]*nlm2[m2] + nlm1[m1]*nlm2[m2+2*m_size]*dis2[2]); + stress[5] += tmp*(nlm1[m1+3*m_size]*dis1[2]*nlm2[m2] + nlm1[m1]*nlm2[m2+3*m_size]*dis2[2]); + } +} + +class StressIJRTest : public ::testing::Test { protected: void SetUp() override {} }; + +TEST_F(StressIJRTest, SingleOrbital_XDisplacement) +{ + std::vector pot_onsite = {1.0}, nlm1 = {1.0, 0.1, 0.0, 0.0}, nlm2 = {1.0, 0.2, 0.0, 0.0}; + double dm_val = 1.0, dis1[3] = {1.0, 0.0, 0.0}, dis2[3] = {-1.0, 0.0, 0.0}, stress[6] = {0.0}; + cal_stress_IJR_core(pot_onsite, nlm1, nlm2, dm_val, 1, 1, dis1, dis2, stress); + EXPECT_NEAR(stress[0], -0.1, 1e-15); + EXPECT_NEAR(stress[1], 0.0, 1e-15); EXPECT_NEAR(stress[2], 0.0, 1e-15); +} + +TEST_F(StressIJRTest, SymmetricDisplacement) +{ + std::vector pot_onsite = {2.0}, nlm1 = {1.0, 0.1, 0.2, 0.3}, nlm2 = {1.0, 0.1, 0.2, 0.3}; + double dm_val = 1.0, dis1[3] = {1.0, 2.0, 3.0}, dis2[3] = {1.0, 2.0, 3.0}, stress[6] = {0.0}; + cal_stress_IJR_core(pot_onsite, nlm1, nlm2, dm_val, 1, 1, dis1, dis2, stress); + EXPECT_NEAR(stress[0], 2.0*(0.1*1.0 + 1.0*0.1*1.0), 1e-15); // xx + EXPECT_NEAR(stress[4], 2.0*(0.2*3.0 + 1.0*0.2*3.0), 1e-15); // yz +} + +// ===================================================================== +// 2. Stress Voigt -> matrix mapping +// ===================================================================== + +static void voigt_to_matrix(double stress_6[6], double matrix[9]) +{ + for (int i = 0; i < 9; i++) matrix[i] = 0.0; + matrix[0]=stress_6[0]; matrix[1]=stress_6[1]; matrix[2]=stress_6[2]; + matrix[3]=stress_6[1]; matrix[4]=stress_6[4]; matrix[5]=stress_6[3]; + matrix[6]=stress_6[2]; matrix[7]=stress_6[3]; matrix[8]=stress_6[5]; +} + +class VoigtToMatrixTest : public ::testing::Test { protected: void SetUp() override {} }; + +TEST_F(VoigtToMatrixTest, FullMappingAndSymmetry) +{ + double stress_6[6] = {1.0, 2.0, 3.0, 4.0, 5.0, 6.0}, matrix[9]; + voigt_to_matrix(stress_6, matrix); + EXPECT_NEAR(matrix[0], 1.0, 1e-15); EXPECT_NEAR(matrix[1], 2.0, 1e-15); + EXPECT_NEAR(matrix[2], 3.0, 1e-15); EXPECT_NEAR(matrix[3], 2.0, 1e-15); + EXPECT_NEAR(matrix[4], 5.0, 1e-15); EXPECT_NEAR(matrix[5], 4.0, 1e-15); + EXPECT_NEAR(matrix[6], 3.0, 1e-15); EXPECT_NEAR(matrix[7], 4.0, 1e-15); + EXPECT_NEAR(matrix[8], 6.0, 1e-15); + EXPECT_NEAR(matrix[1], matrix[3], 1e-15); EXPECT_NEAR(matrix[2], matrix[6], 1e-15); + EXPECT_NEAR(matrix[5], matrix[7], 1e-15); +} diff --git a/source/source_lcao/module_lr/CMakeLists.txt b/source/source_lcao/module_lr/CMakeLists.txt index cfcd3bb7c8b..d57d02afc4f 100644 --- a/source/source_lcao/module_lr/CMakeLists.txt +++ b/source/source_lcao/module_lr/CMakeLists.txt @@ -8,7 +8,6 @@ if(ENABLE_LCAO) utils/lr_util.cpp utils/lr_util_hcontainer.cpp utils/lr_io.cpp - utils/lr_io_krlist.cpp utils/exciton_plotter.cpp ao_to_mo_transformer/ao_to_mo_parallel.cpp ao_to_mo_transformer/ao_to_mo_serial.cpp @@ -22,6 +21,15 @@ if(ENABLE_LCAO) hamilt_casida.cpp potentials/xc_kernel.cpp) + # BSE-related code: only compiled with LibRI (__EXX), all its + # consumers (ESolver_BSE, RI benchmark in hamilt_casida.h) are + # already guarded by __EXX + if(ENABLE_LIBRI) + list(APPEND objects + utils/lr_io_krlist.cpp + ) + endif() + add_library( lr OBJECT diff --git a/source/source_lcao/module_lr/ao_to_mo_transformer/ao_to_mo.h b/source/source_lcao/module_lr/ao_to_mo_transformer/ao_to_mo.h index 0cfcba9348c..cc812f4b1d9 100644 --- a/source/source_lcao/module_lr/ao_to_mo_transformer/ao_to_mo.h +++ b/source/source_lcao/module_lr/ao_to_mo_transformer/ao_to_mo.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_LCAO_MODULE_LR_AO_TO_MO_TRANSFORMER_AO_TO_MO_H +#define ABACUS_SOURCE_LCAO_MODULE_LR_AO_TO_MO_TRANSFORMER_AO_TO_MO_H + #include #include "source_psi/psi.h" #include "source_lcao/module_lr/utils/mo_type.h" @@ -41,4 +43,6 @@ namespace LR const bool add_on = true, const LR_Util::MO_TYPE type = LR_Util::VO); #endif -} \ No newline at end of file +} + +#endif // ABACUS_SOURCE_LCAO_MODULE_LR_AO_TO_MO_TRANSFORMER_AO_TO_MO_H diff --git a/source/source_lcao/module_lr/dm_trans/dm_trans.h b/source/source_lcao/module_lr/dm_trans/dm_trans.h index 36bbd308772..3f4c81d82e4 100644 --- a/source/source_lcao/module_lr/dm_trans/dm_trans.h +++ b/source/source_lcao/module_lr/dm_trans/dm_trans.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_LCAO_MODULE_LR_DM_TRANS_DM_TRANS_H +#define ABACUS_SOURCE_LCAO_MODULE_LR_DM_TRANS_DM_TRANS_H + // use tensor or basematrix in the future #include #include "source_psi/psi.h" @@ -45,4 +47,6 @@ namespace LR const int& nocc, const int& nvirt, const T factor = (T)1.0, const LR_Util::MO_TYPE type = LR_Util::MO_TYPE::VO); -} \ No newline at end of file +} + +#endif // ABACUS_SOURCE_LCAO_MODULE_LR_DM_TRANS_DM_TRANS_H diff --git a/source/source_lcao/module_lr/dm_trans/dmr_complex.cpp b/source/source_lcao/module_lr/dm_trans/dmr_complex.cpp deleted file mode 100644 index 0b65bc610d8..00000000000 --- a/source/source_lcao/module_lr/dm_trans/dmr_complex.cpp +++ /dev/null @@ -1,82 +0,0 @@ -#include "source_estate/module_dm/density_matrix.h" -#include "source_base/timer.h" -#include "source_io/module_parameter/parameter.h" -#include "source_base/libm/libm.h" -namespace elecstate -{ - template<> - void DensityMatrix, std::complex>::cal_DMR(int ik_in) - { - ModuleBase::TITLE("DensityMatrix", "cal_DMR"); - ModuleBase::timer::start("DensityMatrix", "cal_DMR"); - for (int is = 1; is <= this->_nspin; ++is) - { - const int ik_begin = this->_nk * (is - 1); // jump this->_nk for spin_down if nspin==2 - hamilt::HContainer>*const tmp_DMR = this->_DMR[is - 1]; - // set zero since this function is called in every scf step - tmp_DMR->set_zero(); -#ifdef _OPENMP -#pragma omp parallel for -#endif - for (int i = 0; i < tmp_DMR->size_atom_pairs(); ++i) - { - hamilt::AtomPair>& tmp_ap = tmp_DMR->get_atom_pair(i); - const int iat1 = tmp_ap.get_atom_i(); - const int iat2 = tmp_ap.get_atom_j(); - // get global indexes of whole matrix for each atom in this process - const int row_ap = this->_paraV->atom_begin_row[iat1]; - const int col_ap = this->_paraV->atom_begin_col[iat2]; - assert(row_ap != -1 && col_ap != -1 && "Atom-pair not belong this process"); - for (int ir = 0; ir < tmp_ap.get_R_size(); ++ir) - { - const ModuleBase::Vector3 r_index = tmp_ap.get_R_index(ir); - hamilt::BaseMatrix>*const tmp_matrix = tmp_ap.find_matrix(r_index); -#ifdef __DEBUG - if (tmp_matrix == nullptr) - { - std::cout << "tmp_matrix is nullptr" << std::endl; - continue; - } -#endif - // loop over k-points - if (PARAM.inp.nspin != 4) { - for (int ik = 0; ik < this->_nk; ++ik) - { - if (ik_in >= 0 && ik_in != ik) continue; - // cal k_phase - // if TK==std::complex, kphase is e^{ikR} - const ModuleBase::Vector3 dR(r_index[0], r_index[1], r_index[2]); - const double arg = (this->_kvec_d[ik] * dR) * ModuleBase::TWO_PI; - double sinp = 0.0, cosp = 0.0; - ModuleBase::libm::sincos(arg, &sinp, &cosp); - const std::complex kphase = std::complex(cosp, sinp); - // set DMR element - std::complex* tmp_DMR_pointer = tmp_matrix->get_pointer(); - const std::complex* tmp_DMK_pointer - = this->_DMK[ik + ik_begin].data() - + col_ap * this->_paraV->nrow + row_ap; - // jump DMK to fill DMR - // DMR is row-major, DMK is column-major - for (int mu = 0; mu < this->_paraV->get_nrow_atom(iat1); ++mu) - { - BlasConnector::axpy(this->_paraV->get_ncol_atom(iat2), - kphase, - tmp_DMK_pointer, - this->_paraV->get_row_size(), - tmp_DMR_pointer, - 1); - tmp_DMK_pointer += 1; - tmp_DMR_pointer += this->_paraV->get_ncol_atom(iat2); - } - } - } - // treat DMR as pauli matrix when NSPIN=4 - if (PARAM.inp.nspin == 4) - { throw std::runtime_error("complex DM(R) with NSPIN=4 is not implemented yet"); } - } - } - } - ModuleBase::timer::end("DensityMatrix", "cal_DMR"); - } - // template class DensityMatrix, std::complex>; -} \ No newline at end of file diff --git a/source/source_lcao/module_lr/hamilt_casida.h b/source/source_lcao/module_lr/hamilt_casida.h index 4bf2ea98d0b..b17a13fdc45 100644 --- a/source/source_lcao/module_lr/hamilt_casida.h +++ b/source/source_lcao/module_lr/hamilt_casida.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_LCAO_MODULE_LR_HAMILT_CASIDA_H +#define ABACUS_SOURCE_LCAO_MODULE_LR_HAMILT_CASIDA_H + #include #include "source_hamilt/hamilt.h" #include "source_estate/module_dm/density_matrix.h" @@ -47,9 +49,9 @@ namespace LR ModuleBase::TITLE("HamiltLR", "HamiltLR"); if (ri_hartree_benchmark != "aims" && ri_hartree_benchmark !="aims-librpa") { assert(aims_nbasis.empty()); } // always use nspin=1 for transition density matrix - this->DM_trans = LR_Util::make_unique>(&pmat_in, 1, kv_in.kvec_d, nk); + this->DM_trans = LR_Util::make_unique>(&pmat_in, 1, kv_in.kvec_d, nk); if (ri_hartree_benchmark == "none") { LR_Util::initialize_DMR(*this->DM_trans, pmat_in, ucell_in, gd_in, orb_cutoff); } - // this->DM_trans->init_DMR(&gd_in, &ucell_in); // too large due to not restricted by orb_cutoff + // this->DM_trans->init_dmr(&gd_in, &ucell_in); // too large due to not restricted by orb_cutoff // 1.add the diag operator (the first one) this->ops = new OperatorLRDiag(eig_ks.c, pX[0], nk, nocc[0], nvirt[0]); @@ -141,7 +143,7 @@ namespace LR #endif // LR_Util::print_tensor(dm_trans_2d[0], "dm_trans_2d[0]", &pmat_in); // tensor to vector, then set DMK - for (int ik = 0;ik < nk;++ik) { this->DM_trans->set_DMK_pointer(ik, dm_trans_2d[ik].data()); } + for (int ik = 0;ik < nk;++ik) { this->DM_trans->set_dmk_ptr(ik, dm_trans_2d[ik].data()); } }; } ~HamiltLR() { delete this->ops; } @@ -198,7 +200,7 @@ namespace LR T one()const; /// transition density matrix in AO representation /// calculate on the same address for each bands, and commonly used by all the operators - std::unique_ptr> DM_trans; + std::unique_ptr> DM_trans; /// first node operator, add operations from each operators hamilt::Operator* ops = nullptr; @@ -206,3 +208,5 @@ namespace LR std::function cal_dm_trans; }; } + +#endif // ABACUS_SOURCE_LCAO_MODULE_LR_HAMILT_CASIDA_H diff --git a/source/source_lcao/module_lr/hamilt_ulr.hpp b/source/source_lcao/module_lr/hamilt_ulr.hpp index 38d77e753bb..b9b88cd42d6 100644 --- a/source/source_lcao/module_lr/hamilt_ulr.hpp +++ b/source/source_lcao/module_lr/hamilt_ulr.hpp @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_LCAO_MODULE_LR_HAMILT_ULR_HPP +#define ABACUS_SOURCE_LCAO_MODULE_LR_HAMILT_ULR_HPP + #include "source_hamilt/hamilt.h" #include "source_estate/module_dm/density_matrix.h" #include "source_lcao/module_lr/operator_casida/operator_lr_diag.h" @@ -38,9 +40,9 @@ namespace LR gdim(nk* std::inner_product(nocc.begin(), nocc.end(), nvirt.begin(), 0)) { ModuleBase::TITLE("HamiltULR", "HamiltULR"); - this->DM_trans = LR_Util::make_unique>(&pmat_in, 1, kv_in.kvec_d, nk); + this->DM_trans = LR_Util::make_unique>(&pmat_in, 1, kv_in.kvec_d, nk); LR_Util::initialize_DMR(*this->DM_trans, pmat_in, ucell_in, gd_in, orb_cutoff); - // this->DM_trans->init_DMR(&gd_in, &ucell_in); // too large due to not restricted by orb_cutoff + // this->DM_trans->init_dmr(&gd_in, &ucell_in); // too large due to not restricted by orb_cutoff this->ops.resize(4); this->ops[0] = new OperatorLRDiag(eig_ks.c, pX_in[0], nk, nocc[0], nvirt[0]); @@ -79,7 +81,7 @@ namespace LR #endif // LR_Util::print_tensor(dm_trans_2d[0], "DMtrans(k=0)", &pmat_in); // tensor to vector, then set DMK - for (int ik = 0;ik < nk;++ik) { this->DM_trans->set_DMK_pointer(ik, dm_trans_2d[ik].data()); } + for (int ik = 0;ik < nk;++ik) { this->DM_trans->set_dmk_ptr(ik, dm_trans_2d[ik].data()); } }; } ~HamiltULR() @@ -220,9 +222,11 @@ namespace LR /// transition density matrix in AO representation /// Hxc only: size=1, calculate on the same address for each bands /// Hxc+Exx: size=nbands, store the result of each bands for common use - std::unique_ptr> DM_trans; + std::unique_ptr> DM_trans; std::function cal_dm_trans; const bool tdm_sym = false; ///< whether to symmetrize the transition density matrix }; -} \ No newline at end of file +} + +#endif // ABACUS_SOURCE_LCAO_MODULE_LR_HAMILT_ULR_HPP diff --git a/source/source_lcao/module_lr/hsolver_lrtd.hpp b/source/source_lcao/module_lr/hsolver_lrtd.hpp index 629bf2b9628..55880f0d294 100644 --- a/source/source_lcao/module_lr/hsolver_lrtd.hpp +++ b/source/source_lcao/module_lr/hsolver_lrtd.hpp @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_LCAO_MODULE_LR_HSOLVER_LRTD_HPP +#define ABACUS_SOURCE_LCAO_MODULE_LR_HSOLVER_LRTD_HPP + #include "source_io/module_parameter/parameter.h" #include "source_hsolver/diag_comm_info.h" #include "source_hsolver/diago_david.h" @@ -16,6 +18,23 @@ namespace LR namespace HSolver { + /// The LR Hamiltonians (HamiltLR, HamiltULR) are not hamilt::Hamilt, so + /// they get their own hsolver::HSOperator view. S is the identity. + template + class LRHSOperator : public hsolver::HSOperator + { + public: + explicit LRHSOperator(const THamilt& hm) : hm_(hm) {} + void update_k(const int ik) override {} + void hpsi(const T* x, T* hx, const int ld, const int nvec) const override { hm_.hPsi(x, hx, ld, nvec); } + void spsi(const T* x, T* sx, const int ld, const int nvec) const override + { + std::memcpy(sx, x, sizeof(T) * static_cast(ld) * static_cast(nvec)); + } + private: + const THamilt& hm_; + }; + template inline void print_eigs(const std::vector& eigs, const std::string& label = "", const double factor = 1.0) { @@ -78,12 +97,9 @@ namespace LR } else { - // 3. set maxiter and funcs + // 3. set maxiter and the operator const int maxiter = hsolver::DiagoIterAssist::PW_DIAG_NMAX; - - auto hpsi_func = [&hm](T* psi_in, T* hpsi, const int ld_psi, const int nvec) {hm.hPsi(psi_in, hpsi, ld_psi, nvec);}; - auto spsi_func = [&hm](const T* psi_in, T* spsi, const int ld_psi, const int nvec) - { std::memcpy(spsi, psi_in, sizeof(T) * ld_psi * nvec); }; + const LRHSOperator op(hm); if (method == "dav") { @@ -99,7 +115,7 @@ namespace LR PARAM.inp.pw_diag_ndim, comm_info); std::vector ethr_band(nband, diag_ethr); - hsolver::DiagoIterAssist::avg_iter += static_cast(david.diag(hpsi_func, spsi_func, + hsolver::DiagoIterAssist::avg_iter += static_cast(david.diag(op, dim, psi, eigenvalue.data(), ethr_band, maxiter, ntry_max, 0)); } else if (method == "dav_subspace") //need refactor @@ -115,49 +131,16 @@ namespace LR PARAM.inp.nb2d); std::vector ethr_band(nband, diag_ethr); hsolver::DiagoIterAssist::avg_iter += static_cast( - dav_subspace.diag(hpsi_func, spsi_func, psi, dim, eigenvalue.data(), ethr_band, false /*scf*/)); + dav_subspace.diag(op, psi, dim, eigenvalue.data(), ethr_band, false /*scf*/)); } else if (method == "cg") { - ////// `diagH_subspace` needs refactor: - ////// replace `Hamilt*` with `hpsi_func` - ////// or I cannot use `is_subspace=true` as my `HamiltLR` does not inherit `Hamilt`. - - // auto subspace_func = [&hm](const ct::Tensor& psi_in, ct::Tensor& psi_out) { - // const auto ndim = psi_in.shape().ndim(); - // REQUIRES_OK(ndim == 2, "dims of psi_in should be less than or equal to 2"); - // // Convert a Tensor object to a psi::Psi object - // auto psi_in_wrapper = psi::Psi(psi_in.data(), - // 1, - // psi_in.shape().dim_size(0), - // psi_in.shape().dim_size(1)); - // auto psi_out_wrapper = psi::Psi(psi_out.data(), - // 1, - // psi_out.shape().dim_size(0), - // psi_out.shape().dim_size(1)); - // auto eigen = ct::Tensor(ct::DataTypeToEnum>::value, - // ct::DeviceType::CpuDevice, - // ct::TensorShape({ psi_in.shape().dim_size(0) })); - // hsolver::DiagoIterAssist::diagH_subspace(hm, psi_in_wrapper, psi_out_wrapper, eigen.data>()); - // }; - - ////// why diago_cg depends on basis_type? - // hsolver::DiagoCG cg("lcao", "nscf", true, subspace_func, diag_ethr, maxiter, GlobalV::NPROC_IN_POOL); - - auto subspace_func = [](T* psi_in, T* psi_out, const int ld_psi, const int nband, const bool S_orth) { - }; - hsolver::DiagoCG cg("lcao", "nscf", false, subspace_func, diag_ethr, maxiter, GlobalV::NPROC_IN_POOL); - - auto hpsi_func = [&hm](T* psi_in, T* hpsi, const int ld_psi, const int nvec) { - hm.hPsi(psi_in, hpsi, ld_psi, nvec); - }; - auto spsi_func = [](T* psi_in, T* spsi, const int ld_psi, const int nvec) { - std::memcpy(spsi, psi_in, sizeof(T) * static_cast(ld_psi) * static_cast(nvec)); - }; + // the subspace rotation of DiagoCG now works on any HSOperator, so it could be + // switched on here; it is kept off to leave the LR results unchanged + hsolver::DiagoCG cg("lcao", "nscf", false, comm_info, diag_ethr, maxiter); std::vector ethr_band(nband, diag_ethr); - cg.diag(hpsi_func, - spsi_func, + cg.diag(op, dim, nband, dim, @@ -198,3 +181,5 @@ namespace LR } } } + +#endif // ABACUS_SOURCE_LCAO_MODULE_LR_HSOLVER_LRTD_HPP diff --git a/source/source_lcao/module_lr/lr_spectrum.cpp b/source/source_lcao/module_lr/lr_spectrum.cpp index 4f184fb80ea..7f38537eab1 100644 --- a/source/source_lcao/module_lr/lr_spectrum.cpp +++ b/source/source_lcao/module_lr/lr_spectrum.cpp @@ -9,11 +9,11 @@ #include "source_hamilt/module_gint/gint_interface.h" template -elecstate::DensityMatrix LR::LR_Spectrum::cal_transition_density_matrix(const int istate, const T* X_in, const bool need_R) +module_dm::DensityMatrix LR::LR_Spectrum::cal_transition_density_matrix(const int istate, const T* X_in, const bool need_R) { const T* const X = X_in == nullptr ? this->X : X_in; const int offset_b = istate * ldim; //start index of band istate - elecstate::DensityMatrix DM_trans(&this->pmat, this->nspin_x, this->kv.kvec_d, this->nk); + module_dm::DensityMatrix DM_trans(&this->pmat, this->nspin_x, this->kv.kvec_d, this->nk); for (int is = 0;is < this->nspin_x; ++is) { const int offset_x = offset_b + is * nk * this->pX[0].get_local_size(); @@ -25,12 +25,12 @@ elecstate::DensityMatrix LR::LR_Spectrum::cal_transition_density_matrix std::vector dm_trans_2d = cal_dm_trans_blas(X + offset_x, this->psi_ks_vec[is], this->nocc[is], this->nvirt[is], (T)1.0 / (T)nk); // if (this->tdm_sym) for (auto& t : dm_trans_2d) LR_Util::matsym(t.data(), naos); #endif - for (int ik = 0;ik < this->nk;++ik) { DM_trans.set_DMK_pointer(ik + is * nk, dm_trans_2d[ik].data()); } + for (int ik = 0;ik < this->nk;++ik) { DM_trans.set_dmk_ptr(ik + is * nk, dm_trans_2d[ik].data()); } } if (need_R) { LR_Util::initialize_DMR(DM_trans, this->pmat, this->ucell, this->gd_, this->orb_cutoff_); - DM_trans.cal_DMR(); + DM_trans.cal_dmr(-1); } return DM_trans; } @@ -50,14 +50,14 @@ ModuleBase::Vector3 LR::LR_Spectrum::cal_transition_dipole_istat { ModuleBase::Vector3 trans_dipole(0.0, 0.0, 0.0); // 1. transition density matrix - const elecstate::DensityMatrix DM_trans = this->cal_transition_density_matrix(istate); + const module_dm::DensityMatrix DM_trans = this->cal_transition_density_matrix(istate); for (int is = 0;is < this->nspin_x;++is) { // 2. transition density double** rho_trans = nullptr; LR_Util::_allocate_2order_nested_ptr(rho_trans, 1, this->rho_basis.nrxx); ModuleBase::GlobalFunc::ZEROS(rho_trans[0], this->rho_basis.nrxx); - ModuleGint::cal_gint_rho({ DM_trans.get_DMR_vector().at(is) }, 1, rho_trans, false); + ModuleGint::cal_gint_rho({ DM_trans.get_dmr_vec().at(is) }, 1, rho_trans, false); // 3. transition dipole moment for (int ir = 0; ir < rho_basis.nrxx; ++ir) @@ -87,7 +87,7 @@ ModuleBase::Vector3> LR::LR_Spectrum>: //1. transition density matrix ModuleBase::Vector3> trans_dipole(0.0, 0.0, 0.0); - const elecstate::DensityMatrix, std::complex> DM_trans = this->cal_transition_density_matrix(istate); + const module_dm::DensityMatrix, std::complex> DM_trans = this->cal_transition_density_matrix(istate); for (int is = 0;is < this->nspin_x;++is) { // 2. transition density @@ -96,19 +96,19 @@ ModuleBase::Vector3> LR::LR_Spectrum>: LR_Util::_allocate_2order_nested_ptr(rho_trans_real, 1, this->rho_basis.nrxx); LR_Util::_allocate_2order_nested_ptr(rho_trans_imag, 1, this->rho_basis.nrxx); - elecstate::DensityMatrix, double> DM_trans_real_imag(&this->pmat, 1, this->kv.kvec_d, this->nk); + module_dm::DensityMatrix, double> DM_trans_real_imag(&this->pmat, 1, this->kv.kvec_d, this->nk); LR_Util::initialize_DMR(DM_trans_real_imag, this->pmat, this->ucell, this->gd_, this->orb_cutoff_); // real part - LR_Util::get_DMR_real_imag_part(DM_trans, DM_trans_real_imag, ucell.nat, 'R'); + LR_Util::get_DMR_real_imag_part(DM_trans, DM_trans_real_imag, ucell.nat, is, 'R'); ModuleBase::GlobalFunc::ZEROS(rho_trans_real[0], this->rho_basis.nrxx); - ModuleGint::cal_gint_rho(DM_trans_real_imag.get_DMR_vector(), 1, rho_trans_real, false); + ModuleGint::cal_gint_rho(DM_trans_real_imag.get_dmr_vec(), 1, rho_trans_real, false); // LR_Util::print_grid_nonzero(rho_trans_real[0], this->rho_basis.nrxx, 10, "rho_trans"); // imag part - LR_Util::get_DMR_real_imag_part(DM_trans, DM_trans_real_imag, ucell.nat, 'I'); + LR_Util::get_DMR_real_imag_part(DM_trans, DM_trans_real_imag, ucell.nat, is, 'I'); ModuleBase::GlobalFunc::ZEROS(rho_trans_imag[0], this->rho_basis.nrxx); - ModuleGint::cal_gint_rho(DM_trans_real_imag.get_DMR_vector(), 1, rho_trans_imag, false); + ModuleGint::cal_gint_rho(DM_trans_real_imag.get_dmr_vec(), 1, rho_trans_imag, false); // LR_Util::print_grid_nonzero(rho_trans_imag[0], this->rho_basis.nrxx, 10, "rho_trans"); // 3. transition dipole moment @@ -192,10 +192,10 @@ void LR::LR_Spectrum::optical_absorption_method1(const std::vector& f std::vector& osc = this->oscillator_strength_; std::ofstream ofs(this->out_dir + "absorption.dat"); - if (this->my_rank == 0) - { - ofs << "Frequency (eV) | wave length(nm) | Absorption (a.u.)" << std::endl; - } + if (this->my_rank == 0) + { + ofs << "Frequency (eV) | wave length(nm) | Absorption (a.u.)" << std::endl; + } double FourPI_div_c = ModuleBase::FOUR_PI / 137.036; double fac = 4 * M_PI / ucell.omega * ModuleBase::e2 / this->nk; // e2 for Ry to Hartree in the denominator diff --git a/source/source_lcao/module_lr/lr_spectrum.h b/source/source_lcao/module_lr/lr_spectrum.h index 91621080bf2..e60fe3d6efc 100644 --- a/source/source_lcao/module_lr/lr_spectrum.h +++ b/source/source_lcao/module_lr/lr_spectrum.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_LCAO_MODULE_LR_LR_SPECTRUM_H +#define ABACUS_SOURCE_LCAO_MODULE_LR_LR_SPECTRUM_H + #include "source_cell/klist.h" #include "source_psi/psi.h" #include "source_estate/module_dm/density_matrix.h" @@ -72,7 +74,7 @@ namespace LR void cal_transition_dipoles_velocity(const double* const eig_ks); double cal_mean_squared_dipole(ModuleBase::Vector3 dipole); /// calculate the transition density matrix - elecstate::DensityMatrix cal_transition_density_matrix(const int istate, const T* X_in = nullptr, const bool need_R = true); + module_dm::DensityMatrix cal_transition_density_matrix(const int istate, const T* X_in = nullptr, const bool need_R = true); const int my_rank; const int nspin_x = 1; ///< 1 for singlet/triplet, 2 for updown(openshell) @@ -111,3 +113,5 @@ namespace LR std::vector oscillator_strength_;///< $2/3\Omega |\sum_{ia\sigma} \braket{\psi_{i}|\mathbf{r}|\psi_{a}} |^2$, atomic unit (Hartree) }; } + +#endif // ABACUS_SOURCE_LCAO_MODULE_LR_LR_SPECTRUM_H diff --git a/source/source_lcao/module_lr/lr_spectrum_velocity.cpp b/source/source_lcao/module_lr/lr_spectrum_velocity.cpp index 21d1a97c4ef..b94b8d3bd97 100644 --- a/source/source_lcao/module_lr/lr_spectrum_velocity.cpp +++ b/source/source_lcao/module_lr/lr_spectrum_velocity.cpp @@ -84,7 +84,7 @@ namespace LR ModuleBase::Vector3 LR::LR_Spectrum::cal_transition_dipole_istate_velocity_R(const int istate, const Velocity_op>& vR) { // transition density matrix D(R) - const elecstate::DensityMatrix& DM_trans = this->cal_transition_density_matrix(istate); + const module_dm::DensityMatrix& DM_trans = this->cal_transition_density_matrix(istate); std::vector> trans_dipole(3, 0.0); // $=\sum_{uvR} v(R) D(R) = \sum_{aik}X_{aik}$ const std::complex fac = ModuleBase::IMAG_UNIT / (omega[istate] / ModuleBase::e2); // Ry to Hartree @@ -92,7 +92,7 @@ namespace LR { for (int is = 0;is < this->nspin_x; ++is) { - trans_dipole[i] += LR_Util::dot_R_matrix(*vR.get_current_term_pointer(i), *DM_trans.get_DMR_pointer(is + 1), ucell.nat) * fac; + trans_dipole[i] += LR_Util::dot_R_matrix(*vR.get_current_term_pointer(i), *DM_trans.get_dmr_ptr(is + 1), ucell.nat) * fac; } // end for spin_x, only matter in open-shell system trans_dipole[i] *= static_cast(this->nk); // nk is divided inside DM_trans, now recover it if (this->nspin_x == 1) { trans_dipole[i] *= sqrt(2.0); } // *2 for 2 spins, /sqrt(2) for the halfed dimension of X in the normalizaiton @@ -106,7 +106,7 @@ namespace LR ModuleBase::Vector3 LR::LR_Spectrum::cal_transition_dipole_istate_velocity_k(const int istate, const Velocity_op>& vR) { // transition density matrix D(R) - const elecstate::DensityMatrix& DM_trans = this->cal_transition_density_matrix(istate, this->X, false); + const module_dm::DensityMatrix& DM_trans = this->cal_transition_density_matrix(istate, this->X, false); std::vector> trans_dipole(3, 0.0); // $=\sum_{uvk} v(k) D(k) = \sum_{aik}X_{aik}$ const std::complex fac = ModuleBase::IMAG_UNIT / (omega[istate] / ModuleBase::e2); // Ry to Hartree @@ -118,7 +118,7 @@ namespace LR { std::vector> vk(pmat.get_local_size(), 0.0); hamilt::folding_HR(*vR.get_current_term_pointer(i), vk.data(), kv.kvec_d[ik], pmat.get_row_size(), 1); - trans_dipole[i] += std::inner_product(vk.begin(), vk.end(), DM_trans.get_DMK_pointer(is * nk + ik), std::complex(0., 0.)) * fac; + trans_dipole[i] += std::inner_product(vk.begin(), vk.end(), DM_trans.get_dmk_ptr(is * nk + ik), std::complex(0., 0.)) * fac; } } // end for spin_x, only matter in open-shell system trans_dipole[i] *= static_cast(this->nk); // nk is divided inside DM_trans, now recover it diff --git a/source/source_lcao/module_lr/operator_casida/operator_lr_diag.h b/source/source_lcao/module_lr/operator_casida/operator_lr_diag.h index b651be3fb71..041a43c2063 100644 --- a/source/source_lcao/module_lr/operator_casida/operator_lr_diag.h +++ b/source/source_lcao/module_lr/operator_casida/operator_lr_diag.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_LCAO_MODULE_LR_OPERATOR_CASIDA_OPERATOR_LR_DIAG_H +#define ABACUS_SOURCE_LCAO_MODULE_LR_OPERATOR_CASIDA_OPERATOR_LR_DIAG_H + #include "source_base/kernels/math_kernel_op.h" #include "source_hamilt/operator.h" #ifdef __MPI @@ -61,3 +63,5 @@ namespace LR Device* ctx = {}; }; } + +#endif // ABACUS_SOURCE_LCAO_MODULE_LR_OPERATOR_CASIDA_OPERATOR_LR_DIAG_H diff --git a/source/source_lcao/module_lr/operator_casida/operator_lr_exx.cpp b/source/source_lcao/module_lr/operator_casida/operator_lr_exx.cpp index 3879569b2f5..7ffc500f946 100644 --- a/source/source_lcao/module_lr/operator_casida/operator_lr_exx.cpp +++ b/source/source_lcao/module_lr/operator_casida/operator_lr_exx.cpp @@ -107,7 +107,7 @@ namespace LR // 1. set_Ds (once) // convert to vector for the interface of RI_2D_Comm::split_m2D_ktoR (interface will be unified to ct::Tensor) - std::vector> DMk_trans_vector = this->DM_trans->get_DMK_vector(); + std::vector> DMk_trans_vector = this->DM_trans->get_dmk_vec(); // assert(DMk_trans_vector.size() == nk); std::vector*> DMk_trans_pointer(nk); for (int ik = 0;ik < nk;++ik) { DMk_trans_pointer[ik] = &DMk_trans_vector[ik]; } diff --git a/source/source_lcao/module_lr/operator_casida/operator_lr_exx.h b/source/source_lcao/module_lr/operator_casida/operator_lr_exx.h index e6e1b45ff92..d426cdf6d39 100644 --- a/source/source_lcao/module_lr/operator_casida/operator_lr_exx.h +++ b/source/source_lcao/module_lr/operator_casida/operator_lr_exx.h @@ -1,5 +1,7 @@ #ifdef __EXX -#pragma once +#ifndef ABACUS_SOURCE_LCAO_MODULE_LR_OPERATOR_CASIDA_OPERATOR_LR_EXX_H +#define ABACUS_SOURCE_LCAO_MODULE_LR_OPERATOR_CASIDA_OPERATOR_LR_EXX_H + #include "source_hamilt/operator.h" #include "source_estate/module_dm/density_matrix.h" #include "source_lcao/module_ri/exx_lri.h" @@ -23,7 +25,7 @@ namespace LR const int& nvirt, const UnitCell& ucell_in, const psi::Psi& psi_ks_in, - std::unique_ptr>& DM_trans_in, + std::unique_ptr>& DM_trans_in, // HContainer* hR_in, std::weak_ptr> exx_lri_in, const K_Vectors& kv_in, @@ -82,12 +84,12 @@ namespace LR psi::Psi psi_ks_full; /// transition density matrix - std::unique_ptr>& DM_trans; + std::unique_ptr>& DM_trans; /// density matrix of a certain (i, a, k), with full naos*naos size for each key /// D^{iak}_{\mu\nu}(k): 1/N_k * c_{ak,\mu} c^*_{ik,\nu} /// D^{iak}_{\mu\nu}(R): D^{iak}_{\mu\nu}(k)e^{-ikR} - // elecstate::DensityMatrix* DM_onebase; + // module_dm::DensityMatrix* DM_onebase; mutable std::map>> Ds_onebase; // cells in the Born von Karmen supercell (direct) @@ -116,4 +118,6 @@ namespace LR }; } -#endif \ No newline at end of file +#endif + +#endif // ABACUS_SOURCE_LCAO_MODULE_LR_OPERATOR_CASIDA_OPERATOR_LR_EXX_H diff --git a/source/source_lcao/module_lr/operator_casida/operator_lr_hxc.cpp b/source/source_lcao/module_lr/operator_casida/operator_lr_hxc.cpp index 675a15b91d0..64c38a184bb 100644 --- a/source/source_lcao/module_lr/operator_casida/operator_lr_hxc.cpp +++ b/source/source_lcao/module_lr/operator_casida/operator_lr_hxc.cpp @@ -24,7 +24,7 @@ namespace LR const int& sl = ispin_ks[0]; const auto psil_ks = LR_Util::get_psi_spin(psi_ks, sl, nk); - this->DM_trans->cal_DMR(); //DM_trans->get_DMR_vector() is 2d-block parallized + this->DM_trans->cal_dmr(-1); //DM_trans->get_dmr_vec() is 2d-block parallized // LR_Util::print_DMR(*DM_trans, ucell.nat, "DMR"); // ========================= begin grid calculation========================= @@ -67,7 +67,7 @@ namespace LR const int& nrxx = this->pot.lock()->nrxx; LR_Util::_allocate_2order_nested_ptr(rho_trans, 1, nrxx); // currently gint_kernel_rho uses PARAM.inp.nspin, it needs refactor ModuleBase::GlobalFunc::ZEROS(rho_trans[0], nrxx); - ModuleGint::cal_gint_rho(this->DM_trans->get_DMR_vector(), 1, rho_trans, false); + ModuleGint::cal_gint_rho(this->DM_trans->get_dmr_vec(), 1, rho_trans, false); // 3. v_hxc = f_hxc * rho_trans ModuleBase::matrix vr_hxc(1, nrxx); //grid this->pot.lock()->cal_v_eff(rho_trans, ucell, vr_hxc, ispin_ks); @@ -85,8 +85,8 @@ namespace LR ModuleBase::TITLE("OperatorLRHxc", "grid_calculation(complex)"); ModuleBase::timer::start("OperatorLRHxc", "grid_calculation"); - elecstate::DensityMatrix, double> DM_trans_real_imag(&pmat, 1, kv.kvec_d, kv.get_nks() / nspin); - DM_trans_real_imag.init_DMR(*this->hR); + module_dm::DensityMatrix, double> DM_trans_real_imag(&pmat, 1, kv.kvec_d, kv.get_nks() / nspin); + DM_trans_real_imag.init_dmr(*this->hR); hamilt::HContainer HR_real_imag(ucell, &this->pmat); LR_Util::initialize_HR, double>(HR_real_imag, ucell, gd, orb_cutoff_); @@ -102,7 +102,7 @@ namespace LR LR_Util::_allocate_2order_nested_ptr(rho_trans, 1, nrxx); // nspin=1 for transition density ModuleBase::GlobalFunc::ZEROS(rho_trans[0], nrxx); - ModuleGint::cal_gint_rho(DM_trans_real_imag.get_DMR_vector(), 1, rho_trans, false); + ModuleGint::cal_gint_rho(DM_trans_real_imag.get_dmr_vec(), 1, rho_trans, false); // print_grid_nonzero(rho_trans[0], nrxx, 10, "rho_trans"); // 3. v_hxc = f_hxc * rho_trans diff --git a/source/source_lcao/module_lr/operator_casida/operator_lr_hxc.h b/source/source_lcao/module_lr/operator_casida/operator_lr_hxc.h index 2318056bfb5..0abb437ef50 100644 --- a/source/source_lcao/module_lr/operator_casida/operator_lr_hxc.h +++ b/source/source_lcao/module_lr/operator_casida/operator_lr_hxc.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_LCAO_MODULE_LR_OPERATOR_CASIDA_OPERATOR_LR_HXC_H +#define ABACUS_SOURCE_LCAO_MODULE_LR_OPERATOR_CASIDA_OPERATOR_LR_HXC_H + #include "source_cell/klist.h" #include "source_hamilt/operator.h" #include "source_estate/module_dm/density_matrix.h" @@ -18,7 +20,7 @@ namespace LR const std::vector& nocc, const std::vector& nvirt, const psi::Psi& psi_ks_in, - std::unique_ptr>& DM_trans_in, + std::unique_ptr>& DM_trans_in, std::weak_ptr pot_in, const UnitCell& ucell_in, const std::vector& orb_cutoff, @@ -68,7 +70,7 @@ namespace LR const psi::Psi& psi_ks = nullptr; /// transition density matrix - std::unique_ptr>& DM_trans; + std::unique_ptr>& DM_trans; /// transition hamiltonian in AO representation std::unique_ptr> hR = nullptr; @@ -88,3 +90,5 @@ namespace LR mutable bool first_print = true; }; } + +#endif // ABACUS_SOURCE_LCAO_MODULE_LR_OPERATOR_CASIDA_OPERATOR_LR_HXC_H diff --git a/source/source_lcao/module_lr/potentials/pot_hxc_lrtd.h b/source/source_lcao/module_lr/potentials/pot_hxc_lrtd.h index bb8a28274db..8a712253727 100644 --- a/source/source_lcao/module_lr/potentials/pot_hxc_lrtd.h +++ b/source/source_lcao/module_lr/potentials/pot_hxc_lrtd.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_LCAO_MODULE_LR_POTENTIALS_POT_HXC_LRTD_H +#define ABACUS_SOURCE_LCAO_MODULE_LR_POTENTIALS_POT_HXC_LRTD_H + #include "source_estate/module_pot/h_hartree_pw.h" #include "xc_kernel.h" #include @@ -51,3 +53,5 @@ namespace LR }; } // namespace LR + +#endif // ABACUS_SOURCE_LCAO_MODULE_LR_POTENTIALS_POT_HXC_LRTD_H diff --git a/source/source_lcao/module_lr/potentials/xc_kernel.h b/source/source_lcao/module_lr/potentials/xc_kernel.h index 06cb78c9e3d..96f4e42bc18 100644 --- a/source/source_lcao/module_lr/potentials/xc_kernel.h +++ b/source/source_lcao/module_lr/potentials/xc_kernel.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_LCAO_MODULE_LR_POTENTIALS_XC_KERNEL_H +#define ABACUS_SOURCE_LCAO_MODULE_LR_POTENTIALS_XC_KERNEL_H + #include "source_basis/module_pw/pw_basis.h" #include "source_cell/unitcell.h" #include "source_base/parallel_grid.h" @@ -73,3 +75,4 @@ namespace LR }; } +#endif // ABACUS_SOURCE_LCAO_MODULE_LR_POTENTIALS_XC_KERNEL_H diff --git a/source/source_lcao/module_lr/ri_benchmark/operator_ri_hartree.h b/source/source_lcao/module_lr/ri_benchmark/operator_ri_hartree.h index d2f497878e3..690a46a2918 100644 --- a/source/source_lcao/module_lr/ri_benchmark/operator_ri_hartree.h +++ b/source/source_lcao/module_lr/ri_benchmark/operator_ri_hartree.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_LCAO_MODULE_LR_RI_BENCHMARK_OPERATOR_RI_HARTREE_H +#define ABACUS_SOURCE_LCAO_MODULE_LR_RI_BENCHMARK_OPERATOR_RI_HARTREE_H + #include "source_hamilt/operator.h" #include "source_lcao/module_lr/ri_benchmark/ri_benchmark.h" #include "source_lcao/module_lr/utils/lr_util_print.h" @@ -80,4 +82,6 @@ namespace RI_Benchmark const TLRI CV_ov; const TLRI CV_vo; }; -} \ No newline at end of file +} + +#endif // ABACUS_SOURCE_LCAO_MODULE_LR_RI_BENCHMARK_OPERATOR_RI_HARTREE_H diff --git a/source/source_lcao/module_lr/ri_benchmark/ri_benchmark.h b/source/source_lcao/module_lr/ri_benchmark/ri_benchmark.h index 0615bfe1df6..a781ece5fca 100644 --- a/source/source_lcao/module_lr/ri_benchmark/ri_benchmark.h +++ b/source/source_lcao/module_lr/ri_benchmark/ri_benchmark.h @@ -1,5 +1,7 @@ #ifdef __EXX -#pragma once +#ifndef ABACUS_SOURCE_LCAO_MODULE_LR_RI_BENCHMARK_RI_BENCHMARK_H +#define ABACUS_SOURCE_LCAO_MODULE_LR_RI_BENCHMARK_RI_BENCHMARK_H + #include "source_cell/unitcell.h" #include "source_psi/psi.h" #include "source_lcao/module_lr/utils/lr_io.h" @@ -74,4 +76,6 @@ namespace RI_Benchmark std::vector> split_Ds(const std::vector>& Ds, const std::vector& aims_nbasis, const UnitCell& ucell); } #include "ri_benchmark.hpp" -#endif \ No newline at end of file +#endif + +#endif // ABACUS_SOURCE_LCAO_MODULE_LR_RI_BENCHMARK_RI_BENCHMARK_H diff --git a/source/source_lcao/module_lr/ri_benchmark/ri_benchmark.hpp b/source/source_lcao/module_lr/ri_benchmark/ri_benchmark.hpp index a7306047edc..6c097827dc6 100644 --- a/source/source_lcao/module_lr/ri_benchmark/ri_benchmark.hpp +++ b/source/source_lcao/module_lr/ri_benchmark/ri_benchmark.hpp @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_LCAO_MODULE_LR_RI_BENCHMARK_RI_BENCHMARK_HPP +#define ABACUS_SOURCE_LCAO_MODULE_LR_RI_BENCHMARK_RI_BENCHMARK_HPP + #include #include #include "ri_benchmark.h" @@ -378,7 +380,7 @@ namespace RI_Benchmark template std::vector> split_Ds(const std::vector>& Ds, const std::vector& aims_nbasis, const UnitCell& ucell) // vector index: ispin { - // Due to the hard-coded constructor of elecstate::DensityMatrix, singlet-triplet with nspin=2 cannot use DM_trans with size 1 + // Due to the hard-coded constructor of module_dm::DensityMatrix, singlet-triplet with nspin=2 cannot use DM_trans with size 1 // if(Ds.size()>1) { throw std::runtime_error("split_Ds only supports gamma-only spin-1 Ds now."); } std::vector> Ds_split; for (const auto& D : Ds) @@ -414,3 +416,5 @@ namespace RI_Benchmark return Ds_split; } } + +#endif // ABACUS_SOURCE_LCAO_MODULE_LR_RI_BENCHMARK_RI_BENCHMARK_HPP diff --git a/source/source_lcao/module_lr/utils/exciton_plotter.cpp b/source/source_lcao/module_lr/utils/exciton_plotter.cpp index 6bcfa64794b..b7c4483198f 100644 --- a/source/source_lcao/module_lr/utils/exciton_plotter.cpp +++ b/source/source_lcao/module_lr/utils/exciton_plotter.cpp @@ -478,13 +478,13 @@ void ExcitonPlotter::plot_average_density(const int istate, const std::string ModuleBase::WARNING_QUIT("ExcitonPlotter", "Unknown average density type: " + type + ". Use hole or elec."); } const auto dmk = type == "hole" ? cal_effective_dmk_hole(istate) : cal_effective_dmk_elec(istate); - elecstate::DensityMatrix dm(&this->pmat, this->nspin_x, this->kv.kvec_d, this->nk); + module_dm::DensityMatrix dm(&this->pmat, this->nspin_x, this->kv.kvec_d, this->nk); for (int ik = 0; ik < this->nk; ++ik) { - dm.set_DMK_pointer(ik, dmk[ik].template data()); + dm.set_dmk_ptr(ik, dmk[ik].template data()); } LR_Util::initialize_DMR(dm, this->pmat, this->ucell, this->gd_, this->orb_cutoff_); - dm.cal_DMR(); + dm.cal_dmr(-1); double** rho_result = nullptr; LR_Util::_allocate_2order_nested_ptr(rho_result, this->nspin_x, this->rho_basis.nrxx); @@ -492,7 +492,7 @@ void ExcitonPlotter::plot_average_density(const int istate, const std::string { ModuleBase::GlobalFunc::ZEROS(rho_result[is], this->rho_basis.nrxx); } - ModuleGint::cal_gint_rho(dm.get_DMR_vector(), this->nspin_x, rho_result, false); + ModuleGint::cal_gint_rho(dm.get_dmr_vec(), this->nspin_x, rho_result, false); for (int is = 0; is < this->nspin_x; ++is) { diff --git a/source/source_lcao/module_lr/utils/exciton_plotter.h b/source/source_lcao/module_lr/utils/exciton_plotter.h index 18ad7521233..87a5ea71884 100644 --- a/source/source_lcao/module_lr/utils/exciton_plotter.h +++ b/source/source_lcao/module_lr/utils/exciton_plotter.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_LCAO_MODULE_LR_UTILS_EXCITON_PLOTTER_H +#define ABACUS_SOURCE_LCAO_MODULE_LR_UTILS_EXCITON_PLOTTER_H + #include "source_base/tool_title.h" #include "source_base/ylm.h" #include "source_basis/module_ao/orb_read.h" @@ -166,7 +168,7 @@ class ExcitonPlotter /// @brief Compute effective DMK for average hole density /// Produces D_hole(k)^T, where D_hole(k) = C_occ(k) * (X_k^H * X_k) * C_occ(k)^H, - /// matching the AO-index order expected by DensityMatrix::cal_DMR(). + /// matching the AO-index order expected by DensityMatrix::cal_dmr(). /// Marginalizes over conduction bands: M_k[v,v'] = Sum_c A_{kvc} * conj(A_{kv'c}). /// @param istate BSE state index /// @return dmk_per_kpoint @@ -174,7 +176,7 @@ class ExcitonPlotter /// @brief Compute effective DMK for average electron density /// Produces D_elec(k)^T, where D_elec(k) = C_virt(k) * (X_k * X_k^H) * C_virt(k)^H, - /// matching the AO-index order expected by DensityMatrix::cal_DMR(). + /// matching the AO-index order expected by DensityMatrix::cal_dmr(). /// Marginalizes over valence bands: N_k[c,c'] = Sum_v A_{kvc} * conj(A_{kvc'}). /// @param istate BSE state index /// @return dmk_per_kpoint @@ -254,3 +256,5 @@ class ExcitonPlotter }; } // namespace LR_Util + +#endif // ABACUS_SOURCE_LCAO_MODULE_LR_UTILS_EXCITON_PLOTTER_H diff --git a/source/source_lcao/module_lr/utils/lr_io.cpp b/source/source_lcao/module_lr/utils/lr_io.cpp index e27d66e14d4..f61a8a6a16c 100644 --- a/source/source_lcao/module_lr/utils/lr_io.cpp +++ b/source/source_lcao/module_lr/utils/lr_io.cpp @@ -749,7 +749,7 @@ TLRI read_Ws(const TLRI& Vs, const std::vector& Rlist) const std::string filename = "librpa.d/Wc_iR_" + std::to_string(iR) + "_ifreq_0.mtx"; std::ifstream infileW(filename); if(!infileW) throw std::runtime_error(filename + " not found!"); - if(GlobalV::MY_RANK == 0) std::cout << "reading Wc file: " << filename << std::endl; + // if(GlobalV::MY_RANK == 0) std::cout << "reading Wc file: " << filename << std::endl; TC R{}; // iR of Wc file is not equal to iR in Rlist !!! bool R_is_found = false; diff --git a/source/source_lcao/module_lr/utils/lr_io.h b/source/source_lcao/module_lr/utils/lr_io.h index 28c6efe999b..af38f718856 100644 --- a/source/source_lcao/module_lr/utils/lr_io.h +++ b/source/source_lcao/module_lr/utils/lr_io.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_LCAO_MODULE_LR_UTILS_LR_IO_H +#define ABACUS_SOURCE_LCAO_MODULE_LR_UTILS_LR_IO_H + #include "source_base/tool_title.h" #include "source_basis/module_ao/parallel_orbitals.h" #include "source_cell/klist.h" @@ -114,3 +116,5 @@ void write_lri_R_max_norm(const TLRI& tensors, const UnitCell& ucell, const s #endif } // namespace LR_IO + +#endif // ABACUS_SOURCE_LCAO_MODULE_LR_UTILS_LR_IO_H diff --git a/source/source_lcao/module_lr/utils/lr_io_krlist.cpp b/source/source_lcao/module_lr/utils/lr_io_krlist.cpp index 84c05a2e14e..1d9b53d9fc2 100644 --- a/source/source_lcao/module_lr/utils/lr_io_krlist.cpp +++ b/source/source_lcao/module_lr/utils/lr_io_krlist.cpp @@ -1,6 +1,6 @@ #include "lr_io_krlist.h" #include "lr_io.h" -#include "source_lcao/module_ri/ri_util.h" +#include "source_lcao/module_ri/ri_util_bvk.h" #include "source_base/constants.h" #include #include diff --git a/source/source_lcao/module_lr/utils/lr_io_krlist.h b/source/source_lcao/module_lr/utils/lr_io_krlist.h index a6aef40edc7..87ef66fdf91 100644 --- a/source/source_lcao/module_lr/utils/lr_io_krlist.h +++ b/source/source_lcao/module_lr/utils/lr_io_krlist.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_LCAO_MODULE_LR_UTILS_LR_IO_KRLIST_H +#define ABACUS_SOURCE_LCAO_MODULE_LR_UTILS_LR_IO_KRLIST_H + #include "source_cell/klist.h" #include "source_cell/unitcell.h" @@ -29,3 +31,5 @@ class RI_kRlist }; } // namespace LR_IO + +#endif // ABACUS_SOURCE_LCAO_MODULE_LR_UTILS_LR_IO_KRLIST_H diff --git a/source/source_lcao/module_lr/utils/lr_util.h b/source/source_lcao/module_lr/utils/lr_util.h index 0f1ff481c27..c8c716c7402 100644 --- a/source/source_lcao/module_lr/utils/lr_util.h +++ b/source/source_lcao/module_lr/utils/lr_util.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_LCAO_MODULE_LR_UTILS_LR_UTIL_H +#define ABACUS_SOURCE_LCAO_MODULE_LR_UTILS_LR_UTIL_H + #include #include #include @@ -169,3 +171,5 @@ namespace LR_Util std::string toupper(const std::string& str); } #include "lr_util.hpp" + +#endif // ABACUS_SOURCE_LCAO_MODULE_LR_UTILS_LR_UTIL_H diff --git a/source/source_lcao/module_lr/utils/lr_util.hpp b/source/source_lcao/module_lr/utils/lr_util.hpp index ca084c50981..c5bd9440d49 100644 --- a/source/source_lcao/module_lr/utils/lr_util.hpp +++ b/source/source_lcao/module_lr/utils/lr_util.hpp @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_LCAO_MODULE_LR_UTILS_LR_UTIL_HPP +#define ABACUS_SOURCE_LCAO_MODULE_LR_UTILS_LR_UTIL_HPP + #include "lr_util.h" #include "source_cell/unitcell.h" #include "source_base/constants.h" @@ -583,3 +585,5 @@ namespace LR_Util #endif } + +#endif // ABACUS_SOURCE_LCAO_MODULE_LR_UTILS_LR_UTIL_HPP diff --git a/source/source_lcao/module_lr/utils/lr_util_hcontainer.cpp b/source/source_lcao/module_lr/utils/lr_util_hcontainer.cpp index d04952e615b..b655f00c99a 100644 --- a/source/source_lcao/module_lr/utils/lr_util_hcontainer.cpp +++ b/source/source_lcao/module_lr/utils/lr_util_hcontainer.cpp @@ -1,17 +1,17 @@ #include "lr_util_hcontainer.h" namespace LR_Util { - void get_DMR_real_imag_part(const elecstate::DensityMatrix, std::complex>& DMR, - elecstate::DensityMatrix, double>& DMR_real, + void get_DMR_real_imag_part(const module_dm::DensityMatrix, std::complex>& DMR, + module_dm::DensityMatrix, double>& DMR_real, const int& nat, const char& type) { - assert(DMR.get_DMR_vector().size() == DMR_real.get_DMR_vector().size()); + assert(DMR.get_dmr_vec().size() == DMR_real.get_dmr_vec().size()); bool get_imag = (type == 'I' || type == 'i'); - for (int is = 0;is < DMR.get_DMR_vector().size();++is) + for (int is = 0;is < DMR.get_dmr_vec().size();++is) { - auto dr = DMR.get_DMR_vector()[is]; //get_DMR_pointer() has bug when is=0 - auto dr_real = DMR_real.get_DMR_vector()[is]; + auto dr = DMR.get_dmr_vec()[is]; //get_dmr_ptr() has bug when is=0 + auto dr_real = DMR_real.get_dmr_vec()[is]; assert(dr != nullptr); assert(dr_real != nullptr); for (int ia = 0;ia < nat;ia++) { @@ -19,6 +19,9 @@ namespace LR_Util { auto ap = dr->find_pair(ia, ja); auto ap_real = dr_real->find_pair(ia, ja); + // under MPI-parallel (2D block-cyclic) HContainer, an atom pair not owned by this rank + // is absent from find_pair() and returns nullptr here; skip it + if (!ap || !ap_real) { continue; } for (int iR = 0;iR < ap->get_R_size();++iR) { // R index may be different between the two HContainers, find by R value instead of R-index @@ -32,6 +35,39 @@ namespace LR_Util } } + void get_DMR_real_imag_part(const module_dm::DensityMatrix, std::complex>& DMR, + module_dm::DensityMatrix, double>& DMR_real, + const int& nat, + const int& is, + const char& type) + { + assert(is < DMR.get_dmr_vec().size()); + assert(DMR_real.get_dmr_vec().size() == 1); + bool get_imag = (type == 'I' || type == 'i'); + auto dr = DMR.get_dmr_vec()[is]; //get_dmr_ptr() has bug when is=0 + auto dr_real = DMR_real.get_dmr_vec()[0]; + assert(dr != nullptr); + assert(dr_real != nullptr); + for (int ia = 0;ia < nat;ia++) { + for (int ja = 0;ja < nat;ja++) + { + auto ap = dr->find_pair(ia, ja); + auto ap_real = dr_real->find_pair(ia, ja); + // under MPI-parallel (2D block-cyclic) HContainer, an atom pair not owned by this rank + // is absent from find_pair() and returns nullptr here; skip it + if (!ap || !ap_real) { continue; } + for (int iR = 0;iR < ap->get_R_size();++iR) + { + // R index may be different between the two HContainers, find by R value instead of R-index + auto dR = ap->get_R_index(iR); + auto ptr = ap->get_HR_values(iR).get_pointer(); + auto ptr_real = ap_real->get_HR_values(dR.x, dR.y, dR.z).get_pointer(); + for (int i = 0;i < ap->get_size();++i) { ptr_real[i] = (get_imag ? ptr[i].imag() : ptr[i].real()); } + } + } + } + } + void set_HR_real_imag_part(const hamilt::HContainer& HR_real, hamilt::HContainer>& HR, const int& nat, @@ -43,6 +79,9 @@ namespace LR_Util { auto ap = HR.find_pair(ia, ja); auto ap_real = HR_real.find_pair(ia, ja); + // under MPI-parallel (2D block-cyclic) HContainer, an atom pair not owned by this rank + // is absent from find_pair() and returns nullptr here; skip it + if (!ap || !ap_real) { continue; } for (int iR = 0;iR < ap->get_R_size();++iR) { // R index may be different between the two HContainers, find by R value instead of R-index diff --git a/source/source_lcao/module_lr/utils/lr_util_hcontainer.h b/source/source_lcao/module_lr/utils/lr_util_hcontainer.h index e75b13ff437..2c5fa6bb7f9 100644 --- a/source/source_lcao/module_lr/utils/lr_util_hcontainer.h +++ b/source/source_lcao/module_lr/utils/lr_util_hcontainer.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_LCAO_MODULE_LR_UTILS_LR_UTIL_HCONTAINER_H +#define ABACUS_SOURCE_LCAO_MODULE_LR_UTILS_LR_UTIL_HCONTAINER_H + #include "source_estate/module_dm/density_matrix.h" #include #include "source_base/parallel_reduce.h" @@ -31,16 +33,23 @@ namespace LR_Util } } template - void print_DMR(const elecstate::DensityMatrix& DMR, const int& nat, const std::string& label, const double& threshold = 1e-10) + void print_DMR(const module_dm::DensityMatrix& DMR, const int& nat, const std::string& label, const double& threshold = 1e-10) { std::cout << label << "\n"; int is = 0; - for (auto& dr : DMR.get_DMR_vector()) + for (auto& dr : DMR.get_dmr_vec()) print_HR(*dr, nat, "DMR[" + std::to_string(is++) + "]", threshold); } - void get_DMR_real_imag_part(const elecstate::DensityMatrix, std::complex>& DMR, - elecstate::DensityMatrix, double>& DMR_real, + void get_DMR_real_imag_part(const module_dm::DensityMatrix, std::complex>& DMR, + module_dm::DensityMatrix, double>& DMR_real, + const int& nat, + const char& type = 'R'); + /// overload: only copy the `is`-th spin channel of DMR (source) into the (single-channel) DMR_real, + /// to avoid mixing/overlapping spin channels when DMR has more than one spin channel + void get_DMR_real_imag_part(const module_dm::DensityMatrix, std::complex>& DMR, + module_dm::DensityMatrix, double>& DMR_real, const int& nat, + const int& is, const char& type = 'R'); void set_HR_real_imag_part(const hamilt::HContainer& HR_real, hamilt::HContainer>& HR, @@ -78,7 +87,7 @@ namespace LR_Util if (std::is_same::value) { hR.fix_gamma(); } } template - void initialize_DMR(elecstate::DensityMatrix& dm, + void initialize_DMR(module_dm::DensityMatrix& dm, const Parallel_Orbitals& pmat, const UnitCell& ucell, const Grid_Driver& gd, @@ -86,7 +95,7 @@ namespace LR_Util { hamilt::HContainer hR_tmp(&pmat); initialize_HR(hR_tmp, ucell, gd, orb_cutoff); - dm.init_DMR(hR_tmp); + dm.init_dmr(hR_tmp); } /// $\sum_{uvR} H1_{uv}(R) H2_{uv}(R)$ @@ -118,4 +127,6 @@ namespace LR_Util return sum; } -} \ No newline at end of file +} + +#endif // ABACUS_SOURCE_LCAO_MODULE_LR_UTILS_LR_UTIL_HCONTAINER_H diff --git a/source/source_lcao/module_lr/utils/lr_util_print.h b/source/source_lcao/module_lr/utils/lr_util_print.h index 789306b9bd9..0d46af613b9 100644 --- a/source/source_lcao/module_lr/utils/lr_util_print.h +++ b/source/source_lcao/module_lr/utils/lr_util_print.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_LCAO_MODULE_LR_UTILS_LR_UTIL_PRINT_H +#define ABACUS_SOURCE_LCAO_MODULE_LR_UTILS_LR_UTIL_PRINT_H + #include #include "source_psi/psi.h" #ifdef __EXX @@ -245,4 +247,6 @@ namespace LR_Util } } #endif -} \ No newline at end of file +} + +#endif // ABACUS_SOURCE_LCAO_MODULE_LR_UTILS_LR_UTIL_PRINT_H diff --git a/source/source_lcao/module_lr/utils/lr_util_xc.hpp b/source/source_lcao/module_lr/utils/lr_util_xc.hpp index a270482ba7a..e8992bfd429 100644 --- a/source/source_lcao/module_lr/utils/lr_util_xc.hpp +++ b/source/source_lcao/module_lr/utils/lr_util_xc.hpp @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_LCAO_MODULE_LR_UTILS_LR_UTIL_XC_HPP +#define ABACUS_SOURCE_LCAO_MODULE_LR_UTILS_LR_UTIL_XC_HPP + #include "lr_util.h" namespace LR_Util { @@ -46,3 +48,5 @@ namespace LR_Util lapl(rhor.data(), lapn.data(), rho_basis, tpiba2); } } + +#endif // ABACUS_SOURCE_LCAO_MODULE_LR_UTILS_LR_UTIL_XC_HPP diff --git a/source/source_lcao/module_lr/utils/mo_type.h b/source/source_lcao/module_lr/utils/mo_type.h index 5f38f527e89..6d4e825a3f6 100644 --- a/source/source_lcao/module_lr/utils/mo_type.h +++ b/source/source_lcao/module_lr/utils/mo_type.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_LCAO_MODULE_LR_UTILS_MO_TYPE_H +#define ABACUS_SOURCE_LCAO_MODULE_LR_UTILS_MO_TYPE_H + #include namespace LR_Util { @@ -40,4 +42,6 @@ inline void set_dim(const MO_TYPE type, const int& nocc, const int& nvirt, throw std::runtime_error("Error in LR::set_dim: unknown MO_TYPE"); } } -} \ No newline at end of file +} + +#endif // ABACUS_SOURCE_LCAO_MODULE_LR_UTILS_MO_TYPE_H diff --git a/source/source_lcao/module_lr/utils/spectrum_mo.hpp b/source/source_lcao/module_lr/utils/spectrum_mo.hpp index 4a19ffa6912..ae3818331a5 100644 --- a/source/source_lcao/module_lr/utils/spectrum_mo.hpp +++ b/source/source_lcao/module_lr/utils/spectrum_mo.hpp @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_LCAO_MODULE_LR_UTILS_SPECTRUM_MO_HPP +#define ABACUS_SOURCE_LCAO_MODULE_LR_UTILS_SPECTRUM_MO_HPP + #include "source_base/tool_title.h" #include "source_basis/module_nao/two_center_bundle.h" #include "source_cell/klist.h" @@ -203,4 +205,6 @@ inline void output_spectrum_mo_librpa(const std::vector>& o } } } -} \ No newline at end of file +} + +#endif // ABACUS_SOURCE_LCAO_MODULE_LR_UTILS_SPECTRUM_MO_HPP diff --git a/source/source_lcao/module_operator_lcao/deepks_lcao.cpp b/source/source_lcao/module_operator_lcao/deepks_lcao.cpp index b97c752d5ba..4e273aba26e 100644 --- a/source/source_lcao/module_operator_lcao/deepks_lcao.cpp +++ b/source/source_lcao/module_operator_lcao/deepks_lcao.cpp @@ -24,7 +24,7 @@ DeePKS>::DeePKS(HS_Matrix_K* hsk_in, const TwoCenterIntegrator* intor_orb_alpha, const LCAO_Orbitals* ptr_orb, const int& nks_in, - elecstate::DensityMatrix* DM_in + module_dm::DensityMatrix* DM_in #ifdef __MLALGO , LCAO_Deepks* ld_in diff --git a/source/source_lcao/module_operator_lcao/deepks_lcao.h b/source/source_lcao/module_operator_lcao/deepks_lcao.h index dd8a2b937e4..8916ba62e9d 100644 --- a/source/source_lcao/module_operator_lcao/deepks_lcao.h +++ b/source/source_lcao/module_operator_lcao/deepks_lcao.h @@ -39,7 +39,7 @@ class DeePKS> : public OperatorLCAO const TwoCenterIntegrator* intor_orb_alpha, const LCAO_Orbitals* ptr_orb, const int& nks_in, - elecstate::DensityMatrix* DM_in + module_dm::DensityMatrix* DM_in #ifdef __MLALGO , LCAO_Deepks* ld_in @@ -67,7 +67,7 @@ class DeePKS> : public OperatorLCAO #endif private: - elecstate::DensityMatrix* DM; + module_dm::DensityMatrix* DM; const UnitCell* ucell = nullptr; Grid_Driver* gridD = nullptr; diff --git a/source/source_lcao/module_operator_lcao/dspin_lcao.cpp b/source/source_lcao/module_operator_lcao/dspin_lcao.cpp index 6d487b2fe21..8f8e975587b 100644 --- a/source/source_lcao/module_operator_lcao/dspin_lcao.cpp +++ b/source/source_lcao/module_operator_lcao/dspin_lcao.cpp @@ -228,8 +228,21 @@ void hamilt::DeltaSpin>::cal_pre_HR() } this->paraV = this->hR->get_paraV(); ModuleBase::timer::start("DeltaSpin", "cal_pre_HR"); + // Reset the caches from a possible previous initialization: + // pre_hr holds raw HContainer pointers, so clear() alone would leak them, + // and stale B_I_data would survive for atoms that are no longer constrained. + for (auto& hr : this->pre_hr) + { + delete hr; + } this->pre_hr.clear(); this->pre_hr.resize(this->ucell->nat, nullptr); + for (auto& b : this->B_I_data) + { + b.clear(); + } + this->B_I_data.clear(); + this->B_I_data.resize(this->ucell->nat); const int npol = this->ucell->get_npol(); size_t memory_cost = 0; diff --git a/source/source_lcao/module_operator_lcao/op_exx_lcao.cpp b/source/source_lcao/module_operator_lcao/op_exx_lcao.cpp index 619e6462dd5..5a6c6769830 100644 --- a/source/source_lcao/module_operator_lcao/op_exx_lcao.cpp +++ b/source/source_lcao/module_operator_lcao/op_exx_lcao.cpp @@ -10,6 +10,8 @@ #include "source_lcao/module_ri/exx_lri_interface.h" #include "source_lcao/module_ri/ri_2d_comm.h" +#include + namespace hamilt { RI::Cell_Nearest init_cell_nearest(const UnitCell& ucell, const std::array& Rs_period) @@ -222,16 +224,27 @@ OperatorEXX>::OperatorEXX(HS_Matrix_K* hsk_in, // 2. read DM const int nspin_dm = (PARAM.inp.nspin == 2) ? 2 : 1; + // dmR_owner owns the HContainers (RAII); dmR_vec is a non-owning view + // passed to dm_container_to_Ds which expects raw pointers. + std::vector>> dmR_owner(nspin_dm); std::vector*> dmR_vec(nspin_dm); for (int is = 0; is < nspin_dm; ++is) { + // global_readin_dir is normalized by to_dir() and always ends with '/' const std::string dmfile - = PARAM.globalv.global_readin_dir + "/dmrs" + std::to_string(is + 1) + "_nao.csr"; - dmR_vec[is] = new hamilt::HContainer(const_cast(pv)); + = PARAM.globalv.global_readin_dir + "dmrs" + std::to_string(is + 1) + "_nao.csr"; + // EXX-specific: add rank guard because OperatorEXX is constructed on all MPI ranks, + // unlike most other places where ofs_running is only written on rank 0 + if (GlobalV::MY_RANK == 0) + { + GlobalV::ofs_running << " Read density matrix for EXX from " << dmfile << std::endl; + } + dmR_owner[is] = std::unique_ptr>( + new hamilt::HContainer(const_cast(pv))); + dmR_vec[is] = dmR_owner[is].get(); hamilt::Read_HContainer reader_dm(dmR_vec[is], dmfile, PARAM.globalv.nlocal, &ucell, GlobalV::MY_RANK); reader_dm.read(); } - // 3. DM->Ds->Hexx (do not use symmetry for nscf) XC_Functional::set_xc_type(ucell.atoms[0].ncpp.xc_func); if (exx_info_ptr->info_ri.real_number) @@ -592,35 +605,6 @@ void OperatorEXX>::contributeHk(int ik) } } -template -template -void OperatorEXX>::cal_dH( - const int ispin, - std::array*>, 3>& dhR, - const std::array>>>>, 3>& dHexxs) -{ - // dhR is the set of per-atom-I HContainers to fill (not this->hR, which may be a dummy here). - const Parallel_Orbitals* const paraV = dhR[0][0]->get_paraV(); - const RI::Cell_Nearest* const cell_nearest - = this->use_cell_nearest ? &this->cell_nearest : nullptr; - for (int idir = 0; idir < 3; ++idir) - { - for (int iat = 0; iat < ucell.nat; ++iat) - { - // add_HexxR only fills existing matrices, so first allocate the atom-pair - // structure of this per-I container from the exx-form data (same cell mapping). - reallocate_hcontainer(dHexxs[idir][iat], dhR[idir][iat], cell_nearest); - RI_2D_Comm::add_HexxR(ispin, - exx_info_ptr->info_global.hybrid_alpha, - dHexxs[idir][iat], - *paraV, - PARAM.globalv.npol, - *dhR[idir][iat], - cell_nearest); - } - } -} - // explicit member function instantiations for constructors template OperatorEXX>::OperatorEXX( HS_Matrix_K*, HContainer*, const UnitCell&, const K_Vectors&, @@ -668,26 +652,6 @@ template void OperatorEXX>::contributeHk(int); template void OperatorEXX, double>>::contributeHk(int); template void OperatorEXX, std::complex>>::contributeHk(int); -// explicit member function instantiations for cal_dH (template member function) -template void OperatorEXX>::cal_dH( - const int, std::array*>, 3>&, - const std::array>>>>, 3>&); -template void OperatorEXX>::cal_dH>( - const int, std::array*>, 3>&, - const std::array>>>>>, 3>&); -template void OperatorEXX, double>>::cal_dH( - const int, std::array*>, 3>&, - const std::array>>>>, 3>&); -template void OperatorEXX, double>>::cal_dH>( - const int, std::array*>, 3>&, - const std::array>>>>>, 3>&); -template void OperatorEXX, std::complex>>::cal_dH( - const int, std::array*>, 3>&, - const std::array>>>>, 3>&); -template void OperatorEXX, std::complex>>::cal_dH>( - const int, std::array*>, 3>&, - const std::array>>>>>, 3>&); - // explicit instantiations for reallocate_hcontainer (first overload) template void reallocate_hcontainer( const std::vector>>>&, diff --git a/source/source_lcao/module_operator_lcao/op_exx_lcao.h b/source/source_lcao/module_operator_lcao/op_exx_lcao.h index 72451fef685..c1e792aa277 100644 --- a/source/source_lcao/module_operator_lcao/op_exx_lcao.h +++ b/source/source_lcao/module_operator_lcao/op_exx_lcao.h @@ -67,11 +67,6 @@ class OperatorEXX> : public OperatorLCAO virtual void contributeHk(int ik) override; virtual void contributeHR() override; - template - void cal_dH(const int ispin, - std::array*>, 3>& dhR, - const std::array>>>>, 3>& dHexxs); - private: Add_Hexx_Type add_hexx_type = Add_Hexx_Type::R; int current_spin = 0; diff --git a/source/source_lcao/module_operator_lcao/operator_lcao.cpp b/source/source_lcao/module_operator_lcao/operator_lcao.cpp index f0560e69492..d98ce868d4e 100644 --- a/source/source_lcao/module_operator_lcao/operator_lcao.cpp +++ b/source/source_lcao/module_operator_lcao/operator_lcao.cpp @@ -78,6 +78,14 @@ void OperatorLCAO::init(const int ik_in) { this->hR->set_zero(); } } + // propagate current_spin to next operator so all nodes in the chain + // share the same spin state, set by HamiltLCAO::updateHk via set_current_spin. + // This is done before processing this node so that children receive the + // correct spin regardless of any local toggling that may happen inside + // this operator's contributeHR(). + if (this->next_op != nullptr) { + dynamic_cast*>(this->next_op)->current_spin = this->current_spin; + } switch (this->cal_type) { case calculation_type::lcao_overlap: { // cal_type=lcao_overlap refer to overlap matrix operators, which are @@ -254,7 +262,7 @@ void OperatorLCAO::contributeHk(int ik) { const int nrow = this->hsk->get_pv()->get_row_size(); if(PARAM.inp.td_stype == 2) { - module_rt::folding_HR_td(*this->hR, this->hsk->get_hk(), this->kvec_d[ik], TD_info::cart_At, TD_info::td_vel_op->get_phase_hybrid(), nrow, 1); + module_rt::folding_HR_td(*this->hR, this->hsk->get_hk(), this->kvec_d[ik], TD_info::A_prop_ha, TD_info::td_vel_op->get_phase_hybrid(), nrow, 1); } else { @@ -266,7 +274,7 @@ void OperatorLCAO::contributeHk(int ik) { const int ncol = this->hsk->get_pv()->get_col_size(); if(PARAM.inp.td_stype == 2) { - module_rt::folding_HR_td(*this->hR, this->hsk->get_hk(), this->kvec_d[ik], TD_info::cart_At, TD_info::td_vel_op->get_phase_hybrid(), ncol, 0); + module_rt::folding_HR_td(*this->hR, this->hsk->get_hk(), this->kvec_d[ik], TD_info::A_prop_ha, TD_info::td_vel_op->get_phase_hybrid(), ncol, 0); } else { diff --git a/source/source_lcao/module_operator_lcao/overlap.cpp b/source/source_lcao/module_operator_lcao/overlap.cpp index bed895eee20..6b206c60a01 100644 --- a/source/source_lcao/module_operator_lcao/overlap.cpp +++ b/source/source_lcao/module_operator_lcao/overlap.cpp @@ -290,7 +290,7 @@ void Overlap>::contributeHk(int ik) const int nrow = this->SR->get_atom_pair(0).get_paraV()->get_row_size(); if(PARAM.inp.td_stype == 2) { - module_rt::folding_HR_td(*this->SR, this->hsk->get_sk(), this->kvec_d[ik], TD_info::cart_At, TD_info::td_vel_op->get_phase_hybrid(), nrow, 1); + module_rt::folding_HR_td(*this->SR, this->hsk->get_sk(), this->kvec_d[ik], TD_info::A_prop_ha, TD_info::td_vel_op->get_phase_hybrid(), nrow, 1); } else { @@ -302,7 +302,7 @@ void Overlap>::contributeHk(int ik) const int ncol = this->SR->get_atom_pair(0).get_paraV()->get_col_size(); if(PARAM.inp.td_stype == 2) { - module_rt::folding_HR_td(*this->SR, this->hsk->get_sk(), this->kvec_d[ik], TD_info::cart_At, TD_info::td_vel_op->get_phase_hybrid(), ncol, 0); + module_rt::folding_HR_td(*this->SR, this->hsk->get_sk(), this->kvec_d[ik], TD_info::A_prop_ha, TD_info::td_vel_op->get_phase_hybrid(), ncol, 0); } else { diff --git a/source/source_lcao/module_operator_lcao/td_ekinetic_lcao.cpp b/source/source_lcao/module_operator_lcao/td_ekinetic_lcao.cpp index 7a74fd4e336..35987012d85 100644 --- a/source/source_lcao/module_operator_lcao/td_ekinetic_lcao.cpp +++ b/source/source_lcao/module_operator_lcao/td_ekinetic_lcao.cpp @@ -215,11 +215,11 @@ void TDEkinetic>::cal_HR_IJR(const int& iat1, hr_mat_p += (npol - 1) * col_indexes.size(); if (current_mat_p != nullptr) { - for (int dir = 0; dir < 3; dir++) - { - current_mat_p[dir] += (npol - 1) * col_indexes.size(); - } - } + for (int dir = 0; dir < 3; dir++) + { + current_mat_p[dir] += (npol - 1) * col_indexes.size(); + } + } } } //update vector potential for td_ekintic term @@ -227,7 +227,7 @@ template void TDEkinetic>::update_td() { //std::cout<<"velocity"<cart_At = TD_info::cart_At; + this->cart_At = TD_info::A_prop_ha; } template diff --git a/source/source_lcao/module_operator_lcao/td_nonlocal_lcao.cpp b/source/source_lcao/module_operator_lcao/td_nonlocal_lcao.cpp index 4b1bd6c92a9..f045ba5939b 100644 --- a/source/source_lcao/module_operator_lcao/td_nonlocal_lcao.cpp +++ b/source/source_lcao/module_operator_lcao/td_nonlocal_lcao.cpp @@ -52,7 +52,7 @@ template void hamilt::TDNonlocal>::update_td() { // calculate At in cartesian coorinates. - this->cart_At = TD_info::cart_At; + this->cart_At = TD_info::A_prop_ha; } // initialize_HR() template diff --git a/source/source_lcao/module_operator_lcao/td_pot_hybrid.cpp b/source/source_lcao/module_operator_lcao/td_pot_hybrid.cpp index 470bc384691..47198c25e40 100644 --- a/source/source_lcao/module_operator_lcao/td_pot_hybrid.cpp +++ b/source/source_lcao/module_operator_lcao/td_pot_hybrid.cpp @@ -237,9 +237,9 @@ template void hamilt::TD_pot_hybrid>::update_td() { //std::cout<<"hybrid gauge" <cart_At = TD_info::cart_At; + this->cart_At = TD_info::A_prop_ha; //std::cout<<"At: "<< TD_info::td_vel_op->cart_At[0] <<" "<cart_At[1]<<" "<cart_At[2]<<" "< diff --git a/source/source_lcao/module_operator_lcao/td_pot_hybrid.h b/source/source_lcao/module_operator_lcao/td_pot_hybrid.h index 563ffc956c9..9b82e9b75e2 100644 --- a/source/source_lcao/module_operator_lcao/td_pot_hybrid.h +++ b/source/source_lcao/module_operator_lcao/td_pot_hybrid.h @@ -8,7 +8,7 @@ #include "source_lcao/module_operator_lcao/operator_lcao.h" #include "source_hamilt/module_hcontainer/hcontainer.h" #include -#include "source_io/module_hs/cal_r_overlap_r.h" +#include "source_io/module_hs/pos_op_mat.h" namespace hamilt { @@ -85,7 +85,7 @@ class TD_pot_hybrid> : public OperatorLCAO bool HR_fixed_done = false; //tddft part - cal_r_overlap_R* r_calculator; + Position_op* r_calculator; //ETD //std::vector> hk_hybrid; //ETD diff --git a/source/source_lcao/module_operator_lcao/td_pot_hybrid_fs.cpp b/source/source_lcao/module_operator_lcao/td_pot_hybrid_fs.cpp index 1f2722a598c..ecc395b11ca 100644 --- a/source/source_lcao/module_operator_lcao/td_pot_hybrid_fs.cpp +++ b/source/source_lcao/module_operator_lcao/td_pot_hybrid_fs.cpp @@ -15,7 +15,7 @@ void TD_pot_hybrid>::cal_force_stress(const bool cal_force, { return; } - Et = elecstate::H_TDDFT_pw::Et; + Et = elecstate::H_TDDFT_pw::efield_ha * 2.0; // Hamiltonian and forces use Rydberg units. const Parallel_Orbitals* paraV = dmR->get_paraV(); #pragma omp parallel { @@ -201,4 +201,4 @@ template void TD_pot_hybrid, double>>::cal_for double* dmR_pointer, double* force1, double* force2); -}// namespace hamilt \ No newline at end of file +}// namespace hamilt diff --git a/source/source_lcao/module_operator_lcao/test/test_nonlocal.cpp b/source/source_lcao/module_operator_lcao/test/test_nonlocal.cpp index c03716445fb..5eea2ba121a 100644 --- a/source/source_lcao/module_operator_lcao/test/test_nonlocal.cpp +++ b/source/source_lcao/module_operator_lcao/test/test_nonlocal.cpp @@ -34,7 +34,7 @@ class NonlocalTest : public ::testing::Test // set up a unitcell, with one element and test_size atoms, each atom has test_nw orbitals ucell.ntype = 1; auto* lcao_nl = new LCAONonlocalInfo(); - lcao_nl->get_nonlocal().Beta = new Numerical_Nonlocal[ucell.ntype]; + lcao_nl->get_nonlocal().resize_Beta(ucell.ntype); ucell.infoNL.reset(lcao_nl); ucell.nat = test_size; ucell.atoms = new Atom[ucell.ntype]; diff --git a/source/source_lcao/module_operator_lcao/test/test_t_nl_cd.cpp b/source/source_lcao/module_operator_lcao/test/test_t_nl_cd.cpp index 972407d01fd..5b9e6c1c572 100644 --- a/source/source_lcao/module_operator_lcao/test/test_t_nl_cd.cpp +++ b/source/source_lcao/module_operator_lcao/test/test_t_nl_cd.cpp @@ -60,7 +60,7 @@ class TNLTest : public ::testing::Test ucell.set_iat2iwt(2); // for Nonlocal auto* lcao_nl = new LCAONonlocalInfo(); - lcao_nl->get_nonlocal().Beta = new Numerical_Nonlocal[ucell.ntype]; + lcao_nl->get_nonlocal().resize_Beta(ucell.ntype); ucell.infoNL.reset(lcao_nl); ucell.atoms[0].ncpp.d_real.create(5, 5); ucell.atoms[0].ncpp.d_real.zero_out(); diff --git a/source/source_lcao/module_operator_lcao/test/tmp_mocks.cpp b/source/source_lcao/module_operator_lcao/test/tmp_mocks.cpp index 9e7ce8dce7e..1dfa71c0b75 100644 --- a/source/source_lcao/module_operator_lcao/test/tmp_mocks.cpp +++ b/source/source_lcao/module_operator_lcao/test/tmp_mocks.cpp @@ -202,15 +202,10 @@ Numerical_Nonlocal::~Numerical_Nonlocal() {} #include "../../setup_nonlocal.h" InfoNonlocal::InfoNonlocal() { - this->Beta = new Numerical_Nonlocal[1]; - this->nproj = nullptr; this->nprojmax = 0; this->rcutmax_Beta = 0.0; } -InfoNonlocal::~InfoNonlocal() { - delete[] Beta; - delete[] nproj; -} +InfoNonlocal::~InfoNonlocal() = default; Numerical_Orbital::Numerical_Orbital() { this->rcut = 1.0; } Numerical_Orbital::~Numerical_Orbital() {} @@ -226,7 +221,7 @@ class TD_info { public: TD_info(): phase_hybrid{} {td_vel_op = this;} ~TD_info() {} - static ModuleBase::Vector3 cart_At; + static ModuleBase::Vector3 A_prop_ha; static TD_info* td_vel_op; std::map, std::complex> phase_hybrid; const std::map, std::complex>& get_phase_hybrid() const @@ -234,7 +229,7 @@ class TD_info { return this->phase_hybrid; } }; -ModuleBase::Vector3 TD_info::cart_At(0.0, 0.0, 0.0); +ModuleBase::Vector3 TD_info::A_prop_ha(0.0, 0.0, 0.0); TD_info* TD_info::td_vel_op = nullptr; -static TD_info mock_td_info; +static TD_info mock_td_info; diff --git a/source/source_lcao/module_operator_lcao/veff_dh.cpp b/source/source_lcao/module_operator_lcao/veff_dh.cpp index 34eea550af1..418c8a9040f 100644 --- a/source/source_lcao/module_operator_lcao/veff_dh.cpp +++ b/source/source_lcao/module_operator_lcao/veff_dh.cpp @@ -176,7 +176,7 @@ void Veff>::cal_dH(std::array(rho_basis)); - chr.allocate(PARAM.inp.nspin, false); + chr.allocate(PARAM.inp.nspin, false, false, PARAM.inp.test_charge); // cal_force_loc returns the local Hellmann-Feynman force on every atom: // F_I = -Omega * sum_G e^{iG.tau_I} iG . V^{L,Z_I}(G) rho*(G) @@ -325,7 +325,7 @@ void Veff>::cal_dH(std::array>::cal_dH(std::arraynrxx; // finite-difference XC: delta V^XC(r) = V^XC[rho0 + drho](r) - V^XC[rho0](r) - elecstate::PotXC_FDM dvxcr_fdm_op(rho_basis, chg, this->ucell); + elecstate::PotXC_FDM dvxcr_fdm_op(chg->nspin, PARAM.globalv.domag, PARAM.globalv.domag_z, + PARAM.inp.gga_grad, PARAM.inp.out_elf[0] > 0, PARAM.inp.test_charge, + rho_basis, chg, this->ucell); std::vector chg_drho(3); for (int d = 0; d < 3; ++d) { chg_drho[d].set_rhopw(const_cast(rho_basis)); - chg_drho[d].allocate(chg->nspin, false); + chg_drho[d].allocate(chg->nspin, false, false, PARAM.inp.test_charge); } diff --git a/source/source_lcao/module_rdmft/rdmft.cpp b/source/source_lcao/module_rdmft/rdmft.cpp index 5a27d9d86f8..22883e46d6d 100644 --- a/source/source_lcao/module_rdmft/rdmft.cpp +++ b/source/source_lcao/module_rdmft/rdmft.cpp @@ -158,8 +158,8 @@ void RDMFT::init(Parallel_Orbitals& ParaV_in, { const std::array& period = RI_Util::get_Born_vonKarmen_period(*kv); this->symrot_exx.find_irreducible_sector(ucell->symm, ucell->atoms, ucell->st, - RI_Util::get_Born_von_Karmen_cells(period), period, ucell->lat); - this->symrot_exx.cal_Ms(*kv, *ucell, *ParaV); + RI_Util::get_Born_von_Karmen_cells(period), period, ucell->lat, PARAM.globalv.global_out_dir); + this->symrot_exx.cal_Ms(*kv, *ucell, *ParaV, nspin); } if (this->exx_info_->info_ri.real_number) @@ -345,7 +345,11 @@ void RDMFT::cal_Energy(const int cal_type) { this->pelec->f_en.deband = this->pelec->cal_delta_eband(*ucell); E_descf = pelec->f_en.descf = 0.0; - this->pelec->cal_energies(2); + this->pelec->cal_energies(2, + PARAM.inp.imp_sol, + PARAM.inp.sc_mag_switch, + PARAM.inp.dft_plus_u, + PARAM.inp.assume_isolated); Etotal = this->pelec->f_en.etot; } diff --git a/source/source_lcao/module_rdmft/rdmft_pot.cpp b/source/source_lcao/module_rdmft/rdmft_pot.cpp index 1f692833aad..604b217d819 100644 --- a/source/source_lcao/module_rdmft/rdmft_pot.cpp +++ b/source/source_lcao/module_rdmft/rdmft_pot.cpp @@ -6,7 +6,7 @@ #include "rdmft.h" #include "source_lcao/module_rdmft/rdmft_tools.h" #include "source_psi/psi.h" -#include "source_estate/module_dm/cal_dm_psi.h" +#include "source_estate/module_dm/dm_from_psi.h" #ifdef __EXX #include "source_lcao/module_ri/ri_2d_comm.h" @@ -25,23 +25,14 @@ namespace rdmft template void RDMFT::get_DM_XC(std::vector< std::vector >& DM_XC) { - // get wk_funEta_wfc = wk*g(eta)*conj(wfc) - psi::Psi wk_funEta_wfc(wfc); - conj_psi(wk_funEta_wfc); - occNum_MulPsi(ParaV, wk_fun_occNum, wk_funEta_wfc, 0); - - // get the special DM_XC used in constructing V_exx_XC + // get the special DM_XC used in constructing V_exx_XC. + // wk_fun_occNum holds wk*g(eta); dmk_from_psi applies the conjugation + // and the band weighting internally (g(eta) is used with symbol = 0, + // where occNum_func is the identity, so wk_fun_occNum is the weight itself). for(int ik=0; ikdesc_wfc, ParaV->desc); -#else - elecstate::psiMulPsi(wk_funEta_wfc, wfc, DM_Kpointer); -#endif + module_dm::dmk_from_psi(ParaV, wk_fun_occNum, ik, wfc, DM_Kpointer); } } @@ -164,17 +155,17 @@ void RDMFT::cal_V_XC(const UnitCell& ucell) // // //test // DM_XC_pass = DM_XC; - // elecstate::DensityMatrix DM_test(ParaV, nspin, kv->kvec_d, nk_total); - // elecstate::cal_dm_psi(ParaV, wg, wfc, DM_test); - // DM_test.init_DMR(this->gd, this->ucell); - // DM_test.cal_DMR(); + // module_dm::DensityMatrix DM_test(ParaV, nspin, kv->kvec_d, nk_total); + // module_dm::dm_from_psi(ParaV, wg, wfc, DM_test); + // DM_test.init_dmr(this->gd, this->ucell); + // DM_test.cal_dmr(-1); // // compare DM_XC and DM get in update_charge(or ABACUS) // std::cout << "\n\ntest DM_XC - DM in ABACUS: \n" << std::endl; // double DM_XC_minus_DMtest = 0.0; // for(int ik=0; iknloc; ++iloc) // { // double test = std::abs(DM_XC[ik][iloc] - dmk_pointer[iloc]); diff --git a/source/source_lcao/module_rdmft/rdmft_tools.cpp b/source/source_lcao/module_rdmft/rdmft_tools.cpp index 84998880611..7e72251cf1d 100644 --- a/source/source_lcao/module_rdmft/rdmft_tools.cpp +++ b/source/source_lcao/module_rdmft/rdmft_tools.cpp @@ -24,10 +24,7 @@ namespace rdmft { template <> -void conj_psi(psi::Psi& wfc) {} - -template <> -void HkPsi(const Parallel_Orbitals* ParaV, +void HkPsi(const Parallel_Orbitals* ParaV, const double& HK, const double& wfc, double& H_wfc) @@ -281,7 +278,7 @@ void Veff_rdmft, double>::contributeHR() ModuleBase::matrix vofk = *vloc_; vofk.zero_out(); ModuleBase::matrix v_matrix_XC(this->nspin, charge_->nrxx); - elecstate::PotXC potXC(rho_basis_, etxc, vtxc, &vofk); + elecstate::PotXC potXC(PARAM.globalv.domag, PARAM.globalv.domag_z, PARAM.inp.gga_grad, rho_basis_, etxc, vtxc, &vofk); potXC.cal_v_eff(charge_, ucell, v_matrix_XC); // if need meta-GGA, go to study veff_lcao.cpp and modify the code @@ -363,7 +360,7 @@ void Veff_rdmft::contributeHR() ModuleBase::matrix vofk = *vloc_; vofk.zero_out(); ModuleBase::matrix v_matrix_XC(this->nspin, charge_->nrxx); - elecstate::PotXC potXC(rho_basis_, etxc, vtxc, &vofk); + elecstate::PotXC potXC(PARAM.globalv.domag, PARAM.globalv.domag_z, PARAM.inp.gga_grad, rho_basis_, etxc, vtxc, &vofk); potXC.cal_v_eff(charge_, ucell, v_matrix_XC); for(int is=0; isnspin; ++is) diff --git a/source/source_lcao/module_rdmft/rdmft_tools.h b/source/source_lcao/module_rdmft/rdmft_tools.h index d6400a40b60..9bf240a9214 100644 --- a/source/source_lcao/module_rdmft/rdmft_tools.h +++ b/source/source_lcao/module_rdmft/rdmft_tools.h @@ -15,7 +15,7 @@ #include "source_base/module_external/scalapack_connector.h" #include "source_base/parallel_2d.h" #include "source_basis/module_ao/parallel_orbitals.h" -#include "source_estate/module_dm/cal_dm_psi.h" +#include "source_estate/module_dm/dm_from_psi.h" #include "source_estate/module_dm/density_matrix.h" #include "source_hamilt/module_hcontainer/hcontainer.h" @@ -44,19 +44,6 @@ namespace rdmft double occNum_func(double eta, int symbol = 0, const std::string XC_func_rdmft = "hf", const double alpha_power = 1.0); -template -void conj_psi(psi::Psi& wfc) -{ - TK* pwfc = &wfc(0, 0, 0); - for(int i=0; i -void conj_psi(psi::Psi& wfc); - - // wfc and H_wfc need to be k_firest and provide wfc(ik, 0, 0) and H_wfc(ik, 0, 0) //! implement matrix multiplication of Hk^dagger and psi template diff --git a/source/source_lcao/module_rdmft/update_state_rdmft.cpp b/source/source_lcao/module_rdmft/update_state_rdmft.cpp index 69ca9d2d528..2cb1263e1cb 100644 --- a/source/source_lcao/module_rdmft/update_state_rdmft.cpp +++ b/source/source_lcao/module_rdmft/update_state_rdmft.cpp @@ -5,9 +5,9 @@ #include "rdmft.h" #include "source_lcao/module_rdmft/rdmft_tools.h" -#include "source_estate/module_dm/cal_dm_psi.h" +#include "source_estate/module_dm/dm_from_psi.h" #include "source_estate/module_dm/density_matrix.h" -#include "source_estate/module_charge/symm_rho.h" +#include "source_estate/module_charge/chg_symm.h" #include "source_hamilt/module_gint/gint_interface.h" #include "source_hamilt/module_xc/xc_functional.h" @@ -97,52 +97,51 @@ void RDMFT::update_charge(UnitCell& ucell) if( PARAM.inp.gamma_only ) { // calculate DMK and DMR - elecstate::DensityMatrix DM_gamma_only(ParaV, nspin); - elecstate::cal_dm_psi(ParaV, wg, wfc, DM_gamma_only); - DM_gamma_only.init_DMR(this->gd, &ucell); - DM_gamma_only.cal_DMR(); + module_dm::DensityMatrix DM_gamma_only(ParaV, nspin); + module_dm::dm_from_psi(ParaV, wg, wfc, DM_gamma_only); + DM_gamma_only.init_dmr(this->gd, &ucell); + DM_gamma_only.cal_dmr(-1); for (int is = 0; is < nspin; is++) { ModuleBase::GlobalFunc::ZEROS(charge->rho[is], charge->nrxx); } - ModuleGint::cal_gint_rho(DM_gamma_only.get_DMR_vector(), nspin, charge->rho); + ModuleGint::cal_gint_rho(DM_gamma_only.get_dmr_vec(), nspin, charge->rho); if (XC_Functional::get_ked_flag()) { this->pelec->cal_tau(wfc); } - charge->renormalize_rho(); + charge->renormalize_rho(PARAM.inp.nelec, ucell.omega); } else { // calculate DMK and DMR - elecstate::DensityMatrix DM(ParaV, nspin, kv->kvec_d, nk_total); - elecstate::cal_dm_psi(ParaV, wg, wfc, DM); - DM.init_DMR(this->gd, &ucell); - DM.cal_DMR(); + module_dm::DensityMatrix DM(ParaV, nspin, kv->kvec_d, nk_total); + module_dm::dm_from_psi(ParaV, wg, wfc, DM); + DM.init_dmr(this->gd, &ucell); + DM.cal_dmr(-1); for (int is = 0; is < nspin; is++) { ModuleBase::GlobalFunc::ZEROS(charge->rho[is], charge->nrxx); } - ModuleGint::cal_gint_rho(DM.get_DMR_vector(), nspin, charge->rho); + ModuleGint::cal_gint_rho(DM.get_dmr_vec(), nspin, charge->rho); if (XC_Functional::get_ked_flag()) { this->pelec->cal_tau(wfc); } - charge->renormalize_rho(); + charge->renormalize_rho(PARAM.inp.nelec, ucell.omega); } // charge density symmetrization - Symmetry_rho srho; for (int is = 0; is < nspin; is++) { - srho.begin(is, *(this->charge), rho_basis, ucell.symm); + module_charge::cal_rhog_symm(is, *(this->charge), rho_basis, ucell.symm); } } diff --git a/source/source_lcao/module_ri/CMakeLists.txt b/source/source_lcao/module_ri/CMakeLists.txt index c45ca9751a4..cf9f511eb2c 100644 --- a/source/source_lcao/module_ri/CMakeLists.txt +++ b/source/source_lcao/module_ri/CMakeLists.txt @@ -22,6 +22,7 @@ if (ENABLE_LIBRI) gaussian_abfs.cpp singular_value.cpp exx_lri_detail.cpp + ri_util.cpp ) endif() add_library( diff --git a/source/source_lcao/module_ri/abfs.h b/source/source_lcao/module_ri/abfs.h index d3b774d9734..bdc6b94f79b 100644 --- a/source/source_lcao/module_ri/abfs.h +++ b/source/source_lcao/module_ri/abfs.h @@ -12,8 +12,8 @@ class Abfs { public: - - template class Vector3_Order; + + template class Vector3_Order; }; #endif // ABFS_H diff --git a/source/source_lcao/module_ri/abfs_construct_pca.cpp b/source/source_lcao/module_ri/abfs_construct_pca.cpp index b07e49fb1df..751f9405fa0 100644 --- a/source/source_lcao/module_ri/abfs_construct_pca.cpp +++ b/source/source_lcao/module_ri/abfs_construct_pca.cpp @@ -16,20 +16,20 @@ namespace PCA { template <> void tensor_syev(char jobz, char uplo, RI::Tensor& a, double* w, int& info) - { - assert(a.shape.size() == 2); - assert(a.shape[0] == a.shape[1]); + { + assert(a.shape.size() == 2); + assert(a.shape[0] == a.shape[1]); const int n = a.shape[0]; const int lda = a.shape[1]; double work_query = 0.0; - constexpr int minus_one = -1; + constexpr int minus_one = -1; dsyev_(&jobz, &uplo, &n, a.ptr(), &lda, w, &work_query, &minus_one, &info); const int lwork = static_cast(work_query); - std::vector work(std::max(1, lwork)); + std::vector work(std::max(1, lwork)); dsyev_(&jobz, &uplo, &n, a.ptr(), &lda, w, work.data(), &lwork, &info); } @@ -52,7 +52,7 @@ void tensor_syev(char jobz, char uplo, RI::Tensor& a, float* w, in std::vector work(std::max(1, lwork)); ssyev_(&jobz, &uplo, &n, a.ptr(), &lda, w, work.data(), &lwork, &info); - } + } template <> void tensor_syev>(char jobz, char uplo, RI::Tensor>& a, double* w, int& info) @@ -117,9 +117,9 @@ RI::Tensor get_sub_matrix(const RI::Tensor& m, // size: (lcaos, const std::size_t& L, const ModuleBase::Element_Basis_Index::Range& range, const ModuleBase::Element_Basis_Index::IndexLNM& index) - { - ModuleBase::TITLE("ABFs_Construct::PCA::get_sub_matrix"); - assert(m.shape.size() == 3); + { + ModuleBase::TITLE("ABFs_Construct::PCA::get_sub_matrix"); + assert(m.shape.size() == 3); RI::Tensor m_sub({m.shape[0], m.shape[1], range[T][L].N}); for (std::size_t ir = 0; ir != m.shape[0]; ++ir) { @@ -127,20 +127,20 @@ RI::Tensor get_sub_matrix(const RI::Tensor& m, // size: (lcaos, { for (std::size_t N = 0; N != range[T][L].N; ++N) { - m_sub(ir, jr, N) = m(ir, jr, index[T][L][N][0]); + m_sub(ir, jr, N) = m(ir, jr, index[T][L][N][0]); } } } m_sub = m_sub.reshape({m.shape[0] * m.shape[1], range[T][L].N}); - return m_sub; - } + return m_sub; + } RI::Tensor get_column_mean0_matrix(const RI::Tensor& m) - { - ModuleBase::TITLE("ABFs_Construct::PCA::get_column_mean0_matrix"); + { + ModuleBase::TITLE("ABFs_Construct::PCA::get_column_mean0_matrix"); RI::Tensor m_new(m.shape); for (std::size_t ic = 0; ic != m.shape[1]; ++ic) - { + { double sum = 0; for (std::size_t ir = 0; ir != m.shape[0]; ++ir) { @@ -151,19 +151,19 @@ RI::Tensor get_column_mean0_matrix(const RI::Tensor& m) { m_new(ir, ic) = m(ir, ic) - mean; } - } - return m_new; - } + } + return m_new; + } - std::vector, RI::Tensor>>> cal_PCA( + std::vector, RI::Tensor>>> cal_PCA( const UnitCell& ucell, const LCAO_Orbitals& orb, const std::vector>>& lcaos, const std::vector>>& abfs, const double kmesh_times) - { - ModuleBase::TITLE("ABFs_Construct::PCA::cal_PCA"); - + { + ModuleBase::TITLE("ABFs_Construct::PCA::cal_PCA"); + const ModuleBase::Element_Basis_Index::Range range_lcaos = ModuleBase::Element_Basis_Index::construct_range(lcaos); const ModuleBase::Element_Basis_Index::IndexLNM index_lcaos = ModuleBase::Element_Basis_Index::construct_index(range_lcaos); @@ -172,68 +172,68 @@ RI::Tensor get_column_mean0_matrix(const RI::Tensor& m) const ModuleBase::Element_Basis_Index::IndexLNM index_abfs = ModuleBase::Element_Basis_Index::construct_index(range_abfs); - Matrix_Orbs21 m_abfslcaos_lcaos; + Matrix_Orbs21 m_abfslcaos_lcaos; m_abfslcaos_lcaos.init(abfs, lcaos, lcaos, ucell, orb, kmesh_times); std::map>> delta_R; for (std::size_t it = 0; it != abfs.size(); ++it) { delta_R[it][it] = {0.0}; } - m_abfslcaos_lcaos.init_radial_table(delta_R); - + m_abfslcaos_lcaos.init_radial_table(delta_R); + std::vector, RI::Tensor>>> eig(abfs.size()); for (std::size_t T = 0; T != abfs.size(); ++T) - { + { const RI::Tensor A = m_abfslcaos_lcaos.cal_overlap_matrix(T, - T, + T, ModuleBase::Vector3{0, 0, 0}, ModuleBase::Vector3{0, 0, 0}, - index_abfs, - index_lcaos, - index_lcaos, - Matrix_Orbs21::Matrix_Order::A2BA1); - - eig[T].resize(abfs[T].size()); + index_abfs, + index_lcaos, + index_lcaos, + Matrix_Orbs21::Matrix_Order::A2BA1); + + eig[T].resize(abfs[T].size()); for (std::size_t L = 0; L != abfs[T].size(); ++L) - { + { const RI::Tensor A_sub = get_sub_matrix(A, T, L, range_abfs, index_abfs); - RI::Tensor mm = A_sub.transpose() * A_sub; - std::vector eig_value(mm.shape[0]); - + RI::Tensor mm = A_sub.transpose() * A_sub; + std::vector eig_value(mm.shape[0]); + int info = 1; tensor_syev('V', 'L', mm, eig_value.data(), info); if (info) - { - std::cout << std::endl << "info_dsyev = " << info << std::endl; + { + std::cout << std::endl << "info_dsyev = " << info << std::endl; auto tensor_print = [](RI::Tensor& m, std::ostream& os, const double threshold) { - for (int ir = 0; ir != m.shape[0]; ++ir) - { - for (int ic = 0; ic != m.shape[1]; ++ic) - { + for (int ir = 0; ir != m.shape[0]; ++ir) + { + for (int ic = 0; ic != m.shape[1]; ++ic) + { if (std::abs(m(ir, ic)) > threshold) { - os << m(ir, ic) << "\t"; + os << m(ir, ic) << "\t"; } else { - os << 0 << "\t"; + os << 0 << "\t"; } - } - os << std::endl; - } - os << std::endl; - }; - tensor_print(mm, GlobalV::ofs_warning, 0.0); + } + os << std::endl; + } + os << std::endl; + }; + tensor_print(mm, GlobalV::ofs_warning, 0.0); std::cout << "in file " << __FILE__ << " line " << __LINE__ << std::endl; - ModuleBase::QUIT(); - } + ModuleBase::QUIT(); + } eig[T][L] = std::make_pair(eig_value, mm); - } - } - - return eig; - } + } + } + + return eig; + } } // namespace PCA } // namespace ABFs_Construct diff --git a/source/source_lcao/module_ri/abfs_construct_pca.h b/source/source_lcao/module_ri/abfs_construct_pca.h index d6e5409710a..9bfc4d36a4c 100644 --- a/source/source_lcao/module_ri/abfs_construct_pca.h +++ b/source/source_lcao/module_ri/abfs_construct_pca.h @@ -16,7 +16,7 @@ namespace ABFs_Construct namespace PCA { extern std::vector, RI::Tensor>>> cal_PCA( - const UnitCell& ucell, + const UnitCell& ucell, const LCAO_Orbitals& orb, const std::vector>>& lcaos, const std::vector>>& abfs, // abfs must be orthonormal diff --git a/source/source_lcao/module_ri/abfs_vector3_order.h b/source/source_lcao/module_ri/abfs_vector3_order.h index 4857770ea0f..d37c35421ed 100644 --- a/source/source_lcao/module_ri/abfs_vector3_order.h +++ b/source/source_lcao/module_ri/abfs_vector3_order.h @@ -15,9 +15,9 @@ template class Abfs::Vector3_Order: public ModuleBase::Vector3 { public: - Vector3_Order(const ModuleBase::Vector3 &v):ModuleBase::Vector3(v){} - Vector3_Order(const T &x,const T &y,const T &z):ModuleBase::Vector3(x,y,z){} - Vector3_Order()=default; + Vector3_Order(const ModuleBase::Vector3 &v):ModuleBase::Vector3(v){} + Vector3_Order(const T &x,const T &y,const T &z):ModuleBase::Vector3(x,y,z){} + Vector3_Order()=default; }; template @@ -26,65 +26,65 @@ bool operator< ( const Abfs::Vector3_Order &v1, const Abfs::Vector3_Order template<> inline bool operator< ( const Abfs::Vector3_Order &v1, const Abfs::Vector3_Order &v2 ) { - // threshold is used when v1 and v2 are approximately equal with double numerical error, - // but a strange bug occurs at LRI_CV::dVws. - // Peize Lin change at 2023.04.29 - if ( v1.x < v2.x ) return true; - else if ( v1.x > v2.x ) return false; - if ( v1.y < v2.y ) return true; - else if ( v1.y > v2.y ) return false; - if ( v1.z < v2.z ) return true; - else if ( v1.z > v2.z ) return false; - return false; + // threshold is used when v1 and v2 are approximately equal with double numerical error, + // but a strange bug occurs at LRI_CV::dVws. + // Peize Lin change at 2023.04.29 + if ( v1.x < v2.x ) return true; + else if ( v1.x > v2.x ) return false; + if ( v1.y < v2.y ) return true; + else if ( v1.y > v2.y ) return false; + if ( v1.z < v2.z ) return true; + else if ( v1.z > v2.z ) return false; + return false; } template<> inline bool operator< ( const Abfs::Vector3_Order &v1, const Abfs::Vector3_Order &v2 ) { - if ( v1.x < v2.x ) return true; - else if ( v1.x > v2.x ) return false; - if ( v1.y < v2.y ) return true; - else if ( v1.y > v2.y ) return false; - if ( v1.z < v2.z ) return true; - else if ( v1.z > v2.z ) return false; - return false; + if ( v1.x < v2.x ) return true; + else if ( v1.x > v2.x ) return false; + if ( v1.y < v2.y ) return true; + else if ( v1.y > v2.y ) return false; + if ( v1.z < v2.z ) return true; + else if ( v1.z > v2.z ) return false; + return false; } /* template bool operator> ( const Abfs::Vector3_Order &v1, const Abfs::Vector3_Order &v2 ) { - if( v1.x>v2.x ) return true; - else if ( v1.xv2.y ) return true; - else if ( v1.yv2.z ) return true; - else if ( v1.zv2.x ) return true; + else if ( v1.xv2.y ) return true; + else if ( v1.yv2.z ) return true; + else if ( v1.z bool operator<= ( const Abfs::Vector3_Order &v1, const Abfs::Vector3_Order &v2 ) { - if( v1.xv2.x ) return false; - if( v1.yv2.y ) return false; - if( v1.zv2.z ) return false; - return true; + if( v1.xv2.x ) return false; + if( v1.yv2.y ) return false; + if( v1.zv2.z ) return false; + return true; } template bool operator>= ( const Abfs::Vector3_Order &v1, const Abfs::Vector3_Order &v2 ) { - if( v1.x>v2.x ) return true; - else if ( v1.xv2.y ) return true; - else if ( v1.yv2.z ) return true; - else if ( v1.zv2.x ) return true; + else if ( v1.xv2.y ) return true; + else if ( v1.yv2.z ) return true; + else if ( v1.z= ( const Abfs::Vector3_Order &v1, const Abfs::Vector3_Order template Abfs::Vector3_Order operator% ( const Abfs::Vector3_Order &v1, const Abfs::Vector3_Order &v2 ) { - auto mod = [](const int i, const int n){ return (i%n+3*n/2)%n-n/2; }; // [-n/2,n/2] + auto mod = [](const int i, const int n){ return (i%n+3*n/2)%n-n/2; }; // [-n/2,n/2] // auto mod = [](const int i, const int n){ return (i%n+n)%n; }; // [0,n] - return Abfs::Vector3_Order{ mod(v1.x,v2.x), mod(v1.y,v2.y), mod(v1.z,v2.z) }; + return Abfs::Vector3_Order{ mod(v1.x,v2.x), mod(v1.y,v2.y), mod(v1.z,v2.z) }; } #endif // ABFS_VECTOR3_ORDER_H diff --git a/source/source_lcao/module_ri/conv_coulomb_pot_k.cpp b/source/source_lcao/module_ri/conv_coulomb_pot_k.cpp index 3970b98f41d..6903b172154 100644 --- a/source/source_lcao/module_ri/conv_coulomb_pot_k.cpp +++ b/source/source_lcao/module_ri/conv_coulomb_pot_k.cpp @@ -5,146 +5,146 @@ namespace Conv_Coulomb_Pot_K { - std::vector cal_psi_fock_limits( - const std::vector & psif) - { - std::vector psik2_ccp(psif.size()); - for( size_t ik=0; ik cal_psi_fock_limits( + const std::vector & psif) + { + std::vector psik2_ccp(psif.size()); + for( size_t ik=0; ik cal_psi_fock_spencer( - const std::vector &psif, - const std::vector &k_radial, - const double rcut) - { - std::vector psik2_ccp(psif.size()); - for (size_t ik = 0; ik < psif.size(); ++ik) - { psik2_ccp[ik] = ModuleBase::FOUR_PI * psif[ik] * (1 - std::cos(k_radial[ik] * rcut)); } - return psik2_ccp; - } + // rongshi add 2022-07-27 + // Sphere truction -- Spencer + std::vector cal_psi_fock_spencer( + const std::vector &psif, + const std::vector &k_radial, + const double rcut) + { + std::vector psik2_ccp(psif.size()); + for (size_t ik = 0; ik < psif.size(); ++ik) + { psik2_ccp[ik] = ModuleBase::FOUR_PI * psif[ik] * (1 - std::cos(k_radial[ik] * rcut)); } + return psik2_ccp; + } - std::vector cal_psi_erfc_limits( - const std::vector & psif, - const std::vector & k_radial, - const double erfc_omega) - { - std::vector psik2_ccp(psif.size()); - for( size_t ik=0; ik cal_psi_erfc_limits( + const std::vector & psif, + const std::vector & k_radial, + const double erfc_omega) + { + std::vector psik2_ccp(psif.size()); + for( size_t ik=0; ik cal_psi_erfc_spencer( - const std::vector& psif, + std::vector cal_psi_erfc_spencer( + const std::vector& psif, const std::vector& k_radial, const double erfc_omega, const double rcut) - { - double eps = 1e-14; - std::vector psik2_ccp(psif.size()); - for (size_t ik = 0; ik < psif.size(); ++ik) - { - double temp0 = std::cos(k_radial[ik] * rcut) * std::erfc(erfc_omega * rcut); - double temp1 = std::exp(-(k_radial[ik] * k_radial[ik]) / (4 * erfc_omega * erfc_omega)); - std::complex temp2 = std::complex(0, 0); - std::complex temp3 = std::complex(0, 0); - if (temp1 >= eps) - { - temp2 = ModuleBase::ErrorFunc::erf(0.5 * (ModuleBase::IMAG_UNIT * k_radial[ik] + 2 * erfc_omega * erfc_omega * rcut) - / erfc_omega); - temp3 = ModuleBase::NEG_IMAG_UNIT - * ModuleBase::ErrorFunc::erfi(0.5 * k_radial[ik] / erfc_omega + ModuleBase::IMAG_UNIT * erfc_omega * rcut); - } - std::complex fock_part = -0.5 * (-2 + 2 * temp0 + temp1 * (temp2 + temp3)); - psik2_ccp[ik] = ModuleBase::FOUR_PI * psif[ik] * fock_part.real(); - } - return psik2_ccp; - } + { + double eps = 1e-14; + std::vector psik2_ccp(psif.size()); + for (size_t ik = 0; ik < psif.size(); ++ik) + { + double temp0 = std::cos(k_radial[ik] * rcut) * std::erfc(erfc_omega * rcut); + double temp1 = std::exp(-(k_radial[ik] * k_radial[ik]) / (4 * erfc_omega * erfc_omega)); + std::complex temp2 = std::complex(0, 0); + std::complex temp3 = std::complex(0, 0); + if (temp1 >= eps) + { + temp2 = ModuleBase::ErrorFunc::erf(0.5 * (ModuleBase::IMAG_UNIT * k_radial[ik] + 2 * erfc_omega * erfc_omega * rcut) + / erfc_omega); + temp3 = ModuleBase::NEG_IMAG_UNIT + * ModuleBase::ErrorFunc::erfi(0.5 * k_radial[ik] / erfc_omega + ModuleBase::IMAG_UNIT * erfc_omega * rcut); + } + std::complex fock_part = -0.5 * (-2 + 2 * temp0 + temp1 * (temp2 + temp3)); + psik2_ccp[ik] = ModuleBase::FOUR_PI * psif[ik] * fock_part.real(); + } + return psik2_ccp; + } - template<> - Numerical_Orbital_Lm cal_orbs_ccp( - const Numerical_Orbital_Lm &orbs, - const std::map>> &coulomb_param, - const double rmesh_times) - { - std::vector psik2_ccp(orbs.get_psif().size(), 0.0); + template<> + Numerical_Orbital_Lm cal_orbs_ccp( + const Numerical_Orbital_Lm &orbs, + const std::map>> &coulomb_param, + const double rmesh_times) + { + std::vector psik2_ccp(orbs.get_psif().size(), 0.0); - for(const auto ¶m_list : coulomb_param) - { - switch(param_list.first) - { - case Conv_Coulomb_Pot_K::Coulomb_Type::Fock: - { - for(const auto ¶m : param_list.second) - { - if(param.at("singularity_correction") == "limits" || param.at("singularity_correction") == "massidda" || param.at("singularity_correction") == "carrier") - { psik2_ccp = psik2_ccp + std::stod(param.at("alpha")) * cal_psi_fock_limits( orbs.get_psif() ); } - else if(param.at("singularity_correction") == "spencer" || param.at("singularity_correction") == "revised_spencer") - { psik2_ccp = psik2_ccp + std::stod(param.at("alpha")) * cal_psi_fock_spencer( orbs.get_psif(), orbs.get_k_radial(), std::stod(param.at("Rcut")) ); } - else - { throw std::invalid_argument( "singularity_correction = " + param.at("singularity_correction") + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__) ); } - } - break; - } - case Conv_Coulomb_Pot_K::Coulomb_Type::Erfc: - { - for(const auto ¶m : param_list.second) - { - if(param.at("singularity_correction") == "limits" || param.at("singularity_correction") == "massidda" || param.at("singularity_correction") == "carrier") - { psik2_ccp = psik2_ccp + std::stod(param.at("alpha")) * cal_psi_erfc_limits( orbs.get_psif(), orbs.get_k_radial(), std::stod(param.at("omega")) ); } - else if(param.at("singularity_correction") == "spencer" || param.at("singularity_correction") == "revised_spencer") - { psik2_ccp = psik2_ccp + std::stod(param.at("alpha")) * cal_psi_erfc_spencer( orbs.get_psif(), orbs.get_k_radial(), std::stod(param.at("omega")), std::stod(param.at("Rcut")) ); } - else - { throw std::invalid_argument( "singularity_correction = " + param.at("singularity_correction") + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__) ); } - } - break; - } - default: - { - throw std::invalid_argument( std::string(__FILE__) + " line " + std::to_string(__LINE__) ); - } - } - } + for(const auto ¶m_list : coulomb_param) + { + switch(param_list.first) + { + case Conv_Coulomb_Pot_K::Coulomb_Type::Fock: + { + for(const auto ¶m : param_list.second) + { + if(param.at("singularity_correction") == "limits" || param.at("singularity_correction") == "massidda" || param.at("singularity_correction") == "carrier") + { psik2_ccp = psik2_ccp + std::stod(param.at("alpha")) * cal_psi_fock_limits( orbs.get_psif() ); } + else if(param.at("singularity_correction") == "spencer" || param.at("singularity_correction") == "revised_spencer") + { psik2_ccp = psik2_ccp + std::stod(param.at("alpha")) * cal_psi_fock_spencer( orbs.get_psif(), orbs.get_k_radial(), std::stod(param.at("Rcut")) ); } + else + { throw std::invalid_argument( "singularity_correction = " + param.at("singularity_correction") + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__) ); } + } + break; + } + case Conv_Coulomb_Pot_K::Coulomb_Type::Erfc: + { + for(const auto ¶m : param_list.second) + { + if(param.at("singularity_correction") == "limits" || param.at("singularity_correction") == "massidda" || param.at("singularity_correction") == "carrier") + { psik2_ccp = psik2_ccp + std::stod(param.at("alpha")) * cal_psi_erfc_limits( orbs.get_psif(), orbs.get_k_radial(), std::stod(param.at("omega")) ); } + else if(param.at("singularity_correction") == "spencer" || param.at("singularity_correction") == "revised_spencer") + { psik2_ccp = psik2_ccp + std::stod(param.at("alpha")) * cal_psi_erfc_spencer( orbs.get_psif(), orbs.get_k_radial(), std::stod(param.at("omega")), std::stod(param.at("Rcut")) ); } + else + { throw std::invalid_argument( "singularity_correction = " + param.at("singularity_correction") + " in " + std::string(__FILE__) + " line " + std::to_string(__LINE__) ); } + } + break; + } + default: + { + throw std::invalid_argument( std::string(__FILE__) + " line " + std::to_string(__LINE__) ); + } + } + } - const double dr = orbs.get_rab().back(); - const int Nr = (static_cast(orbs.getNr()*rmesh_times)) | 1; + const double dr = orbs.get_rab().back(); + const int Nr = (static_cast(orbs.getNr()*rmesh_times)) | 1; - std::vector rab(Nr); - for( size_t ir=0; ir rab(Nr); + for( size_t ir=0; ir r_radial(Nr); - for( size_t ir=0; ir r_radial(Nr); + for( size_t ir=0; ir Numerical_Orbital_Lm cal_orbs_ccp_spencer( @@ -251,16 +251,16 @@ namespace Conv_Coulomb_Pot_K } template<> - double get_rmesh_proportion( - const Numerical_Orbital_Lm &orbs, - const double psi_threshold) - { - for(int ir=orbs.getNr()-1; ir>=0; --ir) - { - if(std::abs(orbs.getPsi(ir))>=psi_threshold) - { return static_cast(ir)/orbs.getNr(); } - } - return 0.0; - } + double get_rmesh_proportion( + const Numerical_Orbital_Lm &orbs, + const double psi_threshold) + { + for(int ir=orbs.getNr()-1; ir>=0; --ir) + { + if(std::abs(orbs.getPsi(ir))>=psi_threshold) + { return static_cast(ir)/orbs.getNr(); } + } + return 0.0; + } } diff --git a/source/source_lcao/module_ri/conv_coulomb_pot_k.h b/source/source_lcao/module_ri/conv_coulomb_pot_k.h index bae445b6070..1b5657767c1 100644 --- a/source/source_lcao/module_ri/conv_coulomb_pot_k.h +++ b/source/source_lcao/module_ri/conv_coulomb_pot_k.h @@ -7,43 +7,43 @@ namespace Conv_Coulomb_Pot_K { - // Constrains the scalar cal_orbs_ccp/cal_orbs_ccp_spencer overloads below: - // icpc cannot order them against the recursive std::vector overloads and - // reports an ambiguity, so the scalar overload is disabled for vectors. - template struct is_std_vector : std::false_type {}; - template struct is_std_vector> : std::true_type {}; - - template extern typename std::enable_if::value, T>::type cal_orbs_ccp( - const T &orbs, - const CoulombParam &coulomb_param, - const double rmesh_times); - - template extern typename std::enable_if::value, T>::type cal_orbs_ccp_spencer( - const T &orbs, - const CoulombParam &coulomb_param, - const double rmesh_times); + // Constrains the scalar cal_orbs_ccp/cal_orbs_ccp_spencer overloads below: + // icpc cannot order them against the recursive std::vector overloads and + // reports an ambiguity, so the scalar overload is disabled for vectors. + template struct is_std_vector : std::false_type {}; + template struct is_std_vector> : std::true_type {}; + + template extern typename std::enable_if::value, T>::type cal_orbs_ccp( + const T &orbs, + const CoulombParam &coulomb_param, + const double rmesh_times); + + template extern typename std::enable_if::value, T>::type cal_orbs_ccp_spencer( + const T &orbs, + const CoulombParam &coulomb_param, + const double rmesh_times); //private: - template< typename T > extern double get_rmesh_proportion( - const T &orbs, - const double psi_threshold); + template< typename T > extern double get_rmesh_proportion( + const T &orbs, + const double psi_threshold); //private: - extern std::vector cal_psi_fock_limits( - const std::vector & psif); - extern std::vector cal_psi_fock_spencer( - const std::vector &psif, - const std::vector &k_radial, - const double rcut); - extern std::vector cal_psi_erfc_limits( - const std::vector & psif, - const std::vector & k_radial, - const double erfc_omega); - extern std::vector cal_psi_erfc_spencer( - const std::vector & psif, - const std::vector & k_radial, - const double erfc_omega, - const double rcut); + extern std::vector cal_psi_fock_limits( + const std::vector & psif); + extern std::vector cal_psi_fock_spencer( + const std::vector &psif, + const std::vector &k_radial, + const double rcut); + extern std::vector cal_psi_erfc_limits( + const std::vector & psif, + const std::vector & k_radial, + const double erfc_omega); + extern std::vector cal_psi_erfc_spencer( + const std::vector & psif, + const std::vector & k_radial, + const double erfc_omega, + const double rcut); } #include "conv_coulomb_pot_k.hpp" diff --git a/source/source_lcao/module_ri/conv_coulomb_pot_k.hpp b/source/source_lcao/module_ri/conv_coulomb_pot_k.hpp index bfd07fc17b2..1c1627f1fb3 100644 --- a/source/source_lcao/module_ri/conv_coulomb_pot_k.hpp +++ b/source/source_lcao/module_ri/conv_coulomb_pot_k.hpp @@ -9,59 +9,59 @@ namespace Conv_Coulomb_Pot_K { - template< typename T > - std::vector cal_orbs_ccp( - const std::vector & orbs, - const std::map>> &coulomb_param, - const double rmesh_times) - { - std::vector orbs_ccp(orbs.size()); - for( size_t i=0; i!=orbs.size(); ++i ) - orbs_ccp[i] = cal_orbs_ccp(orbs[i], coulomb_param, rmesh_times); - return orbs_ccp; - } + template< typename T > + std::vector cal_orbs_ccp( + const std::vector & orbs, + const std::map>> &coulomb_param, + const double rmesh_times) + { + std::vector orbs_ccp(orbs.size()); + for( size_t i=0; i!=orbs.size(); ++i ) + orbs_ccp[i] = cal_orbs_ccp(orbs[i], coulomb_param, rmesh_times); + return orbs_ccp; + } - template< typename T > - double get_rmesh_proportion( - const std::vector & orbs, - const double psi_threshold) - { - double rmesh_proportion=0; - for( const auto &orb : orbs ) - rmesh_proportion = std::max(rmesh_proportion, get_rmesh_proportion(orb,psi_threshold)); - return rmesh_proportion; - } + template< typename T > + double get_rmesh_proportion( + const std::vector & orbs, + const double psi_threshold) + { + double rmesh_proportion=0; + for( const auto &orb : orbs ) + rmesh_proportion = std::max(rmesh_proportion, get_rmesh_proportion(orb,psi_threshold)); + return rmesh_proportion; + } - template< typename T > - std::vector cal_orbs_ccp_spencer( - const std::vector & orbs, - const std::map>> &coulomb_param, - const double rmesh_times) - { - std::vector orbs_ccp(orbs.size()); - for( size_t i=0; i!=orbs.size(); ++i ) - orbs_ccp[i] = cal_orbs_ccp_spencer(orbs[i], coulomb_param, rmesh_times); - return orbs_ccp; - } + template< typename T > + std::vector cal_orbs_ccp_spencer( + const std::vector & orbs, + const std::map>> &coulomb_param, + const double rmesh_times) + { + std::vector orbs_ccp(orbs.size()); + for( size_t i=0; i!=orbs.size(); ++i ) + orbs_ccp[i] = cal_orbs_ccp_spencer(orbs[i], coulomb_param, rmesh_times); + return orbs_ccp; + } - // for cal_orbs_ccp() - template - std::vector operator*(const T &s, const std::vector &v_in) - { - std::vector v(v_in.size()); - for(std::size_t i=0; i - std::vector operator+ (const std::vector &v1, const std::vector &v2) - { - assert(v1.size()==v2.size()); - std::vector v(v1.size()); - for(std::size_t i=0; i + std::vector operator*(const T &s, const std::vector &v_in) + { + std::vector v(v_in.size()); + for(std::size_t i=0; i + std::vector operator+ (const std::vector &v1, const std::vector &v2) + { + assert(v1.size()==v2.size()); + std::vector v(v1.size()); + for(std::size_t i=0; i::get_singular_chi(const UnitCell& ucell, const std::vecto double chi = 0.0; for(const auto ¶m : param_list) - { + { if(param.at("singularity_correction") == "carrier") - { + { chi = Singular_Value::cal_carrier(ucell, this->kvec_c, qdiv, 100, 30, 1e-6, 3); } else if(param.at("singularity_correction") == "massidda") - { + { chi = Singular_Value::cal_massidda(ucell, this->nmp, qdiv, 1, 5, 1e-4); } else @@ -413,7 +413,7 @@ auto Ewald_Vq::set_Vq_dVq_gauss(const UnitCell& ucell, std::map> Vq_dVq_gauss_out; for(const auto ¶m_list : this->coulomb_param) - { + { std::complex alpha; for(const auto ¶m : param_list.second) { diff --git a/source/source_lcao/module_ri/exx_abfs.h b/source/source_lcao/module_ri/exx_abfs.h index 0c5c84f1628..be6101af4c8 100644 --- a/source/source_lcao/module_ri/exx_abfs.h +++ b/source/source_lcao/module_ri/exx_abfs.h @@ -11,13 +11,13 @@ using std::map; class Exx_Abfs { public: - class Jle; - class IO; - class Construct_Orbs; - class PCA; - - int rmesh_times = 5; // Peize Lin test - int kmesh_times = 1; // Peize Lin test + class Jle; + class IO; + class Construct_Orbs; + class PCA; + + int rmesh_times = 5; // Peize Lin test + int kmesh_times = 1; // Peize Lin test }; #endif diff --git a/source/source_lcao/module_ri/exx_abfs_ctor_orbs.cpp b/source/source_lcao/module_ri/exx_abfs_ctor_orbs.cpp index 2c23f35e02f..3efac4ff9cb 100644 --- a/source/source_lcao/module_ri/exx_abfs_ctor_orbs.cpp +++ b/source/source_lcao/module_ri/exx_abfs_ctor_orbs.cpp @@ -8,435 +8,435 @@ #include "source_lcao/module_ri/test_code/exx_abfs_ctor_orbs_test.h" // Peize Lin test std::vector>> Exx_Abfs::Construct_Orbs::change_orbs( - const LCAO_Orbitals &orbs_in, - const double kmesh_times ) + const LCAO_Orbitals &orbs_in, + const double kmesh_times ) { ModuleBase::TITLE("Exx_Abfs::Construct_Orbs::change_orbs"); - std::vector>> orbs; - orbs.resize( orbs_in.get_ntype() ); - for (int T = 0; T < orbs_in.get_ntype() ; T++) - { - orbs[T].resize( orbs_in.Phi[T].getLmax()+1 ); - for (int L=0; L <= orbs_in.Phi[T].getLmax() ; L++) - { - orbs[T][L].resize( orbs_in.Phi[T].getNchi(L) ); - for (int N = 0; N < orbs_in.Phi[T].getNchi(L); ++N) - { - const auto &orb_origin = orbs_in.Phi[T].PhiLN(L,N); - orbs[T][L][N].set_orbital_info( - orb_origin.getLabel(), - orb_origin.getType(), - orb_origin.getL(), - orb_origin.getChi(), - orb_origin.getNr(), - orb_origin.getRab(), - orb_origin.getRadial(), - Numerical_Orbital_Lm::Psi_Type::Psi, - orb_origin.getPsi(), - static_cast(orb_origin.getNk() * kmesh_times) | 1, // Nk must be odd - orb_origin.getDk(), // Peize Lin change 2017-04-16 + std::vector>> orbs; + orbs.resize( orbs_in.get_ntype() ); + for (int T = 0; T < orbs_in.get_ntype() ; T++) + { + orbs[T].resize( orbs_in.Phi[T].getLmax()+1 ); + for (int L=0; L <= orbs_in.Phi[T].getLmax() ; L++) + { + orbs[T][L].resize( orbs_in.Phi[T].getNchi(L) ); + for (int N = 0; N < orbs_in.Phi[T].getNchi(L); ++N) + { + const auto &orb_origin = orbs_in.Phi[T].PhiLN(L,N); + orbs[T][L][N].set_orbital_info( + orb_origin.getLabel(), + orb_origin.getType(), + orb_origin.getL(), + orb_origin.getChi(), + orb_origin.getNr(), + orb_origin.getRab(), + orb_origin.getRadial(), + Numerical_Orbital_Lm::Psi_Type::Psi, + orb_origin.getPsi(), + static_cast(orb_origin.getNk() * kmesh_times) | 1, // Nk must be odd + orb_origin.getDk(), // Peize Lin change 2017-04-16 // orb_origin.getDk() / kmesh_times, - orb_origin.getDruniform(), - false, - true, PARAM.inp.cal_force); - } - } - } - - Exx_Abfs::Construct_Orbs::filter_empty_orbs(orbs); - return orbs; + orb_origin.getDruniform(), + false, + true, PARAM.inp.cal_force); + } + } + } + + Exx_Abfs::Construct_Orbs::filter_empty_orbs(orbs); + return orbs; } std::vector>> Exx_Abfs::Construct_Orbs::change_orbs( - const std::vector>> &orbs_in, - const double kmesh_times ) + const std::vector>> &orbs_in, + const double kmesh_times ) { - ModuleBase::TITLE("Exx_Abfs::Construct_Orbs::change_orbs"); - return orbital( get_psi(orbs_in), orbs_in, kmesh_times ); + ModuleBase::TITLE("Exx_Abfs::Construct_Orbs::change_orbs"); + return orbital( get_psi(orbs_in), orbs_in, kmesh_times ); } // P = u/r * Y /* template std::vector>> Exx_Abfs::Construct_Orbs::abfs_same_atom( - const Orbs_Type &orbs, - const double kmesh_times, - const double norm_threshold ) + const Orbs_Type &orbs, + const double kmesh_times, + const double norm_threshold ) { - const std::vector>>> - abfs_same_atom_psir = psir_mult_psir( orbs ); - const std::vector>>> - abfs_same_atom_psir_orth = orth( abfs_same_atom_psir, orbs, norm_threshold ); - const std::vector>>> - abfs_same_atom_psi_orth = div_r( abfs_same_atom_psir_orth, orbs.get_r_radial ); - const std::vector>> - abfs_same_atom = orbital( abfs_same_atom_psi_orth, orbs, kmesh_times ); - return abfs_same_atom; + const std::vector>>> + abfs_same_atom_psir = psir_mult_psir( orbs ); + const std::vector>>> + abfs_same_atom_psir_orth = orth( abfs_same_atom_psir, orbs, norm_threshold ); + const std::vector>>> + abfs_same_atom_psi_orth = div_r( abfs_same_atom_psir_orth, orbs.get_r_radial ); + const std::vector>> + abfs_same_atom = orbital( abfs_same_atom_psi_orth, orbs, kmesh_times ); + return abfs_same_atom; } */ // P = f * Y std::vector>> Exx_Abfs::Construct_Orbs::abfs_same_atom( - const UnitCell &ucell, + const UnitCell &ucell, const LCAO_Orbitals& orb, - const std::vector>> &orbs, - const double kmesh_times_mot, - const double times_threshold ) + const std::vector>> &orbs, + const double kmesh_times_mot, + const double times_threshold ) { - ModuleBase::TITLE("Exx_Abfs::Construct_Orbs::abfs_same_atom"); - if(times_threshold>1) - { return std::vector>>(orb.get_ntype()); } - - const std::vector>>> - abfs_same_atom_psi = psi_mult_psi( orbs ); - - const std::vector>>> - abfs_same_atom_orth_psi = orth( abfs_same_atom_psi, orbs ); - const std::vector>> - abfs_same_atom = orbital( abfs_same_atom_orth_psi, orbs, 1 ); - - #if TEST_EXX_LCAO==1 - print_orbs(abfs_same_atom_psi,"abfs_same_atom_psi.dat"); - print_orbs(abfs_same_atom_orth_psi,"abfs_same_atom_orth_psi.dat"); - #elif TEST_EXX_LCAO==-1 - #error "TEST_EXX_LCAO" - #endif - - const std::vector>>> - abfs_same_atom_pca_psi = pca(ucell,orb, abfs_same_atom, orbs, kmesh_times_mot, times_threshold ); - - #if TEST_EXX_LCAO==1 - print_orbs(abfs_same_atom_pca_psi,"abfs_same_atom_pca_psi.dat"); - #elif TEST_EXX_LCAO==-1 - #error "TEST_EXX_LCAO" - #endif + ModuleBase::TITLE("Exx_Abfs::Construct_Orbs::abfs_same_atom"); + if(times_threshold>1) + { return std::vector>>(orb.get_ntype()); } + + const std::vector>>> + abfs_same_atom_psi = psi_mult_psi( orbs ); + const std::vector>>> + abfs_same_atom_orth_psi = orth( abfs_same_atom_psi, orbs ); const std::vector>> - abfs_same_atom_pca = orbital( abfs_same_atom_pca_psi, orbs, 1 ); + abfs_same_atom = orbital( abfs_same_atom_orth_psi, orbs, 1 ); + + #if TEST_EXX_LCAO==1 + print_orbs(abfs_same_atom_psi,"abfs_same_atom_psi.dat"); + print_orbs(abfs_same_atom_orth_psi,"abfs_same_atom_orth_psi.dat"); + #elif TEST_EXX_LCAO==-1 + #error "TEST_EXX_LCAO" + #endif + + const std::vector>>> + abfs_same_atom_pca_psi = pca(ucell,orb, abfs_same_atom, orbs, kmesh_times_mot, times_threshold ); + + #if TEST_EXX_LCAO==1 + print_orbs(abfs_same_atom_pca_psi,"abfs_same_atom_pca_psi.dat"); + #elif TEST_EXX_LCAO==-1 + #error "TEST_EXX_LCAO" + #endif + + const std::vector>> + abfs_same_atom_pca = orbital( abfs_same_atom_pca_psi, orbs, 1 ); return abfs_same_atom_pca; } /* template<> std::vector>>> Exx_Abfs::Construct_Orbs::psi_mult_psi( - const LCAO_Orbitals &orbs ) + const LCAO_Orbitals &orbs ) { - std::vector>>> psi_mult_psi( orbs.get_ntype() ); - for( int T=0; T!=orbs.get_ntype(); ++T ) - { - psi_mult_psi[T].resize( 2*orbs.Phi[T].getLmax()+1 ); - for( int L1=0; L1<=orbs.Phi[T].getLmax(); ++L1 ) - { - for( int N1=0; N1!=orbs.Phi[T].getNchi(L1); ++N1 ) - { - for( int L2=L1; L2<=orbs.Phi[T].getLmax(); ++L2 ) - { - for( int N2=((L2==L1)?N1:0); N2!=orbs.Phi[T].getNchi(L2); ++N2 ) - { - assert( orbs.Phi[T].PhiLN(L1,N1).getNr()==orbs.Phi[T].PhiLN(L2,N2).getNr() ); - - std::vector mult_psir( orbs.Phi[T].PhiLN(L1,N1).getNr() ); - for( int ir=0; ir!=orbs.Phi[T].PhiLN(L1,N1).getNr(); ++ir) - { - mult_psir[ir] = orbs.Phi[T].PhiLN(L1,N1).getPsi(ir) * orbs.Phi[T].PhiLN(L2,N2).getPsi(ir) ; - } - for( int L_new=std::abs(L2-L1); L_new<=L1+L2; ++L_new ) - { - psi_mult_psi[T][L_new].push_back(mult_psir); - } - } - } - } - } - } - return psi_mult_psi; + std::vector>>> psi_mult_psi( orbs.get_ntype() ); + for( int T=0; T!=orbs.get_ntype(); ++T ) + { + psi_mult_psi[T].resize( 2*orbs.Phi[T].getLmax()+1 ); + for( int L1=0; L1<=orbs.Phi[T].getLmax(); ++L1 ) + { + for( int N1=0; N1!=orbs.Phi[T].getNchi(L1); ++N1 ) + { + for( int L2=L1; L2<=orbs.Phi[T].getLmax(); ++L2 ) + { + for( int N2=((L2==L1)?N1:0); N2!=orbs.Phi[T].getNchi(L2); ++N2 ) + { + assert( orbs.Phi[T].PhiLN(L1,N1).getNr()==orbs.Phi[T].PhiLN(L2,N2).getNr() ); + + std::vector mult_psir( orbs.Phi[T].PhiLN(L1,N1).getNr() ); + for( int ir=0; ir!=orbs.Phi[T].PhiLN(L1,N1).getNr(); ++ir) + { + mult_psir[ir] = orbs.Phi[T].PhiLN(L1,N1).getPsi(ir) * orbs.Phi[T].PhiLN(L2,N2).getPsi(ir) ; + } + for( int L_new=std::abs(L2-L1); L_new<=L1+L2; ++L_new ) + { + psi_mult_psi[T][L_new].push_back(mult_psir); + } + } + } + } + } + } + return psi_mult_psi; } */ std::vector>>> Exx_Abfs::Construct_Orbs::psi_mult_psi( - const std::vector>> &orbs ) + const std::vector>> &orbs ) { - std::vector>>> psi_mult_psi( orbs.size() ); - for( int T=0; T!=orbs.size(); ++T ) - { - psi_mult_psi[T].resize( 2*orbs[T].size()-1 ); - for( int L1=0; L1!=orbs[T].size(); ++L1 ) - { - for( int N1=0; N1!=orbs[T][L1].size(); ++N1 ) - { - for( int L2=L1; L2!=orbs[T].size(); ++L2 ) - { - for( int N2=((L2==L1)?N1:0); N2!=orbs[T][L2].size(); ++N2 ) - { - assert( orbs[T][L1][N1].getNr()==orbs[T][L2][N2].getNr() ); - - std::vector mult_psir( orbs[T][L1][N1].getNr() ); - for( int ir=0; ir!=orbs[T][L1][N1].getNr(); ++ir) - { - mult_psir[ir] = orbs[T][L1][N1].getPsi(ir) * orbs[T][L2][N2].getPsi(ir) ; - } - for( int L_new=std::abs(L1-L2); L_new<=L1+L2; ++L_new ) - { - psi_mult_psi[T][L_new].push_back(mult_psir); - } - } - } - } - } - } - return psi_mult_psi; + std::vector>>> psi_mult_psi( orbs.size() ); + for( int T=0; T!=orbs.size(); ++T ) + { + psi_mult_psi[T].resize( 2*orbs[T].size()-1 ); + for( int L1=0; L1!=orbs[T].size(); ++L1 ) + { + for( int N1=0; N1!=orbs[T][L1].size(); ++N1 ) + { + for( int L2=L1; L2!=orbs[T].size(); ++L2 ) + { + for( int N2=((L2==L1)?N1:0); N2!=orbs[T][L2].size(); ++N2 ) + { + assert( orbs[T][L1][N1].getNr()==orbs[T][L2][N2].getNr() ); + + std::vector mult_psir( orbs[T][L1][N1].getNr() ); + for( int ir=0; ir!=orbs[T][L1][N1].getNr(); ++ir) + { + mult_psir[ir] = orbs[T][L1][N1].getPsi(ir) * orbs[T][L2][N2].getPsi(ir) ; + } + for( int L_new=std::abs(L1-L2); L_new<=L1+L2; ++L_new ) + { + psi_mult_psi[T][L_new].push_back(mult_psir); + } + } + } + } + } + } + return psi_mult_psi; } /* template<> std::vector>>> Exx_Abfs::Construct_Orbs::psir_mult_psir( - const LCAO_Orbitals &orbs ) + const LCAO_Orbitals &orbs ) { - std::vector>>> psir_mult_psir( orbs.get_ntype() ); - for( int T=0; T!=orbs.get_ntype(); ++T ) - { - psir_mult_psir[T].resize( 2*orbs.Phi[T].getLmax()+1 ); - for( int L1=0; L1<=orbs.Phi[T].getLmax(); ++L1 ) - { - for( int N1=0; N1!=orbs.Phi[T].getNchi(L1); ++N1 ) - { - for( int L2=L1; L2<=orbs.Phi[T].getLmax(); ++L2 ) - { - for( int N2=((L2==L1)?N1:0); N2!=orbs.Phi[T].getNchi(L2); ++N2 ) - { - assert( orbs.Phi[T].PhiLN(L1,N1).getNr()==orbs.Phi[T].PhiLN(L2,N2).getNr() ); - - std::vector mult_psir( orbs.Phi[T].PhiLN(L1,N1).getNr() ); - for( int ir=0; ir!=orbs.Phi[T].PhiLN(L1,N1).getNr(); ++ir) - { - mult_psir[ir] = orbs.Phi[T].PhiLN(L1,N1).getPsi_r(ir) * orbs.Phi[T].PhiLN(L2,N2).getPsi_r(ir) ; - } - for( int L_new=std::abs(L1-L2); L_new<=L1+L2; ++L_new ) - { - psir_mult_psir[T][L_new].push_back(mult_psir); - } - } - } - } - } - } - return psir_mult_psir; + std::vector>>> psir_mult_psir( orbs.get_ntype() ); + for( int T=0; T!=orbs.get_ntype(); ++T ) + { + psir_mult_psir[T].resize( 2*orbs.Phi[T].getLmax()+1 ); + for( int L1=0; L1<=orbs.Phi[T].getLmax(); ++L1 ) + { + for( int N1=0; N1!=orbs.Phi[T].getNchi(L1); ++N1 ) + { + for( int L2=L1; L2<=orbs.Phi[T].getLmax(); ++L2 ) + { + for( int N2=((L2==L1)?N1:0); N2!=orbs.Phi[T].getNchi(L2); ++N2 ) + { + assert( orbs.Phi[T].PhiLN(L1,N1).getNr()==orbs.Phi[T].PhiLN(L2,N2).getNr() ); + + std::vector mult_psir( orbs.Phi[T].PhiLN(L1,N1).getNr() ); + for( int ir=0; ir!=orbs.Phi[T].PhiLN(L1,N1).getNr(); ++ir) + { + mult_psir[ir] = orbs.Phi[T].PhiLN(L1,N1).getPsi_r(ir) * orbs.Phi[T].PhiLN(L2,N2).getPsi_r(ir) ; + } + for( int L_new=std::abs(L1-L2); L_new<=L1+L2; ++L_new ) + { + psir_mult_psir[T][L_new].push_back(mult_psir); + } + } + } + } + } + } + return psir_mult_psir; } */ std::vector>>> Exx_Abfs::Construct_Orbs::psir_mult_psir( - const std::vector>> &orbs ) + const std::vector>> &orbs ) { - std::vector>>> psir_mult_psir( orbs.size() ); - for( int T=0; T!=orbs.size(); ++T ) - { - psir_mult_psir[T].resize( 2*orbs[T].size()-1 ); - for( int L1=0; L1!=orbs[T].size(); ++L1 ) - { - for( int N1=0; N1!=orbs[T][L1].size(); ++N1 ) - { - for( int L2=L1; L2!=orbs[T].size(); ++L2 ) - { - for( int N2=((L2==L1)?N1:0); N2!=orbs[T][L2].size(); ++N2 ) - { - assert( orbs[T][L1][N1].getNr()==orbs[T][L2][N2].getNr() ); - - std::vector mult_psir( orbs[T][L1][N1].getNr() ); - for( int ir=0; ir!=orbs[T][L1][N1].getNr(); ++ir) - { - mult_psir[ir] = orbs[T][L1][N1].getPsi_r(ir) * orbs[T][L2][N2].getPsi_r(ir) ; - } - for( int L_new=std::abs(L1-L2); L_new<=L1+L2; ++L_new ) - { - psir_mult_psir[T][L_new].push_back(mult_psir); - } - } - } - } - } - } - return psir_mult_psir; + std::vector>>> psir_mult_psir( orbs.size() ); + for( int T=0; T!=orbs.size(); ++T ) + { + psir_mult_psir[T].resize( 2*orbs[T].size()-1 ); + for( int L1=0; L1!=orbs[T].size(); ++L1 ) + { + for( int N1=0; N1!=orbs[T][L1].size(); ++N1 ) + { + for( int L2=L1; L2!=orbs[T].size(); ++L2 ) + { + for( int N2=((L2==L1)?N1:0); N2!=orbs[T][L2].size(); ++N2 ) + { + assert( orbs[T][L1][N1].getNr()==orbs[T][L2][N2].getNr() ); + + std::vector mult_psir( orbs[T][L1][N1].getNr() ); + for( int ir=0; ir!=orbs[T][L1][N1].getNr(); ++ir) + { + mult_psir[ir] = orbs[T][L1][N1].getPsi_r(ir) * orbs[T][L2][N2].getPsi_r(ir) ; + } + for( int L_new=std::abs(L1-L2); L_new<=L1+L2; ++L_new ) + { + psir_mult_psir[T][L_new].push_back(mult_psir); + } + } + } + } + } + } + return psir_mult_psir; } std::vector>>> Exx_Abfs::Construct_Orbs::pca( - const UnitCell &ucell, + const UnitCell &ucell, const LCAO_Orbitals& orb, - const std::vector>> &abfs, - const std::vector>> &orbs, - const double kmesh_times_mot, - const double times_threshold ) + const std::vector>> &abfs, + const std::vector>> &orbs, + const double kmesh_times_mot, + const double times_threshold ) { - if(times_threshold>1) - return std::vector>>>(abfs.size()); + if(times_threshold>1) + return std::vector>>>(abfs.size()); - const std::vector,RI::Tensor>>> - eig = ABFs_Construct::PCA::cal_PCA(ucell, orb, orbs, abfs, kmesh_times_mot ); + const std::vector,RI::Tensor>>> + eig = ABFs_Construct::PCA::cal_PCA(ucell, orb, orbs, abfs, kmesh_times_mot ); - const std::vector>>> psis = get_psi( abfs ); - std::vector>>> psis_new( psis.size() ); + const std::vector>>> psis = get_psi( abfs ); + std::vector>>> psis_new( psis.size() ); - for( size_t T=0; T!=eig.size(); ++T ) - { - double eig_value_max = 0; - for( size_t L=0; L!=eig[T].size(); ++L ) - for( size_t M=0; M!=eig[T][L].first.size(); ++M ) - { + for( size_t T=0; T!=eig.size(); ++T ) + { + double eig_value_max = 0; + for( size_t L=0; L!=eig[T].size(); ++L ) + for( size_t M=0; M!=eig[T][L].first.size(); ++M ) + { //ofs< &eig_value = eig[T][L].first; - const RI::Tensor &eig_vec = eig[T][L].second; - for( size_t M=0; M!=eig_value.size(); ++M ) - { - if( eig_value[M] > eig_value_threshold ) - { - std::vector psi_new( psis[T][L][0].size() ); - for( size_t N=0; N!=psis[T][L].size(); ++N ) - for( size_t ir=0; ir!=psi_new.size(); ++ir ) - psi_new[ir] += eig_vec(M,N) * psis[T][L][N][ir]; - psis_new[T][L].push_back( psi_new ); - } - } - } - } - else - { - ModuleBase::WARNING(std::string(__FILE__), - "Element "+ModuleBase::GlobalFunc::TO_STRING(T)+" , all training data (lcao[i]*lcao[j]) are all the same. So PCA randomly choose an abf as the result."); - psis_new[T].resize( psis[T].size() ); - for( size_t L=0; L!=psis[T].size(); ++L ) - if( !psis[T][L].empty() ) - { - psis_new[T][L].push_back(psis[T][L][0]); - break; - } - } - } + if(eig_value_max) + { + psis_new[T].resize( psis[T].size() ); + for( size_t L=0; L!=eig[T].size(); ++L ) + { + const std::vector &eig_value = eig[T][L].first; + const RI::Tensor &eig_vec = eig[T][L].second; + for( size_t M=0; M!=eig_value.size(); ++M ) + { + if( eig_value[M] > eig_value_threshold ) + { + std::vector psi_new( psis[T][L][0].size() ); + for( size_t N=0; N!=psis[T][L].size(); ++N ) + for( size_t ir=0; ir!=psi_new.size(); ++ir ) + psi_new[ir] += eig_vec(M,N) * psis[T][L][N][ir]; + psis_new[T][L].push_back( psi_new ); + } + } + } + } + else + { + ModuleBase::WARNING(std::string(__FILE__), + "Element "+ModuleBase::GlobalFunc::TO_STRING(T)+" , all training data (lcao[i]*lcao[j]) are all the same. So PCA randomly choose an abf as the result."); + psis_new[T].resize( psis[T].size() ); + for( size_t L=0; L!=psis[T].size(); ++L ) + if( !psis[T][L].empty() ) + { + psis_new[T][L].push_back(psis[T][L][0]); + break; + } + } + } //ofs.close(); - return psis_new; + return psis_new; } std::vector>>> Exx_Abfs::Construct_Orbs::orth( - const std::vector>>> &psis, - const std::vector>> &orbs, - const double norm_threshold ) + const std::vector>>> &psis, + const std::vector>> &orbs, + const double norm_threshold ) { - std::vector>>> psis_orth( psis.size() ); - for( int T=0; T!=psis.size(); ++T ) - { - const Numerical_Orbital_Lm &orb = orbs[T][0][0]; - ModuleBase::Gram_Schmidt_Orth gso( - orb.get_rab(), - ModuleBase::Gram_Schmidt_Orth::Coordinate::Sphere ); - psis_orth[T].resize( psis[T].size() ); - for( int L=0; L!=psis[T].size(); ++L ) - { - psis_orth[T][L] = gso.cal_orth( psis[T][L], norm_threshold ); // Peize Lin test 2016-10-08 + std::vector>>> psis_orth( psis.size() ); + for( int T=0; T!=psis.size(); ++T ) + { + const Numerical_Orbital_Lm &orb = orbs[T][0][0]; + ModuleBase::Gram_Schmidt_Orth gso( + orb.get_rab(), + ModuleBase::Gram_Schmidt_Orth::Coordinate::Sphere ); + psis_orth[T].resize( psis[T].size() ); + for( int L=0; L!=psis[T].size(); ++L ) + { + psis_orth[T][L] = gso.cal_orth( psis[T][L], norm_threshold ); // Peize Lin test 2016-10-08 // psis_orth[T][L] = gso.cal_orth( psis[T][L] ); // Peize Lin test 2016-10-08 - } - } - return psis_orth; + } + } + return psis_orth; } std::vector>>> Exx_Abfs::Construct_Orbs::div_r( - const std::vector>>> &psirs, - const std::vector &r_radial ) + const std::vector>>> &psirs, + const std::vector &r_radial ) { - std::vector>>> psis( psirs.size() ); - for( auto T=0; T!=psirs.size(); ++T ) - { - psis[T].resize( psirs[T].size() ); - for( auto L=0; L!=psirs[T].size(); ++L ) - { - psis[T][L].resize( psirs[T][L].size() ); - for( auto N=0; N!=psirs[T][L].size(); ++N ) - { - psis[T][L][N].resize( psirs[T][L][N].size() ); - psis[T][L][N][0] = 0; - for( auto ir=1; ir!=psirs[T][L][N].size(); ++ir ) - { - psis[T][L][N][ir] = psirs[T][L][N][ir] / r_radial[ir]; - } - } - } - } - return psis; + std::vector>>> psis( psirs.size() ); + for( auto T=0; T!=psirs.size(); ++T ) + { + psis[T].resize( psirs[T].size() ); + for( auto L=0; L!=psirs[T].size(); ++L ) + { + psis[T][L].resize( psirs[T][L].size() ); + for( auto N=0; N!=psirs[T][L].size(); ++N ) + { + psis[T][L][N].resize( psirs[T][L][N].size() ); + psis[T][L][N][0] = 0; + for( auto ir=1; ir!=psirs[T][L][N].size(); ++ir ) + { + psis[T][L][N][ir] = psirs[T][L][N][ir] / r_radial[ir]; + } + } + } + } + return psis; } std::vector>>> Exx_Abfs::Construct_Orbs::get_psi( - const std::vector>> &orbs ) + const std::vector>> &orbs ) { - std::vector>>> orbs_psi( orbs.size() ); - for( int T=0; T!=orbs.size(); ++T ) - { - orbs_psi[T].resize( orbs[T].size() ); - for( int L=0; L!=orbs[T].size(); ++L ) - { - orbs_psi[T][L].resize( orbs[T][L].size() ); - for( int N=0; N!=orbs[T][L].size(); ++N ) - { - orbs_psi[T][L][N] = orbs[T][L][N].get_psi(); - } - } - } - return orbs_psi; + std::vector>>> orbs_psi( orbs.size() ); + for( int T=0; T!=orbs.size(); ++T ) + { + orbs_psi[T].resize( orbs[T].size() ); + for( int L=0; L!=orbs[T].size(); ++L ) + { + orbs_psi[T][L].resize( orbs[T][L].size() ); + for( int N=0; N!=orbs[T][L].size(); ++N ) + { + orbs_psi[T][L][N] = orbs[T][L][N].get_psi(); + } + } + } + return orbs_psi; } std::vector>> Exx_Abfs::Construct_Orbs::orbital( - const std::vector>>> &psis, - const std::vector>> &orbs_info, - const double kmesh_times) + const std::vector>>> &psis, + const std::vector>> &orbs_info, + const double kmesh_times) { - std::vector>> orbs_new( psis.size() ); - for( int T=0; T!=psis.size(); ++T ) - { - const Numerical_Orbital_Lm &orb_info = orbs_info[T][0][0]; - orbs_new[T].resize( psis[T].size() ); - for( int L=0; L!=psis[T].size(); ++L ) - { - orbs_new[T][L].resize( psis[T][L].size() ); - for( int N=0; N!=psis[T][L].size(); ++N ) - { - orbs_new[T][L][N].set_orbital_info( - orb_info.getLabel(), - T, - L, - N, - orb_info.getNr(), - orb_info.getRab(), - orb_info.getRadial(), - Numerical_Orbital_Lm::Psi_Type::Psi, - ModuleBase::GlobalFunc::VECTOR_TO_PTR(psis[T][L][N]), - static_cast(orb_info.getNk() * kmesh_times) | 1, // Nk must be odd - orb_info.getDk(), // Peize Lin test 2017-04-16 + std::vector>> orbs_new( psis.size() ); + for( int T=0; T!=psis.size(); ++T ) + { + const Numerical_Orbital_Lm &orb_info = orbs_info[T][0][0]; + orbs_new[T].resize( psis[T].size() ); + for( int L=0; L!=psis[T].size(); ++L ) + { + orbs_new[T][L].resize( psis[T][L].size() ); + for( int N=0; N!=psis[T][L].size(); ++N ) + { + orbs_new[T][L][N].set_orbital_info( + orb_info.getLabel(), + T, + L, + N, + orb_info.getNr(), + orb_info.getRab(), + orb_info.getRadial(), + Numerical_Orbital_Lm::Psi_Type::Psi, + ModuleBase::GlobalFunc::VECTOR_TO_PTR(psis[T][L][N]), + static_cast(orb_info.getNk() * kmesh_times) | 1, // Nk must be odd + orb_info.getDk(), // Peize Lin test 2017-04-16 // orb_info.getDk() / kmesh_times, - orb_info.getDruniform(), - false, - true, PARAM.inp.cal_force); - } - } - } - return orbs_new; + orb_info.getDruniform(), + false, + true, PARAM.inp.cal_force); + } + } + } + return orbs_new; } /* std::vector>>> Exx_Abfs::Construct_Orbs::get_psi( - const LCAO_Orbitals &orbs ) + const LCAO_Orbitals &orbs ) { - std::vector>>> orbs_psi( orbs.get_ntype() ); - for( int T=0; T!=orbs.get_ntype(); ++T ) + std::vector>>> orbs_psi( orbs.get_ntype() ); + for( int T=0; T!=orbs.get_ntype(); ++T ) { orbs_psi[T].resize( orbs.Phi[T].getLmax()+1 ); for( int L=0; L<=orbs.Phi[T].getLmax(); ++L ) @@ -473,25 +473,25 @@ inline const Numerical_Orbital_Lm &Exx_Abfs::Construct_Orbs::get_orbital( */ void Exx_Abfs::Construct_Orbs::print_orbs_size( - const UnitCell& ucell, - const std::vector>> &orbs, - std::ostream &os) + const UnitCell& ucell, + const std::vector>> &orbs, + std::ostream &os) { - os<<" Auxiliary basis functions"< L_labels = {'s', 'p', 'd'}; - for(std::size_t T=0; T L_labels = {'s', 'p', 'd'}; + for(std::size_t T=0; T>> Exx_Abfs::Construct_Orbs::get_multipole( @@ -544,14 +544,14 @@ std::vector Exx_Abfs::Construct_Orbs::get_Rcut(const std::vector>> &orbs) { - for (int T=0; T=0 ; --L) - { - if (orbs[T][L].size()>0) - { break; } - else - { orbs[T].resize(L); } - } - } + for (int T=0; T=0 ; --L) + { + if (orbs[T][L].size()>0) + { break; } + else + { orbs[T].resize(L); } + } + } } diff --git a/source/source_lcao/module_ri/exx_abfs_ctor_orbs.h b/source/source_lcao/module_ri/exx_abfs_ctor_orbs.h index 5981e6704f6..f72c6830a1c 100644 --- a/source/source_lcao/module_ri/exx_abfs_ctor_orbs.h +++ b/source/source_lcao/module_ri/exx_abfs_ctor_orbs.h @@ -13,90 +13,90 @@ class LCAO_Orbitals; class Exx_Abfs::Construct_Orbs { public: - static std::vector>> change_orbs( - const LCAO_Orbitals &orb_in, - const double kmesh_times ); - static std::vector>> change_orbs( - const std::vector>> &orb_in, - const double kmesh_times ); + static std::vector>> change_orbs( + const LCAO_Orbitals &orb_in, + const double kmesh_times ); + static std::vector>> change_orbs( + const std::vector>> &orb_in, + const double kmesh_times ); - static std::vector>> abfs_same_atom( - const UnitCell &ucell, - const LCAO_Orbitals& orb, - const std::vector>> &lcaos, - const double kmesh_times_mot, - const double times_threshold=0); - - static void print_orbs_size( - const UnitCell& ucell, - const std::vector>> &orbs, - std::ostream &os); + static std::vector>> abfs_same_atom( + const UnitCell &ucell, + const LCAO_Orbitals& orb, + const std::vector>> &lcaos, + const double kmesh_times_mot, + const double times_threshold=0); + + static void print_orbs_size( + const UnitCell& ucell, + const std::vector>> &orbs, + std::ostream &os); - // get the max number of orbitals among all elements - // static int get_nmax_total(const - // std::vector>> &orb_in); get - // number of orbitals for each element static std::map - // get_nw(const std::vector>> - // &orb_in); + // get the max number of orbitals among all elements + // static int get_nmax_total(const + // std::vector>> &orb_in); get + // number of orbitals for each element static std::map + // get_nw(const std::vector>> + // &orb_in); - // get multipole of orbitals for each element and angular moment - static std::vector>> get_multipole( - const std::vector>> &orb_in); + // get multipole of orbitals for each element and angular moment + static std::vector>> get_multipole( + const std::vector>> &orb_in); - static std::vector get_Rcut( - const std::vector>> &orb_in); - static inline double get_Rmax(const std::vector& rcut) - { - return *std::max_element(rcut.begin(), rcut.end()); - } - static inline double get_Rmax( - const std::vector>> &orb_in) - { - std::vector rcut = get_Rcut(orb_in); - return get_Rmax(rcut); - } - template - static int get_Lmax(const std::vector> &orb) - { - return max_element(orb.begin(), orb.end(), - [](const std::vector &orb_A, const std::vector &orb_B){ return orb_A.size() < orb_B.size(); }) - ->size() - 1; - } + static std::vector get_Rcut( + const std::vector>> &orb_in); + static inline double get_Rmax(const std::vector& rcut) + { + return *std::max_element(rcut.begin(), rcut.end()); + } + static inline double get_Rmax( + const std::vector>> &orb_in) + { + std::vector rcut = get_Rcut(orb_in); + return get_Rmax(rcut); + } + template + static int get_Lmax(const std::vector> &orb) + { + return max_element(orb.begin(), orb.end(), + [](const std::vector &orb_A, const std::vector &orb_B){ return orb_A.size() < orb_B.size(); }) + ->size() - 1; + } - static void filter_empty_orbs( - std::vector>> &orbs); + static void filter_empty_orbs( + std::vector>> &orbs); private: - static std::vector>>> psi_mult_psi( - const std::vector>> &lcaos ); - - static std::vector>>> psir_mult_psir( - const std::vector>> &lcaos ); - - static std::vector>>> orth( - const std::vector>>> &psis, - const std::vector>> &lcaos, - const double norm_threshold = std::numeric_limits::min() ); + static std::vector>>> psi_mult_psi( + const std::vector>> &lcaos ); + + static std::vector>>> psir_mult_psir( + const std::vector>> &lcaos ); + + static std::vector>>> orth( + const std::vector>>> &psis, + const std::vector>> &lcaos, + const double norm_threshold = std::numeric_limits::min() ); - static std::vector>>> pca( - const UnitCell &ucell, - const LCAO_Orbitals& orb, - const std::vector>> &abfs, - const std::vector>> &orbs, - const double kmesh_times_mot, - const double times_threshold ); - - static std::vector>>> div_r( - const std::vector>>> &psirs, - const std::vector &r_radial ); - - static std::vector>> orbital( - const std::vector>>> &psis, - const std::vector>> &orbs_info, - const double kmesh_times); - - static std::vector>>> get_psi( - const std::vector>> &orbs ); + static std::vector>>> pca( + const UnitCell &ucell, + const LCAO_Orbitals& orb, + const std::vector>> &abfs, + const std::vector>> &orbs, + const double kmesh_times_mot, + const double times_threshold ); + + static std::vector>>> div_r( + const std::vector>>> &psirs, + const std::vector &r_radial ); + + static std::vector>> orbital( + const std::vector>>> &psis, + const std::vector>> &orbs_info, + const double kmesh_times); + + static std::vector>>> get_psi( + const std::vector>> &orbs ); }; #endif // EXX_ABFS_IO_ASA_H diff --git a/source/source_lcao/module_ri/exx_abfs_io.cpp b/source/source_lcao/module_ri/exx_abfs_io.cpp index 0c959dac6d8..0a525c1cfe7 100644 --- a/source/source_lcao/module_ri/exx_abfs_io.cpp +++ b/source/source_lcao/module_ri/exx_abfs_io.cpp @@ -10,250 +10,250 @@ std::vector>> Exx_Abfs::IO::construct_abfs( - const LCAO_Orbitals &orbs, - const std::vector &files_abfs, - const double kmesh_times ) + const LCAO_Orbitals &orbs, + const std::vector &files_abfs, + const double kmesh_times ) { - ModuleBase::TITLE("Exx_Abfs::IO::construct_abfs"); - std::vector>> abfs( files_abfs.size() ); - for( size_t T=0; T!=files_abfs.size(); ++T ) - abfs[T] = construct_abfs_T( - files_abfs[T], - T, - static_cast(orbs.get_kmesh() * kmesh_times) | 1, // Nk must be odd + ModuleBase::TITLE("Exx_Abfs::IO::construct_abfs"); + std::vector>> abfs( files_abfs.size() ); + for( size_t T=0; T!=files_abfs.size(); ++T ) + abfs[T] = construct_abfs_T( + files_abfs[T], + T, + static_cast(orbs.get_kmesh() * kmesh_times) | 1, // Nk must be odd // orbs.get_dk() / kmesh_times, - orbs.get_dk(), // Peize Lin change 2017-04-16 - orbs.get_dr_uniform() ); - - return abfs; + orbs.get_dk(), // Peize Lin change 2017-04-16 + orbs.get_dr_uniform() ); + + return abfs; } std::vector>> Exx_Abfs::IO::construct_abfs( - const std::vector>> & abfs_pre, - const LCAO_Orbitals &orbs, - const std::vector &files_abfs, - const double kmesh_times ) + const std::vector>> & abfs_pre, + const LCAO_Orbitals &orbs, + const std::vector &files_abfs, + const double kmesh_times ) { - std::vector>> - abfs = construct_abfs( orbs, files_abfs, kmesh_times ); - - assert( abfs.size() == abfs_pre.size() ); - for( size_t T=0; T!=abfs.size(); ++T ) - { - if( abfs[T].size() < abfs_pre[T].size() ) - abfs[T].resize( abfs_pre[T].size() ); - for( size_t L=0; L!=abfs_pre[T].size(); ++L ) - { - abfs[T][L].insert( abfs[T][L].begin(), abfs_pre[T][L].begin(), abfs_pre[T][L].end() ); - } - } - - return abfs; + std::vector>> + abfs = construct_abfs( orbs, files_abfs, kmesh_times ); + + assert( abfs.size() == abfs_pre.size() ); + for( size_t T=0; T!=abfs.size(); ++T ) + { + if( abfs[T].size() < abfs_pre[T].size() ) + abfs[T].resize( abfs_pre[T].size() ); + for( size_t L=0; L!=abfs_pre[T].size(); ++L ) + { + abfs[T][L].insert( abfs[T][L].begin(), abfs_pre[T][L].begin(), abfs_pre[T][L].end() ); + } + } + + return abfs; } std::vector> Exx_Abfs::IO::construct_abfs_T( - const std::string & file_name, - const int &T, - const int &nk, - const double &dk, - const double &dr_uniform) + const std::string & file_name, + const int &T, + const int &nk, + const double &dk, + const double &dr_uniform) { - std::string label; - size_t L_size; - std::map N_size; - size_t meshr; - double dr = 0.0; - std::map>> psis; - - /*---------------------- - 1.read abfs - ----------------------*/ - std::string word; - - std::ifstream ifs( file_name.c_str() ); - if(!ifs) - throw std::runtime_error(" Can't find the abfs ORBITAL file " + file_name); - - while( ifs.good() ) - { - ifs >> word; - - if( "Element"==word ) - { - ModuleBase::GlobalFunc::READ_VALUE( ifs, label ); - } - else if ( "Lmax"==word ) - { - ModuleBase::GlobalFunc::READ_VALUE( ifs, L_size ); - - if( L_size>=9 ) - { - std::stringstream ss; - ss<<"Lmax>=9 error in "<<__FILE__<<" line "<<__LINE__; - throw std::invalid_argument(ss.str()); - } - } - else if ( "Sorbital-->"==word ) - { - ModuleBase::GlobalFunc::READ_VALUE( ifs, N_size[0] ); - } - else if ( "Porbital-->"==word ) - { - ModuleBase::GlobalFunc::READ_VALUE( ifs, N_size[1] ); - } - else if ( "Dorbital-->"==word ) - { - ModuleBase::GlobalFunc::READ_VALUE( ifs, N_size[2] ); - } - else if ( "Forbital-->"==word ) - { - ModuleBase::GlobalFunc::READ_VALUE( ifs, N_size[3] ); - } - else if ( "Gorbital-->"==word ) - { - ModuleBase::GlobalFunc::READ_VALUE( ifs, N_size[4] ); - } - else if ( "Horbital-->"==word ) - { - ModuleBase::GlobalFunc::READ_VALUE( ifs, N_size[5] ); - } - else if ( "Iorbital-->"==word ) - { - ModuleBase::GlobalFunc::READ_VALUE( ifs, N_size[6] ); - } - else if ( "Jorbital-->"==word ) - { - ModuleBase::GlobalFunc::READ_VALUE( ifs, N_size[7] ); - } - else if ( "Korbital-->"==word ) - { - ModuleBase::GlobalFunc::READ_VALUE( ifs, N_size[8] ); - } - else if ( "Lorbital-->"==word ) - { - ModuleBase::GlobalFunc::READ_VALUE( ifs, N_size[9] ); - } - else if ( "Morbital-->"==word ) - { - ModuleBase::GlobalFunc::READ_VALUE( ifs, N_size[10] ); - } - else if ( "Norbital-->"==word ) - { - ModuleBase::GlobalFunc::READ_VALUE( ifs, N_size[11] ); - } - else if ( "END"==word ) - { - break; - } - } - - ModuleBase::CHECK_NAME(ifs, "Mesh"); - ifs >> meshr; - - ModuleBase::CHECK_NAME(ifs, "dr"); - ifs >> dr; + std::string label; + size_t L_size; + std::map N_size; + size_t meshr; + double dr = 0.0; + std::map>> psis; + + /*---------------------- + 1.read abfs + ----------------------*/ + std::string word; + + std::ifstream ifs( file_name.c_str() ); + if(!ifs) + throw std::runtime_error(" Can't find the abfs ORBITAL file " + file_name); + + while( ifs.good() ) + { + ifs >> word; + + if( "Element"==word ) + { + ModuleBase::GlobalFunc::READ_VALUE( ifs, label ); + } + else if ( "Lmax"==word ) + { + ModuleBase::GlobalFunc::READ_VALUE( ifs, L_size ); + + if( L_size>=9 ) + { + std::stringstream ss; + ss<<"Lmax>=9 error in "<<__FILE__<<" line "<<__LINE__; + throw std::invalid_argument(ss.str()); + } + } + else if ( "Sorbital-->"==word ) + { + ModuleBase::GlobalFunc::READ_VALUE( ifs, N_size[0] ); + } + else if ( "Porbital-->"==word ) + { + ModuleBase::GlobalFunc::READ_VALUE( ifs, N_size[1] ); + } + else if ( "Dorbital-->"==word ) + { + ModuleBase::GlobalFunc::READ_VALUE( ifs, N_size[2] ); + } + else if ( "Forbital-->"==word ) + { + ModuleBase::GlobalFunc::READ_VALUE( ifs, N_size[3] ); + } + else if ( "Gorbital-->"==word ) + { + ModuleBase::GlobalFunc::READ_VALUE( ifs, N_size[4] ); + } + else if ( "Horbital-->"==word ) + { + ModuleBase::GlobalFunc::READ_VALUE( ifs, N_size[5] ); + } + else if ( "Iorbital-->"==word ) + { + ModuleBase::GlobalFunc::READ_VALUE( ifs, N_size[6] ); + } + else if ( "Jorbital-->"==word ) + { + ModuleBase::GlobalFunc::READ_VALUE( ifs, N_size[7] ); + } + else if ( "Korbital-->"==word ) + { + ModuleBase::GlobalFunc::READ_VALUE( ifs, N_size[8] ); + } + else if ( "Lorbital-->"==word ) + { + ModuleBase::GlobalFunc::READ_VALUE( ifs, N_size[9] ); + } + else if ( "Morbital-->"==word ) + { + ModuleBase::GlobalFunc::READ_VALUE( ifs, N_size[10] ); + } + else if ( "Norbital-->"==word ) + { + ModuleBase::GlobalFunc::READ_VALUE( ifs, N_size[11] ); + } + else if ( "END"==word ) + { + break; + } + } + + ModuleBase::CHECK_NAME(ifs, "Mesh"); + ifs >> meshr; + + ModuleBase::CHECK_NAME(ifs, "dr"); + ifs >> dr; - while(ifs.good()) - { - ifs >> word; - if(word=="Type") - { - std::string s_L, s_N; - ifs >> s_L >> s_N; - - size_t T,L,N; - ifs >> T >> L >> N; - - psis[L][N].resize(meshr); - for(int ir=0; ir!=meshr; ir++) - { - ifs >> psis[L][N][ir]; - } - } + while(ifs.good()) + { + ifs >> word; + if(word=="Type") + { + std::string s_L, s_N; + ifs >> s_L >> s_N; + + size_t T,L,N; + ifs >> T >> L >> N; + + psis[L][N].resize(meshr); + for(int ir=0; ir!=meshr; ir++) + { + ifs >> psis[L][N][ir]; + } + } - } - ifs.close(); + } + ifs.close(); - - /*---------------------- - 2.check L,N orbital - ----------------------*/ - for( size_t L=0; L<=L_size; ++L ) - if( N_size.find(L) == N_size.end() ) - { throw std::domain_error("Can't find N of L="+std::to_string(L)+" in "+file_name); } - for( size_t L=0; L<=L_size; ++L ) - for( size_t N=0; N!=N_size[L]; ++N ) - if( psis.find(L)==psis.end() || psis[L].find(N)==psis[L].end() ) - { throw std::domain_error("Can't find abf of L="+std::to_string(L)+" T="+std::to_string(T)+" in "+file_name); } + + /*---------------------- + 2.check L,N orbital + ----------------------*/ + for( size_t L=0; L<=L_size; ++L ) + if( N_size.find(L) == N_size.end() ) + { throw std::domain_error("Can't find N of L="+std::to_string(L)+" in "+file_name); } + for( size_t L=0; L<=L_size; ++L ) + for( size_t N=0; N!=N_size[L]; ++N ) + if( psis.find(L)==psis.end() || psis[L].find(N)==psis[L].end() ) + { throw std::domain_error("Can't find abf of L="+std::to_string(L)+" T="+std::to_string(T)+" in "+file_name); } - - /*---------------------- - 3.rab, radial - ----------------------*/ - if(meshr%2==0) ++meshr; - - std::vector rab(meshr); - std::vector radial(meshr); - for( int ir=0; ir!=meshr; ++ir ) - { - rab[ir] = dr; - radial[ir] = ir*dr; //mohan 2010-04-19 - } + + /*---------------------- + 3.rab, radial + ----------------------*/ + if(meshr%2==0) ++meshr; + + std::vector rab(meshr); + std::vector radial(meshr); + for( int ir=0; ir!=meshr; ++ir ) + { + rab[ir] = dr; + radial[ir] = ir*dr; //mohan 2010-04-19 + } - - /*---------------------- - 4.normalize psi - ----------------------*/ - for( size_t L=0; L<=L_size; ++L ) - { - for( size_t N=0; N!=N_size[L]; ++N ) - { - std::vector psir(meshr); - std::vector inner(meshr); - psis[L][N].resize(meshr); - for( int ir=0; ir!=meshr; ++ir ) - { - psir[ir] = psis[L][N][ir] * radial[ir]; - inner[ir] = psir[ir] * psir[ir]; - } - double unit = 0.0; - ModuleBase::Integral::Simpson_Integral(meshr, ModuleBase::GlobalFunc::VECTOR_TO_PTR(inner), ModuleBase::GlobalFunc::VECTOR_TO_PTR(rab), unit); - for( int ir=0; ir!=meshr; ++ir ) - { - psis[L][N][ir] /= sqrt(unit); - } - } - } + + /*---------------------- + 4.normalize psi + ----------------------*/ + for( size_t L=0; L<=L_size; ++L ) + { + for( size_t N=0; N!=N_size[L]; ++N ) + { + std::vector psir(meshr); + std::vector inner(meshr); + psis[L][N].resize(meshr); + for( int ir=0; ir!=meshr; ++ir ) + { + psir[ir] = psis[L][N][ir] * radial[ir]; + inner[ir] = psir[ir] * psir[ir]; + } + double unit = 0.0; + ModuleBase::Integral::Simpson_Integral(meshr, ModuleBase::GlobalFunc::VECTOR_TO_PTR(inner), ModuleBase::GlobalFunc::VECTOR_TO_PTR(rab), unit); + for( int ir=0; ir!=meshr; ++ir ) + { + psis[L][N][ir] /= sqrt(unit); + } + } + } - - /*---------------------- - 5.construct abfs - ----------------------*/ - std::vector> abfs_T; + + /*---------------------- + 5.construct abfs + ----------------------*/ + std::vector> abfs_T; - abfs_T.resize(L_size+1); - for( size_t L=0; L<=L_size; ++L ) - { - abfs_T[L].resize(N_size[L]); - for( size_t N=0; N!=N_size[L]; ++N ) - { - abfs_T[L][N].set_orbital_info( - label, - T, //type - L, //angular momentum L - N, // number of orbitals of this L - meshr, // number of radial mesh - ModuleBase::GlobalFunc::VECTOR_TO_PTR(rab), - ModuleBase::GlobalFunc::VECTOR_TO_PTR(radial),// radial mesh value(a.u.) - Numerical_Orbital_Lm::Psi_Type::Psi, - ModuleBase::GlobalFunc::VECTOR_TO_PTR(psis[L][N]), // radial wave function - nk, - dk, - dr_uniform, - false, - true, PARAM.inp.cal_force); - } - } - - return abfs_T; + abfs_T.resize(L_size+1); + for( size_t L=0; L<=L_size; ++L ) + { + abfs_T[L].resize(N_size[L]); + for( size_t N=0; N!=N_size[L]; ++N ) + { + abfs_T[L][N].set_orbital_info( + label, + T, //type + L, //angular momentum L + N, // number of orbitals of this L + meshr, // number of radial mesh + ModuleBase::GlobalFunc::VECTOR_TO_PTR(rab), + ModuleBase::GlobalFunc::VECTOR_TO_PTR(radial),// radial mesh value(a.u.) + Numerical_Orbital_Lm::Psi_Type::Psi, + ModuleBase::GlobalFunc::VECTOR_TO_PTR(psis[L][N]), // radial wave function + nk, + dk, + dr_uniform, + false, + true, PARAM.inp.cal_force); + } + } + + return abfs_T; } diff --git a/source/source_lcao/module_ri/exx_abfs_io.h b/source/source_lcao/module_ri/exx_abfs_io.h index b95fc13235b..9b113a47dce 100644 --- a/source/source_lcao/module_ri/exx_abfs_io.h +++ b/source/source_lcao/module_ri/exx_abfs_io.h @@ -16,25 +16,25 @@ class LCAO_Orbitals; class Exx_Abfs::IO { public: - - static std::vector>> construct_abfs( - const LCAO_Orbitals &orbs, - const std::vector &files_abfs, - const double kmesh_times=1 ); // close dK, keep Kcut - - static std::vector>> construct_abfs( - const std::vector>> &abfs_pre, - const LCAO_Orbitals &orbs, - const std::vector &files_abfs, - const double kmesh_times=1 ); // close dK, keep Kcut + + static std::vector>> construct_abfs( + const LCAO_Orbitals &orbs, + const std::vector &files_abfs, + const double kmesh_times=1 ); // close dK, keep Kcut + + static std::vector>> construct_abfs( + const std::vector>> &abfs_pre, + const LCAO_Orbitals &orbs, + const std::vector &files_abfs, + const double kmesh_times=1 ); // close dK, keep Kcut private: - static std::vector> construct_abfs_T( - const std::string & file_name, - const int &T, - const int &nk, - const double &dk, - const double &dr_uniform); + static std::vector> construct_abfs_T( + const std::string & file_name, + const int &T, + const int &nk, + const double &dk, + const double &dr_uniform); }; #endif // EXX_ABFS_IO_H diff --git a/source/source_lcao/module_ri/exx_abfs_jle.cpp b/source/source_lcao/module_ri/exx_abfs_jle.cpp index 21d409df0d0..fe1182a01a9 100644 --- a/source/source_lcao/module_ri/exx_abfs_jle.cpp +++ b/source/source_lcao/module_ri/exx_abfs_jle.cpp @@ -9,58 +9,58 @@ std::vector>> Exx_Abfs::Jle::init_jle( - const Exx_Info_Opt_ABFs &info, - const double kmesh_times, - const UnitCell& ucell, - const LCAO_Orbitals& orb) + const Exx_Info_Opt_ABFs &info, + const double kmesh_times, + const UnitCell& ucell, + const LCAO_Orbitals& orb) { - ModuleBase::TITLE("Exx_Abfs::Jle","init_jle"); - std::vector>> jle( ucell.ntype ); + ModuleBase::TITLE("Exx_Abfs::Jle","init_jle"); + std::vector>> jle( ucell.ntype ); - for(int T=0; T0) - { jle[T].resize( info.abfs_Lmax+1 ); } - for(int L=0; L<=info.abfs_Lmax; ++L) - { - const size_t ecut_number - = static_cast( std::sqrt( info.ecut_exx ) * orb.Phi[T].getRcut() / ModuleBase::PI ); // Rydberg Unit. - if(ecut_number<=0) - { continue; } + for(int T=0; T0) + { jle[T].resize( info.abfs_Lmax+1 ); } + for(int L=0; L<=info.abfs_Lmax; ++L) + { + const size_t ecut_number + = static_cast( std::sqrt( info.ecut_exx ) * orb.Phi[T].getRcut() / ModuleBase::PI ); // Rydberg Unit. + if(ecut_number<=0) + { continue; } - jle[T][L].resize( ecut_number ); + jle[T][L].resize( ecut_number ); - std::vector en(ecut_number, 0.0); - ModuleBase::Sphbes::Spherical_Bessel_Roots(ecut_number, L, info.tolerence, en.data(), orb.Phi[T].getRcut()); + std::vector en(ecut_number, 0.0); + ModuleBase::Sphbes::Spherical_Bessel_Roots(ecut_number, L, info.tolerence, en.data(), orb.Phi[T].getRcut()); - for(size_t E=0; E jle_r( orb.Phi[T].PhiLN(0,0).getNr() ); - ModuleBase::Sphbes::Spherical_Bessel( - orb.Phi[T].PhiLN(0,0).getNr(), - orb.Phi[T].PhiLN(0,0).getRadial(), - en[E], - L, - jle_r.data()); - jle[T][L][E].set_orbital_info( - orb.Phi[T].PhiLN(0,0).getLabel(), - orb.Phi[T].PhiLN(0,0).getType(), - L, - E, // N? - orb.Phi[T].PhiLN(0,0).getNr(), - orb.Phi[T].PhiLN(0,0).getRab(), - orb.Phi[T].PhiLN(0,0).getRadial(), - Numerical_Orbital_Lm::Psi_Type::Psi, - jle_r.data(), - static_cast(orb.Phi[T].PhiLN(0,0).getNk() * kmesh_times) | 1, - orb.Phi[T].PhiLN(0,0).getDk(), - orb.Phi[T].PhiLN(0,0).getDruniform(), - false, - true, - PARAM.inp.cal_force); - } - } - } + for(size_t E=0; E jle_r( orb.Phi[T].PhiLN(0,0).getNr() ); + ModuleBase::Sphbes::Spherical_Bessel( + orb.Phi[T].PhiLN(0,0).getNr(), + orb.Phi[T].PhiLN(0,0).getRadial(), + en[E], + L, + jle_r.data()); + jle[T][L][E].set_orbital_info( + orb.Phi[T].PhiLN(0,0).getLabel(), + orb.Phi[T].PhiLN(0,0).getType(), + L, + E, // N? + orb.Phi[T].PhiLN(0,0).getNr(), + orb.Phi[T].PhiLN(0,0).getRab(), + orb.Phi[T].PhiLN(0,0).getRadial(), + Numerical_Orbital_Lm::Psi_Type::Psi, + jle_r.data(), + static_cast(orb.Phi[T].PhiLN(0,0).getNk() * kmesh_times) | 1, + orb.Phi[T].PhiLN(0,0).getDk(), + orb.Phi[T].PhiLN(0,0).getDruniform(), + false, + true, + PARAM.inp.cal_force); + } + } + } - return jle; + return jle; } diff --git a/source/source_lcao/module_ri/exx_abfs_jle.h b/source/source_lcao/module_ri/exx_abfs_jle.h index 4b31b5c0137..51e00d3b122 100644 --- a/source/source_lcao/module_ri/exx_abfs_jle.h +++ b/source/source_lcao/module_ri/exx_abfs_jle.h @@ -7,18 +7,18 @@ #include - class LCAO_Orbitals; - class UnitCell; + class LCAO_Orbitals; + class UnitCell; class Exx_Abfs::Jle { public: - static std::vector>> - init_jle( - const Exx_Info_Opt_ABFs &info, - const double kmesh_times, - const UnitCell& ucell, - const LCAO_Orbitals& orb); + static std::vector>> + init_jle( + const Exx_Info_Opt_ABFs &info, + const double kmesh_times, + const UnitCell& ucell, + const LCAO_Orbitals& orb); }; #endif // EXX_ABFS_JLE_H diff --git a/source/source_lcao/module_ri/exx_lri.h b/source/source_lcao/module_ri/exx_lri.h index 3085c9e571c..67eaa321c28 100644 --- a/source/source_lcao/module_ri/exx_lri.h +++ b/source/source_lcao/module_ri/exx_lri.h @@ -21,52 +21,52 @@ #include "module_exx_symmetry/symm_rotation.h" - class Parallel_Orbitals; + class Parallel_Orbitals; - template - class RPA_LRI; + template + class RPA_LRI; - template - class Exx_LRI_Interface; + template + class Exx_LRI_Interface; - namespace LR - { - template - class OperatorLREXX; - } + namespace LR + { + template + class OperatorLREXX; + } template class Exx_Obj { - // match with Conv_Coulomb_Pot_K::Coulomb_Method - public: - LRI_CV cv; - Ewald_Vq evq; - std::vector>> abfs_ccp; + // match with Conv_Coulomb_Pot_K::Coulomb_Method + public: + LRI_CV cv; + Ewald_Vq evq; + std::vector>> abfs_ccp; }; template class Exx_LRI { private: - using TA = int; - using Tcell = int; - static constexpr std::size_t Ndim = 3; - using TC = std::array; - using TAC = std::pair; - using TatomR = std::array; // tmp + using TA = int; + using Tcell = int; + static constexpr std::size_t Ndim = 3; + using TC = std::array; + using TAC = std::pair; + using TatomR = std::array; // tmp public: - Exx_LRI(const Exx_Info_RI& info_in) :info(info_in) {} - Exx_LRI operator=(const Exx_LRI&) = delete; - Exx_LRI operator=(Exx_LRI&&); - - void init( - const MPI_Comm &mpi_comm_in, - const UnitCell &ucell, - const K_Vectors &kv_in, - const LCAO_Orbitals& orb, - const std::vector>>& abfs_in = {}); + Exx_LRI(const Exx_Info_RI& info_in) :info(info_in) {} + Exx_LRI operator=(const Exx_LRI&) = delete; + Exx_LRI operator=(Exx_LRI&&); + + void init( + const MPI_Comm &mpi_comm_in, + const UnitCell &ucell, + const K_Vectors &kv_in, + const LCAO_Orbitals& orb, + const std::vector>>& abfs_in = {}); void init_spencer(const MPI_Comm& mpi_comm_in, const UnitCell& ucell, const K_Vectors& kv_in, @@ -74,76 +74,74 @@ class Exx_LRI const std::vector>>& abfs_in = {}); void cal_exx_ions(const UnitCell& ucell, const bool write_cv = false); void cal_cut_coulomb_cs( - std::map>>& Vs_cut_IJR, - std::map>>& Cs, - const UnitCell& ucell, - const bool write_cv = false); - void cal_ewald_coulomb( - std::map>>& Vs_full_IJR, - std::map>>& Cs, - const UnitCell& ucell, - const bool write_cv = false); - void cal_exx_elec( - const std::vector>>>& Ds, - const UnitCell& ucell, - const Parallel_Orbitals& pv, - const ModuleSymmetry::Symmetry_rotation* p_symrot = nullptr); - // (nspin=4) real-space symmetry EXX: the spinor H(R) rotation couples the 4 spin channels via - // the SU(2) part U(isym), so the 4 channels must be rotated together (not one-per-outer-loop). - // Gathers the irreducible Hs of all 4 channels, calls Symmetry_rotation::restore_HR_nspin4, then - // finishes energy/gather per channel. Called from cal_exx_elec when p_symrot && nspin==4. - void cal_exx_elec_soc( - const std::vector>>>& Ds, - const UnitCell& ucell, - const std::vector, std::set>>& judge, - const ModuleSymmetry::Symmetry_rotation* p_symrot); - void cal_exx_force(const int& nat); - void cal_exx_stress(const double& omega, const double& lat0); - void cal_exx_dHs(const std::vector>>>& Ds, - const UnitCell& ucell, - const Parallel_Orbitals& pv); - - void reset_Cs(const std::map>>& Cs_in) { this->exx_lri.set_Cs(Cs_in, this->info.C_threshold); } - void reset_Vs(const std::map>>& Vs_in) { this->exx_lri.set_Vs(Vs_in, this->info.V_threshold); } - //std::vector> get_abfs_nchis() const; - - std::vector< std::map>>> Hexxs; - std::array>>>>, 3> dHexxs; // direction, atom, spin, (i,j,R) - double Eexx; - ModuleBase::matrix force_exx; - ModuleBase::matrix stress_exx; - - int abfs_Lmax() const { return abfs_Lmax_; } - const Exx_Info_RI& get_info_ri() const { return info; } + std::map>>& Vs_cut_IJR, + std::map>>& Cs, + const UnitCell& ucell, + const bool write_cv = false); + void cal_ewald_coulomb( + std::map>>& Vs_full_IJR, + std::map>>& Cs, + const UnitCell& ucell, + const bool write_cv = false); + void cal_exx_elec( + const std::vector>>>& Ds, + const UnitCell& ucell, + const Parallel_Orbitals& pv, + const ModuleSymmetry::Symmetry_rotation* p_symrot = nullptr); + // (nspin=4) real-space symmetry EXX: the spinor H(R) rotation couples the 4 spin channels via + // the SU(2) part U(isym), so the 4 channels must be rotated together (not one-per-outer-loop). + // Gathers the irreducible Hs of all 4 channels, calls Symmetry_rotation::restore_HR_nspin4, then + // finishes energy/gather per channel. Called from cal_exx_elec when p_symrot && nspin==4. + void cal_exx_elec_soc( + const std::vector>>>& Ds, + const UnitCell& ucell, + const std::vector, std::set>>& judge, + const ModuleSymmetry::Symmetry_rotation* p_symrot); + void cal_exx_force(const int& nat); + void cal_exx_stress(const double& omega, const double& lat0); + + void reset_Cs(const std::map>>& Cs_in) { this->exx_lri.set_Cs(Cs_in, this->info.C_threshold); } + void reset_Vs(const std::map>>& Vs_in) { this->exx_lri.set_Vs(Vs_in, this->info.V_threshold); } + //std::vector> get_abfs_nchis() const; + + std::vector< std::map>>> Hexxs; + double Eexx = 0.0; // initialize: in the HexxR-restart branch (init_chg=file) Eexx is + // never computed, and OperatorEXX still reads get_Eexx() for the NSCF + // energy report; leaving it uninitialized leaked an indeterminate value. + ModuleBase::matrix force_exx; + ModuleBase::matrix stress_exx; + + int abfs_Lmax() const { return abfs_Lmax_; } + const Exx_Info_RI& get_info_ri() const { return info; } private: - Exx_Info_RI info; - int abfs_Lmax_ = 0; - MPI_Comm mpi_comm; - const K_Vectors *p_kv = nullptr; - std::shared_ptr MGT; - std::vector orb_cutoff_; - - std::vector>> lcaos; - std::vector>> abfs; - //std::vector>> abfs_ccp; - std::map> exx_objs; - //LRI_CV cv; - RI::Exx exx_lri; - std::map MGT; + std::vector orb_cutoff_; + + std::vector>> lcaos; + std::vector>> abfs; + //std::vector>> abfs_ccp; + std::map> exx_objs; + //LRI_CV cv; + RI::Exx exx_lri; + std::map>>>> coulomb_settings; - void post_process_Hexx( std::map>> &Hexxs_io ) const; - double post_process_Eexx(const double& Eexx_in) const; + void post_process_Hexx( std::map>> &Hexxs_io ) const; + double post_process_Eexx(const double& Eexx_in) const; - friend class RPA_LRI; - friend class RPA_LRI, Tdata>; - friend class Exx_LRI_Interface; - friend class Exx_LRI_Interface, Tdata>; - friend class LR::OperatorLREXX; - friend class LR::OperatorLREXX>; + friend class RPA_LRI; + friend class RPA_LRI, Tdata>; + friend class Exx_LRI_Interface; + friend class Exx_LRI_Interface, Tdata>; + friend class LR::OperatorLREXX; + friend class LR::OperatorLREXX>; }; #include "exx_lri.hpp" diff --git a/source/source_lcao/module_ri/exx_lri.hpp b/source/source_lcao/module_ri/exx_lri.hpp index 9cb8f75d720..04a134a7748 100644 --- a/source/source_lcao/module_ri/exx_lri.hpp +++ b/source/source_lcao/module_ri/exx_lri.hpp @@ -32,1010 +32,935 @@ template void Exx_LRI::init(const MPI_Comm &mpi_comm_in, - const UnitCell &ucell, - const K_Vectors &kv_in, - const LCAO_Orbitals& orb, - const std::vector>>& abfs_in) + const UnitCell &ucell, + const K_Vectors &kv_in, + const LCAO_Orbitals& orb, + const std::vector>>& abfs_in) { - ModuleBase::TITLE("Exx_LRI","init"); - ModuleBase::timer::start("Exx_LRI", "init"); - - this->mpi_comm = mpi_comm_in; - this->p_kv = &kv_in; - this->orb_cutoff_ = orb.cutoffs(); - - this->lcaos = Exx_Abfs::Construct_Orbs::change_orbs( orb, this->info.kmesh_times ); - Exx_Abfs::Construct_Orbs::filter_empty_orbs(this->lcaos); - - if(abfs_in.empty()) - { - const std::vector>> - abfs_same_atom = Exx_Abfs::Construct_Orbs::abfs_same_atom(ucell, orb, this->lcaos, this->info.kmesh_times, this->info.pca_threshold ); - if(this->info.files_abfs.empty()) - { this->abfs = abfs_same_atom;} - else - { this->abfs = Exx_Abfs::IO::construct_abfs( abfs_same_atom, orb, this->info.files_abfs, this->info.kmesh_times ); } - } - else - { - this->abfs = abfs_in; - } - Exx_Abfs::Construct_Orbs::filter_empty_orbs(this->abfs); - Exx_Abfs::Construct_Orbs::print_orbs_size(ucell, this->abfs, GlobalV::ofs_running); - - for( size_t T=0; T!=this->abfs.size(); ++T ) - { - this->abfs_Lmax_ = std::max(this->abfs_Lmax_, static_cast(this->abfs[T].size())-1); - } - - this->exx_objs.clear(); - this->coulomb_settings = RI_Util::update_coulomb_settings(this->info.coulomb_param, ucell, this->p_kv); - - this->MGT = std::make_shared(); - for(const auto &settings_list : this->coulomb_settings) - { - this->exx_objs[settings_list.first].abfs_ccp = Conv_Coulomb_Pot_K::cal_orbs_ccp(this->abfs, settings_list.second.second, this->info.ccp_rmesh_times); - this->exx_objs[settings_list.first].cv.set_orbitals(ucell, orb, - this->lcaos, this->abfs, this->exx_objs[settings_list.first].abfs_ccp, - this->info.kmesh_times, this->MGT, settings_list.second.first ); - this->exx_objs[settings_list.first].cv.set_info_ri(&this->info); - if (settings_list.first == Conv_Coulomb_Pot_K::Coulomb_Method::Ewald) - { - const int evq_abfs_Lmax = this->abfs_Lmax_; - this->exx_objs[settings_list.first].evq.init(ucell, orb, - this->mpi_comm, this->p_kv, this->lcaos, this->abfs, - settings_list.second.second, this->MGT, this->info.ccp_rmesh_times, this->info.kmesh_times, - evq_abfs_Lmax); - } - } - - ModuleBase::timer::end("Exx_LRI", "init"); + ModuleBase::TITLE("Exx_LRI","init"); + ModuleBase::timer::start("Exx_LRI", "init"); + + this->mpi_comm = mpi_comm_in; + this->p_kv = &kv_in; + this->orb_cutoff_ = orb.cutoffs(); + + this->lcaos = Exx_Abfs::Construct_Orbs::change_orbs( orb, this->info.kmesh_times ); + Exx_Abfs::Construct_Orbs::filter_empty_orbs(this->lcaos); + + if(abfs_in.empty()) + { + const std::vector>> + abfs_same_atom = Exx_Abfs::Construct_Orbs::abfs_same_atom(ucell, orb, this->lcaos, this->info.kmesh_times, this->info.pca_threshold ); + if(this->info.files_abfs.empty()) + { this->abfs = abfs_same_atom;} + else + { this->abfs = Exx_Abfs::IO::construct_abfs( abfs_same_atom, orb, this->info.files_abfs, this->info.kmesh_times ); } + } + else + { + this->abfs = abfs_in; + } + Exx_Abfs::Construct_Orbs::filter_empty_orbs(this->abfs); + Exx_Abfs::Construct_Orbs::print_orbs_size(ucell, this->abfs, GlobalV::ofs_running); + + for( size_t T=0; T!=this->abfs.size(); ++T ) + { + this->abfs_Lmax_ = std::max(this->abfs_Lmax_, static_cast(this->abfs[T].size())-1); + } + + this->exx_objs.clear(); + this->coulomb_settings = RI_Util::update_coulomb_settings(this->info.coulomb_param, ucell, this->p_kv); + + this->MGT = std::make_shared(); + for(const auto &settings_list : this->coulomb_settings) + { + this->exx_objs[settings_list.first].abfs_ccp = Conv_Coulomb_Pot_K::cal_orbs_ccp(this->abfs, settings_list.second.second, this->info.ccp_rmesh_times); + this->exx_objs[settings_list.first].cv.set_orbitals(ucell, orb, + this->lcaos, this->abfs, this->exx_objs[settings_list.first].abfs_ccp, + this->info.kmesh_times, this->MGT, settings_list.second.first ); + this->exx_objs[settings_list.first].cv.set_info_ri(&this->info); + if (settings_list.first == Conv_Coulomb_Pot_K::Coulomb_Method::Ewald) + { + const int evq_abfs_Lmax = this->abfs_Lmax_; + this->exx_objs[settings_list.first].evq.init(ucell, orb, + this->mpi_comm, this->p_kv, this->lcaos, this->abfs, + settings_list.second.second, this->MGT, this->info.ccp_rmesh_times, this->info.kmesh_times, + evq_abfs_Lmax); + } + } + + ModuleBase::timer::end("Exx_LRI", "init"); } template void Exx_LRI::init_spencer( - const MPI_Comm& mpi_comm_in, - const UnitCell& ucell, - const K_Vectors& kv_in, - const LCAO_Orbitals& orb, - const std::vector>>& abfs_in) + const MPI_Comm& mpi_comm_in, + const UnitCell& ucell, + const K_Vectors& kv_in, + const LCAO_Orbitals& orb, + const std::vector>>& abfs_in) { - ModuleBase::TITLE("Exx_LRI", "init_spencer"); - ModuleBase::timer::start("Exx_LRI", "init_spencer"); - - this->mpi_comm = mpi_comm_in; - this->p_kv = &kv_in; - this->orb_cutoff_ = orb.cutoffs(); - - this->lcaos = Exx_Abfs::Construct_Orbs::change_orbs(orb, this->info.kmesh_times); - Exx_Abfs::Construct_Orbs::filter_empty_orbs(this->lcaos); - - if (abfs_in.empty()) - { - const std::vector>> abfs_same_atom - = Exx_Abfs::Construct_Orbs::abfs_same_atom( - ucell, orb, this->lcaos, this->info.kmesh_times, this->info.pca_threshold); - if (this->info.files_abfs.empty()) - { this->abfs = abfs_same_atom; } - else - { this->abfs = Exx_Abfs::IO::construct_abfs(abfs_same_atom, orb, this->info.files_abfs, this->info.kmesh_times); } - } - else - { - this->abfs = abfs_in; - } - Exx_Abfs::Construct_Orbs::filter_empty_orbs(this->abfs); - Exx_Abfs::Construct_Orbs::print_orbs_size(ucell, this->abfs, GlobalV::ofs_running); - - for (size_t T = 0; T != this->abfs.size(); ++T) - { - this->abfs_Lmax_ = std::max(this->abfs_Lmax_, static_cast(this->abfs[T].size()) - 1); - } - - this->exx_objs.clear(); - this->coulomb_settings.clear(); - this->coulomb_settings[Conv_Coulomb_Pot_K::Coulomb_Method::Center2] = std::make_pair( - true, ExxLriDetail::build_center2_cut_coulomb_param(this->info.coulomb_param, ucell, kv_in)); - - this->MGT = std::make_shared(); - const auto center2_settings = this->coulomb_settings.find(Conv_Coulomb_Pot_K::Coulomb_Method::Center2); - if (center2_settings == this->coulomb_settings.end()) - { throw std::invalid_argument("Exx_LRI::init_spencer failed to prepare Center2 settings."); } - - this->exx_objs[Conv_Coulomb_Pot_K::Coulomb_Method::Center2].abfs_ccp = Conv_Coulomb_Pot_K::cal_orbs_ccp_spencer( - this->abfs, - center2_settings->second.second, - this->info.ccp_rmesh_times); - this->exx_objs[Conv_Coulomb_Pot_K::Coulomb_Method::Center2].cv.set_orbitals( - ucell, - orb, - this->lcaos, - this->abfs, - this->exx_objs[Conv_Coulomb_Pot_K::Coulomb_Method::Center2].abfs_ccp, - this->info.kmesh_times, - this->MGT, - center2_settings->second.first); - this->exx_objs[Conv_Coulomb_Pot_K::Coulomb_Method::Center2].cv.set_info_ri(&this->info); - - ModuleBase::timer::end("Exx_LRI", "init_spencer"); + ModuleBase::TITLE("Exx_LRI", "init_spencer"); + ModuleBase::timer::start("Exx_LRI", "init_spencer"); + + this->mpi_comm = mpi_comm_in; + this->p_kv = &kv_in; + this->orb_cutoff_ = orb.cutoffs(); + + this->lcaos = Exx_Abfs::Construct_Orbs::change_orbs(orb, this->info.kmesh_times); + Exx_Abfs::Construct_Orbs::filter_empty_orbs(this->lcaos); + + if (abfs_in.empty()) + { + const std::vector>> abfs_same_atom + = Exx_Abfs::Construct_Orbs::abfs_same_atom( + ucell, orb, this->lcaos, this->info.kmesh_times, this->info.pca_threshold); + if (this->info.files_abfs.empty()) + { this->abfs = abfs_same_atom; } + else + { this->abfs = Exx_Abfs::IO::construct_abfs(abfs_same_atom, orb, this->info.files_abfs, this->info.kmesh_times); } + } + else + { + this->abfs = abfs_in; + } + Exx_Abfs::Construct_Orbs::filter_empty_orbs(this->abfs); + Exx_Abfs::Construct_Orbs::print_orbs_size(ucell, this->abfs, GlobalV::ofs_running); + + for (size_t T = 0; T != this->abfs.size(); ++T) + { + this->abfs_Lmax_ = std::max(this->abfs_Lmax_, static_cast(this->abfs[T].size()) - 1); + } + + this->exx_objs.clear(); + this->coulomb_settings.clear(); + this->coulomb_settings[Conv_Coulomb_Pot_K::Coulomb_Method::Center2] = std::make_pair( + true, ExxLriDetail::build_center2_cut_coulomb_param(this->info.coulomb_param, ucell, kv_in)); + + this->MGT = std::make_shared(); + const auto center2_settings = this->coulomb_settings.find(Conv_Coulomb_Pot_K::Coulomb_Method::Center2); + if (center2_settings == this->coulomb_settings.end()) + { throw std::invalid_argument("Exx_LRI::init_spencer failed to prepare Center2 settings."); } + + this->exx_objs[Conv_Coulomb_Pot_K::Coulomb_Method::Center2].abfs_ccp = Conv_Coulomb_Pot_K::cal_orbs_ccp_spencer( + this->abfs, + center2_settings->second.second, + this->info.ccp_rmesh_times); + this->exx_objs[Conv_Coulomb_Pot_K::Coulomb_Method::Center2].cv.set_orbitals( + ucell, + orb, + this->lcaos, + this->abfs, + this->exx_objs[Conv_Coulomb_Pot_K::Coulomb_Method::Center2].abfs_ccp, + this->info.kmesh_times, + this->MGT, + center2_settings->second.first); + this->exx_objs[Conv_Coulomb_Pot_K::Coulomb_Method::Center2].cv.set_info_ri(&this->info); + + ModuleBase::timer::end("Exx_LRI", "init_spencer"); } template void Exx_LRI::cal_exx_ions(const UnitCell& ucell, - const bool write_cv) + const bool write_cv) { - ModuleBase::TITLE("Exx_LRI","cal_exx_ions"); - ModuleBase::timer::start("Exx_LRI", "cal_exx_ions"); - - // init_radial_table_ions( cal_atom_centres_core(atom_pairs_core_origin), atom_pairs_core_origin ); - - // this->m_abfsabfs.init_radial_table(Rradial); - // this->m_abfslcaos_lcaos.init_radial_table(Rradial); - - std::vector atoms(ucell.nat); - for(int iat=0; iat atoms_pos; - for(int iat=0; iat latvec - = {RI_Util::Vector3_to_array3(ucell.a1), - RI_Util::Vector3_to_array3(ucell.a2), - RI_Util::Vector3_to_array3(ucell.a3)}; - const std::array period = {this->p_kv->nmp[0], this->p_kv->nmp[1], this->p_kv->nmp[2]}; - - this->exx_lri.set_parallel(this->mpi_comm, atoms_pos, latvec, period); - - // std::max(3) for gamma_only, list_A2 should contain cell {-1,0,1}. In the future distribute will be neighbour. - const std::array period_Vs = LRI_CV_Tools::cal_latvec_range(1+this->info.ccp_rmesh_times, ucell, orb_cutoff_); - const std::pair, std::vector>>>> - list_As_Vs = RI::Distribute_Equally::distribute_atoms_periods(this->mpi_comm, atoms, period_Vs, 2, false); - - std::map>> Vs; - std::map, Ndim>>> dVs; - const bool cal_dCV = PARAM.inp.cal_force || PARAM.inp.cal_stress || PARAM.inp.out_mat_dh_exx[0]; - for(const auto &settings_list : this->coulomb_settings) - { - std::map>> - Vs_temp = this->exx_objs[settings_list.first].cv.cal_Vs(ucell, - list_As_Vs.first, list_As_Vs.second[0], - {{"writable_Vws",true}}); - this->exx_objs[settings_list.first].cv.Vws = LRI_CV_Tools::get_CVws(ucell,Vs_temp); - if (settings_list.first == Conv_Coulomb_Pot_K::Coulomb_Method::Ewald) - { - this->exx_objs[settings_list.first].evq.init_ions(ucell, period_Vs); - const auto &coulomb_param = settings_list.second.second; - std::map>> Vs_ewald; - for(const auto ¶m_list : coulomb_param) - { - std::map>> Vs_ewald_temp; - switch(param_list.first) - { - case Conv_Coulomb_Pot_K::Coulomb_Type::Fock: - { - double chi = this->exx_objs[settings_list.first].evq.get_singular_chi(ucell, param_list.second, 2.0); - Vs_ewald_temp = this->exx_objs[settings_list.first].evq.cal_Vs(ucell, chi, Vs_temp); - break; - } - default: - { - throw std::invalid_argument( std::string(__FILE__) + " line " + std::to_string(__LINE__) ); - } - } - // Vs_temp cannot be covered here - Vs_ewald = Vs_ewald.empty() ? Vs_ewald_temp : LRI_CV_Tools::add(Vs_ewald, Vs_ewald_temp); - } - Vs_temp = Vs_ewald; - } - Vs = Vs.empty() ? Vs_temp : LRI_CV_Tools::add(Vs, Vs_temp); + ModuleBase::TITLE("Exx_LRI","cal_exx_ions"); + ModuleBase::timer::start("Exx_LRI", "cal_exx_ions"); + + // init_radial_table_ions( cal_atom_centres_core(atom_pairs_core_origin), atom_pairs_core_origin ); + + // this->m_abfsabfs.init_radial_table(Rradial); + // this->m_abfslcaos_lcaos.init_radial_table(Rradial); + + std::vector atoms(ucell.nat); + for(int iat=0; iat atoms_pos; + for(int iat=0; iat latvec + = {RI_Util::Vector3_to_array3(ucell.a1), + RI_Util::Vector3_to_array3(ucell.a2), + RI_Util::Vector3_to_array3(ucell.a3)}; + const std::array period = {this->p_kv->nmp[0], this->p_kv->nmp[1], this->p_kv->nmp[2]}; + + this->exx_lri.set_parallel(this->mpi_comm, atoms_pos, latvec, period); + + // std::max(3) for gamma_only, list_A2 should contain cell {-1,0,1}. In the future distribute will be neighbour. + const std::array period_Vs = LRI_CV_Tools::cal_latvec_range(1+this->info.ccp_rmesh_times, ucell, orb_cutoff_); + const std::pair, std::vector>>>> + list_As_Vs = RI::Distribute_Equally::distribute_atoms_periods(this->mpi_comm, atoms, period_Vs, 2, false); + + std::map>> Vs; + std::map, Ndim>>> dVs; + const bool cal_dCV = PARAM.inp.cal_force || PARAM.inp.cal_stress; + for(const auto &settings_list : this->coulomb_settings) + { + std::map>> + Vs_temp = this->exx_objs[settings_list.first].cv.cal_Vs(ucell, + list_As_Vs.first, list_As_Vs.second[0], + {{"writable_Vws",true}}); + this->exx_objs[settings_list.first].cv.Vws = LRI_CV_Tools::get_CVws(ucell,Vs_temp); + if (settings_list.first == Conv_Coulomb_Pot_K::Coulomb_Method::Ewald) + { + this->exx_objs[settings_list.first].evq.init_ions(ucell, period_Vs); + const auto &coulomb_param = settings_list.second.second; + std::map>> Vs_ewald; + for(const auto ¶m_list : coulomb_param) + { + std::map>> Vs_ewald_temp; + switch(param_list.first) + { + case Conv_Coulomb_Pot_K::Coulomb_Type::Fock: + { + double chi = this->exx_objs[settings_list.first].evq.get_singular_chi(ucell, param_list.second, 2.0); + Vs_ewald_temp = this->exx_objs[settings_list.first].evq.cal_Vs(ucell, chi, Vs_temp); + break; + } + default: + { + throw std::invalid_argument( std::string(__FILE__) + " line " + std::to_string(__LINE__) ); + } + } + // Vs_temp cannot be covered here + Vs_ewald = Vs_ewald.empty() ? Vs_ewald_temp : LRI_CV_Tools::add(Vs_ewald, Vs_ewald_temp); + } + Vs_temp = Vs_ewald; + } + Vs = Vs.empty() ? Vs_temp : LRI_CV_Tools::add(Vs, Vs_temp); if (cal_dCV) - { - std::map, Ndim>>> - dVs_temp = this->exx_objs[settings_list.first].cv.cal_dVs(ucell, - list_As_Vs.first, list_As_Vs.second[0], - {{"writable_dVws",true}}); - this->exx_objs[settings_list.first].cv.dVws = LRI_CV_Tools::get_dCVws(ucell,dVs_temp); - dVs = dVs.empty() ? dVs_temp : LRI_CV_Tools::add(dVs, dVs_temp); - } - } - if (write_cv && GlobalV::MY_RANK == 0) - { LRI_CV_Tools::write_Vs_abf(Vs, PARAM.globalv.global_out_dir + "Vs"); } - this->exx_lri.set_Vs(std::move(Vs), this->info.V_threshold); + { + std::map, Ndim>>> + dVs_temp = this->exx_objs[settings_list.first].cv.cal_dVs(ucell, + list_As_Vs.first, list_As_Vs.second[0], + {{"writable_dVws",true}}); + this->exx_objs[settings_list.first].cv.dVws = LRI_CV_Tools::get_dCVws(ucell,dVs_temp); + dVs = dVs.empty() ? dVs_temp : LRI_CV_Tools::add(dVs, dVs_temp); + } + } + if (write_cv && GlobalV::MY_RANK == 0) + { LRI_CV_Tools::write_Vs_abf(Vs, PARAM.globalv.global_out_dir + "Vs"); } + this->exx_lri.set_Vs(std::move(Vs), this->info.V_threshold); if (cal_dCV) - { - std::array>>, Ndim> - dVs_order = LRI_CV_Tools::change_order(std::move(dVs)); - this->exx_lri.set_dVs(std::move(dVs_order), this->info.V_grad_threshold); - if(PARAM.inp.cal_stress) - { - std::array>>,3>,3> dVRs = LRI_CV_Tools::cal_dMRs(ucell,dVs_order); - this->exx_lri.set_dVRs(std::move(dVRs), this->info.V_grad_R_threshold); - } - } - - const std::array period_Cs = LRI_CV_Tools::cal_latvec_range(2, ucell,orb_cutoff_); - const std::pair, std::vector>>>> - list_As_Cs = RI::Distribute_Equally::distribute_atoms_periods(this->mpi_comm, atoms, period_Cs, 2, false); - - std::map>> Cs; - std::map, 3>>> dCs; - for(const auto &settings_list : this->coulomb_settings) - { - if(settings_list.second.first) - { - std::pair>>, - std::map, 3>>>> - Cs_dCs = this->exx_objs[settings_list.first].cv.cal_Cs_dCs( - ucell, - list_As_Cs.first, list_As_Cs.second[0], + { + std::array>>, Ndim> + dVs_order = LRI_CV_Tools::change_order(std::move(dVs)); + this->exx_lri.set_dVs(std::move(dVs_order), this->info.V_grad_threshold); + if(PARAM.inp.cal_stress) + { + std::array>>,3>,3> dVRs = LRI_CV_Tools::cal_dMRs(ucell,dVs_order); + this->exx_lri.set_dVRs(std::move(dVRs), this->info.V_grad_R_threshold); + } + } + + const std::array period_Cs = LRI_CV_Tools::cal_latvec_range(2, ucell,orb_cutoff_); + const std::pair, std::vector>>>> + list_As_Cs = RI::Distribute_Equally::distribute_atoms_periods(this->mpi_comm, atoms, period_Cs, 2, false); + + std::map>> Cs; + std::map, 3>>> dCs; + for(const auto &settings_list : this->coulomb_settings) + { + if(settings_list.second.first) + { + std::pair>>, + std::map, 3>>>> + Cs_dCs = this->exx_objs[settings_list.first].cv.cal_Cs_dCs( + ucell, + list_As_Cs.first, list_As_Cs.second[0], { {"cal_dC",cal_dCV}, - {"writable_Cws",true}, {"writable_dCws",true}, {"writable_Vws",false}, {"writable_dVws",false}}); - std::map>> &Cs_temp = std::get<0>(Cs_dCs); - this->exx_objs[settings_list.first].cv.Cws = LRI_CV_Tools::get_CVws(ucell,Cs_temp); - Cs = Cs.empty() ? Cs_temp : LRI_CV_Tools::add(Cs, Cs_temp); + {"writable_Cws",true}, {"writable_dCws",true}, {"writable_Vws",false}, {"writable_dVws",false}}); + std::map>> &Cs_temp = std::get<0>(Cs_dCs); + this->exx_objs[settings_list.first].cv.Cws = LRI_CV_Tools::get_CVws(ucell,Cs_temp); + Cs = Cs.empty() ? Cs_temp : LRI_CV_Tools::add(Cs, Cs_temp); if (cal_dCV) - { - std::map, 3>>> &dCs_temp = std::get<1>(Cs_dCs); - this->exx_objs[settings_list.first].cv.dCws = LRI_CV_Tools::get_dCVws(ucell,dCs_temp); - dCs = dCs.empty() ? dCs_temp : LRI_CV_Tools::add(dCs, dCs_temp); - } - } - } - if (write_cv && GlobalV::MY_RANK == 0) - { LRI_CV_Tools::write_Cs_ao(Cs, PARAM.globalv.global_out_dir + "Cs"); } - this->exx_lri.set_Cs(std::move(Cs), this->info.C_threshold); + { + std::map, 3>>> &dCs_temp = std::get<1>(Cs_dCs); + this->exx_objs[settings_list.first].cv.dCws = LRI_CV_Tools::get_dCVws(ucell,dCs_temp); + dCs = dCs.empty() ? dCs_temp : LRI_CV_Tools::add(dCs, dCs_temp); + } + } + } + if (write_cv && GlobalV::MY_RANK == 0) + { LRI_CV_Tools::write_Cs_ao(Cs, PARAM.globalv.global_out_dir + "Cs"); } + this->exx_lri.set_Cs(std::move(Cs), this->info.C_threshold); if (cal_dCV) - { - std::array>>, Ndim> - dCs_order = LRI_CV_Tools::change_order(std::move(dCs)); - this->exx_lri.set_dCs(std::move(dCs_order), this->info.C_grad_threshold); - if(PARAM.inp.cal_stress) - { - std::array>>,3>,3> dCRs = LRI_CV_Tools::cal_dMRs(ucell,dCs_order); - this->exx_lri.set_dCRs(std::move(dCRs), this->info.C_grad_R_threshold); - } - } - ModuleBase::timer::end("Exx_LRI", "cal_exx_ions"); + { + std::array>>, Ndim> + dCs_order = LRI_CV_Tools::change_order(std::move(dCs)); + this->exx_lri.set_dCs(std::move(dCs_order), this->info.C_grad_threshold); + if(PARAM.inp.cal_stress) + { + std::array>>,3>,3> dCRs = LRI_CV_Tools::cal_dMRs(ucell,dCs_order); + this->exx_lri.set_dCRs(std::move(dCRs), this->info.C_grad_R_threshold); + } + } + ModuleBase::timer::end("Exx_LRI", "cal_exx_ions"); } - #if 0 - template - void Exx_LRI::cal_cut_coulomb_cs( - std::map>>& Vs_cut, - std::map>>& Cs, - const UnitCell& ucell, - const bool write_cv) + #if 0 + template + void Exx_LRI::cal_cut_coulomb_cs( + std::map>>& Vs_cut, + std::map>>& Cs, + const UnitCell& ucell, + const bool write_cv) { - ModuleBase::TITLE("Exx_LRI", "cal_cut_coulomb_cs"); - ModuleBase::timer::start("Exx_LRI", "cal_cut_coulomb_cs"); - - std::vector atoms(ucell.nat); - for(int iat=0; iat atoms_pos; - for(int iat=0; iat latvec - = {RI_Util::Vector3_to_array3(ucell.a1), - RI_Util::Vector3_to_array3(ucell.a2), - RI_Util::Vector3_to_array3(ucell.a3)}; - const std::array period = {this->p_kv->nmp[0], this->p_kv->nmp[1], this->p_kv->nmp[2]}; - - this->exx_lri.set_parallel(this->mpi_comm, atoms_pos, latvec, period); - - // std::max(3) for gamma_only, list_A2 should contain cell {-1,0,1}. In the future distribute will be neighbour. - const std::array period_Vs - = LRI_CV_Tools::cal_latvec_range(1 + this->info.ccp_rmesh_times, ucell, orb_cutoff_); - const std::pair, std::vector>>>> - list_As_Vs = RI::Distribute_Equally::distribute_atoms_periods(this->mpi_comm, atoms, period_Vs, 2, false); - - std::map, Ndim>>> dVs; - for(const auto &settings_list : this->coulomb_settings) - { - if(!settings_list.second.first) continue; - std::map>> - Vs_temp = this->exx_objs[settings_list.first].cv.cal_Vs(ucell, - list_As_Vs.first, list_As_Vs.second[0], - {{"writable_Vws",true}}); - this->exx_objs[settings_list.first].cv.Vws = LRI_CV_Tools::get_CVws(ucell,Vs_temp); - - // ======rotate ABFs begin====== - int flag = 0; - for (const auto& IJRc: this->exx_objs[settings_list.first].cv.Vws) - { - const TA& I = IJRc.first; - const auto& JRc = IJRc.second; - for (const auto& JRc_tensor: JRc) - { - const TA& J = JRc_tensor.first; - const auto Rc = JRc_tensor.second; - for (const auto& Rc_tensor: Rc) - { - const auto& R = Rc_tensor.first; - flag += 1; - } - } - } - std::cout << "Coulomb: number of atom-pairs inside atomic overlap is " << flag << ". " << std::endl; - if (this->info.coul_moment == true) - { - double hf_Rcut = std::pow(0.75 * this->p_kv->get_nkstot_nospin() * ucell.omega / (ModuleBase::PI), 1.0 / 3.0); - // To cal Cs, we still cal all Vs(R) in r space - // moment_abfs->cal_VR(ucell, - // this->abfs, - // list_As_Vs, - // orb_cutoff_, - // hf_Rcut, - // this->exx_objs[settings_list.first].cv, - // Vs_cut); - delete moment_abfs; - moment_abfs = nullptr; - malloc_trim(0); - } - - flag = 0; - for (const auto& IJRc: this->exx_objs[settings_list.first].cv.Vws) - { - const auto& JRc = IJRc.second; - for (const auto& JRc_tensor: JRc) - { - const auto Rc = JRc_tensor.second; - for (const auto& Rc_tensor: Rc) - { - flag += 1; - } - } - } - std::cout << "Coulomb: number of all atom-pairs is " << flag << ". " << std::endl; - // ======rotate ABFs end====== - - Vs_cut = Vs.empty() ? Vs_temp : LRI_CV_Tools::add(Vs_cut, Vs_temp); - - if(PARAM.inp.cal_force || PARAM.inp.cal_stress) - { - std::map, Ndim>>> - dVs_temp = this->exx_objs[settings_list.first].cv.cal_dVs(ucell, - list_As_Vs.first, list_As_Vs.second[0], - {{"writable_dVws",true}}); - this->exx_objs[settings_list.first].cv.dVws = LRI_CV_Tools::get_dCVws(ucell,dVs_temp); - dVs = dVs.empty() ? dVs_temp : LRI_CV_Tools::add(dVs, dVs_temp); - } - } - - if (write_cv && GlobalV::MY_RANK == 0) - { - LRI_CV_Tools::write_Vs_abf(Vs_cut, PARAM.globalv.global_out_dir + "Vs_cut"); - } - this->exx_lri.set_Vs(std::move(Vs_cut), this->info.V_threshold); - - if(PARAM.inp.cal_force || PARAM.inp.cal_stress) - { - std::map,Ndim>>> dVs - = this->exx_objs[coulomb_method].cv.cal_dVs(ucell, - list_As_Vs.first, list_As_Vs.second[0], - {{"writable_dVws",true}}); - this->exx_objs[coulomb_method].cv.dVws = LRI_CV_Tools::get_dCVws(ucell,dVs); - - std::array>>,Ndim> dVs_order - = LRI_CV_Tools::change_order(std::move(dVs)); - this->exx_lri.set_dVs(std::move(dVs_order), this->info.V_grad_threshold); - if(PARAM.inp.cal_stress) - { - std::array>>,3>,3> dVRs - = LRI_CV_Tools::cal_dMRs(ucell,dVs_order); - this->exx_lri.set_dVRs(std::move(dVRs), this->info.V_grad_R_threshold); - } - } - - const std::array period_Cs = LRI_CV_Tools::cal_latvec_range(2, ucell,orb_cutoff_); - const std::pair, std::vector>>>> - list_As_Cs = RI::Distribute_Equally::distribute_atoms_periods(this->mpi_comm, atoms, period_Cs, 2, false); - std::pair>>, - std::map,3>>>> - Cs_dCs = this->exx_objs[coulomb_method].cv.cal_Cs_dCs(ucell, - list_As_Cs.first, list_As_Cs.second[0], - {{"cal_dC",PARAM.inp.cal_force||PARAM.inp.cal_stress}, - {"writable_Cws",true}, - {"writable_dCws",true}, - {"writable_Vws",false}, - {"writable_dVws",false}}); - Cs = std::get<0>(Cs_dCs); - this->exx_objs[coulomb_method].cv.Cws = LRI_CV_Tools::get_CVws(ucell,Cs); - if(write_cv && GlobalV::MY_RANK==0) - { - LRI_CV_Tools::write_Cs_ao(Cs, PARAM.globalv.global_out_dir + "Cs"); - } - this->exx_lri.set_Cs(Cs, this->info.C_threshold); - - if(PARAM.inp.cal_force || PARAM.inp.cal_stress) - { - std::map,3>>>& dCs = std::get<1>(Cs_dCs); - this->exx_objs[coulomb_method].cv.dCws = LRI_CV_Tools::get_dCVws(ucell,dCs); - std::array>>,Ndim> dCs_order - = LRI_CV_Tools::change_order(std::move(dCs)); - this->exx_lri.set_dCs(std::move(dCs_order), this->info.C_grad_threshold); - if(PARAM.inp.cal_stress) - { - std::array>>,3>,3> dCRs - = LRI_CV_Tools::cal_dMRs(ucell,dCs_order); - this->exx_lri.set_dCRs(std::move(dCRs), this->info.C_grad_R_threshold); - } - } - ModuleBase::timer::end("Exx_LRI", "cal_cut_coulomb_cs"); + ModuleBase::TITLE("Exx_LRI", "cal_cut_coulomb_cs"); + ModuleBase::timer::start("Exx_LRI", "cal_cut_coulomb_cs"); + + std::vector atoms(ucell.nat); + for(int iat=0; iat atoms_pos; + for(int iat=0; iat latvec + = {RI_Util::Vector3_to_array3(ucell.a1), + RI_Util::Vector3_to_array3(ucell.a2), + RI_Util::Vector3_to_array3(ucell.a3)}; + const std::array period = {this->p_kv->nmp[0], this->p_kv->nmp[1], this->p_kv->nmp[2]}; + + this->exx_lri.set_parallel(this->mpi_comm, atoms_pos, latvec, period); + + // std::max(3) for gamma_only, list_A2 should contain cell {-1,0,1}. In the future distribute will be neighbour. + const std::array period_Vs + = LRI_CV_Tools::cal_latvec_range(1 + this->info.ccp_rmesh_times, ucell, orb_cutoff_); + const std::pair, std::vector>>>> + list_As_Vs = RI::Distribute_Equally::distribute_atoms_periods(this->mpi_comm, atoms, period_Vs, 2, false); + + std::map, Ndim>>> dVs; + for(const auto &settings_list : this->coulomb_settings) + { + if(!settings_list.second.first) continue; + std::map>> + Vs_temp = this->exx_objs[settings_list.first].cv.cal_Vs(ucell, + list_As_Vs.first, list_As_Vs.second[0], + {{"writable_Vws",true}}); + this->exx_objs[settings_list.first].cv.Vws = LRI_CV_Tools::get_CVws(ucell,Vs_temp); + + // ======rotate ABFs begin====== + int flag = 0; + for (const auto& IJRc: this->exx_objs[settings_list.first].cv.Vws) + { + const TA& I = IJRc.first; + const auto& JRc = IJRc.second; + for (const auto& JRc_tensor: JRc) + { + const TA& J = JRc_tensor.first; + const auto Rc = JRc_tensor.second; + for (const auto& Rc_tensor: Rc) + { + const auto& R = Rc_tensor.first; + flag += 1; + } + } + } + std::cout << "Coulomb: number of atom-pairs inside atomic overlap is " << flag << ". " << std::endl; + if (this->info.coul_moment == true) + { + double hf_Rcut = std::pow(0.75 * this->p_kv->get_nkstot_nospin() * ucell.omega / (ModuleBase::PI), 1.0 / 3.0); + // To cal Cs, we still cal all Vs(R) in r space + // moment_abfs->cal_VR(ucell, + // this->abfs, + // list_As_Vs, + // orb_cutoff_, + // hf_Rcut, + // this->exx_objs[settings_list.first].cv, + // Vs_cut); + delete moment_abfs; + moment_abfs = nullptr; + malloc_trim(0); + } + + flag = 0; + for (const auto& IJRc: this->exx_objs[settings_list.first].cv.Vws) + { + const auto& JRc = IJRc.second; + for (const auto& JRc_tensor: JRc) + { + const auto Rc = JRc_tensor.second; + for (const auto& Rc_tensor: Rc) + { + flag += 1; + } + } + } + std::cout << "Coulomb: number of all atom-pairs is " << flag << ". " << std::endl; + // ======rotate ABFs end====== + + Vs_cut = Vs.empty() ? Vs_temp : LRI_CV_Tools::add(Vs_cut, Vs_temp); + + if(PARAM.inp.cal_force || PARAM.inp.cal_stress) + { + std::map, Ndim>>> + dVs_temp = this->exx_objs[settings_list.first].cv.cal_dVs(ucell, + list_As_Vs.first, list_As_Vs.second[0], + {{"writable_dVws",true}}); + this->exx_objs[settings_list.first].cv.dVws = LRI_CV_Tools::get_dCVws(ucell,dVs_temp); + dVs = dVs.empty() ? dVs_temp : LRI_CV_Tools::add(dVs, dVs_temp); + } + } + + if (write_cv && GlobalV::MY_RANK == 0) + { + LRI_CV_Tools::write_Vs_abf(Vs_cut, PARAM.globalv.global_out_dir + "Vs_cut"); + } + this->exx_lri.set_Vs(std::move(Vs_cut), this->info.V_threshold); + + if(PARAM.inp.cal_force || PARAM.inp.cal_stress) + { + std::map,Ndim>>> dVs + = this->exx_objs[coulomb_method].cv.cal_dVs(ucell, + list_As_Vs.first, list_As_Vs.second[0], + {{"writable_dVws",true}}); + this->exx_objs[coulomb_method].cv.dVws = LRI_CV_Tools::get_dCVws(ucell,dVs); + + std::array>>,Ndim> dVs_order + = LRI_CV_Tools::change_order(std::move(dVs)); + this->exx_lri.set_dVs(std::move(dVs_order), this->info.V_grad_threshold); + if(PARAM.inp.cal_stress) + { + std::array>>,3>,3> dVRs + = LRI_CV_Tools::cal_dMRs(ucell,dVs_order); + this->exx_lri.set_dVRs(std::move(dVRs), this->info.V_grad_R_threshold); + } + } + + const std::array period_Cs = LRI_CV_Tools::cal_latvec_range(2, ucell,orb_cutoff_); + const std::pair, std::vector>>>> + list_As_Cs = RI::Distribute_Equally::distribute_atoms_periods(this->mpi_comm, atoms, period_Cs, 2, false); + std::pair>>, + std::map,3>>>> + Cs_dCs = this->exx_objs[coulomb_method].cv.cal_Cs_dCs(ucell, + list_As_Cs.first, list_As_Cs.second[0], + {{"cal_dC",PARAM.inp.cal_force||PARAM.inp.cal_stress}, + {"writable_Cws",true}, + {"writable_dCws",true}, + {"writable_Vws",false}, + {"writable_dVws",false}}); + Cs = std::get<0>(Cs_dCs); + this->exx_objs[coulomb_method].cv.Cws = LRI_CV_Tools::get_CVws(ucell,Cs); + if(write_cv && GlobalV::MY_RANK==0) + { + LRI_CV_Tools::write_Cs_ao(Cs, PARAM.globalv.global_out_dir + "Cs"); + } + this->exx_lri.set_Cs(Cs, this->info.C_threshold); + + if(PARAM.inp.cal_force || PARAM.inp.cal_stress) + { + std::map,3>>>& dCs = std::get<1>(Cs_dCs); + this->exx_objs[coulomb_method].cv.dCws = LRI_CV_Tools::get_dCVws(ucell,dCs); + std::array>>,Ndim> dCs_order + = LRI_CV_Tools::change_order(std::move(dCs)); + this->exx_lri.set_dCs(std::move(dCs_order), this->info.C_grad_threshold); + if(PARAM.inp.cal_stress) + { + std::array>>,3>,3> dCRs + = LRI_CV_Tools::cal_dMRs(ucell,dCs_order); + this->exx_lri.set_dCRs(std::move(dCRs), this->info.C_grad_R_threshold); + } + } + ModuleBase::timer::end("Exx_LRI", "cal_cut_coulomb_cs"); } template void Exx_LRI::cal_ewald_coulomb( - std::map>>& Vs_full, - std::map>>& Cs, - const UnitCell& ucell, - const bool write_cv) + std::map>>& Vs_full, + std::map>>& Cs, + const UnitCell& ucell, + const bool write_cv) { - ModuleBase::TITLE("Exx_LRI", "cal_ewald_coulomb"); - ModuleBase::timer::start("Exx_LRI", "cal_ewald_coulomb"); - - std::vector atoms(ucell.nat); - for (int iat = 0; iat < ucell.nat; ++iat) - atoms[iat] = iat; - std::map atoms_pos; - for (int iat = 0; iat < ucell.nat; ++iat) - atoms_pos[iat] = RI_Util::Vector3_to_array3(ucell.atoms[ucell.iat2it[iat]].tau[ucell.iat2ia[iat]]); - const std::array latvec = { - RI_Util::Vector3_to_array3(ucell.a1), - RI_Util::Vector3_to_array3(ucell.a2), - RI_Util::Vector3_to_array3(ucell.a3)}; - const std::array period = {this->p_kv->nmp[0], this->p_kv->nmp[1], this->p_kv->nmp[2]}; - - this->exx_lri.set_parallel(this->mpi_comm, atoms_pos, latvec, period); - - // std::max(3) for gamma_only, list_A2 should contain cell {-1,0,1}. In the future distribute will be neighbour. - const std::array period_Vs - = LRI_CV_Tools::cal_latvec_range(1 + this->info.ccp_rmesh_times, ucell, orb_cutoff_); - const std::pair, std::vector>>>> list_As_Vs - = RI::Distribute_Equally::distribute_atoms_periods(this->mpi_comm, atoms, period_Vs, 2, false); - - std::map, Ndim>>> dVs; - for (const auto& settings_list: this->coulomb_settings) - { - if (!settings_list.second.first) - continue; - std::map>> Vs_temp - = this->exx_objs[settings_list.first].cv.cal_Vs( - ucell, - list_As_Vs.first, - list_As_Vs.second[0], - {{"writable_Vws", true}}); - this->exx_objs[settings_list.first].cv.Vws = LRI_CV_Tools::get_CVws(ucell, Vs_temp); - - // ======rotate ABFs begin====== - int flag = 0; - for (const auto& IJRc: this->exx_objs[settings_list.first].cv.Vws) - { - const TA& I = IJRc.first; - const auto& JRc = IJRc.second; - for (const auto& JRc_tensor: JRc) - { - const TA& J = JRc_tensor.first; - const auto Rc = JRc_tensor.second; - for (const auto& Rc_tensor: Rc) - { - const auto& R = Rc_tensor.first; - flag += 1; - } - } - } - std::cout << "Coulomb: number of atom-pairs inside atomic overlap is " << flag << ". " << std::endl; - if (this->info.coul_moment == true) - { - double hf_Rcut = std::pow(0.75 * this->p_kv->get_nkstot_nospin() * ucell.omega / (ModuleBase::PI), 1.0 / 3.0); - // To cal Cs, we still cal all Vs(R) in r space - // moment_abfs->cal_VR(ucell, - // this->abfs, - // list_As_Vs, - // orb_cutoff_, - // hf_Rcut, - // this->exx_objs[settings_list.first].cv, - // Vs_full); - delete moment_abfs; - moment_abfs = nullptr; - malloc_trim(0); - } - - flag = 0; - for (const auto& IJRc: this->exx_objs[settings_list.first].cv.Vws) - { - const auto& JRc = IJRc.second; - for (const auto& JRc_tensor: JRc) - { - const auto Rc = JRc_tensor.second; - for (const auto& Rc_tensor: Rc) - { - flag += 1; - } - } - } - std::cout << "Coulomb: number of all atom-pairs is " << flag << ". " << std::endl; - // ======rotate ABFs end====== - - if (settings_list.first == Conv_Coulomb_Pot_K::Coulomb_Method::Ewald) - { - this->exx_objs[settings_list.first].evq.init_ions(ucell, period_Vs); - const auto& coulomb_param = settings_list.second.second; - std::map>> Vs_ewald; - for (const auto& param_list: coulomb_param) - { - std::map>> Vs_ewald_temp; - switch (param_list.first) - { - case Conv_Coulomb_Pot_K::Coulomb_Type::Fock: { - double chi - = this->exx_objs[settings_list.first].evq.get_singular_chi(ucell, param_list.second, 2.0); - Vs_ewald_temp = this->exx_objs[settings_list.first].evq.cal_Vs(ucell, chi, Vs_temp); - break; - } - default: { - throw std::invalid_argument(std::string(__FILE__) + " line " + std::to_string(__LINE__)); - } - } - // Vs_temp cannot be covered here - Vs_ewald = Vs_ewald.empty() ? Vs_ewald_temp : LRI_CV_Tools::add(Vs_ewald, Vs_ewald_temp); - } - Vs_temp = Vs_ewald; - } - - Vs_full = Vs.empty() ? Vs_temp : LRI_CV_Tools::add(Vs_full, Vs_temp); - } - - if (write_cv && GlobalV::MY_RANK == 0) - { - LRI_CV_Tools::write_Vs_abf(Vs_full, PARAM.globalv.global_out_dir + "Vs_full"); - } - // this->exx_lri.set_Vs(std::move(Vs_full), this->info.V_threshold); - - // const std::array period_Cs = LRI_CV_Tools::cal_latvec_range(2, ucell,orb_cutoff_); - // const std::pair, std::vector>>>> - // list_As_Cs = RI::Distribute_Equally::distribute_atoms_periods(this->mpi_comm, atoms, period_Cs, 2, false); - // std::pair>>, - // std::map,3>>>> - // Cs_dCs = this->exx_objs[coulomb_method].cv.cal_Cs_dCs(ucell, - // list_As_Cs.first, list_As_Cs.second[0], - // {{"cal_dC",PARAM.inp.cal_force||PARAM.inp.cal_stress}, - // {"writable_Cws",true}, - // {"writable_dCws",true}, - // {"writable_Vws",false}, - // {"writable_dVws",false}}); - // Cs = std::get<0>(Cs_dCs); - // this->exx_objs[coulomb_method].cv.Cws = LRI_CV_Tools::get_CVws(ucell,Cs); - // if(write_cv && GlobalV::MY_RANK==0) - // { - // LRI_CV_Tools::write_Cs_ao(Cs, PARAM.globalv.global_out_dir + "Cs"); - // } - // this->exx_lri.set_Cs(Cs, this->info.C_threshold); - ModuleBase::timer::end("Exx_LRI", "cal_ewald_coulomb"); + ModuleBase::TITLE("Exx_LRI", "cal_ewald_coulomb"); + ModuleBase::timer::start("Exx_LRI", "cal_ewald_coulomb"); + + std::vector atoms(ucell.nat); + for (int iat = 0; iat < ucell.nat; ++iat) + atoms[iat] = iat; + std::map atoms_pos; + for (int iat = 0; iat < ucell.nat; ++iat) + atoms_pos[iat] = RI_Util::Vector3_to_array3(ucell.atoms[ucell.iat2it[iat]].tau[ucell.iat2ia[iat]]); + const std::array latvec = { + RI_Util::Vector3_to_array3(ucell.a1), + RI_Util::Vector3_to_array3(ucell.a2), + RI_Util::Vector3_to_array3(ucell.a3)}; + const std::array period = {this->p_kv->nmp[0], this->p_kv->nmp[1], this->p_kv->nmp[2]}; + + this->exx_lri.set_parallel(this->mpi_comm, atoms_pos, latvec, period); + + // std::max(3) for gamma_only, list_A2 should contain cell {-1,0,1}. In the future distribute will be neighbour. + const std::array period_Vs + = LRI_CV_Tools::cal_latvec_range(1 + this->info.ccp_rmesh_times, ucell, orb_cutoff_); + const std::pair, std::vector>>>> list_As_Vs + = RI::Distribute_Equally::distribute_atoms_periods(this->mpi_comm, atoms, period_Vs, 2, false); + + std::map, Ndim>>> dVs; + for (const auto& settings_list: this->coulomb_settings) + { + if (!settings_list.second.first) + continue; + std::map>> Vs_temp + = this->exx_objs[settings_list.first].cv.cal_Vs( + ucell, + list_As_Vs.first, + list_As_Vs.second[0], + {{"writable_Vws", true}}); + this->exx_objs[settings_list.first].cv.Vws = LRI_CV_Tools::get_CVws(ucell, Vs_temp); + + // ======rotate ABFs begin====== + int flag = 0; + for (const auto& IJRc: this->exx_objs[settings_list.first].cv.Vws) + { + const TA& I = IJRc.first; + const auto& JRc = IJRc.second; + for (const auto& JRc_tensor: JRc) + { + const TA& J = JRc_tensor.first; + const auto Rc = JRc_tensor.second; + for (const auto& Rc_tensor: Rc) + { + const auto& R = Rc_tensor.first; + flag += 1; + } + } + } + std::cout << "Coulomb: number of atom-pairs inside atomic overlap is " << flag << ". " << std::endl; + if (this->info.coul_moment == true) + { + double hf_Rcut = std::pow(0.75 * this->p_kv->get_nkstot_nospin() * ucell.omega / (ModuleBase::PI), 1.0 / 3.0); + // To cal Cs, we still cal all Vs(R) in r space + // moment_abfs->cal_VR(ucell, + // this->abfs, + // list_As_Vs, + // orb_cutoff_, + // hf_Rcut, + // this->exx_objs[settings_list.first].cv, + // Vs_full); + delete moment_abfs; + moment_abfs = nullptr; + malloc_trim(0); + } + + flag = 0; + for (const auto& IJRc: this->exx_objs[settings_list.first].cv.Vws) + { + const auto& JRc = IJRc.second; + for (const auto& JRc_tensor: JRc) + { + const auto Rc = JRc_tensor.second; + for (const auto& Rc_tensor: Rc) + { + flag += 1; + } + } + } + std::cout << "Coulomb: number of all atom-pairs is " << flag << ". " << std::endl; + // ======rotate ABFs end====== + + if (settings_list.first == Conv_Coulomb_Pot_K::Coulomb_Method::Ewald) + { + this->exx_objs[settings_list.first].evq.init_ions(ucell, period_Vs); + const auto& coulomb_param = settings_list.second.second; + std::map>> Vs_ewald; + for (const auto& param_list: coulomb_param) + { + std::map>> Vs_ewald_temp; + switch (param_list.first) + { + case Conv_Coulomb_Pot_K::Coulomb_Type::Fock: { + double chi + = this->exx_objs[settings_list.first].evq.get_singular_chi(ucell, param_list.second, 2.0); + Vs_ewald_temp = this->exx_objs[settings_list.first].evq.cal_Vs(ucell, chi, Vs_temp); + break; + } + default: { + throw std::invalid_argument(std::string(__FILE__) + " line " + std::to_string(__LINE__)); + } + } + // Vs_temp cannot be covered here + Vs_ewald = Vs_ewald.empty() ? Vs_ewald_temp : LRI_CV_Tools::add(Vs_ewald, Vs_ewald_temp); + } + Vs_temp = Vs_ewald; + } + + Vs_full = Vs.empty() ? Vs_temp : LRI_CV_Tools::add(Vs_full, Vs_temp); + } + + if (write_cv && GlobalV::MY_RANK == 0) + { + LRI_CV_Tools::write_Vs_abf(Vs_full, PARAM.globalv.global_out_dir + "Vs_full"); + } + // this->exx_lri.set_Vs(std::move(Vs_full), this->info.V_threshold); + + // const std::array period_Cs = LRI_CV_Tools::cal_latvec_range(2, ucell,orb_cutoff_); + // const std::pair, std::vector>>>> + // list_As_Cs = RI::Distribute_Equally::distribute_atoms_periods(this->mpi_comm, atoms, period_Cs, 2, false); + // std::pair>>, + // std::map,3>>>> + // Cs_dCs = this->exx_objs[coulomb_method].cv.cal_Cs_dCs(ucell, + // list_As_Cs.first, list_As_Cs.second[0], + // {{"cal_dC",PARAM.inp.cal_force||PARAM.inp.cal_stress}, + // {"writable_Cws",true}, + // {"writable_dCws",true}, + // {"writable_Vws",false}, + // {"writable_dVws",false}}); + // Cs = std::get<0>(Cs_dCs); + // this->exx_objs[coulomb_method].cv.Cws = LRI_CV_Tools::get_CVws(ucell,Cs); + // if(write_cv && GlobalV::MY_RANK==0) + // { + // LRI_CV_Tools::write_Cs_ao(Cs, PARAM.globalv.global_out_dir + "Cs"); + // } + // this->exx_lri.set_Cs(Cs, this->info.C_threshold); + ModuleBase::timer::end("Exx_LRI", "cal_ewald_coulomb"); } - #endif + #endif template void Exx_LRI::cal_cut_coulomb_cs( - std::map>>& Vs_cut, - std::map>>& Cs, - const UnitCell& ucell, - const bool write_cv) + std::map>>& Vs_cut, + std::map>>& Cs, + const UnitCell& ucell, + const bool write_cv) { - ModuleBase::TITLE("Exx_LRI", "cal_cut_coulomb_cs"); - ModuleBase::timer::start("Exx_LRI", "cal_cut_coulomb_cs"); - - std::vector atoms(ucell.nat); - for (int iat = 0; iat < ucell.nat; ++iat) - { - atoms[iat] = iat; - } - std::map atoms_pos; - for (int iat = 0; iat < ucell.nat; ++iat) - { - atoms_pos[iat] = RI_Util::Vector3_to_array3(ucell.atoms[ucell.iat2it[iat]].tau[ucell.iat2ia[iat]]); - } - const std::array latvec = { - RI_Util::Vector3_to_array3(ucell.a1), - RI_Util::Vector3_to_array3(ucell.a2), - RI_Util::Vector3_to_array3(ucell.a3)}; - const std::array period = {this->p_kv->nmp[0], this->p_kv->nmp[1], this->p_kv->nmp[2]}; - - this->exx_lri.set_parallel(this->mpi_comm, atoms_pos, latvec, period); - - const auto center2_method = Conv_Coulomb_Pot_K::Coulomb_Method::Center2; - auto center2_obj_it = this->exx_objs.find(center2_method); - if (center2_obj_it == this->exx_objs.end()) - { - throw std::invalid_argument(std::string(__FILE__) + " line " + std::to_string(__LINE__)); - } - - const std::array period_Vs - = LRI_CV_Tools::cal_latvec_range(1 + this->info.ccp_rmesh_times, ucell, orb_cutoff_); - const std::pair, std::vector>>>> list_As_Vs - = RI::Distribute_Equally::distribute_atoms_periods(this->mpi_comm, atoms, period_Vs, 2, false); - - Vs_cut = center2_obj_it->second.cv.cal_Vs( - ucell, - list_As_Vs.first, - list_As_Vs.second[0], - {{"writable_Vws", true}}); - center2_obj_it->second.cv.Vws = LRI_CV_Tools::get_CVws(ucell, Vs_cut); - if (write_cv && GlobalV::MY_RANK == 0) - { - LRI_CV_Tools::write_Vs_abf(Vs_cut, PARAM.globalv.global_out_dir + "Vs_cut"); - } - this->exx_lri.set_Vs(Vs_cut, this->info.V_threshold); - - const std::array period_Cs = LRI_CV_Tools::cal_latvec_range(2, ucell, orb_cutoff_); - const std::pair, std::vector>>>> list_As_Cs - = RI::Distribute_Equally::distribute_atoms_periods(this->mpi_comm, atoms, period_Cs, 2, false); - std::pair>>, - std::map, 3>>>> - Cs_dCs = center2_obj_it->second.cv.cal_Cs_dCs( - ucell, - list_As_Cs.first, - list_As_Cs.second[0], - {{"cal_dC", false}, - {"writable_Cws", true}, - {"writable_dCws", true}, - {"writable_Vws", false}, - {"writable_dVws", false}}); - Cs = std::get<0>(Cs_dCs); - center2_obj_it->second.cv.Cws = LRI_CV_Tools::get_CVws(ucell, Cs); - if (write_cv && GlobalV::MY_RANK == 0) - { - LRI_CV_Tools::write_Cs_ao(Cs, PARAM.globalv.global_out_dir + "Cs"); - } - this->exx_lri.set_Cs(Cs, this->info.C_threshold); - - ModuleBase::timer::end("Exx_LRI", "cal_cut_coulomb_cs"); + ModuleBase::TITLE("Exx_LRI", "cal_cut_coulomb_cs"); + ModuleBase::timer::start("Exx_LRI", "cal_cut_coulomb_cs"); + + std::vector atoms(ucell.nat); + for (int iat = 0; iat < ucell.nat; ++iat) + { + atoms[iat] = iat; + } + std::map atoms_pos; + for (int iat = 0; iat < ucell.nat; ++iat) + { + atoms_pos[iat] = RI_Util::Vector3_to_array3(ucell.atoms[ucell.iat2it[iat]].tau[ucell.iat2ia[iat]]); + } + const std::array latvec = { + RI_Util::Vector3_to_array3(ucell.a1), + RI_Util::Vector3_to_array3(ucell.a2), + RI_Util::Vector3_to_array3(ucell.a3)}; + const std::array period = {this->p_kv->nmp[0], this->p_kv->nmp[1], this->p_kv->nmp[2]}; + + this->exx_lri.set_parallel(this->mpi_comm, atoms_pos, latvec, period); + + const auto center2_method = Conv_Coulomb_Pot_K::Coulomb_Method::Center2; + auto center2_obj_it = this->exx_objs.find(center2_method); + if (center2_obj_it == this->exx_objs.end()) + { + throw std::invalid_argument(std::string(__FILE__) + " line " + std::to_string(__LINE__)); + } + + const std::array period_Vs + = LRI_CV_Tools::cal_latvec_range(1 + this->info.ccp_rmesh_times, ucell, orb_cutoff_); + const std::pair, std::vector>>>> list_As_Vs + = RI::Distribute_Equally::distribute_atoms_periods(this->mpi_comm, atoms, period_Vs, 2, false); + + Vs_cut = center2_obj_it->second.cv.cal_Vs( + ucell, + list_As_Vs.first, + list_As_Vs.second[0], + {{"writable_Vws", true}}); + center2_obj_it->second.cv.Vws = LRI_CV_Tools::get_CVws(ucell, Vs_cut); + if (write_cv && GlobalV::MY_RANK == 0) + { + LRI_CV_Tools::write_Vs_abf(Vs_cut, PARAM.globalv.global_out_dir + "Vs_cut"); + } + this->exx_lri.set_Vs(Vs_cut, this->info.V_threshold); + + const std::array period_Cs = LRI_CV_Tools::cal_latvec_range(2, ucell, orb_cutoff_); + const std::pair, std::vector>>>> list_As_Cs + = RI::Distribute_Equally::distribute_atoms_periods(this->mpi_comm, atoms, period_Cs, 2, false); + std::pair>>, + std::map, 3>>>> + Cs_dCs = center2_obj_it->second.cv.cal_Cs_dCs( + ucell, + list_As_Cs.first, + list_As_Cs.second[0], + {{"cal_dC", false}, + {"writable_Cws", true}, + {"writable_dCws", true}, + {"writable_Vws", false}, + {"writable_dVws", false}}); + Cs = std::get<0>(Cs_dCs); + center2_obj_it->second.cv.Cws = LRI_CV_Tools::get_CVws(ucell, Cs); + if (write_cv && GlobalV::MY_RANK == 0) + { + LRI_CV_Tools::write_Cs_ao(Cs, PARAM.globalv.global_out_dir + "Cs"); + } + this->exx_lri.set_Cs(Cs, this->info.C_threshold); + + ModuleBase::timer::end("Exx_LRI", "cal_cut_coulomb_cs"); } template void Exx_LRI::cal_ewald_coulomb( - std::map>>& Vs_full, - std::map>>& Cs, - const UnitCell& ucell, - const bool write_cv) + std::map>>& Vs_full, + std::map>>& Cs, + const UnitCell& ucell, + const bool write_cv) { - ModuleBase::TITLE("Exx_LRI", "cal_ewald_coulomb"); - ModuleBase::timer::start("Exx_LRI", "cal_ewald_coulomb"); - - std::vector atoms(ucell.nat); - for (int iat = 0; iat < ucell.nat; ++iat) - { - atoms[iat] = iat; - } - std::map atoms_pos; - for (int iat = 0; iat < ucell.nat; ++iat) - { - atoms_pos[iat] = RI_Util::Vector3_to_array3(ucell.atoms[ucell.iat2it[iat]].tau[ucell.iat2ia[iat]]); - } - const std::array latvec = { - RI_Util::Vector3_to_array3(ucell.a1), - RI_Util::Vector3_to_array3(ucell.a2), - RI_Util::Vector3_to_array3(ucell.a3)}; - const std::array period = {this->p_kv->nmp[0], this->p_kv->nmp[1], this->p_kv->nmp[2]}; - - this->exx_lri.set_parallel(this->mpi_comm, atoms_pos, latvec, period); - - const std::array period_Vs - = LRI_CV_Tools::cal_latvec_range(1 + this->info.ccp_rmesh_times, ucell, orb_cutoff_); - const std::pair, std::vector>>>> list_As_Vs - = RI::Distribute_Equally::distribute_atoms_periods(this->mpi_comm, atoms, period_Vs, 2, false); - - for (const auto& settings_list : this->coulomb_settings) - { - std::map>> Vs_temp - = this->exx_objs[settings_list.first].cv.cal_Vs( - ucell, - list_As_Vs.first, - list_As_Vs.second[0], - {{"writable_Vws", true}}); - this->exx_objs[settings_list.first].cv.Vws = LRI_CV_Tools::get_CVws(ucell, Vs_temp); - - if (settings_list.first == Conv_Coulomb_Pot_K::Coulomb_Method::Ewald) - { - this->exx_objs[settings_list.first].evq.init_ions(ucell, period_Vs); - std::map>> Vs_ewald; - for (const auto& param_list : settings_list.second.second) - { - std::map>> Vs_ewald_temp; - switch (param_list.first) - { - case Conv_Coulomb_Pot_K::Coulomb_Type::Fock: - { - double chi = this->exx_objs[settings_list.first].evq.get_singular_chi(ucell, param_list.second, 2.0); - Vs_ewald_temp = this->exx_objs[settings_list.first].evq.cal_Vs(ucell, chi, Vs_temp); - break; - } - default: - { - throw std::invalid_argument(std::string(__FILE__) + " line " + std::to_string(__LINE__)); - } - } - Vs_ewald = Vs_ewald.empty() ? Vs_ewald_temp : LRI_CV_Tools::add(Vs_ewald, Vs_ewald_temp); - } - Vs_temp = Vs_ewald; - } - - Vs_full = Vs_full.empty() ? Vs_temp : LRI_CV_Tools::add(Vs_full, Vs_temp); - } - - if (write_cv && GlobalV::MY_RANK == 0) - { - LRI_CV_Tools::write_Vs_abf(Vs_full, PARAM.globalv.global_out_dir + "Vs_full"); - } - - Cs.clear(); - ModuleBase::timer::end("Exx_LRI", "cal_ewald_coulomb"); + ModuleBase::TITLE("Exx_LRI", "cal_ewald_coulomb"); + ModuleBase::timer::start("Exx_LRI", "cal_ewald_coulomb"); + + std::vector atoms(ucell.nat); + for (int iat = 0; iat < ucell.nat; ++iat) + { + atoms[iat] = iat; + } + std::map atoms_pos; + for (int iat = 0; iat < ucell.nat; ++iat) + { + atoms_pos[iat] = RI_Util::Vector3_to_array3(ucell.atoms[ucell.iat2it[iat]].tau[ucell.iat2ia[iat]]); + } + const std::array latvec = { + RI_Util::Vector3_to_array3(ucell.a1), + RI_Util::Vector3_to_array3(ucell.a2), + RI_Util::Vector3_to_array3(ucell.a3)}; + const std::array period = {this->p_kv->nmp[0], this->p_kv->nmp[1], this->p_kv->nmp[2]}; + + this->exx_lri.set_parallel(this->mpi_comm, atoms_pos, latvec, period); + + const std::array period_Vs + = LRI_CV_Tools::cal_latvec_range(1 + this->info.ccp_rmesh_times, ucell, orb_cutoff_); + const std::pair, std::vector>>>> list_As_Vs + = RI::Distribute_Equally::distribute_atoms_periods(this->mpi_comm, atoms, period_Vs, 2, false); + + for (const auto& settings_list : this->coulomb_settings) + { + std::map>> Vs_temp + = this->exx_objs[settings_list.first].cv.cal_Vs( + ucell, + list_As_Vs.first, + list_As_Vs.second[0], + {{"writable_Vws", true}}); + this->exx_objs[settings_list.first].cv.Vws = LRI_CV_Tools::get_CVws(ucell, Vs_temp); + + if (settings_list.first == Conv_Coulomb_Pot_K::Coulomb_Method::Ewald) + { + this->exx_objs[settings_list.first].evq.init_ions(ucell, period_Vs); + std::map>> Vs_ewald; + for (const auto& param_list : settings_list.second.second) + { + std::map>> Vs_ewald_temp; + switch (param_list.first) + { + case Conv_Coulomb_Pot_K::Coulomb_Type::Fock: + { + double chi = this->exx_objs[settings_list.first].evq.get_singular_chi(ucell, param_list.second, 2.0); + Vs_ewald_temp = this->exx_objs[settings_list.first].evq.cal_Vs(ucell, chi, Vs_temp); + break; + } + default: + { + throw std::invalid_argument(std::string(__FILE__) + " line " + std::to_string(__LINE__)); + } + } + Vs_ewald = Vs_ewald.empty() ? Vs_ewald_temp : LRI_CV_Tools::add(Vs_ewald, Vs_ewald_temp); + } + Vs_temp = Vs_ewald; + } + + Vs_full = Vs_full.empty() ? Vs_temp : LRI_CV_Tools::add(Vs_full, Vs_temp); + } + + if (write_cv && GlobalV::MY_RANK == 0) + { + LRI_CV_Tools::write_Vs_abf(Vs_full, PARAM.globalv.global_out_dir + "Vs_full"); + } + + Cs.clear(); + ModuleBase::timer::end("Exx_LRI", "cal_ewald_coulomb"); } template void Exx_LRI::cal_exx_elec(const std::vector>>>& Ds, - const UnitCell& ucell, - const Parallel_Orbitals& pv, - const ModuleSymmetry::Symmetry_rotation* p_symrot) + const UnitCell& ucell, + const Parallel_Orbitals& pv, + const ModuleSymmetry::Symmetry_rotation* p_symrot) { - ModuleBase::TITLE("Exx_LRI","cal_exx_elec"); - ModuleBase::timer::start("Exx_LRI", "cal_exx_elec"); - - const std::vector, std::set>> judge = RI_2D_Comm::get_2D_judge(ucell,pv); - - if(p_symrot) - { this->exx_lri.set_symmetry(true, p_symrot->get_irreducible_sector()); } - else - { this->exx_lri.set_symmetry(false, {}); } - - // (nspin=4) the spinor H(R) rotation mixes the 4 spin channels, so they cannot be rotated - // independently in the per-spin loop below; hand off to the SOC implementation. - if (p_symrot && PARAM.inp.nspin == 4) - { - this->cal_exx_elec_soc(Ds, ucell, judge, p_symrot); - this->exx_lri.set_symmetry(false, {}); - ModuleBase::timer::end("Exx_LRI", "cal_exx_elec"); - return; - } - - this->Hexxs.resize(PARAM.inp.nspin); - this->Eexx = 0; - for(int is=0; isexx_lri.set_Ds(Ds[is], this->info.dm_threshold, suffix); - this->exx_lri.cal_Hs({ "","",suffix }); - - if (!p_symrot) - { - this->Hexxs[is] = RI::Communicate_Tensors_Map_Judge::comm_map2_first( - this->mpi_comm, std::move(this->exx_lri.Hs), std::get<0>(judge[is]), std::get<1>(judge[is])); - } - else - { - // reduce but not repeat - auto Hs_a2D = this->exx_lri.post_2D.set_tensors_map2(this->exx_lri.Hs); - // rotate locally without repeat - Hs_a2D = p_symrot->restore_HR(ucell.symm, ucell.atoms, ucell.st, 'H', Hs_a2D); - // cal energy using full Hs without repeat - this->exx_lri.energy = this->exx_lri.post_2D.cal_energy( - this->exx_lri.post_2D.saves["Ds_" + suffix], - this->exx_lri.post_2D.set_tensors_map2(Hs_a2D)); - // get repeated full Hs for abacus - this->Hexxs[is] = RI::Communicate_Tensors_Map_Judge::comm_map2_first( - this->mpi_comm, std::move(Hs_a2D), std::get<0>(judge[is]), std::get<1>(judge[is])); - } - this->Eexx += std::real(this->exx_lri.energy); - post_process_Hexx(this->Hexxs[is]); - } - this->Eexx = post_process_Eexx(this->Eexx); - this->exx_lri.set_symmetry(false, {}); - ModuleBase::timer::end("Exx_LRI", "cal_exx_elec"); + ModuleBase::TITLE("Exx_LRI","cal_exx_elec"); + ModuleBase::timer::start("Exx_LRI", "cal_exx_elec"); + + const std::vector, std::set>> judge = RI_2D_Comm::get_2D_judge(ucell,pv); + + if(p_symrot) + { this->exx_lri.set_symmetry(true, p_symrot->get_irreducible_sector()); } + else + { this->exx_lri.set_symmetry(false, {}); } + + // (nspin=4) the spinor H(R) rotation mixes the 4 spin channels, so they cannot be rotated + // independently in the per-spin loop below; hand off to the SOC implementation. + if (p_symrot && PARAM.inp.nspin == 4) + { + this->cal_exx_elec_soc(Ds, ucell, judge, p_symrot); + this->exx_lri.set_symmetry(false, {}); + ModuleBase::timer::end("Exx_LRI", "cal_exx_elec"); + return; + } + + this->Hexxs.resize(PARAM.inp.nspin); + this->Eexx = 0; + for(int is=0; isexx_lri.set_Ds(Ds[is], this->info.dm_threshold, suffix); + this->exx_lri.cal_Hs({ "","",suffix }); + + if (!p_symrot) + { + this->Hexxs[is] = RI::Communicate_Tensors_Map_Judge::comm_map2_first( + this->mpi_comm, std::move(this->exx_lri.Hs), std::get<0>(judge[is]), std::get<1>(judge[is])); + } + else + { + // reduce but not repeat + auto Hs_a2D = this->exx_lri.post_2D.set_tensors_map2(this->exx_lri.Hs); + // rotate locally without repeat + Hs_a2D = p_symrot->restore_HR(ucell.symm, ucell.atoms, ucell.st, 'H', Hs_a2D); + // cal energy using full Hs without repeat + this->exx_lri.energy = this->exx_lri.post_2D.cal_energy( + this->exx_lri.post_2D.saves["Ds_" + suffix], + this->exx_lri.post_2D.set_tensors_map2(Hs_a2D)); + // get repeated full Hs for abacus + this->Hexxs[is] = RI::Communicate_Tensors_Map_Judge::comm_map2_first( + this->mpi_comm, std::move(Hs_a2D), std::get<0>(judge[is]), std::get<1>(judge[is])); + } + this->Eexx += std::real(this->exx_lri.energy); + post_process_Hexx(this->Hexxs[is]); + } + this->Eexx = post_process_Eexx(this->Eexx); + this->exx_lri.set_symmetry(false, {}); + ModuleBase::timer::end("Exx_LRI", "cal_exx_elec"); } template void Exx_LRI::cal_exx_elec_soc( - const std::vector>>>& Ds, - const UnitCell& ucell, - const std::vector, std::set>>& judge, - const ModuleSymmetry::Symmetry_rotation* p_symrot) + const std::vector>>>& Ds, + const UnitCell& ucell, + const std::vector, std::set>>& judge, + const ModuleSymmetry::Symmetry_rotation* p_symrot) { - ModuleBase::TITLE("Exx_LRI", "cal_exx_elec_soc"); - this->Hexxs.resize(PARAM.inp.nspin); // nspin==4 - this->Eexx = 0; - - // pass 1: compute the irreducible-sector Hs of all 4 spin channels. - // distinct suffix per channel keeps all 4 "Ds_*" saves alive for the energy in pass 3. - std::array>>, 4> Hs_irr; - std::array suffix; - for (int is = 0; is < 4; ++is) - { - suffix[is] = std::to_string(is); - this->exx_lri.set_Ds(Ds[is], this->info.dm_threshold, suffix[is]); - this->exx_lri.cal_Hs({ "","",suffix[is] }); - Hs_irr[is] = this->exx_lri.post_2D.set_tensors_map2(this->exx_lri.Hs); - } - - // pass 2: spinor-coupled rotation of the 4 channels from the irreducible sector to the full BZ - std::array>>, 4> Hs_full = - p_symrot->restore_HR_nspin4(ucell.symm, ucell.atoms, ucell.st, 'H', Hs_irr); - - // pass 3: per-channel energy (full Hs, no repeat), then gather the repeated full Hs for abacus - for (int is = 0; is < 4; ++is) - { - this->exx_lri.energy = this->exx_lri.post_2D.cal_energy( - this->exx_lri.post_2D.saves["Ds_" + suffix[is]], - this->exx_lri.post_2D.set_tensors_map2(Hs_full[is])); - this->Hexxs[is] = RI::Communicate_Tensors_Map_Judge::comm_map2_first( - this->mpi_comm, std::move(Hs_full[is]), std::get<0>(judge[is]), std::get<1>(judge[is])); - this->Eexx += std::real(this->exx_lri.energy); - post_process_Hexx(this->Hexxs[is]); - } - this->Eexx = post_process_Eexx(this->Eexx); + ModuleBase::TITLE("Exx_LRI", "cal_exx_elec_soc"); + this->Hexxs.resize(PARAM.inp.nspin); // nspin==4 + this->Eexx = 0; + + // pass 1: compute the irreducible-sector Hs of all 4 spin channels. + // distinct suffix per channel keeps all 4 "Ds_*" saves alive for the energy in pass 3. + std::array>>, 4> Hs_irr; + std::array suffix; + for (int is = 0; is < 4; ++is) + { + suffix[is] = std::to_string(is); + this->exx_lri.set_Ds(Ds[is], this->info.dm_threshold, suffix[is]); + this->exx_lri.cal_Hs({ "","",suffix[is] }); + Hs_irr[is] = this->exx_lri.post_2D.set_tensors_map2(this->exx_lri.Hs); + } + + // pass 2: spinor-coupled rotation of the 4 channels from the irreducible sector to the full BZ + std::array>>, 4> Hs_full = + p_symrot->restore_HR_nspin4(ucell.symm, ucell.atoms, ucell.st, 'H', Hs_irr); + + // pass 3: per-channel energy (full Hs, no repeat), then gather the repeated full Hs for abacus + for (int is = 0; is < 4; ++is) + { + this->exx_lri.energy = this->exx_lri.post_2D.cal_energy( + this->exx_lri.post_2D.saves["Ds_" + suffix[is]], + this->exx_lri.post_2D.set_tensors_map2(Hs_full[is])); + this->Hexxs[is] = RI::Communicate_Tensors_Map_Judge::comm_map2_first( + this->mpi_comm, std::move(Hs_full[is]), std::get<0>(judge[is]), std::get<1>(judge[is])); + this->Eexx += std::real(this->exx_lri.energy); + post_process_Hexx(this->Hexxs[is]); + } + this->Eexx = post_process_Eexx(this->Eexx); } template void Exx_LRI::post_process_Hexx( std::map>> &Hexxs_io ) const { - ModuleBase::TITLE("Exx_LRI","post_process_Hexx"); - constexpr Tdata frac = -1 * 2; // why? Hartree to Ry? - const std::function&)> - multiply_frac = [&frac](RI::Tensor &t) - { t = t*frac; }; - RI::Map_Operator::for_each( Hexxs_io, multiply_frac ); + ModuleBase::TITLE("Exx_LRI","post_process_Hexx"); + constexpr Tdata frac = -1 * 2; // why? Hartree to Ry? + const std::function&)> + multiply_frac = [&frac](RI::Tensor &t) + { t = t*frac; }; + RI::Map_Operator::for_each( Hexxs_io, multiply_frac ); } template double Exx_LRI::post_process_Eexx(const double& Eexx_in) const { - ModuleBase::TITLE("Exx_LRI","post_process_Eexx"); - const double SPIN_multiple = std::map{ {1,2}, {2,1}, {4,1} }.at(PARAM.inp.nspin); // why? - const double frac = -SPIN_multiple; - return frac * Eexx_in; + ModuleBase::TITLE("Exx_LRI","post_process_Eexx"); + const double SPIN_multiple = std::map{ {1,2}, {2,1}, {4,1} }.at(PARAM.inp.nspin); // why? + const double frac = -SPIN_multiple; + return frac * Eexx_in; } /* post_process_old { - // D - const std::map SPIN_multiple = {{1,0.5}, {2,1}, {4,1}}; // ??? - DR *= SPIN_multiple.at(NSPIN); + // D + const std::map SPIN_multiple = {{1,0.5}, {2,1}, {4,1}}; // ??? + DR *= SPIN_multiple.at(NSPIN); - // H - HR *= -2; + // H + HR *= -2; - // E - const std::map SPIN_multiple = {{1,2}, {2,1}, {4,1}}; // ??? - energy *= SPIN_multiple.at(PARAM.inp.nspin); // ? - energy /= 2; // /2 for Ry + // E + const std::map SPIN_multiple = {{1,2}, {2,1}, {4,1}}; // ??? + energy *= SPIN_multiple.at(PARAM.inp.nspin); // ? + energy /= 2; // /2 for Ry } */ template void Exx_LRI::cal_exx_force(const int& nat) { - ModuleBase::TITLE("Exx_LRI","cal_exx_force"); - ModuleBase::timer::start("Exx_LRI", "cal_exx_force"); - - this->force_exx.create(nat, Ndim); - for(int is=0; isexx_lri.cal_force({"","",std::to_string(is),"",""}); - for(std::size_t idim=0; idimexx_lri.force[idim]) { - this->force_exx(force_item.first, idim) += std::real(force_item.second); - } } - } - - const double SPIN_multiple = std::map{{1,2}, {2,1}, {4,1}}.at(PARAM.inp.nspin); // why? - const double frac = -2 * SPIN_multiple; // why? - this->force_exx *= frac; - ModuleBase::timer::end("Exx_LRI", "cal_exx_force"); -} + ModuleBase::TITLE("Exx_LRI","cal_exx_force"); + ModuleBase::timer::start("Exx_LRI", "cal_exx_force"); + this->force_exx.create(nat, Ndim); + for(int is=0; isexx_lri.cal_force({"","",std::to_string(is),"",""}); + for(std::size_t idim=0; idimexx_lri.force[idim]) { + this->force_exx(force_item.first, idim) += std::real(force_item.second); + } } + } -template -void Exx_LRI::cal_exx_stress(const double& omega, const double& lat0) -{ - ModuleBase::TITLE("Exx_LRI","cal_exx_stress"); - ModuleBase::timer::start("Exx_LRI", "cal_exx_stress"); - - this->stress_exx.create(Ndim, Ndim); - for(int is=0; isexx_lri.cal_stress({"","",std::to_string(is),"",""}); - for(std::size_t idim0=0; idim0stress_exx(idim0,idim1) += std::real(this->exx_lri.stress(idim0,idim1)); - } } - } - - const double SPIN_multiple = std::map{{1,2}, {2,1}, {4,1}}.at(PARAM.inp.nspin); // why? - const double frac = 2 * SPIN_multiple / omega * lat0; // why? - this->stress_exx *= frac; - - ModuleBase::timer::end("Exx_LRI", "cal_exx_stress"); + const double SPIN_multiple = std::map{{1,2}, {2,1}, {4,1}}.at(PARAM.inp.nspin); // why? + const double frac = -2 * SPIN_multiple; // why? + this->force_exx *= frac; + ModuleBase::timer::end("Exx_LRI", "cal_exx_force"); } + template -void Exx_LRI::cal_exx_dHs(const std::vector>>>& Ds, - const UnitCell& ucell, - const Parallel_Orbitals& pv) +void Exx_LRI::cal_exx_stress(const double& omega, const double& lat0) { - ModuleBase::TITLE("Exx_LRI", "cal_exx_dHs"); - ModuleBase::timer::start("Exx_LRI", "cal_exx_dHs"); -#ifdef __EXX_DEV + ModuleBase::TITLE("Exx_LRI","cal_exx_stress"); + ModuleBase::timer::start("Exx_LRI", "cal_exx_stress"); - const std::vector, std::set>> judge = RI_2D_Comm::get_2D_judge(ucell, pv); - - const int nspin = PARAM.inp.nspin; - // dHexxs is indexed [direction(3)][atom][spin]; std::array has no resize(), so size the - // atom/spin dimensions explicitly before filling them below. - for (int ipos = 0; ipos < 3; ++ipos) + this->stress_exx.create(Ndim, Ndim); + for(int is=0; isdHexxs[ipos].resize(ucell.nat); - for (int iat = 0; iat < ucell.nat; ++iat) - { - this->dHexxs[ipos][iat].resize(nspin); - } + this->exx_lri.cal_stress({"","",std::to_string(is),"",""}); + for(std::size_t idim0=0; idim0stress_exx(idim0,idim1) += std::real(this->exx_lri.stress(idim0,idim1)); + } } } - for (int is = 0; is < nspin; ++is) - { - using namespace RI::Map_Operator; - const std::string suffix = std::to_string(is); // the same as cal_force/cal_stress case + const double SPIN_multiple = std::map{{1,2}, {2,1}, {4,1}}.at(PARAM.inp.nspin); // why? + const double frac = 2 * SPIN_multiple / omega * lat0; // why? + this->stress_exx *= frac; - this->exx_lri.set_Ds(Ds[is], this->info.dm_threshold, suffix); - this->exx_lri.cal_dHs({ "","",suffix,"","" }); // get lri-distributed exx_lri.dHs (all 3 directions) - // postprocess exx_lri.dHs[x/y/z] - for (int ipos = 0; ipos < 3; ++ipos) - { - // 1. Pulay terms: regroup by the differentiated atom: relative row/col [0]/[1] to absolute [0, nat-1] - std::vector>>> dHs_ipos_ispin(ucell.nat); - // dHs[ipos][0]: derivative w.r.t. the first (row) atom -> group by the row atom - for (const auto& item : this->exx_lri.dHs[ipos][0]) - { - const TA iat = item.first; - dHs_ipos_ispin[iat] = dHs_ipos_ispin[iat] - + std::map>>{ item }; - } - // dHs[ipos][1]: derivative w.r.t. the second (col) atom -> group each (col,R) entry by its col atom - for (const auto& row_item : this->exx_lri.dHs[ipos][1]) - { - const TA iat_row = row_item.first; - for (const auto& col_item : row_item.second) - { - const TA iat_col = col_item.first.first; // col atom (TA) of the TAC key - // Accumulate, do NOT drop on key collision: - // when iat_col==iat_row==I, directly insert would make col_item lost. - const auto ins = dHs_ipos_ispin[iat_col][iat_row].insert(col_item); - if (!ins.second) - ins.first->second = ins.first->second + col_item.second; - } - } - // 2. add Hellmann-Feynman terms and convert to 2D-distribution for abacus - for (int iat = 0; iat < ucell.nat; ++iat) - { - dHs_ipos_ispin[iat] = dHs_ipos_ispin[iat] + this->exx_lri.dHs_HF[ipos][iat]; - dHs_ipos_ispin[iat] = RI::Communicate_Tensors_Map_Judge::comm_map2_first( - this->mpi_comm, std::move(dHs_ipos_ispin[iat]), std::get<0>(judge[is]), std::get<1>(judge[is])); - this->dHexxs[ipos][iat][is] = dHs_ipos_ispin[iat]; - // Reuse the same post-processing as the Hexx writer (general nspin=1,2,4 factor) - // -1 factor compared to cal_force (F=-dE/dτ) is already included in LibRI's cal_dHs function. - this->post_process_Hexx(this->dHexxs[ipos][iat][is]); - } - } - } - ModuleBase::timer::end("Exx_LRI", "cal_exx_dHs"); -#else - ModuleBase::WARNING_QUIT("cal_exx_dHs","Compile with the developing version of LibRI and -DEXX_DEV flag to calculate dHexx."); -#endif + ModuleBase::timer::end("Exx_LRI", "cal_exx_stress"); } @@ -1043,15 +968,15 @@ void Exx_LRI::cal_exx_dHs(const std::vector std::vector> Exx_LRI::get_abfs_nchis() const { - std::vector> abfs_nchis; - for (const auto& abfs_T : this->abfs) - { - std::vector abfs_nchi_T; - for (const auto& abfs_L : abfs_T) - { abfs_nchi_T.push_back(abfs_L.size()); } - abfs_nchis.push_back(abfs_nchi_T); - } - return abfs_nchis; + std::vector> abfs_nchis; + for (const auto& abfs_T : this->abfs) + { + std::vector abfs_nchi_T; + for (const auto& abfs_L : abfs_T) + { abfs_nchi_T.push_back(abfs_L.size()); } + abfs_nchis.push_back(abfs_nchi_T); + } + return abfs_nchis; } */ diff --git a/source/source_lcao/module_ri/exx_lri_detail.h b/source/source_lcao/module_ri/exx_lri_detail.h index e72e3852dc3..5d1da32ffe2 100644 --- a/source/source_lcao/module_ri/exx_lri_detail.h +++ b/source/source_lcao/module_ri/exx_lri_detail.h @@ -18,8 +18,8 @@ #include #endif - class UnitCell; - class K_Vectors; + class UnitCell; + class K_Vectors; namespace ExxLriDetail { @@ -29,7 +29,7 @@ using CoulombParam inline void trim_malloc_cache() { #if defined(__GLIBC__) - malloc_trim(0); + malloc_trim(0); #endif } diff --git a/source/source_lcao/module_ri/exx_lri_interface.h b/source/source_lcao/module_ri/exx_lri_interface.h index 75df9fd9761..888f5ad91a6 100644 --- a/source/source_lcao/module_ri/exx_lri_interface.h +++ b/source/source_lcao/module_ri/exx_lri_interface.h @@ -2,7 +2,7 @@ #define EXX_LRI_INTERFACE_H #include "exx_lri.h" -#include "source_estate/module_charge/charge_mixing.h" +#include "source_estate/module_charge/chg_mix.h" #include "source_lcao/module_ri/mix_dmk_2d.h" #include "source_lcao/module_ri/module_exx_symmetry/symm_rotation.h" #include "source_estate/module_dm/density_matrix.h" // mohan add 2025-11-04 @@ -15,9 +15,14 @@ class Charge_Mixing; namespace elecstate { class ElecState; +} +namespace module_dm +{ template class DensityMatrix; - +} +namespace elecstate +{ /// for symmetry, multi-k, nspin<4: restore DM(k) form DM(k_ibz) std::vector>> restore_dm(const K_Vectors& kv, const std::vector>>& dm_k_ibz, @@ -56,7 +61,6 @@ class Exx_LRI_Interface double &get_Eexx() const { return this->exx_ptr->Eexx; } ModuleBase::matrix &get_force() const { return this->exx_ptr->force_exx; } ModuleBase::matrix &get_stress() const { return this->exx_ptr->stress_exx; } - auto& get_dHexxs() const { return this->exx_ptr->dHexxs; } int get_two_level_step() const { return this->two_level_step; @@ -84,14 +88,6 @@ class Exx_LRI_Interface /// @brief: in cal_exx_stress: Exx_LRI::cal_exx_stress() void cal_exx_stress(const double& omega, const double& lat0); - /// @brief: in cal_exx_dHs: Exx_LRI::cal_exx_dHs() - void cal_exx_dHs(const std::vector>>>& Ds, - const UnitCell& ucell, - const Parallel_Orbitals& pv); - - /// @brief build the exx-form dH (dHexxs) from the current mixed DM (for dH/dR output) - void cal_exx_dHs(const UnitCell& ucell, const Parallel_Orbitals& pv, const int nspin); - // Processes in ESolver_KS_LCAO /// @brief in before_all_runners: set symmetry according to irreducible k-points /// since k-points are not reduced again after the variation of the cell and exx-symmetry must be consistent with k-points. @@ -104,8 +100,9 @@ class Exx_LRI_Interface /// @brief in eachiterinit: do DM mixing and calculate Hexx when entering 2nd SCF void exx_eachiterinit(const int istep, const UnitCell& ucell, - const elecstate::DensityMatrix& dm/**< double should be Tdata if complex-PBE-DM is supported*/, + const module_dm::DensityMatrix& dm/**< double should be Tdata if complex-PBE-DM is supported*/, const K_Vectors& kv, + const Parallel_Orbitals& pv, const int& iter); /// @brief in hamilt2rho: calculate Hexx and Eexx @@ -115,8 +112,9 @@ class Exx_LRI_Interface void exx_iter_finish(const K_Vectors& kv, const UnitCell& ucell, hamilt::Hamilt& hamilt, - elecstate::ElecState& elec, - elecstate::DensityMatrix* dm, // mohan add 2025-11-04 + elecstate::ElecState& elec, + module_dm::DensityMatrix* dm, // mohan add 2025-11-04 + const Parallel_Orbitals& pv, Charge_Mixing& chgmix, const double& scf_ene_thr, int& iter, @@ -125,8 +123,9 @@ class Exx_LRI_Interface /// @brief: in do_after_converge: add exx operators; do DM mixing if seperate loop bool exx_after_converge(const UnitCell& ucell, hamilt::Hamilt& hamilt, - const elecstate::DensityMatrix& dm/**< double should be Tdata if complex-PBE-DM is supported*/, + const module_dm::DensityMatrix& dm/**< double should be Tdata if complex-PBE-DM is supported*/, const K_Vectors& kv, + const Parallel_Orbitals& pv, const int& nspin, int& iter, const int& istep, @@ -139,7 +138,7 @@ class Exx_LRI_Interface /// >0: not the first outer loop. contributeHk will do enerything normally. int two_level_step = 0; double etot_last_outer_loop = 0.0; - elecstate::DensityMatrix* dm_last_step; + module_dm::DensityMatrix* dm_last_step; size_t hybrid_step() const { return hybrid_step_; } void set_hybrid_step(size_t s) { hybrid_step_ = s; } @@ -169,7 +168,6 @@ class Exx_LRI_Interface bool elec = false; bool force = false; bool stress = false; - bool dHs = false; }; Flag_Finish flag_finish; }; diff --git a/source/source_lcao/module_ri/exx_lri_interface.hpp b/source/source_lcao/module_ri/exx_lri_interface.hpp index f00011ab833..c09592f5fca 100644 --- a/source/source_lcao/module_ri/exx_lri_interface.hpp +++ b/source/source_lcao/module_ri/exx_lri_interface.hpp @@ -5,7 +5,7 @@ #include "source_base/parallel_common.h" #include "source_estate/elecstate_lcao.h" #include "source_hamilt/module_xc/xc_functional.h" -#include "source_io/module_hs/write_hs_sparse.h" +#include "source_io/module_hs/hs_sparse_io.h" #include "source_base/module_out/csr_reader.h" #include "source_io/module_parameter/parameter.h" #include "source_io/module_restart/restart.h" @@ -78,35 +78,6 @@ void Exx_LRI_Interface::cal_exx_stress(const double& omega, const doub this->flag_finish.stress = true; } -template -void Exx_LRI_Interface::cal_exx_dHs(const std::vector>>>& Ds, - const UnitCell& ucell, - const Parallel_Orbitals& pv) -{ - ModuleBase::TITLE("Exx_LRI_Interface", "cal_exx_dHs"); - if (!this->flag_finish.init || !this->flag_finish.ions) - { - throw std::runtime_error("Exx init unfinished when " + std::string(__FILE__) + " line " + std::to_string(__LINE__)); - } - - this->exx_ptr->cal_exx_dHs(Ds, ucell, pv); - - this->flag_finish.dHs = true; -} - -template -void Exx_LRI_Interface::cal_exx_dHs(const UnitCell& ucell, - const Parallel_Orbitals& pv, - const int nspin) -{ - // build D(R) from the current mixed D(k) (mirrors the Ds construction in exx_iter_finish) - const std::vector>>> Ds - = PARAM.globalv.gamma_only_local - ? RI_2D_Comm::split_m2D_ktoR(ucell, *this->exx_ptr->p_kv, this->mix_DMk_2D.get_DMk_out(), pv, nspin) - : RI_2D_Comm::split_m2D_ktoR(ucell, *this->exx_ptr->p_kv, this->mix_DMk_2D.get_DMk_out(), pv, nspin, this->exx_spacegroup_symmetry); - this->cal_exx_dHs(Ds, ucell, pv); -} - template void Exx_LRI_Interface::exx_before_all_runners( const K_Vectors& kv, @@ -121,9 +92,9 @@ void Exx_LRI_Interface::exx_before_all_runners( const std::array& period = RI_Util::get_Born_vonKarmen_period(kv); this->symrot_.find_irreducible_sector( ucell.symm, ucell.atoms, ucell.st, - RI_Util::get_Born_von_Karmen_cells(period), period, ucell.lat); + RI_Util::get_Born_von_Karmen_cells(period), period, ucell.lat, PARAM.globalv.global_out_dir); this->symrot_.set_abfs_Lmax(Exx_Abfs::Construct_Orbs::get_Lmax(this->exx_ptr->abfs)); - this->symrot_.cal_Ms(kv, ucell, pv); + this->symrot_.cal_Ms(kv, ucell, pv, PARAM.inp.nspin); } } @@ -178,8 +149,9 @@ void Exx_LRI_Interface::exx_beforescf(const int istep, template void Exx_LRI_Interface::exx_eachiterinit(const int istep, const UnitCell& ucell, - const elecstate::DensityMatrix& dm, + const module_dm::DensityMatrix& dm, const K_Vectors& kv, + const Parallel_Orbitals& pv, const int& iter) { ModuleBase::TITLE("Exx_LRI_Interface","exx_eachiterinit"); @@ -211,24 +183,24 @@ void Exx_LRI_Interface::exx_eachiterinit(const int istep, this->mix_DMk_2D.set_mixing(this->p_chgmix_->get_mixing()); } - auto cal = [this, &ucell,&kv, &flag_restart](const elecstate::DensityMatrix& dm_in) + auto cal = [this, &ucell,&kv, &pv, &flag_restart](const module_dm::DensityMatrix& dm_in) { if (this->exx_spacegroup_symmetry) - { this->mix_DMk_2D.mix(symrot_.restore_dm(kv, dm_in.get_DMK_vector(), *dm_in.get_paraV_pointer()), flag_restart); } + { this->mix_DMk_2D.mix(symrot_.restore_dm(kv, dm_in.get_dmk_vec(), pv), flag_restart); } else - { this->mix_DMk_2D.mix(dm_in.get_DMK_vector(), flag_restart); } + { this->mix_DMk_2D.mix(dm_in.get_dmk_vec(), flag_restart); } const std::vector>>> Ds = RI_2D_Comm::split_m2D_ktoR( ucell, *this->exx_ptr->p_kv, this->mix_DMk_2D.get_DMk_out(), - *dm_in.get_paraV_pointer(), + pv, PARAM.inp.nspin, this->exx_spacegroup_symmetry); if(this->exx_spacegroup_symmetry && this->exx_ptr->info.exx_symmetry_realspace) - { this->cal_exx_elec(Ds, ucell,*dm_in.get_paraV_pointer(), &this->symrot_); } + { this->cal_exx_elec(Ds, ucell, pv, &this->symrot_); } else - { this->cal_exx_elec(Ds, ucell,*dm_in.get_paraV_pointer()); } + { this->cal_exx_elec(Ds, ucell, pv); } }; if(istep > 0 && flag_restart) @@ -273,15 +245,16 @@ void Exx_LRI_Interface::exx_hamilt2rho(elecstate::ElecState& elec, con template void Exx_LRI_Interface::exx_iter_finish(const K_Vectors& kv, - const UnitCell& ucell, - hamilt::Hamilt& hamilt, - elecstate::ElecState& elec, - elecstate::DensityMatrix* dm, // mohan add 2025-11-04 - Charge_Mixing& chgmix, - const double& scf_ene_thr, - int& iter, - const int istep, - bool& conv_esolver) + const UnitCell& ucell, + hamilt::Hamilt& hamilt, + elecstate::ElecState& elec, + module_dm::DensityMatrix* dm, // mohan add 2025-11-04 + const Parallel_Orbitals& pv, + Charge_Mixing& chgmix, + const double& scf_ene_thr, + int& iter, + const int istep, + bool& conv_esolver) { ModuleBase::TITLE("Exx_LRI_Interface","exx_iter_finish"); if (GlobalC::restart.info_save.save_H && (this->two_level_step > 0 || istep > 0) @@ -332,6 +305,7 @@ void Exx_LRI_Interface::exx_iter_finish(const K_Vectors& kv, hamilt, *dm, kv, + pv, PARAM.inp.nspin, iter, istep, @@ -359,8 +333,9 @@ template bool Exx_LRI_Interface::exx_after_converge( const UnitCell& ucell, hamilt::Hamilt& hamilt, - const elecstate::DensityMatrix& dm, + const module_dm::DensityMatrix& dm, const K_Vectors& kv, + const Parallel_Orbitals& pv, const int& nspin, int& iter, const int& istep, @@ -425,21 +400,21 @@ bool Exx_LRI_Interface::exx_after_converge( const bool flag_restart = (this->two_level_step == 0 && PARAM.inp.init_wfc != "file") ? true : false; if(this->exx_spacegroup_symmetry) - { this->mix_DMk_2D.mix(symrot_.restore_dm(kv, dm.get_DMK_vector(), *dm.get_paraV_pointer()), flag_restart); } + { this->mix_DMk_2D.mix(symrot_.restore_dm(kv, dm.get_dmk_vec(), pv), flag_restart); } else - { this->mix_DMk_2D.mix(dm.get_DMK_vector(), flag_restart); } + { this->mix_DMk_2D.mix(dm.get_dmk_vec(), flag_restart); } const std::vector>>> Ds = RI_2D_Comm::split_m2D_ktoR( ucell, *this->exx_ptr->p_kv, this->mix_DMk_2D.get_DMk_out(), - *dm.get_paraV_pointer(), + pv, nspin, this->exx_spacegroup_symmetry); if(this->exx_spacegroup_symmetry && this->exx_ptr->info.exx_symmetry_realspace) - { this->cal_exx_elec(Ds, ucell, *dm.get_paraV_pointer(), &this->symrot_); } + { this->cal_exx_elec(Ds, ucell, pv, &this->symrot_); } else - { this->cal_exx_elec(Ds, ucell, *dm.get_paraV_pointer()); } // restore DM but not Hexx + { this->cal_exx_elec(Ds, ucell, pv); } // restore DM but not Hexx iter = 0; this->two_level_step++; diff --git a/source/source_lcao/module_ri/exx_opt_orb.cpp b/source/source_lcao/module_ri/exx_opt_orb.cpp index 645650134b6..7d01e7b07c5 100644 --- a/source/source_lcao/module_ri/exx_opt_orb.cpp +++ b/source/source_lcao/module_ri/exx_opt_orb.cpp @@ -10,413 +10,414 @@ #include "source_lcao/module_ri/matrix_orbs21.h" #include "source_lcao/module_ri/matrix_orbs22.h" #include "source_lcao/module_ri/lri_cv_tools.h" +#include "source_io/module_parameter/parameter.h" #include void Exx_Opt_Orb::generate_matrix( - const Exx_Info_Opt_ABFs &info, - const K_Vectors &kv, - const UnitCell &ucell, - const LCAO_Orbitals &orb) const + const Exx_Info_Opt_ABFs &info, + const K_Vectors &kv, + const UnitCell &ucell, + const LCAO_Orbitals &orb) const { - ModuleBase::TITLE("Exx_Opt_Orb::generate_matrix"); + ModuleBase::TITLE("Exx_Opt_Orb::generate_matrix"); - auto judge_orbs_empty = [](const std::vector>> &orbs) -> bool - { - for(const auto &orb_t : orbs) { - for(const auto &orb_tl : orb_t) { - if(orb_tl.size()>0) { - return false; - }}} - return true; - }; + auto judge_orbs_empty = [](const std::vector>> &orbs) -> bool + { + for(const auto &orb_t : orbs) { + for(const auto &orb_tl : orb_t) { + if(orb_tl.size()>0) { + return false; + }}} + return true; + }; - std::vector>> - lcaos = Exx_Abfs::Construct_Orbs::change_orbs( orb, info.kmesh_times ); - Exx_Abfs::Construct_Orbs::filter_empty_orbs(lcaos); + std::vector>> + lcaos = Exx_Abfs::Construct_Orbs::change_orbs( orb, info.kmesh_times ); + Exx_Abfs::Construct_Orbs::filter_empty_orbs(lcaos); - std::vector>> - abfs = Exx_Abfs::Construct_Orbs::abfs_same_atom(ucell,orb, lcaos, info.kmesh_times, info.pca_threshold ); - if(!info.files_abfs.empty()) - { abfs = Exx_Abfs::IO::construct_abfs( abfs, orb, info.files_abfs, info.kmesh_times ); } - Exx_Abfs::Construct_Orbs::filter_empty_orbs(abfs); + std::vector>> + abfs = Exx_Abfs::Construct_Orbs::abfs_same_atom(ucell,orb, lcaos, info.kmesh_times, info.pca_threshold ); + if(!info.files_abfs.empty()) + { abfs = Exx_Abfs::IO::construct_abfs( abfs, orb, info.files_abfs, info.kmesh_times ); } + Exx_Abfs::Construct_Orbs::filter_empty_orbs(abfs); - std::vector< std::vector< std::vector< Numerical_Orbital_Lm>>> - jle = Exx_Abfs::Jle::init_jle(info, info.kmesh_times, ucell , orb); - if(!info.files_jles.empty()) - { jle = Exx_Abfs::IO::construct_abfs( jle, orb, info.files_jles, info.kmesh_times ); } - Exx_Abfs::Construct_Orbs::filter_empty_orbs(jle); + std::vector< std::vector< std::vector< Numerical_Orbital_Lm>>> + jle = Exx_Abfs::Jle::init_jle(info, info.kmesh_times, ucell , orb); + if(!info.files_jles.empty()) + { jle = Exx_Abfs::IO::construct_abfs( jle, orb, info.files_jles, info.kmesh_times ); } + Exx_Abfs::Construct_Orbs::filter_empty_orbs(jle); - const ModuleBase::Element_Basis_Index::Range range_lcaos = ModuleBase::Element_Basis_Index::construct_range( lcaos ); - const ModuleBase::Element_Basis_Index::IndexLNM index_lcaos = ModuleBase::Element_Basis_Index::construct_index( range_lcaos ); + const ModuleBase::Element_Basis_Index::Range range_lcaos = ModuleBase::Element_Basis_Index::construct_range( lcaos ); + const ModuleBase::Element_Basis_Index::IndexLNM index_lcaos = ModuleBase::Element_Basis_Index::construct_index( range_lcaos ); - const ModuleBase::Element_Basis_Index::Range range_abfs = ModuleBase::Element_Basis_Index::construct_range( abfs ); - const ModuleBase::Element_Basis_Index::IndexLNM index_abfs = ModuleBase::Element_Basis_Index::construct_index( range_abfs ); + const ModuleBase::Element_Basis_Index::Range range_abfs = ModuleBase::Element_Basis_Index::construct_range( abfs ); + const ModuleBase::Element_Basis_Index::IndexLNM index_abfs = ModuleBase::Element_Basis_Index::construct_index( range_abfs ); - const ModuleBase::Element_Basis_Index::Range range_jys = ModuleBase::Element_Basis_Index::construct_range( jle ); - const ModuleBase::Element_Basis_Index::IndexLNM index_jys = ModuleBase::Element_Basis_Index::construct_index( range_jys ); + const ModuleBase::Element_Basis_Index::Range range_jys = ModuleBase::Element_Basis_Index::construct_range( jle ); + const ModuleBase::Element_Basis_Index::IndexLNM index_jys = ModuleBase::Element_Basis_Index::construct_index( range_jys ); - Exx_Abfs::Construct_Orbs::print_orbs_size(ucell, abfs, GlobalV::ofs_running); - Exx_Abfs::Construct_Orbs::print_orbs_size(ucell, jle, GlobalV::ofs_running); + Exx_Abfs::Construct_Orbs::print_orbs_size(ucell, abfs, GlobalV::ofs_running); + Exx_Abfs::Construct_Orbs::print_orbs_size(ucell, jle, GlobalV::ofs_running); - const std::map>> radial_R = get_radial_R(ucell); + const std::map>> radial_R = get_radial_R(ucell); - // < lcaos lcaos | lcaos lcaos > - const auto ms_lcaoslcaos_lcaoslcaos = [&]() -> std::map>>>> - { - if(judge_orbs_empty(lcaos)) { return {}; } - Matrix_Orbs22 m_lcaoslcaos_lcaoslcaos; - m_lcaoslcaos_lcaoslcaos.init( lcaos, lcaos, lcaos, lcaos, ucell,orb, info.kmesh_times ); - #if TEST_EXX_RADIAL>=1 - m_lcaoslcaos_lcaoslcaos.init_radial_table(radial_R); - #else - m_lcaoslcaos_lcaoslcaos.init_radial_table(); - #endif - return m_lcaoslcaos_lcaoslcaos.cal_overlap_matrix_all(ucell,index_lcaos, index_lcaos, index_lcaos, index_lcaos); - }(); + // < lcaos lcaos | lcaos lcaos > + const auto ms_lcaoslcaos_lcaoslcaos = [&]() -> std::map>>>> + { + if(judge_orbs_empty(lcaos)) { return {}; } + Matrix_Orbs22 m_lcaoslcaos_lcaoslcaos; + m_lcaoslcaos_lcaoslcaos.init( lcaos, lcaos, lcaos, lcaos, ucell,orb, info.kmesh_times ); + #if TEST_EXX_RADIAL>=1 + m_lcaoslcaos_lcaoslcaos.init_radial_table(radial_R); + #else + m_lcaoslcaos_lcaoslcaos.init_radial_table(); + #endif + return m_lcaoslcaos_lcaoslcaos.cal_overlap_matrix_all(ucell,index_lcaos, index_lcaos, index_lcaos, index_lcaos); + }(); - // < lcaos lcaos | jys > - const auto ms_lcaoslcaos_jys = [&]() -> std::map>>>>> - { - if(judge_orbs_empty(lcaos)) { return {}; } - if(judge_orbs_empty(jle)) { return {}; } - Matrix_Orbs21 m_jyslcaos_lcaos; - m_jyslcaos_lcaos.init( jle, lcaos, lcaos, ucell , orb, info.kmesh_times ); - #if TEST_EXX_RADIAL>=1 - m_jyslcaos_lcaos.init_radial_table( radial_R); - #else - m_jyslcaos_lcaos.init_radial_table(); - #endif - return m_jyslcaos_lcaos.cal_overlap_matrix_all(ucell,index_jys, index_lcaos, index_lcaos ); - }(); + // < lcaos lcaos | jys > + const auto ms_lcaoslcaos_jys = [&]() -> std::map>>>>> + { + if(judge_orbs_empty(lcaos)) { return {}; } + if(judge_orbs_empty(jle)) { return {}; } + Matrix_Orbs21 m_jyslcaos_lcaos; + m_jyslcaos_lcaos.init( jle, lcaos, lcaos, ucell , orb, info.kmesh_times ); + #if TEST_EXX_RADIAL>=1 + m_jyslcaos_lcaos.init_radial_table( radial_R); + #else + m_jyslcaos_lcaos.init_radial_table(); + #endif + return m_jyslcaos_lcaos.cal_overlap_matrix_all(ucell,index_jys, index_lcaos, index_lcaos ); + }(); - // < jys | jys > - const auto ms_jys_jys = [&]() -> std::map>>>> - { - if(judge_orbs_empty(jle)) { return {}; } - Matrix_Orbs11 m_jys_jys; - m_jys_jys.init( jle, jle, ucell,orb, info.kmesh_times ); - #if TEST_EXX_RADIAL>=1 - m_jys_jys.init_radial_table(radial_R); - #else - m_jys_jys.init_radial_table(); - #endif - return m_jys_jys.cal_overlap_matrix_all(ucell,index_jys, index_jys ); - }(); + // < jys | jys > + const auto ms_jys_jys = [&]() -> std::map>>>> + { + if(judge_orbs_empty(jle)) { return {}; } + Matrix_Orbs11 m_jys_jys; + m_jys_jys.init( jle, jle, ucell,orb, info.kmesh_times ); + #if TEST_EXX_RADIAL>=1 + m_jys_jys.init_radial_table(radial_R); + #else + m_jys_jys.init_radial_table(); + #endif + return m_jys_jys.cal_overlap_matrix_all(ucell,index_jys, index_jys ); + }(); - // < abfs | abfs > - const auto ms_abfs_abfs = [&]() -> std::map>>>> - { - if(judge_orbs_empty(abfs)) { return {}; } - Matrix_Orbs11 m_abfs_abfs; - m_abfs_abfs.init( abfs, abfs, ucell, orb, info.kmesh_times ); - #if TEST_EXX_RADIAL>=1 - m_abfs_abfs.init_radial_table(radial_R); - #else - m_abfs_abfs.init_radial_table(); - #endif - return m_abfs_abfs.cal_overlap_matrix_all(ucell,index_abfs, index_abfs ); - }(); + // < abfs | abfs > + const auto ms_abfs_abfs = [&]() -> std::map>>>> + { + if(judge_orbs_empty(abfs)) { return {}; } + Matrix_Orbs11 m_abfs_abfs; + m_abfs_abfs.init( abfs, abfs, ucell, orb, info.kmesh_times ); + #if TEST_EXX_RADIAL>=1 + m_abfs_abfs.init_radial_table(radial_R); + #else + m_abfs_abfs.init_radial_table(); + #endif + return m_abfs_abfs.cal_overlap_matrix_all(ucell,index_abfs, index_abfs ); + }(); - // < lcaos lcaos | abfs > - const auto ms_lcaoslcaos_abfs = [&]() -> std::map>>>>> - { - if(judge_orbs_empty(lcaos)) { return {}; } - if(judge_orbs_empty(abfs)) { return {}; } - Matrix_Orbs21 m_abfslcaos_lcaos; - m_abfslcaos_lcaos.init( abfs, lcaos, lcaos, ucell , orb, info.kmesh_times ); - #if TEST_EXX_RADIAL>=1 - m_abfslcaos_lcaos.init_radial_table(radial_R); - #else - m_abfslcaos_lcaos.init_radial_table(); - #endif - return m_abfslcaos_lcaos.cal_overlap_matrix_all(ucell,index_abfs, index_lcaos, index_lcaos ); - }(); + // < lcaos lcaos | abfs > + const auto ms_lcaoslcaos_abfs = [&]() -> std::map>>>>> + { + if(judge_orbs_empty(lcaos)) { return {}; } + if(judge_orbs_empty(abfs)) { return {}; } + Matrix_Orbs21 m_abfslcaos_lcaos; + m_abfslcaos_lcaos.init( abfs, lcaos, lcaos, ucell , orb, info.kmesh_times ); + #if TEST_EXX_RADIAL>=1 + m_abfslcaos_lcaos.init_radial_table(radial_R); + #else + m_abfslcaos_lcaos.init_radial_table(); + #endif + return m_abfslcaos_lcaos.cal_overlap_matrix_all(ucell,index_abfs, index_lcaos, index_lcaos ); + }(); - // < jys | abfs > - const auto ms_jys_abfs = [&]() -> std::map>>>> - { - if(judge_orbs_empty(jle)) { return {}; } - if(judge_orbs_empty(abfs)) { return {}; } - Matrix_Orbs11 m_jys_abfs; - m_jys_abfs.init( jle, abfs, ucell,orb, info.kmesh_times ); - #if TEST_EXX_RADIAL>=1 - m_jys_abfs.init_radial_table(radial_R); - #else - m_jys_abfs.init_radial_table(); - #endif - return m_jys_abfs.cal_overlap_matrix_all(ucell,index_jys, index_abfs ); - }(); + // < jys | abfs > + const auto ms_jys_abfs = [&]() -> std::map>>>> + { + if(judge_orbs_empty(jle)) { return {}; } + if(judge_orbs_empty(abfs)) { return {}; } + Matrix_Orbs11 m_jys_abfs; + m_jys_abfs.init( jle, abfs, ucell,orb, info.kmesh_times ); + #if TEST_EXX_RADIAL>=1 + m_jys_abfs.init_radial_table(radial_R); + #else + m_jys_abfs.init_radial_table(); + #endif + return m_jys_abfs.cal_overlap_matrix_all(ucell,index_jys, index_abfs ); + }(); - for( size_t TA=0; TA!=ucell.ntype; ++TA ) - { - for( size_t IA=0; IA!=ucell.atoms[TA].na; ++IA ) - { - for( size_t TB=0; TB!=ucell.ntype; ++TB ) - { - for( size_t IB=0; IB!=ucell.atoms[TB].na; ++IB ) - { - if( TA==TB && IA==IB ) - { - const size_t T=TA, I=IA; - if(!judge_orbs_empty(abfs)) - { - // < abfs | abfs >.I - const std::vector>> ms_abfs_abfs_I = cal_I( ms_abfs_abfs, T,I,T,I ); - // < lcaos lcaos | lcaos lcaos > - < lcaos lcaos | abfs > * < abfs | abfs >.I * < abfs | lcaos lcaos > - const RI::Tensor m_lcaoslcaos_lcaoslcaos_proj = - ms_lcaoslcaos_lcaoslcaos.at(T).at(I).at(T).at(I) - cal_mul_22( - ms_lcaoslcaos_abfs.at(T).at(I).at(T).at(I), - ms_abfs_abfs_I, - ms_lcaoslcaos_abfs.at(T).at(I).at(T).at(I)); - // < lcaos lcaos | jys > - < lcaos lcaos | abfs > * < abfs | abfs >.I * < abfs | jys > - const std::vector> m_lcaoslcaos_jys_proj = - {ms_lcaoslcaos_jys.at(T).at(I).at(T).at(I)[0] - cal_mul_21( - ms_lcaoslcaos_abfs.at(T).at(I).at(T).at(I), - ms_abfs_abfs_I, - {ms_jys_abfs.at(T).at(I).at(T).at(I)})}; - // < jys | jys > - < jys | abfs > * < abfs | abfs >.I * < abfs | jys > - const std::vector>> m_jys_jys_proj = - {{ms_jys_jys.at(T).at(I).at(T).at(I) - cal_mul_11( - {ms_jys_abfs.at(T).at(I).at(T).at(I)}, - ms_abfs_abfs_I, - {ms_jys_abfs.at(T).at(I).at(T).at(I)})}}; - print_matrix( - info, - ucell, - kv, - jle.at(T).size()-1, - {jle.at(T).at(0).size()}, - PARAM.globalv.global_out_dir+"/matrix-opt-abfs", - m_lcaoslcaos_jys_proj, - m_jys_jys_proj, - m_lcaoslcaos_lcaoslcaos_proj, - T, I, T, I, + for( size_t TA=0; TA!=ucell.ntype; ++TA ) + { + for( size_t IA=0; IA!=ucell.atoms[TA].na; ++IA ) + { + for( size_t TB=0; TB!=ucell.ntype; ++TB ) + { + for( size_t IB=0; IB!=ucell.atoms[TB].na; ++IB ) + { + if( TA==TB && IA==IB ) + { + const size_t T=TA, I=IA; + if(!judge_orbs_empty(abfs)) + { + // < abfs | abfs >.I + const std::vector>> ms_abfs_abfs_I = cal_I( ms_abfs_abfs, T,I,T,I ); + // < lcaos lcaos | lcaos lcaos > - < lcaos lcaos | abfs > * < abfs | abfs >.I * < abfs | lcaos lcaos > + const RI::Tensor m_lcaoslcaos_lcaoslcaos_proj = + ms_lcaoslcaos_lcaoslcaos.at(T).at(I).at(T).at(I) - cal_mul_22( + ms_lcaoslcaos_abfs.at(T).at(I).at(T).at(I), + ms_abfs_abfs_I, + ms_lcaoslcaos_abfs.at(T).at(I).at(T).at(I)); + // < lcaos lcaos | jys > - < lcaos lcaos | abfs > * < abfs | abfs >.I * < abfs | jys > + const std::vector> m_lcaoslcaos_jys_proj = + {ms_lcaoslcaos_jys.at(T).at(I).at(T).at(I)[0] - cal_mul_21( + ms_lcaoslcaos_abfs.at(T).at(I).at(T).at(I), + ms_abfs_abfs_I, + {ms_jys_abfs.at(T).at(I).at(T).at(I)})}; + // < jys | jys > - < jys | abfs > * < abfs | abfs >.I * < abfs | jys > + const std::vector>> m_jys_jys_proj = + {{ms_jys_jys.at(T).at(I).at(T).at(I) - cal_mul_11( + {ms_jys_abfs.at(T).at(I).at(T).at(I)}, + ms_abfs_abfs_I, + {ms_jys_abfs.at(T).at(I).at(T).at(I)})}}; + print_matrix( + info, + ucell, + kv, + jle.at(T).size()-1, + {jle.at(T).at(0).size()}, + PARAM.globalv.global_out_dir+"/matrix-opt-abfs", + m_lcaoslcaos_jys_proj, + m_jys_jys_proj, + m_lcaoslcaos_lcaoslcaos_proj, + T, I, T, I, orb.cutoffs(), - range_jys, index_jys ); - } - else - { - print_matrix( - info, - ucell, - kv, - jle.at(T).size()-1, - {jle.at(T).at(0).size()}, - PARAM.globalv.global_out_dir+"/matrix-opt-abfs", - ms_lcaoslcaos_jys.at(T).at(I).at(T).at(I), - {{ms_jys_jys.at(T).at(I).at(T).at(I)}}, - ms_lcaoslcaos_lcaoslcaos.at(T).at(I).at(T).at(I), - T, I, T, I, + range_jys, index_jys ); + } + else + { + print_matrix( + info, + ucell, + kv, + jle.at(T).size()-1, + {jle.at(T).at(0).size()}, + PARAM.globalv.global_out_dir+"/matrix-opt-abfs", + ms_lcaoslcaos_jys.at(T).at(I).at(T).at(I), + {{ms_jys_jys.at(T).at(I).at(T).at(I)}}, + ms_lcaoslcaos_lcaoslcaos.at(T).at(I).at(T).at(I), + T, I, T, I, orb.cutoffs(), - range_jys, index_jys ); - } - } - else - { - if(!judge_orbs_empty(abfs)) - { - // < abfs | abfs >.I - const std::vector>> ms_abfs_abfs_I = cal_I( ms_abfs_abfs, TA,IA,TB,IB ); - // < lcaos lcaos | lcaos lcaos > - < lcaos lcaos | abfs > * < abfs | abfs >.I * < abfs | lcaos lcaos > - const RI::Tensor m_lcaoslcaos_lcaoslcaos_proj = - ms_lcaoslcaos_lcaoslcaos.at(TA).at(IA).at(TB).at(IB) - cal_mul_22( - ms_lcaoslcaos_abfs.at(TA).at(IA).at(TB).at(IB), - ms_abfs_abfs_I, - ms_lcaoslcaos_abfs.at(TA).at(IA).at(TB).at(IB)); - // < lcaos lcaos | jys > - < lcaos lcaos | abfs > * < abfs | abfs >.I * < abfs | jys > - const std::vector> m_lcaoslcaos_jys_proj = - {ms_lcaoslcaos_jys.at(TA).at(IA).at(TB).at(IB)[0] - cal_mul_21( - ms_lcaoslcaos_abfs.at(TA).at(IA).at(TB).at(IB), - ms_abfs_abfs_I, - { ms_jys_abfs.at(TA).at(IA).at(TA).at(IA), ms_jys_abfs.at(TA).at(IA).at(TB).at(IB) }), - ms_lcaoslcaos_jys.at(TA).at(IA).at(TB).at(IB)[1] - cal_mul_21( - ms_lcaoslcaos_abfs.at(TA).at(IA).at(TB).at(IB), - ms_abfs_abfs_I, - { ms_jys_abfs.at(TB).at(IB).at(TA).at(IA), ms_jys_abfs.at(TB).at(IB).at(TB).at(IB) })}; - // < jys | jys > - < jys | abfs > * < abfs | abfs >.I * < abfs | jys > - const std::vector>> m_jys_jys_proj = - {{ms_jys_jys.at(TA).at(IA).at(TA).at(IA) - cal_mul_11( - { ms_jys_abfs.at(TA).at(IA).at(TA).at(IA), ms_jys_abfs.at(TA).at(IA).at(TB).at(IB) }, - ms_abfs_abfs_I, - { ms_jys_abfs.at(TA).at(IA).at(TA).at(IA), ms_jys_abfs.at(TA).at(IA).at(TB).at(IB) }), - ms_jys_jys.at(TA).at(IA).at(TB).at(IB) - cal_mul_11( - { ms_jys_abfs.at(TA).at(IA).at(TA).at(IA), ms_jys_abfs.at(TA).at(IA).at(TB).at(IB) }, - ms_abfs_abfs_I, - { ms_jys_abfs.at(TB).at(IB).at(TA).at(IA), ms_jys_abfs.at(TB).at(IB).at(TB).at(IB) }) }, - {ms_jys_jys.at(TB).at(IB).at(TA).at(IA) - cal_mul_11( - { ms_jys_abfs.at(TB).at(IB).at(TA).at(IA), ms_jys_abfs.at(TB).at(IB).at(TB).at(IB) }, - ms_abfs_abfs_I, - { ms_jys_abfs.at(TA).at(IA).at(TA).at(IA), ms_jys_abfs.at(TA).at(IA).at(TB).at(IB) }), - ms_jys_jys.at(TB).at(IB).at(TB).at(IB) - cal_mul_11( - { ms_jys_abfs.at(TB).at(IB).at(TA).at(IA), ms_jys_abfs.at(TB).at(IB).at(TB).at(IB) }, - ms_abfs_abfs_I, - { ms_jys_abfs.at(TB).at(IB).at(TA).at(IA), ms_jys_abfs.at(TB).at(IB).at(TB).at(IB) }) }}; - print_matrix( - info, - ucell, - kv, - std::max(jle.at(TA).size(), jle.at(TB).size())-1, - {jle.at(TA).at(0).size(), jle.at(TB).at(0).size()}, - PARAM.globalv.global_out_dir+"/matrix-opt-abfs", - m_lcaoslcaos_jys_proj, - m_jys_jys_proj, - m_lcaoslcaos_lcaoslcaos_proj, - TA, IA, TB, IB, + range_jys, index_jys ); + } + } + else + { + if(!judge_orbs_empty(abfs)) + { + // < abfs | abfs >.I + const std::vector>> ms_abfs_abfs_I = cal_I( ms_abfs_abfs, TA,IA,TB,IB ); + // < lcaos lcaos | lcaos lcaos > - < lcaos lcaos | abfs > * < abfs | abfs >.I * < abfs | lcaos lcaos > + const RI::Tensor m_lcaoslcaos_lcaoslcaos_proj = + ms_lcaoslcaos_lcaoslcaos.at(TA).at(IA).at(TB).at(IB) - cal_mul_22( + ms_lcaoslcaos_abfs.at(TA).at(IA).at(TB).at(IB), + ms_abfs_abfs_I, + ms_lcaoslcaos_abfs.at(TA).at(IA).at(TB).at(IB)); + // < lcaos lcaos | jys > - < lcaos lcaos | abfs > * < abfs | abfs >.I * < abfs | jys > + const std::vector> m_lcaoslcaos_jys_proj = + {ms_lcaoslcaos_jys.at(TA).at(IA).at(TB).at(IB)[0] - cal_mul_21( + ms_lcaoslcaos_abfs.at(TA).at(IA).at(TB).at(IB), + ms_abfs_abfs_I, + { ms_jys_abfs.at(TA).at(IA).at(TA).at(IA), ms_jys_abfs.at(TA).at(IA).at(TB).at(IB) }), + ms_lcaoslcaos_jys.at(TA).at(IA).at(TB).at(IB)[1] - cal_mul_21( + ms_lcaoslcaos_abfs.at(TA).at(IA).at(TB).at(IB), + ms_abfs_abfs_I, + { ms_jys_abfs.at(TB).at(IB).at(TA).at(IA), ms_jys_abfs.at(TB).at(IB).at(TB).at(IB) })}; + // < jys | jys > - < jys | abfs > * < abfs | abfs >.I * < abfs | jys > + const std::vector>> m_jys_jys_proj = + {{ms_jys_jys.at(TA).at(IA).at(TA).at(IA) - cal_mul_11( + { ms_jys_abfs.at(TA).at(IA).at(TA).at(IA), ms_jys_abfs.at(TA).at(IA).at(TB).at(IB) }, + ms_abfs_abfs_I, + { ms_jys_abfs.at(TA).at(IA).at(TA).at(IA), ms_jys_abfs.at(TA).at(IA).at(TB).at(IB) }), + ms_jys_jys.at(TA).at(IA).at(TB).at(IB) - cal_mul_11( + { ms_jys_abfs.at(TA).at(IA).at(TA).at(IA), ms_jys_abfs.at(TA).at(IA).at(TB).at(IB) }, + ms_abfs_abfs_I, + { ms_jys_abfs.at(TB).at(IB).at(TA).at(IA), ms_jys_abfs.at(TB).at(IB).at(TB).at(IB) }) }, + {ms_jys_jys.at(TB).at(IB).at(TA).at(IA) - cal_mul_11( + { ms_jys_abfs.at(TB).at(IB).at(TA).at(IA), ms_jys_abfs.at(TB).at(IB).at(TB).at(IB) }, + ms_abfs_abfs_I, + { ms_jys_abfs.at(TA).at(IA).at(TA).at(IA), ms_jys_abfs.at(TA).at(IA).at(TB).at(IB) }), + ms_jys_jys.at(TB).at(IB).at(TB).at(IB) - cal_mul_11( + { ms_jys_abfs.at(TB).at(IB).at(TA).at(IA), ms_jys_abfs.at(TB).at(IB).at(TB).at(IB) }, + ms_abfs_abfs_I, + { ms_jys_abfs.at(TB).at(IB).at(TA).at(IA), ms_jys_abfs.at(TB).at(IB).at(TB).at(IB) }) }}; + print_matrix( + info, + ucell, + kv, + std::max(jle.at(TA).size(), jle.at(TB).size())-1, + {jle.at(TA).at(0).size(), jle.at(TB).at(0).size()}, + PARAM.globalv.global_out_dir+"/matrix-opt-abfs", + m_lcaoslcaos_jys_proj, + m_jys_jys_proj, + m_lcaoslcaos_lcaoslcaos_proj, + TA, IA, TB, IB, orb.cutoffs(), - range_jys, index_jys ); - } - else - { - print_matrix( - info, - ucell, - kv, - std::max(jle.at(TA).size(), jle.at(TB).size())-1, - {jle.at(TA).at(0).size(), jle.at(TB).at(0).size()}, - PARAM.globalv.global_out_dir+"/matrix-opt-abfs", - ms_lcaoslcaos_jys.at(TA).at(IA).at(TB).at(IB), - {{ms_jys_jys.at(TA).at(IA).at(TA).at(IA), ms_jys_jys.at(TA).at(IA).at(TB).at(IB)}, - {ms_jys_jys.at(TB).at(IB).at(TA).at(IA), ms_jys_jys.at(TB).at(IB).at(TB).at(IB)}}, - ms_lcaoslcaos_lcaoslcaos.at(TA).at(IA).at(TB).at(IB), - TA, IA, TB, IB, + range_jys, index_jys ); + } + else + { + print_matrix( + info, + ucell, + kv, + std::max(jle.at(TA).size(), jle.at(TB).size())-1, + {jle.at(TA).at(0).size(), jle.at(TB).at(0).size()}, + PARAM.globalv.global_out_dir+"/matrix-opt-abfs", + ms_lcaoslcaos_jys.at(TA).at(IA).at(TB).at(IB), + {{ms_jys_jys.at(TA).at(IA).at(TA).at(IA), ms_jys_jys.at(TA).at(IA).at(TB).at(IB)}, + {ms_jys_jys.at(TB).at(IB).at(TA).at(IA), ms_jys_jys.at(TB).at(IB).at(TB).at(IB)}}, + ms_lcaoslcaos_lcaoslcaos.at(TA).at(IA).at(TB).at(IB), + TA, IA, TB, IB, orb.cutoffs(), - range_jys, index_jys ); - } - } - } - } - } - } + range_jys, index_jys ); + } + } + } + } + } + } } // m_left * m_middle * m_right.T RI::Tensor Exx_Opt_Orb::cal_mul_22( - const std::vector> & m_left, - const std::vector>> & m_middle, - const std::vector> & m_right ) const + const std::vector> & m_left, + const std::vector>> & m_middle, + const std::vector> & m_right ) const { - ModuleBase::TITLE("Exx_Opt_Orb::cal_mul_22"); - RI::Tensor m_mul; - for( size_t il=0; il!=m_left.size(); ++il ) - { - for( size_t ir=0; ir!=m_right.size(); ++ir ) - { - // m_mul += m_left[il] * m_middle[il][ir] * m_right[ir].T; - const RI::Tensor m_lm = RI::Tensor_Multiply::x0x1y1_x0x1a_ay1(m_left[il], m_middle[il][ir]); - const RI::Tensor m_lmr = RI::Tensor_Multiply::x0x1y0y1_x0x1a_y0y1a(m_lm, m_right[ir]); - if(m_mul.empty()) - { m_mul = std::move(m_lmr); } - else - { m_mul += m_lmr; } - } - } - return m_mul; + ModuleBase::TITLE("Exx_Opt_Orb::cal_mul_22"); + RI::Tensor m_mul; + for( size_t il=0; il!=m_left.size(); ++il ) + { + for( size_t ir=0; ir!=m_right.size(); ++ir ) + { + // m_mul += m_left[il] * m_middle[il][ir] * m_right[ir].T; + const RI::Tensor m_lm = RI::Tensor_Multiply::x0x1y1_x0x1a_ay1(m_left[il], m_middle[il][ir]); + const RI::Tensor m_lmr = RI::Tensor_Multiply::x0x1y0y1_x0x1a_y0y1a(m_lm, m_right[ir]); + if(m_mul.empty()) + { m_mul = std::move(m_lmr); } + else + { m_mul += m_lmr; } + } + } + return m_mul; } RI::Tensor Exx_Opt_Orb::cal_mul_21( - const std::vector> & m_left, - const std::vector>> & m_middle, - const std::vector> & m_right ) const + const std::vector> & m_left, + const std::vector>> & m_middle, + const std::vector> & m_right ) const { - ModuleBase::TITLE("Exx_Opt_Orb::cal_mul_21"); - RI::Tensor m_mul; - for( size_t il=0; il!=m_left.size(); ++il ) - { - for( size_t ir=0; ir!=m_right.size(); ++ir ) - { - // m_mul += m_left[il] * m_middle[il][ir] * m_right[ir].T; - const RI::Tensor m_lm = RI::Tensor_Multiply::x0x1y1_x0x1a_ay1(m_left[il], m_middle[il][ir]); - const RI::Tensor m_lmr = RI::Tensor_Multiply::x0x1y0_x0x1a_y0a(m_lm, m_right[ir]); - if(m_mul.empty()) - { m_mul = std::move(m_lmr); } - else - { m_mul += m_lmr; } - } - } - return m_mul; + ModuleBase::TITLE("Exx_Opt_Orb::cal_mul_21"); + RI::Tensor m_mul; + for( size_t il=0; il!=m_left.size(); ++il ) + { + for( size_t ir=0; ir!=m_right.size(); ++ir ) + { + // m_mul += m_left[il] * m_middle[il][ir] * m_right[ir].T; + const RI::Tensor m_lm = RI::Tensor_Multiply::x0x1y1_x0x1a_ay1(m_left[il], m_middle[il][ir]); + const RI::Tensor m_lmr = RI::Tensor_Multiply::x0x1y0_x0x1a_y0a(m_lm, m_right[ir]); + if(m_mul.empty()) + { m_mul = std::move(m_lmr); } + else + { m_mul += m_lmr; } + } + } + return m_mul; } RI::Tensor Exx_Opt_Orb::cal_mul_12( - const std::vector> & m_left, - const std::vector>> & m_middle, - const std::vector> & m_right ) const + const std::vector> & m_left, + const std::vector>> & m_middle, + const std::vector> & m_right ) const { - ModuleBase::TITLE("Exx_Opt_Orb::cal_mul_12"); - RI::Tensor m_mul; - for( size_t il=0; il!=m_left.size(); ++il ) - { - for( size_t ir=0; ir!=m_right.size(); ++ir ) - { - // m_mul += m_left[il] * m_middle[il][ir] * m_right[ir].T; - const RI::Tensor m_lm = RI::Tensor_Multiply::x0y1_x0a_ay1(m_left[il], m_middle[il][ir]); - const RI::Tensor m_lmr = RI::Tensor_Multiply::x0y0y1_x0a_y0y1a(m_lm, m_right[ir]); - if(m_mul.empty()) - { m_mul = std::move(m_lmr); } - else - { m_mul += m_lmr; } - } - } - return m_mul; + ModuleBase::TITLE("Exx_Opt_Orb::cal_mul_12"); + RI::Tensor m_mul; + for( size_t il=0; il!=m_left.size(); ++il ) + { + for( size_t ir=0; ir!=m_right.size(); ++ir ) + { + // m_mul += m_left[il] * m_middle[il][ir] * m_right[ir].T; + const RI::Tensor m_lm = RI::Tensor_Multiply::x0y1_x0a_ay1(m_left[il], m_middle[il][ir]); + const RI::Tensor m_lmr = RI::Tensor_Multiply::x0y0y1_x0a_y0y1a(m_lm, m_right[ir]); + if(m_mul.empty()) + { m_mul = std::move(m_lmr); } + else + { m_mul += m_lmr; } + } + } + return m_mul; } RI::Tensor Exx_Opt_Orb::cal_mul_11( - const std::vector> & m_left, - const std::vector>> & m_middle, - const std::vector> & m_right ) const + const std::vector> & m_left, + const std::vector>> & m_middle, + const std::vector> & m_right ) const { - ModuleBase::TITLE("Exx_Opt_Orb::cal_mul_11"); - RI::Tensor m_mul; - for( size_t il=0; il!=m_left.size(); ++il ) - { - for( size_t ir=0; ir!=m_right.size(); ++ir ) - { - // m_mul += m_left[il] * m_middle[il][ir] * m_right[ir].T; - const RI::Tensor m_lm = RI::Tensor_Multiply::x0y1_x0a_ay1(m_left[il], m_middle[il][ir]); - const RI::Tensor m_lmr = RI::Tensor_Multiply::x0y0_x0a_y0a(m_lm, m_right[ir]); - if(m_mul.empty()) - { m_mul = std::move(m_lmr); } - else - { m_mul += m_lmr; } - } - } - return m_mul; + ModuleBase::TITLE("Exx_Opt_Orb::cal_mul_11"); + RI::Tensor m_mul; + for( size_t il=0; il!=m_left.size(); ++il ) + { + for( size_t ir=0; ir!=m_right.size(); ++ir ) + { + // m_mul += m_left[il] * m_middle[il][ir] * m_right[ir].T; + const RI::Tensor m_lm = RI::Tensor_Multiply::x0y1_x0a_ay1(m_left[il], m_middle[il][ir]); + const RI::Tensor m_lmr = RI::Tensor_Multiply::x0y0_x0a_y0a(m_lm, m_right[ir]); + if(m_mul.empty()) + { m_mul = std::move(m_lmr); } + else + { m_mul += m_lmr; } + } + } + return m_mul; } std::vector>> Exx_Opt_Orb::cal_I( - const std::map>>>> &ms, - const size_t TA, const size_t IA, const size_t TB, const size_t IB ) const + const std::map>>>> &ms, + const size_t TA, const size_t IA, const size_t TB, const size_t IB ) const { - ModuleBase::TITLE("Exx_Opt_Orb::cal_I"); + ModuleBase::TITLE("Exx_Opt_Orb::cal_I"); - if( TA==TB && IA==IB ) - { - return {{LRI_CV_Tools::cal_I(RI::Tensor(ms.at(TA).at(IA).at(TA).at(IA)))}}; - } - else - { - std::vector>> m_in - {{ ms.at(TA).at(IA).at(TA).at(IA), - ms.at(TA).at(IA).at(TB).at(IB) }, - { ms.at(TB).at(IB).at(TA).at(IA), - ms.at(TB).at(IB).at(TB).at(IB) }}; - return LRI_CV_Tools::cal_I(m_in); - } + if( TA==TB && IA==IB ) + { + return {{LRI_CV_Tools::cal_I(RI::Tensor(ms.at(TA).at(IA).at(TA).at(IA)))}}; + } + else + { + std::vector>> m_in + {{ ms.at(TA).at(IA).at(TA).at(IA), + ms.at(TA).at(IA).at(TB).at(IB) }, + { ms.at(TB).at(IB).at(TA).at(IA), + ms.at(TB).at(IB).at(TB).at(IB) }}; + return LRI_CV_Tools::cal_I(m_in); + } } std::map>> Exx_Opt_Orb::get_radial_R(const UnitCell& ucell) const { - ModuleBase::TITLE("Exx_Opt_Orb::get_radial_R"); - std::map>> radial_R; - for( size_t TA=0; TA!=ucell.ntype; ++TA ) { - for( size_t IA=0; IA!=ucell.atoms[TA].na; ++IA ) { - for( size_t TB=0; TB!=ucell.ntype; ++TB ) { - for( size_t IB=0; IB!=ucell.atoms[TB].na; ++IB ) - { - const ModuleBase::Vector3 &tauA = ucell.atoms[TA].tau[IA]; - const ModuleBase::Vector3 &tauB = ucell.atoms[TB].tau[IB]; - const double delta_R = (-tauA+tauB).norm(); - radial_R[TA][TB].insert( delta_R ); - radial_R[TB][TA].insert( delta_R ); - } - } - } - } - return radial_R; + ModuleBase::TITLE("Exx_Opt_Orb::get_radial_R"); + std::map>> radial_R; + for( size_t TA=0; TA!=ucell.ntype; ++TA ) { + for( size_t IA=0; IA!=ucell.atoms[TA].na; ++IA ) { + for( size_t TB=0; TB!=ucell.ntype; ++TB ) { + for( size_t IB=0; IB!=ucell.atoms[TB].na; ++IB ) + { + const ModuleBase::Vector3 &tauA = ucell.atoms[TA].tau[IA]; + const ModuleBase::Vector3 &tauB = ucell.atoms[TB].tau[IB]; + const double delta_R = (-tauA+tauB).norm(); + radial_R[TA][TB].insert( delta_R ); + radial_R[TB][TA].insert( delta_R ); + } + } + } + } + return radial_R; } diff --git a/source/source_lcao/module_ri/exx_opt_orb.h b/source/source_lcao/module_ri/exx_opt_orb.h index 418db018fa7..d6070720911 100644 --- a/source/source_lcao/module_ri/exx_opt_orb.h +++ b/source/source_lcao/module_ri/exx_opt_orb.h @@ -14,45 +14,45 @@ class Exx_Opt_Orb { public: - void generate_matrix( - const Exx_Info_Opt_ABFs &info, - const K_Vectors &kv, - const UnitCell &ucell, - const LCAO_Orbitals &orb) const; + void generate_matrix( + const Exx_Info_Opt_ABFs &info, + const K_Vectors &kv, + const UnitCell &ucell, + const LCAO_Orbitals &orb) const; private: - std::vector>> cal_I( - const std::map>>>> &ms, - const size_t TA, const size_t IA, const size_t TB, const size_t IB ) const; - RI::Tensor cal_mul_22( - const std::vector> & m_left, - const std::vector>> & m_middle, - const std::vector> & m_right ) const; - RI::Tensor cal_mul_21( - const std::vector> & m_left, - const std::vector>> & m_middle, - const std::vector> & m_right ) const; - RI::Tensor cal_mul_12( - const std::vector> & m_left, - const std::vector>> & m_middle, - const std::vector> & m_right ) const; - RI::Tensor cal_mul_11( - const std::vector> & m_left, - const std::vector>> & m_middle, - const std::vector> & m_right ) const; - void print_matrix( - const Exx_Info_Opt_ABFs &info, - const UnitCell& ucell, - const K_Vectors &kv, - const int Lmax, - const std::vector &ecut_number, - const std::string& file_name, - const std::vector> &matrix_Q, - const std::vector>> &matrix_S, - const RI::Tensor &matrix_V, - const size_t TA, const size_t IA, const size_t TB, const size_t IB, - const std::vector& orb_cutoff, - const ModuleBase::Element_Basis_Index::Range &range_jles, - const ModuleBase::Element_Basis_Index::IndexLNM &index_jles) const; - std::map>> get_radial_R(const UnitCell& ucell) const; + std::vector>> cal_I( + const std::map>>>> &ms, + const size_t TA, const size_t IA, const size_t TB, const size_t IB ) const; + RI::Tensor cal_mul_22( + const std::vector> & m_left, + const std::vector>> & m_middle, + const std::vector> & m_right ) const; + RI::Tensor cal_mul_21( + const std::vector> & m_left, + const std::vector>> & m_middle, + const std::vector> & m_right ) const; + RI::Tensor cal_mul_12( + const std::vector> & m_left, + const std::vector>> & m_middle, + const std::vector> & m_right ) const; + RI::Tensor cal_mul_11( + const std::vector> & m_left, + const std::vector>> & m_middle, + const std::vector> & m_right ) const; + void print_matrix( + const Exx_Info_Opt_ABFs &info, + const UnitCell& ucell, + const K_Vectors &kv, + const int Lmax, + const std::vector &ecut_number, + const std::string& file_name, + const std::vector> &matrix_Q, + const std::vector>> &matrix_S, + const RI::Tensor &matrix_V, + const size_t TA, const size_t IA, const size_t TB, const size_t IB, + const std::vector& orb_cutoff, + const ModuleBase::Element_Basis_Index::Range &range_jles, + const ModuleBase::Element_Basis_Index::IndexLNM &index_jles) const; + std::map>> get_radial_R(const UnitCell& ucell) const; }; #endif diff --git a/source/source_lcao/module_ri/exx_opt_orb_print.cpp b/source/source_lcao/module_ri/exx_opt_orb_print.cpp index d974faf892e..6d76c46fb8c 100644 --- a/source/source_lcao/module_ri/exx_opt_orb_print.cpp +++ b/source/source_lcao/module_ri/exx_opt_orb_print.cpp @@ -4,207 +4,207 @@ #include void Exx_Opt_Orb::print_matrix( - const Exx_Info_Opt_ABFs &info, - const UnitCell& ucell, - const K_Vectors &kv, - const int Lmax, - const std::vector &ecut_number, - const std::string &file_name, - const std::vector> &matrix_Q, - const std::vector>> &matrix_S, - const RI::Tensor &matrix_V, - const std::size_t TA, const std::size_t IA, const std::size_t TB, const std::size_t IB, - const std::vector& orb_cutoff, - const ModuleBase::Element_Basis_Index::Range &range_jles, - const ModuleBase::Element_Basis_Index::IndexLNM &index_jles) const + const Exx_Info_Opt_ABFs &info, + const UnitCell& ucell, + const K_Vectors &kv, + const int Lmax, + const std::vector &ecut_number, + const std::string &file_name, + const std::vector> &matrix_Q, + const std::vector>> &matrix_S, + const RI::Tensor &matrix_V, + const std::size_t TA, const std::size_t IA, const std::size_t TB, const std::size_t IB, + const std::vector& orb_cutoff, + const ModuleBase::Element_Basis_Index::Range &range_jles, + const ModuleBase::Element_Basis_Index::IndexLNM &index_jles) const { - auto print_header = [&]( std::ofstream &ofs ) - { - ofs << ucell.lat0 << std::endl; + auto print_header = [&]( std::ofstream &ofs ) + { + ofs << ucell.lat0 << std::endl; - ofs << ucell.latvec.e11 << " " << ucell.latvec.e12 << " " << ucell.latvec.e13 << std::endl; - ofs << ucell.latvec.e21 << " " << ucell.latvec.e22 << " " << ucell.latvec.e23 << std::endl; - ofs << ucell.latvec.e31 << " " << ucell.latvec.e32 << " " << ucell.latvec.e33 << std::endl; - - if( TA==TB ) - { - ofs << 1 << " ntype" << std::endl; - ofs << ucell.atoms[TA].label << " label" << std::endl; - if( IA==IB ) - { - ofs << 1 << " na" << std::endl; - ofs << ucell.atoms[TA].tau[IA].x << " " - << ucell.atoms[TA].tau[IA].y << " " - << ucell.atoms[TA].tau[IA].z << std::endl; - } - else - { - ofs << 2 << " na" << std::endl; - ofs << ucell.atoms[TA].tau[IA].x << " " - << ucell.atoms[TA].tau[IA].y << " " - << ucell.atoms[TA].tau[IA].z << std::endl; - ofs << ucell.atoms[TB].tau[IB].x << " " - << ucell.atoms[TB].tau[IB].y << " " - << ucell.atoms[TB].tau[IB].z << std::endl; - } - } - else - { - ofs << 2 << " ntype" << std::endl; - ofs << ucell.atoms[TA].label << " label" << std::endl; - ofs << 1 << " na" << std::endl; - ofs << ucell.atoms[TA].tau[IA].x << " " - << ucell.atoms[TA].tau[IA].y << " " - << ucell.atoms[TA].tau[IA].z << std::endl; - ofs << ucell.atoms[TB].label << " label" << std::endl; - ofs << 1 << " na" << std::endl; - ofs << ucell.atoms[TB].tau[IB].x << " " - << ucell.atoms[TB].tau[IB].y << " " - << ucell.atoms[TB].tau[IB].z << std::endl; - } - - ofs << info.ecut_exx << " ecutwfc" << std::endl; + ofs << ucell.latvec.e11 << " " << ucell.latvec.e12 << " " << ucell.latvec.e13 << std::endl; + ofs << ucell.latvec.e21 << " " << ucell.latvec.e22 << " " << ucell.latvec.e23 << std::endl; + ofs << ucell.latvec.e31 << " " << ucell.latvec.e32 << " " << ucell.latvec.e33 << std::endl; + + if( TA==TB ) + { + ofs << 1 << " ntype" << std::endl; + ofs << ucell.atoms[TA].label << " label" << std::endl; + if( IA==IB ) + { + ofs << 1 << " na" << std::endl; + ofs << ucell.atoms[TA].tau[IA].x << " " + << ucell.atoms[TA].tau[IA].y << " " + << ucell.atoms[TA].tau[IA].z << std::endl; + } + else + { + ofs << 2 << " na" << std::endl; + ofs << ucell.atoms[TA].tau[IA].x << " " + << ucell.atoms[TA].tau[IA].y << " " + << ucell.atoms[TA].tau[IA].z << std::endl; + ofs << ucell.atoms[TB].tau[IB].x << " " + << ucell.atoms[TB].tau[IB].y << " " + << ucell.atoms[TB].tau[IB].z << std::endl; + } + } + else + { + ofs << 2 << " ntype" << std::endl; + ofs << ucell.atoms[TA].label << " label" << std::endl; + ofs << 1 << " na" << std::endl; + ofs << ucell.atoms[TA].tau[IA].x << " " + << ucell.atoms[TA].tau[IA].y << " " + << ucell.atoms[TA].tau[IA].z << std::endl; + ofs << ucell.atoms[TB].label << " label" << std::endl; + ofs << 1 << " na" << std::endl; + ofs << ucell.atoms[TB].tau[IB].x << " " + << ucell.atoms[TB].tau[IB].y << " " + << ucell.atoms[TB].tau[IB].z << std::endl; + } + + ofs << info.ecut_exx << " ecutwfc" << std::endl; - // this parameter determine the total number of jlq. - ofs << info.ecut_exx << " ecutwfc_jlq" << std::endl; + // this parameter determine the total number of jlq. + ofs << info.ecut_exx << " ecutwfc_jlq" << std::endl; - if(TA==TB) - { ofs << orb_cutoff[TA] << " rcut_Jlq" << std::endl; } - else - { ofs << orb_cutoff[TA] << " " << orb_cutoff[TB] << " rcut_Jlq" << std::endl; } + if(TA==TB) + { ofs << orb_cutoff[TA] << " rcut_Jlq" << std::endl; } + else + { ofs << orb_cutoff[TA] << " " << orb_cutoff[TB] << " rcut_Jlq" << std::endl; } - ofs << 0 << " smooth" << std::endl; - ofs << 0 << " smearing_sigma" << std::endl; + ofs << 0 << " smooth" << std::endl; + ofs << 0 << " smearing_sigma" << std::endl; - ofs << info.tolerence << " tolerence" << std::endl; + ofs << info.tolerence << " tolerence" << std::endl; - ofs << Lmax << " lmax" << std::endl; + ofs << Lmax << " lmax" << std::endl; - ofs << kv.get_nkstot() << " nks" << std::endl; - assert( matrix_V.shape[0]*matrix_V.shape[1] == matrix_V.shape[2]*matrix_V.shape[3] ); - ofs << matrix_V.shape[0]*matrix_V.shape[1] << " nbands" << std::endl; - - auto cal_sum_M = [&range_jles](std::size_t T) -> std::size_t - { - std::size_t sum_M = 0; - for( std::size_t L = 0; L!=range_jles[T].size(); ++L ) - { sum_M += range_jles[T][L].M; } - return sum_M; - }; - const std::size_t nwfc = (TA==TB && IA==IB) ? cal_sum_M(TA) : cal_sum_M(TA)+cal_sum_M(TB); - ofs << nwfc << " nwfc" << std::endl; - - for(const std::size_t ne : ecut_number) - { ofs << ne << " "; } - ofs << "ne" << std::endl; - - ofs << "" << std::endl; - for( int ik=0; ik!=kv.get_nkstot(); ++ik ) - { - ofs << kv.kvec_c[ik].x << " " << kv.kvec_c[ik].y << " " << kv.kvec_c[ik].z; - ofs << " " << kv.wk[ik] * 0.5 << std::endl; - } - ofs << "" << std::endl; + ofs << kv.get_nkstot() << " nks" << std::endl; + assert( matrix_V.shape[0]*matrix_V.shape[1] == matrix_V.shape[2]*matrix_V.shape[3] ); + ofs << matrix_V.shape[0]*matrix_V.shape[1] << " nbands" << std::endl; + + auto cal_sum_M = [&range_jles](std::size_t T) -> std::size_t + { + std::size_t sum_M = 0; + for( std::size_t L = 0; L!=range_jles[T].size(); ++L ) + { sum_M += range_jles[T][L].M; } + return sum_M; + }; + const std::size_t nwfc = (TA==TB && IA==IB) ? cal_sum_M(TA) : cal_sum_M(TA)+cal_sum_M(TB); + ofs << nwfc << " nwfc" << std::endl; + + for(const std::size_t ne : ecut_number) + { ofs << ne << " "; } + ofs << "ne" << std::endl; + + ofs << "" << std::endl; + for( int ik=0; ik!=kv.get_nkstot(); ++ik ) + { + ofs << kv.kvec_c[ik].x << " " << kv.kvec_c[ik].y << " " << kv.kvec_c[ik].z; + ofs << " " << kv.wk[ik] * 0.5 << std::endl; + } + ofs << "" << std::endl; - ofs << std::endl; - }; - - - auto print_Q = [&]( std::ofstream &ofs ) - { - //--------------------- - // < Psi | jY > - //--------------------- - ofs<< "" << std::endl; - for( std::size_t iw0=0; iw0!=matrix_V.shape[0]; ++iw0 ) - { - for( std::size_t iw1=0; iw1!=matrix_V.shape[1]; ++iw1 ) - { - for( std::size_t iat=0; iat!=matrix_Q.size(); ++iat ) - { - const std::size_t it = (iat==0) ? TA : TB; - for( std::size_t il=0; il!=range_jles[it].size(); ++il ) - { - for( std::size_t im=0; im!=range_jles[it][il].M; ++im ) - { - for( std::size_t iq=0; iq!=range_jles[it][il].N; ++iq ) - { - ofs<" << std::endl << std::endl; - }; - - - auto print_S = [&]( std::ofstream &ofs, const double scale=1 ) - { - //--------------------- - // < jY | jY > - //--------------------- - ofs<< "" <" << std::endl << std::endl; - }; - - - auto print_V = [&]( std::ofstream &ofs, const double scale=1 ) - { - //--------------------- - // < Psi | Psi > - //--------------------- - ofs << "" << std::endl; - for( std::size_t iw0=0; iw0!=matrix_V.shape[0]; ++iw0 ) - { - for( std::size_t iw1=0; iw1!=matrix_V.shape[1]; ++iw1 ) - { - for( std::size_t iw2=0; iw2!=matrix_V.shape[2]; ++iw2 ) - { - for( std::size_t iw3=0; iw3!=matrix_V.shape[3]; ++iw3 ) - { - ofs<" << std::endl << std::endl; - }; - - ModuleBase::TITLE("Exx_Opt_Orb","print_matrix"); - std::ofstream ofs(file_name+"_"+std::to_string(TA)+"_"+std::to_string(IA)+"_"+std::to_string(TB)+"_"+std::to_string(IB)); - print_header(ofs); - ofs< + //--------------------- + ofs<< "" << std::endl; + for( std::size_t iw0=0; iw0!=matrix_V.shape[0]; ++iw0 ) + { + for( std::size_t iw1=0; iw1!=matrix_V.shape[1]; ++iw1 ) + { + for( std::size_t iat=0; iat!=matrix_Q.size(); ++iat ) + { + const std::size_t it = (iat==0) ? TA : TB; + for( std::size_t il=0; il!=range_jles[it].size(); ++il ) + { + for( std::size_t im=0; im!=range_jles[it][il].M; ++im ) + { + for( std::size_t iq=0; iq!=range_jles[it][il].N; ++iq ) + { + ofs<" << std::endl << std::endl; + }; + + + auto print_S = [&]( std::ofstream &ofs, const double scale=1 ) + { + //--------------------- + // < jY | jY > + //--------------------- + ofs<< "" <" << std::endl << std::endl; + }; + + + auto print_V = [&]( std::ofstream &ofs, const double scale=1 ) + { + //--------------------- + // < Psi | Psi > + //--------------------- + ofs << "" << std::endl; + for( std::size_t iw0=0; iw0!=matrix_V.shape[0]; ++iw0 ) + { + for( std::size_t iw1=0; iw1!=matrix_V.shape[1]; ++iw1 ) + { + for( std::size_t iw2=0; iw2!=matrix_V.shape[2]; ++iw2 ) + { + for( std::size_t iw3=0; iw3!=matrix_V.shape[3]; ++iw3 ) + { + ofs<" << std::endl << std::endl; + }; + + ModuleBase::TITLE("Exx_Opt_Orb","print_matrix"); + std::ofstream ofs(file_name+"_"+std::to_string(TA)+"_"+std::to_string(IA)+"_"+std::to_string(TB)+"_"+std::to_string(IB)); + print_header(ofs); + ofs< diff --git a/source/source_lcao/module_ri/invmat_ri.h b/source/source_lcao/module_ri/invmat_ri.h index 4a79a1458e7..a1cb75cf1a9 100644 --- a/source/source_lcao/module_ri/invmat_ri.h +++ b/source/source_lcao/module_ri/invmat_ri.h @@ -3,7 +3,9 @@ // DATE : 2022-08-17 //======================= -#pragma once +#ifndef ABACUS_SOURCE_LCAO_MODULE_RI_INVMAT_RI_H +#define ABACUS_SOURCE_LCAO_MODULE_RI_INVMAT_RI_H + #include "abfs_construct_pca.h" @@ -34,4 +36,6 @@ class Inverse_Matrix RI::Tensor A; }; -#include "invmat_ri.hpp" \ No newline at end of file +#include "invmat_ri.hpp" + +#endif // ABACUS_SOURCE_LCAO_MODULE_RI_INVMAT_RI_H diff --git a/source/source_lcao/module_ri/lri_cv.h b/source/source_lcao/module_ri/lri_cv.h index c78839bdac4..12359d3b5e3 100644 --- a/source/source_lcao/module_ri/lri_cv.h +++ b/source/source_lcao/module_ri/lri_cv.h @@ -25,83 +25,83 @@ template class LRI_CV { private: - using TA = int; - using TC = std::array; - using TAC = std::pair; - using Tdata_real = RI::Global_Func::To_Real_t; + using TA = int; + using TC = std::array; + using TAC = std::pair; + using Tdata_real = RI::Global_Func::To_Real_t; public: - LRI_CV(); - ~LRI_CV(); + LRI_CV(); + ~LRI_CV(); - void set_info_ri(const Exx_Info_RI* p) { p_info_ri = p; } + void set_info_ri(const Exx_Info_RI* p) { p_info_ri = p; } - void set_orbitals( - const UnitCell &ucell, + void set_orbitals( + const UnitCell &ucell, const LCAO_Orbitals& orb, - const std::vector>> &lcaos_in, - const std::vector>> &abfs_in, - const std::vector>> &abfs_ccp_in, - const double &kmesh_times, - std::shared_ptr MGT, + const std::vector>> &lcaos_in, + const std::vector>> &abfs_in, + const std::vector>> &abfs_ccp_in, + const double &kmesh_times, + std::shared_ptr MGT, const bool& init_C); - inline std::map>> - cal_Vs( - const UnitCell &ucell, - const std::vector &list_A0, - const std::vector &list_A1, - const std::map &flags); // "writable_Vws" - inline std::map, 3>>> - cal_dVs( - const UnitCell &ucell, - const std::vector &list_A0, - const std::vector &list_A1, - const std::map &flags); // "writable_dVws" - std::pair>>, + inline std::map>> + cal_Vs( + const UnitCell &ucell, + const std::vector &list_A0, + const std::vector &list_A1, + const std::map &flags); // "writable_Vws" + inline std::map, 3>>> + cal_dVs( + const UnitCell &ucell, + const std::vector &list_A0, + const std::vector &list_A1, + const std::map &flags); // "writable_dVws" + std::pair>>, std::map, 3>>>> - cal_Cs_dCs( - const UnitCell &ucell, - const std::vector &list_A0, - const std::vector &list_A1, - const std::map &flags); // "cal_dC", "writable_Cws", "writable_dCws", "writable_Vws", "writable_dVws" + cal_Cs_dCs( + const UnitCell &ucell, + const std::vector &list_A0, + const std::vector &list_A1, + const std::map &flags); // "cal_dC", "writable_Cws", "writable_dCws", "writable_Vws", "writable_dVws" - size_t get_index_abfs_size(const size_t &iat){return this->index_abfs[iat].count_size; } + size_t get_index_abfs_size(const size_t &iat){return this->index_abfs[iat].count_size; } private: - std::vector>> lcaos; - std::vector>> abfs; - std::vector>> abfs_ccp; - ModuleBase::Element_Basis_Index::IndexLNM index_lcaos; - ModuleBase::Element_Basis_Index::IndexLNM index_abfs; - std::vector lcaos_rcut; + std::vector>> lcaos; + std::vector>> abfs; + std::vector>> abfs_ccp; + ModuleBase::Element_Basis_Index::IndexLNM index_lcaos; + ModuleBase::Element_Basis_Index::IndexLNM index_abfs; + std::vector lcaos_rcut; std::vector abfs_ccp_rcut; - const Exx_Info_RI* p_info_ri = nullptr; + const Exx_Info_RI* p_info_ri = nullptr; public: - std::map,RI::Tensor>>> Vws; - std::map,RI::Tensor>>> Cws; - std::map,std::array,3>>>> dVws; - std::map,std::array,3>>>> dCws; + std::map,RI::Tensor>>> Vws; + std::map,RI::Tensor>>> Cws; + std::map,std::array,3>>>> dVws; + std::map,std::array,3>>>> dCws; private: - pthread_rwlock_t rwlock_Vw; - pthread_rwlock_t rwlock_Cw; - pthread_rwlock_t rwlock_dVw; - pthread_rwlock_t rwlock_dCw; - - Matrix_Orbs11 m_abfs_abfs; - Matrix_Orbs21 m_abfslcaos_lcaos; - - template - using T_func_DPcal_data = std::function &R, - const std::map &flags)>; - using T_func_cal_Rcut = std::function; - template - std::map> - cal_datas( - const UnitCell &ucell, + pthread_rwlock_t rwlock_Vw; + pthread_rwlock_t rwlock_Cw; + pthread_rwlock_t rwlock_dVw; + pthread_rwlock_t rwlock_dCw; + + Matrix_Orbs11 m_abfs_abfs; + Matrix_Orbs21 m_abfslcaos_lcaos; + + template + using T_func_DPcal_data = std::function &R, + const std::map &flags)>; + using T_func_cal_Rcut = std::function; + template + std::map> + cal_datas( + const UnitCell &ucell, const std::vector& list_A0, const std::vector& list_A1, const std::map& flags, @@ -111,34 +111,34 @@ class LRI_CV inline double cal_V_Rcut(const int it0, const int it1); inline double cal_C_Rcut(const int it0, const int it1); - inline RI::Tensor - DPcal_V( - const int it0, - const int it1, - const Abfs::Vector3_Order &R, - const std::map &flags); // "writable_Vws" - inline std::array,3> - DPcal_dV( - const int it0, - const int it1, - const Abfs::Vector3_Order &R, - const std::map &flags); // "writable_dVws" - std::pair, std::array,3>> - DPcal_C_dC( - const int it0, - const int it1, - const Abfs::Vector3_Order &R, - const std::map &flags); // "cal_dC", "writable_Cws", "writable_dCws", "writable_Vws", "writable_dVws" - - template - To11 DPcal_o11( - const int it0, - const int it1, - const Abfs::Vector3_Order &R, - const bool &flag_writable_o11ws, - pthread_rwlock_t &rwlock_o11, - std::map,To11>>> &o11ws, - const Tfunc &func_cal_o11); + inline RI::Tensor + DPcal_V( + const int it0, + const int it1, + const Abfs::Vector3_Order &R, + const std::map &flags); // "writable_Vws" + inline std::array,3> + DPcal_dV( + const int it0, + const int it1, + const Abfs::Vector3_Order &R, + const std::map &flags); // "writable_dVws" + std::pair, std::array,3>> + DPcal_C_dC( + const int it0, + const int it1, + const Abfs::Vector3_Order &R, + const std::map &flags); // "cal_dC", "writable_Cws", "writable_dCws", "writable_Vws", "writable_dVws" + + template + To11 DPcal_o11( + const int it0, + const int it1, + const Abfs::Vector3_Order &R, + const bool &flag_writable_o11ws, + pthread_rwlock_t &rwlock_o11, + std::map,To11>>> &o11ws, + const Tfunc &func_cal_o11); }; #include "lri_cv.hpp" diff --git a/source/source_lcao/module_ri/lri_cv.hpp b/source/source_lcao/module_ri/lri_cv.hpp index e3eaa2d6148..49f197816c0 100644 --- a/source/source_lcao/module_ri/lri_cv.hpp +++ b/source/source_lcao/module_ri/lri_cv.hpp @@ -20,66 +20,66 @@ template LRI_CV::LRI_CV() { - pthread_rwlock_init(&rwlock_Vw,NULL); - pthread_rwlock_init(&rwlock_Cw,NULL); - pthread_rwlock_init(&rwlock_dVw,NULL); - pthread_rwlock_init(&rwlock_dCw,NULL); + pthread_rwlock_init(&rwlock_Vw,NULL); + pthread_rwlock_init(&rwlock_Cw,NULL); + pthread_rwlock_init(&rwlock_dVw,NULL); + pthread_rwlock_init(&rwlock_dCw,NULL); } template LRI_CV::~LRI_CV() { - pthread_rwlock_destroy(&rwlock_Vw); - pthread_rwlock_destroy(&rwlock_Cw); - pthread_rwlock_destroy(&rwlock_dVw); - pthread_rwlock_destroy(&rwlock_dCw); + pthread_rwlock_destroy(&rwlock_Vw); + pthread_rwlock_destroy(&rwlock_Cw); + pthread_rwlock_destroy(&rwlock_dVw); + pthread_rwlock_destroy(&rwlock_dCw); } template void LRI_CV::set_orbitals( - const UnitCell &ucell, - const LCAO_Orbitals& orb, - const std::vector>> &lcaos_in, - const std::vector>> &abfs_in, - const std::vector>> &abfs_ccp_in, - const double &kmesh_times, - std::shared_ptr MGT, + const UnitCell &ucell, + const LCAO_Orbitals& orb, + const std::vector>> &lcaos_in, + const std::vector>> &abfs_in, + const std::vector>> &abfs_ccp_in, + const double &kmesh_times, + std::shared_ptr MGT, const bool& init_C) { - ModuleBase::TITLE("LRI_CV", "set_orbitals"); - ModuleBase::timer::start("LRI_CV", "set_orbitals"); + ModuleBase::TITLE("LRI_CV", "set_orbitals"); + ModuleBase::timer::start("LRI_CV", "set_orbitals"); - this->lcaos = lcaos_in; - this->abfs = abfs_in; - this->abfs_ccp = abfs_ccp_in; + this->lcaos = lcaos_in; + this->abfs = abfs_in; + this->abfs_ccp = abfs_ccp_in; - this->lcaos_rcut = Exx_Abfs::Construct_Orbs::get_Rcut(this->lcaos); + this->lcaos_rcut = Exx_Abfs::Construct_Orbs::get_Rcut(this->lcaos); this->abfs_ccp_rcut = Exx_Abfs::Construct_Orbs::get_Rcut(this->abfs_ccp); - const ModuleBase::Element_Basis_Index::Range - range_lcaos = ModuleBase::Element_Basis_Index::construct_range( lcaos ); - this->index_lcaos = ModuleBase::Element_Basis_Index::construct_index( range_lcaos ); + const ModuleBase::Element_Basis_Index::Range + range_lcaos = ModuleBase::Element_Basis_Index::construct_range( lcaos ); + this->index_lcaos = ModuleBase::Element_Basis_Index::construct_index( range_lcaos ); - const ModuleBase::Element_Basis_Index::Range - range_abfs = ModuleBase::Element_Basis_Index::construct_range( abfs ); - this->index_abfs = ModuleBase::Element_Basis_Index::construct_index( range_abfs ); + const ModuleBase::Element_Basis_Index::Range + range_abfs = ModuleBase::Element_Basis_Index::construct_range( abfs ); + this->index_abfs = ModuleBase::Element_Basis_Index::construct_index( range_abfs ); - this->m_abfs_abfs.MGT = this->m_abfslcaos_lcaos.MGT = MGT; + this->m_abfs_abfs.MGT = this->m_abfslcaos_lcaos.MGT = MGT; this->m_abfs_abfs.init( - this->abfs_ccp, this->abfs, - ucell, orb, kmesh_times); + this->abfs_ccp, this->abfs, + ucell, orb, kmesh_times); if (init_C) this->m_abfslcaos_lcaos.init( - this->abfs_ccp, this->lcaos, this->lcaos, - ucell, orb, kmesh_times); + this->abfs_ccp, this->lcaos, this->lcaos, + ucell, orb, kmesh_times); this->m_abfs_abfs.init_radial_table(); if (init_C) { this->m_abfslcaos_lcaos.init_radial_table(); } - ModuleBase::timer::end("LRI_CV", "set_orbitals"); + ModuleBase::timer::end("LRI_CV", "set_orbitals"); } template @@ -95,7 +95,7 @@ double LRI_CV::cal_C_Rcut(const int it0, const int it1) { template template auto LRI_CV::cal_datas( - const UnitCell &ucell, + const UnitCell &ucell, const std::vector& list_A0, const std::vector& list_A1, const std::map& flags, @@ -103,111 +103,111 @@ auto LRI_CV::cal_datas( const T_func_DPcal_data& func_DPcal_data) -> std::map> { - ModuleBase::TITLE("LRI_CV","cal_datas"); - ModuleBase::timer::start("LRI_CV", "cal_datas"); - - std::map> Datas; - #pragma omp parallel - for(size_t i0=0; i0 tau0 = ucell.atoms[it0].tau[ia0]; - const ModuleBase::Vector3 tau1 = ucell.atoms[it1].tau[ia1]; - const double Rcut + ModuleBase::TITLE("LRI_CV","cal_datas"); + ModuleBase::timer::start("LRI_CV", "cal_datas"); + + std::map> Datas; + #pragma omp parallel + for(size_t i0=0; i0 tau0 = ucell.atoms[it0].tau[ia0]; + const ModuleBase::Vector3 tau1 = ucell.atoms[it1].tau[ia1]; + const double Rcut = std::min(func_cal_Rcut(it0, it1), func_cal_Rcut(it1, it0)); - const Abfs::Vector3_Order R_delta = -tau0+tau1+(RI_Util::array3_to_Vector3(cell1)*ucell.latvec); - if( R_delta.norm()*ucell.lat0 < Rcut ) - { - const Tresult Data = func_DPcal_data(it0, it1, R_delta, flags); - // if(Data.norm(std::numeric_limits::max()) > threshold) - // { - #pragma omp critical(LRI_CV_cal_datas) - Datas[list_A0[i0]][list_A1[i1]] = Data; - // } - } - } - } - ModuleBase::timer::end("LRI_CV", "cal_datas"); - return Datas; + const Abfs::Vector3_Order R_delta = -tau0+tau1+(RI_Util::array3_to_Vector3(cell1)*ucell.latvec); + if( R_delta.norm()*ucell.lat0 < Rcut ) + { + const Tresult Data = func_DPcal_data(it0, it1, R_delta, flags); + // if(Data.norm(std::numeric_limits::max()) > threshold) + // { + #pragma omp critical(LRI_CV_cal_datas) + Datas[list_A0[i0]][list_A1[i1]] = Data; + // } + } + } + } + ModuleBase::timer::end("LRI_CV", "cal_datas"); + return Datas; } template auto LRI_CV::cal_Vs( - const UnitCell &ucell, - const std::vector &list_A0, - const std::vector &list_A1, - const std::map &flags) // + "writable_Vws" + const UnitCell &ucell, + const std::vector &list_A0, + const std::vector &list_A1, + const std::map &flags) // + "writable_Vws" -> std::map>> { - ModuleBase::TITLE("LRI_CV","cal_Vs"); - const T_func_DPcal_data> - func_DPcal_V = std::bind( - &LRI_CV::DPcal_V, this, - std::placeholders::_1, std::placeholders::_2, std::placeholders::_3, std::placeholders::_4); - const T_func_cal_Rcut func_cal_Rcut = std::bind(&LRI_CV::cal_V_Rcut, + ModuleBase::TITLE("LRI_CV","cal_Vs"); + const T_func_DPcal_data> + func_DPcal_V = std::bind( + &LRI_CV::DPcal_V, this, + std::placeholders::_1, std::placeholders::_2, std::placeholders::_3, std::placeholders::_4); + const T_func_cal_Rcut func_cal_Rcut = std::bind(&LRI_CV::cal_V_Rcut, this, std::placeholders::_1, std::placeholders::_2); - return this->cal_datas(ucell,list_A0, list_A1, flags, func_cal_Rcut, func_DPcal_V); + return this->cal_datas(ucell,list_A0, list_A1, flags, func_cal_Rcut, func_DPcal_V); } template auto LRI_CV::cal_dVs( - const UnitCell &ucell, - const std::vector &list_A0, - const std::vector &list_A1, - const std::map &flags) // + "writable_dVws" + const UnitCell &ucell, + const std::vector &list_A0, + const std::vector &list_A1, + const std::map &flags) // + "writable_dVws" -> std::map, 3>>> { - ModuleBase::TITLE("LRI_CV","cal_dVs"); - const T_func_DPcal_data,3>> - func_DPcal_dV = std::bind( - &LRI_CV::DPcal_dV, this, - std::placeholders::_1, std::placeholders::_2, std::placeholders::_3, std::placeholders::_4); + ModuleBase::TITLE("LRI_CV","cal_dVs"); + const T_func_DPcal_data,3>> + func_DPcal_dV = std::bind( + &LRI_CV::DPcal_dV, this, + std::placeholders::_1, std::placeholders::_2, std::placeholders::_3, std::placeholders::_4); - const T_func_cal_Rcut func_cal_Rcut = std::bind(&LRI_CV::cal_V_Rcut, + const T_func_cal_Rcut func_cal_Rcut = std::bind(&LRI_CV::cal_V_Rcut, this, std::placeholders::_1, std::placeholders::_2); - return this->cal_datas(ucell,list_A0, list_A1, flags, func_cal_Rcut, func_DPcal_dV); + return this->cal_datas(ucell,list_A0, list_A1, flags, func_cal_Rcut, func_DPcal_dV); } template auto LRI_CV::cal_Cs_dCs( - const UnitCell &ucell, - const std::vector &list_A0, - const std::vector &list_A1, - const std::map &flags) // "cal_dC" + "writable_Cws", "writable_dCws", "writable_Vws", "writable_dVws" + const UnitCell &ucell, + const std::vector &list_A0, + const std::vector &list_A1, + const std::map &flags) // "cal_dC" + "writable_Cws", "writable_dCws", "writable_Vws", "writable_dVws" -> std::pair< std::map>>, std::map, 3>>>> { - ModuleBase::TITLE("LRI_CV","cal_Cs_dCs"); - const T_func_DPcal_data, std::array,3>>> - func_DPcal_C_dC = std::bind( - &LRI_CV::DPcal_C_dC, this, - std::placeholders::_1, std::placeholders::_2, std::placeholders::_3, std::placeholders::_4); + ModuleBase::TITLE("LRI_CV","cal_Cs_dCs"); + const T_func_DPcal_data, std::array,3>>> + func_DPcal_C_dC = std::bind( + &LRI_CV::DPcal_C_dC, this, + std::placeholders::_1, std::placeholders::_2, std::placeholders::_3, std::placeholders::_4); const T_func_cal_Rcut func_cal_Rcut = std::bind(&LRI_CV::cal_C_Rcut, this, std::placeholders::_1, std::placeholders::_2); - std::map, std::array,3>>>> - Cs_dCs_tmp = this->cal_datas(ucell,list_A0, list_A1, flags, func_cal_Rcut, func_DPcal_C_dC); + std::map, std::array,3>>>> + Cs_dCs_tmp = this->cal_datas(ucell,list_A0, list_A1, flags, func_cal_Rcut, func_DPcal_C_dC); - std::map>> Cs; + std::map>> Cs; std::map, 3>>> dCs; for (auto& Cs_dCs_A: Cs_dCs_tmp) for (auto& Cs_dCs_B: Cs_dCs_A.second) { @@ -217,140 +217,140 @@ auto LRI_CV::cal_Cs_dCs( dCs[Cs_dCs_A.first][Cs_dCs_B.first] = std::move(std::get<1>(Cs_dCs_B.second)); } - return std::make_pair(Cs, dCs); + return std::make_pair(Cs, dCs); } template template To11 LRI_CV::DPcal_o11( - const int it0, - const int it1, - const Abfs::Vector3_Order &R, - const bool &flag_writable_o11ws, - pthread_rwlock_t &rwlock_o11, - std::map,To11>>> &o11ws, - const Tfunc &func_cal_o11) + const int it0, + const int it1, + const Abfs::Vector3_Order &R, + const bool &flag_writable_o11ws, + pthread_rwlock_t &rwlock_o11, + std::map,To11>>> &o11ws, + const Tfunc &func_cal_o11) { - const Abfs::Vector3_Order Rm = -R; - pthread_rwlock_rdlock(&rwlock_o11); - const To11 o11_read = RI::Global_Func::find(o11ws, it0, it1, R); - pthread_rwlock_unlock(&rwlock_o11); - - if(LRI_CV_Tools::exist(o11_read)) - { - return o11_read; - } - else - { - pthread_rwlock_rdlock(&rwlock_o11); - const To11 o11_transform_read = RI::Global_Func::find(o11ws, it1, it0, Rm); - pthread_rwlock_unlock(&rwlock_o11); - - if(LRI_CV_Tools::exist(o11_transform_read)) - { - const To11 o11 = LRI_CV_Tools::transform_Rm(o11_transform_read); - if(flag_writable_o11ws) // such write may be deleted for memory saving with transform_Rm() every time - { - pthread_rwlock_wrlock(&rwlock_o11); - o11ws[it0][it1][R] = o11; - pthread_rwlock_unlock(&rwlock_o11); - } - return o11; - } - else - { - const To11 o11 = func_cal_o11( - it0, it1, ModuleBase::Vector3{0,0,0}, R, - this->index_abfs, this->index_abfs, - Matrix_Orbs11::Matrix_Order::AB); - if(flag_writable_o11ws) - { - pthread_rwlock_wrlock(&rwlock_o11); - o11ws[it0][it1][R] = o11; - pthread_rwlock_unlock(&rwlock_o11); - } - return o11; - } // end else (!exist(o11_transform_read)) - } // end else (!exist(o11_read)) + const Abfs::Vector3_Order Rm = -R; + pthread_rwlock_rdlock(&rwlock_o11); + const To11 o11_read = RI::Global_Func::find(o11ws, it0, it1, R); + pthread_rwlock_unlock(&rwlock_o11); + + if(LRI_CV_Tools::exist(o11_read)) + { + return o11_read; + } + else + { + pthread_rwlock_rdlock(&rwlock_o11); + const To11 o11_transform_read = RI::Global_Func::find(o11ws, it1, it0, Rm); + pthread_rwlock_unlock(&rwlock_o11); + + if(LRI_CV_Tools::exist(o11_transform_read)) + { + const To11 o11 = LRI_CV_Tools::transform_Rm(o11_transform_read); + if(flag_writable_o11ws) // such write may be deleted for memory saving with transform_Rm() every time + { + pthread_rwlock_wrlock(&rwlock_o11); + o11ws[it0][it1][R] = o11; + pthread_rwlock_unlock(&rwlock_o11); + } + return o11; + } + else + { + const To11 o11 = func_cal_o11( + it0, it1, ModuleBase::Vector3{0,0,0}, R, + this->index_abfs, this->index_abfs, + Matrix_Orbs11::Matrix_Order::AB); + if(flag_writable_o11ws) + { + pthread_rwlock_wrlock(&rwlock_o11); + o11ws[it0][it1][R] = o11; + pthread_rwlock_unlock(&rwlock_o11); + } + return o11; + } // end else (!exist(o11_transform_read)) + } // end else (!exist(o11_read)) } template RI::Tensor LRI_CV::DPcal_V( - const int it0, - const int it1, - const Abfs::Vector3_Order &R, - const std::map &flags) // "writable_Vws" + const int it0, + const int it1, + const Abfs::Vector3_Order &R, + const std::map &flags) // "writable_Vws" { - const auto cal_overlap_matrix = std::bind( - &Matrix_Orbs11::cal_overlap_matrix, - &this->m_abfs_abfs, - std::placeholders::_1, std::placeholders::_2, std::placeholders::_3, std::placeholders::_4, std::placeholders::_5, std::placeholders::_6, std::placeholders::_7); - return this->DPcal_o11(it0, it1, R, flags.at("writable_Vws"), this->rwlock_Vw, this->Vws, cal_overlap_matrix); + const auto cal_overlap_matrix = std::bind( + &Matrix_Orbs11::cal_overlap_matrix, + &this->m_abfs_abfs, + std::placeholders::_1, std::placeholders::_2, std::placeholders::_3, std::placeholders::_4, std::placeholders::_5, std::placeholders::_6, std::placeholders::_7); + return this->DPcal_o11(it0, it1, R, flags.at("writable_Vws"), this->rwlock_Vw, this->Vws, cal_overlap_matrix); } template std::array, 3> LRI_CV::DPcal_dV( - const int it0, - const int it1, - const Abfs::Vector3_Order &R, - const std::map &flags) // "writable_dVws" + const int it0, + const int it1, + const Abfs::Vector3_Order &R, + const std::map &flags) // "writable_dVws" { - if(ModuleBase::Vector3(0,0,0)==R) - { - assert(it0==it1); - const size_t size = this->index_abfs[it0].count_size; - const std::array, 3> dV = { RI::Tensor({size,size}), RI::Tensor({size,size}), RI::Tensor({size,size}) }; - if(flags.at("writable_dVws")) - { - pthread_rwlock_wrlock(&this->rwlock_dVw); - this->dVws[it0][it1][R] = dV; - pthread_rwlock_unlock(&this->rwlock_dVw); - } - return dV; - } - - const auto cal_grad_overlap_matrix = std::bind( - &Matrix_Orbs11::cal_grad_overlap_matrix, - &this->m_abfs_abfs, - std::placeholders::_1, std::placeholders::_2, std::placeholders::_3, std::placeholders::_4, std::placeholders::_5, std::placeholders::_6, std::placeholders::_7); - return this->DPcal_o11(it0, it1, R, flags.at("writable_dVws"), this->rwlock_dVw, this->dVws, cal_grad_overlap_matrix); + if(ModuleBase::Vector3(0,0,0)==R) + { + assert(it0==it1); + const size_t size = this->index_abfs[it0].count_size; + const std::array, 3> dV = { RI::Tensor({size,size}), RI::Tensor({size,size}), RI::Tensor({size,size}) }; + if(flags.at("writable_dVws")) + { + pthread_rwlock_wrlock(&this->rwlock_dVw); + this->dVws[it0][it1][R] = dV; + pthread_rwlock_unlock(&this->rwlock_dVw); + } + return dV; + } + + const auto cal_grad_overlap_matrix = std::bind( + &Matrix_Orbs11::cal_grad_overlap_matrix, + &this->m_abfs_abfs, + std::placeholders::_1, std::placeholders::_2, std::placeholders::_3, std::placeholders::_4, std::placeholders::_5, std::placeholders::_6, std::placeholders::_7); + return this->DPcal_o11(it0, it1, R, flags.at("writable_dVws"), this->rwlock_dVw, this->dVws, cal_grad_overlap_matrix); } template std::pair, std::array,3>> LRI_CV::DPcal_C_dC( - const int it0, - const int it1, - const Abfs::Vector3_Order &R, - const std::map &flags) // "cal_dC", "writable_Cws", "writable_dCws" + "writable_Vws", "writable_dVws" + const int it0, + const int it1, + const Abfs::Vector3_Order &R, + const std::map &flags) // "cal_dC", "writable_Cws", "writable_dCws" + "writable_Vws", "writable_dVws" { - using namespace LRI_CV_Tools; - - const Abfs::Vector3_Order Rm = -R; - pthread_rwlock_rdlock(&this->rwlock_Cw); - const RI::Tensor C_read = RI::Global_Func::find(this->Cws, it0, it1, R); - pthread_rwlock_unlock(&this->rwlock_Cw); - pthread_rwlock_rdlock(&this->rwlock_dCw); - const std::array,3> dC_read = RI::Global_Func::find(this->dCws, it0, it1, R); - pthread_rwlock_unlock(&this->rwlock_dCw); - const bool flag_finish_dC = (!flags.at("cal_dC")) || LRI_CV_Tools::exist(dC_read); - - if(!C_read.empty() && flag_finish_dC) - { - return std::make_pair(C_read, dC_read); - } - else - { - if( (ModuleBase::Vector3(0,0,0)==R) && (it0==it1) ) - { - const RI::Tensor - A = this->m_abfslcaos_lcaos.template cal_overlap_matrix( - it0, it1, {0,0,0}, {0,0,0}, - this->index_abfs, this->index_lcaos, this->index_lcaos, - Matrix_Orbs21::Matrix_Order::A1A2B); + using namespace LRI_CV_Tools; + + const Abfs::Vector3_Order Rm = -R; + pthread_rwlock_rdlock(&this->rwlock_Cw); + const RI::Tensor C_read = RI::Global_Func::find(this->Cws, it0, it1, R); + pthread_rwlock_unlock(&this->rwlock_Cw); + pthread_rwlock_rdlock(&this->rwlock_dCw); + const std::array,3> dC_read = RI::Global_Func::find(this->dCws, it0, it1, R); + pthread_rwlock_unlock(&this->rwlock_dCw); + const bool flag_finish_dC = (!flags.at("cal_dC")) || LRI_CV_Tools::exist(dC_read); + + if(!C_read.empty() && flag_finish_dC) + { + return std::make_pair(C_read, dC_read); + } + else + { + if( (ModuleBase::Vector3(0,0,0)==R) && (it0==it1) ) + { + const RI::Tensor + A = this->m_abfslcaos_lcaos.template cal_overlap_matrix( + it0, it1, {0,0,0}, {0,0,0}, + this->index_abfs, this->index_lcaos, this->index_lcaos, + Matrix_Orbs21::Matrix_Order::A1A2B); const RI::Tensor V = this->DPcal_V(it0, it0, {0, 0, 0}, {{"writable_Vws", true}}); RI::Tensor L; const double cs_inv_thr = this->p_info_ri != nullptr ? this->p_info_ri->Cs_inv_thr : 0.0; @@ -359,109 +359,109 @@ LRI_CV::DPcal_C_dC( else L = LRI_CV_Tools::cal_I(V); - const RI::Tensor C = RI::Global_Func::convert(0.5) * LRI_CV_Tools::mul1(L,A); // Attention 0.5! - if(flags.at("writable_Cws")) - { - pthread_rwlock_wrlock(&this->rwlock_Cw); - this->Cws[it0][it1][{0,0,0}] = C; - pthread_rwlock_unlock(&this->rwlock_Cw); - } - - if(flag_finish_dC) - { - return std::make_pair(C, dC_read); - } - else - { - const RI::Shape_Vector sizes = {this->index_abfs[it0].count_size, - this->index_lcaos[it0].count_size, - this->index_lcaos[it0].count_size}; - const std::array,3> - dC({RI::Tensor({sizes}), RI::Tensor({sizes}), RI::Tensor({sizes})}); - if(flags.at("writable_dCws")) - { - pthread_rwlock_wrlock(&this->rwlock_dCw); - this->dCws[it0][it1][{0,0,0}] = dC; - pthread_rwlock_unlock(&this->rwlock_dCw); - } - return std::make_pair(C, dC); - } - } // end if( (ModuleBase::Vector3(0,0,0)==R) && (it0==it1) ) - else - { - const std::vector> - A = {this->m_abfslcaos_lcaos.template cal_overlap_matrix( - it0, it1, {0,0,0}, R, - this->index_abfs, this->index_lcaos, this->index_lcaos, - Matrix_Orbs21::Matrix_Order::A1A2B), - this->m_abfslcaos_lcaos.template cal_overlap_matrix( - it1, it0, {0,0,0}, Rm, - this->index_abfs, this->index_lcaos, this->index_lcaos, - Matrix_Orbs21::Matrix_Order::A1BA2)}; - - const std::vector>> - V = {{DPcal_V(it0, it0, {0,0,0}, {{"writable_Vws",true}}), - DPcal_V(it0, it1, R, flags)}, - {DPcal_V(it1, it0, Rm, flags), - DPcal_V(it1, it1, {0,0,0}, {{"writable_Vws",true}})}}; + const RI::Tensor C = RI::Global_Func::convert(0.5) * LRI_CV_Tools::mul1(L,A); // Attention 0.5! + if(flags.at("writable_Cws")) + { + pthread_rwlock_wrlock(&this->rwlock_Cw); + this->Cws[it0][it1][{0,0,0}] = C; + pthread_rwlock_unlock(&this->rwlock_Cw); + } + + if(flag_finish_dC) + { + return std::make_pair(C, dC_read); + } + else + { + const RI::Shape_Vector sizes = {this->index_abfs[it0].count_size, + this->index_lcaos[it0].count_size, + this->index_lcaos[it0].count_size}; + const std::array,3> + dC({RI::Tensor({sizes}), RI::Tensor({sizes}), RI::Tensor({sizes})}); + if(flags.at("writable_dCws")) + { + pthread_rwlock_wrlock(&this->rwlock_dCw); + this->dCws[it0][it1][{0,0,0}] = dC; + pthread_rwlock_unlock(&this->rwlock_dCw); + } + return std::make_pair(C, dC); + } + } // end if( (ModuleBase::Vector3(0,0,0)==R) && (it0==it1) ) + else + { + const std::vector> + A = {this->m_abfslcaos_lcaos.template cal_overlap_matrix( + it0, it1, {0,0,0}, R, + this->index_abfs, this->index_lcaos, this->index_lcaos, + Matrix_Orbs21::Matrix_Order::A1A2B), + this->m_abfslcaos_lcaos.template cal_overlap_matrix( + it1, it0, {0,0,0}, Rm, + this->index_abfs, this->index_lcaos, this->index_lcaos, + Matrix_Orbs21::Matrix_Order::A1BA2)}; + + const std::vector>> + V = {{DPcal_V(it0, it0, {0,0,0}, {{"writable_Vws",true}}), + DPcal_V(it0, it1, R, flags)}, + {DPcal_V(it1, it0, Rm, flags), + DPcal_V(it1, it1, {0,0,0}, {{"writable_Vws",true}})}}; std::vector>> L; const double cs_inv_thr = this->p_info_ri != nullptr ? this->p_info_ri->Cs_inv_thr : 0.0; if (cs_inv_thr > 0) L = LRI_CV_Tools::cal_I(V, Inverse_Matrix::Method::syev, cs_inv_thr); else - L = LRI_CV_Tools::cal_I(V); - - const std::vector> C = LRI_CV_Tools::mul2(L,A); - if(flags.at("writable_Cws")) - { - pthread_rwlock_wrlock(&this->rwlock_Cw); - this->Cws[it0][it1][R] = C[0]; - this->Cws[it1][it0][Rm] = LRI_CV_Tools::transpose12(C[1]); - pthread_rwlock_unlock(&this->rwlock_Cw); - } - - if(flag_finish_dC) - { - return std::make_pair(C[0], dC_read); - } - else - { - const std::vector,3>> - dA = {this->m_abfslcaos_lcaos.template cal_grad_overlap_matrix( - it0, it1, {0,0,0}, R, - this->index_abfs, this->index_lcaos, this->index_lcaos, - Matrix_Orbs21::Matrix_Order::A1A2B), - LRI_CV_Tools::negative( - this->m_abfslcaos_lcaos.template cal_grad_overlap_matrix( - it1, it0, {0,0,0}, Rm, - this->index_abfs, this->index_lcaos, this->index_lcaos, - Matrix_Orbs21::Matrix_Order::A1BA2))}; - - const std::array,3> dV_01 = DPcal_dV(it0, it1, R, flags); - const std::array,3> dV_10 = LRI_CV_Tools::negative(DPcal_dV(it1, it0, Rm, flags)); - - std::array>,3> // dC = L*(dA-dV*C) - dC_tmp = LRI_CV_Tools::mul2( - L, - LRI_CV_Tools::change_order( LRI_CV_Tools::minus( - dA, - std::vector,3>>{ - LRI_CV_Tools::mul1(dV_01, C[1]), - LRI_CV_Tools::mul1(dV_10, C[0])}))); - const std::vector,3>> - dC = LRI_CV_Tools::change_order(std::move(dC_tmp)); - if(flags.at("writable_dCws")) - { - pthread_rwlock_wrlock(&this->rwlock_dCw); - this->dCws[it0][it1][R] = dC[0]; - this->dCws[it1][it0][Rm] = LRI_CV_Tools::negative(LRI_CV_Tools::transpose12(dC[1])); - pthread_rwlock_unlock(&this->rwlock_dCw); - } - return std::make_pair(C[0], dC[0]); - } // end else (!flag_finish_dC) - } // end else ( (ModuleBase::Vector3(0,0,0)!=R) || (it0!=it1) ) - } // end else (!(C_read && flag_finish_dC)) + L = LRI_CV_Tools::cal_I(V); + + const std::vector> C = LRI_CV_Tools::mul2(L,A); + if(flags.at("writable_Cws")) + { + pthread_rwlock_wrlock(&this->rwlock_Cw); + this->Cws[it0][it1][R] = C[0]; + this->Cws[it1][it0][Rm] = LRI_CV_Tools::transpose12(C[1]); + pthread_rwlock_unlock(&this->rwlock_Cw); + } + + if(flag_finish_dC) + { + return std::make_pair(C[0], dC_read); + } + else + { + const std::vector,3>> + dA = {this->m_abfslcaos_lcaos.template cal_grad_overlap_matrix( + it0, it1, {0,0,0}, R, + this->index_abfs, this->index_lcaos, this->index_lcaos, + Matrix_Orbs21::Matrix_Order::A1A2B), + LRI_CV_Tools::negative( + this->m_abfslcaos_lcaos.template cal_grad_overlap_matrix( + it1, it0, {0,0,0}, Rm, + this->index_abfs, this->index_lcaos, this->index_lcaos, + Matrix_Orbs21::Matrix_Order::A1BA2))}; + + const std::array,3> dV_01 = DPcal_dV(it0, it1, R, flags); + const std::array,3> dV_10 = LRI_CV_Tools::negative(DPcal_dV(it1, it0, Rm, flags)); + + std::array>,3> // dC = L*(dA-dV*C) + dC_tmp = LRI_CV_Tools::mul2( + L, + LRI_CV_Tools::change_order( LRI_CV_Tools::minus( + dA, + std::vector,3>>{ + LRI_CV_Tools::mul1(dV_01, C[1]), + LRI_CV_Tools::mul1(dV_10, C[0])}))); + const std::vector,3>> + dC = LRI_CV_Tools::change_order(std::move(dC_tmp)); + if(flags.at("writable_dCws")) + { + pthread_rwlock_wrlock(&this->rwlock_dCw); + this->dCws[it0][it1][R] = dC[0]; + this->dCws[it1][it0][Rm] = LRI_CV_Tools::negative(LRI_CV_Tools::transpose12(dC[1])); + pthread_rwlock_unlock(&this->rwlock_dCw); + } + return std::make_pair(C[0], dC[0]); + } // end else (!flag_finish_dC) + } // end else ( (ModuleBase::Vector3(0,0,0)!=R) || (it0!=it1) ) + } // end else (!(C_read && flag_finish_dC)) } diff --git a/source/source_lcao/module_ri/lri_cv_tools.hpp b/source/source_lcao/module_ri/lri_cv_tools.hpp index 37b35445c60..1f96ec8c33d 100644 --- a/source/source_lcao/module_ri/lri_cv_tools.hpp +++ b/source/source_lcao/module_ri/lri_cv_tools.hpp @@ -365,46 +365,46 @@ std::map>> template std::array LRI_CV_Tools::cal_latvec_range(const double& rcut_times, - const UnitCell &ucell, - const std::vector& orb_cutoff) { + const UnitCell &ucell, + const std::vector& orb_cutoff) { double Rcut_max = 0; for(int T=0; T proj = ModuleBase::Mathzone::latvec_projection( - std::array,3>{ucell.a1, ucell.a2, ucell.a3}); - const ModuleBase::Vector3 latvec_times = Rcut_max * rcut_times / (proj * ucell.lat0); + Rcut_max = std::max(Rcut_max, orb_cutoff[T]); + const ModuleBase::Vector3 proj = ModuleBase::Mathzone::latvec_projection( + std::array,3>{ucell.a1, ucell.a2, ucell.a3}); + const ModuleBase::Vector3 latvec_times = Rcut_max * rcut_times / (proj * ucell.lat0); const ModuleBase::Vector3 latvec_times_ceil = {static_cast(std::ceil(latvec_times.x)), static_cast(std::ceil(latvec_times.y)), static_cast(std::ceil(latvec_times.z))}; const ModuleBase::Vector3 period = 2 * latvec_times_ceil + ModuleBase::Vector3{1,1,1}; - return std::array{period.x, period.y, period.z}; + return std::array{period.x, period.y, period.z}; } template std::map,RI::Tensor>>> LRI_CV_Tools::get_CVws( - const UnitCell &ucell, - const std::map>,RI::Tensor>> &CVs) + const UnitCell &ucell, + const std::map>,RI::Tensor>> &CVs) { - std::map,RI::Tensor>>> CVws; - for(const auto &CVs_A : CVs) - { - const TA iat0 = CVs_A.first; - const int it0 = ucell.iat2it[iat0]; - const int ia0 = ucell.iat2ia[iat0]; - const ModuleBase::Vector3 tau0 = ucell.atoms[it0].tau[ia0]; - for(const auto &CVs_B : CVs_A.second) - { - const TA iat1 = CVs_B.first.first; - const int it1 = ucell.iat2it[iat1]; - const int ia1 = ucell.iat2ia[iat1]; - const std::array &cell1 = CVs_B.first.second; - const ModuleBase::Vector3 tau1 = ucell.atoms[it1].tau[ia1]; - const Abfs::Vector3_Order R_delta = -tau0+tau1+(RI_Util::array3_to_Vector3(cell1)*ucell.latvec); - CVws[it0][it1][R_delta] = CVs_B.second; - } - } - return CVws; + std::map,RI::Tensor>>> CVws; + for(const auto &CVs_A : CVs) + { + const TA iat0 = CVs_A.first; + const int it0 = ucell.iat2it[iat0]; + const int ia0 = ucell.iat2ia[iat0]; + const ModuleBase::Vector3 tau0 = ucell.atoms[it0].tau[ia0]; + for(const auto &CVs_B : CVs_A.second) + { + const TA iat1 = CVs_B.first.first; + const int it1 = ucell.iat2it[iat1]; + const int ia1 = ucell.iat2ia[iat1]; + const std::array &cell1 = CVs_B.first.second; + const ModuleBase::Vector3 tau1 = ucell.atoms[it1].tau[ia1]; + const Abfs::Vector3_Order R_delta = -tau0+tau1+(RI_Util::array3_to_Vector3(cell1)*ucell.latvec); + CVws[it0][it1][R_delta] = CVs_B.second; + } + } + return CVws; } template @@ -521,42 +521,42 @@ std::array, N> template std::array,RI::Tensor>>,3>,3> LRI_CV_Tools::cal_dMRs( - const UnitCell &ucell, - const std::array,RI::Tensor>>,3> &dMs) + const UnitCell &ucell, + const std::array,RI::Tensor>>,3> &dMs) { - auto get_R_delta = [&](const TA &iat0, const std::pair &A1) -> std::array - { - const TA iat1 = A1.first; - const TC &cell1 = A1.second; - const int it0 = ucell.iat2it[iat0]; - const int ia0 = ucell.iat2ia[iat0]; - const int it1 = ucell.iat2it[iat1]; - const int ia1 = ucell.iat2ia[iat1]; - const ModuleBase::Vector3 tau0 = ucell.atoms[it0].tau[ia0]; - const ModuleBase::Vector3 tau1 = ucell.atoms[it1].tau[ia1]; - const Abfs::Vector3_Order R_delta = -tau0+tau1+(RI_Util::array3_to_Vector3(cell1)*ucell.latvec); - return std::array{R_delta.x, R_delta.y, R_delta.z}; - }; - constexpr int Npos = 3; - std::array,RI::Tensor>>,Npos>,Npos> dMRs; - for(int ipos0=0; ipos0 A1 = dMs_B.first; - const RI::Tensor &dM = dMs_B.second; - const std::array R_delta = get_R_delta(iat0, A1); - dMRs[ipos0][ipos1][iat0][A1] = dM * R_delta[ipos1]; - } - } - } - } - return dMRs; + auto get_R_delta = [&](const TA &iat0, const std::pair &A1) -> std::array + { + const TA iat1 = A1.first; + const TC &cell1 = A1.second; + const int it0 = ucell.iat2it[iat0]; + const int ia0 = ucell.iat2ia[iat0]; + const int it1 = ucell.iat2it[iat1]; + const int ia1 = ucell.iat2ia[iat1]; + const ModuleBase::Vector3 tau0 = ucell.atoms[it0].tau[ia0]; + const ModuleBase::Vector3 tau1 = ucell.atoms[it1].tau[ia1]; + const Abfs::Vector3_Order R_delta = -tau0+tau1+(RI_Util::array3_to_Vector3(cell1)*ucell.latvec); + return std::array{R_delta.x, R_delta.y, R_delta.z}; + }; + constexpr int Npos = 3; + std::array,RI::Tensor>>,Npos>,Npos> dMRs; + for(int ipos0=0; ipos0 A1 = dMs_B.first; + const RI::Tensor &dM = dMs_B.second; + const std::array R_delta = get_R_delta(iat0, A1); + dMRs[ipos0][ipos1][iat0][A1] = dM * R_delta[ipos1]; + } + } + } + } + return dMRs; } #endif diff --git a/source/source_lcao/module_ri/matrix_orbs11.hpp b/source/source_lcao/module_ri/matrix_orbs11.hpp index 8a0d553d94c..2d6dff345f7 100644 --- a/source/source_lcao/module_ri/matrix_orbs11.hpp +++ b/source/source_lcao/module_ri/matrix_orbs11.hpp @@ -11,150 +11,150 @@ template RI::Tensor Matrix_Orbs11::cal_overlap_matrix( - const size_t TA, - const size_t TB, - const ModuleBase::Vector3 &tauA, - const ModuleBase::Vector3 &tauB, - const ModuleBase::Element_Basis_Index::IndexLNM &index_A, - const ModuleBase::Element_Basis_Index::IndexLNM &index_B, - const Matrix_Order &matrix_order) const + const size_t TA, + const size_t TB, + const ModuleBase::Vector3 &tauA, + const ModuleBase::Vector3 &tauB, + const ModuleBase::Element_Basis_Index::IndexLNM &index_A, + const ModuleBase::Element_Basis_Index::IndexLNM &index_B, + const Matrix_Order &matrix_order) const { - RI::Tensor m; - const double lat0 = *this->lat0; - const size_t sizeA = index_A[TA].count_size; - const size_t sizeB = index_B[TB].count_size; - switch(matrix_order) - { - case Matrix_Order::AB: m = RI::Tensor({sizeA, sizeB}); break; - case Matrix_Order::BA: m = RI::Tensor({sizeB, sizeA}); break; - default: throw std::invalid_argument(std::string(__FILE__)+" line "+std::to_string(__LINE__)); - } + RI::Tensor m; + const double lat0 = *this->lat0; + const size_t sizeA = index_A[TA].count_size; + const size_t sizeB = index_B[TB].count_size; + switch(matrix_order) + { + case Matrix_Order::AB: m = RI::Tensor({sizeA, sizeB}); break; + case Matrix_Order::BA: m = RI::Tensor({sizeB, sizeA}); break; + default: throw std::invalid_argument(std::string(__FILE__)+" line "+std::to_string(__LINE__)); + } - for( const auto &co3 : center2_orb11_s.at(TA).at(TB) ) - { - const int LA = co3.first; - for( const auto &co4 : co3.second ) - { - const size_t NA = co4.first; - for( size_t MA=0; MA!=2*LA+1; ++MA ) - { - for( const auto &co5 : co4.second ) - { - const int LB = co5.first; - for( const auto &co6 : co5.second ) - { - const size_t NB = co6.first; - for( size_t MB=0; MB!=2*LB+1; ++MB ) - { - const Tdata overlap = co6.second.cal_overlap( tauA*lat0, tauB*lat0, MA, MB ); - const size_t iA = index_A[TA][LA][NA][MA]; - const size_t iB = index_B[TB][LB][NB][MB]; - switch(matrix_order) - { - case Matrix_Order::AB: m(iA,iB) = overlap; break; - case Matrix_Order::BA: m(iB,iA) = overlap; break; - default: throw std::invalid_argument(std::string(__FILE__)+" line "+std::to_string(__LINE__)); - } - } - } - } - } - } - } - return m; + for( const auto &co3 : center2_orb11_s.at(TA).at(TB) ) + { + const int LA = co3.first; + for( const auto &co4 : co3.second ) + { + const size_t NA = co4.first; + for( size_t MA=0; MA!=2*LA+1; ++MA ) + { + for( const auto &co5 : co4.second ) + { + const int LB = co5.first; + for( const auto &co6 : co5.second ) + { + const size_t NB = co6.first; + for( size_t MB=0; MB!=2*LB+1; ++MB ) + { + const Tdata overlap = co6.second.cal_overlap( tauA*lat0, tauB*lat0, MA, MB ); + const size_t iA = index_A[TA][LA][NA][MA]; + const size_t iB = index_B[TB][LB][NB][MB]; + switch(matrix_order) + { + case Matrix_Order::AB: m(iA,iB) = overlap; break; + case Matrix_Order::BA: m(iB,iA) = overlap; break; + default: throw std::invalid_argument(std::string(__FILE__)+" line "+std::to_string(__LINE__)); + } + } + } + } + } + } + } + return m; } template std::array,3> Matrix_Orbs11::cal_grad_overlap_matrix( - const size_t TA, - const size_t TB, - const ModuleBase::Vector3 &tauA, - const ModuleBase::Vector3 &tauB, - const ModuleBase::Element_Basis_Index::IndexLNM &index_A, - const ModuleBase::Element_Basis_Index::IndexLNM &index_B, - const Matrix_Order &matrix_order) const + const size_t TA, + const size_t TB, + const ModuleBase::Vector3 &tauA, + const ModuleBase::Vector3 &tauB, + const ModuleBase::Element_Basis_Index::IndexLNM &index_A, + const ModuleBase::Element_Basis_Index::IndexLNM &index_B, + const Matrix_Order &matrix_order) const { - std::array,3> m; - const double lat0 = *this->lat0; - const size_t sizeA = index_A[TA].count_size; - const size_t sizeB = index_B[TB].count_size; - for(int i=0; i({sizeA, sizeB}); break; - case Matrix_Order::BA: m[i] = RI::Tensor({sizeB, sizeA}); break; - default: throw std::invalid_argument(std::string(__FILE__)+" line "+std::to_string(__LINE__)); - } - } + std::array,3> m; + const double lat0 = *this->lat0; + const size_t sizeA = index_A[TA].count_size; + const size_t sizeB = index_B[TB].count_size; + for(int i=0; i({sizeA, sizeB}); break; + case Matrix_Order::BA: m[i] = RI::Tensor({sizeB, sizeA}); break; + default: throw std::invalid_argument(std::string(__FILE__)+" line "+std::to_string(__LINE__)); + } + } - for( const auto &co3 : center2_orb11_s.at(TA).at(TB) ) - { - const int LA = co3.first; - for( const auto &co4 : co3.second ) - { - const size_t NA = co4.first; - for( size_t MA=0; MA!=2*LA+1; ++MA ) - { - for( const auto &co5 : co4.second ) - { - const int LB = co5.first; - for( const auto &co6 : co5.second ) - { - const size_t NB = co6.first; - for( size_t MB=0; MB!=2*LB+1; ++MB ) - { - const std::array grad_overlap = RI_Util::Vector3_to_array3(co6.second.cal_grad_overlap( tauA*lat0, tauB*lat0, MA, MB )); - const size_t iA = index_A[TA][LA][NA][MA]; - const size_t iB = index_B[TB][LB][NB][MB]; - for(size_t i=0; i grad_overlap = RI_Util::Vector3_to_array3(co6.second.cal_grad_overlap( tauA*lat0, tauB*lat0, MA, MB )); + const size_t iA = index_A[TA][LA][NA][MA]; + const size_t iB = index_B[TB][LB][NB][MB]; + for(size_t i=0; i std::map>>>> Matrix_Orbs11::cal_overlap_matrix_all( - const UnitCell &ucell, - const ModuleBase::Element_Basis_Index::IndexLNM &index_r, - const ModuleBase::Element_Basis_Index::IndexLNM &index_c ) const + const UnitCell &ucell, + const ModuleBase::Element_Basis_Index::IndexLNM &index_r, + const ModuleBase::Element_Basis_Index::IndexLNM &index_c ) const { - ModuleBase::TITLE("Matrix_Orbs11","cal_overlap_matrix"); + ModuleBase::TITLE("Matrix_Orbs11","cal_overlap_matrix"); - std::map>>>> matrixes; + std::map>>>> matrixes; - for( const auto &co1 : center2_orb11_s ) - { - const size_t TA = co1.first; - for (size_t IA=0; IA!=ucell.atoms[TA].na; ++IA) - { - const ModuleBase::Vector3 &tauA( ucell.atoms[TA].tau[IA] ); + for( const auto &co1 : center2_orb11_s ) + { + const size_t TA = co1.first; + for (size_t IA=0; IA!=ucell.atoms[TA].na; ++IA) + { + const ModuleBase::Vector3 &tauA( ucell.atoms[TA].tau[IA] ); - for( const auto &co2 : co1.second ) - { - const size_t TB = co2.first; - for (size_t IB=0; IB!=ucell.atoms[TB].na; ++IB) - { - const ModuleBase::Vector3 &tauB( ucell.atoms[TB].tau[IB] ); + for( const auto &co2 : co1.second ) + { + const size_t TB = co2.first; + for (size_t IB=0; IB!=ucell.atoms[TB].na; ++IB) + { + const ModuleBase::Vector3 &tauB( ucell.atoms[TB].tau[IB] ); - matrixes[TA][IA][TB][IB] = cal_overlap_matrix( TA, TB, tauA, tauB, index_r, index_c, Matrix_Order::AB ); - } - } - } - } - return matrixes; + matrixes[TA][IA][TB][IB] = cal_overlap_matrix( TA, TB, tauA, tauB, index_r, index_c, Matrix_Order::AB ); + } + } + } + } + return matrixes; } #endif diff --git a/source/source_lcao/module_ri/matrix_orbs21.hpp b/source/source_lcao/module_ri/matrix_orbs21.hpp index 11018b8bb74..b6f464d7440 100644 --- a/source/source_lcao/module_ri/matrix_orbs21.hpp +++ b/source/source_lcao/module_ri/matrix_orbs21.hpp @@ -11,210 +11,210 @@ template RI::Tensor Matrix_Orbs21::cal_overlap_matrix( - const size_t TA, - const size_t TB, - const ModuleBase::Vector3 &tauA, - const ModuleBase::Vector3 &tauB, - const ModuleBase::Element_Basis_Index::IndexLNM &index_A1, - const ModuleBase::Element_Basis_Index::IndexLNM &index_A2, - const ModuleBase::Element_Basis_Index::IndexLNM &index_B, - const Matrix_Order &matrix_order) const + const size_t TA, + const size_t TB, + const ModuleBase::Vector3 &tauA, + const ModuleBase::Vector3 &tauB, + const ModuleBase::Element_Basis_Index::IndexLNM &index_A1, + const ModuleBase::Element_Basis_Index::IndexLNM &index_A2, + const ModuleBase::Element_Basis_Index::IndexLNM &index_B, + const Matrix_Order &matrix_order) const { - RI::Tensor m; - const double lat0 = *this->lat0; - const size_t sizeA1 = index_A1[TA].count_size; - const size_t sizeA2 = index_A2[TA].count_size; - const size_t sizeB = index_B[TB].count_size; - switch(matrix_order) - { - case Matrix_Order::A1A2B: m = RI::Tensor({sizeA1, sizeA2, sizeB}); break; - case Matrix_Order::A1BA2: m = RI::Tensor({sizeA1, sizeB, sizeA2}); break; - case Matrix_Order::BA1A2: m = RI::Tensor({sizeB, sizeA1, sizeA2}); break; - case Matrix_Order::BA2A1: m = RI::Tensor({sizeB, sizeA2, sizeA1}); break; - case Matrix_Order::A2A1B: m = RI::Tensor({sizeA2, sizeA1, sizeB}); break; - case Matrix_Order::A2BA1: m = RI::Tensor({sizeA2, sizeB, sizeA1}); break; - default: throw std::invalid_argument(std::string(__FILE__)+" line "+std::to_string(__LINE__)); - } + RI::Tensor m; + const double lat0 = *this->lat0; + const size_t sizeA1 = index_A1[TA].count_size; + const size_t sizeA2 = index_A2[TA].count_size; + const size_t sizeB = index_B[TB].count_size; + switch(matrix_order) + { + case Matrix_Order::A1A2B: m = RI::Tensor({sizeA1, sizeA2, sizeB}); break; + case Matrix_Order::A1BA2: m = RI::Tensor({sizeA1, sizeB, sizeA2}); break; + case Matrix_Order::BA1A2: m = RI::Tensor({sizeB, sizeA1, sizeA2}); break; + case Matrix_Order::BA2A1: m = RI::Tensor({sizeB, sizeA2, sizeA1}); break; + case Matrix_Order::A2A1B: m = RI::Tensor({sizeA2, sizeA1, sizeB}); break; + case Matrix_Order::A2BA1: m = RI::Tensor({sizeA2, sizeB, sizeA1}); break; + default: throw std::invalid_argument(std::string(__FILE__)+" line "+std::to_string(__LINE__)); + } - for( const auto &co3 : center2_orb21_s.at(TA).at(TB) ) - { - const int LA1 = co3.first; - for( const auto &co4 : co3.second ) - { - const size_t NA1 = co4.first; - for( size_t MA1=0; MA1!=2*LA1+1; ++MA1 ) - { - for( const auto &co5 : co4.second ) - { - const int LA2 = co5.first; - for( const auto &co6 : co5.second ) - { - const size_t NA2 = co6.first; - for( size_t MA2=0; MA2!=2*LA2+1; ++MA2 ) - { - for( const auto &co7 : co6.second ) - { - const int LB = co7.first; - for( const auto &co8 : co7.second ) - { - const size_t NB = co8.first; - for( size_t MB=0; MB!=2*LB+1; ++MB ) - { - const Tdata overlap = co8.second.cal_overlap( tauA*lat0, tauB*lat0, MA1, MA2, MB ); - const size_t iA1 = index_A1[TA][LA1][NA1][MA1]; - const size_t iA2 = index_A2[TA][LA2][NA2][MA2]; - const size_t iB = index_B[TB][LB][NB][MB]; - switch(matrix_order) - { - case Matrix_Order::A1A2B: m(iA1,iA2,iB) = overlap; break; - case Matrix_Order::A1BA2: m(iA1,iB,iA2) = overlap; break; - case Matrix_Order::A2A1B: m(iA2,iA1,iB) = overlap; break; - case Matrix_Order::A2BA1: m(iA2,iB,iA1) = overlap; break; - case Matrix_Order::BA1A2: m(iB,iA1,iA2) = overlap; break; - case Matrix_Order::BA2A1: m(iB,iA2,iA1) = overlap; break; - default: throw std::invalid_argument(std::string(__FILE__)+" line "+std::to_string(__LINE__)); - } - } - } - } - } - } - } - } - } - } - return m; + for( const auto &co3 : center2_orb21_s.at(TA).at(TB) ) + { + const int LA1 = co3.first; + for( const auto &co4 : co3.second ) + { + const size_t NA1 = co4.first; + for( size_t MA1=0; MA1!=2*LA1+1; ++MA1 ) + { + for( const auto &co5 : co4.second ) + { + const int LA2 = co5.first; + for( const auto &co6 : co5.second ) + { + const size_t NA2 = co6.first; + for( size_t MA2=0; MA2!=2*LA2+1; ++MA2 ) + { + for( const auto &co7 : co6.second ) + { + const int LB = co7.first; + for( const auto &co8 : co7.second ) + { + const size_t NB = co8.first; + for( size_t MB=0; MB!=2*LB+1; ++MB ) + { + const Tdata overlap = co8.second.cal_overlap( tauA*lat0, tauB*lat0, MA1, MA2, MB ); + const size_t iA1 = index_A1[TA][LA1][NA1][MA1]; + const size_t iA2 = index_A2[TA][LA2][NA2][MA2]; + const size_t iB = index_B[TB][LB][NB][MB]; + switch(matrix_order) + { + case Matrix_Order::A1A2B: m(iA1,iA2,iB) = overlap; break; + case Matrix_Order::A1BA2: m(iA1,iB,iA2) = overlap; break; + case Matrix_Order::A2A1B: m(iA2,iA1,iB) = overlap; break; + case Matrix_Order::A2BA1: m(iA2,iB,iA1) = overlap; break; + case Matrix_Order::BA1A2: m(iB,iA1,iA2) = overlap; break; + case Matrix_Order::BA2A1: m(iB,iA2,iA1) = overlap; break; + default: throw std::invalid_argument(std::string(__FILE__)+" line "+std::to_string(__LINE__)); + } + } + } + } + } + } + } + } + } + } + return m; } template std::array,3> Matrix_Orbs21::cal_grad_overlap_matrix( - const size_t TA, - const size_t TB, - const ModuleBase::Vector3 &tauA, - const ModuleBase::Vector3 &tauB, - const ModuleBase::Element_Basis_Index::IndexLNM &index_A1, - const ModuleBase::Element_Basis_Index::IndexLNM &index_A2, - const ModuleBase::Element_Basis_Index::IndexLNM &index_B, - const Matrix_Order &matrix_order) const + const size_t TA, + const size_t TB, + const ModuleBase::Vector3 &tauA, + const ModuleBase::Vector3 &tauB, + const ModuleBase::Element_Basis_Index::IndexLNM &index_A1, + const ModuleBase::Element_Basis_Index::IndexLNM &index_A2, + const ModuleBase::Element_Basis_Index::IndexLNM &index_B, + const Matrix_Order &matrix_order) const { - std::array,3> m; - const double lat0 = *this->lat0; - const size_t sizeA1 = index_A1[TA].count_size; - const size_t sizeA2 = index_A2[TA].count_size; - const size_t sizeB = index_B[TB].count_size; - for(int i=0; i({sizeA1, sizeA2, sizeB}); break; - case Matrix_Order::A1BA2: m[i] = RI::Tensor({sizeA1, sizeB, sizeA2}); break; - case Matrix_Order::BA1A2: m[i] = RI::Tensor({sizeB, sizeA1, sizeA2}); break; - case Matrix_Order::BA2A1: m[i] = RI::Tensor({sizeB, sizeA2, sizeA1}); break; - case Matrix_Order::A2A1B: m[i] = RI::Tensor({sizeA2, sizeA1, sizeB}); break; - case Matrix_Order::A2BA1: m[i] = RI::Tensor({sizeA2, sizeB, sizeA1}); break; - default: throw std::invalid_argument(std::string(__FILE__)+" line "+std::to_string(__LINE__)); - } - } + std::array,3> m; + const double lat0 = *this->lat0; + const size_t sizeA1 = index_A1[TA].count_size; + const size_t sizeA2 = index_A2[TA].count_size; + const size_t sizeB = index_B[TB].count_size; + for(int i=0; i({sizeA1, sizeA2, sizeB}); break; + case Matrix_Order::A1BA2: m[i] = RI::Tensor({sizeA1, sizeB, sizeA2}); break; + case Matrix_Order::BA1A2: m[i] = RI::Tensor({sizeB, sizeA1, sizeA2}); break; + case Matrix_Order::BA2A1: m[i] = RI::Tensor({sizeB, sizeA2, sizeA1}); break; + case Matrix_Order::A2A1B: m[i] = RI::Tensor({sizeA2, sizeA1, sizeB}); break; + case Matrix_Order::A2BA1: m[i] = RI::Tensor({sizeA2, sizeB, sizeA1}); break; + default: throw std::invalid_argument(std::string(__FILE__)+" line "+std::to_string(__LINE__)); + } + } - for( const auto &co3 : center2_orb21_s.at(TA).at(TB) ) - { - const int LA1 = co3.first; - for( const auto &co4 : co3.second ) - { - const size_t NA1 = co4.first; - for( size_t MA1=0; MA1!=2*LA1+1; ++MA1 ) - { - for( const auto &co5 : co4.second ) - { - const int LA2 = co5.first; - for( const auto &co6 : co5.second ) - { - const size_t NA2 = co6.first; - for( size_t MA2=0; MA2!=2*LA2+1; ++MA2 ) - { - for( const auto &co7 : co6.second ) - { - const int LB = co7.first; - for( const auto &co8 : co7.second ) - { - const size_t NB = co8.first; - for( size_t MB=0; MB!=2*LB+1; ++MB ) - { - const std::array grad_overlap = RI_Util::Vector3_to_array3(co8.second.cal_grad_overlap( tauA*lat0, tauB*lat0, MA1, MA2, MB )); - const size_t iA1 = index_A1[TA][LA1][NA1][MA1]; - const size_t iA2 = index_A2[TA][LA2][NA2][MA2]; - const size_t iB = index_B[TB][LB][NB][MB]; - for(size_t i=0; i grad_overlap = RI_Util::Vector3_to_array3(co8.second.cal_grad_overlap( tauA*lat0, tauB*lat0, MA1, MA2, MB )); + const size_t iA1 = index_A1[TA][LA1][NA1][MA1]; + const size_t iA2 = index_A2[TA][LA2][NA2][MA2]; + const size_t iB = index_B[TB][LB][NB][MB]; + for(size_t i=0; i std::map>>>>> Matrix_Orbs21::cal_overlap_matrix_all( - const UnitCell &ucell, - const ModuleBase::Element_Basis_Index::IndexLNM &index_A1, - const ModuleBase::Element_Basis_Index::IndexLNM &index_A2, - const ModuleBase::Element_Basis_Index::IndexLNM &index_B) const + const UnitCell &ucell, + const ModuleBase::Element_Basis_Index::IndexLNM &index_A1, + const ModuleBase::Element_Basis_Index::IndexLNM &index_A2, + const ModuleBase::Element_Basis_Index::IndexLNM &index_B) const { - ModuleBase::TITLE("Matrix_Orbs21","cal_overlap_matrix"); + ModuleBase::TITLE("Matrix_Orbs21","cal_overlap_matrix"); - std::map>>>>> matrixes; + std::map>>>>> matrixes; - for( const auto &co1 : center2_orb21_s ) - { - const size_t TA = co1.first; - for( size_t IA=0; IA!=ucell.atoms[TA].na; ++IA ) - { - const ModuleBase::Vector3 &tauA( ucell.atoms[TA].tau[IA] ); + for( const auto &co1 : center2_orb21_s ) + { + const size_t TA = co1.first; + for( size_t IA=0; IA!=ucell.atoms[TA].na; ++IA ) + { + const ModuleBase::Vector3 &tauA( ucell.atoms[TA].tau[IA] ); - for( const auto &co2 : co1.second ) - { - const size_t TB = co2.first; - for( size_t IB=0; IB!=ucell.atoms[TB].na; ++IB ) - { - const ModuleBase::Vector3 &tauB( ucell.atoms[TB].tau[IB] ); + for( const auto &co2 : co1.second ) + { + const size_t TB = co2.first; + for( size_t IB=0; IB!=ucell.atoms[TB].na; ++IB ) + { + const ModuleBase::Vector3 &tauB( ucell.atoms[TB].tau[IB] ); - const RI::Tensor &&m = cal_overlap_matrix( TA, TB, tauA, tauB, index_A1, index_A2, index_B, Matrix_Order::A2BA1 ); - matrixes[TA][IA][TB][IB].resize(2); - matrixes[TA][IA][TB][IB][0] = std::move(m); - const RI::Tensor &&n = cal_overlap_matrix( TA, TB, tauA, tauB, index_A1, index_A2, index_B, Matrix_Order::BA2A1 ); - matrixes[TB][IB][TA][IA].resize(2); - matrixes[TB][IB][TA][IA][1] = std::move(n); + const RI::Tensor &&m = cal_overlap_matrix( TA, TB, tauA, tauB, index_A1, index_A2, index_B, Matrix_Order::A2BA1 ); + matrixes[TA][IA][TB][IB].resize(2); + matrixes[TA][IA][TB][IB][0] = std::move(m); + const RI::Tensor &&n = cal_overlap_matrix( TA, TB, tauA, tauB, index_A1, index_A2, index_B, Matrix_Order::BA2A1 ); + matrixes[TB][IB][TA][IA].resize(2); + matrixes[TB][IB][TA][IA][1] = std::move(n); } - } - } + } + } } - // matrixes[T][I][T][I][0] = matrixes[T][I][T][I][1], so delete repeat + // matrixes[T][I][T][I][0] = matrixes[T][I][T][I][1], so delete repeat for (auto m1 : matrixes) - { - const size_t T = m1.first; - for( auto m2 : m1.second ) - { - const size_t I = m2.first; - matrixes[T][I][T][I].resize(1); - } - } + { + const size_t T = m1.first; + for( auto m2 : m1.second ) + { + const size_t I = m2.first; + matrixes[T][I][T][I].resize(1); + } + } - return matrixes; + return matrixes; } #endif diff --git a/source/source_lcao/module_ri/matrix_orbs22.hpp b/source/source_lcao/module_ri/matrix_orbs22.hpp index 8df841616d8..972f61e0add 100644 --- a/source/source_lcao/module_ri/matrix_orbs22.hpp +++ b/source/source_lcao/module_ri/matrix_orbs22.hpp @@ -11,302 +11,302 @@ template RI::Tensor Matrix_Orbs22::cal_overlap_matrix( - const size_t TA, - const size_t TB, - const ModuleBase::Vector3 &tauA, - const ModuleBase::Vector3 &tauB, - const ModuleBase::Element_Basis_Index::IndexLNM &index_A1, - const ModuleBase::Element_Basis_Index::IndexLNM &index_A2, - const ModuleBase::Element_Basis_Index::IndexLNM &index_B1, - const ModuleBase::Element_Basis_Index::IndexLNM &index_B2, - const Matrix_Order &matrix_order) const + const size_t TA, + const size_t TB, + const ModuleBase::Vector3 &tauA, + const ModuleBase::Vector3 &tauB, + const ModuleBase::Element_Basis_Index::IndexLNM &index_A1, + const ModuleBase::Element_Basis_Index::IndexLNM &index_A2, + const ModuleBase::Element_Basis_Index::IndexLNM &index_B1, + const ModuleBase::Element_Basis_Index::IndexLNM &index_B2, + const Matrix_Order &matrix_order) const { - const double lat0 = *this->lat0; - RI::Tensor m; - const size_t sizeA1 = index_A1[TA].count_size; - const size_t sizeA2 = index_A2[TA].count_size; - const size_t sizeB1 = index_B1[TB].count_size; - const size_t sizeB2 = index_B2[TB].count_size; - switch(matrix_order) - { - case Matrix_Order::A1A2B1B2: m = RI::Tensor({sizeA1, sizeA2, sizeB1, sizeB2}); break; - case Matrix_Order::A1A2B2B1: m = RI::Tensor({sizeA1, sizeA2, sizeB2, sizeB1}); break; - case Matrix_Order::A1B1A2B2: m = RI::Tensor({sizeA1, sizeB1, sizeA2, sizeB2}); break; - case Matrix_Order::A1B1B2A2: m = RI::Tensor({sizeA1, sizeB1, sizeB2, sizeA2}); break; - case Matrix_Order::A1B2A2B1: m = RI::Tensor({sizeA1, sizeB2, sizeA2, sizeB1}); break; - case Matrix_Order::A1B2B1A2: m = RI::Tensor({sizeA1, sizeB2, sizeB1, sizeA2}); break; - case Matrix_Order::A2A1B1B2: m = RI::Tensor({sizeA2, sizeA1, sizeB1, sizeB2}); break; - case Matrix_Order::A2A1B2B1: m = RI::Tensor({sizeA2, sizeA1, sizeB2, sizeB1}); break; - case Matrix_Order::A2B1A1B2: m = RI::Tensor({sizeA2, sizeB1, sizeA1, sizeB2}); break; - case Matrix_Order::A2B1B2A1: m = RI::Tensor({sizeA2, sizeB1, sizeB2, sizeA1}); break; - case Matrix_Order::A2B2A1B1: m = RI::Tensor({sizeA2, sizeB2, sizeA1, sizeB1}); break; - case Matrix_Order::A2B2B1A1: m = RI::Tensor({sizeA2, sizeB2, sizeB1, sizeA1}); break; - case Matrix_Order::B1A1A2B2: m = RI::Tensor({sizeB1, sizeA1, sizeA2, sizeB2}); break; - case Matrix_Order::B1A1B2A2: m = RI::Tensor({sizeB1, sizeA1, sizeB2, sizeA2}); break; - case Matrix_Order::B1A2A1B2: m = RI::Tensor({sizeB1, sizeA2, sizeA1, sizeB2}); break; - case Matrix_Order::B1A2B2A1: m = RI::Tensor({sizeB1, sizeA2, sizeB2, sizeA1}); break; - case Matrix_Order::B1B2A1A2: m = RI::Tensor({sizeB1, sizeB2, sizeA1, sizeA2}); break; - case Matrix_Order::B1B2A2A1: m = RI::Tensor({sizeB1, sizeB2, sizeA2, sizeA1}); break; - case Matrix_Order::B2A1A2B1: m = RI::Tensor({sizeB2, sizeA1, sizeA2, sizeB1}); break; - case Matrix_Order::B2A1B1A2: m = RI::Tensor({sizeB2, sizeA1, sizeB1, sizeA2}); break; - case Matrix_Order::B2A2A1B1: m = RI::Tensor({sizeB2, sizeA2, sizeA1, sizeB1}); break; - case Matrix_Order::B2A2B1A1: m = RI::Tensor({sizeB2, sizeA2, sizeB1, sizeA1}); break; - case Matrix_Order::B2B1A1A2: m = RI::Tensor({sizeB2, sizeB1, sizeA1, sizeA2}); break; - case Matrix_Order::B2B1A2A1: m = RI::Tensor({sizeB2, sizeB1, sizeA2, sizeA1}); break; - default: throw std::invalid_argument(std::string(__FILE__)+" line "+std::to_string(__LINE__)); - } + const double lat0 = *this->lat0; + RI::Tensor m; + const size_t sizeA1 = index_A1[TA].count_size; + const size_t sizeA2 = index_A2[TA].count_size; + const size_t sizeB1 = index_B1[TB].count_size; + const size_t sizeB2 = index_B2[TB].count_size; + switch(matrix_order) + { + case Matrix_Order::A1A2B1B2: m = RI::Tensor({sizeA1, sizeA2, sizeB1, sizeB2}); break; + case Matrix_Order::A1A2B2B1: m = RI::Tensor({sizeA1, sizeA2, sizeB2, sizeB1}); break; + case Matrix_Order::A1B1A2B2: m = RI::Tensor({sizeA1, sizeB1, sizeA2, sizeB2}); break; + case Matrix_Order::A1B1B2A2: m = RI::Tensor({sizeA1, sizeB1, sizeB2, sizeA2}); break; + case Matrix_Order::A1B2A2B1: m = RI::Tensor({sizeA1, sizeB2, sizeA2, sizeB1}); break; + case Matrix_Order::A1B2B1A2: m = RI::Tensor({sizeA1, sizeB2, sizeB1, sizeA2}); break; + case Matrix_Order::A2A1B1B2: m = RI::Tensor({sizeA2, sizeA1, sizeB1, sizeB2}); break; + case Matrix_Order::A2A1B2B1: m = RI::Tensor({sizeA2, sizeA1, sizeB2, sizeB1}); break; + case Matrix_Order::A2B1A1B2: m = RI::Tensor({sizeA2, sizeB1, sizeA1, sizeB2}); break; + case Matrix_Order::A2B1B2A1: m = RI::Tensor({sizeA2, sizeB1, sizeB2, sizeA1}); break; + case Matrix_Order::A2B2A1B1: m = RI::Tensor({sizeA2, sizeB2, sizeA1, sizeB1}); break; + case Matrix_Order::A2B2B1A1: m = RI::Tensor({sizeA2, sizeB2, sizeB1, sizeA1}); break; + case Matrix_Order::B1A1A2B2: m = RI::Tensor({sizeB1, sizeA1, sizeA2, sizeB2}); break; + case Matrix_Order::B1A1B2A2: m = RI::Tensor({sizeB1, sizeA1, sizeB2, sizeA2}); break; + case Matrix_Order::B1A2A1B2: m = RI::Tensor({sizeB1, sizeA2, sizeA1, sizeB2}); break; + case Matrix_Order::B1A2B2A1: m = RI::Tensor({sizeB1, sizeA2, sizeB2, sizeA1}); break; + case Matrix_Order::B1B2A1A2: m = RI::Tensor({sizeB1, sizeB2, sizeA1, sizeA2}); break; + case Matrix_Order::B1B2A2A1: m = RI::Tensor({sizeB1, sizeB2, sizeA2, sizeA1}); break; + case Matrix_Order::B2A1A2B1: m = RI::Tensor({sizeB2, sizeA1, sizeA2, sizeB1}); break; + case Matrix_Order::B2A1B1A2: m = RI::Tensor({sizeB2, sizeA1, sizeB1, sizeA2}); break; + case Matrix_Order::B2A2A1B1: m = RI::Tensor({sizeB2, sizeA2, sizeA1, sizeB1}); break; + case Matrix_Order::B2A2B1A1: m = RI::Tensor({sizeB2, sizeA2, sizeB1, sizeA1}); break; + case Matrix_Order::B2B1A1A2: m = RI::Tensor({sizeB2, sizeB1, sizeA1, sizeA2}); break; + case Matrix_Order::B2B1A2A1: m = RI::Tensor({sizeB2, sizeB1, sizeA2, sizeA1}); break; + default: throw std::invalid_argument(std::string(__FILE__)+" line "+std::to_string(__LINE__)); + } - for( const auto &co3 : center2_orb22_s.at(TA).at(TB) ) - { - const int LA1 = co3.first; - for( const auto &co4 : co3.second ) - { - const size_t NA1 = co4.first; - for( size_t MA1=0; MA1!=2*LA1+1; ++MA1 ) - { - for( const auto &co5 : co4.second ) - { - const int LA2 = co5.first; - for( const auto &co6 : co5.second ) - { - const size_t NA2 = co6.first; - for( size_t MA2=0; MA2!=2*LA2+1; ++MA2 ) - { - for( const auto &co7 : co6.second ) - { - const int LB1 = co7.first; - for( const auto &co8 : co7.second ) - { - const size_t NB1 = co8.first; - for( size_t MB1=0; MB1!=2*LB1+1; ++MB1 ) - { - for( const auto &co9 : co8.second ) - { - const int LB2 = co9.first; - for( const auto &co10 : co9.second ) - { - const size_t NB2 = co10.first; - for( size_t MB2=0; MB2!=2*LB2+1; ++MB2 ) - { - const Tdata overlap = co10.second.cal_overlap( tauA*lat0, tauB*lat0, MA1, MA2, MB1, MB2 ); - const size_t iA1 = index_A1[TA][LA1][NA1][MA1]; - const size_t iA2 = index_A2[TA][LA2][NA2][MA2]; - const size_t iB1 = index_B1[TB][LB1][NB1][MB1]; - const size_t iB2 = index_B2[TB][LB2][NB2][MB2]; - switch(matrix_order) - { - case Matrix_Order::A1A2B1B2: m(iA1,iA2,iB1,iB2) = overlap; break; - case Matrix_Order::A1A2B2B1: m(iA1,iA2,iB2,iB1) = overlap; break; - case Matrix_Order::A1B1A2B2: m(iA1,iB1,iA2,iB2) = overlap; break; - case Matrix_Order::A1B1B2A2: m(iA1,iB1,iB2,iA2) = overlap; break; - case Matrix_Order::A1B2A2B1: m(iA1,iB2,iA2,iB1) = overlap; break; - case Matrix_Order::A1B2B1A2: m(iA1,iB2,iB1,iA2) = overlap; break; - case Matrix_Order::A2A1B1B2: m(iA2,iA1,iB1,iB2) = overlap; break; - case Matrix_Order::A2A1B2B1: m(iA2,iA1,iB2,iB1) = overlap; break; - case Matrix_Order::A2B1A1B2: m(iA2,iB1,iA1,iB2) = overlap; break; - case Matrix_Order::A2B1B2A1: m(iA2,iB1,iB2,iA1) = overlap; break; - case Matrix_Order::A2B2A1B1: m(iA2,iB2,iA1,iB1) = overlap; break; - case Matrix_Order::A2B2B1A1: m(iA2,iB2,iB1,iA1) = overlap; break; - case Matrix_Order::B1A1A2B2: m(iB1,iA1,iA2,iB2) = overlap; break; - case Matrix_Order::B1A1B2A2: m(iB1,iA1,iB2,iA2) = overlap; break; - case Matrix_Order::B1A2A1B2: m(iB1,iA2,iA1,iB2) = overlap; break; - case Matrix_Order::B1A2B2A1: m(iB1,iA2,iB2,iA1) = overlap; break; - case Matrix_Order::B1B2A1A2: m(iB1,iB2,iA1,iA2) = overlap; break; - case Matrix_Order::B1B2A2A1: m(iB1,iB2,iA2,iA1) = overlap; break; - case Matrix_Order::B2A1A2B1: m(iB2,iA1,iA2,iB1) = overlap; break; - case Matrix_Order::B2A1B1A2: m(iB2,iA1,iB1,iA2) = overlap; break; - case Matrix_Order::B2A2A1B1: m(iB2,iA2,iA1,iB1) = overlap; break; - case Matrix_Order::B2A2B1A1: m(iB2,iA2,iB1,iA1) = overlap; break; - case Matrix_Order::B2B1A1A2: m(iB2,iB1,iA1,iA2) = overlap; break; - case Matrix_Order::B2B1A2A1: m(iB2,iB1,iA2,iA1) = overlap; break; - default: throw std::invalid_argument(std::string(__FILE__)+" line "+std::to_string(__LINE__)); - } - } - } - } - } - } - } - } - } - } - } - } - } - return m; + for( const auto &co3 : center2_orb22_s.at(TA).at(TB) ) + { + const int LA1 = co3.first; + for( const auto &co4 : co3.second ) + { + const size_t NA1 = co4.first; + for( size_t MA1=0; MA1!=2*LA1+1; ++MA1 ) + { + for( const auto &co5 : co4.second ) + { + const int LA2 = co5.first; + for( const auto &co6 : co5.second ) + { + const size_t NA2 = co6.first; + for( size_t MA2=0; MA2!=2*LA2+1; ++MA2 ) + { + for( const auto &co7 : co6.second ) + { + const int LB1 = co7.first; + for( const auto &co8 : co7.second ) + { + const size_t NB1 = co8.first; + for( size_t MB1=0; MB1!=2*LB1+1; ++MB1 ) + { + for( const auto &co9 : co8.second ) + { + const int LB2 = co9.first; + for( const auto &co10 : co9.second ) + { + const size_t NB2 = co10.first; + for( size_t MB2=0; MB2!=2*LB2+1; ++MB2 ) + { + const Tdata overlap = co10.second.cal_overlap( tauA*lat0, tauB*lat0, MA1, MA2, MB1, MB2 ); + const size_t iA1 = index_A1[TA][LA1][NA1][MA1]; + const size_t iA2 = index_A2[TA][LA2][NA2][MA2]; + const size_t iB1 = index_B1[TB][LB1][NB1][MB1]; + const size_t iB2 = index_B2[TB][LB2][NB2][MB2]; + switch(matrix_order) + { + case Matrix_Order::A1A2B1B2: m(iA1,iA2,iB1,iB2) = overlap; break; + case Matrix_Order::A1A2B2B1: m(iA1,iA2,iB2,iB1) = overlap; break; + case Matrix_Order::A1B1A2B2: m(iA1,iB1,iA2,iB2) = overlap; break; + case Matrix_Order::A1B1B2A2: m(iA1,iB1,iB2,iA2) = overlap; break; + case Matrix_Order::A1B2A2B1: m(iA1,iB2,iA2,iB1) = overlap; break; + case Matrix_Order::A1B2B1A2: m(iA1,iB2,iB1,iA2) = overlap; break; + case Matrix_Order::A2A1B1B2: m(iA2,iA1,iB1,iB2) = overlap; break; + case Matrix_Order::A2A1B2B1: m(iA2,iA1,iB2,iB1) = overlap; break; + case Matrix_Order::A2B1A1B2: m(iA2,iB1,iA1,iB2) = overlap; break; + case Matrix_Order::A2B1B2A1: m(iA2,iB1,iB2,iA1) = overlap; break; + case Matrix_Order::A2B2A1B1: m(iA2,iB2,iA1,iB1) = overlap; break; + case Matrix_Order::A2B2B1A1: m(iA2,iB2,iB1,iA1) = overlap; break; + case Matrix_Order::B1A1A2B2: m(iB1,iA1,iA2,iB2) = overlap; break; + case Matrix_Order::B1A1B2A2: m(iB1,iA1,iB2,iA2) = overlap; break; + case Matrix_Order::B1A2A1B2: m(iB1,iA2,iA1,iB2) = overlap; break; + case Matrix_Order::B1A2B2A1: m(iB1,iA2,iB2,iA1) = overlap; break; + case Matrix_Order::B1B2A1A2: m(iB1,iB2,iA1,iA2) = overlap; break; + case Matrix_Order::B1B2A2A1: m(iB1,iB2,iA2,iA1) = overlap; break; + case Matrix_Order::B2A1A2B1: m(iB2,iA1,iA2,iB1) = overlap; break; + case Matrix_Order::B2A1B1A2: m(iB2,iA1,iB1,iA2) = overlap; break; + case Matrix_Order::B2A2A1B1: m(iB2,iA2,iA1,iB1) = overlap; break; + case Matrix_Order::B2A2B1A1: m(iB2,iA2,iB1,iA1) = overlap; break; + case Matrix_Order::B2B1A1A2: m(iB2,iB1,iA1,iA2) = overlap; break; + case Matrix_Order::B2B1A2A1: m(iB2,iB1,iA2,iA1) = overlap; break; + default: throw std::invalid_argument(std::string(__FILE__)+" line "+std::to_string(__LINE__)); + } + } + } + } + } + } + } + } + } + } + } + } + } + return m; } template std::array,3> Matrix_Orbs22::cal_grad_overlap_matrix( - const size_t TA, - const size_t TB, - const ModuleBase::Vector3 &tauA, - const ModuleBase::Vector3 &tauB, - const ModuleBase::Element_Basis_Index::IndexLNM &index_A1, - const ModuleBase::Element_Basis_Index::IndexLNM &index_A2, - const ModuleBase::Element_Basis_Index::IndexLNM &index_B1, - const ModuleBase::Element_Basis_Index::IndexLNM &index_B2, - const Matrix_Order &matrix_order) const + const size_t TA, + const size_t TB, + const ModuleBase::Vector3 &tauA, + const ModuleBase::Vector3 &tauB, + const ModuleBase::Element_Basis_Index::IndexLNM &index_A1, + const ModuleBase::Element_Basis_Index::IndexLNM &index_A2, + const ModuleBase::Element_Basis_Index::IndexLNM &index_B1, + const ModuleBase::Element_Basis_Index::IndexLNM &index_B2, + const Matrix_Order &matrix_order) const { - std::array,3> m; - const size_t sizeA1 = index_A1[TA].count_size; - const size_t sizeA2 = index_A2[TA].count_size; - const size_t sizeB1 = index_B1[TB].count_size; - const size_t sizeB2 = index_B2[TB].count_size; - for(int i=0; i({sizeA1, sizeA2, sizeB1, sizeB2}); break; - case Matrix_Order::A1A2B2B1: m[i] = RI::Tensor({sizeA1, sizeA2, sizeB2, sizeB1}); break; - case Matrix_Order::A1B1A2B2: m[i] = RI::Tensor({sizeA1, sizeB1, sizeA2, sizeB2}); break; - case Matrix_Order::A1B1B2A2: m[i] = RI::Tensor({sizeA1, sizeB1, sizeB2, sizeA2}); break; - case Matrix_Order::A1B2A2B1: m[i] = RI::Tensor({sizeA1, sizeB2, sizeA2, sizeB1}); break; - case Matrix_Order::A1B2B1A2: m[i] = RI::Tensor({sizeA1, sizeB2, sizeB1, sizeA2}); break; - case Matrix_Order::A2A1B1B2: m[i] = RI::Tensor({sizeA2, sizeA1, sizeB1, sizeB2}); break; - case Matrix_Order::A2A1B2B1: m[i] = RI::Tensor({sizeA2, sizeA1, sizeB2, sizeB1}); break; - case Matrix_Order::A2B1A1B2: m[i] = RI::Tensor({sizeA2, sizeB1, sizeA1, sizeB2}); break; - case Matrix_Order::A2B1B2A1: m[i] = RI::Tensor({sizeA2, sizeB1, sizeB2, sizeA1}); break; - case Matrix_Order::A2B2A1B1: m[i] = RI::Tensor({sizeA2, sizeB2, sizeA1, sizeB1}); break; - case Matrix_Order::A2B2B1A1: m[i] = RI::Tensor({sizeA2, sizeB2, sizeB1, sizeA1}); break; - case Matrix_Order::B1A1A2B2: m[i] = RI::Tensor({sizeB1, sizeA1, sizeA2, sizeB2}); break; - case Matrix_Order::B1A1B2A2: m[i] = RI::Tensor({sizeB1, sizeA1, sizeB2, sizeA2}); break; - case Matrix_Order::B1A2A1B2: m[i] = RI::Tensor({sizeB1, sizeA2, sizeA1, sizeB2}); break; - case Matrix_Order::B1A2B2A1: m[i] = RI::Tensor({sizeB1, sizeA2, sizeB2, sizeA1}); break; - case Matrix_Order::B1B2A1A2: m[i] = RI::Tensor({sizeB1, sizeB2, sizeA1, sizeA2}); break; - case Matrix_Order::B1B2A2A1: m[i] = RI::Tensor({sizeB1, sizeB2, sizeA2, sizeA1}); break; - case Matrix_Order::B2A1A2B1: m[i] = RI::Tensor({sizeB2, sizeA1, sizeA2, sizeB1}); break; - case Matrix_Order::B2A1B1A2: m[i] = RI::Tensor({sizeB2, sizeA1, sizeB1, sizeA2}); break; - case Matrix_Order::B2A2A1B1: m[i] = RI::Tensor({sizeB2, sizeA2, sizeA1, sizeB1}); break; - case Matrix_Order::B2A2B1A1: m[i] = RI::Tensor({sizeB2, sizeA2, sizeB1, sizeA1}); break; - case Matrix_Order::B2B1A1A2: m[i] = RI::Tensor({sizeB2, sizeB1, sizeA1, sizeA2}); break; - case Matrix_Order::B2B1A2A1: m[i] = RI::Tensor({sizeB2, sizeB1, sizeA2, sizeA1}); break; - default: throw std::invalid_argument(std::string(__FILE__)+" line "+std::to_string(__LINE__)); - } - } - const double lat0 = *this->lat0; - for( const auto &co3 : center2_orb22_s.at(TA).at(TB) ) - { - const int LA1 = co3.first; - for( const auto &co4 : co3.second ) - { - const size_t NA1 = co4.first; - for( size_t MA1=0; MA1!=2*LA1+1; ++MA1 ) - { - for( const auto &co5 : co4.second ) - { - const int LA2 = co5.first; - for( const auto &co6 : co5.second ) - { - const size_t NA2 = co6.first; - for( size_t MA2=0; MA2!=2*LA2+1; ++MA2 ) - { - for( const auto &co7 : co6.second ) - { - const int LB1 = co7.first; - for( const auto &co8 : co7.second ) - { - const size_t NB1 = co8.first; - for( size_t MB1=0; MB1!=2*LB1+1; ++MB1 ) - { - for( const auto &co9 : co8.second ) - { - const int LB2 = co9.first; - for( const auto &co10 : co9.second ) - { - const size_t NB2 = co10.first; - for( size_t MB2=0; MB2!=2*LB2+1; ++MB2 ) - { - const std::array grad_overlap = RI_Util::Vector3_to_array3(co10.second.cal_grad_overlap( tauA*lat0, tauB*lat0, MA1, MA2, MB1, MB2 )); - const size_t iA1 = index_A1[TA][LA1][NA1][MA1]; - const size_t iA2 = index_A2[TA][LA2][NA2][MA2]; - const size_t iB1 = index_B1[TB][LB1][NB1][MB1]; - const size_t iB2 = index_B2[TB][LB2][NB2][MB2]; - for(size_t i=0; i,3> m; + const size_t sizeA1 = index_A1[TA].count_size; + const size_t sizeA2 = index_A2[TA].count_size; + const size_t sizeB1 = index_B1[TB].count_size; + const size_t sizeB2 = index_B2[TB].count_size; + for(int i=0; i({sizeA1, sizeA2, sizeB1, sizeB2}); break; + case Matrix_Order::A1A2B2B1: m[i] = RI::Tensor({sizeA1, sizeA2, sizeB2, sizeB1}); break; + case Matrix_Order::A1B1A2B2: m[i] = RI::Tensor({sizeA1, sizeB1, sizeA2, sizeB2}); break; + case Matrix_Order::A1B1B2A2: m[i] = RI::Tensor({sizeA1, sizeB1, sizeB2, sizeA2}); break; + case Matrix_Order::A1B2A2B1: m[i] = RI::Tensor({sizeA1, sizeB2, sizeA2, sizeB1}); break; + case Matrix_Order::A1B2B1A2: m[i] = RI::Tensor({sizeA1, sizeB2, sizeB1, sizeA2}); break; + case Matrix_Order::A2A1B1B2: m[i] = RI::Tensor({sizeA2, sizeA1, sizeB1, sizeB2}); break; + case Matrix_Order::A2A1B2B1: m[i] = RI::Tensor({sizeA2, sizeA1, sizeB2, sizeB1}); break; + case Matrix_Order::A2B1A1B2: m[i] = RI::Tensor({sizeA2, sizeB1, sizeA1, sizeB2}); break; + case Matrix_Order::A2B1B2A1: m[i] = RI::Tensor({sizeA2, sizeB1, sizeB2, sizeA1}); break; + case Matrix_Order::A2B2A1B1: m[i] = RI::Tensor({sizeA2, sizeB2, sizeA1, sizeB1}); break; + case Matrix_Order::A2B2B1A1: m[i] = RI::Tensor({sizeA2, sizeB2, sizeB1, sizeA1}); break; + case Matrix_Order::B1A1A2B2: m[i] = RI::Tensor({sizeB1, sizeA1, sizeA2, sizeB2}); break; + case Matrix_Order::B1A1B2A2: m[i] = RI::Tensor({sizeB1, sizeA1, sizeB2, sizeA2}); break; + case Matrix_Order::B1A2A1B2: m[i] = RI::Tensor({sizeB1, sizeA2, sizeA1, sizeB2}); break; + case Matrix_Order::B1A2B2A1: m[i] = RI::Tensor({sizeB1, sizeA2, sizeB2, sizeA1}); break; + case Matrix_Order::B1B2A1A2: m[i] = RI::Tensor({sizeB1, sizeB2, sizeA1, sizeA2}); break; + case Matrix_Order::B1B2A2A1: m[i] = RI::Tensor({sizeB1, sizeB2, sizeA2, sizeA1}); break; + case Matrix_Order::B2A1A2B1: m[i] = RI::Tensor({sizeB2, sizeA1, sizeA2, sizeB1}); break; + case Matrix_Order::B2A1B1A2: m[i] = RI::Tensor({sizeB2, sizeA1, sizeB1, sizeA2}); break; + case Matrix_Order::B2A2A1B1: m[i] = RI::Tensor({sizeB2, sizeA2, sizeA1, sizeB1}); break; + case Matrix_Order::B2A2B1A1: m[i] = RI::Tensor({sizeB2, sizeA2, sizeB1, sizeA1}); break; + case Matrix_Order::B2B1A1A2: m[i] = RI::Tensor({sizeB2, sizeB1, sizeA1, sizeA2}); break; + case Matrix_Order::B2B1A2A1: m[i] = RI::Tensor({sizeB2, sizeB1, sizeA2, sizeA1}); break; + default: throw std::invalid_argument(std::string(__FILE__)+" line "+std::to_string(__LINE__)); + } + } + const double lat0 = *this->lat0; + for( const auto &co3 : center2_orb22_s.at(TA).at(TB) ) + { + const int LA1 = co3.first; + for( const auto &co4 : co3.second ) + { + const size_t NA1 = co4.first; + for( size_t MA1=0; MA1!=2*LA1+1; ++MA1 ) + { + for( const auto &co5 : co4.second ) + { + const int LA2 = co5.first; + for( const auto &co6 : co5.second ) + { + const size_t NA2 = co6.first; + for( size_t MA2=0; MA2!=2*LA2+1; ++MA2 ) + { + for( const auto &co7 : co6.second ) + { + const int LB1 = co7.first; + for( const auto &co8 : co7.second ) + { + const size_t NB1 = co8.first; + for( size_t MB1=0; MB1!=2*LB1+1; ++MB1 ) + { + for( const auto &co9 : co8.second ) + { + const int LB2 = co9.first; + for( const auto &co10 : co9.second ) + { + const size_t NB2 = co10.first; + for( size_t MB2=0; MB2!=2*LB2+1; ++MB2 ) + { + const std::array grad_overlap = RI_Util::Vector3_to_array3(co10.second.cal_grad_overlap( tauA*lat0, tauB*lat0, MA1, MA2, MB1, MB2 )); + const size_t iA1 = index_A1[TA][LA1][NA1][MA1]; + const size_t iA2 = index_A2[TA][LA2][NA2][MA2]; + const size_t iB1 = index_B1[TB][LB1][NB1][MB1]; + const size_t iB2 = index_B2[TB][LB2][NB2][MB2]; + for(size_t i=0; i std::map < size_t, std::map>>>> Matrix_Orbs22::cal_overlap_matrix_all( - const UnitCell &ucell, - const ModuleBase::Element_Basis_Index::IndexLNM &index_A1, - const ModuleBase::Element_Basis_Index::IndexLNM &index_A2, - const ModuleBase::Element_Basis_Index::IndexLNM &index_B1, - const ModuleBase::Element_Basis_Index::IndexLNM &index_B2 ) const + const UnitCell &ucell, + const ModuleBase::Element_Basis_Index::IndexLNM &index_A1, + const ModuleBase::Element_Basis_Index::IndexLNM &index_A2, + const ModuleBase::Element_Basis_Index::IndexLNM &index_B1, + const ModuleBase::Element_Basis_Index::IndexLNM &index_B2 ) const { - std::map>>>> matrixes; + std::map>>>> matrixes; - for( const auto &co1 : center2_orb22_s ) - { - const size_t TA = co1.first; - for( size_t IA=0; IA!=ucell.atoms[TA].na; ++IA ) - { - const ModuleBase::Vector3 &tauA( ucell.atoms[TA].tau[IA] ); + for( const auto &co1 : center2_orb22_s ) + { + const size_t TA = co1.first; + for( size_t IA=0; IA!=ucell.atoms[TA].na; ++IA ) + { + const ModuleBase::Vector3 &tauA( ucell.atoms[TA].tau[IA] ); - for( const auto &co2 : co1.second ) - { - const size_t TB = co2.first; - for( size_t IB=0; IB!=ucell.atoms[TB].na; ++IB ) - { - const ModuleBase::Vector3 &tauB( ucell.atoms[TB].tau[IB] ); + for( const auto &co2 : co1.second ) + { + const size_t TB = co2.first; + for( size_t IB=0; IB!=ucell.atoms[TB].na; ++IB ) + { + const ModuleBase::Vector3 &tauB( ucell.atoms[TB].tau[IB] ); - matrixes[TA][IA][TB][IB] = cal_overlap_matrix( - TA, - TB, - ucell.atoms[TA].tau[IA], - ucell.atoms[TB].tau[IB], - index_A1, - index_A2, - index_B1, - index_B2, - Matrix_Order::A1B1A2B2); - } - } - } - } - return matrixes; + matrixes[TA][IA][TB][IB] = cal_overlap_matrix( + TA, + TB, + ucell.atoms[TA].tau[IA], + ucell.atoms[TB].tau[IB], + index_A1, + index_A2, + index_B1, + index_B2, + Matrix_Order::A1B1A2B2); + } + } + } + } + return matrixes; } #endif diff --git a/source/source_lcao/module_ri/mix_dmk_2d.h b/source/source_lcao/module_ri/mix_dmk_2d.h index 042935ccb74..62c61c49d9b 100644 --- a/source/source_lcao/module_ri/mix_dmk_2d.h +++ b/source/source_lcao/module_ri/mix_dmk_2d.h @@ -15,37 +15,37 @@ template class Mix_DMk_2D { public: - ~Mix_DMk_2D(); + ~Mix_DMk_2D(); - /** - * @brief Sets the number of k-points. - * @param nks Number of k-points. - */ - void set_nks(const int nks); + /** + * @brief Sets the number of k-points. + * @param nks Number of k-points. + */ + void set_nks(const int nks); - /** - * @brief Sets the mixing mode. - * @param Mixing Mixing pointer. - */ - void set_mixing(Base_Mixing::Mixing* mixing_in); + /** + * @brief Sets the mixing mode. + * @param Mixing Mixing pointer. + */ + void set_mixing(Base_Mixing::Mixing* mixing_in); - /** - * @brief Sets Base_Mixing::Plain_Mixing. - * @param mixing_beta mixing beta for plain mixing. - */ - void set_mixing_plain(const double& mixing_beta); + /** + * @brief Sets Base_Mixing::Plain_Mixing. + * @param mixing_beta mixing beta for plain mixing. + */ + void set_mixing_plain(const double& mixing_beta); - /** - * @brief Mixes the density matrix. - * @param dm Density matrix. - * @param flag_restart Flag indicating whether restart mixing. - */ + /** + * @brief Mixes the density matrix. + * @param dm Density matrix. + * @param flag_restart Flag indicating whether restart mixing. + */ void mix(const std::vector>& dm, const bool flag_restart); - /** - * @brief Returns the density matrix. - * @return Density matrices for each k-points. - */ + /** + * @brief Returns the density matrix. + * @return Density matrices for each k-points. + */ std::vector*> get_DMk_out() const; private: @@ -61,7 +61,7 @@ class Mix_DMk_2D std::vector mix_DMk; Base_Mixing::Mixing* mixing = nullptr; - bool flag_del_mixing = false; + bool flag_del_mixing = false; }; #endif diff --git a/source/source_lcao/module_ri/module_exx_symmetry/CMakeLists.txt b/source/source_lcao/module_ri/module_exx_symmetry/CMakeLists.txt index 2af8a5bf7fa..e5b6e32ce24 100644 --- a/source/source_lcao/module_ri/module_exx_symmetry/CMakeLists.txt +++ b/source/source_lcao/module_ri/module_exx_symmetry/CMakeLists.txt @@ -1,8 +1,6 @@ if (ENABLE_LIBRI) list(APPEND objects - irreducible_sector.cpp - irred_sec_bvk.cpp symm_rotation.cpp symm_rot_out.cpp ) diff --git a/source/source_lcao/module_ri/module_exx_symmetry/symm_rot_out.cpp b/source/source_lcao/module_ri/module_exx_symmetry/symm_rot_out.cpp index 5dbecc1a40f..0ff3c9be33b 100644 --- a/source/source_lcao/module_ri/module_exx_symmetry/symm_rot_out.cpp +++ b/source/source_lcao/module_ri/module_exx_symmetry/symm_rot_out.cpp @@ -1,4 +1,6 @@ #include "./symm_rotation.h" +#include "source_io/module_parameter/parameter.h" + namespace ModuleSymmetry { std::string mat3_fmt(const ModuleBase::Matrix3& m) diff --git a/source/source_lcao/module_ri/module_exx_symmetry/symm_rotation.cpp b/source/source_lcao/module_ri/module_exx_symmetry/symm_rotation.cpp index a1942918f70..b8261ad6fe1 100644 --- a/source/source_lcao/module_ri/module_exx_symmetry/symm_rotation.cpp +++ b/source/source_lcao/module_ri/module_exx_symmetry/symm_rotation.cpp @@ -7,6 +7,7 @@ #include "source_base/tool_title.h" #include "source_base/timer.h" #include "source_base/mathzone.h" +#include "source_lcao/module_ri/ri_util.h" namespace ModuleSymmetry { @@ -16,590 +17,6 @@ namespace ModuleSymmetry this->abfs_l_nchi_ = abfs_l_nchi; for (auto& abfs_T : abfs_l_nchi) { this->abfs_Lmax_ = std::max(this->abfs_Lmax_, static_cast(abfs_T.size()) - 1); } } - void Symmetry_rotation::cal_Ms(const K_Vectors& kv, - //const std::vector>& kstars, - const UnitCell& ucell, const Parallel_2D& pv) - { - ModuleBase::TITLE("Symmetry_rotation", "cal_Ms"); - ModuleBase::timer::start("Symmetry_rotation", "cal_Ms"); - - this->nsym_ = ucell.symm.nrotk; - this->nanti_ = ucell.symm.nrotk_anti; - this->magnetic_nspin4_ = ucell.symm.magnetic_nspin4; - this->eps_ = ucell.symm.epsilon; - if (this->irs_.invmap_.empty()) - { - this->irs_.invmap_.resize(ucell.symm.nrotk); - ucell.symm.gmatrix_invmap(ucell.symm.gmatrix, ucell.symm.nrotk, this->irs_.invmap_.data()); - } - // 1. calculate the rotation matrix in real spherical harmonics representation for each symmetry operation: [T_l (isym)]_mm' - const int nop_tot = this->nsym_ + this->nanti_; - std::vector gmatc(nop_tot); - for (int i = 0;i < nsym_;++i) { gmatc[i] = this->irs_.direct_to_cartesian(ucell.symm.gmatrix[i], ucell.latvec); } - for (int j = 0;j < this->nanti_;++j) - { gmatc[nsym_ + j] = this->irs_.direct_to_cartesian(ucell.symm.gmatrix_anti[j], ucell.latvec); } - this->cal_rotmat_Slm(gmatc.data(), std::max(this->abfs_Lmax_, ucell.lmax), nop_tot); - - // 1.5 (nspin=4) the SU(2) spin-1/2 rotation U(isym) for each symmetry operation. The AO - // rotation matrix M becomes the spinor operator T(isym) (x) U(isym) so that the same - // gemm D(k)=M^dagger D(k_ibz) M rotates both the orbital and the spin part at once. - // For an antiunitary element Theta*g only the spatial part g enters M here; the Theta - // (sigma_y (.)^* sigma_y) is applied afterwards in restore_dm. - std::vector spin_U(nop_tot, SpinRotation::Su2{ 1.0, 0.0, 0.0, 1.0 }); - if (PARAM.inp.nspin == 4) - { - for (int i = 0;i < nop_tot;++i) { spin_U[i] = SpinRotation::so3_to_su2(gmatc[i]); } - } - this->spin_U_ = spin_U; // keep for restore_HR_nspin4 (real-space EXX H(R) spin mixing) - - // A k-star contains only one operation per distinct k point. The other - // operations fixing k (modulo a reciprocal lattice vector) must still - // be averaged: a finite-grid SCF density need not respect this little group. - const int nks_ibz = kv.kstars.size(); - this->Ms_.assign(nks_ibz, {}); - this->little_groups_.assign(nks_ibz, {}); - for (int ik_ibz = 0; ik_ibz < nks_ibz; ++ik_ibz) - { - std::set needed; - for (const auto& member : kv.kstars[ik_ibz]) - { - const int op = (!this->magnetic_nspin4_ && member.first >= nsym_) - ? member.first - nsym_ : member.first; - needed.insert(op); - } - for (int op = 0; op < nsym_; ++op) - { - const auto delta = kv.kvec_d[ik_ibz] * ucell.symm.kgmatrix[op] - kv.kvec_d[ik_ibz]; - if (std::abs(delta.x - std::round(delta.x)) < this->eps_ - && std::abs(delta.y - std::round(delta.y)) < this->eps_ - && std::abs(delta.z - std::round(delta.z)) < this->eps_) - { - this->little_groups_[ik_ibz].push_back(op); - needed.insert(op); - } - } - for (const int op : needed) - { - this->Ms_[ik_ibz][op] = this->contruct_2d_rot_mat_ao( - ucell.symm, ucell.atoms, ucell.st, kv.kvec_d[ik_ibz], op, pv, spin_U[op]); - } - } - // output Ms of isym=1 - // std::ofstream ofs("Ms_kibz7_sym7.dat"); - // for (int i = 0;i < pv.get_row_size();++i) - // { - // for (int j = 0;j < pv.get_col_size();++j) - // { - // ofs << std::setprecision(10) << this->Ms_[7][7][j * pv.get_col_size() + i] << " "; - // } - // ofs << std::endl; - // } - // ofs << std::endl; - // ofs.close(); - - ModuleBase::timer::end("Symmetry_rotation", "cal_Ms"); - } - - std::vector>> Symmetry_rotation::restore_dm(const K_Vectors& kv, - const std::vector>>& dm_k_ibz, const Parallel_2D& pv)const - { - ModuleBase::TITLE("Symmetry_rotation", "restore_dm"); - ModuleBase::timer::start("Symmetry_rotation", "restore_dm"); - std::vector>> dm_k_full; - int nspin0 = PARAM.inp.nspin == 2 ? 2 : 1; - dm_k_full.reserve(kv.get_nkstot_nospin() * nspin0); //nkstot_nospin didn't doubled by spin - int nk = kv.get_nkstot() / nspin0; - - // (nspin=4) Sigma_y = I (x) sigma_y for the time-reversal spin flip; k-independent, build once. - std::vector> sigma_y; - if (PARAM.inp.nspin == 4) { sigma_y = this->set_sigma_y_2d(pv); } - - for (int is = 0;is < nspin0;++is) - { - for (int ik_ibz = 0;ik_ibz < nk;++ik_ibz) - { - // P_k D = |G_k|^{-1} sum_g M_g^T D M_g^*. This preserves - // Hermiticity and makes restoration independent of the chosen - // star representative; rotating just one arbitrary D does not. - const auto& little_group = this->little_groups_.at(ik_ibz); - assert(!little_group.empty()); - std::vector> projected = dm_k_ibz[ik_ibz + is * nk]; - if (little_group.size() > 1) - { - std::fill(projected.begin(), projected.end(), 0.0); - for (const int op : little_group) - { - const auto rotated = this->rot_matrix_ao( - dm_k_ibz[ik_ibz + is * nk], ik_ibz, little_group.size(), op, pv); - for (size_t i = 0; i < projected.size(); ++i) - { - projected[i] += rotated[i]; - } - } - } - for (auto& isym_kvd : kv.kstars[ik_ibz]) - { - if (isym_kvd.first == 0) - { - double factor = 1.0 / static_cast(kv.kstars[ik_ibz].size()); - std::vector> dm_scaled(pv.get_local_size()); - for (int i = 0;i < pv.get_local_size();++i) { dm_scaled[i] = factor * projected[i]; } - dm_k_full.push_back(dm_scaled); - } - else if (isym_kvd.first < nsym_) - { //space group operations - dm_k_full.push_back(this->rot_matrix_ao(projected, ik_ibz, kv.kstars[ik_ibz].size(), isym_kvd.first, pv)); - } - else - { // antiunitary elements: Theta * (spatial operation) - // D(Theta*g k_ibz) = sigma_y [D(g k_ibz)]^* sigma_y with D(g k_ibz) = M^dagger D M. - // For nspin=4, first do the (non-conjugated) spatial rotation, then the spin flip; - // for nspin<4 (Theta=K) the original TRS_conj path already gives the conjugate. - // - // Which spatial operation the index denotes depends on the regime, matching - // how the k-reduction filled kgmatrix[] (see K_Vectors::reduce_by_symmetry): - // - nspin=4 magnetic (Shubnikov): index j+nsym_ is the antiunitary element - // Theta*gmatrix_anti[j]; its Ms is stored under the RAW key j+nsym_. - // - otherwise (grey group / nspin<4): index i+nsym_ is Theta*gmatrix[i], - // i.e. the unitary operation i, whose Ms is stored under key i. - const int isym_M = this->magnetic_nspin4_ ? isym_kvd.first : (isym_kvd.first - nsym_); - if (PARAM.inp.nspin == 4) - { - // m=0: gray group: the space-group part of anti-unitary elements are the same of the unitary elements, isym_M < nsym_ - // m!=0: Shubnikov group: using different space-group part of anti-unitary elements stored in gmatrix_anti with isym_M >= nsym_ - dm_k_full.push_back(this->trs_spin_rotate( - this->rot_matrix_ao(projected, ik_ibz, kv.kstars[ik_ibz].size(), isym_M, pv, false), - sigma_y, pv, 1.0)); - } - else - { - dm_k_full.push_back(this->rot_matrix_ao(projected, ik_ibz, kv.kstars[ik_ibz].size(), isym_M, pv, true)); - } - } - } - } - } - // test for output -/* - std::ofstream ofs("DM.dat"); - int ik = 0; - for (int ikibz = 0;ikibz < kv.get_nkstot() / nspin0;++ikibz) - for (auto& isym_kvd : kv.kstars[ikibz]) - { - ofs << "isym=" << isym_kvd.first << std::endl; - ofs << " k = " << isym_kvd.second.x << " " << isym_kvd.second.y << " " << isym_kvd.second.z << std::endl; - ofs << "DM(k):" << std::endl; - for (int i = 0;i < pv.get_row_size();++i) - { - for (int j = 0;j < pv.get_col_size();++j) - { - ofs << dm_k_full[ik][j * pv.get_row_size() + i] << " "; - } - ofs << std::endl; - } - ++ik; - ofs << std::endl; - } - ofs.close(); -*/ - ModuleBase::timer::end("Symmetry_rotation", "restore_dm"); - return dm_k_full; - } - std::vector> Symmetry_rotation::restore_dm(const K_Vectors& kv, - const std::vector>& dm_k_ibz, const Parallel_2D& pv)const - { - return dm_k_ibz;// do nothing for gamma_only - } - - // calculate Wigner D matrix - double Symmetry_rotation::wigner_d(const double beta, const int l, const int m1, const int m2) const - { - auto factorial = [](int n) -> int { - int result = 1; - for (int i = 1;i <= n;++i) { result *= i; -} - return result; - }; - double result = 0.0; - for (int i = std::max(0, m2 - m1);i <= std::min(l - m1, l + m2);++i) { - result += std::pow(-1, i) * std::sqrt(factorial(l + m1) * factorial(l - m1) * factorial(l + m2) * factorial(l - m2)) - * std::pow(std::cos(beta / 2), 2 * l + m2 - m1 - 2 * i) * std::pow(-std::sin(beta / 2), m1 - m2 + 2 * i) - / (factorial(i) * factorial(l - m1 - i) * factorial(l + m2 - i) * factorial(i - m2 + m1)); -} - return result; - } - - std::complex Symmetry_rotation::wigner_D(const TCdouble& euler_angle, const int l, const int m1, const int m2, const bool inv) const - { - std::complex prefac(inv ? std::pow(-1, l) : 1, 0); - return std::exp(-ModuleBase::IMAG_UNIT * static_cast(m1) * euler_angle.x) - * std::exp(-ModuleBase::IMAG_UNIT * static_cast(m2) * euler_angle.z) - * wigner_d(euler_angle.y, l, m1, m2) * prefac; - } - - // c^l_{m1, m2}= - std::complex Symmetry_rotation::ovlp_Ylm_Slm(const int l, const int m1, const int m2) const - { - if (m1 == m2) - { - if (m1 == 0) { return 1.0; -} - if (m1 > 0) { return 1 / std::sqrt(2); -} - if (m1 < 0) { return std::pow(-1, m1) * ModuleBase::IMAG_UNIT / std::sqrt(2); -} - } - else if (m1 == -m2) - { - if (m1 > 0) { return -ModuleBase::IMAG_UNIT / std::sqrt(2); -} - if (m1 < 0) { return std::pow(-1, m1) / std::sqrt(2); -} - } - return 0.0; - } - - // reference: https://github.com/minyez/abf_trans/blob/f9e68e68069a94610d89e077bfe6e8ffac0b097d/src/rotate.cpp#L118 - // because the atom position here is row vector, the original gmatrix(eular angle) is transposed. - // gmatc: the rotation matrix under the basis of cartesian coordinates - // gmatc should be a rotation matrix, i.e. det(gmatc)=1 - TCdouble Symmetry_rotation::get_euler_angle(const ModuleBase::Matrix3& gmatc) const - { - double threshold = this->eps_; - double alpha = 0.0, beta = 0.0, gamma = 0.0; - if (std::fabs(gmatc.e32) > threshold || std::fabs(gmatc.e31) > threshold) // sin(beta) is not zero - { - // use the 2-angle elements to get alpha and gamma - alpha = std::atan2(gmatc.e32, gmatc.e31); - if (alpha < 0) { alpha += 2 * ModuleBase::PI; -} - gamma = std::atan2(gmatc.e23, -gmatc.e13); - if (gamma < 0) { gamma += 2 * ModuleBase::PI; -} - // use the larger one of 2-angle elements to calculate beta - if (std::fabs(gmatc.e32) > std::fabs(gmatc.e31)) { - beta = std::atan2(gmatc.e32 / std::sin(alpha), gmatc.e33); - } else { - beta = std::atan2(gmatc.e31 / std::cos(alpha), gmatc.e33); -} - } - else - {//sin(beta)=0, beta = 0 or pi, only (alpha+gamma) or (alpha-gamma) is important. now assign this to alpha. - alpha = std::atan2(gmatc.e12, gmatc.e11); - if (alpha < 0) { alpha += 2 * ModuleBase::PI; -} - // if beta=0, gmatc.e11=cos(alpha+gamma), gmatc.e21=sin(alpha+gamma) - // if beta=pi, gmatc.e11=cos(pi+alpha-gamma), gmatc.e21=sin(pi+alpha-gamma) - if (gmatc.e33 > 0) - { - beta = 0; - gamma = 0; //alpha+gamma=alpha => gamma=0 - } - else - { - beta = ModuleBase::PI; - gamma = ModuleBase::PI;// pi+alpha-gamma=alpha => gamma=pi - } - } - return TCdouble(alpha, beta, gamma); - } - - // in: the real value of m in range {-l, -l+1, ..., 0, ..., l-1, l} - // out: the index of the orbital in a fixed {n， l}, i.e. the index in array [0, 1, -1, 2, -2, ...] - inline int m2im(int m) - { - return (m > 0 ? 2 * m - 1 : -2 * m); - } - - /// T_mm' = [c^\dagger D c]_mm' - void Symmetry_rotation::cal_rotmat_Slm(const ModuleBase::Matrix3* gmatc, const int lmax, const int nop) - { - const int nop_tot = (nop < 0) ? this->nsym_ : nop; - auto set_integer = [](RI::Tensor>& mat) -> void - { - double zero_thres = 1e-10; - for (int i = 0;i < mat.shape[0];++i) { - for (int j = 0;j < mat.shape[1];++j) - { - if (std::abs(mat(i, j).real() - std::round(mat(i, j).real())) < zero_thres) { mat(i, j).real(std::round(mat(i, j).real())); -} - if (std::abs(mat(i, j).imag() - std::round(mat(i, j).imag())) < zero_thres) { mat(i, j).imag(std::round(mat(i, j).imag())); -} - } -} - }; - this->rotmat_Slm_.resize(nop_tot); - // c matrix is independent on isym - std::vector>> c_mm(lmax + 1); - for (int l = 0;l <= lmax;++l) { - c_mm[l] = RI::Tensor>({ size_t(2 * l + 1), size_t(2 * l + 1) }); -} - for (int l = 0;l <= lmax;++l) { - for (int m1 = -l;m1 <= l;++m1) { - for (int m2 = -l;m2 <= l;++m2) { - c_mm[l](m2im(m1), m2im(m2)) = ovlp_Ylm_Slm(l, m1, m2); -} -} -} - - for (int isym = 0;isym < nop_tot;++isym) - { - // if R is a reflection operation, calculate D^l(R)=(-1)^l*D^l(IR), so the euler angle of (IR) is needed. - TCdouble euler_angle = get_euler_angle(gmatc[isym].Det() > 0 ? - gmatc[isym] : gmatc[isym] * ModuleBase::Matrix3(-1, 0, 0, 0, -1, 0, 0, 0, -1)); - - this->rotmat_Slm_[isym].resize(lmax + 1); - for (int l = 0;l <= lmax;++l) - {// wigner D matrix - RI::Tensor> D_mm({ size_t(2 * l + 1), size_t(2 * l + 1) }); - for (int m1 = -l;m1 <= l;++m1) { - for (int m2 = -l;m2 <= l;++m2) { - D_mm(m2im(m1), m2im(m2)) = wigner_D(euler_angle, l, m1, m2, (gmatc[isym].Det() < 0)); -} -} - this->rotmat_Slm_[isym][l] = c_mm[l].dagger() * D_mm * c_mm[l]; - // set_integer(this->rotmat_Slm_[isym][l]); - } - } - /* - std::vector euler_angles_test(nsym_); - for (int isym = 0;isym < nsym_;++isym) euler_angles_test[isym] = - get_euler_angle(gmatc[isym].Det() > 0 ? gmatc[isym] : gmatc[isym] * ModuleBase::Matrix3(-1, 0, 0, 0, -1, 0, 0, 0, -1)); - - auto test_Tmm = [&]()-> void - { - std::ofstream ofs("Tlm.dat"); - for (int isym = 0;isym < nsym_;++isym) - { - ofs << "isym=" << isym << std::endl; - ofs << "gmatrix_cart=" << std::endl; - ofs << gmatc[isym].e11 << " " << gmatc[isym].e12 << " " << gmatc[isym].e13 << std::endl; - ofs << gmatc[isym].e21 << " " << gmatc[isym].e22 << " " << gmatc[isym].e23 << std::endl; - ofs << gmatc[isym].e31 << " " << gmatc[isym].e32 << " " << gmatc[isym].e33 << std::endl; - ofs << "gmatrix_direct=" << std::endl; - ofs << ucell.symm.gmatrix[isym].e11 << " " << ucell.symm.gmatrix[isym].e12 << " " << ucell.symm.gmatrix[isym].e13 << std::endl; - ofs << ucell.symm.gmatrix[isym].e21 << " " << ucell.symm.gmatrix[isym].e22 << " " << ucell.symm.gmatrix[isym].e23 << std::endl; - ofs << ucell.symm.gmatrix[isym].e31 << " " << ucell.symm.gmatrix[isym].e32 << " " << ucell.symm.gmatrix[isym].e33 << std::endl; - ofs << "kgmatrix_direct=" << std::endl; - ofs << ucell.symm.kgmatrix[isym].e11 << " " << ucell.symm.kgmatrix[isym].e12 << " " << ucell.symm.kgmatrix[isym].e13 << std::endl; - ofs << ucell.symm.kgmatrix[isym].e21 << " " << ucell.symm.kgmatrix[isym].e22 << " " << ucell.symm.kgmatrix[isym].e23 << std::endl; - ofs << ucell.symm.kgmatrix[isym].e31 << " " << ucell.symm.kgmatrix[isym].e32 << " " << ucell.symm.kgmatrix[isym].e33 << std::endl; - ofs << "euler_angle/pi: " << euler_angles_test[isym].x / ModuleBase::PI << " " - << euler_angles_test[isym].y / ModuleBase::PI << " " << euler_angles_test[isym].z / ModuleBase::PI << std::endl; - for (int l = 0;l <= lmax;++l) - for (int i = 0;i < 2 * l + 1;++i) - { - for (int j = 0;j < 2 * l + 1;++j) ofs << this->rotmat_Slm_[isym][l](i, j) << " "; - ofs << std::endl; - } - } - ofs.close(); - }; - test_Tmm(); - */ - } - - void Symmetry_rotation::set_block_to_mat2d(const int starti, const int startj, const RI::Tensor>& block, - std::vector>& obj_mat, const Parallel_2D& pv, const bool trans) const - { // caution: ComplaxMatrix is row-major(col-continuous), but obj_mat is col-major(row-continuous) - for (int j = 0;j < block.shape[0];++j) {//outside dimension - for (int i = 0;i < block.shape[1];++i) { //inside dimension - if (pv.in_this_processor(starti + i, startj + j)) - { - int index = pv.global2local_col(startj + j) * pv.get_row_size() + pv.global2local_row(starti + i); - obj_mat[index] = trans ? block(i, j) : block(j, i); - } -} -} - } - - void Symmetry_rotation::set_block_to_mat2d(const int starti, const int startj, const RI::Tensor>& block, - std::vector& obj_mat, const Parallel_2D& pv, const bool trans) const - { // caution: ComplaxMatrix is row-major(col-continuous), but obj_mat is col-major(row-continuous) - for (int j = 0;j < block.shape[0];++j) {//outside dimension - for (int i = 0;i < block.shape[1];++i) { //inside dimension - if (pv.in_this_processor(starti + i, startj + j)) - { - int index = pv.global2local_col(startj + j) * pv.get_row_size() + pv.global2local_row(starti + i); - obj_mat[index] = trans ? block(i, j).real() : block(j, i).real(); - } -} -} - } - - // 2d-block parallized rotation matrix in AO-representation, denoted as M. - // finally we will use D(k)=M(R, k)^\dagger*D(Rk)*M(R, k) to D(k) from D(Rk) in cal_Ms. - std::vector> Symmetry_rotation::contruct_2d_rot_mat_ao(const Symmetry& symm, const Atom* atoms, const Statistics& cell_st, - const TCdouble& kvec_d_ibz, int isym, const Parallel_2D& pv, const SpinRotation::Su2& spin_U) const - { - const bool soc = (PARAM.inp.nspin == 4); - const int npol = soc ? 2 : 1; // spinor: global AO index is spin-fast interleaved, I = npol*iw_orb + s - std::vector> M_isym(pv.get_local_size(), 0.0); - // isym >= symm.nrotk addresses the antiunitary coset (spatial part gmatrix_anti[isym-nrotk]), - // whose atom map lives in a separate table. - const int nrotk_u = symm.nrotk; - auto rotated_atom = [&symm, nrotk_u](const int is, const int iat) -> int - { - return (is < nrotk_u) ? symm.get_rotated_atom(is, iat) - : symm.get_rotated_atom_anti(is - nrotk_u, iat); - }; - for (int iat1 = 0;iat1 < cell_st.nat;++iat1) - { - int it = cell_st.iat2it[iat1]; // it1=it2 - int ia1 = cell_st.iat2ia[iat1]; - int iat2 = rotated_atom(isym, iat1); //iat2=rot(iat1) - int ia2 = cell_st.iat2ia[iat2]; - // cal phase factor from return lattice: exp(-ik_ibz*O) - double arg = -2 * ModuleBase::PI * kvec_d_ibz * this->irs_.return_lattice_[iat1][isym]; - std::complexphase_factor = std::complex(std::cos(arg), std::sin(arg)); - int iw1start = atoms[it].stapos_wf + ia1 * atoms[it].nw; - int iw2start = atoms[it].stapos_wf + ia2 * atoms[it].nw; - int iw = 0; - while (iw < atoms[it].nw) - { - int l = atoms[it].iw2l[iw]; - int nm = 2 * l + 1; - //caution: the order of m in orbitals may be different from increasing - if (!soc) - { - set_block_to_mat2d(iw2start + iw, iw1start + iw, - phase_factor * this->rotmat_Slm_[isym][l], M_isym, pv, true); - } - else - { - // M = T(isym) (x) U(isym): scatter phase * T_l(m,m') * U(a,b) to the interleaved - // spinor positions (row = rotated atom/spin, col = original atom/spin). For nspin=4 - // stapos_wf already carries the npol factor, so the per-atom offset is ia*nw*npol - // and the within-atom spinor index is (iw_orb)*npol + spin (spin is the fast index). - const int base2 = atoms[it].stapos_wf + ia2 * atoms[it].nw * npol; - const int base1 = atoms[it].stapos_wf + ia1 * atoms[it].nw * npol; - const RI::Tensor>& Tl = this->rotmat_Slm_[isym][l]; - for (int m = 0;m < nm;++m) - { - for (int mp = 0;mp < nm;++mp) - { - const std::complex t = phase_factor * Tl(m, mp); - for (int a = 0;a < npol;++a) - { - for (int b = 0;b < npol;++b) - { - const int gi = base2 + (iw + m) * npol + a; - const int gj = base1 + (iw + mp) * npol + b; - if (pv.in_this_processor(gi, gj)) - { - const int index = pv.global2local_col(gj) * pv.get_row_size() + pv.global2local_row(gi); - // M(isym) = T_l (x) U is the spinor rep, with U = so3_to_su2 placed as-is: - // M[(m,a),(m',b)] = phase * T_l(m,m') * U_{ab}, U_{ab} = spin_U[a*npol + b]. - // Both T_l (rotmat_Slm) and U are ANTI-homomorphisms here (row-vector / R^T convention: - // rotmat_Slm(g)=R_orb(g)^{-1}, so3_to_su2 likewise), so this M is a consistent rep - // and rot_matrix_ao's stored-DM rotation M^T D M^* is exact for ALL ops. - M_isym[index] = t * spin_U[a * npol + b]; - } - } - } - } - } - } - iw += nm; - } - } - return M_isym; - } - - // D(k) = M^T(R, k) D(k_ibz) M^*(R, k), if D(k) is col-maj - // D^T(k) = M^\dagger(R, k) D^T(k_ibz) M(R, k), if D(k) is row-maj - // Ds from RI_2D_Comm are row-maj - // the link ik_ibz-isym-ik can be found in kstars. - std::vector> Symmetry_rotation::rot_matrix_ao(const std::vector>& DMkibz, - const int ik_ibz, const int kstar_size, const int isym, const Parallel_2D& pv, const bool TRS_conj) const - { - std::vector> DMk(pv.nloc, 0.0); - std::vector> DMkibz_M(pv.nloc, 0.0); // intermediate result - const char dagger = 'C'; - const char transpose = 'T'; - const char notrans = 'N'; - std::complex alpha(1.0, 0.0); - const std::complex beta(0.0, 0.0); - const int nbasis = pv.get_global_row_size(); - const int i1 = 1; - if (TRS_conj) - { - // D^T* = M^T [M^T (D^T)^T]^\dagger - ScalapackConnector::gemm(transpose, transpose, nbasis, nbasis, nbasis, - alpha, this->Ms_[ik_ibz].at(isym).data(), i1, i1, pv.desc, DMkibz.data(), i1, i1, pv.desc, - beta, DMkibz_M.data(), i1, i1, pv.desc); - alpha.real(1.0 / static_cast(kstar_size)); - ScalapackConnector::gemm(transpose, dagger, nbasis, nbasis, nbasis, - alpha, this->Ms_[ik_ibz].at(isym).data(), i1, i1, pv.desc, DMkibz_M.data(), i1, i1, pv.desc, - beta, DMk.data(), i1, i1, pv.desc); - } - else - { - // Physical DM rotation D(k) = M^dagger D(k_ibz) M, with M = T (x) U is the anti-homomorphism rep in row-major convention. - // ABACUS stores the DM transposed (S = D^T), for which this becomes S(gk) = M^T S(k_ibz) M^* = (conj M)^dagger S (conj M) - // For nspin<4 the orbital-only M is real, so Mc = M and this is bit-identical to the old M^dagger D M. - const std::vector>& Mref = this->Ms_[ik_ibz].at(isym); - std::vector> Mc(Mref.size()); - for (size_t i = 0; i < Mref.size(); ++i) { Mc[i] = std::conj(Mref[i]); } - ScalapackConnector::gemm(dagger, notrans, nbasis, nbasis, nbasis, - alpha, Mc.data(), i1, i1, pv.desc, DMkibz.data(), i1, i1, pv.desc, - beta, DMkibz_M.data(), i1, i1, pv.desc); - alpha.real(1.0 / static_cast(kstar_size)); - ScalapackConnector::gemm(notrans, notrans, nbasis, nbasis, nbasis, - alpha, DMkibz_M.data(), i1, i1, pv.desc, Mc.data(), i1, i1, pv.desc, - beta, DMk.data(), i1, i1, pv.desc); - } - return DMk; - } - - std::vector> Symmetry_rotation::set_sigma_y_2d(const Parallel_2D& pv) const - { - std::vector> sigma_y(pv.get_local_size(), 0.0); - const int nlocal = pv.get_global_row_size(); // = 2*nao for nspin=4 - // sigma_y = [[0, -i], [i, 0]] on the interleaved spin index (I = 2*iorb + spin) - const std::complex sy[2][2] = { {std::complex(0.0, 0.0), std::complex(0.0, -1.0)}, - {std::complex(0.0, 1.0), std::complex(0.0, 0.0)} }; - for (int iorb = 0; 2 * iorb < nlocal; ++iorb) - { - for (int a = 0; a < 2; ++a) - { - const int b = 1 - a; // only the off-diagonal spin entries are non-zero - const int gi = 2 * iorb + a; - const int gj = 2 * iorb + b; - if (pv.in_this_processor(gi, gj)) - { - const int index = pv.global2local_col(gj) * pv.get_row_size() + pv.global2local_row(gi); - sigma_y[index] = sy[a][b]; - } - } - } - return sigma_y; - } - - std::vector> Symmetry_rotation::trs_spin_rotate(const std::vector>& X, - const std::vector>& sigma_y, const Parallel_2D& pv, const double scale) const - { - // stored (transposed 2d-block) form of D_new = sigma_y * conj(D) * sigma_y is - // Sigma_y * conj(X) * Sigma_y (Sigma_y^T = -Sigma_y, the two minus signs cancel). - const char notrans = 'N'; - const int nbasis = pv.get_global_row_size(); - const int i1 = 1; - const std::complex one(1.0, 0.0); - const std::complex beta(0.0, 0.0); - std::vector> Xc(X.size()); - for (size_t i = 0; i < X.size(); ++i) { Xc[i] = std::conj(X[i]); } - std::vector> tmp(pv.get_local_size(), 0.0); - std::vector> out(pv.get_local_size(), 0.0); - // tmp = Sigma_y * conj(X) - ScalapackConnector::gemm(notrans, notrans, nbasis, nbasis, nbasis, - one, sigma_y.data(), i1, i1, pv.desc, Xc.data(), i1, i1, pv.desc, - beta, tmp.data(), i1, i1, pv.desc); - // out = scale * tmp * Sigma_y - ScalapackConnector::gemm(notrans, notrans, nbasis, nbasis, nbasis, - std::complex(scale, 0.0), tmp.data(), i1, i1, pv.desc, sigma_y.data(), i1, i1, pv.desc, - beta, out.data(), i1, i1, pv.desc); - return out; - } std::vector Symmetry_rotation::get_Rs_from_adjacent_list(const UnitCell& ucell, const Grid_Driver& gd, diff --git a/source/source_lcao/module_ri/module_exx_symmetry/symm_rotation.h b/source/source_lcao/module_ri/module_exx_symmetry/symm_rotation.h index 1d27edfef2f..16d7ba91928 100644 --- a/source/source_lcao/module_ri/module_exx_symmetry/symm_rotation.h +++ b/source/source_lcao/module_ri/module_exx_symmetry/symm_rotation.h @@ -1,119 +1,30 @@ -#pragma once -#include "irreducible_sector.h" +#ifndef ABACUS_SOURCE_LCAO_MODULE_RI_MODULE_EXX_SYMMETRY_SYMM_ROTATION_H +#define ABACUS_SOURCE_LCAO_MODULE_RI_MODULE_EXX_SYMMETRY_SYMM_ROTATION_H + +#include "source_cell/module_symmetry/symm_rotation_k.h" #include "source_basis/module_ao/parallel_orbitals.h" #include #include "source_hamilt/module_hcontainer/hcontainer.h" #include "source_cell/module_neighbor/sltk_grid_driver.h" -#include "source_cell/module_symmetry/symm_rot_spin.h" namespace ModuleSymmetry { - using Tap = std::pair; - using TC = std::array; - using TapR = std::pair; - using TCdouble = Abfs::Vector3_Order; - - class Symmetry_rotation + /// Real-space (RI::Tensor / HContainer) H(R) and RI-coefficient symmetry restoration for + /// EXX/RPA, built on top of the LibRI-independent k-space restoration in + /// ModuleSymmetry::Symmetry_rotation_k (source_cell/module_symmetry/symm_rotation_k.h), + /// which provides cal_Ms/restore_dm/rot_matrix_ao and the shared rotation-matrix machinery + /// (rotmat_Slm_, irs_, Ms_, spin_U_, ...). Only the parts that genuinely need LibRI (RI::Tensor + /// atom-pair maps, HContainer real-space rotation) live here. + class Symmetry_rotation : public Symmetry_rotation_k { public: Symmetry_rotation() {}; ~Symmetry_rotation() {}; - //-------------------------------------------------------------------------------- - // getters - const std::map>& get_irreducible_sector()const { return this->irs_.get_irreducible_sector(); } - TCdouble get_return_lattice(const Symmetry& symm, - const ModuleBase::Matrix3& gmatd, const TCdouble gtransd, - const TCdouble& posd_a1, const TCdouble& posd_a2)const - { - return this->irs_.get_return_lattice(symm, gmatd, gtransd, posd_a1, posd_a2); - } - TCdouble get_return_lattice(const int iat, const int isym) const - { - return this->irs_.get_return_lattice(iat, isym); - } - /// the rotation matrix under the basis of S_l^m. size: [nsym][lmax][nm*nm] - const std::vector>>>& rotmat_Slm = this->rotmat_Slm_; - const int& abfs_Lmax = this->abfs_Lmax_; //-------------------------------------------------------------------------------- // setters - void find_irreducible_sector(const Symmetry& symm, const Atom* atoms, const Statistics& st, - const std::vector& Rs, const TC& period, const Lattice& lat) - { - this->irs_.find_irreducible_sector(symm, atoms, st, Rs, period, lat); - } - void set_abfs_Lmax(const int l) { this->abfs_Lmax_ = l; } void set_Cs_rotation(const std::vector>& abfs_l_nchi); //-------------------------------------------------------------------------------- - /// functions to contruct rotation matrix in AO-representation - - /// The top-level calculation interface of this class. calculate the rotation matrix in AO representation: M - /// only need once call in each ion step (decided by the configuration) - /// @param kstars equal k points to each ibz-kpont, corresponding to a certain symmetry operations. - void cal_Ms(const K_Vectors& kv, - //const std::vector>& kstars, - const UnitCell& ucell, const Parallel_2D& pv); - - /// Use calculated M matrix to recover D(k) from D(k_ibz): D(k) = M(R, k)^\dagger D(k_ibz) M(R, k) - /// the link "ik_ibz-isym-ik" can be found in kstars: k_bz = gmat[isym](k) - std::vector>>restore_dm(const K_Vectors& kv, - const std::vector>>& dm_k_ibz, - const Parallel_2D& pv)const; - std::vector>restore_dm(const K_Vectors& kv, - const std::vector>& dm_k_ibz, - const Parallel_2D& pv)const; - std::vector> rot_matrix_ao(const std::vector>& DMkibz, - const int ik_ibz, const int kstar_size, const int isym, const Parallel_2D& pv, const bool TRS_conj = false) const; - - /// (nspin=4) build the 2*nao spin operator Sigma_y = I_nao (x) sigma_y in 2d-block layout. - std::vector> set_sigma_y_2d(const Parallel_2D& pv) const; - - /// (nspin=4) time-reversal on the spin density matrix: D(k) = sigma_y D^*(-k) sigma_y, - /// realized distribution-safely as scale * Sigma_y * conj(X) * Sigma_y (X is the already - /// space-group-rotated D(-k) stored in the transposed 2d-block convention). - std::vector> trs_spin_rotate(const std::vector>& X, - const std::vector>& sigma_y, const Parallel_2D& pv, const double scale) const; - - /// Inject synthetic AO rotations for density-restoration regression tests. - void set_density_rotations_for_testing( - const std::vector>>>& rotations, - const std::vector>& little_groups, - const int nrot) - { - this->Ms_ = rotations; - this->little_groups_ = little_groups; - this->nsym_ = nrot; - } - - /// calculate Wigner D matrix - double wigner_d(const double beta, const int l, const int m1, const int m2) const; - std::complex wigner_D(const TCdouble& euler_angle, const int l, const int m1, const int m2, const bool inv) const; - - /// c^l_{m1, m2}= - std::complex ovlp_Ylm_Slm(const int l, const int m1, const int m2) const; - - /// calculate euler angle from rotation matrix - TCdouble get_euler_angle(const ModuleBase::Matrix3& gmatc) const; - - /// T_mm' = [c^\dagger D c]_mm', the rotation matrix in the representation of real sphere harmonics - /// @param nop number of operations in gmatc; <0 means nsym_ (the unitary ones only). - /// Pass nsym_+nanti_ to also build the antiunitary operations' T_l. - void cal_rotmat_Slm(const ModuleBase::Matrix3* gmatc, const int lmax, const int nop); - - /// set a block matrix onto a 2d-parallelized matrix(col-maj), at the position (starti, startj) - /// if trans=true, the block matrix is transposed before setting - void set_block_to_mat2d(const int starti, const int startj, const RI::Tensor>& block, - std::vector>& obj_mat, const Parallel_2D& pv, const bool trans = false) const; - void set_block_to_mat2d(const int starti, const int startj, const RI::Tensor>& block, - std::vector& obj_mat, const Parallel_2D& pv, const bool trans = false) const; - - /// 2d-block parallized rotation matrix in AO-representation, denoted as M. - /// finally we will use D(k)=M(R, k)^\dagger*D(Rk)*M(R, k) to recover D(k) from D(Rk). - std::vector> contruct_2d_rot_mat_ao(const Symmetry& symm, const Atom* atoms, const Statistics& cell_st, - const TCdouble& kvec_d_ibz, int isym, const Parallel_2D& pv, - const SpinRotation::Su2& spin_U /*= SpinRotation::Su2{ 1.0, 0.0, 0.0, 1.0 }*/) const; - - std::vector>>>& get_rotmat_Slm() { return this->rotmat_Slm_; } //-------------------------------------------------------------------------------- /// The main functions to rotate matrices @@ -145,7 +56,7 @@ namespace ModuleSymmetry template // RI::Tensor type, using col-major implementation void test_HR_rotation(const Symmetry& symm, const Atom* atoms, const Statistics& st, const char mode, const std::map, RI::Tensor>>& HR_full); - template // test the rotation of RI coefficients + template // test the rotation of RI coefficients void test_Cs_rotation(const Symmetry& symm, const Atom* atoms, const Statistics& st, const std::map, RI::Tensor>>& Cs_full)const; template // HContainer type, using row-major implementation @@ -185,53 +96,19 @@ namespace ModuleSymmetry RI::Tensor set_rotation_matrix(const Atom& a, const int& isym)const; template RI::Tensor set_rotation_matrix_abf(const int& type, const int& isym)const; - //-------------------------------------------------------------------------------- - - int nsym_ = 1; - /// (nspin=4, magnetic) number of ANTIUNITARY elements Theta*g of the Shubnikov group. - /// Their orbital rotations / return lattices / Ms are appended after the nsym_ unitary - /// ones, so the raw index isym in [nsym_, nsym_+nanti_) addresses gmatrix_anti[isym-nsym_]. - int nanti_ = 0; - /// (nspin=4) true when the configuration carries a non-zero local moment. Then pure time - /// reversal is NOT a symmetry (it reverses m) and the k-star must be restored with the - /// Shubnikov elements Theta*gmatrix_anti[] instead of the generic -k shortcut. - bool magnetic_nspin4_ = false; - double eps_ = 1e-6; - - // (removed, not needed) TRS_first_: - // it used to short-circuit any star member equal to -k to pure time reversal, - // which silently pre-empted the genuine space-group operation that produced it. - // The operation is now decided by the index alone: isym=nsym_ antiunitary. - // A -k member reached through the TRS doubling lands on the antiunitary branch with M=I, - // which reduces exactly to the direct conjugation. + /// RI::Tensor mirror of rotmat_Slm_ (which is stored as ModuleBase::ComplexMatrix, shared + /// with the LibRI-free k-space code), rebuilt lazily and cached across the many + /// set_rotation_matrix/set_rotation_matrix_abf calls within one ion step (one per atom + /// pair/cell), instead of reconverting the same small block every time. + const RI::Tensor>& get_rotmat_Slm_tensor(const int isym, const int l)const; + mutable std::vector>>> rotmat_Slm_tensor_; + mutable int rotmat_Slm_tensor_version_ = -1; + //-------------------------------------------------------------------------------- bool reduce_Cs_ = false; - int abfs_Lmax_ = 0; std::vector> abfs_l_nchi_;///< number of abfs for each angular momentum - - /// the rotation matrix under the basis of S_l^m. size: [nsym][lmax][nm*nm] - std::vector>>> rotmat_Slm_; - // [natom][nsym], phase factor corresponding to a certain kvec_d_ibz - // std::vector>> phase_factor_; - - /// The unitary matrix associate D(Rk) with D(k) for each ibz-kpoint Rk and each symmetry operation. - /// size: [nks_ibz][nsym][nbasis*nbasis], only need to calculate once. - std::vector>>> Ms_; - - /// Unitary operations fixing each IBZ k point modulo reciprocal lattice vectors. - /// Geometry data built with Ms_ in cal_Ms, not an SCF workflow switch. - std::vector> little_groups_; - - /// (nspin=4) the SU(2) spin-1/2 rotation U(isym) for each symmetry operation, size [nsym]. - /// The spinor AO rotation is T(isym) (x) U(isym); restore_HR_nspin4 uses it to mix the 4 spin - /// channels of the real-space EXX H(R). Filled in cal_Ms (identity for nspin<4). - std::vector spin_U_; - - /// irreducible sector - Irreducible_Sector irs_; - }; template std::string vec3_fmt(const T& x, const T& y, const T& z) @@ -250,4 +127,6 @@ namespace ModuleSymmetry } #include "symm_rotation_r.hpp" -#include "symm_rotation_r_hcontainer.hpp" \ No newline at end of file +#include "symm_rotation_r_hcontainer.hpp" + +#endif // ABACUS_SOURCE_LCAO_MODULE_RI_MODULE_EXX_SYMMETRY_SYMM_ROTATION_H diff --git a/source/source_lcao/module_ri/module_exx_symmetry/symm_rotation_r.hpp b/source/source_lcao/module_ri/module_exx_symmetry/symm_rotation_r.hpp index 6c7cd64f8a3..381aaf5f435 100644 --- a/source/source_lcao/module_ri/module_exx_symmetry/symm_rotation_r.hpp +++ b/source/source_lcao/module_ri/module_exx_symmetry/symm_rotation_r.hpp @@ -6,6 +6,38 @@ #include namespace ModuleSymmetry { + /// rotmat_Slm_ (inherited from Symmetry_rotation_k, shared with the LibRI-free k-space + /// restoration) is stored as ModuleBase::ComplexMatrix; convert to RI::Tensor here, at the + /// boundary of the real-space (RI::Tensor atom-pair) rotation code that needs that type. + inline RI::Tensor> rotmat_Slm_to_tensor(const ModuleBase::ComplexMatrix& block) + { + RI::Tensor> t({ static_cast(block.nr), static_cast(block.nc) }); + for (int i = 0; i < block.nr; ++i) { + for (int j = 0; j < block.nc; ++j) { + t(i, j) = block(i, j); + } } + return t; + } + + /// Cached (isym, l) lookup into rotmat_Slm_tensor_: rebuilt only when rotmat_Slm_version_ + /// (bumped by cal_rotmat_Slm, i.e. once per ion step) changes, instead of reconverting the + /// same block on every set_rotation_matrix/set_rotation_matrix_abf call (once per atom pair). + inline const RI::Tensor>& Symmetry_rotation::get_rotmat_Slm_tensor(const int isym, const int l)const + { + if (this->rotmat_Slm_tensor_version_ != this->rotmat_Slm_version_) + { + this->rotmat_Slm_tensor_.resize(this->rotmat_Slm_.size()); + for (size_t is = 0; is < this->rotmat_Slm_.size(); ++is) + { + this->rotmat_Slm_tensor_[is].resize(this->rotmat_Slm_[is].size()); + for (size_t il = 0; il < this->rotmat_Slm_[is].size(); ++il) { + this->rotmat_Slm_tensor_[is][il] = rotmat_Slm_to_tensor(this->rotmat_Slm_[is][il]); + } } + this->rotmat_Slm_tensor_version_ = this->rotmat_Slm_version_; + } + return this->rotmat_Slm_tensor_[isym][l]; + } + /// Elementwise complex conjugation used by the time-reversal branch of restore_HR_nspin4. /// Overloaded (not specialized) so a real Tdata compiles to the identity. inline float conj_elem(const float v) { return v; } @@ -268,7 +300,7 @@ namespace ModuleSymmetry { int l = a.iw2l[iw]; int nm = 2 * l + 1; - set_block(iw, iw, this->rotmat_Slm_[isym][l], T); + set_block(iw, iw, this->get_rotmat_Slm_tensor(isym, l), T); iw += nm; } return T; @@ -328,7 +360,7 @@ namespace ModuleSymmetry int nm = 2 * L + 1; for (int N = 0;N < this->abfs_l_nchi_[type][L];++N) { - set_block(iw, iw, this->rotmat_Slm_[isym][L], T); + set_block(iw, iw, this->get_rotmat_Slm_tensor(isym, L), T); iw += nm; // std::cout << "L=" << L << ", N=" << N << ", iw=" << iw << "\n"; } diff --git a/source/source_lcao/module_ri/module_exx_symmetry/test/CMakeLists.txt b/source/source_lcao/module_ri/module_exx_symmetry/test/CMakeLists.txt index ba45e581124..46fe581b419 100644 --- a/source/source_lcao/module_ri/module_exx_symmetry/test/CMakeLists.txt +++ b/source/source_lcao/module_ri/module_exx_symmetry/test/CMakeLists.txt @@ -4,7 +4,7 @@ abacus_disable_feature_definitions(__ROCM) AddTest( TARGET MODULE_RI_EXX_SYMMETRY_rotation LIBS base device symmetry neighbor parameter - SOURCES symm_rotation_test.cpp test_symm_rotation.cpp ../symm_rotation.cpp ../symm_rot_out.cpp ../irreducible_sector.cpp ../irred_sec_bvk.cpp + SOURCES symm_rotation_test.cpp test_symm_rotation.cpp ../symm_rotation.cpp ../symm_rot_out.cpp ../../../../source_basis/module_ao/parallel_orbitals.cpp ) \ No newline at end of file diff --git a/source/source_lcao/module_ri/module_exx_symmetry/test/symm_rotation_test.cpp b/source/source_lcao/module_ri/module_exx_symmetry/test/symm_rotation_test.cpp index e3dc3ef3db5..483df5d894a 100644 --- a/source/source_lcao/module_ri/module_exx_symmetry/test/symm_rotation_test.cpp +++ b/source/source_lcao/module_ri/module_exx_symmetry/test/symm_rotation_test.cpp @@ -116,7 +116,7 @@ TEST_F(SymmetryRotationTest, OvlpYS) TEST_F(SymmetryRotationTest, RotMat) { symrot.cal_rotmat_Slm(&C41, 1, -1); - RI::Tensor>& rotmat = symrot.get_rotmat_Slm()[0][1]; + ModuleBase::ComplexMatrix& rotmat = symrot.get_rotmat_Slm()[0][1]; int l = 1; for (int m1 = -l;m1 <= l;++m1) for (int m2 = -l;m2 <= l;++m2) @@ -145,7 +145,7 @@ TEST_F(SymmetryRotationTest, SetBlockToMat2d) for (int j = 0;j < pv.get_col_size();++j) for (int i = 0;i < pv.get_row_size();++i) obj_mat[j * pv.get_row_size() + i] = std::complex(static_cast(pv.local2global_row(i)), static_cast(pv.local2global_col(j))); - RI::Tensor> block({ 2, 2 }); + ModuleBase::ComplexMatrix block(2, 2); block(0, 0) = 0; block(0, 1) = -1; block(1, 0) = -2; block(1, 1) = -3; symrot.set_block_to_mat2d(2, 3, block, obj_mat, pv); for (int i = 2;i < 4;++i) diff --git a/source/source_lcao/module_ri/module_exx_symmetry/test/test_symm_rotation.cpp b/source/source_lcao/module_ri/module_exx_symmetry/test/test_symm_rotation.cpp index 01ec4938bfb..64b8ad020ca 100644 --- a/source/source_lcao/module_ri/module_exx_symmetry/test/test_symm_rotation.cpp +++ b/source/source_lcao/module_ri/module_exx_symmetry/test/test_symm_rotation.cpp @@ -1,22 +1,6 @@ #include "../symm_rotation.h" -#include "source_io/module_parameter/parameter.h" #include "gtest/gtest.h" -class TestParameters -{ - public: - TestParameters(Parameter& parameters, const int nspin) - : parameters_(parameters), original_nspin_(parameters.inp.nspin) - { - parameters_.input.nspin = nspin; - } - ~TestParameters() { parameters_.input.nspin = original_nspin_; } - - private: - Parameter& parameters_; - const int original_nspin_; -}; - // K-point generation is outside this test: use explicit stars, but provide // the virtual symbols needed by the existing lightweight rotation test target. void ModuleCell::ReciprocalGrid::renew(const int&) @@ -62,7 +46,6 @@ std::vector rotate_reference(const std::vector& density, void check_little_group_restoration(const int nspin) { - const TestParameters parameters(PARAM, nspin); const int channels = nspin == 2 ? 2 : 1; const int n = 4; Parallel_2D pv; @@ -98,10 +81,15 @@ void check_little_group_restoration(const int nspin) }; rotation.set_density_rotations_for_testing( {{{0, local(identity)}, {1, local(little)}, {2, local(representative)}, {3, local(alternate)}}}, - {{0, 1}}, 4); + {{0, 1}}, 4, nspin); K_Vectors kv; kv.set_nkstot(channels); kv.set_nkstot_nospin(2); + // Single-pool (KPAR=1) scenario: kv.get_nks() (local) equals kv.get_nkstot() (global), + // and ik2iktot is the identity map. With only one global ibz-k here, any value mod + // kv.kstars.size()==1 is 0, so the exact ik2iktot values don't matter, only its size. + kv.set_nks(channels); + kv.ik2iktot.assign(channels, 0); kv.kstars = {{{0, {0.25, 0.0, 0.0}}, {2, {0.0, 0.25, 0.0}}}}; std::vector> inputs; std::vector> expected; diff --git a/source/source_lcao/module_ri/ri_2d_comm.cpp b/source/source_lcao/module_ri/ri_2d_comm.cpp index 498907a9343..ad26a2daa3a 100644 --- a/source/source_lcao/module_ri/ri_2d_comm.cpp +++ b/source/source_lcao/module_ri/ri_2d_comm.cpp @@ -15,59 +15,124 @@ auto RI_2D_Comm::get_2D_judge(const UnitCell& ucell, const Parallel_2D& pv) -> std::vector, std::set>> { - ModuleBase::TITLE("RI_2D_Comm","get_2D_judge"); + ModuleBase::TITLE("RI_2D_Comm","get_2D_judge"); - const std::map nspin_b = {{1,1}, {2,1}, {4,2}}; + const std::map nspin_b = {{1,1}, {2,1}, {4,2}}; - std::vector> iat0_list(nspin_b.at(PARAM.inp.nspin)); + std::vector> iat0_list(nspin_b.at(PARAM.inp.nspin)); for (int iwt0_2D = 0; iwt0_2D < pv.nrow; ++iwt0_2D) - { + { const int iwt0 = pv.local2global_row(iwt0_2D); - int iat0=0;int iw0_b=0;int is0_b=0; - std::tie(iat0,iw0_b,is0_b) = RI_2D_Comm::get_iat_iw_is_block(ucell,iwt0); - iat0_list[is0_b].insert(iat0); - } + int iat0=0;int iw0_b=0;int is0_b=0; + std::tie(iat0,iw0_b,is0_b) = RI_2D_Comm::get_iat_iw_is_block(ucell,iwt0); + iat0_list[is0_b].insert(iat0); + } - std::vector> iat1_list(nspin_b.at(PARAM.inp.nspin)); + std::vector> iat1_list(nspin_b.at(PARAM.inp.nspin)); for (int iwt1_2D = 0; iwt1_2D < pv.ncol; ++iwt1_2D) - { + { const int iwt1 = pv.local2global_col(iwt1_2D); - int iat1=0;int iw1_b=0;int is1_b=0; - std::tie(iat1,iw1_b,is1_b) = RI_2D_Comm::get_iat_iw_is_block(ucell,iwt1); - iat1_list[is1_b].insert(iat1); - } - - std::vector, std::set>> judge(PARAM.inp.nspin); - switch(PARAM.inp.nspin) - { - case 1: - judge[0] = std::make_tuple( std::move(iat0_list[0]), std::move(iat1_list[0]) ); - break; - case 2: - judge[0] = judge[1] = std::make_tuple( std::move(iat0_list[0]), std::move(iat1_list[0]) ); - break; - case 4: - for(int is0_b=0; is0_b<2; ++is0_b) - for(int is1_b=0; is1_b<2; ++is1_b) - { - const int is_b = RI_2D_Comm::get_is_block(-1, is0_b, is1_b); - judge[is_b] = std::make_tuple( iat0_list[is0_b], iat1_list[is1_b] ); - } - break; - default: - throw std::invalid_argument(std::string(__FILE__)+" line "+std::to_string(__LINE__)); - } - return judge; + int iat1=0;int iw1_b=0;int is1_b=0; + std::tie(iat1,iw1_b,is1_b) = RI_2D_Comm::get_iat_iw_is_block(ucell,iwt1); + iat1_list[is1_b].insert(iat1); + } + + std::vector, std::set>> judge(PARAM.inp.nspin); + switch(PARAM.inp.nspin) + { + case 1: + judge[0] = std::make_tuple( std::move(iat0_list[0]), std::move(iat1_list[0]) ); + break; + case 2: + judge[0] = judge[1] = std::make_tuple( std::move(iat0_list[0]), std::move(iat1_list[0]) ); + break; + case 4: + for(int is0_b=0; is0_b<2; ++is0_b) + for(int is1_b=0; is1_b<2; ++is1_b) + { + const int is_b = RI_2D_Comm::get_is_block(-1, is0_b, is1_b); + judge[is_b] = std::make_tuple( iat0_list[is0_b], iat1_list[is1_b] ); + } + break; + default: + throw std::invalid_argument(std::string(__FILE__)+" line "+std::to_string(__LINE__)); + } + return judge; } std::vector RI_2D_Comm::get_ik_list(const K_Vectors &kv, const int is_k) { - std::vector ik_list; - for(int ik=0; ik ik_list; + for(int ik=0; ik +RI_2D_Comm::get_iat_iw_is_block(const UnitCell& ucell,const int& iwt) +{ + const int iat = ucell.iwt2iat[iwt]; + const int iw = ucell.iwt2iw[iwt]; + switch(PARAM.inp.nspin) + { + case 1: case 2: + return std::make_tuple(iat, iw, 0); + case 4: + return std::make_tuple(iat, iw/2, iw%2); + default: + throw std::invalid_argument(std::string(__FILE__)+" line "+std::to_string(__LINE__)); + } +} + +int RI_2D_Comm::get_is_block(const int is_k, const int is_row_b, const int is_col_b) +{ + switch(PARAM.inp.nspin) + { + case 1: return 0; + case 2: return is_k; + case 4: return is_row_b*2+is_col_b; + default: throw std::invalid_argument(std::string(__FILE__)+" line "+std::to_string(__LINE__)); + } +} + +std::tuple +RI_2D_Comm::split_is_block(const int is_b) +{ + switch(PARAM.inp.nspin) + { + case 1: case 2: + return std::make_tuple(0, 0); + case 4: + return std::make_tuple(is_b/2, is_b%2); + default: + throw std::invalid_argument(std::string(__FILE__)+" line "+std::to_string(__LINE__)); + } +} + + + +int RI_2D_Comm::get_iwt(const UnitCell& ucell, + const int iat, + const int iw_b, + const int is_b) +{ + const int it = ucell.iat2it[iat]; + const int ia = ucell.iat2ia[iat]; + int iw=-1; + switch(PARAM.inp.nspin) + { + case 1: case 2: + iw = iw_b; break; + case 4: + iw = iw_b*2+is_b; break; + default: + throw std::invalid_argument(std::string(__FILE__)+" line "+std::to_string(__LINE__)); + } + const int iwt = ucell.itiaiw2iwt(it,ia,iw); + return iwt; } diff --git a/source/source_lcao/module_ri/ri_2d_comm.h b/source/source_lcao/module_ri/ri_2d_comm.h index df24ecd3527..0720ad7ccbe 100644 --- a/source/source_lcao/module_ri/ri_2d_comm.h +++ b/source/source_lcao/module_ri/ri_2d_comm.h @@ -54,8 +54,8 @@ extern std::vector>>> split_m2D_kto const int nspin, const bool spgsym = false); - // judge[is] = {s0, s1} - extern std::vector, std::set>> + // judge[is] = {s0, s1} + extern std::vector, std::set>> get_2D_judge(const UnitCell& ucell, const Parallel_2D& pv); template @@ -78,22 +78,22 @@ extern std::vector>>> split_m2D_kto hamilt::HContainer& HlocR, const RI::Cell_Nearest* const cell_nearest = nullptr); - template - extern std::vector> Hexxs_to_Hk( - const K_Vectors &kv, - const Parallel_Orbitals &pv, - const std::vector< std::map>>> &Hexxs, - const int ik); - template - std::vector> pulay_mixing( - const Parallel_Orbitals &pv, - std::deque>> &Hk_seq, - const std::vector> &Hk_new, - const double mixing_beta, - const std::string mixing_mode); + template + extern std::vector> Hexxs_to_Hk( + const K_Vectors &kv, + const Parallel_Orbitals &pv, + const std::vector< std::map>>> &Hexxs, + const int ik); + template + std::vector> pulay_mixing( + const Parallel_Orbitals &pv, + std::deque>> &Hk_seq, + const std::vector> &Hk_new, + const double mixing_beta, + const std::string mixing_mode); // DM(R) format conversion: the real-space (DM(R)) counterpart of split_m2D_ktoR, - // and the inverse of add_HexxR. dm_container is DensityMatrix::get_DMR_vector(): + // and the inverse of add_HexxR. dm_container is DensityMatrix::get_dmr_vec(): // nspin==1 : size 1 (container 0 -> spin-block 0) // nspin==2 : size 2 (container is -> spin-block is) // nspin==4 : size 1 (container 0 holds the 2x2 npol blocks -> spin-blocks 0,1,2,3) @@ -105,11 +105,11 @@ extern std::vector>>> split_m2D_kto const int nspin); //private: - extern std::vector get_ik_list(const K_Vectors &kv, const int is_k); - extern inline std::tuple get_iat_iw_is_block(const UnitCell& ucell,const int& iwt); - extern inline int get_is_block(const int is_k, const int is_row_b, const int is_col_b); - extern inline std::tuple split_is_block(const int is_b); - extern inline int get_iwt(const UnitCell& ucell, const int iat, const int iw_b, const int is_b); + extern std::vector get_ik_list(const K_Vectors &kv, const int is_k); + extern std::tuple get_iat_iw_is_block(const UnitCell& ucell,const int& iwt); + extern int get_is_block(const int is_k, const int is_row_b, const int is_col_b); + extern std::tuple split_is_block(const int is_b); + extern int get_iwt(const UnitCell& ucell, const int iat, const int iw_b, const int is_b); template extern std::map> comm_map2_first(const MPI_Comm& mpi_comm, diff --git a/source/source_lcao/module_ri/ri_2d_comm.hpp b/source/source_lcao/module_ri/ri_2d_comm.hpp index 64cdc8786e5..2d8a6294b08 100644 --- a/source/source_lcao/module_ri/ri_2d_comm.hpp +++ b/source/source_lcao/module_ri/ri_2d_comm.hpp @@ -44,15 +44,15 @@ auto RI_2D_Comm::split_m2D_ktoR(const UnitCell& ucell, const bool spgsym) -> std::vector>>> { - ModuleBase::TITLE("RI_2D_Comm","split_m2D_ktoR"); - ModuleBase::timer::start("RI_2D_Comm", "split_m2D_ktoR"); - const TC period = RI_Util::get_Born_vonKarmen_period(kv); + ModuleBase::TITLE("RI_2D_Comm","split_m2D_ktoR"); + ModuleBase::timer::start("RI_2D_Comm", "split_m2D_ktoR"); + const TC period = RI_Util::get_Born_vonKarmen_period(kv); std::vector>>> mRs_a2D = (period == TC{1, 1, 1}) ? RI_2D_Comm::split_m2D_ktoR_gamma(ucell, mks_2D, pv, nspin) : RI_2D_Comm::split_m2D_ktoR_k(ucell, kv, mks_2D, pv, nspin, spgsym); - ModuleBase::timer::end("RI_2D_Comm", "split_m2D_ktoR"); - return mRs_a2D; + ModuleBase::timer::end("RI_2D_Comm", "split_m2D_ktoR"); + return mRs_a2D; } template @@ -62,10 +62,10 @@ auto RI_2D_Comm::split_m2D_ktoR_gamma(const UnitCell& ucell, const int nspin) -> std::vector>>> { - ModuleBase::TITLE("RI_2D_Comm","split_m2D_ktoR_gamma"); - ModuleBase::timer::start("RI_2D_Comm", "split_m2D_ktoR_gamma"); + ModuleBase::TITLE("RI_2D_Comm","split_m2D_ktoR_gamma"); + ModuleBase::timer::start("RI_2D_Comm", "split_m2D_ktoR_gamma"); - const std::map nspin_k = {{1,1}, {2,2}, {4,1}}; + const std::map nspin_k = {{1,1}, {2,2}, {4,1}}; const double SPIN_multiple = std::map{ {1,0.5}, {2,1}, {4,1} }.at(nspin); // why? const TC cell = {0, 0, 0}; @@ -144,8 +144,8 @@ auto RI_2D_Comm::split_m2D_ktoR_gamma(const UnitCell& ucell, } #endif - ModuleBase::timer::end("RI_2D_Comm", "split_m2D_ktoR_gamma"); - return mRs_a2D; + ModuleBase::timer::end("RI_2D_Comm", "split_m2D_ktoR_gamma"); + return mRs_a2D; } template @@ -157,11 +157,11 @@ auto RI_2D_Comm::split_m2D_ktoR_k(const UnitCell& ucell, const bool spgsym) -> std::vector>>> { - ModuleBase::TITLE("RI_2D_Comm","split_m2D_ktoR_k"); - ModuleBase::timer::start("RI_2D_Comm", "split_m2D_ktoR_k"); + ModuleBase::TITLE("RI_2D_Comm","split_m2D_ktoR_k"); + ModuleBase::timer::start("RI_2D_Comm", "split_m2D_ktoR_k"); - const TC period = RI_Util::get_Born_vonKarmen_period(kv); - const std::map nspin_k = {{1,1}, {2,2}, {4,1}}; + const TC period = RI_Util::get_Born_vonKarmen_period(kv); + const std::map nspin_k = {{1,1}, {2,2}, {4,1}}; const double SPIN_multiple = std::map{ {1,0.5}, {2,1}, {4,1} }.at(nspin); // why? std::vector>>> mRs_a2D(nspin); @@ -258,125 +258,61 @@ auto RI_2D_Comm::split_m2D_ktoR_k(const UnitCell& ucell, } } } // end #pragma omp parallel - ModuleBase::timer::end("RI_2D_Comm", "split_m2D_ktoR_k"); - return mRs_a2D; + ModuleBase::timer::end("RI_2D_Comm", "split_m2D_ktoR_k"); + return mRs_a2D; } template void RI_2D_Comm::add_Hexx( const UnitCell &ucell, - const K_Vectors &kv, - const int ik, + const K_Vectors &kv, + const int ik, const double alpha, - const std::vector>>> &Hs, + const std::vector>>> &Hs, const Parallel_Orbitals& pv, TK* hk) { - ModuleBase::TITLE("RI_2D_Comm","add_Hexx"); - ModuleBase::timer::start("RI_2D_Comm", "add_Hexx"); - - const std::map> is_list = {{1,{0}}, {2,{kv.isk[ik]}}, {4,{0,1,2,3}}}; - for(const int is_b : is_list.at(PARAM.inp.nspin)) - { - int is0_b, is1_b; - std::tie(is0_b,is1_b) = RI_2D_Comm::split_is_block(is_b); - for(const auto &Hs_tmpA : Hs[is_b]) - { - const TA &iat0 = Hs_tmpA.first; - for(const auto &Hs_tmpB : Hs_tmpA.second) - { - const TA &iat1 = Hs_tmpB.first.first; - const TC &cell1 = Hs_tmpB.first.second; + ModuleBase::TITLE("RI_2D_Comm","add_Hexx"); + ModuleBase::timer::start("RI_2D_Comm", "add_Hexx"); + + const std::map> is_list = {{1,{0}}, {2,{kv.isk[ik]}}, {4,{0,1,2,3}}}; + for(const int is_b : is_list.at(PARAM.inp.nspin)) + { + int is0_b, is1_b; + std::tie(is0_b,is1_b) = RI_2D_Comm::split_is_block(is_b); + for(const auto &Hs_tmpA : Hs[is_b]) + { + const TA &iat0 = Hs_tmpA.first; + for(const auto &Hs_tmpB : Hs_tmpA.second) + { + const TA &iat1 = Hs_tmpB.first.first; + const TC &cell1 = Hs_tmpB.first.second; const std::complex frac = alpha - * std::exp( ModuleBase::TWO_PI*ModuleBase::IMAG_UNIT * (kv.kvec_c[ik] * (RI_Util::array3_to_Vector3(cell1)*ucell.latvec)) ); - const RI::Tensor &H = Hs_tmpB.second; - for(size_t iw0_b=0; iw0_b &H = Hs_tmpB.second; + for(size_t iw0_b=0; iw0_b(H(iw0_b, iw1_b)) * RI::Global_Func::convert(frac), pv, hk); - } - } - } - } - } - ModuleBase::timer::end("RI_2D_Comm", "add_Hexx"); -} - - -std::tuple -RI_2D_Comm::get_iat_iw_is_block(const UnitCell& ucell,const int& iwt) -{ - const int iat = ucell.iwt2iat[iwt]; - const int iw = ucell.iwt2iw[iwt]; - switch(PARAM.inp.nspin) - { - case 1: case 2: - return std::make_tuple(iat, iw, 0); - case 4: - return std::make_tuple(iat, iw/2, iw%2); - default: - throw std::invalid_argument(std::string(__FILE__)+" line "+std::to_string(__LINE__)); - } -} - -int RI_2D_Comm::get_is_block(const int is_k, const int is_row_b, const int is_col_b) -{ - switch(PARAM.inp.nspin) - { - case 1: return 0; - case 2: return is_k; - case 4: return is_row_b*2+is_col_b; - default: throw std::invalid_argument(std::string(__FILE__)+" line "+std::to_string(__LINE__)); - } -} - -std::tuple -RI_2D_Comm::split_is_block(const int is_b) -{ - switch(PARAM.inp.nspin) - { - case 1: case 2: - return std::make_tuple(0, 0); - case 4: - return std::make_tuple(is_b/2, is_b%2); - default: - throw std::invalid_argument(std::string(__FILE__)+" line "+std::to_string(__LINE__)); - } + } + } + } + } + } + ModuleBase::timer::end("RI_2D_Comm", "add_Hexx"); } - -int RI_2D_Comm::get_iwt(const UnitCell& ucell, - const int iat, - const int iw_b, - const int is_b) -{ - const int it = ucell.iat2it[iat]; - const int ia = ucell.iat2ia[iat]; - int iw=-1; - switch(PARAM.inp.nspin) - { - case 1: case 2: - iw = iw_b; break; - case 4: - iw = iw_b*2+is_b; break; - default: - throw std::invalid_argument(std::string(__FILE__)+" line "+std::to_string(__LINE__)); - } - const int iwt = ucell.itiaiw2iwt(it,ia,iw); - return iwt; -} - template void RI_2D_Comm::add_HexxR( const int current_spin, @@ -431,7 +367,7 @@ void RI_2D_Comm::add_HexxR( // This is the real-space (DM(R)) counterpart of split_m2D_ktoR (which converts DM(k)), // and the inverse of add_HexxR (which converts Ds -> HContainer); it therefore reuses the // exact same atom/orbital/spin-block index mapping as add_HexxR. -// dm_container is the DM(R) vector returned by DensityMatrix::get_DMR_vector(): +// dm_container is the DM(R) vector returned by DensityMatrix::get_dmr_vec(): // nspin==1 : size 1 (container 0 -> spin-block 0) // nspin==2 : size 2 (container is -> spin-block is) // nspin==4 : size 1 (container 0 holds the 2x2 npol blocks -> spin-blocks 0,1,2,3) diff --git a/source/source_lcao/module_ri/ri_util.cpp b/source/source_lcao/module_ri/ri_util.cpp new file mode 100644 index 00000000000..c122bf529c0 --- /dev/null +++ b/source/source_lcao/module_ri/ri_util.cpp @@ -0,0 +1,118 @@ +//======================= +// AUTHOR : Peize Lin +// DATE : 2022-08-17 +//======================= + +#include "ri_util.h" + +#include "source_base/constants.h" +#include "source_base/global_function.h" +#include "source_cell/unitcell.h" + +#include +#include +#include +#include + +namespace RI_Util +{ + std::map>> + update_coulomb_param( + const std::map>> &coulomb_param, + const UnitCell &ucell, + const K_Vectors *p_kv) + { + std::map>> coulomb_param_updated = coulomb_param; + for(auto ¶m_list : coulomb_param_updated) + { + for(auto ¶m : param_list.second) + { + if(param.at("singularity_correction") == "spencer") + { + // 4/3 * pi * Rcut^3 = V_{supercell} = V_{unitcell} * Nk + const double Rcut = std::pow(0.75 * p_kv->get_nkstot_nospin() * ucell.omega / (ModuleBase::PI), 1.0/3.0); + param["Rcut"] = ModuleBase::GlobalFunc::TO_STRING(Rcut); + } + else if(param.at("singularity_correction") == "revised_spencer") + { + const double bvk_a1 = ucell.a1.norm() * p_kv->nmp[0]; + const double bvk_a2 = ucell.a2.norm() * p_kv->nmp[1]; + const double bvk_a3 = ucell.a3.norm() * p_kv->nmp[2]; + const double Rcut = 0.5 * std::min({bvk_a1, bvk_a2, bvk_a3}); + param["Rcut"] = ModuleBase::GlobalFunc::TO_STRING(Rcut); + } + } + } + return coulomb_param_updated; + } + + std::map>>>> + update_coulomb_settings( + const std::map>> &coulomb_param, + const UnitCell &ucell, + const K_Vectors *p_kv) + { + const std::map>> + coulomb_param_updated = update_coulomb_param(coulomb_param, ucell, p_kv); + + // Separate the parameters into Center2 and Ewald methods + std::map>> coulomb_param_center2; + std::map>> coulomb_param_ewald; + for(auto ¶m_list : coulomb_param_updated) + { + switch(param_list.first) + { + case Conv_Coulomb_Pot_K::Coulomb_Type::Fock: + { + for(auto ¶m : param_list.second) + { + if(param.at("singularity_correction") == "spencer" || param.at("singularity_correction") == "limits" + || param.at("singularity_correction") == "revised_spencer") + { + coulomb_param_center2[param_list.first].push_back(param); + } + else if (param.at("singularity_correction") == "massidda" || param.at("singularity_correction") == "carrier" ) + { + coulomb_param_ewald[param_list.first].push_back(param); + } + } + break; + } + case Conv_Coulomb_Pot_K::Coulomb_Type::Erfc: + { + coulomb_param_center2[param_list.first] = param_list.second; // Erfc is always calculated with Center2 method. + break; + } + default: + { + throw std::invalid_argument( std::string(__FILE__) + " line " + std::to_string(__LINE__) ); + } + } + } + + std::map>>>> coulomb_settings; + + const bool cal_center = !coulomb_param_center2.empty(); + const bool cal_ewald = !coulomb_param_ewald.empty(); + if(cal_center) + { + coulomb_settings[Conv_Coulomb_Pot_K::Coulomb_Method::Center2] = std::make_pair(cal_center, coulomb_param_center2); + } + if(cal_ewald) + { + coulomb_settings[Conv_Coulomb_Pot_K::Coulomb_Method::Ewald] = std::make_pair(cal_ewald, coulomb_param_ewald); + } + if(cal_center && cal_ewald) + { + coulomb_settings[Conv_Coulomb_Pot_K::Coulomb_Method::Center2].first = false; // If both methods are available, only HF for C is needed. + } + + return coulomb_settings; + } +} diff --git a/source/source_lcao/module_ri/ri_util.h b/source/source_lcao/module_ri/ri_util.h index 67aef442ddc..090e9a7e80a 100644 --- a/source/source_lcao/module_ri/ri_util.h +++ b/source/source_lcao/module_ri/ri_util.h @@ -8,6 +8,7 @@ #include "source_cell/klist.h" #include "source_lcao/module_ri/conv_coulomb_pot_k.h" +#include "source_lcao/module_ri/ri_util_bvk.h" #include #include @@ -20,35 +21,28 @@ namespace RI_Util { - inline extern std::array - get_Born_vonKarmen_period(const K_Vectors &kv); - - template - extern std::vector> - get_Born_von_Karmen_cells( const std::array &Born_von_Karman_period ); - - template - inline std::array - Vector3_to_array3(const ModuleBase::Vector3 &v) - { - return std::array {v.x, v.y, v.z}; - } - template - inline ModuleBase::Vector3 - array3_to_Vector3(const std::array &v) - { - return ModuleBase::Vector3 {v[0], v[1], v[2]}; - } + template + inline std::array + Vector3_to_array3(const ModuleBase::Vector3 &v) + { + return std::array {v.x, v.y, v.z}; + } + template + inline ModuleBase::Vector3 + array3_to_Vector3(const std::array &v) + { + return ModuleBase::Vector3 {v[0], v[1], v[2]}; + } - template - RI::Tensor - Matrix_to_Tensor(const Tmatrix &m_old) - { - RI::Tensor m_new({static_cast(m_old.nr), static_cast(m_old.nc)}); - for(int ir=0; ir(m_old(ir,ic)); - return m_new; + template + RI::Tensor + Matrix_to_Tensor(const Tmatrix &m_old) + { + RI::Tensor m_new({static_cast(m_old.nr), static_cast(m_old.nc)}); + for(int ir=0; ir(m_old(ir,ic)); + return m_new; } template @@ -63,22 +57,20 @@ namespace RI_Util return m_new; } - std::map>> - update_coulomb_param( - const std::map>> &coulomb_param, - const UnitCell &ucell, - const K_Vectors *p_kv); + std::map>> + update_coulomb_param( + const std::map>> &coulomb_param, + const UnitCell &ucell, + const K_Vectors *p_kv); - std::map>>>> - update_coulomb_settings( - const std::map>> &coulomb_param, - const UnitCell &ucell, - const K_Vectors *p_kv); + update_coulomb_settings( + const std::map>> &coulomb_param, + const UnitCell &ucell, + const K_Vectors *p_kv); } -#include "ri_util.hpp" - -#endif \ No newline at end of file +#endif diff --git a/source/source_lcao/module_ri/ri_util.hpp b/source/source_lcao/module_ri/ri_util.hpp deleted file mode 100644 index 886c5f1b377..00000000000 --- a/source/source_lcao/module_ri/ri_util.hpp +++ /dev/null @@ -1,172 +0,0 @@ -//======================= -// AUTHOR : Peize Lin -// DATE : 2022-08-17 -//======================= - -#ifndef RI_UTIL_HPP -#define RI_UTIL_HPP - -#include "ri_util.h" -#include "source_base/global_function.h" -#include "source_io/module_parameter/parameter.h" - -namespace RI_Util -{ - inline std::array - get_Born_vonKarmen_period(const K_Vectors &kv) - { - return std::array{kv.nmp[0], kv.nmp[1], kv.nmp[2]}; - } - - template - std::vector> - get_Born_von_Karmen_cells( const std::array &Born_von_Karman_period ) - { - using namespace RI::Array_Operator; - std::vector> Born_von_Karman_cells; - for( int c=0; c{c} % Born_von_Karman_period ); - return Born_von_Karman_cells; - } - - template - std::vector> - get_Born_von_Karmen_cells( const std::array &Born_von_Karman_period ) - { - using namespace RI::Array_Operator; - - std::array sub_Born_von_Karman_period; - for(int i=0; i> Born_von_Karman_cells; - for( const std::array &sub_cell : get_Born_von_Karmen_cells(sub_Born_von_Karman_period) ) - for( Tcell c=0; c cell; - for(int i=0; i{c} % std::array{Born_von_Karman_period.back()})[0]; - Born_von_Karman_cells.emplace_back(std::move(cell)); - } - return Born_von_Karman_cells; - } - - /* example for Ndim=3: - template - std::vector> - get_Born_von_Karmen_cells( const std::array &Born_von_Karman_period ) - { - using namespace Array_Operator; - std::vector> Born_von_Karman_cells; - for( int ix=0; ix{ix,iy,iz} % Born_von_Karman_period ); - return Born_von_Karman_cells; - } - */ - - inline std::map>> - update_coulomb_param( - const std::map>> &coulomb_param, - const UnitCell &ucell, - const K_Vectors *p_kv) - { - std::map>> coulomb_param_updated = coulomb_param; - for(auto ¶m_list : coulomb_param_updated) - { - for(auto ¶m : param_list.second) - { - if(param.at("singularity_correction") == "spencer") - { - // 4/3 * pi * Rcut^3 = V_{supercell} = V_{unitcell} * Nk - const int nspin0 = (PARAM.inp.nspin==2) ? 2 : 1; - const double Rcut = std::pow(0.75 * p_kv->get_nkstot_nospin() * ucell.omega / (ModuleBase::PI), 1.0/3.0); - param["Rcut"] = ModuleBase::GlobalFunc::TO_STRING(Rcut); - } - else if(param.at("singularity_correction") == "revised_spencer") - { - const double bvk_a1 = ucell.a1.norm() * p_kv->nmp[0]; - const double bvk_a2 = ucell.a2.norm() * p_kv->nmp[1]; - const double bvk_a3 = ucell.a3.norm() * p_kv->nmp[2]; - const double Rcut = 0.5 * std::min({bvk_a1, bvk_a2, bvk_a3}); - param["Rcut"] = ModuleBase::GlobalFunc::TO_STRING(Rcut); - } - } - } - return coulomb_param_updated; - } - - inline std::map>>>> - update_coulomb_settings( - const std::map>> &coulomb_param, - const UnitCell &ucell, - const K_Vectors *p_kv) - { - const std::map>> - coulomb_param_updated = update_coulomb_param(coulomb_param, ucell, p_kv); - - // Separate the parameters into Center2 and Ewald methods - std::map>> coulomb_param_center2; - std::map>> coulomb_param_ewald; - for(auto ¶m_list : coulomb_param_updated) - { - switch(param_list.first) - { - case Conv_Coulomb_Pot_K::Coulomb_Type::Fock: - { - for(auto ¶m : param_list.second) - { - if(param.at("singularity_correction") == "spencer" || param.at("singularity_correction") == "limits" - || param.at("singularity_correction") == "revised_spencer") - { - coulomb_param_center2[param_list.first].push_back(param); - } - else if (param.at("singularity_correction") == "massidda" || param.at("singularity_correction") == "carrier" ) - { - coulomb_param_ewald[param_list.first].push_back(param); - } - } - break; - } - case Conv_Coulomb_Pot_K::Coulomb_Type::Erfc: - { - coulomb_param_center2[param_list.first] = param_list.second; // Erfc is always calculated with Center2 method. - break; - } - default: - { - throw std::invalid_argument( std::string(__FILE__) + " line " + std::to_string(__LINE__) ); - } - } - } - - std::map>>>> coulomb_settings; - - const bool cal_center = !coulomb_param_center2.empty(); - const bool cal_ewald = !coulomb_param_ewald.empty(); - if(cal_center) - { - coulomb_settings[Conv_Coulomb_Pot_K::Coulomb_Method::Center2] = std::make_pair(cal_center, coulomb_param_center2); - } - if(cal_ewald) - { - coulomb_settings[Conv_Coulomb_Pot_K::Coulomb_Method::Ewald] = std::make_pair(cal_ewald, coulomb_param_ewald); - } - if(cal_center && cal_ewald) - { - coulomb_settings[Conv_Coulomb_Pot_K::Coulomb_Method::Center2].first = false; // If both methods are available, only HF for C is needed. - } - - return coulomb_settings; - } -} - -#endif diff --git a/source/source_lcao/module_ri/ri_util_bvk.h b/source/source_lcao/module_ri/ri_util_bvk.h new file mode 100644 index 00000000000..0079e0bab2d --- /dev/null +++ b/source/source_lcao/module_ri/ri_util_bvk.h @@ -0,0 +1,81 @@ +//======================= +// AUTHOR : Peize Lin +// DATE : 2022-08-17 +//======================= + +// Born-von Karmen supercell utilities. +// This header is free of LibRI dependencies so that modules built without +// LibRI (e.g. module_lr) can use it. + +#ifndef RI_UTIL_BVK_H +#define RI_UTIL_BVK_H + +#include "source_cell/klist.h" + +#include +#include + +namespace RI_Util +{ + inline std::array + get_Born_vonKarmen_period(const K_Vectors &kv) + { + return std::array{kv.nmp[0], kv.nmp[1], kv.nmp[2]}; + } + + // Fold a cell index into [-period/2, period/2) to match LibRI's + // Array_Operator::operator%: (c % period + 3 * period / 2) % period - period / 2 + template + inline Tcell fold_cell_centered(const Tcell c, const Tcell period) + { + return (c % period + 3 * period / 2) % period - period / 2; + } + + template + std::vector> + get_Born_von_Karmen_cells( const std::array &Born_von_Karman_period ) + { + std::vector> Born_von_Karman_cells; + for( Tcell c=0; c{ fold_cell_centered(c, Born_von_Karman_period[0]) }); + return Born_von_Karman_cells; + } + + template + std::vector> + get_Born_von_Karmen_cells( const std::array &Born_von_Karman_period ) + { + std::array sub_Born_von_Karman_period; + for(size_t i=0; i> Born_von_Karman_cells; + for( const std::array &sub_cell : get_Born_von_Karmen_cells(sub_Born_von_Karman_period) ) + for( Tcell c=0; c cell; + for(size_t i=0; i + std::vector> + get_Born_von_Karmen_cells( const std::array &Born_von_Karman_period ) + { + std::vector> Born_von_Karman_cells; + for( int ix=0; ix{ix,iy,iz} ); + return Born_von_Karman_cells; + } + */ +} + +#endif diff --git a/source/source_lcao/module_ri/rpa_lri.h b/source/source_lcao/module_ri/rpa_lri.h index 31e6ccedd25..675ed5ac251 100644 --- a/source/source_lcao/module_ri/rpa_lri.h +++ b/source/source_lcao/module_ri/rpa_lri.h @@ -16,11 +16,13 @@ #include #include #include +#include #include #include class Parallel_Orbitals; class K_Vectors; +template class Exx_LRI; template class RPA_LRI { @@ -38,21 +40,22 @@ template class RPA_LRI RPA_LRI(const Exx_Info_RI &info_in) : info(info_in) { } - ~RPA_LRI(){}; + ~RPA_LRI(); void postSCF(const UnitCell& ucell, const MPI_Comm& mpi_comm_in, - const elecstate::DensityMatrix& dm, + const module_dm::DensityMatrix& dm, const elecstate::ElecState* pelec, const K_Vectors& kv, const LCAO_Orbitals& orb, const Parallel_Orbitals& parav, const psi::Psi& psi); void init(const MPI_Comm &mpi_comm_in, const K_Vectors &kv_in, const std::vector& orb_cutoff); - void cal_postSCF_exx(const elecstate::DensityMatrix& dm, + void cal_postSCF_exx(const module_dm::DensityMatrix& dm, const MPI_Comm& mpi_comm_in, const UnitCell& ucell, const K_Vectors& kv, - const LCAO_Orbitals& orb); + const LCAO_Orbitals& orb, + const Parallel_Orbitals& parav); void output_ewald_coulomb(const UnitCell& ucell, const K_Vectors& kv, const LCAO_Orbitals& orb); void cal_large_Cs(const UnitCell& ucell, const LCAO_Orbitals& orb, const K_Vectors& kv); void cal_abfs_overlap(const UnitCell& ucell, const LCAO_Orbitals& orb, const K_Vectors& kv); @@ -94,6 +97,7 @@ template class RPA_LRI MPI_Comm mpi_comm; std::vector orb_cutoff_; double ccp_rmesh_times_ewald = 0.0; + double ccp_rmesh_times_cut = 0.0; std::vector>> lcaos; std::vector>> abfs; @@ -110,8 +114,8 @@ template class RPA_LRI // Tdata post_process_Erpa( const Tdata &Erpa_in ) const; - Exx_LRI* exx_cut_coulomb = nullptr; - Exx_LRI* exx_full_coulomb = nullptr; + std::unique_ptr> exx_cut_coulomb; + std::unique_ptr> exx_full_coulomb; }; #include "rpa_lri.hpp" diff --git a/source/source_lcao/module_ri/rpa_lri.hpp b/source/source_lcao/module_ri/rpa_lri.hpp index b67d8163061..5f19a66120b 100644 --- a/source/source_lcao/module_ri/rpa_lri.hpp +++ b/source/source_lcao/module_ri/rpa_lri.hpp @@ -18,6 +18,7 @@ #include "source_lcao/module_ri/module_exx_symmetry/symm_rotation.h" #include "rpa_lri.h" +#include "exx_lri.h" #include "source_basis/module_ao/elem_basis_idx_orb.h" #include "source_base/global_function.h" #include "source_estate/elecstate_lcao.h" @@ -36,12 +37,15 @@ inline void trim_malloc_cache() malloc_trim(0); #endif } -} +} // namespace RpaLriDetail + +template +RPA_LRI::~RPA_LRI() = default; template void RPA_LRI::postSCF(const UnitCell& ucell, const MPI_Comm& mpi_comm_in, - const elecstate::DensityMatrix& dm, + const module_dm::DensityMatrix& dm, const elecstate::ElecState* pelec, const K_Vectors& kv, const LCAO_Orbitals& orb, @@ -51,23 +55,25 @@ void RPA_LRI::postSCF(const UnitCell& ucell, ModuleBase::TITLE("RPA_LRI", "postSCF"); ModuleBase::timer::start("RPA_LRI", "postSCF"); ModuleBase::GlobalFunc::MAKE_DIR(outdir); + this->ccp_rmesh_times_cut = PARAM.inp.rpa_ccp_rmesh_times; + this->ccp_rmesh_times_ewald = this->info.ccp_rmesh_times; // should be `exx_ccp_rmesh_times` - this->cal_postSCF_exx(dm, mpi_comm_in, ucell, kv, orb); + this->cal_postSCF_exx(dm, mpi_comm_in, ucell, kv, orb, parav); this->init(mpi_comm_in, kv, orb.cutoffs()); this->out_bands(pelec); this->out_eigen_vector(parav, psi); this->out_struc(ucell); std::cout << "rpa_pca_threshold: " << this->info.pca_threshold << std::endl; - std::cout << "rpa_ccp_rmesh_times: " << this->info.ccp_rmesh_times << std::endl; + std::cout << "rpa_ccp_rmesh_times_cut: " << this->ccp_rmesh_times_cut << std::endl; + std::cout << "rpa_ccp_rmesh_times_ewald: " << this->ccp_rmesh_times_ewald << std::endl; std::cout << "rpa_lcao_exx(Ha): " << std::fixed << std::setprecision(15) << exx_cut_coulomb->Eexx / 2.0 << std::endl; std::cout << "etxc(Ha): " << std::fixed << std::setprecision(15) << pelec->f_en.etxc / 2.0 << std::endl; std::cout << "etot(Ha): " << std::fixed << std::setprecision(15) << pelec->f_en.etot / 2.0 << std::endl; std::cout << "Etot_without_rpa(Ha): " << std::fixed << std::setprecision(15) << (pelec->f_en.etot - pelec->f_en.etxc + exx_cut_coulomb->Eexx) / 2.0 << std::endl; - delete exx_cut_coulomb; - exx_cut_coulomb = nullptr; + exx_cut_coulomb.reset(); RpaLriDetail::trim_malloc_cache(); if (this->info.shrink_abfs_pca_thr >= 0.0) @@ -106,11 +112,12 @@ void RPA_LRI::init(const MPI_Comm& mpi_comm_in, const K_Vectors& kv_in } template -void RPA_LRI::cal_postSCF_exx(const elecstate::DensityMatrix& dm, +void RPA_LRI::cal_postSCF_exx(const module_dm::DensityMatrix& dm, const MPI_Comm& mpi_comm_in, const UnitCell& ucell, const K_Vectors& kv, - const LCAO_Orbitals& orb) + const LCAO_Orbitals& orb, + const Parallel_Orbitals& parav) { ModuleBase::TITLE("RPA_LRI", "cal_postSCF_exx"); ModuleBase::timer::start("RPA_LRI", "cal_postSCF_exx"); @@ -160,39 +167,36 @@ void RPA_LRI::cal_postSCF_exx(const elecstate::DensityMatrix { const std::array period = RI_Util::get_Born_vonKarmen_period(kv); const auto& Rs = RI_Util::get_Born_von_Karmen_cells(period); - symrot.find_irreducible_sector(ucell.symm, ucell.atoms, ucell.st, Rs, period, ucell.lat); + symrot.find_irreducible_sector(ucell.symm, ucell.atoms, ucell.st, Rs, period, ucell.lat, PARAM.globalv.global_out_dir); // set Lmax of the rotation matrices to max(l_ao, l_abf), to support rotation under ABF // NOTE: Using Exx_Abfs::Construct_Orbs::get_Lmax() to compute Lmax from the actual ABFs // instead of relying on exx_cut_coulomb->abfs_Lmax() (not yet initialized) or // this->info.abfs_Lmax (defaults to 0). This ensures correct Lmax for symmetry rotation. symrot.set_abfs_Lmax(Exx_Abfs::Construct_Orbs::get_Lmax(abfs_for_lmax)); - symrot.cal_Ms(kv, ucell, *dm.get_paraV_pointer()); + symrot.cal_Ms(kv, ucell, parav, PARAM.inp.nspin); // output Ts (symrot_R.txt) and Ms (symrot_k.txt) ModuleSymmetry::print_symrot_info_R(symrot, ucell.symm, ucell.lmax, Rs); ModuleSymmetry::print_symrot_info_k(symrot, kv, ucell); - mix_DMk_2D.mix(symrot.restore_dm(kv, dm.get_DMK_vector(), *dm.get_paraV_pointer()), true); + mix_DMk_2D.mix(symrot.restore_dm(kv, dm.get_dmk_vec(), parav), true); } - else { mix_DMk_2D.mix(dm.get_DMK_vector(), true); } + else { mix_DMk_2D.mix(dm.get_dmk_vec(), true); } const std::vector>>> Ds = RI_2D_Comm::split_m2D_ktoR( ucell, kv, mix_DMk_2D.get_DMk_out(), - *dm.get_paraV_pointer(), + parav, PARAM.inp.nspin, exx_spacegroup_symmetry); - - // reserve exx_ccp_rmesh_times to calculate full Coulomb - // Note: ccp_type=Hf and hybrid_alpha=1 were previously set on the global Exx_Info - // and sync_from_global() was called, but this->info (value copy) already has the correct - // coulomb_param from construction time, so those writes are redundant and removed. - this->ccp_rmesh_times_ewald = this->info.ccp_rmesh_times; - // Using rpa_ccp_rmesh_times to calculate cut Coulomb this->Vs_period - Exx_Info_RI local_info = this->info; - local_info.ccp_rmesh_times = PARAM.inp.rpa_ccp_rmesh_times; + if (!exx_cut_coulomb) - exx_cut_coulomb = new Exx_LRI(local_info); + { + Exx_Info_RI local_info = this->info; + local_info.ccp_rmesh_times = this->ccp_rmesh_times_cut; + Exx_LRI* new_exx = new Exx_LRI(local_info); + exx_cut_coulomb.reset(new_exx); + } if (this->info.shrink_abfs_pca_thr >= 0.0) { @@ -207,15 +211,16 @@ void RPA_LRI::cal_postSCF_exx(const elecstate::DensityMatrix // NOTE: Reuse abfs_for_lmax constructed earlier to avoid redundant ABFs construction. exx_cut_coulomb->init_spencer(mpi_comm_in, ucell, kv, orb, abfs_for_lmax); // cal C and V for exx - this->output_cut_coulomb_cs(ucell, exx_cut_coulomb); + Exx_LRI* cut_coulomb = exx_cut_coulomb.get(); + this->output_cut_coulomb_cs(ucell, cut_coulomb); // cal CVCD if (exx_spacegroup_symmetry && PARAM.inp.exx_symmetry_realspace) { - exx_cut_coulomb->cal_exx_elec(Ds, ucell, *dm.get_paraV_pointer(), &symrot); + exx_cut_coulomb->cal_exx_elec(Ds, ucell, parav, &symrot); } else { - exx_cut_coulomb->cal_exx_elec(Ds, ucell, *dm.get_paraV_pointer()); + exx_cut_coulomb->cal_exx_elec(Ds, ucell, parav); } // cout<<"postSCF_Eexx: "<::output_cut_coulomb_cs(const UnitCell& ucell, Exx_LRI>> Vs_cut_IJR; std::map>> Cs; std::map>> tmp; - std::cout << "Use rpa_ccp_rmesh_times=" << this->info.ccp_rmesh_times << " to calculate cut Coulomb" << std::endl; + std::cout << "Use rpa_ccp_rmesh_times=" << this->ccp_rmesh_times_cut << " to calculate cut Coulomb" << std::endl; // Shrink_ABFS_ORBITAL cannot exceed this angular momentum of MGT exx_lri_rpa->cal_cut_coulomb_cs(Vs_cut_IJR, Cs, ucell, PARAM.inp.out_ri_cv); // MPI: {ia0, {ia1, R}} to {ia0, ia1} @@ -240,7 +245,7 @@ void RPA_LRI::output_cut_coulomb_cs(const UnitCell& ucell, Exx_LRI period_Vs - = LRI_CV_Tools::cal_latvec_range(1 + this->info.ccp_rmesh_times, ucell, this->orb_cutoff_); + = LRI_CV_Tools::cal_latvec_range(1 + this->ccp_rmesh_times_cut, ucell, this->orb_cutoff_); const std::pair, std::vector>>> list_As_Vs_atoms = RI::Distribute_Equally::distribute_atoms(this->mpi_comm, atoms, period_Vs, 2, false); const auto list_A0_pair_R = list_As_Vs_atoms.first; @@ -286,7 +291,10 @@ void RPA_LRI::output_ewald_coulomb(const UnitCell& ucell, const K_Vect Exx_Info_RI local_info = this->info; local_info.ccp_rmesh_times = this->ccp_rmesh_times_ewald; if (!exx_full_coulomb) - exx_full_coulomb = new Exx_LRI(local_info); + { + Exx_LRI* new_exx = new Exx_LRI(local_info); + exx_full_coulomb.reset(new_exx); + } if (this->info.shrink_abfs_pca_thr >= 0.0) exx_full_coulomb->init(mpi_comm, ucell, kv, orb, this->abfs_shrink); @@ -322,14 +330,14 @@ void RPA_LRI::output_ewald_coulomb(const UnitCell& ucell, const K_Vect const std::array period = {p_kv->nmp[0], p_kv->nmp[1], p_kv->nmp[2]}; this->Vs_period = RI::RI_Tools::cal_period(Vs_full_IJ, period); - this->out_coulomb_k(ucell, this->Vs_period, "coulomb_mat_", exx_full_coulomb); + Exx_LRI* full_coulomb = exx_full_coulomb.get(); + this->out_coulomb_k(ucell, this->Vs_period, "coulomb_mat_", full_coulomb); Vs_period.clear(); Vs_period.swap(tmp); Cs.clear(); Cs.swap(tmp); - delete exx_full_coulomb; - exx_full_coulomb = nullptr; + exx_full_coulomb.reset(); RpaLriDetail::trim_malloc_cache(); ModuleBase::timer::end("RPA_LRI", "output_ewald_coulomb"); @@ -341,7 +349,12 @@ void RPA_LRI::cal_large_Cs(const UnitCell& ucell, const LCAO_Orbitals& ModuleBase::TITLE("RPA_LRI", "cal_large_Cs"); ModuleBase::timer::start("RPA_LRI", "cal_large_Cs"); if (!exx_cut_coulomb) - exx_cut_coulomb = new Exx_LRI(this->info); + { + Exx_Info_RI local_info = this->info; + local_info.ccp_rmesh_times = this->ccp_rmesh_times_cut; + Exx_LRI* new_exx = new Exx_LRI(local_info); + exx_cut_coulomb.reset(new_exx); + } exx_cut_coulomb->init_spencer(this->mpi_comm, ucell, kv, orb); ModuleBase::GlobalFunc::DONE(GlobalV::ofs_running, "exx_cut_coulomb->init"); this->abfs = exx_cut_coulomb->abfs; @@ -374,7 +387,7 @@ void RPA_LRI::cal_large_Cs(const UnitCell& ucell, const LCAO_Orbitals& true); const std::array period_Vs - = LRI_CV_Tools::cal_latvec_range(1 + this->info.ccp_rmesh_times, ucell, orb_cutoff_); + = LRI_CV_Tools::cal_latvec_range(1 + this->ccp_rmesh_times_cut, ucell, orb_cutoff_); std::pair, std::vector>>>> list_As_Vs = RI::Distribute_Equally::distribute_atoms_periods(this->mpi_comm, atoms, period_Vs, 2, false); ModuleBase::GlobalFunc::DONE(GlobalV::ofs_running, "cal_large_Vs start"); @@ -422,7 +435,8 @@ void RPA_LRI::cal_large_Cs(const UnitCell& ucell, const LCAO_Orbitals& this->Vs_period = RI_2D_Comm::comm_map2_first(this->mpi_comm, this->Vs_period, atoms00, atoms01); ModuleBase::GlobalFunc::DONE(GlobalV::ofs_running, "Vs_period_comm"); - this->out_coulomb_k(ucell, this->Vs_period, "coulomb_unshrinked_cut_", exx_cut_coulomb); + Exx_LRI* cut_coulomb = exx_cut_coulomb.get(); + this->out_coulomb_k(ucell, this->Vs_period, "coulomb_unshrinked_cut_", cut_coulomb); ModuleBase::GlobalFunc::DONE(GlobalV::ofs_running, "out_large_Vs"); this->Vs_period.clear(); this->Vs_period.swap(tmp); @@ -435,8 +449,7 @@ void RPA_LRI::cal_large_Cs(const UnitCell& ucell, const LCAO_Orbitals& ModuleBase::GlobalFunc::DONE(GlobalV::ofs_running, "out_large_Cs"); this->Cs_period.clear(); this->Cs_period.swap(tmp); - delete exx_cut_coulomb; - exx_cut_coulomb = nullptr; + exx_cut_coulomb.reset(); RpaLriDetail::trim_malloc_cache(); ModuleBase::timer::end("RPA_LRI", "cal_large_Cs"); diff --git a/source/source_lcao/module_ri/serialization_boost.h b/source/source_lcao/module_ri/serialization_boost.h index 8cc87ee5793..c53d8d9b482 100644 --- a/source/source_lcao/module_ri/serialization_boost.h +++ b/source/source_lcao/module_ri/serialization_boost.h @@ -18,74 +18,74 @@ namespace boost { - namespace serialization - { - // ModuleBase::Vector3 Peize Lin add 2018-07-01 - template - inline void serialize(Archive & ar, ModuleBase::Vector3 & v, const unsigned int version) - { - ar & v.x; - ar & v.y; - ar & v.z; - } + namespace serialization + { + // ModuleBase::Vector3 Peize Lin add 2018-07-01 + template + inline void serialize(Archive & ar, ModuleBase::Vector3 & v, const unsigned int version) + { + ar & v.x; + ar & v.y; + ar & v.z; + } - // Abfs::Vector3_Order Peize Lin add 2018-07-01 - template - inline void serialize(Archive & ar, Abfs::Vector3_Order & v, const unsigned int version) - { - ar & v.x; - ar & v.y; - ar & v.z; - } + // Abfs::Vector3_Order Peize Lin add 2018-07-01 + template + inline void serialize(Archive & ar, Abfs::Vector3_Order & v, const unsigned int version) + { + ar & v.x; + ar & v.y; + ar & v.z; + } - // matrix Peize Lin add 2018-07-01 - template - inline void save( Archive & ar, const ModuleBase::matrix & m, const unsigned int /*file_version*/ ) - { - const collection_size_type nr(m.nr), nc(m.nc); - ar << BOOST_SERIALIZATION_NVP(nr) << BOOST_SERIALIZATION_NVP(nc); - if( nr && nc ) - ar << serialization::make_array( m.c, nr*nc ); - } - template - inline void load( Archive & ar, ModuleBase::matrix &m, const unsigned int /*file_version*/ ) - { - collection_size_type nr, nc; - ar >> BOOST_SERIALIZATION_NVP(nr) >> BOOST_SERIALIZATION_NVP(nc); - m.create(nr,nc,false); - if( nr && nc ) - ar >> serialization::make_array( m.c, nr*nc ); - } - template - inline void serialize( Archive & ar, ModuleBase::matrix & m, const unsigned int file_version ) - { - boost::serialization::split_free(ar, m, file_version); - } + // matrix Peize Lin add 2018-07-01 + template + inline void save( Archive & ar, const ModuleBase::matrix & m, const unsigned int /*file_version*/ ) + { + const collection_size_type nr(m.nr), nc(m.nc); + ar << BOOST_SERIALIZATION_NVP(nr) << BOOST_SERIALIZATION_NVP(nc); + if( nr && nc ) + ar << serialization::make_array( m.c, nr*nc ); + } + template + inline void load( Archive & ar, ModuleBase::matrix &m, const unsigned int /*file_version*/ ) + { + collection_size_type nr, nc; + ar >> BOOST_SERIALIZATION_NVP(nr) >> BOOST_SERIALIZATION_NVP(nc); + m.create(nr,nc,false); + if( nr && nc ) + ar >> serialization::make_array( m.c, nr*nc ); + } + template + inline void serialize( Archive & ar, ModuleBase::matrix & m, const unsigned int file_version ) + { + boost::serialization::split_free(ar, m, file_version); + } - // MatrixWrapper Peize Lin add 2018-07-31 - template - inline void save( Archive & ar, const ModuleBase::Matrix_Wrapper & m, const unsigned int /*file_version*/ ) - { - const collection_size_type nr(m.nr), nc(m.nc); - ar << BOOST_SERIALIZATION_NVP(nr) << BOOST_SERIALIZATION_NVP(nc); - if( nr && nc ) - ar << serialization::make_array( m.c, nr*nc ); - } - template - inline void load( Archive & ar, ModuleBase::Matrix_Wrapper &m, const unsigned int /*file_version*/ ) - { - collection_size_type nr, nc; - ar >> BOOST_SERIALIZATION_NVP(nr) >> BOOST_SERIALIZATION_NVP(nc); - m.create(nr,nc,false); - if( nr && nc ) - ar >> serialization::make_array( m.c, nr*nc ); - } - template - inline void serialize( Archive & ar, ModuleBase::Matrix_Wrapper & m, const unsigned int file_version ) - { - boost::serialization::split_free(ar, m, file_version); - } - } + // MatrixWrapper Peize Lin add 2018-07-31 + template + inline void save( Archive & ar, const ModuleBase::Matrix_Wrapper & m, const unsigned int /*file_version*/ ) + { + const collection_size_type nr(m.nr), nc(m.nc); + ar << BOOST_SERIALIZATION_NVP(nr) << BOOST_SERIALIZATION_NVP(nc); + if( nr && nc ) + ar << serialization::make_array( m.c, nr*nc ); + } + template + inline void load( Archive & ar, ModuleBase::Matrix_Wrapper &m, const unsigned int /*file_version*/ ) + { + collection_size_type nr, nc; + ar >> BOOST_SERIALIZATION_NVP(nr) >> BOOST_SERIALIZATION_NVP(nc); + m.create(nr,nc,false); + if( nr && nc ) + ar >> serialization::make_array( m.c, nr*nc ); + } + template + inline void serialize( Archive & ar, ModuleBase::Matrix_Wrapper & m, const unsigned int file_version ) + { + boost::serialization::split_free(ar, m, file_version); + } + } } diff --git a/source/source_lcao/module_ri/serialization_cereal.h b/source/source_lcao/module_ri/serialization_cereal.h index e533db88986..d0c0fa0af64 100644 --- a/source/source_lcao/module_ri/serialization_cereal.h +++ b/source/source_lcao/module_ri/serialization_cereal.h @@ -2,7 +2,6 @@ #define SERIALIZATION_CEREAL_H #include -#include #include #include #include @@ -19,20 +18,20 @@ template void serialize( Archive & ar, Abfs::Vector3_ namespace ModuleBase { - template void serialize( Archive & ar, Vector3 & v ){ ar(v.x); ar(v.y); ar(v.z); } + template void serialize( Archive & ar, Vector3 & v ){ ar(v.x); ar(v.y); ar(v.z); } - template void save( Archive & ar, const matrix & m ) - { - ar(m.nr); ar(m.nc); - ar(cereal::binary_data(m.c, m.nr*m.nc*sizeof(double))); - } - template void load( Archive & ar, matrix & m ) - { - int nr, nc; - ar(nr); ar(nc); - m.create(nr,nc); - ar(cereal::binary_data(m.c, m.nr*m.nc*sizeof(double))); - } + template void save( Archive & ar, const matrix & m ) + { + ar(m.nr); ar(m.nc); + ar(cereal::binary_data(m.c, m.nr*m.nc*sizeof(double))); + } + template void load( Archive & ar, matrix & m ) + { + int nr, nc; + ar(nr); ar(nc); + m.create(nr,nc); + ar(cereal::binary_data(m.c, m.nr*m.nc*sizeof(double))); + } } #include @@ -40,35 +39,35 @@ namespace ModuleBase namespace ModuleBase { - template - void bcast_data_cereal(T &data, const MPI_Comm &mpi_comm, const int &rank_bcast) - { - int my_rank; MPI_Comm_rank( mpi_comm, &my_rank ); - if(my_rank==rank_bcast) - { - std::stringstream ss; - { - cereal::BinaryOutputArchive ar(ss); - ar(data); - } - const int size = ss.str().size(); - MPI_Bcast( const_cast(&size), 1, MPI_INT, rank_bcast, mpi_comm ); - MPI_Bcast( const_cast(ss.str().c_str()), size, MPI_CHAR, rank_bcast, mpi_comm ); - } - else - { - int size; - MPI_Bcast( &size, 1, MPI_INT, rank_bcast, mpi_comm ); - std::vector c(size); - MPI_Bcast( c.data(), size, MPI_CHAR, rank_bcast, mpi_comm ); - std::stringstream ss; - ss.rdbuf()->pubsetbuf(c.data(),size); - { - cereal::BinaryInputArchive ar(ss); - ar(data); - } - } - } + template + void bcast_data_cereal(T &data, const MPI_Comm &mpi_comm, const int &rank_bcast) + { + int my_rank; MPI_Comm_rank( mpi_comm, &my_rank ); + if(my_rank==rank_bcast) + { + std::stringstream ss; + { + cereal::BinaryOutputArchive ar(ss); + ar(data); + } + const int size = ss.str().size(); + MPI_Bcast( const_cast(&size), 1, MPI_INT, rank_bcast, mpi_comm ); + MPI_Bcast( const_cast(ss.str().c_str()), size, MPI_CHAR, rank_bcast, mpi_comm ); + } + else + { + int size; + MPI_Bcast( &size, 1, MPI_INT, rank_bcast, mpi_comm ); + std::vector c(size); + MPI_Bcast( c.data(), size, MPI_CHAR, rank_bcast, mpi_comm ); + std::stringstream ss; + ss.rdbuf()->pubsetbuf(c.data(),size); + { + cereal::BinaryInputArchive ar(ss); + ar(data); + } + } + } } #endif diff --git a/source/source_lcao/module_ri/test/dm_mixing_test.cpp b/source/source_lcao/module_ri/test/dm_mixing_test.cpp index 8e2afe3f873..462bc013757 100644 --- a/source/source_lcao/module_ri/test/dm_mixing_test.cpp +++ b/source/source_lcao/module_ri/test/dm_mixing_test.cpp @@ -4,7 +4,7 @@ #include "source_lcao/module_ri/mix_dmk_2d.h" /************************************************ - * unit test of charge_mixing.cpp & mix_dmk_2d.cpp + * unit test of chg_mix.cpp & mix_dmk_2d.cpp ***********************************************/ /** diff --git a/source/source_lcao/module_ri/test_code/element_basis_index_test.cpp b/source/source_lcao/module_ri/test_code/element_basis_index_test.cpp index 0777130374c..205077127eb 100644 --- a/source/source_lcao/module_ri/test_code/element_basis_index_test.cpp +++ b/source/source_lcao/module_ri/test_code/element_basis_index_test.cpp @@ -8,83 +8,83 @@ ModuleBase::Element_Basis_Index::Range construct_range() { - ModuleBase::Element_Basis_Index::Range range; - range.resize(2); - - range[0].resize(3); - range[0][0].N = range[0][1].N = 2; - range[0][2].N = 1; - - range[1].resize(2); - range[1][0].N = 3; - range[1][1].N = 2; - - for( size_t T=0; T!=range.size(); ++T ) - for( size_t L=0; L!=range[T].size(); ++L ) - range[T][L].M=2*L+1; - - return range; + ModuleBase::Element_Basis_Index::Range range; + range.resize(2); + + range[0].resize(3); + range[0][0].N = range[0][1].N = 2; + range[0][2].N = 1; + + range[1].resize(2); + range[1][0].N = 3; + range[1][1].N = 2; + + for( size_t T=0; T!=range.size(); ++T ) + for( size_t L=0; L!=range[T].size(); ++L ) + range[T][L].M=2*L+1; + + return range; } int main() { - const ModuleBase::Element_Basis_Index::Range range = construct_range(); - std::cout<<"range:"<>>> & orbs, const std::string & file_name ) { - std::ofstream ofsN(file_name); - for( size_t T=0; T!=orbs.size(); ++T ) - for( size_t L=0; L!=orbs[T].size(); ++L ) - for( size_t N=0; N!=orbs[T][L].size(); ++N ) - { - ofsN<>> & orbs, const std::string & file_name ) { - std::ofstream ofsN(file_name); - for( size_t T=0; T!=orbs.size(); ++T ) - for( size_t L=0; L!=orbs[T].size(); ++L ) - for( size_t N=0; N!=orbs[T][L].size(); ++N ) - { - ofsN< - Tensor init_Tensor(const std::vector &shape) - { - Tensor t(shape); - for(size_t i=0; isize(); ++i) - t.ptr()[i] = i; - return t; - } - - template - Tensor init_Tensor2(const std::vector &shape) - { - Tensor t(shape); - for(size_t i0=0; i0 - void test_input_output() - { - Inverse_Matrix inv; - - const size_t n_all = 5; - const std::vector n0 = {2,3}; - const std::vector n1 = {1,2,2}; - - Tensor m = init_Tensor({n_all,n_all}); - - std::vector>> ms(n0.size(), std::vector>(n1.size())); - for(size_t Im0=0; Im0({n0[Im0], n1[Im1]}); - - inv.input(m); - std::cout< - void test_inverse() - { - Tensor t = init_Tensor2({5,5}); - Inverse_Matrix inv; - inv.input(t); - inv.cal_inverse(Inverse_Matrix::Method::potrf); - //inv.cal_inverse(Inverse_Matrix::Method::syev); - Tensor tI = inv.output(); - - std::cout< + Tensor init_Tensor(const std::vector &shape) + { + Tensor t(shape); + for(size_t i=0; isize(); ++i) + t.ptr()[i] = i; + return t; + } + + template + Tensor init_Tensor2(const std::vector &shape) + { + Tensor t(shape); + for(size_t i0=0; i0 + void test_input_output() + { + Inverse_Matrix inv; + + const size_t n_all = 5; + const std::vector n0 = {2,3}; + const std::vector n1 = {1,2,2}; + + Tensor m = init_Tensor({n_all,n_all}); + + std::vector>> ms(n0.size(), std::vector>(n1.size())); + for(size_t Im0=0; Im0({n0[Im0], n1[Im1]}); + + inv.input(m); + std::cout< + void test_inverse() + { + Tensor t = init_Tensor2({5,5}); + Inverse_Matrix inv; + inv.input(t); + inv.cal_inverse(Inverse_Matrix::Method::potrf); + //inv.cal_inverse(Inverse_Matrix::Method::syev); + Tensor tI = inv.output(); + + std::cout< descv(9); - ScalapackConnector::descinit(ModuleBase::GlobalFunc::VECTOR_TO_PTR(descv), m, n, mb, nb, irsrc, icsrc, ictxt, mlld, info); - return descv; - }; + auto ofs_pro = [&]() + { + std::ofstream ofs("ofs_"+ModuleBase::GlobalFunc::TO_STRING(my_rank)); + ofs< descv(9); + ScalapackConnector::descinit(ModuleBase::GlobalFunc::VECTOR_TO_PTR(descv), m, n, mb, nb, irsrc, icsrc, ictxt, mlld, info); + return descv; + }; // return init_desc; // }; - auto ofs_matrix = [&](const std::vector& v) - { - std::ofstream ofs("matrix_"+ModuleBase::GlobalFunc::TO_STRING(my_rank)); - for(const double i : v ) - ofs< &descv) - { - std::ofstream ofs("desc_"+ModuleBase::GlobalFunc::TO_STRING(my_rank)); - for(int i=0; i<9; ++i) - ofs<& v) + { + std::ofstream ofs("matrix_"+ModuleBase::GlobalFunc::TO_STRING(my_rank)); + for(const double i : v ) + ofs< &descv) + { + std::ofstream ofs("desc_"+ModuleBase::GlobalFunc::TO_STRING(my_rank)); + for(int i=0; i<9; ++i) + ofs< descv_5321_32 = init_ictxt('R',true)(5,3,2,1,0,0); - int *desc_5321_32 = ModuleBase::GlobalFunc::VECTOR_TO_PTR(descv_5321_32); - - std::vector descv_5521_32 = init_ictxt('R',true)(5,5,2,1,0,0); - int *desc_5521_32 = ModuleBase::GlobalFunc::VECTOR_TO_PTR(descv_5521_32); - - std::vector descv_5522_32 = init_ictxt('R',true)(5,5,2,2,0,0); - int *desc_5522_32 = ModuleBase::GlobalFunc::VECTOR_TO_PTR(descv_5522_32); - - std::vector descv_5511_32 = init_ictxt('R',true)(5,5,1,1,0,0); - int *desc_5511_32 = ModuleBase::GlobalFunc::VECTOR_TO_PTR(descv_5511_32); - ofs_desc(descv_5511_32); + std::vector descv_5321_32 = init_ictxt('R',true)(5,3,2,1,0,0); + int *desc_5321_32 = ModuleBase::GlobalFunc::VECTOR_TO_PTR(descv_5321_32); + + std::vector descv_5521_32 = init_ictxt('R',true)(5,5,2,1,0,0); + int *desc_5521_32 = ModuleBase::GlobalFunc::VECTOR_TO_PTR(descv_5521_32); + + std::vector descv_5522_32 = init_ictxt('R',true)(5,5,2,2,0,0); + int *desc_5522_32 = ModuleBase::GlobalFunc::VECTOR_TO_PTR(descv_5522_32); + + std::vector descv_5511_32 = init_ictxt('R',true)(5,5,1,1,0,0); + int *desc_5511_32 = ModuleBase::GlobalFunc::VECTOR_TO_PTR(descv_5511_32); + ofs_desc(descv_5511_32); */ - std::vector descv_5321_32 = init_desc(5,3,2,1,0,0); - int *desc_5321_32 = ModuleBase::GlobalFunc::VECTOR_TO_PTR(descv_5321_32); - - std::vector descv_5521_32 = init_desc(5,5,2,1,0,0); - int *desc_5521_32 = ModuleBase::GlobalFunc::VECTOR_TO_PTR(descv_5521_32); - - std::vector descv_5522_32 = init_desc(5,5,2,2,0,0); - int *desc_5522_32 = ModuleBase::GlobalFunc::VECTOR_TO_PTR(descv_5522_32); - - std::vector descv_5511_32 = init_desc(5,5,1,1,0,0); - int *desc_5511_32 = ModuleBase::GlobalFunc::VECTOR_TO_PTR(descv_5511_32); - - /* - auto init_matrix_A5624_32 = [&]() - { - switch(my_rank) - { - case 0: return std::vector{0,1,2,3,6,7,8,9}; - case 1: return std::vector{4,5,10,11}; - case 2: return std::vector{12,13,14,15,18,19,20,21}; - case 3: return std::vector{16,17,22,23}; - case 4: return std::vector{24,25,26,27}; - case 5: return std::vector{28,29}; - } - }; - auto init_matrix_I21 = [&]() - { - switch(my_rank) - { - case 0: return std::vector{1,0,0,0,0,0}; - case 1: return std::vector{0,0,0,1,0,0}; - case 2: return std::vector{0,1,0,0,0,0}; - case 3: return std::vector{0,0,0,0,1,0}; - case 4: return std::vector{0,0,1,0,0,0}; - case 5: return std::vector{0,0,0,0,0,1}; - } - }; - auto init_matrix_I11 = [&]() - { - switch(my_rank) - { - case 0: return std::vector{1,0,0,0,0,0}; - case 1: return std::vector{0,0,0,0,1,0}; - case 2: return std::vector{0,0,0,0,0,1}; - case 3: return std::vector{1,0,0,0,0,0}; - case 4: return std::vector{0,1,0,0,0,0}; - case 5: return std::vector{0,0,0,0,0,1}; - } - }; - auto init_matrix_I4411 = [&]() - { - switch(my_rank) - { - case 0: return std::vector{1,0,0,1}; - case 1: return std::vector{0,0,0,0}; - case 2: return std::vector{0,0,0,0}; - case 3: return std::vector{1,0,0,1}; - } - }; - auto init_matrix_I4422 = [&]() - { - switch(my_rank) - { - case 0: return std::vector{1,0,0,1}; - case 1: return std::vector{0,0,0,0}; - case 2: return std::vector{0,0,0,0}; - case 3: return std::vector{1,0,0,1}; - } - }; - auto init_matrix_I5522 = [&]() - { - switch(my_rank) - { - case 0: return std::vector{1,0,0,0,1,0,0,0,1}; - case 1: return std::vector{0,0,0,0,0,0}; - case 2: return std::vector{0,0,0,0,0,0}; - case 3: return std::vector{1,0,0,1}; - } - }; - auto init_matrix_I5511_32 = [&]() - { - switch(my_rank) - { - case 0: return std::vector{1,0,0,0,0}; - case 1: return std::vector{0,0,0,1}; - case 2: return std::vector{0,0,0,0,0,1}; - case 3: return std::vector{1,0,0,0}; - case 4: return std::vector{0,1,0}; - case 5: return std::vector{0,0}; - } - }; - auto init_matrix_I5521_32 = [&]() - { - switch(my_rank) - { - case 0: return std::vector{1,0,0,0,0,0}; - case 1: return std::vector{0,0,1,0}; - case 2: return std::vector{0,1,0,0,0,0}; - case 3: return std::vector{0,0,0,1}; - case 4: return std::vector{0,0,1}; - case 5: return std::vector{0,0}; - } - }; - auto init_matrix_I5522_32 = [&]() - { - switch(my_rank) - { - case 0: return std::vector{1,0,0,0,1,0}; - case 1: return std::vector{0,0,0,0}; - case 2: return std::vector{0,0,0,0,0,0}; - case 3: return std::vector{1,0,0,1}; - case 4: return std::vector{0,0,1}; - case 5: return std::vector{0,0}; - } - }; - auto init_matrix_I3321_11 = [&]() - { - switch(my_rank) - { - case 0: return std::vector{1,0,0,0,1,0,0,0,1}; - } - }; - auto init_matrix_I3321_21 = [&]() - { - switch(my_rank) - { - case 0: return std::vector{1,0,0,0,0,1}; - case 1: return std::vector{0,1,0}; - } - }; - auto init_matrix_A5522_32 = [&]() - { - switch(my_rank) - { - case 0: return std::vector{1,2,5,6,7,10}; - case 1: return std::vector{3,4,8,9}; - case 2: return std::vector{11,12,15,16,17,20}; - case 3: return std::vector{13,14,18,19}; - case 4: return std::vector{21,22,25}; - case 5: return std::vector{23,24}; - } - }; - */ - auto init_matrix_I3311_32 = [&]() - { - switch(my_rank) - { - case 0: return std::vector{1,0}; - case 1: return std::vector{0}; - case 2: return std::vector{0,0}; - case 3: return std::vector{1}; - case 4: return std::vector{0,1}; - case 5: return std::vector{0}; - } - }; - auto init_matrix_I5511_32 = [&]() - { - switch(my_rank) - { - case 0: return std::vector{1,0,0,0,0,0}; - case 1: return std::vector{0,0,0,1}; - case 2: return std::vector{0,0,0,0,0,1}; - case 3: return std::vector{1,0,0,0}; - case 4: return std::vector{0,1,0}; - case 5: return std::vector{0,0}; - } - }; - auto init_matrix_I5521_32 = [&]() - { - switch(my_rank) - { - case 0: return std::vector{1,0,0,0,0,0}; - case 1: return std::vector{0,1,0,0}; - case 2: return std::vector{0,0,1,0,0,0}; - case 3: return std::vector{0,0,0,1}; - case 4: return std::vector{0,0,1}; - case 5: return std::vector{0,0}; - } - }; - auto init_matrix_I5522_32 = [&]() - { - switch(my_rank) - { - case 0: return std::vector{1,0,0,1,0,0}; - case 1: return std::vector{0,0,0,0}; - case 2: return std::vector{0,0,0,0,0,0}; - case 3: return std::vector{1,0,0,1}; - case 4: return std::vector{0,0,1}; - case 5: return std::vector{0,0}; - } - }; - auto init_matrix_A5321_32 = [&]() - { - switch(my_rank) - { - case 0: return std::vector{1,6,3,8}; - case 1: return std::vector{2,7}; - case 2: return std::vector{11,16,13,18}; - case 3: return std::vector{12,17}; - case 4: return std::vector{21,23}; - case 5: return std::vector{22}; - } - }; - auto init_matrix_A5521_32 = [&]() - { - switch(my_rank) - { - case 0: return std::vector{1,6,3,8,5,10}; - case 1: return std::vector{2,7,4,9}; - case 2: return std::vector{11,16,13,18,15,20}; - case 3: return std::vector{12,17,14,19}; - case 4: return std::vector{21,23,25}; - case 5: return std::vector{22,24}; - } - }; - auto init_matrix_A5522_32 = [&]() - { - switch(my_rank) - { - case 0: return std::vector{1,6,2,7,5,10}; - case 1: return std::vector{3,8,4,9}; - case 2: return std::vector{11,16,12,17,15,20}; - case 3: return std::vector{13,18,14,19}; - case 4: return std::vector{21,22,25}; - case 5: return std::vector{23,24}; - } - }; + std::vector descv_5321_32 = init_desc(5,3,2,1,0,0); + int *desc_5321_32 = ModuleBase::GlobalFunc::VECTOR_TO_PTR(descv_5321_32); + + std::vector descv_5521_32 = init_desc(5,5,2,1,0,0); + int *desc_5521_32 = ModuleBase::GlobalFunc::VECTOR_TO_PTR(descv_5521_32); + + std::vector descv_5522_32 = init_desc(5,5,2,2,0,0); + int *desc_5522_32 = ModuleBase::GlobalFunc::VECTOR_TO_PTR(descv_5522_32); + + std::vector descv_5511_32 = init_desc(5,5,1,1,0,0); + int *desc_5511_32 = ModuleBase::GlobalFunc::VECTOR_TO_PTR(descv_5511_32); + + /* + auto init_matrix_A5624_32 = [&]() + { + switch(my_rank) + { + case 0: return std::vector{0,1,2,3,6,7,8,9}; + case 1: return std::vector{4,5,10,11}; + case 2: return std::vector{12,13,14,15,18,19,20,21}; + case 3: return std::vector{16,17,22,23}; + case 4: return std::vector{24,25,26,27}; + case 5: return std::vector{28,29}; + } + }; + auto init_matrix_I21 = [&]() + { + switch(my_rank) + { + case 0: return std::vector{1,0,0,0,0,0}; + case 1: return std::vector{0,0,0,1,0,0}; + case 2: return std::vector{0,1,0,0,0,0}; + case 3: return std::vector{0,0,0,0,1,0}; + case 4: return std::vector{0,0,1,0,0,0}; + case 5: return std::vector{0,0,0,0,0,1}; + } + }; + auto init_matrix_I11 = [&]() + { + switch(my_rank) + { + case 0: return std::vector{1,0,0,0,0,0}; + case 1: return std::vector{0,0,0,0,1,0}; + case 2: return std::vector{0,0,0,0,0,1}; + case 3: return std::vector{1,0,0,0,0,0}; + case 4: return std::vector{0,1,0,0,0,0}; + case 5: return std::vector{0,0,0,0,0,1}; + } + }; + auto init_matrix_I4411 = [&]() + { + switch(my_rank) + { + case 0: return std::vector{1,0,0,1}; + case 1: return std::vector{0,0,0,0}; + case 2: return std::vector{0,0,0,0}; + case 3: return std::vector{1,0,0,1}; + } + }; + auto init_matrix_I4422 = [&]() + { + switch(my_rank) + { + case 0: return std::vector{1,0,0,1}; + case 1: return std::vector{0,0,0,0}; + case 2: return std::vector{0,0,0,0}; + case 3: return std::vector{1,0,0,1}; + } + }; + auto init_matrix_I5522 = [&]() + { + switch(my_rank) + { + case 0: return std::vector{1,0,0,0,1,0,0,0,1}; + case 1: return std::vector{0,0,0,0,0,0}; + case 2: return std::vector{0,0,0,0,0,0}; + case 3: return std::vector{1,0,0,1}; + } + }; + auto init_matrix_I5511_32 = [&]() + { + switch(my_rank) + { + case 0: return std::vector{1,0,0,0,0}; + case 1: return std::vector{0,0,0,1}; + case 2: return std::vector{0,0,0,0,0,1}; + case 3: return std::vector{1,0,0,0}; + case 4: return std::vector{0,1,0}; + case 5: return std::vector{0,0}; + } + }; + auto init_matrix_I5521_32 = [&]() + { + switch(my_rank) + { + case 0: return std::vector{1,0,0,0,0,0}; + case 1: return std::vector{0,0,1,0}; + case 2: return std::vector{0,1,0,0,0,0}; + case 3: return std::vector{0,0,0,1}; + case 4: return std::vector{0,0,1}; + case 5: return std::vector{0,0}; + } + }; + auto init_matrix_I5522_32 = [&]() + { + switch(my_rank) + { + case 0: return std::vector{1,0,0,0,1,0}; + case 1: return std::vector{0,0,0,0}; + case 2: return std::vector{0,0,0,0,0,0}; + case 3: return std::vector{1,0,0,1}; + case 4: return std::vector{0,0,1}; + case 5: return std::vector{0,0}; + } + }; + auto init_matrix_I3321_11 = [&]() + { + switch(my_rank) + { + case 0: return std::vector{1,0,0,0,1,0,0,0,1}; + } + }; + auto init_matrix_I3321_21 = [&]() + { + switch(my_rank) + { + case 0: return std::vector{1,0,0,0,0,1}; + case 1: return std::vector{0,1,0}; + } + }; + auto init_matrix_A5522_32 = [&]() + { + switch(my_rank) + { + case 0: return std::vector{1,2,5,6,7,10}; + case 1: return std::vector{3,4,8,9}; + case 2: return std::vector{11,12,15,16,17,20}; + case 3: return std::vector{13,14,18,19}; + case 4: return std::vector{21,22,25}; + case 5: return std::vector{23,24}; + } + }; + */ + auto init_matrix_I3311_32 = [&]() + { + switch(my_rank) + { + case 0: return std::vector{1,0}; + case 1: return std::vector{0}; + case 2: return std::vector{0,0}; + case 3: return std::vector{1}; + case 4: return std::vector{0,1}; + case 5: return std::vector{0}; + } + }; + auto init_matrix_I5511_32 = [&]() + { + switch(my_rank) + { + case 0: return std::vector{1,0,0,0,0,0}; + case 1: return std::vector{0,0,0,1}; + case 2: return std::vector{0,0,0,0,0,1}; + case 3: return std::vector{1,0,0,0}; + case 4: return std::vector{0,1,0}; + case 5: return std::vector{0,0}; + } + }; + auto init_matrix_I5521_32 = [&]() + { + switch(my_rank) + { + case 0: return std::vector{1,0,0,0,0,0}; + case 1: return std::vector{0,1,0,0}; + case 2: return std::vector{0,0,1,0,0,0}; + case 3: return std::vector{0,0,0,1}; + case 4: return std::vector{0,0,1}; + case 5: return std::vector{0,0}; + } + }; + auto init_matrix_I5522_32 = [&]() + { + switch(my_rank) + { + case 0: return std::vector{1,0,0,1,0,0}; + case 1: return std::vector{0,0,0,0}; + case 2: return std::vector{0,0,0,0,0,0}; + case 3: return std::vector{1,0,0,1}; + case 4: return std::vector{0,0,1}; + case 5: return std::vector{0,0}; + } + }; + auto init_matrix_A5321_32 = [&]() + { + switch(my_rank) + { + case 0: return std::vector{1,6,3,8}; + case 1: return std::vector{2,7}; + case 2: return std::vector{11,16,13,18}; + case 3: return std::vector{12,17}; + case 4: return std::vector{21,23}; + case 5: return std::vector{22}; + } + }; + auto init_matrix_A5521_32 = [&]() + { + switch(my_rank) + { + case 0: return std::vector{1,6,3,8,5,10}; + case 1: return std::vector{2,7,4,9}; + case 2: return std::vector{11,16,13,18,15,20}; + case 3: return std::vector{12,17,14,19}; + case 4: return std::vector{21,23,25}; + case 5: return std::vector{22,24}; + } + }; + auto init_matrix_A5522_32 = [&]() + { + switch(my_rank) + { + case 0: return std::vector{1,6,2,7,5,10}; + case 1: return std::vector{3,8,4,9}; + case 2: return std::vector{11,16,12,17,15,20}; + case 3: return std::vector{13,18,14,19}; + case 4: return std::vector{21,22,25}; + case 5: return std::vector{23,24}; + } + }; - std::vector mA5321_32 = init_matrix_A5321_32(); - std::vector mA5522_32 = init_matrix_A5522_32(); - std::vector mI3311_32 = init_matrix_I3311_32(); - std::vector mI5511_32 = init_matrix_I5511_32(); - std::vector mI5521_32 = init_matrix_I5521_32(); - std::vector mC(100); - ScalapackConnector::pgemm( 'T','N', 3,5,5, 1, ModuleBase::GlobalFunc::VECTOR_TO_PTR(mA5321_32),1,1,desc_5321_32, ModuleBase::GlobalFunc::VECTOR_TO_PTR(mI5511_32),1,1,desc_5511_32, 0, ModuleBase::GlobalFunc::VECTOR_TO_PTR(mC),1,1,desc_5522_32 ); - - auto check_matrix = [&](const std::vector &v1, const std::vector &v2) - { - const size_t len = min(v1.size(),v2.size()); - for(int i=0; i mA5321_32 = init_matrix_A5321_32(); + std::vector mA5522_32 = init_matrix_A5522_32(); + std::vector mI3311_32 = init_matrix_I3311_32(); + std::vector mI5511_32 = init_matrix_I5511_32(); + std::vector mI5521_32 = init_matrix_I5521_32(); + std::vector mC(100); + ScalapackConnector::pgemm( 'T','N', 3,5,5, 1, ModuleBase::GlobalFunc::VECTOR_TO_PTR(mA5321_32),1,1,desc_5321_32, ModuleBase::GlobalFunc::VECTOR_TO_PTR(mI5511_32),1,1,desc_5511_32, 0, ModuleBase::GlobalFunc::VECTOR_TO_PTR(mC),1,1,desc_5522_32 ); + + auto check_matrix = [&](const std::vector &v1, const std::vector &v2) + { + const size_t len = min(v1.size(),v2.size()); + for(int i=0; i static std::ostream & operator<<( std::ostream & os, const std::pair &p ) { - os<<"<"<"; - return os; + os<<"<"<"; + return os; } // Peize Lin add 2016-06-06 template static std::ostream & operator<<( std::ostream & os, const std::vector &v ) { - os<<"["; - for( const T &i : v ) - os< static std::ostream & operator<<( std::ostream & os, const std::valarray &v ) { - os<<"["; - for( const T &i : v ) - os< static std::ostream & operator<<( std::ostream & os, const std::array &v ) { - os<<"["; - for( const T &i : v ) - os< static std::ostream & operator<<( std::ostream & os, const std::set &v ) { - os<<"("; - for( const T &i : v ) - os< static std::ostream & operator<<( std::ostream & os, const std::map &v ) { - for( const auto &i : v ) + for( const auto &i : v ) // os<<"{"< get_memory(const int N) { - std::vector m; - std::string s; - std::ifstream ifs("/proc/meminfo"); - ifs>>s>>s; - m.push_back(stoi(s)/1024); - for(int i=1; i>s>>s>>s; - m.push_back(stoi(s)/1024); - } - ifs.close(); - return m; + std::vector m; + std::string s; + std::ifstream ifs("/proc/meminfo"); + ifs>>s>>s; + m.push_back(stoi(s)/1024); + for(int i=1; i>s>>s>>s; + m.push_back(stoi(s)/1024); + } + ifs.close(); + return m; } #endif // TEST_FUNCTION_H diff --git a/source/source_lcao/module_rt/band_energy.h b/source/source_lcao/module_rt/band_energy.h index 107cd749e59..a2d4736824b 100644 --- a/source/source_lcao/module_rt/band_energy.h +++ b/source/source_lcao/module_rt/band_energy.h @@ -1,5 +1,5 @@ /** - * @file bandenegy.h + * @file band_energy.h * @brief compute band energy ekb * This file originally belonged to file LCAO_evolve.cpp */ diff --git a/source/source_lcao/module_rt/force_rt_overlap.cpp b/source/source_lcao/module_rt/force_rt_overlap.cpp index 86f70c3e6d1..292dfd5301e 100644 --- a/source/source_lcao/module_rt/force_rt_overlap.cpp +++ b/source/source_lcao/module_rt/force_rt_overlap.cpp @@ -9,7 +9,7 @@ #include "td_info.h" template <> void cal_foverlap_rt(ModuleBase::matrix& foverlap, - const LCAO_domain::Setup_DM>& dmat, + const module_dm::Setup_DM>& dmat, hamilt::Hamilt>* p_hamilt, const K_Vectors& kv, Parallel_Orbitals& pv, @@ -25,31 +25,26 @@ void cal_foverlap_rt(ModuleBase::matrix& foverlap, // allocate matrix const long nloc = pv.nloc; const int nrow = pv.nrow; - std::complex* Htmp = new std::complex[nloc]; - std::complex* Sinv = new std::complex[nloc]; - std::complex* dsxk = new std::complex[nloc]; - std::complex* pdsxk = new std::complex[nloc]; - std::complex* tmp1 = new std::complex[nloc]; - std::complex* tmp2 = new std::complex[nloc]; - std::complex* tmp3 = new std::complex[nloc]; - std::complex* Hybridtmp = new std::complex[nloc]; - std::vector*> tmp_out = {nullptr, nullptr, nullptr}; - for (int dir = 0; dir < 3; dir++) - { - tmp_out[dir] = new std::complex[nloc]; - ModuleBase::GlobalFunc::ZEROS(tmp_out[dir], nloc); - } + std::vector> Htmp(nloc); + std::vector> Sinv(nloc); + std::vector> dsxk(nloc); + std::vector> pdsxk(nloc); + std::vector> tmp1(nloc); + std::vector> tmp2(nloc); + std::vector> tmp3(nloc); + std::vector> Hybridtmp(nloc); + std::vector>> tmp_out(3, std::vector>(nloc, 0.0)); for (int ik = 0; ik < kv.get_nks(); ++ik) { p_hamilt->updateHk(ik); // get dmk - std::complex* tmp_dmk = dmat.dm->get_DMK_pointer(ik); + std::complex* tmp_dmk = dmat.dm->get_dmk_ptr(ik); - ModuleBase::GlobalFunc::ZEROS(Htmp, nloc); - ModuleBase::GlobalFunc::ZEROS(Sinv, nloc); - ModuleBase::GlobalFunc::ZEROS(tmp1, nloc); - ModuleBase::GlobalFunc::ZEROS(tmp2, nloc); - ModuleBase::GlobalFunc::ZEROS(Hybridtmp, nloc); + ModuleBase::GlobalFunc::ZEROS(Htmp.data(), nloc); + ModuleBase::GlobalFunc::ZEROS(Sinv.data(), nloc); + ModuleBase::GlobalFunc::ZEROS(tmp1.data(), nloc); + ModuleBase::GlobalFunc::ZEROS(tmp2.data(), nloc); + ModuleBase::GlobalFunc::ZEROS(Hybridtmp.data(), nloc); const int inc = 1; @@ -57,8 +52,8 @@ void cal_foverlap_rt(ModuleBase::matrix& foverlap, hamilt::MatrixBlock> s_mat; // get Hk Sk p_hamilt->matrix(h_mat, s_mat); - BlasConnector::copy(nloc, h_mat.p, inc, Htmp, inc); - BlasConnector::copy(nloc, s_mat.p, inc, Sinv, inc); + BlasConnector::copy(nloc, h_mat.p, inc, Htmp.data(), inc); + BlasConnector::copy(nloc, s_mat.p, inc, Sinv.data(), inc); vector ipiv(nloc, 0); int info = 0; @@ -68,7 +63,7 @@ void cal_foverlap_rt(ModuleBase::matrix& foverlap, const std::complex mone_complex = {-1.0, 0.0}; const std::complex zero_complex = {0.0, 0.0}; - ScalapackConnector::getrf(nlocal, nlocal, Sinv, one_int, one_int, pv.desc, ipiv.data(), &info); + ScalapackConnector::getrf(nlocal, nlocal, Sinv.data(), one_int, one_int, pv.desc, ipiv.data(), &info); int lwork = -1; int liwork = -1; @@ -79,14 +74,14 @@ void cal_foverlap_rt(ModuleBase::matrix& foverlap, // if liwork = -1, then the size of iwork is (at least) of length 1. std::vector iwork(1, 0); - ScalapackConnector::getri(nlocal, Sinv, one_int, one_int, pv.desc, ipiv.data(), work.data(), &lwork, iwork.data(), &liwork, &info); + ScalapackConnector::getri(nlocal, Sinv.data(), one_int, one_int, pv.desc, ipiv.data(), work.data(), &lwork, iwork.data(), &liwork, &info); lwork = work[0].real(); work.resize(lwork, 0); liwork = iwork[0]; iwork.resize(liwork, 0); - ScalapackConnector::getri(nlocal, Sinv, one_int, one_int, pv.desc, ipiv.data(), work.data(), &lwork, iwork.data(), &liwork, &info); + ScalapackConnector::getri(nlocal, Sinv.data(), one_int, one_int, pv.desc, ipiv.data(), work.data(), &lwork, iwork.data(), &liwork, &info); const char N_char = 'N'; const char T_char = 'T'; @@ -102,12 +97,12 @@ void cal_foverlap_rt(ModuleBase::matrix& foverlap, one_int, one_int, pv.desc, - Htmp, + Htmp.data(), one_int, one_int, pv.desc, zero_complex, - tmp1, + tmp1.data(), one_int, one_int, pv.desc); @@ -118,50 +113,51 @@ void cal_foverlap_rt(ModuleBase::matrix& foverlap, nlocal, nlocal, one_complex, - tmp1, + tmp1.data(), one_int, one_int, pv.desc, - Sinv, + Sinv.data(), one_int, one_int, pv.desc, zero_complex, - tmp2, + tmp2.data(), one_int, one_int, pv.desc); for (int dir = 0; dir < 3; dir++) { - ModuleBase::GlobalFunc::ZEROS(dsxk, nloc); - ModuleBase::GlobalFunc::ZEROS(pdsxk, nloc); - ModuleBase::GlobalFunc::ZEROS(tmp3, nloc); - module_rt::folding_HR_td(*dsxr[dir], dsxk, kv.kvec_d[ik], TD_info::cart_At, TD_info::td_vel_op->get_phase_hybrid(), nrow, 1); + ModuleBase::GlobalFunc::ZEROS(dsxk.data(), nloc); + ModuleBase::GlobalFunc::ZEROS(pdsxk.data(), nloc); + ModuleBase::GlobalFunc::ZEROS(tmp3.data(), nloc); + module_rt::folding_HR_td(*dsxr[dir], dsxk.data(), kv.kvec_d[ik], TD_info::A_prop_ha, TD_info::td_vel_op->get_phase_hybrid(), nrow, 1); + const ModuleBase::Vector3 electric_field_ry = elecstate::H_TDDFT_pw::efield_ha * 2.0; module_rt::folding_partial_dot(*dsxr[dir], - pdsxk, + pdsxk.data(), kv.kvec_d[ik], nrow, 1, &ucell, TD_info::td_vel_op->get_phase_hybrid(), - TD_info::cart_At, - elecstate::H_TDDFT_pw::Et); + TD_info::A_prop_ha, + electric_field_ry); ScalapackConnector::gemm(N_char, N_char, nlocal, nlocal, nlocal, two_complex, - tmp2, + tmp2.data(), one_int, one_int, pv.desc, - dsxk, + dsxk.data(), one_int, one_int, pv.desc, one_complex, - tmp_out[dir], + tmp_out[dir].data(), one_int, one_int, pv.desc); @@ -176,12 +172,12 @@ void cal_foverlap_rt(ModuleBase::matrix& foverlap, one_int, one_int, pv.desc, - pdsxk, + pdsxk.data(), one_int, one_int, pv.desc, one_complex, - tmp3, + tmp3.data(), one_int, one_int, pv.desc); @@ -190,23 +186,17 @@ void cal_foverlap_rt(ModuleBase::matrix& foverlap, nlocal, nlocal, one_complex, - tmp3, + tmp3.data(), one_int, one_int, pv.desc, one_complex, - tmp_out[dir], + tmp_out[dir].data(), one_int, one_int, pv.desc); } } - delete[] Htmp; - delete[] Sinv; - delete[] tmp1; - delete[] tmp2; - delete[] dsxk; - delete[] Hybridtmp; // std::string filename = "process_debug_" + std::to_string(GlobalV::MY_RANK) + ".txt"; // std::ofstream debug_file(filename); // debug_file << "=== Process " << GlobalV::MY_RANK << " ===" << std::endl; @@ -236,7 +226,7 @@ void cal_foverlap_rt(ModuleBase::matrix& foverlap, int row0 = pv.atom_begin_row[iat]; int col0 = pv.atom_begin_col[iat]; const int row_size = pv.get_row_size(); - std::vector*> p_diag = {tmp_out[0], tmp_out[1], tmp_out[2]}; + std::vector*> p_diag = {tmp_out[0].data(), tmp_out[1].data(), tmp_out[2].data()}; for (int mu = 0; mu < pv.get_nrow_atom(iat); ++mu) { for (int nu = 0; nu < pv.get_ncol_atom(iat); ++nu) @@ -254,16 +244,12 @@ void cal_foverlap_rt(ModuleBase::matrix& foverlap, } } Parallel_Reduce::reduce_all(foverlap.c, foverlap.nr * foverlap.nc); - for (int dir = 0; dir < 3; dir++) - { - delete[] tmp_out[dir]; - } return; #endif } template <> void cal_foverlap_rt(ModuleBase::matrix& foverlap, - const LCAO_domain::Setup_DM& dmat, + const module_dm::Setup_DM& dmat, hamilt::Hamilt* p_hamilt, const K_Vectors& kv, Parallel_Orbitals& pv, diff --git a/source/source_lcao/module_rt/force_rt_overlap.h b/source/source_lcao/module_rt/force_rt_overlap.h index cfffc32bb95..7c18e5a2d55 100644 --- a/source/source_lcao/module_rt/force_rt_overlap.h +++ b/source/source_lcao/module_rt/force_rt_overlap.h @@ -3,12 +3,12 @@ #include "source_basis/module_ao/parallel_orbitals.h" #include "source_cell/klist.h" -#include "source_lcao/setup_dm.h" +#include "source_lcao/allocate_dm.h" #include "source_hamilt/hamilt.h" template void cal_foverlap_rt(ModuleBase::matrix& foverlap, - const LCAO_domain::Setup_DM& dmat, + const module_dm::Setup_DM& dmat, hamilt::Hamilt* p_hamilt, const K_Vectors& kv, Parallel_Orbitals& pv, diff --git a/source/source_lcao/module_rt/kernels/cuda/snap_psibeta_gpu.cu b/source/source_lcao/module_rt/kernels/cuda/snap_psibeta_gpu.cu index 37ad83f4d63..9a4e4b41ccc 100644 --- a/source/source_lcao/module_rt/kernels/cuda/snap_psibeta_gpu.cu +++ b/source/source_lcao/module_rt/kernels/cuda/snap_psibeta_gpu.cu @@ -126,7 +126,7 @@ void snap_psibeta_atom_batch_gpu( // Early exit if no projectors on center atom //========================================================================= - const int nproj = infoNL_.nproj[T0]; + const int nproj = infoNL_.get_nproj(T0); if (nproj == 0) { ModuleBase::timer::end("module_rt", "snap_psibeta_gpu"); @@ -143,7 +143,7 @@ void snap_psibeta_atom_batch_gpu( for (int ip = 0; ip < nproj; ip++) { proj_m0_offset_h[ip] = natomwfc; - int L0 = infoNL_.Beta[T0].Proj[ip].getL(); + int L0 = infoNL_.get_Beta(T0).Proj[ip].getL(); // Validate angular momentum if (L0 > MAX_L) @@ -257,7 +257,7 @@ void snap_psibeta_atom_batch_gpu( for (int ip = 0; ip < nproj; ip++) { - const auto& proj = infoNL_.Beta[T0].Proj[ip]; + const auto& proj = infoNL_.get_Beta(T0).Proj[ip]; const int L0 = proj.getL(); const int mesh = proj.getNr(); const double rcut = proj.getRcut(); diff --git a/source/source_lcao/module_rt/snap_psb_half_tddft.cpp b/source/source_lcao/module_rt/snap_psb_half_tddft.cpp index bd00c18cba3..fd5ac163c1c 100644 --- a/source/source_lcao/module_rt/snap_psb_half_tddft.cpp +++ b/source/source_lcao/module_rt/snap_psb_half_tddft.cpp @@ -35,12 +35,12 @@ void snap_psibeta_half_tddft(const LCAO_Orbitals& orb, const SnapIntegrationOptions& options) { std::vector channels; - channels.reserve(infoNL_.nproj[T0]); + channels.reserve(infoNL_.get_nproj(T0)); // UPF nonlocal beta projectors already follow the r * beta_l(r) convention. - for (int ip = 0; ip < infoNL_.nproj[T0]; ++ip) + for (int ip = 0; ip < infoNL_.get_nproj(T0); ++ip) { - const auto& proj = infoNL_.Beta[T0].Proj[ip]; + const auto& proj = infoNL_.get_Beta(T0).Proj[ip]; ProjectorChannel channel; channel.l = proj.getL(); channel.mesh = proj.getNr(); diff --git a/source/source_lcao/module_rt/td_info.cpp b/source/source_lcao/module_rt/td_info.cpp index fd3029a0c4c..4394d6a19aa 100644 --- a/source/source_lcao/module_rt/td_info.cpp +++ b/source/source_lcao/module_rt/td_info.cpp @@ -1,50 +1,29 @@ #include "td_info.h" -#include "source_base/global_variable.h" #include "source_base/libm/libm.h" +#include "source_cell/module_neighbor/sltk_grid_driver.h" #include "source_estate/module_pot/h_tddft_pw.h" -#include "source_io/module_efield/td_vector_pot_io.h" #include "source_io/module_parameter/parameter.h" bool TD_info::out_mat_R = false; -bool TD_info::out_vecpot = false; int TD_info::out_current = 0; bool TD_info::out_current_k = false; -bool TD_info::init_vecpot_file = false; bool TD_info::evolve_once = false; TD_info* TD_info::td_vel_op = nullptr; int TD_info::estep_shift = 0; int TD_info::istep = -1; -int TD_info::max_istep = -1; -ModuleBase::Vector3 TD_info::cart_At; -std::vector> TD_info::At_from_file; +ModuleBase::Vector3 TD_info::A_prop_ha; -TD_info::TD_info(const UnitCell* ucell_in, const Parallel_Orbitals& pv, const LCAO_Orbitals& orb) +TD_info::TD_info(const UnitCell* ucell_in, const Parallel_Orbitals& pv, const LCAO_Orbitals& orb, const int restart_step) { - if (init_vecpot_file && istep == -1) - { - At_from_file = ModuleIO::read_td_vector_pot(""); - max_istep = At_from_file.size() - 1; - } - // read in restart step - if (PARAM.inp.mdp.md_restart) - { - std::stringstream ssc; - ssc << PARAM.globalv.global_readin_dir << "Restart_td.txt"; - std::ifstream file(ssc.str().c_str()); - if (!file) - { - ModuleBase::WARNING_QUIT("TD_info::TD_info", "No Restart_td.txt!"); - } - file >> estep_shift; - // std::cout<<"estep_shift"<istep += estep_shift; + estep_shift = restart_step; if (out_current == 2 || elecstate::H_TDDFT_pw::stype == 2) { - r_calculator.init(*ucell_in, pv, orb); + const bool cal_force = PARAM.inp.cal_force; + const int nlocal = PARAM.globalv.nlocal; + r_calculator.init(*ucell_in, pv, orb, cal_force, nlocal); } return; } @@ -70,23 +49,10 @@ TD_info::~TD_info() } } -void TD_info::cal_cart_At(const ModuleBase::Vector3& At) +void TD_info::set_A_prop(const int step, const ModuleBase::Vector3& A_ha) { - istep++; - if (init_vecpot_file) - { - cart_At = At_from_file[istep > max_istep ? max_istep : istep]; - } - else - { - // transfrom into atomic unit - cart_At = At / 2.0; - } - // output the vector potential if needed - if (out_vecpot && GlobalV::MY_RANK == 0) - { - ModuleIO::write_td_vector_pot(PARAM.globalv.global_out_dir, istep, cart_At); - } + istep = step; + A_prop_ha = A_ha; // update hybrid gauge phase if (elecstate::H_TDDFT_pw::stype == 2) { @@ -94,7 +60,7 @@ void TD_info::cal_cart_At(const ModuleBase::Vector3& At) { const ModuleBase::Vector3& r_index = phase_pair.first; ModuleBase::Vector3 dR = double(r_index.x) * a1 + double(r_index.y) * a2 + double(r_index.z) * a3; - const double arg_td = cart_At * dR * lat0; + const double arg_td = A_prop_ha * dR * lat0; double sinp, cosp; ModuleBase::libm::sincos(arg_td, &sinp, &cosp); phase_hybrid[r_index] = std::complex(cosp, sinp); @@ -102,28 +68,6 @@ void TD_info::cal_cart_At(const ModuleBase::Vector3& At) } } -void TD_info::out_restart_info(const int nstep, - const ModuleBase::Vector3& At_current, - const ModuleBase::Vector3& At_laststep) -{ - if (GlobalV::MY_RANK == 0) - { - // open file - std::string outdir = PARAM.globalv.global_out_dir + "Restart_td.txt"; - std::ofstream outFile(outdir); - if (!outFile) - { - ModuleBase::WARNING_QUIT("out_restart_info", "no Restart_td.txt!"); - } - // write data - outFile << nstep << std::endl; - outFile << At_current[0] << " " << At_current[1] << " " << At_current[2] << std::endl; - outFile << At_laststep[0] << " " << At_laststep[1] << " " << At_laststep[2] << std::endl; - outFile.close(); - } - - return; -} template void TD_info::initialize_phase_hybrid(const UnitCell& ucell, const hamilt::HContainer* hR) { @@ -141,7 +85,7 @@ void TD_info::initialize_phase_hybrid(const UnitCell& ucell, const hamilt::HCont continue; ModuleBase::Vector3 dR = double(r_index.x) * a1 + double(r_index.y) * a2 + double(r_index.z) * a3; - const double arg_td = cart_At * dR * lat0; + const double arg_td = A_prop_ha * dR * lat0; double sinp, cosp; ModuleBase::libm::sincos(arg_td, &sinp, &cosp); phase_hybrid[r_index] = std::complex(cosp, sinp); @@ -334,5 +278,6 @@ void TD_info::calculate_grad_overlap(const Parallel_Orbitals& paraV, } ModuleBase::timer::end("TD_info", "calculate_grad_overlap"); } -template void TD_info::initialize_phase_hybrid>(const UnitCell& ucell, const hamilt::HContainer>* hR); +template void TD_info::initialize_phase_hybrid>(const UnitCell& ucell, + const hamilt::HContainer>* hR); template void TD_info::initialize_phase_hybrid(const UnitCell& ucell, const hamilt::HContainer* hR); diff --git a/source/source_lcao/module_rt/td_info.h b/source/source_lcao/module_rt/td_info.h index 797b0c83fb3..40960aa9ec3 100644 --- a/source/source_lcao/module_rt/td_info.h +++ b/source/source_lcao/module_rt/td_info.h @@ -1,17 +1,17 @@ #ifndef TD_INFO_H #define TD_INFO_H -#include "source_lcao/module_ri/abfs_vector3_order.h" #include "source_base/timer.h" -#include "source_hamilt/module_hcontainer/hcontainer.h" -#include "source_io/module_hs/cal_r_overlap_r.h" #include "source_basis/module_nao/two_center_integrator.h" +#include "source_hamilt/module_hcontainer/hcontainer.h" +#include "source_io/module_hs/pos_op_mat.h" +#include "source_lcao/module_ri/abfs_vector3_order.h" #include // Class to store TDDFT infos, mainly for periodic system. class TD_info { public: - TD_info(const UnitCell* ucell_in,const Parallel_Orbitals& pv, const LCAO_Orbitals& orb); + TD_info(const UnitCell* ucell_in, const Parallel_Orbitals& pv, const LCAO_Orbitals& orb, const int restart_step); ~TD_info(); /// @brief switch to control the output of HR @@ -20,34 +20,23 @@ class TD_info /// @brief pointer to the only TD_info object itself static TD_info* td_vel_op; - /// @brief switch to control the output of At - static bool out_vecpot; - /// @brief switch to control the output of current static int out_current; /// @brief switch to control the format of the output current, in total or in each k-point static bool out_current_k; - /// @brief switch to control the source of At - static bool init_vecpot_file; - /// @brief if need to calculate more than once static bool evolve_once; /// @brief Restart step static int estep_shift; - /// @brief Store the vector potential for tddft calculation - static ModuleBase::Vector3 cart_At; - - /// @brief calculate the At in cartesian coordinate - void cal_cart_At(const ModuleBase::Vector3& At); + /** @brief Propagation vector potential in Hartree atomic units, published by the ESolver. */ + static ModuleBase::Vector3 A_prop_ha; - /// @brief output RT-TDDFT info for restart - void out_restart_info(const int nstep, - const ModuleBase::Vector3& At_current, - const ModuleBase::Vector3& At_laststep); + /** @brief Bind the manager-selected propagation A and refresh hybrid phases at an explicit step. */ + void set_A_prop(const int step, const ModuleBase::Vector3& A_ha); // allocate memory for current term. void initialize_current_term(const hamilt::HContainer>* HR, const Parallel_Orbitals* paraV); @@ -59,7 +48,7 @@ class TD_info // allocate memory for phase_hybrid. template void initialize_phase_hybrid(const UnitCell& ucell, const hamilt::HContainer* hR); - + const std::map, std::complex>& get_phase_hybrid() const { return this->phase_hybrid; @@ -72,7 +61,7 @@ class TD_info const TwoCenterIntegrator* intor); std::vector*> get_grad_overlap() const { - return this->grad_overlap; + return this->grad_overlap; } // set velocity HR. void set_velocity_HR(hamilt::HContainer>* HR) @@ -86,17 +75,17 @@ class TD_info int get_istep() { - return istep; + return istep; } // For TDDFT velocity gauge, to fix the output of HR std::map, std::map>>> HR_sparse_td_vel[2]; //r_calculator - cal_r_overlap_R r_calculator; + Position_op r_calculator; private: /// @brief lattice vectors, used to calculate the extra phase for hybrid gauge - ModuleBase::Vector3a1, a2, a3; + ModuleBase::Vector3 a1, a2, a3; double lat0; /// @brief store time-dependent phase for hybrid gauge @@ -105,12 +94,6 @@ class TD_info /// @brief store isteps now static int istep; - /// @brief total steps of read in At - static int max_istep; - - /// @brief store the read in At_data - static std::vector> At_from_file; - /// @brief store the dS/dD matrix std::vector*> grad_overlap = {nullptr, nullptr, nullptr}; diff --git a/source/source_lcao/module_rt/test/CMakeLists.txt b/source/source_lcao/module_rt/test/CMakeLists.txt index 6423db20032..07f0018a68f 100644 --- a/source/source_lcao/module_rt/test/CMakeLists.txt +++ b/source/source_lcao/module_rt/test/CMakeLists.txt @@ -56,9 +56,12 @@ AddTest( ../../../source_cell/read_pp_upf100.cpp ../../../source_cell/read_pp_vwr.cpp ../../../source_cell/read_pp_blps.cpp - ../../../source_io/module_hs/cal_r_overlap_r.cpp - ../../../source_io/module_hs/single_r_io.cpp - ../../../source_io/module_hs/rr_sparse_writer.cpp + ../../../source_io/module_hs/pos_op_mat.cpp + ../../../source_io/module_hs/pos_op_basis.cpp + ../../../source_io/module_hs/pos_op_calc.cpp + ../../../source_io/module_hs/pos_op_writer.cpp + ../../../source_io/module_hs/lat_r_csr.cpp + ../../../source_io/module_hs/pos_op_csr.cpp ../../../source_pw/module_pwdft/soc.cpp ${SNAP_PSIBETA_CUDA_TEST_SOURCES} ) diff --git a/source/source_lcao/module_rt/test/propagator_test1.cpp b/source/source_lcao/module_rt/test/propagator_test1.cpp index be426d6a857..0908535ceb2 100644 --- a/source/source_lcao/module_rt/test/propagator_test1.cpp +++ b/source/source_lcao/module_rt/test/propagator_test1.cpp @@ -1,9 +1,7 @@ #include -#define private public -#define protected public +#include "source_base/global_variable.h" #include "source_basis/module_ao/parallel_orbitals.h" #include "source_lcao/module_rt/propagator.h" -#include "source_io/module_parameter/parameter.h" #include "tddft_test.h" #include @@ -38,7 +36,7 @@ TEST(PropagatorTest, testPropagatorCN) pv->nloc = nlocal * nlocal; pv->ncol = nlocal; pv->set_coord(0, 0); - PARAM.input.mdp.md_dt = 4 * ModuleBase::AU_to_FS; + const double md_dt = 4 * ModuleBase::AU_to_FS; // Initialize input matrices int info; @@ -71,7 +69,7 @@ TEST(PropagatorTest, testPropagatorCN) // Call the function int propagator = 0; - module_rt::Propagator prop(propagator, pv, PARAM.mdp.md_dt); + module_rt::Propagator prop(propagator, pv, md_dt); prop.compute_propagator(nlocal, Stmp, Htmp, nullptr, U_operator, GlobalV::ofs_running, print_matrix); // Check the results diff --git a/source/source_lcao/module_rt/test/propagator_test2.cpp b/source/source_lcao/module_rt/test/propagator_test2.cpp index d7da6e27422..283ac9035a9 100644 --- a/source/source_lcao/module_rt/test/propagator_test2.cpp +++ b/source/source_lcao/module_rt/test/propagator_test2.cpp @@ -1,9 +1,7 @@ #include -#define private public -#define protected public +#include "source_base/global_variable.h" #include "source_basis/module_ao/parallel_orbitals.h" #include "source_lcao/module_rt/propagator.h" -#include "source_io/module_parameter/parameter.h" #include "tddft_test.h" #include @@ -43,7 +41,7 @@ TEST(PropagatorTest, testPropagatorTaylor) dim[0] = nprow; dim[1] = npcol; - PARAM.input.mdp.md_dt = 4 * ModuleBase::AU_to_FS; + const double md_dt = 4 * ModuleBase::AU_to_FS; // Initialize input matrices int info; @@ -76,7 +74,7 @@ TEST(PropagatorTest, testPropagatorTaylor) // Call the function int propagator = 1; - module_rt::Propagator prop(propagator, pv, PARAM.mdp.md_dt); + module_rt::Propagator prop(propagator, pv, md_dt); prop.compute_propagator(nlocal, Stmp, Htmp, nullptr, U_operator, GlobalV::ofs_running, print_matrix); // Check the results diff --git a/source/source_lcao/module_rt/test/propagator_test3.cpp b/source/source_lcao/module_rt/test/propagator_test3.cpp index 6b7f463589f..c96569e762e 100644 --- a/source/source_lcao/module_rt/test/propagator_test3.cpp +++ b/source/source_lcao/module_rt/test/propagator_test3.cpp @@ -1,9 +1,7 @@ #include -#define private public -#define protected public +#include "source_base/global_variable.h" #include "source_basis/module_ao/parallel_orbitals.h" #include "source_lcao/module_rt/propagator.h" -#include "source_io/module_parameter/parameter.h" #include "tddft_test.h" #include @@ -44,7 +42,7 @@ TEST(PropagatorTest, testPropagatorETRS) dim[0] = nprow; dim[1] = npcol; - PARAM.input.mdp.md_dt = 4 * ModuleBase::AU_to_FS; + const double md_dt = 4 * ModuleBase::AU_to_FS; // Initialize input matrices int info; @@ -80,7 +78,7 @@ TEST(PropagatorTest, testPropagatorETRS) // Call the function int propagator = 2; - module_rt::Propagator prop(propagator, pv, PARAM.mdp.md_dt); + module_rt::Propagator prop(propagator, pv, md_dt); prop.compute_propagator(nlocal, Stmp, Htmp, Hlaststep, U_operator, GlobalV::ofs_running, print_matrix); // Check the results diff --git a/source/source_lcao/module_rt/test/snap_psb_half_tddft_test.cpp b/source/source_lcao/module_rt/test/snap_psb_half_tddft_test.cpp index c8d1e4981a7..c289e079e75 100644 --- a/source/source_lcao/module_rt/test/snap_psb_half_tddft_test.cpp +++ b/source/source_lcao/module_rt/test/snap_psb_half_tddft_test.cpp @@ -1,9 +1,10 @@ #include "source_lcao/module_rt/snap_psb_half_tddft.h" #include "source_base/ylm.h" +#include "source_basis/module_ao/parallel_orbitals.h" #include "source_cell/read_pp.h" #include "source_cell/unitcell.h" -#include "source_io/module_hs/cal_r_overlap_r.h" +#include "source_io/module_hs/pos_op_mat.h" #include "../../lcao_nonlocal_info.h" #ifdef __CUDA @@ -84,7 +85,7 @@ void print_comparison_stats(const char* label, const ComparisonStats& stats) ComparisonStats compare_zero_vector_potential(const LCAO_Orbitals& orb, const UnitCell& ucell, - cal_r_overlap_R& r_calculator, + Position_op& r_calculator, const int radial_grid_num, const int lebedev_grid_points) { @@ -334,28 +335,30 @@ class SnapPsibetaHalfTddftTest : public ::testing::Test ASSERT_EQ(atom.ncpp.jjj.size(), 6); auto* lcao_nl = new LCAONonlocalInfo(); - lcao_nl->get_nonlocal().nproj = new int[1]; + lcao_nl->get_nonlocal().assign_nproj(1, 0); + lcao_nl->get_nonlocal().resize_Beta(1); std::ofstream log("snap_psibeta_half_tddft_nonlocal.log"); lcao_nl->get_nonlocal().Set_NonLocal(0, &atom, - lcao_nl->get_nonlocal().nproj[0], + lcao_nl->get_nonlocal().get_nproj_ref(0), orb.get_kmesh(), orb.get_dk(), orb.get_dr_uniform(), log, false, false, - 1); + 1, + 0); - ASSERT_EQ(lcao_nl->get_nonlocal().nproj[0], 6); - lcao_nl->get_nonlocal().nprojmax = lcao_nl->get_nonlocal().nproj[0]; - lcao_nl->get_nonlocal().rcutmax_Beta = lcao_nl->get_nonlocal().Beta[0].get_rcut_max(); + ASSERT_EQ(lcao_nl->get_nonlocal().get_nproj(0), 6); + lcao_nl->get_nonlocal().set_nprojmax(lcao_nl->get_nonlocal().get_nproj(0)); + lcao_nl->get_nonlocal().set_rcutmax_Beta(lcao_nl->get_nonlocal().get_Beta(0).get_rcut_max()); ucell.infoNL.reset(lcao_nl); } void initialize_r_overlap_reference() { - r_calculator.init_nonlocal(ucell, pv, orb); + r_calculator.init_nonlocal(ucell, pv, orb, false, ucell.atoms[0].nw); } ComparisonStats compare_zero_vector_potential(const int radial_grid_num, const int lebedev_grid_points) @@ -366,7 +369,7 @@ class SnapPsibetaHalfTddftTest : public ::testing::Test LCAO_Orbitals orb; UnitCell ucell; Parallel_Orbitals pv; - cal_r_overlap_R r_calculator; + Position_op r_calculator; }; class SnapPsibetaNonuniformHalfTddftTest : public ::testing::Test @@ -418,30 +421,32 @@ class SnapPsibetaNonuniformHalfTddftTest : public ::testing::Test pseudo_reader.complete_default(atom.ncpp, 15.0); auto* lcao_nl = new LCAONonlocalInfo(); - lcao_nl->get_nonlocal().nproj = new int[1]; + lcao_nl->get_nonlocal().assign_nproj(1, 0); + lcao_nl->get_nonlocal().resize_Beta(1); std::ofstream log("snap_psibeta_half_tddft_al_nonlocal.log"); lcao_nl->get_nonlocal().Set_NonLocal(0, &atom, - lcao_nl->get_nonlocal().nproj[0], + lcao_nl->get_nonlocal().get_nproj_ref(0), orb.get_kmesh(), orb.get_dk(), orb.get_dr_uniform(), log, false, false, - 1); - ASSERT_EQ(lcao_nl->get_nonlocal().nproj[0], 4); - lcao_nl->get_nonlocal().nprojmax = lcao_nl->get_nonlocal().nproj[0]; - lcao_nl->get_nonlocal().rcutmax_Beta = lcao_nl->get_nonlocal().Beta[0].get_rcut_max(); + 1, + 0); + ASSERT_EQ(lcao_nl->get_nonlocal().get_nproj(0), 4); + lcao_nl->get_nonlocal().set_nprojmax(lcao_nl->get_nonlocal().get_nproj(0)); + lcao_nl->get_nonlocal().set_rcutmax_Beta(lcao_nl->get_nonlocal().get_Beta(0).get_rcut_max()); ucell.infoNL.reset(lcao_nl); - r_calculator.init_nonlocal(ucell, pv, orb); + r_calculator.init_nonlocal(ucell, pv, orb, false, atom.nw); } LCAO_Orbitals orb; UnitCell ucell; Parallel_Orbitals pv; - cal_r_overlap_R r_calculator; + Position_op r_calculator; }; } // namespace @@ -490,9 +495,9 @@ TEST_F(SnapPsibetaNonuniformHalfTddftTest, NonuniformAlProjectorMatchesTwoCenter ASSERT_NE(lcao_nl, nullptr); bool found_nonuniform_spacing = false; - for (int ip = 0; ip < lcao_nl->get_nonlocal().nproj[0]; ++ip) + for (int ip = 0; ip < lcao_nl->get_nonlocal().get_nproj(0); ++ip) { - const auto& projector = lcao_nl->get_nonlocal().Beta[0].Proj[ip]; + const auto& projector = lcao_nl->get_nonlocal().get_Beta(0).Proj[ip]; ASSERT_GT(projector.getNr(), 2); const double first_spacing = projector.getRadial(1) - projector.getRadial(0); for (int ir = 2; ir < projector.getNr(); ++ir) diff --git a/source/source_lcao/module_rt/test/tddft_test.cpp b/source/source_lcao/module_rt/test/tddft_test.cpp index d70fb3a8f13..26d4793a45f 100644 --- a/source/source_lcao/module_rt/test/tddft_test.cpp +++ b/source/source_lcao/module_rt/test/tddft_test.cpp @@ -44,7 +44,7 @@ int main(int argc, char** argv) testing::InitGoogleTest(&argc, argv); result = RUN_ALL_TESTS(); - Cblacs_exit(ictxt); + Cblacs_gridexit(ictxt); // MPI_Finalize(); return result; diff --git a/source/source_lcao/module_rt/velocity_op.cpp b/source/source_lcao/module_rt/velocity_op.cpp index a34fd8492ed..f22f313d6ec 100644 --- a/source/source_lcao/module_rt/velocity_op.cpp +++ b/source/source_lcao/module_rt/velocity_op.cpp @@ -8,7 +8,7 @@ #endif #include "source_io/module_parameter/parameter.h" template -cal_r_overlap_R Velocity_op::r_calculator; +Position_op Velocity_op::r_calculator; template bool Velocity_op::init_done = false; template @@ -40,7 +40,9 @@ void Velocity_op::initialize_vcomm_r(const Grid_Driver* GridD, const Paralle ModuleBase::timer::start("Velocity_op", "initialize_vcomm_r"); if(!init_done) { - r_calculator.init_nonlocal(*ucell, *paraV, orb_); + const bool cal_force = PARAM.inp.cal_force; + const int nlocal = PARAM.globalv.nlocal; + r_calculator.init_nonlocal(*ucell, *paraV, orb_, cal_force, nlocal); init_done = true; } diff --git a/source/source_lcao/module_rt/velocity_op.h b/source/source_lcao/module_rt/velocity_op.h index 6dfb6124280..09a390d7549 100644 --- a/source/source_lcao/module_rt/velocity_op.h +++ b/source/source_lcao/module_rt/velocity_op.h @@ -7,7 +7,7 @@ #include "source_hamilt/module_hcontainer/hcontainer.h" #include "source_basis/module_nao/two_center_integrator.h" #include "source_base/vector3.h" -#include "source_io/module_hs/cal_r_overlap_r.h" +#include "source_io/module_hs/pos_op_mat.h" //design to calculate velocity operator template @@ -71,7 +71,7 @@ class Velocity_op /// @brief Store the vector potential for td_ekinetic term ModuleBase::Vector3 cart_At; - static cal_r_overlap_R r_calculator; + static Position_op r_calculator; static bool init_done; }; diff --git a/source/source_lcao/pulay_fs.h b/source/source_lcao/pulay_fs.h index a2f1d1ee1a0..3a8091df392 100644 --- a/source/source_lcao/pulay_fs.h +++ b/source/source_lcao/pulay_fs.h @@ -1,4 +1,6 @@ -#pragma once +#ifndef ABACUS_SOURCE_LCAO_PULAY_FS_H +#define ABACUS_SOURCE_LCAO_PULAY_FS_H + #include "source_estate/module_dm/density_matrix.h" #include "source_estate/module_pot/potential_new.h" #include "source_cell/unitcell.h" @@ -15,7 +17,7 @@ namespace PulayForceStress void cal_pulay_fs( ModuleBase::matrix& f, ///< [out] force ModuleBase::matrix& s, ///< [out] stress - const elecstate::DensityMatrix& dm, ///< [in] density matrix or energy density matrix + const module_dm::DensityMatrix& dm, ///< [in] density matrix or energy density matrix const UnitCell& ucell, ///< [in] unit cell const Parallel_Orbitals& pv, ///< [in] parallel orbitals const double* (&dHSx)[3], ///< [in] dHSx x, y, z, for force @@ -31,7 +33,7 @@ namespace PulayForceStress void cal_pulay_fs( ModuleBase::matrix& f, ///< [out] force ModuleBase::matrix& s, ///< [out] stress - const elecstate::DensityMatrix& dm, ///< [in] density matrix or energy density matrix + const module_dm::DensityMatrix& dm, ///< [in] density matrix or energy density matrix const UnitCell& ucell, ///< [in] unit cell const Parallel_Orbitals& pv, ///< [in] parallel orbitals const double* (&dHSx)[3], ///< [in] dHSx x, y, z, for force and stress @@ -47,12 +49,14 @@ namespace PulayForceStress void cal_pulay_fs( ModuleBase::matrix& f, ///< [out] force ModuleBase::matrix& s, ///< [out] stress - const elecstate::DensityMatrix& dm, ///< [in] density matrix or energy density matrix + const module_dm::DensityMatrix& dm, ///< [in] density matrix or energy density matrix const UnitCell& ucell, ///< [in] unit cell const elecstate::Potential* pot, ///< [in] potential on grid const bool& isforce, const bool& isstress, const bool& set_dmr_gint = true); } -#include "pulay_fs_temp.hpp" -#include "pulay_fs_gint.hpp" +#include "pulay_fs_temp.h" +#include "pulay_fs_gint.h" + +#endif // ABACUS_SOURCE_LCAO_PULAY_FS_H diff --git a/source/source_lcao/pulay_fs_center2.cpp b/source/source_lcao/pulay_fs_center2.cpp index 8511bda5f95..eacbc5088ab 100644 --- a/source/source_lcao/pulay_fs_center2.cpp +++ b/source/source_lcao/pulay_fs_center2.cpp @@ -2,108 +2,108 @@ template<> // gamma-only, provided xy void PulayForceStress::cal_pulay_fs( - ModuleBase::matrix& force, - ModuleBase::matrix& stress, - const elecstate::DensityMatrix& dm, - const UnitCell& ucell, - const Parallel_Orbitals& pv, - const double* (&dHSx)[3], - const double* (&dHSxy)[6], - const bool& isforce, - const bool& isstress, - Record_adj* ra, - const double& factor_force, - const double& factor_stress) + ModuleBase::matrix& force, + ModuleBase::matrix& stress, + const module_dm::DensityMatrix& dm, + const UnitCell& ucell, + const Parallel_Orbitals& pv, + const double* (&dHSx)[3], + const double* (&dHSxy)[6], + const bool& isforce, + const bool& isstress, + Record_adj* ra, + const double& factor_force, + const double& factor_stress) { - ModuleBase::TITLE("Forces", "cal_pulay_fs"); - ModuleBase::timer::start("Forces", "cal_pulay_fs"); + ModuleBase::TITLE("Forces", "cal_pulay_fs"); + ModuleBase::timer::start("Forces", "cal_pulay_fs"); - const int nspin = PARAM.inp.nspin; - const int nlocal = PARAM.globalv.nlocal; + const int nspin = PARAM.inp.nspin; + const int nlocal = PARAM.globalv.nlocal; - for (int i = 0; i < nlocal; ++i) - { - const int iat = ucell.iwt2iat[i]; - for (int j = 0; j < nlocal; ++j) - { - const int mu = pv.global2local_row(j); - const int nu = pv.global2local_col(i); + for (int i = 0; i < nlocal; ++i) + { + const int iat = ucell.iwt2iat[i]; + for (int j = 0; j < nlocal; ++j) + { + const int mu = pv.global2local_row(j); + const int nu = pv.global2local_col(i); - if (mu >= 0 && nu >= 0) - { - const int index = mu * pv.ncol + nu; - double sum = 0.0; - for (int is = 0; is < nspin; ++is) - { - sum += dm.get_DMK(is + 1, 0, nu, mu); - } - if (isforce) - { - const double sumf = sum * factor_force; - for (int i = 0; i < 3; ++i) - { - force(iat, i) += sumf * 2.0 * dHSx[i][index]; - } - } - if (isstress) - { - const double sums = sum * factor_stress; - int ij = 0; - for (int i = 0; i < 3;++i) - { - for (int j = i; j < 3; ++j) - { - stress(i, j) += sums * dHSxy[ij++][index]; - } - } - } - } - } - } + if (mu >= 0 && nu >= 0) + { + const int index = mu * pv.ncol + nu; + double sum = 0.0; + for (int is = 0; is < nspin; ++is) + { + sum += dm.get_dmk(is + 1, 0, nu, mu); + } + if (isforce) + { + const double sumf = sum * factor_force; + for (int i = 0; i < 3; ++i) + { + force(iat, i) += sumf * 2.0 * dHSx[i][index]; + } + } + if (isstress) + { + const double sums = sum * factor_stress; + int ij = 0; + for (int i = 0; i < 3;++i) + { + for (int j = i; j < 3; ++j) + { + stress(i, j) += sums * dHSxy[ij++][index]; + } + } + } + } + } + } - if (isstress) - { - StressTools::stress_fill(ucell.lat0, ucell.omega, stress); - } + if (isstress) + { + StressTools::stress_fill(ucell.lat0, ucell.omega, stress); + } - ModuleBase::timer::end("Forces", "cal_pulay_fs"); + ModuleBase::timer::end("Forces", "cal_pulay_fs"); } template<> //multi-k, provided xy void PulayForceStress::cal_pulay_fs( - ModuleBase::matrix& force, - ModuleBase::matrix& stress, - const elecstate::DensityMatrix, double>& dm, - const UnitCell& ucell, - const Parallel_Orbitals& pv, - const double* (&dHSx)[3], - const double* (&dHSxy)[6], - const bool& isforce, - const bool& isstress, - Record_adj* ra, - const double& factor_force, - const double& factor_stress) + ModuleBase::matrix& force, + ModuleBase::matrix& stress, + const module_dm::DensityMatrix, double>& dm, + const UnitCell& ucell, + const Parallel_Orbitals& pv, + const double* (&dHSx)[3], + const double* (&dHSxy)[6], + const bool& isforce, + const bool& isstress, + Record_adj* ra, + const double& factor_force, + const double& factor_stress) { - auto stress_func = [](ModuleBase::matrix& local_s, - const double& dm2d1_s, - const double** dHSx, - const double** dHSxy, - const double* dtau, - const int& irr) - { - int ij = 0; - for (int i = 0; i < 3; ++i) - { - for (int j = i; j < 3; ++j) - { - local_s(i, j) += dm2d1_s * dHSxy[ij++][irr]; - } - } - }; - cal_pulay_fs(force, stress, dm, ucell, pv, dHSx, dHSxy, - nullptr, isforce, isstress, ra, - factor_force, factor_stress, stress_func); + auto stress_func = [](ModuleBase::matrix& local_s, + const double& dm2d1_s, + const double** dHSx, + const double** dHSxy, + const double* dtau, + const int& irr) + { + int ij = 0; + for (int i = 0; i < 3; ++i) + { + for (int j = i; j < 3; ++j) + { + local_s(i, j) += dm2d1_s * dHSxy[ij++][irr]; + } + } + }; + cal_pulay_fs(force, stress, dm, ucell, pv, dHSx, dHSxy, + nullptr, isforce, isstress, ra, + factor_force, factor_stress, stress_func); } @@ -111,7 +111,7 @@ template<> // multi-k, provided x void PulayForceStress::cal_pulay_fs( ModuleBase::matrix& force, ModuleBase::matrix& stress, - const elecstate::DensityMatrix, double>& dm, + const module_dm::DensityMatrix, double>& dm, const UnitCell& ucell, const Parallel_Orbitals& pv, const double* (&dHSx)[3], @@ -122,23 +122,23 @@ void PulayForceStress::cal_pulay_fs( const double& factor_force, const double& factor_stress) { - auto stress_func = [](ModuleBase::matrix& local_s, - const double& dm2d1_s, - const double** dHSx, - const double** dHSxy, - const double* dtau, - const int& irr) - { - for (int i = 0; i < 3; ++i) - { - for (int j = i; j < 3; ++j) - { - local_s(i, j) += dm2d1_s * dHSx[i][irr] * dtau[irr * 3 + j]; - } - } - }; - cal_pulay_fs(force, stress, dm, ucell, pv, dHSx, - nullptr, dtau, isforce, isstress, ra, - factor_force, factor_stress, stress_func); + auto stress_func = [](ModuleBase::matrix& local_s, + const double& dm2d1_s, + const double** dHSx, + const double** dHSxy, + const double* dtau, + const int& irr) + { + for (int i = 0; i < 3; ++i) + { + for (int j = i; j < 3; ++j) + { + local_s(i, j) += dm2d1_s * dHSx[i][irr] * dtau[irr * 3 + j]; + } + } + }; + cal_pulay_fs(force, stress, dm, ucell, pv, dHSx, + nullptr, dtau, isforce, isstress, ra, + factor_force, factor_stress, stress_func); } diff --git a/source/source_lcao/pulay_fs_gint.h b/source/source_lcao/pulay_fs_gint.h new file mode 100644 index 00000000000..a2eab535178 --- /dev/null +++ b/source/source_lcao/pulay_fs_gint.h @@ -0,0 +1,47 @@ +#ifndef ABACUS_SOURCE_LCAO_PULAY_FS_GINT_H +#define ABACUS_SOURCE_LCAO_PULAY_FS_GINT_H + +#include "pulay_fs.h" +#include "source_lcao/stress_tools.h" +#include "source_hamilt/module_xc/xc_functional.h" +#include "source_io/module_parameter/parameter.h" +#include "source_hamilt/module_gint/gint_interface.h" +namespace PulayForceStress +{ + template + void cal_pulay_fs( + ModuleBase::matrix& f, ///< [out] force + ModuleBase::matrix& s, ///< [out] stress + const module_dm::DensityMatrix& dm, ///< [in] density matrix + const UnitCell& ucell, ///< [in] unit cell + const elecstate::Potential* pot, ///< [in] potential on grid + const bool& isforce, + const bool& isstress, + const bool& set_dmr_gint) + { + const int nspin = PARAM.inp.nspin; + std::vector vr_eff(nspin, nullptr); + std::vector vofk_eff(nspin, nullptr); + if (XC_Functional::get_func_type() == 3 || XC_Functional::get_func_type() == 5) + { + for (int is = 0; is < nspin; ++is) + { + vr_eff[is] = pot->get_eff_v(is); + vofk_eff[is] = pot->get_eff_vofk(is); + } + ModuleGint::cal_gint_fvl_meta(nspin, vr_eff, vofk_eff, dm.get_dmr_vec(), isforce, isstress, &f, &s); + } + else + { + for(int is = 0; is < nspin; ++is) + { + vr_eff[is] = pot->get_eff_v(is); + } + ModuleGint::cal_gint_fvl(nspin, vr_eff, dm.get_dmr_vec(), isforce, isstress, &f, &s); + } + + if (isstress) { StressTools::stress_fill(-1.0, ucell.omega, s); } + } +} + +#endif // ABACUS_SOURCE_LCAO_PULAY_FS_GINT_H diff --git a/source/source_lcao/pulay_fs_gint.hpp b/source/source_lcao/pulay_fs_gint.hpp deleted file mode 100644 index b31a97f0518..00000000000 --- a/source/source_lcao/pulay_fs_gint.hpp +++ /dev/null @@ -1,43 +0,0 @@ -#pragma once -#include "pulay_fs.h" -#include "source_lcao/stress_tools.h" -#include "source_hamilt/module_xc/xc_functional.h" -#include "source_io/module_parameter/parameter.h" -#include "source_hamilt/module_gint/gint_interface.h" -namespace PulayForceStress -{ - template - void cal_pulay_fs( - ModuleBase::matrix& f, ///< [out] force - ModuleBase::matrix& s, ///< [out] stress - const elecstate::DensityMatrix& dm, ///< [in] density matrix - const UnitCell& ucell, ///< [in] unit cell - const elecstate::Potential* pot, ///< [in] potential on grid - const bool& isforce, - const bool& isstress, - const bool& set_dmr_gint) - { - const int nspin = PARAM.inp.nspin; - std::vector vr_eff(nspin, nullptr); - std::vector vofk_eff(nspin, nullptr); - if (XC_Functional::get_func_type() == 3 || XC_Functional::get_func_type() == 5) - { - for (int is = 0; is < nspin; ++is) - { - vr_eff[is] = pot->get_eff_v(is); - vofk_eff[is] = pot->get_eff_vofk(is); - } - ModuleGint::cal_gint_fvl_meta(nspin, vr_eff, vofk_eff, dm.get_DMR_vector(), isforce, isstress, &f, &s); - } - else - { - for(int is = 0; is < nspin; ++is) - { - vr_eff[is] = pot->get_eff_v(is); - } - ModuleGint::cal_gint_fvl(nspin, vr_eff, dm.get_DMR_vector(), isforce, isstress, &f, &s); - } - - if (isstress) { StressTools::stress_fill(-1.0, ucell.omega, s); } - } -} diff --git a/source/source_lcao/pulay_fs_temp.h b/source/source_lcao/pulay_fs_temp.h new file mode 100644 index 00000000000..2cd35142465 --- /dev/null +++ b/source/source_lcao/pulay_fs_temp.h @@ -0,0 +1,138 @@ +#ifndef ABACUS_SOURCE_LCAO_PULAY_FS_TEMP_H +#define ABACUS_SOURCE_LCAO_PULAY_FS_TEMP_H + +#include +#include "pulay_fs.h" +#include "source_base/timer.h" +#include "source_io/module_parameter/parameter.h" +#ifdef _OPENMP +#include +#endif + + +namespace PulayForceStress +{ + // common kernel + template + inline void cal_pulay_fs( + ModuleBase::matrix& f, + ModuleBase::matrix& s, + const module_dm::DensityMatrix& dm, + const UnitCell& ucell, + const Parallel_Orbitals& pv, + const double** dHSx, + const double** dHSxy, + const double* dtau, + const bool& isforce, + const bool& isstress, + Record_adj* ra, + const double& factor_force, + const double& factor_stress, + Tfunc& stress_func) + { + ModuleBase::TITLE("Force_LCAO", "cal_pulay_fs_center2"); + ModuleBase::timer::start("Force_LCAO", "cal_pulay_fs_center2"); + + const int nspin_DMR = (PARAM.inp.nspin == 2) ? 2 : 1; + int total_irr = 0; +#ifdef _OPENMP +#pragma omp parallel + { + int num_threads = omp_get_num_threads(); + ModuleBase::matrix local_s(3, 3); + int local_total_irr = 0; +#else + ModuleBase::matrix& local_s = s; + int& local_total_irr = total_irr; +#endif + +#ifdef _OPENMP +#pragma omp for schedule(dynamic) +#endif + for (int iat = 0; iat < ucell.nat; iat++) + { + const int T1 = ucell.iat2it[iat]; + Atom* atom1 = &ucell.atoms[T1]; + const int I1 = ucell.iat2ia[iat]; + // get iat1 + int iat1 = ucell.itia2iat(T1, I1); + double* f_iat = nullptr; + if (isforce) { f_iat = &f(iat, 0); } +#ifdef _OPENMP + // using local stack to avoid false sharing in multi-threaded case + double f_tmp[3] = { 0.0, 0.0, 0.0 }; + if (num_threads > 1) { f_iat = f_tmp; } +#endif + int irr = pv.nlocstart[iat]; + const int start1 = ucell.itiaiw2iwt(T1, I1, 0); + for (int cb = 0; cb < ra->na_each[iat]; ++cb) + { + const int T2 = ra->get_info(iat, cb)[3]; + const int I2 = ra->get_info(iat, cb)[4]; + const int start2 = ucell.itiaiw2iwt(T2, I2, 0); + Atom* atom2 = &ucell.atoms[T2]; + // get iat2 + int iat2 = ucell.itia2iat(T2, I2); + double Rx = ra->get_info(iat, cb)[0]; + double Ry = ra->get_info(iat, cb)[1]; + double Rz = ra->get_info(iat, cb)[2]; + // get BaseMatrix + if (pv.is_invalid_atom_pair(iat1, iat2)) { continue; } + std::vector*> tmp_matrix; + for (int is = 0; is < nspin_DMR; ++is) + { + tmp_matrix.push_back(dm.get_dmr_ptr(is + 1)->find_matrix(iat1, iat2, Rx, Ry, Rz)); + } + for (int mu = 0; mu < pv.get_nrow_atom(iat1); ++mu) + { + for (int nu = 0; nu < pv.get_ncol_atom(iat2); ++nu) + { + // the DMR should not be summed over spin, do the summation here + double dm2d1 = 0.0; + for (int is = 0; is < nspin_DMR; ++is) { dm2d1 += tmp_matrix[is]->get_value(mu, nu); } + double dm2d2 = 2.0 * dm2d1; + if (isforce) + { + const double dm2d2_f = dm2d2 * factor_force; + for (int i = 0; i < 3; ++i) { f_iat[i] += dm2d2_f * dHSx[i][irr]; } + } + if (isstress) + { + const double dm2d1_s = dm2d1 * factor_stress; + stress_func(local_s, dm2d1_s, dHSx, dHSxy, dtau, irr); + } + ++local_total_irr; + ++irr; + } + } + } +#ifdef _OPENMP + if (isforce && num_threads > 1) { for (int i = 0; i < 3; ++i) { f(iat, i) += f_iat[i]; } } +#endif + } // end iat +#ifdef _OPENMP +#pragma omp critical(cal_foverlap_k_reduce) + { + total_irr += local_total_irr; + if (isstress) + { + for (int i = 0; i < 3; ++i) { for (int j = i; j < 3; ++j) { s(i, j) += local_s(i, j); } } + } + } + } +#endif + + if (total_irr != pv.nnr) + { + ModuleBase::GlobalFunc::OUT(GlobalV::ofs_running, "wrong irr", total_irr); + ModuleBase::GlobalFunc::OUT(GlobalV::ofs_running, "wrong pv.nnr", pv.nnr); + ModuleBase::WARNING_QUIT("Force_LCAO::cal_pulay_fs_center2", "irr!=pv.nnr"); + } + + if (isstress) { StressTools::stress_fill(ucell.lat0, ucell.omega, s); } + + ModuleBase::timer::end("Force_LCAO", "cal_pulay_fs_center2"); + } +} + +#endif // ABACUS_SOURCE_LCAO_PULAY_FS_TEMP_H diff --git a/source/source_lcao/pulay_fs_temp.hpp b/source/source_lcao/pulay_fs_temp.hpp deleted file mode 100644 index 7d222318f95..00000000000 --- a/source/source_lcao/pulay_fs_temp.hpp +++ /dev/null @@ -1,134 +0,0 @@ -#pragma once -#include -#include "pulay_fs.h" -#include "source_base/timer.h" -#include "source_io/module_parameter/parameter.h" -#ifdef _OPENMP -#include -#endif - - -namespace PulayForceStress -{ - // common kernel - template - inline void cal_pulay_fs( - ModuleBase::matrix& f, - ModuleBase::matrix& s, - const elecstate::DensityMatrix& dm, - const UnitCell& ucell, - const Parallel_Orbitals& pv, - const double** dHSx, - const double** dHSxy, - const double* dtau, - const bool& isforce, - const bool& isstress, - Record_adj* ra, - const double& factor_force, - const double& factor_stress, - Tfunc& stress_func) - { - ModuleBase::TITLE("Force_LCAO", "cal_pulay_fs_center2"); - ModuleBase::timer::start("Force_LCAO", "cal_pulay_fs_center2"); - - const int nspin_DMR = (PARAM.inp.nspin == 2) ? 2 : 1; - int total_irr = 0; -#ifdef _OPENMP -#pragma omp parallel - { - int num_threads = omp_get_num_threads(); - ModuleBase::matrix local_s(3, 3); - int local_total_irr = 0; -#else - ModuleBase::matrix& local_s = s; - int& local_total_irr = total_irr; -#endif - -#ifdef _OPENMP -#pragma omp for schedule(dynamic) -#endif - for (int iat = 0; iat < ucell.nat; iat++) - { - const int T1 = ucell.iat2it[iat]; - Atom* atom1 = &ucell.atoms[T1]; - const int I1 = ucell.iat2ia[iat]; - // get iat1 - int iat1 = ucell.itia2iat(T1, I1); - double* f_iat = nullptr; - if (isforce) { f_iat = &f(iat, 0); } -#ifdef _OPENMP - // using local stack to avoid false sharing in multi-threaded case - double f_tmp[3] = { 0.0, 0.0, 0.0 }; - if (num_threads > 1) { f_iat = f_tmp; } -#endif - int irr = pv.nlocstart[iat]; - const int start1 = ucell.itiaiw2iwt(T1, I1, 0); - for (int cb = 0; cb < ra->na_each[iat]; ++cb) - { - const int T2 = ra->info[iat][cb][3]; - const int I2 = ra->info[iat][cb][4]; - const int start2 = ucell.itiaiw2iwt(T2, I2, 0); - Atom* atom2 = &ucell.atoms[T2]; - // get iat2 - int iat2 = ucell.itia2iat(T2, I2); - double Rx = ra->info[iat][cb][0]; - double Ry = ra->info[iat][cb][1]; - double Rz = ra->info[iat][cb][2]; - // get BaseMatrix - if (pv.is_invalid_atom_pair(iat1, iat2)) { continue; } - std::vector*> tmp_matrix; - for (int is = 0; is < nspin_DMR; ++is) - { - tmp_matrix.push_back(dm.get_DMR_pointer(is + 1)->find_matrix(iat1, iat2, Rx, Ry, Rz)); - } - for (int mu = 0; mu < pv.get_nrow_atom(iat1); ++mu) - { - for (int nu = 0; nu < pv.get_ncol_atom(iat2); ++nu) - { - // the DMR should not be summed over spin, do the summation here - double dm2d1 = 0.0; - for (int is = 0; is < nspin_DMR; ++is) { dm2d1 += tmp_matrix[is]->get_value(mu, nu); } - double dm2d2 = 2.0 * dm2d1; - if (isforce) - { - const double dm2d2_f = dm2d2 * factor_force; - for (int i = 0; i < 3; ++i) { f_iat[i] += dm2d2_f * dHSx[i][irr]; } - } - if (isstress) - { - const double dm2d1_s = dm2d1 * factor_stress; - stress_func(local_s, dm2d1_s, dHSx, dHSxy, dtau, irr); - } - ++local_total_irr; - ++irr; - } - } - } -#ifdef _OPENMP - if (isforce && num_threads > 1) { for (int i = 0; i < 3; ++i) { f(iat, i) += f_iat[i]; } } -#endif - } // end iat -#ifdef _OPENMP -#pragma omp critical(cal_foverlap_k_reduce) - { - total_irr += local_total_irr; - if (isstress) - { - for (int i = 0; i < 3; ++i) { for (int j = i; j < 3; ++j) { s(i, j) += local_s(i, j); } } - } - } - } -#endif - - if (total_irr != pv.nnr) - { - ModuleBase::GlobalFunc::OUT(GlobalV::ofs_running, "wrong irr", total_irr); - ModuleBase::GlobalFunc::OUT(GlobalV::ofs_running, "wrong pv.nnr", pv.nnr); - ModuleBase::WARNING_QUIT("Force_LCAO::cal_pulay_fs_center2", "irr!=pv.nnr"); - } - - if (isstress) { StressTools::stress_fill(ucell.lat0, ucell.omega, s); } - - ModuleBase::timer::end("Force_LCAO", "cal_pulay_fs_center2"); - } -} diff --git a/source/source_lcao/record_adj.cpp b/source/source_lcao/record_adj.cpp deleted file mode 100644 index e4722e8cdc3..00000000000 --- a/source/source_lcao/record_adj.cpp +++ /dev/null @@ -1,281 +0,0 @@ -#include "record_adj.h" -#include "source_base/timer.h" -#include "source_cell/module_neighbor/sltk_grid_driver.h" -#include "source_io/module_parameter/parameter.h" - -Record_adj::Record_adj() -{ -} -Record_adj::~Record_adj() -{ - if (info_modified) - { - this->delete_grid(); - } -} - -void Record_adj::delete_grid() -{ - for (int i = 0; i < na_proc; i++) - { - // how many 'numerical orbital' adjacents - // for each atom in this process. - for (int j = 0; j < na_each[i]; j++) - { - delete[] info[i][j]; - } - delete[] info[i]; - } - delete[] info; - delete[] na_each; - if (iat2ca) - { - delete[] iat2ca; - } - info_modified = false; -} - -//-------------------------------------------- -// This will record the orbitals according to -// HPSEPS's 2D block division. -// If multi-k, calculate nnr at the same time. -// be called only once in an ion-step. -//-------------------------------------------- -void Record_adj::for_2d(const UnitCell& ucell, - const Grid_Driver& grid_d, - Parallel_Orbitals& pv, - bool gamma_only, - const std::vector& orb_cutoff) -{ - ModuleBase::TITLE("Record_adj", "for_2d"); - ModuleBase::timer::start("Record_adj", "for_2d"); - - assert(ucell.nat > 0); - if (!gamma_only) - { - // Record_adj should not modify members of pv, need refactor! mohan add 2025-03-10 - pv.nlocdim.assign(ucell.nat, 0); - pv.nlocstart.assign(ucell.nat, 0); - pv.nnr = 0; - } - { - // (1) find the adjacent atoms of atom[T1,I1]; - ModuleBase::Vector3 tau1, tau2, dtau; - ModuleBase::Vector3 dtau1, dtau2, tau0; - - this->na_proc = ucell.nat; - - // number of adjacents for each atom. - this->na_each = new int[na_proc]; - ModuleBase::GlobalFunc::ZEROS(na_each, na_proc); - int iat = 0; - - for (int T1 = 0; T1 < ucell.ntype; ++T1) - { - Atom* atom1 = &ucell.atoms[T1]; - for (int I1 = 0; I1 < atom1->na; ++I1) - { - tau1 = atom1->tau[I1]; - // grid_d.Find_atom( tau1 ); - grid_d.Find_atom(ucell, tau1, T1, I1); - const int start1 = ucell.itiaiw2iwt(T1, I1, 0); - if (!gamma_only) - { - pv.nlocstart[iat] = pv.nnr; - } - - // (2) search among all adjacent atoms. - for (int ad = 0; ad < grid_d.getAdjacentNum() + 1; ++ad) - { - const int T2 = grid_d.getType(ad); - const int I2 = grid_d.getNatom(ad); - const int start2 = ucell.itiaiw2iwt(T2, I2, 0); - tau2 = grid_d.getAdjacentTau(ad); - dtau = tau2 - tau1; - double distance = dtau.norm() * ucell.lat0; - double rcut = orb_cutoff[T1] + orb_cutoff[T2]; - - bool is_adj = false; - if (distance < rcut) - { - is_adj = true; - // there is another possibility that i and j are adjacent atoms. - // which is that are adjacents while are also - // adjacents, these considerations are only considered in k-point - // algorithm, - } - else if (distance >= rcut) - { - for (int ad0 = 0; ad0 < grid_d.getAdjacentNum() + 1; ++ad0) - { - const int T0 = grid_d.getType(ad0); - // const int I0 = grid_d.getNatom(ad0); - // const int iat0 = ucell.itia2iat(T0, I0); - // const int start0 = ucell.itiaiw2iwt(T0, I0, 0); - - tau0 = grid_d.getAdjacentTau(ad0); - dtau1 = tau0 - tau1; - double distance1 = dtau1.norm() * ucell.lat0; - double rcut1 = orb_cutoff[T1] + ucell.infoNL->get_rcut_max(T0); - - dtau2 = tau0 - tau2; - double distance2 = dtau2.norm() * ucell.lat0; - double rcut2 = orb_cutoff[T2] + ucell.infoNL->get_rcut_max(T0); - - if (distance1 < rcut1 && distance2 < rcut2) - { - is_adj = true; - break; - } // dis1, dis2 - } - } - - if (is_adj) - { - ++na_each[iat]; - if (!gamma_only) - { - for (int ii = 0; ii < atom1->nw * PARAM.globalv.npol; ++ii) - { - // the index of orbitals in this processor - const int iw1_all = start1 + ii; - const int mu = pv.global2local_row(iw1_all); - if (mu < 0) - { - continue; - } - - for (int jj = 0; jj < ucell.atoms[T2].nw * PARAM.globalv.npol; ++jj) - { - const int iw2_all = start2 + jj; - const int nu = pv.global2local_col(iw2_all); - if (nu < 0) - { - continue; - } - - pv.nlocdim[iat]++; - ++(pv.nnr); - } - } - } - } // end is_adj - } // end ad - ++iat; - } // end I1 - } // end T1 - } - // xiaohui add "OUT_LEVEL", 2015-09-16 - if (PARAM.inp.out_level != "m" && !gamma_only) - { - ModuleBase::GlobalFunc::OUT(GlobalV::ofs_running, "ParaV.nnr", pv.nnr); - } - - //------------------------------------------------ - // info will identify each atom in each unitcell. - //------------------------------------------------ - this->info = new int**[na_proc]; -#ifdef _OPENMP -#pragma omp parallel - { -#endif - - ModuleBase::Vector3 tau1, tau2, dtau; - ModuleBase::Vector3 dtau1, dtau2, tau0; - -#ifdef _OPENMP -#pragma omp for schedule(dynamic) -#endif - for (int i = 0; i < na_proc; i++) - { - // GlobalV::ofs_running << " atom" << std::setw(5) << i << std::setw(10) << na_each[i] << std::endl; - if (na_each[i] > 0) - { - info[i] = new int*[na_each[i]]; - for (int j = 0; j < na_each[i]; j++) - { - // (Rx, Ry, Rz, T, I) - info[i][j] = new int[5]; - ModuleBase::GlobalFunc::ZEROS(info[i][j], 5); - } - } - } - -#ifdef _OPENMP -#pragma omp for schedule(dynamic) -#endif - for (int iat = 0; iat < ucell.nat; ++iat) - { - const int T1 = ucell.iat2it[iat]; - Atom* atom1 = &ucell.atoms[T1]; - const int I1 = ucell.iat2ia[iat]; - { - tau1 = atom1->tau[I1]; - // grid_d.Find_atom( tau1 ); - AdjacentAtomInfo adjs; - grid_d.Find_atom(ucell, tau1, T1, I1, &adjs); - - // (2) search among all adjacent atoms. - int cb = 0; - for (int ad = 0; ad < adjs.adj_num + 1; ++ad) - { - const int T2 = adjs.ntype[ad]; - const int I2 = adjs.natom[ad]; - tau2 = adjs.adjacent_tau[ad]; - dtau = tau2 - tau1; - double distance = dtau.norm() * ucell.lat0; - double rcut = orb_cutoff[T1] + orb_cutoff[T2]; - - bool is_adj = false; - if (distance < rcut) - { - is_adj = true; - } - else if (distance >= rcut) - { - for (int ad0 = 0; ad0 < adjs.adj_num + 1; ++ad0) - { - const int T0 = adjs.ntype[ad0]; - // const int I0 = grid_d.getNatom(ad0); - // const int iat0 = ucell.itia2iat(T0, I0); - // const int start0 = ucell.itiaiw2iwt(T0, I0, 0); - - tau0 = adjs.adjacent_tau[ad0]; - dtau1 = tau0 - tau1; - double distance1 = dtau1.norm() * ucell.lat0; - double rcut1 = orb_cutoff[T1] + ucell.infoNL->get_rcut_max(T0); - - dtau2 = tau0 - tau2; - double distance2 = dtau2.norm() * ucell.lat0; - double rcut2 = orb_cutoff[T2] + ucell.infoNL->get_rcut_max(T0); - - if (distance1 < rcut1 && distance2 < rcut2) - { - is_adj = true; - break; - } // dis1, dis2 - } - } - - if (is_adj) - { - info[iat][cb][0] = adjs.box[ad].x; - info[iat][cb][1] = adjs.box[ad].y; - info[iat][cb][2] = adjs.box[ad].z; - info[iat][cb][3] = T2; - info[iat][cb][4] = I2; - ++cb; - } - } // end ad - // GlobalV::ofs_running << " nadj = " << cb << std::endl; - } // end I1 - } // end T1 -#ifdef _OPENMP - } -#endif - ModuleBase::timer::end("Record_adj", "for_2d"); - info_modified = true; - return; -} - - diff --git a/source/source_lcao/record_adj.h b/source/source_lcao/record_adj.h deleted file mode 100644 index 871403ca140..00000000000 --- a/source/source_lcao/record_adj.h +++ /dev/null @@ -1,59 +0,0 @@ -#ifndef RECORD_ADJ_H -#define RECORD_ADJ_H - -#include "source_basis/module_ao/parallel_orbitals.h" -#include "source_cell/unitcell.h" -#include "source_cell/module_neighbor/sltk_grid_driver.h" - -//--------------------------------------------------- -// FUNCTION: record the adjacent atoms for each atom -//--------------------------------------------------- -class Record_adj -{ - private: - bool info_modified = false; - - public: - Record_adj(); - ~Record_adj(); - - //-------------------------------------------- - // This will record the orbitals according to - // HPSEPS's 2D block division. - //-------------------------------------------- - void for_2d(const UnitCell& ucell, - const Grid_Driver& grid_d, - Parallel_Orbitals& pv, - bool gamma_only, - const std::vector& orb_cutoff); - - - void delete_grid(); - - int na_proc=0; - int* na_each=nullptr; - - //-------------------------------------------- - // record sparse atom index in for_grid(); - // Map iat(dense atom index) to sparse atom index - // Mainly removing the index dependency for OpenMP parallel loop - // - // Meaning: - // 1. if iat2ca[iat] > 0, it contains the sparse atom index - // 2. if iat2ca[iat] < 0, the sparse atom index of iat does not exist - // - // Usage: - // 1. iat2ca[iat] > 0 ? na_each[iat2ca[iat]] : 0 - // 2. iat2ca[iat] > 0 ? info[iat2ca[iat]] : nullptr - //-------------------------------------------- - int* iat2ca=nullptr; - - //------------------------------------------------ - // info will identify each atom in each unitcell. - //------------------------------------------------ - int*** info=nullptr; - - private: -}; - -#endif diff --git a/source/source_lcao/rho_tau_lcao.cpp b/source/source_lcao/rho_tau_lcao.cpp index 6e42883fc7b..a448165abb5 100644 --- a/source/source_lcao/rho_tau_lcao.cpp +++ b/source/source_lcao/rho_tau_lcao.cpp @@ -5,7 +5,9 @@ void LCAO_domain::dm2rho(std::vector*> &dmr, const int nspin, Charge* chr, - bool skip_normalize) + const double nelec, + const double omega, + const bool skip_normalize) { ModuleBase::TITLE("LCAO_domain", "dm2rho"); ModuleBase::timer::start("LCAO_domain", "dm2rho"); @@ -17,13 +19,16 @@ void LCAO_domain::dm2rho(std::vector*> &dmr, ModuleGint::cal_gint_rho(dmr, nspin, chr->rho); - if(!skip_normalize)chr->renormalize_rho(); + // omega must be ucell.omega (current cell volume), NOT rhopw->omega which + // is stale in variable-cell calculations (pw_rho/pw_rhod are not rebuilt + // on cell change). See Charge::renormalize_rho for details. + if(!skip_normalize)chr->renormalize_rho(nelec, omega); // should be moved somewhere else, mohan 20251024 - if (XC_Functional::get_ked_flag()) - { - dm2tau(dmr, nspin, chr); - } + if (XC_Functional::get_ked_flag()) + { + dm2tau(dmr, nspin, chr); + } // symmetrize of charge density should be here, mohan 20251023 @@ -39,11 +44,11 @@ void LCAO_domain::dm2tau(std::vector*> &dmr, ModuleBase::TITLE("LCAO_domain", "dm2tau"); ModuleBase::timer::start("LCAO_domain", "dm2tau"); - for (int is = 0; is < nspin; is++) - { - ModuleBase::GlobalFunc::ZEROS(chr->kin_r[is], chr->nrxx); - } - ModuleGint::cal_gint_tau(dmr, nspin, chr->kin_r); + for (int is = 0; is < nspin; is++) + { + ModuleBase::GlobalFunc::ZEROS(chr->kin_r[is], chr->nrxx); + } + ModuleGint::cal_gint_tau(dmr, nspin, chr->kin_r); ModuleBase::timer::end("LCAO_domain", "dm2tau"); } diff --git a/source/source_lcao/rho_tau_lcao.h b/source/source_lcao/rho_tau_lcao.h index 5a3469b857e..174e8ed9784 100644 --- a/source/source_lcao/rho_tau_lcao.h +++ b/source/source_lcao/rho_tau_lcao.h @@ -7,14 +7,16 @@ // generate charge density from different basis or methods namespace LCAO_domain { - void dm2rho(std::vector*> &dmr, - const int nspin, - Charge* chr, - bool skip_normalize = false); + void dm2rho(std::vector*> &dmr, + const int nspin, + Charge* chr, + const double nelec, + const double omega, + const bool skip_normalize); - void dm2tau(std::vector*> &dmr, - const int nspin, - Charge* chr); + void dm2tau(std::vector*> &dmr, + const int nspin, + Charge* chr); } #endif diff --git a/source/source_lcao/setup_deepks.cpp b/source/source_lcao/setup_deepks.cpp index c96d805cf89..9af6e457f4a 100644 --- a/source/source_lcao/setup_deepks.cpp +++ b/source/source_lcao/setup_deepks.cpp @@ -96,16 +96,19 @@ void Setup_DeePKS::delta_e(const UnitCell& ucell, #endif } -template -void Setup_DeePKS::write_forces(const ModuleBase::matrix& fcs, - const ModuleBase::matrix& fvnl_dalpha, - const Input_para& inp) +namespace DeePKS_domain +{ + +void write_forces(const ModuleBase::matrix& fcs, + const ModuleBase::matrix& fvnl_dalpha, + const std::string& dpks_out_type, + const Input_para& inp) { #ifdef __MLALGO // DeePKS force if (inp.deepks_out_labels) // not parallelized yet { - if (inp.deepks_out_base == "none" || (inp.deepks_out_base != "none" && this->dpks_out_type == "tot")) + if (inp.deepks_out_base == "none" || (inp.deepks_out_base != "none" && dpks_out_type == "tot")) { const std::string file_ftot = PARAM.globalv.global_out_dir + (inp.deepks_out_labels == 1 ? "deepks_ftot.npy" : "deepks_force.npy"); @@ -139,18 +142,18 @@ void Setup_DeePKS::write_forces(const ModuleBase::matrix& fcs, #endif } -template -void Setup_DeePKS::write_stress(const ModuleBase::matrix& scs, - const ModuleBase::matrix& svnl_dalpha, - const double& omega, - const Input_para& inp) +void write_stress(const ModuleBase::matrix& scs, + const ModuleBase::matrix& svnl_dalpha, + const double& omega, + const std::string& dpks_out_type, + const Input_para& inp) { #ifdef __MLALGO if (inp.deepks_out_labels == 1) { assert(omega > 0.0); - if (inp.deepks_out_base == "none" || (inp.deepks_out_base != "none" && this->dpks_out_type == "tot")) + if (inp.deepks_out_base == "none" || (inp.deepks_out_base != "none" && dpks_out_type == "tot")) { const std::string file_stot = PARAM.globalv.global_out_dir + "deepks_stot.npy"; LCAO_deepks_io::save_matrix2npy(file_stot, @@ -180,7 +183,7 @@ void Setup_DeePKS::write_stress(const ModuleBase::matrix& scs, // output scs as tot or base in another dir // this base considers changing xc functional to base functional const std::string file_s = PARAM.globalv.global_deepks_label_elec_dir - + (this->dpks_out_type == "tot" ? "stot.npy" : "sbase.npy"); + + (dpks_out_type == "tot" ? "stot.npy" : "sbase.npy"); LCAO_deepks_io::save_matrix2npy(file_s, scs, GlobalV::MY_RANK, @@ -197,5 +200,7 @@ void Setup_DeePKS::write_stress(const ModuleBase::matrix& scs, #endif } +} // namespace DeePKS_domain + template class Setup_DeePKS; template class Setup_DeePKS>; diff --git a/source/source_lcao/setup_deepks.h b/source/source_lcao/setup_deepks.h index 07d6a74fa08..1948aea0443 100644 --- a/source/source_lcao/setup_deepks.h +++ b/source/source_lcao/setup_deepks.h @@ -32,43 +32,53 @@ class Setup_DeePKS std::string dpks_out_type; - void before_runner( - const UnitCell &ucell, // unitcell - const int nks, // k points + void before_runner( + const UnitCell &ucell, // unitcell + const int nks, // k points const LCAO_Orbitals &orb, // orbital info - Parallel_Orbitals &pv, // parallel orbitals - const Input_para &inp); + Parallel_Orbitals &pv, // parallel orbitals + const Input_para &inp); void build_overlap( - const UnitCell &ucell, - const LCAO_Orbitals &orb, - const Parallel_Orbitals &pv, - const Grid_Driver &gd, + const UnitCell &ucell, + const LCAO_Orbitals &orb, + const Parallel_Orbitals &pv, + const Grid_Driver &gd, TwoCenterIntegrator &overlap_orb_alpha, - const Input_para &inp); + const Input_para &inp); void delta_e( - const UnitCell& ucell, + const UnitCell& ucell, const K_Vectors &kv, - const LCAO_Orbitals& orb, - const Parallel_Orbitals &pv, // parallel orbitals - const Grid_Driver &gd, - const std::vector>& dm_vec, + const LCAO_Orbitals& orb, + const Parallel_Orbitals &pv, // parallel orbitals + const Grid_Driver &gd, + const std::vector>& dm_vec, elecstate::fenergy &f_en, - const Input_para &inp); + const Input_para &inp); - void write_forces( - const ModuleBase::matrix &fcs, - const ModuleBase::matrix &fvnl_dalpha, - const Input_para &inp); +}; - void write_stress( - const ModuleBase::matrix &scs, - const ModuleBase::matrix &svnl_dalpha, - const double &omega, - const Input_para &inp); +// Free functions that write the DeePKS force/stress labels to .npy files. +// They are independent of the electronic type TK, so they live outside +// Setup_DeePKS and only need the output-type selector dpks_out_type. +namespace DeePKS_domain +{ -}; +void write_forces( + const ModuleBase::matrix &fcs, + const ModuleBase::matrix &fvnl_dalpha, + const std::string &dpks_out_type, + const Input_para &inp); + +void write_stress( + const ModuleBase::matrix &scs, + const ModuleBase::matrix &svnl_dalpha, + const double &omega, + const std::string &dpks_out_type, + const Input_para &inp); + +} // namespace DeePKS_domain #endif diff --git a/source/source_lcao/setup_dm.cpp b/source/source_lcao/setup_dm.cpp deleted file mode 100644 index 4c8ff2dd458..00000000000 --- a/source/source_lcao/setup_dm.cpp +++ /dev/null @@ -1,25 +0,0 @@ -#include "source_lcao/setup_dm.h" -#include "source_estate/cal_dm.h" -#include "source_base/timer.h" -#include "source_estate/module_dm/cal_dm_psi.h" -#include "source_hamilt/module_xc/xc_functional.h" -#include "source_lcao/module_deltaspin/spin_constrain.h" -#include "source_io/module_parameter/parameter.h" -#include "source_hamilt/module_gint/gint_interface.h" -#include - -namespace LCAO_domain -{ - -// change init_dm to allocate_dm, mohan 2025-10-31 -template -void Setup_DM::allocate_dm(const K_Vectors* kv, const Parallel_Orbitals* pv, const int nspin) -{ - const int nspin_dm = nspin == 2 ? 2 : 1; - this->dm = new elecstate::DensityMatrix(pv, nspin_dm, kv->kvec_d, kv->get_nks() / nspin_dm); -} - -template class Setup_DM; // Gamma_only case -template class Setup_DM>; // multi-k case - -} // namespace elecstate diff --git a/source/source_lcao/setup_dm.h b/source/source_lcao/setup_dm.h deleted file mode 100644 index 672a50c8781..00000000000 --- a/source/source_lcao/setup_dm.h +++ /dev/null @@ -1,39 +0,0 @@ -#ifndef SETUP_DM_H -#define SETUP_DM_H - -#include "source_cell/klist.h" -#include "source_basis/module_ao/parallel_orbitals.h" -#include "source_estate/module_dm/density_matrix.h" - -#include - -namespace LCAO_domain -{ -template -class Setup_DM -{ - public: - - Setup_DM() - { - } // will be called by ElecStateLCAO_TDDFT - - ~Setup_DM() - { - if (this->dm != nullptr) - { - delete this->dm; - } - } - - // allocate density matrix - void allocate_dm(const K_Vectors* kv, const Parallel_Orbitals* pv, const int nspin); - - elecstate::DensityMatrix* dm = nullptr; - -}; - - -} // namespace elecstate - -#endif diff --git a/source/source_lcao/setup_exx.cpp b/source/source_lcao/setup_exx.cpp index 0285c786a9c..c1371d33b96 100644 --- a/source/source_lcao/setup_exx.cpp +++ b/source/source_lcao/setup_exx.cpp @@ -42,12 +42,12 @@ void Exx_NAO::init(const UnitCell& ucell, Exx_Info& exx_info) template void Exx_NAO::before_runner( - UnitCell& ucell, // unitcell - K_Vectors &kv, // k points + UnitCell& ucell, // unitcell + K_Vectors &kv, // k points const LCAO_Orbitals &orb, // orbital info const Parallel_Orbitals &pv, // parallel orbitals - const Input_para& inp, - Exx_Info& exx_info) + const Input_para& inp, + Exx_Info& exx_info) { #ifdef __EXX if (inp.calculation == "scf" || inp.calculation == "relax" || inp.calculation == "cell-relax" @@ -94,13 +94,13 @@ void Exx_NAO::before_runner( template void Exx_NAO::before_scf( - const UnitCell &ucell, // unitcell - const K_Vectors &kv, - const LCAO_Orbitals &orb, // orbital info - Charge_Mixing* p_chgmix, - const int istep, - const Input_para& inp, - Exx_Info& exx_info) + const UnitCell &ucell, // unitcell + const K_Vectors &kv, + const LCAO_Orbitals &orb, // orbital info + Charge_Mixing* p_chgmix, + const int istep, + const Input_para& inp, + Exx_Info& exx_info) { #ifdef __EXX if (inp.calculation != "nscf") @@ -113,11 +113,11 @@ void Exx_NAO::before_scf( { this->exc->exx_beforescf(istep, kv, *p_chgmix, ucell, orb); } - } - else - { - // do nothing - } + } + else + { + // do nothing + } #endif } diff --git a/source/source_lcao/setup_exx.h b/source/source_lcao/setup_exx.h index d91c15b0b91..11f1fb52d96 100644 --- a/source/source_lcao/setup_exx.h +++ b/source/source_lcao/setup_exx.h @@ -6,7 +6,7 @@ #include "source_io/module_parameter/input_parameter.h" // Input_para #include "source_basis/module_ao/parallel_orbitals.h" // parallel orbitals #include "source_basis/module_ao/orb_read.h" // orb -#include "source_estate/module_charge/charge_mixing.h" // use charge mixing +#include "source_estate/module_charge/chg_mix.h" // use charge mixing #include "source_hamilt/module_xc/exx_info.h" // for Exx_Info // for EXX @@ -31,22 +31,22 @@ class Exx_NAO void init(const UnitCell& ucell, Exx_Info& exx_info); - void before_runner( - UnitCell& ucell, // unitcell - K_Vectors &kv, // k points + void before_runner( + UnitCell& ucell, // unitcell + K_Vectors &kv, // k points const LCAO_Orbitals &orb, // orbital info - const Parallel_Orbitals &pv, // parallel orbitals - const Input_para& inp, - Exx_Info& exx_info); - - void before_scf( - const UnitCell &ucell, // unitcell - const K_Vectors &kv, - const LCAO_Orbitals &orb, // orbital info - Charge_Mixing* p_chgmix, - const int istep, - const Input_para& inp, - Exx_Info& exx_info); + const Parallel_Orbitals &pv, // parallel orbitals + const Input_para& inp, + Exx_Info& exx_info); + + void before_scf( + const UnitCell &ucell, // unitcell + const K_Vectors &kv, + const LCAO_Orbitals &orb, // orbital info + Charge_Mixing* p_chgmix, + const int istep, + const Input_para& inp, + Exx_Info& exx_info); }; diff --git a/source/source_lcao/setup_nonlocal.cpp b/source/source_lcao/setup_nonlocal.cpp index 7a93b65308e..936c16516cf 100644 --- a/source/source_lcao/setup_nonlocal.cpp +++ b/source/source_lcao/setup_nonlocal.cpp @@ -1,58 +1,32 @@ #include "setup_nonlocal.h" #include "source_base/parallel_common.h" -#include "source_io/module_parameter/parameter.h" #ifdef __LCAO #include "source_pw/module_pwdft/soc.h" +#include "../source_base/complexmatrix.h" // mohan add 2013-08-02 // In order to get rid of the read in file .NONLOCAL. InfoNonlocal::InfoNonlocal() { - this->Beta = new Numerical_Nonlocal[1]; - this->nproj = nullptr; this->nprojmax = 0; this->rcutmax_Beta = 0.0; } -InfoNonlocal::~InfoNonlocal() -{ - delete[] Beta; - delete[] nproj; -} +InfoNonlocal::~InfoNonlocal() = default; -#include "../source_base/complexmatrix.h" -void InfoNonlocal::Set_NonLocal(const int& it, - Atom* atom, - int& n_projectors, - const int& kmesh, - const double& dk, - const double& dr_uniform, - std::ofstream& log, - const bool& out_element_info, - const bool& lspinorb, - const int& nspin) +void InfoNonlocal::build_soc_coefficients(const Atom* atom, + const int& n_projectors, + ModuleBase::ComplexMatrix& coefficient_D_nc_in) { - ModuleBase::TITLE("InfoNonlocal", "Set_NonLocal"); - - // set a pointer - // Atom* atom = &ucell.atoms[it]; - - // get the number of non-local projectors - n_projectors = atom->ncpp.nbeta; - - const int nh = atom->ncpp.nh; // zhengdy-soc - - // set the nonlocal projector objects - Numerical_Nonlocal_Lm* tmpBeta_lm = new Numerical_Nonlocal_Lm[n_projectors]; - - ModuleBase::ComplexMatrix coefficient_D_nc_in(nh * 2, nh * 2); // zhengdy-soc + const int nh = atom->ncpp.nh; int lmaxkb = -1; for (int ibeta = 0; ibeta < atom->ncpp.nbeta; ibeta++) { lmaxkb = std::max(lmaxkb, atom->ncpp.lll[ibeta]); } + Soc soc; if (atom->ncpp.has_so) { @@ -61,10 +35,8 @@ void InfoNonlocal::Set_NonLocal(const int& it, } int ip1 = 0; - for (int p1 = 0; p1 < n_projectors; p1++) // nbeta + for (int p1 = 0; p1 < n_projectors; p1++) { - const int lnow = atom->ncpp.lll[p1]; - const int l1 = atom->ncpp.lll[p1]; const double j1 = atom->ncpp.jjj[p1]; for (int m1 = 0; m1 < 2 * l1 + 1; m1++) @@ -109,307 +81,103 @@ void InfoNonlocal::Set_NonLocal(const int& it, assert(ip2 == nh); ip1++; } // end m1 - - // only keep the nonzero part. - int cut_mesh = atom->ncpp.mesh; - for (int ir = atom->ncpp.mesh - 1; ir >= 0; --ir) - { - if (std::abs(atom->ncpp.betar(p1, ir)) > 1.0e-10) - { - cut_mesh = ir; - break; - } - } - if (cut_mesh % 2 == 0) - { - ++cut_mesh; - } - - double* beta_r = new double[cut_mesh]; - ModuleBase::GlobalFunc::ZEROS(beta_r, cut_mesh); - for (int ir = 0; ir < cut_mesh; ++ir) - { - beta_r[ir] = atom->ncpp.betar(p1, ir); - } - - tmpBeta_lm[p1].set_NL_proj(atom->label, - it, // type - lnow, // angular momentum L - cut_mesh, // number of radial mesh - atom->ncpp.rab.data(), - atom->ncpp.r.data(), // radial mesh value (a.u.) - beta_r, - kmesh, - dk, - dr_uniform); // delta k mesh in reciprocal space - - if (out_element_info) { - tmpBeta_lm[p1].plot(GlobalV::MY_RANK); - } - - delete[] beta_r; } - - assert(ip1 == nh); - - this->Beta[it].set_type_info(it, - atom->label, - atom->ncpp.pp_type, - atom->ncpp.lmax, - n_projectors, - tmpBeta_lm); // zhengdy-soc 2018-09-10 - - // mohan add 2021-05-07 - atom->ncpp.set_d_so(coefficient_D_nc_in, n_projectors, nh, atom->ncpp.has_so, lspinorb, nspin); - - delete[] tmpBeta_lm; - - log << " SET NONLOCAL PSEUDOPOTENTIAL PROJECTORS FOR ELEMENT " << atom->label << std::endl; - return; } -void InfoNonlocal::Read_NonLocal(const int& it, - Atom* atom, - int& n_projectors, - const int& my_rank, - const int& kmesh, - const double& dk, - const double& dr_uniform, - const std::string& nonlocalFile) +void InfoNonlocal::build_beta_r(const Atom* atom, + const int& p1, + std::vector& beta_r, + int& cut_mesh) { - ModuleBase::TITLE("InfoNonlocal", "Read_NonLocal"); - - std::ifstream ifs; - - // mohan add 2010-09-08. - // check if the non-local pseudopotential file exist. - bool open = false; - if (my_rank == 0) + cut_mesh = atom->ncpp.mesh; + for (int ir = atom->ncpp.mesh - 1; ir >= 0; --ir) { - ifs.open(nonlocalFile.c_str()); - if (ifs) + if (std::abs(atom->ncpp.betar(p1, ir)) > 1.0e-10) { - open = true; + cut_mesh = ir; + break; } } -#ifdef __MPI - Parallel_Common::bcast_bool(open); -#endif - if (!open) - { - std::cout << " Non-local File : " << nonlocalFile << std::endl; - ModuleBase::WARNING_QUIT("InfoNonlocal::Read_NonLocal", "Can not find the NONLOCAL file."); - } - else + if (cut_mesh % 2 == 0) { -// GlobalV::ofs_running << " Open nonlocal pseudopotential file: " << nonlocalFile << std::endl; + ++cut_mesh; } - std::string label; - std::string ps_type; - - // maximal lmax allowed in this calculation - int nlmax = 0; - - if (my_rank == 0) - { - if (ModuleBase::GlobalFunc::SCAN_BEGIN(ifs, "
")) - { - ModuleBase::GlobalFunc::READ_VALUE(ifs, label); - ModuleBase::GlobalFunc::READ_VALUE(ifs, ps_type); - if (ps_type != "NC") - { - ModuleBase::WARNING_QUIT("InfoNonlocal::Read_NonLocal", - "Only available for NC nonlocal pseudopotential"); - } - ModuleBase::GlobalFunc::READ_VALUE(ifs, nlmax); - // std::cout << " " << label << " " << ps_type << " " << nlmax << std::endl; - assert(nlmax >= -1); - ModuleBase::GlobalFunc::SCAN_END(ifs, "
"); - } - } - -#ifdef __MPI - Parallel_Common::bcast_string(label); - Parallel_Common::bcast_string(ps_type); - Parallel_Common::bcast_int(nlmax); -#endif - - // mohan add 2012-06-09 - if (nlmax != -1) + beta_r.resize(cut_mesh, 0.0); + for (int ir = 0; ir < cut_mesh; ++ir) { - bool find_lmax = false; - for (int ic = 0; ic < atom->ncpp.nbeta; ic++) - { - if (nlmax == atom->ncpp.lll[ic]) - { - find_lmax = true; - break; - } - } - - if (!find_lmax) - { - std::cout << " For element " << label << std::endl; - std::cout << " Max L Read in from NONLOCAL = " << nlmax << std::endl; - for (int ib = 0; ib < atom->ncpp.nbeta; ++ib) - { - std::cout << " Max L Read in from pseudopotential file = " << atom->ncpp.lll[ib] << std::endl; - } - ModuleBase::WARNING_QUIT("InfoNonlocal::Read_NonLocal", "nlmax != atom->lll"); - } + beta_r[ir] = atom->ncpp.betar(p1, ir); } +} - ModuleBase::GlobalFunc::OUT(GlobalV::ofs_running, "label", label); - ModuleBase::GlobalFunc::OUT(GlobalV::ofs_running, "nlmax", nlmax); - - //------------------------------------------- - // if each L has projectors more than once, - // this needed to be modified. - //------------------------------------------- - int nproj_allowed = nlmax + 1; - ModuleBase::matrix coefficient_D_in(nproj_allowed, nproj_allowed); - ModuleBase::ComplexMatrix coefficient_D_nc_in(nproj_allowed * 2, nproj_allowed * 2); - - if (my_rank == 0) - { - if (ModuleBase::GlobalFunc::SCAN_BEGIN(ifs, "")) - { - //-------------------------------------- - // this parameter is very important!!! - //-------------------------------------- - ModuleBase::GlobalFunc::READ_VALUE(ifs, n_projectors); - ModuleBase::GlobalFunc::OUT(GlobalV::ofs_running, "n_projectors", n_projectors); - - for (int p1 = 0; p1 < n_projectors; p1++) - { - for (int p2 = 0; p2 < n_projectors; p2++) - { - int L1_read, L2_read; - - ifs >> L1_read >> L2_read; - - assert(L1_read <= nlmax); - assert(L2_read <= nlmax); - - ifs >> coefficient_D_in(L1_read, L2_read); +void InfoNonlocal::Set_NonLocal(const int& it, + Atom* atom, + int& n_projectors, + const int& kmesh, + const double& dk, + const double& dr_uniform, + std::ofstream& log, + const bool& out_element_info, + const bool& lspinorb, + const int& nspin, + const int& my_rank) +{ + ModuleBase::TITLE("InfoNonlocal", "Set_NonLocal"); - } - } - ModuleBase::GlobalFunc::SCAN_END(ifs, ""); - } - } + // get the number of non-local projectors + n_projectors = atom->ncpp.nbeta; + const int nh = atom->ncpp.nh; // zhengdy-soc -#ifdef __MPI - Parallel_Common::bcast_int(n_projectors); // mohan add 2010-12-20 -#endif + // set the nonlocal projector objects + std::vector tmpBeta_lm(n_projectors); + ModuleBase::ComplexMatrix coefficient_D_nc_in(nh * 2, nh * 2); // zhengdy-soc - Numerical_Nonlocal_Lm* tmpBeta_lm = new Numerical_Nonlocal_Lm[n_projectors]; - int* LfromBeta = new int[n_projectors]; - ModuleBase::GlobalFunc::ZEROS(LfromBeta, n_projectors); + build_soc_coefficients(atom, n_projectors, coefficient_D_nc_in); for (int p1 = 0; p1 < n_projectors; p1++) { - int meshr_ps = 0; - if (my_rank == 0) - { - if (ModuleBase::GlobalFunc::SCAN_BEGIN(ifs, "", false)) - { - int iproj = 0; - ModuleBase::GlobalFunc::READ_VALUE(ifs, iproj); - if (iproj != p1) - { - std::cout << " iproj=" << iproj << " p1=" << p1 << std::endl; - ModuleBase::WARNING_QUIT("InfoNonlocal::Read_NonLocal", "Check non-local projector index."); - } - - ModuleBase::GlobalFunc::READ_VALUE(ifs, LfromBeta[p1]); - assert(LfromBeta[p1] >= 0); - assert(LfromBeta[p1] <= nlmax); - - ModuleBase::GlobalFunc::READ_VALUE(ifs, meshr_ps); - if (meshr_ps % 2 == 0) - { - std::cout << " meshr_ps = " << meshr_ps << std::endl; - ModuleBase::WARNING_QUIT("InfoNonlocal::Read_NonLocal", "meshr_ps must be odd!"); - } - } - else - { - ModuleBase::WARNING_QUIT("InfoNonlocal::Read_NonLocal", " doesn't match!"); - } - } // end my_rank==0 - -#ifdef __MPI - Parallel_Common::bcast_int(meshr_ps); - Parallel_Common::bcast_int(LfromBeta[p1]); -#endif - - double* radial_ps = new double[meshr_ps]; - double* rab_ps = new double[meshr_ps]; - double* beta_r = new double[meshr_ps]; - ModuleBase::GlobalFunc::ZEROS(radial_ps, meshr_ps); - ModuleBase::GlobalFunc::ZEROS(rab_ps, meshr_ps); - ModuleBase::GlobalFunc::ZEROS(beta_r, meshr_ps); - - if (my_rank == 0) - { - for (int ir = 0; ir < meshr_ps; ir++) - { - ifs >> radial_ps[ir]; - ifs >> beta_r[ir]; - ifs >> rab_ps[ir]; - } - } - -#ifdef __MPI - Parallel_Common::bcast_double(radial_ps, meshr_ps); - Parallel_Common::bcast_double(beta_r, meshr_ps); - Parallel_Common::bcast_double(rab_ps, meshr_ps); -#endif + const int lnow = atom->ncpp.lll[p1]; + int cut_mesh = 0; + std::vector beta_r; + build_beta_r(atom, p1, beta_r, cut_mesh); - tmpBeta_lm[p1].set_NL_proj(label, - it, // type - LfromBeta[p1], // angular momentum L - meshr_ps, // number of radial mesh - rab_ps, - radial_ps, // radial mesh value(a.u.) - beta_r, + tmpBeta_lm[p1].set_NL_proj(atom->label, + it, // type + lnow, // angular momentum L + cut_mesh, // number of radial mesh + atom->ncpp.rab.data(), + atom->ncpp.r.data(), // radial mesh value (a.u.) + beta_r.data(), kmesh, dk, dr_uniform); // delta k mesh in reciprocal space - if (PARAM.inp.out_element_info) - { - tmpBeta_lm[p1].plot(my_rank); - } - - delete[] radial_ps; - delete[] rab_ps; - delete[] beta_r; - - if (my_rank == 0) + if (out_element_info) { - ModuleBase::GlobalFunc::SCAN_END(ifs, ""); + tmpBeta_lm[p1].plot(my_rank); } - } // end projectors. - - this->Beta[it].set_type_info(it, label, ps_type, nlmax, n_projectors, tmpBeta_lm); + } - ifs.close(); + this->Beta[it].set_type_info(it, + atom->label, + atom->ncpp.pp_type, + atom->ncpp.lmax, + n_projectors, + tmpBeta_lm.data()); // zhengdy-soc 2018-09-10 - delete[] LfromBeta; - delete[] tmpBeta_lm; + // mohan add 2021-05-07 + atom->ncpp.set_d_so(coefficient_D_nc_in, n_projectors, nh, atom->ncpp.has_so, lspinorb, nspin); - return; + log << " SET NONLOCAL PSEUDOPOTENTIAL PROJECTORS FOR ELEMENT " << atom->label << std::endl; } void InfoNonlocal::setupNonlocal(const int& ntype, Atom* atoms, std::ofstream& log, LCAO_Orbitals& orb, const std::string& basis_type, const bool& out_element_info, const bool& lspinorb, - const int& nspin) + const int& nspin, + const int& my_rank) { //>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>> //~~~~~~~~~~~~~~~~~~~~~~ 2 ~~~~~~~~~~~~~~~~~~~~~~~~~ @@ -421,38 +189,19 @@ void InfoNonlocal::setupNonlocal(const int& ntype, Atom* atoms, std::ofstream& l //>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>> if (basis_type == "lcao" || basis_type == "lcao_in_pw") { - delete[] this->Beta; - this->Beta = new Numerical_Nonlocal[ntype]; - - delete[] this->nproj; - this->nproj = new int[ntype]; - ModuleBase::GlobalFunc::ZEROS(this->nproj, ntype); + // Replace the whole vector in one shot: elements are constructed in + // place and the move assignment only swaps the buffer, so no + // Numerical_Nonlocal object is ever copied or moved. + this->Beta = std::vector(ntype); + this->nproj.assign(ntype, 0); this->nprojmax = 0; - // if true: read in the nonlocal file from file. - // if false: get nonlocal information from .upf or .vwr directly - bool readin_nonlocal = false; - for (int it = 0; it < ntype; it++) { Atom* atom = &atoms[it]; - if (readin_nonlocal) - { - this->Read_NonLocal(it, - atom, - this->nproj[it], - GlobalV::MY_RANK, - orb.get_kmesh(), - orb.get_dk(), - orb.get_dr_uniform(), - orb.orbital_file[it]); - } - else - { - this->Set_NonLocal(it, atom, this->nproj[it], orb.get_kmesh(), orb.get_dk(), orb.get_dr_uniform(), log, - out_element_info, lspinorb, nspin); - } + this->Set_NonLocal(it, atom, this->nproj[it], orb.get_kmesh(), orb.get_dk(), orb.get_dr_uniform(), log, + out_element_info, lspinorb, nspin, my_rank); this->nprojmax = std::max(this->nprojmax, this->nproj[it]); // caoyu add 2021-05-24 to reconstruct atom_arrange::set_sr_NL this->rcutmax_Beta = std::max(this->rcutmax_Beta, this->Beta[it].get_rcut_max()); diff --git a/source/source_lcao/setup_nonlocal.h b/source/source_lcao/setup_nonlocal.h index 16959ea38ad..ad4605347ed 100644 --- a/source/source_lcao/setup_nonlocal.h +++ b/source/source_lcao/setup_nonlocal.h @@ -1,55 +1,89 @@ #ifndef INFONONLOCAL_H #define INFONONLOCAL_H +#include +#include +#include + #include "../source_cell/atom_spec.h" #include "../source_basis/module_ao/orb_nonlocal.h" #include "../source_basis/module_ao/orb_read.h" + class InfoNonlocal { - public: - InfoNonlocal(); - ~InfoNonlocal(); - /// - ///NON-LOCAL part for LCAO - /// - Numerical_Nonlocal* Beta = nullptr;/// nonlocal projectors (1-dimension array) - int * nproj = nullptr; //mohan add 2010-12-19 - int nprojmax; // mohan add 2010-03-07 - double rcutmax_Beta; //caoyu add 2021-05-24 - const double& get_rcutmax_Beta(void) const { return rcutmax_Beta; } - /// in order to get rid of the .NONLOCAL file. - void Set_NonLocal( - const int &it, - Atom* atom, - int &n_projectors, - const int& kmesh, - const double& dk, - const double& dr_uniform, - std::ofstream &log, - const bool& out_element_info, - const bool& lspinorb, - const int& nspin); - /// read in the NONLOCAL projector from file. - void Read_NonLocal( - const int &it, - Atom* atom, - int &n_projectors, - const int &my_rank, - const int& kmesh, - const double& dk, - const double& dr_uniform, - const std::string& nonlocalFile); - //workflow to setup nonlocal part for LCAO - void setupNonlocal( - const int& ntype, - Atom* atoms, - std::ofstream &log, - LCAO_Orbitals &orb, - const std::string& basis_type, - const bool& out_element_info, - const bool& lspinorb, - const int& nspin - ); +public: + InfoNonlocal(); + ~InfoNonlocal(); + + /// NON-LOCAL part for LCAO +private: + std::vector Beta; ///< nonlocal projectors (one per atom type) + std::vector nproj; ///< number of projectors per atom type, mohan add 2010-12-19 + int nprojmax; ///< max number of projectors among all types, mohan add 2010-03-07 + double rcutmax_Beta; ///< max cutoff radius among all projectors, caoyu add 2021-05-24 + +public: + + const std::vector& get_Beta() const { return Beta; } + std::vector& get_Beta() { return Beta; } + const Numerical_Nonlocal& get_Beta(const int& it) const { return Beta[it]; } + const Numerical_Nonlocal* get_Beta_data() const { return Beta.data(); } + Numerical_Nonlocal* get_Beta_data() { return Beta.data(); } + // Replaces the whole vector without copying or moving any element, + // since Numerical_Nonlocal is non-copyable (see orb_nonlocal.h). + void resize_Beta(const int& ntype) { Beta = std::vector(ntype); } + + const std::vector& get_nproj() const { return nproj; } + std::vector& get_nproj() { return nproj; } + int get_nproj(const int& it) const { return nproj[it]; } + int& get_nproj_ref(const int& it) { return nproj[it]; } + void assign_nproj(const int& ntype, const int& value) { nproj.assign(ntype, value); } + + const int& get_nprojmax() const { return nprojmax; } + void set_nprojmax(const int& value) { nprojmax = value; } + + const double& get_rcutmax_Beta(void) const { return rcutmax_Beta; } + void set_rcutmax_Beta(const double& value) { rcutmax_Beta = value; } + + /// in order to get rid of the .NONLOCAL file. + void Set_NonLocal( + const int& it, + Atom* atom, + int& n_projectors, + const int& kmesh, + const double& dk, + const double& dr_uniform, + std::ofstream& log, + const bool& out_element_info, + const bool& lspinorb, + const int& nspin, + const int& my_rank); + + /// workflow to setup nonlocal part for LCAO + void setupNonlocal( + const int& ntype, + Atom* atoms, + std::ofstream& log, + LCAO_Orbitals& orb, + const std::string& basis_type, + const bool& out_element_info, + const bool& lspinorb, + const int& nspin, + const int& my_rank); + +private: + /// build SOC coefficient matrix for nonlocal projectors + void build_soc_coefficients( + const Atom* atom, + const int& n_projectors, + ModuleBase::ComplexMatrix& coefficient_D_nc_in); + + /// build radial projector beta_r on truncated mesh + void build_beta_r( + const Atom* atom, + const int& p1, + std::vector& beta_r, + int& cut_mesh); }; -#endif \ No newline at end of file +#endif diff --git a/source/source_lcao/spar_dh.cpp b/source/source_lcao/spar_dh.cpp index c92d97f5b57..e14e6c6ad4d 100644 --- a/source/source_lcao/spar_dh.cpp +++ b/source/source_lcao/spar_dh.cpp @@ -11,24 +11,20 @@ void sparse_format::cal_dS(const UnitCell& ucell, const Grid_Driver& grid, const TwoCenterBundle& two_center_bundle, const LCAO_Orbitals& orb, - const double& sparse_thr) + const double& sparse_thr, + const bool gamma_only_local, + const int nspin, + const int npol) { ModuleBase::TITLE("sparse_format", "cal_dS"); sparse_format::set_R_range(HS_Arrays.all_R_coor, grid); -const int nnr = PARAM.globalv.gamma_only_local ? pv.nloc : pv.nnr; +const int nnr = gamma_only_local ? pv.nloc : pv.nnr; ForceStressArrays fsr_dh; -fsr_dh.DHloc_fixedR_x = new double[nnr]; -fsr_dh.DHloc_fixedR_y = new double[nnr]; -fsr_dh.DHloc_fixedR_z = new double[nnr]; -ModuleBase::GlobalFunc::ZEROS(fsr_dh.DHloc_fixedR_x, nnr); -ModuleBase::GlobalFunc::ZEROS(fsr_dh.DHloc_fixedR_y, nnr); -ModuleBase::GlobalFunc::ZEROS(fsr_dh.DHloc_fixedR_z, nnr); -// the pointers of dS is different from dH, use the dh pointers to reuse the print functions -fsr_dh.DSloc_Rx = fsr_dh.DHloc_fixedR_x; -fsr_dh.DSloc_Ry = fsr_dh.DHloc_fixedR_y; -fsr_dh.DSloc_Rz = fsr_dh.DHloc_fixedR_z; +fsr_dh.DHloc_fixedR_x.resize(nnr, 0.0); +fsr_dh.DHloc_fixedR_y.resize(nnr, 0.0); +fsr_dh.DHloc_fixedR_z.resize(nnr, 0.0); // cal dS= in LCAO const bool cal_deri = true; const bool cal_stress = false; @@ -44,10 +40,7 @@ LCAO_domain::build_ST_new(fsr_dh, nullptr, false); // delete unused parameter lm.Hloc_fixedR -sparse_format::cal_dSTN_R(ucell,pv, HS_Arrays, fsr_dh, grid, orb.cutoffs(), 0, sparse_thr); -delete[] fsr_dh.DHloc_fixedR_x; -delete[] fsr_dh.DHloc_fixedR_y; -delete[] fsr_dh.DHloc_fixedR_z; +sparse_format::cal_dSTN_R(ucell, pv, HS_Arrays, fsr_dh, grid, orb.cutoffs(), 0, sparse_thr, nspin, npol); return; } void sparse_format::cal_dH(const UnitCell& ucell, @@ -58,23 +51,22 @@ void sparse_format::cal_dH(const UnitCell& ucell, const LCAO_Orbitals& orb, const int& current_spin, const double& sparse_thr, - const ModuleBase::matrix& v_eff) + const ModuleBase::matrix& v_eff, + const bool gamma_only_local, + const int nspin, + const int npol) { ModuleBase::TITLE("sparse_format", "cal_dH"); sparse_format::set_R_range(HS_Arrays.all_R_coor, grid); - const int nnr = PARAM.globalv.gamma_only_local ? pv.nloc : pv.nnr; + const int nnr = gamma_only_local ? pv.nloc : pv.nnr; ForceStressArrays fsr_dh; - fsr_dh.DHloc_fixedR_x = new double[nnr]; - fsr_dh.DHloc_fixedR_y = new double[nnr]; - fsr_dh.DHloc_fixedR_z = new double[nnr]; - - ModuleBase::GlobalFunc::ZEROS(fsr_dh.DHloc_fixedR_x, nnr); - ModuleBase::GlobalFunc::ZEROS(fsr_dh.DHloc_fixedR_y, nnr); - ModuleBase::GlobalFunc::ZEROS(fsr_dh.DHloc_fixedR_z, nnr); + fsr_dh.DHloc_fixedR_x.resize(nnr, 0.0); + fsr_dh.DHloc_fixedR_y.resize(nnr, 0.0); + fsr_dh.DHloc_fixedR_z.resize(nnr, 0.0); // cal dT= in LCAO // cal T + VNL(P1) in LCAO basis const bool cal_deri = true; @@ -100,20 +92,16 @@ void sparse_format::cal_dH(const UnitCell& ucell, *(two_center_bundle.overlap_orb_beta), &grid); - sparse_format::cal_dSTN_R(ucell,pv, HS_Arrays, fsr_dh, grid, orb.cutoffs(), current_spin, sparse_thr); - - delete[] fsr_dh.DHloc_fixedR_x; - delete[] fsr_dh.DHloc_fixedR_y; - delete[] fsr_dh.DHloc_fixedR_z; + sparse_format::cal_dSTN_R(ucell, pv, HS_Arrays, fsr_dh, grid, orb.cutoffs(), current_spin, sparse_thr, nspin, npol); - if(PARAM.inp.nspin==2) + if(nspin==2) { const double* vr_eff1 = v_eff.nc * v_eff.nr > 0 ? &(v_eff(current_spin, 0)) : nullptr; - if (!PARAM.globalv.gamma_only_local) + if (!gamma_only_local) { ModuleGint::cal_dvlocal_R_sparse( - PARAM.inp.nspin, PARAM.globalv.npol, current_spin, PARAM.globalv.nlocal, + nspin, npol, current_spin, PARAM.globalv.nlocal, sparse_thr, vr_eff1, pv, ucell, grid, HS_Arrays); } } @@ -152,7 +140,9 @@ void sparse_format::cal_dSTN_R(const UnitCell& ucell, const Grid_Driver& grid, const std::vector& orb_cutoff, const int& current_spin, - const double& sparse_thr) + const double& sparse_thr, + const int nspin, + const int npol) { ModuleBase::TITLE("sparse_format", "cal_dSTN_R"); @@ -220,7 +210,7 @@ void sparse_format::cal_dSTN_R(const UnitCell& ucell, Abfs::Vector3_Order dR(grid.getBox(ad).x, grid.getBox(ad).y, grid.getBox(ad).z); - for (int ii = 0; ii < atom1->nw * PARAM.globalv.npol; ii++) + for (int ii = 0; ii < atom1->nw * npol; ii++) { const int iw1_all = start + ii; const int mu = pv.global2local_row(iw1_all); @@ -230,7 +220,7 @@ void sparse_format::cal_dSTN_R(const UnitCell& ucell, continue; } - for (int jj = 0; jj < atom2->nw * PARAM.globalv.npol; jj++) + for (int jj = 0; jj < atom2->nw * npol; jj++) { int iw2_all = start2 + jj; const int nu = pv.global2local_col(iw2_all); @@ -240,7 +230,7 @@ void sparse_format::cal_dSTN_R(const UnitCell& ucell, continue; } - if (PARAM.inp.nspin != 4) + if (nspin != 4) { temp_value_double = fsr.DHloc_fixedR_x[index]; if (std::abs(temp_value_double) > sparse_thr) @@ -273,11 +263,11 @@ void sparse_format::cal_dSTN_R(const UnitCell& ucell, return; } -void sparse_format::destroy_dH_R_sparse(LCAO_HS_Arrays& HS_Arrays) +void sparse_format::destroy_dH_R_sparse(LCAO_HS_Arrays& HS_Arrays, const int nspin) { ModuleBase::TITLE("LCAO_domain", "destroy_dH_R_sparse"); - if (PARAM.inp.nspin != 4) + if (nspin != 4) { std::map, std::map>> empty_dHRx_sparse_up; std::map, std::map>> empty_dHRx_sparse_down; diff --git a/source/source_lcao/spar_dh.h b/source/source_lcao/spar_dh.h index 25ee2d42cba..bb59e863427 100644 --- a/source/source_lcao/spar_dh.h +++ b/source/source_lcao/spar_dh.h @@ -6,7 +6,7 @@ #include "source_basis/module_nao/two_center_bundle.h" #include "source_basis/module_ao/orb_read.h" #include "source_cell/module_neighbor/sltk_grid_driver.h" -#include "source_lcao/lcao_hs_arrays.hpp" +#include "source_lcao/lcao_hs_arrays.h" #include "source_lcao/force_stress_arrays.h" #include @@ -20,7 +20,10 @@ void cal_dH(const UnitCell& ucell, const LCAO_Orbitals& orb, const int& current_spin, const double& sparse_thr, - const ModuleBase::matrix& v_eff); + const ModuleBase::matrix& v_eff, + const bool gamma_only_local, + const int nspin, + const int npol); // calculated the derivative of the overlap matrix: void cal_dS(const UnitCell& ucell, @@ -29,7 +32,10 @@ void cal_dS(const UnitCell& ucell, const Grid_Driver& grid, const TwoCenterBundle& two_center_bundle, const LCAO_Orbitals& orb, - const double& sparse_thr); + const double& sparse_thr, + const bool gamma_only_local, + const int nspin, + const int npol); // be called by 'cal_dH_sparse' void set_R_range(std::set>& all_R_coor, const Grid_Driver& grid); @@ -42,9 +48,11 @@ void cal_dSTN_R(const UnitCell& ucell, const Grid_Driver& grid, const std::vector& orb_cutoff, const int& current_spin, - const double& sparse_thr); + const double& sparse_thr, + const int nspin, + const int npol); -void destroy_dH_R_sparse(LCAO_HS_Arrays& HS_Arrays); +void destroy_dH_R_sparse(LCAO_HS_Arrays& HS_Arrays, const int nspin); } // namespace sparse_format diff --git a/source/source_lcao/spar_exx.cpp b/source/source_lcao/spar_exx.cpp index 376d2731b82..57c38920507 100644 --- a/source/source_lcao/spar_exx.cpp +++ b/source/source_lcao/spar_exx.cpp @@ -16,7 +16,7 @@ #include "source_hamilt/module_xc/exx_info.h" #include "source_io/module_parameter/parameter.h" #include "source_lcao/module_ri/ri_2d_comm.h" -#include "source_lcao/module_ri/ri_util.hpp" +#include "source_lcao/module_ri/ri_util.h" // -------------------------------------------------------- // Implementation of the cal_HR_exx function diff --git a/source/source_lcao/spar_exx.h b/source/source_lcao/spar_exx.h index 9a3645fd937..99fe20aa569 100644 --- a/source/source_lcao/spar_exx.h +++ b/source/source_lcao/spar_exx.h @@ -14,7 +14,7 @@ #include "source_basis/module_ao/parallel_orbitals.h" #include "source_cell/unitcell.h" -#include "source_lcao/lcao_hs_arrays.hpp" +#include "source_lcao/lcao_hs_arrays.h" // -------------------------------------------------------- // Namespace - merged into one block diff --git a/source/source_lcao/spar_hsr.h b/source/source_lcao/spar_hsr.h index 281025eba24..8a545826f69 100644 --- a/source/source_lcao/spar_hsr.h +++ b/source/source_lcao/spar_hsr.h @@ -1,7 +1,7 @@ #ifndef SPARSE_FORMAT_HSR_H #define SPARSE_FORMAT_HSR_H -#include "source_lcao/lcao_hs_arrays.hpp" +#include "source_lcao/lcao_hs_arrays.h" #include "source_hamilt/module_hcontainer/hcontainer.h" #include "source_hamilt/hamilt.h" diff --git a/source/source_lcao/spar_st.h b/source/source_lcao/spar_st.h index c0c20019f66..28db2a929fe 100644 --- a/source/source_lcao/spar_st.h +++ b/source/source_lcao/spar_st.h @@ -1,7 +1,7 @@ #ifndef SPARSE_FORMAT_ST_H #define SPARSE_FORMAT_ST_H -#include "source_lcao/lcao_hs_arrays.hpp" +#include "source_lcao/lcao_hs_arrays.h" #include "source_lcao/hamilt_lcao.h" namespace sparse_format diff --git a/source/source_lcao/stress_tools.h b/source/source_lcao/stress_tools.h index 305870e3d74..485cbfb8f45 100644 --- a/source/source_lcao/stress_tools.h +++ b/source/source_lcao/stress_tools.h @@ -1,8 +1,12 @@ -#pragma once +#ifndef ABACUS_SOURCE_LCAO_STRESS_TOOLS_H +#define ABACUS_SOURCE_LCAO_STRESS_TOOLS_H + #include "source_base/matrix.h" // this namespace used to store global function for some stress operation namespace StressTools { // set upper matrix to whole matrix void stress_fill(const double& lat0_, const double& omega_, ModuleBase::matrix& stress_matrix); -} // namespace StressTools \ No newline at end of file +} // namespace StressTools + +#endif // ABACUS_SOURCE_LCAO_STRESS_TOOLS_H diff --git a/source/source_lcao/test/CMakeLists.txt b/source/source_lcao/test/CMakeLists.txt index 8f5d6a1aee6..4c993f3023b 100644 --- a/source/source_lcao/test/CMakeLists.txt +++ b/source/source_lcao/test/CMakeLists.txt @@ -8,7 +8,10 @@ AddTest( LIBS parameter base device symmetry SOURCES test_init_dm_from_file.cpp tmp_mocks.cpp ${ABACUS_SOURCE_DIR}/source_estate/module_dm/density_matrix.cpp - ${ABACUS_SOURCE_DIR}/source_estate/module_dm/density_matrix_io.cpp + ${ABACUS_SOURCE_DIR}/source_estate/module_dm/dmr_gamma.cpp + ${ABACUS_SOURCE_DIR}/source_estate/module_dm/dmr_init.cpp + ${ABACUS_SOURCE_DIR}/source_estate/module_dm/dm_setter.cpp + ${ABACUS_SOURCE_DIR}/source_estate/module_dm/dm_getter.cpp ${ABACUS_SOURCE_DIR}/source_hamilt/module_hcontainer/base_matrix.cpp ${ABACUS_SOURCE_DIR}/source_hamilt/module_hcontainer/hcontainer.cpp ${ABACUS_SOURCE_DIR}/source_hamilt/module_hcontainer/atom_pair.cpp diff --git a/source/source_lcao/test/test_init_dm_from_file.cpp b/source/source_lcao/test/test_init_dm_from_file.cpp index c0edf367737..fb2ab91d67d 100644 --- a/source/source_lcao/test/test_init_dm_from_file.cpp +++ b/source/source_lcao/test/test_init_dm_from_file.cpp @@ -5,13 +5,11 @@ #include "gmock/gmock.h" #include "gtest/gtest.h" -#define private public #include "source_estate/module_dm/density_matrix.h" #include "source_hamilt/module_hcontainer/hcontainer.h" #include "source_hamilt/module_hcontainer/read_hcontainer.h" -#include "source_lcao/setup_dm.h" +#include "source_lcao/allocate_dm.h" #include "source_cell/klist.h" -#undef private #include "source_io/module_dm/write_dmr.h" /************************************************ @@ -113,7 +111,7 @@ class InitDMFileTest : public testing::Test } /// Create DensityMatrix with given nspin and initialize DMR from an HContainer template - elecstate::DensityMatrix* create_dm(int nspin) + module_dm::DensityMatrix* create_dm(int nspin) { K_Vectors kv; int nks = (nspin == 2) ? 2 : 1; @@ -121,7 +119,7 @@ class InitDMFileTest : public testing::Test kv.kvec_d.resize(kv.get_nks()); int nspin_dm = (nspin == 2) ? 2 : 1; - auto* dm = new elecstate::DensityMatrix( + auto* dm = new module_dm::DensityMatrix( paraV, nspin_dm, kv.kvec_d, kv.get_nks() / nspin_dm); // Create a template HContainer and init DMR from it @@ -135,7 +133,7 @@ class InitDMFileTest : public testing::Test } } tmp_HR.allocate(nullptr, true); - dm->init_DMR(tmp_HR); + dm->init_dmr(tmp_HR); return dm; } }; @@ -146,9 +144,9 @@ TEST_F(InitDMFileTest, Nspin1_ReadSingleFile) write_test_csr("./test_dm_dir/dmrs1_nao.csr", 1.0, 0, 1); auto* dm = create_dm(1); - ASSERT_EQ(dm->_DMR.size(), 1); + ASSERT_EQ(dm->get_dmr_vec().size(), 1); - hamilt::HContainer* dmr0 = dm->get_DMR_vector()[0]; + hamilt::HContainer* dmr0 = dm->get_dmr_vec()[0]; hamilt::Read_HContainer reader(dmr0, "./test_dm_dir/dmrs1_nao.csr", nlocal, &ucell, 0); reader.read(); @@ -179,15 +177,15 @@ TEST_F(InitDMFileTest, Nspin2_ReadTwoFiles) write_test_csr("./test_dm_dir/dmrs2_nao.csr", 0.5, 1, 2); // spin-down auto* dm = create_dm(2); - ASSERT_EQ(dm->_DMR.size(), 2); + ASSERT_EQ(dm->get_dmr_vec().size(), 2); // Read spin-up - hamilt::HContainer* dmr0 = dm->get_DMR_vector()[0]; + hamilt::HContainer* dmr0 = dm->get_dmr_vec()[0]; hamilt::Read_HContainer reader0(dmr0, "./test_dm_dir/dmrs1_nao.csr", nlocal, &ucell, 0); reader0.read(); // Read spin-down - hamilt::HContainer* dmr1 = dm->get_DMR_vector()[1]; + hamilt::HContainer* dmr1 = dm->get_dmr_vec()[1]; hamilt::Read_HContainer reader1(dmr1, "./test_dm_dir/dmrs2_nao.csr", nlocal, &ucell, 0); reader1.read(); @@ -222,17 +220,17 @@ TEST_F(InitDMFileTest, Nspin2_ReadTwoFiles) TEST_F(InitDMFileTest, Nspin2_DMRVectorSize) { auto* dm = create_dm(2); - EXPECT_EQ(dm->_DMR.size(), 2); - EXPECT_NE(dm->_DMR[0], nullptr); - EXPECT_NE(dm->_DMR[1], nullptr); + EXPECT_EQ(dm->get_dmr_vec().size(), 2); + EXPECT_NE(dm->get_dmr_vec()[0], nullptr); + EXPECT_NE(dm->get_dmr_vec()[1], nullptr); delete dm; } TEST_F(InitDMFileTest, Nspin1_DMRVectorSize) { auto* dm = create_dm(1); - EXPECT_EQ(dm->_DMR.size(), 1); - EXPECT_NE(dm->_DMR[0], nullptr); + EXPECT_EQ(dm->get_dmr_vec().size(), 1); + EXPECT_NE(dm->get_dmr_vec()[0], nullptr); delete dm; } diff --git a/source/source_lcao/test/test_output_hcontainer_consistency.cpp b/source/source_lcao/test/test_output_hcontainer_consistency.cpp index f722a580696..075495d4f1b 100644 --- a/source/source_lcao/test/test_output_hcontainer_consistency.cpp +++ b/source/source_lcao/test/test_output_hcontainer_consistency.cpp @@ -7,11 +7,9 @@ #include #include -#define private public #include "source_hamilt/module_hcontainer/hcontainer.h" #include "source_hamilt/module_hcontainer/output_hcontainer.h" #include "source_hamilt/module_hcontainer/read_hcontainer.h" -#undef private #include "source_cell/unitcell.h" #include "source_basis/module_ao/parallel_orbitals.h" diff --git a/source/source_lcao/test/tmp_mocks.cpp b/source/source_lcao/test/tmp_mocks.cpp index 10209add16b..18535dfb319 100644 --- a/source/source_lcao/test/tmp_mocks.cpp +++ b/source/source_lcao/test/tmp_mocks.cpp @@ -43,7 +43,7 @@ void UnitCell::set_iat2iwt(const int& npol_in) return; } -// stub for Grid_Driver::Find_atom (used by density_matrix_io.cpp but not exercised in test) +// stub for Grid_Driver::Find_atom (used by dmr_init.cpp but not exercised in test) void Grid_Driver::Find_atom(const UnitCell& ucell, const ModuleBase::Vector3& tau, const int& T, diff --git a/source/source_main/driver_run.cpp b/source/source_main/driver_run.cpp index e55be43c9b0..bb637ce69a2 100644 --- a/source/source_main/driver_run.cpp +++ b/source/source_main/driver_run.cpp @@ -2,7 +2,7 @@ #include "source_base/kernels/math_kernel_op.h" #include "source_cell/check_atomic_stru.h" #include "source_cell/mdcell.h" -#include "source_cell/module_neighlist/domain_decomposition.h" +#include "source_cell/domain_decomposition.h" #include "source_esolver/esolver_factory.h" #include "source_hsolver/kernels/hegvd_op.h" #include "source_io/module_json/para_json.h" @@ -87,7 +87,7 @@ void Driver::driver_run() unitcell::check_atomic_stru(ucell, input.min_dist_coef); ucell_initialized = true; -#ifdef __RAPIDJSON +#ifdef __JSON Json::gen_stru_wrapper(&ucell, input); #endif }; diff --git a/source/source_main/version.h b/source/source_main/version.h index 9e0703bbeaf..5aaed3f0baf 100644 --- a/source/source_main/version.h +++ b/source/source_main/version.h @@ -1,3 +1,3 @@ #ifndef VERSION -#define VERSION "v3.11.0-beta9" +#define VERSION "v3.11.0-beta10" #endif diff --git a/source/source_md/CMakeLists.txt b/source/source_md/CMakeLists.txt index 73f777a1256..000417d0d78 100644 --- a/source/source_md/CMakeLists.txt +++ b/source/source_md/CMakeLists.txt @@ -15,6 +15,8 @@ if(ENABLE_COVERAGE) add_coverage(md) endif() +add_subdirectory(module_plumed) + if(BUILD_TESTING) if(ENABLE_MPI) add_subdirectory(test) diff --git a/source/source_md/fire.cpp b/source/source_md/fire.cpp index 8e102160696..cb4611c6e9c 100644 --- a/source/source_md/fire.cpp +++ b/source/source_md/fire.cpp @@ -6,9 +6,15 @@ #endif #include "source_base/timer.h" -FIRE::FIRE(const Parameter& param_in, MDCell& mdcell_in) : MD_base(param_in, mdcell_in) +FIRE::FIRE(const MD_para& mdp_in, + const bool cal_stress_in, + const bool init_vel, + const int my_rank_in, + const double force_thr_in, + MDCell& mdcell_in) + : MD_base(mdp_in, cal_stress_in, init_vel, my_rank_in, mdcell_in) { - force_thr = param_in.inp.force_thr; + force_thr = force_thr_in; dt_max = -1.0; alpha_start = 0.10; alpha = alpha_start; @@ -82,8 +88,8 @@ void FIRE::print_md(std::ofstream& ofs, const bool& cal_stress) const double max_force = max * ModuleBase::Hartree_to_eV * ModuleBase::ANGSTROM_AU; - ofs << " LARGEST FORCE (eV/A) : " << max_force << std::endl; - std::cout << " LARGEST FORCE (eV/A) : " << max_force << std::endl; + ofs << " LARGEST FORCE (eV/A) : " << max_force << std::endl; + std::cout << " LARGEST FORCE (eV/A) : " << max_force << std::endl; return; } diff --git a/source/source_md/fire.h b/source/source_md/fire.h index 8ae54b77723..5b84c3be9ce 100644 --- a/source/source_md/fire.h +++ b/source/source_md/fire.h @@ -13,10 +13,34 @@ class FIRE : public MD_base { public: - FIRE(const Parameter& param_in, MDCell& mdcell_in); + /// @param force_thr_in force convergence threshold used by the FIRE descent + FIRE(const MD_para& mdp_in, + const bool cal_stress_in, + const bool init_vel, + const int my_rank_in, + const double force_thr_in, + MDCell& mdcell_in); ~FIRE(); + /// @brief mixing coefficient of the FIRE velocity update + double get_alpha() const + { + return alpha; + } + + /// @brief largest time step the adaptive scheme is allowed to reach + double get_dt_max() const + { + return dt_max; + } + + /// @brief number of consecutive steps with negative power + int get_negative_count() const + { + return negative_count; + } + private: void setup(ModuleESolver::ESolver* p_esolver, const std::string& global_readin_dir, DomainDecomposition& decomp); diff --git a/source/source_md/langevin.cpp b/source/source_md/langevin.cpp index a695d351146..204e946bbc6 100644 --- a/source/source_md/langevin.cpp +++ b/source/source_md/langevin.cpp @@ -4,7 +4,12 @@ #include "source_cell/unitcell.h" #include "source_base/timer.h" -Langevin::Langevin(const Parameter& param_in, MDCell& mdcell_in) : MD_base(param_in, mdcell_in) +Langevin::Langevin(const MD_para& mdp_in, + const bool cal_stress_in, + const bool init_vel, + const int my_rank_in, + MDCell& mdcell_in) + : MD_base(mdp_in, cal_stress_in, init_vel, my_rank_in, mdcell_in) { /// convert to a.u. unit assert(ModuleBase::AU_to_FS!=0.0); diff --git a/source/source_md/langevin.h b/source/source_md/langevin.h index f6c842e537f..58ad4b63a66 100644 --- a/source/source_md/langevin.h +++ b/source/source_md/langevin.h @@ -16,7 +16,11 @@ class Langevin : public MD_base { public: - Langevin(const Parameter& param_in, MDCell& mdcell_in); + Langevin(const MD_para& mdp_in, + const bool cal_stress_in, + const bool init_vel, + const int my_rank_in, + MDCell& mdcell_in); private: void setup(ModuleESolver::ESolver* p_esolver, const std::string& global_readin_dir, DomainDecomposition& decomp); diff --git a/source/source_md/md_base.cpp b/source/source_md/md_base.cpp index 100debadcdd..cb00647a865 100644 --- a/source/source_md/md_base.cpp +++ b/source/source_md/md_base.cpp @@ -8,15 +8,19 @@ #include #include -MD_base::MD_base(const Parameter& param_in, MDCell& mdcell_in) -: mdp(param_in.mdp), mdcell(mdcell_in) +MD_base::MD_base(const MD_para& mdp_in, + const bool cal_stress_in, + const bool init_vel, + const int my_rank_in, + MDCell& mdcell_in) +: mdp(mdp_in), mdcell(mdcell_in) { #ifdef __MPI my_rank = mdcell.mpi_rank(); #else - my_rank = param_in.globalv.myrank; + my_rank = my_rank_in; #endif - cal_stress = param_in.inp.cal_stress; + cal_stress = cal_stress_in; srand((mdp.md_seed >= 0 ? mdp.md_seed : 1) + my_rank); stop = false; @@ -35,7 +39,7 @@ MD_base::MD_base(const Parameter& param_in, MDCell& mdcell_in) step_ = 0; step_rst_ = 0; - MD_func::init_vel(mdcell, param_in.inp.init_vel, mdp.md_restart, md_tfirst, frozen_freedom_); + MD_func::init_vel(mdcell, init_vel, mdp.md_restart, md_tfirst, frozen_freedom_); t_current = MD_func::current_temp(kinetic, mdcell, frozen_freedom_); } @@ -164,9 +168,11 @@ void MD_base::print_md(std::ofstream& ofs, const bool& cal_stress) } std::cout << std::endl; + std::cout << std::setprecision(10); std::cout << " " << std::left << std::setw(20) << 2 * (potential + kinetic) << std::left << std::setw(20) << 2 * potential << std::left << std::setw(20) << 2 * kinetic << std::left << std::setw(20) << t_current * ModuleBase::Hartree_to_K; + std::cout << std::setprecision(8); if (cal_stress) { @@ -198,9 +204,11 @@ void MD_base::print_md(std::ofstream& ofs, const bool& cal_stress) } ofs << std::endl; + ofs << std::setprecision(10); ofs << " " << std::left << std::setw(20) << 2 * (potential + kinetic) << std::left << std::setw(20) << 2 * potential << std::left << std::setw(20) << 2 * kinetic << std::left << std::setw(20) << t_current * ModuleBase::Hartree_to_K; + ofs << std::setprecision(8); if (cal_stress) { diff --git a/source/source_md/md_base.h b/source/source_md/md_base.h index c3f870ee1e8..4095c409a0c 100644 --- a/source/source_md/md_base.h +++ b/source/source_md/md_base.h @@ -3,7 +3,7 @@ #include "source_cell/mdcell.h" #include "source_esolver/esolver.h" -#include "source_io/module_parameter/parameter.h" +#include "source_io/module_parameter/md_parameter.h" #include @@ -20,7 +20,21 @@ class DomainDecomposition; class MD_base { public: - MD_base(const Parameter& param_in, MDCell& mdcell_in); + /** + * @brief construct the integrator from the values it actually uses + * @param mdp_in the md input parameters; the reference is kept, so it must + * outlive the integrator + * @param cal_stress_in whether stress is calculated + * @param init_vel whether initial velocities are read from STRU + * @param my_rank_in MPI rank of the processor; only consulted in serial + * builds, where MDCell cannot supply it + * @param mdcell_in mdcell information + */ + MD_base(const MD_para& mdp_in, + const bool cal_stress_in, + const bool init_vel, + const int my_rank_in, + MDCell& mdcell_in); virtual ~MD_base(); /** @@ -56,13 +70,18 @@ class MD_base */ virtual void write_restart(const std::string& global_out_dir); - protected: /** * @brief restart MD when md_restart is true + * + * Public counterpart of write_restart(): setup() calls it internally when + * md_restart is set, and a caller that has just written a restart file may + * read it back through here. + * * @param global_readin_dir directory of files for reading */ virtual void restart(const std::string& global_readin_dir); + protected: /** * @brief perform one step update of pos due to atomic velocity */ @@ -75,6 +94,12 @@ class MD_base virtual void update_vel(); public: + /// @brief the time increment in a.u., converted from mdp.md_dt + double get_md_dt() const + { + return md_dt; + } + bool stop; ///< MD stop or not double t_current; ///< current temperature int step_; ///< the MD step finished in current calculation diff --git a/source/source_md/md_func.cpp b/source/source_md/md_func.cpp index 0bd89756e6c..a549e87732c 100644 --- a/source/source_md/md_func.cpp +++ b/source/source_md/md_func.cpp @@ -1,10 +1,9 @@ #include "md_func.h" -#include "source_cell/module_neighlist/domain_decomposition.h" +#include "source_cell/domain_decomposition.h" #include "source_base/global_variable.h" #include "source_base/timer.h" #include "source_io/module_output/output_log.h" -#include "source_io/module_parameter/parameter.h" #include #include @@ -411,7 +410,8 @@ void print_stress(std::ofstream& ofs, const ModuleBase::matrix& virial, const Mo void dump_info(const int& step, const std::string& global_out_dir, const MDCell& mdcell, - const Parameter& param_in, + const MD_para& mdp, + const bool cal_stress, const ModuleBase::matrix& virial) { std::stringstream file; @@ -426,15 +426,15 @@ void dump_info(const int& step, header << " " << mdcell.latvec().e11 << " " << mdcell.latvec().e12 << " " << mdcell.latvec().e13 << "\n"; header << " " << mdcell.latvec().e21 << " " << mdcell.latvec().e22 << " " << mdcell.latvec().e23 << "\n"; header << " " << mdcell.latvec().e31 << " " << mdcell.latvec().e32 << " " << mdcell.latvec().e33 << "\n"; - if (param_in.inp.cal_stress && param_in.mdp.dump_virial) + if (cal_stress && mdp.dump_virial) { header << "VIRIAL (kbar)\n"; for (int i = 0; i < 3; ++i) header << " " << virial(i, 0) * unit_virial << " " << virial(i, 1) * unit_virial << " " << virial(i, 2) * unit_virial << "\n"; } header << "INDEX LABEL POSITION (Angstrom)"; - if (param_in.mdp.dump_force) header << " FORCE (eV/Angstrom)"; - if (param_in.mdp.dump_vel) header << " VELOCITY (Angstrom/fs)"; + if (mdp.dump_force) header << " FORCE (eV/Angstrom)"; + if (mdp.dump_vel) header << " VELOCITY (Angstrom/fs)"; header << "\n"; std::vector type_offsets(mdcell.type_atom_counts().size() + 1, 0); for (std::size_t it = 0; it < mdcell.type_atom_counts().size(); ++it) @@ -449,9 +449,9 @@ void dump_info(const int& step, local << " " << type_offsets[static_cast(atom.type)] + atom.type_index << " " << mdcell.type_labels()[static_cast(atom.type)] << " " << atom.cart.x * unit_pos << " " << atom.cart.y * unit_pos << " " << atom.cart.z * unit_pos; - if (param_in.mdp.dump_force) + if (mdp.dump_force) local << " " << atom.force.x * unit_force << " " << atom.force.y * unit_force << " " << atom.force.z * unit_force; - if (param_in.mdp.dump_vel) + if (mdp.dump_vel) local << " " << atom.vel.x * unit_vel << " " << atom.vel.y * unit_vel << " " << atom.vel.z * unit_vel; local << "\n"; } diff --git a/source/source_md/md_func.h b/source/source_md/md_func.h index feab9518773..c8095d069e8 100644 --- a/source/source_md/md_func.h +++ b/source/source_md/md_func.h @@ -3,6 +3,7 @@ #include "source_cell/mdcell.h" #include "source_esolver/esolver.h" +#include "source_io/module_parameter/md_parameter.h" #include @@ -137,13 +138,15 @@ void print_stress(std::ofstream& ofs, const ModuleBase::matrix& virial, const Mo * @param step current md step * @param global_out_dir directory of output files * @param mdcell MD cell information - * @param param_in input parameters used in MD + * @param mdp the md input parameters + * @param cal_stress whether stress is calculated * @param virial lattice virial tensor */ void dump_info(const int& step, const std::string& global_out_dir, const MDCell& mdcell, - const Parameter& param_in, + const MD_para& mdp, + const bool cal_stress, const ModuleBase::matrix& virial); /** diff --git a/source/source_md/module_plumed/CMakeLists.txt b/source/source_md/module_plumed/CMakeLists.txt new file mode 100644 index 00000000000..4a80bce958e --- /dev/null +++ b/source/source_md/module_plumed/CMakeLists.txt @@ -0,0 +1,15 @@ +add_library( + module_plumed + OBJECT + plumed_interface.cpp +) + +# The PLUMED interface is only compiled in when the build is configured with +# -DENABLE_PLUMED=ON +if(ENABLE_PLUMED) + target_compile_definitions(module_plumed PRIVATE USE_PLUMED) +endif() + +if(ENABLE_COVERAGE) + add_coverage(module_plumed) +endif() diff --git a/source/source_md/module_plumed/plumed_interface.cpp b/source/source_md/module_plumed/plumed_interface.cpp new file mode 100644 index 00000000000..ca92f8f3d81 --- /dev/null +++ b/source/source_md/module_plumed/plumed_interface.cpp @@ -0,0 +1,203 @@ +#include "plumed_interface.h" + +#include + +#include "source_base/tool_quit.h" + +#ifdef USE_PLUMED +// C API of the PLUMED wrapper library, matching in a +// PLUMED installation. The handle is an opaque pointer-sized structure; the +// wrapper library takes care of loading the PLUMED kernel. +extern "C" +{ + plumed_handle plumed_create(); + void plumed_cmd(plumed_handle plumed, const char* key, const void* val); + void plumed_finalize(plumed_handle plumed); +} +#endif + +namespace ModulePlumed +{ + +namespace +{ +// Conversion factors from ABACUS atomic units to the PLUMED default units +// (length: nm, energy: kJ/mol, time: ps, mass: amu). They are declared to +// PLUMED once in init() with setMD*Units; afterwards all the arrays handed +// over to PLUMED are in ABACUS native units. +constexpr double BOHR_TO_NM = 0.052917721067; // 1 Bohr = 0.052917721067 nm +constexpr double HARTREE_TO_KJMOL = 2625.499638; // 1 Hartree = 2625.499638 kJ/mol +constexpr double AU_TIME_TO_PS = 2.418884326509e-5; // atomic unit of time = 2.418884326509e-5 ps +constexpr double ELECTRON_MASS_TO_AMU = 5.48579909065e-4; // 1 electron mass = 5.48579909065e-4 amu +} // namespace + +PlumedInterface::~PlumedInterface() +{ + this->finalize(); +} + +bool PlumedInterface::init(const std::string& plumed_file, + const int nat, + const double dt_au, + const double* masses_au) +{ +#ifdef USE_PLUMED + this->active_run_ = true; + this->nat_ = nat; + + // ABACUS stores atomic masses in atomic units (electron masses), while + // PLUMED expects them in amu + this->masses_amu_.resize(nat); + for (int i = 0; i < nat; ++i) + { + this->masses_amu_[i] = masses_au[i] * ELECTRON_MASS_TO_AMU; + } + + this->plumed_ = plumed_create(); + this->has_plumed_ = true; + if (this->plumed_.p == nullptr) + { + ModuleBase::WARNING_QUIT("ModulePlumed::PlumedInterface::init", + "could not create a PLUMED object, check the PLUMED installation"); + } + + int real_precision = 8; // double precision reals + int natoms = nat; + double length_units = BOHR_TO_NM; + double energy_units = HARTREE_TO_KJMOL; + double time_units = AU_TIME_TO_PS; + double timestep = dt_au; + const char* engine = "abacus"; + + // commands that configure PLUMED must be issued before "init" + plumed_cmd(this->plumed_, "setRealPrecision", &real_precision); + plumed_cmd(this->plumed_, "setMDLengthUnits", &length_units); + plumed_cmd(this->plumed_, "setMDEnergyUnits", &energy_units); + plumed_cmd(this->plumed_, "setMDTimeUnits", &time_units); + plumed_cmd(this->plumed_, "setPlumedDat", plumed_file.c_str()); + plumed_cmd(this->plumed_, "setNatoms", &natoms); + plumed_cmd(this->plumed_, "setMDEngine", engine); + plumed_cmd(this->plumed_, "setTimestep", ×tep); + plumed_cmd(this->plumed_, "init", nullptr); + + return true; +#else + (void)plumed_file; + (void)nat; + (void)dt_au; + (void)masses_au; + + ModuleBase::WARNING_QUIT("ModulePlumed::PlumedInterface::init", + "plumed was requested in INPUT but ABACUS was not compiled with PLUMED support; " + "reconfigure with -DENABLE_PLUMED=ON"); + return false; +#endif +} + +void PlumedInterface::prepare_virial_(const double* cell, const double* virial) +{ + // The Box (cell) data receives forces whenever a periodic collective + // variable is biased, so PLUMED always needs a valid force buffer here; + // when the MD run does not provide a virial, a zero buffer is registered + // instead (as in the reference implementation distributed with GROMACS). + // The virial is handed over in energy units with the convention + // virial_ij = -stress_ij * volume (the same quantity used by the official + // Quantum ESPRESSO interface); ABACUS keeps the potential part of the + // stress (Hartree/Bohr^3) in its MD virial matrix, so the volume is + // multiplied back here. + for (int i = 0; i < 9; ++i) + { + this->virial_buf_[i] = 0.0; + } + if (virial == nullptr || cell == nullptr) + { + return; + } + const double volume = std::fabs(cell[0] * (cell[4] * cell[8] - cell[5] * cell[7]) + - cell[1] * (cell[3] * cell[8] - cell[5] * cell[6]) + + cell[2] * (cell[3] * cell[7] - cell[4] * cell[6])); + for (int i = 0; i < 9; ++i) + { + this->virial_buf_[i] = -virial[i] * volume; + } +} + +void PlumedInterface::compute_forces_(const int istep, + const ModuleBase::Vector3* pos, + double* force, + const double* cell, + const double potential, + const double* virial) +{ +#ifdef USE_PLUMED + int step = istep; + double energy = potential; + + plumed_cmd(this->plumed_, "setStep", &step); + // Vector3 is a plain {double x, y, z} structure, so the arrays can + // be handed to PLUMED as contiguous [nat][3] buffers + plumed_cmd(this->plumed_, "setPositions", + reinterpret_cast(const_cast*>(pos))); + plumed_cmd(this->plumed_, "setMasses", this->masses_amu_.data()); + plumed_cmd(this->plumed_, "setBox", cell); + plumed_cmd(this->plumed_, "setEnergy", &energy); + // prepare the calculation graph; the arrays that receive the biasing + // forces are registered afterwards, as done in the reference + // implementations distributed with PLUMED + plumed_cmd(this->plumed_, "prepareCalc", nullptr); + plumed_cmd(this->plumed_, "setForces", force); + this->prepare_virial_(cell, virial); + plumed_cmd(this->plumed_, "setVirial", this->virial_buf_); + // flag this step as a checkpointing step: PLUMED then flushes all its + // output files (COLVAR, ...) in update(), so that the collective + // variables can be followed while the MD is running. Without it, + // PLUMED only flushes every 10000 steps. + int check_point = 1; + plumed_cmd(this->plumed_, "doCheckPoint", &check_point); + plumed_cmd(this->plumed_, "performCalc", nullptr); +#else + (void)istep; + (void)pos; + (void)force; + (void)cell; + (void)potential; + (void)virial; +#endif +} + +void PlumedInterface::move(const int istep, + const ModuleBase::Vector3* pos, + ModuleBase::Vector3* force, + const double* cell, + const double potential, + const double* virial) +{ +#ifdef USE_PLUMED + if (!this->active_run_) + { + return; + } + this->compute_forces_(istep, pos, reinterpret_cast(force), cell, potential, virial); +#else + (void)istep; + (void)pos; + (void)force; + (void)cell; + (void)potential; + (void)virial; +#endif +} + +void PlumedInterface::finalize() +{ +#ifdef USE_PLUMED + if (this->has_plumed_) + { + plumed_finalize(this->plumed_); + this->has_plumed_ = false; + } +#endif + this->active_run_ = false; +} + +} // namespace ModulePlumed diff --git a/source/source_md/module_plumed/plumed_interface.h b/source/source_md/module_plumed/plumed_interface.h new file mode 100644 index 00000000000..8a1d08e4a3c --- /dev/null +++ b/source/source_md/module_plumed/plumed_interface.h @@ -0,0 +1,133 @@ +#ifndef PLUMED_INTERFACE_H +#define PLUMED_INTERFACE_H + +#include +#include + +#include "source_base/vector3.h" + +/// opaque handle of the PLUMED wrapper library, see +/// in a PLUMED installation; it is a plain pointer-sized structure, so it can +/// be stored by value and initialized to a null pointer +struct plumed_handle +{ + void* p; +}; + +namespace ModulePlumed +{ + +/** + * @brief interface between an ABACUS MD run and the PLUMED plugin + * + * This class wraps the C API of the PLUMED wrapper library (``libplumed.so``, + * ``plumed_create`` / ``plumed_cmd`` / ``plumed_finalize``, see + * in a PLUMED installation) so that + * collective variables, biasing potentials and free-energy methods provided + * by PLUMED (https://www.plumed.org) can be used in ABACUS MD runs. + * + * Units: ABACUS atomic units differ from the PLUMED defaults, so the + * conversion factors are declared once in init() through the commands + * ``setMDEnergyUnits`` (Hartree -> kJ/mol), ``setMDLengthUnits`` + * (Bohr -> nm) and ``setMDTimeUnits`` (atomic unit of time -> ps). From then + * on all quantities are passed in ABACUS native units and PLUMED performs the + * conversion internally: + * - positions / cell : Bohr + * - forces : Hartree/Bohr (biasing forces are + * added in place by PLUMED) + * - potential energy : Hartree + * - virial : Hartree (=-stress * volume, + * following the Quantum ESPRESSO + * interface convention) + * - masses : electron masses (converted to amu + * internally) + * + * The interface currently supports a single MPI rank only, because PLUMED + * must see the whole system while ABACUS distributes the atoms of an MD run + * over the ranks; run_md.cpp stops the run before the MD loop when more than + * one rank is used. + * + * This interface is a no-op stub unless ABACUS is configured with + * ``-DENABLE_PLUMED=ON``; requesting PLUMED in INPUT while ABACUS was built + * without it stops the run with an error message. + */ +class PlumedInterface +{ + public: + PlumedInterface() = default; + ~PlumedInterface(); + + PlumedInterface(const PlumedInterface&) = delete; + PlumedInterface& operator=(const PlumedInterface&) = delete; + + /** + * @brief create a PLUMED object and read the PLUMED input file + * + * Must be called once before the MD loop. + * + * @param plumed_file [in] path of the PLUMED input file + * @param nat [in] number of atoms + * @param dt_au [in] MD time step in atomic units of time + * @param masses_au [in] atomic masses in ABACUS atomic units (electron + * masses), nat entries + * @return true when the interface is ready to be used in the MD loop + */ + bool init(const std::string& plumed_file, + const int nat, + const double dt_au, + const double* masses_au); + + /** + * @brief one MD step: hand over the configuration, let PLUMED compute the + * collective variables and add the biasing forces to ``force`` + * + * @param istep [in] global MD step index + * @param pos [in] atomic positions in Bohr (nat entries) + * @param force [in,out] atomic forces in Hartree/Bohr; PLUMED adds the + * biasing forces in place + * @param cell [in] 3x3 cell matrix in Bohr, row-major, the rows being + * the lattice vectors + * @param potential [in] potential energy in Hartree + * @param virial [in] 3x3 virial in Hartree (=-stress * volume), or + * nullptr when the stress is not available + */ + void move(const int istep, + const ModuleBase::Vector3* pos, + ModuleBase::Vector3* force, + const double* cell, + const double potential, + const double* virial); + + /// @brief finalize PLUMED (flushes the COLVAR and other output files) + void finalize(); + + /// @brief whether PLUMED is active in this run + bool active() const + { + return this->active_run_; + } + + private: + /// @brief register the configuration of one step with PLUMED and let it + /// compute the collective variables and the biasing forces + void compute_forces_(const int istep, + const ModuleBase::Vector3* pos, + double* force, + const double* cell, + const double potential, + const double* virial); + + /// @brief build the 3x3 virial (in energy units) handed over to PLUMED + void prepare_virial_(const double* cell, const double* virial); + + plumed_handle plumed_{nullptr}; ///< handle returned by plumed_create() + bool has_plumed_ = false; ///< whether a PLUMED object is alive + bool active_run_ = false; ///< whether init() has been called + int nat_ = 0; ///< number of atoms + std::vector masses_amu_; ///< atomic masses in amu + double virial_buf_[9] = {0.0}; ///< virial handed over to PLUMED (energy units) +}; + +} // namespace ModulePlumed + +#endif // PLUMED_INTERFACE_H diff --git a/source/source_md/msst.cpp b/source/source_md/msst.cpp index 3ed4eaca509..02f311a7395 100644 --- a/source/source_md/msst.cpp +++ b/source/source_md/msst.cpp @@ -7,7 +7,12 @@ #endif #include "source_base/timer.h" -MSST::MSST(const Parameter& param_in, MDCell& mdcell_in) : MD_base(param_in, mdcell_in) +MSST::MSST(const MD_para& mdp_in, + const bool cal_stress_in, + const bool init_vel, + const int my_rank_in, + MDCell& mdcell_in) + : MD_base(mdp_in, cal_stress_in, init_vel, my_rank_in, mdcell_in) { msst_qmass = mdp.msst_qmass / pow(ModuleBase::ANGSTROM_AU, 4) / pow(ModuleBase::AU_to_MASS, 2); msst_vel = mdp.msst_vel * ModuleBase::ANGSTROM_AU * ModuleBase::AU_to_FS; diff --git a/source/source_md/msst.h b/source/source_md/msst.h index 180213bf075..00ba2a7914a 100644 --- a/source/source_md/msst.h +++ b/source/source_md/msst.h @@ -15,9 +15,43 @@ class MSST : public MD_base { public: - MSST(const Parameter& param_in, MDCell& mdcell_in); + MSST(const MD_para& mdp_in, + const bool cal_stress_in, + const bool init_vel, + const int my_rank_in, + MDCell& mdcell_in); ~MSST(); + /// @brief time derivative of the volume, per lattice direction + const ModuleBase::Vector3& get_omega() const + { + return omega; + } + + /// @brief energy of the initial configuration + double get_e0() const + { + return e0; + } + + /// @brief volume of the initial configuration + double get_v0() const + { + return v0; + } + + /// @brief pressure of the initial configuration + double get_p0() const + { + return p0; + } + + /// @brief Lagrangian location of the cell + double get_lag_pos() const + { + return lag_pos; + } + private: void setup(ModuleESolver::ESolver* p_esolver, const std::string& global_readin_dir, DomainDecomposition& decomp); void first_half(std::ofstream& ofs); diff --git a/source/source_md/nhchain.cpp b/source/source_md/nhchain.cpp index 76d5e85c14e..9a99f1b05e8 100644 --- a/source/source_md/nhchain.cpp +++ b/source/source_md/nhchain.cpp @@ -6,7 +6,12 @@ #endif #include "source_base/timer.h" -Nose_Hoover::Nose_Hoover(const Parameter& param_in, MDCell& mdcell_in) : MD_base(param_in, mdcell_in) +Nose_Hoover::Nose_Hoover(const MD_para& mdp_in, + const bool cal_stress_in, + const bool init_vel, + const int my_rank_in, + MDCell& mdcell_in) + : MD_base(mdp_in, cal_stress_in, init_vel, my_rank_in, mdcell_in) { const double unit_transform = ModuleBase::HARTREE_SI / pow(ModuleBase::BOHR_RADIUS_SI, 3) * 1.0e-8; diff --git a/source/source_md/nhchain.h b/source/source_md/nhchain.h index d1514d201b0..6e0db3bec60 100644 --- a/source/source_md/nhchain.h +++ b/source/source_md/nhchain.h @@ -12,9 +12,47 @@ class Nose_Hoover : public MD_base { public: - Nose_Hoover(const Parameter& param_in, MDCell& mdcell_in); + Nose_Hoover(const MD_para& mdp_in, + const bool cal_stress_in, + const bool init_vel, + const int my_rank_in, + MDCell& mdcell_in); ~Nose_Hoover(); + /// @brief positions of the thermostats coupled with the particles + /// (md_tchain entries) + const double* get_eta() const + { + return eta; + } + + /// @brief velocities of the thermostats coupled with the particles + /// (md_tchain entries) + const double* get_v_eta() const + { + return v_eta; + } + + /// @brief positions of the thermostats coupled with the barostat + /// (md_pchain entries) + const double* get_peta() const + { + return peta; + } + + /// @brief velocities of the thermostats coupled with the barostat + /// (md_pchain entries) + const double* get_v_peta() const + { + return v_peta; + } + + /// @brief velocities of the six lattice components + const double* get_v_omega() const + { + return v_omega; + } + private: void setup(ModuleESolver::ESolver* p_esolver, const std::string& global_readin_dir, DomainDecomposition& decomp); void first_half(std::ofstream& ofs); diff --git a/source/source_md/run_md.cpp b/source/source_md/run_md.cpp index b2eb16420da..114f9cbb3a5 100644 --- a/source/source_md/run_md.cpp +++ b/source/source_md/run_md.cpp @@ -6,21 +6,28 @@ #include "source_base/parallel_cell.h" #include "source_cell/mdcell_reader.h" #include "source_cell/mdcell.h" -#include "source_cell/module_neighlist/domain_decomposition.h" +#include "source_cell/domain_decomposition.h" #include "source_io/module_parameter/parameter.h" #include "fire.h" #include "langevin.h" #include "md_func.h" #include "source_base/global_file.h" #include "source_base/timer.h" +#ifdef __JSON +#include "source_io/module_json/output_info.h" +#endif #include "source_io/module_output/print_info.h" #include "msst.h" #include "nhchain.h" #include "verlet.h" #include "source_cell/update_cell.h" #include "source_cell/print_cell.h" +#include "source_md/module_plumed/plumed_interface.h" #include +#ifdef __MPI +#include +#endif namespace Run_MD { @@ -74,6 +81,54 @@ void prepare_mdcell(MDCell& mdcell, UnitCell& ucell, DomainDecomposition& decomp mdcell.mutable_stru_meta() = unitcell::make_stru_meta(ucell); } +namespace +{ +/// @brief collect the absolute Cartesian positions (in Bohr) of the atoms +/// owned by this rank; the cell stores cart in units of the lattice constant +void collect_plumed_positions(const MDCell& mdcell, std::vector>& positions) +{ + const double lat0 = mdcell.lat0(); + const std::vector& atoms = mdcell.owned_atoms(); + positions.resize(atoms.size()); + for (std::size_t i = 0; i < atoms.size(); ++i) + { + positions[i] = atoms[i].cart * lat0; + } +} + +/// @brief gather the atomic forces (in Hartree/Bohr) +void collect_plumed_forces(const MDCell& mdcell, std::vector>& forces) +{ + const std::vector& atoms = mdcell.owned_atoms(); + forces.resize(atoms.size()); + for (std::size_t i = 0; i < atoms.size(); ++i) + { + forces[i] = atoms[i].force; + } +} + +/// @brief write the (PLUMED-biased) forces back into the cell +void apply_plumed_forces(MDCell& mdcell, const std::vector>& forces) +{ + std::vector& atoms = mdcell.owned_atoms(); + for (std::size_t i = 0; i < atoms.size(); ++i) + { + atoms[i].force = forces[i]; + } +} + +/// @brief 3x3 cell matrix (in Bohr, row-major) expected by PLUMED +void collect_plumed_cell(const MDCell& mdcell, double* cell) +{ + const double lat0 = mdcell.lat0(); + const double* latvec = &mdcell.latvec().e11; // unitless lattice vectors + for (int i = 0; i < 9; ++i) + { + cell[i] = latvec[i] * lat0; + } +} +} // namespace + void md_line(MDCell& mdcell, ModuleESolver::ESolver* p_esolver, const Parameter& param_in, @@ -83,37 +138,97 @@ void md_line(MDCell& mdcell, ModuleBase::timer::start("Run_MD", "md_line"); /// determine the md_type MD_base* mdrun = nullptr; + /// the integrators take the values they use, not the whole Parameter + const MD_para& mdp = param_in.mdp; + const bool cal_stress = param_in.inp.cal_stress; + const bool init_vel = param_in.inp.init_vel; + const int my_rank = param_in.globalv.myrank; if (param_in.mdp.md_type == "fire") { - mdrun = new FIRE(param_in, mdcell); + mdrun = new FIRE(mdp, cal_stress, init_vel, my_rank, param_in.inp.force_thr, mdcell); } else if ((param_in.mdp.md_type == "nvt" && param_in.mdp.md_thermostat == "nhc") || param_in.mdp.md_type == "npt") { - mdrun = new Nose_Hoover(param_in, mdcell); + mdrun = new Nose_Hoover(mdp, cal_stress, init_vel, my_rank, mdcell); } else if (param_in.mdp.md_type == "nve" || param_in.mdp.md_type == "nvt") { - mdrun = new Verlet(param_in, mdcell); + mdrun = new Verlet(mdp, cal_stress, init_vel, my_rank, mdcell); } else if (param_in.mdp.md_type == "langevin") { - mdrun = new Langevin(param_in, mdcell); + mdrun = new Langevin(mdp, cal_stress, init_vel, my_rank, mdcell); } else if (param_in.mdp.md_type == "msst") { - mdrun = new MSST(param_in, mdcell); + mdrun = new MSST(mdp, cal_stress, init_vel, my_rank, mdcell); } else { ModuleBase::WARNING_QUIT("md_line", "no such md_type!"); } + /// interface to PLUMED: no-op unless `plumed` is set in INPUT and ABACUS + /// was compiled with PLUMED support (-DPLUMED_ROOT=...) + ModulePlumed::PlumedInterface plumed; + std::vector> plumed_positions; + std::vector> plumed_forces; + double plumed_cell[9]; + + auto run_plumed_step = [&](const int istep) { + collect_plumed_positions(mdcell, plumed_positions); + collect_plumed_forces(mdcell, plumed_forces); + collect_plumed_cell(mdcell, plumed_cell); + plumed.move(istep, + plumed_positions.data(), + plumed_forces.data(), + plumed_cell, + mdrun->potential, + param_in.inp.cal_stress ? mdrun->virial.c : nullptr); + apply_plumed_forces(mdcell, plumed_forces); + }; + /// md cycle, mohan update 2026-01-04, change '<=' to '<' while ((mdrun->step_ + mdrun->step_rst_) < param_in.mdp.md_nstep && !mdrun->stop) { +#ifdef __JSON + // JSON output currently follows the UnitCell-backed electronic-structure path. + // Start one output record before the solver appends SCF information for this MD step. + if (mdcell.has_backing_unitcell()) + { + Json::init_output_array_obj(); + } +#endif + if (mdrun->step_ == 0) { mdrun->setup(p_esolver, param_in.globalv.global_readin_dir, decomp); + + if (param_in.mdp.plumed) + { + int nproc = 1; +#ifdef __MPI + MPI_Comm_size(MPI_COMM_WORLD, &nproc); +#endif + if (nproc > 1) + { + ModuleBase::WARNING_QUIT("md_line", + "the PLUMED interface currently supports a single MPI rank only; " + "run the MD with one process (OpenMP threads are fine)"); + } + std::vector plumed_masses(mdcell.owned_atoms().size()); + for (std::size_t i = 0; i < mdcell.owned_atoms().size(); ++i) + { + plumed_masses[i] = mdcell.owned_atoms()[i].mass; + } + plumed.init(param_in.mdp.plumed_file, + static_cast(mdcell.owned_atoms().size()), + mdrun->get_md_dt(), + plumed_masses.data()); + /// the initial configuration already carries positions and + /// forces, so PLUMED can be called right away + run_plumed_step(mdrun->step_ + mdrun->step_rst_); + } } else { @@ -134,6 +249,14 @@ void md_line(MDCell& mdcell, mdrun->virial, param_in.mdp.md_out_force); + if (plumed.active()) + { + /// let PLUMED compute the collective variables and add the + /// biasing forces; they are used by the second half of the + /// current step and by the first half of the next one + run_plumed_step(mdrun->step_ + mdrun->step_rst_); + } + mdrun->second_half(); MD_func::compute_stress(mdcell, @@ -152,7 +275,8 @@ void md_line(MDCell& mdcell, MD_func::dump_info(mdrun->step_ + mdrun->step_rst_, PARAM.globalv.global_out_dir, mdcell, - param_in, + mdp, + cal_stress, mdrun->virial); } @@ -172,6 +296,9 @@ void md_line(MDCell& mdcell, mdrun->step_++; } + /// flush the COLVAR and other PLUMED output files + plumed.finalize(); + delete mdrun; ModuleBase::timer::end("Run_MD", "md_line"); return; diff --git a/source/source_md/test/CMakeLists.txt b/source/source_md/test/CMakeLists.txt index 5fd8b8babe6..406cd5be5d8 100644 --- a/source/source_md/test/CMakeLists.txt +++ b/source/source_md/test/CMakeLists.txt @@ -50,7 +50,7 @@ list(APPEND depend_files ../../source_cell/module_neighlist/neighbor_search.cpp ../../source_cell/module_neighlist/bin_manager.cpp ../../source_cell/module_neighlist/page_allocator.cpp - ../../source_cell/module_neighlist/domain_decomposition.cpp + ../../source_cell/domain_decomposition.cpp ../../source_cell/mdcell.cpp ../../source_base/output.cpp ../../source_io/module_output/output_log.cpp @@ -85,11 +85,20 @@ AddTest( ${depend_files} ) +set(md_run_json_sources) +if(ENABLE_JSON) + list(APPEND md_run_json_sources + ../../source_io/module_json/abacusjson.cpp + ../../source_io/module_json/output_info.cpp + ) +endif() + AddTest( TARGET MODULE_MD_run LIBS parameter psi device SOURCES run_md_test.cpp ../run_md.cpp + ../module_plumed/plumed_interface.cpp ../md_base.cpp ../fire.cpp ../langevin.cpp @@ -98,6 +107,7 @@ AddTest( ../verlet.cpp ../../source_cell/mdcell_reader.cpp ../../source_cell/magnetism.cpp + ${md_run_json_sources} ${depend_files} ) diff --git a/source/source_md/test/fire_test.cpp b/source/source_md/test/fire_test.cpp index c337dad939f..8655b5aa90b 100644 --- a/source/source_md/test/fire_test.cpp +++ b/source/source_md/test/fire_test.cpp @@ -1,11 +1,6 @@ -#include "source_cell/module_neighlist/domain_decomposition.h" +#include "source_cell/domain_decomposition.h" #include "gmock/gmock.h" #include "gtest/gtest.h" -#define private public -#include "source_io/module_parameter/parameter.h" -#undef private -#define private public -#define protected public #include "setcell.h" #include "source_esolver/esolver_lj.h" #include "source_md/fire.h" @@ -43,20 +38,23 @@ class FIREtest : public testing::Test UnitCell ucell; MDCell mdcell; DomainDecomposition decomp; - Parameter param_in; + Input_para inp; ModuleESolver::ESolver* p_esolver; + /// the md writers take the directory explicitly, so the test owns it + const std::string out_dir = "./"; + const std::string readin_dir = "./"; void SetUp() { Setcell::setupcell(ucell); - Setcell::parameters(param_in.input); + Setcell::parameters(inp); p_esolver = new ModuleESolver::ESolver_LJ(); mdcell = Setcell::setup_mdcell(ucell); decomp.init(ModuleBase::world_comm_domain(), mdcell.latvec(), mdcell.lat0(), 0.0, 0.0); - p_esolver->before_all_runners(mdcell, param_in.inp); - mdrun = new FIRE(param_in, mdcell); - mdrun->setup(p_esolver, PARAM.sys.global_readin_dir, decomp); + p_esolver->before_all_runners(mdcell, inp); + mdrun = new FIRE(inp.mdp, inp.cal_stress, inp.init_vel, /*my_rank=*/0, inp.force_thr, mdcell); + mdrun->setup(p_esolver, readin_dir, decomp); } void TearDown() @@ -147,7 +145,7 @@ TEST_F(FIREtest, WriteRestart) { mdrun->step_ = 1; mdrun->step_rst_ = 2; - mdrun->write_restart(PARAM.sys.global_out_dir); + mdrun->write_restart(out_dir); std::ifstream ifs("Restart_md.txt"); std::string output_str; @@ -168,15 +166,15 @@ TEST_F(FIREtest, WriteRestart) TEST_F(FIREtest, Restart) { - mdrun->restart(PARAM.sys.global_readin_dir); + mdrun->restart(readin_dir); remove("Restart_md.txt"); FIRE* fire = dynamic_cast(mdrun); EXPECT_EQ(mdrun->step_rst_, 3); - EXPECT_EQ(fire->alpha, 0.1); - EXPECT_EQ(fire->negative_count, 0); - EXPECT_EQ(fire->dt_max, -1); - EXPECT_EQ(fire->md_dt, 41.3414); + EXPECT_EQ(fire->get_alpha(), 0.1); + EXPECT_EQ(fire->get_negative_count(), 0); + EXPECT_EQ(fire->get_dt_max(), -1); + EXPECT_EQ(fire->get_md_dt(), 41.3414); } TEST_F(FIREtest, PrintMD) @@ -209,7 +207,7 @@ TEST_F(FIREtest, PrintMD) EXPECT_THAT( output_str, testing::HasSubstr( - " -0.015365236 -0.023915637 0.0085504016 300 1.0846391 ")); + " -0.01536523561 -0.02391563725 0.008550401641 300 1.0846391 ")); getline(ifs, output_str); EXPECT_THAT( output_str, diff --git a/source/source_md/test/langevin_test.cpp b/source/source_md/test/langevin_test.cpp index bd146a5473e..784b6a9fd18 100644 --- a/source/source_md/test/langevin_test.cpp +++ b/source/source_md/test/langevin_test.cpp @@ -1,11 +1,6 @@ -#include "source_cell/module_neighlist/domain_decomposition.h" +#include "source_cell/domain_decomposition.h" #include "gmock/gmock.h" #include "gtest/gtest.h" -#define private public -#include "source_io/module_parameter/parameter.h" -#undef private -#define private public -#define protected public #include "setcell.h" #include "source_esolver/esolver_lj.h" #include "source_md/langevin.h" @@ -43,20 +38,23 @@ class Langevin_test : public testing::Test UnitCell ucell; MDCell mdcell; DomainDecomposition decomp; - Parameter param_in; + Input_para inp; ModuleESolver::ESolver* p_esolver; + /// the md writers take the directory explicitly, so the test owns it + const std::string out_dir = "./"; + const std::string readin_dir = "./"; void SetUp() { Setcell::setupcell(ucell); - Setcell::parameters(param_in.input); + Setcell::parameters(inp); p_esolver = new ModuleESolver::ESolver_LJ(); mdcell = Setcell::setup_mdcell(ucell); decomp.init(ModuleBase::world_comm_domain(), mdcell.latvec(), mdcell.lat0(), 0.0, 0.0); - p_esolver->before_all_runners(mdcell, param_in.inp); - mdrun = new Langevin(param_in, mdcell); - mdrun->setup(p_esolver, PARAM.sys.global_readin_dir, decomp); + p_esolver->before_all_runners(mdcell, inp); + mdrun = new Langevin(inp.mdp, inp.cal_stress, inp.init_vel, /*my_rank=*/0, mdcell); + mdrun->setup(p_esolver, readin_dir, decomp); } void TearDown() @@ -148,7 +146,7 @@ TEST_F(Langevin_test, write_restart) { mdrun->step_ = 1; mdrun->step_rst_ = 2; - mdrun->write_restart(PARAM.sys.global_out_dir); + mdrun->write_restart(out_dir); std::ifstream ifs("Restart_md.txt"); std::string output_str; @@ -159,7 +157,7 @@ TEST_F(Langevin_test, write_restart) TEST_F(Langevin_test, restart) { - mdrun->restart(PARAM.sys.global_readin_dir); + mdrun->restart(readin_dir); remove("Restart_md.txt"); EXPECT_EQ(mdrun->step_rst_, 3); @@ -194,7 +192,7 @@ TEST_F(Langevin_test, print_md) EXPECT_THAT( output_str, testing::HasSubstr( - " -0.015365236 -0.023915637 0.0085504016 300 1.0846391 ")); + " -0.01536523561 -0.02391563725 0.008550401641 300 1.0846391 ")); getline(ifs, output_str); EXPECT_THAT( output_str, diff --git a/source/source_md/test/lj_pot_test.cpp b/source/source_md/test/lj_pot_test.cpp index ecf2ae0214f..0ae49fb2b41 100644 --- a/source/source_md/test/lj_pot_test.cpp +++ b/source/source_md/test/lj_pot_test.cpp @@ -1,11 +1,9 @@ -#include "source_cell/module_neighlist/domain_decomposition.h" +#include "source_cell/domain_decomposition.h" #include "gtest/gtest.h" -#define private public #include "setcell.h" #include "source_esolver/esolver_lj.h" #include "source_io/module_parameter/parameter.h" #include "source_md/md_func.h" -#undef private #define doublethreshold 1e-12 /************************************************ @@ -25,7 +23,7 @@ class LJ_pot_test : public testing::Test double potential; int natom; UnitCell ucell; - Parameter param; + Input_para inp; void SetUp() { @@ -34,7 +32,7 @@ class LJ_pot_test : public testing::Test natom = ucell.nat; stress.create(3, 3); - Setcell::parameters(param.input); + Setcell::parameters(inp); } void TearDown() @@ -49,7 +47,7 @@ TEST_F(LJ_pot_test, potential) mdcell = Setcell::setup_mdcell(ucell); DomainDecomposition decomp; decomp.init(ModuleBase::world_comm_domain(), mdcell.latvec(), mdcell.lat0(), 0.0, 0.0); - p_esolver->before_all_runners(mdcell, param.inp); + p_esolver->before_all_runners(mdcell, inp); MD_func::force_virial(p_esolver, 0, mdcell, decomp, potential, true, stress, false); EXPECT_NEAR(potential, -0.011957818623534381, doublethreshold); } @@ -57,7 +55,7 @@ TEST_F(LJ_pot_test, potential) TEST_F(LJ_pot_test, unitcell_compatibility) { ModuleESolver::ESolver* p_esolver = new ModuleESolver::ESolver_LJ(); - p_esolver->before_all_runners(ucell, param.inp); + p_esolver->before_all_runners(ucell, inp); p_esolver->runner(ucell, 0); p_esolver->cal_force(ucell, stress); @@ -74,7 +72,7 @@ TEST_F(LJ_pot_test, force) mdcell = Setcell::setup_mdcell(ucell); DomainDecomposition decomp; decomp.init(ModuleBase::world_comm_domain(), mdcell.latvec(), mdcell.lat0(), 0.0, 0.0); - p_esolver->before_all_runners(mdcell, param.inp); + p_esolver->before_all_runners(mdcell, inp); MD_func::force_virial(p_esolver, 0, mdcell, decomp, potential, true, stress, false); const std::vector& atoms = mdcell.owned_atoms(); EXPECT_NEAR(atoms[0].force.x, 0.00049817733089377704, doublethreshold); @@ -98,7 +96,7 @@ TEST_F(LJ_pot_test, mdcell_cal_force) mdcell = Setcell::setup_mdcell(ucell); DomainDecomposition decomp; decomp.init(ModuleBase::world_comm_domain(), mdcell.latvec(), mdcell.lat0(), 0.0, 0.0); - p_esolver.before_all_runners(mdcell, param.inp); + p_esolver.before_all_runners(mdcell, inp); decomp.prepare_neighbors(mdcell); p_esolver.runner(mdcell, 0); @@ -120,7 +118,7 @@ TEST_F(LJ_pot_test, stress) mdcell = Setcell::setup_mdcell(ucell); DomainDecomposition decomp; decomp.init(ModuleBase::world_comm_domain(), mdcell.latvec(), mdcell.lat0(), 0.0, 0.0); - p_esolver->before_all_runners(mdcell, param.inp); + p_esolver->before_all_runners(mdcell, inp); MD_func::force_virial(p_esolver, 0, mdcell, decomp, potential, true, stress, false); EXPECT_NEAR(stress(0, 0), 8.0360222227631859e-07, doublethreshold); EXPECT_NEAR(stress(0, 1), 1.7207745586539077e-07, doublethreshold); @@ -140,7 +138,7 @@ TEST_F(LJ_pot_test, mdcell_stress_includes_external_pressure) mdcell = Setcell::setup_mdcell(ucell); DomainDecomposition decomp; decomp.init(ModuleBase::world_comm_domain(), mdcell.latvec(), mdcell.lat0(), 0.0, 0.0); - Input_para input = param.inp; + Input_para input = inp; p_esolver.before_all_runners(mdcell, input); decomp.prepare_neighbors(mdcell); p_esolver.runner(mdcell, 0); @@ -155,9 +153,9 @@ TEST_F(LJ_pot_test, mdcell_stress_includes_external_pressure) p_esolver.cal_stress(mdcell, stress); const double unit_transform = ModuleBase::RYDBERG_SI / pow(ModuleBase::BOHR_RADIUS_SI, 3) * 1.0e-8; - EXPECT_NEAR(stress(0, 0), p_esolver.lj_virial(0, 0) - 1.0 / unit_transform, doublethreshold); - EXPECT_NEAR(stress(1, 1), p_esolver.lj_virial(1, 1) - 2.0 / unit_transform, doublethreshold); - EXPECT_NEAR(stress(2, 2), p_esolver.lj_virial(2, 2) - 3.0 / unit_transform, doublethreshold); + EXPECT_NEAR(stress(0, 0), p_esolver.get_lj_virial()(0, 0) - 1.0 / unit_transform, doublethreshold); + EXPECT_NEAR(stress(1, 1), p_esolver.get_lj_virial()(1, 1) - 2.0 / unit_transform, doublethreshold); + EXPECT_NEAR(stress(2, 2), p_esolver.get_lj_virial()(2, 2) - 3.0 / unit_transform, doublethreshold); input.press1 = saved_press1; input.press2 = saved_press2; @@ -169,24 +167,24 @@ TEST_F(LJ_pot_test, RcutSearchRadius) ModuleESolver::ESolver_LJ* p_esolver = new ModuleESolver::ESolver_LJ(); ucell.ntype = 2; std::vector rcut = {3.0}; - p_esolver->rcut_search_radius(ucell.ntype, rcut); + p_esolver->rcut_search_radius_for_testing(ucell.ntype, rcut); for (int i = 0; i < ucell.ntype; i++) { for (int j = 0; j < ucell.ntype; j++) { - EXPECT_NEAR(p_esolver->lj_rcut(i, j), 3.0 * ModuleBase::ANGSTROM_AU, doublethreshold); + EXPECT_NEAR(p_esolver->get_lj_rcut()(i, j), 3.0 * ModuleBase::ANGSTROM_AU, doublethreshold); } } - EXPECT_NEAR(p_esolver->search_radius, 3.0 * ModuleBase::ANGSTROM_AU + 0.01, doublethreshold); + EXPECT_NEAR(p_esolver->get_search_radius(), 3.0 * ModuleBase::ANGSTROM_AU + 0.01, doublethreshold); rcut = {3.0, 4.0, 5.0}; - p_esolver->rcut_search_radius(ucell.ntype, rcut); - EXPECT_NEAR(p_esolver->lj_rcut(0, 0), 3.0 * ModuleBase::ANGSTROM_AU, doublethreshold); - EXPECT_NEAR(p_esolver->lj_rcut(0, 1), 4.0 * ModuleBase::ANGSTROM_AU, doublethreshold); - EXPECT_NEAR(p_esolver->lj_rcut(1, 0), 4.0 * ModuleBase::ANGSTROM_AU, doublethreshold); - EXPECT_NEAR(p_esolver->lj_rcut(1, 1), 5.0 * ModuleBase::ANGSTROM_AU, doublethreshold); - EXPECT_NEAR(p_esolver->search_radius, 5.0 * ModuleBase::ANGSTROM_AU + 0.01, doublethreshold); + p_esolver->rcut_search_radius_for_testing(ucell.ntype, rcut); + EXPECT_NEAR(p_esolver->get_lj_rcut()(0, 0), 3.0 * ModuleBase::ANGSTROM_AU, doublethreshold); + EXPECT_NEAR(p_esolver->get_lj_rcut()(0, 1), 4.0 * ModuleBase::ANGSTROM_AU, doublethreshold); + EXPECT_NEAR(p_esolver->get_lj_rcut()(1, 0), 4.0 * ModuleBase::ANGSTROM_AU, doublethreshold); + EXPECT_NEAR(p_esolver->get_lj_rcut()(1, 1), 5.0 * ModuleBase::ANGSTROM_AU, doublethreshold); + EXPECT_NEAR(p_esolver->get_search_radius(), 5.0 * ModuleBase::ANGSTROM_AU + 0.01, doublethreshold); } TEST_F(LJ_pot_test, SetC6C12) @@ -199,7 +197,7 @@ TEST_F(LJ_pot_test, SetC6C12) std::vector lj_epsilon = {0.1, 0.2, 0.3}; std::vector lj_sigma = {0.2, 0.4, 0.6}; - p_esolver->set_c6_c12(ucell.ntype, rule, lj_epsilon, lj_sigma); + p_esolver->set_c6_c12_for_testing(ucell.ntype, rule, lj_epsilon, lj_sigma); for (int i = 0; i < ucell.ntype; i++) { @@ -207,10 +205,10 @@ TEST_F(LJ_pot_test, SetC6C12) { int k = i * (i + 1) / 2 + j; double temp = pow(lj_sigma[k] * ModuleBase::ANGSTROM_AU, 6); - EXPECT_NEAR(p_esolver->lj_c6(i, j), 4.0 * lj_epsilon[k] * temp / ModuleBase::Ry_to_eV, doublethreshold); - EXPECT_NEAR(p_esolver->lj_c12(i, j), p_esolver->lj_c6(i, j) * temp, doublethreshold); - EXPECT_DOUBLE_EQ(p_esolver->lj_c6(i, j), p_esolver->lj_c6(j, i)); - EXPECT_DOUBLE_EQ(p_esolver->lj_c12(i, j), p_esolver->lj_c12(j, i)); + EXPECT_NEAR(p_esolver->get_lj_c6()(i, j), 4.0 * lj_epsilon[k] * temp / ModuleBase::Ry_to_eV, doublethreshold); + EXPECT_NEAR(p_esolver->get_lj_c12()(i, j), p_esolver->get_lj_c6()(i, j) * temp, doublethreshold); + EXPECT_DOUBLE_EQ(p_esolver->get_lj_c6()(i, j), p_esolver->get_lj_c6()(j, i)); + EXPECT_DOUBLE_EQ(p_esolver->get_lj_c12()(i, j), p_esolver->get_lj_c12()(j, i)); } } @@ -219,24 +217,24 @@ TEST_F(LJ_pot_test, SetC6C12) lj_epsilon = {0.1, 0.2}; lj_sigma = {0.2, 0.4}; - p_esolver->set_c6_c12(ucell.ntype, rule, lj_epsilon, lj_sigma); + p_esolver->set_c6_c12_for_testing(ucell.ntype, rule, lj_epsilon, lj_sigma); for (int i = 0; i < ucell.ntype; i++) { double temp = pow(lj_sigma[i] * ModuleBase::ANGSTROM_AU, 6); - EXPECT_NEAR(p_esolver->lj_c6(i, i), 4.0 * lj_epsilon[i] * temp / ModuleBase::Ry_to_eV, doublethreshold); - EXPECT_NEAR(p_esolver->lj_c12(i, i), p_esolver->lj_c6(i, i) * temp, doublethreshold); + EXPECT_NEAR(p_esolver->get_lj_c6()(i, i), 4.0 * lj_epsilon[i] * temp / ModuleBase::Ry_to_eV, doublethreshold); + EXPECT_NEAR(p_esolver->get_lj_c12()(i, i), p_esolver->get_lj_c6()(i, i) * temp, doublethreshold); for (int j = 0; j < i; j++) { - EXPECT_NEAR(p_esolver->lj_c6(i, j), - std::sqrt(p_esolver->lj_c6(i, i) * p_esolver->lj_c6(j, j)), + EXPECT_NEAR(p_esolver->get_lj_c6()(i, j), + std::sqrt(p_esolver->get_lj_c6()(i, i) * p_esolver->get_lj_c6()(j, j)), doublethreshold); - EXPECT_NEAR(p_esolver->lj_c12(i, j), - std::sqrt(p_esolver->lj_c12(i, i) * p_esolver->lj_c12(j, j)), + EXPECT_NEAR(p_esolver->get_lj_c12()(i, j), + std::sqrt(p_esolver->get_lj_c12()(i, i) * p_esolver->get_lj_c12()(j, j)), doublethreshold); - EXPECT_DOUBLE_EQ(p_esolver->lj_c6(i, j), p_esolver->lj_c6(j, i)); - EXPECT_DOUBLE_EQ(p_esolver->lj_c12(i, j), p_esolver->lj_c12(j, i)); + EXPECT_DOUBLE_EQ(p_esolver->get_lj_c6()(i, j), p_esolver->get_lj_c6()(j, i)); + EXPECT_DOUBLE_EQ(p_esolver->get_lj_c12()(i, j), p_esolver->get_lj_c12()(j, i)); } } @@ -245,19 +243,19 @@ TEST_F(LJ_pot_test, SetC6C12) lj_epsilon = {0.1, 0.2}; lj_sigma = {0.2, 0.4}; - p_esolver->set_c6_c12(ucell.ntype, rule, lj_epsilon, lj_sigma); + p_esolver->set_c6_c12_for_testing(ucell.ntype, rule, lj_epsilon, lj_sigma); for (int i = 0; i < ucell.ntype; i++) { for (int j = 0; j <= i; j++) { double temp = pow((lj_sigma[i] + lj_sigma[j]) / 2 * ModuleBase::ANGSTROM_AU, 6); - EXPECT_NEAR(p_esolver->lj_c6(i, j), + EXPECT_NEAR(p_esolver->get_lj_c6()(i, j), 4.0 * std::sqrt(lj_epsilon[i] * lj_epsilon[j]) * temp / ModuleBase::Ry_to_eV, doublethreshold); - EXPECT_NEAR(p_esolver->lj_c12(i, j), p_esolver->lj_c6(i, j) * temp, doublethreshold); - EXPECT_DOUBLE_EQ(p_esolver->lj_c6(i, j), p_esolver->lj_c6(j, i)); - EXPECT_DOUBLE_EQ(p_esolver->lj_c12(i, j), p_esolver->lj_c12(j, i)); + EXPECT_NEAR(p_esolver->get_lj_c12()(i, j), p_esolver->get_lj_c6()(i, j) * temp, doublethreshold); + EXPECT_DOUBLE_EQ(p_esolver->get_lj_c6()(i, j), p_esolver->get_lj_c6()(j, i)); + EXPECT_DOUBLE_EQ(p_esolver->get_lj_c12()(i, j), p_esolver->get_lj_c12()(j, i)); } } } @@ -268,27 +266,27 @@ TEST_F(LJ_pot_test, CalEnShift) ucell.ntype = 2; std::vector rcut = {3.0}; - p_esolver->rcut_search_radius(ucell.ntype, rcut); + p_esolver->rcut_search_radius_for_testing(ucell.ntype, rcut); int rule = 1; std::vector lj_epsilon = {0.1, 0.2, 0.3}; std::vector lj_sigma = {0.2, 0.4, 0.6}; - p_esolver->set_c6_c12(ucell.ntype, rule, lj_epsilon, lj_sigma); + p_esolver->set_c6_c12_for_testing(ucell.ntype, rule, lj_epsilon, lj_sigma); // false - p_esolver->cal_en_shift(ucell.ntype, false); + p_esolver->cal_en_shift_for_testing(ucell.ntype, false); for (int i = 0; i < ucell.ntype; i++) { for (int j = 0; j < ucell.ntype; j++) { - EXPECT_DOUBLE_EQ(p_esolver->en_shift(i, j), 0.0); + EXPECT_DOUBLE_EQ(p_esolver->get_en_shift()(i, j), 0.0); } } // true - p_esolver->cal_en_shift(ucell.ntype, true); - EXPECT_NEAR(p_esolver->en_shift(0, 0), -2.5810212013100967e-09, doublethreshold); - EXPECT_NEAR(p_esolver->en_shift(0, 1), -3.303688865319793e-07, doublethreshold); - EXPECT_NEAR(p_esolver->en_shift(1, 0), -3.303688865319793e-07, doublethreshold); - EXPECT_NEAR(p_esolver->en_shift(1, 1), -5.6443326024140752e-06, doublethreshold); + p_esolver->cal_en_shift_for_testing(ucell.ntype, true); + EXPECT_NEAR(p_esolver->get_en_shift()(0, 0), -2.5810212013100967e-09, doublethreshold); + EXPECT_NEAR(p_esolver->get_en_shift()(0, 1), -3.303688865319793e-07, doublethreshold); + EXPECT_NEAR(p_esolver->get_en_shift()(1, 0), -3.303688865319793e-07, doublethreshold); + EXPECT_NEAR(p_esolver->get_en_shift()(1, 1), -5.6443326024140752e-06, doublethreshold); } diff --git a/source/source_md/test/md_func_test.cpp b/source/source_md/test/md_func_test.cpp index f1011aa51ac..7eeccd78691 100644 --- a/source/source_md/test/md_func_test.cpp +++ b/source/source_md/test/md_func_test.cpp @@ -1,11 +1,6 @@ -#include "source_cell/module_neighlist/domain_decomposition.h" +#include "source_cell/domain_decomposition.h" #include "gmock/gmock.h" #include "gtest/gtest.h" -#define private public -#include "source_io/module_parameter/parameter.h" -#undef private -#define private public -#define protected public #include "md_test_fixture.h" #include "source_md/md_func.h" @@ -166,7 +161,7 @@ TEST_F(MD_func_test, dump_info) { atom.vel = ModuleBase::Vector3(0.0, 0.0, 0.0); } - MD_func::dump_info(0, PARAM.sys.global_out_dir, mdcell, param_in, virial); + MD_func::dump_info(0, out_dir, mdcell, inp.mdp, inp.cal_stress, virial); std::ifstream ifs("MD_dump"); std::string output_str; getline(ifs, output_str); @@ -212,7 +207,7 @@ TEST_F(MD_func_test, dump_info) ifs.close(); // append - MD_func::dump_info(1, PARAM.sys.global_out_dir, mdcell, param_in, virial); + MD_func::dump_info(1, out_dir, mdcell, inp.mdp, inp.cal_stress, virial); std::ifstream ifs2("MD_dump"); getline(ifs2, output_str); EXPECT_THAT(output_str, testing::HasSubstr("MDSTEP: 0")); diff --git a/source/source_md/test/md_test_fixture.h b/source/source_md/test/md_test_fixture.h index a5b8034a726..853ccd1a3fc 100644 --- a/source/source_md/test/md_test_fixture.h +++ b/source/source_md/test/md_test_fixture.h @@ -3,28 +3,29 @@ #include "gtest/gtest.h" #include "source_esolver/esolver_lj.h" -#include "source_io/module_parameter/parameter.h" +#include "source_io/module_parameter/input_parameter.h" #include "source_md/md_base.h" -#include "source_cell/module_neighlist/domain_decomposition.h" +#include "source_cell/domain_decomposition.h" #include "setcell.h" #include +#include #include class MdTestBase : public testing::Test { protected: UnitCell ucell; - Parameter param_in; + Input_para inp; std::unique_ptr p_esolver; void SetUp() override { Setcell::setupcell(ucell); - Setcell::parameters(param_in.input); + Setcell::parameters(inp); p_esolver.reset(new ModuleESolver::ESolver_LJ()); - p_esolver->before_all_runners(ucell, param_in.inp); + p_esolver->before_all_runners(ucell, inp); } }; @@ -35,15 +36,17 @@ class MdIntegratorFixture : public MdTestBase MDCell mdcell; DomainDecomposition decomp; std::unique_ptr mdrun; + /// setup() takes the directory explicitly, so the fixture owns it + const std::string readin_dir = "./"; void SetUp() override { MdTestBase::SetUp(); mdcell = Setcell::setup_mdcell(ucell); decomp.init(ModuleBase::world_comm_domain(), mdcell.latvec(), mdcell.lat0(), 0.0, 0.0); - p_esolver->before_all_runners(mdcell, param_in.inp); - mdrun.reset(new Integrator(param_in, mdcell)); - mdrun->setup(p_esolver.get(), param_in.globalv.global_readin_dir, decomp); + p_esolver->before_all_runners(mdcell, inp); + mdrun.reset(new Integrator(this->inp.mdp, this->inp.cal_stress, this->inp.init_vel, /*my_rank=*/0, mdcell)); + mdrun->setup(p_esolver.get(), readin_dir, decomp); } }; @@ -67,12 +70,14 @@ class MdFuncTestFixture : public testing::Test int natom = 0; double temperature = 0.0; int frozen_freedom = 0; - Parameter param_in; + Input_para inp; + /// the md writers take the directory explicitly, so the fixture owns it + const std::string out_dir = "./"; void SetUp() override { Setcell::setupcell(ucell); - Setcell::parameters(param_in.input); + Setcell::parameters(inp); natom = ucell.nat; allmass_store.resize(natom); diff --git a/source/source_md/test/msst_test.cpp b/source/source_md/test/msst_test.cpp index dcefcada9d9..6a7e7cd3d29 100644 --- a/source/source_md/test/msst_test.cpp +++ b/source/source_md/test/msst_test.cpp @@ -1,11 +1,6 @@ -#include "source_cell/module_neighlist/domain_decomposition.h" +#include "source_cell/domain_decomposition.h" #include "gmock/gmock.h" #include "gtest/gtest.h" -#define private public -#include "source_io/module_parameter/parameter.h" -#undef private -#define private public -#define protected public #include "setcell.h" #include "source_esolver/esolver_lj.h" #include "source_md/msst.h" @@ -43,20 +38,23 @@ class MSST_test : public testing::Test UnitCell ucell; MDCell mdcell; DomainDecomposition decomp; - Parameter param_in; + Input_para inp; ModuleESolver::ESolver* p_esolver; + /// the md writers take the directory explicitly, so the test owns it + const std::string out_dir = "./"; + const std::string readin_dir = "./"; void SetUp() { Setcell::setupcell(ucell); - Setcell::parameters(param_in.input); + Setcell::parameters(inp); p_esolver = new ModuleESolver::ESolver_LJ(); mdcell = Setcell::setup_mdcell(ucell); decomp.init(ModuleBase::world_comm_domain(), mdcell.latvec(), mdcell.lat0(), 0.0, 0.0); - p_esolver->before_all_runners(mdcell, param_in.inp); - mdrun = new MSST(param_in, mdcell); - mdrun->setup(p_esolver, PARAM.sys.global_readin_dir, decomp); + p_esolver->before_all_runners(mdcell, inp); + mdrun = new MSST(inp.mdp, inp.cal_stress, inp.init_vel, /*my_rank=*/0, mdcell); + mdrun->setup(p_esolver, readin_dir, decomp); } void TearDown() @@ -186,7 +184,7 @@ TEST_F(MSST_test, write_restart) { mdrun->step_ = 1; mdrun->step_rst_ = 2; - mdrun->write_restart(PARAM.sys.global_out_dir); + mdrun->write_restart(out_dir); std::ifstream ifs("Restart_md.txt"); std::string output_str; @@ -209,16 +207,16 @@ TEST_F(MSST_test, write_restart) TEST_F(MSST_test, restart) { - mdrun->restart(PARAM.sys.global_readin_dir); + mdrun->restart(readin_dir); remove("Restart_md.txt"); MSST* msst = dynamic_cast(mdrun); EXPECT_EQ(mdrun->step_rst_, 3); - EXPECT_EQ(msst->omega[mdrun->mdp.msst_direction], -0.00977662); - EXPECT_EQ(msst->e0, -0.00768262); - EXPECT_EQ(msst->v0, 1000); - EXPECT_EQ(msst->p0, 1.60606e-06); - EXPECT_EQ(msst->lag_pos, 0); + EXPECT_EQ(msst->get_omega()[inp.mdp.msst_direction], -0.00977662); + EXPECT_EQ(msst->get_e0(), -0.00768262); + EXPECT_EQ(msst->get_v0(), 1000); + EXPECT_EQ(msst->get_p0(), 1.60606e-06); + EXPECT_EQ(msst->get_lag_pos(), 0); } TEST_F(MSST_test, print_md) @@ -254,7 +252,7 @@ TEST_F(MSST_test, print_md) EXPECT_THAT( output_str, testing::HasSubstr( - " -0.01545074 -0.023915637 0.0084648976 297 1.0762537 ")); + " -0.01545073962 -0.02391563725 0.008464897624 297 1.0762537 ")); getline(ifs, output_str); EXPECT_THAT( output_str, diff --git a/source/source_md/test/nhchain_test.cpp b/source/source_md/test/nhchain_test.cpp index 794927abb40..a73b9811c41 100644 --- a/source/source_md/test/nhchain_test.cpp +++ b/source/source_md/test/nhchain_test.cpp @@ -1,11 +1,6 @@ -#include "source_cell/module_neighlist/domain_decomposition.h" +#include "source_cell/domain_decomposition.h" #include "gmock/gmock.h" #include "gtest/gtest.h" -#define private public -#include "source_io/module_parameter/parameter.h" -#undef private -#define private public -#define protected public #include "setcell.h" #include "source_esolver/esolver_lj.h" #include "source_md/nhchain.h" @@ -41,24 +36,27 @@ class NHC_test : public testing::Test UnitCell ucell; MDCell mdcell; DomainDecomposition decomp; - Parameter param_in; + Input_para inp; ModuleESolver::ESolver* p_esolver; + /// the md writers take the directory explicitly, so the test owns it + const std::string out_dir = "./"; + const std::string readin_dir = "./"; void SetUp() { Setcell::setupcell(ucell); - Setcell::parameters(param_in.input); + Setcell::parameters(inp); p_esolver = new ModuleESolver::ESolver_LJ(); - param_in.input.mdp.md_type = "npt"; - param_in.input.mdp.md_pmode = "tri"; - param_in.input.mdp.md_pfirst = 1; - param_in.input.mdp.md_plast = 1; + inp.mdp.md_type = "npt"; + inp.mdp.md_pmode = "tri"; + inp.mdp.md_pfirst = 1; + inp.mdp.md_plast = 1; mdcell = Setcell::setup_mdcell(ucell); decomp.init(ModuleBase::world_comm_domain(), mdcell.latvec(), mdcell.lat0(), 0.0, 0.0); - p_esolver->before_all_runners(mdcell, param_in.inp); - mdrun = new Nose_Hoover(param_in, mdcell); - mdrun->setup(p_esolver, PARAM.sys.global_readin_dir, decomp); + p_esolver->before_all_runners(mdcell, inp); + mdrun = new Nose_Hoover(inp.mdp, inp.cal_stress, inp.init_vel, /*my_rank=*/0, mdcell); + mdrun->setup(p_esolver, readin_dir, decomp); } void TearDown() @@ -152,7 +150,7 @@ TEST_F(NHC_test, write_restart) mdrun->step_ = 1; mdrun->step_rst_ = 2; - mdrun->write_restart(PARAM.sys.global_out_dir); + mdrun->write_restart(out_dir); std::ifstream ifs("Restart_md.txt"); std::string output_str; @@ -180,35 +178,35 @@ TEST_F(NHC_test, write_restart) TEST_F(NHC_test, restart) { - mdrun->restart(PARAM.sys.global_readin_dir); + mdrun->restart(readin_dir); remove("Restart_md.txt"); Nose_Hoover* nhc = dynamic_cast(mdrun); EXPECT_EQ(mdrun->step_rst_, 3); - EXPECT_EQ(mdrun->mdp.md_tchain, 4); - EXPECT_EQ(mdrun->mdp.md_pchain, 4); - EXPECT_EQ(nhc->eta[0], -0.0626326); - EXPECT_EQ(nhc->eta[1], -0.578523); - EXPECT_EQ(nhc->eta[2], -0.462472); - EXPECT_EQ(nhc->eta[3], -0.424503); - EXPECT_EQ(nhc->v_eta[0], -0.00658882); - EXPECT_EQ(nhc->v_eta[1], -0.0304055); - EXPECT_EQ(nhc->v_eta[2], -0.0188618); - EXPECT_EQ(nhc->v_eta[3], -0.0175663); - EXPECT_EQ(nhc->v_omega[0], 0.583152); - EXPECT_EQ(nhc->v_omega[1], -0.106519); - EXPECT_EQ(nhc->v_omega[2], -0.895936); - EXPECT_EQ(nhc->v_omega[3], -0.634424); - EXPECT_EQ(nhc->v_omega[4], 0.627532); - EXPECT_EQ(nhc->v_omega[5], -0.473422); - EXPECT_EQ(nhc->peta[0], -6.08823); - EXPECT_EQ(nhc->peta[1], -0.525329); - EXPECT_EQ(nhc->peta[2], 0.121814); - EXPECT_EQ(nhc->peta[3], 4771.79); - EXPECT_EQ(nhc->v_peta[0], 255.853); - EXPECT_EQ(nhc->v_peta[1], -0.266732); - EXPECT_EQ(nhc->v_peta[2], 0); - EXPECT_EQ(nhc->v_peta[3], 226.197); + EXPECT_EQ(inp.mdp.md_tchain, 4); + EXPECT_EQ(inp.mdp.md_pchain, 4); + EXPECT_EQ(nhc->get_eta()[0], -0.0626326); + EXPECT_EQ(nhc->get_eta()[1], -0.578523); + EXPECT_EQ(nhc->get_eta()[2], -0.462472); + EXPECT_EQ(nhc->get_eta()[3], -0.424503); + EXPECT_EQ(nhc->get_v_eta()[0], -0.00658882); + EXPECT_EQ(nhc->get_v_eta()[1], -0.0304055); + EXPECT_EQ(nhc->get_v_eta()[2], -0.0188618); + EXPECT_EQ(nhc->get_v_eta()[3], -0.0175663); + EXPECT_EQ(nhc->get_v_omega()[0], 0.583152); + EXPECT_EQ(nhc->get_v_omega()[1], -0.106519); + EXPECT_EQ(nhc->get_v_omega()[2], -0.895936); + EXPECT_EQ(nhc->get_v_omega()[3], -0.634424); + EXPECT_EQ(nhc->get_v_omega()[4], 0.627532); + EXPECT_EQ(nhc->get_v_omega()[5], -0.473422); + EXPECT_EQ(nhc->get_peta()[0], -6.08823); + EXPECT_EQ(nhc->get_peta()[1], -0.525329); + EXPECT_EQ(nhc->get_peta()[2], 0.121814); + EXPECT_EQ(nhc->get_peta()[3], 4771.79); + EXPECT_EQ(nhc->get_v_peta()[0], 255.853); + EXPECT_EQ(nhc->get_v_peta()[1], -0.266732); + EXPECT_EQ(nhc->get_v_peta()[2], 0); + EXPECT_EQ(nhc->get_v_peta()[3], 226.197); } TEST_F(NHC_test, print_md) @@ -240,7 +238,7 @@ TEST_F(NHC_test, print_md) EXPECT_THAT( output_str, testing::HasSubstr( - " -0.015365236 -0.023915637 0.0085504016 300 1.0846391 ")); + " -0.01536523561 -0.02391563725 0.008550401641 300 1.0846391 ")); getline(ifs, output_str); EXPECT_THAT( output_str, diff --git a/source/source_md/test/run_md_test.cpp b/source/source_md/test/run_md_test.cpp index 6ee7d102a3d..9c459d3e886 100644 --- a/source/source_md/test/run_md_test.cpp +++ b/source/source_md/test/run_md_test.cpp @@ -1,4 +1,4 @@ -#include "source_cell/module_neighlist/domain_decomposition.h" +#include "source_cell/domain_decomposition.h" #include "gtest/gtest.h" #include "source_cell/mdcell.h" diff --git a/source/source_md/test/setcell.h b/source/source_md/test/setcell.h index dccb8ce1edc..0cd0809c1ab 100644 --- a/source/source_md/test/setcell.h +++ b/source/source_md/test/setcell.h @@ -10,7 +10,7 @@ #include "source_cell/unitcell.h" #include "source_base/constants.h" #include "source_base/parallel_cell.h" -#include "source_io/module_parameter/parameter.h" +#include "source_io/module_parameter/input_parameter.h" #include #include @@ -98,13 +98,16 @@ class Setcell ucell.set_iat2itia(); }; + /// @brief fill an Input_para with the settings the md tests share. + /// Only the passed-in object is touched: no source compiled by any + /// MODULE_MD target reads esolver_type, search_radius or cal_stress + /// from the global parameter singleton, and the output/readin + /// directories are passed explicitly to setup(), write_restart() + /// and restart(). static void parameters(Input_para& input) { - PARAM.sys.global_out_dir = "./"; - PARAM.sys.global_readin_dir = "./"; - PARAM.input.search_radius = 8.5 * ModuleBase::ANGSTROM_AU; - PARAM.input.cal_stress = true; - + input.search_radius = 8.5 * ModuleBase::ANGSTROM_AU; + input.esolver_type = "lj"; input.mdp.dump_virial = true; input.mdp.dump_force = true; @@ -116,7 +119,6 @@ class Setcell input.mdp.md_dt = 1; input.mdp.md_tfirst = input.mdp.md_tlast = 300; - PARAM.input.esolver_type = "lj"; input.mdp.lj_rcut = {8.5}; input.mdp.lj_epsilon = {0.01032}; input.mdp.lj_sigma = {3.405}; diff --git a/source/source_md/test/test_md_func.cpp b/source/source_md/test/test_md_func.cpp index 31c34bc3bdd..9e0aef2f6ee 100644 --- a/source/source_md/test/test_md_func.cpp +++ b/source/source_md/test/test_md_func.cpp @@ -1,6 +1,6 @@ #include "source_md/md_func.h" #include "source_cell/mdcell.h" -#include "source_cell/module_neighlist/domain_decomposition.h" +#include "source_cell/domain_decomposition.h" #include "source_base/parallel_cell.h" #include "source_esolver/esolver.h" diff --git a/source/source_md/test/verlet_test.cpp b/source/source_md/test/verlet_test.cpp index b2509c5e2c1..8d960f04d38 100644 --- a/source/source_md/test/verlet_test.cpp +++ b/source/source_md/test/verlet_test.cpp @@ -1,11 +1,6 @@ -#include "source_cell/module_neighlist/domain_decomposition.h" +#include "source_cell/domain_decomposition.h" #include "gmock/gmock.h" #include "gtest/gtest.h" -#define private public -#include "source_io/module_parameter/parameter.h" -#undef private -#define private public -#define protected public #include "setcell.h" #include "source_esolver/esolver_lj.h" #include "source_md/verlet.h" @@ -45,20 +40,23 @@ class Verlet_test : public testing::Test UnitCell ucell; MDCell mdcell; DomainDecomposition decomp; - Parameter param_in; + Input_para inp; ModuleESolver::ESolver* p_esolver; + /// the md writers take the directory explicitly, so the test owns it + const std::string out_dir = "./"; + const std::string readin_dir = "./"; void SetUp() { Setcell::setupcell(ucell); - Setcell::parameters(param_in.input); + Setcell::parameters(inp); p_esolver = new ModuleESolver::ESolver_LJ(); mdcell = Setcell::setup_mdcell(ucell); decomp.init(ModuleBase::world_comm_domain(), mdcell.latvec(), mdcell.lat0(), 0.0, 0.0); - p_esolver->before_all_runners(mdcell, param_in.inp); - mdrun = new Verlet(param_in, mdcell); - mdrun->setup(p_esolver, PARAM.sys.global_readin_dir, decomp); + p_esolver->before_all_runners(mdcell, inp); + mdrun = new Verlet(inp.mdp, inp.cal_stress, inp.init_vel, /*my_rank=*/0, mdcell); + mdrun->setup(p_esolver, readin_dir, decomp); } void TearDown() @@ -116,7 +114,7 @@ TEST_F(Verlet_test, first_half) TEST_F(Verlet_test, NVE) { mdrun->first_half(GlobalV::ofs_running); - param_in.input.mdp.md_type = "nve"; + inp.mdp.md_type = "nve"; mdrun->second_half(); ; @@ -150,8 +148,8 @@ TEST_F(Verlet_test, NVE) TEST_F(Verlet_test, Anderson) { mdrun->first_half(GlobalV::ofs_running); - param_in.input.mdp.md_type = "nvt"; - param_in.input.mdp.md_thermostat = "anderson"; + inp.mdp.md_type = "nvt"; + inp.mdp.md_thermostat = "anderson"; mdrun->second_half(); ; @@ -185,8 +183,8 @@ TEST_F(Verlet_test, Anderson) TEST_F(Verlet_test, Berendsen) { mdrun->first_half(GlobalV::ofs_running); - param_in.input.mdp.md_type = "nvt"; - param_in.input.mdp.md_thermostat = "berendsen"; + inp.mdp.md_type = "nvt"; + inp.mdp.md_thermostat = "berendsen"; mdrun->second_half(); ; @@ -220,8 +218,8 @@ TEST_F(Verlet_test, Berendsen) TEST_F(Verlet_test, rescaling) { mdrun->first_half(GlobalV::ofs_running); - param_in.input.mdp.md_type = "nvt"; - param_in.input.mdp.md_thermostat = "rescaling"; + inp.mdp.md_type = "nvt"; + inp.mdp.md_thermostat = "rescaling"; mdrun->second_half(); ; @@ -255,8 +253,8 @@ TEST_F(Verlet_test, rescaling) TEST_F(Verlet_test, rescale_v) { mdrun->first_half(GlobalV::ofs_running); - param_in.input.mdp.md_type = "nvt"; - param_in.input.mdp.md_thermostat = "rescale_v"; + inp.mdp.md_type = "nvt"; + inp.mdp.md_thermostat = "rescale_v"; mdrun->second_half(); ; @@ -291,10 +289,10 @@ TEST_F(Verlet_test, CSVR) { std::ofstream ofs; mdrun->first_half(ofs); - param_in.input.mdp.md_type = "nvt"; - param_in.input.mdp.md_thermostat = "csvr"; - param_in.input.mdp.md_csvr_tau = 100.0; - param_in.input.mdp.md_seed = 12345; + inp.mdp.md_type = "nvt"; + inp.mdp.md_thermostat = "csvr"; + inp.mdp.md_csvr_tau = 100.0; + inp.mdp.md_seed = 12345; mdrun->second_half(); // Check that positions are updated correctly @@ -312,7 +310,7 @@ TEST_F(Verlet_test, write_restart) { mdrun->step_ = 1; mdrun->step_rst_ = 2; - mdrun->write_restart(PARAM.sys.global_out_dir); + mdrun->write_restart(out_dir); std::ifstream ifs("Restart_md.txt"); std::string output_str; @@ -323,7 +321,7 @@ TEST_F(Verlet_test, write_restart) TEST_F(Verlet_test, restart) { - mdrun->restart(PARAM.sys.global_readin_dir); + mdrun->restart(readin_dir); remove("Restart_md.txt"); EXPECT_EQ(mdrun->step_rst_, 3); @@ -358,7 +356,7 @@ TEST_F(Verlet_test, print_md) EXPECT_THAT( output_str, testing::HasSubstr( - " -0.015365236 -0.023915637 0.0085504016 300 1.0846391 ")); + " -0.01536523561 -0.02391563725 0.008550401641 300 1.0846391 ")); getline(ifs, output_str); EXPECT_THAT( output_str, diff --git a/source/source_md/verlet.cpp b/source/source_md/verlet.cpp index dc1071ef7af..05c9c0f38ae 100644 --- a/source/source_md/verlet.cpp +++ b/source/source_md/verlet.cpp @@ -7,7 +7,12 @@ #include #endif -Verlet::Verlet(const Parameter& param_in, MDCell& mdcell_in) : MD_base(param_in, mdcell_in) +Verlet::Verlet(const MD_para& mdp_in, + const bool cal_stress_in, + const bool init_vel, + const int my_rank_in, + MDCell& mdcell_in) + : MD_base(mdp_in, cal_stress_in, init_vel, my_rank_in, mdcell_in) { } diff --git a/source/source_md/verlet.h b/source/source_md/verlet.h index 220b0190777..670688251ef 100644 --- a/source/source_md/verlet.h +++ b/source/source_md/verlet.h @@ -10,7 +10,11 @@ class Verlet : public MD_base { public: - Verlet(const Parameter& param_in, MDCell& mdcell_in); + Verlet(const MD_para& mdp_in, + const bool cal_stress_in, + const bool init_vel, + const int my_rank_in, + MDCell& mdcell_in); ~Verlet(); private: diff --git a/source/source_psi/psi_init_file.cpp b/source/source_psi/psi_init_file.cpp index 40478b5ffef..29a8619e30b 100644 --- a/source/source_psi/psi_init_file.cpp +++ b/source/source_psi/psi_init_file.cpp @@ -57,13 +57,13 @@ void psi_init_file::init_psig(T* psig, const int& ik) const int istep = -1; std::string fn = ModuleIO::filename_output(this->global_readin_dir_,"wf","pw", - ik,this->ik2iktot_,this->nspin_,nkstot, - out_type,out_app_flag,gamma_only,istep); + ik,this->ik2iktot_,this->nspin_,nkstot, + out_type,out_app_flag,gamma_only,istep); ModuleIO::read_wfc_pw(fn, this->pw_wfc_, - this->rank_in_pool_, this->nproc_in_pool_, - this->nbands_start_, this->npol_, - ik, ik_tot, nkstot, wfcatom); + this->rank_in_pool_, this->nproc_in_pool_, + this->nbands_start_, this->npol_, + ik, ik_tot, nkstot, wfcatom); assert(this->nbands_start_ <= wfcatom.nr); for (int ib = 0; ib < this->nbands_start_; ib++) diff --git a/source/source_psi/psi_prepare.cpp b/source/source_psi/psi_prepare.cpp index a0664292eed..353e58adce7 100644 --- a/source/source_psi/psi_prepare.cpp +++ b/source/source_psi/psi_prepare.cpp @@ -11,6 +11,7 @@ #include "source_basis/module_pw/pw_basis_k.h" #include "source_cell/unitcell.h" #include "source_hsolver/diag_comm_info.h" +#include "source_hamilt/hamilt_hs_adapter.h" #include "source_hsolver/diago_iter_assist.h" #include "source_io/module_parameter/parameter.h" #include "source_psi/psi_init_atomic.h" @@ -221,6 +222,9 @@ void PSIPrepare::initialize_psi(Psi>* psi, } } + // the subspace diagonalization sees the Hamiltonian only through this operator + hamilt::HamiltHSOperator op(p_hamilt, &this->pw_wfc); + // loop over kpoints, make it possible to only allocate memory for psig at the only one kpt // like (1, nbands, npwx), in which npwx is the maximal npw of all kpoints for (int ik = 0; ik < this->pw_wfc.nks; ik++) @@ -231,7 +235,7 @@ void PSIPrepare::initialize_psi(Psi>* psi, kspw_psi->fix_k(ik); //! Update Hamiltonian from other kpoint to the given one - p_hamilt->updateHk(ik); + op.update_k(ik); if (fill) { //! initialize psi_cpu @@ -249,7 +253,7 @@ void PSIPrepare::initialize_psi(Psi>* psi, { // for diagH_subspace_init, psi_device->get_pointer() and kspw_psi->get_pointer() should be // different - hsolver::DiagoIterAssist::diag_subspace_init(p_hamilt, + hsolver::DiagoIterAssist::diag_subspace_init(op, psi_device->get_pointer(), nbands_start, nbasis, @@ -262,7 +266,7 @@ void PSIPrepare::initialize_psi(Psi>* psi, else { // for diagH_subspace, psi_device->get_pointer() and kspw_psi->get_pointer() can be the same - hsolver::DiagoIterAssist::diag_subspace(p_hamilt, + hsolver::DiagoIterAssist::diag_subspace(op, *psi_device, *kspw_psi, etatom.data(), diff --git a/source/source_psi/setup_psi_pw.cpp b/source/source_psi/setup_psi_pw.cpp index eec367ad0f0..795222e967b 100644 --- a/source/source_psi/setup_psi_pw.cpp +++ b/source/source_psi/setup_psi_pw.cpp @@ -231,6 +231,18 @@ void Setup_Psi_pw::copy_d2h() #endif } +void Setup_Psi_pw::sync_cpu() +{ + if (this->device_type_ == base_device::GpuDevice) + { + this->copy_d2h(); + } + else if (this->precision_type_ == PrecisionType::ComplexFloat) + { + this->copy_d2h_impl, base_device::DEVICE_CPU>(); + } +} + template void Setup_Psi_pw::castmem_d2h_impl(std::complex* dst, const std::complex* src, const std::size_t size) { diff --git a/source/source_psi/setup_psi_pw.h b/source/source_psi/setup_psi_pw.h index 55cb8ef703f..a435ea0e040 100644 --- a/source/source_psi/setup_psi_pw.h +++ b/source/source_psi/setup_psi_pw.h @@ -60,6 +60,9 @@ class Setup_Psi_pw // Transfer data from device to host in pw basis void copy_d2h(); + /** @brief Refresh the double CPU mirror from the native device and precision. */ + void sync_cpu(); + void clean(); //------------ diff --git a/source/source_psi/test/psi_init_unit_test.cpp b/source/source_psi/test/psi_init_unit_test.cpp index 4e2dd16d740..a541bc1bb5d 100644 --- a/source/source_psi/test/psi_init_unit_test.cpp +++ b/source/source_psi/test/psi_init_unit_test.cpp @@ -82,7 +82,7 @@ ORB_gaunt_table::~ORB_gaunt_table() {} Structure_Factor::Structure_Factor() {} Structure_Factor::~Structure_Factor() {} -void Structure_Factor::setup(const UnitCell* Ucell, const Parallel_Grid&, const ModulePW::PW_Basis* rho_basis) {} +void Structure_Factor::setup(const UnitCell* Ucell, const Parallel_Grid&, const ModulePW::PW_Basis* rho_basis, const bool) {} std::complex* Structure_Factor::get_sk(int ik, int it, int ia, ModulePW::PW_Basis_K const*wfc_basis) const { int npw = wfc_basis->npwk[ik]; diff --git a/source/source_pw/CMakeLists.txt b/source/source_pw/CMakeLists.txt index ee58a96433f..0efea3ccb93 100644 --- a/source/source_pw/CMakeLists.txt +++ b/source/source_pw/CMakeLists.txt @@ -1,3 +1,4 @@ +add_subdirectory(module_proj) add_subdirectory(module_pwdft) add_subdirectory(module_ofdft) add_subdirectory(module_stodft) diff --git a/source/source_pw/module_dfpt/dfpt_pw_init.cpp b/source/source_pw/module_dfpt/dfpt_pw_init.cpp index 0b175c0075a..2b702514f6d 100644 --- a/source/source_pw/module_dfpt/dfpt_pw_init.cpp +++ b/source/source_pw/module_dfpt/dfpt_pw_init.cpp @@ -257,9 +257,12 @@ void DFPT_PW::Impl::copy_occ_state_ball(int ik, for (int igl = 0; igl < npw_kq; ++igl) { const ModuleBase::Vector3 gf = kq.get_gcar(igl) * ginv; - const std::vector key = {static_cast(std::round(gf.x)) + dn.x, - static_cast(std::round(gf.y)) + dn.y, - static_cast(std::round(gf.z)) + dn.z}; + // dn = k_d(ikq) - k_d(ik) - q (integer, reciprocal-basis coords). + // A k+q-ball vector G and an ikq-ball vector G' describe the same + // plane wave when G' + k_d(ikq) == G + k_d(ik) + q, i.e. G' = G - dn. + const std::vector key = {static_cast(std::round(gf.x)) - dn.x, + static_cast(std::round(gf.y)) - dn.y, + static_cast(std::round(gf.z)) - dn.z}; const auto it = jgl_of_n.find(key); if (it != jgl_of_n.end()) { diff --git a/source/source_pw/module_dfpt/test/dfpt_pw_run_test.cpp b/source/source_pw/module_dfpt/test/dfpt_pw_run_test.cpp index 73ed2219c4e..881eca4d94f 100644 --- a/source/source_pw/module_dfpt/test/dfpt_pw_run_test.cpp +++ b/source/source_pw/module_dfpt/test/dfpt_pw_run_test.cpp @@ -11,7 +11,7 @@ #include "dfpt_stru_fixture.h" #include "source_base/global_variable.h" #include "source_base/parallel_global.h" -#include "source_estate/module_charge/charge_mixing.h" +#include "source_estate/module_charge/chg_mix.h" #include "source_pw/module_dfpt/dfpt_pw.h" #include "source_pw/module_pwdft/dftu_base.h" diff --git a/source/source_pw/module_dfpt/test/dfpt_test_mocks.cpp b/source/source_pw/module_dfpt/test/dfpt_test_mocks.cpp index 48ff22ad59e..e4b45c89d1d 100644 --- a/source/source_pw/module_dfpt/test/dfpt_test_mocks.cpp +++ b/source/source_pw/module_dfpt/test/dfpt_test_mocks.cpp @@ -19,7 +19,7 @@ #include "source_cell/magnetism.h" #include "source_cell/pseudo.h" #include "source_cell/unitcell.h" -#include "source_estate/module_charge/charge_mixing.h" +#include "source_estate/module_charge/chg_mix.h" #include "source_pw/module_pwdft/dftu_base.h" #include "source_pw/module_pwdft/stru_fac.h" diff --git a/source/source_pw/module_ofdft/evolve_ofdft.cpp b/source/source_pw/module_ofdft/evolve_ofdft.cpp index 7f276781ddb..3e89a117d9c 100644 --- a/source/source_pw/module_ofdft/evolve_ofdft.cpp +++ b/source/source_pw/module_ofdft/evolve_ofdft.cpp @@ -32,7 +32,7 @@ void Evolve_OFDFT::cal_Hpsi(elecstate::ElecState* pelec, chr.rho[is][ir] = std::norm(psi_[is * nrxx + ir]); } } - this->renormalize_psi(chr, pw_rho, psi_); + this->renormalize_psi(chr, pw_rho, psi_, ucell.omega); pelec->pot->update_from_charge(&chr, &ucell); // Hartree + XC + external this->cal_tf_potential(chr.rho, pw_rho, pelec->pot->get_eff_v()); // TF potential @@ -55,9 +55,9 @@ void Evolve_OFDFT::cal_Hpsi(elecstate::ElecState* pelec, this->cal_vw_potential_phi(psi_, pw_rho, Hpsi); } -void Evolve_OFDFT::renormalize_psi(Charge& chr, ModulePW::PW_Basis* pw_rho, std::vector>& pphi_) +void Evolve_OFDFT::renormalize_psi(Charge& chr, ModulePW::PW_Basis* pw_rho, std::vector>& pphi_, const double omega) { - const double sr = chr.sum_rho(); + const double sr = chr.sum_rho(omega); const double normalize_factor = PARAM.inp.nelec / sr; const int nspin = PARAM.inp.nspin; const int nrxx = pw_rho->nrxx; @@ -311,7 +311,7 @@ void Evolve_OFDFT::propagate_psi_RK4(elecstate::ElecState* pelec, chr.rho[is][ir] = abs(pphi_[is * nrxx + ir])*abs(pphi_[is * nrxx + ir]); } } - this->renormalize_psi(chr, pw_rho, pphi_); + this->renormalize_psi(chr, pw_rho, pphi_, ucell.omega); ModuleBase::timer::end("ESolver_OF_TDDFT", "propagate_psi_RK4"); } @@ -369,7 +369,7 @@ void Evolve_OFDFT::propagate_psi_RK2(elecstate::ElecState* pelec, } } - this->renormalize_psi(chr, pw_rho, pphi_); + this->renormalize_psi(chr, pw_rho, pphi_, ucell.omega); ModuleBase::timer::end("ESolver_OF_TDDFT", "propagate_psi_RK2"); } diff --git a/source/source_pw/module_ofdft/evolve_ofdft.h b/source/source_pw/module_ofdft/evolve_ofdft.h index 63b6b4f86f7..da4df0bf4ee 100644 --- a/source/source_pw/module_ofdft/evolve_ofdft.h +++ b/source/source_pw/module_ofdft/evolve_ofdft.h @@ -35,7 +35,7 @@ class Evolve_OFDFT std::vector>& pphi_, ModulePW::PW_Basis* pw_rho); - void renormalize_psi(Charge& chr, ModulePW::PW_Basis* pw_rho, std::vector>& pphi_); + void renormalize_psi(Charge& chr, ModulePW::PW_Basis* pw_rho, std::vector>& pphi_, const double omega); private: const double c_tf_ diff --git a/source/source_pw/module_ofdft/kedf_manager.cpp b/source/source_pw/module_ofdft/kedf_manager.cpp index 96a8be470e6..953f406698b 100644 --- a/source/source_pw/module_ofdft/kedf_manager.cpp +++ b/source/source_pw/module_ofdft/kedf_manager.cpp @@ -292,7 +292,7 @@ double KEDF_Manager::get_energy() const { GlobalV::ofs_running << " WARNING: ML >= TF" << std::endl; GlobalV::ofs_running << " ML Term = " << this->ml_->ml_energy - << " Ry, TF Term = " << this->tf_->tf_energy << " Ry." << std::endl; + << " Ry, TF Term = " << this->tf_->tf_energy << " Ry." << std::endl; } } #endif diff --git a/source/source_pw/module_ofdft/kedf_ml.cpp b/source/source_pw/module_ofdft/kedf_ml.cpp index 4c439b8f30f..4fd92a929e8 100644 --- a/source/source_pw/module_ofdft/kedf_ml.cpp +++ b/source/source_pw/module_ofdft/kedf_ml.cpp @@ -109,7 +109,7 @@ void KEDF_ML::set_para( } ofs_running << " feg_net_F = " << this->feg_net_F - << " (Pauli energy enhancement factor in free electron gas)" << std::endl << std::endl; + << " (Pauli energy enhancement factor in free electron gas)" << std::endl << std::endl; } } else @@ -129,8 +129,8 @@ void KEDF_ML::set_para( this->cal_tool->set_para(nx, nelec, tf_weight, vw_weight, chi_p, chi_q, chi_xi, chi_pnl, chi_qnl, nkernel, kernel_type, - kernel_scaling, yukawa_alpha, kernel_file, - this->dV * pw_rho->nxyz, pw_rho, ofs_running); + kernel_scaling, yukawa_alpha, kernel_file, + this->dV * pw_rho->nxyz, pw_rho, ofs_running); } else { diff --git a/source/source_pw/module_ofdft/kedf_ml.h b/source/source_pw/module_ofdft/kedf_ml.h index 9ac90a2bf64..ea3f7d4383e 100644 --- a/source/source_pw/module_ofdft/kedf_ml.h +++ b/source/source_pw/module_ofdft/kedf_ml.h @@ -13,7 +13,7 @@ class KEDF_ML : public ML_Base KEDF_ML() { this->energy_prefactor = 3. /10. * std::pow(3*std::pow(M_PI, 2.0), 2.0/3.0) * 2; - // 10/3*(3*pi^2)^{2/3}, multiply by 2 to convert unit from Hartree to Ry, finally in Ry*Bohr^(-2) + // 10/3*(3*pi^2)^{2/3}, multiply by 2 to convert unit from Hartree to Ry, finally in Ry*Bohr^(-2) this->energy_exponent = 5. / 3.; } diff --git a/source/source_pw/module_ofdft/ml_base.cpp b/source/source_pw/module_ofdft/ml_base.cpp index 2e04cb6231d..e209fda7800 100644 --- a/source/source_pw/module_ofdft/ml_base.cpp +++ b/source/source_pw/module_ofdft/ml_base.cpp @@ -27,7 +27,7 @@ void ML_Base::set_device(const std::string& device_inpt, std::ostream& ofs_runni } else { - std::cout << "--------------- Warning: GPU is unavailable ---------------" << std::endl; + std::cout << "--------------- Warning: GPU is unavailable ---------------" << std::endl; ofs_running << "--------------- Warning: GPU is unavailable ---------------" << std::endl; ofs_running << "------------------- Running Neural Network on CPU -------------------" << std::endl; @@ -264,9 +264,9 @@ void ML_Base::get_potential_(const double * const * prho, const ModulePW::PW_Bas pauli_potential[ir] += factor * (this->energy_exponent * this->enhancement_cpu_ptr[ir] - + this->pot_gamma_term(ir) + this->pot_p_term_1(ir) + this->pot_q_term_1(ir) + + this->pot_gamma_term(ir) + this->pot_p_term_1(ir) + this->pot_q_term_1(ir) + this->pot_xi_term_1(ir) + this->pot_tanhxi_term_1(ir) - + this->pot_tanhp_term_1(ir) + this->pot_tanhq_term_1(ir)); + + this->pot_tanhp_term_1(ir) + this->pot_tanhq_term_1(ir)); rpotential(0, ir) += pauli_potential[ir]; } diff --git a/source/source_pw/module_ofdft/of_print_info.cpp b/source/source_pw/module_ofdft/of_print_info.cpp index 1f3a14b7910..53a3addf769 100644 --- a/source/source_pw/module_ofdft/of_print_info.cpp +++ b/source/source_pw/module_ofdft/of_print_info.cpp @@ -8,13 +8,13 @@ * and write the components of the total energy into running_log. */ void OFDFT::print_info(const int iter, - ModuleBase::TimePoint &iter_time, - const double &energy_current, - const double &energy_last, - const double &normdLdphi, - const elecstate::ElecState *pelec, - KEDF_Manager *kedf_manager, - const bool conv_esolver) + ModuleBase::TimePoint &iter_time, + const double &energy_current, + const double &energy_last, + const double &normdLdphi, + const elecstate::ElecState *pelec, + KEDF_Manager *kedf_manager, + const bool conv_esolver) { if (iter == 0) { @@ -47,11 +47,11 @@ void OFDFT::print_info(const int iter, std::vector titles; std::vector energies_Ry; std::vector energies_eV; - if ((PARAM.inp.out_band[0] > 0 && - ((iter + 1) % PARAM.inp.out_band[0] == 0 || - conv_esolver || - iter == PARAM.inp.scf_nmax)) || - PARAM.inp.init_chg == "file") + if ((PARAM.inp.out_band[0] > 0 && + ((iter + 1) % PARAM.inp.out_band[0] == 0 || + conv_esolver || + iter == PARAM.inp.scf_nmax)) || + PARAM.inp.init_chg == "file") { titles.push_back("E_Total"); energies_Ry.push_back(pelec->f_en.etot); diff --git a/source/source_pw/module_ofdft/of_print_info.h b/source/source_pw/module_ofdft/of_print_info.h index dd45e6bbc6d..eaced920260 100644 --- a/source/source_pw/module_ofdft/of_print_info.h +++ b/source/source_pw/module_ofdft/of_print_info.h @@ -11,13 +11,13 @@ namespace OFDFT { void print_info(const int iter, - ModuleBase::TimePoint &iter_time, - const double &energy_current, - const double &energy_last, - const double &normdLdphi, - const elecstate::ElecState *pelec, - KEDF_Manager *kedf_manager, - const bool conv_esolver); + ModuleBase::TimePoint &iter_time, + const double &energy_current, + const double &energy_last, + const double &normdLdphi, + const elecstate::ElecState *pelec, + KEDF_Manager *kedf_manager, + const bool conv_esolver); } diff --git a/source/source_pw/module_ofdft/of_stress_pw.cpp b/source/source_pw/module_ofdft/of_stress_pw.cpp index 6b988baf786..20189bd6f62 100644 --- a/source/source_pw/module_ofdft/of_stress_pw.cpp +++ b/source/source_pw/module_ofdft/of_stress_pw.cpp @@ -7,7 +7,12 @@ // Since the kinetic stress of OFDFT is calculated by kinetic functionals in esolver_of.cpp, here we regard it as an // input variable. -void OF_Stress_PW::cal_stress(ModuleBase::matrix& sigmatot, +void OF_Stress_PW::cal_stress(const int nspin, + const bool domag, + const bool domag_z, + const int gga_grad, + const bool gamma_only_pw, + ModuleBase::matrix& sigmatot, ModuleBase::matrix& kinetic_stress, UnitCell& ucell, const vdw::VdwResult* vdw_result, @@ -74,13 +79,15 @@ void OF_Stress_PW::cal_stress(ModuleBase::matrix& sigmatot, { sigmaxc(i, i) = -(pelec->f_en.etxc - pelec->f_en.vtxc) / ucell.omega; } - stress_gga(ucell,sigmaxc, this->rhopw, pelec->charge); + stress_gga(ucell,sigmaxc, this->rhopw, pelec->charge, + nspin, domag, domag_z, gga_grad); // local contribution stress_loc(ucell,sigmaloc, this->rhopw, locpp.vloc, p_sf, true, pelec->charge); // nlcc - stress_cc(sigmaxcc, this->rhopw, ucell, p_sf, true, locpp.numeric, pelec->charge); + stress_cc(sigmaxcc, this->rhopw, ucell, p_sf, true, locpp.numeric, pelec->charge, + nspin, domag, domag_z, gga_grad, gamma_only_pw); // vdW term prepared before SCF for this ionic configuration. if (vdw_result != nullptr) diff --git a/source/source_pw/module_ofdft/of_stress_pw.h b/source/source_pw/module_ofdft/of_stress_pw.h index 17fc50f4775..edb62c9e04c 100644 --- a/source/source_pw/module_ofdft/of_stress_pw.h +++ b/source/source_pw/module_ofdft/of_stress_pw.h @@ -17,7 +17,12 @@ class OF_Stress_PW : public Stress_Func : pelec(pelec_in), rhopw(rhopw_in){}; // calculate the stress in OFDFT - void cal_stress(ModuleBase::matrix& sigmatot, + void cal_stress(const int nspin, + const bool domag, + const bool domag_z, + const int gga_grad, + const bool gamma_only_pw, + ModuleBase::matrix& sigmatot, ModuleBase::matrix& kinetic_stress, UnitCell& ucell, const vdw::VdwResult* vdw_result, diff --git a/source/source_pw/module_proj/CMakeLists.txt b/source/source_pw/module_proj/CMakeLists.txt new file mode 100644 index 00000000000..bd34f0ddb3f --- /dev/null +++ b/source/source_pw/module_proj/CMakeLists.txt @@ -0,0 +1,39 @@ +# module_proj: projection of plane-wave wavefunctions onto local (atomic) orbitals. +# +# This module hosts the onsite-projector code extracted from module_pwdft: +# radial projectors (SBT), onsite projector init/tabulate/overlap, and the +# Onsite_Proj_tools computation backend. +# +# Known temporary dependency (recorded during extraction): +# - Onsite_Proj_tools includes source_pw/module_pwdft/vnl_pw.h and +# source_pw/module_pwdft/kernels/{force_op,stress_op}.h, i.e. module_proj +# currently depends on module_pwdft headers. This is tolerated short-term; +# the planned cleanup is to split the pseudopotential-specific constructor +# of Onsite_Proj_tools so this reverse dependency can be removed. + +list(APPEND objects + radial_proj.cpp + onsite_proj_print.cpp + onsite_proj.cpp + onsite_proj_init.cpp + onsite_proj_overlap.cpp + onsite_proj_tools.cpp + onsite_proj_tools_becp.cpp + onsite_proj_tools_dbecp.cpp + onsite_proj_tools_force.cpp + onsite_proj_tools_stress.cpp +) + +add_library( + module_proj + OBJECT + ${objects} +) + +if(ENABLE_COVERAGE) + add_coverage(module_proj) +endif() + +if(BUILD_TESTING) + add_subdirectory(unittests) +endif() diff --git a/source/source_pw/module_pwdft/onsite_proj.cpp b/source/source_pw/module_proj/onsite_proj.cpp similarity index 95% rename from source/source_pw/module_pwdft/onsite_proj.cpp rename to source/source_pw/module_proj/onsite_proj.cpp index 2b40f78f970..9ce54853965 100644 --- a/source/source_pw/module_pwdft/onsite_proj.cpp +++ b/source/source_pw/module_proj/onsite_proj.cpp @@ -1,4 +1,4 @@ -#include "source_pw/module_pwdft/onsite_proj.h" +#include "source_pw/module_proj/onsite_proj.h" #include "source_base/kernels/math_kernel_op.h" template diff --git a/source/source_pw/module_pwdft/onsite_proj.h b/source/source_pw/module_proj/onsite_proj.h similarity index 93% rename from source/source_pw/module_pwdft/onsite_proj.h rename to source/source_pw/module_proj/onsite_proj.h index a4181e4b4bb..fcfe8553b14 100644 --- a/source/source_pw/module_pwdft/onsite_proj.h +++ b/source/source_pw/module_proj/onsite_proj.h @@ -4,9 +4,8 @@ #include "source_base/kernels/math_kernel_op.h" #include "source_pw/module_pwdft/stru_fac.h" #include "source_basis/module_pw/pw_basis_k.h" -#include "source_pw/module_pwdft/radial_proj.h" #include "source_psi/psi.h" -#include "source_pw/module_pwdft/onsite_proj_tools.h" +#include "source_pw/module_proj/onsite_proj_tools.h" #include "source_pw/module_pwdft/dftu_base.h" #include @@ -52,14 +51,6 @@ namespace projectors const std::complex* ppsi, const int ld_psi = 0 ); - void read_abacus_orb(std::ifstream& ifs, - std::string& elem, - double& ecut, - int& nr, - double& dr, - std::vector& nzeta, - std::vector>& radials, - const int rank = 0); /// @brief static access to this class instance static OnsiteProjector* get_instance(); void init(const std::string& orbital_dir, @@ -75,7 +66,9 @@ namespace projectors const ModuleBase::matrix& ekb); /// @brief calculate and print the occupations of all lm orbitals - void cal_occupations(const psi::Psi, Device>* psi, const ModuleBase::matrix& wg_in); + void cal_occupations(const psi::Psi, Device>* psi, + const ModuleBase::matrix& wg_in, + const int nspin_in); int get_size_becp() const { return size_becp; } std::complex* get_becp() const { return becp; } @@ -145,7 +138,6 @@ namespace projectors Structure_Factor* sf_ = nullptr; // level2: the structure factor calculator int ntype = 0; - RadialProjection::RadialProjector rp_; std::vector irow2it_; std::vector irow2iproj_; std::vector irow2m_; diff --git a/source/source_pw/module_proj/onsite_proj_init.cpp b/source/source_pw/module_proj/onsite_proj_init.cpp new file mode 100644 index 00000000000..76f0faa7af2 --- /dev/null +++ b/source/source_pw/module_proj/onsite_proj_init.cpp @@ -0,0 +1,302 @@ +#include +#include +#include +#include +#include +#include + +#include "source_base/module_out/orb_io.h" +#include "source_pw/module_proj/onsite_proj.h" +#include "source_pw/module_proj/onsite_proj_print.h" +#include "source_pw/module_proj/radial_proj.h" +#include "source_base/projgen.h" +#include "source_base/kernels/math_kernel_op.h" +#include "source_base/tool_quit.h" +#include "source_base/timer.h" +#include "source_io/module_parameter/parameter.h" + +/** + * =============================================================================================== + * + * README + * + * =============================================================================================== + * + * This is a code demo for illustrating how to use unified radial projection in implementation of + * Operators involving local radial projectors on PW-expanded wavefunctions. + * + * Example usage: + * ```c++ + * // select the range of atoms that impose the operator in std::vector> it2ia like + * // it2ia[it] = {ia1, ia2, ...} for each type + * // if all atoms in present kind is "selected", just set it2ia[it].resize(na) and call + * // std::iota(it2ia[it].begin(), it2ia[it].end(), 0) + * + * std::vector> it2ia; // as if we have given its value... + * + * // you should have the `orbital_dir` as the directory containing the orbital files, then those + * // will be read by a static function `AtomicRadials::read_abacus_orb` to get the radial orbitals + * + * // call `init_proj` to initialize the radial projector, this function only needs to be called + * // once during the runtime. + * // its input... + * // the `nproj`, is for specifying number of projectors of each atom type, can be zero, + * // but cannot be the value larger than the number of zeta functions for the given angular momentum. + * // the `lproj` is the angular momentum of the projectors, and `iproj` is the index of zeta function + * // that each projector generated from. + * // the `lproj` along with `iproj` can enable radial projectors in any number developer wants. + * + * // the `onsite_r` is the onsite-radius for all valid projectors, it is used to generate the new + * // radial function that more localized than the original one, which is expected to have enhanced + * // projection efficiency. + * + * std::vector rgrid; + * std::vector> projs; + * std::vector> it2iproj; + * init_proj(orbital_dir, ucell, nproj, lproj, iproj, onsite_r, rgrid, projs, it2iproj); + * + * // then call the function `cal_becp` to calculate the becp. HOWEVER, there are quantities that + * // can be calculated in advance and reused in the following calculations. Please see the function + * // implementation, especially the comments about CACHE 0, CACHE 1, CACHE 2..., etc. + * + * // the input param of `cal_becp`... + * // the `it2ia` has been explained above + * // the `it2iproj` is the output of function `init_proj`, so you do not need to worry about it + * // the `rgrid` and `projs` are also the output of function `init_proj` + * // the `lproj` is the angular momentum for each projector, actually you have used it in `init_proj`, it + * // is the same as `lproj` + * // the `nq` is the number of G+k vectors, typically it is always GlobalV::NQX + * // the `dq` is the step size of G+k vectors, typically it is always GlobalV::DQ + * // the `ik` is the k-point index + * // the `pw_basis` is the plane wave basis, need ik + * // the `omega` is the cell volume + * // the `tpiba` is 2*pi/lat0 + * // the `sf` is the structure factor calculator + * // the `psi` is the wavefunction + * // the `becp` is the output of the function, it is the becp + * cal_becp(it2ia, it2iproj, rgrid, projs, lproj, nq, dq, ik, pw_basis, omega, tpiba, sf, psi, becp); + * + * // About parallelization, presently, the function `AtomicRadials::read_abacus_orb` is actually parallelized + * // by MPI, so after the reading of orbital, actually all processors have the same data. Therefore it is not + * // needed to call functions like `Parallel_Reduce` or `Parallel_Bcast` to synchronize the data. + * // However, what is strikingly memory-consuming is the table `tab_atomic_`. Performance optimization will + * // be needed if the memory is not enough. + */ + +template +void projectors::OnsiteProjector::init(const std::string& orbital_dir, + const UnitCell* ucell_in, + const psi::Psi, Device>& psi, + const K_Vectors& kv, + const ModulePW::PW_Basis_K& pw_basis, // level1: the plane wave basis, need ik + Structure_Factor& sf, // level2: the structure factor calculator + const double onsite_radius, + const int nq, + const double dq, + const ModuleBase::matrix& wg, + const ModuleBase::matrix& ekb) +{ + this->device = base_device::get_device_type(this->ctx); + + if(!this->initialed) + { + this->ucell = ucell_in; + this->ntype = ucell_in->ntype; + this->isk_ = kv.isk.data(); + + this->pw_basis_ = &pw_basis; + this->sf_ = &sf; + + std::vector orb_files(ntype); + std::vector nproj(ntype); + int sum_nproj = 0; + for(int it=0; itorbital_fn[it]; + nproj[it] = ucell->atoms[it].nwl; + sum_nproj += nproj[it]; + } + this->lproj.resize(sum_nproj); + int index = 0; + for(int it=0; itlproj[index++] = il; + } + } + std::vector iproj(sum_nproj, 0); + std::vector onsite_r(sum_nproj, onsite_radius); + + this->it2ia.resize(this->ntype); + this->iat_nh.resize(this->ucell->nat); + int iat = 0; + for(int it = 0; it < it2ia.size(); it++) + { + it2ia[it].resize(this->ucell->atoms[it].na); + std::iota(it2ia[it].begin(), it2ia[it].end(), 0); + for(int ia = 0; ia < it2ia[it].size(); ia++) + { + iat_nh[iat++] = nproj[it] * nproj[it]; + } + } + + this->init_proj(PARAM.inp.orbital_dir, + orb_files, + nproj, + lproj, + iproj, + onsite_r); + + ModuleBase::timer::start("OnsiteProj", "cubspl_tabulate"); + // STAGE 0 - making the interpolation table + // CACHE 0 - if cache the irow2it, irow2iproj, irow2m, itiaiprojm2irow, can be reused for + // SCF, RELAX and CELL-RELAX calculation + // [in] rgrid, projs, lproj, it2ia, it2iproj, nq, dq + RadialProjection::build_backward_map(it2iproj, lproj, irow2it_, irow2iproj_, irow2m_); + RadialProjection::build_forward_map(it2ia, it2iproj, lproj, itiaiprojm2irow_); + RadialProjection::build_sbt_tab(nproj, rgrid, projs, lproj, nq, dq, ucell_in->omega, psi.get_npol(), tab, nhtol); + // For being compatible with present cal_force and cal_stress framework + // uncomment the following code block if you want to use the Onsite_Proj_tools + if(this->tab_atomic_ == nullptr) + { + this->tot_nproj = itiaiprojm2irow_.size(); + this->npwx_ = this->pw_basis_->npwk_max; + this->size_vproj = this->tot_nproj * this->npwx_; + resmem_complex_op()(this->tab_atomic_, this->size_vproj, "OnsiteP::tab_atomic_"); + } + + delete this->fs_tools; // it is okay to delete nullptr + this->fs_tools = new hamilt::Onsite_Proj_tools( + nproj, lproj, tab, nhtol, this->tab_atomic_, ucell_in, &psi, &kv, &pw_basis, &sf, wg, ekb); + + ModuleBase::timer::end("OnsiteProj", "cubspl_tabulate"); + + this->initialed = true; + } +} + +template +void projectors::OnsiteProjector::init_proj(const std::string& orbital_dir, + const std::vector& orb_files, + const std::vector& nproj, // for each type, the number of projectors + const std::vector& lproj, // angular momentum of projectors within the type (l of zeta function) + const std::vector& iproj, // index of projectors within the type (izeta) + const std::vector& onsite_r) +{ + // extract the information from ucell + const int ntype = nproj.size(); + assert(ntype == orb_files.size()); + this->it2iproj.resize(ntype); + + int nproj_tot = 0; + nproj_tot = std::accumulate(nproj.begin(), nproj.end(), nproj_tot, std::plus()); + assert(nproj_tot == lproj.size()); + assert(nproj_tot == iproj.size()); + assert(nproj_tot == onsite_r.size()); + this->projs.resize(nproj_tot); + + int idx = 0; + int nr = -1; + double dr = -1.0; + for(int it = 0; it < ntype; ++it) + { + const int nproj_it = nproj[it]; + this->it2iproj[it].resize(nproj_it); + print::print_proj_status(it, nproj_it); + if(nproj_it == 0) + { + continue; + } + std::ifstream ifs(orbital_dir + orb_files[it]); + std::string elem = ""; + double ecut = -1.0; + int nr_ = -1; + double dr_ = -1.0; + std::vector nzeta; // number of radials for each l + std::vector> radials; // radials arranged in serial + ModuleIO::read_abacus_orb(ifs, elem, ecut, nr_, dr_, nzeta, radials); +#ifdef __DEBUG + assert(elem != ""); + assert(ecut != -1.0); + assert(nr_ != -1); + assert(dr_ != -1.0); +#endif + nr = std::max(nr, nr_); // the maximal nr + assert(dr == -1.0 || dr == dr_); // the dr should be the same for all types + dr = (dr == -1.0) ? dr_ : dr; + for(int ip = 0; ip < nproj_it; ++ip) + { + int l = lproj[idx]; + int izeta = iproj[idx]; + int irad = 0; + irad = std::accumulate(nzeta.begin(), nzeta.begin() + l, irad); + irad += izeta; + std::vector temp = radials[irad]; + rgrid.resize(nr); + std::iota(rgrid.begin(), rgrid.end(), 0); + std::for_each(rgrid.begin(), rgrid.end(), [dr](double& r_i) { r_i *= dr; }); + smoothgen(nr, rgrid.data(), temp.data(), onsite_r[idx], projs[idx]); + it2iproj[it][ip] = idx; + ++idx; + } + } + // do zero padding + if(nr != -1) + { + std::for_each(projs.begin(), projs.end(), [nr](std::vector& proj) { proj.resize(nr, 0.0); }); + } + // generate the rgrid + this->rgrid.resize(nr); + std::iota(rgrid.begin(), rgrid.end(), 0); + std::for_each(rgrid.begin(), rgrid.end(), [dr](double& r_i) { r_i *= dr; }); +} + +// explicit method instantiation +template +void projectors::OnsiteProjector::init( + const std::string&, + const UnitCell*, + const psi::Psi, base_device::DEVICE_CPU>&, + const K_Vectors&, + const ModulePW::PW_Basis_K&, + Structure_Factor&, + const double, + const int, + const double, + const ModuleBase::matrix&, + const ModuleBase::matrix&); + +template +void projectors::OnsiteProjector::init_proj( + const std::string&, + const std::vector&, + const std::vector&, + const std::vector&, + const std::vector&, + const std::vector&); + +#if ((defined __CUDA) || (defined __ROCM)) +template +void projectors::OnsiteProjector::init( + const std::string&, + const UnitCell*, + const psi::Psi, base_device::DEVICE_GPU>&, + const K_Vectors&, + const ModulePW::PW_Basis_K&, + Structure_Factor&, + const double, + const int, + const double, + const ModuleBase::matrix&, + const ModuleBase::matrix&); + +template +void projectors::OnsiteProjector::init_proj( + const std::string&, + const std::vector&, + const std::vector&, + const std::vector&, + const std::vector&, + const std::vector&); +#endif diff --git a/source/source_pw/module_proj/onsite_proj_overlap.cpp b/source/source_pw/module_proj/onsite_proj_overlap.cpp new file mode 100644 index 00000000000..79776c7496e --- /dev/null +++ b/source/source_pw/module_proj/onsite_proj_overlap.cpp @@ -0,0 +1,139 @@ +#include +#include + +#include "source_pw/module_proj/onsite_proj.h" +#include "source_pw/module_proj/onsite_proj_print.h" +#include "source_cell/cell_tools.h" +#include "source_base/kernels/math_kernel_op.h" +#include "source_base/parallel_reduce.h" +#include "source_base/timer.h" +#include "source_estate/occ_comput.h" +#include "source_io/module_parameter/parameter.h" + +template +void projectors::OnsiteProjector::tabulate_atomic(const int ik, const char grad) +{ + ModuleBase::timer::start("OnsiteProj", "tabulate_atomic"); + // The actual tabulation of (STAGE 1 + STAGE 2) is performed by + // Onsite_Proj_tools; this member only records the k-point dimensions. + this->ik_ = ik; + this->npw_ = pw_basis_->npwk[ik]; + this->npwx_ = pw_basis_->npwk_max; + ModuleBase::timer::end("OnsiteProj", "tabulate_atomic"); +} + +template +void projectors::OnsiteProjector::overlap_proj_psi( + const int npm, + const std::complex* ppsi, + const int ld_psi) +{ + ModuleBase::timer::start("OnsiteProj", "overlap"); + // notes on refactor for DCU calculation + // the npm here is nbands(occ) * npol, for calling cal_becp, the npol should be divided. + int npol = this->ucell->get_npol(); + if(this->becp == nullptr || this->size_becp < npm*this->tot_nproj) + { + this->size_becp = npm*this->tot_nproj; + resmem_complex_op()(this->becp, this->size_becp); + if(this->device == base_device::GpuDevice ) + { + resmem_complex_h_op()(this->h_becp, this->size_becp); + } + else + { + this->h_becp = this->becp; + } + } + this->fs_tools->cal_becp(ik_, npm/npol, this->becp, ppsi, ld_psi > 0 ? ld_psi : this->npwx_); // in cal_becp, npm should be the one not multiplied by npol + if(this->device == base_device::GpuDevice) + { + syncmem_complex_d2h_op()(h_becp, this->becp, this->size_becp); + } + ModuleBase::timer::end("OnsiteProj", "overlap"); +} + +template +void projectors::OnsiteProjector::cal_occupations( + const psi::Psi, Device>* psi_in, + const ModuleBase::matrix& wg_in, + const int nspin_in) +{ + ModuleBase::timer::start("OnsiteProj", "cal_occupation"); + this->tabulate_atomic(0); + std::vector> occs(this->tot_nproj * 4, 0.0); + + // loop over k-points to calculate Mi of \sum_{k,i,l,m} + const int nbands = psi_in->get_nbands(); + const int npol = psi_in->get_npol(); + for(int ik = 0; ik < psi_in->get_nk(); ik++) + { + psi_in->fix_k(ik); + if(ik != 0) + { + this->tabulate_atomic(ik); + } + this->overlap_proj_psi(nbands * npol, psi_in->get_pointer()); + // proj(nbands*npol , nkb) holds . + // nspin=2 (npol=1): the spin-up and spin-down channels are separate + // k-points, selected by isk. nspin=1 (npol=1): no spin polarization, + // the occupancy is split evenly so the printed magnetization is zero. + // nspin=4 (npol=2): both spinor components are interleaved per band. + const std::complex* proj_p = this->get_h_becp(); + const double* wg_ik = &wg_in(ik, 0); + const int isk = (nspin_in == 2 && this->isk_ != nullptr) ? this->isk_[ik] : 0; + const int nat = static_cast(this->iat_nh.size()); + elecstate::occ_from_proj( + proj_p, + wg_ik, + nbands, + npol, + this->tot_nproj, + nspin_in, + isk, + this->iat_nh.data(), + nat, + occs.data()); + } + // reduce mag from all k-pools + const int npool = GlobalV::KPAR * PARAM.inp.bndpar; + Parallel_Reduce::reduce_double_allpool(npool, GlobalV::NPROC_IN_POOL, (double*)(&(occs[0])), occs.size()*2); + // occ has been reduced and calculate mag + // Print orbital charge analysis + auto atom_labels = unitcell::get_atomLabels(this->ucell->atoms, this->ucell->ntype); + print::print_orb_chg(this->ucell, occs, this->iat_nh, atom_labels); + + // print charge + ModuleBase::timer::end("OnsiteProj", "cal_occupation"); +} + +// explicit method instantiation +template +void projectors::OnsiteProjector::tabulate_atomic( + const int, const char); + +template +void projectors::OnsiteProjector::overlap_proj_psi( + const int, const std::complex*, const int); + +template +void projectors::OnsiteProjector::cal_occupations( + const psi::Psi, base_device::DEVICE_CPU>*, + const ModuleBase::matrix&, + const int); + +#if ((defined __CUDA) || (defined __ROCM)) +template +void projectors::OnsiteProjector::tabulate_atomic( + const int, const char); + +template +void projectors::OnsiteProjector::overlap_proj_psi( + const int, const std::complex*, const int); + +template +void projectors::OnsiteProjector::cal_occupations( + const psi::Psi, base_device::DEVICE_GPU>*, + const ModuleBase::matrix&, + const int); +#endif diff --git a/source/source_pw/module_pwdft/onsite_proj_print.cpp b/source/source_pw/module_proj/onsite_proj_print.cpp similarity index 94% rename from source/source_pw/module_pwdft/onsite_proj_print.cpp rename to source/source_pw/module_proj/onsite_proj_print.cpp index adb6e1246c0..a15f186e8ae 100644 --- a/source/source_pw/module_pwdft/onsite_proj_print.cpp +++ b/source/source_pw/module_proj/onsite_proj_print.cpp @@ -1,4 +1,4 @@ -#include "source_pw/module_pwdft/onsite_proj_print.h" +#include "source_pw/module_proj/onsite_proj_print.h" #include "source_base/formatter.h" namespace projectors { @@ -32,11 +32,13 @@ void print_orb_chg( /*align=*/{/*value*/FmtTable::Align::RIGHT, /*title*/FmtTable::Align::LEFT}); // parameters for mag output int occ_index = 0; + std::vector atom_labels_iat(ucell->nat); for(int iat=0; iatnat; iat++) { const int it = ucell->iat2it[iat]; std::string atom_label = atom_labels[it]; int ia = ucell->iat2ia[iat]; + atom_labels_iat[iat] = atom_label + std::to_string(ia+1); GlobalV::ofs_running << FmtCore::format("%-20s", atom_label+std::to_string(ia+1)) << std::endl; std::vector sum(4, 0.0); int current_l = 1; @@ -72,7 +74,7 @@ void print_orb_chg( GlobalV::ofs_running << std::endl; // Print magnetism table - print_mag_table(atom_labels, mag_x, mag_y, mag_z); + print_mag_table(atom_labels_iat, mag_x, mag_y, mag_z); } void print_mag_table( diff --git a/source/source_pw/module_pwdft/onsite_proj_print.h b/source/source_pw/module_proj/onsite_proj_print.h similarity index 100% rename from source/source_pw/module_pwdft/onsite_proj_print.h rename to source/source_pw/module_proj/onsite_proj_print.h diff --git a/source/source_pw/module_pwdft/onsite_proj_tools.cpp b/source/source_pw/module_proj/onsite_proj_tools.cpp similarity index 99% rename from source/source_pw/module_pwdft/onsite_proj_tools.cpp rename to source/source_pw/module_proj/onsite_proj_tools.cpp index 82e3f07bc7f..a1d803816a2 100644 --- a/source/source_pw/module_pwdft/onsite_proj_tools.cpp +++ b/source/source_pw/module_proj/onsite_proj_tools.cpp @@ -1,4 +1,4 @@ -#include "onsite_proj_tools.h" +#include "source_pw/module_proj/onsite_proj_tools.h" #include "source_base/math_polyint.h" #include "source_base/math_ylmreal.h" @@ -7,7 +7,7 @@ #include "source_base/tool_title.h" #include "source_pw/module_pwdft/kernels/force_op.h" #include "source_io/module_parameter/parameter.h" -#include "nonlocal_maths.hpp" +#include "source_pw/module_pwdft/nonlocal_maths.h" #include diff --git a/source/source_pw/module_pwdft/onsite_proj_tools.h b/source/source_pw/module_proj/onsite_proj_tools.h similarity index 99% rename from source/source_pw/module_pwdft/onsite_proj_tools.h rename to source/source_pw/module_proj/onsite_proj_tools.h index 7e8bfd2facf..facbb9faf14 100644 --- a/source/source_pw/module_pwdft/onsite_proj_tools.h +++ b/source/source_pw/module_proj/onsite_proj_tools.h @@ -167,7 +167,7 @@ class Onsite_Proj_tools int gcar_zero_counts[3] = {0, 0, 0}; std::complex* vkb_save = nullptr; /// @brief count zero gcar indexes and prepare zero_indexes, do gcar_y /= gcar_x, gcar_z /= gcar_y - void transfer_gcar(int npw, int npw_max, const FPTYPE* gcar_in); + void transfer_gcar(int npw, int npw_max, const ModuleBase::Vector3* gcar_in); /// @brief save the 0-value dvkbs for calculating the dbecp_i in the force calculation void save_vkb(int npw, int ipol); /// @brief revert the 0-value dvkbs for calculating the dbecp_i in the force calculation diff --git a/source/source_pw/module_pwdft/onsite_proj_tools_becp.cpp b/source/source_pw/module_proj/onsite_proj_tools_becp.cpp similarity index 98% rename from source/source_pw/module_pwdft/onsite_proj_tools_becp.cpp rename to source/source_pw/module_proj/onsite_proj_tools_becp.cpp index c0afe1b13b0..170f1ecb931 100644 --- a/source/source_pw/module_pwdft/onsite_proj_tools_becp.cpp +++ b/source/source_pw/module_proj/onsite_proj_tools_becp.cpp @@ -1,4 +1,4 @@ -#include "onsite_proj_tools.h" +#include "source_pw/module_proj/onsite_proj_tools.h" #include "source_base/math_polyint.h" #include "source_base/math_ylmreal.h" @@ -6,7 +6,7 @@ #include "source_base/timer.h" #include "source_base/tool_title.h" #include "source_io/module_parameter/parameter.h" -#include "nonlocal_maths.hpp" +#include "source_pw/module_pwdft/nonlocal_maths.h" // cal_becp // starts from vkb (nkb, ng) table diff --git a/source/source_pw/module_pwdft/onsite_proj_tools_dbecp.cpp b/source/source_pw/module_proj/onsite_proj_tools_dbecp.cpp similarity index 87% rename from source/source_pw/module_pwdft/onsite_proj_tools_dbecp.cpp rename to source/source_pw/module_proj/onsite_proj_tools_dbecp.cpp index 7b77c64e86e..a6a121c8a0a 100644 --- a/source/source_pw/module_pwdft/onsite_proj_tools_dbecp.cpp +++ b/source/source_pw/module_proj/onsite_proj_tools_dbecp.cpp @@ -1,13 +1,12 @@ -#include "onsite_proj_tools.h" - #include "source_base/math_polyint.h" #include "source_base/math_ylmreal.h" #include "source_base/parallel_reduce.h" #include "source_base/timer.h" #include "source_base/tool_title.h" -#include "source_pw/module_pwdft/kernels/force_op.h" #include "source_io/module_parameter/parameter.h" -#include "nonlocal_maths.hpp" +#include "source_pw/module_proj/onsite_proj_tools.h" +#include "source_pw/module_pwdft/kernels/force_op.h" +#include "source_pw/module_pwdft/nonlocal_maths.h" namespace hamilt { @@ -80,14 +79,7 @@ void Onsite_Proj_tools::cal_dbecp_s(int ik, int npm, int ipol, i int nh = pref.size(); // prepare indexes for calculate vkb_deri this->dvkb_indexes.resize(nh * 4); - maths.cal_dvkb_index(this->nproj[it], - this->nhtol->c, - this->nhtol->nc, - npw, - it, - ipol, - jpol, - this->dvkb_indexes.data()); + maths.cal_dvkb_index(this->nproj[it], this->nhtol->c, this->nhtol->nc, npw, it, ipol, jpol, this->dvkb_indexes.data()); if (this->device == base_device::GpuDevice) { syncmem_int_h2d_op()(d_dvkb_indexes, dvkb_indexes.data(), nh * 4); @@ -126,19 +118,7 @@ void Onsite_Proj_tools::cal_dbecp_s(int ik, int npm, int ipol, i const char transa = 'C'; const char transb = 'N'; - gemm_op()(transa, - transb, - nkb, - npm_npol, - npw, - &ModuleBase::ONE, - ppcell_vkb, - npw, - ppsi, - this->max_npw, - &ModuleBase::ZERO, - dbecp, - nkb); + gemm_op()(transa, transb, nkb, npm_npol, npw, &ModuleBase::ONE, ppcell_vkb, npw, ppsi, this->max_npw, &ModuleBase::ZERO, dbecp, nkb); ModuleBase::timer::end("Onsite_Proj_tools", "cal_dbecp_s"); } @@ -163,15 +143,13 @@ void Onsite_Proj_tools::cal_dbecp_f(int ik, int npm, int ipol) if (this->pre_ik_f == -1) // if it is the very first run, we allocate { resmem_var_op()(gcar, 3 * this->wfc_basis_->npwk_max); - resmem_int_op()(gcar_zero_indexes, 3 * this->wfc_basis_->npwk_max); + // Each row stores a count followed by up to npwk_max plane-wave indices. + resmem_int_op()(gcar_zero_indexes, 3 * (this->wfc_basis_->npwk_max + 1)); } // first refresh the value of gcar_zero_indexes, gcar_zero_counts if (this->pre_ik_f != ik) - { // the following lines will cause UNDEFINED BEHAVIOR because memory layout of vector3 instance - // is assumed to be always contiguous but it is not guaranteed. - this->transfer_gcar(npw, - this->wfc_basis_->npwk_max, - &(this->wfc_basis_->gcar[ik * this->wfc_basis_->npwk_max].x)); + { + this->transfer_gcar(npw, this->wfc_basis_->npwk_max, &this->wfc_basis_->gcar[ik * this->wfc_basis_->npwk_max]); } // backup vkb values to vkb_save @@ -226,10 +204,15 @@ void Onsite_Proj_tools::cal_dbecp_f(int ik, int npm, int ipol) template void Onsite_Proj_tools::save_vkb(int npw, int ipol) { + if (this->gcar_zero_counts[ipol] == 0) + { + return; + } + const int index_stride = this->wfc_basis_->npwk_max + 1; if (this->device == base_device::CpuDevice) { - const int gcar_zero_count = this->gcar_zero_indexes[ipol * this->wfc_basis_->npwk_max]; - const int* gcar_zero_ptrs = &this->gcar_zero_indexes[ipol * this->wfc_basis_->npwk_max + 1]; + const int gcar_zero_count = this->gcar_zero_indexes[ipol * index_stride]; + const int* gcar_zero_ptrs = &this->gcar_zero_indexes[ipol * index_stride + 1]; const std::complex* vkb_ptr = this->ppcell_vkb; std::complex* vkb_save_ptr = this->vkb_save; // find the zero indexes to save the vkb values to vkb_save @@ -253,7 +236,7 @@ void Onsite_Proj_tools::save_vkb(int npw, int ipol) this->gcar_zero_counts[ipol], npw, ipol, - this->wfc_basis_->npwk_max); + index_stride); #endif } } @@ -262,11 +245,16 @@ void Onsite_Proj_tools::save_vkb(int npw, int ipol) template void Onsite_Proj_tools::revert_vkb(int npw, int ipol) { + if (this->gcar_zero_counts[ipol] == 0) + { + return; + } + const int index_stride = this->wfc_basis_->npwk_max + 1; const std::complex coeff = ipol == 0 ? ModuleBase::NEG_IMAG_UNIT : ModuleBase::ONE; if (this->device == base_device::CpuDevice) { - const int gcar_zero_count = this->gcar_zero_indexes[ipol * this->wfc_basis_->npwk_max]; - const int* gcar_zero_ptrs = &this->gcar_zero_indexes[ipol * this->wfc_basis_->npwk_max + 1]; + const int gcar_zero_count = this->gcar_zero_indexes[ipol * index_stride]; + const int* gcar_zero_ptrs = &this->gcar_zero_indexes[ipol * index_stride + 1]; std::complex* vkb_ptr = this->ppcell_vkb; const std::complex* vkb_save_ptr = this->vkb_save; // find the zero indexes to save the vkb values to vkb_save @@ -290,24 +278,32 @@ void Onsite_Proj_tools::revert_vkb(int npw, int ipol) this->gcar_zero_counts[ipol], npw, ipol, - this->wfc_basis_->npwk_max, + index_stride, coeff); #endif } } template -void Onsite_Proj_tools::transfer_gcar(int npw, int npw_max, const FPTYPE* gcar_in) +void Onsite_Proj_tools::transfer_gcar(int npw, int npw_max, const ModuleBase::Vector3* gcar_in) { + // unpack Vector3 into a contiguous buffer elementwise: + // the memory layout of Vector3 is not guaranteed, so copying through + // a raw FPTYPE* pointer would be undefined behavior. std::vector gcar_tmp(3 * npw_max); // [out], will overwritten this->gcar - gcar_tmp.assign(gcar_in, - gcar_in + 3 * npw_max); // UNDEFINED BEHAVIOR!!! nobody always knows the memory layout of vector3 - std::vector gcar_zero_indexes_tmp(3 * npw_max); // a "checklist" + for (int ig = 0; ig < npw; ++ig) + { + gcar_tmp[ig * 3] = gcar_in[ig].x; + gcar_tmp[ig * 3 + 1] = gcar_in[ig].y; + gcar_tmp[ig * 3 + 2] = gcar_in[ig].z; + } + const int index_stride = npw_max + 1; + std::vector gcar_zero_indexes_tmp(3 * index_stride); int* gcar_zero_ptrs[3]; for (int i = 0; i < 3; i++) { - gcar_zero_ptrs[i] = &gcar_zero_indexes_tmp[i * npw_max]; + gcar_zero_ptrs[i] = &gcar_zero_indexes_tmp[i * index_stride]; gcar_zero_ptrs[i][0] = -1; this->gcar_zero_counts[i] = 0; } @@ -350,7 +346,7 @@ void Onsite_Proj_tools::transfer_gcar(int npw, int npw_max, cons resmem_complex_op()(this->vkb_save, this->nkb * max_count); // transfer the gcar and gcar_zero_indexes to the device syncmem_var_h2d_op()(gcar, gcar_tmp.data(), 3 * npw_max); - syncmem_int_h2d_op()(gcar_zero_indexes, gcar_zero_indexes_tmp.data(), 3 * npw_max); + syncmem_int_h2d_op()(gcar_zero_indexes, gcar_zero_indexes_tmp.data(), 3 * index_stride); } // template instantiation @@ -358,13 +354,13 @@ template void Onsite_Proj_tools::cal_dbecp_s(in template void Onsite_Proj_tools::cal_dbecp_f(int, int, int); template void Onsite_Proj_tools::save_vkb(int, int); template void Onsite_Proj_tools::revert_vkb(int, int); -template void Onsite_Proj_tools::transfer_gcar(int, int, const double*); +template void Onsite_Proj_tools::transfer_gcar(int, int, const ModuleBase::Vector3*); #if ((defined __CUDA) || (defined __ROCM)) template void Onsite_Proj_tools::cal_dbecp_s(int, int, int, int); template void Onsite_Proj_tools::cal_dbecp_f(int, int, int); template void Onsite_Proj_tools::save_vkb(int, int); template void Onsite_Proj_tools::revert_vkb(int, int); -template void Onsite_Proj_tools::transfer_gcar(int, int, const double*); +template void Onsite_Proj_tools::transfer_gcar(int, int, const ModuleBase::Vector3*); #endif } // namespace hamilt diff --git a/source/source_pw/module_pwdft/onsite_proj_tools_force.cpp b/source/source_pw/module_proj/onsite_proj_tools_force.cpp similarity index 98% rename from source/source_pw/module_pwdft/onsite_proj_tools_force.cpp rename to source/source_pw/module_proj/onsite_proj_tools_force.cpp index 7c326c0ab8f..a2c8b9278e8 100644 --- a/source/source_pw/module_pwdft/onsite_proj_tools_force.cpp +++ b/source/source_pw/module_proj/onsite_proj_tools_force.cpp @@ -1,4 +1,4 @@ -#include "onsite_proj_tools.h" +#include "source_pw/module_proj/onsite_proj_tools.h" #include "source_base/math_polyint.h" #include "source_base/math_ylmreal.h" @@ -7,7 +7,7 @@ #include "source_base/tool_title.h" #include "source_pw/module_pwdft/kernels/force_op.h" #include "source_io/module_parameter/parameter.h" -#include "nonlocal_maths.hpp" +#include "source_pw/module_pwdft/nonlocal_maths.h" #include diff --git a/source/source_pw/module_pwdft/onsite_proj_tools_stress.cpp b/source/source_pw/module_proj/onsite_proj_tools_stress.cpp similarity index 96% rename from source/source_pw/module_pwdft/onsite_proj_tools_stress.cpp rename to source/source_pw/module_proj/onsite_proj_tools_stress.cpp index 99f4da432d8..5ba09ff6416 100644 --- a/source/source_pw/module_pwdft/onsite_proj_tools_stress.cpp +++ b/source/source_pw/module_proj/onsite_proj_tools_stress.cpp @@ -1,4 +1,4 @@ -#include "onsite_proj_tools.h" +#include "source_pw/module_proj/onsite_proj_tools.h" #include "source_base/math_polyint.h" #include "source_base/math_ylmreal.h" @@ -7,7 +7,7 @@ #include "source_base/tool_title.h" #include "source_pw/module_pwdft/kernels/force_op.h" #include "source_io/module_parameter/parameter.h" -#include "nonlocal_maths.hpp" +#include "source_pw/module_pwdft/nonlocal_maths.h" #include @@ -30,15 +30,15 @@ double Onsite_Proj_tools::cal_stress_dftu(int ik, #if defined(__CUDA) || defined(__ROCM) if (this->device == base_device::GpuDevice) { - // orb_corr_tmp + // orb_corr_tmp resmem_int_op()(orb_corr_tmp, this->ucell_->ntype); syncmem_int_h2d_op()(orb_corr_tmp, orb_corr, this->ucell_->ntype); - // pot_onsite_tmp + // pot_onsite_tmp resmem_complex_op()(pot_onsite_tmp, size_pot_onsite); syncmem_complex_h2d_op()(pot_onsite_tmp, pot_onsite, size_pot_onsite); - // transfer data from from host to device + // transfer data from from host to device syncmem_var_h2d_op()(d_wg, h_wg, this->nbands * (ik+1)); // Allocate device memory for stress @@ -99,7 +99,7 @@ double Onsite_Proj_tools::cal_stress_dftu(int ik, template double Onsite_Proj_tools::cal_stress_dspin(int ik, int npm, - const ModuleBase::Vector3* lambda, + const ModuleBase::Vector3* lambda, const FPTYPE* h_wg) { double stress_out = 0.0; diff --git a/source/source_pw/module_proj/radial_proj.cpp b/source/source_pw/module_proj/radial_proj.cpp new file mode 100644 index 00000000000..027c6daa046 --- /dev/null +++ b/source/source_pw/module_proj/radial_proj.cpp @@ -0,0 +1,300 @@ +#include +#include +#include +#include +#include +#include +#include "source_pw/module_proj/radial_proj.h" +#include "source_base/constants.h" +#include "source_base/math_ylmreal.h" +#include "source_base/sph_bessel_tf.h" +#include "source_base/timer.h" + +void RadialProjection::build_backward_map(const std::vector>& it2iproj, + const std::vector& iproj2l, + std::vector& irow2it, + std::vector& irow2iproj, + std::vector& irow2m) +{ + const int ntype = it2iproj.size(); // the ntype here only count the valid, that is, with the projector. + + int nproj_tot = 0; // count the total number of projectors (atom-position-irrelevant) + for(int it = 0; it < ntype; it++) // for all types with projector... + { + for(auto& iproj: it2iproj[it]) // for each projector, the projector is indexed with global iproj + { + const int l = iproj2l[iproj]; // easily get the angular momentum of this projector + nproj_tot += (2*l + 1); // add 2l+1 to the total number of projectors + } + } + // resize/allcoate the memory for the output + irow2it.resize(nproj_tot); + irow2iproj.resize(nproj_tot); + irow2m.resize(nproj_tot); + + int irow = 0; + for(int it = 0; it < ntype; it++) + { + const int nproj = it2iproj[it].size(); + for(int iproj = 0; iproj < nproj; iproj++) + { + const int l = iproj2l[it2iproj[it][iproj]]; + for(int m = -l; m <= l; m++) + { + irow2it[irow] = it; + irow2iproj[irow] = iproj; + irow2m[irow] = m; + irow++; + } + } + } +} + +void RadialProjection::build_forward_map(const std::vector>& it2ia, + const std::vector>& it2iproj, + const std::vector& iproj2l, + std::map, int>& itiaiprojm2irow) +{ + const int ntype = it2ia.size(); + int irow = 0; + for(int it = 0; it < ntype; it++) + { + const int nproj = it2iproj[it].size(); + for(auto& ia: it2ia[it]) // the index is from UnitCell, so it allows the case ia is not continuous + { + for(int iproj = 0; iproj < nproj; iproj++) + { + const int l = iproj2l[it2iproj[it][iproj]]; // what is the iproj of the i-th projector of it atomtype? + for(int m = -l; m <= l; m++) + { + itiaiprojm2irow[std::make_tuple(it, ia, iproj, m)] = irow; + irow++; + } + } + } + } +} + +void RadialProjection::build_sbt_tab(const std::vector& nproj, + const std::vector& r, + const std::vector>& radials, + const std::vector& l, + const int nq, + const double dq, + const double omega, + const int npol, + ModuleBase::realArray& tab, + ModuleBase::matrix& nhtol) +{ + int nprojmax = *std::max_element(nproj.begin(), nproj.end()); + const int ntype = nproj.size(); + + tab.create(ntype, nprojmax*npol, nq); + tab.zero_out(); + + std::vector qgrid(nq); + std::iota(qgrid.begin(), qgrid.end(), 0); + std::transform(qgrid.begin(), qgrid.end(), qgrid.begin(), [dq](const double& q){return q*dq;}); + + ModuleBase::SphericalBesselTransformer sbt_(true); // bool: enable cache + int iproj = 0; + int nchmax = 0; + const double pref = 4*M_PI/std::sqrt(omega) / std::sqrt(2.0/std::acos(-1.0)); + for (int it = 0; it < ntype; it++) + { + int nch = 0; + const int nproj_it = nproj[it]; + for (int ip = 0; ip < nproj_it; ip++) + { + const int l_ = l[iproj]; + nch += 2*l_ + 1; + std::vector _temp(nq); + sbt_.direct(l_, r.size(), r.data(), radials[iproj].data(), nq, qgrid.data(), _temp.data()); + std::for_each(_temp.begin(), _temp.end(), [pref](double& x){x = x*pref;}); + for (int iq = 0; iq < nq; iq++) + { + tab(it, ip, iq) = _temp[iq]; + } + iproj++; + } + nchmax = std::max(nchmax, nch); + } + + nhtol.create(ntype, nchmax); + nhtol.zero_out(); + iproj = 0; + for (int it = 0; it < ntype; it++) + { + int ih = 0; // channel index, across all projectors of present type + for (int ip = 0; ip < nproj[it]; ip++) + { + const int l_ = l[iproj]; + for (int m = -l_; m <= l_; m++) + { + nhtol(it, ih) = l_; + ih++; + } + iproj++; + } + } +} + +void RadialProjection::RadialProjector::build_sbt_tab(const int nr, + const double* r, + const std::vector& radials, + const std::vector& l, + const int nq, + const double dq) +{ + ModuleBase::timer::start("RadialProjection", "cubspl_tabulate_vq_each_radial"); + l_ = l; + const int nrad = radials.size(); + assert(nrad == l.size()); + std::vector qgrid(nq); + std::iota(qgrid.begin(), qgrid.end(), 0); + std::transform(qgrid.begin(), qgrid.end(), qgrid.begin(), [dq](const double& q){return q*dq;}); + + if(cubspl_.get()) { cubspl_.reset(); } // release the old one if it is not the first time + cubspl_ = std::unique_ptr(new ModuleBase::CubicSpline(nq, // int + qgrid.data())); // double* + cubspl_->reserve(nrad); + ModuleBase::SphericalBesselTransformer sbt_(true); // bool: enable cache + + std::vector _temp(nq); + // the SphericalBesselTransformer's result is multiplied by one extra factor sqrt(2/pi), should remove it + // see source_base/sph_bessel_tf.h and source_base/sph_bessel_tf.cpp:328 + const double pref = std::sqrt(2.0/std::acos(-1.0)); + for(int i = 0; i < nrad; i++) + { + sbt_.direct(l[i], nr, r, radials[i], nq, qgrid.data(), _temp.data()); + std::for_each(_temp.begin(), _temp.end(), [pref](double& x){x = x/pref;}); + cubspl_->add(_temp.data()); + } + ModuleBase::timer::end("RadialProjection", "cubspl_tabulate_vq_each_radial"); +} + +void RadialProjection::RadialProjector::build_sbt_tab(const std::vector& r, + const std::vector>& radials, + const std::vector& l, + const int nq, + const double dq) +{ + ModuleBase::timer::start("RadialProjection", "cubspl_tabulate_vq_each_radial"); + const int nr = r.size(); + const int nrad = radials.size(); + for(int i = 0; i < nrad; i++) { assert(radials[i].size() == nr); } + std::vector radptrs(radials.size()); + for(int i = 0; i < radials.size(); i++) { radptrs[i] = const_cast(radials[i].data()); } + ModuleBase::timer::end("RadialProjection", "cubspl_tabulate_vq_each_radial"); + build_sbt_tab(nr, r.data(), radptrs, l, nq, dq); +} + +void RadialProjection::RadialProjector::sbtft(const std::vector>& qs, + std::vector>& out, + const char type, + const double omega, + const double tpiba) +{ + ModuleBase::timer::start("RadialProjection", "interp_sphbes_ft_flzYlm"); + assert(type == 'r' || type == 'l'); // type must be one of 'r' or 'l' + // first cache the Ylm values + const int lmax_ = *std::max_element(l_.begin(), l_.end()); + const int total_lm = std::pow(lmax_+1, 2); + const int npw = qs.size(); + ModuleBase::matrix ylm_(total_lm, npw); + ModuleBase::YlmReal::Ylm_Real(total_lm, npw, qs.data(), ylm_); + + const int nrad = l_.size(); + int nchannel = 0; + for(auto l: l_) { nchannel += 2*l+1; } + out.resize(nchannel*npw); + + std::vector qnorm(npw); + std::transform(qs.begin(), qs.end(), qnorm.begin(), [tpiba](const ModuleBase::Vector3& q){return tpiba*q.norm();}); + + std::vector Jlfq(npw); + int iproj = 0; + for(int i = 0; i < nrad; i++) + { + const int l = l_[i]; + // here is bug-prone + // we define l as and r as . The former is int{p(r)exp(iqr)} and the latter is int{p(r)exp(-iqr)} + // , in which we have use G+k=q notation. So once do Ylm expansion on exp(iqr), will get a pure imaginary + // prefactor i^l. + std::complex pref = (type == 'l')? std::pow(ModuleBase::IMAG_UNIT, l) : std::pow(ModuleBase::NEG_IMAG_UNIT, l); + pref = pref * ModuleBase::FOUR_PI/std::sqrt(omega); + cubspl_->eval(npw, qnorm.data(), Jlfq.data(), nullptr, nullptr, i); + for(int m = -l; m <= l; m++) + { + for(int iq = 0; iq < npw; iq++) + { + out[iproj*npw+iq] = pref * Jlfq[iq] * ylm_(l*l + l + m, iq); + } + iproj++; + } + } + assert(iproj == nchannel); // should write to inflate each radial to 2l+1 channels + ModuleBase::timer::end("RadialProjection", "interp_sphbes_ft_flzYlm"); +} + +void RadialProjection::mask_func(std::vector& mask) +{ + /* mask function is hard coded here, eta = 15 */ + mask.resize(201); + std::string src = "0.10000000E+01"; + src += " 0.10000000E+01 0.99948662E+00 0.99863154E+00 0.99743557E+00"; + src += " 0.99589985E+00 0.99402586E+00 0.99181538E+00 0.98927052E+00"; + src += " 0.98639370E+00 0.98318766E+00 0.97965544E+00 0.97580040E+00"; + src += " 0.97162618E+00 0.96713671E+00 0.96233623E+00 0.95722924E+00"; + src += " 0.95182053E+00 0.94611516E+00 0.94011842E+00 0.93383589E+00"; + src += " 0.92727338E+00 0.92043693E+00 0.91333282E+00 0.90596753E+00"; + src += " 0.89834777E+00 0.89048044E+00 0.88237263E+00 0.87403161E+00"; + src += " 0.86546483E+00 0.85667987E+00 0.84768450E+00 0.83848659E+00"; + src += " 0.82909416E+00 0.81951535E+00 0.80975838E+00 0.79983160E+00"; + src += " 0.78974340E+00 0.77950227E+00 0.76911677E+00 0.75859548E+00"; + src += " 0.74794703E+00 0.73718009E+00 0.72630334E+00 0.71532544E+00"; + src += " 0.70425508E+00 0.69310092E+00 0.68187158E+00 0.67057566E+00"; + src += " 0.65922170E+00 0.64781819E+00 0.63637355E+00 0.62489612E+00"; + src += " 0.61339415E+00 0.60187581E+00 0.59034914E+00 0.57882208E+00"; + src += " 0.56730245E+00 0.55579794E+00 0.54431609E+00 0.53286431E+00"; + src += " 0.52144984E+00 0.51007978E+00 0.49876105E+00 0.48750040E+00"; + src += " 0.47630440E+00 0.46517945E+00 0.45413176E+00 0.44316732E+00"; + src += " 0.43229196E+00 0.42151128E+00 0.41083069E+00 0.40025539E+00"; + src += " 0.38979038E+00 0.37944042E+00 0.36921008E+00 0.35910371E+00"; + src += " 0.34912542E+00 0.33927912E+00 0.32956851E+00 0.31999705E+00"; + src += " 0.31056799E+00 0.30128436E+00 0.29214897E+00 0.28316441E+00"; + src += " 0.27433307E+00 0.26565709E+00 0.25713844E+00 0.24877886E+00"; + src += " 0.24057988E+00 0.23254283E+00 0.22466884E+00 0.21695884E+00"; + src += " 0.20941357E+00 0.20203357E+00 0.19481920E+00 0.18777065E+00"; + src += " 0.18088790E+00 0.17417080E+00 0.16761900E+00 0.16123200E+00"; + src += " 0.15500913E+00 0.14894959E+00 0.14305240E+00 0.13731647E+00"; + src += " 0.13174055E+00 0.12632327E+00 0.12106315E+00 0.11595855E+00"; + src += " 0.11100775E+00 0.10620891E+00 0.10156010E+00 0.97059268E-01"; + src += " 0.92704295E-01 0.88492966E-01 0.84422989E-01 0.80492001E-01"; + src += " 0.76697569E-01 0.73037197E-01 0.69508335E-01 0.66108380E-01"; + src += " 0.62834685E-01 0.59684561E-01 0.56655284E-01 0.53744102E-01"; + src += " 0.50948236E-01 0.48264886E-01 0.45691239E-01 0.43224469E-01"; + src += " 0.40861744E-01 0.38600231E-01 0.36437098E-01 0.34369520E-01"; + src += " 0.32394681E-01 0.30509780E-01 0.28712032E-01 0.26998673E-01"; + src += " 0.25366964E-01 0.23814193E-01 0.22337676E-01 0.20934765E-01"; + src += " 0.19602844E-01 0.18339338E-01 0.17141711E-01 0.16007467E-01"; + src += " 0.14934157E-01 0.13919377E-01 0.12960772E-01 0.12056034E-01"; + src += " 0.11202905E-01 0.10399183E-01 0.96427132E-02 0.89313983E-02"; + src += " 0.82631938E-02 0.76361106E-02 0.70482151E-02 0.64976294E-02"; + src += " 0.59825322E-02 0.55011581E-02 0.50517982E-02 0.46327998E-02"; + src += " 0.42425662E-02 0.38795566E-02 0.35422853E-02 0.32293218E-02"; + src += " 0.29392897E-02 0.26708663E-02 0.24227820E-02 0.21938194E-02"; + src += " 0.19828122E-02 0.17886449E-02 0.16102512E-02 0.14466132E-02"; + src += " 0.12967606E-02 0.11597692E-02 0.10347601E-02 0.92089812E-03"; + src += " 0.81739110E-03 0.72348823E-03 0.63847906E-03 0.56169212E-03"; + src += " 0.49249371E-03 0.43028657E-03 0.37450862E-03 0.32463165E-03"; + src += " 0.28016004E-03 0.24062948E-03 0.20560566E-03 0.17468305E-03"; + src += " 0.14748362E-03 0.12365560E-03 0.10287226E-03 0.84830727E-04"; + src += " 0.69250769E-04 0.55873673E-04 0.44461100E-04 0.34793983E-04"; + src += " 0.26671449E-04 0.19909778E-04 0.14341381E-04 0.98138215E-05"; + std::stringstream ss(src); + for(int i = 0; i < mask.size(); i++) + { + ss >> mask[i]; + } +} diff --git a/source/source_pw/module_proj/radial_proj.h b/source/source_pw/module_proj/radial_proj.h new file mode 100644 index 00000000000..8ee8ec82a9f --- /dev/null +++ b/source/source_pw/module_proj/radial_proj.h @@ -0,0 +1,192 @@ +#ifndef RADIAL_PROJECTION_H +#define RADIAL_PROJECTION_H + +/** + * @file radial_proj.h + * + * project any atom-centered function that has separable radial and angular parts + * or any function that can be expanded with spherical harmonics onto the planewave basis, + * although the latter will be somewhat cumbersome: + * f(r) = sum_{l,m} f_{lm}(r) * Ylm(theta, phi) + * F(q) = sum_{l,m} i^l * 4*pi/sqrt(omega) * Jl[f_{lm}](q) * Ylm(q) + */ + +#include "source_base/vector3.h" +#include "source_base/cubic_spline.h" +#include "source_base/matrix.h" +#include "source_base/realarray.h" +#include +#include +#include +#include +#include + +namespace RadialProjection +{ + /** + * Notation of the following two functions: + * + * Given all the projectors are listed in a series, so the `iproj` is the index goes across + * all atomtypes, which means if for the first type, the iproj goes from 0 to 4, then the + * second atomtypes the iproj will start from 5, and so on... + * However, there is also another convention, like numerical atomic orbitals, developer always + * use "l" to index orbitals, here, in all output map, the `iproj` will start from 0, which + * means in output the `iproj` is local index. + * ----------------------------------------------------------------------------------------- + * First, the following lists should be prepared as early as possible, + * + * it2iproj: for given it, the index of atom type, return the list of index of projectors. + * + * iproj2l: for given iproj, the index of projectors, return the l of this projector. More + * simply explaning, it is just the list of angular momentum of projectors. + * + * it2ia: just a list that stolen information from UnitCell, for given it, the index of atom + * within the range of it. So this list is different from the it2iproj, iproj is the index + * across type but ia is the index within the type. So for each it2ia[it], the ia, in principle + * , always/can start from 0. + * + * One may question that does the indexing support one atom type with multiple projectors? The + * answer is YES. Combining the it2iproj and it2ia, one can even support PART of atoms of one + * type has multiple projectors. + * ----------------------------------------------------------------------------------------- + * Then the returned lists, + * + * irow2it: for given `irow`, the index of row, return the `it`: the index of atom type. + * + * irow2iproj: for given `irow`, the index of row, return the `iproj`, the index of projectors, + * note that this `iproj` is the local index. + * + * irow2m: for given irow, the index of row, return the m, the magnetic quantum number of this + * projector. + * + * One may complain that cannot get `l` from the `irow`, but the truth is, not exactly. One can + * get the `l` starting from `irow` by: + * ```c++ + * const int iproj = irow2iproj[irow]; + * const int it = irow2it[irow]; + * const int iproj_g = it2iproj[it][iproj]; + * const int l = iproj2l[iproj_g]; + * ``` + */ + void build_backward_map(const std::vector>& it2iproj, + const std::vector& iproj2l, + std::vector& irow2it, + std::vector& irow2iproj, + std::vector& irow2m); + + void build_forward_map(const std::vector>& it2ia, + const std::vector>& it2iproj, + const std::vector& iproj2l, + std::map, int>& itiaiprojm2irow); + + /** + * @brief make interpolation tables for the Spherical Bessel Transform of + * type-wise radial projectors, in the (ntype, nprojmax*npol, nq) realArray + * layout used by the nonlocal-operator kernels, plus the (it, ih) -> l map. + * + * @param nproj number of projectors for each atom type + * @param r radial grids, shared by all radial functions + * @param radials radial functions, each element is a radial function + * @param l angular momentum quantum number for each radial function + * @param nq number of q-points + * @param dq space between q-points + * @param omega cell volume, used in the prefactor 4*pi/sqrt(omega) + * @param npol number of spinor components (for nspin 4) + * @param tab [out] interpolation table, (ntype, nprojmax*npol, nq) + * @param nhtol [out] map from (it, ih) to l, with ih the (l, m)-distinctive index + */ + void build_sbt_tab(const std::vector& nproj, + const std::vector& r, + const std::vector>& radials, + const std::vector& l, + const int nq, + const double dq, + const double omega, + const int npol, + ModuleBase::realArray& tab, + ModuleBase::matrix& nhtol); + + /** + * @brief RadialProjector holds the interpolation table of the Spherical Bessel + * Transform of a set of radial functions and evaluates the analytical Fourier + * transform on arbitrary q vectors. + * + * Usage: + * + * reciprocal space integration + * ```c++ + * RadialProjector rp; + * const int nq = 1000; + * const double dq = 0.01; + * // given `r` is the real space grid and `radials` is the collection of radial + * // functions, `l` is the angular momentum quantum number for each radial function + * // then the interpolation table can be rapidly built by calling SphericalBesselTransformer + * // and CubicSpline modules. + * rp.build_sbt_tab(r, radials, l, nq, dq); + * // then the set of q will used to calculate the Fourier transform + * rp.sbtft(qs, out, 'r', omega, tpiba); + * // in `out`, there will be the Fourier transform of the radial functions organized + * // in the same way as the input `radials` and `qs`, as row and column respectively. + * // but one should note for each radials, there are 2*l+1 components now instead of + * // just one. + * ``` + */ + class RadialProjector + { + public: + RadialProjector() = default; + ~RadialProjector() = default; + + /** + * @brief make a interpolation table for the Spherical Bessel Transform of f(r) + * + * @param nr number of grid points, shared by all radial functions + * @param r radial grids, shared by all radial functions + * @param radials radial functions, each element is a radial function + * @param l angular momentum quantum number for each radial function + * @param nq number of q-points + * @param dq space between q-points + */ + void build_sbt_tab(const int nr, + const double* r, + const std::vector& radials, + const std::vector& l, + const int nq, + const double dq); + void build_sbt_tab(const std::vector& r, + const std::vector>& radials, + const std::vector& l, + const int nq, + const double dq); + /** + * @brief perform analytical version of the Fourier transform: + * F(q) = int(f(r)*exp(-iq.r) d^3r) + * = 4*pi/sqrt(omega) * (-i)^l * Jl[f](q) * Ylm(q) + * , where Ylm(q) is real spherical harmonic function, and Jl[f](q) is + * the Spherial Bessel Transform of f(r): + * Jl[f](q) = int(f(r)*j_l(q*r)*r^2 dr) + * , where j_l(q*r) is the spherical Bessel function of the first kind. + * . If use another notation, F(q) = , this is denoted as type + * "r" for ket |>, and "l" for bra <|. + */ + + void sbtft(const std::vector>& qs, + std::vector>& out, + const char type = 'r', // 'r' for ket |>, 'l' for bra <| + const double omega = 1.0, + const double tpiba = 1.0); + + private: + std::unique_ptr cubspl_; + std::vector l_; + }; + + /** + * @brief get the mask function for SBFFT + * + * @param mask mask function + */ + void mask_func(std::vector& mask); +} + +#endif // RADIAL_PROJECTION_H diff --git a/source/source_pw/module_proj/unittests/CMakeLists.txt b/source/source_pw/module_proj/unittests/CMakeLists.txt new file mode 100644 index 00000000000..b9b9154c02b --- /dev/null +++ b/source/source_pw/module_proj/unittests/CMakeLists.txt @@ -0,0 +1,5 @@ +AddTest( + TARGET MODULE_PW_radial_proj_test + LIBS parameter base device + SOURCES test_radial_proj.cpp ../radial_proj.cpp +) diff --git a/source/source_pw/module_proj/unittests/test_radial_proj.cpp b/source/source_pw/module_proj/unittests/test_radial_proj.cpp new file mode 100644 index 00000000000..7d0243967ff --- /dev/null +++ b/source/source_pw/module_proj/unittests/test_radial_proj.cpp @@ -0,0 +1,295 @@ +#include "source_pw/module_proj/radial_proj.h" +#include +#include +#include +#include +#include + +#define DOUBLETHRESHOLD 1e-15 + +TEST(RadialProjectionTest, BuildBackwardMapTest) +{ + const std::vector> it2iproj = { + {0, 1}, + {2, 3, 4, 5}, + {6, 7, 8} + }; + const std::vector iproj2l = {0, 1, 0, 1, 2, 3, 0, 1, 2}; + /** + * a backward map will index irow to (it, iproj, m). For the above example, + * it is actually the first atom type has two projectors, the first projector + * has l = 0, the second has l = 1. The former will have only one channel, + * the second will have 3. The second atom type has 4 projectors... + * + * Therefore, it is expected there are 1+3+1+3+5+7+1+3+5 = 29 rows in total. + */ + std::vector itref(4, 0); // four 0, sixteen 1, nine 2 + itref.insert(itref.end(), 16, 1); + itref.insert(itref.end(), 9, 2); + std::vector iprojref(1, 0); + iprojref.insert(iprojref.end(), 3, 1); + iprojref.insert(iprojref.end(), 1, 0); + iprojref.insert(iprojref.end(), 3, 1); + iprojref.insert(iprojref.end(), 5, 2); + iprojref.insert(iprojref.end(), 7, 3); + iprojref.insert(iprojref.end(), 1, 0); + iprojref.insert(iprojref.end(), 3, 1); + iprojref.insert(iprojref.end(), 5, 2); + std::vector mref(1, 0); + mref.insert(mref.end(), {-1, 0, 1}); + mref.insert(mref.end(), {0}); + mref.insert(mref.end(), {-1, 0, 1}); + mref.insert(mref.end(), {-2, -1, 0, 1, 2}); + mref.insert(mref.end(), {-3, -2, -1, 0, 1, 2, 3}); + mref.insert(mref.end(), {0}); + mref.insert(mref.end(), {-1, 0, 1}); + mref.insert(mref.end(), {-2, -1, 0, 1, 2}); + + std::vector irow2it, irow2iproj, irow2m; + RadialProjection::build_backward_map(it2iproj, iproj2l, irow2it, irow2iproj, irow2m); + for(int i = 0; i < 29; ++i) + { + EXPECT_EQ(irow2it[i], itref[i]); + EXPECT_EQ(irow2iproj[i], iprojref[i]); + EXPECT_EQ(irow2m[i], mref[i]); + } +} + +TEST(RadialProjectionTest, BuildForwardMapTest) +{ + // forward means from (it, ia, iproj, m) to irow + const std::vector> it2ia = { + {0, 1}, + {0, 1, 3}, + {12} + }; + const std::vector> it2iproj = { + {0, 1}, + {2, 3, 4, 5}, + {6, 7, 8} + }; + const std::vector iproj2l = {0, 1, 0, 1, 2, 3, 0, 1, 2}; + std::map, int> forward_map; + RadialProjection::build_forward_map(it2ia, it2iproj, iproj2l, forward_map); + // this is a terribly large map, for type 0, there are 2 atoms, each atom has two projectors + // with angular momentum as 0 and 1, respectively. For type 1, there are 3 atoms, each has + // 4 projectors with angular momentum from 0 to 3... + const int nrow = 2*4 + 3*16 + 1*9; + EXPECT_EQ(forward_map.size(), nrow); + // it ia iproj m + // it = 0 + // ia = 0 + // iproj = 0: l = 0 + EXPECT_EQ(forward_map[std::make_tuple(0, 0, 0, 0)], 0); + // iproj = 1: l = 1: m = -1, 0, 1 + EXPECT_EQ(forward_map[std::make_tuple(0, 0, 1, -1)], 1); + EXPECT_EQ(forward_map[std::make_tuple(0, 0, 1, 0)], 2); + EXPECT_EQ(forward_map[std::make_tuple(0, 0, 1, 1)], 3); + // ia = 1 + // iproj = 0: l = 0 + EXPECT_EQ(forward_map[std::make_tuple(0, 1, 0, 0)], 4); + // iproj = 1: l = 1: m = -1, 0, 1 + EXPECT_EQ(forward_map[std::make_tuple(0, 1, 1, -1)], 5); + EXPECT_EQ(forward_map[std::make_tuple(0, 1, 1, 0)], 6); + EXPECT_EQ(forward_map[std::make_tuple(0, 1, 1, 1)], 7); + // it = 1 + // ia = 0 + // iproj = 0: l = 0 + EXPECT_EQ(forward_map[std::make_tuple(1, 0, 0, 0)], 8); + // iproj = 1: l = 1: m = -1, 0, 1 + EXPECT_EQ(forward_map[std::make_tuple(1, 0, 1, -1)], 9); + EXPECT_EQ(forward_map[std::make_tuple(1, 0, 1, 0)], 10); + EXPECT_EQ(forward_map[std::make_tuple(1, 0, 1, 1)], 11); + // iproj = 2: l = 2: m = -2, -1, 0, 1, 2 + EXPECT_EQ(forward_map[std::make_tuple(1, 0, 2, -2)], 12); + EXPECT_EQ(forward_map[std::make_tuple(1, 0, 2, -1)], 13); + EXPECT_EQ(forward_map[std::make_tuple(1, 0, 2, 0)], 14); + EXPECT_EQ(forward_map[std::make_tuple(1, 0, 2, 1)], 15); + EXPECT_EQ(forward_map[std::make_tuple(1, 0, 2, 2)], 16); + // iproj = 3: l = 3: m = -3, -2, -1, 0, 1, 2, 3 + EXPECT_EQ(forward_map[std::make_tuple(1, 0, 3, -3)], 17); + EXPECT_EQ(forward_map[std::make_tuple(1, 0, 3, -2)], 18); + EXPECT_EQ(forward_map[std::make_tuple(1, 0, 3, -1)], 19); + EXPECT_EQ(forward_map[std::make_tuple(1, 0, 3, 0)], 20); + EXPECT_EQ(forward_map[std::make_tuple(1, 0, 3, 1)], 21); + EXPECT_EQ(forward_map[std::make_tuple(1, 0, 3, 2)], 22); + EXPECT_EQ(forward_map[std::make_tuple(1, 0, 3, 3)], 23); + // ia = 1... +} + +TEST(RadialProjectionTest, MaskfunctionGenerationTest) +{ + std::vector mask; + RadialProjection::mask_func(mask); + EXPECT_EQ(mask.size(), 201); + EXPECT_EQ(mask[0], 1.0); // the rescaled value of the mask function, at 0, is 1 + EXPECT_NEAR(mask[200], 0.98138215E-05, 1e-10); // real space cut, at rc, is 0 +} + +TEST(RadialProjectionTest, BuildSbtTabCorrectnessTest) +{ + RadialProjection::RadialProjector rp; + + // use mask function as the example + std::vector mask; + RadialProjection::mask_func(mask); + // suppose the r from 0 to 2.0 (inclusive) with 0.01 step + std::vector r(mask.size()); + std::iota(r.begin(), r.end(), 0); + std::for_each(r.begin(), r.end(), [](double& x){x = x * 0.01;}); + + // fill into the container + std::vector> radials(1, mask); + + // suppose the angular momentum of function to transform is 0 + // but this is true because the mask function is really spherically + // symmetric. + std::vector l(1, 0); + + // build the interpolation table + rp.build_sbt_tab(r, radials, l, 201, 0.01); // build an interpolation table + + // only one q point: (0, 0, 0), is Gamma + std::vector> q(1); + q[0] = ModuleBase::Vector3(0.0, 0.0, 0.0); + + // so the output should have the same size as q + std::vector> out(q.size()); + + // transform, the last argument is 'r' for . + // for q = 0, the integration is actually performed on the function + // itself, so it is actually the integration of the function, on the + // grid from 0 to rc (2.0) with stepsize 0.01. + rp.sbtft(q, out, 'r', 1.0, 1.0); + // print, each 5 numbers in a row + + /** + * The following Python code is used to generate the reference data + * from scipy.integrate import simps + * import numpy as np + * + * r = np.arange(0, 2.01, 0.01) + * mask = np.array(mask) + * omega = 1 + * # Y00 is 1/sqrt(4*pi) + * val = simps(mask*r**2, x=r) * np.sqrt(4*np.pi) + * print(val/np.sqrt(omega)) + */ + std::complex ref(0.6143605159327622, 0.0); + EXPECT_NEAR(out[0].real(), ref.real(), DOUBLETHRESHOLD); +} + +TEST(RadialProjectionTest, BuildSbtTabStabilityTest) +{ + // still use mask function but scale with different Gaussian function + std::vector mask; + RadialProjection::mask_func(mask); + // suppose the r from 0 to 2.0 (inclusive) with 0.01 step + std::vector r(mask.size()); + std::iota(r.begin(), r.end(), 0); + std::for_each(r.begin(), r.end(), [](double& x){x = x * 0.01;}); + + std::vector aug1(mask.size()); + std::vector aug2(mask.size()); + std::vector aug3(mask.size()); + const double sigma1 = 0.1; + const double sigma2 = 1.0; + const double sigma3 = 10.0; + std::transform(r.begin(), r.end(), aug1.begin(), [sigma1](double x){return std::exp(-x*x/(2*sigma1*sigma1));}); + std::transform(r.begin(), r.end(), aug2.begin(), [sigma2](double x){return std::exp(-x*x/(2*sigma2*sigma2));}); + std::transform(r.begin(), r.end(), aug3.begin(), [sigma3](double x){return std::exp(-x*x/(2*sigma3*sigma3));}); + + std::vector mask1(mask.size()); + std::vector mask2(mask.size()); + std::vector mask3(mask.size()); + std::transform(mask.begin(), mask.end(), aug1.begin(), mask1.begin(), std::multiplies()); + std::transform(mask.begin(), mask.end(), aug2.begin(), mask2.begin(), std::multiplies()); + std::transform(mask.begin(), mask.end(), aug3.begin(), mask3.begin(), std::multiplies()); + + // then prepare the q vectors, in total 50, each with 3 components among which each is random + // number ranging from 0 to 2, use the random seed as 1 + const int npw = 50; + std::vector> q(npw); + std::mt19937 gen(1); + std::uniform_real_distribution dis(0.0, 2.0); + std::for_each(q.begin(), q.end(), [&gen, &dis](ModuleBase::Vector3& x){ + x[0] = dis(gen); + x[1] = dis(gen); + x[2] = dis(gen); + }); + + // fill into the container + std::vector> radials(4); // do not remember mask0 = mask + radials[0] = mask; + radials[1] = mask1; + radials[2] = mask2; + radials[3] = mask3; + + // angular momentum set as 0, 1, 2, 3 + std::vector l(4); + std::iota(l.begin(), l.end(), 0); + + RadialProjection::RadialProjector rp; + rp.build_sbt_tab(r, radials, l, 401, 0.01); // build an interpolation table + + // then perform the transform + std::vector> out; + rp.sbtft(q, out, 'r', 1.0, 1.0); + + // there are 50 q points, and radials have l as 0, 1, 2, 3, which means the + // number of lm components in total is 1 + 3 + 5 + 7 = 16 + // therefore there should be 50 * 16 = 800 complex numbers in the output + EXPECT_EQ(out.size(), npw*16); + // check if they are listed in expected sequence, say the first should be + // Fourier transform of the first function, mask, l = 0, m = 0 + RadialProjection::RadialProjector rp1; + rp1.build_sbt_tab(r, + std::vector>(1, mask), + std::vector(1, 0), 401, 0.01); + std::vector> out1; + rp1.sbtft(q, out1, 'r', 1.0, 1.0); + for(int iq = 0; iq < 50; ++iq) // 50 q-points + { + EXPECT_NEAR(out[iq].real(), out1[iq].real(), DOUBLETHRESHOLD); + EXPECT_NEAR(out[iq].imag(), out1[iq].imag(), DOUBLETHRESHOLD); + } + RadialProjection::RadialProjector rp2; + rp2.build_sbt_tab(r, + std::vector>(1, mask1), + std::vector(1, 1), 401, 0.01); + std::vector> out2; + rp2.sbtft(q, out2, 'r', 1.0, 1.0); + for(int iq = 0; iq < 50*(2*1+1); ++iq) // 50 q-points + { + EXPECT_NEAR(out[iq+50].real(), out2[iq].real(), DOUBLETHRESHOLD); + EXPECT_NEAR(out[iq+50].imag(), out2[iq].imag(), DOUBLETHRESHOLD); + } + RadialProjection::RadialProjector rp3; + rp3.build_sbt_tab(r, + std::vector>(1, mask2), + std::vector(1, 2), 401, 0.01); + std::vector> out3; + rp3.sbtft(q, out3, 'r', 1.0, 1.0); + for(int iq = 0; iq < 50*(2*2+1); ++iq) // 50 q-points + { + EXPECT_NEAR(out[iq+50+50*(2*1+1)].real(), out3[iq].real(), DOUBLETHRESHOLD); + EXPECT_NEAR(out[iq+50+50*(2*1+1)].imag(), out3[iq].imag(), DOUBLETHRESHOLD); + } + RadialProjection::RadialProjector rp4; + rp4.build_sbt_tab(r, + std::vector>(1, mask3), + std::vector(1, 3), 401, 0.01); + std::vector> out4; + rp4.sbtft(q, out4, 'r', 1.0, 1.0); + for(int iq = 0; iq < 50*(2*3+1); ++iq) // 50 q-points + { + EXPECT_NEAR(out[iq+50+50*(2*1+1)+50*(2*2+1)].real(), out4[iq].real(), DOUBLETHRESHOLD); + EXPECT_NEAR(out[iq+50+50*(2*1+1)+50*(2*2+1)].imag(), out4[iq].imag(), DOUBLETHRESHOLD); + } +} + +int main() +{ + testing::InitGoogleTest(); + return RUN_ALL_TESTS(); +} diff --git a/source/source_pw/module_pwdft/CMakeLists.txt b/source/source_pw/module_pwdft/CMakeLists.txt index f6b8597c806..cf652068f3d 100644 --- a/source/source_pw/module_pwdft/CMakeLists.txt +++ b/source/source_pw/module_pwdft/CMakeLists.txt @@ -2,9 +2,16 @@ list(APPEND objects hamilt_pw.cpp + td_history_pw.cpp + td_pw.cpp op_pw_ekin.cpp + op_pw_ekin_td.cpp op_pw_veff.cpp + projector_gradient.cpp + velocity_workspace.cpp + nonlocal_workspace.cpp op_pw_nl.cpp + op_pw_nl_td.cpp op_pw_meta.cpp op_pw_vel.cpp op_pw.cpp @@ -12,6 +19,16 @@ list(APPEND objects op_pw_exx.cpp op_pw_exx_ace.cpp op_pw_exx_pot.cpp + kernels/exx_batch_op.cpp +) + +if(ENABLE_FLOAT_FFTW) + list(APPEND objects kernels/exx_batch_op_float.cpp) +else() + list(APPEND objects kernels/exx_batch_op_float_stub.cpp) +endif() + +list(APPEND objects dftu_base.cpp dftu_base_io.cpp dftu_pw.cpp @@ -50,6 +67,7 @@ list(APPEND objects vnl_pw_grad.cpp vnl_pw_init_vnl.cpp vnl_pw_qrad.cpp + vnl_pw_table.cpp stru_fac.cpp stru_fac_k.cpp soc.cpp @@ -59,17 +77,8 @@ list(APPEND objects fs_nonlocal_tools_stress.cpp fs_nonlocal_tools_force.cpp fs_kin_tools.cpp - radial_proj.cpp - onsite_proj.cpp - onsite_proj_init.cpp - onsite_proj_overlap.cpp + nonlocal_maths.cpp onsite_proj_force_stress.cpp - onsite_proj_tools.cpp - onsite_proj_tools_becp.cpp - onsite_proj_tools_dbecp.cpp - onsite_proj_tools_force.cpp - onsite_proj_tools_stress.cpp - onsite_proj_print.cpp vsep_pw.cpp ) diff --git a/source/source_pw/module_pwdft/deltaspin_pw.cpp b/source/source_pw/module_pwdft/deltaspin_pw.cpp index 53966859998..bd14a63920a 100644 --- a/source/source_pw/module_pwdft/deltaspin_pw.cpp +++ b/source/source_pw/module_pwdft/deltaspin_pw.cpp @@ -1,6 +1,6 @@ #include "source_pw/module_pwdft/deltaspin_pw.h" #include "source_lcao/module_deltaspin/spin_constrain.h" -#include "source_estate/module_charge/charge_mixing.h" +#include "source_estate/module_charge/chg_mix.h" #include "source_io/module_parameter/parameter.h" namespace pw diff --git a/source/source_pw/module_pwdft/dftu_base.cpp b/source/source_pw/module_pwdft/dftu_base.cpp index 3f5f5ca3eb7..41f6dd993d3 100644 --- a/source/source_pw/module_pwdft/dftu_base.cpp +++ b/source/source_pw/module_pwdft/dftu_base.cpp @@ -99,18 +99,14 @@ void Plus_U_Base::init_base(UnitCell& cell, for (int l = 0; l <= cell.atoms[it].nwl; l++) { - const int N = cell.atoms[it].l_nchi[l]; - - for (int n = 0; n < N; n++) + // only the first radial channel (n=0) of each l is stored + if (nspin == 1 || nspin == 2) + { + num_locale += (2 * l + 1) * (2 * l + 1) * 2; + } + else if (nspin == 4) { - if (nspin == 1 || nspin == 2) - { - num_locale += (2 * l + 1) * (2 * l + 1) * 2; - } - else if (nspin == 4) - { - num_locale += (2 * l + 1) * (2 * l + 1) * npol * npol; - } + num_locale += (2 * l + 1) * (2 * l + 1) * npol * npol; } } } diff --git a/source/source_pw/module_pwdft/dftu_base_io.cpp b/source/source_pw/module_pwdft/dftu_base_io.cpp index c74614d349d..225b0fdb4c0 100644 --- a/source/source_pw/module_pwdft/dftu_base_io.cpp +++ b/source/source_pw/module_pwdft/dftu_base_io.cpp @@ -167,6 +167,12 @@ void read_occup_m(const UnitCell& ucell, ifdftu >> zeta; ifdftu.ignore(150, '\n'); + if (zeta != 0) + { + ModuleBase::WARNING_QUIT("DFTU_BASE::read_occup_m", + "only the first radial channel (ORBITAL=0) is supported"); + } + T = ucell.iat2it[iat]; const int NL = ucell.atoms[T].nwl + 1; @@ -194,7 +200,7 @@ void read_occup_m(const UnitCell& ucell, for (int m1 = 0; m1 < 2 * L + 1; m1++) { ifdftu >> value; - occ.set(iat, L, zeta, spin, m0, m1, value); + occ.set(iat, L, spin, m0, m1, value); } ifdftu.ignore(150, '\n'); } @@ -220,7 +226,7 @@ void read_occup_m(const UnitCell& ucell, { int m1_all = m1 + (2 * L + 1) * ipol1; ifdftu >> value; - occ.set(iat, L, zeta, 0, m0_all, m1_all, value); + occ.set(iat, L, 0, m0_all, m1_all, value); } } ifdftu.ignore(150, '\n'); @@ -278,27 +284,19 @@ void local_occup_bcast(const UnitCell& ucell, continue; } - for (int n = 0; n < ucell.atoms[T].l_nchi[l]; n++) + if (nspin == 1 || nspin == 2) { - if (n != 0) + for (int spin = 0; spin < 2; spin++) { - continue; - } - - if (nspin == 1 || nspin == 2) - { - for (int spin = 0; spin < 2; spin++) - { - Parallel_Common::bcast_double(occ.mat(iat, l, n, spin).c, - occ.mat(iat, l, n, spin).nr * occ.mat(iat, l, n, spin).nc); - } - } - else if (nspin == 4) // SOC - { - Parallel_Common::bcast_double(occ.mat(iat, l, n, 0).c, - occ.mat(iat, l, n, 0).nr * occ.mat(iat, l, n, 0).nc); + Parallel_Common::bcast_double(occ.mat(iat, l, spin).c, + occ.mat(iat, l, spin).nr * occ.mat(iat, l, spin).nc); } } + else if (nspin == 4) // SOC + { + Parallel_Common::bcast_double(occ.mat(iat, l, 0).c, + occ.mat(iat, l, 0).nr * occ.mat(iat, l, 0).nc); + } } } } @@ -342,18 +340,11 @@ void output(const Plus_U_Base& dftu, } else { - for (int n = 0; n < N; n++) - { - if (n != 0) - { - continue; - } - double Ueff = (dftu.yukawa().get_U(T, L, n) - dftu.yukawa().get_J(T, L, n)) * ModuleBase::Ry_to_eV; - GlobalV::ofs_running << " Type=" << T+1 << " L=" << L << " ORBITAL=" << n - << " U=" << dftu.yukawa().get_U(T, L, n) * ModuleBase::Ry_to_eV << " eV" - << " J=" << dftu.yukawa().get_J(T, L, n) * ModuleBase::Ry_to_eV << " eV" - << std::endl; - } + double Ueff = (dftu.yukawa().get_U(T, L) - dftu.yukawa().get_J(T, L)) * ModuleBase::Ry_to_eV; + GlobalV::ofs_running << " Type=" << T+1 << " L=" << L << " ORBITAL=" << 0 + << " U=" << dftu.yukawa().get_U(T, L) * ModuleBase::Ry_to_eV << " eV" + << " J=" << dftu.yukawa().get_J(T, L) * ModuleBase::Ry_to_eV << " eV" + << std::endl; } } } @@ -417,121 +408,111 @@ void write_occup_m(const Plus_U_Base& dftu, continue; } - const int N = ucell.atoms[T].l_nchi[l]; + ofs << "\n Atom= " << iat+1; + ofs << " L= " << l; + ofs << " ORBITAL= " << 0 << std::endl; - for (int n = 0; n < N; n++) + if (nspin == 1 || nspin == 2) { - if (n != 0) - { - continue; - } - - ofs << "\n Atom= " << iat+1; - ofs << " L= " << l; - ofs << " ORBITAL= " << n << std::endl; - - if (nspin == 1 || nspin == 2) + double sum0[2]; + for (int is = 0; is < 2; is++) { - double sum0[2]; - for (int is = 0; is < 2; is++) + if (diag) { - if (diag) + std::vector> A(2 * l + 1, std::vector(2 * l + 1)); + for (int m0 = 0; m0 < 2 * l + 1; m0++) { - std::vector> A(2 * l + 1, std::vector(2 * l + 1)); - for (int m0 = 0; m0 < 2 * l + 1; m0++) - { - for (int m1 = 0; m1 < 2 * l + 1; m1++) - { - A[m0][m1] = dftu.occmat().get(iat, l, n, is, m0, m1); - } - } - std::vector eigenvalues = CalculateEigenvalues(A, 2 * l + 1); - sum0[is] = 0.0; - ofs << " Eigenvalues for spin=" << is+1 << std::endl; - ofs << std::setprecision(8) << std::fixed; - for (int i = 0; i < 2 * l + 1; i++) + for (int m1 = 0; m1 < 2 * l + 1; m1++) { - ofs << std::setw(12) << eigenvalues[i]; - sum0[is] += eigenvalues[i]; + A[m0][m1] = dftu.occmat().get(iat, l, is, m0, m1); } - ofs << std::endl; - ofs << " sum is " << std::setw(12) << sum0[is] << std::endl; } - ofs << " spin= " << is+1 << std::endl; + std::vector eigenvalues = CalculateEigenvalues(A, 2 * l + 1); + sum0[is] = 0.0; + ofs << " Eigenvalues for spin=" << is+1 << std::endl; ofs << std::setprecision(8) << std::fixed; - for (int m0 = 0; m0 < 2 * l + 1; m0++) + for (int i = 0; i < 2 * l + 1; i++) { - for (int m1 = 0; m1 < 2 * l + 1; m1++) - { - ofs << std::setw(12) - << dftu.occmat().get(iat, l, n, is, m0, m1); - } - ofs << std::endl; + ofs << std::setw(12) << eigenvalues[i]; + sum0[is] += eigenvalues[i]; } + ofs << std::endl; + ofs << " sum is " << std::setw(12) << sum0[is] << std::endl; } - if (diag) + ofs << " spin= " << is+1 << std::endl; + ofs << std::setprecision(8) << std::fixed; + for (int m0 = 0; m0 < 2 * l + 1; m0++) { - ofs << std::setw(12) << std::setprecision(8) - << std::fixed << " Magnetism for atom " << iat+1 << ": " << sum0[0] - sum0[1] - << std::endl; + for (int m1 = 0; m1 < 2 * l + 1; m1++) + { + ofs << std::setw(12) + << dftu.occmat().get(iat, l, is, m0, m1); + } + ofs << std::endl; } } - else if (nspin == 4) // SOC + if (diag) { - if (diag) + ofs << std::setw(12) << std::setprecision(8) + << std::fixed << " Magnetism for atom " << iat+1 << ": " << sum0[0] - sum0[1] + << std::endl; + } + } + else if (nspin == 4) // SOC + { + if (diag) + { + double sum0[4]; + std::vector> A(2 * l + 1, std::vector(2 * l + 1)); + int index = 0; + for (int is = 0; is < 4; is++) { - double sum0[4]; - std::vector> A(2 * l + 1, std::vector(2 * l + 1)); - int index = 0; - for (int is = 0; is < 4; is++) + for (int m0 = 0; m0 < 2 * l + 1; m0++) { - for (int m0 = 0; m0 < 2 * l + 1; m0++) - { - for (int m1 = 0; m1 < 2 * l + 1; m1++) - { - A[m0][m1] = dftu.occmat().get(iat, l, n, 0, m0, m1); - index++; - } - } - std::vector eigenvalues = CalculateEigenvalues(A, 2 * l + 1); - sum0[is] = 0.0; - ofs << " Eigenvalues for is=" << is << std::endl; - ofs << std::setprecision(8) << std::fixed; - for (int i = 0; i < 2 * l + 1; i++) + for (int m1 = 0; m1 < 2 * l + 1; m1++) { - ofs << std::setw(12) << eigenvalues[i]; - sum0[is] += eigenvalues[i]; + A[m0][m1] = dftu.occmat().get(iat, l, 0, m0, m1); + index++; } - ofs << std::endl; - ofs << " sum is " << std::setw(12) << sum0[is] << std::endl; } - ofs << std::setw(12) << std::setprecision(8) - << std::fixed << " Magnetism for atom " << iat + 1 << ": " - << sum0[1] << " " << sum0[2] << " " << sum0[3] << std::endl; + std::vector eigenvalues = CalculateEigenvalues(A, 2 * l + 1); + sum0[is] = 0.0; + ofs << " Eigenvalues for is=" << is << std::endl; + ofs << std::setprecision(8) << std::fixed; + for (int i = 0; i < 2 * l + 1; i++) + { + ofs << std::setw(12) << eigenvalues[i]; + sum0[is] += eigenvalues[i]; + } + ofs << std::endl; + ofs << " sum is " << std::setw(12) << sum0[is] << std::endl; } - else + ofs << std::setw(12) << std::setprecision(8) + << std::fixed << " Magnetism for atom " << iat + 1 << ": " + << sum0[1] << " " << sum0[2] << " " << sum0[3] << std::endl; + } + else + { + for (int m0 = 0; m0 < 2 * l + 1; m0++) { - for (int m0 = 0; m0 < 2 * l + 1; m0++) + for (int ipol0 = 0; ipol0 < npol; ipol0++) { - for (int ipol0 = 0; ipol0 < npol; ipol0++) - { - const int m0_all = m0 + (2 * l + 1) * ipol0; + const int m0_all = m0 + (2 * l + 1) * ipol0; - for (int m1 = 0; m1 < 2 * l + 1; m1++) + for (int m1 = 0; m1 < 2 * l + 1; m1++) + { + for (int ipol1 = 0; ipol1 < npol; ipol1++) { - for (int ipol1 = 0; ipol1 < npol; ipol1++) - { - int m1_all = m1 + (2 * l + 1) * ipol1; - ofs << std::setw(12) << std::setprecision(8) << std::fixed - << dftu.occmat().get(iat, l, n, 0, m0_all, m1_all); - } + int m1_all = m1 + (2 * l + 1) * ipol1; + ofs << std::setw(12) << std::setprecision(8) << std::fixed + << dftu.occmat().get(iat, l, 0, m0_all, m1_all); } - ofs << std::endl; } + ofs << std::endl; } } } - } // n + } } // l } // I } // T diff --git a/source/source_pw/module_pwdft/dftu_base_io.h b/source/source_pw/module_pwdft/dftu_base_io.h index 291602f3a22..efa721d74b6 100644 --- a/source/source_pw/module_pwdft/dftu_base_io.h +++ b/source/source_pw/module_pwdft/dftu_base_io.h @@ -14,8 +14,8 @@ class UnitCell; namespace DFTU_BASE { -/// nested occupation-matrix type used by DFT+U: occ_mat[iat][l][n][spin](m0, m1) -using OccMatData = std::vector>>>; +/// nested occupation-matrix type used by DFT+U: occ_mat[iat][l][spin](m0, m1) +using OccMatData = std::vector>>; /// Read the local occupation number matrix from file (rank 0 only). /// diff --git a/source/source_pw/module_pwdft/dftu_pw.cpp b/source/source_pw/module_pwdft/dftu_pw.cpp index 53a48b39bc5..e7eda6c55dd 100644 --- a/source/source_pw/module_pwdft/dftu_pw.cpp +++ b/source/source_pw/module_pwdft/dftu_pw.cpp @@ -3,9 +3,9 @@ #include "source_pw/module_pwdft/dftu_pw_tools.h" #include "source_pw/module_pwdft/dftu_base.h" #include "source_pw/module_pwdft/dftu_base_io.h" -#include "source_pw/module_pwdft/onsite_proj.h" +#include "source_pw/module_proj/onsite_proj.h" #include "source_cell/unitcell.h" -#include "source_estate/module_charge/charge_mixing.h" +#include "source_estate/module_charge/chg_mix.h" #include "source_base/timer.h" namespace DFTU_BASE { @@ -50,14 +50,14 @@ void accumulate_occ_one_k(const void* psi_in, if(nspin == 4) { pw::accumulate_occ_spinor( - occmat.mat(iat, target_l, 0, 0).c, + occmat.mat(iat, target_l, 0).c, becp, nbands, npol, nkb, begin_ih, m_begin, tlp1, wg_in, ik); } else // nspin=1 or nspin=2 { pw::accumulate_occ_scalar( - occmat.mat(iat, target_l, 0, is).c, + occmat.mat(iat, target_l, is).c, becp, nbands, nkb, begin_ih, m_begin, tlp1, wg_in, ik); } diff --git a/source/source_pw/module_pwdft/dftu_pw_tools.cpp b/source/source_pw/module_pwdft/dftu_pw_tools.cpp index e3e8c860960..b068df38654 100644 --- a/source/source_pw/module_pwdft/dftu_pw_tools.cpp +++ b/source/source_pw/module_pwdft/dftu_pw_tools.cpp @@ -171,13 +171,13 @@ void reduce_occ_mat(const UnitCell& cell, { Parallel_Reduce::reduce_double_allpool(kpar, GlobalV::NPROC_IN_POOL, - occmat.mat(iat, target_l, 0, 0).c, + occmat.mat(iat, target_l, 0).c, size); if(nspin == 2) { Parallel_Reduce::reduce_double_allpool(kpar, GlobalV::NPROC_IN_POOL, - occmat.mat(iat, target_l, 0, 1).c, + occmat.mat(iat, target_l, 1).c, size); } } @@ -185,7 +185,7 @@ void reduce_occ_mat(const UnitCell& cell, { Parallel_Reduce::reduce_double_allpool(kpar, GlobalV::NPROC_IN_POOL, - occmat.mat(iat, target_l, 0, 0).c, + occmat.mat(iat, target_l, 0).c, size * 4); } } @@ -229,7 +229,7 @@ void compute_pot_uterm_and_energy(const UnitCell& cell, // contiguously, each of size m_size*m_size. energy_u += compute_pot_onsite_spinor( pot_onsite_iat, - occmat.mat(iat, target_l, 0, 0).c, + occmat.mat(iat, target_l, 0).c, u_value, diag_coeff, weight_eu, m_size); } else // nspin=1 or nspin=2 @@ -237,7 +237,7 @@ void compute_pot_uterm_and_energy(const UnitCell& cell, // spin-up channel energy_u += compute_pot_onsite_scalar( pot_onsite_iat, - occmat.mat(iat, target_l, 0, 0).c, + occmat.mat(iat, target_l, 0).c, u_value, diag_coeff, weight_eu, m_size); // spin-down channel for nspin=2 if(nspin == 2) @@ -245,7 +245,7 @@ void compute_pot_uterm_and_energy(const UnitCell& cell, std::complex* pot_onsite_iat1 = &(uterm_mat[uterm_mat.size()/2 + uterm_mat_index[iat]]); energy_u += compute_pot_onsite_scalar( pot_onsite_iat1, - occmat.mat(iat, target_l, 0, 1).c, + occmat.mat(iat, target_l, 1).c, u_value, diag_coeff, weight_eu, m_size); } } diff --git a/source/source_pw/module_pwdft/dftu_pw_tools.h b/source/source_pw/module_pwdft/dftu_pw_tools.h index 2b123406ff8..23edb2871b3 100644 --- a/source/source_pw/module_pwdft/dftu_pw_tools.h +++ b/source/source_pw/module_pwdft/dftu_pw_tools.h @@ -34,7 +34,7 @@ void pauli_to_spin_basis(std::complex* pot_onsite, int m_size); /// to convert pot_onsite to spin basis in-place. /// /// pot_onsite: pointer to uterm_mat[uterm_mat_index[iat]] -/// occ: pointer to occ_mat[iat][target_l][0][0].c (4 Pauli blocks packed) +/// occ: pointer to occ_mat[iat][target_l][0].c (4 Pauli blocks packed) double compute_pot_onsite_spinor( std::complex* pot_onsite, const double* occ, @@ -47,7 +47,7 @@ double compute_pot_onsite_spinor( /// (nspin==1 or nspin==2). Returns the energy_u increment. /// /// pot_onsite: pointer to the spin channel's pot_onsite block (size m_size * m_size) -/// occ: pointer to occ_mat[iat][target_l][0][is].c for this channel +/// occ: pointer to occ_mat[iat][target_l][is].c for this channel double compute_pot_onsite_scalar( std::complex* pot_onsite, const double* occ, @@ -58,7 +58,7 @@ double compute_pot_onsite_scalar( /// accumulate occ_mat from becp for one atom, one k-point (nspin==4, spinor). /// -/// occ_mat_out points to occ_mat[iat][target_l][0][0].c, which packs 4 +/// occ_mat_out points to occ_mat[iat][target_l][0].c, which packs 4 /// Pauli blocks contiguously (each of size tlp1*tlp1). The function adds /// the contributions from all nbands bands for the given k-point. /// @@ -82,7 +82,7 @@ void accumulate_occ_spinor( /// accumulate occ_mat from becp for one atom, one k-point (nspin==1 or 2). /// -/// occ_mat_out points to occ_mat[iat][target_l][0][is].c, a single channel +/// occ_mat_out points to occ_mat[iat][target_l][is].c, a single channel /// of size tlp1*tlp1. The caller selects the spin channel by passing the /// corresponding occ_mat pointer; this function does not need is. /// Adds contributions from all nbands bands. diff --git a/source/source_pw/module_pwdft/force_pw.cpp b/source/source_pw/module_pwdft/force_pw.cpp index 787eb528bd1..d37e2379954 100644 --- a/source/source_pw/module_pwdft/force_pw.cpp +++ b/source/source_pw/module_pwdft/force_pw.cpp @@ -14,6 +14,7 @@ #include "source_base/timer.h" #include "source_base/tool_threading.h" #include "source_base/tool_quit.h" +#include "source_cell/module_symmetry/symmetry.h" #include "source_estate/module_pot/efield.h" #include "source_estate/module_pot/gatefield.h" #include "source_hamilt/module_ewald/h_ewald_pw.h" @@ -26,7 +27,12 @@ template -void Forces::cal_force(UnitCell& ucell, +void Forces::cal_force(const int nspin, + const bool domag, + const bool domag_z, + const int gga_grad, + const bool use_onsite_projection, + UnitCell& ucell, ModuleBase::matrix& force, const vdw::VdwResult* vdw_result, const elecstate::ElecState& elec, @@ -76,14 +82,15 @@ void Forces::cal_force(UnitCell& ucell, // DFT+U and DeltaSpin // here maybe a bug when OFDFT calls +U, mohan add 20251107 - if(PARAM.inp.dft_plus_u || PARAM.inp.sc_mag_switch) + if(use_onsite_projection) { this->cal_force_onsite(forceonsite, wg, wfc_basis, ucell, *p_dftu, psi_in); } } // non-linear core correction - Forces::cal_force_cc(forcecc, rho_basis, chr, locpp->numeric, ucell); + Forces::cal_force_cc(forcecc, rho_basis, chr, locpp->numeric, ucell, + nspin, domag, domag_z, gga_grad); // force due to core charge this->cal_force_scc(forcescc, rho_basis, elec.vnew, elec.vnew_exist, locpp->numeric, ucell); @@ -134,20 +141,18 @@ void Forces::cal_force(UnitCell& ucell, if (PARAM.inp.imp_sol) { forcesol.create(this->nat, 3); - solvent.cal_force_sol(ucell, rho_basis, locpp->vloc, PARAM.inp.nspin, forcesol); + solvent.cal_force_sol(ucell, rho_basis, locpp->vloc, nspin, forcesol); if (PARAM.inp.test_force) { ModuleIO::print_force(GlobalV::ofs_running, ucell, "IMP_SOL FORCE (Ry/Bohr)", forcesol); } } - // impose total force = 0 + // sum all force terms into the total force int iat = 0; for (int ipol = 0; ipol < 3; ipol++) { - double sum = 0.0; iat = 0; - for (int it = 0; it < ucell.ntype; it++) { for (int ia = 0; ia < ucell.atoms[it].na; ia++) @@ -175,79 +180,28 @@ void Forces::cal_force(UnitCell& ucell, force(iat, ipol) = force(iat, ipol) + forcesol(iat, ipol); } - if(PARAM.inp.dft_plus_u || PARAM.inp.sc_mag_switch) + if(use_onsite_projection) { force(iat, ipol) += forceonsite(iat, ipol); } - sum += force(iat, ipol); - iat++; } } - - if (!(PARAM.inp.gate_flag || PARAM.inp.efield_flag)) - { - double compen = sum / this->nat; - for (int iat = 0; iat < this->nat; ++iat) - { - force(iat, ipol) = force(iat, ipol) - compen; - } - } } if (PARAM.inp.gate_flag || PARAM.inp.efield_flag) { GlobalV::ofs_running << "Atomic forces are not shifted if gate_flag or efield_flag == true!" << std::endl; } + else + { + ModuleBase::remove_net_force(this->nat, force); + } if (ModuleSymmetry::Symmetry::symm_flag == 1) { - double d1 = 0.0, d2 = 0.0, d3 = 0.0; - for (int iat = 0; iat < this->nat; iat++) - { - ModuleBase::Mathzone::Cartesian_to_Direct(force(iat, 0), - force(iat, 1), - force(iat, 2), - ucell.a1.x, - ucell.a1.y, - ucell.a1.z, - ucell.a2.x, - ucell.a2.y, - ucell.a2.z, - ucell.a3.x, - ucell.a3.y, - ucell.a3.z, - d1, - d2, - d3); - - force(iat, 0) = d1; - force(iat, 1) = d2; - force(iat, 2) = d3; - } - p_symm->symmetrize_vec3_nat(force.c); - for (int iat = 0; iat < this->nat; iat++) - { - ModuleBase::Mathzone::Direct_to_Cartesian(force(iat, 0), - force(iat, 1), - force(iat, 2), - ucell.a1.x, - ucell.a1.y, - ucell.a1.z, - ucell.a2.x, - ucell.a2.y, - ucell.a2.z, - ucell.a3.x, - ucell.a3.y, - ucell.a3.z, - d1, - d2, - d3); - force(iat, 0) = d1; - force(iat, 1) = d2; - force(iat, 2) = d3; - } + ModuleSymmetry::symmetrize_force_cartesian(p_symm, this->nat, ucell.a1, ucell.a2, ucell.a3, force); } GlobalV::ofs_running << std::setiosflags(std::ios::fixed) << std::setprecision(6) << std::endl; @@ -313,7 +267,7 @@ void Forces::cal_force(UnitCell& ucell, forcesol, false); } - if (PARAM.inp.dft_plus_u || PARAM.inp.sc_mag_switch) + if (use_onsite_projection) { ModuleIO::print_force(GlobalV::ofs_running, ucell, diff --git a/source/source_pw/module_pwdft/force_pw.h b/source/source_pw/module_pwdft/force_pw.h index 3a2f3c5fa9f..6aeaff6069e 100644 --- a/source/source_pw/module_pwdft/force_pw.h +++ b/source/source_pw/module_pwdft/force_pw.h @@ -45,7 +45,12 @@ class Forces Forces(const int nat_in) : nat(nat_in){}; ~Forces(){}; - void cal_force(UnitCell& ucell, + void cal_force(const int nspin, + const bool domag, + const bool domag_z, + const int gga_grad, + const bool use_onsite_projection, + UnitCell& ucell, ModuleBase::matrix& force, const vdw::VdwResult* vdw_result, const elecstate::ElecState& elec, @@ -77,7 +82,11 @@ class Forces const ModulePW::PW_Basis* const rho_basis, const Charge* const chr, const bool* numeric, - UnitCell& ucell_in); + UnitCell& ucell_in, + const int nspin, + const bool domag, + const bool domag_z, + const int gga_grad); /** * @brief This routine computes the atomic force of non-local pseudopotential * F^{NL}_i = \sum_{n,k}f_{nk}\sum_I \sum_{lm,l'm'}D_{l,l'}^{I} [ diff --git a/source/source_pw/module_pwdft/force_pw_cc.cpp b/source/source_pw/module_pwdft/force_pw_cc.cpp index 56575ea0925..a0d0b944e4c 100644 --- a/source/source_pw/module_pwdft/force_pw_cc.cpp +++ b/source/source_pw/module_pwdft/force_pw_cc.cpp @@ -1,7 +1,6 @@ #include "force_pw.h" #include "stress_func.h" #include "source_base/parallel_reduce.h" -#include "source_io/module_parameter/parameter.h" // new #include "source_base/complexmatrix.h" #include "source_base/libm/libm.h" @@ -32,7 +31,11 @@ void Forces::cal_force_cc(ModuleBase::matrix& forcecc, const ModulePW::PW_Basis* const rho_basis, const Charge* const chr, const bool* numeric, - UnitCell& ucell_in) + UnitCell& ucell_in, + const int nspin, + const bool domag, + const bool domag_z, + const int gga_grad) { ModuleBase::TITLE("Forces", "cal_force_cc"); // recalculate the exchange-correlation potential. @@ -53,7 +56,7 @@ void Forces::cal_force_cc(ModuleBase::matrix& forcecc, return; } - ModuleBase::matrix v(PARAM.inp.nspin, rho_basis->nrxx); + ModuleBase::matrix v(nspin, rho_basis->nrxx); const double hybrid_alpha = XC_Functional::get_hybrid_alpha(); #ifdef __EXX @@ -66,7 +69,7 @@ void Forces::cal_force_cc(ModuleBase::matrix& forcecc, #ifdef __LIBXC const auto etxc_vtxc_v = XC_Functional_Libxc::v_xc_meta(XC_Functional::get_func_id(), rho_basis->nrxx, ucell_in.omega, ucell_in.tpiba, chr, - PARAM.inp.nspin, hybrid_alpha, hse_omega); + nspin, hybrid_alpha, hse_omega); // etxc = std::get<0>(etxc_vtxc_v); // vtxc = std::get<1>(etxc_vtxc_v); @@ -77,11 +80,12 @@ void Forces::cal_force_cc(ModuleBase::matrix& forcecc, } else { - unitcell::cal_ux(ucell_in, PARAM.inp.nspin); + unitcell::cal_ux(ucell_in, nspin); const auto etxc_vtxc_v = XC_Functional::v_xc(rho_basis->nrxx, chr, &ucell_in, - PARAM.inp.nspin, - PARAM.globalv.domag, - PARAM.globalv.domag_z, + nspin, + domag, + domag_z, + gga_grad, hybrid_alpha, hse_omega); @@ -92,7 +96,7 @@ void Forces::cal_force_cc(ModuleBase::matrix& forcecc, const ModuleBase::matrix vxc = v; std::complex* psiv = new std::complex[rho_basis->nmaxgr]; - if (PARAM.inp.nspin == 1 || PARAM.inp.nspin == 4) + if (nspin == 1 || nspin == 4) { #ifdef _OPENMP #pragma omp parallel for schedule(static, 1024) @@ -368,4 +372,4 @@ void Forces::deriv_drhoc template class Forces; #if ((defined __CUDA) || (defined __ROCM)) template class Forces; -#endif \ No newline at end of file +#endif diff --git a/source/source_pw/module_pwdft/force_pw_onsite.cpp b/source/source_pw/module_pwdft/force_pw_onsite.cpp index 56c5943a50e..859eed2a780 100644 --- a/source/source_pw/module_pwdft/force_pw_onsite.cpp +++ b/source/source_pw/module_pwdft/force_pw_onsite.cpp @@ -2,7 +2,7 @@ #include "source_base/parallel_reduce.h" #include "source_base/timer.h" #include "source_base/tool_title.h" -#include "source_pw/module_pwdft/onsite_proj.h" +#include "source_pw/module_proj/onsite_proj.h" #include "source_pw/module_pwdft/kernels/force_op.h" #include "source_io/module_parameter/parameter.h" #include "source_pw/module_pwdft/dftu_base.h" @@ -12,9 +12,9 @@ template void Forces::cal_force_onsite(ModuleBase::matrix& force_onsite, const ModuleBase::matrix& wg, const ModulePW::PW_Basis_K* wfc_basis, - const UnitCell& ucell_in, - const Plus_U_Base& dftu, - const psi::Psi , Device>* psi_in) + const UnitCell& ucell_in, + const Plus_U_Base& dftu, + const psi::Psi , Device>* psi_in) { ModuleBase::TITLE("Forces", "cal_force_onsite"); if(psi_in == nullptr || wfc_basis == nullptr) diff --git a/source/source_pw/module_pwdft/force_pw_scc.cpp b/source/source_pw/module_pwdft/force_pw_scc.cpp index dbfce6945a3..32b6f7deeec 100644 --- a/source/source_pw/module_pwdft/force_pw_scc.cpp +++ b/source/source_pw/module_pwdft/force_pw_scc.cpp @@ -68,72 +68,72 @@ void Forces::cal_force_scc(ModuleBase::matrix& forcescc, } // work space - std::vector rhocgnt(rho_basis->ngg); + std::vector rhocgnt(rho_basis->ngg); ModuleBase::GlobalFunc::ZEROS(rhocgnt.data(), rho_basis->ngg); rho_basis->real2recip(psic.data(), psic.data()); int igg0 = 0; const int ig0 = rho_basis->ig_gge0; - if (rho_basis->gg_uniq[0] < 1.0e-8) - { - igg0 = 1; - } + if (rho_basis->gg_uniq[0] < 1.0e-8) + { + igg0 = 1; + } double fact = 2.0; - for (int nt = 0; nt < ucell_in.ntype; nt++) - { - // Here we compute the G.ne.0 term - const int mesh = ucell_in.atoms[nt].ncpp.msh; - this->deriv_drhoc_scc(numeric, - mesh, - ucell_in.atoms[nt].ncpp.r.data(), - ucell_in.atoms[nt].ncpp.rab.data(), - ucell_in.atoms[nt].ncpp.rho_at.data(), - rhocgnt.data(), - rho_basis, - ucell_in); - int iat = 0; - for (int it = 0; it < ucell_in.ntype; it++) { - for (int ia = 0; ia < ucell_in.atoms[it].na; ia++) { - if (nt == it) { - const ModuleBase::Vector3 pos - = ucell_in.atoms[it].tau[ia]; - double &force0 = forcescc(iat, 0), - &force1 = forcescc(iat, 1), - &force2 = forcescc(iat, 2); + for (int nt = 0; nt < ucell_in.ntype; nt++) + { + // Here we compute the G.ne.0 term + const int mesh = ucell_in.atoms[nt].ncpp.msh; + this->deriv_drhoc_scc(numeric, + mesh, + ucell_in.atoms[nt].ncpp.r.data(), + ucell_in.atoms[nt].ncpp.rab.data(), + ucell_in.atoms[nt].ncpp.rho_at.data(), + rhocgnt.data(), + rho_basis, + ucell_in); + int iat = 0; + for (int it = 0; it < ucell_in.ntype; it++) { + for (int ia = 0; ia < ucell_in.atoms[it].na; ia++) { + if (nt == it) { + const ModuleBase::Vector3 pos + = ucell_in.atoms[it].tau[ia]; + double &force0 = forcescc(iat, 0), + &force1 = forcescc(iat, 1), + &force2 = forcescc(iat, 2); #ifdef _OPENMP #pragma omp parallel for reduction(+ : force0) reduction(+ : force1) reduction(+ : force2) #endif - for (int ig = 0; ig < rho_basis->npw; ++ig) { - if (ig == ig0) { - continue; - } - const ModuleBase::Vector3 gv - = rho_basis->gcar[ig]; - const double rhocgntigg - = rhocgnt[rho_basis->ig2igg[ig]]; - const double arg = ModuleBase::TWO_PI * (gv * pos); - double sinp = 0.0, cosp = 0.0; - ModuleBase::libm::sincos(arg, &sinp, &cosp); - const std::complex cpm - = std::complex(sinp, cosp) * conj(psic[ig]); - - force0 += fact * rhocgntigg * ucell_in.tpiba - * gv.x * cpm.real(); - force1 += fact * rhocgntigg * ucell_in.tpiba - * gv.y * cpm.real(); - force2 += fact * rhocgntigg * ucell_in.tpiba - * gv.z * cpm.real(); - } - } - iat++; - } - } - } - - - Parallel_Reduce::reduce_pool(forcescc.c, forcescc.nr * forcescc.nc); + for (int ig = 0; ig < rho_basis->npw; ++ig) { + if (ig == ig0) { + continue; + } + const ModuleBase::Vector3 gv + = rho_basis->gcar[ig]; + const double rhocgntigg + = rhocgnt[rho_basis->ig2igg[ig]]; + const double arg = ModuleBase::TWO_PI * (gv * pos); + double sinp = 0.0, cosp = 0.0; + ModuleBase::libm::sincos(arg, &sinp, &cosp); + const std::complex cpm + = std::complex(sinp, cosp) * conj(psic[ig]); + + force0 += fact * rhocgntigg * ucell_in.tpiba + * gv.x * cpm.real(); + force1 += fact * rhocgntigg * ucell_in.tpiba + * gv.y * cpm.real(); + force2 += fact * rhocgntigg * ucell_in.tpiba + * gv.z * cpm.real(); + } + } + iat++; + } + } + } + + + Parallel_Reduce::reduce_pool(forcescc.c, forcescc.nr * forcescc.nc); ModuleBase::timer::end("Forces", "cal_force_scc"); return; @@ -156,7 +156,7 @@ void Forces::deriv_drhoc_scc(const bool& numeric, /// the modulus of g for a given shell /// the fourier transform /// auxiliary memory for integration - std::vector gx_arr(rho_basis->ngg); + std::vector gx_arr(rho_basis->ngg); double* gx_arr_d = nullptr; /// counter on radial mesh points /// counter on g shells @@ -165,12 +165,12 @@ void Forces::deriv_drhoc_scc(const bool& numeric, /// /// G=0 term /// - if (rho_basis->gg_uniq[0] < 1.0e-8) - { - drhocg[0] = 0.0; + if (rho_basis->gg_uniq[0] < 1.0e-8) + { + drhocg[0] = 0.0; igl0 = 1; - } else - { + } else + { igl0 = 0; } @@ -182,19 +182,19 @@ void Forces::deriv_drhoc_scc(const bool& numeric, #ifdef _OPENMP #pragma omp parallel for #endif - for (int igl = igl0; igl < rho_basis->ngg; igl++) - { - gx_arr[igl] = sqrt(rho_basis->gg_uniq[igl]) * ucell_in.tpiba; - } + for (int igl = igl0; igl < rho_basis->ngg; igl++) + { + gx_arr[igl] = sqrt(rho_basis->gg_uniq[igl]) * ucell_in.tpiba; + } - double *r_d = nullptr; + double *r_d = nullptr; double *rhoc_d = nullptr; double *rab_d = nullptr; double *aux_d = nullptr; double *drhocg_d = nullptr; - if (this->device == base_device::GpuDevice) - { - resmem_var_op()(r_d, mesh); + if (this->device == base_device::GpuDevice) + { + resmem_var_op()(r_d, mesh); resmem_var_op()(rhoc_d, mesh); resmem_var_op()(rab_d, mesh); @@ -210,43 +210,43 @@ void Forces::deriv_drhoc_scc(const bool& numeric, syncmem_var_h2d_op()(rhoc_d, rhoc, mesh); } - if(this->device == base_device::GpuDevice) - { - hamilt::cal_stress_drhoc_aux_op()( - r_d, - rhoc_d, - gx_arr_d+igl0, - rab_d, - drhocg_d+igl0, - mesh, - igl0, - rho_basis->ngg-igl0, - ucell_in.omega, - 2); - - syncmem_var_d2h_op()(drhocg+igl0, drhocg_d+igl0, rho_basis->ngg-igl0); - - } - else - { - hamilt::cal_stress_drhoc_aux_op()( - r, - rhoc, - gx_arr.data()+igl0, - rab, - drhocg+igl0, - mesh, - igl0, - rho_basis->ngg-igl0, - ucell_in.omega, - 2); - - } + if(this->device == base_device::GpuDevice) + { + hamilt::cal_stress_drhoc_aux_op()( + r_d, + rhoc_d, + gx_arr_d+igl0, + rab_d, + drhocg_d+igl0, + mesh, + igl0, + rho_basis->ngg-igl0, + ucell_in.omega, + 2); + + syncmem_var_d2h_op()(drhocg+igl0, drhocg_d+igl0, rho_basis->ngg-igl0); + + } + else + { + hamilt::cal_stress_drhoc_aux_op()( + r, + rhoc, + gx_arr.data()+igl0, + rab, + drhocg+igl0, + mesh, + igl0, + rho_basis->ngg-igl0, + ucell_in.omega, + 2); + + } delmem_var_op()(r_d); delmem_var_op()(rhoc_d); delmem_var_op()(rab_d); - delmem_var_op()(aux_d); + delmem_var_op()(aux_d); delmem_var_op()(gx_arr_d); delmem_var_op()(drhocg_d); return; diff --git a/source/source_pw/module_pwdft/fs_nonlocal_tools.cpp b/source/source_pw/module_pwdft/fs_nonlocal_tools.cpp index 4a380349c69..ee0b7d3ada2 100644 --- a/source/source_pw/module_pwdft/fs_nonlocal_tools.cpp +++ b/source/source_pw/module_pwdft/fs_nonlocal_tools.cpp @@ -7,7 +7,7 @@ #include "source_base/tool_title.h" #include "source_pw/module_pwdft/kernels/force_op.h" #include "source_io/module_parameter/parameter.h" -#include "nonlocal_maths.hpp" +#include "nonlocal_maths.h" #include "source_base/parallel_comm.h" // different MPI worlds (POOL_WORLD) diff --git a/source/source_pw/module_pwdft/fs_nonlocal_tools_force.cpp b/source/source_pw/module_pwdft/fs_nonlocal_tools_force.cpp index 714b1f1653b..24eb88854da 100644 --- a/source/source_pw/module_pwdft/fs_nonlocal_tools_force.cpp +++ b/source/source_pw/module_pwdft/fs_nonlocal_tools_force.cpp @@ -1,15 +1,13 @@ #include "fs_nonlocal_tools.h" - +#include "nonlocal_maths.h" #include "source_base/math_polyint.h" #include "source_base/math_ylmreal.h" +#include "source_base/parallel_comm.h" // different MPI worlds (POOL_WORLD) #include "source_base/parallel_device.h" #include "source_base/timer.h" #include "source_base/tool_title.h" -#include "source_pw/module_pwdft/kernels/force_op.h" #include "source_io/module_parameter/parameter.h" -#include "nonlocal_maths.hpp" - -#include "source_base/parallel_comm.h" // different MPI worlds (POOL_WORLD) +#include "source_pw/module_pwdft/kernels/force_op.h" namespace hamilt { @@ -32,14 +30,13 @@ void FS_Nonlocal_tools::cal_vkb_deri_f(const int& ik, const int& if (this->pre_ik_f == -1) { resmem_var_op()(gcar, 3 * this->wfc_basis_->npwk_max); - resmem_int_op()(gcar_zero_indexes, 3 * this->wfc_basis_->npwk_max); + // Each row stores a count followed by up to npwk_max plane-wave indices. + resmem_int_op()(gcar_zero_indexes, 3 * (this->wfc_basis_->npwk_max + 1)); } if (this->pre_ik_f != ik) { - this->transfer_gcar(npw, - this->wfc_basis_->npwk_max, - &(this->wfc_basis_->gcar[ik * this->wfc_basis_->npwk_max].x)); + this->transfer_gcar(npw, this->wfc_basis_->npwk_max, &(this->wfc_basis_->gcar[ik * this->wfc_basis_->npwk_max].x)); } this->save_vkb(ik, ipol); @@ -48,7 +45,6 @@ void FS_Nonlocal_tools::cal_vkb_deri_f(const int& ik, const int& // calculate the vkb_deri for ipol with the memory of ppcell_vkb cal_vkb1_nl_op()(this->ctx, nkb, npw, npw, npw, ipol, coeff, vkb_ptr, gcar, vkb_deri_ptr); - } template @@ -90,11 +86,16 @@ void FS_Nonlocal_tools::cal_dbecp_f(const int& ik, template void FS_Nonlocal_tools::save_vkb(const int& ik, const int& ipol) { + if (this->gcar_zero_counts[ipol] == 0) + { + return; + } const int npw = this->wfc_basis_->npwk[ik]; + const int index_stride = this->wfc_basis_->npwk_max + 1; if (this->device == base_device::CpuDevice) { - const int gcar_zero_count = this->gcar_zero_indexes[ipol * this->wfc_basis_->npwk_max]; - const int* gcar_zero_ptrs = &this->gcar_zero_indexes[ipol * this->wfc_basis_->npwk_max + 1]; + const int gcar_zero_count = this->gcar_zero_indexes[ipol * index_stride]; + const int* gcar_zero_ptrs = &this->gcar_zero_indexes[ipol * index_stride + 1]; const std::complex* vkb_ptr = this->ppcell_vkb; std::complex* vkb_save_ptr = this->vkb_save; // find the zero indexes to save the vkb values to vkb_save @@ -118,7 +119,7 @@ void FS_Nonlocal_tools::save_vkb(const int& ik, const int& ipol) this->gcar_zero_counts[ipol], npw, ipol, - this->wfc_basis_->npwk_max); + index_stride); #endif } } @@ -127,12 +128,18 @@ void FS_Nonlocal_tools::save_vkb(const int& ik, const int& ipol) template void FS_Nonlocal_tools::revert_vkb(const int& ik, const int& ipol) { + if (this->gcar_zero_counts[ipol] == 0) + { + this->pre_ik_f = ik; + return; + } const int npw = this->wfc_basis_->npwk[ik]; + const int index_stride = this->wfc_basis_->npwk_max + 1; const std::complex coeff = ipol == 0 ? ModuleBase::NEG_IMAG_UNIT : ModuleBase::ONE; if (this->device == base_device::CpuDevice) { - const int gcar_zero_count = this->gcar_zero_indexes[ipol * this->wfc_basis_->npwk_max]; - const int* gcar_zero_ptrs = &this->gcar_zero_indexes[ipol * this->wfc_basis_->npwk_max + 1]; + const int gcar_zero_count = this->gcar_zero_indexes[ipol * index_stride]; + const int* gcar_zero_ptrs = &this->gcar_zero_indexes[ipol * index_stride + 1]; std::complex* vkb_ptr = this->ppcell_vkb; const std::complex* vkb_save_ptr = this->vkb_save; // find the zero indexes to save the vkb values to vkb_save @@ -156,7 +163,7 @@ void FS_Nonlocal_tools::revert_vkb(const int& ik, const int& ipo this->gcar_zero_counts[ipol], npw, ipol, - this->wfc_basis_->npwk_max, + index_stride, coeff); #endif } @@ -168,12 +175,13 @@ void FS_Nonlocal_tools::transfer_gcar(const int& npw, const int& { std::vector gcar_tmp(3 * npw_max); gcar_tmp.assign(gcar_in, gcar_in + 3 * npw_max); - std::vector gcar_zero_indexes_tmp(3 * npw_max); + const int index_stride = npw_max + 1; + std::vector gcar_zero_indexes_tmp(3 * index_stride); int* gcar_zero_ptrs[3]; for (int i = 0; i < 3; i++) { - gcar_zero_ptrs[i] = &gcar_zero_indexes_tmp[i * npw_max]; + gcar_zero_ptrs[i] = &gcar_zero_indexes_tmp[i * index_stride]; gcar_zero_ptrs[i][0] = -1; this->gcar_zero_counts[i] = 0; } @@ -216,7 +224,7 @@ void FS_Nonlocal_tools::transfer_gcar(const int& npw, const int& resmem_complex_op()(this->vkb_save, this->nkb * max_count); // transfer the gcar and gcar_zero_indexes to the device syncmem_var_h2d_op()(gcar, gcar_tmp.data(), 3 * npw_max); - syncmem_int_h2d_op()(gcar_zero_indexes, gcar_zero_indexes_tmp.data(), 3 * npw_max); + syncmem_int_h2d_op()(gcar_zero_indexes, gcar_zero_indexes_tmp.data(), 3 * index_stride); } // cal_force @@ -245,7 +253,7 @@ void FS_Nonlocal_tools::cal_force(const int& ik, { d_wg_ik = d_wg + this->nbands * ik; } - + cal_force_nl_op()(this->ctx, nondiagonal, npm, @@ -305,20 +313,16 @@ void FS_Nonlocal_tools::cal_force(const int& ik, } // template instantiation -template void FS_Nonlocal_tools::cal_vkb_deri_f(const int&, - const int&, - const int&); +template void FS_Nonlocal_tools::cal_vkb_deri_f(const int&, const int&, const int&); template void FS_Nonlocal_tools::cal_dbecp_f(const int&, - const int&, - const int&, - const int&, - const std::complex*, - const int&); + const int&, + const int&, + const int&, + const std::complex*, + const int&); template void FS_Nonlocal_tools::save_vkb(const int&, const int&); template void FS_Nonlocal_tools::revert_vkb(const int&, const int&); -template void FS_Nonlocal_tools::transfer_gcar(const int&, - const int&, - const double*); +template void FS_Nonlocal_tools::transfer_gcar(const int&, const int&, const double*); template void FS_Nonlocal_tools::cal_force(const int&, const int&, const int&, @@ -326,20 +330,16 @@ template void FS_Nonlocal_tools::cal_force(cons double*, const int&); #if ((defined __CUDA) || (defined __ROCM)) -template void FS_Nonlocal_tools::cal_vkb_deri_f(const int&, - const int&, - const int&); +template void FS_Nonlocal_tools::cal_vkb_deri_f(const int&, const int&, const int&); template void FS_Nonlocal_tools::cal_dbecp_f(const int&, - const int&, - const int&, - const int&, - const std::complex*, - const int&); + const int&, + const int&, + const int&, + const std::complex*, + const int&); template void FS_Nonlocal_tools::save_vkb(const int&, const int&); template void FS_Nonlocal_tools::revert_vkb(const int&, const int&); -template void FS_Nonlocal_tools::transfer_gcar(const int&, - const int&, - const double*); +template void FS_Nonlocal_tools::transfer_gcar(const int&, const int&, const double*); template void FS_Nonlocal_tools::cal_force(const int&, const int&, const int&, diff --git a/source/source_pw/module_pwdft/fs_nonlocal_tools_stress.cpp b/source/source_pw/module_pwdft/fs_nonlocal_tools_stress.cpp index 7b6c5c2a0f1..a92a4d0a6a2 100644 --- a/source/source_pw/module_pwdft/fs_nonlocal_tools_stress.cpp +++ b/source/source_pw/module_pwdft/fs_nonlocal_tools_stress.cpp @@ -7,7 +7,7 @@ #include "source_base/tool_title.h" #include "source_pw/module_pwdft/kernels/force_op.h" #include "source_io/module_parameter/parameter.h" -#include "nonlocal_maths.hpp" +#include "nonlocal_maths.h" #include "source_base/parallel_comm.h" // different MPI worlds (POOL_WORLD) diff --git a/source/source_pw/module_pwdft/fs_nonlocal_tools_vkb.cpp b/source/source_pw/module_pwdft/fs_nonlocal_tools_vkb.cpp index 38767608675..79c105ef7d1 100644 --- a/source/source_pw/module_pwdft/fs_nonlocal_tools_vkb.cpp +++ b/source/source_pw/module_pwdft/fs_nonlocal_tools_vkb.cpp @@ -7,7 +7,7 @@ #include "source_base/tool_title.h" #include "source_pw/module_pwdft/kernels/force_op.h" #include "source_io/module_parameter/parameter.h" -#include "nonlocal_maths.hpp" +#include "nonlocal_maths.h" #include "source_base/parallel_comm.h" // different MPI worlds (POOL_WORLD) diff --git a/source/source_pw/module_pwdft/hamilt_lcaopw.h b/source/source_pw/module_pwdft/hamilt_lcaopw.h index 414c5cddb95..24038ae3505 100644 --- a/source/source_pw/module_pwdft/hamilt_lcaopw.h +++ b/source/source_pw/module_pwdft/hamilt_lcaopw.h @@ -1,6 +1,7 @@ #ifndef HAMILTLIP_H #define HAMILTLIP_H +#include "source_hamilt/hamilt_hs_adapter.h" #include "source_pw/module_pwdft/hamilt_pw.h" #ifdef __EXX #include "source_lcao/module_ri/exx_lip.h" @@ -32,6 +33,52 @@ namespace hamilt #endif }; + /// HamiltLIP seen through hsolver::HSOperator. Besides H and S it feeds the + /// EXX term into the subspace Hamiltonian and hands the subspace + /// eigenvectors back to Exx_Lip, which is what HSolverLIP needs. + template + class HamiltLIPHSOperator : public HamiltHSOperator + { + public: + HamiltLIPHSOperator(HamiltLIP* hm, + const ModulePW::PW_Basis_K* wfc_basis, + const bool cal_exx, + const double hybrid_alpha) + : HamiltHSOperator(hm, wfc_basis), hm_lip_(hm), cal_exx_(cal_exx), + hybrid_alpha_(hybrid_alpha){}; + +#ifdef __EXX + void add_to_subspace_h(T* hcc, const int naos) const override + { + if (!cal_exx_) + { + return; + } + const int ik = this->ik_; + for (int n = 0; n < naos; ++n) + { + for (int m = 0; m < naos; ++m) + { + hcc[n * naos + m] += (T)hybrid_alpha_ * hm_lip_->exx_lip.get_exx_matrix()[ik][m][n]; + } + } + } + + void export_subspace_vec(const T* vcc, const int naos, const int nbands) const override + { + if (cal_exx_) + { + hm_lip_->exx_lip.set_hvec(this->ik_, vcc, naos, nbands); + } + } +#endif + + private: + HamiltLIP* hm_lip_ = nullptr; + const bool cal_exx_; + const double hybrid_alpha_; + }; + } // namespace hamilt #endif diff --git a/source/source_pw/module_pwdft/hamilt_pw.cpp b/source/source_pw/module_pwdft/hamilt_pw.cpp index 01dc1f3188a..18beccada0c 100644 --- a/source/source_pw/module_pwdft/hamilt_pw.cpp +++ b/source/source_pw/module_pwdft/hamilt_pw.cpp @@ -1,9 +1,11 @@ #include "hamilt_pw.h" #include "op_pw_ekin.h" +#include "op_pw_ekin_td.h" #include "op_pw_exx.h" #include "op_pw_meta.h" #include "op_pw_nl.h" +#include "op_pw_nl_td.h" #include "op_pw_proj.h" #include "op_pw_veff.h" #include "source_base/global_function.h" @@ -34,6 +36,8 @@ HamiltPW::HamiltPW(elecstate::Potential* pot_in, const auto tpiba = static_cast(ucell->tpiba); const int* isk = pkv->isk.data(); const Real* gk2 = wfc_basis->get_gk2_data(); + const bool is_tddft = PARAM.inp.esolver_type == "tddft"; + const bool velocity_gauge = is_tddft && PARAM.inp.td_stype == 1; if (PARAM.inp.t_in_h) { @@ -48,6 +52,21 @@ HamiltPW::HamiltPW(elecstate::Potential* pot_in, { this->ops->add(ekinetic); } + + // Add the velocity-gauge kinetic correction 2*A*p + A^2. + if (velocity_gauge) + { + Operator* td_ekinetic = new TDEkineticPW(wfc_basis); + + if (this->ops == nullptr) + { + this->ops = td_ekinetic; + } + else + { + this->ops->add(td_ekinetic); + } + } } if (PARAM.inp.vl_in_h) { @@ -74,6 +93,10 @@ HamiltPW::HamiltPW(elecstate::Potential* pot_in, { pot_register_in.push_back("gatefield"); } + if (is_tddft) + { + pot_register_in.push_back("tddft"); + } if (PARAM.inp.ml_exx) // sunliang { pot_register_in.push_back("ml_exx"); @@ -112,7 +135,19 @@ HamiltPW::HamiltPW(elecstate::Potential* pot_in, } if (PARAM.inp.vnl_in_h) { - Operator* nonlocal = new Nonlocal>(isk, this->ppcell, ucell, wfc_basis); + Operator* nonlocal = nullptr; + + // Use vector-potential-shifted projectors in the velocity gauge. + if (velocity_gauge) + { + nonlocal = new TDNonlocalPW(isk, this->ppcell, ucell, wfc_basis); + } + else + { + // Ground-state and length-gauge calculations use the standard operator. + nonlocal = new Nonlocal>(isk, this->ppcell, ucell, wfc_basis); + } + if (this->ops == nullptr) { this->ops = nonlocal; @@ -133,10 +168,16 @@ HamiltPW::HamiltPW(elecstate::Potential* pot_in, } if (exx_info && exx_info->cal_exx) { - bool separate_loop = exx_info->separate_loop; - double hybrid_alpha = exx_info->hybrid_alpha; - auto coulomb_param = exx_info->coulomb_param; - auto exx = new OperatorEXXPW(isk, wfc_basis, pot_in->get_rho_basis(), pkv, ucell, separate_loop, hybrid_alpha, coulomb_param); + auto exx = new OperatorEXXPW(isk, + wfc_basis, + pot_in->get_rho_basis(), + pkv, + ucell, + *exx_info, + PARAM.inp.nspin, + GlobalV::KPAR, + GlobalV::MY_RANK, + GlobalV::MY_POOL); if (this->ops == nullptr) { this->ops = exx; @@ -159,6 +200,33 @@ HamiltPW::~HamiltPW() } } +template +void HamiltPW::bind_td_state(const Real* veff, const Real* vofk, const ModuleBase::Vector3& A_ha) +{ + Operator* op = this->ops; + while (op != nullptr) + { + if (op->get_cal_type() == calculation_type::pw_veff) + { + static_cast>*>(op)->set_veff(veff); + } + else if (op->get_cal_type() == calculation_type::pw_meta) + { + static_cast>*>(op)->set_vk(vofk); + } + else if (op->get_cal_type() == calculation_type::pw_ekinetic_td) + { + static_cast*>(op)->set_A_ha(A_ha); + } + else if (op->get_cal_type() == calculation_type::pw_nonlocal) + { + TDNonlocalPW* td_nonlocal = dynamic_cast*>(op); + if (td_nonlocal != nullptr) td_nonlocal->set_A_ha(A_ha); + } + op = op->next_op; + } +} + template void HamiltPW::updateHk(const int ik) { diff --git a/source/source_pw/module_pwdft/hamilt_pw.h b/source/source_pw/module_pwdft/hamilt_pw.h index 732fc4833d9..72158eadaf4 100644 --- a/source/source_pw/module_pwdft/hamilt_pw.h +++ b/source/source_pw/module_pwdft/hamilt_pw.h @@ -40,6 +40,11 @@ class HamiltPW : public Hamilt // for target K point, update consequence of hPsi() and matrix() void updateHk(const int ik) override; + /** @brief Bind local-potential buffers and Hartree vector potential for the same operation. + * @note The caller must keep the buffers alive and restore endpoint state after propagation. + */ + void bind_td_state(const Real* veff, const Real* vofk, const ModuleBase::Vector3& A_ha); + void sPsi(const T* psi_in, // psi T* spsi, // spsi const int nrow, // dimension of spsi: nbands * nrow diff --git a/source/source_pw/module_pwdft/kernels/cuda/exx_batch_op.cu b/source/source_pw/module_pwdft/kernels/cuda/exx_batch_op.cu new file mode 100644 index 00000000000..f01461ca508 --- /dev/null +++ b/source/source_pw/module_pwdft/kernels/cuda/exx_batch_op.cu @@ -0,0 +1,412 @@ +#include "source_pw/module_pwdft/kernels/exx_batch_op.h" + +#include +#include +#include +#include +#include + +namespace hamilt +{ + +#define THREADS_PER_BLOCK 256 + +template +struct CufftTraits; + +template <> +struct CufftTraits> +{ + using cufft_t = cufftDoubleComplex; + static constexpr cufftType type = CUFFT_Z2Z; + static cufftResult exec(cufftHandle plan, cufft_t* data, int dir) { return cufftExecZ2Z(plan, data, data, dir); } +}; + +template <> +struct CufftTraits> +{ + using cufft_t = cufftComplex; + static constexpr cufftType type = CUFFT_C2C; + static cufftResult exec(cufftHandle plan, cufft_t* data, int dir) { return cufftExecC2C(plan, data, data, dir); } +}; + +template +static void exx_batch_fft_plan_create_gpu(void** plan, const int nx, const int ny, const int nz, const int batch) +{ + cufftHandle handle; + int n[3] = {nx, ny, nz}; + CHECK_CUFFT(cufftPlanMany(&handle, + 3, + n, + nullptr, 1, 0, // input: contiguous, distance = nx*ny*nz + nullptr, 1, 0, // output: same (in-place) + CufftTraits::type, + batch)); + *plan = new cufftHandle(handle); +} + +template +static void exx_batch_fft_exec_gpu(void* plan, T* data, const bool forward) +{ + cufftHandle handle = *reinterpret_cast(plan); + CHECK_CUFFT(CufftTraits::exec(handle, + reinterpret_cast::cufft_t*>(data), + forward ? CUFFT_FORWARD : CUFFT_INVERSE)); +} + +template +static void exx_batch_fft_plan_destroy_gpu(void** plan) +{ + if (*plan != nullptr) + { + cufftHandle handle = *reinterpret_cast(plan); + // no CHECK_CUFFT here: at process teardown the CUDA context may already + // be gone, in which case cufftDestroy reports CUFFT_INVALID_PLAN; that + // is harmless and must not abort the run + cufftDestroy(handle); + delete reinterpret_cast(*plan); + *plan = nullptr; + } +} + +template +__global__ void batch_density_real(const int nbands, + const int nrxx, + const thrust::complex* nk_real_all, + const thrust::complex* mq_real, + const FPTYPE omega_inv, + thrust::complex* out) +{ + const long long idx = static_cast(blockIdx.x) * blockDim.x + threadIdx.x; + if (idx < static_cast(nbands) * nrxx) + { + const int i = idx % nrxx; + out[idx] = nk_real_all[idx] * thrust::conj(mq_real[i]) * omega_inv; + } +} + +template +__global__ void batch_gather_pw(const int nbands, + const int npw, + const int nxyz, + const int* box_map, + const thrust::complex* box, + thrust::complex* pw) +{ + const long long idx = static_cast(blockIdx.x) * blockDim.x + threadIdx.x; + if (idx < static_cast(nbands) * npw) + { + const long long n = idx / npw; + const int ig = idx % npw; + pw[idx] = box[n * nxyz + box_map[ig]] / static_cast(nxyz); + } +} + +template +__global__ void batch_mul_pot(const int nbands, + const int npw, + const FPTYPE* pot, + thrust::complex* pw) +{ + const long long idx = static_cast(blockIdx.x) * blockDim.x + threadIdx.x; + if (idx < static_cast(nbands) * npw) + { + const int ig = idx % npw; + pw[idx] *= pot[ig]; + } +} + +template +__global__ void batch_scatter_pw(const int nbands, + const int npw, + const int nxyz, + const int* box_map, + const thrust::complex* pw, + thrust::complex* box) +{ + const long long idx = static_cast(blockIdx.x) * blockDim.x + threadIdx.x; + if (idx < static_cast(nbands) * npw) + { + const long long n = idx / npw; + const int ig = idx % npw; + box[n * nxyz + box_map[ig]] = pw[idx]; + } +} + +template +__global__ void batch_mul_real(const int nbands, + const int nrxx, + const thrust::complex* mq_real, + thrust::complex* data) +{ + const long long idx = static_cast(blockIdx.x) * blockDim.x + threadIdx.x; + if (idx < static_cast(nbands) * nrxx) + { + const int i = idx % nrxx; + data[idx] *= mq_real[i]; + } +} + +template +__global__ void batch_gather_accum(const int nbands, + const int npwk, + const int nxyz, + const int* box_map, + const thrust::complex* box, + const FPTYPE factor, + thrust::complex* out, + const int out_stride) +{ + const long long idx = static_cast(blockIdx.x) * blockDim.x + threadIdx.x; + if (idx < static_cast(nbands) * npwk) + { + const long long n = idx / npwk; + const int ig = idx % npwk; + out[n * out_stride + ig] += factor / static_cast(nxyz) * box[n * nxyz + box_map[ig]]; + } +} + +template +__global__ void batch_scatter_wfc(const int nbands, + const int npwk, + const int nxyz, + const int* map, + const thrust::complex* psi, + const int psi_stride, + thrust::complex* box) +{ + const long long idx = static_cast(blockIdx.x) * blockDim.x + threadIdx.x; + if (idx < static_cast(nbands) * npwk) + { + const long long n = idx / npwk; + const int ig = idx % npwk; + box[n * nxyz + map[ig]] = psi[n * psi_stride + ig]; + } +} + +static inline int nblocks(const long long n) +{ + return static_cast((n + THREADS_PER_BLOCK - 1) / THREADS_PER_BLOCK); +} + +template +static void exx_batch_density_real_gpu(const int nbands, + const int nrxx, + const T* nk_real_all, + const T* mq_real, + const double omega, + T* out) +{ + using FPTYPE = typename T::value_type; + batch_density_real<<(nbands) * nrxx), THREADS_PER_BLOCK>>>( + nbands, + nrxx, + reinterpret_cast*>(nk_real_all), + reinterpret_cast*>(mq_real), + static_cast(1.0 / omega), + reinterpret_cast*>(out)); + CHECK_CUDA_SYNC(); +} + +template +static void exx_batch_gather_pw_gpu(const int nbands, + const int npw, + const int nxyz, + const int* box_map, + const T* box, + T* pw) +{ + using FPTYPE = typename T::value_type; + batch_gather_pw<<(nbands) * npw), THREADS_PER_BLOCK>>>( + nbands, + npw, + nxyz, + box_map, + reinterpret_cast*>(box), + reinterpret_cast*>(pw)); + CHECK_CUDA_SYNC(); +} + +template +static void exx_batch_mul_pot_gpu(const int nbands, const int npw, const Real* pot, T* pw) +{ + using FPTYPE = typename T::value_type; + batch_mul_pot<<(nbands) * npw), THREADS_PER_BLOCK>>>( + nbands, + npw, + pot, + reinterpret_cast*>(pw)); + CHECK_CUDA_SYNC(); +} + +template +static void exx_batch_scatter_pw_gpu(const int nbands, + const int npw, + const int nxyz, + const int* box_map, + const T* pw, + T* box) +{ + using FPTYPE = typename T::value_type; + batch_scatter_pw<<(nbands) * npw), THREADS_PER_BLOCK>>>( + nbands, + npw, + nxyz, + box_map, + reinterpret_cast*>(pw), + reinterpret_cast*>(box)); + CHECK_CUDA_SYNC(); +} + +template +static void exx_batch_mul_real_gpu(const int nbands, const int nrxx, const T* mq_real, T* data) +{ + using FPTYPE = typename T::value_type; + batch_mul_real<<(nbands) * nrxx), THREADS_PER_BLOCK>>>( + nbands, + nrxx, + reinterpret_cast*>(mq_real), + reinterpret_cast*>(data)); + CHECK_CUDA_SYNC(); +} + +template +static void exx_batch_scatter_wfc_gpu(const int nbands, + const int npwk, + const int nxyz, + const int* map, + const T* psi, + const int psi_stride, + T* box) +{ + using FPTYPE = typename T::value_type; + batch_scatter_wfc<<(nbands) * npwk), THREADS_PER_BLOCK>>>( + nbands, + npwk, + nxyz, + map, + reinterpret_cast*>(psi), + psi_stride, + reinterpret_cast*>(box)); + CHECK_CUDA_SYNC(); +} + +template +static void exx_batch_gather_accum_gpu(const int nbands, + const int npwk, + const int nxyz, + const int* box_map, + const T* box, + const Real factor, + T* out, + const int out_stride) +{ + using FPTYPE = typename T::value_type; + batch_gather_accum<<(nbands) * npwk), THREADS_PER_BLOCK>>>( + nbands, + npwk, + nxyz, + box_map, + reinterpret_cast*>(box), + static_cast(factor), + reinterpret_cast*>(out), + out_stride); + CHECK_CUDA_SYNC(); +} + +// DEVICE_GPU explicit specializations of the API declared in exx_batch_op.h, +// forwarding to the CUDA implementations above. +#define EXX_BATCH_GPU_SPEC(T, Real) \ + template <> \ + void exx_batch_density_real(const int nbands, \ + const int nrxx, \ + const T* nk_real_all, \ + const T* mq_real, \ + const double omega, \ + T* out) \ + { \ + exx_batch_density_real_gpu(nbands, nrxx, nk_real_all, mq_real, omega, out); \ + } \ + template <> \ + void exx_batch_gather_pw(const int nbands, \ + const int npw, \ + const int nxyz, \ + const int* box_map, \ + const T* box, \ + T* pw) \ + { \ + exx_batch_gather_pw_gpu(nbands, npw, nxyz, box_map, box, pw); \ + } \ + template <> \ + void exx_batch_mul_pot(const int nbands, \ + const int npw, \ + const Real* pot, \ + T* pw) \ + { \ + exx_batch_mul_pot_gpu(nbands, npw, pot, pw); \ + } \ + template <> \ + void exx_batch_scatter_pw(const int nbands, \ + const int npw, \ + const int nxyz, \ + const int* box_map, \ + const T* pw, \ + T* box) \ + { \ + exx_batch_scatter_pw_gpu(nbands, npw, nxyz, box_map, pw, box); \ + } \ + template <> \ + void exx_batch_mul_real(const int nbands, \ + const int nrxx, \ + const T* mq_real, \ + T* data) \ + { \ + exx_batch_mul_real_gpu(nbands, nrxx, mq_real, data); \ + } \ + template <> \ + void exx_batch_gather_accum(const int nbands, \ + const int npwk, \ + const int nxyz, \ + const int* box_map, \ + const T* box, \ + const Real factor, \ + T* out, \ + const int out_stride) \ + { \ + exx_batch_gather_accum_gpu(nbands, npwk, nxyz, box_map, box, factor, out, out_stride); \ + } \ + template <> \ + void exx_batch_scatter_wfc(const int nbands, \ + const int npwk, \ + const int nxyz, \ + const int* map, \ + const T* psi, \ + const int psi_stride, \ + T* box) \ + { \ + exx_batch_scatter_wfc_gpu(nbands, npwk, nxyz, map, psi, psi_stride, box); \ + } \ + template <> \ + void exx_batch_fft_plan_create(void** plan, \ + const int nx, \ + const int ny, \ + const int nz, \ + const int batch) \ + { \ + exx_batch_fft_plan_create_gpu(plan, nx, ny, nz, batch); \ + } \ + template <> \ + void exx_batch_fft_exec(void* plan, T* data, const bool forward) \ + { \ + exx_batch_fft_exec_gpu(plan, data, forward); \ + } \ + template <> \ + void exx_batch_fft_plan_destroy(void** plan) \ + { \ + exx_batch_fft_plan_destroy_gpu(plan); \ + } + +EXX_BATCH_GPU_SPEC(std::complex, double) +EXX_BATCH_GPU_SPEC(std::complex, float) + +} // namespace hamilt diff --git a/source/source_pw/module_pwdft/kernels/cuda/nonlocal_op.cu b/source/source_pw/module_pwdft/kernels/cuda/nonlocal_op.cu index c02fc785e22..dcf46b7d077 100644 --- a/source/source_pw/module_pwdft/kernels/cuda/nonlocal_op.cu +++ b/source/source_pw/module_pwdft/kernels/cuda/nonlocal_op.cu @@ -1,6 +1,7 @@ #include "source_pw/module_pwdft/kernels/nonlocal_op.h" #include +#include #include #include @@ -30,12 +31,16 @@ __global__ void nonlocal_pw( const int jj = blockIdx.x % l2; for (int kk = threadIdx.x; kk < l3; kk += blockDim.x) { thrust::complex res(0.0, 0.0); + const std::int64_t projector_offset = sum + static_cast(ii) * l3; for (int xx = 0; xx < l3; xx++) { + const std::int64_t deeq_index = ((static_cast(spin) * deeq_x + iat + ii) * deeq_y + xx) * deeq_z + kk; + const std::int64_t input_index = static_cast(jj) * nkb + projector_offset + xx; res - += deeq[((spin * deeq_x + iat + ii) * deeq_y + xx) * deeq_z + kk] - * becp[jj * nkb + sum + ii * l3 + xx]; + += deeq[deeq_index] + * becp[input_index]; } - ps[(sum + ii * l3 + kk) * l2 + jj] += res; + const std::int64_t output_index = (projector_offset + kk) * l2 + jj; + ps[output_index] += res; } } @@ -54,20 +59,24 @@ __global__ void nonlocal_pw( thrust::complex* ps, const thrust::complex* becp) { - const int ii = blockIdx.x * 2 / l2; - const int jj = blockIdx.x * 2 % l2; + const std::int64_t spinor_index = static_cast(blockIdx.x) * 2; + const int ii = spinor_index / l2; + const int jj = spinor_index % l2; for (int kk = threadIdx.x; kk < l3; kk += blockDim.x) { thrust::complex res1(0.0, 0.0); thrust::complex res2(0.0, 0.0); - int psind = (sum + ii * l3 + kk) * l2 + jj; + const std::int64_t projector_offset = sum + static_cast(ii) * l3; + const std::int64_t psind = (projector_offset + kk) * l2 + jj; + const std::int64_t spin_stride = static_cast(deeq_x) * deeq_y * deeq_z; for (int xx = 0; xx < l3; xx++) { - int becpind = jj * nkb + sum + ii * l3 + xx; + const std::int64_t becpind = static_cast(jj) * nkb + projector_offset + xx; + const std::int64_t deeq_index = ((static_cast(iat) + ii) * deeq_y + kk) * deeq_z + xx; thrust::complex becp1 = becp[becpind]; thrust::complex becp2 = becp[becpind + nkb]; - res1 += deeq_nc[((0 * deeq_x + iat + ii) * deeq_y + kk) * deeq_z + xx] * becp1 - + deeq_nc[((1 * deeq_x + iat + ii) * deeq_y + kk) * deeq_z + xx] * becp2; - res2 += deeq_nc[((2 * deeq_x + iat + ii) * deeq_y + kk) * deeq_z + xx] * becp1 - + deeq_nc[((3 * deeq_x + iat + ii) * deeq_y + kk) * deeq_z + xx] * becp2; + res1 += deeq_nc[deeq_index] * becp1 + + deeq_nc[spin_stride + deeq_index] * becp2; + res2 += deeq_nc[2 * spin_stride + deeq_index] * becp1 + + deeq_nc[3 * spin_stride + deeq_index] * becp2; } ps[psind] += res1; ps[psind + 1] += res2; @@ -121,7 +130,8 @@ void hamilt::nonlocal_pw_op::operator()(const b { // denghui implement 20221109 // <<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<< - nonlocal_pw<<>>( + const std::int64_t block_count = static_cast(l1) * l2 / 2; + nonlocal_pw<<>>( l1, l2, l3, // loop size sum, iat, nkb, // control params deeq_x, deeq_y, deeq_z, diff --git a/source/source_pw/module_pwdft/kernels/cuda/projector_gradient_op.cu b/source/source_pw/module_pwdft/kernels/cuda/projector_gradient_op.cu new file mode 100644 index 00000000000..67d83694de2 --- /dev/null +++ b/source/source_pw/module_pwdft/kernels/cuda/projector_gradient_op.cu @@ -0,0 +1,27 @@ +#include "source_pw/module_pwdft/kernels/projector_gradient_op.h" +#include +#include +#include +namespace hamilt { +template +__global__ void gradient_kernel(int npw, int ld, int nkb, int nbeta, int nq, Real dq, Real tpiba, + const int* metadata, const Real* q, const Real* tab, const Real* derivative, + const thrust::complex* sk, thrust::complex* out) +{ + const int ig=blockIdx.x*blockDim.x+threadIdx.x; + if(ig(blockIdx.y),npw,ld,nkb,nbeta,nq,dq,tpiba,metadata,q,tab,derivative,sk,out); +} +template +void projector_gradient_op::operator()(int npw, int ld, int nkb, int nbeta, int nq, + Real dq, Real tpiba, const int* metadata, const Real* q, const Real* tab, const Real* derivative, + const std::complex* sk, std::complex* out) const +{ + if(npw==0 || nkb==0) return; + gradient_kernel<<>>(npw,ld,nkb,nbeta,nq,dq,tpiba,metadata,q,tab,derivative, + reinterpret_cast*>(sk),reinterpret_cast*>(out)); + const cudaError_t error=cudaGetLastError(); + if(error!=cudaSuccess) throw std::runtime_error(cudaGetErrorString(error)); +} +template class projector_gradient_op; +template class projector_gradient_op; +} diff --git a/source/source_pw/module_pwdft/kernels/cuda/stress_op.cu b/source/source_pw/module_pwdft/kernels/cuda/stress_op.cu index 9e29a364b6e..3edc438302a 100644 --- a/source/source_pw/module_pwdft/kernels/cuda/stress_op.cu +++ b/source/source_pw/module_pwdft/kernels/cuda/stress_op.cu @@ -344,7 +344,7 @@ __global__ void cal_stress_nl( { for (int ii = threadIdx.x; ii < nproj * nproj; ii += blockDim.x) { const int ip1 = ii / nproj; - const int ip2 = ii % nproj; + const int ip2 = ii % nproj; thrust::complex ps_qq = 0; if(ekb_now != 0) { diff --git a/source/source_pw/module_pwdft/kernels/cuda/vnl_op.cu b/source/source_pw/module_pwdft/kernels/cuda/vnl_op.cu index 689b958282c..551f44630cd 100644 --- a/source/source_pw/module_pwdft/kernels/cuda/vnl_op.cu +++ b/source/source_pw/module_pwdft/kernels/cuda/vnl_op.cu @@ -1,4 +1,6 @@ #include "source_pw/module_pwdft/kernels/vnl_op.h" + +#include #include "vnl_tools_cu.hpp" #include @@ -44,17 +46,18 @@ __global__ void cal_vnl( const int nbeta = atom_nb[it]; for (int nb = 0; nb < nbeta; nb++) { - const FPTYPE gnorm = sqrt(gk[ig * 3 + 0] * gk[ig * 3 + 0] + gk[ig * 3 + 1] * gk[ig * 3 + 1] + - gk[ig * 3 + 2] * gk[ig * 3 + 2]) * tpiba; + const std::int64_t g_offset = static_cast(ig) * 3; + const FPTYPE gnorm = sqrt(gk[g_offset + 0] * gk[g_offset + 0] + gk[g_offset + 1] * gk[g_offset + 1] + + gk[g_offset + 2] * gk[g_offset + 2]) * tpiba; vq = _polynomial_interpolation( tab, it, nb, tab_2, tab_3, DQ, gnorm); // add spherical harmonic part for (int ih = 0; ih < nh; ih++) { - if (nb == indv[it * nhm + ih]) { - const int lm = static_cast(nhtolm[it * nhm + ih]); - vkb1[ih * npw + ig] = ylm[lm * npw + ig] * vq; + if (nb == indv[static_cast(it) * nhm + ih]) { + const int lm = static_cast(nhtolm[static_cast(it) * nhm + ih]); + vkb1[static_cast(ih) * npw + ig] = ylm[static_cast(lm) * npw + ig] * vq; } } // end ih } // end nbeta @@ -63,9 +66,9 @@ __global__ void cal_vnl( // now add the structure factor and factor (-i)^l for (int ia = 0; ia < atom_na[it]; ia++) { for (int ih = 0; ih < nh; ih++) { - thrust::complex pref = pow(NEG_IMAG_UNIT, nhtol[it * nhm + ih]); //? - thrust::complex *pvkb = vkb_in + jkb * npwx; - pvkb[ig] = vkb1[ih * npw + ig] * sk[iat * npw + ig] * pref; + thrust::complex pref = pow(NEG_IMAG_UNIT, nhtol[static_cast(it) * nhm + ih]); //? + thrust::complex *pvkb = vkb_in + static_cast(jkb) * npwx; + pvkb[ig] = vkb1[static_cast(ih) * npw + ig] * sk[static_cast(iat) * npw + ig] * pref; ++jkb; } // end ih iat++; @@ -98,7 +101,7 @@ void cal_vnl_op::operator() ( const std::complex *sk, std::complex *vkb_in) { - int block = (npw + THREADS_PER_BLOCK - 1) / THREADS_PER_BLOCK; + int block = npw / THREADS_PER_BLOCK + (npw % THREADS_PER_BLOCK != 0); cal_vnl<<>>( ntype, npw, npwx, nhm, tab_2, tab_3, atom_na, atom_nb, atom_nh, @@ -114,4 +117,4 @@ void cal_vnl_op::operator() ( template struct cal_vnl_op; template struct cal_vnl_op; -} // namespace hamilt \ No newline at end of file +} // namespace hamilt diff --git a/source/source_pw/module_pwdft/kernels/cuda/vnl_tools_cu.hpp b/source/source_pw/module_pwdft/kernels/cuda/vnl_tools_cu.hpp index aaf7623f3a4..8f9f689a022 100644 --- a/source/source_pw/module_pwdft/kernels/cuda/vnl_tools_cu.hpp +++ b/source/source_pw/module_pwdft/kernels/cuda/vnl_tools_cu.hpp @@ -2,6 +2,8 @@ #define W_ABACUS_DEVELOP_ABACUS_DEVELOP_SOURCE_source_pw_HAMILT_PWDFT_KERNELS_CUDA_VNL_TOOLS_CU_HPP #include + +#include #ifdef __CUDA #include #endif @@ -28,10 +30,10 @@ __device__ FPTYPE _polynomial_interpolation(const FPTYPE* table, const FPTYPE x1 = 1.0 - x0; const FPTYPE x2 = 2.0 - x0; const FPTYPE x3 = 3.0 - x0; - const FPTYPE y = table[(dim1 * tab_2 + dim2) * tab_3 + iq + 0] * x1 * x2 * x3 / 6.0 - + table[(dim1 * tab_2 + dim2) * tab_3 + iq + 0 + 1] * x0 * x2 * x3 / 2.0 - - table[(dim1 * tab_2 + dim2) * tab_3 + iq + 0 + 2] * x1 * x0 * x3 / 2.0 - + table[(dim1 * tab_2 + dim2) * tab_3 + iq + 0 + 3] * x1 * x2 * x0 / 6.0; + const FPTYPE y = table[(static_cast(dim1) * tab_2 + dim2) * tab_3 + iq + 0] * x1 * x2 * x3 / 6.0 + + table[(static_cast(dim1) * tab_2 + dim2) * tab_3 + iq + 0 + 1] * x0 * x2 * x3 / 2.0 + - table[(static_cast(dim1) * tab_2 + dim2) * tab_3 + iq + 0 + 2] * x1 * x0 * x3 / 2.0 + + table[(static_cast(dim1) * tab_2 + dim2) * tab_3 + iq + 0 + 3] * x1 * x2 * x0 / 6.0; return y; } @@ -52,10 +54,10 @@ __device__ FPTYPE _polynomial_interpolation_nl(const FPTYPE* table, const FPTYPE x1 = 1.0 - x0; const FPTYPE x2 = 2.0 - x0; const FPTYPE x3 = 3.0 - x0; - const FPTYPE y = (table[(dim1 * tab_2 + dim2) * tab_3 + iq + 0] * (-x2 * x3 - x1 * x3 - x1 * x2) / 6.0 - + table[(dim1 * tab_2 + dim2) * tab_3 + iq + 0 + 1] * (+x2 * x3 - x0 * x3 - x0 * x2) / 2.0 - - table[(dim1 * tab_2 + dim2) * tab_3 + iq + 0 + 2] * (+x1 * x3 - x0 * x3 - x0 * x1) / 2.0 - + table[(dim1 * tab_2 + dim2) * tab_3 + iq + 0 + 3] * (+x1 * x2 - x0 * x2 - x0 * x1) / 6.0) + const FPTYPE y = (table[(static_cast(dim1) * tab_2 + dim2) * tab_3 + iq + 0] * (-x2 * x3 - x1 * x3 - x1 * x2) / 6.0 + + table[(static_cast(dim1) * tab_2 + dim2) * tab_3 + iq + 0 + 1] * (+x2 * x3 - x0 * x3 - x0 * x2) / 2.0 + - table[(static_cast(dim1) * tab_2 + dim2) * tab_3 + iq + 0 + 2] * (+x1 * x3 - x0 * x3 - x0 * x1) / 2.0 + + table[(static_cast(dim1) * tab_2 + dim2) * tab_3 + iq + 0 + 3] * (+x1 * x2 - x0 * x2 - x0 * x1) / 6.0) / table_interval; return y; diff --git a/source/source_pw/module_pwdft/kernels/cuda/wf_op.cu b/source/source_pw/module_pwdft/kernels/cuda/wf_op.cu index a5ab6cc0870..8b176340da7 100644 --- a/source/source_pw/module_pwdft/kernels/cuda/wf_op.cu +++ b/source/source_pw/module_pwdft/kernels/cuda/wf_op.cu @@ -1,5 +1,7 @@ #include "source_pw/module_pwdft/kernels/wf_op.h" +#include + #include #include @@ -44,11 +46,11 @@ __global__ void cal_sk( for (int ia = 0; ia < atom_na[it]; ia++) { FPTYPE arg = 0.0; for (int ii = 0; ii < 3; ii++) { - arg += kvec_c[ik * 3 + ii] * atom_tau[iat * 3 + ii]; + arg += kvec_c[static_cast(ik) * 3 + ii] * atom_tau[static_cast(iat) * 3 + ii]; } arg *= TWO_PI; const thrust::complex kphase = thrust::complex(cos(arg), -sin(arg)); - const int isz = igl2isz[ik * npwx + igl]; + const int isz = igl2isz[static_cast(ik) * npwx + igl]; int iz = isz % nz; const int is = isz / nz; const int ixy = is2fftixy[is]; @@ -63,8 +65,8 @@ __global__ void cal_sk( ix += rho_nx; iy += rho_ny; iz += rho_nz; - sk[iat * npw + igl] = kphase * eigts1[iat * eigts1_nc + ix] * eigts2[iat * eigts2_nc + iy] - * eigts3[iat * eigts3_nc + iz]; + sk[static_cast(iat) * npw + igl] = kphase * eigts1[static_cast(iat) * eigts1_nc + ix] * eigts2[static_cast(iat) * eigts2_nc + iy] + * eigts3[static_cast(iat) * eigts3_nc + iz]; iat++; } } @@ -97,7 +99,7 @@ void cal_sk_op::operator()(const base_device::D std::complex* eigts3, std::complex* sk) { - int block = (npw + THREADS_PER_BLOCK - 1) / THREADS_PER_BLOCK; + int block = npw / THREADS_PER_BLOCK + (npw % THREADS_PER_BLOCK != 0); cal_sk<<>>( ik, ntype, nx, ny, nz, @@ -120,4 +122,4 @@ void cal_sk_op::operator()(const base_device::D template struct cal_sk_op; template struct cal_sk_op; -} // namespace hamilt \ No newline at end of file +} // namespace hamilt diff --git a/source/source_pw/module_pwdft/kernels/exx_batch_op.cpp b/source/source_pw/module_pwdft/kernels/exx_batch_op.cpp new file mode 100644 index 00000000000..ed4a2527cc2 --- /dev/null +++ b/source/source_pw/module_pwdft/kernels/exx_batch_op.cpp @@ -0,0 +1,261 @@ +#include "source_pw/module_pwdft/kernels/exx_batch_op.h" + +#include + +namespace hamilt +{ + +// Generic host implementation of the EXX batched kernels: plain loops plus +// FFTW plans. The CUDA explicit specializations in kernels/cuda/exx_batch_op.cu +// provide the DEVICE_GPU versions; on non-CUDA builds the DEVICE_GPU +// instantiations below exist only to satisfy the linker (the ROCm GPU path +// never activates the batched path, so they are never executed). + +template +void exx_batch_density_real(const int nbands, + const int nrxx, + const T* nk_real_all, + const T* mq_real, + const double omega, + T* out) +{ + using Real = typename T::value_type; + const Real omega_inv = static_cast(1.0 / omega); + for (int n = 0; n < nbands; n++) + { + const T* nk_real = nk_real_all + n * nrxx; + T* out_n = out + n * nrxx; + for (int i = 0; i < nrxx; i++) + { + out_n[i] = nk_real[i] * std::conj(mq_real[i]) * omega_inv; + } + } +} + +template +void exx_batch_gather_pw(const int nbands, + const int npw, + const int nxyz, + const int* box_map, + const T* box, + T* pw) +{ + using Real = typename T::value_type; + const Real nxyz_inv = static_cast(1.0) / static_cast(nxyz); + for (int n = 0; n < nbands; n++) + { + const T* box_n = box + n * nxyz; + T* pw_n = pw + n * npw; + for (int ig = 0; ig < npw; ig++) + { + pw_n[ig] = box_n[box_map[ig]] * nxyz_inv; + } + } +} + +template +void exx_batch_mul_pot(const int nbands, const int npw, const Real* pot, T* pw) +{ + for (int n = 0; n < nbands; n++) + { + T* pw_n = pw + n * npw; + for (int ig = 0; ig < npw; ig++) + { + pw_n[ig] *= pot[ig]; + } + } +} + +template +void exx_batch_scatter_pw(const int nbands, + const int npw, + const int nxyz, + const int* box_map, + const T* pw, + T* box) +{ + for (int n = 0; n < nbands; n++) + { + const T* pw_n = pw + n * npw; + T* box_n = box + n * nxyz; + for (int ig = 0; ig < npw; ig++) + { + box_n[box_map[ig]] = pw_n[ig]; + } + } +} + +template +void exx_batch_mul_real(const int nbands, const int nrxx, const T* mq_real, T* data) +{ + for (int n = 0; n < nbands; n++) + { + T* data_n = data + n * nrxx; + for (int i = 0; i < nrxx; i++) + { + data_n[i] *= mq_real[i]; + } + } +} + +template +void exx_batch_gather_accum(const int nbands, + const int npwk, + const int nxyz, + const int* box_map, + const T* box, + const Real factor, + T* out, + const int out_stride) +{ + const Real factor_scaled = factor / static_cast(nxyz); + for (int n = 0; n < nbands; n++) + { + const T* box_n = box + n * nxyz; + T* out_n = out + n * out_stride; + for (int ig = 0; ig < npwk; ig++) + { + out_n[ig] += factor_scaled * box_n[box_map[ig]]; + } + } +} + +template +void exx_batch_scatter_wfc(const int nbands, + const int npwk, + const int nxyz, + const int* map, + const T* psi, + const int psi_stride, + T* box) +{ + for (int n = 0; n < nbands; n++) + { + const T* psi_n = psi + n * psi_stride; + T* box_n = box + n * nxyz; + for (int ig = 0; ig < npwk; ig++) + { + box_n[map[ig]] = psi_n[ig]; + } + } +} + +#define EXX_BATCH_INSTANTIATE(T, Real, DEV) \ + template void exx_batch_density_real(int, int, const T*, const T*, double, T*); \ + template void exx_batch_gather_pw(int, int, int, const int*, const T*, T*); \ + template void exx_batch_mul_pot(int, int, const Real*, T*); \ + template void exx_batch_scatter_pw(int, int, int, const int*, const T*, T*); \ + template void exx_batch_mul_real(int, int, const T*, T*); \ + template void exx_batch_gather_accum(int, int, int, const int*, const T*, Real, T*, int); \ + template void exx_batch_scatter_wfc(int, int, int, const int*, const T*, int, T*); + +EXX_BATCH_INSTANTIATE(std::complex, double, base_device::DEVICE_CPU) +EXX_BATCH_INSTANTIATE(std::complex, float, base_device::DEVICE_CPU) +#if !defined(__CUDA) +// Non-CUDA builds (including ROCm): the DEVICE_GPU instantiations of +// OperatorEXXPW reference these symbols, but the batched path is never +// activated there, so the host implementations are only link fodder. +EXX_BATCH_INSTANTIATE(std::complex, double, base_device::DEVICE_GPU) +EXX_BATCH_INSTANTIATE(std::complex, float, base_device::DEVICE_GPU) +#endif + +// Batched FFTW c2c plans (double precision; the single-precision ones live in +// exx_batch_op_float.cpp / exx_batch_op_float_stub.cpp). FFTW encodes the +// transform direction in the plan (unlike cuFFT), so the handle holds one +// plan per direction. The plans are created on a scratch buffer with +// FFTW_UNALIGNED and run through fftw_execute_dft, so any buffer of the right +// size can be transformed. +namespace +{ +struct ExxFftwPlanD +{ + fftw_plan fwd = nullptr; + fftw_plan bac = nullptr; +}; + +void exx_fftw_plan_create_d(void** plan, const int nx, const int ny, const int nz, const int batch) +{ + const int n[3] = {nx, ny, nz}; + const int nxyz = nx * ny * nz; + fftw_complex* tmp = reinterpret_cast(fftw_malloc(sizeof(fftw_complex) * nxyz * batch)); + auto* p = new ExxFftwPlanD; + const unsigned flag = FFTW_ESTIMATE | FFTW_UNALIGNED; + p->fwd = fftw_plan_many_dft(3, n, batch, tmp, nullptr, 1, nxyz, tmp, nullptr, 1, nxyz, FFTW_FORWARD, flag); + p->bac = fftw_plan_many_dft(3, n, batch, tmp, nullptr, 1, nxyz, tmp, nullptr, 1, nxyz, FFTW_BACKWARD, flag); + fftw_free(tmp); + *plan = p; +} + +void exx_fftw_exec_d(void* plan, std::complex* data, const bool forward) +{ + auto* p = reinterpret_cast(plan); + fftw_execute_dft(forward ? p->fwd : p->bac, + reinterpret_cast(data), + reinterpret_cast(data)); +} + +void exx_fftw_plan_destroy_d(void** plan) +{ + if (*plan != nullptr) + { + auto* p = reinterpret_cast(*plan); + fftw_destroy_plan(p->fwd); + fftw_destroy_plan(p->bac); + delete p; + *plan = nullptr; + } +} +} // namespace + +template <> +void exx_batch_fft_plan_create, base_device::DEVICE_CPU>(void** plan, + const int nx, + const int ny, + const int nz, + const int batch) +{ + exx_fftw_plan_create_d(plan, nx, ny, nz, batch); +} + +template <> +void exx_batch_fft_exec, base_device::DEVICE_CPU>(void* plan, + std::complex* data, + const bool forward) +{ + exx_fftw_exec_d(plan, data, forward); +} + +template <> +void exx_batch_fft_plan_destroy, base_device::DEVICE_CPU>(void** plan) +{ + exx_fftw_plan_destroy_d(plan); +} + +#if !defined(__CUDA) +// link fodder for the DEVICE_GPU instantiations, never executed (see above) +template <> +void exx_batch_fft_plan_create, base_device::DEVICE_GPU>(void** plan, + const int nx, + const int ny, + const int nz, + const int batch) +{ + exx_fftw_plan_create_d(plan, nx, ny, nz, batch); +} + +template <> +void exx_batch_fft_exec, base_device::DEVICE_GPU>(void* plan, + std::complex* data, + const bool forward) +{ + exx_fftw_exec_d(plan, data, forward); +} + +template <> +void exx_batch_fft_plan_destroy, base_device::DEVICE_GPU>(void** plan) +{ + exx_fftw_plan_destroy_d(plan); +} +#endif + +} // namespace hamilt diff --git a/source/source_pw/module_pwdft/kernels/exx_batch_op.h b/source/source_pw/module_pwdft/kernels/exx_batch_op.h new file mode 100644 index 00000000000..156c0d43489 --- /dev/null +++ b/source/source_pw/module_pwdft/kernels/exx_batch_op.h @@ -0,0 +1,91 @@ +#ifndef EXX_BATCH_OP_H +#define EXX_BATCH_OP_H + +#include "source_base/module_device/types.h" + +#include + +namespace hamilt +{ + +// Batched elementwise / gather-scatter kernels and batched FFT helpers for +// the n-band inner loop of OperatorEXXPW. All bands of one (iq, m_iband) +// pair are processed with a single kernel launch / batched FFT instead of +// one launch per band. The generic implementation (exx_batch_op.cpp) runs on +// the host with plain loops and FFTW plans; the CUDA explicit +// specializations (kernels/cuda/exx_batch_op.cu) run the same operations on +// the GPU with kernels and cuFFT. Instantiated for std::complex and +// std::complex. + +// out[n*nrxx + i] = nk_real_all[n*nrxx + i] * conj(mq_real[i]) / omega +template +void exx_batch_density_real(const int nbands, + const int nrxx, + const T* nk_real_all, + const T* mq_real, + const double omega, + T* out); + +// pw[n*npw + ig] = box[n*nxyz + box_map[ig]] / nxyz (real -> recip, gather) +template +void exx_batch_gather_pw(const int nbands, + const int npw, + const int nxyz, + const int* box_map, + const T* box, + T* pw); + +// pw[n*npw + ig] *= pot[ig] +template +void exx_batch_mul_pot(const int nbands, const int npw, const Real* pot, T* pw); + +// box[n*nxyz + box_map[ig]] = pw[n*npw + ig] (recip -> real, scatter; box pre-zeroed) +template +void exx_batch_scatter_pw(const int nbands, + const int npw, + const int nxyz, + const int* box_map, + const T* pw, + T* box); + +// data[n*nrxx + i] *= mq_real[i] +template +void exx_batch_mul_real(const int nbands, const int nrxx, const T* mq_real, T* data); + +// out[n*out_stride + ig] += factor / nxyz * box[n*nxyz + box_map[ig]] (real -> recip, gather + accumulate) +template +void exx_batch_gather_accum(const int nbands, + const int npwk, + const int nxyz, + const int* box_map, + const T* box, + const Real factor, + T* out, + const int out_stride); + +// box[n*nxyz + map[ig]] = psi[n*psi_stride + ig] (scatter PW coefficients into the box; box pre-zeroed) +template +void exx_batch_scatter_wfc(const int nbands, + const int npwk, + const int nxyz, + const int* map, + const T* psi, + const int psi_stride, + T* box); + +// Batched in-place 3D complex-to-complex FFT over the (nx, ny, nz) grid, +// batch = nbands transforms with contiguous distance nx*ny*nz, matching the +// layout of cufftPlan3d used in FFT_CUDA. +template +void exx_batch_fft_plan_create(void** plan, const int nx, const int ny, const int nz, const int batch); + +// forward = FFT_FORWARD, otherwise FFT_BACKWARD (unnormalized, as in FFT_CUDA) +template +void exx_batch_fft_exec(void* plan, T* data, const bool forward); + +template +void exx_batch_fft_plan_destroy(void** plan); + +} // namespace hamilt + +#endif // EXX_BATCH_OP_H diff --git a/source/source_pw/module_pwdft/kernels/exx_batch_op_float.cpp b/source/source_pw/module_pwdft/kernels/exx_batch_op_float.cpp new file mode 100644 index 00000000000..baeda9ea0ff --- /dev/null +++ b/source/source_pw/module_pwdft/kernels/exx_batch_op_float.cpp @@ -0,0 +1,107 @@ +#include "source_pw/module_pwdft/kernels/exx_batch_op.h" + +#include + +// Single-precision batched FFTW plans for the EXX batched path, compiled only +// when ENABLE_FLOAT_FFTW is on (see module_pwdft/CMakeLists.txt); otherwise +// exx_batch_op_float_stub.cpp provides the same symbols with a runtime error. +// Mirrors the double-precision implementation in exx_batch_op.cpp. + +namespace hamilt +{ + +namespace +{ +struct ExxFftwPlanF +{ + fftwf_plan fwd = nullptr; + fftwf_plan bac = nullptr; +}; + +void exx_fftw_plan_create_f(void** plan, const int nx, const int ny, const int nz, const int batch) +{ + const int n[3] = {nx, ny, nz}; + const int nxyz = nx * ny * nz; + fftwf_complex* tmp = reinterpret_cast(fftwf_malloc(sizeof(fftwf_complex) * nxyz * batch)); + auto* p = new ExxFftwPlanF; + const unsigned flag = FFTW_ESTIMATE | FFTW_UNALIGNED; + p->fwd = fftwf_plan_many_dft(3, n, batch, tmp, nullptr, 1, nxyz, tmp, nullptr, 1, nxyz, FFTW_FORWARD, flag); + p->bac = fftwf_plan_many_dft(3, n, batch, tmp, nullptr, 1, nxyz, tmp, nullptr, 1, nxyz, FFTW_BACKWARD, flag); + fftwf_free(tmp); + *plan = p; +} + +void exx_fftw_exec_f(void* plan, std::complex* data, const bool forward) +{ + auto* p = reinterpret_cast(plan); + fftwf_execute_dft(forward ? p->fwd : p->bac, + reinterpret_cast(data), + reinterpret_cast(data)); +} + +void exx_fftw_plan_destroy_f(void** plan) +{ + if (*plan != nullptr) + { + auto* p = reinterpret_cast(*plan); + fftwf_destroy_plan(p->fwd); + fftwf_destroy_plan(p->bac); + delete p; + *plan = nullptr; + } +} +} // namespace + +template <> +void exx_batch_fft_plan_create, base_device::DEVICE_CPU>(void** plan, + const int nx, + const int ny, + const int nz, + const int batch) +{ + exx_fftw_plan_create_f(plan, nx, ny, nz, batch); +} + +template <> +void exx_batch_fft_exec, base_device::DEVICE_CPU>(void* plan, + std::complex* data, + const bool forward) +{ + exx_fftw_exec_f(plan, data, forward); +} + +template <> +void exx_batch_fft_plan_destroy, base_device::DEVICE_CPU>(void** plan) +{ + exx_fftw_plan_destroy_f(plan); +} + +#if !defined(__CUDA) +// link fodder for the DEVICE_GPU instantiations, never executed (ROCm keeps +// the batched path disabled) +template <> +void exx_batch_fft_plan_create, base_device::DEVICE_GPU>(void** plan, + const int nx, + const int ny, + const int nz, + const int batch) +{ + exx_fftw_plan_create_f(plan, nx, ny, nz, batch); +} + +template <> +void exx_batch_fft_exec, base_device::DEVICE_GPU>(void* plan, + std::complex* data, + const bool forward) +{ + exx_fftw_exec_f(plan, data, forward); +} + +template <> +void exx_batch_fft_plan_destroy, base_device::DEVICE_GPU>(void** plan) +{ + exx_fftw_plan_destroy_f(plan); +} +#endif + +} // namespace hamilt diff --git a/source/source_pw/module_pwdft/kernels/exx_batch_op_float_stub.cpp b/source/source_pw/module_pwdft/kernels/exx_batch_op_float_stub.cpp new file mode 100644 index 00000000000..a12db41c9ad --- /dev/null +++ b/source/source_pw/module_pwdft/kernels/exx_batch_op_float_stub.cpp @@ -0,0 +1,72 @@ +#include "source_pw/module_pwdft/kernels/exx_batch_op.h" + +#include "source_base/tool_quit.h" + +// Stubs for the single-precision batched FFTW plans, compiled when +// ENABLE_FLOAT_FFTW is off (see module_pwdft/CMakeLists.txt); the real +// implementation lives in exx_batch_op_float.cpp. Float PW on CPU needs the +// single-precision FFTW library anyway, so reaching these means the build +// lacks it: fail with a clear message instead of a null-symbol crash. + +namespace hamilt +{ + +namespace +{ +[[noreturn]] void exx_fftw_float_missing() +{ + ModuleBase::WARNING_QUIT("exx_batch_fft", + "single-precision FFTW is required for the float EXX FFT on CPU; " + "rebuild with ENABLE_FLOAT_FFTW=ON"); +} +} // namespace + +template <> +void exx_batch_fft_plan_create, base_device::DEVICE_CPU>(void**, + const int, + const int, + const int, + const int) +{ + exx_fftw_float_missing(); +} + +template <> +void exx_batch_fft_exec, base_device::DEVICE_CPU>(void*, std::complex*, const bool) +{ + exx_fftw_float_missing(); +} + +template <> +void exx_batch_fft_plan_destroy, base_device::DEVICE_CPU>(void** plan) +{ + // no plan could have been created, but stay harmless at process teardown + *plan = nullptr; +} + +#if !defined(__CUDA) +// link fodder for the DEVICE_GPU instantiations, never executed +template <> +void exx_batch_fft_plan_create, base_device::DEVICE_GPU>(void**, + const int, + const int, + const int, + const int) +{ + exx_fftw_float_missing(); +} + +template <> +void exx_batch_fft_exec, base_device::DEVICE_GPU>(void*, std::complex*, const bool) +{ + exx_fftw_float_missing(); +} + +template <> +void exx_batch_fft_plan_destroy, base_device::DEVICE_GPU>(void** plan) +{ + *plan = nullptr; +} +#endif + +} // namespace hamilt diff --git a/source/source_pw/module_pwdft/kernels/nonlocal_op.cpp b/source/source_pw/module_pwdft/kernels/nonlocal_op.cpp index cfa6c51c03c..233a0fceda1 100644 --- a/source/source_pw/module_pwdft/kernels/nonlocal_op.cpp +++ b/source/source_pw/module_pwdft/kernels/nonlocal_op.cpp @@ -1,6 +1,9 @@ #include "source_pw/module_pwdft/kernels/nonlocal_op.h" -namespace hamilt { +#include + +namespace hamilt +{ template struct nonlocal_pw_op @@ -23,61 +26,77 @@ struct nonlocal_pw_op #ifdef _OPENMP #pragma omp parallel for collapse(3) #endif - for (int ii = 0; ii < l1; ii++) { - // each atom has nproj, means this is with structure factor; - // each projector (each atom) must multiply coefficient - // with all the other projectors. - for (int jj = 0; jj < l2; ++jj) - for (int kk = 0; kk < l3; kk++) - for (int xx = 0; xx < l3; xx++) - ps[(sum + ii * l3 + kk) * l2 + jj] - += deeq[((spin * deeq_x + iat + ii) * deeq_y + xx) * deeq_z + kk] - * becp[jj * nkb + sum + ii * l3 + xx]; + for (int ii = 0; ii < l1; ii++) + { + // each atom has nproj, means this is with structure factor; + // each projector (each atom) must multiply coefficient + // with all the other projectors. + for (int jj = 0; jj < l2; ++jj) + { + for (int kk = 0; kk < l3; kk++) + { + for (int xx = 0; xx < l3; xx++) + { + const std::int64_t projector_offset = sum + static_cast(ii) * l3; + const std::int64_t output_index = (projector_offset + kk) * l2 + jj; + const std::int64_t input_index = static_cast(jj) * nkb + projector_offset + xx; + const std::int64_t deeq_index = ((static_cast(spin) * deeq_x + iat + ii) * deeq_y + xx) * deeq_z + kk; + ps[output_index] += deeq[deeq_index] * becp[input_index]; + } + } + } + } + sum += l1 * l3; + iat += l1; } - sum += l1 * l3; - iat += l1; - } - void operator()(const base_device::DEVICE_CPU* dev, - const int& l1, - const int& l2, - const int& l3, - int& sum, - int& iat, - const int& nkb, - const int& deeq_x, - const int& deeq_y, - const int& deeq_z, - const std::complex* deeq_nc, - std::complex* ps, - const std::complex* becp) - { + void operator()(const base_device::DEVICE_CPU* dev, + const int& l1, + const int& l2, + const int& l3, + int& sum, + int& iat, + const int& nkb, + const int& deeq_x, + const int& deeq_y, + const int& deeq_z, + const std::complex* deeq_nc, + std::complex* ps, + const std::complex* becp) + { #ifdef _OPENMP #pragma omp parallel for collapse(3) #endif - for (int ii = 0; ii < l1; ii++) - // each atom has nproj, means this is with structure factor; - // each projector (each atom) must multiply coefficient - // with all the other projectors. - for (int jj = 0; jj < l2; jj+=2) - for (int kk = 0; kk < l3; kk++) - for (int xx = 0; xx < l3; xx++) - { - int psind = (sum + ii * l3 + kk) * l2 + jj; - int becpind = jj * nkb + sum + ii * l3 + xx; - auto &becp1 = becp[becpind]; - auto &becp2 = becp[becpind + nkb]; - ps[psind] += deeq_nc[((0 * deeq_x + iat + ii) * deeq_y + kk) * deeq_z + xx] * becp1 - + deeq_nc[((1 * deeq_x + iat + ii) * deeq_y + kk) * deeq_z + xx] * becp2; - ps[psind + 1] += deeq_nc[((2 * deeq_x + iat + ii) * deeq_y + kk) * deeq_z + xx] * becp1 - + deeq_nc[((3 * deeq_x + iat + ii) * deeq_y + kk) * deeq_z + xx] * becp2; - } // end jj - iat += l1; - sum += l1 * l3; - } + for (int ii = 0; ii < l1; ii++) + { + // each atom has nproj, means this is with structure factor; + // each projector (each atom) must multiply coefficient + // with all the other projectors. + for (int jj = 0; jj < l2; jj += 2) + { + for (int kk = 0; kk < l3; kk++) + { + for (int xx = 0; xx < l3; xx++) + { + const std::int64_t projector_offset = sum + static_cast(ii) * l3; + const std::int64_t psind = (projector_offset + kk) * l2 + jj; + const std::int64_t becpind = static_cast(jj) * nkb + projector_offset + xx; + const std::int64_t spin_stride = static_cast(deeq_x) * deeq_y * deeq_z; + const std::int64_t deeq_index = ((static_cast(iat) + ii) * deeq_y + kk) * deeq_z + xx; + auto& becp1 = becp[becpind]; + auto& becp2 = becp[becpind + nkb]; + ps[psind] += deeq_nc[deeq_index] * becp1 + deeq_nc[spin_stride + deeq_index] * becp2; + ps[psind + 1] += deeq_nc[2 * spin_stride + deeq_index] * becp1 + deeq_nc[3 * spin_stride + deeq_index] * becp2; + } // end jj + } + } + } + iat += l1; + sum += l1 * l3; + } }; template struct nonlocal_pw_op; template struct nonlocal_pw_op; -} // namespace hamilt \ No newline at end of file +} // namespace hamilt diff --git a/source/source_pw/module_pwdft/kernels/projector_gradient_op.cpp b/source/source_pw/module_pwdft/kernels/projector_gradient_op.cpp new file mode 100644 index 00000000000..60196eea48c --- /dev/null +++ b/source/source_pw/module_pwdft/kernels/projector_gradient_op.cpp @@ -0,0 +1,17 @@ +#include "source_pw/module_pwdft/kernels/projector_gradient_op.h" +namespace hamilt { +template +void projector_gradient_op::operator()(int npw, int ld, int nkb, int nbeta, int nq, + Real dq, Real tpiba, const int* metadata, const Real* q, const Real* tab, const Real* derivative, + const std::complex* sk, std::complex* out) const +{ +#ifdef _OPENMP +#pragma omp parallel for collapse(2) +#endif + for (int p=0; p; +template class projector_gradient_op; +} diff --git a/source/source_pw/module_pwdft/kernels/projector_gradient_op.h b/source/source_pw/module_pwdft/kernels/projector_gradient_op.h new file mode 100644 index 00000000000..8f37db20a3f --- /dev/null +++ b/source/source_pw/module_pwdft/kernels/projector_gradient_op.h @@ -0,0 +1,146 @@ +#ifndef PW_PROJECTOR_GRADIENT_OP_H +#define PW_PROJECTOR_GRADIENT_OP_H +#include "source_base/module_device/types.h" +#include +#include +#include +namespace hamilt +{ +/** @brief Evaluate a normalized real solid harmonic and its Cartesian derivatives. */ +template +#if defined(__CUDACC__) || defined(__HIPCC__) +__host__ __device__ +#endif +void gradient_harmonic(const int l, const int lm, const Real* q, Real* value) +{ + const int index = lm - l * l; + const int m = (index + 1) / 2; + const bool imaginary = index > 0 && index % 2 == 0; + const Real radius = sqrt(q[0]*q[0] + q[1]*q[1] + q[2]*q[2]); + for (int d = 0; d < 4; ++d) { value[d] = Real(0); } + if (radius < Real(1e-9)) + { + if (l == 0) { value[0] = Real(0.28209479177387814347); } + return; + } + const Real x = q[0]/radius; + const Real y = q[1]/radius; + const Real z = q[2]/radius; + Real re[4] = {Real(1), Real(0), Real(0), Real(0)}; + Real im[4] = {Real(0), Real(0), Real(0), Real(0)}; + // Cartesian solid harmonics avoid polar-axis singularities. + for (int k = 1; k <= m; ++k) + { + const Real factor = Real(-(2*k-1)); + const Real old_re = re[0]; + const Real old_im = im[0]; + for (int d = 0; d < 4; ++d) + { + const Real r = re[d]; + re[d] = factor * (x*r-y*im[d] + (d==1 ? old_re : Real(0)) - (d==2 ? old_im : Real(0))); + im[d] = factor * (x*im[d]+y*r + (d==1 ? old_im : Real(0)) + (d==2 ? old_re : Real(0))); + } + } + Real previous[4] = {}; + Real current[4]; + for (int d = 0; d < 4; ++d) { current[d] = imaginary ? im[d] : re[d]; } + for (int n = m+1; n <= l; ++n) + { + Real next[4]; + for (int d = 0; d < 4; ++d) + { + const Real coordinate = d==1 ? x : (d==2 ? y : z); + next[d] = (Real(2*n-1)*(z*current[d] + (d==3 ? current[0] : Real(0))) + - Real(n+m-1)*(previous[d] + (d>0 ? Real(2)*coordinate*previous[0] : Real(0)))) / Real(n-m); + } + for (int d = 0; d < 4; ++d) { previous[d]=current[d]; current[d]=next[d]; } + } + Real normalization = Real(2*l+1) / Real(12.56637061435917295385); + for (int k=l-m+1; k<=l+m; ++k) { normalization /= Real(k); } + normalization = sqrt(normalization * (m==0 ? Real(1) : Real(2))); + value[0] = normalization * current[0]; + for (int d=1; d<4; ++d) + { + value[d] = normalization * (current[d] - Real(l)*current[0]*q[d-1]/radius) / radius; + } +} + +/** @brief Device projector-gradient assembly; metadata is [type,beta,l,lm,atom]. */ +template +class projector_gradient_op +{ + public: + void operator()(int npw, int ld, int nkb, int nbeta, int nq, Real dq, Real tpiba, + const int* metadata, const Real* q, const Real* tab, const Real* derivative, + const std::complex* sk, std::complex* out) const; +}; + +#if defined(__CUDA) || defined(__ROCM) +template +class projector_gradient_op +{ + public: + void operator()(int npw, int ld, int nkb, int nbeta, int nq, Real dq, Real tpiba, + const int* metadata, const Real* q, const Real* tab, const Real* derivative, + const std::complex* sk, std::complex* out) const; +}; +#endif + +template +#if defined(__CUDACC__) || defined(__HIPCC__) +__host__ __device__ +#endif +Real gradient_interpolate(const Real* table, const Real position) +{ + const int index = static_cast(position); + const Real x = position - Real(index); + return table[index]*(Real(1)-x)*(Real(2)-x)*(Real(3)-x)/Real(6) + + table[index+1]*x*(Real(2)-x)*(Real(3)-x)/Real(2) + - table[index+2]*x*(Real(1)-x)*(Real(3)-x)/Real(2) + + table[index+3]*x*(Real(1)-x)*(Real(2)-x)/Real(6); +} + +template +#if defined(__CUDACC__) || defined(__HIPCC__) +__host__ __device__ +#endif +void gradient_element(const int ig, const int projector, const int npw, const int ld, const int nkb, + const int nbeta, const int nq, const Real dq, const Real tpiba, + const int* metadata, const Real* q, const Real* tab, const Real* derivative, + const Complex* sk, Complex* out) +{ + const int* meta = metadata + 5 * static_cast(projector); + const int l = meta[2]; + const int lm = meta[3]; + const Real* momentum = q + 3 * static_cast(ig); + const Real radius = sqrt(momentum[0]*momentum[0]+momentum[1]*momentum[1]+momentum[2]*momentum[2]); + const Real position = radius*tpiba/dq; + Real harmonic[4]; + gradient_harmonic(l, lm, momentum, harmonic); + const std::int64_t radial_offset = (static_cast(meta[0]) * nbeta + meta[1]) * nq; + const Real* radial = tab + radial_offset; + const Real* radial_d = derivative + radial_offset; + const Real v = gradient_interpolate(radial, position); + const Real dv = gradient_interpolate(radial_d, position); + const int phase_index = l % 4; + const Complex phase = phase_index==0 ? Complex(1,0) : (phase_index==1 ? Complex(0,-1) + : (phase_index==2 ? Complex(-1,0) : Complex(0,1))); + for (int d=0; d<3; ++d) + { + Real gradient = Real(0); + if (radius < Real(1e-8)) + { + if (lm==1 && d==2) { gradient = dv*Real(0.48860251190291992159); } + if ((lm==2 && d==0) || (lm==3 && d==1)) { gradient = -dv*Real(0.48860251190291992159); } + } + else + { + gradient = harmonic[0]*dv*momentum[d]/radius + harmonic[d+1]*v/tpiba; + } + const std::int64_t output_index = (static_cast(d) * nkb + projector) * ld + ig; + const std::int64_t structure_index = static_cast(meta[4]) * npw + ig; + out[output_index] = gradient * sk[structure_index] * phase; + } +} +} // namespace hamilt +#endif diff --git a/source/source_pw/module_pwdft/kernels/rocm/nonlocal_op.hip.cu b/source/source_pw/module_pwdft/kernels/rocm/nonlocal_op.hip.cu index e4c14cfd5a1..b686534ccac 100644 --- a/source/source_pw/module_pwdft/kernels/rocm/nonlocal_op.hip.cu +++ b/source/source_pw/module_pwdft/kernels/rocm/nonlocal_op.hip.cu @@ -1,6 +1,7 @@ #include "source_pw/module_pwdft/kernels/nonlocal_op.h" #include +#include #include #include @@ -30,12 +31,16 @@ __global__ void nonlocal_pw( const int jj = blockIdx.x % l2; for (int kk = threadIdx.x; kk < l3; kk += blockDim.x) { thrust::complex res(0.0, 0.0); + const std::int64_t projector_offset = sum + static_cast(ii) * l3; for (int xx = 0; xx < l3; xx++) { + const std::int64_t deeq_index = ((static_cast(spin) * deeq_x + iat + ii) * deeq_y + xx) * deeq_z + kk; + const std::int64_t input_index = static_cast(jj) * nkb + projector_offset + xx; res - += deeq[((spin * deeq_x + iat + ii) * deeq_y + xx) * deeq_z + kk] - * becp[jj * nkb + sum + ii * l3 + xx]; + += deeq[deeq_index] + * becp[input_index]; } - ps[(sum + ii * l3 + kk) * l2 + jj] += res; + const std::int64_t output_index = (projector_offset + kk) * l2 + jj; + ps[output_index] += res; } } @@ -54,20 +59,24 @@ __global__ void nonlocal_pw( thrust::complex* ps, const thrust::complex* becp) { - const int ii = blockIdx.x * 2 / l2; - const int jj = blockIdx.x * 2 % l2; + const std::int64_t spinor_index = static_cast(blockIdx.x) * 2; + const int ii = spinor_index / l2; + const int jj = spinor_index % l2; for (int kk = threadIdx.x; kk < l3; kk += blockDim.x) { thrust::complex res1(0.0, 0.0); thrust::complex res2(0.0, 0.0); - int psind = (sum + ii * l3 + kk) * l2 + jj; + const std::int64_t projector_offset = sum + static_cast(ii) * l3; + const std::int64_t psind = (projector_offset + kk) * l2 + jj; + const std::int64_t spin_stride = static_cast(deeq_x) * deeq_y * deeq_z; for (int xx = 0; xx < l3; xx++) { - int becpind = jj * nkb + sum + ii * l3 + xx; + const std::int64_t becpind = static_cast(jj) * nkb + projector_offset + xx; + const std::int64_t deeq_index = ((static_cast(iat) + ii) * deeq_y + kk) * deeq_z + xx; thrust::complex becp1 = becp[becpind]; thrust::complex becp2 = becp[becpind + nkb]; - res1 += deeq_nc[((0 * deeq_x + iat + ii) * deeq_y + kk) * deeq_z + xx] * becp1 - + deeq_nc[((1 * deeq_x + iat + ii) * deeq_y + kk) * deeq_z + xx] * becp2; - res2 += deeq_nc[((2 * deeq_x + iat + ii) * deeq_y + kk) * deeq_z + xx] * becp1 - + deeq_nc[((3 * deeq_x + iat + ii) * deeq_y + kk) * deeq_z + xx] * becp2; + res1 += deeq_nc[deeq_index] * becp1 + + deeq_nc[spin_stride + deeq_index] * becp2; + res2 += deeq_nc[2 * spin_stride + deeq_index] * becp1 + + deeq_nc[3 * spin_stride + deeq_index] * becp2; } ps[psind] += res1; ps[psind + 1] += res2; @@ -122,7 +131,9 @@ void hamilt::nonlocal_pw_op::operator()(const b { // denghui implement 20221109 // <<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<< - hipLaunchKernelGGL(HIP_KERNEL_NAME(nonlocal_pw), dim3(l1 * l2 / 2), dim3(THREADS_PER_BLOCK), 0, 0, + const std::int64_t block_count = static_cast(l1) * l2 / 2; + const dim3 grid(block_count); + hipLaunchKernelGGL(HIP_KERNEL_NAME(nonlocal_pw), grid, dim3(THREADS_PER_BLOCK), 0, 0, l1, l2, l3, // loop size sum, iat, nkb, // control params deeq_x, deeq_y, deeq_z, diff --git a/source/source_pw/module_pwdft/kernels/rocm/projector_gradient_op.hip.cu b/source/source_pw/module_pwdft/kernels/rocm/projector_gradient_op.hip.cu new file mode 100644 index 00000000000..b79526666ee --- /dev/null +++ b/source/source_pw/module_pwdft/kernels/rocm/projector_gradient_op.hip.cu @@ -0,0 +1,27 @@ +#include "source_pw/module_pwdft/kernels/projector_gradient_op.h" +#include +#include +#include +namespace hamilt { +template +__global__ void gradient_kernel(int npw, int ld, int nkb, int nbeta, int nq, Real dq, Real tpiba, + const int* metadata, const Real* q, const Real* tab, const Real* derivative, + const thrust::complex* sk, thrust::complex* out) +{ + const int ig=blockIdx.x*blockDim.x+threadIdx.x; + if(ig(blockIdx.y),npw,ld,nkb,nbeta,nq,dq,tpiba,metadata,q,tab,derivative,sk,out); +} +template +void projector_gradient_op::operator()(int npw, int ld, int nkb, int nbeta, int nq, + Real dq, Real tpiba, const int* metadata, const Real* q, const Real* tab, const Real* derivative, + const std::complex* sk, std::complex* out) const +{ + if(npw==0 || nkb==0) return; + gradient_kernel<<>>(npw,ld,nkb,nbeta,nq,dq,tpiba,metadata,q,tab,derivative, + reinterpret_cast*>(sk),reinterpret_cast*>(out)); + const hipError_t error=hipGetLastError(); + if(error!=hipSuccess) throw std::runtime_error(hipGetErrorString(error)); +} +template class projector_gradient_op; +template class projector_gradient_op; +} diff --git a/source/source_pw/module_pwdft/kernels/rocm/vnl_op.hip.cu b/source/source_pw/module_pwdft/kernels/rocm/vnl_op.hip.cu index 94a234e0bbc..76de8dd80e1 100644 --- a/source/source_pw/module_pwdft/kernels/rocm/vnl_op.hip.cu +++ b/source/source_pw/module_pwdft/kernels/rocm/vnl_op.hip.cu @@ -1,4 +1,6 @@ #include "source_pw/module_pwdft/kernels/vnl_op.h" + +#include #include "source_pw/module_pwdft/kernels/cuda/vnl_tools_cu.hpp" #include @@ -44,17 +46,18 @@ __global__ void cal_vnl( const int nbeta = atom_nb[it]; for (int nb = 0; nb < nbeta; nb++) { - const FPTYPE gnorm = sqrt(gk[ig * 3 + 0] * gk[ig * 3 + 0] + gk[ig * 3 + 1] * gk[ig * 3 + 1] + - gk[ig * 3 + 2] * gk[ig * 3 + 2]) * tpiba; + const std::int64_t g_offset = static_cast(ig) * 3; + const FPTYPE gnorm = sqrt(gk[g_offset + 0] * gk[g_offset + 0] + gk[g_offset + 1] * gk[g_offset + 1] + + gk[g_offset + 2] * gk[g_offset + 2]) * tpiba; vq = _polynomial_interpolation( tab, it, nb, tab_2, tab_3, DQ, gnorm); // add spherical harmonic part for (int ih = 0; ih < nh; ih++) { - if (nb == indv[it * nhm + ih]) { - const int lm = static_cast(nhtolm[it * nhm + ih]); - vkb1[ih * npw + ig] = ylm[lm * npw + ig] * vq; + if (nb == indv[static_cast(it) * nhm + ih]) { + const int lm = static_cast(nhtolm[static_cast(it) * nhm + ih]); + vkb1[static_cast(ih) * npw + ig] = ylm[static_cast(lm) * npw + ig] * vq; } } // end ih } // end nbeta @@ -63,9 +66,9 @@ __global__ void cal_vnl( // now add the structure factor and factor (-i)^l for (int ia = 0; ia < atom_na[it]; ia++) { for (int ih = 0; ih < nh; ih++) { - thrust::complex pref = pow(NEG_IMAG_UNIT, nhtol[it * nhm + ih]); //? - thrust::complex *pvkb = vkb_in + jkb * npwx; - pvkb[ig] = vkb1[ih * npw + ig] * sk[iat * npw + ig] * pref; + thrust::complex pref = pow(NEG_IMAG_UNIT, nhtol[static_cast(it) * nhm + ih]); //? + thrust::complex *pvkb = vkb_in + static_cast(jkb) * npwx; + pvkb[ig] = vkb1[static_cast(ih) * npw + ig] * sk[static_cast(iat) * npw + ig] * pref; ++jkb; } // end ih iat++; @@ -98,7 +101,7 @@ void cal_vnl_op::operator() ( const std::complex *sk, std::complex *vkb_in) { - int block = (npw + THREADS_PER_BLOCK - 1) / THREADS_PER_BLOCK; + int block = npw / THREADS_PER_BLOCK + (npw % THREADS_PER_BLOCK != 0); hipLaunchKernelGGL(HIP_KERNEL_NAME(cal_vnl), dim3(block), dim3(THREADS_PER_BLOCK), 0, 0, ntype, npw, npwx, nhm, tab_2, tab_3, atom_na, atom_nb, atom_nh, @@ -114,4 +117,4 @@ void cal_vnl_op::operator() ( template struct cal_vnl_op; template struct cal_vnl_op; -} // namespace hamilt \ No newline at end of file +} // namespace hamilt diff --git a/source/source_pw/module_pwdft/kernels/rocm/wf_op.hip.cu b/source/source_pw/module_pwdft/kernels/rocm/wf_op.hip.cu index d03a35bc616..c7bb7f9a056 100644 --- a/source/source_pw/module_pwdft/kernels/rocm/wf_op.hip.cu +++ b/source/source_pw/module_pwdft/kernels/rocm/wf_op.hip.cu @@ -1,5 +1,7 @@ #include "source_pw/module_pwdft/kernels/wf_op.h" +#include + #include #include @@ -43,11 +45,11 @@ __global__ void cal_sk( for (int ia = 0; ia < atom_na[it]; ia++) { FPTYPE arg = 0.0; for (int ii = 0; ii < 3; ii++) { - arg += kvec_c[ik * 3 + ii] * atom_tau[iat * 3 + ii]; + arg += kvec_c[static_cast(ik) * 3 + ii] * atom_tau[static_cast(iat) * 3 + ii]; } arg *= TWO_PI; const thrust::complex kphase = thrust::complex(cos(arg), -sin(arg)); - const int isz = igl2isz[ik * npwx + igl]; + const int isz = igl2isz[static_cast(ik) * npwx + igl]; int iz = isz % nz; const int is = isz / nz; const int ixy = is2fftixy[is]; @@ -62,8 +64,8 @@ __global__ void cal_sk( ix += rho_nx; iy += rho_ny; iz += rho_nz; - sk[iat * npw + igl] = kphase * eigts1[iat * eigts1_nc + ix] * eigts2[iat * eigts2_nc + iy] - * eigts3[iat * eigts3_nc + iz]; + sk[static_cast(iat) * npw + igl] = kphase * eigts1[static_cast(iat) * eigts1_nc + ix] * eigts2[static_cast(iat) * eigts2_nc + iy] + * eigts3[static_cast(iat) * eigts3_nc + iz]; iat++; } } @@ -96,7 +98,7 @@ void cal_sk_op::operator()(const base_device::D std::complex* eigts3, std::complex* sk) { - int block = (npw + THREADS_PER_BLOCK - 1) / THREADS_PER_BLOCK; + int block = npw / THREADS_PER_BLOCK + (npw % THREADS_PER_BLOCK != 0); hipLaunchKernelGGL(HIP_KERNEL_NAME(cal_sk), dim3(block), dim3(THREADS_PER_BLOCK), 0, 0, ik, ntype, nx, ny, nz, @@ -119,4 +121,4 @@ void cal_sk_op::operator()(const base_device::D template struct cal_sk_op; template struct cal_sk_op; -} // namespace hamilt \ No newline at end of file +} // namespace hamilt diff --git a/source/source_pw/module_pwdft/kernels/stress_op.cpp b/source/source_pw/module_pwdft/kernels/stress_op.cpp index aff61a53d69..66f45f3672b 100644 --- a/source/source_pw/module_pwdft/kernels/stress_op.cpp +++ b/source/source_pw/module_pwdft/kernels/stress_op.cpp @@ -590,17 +590,17 @@ void Simpson_Integral assert(mesh&1); asum = 0.00; - const size_t end = mesh-2; + const size_t end = mesh-2; for( size_t i=1; i!=end; i+=2 ) { - const double f1 = func[i]*rab[i]; - asum += f1 + f1 + func[i+1]*rab[i+1]; + const double f1 = func[i]*rab[i]; + asum += f1 + f1 + func[i+1]*rab[i+1]; } - const double f1 = func[mesh-2]*rab[mesh-2]; - asum += f1+f1; - asum += asum; - asum += func[0]*rab[0] + func[mesh-1]*rab[mesh-1]; - asum /= 3.0; + const double f1 = func[mesh-2]*rab[mesh-2]; + asum += f1+f1; + asum += asum; + asum += func[0]*rab[0] + func[mesh-1]*rab[mesh-1]; + asum /= 3.0; return; }// end subroutine simpson diff --git a/source/source_pw/module_pwdft/kernels/vnl_op.cpp b/source/source_pw/module_pwdft/kernels/vnl_op.cpp index 55b73a5af56..ca4ac7f12cb 100644 --- a/source/source_pw/module_pwdft/kernels/vnl_op.cpp +++ b/source/source_pw/module_pwdft/kernels/vnl_op.cpp @@ -1,5 +1,7 @@ #include "source_pw/module_pwdft/kernels/vnl_op.h" +#include + #include "vnl_tools.hpp" namespace hamilt @@ -53,8 +55,9 @@ struct cal_vnl_op #endif for (int ig = 0; ig < npw; ig++) { - const FPTYPE gnorm = sqrt(gk[ig * 3 + 0] * gk[ig * 3 + 0] + gk[ig * 3 + 1] * gk[ig * 3 + 1] - + gk[ig * 3 + 2] * gk[ig * 3 + 2]) + const std::int64_t g_offset = static_cast(ig) * 3; + const FPTYPE gnorm = sqrt(gk[g_offset + 0] * gk[g_offset + 0] + gk[g_offset + 1] * gk[g_offset + 1] + + gk[g_offset + 2] * gk[g_offset + 2]) * tpiba; vq = _polynomial_interpolation(tab, it, nb, tab_2, tab_3, DQ, gnorm); @@ -62,10 +65,10 @@ struct cal_vnl_op // add spherical harmonic part for (int ih = 0; ih < nh; ih++) { - if (nb == indv[it * nhm + ih]) + if (nb == indv[static_cast(it) * nhm + ih]) { - const int lm = static_cast(nhtolm[it * nhm + ih]); - vkb1[ih * npw + ig] = ylm[lm * npw + ig] * vq; + const int lm = static_cast(nhtolm[static_cast(it) * nhm + ih]); + vkb1[static_cast(ih) * npw + ig] = ylm[static_cast(lm) * npw + ig] * vq; } } // end ih } @@ -78,14 +81,14 @@ struct cal_vnl_op for (int ih = 0; ih < nh; ih++) { // std::complex pref = pow(NEG_IMAG_UNIT, nhtol[it * nhm + ih]); //? - std::complex pref = pref_tab[int(nhtol[it * nhm + ih]) % imag_pow_period]; - std::complex* pvkb = vkb_in + jkb * npwx; + std::complex pref = pref_tab[int(nhtol[static_cast(it) * nhm + ih]) % imag_pow_period]; + std::complex* pvkb = vkb_in + static_cast(jkb) * npwx; #ifdef _OPENMP #pragma omp for #endif for (int ig = 0; ig < npw; ig++) { - pvkb[ig] = vkb1[ih * npw + ig] * sk[iat * npw + ig] * pref; + pvkb[ig] = vkb1[static_cast(ih) * npw + ig] * sk[static_cast(iat) * npw + ig] * pref; } ++jkb; } // end ih diff --git a/source/source_pw/module_pwdft/kernels/vnl_tools.hpp b/source/source_pw/module_pwdft/kernels/vnl_tools.hpp index eb9f42d6a27..28ca8779dc3 100644 --- a/source/source_pw/module_pwdft/kernels/vnl_tools.hpp +++ b/source/source_pw/module_pwdft/kernels/vnl_tools.hpp @@ -1,4 +1,6 @@ #ifndef W_ABACUS_DEVELOP_ABACUS_DEVELOP_SOURCE_source_pw_HAMILT_PWDFT_KERNELS_VNL_TOOLS_HPP + +#include #define W_ABACUS_DEVELOP_ABACUS_DEVELOP_SOURCE_source_pw_HAMILT_PWDFT_KERNELS_VNL_TOOLS_HPP namespace hamilt @@ -19,10 +21,10 @@ static inline FPTYPE _polynomial_interpolation(const FPTYPE* table, const FPTYPE x1 = 1.0 - x0; const FPTYPE x2 = 2.0 - x0; const FPTYPE x3 = 3.0 - x0; - const FPTYPE y = table[(dim1 * tab_2 + dim2) * tab_3 + iq + 0] * x1 * x2 * x3 / 6.0 - + table[(dim1 * tab_2 + dim2) * tab_3 + iq + 0 + 1] * x0 * x2 * x3 / 2.0 - - table[(dim1 * tab_2 + dim2) * tab_3 + iq + 0 + 2] * x1 * x0 * x3 / 2.0 - + table[(dim1 * tab_2 + dim2) * tab_3 + iq + 0 + 3] * x1 * x2 * x0 / 6.0; + const FPTYPE y = table[(static_cast(dim1) * tab_2 + dim2) * tab_3 + iq + 0] * x1 * x2 * x3 / 6.0 + + table[(static_cast(dim1) * tab_2 + dim2) * tab_3 + iq + 0 + 1] * x0 * x2 * x3 / 2.0 + - table[(static_cast(dim1) * tab_2 + dim2) * tab_3 + iq + 0 + 2] * x1 * x0 * x3 / 2.0 + + table[(static_cast(dim1) * tab_2 + dim2) * tab_3 + iq + 0 + 3] * x1 * x2 * x0 / 6.0; // ModuleBase::timer::start("PolyInt","Poly_Interpo_2"); return y; @@ -44,10 +46,10 @@ static inline FPTYPE _polynomial_interpolation_nl(const FPTYPE* table, const FPTYPE x1 = 1.0 - x0; const FPTYPE x2 = 2.0 - x0; const FPTYPE x3 = 3.0 - x0; - const FPTYPE y = (table[(dim1 * tab_2 + dim2) * tab_3 + iq + 0] * (-x2 * x3 - x1 * x3 - x1 * x2) / 6.0 - + table[(dim1 * tab_2 + dim2) * tab_3 + iq + 0 + 1] * (+x2 * x3 - x0 * x3 - x0 * x2) / 2.0 - - table[(dim1 * tab_2 + dim2) * tab_3 + iq + 0 + 2] * (+x1 * x3 - x0 * x3 - x0 * x1) / 2.0 - + table[(dim1 * tab_2 + dim2) * tab_3 + iq + 0 + 3] * (+x1 * x2 - x0 * x2 - x0 * x1) / 6.0) + const FPTYPE y = (table[(static_cast(dim1) * tab_2 + dim2) * tab_3 + iq + 0] * (-x2 * x3 - x1 * x3 - x1 * x2) / 6.0 + + table[(static_cast(dim1) * tab_2 + dim2) * tab_3 + iq + 0 + 1] * (+x2 * x3 - x0 * x3 - x0 * x2) / 2.0 + - table[(static_cast(dim1) * tab_2 + dim2) * tab_3 + iq + 0 + 2] * (+x1 * x3 - x0 * x3 - x0 * x1) / 2.0 + + table[(static_cast(dim1) * tab_2 + dim2) * tab_3 + iq + 0 + 3] * (+x1 * x2 - x0 * x2 - x0 * x1) / 6.0) / table_interval; return y; diff --git a/source/source_pw/module_pwdft/kernels/wf_op.cpp b/source/source_pw/module_pwdft/kernels/wf_op.cpp index 80c4b3eb04c..c942e484326 100644 --- a/source/source_pw/module_pwdft/kernels/wf_op.cpp +++ b/source/source_pw/module_pwdft/kernels/wf_op.cpp @@ -1,4 +1,6 @@ #include "source_pw/module_pwdft/kernels/wf_op.h" + +#include #include "source_base/libm/libm.h" namespace hamilt{ @@ -41,7 +43,7 @@ struct cal_sk_op for (int ia = 0; ia < atom_na[it]; ia++) { FPTYPE arg = 0.0; for (int ii = 0; ii < 3; ii++) { - arg += kvec_c[ik * 3 + ii] * atom_tau[iat * 3 + ii]; + arg += kvec_c[static_cast(ik) * 3 + ii] * atom_tau[static_cast(iat) * 3 + ii]; } arg *= TWO_PI; FPTYPE sinp, cosp; @@ -51,7 +53,7 @@ struct cal_sk_op #pragma omp for #endif for (int igl = 0; igl < npw; ++igl) { - const int isz = igl2isz[ik * npwx + igl]; + const int isz = igl2isz[static_cast(ik) * npwx + igl]; int iz = isz % nz; const int is = isz / nz; const int ixy = is2fftixy[is]; @@ -66,8 +68,8 @@ struct cal_sk_op ix += rho_nx; iy += rho_ny; iz += rho_nz; - sk[iat * npw + igl] = kphase * eigts1[iat * eigts1_nc + ix] * eigts2[iat * eigts2_nc + iy] - * eigts3[iat * eigts3_nc + iz]; + sk[static_cast(iat) * npw + igl] = kphase * eigts1[static_cast(iat) * eigts1_nc + ix] * eigts2[static_cast(iat) * eigts2_nc + iy] + * eigts3[static_cast(iat) * eigts3_nc + iz]; } iat++; } @@ -82,4 +84,3 @@ template struct cal_sk_op; template struct cal_sk_op; } // namespace hamilt - diff --git a/source/source_pw/module_pwdft/nonlocal_maths.cpp b/source/source_pw/module_pwdft/nonlocal_maths.cpp new file mode 100644 index 00000000000..fb1ff95a912 --- /dev/null +++ b/source/source_pw/module_pwdft/nonlocal_maths.cpp @@ -0,0 +1,426 @@ +#include "nonlocal_maths.h" + +namespace hamilt +{ + +template +Nonlocal_maths::Nonlocal_maths(const pseudopot_cell_vnl* nlpp_in, const UnitCell* ucell_in) +{ + this->device = base_device::get_device_type(this->ctx); + this->nhtol_ = nlpp_in->nhtol; + this->lmax_ = nlpp_in->lmaxkb; + this->ucell_ = ucell_in; +} + +template +Nonlocal_maths::Nonlocal_maths(const ModuleBase::matrix& nhtol, + const int lmax, + const UnitCell* ucell_in) +{ + this->device = base_device::get_device_type(this->ctx); + this->nhtol_ = nhtol; + this->lmax_ = lmax; + this->ucell_ = ucell_in; +} + +// prepare a memory block containing information of vector G+k, this function can be named as eval_q or eval_gk +// seems this operation is not on gpu +template +std::vector Nonlocal_maths::cal_gk(int ik, const ModulePW::PW_Basis_K* pw_basis) +{ + int npw = pw_basis->npwk[ik]; + std::vector gk(npw * 5); + ModuleBase::Vector3 q; + for (int ig = 0; ig < npw; ++ig) + { + // written in memory block from 0 to 3*npw. This is like a matrix with npw rows and 3 columns + q = pw_basis->getgpluskcar(ik, ig); + gk[ig * 3] = q.x; + gk[ig * 3 + 1] = q.y; + gk[ig * 3 + 2] = q.z; + // the following written in memory block from 3*npw to 5*npw, the excess 2*npw is for norm and 1/norm + // for memory consecutive consideration, there are blocks storing the norm and 1/norm. + FPTYPE norm = sqrt(q.norm2()); + gk[3 * npw + ig] = norm * this->ucell_->tpiba; // one line with length npw, storing the norm + gk[4 * npw + ig] = norm < 1e-8 ? 0.0 : 1.0 / norm * this->ucell_->tpiba; // one line with length npw, storing 1/norm + } + return gk; +} + +// tabulate the spherical haromonic functions up to lmax. The q vector is given as input. +// I would rather call this function as cal_ylm_cpu2gpu, distincting from the pure cpu implementation +template +void Nonlocal_maths::cal_ylm(int lmax, int npw, const FPTYPE* q, FPTYPE* ylm) +{ + const int ntot_ylm = (lmax + 1) * (lmax + 1); + if (this->device == base_device::GpuDevice) + { + // alias + using syncmem_var_h2d_op = base_device::memory::synchronize_memory_op; + // allocate + std::vector ylm_cpu(ntot_ylm * npw); + // calculate + ModuleBase::YlmReal::Ylm_Real(cpu_ctx, ntot_ylm, npw, q, ylm_cpu.data()); + // send from cpu to gpu + syncmem_var_h2d_op()(ylm, ylm_cpu.data(), ylm_cpu.size()); + } + else + { + // calculate. Why not implement this logic branch inside some function??? + ModuleBase::YlmReal::Ylm_Real(cpu_ctx, ntot_ylm, npw, q, ylm); + } + return; +} + +// this function calculate the numerical derivate of the spherical harmonic functions respect to the G vector... +// maybe called eval_dylmdq_cpu2gpu? +template +void Nonlocal_maths::cal_ylm_deri(int lmax, int npw, const FPTYPE* q, FPTYPE* out) +{ + const int ntot_ylm = (lmax + 1) * (lmax + 1); + + if (this->device == base_device::GpuDevice) + { + // alias + using syncmem_var_h2d_op = base_device::memory::synchronize_memory_op; + // allocate + std::vector dylmdq_cpu(3 * ntot_ylm * npw); + // calculate + for (int ipol = 0; ipol < 3; ipol++) + { + Nonlocal_maths::dylmr2(ntot_ylm, npw, q, &dylmdq_cpu[ipol * ntot_ylm * npw], ipol); + } + // send from cpu to gpu + syncmem_var_h2d_op()(out, dylmdq_cpu.data(), dylmdq_cpu.size()); + } + else + { + for (int ipol = 0; ipol < 3; ipol++) + { + Nonlocal_maths::dylmr2(ntot_ylm, npw, q, &out[ipol * ntot_ylm * npw], ipol); + } + } + + return; +} +// cal_pref +template +std::vector> Nonlocal_maths::cal_pref(int it, const int nh) +{ + // nh is the total number of m-channels of the beta functions + // for example, if angular momentum of beta functions are 0, 0, 1, 1, 1, 1, the nh will be + // 1 + 1 + 3 + 3 + 3 + 3 = 14 + std::vector> pref(nh); + for (int ih = 0; ih < nh; ih++) + { + pref[ih] = std::pow(std::complex(0.0, -1.0), this->nhtol_(it, ih)); + // it is actually nh2l, which means to get the angular momentum... + } + return pref; +} + +// cal_vkb +// cpu version first, gpu version later +template +void Nonlocal_maths::cal_vkb(int it, + int ia, + int npw, + const FPTYPE* vq_in, + const FPTYPE* ylm_in, + const std::complex* sk_in, + const std::complex* pref_in, + std::complex* vkb_out) +{ + int ih = 0; + // loop over all beta functions + for (int ib = 0; ib < this->ucell_->atoms[it].ncpp.nbeta; ib++) + { + int l = this->nhtol_(it, ih); + // loop over all m angular momentum + for (int m = 0; m < 2 * l + 1; m++) + { + int lm = l * l + m; + std::complex* vkb_ptr = &vkb_out[ih * npw]; + const FPTYPE* ylm_ptr = &ylm_in[lm * npw]; + const FPTYPE* vq_ptr = &vq_in[ib * npw]; + // loop over all G-vectors + for (int ig = 0; ig < npw; ig++) + { + vkb_ptr[ig] = ylm_ptr[ig] * vq_ptr[ig] * sk_in[ig] * pref_in[ih]; + } + ih++; + } + } +} + +// cal_vkb +// cpu version first, gpu version later +template +void Nonlocal_maths::cal_vkb_deri(int it, + int ia, + int npw, + int ipol, + int jpol, + const FPTYPE* vq_in, + const FPTYPE* vq_deri_in, + const FPTYPE* ylm_in, + const FPTYPE* ylm_deri_in, + const std::complex* sk_in, + const std::complex* pref_in, + const FPTYPE* gk_in, + std::complex* vkb_out) +{ + const int x1 = (this->lmax_ + 1) * (this->lmax_ + 1); + int ih = 0; + // loop over all beta functions + for (int nb = 0; nb < this->ucell_->atoms[it].ncpp.nbeta; nb++) + { + const int l = this->nhtol_(it, ih); + // loop over all m angular momentum + for (int m = 0; m < 2 * l + 1; m++) + { + const int lm = l * l + m; + std::complex* vkb_ptr = &vkb_out[ih * npw]; + const FPTYPE* ylm_ptr = &ylm_in[lm * npw]; + const FPTYPE* vq_ptr = &vq_in[nb * npw]; + // set vkb to zero + for (int ig = 0; ig < npw; ig++) + { + vkb_ptr[ig] = std::complex(0.0, 0.0); + } + // first term: ylm * vq * sk * pref + // loop over all G-vectors + if (ipol == jpol) + { + for (int ig = 0; ig < npw; ig++) + { + vkb_ptr[ig] -= ylm_ptr[ig] * vq_ptr[ig] * sk_in[ig] * pref_in[ih]; + } + } + // second term: ylm_deri * vq_deri * sk * pref + // loop over all G-vectors + const FPTYPE* ylm_deri_ptr1 = &ylm_deri_in[(ipol * x1 + lm) * npw]; + const FPTYPE* ylm_deri_ptr2 = &ylm_deri_in[(jpol * x1 + lm) * npw]; + const FPTYPE* vq_deri_ptr = &vq_deri_in[nb * npw]; + const FPTYPE* qnorm = &gk_in[4 * npw]; + for (int ig = 0; ig < npw; ig++) + { + vkb_ptr[ig] -= (gk_in[ig * 3 + ipol] * ylm_deri_ptr2[ig] + gk_in[ig * 3 + jpol] * ylm_deri_ptr1[ig]) + * vq_ptr[ig] * sk_in[ig] * pref_in[ih]; + } + // third term: ylm * vq_deri * sk * pref + // loop over all G-vectors + for (int ig = 0; ig < npw; ig++) + { + vkb_ptr[ig] -= 2.0 * ylm_ptr[ig] * vq_deri_ptr[ig] * sk_in[ig] * pref_in[ih] * gk_in[ig * 3 + ipol] + * gk_in[ig * 3 + jpol] * qnorm[ig]; + } + ih++; + } + } +} + +template +void Nonlocal_maths::prepare_vkb_ptr(int nbeta, + double* nhtol, + int nhtol_nc, + int npw, + int it, + std::complex* vkb_out, + std::complex** vkb_ptrs, + FPTYPE* ylm_in, + FPTYPE** ylm_ptrs, + FPTYPE* vq_in, + FPTYPE** vq_ptrs) +{ + // std::complex** vkb_ptrs[nh]; + // const FPTYPE** ylm_ptrs[nh]; + // const FPTYPE** vq_ptrs[nh]; + int ih = 0; + for (int nb = 0; nb < nbeta; nb++) + { + int l = nhtol[it * nhtol_nc + ih]; + for (int m = 0; m < 2 * l + 1; m++) + { + int lm = l * l + m; + vkb_ptrs[ih] = &vkb_out[ih * npw]; + ylm_ptrs[ih] = &ylm_in[lm * npw]; + vq_ptrs[ih] = &vq_in[nb * npw]; + ih++; + } + } +} + +template +void Nonlocal_maths::cal_dvkb_index(const int nbeta, + const double* nhtol, + const int nhtol_nc, + const int npw, + const int it, + const int ipol, + const int jpol, + int* indexes) +{ + int ih = 0; + const int x1 = (this->lmax_ + 1) * (this->lmax_ + 1); + for (int nb = 0; nb < nbeta; nb++) + { + int l = nhtol[it * nhtol_nc + ih]; + for (int m = 0; m < 2 * l + 1; m++) + { + //std::cout << "in function cal_dvkb_index, nhtol(" << it << ", " << ih << ") = " << l << std::endl; + int lm = l * l + m; + indexes[ih * 4] = lm; // the index of ylm matrix, for given l and m, together with ig to get value + indexes[ih * 4 + 1] = nb; // the iproj of present atom type + indexes[ih * 4 + 2] = (ipol * x1 + lm); + indexes[ih * 4 + 3] = (jpol * x1 + lm); + + ih++; + } + } +} + +template +void Nonlocal_maths::dylmr2(const int nylm, + const int ngy, + const FPTYPE* gk, + FPTYPE* dylm, + const int ipol) +{ + //----------------------------------------------------------------------- + // + // compute \partial Y_lm(G) \over \partial (G)_ipol + // using simple numerical derivation (SdG) + // The spherical harmonics are calculated in ylmr2 + // + // int nylm, ngy, ipol; + // number of spherical harmonics + // the number of g vectors to compute + // desired polarization + // FPTYPE g (3, ngy), gg (ngy), dylm (ngy, nylm) + // the coordinates of g vectors + // the moduli of g vectors + // the spherical harmonics + // the spherical harmonics derivatives + // + const FPTYPE delta = 1e-6; + const FPTYPE small = 1e-15; + + ModuleBase::matrix ylmaux; + // dg is the finite increment for numerical derivation: + // dg = delta |G| = delta * sqrt(gg) + // dgi= 1 /(delta * sqrt(gg)) + // gx = g +/- dg + + std::vector gx(ngy * 3); + + std::vector dg(ngy); + std::vector dgi(ngy); + + ylmaux.create(nylm, ngy); + + ModuleBase::GlobalFunc::ZEROS(dylm, nylm * ngy); + ylmaux.zero_out(); + +#ifdef _OPENMP +#pragma omp parallel for +#endif + for (int ig = 0; ig < 3 * ngy; ig++) + { + gx[ig] = gk[ig]; + } + //$OMP PARALLEL DO DEFAULT(SHARED) PRIVATE(ig) +#ifdef _OPENMP +#pragma omp parallel for +#endif + for (int ig = 0; ig < ngy; ig++) + { + const int igx = ig * 3, igy = ig * 3 + 1, igz = ig * 3 + 2; + FPTYPE norm2 = gx[igx] * gx[igx] + gx[igy] * gx[igy] + gx[igz] * gx[igz]; + dg[ig] = delta * sqrt(norm2); + if (dg[ig] > small) + { + dgi[ig] = 1.0 / dg[ig]; + } + else + { + dgi[ig] = 0.0; + } + } + //$OMP END PARALLEL DO + + //$OMP PARALLEL DO DEFAULT(SHARED) PRIVATE(ig) +#ifdef _OPENMP +#pragma omp parallel for +#endif + for (int ig = 0; ig < ngy; ig++) + { + const int index = ig * 3 + ipol; + gx[index] = gk[index] + dg[ig]; + } + //$OMP END PARALLEL DO + + base_device::DEVICE_CPU* cpu = {}; + ModuleBase::YlmReal::Ylm_Real(cpu, nylm, ngy, gx.data(), dylm); + //$OMP PARALLEL DO DEFAULT(SHARED) PRIVATE(ig) +#ifdef _OPENMP +#pragma omp parallel for +#endif + for (int ig = 0; ig < ngy; ig++) + { + const int index = ig * 3 + ipol; + gx[index] = gk[index] - dg[ig]; + } + //$OMP END PARALLEL DO + + ModuleBase::YlmReal::Ylm_Real(cpu, nylm, ngy, gx.data(), ylmaux.c); + + // zaxpy ( - 1.0, ylmaux, 1, dylm, 1); +#ifdef _OPENMP +#pragma omp parallel for collapse(2) +#endif + for (int lm = 0; lm < nylm; lm++) + { + for (int ig = 0; ig < ngy; ig++) + { + dylm[lm * ngy + ig] -= ylmaux(lm, ig); + dylm[lm * ngy + ig] *= 0.5 * dgi[ig]; + } + } + + return; +} + +template +FPTYPE Nonlocal_maths::Polynomial_Interpolation_nl(const ModuleBase::realArray& table, + const int& dim1, + const int& dim2, + const FPTYPE& table_interval, + const FPTYPE& x // input value +) +{ + + assert(table_interval > 0.0); + const FPTYPE position = x / table_interval; + const int iq = static_cast(position); + + const FPTYPE x0 = position - static_cast(iq); + const FPTYPE x1 = 1.0 - x0; + const FPTYPE x2 = 2.0 - x0; + const FPTYPE x3 = 3.0 - x0; + const FPTYPE y = (table(dim1, dim2, iq) * (-x2 * x3 - x1 * x3 - x1 * x2) / 6.0 + + table(dim1, dim2, iq + 1) * (+x2 * x3 - x0 * x3 - x0 * x2) / 2.0 + - table(dim1, dim2, iq + 2) * (+x1 * x3 - x0 * x3 - x0 * x1) / 2.0 + + table(dim1, dim2, iq + 3) * (+x1 * x2 - x0 * x2 - x0 * x1) / 6.0) + / table_interval; + + return y; +} + +// Explicit template instantiation +template class Nonlocal_maths; +#if ((defined __CUDA) || (defined __ROCM)) +template class Nonlocal_maths; +#endif + +} // namespace hamilt diff --git a/source/source_pw/module_pwdft/nonlocal_maths.h b/source/source_pw/module_pwdft/nonlocal_maths.h new file mode 100644 index 00000000000..74489f76ed5 --- /dev/null +++ b/source/source_pw/module_pwdft/nonlocal_maths.h @@ -0,0 +1,119 @@ +#ifndef HAMILTPW_NONLOCAL_MATHS_H +#define HAMILTPW_NONLOCAL_MATHS_H + +#include "source_base/math_ylmreal.h" +#include "source_base/module_device/device.h" +#include "source_basis/module_pw/pw_basis_k.h" +#include "source_cell/klist.h" +#include "source_cell/unitcell.h" +#include "source_pw/module_pwdft/vnl_pw.h" +#include "source_pw/module_pwdft/kernels/stress_op.h" +#include "source_base/kernels/math_kernel_op.h" + +namespace hamilt +{ + +template +class Nonlocal_maths +{ + public: + Nonlocal_maths(const pseudopot_cell_vnl* nlpp_in, const UnitCell* ucell_in); + Nonlocal_maths(const ModuleBase::matrix& nhtol, const int lmax, const UnitCell* ucell_in); + + private: + ModuleBase::matrix nhtol_; + int lmax_ = 0; + const UnitCell* ucell_ = nullptr; + + Device* ctx = {}; + base_device::DEVICE_CPU* cpu_ctx = {}; + base_device::AbacusDevice_t device = {}; + + public: + // functions + /** + * @brief this function prepares all the q (G+k) information in one contiguous memory block + * including the x, y and z components, its norm and the reciprocal of its norm + * + * @param ik index of k point + * @param pw_basis the plane wave basis + * @return std::vector 1d contiguous memory block containing all the q information. The + * first 3*npw are data of x, y and z components, the next 2*npw are data of norm and 1/norm. + * This is beneficial for GPU memory access. + */ + std::vector cal_gk(int ik, const ModulePW::PW_Basis_K* pw_basis); + /** + * @brief calculate the real spherical harmonic functions on cpu (and optionally send to gpu, + * if gpu is available) + * + * @param lmax [in] maximum angular momentum to calculate + * @param npw [in] number of G+k vectors + * @param gk_in [in] the G+k vectors + * @param ylm [out] the spherical harmonic functions + */ + void cal_ylm(int lmax, int npw, const FPTYPE* gk_in, FPTYPE* ylm); + /// calculate the derivate of the sperical bessel function for projections + void cal_ylm_deri(int lmax, int npw, const FPTYPE* gk_in, FPTYPE* ylm_deri); + /// calculate the (-i)^l factors + std::vector > cal_pref(int it, const int nh); + /// calculate the vkb matrix for this atom + /// vkb = sum_lm (-i)^l * ylm(g^) * vq(g^) * sk(g^) + void cal_vkb(int it, + int ia, + int npw, + const FPTYPE* vq_in, + const FPTYPE* ylm_in, + const std::complex* sk_in, + const std::complex* pref_in, + std::complex* vkb_out); + /// calculate the dvkb matrix for this atom + void cal_vkb_deri(int it, + int ia, + int npw, + int ipol, + int jpol, + const FPTYPE* vq_in, + const FPTYPE* vq_deri_in, + const FPTYPE* ylm_in, + const FPTYPE* ylm_deri_in, + const std::complex* sk_in, + const std::complex* pref_in, + const FPTYPE* gk_in, + std::complex* vkb_out); + + /// calculate the ptr used in vkb_op + void prepare_vkb_ptr(int nbeta, + double* nhtol, + int nhtol_nc, + int npw, + int it, + std::complex* vkb_out, + std::complex** vkb_ptrs, + FPTYPE* ylm_in, + FPTYPE** ylm_ptrs, + FPTYPE* vq_in, + FPTYPE** vq_ptrs); + + /// calculate the indexes used in vkb_deri_op + /// indexes save (lm, nb, dylm_lm_ipol, dylm_lm_jpol) for nh + void cal_dvkb_index(const int nbeta, + const double* nhtol, + const int nhtol_nc, + const int npw, + const int it, + const int ipol, + const int jpol, + int* indexes); + + static void dylmr2(const int nylm, const int ngy, const FPTYPE* gk, FPTYPE* dylm, const int ipol); + /// polynomial interpolation tool for calculate derivate of vq + static FPTYPE Polynomial_Interpolation_nl(const ModuleBase::realArray& table, + const int& dim1, + const int& dim2, + const FPTYPE& table_interval, + const FPTYPE& x); +}; + +} // namespace hamilt + +#endif diff --git a/source/source_pw/module_pwdft/nonlocal_maths.hpp b/source/source_pw/module_pwdft/nonlocal_maths.hpp deleted file mode 100644 index d53b1cc6e4d..00000000000 --- a/source/source_pw/module_pwdft/nonlocal_maths.hpp +++ /dev/null @@ -1,522 +0,0 @@ -#ifndef HAMILTPW_NONLOCAL_MATHS_H -#define HAMILTPW_NONLOCAL_MATHS_H - -#include "source_base/module_device/device.h" -#include "source_basis/module_pw/pw_basis_k.h" -#include "source_cell/klist.h" -#include "source_cell/unitcell.h" -#include "source_pw/module_pwdft/vnl_pw.h" -#include "source_pw/module_pwdft/kernels/stress_op.h" -#include "source_base/kernels/math_kernel_op.h" - -namespace hamilt -{ - -template -class Nonlocal_maths -{ - public: - Nonlocal_maths(const pseudopot_cell_vnl* nlpp_in, const UnitCell* ucell_in) - { - this->device = base_device::get_device_type(this->ctx); - this->nhtol_ = nlpp_in->nhtol; - this->lmax_ = nlpp_in->lmaxkb; - this->ucell_ = ucell_in; - } - Nonlocal_maths(const ModuleBase::matrix& nhtol, const int lmax, const UnitCell* ucell_in) - { - this->device = base_device::get_device_type(this->ctx); - this->nhtol_ = nhtol; - this->lmax_ = lmax; - this->ucell_ = ucell_in; - } - - private: - ModuleBase::matrix nhtol_; - int lmax_ = 0; - const UnitCell* ucell_ = nullptr; - - Device* ctx = {}; - base_device::DEVICE_CPU* cpu_ctx = {}; - base_device::AbacusDevice_t device = {}; - - public: - // functions - /** - * @brief this function prepares all the q (G+k) information in one contiguous memory block - * including the x, y and z components, its norm and the reciprocal of its norm - * - * @param ik index of k point - * @param pw_basis the plane wave basis - * @return std::vector 1d contiguous memory block containing all the q information. The - * first 3*npw are data of x, y and z components, the next 2*npw are data of norm and 1/norm. - * This is beneficial for GPU memory access. - */ - std::vector cal_gk(int ik, const ModulePW::PW_Basis_K* pw_basis); - /** - * @brief calculate the real spherical harmonic functions on cpu (and optionally send to gpu, - * if gpu is available) - * - * @param lmax [in] maximum angular momentum to calculate - * @param npw [in] number of G+k vectors - * @param gk_in [in] the G+k vectors - * @param ylm [out] the spherical harmonic functions - */ - void cal_ylm(int lmax, int npw, const FPTYPE* gk_in, FPTYPE* ylm); - /// calculate the derivate of the sperical bessel function for projections - void cal_ylm_deri(int lmax, int npw, const FPTYPE* gk_in, FPTYPE* ylm_deri); - /// calculate the (-i)^l factors - std::vector > cal_pref(int it, const int nh); - /// calculate the vkb matrix for this atom - /// vkb = sum_lm (-i)^l * ylm(g^) * vq(g^) * sk(g^) - void cal_vkb(int it, - int ia, - int npw, - const FPTYPE* vq_in, - const FPTYPE* ylm_in, - const std::complex* sk_in, - const std::complex* pref_in, - std::complex* vkb_out); - /// calculate the dvkb matrix for this atom - void cal_vkb_deri(int it, - int ia, - int npw, - int ipol, - int jpol, - const FPTYPE* vq_in, - const FPTYPE* vq_deri_in, - const FPTYPE* ylm_in, - const FPTYPE* ylm_deri_in, - const std::complex* sk_in, - const std::complex* pref_in, - const FPTYPE* gk_in, - std::complex* vkb_out); - - /// calculate the ptr used in vkb_op - void prepare_vkb_ptr(int nbeta, - double* nhtol, - int nhtol_nc, - int npw, - int it, - std::complex* vkb_out, - std::complex** vkb_ptrs, - FPTYPE* ylm_in, - FPTYPE** ylm_ptrs, - FPTYPE* vq_in, - FPTYPE** vq_ptrs); - - /// calculate the indexes used in vkb_deri_op - /// indexes save (lm, nb, dylm_lm_ipol, dylm_lm_jpol) for nh - void cal_dvkb_index(const int nbeta, - const double* nhtol, - const int nhtol_nc, - const int npw, - const int it, - const int ipol, - const int jpol, - int* indexes); - - static void dylmr2(const int nylm, const int ngy, const FPTYPE* gk, FPTYPE* dylm, const int ipol); - /// polynomial interpolation tool for calculate derivate of vq - static FPTYPE Polynomial_Interpolation_nl(const ModuleBase::realArray& table, - const int& dim1, - const int& dim2, - const FPTYPE& table_interval, - const FPTYPE& x); -}; - -// prepare a memory block containing information of vector G+k, this function can be named as eval_q or eval_gk -// seems this operation is not on gpu -template -std::vector Nonlocal_maths::cal_gk(int ik, const ModulePW::PW_Basis_K* pw_basis) -{ - int npw = pw_basis->npwk[ik]; - std::vector gk(npw * 5); - ModuleBase::Vector3 q; - for (int ig = 0; ig < npw; ++ig) - { - // written in memory block from 0 to 3*npw. This is like a matrix with npw rows and 3 columns - q = pw_basis->getgpluskcar(ik, ig); - gk[ig * 3] = q.x; - gk[ig * 3 + 1] = q.y; - gk[ig * 3 + 2] = q.z; - // the following written in memory block from 3*npw to 5*npw, the excess 2*npw is for norm and 1/norm - // for memory consecutive consideration, there are blocks storing the norm and 1/norm. - FPTYPE norm = sqrt(q.norm2()); - gk[3 * npw + ig] = norm * this->ucell_->tpiba; // one line with length npw, storing the norm - gk[4 * npw + ig] = norm < 1e-8 ? 0.0 : 1.0 / norm * this->ucell_->tpiba; // one line with length npw, storing 1/norm - } - return gk; -} - -// tabulate the spherical haromonic functions up to lmax. The q vector is given as input. -// I would rather call this function as cal_ylm_cpu2gpu, distincting from the pure cpu implementation -template -void Nonlocal_maths::cal_ylm(int lmax, int npw, const FPTYPE* q, FPTYPE* ylm) -{ - const int ntot_ylm = (lmax + 1) * (lmax + 1); - if (this->device == base_device::GpuDevice) - { - // alias - using syncmem_var_h2d_op = base_device::memory::synchronize_memory_op; - // allocate - std::vector ylm_cpu(ntot_ylm * npw); - // calculate - ModuleBase::YlmReal::Ylm_Real(cpu_ctx, ntot_ylm, npw, q, ylm_cpu.data()); - // send from cpu to gpu - syncmem_var_h2d_op()(ylm, ylm_cpu.data(), ylm_cpu.size()); - } - else - { - // calculate. Why not implement this logic branch inside some function??? - ModuleBase::YlmReal::Ylm_Real(cpu_ctx, ntot_ylm, npw, q, ylm); - } - return; -} - -// this function calculate the numerical derivate of the spherical harmonic functions respect to the G vector... -// maybe called eval_dylmdq_cpu2gpu? -template -void Nonlocal_maths::cal_ylm_deri(int lmax, int npw, const FPTYPE* q, FPTYPE* out) -{ - const int ntot_ylm = (lmax + 1) * (lmax + 1); - - if (this->device == base_device::GpuDevice) - { - // alias - using syncmem_var_h2d_op = base_device::memory::synchronize_memory_op; - // allocate - std::vector dylmdq_cpu(3 * ntot_ylm * npw); - // calculate - for (int ipol = 0; ipol < 3; ipol++) - { - Nonlocal_maths::dylmr2(ntot_ylm, npw, q, &dylmdq_cpu[ipol * ntot_ylm * npw], ipol); - } - // send from cpu to gpu - syncmem_var_h2d_op()(out, dylmdq_cpu.data(), dylmdq_cpu.size()); - } - else - { - for (int ipol = 0; ipol < 3; ipol++) - { - Nonlocal_maths::dylmr2(ntot_ylm, npw, q, &out[ipol * ntot_ylm * npw], ipol); - } - } - - return; -} -// cal_pref -template -std::vector> Nonlocal_maths::cal_pref(int it, const int nh) -{ - // nh is the total number of m-channels of the beta functions - // for example, if angular momentum of beta functions are 0, 0, 1, 1, 1, 1, the nh will be - // 1 + 1 + 3 + 3 + 3 + 3 = 14 - std::vector> pref(nh); - for (int ih = 0; ih < nh; ih++) - { - pref[ih] = std::pow(std::complex(0.0, -1.0), this->nhtol_(it, ih)); - // it is actually nh2l, which means to get the angular momentum... - } - return pref; -} - -// cal_vkb -// cpu version first, gpu version later -template -void Nonlocal_maths::cal_vkb(int it, - int ia, - int npw, - const FPTYPE* vq_in, - const FPTYPE* ylm_in, - const std::complex* sk_in, - const std::complex* pref_in, - std::complex* vkb_out) -{ - int ih = 0; - // loop over all beta functions - for (int ib = 0; ib < this->ucell_->atoms[it].ncpp.nbeta; ib++) - { - int l = this->nhtol_(it, ih); - // loop over all m angular momentum - for (int m = 0; m < 2 * l + 1; m++) - { - int lm = l * l + m; - std::complex* vkb_ptr = &vkb_out[ih * npw]; - const FPTYPE* ylm_ptr = &ylm_in[lm * npw]; - const FPTYPE* vq_ptr = &vq_in[ib * npw]; - // loop over all G-vectors - for (int ig = 0; ig < npw; ig++) - { - vkb_ptr[ig] = ylm_ptr[ig] * vq_ptr[ig] * sk_in[ig] * pref_in[ih]; - } - ih++; - } - } -} - -// cal_vkb -// cpu version first, gpu version later -template -void Nonlocal_maths::cal_vkb_deri(int it, - int ia, - int npw, - int ipol, - int jpol, - const FPTYPE* vq_in, - const FPTYPE* vq_deri_in, - const FPTYPE* ylm_in, - const FPTYPE* ylm_deri_in, - const std::complex* sk_in, - const std::complex* pref_in, - const FPTYPE* gk_in, - std::complex* vkb_out) -{ - const int x1 = (this->lmax_ + 1) * (this->lmax_ + 1); - int ih = 0; - // loop over all beta functions - for (int nb = 0; nb < this->ucell_->atoms[it].ncpp.nbeta; nb++) - { - const int l = this->nhtol_(it, ih); - // loop over all m angular momentum - for (int m = 0; m < 2 * l + 1; m++) - { - const int lm = l * l + m; - std::complex* vkb_ptr = &vkb_out[ih * npw]; - const FPTYPE* ylm_ptr = &ylm_in[lm * npw]; - const FPTYPE* vq_ptr = &vq_in[nb * npw]; - // set vkb to zero - for (int ig = 0; ig < npw; ig++) - { - vkb_ptr[ig] = std::complex(0.0, 0.0); - } - // first term: ylm * vq * sk * pref - // loop over all G-vectors - if (ipol == jpol) - { - for (int ig = 0; ig < npw; ig++) - { - vkb_ptr[ig] -= ylm_ptr[ig] * vq_ptr[ig] * sk_in[ig] * pref_in[ih]; - } - } - // second term: ylm_deri * vq_deri * sk * pref - // loop over all G-vectors - const FPTYPE* ylm_deri_ptr1 = &ylm_deri_in[(ipol * x1 + lm) * npw]; - const FPTYPE* ylm_deri_ptr2 = &ylm_deri_in[(jpol * x1 + lm) * npw]; - const FPTYPE* vq_deri_ptr = &vq_deri_in[nb * npw]; - const FPTYPE* qnorm = &gk_in[4 * npw]; - for (int ig = 0; ig < npw; ig++) - { - vkb_ptr[ig] -= (gk_in[ig * 3 + ipol] * ylm_deri_ptr2[ig] + gk_in[ig * 3 + jpol] * ylm_deri_ptr1[ig]) - * vq_ptr[ig] * sk_in[ig] * pref_in[ih]; - } - // third term: ylm * vq_deri * sk * pref - // loop over all G-vectors - for (int ig = 0; ig < npw; ig++) - { - vkb_ptr[ig] -= 2.0 * ylm_ptr[ig] * vq_deri_ptr[ig] * sk_in[ig] * pref_in[ih] * gk_in[ig * 3 + ipol] - * gk_in[ig * 3 + jpol] * qnorm[ig]; - } - ih++; - } - } -} - -template -void Nonlocal_maths::prepare_vkb_ptr(int nbeta, - double* nhtol, - int nhtol_nc, - int npw, - int it, - std::complex* vkb_out, - std::complex** vkb_ptrs, - FPTYPE* ylm_in, - FPTYPE** ylm_ptrs, - FPTYPE* vq_in, - FPTYPE** vq_ptrs) -{ - // std::complex** vkb_ptrs[nh]; - // const FPTYPE** ylm_ptrs[nh]; - // const FPTYPE** vq_ptrs[nh]; - int ih = 0; - for (int nb = 0; nb < nbeta; nb++) - { - int l = nhtol[it * nhtol_nc + ih]; - for (int m = 0; m < 2 * l + 1; m++) - { - int lm = l * l + m; - vkb_ptrs[ih] = &vkb_out[ih * npw]; - ylm_ptrs[ih] = &ylm_in[lm * npw]; - vq_ptrs[ih] = &vq_in[nb * npw]; - ih++; - } - } -} - -template -void Nonlocal_maths::cal_dvkb_index(const int nbeta, - const double* nhtol, - const int nhtol_nc, - const int npw, - const int it, - const int ipol, - const int jpol, - int* indexes) -{ - int ih = 0; - const int x1 = (this->lmax_ + 1) * (this->lmax_ + 1); - for (int nb = 0; nb < nbeta; nb++) - { - int l = nhtol[it * nhtol_nc + ih]; - for (int m = 0; m < 2 * l + 1; m++) - { - //std::cout << "in function cal_dvkb_index, nhtol(" << it << ", " << ih << ") = " << l << std::endl; - int lm = l * l + m; - indexes[ih * 4] = lm; // the index of ylm matrix, for given l and m, together with ig to get value - indexes[ih * 4 + 1] = nb; // the iproj of present atom type - indexes[ih * 4 + 2] = (ipol * x1 + lm); - indexes[ih * 4 + 3] = (jpol * x1 + lm); - - ih++; - } - } -} - -template -void Nonlocal_maths::dylmr2(const int nylm, - const int ngy, - const FPTYPE* gk, - FPTYPE* dylm, - const int ipol) -{ - //----------------------------------------------------------------------- - // - // compute \partial Y_lm(G) \over \partial (G)_ipol - // using simple numerical derivation (SdG) - // The spherical harmonics are calculated in ylmr2 - // - // int nylm, ngy, ipol; - // number of spherical harmonics - // the number of g vectors to compute - // desired polarization - // FPTYPE g (3, ngy), gg (ngy), dylm (ngy, nylm) - // the coordinates of g vectors - // the moduli of g vectors - // the spherical harmonics derivatives - // - const FPTYPE delta = 1e-6; - const FPTYPE small = 1e-15; - - ModuleBase::matrix ylmaux; - // dg is the finite increment for numerical derivation: - // dg = delta |G| = delta * sqrt(gg) - // dgi= 1 /(delta * sqrt(gg)) - // gx = g +/- dg - - std::vector gx(ngy * 3); - - std::vector dg(ngy); - std::vector dgi(ngy); - - ylmaux.create(nylm, ngy); - - ModuleBase::GlobalFunc::ZEROS(dylm, nylm * ngy); - ylmaux.zero_out(); - -#ifdef _OPENMP -#pragma omp parallel for -#endif - for (int ig = 0; ig < 3 * ngy; ig++) - { - gx[ig] = gk[ig]; - } - //$OMP PARALLEL DO DEFAULT(SHARED) PRIVATE(ig) -#ifdef _OPENMP -#pragma omp parallel for -#endif - for (int ig = 0; ig < ngy; ig++) - { - const int igx = ig * 3, igy = ig * 3 + 1, igz = ig * 3 + 2; - FPTYPE norm2 = gx[igx] * gx[igx] + gx[igy] * gx[igy] + gx[igz] * gx[igz]; - dg[ig] = delta * sqrt(norm2); - if (dg[ig] > small) - { - dgi[ig] = 1.0 / dg[ig]; - } - else - { - dgi[ig] = 0.0; - } - } - //$OMP END PARALLEL DO - - //$OMP PARALLEL DO DEFAULT(SHARED) PRIVATE(ig) -#ifdef _OPENMP -#pragma omp parallel for -#endif - for (int ig = 0; ig < ngy; ig++) - { - const int index = ig * 3 + ipol; - gx[index] = gk[index] + dg[ig]; - } - //$OMP END PARALLEL DO - - base_device::DEVICE_CPU* cpu = {}; - ModuleBase::YlmReal::Ylm_Real(cpu, nylm, ngy, gx.data(), dylm); - //$OMP PARALLEL DO DEFAULT(SHARED) PRIVATE(ig) -#ifdef _OPENMP -#pragma omp parallel for -#endif - for (int ig = 0; ig < ngy; ig++) - { - const int index = ig * 3 + ipol; - gx[index] = gk[index] - dg[ig]; - } - //$OMP END PARALLEL DO - - ModuleBase::YlmReal::Ylm_Real(cpu, nylm, ngy, gx.data(), ylmaux.c); - - // zaxpy ( - 1.0, ylmaux, 1, dylm, 1); -#ifdef _OPENMP -#pragma omp parallel for collapse(2) -#endif - for (int lm = 0; lm < nylm; lm++) - { - for (int ig = 0; ig < ngy; ig++) - { - dylm[lm * ngy + ig] -= ylmaux(lm, ig); - dylm[lm * ngy + ig] *= 0.5 * dgi[ig]; - } - } - - return; -} - -template -FPTYPE Nonlocal_maths::Polynomial_Interpolation_nl(const ModuleBase::realArray& table, - const int& dim1, - const int& dim2, - const FPTYPE& table_interval, - const FPTYPE& x // input value -) -{ - - assert(table_interval > 0.0); - const FPTYPE position = x / table_interval; - const int iq = static_cast(position); - - const FPTYPE x0 = position - static_cast(iq); - const FPTYPE x1 = 1.0 - x0; - const FPTYPE x2 = 2.0 - x0; - const FPTYPE x3 = 3.0 - x0; - const FPTYPE y = (table(dim1, dim2, iq) * (-x2 * x3 - x1 * x3 - x1 * x2) / 6.0 - + table(dim1, dim2, iq + 1) * (+x2 * x3 - x0 * x3 - x0 * x2) / 2.0 - - table(dim1, dim2, iq + 2) * (+x1 * x3 - x0 * x3 - x0 * x1) / 2.0 - + table(dim1, dim2, iq + 3) * (+x1 * x2 - x0 * x2 - x0 * x1) / 6.0) - / table_interval; - - return y; -} -} // namespace hamilt - -#endif \ No newline at end of file diff --git a/source/source_pw/module_pwdft/nonlocal_workspace.cpp b/source/source_pw/module_pwdft/nonlocal_workspace.cpp new file mode 100644 index 00000000000..5530baef6a5 --- /dev/null +++ b/source/source_pw/module_pwdft/nonlocal_workspace.cpp @@ -0,0 +1,162 @@ +#include "source_pw/module_pwdft/nonlocal_workspace.h" + +#include "source_base/kernels/math_kernel_op.h" +#include "source_base/parallel_device.h" +#include "source_base/timer.h" +#include "source_pw/module_pwdft/kernels/nonlocal_op.h" + +#include +#include + +namespace hamilt +{ +namespace +{ +#ifdef __DSP +template +using Resize = base_device::memory::resize_memory_op_mt; +template +using Delete = base_device::memory::delete_memory_op_mt; +template +using Zero = base_device::memory::set_memory_op_mt; +template +using Gemm = ModuleBase::gemm_op_mt; +#else +template +using Resize = base_device::memory::resize_memory_op; +template +using Delete = base_device::memory::delete_memory_op; +template +using Zero = base_device::memory::set_memory_op; +template +using Gemm = ModuleBase::gemm_op; +#endif +} + +template +NonlocalWorkspace::~NonlocalWorkspace() +{ + Delete()(becp_); + Delete()(ps_); +} + +template +void NonlocalWorkspace::project(const T* vkb, const T* psi, const int npw, + const int ldv, const int ldp, const int nkb, const int bands) +{ + const T one(1, 0); + const T zero(0, 0); + if (npw == 0) + { + const size_t count = static_cast(nkb) * bands; + Zero()(becp_, 0, count); + } + else if (bands == 1) + { + ModuleBase::gemv_op()('C', npw, nkb, &one, vkb, ldv, psi, 1, &zero, becp_, 1); + } + else + { + Gemm()('C', 'N', nkb, bands, npw, &one, vkb, ldv, psi, ldp, &zero, becp_, nkb); + } +} + +template +void NonlocalWorkspace::contract(const UnitCell& cell, const pseudopot_cell_vnl& pp, + const int spin, const int npol, const int bands) +{ + using Real = typename GetTypeReal::type; + const size_t count = static_cast(pp.nkb) * bands; + Zero()(ps_, 0, count); + int sum = 0; + int atom = 0; + for (int type = 0; type < cell.ntype; ++type) + { + const int nproj = cell.atoms[type].ncpp.nh; + if (npol == 1) + { + nonlocal_pw_op()(nullptr, cell.atoms[type].na, bands, nproj, sum, atom, + spin, pp.nkb, pp.deeq.getBound2(), pp.deeq.getBound3(), pp.deeq.getBound4(), + pp.template get_deeq_data(), ps_, becp_); + } + else + { + nonlocal_pw_op()(nullptr, cell.atoms[type].na, bands, nproj, sum, atom, + pp.nkb, pp.deeq_nc.getBound2(), pp.deeq_nc.getBound3(), pp.deeq_nc.getBound4(), + pp.template get_deeq_nc_data(), ps_, becp_); + } + } +} + +template +void NonlocalWorkspace::back_project(const T* vkb, T* hpsi, const int npw, + const int ldv, const int ldp, const int nkb, const int bands) +{ + const T one(1, 0); + if (bands == 1) + { + ModuleBase::gemv_op()('N', npw, nkb, &one, vkb, ldv, ps_, 1, &one, hpsi, 1); + } + else + { + Gemm()('N', 'T', npw, bands, nkb, &one, vkb, ldv, ps_, bands, &one, hpsi, ldp); + } +} + +template +void NonlocalWorkspace::apply(const UnitCell& cell, const pseudopot_cell_vnl& pp, + const ModulePW::PW_Basis_K& basis, const int spin, + const int bands, const int nbasis, const int npol, const int npw, + const bool is_first_node, const T* vkb, const T* psi, T* hpsi) +{ + ModuleBase::timer::start("NonlocalWorkspace", "apply"); + if (is_first_node) + { + const size_t count = static_cast(nbasis) * bands / npol; + Zero()(hpsi, 0, count); + } + if (pp.nkb > 0 && bands > 0) + { + const size_t count = static_cast(pp.nkb) * bands; + if (count > becp_capacity_) + { + Resize()(becp_, count, "Nonlocal::becp"); + becp_capacity_ = count; + } + if (count > ps_capacity_) + { + Resize()(ps_, count, "Nonlocal::ps"); + ps_capacity_ = count; + } + this->project(vkb, psi, npw, pp.vkbnc, nbasis / npol, pp.nkb, bands); +#ifdef __MPI + if (basis.poolnproc > 1) + { + // The complex reduction wrapper doubles the MPI element count. + const size_t max_chunk = std::numeric_limits::max() / 2; + for (size_t offset = 0; offset < count;) + { + const size_t remaining = count - offset; + const int chunk_size = static_cast(std::min(max_chunk, remaining)); + T* chunk = becp_ + offset; + Parallel_Common::reduce_dev(chunk, chunk_size, basis.pool_world); + offset += chunk_size; + } + } +#endif + if (npw > 0) + { + this->contract(cell, pp, spin, npol, bands); + this->back_project(vkb, hpsi, npw, pp.vkbnc, nbasis / npol, pp.nkb, bands); + } + } + ModuleBase::timer::end("NonlocalWorkspace", "apply"); +} + +template class NonlocalWorkspace, base_device::DEVICE_CPU>; +template class NonlocalWorkspace, base_device::DEVICE_CPU>; +#if defined(__CUDA) || defined(__ROCM) +template class NonlocalWorkspace, base_device::DEVICE_GPU>; +template class NonlocalWorkspace, base_device::DEVICE_GPU>; +#endif +} // namespace hamilt diff --git a/source/source_pw/module_pwdft/nonlocal_workspace.h b/source/source_pw/module_pwdft/nonlocal_workspace.h new file mode 100644 index 00000000000..5a659645353 --- /dev/null +++ b/source/source_pw/module_pwdft/nonlocal_workspace.h @@ -0,0 +1,42 @@ +#ifndef PW_NONLOCAL_WORKSPACE_H +#define PW_NONLOCAL_WORKSPACE_H + +#include "source_pw/module_pwdft/vnl_pw.h" +#include + +namespace hamilt +{ +/** @brief Reusable device buffers for applying a nonlocal projector operator. */ +template +class NonlocalWorkspace +{ + private: + T* becp_ = nullptr; + T* ps_ = nullptr; + size_t becp_capacity_ = 0; + size_t ps_capacity_ = 0; + + void project(const T* vkb, const T* psi, const int npw, const int ldv, + const int ldp, const int nkb, const int bands); + void contract(const UnitCell& cell, const pseudopot_cell_vnl& pp, + const int spin, const int npol, const int bands); + void back_project(const T* vkb, T* hpsi, const int npw, const int ldv, + const int ldp, const int nkb, const int bands); + + public: + NonlocalWorkspace() = default; + ~NonlocalWorkspace(); + NonlocalWorkspace(const NonlocalWorkspace&) = delete; + NonlocalWorkspace& operator=(const NonlocalWorkspace&) = delete; + + /** @brief Apply beta D beta-dagger, preserving pool collectives on empty ranks. */ + void apply(const UnitCell& cell, const pseudopot_cell_vnl& pp, + const ModulePW::PW_Basis_K& basis, const int spin, + const int bands, const int nbasis, const int npol, const int npw, + const bool is_first_node, const T* vkb, const T* psi, T* hpsi); + + /** @brief Borrow the most recently computed projection coefficients. */ + T* get_becp() const { return becp_; } +}; +} // namespace hamilt +#endif diff --git a/source/source_pw/module_pwdft/onsite_proj_force_stress.cpp b/source/source_pw/module_pwdft/onsite_proj_force_stress.cpp index 024e077edb3..aaf871e0b2b 100644 --- a/source/source_pw/module_pwdft/onsite_proj_force_stress.cpp +++ b/source/source_pw/module_pwdft/onsite_proj_force_stress.cpp @@ -1,7 +1,7 @@ #include #include "source_io/module_parameter/parameter.h" -#include "source_pw/module_pwdft/onsite_proj.h" +#include "source_pw/module_proj/onsite_proj.h" #include "source_pw/module_pwdft/dftu_base.h" template diff --git a/source/source_pw/module_pwdft/onsite_proj_init.cpp b/source/source_pw/module_pwdft/onsite_proj_init.cpp deleted file mode 100644 index dcde05c7dcd..00000000000 --- a/source/source_pw/module_pwdft/onsite_proj_init.cpp +++ /dev/null @@ -1,442 +0,0 @@ -#include -#include -#include -#include -#include -#include - -#include "source_pw/module_pwdft/onsite_proj.h" -#include "source_pw/module_pwdft/onsite_proj_print.h" -#include "source_base/projgen.h" -#include "source_base/kernels/math_kernel_op.h" -#include "source_base/tool_quit.h" -#include "source_base/timer.h" -#include "source_io/module_parameter/parameter.h" - -#ifdef __MPI -#include "source_base/parallel_common.h" -#endif - -/** - * =============================================================================================== - * - * README - * - * =============================================================================================== - * - * This is a code demo for illustrating how to use unified radial projection in implementation of - * Operators involving local radial projectors on PW-expanded wavefunctions. - * - * Example usage: - * ```c++ - * // select the range of atoms that impose the operator in std::vector> it2ia like - * // it2ia[it] = {ia1, ia2, ...} for each type - * // if all atoms in present kind is "selected", just set it2ia[it].resize(na) and call - * // std::iota(it2ia[it].begin(), it2ia[it].end(), 0) - * - * std::vector> it2ia; // as if we have given its value... - * - * // you should have the `orbital_dir` as the directory containing the orbital files, then those - * // will be read by a static function `AtomicRadials::read_abacus_orb` to get the radial orbitals - * - * // call `init_proj` to initialize the radial projector, this function only needs to be called - * // once during the runtime. - * // its input... - * // the `nproj`, is for specifying number of projectors of each atom type, can be zero, - * // but cannot be the value larger than the number of zeta functions for the given angular momentum. - * // the `lproj` is the angular momentum of the projectors, and `iproj` is the index of zeta function - * // that each projector generated from. - * // the `lproj` along with `iproj` can enable radial projectors in any number developer wants. - * - * // the `onsite_r` is the onsite-radius for all valid projectors, it is used to generate the new - * // radial function that more localized than the original one, which is expected to have enhanced - * // projection efficiency. - * - * std::vector rgrid; - * std::vector> projs; - * std::vector> it2iproj; - * init_proj(orbital_dir, ucell, nproj, lproj, iproj, onsite_r, rgrid, projs, it2iproj); - * - * // then call the function `cal_becp` to calculate the becp. HOWEVER, there are quantities that - * // can be calculated in advance and reused in the following calculations. Please see the function - * // implementation, especially the comments about CACHE 0, CACHE 1, CACHE 2..., etc. - * - * // the input param of `cal_becp`... - * // the `it2ia` has been explained above - * // the `it2iproj` is the output of function `init_proj`, so you do not need to worry about it - * // the `rgrid` and `projs` are also the output of function `init_proj` - * // the `lproj` is the angular momentum for each projector, actually you have used it in `init_proj`, it - * // is the same as `lproj` - * // the `nq` is the number of G+k vectors, typically it is always GlobalV::NQX - * // the `dq` is the step size of G+k vectors, typically it is always GlobalV::DQ - * // the `ik` is the k-point index - * // the `pw_basis` is the plane wave basis, need ik - * // the `omega` is the cell volume - * // the `tpiba` is 2*pi/lat0 - * // the `sf` is the structure factor calculator - * // the `psi` is the wavefunction - * // the `becp` is the output of the function, it is the becp - * cal_becp(it2ia, it2iproj, rgrid, projs, lproj, nq, dq, ik, pw_basis, omega, tpiba, sf, psi, becp); - * - * // About parallelization, presently, the function `AtomicRadials::read_abacus_orb` is actually parallelized - * // by MPI, so after the reading of orbital, actually all processors have the same data. Therefore it is not - * // needed to call functions like `Parallel_Reduce` or `Parallel_Bcast` to synchronize the data. - * // However, what is strikingly memory-consuming is the table `tab_atomic_`. Performance optimization will - * // be needed if the memory is not enough. - */ - -template -void projectors::OnsiteProjector::init(const std::string& orbital_dir, - const UnitCell* ucell_in, - const psi::Psi, Device>& psi, - const K_Vectors& kv, - const ModulePW::PW_Basis_K& pw_basis, // level1: the plane wave basis, need ik - Structure_Factor& sf, // level2: the structure factor calculator - const double onsite_radius, - const int nq, - const double dq, - const ModuleBase::matrix& wg, - const ModuleBase::matrix& ekb) -{ - this->device = base_device::get_device_type(this->ctx); - - if(!this->initialed) - { - this->ucell = ucell_in; - this->ntype = ucell_in->ntype; - this->isk_ = kv.isk.data(); - - this->pw_basis_ = &pw_basis; - this->sf_ = &sf; - - std::vector orb_files(ntype); - std::vector nproj(ntype); - int sum_nproj = 0; - for(int it=0; itorbital_fn[it]; - nproj[it] = ucell->atoms[it].nwl; - sum_nproj += nproj[it]; - } - this->lproj.resize(sum_nproj); - int index = 0; - for(int it=0; itlproj[index++] = il; - } - } - std::vector iproj(sum_nproj, 0); - std::vector onsite_r(sum_nproj, onsite_radius); - - this->it2ia.resize(this->ntype); - this->iat_nh.resize(this->ucell->nat); - int iat = 0; - for(int it = 0; it < it2ia.size(); it++) - { - it2ia[it].resize(this->ucell->atoms[it].na); - std::iota(it2ia[it].begin(), it2ia[it].end(), 0); - for(int ia = 0; ia < it2ia[it].size(); ia++) - { - iat_nh[iat++] = nproj[it] * nproj[it]; - } - } - - this->init_proj(PARAM.inp.orbital_dir, - orb_files, - nproj, - lproj, - iproj, - onsite_r); - - ModuleBase::timer::start("OnsiteProj", "cubspl_tabulate"); - // STAGE 0 - making the interpolation table - // CACHE 0 - if cache the irow2it, irow2iproj, irow2m, itiaiprojm2irow, can be reused for - // SCF, RELAX and CELL-RELAX calculation - // [in] rgrid, projs, lproj, it2ia, it2iproj, nq, dq - RadialProjection::RadialProjector::_build_backward_map(it2iproj, lproj, irow2it_, irow2iproj_, irow2m_); - RadialProjection::RadialProjector::_build_forward_map(it2ia, it2iproj, lproj, itiaiprojm2irow_); - //rp_._build_sbt_tab(rgrid, projs, lproj, nq, dq); - rp_._build_sbt_tab(nproj, rgrid, projs, lproj, nq, dq, ucell_in->omega, psi.get_npol(), tab, nhtol); - // For being compatible with present cal_force and cal_stress framework - // uncomment the following code block if you want to use the Onsite_Proj_tools - if(this->tab_atomic_ == nullptr) - { - this->tot_nproj = itiaiprojm2irow_.size(); - this->npwx_ = this->pw_basis_->npwk_max; - this->size_vproj = this->tot_nproj * this->npwx_; - resmem_complex_op()(this->tab_atomic_, this->size_vproj, "OnsiteP::tab_atomic_"); - } - - delete this->fs_tools; // it is okay to delete nullptr - this->fs_tools = new hamilt::Onsite_Proj_tools( - nproj, lproj, tab, nhtol, this->tab_atomic_, ucell_in, &psi, &kv, &pw_basis, &sf, wg, ekb); - - ModuleBase::timer::end("OnsiteProj", "cubspl_tabulate"); - - this->initialed = true; - } -} - -template -void projectors::OnsiteProjector::init_proj(const std::string& orbital_dir, - const std::vector& orb_files, - const std::vector& nproj, // for each type, the number of projectors - const std::vector& lproj, // angular momentum of projectors within the type (l of zeta function) - const std::vector& iproj, // index of projectors within the type (izeta) - const std::vector& onsite_r) -{ - // extract the information from ucell - const int ntype = nproj.size(); - assert(ntype == orb_files.size()); - this->it2iproj.resize(ntype); - - int nproj_tot = 0; - nproj_tot = std::accumulate(nproj.begin(), nproj.end(), nproj_tot, std::plus()); - assert(nproj_tot == lproj.size()); - assert(nproj_tot == iproj.size()); - assert(nproj_tot == onsite_r.size()); - this->projs.resize(nproj_tot); - - int idx = 0; - int nr = -1; - double dr = -1.0; - for(int it = 0; it < ntype; ++it) - { - const int nproj_it = nproj[it]; - this->it2iproj[it].resize(nproj_it); - print::print_proj_status(it, nproj_it); - if(nproj_it == 0) - { - continue; - } - std::ifstream ifs(orbital_dir + orb_files[it]); - std::string elem = ""; - double ecut = -1.0; - int nr_ = -1; - double dr_ = -1.0; - std::vector nzeta; // number of radials for each l - std::vector> radials; // radials arranged in serial - this->read_abacus_orb(ifs, elem, ecut, nr_, dr_, nzeta, radials); -#ifdef __DEBUG - assert(elem != ""); - assert(ecut != -1.0); - assert(nr_ != -1); - assert(dr_ != -1.0); -#endif - nr = std::max(nr, nr_); // the maximal nr - assert(dr == -1.0 || dr == dr_); // the dr should be the same for all types - dr = (dr == -1.0) ? dr_ : dr; - for(int ip = 0; ip < nproj_it; ++ip) - { - int l = lproj[idx]; - int izeta = iproj[idx]; - int irad = 0; - irad = std::accumulate(nzeta.begin(), nzeta.begin() + l, irad); - irad += izeta; - std::vector temp = radials[irad]; - rgrid.resize(nr); - std::iota(rgrid.begin(), rgrid.end(), 0); - std::for_each(rgrid.begin(), rgrid.end(), [dr](double& r_i) { r_i *= dr; }); - smoothgen(nr, rgrid.data(), temp.data(), onsite_r[idx], projs[idx]); - it2iproj[it][ip] = idx; - ++idx; - } - } - // do zero padding - if(nr != -1) - { - std::for_each(projs.begin(), projs.end(), [nr](std::vector& proj) { proj.resize(nr, 0.0); }); - } - // generate the rgrid - this->rgrid.resize(nr); - std::iota(rgrid.begin(), rgrid.end(), 0); - std::for_each(rgrid.begin(), rgrid.end(), [dr](double& r_i) { r_i *= dr; }); -} - -template -void projectors::OnsiteProjector::read_abacus_orb(std::ifstream& ifs, - std::string& elem, - double& ecut, - int& nr, - double& dr, - std::vector& nzeta, - std::vector>& radials, - const int rank) -{ - nr = 0; // number of grid points - dr = 0; // grid spacing - int lmax = 0, nchi = 0; // number of radial functions - std::vector> radial_map_; // build a map from [l][izeta] to 1-d array index - std::string tmp; - // first read the header - if (rank == 0) - { - if (!ifs.is_open()) - { - ModuleBase::WARNING_QUIT("AtomicRadials::read_abacus_orb", "Couldn't open orbital file."); - } - while (ifs >> tmp) - { - if (tmp == "Element") - { - ifs >> elem; - } - else if (tmp == "Cutoff(Ry)") - { - ifs >> ecut; - } - else if (tmp == "Lmax") - { - ifs >> lmax; - nzeta.resize(lmax + 1); - for (int l = 0; l <= lmax; ++l) - { - ifs >> tmp >> tmp >> tmp >> nzeta[l]; - } - } - else if (tmp == "Mesh") - { - ifs >> nr; - continue; - } - else if (tmp == "dr") - { - ifs >> dr; - break; - } - } - radial_map_.resize(lmax + 1); - for (int l = 0; l <= lmax; ++l) - { - radial_map_[l].resize(nzeta[l]); - } - int ichi = 0; - for (int l = 0; l <= lmax; ++l) - { - for (int iz = 0; iz < nzeta[l]; ++iz) - { - radial_map_[l][iz] = ichi++; // return the value of ichi, then increment - } - } - nchi = ichi; // total number of radial functions - radials.resize(nchi); - std::for_each(radials.begin(), radials.end(), [nr](std::vector& v) { v.resize(nr); }); - } - - // broadcast the header information -#ifdef __MPI - Parallel_Common::bcast_string(elem); - Parallel_Common::bcast_double(ecut); - Parallel_Common::bcast_int(lmax); - Parallel_Common::bcast_int(nchi); - Parallel_Common::bcast_int(nr); - Parallel_Common::bcast_double(dr); -#endif - - // then adjust the size of the vectors - if (rank != 0) - { - nzeta.resize(lmax + 1); - radials.resize(nchi); - std::for_each(radials.begin(), radials.end(), [nr](std::vector& v) { v.resize(nr); }); - } - // broadcast the number of zeta functions for each angular momentum -#ifdef __MPI - Parallel_Common::bcast_int(nzeta.data(), lmax + 1); -#endif - - // read the radial functions by rank0 - int ichi = 0; - for (int i = 0; i != nchi; ++i) - { - if (rank == 0) - { - int l = 0, izeta = 0; - ifs >> tmp >> tmp >> tmp; - ifs >> tmp >> l >> izeta; - ichi = radial_map_[l][izeta]; - for (int ir = 0; ir != nr; ++ir) - { - ifs >> radials[ichi][ir]; - } - } - // broadcast the radial functions -#ifdef __MPI - Parallel_Common::bcast_int(ichi); // let other ranks know where to store the radial function - Parallel_Common::bcast_double(radials[ichi].data(), nr); -#endif - } -} // end of read_abacus_orb - -// explicit method instantiation -template -void projectors::OnsiteProjector::init( - const std::string&, - const UnitCell*, - const psi::Psi, base_device::DEVICE_CPU>&, - const K_Vectors&, - const ModulePW::PW_Basis_K&, - Structure_Factor&, - const double, - const int, - const double, - const ModuleBase::matrix&, - const ModuleBase::matrix&); - -template -void projectors::OnsiteProjector::init_proj( - const std::string&, - const std::vector&, - const std::vector&, - const std::vector&, - const std::vector&, - const std::vector&); - -template -void projectors::OnsiteProjector::read_abacus_orb( - std::ifstream&, - std::string&, - double&, - int&, - double&, - std::vector&, - std::vector>&, - const int); - -#if ((defined __CUDA) || (defined __ROCM)) -template -void projectors::OnsiteProjector::init( - const std::string&, - const UnitCell*, - const psi::Psi, base_device::DEVICE_GPU>&, - const K_Vectors&, - const ModulePW::PW_Basis_K&, - Structure_Factor&, - const double, - const int, - const double, - const ModuleBase::matrix&, - const ModuleBase::matrix&); - -template -void projectors::OnsiteProjector::init_proj( - const std::string&, - const std::vector&, - const std::vector&, - const std::vector&, - const std::vector&, - const std::vector&); - -template -void projectors::OnsiteProjector::read_abacus_orb( - std::ifstream&, - std::string&, - double&, - int&, - double&, - std::vector&, - std::vector>&, - const int); -#endif diff --git a/source/source_pw/module_pwdft/onsite_proj_overlap.cpp b/source/source_pw/module_pwdft/onsite_proj_overlap.cpp deleted file mode 100644 index 989a6092148..00000000000 --- a/source/source_pw/module_pwdft/onsite_proj_overlap.cpp +++ /dev/null @@ -1,231 +0,0 @@ -#include -#include - -#include "source_pw/module_pwdft/onsite_proj.h" -#include "source_pw/module_pwdft/onsite_proj_print.h" -#include "source_cell/cell_tools.h" -#include "source_base/kernels/math_kernel_op.h" -#include "source_base/parallel_reduce.h" -#include "source_base/timer.h" -#include "source_io/module_parameter/parameter.h" - -template -void projectors::OnsiteProjector::tabulate_atomic(const int ik, const char grad) -{ - ModuleBase::timer::start("OnsiteProj", "tabulate_atomic"); - // assert(grad == 'n' || grad == 'x' || grad == 'y' || grad == 'z'); - // grad = 'n' means no gradient, grad = 'x' means gradient along x, etc. - - // STAGE 1 - calculate the for the given G+k vector - // CACHE 1 - if cache the tab_, can be reused for SCF and RELAX calculation - // [in] pw_basis, ik, omega, tpiba, irow2it - this->ik_ = ik; - this->npw_ = pw_basis_->npwk[ik]; - this->npwx_ = pw_basis_->npwk_max; - // std::vector> q(this->npw_); - // for(int ig = 0; ig < this->npw_; ++ig) - // { - // q[ig] = pw_basis_->getgpluskcar(ik, ig); // get the G+k vector, G+k will change during CELL-RELAX - // } - // const int nrow = irow2it_.size(); - // std::vector> tab_(nrow*this->npw_); - // // convention used here: 'l': , 'r': - // // denote q=G+k, = exp(iqr), the routine Fourier Transform written as F(q) = - // // what is calculated is here - // rp_.sbtft(q, tab_, 'l', this->ucell->omega, this->ucell->tpiba); - - // STAGE 2 - make_atomic: multiply e^iqtau and extend the to for each atom - // CACHE 2 - if cache the tab_atomic_, can be reused for SCF calculation - // [in] it2ia, itiaiprojm2irow, tab_, npw, sf - // for(int irow = 0; irow < nrow; ++irow) - // { - // const int it = irow2it_[irow]; - // const int iproj = irow2iproj_[irow]; - // const int m = irow2m_[irow]; - // for(int ia = 0; ia < na[it]; ++ia) - // { - // // why Structure_Factor needs the FULL pw_basis??? - // std::complex* sk = this->sf_->get_sk(ik, it, ia, pw_basis_); // exp(-iqtau) - // // Note: idea on extending the param list of get_sk - // // the get_sk should have an extra param 'grad' to calculate the gradient of S(q), which - // // is actually very simple to be - // // d(S(q))/dq = -i S(q) * tau, for one direction it is just -i S(q) * tau_x (if x is the direction) - // const int irow_out = itiaiprojm2irow_.at(std::make_tuple(it, ia, iproj, m)); - // for(int ig = 0; ig < this->npw_; ++ig) - // { - // std::complex deriv = (grad == 'n')? 1.0: ModuleBase::NEG_IMAG_UNIT; // because sk is exp(-iqtau) - // deriv = (grad == 'n')? 1.0: (grad == 'x')? deriv * q[ig].x: (grad == 'y')? deriv * q[ig].y: deriv * q[ig].z; - // // there must be something twisted in ABACUS - // // because the tab_ is , but the sk is exp(-iqtau). How can it get the - // // correct result? - // this->tab_atomic_[irow_out*this->npw_ + ig] = sk[ig] * tab_[irow*this->npw_ + ig] * deriv; - // } - // delete[] sk; - // } - // } - // q.clear(); - // q.shrink_to_fit(); // release memory - // tab_.clear(); - // tab_.shrink_to_fit(); // release memory - ModuleBase::timer::end("OnsiteProj", "tabulate_atomic"); -} - -template -void projectors::OnsiteProjector::overlap_proj_psi( - const int npm, - const std::complex* ppsi, - const int ld_psi) -{ - ModuleBase::timer::start("OnsiteProj", "overlap"); - // STAGE 3 - cal_becp - // CACHE 3 - it is no use to cache becp, it will change in each SCF iteration - // [in] psi, tab_atomic_, npw, becp, ik -// const char transa = 'C'; -// const char transb = 'N'; -// const int ldb = this->npwx_; -// const int ldc = this->tot_nproj; -// const std::complex alpha = 1.0; -// const std::complex beta = 0.0; -// if(this->becp == nullptr || this->size_becp < npm*ldc) -// { -// delete[] this->becp; -// this->becp = new std::complex[npm*ldc]; -// this->size_becp = npm*ldc; -// } -// setmem_complex_op()(ctx, this->becp, 0.0, this->size_becp); -// gemm_op()( -// this->ctx, -// transa, // const char transa -// transb, // const char transb -// ldc, // const int m -// npm, // const int n -// this->npw_, // const int k -// &alpha, // const std::complex alpha -// this->tab_atomic_, // const std::complex* a -// this->npw_, // const int lda -// ppsi, // const std::complex* b -// ldb, // const int ldb -// &beta, // const std::complex beta -// becp, // std::complex* c -// ldc); // const int ldc -// #ifdef __MPI -// Parallel_Reduce::reduce_pool(becp, size_becp); -// #endif - - // notes on refactor for DCU calculation - // the npm here is nbands(occ) * npol, for calling cal_becp, the npol should be divided. - // std::cout << "npm: " << npm << std::endl; - // std::cout << "at " << __FILE__ << ": " << __LINE__ << " output tot_nproj: " << this->tot_nproj << std::endl; - // std::cout << "at " << __FILE__ << ": " << __LINE__ << " output npm: " << npm << std::endl; - // std::cout << "at " << __FILE__ << ": " << __LINE__ << " ik_: " << ik_ << std::endl; - int npol = this->ucell->get_npol(); - if(this->becp == nullptr || this->size_becp < npm*this->tot_nproj) - { - this->size_becp = npm*this->tot_nproj; - resmem_complex_op()(this->becp, this->size_becp); - if(this->device == base_device::GpuDevice ) - { - resmem_complex_h_op()(this->h_becp, this->size_becp); - } - else - { - this->h_becp = this->becp; - } - } - this->fs_tools->cal_becp(ik_, npm/npol, this->becp, ppsi, ld_psi > 0 ? ld_psi : this->npwx_); // in cal_becp, npm should be the one not multiplied by npol - if(this->device == base_device::GpuDevice) - { - syncmem_complex_d2h_op()(h_becp, this->becp, this->size_becp); - } - ModuleBase::timer::end("OnsiteProj", "overlap"); -} - -template -void projectors::OnsiteProjector::cal_occupations( - const psi::Psi, Device>* psi_in, - const ModuleBase::matrix& wg_in) -{ - ModuleBase::timer::start("OnsiteProj", "cal_occupation"); - this->tabulate_atomic(0); - std::vector> occs(this->tot_nproj * 4, 0.0); - - // loop over k-points to calculate Mi of \sum_{k,i,l,m} - const int nbands = psi_in->get_nbands(); - for(int ik = 0; ik < psi_in->get_nk(); ik++) - { - psi_in->fix_k(ik); - if(ik != 0) - { - this->tabulate_atomic(ik); - } - // std::cout << __FILE__ << ":" << __LINE__ << " nbands = " << nbands << std::endl; - this->overlap_proj_psi( - nbands * psi_in->get_npol(), - psi_in->get_pointer()); - const std::complex* becp_p = this->get_h_becp(); - // becp(nbands*npol , nkb) - // mag = wg * \sum_{nh}becp * becp - int nkb = this->tot_nproj; - //nkb = 18; - //std::cout << "at " << __FILE__ << ": " << __LINE__ << " output nbands: " << nbands << std::endl; - //std::cout << "at " << __FILE__ << ": " << __LINE__ << " output nkb: " << nkb << std::endl; - for(int ib = 0;ibiat_nh.size(); iat++) - { - const int nh = this->get_nh(iat); - for(int ih = 0; ih < nh; ih++) - { - const int occ_index = (begin_ih + ih) * 4; - const int index = ib*2*nkb + begin_ih + ih; - occs[occ_index] += weight * conj(becp_p[index]) * becp_p[index]; - occs[occ_index + 1] += weight * conj(becp_p[index]) * becp_p[index + nkb]; - occs[occ_index + 2] += weight * conj(becp_p[index + nkb]) * becp_p[index]; - occs[occ_index + 3] += weight * conj(becp_p[index + nkb]) * becp_p[index + nkb]; - } - begin_ih += nh; - } - } - } - // reduce mag from all k-pools - const int npool = GlobalV::KPAR * PARAM.inp.bndpar; - Parallel_Reduce::reduce_double_allpool(npool, GlobalV::NPROC_IN_POOL, (double*)(&(occs[0])), occs.size()*2); - // occ has been reduced and calculate mag - // Print orbital charge analysis - auto atom_labels = unitcell::get_atomLabels(this->ucell->atoms, this->ucell->ntype); - print::print_orb_chg(this->ucell, occs, this->iat_nh, atom_labels); - - // print charge - ModuleBase::timer::end("OnsiteProj", "cal_occupation"); -} - -// explicit method instantiation -template -void projectors::OnsiteProjector::tabulate_atomic( - const int, const char); - -template -void projectors::OnsiteProjector::overlap_proj_psi( - const int, const std::complex*, const int); - -template -void projectors::OnsiteProjector::cal_occupations( - const psi::Psi, base_device::DEVICE_CPU>*, - const ModuleBase::matrix&); - -#if ((defined __CUDA) || (defined __ROCM)) -template -void projectors::OnsiteProjector::tabulate_atomic( - const int, const char); - -template -void projectors::OnsiteProjector::overlap_proj_psi( - const int, const std::complex*, const int); - -template -void projectors::OnsiteProjector::cal_occupations( - const psi::Psi, base_device::DEVICE_GPU>*, - const ModuleBase::matrix&); -#endif diff --git a/source/source_pw/module_pwdft/op_pw_ekin_td.cpp b/source/source_pw/module_pwdft/op_pw_ekin_td.cpp new file mode 100644 index 00000000000..f604ea3d9c0 --- /dev/null +++ b/source/source_pw/module_pwdft/op_pw_ekin_td.cpp @@ -0,0 +1,87 @@ +#include "source_pw/module_pwdft/op_pw_ekin_td.h" + +#include "source_base/module_device/memory_op.h" +#include "source_base/timer.h" +#include "source_basis/module_pw/pw_momentum.h" +#include "source_pw/module_pwdft/kernels/ekinetic_op.h" + +#include +#include + +namespace hamilt +{ + +template +TDEkineticPW::TDEkineticPW(const ModulePW::PW_Basis_K* wfc_basis_in) : wfc_basis(wfc_basis_in) +{ + this->classname = "TDEkineticPW"; + this->cal_type = calculation_type::pw_ekinetic_td; +} + +template +TDEkineticPW::~TDEkineticPW() = default; + +template +void TDEkineticPW::init(const int ik) +{ + ModuleBase::timer::start("TDEkineticPW", "init"); + this->ik = ik; + const int npw = this->wfc_basis->npwk[ik]; + using CtDevice = typename ct::PsiToContainer::type; + if (this->factor_.NumElements() < std::max(1, npw) || this->factor_.data_type() != ct::DataTypeToEnum::value + || this->factor_.device_type() != ct::DeviceTypeToEnum::value) + { + this->factor_ = ct::Tensor(ct::DataTypeToEnum::value, ct::DeviceTypeToEnum::value, {std::max(1, npw)}); + } + std::vector factor(npw); + for (int ig = 0; ig < npw; ++ig) + { + const double shift = ModulePW::kinetic_shift(*this->wfc_basis, ik, ig, A_ha_); + factor[ig] = static_cast(shift); + } + if (npw > 0) + { + base_device::memory::synchronize_memory_op()(this->factor_.template data(), + factor.data(), + npw); + } + if (this->next_op != nullptr) + { + this->next_op->init(ik); + } + ModuleBase::timer::end("TDEkineticPW", "init"); +} + +template +void TDEkineticPW::act(const int nbands, + const int nbasis, + const int npol, + const T* tmpsi_in, + T* tmhpsi, + const int ngk_ik, + const bool is_first_node) const +{ + ModuleBase::timer::start("TDEkineticPW", "act"); + // Reuse the CPU/CUDA/HIP diagonal kinetic kernel for 2*A*p + A^2. + if (nbands > 0) + { + ekinetic_pw_op()(this->ctx, + nbands, + ngk_ik, + nbasis / npol, + is_first_node, + Real(1), + this->factor_.template data(), + tmhpsi, + tmpsi_in); + } + ModuleBase::timer::end("TDEkineticPW", "act"); +} + +template class TDEkineticPW, base_device::DEVICE_CPU>; +template class TDEkineticPW, base_device::DEVICE_CPU>; +#if defined(__CUDA) || defined(__ROCM) +template class TDEkineticPW, base_device::DEVICE_GPU>; +template class TDEkineticPW, base_device::DEVICE_GPU>; +#endif +} // namespace hamilt diff --git a/source/source_pw/module_pwdft/op_pw_ekin_td.h b/source/source_pw/module_pwdft/op_pw_ekin_td.h new file mode 100644 index 00000000000..a02663a5011 --- /dev/null +++ b/source/source_pw/module_pwdft/op_pw_ekin_td.h @@ -0,0 +1,43 @@ +#ifndef OP_PW_EKIN_TD_H +#define OP_PW_EKIN_TD_H + +#include "op_pw.h" +#include "source_base/module_container/ATen/core/tensor.h" +#include "source_basis/module_pw/pw_basis_k.h" + +namespace hamilt +{ + +template +class TDEkineticPW : public OperatorPW +{ + private: + using Real = typename GetTypeReal::type; + + public: + /** @brief Bind the vector potential in Hartree atomic units before init(). */ + void set_A_ha(const ModuleBase::Vector3& A) { A_ha_ = A; } + TDEkineticPW(const ModulePW::PW_Basis_K* wfc_basis_in); + + virtual ~TDEkineticPW(); + + void init(const int ik) override; + + virtual void act(const int nbands, + const int nbasis, + const int npol, + const T* tmpsi_in, + T* tmhpsi, + const int ngk_ik = 0, + const bool is_first_node = false) const override; + + private: + const ModulePW::PW_Basis_K* wfc_basis = nullptr; + ModuleBase::Vector3 A_ha_; + Device* ctx = {}; + ct::Tensor factor_; +}; + +} // namespace hamilt + +#endif // OP_PW_EKIN_TD_H diff --git a/source/source_pw/module_pwdft/op_pw_exx.cpp b/source/source_pw/module_pwdft/op_pw_exx.cpp index 94f19ce3dec..a6057a1e2cd 100644 --- a/source/source_pw/module_pwdft/op_pw_exx.cpp +++ b/source/source_pw/module_pwdft/op_pw_exx.cpp @@ -1,7 +1,6 @@ #include "op_pw_exx.h" #include "source_base/constants.h" -#include "source_base/global_variable.h" #include "source_base/parallel_common.h" #include "source_base/parallel_device.h" #include "source_base/parallel_comm.h" // use KP_WORLD @@ -11,13 +10,15 @@ #include "source_base/tool_quit.h" #include "source_cell/klist.h" #include "source_hamilt/operator.h" +#include "source_hamilt/module_xc/general_exx_info.h" #include "source_psi/psi.h" #include "source_pw/module_pwdft/kernels/cal_density_real_op.h" +#include "source_pw/module_pwdft/kernels/exx_batch_op.h" #include "source_pw/module_pwdft/kernels/exx_cal_energy_op.h" #include "source_pw/module_pwdft/kernels/mul_potential_op.h" #include "source_pw/module_pwdft/kernels/vec_mul_cx_op.h" -#include "source_io/module_parameter/parameter.h" +#include #include #include #include @@ -37,19 +38,22 @@ OperatorEXXPW::OperatorEXXPW(const int* isk_in, const ModulePW::PW_Basis* rhopw_in, K_Vectors *kv_in, const UnitCell *ucell, - const bool separate_loop_in, - const Real hybrid_alpha_in, - const CoulombParam& coulomb_param_in) + const General_Exx_Info& exx_info, + const int nspin_in, + const int kpar_in, + const int my_rank_in, + const int my_pool_in) : isk(isk_in), wfcpw(wfcpw_in), rhopw(rhopw_in), kv(kv_in), ucell(ucell), - separate_loop(separate_loop_in), hybrid_alpha(hybrid_alpha_in), - coulomb_param(coulomb_param_in) + separate_loop(exx_info.separate_loop), hybrid_alpha(exx_info.hybrid_alpha), + coulomb_param(exx_info.coulomb_param), nspin_(nspin_in), ecut_exx_(exx_info.ecut_exx), + ecutexx_user_set_(exx_info.ecutexx_user_set), exx_batch_size_(exx_info.exx_batch_size), + exxace_(exx_info.exxace), my_rank_(my_rank_in), my_pool_(my_pool_in) { - if (GlobalV::KPAR != 1 && PARAM.inp.exxace == false) + if (kpar_in != 1 && !exxace_) { - // GlobalV::ofs_running << "EXX Calculation does not support k-point parallelism" << std::endl; ModuleBase::WARNING_QUIT("OperatorEXXPW", "EXX Calculation does not support k-point parallelism when exxace is set to false"); } - gamma_extrapolation = PARAM.inp.exx_gamma_extrapolation; + gamma_extrapolation = exx_info.gamma_extrapolation; bool is_mp = kv_in->get_is_mp(); #ifdef __MPI Parallel_Common::bcast_bool(is_mp); @@ -77,28 +81,22 @@ OperatorEXXPW::OperatorEXXPW(const int* isk_in, // resmem_complex_op()(this->ctx, psi_all_real, wfcpw->nrxx * GlobalV::NBANDS); int nks = wfcpw->nks; - int nk_fac = PARAM.inp.nspin == 2 ? 2 : 1; + int nk_fac = nspin_ == 2 ? 2 : 1; resmem_real_op()(pot, rhopw->npw); tpiba = ucell->tpiba; Real tpiba2 = tpiba * tpiba; - // initialize rhopw_dev - double ecut_exx = PARAM.inp.ecutexx; - if (ecut_exx == 0.0) - { - ecut_exx = PARAM.inp.ecutrho; - } - + // initialize rhopw_dev on the resolved EXX cutoff rhopw_dev = new ModulePW::PW_Basis(wfcpw->get_device(), rhopw->get_precision()); rhopw_dev->fft_bundle.setfft(wfcpw->get_device(), rhopw->get_precision()); #ifdef __MPI rhopw_dev->initmpi(rhopw->poolnproc, rhopw->poolrank, rhopw->pool_world); #endif // here we can actually use different ecut to init the grids - rhopw_dev->initgrids(rhopw->lat0, rhopw->latvec, ecut_exx); + rhopw_dev->initgrids(rhopw->lat0, rhopw->latvec, ecut_exx_); rhopw_dev->initgrids(rhopw->lat0, rhopw->latvec, rhopw->nx, rhopw->ny, rhopw->nz); - rhopw_dev->initparameters(rhopw->gamma_only, ecut_exx, rhopw->distribution_type, rhopw->xprime); + rhopw_dev->initparameters(rhopw->gamma_only, ecut_exx_, rhopw->distribution_type, rhopw->xprime); rhopw_dev->setuptransform(); rhopw_dev->collect_local_pw(); @@ -135,6 +133,33 @@ OperatorEXXPW::~OperatorEXXPW() // use delete_memory_op to delete the allocated pws delmem_complex_op()(psi_nk_real); delmem_complex_op()(psi_mq_real); + if (psi_nk_real_cache != nullptr) + { + delmem_complex_op()(psi_nk_real_cache); + } + if (dens_box_batch != nullptr) + { + delmem_complex_op()(dens_box_batch); + delmem_complex_op()(dens_pw_batch); + exx_batch_fft_plan_destroy(&exx_fft_plan); + exx_batch_fft_plan_destroy(&exx_fft_plan1); + } + if (sg_map_rho != nullptr) + { + delmem_int_op()(sg_map_rho); + } + if (sg_map_wfc != nullptr) + { + delmem_int_op()(sg_map_wfc); + } + if (full_map_rho != nullptr) + { + delmem_int_op()(full_map_rho); + } + if (full_map_wfc != nullptr) + { + delmem_int_op()(full_map_wfc); + } delmem_complex_op()(density_real); delmem_complex_op()(h_psi_real); delmem_complex_op()(density_recip); @@ -190,7 +215,7 @@ void OperatorEXXPW::act(const int nbands, setmem_complex_op()(tmhpsi, 0, nbasis*nbands/npol); } - if (PARAM.inp.exxace && this->separate_loop) + if (exxace_ && this->separate_loop) { act_op_ace(nbands, nbasis, npol, tmpsi_in, tmhpsi, ngk_ik, is_first_node); } @@ -219,77 +244,35 @@ void OperatorEXXPW::act_op(const int nbands, setmem_complex_op()(psi_mq_real, 0, wfcpw->nrxx); auto q_points = get_q_points(this->ik); - // std::cout << "kpoint " << this->ik << ", qpoints: "; - // for (auto iq: q_points) - // std::cout << iq << ", "; - // std::cout << std::endl; - int nk_fac = PARAM.inp.nspin == 2 ? 2 : 1; + int nk_fac = nspin_ == 2 ? 2 : 1; int nk = wfcpw->nks / nk_fac; + const Real nqs = q_points.size(); - // ik fixed here, select band n - for (int n_iband = 0; n_iband < nbands; n_iband++) - { - const T *psi_nk = tmpsi_in + n_iband * nbasis; - // retrieve \psi_nk in real space - wfcpw->recip_to_real(ctx, psi_nk, psi_nk_real, this->ik); - - // for \psi_nk, get the pw of iq and band m + maybe_setup_exx_grid(); + // psi_nk in real space for all bands once per ik, reused over (iq, m) + cache_psi_nk_real(nbands, nbasis, tmpsi_in, this->ik); - Real nqs = q_points.size(); - for (int iq: q_points) + for (int iq: q_points) + { + get_exx_potential(kv, wfcpw, rhopw_dev, pot, tpiba, gamma_extrapolation, ucell->omega, this->ik, iq % nk, false, this->coulomb_param); + for (int m_iband = 0; m_iband < psi.get_nbands(); m_iband++) { - get_exx_potential(kv, wfcpw, rhopw_dev, pot, tpiba, gamma_extrapolation, ucell->omega, this->ik, iq % nk, false, this->coulomb_param); - for (int m_iband = 0; m_iband < psi.get_nbands(); m_iband++) + // occupation of the source state (m, iq), not of the target k-point + double wg_mqb_real = (*wg)(iq, m_iband); + if (wg_mqb_real < 1e-12) { - double wg_mqb_real = (*wg)(this->ik, m_iband); - T wg_mqb = wg_mqb_real; - if (wg_mqb_real < 1e-12) - { - continue; - } - - const T* psi_mq = get_pw(m_iband, iq); - wfcpw->recip_to_real(ctx, psi_mq, psi_mq_real, iq); - - // direct multiplication in real space, \psi_nk(r) * \psi_mq(r) - cal_density_recip(psi_nk_real, psi_mq_real, ucell->omega); - - // multiply the density with the potential in recip space - multiply_potential(density_recip, this->ik, iq); - - // bring the potential back to real space - rho_recip2real(density_recip, density_real); - - if (false) - { - // do nothing - } - else - { - vec_mul_vec_complex_op()(density_real, psi_mq_real, density_real, wfcpw->nrxx); - } - - T wk_iq = kv->wk[iq]; + continue; + } - T tmp_scalar = wg_mqb / wk_iq / nqs; - axpy_complex_op()(wfcpw->nrxx, - &tmp_scalar, - density_real, - 1, - h_psi_real, - 1); + wfc_to_real_exx(get_pw(m_iband, iq), iq, nbasis); - } // end of m_iband - setmem_complex_op()(density_real, 0, rhopw_dev->nrxx); - setmem_complex_op()(density_recip, 0, rhopw_dev->npw); - setmem_complex_op()(psi_mq_real, 0, wfcpw->nrxx); + // full accumulation weight, hybrid_alpha included + const Real factor = this->hybrid_alpha * wg_mqb_real / kv->wk[iq] / nqs; + apply_fock_all_bands(nbands, nbasis, iq, factor, tmhpsi); - } // end of iq - T* h_psi_nk = tmhpsi + n_iband * nbasis; - wfcpw->real_to_recip(ctx, h_psi_real, h_psi_nk, this->ik, true, this->hybrid_alpha); - setmem_complex_op()(h_psi_real, 0, rhopw_dev->nrxx); + } // end of m_iband - } + } // end of iq ModuleBase::timer::end("OperatorEXXPW", "act_op"); @@ -297,12 +280,11 @@ void OperatorEXXPW::act_op(const int nbands, template void OperatorEXXPW::act_op_kpar(const int nbands, - const int nbasis, - const int npol, - const T *tmpsi_in, - T *tmhpsi, - const int ngk_ik, - const bool is_first_node) const + const int nbasis, + const T* tmpsi_in, + T* tmhpsi, + const int ispin, + const bool has_kpoint) const { ModuleBase::timer::start("OperatorEXXPW", "act_op_kpar"); @@ -315,8 +297,15 @@ void OperatorEXXPW::act_op_kpar(const int nbands, setmem_complex_op()(psi_nk_real, 0, wfcpw->nrxx); setmem_complex_op()(psi_mq_real, 0, wfcpw->nrxx); int nqs = kv->get_nkstot_nospin(); - int nspin_fac = PARAM.inp.nspin == 2 ? 2 : 1; - int ispin = this->ik < (wfcpw->nks / nspin_fac) ? 0 : 1; + int nspin_fac = nspin_ == 2 ? 2 : 1; + + maybe_setup_exx_grid(); + // psi_nk in real space for all bands once per ik, reused over (iq, m); + // the MPI communication order below is unchanged + if (has_kpoint) + { + cache_psi_nk_real(nbands, nbasis, tmpsi_in, this->ik); + } // ik fixed here, select band n for (int iq = 0; iq < nqs; iq++) @@ -333,75 +322,543 @@ void OperatorEXXPW::act_op_kpar(const int nbands, iq_loc_spin += wfcpw->nks / nspin_fac; } - for (int m_iband = 0; m_iband < psi.get_nbands(); m_iband++) + // occupation row and k weight of the source k-point, fetched from the + // pool that owns it in a single broadcast + const int nb = psi.get_nbands(); + const int occupation_count = nb + 1; + std::vector occ_q(occupation_count); + if (iq_pool == my_pool_) { - double wg_mqb = 0; - if (iq_pool == GlobalV::MY_POOL) + for (int m = 0; m < nb; m++) { - wg_mqb = (*wg)(iq_loc_spin, m_iband); + occ_q[m] = (*wg)(iq_loc_spin, m); } + occ_q[nb] = kv->wk[iq_loc_spin]; + } #ifdef __MPI - MPI_Bcast(&wg_mqb, 1, MPI_DOUBLE, kv->para_k.get_startpro_pool(iq_pool), MPI_COMM_WORLD); + const int source_rank = kv->para_k.get_startpro_pool(iq_pool); + Parallel_Common::bcast_dev( + occ_q.data(), occupation_count, MPI_COMM_WORLD, source_rank); #endif + const Real wk_q = occ_q[nb]; + + for (int m_iband = 0; m_iband < nb; m_iband++) + { + const double wg_mqb = occ_q[m_iband]; if (wg_mqb < 1e-12) continue; - if (iq_pool == GlobalV::MY_POOL) + if (iq_pool == my_pool_) { - const T* psi_mq = get_pw(m_iband, iq_loc_spin); - wfcpw->recip_to_real(ctx, psi_mq, psi_mq_real, iq_loc); - // send + wfc_to_real_exx(get_pw(m_iband, iq_loc_spin), iq_loc, nbasis); } #ifdef __MPI - Parallel_Common::bcast_dev(psi_mq_real, wfcpw->nrxx, KP_WORLD, iq_pool); + const int grid_size = exx_grid_size(); + Parallel_Common::bcast_dev(psi_mq_real, grid_size, KP_WORLD, iq_pool); #endif - for (int n_iband = 0; n_iband < nbands; n_iband++) + + // k weight of the source k-point (identical to wk[this->ik] on the + // uniform k-grids without symmetry reduction this scheme assumes) + const Real factor = this->hybrid_alpha * wg_mqb / wk_q / nqs; + if (has_kpoint) { - const T* psi_nk = tmpsi_in + n_iband * nbasis; - // retrieve \psi_nk in real space - wfcpw->recip_to_real(ctx, psi_nk, psi_nk_real, this->ik); + apply_fock_all_bands(nbands, nbasis, iq, factor, tmhpsi); + } + } // end of m_iband - // direct multiplication in real space, \psi_nk(r) * \psi_mq(r) - cal_density_recip(psi_nk_real, psi_mq_real, ucell->omega); + } // end of iq - mul_potential_op()(pot, density_recip, rhopw_dev->npw, wfcpw->nks, this->ik, iq); + ModuleBase::timer::end("OperatorEXXPW", "act_op_kpar"); - // bring the potential back to real space - rho_recip2real(density_recip, density_real); +} - if (false) - { - // do nothing - } - else - { - vec_mul_vec_complex_op()(density_real, psi_mq_real, density_real, wfcpw->nrxx); - } +template +void OperatorEXXPW::maybe_setup_exx_grid() const +{ + if (exx_sg_init) + { + return; + } +#if !defined(__CUDA) + // the small-grid/batched path on GPU is implemented for CUDA only; on + // ROCm builds the GPU instantiation keeps the full-grid path + if (!std::is_same::value) + { + exx_sg_init = true; + return; + } +#endif + setup_exx_small_grid(); + if (std::is_same::value && !exx_sg_ok) + { + // the GPU bases own these maps as ig2ixyz*; the CPU batched path + // needs host-built ones for the full grid + setup_full_grid_maps(); + } +} - Real wk_iq = kv->wk[iq]; - Real wk_ik = kv->wk[this->ik]; +template +bool OperatorEXXPW::exx_grid_active() const +{ +#if !defined(__CUDA) + if (!std::is_same::value) + { + return false; // ROCm GPU + } +#endif + // the map-based FFT path needs the whole box local to this rank; the + // small grid additionally cuts the FFT work, the full-grid path mainly + // keeps the code path unified + return exx_sg_ok + || (wfcpw->nrxx == rhopw_dev->nrxx && wfcpw->nxyz == rhopw_dev->nxyz + && wfcpw->nrxx == wfcpw->nxyz); +} - Real tmp_scalar = wg_mqb / wk_ik / nqs; // wk_ik works for now, but wrong for symmetry. +template +int OperatorEXXPW::exx_band_chunk(const int nbands) const +{ + if (exx_batch_size_ > 0 && exx_batch_size_ < nbands) + { + return exx_batch_size_; + } + return nbands; +} - T* h_psi_nk = tmhpsi + n_iband * nbasis; - wfcpw->real_to_recip(ctx, density_real, h_psi_nk, this->ik, true, this->hybrid_alpha * tmp_scalar); +template +const int* OperatorEXXPW::active_map_rho() const +{ + if (exx_sg_ok) + { + return sg_map_rho; + } +#if defined(__CUDA) + if (std::is_same::value) + { + return rhopw_dev->ig2ixyz_gpu; + } +#endif + return full_map_rho; +} +template +const int* OperatorEXXPW::active_map_wfc(const int ik) const +{ + if (exx_sg_ok) + { + return sg_map_wfc + ik * wfcpw->npwk_max; + } +#if defined(__CUDA) + if (std::is_same::value) + { + return wfcpw->ig2ixyz_k + ik * wfcpw->npwk_max; + } +#endif + return full_map_wfc + ik * wfcpw->npwk_max; +} - } // end of m_iband - setmem_complex_op()(density_real, 0, rhopw_dev->nrxx); - setmem_complex_op()(density_recip, 0, rhopw_dev->npw); - setmem_complex_op()(psi_mq_real, 0, wfcpw->nrxx); +template +void OperatorEXXPW::cache_psi_nk_real(const int nbands, const int nbasis, const T* psi_in, const int ik) const +{ + if (psi_nk_cache_size < nbands) + { + // resize_memory_op frees the old block itself + resmem_complex_op()(psi_nk_real_cache, nbands * exx_grid_size()); + psi_nk_cache_size = nbands; + } + if (exx_grid_active()) + { + // scatter the PW coefficients into the active box and do batched + // backward FFTs (unnormalized, like recip_to_real), one band chunk + // at a time; the full cache stays valid for reuse over (iq, m) + const int chunk = exx_band_chunk(nbands); + for (int n0 = 0; n0 < nbands; n0 += chunk) + { + const int bn = std::min(chunk, nbands - n0); + ensure_exx_batch(bn); + T* cache_n = psi_nk_real_cache + n0 * exx_grid_size(); + setmem_complex_op()(cache_n, 0, bn * exx_grid_size()); + exx_batch_scatter_wfc(bn, + wfcpw->npwk[ik], + exx_grid_size(), + active_map_wfc(ik), + psi_in + n0 * nbasis, + nbasis, + cache_n); + exx_batch_fft_exec(exx_fft_plan, cache_n, false); + } + return; + } + for (int n = 0; n < nbands; n++) + { + wfcpw->recip_to_real(ctx, psi_in + n * nbasis, psi_nk_real_cache + n * wfcpw->nrxx, ik); + } +} - } // end of iq +template +void OperatorEXXPW::wfc_to_real_exx(const T* psi_m, const int iq, const int psi_stride) const +{ + if (exx_grid_active()) + { + // scatter + backward FFT on the active grid + setmem_complex_op()(psi_mq_real, 0, exx_grid_size()); + exx_batch_scatter_wfc(1, + wfcpw->npwk[iq], + exx_grid_size(), + active_map_wfc(iq), + psi_m, + psi_stride, + psi_mq_real); + exx_batch_fft_exec(exx_fft_plan1, psi_mq_real, false); + return; + } + wfcpw->recip_to_real(ctx, psi_m, psi_mq_real, iq); +} +template +void OperatorEXXPW::apply_fock_all_bands(const int nbands, + const int nbasis, + const int iq, + const Real factor, + T* tmhpsi) const +{ + if (exx_grid_active()) + { + apply_exx_nbatched(nbands, nbasis, psi_mq_real, factor, tmhpsi); + return; } + for (int n = 0; n < nbands; n++) + { + const T* nk_real = psi_nk_real_cache + n * wfcpw->nrxx; - ModuleBase::timer::end("OperatorEXXPW", "act_op_kpar"); + // direct multiplication in real space, \psi_nk(r) * \psi_mq(r) + cal_density_recip(nk_real, psi_mq_real, ucell->omega); + + // multiply the density with the potential in recip space + multiply_potential(density_recip, this->ik, iq); + + // bring the potential back to real space + rho_recip2real(density_recip, density_real); + vec_mul_vec_complex_op()(density_real, psi_mq_real, density_real, wfcpw->nrxx); + + wfcpw->real_to_recip(ctx, density_real, tmhpsi + n * nbasis, this->ik, true, factor); + } } + +template +void OperatorEXXPW::prepare_pair_densities(const int start, const int count) const +{ + if (exx_grid_active()) + { + // pair densities of one band chunk in one batched round, rhopw_dev G-space + dens_chunk_base = start; + ensure_exx_batch(count); + calc_density_pw_nbatched(count, psi_nk_real_cache + start * exx_grid_size(), psi_mq_real); + } +} + +template +const T* OperatorEXXPW::pair_density(const int n) const +{ + if (exx_grid_active()) + { + return dens_pw_batch + (n - dens_chunk_base) * rhopw_dev->npw; + } + cal_density_recip(psi_nk_real_cache + n * wfcpw->nrxx, psi_mq_real, ucell->omega); + return density_recip; +} + +template +void OperatorEXXPW::setup_exx_small_grid() const +{ + exx_sg_init = true; + exx_sg_ok = false; + + const bool user_set = ecutexx_user_set_; + const double ecut_exx = ecut_exx_; + + // FFT box dims for ecut_exx, obtained the same way as rhopw_dev + ModulePW::PW_Basis gridt(wfcpw->get_device(), rhopw->get_precision()); + gridt.fft_bundle.setfft(wfcpw->get_device(), rhopw->get_precision()); +#ifdef __MPI + gridt.initmpi(rhopw->poolnproc, rhopw->poolrank, rhopw->pool_world); +#endif + gridt.initgrids(rhopw->lat0, rhopw->latvec, ecut_exx); + // The small grid must be strictly smaller than the wavefunction box: + // psi_mq_real and psi_nk_real hold wfcpw->nrxx entries, so any larger + // box (e.g. ecutexx > ecutrho) would overflow them. initgrids rounds + // dimensions to FFT-friendly sizes, so compare volumes, not dimensions. + const long long sg_volume = static_cast(gridt.nx) * gridt.ny * gridt.nz; + if (sg_volume >= wfcpw->nrxx) + { + if (user_set && my_rank_ == 0) + { + ModuleBase::WARNING("OperatorEXXPW", + "ecutexx gives no smaller FFT grid than ecutrho; EXX stays on the full grid"); + } + return; // no gain over the current grid + } + if (rhopw->poolnproc != 1) + { + if (user_set && my_rank_ == 0) + { + ModuleBase::WARNING("OperatorEXXPW", + "ecutexx is set but the FFT box is distributed over the plane-wave pool; " + "the small-grid EXX path needs a local box, so EXX stays on the full grid"); + } + return; // batched path assumes the whole box is local + } + + // QE-style guard (gcutmt >= gkcut): every |k+G|^2 must fit the ecut_exx + // sphere, otherwise the wavefunctions cannot live on the small grid. + // Note: gk2 entries beyond npwk[ik] are uninitialized, so loop per k. + const Real tpiba2 = tpiba * tpiba; + for (int ik = 0; ik < wfcpw->nks; ik++) + { + for (int ig = 0; ig < wfcpw->npwk[ik]; ig++) + { + if (wfcpw->gk2[ik * wfcpw->npwk_max + ig] * tpiba2 > ecut_exx) + { + if (user_set && my_rank_ == 0) + { + ModuleBase::WARNING("OperatorEXXPW", + "ecutexx is smaller than the wavefunction cutoff |k+G|^2 of some " + "k-point; the wavefunctions would not fit the small grid, so EXX " + "stays on the full grid. Raise ecutexx above ~ecutwfc to enable it"); + } + return; + } + } + } + + sg_nx = gridt.nx; + sg_ny = gridt.ny; + sg_nz = gridt.nz; + sg_nxyz = sg_nx * sg_ny * sg_nz; + + // Remap a box index of the (nx, ny, nz) grid to the small grid, going + // through the Miller indices (both boxes are centered at G=0). + auto remap = [](const int idx, const int nx, const int ny, const int nz, + const int tnx, const int tny, const int tnz) { + const int iz = idx % nz; + const int iy = (idx / nz) % ny; + const int ix = idx / (ny * nz); + const int hx = ix >= nx / 2 + 1 ? ix - nx : ix; + const int hy = iy >= ny / 2 + 1 ? iy - ny : iy; + const int hz = iz >= nz / 2 + 1 ? iz - nz : iz; + const int tx = hx < 0 ? hx + tnx : hx; + const int ty = hy < 0 ? hy + tny : hy; + const int tz = hz < 0 ? hz + tnz : hz; + return tz + ty * tnz + tx * tny * tnz; + }; + + // rho map: rhopw_dev G-vectors (box = wfcpw/rhopw grid) -> small box + const int npw_rho = rhopw_dev->npw; + std::vector host_map(npw_rho); + std::vector host_ig2ixyz(npw_rho); + if (rhopw_dev->ig2ixyz_gpu != nullptr) + { + syncmem_int_d2h_op()(host_ig2ixyz.data(), rhopw_dev->ig2ixyz_gpu, npw_rho); + } + else + { + // host basis without ig2ixyz: rebuild the big-box index from the + // Miller indices (gdirect holds integer values as doubles) + for (int ig = 0; ig < npw_rho; ig++) + { + const int hx = static_cast(std::lround(rhopw_dev->gdirect[ig].x)); + const int hy = static_cast(std::lround(rhopw_dev->gdirect[ig].y)); + const int hz = static_cast(std::lround(rhopw_dev->gdirect[ig].z)); + const int ix = hx < 0 ? hx + wfcpw->nx : hx; + const int iy = hy < 0 ? hy + wfcpw->ny : hy; + const int iz = hz < 0 ? hz + wfcpw->nz : hz; + host_ig2ixyz[ig] = iz + iy * wfcpw->nz + ix * wfcpw->ny * wfcpw->nz; + } + } + for (int ig = 0; ig < npw_rho; ig++) + { + host_map[ig] = remap(host_ig2ixyz[ig], wfcpw->nx, wfcpw->ny, wfcpw->nz, sg_nx, sg_ny, sg_nz); + } + resmem_int_op()(sg_map_rho, npw_rho); + syncmem_int_h2d_op()(sg_map_rho, host_map.data(), npw_rho); + + // wfc map: (G+k) components of every k-point -> small box + const int n_tot = wfcpw->npwk_max * wfcpw->nks; + host_map.resize(n_tot); + if (!wfcpw->ig2ixyz_k_cpu.empty()) + { + for (int i = 0; i < n_tot; i++) + { + host_map[i] = remap(wfcpw->ig2ixyz_k_cpu[i], wfcpw->nx, wfcpw->ny, wfcpw->nz, sg_nx, sg_ny, sg_nz); + } + } + else + { + // host basis without ig2ixyz_k_cpu: same construction as + // PW_Basis_K::get_ig2ixyz_k, from the stick decomposition (the small + // grid requires a local box, so all sticks are on this rank) + for (int ik = 0; ik < wfcpw->nks; ik++) + { + for (int igl = 0; igl < wfcpw->npwk[ik]; igl++) + { + const int isz = wfcpw->igl2isz_k[igl + ik * wfcpw->npwk_max]; + const int iz = isz % wfcpw->nz; + const int is = isz / wfcpw->nz; + const int ixy = wfcpw->is2fftixy[is]; + const int iy = ixy % wfcpw->ny; + const int ix = ixy / wfcpw->ny; + host_map[igl + ik * wfcpw->npwk_max] = + remap(iz + iy * wfcpw->nz + ix * wfcpw->ny * wfcpw->nz, + wfcpw->nx, wfcpw->ny, wfcpw->nz, sg_nx, sg_ny, sg_nz); + } + } + } + resmem_int_op()(sg_map_wfc, n_tot); + syncmem_int_h2d_op()(sg_map_wfc, host_map.data(), n_tot); + + exx_sg_ok = true; + if (my_rank_ == 0) + { + std::cout << " EXX small grid enabled: FFT (" << sg_nx << "," << sg_ny << "," << sg_nz << ") instead of (" + << wfcpw->nx << "," << wfcpw->ny << "," << wfcpw->nz << ") for ecut_exx = " << ecut_exx << " Ry" + << std::endl; + } +} + +template +void OperatorEXXPW::setup_full_grid_maps() const +{ + // host-built full-box maps for the CPU batched path when the small grid + // is not active; same sources as in setup_exx_small_grid + const int npw_rho = rhopw_dev->npw; + std::vector host_map(npw_rho); + for (int ig = 0; ig < npw_rho; ig++) + { + const int hx = static_cast(std::lround(rhopw_dev->gdirect[ig].x)); + const int hy = static_cast(std::lround(rhopw_dev->gdirect[ig].y)); + const int hz = static_cast(std::lround(rhopw_dev->gdirect[ig].z)); + const int ix = hx < 0 ? hx + wfcpw->nx : hx; + const int iy = hy < 0 ? hy + wfcpw->ny : hy; + const int iz = hz < 0 ? hz + wfcpw->nz : hz; + host_map[ig] = iz + iy * wfcpw->nz + ix * wfcpw->ny * wfcpw->nz; + } + resmem_int_op()(full_map_rho, npw_rho); + syncmem_int_h2d_op()(full_map_rho, host_map.data(), npw_rho); + + const int n_tot = wfcpw->npwk_max * wfcpw->nks; + host_map.resize(n_tot); + for (int ik = 0; ik < wfcpw->nks; ik++) + { + for (int igl = 0; igl < wfcpw->npwk[ik]; igl++) + { + const int isz = wfcpw->igl2isz_k[igl + ik * wfcpw->npwk_max]; + const int iz = isz % wfcpw->nz; + const int is = isz / wfcpw->nz; + const int ixy = wfcpw->is2fftixy[is]; + const int iy = ixy % wfcpw->ny; + const int ix = ixy / wfcpw->ny; + host_map[igl + ik * wfcpw->npwk_max] = iz + iy * wfcpw->nz + ix * wfcpw->ny * wfcpw->nz; + } + } + resmem_int_op()(full_map_wfc, n_tot); + syncmem_int_h2d_op()(full_map_wfc, host_map.data(), n_tot); +} + +template +void OperatorEXXPW::ensure_exx_batch(const int nbands) const +{ + const int nx = exx_sg_ok ? sg_nx : wfcpw->nx; + const int ny = exx_sg_ok ? sg_ny : wfcpw->ny; + const int nz = exx_sg_ok ? sg_nz : wfcpw->nz; + const int nxyz = exx_sg_ok ? sg_nxyz : wfcpw->nxyz; + if (exx_batch_alloc == nbands && exx_fft_plan_nx == nx && exx_fft_plan_ny == ny && exx_fft_plan_nz == nz) + { + return; + } + if (dens_box_batch != nullptr) + { + // resize_memory_op frees the old block itself; the plans need an + // explicit destroy (they are plain library handles) + exx_batch_fft_plan_destroy(&exx_fft_plan); + exx_batch_fft_plan_destroy(&exx_fft_plan1); + } + resmem_complex_op()(dens_box_batch, nbands * nxyz); + resmem_complex_op()(dens_pw_batch, nbands * rhopw_dev->npw); + exx_batch_fft_plan_create(&exx_fft_plan, nx, ny, nz, nbands); + exx_batch_fft_plan_create(&exx_fft_plan1, nx, ny, nz, 1); + exx_batch_alloc = nbands; + exx_fft_plan_nx = nx; + exx_fft_plan_ny = ny; + exx_fft_plan_nz = nz; +} + +template +void OperatorEXXPW::apply_exx_nbatched(const int nbands, + const int nbasis, + const T* psi_mq_real, + const Real factor, + T* tmhpsi) const +{ + // active FFT grid: the small ecut_exx grid when usable, else the wfcpw grid + const int nxyz = exx_sg_ok ? sg_nxyz : wfcpw->nxyz; + const int nrxx = nxyz; // batched path requires the whole box to be local + const int npw_rho = rhopw_dev->npw; + const int npwk = wfcpw->npwk[this->ik]; + const int* map_rho = active_map_rho(); + const int* map_wfc = active_map_wfc(this->ik); + + const int chunk = exx_band_chunk(nbands); + for (int n0 = 0; n0 < nbands; n0 += chunk) + { + const int bn = std::min(chunk, nbands - n0); + ensure_exx_batch(bn); + // 1. reciprocal-space density of this band chunk, ends up in dens_pw_batch + calc_density_pw_nbatched(bn, psi_nk_real_cache + n0 * exx_grid_size(), psi_mq_real); + // 2. multiply by the Coulomb potential in recip space + exx_batch_mul_pot(bn, npw_rho, pot, dens_pw_batch); + // 3. scatter back to the (pre-zeroed) box + setmem_complex_op()(dens_box_batch, 0, bn * nxyz); + exx_batch_scatter_pw(bn, npw_rho, nxyz, map_rho, dens_pw_batch, dens_box_batch); + // 4. batched backward FFT (unnormalized, matching recip2real) + exx_batch_fft_exec(exx_fft_plan, dens_box_batch, false); + // 5. multiply by psi_mq(r) in real space + exx_batch_mul_real(bn, nrxx, psi_mq_real, dens_box_batch); + // 6. batched forward FFT + exx_batch_fft_exec(exx_fft_plan, dens_box_batch, true); + // 7. gather and accumulate into hpsi with the EXX weight + exx_batch_gather_accum(bn, + npwk, + nxyz, + map_wfc, + dens_box_batch, + factor, + tmhpsi + n0 * nbasis, + nbasis); + } +} + +template +void OperatorEXXPW::calc_density_pw_nbatched(const int nbands, const T* nk_real, const T* psi_mq_real) const +{ + // active FFT grid: the small ecut_exx grid when usable, else the wfcpw grid + const int nxyz = exx_sg_ok ? sg_nxyz : wfcpw->nxyz; + const int nrxx = nxyz; // batched path requires the whole box to be local + const int npw_rho = rhopw_dev->npw; + const int* map_rho = active_map_rho(); + + // 1. density_real(r) = psi_nk(r) * conj(psi_mq(r)) / omega for all bands at once + exx_batch_density_real(nbands, nrxx, nk_real, psi_mq_real, ucell->omega, dens_box_batch); + // 2. batched forward FFT to the box + exx_batch_fft_exec(exx_fft_plan, dens_box_batch, true); + // 3. gather the PW components (with the 1/nxyz of real_to_recip) + exx_batch_gather_pw(nbands, npw_rho, nxyz, map_rho, dens_box_batch, dens_pw_batch); +} + + + template std::vector OperatorEXXPW::get_q_points(const int ik) const { @@ -417,11 +874,11 @@ std::vector OperatorEXXPW::get_q_points(const int ik) const { for (int iq = 0; iq < wfcpw->nks; iq++) { - if (PARAM.inp.nspin ==1 ) + if (nspin_ ==1 ) { q_points_ik.push_back(iq); } - else if (PARAM.inp.nspin == 2) + else if (nspin_ == 2) { int nk_fac = 2; int nk = wfcpw->nks / nk_fac; @@ -454,7 +911,7 @@ void OperatorEXXPW::multiply_potential(T *density_recip, int ik, int ModuleBase::timer::start("OperatorEXXPW", "multiply_potential"); int npw = rhopw_dev->npw; int nks = wfcpw->nks; - int nk_fac = PARAM.inp.nspin == 2 ? 2 : 1; + int nk_fac = nspin_ == 2 ? 2 : 1; int nk = nks / nk_fac; mul_potential_op()(pot, density_recip, npw, nks, ik, iq); @@ -483,6 +940,13 @@ OperatorEXXPW::OperatorEXXPW(const OperatorEXXPW *op this->psi = op->psi; this->ctx = op->ctx; this->cpu_ctx = op->cpu_ctx; + this->nspin_ = op->nspin_; + this->ecut_exx_ = op->ecut_exx_; + this->ecutexx_user_set_ = op->ecutexx_user_set_; + this->exx_batch_size_ = op->exx_batch_size_; + this->exxace_ = op->exxace_; + this->my_rank_ = op->my_rank_; + this->my_pool_ = op->my_pool_; resmem_complex_op()(this->ctx, psi_nk_real, wfcpw->nrxx); resmem_complex_op()(this->ctx, psi_mq_real, wfcpw->nrxx); resmem_complex_op()(this->ctx, density_real, rhopw_dev->nrxx); @@ -497,7 +961,7 @@ OperatorEXXPW::OperatorEXXPW(const OperatorEXXPW *op template double OperatorEXXPW::cal_exx_energy(psi::Psi *psi_) const { - if (PARAM.inp.exxace && this->separate_loop) + if (exxace_ && this->separate_loop) { return cal_exx_energy_ace(psi_); } @@ -522,91 +986,88 @@ double OperatorEXXPW::cal_exx_energy_op(psi::Psi *ppsi_) c setmem_complex_op()(density_recip, 0, rhopw_dev->npw); if (wg == nullptr) return 0.0; - const int nk_fac = PARAM.inp.nspin == 2 ? 2 : 1; + const int nk_fac = nspin_ == 2 ? 2 : 1; + const int nb = psi.get_nbands(); + const int nbasis = psi_.get_nbasis(); + const int npw = rhopw_dev->npw; + maybe_setup_exx_grid(); + double Eexx_ik_real = 0.0; for (int ik = 0; ik < wfcpw->nks; ik++) { - // auto k = this->pw_wfc->kvec_c[ik]; - // std::cout << k << std::endl; - for (int n_iband = 0; n_iband < psi.get_nbands(); n_iband++) + // NOTE: psi_nk intentionally comes from the member psi (state of the + // previous set_psi), while psi_mq comes from the argument psi_ — + // iter_finish computes dexx as the difference of two cal_exx_energy + // calls around set_psi, so the asymmetry is load-bearing for the EXX + // outer-loop convergence check. + psi.fix_kb(ik, 0); + cache_psi_nk_real(nb, nbasis, psi.get_pointer(), ik); + + // q points of the same spin channel as ik + std::vector q_points_ik; + if (nspin_ == 1) { - setmem_complex_op()(h_psi_recip, 0, wfcpw->npwk_max); - setmem_complex_op()(h_psi_real, 0, rhopw_dev->nrxx); - setmem_complex_op()(density_real, 0, rhopw_dev->nrxx); - setmem_complex_op()(density_recip, 0, rhopw_dev->npw); - - double wg_ikb_real = (*wg)(ik, n_iband); - T wg_ikb = wg_ikb_real; - if (wg_ikb_real < 1e-12) + for (int iq = 0; iq < wfcpw->nks; iq++) { - continue; + q_points_ik.push_back(iq); } - - // const T *psi_nk = get_pw(n_iband, ik); - psi.fix_kb(ik, n_iband); - const T* psi_nk = psi.get_pointer(); - // retrieve \psi_nk in real space - wfcpw->recip_to_real(ctx, psi_nk, psi_nk_real, ik); - - // for \psi_nk, get the pw of iq and band m - // q_points is a vector of integers, 0 to nks-1 - std::vector q_points; - if (PARAM.inp.nspin == 1) + } + else if (nspin_ == 2) + { + const int nk = wfcpw->nks / nk_fac; + const int k_spin = ik / nk; + for (int iq = k_spin * nk; iq < (k_spin + 1) * nk; iq++) { - for (int iq = 0; iq < wfcpw->nks; iq++) - { - q_points.push_back(iq); - } + q_points_ik.push_back(iq); } - else if (PARAM.inp.nspin == 2) + } + else + { + ModuleBase::WARNING_QUIT("OperatorEXXPW", "nspin == 4 not supported"); + } + const double nqs = q_points_ik.size(); + const int nk = wfcpw->nks / nk_fac; + + for (int iq: q_points_ik) + { + get_exx_potential(kv, wfcpw, rhopw_dev, pot, tpiba, gamma_extrapolation, ucell->omega, ik, iq % nk, false, this->coulomb_param); + for (int m_iband = 0; m_iband < nb; m_iband++) { - int nk = wfcpw->nks / nk_fac; - int k_spin = ik / nk; - for (int iq = 0; iq < wfcpw->nks; iq++) + const double wg_iqb_real = (*wg)(iq, m_iband); + if (wg_iqb_real < 1e-12) { - int q_spin = iq / nk; - if (k_spin == q_spin) - { - q_points.push_back(iq); - } + continue; } - } - else - { - ModuleBase::WARNING_QUIT("OperatorEXXPW", "nspin == 4 not supported"); - } - double nqs = q_points.size(); + psi_.fix_kb(iq, m_iband); + wfc_to_real_exx(psi_.get_pointer(), iq, nbasis); - for (int iq: q_points) - { - int nk = wfcpw->nks / nk_fac; - get_exx_potential(kv, wfcpw, rhopw_dev, pot, tpiba, gamma_extrapolation, ucell->omega, ik, iq % nk, false, this->coulomb_param); - for (int m_iband = 0; m_iband < psi.get_nbands(); m_iband++) + // pair densities on the active grid, one band chunk per + // batched round; the energy kernel works purely in G-space + const int chunk = exx_grid_active() ? exx_band_chunk(nb) : nb; + for (int n0 = 0; n0 < nb; n0 += chunk) { - double wg_iqb_real = (*wg)(iq, m_iband); - T wg_iqb = wg_iqb_real; - if (wg_iqb_real < 1e-12) + const int bn = std::min(chunk, nb - n0); + prepare_pair_densities(n0, bn); + for (int nn = 0; nn < bn; nn++) { - continue; + const int n_iband = n0 + nn; + const double wg_ikb_real = (*wg)(ik, n_iband); + if (wg_ikb_real < 1e-12) + { + continue; + } + // pure occupations f = wg / wk; divide by the q-mesh + // size nqs to apply the uniform q-point BZ weight, and + // by wk[ik] to convert wg(ik) into the occupation. + const double exx_weight = wg_iqb_real / nqs * wg_ikb_real / kv->wk[ik]; + Eexx_ik_real += exx_cal_energy_op()(pair_density(n_iband), + pot, + exx_weight, + npw); } - - psi_.fix_kb(iq, m_iband); - const T* psi_mq = psi_.get_pointer(); - // const T* psi_mq = get_pw(m_iband, iq); - wfcpw->recip_to_real(ctx, psi_mq, psi_mq_real, iq); - - cal_density_recip(psi_nk_real, psi_mq_real, ucell->omega); - - int nks = wfcpw->nks; - int npw = rhopw_dev->npw; - // int nk = nks / nk_fac; - Eexx_ik_real += exx_cal_energy_op()(density_recip, pot, wg_iqb_real / nqs * wg_ikb_real / kv->wk[ik], npw); - - } // m_iband - - } // iq - - } // n_iband + } + } // m_iband + } // iq } // ik Eexx_ik_real *= 0.5 * ucell->omega; diff --git a/source/source_pw/module_pwdft/op_pw_exx.h b/source/source_pw/module_pwdft/op_pw_exx.h index 4d378325b28..6e0131ab83b 100644 --- a/source/source_pw/module_pwdft/op_pw_exx.h +++ b/source/source_pw/module_pwdft/op_pw_exx.h @@ -16,6 +16,9 @@ #include #include +/// General_Exx_Info forward declaration, full definition in general_exx_info.h +struct General_Exx_Info; + namespace hamilt { @@ -31,9 +34,11 @@ class OperatorEXXPW : public OperatorPW const ModulePW::PW_Basis* rhopw_in, K_Vectors* kv_in, const UnitCell* ucell, - const bool separate_loop_in, - const Real hybrid_alpha_in, - const CoulombParam& coulomb_param_in); + const General_Exx_Info& exx_info, + const int nspin_in, + const int kpar_in, + const int my_rank_in, + const int my_pool_in); template explicit OperatorEXXPW(const OperatorEXXPW *op_exx); @@ -70,7 +75,17 @@ class OperatorEXXPW : public OperatorPW ModulePW::PW_Basis* rhopw_dev = nullptr; // for device const UnitCell *ucell = nullptr; Real tpiba = 0; - + + // INPUT/runtime configuration snapshotted at construction, so the + // operator never touches the global parameter objects itself + int nspin_ = 1; + double ecut_exx_ = 0.0; // resolved EXX cutoff (Ry) + bool ecutexx_user_set_ = false; // gates the small-grid fallback warnings + int exx_batch_size_ = 0; // band chunk width, 0 = all bands + bool exxace_ = false; + int my_rank_ = 0; // world rank, gates the one-time reports + int my_pool_ = 0; // k-point pool of this rank + std::vector get_q_points(const int ik) const; const T *get_pw(const int m, const int iq) const; @@ -84,13 +99,14 @@ class OperatorEXXPW : public OperatorPW const int ngk_ik = 0, const bool is_first_node = false) const; + // Both active and idle ACE pools join the same source-q broadcasts. + // has_kpoint identifies a local target k point in the current spin channel. void act_op_kpar(const int nbands, - const int nbasis, - const int npol, - const T *tmpsi_in, - T *tmhpsi, - const int ngk_ik = 0, - const bool is_first_node = false) const; + const int nbasis, + const T* tmpsi_in, + T* tmhpsi, + const int ispin, + const bool has_kpoint) const; void act_op_ace(const int nbands, const int nbasis, @@ -125,6 +141,90 @@ class OperatorEXXPW : public OperatorPW // real space memory T *psi_nk_real = nullptr; T *psi_mq_real = nullptr; + // cache of psi_nk in real space for all bands on the active grid + mutable T* psi_nk_real_cache = nullptr; + mutable int psi_nk_cache_size = 0; // number of bands currently allocated in the cache + + // ---- EXX FFT grid + band-batch layer ---- + // All EXX PW entry points (act_op, act_op_kpar, + // cal_exx_energy_op) share one code path built on these primitives; the + // batched kernels are a specialization of the per-band operations, + // selected inside the primitives (exx_grid_active). The active real-space + // grid is the small ecut_exx box when usable, else the wfcpw box; + // callers must not branch on this themselves. + void maybe_setup_exx_grid() const; + // whether the map-based (batched-kernel) path can run at all: the small + // grid is usable, or the full box is local to this rank + bool exx_grid_active() const; + // band chunk width of one batched round, 1..nbands: the exx_batch_size + // INPUT (0 = all bands). Only meaningful when exx_grid_active(); the + // result is independent of the chunking. + int exx_band_chunk(const int nbands) const; + int exx_grid_size() const { return exx_sg_ok ? sg_nxyz : wfcpw->nrxx; } + // G-vector -> active FFT box index maps, hiding the grid choice: the + // small-grid maps when exx_sg_ok, the basis's own ig2ixyz arrays on CUDA, + // and the host-built full-box maps (full_map_*) on CPU otherwise + const int* active_map_rho() const; + const int* active_map_wfc(const int ik) const; + // fill psi_nk_real_cache: all bands of psi_in at k-point ik, real space on + // the active grid + void cache_psi_nk_real(const int nbands, const int nbasis, const T* psi_in, const int ik) const; + // psi_mq_real: single band psi_m at k-point iq, real space on the active + // grid (psi_m points to a band block with row stride psi_stride) + void wfc_to_real_exx(const T* psi_m, const int iq, const int psi_stride) const; + // tmhpsi += factor * V_x|psi> for all bands; needs psi_nk_real_cache and + // psi_mq_real filled, and `pot` holding the Coulomb kernel for (ik, iq) + void apply_fock_all_bands(const int nbands, const int nbasis, const int iq, const Real factor, T* tmhpsi) const; + // pair densities psi_nk* psi_mq of bands [start, start+count) in the + // rhopw_dev G-space; pair_density(n) then returns band n (global index) + // without recomputation + void prepare_pair_densities(const int start, const int count) const; + const T* pair_density(const int n) const; + + // Small EXX FFT grid sized by ecut_exx (like QE's dfftt): when usable, the + // batched path runs all FFTs on this smaller grid instead of the wfcpw grid. + // Requires all |k+G|^2 < ecut_exx (otherwise the wavefunctions do not fit). + // Available on CPU and CUDA; on ROCm the GPU path keeps the full grid + // (exx_sg_ok stays false there). + void setup_exx_small_grid() const; + // host-built full-box maps for the batched path on CPU when the small + // grid is not active (the GPU bases own these as ig2ixyz*) + void setup_full_grid_maps() const; + mutable bool exx_sg_init = false; + mutable bool exx_sg_ok = false; + mutable int sg_nx = 0, sg_ny = 0, sg_nz = 0, sg_nxyz = 0; + mutable int* sg_map_rho = nullptr; // rhopw_dev G-vector ig -> small box index [rhopw_dev->npw] + mutable int* sg_map_wfc = nullptr; // wfcpw (G+k) ig -> small box index [npwk_max * nks] + mutable int* full_map_rho = nullptr; // rhopw_dev G-vector ig -> full box index (CPU only) + mutable int* full_map_wfc = nullptr; // wfcpw (G+k) ig -> full box index (CPU only) + + // Batched n-band loop of act_op / act_op_kpar: all bands of one + // (iq, m_iband) pair are processed with batched FFTs and a handful of + // kernel launches instead of ~8 launches per band. Requires the psi_nk + // real-space cache and a local (non-distributed) FFT box. `factor` is the + // full accumulation weight (including hybrid_alpha and the k/q weights). + void apply_exx_nbatched(const int nbands, + const int nbasis, + const T* psi_mq_real, + const Real factor, + T* tmhpsi) const; + + // reciprocal-space density of bands [0, nbands) of the nk_real block + // (rhopw_dev G-space), ends up in dens_pw_batch + void calc_density_pw_nbatched(const int nbands, const T* nk_real, const T* psi_mq_real) const; + + mutable void* exx_fft_plan = nullptr; // batched FFT plan (void* to keep FFTW/cuFFT out of the header) + mutable void* exx_fft_plan1 = nullptr; // batch-1 FFT plan (psi_mq on the active grid) + mutable int exx_fft_plan_nx = 0; // grid dims the plans were created for + mutable int exx_fft_plan_ny = 0; + mutable int exx_fft_plan_nz = 0; + mutable T* dens_box_batch = nullptr; // band-chunk * nxyz box buffer + mutable T* dens_pw_batch = nullptr; // band-chunk * npw (rhopw_dev) plane-wave buffer + mutable int exx_batch_alloc = 0; // band chunk width the buffers/plan are allocated for + mutable int dens_chunk_base = 0; // global band index of dens_pw_batch[0] + + // create/recreate the batched FFT plans and work buffers for the active grid + void ensure_exx_batch(const int nbands) const; T *density_real = nullptr; T *h_psi_real = nullptr; // density recip space memory @@ -168,6 +268,10 @@ class OperatorEXXPW : public OperatorPW using syncmem_complex_d2c_op = base_device::memory::synchronize_memory_op; using syncmem_real_c2d_op = base_device::memory::synchronize_memory_op; using syncmem_real_d2c_op = base_device::memory::synchronize_memory_op; + using resmem_int_op = base_device::memory::resize_memory_op; + using delmem_int_op = base_device::memory::delete_memory_op; + using syncmem_int_h2d_op = base_device::memory::synchronize_memory_op; + using syncmem_int_d2h_op = base_device::memory::synchronize_memory_op; using lapack_potrf = container::kernels::lapack_potrf; using lapack_trtri = container::kernels::lapack_trtri; diff --git a/source/source_pw/module_pwdft/op_pw_exx_ace.cpp b/source/source_pw/module_pwdft/op_pw_exx_ace.cpp index 67322bdc493..e598df6ea83 100644 --- a/source/source_pw/module_pwdft/op_pw_exx_ace.cpp +++ b/source/source_pw/module_pwdft/op_pw_exx_ace.cpp @@ -78,6 +78,7 @@ void OperatorEXXPW::construct_ace() const int nbands = psi.get_nbands(); int nbasis = psi.get_nbasis(); int nk = psi.get_nk(); + const int hpsi_size = nbands * nbasis; int* ik_ = const_cast(&this->ik); int ik_save = this->ik; @@ -119,82 +120,30 @@ void OperatorEXXPW::construct_ace() const ModuleBase::timer::start("OperatorEXXPW", "construct_ace"); int nk_max = kv->para_k.get_max_nks_pool(); - int nspin_fac = PARAM.inp.nspin == 2 ? 2 : 1; + int nspin_fac = nspin_ == 2 ? 2 : 1; for (int ispin = 0; ispin < nspin_fac; ispin++) { for (int ik0 = 0; ik0 < nk_max; ik0++) { - int ik = ik0 + ispin * wfcpw->nks / nspin_fac; - // printf("ik: %d\n", ik); - int npwk = wfcpw->npwk[ik]; - - T* Xi_ace = Xi_ace_k[ik]; - psi.fix_kb(ik, 0); - T* p_psi = psi.get_pointer(); - - setmem_complex_op()(h_psi_ace, 0, nbands * nbasis); - - setmem_complex_op()(h_psi_recip, 0, wfcpw->npwk_max); - setmem_complex_op()(h_psi_real, 0, rhopw_dev->nrxx); - setmem_complex_op()(density_real, 0, rhopw_dev->nrxx); - setmem_complex_op()(density_recip, 0, rhopw_dev->npw); - setmem_complex_op()(psi_nk_real, 0, wfcpw->nrxx); - setmem_complex_op()(psi_mq_real, 0, wfcpw->nrxx); - int nqs = kv->get_nkstot_nospin(); - - bool skip_ik = false; - if (ik >= wfcpw->nks) + const int nks_per_spin = wfcpw->nks / nspin_fac; + const bool has_kpoint = ik0 < nks_per_spin; + // Use an out-of-range sentinel on idle pools. They still enter the + // same q-point collectives, but never access local k-point data. + const int ik = has_kpoint ? ik0 + ispin * nks_per_spin : wfcpw->nks; + *ik_ = ik; + T* p_psi = nullptr; + if (has_kpoint) { - skip_ik = true; + psi.fix_kb(ik, 0); + p_psi = psi.get_pointer(); } - if (skip_ik) - { - // ik fixed here, select band n - for (int iq0 = 0; iq0 < nqs; iq0++) - { - // For nspin=2, iq should be in the same spin channel as ik - int iq = 0; - - int nk = wfcpw->nks / 2; - iq = iq0 + ispin * nk; // iq in the same spin channel - - // for \psi_nk, get the pw of iq and band m - get_exx_potential(kv, wfcpw, rhopw_dev, pot, tpiba, gamma_extrapolation, ucell->omega, ik, iq, false, this->coulomb_param); - - // decide which pool does the iq belong to - int iq_pool = kv->para_k.whichpool[iq0]; - int iq_loc = iq - kv->para_k.startk_pool[iq_pool]; - - for (int m_iband = 0; m_iband < psi.get_nbands(); m_iband++) - { - double wg_mqb = 0; - if (iq_pool == GlobalV::MY_POOL) - { - wg_mqb = (*wg)(iq_loc, m_iband); - } -#ifdef __MPI - MPI_Bcast(&wg_mqb, 1, MPI_DOUBLE, kv->para_k.get_startpro_pool(iq_pool), MPI_COMM_WORLD); -#endif - if (wg_mqb < 1e-12) - continue; - - if (iq_pool == GlobalV::MY_POOL) - { - const T* psi_mq = get_pw(m_iband, iq_loc); - wfcpw->recip_to_real(ctx, psi_mq, psi_mq_real, iq_loc); - } -#ifdef __MPI - Parallel_Common::bcast_dev(psi_mq_real, wfcpw->nrxx, KP_WORLD, iq_pool); -#endif - - } // end of iq + setmem_complex_op()(h_psi_ace, 0, hpsi_size); + act_op_kpar(nbands, nbasis, p_psi, h_psi_ace, ispin, has_kpoint); - } - } - else + if (has_kpoint) { - *ik_ = ik; - act_op_kpar(nbands, nbasis, 1, p_psi, h_psi_ace, nbasis, false); + const int npwk = wfcpw->npwk[ik]; + T* Xi_ace = Xi_ace_k[ik]; // psi_h_psi_ace = psi^\dagger * h_psi_ace // p_exx_helper->psi.fix_kb(0, 0); gemm_complex_op()('C', @@ -274,7 +223,7 @@ void OperatorEXXPW::construct_ace() const nbands); // clear mem - setmem_complex_op()(h_psi_ace, 0, nbands * nbasis); + setmem_complex_op()(h_psi_ace, 0, hpsi_size); setmem_complex_op()(psi_h_psi_ace, 0, nbands * nbands); setmem_complex_op()(L_ace, 0, nbands * nbands); } @@ -291,7 +240,7 @@ template double OperatorEXXPW::cal_exx_energy_ace(psi::Psi* ppsi_) const { double Eexx = 0; - int nspin_fac = PARAM.inp.nspin == 2 ? 2 : 1; + int nspin_fac = nspin_ == 2 ? 2 : 1; psi::Psi psi_ = *ppsi_; int* ik_ = const_cast(&this->ik); int ik_save = this->ik; @@ -329,4 +278,4 @@ template class OperatorEXXPW, base_device::DEVICE_GPU>; template class OperatorEXXPW, base_device::DEVICE_GPU>; #endif -} // namespace hamilt \ No newline at end of file +} // namespace hamilt diff --git a/source/source_pw/module_pwdft/op_pw_exx_pot.cpp b/source/source_pw/module_pwdft/op_pw_exx_pot.cpp index 2a99833bbfa..6094868defe 100644 --- a/source/source_pw/module_pwdft/op_pw_exx_pot.cpp +++ b/source/source_pw/module_pwdft/op_pw_exx_pot.cpp @@ -2,8 +2,53 @@ #include "source_base/parallel_reduce.h" #include "source_io/module_parameter/parameter.h" +#include +#include +#include + +#ifdef __CUDA +#include +#endif + namespace hamilt { +#ifdef __CUDA +namespace +{ +class CudaHostRegistration +{ + public: + CudaHostRegistration(void* data, std::size_t bytes, bool enabled) : data_(data) + { + if (enabled) + { + const cudaError_t err = cudaHostRegister(data_, bytes, cudaHostRegisterPortable); + if (err != cudaSuccess) + { + throw std::runtime_error("failed to register potential CPU memory operations"); + } + registered_ = true; + } + } + + ~CudaHostRegistration() + { + if (registered_) + { + cudaHostUnregister(data_); + } + } + + CudaHostRegistration(const CudaHostRegistration&) = delete; + CudaHostRegistration& operator=(const CudaHostRegistration&) = delete; + + private: + void* data_; + bool registered_ = false; +}; +} // namespace +#endif + extern template class OperatorEXXPW, base_device::DEVICE_CPU>; extern template class OperatorEXXPW, base_device::DEVICE_CPU>; #if ((defined __CUDA) || (defined __ROCM)) @@ -24,20 +69,15 @@ void get_exx_potential(const K_Vectors* kv, bool is_stress, const CoulombParam& coulomb_param_in) { - using setmem_real_cpu_op = base_device::memory::set_memory_op; using syncmem_real_c2d_op = base_device::memory::synchronize_memory_op; Real nqs_half1 = 0.5 * kv->nmp[0]; Real nqs_half2 = 0.5 * kv->nmp[1]; Real nqs_half3 = 0.5 * kv->nmp[2]; - Real* pot_cpu = nullptr; - int nks = wfcpw->nks, npw = rhopw_dev->npw; + int nks = wfcpw->nks; + int npw = rhopw_dev->npw; double tpiba2 = tpiba * tpiba; - pot_cpu = new Real[npw]; - // fill zero - setmem_real_cpu_op()(pot_cpu, 0, npw); - std::vector> qvec_c, qvec_d; #ifdef __MPI kv->para_k.gatherkvec(kv->kvec_c, qvec_c); @@ -47,11 +87,13 @@ void get_exx_potential(const K_Vectors* kv, qvec_d = kv->kvec_d; #endif - if (ik > nks) + if (ik >= nks) { return; } + std::vector pot_cpu(npw, 0); + // calculate Fock pot auto it_fock = coulomb_param_in.find(Conv_Coulomb_Pot_K::Coulomb_Type::Fock); if (it_fock != coulomb_param_in.end()) @@ -125,15 +167,10 @@ void get_exx_potential(const K_Vectors* kv, double erfc_omega = std::stod(param["omega"]); double erfc_omega2 = erfc_omega * erfc_omega; double alpha = std::stod(param["alpha"]); - // double exx_div = OperatorEXXPW, Device>::erfc_div[i]; - double exx_div = exx_divergence(Conv_Coulomb_Pot_K::Coulomb_Type::Erfc, - erfc_omega, - kv, - wfcpw, - rhopw_dev, - tpiba, - gamma_extrapolation, - ucell_omega); + // Like the Fock correction, this global q sum is initialized by + // the operator. Recomputing it here would require idle pools to + // join a collective after their local k-point bounds check. + const double exx_div = OperatorEXXPW, Device>::erfc_div[i]; const ModuleBase::Vector3 k_c = wfcpw->kvec_c[ik]; const ModuleBase::Vector3 k_d = wfcpw->kvec_d[ik]; const ModuleBase::Vector3 q_c = qvec_c[iq]; @@ -205,24 +242,13 @@ void get_exx_potential(const K_Vectors* kv, } // copy the potential to the device memory + Real* pot_cpu_data = pot_cpu.data(); #ifdef __CUDA - if (PARAM.inp.device == "gpu") - { - cudaError_t err = cudaHostRegister(pot_cpu, sizeof(Real) * npw, cudaHostRegisterPortable); - if (err != cudaSuccess) { - throw std::runtime_error("failed to register potential CPU memory operations"); - } - } -#endif - syncmem_real_c2d_op()(pot, pot_cpu, rhopw_dev->npw); -#ifdef __CUDA - if (PARAM.inp.device == "gpu") - { - cudaHostUnregister(pot_cpu); - } + const bool use_gpu = PARAM.inp.device == "gpu"; + const std::size_t bytes = sizeof(Real) * npw; + CudaHostRegistration registration(pot_cpu_data, bytes, use_gpu); #endif - - delete[] pot_cpu; + syncmem_real_c2d_op()(pot, pot_cpu_data, npw); } template @@ -237,19 +263,16 @@ void get_exx_stress_potential(const K_Vectors* kv, int iq, const CoulombParam& coulomb_param_in) { - using setmem_real_cpu_op = base_device::memory::set_memory_op; using syncmem_real_c2d_op = base_device::memory::synchronize_memory_op; Real nqs_half1 = 0.5 * kv->nmp[0]; Real nqs_half2 = 0.5 * kv->nmp[1]; Real nqs_half3 = 0.5 * kv->nmp[2]; - Real* pot_cpu = nullptr; - int nks = wfcpw->nks, npw = rhopw_dev->npw; + int nks = wfcpw->nks; + int npw = rhopw_dev->npw; double tpiba2 = tpiba * tpiba; - pot_cpu = new Real[npw]; - // fill zero - setmem_real_cpu_op()(pot_cpu, 0, npw); + std::vector pot_cpu(npw, 0); // calculate Fock pot auto it_fock = coulomb_param_in.find(Conv_Coulomb_Pot_K::Coulomb_Type::Fock); @@ -395,24 +418,13 @@ void get_exx_stress_potential(const K_Vectors* kv, } // copy the potential to the device memory + Real* pot_cpu_data = pot_cpu.data(); #ifdef __CUDA - if (PARAM.inp.device == "gpu") - { - cudaError_t err = cudaHostRegister(pot_cpu, sizeof(Real) * npw, cudaHostRegisterPortable); - if (err != cudaSuccess) { - throw std::runtime_error("failed to register potential CPU memory operations"); - } - } -#endif - syncmem_real_c2d_op()(pot, pot_cpu, rhopw_dev->npw); -#ifdef __CUDA - if (PARAM.inp.device == "gpu") - { - cudaHostUnregister(pot_cpu); - } + const bool use_gpu = PARAM.inp.device == "gpu"; + const std::size_t bytes = sizeof(Real) * npw; + CudaHostRegistration registration(pot_cpu_data, bytes, use_gpu); #endif - - delete[] pot_cpu; + syncmem_real_c2d_op()(pot, pot_cpu_data, npw); } double exx_divergence(Conv_Coulomb_Pot_K::Coulomb_Type coulomb_type, diff --git a/source/source_pw/module_pwdft/op_pw_meta.h b/source/source_pw/module_pwdft/op_pw_meta.h index bb9ffbc21b3..533eba97d1e 100644 --- a/source/source_pw/module_pwdft/op_pw_meta.h +++ b/source/source_pw/module_pwdft/op_pw_meta.h @@ -53,6 +53,8 @@ class Meta> : public OperatorPW } const int * get_isk() const {return this->isk;} const Real* get_vk() const {return this->vk;} + /** @brief Select the borrowed endpoint or midpoint kinetic-energy-density potential. */ + void set_vk(const Real* potential) { this->vk = potential; } int get_vk_row() const {return this->vk_row;} int get_vk_col() const {return this->vk_col;} const ModulePW::PW_Basis_K* get_wfcpw() const diff --git a/source/source_pw/module_pwdft/op_pw_nl.cpp b/source/source_pw/module_pwdft/op_pw_nl.cpp index d3551808ea7..97f08a306e2 100644 --- a/source/source_pw/module_pwdft/op_pw_nl.cpp +++ b/source/source_pw/module_pwdft/op_pw_nl.cpp @@ -1,20 +1,18 @@ #include "op_pw_nl.h" -#include "source_io/module_parameter/parameter.h" #include "source_base/timer.h" -#include "source_base/parallel_reduce.h" #include "source_base/tool_quit.h" +namespace hamilt +{ -namespace hamilt { - -template +template Nonlocal>::Nonlocal(const int* isk_in, - const pseudopot_cell_vnl* ppcell_in, - const UnitCell* ucell_in, - const ModulePW::PW_Basis_K* wfc_basis) + const pseudopot_cell_vnl* ppcell_in, + const UnitCell* ucell_in, + const ModulePW::PW_Basis_K* wfc_basis) { - if( isk_in == nullptr || ppcell_in == nullptr || ucell_in == nullptr) + if (isk_in == nullptr || ppcell_in == nullptr || ucell_in == nullptr || wfc_basis == nullptr) { ModuleBase::WARNING_QUIT("NonlocalPW", "Constuctor of Operator::NonlocalPW is failed, please check your code!"); } @@ -24,278 +22,59 @@ Nonlocal>::Nonlocal(const int* isk_in, this->isk = isk_in; this->ppcell = ppcell_in; this->ucell = ucell_in; - this->deeq = this->ppcell->template get_deeq_data(); - this->deeq_nc = this->ppcell->template get_deeq_nc_data(); this->vkb = this->ppcell->template get_vkb_data(); - } -template -Nonlocal>::~Nonlocal() { - delmem_complex_op()(this->ps); - delmem_complex_op()(this->becp); +template +Nonlocal>::~Nonlocal() +{ } -template +template void Nonlocal>::init(const int ik_in) { - ModuleBase::timer::start("Nonlocal", "getvnl"); + ModuleBase::timer::start("Nonlocal", "init"); this->ik = ik_in; // Calculate nonlocal pseudopotential vkb - if(this->ppcell->nkb > 0) //xiaohui add 2013-09-02. Attention... - { - this->ppcell->getvnl(this->ctx, *this->ucell, this->ik, this->vkb); - } - - if(this->next_op != nullptr) + if (this->ppcell->nkb > 0) // xiaohui add 2013-09-02. Attention... { - this->next_op->init(ik_in); + this->ppcell->getvnl(this->ctx, *this->ucell, this->ik, ModuleBase::Vector3(0.0, 0.0, 0.0), this->vkb); } - ModuleBase::timer::end("Nonlocal", "getvnl"); -} - -//-------------------------------------------------------------------------- -// this function sum up each non-local pseudopotential located on each atom, -//-------------------------------------------------------------------------- -template -void Nonlocal>::add_nonlocal_pp(T *hpsi_in, const T *becp, const int m) const -{ - ModuleBase::timer::start("Nonlocal", "add_nonlocal_pp"); - - // number of projectors - int nkb = this->ppcell->nkb; - - // T *ps = new T[nkb * m]; - // ModuleBase::GlobalFunc::ZEROS(ps, m * nkb); - if (this->nkb_m < m * nkb) { - resmem_complex_op()(this->ps, nkb * m, "Nonlocal::ps"); - this->nkb_m = m * nkb; - } - setmem_complex_op()(this->ps, 0, nkb * m); - - int sum = 0; - int iat = 0; - if (this->npol == 1) - { - const int current_spin = this->isk[this->ik]; - for (int it = 0; it < this->ucell->ntype; it++) - { - const int nproj = this->ucell->atoms[it].ncpp.nh; - // denghui replace 2022-10-20 - // <<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<< - nonlocal_op()( - this->ctx, // device context - this->ucell->atoms[it].na, m, nproj, // four loop size - sum, iat, current_spin, nkb, // additional index params - this->ppcell->deeq.getBound2(), this->ppcell->deeq.getBound3(), this->ppcell->deeq.getBound4(), // realArray operator() - this->deeq, // array of data - this->ps, this->becp); // array of data - // <<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<< - // for (int ia = 0; ia < this->ucell->atoms[it].na; ia++) - // { - // // each atom has nproj, means this is with structure factor; - // // each projector (each atom) must multiply coefficient - // // with all the other projectors. - // for (int ib = 0; ib < m; ++ib) - // { - // for (int ip2 = 0; ip2 < nproj; ip2++) - // { - // for (int ip = 0; ip < nproj; ip++) - // { - // this->ps[(sum + ip2) * m + ib] - // += this->ppcell->deeq(current_spin, iat, ip, ip2) * this->becp[ib * nkb + sum + ip]; - // } // end ib - // } // end ih - // } // end jh - // sum += nproj; - // ++iat; - // } // end na - } // end nt - } - else + if (this->next_op != nullptr) { - for (int it = 0; it < this->ucell->ntype; it++) - { - const int nproj = this->ucell->atoms[it].ncpp.nh; - // added by denghui at 20221109 - // <<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<< - nonlocal_op()( - this->ctx, // device context - this->ucell->atoms[it].na, m, nproj, // four loop size - sum, iat, nkb, // additional index params - this->ppcell->deeq_nc.getBound2(), this->ppcell->deeq_nc.getBound3(), this->ppcell->deeq_nc.getBound4(), // realArray operator() - this->deeq_nc, // array of data - this->ps, this->becp); // array of data - // >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>> - // for (int ia = 0; ia < this->ucell->atoms[it].na; ia++) - // { - // // each atom has nproj, means this is with structure factor; - // // each projector (each atom) must multiply coefficient - // // with all the other projectors. - // for (int ib = 0; ib < m; ib+=2) - // { - // for (int ip2 = 0; ip2 < nproj; ip2++) - // { - // for (int ip = 0; ip < nproj; ip++) - // { - // psind = (sum + ip2) * m + ib; - // becpind = ib * nkb + sum + ip; - // becp1 = becp[becpind]; - // becp2 = becp[becpind + nkb]; - // ps[psind] += this->ppcell->deeq_nc(0, iat, ip2, ip) * becp1 - // + this->ppcell->deeq_nc(1, iat, ip2, ip) * becp2; - // ps[psind + 1] += this->ppcell->deeq_nc(2, iat, ip2, ip) * becp1 - // + this->ppcell->deeq_nc(3, iat, ip2, ip) * becp2; - // } // end ib - // } // end ih - // } // end jh - // sum += nproj; - // ++iat; - // } // end na - } // end nt + this->next_op->init(ik_in); } - // use simple method. - //<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<< - // qianrui optimize 2021-3-31 - char transa = 'N'; - char transb = 'T'; - if (m == 1) - { - int inc = 1; - // denghui replace 2022-10-20 - // <<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<< - gemv_op()( - transa, - this->npw, - this->ppcell->nkb, - &this->one, - this->vkb, - this->ppcell->vkbnc, - this->ps, - inc, - &this->one, - hpsi_in, - inc); - } - else - { - int npm = m; - //<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<< - // denghui replace 2022-10-20 - #ifdef __DSP - ModuleBase::gemm_op_mt() - #else - gemm_op() - #endif - ( - transa, - transb, - this->npw, - npm, - this->ppcell->nkb, - &this->one, - this->vkb, - this->ppcell->vkbnc, - this->ps, - npm, - &this->one, - hpsi_in, - this->max_npw - ); - } - ModuleBase::timer::end("Nonlocal", "add_nonlocal_pp"); + ModuleBase::timer::end("Nonlocal", "init"); } -template -void Nonlocal>::act( - const int nbands, - const int nbasis, - const int npol, - const T* tmpsi_in, - T* tmhpsi, - const int ngk_ik, - const bool is_first_node)const +template +void Nonlocal>::act(const int nbands, + const int nbasis, + const int npol, + const T* tmpsi_in, + T* tmhpsi, + const int ngk_ik, + const bool is_first_node) const { - ModuleBase::timer::start("Operator", "nonlocal_pw"); - if(is_first_node) - { - setmem_complex_op()(tmhpsi, 0, nbasis*nbands/npol); - } - - this->npw = ngk_ik; - this->max_npw = nbasis / npol; - this->npol = npol; - - if (this->ppcell->nkb > 0) - { - //<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<< - // qianrui optimize 2021-3-31 - int nkb = this->ppcell->nkb; - if (this->nkb_m < nbands * nkb) - { - resmem_complex_op()(this->becp, nbands * nkb, "Nonlocal::becp"); - } - // ModuleBase::ComplexMatrix becp(nbands, nkb, false); - char transa = 'C'; - char transb = 'N'; - if (nbands == 1) - { - int inc = 1; - // denghui replace 2022-10-20 - // <<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<< - gemv_op()( - transa, - this->npw, - nkb, - &this->one, - this->vkb, - this->ppcell->vkbnc, - tmpsi_in, - inc, - &this->zero, - this->becp, - inc); - } - else - { - int npm = nbands; - //<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<< - // denghui replace 2022-10-20 - #ifdef __DSP - ModuleBase::gemm_op_mt() - #else - gemm_op() - #endif - ( - transa, - transb, - nkb, - npm, - this->npw, - &this->one, - this->vkb, - this->ppcell->vkbnc, - tmpsi_in, - max_npw, - &this->zero, - this->becp, - nkb - ); - } - - Parallel_Reduce::reduce_pool(becp, nkb * nbands); - - this->add_nonlocal_pp(tmhpsi, becp, nbands); - } - - ModuleBase::timer::end("Operator", "nonlocal_pw"); + this->workspace_.apply(*this->ucell, + *this->ppcell, + *this->wfcpw, + this->isk[this->ik], + nbands, + nbasis, + npol, + ngk_ik, + is_first_node, + this->vkb, + tmpsi_in, + tmhpsi); } -template -template -hamilt::Nonlocal>::Nonlocal(const Nonlocal> *nonlocal) +template +template +hamilt::Nonlocal>::Nonlocal(const Nonlocal>* nonlocal) { this->classname = "Nonlocal"; this->cal_type = calculation_type::pw_nonlocal; @@ -303,10 +82,9 @@ hamilt::Nonlocal>::Nonlocal(const Nonlocalisk = nonlocal->get_isk(); this->ppcell = nonlocal->get_ppcell(); this->ucell = nonlocal->get_ucell(); - this->deeq = this->ppcell->d_deeq; - this->deeq_nc = this->ppcell->template get_deeq_nc_data(); + this->wfcpw = nonlocal->get_wfcpw(); this->vkb = this->ppcell->template get_vkb_data(); - if( this->isk == nullptr || this->ppcell == nullptr || this->ucell == nullptr) + if (this->isk == nullptr || this->ppcell == nullptr || this->ucell == nullptr || this->wfcpw == nullptr) { ModuleBase::WARNING_QUIT("NonlocalPW", "Constuctor of Operator::NonlocalPW is failed, please check your code!"); } diff --git a/source/source_pw/module_pwdft/op_pw_nl.h b/source/source_pw/module_pwdft/op_pw_nl.h index dcdbf889a8c..835f389c5f4 100644 --- a/source/source_pw/module_pwdft/op_pw_nl.h +++ b/source/source_pw/module_pwdft/op_pw_nl.h @@ -4,8 +4,7 @@ #include "op_pw.h" #include "source_cell/unitcell.h" -#include "source_pw/module_pwdft/kernels/nonlocal_op.h" -#include "source_base/kernels/math_kernel_op.h" +#include "source_pw/module_pwdft/nonlocal_workspace.h" #include "source_pw/module_pwdft/vnl_pw.h" @@ -49,25 +48,19 @@ class Nonlocal> : public OperatorPW const int *get_isk() const {return this->isk;} const pseudopot_cell_vnl *get_ppcell() const {return this->ppcell;} const UnitCell *get_ucell() const {return this->ucell;} + /** @brief Return the borrowed plane-wave basis and pool communicator. */ + const ModulePW::PW_Basis_K* get_wfcpw() const { return this->wfcpw; } T* get_vkb() const { return this->vkb; } T* get_becp() const { - return this->becp; + return this->workspace_.get_becp(); } private: - void add_nonlocal_pp(T *hpsi_in, const T *becp, const int m) const; - - mutable int max_npw = 0; - - mutable int npw = 0; - - mutable int npol = 0; - - mutable size_t nkb_m = 0; + mutable NonlocalWorkspace workspace_; const int* isk = nullptr; @@ -77,30 +70,9 @@ class Nonlocal> : public OperatorPW const ModulePW::PW_Basis_K* wfcpw = nullptr; - mutable T *ps = nullptr; mutable T *vkb = nullptr; - mutable T *becp = nullptr; Device* ctx = {}; - base_device::DEVICE_CPU* cpu_ctx = {}; - Real * deeq = nullptr; - T * deeq_nc = nullptr; - // using nonlocal_op = nonlocal_pw_op; - using gemv_op = ModuleBase::gemv_op; - using gemm_op = ModuleBase::gemm_op; - using nonlocal_op = nonlocal_pw_op; -#ifdef __DSP - using setmem_complex_op = base_device::memory::set_memory_op_mt; - using resmem_complex_op = base_device::memory::resize_memory_op_mt; - using delmem_complex_op = base_device::memory::delete_memory_op_mt; -#else - using setmem_complex_op = base_device::memory::set_memory_op; - using resmem_complex_op = base_device::memory::resize_memory_op; - using delmem_complex_op = base_device::memory::delete_memory_op; -#endif - using syncmem_complex_h2d_op = base_device::memory::synchronize_memory_op; - T one{1, 0}; - T zero{0, 0}; }; } // namespace hamilt diff --git a/source/source_pw/module_pwdft/op_pw_nl_td.cpp b/source/source_pw/module_pwdft/op_pw_nl_td.cpp new file mode 100644 index 00000000000..a2856e3b68a --- /dev/null +++ b/source/source_pw/module_pwdft/op_pw_nl_td.cpp @@ -0,0 +1,100 @@ +#include "op_pw_nl_td.h" + +#include "source_base/timer.h" +#include "source_base/tool_quit.h" + +#include + +namespace hamilt +{ + +template +TDNonlocalPW::TDNonlocalPW(const int* isk_in, + const pseudopot_cell_vnl* ppcell_in, + const UnitCell* ucell_in, + const ModulePW::PW_Basis_K* wfc_basis) +{ + if (isk_in == nullptr || ppcell_in == nullptr || ucell_in == nullptr || wfc_basis == nullptr) + { + ModuleBase::WARNING_QUIT("TDNonlocalPW", "Constructor failed, null pointers detected!"); + } + + this->classname = "TDNonlocalPW"; + this->cal_type = calculation_type::pw_nonlocal; + this->isk = isk_in; + this->ppcell = ppcell_in; + this->ucell = ucell_in; + this->wfcpw = wfc_basis; +} + +template +TDNonlocalPW::~TDNonlocalPW() +{ + delmem_complex_op()(this->vkb_td); +} + +template +void TDNonlocalPW::init(const int ik_in) +{ + ModuleBase::timer::start("TDNonlocalPW", "init"); + this->ik = ik_in; + + // Refresh the shifted projectors when nonlocal projectors are present. + if (this->ppcell->nkb > 0 && this->wfcpw->npwk[ik_in] > 0) + { + // Allocate the time-dependent projector cache on first use. + if (this->vkb_td == nullptr) + { + const std::int64_t projector_elements = static_cast(this->ppcell->nkb) * this->wfcpw->npwk_max; + resmem_complex_op()(this->vkb_td, projector_elements, "TDNL::vkb_td"); + } + + // Hartree-unit vector potential. + ModuleBase::Vector3 A_au = A_ha_; + + // Generate projectors shifted by the current vector potential. + this->ppcell->getvnl(this->ctx, *this->ucell, this->ik, A_au, this->vkb_td); + } + + if (this->next_op != nullptr) + { + this->next_op->init(ik_in); + } + ModuleBase::timer::end("TDNonlocalPW", "init"); +} + +template +void TDNonlocalPW::act(const int nbands, + const int nbasis, + const int npol, + const T* tmpsi_in, + T* tmhpsi, + const int ngk_ik, + const bool is_first_node) const +{ + this->workspace_.apply(*this->ucell, + *this->ppcell, + *this->wfcpw, + this->isk[this->ik], + nbands, + nbasis, + npol, + ngk_ik, + is_first_node, + this->vkb_td, + tmpsi_in, + tmhpsi); +} + +// Explicit CPU template instantiations. +template class TDNonlocalPW, base_device::DEVICE_CPU>; +template class TDNonlocalPW, base_device::DEVICE_CPU>; + +// ================= GPU explicit instantiations ================= +#if ((defined __CUDA) || (defined __ROCM)) +template class TDNonlocalPW, base_device::DEVICE_GPU>; +template class TDNonlocalPW, base_device::DEVICE_GPU>; +#endif +// =============================================================== + +} // namespace hamilt diff --git a/source/source_pw/module_pwdft/op_pw_nl_td.h b/source/source_pw/module_pwdft/op_pw_nl_td.h new file mode 100644 index 00000000000..ad3756c6057 --- /dev/null +++ b/source/source_pw/module_pwdft/op_pw_nl_td.h @@ -0,0 +1,50 @@ +#ifndef OP_PW_NL_TD_H +#define OP_PW_NL_TD_H + +#include "op_pw.h" +#include "source_pw/module_pwdft/nonlocal_workspace.h" +#include "source_pw/module_pwdft/vnl_pw.h" + +namespace hamilt +{ + +template +class TDNonlocalPW : public OperatorPW +{ + public: + /** @brief Bind the vector potential in Hartree atomic units before init(). */ + void set_A_ha(const ModuleBase::Vector3& A) + { + A_ha_ = A; + } + TDNonlocalPW(const int* isk_in, const pseudopot_cell_vnl* ppcell_in, const UnitCell* ucell_in, const ModulePW::PW_Basis_K* wfc_basis); + + virtual ~TDNonlocalPW(); + virtual void init(const int ik_in) override; + + virtual void act(const int nbands, + const int nbasis, + const int npol, + const T* tmpsi_in, + T* tmhpsi, + const int ngk_ik = 0, + const bool is_first_node = false) const override; + + private: + mutable NonlocalWorkspace workspace_; + + const int* isk = nullptr; + const pseudopot_cell_vnl* ppcell = nullptr; + const UnitCell* ucell = nullptr; + const ModulePW::PW_Basis_K* wfcpw = nullptr; + + mutable T* vkb_td = nullptr; // Time-dependent projector cache. + + ModuleBase::Vector3 A_ha_; + Device* ctx = {}; + using resmem_complex_op = base_device::memory::resize_memory_op; + using delmem_complex_op = base_device::memory::delete_memory_op; +}; + +} // namespace hamilt +#endif diff --git a/source/source_pw/module_pwdft/op_pw_proj.cpp b/source/source_pw/module_pwdft/op_pw_proj.cpp index d783e34db69..fd469e7ce53 100644 --- a/source/source_pw/module_pwdft/op_pw_proj.cpp +++ b/source/source_pw/module_pwdft/op_pw_proj.cpp @@ -5,7 +5,7 @@ #include "source_base/tool_quit.h" #include "source_lcao/module_deltaspin/spin_constrain.h" #include "source_pw/module_pwdft/dftu_base.h" -#include "source_pw/module_pwdft/onsite_proj.h" +#include "source_pw/module_proj/onsite_proj.h" #include "source_pw/module_pwdft/kernels/onsite_op.h" @@ -13,10 +13,10 @@ namespace hamilt { template OnsiteProj>::OnsiteProj(const int* isk_in, - const UnitCell* ucell_in, - Plus_U_Base* p_dftu, - const bool cal_delta_spin, - const bool cal_dftu) + const UnitCell* ucell_in, + Plus_U_Base* p_dftu, + const bool cal_delta_spin, + const bool cal_dftu) { this->classname = "OnsiteProj"; this->cal_type = calculation_type::pw_onsite; @@ -265,13 +265,13 @@ void OnsiteProj>::setup_pw_dftu_indices() const template void OnsiteProj>::cal_ps_dftu( - const int npol, - const int m) const + const int npol, + const int m) const { - if(!this->has_dftu) - { - return; - } + if(!this->has_dftu) + { + return; + } auto* onsite_p = projectors::OnsiteProjector::get_instance(); const std::complex* becp = onsite_p->get_becp(); @@ -309,59 +309,59 @@ void OnsiteProj>::cal_ps_dftu( template<> void OnsiteProj, base_device::DEVICE_CPU>>::add_onsite_proj( - std::complex *hpsi_in, - const int npol, - const int m, - const int npwx) const + std::complex *hpsi_in, + const int npol, + const int m, + const int npwx) const {} template<> void OnsiteProj, base_device::DEVICE_CPU>>::update_becp( - const std::complex *psi_in, - const int npol, - const int m, - const int npwx) const + const std::complex *psi_in, + const int npol, + const int m, + const int npwx) const {} template<> void OnsiteProj, base_device::DEVICE_CPU>>::cal_ps_delta_spin( - const int npol, - const int m) const + const int npol, + const int m) const {} template<> void OnsiteProj, base_device::DEVICE_CPU>>::cal_ps_dftu( - const int npol, - const int m) const + const int npol, + const int m) const {} #if ((defined __CUDA) || (defined __ROCM)) template<> void OnsiteProj, base_device::DEVICE_GPU>>::add_onsite_proj( - std::complex *hpsi_in, - const int npol, - const int m, - const int npwx) const + std::complex *hpsi_in, + const int npol, + const int m, + const int npwx) const {} template<> void OnsiteProj, base_device::DEVICE_GPU>>::update_becp( - const std::complex *psi_in, - const int npol, - const int m, - const int npwx) const + const std::complex *psi_in, + const int npol, + const int m, + const int npwx) const {} template<> void OnsiteProj, base_device::DEVICE_GPU>>::cal_ps_delta_spin( - const int npol, - const int m) const + const int npol, + const int m) const {} template<> void OnsiteProj, base_device::DEVICE_GPU>>::cal_ps_dftu( - const int npol, - const int m) const + const int npol, + const int m) const {} #endif diff --git a/source/source_pw/module_pwdft/op_pw_proj.h b/source/source_pw/module_pwdft/op_pw_proj.h index 4cece482992..9a07a561ac0 100644 --- a/source/source_pw/module_pwdft/op_pw_proj.h +++ b/source/source_pw/module_pwdft/op_pw_proj.h @@ -43,8 +43,8 @@ class OnsiteProj> : public OperatorPW const int npol, const T* tmpsi_in, T* tmhpsi, - const int ngk = 0, - const bool is_first_node = false)const override; + const int ngk = 0, + const bool is_first_node = false)const override; const int *get_isk() const {return this->isk;} const UnitCell *get_ucell() const {return this->ucell;} diff --git a/source/source_pw/module_pwdft/op_pw_veff.h b/source/source_pw/module_pwdft/op_pw_veff.h index fa7fe40872c..52b90dff51e 100644 --- a/source/source_pw/module_pwdft/op_pw_veff.h +++ b/source/source_pw/module_pwdft/op_pw_veff.h @@ -54,6 +54,9 @@ class Veff> : public OperatorPW return this->wfcpw; } + // Override the effective potential for the TDDFT midpoint Hamiltonian. + void set_veff(const Real* custom_veff) { this->veff = custom_veff; } + private: const int* isk = nullptr; @@ -78,4 +81,4 @@ class Veff> : public OperatorPW } // namespace hamilt -#endif \ No newline at end of file +#endif diff --git a/source/source_pw/module_pwdft/op_pw_vel.cpp b/source/source_pw/module_pwdft/op_pw_vel.cpp index 6d003220ddf..4038442bbee 100644 --- a/source/source_pw/module_pwdft/op_pw_vel.cpp +++ b/source/source_pw/module_pwdft/op_pw_vel.cpp @@ -33,7 +33,7 @@ Velocity::Velocity(const ModulePW::PW_Basis_K* wfcpw_in, this->vtau_row_ = vtau_row_in; if (this->nonlocal) { - this->ppcell->initgradq_vnl(*this->ucell); + this->ppcell->ensure_grad_table(*this->ucell); } } @@ -52,43 +52,57 @@ Velocity::~Velocity() template void Velocity::init(const int ik_in) { + this->init(ik_in, ModuleBase::Vector3(0.0, 0.0, 0.0)); +} + +template +void Velocity::init(const int ik_in, const ModuleBase::Vector3& vector_potential) +{ + ModuleBase::timer::start("Velocity", "init"); this->ik = ik_in; + this->tpiba = this->ucell->tpiba; // init G+K const int npw = this->wfcpw->npwk[ik_in]; const int npwk_max = this->wfcpw->npwk_max; std::vector gtmp(npw); - resmem_var_op()(gx_, npw); - resmem_var_op()(gy_, npw); - resmem_var_op()(gz_, npw); + if (npw > momentum_capacity_) + { + resmem_var_op()(gx_, npw); + resmem_var_op()(gy_, npw); + resmem_var_op()(gz_, npw); + momentum_capacity_ = npw; + } std::vector gtmp_ptr = {this->gx_, this->gy_, this->gz_}; - for(int i=0; i<3; ++i) + for (int i = 0; i < 3; ++i) { for (int ig = 0; ig < npw; ++ig) { const ModuleBase::Vector3 tmpg = wfcpw->getgpluskcar(this->ik, ig); - gtmp[ig] = static_cast(tmpg[i] * tpiba); + gtmp[ig] = static_cast(tmpg[i] * tpiba + vector_potential[i]); + } + if (npw > 0) + { + syncmem_var_h2d_op()(gtmp_ptr[i], gtmp.data(), npw); } - syncmem_var_h2d_op()(gtmp_ptr[i], gtmp.data(), npw); } // Calculate nonlocal pseudopotential vkb - if (this->ppcell->nkb > 0 && this->nonlocal) + if (npw > 0 && this->ppcell->nkb > 0 && this->nonlocal) { - this->ppcell->getgradq_vnl(*this->ucell, ik_in); - // sync to device - if (std::is_same::value || std::is_same::value) + const int nkb = this->ppcell->nkb; + const std::int64_t projector_elements = static_cast(nkb) * npwk_max; + if (projector_elements > projector_capacity_) { - const int nkb = this->ppcell->nkb; - // vkb - resmem_complex_op()(vkb_, nkb * npwk_max); - castmem_complex_h2d_op()(vkb_, this->ppcell->vkb.c, nkb * npwk_max); - - // gradvkb - resmem_complex_op()(gradvkb_, 3 * nkb * npwk_max); - castmem_complex_h2d_op()(gradvkb_, this->ppcell->gradvkb.ptr, 3 * nkb * npwk_max); + const std::int64_t gradient_elements = 3 * projector_elements; + resmem_complex_op()(vkb_, projector_elements); + resmem_complex_op()(gradvkb_, gradient_elements); + projector_capacity_ = projector_elements; } + this->ppcell->getvnl(this->ctx, *this->ucell, ik_in, vector_potential, vkb_); + this->gradient_.calculate(this->ppcell, *this->ucell, *this->wfcpw, ik_in, vector_potential, gradvkb_); } + ModuleBase::timer::end("Velocity", "init"); } template @@ -102,9 +116,10 @@ void Velocity::act(const psi::Psi, Device>* const int npw = this->wfcpw->npwk[this->ik]; const int max_npw = this->wfcpw->npwk_max; + const std::int64_t component_elements = static_cast(n_npwx) * max_npw; const int npol = psi_in->get_npol(); using Real = typename GetTypeReal::type; - + std::vector gtmp_ptr = {this->gx_, this->gy_, this->gz_}; // ------------- // p @@ -112,8 +127,8 @@ void Velocity::act(const psi::Psi, Device>* for (int id = 0; id < 3; ++id) { const Complex* tmpsi_in = psi0; - Complex* tmpvpsi = vpsi + id * n_npwx * max_npw; - for (int ib = 0; ib < n_npwx; ++ib) + Complex* tmpvpsi = vpsi + id * component_elements; + for (int ib = 0; npw > 0 && ib < n_npwx; ++ib) { ModuleBase::vector_mul_vector_op()(npw, tmpvpsi, tmpsi_in, gtmp_ptr[id], add); tmpvpsi += max_npw; @@ -131,15 +146,18 @@ void Velocity::act(const psi::Psi, Device>* // In real space this is implemented as // -i/2 [\partial_\alpha(v_tau psi) + v_tau \partial_\alpha psi]. // --------------------------------------------- - if (this->vtau_ != nullptr && this->vtau_col_ > 0 && XC_Functional::get_ked_flag()) + // The row count also enables the correction on empty real-space partitions. + if (this->vtau_row_ > 0 && XC_Functional::get_ked_flag()) { - if (this->porter1_ == nullptr) + if (this->vtau_col_ != this->wfcpw->nrxx || (this->vtau_col_ > 0 && this->vtau_ == nullptr)) { - resmem_complex_op()(this->porter1_, this->wfcpw->nmaxgr); + ModuleBase::WARNING_QUIT("Velocity", "Invalid local potential for meta-GGA velocity correction."); } - if (this->porter2_ == nullptr) + if (this->wfcpw->nmaxgr > porter_capacity_) { + resmem_complex_op()(this->porter1_, this->wfcpw->nmaxgr); resmem_complex_op()(this->porter2_, this->wfcpw->nmaxgr); + porter_capacity_ = this->wfcpw->nmaxgr; } int current_spin = 0; if (this->vtau_row_ > 1) @@ -150,58 +168,72 @@ void Velocity::act(const psi::Psi, Device>* ModuleBase::WARNING_QUIT("Velocity", "invalid spin index for meta-GGA velocity correction"); } } - const Real* vtau_spin = this->vtau_ + current_spin * this->vtau_col_; + const std::int64_t spin_offset = static_cast(current_spin) * this->vtau_col_; + const Real* vtau_spin = this->vtau_col_ > 0 ? this->vtau_ + spin_offset : nullptr; Complex minus_half_i(0.0, -0.5); for (int ib = 0; ib < n_npwx; ++ib) { - const Complex* bandpsi = psi0 + ib * max_npw; + const std::int64_t band_offset = static_cast(ib) * max_npw; + const Complex* bandpsi = psi0 + band_offset; this->wfcpw->recip_to_real(this->ctx, bandpsi, this->porter1_, this->ik); - ModuleBase::vector_mul_vector_op()(this->vtau_col_, - this->porter1_, - this->porter1_, - vtau_spin, - false); + if (this->vtau_col_ > 0) + { + ModuleBase::vector_mul_vector_op()(this->vtau_col_, + this->porter1_, + this->porter1_, + vtau_spin, + false); + } this->wfcpw->real_to_recip(this->ctx, this->porter1_, this->porter1_, this->ik); for (int id = 0; id < 3; ++id) { - // term1: partial_id (v_tau * psi) - meta_pw_op()(this->ctx, - this->ik, - id, - npw, - max_npw, - this->tpiba, - this->wfcpw->template get_gcar_data(), - this->wfcpw->template get_kvec_c_data(), - this->porter1_, - this->porter2_, - false); - ModuleBase::scal_op()(npw, &minus_half_i, this->porter2_, 1); - Complex* vpsi_slice = vpsi + id * n_npwx * max_npw + ib * max_npw; + Complex* vpsi_slice = vpsi + id * component_elements + band_offset; Complex one = 1.0; - ModuleBase::axpy_op()(npw, &one, this->porter2_, 1, vpsi_slice, 1); + // term1: partial_id (v_tau * psi) + if (npw > 0) + { + meta_pw_op()(this->ctx, + this->ik, + id, + npw, + max_npw, + this->tpiba, + this->wfcpw->template get_gcar_data(), + this->wfcpw->template get_kvec_c_data(), + this->porter1_, + this->porter2_, + false); + ModuleBase::scal_op()(npw, &minus_half_i, this->porter2_, 1); + ModuleBase::axpy_op()(npw, &one, this->porter2_, 1, vpsi_slice, 1); - // term2: v_tau * partial_id psi - meta_pw_op()(this->ctx, - this->ik, - id, - npw, - max_npw, - this->tpiba, - this->wfcpw->template get_gcar_data(), - this->wfcpw->template get_kvec_c_data(), - bandpsi, - this->porter2_, - false); + // term2: v_tau * partial_id psi + meta_pw_op()(this->ctx, + this->ik, + id, + npw, + max_npw, + this->tpiba, + this->wfcpw->template get_gcar_data(), + this->wfcpw->template get_kvec_c_data(), + bandpsi, + this->porter2_, + false); + } this->wfcpw->recip_to_real(this->ctx, this->porter2_, this->porter2_, this->ik); - ModuleBase::vector_mul_vector_op()(this->vtau_col_, - this->porter2_, - this->porter2_, - vtau_spin, - false); + if (this->vtau_col_ > 0) + { + ModuleBase::vector_mul_vector_op()(this->vtau_col_, + this->porter2_, + this->porter2_, + vtau_spin, + false); + } this->wfcpw->real_to_recip(this->ctx, this->porter2_, this->porter2_, this->ik); - ModuleBase::scal_op()(npw, &minus_half_i, this->porter2_, 1); - ModuleBase::axpy_op()(npw, &one, this->porter2_, 1, vpsi_slice, 1); + if (npw > 0) + { + ModuleBase::scal_op()(npw, &minus_half_i, this->porter2_, 1); + ModuleBase::axpy_op()(npw, &one, this->porter2_, 1, vpsi_slice, 1); + } } } } @@ -217,29 +249,27 @@ void Velocity::act(const psi::Psi, Device>* } // 1. <\beta|\psi> - Complex* becp1_ = nullptr; ///<[Device, n_npwx * nkb] <\beta|\psi> - Complex* becp2_ = nullptr; ///<[Device, n_npwx * 3*nkb] <\nabla\beta|\psi> - Complex* ps1_ = nullptr; ///<[Device, nkb * n_npwx] sum of becp1 - Complex* ps2_ = nullptr; ///<[Device, 3*nkb * n_npwx] sum of becp2 - resmem_complex_op()(ps1_, this->ppcell->nkb * n_npwx); - resmem_complex_op()(ps2_, 3 * this->ppcell->nkb * n_npwx); - resmem_complex_op()(becp1_, this->ppcell->nkb * n_npwx); - resmem_complex_op()(becp2_, 3 * this->ppcell->nkb * n_npwx); + const std::int64_t block = static_cast(this->ppcell->nkb) * n_npwx; + Complex* becp1_ = this->contraction_.prepare(block); + Complex* becp2_ = becp1_ + block; + Complex* ps1_ = becp1_ + 4 * block; + Complex* ps2_ = ps1_ + block; const int nkb = this->ppcell->nkb; const int nkb3 = 3 * nkb; Complex one = 1.0; Complex zero = 0.0; - Complex* vkb_d = reinterpret_cast(this->ppcell->vkb.c); - Complex* gradvkb_d = reinterpret_cast(this->ppcell->gradvkb.ptr); - if (std::is_same::value || std::is_same::value) + const Complex* vkb_d = vkb_; + const Complex* gradvkb_d = gradvkb_; + + if (npw == 0) { - vkb_d = vkb_; - gradvkb_d = gradvkb_; + // Keep the reduction collective even when this rank has no local plane waves. + const std::int64_t projection_elements = 4 * block; + base_device::memory::set_memory_op()(becp1_, 0, projection_elements); } - - if (n_npwx == 1) + else if (n_npwx == 1) { int inc = 1; ModuleBase::gemv_op()('C', npw, nkb, &one, vkb_d, max_npw, psi0, inc, &zero, becp1_, inc); @@ -247,105 +277,21 @@ void Velocity::act(const psi::Psi, Device>* } else { - ModuleBase::gemm_op()('C', - 'N', - nkb, - n_npwx, - npw, - &one, - vkb_d, - max_npw, - psi0, - max_npw, - &zero, - becp1_, - nkb); - ModuleBase::gemm_op()('C', - 'N', - nkb3, - n_npwx, - npw, - &one, - gradvkb_d, - max_npw, - psi0, - max_npw, - &zero, - becp2_, - nkb3); - } - - Complex* becp1_cpu = nullptr; - Complex* becp2_cpu = nullptr; - Complex* ps1_cpu = nullptr; - Complex* ps2_cpu = nullptr; - std::vector tmp_space1, tmp_space2; - if(std::is_same::value) - { - tmp_space1.resize(nkb * n_npwx + nkb3 * n_npwx); - becp1_cpu = tmp_space1.data(); - becp2_cpu = becp1_cpu + nkb * n_npwx; - syncmem_complex_d2h_op()(becp1_cpu, becp1_, nkb * n_npwx); - syncmem_complex_d2h_op()(becp2_cpu, becp2_, nkb3 * n_npwx); - - tmp_space2.resize(nkb * n_npwx + nkb3 * n_npwx, 0.0); - ps1_cpu = tmp_space2.data(); - ps2_cpu = ps1_cpu + nkb * n_npwx; - } - else - { - Parallel_Reduce::reduce_pool(becp1_, nkb * n_npwx); - Parallel_Reduce::reduce_pool(becp2_, nkb3 * n_npwx); - becp1_cpu = becp1_; - becp2_cpu = becp2_; - - setmem_complex_op()(ps1_, 0.0, nkb * n_npwx); - setmem_complex_op()(ps2_, 0.0, nkb3 * n_npwx); - ps1_cpu = ps1_; - ps2_cpu = ps2_; + ModuleBase::gemm_op()('C', 'N', nkb, n_npwx, npw, &one, vkb_d, max_npw, psi0, max_npw, &zero, becp1_, nkb); + ModuleBase::gemm_op()('C', 'N', nkb3, n_npwx, npw, &one, gradvkb_d, max_npw, psi0, max_npw, &zero, becp2_, nkb3); } - // 2. <\beta \psi>ucell->ntype; it++) - { - const int nproj = this->ucell->atoms[it].ncpp.nh; - for (int ia = 0; ia < this->ucell->atoms[it].na; ia++) - { - for (int ip = 0; ip < nproj; ip++) - { - for (int ip2 = 0; ip2 < nproj; ip2++) - { - for (int ib = 0; ib < n_npwx; ++ib) - { - FPTYPE dij = static_cast(this->ppcell->deeq(current_spin, iat, ip, ip2)); - int sumip2 = sum + ip2; - int sumip = sum + ip; - ps1_cpu[sumip2 * n_npwx + ib] += dij * becp1_cpu[ib * nkb + sumip]; - ps2_cpu[sumip2 * n_npwx + ib] += dij * becp2_cpu[ib * nkb3 + sumip]; - ps2_cpu[(sumip2 + nkb) * n_npwx + ib] += dij * becp2_cpu[ib * nkb3 + sumip + nkb]; - ps2_cpu[(sumip2 + 2 * nkb) * n_npwx + ib] += dij * becp2_cpu[ib * nkb3 + sumip + 2 * nkb]; - } - } - } - sum += nproj; - ++iat; - } - } - } - else + if (npol != 1) { - ModuleBase::WARNING_QUIT("Velocity", "Velocity operator does not support the non-collinear case yet!"); + ModuleBase::WARNING_QUIT("Velocity", "Non-collinear velocity is not supported."); } + // deeq stores energies in Ry; velocity uses Hartree atomic units like p. + this->contraction_.contract(*this->ucell, *this->ppcell, this->isk[this->ik], n_npwx, FPTYPE(0.5), *this->wfcpw, becp1_); - if(std::is_same::value) + if (npw == 0) { - syncmem_complex_h2d_op()(ps1_, ps1_cpu, nkb * n_npwx); - syncmem_complex_h2d_op()(ps2_, ps2_cpu, nkb3 * n_npwx); + ModuleBase::timer::end("Operator", "Velocity"); + return; } if (n_npwx == 1) @@ -353,40 +299,21 @@ void Velocity::act(const psi::Psi, Device>* int inc = 1; for (int id = 0; id < 3; ++id) { - int vkbshift = id * max_npw * nkb; - int ps2shift = id * nkb; - int npwshift = id * max_npw; - ModuleBase::gemv_op()('N', - npw, - nkb, - &one, - gradvkb_d + vkbshift, - max_npw, - ps1_, - inc, - &one, - vpsi + npwshift, - inc); - ModuleBase::gemv_op()('N', - npw, - nkb, - &one, - vkb_d, - max_npw, - ps2_ + ps2shift, - inc, - &one, - vpsi + npwshift, - inc); + const std::int64_t vkbshift = static_cast(id) * max_npw * nkb; + const std::int64_t ps2shift = static_cast(id) * nkb; + const std::int64_t npwshift = id * component_elements; + ModuleBase::gemv_op()('N', npw, nkb, &one, gradvkb_d + vkbshift, max_npw, ps1_, inc, &one, vpsi + npwshift, inc); + ModuleBase::gemv_op()('N', npw, nkb, &one, vkb_d, max_npw, ps2_ + ps2shift, inc, &one, vpsi + npwshift, inc); } } else { for (int id = 0; id < 3; ++id) { - int vkbshift = id * max_npw * nkb; - int ps2shift = id * n_npwx * nkb; - int npwshift = id * max_npw * n_npwx; + const std::int64_t vkbshift = static_cast(id) * max_npw * nkb; + const std::int64_t ps2shift = id * block; + const std::int64_t npwshift = id * component_elements; ModuleBase::gemm_op()('N', 'T', npw, @@ -415,10 +342,7 @@ void Velocity::act(const psi::Psi, Device>* max_npw); } } - delmem_complex_op()(ps1_); - delmem_complex_op()(ps2_); - delmem_complex_op()(becp1_); - delmem_complex_op()(becp2_); + ModuleBase::timer::end("Operator", "Velocity"); return; } diff --git a/source/source_pw/module_pwdft/op_pw_vel.h b/source/source_pw/module_pwdft/op_pw_vel.h index 0aa31be0649..1c5096e2389 100644 --- a/source/source_pw/module_pwdft/op_pw_vel.h +++ b/source/source_pw/module_pwdft/op_pw_vel.h @@ -1,52 +1,63 @@ #ifndef VELOCITY_PW_H #define VELOCITY_PW_H #include "op_pw.h" -#include "source_cell/unitcell.h" #include "source_base/module_device/types.h" -#include "source_pw/module_pwdft/vnl_pw.h" #include "source_basis/module_pw/pw_basis_k.h" +#include "source_cell/unitcell.h" +#include "source_pw/module_pwdft/projector_gradient.h" +#include "source_pw/module_pwdft/velocity_workspace.h" +#include "source_pw/module_pwdft/vnl_pw.h" +#include namespace hamilt { -//velocity operator mv = im/\hbar * [H,r] = p + im/\hbar [V_NL, r] +// velocity operator mv = im/\hbar * [H,r] = p + im/\hbar [V_NL, r] template class Velocity { - public: - Velocity( - const ModulePW::PW_Basis_K* wfcpw_in, - const int* isk_in, - pseudopot_cell_vnl* ppcell_in, - const UnitCell* ucell_in, - const bool nonlocal_in = true, - const typename GetTypeReal::type* vtau_in = nullptr, - const int vtau_col_in = 0, - const int vtau_row_in = 0 - ); + public: + Velocity(const ModulePW::PW_Basis_K* wfcpw_in, + const int* isk_in, + pseudopot_cell_vnl* ppcell_in, + const UnitCell* ucell_in, + const bool nonlocal_in = true, + const typename GetTypeReal::type* vtau_in = nullptr, + const int vtau_col_in = 0, + const int vtau_row_in = 0); ~Velocity(); void init(const int ik_in); + /** @brief Refresh momentum and projectors with a Hartree-unit vector potential. */ + void init(const int ik_in, const ModuleBase::Vector3& vector_potential); + /** @brief Refresh borrowed spin and meta-GGA potential views before reuse. */ + void set_state(const int* isk, const FPTYPE* vtau, const int cols, const int rows) + { + this->isk = isk; + vtau_ = vtau; + vtau_col_ = cols; + vtau_row_ = rows; + } /** * @brief calculate \hat{v}|\psi> - * + * * @param psi_in Psi class which contains some information * @param n_npwx nbands * NPOL - * @param tmpsi_in |\psi_i> size: n_npwx*npwx + * @param tmpsi_in |\psi_i> size: n_npwx*npwx * @param tmvpsi \hat{v}|\psi> size: 3*n_npwx*npwx * @param add true : tmvpsi = tmvpsi + v|\psi> false: tmvpsi = v|\psi> - * + * */ void act(const psi::Psi, Device>* psi_in, const int n_npwx, const std::complex* tmpsi_in, std::complex* tmvpsi, const bool add = false) const; - + bool nonlocal = true; - private: + private: const ModulePW::PW_Basis_K* wfcpw = nullptr; const int* isk = nullptr; @@ -55,35 +66,34 @@ class Velocity const UnitCell* ucell = nullptr; - int ik=0; + int ik = 0; - double tpiba=0.0; + double tpiba = 0.0; const typename GetTypeReal::type* vtau_ = nullptr; ///< [CPU] meta-GGA vtau on real grid (nspin x nrxx_smooth) - int vtau_col_ = 0; ///< number of grid points per spin for vtau - int vtau_row_ = 0; ///< number of spin channels stored in vtau_ - mutable std::complex* porter1_ = nullptr; ///< workspace on real grid / recip grid - mutable std::complex* porter2_ = nullptr; ///< workspace on real grid / recip grid + int vtau_col_ = 0; ///< number of grid points per spin for vtau + int vtau_row_ = 0; ///< number of spin channels stored in vtau_ + mutable std::complex* porter1_ = nullptr; ///< workspace on real grid / recip grid + mutable std::complex* porter2_ = nullptr; ///< workspace on real grid / recip grid + int momentum_capacity_ = 0; + std::int64_t projector_capacity_ = 0; + mutable int porter_capacity_ = 0; + ProjectorGradient gradient_; + mutable VelocityWorkspace contraction_; Device* ctx = {}; private: - FPTYPE* gx_ = nullptr; ///<[Device, npwx] x component of G+K - FPTYPE* gy_ = nullptr; ///<[Device, npwx] y component of G+K - FPTYPE* gz_ = nullptr; ///<[Device, npwx] z component of G+K + FPTYPE* gx_ = nullptr; ///<[Device, npwx] x component of G+K + FPTYPE* gy_ = nullptr; ///<[Device, npwx] y component of G+K + FPTYPE* gz_ = nullptr; ///<[Device, npwx] z component of G+K std::complex* vkb_ = nullptr; ///<[Device, nkb * npwk_max] nonlocal pseudopotential vkb std::complex* gradvkb_ = nullptr; ///<[Device, 3*nkb * npwk_max] gradient of nonlocal pseudopotential gradvkb - FPTYPE* deeq_ = nullptr; ///<[Device] D matrix for nonlocal pseudopotential - + using Complex = std::complex; using resmem_var_op = base_device::memory::resize_memory_op; using delmem_var_op = base_device::memory::delete_memory_op; using syncmem_var_h2d_op = base_device::memory::synchronize_memory_op; - using castmem_var_h2d_op = base_device::memory::cast_memory_op; using resmem_complex_op = base_device::memory::resize_memory_op, Device>; - using setmem_complex_op = base_device::memory::set_memory_op, Device>; using delmem_complex_op = base_device::memory::delete_memory_op, Device>; - using castmem_complex_h2d_op = base_device::memory::cast_memory_op, std::complex, Device, base_device::DEVICE_CPU>; - using syncmem_complex_d2h_op = base_device::memory::synchronize_memory_op, base_device::DEVICE_CPU, Device>; - using syncmem_complex_h2d_op = base_device::memory::synchronize_memory_op, Device, base_device::DEVICE_CPU>; }; -} +} // namespace hamilt #endif diff --git a/source/source_pw/module_pwdft/projector_gradient.cpp b/source/source_pw/module_pwdft/projector_gradient.cpp new file mode 100644 index 00000000000..3832aff9bbe --- /dev/null +++ b/source/source_pw/module_pwdft/projector_gradient.cpp @@ -0,0 +1,120 @@ +#include "source_pw/module_pwdft/projector_gradient.h" + +#include "source_base/timer.h" +#include "source_pw/module_pwdft/kernels/projector_gradient_op.h" + +#include +#include + +namespace hamilt +{ +template +template +void ProjectorGradient::reserve(ct::Tensor* tensor, const int64_t count) +{ + using CtDevice = typename ct::PsiToContainer::type; + const int64_t capacity = std::max(1, count); + if (tensor->NumElements() < capacity || tensor->data_type() != ct::DataTypeToEnum::value + || tensor->device_type() != ct::DeviceTypeToEnum::value) + { + *tensor = ct::Tensor(ct::DataTypeToEnum::value, ct::DeviceTypeToEnum::value, {capacity}); + } +} + +template +void ProjectorGradient::calculate(pseudopot_cell_vnl* pp, + const UnitCell& cell, + const ModulePW::PW_Basis_K& basis, + const int ik, + const ModuleBase::Vector3& A, + Complex* output) +{ + ModuleBase::timer::start("ProjectorGradient", "calculate"); + if (pp->nkb == 0) + { + ModuleBase::timer::end("ProjectorGradient", "calculate"); + return; + } + pp->ensure_grad_table(cell); + const double dq = pp->table_dq(); + if (version_ != pp->table_version()) + { + reserve(&radial_, pp->tab.getSize()); + reserve(&derivative_, pp->tab_dq.getSize()); + base_device::memory::cast_memory_op()(radial_.template data(), + pp->tab.ptr, + pp->tab.getSize()); + base_device::memory::cast_memory_op()(derivative_.template data(), + pp->tab_dq.ptr, + pp->tab_dq.getSize()); + host_metadata_.clear(); + int atom = 0; + for (int type = 0; type < cell.ntype; ++type) + { + for (int ia = 0; ia < cell.atoms[type].na; ++ia, ++atom) + { + for (int p = 0; p < cell.atoms[type].ncpp.nh; ++p) + { + host_metadata_.push_back(type); + host_metadata_.push_back(static_cast(pp->indv(type, p))); + host_metadata_.push_back(static_cast(pp->nhtol(type, p))); + host_metadata_.push_back(static_cast(pp->nhtolm(type, p))); + host_metadata_.push_back(atom); + } + } + } + reserve(&metadata_, host_metadata_.size()); + base_device::memory::synchronize_memory_op()(metadata_.template data(), + host_metadata_.data(), + host_metadata_.size()); + version_ = pp->table_version(); + } + const int npw = basis.npwk[ik]; + if (npw == 0) + { + ModuleBase::timer::end("ProjectorGradient", "calculate"); + return; + } + const std::int64_t momentum_elements = 3 * static_cast(npw); + host_q_.resize(momentum_elements); + for (int ig = 0; ig < npw; ++ig) + { + const ModuleBase::Vector3 q = basis.getgpluskcar(ik, ig) + A / cell.tpiba; + for (int d = 0; d < 3; ++d) + { + host_q_[3 * static_cast(ig) + d] = static_cast(q[d]); + } + const Real* v = host_q_.data() + 3 * static_cast(ig); + const Real position = std::sqrt(v[0] * v[0] + v[1] * v[1] + v[2] * v[2]) * static_cast(cell.tpiba) / static_cast(dq); + pp->check_vnl_index(static_cast(position) * (1.0 + 8.0 * std::numeric_limits::epsilon()), true); + } + const std::int64_t structure_elements = static_cast(cell.nat) * npw; + reserve(&momentum_, momentum_elements); + reserve(&structure_, structure_elements); + base_device::memory::synchronize_memory_op()(momentum_.template data(), + host_q_.data(), + momentum_elements); + Device* ctx = nullptr; + pp->psf->get_sk(ctx, ik, &basis, structure_.template data()); + projector_gradient_op()(npw, + basis.npwk_max, + pp->nkb, + pp->tab.getBound2(), + pp->tab.getBound3(), + static_cast(dq), + static_cast(cell.tpiba), + metadata_.template data(), + momentum_.template data(), + radial_.template data(), + derivative_.template data(), + structure_.template data(), + output); + ModuleBase::timer::end("ProjectorGradient", "calculate"); +} +template class ProjectorGradient; +template class ProjectorGradient; +#if defined(__CUDA) || defined(__ROCM) +template class ProjectorGradient; +template class ProjectorGradient; +#endif +} // namespace hamilt diff --git a/source/source_pw/module_pwdft/projector_gradient.h b/source/source_pw/module_pwdft/projector_gradient.h new file mode 100644 index 00000000000..01aef3e9f56 --- /dev/null +++ b/source/source_pw/module_pwdft/projector_gradient.h @@ -0,0 +1,32 @@ +#ifndef PW_PROJECTOR_GRADIENT_H +#define PW_PROJECTOR_GRADIENT_H +#include "source_pw/module_pwdft/vnl_pw.h" +#include "source_base/module_container/ATen/core/tensor.h" + +namespace hamilt +{ +/** @brief One-k-point gradient workspace with versioned radial tables. */ +template +class ProjectorGradient +{ + private: + using Complex = std::complex; + ct::Tensor radial_; + ct::Tensor derivative_; + ct::Tensor metadata_; + ct::Tensor momentum_; + ct::Tensor structure_; + size_t version_ = 0; + std::vector host_q_; + std::vector host_metadata_; + + template + void reserve(ct::Tensor* tensor, const int64_t count); + + public: + /** @brief Refresh the shifted gradient in native precision and device storage. */ + void calculate(pseudopot_cell_vnl* pp, const UnitCell& cell, const ModulePW::PW_Basis_K& basis, + const int ik, const ModuleBase::Vector3& A, Complex* output); +}; +} // namespace hamilt +#endif diff --git a/source/source_pw/module_pwdft/radial_proj.cpp b/source/source_pw/module_pwdft/radial_proj.cpp deleted file mode 100644 index b6fa844c55c..00000000000 --- a/source/source_pw/module_pwdft/radial_proj.cpp +++ /dev/null @@ -1,410 +0,0 @@ -#include -#include -#include -#include -#include -#include "source_pw/module_pwdft/radial_proj.h" -#include "source_base/constants.h" -#include "source_base/matrix.h" -#include "source_base/math_ylmreal.h" -#include "source_base/sph_bessel_tf.h" -#include "source_base/timer.h" - -void RadialProjection::RadialProjector::_build_backward_map(const std::vector>& it2iproj, - const std::vector& iproj2l, - std::vector& irow2it, - std::vector& irow2iproj, - std::vector& irow2m) -{ - const int ntype = it2iproj.size(); // the ntype here only count the valid, that is, with the projector. - - int nproj_tot = 0; // count the total number of projectors (atom-position-irrelevant) - for(int it = 0; it < ntype; it++) // for all types with projector... - { - for(auto& iproj: it2iproj[it]) // for each projector, the projector is indexed with global iproj - { - const int l = iproj2l[iproj]; // easily get the angular momentum of this projector - nproj_tot += (2*l + 1); // add 2l+1 to the total number of projectors - } - } - // resize/allcoate the memory for the output - irow2it.resize(nproj_tot); - irow2iproj.resize(nproj_tot); - irow2m.resize(nproj_tot); - - int irow = 0; - for(int it = 0; it < ntype; it++) - { - const int nproj = it2iproj[it].size(); - for(int iproj = 0; iproj < nproj; iproj++) - { - const int l = iproj2l[it2iproj[it][iproj]]; - for(int m = -l; m <= l; m++) - { - irow2it[irow] = it; - irow2iproj[irow] = iproj; - irow2m[irow] = m; - irow++; - } - } - } -} - -void RadialProjection::RadialProjector::_build_forward_map(const std::vector>& it2ia, - const std::vector>& it2iproj, - const std::vector& iproj2l, - std::map, int>& itiaiprojm2irow) -{ - const int ntype = it2ia.size(); - int irow = 0; - for(int it = 0; it < ntype; it++) - { - const int nproj = it2iproj[it].size(); - for(auto& ia: it2ia[it]) // the index is from UnitCell, so it allows the case ia is not continuous - { - for(int iproj = 0; iproj < nproj; iproj++) - { - const int l = iproj2l[it2iproj[it][iproj]]; // what is the iproj of the i-th projector of it atomtype? - for(int m = -l; m <= l; m++) - { - itiaiprojm2irow[std::make_tuple(it, ia, iproj, m)] = irow; - irow++; - } - } - } - } -} - -void RadialProjection::RadialProjector::_build_sbt_tab(const int nr, - const double* r, - const std::vector& radials, - const std::vector& l, - const int nq, - const double& dq) -{ - ModuleBase::timer::start("RadialProjection", "cubspl_tabulate_vq_each_radial"); - l_ = l; - const int nrad = radials.size(); - assert(nrad == l.size()); - std::vector qgrid(nq); - std::iota(qgrid.begin(), qgrid.end(), 0); - std::transform(qgrid.begin(), qgrid.end(), qgrid.begin(), [dq](const double& q){return q*dq;}); - - if(cubspl_.get()) { cubspl_.reset(); } // release the old one if it is not the first time - cubspl_ = std::unique_ptr(new ModuleBase::CubicSpline(nq, // int - qgrid.data())); // double* - cubspl_->reserve(nrad); - ModuleBase::SphericalBesselTransformer sbt_(true); // bool: enable cache - - std::vector _temp(nq); - // the SphericalBesselTransformer's result is multiplied by one extra factor sqrt(2/pi), should remove it - // see source_base/sph_bessel_tf.h and source_base/sph_bessel_tf.cpp:328 - const double pref = std::sqrt(2.0/std::acos(-1.0)); - for(int i = 0; i < nrad; i++) - { - sbt_.direct(l[i], nr, r, radials[i], nq, qgrid.data(), _temp.data()); - std::for_each(_temp.begin(), _temp.end(), [pref](double& x){x = x/pref;}); - cubspl_->add(_temp.data()); - } - ModuleBase::timer::end("RadialProjection", "cubspl_tabulate_vq_each_radial"); -} - -void RadialProjection::RadialProjector::_build_sbt_tab(const std::vector& r, - const std::vector>& radials, - const std::vector& l, - const int nq, - const double& dq) -{ - ModuleBase::timer::start("RadialProjection", "cubspl_tabulate_vq_each_radial"); - const int nr = r.size(); - const int nrad = radials.size(); - for(int i = 0; i < nrad; i++) { assert(radials[i].size() == nr); } - std::vector radptrs(radials.size()); - for(int i = 0; i < radials.size(); i++) { radptrs[i] = const_cast(radials[i].data()); } - ModuleBase::timer::end("RadialProjection", "cubspl_tabulate_vq_each_radial"); - _build_sbt_tab(nr, r.data(), radptrs, l, nq, dq); -} - -void RadialProjection::RadialProjector::_build_sbt_tab(const std::vector& nproj, - const std::vector& r, - const std::vector>& radials, - const std::vector& l, - const int nq, //< GlobalV::DQ - const double& dq, //< GlobalV::NQX - const double& omega, - const int npol, // for nspin 4 - ModuleBase::realArray& tab, - ModuleBase::matrix& nhtol) // output table -{ - int nprojmax = *std::max_element(nproj.begin(), nproj.end()); - const int ntype = nproj.size(); - - tab.create(ntype, nprojmax*npol, nq); - tab.zero_out(); - - std::vector qgrid(nq); - std::iota(qgrid.begin(), qgrid.end(), 0); - std::transform(qgrid.begin(), qgrid.end(), qgrid.begin(), [dq](const double& q){return q*dq;}); - - ModuleBase::SphericalBesselTransformer sbt_(true); // bool: enable cache - int iproj = 0; - int nchmax = 0; - const double pref = 4*M_PI/std::sqrt(omega) / std::sqrt(2.0/std::acos(-1.0)); - for (int it = 0; it < ntype; it++) - { - int nch = 0; - const int nproj_it = nproj[it]; - for (int ip = 0; ip < nproj_it; ip++) - { - const int l_ = l[iproj]; - nch += 2*l_ + 1; - std::vector _temp(nq); - sbt_.direct(l_, r.size(), r.data(), radials[iproj].data(), nq, qgrid.data(), _temp.data()); - std::for_each(_temp.begin(), _temp.end(), [pref](double& x){x = x*pref;}); - for (int iq = 0; iq < nq; iq++) - { - tab(it, ip, iq) = _temp[iq]; - //std::cout << tab(it, ip, iq) << " "; - } - iproj++; - } - nchmax = std::max(nchmax, nch); - } - //std::cout << std::endl; - //ModuleBase::WARNING_QUIT("RadialProjection", "The following code is not implemented yet."); - - nhtol.create(ntype, nchmax); - nhtol.zero_out(); - iproj = 0; - for (int it = 0; it < ntype; it++) - { - int ih = 0; // channel index, across all projectors of present type - for (int ip = 0; ip < nproj[it]; ip++) - { - const int l_ = l[iproj]; - for (int m = -l_; m <= l_; m++) - { - nhtol(it, ih) = l_; - ih++; - } - iproj++; - } - } -} - -void RadialProjection::RadialProjector::sbtft(const std::vector>& qs, - std::vector>& out, - const char type, - const double& omega, - const double& tpiba) -{ - ModuleBase::timer::start("RadialProjection", "interp_sphbes_ft_flzYlm"); - assert(type == 'r' || type == 'l'); // type must be one of 'r' or 'l' - // first cache the Ylm values - const int lmax_ = *std::max_element(l_.begin(), l_.end()); - const int total_lm = std::pow(lmax_+1, 2); - const int npw = qs.size(); - ModuleBase::matrix ylm_(total_lm, npw); - ModuleBase::YlmReal::Ylm_Real(total_lm, npw, qs.data(), ylm_); - - const int nrad = l_.size(); - int nchannel = 0; - for(auto l: l_) { nchannel += 2*l+1; } - out.resize(nchannel*npw); - - std::vector qnorm(npw); - std::transform(qs.begin(), qs.end(), qnorm.begin(), [tpiba](const ModuleBase::Vector3& q){return tpiba*q.norm();}); - - std::vector Jlfq(npw); - int iproj = 0; - for(int i = 0; i < nrad; i++) - { - const int l = l_[i]; - // here is bug-prone - // we define l as and r as . The former is int{p(r)exp(iqr)} and the latter is int{p(r)exp(-iqr)} - // , in which we have use G+k=q notation. So once do Ylm expansion on exp(iqr), will get a pure imaginary - // prefactor i^l. - std::complex pref = (type == 'l')? std::pow(ModuleBase::IMAG_UNIT, l) : std::pow(ModuleBase::NEG_IMAG_UNIT, l); - pref = pref * ModuleBase::FOUR_PI/std::sqrt(omega); - cubspl_->eval(npw, qnorm.data(), Jlfq.data(), nullptr, nullptr, i); - for(int m = -l; m <= l; m++) - { - for(int iq = 0; iq < npw; iq++) - { - out[iproj*npw+iq] = pref * Jlfq[iq] * ylm_(l*l + l + m, iq); - } - iproj++; - } - } - assert(iproj == nchannel); // should write to inflate each radial to 2l+1 channels - ModuleBase::timer::end("RadialProjection", "interp_sphbes_ft_flzYlm"); -} - -void RadialProjection::_mask_func(std::vector& mask) -{ - /* mask function is hard coded here, eta = 15 */ - mask.resize(201); - std::string src = "0.10000000E+01"; - src += " 0.10000000E+01 0.99948662E+00 0.99863154E+00 0.99743557E+00"; - src += " 0.99589985E+00 0.99402586E+00 0.99181538E+00 0.98927052E+00"; - src += " 0.98639370E+00 0.98318766E+00 0.97965544E+00 0.97580040E+00"; - src += " 0.97162618E+00 0.96713671E+00 0.96233623E+00 0.95722924E+00"; - src += " 0.95182053E+00 0.94611516E+00 0.94011842E+00 0.93383589E+00"; - src += " 0.92727338E+00 0.92043693E+00 0.91333282E+00 0.90596753E+00"; - src += " 0.89834777E+00 0.89048044E+00 0.88237263E+00 0.87403161E+00"; - src += " 0.86546483E+00 0.85667987E+00 0.84768450E+00 0.83848659E+00"; - src += " 0.82909416E+00 0.81951535E+00 0.80975838E+00 0.79983160E+00"; - src += " 0.78974340E+00 0.77950227E+00 0.76911677E+00 0.75859548E+00"; - src += " 0.74794703E+00 0.73718009E+00 0.72630334E+00 0.71532544E+00"; - src += " 0.70425508E+00 0.69310092E+00 0.68187158E+00 0.67057566E+00"; - src += " 0.65922170E+00 0.64781819E+00 0.63637355E+00 0.62489612E+00"; - src += " 0.61339415E+00 0.60187581E+00 0.59034914E+00 0.57882208E+00"; - src += " 0.56730245E+00 0.55579794E+00 0.54431609E+00 0.53286431E+00"; - src += " 0.52144984E+00 0.51007978E+00 0.49876105E+00 0.48750040E+00"; - src += " 0.47630440E+00 0.46517945E+00 0.45413176E+00 0.44316732E+00"; - src += " 0.43229196E+00 0.42151128E+00 0.41083069E+00 0.40025539E+00"; - src += " 0.38979038E+00 0.37944042E+00 0.36921008E+00 0.35910371E+00"; - src += " 0.34912542E+00 0.33927912E+00 0.32956851E+00 0.31999705E+00"; - src += " 0.31056799E+00 0.30128436E+00 0.29214897E+00 0.28316441E+00"; - src += " 0.27433307E+00 0.26565709E+00 0.25713844E+00 0.24877886E+00"; - src += " 0.24057988E+00 0.23254283E+00 0.22466884E+00 0.21695884E+00"; - src += " 0.20941357E+00 0.20203357E+00 0.19481920E+00 0.18777065E+00"; - src += " 0.18088790E+00 0.17417080E+00 0.16761900E+00 0.16123200E+00"; - src += " 0.15500913E+00 0.14894959E+00 0.14305240E+00 0.13731647E+00"; - src += " 0.13174055E+00 0.12632327E+00 0.12106315E+00 0.11595855E+00"; - src += " 0.11100775E+00 0.10620891E+00 0.10156010E+00 0.97059268E-01"; - src += " 0.92704295E-01 0.88492966E-01 0.84422989E-01 0.80492001E-01"; - src += " 0.76697569E-01 0.73037197E-01 0.69508335E-01 0.66108380E-01"; - src += " 0.62834685E-01 0.59684561E-01 0.56655284E-01 0.53744102E-01"; - src += " 0.50948236E-01 0.48264886E-01 0.45691239E-01 0.43224469E-01"; - src += " 0.40861744E-01 0.38600231E-01 0.36437098E-01 0.34369520E-01"; - src += " 0.32394681E-01 0.30509780E-01 0.28712032E-01 0.26998673E-01"; - src += " 0.25366964E-01 0.23814193E-01 0.22337676E-01 0.20934765E-01"; - src += " 0.19602844E-01 0.18339338E-01 0.17141711E-01 0.16007467E-01"; - src += " 0.14934157E-01 0.13919377E-01 0.12960772E-01 0.12056034E-01"; - src += " 0.11202905E-01 0.10399183E-01 0.96427132E-02 0.89313983E-02"; - src += " 0.82631938E-02 0.76361106E-02 0.70482151E-02 0.64976294E-02"; - src += " 0.59825322E-02 0.55011581E-02 0.50517982E-02 0.46327998E-02"; - src += " 0.42425662E-02 0.38795566E-02 0.35422853E-02 0.32293218E-02"; - src += " 0.29392897E-02 0.26708663E-02 0.24227820E-02 0.21938194E-02"; - src += " 0.19828122E-02 0.17886449E-02 0.16102512E-02 0.14466132E-02"; - src += " 0.12967606E-02 0.11597692E-02 0.10347601E-02 0.92089812E-03"; - src += " 0.81739110E-03 0.72348823E-03 0.63847906E-03 0.56169212E-03"; - src += " 0.49249371E-03 0.43028657E-03 0.37450862E-03 0.32463165E-03"; - src += " 0.28016004E-03 0.24062948E-03 0.20560566E-03 0.17468305E-03"; - src += " 0.14748362E-03 0.12365560E-03 0.10287226E-03 0.84830727E-04"; - src += " 0.69250769E-04 0.55873673E-04 0.44461100E-04 0.34793983E-04"; - src += " 0.26671449E-04 0.19909778E-04 0.14341381E-04 0.98138215E-05"; - std::stringstream ss(src); - for(int i = 0; i < mask.size(); i++) - { - ss >> mask[i]; - } -} - -void RadialProjection::_do_mask_on_radial(const int nr1, - const double* r, - const double* in, - const int nr2, - const double* mask, - double* out) -{ - /* the key here is to avoid any float-point overflow */ -} - -/** - * Additional-bidirectional mapping for the projector. - * - * These two methods are commented out because of minimal-implementation consideration. - */ - -// void build_itiprojm_map(const std::vector>& it2iproj, -// const std::vector& iproj2l, -// std::vector& irow2it, -// std::vector& irow2iproj, -// std::vector& irow2m, -// std::map, int>& itiprojm2irow) -// { -// const int ntype = it2iproj.size(); - -// int nproj_tot = 0; -// for(int it = 0; it < ntype; it++) -// { -// for(auto& iproj: it2iproj[it]) -// { -// const int l = iproj2l[iproj]; -// nproj_tot += (2*l + 1); -// } -// } -// irow2it.resize(nproj_tot); -// irow2iproj.resize(nproj_tot); -// irow2m.resize(nproj_tot); - -// int irow = 0; -// for(int it = 0; it < ntype; it++) -// { -// const int nproj = it2iproj[it].size(); -// for(int iproj = 0; iproj < nproj; iproj++) -// { -// const int l = iproj2l[it2iproj[it][iproj]]; -// for(int m = -l; m <= l; m++) -// { -// irow2it[irow] = it; -// irow2iproj[irow] = iproj; -// irow2m[irow] = m; -// itiprojm2irow[std::make_tuple(it, iproj, m)] = irow; -// irow++; -// } -// } -// } -// } - -// void build_itiaiprojm_map(const std::vector>& it2ia, -// const std::vector>& it2iproj, -// const std::vector& iproj2l, -// std::vector& irow2it, -// std::vector& irow2ia, -// std::vector& irow2iproj, -// std::vector& irow2m, -// std::map, int>& itiaiprojm2irow) -// { -// const int ntype = it2ia.size(); -// int nproj_tot = 0; -// for(int it = 0; it < ntype; it++) -// { -// for(auto& ia: it2ia[it]) -// { -// for(auto& iproj: it2iproj[it]) -// { -// const int l = iproj2l[iproj]; -// nproj_tot += (2*l + 1); -// } -// } -// } -// irow2it.resize(nproj_tot); -// irow2ia.resize(nproj_tot); -// irow2iproj.resize(nproj_tot); -// irow2m.resize(nproj_tot); - -// int irow = 0; -// for(int it = 0; it < ntype; it++) -// { -// const int nproj = it2iproj[it].size(); -// for(auto& ia: it2ia[it]) -// { -// for(int iproj = 0; iproj < nproj; iproj++) -// { -// const int l = iproj2l[it2iproj[it][iproj]]; -// for(int m = -l; m <= l; m++) -// { -// irow2it[irow] = it; -// irow2ia[irow] = ia; -// irow2iproj[irow] = iproj; -// irow2m[irow] = m; -// itiaiprojm2irow[std::make_tuple(it, ia, iproj, m)] = irow; -// irow++; -// } -// } -// } -// } -// } \ No newline at end of file diff --git a/source/source_pw/module_pwdft/radial_proj.h b/source/source_pw/module_pwdft/radial_proj.h deleted file mode 100644 index b19590be8e4..00000000000 --- a/source/source_pw/module_pwdft/radial_proj.h +++ /dev/null @@ -1,254 +0,0 @@ -#ifndef RADIAL_PROJECTION_H -#define RADIAL_PROJECTION_H - -/** - * @file radial_projection.h - * - * project any atom-centered function that has seperatable radial and angular parts - * or any function can expanded with spherical harmonics onto the planewave basis, - * although the latter will be somewhat cumbersome: - * f(r) = sum_{l,m} f_{lm}(r) * Ylm(theta, phi) - * F(q) = sum_{l,m} i^l * 4*pi/sqrt(omega) * Jl[f_{lm}](q) * Ylm(q) - */ - - -#include "source_base/vector3.h" -#include "source_base/cubic_spline.h" -#include -#include -#include -#include -#include -#include - -#include "source_cell/unitcell.h" -#include "source_basis/module_pw/pw_basis_k.h" - -namespace RadialProjection -{ - /** - * @brief RadialProjector is for projecting a function who has seperatable radial - * and angular parts: - * f(r) = f(|r|) * Ylm(theta, phi) - * onto planewave basis. - * - * Usage: - * - * 1. classical way: reciprocal space integration - * ```c++ - * RadialProjector rp(false); - * const int nq = 1000; - * const double dq = 0.01; - * // given `r` is the real space grid and `radials` is the collection of radial - * // functions, `l` is the angular momentum quantum number for each radial function - * // then the interpolation table can be rapidly built by calling SphericalBesselTransformer - * // and CubicSpline modules. - * rp._build_sbt_tab(r, radials, l, nq, dq); - * // then the set of q will used to calculate the Fourier transform - * rp.sbtft(qs, out, 'r', omega, tpiba); - * // in `out`, there will be the Fourier transform of the radial functions organized - * // in the same way as the input `radials` and `qs`, as row and column respectively. - * // but one should note for each radials, there are 2*l+1 components now instead of - * // just one. - * - * // one may find it is not easy to maintain such a large table, so here, also provides - * // a tool function to map the 2D index to 1D index, and vice versa. With the angular - * // momentum used in function _build_sbt_tab, one can easily build a map from - * // [irad][im] to 1D index, and use two functions _irad_m_to_idx and _idx_to_irad_m - * // to convert between 1D index and [irad][m] (instead of im!). - * std::vector> map_; - * rp._build_sbtft_map(l, map_); - * ``` - * - * 2. SBFFT: small box fast-fourier-transform (not implemented yet) - */ - class RadialProjector - { - public: - /** - * Notation of following two functions: - * - * Given all the projectors are listed in a series, so the `iproj` is the index goes across - * all atomtypes, which means if for the first type, the iproj goes from 0 to 4, then the - * second atomtypes the iproj will start from 5, and so on... - * However, there is also another convention, like numerical atomic orbitals, developer always - * use "l" to index orbitals, here, in all output map, the `iproj` will start from 0, which - * means in output the `iproj` is local index. - * ----------------------------------------------------------------------------------------- - * First, the following lists should be prepared as early as possible, - * - * it2iproj: for given it, the index of atom type, return the list of index of projectors. - * - * iproj2l: for given iproj, the index of projectors, return the l of this projector. More - * simply explaning, it is just the list of angular momentum of projectors. - * - * it2ia: just a list that stolen information from UnitCell, for given it, the index of atom - * within the range of it. So this list is different from the it2iproj, iproj is the index - * across type but ia is the index within the type. So for each it2ia[it], the ia, in principle - * , always/can start from 0. - * - * One may question that does the indexing support one atom type with multiple projectors? The - * answer is YES. Combining the it2iproj and it2ia, one can even support PART of atoms of one - * type has multiple projectors. - * ----------------------------------------------------------------------------------------- - * Then the returned lists, - * - * irow2it: for given `irow`, the index of row, return the `it`: the index of atom type. - * - * irow2ia: for given `irow`, the index of row, return the `ia`: the index of atom within the range - * of `it`. - * - * irow2iproj: for given `irow`, the index of row, return the `iproj`, the index of projectors, - * note that this `iproj` is the local index. - * - * irow2m: for given irow, the index of row, return the m, the magnetic quantum number of this - * projector. - * - * One may complain that cannot get `l` from the `irow`, but the truth is, not exactly. One can - * get the `l` starting from `irow` by: - * ```c++ - * const int iproj = irow2iproj[irow]; - * const int it = irow2it[irow]; - * const int iproj_g = it2iproj[it][iproj]; - * const int l = iproj2l[iproj_g]; - * ``` - */ - - static void _build_backward_map(const std::vector>& it2iproj, - const std::vector& iproj2l, - std::vector& irow2it, - std::vector& irow2iproj, - std::vector& irow2m); - static void _build_forward_map(const std::vector>& it2ia, - const std::vector>& it2iproj, - const std::vector& iproj2l, - std::map, int>& itiaiprojm2irow); - public: - /** - * @brief Construct a new Radial Projector object - * - * @param realspace if perform integration in real space rather than reciprocal space - * , default is false - * - * @attention Currectly only reciprocal space method is implemented - */ - RadialProjector(const bool realspace = false) {} - ~RadialProjector() {} - - // it is more feasible to build interpolation table. this function will tabulate - // for those functions. This function will write the in-build tab_, to place the - // values of Jl[f](q) for each q and l. - // Here I provide two versions of tabulate, one for the case may be capable to - // avoid the memory copy operation, and the other one is for the case of - // std::vector and std::vector>. - /** - * @brief make a interpolation table for the Spherical Bessel Transform of f(r) - * - * @param nr number of grid points, shared by all radial functions - * @param r radial grids, shared by all radial functions - * @param radials radial functions, each element is a radial function - * @param l angular momentum quantum number for each radial function - * @param nq number of q-points - * @param dq space between q-points - */ - void _build_sbt_tab(const int nr, - const double* r, - const std::vector& radials, - const std::vector& l, - const int nq, //< GlobalV::DQ - const double& dq); //< GlobalV::NQX - void _build_sbt_tab(const std::vector& r, - const std::vector>& radials, - const std::vector& l, - const int nq, //< GlobalV::DQ - const double& dq); //< GlobalV::NQX - // compatibility concern: for FS_Nonlocal_tools. Will not call sbtft so need omega - void _build_sbt_tab(const std::vector& nproj, - const std::vector& r, - const std::vector>& radials, - const std::vector& l, - const int nq, //< GlobalV::DQ - const double& dq, //< GlobalV::NQX - const double& omega, - const int npol, - ModuleBase::realArray& tab, - ModuleBase::matrix& nhtol); - /** - * @brief perform analytical version of the Fourier transform: - * F(q) = int(f(r)*exp(-iq.r) d^3r) - * = 4*pi/sqrt(omega) * (-i)^l * Jl[f](q) * Ylm(q) - * , where Ylm(q) is real spherical harmonic function, and Jl[f](q) is - * the Spherial Bessel Transform of f(r): - * Jl[f](q) = int(f(r)*j_l(q*r)*r^2 dr) - * , where j_l(q*r) is the spherical Bessel function of the first kind. - * . If use another notation, F(q) = , this is denoted as type - * "r" for ket |>, and "l" for bra <|. - */ - - void sbtft(const std::vector>& qs, - std::vector>& out, - const char type = 'r', // 'r' for ket |>, 'l' for bra <| - const double& omega = 1.0, - const double& tpiba = 1.0); // 'n' for no gradient, 'x', 'y', 'z' for gradient in x, y, z direction - - void sbfft(); // interface for SBFFT - - - private: - std::unique_ptr cubspl_; - std::vector l_; - }; - - /** ==================================================================================== - * - * Small box Fast-Fourier-Transform (SBFFT) - * - * ==================================================================================== - * Small box FFT is a technique for quickly intergrating real-space localized functions - * , or say perform FFT in a small box defined by a "mask function" with relatively low - * time complexity, will be out-performing in system where number of atoms are larger - * than ten. For details please refer to the work: - * Mask-function real-space implementations of nonlocal pseudopotentials - * by Wang, L.-W., PHYSICAL REVIEW B, VOLUME64,201107(R) - * - * Following are the brief technical review of this technique. Given the function to - * be transformed w(r): - * 1. Generate the q-grid in range |q| < 2qmax - qc, in which the qmax is the one - * defined by ecutrho, and qc is the cutoff defined by ecutwfc. - * 2. With mask function m(r) generated, make division of w(r) by m(r). The real space - * cutoff (or the radius that w(r) vanishes), r0, must be smaller than the cutoff - * of m(r) to ensure the convergence of the division. Denote wm(r) = w(r)/m(r). - * 3. Make FT on wm(r) to get wm(q) - * 4. Make inverse FT with only the q-grid in range |q| < 2qmax - qc to get the - * wm'(r). - * 5. Perform real-space integration on function w'(r)*m(r)*exp(iqr). - */ - - /** - * @brief get the mask function for SBFFT - * - * @param mask mask function - */ - void _mask_func(std::vector& mask); - - /** - * @brief do operation w(r)/m(r) on a radial function. The cutoff radius of w(r) - * is smaller than the cutoff radius of m(r). The m(r) has been rescaled so that - * r ranges from 0 to 1. - * - * @param nr1 number of grid points of function to operate - * @param r grid points of function to operate - * @param in function to operate - * @param nr2 number of grid points of mask function - * @param mask mask function - * @param out output value - */ - void _do_mask_on_radial(const int nr1, - const double* r, - const double* in, - const int nr2, - const double* mask, - double* out); -} - -#endif // RADIAL_PROJECTION_H diff --git a/source/source_pw/module_pwdft/setup_dftu_pw.h b/source/source_pw/module_pwdft/setup_dftu_pw.h index c9c4731cc49..fa6c43bc74a 100644 --- a/source/source_pw/module_pwdft/setup_dftu_pw.h +++ b/source/source_pw/module_pwdft/setup_dftu_pw.h @@ -3,7 +3,7 @@ #include "source_cell/unitcell.h" #include "source_base/matrix.h" -#include "source_estate/module_charge/charge_mixing.h" +#include "source_estate/module_charge/chg_mix.h" struct Input_para; class Plus_U_Base; // mohan add 2025-11-06 diff --git a/source/source_pw/module_pwdft/setup_pot.cpp b/source/source_pw/module_pwdft/setup_pot.cpp index 4541f891a9d..44da50a44e0 100644 --- a/source/source_pw/module_pwdft/setup_pot.cpp +++ b/source/source_pw/module_pwdft/setup_pot.cpp @@ -1,8 +1,8 @@ #include "source_pw/module_pwdft/setup_pot.h" -#include "source_estate/module_charge/symm_rho.h" +#include "source_estate/module_charge/chg_symm.h" #include "source_lcao/module_deltaspin/spin_constrain.h" -#include "source_pw/module_pwdft/onsite_proj.h" +#include "source_pw/module_proj/onsite_proj.h" #include "source_pw/module_pwdft/vnl_pw.h" #include "source_pw/module_pwdft/dftu_base.h" // mohan add 2025-11-06 #include "source_pw/module_pwdft/vsep_pw.h" @@ -47,13 +47,12 @@ void pw::setup_pot(const int istep, //! 2) Symmetrize the charge density (rho) //---------------------------------------------------------- - //! Symmetry_rho should behind init_scf, because charge should be - //! initialized first. liuyu comment: Symmetry_rho should be + //! Density symmetrization should be behind init_scf, because charge should be + //! initialized first. liuyu comment: it should be //! located between init_rho and v_of_rho? - Symmetry_rho srho; for (int is = 0; is < inp.nspin; is++) { - srho.begin(is, chr, pw_rhod, ucell.symm); + module_charge::cal_rhog_symm(is, chr, pw_rhod, ucell.symm); } //---------------------------------------------------------- diff --git a/source/source_pw/module_pwdft/setup_pwrho.cpp b/source/source_pw/module_pwdft/setup_pwrho.cpp index c0d7c6f5bf2..a0e32dcb0d2 100644 --- a/source/source_pw/module_pwdft/setup_pwrho.cpp +++ b/source/source_pw/module_pwdft/setup_pwrho.cpp @@ -1,16 +1,17 @@ #include "source_pw/module_pwdft/setup_pwrho.h" +#include "source_basis/module_pw/pw_basis_big.h" // use PW_Basis_Big #include "source_io/module_output/print_info.h" // use print_rhofft #include "source_base/parallel_comm.h" // use POOL_WORLD void pw::setup_pwrho( - UnitCell& ucell, // unitcell - const bool double_grid, // for USPP + UnitCell& ucell, // unitcell + const bool double_grid, // for USPP bool &pw_rho_flag, // flag for allocation of pw_rho - ModulePW::PW_Basis* &pw_rho, // pw for rhod - ModulePW::PW_Basis* &pw_rhod, // pw for rhod - ModulePW::PW_Basis_Big* &pw_big, // pw for rhod - const std::string &classname, - const Input_para& inp) // input parameters * + ModulePW::PW_Basis* &pw_rho, // pw for rhod + ModulePW::PW_Basis* &pw_rhod, // pw for rhod + ModulePW::PW_Basis_Big* &pw_big, // pw for rhod + const std::string &classname, + const Input_para& inp) // input parameters * { ModuleBase::TITLE("pw", "setup_pwrho"); @@ -71,6 +72,17 @@ void pw::setup_pwrho( } //! initialize the FFT grid + // NOTE(liuyu): ref_cell_factor is currently forced to 1.0 in + // read_input_item_md.cpp because the reference-cell mechanism is + // disabled. When ref_cell_factor != 1, the PW_Basis lattice members + // (lat0/tpiba/G/GGT/omega) become stale relative to ucell in NPT, + // breaking sum_rho/get_local_pp_energy/cal_delta_escf/makov_payne. + // If the reference-cell feature is re-enabled in the future, this + // call site (and the equivalent in setup_pwwfc.cpp) MUST be updated + // to use the proposed initgrids_ref/initgrids_actual split so that + // only nx/ny/nz come from the reference cell while lat0/tpiba/G/GGT/omega + // track the physical cell. Until then, ref_cell_factor * ucell.lat0 + // below is effectively just ucell.lat0. if (inp.nx * inp.ny * inp.nz == 0) { pw_rho->initgrids(inp.ref_cell_factor * ucell.lat0, ucell.latvec, 4.0 * inp.ecutwfc); @@ -97,6 +109,7 @@ void pw::setup_pwrho( { pw_rhod->setfullpw(inp.of_full_pw, inp.of_full_pw_dim); } + // NOTE(liuyu): same ref_cell_factor warning applies to pw_rhod. if (inp.ndx * inp.ndy * inp.ndz == 0) { pw_rhod->initgrids(inp.ref_cell_factor * ucell.lat0, ucell.latvec, inp.ecutrho); @@ -119,9 +132,9 @@ void pw::setup_pwrho( void pw::teardown_pwrho(bool &pw_rho_flag, - const bool double_grid, - ModulePW::PW_Basis* &pw_rho, // pw for rhod - ModulePW::PW_Basis* &pw_rhod) // pw for rhod + const bool double_grid, + ModulePW::PW_Basis* &pw_rho, // pw for rhod + ModulePW::PW_Basis* &pw_rhod) // pw for rhod { if (pw_rho_flag == true) { diff --git a/source/source_pw/module_pwdft/setup_pwrho.h b/source/source_pw/module_pwdft/setup_pwrho.h index 7a6ad1bfd33..14d7fb56db4 100644 --- a/source/source_pw/module_pwdft/setup_pwrho.h +++ b/source/source_pw/module_pwdft/setup_pwrho.h @@ -2,27 +2,32 @@ #define SETUP_PWRHO_H #include "source_cell/unitcell.h" // use UnitCell -#include "source_basis/module_pw/pw_basis.h" // use PW_Basis #include "source_io/module_parameter/input_parameter.h" // use Input_para +namespace ModulePW +{ +class PW_Basis; +class PW_Basis_Big; +} + namespace pw { void setup_pwrho( - UnitCell& ucell, // unitcell + UnitCell& ucell, // unitcell const bool double_grid, // for USPP bool &pw_rho_flag, // flag for allocation of pw_rho - ModulePW::PW_Basis* &pw_rho, // pw for rhod - ModulePW::PW_Basis* &pw_rhod, // pw for rhod - ModulePW::PW_Basis_Big* &pw_big, // pw for rhod - const std::string &classname, - const Input_para& inp); // input parameters * + ModulePW::PW_Basis* &pw_rho, // pw for rhod + ModulePW::PW_Basis* &pw_rhod, // pw for rhod + ModulePW::PW_Basis_Big* &pw_big, // pw for rhod + const std::string &classname, + const Input_para& inp); // input parameters * void teardown_pwrho(bool &pw_rho_flag, - const bool double_grid, - ModulePW::PW_Basis* &pw_rho, // pw for rhod - ModulePW::PW_Basis* &pw_rhod); // pw for rhod + const bool double_grid, + ModulePW::PW_Basis* &pw_rho, // pw for rhod + ModulePW::PW_Basis* &pw_rhod); // pw for rhod } diff --git a/source/source_pw/module_pwdft/setup_pwwfc.cpp b/source/source_pw/module_pwdft/setup_pwwfc.cpp index bd99afed03a..56d5f0cef7f 100644 --- a/source/source_pw/module_pwdft/setup_pwwfc.cpp +++ b/source/source_pw/module_pwdft/setup_pwwfc.cpp @@ -1,4 +1,5 @@ #include "source_pw/module_pwdft/setup_pwwfc.h" // pw_wfc +#include "source_basis/module_pw/pw_basis_k_big.h" // PW_Basis_K_Big #include "source_base/parallel_comm.h" // POOL_WORLD #include "source_base/parallel_reduce.h" // Parallel_Reduce #include "source_io/module_output/print_info.h" // print information @@ -6,14 +7,14 @@ void pw::teardown_pwwfc(ModulePW::PW_Basis_K* &pw_wfc) { - delete pw_wfc; + delete pw_wfc; } void pw::setup_pwwfc(const Input_para& inp, - const UnitCell& ucell, - const ModulePW::PW_Basis& pw_rho, - K_Vectors& kv, - ModulePW::PW_Basis_K* &pw_wfc) + const UnitCell& ucell, + const ModulePW::PW_Basis& pw_rho, + K_Vectors& kv, + ModulePW::PW_Basis_K* &pw_wfc) { ModuleBase::TITLE("pw", "pw_setup"); @@ -47,11 +48,20 @@ void pw::setup_pwwfc(const Input_para& inp, pw_wfc->initmpi(GlobalV::NPROC_IN_POOL, GlobalV::RANK_IN_POOL, POOL_WORLD); #endif - pw_wfc->initgrids(inp.ref_cell_factor * ucell.lat0, - ucell.latvec, - pw_rho.nx, - pw_rho.ny, - pw_rho.nz); + // NOTE(liuyu): ref_cell_factor is currently forced to 1.0 in + // read_input_item_md.cpp because the reference-cell mechanism is + // disabled for both pw_rho and pw_wfc. The wfc FFT grid shares the + // same nx/ny/nz as pw_rho, so the same staleness issue applies: + // when ref_cell_factor > 1, pw_wfc->lat0/tpiba/G/GGT/omega hold + // reference-cell values, which leaks into wfc IO (read_wfc_pw, + // write_wfc_pw), cal_energies, and other paths that read these + // members as physical-cell quantities. If re-enabled in the future, + // see the comment in setup_pwrho.cpp for the required refactor. + pw_wfc->initgrids(inp.ref_cell_factor * ucell.lat0, + ucell.latvec, + pw_rho.nx, + pw_rho.ny, + pw_rho.nz); pw_wfc->initparameters(false, inp.ecutwfc, kv.get_nks(), kv.kvec_d.data()); #ifdef __MPI diff --git a/source/source_pw/module_pwdft/setup_pwwfc.h b/source/source_pw/module_pwdft/setup_pwwfc.h index d52c0426e0d..b67eb6ceff3 100644 --- a/source/source_pw/module_pwdft/setup_pwwfc.h +++ b/source/source_pw/module_pwdft/setup_pwwfc.h @@ -14,10 +14,10 @@ namespace pw void teardown_pwwfc(ModulePW::PW_Basis_K* &pw_wfc); void setup_pwwfc(const Input_para& inp, - const UnitCell& ucell, - const ModulePW::PW_Basis& pw_rho, - K_Vectors& kv, - ModulePW::PW_Basis_K* &pw_wfc); + const UnitCell& ucell, + const ModulePW::PW_Basis& pw_rho, + K_Vectors& kv, + ModulePW::PW_Basis_K* &pw_wfc); } diff --git a/source/source_pw/module_pwdft/stress_cc.cpp b/source/source_pw/module_pwdft/stress_cc.cpp index ffe5dc85e09..a0b6cb3acb3 100644 --- a/source/source_pw/module_pwdft/stress_cc.cpp +++ b/source/source_pw/module_pwdft/stress_cc.cpp @@ -1,7 +1,6 @@ #include "stress_func.h" #include "source_base/parallel_reduce.h" #include "source_hamilt/module_xc/xc_functional.h" -#include "source_io/module_parameter/parameter.h" #include "source_base/math_integral.h" #include "source_base/timer.h" #include "source_cell/cal_ux.h" @@ -15,18 +14,23 @@ template void Stress_Func::stress_cc(ModuleBase::matrix& sigma, ModulePW::PW_Basis* rho_basis, - UnitCell& ucell, + UnitCell& ucell, const Structure_Factor* p_sf, const bool is_pw, - const bool *numeric, - const Charge* const chr) + const bool *numeric, + const Charge* const chr, + const int nspin, + const bool domag, + const bool domag_z, + const int gga_grad, + const bool gamma_only_pw) { ModuleBase::TITLE("Stress","stress_cc"); ModuleBase::timer::start("Stress","stress_cc"); FPTYPE fact=1.0; - if(is_pw&&PARAM.globalv.gamma_only_pw) + if(is_pw&&gamma_only_pw) { fact = 2.0; //is_pw:PW basis, gamma_only need to FPTYPE. } @@ -62,7 +66,7 @@ void Stress_Func::stress_cc(ModuleBase::matrix& sigma, #ifdef __LIBXC const auto etxc_vtxc_v = XC_Functional_Libxc::v_xc_meta(XC_Functional::get_func_id(), rho_basis->nrxx, ucell.omega, ucell.tpiba, chr, - PARAM.inp.nspin, hybrid_alpha, hse_omega); + nspin, hybrid_alpha, hse_omega); // etxc = std::get<0>(etxc_vtxc_v); // vtxc = std::get<1>(etxc_vtxc_v); @@ -73,11 +77,12 @@ void Stress_Func::stress_cc(ModuleBase::matrix& sigma, } else { - unitcell::cal_ux(ucell, PARAM.inp.nspin); + unitcell::cal_ux(ucell, nspin); const auto etxc_vtxc_v = XC_Functional::v_xc(rho_basis->nrxx, chr, &ucell, - PARAM.inp.nspin, - PARAM.globalv.domag, - PARAM.globalv.domag_z, + nspin, + domag, + domag_z, + gga_grad, hybrid_alpha, hse_omega); // etxc = std::get<0>(etxc_vtxc_v); // may delete? @@ -87,7 +92,7 @@ void Stress_Func::stress_cc(ModuleBase::matrix& sigma, std::complex* psic = new std::complex[rho_basis->nmaxgr]; - if(PARAM.inp.nspin==1||PARAM.inp.nspin==4) + if(nspin==1||nspin==4) { #ifdef _OPENMP #pragma omp parallel for schedule(static, 1024) diff --git a/source/source_pw/module_pwdft/stress_ewa.cpp b/source/source_pw/module_pwdft/stress_ewa.cpp index 8188ccfc8da..8134da9d4c2 100644 --- a/source/source_pw/module_pwdft/stress_ewa.cpp +++ b/source/source_pw/module_pwdft/stress_ewa.cpp @@ -13,101 +13,101 @@ //calcualte the Ewald stress term in PW and LCAO template void Stress_Func::stress_ewa(const UnitCell& ucell, - ModuleBase::matrix& sigma, - ModulePW::PW_Basis* rho_basis, - const bool is_pw) + ModuleBase::matrix& sigma, + ModulePW::PW_Basis* rho_basis, + const bool is_pw) { ModuleBase::TITLE("Stress","stress_ewa"); ModuleBase::timer::start("Stress","stress_ewa"); FPTYPE charge=0; for(int it=0; it < ucell.ntype; it++) - { - charge = charge + ucell.atoms[it].ncpp.zv * ucell.atoms[it].na; - } + { + charge = charge + ucell.atoms[it].ncpp.zv * ucell.atoms[it].na; + } //choose alpha in order to have convergence in the sum over G //upperbound is a safe upper bound for the error ON THE ENERGY FPTYPE alpha=2.9; FPTYPE upperbound=0.0; - do{ - alpha-=0.1; - if(alpha==0.0) - { - ModuleBase::WARNING_QUIT("stres_ew", "optimal alpha not found"); - } - upperbound =ModuleBase::e2 * pow(charge,2) * + do{ + alpha-=0.1; + if(alpha==0.0) + { + ModuleBase::WARNING_QUIT("stres_ew", "optimal alpha not found"); + } + upperbound =ModuleBase::e2 * pow(charge,2) * sqrt( 2 * alpha / (ModuleBase::TWO_PI)) * erfc(sqrt(ucell.tpiba2 * rho_basis->ggecut / 4.0 / alpha)); - } + } while(upperbound>1e-7); //G-space sum here //Determine if this processor contains G=0 and set the constant term FPTYPE sdewald=0.0; - const int ig0 = rho_basis->ig_gge0; + const int ig0 = rho_basis->ig_gge0; if( ig0 >= 0) - { + { sdewald = (ModuleBase::TWO_PI) * ModuleBase::e2 / 4.0 / alpha * pow(charge/ucell.omega,2); } else - { + { sdewald = 0.0; } //sdewald is the diagonal term FPTYPE fact=1.0; - if (PARAM.globalv.gamma_only_pw && is_pw) - { - fact=2.0; - } + if (PARAM.globalv.gamma_only_pw && is_pw) + { + fact=2.0; + } // else fact=1.0; #pragma omp parallel { - ModuleBase::matrix local_sigma(3, 3); - FPTYPE local_sdewald = 0; + ModuleBase::matrix local_sigma(3, 3); + FPTYPE local_sdewald = 0; FPTYPE g2,g2a; FPTYPE arg; std::complex rhostar; FPTYPE sewald; - #pragma omp for + #pragma omp for for(int ig = 0; ig < rho_basis->npw; ig++) - { - if(ig == ig0) - { - continue; - } - g2 = rho_basis->gg[ig]* ucell.tpiba2; - g2a = g2 /4.0/alpha; - rhostar=std::complex(0.0,0.0); - - for(int it=0; it < ucell.ntype; it++) - { - for(int i=0; igcar[ig] * ucell.atoms[it].tau[i]) * (ModuleBase::TWO_PI); - FPTYPE sinp, cosp; + { + if(ig == ig0) + { + continue; + } + g2 = rho_basis->gg[ig]* ucell.tpiba2; + g2a = g2 /4.0/alpha; + rhostar=std::complex(0.0,0.0); + + for(int it=0; it < ucell.ntype; it++) + { + for(int i=0; igcar[ig] * ucell.atoms[it].tau[i]) * (ModuleBase::TWO_PI); + FPTYPE sinp, cosp; ModuleBase::libm::sincos(arg, &sinp, &cosp); - rhostar = rhostar + std::complex(ucell.atoms[it].ncpp.zv * cosp,ucell.atoms[it].ncpp.zv * sinp); - } - } - rhostar /= ucell.omega; - sewald = fact* (ModuleBase::TWO_PI) * ModuleBase::e2 * ModuleBase::libm::exp(-g2a) / g2 * pow(std::abs(rhostar),2); - local_sdewald -= sewald; - for(int l=0;l<3;l++) - { - for(int m=0;mgcar[ig][l] * rho_basis->gcar[ig][m] / g2 * (g2a + 1); - } - } - } + rhostar = rhostar + std::complex(ucell.atoms[it].ncpp.zv * cosp,ucell.atoms[it].ncpp.zv * sinp); + } + } + rhostar /= ucell.omega; + sewald = fact* (ModuleBase::TWO_PI) * ModuleBase::e2 * ModuleBase::libm::exp(-g2a) / g2 * pow(std::abs(rhostar),2); + local_sdewald -= sewald; + for(int l=0;l<3;l++) + { + for(int m=0;mgcar[ig][l] * rho_basis->gcar[ig][m] / g2 * (g2a + 1); + } + } + } //R-space sum here (only for the processor that contains G=0) int *irr=nullptr; @@ -120,89 +120,89 @@ void Stress_Func::stress_ewa(const UnitCell& ucell, int nrm=0; FPTYPE fac=0.0; - if(ig0 >= 0) - { - FPTYPE sqa = sqrt(alpha); - FPTYPE sq8a_2pi = sqrt(8 * alpha / (ModuleBase::TWO_PI)); - rmax = 4.0/sqa/ucell.lat0; - const int mxr = H_Ewald_pw::estimate_mxr(rmax, ucell.G); - - std::vector> r(mxr); - std::vector r2(mxr); - std::vector irr(mxr); - - #pragma omp for - for(long long ijat = 0; ijat < ucell.nat * ucell.nat; ijat++) - { - int it=0; - int i=0; - int jt=0; - int j=0; - ucell.ijat2iaitjajt(ijat, &i, &it, &j, &jt); - if (ucell.atoms[it].na != 0 && ucell.atoms[jt].na != 0) - { - //calculate tau[na]-tau[nb] - d_tau = ucell.atoms[it].tau[i] - ucell.atoms[jt].tau[j]; - //generates nearest-neighbors shells - H_Ewald_pw::rgen(d_tau, rmax, irr.data(), ucell.latvec, ucell.G, r.data(), r2.data(), mxr, nrm, PARAM.inp.test_energy); - for(int nr=0; nr= 0) + { + FPTYPE sqa = sqrt(alpha); + FPTYPE sq8a_2pi = sqrt(8 * alpha / (ModuleBase::TWO_PI)); + rmax = 4.0/sqa/ucell.lat0; + const int mxr = H_Ewald_pw::estimate_mxr(rmax, ucell.G); + + std::vector> r(mxr); + std::vector r2(mxr); + std::vector irr(mxr); + + #pragma omp for + for(long long ijat = 0; ijat < ucell.nat * ucell.nat; ijat++) + { + int it=0; + int i=0; + int jt=0; + int j=0; + ucell.ijat2iaitjajt(ijat, &i, &it, &j, &jt); + if (ucell.atoms[it].na != 0 && ucell.atoms[jt].na != 0) + { + //calculate tau[na]-tau[nb] + d_tau = ucell.atoms[it].tau[i] - ucell.atoms[jt].tau[j]; + //generates nearest-neighbors shells + H_Ewald_pw::rgen(d_tau, rmax, irr.data(), ucell.latvec, ucell.G, r.data(), r2.data(), mxr, nrm, PARAM.inp.test_energy); + for(int nr=0; nr; diff --git a/source/source_pw/module_pwdft/stress_exx.cpp b/source/source_pw/module_pwdft/stress_exx.cpp index 9e7e99e2d01..72353923cbb 100644 --- a/source/source_pw/module_pwdft/stress_exx.cpp +++ b/source/source_pw/module_pwdft/stress_exx.cpp @@ -2,6 +2,7 @@ #include "op_pw_exx.h" #include "source_base/parallel_common.h" #include "source_base/parallel_reduce.h" +#include "source_hamilt/module_xc/general_exx_info.h" #include "stress_pw.h" template @@ -11,10 +12,9 @@ void Stress_PW::stress_exx(ModuleBase::matrix& sigma, ModulePW::PW_Basis_K* wfcpw, const K_Vectors *p_kv, const psi::Psi , Device>* d_psi_in, const UnitCell& ucell, - const double hybrid_alpha, - const CoulombParam& coulomb_param) + const General_Exx_Info& exx_info) { - bool gamma_extrapolation = PARAM.inp.exx_gamma_extrapolation; + bool gamma_extrapolation = exx_info.gamma_extrapolation; bool is_mp = p_kv->get_is_mp(); #ifdef __MPI Parallel_Common::bcast_bool(is_mp); @@ -42,6 +42,12 @@ void Stress_PW::stress_exx(ModuleBase::matrix& sigma, double tpiba2 = ucell.tpiba2; double omega_inv = 1.0 / omega; + // Consistent with OperatorEXXPW: the EXX pair density only carries the + // ecut_exx G-sphere (rhopw_dev there), so truncate the stress sum to the + // same sphere. gg is in lat0^-2 units and ggecut = ecut / tpiba2. + // exx_info.ecut_exx is already resolved: ecutexx when set, else ecutrho. + const double ggecut_exx = exx_info.ecut_exx / tpiba2; + // allocate space T* psi_nk_real = nullptr; T* psi_mq_real = nullptr; @@ -75,8 +81,8 @@ void Stress_PW::stress_exx(ModuleBase::matrix& sigma, for (int iq = 0; iq < nqs; iq++) { - hamilt::get_exx_potential(p_kv, wfcpw, rhopw, pot, tpiba, gamma_extrapolation, omega, ik, iq, true, coulomb_param); - hamilt::get_exx_stress_potential(p_kv, wfcpw, rhopw, pot_stress, tpiba, gamma_extrapolation, omega, ik, iq, coulomb_param); + hamilt::get_exx_potential(p_kv, wfcpw, rhopw, pot, tpiba, gamma_extrapolation, omega, ik, iq, true, exx_info.coulomb_param); + hamilt::get_exx_stress_potential(p_kv, wfcpw, rhopw, pot_stress, tpiba, gamma_extrapolation, omega, ik, iq, exx_info.coulomb_param); for (int mband = 0; mband < d_psi_in->get_nbands(); mband++) { // psi_mq in real space @@ -108,6 +114,10 @@ void Stress_PW::stress_exx(ModuleBase::matrix& sigma, #endif for (int ig = 0; ig < rhopw->npw; ig++) { + if (rhopw->gg[ig] > ggecut_exx) + { + continue; // outside the ecut_exx sphere, see above + } const ModuleBase::Vector3 kqg = wfcpw->kvec_c[ik] - wfcpw->kvec_c[iq] + rhopw->gcar[ig]; double kqg_alpha = kqg[alpha] * tpiba; double kqg_beta = kqg[beta] * tpiba; @@ -120,9 +130,13 @@ void Stress_PW::stress_exx(ModuleBase::matrix& sigma, } - sigma(alpha, beta) -= hybrid_alpha + // pure occupations f = wg / wk; divide by the q-mesh + // size nqs to apply the uniform q-point BZ weight, + // and by wk[ik] to convert wg(ik) into the occupation. + const double exx_weight = wg(ik, nband) * wg(iq, mband) / nqs / p_kv->wk[ik]; + sigma(alpha, beta) -= exx_info.hybrid_alpha * 0.25 * sigma_ab_loc - * wg(ik, nband) * wg(iq, mband) / nqs / p_kv->wk[ik]; + * exx_weight; } } } diff --git a/source/source_pw/module_pwdft/stress_func.h b/source/source_pw/module_pwdft/stress_func.h index 9ac596cb63c..4eb758999b8 100644 --- a/source/source_pw/module_pwdft/stress_func.h +++ b/source/source_pw/module_pwdft/stress_func.h @@ -16,6 +16,8 @@ #include "source_psi/psi.h" #include "source_pw/module_pwdft/dftu_base.h" +class pseudopot_cell_vl; + //------------------------------------------------------------------- // mohan reconstruction note: 2021-02-07 // the stress code needs reconstructions (by Daye Zheng) @@ -117,7 +119,12 @@ class Stress_Func const Structure_Factor* p_sf, const bool is_pw, const bool *numeric, - const Charge* const chr); // nonlinear core correction stress in PW or LCAO basis + const Charge* const chr, + const int nspin, + const bool domag, + const bool domag_z, + const int gga_grad, + const bool gamma_only_pw); // nonlinear core correction stress in PW or LCAO basis void deriv_drhoc(const bool& numeric, const double& omega, @@ -134,7 +141,11 @@ class Stress_Func void stress_gga(const UnitCell& ucell, ModuleBase::matrix& sigma, ModulePW::PW_Basis* rho_basis, - const Charge* const chr); // gga part in both PW and LCAO basis + const Charge* const chr, + const int nspin, + const bool domag, + const bool domag_z, + const int gga_grad); // gga part in both PW and LCAO basis void stress_mgga(const UnitCell& ucell, ModuleBase::matrix& sigma, const ModuleBase::matrix& wg, @@ -144,6 +155,31 @@ class Stress_Func ModulePW::PW_Basis_K* wfc_basis, const psi::Psi , Device>* psi_in); // gga part in PW basis + /** + * @brief Compute the plane-wave-basis stress terms shared by PW and LCAO: + * vlocal, hartree, ewald, non-linear core correction and exchange-correlation. + * + * This is the former LCAO_domain::cal_stress_pw, moved here so the PW-part + * stress assembly lives in the PW module. All terms are double precision and + * independent of the electronic template type T used by LCAO. + */ + void stress_pw_terms(UnitCell& ucell, + ModuleBase::matrix& sigmadvl, + ModuleBase::matrix& sigmahar, + ModuleBase::matrix& sigmaewa, + ModuleBase::matrix& sigmacc, + ModuleBase::matrix& sigmaxc, + const double& etxc, + const Charge* const chr, + ModulePW::PW_Basis* rhopw, + const pseudopot_cell_vl& locpp, + const Structure_Factor& sf, + const int nspin, + const bool domag, + const bool domag_z, + const int gga_grad, + const bool gamma_only_pw); + // 7) the stress from the non-local pseudopotentials /** * @brief This routine computes the atomic force of non-local pseudopotential diff --git a/source/source_pw/module_pwdft/stress_gga.cpp b/source/source_pw/module_pwdft/stress_gga.cpp index bb175514803..1716bd093db 100644 --- a/source/source_pw/module_pwdft/stress_gga.cpp +++ b/source/source_pw/module_pwdft/stress_gga.cpp @@ -1,14 +1,17 @@ #include "stress_func.h" #include "source_base/parallel_reduce.h" #include "source_hamilt/module_xc/xc_functional.h" -#include "source_io/module_parameter/parameter.h" //calculate the GGA stress correction in PW and LCAO template void Stress_Func::stress_gga(const UnitCell& ucell, ModuleBase::matrix& sigma, ModulePW::PW_Basis* rho_basis, - const Charge* const chr) + const Charge* const chr, + const int nspin, + const bool domag, + const bool domag_z, + const int gga_grad) { ModuleBase::TITLE("Stress","stress_gga"); ModuleBase::timer::start("Stress","stress_gga"); @@ -31,7 +34,7 @@ void Stress_Func::stress_gga(const UnitCell& ucell, XC_Functional::gradcorr( dum1, dum2, dum3, chr, rho_basis, &ucell, stress_gga, is_stress, - PARAM.inp.nspin, PARAM.globalv.domag, PARAM.globalv.domag_z, + nspin, domag, domag_z, gga_grad, hybrid_alpha, hse_omega); for(int l = 0;l< 3;l++) diff --git a/source/source_pw/module_pwdft/stress_har.cpp b/source/source_pw/module_pwdft/stress_har.cpp index 4f01c03b503..24d572d3a5e 100644 --- a/source/source_pw/module_pwdft/stress_har.cpp +++ b/source/source_pw/module_pwdft/stress_har.cpp @@ -7,21 +7,21 @@ //calculate the Hartree part in PW or LCAO base template void Stress_Func::stress_har(const UnitCell& ucell, - ModuleBase::matrix& sigma, - ModulePW::PW_Basis* rho_basis, - const bool is_pw, - const Charge* const chr) + ModuleBase::matrix& sigma, + ModulePW::PW_Basis* rho_basis, + const bool is_pw, + const Charge* const chr) { ModuleBase::TITLE("Stress","stress_har"); - ModuleBase::timer::start("Stress","stress_har"); + ModuleBase::timer::start("Stress","stress_har"); assert(rho_basis->nmaxgr>0); - std::complex *aux = new std::complex[rho_basis->nmaxgr]; + std::complex *aux = new std::complex[rho_basis->nmaxgr]; - const int nspin_rho = (PARAM.inp.nspin == 2) ? 2 : 1; + const int nspin_rho = (PARAM.inp.nspin == 2) ? 2 : 1; - // Hartree potential VH(r) from n(r) + // Hartree potential VH(r) from n(r) /* blocking rho_basis->nrxx for data locality. @@ -33,125 +33,125 @@ void Stress_Func::stress_har(const UnitCell& ucell, #ifdef _OPENMP #pragma omp parallel for #endif - for (int irb = 0; irb < rho_basis->nrxx; irb += block_ir) - { - // calculate the actual task length of this block - int ir_end = std::min(irb + block_ir, rho_basis->nrxx); + for (int irb = 0; irb < rho_basis->nrxx; irb += block_ir) + { + // calculate the actual task length of this block + int ir_end = std::min(irb + block_ir, rho_basis->nrxx); - { // is = 0 - for (int ir = irb; ir < ir_end; ++ir) - { // initialize aux - aux[ir] = std::complex( chr->rho[0][ir], 0.0 ); - } - } - for (int is = 1; is < nspin_rho; is++) - { - for (int ir = irb; ir < ir_end; ++ir) - { // accumulate aux - aux[ir] += std::complex( chr->rho[is][ir], 0.0 ); - } - } - } - //============================= - // bring rho (aux) to G space - //============================= - rho_basis->real2recip(aux, aux); + { // is = 0 + for (int ir = irb; ir < ir_end; ++ir) + { // initialize aux + aux[ir] = std::complex( chr->rho[0][ir], 0.0 ); + } + } + for (int is = 1; is < nspin_rho; is++) + { + for (int ir = irb; ir < ir_end; ++ir) + { // accumulate aux + aux[ir] += std::complex( chr->rho[is][ir], 0.0 ); + } + } + } + //============================= + // bring rho (aux) to G space + //============================= + rho_basis->real2recip(aux, aux); - const int ig0 = rho_basis->ig_gge0; + const int ig0 = rho_basis->ig_gge0; #ifndef _OPENMP - ModuleBase::matrix& local_sigma = sigma; + ModuleBase::matrix& local_sigma = sigma; #else #pragma omp parallel - { - ModuleBase::matrix local_sigma(3, 3); + { + ModuleBase::matrix local_sigma(3, 3); #pragma omp for #endif - for (int ig = 0 ; ig < rho_basis->npw ; ++ig) - { - if (ig == ig0) - { - continue; - } - const FPTYPE g2 = rho_basis->gg[ig]; - FPTYPE shart= ( conj( aux[ig] ) * aux[ig] ).real()/(ucell.tpiba2 * g2); - for(int l=0;l<3;l++) - { - for(int m=0;mgcar[ig][l] * rho_basis->gcar[ig][m] / g2; - } - } - } + for (int ig = 0 ; ig < rho_basis->npw ; ++ig) + { + if (ig == ig0) + { + continue; + } + const FPTYPE g2 = rho_basis->gg[ig]; + FPTYPE shart= ( conj( aux[ig] ) * aux[ig] ).real()/(ucell.tpiba2 * g2); + for(int l=0;l<3;l++) + { + for(int m=0;mgcar[ig][l] * rho_basis->gcar[ig][m] / g2; + } + } + } #ifdef _OPENMP #pragma omp critical(stress_har_reduce) - { - for(int l=0;l<3;l++) - { - for(int m=0;m; diff --git a/source/source_pw/module_pwdft/stress_kin.cpp b/source/source_pw/module_pwdft/stress_kin.cpp index ba30034d877..048a7c73f1d 100644 --- a/source/source_pw/module_pwdft/stress_kin.cpp +++ b/source/source_pw/module_pwdft/stress_kin.cpp @@ -14,11 +14,11 @@ void Stress_Func::stress_kin(ModuleBase::matrix& sigma, const psi::Psi , Device>* psi_in) { ModuleBase::TITLE("Stress","stress_kin"); - ModuleBase::timer::start("Stress","stress_kin"); + ModuleBase::timer::start("Stress","stress_kin"); - this->ucell = &ucell_in; + this->ucell = &ucell_in; - hamilt::FS_Kin_tools kin_tool(*this->ucell, p_kv, wfc_basis, wg); + hamilt::FS_Kin_tools kin_tool(*this->ucell, p_kv, wfc_basis, wg); for (int ik = 0; ik < wfc_basis->nks; ++ik) { int nbands_occ = wg.nc; @@ -35,9 +35,9 @@ void Stress_Func::stress_kin(ModuleBase::matrix& sigma, kin_tool.cal_stress_kin(ik, npm, true, &psi_in[0](ik, 0, 0)); } kin_tool.symmetrize_stress(p_symm, sigma); - - ModuleBase::timer::end("Stress","stress_kin"); - return; + + ModuleBase::timer::end("Stress","stress_kin"); + return; } template class Stress_Func; diff --git a/source/source_pw/module_pwdft/stress_loc.cpp b/source/source_pw/module_pwdft/stress_loc.cpp index 29fa1ee0900..94f1746dbe0 100644 --- a/source/source_pw/module_pwdft/stress_loc.cpp +++ b/source/source_pw/module_pwdft/stress_loc.cpp @@ -1,4 +1,5 @@ #include "stress_func.h" +#include "source_pw/module_pwdft/vl_pw.h" #include "source_base/parallel_reduce.h" #include "source_base/math_integral.h" #include "source_io/module_parameter/parameter.h" @@ -300,6 +301,52 @@ void Stress_Func::dvloc_coulomb(const UnitCell& ucell, return; } +// plane-wave-basis stress terms shared by PW and LCAO: +// vlocal, hartree, ewald, non-linear core correction and exchange-correlation. +template +void Stress_Func::stress_pw_terms(UnitCell& ucell, + ModuleBase::matrix& sigmadvl, + ModuleBase::matrix& sigmahar, + ModuleBase::matrix& sigmaewa, + ModuleBase::matrix& sigmacc, + ModuleBase::matrix& sigmaxc, + const double& etxc, + const Charge* const chr, + ModulePW::PW_Basis* rhopw, + const pseudopot_cell_vl& locpp, + const Structure_Factor& sf, + const int nspin, + const bool domag, + const bool domag_z, + const int gga_grad, + const bool gamma_only_pw) +{ + ModuleBase::TITLE("Stress", "stress_pw_terms"); + + // local pseudopotential stress: + this->stress_loc(ucell, sigmadvl, rhopw, locpp.vloc, &sf, 0, chr); + + // hartree term + this->stress_har(ucell, sigmahar, rhopw, 0, chr); + + // ewald stress: use plane wave only. + this->stress_ewa(ucell, sigmaewa, rhopw, 0); // remain problem + + // stress due to core correlation. + this->stress_cc(sigmacc, rhopw, ucell, &sf, 0, locpp.numeric, chr, + nspin, domag, domag_z, gga_grad, gamma_only_pw); + + // stress due to self-consistent charge. + for (int i = 0; i < 3; i++) + { + sigmaxc(i, i) = -etxc / ucell.omega; + } + // Exchange-correlation for PBE + this->stress_gga(ucell, sigmaxc, rhopw, chr, nspin, domag, domag_z, gga_grad); + + return; +} + template class Stress_Func; #if ((defined __CUDA) || (defined __ROCM)) template class Stress_Func; diff --git a/source/source_pw/module_pwdft/stress_mgga.cpp b/source/source_pw/module_pwdft/stress_mgga.cpp index 1ee187a4e41..7ecbe1cb914 100644 --- a/source/source_pw/module_pwdft/stress_mgga.cpp +++ b/source/source_pw/module_pwdft/stress_mgga.cpp @@ -1,4 +1,5 @@ #include "source_base/timer.h" +#include "source_estate/module_charge/chg_parallel.h" #include "source_hamilt/module_xc/xc_functional.h" #include "source_io/module_parameter/parameter.h" #include "stress_func.h" @@ -18,10 +19,10 @@ void Stress_Func::stress_mgga(const UnitCell& ucell, ModulePW::PW_Basis_K* wfc_basis, const psi::Psi , Device>* psi_in) { - if (PARAM.inp.nspin == 4) - { - ModuleBase::WARNING_QUIT("stress_mgga", "noncollinear stress + mGGA not implemented"); - } + if (PARAM.inp.nspin == 4) + { + ModuleBase::WARNING_QUIT("stress_mgga", "noncollinear stress + mGGA not implemented"); + } ModuleBase::timer::start("Stress", "stress_mgga"); @@ -49,27 +50,27 @@ void Stress_Func::stress_mgga(const UnitCell& ucell, for (int ik = 0; ik < p_kv->get_nks(); ik++) { if (PARAM.inp.nspin == 2) - { - current_spin = p_kv->isk[ik]; - } + { + current_spin = p_kv->isk[ik]; + } const int npw = p_kv->ngk[ik]; for (int ibnd = 0; ibnd < PARAM.inp.nbands; ibnd++) { const FPTYPE w1 = wg(ik, ibnd) / ucell.omega; const std::complex* psi = &psi_in[0](ik, ibnd, 0); - XC_Functional::grad_wfc, Device>(ik, - ucell.tpiba, - wfc_basis, - psi, - gradwfc.data>()); + XC_Functional::grad_wfc, Device>(ik, + ucell.tpiba, + wfc_basis, + psi, + gradwfc.data>()); - cal_stress_mgga_solver( - current_spin, - nrxx, - w1, - gradwfc.data>(), - crosstaus.data()); + cal_stress_mgga_solver( + current_spin, + nrxx, + w1, + gradwfc.data>(), + crosstaus.data()); } // band loop // delete[] psi; } // k loop @@ -80,7 +81,11 @@ void Stress_Func::stress_mgga(const UnitCell& ucell, { for (int ipol = 0; ipol < 6; ++ipol) { - chr->reduce_diff_pools(&crosstaus_pack[is][ipol * nrxx]); + module_charge::reduce_diff_pools(&crosstaus_pack[is][ipol * nrxx], + *chr, + GlobalV::KPAR, + PARAM.globalv.all_ks_run, + PARAM.inp.bndpar); } } #endif @@ -100,11 +105,11 @@ void Stress_Func::stress_mgga(const UnitCell& ucell, for (int iy = 0; iy < 3; iy++) { FPTYPE delta = 0.0; - if (ix == iy) - { - delta = 1.0; - } - for (int ir = 0; ir < nrxx; ir++) + if (ix == iy) + { + delta = 1.0; + } + for (int ir = 0; ir < nrxx; ir++) { FPTYPE x = v_ofk(is, ir) * (chr->kin_r[is][ir] * delta diff --git a/source/source_pw/module_pwdft/stress_nl.cpp b/source/source_pw/module_pwdft/stress_nl.cpp index 7b9b84925cb..21503be82b6 100644 --- a/source/source_pw/module_pwdft/stress_nl.cpp +++ b/source/source_pw/module_pwdft/stress_nl.cpp @@ -5,7 +5,7 @@ #include "source_base/module_device/device.h" #include "source_base/timer.h" #include "source_pw/module_pwdft/fs_nonlocal_tools.h" -#include "source_pw/module_pwdft/nonlocal_maths.hpp" +#include "source_pw/module_pwdft/nonlocal_maths.h" #include "stress_func.h" // calculate the nonlocal pseudopotential stress in PW template @@ -107,10 +107,10 @@ void Stress_Func::get_dvnl1(ModuleBase::ComplexMatrix& vkb, Structure_Factor* p_sf, ModulePW::PW_Basis_K* wfc_basis) { - if (PARAM.inp.test_pp) - { - ModuleBase::TITLE("Stress", "get_dvnl1"); - } + if (PARAM.inp.test_pp) + { + ModuleBase::TITLE("Stress", "get_dvnl1"); + } const int npw = wfc_basis->npwk[ik]; const int lmaxkb = nlpp->lmaxkb; @@ -146,25 +146,25 @@ void Stress_Func::get_dvnl1(ModuleBase::ComplexMatrix& vkb, int jkb = 0; for (int it = 0; it < this->ucell->ntype; it++) { - if (PARAM.inp.test_pp > 1) - { - ModuleBase::GlobalFunc::OUT("it", it); - } - // calculate beta in G-space using an interpolation table + if (PARAM.inp.test_pp > 1) + { + ModuleBase::GlobalFunc::OUT("it", it); + } + // calculate beta in G-space using an interpolation table const int nbeta = this->ucell->atoms[it].ncpp.nbeta; const int nh = this->ucell->atoms[it].ncpp.nh; - if (PARAM.inp.test_pp > 1) - { - ModuleBase::GlobalFunc::OUT("nbeta", nbeta); - } + if (PARAM.inp.test_pp > 1) + { + ModuleBase::GlobalFunc::OUT("nbeta", nbeta); + } for (int nb = 0; nb < nbeta; nb++) { - if (PARAM.inp.test_pp > 1) - { - ModuleBase::GlobalFunc::OUT("ib", nb); - } + if (PARAM.inp.test_pp > 1) + { + ModuleBase::GlobalFunc::OUT("ib", nb); + } #ifdef _OPENMP #pragma omp parallel for #endif @@ -233,10 +233,10 @@ void Stress_Func::get_dvnl2(ModuleBase::ComplexMatrix& vkb, Structure_Factor* p_sf, ModulePW::PW_Basis_K* wfc_basis) { - if (PARAM.inp.test_pp) - { - ModuleBase::TITLE("Stress", "get_dvnl2"); - } + if (PARAM.inp.test_pp) + { + ModuleBase::TITLE("Stress", "get_dvnl2"); + } // ModuleBase::timer::start("Stress","get_dvnl2"); const int npw = wfc_basis->npwk[ik]; const int lmaxkb = nlpp->lmaxkb; @@ -271,24 +271,24 @@ void Stress_Func::get_dvnl2(ModuleBase::ComplexMatrix& vkb, for (int it = 0; it < this->ucell->ntype; it++) { if (PARAM.inp.test_pp > 1) - { - ModuleBase::GlobalFunc::OUT("it", it); - } + { + ModuleBase::GlobalFunc::OUT("it", it); + } // calculate beta in G-space using an interpolation table const int nbeta = this->ucell->atoms[it].ncpp.nbeta; const int nh = this->ucell->atoms[it].ncpp.nh; - if (PARAM.inp.test_pp > 1) - { - ModuleBase::GlobalFunc::OUT("nbeta", nbeta); - } + if (PARAM.inp.test_pp > 1) + { + ModuleBase::GlobalFunc::OUT("nbeta", nbeta); + } for (int nb = 0; nb < nbeta; nb++) { - if (PARAM.inp.test_pp > 1) - { - ModuleBase::GlobalFunc::OUT("ib", nb); - } + if (PARAM.inp.test_pp > 1) + { + ModuleBase::GlobalFunc::OUT("ib", nb); + } #ifdef _OPENMP #pragma omp parallel for #endif diff --git a/source/source_pw/module_pwdft/stress_onsite.cpp b/source/source_pw/module_pwdft/stress_onsite.cpp index f0ac36f8e3c..3412cfee6c2 100644 --- a/source/source_pw/module_pwdft/stress_onsite.cpp +++ b/source/source_pw/module_pwdft/stress_onsite.cpp @@ -1,7 +1,7 @@ #include "source_base/module_device/device.h" #include "source_base/parallel_reduce.h" #include "source_base/timer.h" -#include "source_pw/module_pwdft/onsite_proj.h" +#include "source_pw/module_proj/onsite_proj.h" #include "source_io/module_parameter/parameter.h" #include "source_pw/module_pwdft/dftu_base.h" #include "source_lcao/module_deltaspin/spin_constrain.h" @@ -106,7 +106,7 @@ void Stress_Func::stress_onsite( sigma_onsite[idx] += dftu_stress; #ifdef __DEBUG - std::cout << " idx=" << idx << " stress=" << sigma_onsite[idx] << std::endl; + std::cout << " idx=" << idx << " stress=" << sigma_onsite[idx] << std::endl; #endif } @@ -155,7 +155,7 @@ void Stress_Func::stress_onsite( // Add to total stress for (int idx = 0; idx < 9; idx++) { - std::cout << " idx=" << idx << " stress=" << sigma_onsite[idx] << std::endl; + std::cout << " idx=" << idx << " stress=" << sigma_onsite[idx] << std::endl; } #endif diff --git a/source/source_pw/module_pwdft/stress_pw.cpp b/source/source_pw/module_pwdft/stress_pw.cpp index bcf6aff8db2..7571d9a44b0 100644 --- a/source/source_pw/module_pwdft/stress_pw.cpp +++ b/source/source_pw/module_pwdft/stress_pw.cpp @@ -9,7 +9,12 @@ #include "source_hamilt/module_xc/general_exx_info.h" // for General_Exx_Info type template -void Stress_PW::cal_stress(ModuleBase::matrix& sigmatot, +void Stress_PW::cal_stress(const int nspin, + const bool domag, + const bool domag_z, + const int gga_grad, + const bool gamma_only_pw, + ModuleBase::matrix& sigmatot, UnitCell& ucell, const vdw::VdwResult* vdw_result, Plus_U_Base& dftu, @@ -91,7 +96,8 @@ void Stress_PW::cal_stress(ModuleBase::matrix& sigmatot, { sigmaxc(i, i) = -(pelec->f_en.etxc - pelec->f_en.vtxc) / ucell.omega; } - this->stress_gga(ucell, sigmaxc, rho_basis, pelec->charge); + this->stress_gga(ucell, sigmaxc, rho_basis, pelec->charge, + nspin, domag, domag_z, gga_grad); if (XC_Functional::get_ked_flag()) { this->stress_mgga(ucell, @@ -108,7 +114,8 @@ void Stress_PW::cal_stress(ModuleBase::matrix& sigmatot, this->stress_loc(ucell, sigmaloc, rho_basis, locpp.vloc, p_sf, 1, pelec->charge); // nlcc - this->stress_cc(sigmaxcc, rho_basis, ucell, p_sf, 1, locpp.numeric, pelec->charge); + this->stress_cc(sigmaxcc, rho_basis, ucell, p_sf, 1, locpp.numeric, pelec->charge, + nspin, domag, domag_z, gga_grad, gamma_only_pw); // nonlocal this->stress_nl(sigmanl, this->pelec->wg, this->pelec->ekb, p_sf, p_kv, p_symm, wfc_basis, d_psi_in, nlpp, ucell); @@ -136,12 +143,9 @@ void Stress_PW::cal_stress(ModuleBase::matrix& sigmatot, } // EXX PW stress - bool cal_exx = exx_info.cal_exx; - double hybrid_alpha = exx_info.hybrid_alpha; - auto coulomb_param = exx_info.coulomb_param; - if (cal_exx) + if (exx_info.cal_exx) { - this->stress_exx(sigmaexx, this->pelec->wg, rho_basis, wfc_basis, p_kv, d_psi_in, ucell, hybrid_alpha, coulomb_param); + this->stress_exx(sigmaexx, this->pelec->wg, rho_basis, wfc_basis, p_kv, d_psi_in, ucell, exx_info); } @@ -182,7 +186,7 @@ void Stress_PW::cal_stress(ModuleBase::matrix& sigmatot, { ModuleIO::print_stress("ONSITE STRESS", sigmaonsite, screen, ry, GlobalV::ofs_running); } - if (cal_exx) + if (exx_info.cal_exx) { ModuleIO::print_stress("EXX STRESS", sigmaexx, screen, ry, GlobalV::ofs_running); } diff --git a/source/source_pw/module_pwdft/stress_pw.h b/source/source_pw/module_pwdft/stress_pw.h index f203a642e71..976aeef2ab2 100644 --- a/source/source_pw/module_pwdft/stress_pw.h +++ b/source/source_pw/module_pwdft/stress_pw.h @@ -21,8 +21,13 @@ class Stress_PW : public Stress_Func Stress_PW(const elecstate::ElecState* pelec_in) : pelec(pelec_in){}; // calculate the stress in PW basis - void cal_stress(ModuleBase::matrix& smearing_sigmatot, - UnitCell& ucell, + void cal_stress(const int nspin, + const bool domag, + const bool domag_z, + const int gga_grad, + const bool gamma_only_pw, + ModuleBase::matrix& smearing_sigmatot, + UnitCell& ucell, const vdw::VdwResult* vdw_result, Plus_U_Base& dftu, const pseudopot_cell_vl& locpp, @@ -52,8 +57,7 @@ class Stress_PW : public Stress_Func const K_Vectors* p_kv, const psi::Psi , Device>* d_psi_in, const UnitCell& ucell, - const double hybrid_alpha, - const CoulombParam& coulomb_param); // exx stress in PW basis + const General_Exx_Info& exx_info); // exx stress in PW basis const elecstate::ElecState* pelec = nullptr; }; diff --git a/source/source_pw/module_pwdft/stress_us.cpp b/source/source_pw/module_pwdft/stress_us.cpp index e294640f470..1d333598d23 100644 --- a/source/source_pw/module_pwdft/stress_us.cpp +++ b/source/source_pw/module_pwdft/stress_us.cpp @@ -8,7 +8,7 @@ #include "source_base/math_ylmreal.h" #include "source_base/timer.h" #include "source_estate/elecstate_pw.h" -#include "source_pw/module_pwdft/nonlocal_maths.hpp" +#include "source_pw/module_pwdft/nonlocal_maths.h" #include "stress_pw.h" // computes the part of the crystal stress which is due @@ -210,10 +210,10 @@ void Stress_Func::dqvan2(const pseudopot_cell_vnl& nlpp, const ModuleBase::matrix& dylmk0, std::complex* dqg) { - if (PARAM.inp.test_pp) - { - ModuleBase::TITLE("Stress", "dqvan2"); - } + if (PARAM.inp.test_pp) + { + ModuleBase::TITLE("Stress", "dqvan2"); + } // computes the indices which correspond to ih,jh const int nb = nlpp.indv(itype, ih); diff --git a/source/source_pw/module_pwdft/stru_fac.cpp b/source/source_pw/module_pwdft/stru_fac.cpp index e98c8ef7497..20531268038 100644 --- a/source/source_pw/module_pwdft/stru_fac.cpp +++ b/source/source_pw/module_pwdft/stru_fac.cpp @@ -53,7 +53,10 @@ void Structure_Factor::set(const ModulePW::PW_Basis* rho_basis_in, const int& nb // Peize Lin optimize and add OpenMP 2021.04.01 // Calculate structure factor -void Structure_Factor::setup(const UnitCell* Ucell, const Parallel_Grid& pgrid, const ModulePW::PW_Basis* rho_basis) +void Structure_Factor::setup(const UnitCell* Ucell, + const Parallel_Grid& pgrid, + const ModulePW::PW_Basis* rho_basis, + const bool has_float_data) { ModuleBase::TITLE("Structure_Factor","setup"); ModuleBase::timer::start("Structure_Factor","setup"); @@ -161,7 +164,7 @@ void Structure_Factor::setup(const UnitCell* Ucell, const Parallel_Grid& pgrid, } if (device == "gpu") { - if (PARAM.globalv.has_float_data) { + if (has_float_data) { resmem_cd_op()(this->c_eigts1, Ucell->nat * (2 * rho_basis->nx + 1)); resmem_cd_op()(this->c_eigts2, Ucell->nat * (2 * rho_basis->ny + 1)); resmem_cd_op()(this->c_eigts3, Ucell->nat * (2 * rho_basis->nz + 1)); @@ -177,7 +180,7 @@ void Structure_Factor::setup(const UnitCell* Ucell, const Parallel_Grid& pgrid, syncmem_z2z_h2d_op()(this->z_eigts3, this->eigts3.c, Ucell->nat * (2 * rho_basis->nz + 1)); } else { - if (PARAM.globalv.has_float_data) { + if (has_float_data) { resmem_ch_op()(this->c_eigts1, Ucell->nat * (2 * rho_basis->nx + 1)); resmem_ch_op()(this->c_eigts2, Ucell->nat * (2 * rho_basis->ny + 1)); resmem_ch_op()(this->c_eigts3, Ucell->nat * (2 * rho_basis->nz + 1)); diff --git a/source/source_pw/module_pwdft/stru_fac.h b/source/source_pw/module_pwdft/stru_fac.h index d9b99b86d0c..39fdf193f6a 100644 --- a/source/source_pw/module_pwdft/stru_fac.h +++ b/source/source_pw/module_pwdft/stru_fac.h @@ -23,9 +23,13 @@ class Structure_Factor // structure factor (ntype, ngmc) ModuleBase::ComplexMatrix strucFac; + /// @brief Calculate structure factors + /// @param has_float_data whether the Hamiltonian carries float data, which + /// decides if the single-precision eigts copies are built as well void setup(const UnitCell* Ucell, const Parallel_Grid& pgrid, - const ModulePW::PW_Basis* rho_basis); // Calculate structure factors + const ModulePW::PW_Basis* rho_basis, + const bool has_float_data); /// calculate structure factors through Cardinal B-spline interpolation void bspline_sf( diff --git a/source/source_pw/module_pwdft/stru_fac_k.cpp b/source/source_pw/module_pwdft/stru_fac_k.cpp index 6223057992d..063d2d1ce36 100644 --- a/source/source_pw/module_pwdft/stru_fac_k.cpp +++ b/source/source_pw/module_pwdft/stru_fac_k.cpp @@ -1,18 +1,17 @@ +#include "source_base/module_device/device.h" #include "source_base/timer.h" #include "source_pw/module_pwdft/kernels/wf_op.h" -#include "source_base/module_device/device.h" #include "stru_fac.h" -std::complex* Structure_Factor::get_sk(const int ik, - const int it, - const int ia, - const ModulePW::PW_Basis_K* wfc_basis) const +#include + +std::complex* Structure_Factor::get_sk(const int ik, const int it, const int ia, const ModulePW::PW_Basis_K* wfc_basis) const { ModuleBase::timer::start("Structure_Factor", "get_sk"); const double arg = (wfc_basis->kvec_c[ik] * ucell->atoms[it].tau[ia]) * ModuleBase::TWO_PI; const std::complex kphase = std::complex(cos(arg), -sin(arg)); const int npw = wfc_basis->npwk[ik]; - std::complex *sk = new std::complex[npw]; + std::complex* sk = new std::complex[npw]; const int nx = wfc_basis->nx, ny = wfc_basis->ny, nz = wfc_basis->nz; #ifdef _OPENMP #pragma omp parallel for @@ -25,15 +24,15 @@ std::complex* Structure_Factor::get_sk(const int ik, const int ixy = wfc_basis->is2fftixy[is]; int ix = ixy / wfc_basis->fftny; int iy = ixy % wfc_basis->fftny; - if (ix >= int(nx / 2) + 1) + if (ix >= int(nx / 2) + 1) { ix -= nx; } - if (iy >= int(ny / 2) + 1) + if (iy >= int(ny / 2) + 1) { iy -= ny; } - if (iz >= int(nz / 2) + 1) + if (iz >= int(nz / 2) + 1) { iz -= nz; } @@ -48,10 +47,7 @@ std::complex* Structure_Factor::get_sk(const int ik, } template -void Structure_Factor::get_sk(Device* ctx, - const int ik, - const ModulePW::PW_Basis_K* wfc_basis, - std::complex* sk) const +void Structure_Factor::get_sk(Device* ctx, const int ik, const ModulePW::PW_Basis_K* wfc_basis, std::complex* sk) const { ModuleBase::timer::start("Structure_Factor", "get_sk"); @@ -66,12 +62,14 @@ void Structure_Factor::get_sk(Device* ctx, using delmem_var_op = base_device::memory::delete_memory_op; using syncmem_var_op = base_device::memory::synchronize_memory_op; - int iat = 0, _npw = wfc_basis->npwk[ik], eigts1_nc = this->eigts1.nc, eigts2_nc = this->eigts2.nc, - eigts3_nc = this->eigts3.nc; + int iat = 0, _npw = wfc_basis->npwk[ik], eigts1_nc = this->eigts1.nc, eigts2_nc = this->eigts2.nc, eigts3_nc = this->eigts3.nc; int *igl2isz = nullptr, *is2fftixy = nullptr, *atom_na = nullptr, *h_atom_na = new int[ucell->ntype]; - FPTYPE *atom_tau = nullptr, *h_atom_tau = new FPTYPE[ucell->nat * 3], *kvec = wfc_basis->get_kvec_c_data(); - std::complex *eigts1 = this->get_eigts1_data(), *eigts2 = this->get_eigts2_data(), - *eigts3 = this->get_eigts3_data(); + const std::int64_t tau_elements = static_cast(ucell->nat) * 3; + FPTYPE* atom_tau = nullptr; + FPTYPE* h_atom_tau = new FPTYPE[tau_elements]; + FPTYPE* kvec = wfc_basis->get_kvec_c_data(); + std::complex*eigts1 = this->get_eigts1_data(), *eigts2 = this->get_eigts2_data(), + *eigts3 = this->get_eigts3_data(); for (int it = 0; it < ucell->ntype; it++) { h_atom_na[it] = ucell->atoms[it].na; @@ -83,18 +81,20 @@ void Structure_Factor::get_sk(Device* ctx, { int it = ucell->iat2it[iat]; int ia = ucell->iat2ia[iat]; - auto *tau = reinterpret_cast(ucell->atoms[it].tau.data()); - h_atom_tau[iat * 3 + 0] = static_cast(tau[ia * 3 + 0]); - h_atom_tau[iat * 3 + 1] = static_cast(tau[ia * 3 + 1]); - h_atom_tau[iat * 3 + 2] = static_cast(tau[ia * 3 + 2]); + auto* tau = reinterpret_cast(ucell->atoms[it].tau.data()); + const std::int64_t target_offset = static_cast(iat) * 3; + const std::int64_t source_offset = static_cast(ia) * 3; + h_atom_tau[target_offset + 0] = static_cast(tau[source_offset + 0]); + h_atom_tau[target_offset + 1] = static_cast(tau[source_offset + 1]); + h_atom_tau[target_offset + 2] = static_cast(tau[source_offset + 2]); } if (device == base_device::GpuDevice) { resmem_int_op()(atom_na, ucell->ntype); syncmem_int_op()(atom_na, h_atom_na, ucell->ntype); - resmem_var_op()(atom_tau, ucell->nat * 3); - syncmem_var_op()(atom_tau, h_atom_tau, ucell->nat * 3); + resmem_var_op()(atom_tau, tau_elements); + syncmem_var_op()(atom_tau, h_atom_tau, tau_elements); igl2isz = wfc_basis->d_igl2isz_k; is2fftixy = wfc_basis->d_is2fftixy; @@ -149,7 +149,7 @@ std::complex* Structure_Factor::get_skq(int ik, ModuleBase::Vector3 q) const // pengfei 2016-11-23 { const int npw = wfc_basis->npwk[ik]; - std::complex *skq = new std::complex[npw]; + std::complex* skq = new std::complex[npw]; for (int ig = 0; ig < npw; ig++) { diff --git a/source/source_pw/module_pwdft/td_history_pw.cpp b/source/source_pw/module_pwdft/td_history_pw.cpp new file mode 100644 index 00000000000..7e981682939 --- /dev/null +++ b/source/source_pw/module_pwdft/td_history_pw.cpp @@ -0,0 +1,117 @@ +#include "source_pw/module_pwdft/td_history_pw.h" + +#include "source_base/kernels/math_kernel_op.h" +#include "source_base/module_device/memory_op.h" +#include "source_base/timer.h" +#include "source_base/tool_quit.h" + +#include + +namespace pw +{ +namespace +{ + +template +void prepare_pair(const ModuleBase::matrix& potential, ct::Tensor* previous, ct::Tensor* midpoint) +{ + using CtDevice = typename ct::PsiToContainer::type; + const ct::DeviceType device = ct::DeviceTypeToEnum::value; + const int64_t size = std::max(1, static_cast(potential.nr) * potential.nc); + if (previous->NumElements() < size || previous->data_type() != ct::DataTypeToEnum::value || previous->device_type() != device) + { + *previous = ct::Tensor(ct::DataTypeToEnum::value, device, {size}); + *midpoint = ct::Tensor(ct::DataTypeToEnum::value, device, {size}); + } +} + +template +const Real* average_potential(const ModuleBase::matrix& shape, const Real* current, const ct::Tensor& previous, ct::Tensor* midpoint) +{ + const int size = shape.nr * shape.nc; + if (size == 0) + { + return current; + } + Real* buffer = midpoint->template data(); + ModuleBase::vector_add_vector_op()(size, buffer, previous.template data(), Real(0.5), current, Real(0.5)); + return buffer; +} + +template +void save_potential(const ModuleBase::matrix& shape, const Real* current, ct::Tensor* previous) +{ + const int size = shape.nr * shape.nc; + if (size > 0) + { + base_device::memory::synchronize_memory_op()(previous->template data(), current, size); + } +} + +} // namespace + +template +void TDHistoryPW::prepare(const elecstate::Potential& potential, const bool needs_ked) +{ + ModuleBase::timer::start("TDHistoryPW", "prepare"); + prepare_pair(potential.get_veff_smooth(), &veff_prev_, &veff_mid_); + if (needs_ked) + { + prepare_pair(potential.get_vofk_smooth(), &vofk_prev_, &vofk_mid_); + } + ModuleBase::timer::end("TDHistoryPW", "prepare"); +} + +template +TDPotentialView::type> TDHistoryPW::prop_pot(elecstate::Potential& potential, + const bool needs_ked, + const bool corrector) +{ + ModuleBase::timer::start("TDHistoryPW", "prop_pot"); + TDPotentialView view{potential.template get_veff_smooth_data(), potential.template get_vofk_smooth_data()}; + if (corrector) + { + view.veff = average_potential(potential.get_veff_smooth(), view.veff, veff_prev_, &veff_mid_); + if (needs_ked) + { + view.vofk = average_potential(potential.get_vofk_smooth(), view.vofk, vofk_prev_, &vofk_mid_); + } + } + ModuleBase::timer::end("TDHistoryPW", "prop_pot"); + return view; +} + +template +void TDHistoryPW::save(const psi::Psi& current, elecstate::Potential& potential, const bool needs_ked) +{ + ModuleBase::timer::start("TDHistoryPW", "save"); + if (!previous_) + { + previous_.reset(new psi::Psi(current)); + } + *previous_ = current; + save_potential(potential.get_veff_smooth(), potential.template get_veff_smooth_data(), &veff_prev_); + if (needs_ked) + { + save_potential(potential.get_vofk_smooth(), potential.template get_vofk_smooth_data(), &vofk_prev_); + } + ModuleBase::timer::end("TDHistoryPW", "save"); +} + +template +const psi::Psi& TDHistoryPW::previous() const +{ + if (!previous_) + { + ModuleBase::WARNING_QUIT("TDHistoryPW", "No converged state has been saved for propagation."); + } + return *previous_; +} + +template class TDHistoryPW, base_device::DEVICE_CPU>; +template class TDHistoryPW, base_device::DEVICE_CPU>; +#if defined(__CUDA) || defined(__ROCM) +template class TDHistoryPW, base_device::DEVICE_GPU>; +template class TDHistoryPW, base_device::DEVICE_GPU>; +#endif +} // namespace pw diff --git a/source/source_pw/module_pwdft/td_history_pw.h b/source/source_pw/module_pwdft/td_history_pw.h new file mode 100644 index 00000000000..5ebfdd90550 --- /dev/null +++ b/source/source_pw/module_pwdft/td_history_pw.h @@ -0,0 +1,50 @@ +#ifndef TD_HISTORY_PW_H +#define TD_HISTORY_PW_H + +#include "source_base/module_container/ATen/core/tensor.h" +#include "source_estate/module_pot/potential_new.h" +#include "source_psi/psi.h" + +#include + +namespace pw +{ + +/** @brief Non-owning device pointers to the smooth local potentials. */ +template +struct TDPotentialView +{ + const Real* veff; + const Real* vofk; +}; + +/** @brief Converged time-step history and predictor-corrector potential storage. */ +template +class TDHistoryPW +{ + public: + /** @brief Prepare storage after the potential grids have been updated. */ + void prepare(const elecstate::Potential& potential, const bool needs_ked); + + /** @brief Use the current potential for the predictor, then average with the saved potential. */ + TDPotentialView::type> prop_pot(elecstate::Potential& potential, + const bool needs_ked, + const bool corrector); + + /** @brief Save the converged state; the caller owns the convergence decision. */ + void save(const psi::Psi& current, elecstate::Potential& potential, const bool needs_ked); + + /** @brief Access the fixed origin of every corrector solve in the next step. */ + const psi::Psi& previous() const; + + private: + using Real = typename GetTypeReal::type; + std::unique_ptr> previous_; + ct::Tensor veff_prev_; + ct::Tensor veff_mid_; + ct::Tensor vofk_prev_; + ct::Tensor vofk_mid_; +}; + +} // namespace pw +#endif diff --git a/source/source_pw/module_pwdft/td_pw.cpp b/source/source_pw/module_pwdft/td_pw.cpp new file mode 100644 index 00000000000..bff532f7600 --- /dev/null +++ b/source/source_pw/module_pwdft/td_pw.cpp @@ -0,0 +1,102 @@ +#include "source_pw/module_pwdft/td_pw.h" + +#include "source_base/parallel_reduce.h" +#include "source_base/timer.h" +#include "source_base/tool_quit.h" +#include "source_cell/cell_tools.h" + +#include + +namespace pw +{ + +void check_td_input(const Input_para& input, const UnitCell& cell, const bool needs_ked, const bool cal_exx) +{ + if (input.init_vecpot_file) + { + ModuleBase::WARNING_QUIT("ESolver_KS_PW_TDDFT", "PW RT-TDDFT does not support init_vecpot_file=true; set init_vecpot_file=false."); + } + if (input.out_current != 0 && input.out_current != 1) + { + ModuleBase::WARNING_QUIT("ESolver_KS_PW_TDDFT", "PW RT-TDDFT requires out_current=0 or 1."); + } + if (input.scf_nmax < 2) + { + ModuleBase::WARNING_QUIT("ESolver_KS_PW_TDDFT", + "PW RT-TDDFT requires scf_nmax >= 2: one predictor and at least one corrector iteration."); + } + if (unitcell::if_atoms_can_move(cell.atoms, cell.ntype)) + { + ModuleBase::WARNING_QUIT("ESolver_KS_PW_TDDFT", + "PW RT-TDDFT requires fixed ions; set every atom's movement flags to 0 0 0 in STRU."); + } + if (input.calculation == "cell-relax" || (input.calculation == "md" && (input.mdp.md_type == "npt" || input.mdp.md_type == "msst"))) + { + ModuleBase::WARNING_QUIT("ESolver_KS_PW_TDDFT", "PW RT-TDDFT requires a fixed cell; cell-relax, NPT and MSST are not supported."); + } + if (input.socket_driver) + { + ModuleBase::WARNING_QUIT("ESolver_KS_PW_TDDFT", + "PW RT-TDDFT requires fixed ions and a fixed cell; socket_driver is not supported."); + } + if (input.dft_plus_u != 0) + { + ModuleBase::WARNING_QUIT("ESolver_KS_PW_TDDFT", "PW RT-TDDFT does not support DFT+U; set dft_plus_u=0."); + } + if (input.sc_mag_switch) + { + ModuleBase::WARNING_QUIT("ESolver_KS_PW_TDDFT", "PW RT-TDDFT does not support spin constraints; set sc_mag_switch=false."); + } + if (cal_exx) + { + ModuleBase::WARNING_QUIT("ESolver_KS_PW_TDDFT", "PW RT-TDDFT does not support exact exchange (EXX); use a non-hybrid functional."); + } + if ((input.nspin != 1 && input.nspin != 2) || input.bndpar != 1 || (input.td_stype != 0 && input.td_stype != 1)) + { + ModuleBase::WARNING_QUIT("ESolver_KS_PW_TDDFT", "PW RT-TDDFT requires nspin=1 or 2, bndpar=1 and td_stype=0 or 1."); + } + if (input.estep_per_md != 1) + { + ModuleBase::WARNING_QUIT("ESolver_KS_PW_TDDFT", "PW RT-TDDFT currently requires estep_per_md to be 1."); + } + if (input.mdp.md_restart || input.restart_load) + { + ModuleBase::WARNING_QUIT("ESolver_KS_PW_TDDFT", "PW RT-TDDFT restart is not supported yet."); + } + if (input.td_stype == 1 && needs_ked) + { + ModuleBase::WARNING_QUIT("ESolver_KS_PW_TDDFT", + "PW RT-TDDFT with a kinetic-energy-density functional requires td_stype=0; " + "the velocity-gauge coupling of the functional is not implemented."); + } +} + +double td_momentum_bound(const ModulePW::PW_Basis_K& basis, const double tpiba) +{ + ModuleBase::timer::start("pw", "td_momentum_bound"); + double bound = 0.0; + for (int ik = 0; ik < basis.nks; ++ik) + { + for (int ig = 0; ig < basis.npwk[ik]; ++ig) + { + bound = std::max(bound, basis.getgpluskcar(ik, ig).norm() * tpiba); + } + } + Parallel_Reduce::reduce_max(bound); + ModuleBase::timer::end("pw", "td_momentum_bound"); + return bound; +} + +void ensure_td_vnl(const UnitCell& cell, + const double unshifted_bound, + const ModuleBase::Vector3& A_right_ha, + const ModuleBase::Vector3& A_prop_ha, + pseudopot_cell_vnl* projectors) +{ + ModuleBase::timer::start("pw", "ensure_td_vnl"); + const double field_bound = std::max(A_right_ha.norm(), A_prop_ha.norm()); + projectors->ensure_vnl_range(cell, unshifted_bound + field_bound); + ModuleBase::timer::end("pw", "ensure_td_vnl"); +} + +} // namespace pw diff --git a/source/source_pw/module_pwdft/td_pw.h b/source/source_pw/module_pwdft/td_pw.h new file mode 100644 index 00000000000..9bc6908b896 --- /dev/null +++ b/source/source_pw/module_pwdft/td_pw.h @@ -0,0 +1,25 @@ +#ifndef TD_PW_H +#define TD_PW_H + +#include "source_basis/module_pw/pw_basis_k.h" +#include "source_cell/unitcell.h" +#include "source_io/module_parameter/input_parameter.h" +#include "source_pw/module_pwdft/vnl_pw.h" + +namespace pw +{ + +/** @brief Check PW propagation restrictions after resolving the XC functional and EXX configuration. */ +void check_td_input(const Input_para& input, const UnitCell& cell, const bool needs_ked, const bool cal_exx); + +/** @brief Return the global unshifted momentum bound after updating the distributed basis. */ +double td_momentum_bound(const ModulePW::PW_Basis_K& basis, const double tpiba); + +/** @brief Extend the projector table to cover both endpoint and propagation fields in Hartree atomic units. */ +void ensure_td_vnl(const UnitCell& cell, const double unshifted_bound, + const ModuleBase::Vector3& A_right_ha, + const ModuleBase::Vector3& A_prop_ha, + pseudopot_cell_vnl* projectors); + +} // namespace pw +#endif diff --git a/source/source_pw/module_pwdft/test/CMakeLists.txt b/source/source_pw/module_pwdft/test/CMakeLists.txt index 5482637ed79..326f27a64e6 100644 --- a/source/source_pw/module_pwdft/test/CMakeLists.txt +++ b/source/source_pw/module_pwdft/test/CMakeLists.txt @@ -22,12 +22,6 @@ AddTest( ../../../source_base/parallel_reduce.cpp ) -AddTest( - TARGET MODULE_PW_radial_proj_test - LIBS parameter base device - SOURCES radial_proj_test.cpp ../radial_proj.cpp -) - AddTest( TARGET MODULE_PW_dftu_base_test LIBS parameter base device diff --git a/source/source_pw/module_pwdft/test/radial_proj_test.cpp b/source/source_pw/module_pwdft/test/radial_proj_test.cpp deleted file mode 100644 index a5b9b4b2dc9..00000000000 --- a/source/source_pw/module_pwdft/test/radial_proj_test.cpp +++ /dev/null @@ -1,295 +0,0 @@ -#include "source_pw/module_pwdft/radial_proj.h" -#include -#include -#include -#include -#include - -#define DOUBLETHRESHOLD 1e-15 - -TEST(RadialProjectionTest, BuildBackwardMapTest) -{ - const std::vector> it2iproj = { - {0, 1}, - {2, 3, 4, 5}, - {6, 7, 8} - }; - const std::vector iproj2l = {0, 1, 0, 1, 2, 3, 0, 1, 2}; - /** - * a backward map will index irow to (it, iproj, m). For the above example, - * it is actually the first atom type has two projectors, the first projector - * has l = 0, the second has l = 1. The former will have only one channel, - * the second will have 3. The second atom type has 4 projectors... - * - * Therefore, it is expected there are 1+3+1+3+5+7+1+3+5 = 29 rows in total. - */ - std::vector itref(4, 0); // four 0, sixteen 1, nine 2 - itref.insert(itref.end(), 16, 1); - itref.insert(itref.end(), 9, 2); - std::vector iprojref(1, 0); - iprojref.insert(iprojref.end(), 3, 1); - iprojref.insert(iprojref.end(), 1, 0); - iprojref.insert(iprojref.end(), 3, 1); - iprojref.insert(iprojref.end(), 5, 2); - iprojref.insert(iprojref.end(), 7, 3); - iprojref.insert(iprojref.end(), 1, 0); - iprojref.insert(iprojref.end(), 3, 1); - iprojref.insert(iprojref.end(), 5, 2); - std::vector mref(1, 0); - mref.insert(mref.end(), {-1, 0, 1}); - mref.insert(mref.end(), {0}); - mref.insert(mref.end(), {-1, 0, 1}); - mref.insert(mref.end(), {-2, -1, 0, 1, 2}); - mref.insert(mref.end(), {-3, -2, -1, 0, 1, 2, 3}); - mref.insert(mref.end(), {0}); - mref.insert(mref.end(), {-1, 0, 1}); - mref.insert(mref.end(), {-2, -1, 0, 1, 2}); - - std::vector irow2it, irow2iproj, irow2m; - RadialProjection::RadialProjector::_build_backward_map(it2iproj, iproj2l, irow2it, irow2iproj, irow2m); - for(int i = 0; i < 29; ++i) - { - EXPECT_EQ(irow2it[i], itref[i]); - EXPECT_EQ(irow2iproj[i], iprojref[i]); - EXPECT_EQ(irow2m[i], mref[i]); - } -} - -TEST(RadialProjectionTest, BuildForwardMapTest) -{ - // forward means from (it, ia, iproj, m) to irow - const std::vector> it2ia = { - {0, 1}, - {0, 1, 3}, - {12} - }; - const std::vector> it2iproj = { - {0, 1}, - {2, 3, 4, 5}, - {6, 7, 8} - }; - const std::vector iproj2l = {0, 1, 0, 1, 2, 3, 0, 1, 2}; - std::map, int> forward_map; - RadialProjection::RadialProjector::_build_forward_map(it2ia, it2iproj, iproj2l, forward_map); - // this is a terribly large map, for type 0, there are 2 atoms, each atom has two projectors - // with angular momentum as 0 and 1, respectively. For type 1, there are 3 atoms, each has - // 4 projectors with angular momentum from 0 to 3... - const int nrow = 2*4 + 3*16 + 1*9; - EXPECT_EQ(forward_map.size(), nrow); - // it ia iproj m - // it = 0 - // ia = 0 - // iproj = 0: l = 0 - EXPECT_EQ(forward_map[std::make_tuple(0, 0, 0, 0)], 0); - // iproj = 1: l = 1: m = -1, 0, 1 - EXPECT_EQ(forward_map[std::make_tuple(0, 0, 1, -1)], 1); - EXPECT_EQ(forward_map[std::make_tuple(0, 0, 1, 0)], 2); - EXPECT_EQ(forward_map[std::make_tuple(0, 0, 1, 1)], 3); - // ia = 1 - // iproj = 0: l = 0 - EXPECT_EQ(forward_map[std::make_tuple(0, 1, 0, 0)], 4); - // iproj = 1: l = 1: m = -1, 0, 1 - EXPECT_EQ(forward_map[std::make_tuple(0, 1, 1, -1)], 5); - EXPECT_EQ(forward_map[std::make_tuple(0, 1, 1, 0)], 6); - EXPECT_EQ(forward_map[std::make_tuple(0, 1, 1, 1)], 7); - // it = 1 - // ia = 0 - // iproj = 0: l = 0 - EXPECT_EQ(forward_map[std::make_tuple(1, 0, 0, 0)], 8); - // iproj = 1: l = 1: m = -1, 0, 1 - EXPECT_EQ(forward_map[std::make_tuple(1, 0, 1, -1)], 9); - EXPECT_EQ(forward_map[std::make_tuple(1, 0, 1, 0)], 10); - EXPECT_EQ(forward_map[std::make_tuple(1, 0, 1, 1)], 11); - // iproj = 2: l = 2: m = -2, -1, 0, 1, 2 - EXPECT_EQ(forward_map[std::make_tuple(1, 0, 2, -2)], 12); - EXPECT_EQ(forward_map[std::make_tuple(1, 0, 2, -1)], 13); - EXPECT_EQ(forward_map[std::make_tuple(1, 0, 2, 0)], 14); - EXPECT_EQ(forward_map[std::make_tuple(1, 0, 2, 1)], 15); - EXPECT_EQ(forward_map[std::make_tuple(1, 0, 2, 2)], 16); - // iproj = 3: l = 3: m = -3, -2, -1, 0, 1, 2, 3 - EXPECT_EQ(forward_map[std::make_tuple(1, 0, 3, -3)], 17); - EXPECT_EQ(forward_map[std::make_tuple(1, 0, 3, -2)], 18); - EXPECT_EQ(forward_map[std::make_tuple(1, 0, 3, -1)], 19); - EXPECT_EQ(forward_map[std::make_tuple(1, 0, 3, 0)], 20); - EXPECT_EQ(forward_map[std::make_tuple(1, 0, 3, 1)], 21); - EXPECT_EQ(forward_map[std::make_tuple(1, 0, 3, 2)], 22); - EXPECT_EQ(forward_map[std::make_tuple(1, 0, 3, 3)], 23); - // ia = 1... -} - -TEST(RadialProjectionTest, MaskfunctionGenerationTest) -{ - std::vector mask; - RadialProjection::_mask_func(mask); - EXPECT_EQ(mask.size(), 201); - EXPECT_EQ(mask[0], 1.0); // the rescaled value of the mask function, at 0, is 1 - EXPECT_NEAR(mask[200], 0.98138215E-05, 1e-10); // real space cut, at rc, is 0 -} - -TEST(RadialProjectionTest, BuildSbtTabCorrectnessTest) -{ - RadialProjection::RadialProjector rp; - - // use mask function as the example - std::vector mask; - RadialProjection::_mask_func(mask); - // suppose the r from 0 to 2.0 (inclusive) with 0.01 step - std::vector r(mask.size()); - std::iota(r.begin(), r.end(), 0); - std::for_each(r.begin(), r.end(), [](double& x){x = x * 0.01;}); - - // fill into the container - std::vector> radials(1, mask); - - // suppose the angular momentum of function to transform is 0 - // but this is true because the mask function is really spherically - // symmetric. - std::vector l(1, 0); - - // build the interpolation table - rp._build_sbt_tab(r, radials, l, 201, 0.01); // build an interpolation table - - // only one q point: (0, 0, 0), is Gamma - std::vector> q(1); - q[0] = ModuleBase::Vector3(0.0, 0.0, 0.0); - - // so the output should have the same size as q - std::vector> out(q.size()); - - // transform, the last argument is 'r' for . - // for q = 0, the integration is actually performed on the function - // itself, so it is actually the integration of the function, on the - // grid from 0 to rc (2.0) with stepsize 0.01. - rp.sbtft(q, out, 'r', 1.0, 1.0); - // print, each 5 numbers in a row - - /** - * The following Python code is used to generate the reference data - * from scipy.integrate import simps - * import numpy as np - * - * r = np.arange(0, 2.01, 0.01) - * mask = np.array(mask) - * omega = 1 - * # Y00 is 1/sqrt(4*pi) - * val = simps(mask*r**2, x=r) * np.sqrt(4*np.pi) - * print(val/np.sqrt(omega)) - */ - std::complex ref(0.6143605159327622, 0.0); - EXPECT_NEAR(out[0].real(), ref.real(), DOUBLETHRESHOLD); -} - -TEST(RadialProjectionTest, BuildSbtTabStabilityTest) -{ - // still use mask function but scale with different Gaussian function - std::vector mask; - RadialProjection::_mask_func(mask); - // suppose the r from 0 to 2.0 (inclusive) with 0.01 step - std::vector r(mask.size()); - std::iota(r.begin(), r.end(), 0); - std::for_each(r.begin(), r.end(), [](double& x){x = x * 0.01;}); - - std::vector aug1(mask.size()); - std::vector aug2(mask.size()); - std::vector aug3(mask.size()); - const double sigma1 = 0.1; - const double sigma2 = 1.0; - const double sigma3 = 10.0; - std::transform(r.begin(), r.end(), aug1.begin(), [sigma1](double x){return std::exp(-x*x/(2*sigma1*sigma1));}); - std::transform(r.begin(), r.end(), aug2.begin(), [sigma2](double x){return std::exp(-x*x/(2*sigma2*sigma2));}); - std::transform(r.begin(), r.end(), aug3.begin(), [sigma3](double x){return std::exp(-x*x/(2*sigma3*sigma3));}); - - std::vector mask1(mask.size()); - std::vector mask2(mask.size()); - std::vector mask3(mask.size()); - std::transform(mask.begin(), mask.end(), aug1.begin(), mask1.begin(), std::multiplies()); - std::transform(mask.begin(), mask.end(), aug2.begin(), mask2.begin(), std::multiplies()); - std::transform(mask.begin(), mask.end(), aug3.begin(), mask3.begin(), std::multiplies()); - - // then prepare the q vectors, in total 50, each with 3 components among which each is random - // number ranging from 0 to 2, use the random seed as 1 - const int npw = 50; - std::vector> q(npw); - std::mt19937 gen(1); - std::uniform_real_distribution dis(0.0, 2.0); - std::for_each(q.begin(), q.end(), [&gen, &dis](ModuleBase::Vector3& x){ - x[0] = dis(gen); - x[1] = dis(gen); - x[2] = dis(gen); - }); - - // fill into the container - std::vector> radials(4); // do not remember mask0 = mask - radials[0] = mask; - radials[1] = mask1; - radials[2] = mask2; - radials[3] = mask3; - - // angular momentum set as 0, 1, 2, 3 - std::vector l(4); - std::iota(l.begin(), l.end(), 0); - - RadialProjection::RadialProjector rp; - rp._build_sbt_tab(r, radials, l, 401, 0.01); // build an interpolation table - - // then perform the transform - std::vector> out; - rp.sbtft(q, out, 'r', 1.0, 1.0); - - // there are 50 q points, and radials have l as 0, 1, 2, 3, which means the - // number of lm components in total is 1 + 3 + 5 + 7 = 16 - // therefore there should be 50 * 16 = 800 complex numbers in the output - EXPECT_EQ(out.size(), npw*16); - // check if they are listed in expected sequence, say the first should be - // Fourier transform of the first function, mask, l = 0, m = 0 - RadialProjection::RadialProjector rp1; - rp1._build_sbt_tab(r, - std::vector>(1, mask), - std::vector(1, 0), 401, 0.01); - std::vector> out1; - rp1.sbtft(q, out1, 'r', 1.0, 1.0); - for(int iq = 0; iq < 50; ++iq) // 50 q-points - { - EXPECT_NEAR(out[iq].real(), out1[iq].real(), DOUBLETHRESHOLD); - EXPECT_NEAR(out[iq].imag(), out1[iq].imag(), DOUBLETHRESHOLD); - } - RadialProjection::RadialProjector rp2; - rp2._build_sbt_tab(r, - std::vector>(1, mask1), - std::vector(1, 1), 401, 0.01); - std::vector> out2; - rp2.sbtft(q, out2, 'r', 1.0, 1.0); - for(int iq = 0; iq < 50*(2*1+1); ++iq) // 50 q-points - { - EXPECT_NEAR(out[iq+50].real(), out2[iq].real(), DOUBLETHRESHOLD); - EXPECT_NEAR(out[iq+50].imag(), out2[iq].imag(), DOUBLETHRESHOLD); - } - RadialProjection::RadialProjector rp3; - rp3._build_sbt_tab(r, - std::vector>(1, mask2), - std::vector(1, 2), 401, 0.01); - std::vector> out3; - rp3.sbtft(q, out3, 'r', 1.0, 1.0); - for(int iq = 0; iq < 50*(2*2+1); ++iq) // 50 q-points - { - EXPECT_NEAR(out[iq+50+50*(2*1+1)].real(), out3[iq].real(), DOUBLETHRESHOLD); - EXPECT_NEAR(out[iq+50+50*(2*1+1)].imag(), out3[iq].imag(), DOUBLETHRESHOLD); - } - RadialProjection::RadialProjector rp4; - rp4._build_sbt_tab(r, - std::vector>(1, mask3), - std::vector(1, 3), 401, 0.01); - std::vector> out4; - rp4.sbtft(q, out4, 'r', 1.0, 1.0); - for(int iq = 0; iq < 50*(2*3+1); ++iq) // 50 q-points - { - EXPECT_NEAR(out[iq+50+50*(2*1+1)+50*(2*2+1)].real(), out4[iq].real(), DOUBLETHRESHOLD); - EXPECT_NEAR(out[iq+50+50*(2*1+1)+50*(2*2+1)].imag(), out4[iq].imag(), DOUBLETHRESHOLD); - } -} - -int main() -{ - testing::InitGoogleTest(); - return RUN_ALL_TESTS(); -} \ No newline at end of file diff --git a/source/source_pw/module_pwdft/test/structure_factor_test.cpp b/source/source_pw/module_pwdft/test/structure_factor_test.cpp index d1b2fe8c7ce..129e398a77f 100644 --- a/source/source_pw/module_pwdft/test/structure_factor_test.cpp +++ b/source/source_pw/module_pwdft/test/structure_factor_test.cpp @@ -4,11 +4,9 @@ #include #include #include "source_cell/unitcell.h" -#include "source_estate/module_dm/test/prepare_unitcell.h" -#define private public +#include "source_estate/module_dm/unittests/prepare_unitcell.h" #include "source_io/module_parameter/parameter.h" #include "source_pw/module_pwdft/stru_fac.h" -#undef private /************************************************ * unit test of class Structure_factor and ***********************************************/ @@ -70,49 +68,49 @@ TEST_F(StructureFactorTest, set) TEST_F(StructureFactorTest, setup_structure_factor_double) { rho_basis->npw = 10; - SF.setup(ucell,*pgrid,rho_basis); + SF.setup(ucell,*pgrid,rho_basis,false); for (int i=0;i< ucell->nat * (2 * rho_basis->nx + 1);i++) { - EXPECT_EQ(SF.z_eigts1[i].real(),1); - EXPECT_EQ(SF.z_eigts1[i].imag(),0); + EXPECT_EQ(SF.get_eigts1_data()[i].real(),1); + EXPECT_EQ(SF.get_eigts1_data()[i].imag(),0); } for (int i=0;i< ucell->nat * (2 * rho_basis->ny + 1);i++) { - EXPECT_EQ(SF.z_eigts2[i].real(),1); - EXPECT_EQ(SF.z_eigts2[i].imag(),0); + EXPECT_EQ(SF.get_eigts2_data()[i].real(),1); + EXPECT_EQ(SF.get_eigts2_data()[i].imag(),0); } for (int i=0;i< ucell->nat * (2 * rho_basis->nz + 1);i++) { - EXPECT_EQ(SF.z_eigts3[i].real(),1); - EXPECT_EQ(SF.z_eigts3[i].imag(),0); + EXPECT_EQ(SF.get_eigts3_data()[i].real(),1); + EXPECT_EQ(SF.get_eigts3_data()[i].imag(),0); } } TEST_F(StructureFactorTest, setup_structure_factor_float) { - PARAM.sys.has_float_data = true; + // the float eigts copies are what this case checks, so ask setup() for them rho_basis->npw = 10; - SF.setup(ucell,*pgrid,rho_basis); + SF.setup(ucell,*pgrid,rho_basis,true); for (int i=0;i< ucell->nat * (2 * rho_basis->nx + 1);i++) { - EXPECT_EQ(SF.c_eigts1[i].real(),1); - EXPECT_EQ(SF.c_eigts1[i].imag(),0); + EXPECT_EQ(SF.get_eigts1_data()[i].real(),1); + EXPECT_EQ(SF.get_eigts1_data()[i].imag(),0); } for (int i=0;i< ucell->nat * (2 * rho_basis->ny + 1);i++) { - EXPECT_EQ(SF.c_eigts2[i].real(),1); - EXPECT_EQ(SF.c_eigts2[i].imag(),0); + EXPECT_EQ(SF.get_eigts2_data()[i].real(),1); + EXPECT_EQ(SF.get_eigts2_data()[i].imag(),0); } for (int i=0;i< ucell->nat * (2 * rho_basis->nz + 1);i++) { - EXPECT_EQ(SF.c_eigts3[i].real(),1); - EXPECT_EQ(SF.c_eigts3[i].imag(),0); + EXPECT_EQ(SF.get_eigts3_data()[i].real(),1); + EXPECT_EQ(SF.get_eigts3_data()[i].imag(),0); } } diff --git a/source/source_pw/module_pwdft/uspp_support.cpp b/source/source_pw/module_pwdft/uspp_support.cpp index 5e5fd9075a6..1046de82a50 100644 --- a/source/source_pw/module_pwdft/uspp_support.cpp +++ b/source/source_pw/module_pwdft/uspp_support.cpp @@ -23,6 +23,13 @@ void validate_uspp_support(const bool use_uspp, } std::vector violations; + // TODO: allowing LCAO (basis_type=lcao, lcao_in_pw) with USPP would enable + // double_grid on LCAO paths. Before relaxing this guard, verify (1) which + // grid module_charge::symmetrize_rho must use, since LCAO callers pass + // smooth pw_rho while LIP/pchg callers pass dense pw_rhod (see the TODO in + // chg_symm.cpp); (2) that charge init/extrapolation and the smooth/dense + // mixing paths are correct for LCAO; (3) the ndx/ndy/ndz input path in + // read_inp_sys.cpp lacks the LCAO guard that the ecutrho/ecutwfc path has. if (basis_type != "pw") { violations.push_back("basis_type=" + basis_type + " (only pw is supported)"); diff --git a/source/source_pw/module_pwdft/velocity_workspace.cpp b/source/source_pw/module_pwdft/velocity_workspace.cpp new file mode 100644 index 00000000000..c7147c5594e --- /dev/null +++ b/source/source_pw/module_pwdft/velocity_workspace.cpp @@ -0,0 +1,113 @@ +#include "source_pw/module_pwdft/velocity_workspace.h" + +#include "source_base/kernels/math_kernel_op.h" +#include "source_base/module_device/memory_op.h" +#include "source_base/parallel_device.h" +#include "source_base/timer.h" +#include "source_pw/module_pwdft/kernels/nonlocal_op.h" + +#include +#include + +namespace hamilt +{ +template +std::complex* VelocityWorkspace::prepare(const std::int64_t count) +{ + using CtDevice = typename ct::PsiToContainer::type; + if (count > capacity_ || coefficients_.data_type() != ct::DataTypeToEnum::value + || coefficients_.device_type() != ct::DeviceTypeToEnum::value) + { + const std::int64_t workspace_elements = 8 * count; + const std::int64_t capacity = std::max(1, workspace_elements); + const std::int64_t projection_elements = 4 * count; + coefficients_ = ct::Tensor(ct::DataTypeToEnum::value, ct::DeviceTypeToEnum::value, {capacity}); + host_.resize(projection_elements); + capacity_ = count; + } + return coefficients_.template data(); +} + +template +void VelocityWorkspace::contract(const UnitCell& cell, + const pseudopot_cell_vnl& pp, + const int spin, + const int bands, + const Real scale, + const ModulePW::PW_Basis_K& basis, + Complex* buffer) +{ + ModuleBase::timer::start("VelocityWorkspace", "contract"); + const std::int64_t count = static_cast(bands) * pp.nkb; + if (count == 0) + { + ModuleBase::timer::end("VelocityWorkspace", "contract"); + return; + } + const std::int64_t projection_elements = 4 * count; +#ifdef __MPI + if (basis.poolnproc > 1) + { + // The complex reduction wrapper doubles the MPI element count. + const std::int64_t max_chunk = std::numeric_limits::max() / 2; + for (std::int64_t offset = 0; offset < projection_elements;) + { + const std::int64_t remaining = projection_elements - offset; + const int chunk_size = static_cast(std::min(max_chunk, remaining)); + Complex* chunk = buffer + offset; + Complex* host_chunk = host_.data() + offset; + Parallel_Common::reduce_dev(chunk, chunk_size, basis.pool_world, host_chunk); + offset += chunk_size; + } + } +#endif + Complex* output = buffer + projection_elements; + base_device::memory::set_memory_op()(output, 0, projection_elements); + for (int component = 0; component < 4; ++component) + { + int sum = 0; + int atom = 0; + const int stride = component == 0 ? pp.nkb : 3 * pp.nkb; + const std::int64_t gradient_offset = static_cast(component - 1) * pp.nkb; + const Complex* input = component == 0 ? buffer : buffer + count + gradient_offset; + Complex* component_output = output + component * count; + for (int type = 0; type < cell.ntype; ++type) + { + nonlocal_pw_op()(nullptr, + cell.atoms[type].na, + bands, + cell.atoms[type].ncpp.nh, + sum, + atom, + spin, + stride, + pp.deeq.getBound2(), + pp.deeq.getBound3(), + pp.deeq.getBound4(), + pp.template get_deeq_data(), + component_output, + input); + } + } + if (scale != Real(1)) + { + const Complex factor(scale, 0); + const std::int64_t max_chunk = std::numeric_limits::max(); + for (std::int64_t offset = 0; offset < projection_elements;) + { + const std::int64_t remaining = projection_elements - offset; + const int chunk_size = static_cast(std::min(max_chunk, remaining)); + Complex* chunk = output + offset; + ModuleBase::scal_op()(chunk_size, &factor, chunk, 1); + offset += chunk_size; + } + } + ModuleBase::timer::end("VelocityWorkspace", "contract"); +} +template class VelocityWorkspace; +template class VelocityWorkspace; +#if defined(__CUDA) || defined(__ROCM) +template class VelocityWorkspace; +template class VelocityWorkspace; +#endif +} // namespace hamilt diff --git a/source/source_pw/module_pwdft/velocity_workspace.h b/source/source_pw/module_pwdft/velocity_workspace.h new file mode 100644 index 00000000000..7dd2665f63b --- /dev/null +++ b/source/source_pw/module_pwdft/velocity_workspace.h @@ -0,0 +1,35 @@ +#ifndef HAMILT_VELOCITY_WORKSPACE_H +#define HAMILT_VELOCITY_WORKSPACE_H + +#include "source_base/module_container/ATen/core/tensor.h" +#include "source_pw/module_pwdft/vnl_pw.h" + +#include + +namespace hamilt +{ +/** @brief Reusable projection storage and device contraction for a velocity operator. */ +template +class VelocityWorkspace +{ + private: + using Complex = std::complex; + ct::Tensor coefficients_; + std::vector host_; + std::int64_t capacity_ = 0; + + public: + /** @brief Reserve four input and four output projector blocks. */ + Complex* prepare(const std::int64_t count); + + /** @brief Reduce projections and contract D on their native device. */ + void contract(const UnitCell& cell, + const pseudopot_cell_vnl& pp, + const int spin, + const int bands, + const Real scale, + const ModulePW::PW_Basis_K& basis, + Complex* buffer); +}; +} // namespace hamilt +#endif diff --git a/source/source_pw/module_pwdft/vnl_pw.cpp b/source/source_pw/module_pwdft/vnl_pw.cpp index 1c1d3ed35d7..233e0590bca 100644 --- a/source/source_pw/module_pwdft/vnl_pw.cpp +++ b/source/source_pw/module_pwdft/vnl_pw.cpp @@ -1,14 +1,15 @@ #include "vnl_pw.h" -#include "source_io/module_parameter/parameter.h" #include "source_base/global_function.h" #include "source_base/global_variable.h" -#include "source_base/output.h" #include "source_base/memory_recorder.h" #include "source_base/module_device/device.h" +#include "source_base/output.h" #include "source_base/timer.h" +#include "source_io/module_parameter/parameter.h" #include +#include /** * @file vnl_pw.cpp @@ -18,7 +19,7 @@ * - ctor / dtor / release_memory(): device buffer cleanup * - init(): allocate and wire vkb, tab, tab_at, deeq, qq_nt/qq_so, nhtol etc. * - print_vnl(): dump tab to stream - * - rescale_vnl(): rescale tab/tab_at/qrad when the cell volume changes + * - rescale_vnl(): rescale tab/tab_dq/tab_at/qrad when the cell volume changes * - get_*_data() template specializations for CPU/GPU typed pointer access * * Heavy logic (getvnl, init_vnl, qrad, deeq, alpha channel) lives in: @@ -26,7 +27,6 @@ * vnl_pw_deeq.cpp, vnl_pw_alpha.cpp, vnl_pw_grad.cpp */ - pseudopot_cell_vnl::pseudopot_cell_vnl() { this->use_gpu_ = (PARAM.inp.device == "gpu"); @@ -38,9 +38,10 @@ pseudopot_cell_vnl::~pseudopot_cell_vnl() void pseudopot_cell_vnl::release_memory() { - if (this->nhm <= 0 || memory_released) { + if (this->nhm <= 0 || memory_released) + { return; -} + } if (this->use_gpu_) { delmem_sd_op()(this->s_deeq); @@ -113,8 +114,7 @@ void pseudopot_cell_vnl::init(const UnitCell& ucell, GlobalV::ofs_running << " " << ucell.atoms[it].label << " non-local projectors:" << std::endl; for (int ibeta = 0; ibeta < ucell.atoms[it].ncpp.nbeta; ibeta++) { - GlobalV::ofs_running << " projector " << ibeta + 1 << " L=" << ucell.atoms[it].ncpp.lll[ibeta] - << std::endl; + GlobalV::ofs_running << " projector " << ibeta + 1 << " L=" << ucell.atoms[it].ncpp.lll[ibeta] << std::endl; this->lmaxkb = std::max(this->lmaxkb, ucell.atoms[it].ncpp.lll[ibeta]); } } @@ -184,16 +184,12 @@ void pseudopot_cell_vnl::init(const UnitCell& ucell, { if (PARAM.globalv.has_float_data) { - resmem_sh_op()(s_deeq, - PARAM.inp.nspin * ucell.nat * this->nhm * this->nhm, - "VNL::s_deeq"); + resmem_sh_op()(s_deeq, PARAM.inp.nspin * ucell.nat * this->nhm * this->nhm, "VNL::s_deeq"); resmem_sh_op()(s_nhtol, ntype * this->nhm, "VNL::s_nhtol"); resmem_sh_op()(s_nhtolm, ntype * this->nhm, "VNL::s_nhtolm"); resmem_sh_op()(s_indv, ntype * this->nhm, "VNL::s_indv"); resmem_sh_op()(s_qq_nt, ntype * this->nhm * this->nhm, "VNL::s_qq_nt"); - resmem_ch_op()(c_deeq_nc, - PARAM.inp.nspin * ucell.nat * this->nhm * this->nhm, - "VNL::c_deeq_nc"); + resmem_ch_op()(c_deeq_nc, PARAM.inp.nspin * ucell.nat * this->nhm * this->nhm, "VNL::c_deeq_nc"); resmem_ch_op()(c_qq_so, ntype * 4 * this->nhm * this->nhm, "VNL::c_qq_so"); } if (PARAM.globalv.has_double_data) @@ -224,12 +220,13 @@ void pseudopot_cell_vnl::init(const UnitCell& ucell, this->lmaxq = 2 * this->lmaxkb + 1; int npwx = this->wfcpw->npwk_max; this->vkbnc = npwx; + const std::int64_t projector_elements = static_cast(nkb) * npwx; if (nkb > 0 && allocate_vkb) { if (!this->use_gpu_) - { - vkb.create(nkb, npwx); - ModuleBase::Memory::record("VNL::vkb", nkb * npwx * sizeof(std::complex)); + { + vkb.create(nkb, npwx); + ModuleBase::Memory::record("VNL::vkb", projector_elements * sizeof(std::complex)); } // GPU path: vkb ComplexMatrix is not allocated. // Column dimension is stored in vkbnc for gemm/gemv leading dimension. @@ -284,9 +281,9 @@ void pseudopot_cell_vnl::init(const UnitCell& ucell, if (PARAM.globalv.has_float_data) { resmem_sd_op()(s_tab, this->tab.getSize()); - resmem_cd_op()(c_vkb, nkb * npwx); + resmem_cd_op()(c_vkb, projector_elements); } - resmem_zd_op()(z_vkb, nkb * npwx); + resmem_zd_op()(z_vkb, projector_elements); resmem_dd_op()(d_tab, this->tab.getSize()); } else @@ -294,11 +291,12 @@ void pseudopot_cell_vnl::init(const UnitCell& ucell, if (PARAM.globalv.has_float_data) { resmem_sh_op()(s_tab, this->tab.getSize()); - resmem_ch_op()(c_vkb, nkb * npwx); + resmem_ch_op()(c_vkb, projector_elements); } #ifdef __DSP - base_device::memory::resize_memory_op_mt, base_device::DEVICE_CPU>() - (this->z_vkb, this->vkb.size, "VNL::z_vkb"); + base_device::memory::resize_memory_op_mt, base_device::DEVICE_CPU>()(this->z_vkb, + this->vkb.size, + "VNL::z_vkb"); // memcpy(this->z_vkb,this->vkb.c,this->vkb.size*16); #else this->z_vkb = this->vkb.c; @@ -320,6 +318,12 @@ void pseudopot_cell_vnl::print_vnl(std::ofstream& ofs) // scale the non-local pseudopotential tables void pseudopot_cell_vnl::rescale_vnl(const double& omega_in) { + const bool gradient_current = this->gradient_version_ == this->table_version_; + ++this->table_version_; + if (gradient_current) + { + this->gradient_version_ = this->table_version_; + } const double ratio = this->omega_old / omega_in; const double sqrt_ratio = std::sqrt(ratio); this->omega_old = omega_in; @@ -328,6 +332,10 @@ void pseudopot_cell_vnl::rescale_vnl(const double& omega_in) { this->tab.ptr[i] *= sqrt_ratio; } + for (int i = 0; i < this->tab_dq.getSize(); ++i) + { + this->tab_dq.ptr[i] *= sqrt_ratio; + } for (int i = 0; i < this->tab_at.getSize(); i++) { this->tab_at.ptr[i] *= sqrt_ratio; diff --git a/source/source_pw/module_pwdft/vnl_pw.h b/source/source_pw/module_pwdft/vnl_pw.h index ca6a9f06a51..8761d7cc2fa 100644 --- a/source/source_pw/module_pwdft/vnl_pw.h +++ b/source/source_pw/module_pwdft/vnl_pw.h @@ -6,9 +6,9 @@ #include "source_base/intarray.h" #include "source_base/realarray.h" #include "source_cell/unitcell.h" +#include "source_psi/psi.h" #include "source_pw/module_pwdft/soc.h" #include "source_pw/module_pwdft/stru_fac.h" -#include "source_psi/psi.h" #include #ifdef __LCAO @@ -40,16 +40,46 @@ class pseudopot_cell_vnl void rescale_vnl(const double& omega_in); + /** @brief Ensure four-point interpolation covers qmax (inverse Bohr), without changing dq. */ + void ensure_vnl_range(const UnitCell& cell, const double& qmax); + + /** @brief Reject non-finite or out-of-range dimensionless interpolation indices before reading. */ + void check_vnl_index(const double& index, const bool& derivative) const; + + /** @brief Return the momentum spacing shared by the radial projector and derivative tables. */ + double table_dq() const + { + return table_dq_; + } + + /** + * @brief Build nonlocal projectors at G+k+A; pass zero A for ground-state calculations. + * + * @param ctx Device context. + * @param ucell Unit cell defining reciprocal-space units. + * @param ik K-point index. + * @param vector_potential Vector potential in Hartree atomic units. + * @param vkb_in Output projector buffer. + */ template - void getvnl(Device* ctx, const UnitCell& ucell, const int& ik, std::complex* vkb_in) const; + void getvnl(Device* ctx, + const UnitCell& ucell, + const int& ik, + const ModuleBase::Vector3& vector_potential, + std::complex* vkb_in) const; // void getvnl_alpha(const int &ik); void init_vnl_alpha(const UnitCell& cell); void initgradq_vnl(const UnitCell& cell); - - void getgradq_vnl(const UnitCell& ucell, const int ik); + /** @brief Generation of radial tables, including volume rescaling. */ + size_t table_version() const + { + return table_version_; + } + /** @brief Prepare the derivative table only when the radial table changed. */ + void ensure_grad_table(const UnitCell& cell); //=============================================================== // MEMBER VARIABLES : @@ -61,6 +91,12 @@ class pseudopot_cell_vnl //=============================================================== // private: + private: + double table_dq_ = 0.0; + size_t table_version_ = 1; + size_t gradient_version_ = 0; + + public: int nhm = 0; int nbetam = 0; // max number of beta functions @@ -69,7 +105,7 @@ class pseudopot_cell_vnl ModuleBase::matrix indv; // indes linking atomic beta's to beta's in the solid ModuleBase::matrix nhtol; // correspondence n <-> angular momentum l ModuleBase::matrix nhtolm; // correspondence n <-> combined lm index for (l,m) - ModuleBase::matrix nhtoj; // + ModuleBase::matrix nhtoj; // ModuleBase::realArray dvan; //(:,:,:), the D functions of the solid ModuleBase::ComplexArray dvan_so; //(:,:,:), spin-orbit case, added by zhengdy-soc @@ -104,8 +140,7 @@ class pseudopot_cell_vnl std::complex* c_qq_so = nullptr; // GPU array of qq_so std::complex* z_qq_so = nullptr; // GPU array of qq_so - mutable ModuleBase::ComplexMatrix vkb; // all beta functions in reciprocal space - mutable ModuleBase::ComplexArray gradvkb; // gradient of beta functions + mutable ModuleBase::ComplexMatrix vkb; // all beta functions in reciprocal space std::complex*** vkb1_alpha; std::complex*** vkb_alpha; Structure_Factor* psf = nullptr; @@ -215,6 +250,12 @@ class pseudopot_cell_vnl double omega_old = 0; bool use_gpu_ = false; + /** @brief Fill the allocated radial projector table using the original Bessel quadrature. */ + void fill_vnl_table(const UnitCell& cell); + + /** @brief Refresh aliases and allocated precision/device copies after radial-table changes. */ + void sync_vnl_table(); + /** * @brief Compute interpolation table qrad * diff --git a/source/source_pw/module_pwdft/vnl_pw_getvnl.cpp b/source/source_pw/module_pwdft/vnl_pw_getvnl.cpp index 034062b1a6c..1c6625e3bf6 100644 --- a/source/source_pw/module_pwdft/vnl_pw_getvnl.cpp +++ b/source/source_pw/module_pwdft/vnl_pw_getvnl.cpp @@ -1,48 +1,30 @@ -#include "vnl_pw.h" - -#include "source_io/module_parameter/parameter.h" #include "source_base/global_function.h" -#include "source_base/global_variable.h" #include "source_base/math_ylmreal.h" -#include "source_base/timer.h" #include "source_base/module_device/device.h" +#include "source_base/timer.h" #include "source_pw/module_pwdft/kernels/vnl_op.h" +#include "vnl_pw.h" +#include +#include +#include #include /** - * @file vnl_pw_getvnl.cpp - * @brief Compute the Kleinman-Bylander projector functions (beta functions) - * in reciprocal space for a given k-point. - * - * This file contains the getvnl() template implementation and its explicit - * instantiations for CPU/GPU and float/double precision. - */ - -/** - * @brief Calculate beta functions (Kleinman-Bylander projectors) with - * structure factor for all atoms in reciprocal space. - * - * Workflow: - * 1. Collect per-atom-type counts (nbeta/nh/na) into host vectors. - * 2. Build the |G+k| Cartesian vectors for the current k-point. - * 3. Move data to device when GPU is enabled; reuse host vectors otherwise. - * 4. Evaluate real spherical harmonics Ylm on all G+k vectors. - * 5. Build structure factors sk for all atoms. - * 6. Launch cal_vnl_op kernel to assemble vkb = beta * Ylm * sk * tab. + * @brief Calculate velocity-gauge nonlocal projectors at k+G+A(t). * - * @param ctx device context (CPU or GPU) - * @param ucell unit cell (read-only) - * @param ik index of the k-point - * @param vkb_in output projector array of shape (nkb, npw) + * The radial and angular projector factors use the kinetic momentum shifted + * by the Hartree-unit vector potential. The structure factor remains evaluated + * on the static reciprocal grid because its atom-dependent A phase cancels in + * each same-atom projector outer product. */ template void pseudopot_cell_vnl::getvnl(Device* ctx, const UnitCell& ucell, const int& ik, + const ModuleBase::Vector3& vector_potential, std::complex* vkb_in) const { - ModuleBase::TITLE("pseudopot_cell_vnl", "getvnl"); ModuleBase::timer::start("pp_cell_vnl", "getvnl"); using cal_vnl_op = hamilt::cal_vnl_op; @@ -52,84 +34,102 @@ void pseudopot_cell_vnl::getvnl(Device* ctx, using resmem_var_op = base_device::memory::resize_memory_op; using delmem_var_op = base_device::memory::delete_memory_op; using castmem_var_h2d_op = base_device::memory::cast_memory_op; - using castmem_var_h2h_op - = base_device::memory::cast_memory_op; + using castmem_var_h2h_op = base_device::memory::cast_memory_op; using resmem_complex_op = base_device::memory::resize_memory_op, Device>; using delmem_complex_op = base_device::memory::delete_memory_op, Device>; - if (lmaxkb < 0) + if (this->lmaxkb < 0 || this->wfcpw->npwk[ik] == 0) { + ModuleBase::timer::end("pp_cell_vnl", "getvnl"); return; } - const int x1 = (lmaxkb + 1) * (lmaxkb + 1); + const int ylm_count = (this->lmaxkb + 1) * (this->lmaxkb + 1); const int npw = this->wfcpw->npwk[ik]; int* atom_nh = nullptr; int* atom_na = nullptr; int* atom_nb = nullptr; - std::vector h_atom_nh(ucell.ntype); - std::vector h_atom_na(ucell.ntype); - std::vector h_atom_nb(ucell.ntype); - for (int it = 0; it < ucell.ntype; it++) + std::vector host_atom_nh(ucell.ntype); + std::vector host_atom_na(ucell.ntype); + std::vector host_atom_nb(ucell.ntype); + for (int it = 0; it < ucell.ntype; ++it) { - h_atom_nb[it] = ucell.atoms[it].ncpp.nbeta; - h_atom_nh[it] = ucell.atoms[it].ncpp.nh; - h_atom_na[it] = ucell.atoms[it].na; + host_atom_nb[it] = ucell.atoms[it].ncpp.nbeta; + host_atom_nh[it] = ucell.atoms[it].ncpp.nh; + host_atom_na[it] = ucell.atoms[it].na; } - // When the internal memory is large enough, it is better to make vkb1 be the number of pseudopot_cell_vnl. - // We only need to initialize it once as long as the cell is unchanged. - FPTYPE* vkb1 = nullptr; - FPTYPE* gk = nullptr; + + FPTYPE* vkb_radial = nullptr; + FPTYPE* shifted_gk_data = nullptr; FPTYPE* ylm = nullptr; FPTYPE* tab_ptr = this->get_tab_data(); FPTYPE* indv_ptr = this->get_indv_data(); FPTYPE* nhtol_ptr = this->get_nhtol_data(); FPTYPE* nhtolm_ptr = this->get_nhtolm_data(); - resmem_var_op()(ylm, x1 * npw, "VNL::ylm"); - resmem_var_op()(vkb1, nhm * npw, "VNL::vkb1"); + const std::int64_t ylm_elements = static_cast(ylm_count) * npw; + const std::int64_t radial_elements = static_cast(this->nhm) * npw; + const std::int64_t momentum_elements = static_cast(npw) * 3; + resmem_var_op()(ylm, ylm_elements, "VNL::ylm"); + resmem_var_op()(vkb_radial, radial_elements, "VNL::vkb_radial"); - std::vector> gk_vec(npw); + const ModuleBase::Vector3 reduced_vector_potential = vector_potential / ucell.tpiba; + std::vector> shifted_gk(npw); #ifdef _OPENMP #pragma omp parallel for schedule(static) #endif - for (int ig = 0; ig < npw; ig++) + for (int ig = 0; ig < npw; ++ig) + { + shifted_gk[ig] = this->wfcpw->getgpluskcar(ik, ig) + reduced_vector_potential; + } + + // Validate the index in the kernel's precision before obtaining any radial values. + for (int ig = 0; ig < npw; ++ig) { - gk_vec[ig] = this->wfcpw->getgpluskcar(ik, ig); + const FPTYPE x = static_cast(shifted_gk[ig].x); + const FPTYPE y = static_cast(shifted_gk[ig].y); + const FPTYPE z = static_cast(shifted_gk[ig].z); + const FPTYPE q = std::sqrt(x * x + y * y + z * z) * static_cast(ucell.tpiba); + const FPTYPE index = q / static_cast(this->table_dq_); + // Also cover small host/device differences in the norm and division. + const double guarded_index = static_cast(index) * (1.0 + 8.0 * std::numeric_limits::epsilon()); + this->check_vnl_index(guarded_index, false); } + if (this->use_gpu_) { resmem_int_op()(atom_nh, ucell.ntype); resmem_int_op()(atom_nb, ucell.ntype); resmem_int_op()(atom_na, ucell.ntype); - syncmem_int_op()(atom_nh, h_atom_nh.data(), ucell.ntype); - syncmem_int_op()(atom_nb, h_atom_nb.data(), ucell.ntype); - syncmem_int_op()(atom_na, h_atom_na.data(), ucell.ntype); + syncmem_int_op()(atom_nh, host_atom_nh.data(), ucell.ntype); + syncmem_int_op()(atom_nb, host_atom_nb.data(), ucell.ntype); + syncmem_int_op()(atom_na, host_atom_na.data(), ucell.ntype); - resmem_var_op()(gk, npw * 3); - castmem_var_h2d_op()(gk, reinterpret_cast(gk_vec.data()), npw * 3); + resmem_var_op()(shifted_gk_data, momentum_elements); + castmem_var_h2d_op()(shifted_gk_data, reinterpret_cast(shifted_gk.data()), momentum_elements); } else { - atom_nh = h_atom_nh.data(); - atom_nb = h_atom_nb.data(); - atom_na = h_atom_na.data(); + atom_nh = host_atom_nh.data(); + atom_nb = host_atom_nb.data(); + atom_na = host_atom_na.data(); if (std::is_same::value) { - resmem_var_op()(gk, npw * 3); - castmem_var_h2h_op()(gk, reinterpret_cast(gk_vec.data()), npw * 3); + resmem_var_op()(shifted_gk_data, momentum_elements); + castmem_var_h2h_op()(shifted_gk_data, reinterpret_cast(shifted_gk.data()), momentum_elements); } else { - gk = reinterpret_cast(gk_vec.data()); + shifted_gk_data = reinterpret_cast(shifted_gk.data()); } } - ModuleBase::YlmReal::Ylm_Real(ctx, x1, npw, gk, ylm); + ModuleBase::YlmReal::Ylm_Real(ctx, ylm_count, npw, shifted_gk_data, ylm); - std::complex* sk = nullptr; - resmem_complex_op()(sk, ucell.nat * npw); - this->psf->get_sk(ctx, ik, this->wfcpw, sk); + std::complex* structure_factor = nullptr; + const std::int64_t structure_elements = static_cast(ucell.nat) * npw; + resmem_complex_op()(structure_factor, structure_elements); + this->psf->get_sk(ctx, ik, this->wfcpw, structure_factor); cal_vnl_op()(ctx, ucell.ntype, @@ -141,25 +141,25 @@ void pseudopot_cell_vnl::getvnl(Device* ctx, atom_na, atom_nb, atom_nh, - static_cast(PARAM.globalv.dq), + static_cast(this->table_dq_), static_cast(ucell.tpiba), static_cast>(ModuleBase::NEG_IMAG_UNIT), - gk, + shifted_gk_data, ylm, indv_ptr, nhtol_ptr, nhtolm_ptr, tab_ptr, - vkb1, - sk, + vkb_radial, + structure_factor, vkb_in); delmem_var_op()(ylm); - delmem_var_op()(vkb1); - delmem_complex_op()(sk); + delmem_var_op()(vkb_radial); + delmem_complex_op()(structure_factor); if (this->use_gpu_ || std::is_same::value) { - delmem_var_op()(gk); + delmem_var_op()(shifted_gk_data); } if (this->use_gpu_) { @@ -167,26 +167,30 @@ void pseudopot_cell_vnl::getvnl(Device* ctx, delmem_int_op()(atom_nb); delmem_int_op()(atom_na); } + ModuleBase::timer::end("pp_cell_vnl", "getvnl"); -} // end subroutine getvnl +} // Explicit instantiations for CPU/GPU and float/double precision. -// These must stay in the same translation unit as the template definition. template void pseudopot_cell_vnl::getvnl(base_device::DEVICE_CPU*, const UnitCell&, - int const&, + const int&, + const ModuleBase::Vector3&, std::complex*) const; template void pseudopot_cell_vnl::getvnl(base_device::DEVICE_CPU*, const UnitCell&, - int const&, + const int&, + const ModuleBase::Vector3&, std::complex*) const; #if defined(__CUDA) || defined(__ROCM) template void pseudopot_cell_vnl::getvnl(base_device::DEVICE_GPU*, const UnitCell&, - int const&, + const int&, + const ModuleBase::Vector3&, std::complex*) const; template void pseudopot_cell_vnl::getvnl(base_device::DEVICE_GPU*, const UnitCell&, - int const&, + const int&, + const ModuleBase::Vector3&, std::complex*) const; #endif diff --git a/source/source_pw/module_pwdft/vnl_pw_grad.cpp b/source/source_pw/module_pwdft/vnl_pw_grad.cpp index de62b559b9e..cebca7720e9 100644 --- a/source/source_pw/module_pwdft/vnl_pw_grad.cpp +++ b/source/source_pw/module_pwdft/vnl_pw_grad.cpp @@ -1,34 +1,30 @@ -#include "vnl_pw.h" +#include "source_base/math_integral.h" #include "source_base/math_sphbes.h" -#include "source_io/module_parameter/parameter.h" #include "source_base/timer.h" -#include "source_base/math_ylmreal.h" -#include "source_base/math_integral.h" -#include "source_base/math_polyint.h" +#include "vnl_pw.h" #include -void pseudopot_cell_vnl::initgradq_vnl(const UnitCell &cell) +void pseudopot_cell_vnl::ensure_grad_table(const UnitCell& cell) { - const int nbrx = 10; - const int nbrx_nc = 20; - const int ntype = cell.ntype; - if(PARAM.inp.nspin!=4) - { - this->tab_dq.create(ntype, nbrx, PARAM.globalv.nqx); - } - else + if (this->nkb > 0 && this->gradient_version_ != this->table_version_) { - this->tab_dq.create(ntype, nbrx_nc, PARAM.globalv.nqx); + this->initgradq_vnl(cell); } - gradvkb.create(3, nkb, this->wfcpw->npwk_max); +} + +void pseudopot_cell_vnl::initgradq_vnl(const UnitCell& cell) +{ + this->gradient_version_ = this->table_version_; + const int ntype = cell.ntype; + this->tab_dq.create(ntype, this->tab.getBound2(), this->tab.getBound3()); const double pref = ModuleBase::FOUR_PI / sqrt(cell.omega); - for (int it = 0;it < ntype;it++) + for (int it = 0; it < ntype; it++) { const int nbeta = cell.atoms[it].ncpp.nbeta; int kkbeta = cell.atoms[it].ncpp.kkbeta; - if ( (kkbeta%2 == 0) && kkbeta>0 ) + if ((kkbeta % 2 == 0) && kkbeta > 0) { kkbeta--; } @@ -36,154 +32,22 @@ void pseudopot_cell_vnl::initgradq_vnl(const UnitCell &cell) std::vector djl(kkbeta); std::vector aux(kkbeta); - for (int ib = 0;ib < nbeta;ib++) + for (int ib = 0; ib < nbeta; ib++) { const int l = cell.atoms[it].ncpp.lll[ib]; - for (int iq=0; iqtab_dq.getBound3(); iq++) { - const double q = iq * PARAM.globalv.dq; + const double q = iq * this->table_dq_; ModuleBase::Sphbes::dSpherical_Bessel_dx(kkbeta, cell.atoms[it].ncpp.r.data(), q, l, djl.data()); - for (int ir = 0;ir < kkbeta;ir++) + for (int ir = 0; ir < kkbeta; ir++) { - aux[ir] = cell.atoms[it].ncpp.betar(ib, ir) * - djl[ir] * pow(cell.atoms[it].ncpp.r[ir],2); - } + aux[ir] = cell.atoms[it].ncpp.betar(ib, ir) * djl[ir] * pow(cell.atoms[it].ncpp.r[ir], 2); + } double vqint = 0.0; ModuleBase::Integral::Simpson_Integral(kkbeta, aux.data(), cell.atoms[it].ncpp.rab.data(), vqint); this->tab_dq(it, ib, iq) = vqint * pref; } } } - -} - -void pseudopot_cell_vnl::getgradq_vnl(const UnitCell& ucell, - const int ik) -{ - if(PARAM.inp.test_pp) ModuleBase::TITLE("pseudopot_cell_vnl","getvnl"); - ModuleBase::timer::start("pp_cell_vnl","getvnl"); - - if(lmaxkb < 0) - { - return; - } - - const int npw = this->wfcpw->npwk[ik]; - - // When the internal memory is large enough, it is better to make tmpgradvkb and tmpvkb be the number of pseudopot_cell_vnl - // We only need to initialize them once as long as the cell is unchanged. - ModuleBase::realArray tmpgradvkb(3, nhm, npw); - ModuleBase::matrix tmpvkb(nhm, npw); - std::vector vq(npw); - std::vector dvq(npw); - - const int x1= (lmaxkb + 1)*(lmaxkb + 1); - - ModuleBase::matrix ylm(x1, npw); - std::vector dylm(3); - dylm[0].create(x1, npw); - dylm[1].create(x1, npw); - dylm[2].create(x1, npw); - - std::vector> gk(npw); - for (int ig = 0;ig < npw;ig++) - { - gk[ig] = this->wfcpw->getgpluskcar(ik,ig); - } - - ModuleBase::YlmReal::grad_Ylm_Real(x1, npw, gk.data(), ylm, dylm[0], dylm[1], dylm[2]); - - // GPU path skips vkb allocation in init(); allocate now if needed - if (this->vkb.nc == 0 && this->nkb > 0 && this->vkbnc > 0) { - this->vkb.create(this->nkb, this->vkbnc); - } - - int jkb = 0; - for(int it = 0;it < ucell.ntype;it++) - { - // calculate beta in G-space using an interpolation table - const int nbeta = ucell.atoms[it].ncpp.nbeta; - const int nh = ucell.atoms[it].ncpp.nh; - int nb0 = -1; - for( int ih = 0; ih < nh; ++ih) - { - int nb = this->indv(it, ih); - if(nb != nb0) - { - for (int ig = 0;ig < npw;++ig) - { - const double gnorm = gk[ig].norm() * ucell.tpiba; - vq [ig] = ModuleBase::PolyInt::Polynomial_Interpolation( - this->tab, it, nb, PARAM.globalv.nqx, PARAM.globalv.dq, gnorm ); - dvq[ig] =ModuleBase::PolyInt::Polynomial_Interpolation( - this->tab_dq, it, nb, PARAM.globalv.nqx, PARAM.globalv.dq, gnorm ); - } - nb0 = nb; - } - double lmmat[9] = {0, 0, 1, -1, 0, 0, 0, -1, 0}; - for(int id = 0; id < 3; ++id) - { - const int lm = static_cast( nhtolm(it, ih) ); - for (int ig = 0;ig < npw;++ig) - { - ModuleBase::Vector3 gg = gk[ig]; - double ggnorm = gg.norm(); - if(ggnorm < 1e-8) - { - if(lm == 0 || lm > 3) - { - tmpgradvkb(id, ih, ig) = 0.0; - } - else//lm = 1,2,3; l = 1 - { - //q \to 0 : \nabla(f(q)Y(\hat{q})) = f(q)/q*sqrt(3/4/pi)*vec(-delta(lm,2), -delta(lm,3), delta(lm,1)) - // tmpgradvkb(id, ih, ig) = 0.0; - tmpgradvkb(id, ih, ig) = dvq[ig] * sqrt(3.0/4.0/M_PI) * lmmat[(lm-1)*3 + id]; - } - } - else - { - tmpgradvkb(id, ih, ig) = ylm(lm, ig) * dvq[ig] * gg[id] / ggnorm - + dylm[id](lm, ig) / this->wfcpw->tpiba - * vq[ig]; // note: dylm/d(tpiba * gx) = 1/tpiba * dylm/dgx - } - tmpvkb(ih, ig) = ylm(lm,ig) * vq[ig]; - } - } - } - - // vkb1 contains all betas including angular part for type nt - // now add the structure factor and factor (-i)^l - for (int ia=0; ia *sk = this->psf->get_sk(ik, it, ia, this->wfcpw); - - for (int ih = 0;ih < nh;++ih) - { - std::complex pref = pow( ModuleBase::NEG_IMAG_UNIT, nhtol(it, ih)); - std::complex* pvkb = &this->vkb(jkb, 0); - for (int id = 0; id < 3 ; ++id) - { - std::complex* pgvkb = &this->gradvkb(id, jkb, 0); - for (int ig = 0;ig < npw;++ig) - { - std::complex skig = sk[ig]; - pvkb[ig] = tmpvkb(ih, ig) * skig * pref; - // std::complex dskig = ModuleBase::NEG_IMAG_UNIT * (ucell.atoms[it].tau[ia][id] * this->wfcpw->lat0) * skig; - // pgvkb[ig] = tmpgradvkb(id, ih, ig) * skig * pref + tmpvkb(ih, ig) * dskig * pref; - // The second term will be eliminate when doing Dij or we can say (\nabla_q+\nabla_q')S(q'-q) = 0 - pgvkb[ig] = tmpgradvkb(id, ih, ig) * skig * pref; - } - } //end id - ++jkb; - } // end ih - - delete [] sk; - } // end ia - } // enddo - - ModuleBase::timer::end("pp_cell_vnl","getvnl"); - - return; } diff --git a/source/source_pw/module_pwdft/vnl_pw_init_vnl.cpp b/source/source_pw/module_pwdft/vnl_pw_init_vnl.cpp index fb02ef571ba..f44f2187d24 100644 --- a/source/source_pw/module_pwdft/vnl_pw_init_vnl.cpp +++ b/source/source_pw/module_pwdft/vnl_pw_init_vnl.cpp @@ -1,15 +1,11 @@ -#include "vnl_pw.h" - -#include "source_io/module_parameter/parameter.h" #include "source_base/clebsch_gordan_coeff.h" #include "source_base/global_function.h" #include "source_base/global_variable.h" -#include "source_base/math_integral.h" -#include "source_base/math_polyint.h" -#include "source_base/math_sphbes.h" #include "source_base/math_ylmreal.h" #include "source_base/parallel_reduce.h" #include "source_base/timer.h" +#include "source_io/module_parameter/parameter.h" +#include "vnl_pw.h" #include #include @@ -33,6 +29,12 @@ void pseudopot_cell_vnl::init_vnl(UnitCell& cell, const ModulePW::PW_Basis* rho_ ModuleBase::TITLE("pseudopot_cell_vnl", "init_vnl"); ModuleBase::timer::start("ppcell_vnl", "init_vnl"); + this->table_dq_ = PARAM.globalv.dq; + if (!std::isfinite(this->table_dq_) || this->table_dq_ <= 0.0) + { + ModuleBase::WARNING_QUIT("init_vnl", "Radial projector table spacing must be finite and positive."); + } + ++this->table_version_; this->omega_old = cell.omega; // from init_us_1 @@ -53,9 +55,10 @@ void pseudopot_cell_vnl::init_vnl(UnitCell& cell, const ModulePW::PW_Basis* rho_ if (cell.atoms[it].ncpp.tvanp) { cell.atoms[it].ncpp.nqlc = std::min(cell.atoms[it].ncpp.nqlc, lmaxq); - if (cell.atoms[it].ncpp.nqlc < 0) { + if (cell.atoms[it].ncpp.nqlc < 0) + { cell.atoms[it].ncpp.nqlc = 0; -} + } } } @@ -168,18 +171,19 @@ void pseudopot_cell_vnl::init_vnl(UnitCell& cell, const ModulePW::PW_Basis* rho_ { for (int is2 = 0; is2 < 2; ++is2) { - this->dvan_so(ijs, it, ip, ip2) - = cell.atoms[it].ncpp.dion(ir, is) * soc.fcoef(it, is1, is2, ip, ip2); + this->dvan_so(ijs, it, ip, ip2) = cell.atoms[it].ncpp.dion(ir, is) * soc.fcoef(it, is1, is2, ip, ip2); ++ijs; - if (ir != is) { + if (ir != is) + { soc.fcoef(it, is1, is2, ip, ip2) = std::complex(0.0, 0.0); -} + } } } } } } - else { + else + { for (int ip = 0; ip < Nprojectors; ip++) { for (int ip2 = 0; ip2 < Nprojectors; ip2++) @@ -200,7 +204,7 @@ void pseudopot_cell_vnl::init_vnl(UnitCell& cell, const ModulePW::PW_Basis* rho_ } } } -} + } } // e) It computes the coefficients c_{LM}^{nm} which relates the @@ -252,8 +256,7 @@ void pseudopot_cell_vnl::init_vnl(UnitCell& cell, const ModulePW::PW_Basis* rho_ { for (int is = 0; is < 2; is++) { - this->qq_so(it, ijs, kh, lh) += cell.omega * qgm.real() - * soc.fcoef(it, is1, is, kh, ih) + this->qq_so(it, ijs, kh, lh) += cell.omega * qgm.real() * soc.fcoef(it, is1, is, kh, ih) * soc.fcoef(it, is, is2, jh, lh); } } @@ -311,40 +314,7 @@ void pseudopot_cell_vnl::init_vnl(UnitCell& cell, const ModulePW::PW_Basis* rho_ // fill the interpolation table tab ************************************************************/ - const double pref = ModuleBase::FOUR_PI / sqrt(cell.omega); - this->tab.zero_out(); - GlobalV::ofs_running << "\n Init Non-Local PseudoPotential table : "; - for (int it = 0; it < cell.ntype; it++) - { - const int nbeta = cell.atoms[it].ncpp.nbeta; - int kkbeta = cell.atoms[it].ncpp.kkbeta; - - // mohan modify 2008-3-31 - // mohan add kkbeta>0 2009-2-27 - if ((kkbeta % 2 == 0) && kkbeta > 0) - { - kkbeta--; - } - - std::vector jl(kkbeta); - std::vector aux(kkbeta); - for (int ib = 0; ib < nbeta; ib++) - { - const int l = cell.atoms[it].ncpp.lll[ib]; - for (int iq = 0; iq < PARAM.globalv.nqx; iq++) - { - const double q = iq * PARAM.globalv.dq; - ModuleBase::Sphbes::Spherical_Bessel(kkbeta, cell.atoms[it].ncpp.r.data(), q, l, jl.data()); - for (int ir = 0; ir < kkbeta; ir++) - { - aux[ir] = cell.atoms[it].ncpp.betar(ib, ir) * jl[ir] * cell.atoms[it].ncpp.r[ir]; - } - double vqint=0.0; - ModuleBase::Integral::Simpson_Integral(kkbeta, aux.data(), cell.atoms[it].ncpp.rab.data(), vqint); - this->tab(it, ib, iq) = vqint * pref; - } - } - } + this->fill_vnl_table(cell); if (this->use_gpu_) { if (PARAM.globalv.has_float_data) diff --git a/source/source_pw/module_pwdft/vnl_pw_table.cpp b/source/source_pw/module_pwdft/vnl_pw_table.cpp new file mode 100644 index 00000000000..1d86cca94d1 --- /dev/null +++ b/source/source_pw/module_pwdft/vnl_pw_table.cpp @@ -0,0 +1,118 @@ +#include "source_base/global_function.h" +#include "source_base/math_integral.h" +#include "source_base/math_sphbes.h" +#include "source_pw/module_pwdft/vnl_pw.h" + +#include +#include +#include +#include + +void pseudopot_cell_vnl::fill_vnl_table(const UnitCell& cell) +{ + ++this->table_version_; + const double pref = ModuleBase::FOUR_PI / std::sqrt(cell.omega); + this->tab.zero_out(); + for (int it = 0; it < cell.ntype; ++it) + { + const Atom_pseudo& pp = cell.atoms[it].ncpp; + int mesh = pp.kkbeta; + if (mesh > 0 && mesh % 2 == 0) + { + --mesh; + } + std::vector jl(mesh); + std::vector aux(mesh); + for (int ib = 0; ib < pp.nbeta; ++ib) + { + for (int iq = 0; iq < this->tab.getBound3(); ++iq) + { + ModuleBase::Sphbes::Spherical_Bessel(mesh, pp.r.data(), iq * this->table_dq_, pp.lll[ib], jl.data()); + for (int ir = 0; ir < mesh; ++ir) + { + aux[ir] = pp.betar(ib, ir) * jl[ir] * pp.r[ir]; + } + double integral = 0.0; + ModuleBase::Integral::Simpson_Integral(mesh, aux.data(), pp.rab.data(), integral); + this->tab(it, ib, iq) = integral * pref; + } + } + } +} + +void pseudopot_cell_vnl::sync_vnl_table() +{ + const int size = this->tab.getSize(); + if (this->use_gpu_) + { + resmem_dd_op()(this->d_tab, size); + syncmem_d2d_h2d_op()(this->d_tab, this->tab.ptr, size); + if (this->s_tab != nullptr) + { + resmem_sd_op()(this->s_tab, size); + castmem_d2s_h2d_op()(this->s_tab, this->tab.ptr, size); + } + } + else + { + this->d_tab = this->tab.ptr; + if (this->s_tab != nullptr) + { + resmem_sh_op()(this->s_tab, size); + castmem_d2s_h2h_op()(this->s_tab, this->tab.ptr, size); + } + } +} + +void pseudopot_cell_vnl::ensure_vnl_range(const UnitCell& cell, const double& qmax) +{ + const double dq = this->table_dq_; + if (!std::isfinite(qmax) || qmax < 0.0 || !std::isfinite(dq) || dq <= 0.0) + { + ModuleBase::WARNING_QUIT("ensure_vnl_range", "Non-finite or invalid projector momentum/table spacing."); + } + if (this->nkb == 0) + { + return; + } + // Account for float arithmetic in the reduced momentum, norm and table index. + const double padded_q = qmax * (1.0 + 32.0 * std::numeric_limits::epsilon()) + 2.0 * dq; + const double required = std::floor(padded_q / dq) + 4.0; + const int old_size = this->tab.getBound3(); + if (required <= old_size) + { + return; + } + const double capacity = std::max(required + 100.0, std::ceil(1.25 * old_size)); + const double channels = static_cast(this->tab.getBound1()) * this->tab.getBound2(); + if (!std::isfinite(capacity) || channels <= 0.0 || capacity > std::numeric_limits::max() / channels || !std::isfinite(cell.omega) + || cell.omega <= 0.0) + { + ModuleBase::WARNING_QUIT("ensure_vnl_range", "Requested projector table exceeds array limits or has invalid volume."); + } + const bool has_derivative = this->tab_dq.getBound3() >= 4; + try + { + this->tab.create(this->tab.getBound1(), this->tab.getBound2(), static_cast(capacity)); + this->fill_vnl_table(cell); + if (has_derivative) + { + this->initgradq_vnl(cell); + } + this->sync_vnl_table(); + } + catch (const std::bad_alloc&) + { + ModuleBase::WARNING_QUIT("ensure_vnl_range", "Unable to allocate the expanded projector table."); + } +} + +void pseudopot_cell_vnl::check_vnl_index(const double& index, const bool& derivative) const +{ + if (!std::isfinite(index) || index < 0.0 || std::floor(index) + 3.0 >= this->tab.getBound3() + || (derivative && std::floor(index) + 3.0 >= this->tab_dq.getBound3())) + { + ModuleBase::WARNING_QUIT("check_vnl_index", + "Shifted projector momentum exceeds the prepared radial table; prepare its range before evaluation."); + } +} diff --git a/source/source_pw/module_pwdft/yukawa_screening.cpp b/source/source_pw/module_pwdft/yukawa_screening.cpp index 61498299b3f..f9c5f6935b5 100644 --- a/source/source_pw/module_pwdft/yukawa_screening.cpp +++ b/source/source_pw/module_pwdft/yukawa_screening.cpp @@ -43,10 +43,10 @@ void YukawaScreening::init(const UnitCell& cell, { this->Fk_[it][l][n].resize(l + 1, 0.0); } - - this->U_Yukawa_[it][l].resize(N, 0.0); - this->J_Yukawa_[it][l].resize(N, 0.0); } + // only the first radial channel (n=0) of each l carries U/J + this->U_Yukawa_[it].resize(NL, 0.0); + this->J_Yukawa_[it].resize(NL, 0.0); } } @@ -224,25 +224,25 @@ void YukawaScreening::cal_slater_UJ(const UnitCell& ucell, if (L == 1) { - this->U_Yukawa_[T][L][0] = this->Fk_[T][L][0][0]; - this->J_Yukawa_[T][L][0] = this->Fk_[T][L][0][1] / 5.0; + this->U_Yukawa_[T][L] = this->Fk_[T][L][0][0]; + this->J_Yukawa_[T][L] = this->Fk_[T][L][0][1] / 5.0; } else if (L == 2) { - this->U_Yukawa_[T][L][0] = this->Fk_[T][L][0][0]; - this->J_Yukawa_[T][L][0] = (this->Fk_[T][L][0][1] + this->Fk_[T][L][0][2]) / 14.0; + this->U_Yukawa_[T][L] = this->Fk_[T][L][0][0]; + this->J_Yukawa_[T][L] = (this->Fk_[T][L][0][1] + this->Fk_[T][L][0][2]) / 14.0; } else if (L == 3) { - this->U_Yukawa_[T][L][0] = this->Fk_[T][L][0][0]; - this->J_Yukawa_[T][L][0] = (286.0 * this->Fk_[T][L][0][1] + this->U_Yukawa_[T][L] = this->Fk_[T][L][0][0]; + this->J_Yukawa_[T][L] = (286.0 * this->Fk_[T][L][0][1] + 195.0 * this->Fk_[T][L][0][2] + 250.0 * this->Fk_[T][L][0][3]) / 6435.0; } // Hartree to Rydeberg - this->U_Yukawa_[T][L][0] *= 2.0; - this->J_Yukawa_[T][L][0] *= 2.0; + this->U_Yukawa_[T][L] *= 2.0; + this->J_Yukawa_[T][L] *= 2.0; } // end if } // end L } // end T diff --git a/source/source_pw/module_pwdft/yukawa_screening.h b/source/source_pw/module_pwdft/yukawa_screening.h index b81c9a2266a..c908825f338 100644 --- a/source/source_pw/module_pwdft/yukawa_screening.h +++ b/source/source_pw/module_pwdft/yukawa_screening.h @@ -46,13 +46,14 @@ class YukawaScreening const LCAO_Orbitals* orb); double get_lambda() const { return lambda_; } - double get_U(int it, int l, int n) const { return U_Yukawa_[it][l][n]; } - double get_J(int it, int l, int n) const { return J_Yukawa_[it][l][n]; } - /// effective U-J of the correlated orbital (n = 0) for atom type it + /// U/J of the correlated orbital (first radial channel) of atom type it + double get_U(int it, int l) const { return U_Yukawa_[it][l]; } + double get_J(int it, int l) const { return J_Yukawa_[it][l]; } + /// effective U-J of the correlated orbital for atom type it double get_Ueff(int it) const { const int l = l_channel_[it]; - return U_Yukawa_[it][l][0] - J_Yukawa_[it][l][0]; + return U_Yukawa_[it][l] - J_Yukawa_[it][l]; } private: @@ -65,8 +66,8 @@ class YukawaScreening double yukawa_lambda_cfg_ = 0.0; std::vector l_channel_; std::vector>>> Fk_; - std::vector>> U_Yukawa_; - std::vector>> J_Yukawa_; + std::vector> U_Yukawa_; + std::vector> J_Yukawa_; }; #endif diff --git a/source/source_pw/module_stodft/CMakeLists.txt b/source/source_pw/module_stodft/CMakeLists.txt index b0e160c5b21..e284ca8ad17 100644 --- a/source/source_pw/module_stodft/CMakeLists.txt +++ b/source/source_pw/module_stodft/CMakeLists.txt @@ -1,5 +1,6 @@ list(APPEND hamilt_stodft_srcs - hamilt_sdft_pw.cpp + sto_hamilt_pw.cpp + sto_hsolver_pw.cpp sto_iter.cpp sto_che.cpp sto_wf.cpp diff --git a/source/source_pw/module_stodft/hamilt_sdft_pw.cpp b/source/source_pw/module_stodft/hamilt_sdft_pw.cpp deleted file mode 100644 index 90151972285..00000000000 --- a/source/source_pw/module_stodft/hamilt_sdft_pw.cpp +++ /dev/null @@ -1,72 +0,0 @@ -#include "hamilt_sdft_pw.h" -#include "source_base/timer.h" -#include "kernels/hpsi_norm_op.h" - -namespace hamilt -{ - -template -HamiltSdftPW::HamiltSdftPW(elecstate::Potential* pot_in, - ModulePW::PW_Basis_K* wfc_basis, - K_Vectors* p_kv, - pseudopot_cell_vnl* nlpp, - const UnitCell* ucell, - const int& npol, - Real* emin_in, - Real* emax_in) - : HamiltPW(pot_in, wfc_basis, p_kv, nlpp, nullptr, ucell, nullptr), ngk(p_kv->ngk) -{ - this->classname = "HamiltSdftPW"; - this->npwk_max = wfc_basis->npwk_max; - this->npol = npol; - this->emin = emin_in; - this->emax = emax_in; -} - -template -void HamiltSdftPW::hPsi(const T* psi_in, T* hpsi, const int& nbands) -{ - auto call_act = [&, this](const Operator* op, const bool& is_first_node) -> void { - op->act(nbands, this->npwk_max, this->npol, psi_in, hpsi, this->ngk[op->get_ik()], is_first_node); - }; - - ModuleBase::timer::start("HamiltSdftPW", "hPsi"); - call_act(this->ops, true); // first node - Operator* node((Operator*)this->ops->next_op); - while (node != nullptr) - { - call_act(node, false); // other nodes - node = (Operator*)(node->next_op); - } - ModuleBase::timer::end("HamiltSdftPW", "hPsi"); - - return; -} - -template -void HamiltSdftPW::hPsi_norm(const T* psi_in, T* hpsi_norm, const int& nbands) -{ - ModuleBase::timer::start("HamiltSdftPW", "hPsi_norm"); - - this->hPsi(psi_in, hpsi_norm, nbands); - - const int ik = this->ops->get_ik(); - const int npwk_max = this->npwk_max; - const int npwk = this->ngk[ik]; - const Real emin = *this->emin; - const Real emax = *this->emax; - const Real Ebar = (emin + emax) / 2; - const Real DeltaE = (emax - emin) / 2; - - hpsi_norm_op()(this->ctx, nbands, npwk_max, npwk, Ebar, DeltaE, hpsi_norm, psi_in); - ModuleBase::timer::end("HamiltSdftPW", "hPsi_norm"); -} - -template class HamiltSdftPW, base_device::DEVICE_CPU>; -template class HamiltSdftPW, base_device::DEVICE_CPU>; -#if ((defined __CUDA) || (defined __ROCM)) -template class HamiltSdftPW, base_device::DEVICE_GPU>; -template class HamiltSdftPW, base_device::DEVICE_GPU>; -#endif - -} // namespace hamilt diff --git a/source/source_pw/module_stodft/hamilt_sdft_pw.h b/source/source_pw/module_stodft/hamilt_sdft_pw.h deleted file mode 100644 index 282ebec4247..00000000000 --- a/source/source_pw/module_stodft/hamilt_sdft_pw.h +++ /dev/null @@ -1,67 +0,0 @@ -#ifndef HAMILTSDFTPW_H -#define HAMILTSDFTPW_H - -#include "source_pw/module_pwdft/hamilt_pw.h" - -namespace hamilt -{ - -template -class HamiltSdftPW : public HamiltPW -{ - public: - using Real = typename GetTypeReal::type; - /** - * @brief Construct a new HamiltSdftPW object - * - * @param pot_in potential - * @param wfc_basis pw basis for wave functions - * @param p_kv k vectors - * @param npol the length of wave function is npol * npwk_max - * @param emin_in Emin of the Hamiltonian - * @param emax_in Emax of the Hamiltonian - */ - HamiltSdftPW(elecstate::Potential* pot_in, - ModulePW::PW_Basis_K* wfc_basis, - K_Vectors* p_kv, - pseudopot_cell_vnl* nlpp, - const UnitCell* ucell, - const int& npol, - Real* emin_in, - Real* emax_in); - /** - * @brief Destroy the HamiltSdftPW object - * - */ - ~HamiltSdftPW(){}; - - /** - * @brief Calculate \hat{H}|psi> - * - * @param psi_in input wave function - * @param hpsi output wave function - * @param nbands number of bands - */ - void hPsi(const T* psi_in, T* hpsi, const int& nbands = 1); - - /** - * @brief Calculate \hat{H}|psi> and normalize it - * - * @param psi_in input wave function - * @param hpsi output wave function - * @param nbands number of bands - */ - void hPsi_norm(const T* psi_in, T* hpsi, const int& nbands = 1); - - Real* emin = nullptr; ///< Emin of the Hamiltonian - Real* emax = nullptr; ///< Emax of the Hamiltonian - - private: - int npwk_max = 0; ///< maximum number of plane waves - int npol = 0; ///< number of polarizations - std::vector& ngk; ///< number of G vectors -}; - -} // namespace hamilt - -#endif diff --git a/source/source_pw/module_stodft/sto_dos.cpp b/source/source_pw/module_stodft/sto_dos.cpp index 884679a8b83..eb78c6700f3 100644 --- a/source/source_pw/module_stodft/sto_dos.cpp +++ b/source/source_pw/module_stodft/sto_dos.cpp @@ -25,7 +25,7 @@ Sto_DOS::Sto_DOS(ModulePW::PW_Basis_K* p_wfcpw_in, this->p_elec = p_elec_in; this->p_psi = p_psi_in; this->p_hamilt = p_hamilt_in; - this->p_hamilt_sto = static_cast>*>(p_hamilt_in); + this->p_hamilt_sto = static_cast>*>(p_hamilt_in); this->p_stowf = p_stowf_in; this->nbands_ks = p_psi_in->get_nbands(); this->nbands_sto = p_stowf_in->nchi; @@ -51,7 +51,7 @@ void Sto_DOS::decide_param(const int& dos_nche, this->nbands_sto, this->p_kv, reinterpret_cast, Device>*>(this->p_stowf), - reinterpret_cast, Device>*>(this->p_hamilt_sto)); + reinterpret_cast, Device>*>(this->p_hamilt_sto)); if (dos_setemax) { this->emax = dos_emax_ev; @@ -124,7 +124,7 @@ void Sto_DOS::caldos(const double sigmain, const double de, cons p_stowf->chi0->fix_k(ik); pchi = p_stowf->chi0->get_pointer(); } - auto hchi_norm = std::bind(&hamilt::HamiltSdftPW>::hPsi_norm, + auto hchi_norm = std::bind(&StoHamiltPW>::hPsi_norm, p_hamilt_sto, std::placeholders::_1, std::placeholders::_2, @@ -247,10 +247,10 @@ void Sto_DOS::caldos(const double sigmain, const double de, cons double maxerror = 0; double sum = 0; - ofsdos << ndos << " # number of points" << std::endl; - ofsdos << "#" << std::setw(19) << "energy(eV)" - << std::setw(20) << "dos(eV^-1)" << std::setw(20) << "sum" - << std::setw(20) << "error(eV^-1)" << std::endl; + ofsdos << ndos << " # number of points" << std::endl; + ofsdos << "#" << std::setw(19) << "energy(eV)" + << std::setw(20) << "dos(eV^-1)" << std::setw(20) << "sum" + << std::setw(20) << "error(eV^-1)" << std::endl; for (int ie = 0; ie < ndos; ++ie) { double tmperror = 2.0 * std::abs(error[ie]); diff --git a/source/source_pw/module_stodft/sto_dos.h b/source/source_pw/module_stodft/sto_dos.h index 68a433eb24c..ffd1cfc08c0 100644 --- a/source/source_pw/module_stodft/sto_dos.h +++ b/source/source_pw/module_stodft/sto_dos.h @@ -1,7 +1,7 @@ #ifndef STO_DOS #define STO_DOS #include "source_estate/elecstate.h" -#include "source_pw/module_stodft/hamilt_sdft_pw.h" +#include "source_pw/module_stodft/sto_hamilt_pw.h" #include "source_pw/module_stodft/sto_che.h" #include "source_pw/module_stodft/sto_func.h" #include "source_pw/module_stodft/sto_wf.h" @@ -65,7 +65,7 @@ class Sto_DOS = nullptr; ///< pointer to the stochastic wavefunctions Sto_Func stofunc; ///< functions - hamilt::HamiltSdftPW>* p_hamilt_sto = nullptr; ///< pointer to the Hamiltonian for sDFT + StoHamiltPW>* p_hamilt_sto = nullptr; ///< pointer to the Hamiltonian for sDFT }; #endif // STO_DOS \ No newline at end of file diff --git a/source/source_pw/module_stodft/sto_elecond.cpp b/source/source_pw/module_stodft/sto_elecond.cpp index 85516450f99..b705960abc7 100644 --- a/source/source_pw/module_stodft/sto_elecond.cpp +++ b/source/source_pw/module_stodft/sto_elecond.cpp @@ -33,7 +33,7 @@ Sto_EleCond::Sto_EleCond(UnitCell* p_ucell_in, : EleCond(p_ucell_in, p_kv_in, p_elec_in, p_wfcpw_in, p_psi_in, p_ppcell_in) { this->p_hamilt = p_hamilt_in; - this->p_hamilt_sto = static_cast, Device>*>(p_hamilt_in); + this->p_hamilt_sto = static_cast, Device>*>(p_hamilt_in); this->p_stowf = p_stowf_in; this->nbands_ks = p_psi_in->get_nbands(); this->nbands_sto = p_stowf_in->nchi; @@ -42,7 +42,7 @@ Sto_EleCond::Sto_EleCond(UnitCell* p_ucell_in, #ifdef __FLOAT_FFTW if(!std::is_same::value) { - this->hamilt_sto_ = new hamilt::HamiltSdftPW, Device>(p_elec_in->pot, p_wfcpw_in, p_kv_in, p_ppcell_in, p_ucell_in, 1, &this->low_emin_, &this->low_emax_); + this->hamilt_sto_ = new StoHamiltPW, Device>(p_elec_in->pot, p_wfcpw_in, p_kv_in, p_ppcell_in, p_ucell_in, 1, &this->low_emin_, &this->low_emax_); } #endif } @@ -149,33 +149,33 @@ void Sto_EleCond::decide_nche(const FPTYPE dt, } template -void Sto_EleCond::cal_jmatrix(hamilt::HamiltSdftPW, Device>* hamilt, - const psi::Psi, Device>& kspsi_all, - const psi::Psi, Device>& vkspsi, - const double* en, - const double* en_all, - std::complex* leftfact, - std::complex* rightfact, - psi::Psi, Device>& leftchi, - psi::Psi, Device>& rightchi, - psi::Psi, Device>& left_hchi, - psi::Psi, Device>& right_hchi, - psi::Psi, Device>& batch_vchi, - psi::Psi, Device>& batch_vhchi, +void Sto_EleCond::cal_jmatrix(StoHamiltPW, Device>* hamilt, + const psi::Psi, Device>& kspsi_all, + const psi::Psi, Device>& vkspsi, + const double* en, + const double* en_all, + std::complex* leftfact, + std::complex* rightfact, + psi::Psi, Device>& leftchi, + psi::Psi, Device>& rightchi, + psi::Psi, Device>& left_hchi, + psi::Psi, Device>& right_hchi, + psi::Psi, Device>& batch_vchi, + psi::Psi, Device>& batch_vhchi, #ifdef __MPI - psi::Psi, Device>& chi_all, - psi::Psi, Device>& hchi_all, - void* gatherinfo_ks, - void* gatherinfo_sto, + psi::Psi, Device>& chi_all, + psi::Psi, Device>& hchi_all, + void* gatherinfo_ks, + void* gatherinfo_sto, #endif - const int& bsize_psi, - std::complex* j1, - std::complex* j2, - std::complex* tmpj, - hamilt::Velocity& velop, - const int& ik, - const std::complex& factor, - const int bandinfo[6]) + const int& bsize_psi, + std::complex* j1, + std::complex* j2, + std::complex* tmpj, + hamilt::Velocity& velop, + const int& ik, + const std::complex& factor, + const int bandinfo[6]) { ModuleBase::timer::start("Sto_EleCond", "cal_jmatrix"); const std::complex float_factor = factor; @@ -556,14 +556,14 @@ void Sto_EleCond::sKG(const int& smear_type, this->low_emin_ = static_cast(*this->stofunc.Emin); this->low_emax_ = static_cast(*this->stofunc.Emax); lowfunc.set_E_range(&low_emin_, &low_emax_); - hamilt::HamiltSdftPW* p_low_hamilt = nullptr; + StoHamiltPW* p_low_hamilt = nullptr; if(hamilt_sto_ != nullptr) { p_low_hamilt = hamilt_sto_; } else { - p_low_hamilt = reinterpret_cast, Device>*>(this->p_hamilt_sto); + p_low_hamilt = reinterpret_cast, Device>*>(this->p_hamilt_sto); } // Init Chebyshev @@ -794,12 +794,12 @@ void Sto_EleCond::sKG(const int& smear_type, auto nroot_fd = std::bind(&Sto_Func::nroot_fd, &this->stofunc, std::placeholders::_1); che.calcoef_real(nroot_fd); - auto hchi_norm = std::bind(&hamilt::HamiltSdftPW, Device>::hPsi_norm, + auto hchi_norm = std::bind(&StoHamiltPW, Device>::hPsi_norm, p_hamilt_sto, std::placeholders::_1, std::placeholders::_2, std::placeholders::_3); - auto hchi_norm_low = std::bind(&hamilt::HamiltSdftPW::hPsi_norm, + auto hchi_norm_low = std::bind(&StoHamiltPW::hPsi_norm, p_low_hamilt, std::placeholders::_1, std::placeholders::_2, diff --git a/source/source_pw/module_stodft/sto_elecond.h b/source/source_pw/module_stodft/sto_elecond.h index c54a2682f69..4845525dcc6 100644 --- a/source/source_pw/module_stodft/sto_elecond.h +++ b/source/source_pw/module_stodft/sto_elecond.h @@ -2,8 +2,8 @@ #define STOELECOND_H #include "source_hamilt/hamilt.h" -#include "source_hsolver/hsolver_pw_sdft.h" #include "source_pw/module_pwdft/elecond.h" +#include "source_pw/module_stodft/sto_hsolver_pw.h" #include "source_pw/module_stodft/sto_wf.h" template @@ -82,8 +82,8 @@ class Sto_EleCond : protected EleCond Stochastic_WF, Device>* p_stowf = nullptr; ///< pointer to the stochastic wavefunctions Sto_Func stofunc; ///< functions - hamilt::HamiltSdftPW, Device>* p_hamilt_sto = nullptr; ///< pointer to the Hamiltonian for sDFT - hamilt::HamiltSdftPW, Device>* hamilt_sto_ = nullptr; ///< pointer to the Hamiltonian for sDFT + StoHamiltPW, Device>* p_hamilt_sto = nullptr; ///< pointer to the Hamiltonian for sDFT + StoHamiltPW, Device>* hamilt_sto_ = nullptr; ///< pointer to the Hamiltonian for sDFT lowTYPE low_emin_ = 0; ///< Emin of the Hamiltonian for sDFT lowTYPE low_emax_ = 0; ///< Emax of the Hamiltonian for sDFT protected: @@ -91,7 +91,7 @@ class Sto_EleCond : protected EleCond * @brief calculate Jmatrix * */ - void cal_jmatrix(hamilt::HamiltSdftPW, Device>* hamilt, + void cal_jmatrix(StoHamiltPW, Device>* hamilt, const psi::Psi, Device>& kspsi_all, const psi::Psi, Device>& vkspsi, const double* en, diff --git a/source/source_pw/module_stodft/sto_forces.cpp b/source/source_pw/module_stodft/sto_forces.cpp index e092b8f9327..3f455dec4a5 100644 --- a/source/source_pw/module_stodft/sto_forces.cpp +++ b/source/source_pw/module_stodft/sto_forces.cpp @@ -16,7 +16,11 @@ #include "source_hamilt/module_xc/xc_functional.h" template -void Sto_Forces::cal_stoforce(ModuleBase::matrix& force, +void Sto_Forces::cal_stoforce(const int nspin, + const bool domag, + const bool domag_z, + const int gga_grad, + ModuleBase::matrix& force, const elecstate::ElecState& elec, ModulePW::PW_Basis* rho_basis, ModuleSymmetry::Symmetry* p_symm, @@ -44,7 +48,8 @@ void Sto_Forces::cal_stoforce(ModuleBase::matrix& force, this->cal_force_loc(ucell, forcelc, rho_basis, locpp.vloc, chr); this->cal_force_ew(ucell,forceion, rho_basis, p_sf); this->cal_sto_force_nl(forcenl, wg, pkv, wfc_basis, p_sf, nlpp, ucell, psi, stowf); - this->cal_force_cc(forcecc, rho_basis, chr, locpp.numeric, ucell); + this->cal_force_cc(forcecc, rho_basis, chr, locpp.numeric, ucell, + nspin, domag, domag_z, gga_grad); this->cal_force_scc(forcescc, rho_basis, elec.vnew, elec.vnew_exist, locpp.numeric, ucell); // impose total force = 0 diff --git a/source/source_pw/module_stodft/sto_forces.h b/source/source_pw/module_stodft/sto_forces.h index d8416b916bc..4b631f72a5f 100644 --- a/source/source_pw/module_stodft/sto_forces.h +++ b/source/source_pw/module_stodft/sto_forces.h @@ -21,7 +21,11 @@ class Sto_Forces : public Forces Sto_Forces(const int nat_in) : Forces(nat_in){}; ~Sto_Forces(){}; - void cal_stoforce(ModuleBase::matrix& force, + void cal_stoforce(const int nspin, + const bool domag, + const bool domag_z, + const int gga_grad, + ModuleBase::matrix& force, const elecstate::ElecState& elec, ModulePW::PW_Basis* rho_basis, ModuleSymmetry::Symmetry* p_symm, diff --git a/source/source_pw/module_stodft/sto_hamilt_pw.cpp b/source/source_pw/module_stodft/sto_hamilt_pw.cpp new file mode 100644 index 00000000000..2bb5040a361 --- /dev/null +++ b/source/source_pw/module_stodft/sto_hamilt_pw.cpp @@ -0,0 +1,67 @@ +#include "sto_hamilt_pw.h" +#include "source_base/timer.h" +#include "kernels/hpsi_norm_op.h" + +template +StoHamiltPW::StoHamiltPW(elecstate::Potential* pot_in, + ModulePW::PW_Basis_K* wfc_basis, + K_Vectors* p_kv, + pseudopot_cell_vnl* nlpp, + const UnitCell* ucell, + const int& npol, + Real* emin_in, + Real* emax_in) + : hamilt::HamiltPW(pot_in, wfc_basis, p_kv, nlpp, nullptr, ucell, nullptr), ngk(p_kv->ngk) +{ + this->classname = "StoHamiltPW"; + this->npwk_max = wfc_basis->npwk_max; + this->npol = npol; + this->emin = emin_in; + this->emax = emax_in; +} + +template +void StoHamiltPW::hPsi(const T* psi_in, T* hpsi, const int& nbands) +{ + auto call_act = [&, this](const hamilt::Operator* op, const bool& is_first_node) -> void { + op->act(nbands, this->npwk_max, this->npol, psi_in, hpsi, this->ngk[op->get_ik()], is_first_node); + }; + + ModuleBase::timer::start("StoHamiltPW", "hPsi"); + call_act(this->ops, true); // first node + hamilt::Operator* node((hamilt::Operator*)this->ops->next_op); + while (node != nullptr) + { + call_act(node, false); // other nodes + node = (hamilt::Operator*)(node->next_op); + } + ModuleBase::timer::end("StoHamiltPW", "hPsi"); + + return; +} + +template +void StoHamiltPW::hPsi_norm(const T* psi_in, T* hpsi_norm, const int& nbands) +{ + ModuleBase::timer::start("StoHamiltPW", "hPsi_norm"); + + this->hPsi(psi_in, hpsi_norm, nbands); + + const int ik = this->ops->get_ik(); + const int npwk_max = this->npwk_max; + const int npwk = this->ngk[ik]; + const Real emin = *this->emin; + const Real emax = *this->emax; + const Real Ebar = (emin + emax) / 2; + const Real DeltaE = (emax - emin) / 2; + + hamilt::hpsi_norm_op()(this->ctx, nbands, npwk_max, npwk, Ebar, DeltaE, hpsi_norm, psi_in); + ModuleBase::timer::end("StoHamiltPW", "hPsi_norm"); +} + +template class StoHamiltPW, base_device::DEVICE_CPU>; +template class StoHamiltPW, base_device::DEVICE_CPU>; +#if ((defined __CUDA) || (defined __ROCM)) +template class StoHamiltPW, base_device::DEVICE_GPU>; +template class StoHamiltPW, base_device::DEVICE_GPU>; +#endif diff --git a/source/source_pw/module_stodft/sto_hamilt_pw.h b/source/source_pw/module_stodft/sto_hamilt_pw.h new file mode 100644 index 00000000000..31f5716af80 --- /dev/null +++ b/source/source_pw/module_stodft/sto_hamilt_pw.h @@ -0,0 +1,62 @@ +#ifndef STO_HAMILT_PW_H +#define STO_HAMILT_PW_H + +#include "source_pw/module_pwdft/hamilt_pw.h" + +template +class StoHamiltPW : public hamilt::HamiltPW +{ + public: + using Real = typename GetTypeReal::type; + /** + * @brief Construct a new StoHamiltPW object + * + * @param pot_in potential + * @param wfc_basis pw basis for wave functions + * @param p_kv k vectors + * @param npol the length of wave function is npol * npwk_max + * @param emin_in Emin of the Hamiltonian + * @param emax_in Emax of the Hamiltonian + */ + StoHamiltPW(elecstate::Potential* pot_in, + ModulePW::PW_Basis_K* wfc_basis, + K_Vectors* p_kv, + pseudopot_cell_vnl* nlpp, + const UnitCell* ucell, + const int& npol, + Real* emin_in, + Real* emax_in); + /** + * @brief Destroy the StoHamiltPW object + * + */ + ~StoHamiltPW(){}; + + /** + * @brief Calculate \hat{H}|psi> + * + * @param psi_in input wave function + * @param hpsi output wave function + * @param nbands number of bands + */ + void hPsi(const T* psi_in, T* hpsi, const int& nbands = 1); + + /** + * @brief Calculate \hat{H}|psi> and normalize it + * + * @param psi_in input wave function + * @param hpsi output wave function + * @param nbands number of bands + */ + void hPsi_norm(const T* psi_in, T* hpsi, const int& nbands = 1); + + Real* emin = nullptr; ///< Emin of the Hamiltonian + Real* emax = nullptr; ///< Emax of the Hamiltonian + + private: + int npwk_max = 0; ///< maximum number of plane waves + int npol = 0; ///< number of polarizations + std::vector& ngk; ///< number of G vectors +}; + +#endif diff --git a/source/source_pw/module_stodft/sto_hsolver_pw.cpp b/source/source_pw/module_stodft/sto_hsolver_pw.cpp new file mode 100644 index 00000000000..dcb53ec9ddc --- /dev/null +++ b/source/source_pw/module_stodft/sto_hsolver_pw.cpp @@ -0,0 +1,141 @@ +#include "sto_hsolver_pw.h" + +#include "source_base/global_function.h" +#include "source_base/parallel_comm.h" +#include "source_base/parallel_device.h" +#include "source_base/timer.h" +#include "source_base/tool_title.h" +#include "source_estate/module_charge/chg_symm.h" +#include "source_estate/elecstate_tools.h" +#include "source_hamilt/hamilt_hs_adapter.h" + +#include + +template +void StoHSolverPW::solve(const UnitCell& ucell, + hamilt::Hamilt* pHamilt, + psi::Psi& psi, + psi::Psi& psi_cpu, + elecstate::ElecState* pes, + ModulePW::PW_Basis_K* wfc_basis, + Stochastic_WF& stowf, + const int istep, + const int iter, + std::ostream& log, + const bool skip_charge) +{ + ModuleBase::TITLE("StoHSolverPW", "solve"); + ModuleBase::timer::start("StoHSolverPW", "solve"); + + // This override never calls HSolverPW::solve, which is where the base class + // normally establishes the pool communication context. Set it up here so that + // hamiltSolvePsiK builds a diag_comm_info describing the real pool; otherwise it + // would keep the defaults (rank 0, 1 process) and every reduction guarded by + // diag_comm.nproc > 1 would be silently skipped. + this->rank_in_pool = wfc_basis->poolrank; + this->nproc_in_pool = wfc_basis->poolnproc; + + const int npwx = psi.get_nbasis(); + const int nbands = psi.get_nbands(); + const int nks = psi.get_nk(); + + // prepare for the precondition of diagonalization + std::vector precondition(psi.get_nbasis(), 0.0); + + this->ethr_band.resize(psi.get_nbands(), this->diag_thr); + + // report if the specified diagonalization method is not supported + const std::initializer_list _methods = {"cg", "dav", "dav_subspace", "bpcg"}; + if (std::find(std::begin(_methods), std::end(_methods), this->method) == std::end(_methods)) + { + ModuleBase::WARNING_QUIT("HSolverPW::solve", "This type of eigensolver is not supported!"); + } + + // the iterative eigensolvers see the Hamiltonian only through this operator + hamilt::HamiltHSOperator op(pHamilt, wfc_basis); + + // part of KSDFT to get KS orbitals + for (int ik = 0; ik < nks; ++ik) + { + ModuleBase::timer::start("StoHSolverPW", "solve_KS"); + op.update_k(ik); + if (nbands > 0 && this->ks_run) + { + /// update psi pointer for each k point + psi.fix_k(ik); + /// template add precondition calculating here + this->update_precondition(precondition, ik, this->wfc_basis->npwk[ik], pes->pot->get_vl_of_0()); + /// solve eigenvector and eigenvalue for H(k) + double* p_eigenvalues = &(pes->ekb(ik, 0)); + this->hamiltSolvePsiK(op, psi, precondition, p_eigenvalues, nks); + } + +#ifdef __MPI + if (nbands > 0 && !this->all_ks_run) + { + Parallel_Common::bcast_dev(&psi(ik, 0, 0), npwx * nbands, BP_WORLD, 0, &psi_cpu(ik, 0, 0)); + MPI_Bcast(&pes->ekb(ik, 0), nbands, MPI_DOUBLE, 0, BP_WORLD); + } +#endif + ModuleBase::timer::end("StoHSolverPW", "solve_KS"); + stoiter.orthog(ik, psi, stowf); + stoiter.checkemm(ik, istep, iter, stowf); // check and reset emax & emin + } + + this->output_iterInfo(log); + + for (int ik = 0; ik < nks; ik++) + { + // init k + if (nks > 1) + { + pHamilt->updateHk(ik); // necessary , because emax and emin should be decided first + } + stoiter.calPn(ik, stowf); + } + + // iterate to get mu + stoiter.itermu(iter, pes); + + // prepare sqrt{f(\hat{H})}|\chi> to calculate density, force and stress + stoiter.calHsqrtchi(stowf); + + // calculate eband = \sum_{ik,ib} w(ik)f(ik,ib)e_{ikib}, demet = -TS + elecstate::ElecStatePW* pes_pw = static_cast*>(pes); + elecstate::calEBand(pes_pw->ekb,pes_pw->wg,pes_pw->f_en); + if(!this->all_ks_run) + { + pes->f_en.eband /= this->bndpar; + } + stoiter.sum_stoeband(stowf, pes_pw, pHamilt, wfc_basis); + + + + // for nscf, skip charge + if (skip_charge) + { + ModuleBase::timer::end("StoHSolverPW", "solve"); + return; + } + + //(5) calculate new charge density + // calculate KS rho. + pes_pw->init_rho_data(); + if (nbands > 0) + { + pes_pw->psiToRho(psi); + } + // calculate stochastic rho + stoiter.cal_storho(ucell, stowf, pes_pw,wfc_basis); + + // will do rho symmetry and energy calculation in esolver + ModuleBase::timer::end("StoHSolverPW", "solve"); + return; +} + +// template class StoHSolverPW, base_device::DEVICE_CPU>; +template class StoHSolverPW, base_device::DEVICE_CPU>; +#if ((defined __CUDA) || (defined __ROCM)) +// template class StoHSolverPW, base_device::DEVICE_GPU>; +template class StoHSolverPW, base_device::DEVICE_GPU>; +#endif diff --git a/source/source_pw/module_stodft/sto_hsolver_pw.h b/source/source_pw/module_stodft/sto_hsolver_pw.h new file mode 100644 index 00000000000..8c9b8cef9c2 --- /dev/null +++ b/source/source_pw/module_stodft/sto_hsolver_pw.h @@ -0,0 +1,81 @@ +#ifndef STO_HSOLVER_PW_H +#define STO_HSOLVER_PW_H +#include "source_hsolver/hsolver_pw.h" +#include "source_pw/module_stodft/sto_hamilt_pw.h" +#include "source_pw/module_stodft/sto_iter.h" +template +class StoHSolverPW : public hsolver::HSolverPW +{ + protected: + using Real = typename GetTypeReal::type; + + public: + StoHSolverPW(K_Vectors* pkv, + ModulePW::PW_Basis_K* wfc_basis_in, + Stochastic_WF& stowf, + StoChe& stoche, + StoHamiltPW* p_hamilt_sto, + const std::string calculation_type_in, + const std::string basis_type_in, + const std::string method_in, + const bool use_uspp_in, + const int nspin_in, + const int scf_iter_in, + const int diag_iter_max_in, + const double diag_thr_in, + const bool need_subspace_in, + const int nbands_in, + const bool diago_smooth_ethr_in, + const int pw_diag_ndim_in, + const int diag_subspace_in, + const int nb2d_in, + const bool ks_run_in, + const bool all_ks_run_in, + const int bndpar_in) + : hsolver::HSolverPW(wfc_basis_in, + calculation_type_in, + basis_type_in, + method_in, + use_uspp_in, + nspin_in, + scf_iter_in, + diag_iter_max_in, + diag_thr_in, + need_subspace_in, + nbands_in, + diago_smooth_ethr_in, + pw_diag_ndim_in, + diag_subspace_in, + nb2d_in), + ks_run(ks_run_in), all_ks_run(all_ks_run_in), bndpar(bndpar_in) + { + stoiter.init(pkv, wfc_basis_in, stowf, stoche, p_hamilt_sto); + } + + void solve(const UnitCell& ucell, + hamilt::Hamilt* pHamilt, + psi::Psi& psi, + psi::Psi& psi_cpu, + elecstate::ElecState* pes, + ModulePW::PW_Basis_K* wfc_basis, + Stochastic_WF& stowf, + const int istep, + const int iter, + std::ostream& log, + const bool skip_charge); + + Stochastic_Iter stoiter; + + protected: + const bool ks_run; // true if the current process runs the KS part of the SDFT calculation + const bool all_ks_run; // true if every process runs the KS part + const int bndpar; // number of band-parallel groups + + using setmem_complex_op = base_device::memory::set_memory_op; + using setmem_var_op = base_device::memory::set_memory_op; + using syncmem_h2d_op = base_device::memory::synchronize_memory_op; + using syncmem_d2h_op = base_device::memory::synchronize_memory_op; + using syncmem_var_h2d_op = base_device::memory::synchronize_memory_op; + using syncmem_var_d2h_op = base_device::memory::synchronize_memory_op; +}; +#endif diff --git a/source/source_pw/module_stodft/sto_iter.cpp b/source/source_pw/module_stodft/sto_iter.cpp index fc1f28d6e4b..6a121064866 100644 --- a/source/source_pw/module_stodft/sto_iter.cpp +++ b/source/source_pw/module_stodft/sto_iter.cpp @@ -8,6 +8,7 @@ #include "source_base/tool_quit.h" #include "source_base/tool_title.h" #include "source_estate/kernels/elecstate_op.h" +#include "source_estate/module_charge/chg_parallel.h" #include "source_estate/occupy.h" #include "source_hsolver/para_lin_tf.h" #include "source_io/module_parameter/parameter.h" @@ -40,7 +41,7 @@ void Stochastic_Iter::init(K_Vectors* pkv_in, ModulePW::PW_Basis_K* wfc_basis, Stochastic_WF& stowf, StoChe& stoche, - hamilt::HamiltSdftPW* p_hamilt_sto) + StoHamiltPW* p_hamilt_sto) { p_che = stoche.p_che.get(); spolyv = stoche.spolyv.get(); @@ -185,7 +186,7 @@ void Stochastic_Iter::checkemm(const int& ik, while (true) { bool converge; - auto hchi_norm = std::bind(&hamilt::HamiltSdftPW::hPsi_norm, + auto hchi_norm = std::bind(&StoHamiltPW::hPsi_norm, p_hamilt_sto, std::placeholders::_1, std::placeholders::_2, @@ -399,7 +400,7 @@ void Stochastic_Iter::calPn(const int& ik, Stochastic_WF& pchi = stowf.chi0->get_pointer(); } - auto hchi_norm = std::bind(&hamilt::HamiltSdftPW::hPsi_norm, + auto hchi_norm = std::bind(&StoHamiltPW::hPsi_norm, p_hamilt_sto, std::placeholders::_1, std::placeholders::_2, @@ -672,7 +673,11 @@ void Stochastic_Iter::cal_storho(const UnitCell& ucell, { for (int is = 0; is < nspin; ++is) { - pes->charge->reduce_diff_pools(sto_rho[is]); + module_charge::reduce_diff_pools(sto_rho[is], + *pes->charge, + GlobalV::KPAR, + PARAM.globalv.all_ks_run, + PARAM.inp.bndpar); if (!PARAM.globalv.all_ks_run && PARAM.inp.bndpar > 1) { MPI_Allreduce(MPI_IN_PLACE, sto_rho[is], nrxx, MPI_DOUBLE, MPI_SUM, BP_WORLD); @@ -779,7 +784,7 @@ void Stochastic_Iter::calTnchi_ik(const int& ik, Stochastic_WFupdateHk(ik); // necessary, because itermu should be called before this function } - auto hchi_norm = std::bind(&hamilt::HamiltSdftPW::hPsi_norm, + auto hchi_norm = std::bind(&StoHamiltPW::hPsi_norm, p_hamilt_sto, std::placeholders::_1, std::placeholders::_2, diff --git a/source/source_pw/module_stodft/sto_iter.h b/source/source_pw/module_stodft/sto_iter.h index 97df2639310..60c3a7a56af 100644 --- a/source/source_pw/module_stodft/sto_iter.h +++ b/source/source_pw/module_stodft/sto_iter.h @@ -3,7 +3,7 @@ #include "source_base/math_chebyshev.h" #include "source_estate/elecstate_pw.h" #include "source_hamilt/hamilt.h" -#include "source_pw/module_stodft/hamilt_sdft_pw.h" +#include "source_pw/module_stodft/sto_hamilt_pw.h" #include "source_psi/psi.h" #include "sto_che.h" #include "sto_func.h" @@ -42,7 +42,7 @@ class Stochastic_Iter ModulePW::PW_Basis_K* wfc_basis, Stochastic_WF& stowf, StoChe& stoche, - hamilt::HamiltSdftPW* p_hamilt_sto); + StoHamiltPW* p_hamilt_sto); /** * @brief sum demet and eband energies for each k point and each band @@ -117,7 +117,7 @@ class Stochastic_Iter ModuleBase::Chebyshev* p_che = nullptr; Sto_Func stofunc; - hamilt::HamiltSdftPW* p_hamilt_sto = nullptr; + StoHamiltPW* p_hamilt_sto = nullptr; double mu0 = 0.0; // chemical potential; unit in Ry bool change = false; diff --git a/source/source_pw/module_stodft/sto_stress_pw.cpp b/source/source_pw/module_stodft/sto_stress_pw.cpp index b85ab58b503..68a42d67006 100644 --- a/source/source_pw/module_stodft/sto_stress_pw.cpp +++ b/source/source_pw/module_stodft/sto_stress_pw.cpp @@ -9,7 +9,12 @@ #include "source_io/module_parameter/parameter.h" template -void Sto_Stress_PW::cal_stress(ModuleBase::matrix& sigmatot, +void Sto_Stress_PW::cal_stress(const int nspin, + const bool domag, + const bool domag_z, + const int gga_grad, + const bool gamma_only_pw, + ModuleBase::matrix& sigmatot, const elecstate::ElecState& elec, ModulePW::PW_Basis* rho_basis, ModuleSymmetry::Symmetry* p_symm, @@ -50,13 +55,15 @@ void Sto_Stress_PW::cal_stress(ModuleBase::matrix& sigmatot, { sigmaxc(i, i) = -(elec.f_en.etxc - elec.f_en.vtxc) / this->ucell->omega; } - this->stress_gga(ucell_in, sigmaxc, rho_basis, chr); + this->stress_gga(ucell_in, sigmaxc, rho_basis, chr, + nspin, domag, domag_z, gga_grad); // local contribution this->stress_loc(ucell_in, sigmaloc, rho_basis, locpp->vloc, p_sf, true, chr); // nlcc - this->stress_cc(sigmaxcc, rho_basis, ucell_in, p_sf, true, locpp->numeric, chr); + this->stress_cc(sigmaxcc, rho_basis, ucell_in, p_sf, true, locpp->numeric, chr, + nspin, domag, domag_z, gga_grad, gamma_only_pw); // nonlocal this->sto_stress_nl(sigmanl, wg, p_sf, p_symm, p_kv, wfc_basis, *nlpp, ucell_in, psi_in, stowf); diff --git a/source/source_pw/module_stodft/sto_stress_pw.h b/source/source_pw/module_stodft/sto_stress_pw.h index c357cb1fb6f..cb5f41d2926 100644 --- a/source/source_pw/module_stodft/sto_stress_pw.h +++ b/source/source_pw/module_stodft/sto_stress_pw.h @@ -17,7 +17,12 @@ class Sto_Stress_PW : public Stress_Func ~Sto_Stress_PW(){}; // calculate the stress in PW basis - void cal_stress(ModuleBase::matrix& sigmatot, + void cal_stress(const int nspin, + const bool domag, + const bool domag_z, + const int gga_grad, + const bool gamma_only_pw, + ModuleBase::matrix& sigmatot, const elecstate::ElecState& elec, ModulePW::PW_Basis* rho_basis, ModuleSymmetry::Symmetry* p_symm, diff --git a/source/source_pw/module_stodft/sto_tool.cpp b/source/source_pw/module_stodft/sto_tool.cpp index 4c432a687c6..f63a8707236 100644 --- a/source/source_pw/module_stodft/sto_tool.cpp +++ b/source/source_pw/module_stodft/sto_tool.cpp @@ -18,7 +18,7 @@ void check_che_op::operator()(const int& nche_in, const int& nbands_sto, K_Vectors* p_kv, Stochastic_WF, Device>* p_stowf, - hamilt::HamiltSdftPW, Device>* p_hamilt_sto) + StoHamiltPW, Device>* p_hamilt_sto) { //------------------------------ // Convergence test @@ -78,7 +78,7 @@ void check_che_op::operator()(const int& nche_in, while (true) { bool converge; - auto hchi_norm = std::bind(&hamilt::HamiltSdftPW, Device>::hPsi_norm, + auto hchi_norm = std::bind(&StoHamiltPW, Device>::hPsi_norm, p_hamilt_sto, std::placeholders::_1, std::placeholders::_2, diff --git a/source/source_pw/module_stodft/sto_tool.h b/source/source_pw/module_stodft/sto_tool.h index a2eaf2a00bb..e2a04f07da9 100644 --- a/source/source_pw/module_stodft/sto_tool.h +++ b/source/source_pw/module_stodft/sto_tool.h @@ -1,7 +1,7 @@ #ifndef STO_TOOL_H #define STO_TOOL_H #include "source_cell/klist.h" -#include "source_pw/module_stodft/hamilt_sdft_pw.h" +#include "source_pw/module_stodft/sto_hamilt_pw.h" #include "source_pw/module_stodft/sto_wf.h" #include "source_base/module_device/memory_op.h" #include "source_psi/psi.h" @@ -24,7 +24,7 @@ struct check_che_op const int& nbands_sto, K_Vectors* p_kv, Stochastic_WF, Device>* p_stowf, - hamilt::HamiltSdftPW, Device>* p_hamilt_sto); + StoHamiltPW, Device>* p_hamilt_sto); }; /** diff --git a/source/source_pw/module_stodft/test/CMakeLists.txt b/source/source_pw/module_stodft/test/CMakeLists.txt index 836a1ac7dcd..3e933f7d1cd 100644 --- a/source/source_pw/module_stodft/test/CMakeLists.txt +++ b/source/source_pw/module_stodft/test/CMakeLists.txt @@ -1,4 +1,8 @@ abacus_disable_feature_definitions(__MPI) +# These tests exercise CPU code only and link no GPU kernels, so build them +# without the device instantiations (as source_hsolver/test does). +abacus_disable_feature_definitions(__CUDA) +abacus_disable_feature_definitions(__ROCM) AddTest( TARGET MODULE_PW_Sto_Tool_UTs @@ -9,6 +13,20 @@ AddTest( AddTest( TARGET MODULE_PW_Sto_Hamilt_UTs LIBS parameter psi base device planewave_serial symmetry - SOURCES ../hamilt_sdft_pw.cpp test_hamilt_sto.cpp ../../../source_hamilt/operator.cpp + SOURCES ../sto_hamilt_pw.cpp test_sto_hamilt_pw.cpp ../../../source_hamilt/operator.cpp ../../../source_cell/klist.cpp ../../../source_cell/klist_io.cpp ../../../source_cell/parallel_kpoints.cpp ../../../source_cell/reciprocal_grid.cpp +) + +AddTest( + TARGET MODULE_PW_Sto_HSolver_UTs + LIBS parameter psi device base container MPI::MPI_CXX + SOURCES test_sto_hsolver_pw.cpp ../sto_hsolver_pw.cpp + ../../../source_hsolver/hsolver_pw.cpp ../../../source_hsolver/diago_bpcg.cpp + ../../../source_hsolver/diago_dav_subspace.cpp ../../../source_hsolver/diag_const_nums.cpp + ../../../source_hsolver/diago_iter_assist.cpp ../../../source_hsolver/para_lin_tf.cpp + ../../../source_estate/elecstate_tools.cpp ../../../source_estate/occupy.cpp + ../../../source_base/module_fft/fft_bundle.cpp ../../../source_base/module_fft/fft_cpu.cpp + # This test calls MPI_Init in main() and its mocks take MPI_Comm arguments + # unconditionally, so it must keep __MPI even though this directory disables it. + KEEP_FEATURE_DEFINITIONS __MPI ) \ No newline at end of file diff --git a/source/source_pw/module_stodft/test/test_hamilt_sto.cpp b/source/source_pw/module_stodft/test/test_hamilt_sto.cpp deleted file mode 100644 index b749cc67b47..00000000000 --- a/source/source_pw/module_stodft/test/test_hamilt_sto.cpp +++ /dev/null @@ -1,151 +0,0 @@ -#include "../hamilt_sdft_pw.h" -#include "source_pw/module_pwdft/dftu_base.h" -#include "source_hamilt/operator.h" - -#include "gtest/gtest.h" -#include - -elecstate::Potential::~Potential(){} -void elecstate::Potential::cal_v_eff(Charge const*, UnitCell const*, ModuleBase::matrix&){} -void elecstate::Potential::cal_fixed_v(double*){} - -template -hamilt::HamiltPW::HamiltPW( - elecstate::Potential* pot_in, - ModulePW::PW_Basis_K* wfc_basis, - K_Vectors* p_kv, - pseudopot_cell_vnl* ppcell, - Plus_U_Base* p_dftu, - const UnitCell* ucell, - const General_Exx_Info* exx_info){} - -template -hamilt::HamiltPW::~HamiltPW(){ - delete this->ops; -}; -template -void hamilt::HamiltPW::updateHk(int){} -template -void hamilt::HamiltPW::sPsi(T const*, T*, const int, const int, const int) const{} - -template class hamilt::HamiltPW, base_device::DEVICE_CPU>; -template class hamilt::HamiltPW, base_device::DEVICE_CPU>; -template class hamilt::HamiltPW, base_device::DEVICE_GPU>; -template class hamilt::HamiltPW, base_device::DEVICE_GPU>; - -/************************************************ - * unit test of hamilt_sto_pw.cpp - * - Tested Functions: - * - void hPsi(const T* psi_in, T* hpsi, const int& nbands) - * - void hPsi_norm(const T* psi_in, T* hpsi, const int& nbands) - ***********************************************/ - -template -class TestOp : public hamilt::Operator -{ - public: - virtual void act(const int nbands, - const int nbasis, - const int npol, - const T* tmpsi_in, - T* tmhpsi, - const int ngk_ik = 0, - const bool is_first_node = false) const override - { - for (int i = 0; i < nbands; i++) - { - for (int j = 0; j < nbasis; j++) - { - tmhpsi[i * nbasis + j] = tmpsi_in[i * nbasis + j]; - } - } - } -}; - -class TestHamiltSto : public ::testing::Test -{ - public: - TestHamiltSto() - { - const int nbands = 1; - const int nbasis = 2; - const int npol = 1; - // Initialize the hamilt_sto_pw - pot = new elecstate::Potential(); - wfc_basis = new ModulePW::PW_Basis_K(); - wfc_basis->npwk_max = 2; - p_kv = new K_Vectors(); - std::vector ngk = {2}; - p_kv->ngk = ngk; - hamilt_sto = new hamilt::HamiltSdftPW, base_device::DEVICE_CPU>(pot, wfc_basis, p_kv, nullptr, nullptr, npol, &emin, &emax); - hamilt_sto->ops = new TestOp, base_device::DEVICE_CPU>(); - } - - ~TestHamiltSto() - { - delete pot; - delete wfc_basis; - delete p_kv; - delete hamilt_sto; - } - - elecstate::Potential* pot; - ModulePW::PW_Basis_K* wfc_basis; - K_Vectors* p_kv; - hamilt::HamiltSdftPW, base_device::DEVICE_CPU>* hamilt_sto; - double emin = -2.0; - double emax = 2.0; -}; - -TEST_F(TestHamiltSto, hPsi) -{ - const int nbands = 1; - const int nbasis = 2; - // Prepare the input psi - std::vector> psi_in(nbands * nbasis); - std::vector> hpsi(nbands * nbasis); - - for (int i = 0; i < nbands; i++) - { - for (int j = 0; j < nbasis; j++) - { - psi_in[i * nbasis + j] = i + j; - } - } - hamilt_sto->hPsi(psi_in.data(), hpsi.data(), nbands); - // Check the result - for (int i = 0; i < nbands; i++) - { - for (int j = 0; j < nbasis; j++) - { - EXPECT_EQ(hpsi[i * nbasis + j], psi_in[i * nbasis + j]); - } - } -} - -TEST_F(TestHamiltSto, hPsi_norm) -{ - int nbands = 1; - int nbasis = 2; - std::vector> psi_in(nbands * nbasis); - std::vector> hpsi(nbands * nbasis); - - for (int i = 0; i < nbands; i++) - { - for (int j = 0; j < nbasis; j++) - { - psi_in[i * nbasis + j] = i + j; - } - } - - hamilt_sto->hPsi_norm(psi_in.data(), hpsi.data(), nbands); - - // Check the result - for (int i = 0; i < nbands; i++) - { - for (int j = 0; j < nbasis; j++) - { - EXPECT_EQ(hpsi[i * nbasis + j], psi_in[i * nbasis + j] * 0.5); - } - } -} diff --git a/source/source_pw/module_stodft/test/test_sto_hamilt_pw.cpp b/source/source_pw/module_stodft/test/test_sto_hamilt_pw.cpp new file mode 100644 index 00000000000..7023d90719b --- /dev/null +++ b/source/source_pw/module_stodft/test/test_sto_hamilt_pw.cpp @@ -0,0 +1,151 @@ +#include "../sto_hamilt_pw.h" +#include "source_pw/module_pwdft/dftu_base.h" +#include "source_hamilt/operator.h" + +#include "gtest/gtest.h" +#include + +elecstate::Potential::~Potential(){} +void elecstate::Potential::cal_v_eff(Charge const*, UnitCell const*, ModuleBase::matrix&){} +void elecstate::Potential::cal_fixed_v(double*){} + +template +hamilt::HamiltPW::HamiltPW( + elecstate::Potential* pot_in, + ModulePW::PW_Basis_K* wfc_basis, + K_Vectors* p_kv, + pseudopot_cell_vnl* ppcell, + Plus_U_Base* p_dftu, + const UnitCell* ucell, + const General_Exx_Info* exx_info){} + +template +hamilt::HamiltPW::~HamiltPW(){ + delete this->ops; +}; +template +void hamilt::HamiltPW::updateHk(int){} +template +void hamilt::HamiltPW::sPsi(T const*, T*, const int, const int, const int) const{} + +template class hamilt::HamiltPW, base_device::DEVICE_CPU>; +template class hamilt::HamiltPW, base_device::DEVICE_CPU>; +template class hamilt::HamiltPW, base_device::DEVICE_GPU>; +template class hamilt::HamiltPW, base_device::DEVICE_GPU>; + +/************************************************ + * unit test of hamilt_sto_pw.cpp + * - Tested Functions: + * - void hPsi(const T* psi_in, T* hpsi, const int& nbands) + * - void hPsi_norm(const T* psi_in, T* hpsi, const int& nbands) + ***********************************************/ + +template +class TestOp : public hamilt::Operator +{ + public: + virtual void act(const int nbands, + const int nbasis, + const int npol, + const T* tmpsi_in, + T* tmhpsi, + const int ngk_ik = 0, + const bool is_first_node = false) const override + { + for (int i = 0; i < nbands; i++) + { + for (int j = 0; j < nbasis; j++) + { + tmhpsi[i * nbasis + j] = tmpsi_in[i * nbasis + j]; + } + } + } +}; + +class TestHamiltSto : public ::testing::Test +{ + public: + TestHamiltSto() + { + const int nbands = 1; + const int nbasis = 2; + const int npol = 1; + // Initialize the hamilt_sto_pw + pot = new elecstate::Potential(); + wfc_basis = new ModulePW::PW_Basis_K(); + wfc_basis->npwk_max = 2; + p_kv = new K_Vectors(); + std::vector ngk = {2}; + p_kv->ngk = ngk; + hamilt_sto = new StoHamiltPW, base_device::DEVICE_CPU>(pot, wfc_basis, p_kv, nullptr, nullptr, npol, &emin, &emax); + hamilt_sto->ops = new TestOp, base_device::DEVICE_CPU>(); + } + + ~TestHamiltSto() + { + delete pot; + delete wfc_basis; + delete p_kv; + delete hamilt_sto; + } + + elecstate::Potential* pot; + ModulePW::PW_Basis_K* wfc_basis; + K_Vectors* p_kv; + StoHamiltPW, base_device::DEVICE_CPU>* hamilt_sto; + double emin = -2.0; + double emax = 2.0; +}; + +TEST_F(TestHamiltSto, hPsi) +{ + const int nbands = 1; + const int nbasis = 2; + // Prepare the input psi + std::vector> psi_in(nbands * nbasis); + std::vector> hpsi(nbands * nbasis); + + for (int i = 0; i < nbands; i++) + { + for (int j = 0; j < nbasis; j++) + { + psi_in[i * nbasis + j] = i + j; + } + } + hamilt_sto->hPsi(psi_in.data(), hpsi.data(), nbands); + // Check the result + for (int i = 0; i < nbands; i++) + { + for (int j = 0; j < nbasis; j++) + { + EXPECT_EQ(hpsi[i * nbasis + j], psi_in[i * nbasis + j]); + } + } +} + +TEST_F(TestHamiltSto, hPsi_norm) +{ + int nbands = 1; + int nbasis = 2; + std::vector> psi_in(nbands * nbasis); + std::vector> hpsi(nbands * nbasis); + + for (int i = 0; i < nbands; i++) + { + for (int j = 0; j < nbasis; j++) + { + psi_in[i * nbasis + j] = i + j; + } + } + + hamilt_sto->hPsi_norm(psi_in.data(), hpsi.data(), nbands); + + // Check the result + for (int i = 0; i < nbands; i++) + { + for (int j = 0; j < nbasis; j++) + { + EXPECT_EQ(hpsi[i * nbasis + j], psi_in[i * nbasis + j] * 0.5); + } + } +} diff --git a/source/source_pw/module_stodft/test/test_sto_hsolver_pw.cpp b/source/source_pw/module_stodft/test/test_sto_hsolver_pw.cpp new file mode 100644 index 00000000000..22292183184 --- /dev/null +++ b/source/source_pw/module_stodft/test/test_sto_hsolver_pw.cpp @@ -0,0 +1,425 @@ +#include +#include +#include + +#include "source_hsolver/test/hsolver_pw_sup.h" +#include "source_hsolver/test/hsolver_supplementary_mock.h" +#include "source_base/parallel_comm.h" +#include "source_estate/elecstate_pw.h" +#include "source_hsolver/hsolver_pw.h" +#include "source_pw/module_stodft/sto_hsolver_pw.h" +#include "source_pw/module_stodft/sto_wf.h" + +// mock for module_sdft +template +Sto_Func::Sto_Func() +{ +} +template class Sto_Func; + +// mock of Stochastic_WF +template +Stochastic_WF::Stochastic_WF() +{ + chiortho = nullptr; + chi0 = nullptr; + shchi = nullptr; + nchip = nullptr; +} + +template +Stochastic_WF::~Stochastic_WF() +{ + delete[] chi0; + delete[] shchi; + delete[] chiortho; + delete[] nchip; +} + +template +void Stochastic_WF::init(K_Vectors* p_kv, const int npwx_in) +{ +} + +// mock diago_hs_para +namespace hsolver { +template +void diago_hs_para(T* h, + T* s, + const int lda, + const int nband, + typename GetTypeReal::type* const ekb, + T* const wfc, + const MPI_Comm& comm, + const int diag_subspace, + const int block_size = 0) +{} +template void diago_hs_para(double* h, + double* s, + const int lda, + const int nband, + typename GetTypeReal::type* const ekb, + double* const wfc, + const MPI_Comm& comm, + const int diag_subspace, + const int block_size); +template void diago_hs_para>(std::complex* h, + std::complex* s, + const int lda, + const int nband, + typename GetTypeReal>::type* const ekb, + std::complex* const wfc, + const MPI_Comm& comm, + const int diag_subspace, + const int block_size); +template void diago_hs_para(float* h, + float* s, + const int lda, + const int nband, + typename GetTypeReal::type* const ekb, + float* const wfc, + const MPI_Comm& comm, + const int diag_subspace, + const int block_size); +template void diago_hs_para>(std::complex* h, + std::complex* s, + const int lda, + const int nband, + typename GetTypeReal>::type* const ekb, + std::complex* const wfc, + const MPI_Comm& comm, + const int diag_subspace, + const int block_size); + +} + + +template<> +void elecstate::ElecStatePW, base_device::DEVICE_CPU>::init_rho_data() +{ +} + +template class StoChe; + +template +Stochastic_Iter::Stochastic_Iter() +{ + change = false; + mu0 = 0; + method = 2; +} + +template +Stochastic_Iter::~Stochastic_Iter(){}; + +template +void Stochastic_Iter::init(K_Vectors* pkv_in, + ModulePW::PW_Basis_K* wfc_basis, + Stochastic_WF& stowf, + StoChe& stoche, + StoHamiltPW* p_hamilt_sto) +{ + this->nchip = stowf.nchip; + ; + this->targetne = 1; + this->method = stoche.method_sto; +} + +template +void Stochastic_Iter::orthog(const int& ik, psi::Psi& psi, Stochastic_WF& stowf) +{ + // do something to verify this function has been called + for (int i = 0; i < psi.size(); i++) + { + psi.get_pointer()[i] += 1.1; + } + return; +} + +template +void Stochastic_Iter::checkemm(const int& ik, + const int istep, + const int iter, + Stochastic_WF& stowf) +{ + // do something to verify this function has been called + stowf.nchi++; + return; +} + +template +void Stochastic_Iter::calPn(const int& ik, Stochastic_WF& stowf) +{ + // do something to verify this function has been called + stowf.nbands_diag++; + return; +} + +template +void Stochastic_Iter::itermu(int iter, elecstate::ElecState* pes) +{ + // do something to verify this function has been called + pes->f_en.eband += 1.2; + return; +} + +template +void Stochastic_Iter::calHsqrtchi(Stochastic_WF& stowf) +{ + // do something to verify this function has been called + stowf.nchip_max++; + return; +} + +template +void Stochastic_Iter::sum_stoeband(Stochastic_WF& stowf, + elecstate::ElecStatePW* pes, + hamilt::Hamilt* pHamilt, + ModulePW::PW_Basis_K* wfc_basis) +{ + // do something to verify this function has been called + stowf.nbands_total++; + return; +} + +template +void Stochastic_Iter::cal_storho(const UnitCell& ucell, + Stochastic_WF& stowf, + elecstate::ElecStatePW* pes, + ModulePW::PW_Basis_K* wfc_basis) +{ +} +template class Stochastic_Iter, base_device::DEVICE_CPU>; + +Charge::Charge(){}; +Charge::~Charge(){}; + +// Mock implementations for the template functions causing linking errors +namespace ModulePW { + // Mock implementation for recip_to_real + template + void PW_Basis_K::recip_to_real(const Device* ctx, + const std::complex* in, + std::complex* out, + const int ik, + const bool add, + const FPTYPE factor) const + { + // Simple mock implementation that does nothing + // In a real test, you might want to implement behavior that simulates the actual function + } + + // Mock implementation for real_to_recip + template + void PW_Basis_K::real_to_recip(const Device* ctx, + const std::complex* in, + std::complex* out, + const int ik, + const bool add, + const FPTYPE factor) const + { + // Simple mock implementation that does nothing + } + + // Explicit template instantiations + template void PW_Basis_K::recip_to_real( + const base_device::DEVICE_CPU* ctx, + const std::complex* in, + std::complex* out, + const int ik, + const bool add, + const float factor) const; + + template void PW_Basis_K::recip_to_real( + const base_device::DEVICE_CPU* ctx, + const std::complex* in, + std::complex* out, + const int ik, + const bool add, + const double factor) const; + + template void PW_Basis_K::real_to_recip( + const base_device::DEVICE_CPU* ctx, + const std::complex* in, + std::complex* out, + const int ik, + const bool add, + const float factor) const; + + template void PW_Basis_K::real_to_recip( + const base_device::DEVICE_CPU* ctx, + const std::complex* in, + std::complex* out, + const int ik, + const bool add, + const double factor) const; +} +/************************************************ + * unit test of StoHSolverPW class + ***********************************************/ + +/** + * Tested function: + * - 1. solve() + * - with psi; + * - without psi; + * - skip charge; + * - 2. StoHSolverPW::diagethr (for cases below) + * - set_diagethr, for setting diagethr; + */ +class TestStoHSolverPW : public ::testing::Test +{ + public: + TestStoHSolverPW() : elecstate_test(nullptr, nullptr, nullptr, nullptr, nullptr, nullptr, nullptr) + { + stoche.nche = 8; + stoche.method_sto = 1; + } + ModulePW::PW_Basis_K pwbk; + Stochastic_WF> stowf; + K_Vectors kv; + StoChe stoche; + StoHamiltPW>* p_hamilt_sto = nullptr; + StoHSolverPW, base_device::DEVICE_CPU> hs_d + = StoHSolverPW, base_device::DEVICE_CPU>( + &kv, + &pwbk, + stowf, + stoche, + p_hamilt_sto, + "scf", + "pw", + "cg", + false, + 1, + hsolver::DiagoIterAssist>::SCF_ITER, + hsolver::DiagoIterAssist>::PW_DIAG_NMAX, + hsolver::DiagoIterAssist>::PW_DIAG_THR, + hsolver::DiagoIterAssist>::need_subspace, + 0, + false, + 4, + 0, + 0, + false, + true, + 1); + + hamilt::Hamilt> hamilt_test_d; + + psi::Psi> psi_test_cd; + psi::Psi> psi_test_no; + + elecstate::ElecStatePW> elecstate_test; + + std::string method_test = "cg"; + + std::ofstream temp_ofs; +}; + +// TEST_F(TestStoHSolverPW, solve) +// { +// // initial memory and data +// elecstate_test.ekb.create(1, 2); +// elecstate_test.pot = new elecstate::Potential; +// elecstate_test.f_en.eband = 0.0; +// stowf.nbands_diag = 0; +// stowf.nbands_total = 0; +// stowf.nchi = 0; +// stowf.nchip_max = 0; +// psi_test_cd.resize(1, 2, 3); +// const double nelec = 1.0; +// int istep = 0; +// int iter = 0; + +// this->hs_d.solve(&hamilt_test_d, psi_test_cd, psi_test_cd, &elecstate_test, &pwbk, stowf, istep, iter, false); +// EXPECT_DOUBLE_EQ(hsolver::DiagoIterAssist>::avg_iter, 0.0); +// EXPECT_DOUBLE_EQ(elecstate_test.ekb.c[0], 4.0); +// EXPECT_DOUBLE_EQ(elecstate_test.ekb.c[1], 7.0); +// for (int i = 0; i < psi_test_cd.size(); i++) +// { +// // std::cout<<__FILE__<<__LINE__<<" "<rho = new double*[1]; +// elecstate_test.charge->rho[0] = new double[10]; +// elecstate_test.charge->nrxx = 10; +// elecstate_test.rho = new double*[1]; +// elecstate_test.rho[0] = new double[10]; +// int istep = 0; +// int iter = 0; + +// this->hs_d.solve(&hamilt_test_d, psi_test_no, psi_test_no, &elecstate_test, &pwbk, stowf, istep, iter, false); +// EXPECT_DOUBLE_EQ(hsolver::DiagoIterAssist>::avg_iter, 0.0); +// EXPECT_EQ(stowf.nbands_diag, 2); +// EXPECT_EQ(stowf.nbands_total, 1); +// EXPECT_EQ(stowf.nchi, 2); +// EXPECT_EQ(stowf.nchip_max, 1); +// EXPECT_DOUBLE_EQ(elecstate_test.f_en.eband, 1.2); +// /*std::cout<<__FILE__<<__LINE__<<" "<hs_d.solve(&hamilt_test_d, psi_test_no, psi_test_no, &elecstate_test, &pwbk, stowf, istep, iter, true); +// EXPECT_EQ(stowf.nbands_diag, 4); +// EXPECT_EQ(stowf.nbands_total, 1); +// EXPECT_EQ(stowf.nchi, 4); +// EXPECT_EQ(stowf.nchip_max, 2); +// EXPECT_DOUBLE_EQ(elecstate_test.f_en.eband, 2.4); + +// delete[] elecstate_test.charge->rho[0]; +// delete[] elecstate_test.charge->rho; +// delete elecstate_test.charge; +// } + +#ifdef __MPI +#include "source_base/parallel_comm.h" +#include "source_base/timer.h" +#include "mpi.h" +int main(int argc, char** argv) +{ + ModuleBase::timer::disable(); + MPI_Init(&argc, &argv); + testing::InitGoogleTest(&argc, argv); + + MPI_Comm_split(MPI_COMM_WORLD, 0, 1, &BP_WORLD); + int result = RUN_ALL_TESTS(); + + MPI_Comm_free(&BP_WORLD); + MPI_Finalize(); + + return result; +} + +#endif diff --git a/source/source_pw/module_stodft/test/test_sto_tool.cpp b/source/source_pw/module_stodft/test/test_sto_tool.cpp index e4f1c42868e..8ba41f57c8c 100644 --- a/source/source_pw/module_stodft/test/test_sto_tool.cpp +++ b/source/source_pw/module_stodft/test/test_sto_tool.cpp @@ -25,31 +25,31 @@ template void hamilt::HamiltPW::sPsi(T const*, T*, const int, const int, const int) const{} template -hamilt::HamiltSdftPW::HamiltSdftPW(elecstate::Potential* pot_in, - ModulePW::PW_Basis_K* wfc_basis, - K_Vectors* p_kv, - pseudopot_cell_vnl* nlpp, - const UnitCell* ucell, - const int& npol, - Real* emin_in, - Real* emax_in) - : HamiltPW(pot_in, wfc_basis, p_kv, nlpp, nullptr, ucell, nullptr), ngk(p_kv->ngk) +StoHamiltPW::StoHamiltPW(elecstate::Potential* pot_in, + ModulePW::PW_Basis_K* wfc_basis, + K_Vectors* p_kv, + pseudopot_cell_vnl* nlpp, + const UnitCell* ucell, + const int& npol, + Real* emin_in, + Real* emax_in) + : hamilt::HamiltPW(pot_in, wfc_basis, p_kv, nlpp, nullptr, ucell, nullptr), ngk(p_kv->ngk) { } template -void hamilt::HamiltSdftPW::hPsi_norm(const T* psi_in, T* hpsi, const int& nbands){} +void StoHamiltPW::hPsi_norm(const T* psi_in, T* hpsi, const int& nbands){} template class hamilt::HamiltPW, base_device::DEVICE_CPU>; -template class hamilt::HamiltSdftPW, base_device::DEVICE_CPU>; +template class StoHamiltPW, base_device::DEVICE_CPU>; template class hamilt::HamiltPW, base_device::DEVICE_CPU>; -template class hamilt::HamiltSdftPW, base_device::DEVICE_CPU>; +template class StoHamiltPW, base_device::DEVICE_CPU>; #if ((defined __CUDA) || (defined __ROCM)) template class hamilt::HamiltPW, base_device::DEVICE_GPU>; -template class hamilt::HamiltSdftPW, base_device::DEVICE_GPU>; +template class StoHamiltPW, base_device::DEVICE_GPU>; template class hamilt::HamiltPW, base_device::DEVICE_GPU>; -template class hamilt::HamiltSdftPW, base_device::DEVICE_GPU>; +template class StoHamiltPW, base_device::DEVICE_GPU>; #endif /** diff --git a/source/source_relax/CMakeLists.txt b/source/source_relax/CMakeLists.txt index 691f91b20d0..0d16bbef446 100644 --- a/source/source_relax/CMakeLists.txt +++ b/source/source_relax/CMakeLists.txt @@ -10,6 +10,7 @@ add_library( socket_driver.cpp cg_base.cpp relax_driver.cpp + relax_stru_io.cpp relax_sync.cpp line_search.cpp ions_move_lbfgs.cpp @@ -33,7 +34,7 @@ endif() if(BUILD_TESTING) if(ENABLE_MPI) - add_subdirectory(test) + add_subdirectory(unittests) endif() endif() diff --git a/source/source_relax/bfgs_basic.h b/source/source_relax/bfgs_basic.h index 278230ebf85..8b871b5e4ab 100644 --- a/source/source_relax/bfgs_basic.h +++ b/source/source_relax/bfgs_basic.h @@ -23,6 +23,110 @@ class BFGS_Basic BFGS_Basic(); ~BFGS_Basic() = default; + //==================================================================== + // Test seam. + // + // The BFGS state below is protected and the update machinery private, + // because only Ions_Move_BFGS drives them. The unit tests seed that state + // and step the algorithm one stage at a time, so each piece they touch is + // reachable through the accessors and wrappers here rather than by + // reinterpreting the access specifiers. + // + // Production code must keep using the protected/private names directly; + // nothing outside the tests should call the *_for_testing() wrappers. + //==================================================================== + + /// @brief 3N coordinates of the system ( x ) + std::vector& get_pos() + { + return pos; + } + /// @brief 3N components of ( grad( V(x) ) ) + std::vector& get_grad() + { + return grad; + } + /// @brief the step taken, pos = pos_p + move + std::vector& get_move() + { + return move; + } + /// @brief coordinates of the previous step + std::vector& get_pos_p() + { + return pos_p; + } + /// @brief gradient of the previous step + std::vector& get_grad_p() + { + return grad_p; + } + /// @brief step taken at the previous step + std::vector& get_move_p() + { + return move_p; + } + /// @brief whether a bfgs state has been saved + bool& get_save_flag() + { + return save_flag; + } + /// @brief whether the trust radius already hit its minimum last step + bool& get_tr_min_hit() + { + return tr_min_hit; + } + /// @brief whether the Wolfe conditions were satisfied + bool& get_wolfe_flag() + { + return wolfe_flag; + } + /// @brief the inverse Hessian of the BFGS update + ModuleBase::matrix& get_inv_hess() + { + return inv_hess; + } + /// @brief number of previous steps kept by the BFGS update + int& get_bfgs_ndim() + { + return bfgs_ndim; + } + + void allocate_basic_for_testing() + { + allocate_basic(); + } + void new_step_for_testing(const double& lat0, + int& update_iter, + std::ofstream& ofs, + std::vector& etot_info, + const int test_relax_method) + { + new_step(lat0, update_iter, ofs, etot_info, test_relax_method); + } + void reset_hessian_for_testing() + { + reset_hessian(); + } + void save_bfgs_for_testing() + { + save_bfgs(); + } + void update_inverse_hessian_for_testing(const double& lat0, std::ofstream& ofs) + { + update_inverse_hessian(lat0, ofs); + } + void check_wolfe_conditions_for_testing(std::ofstream& ofs, std::vector& etot_info) + { + check_wolfe_conditions(ofs, etot_info); + } + void compute_trust_radius_for_testing(std::ofstream& ofs, + std::vector& etot_info, + const int test_relax_method) + { + compute_trust_radius(ofs, etot_info, test_relax_method); + } + protected: void allocate_basic(void); void new_step(const double& lat0, int& update_iter, std::ofstream& ofs, std::vector& etot_info, const int test_relax_method); diff --git a/source/source_relax/ions_move_basic.cpp b/source/source_relax/ions_move_basic.cpp index b0edadb2844..1e8764ae958 100644 --- a/source/source_relax/ions_move_basic.cpp +++ b/source/source_relax/ions_move_basic.cpp @@ -148,12 +148,8 @@ bool Ions_Move_Basic::check_converged(const UnitCell &ucell, const double ediff = etot_info[0] - etot_info[1]; std::cout << " ETOT DIFF (eV) : " << ediff * ModuleBase::Ry_to_eV << std::endl; std::cout << " LARGEST GRAD (eV/Angstrom) : " - << Ions_Move_Basic::largest_grad * ModuleBase::Ry_to_eV / ModuleBase::BOHR_TO_A + << Ions_Move_Basic::largest_grad * ModuleBase::Ry_to_eV / ModuleBase::BOHR_TO_A << std::endl; - - ofs << "\n Largest force is " << largest_grad * ModuleBase::Ry_to_eV / ModuleBase::BOHR_TO_A - << " eV/Angstrom while threshold is " - << force_thr_ev << " eV/Angstrom" << std::endl; } const double etot_diff = std::abs(etot_info[0] - etot_info[1]); @@ -162,22 +158,17 @@ bool Ions_Move_Basic::check_converged(const UnitCell &ucell, if (Ions_Move_Basic::largest_grad == 0.0) { - ofs << " largest force is 0, no movement is possible." << std::endl; + ofs << "\n largest force is 0, no movement is possible." << std::endl; ofs << " it may converged, otherwise no movement of atom is allowed." << std::endl; return true; } else if (etot_diff < etot_thr && Ions_Move_Basic::largest_grad < force_thr ) { - ofs << "\n Ion relaxation is converged!" << std::endl; - ofs << "\n Energy difference (Ry) = " << etot_diff << std::endl; - ++update_iter; return true; } else { - ofs << "\n Ion relaxation is not converged yet (threshold is " - << force_thr * ModuleBase::Ry_to_eV / ModuleBase::BOHR_TO_A << ")" << std::endl; return false; } } @@ -206,6 +197,7 @@ void Ions_Move_Basic::terminate(const bool converged, const int update_iter, con else { ofs << " the maximum number of steps has been reached." << std::endl; + ofs << " Relaxation is not converged after reaching relax_nmax!" << std::endl; ofs << " end of geometry optimization." << std::endl; } diff --git a/source/source_relax/ions_move_bfgs.h b/source/source_relax/ions_move_bfgs.h index a9e80523527..9f0a85c8b7d 100644 --- a/source/source_relax/ions_move_bfgs.h +++ b/source/source_relax/ions_move_bfgs.h @@ -18,6 +18,40 @@ class Ions_Move_BFGS : public BFGS_Basic void reset(void); bool start(UnitCell& ucell, const ModuleBase::matrix& force, const double& energy_in, const int istep, int& update_iter, std::ofstream& ofs, std::vector& etot_info, const Relax_Criteria& criteria); + //==================================================================== + // Test seam; see the equivalent block in BFGS_Basic. Production code + // must keep using the private names directly. + //==================================================================== + + /// @brief whether allocate() has already run + bool& get_init_done() + { + return init_done; + } + /// @brief whether this is the first step of the relaxation + bool& get_first_step() + { + return first_step; + } + + void bfgs_routine_for_testing(const double& lat0, + const int istep, + int& update_iter, + std::ofstream& ofs, + std::vector& etot_info, + const std::string& out_level, + const int test_relax_method) + { + bfgs_routine(lat0, istep, update_iter, ofs, etot_info, out_level, test_relax_method); + } + void restart_bfgs_for_testing(const double& lat0, + int& update_iter, + std::ofstream& ofs, + const int test_relax_method) + { + restart_bfgs(lat0, update_iter, ofs, test_relax_method); + } + private: bool init_done; void bfgs_routine(const double& lat0, const int istep, int& update_iter, std::ofstream& ofs, std::vector& etot_info, const std::string& out_level, const int test_relax_method); diff --git a/source/source_relax/ions_move_bfgs2.cpp b/source/source_relax/ions_move_bfgs2.cpp index 3a462e14776..5510c586aef 100644 --- a/source/source_relax/ions_move_bfgs2.cpp +++ b/source/source_relax/ions_move_bfgs2.cpp @@ -13,7 +13,6 @@ void Ions_Move_BFGS2::allocate(const int _size) alpha=70;//default value in ase is 70 maxstep=PARAM.inp.relax_bfgs_rmax; size=_size; - largest_grad=0.0; sign=true; H = std::vector>(3*size, std::vector(3*size, 0.0)); @@ -84,15 +83,8 @@ bool Ions_Move_BFGS2::relax_step(const ModuleBase::matrix& _force,UnitCell& ucel this->DetermineStep(steplength,dpos,maxstep); this->UpdatePos(ucell); this->CalculateLargestGrad(_force,ucell); - bool converged = this->IsRestrain(); - if (converged) - { - ofs_running << "\n Largest force is " << largest_grad * ModuleBase::Ry_to_eV / ModuleBase::BOHR_TO_A - << " eV/Angstrom while threshold is " - << PARAM.inp.force_thr_ev << " eV/Angstrom" << std::endl; - ofs_running << "\n Ion relaxation is converged!" << std::endl; - } - // print out geometry information during bfgs_trad relax + bool converged = this->IsRestrain(); + // print out geometry information during bfgs_trad relax unitcell::print_tau(ucell.atoms,ucell.Coordinate,ucell.ntype,ucell.lat0,ofs_running); return converged; } @@ -362,19 +354,19 @@ void Ions_Move_BFGS2::CalculateLargestGrad(const ModuleBase::matrix& _force,Unit ++iat; } } - largest_grad = 0.0; + Ions_Move_Basic::largest_grad = 0.0; for (int i = 0; i < 3*size; i++) { - if (largest_grad < std::abs(grad[i])) + if (Ions_Move_Basic::largest_grad < std::abs(grad[i])) { - largest_grad = std::abs(grad[i]); + Ions_Move_Basic::largest_grad = std::abs(grad[i]); } } assert(ucell.lat0 != 0); Ions_Move_Basic::largest_grad /= ucell.lat0; if (PARAM.inp.out_level == "ie") { - std::cout << " LARGEST GRAD (eV/Angstrom) : " << largest_grad + std::cout << " LARGEST GRAD (eV/Angstrom) : " << Ions_Move_Basic::largest_grad * ModuleBase::Ry_to_eV / ModuleBase::BOHR_TO_A << std::endl; } @@ -382,5 +374,5 @@ void Ions_Move_BFGS2::CalculateLargestGrad(const ModuleBase::matrix& _force,Unit bool Ions_Move_BFGS2::IsRestrain() { - return largest_grad * ModuleBase::Ry_to_eV / ModuleBase::BOHR_TO_A < PARAM.inp.force_thr_ev; + return Ions_Move_Basic::largest_grad * ModuleBase::Ry_to_eV / ModuleBase::BOHR_TO_A < PARAM.inp.force_thr_ev; } diff --git a/source/source_relax/ions_move_bfgs2.h b/source/source_relax/ions_move_bfgs2.h index ff915f4782e..e84a1f6ad07 100644 --- a/source/source_relax/ions_move_bfgs2.h +++ b/source/source_relax/ions_move_bfgs2.h @@ -17,7 +17,13 @@ class Ions_Move_BFGS2 void allocate(const int _size);//initialize parameters void reset(); bool relax_step(const ModuleBase::matrix& _force,UnitCell& ucell, std::ofstream& ofs_running);//a full iteration step - + + /// @brief whether allocate() has run and the optimiser holds live state. + /// reset() clears it; ions_move_methods_test checks that. + bool get_is_initialized() const + { + return is_initialized; + } private: // The unit test drives the private step machinery directly (seeding the @@ -27,7 +33,6 @@ class Ions_Move_BFGS2 bool sign;//check if this is the first iteration double alpha;//initialize H,diagonal element is alpha double maxstep;//every movement smaller than maxstep - double largest_grad; int size;//number of atoms bool is_initialized=false; diff --git a/source/source_relax/ions_move_lbfgs.cpp b/source/source_relax/ions_move_lbfgs.cpp index 7e5f348a347..fed26a3c82f 100644 --- a/source/source_relax/ions_move_lbfgs.cpp +++ b/source/source_relax/ions_move_lbfgs.cpp @@ -1,14 +1,14 @@ #include "ions_move_lbfgs.h" #include "matrix_methods.h" -#include "source_io/module_parameter/parameter.h" #include "ions_move_basic.h" #include "source_cell/update_cell.h" #include "source_cell/print_cell.h" // mohan add 2025-06-19 -void Ions_Move_LBFGS::allocate(const int _size) // initialize H0、H、pos0、force0、force +void Ions_Move_LBFGS::allocate(const int _size, const double relax_bfgs_rmax, const std::string& out_level) // initialize H0、H、pos0、force0、force { alpha=70;//default value in ase is 70 - maxstep=PARAM.inp.relax_bfgs_rmax; + maxstep=relax_bfgs_rmax; + this->out_level=out_level; size=_size; memory=100; iteration=0; @@ -25,7 +25,7 @@ void Ions_Move_LBFGS::allocate(const int _size) // initialize H0、H、pos0、fo //l_search.init_line_search(); } -bool Ions_Move_LBFGS::relax_step(const ModuleBase::matrix _force,UnitCell& ucell,const double &etot, std::ofstream& ofs_running) +bool Ions_Move_LBFGS::relax_step(const ModuleBase::matrix _force,UnitCell& ucell,const double &etot, std::ofstream& ofs_running, const Relax_Criteria& criteria) { get_pos(ucell,pos); get_pos_taud(ucell,pos_taud); @@ -62,7 +62,7 @@ bool Ions_Move_LBFGS::relax_step(const ModuleBase::matrix _force,UnitCell& ucell this->determine_step(steplength,dpos,maxstep); this->update_pos(ucell); this->calculate_largest_grad(_force,ucell); - bool converged = this->is_restrain(); + bool converged = this->is_restrain(criteria); // mohan add 2025-06-22 unitcell::print_tau(ucell.atoms,ucell.Coordinate,ucell.ntype,ucell.lat0,ofs_running); return converged; @@ -251,9 +251,9 @@ void Ions_Move_LBFGS::update_pos(UnitCell& ucell) unitcell::update_pos_tau(ucell.lat,a.data(),ucell.ntype,ucell.nat,ucell.atoms); } -bool Ions_Move_LBFGS::is_restrain() +bool Ions_Move_LBFGS::is_restrain(const Relax_Criteria& criteria) { - return Ions_Move_Basic::largest_grad * ModuleBase::Ry_to_eV / ModuleBase::BOHR_TO_A < PARAM.inp.force_thr_ev; + return Ions_Move_Basic::largest_grad * ModuleBase::Ry_to_eV / ModuleBase::BOHR_TO_A < criteria.force_thr_ev; } void Ions_Move_LBFGS::calculate_largest_grad(const ModuleBase::matrix& _force,UnitCell& ucell) @@ -284,7 +284,7 @@ void Ions_Move_LBFGS::calculate_largest_grad(const ModuleBase::matrix& _force,Un } } Ions_Move_Basic::largest_grad /= ucell.lat0; - if (PARAM.inp.out_level == "ie") + if (out_level == "ie") { std::cout << " LARGEST GRAD (eV/Angstrom) : " << Ions_Move_Basic::largest_grad * ModuleBase::Ry_to_eV / ModuleBase::BOHR_TO_A diff --git a/source/source_relax/ions_move_lbfgs.h b/source/source_relax/ions_move_lbfgs.h index 8ee9b217c96..02d2447e5ce 100644 --- a/source/source_relax/ions_move_lbfgs.h +++ b/source/source_relax/ions_move_lbfgs.h @@ -2,16 +2,18 @@ #define IONS_MOVE_LBFGS_H #include -#include +#include #include #include #include +#include //#include "line_search.h" #include "source_base/matrix.h" #include "source_base/matrix3.h" #include "source_cell/unitcell.h" #include "source_esolver/esolver.h" #include "source_esolver/esolver_ks.h" +#include "relax_criteria.h" /** * @class Ions_Move_LBFGS @@ -23,8 +25,10 @@ class Ions_Move_LBFGS /** * @brief Initialize L-BFGS parameters * @param _size Number of atoms in system + * @param relax_bfgs_rmax Maximum allowed step length (from INPUT) + * @param out_level Output verbosity level (from INPUT) */ - void allocate(const int _size); + void allocate(const int _size, const double relax_bfgs_rmax, const std::string& out_level); /** * @brief Perform one L-BFGS relaxation step @@ -36,7 +40,8 @@ class Ions_Move_LBFGS bool relax_step(const ModuleBase::matrix _force, UnitCell& ucell, const double &etot, - std::ofstream& ofs_running); + std::ofstream& ofs_running, + const Relax_Criteria& criteria); private: //LineSearch l_search; @@ -48,6 +53,7 @@ class Ions_Move_LBFGS int iteration; ///< Current iteration count double energy; ///< Current system energy double alpha_k; ///< Step size parameter + std::string out_level; ///< Output verbosity level ModuleESolver::ESolver* solver = nullptr; ///< Structure solver std::vector steplength;//the length of atoms displacement @@ -77,9 +83,10 @@ class Ions_Move_LBFGS /** * @brief Judge if the relaxation has converged + * @param criteria Convergence thresholds passed from the driver * @return true if converged, false otherwise */ - bool is_restrain(); + bool is_restrain(const Relax_Criteria& criteria); /** * @brief Calculate maximum gradient component diff --git a/source/source_relax/ions_move_methods.cpp b/source/source_relax/ions_move_methods.cpp index d96ee2d0b72..6b2fede7077 100644 --- a/source/source_relax/ions_move_methods.cpp +++ b/source/source_relax/ions_move_methods.cpp @@ -5,6 +5,7 @@ #include "ions_move_basic.h" #include "source_base/global_function.h" #include "source_base/global_variable.h" +#include "source_io/module_parameter/parameter.h" Ions_Move_Methods::Ions_Move_Methods() @@ -45,7 +46,7 @@ void Ions_Move_Methods::allocate(const int &natom, const std::string& relax_meth } else if(relax_method_0 == "lbfgs") { - this->lbfgs.allocate(natom); + this->lbfgs.allocate(natom, PARAM.inp.relax_bfgs_rmax, PARAM.inp.out_level); } else { @@ -87,7 +88,7 @@ void Ions_Move_Methods::cal_movement(const int &istep, } else if (relax_method[0] == "lbfgs") { - converged_ = lbfgs.relax_step(f, ucell, etot, ofs); + converged_ = lbfgs.relax_step(f, ucell, etot, ofs, criteria); } else { diff --git a/source/source_relax/ions_move_methods.h b/source/source_relax/ions_move_methods.h index 36da434ba87..cc4b22fc253 100644 --- a/source/source_relax/ions_move_methods.h +++ b/source/source_relax/ions_move_methods.h @@ -51,6 +51,41 @@ class Ions_Move_Methods return update_iter_; } + //==================================================================== + // Test seam. + // + // reset_after_cell_change() is expected to clear the per-run state of + // whichever method is active, so the test seeds that state through the + // sub-optimiser it belongs to and checks it was cleared. get_converged() + // and get_update_iter() above already cover the reads. + // + // Production code must keep using the private names directly. + //==================================================================== + + void set_converged(const bool value) + { + converged_ = value; + } + void set_update_iter(const int value) + { + update_iter_ = value; + } + /// @brief {etot, etot_p} of the current and previous step + std::vector& get_etot_info() + { + return etot_info_; + } + /// @brief the BFGS optimiser used by relax_method "bfgs" + Ions_Move_BFGS& get_bfgs() + { + return bfgs; + } + /// @brief the traditional BFGS optimiser used by relax_method "bfgs_trad" + Ions_Move_BFGS2& get_bfgs_trad() + { + return bfgs_trad; + } + private: Ions_Move_BFGS bfgs; Ions_Move_CG cg; diff --git a/source/source_relax/lattice_change_basic.cpp b/source/source_relax/lattice_change_basic.cpp index 7a45f109d73..9838980e44c 100644 --- a/source/source_relax/lattice_change_basic.cpp +++ b/source/source_relax/lattice_change_basic.cpp @@ -279,16 +279,11 @@ bool Lattice_Change_Basic::check_converged(const UnitCell &ucell, ModuleBase::ma { if (Lattice_Change_Basic::largest_grad < stress_thr && stress_ii_max < stress_thr) { - ofs << "\n Geometry relaxation is converged!" << std::endl; - ofs << "\n Largest stress is " << largest_grad - << " kbar while threshold is " << stress_thr << " kbar" << std::endl; ++Lattice_Change_Basic::update_iter; return true; } else { - ofs << "\n Geometry relaxation is not converged because threshold is " << stress_thr - << " kbar" << std::endl; return false; } } @@ -297,16 +292,11 @@ bool Lattice_Change_Basic::check_converged(const UnitCell &ucell, ModuleBase::ma // the code is almost the same as previous codes if (Lattice_Change_Basic::largest_grad < 10 * stress_thr) { - ofs << "\n Geometry relaxation is converged!" << std::endl; - ofs << "\n Largest stress is " << largest_grad - << " kbar while threshold is " << stress_thr << " kbar" << std::endl; ++Lattice_Change_Basic::update_iter; return true; } else { - ofs << "\n Geometry relaxation is not converged because threshold is " << stress_thr - << " kbar" << std::endl; return false; } } diff --git a/source/source_relax/relax_driver.cpp b/source/source_relax/relax_driver.cpp index 38a75198b4e..7497fc8e1c9 100644 --- a/source/source_relax/relax_driver.cpp +++ b/source/source_relax/relax_driver.cpp @@ -1,17 +1,14 @@ #include "relax_driver.h" +#include "relax_history.h" +#include "relax_stru_io.h" #include "socket_driver.h" #include "source_base/formatter.h" #include "source_base/global_file.h" -#include "source_cell/cif_io.h" #include "source_io/module_json/output_info.h" #include "source_io/module_output/output_log.h" #include "source_io/module_output/print_info.h" #include "source_base/module_out/read_exit_file.h" #include "source_io/module_parameter/parameter.h" -#include "source_cell/print_cell.h" -#include "source_main/version.h" - -#include void Relax_Driver::relax_driver( ModuleESolver::ESolver* p_esolver, @@ -22,6 +19,12 @@ void Relax_Driver::relax_driver( ModuleBase::TITLE("Relax_Driver", "relax_driver"); ModuleBase::timer::start("Relax_Driver", "relax_driver"); + // Cache global configuration once so downstream helpers do not each reach + // into PARAM / GlobalV. + out_dir_ = PARAM.globalv.global_out_dir; + deepks_setorb_ = PARAM.globalv.deepks_setorb; + my_rank_ = GlobalV::MY_RANK; + if (inp.socket_driver) { Socket_Driver socket_driver; @@ -42,15 +45,24 @@ void Relax_Driver::relax_driver( // so the loop exits after one iteration double etot = 0.0; ModuleBase::matrix stress(3, 3); + ModuleBase::matrix force(ucell.nat, 3); + + // Track whether the current geometry has been evaluated by esolve(). + // After relax_step() proposes a new geometry, it is not evaluated until + // the next esolve() call; if we exit the loop early, force/stress are stale. + bool geometry_evaluated = false; while (steps[0] < inp.relax_nmax) { - ModuleBase::matrix force(ucell.nat, 3); - this->iter_info(steps, inp); this->esolve(steps[0], p_esolver, ucell, inp, force, stress, etot); - this->stru_out(steps[0], ucell, inp, etot, stress); + geometry_evaluated = true; + this->stru_out(steps[0], ucell, inp, etot, stress, force); bool converged = this->relax_step(steps, p_esolver, ucell, inp, force, stress, etot, ofs_running); + if (!converged) + { + geometry_evaluated = false; + } this->json_out(p_esolver, ucell, inp, force, stress); // Check stop conditions @@ -59,7 +71,7 @@ void Relax_Driver::relax_driver( // Relaxation converged, exit loop immediately break; } - else if (ModuleIO::read_exit_file(GlobalV::MY_RANK, "EXIT", ofs_running)) + else if (ModuleIO::read_exit_file(my_rank_, "EXIT", ofs_running)) { // EXIT file detected, exit loop break; @@ -68,7 +80,7 @@ void Relax_Driver::relax_driver( ++steps[0]; } - this->final_out(steps[0], ucell, inp, etot, stress); + this->final_out(steps[0], ucell, inp, etot, stress, force, geometry_evaluated, ofs_running); ModuleBase::timer::end("Relax_Driver", "relax_driver"); return; @@ -101,18 +113,23 @@ void Relax_Driver::iter_info(const std::vector& steps, const Input_para& in ModuleIO::print_screen(steps[2], steps[1], steps[0]+1); } -#ifdef __RAPIDJSON - Json::init_output_array_obj(); +#ifdef __JSON + // ks-lr runs an embedded KS calculation in before_all_runners(), which + // already starts the first output record. + if (inp.esolver_type != "ks-lr" || steps[0] != 0) + { + Json::init_output_array_obj(); + } #endif } void Relax_Driver::esolve(const int istep, - ModuleESolver::ESolver* p_esolver, - UnitCell& ucell, - const Input_para& inp, - ModuleBase::matrix& force, - ModuleBase::matrix& stress, - double& etot) + ModuleESolver::ESolver* p_esolver, + UnitCell& ucell, + const Input_para& inp, + ModuleBase::matrix& force, + ModuleBase::matrix& stress, + double& etot) { p_esolver->runner(ucell, istep); @@ -130,13 +147,13 @@ void Relax_Driver::esolve(const int istep, } bool Relax_Driver::relax_step(std::vector& steps, - ModuleESolver::ESolver* p_esolver, - UnitCell& ucell, - const Input_para& inp, - const ModuleBase::matrix& force, - const ModuleBase::matrix& stress, - const double etot, - std::ofstream& ofs_running) + ModuleESolver::ESolver* p_esolver, + UnitCell& ucell, + const Input_para& inp, + const ModuleBase::matrix& force, + const ModuleBase::matrix& stress, + const double etot, + std::ofstream& ofs_running) { // Guard: For non-relaxation calculations (scf, nscf, etc.), return true immediately // to ensure the main loop exits after one iteration. This provides robustness @@ -151,15 +168,15 @@ bool Relax_Driver::relax_step(std::vector& steps, if (inp.uses_simultaneous_relaxation()) { converged = this->rl.relax_step(ucell, force, stress, etot, ofs_running); - // stress step +1 + // stress step +1 steps[2]++; - // fix force step to 1 + // fix force step to 1 steps[1] = 1; } else { converged = this->rl_old.relax_step(steps[0]+1, etot, ucell, force, - stress, steps[1], steps[2], ofs_running); + stress, steps[1], steps[2], ofs_running); } ModuleIO::output_after_relax(converged, p_esolver->conv_esolver, ofs_running); @@ -167,14 +184,17 @@ bool Relax_Driver::relax_step(std::vector& steps, return converged; } -void Relax_Driver::stru_out(const int istep, UnitCell& ucell, const Input_para& inp, const double etot, const ModuleBase::matrix& stress) +void Relax_Driver::stru_out(const int istep, UnitCell& ucell, const Input_para& inp, const double etot, const ModuleBase::matrix& stress, const ModuleBase::matrix& force) { - // Guard: only output structure files for relaxation calculations - if (inp.calculation != "relax" && inp.calculation != "cell-relax") + // out_stru is effective for scf/nscf/relax/cell-relax (md writes STRU_MD_* via md_restartfreq) + if (inp.calculation != "relax" && inp.calculation != "cell-relax" + && inp.calculation != "scf" && inp.calculation != "nscf") { return; } + const bool is_relax = (inp.calculation == "relax" || inp.calculation == "cell-relax"); + // out_stru: -1 no output, 0 final only, 1 STRU format, 2 CIF format // For -1 and 0, no per-step structure output if (inp.out_stru <= 0) @@ -182,186 +202,138 @@ void Relax_Driver::stru_out(const int istep, UnitCell& ucell, const Input_para& return; } - // cache global parameters to reduce repeated PARAM access - const std::string& out_dir = PARAM.globalv.global_out_dir; - const bool deepks_setorb = PARAM.globalv.deepks_setorb; - - // Build header comment with version, timestamp, energy and stress - std::time_t now = std::time(nullptr); - char time_buf[64]; - std::strftime(time_buf, sizeof(time_buf), "%Y-%m-%d %H:%M:%S", std::localtime(&now)); - std::string header = FmtCore::format("# ABACUS version: %s\n# Written at %s\n# RELAX STEP %d, Energy: %.8f eV\n", - VERSION, - time_buf, - istep + 1, - etot * ModuleBase::Ry_to_eV); - // stress in kbar: Ry/Bohr^3 -> kbar, 3 rows - const double stress_transform = ModuleBase::RYDBERG_SI - / (ModuleBase::BOHR_RADIUS_SI * ModuleBase::BOHR_RADIUS_SI * ModuleBase::BOHR_RADIUS_SI) - * 1.0e-8; - for (int i = 0; i < 3; i++) - { - header += FmtCore::format("# Stress (kbar): %.6f %.6f %.6f\n", - stress(i, 0) * stress_transform, - stress(i, 1) * stress_transform, - stress(i, 2) * stress_transform); - } - - bool need_orb = inp.basis_type == "pw"; - need_orb = need_orb && inp.init_wfc.substr(0, 3) == "nao"; - need_orb = need_orb || inp.basis_type == "lcao"; - need_orb = need_orb || inp.basis_type == "lcao_in_pw"; - + // stru_out is called right after esolve(), so the geometry has been + // evaluated and forces/stress are consistent with the written structure. + const bool geometry_evaluated = true; + const std::string header = relax_stru_io::build_stru_header(istep, etot, stress, inp, false, geometry_evaluated); + const bool need_orb = relax_stru_io::need_orbital(inp); const bool freq_ok = (inp.out_freq_ion > 0 && istep % inp.out_freq_ion == 0); // STRU_NOW: overwrite each step (for out_stru 1 and 2) - if (inp.out_stru == 1) - { - unitcell::print_stru_file(ucell, - ucell.atoms, - ucell.latvec, - out_dir + "STRU_NOW", - header, - inp.nspin, - true, - inp.calculation == "md", - inp.out_mul, - need_orb, - deepks_setorb, - GlobalV::MY_RANK); - } - else if (inp.out_stru == 2) + // For scf/nscf the structure is identical to STRU_FINAL; only STRU_FINAL + // is written in final_out() to avoid a duplicate file. + if (is_relax) { - ModuleIO::CifParser::write(out_dir + "STRU_NOW.cif", - ucell, - header, - "data_?", - GlobalV::MY_RANK); + const std::string now_file = out_dir_ + (inp.out_stru == 1 ? "STRU_NOW" : "STRU_NOW.cif"); + relax_stru_io::write_stru(ucell, inp, now_file, header, force, need_orb, + deepks_setorb_, my_rank_, inp.cal_force); } - // Numbered files per out_freq_ion (for out_stru 1 and 2 only) - if (freq_ok) + // Numbered files per out_freq_ion: only meaningful for relaxation calculations + if (is_relax && freq_ok) { - if (inp.out_stru == 1) - { - unitcell::print_stru_file(ucell, - ucell.atoms, - ucell.latvec, - out_dir + "STRU" + std::to_string(istep + 1), - header, - inp.nspin, - true, - inp.calculation == "md", - inp.out_mul, - need_orb, - deepks_setorb, - GlobalV::MY_RANK); - } - else if (inp.out_stru == 2) - { - ModuleIO::CifParser::write(out_dir + "STRU" + std::to_string(istep + 1) + ".cif", - ucell, - header, - "data_?", - GlobalV::MY_RANK); - } + const std::string step_file = out_dir_ + "STRU" + std::to_string(istep + 1) + + (inp.out_stru == 1 ? "" : ".cif"); + relax_stru_io::write_stru(ucell, inp, step_file, header, force, need_orb, + deepks_setorb_, my_rank_, inp.cal_force); } } void Relax_Driver::json_out(ModuleESolver::ESolver* p_esolver, UnitCell& ucell, const Input_para& inp, const ModuleBase::matrix& force, const ModuleBase::matrix& stress) { -#ifdef __RAPIDJSON +#ifdef __JSON Json::add_output_energy(p_esolver->cal_energy() * ModuleBase::Ry_to_eV); double unit_transform = ModuleBase::RYDBERG_SI / pow(ModuleBase::BOHR_RADIUS_SI, 3) * 1.0e-8; double fac = ModuleBase::Ry_to_eV / 0.529177; - Json::add_output_cell_coo_stress_force(&ucell, force, fac, stress, unit_transform); + Json::add_output_cell_coo_stress_force(ucell, + force, + fac, + stress, + unit_transform, + inp.cal_force, + inp.cal_stress); #endif } -void Relax_Driver::final_out(const int istep, UnitCell& ucell, const Input_para& inp, const double etot, const ModuleBase::matrix& stress) +void Relax_Driver::final_out(const int istep, + UnitCell& ucell, + const Input_para& inp, + const double etot, + const ModuleBase::matrix& stress, + const ModuleBase::matrix& force, + const bool geometry_evaluated, + std::ofstream& ofs_running) { - if (inp.calculation != "relax" && inp.calculation != "cell-relax") - { - return; - } + // Structure final output is effective for scf/nscf/relax/cell-relax; + // relax-specific screen messages remain guarded below. + const bool is_relax = (inp.calculation == "relax" || inp.calculation == "cell-relax"); + const bool stru_effective = is_relax || inp.calculation == "scf" || inp.calculation == "nscf"; // out_stru: 0 no output, 1 STRU format, 2 CIF format // 1: write STRU_FINAL; 2: write STRU_FINAL.cif - if (inp.out_stru == 1 || inp.out_stru == 2) + if (stru_effective && (inp.out_stru == 1 || inp.out_stru == 2)) { - // cache global parameters to reduce repeated PARAM access - const std::string& out_dir = PARAM.globalv.global_out_dir; - const bool deepks_setorb = PARAM.globalv.deepks_setorb; - - // Build header comment for STRU_FINAL - std::time_t now = std::time(nullptr); - char time_buf[64]; - std::strftime(time_buf, sizeof(time_buf), "%Y-%m-%d %H:%M:%S", std::localtime(&now)); - std::string header = FmtCore::format("# ABACUS version: %s\n# Written at %s\n# RELAX STEP %d (FINAL), Energy: %.8f eV\n", - VERSION, - time_buf, - istep + 1, - etot * ModuleBase::Ry_to_eV); - const double stress_transform = ModuleBase::RYDBERG_SI - / (ModuleBase::BOHR_RADIUS_SI * ModuleBase::BOHR_RADIUS_SI * ModuleBase::BOHR_RADIUS_SI) - * 1.0e-8; - for (int i = 0; i < 3; i++) - { - header += FmtCore::format("# Stress (kbar): %.6f %.6f %.6f\n", - stress(i, 0) * stress_transform, - stress(i, 1) * stress_transform, - stress(i, 2) * stress_transform); - } + const std::string header = relax_stru_io::build_stru_header(istep, etot, stress, inp, true, geometry_evaluated); + const bool need_orb = relax_stru_io::need_orbital(inp); + const std::string final_file = out_dir_ + (inp.out_stru == 1 ? "STRU_FINAL" : "STRU_FINAL.cif"); + // Only write forces when they were actually computed and belong + // to the geometry being written. + const bool write_force = inp.cal_force && geometry_evaluated; + relax_stru_io::write_stru(ucell, inp, final_file, header, force, need_orb, + deepks_setorb_, my_rank_, write_force); + } - if (inp.out_stru == 1) + // relax_nmax == 0 is a valid dry-run mode: no relaxation step was ever + // taken, so it must not be reported as a failed relaxation. + if (istep == inp.relax_nmax && inp.relax_nmax > 0) + { + if (is_relax) { - bool need_orb = inp.basis_type == "pw"; - need_orb = need_orb && inp.init_wfc.substr(0, 3) == "nao"; - need_orb = need_orb || inp.basis_type == "lcao"; - need_orb = need_orb || inp.basis_type == "lcao_in_pw"; - - unitcell::print_stru_file(ucell, - ucell.atoms, - ucell.latvec, - out_dir + "STRU_FINAL", - header, - inp.nspin, - true, - inp.calculation == "md", - inp.out_mul, - need_orb, - deepks_setorb, - GlobalV::MY_RANK); + print_not_converged_summary(inp, ofs_running); + + std::cout << "\n ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~" << std::endl; + std::cout << " Geometry relaxation stops here due to reaching the maximum " << std::endl; + std::cout << " relaxation steps. More steps are needed to converge the results " << std::endl; + std::cout << " ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~" << std::endl; } - else if (inp.out_stru == 2) + } + else if (inp.relax_nmax > 0) + { + if (is_relax) { - ModuleIO::CifParser::write(out_dir + "STRU_FINAL.cif", - ucell, - header, - "data_?", - GlobalV::MY_RANK); + std::cout << "\n ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~" << std::endl; + std::cout << " Geometry relaxation thresholds are reached within " << istep << " steps." << std::endl; + std::cout << " ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~" << std::endl; } } - if (istep == inp.relax_nmax) + if (is_relax && inp.relax_nmax == 0) { - std::cout << "\n ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~" << std::endl; - std::cout << " Geometry relaxation stops here due to reaching the maximum " << std::endl; - std::cout << " relaxation steps. More steps are needed to converge the results " << std::endl; - std::cout << " ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~" << std::endl; + print_dry_run_message(); } - else +} + +void Relax_Driver::print_not_converged_summary(const Input_para& inp, std::ofstream& ofs_running) const +{ + // Unified not-converged summary for both relaxation paths so ASE and + // users can read the method, the actual number of steps taken, and + // the per-step force / stress history from the running log. + const std::vector& force_hist = inp.uses_simultaneous_relaxation() + ? rl.get_max_force_history() + : rl_old.get_max_force_history(); + const std::vector& stress_hist = inp.uses_simultaneous_relaxation() + ? rl.get_max_stress_history() + : rl_old.get_max_stress_history(); + const int ionic_steps = static_cast(force_hist.size()); + const std::string method = inp.relax_method.empty() ? "unknown" : inp.relax_method[0]; + ofs_running << " Relaxation method: " << method << std::endl; + ofs_running << " Relaxation stopped after " << ionic_steps << " ionic step(s) (relax_nmax = " + << inp.relax_nmax << " reached)." << std::endl; + if (!force_hist.empty()) { - std::cout << "\n ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~" << std::endl; - std::cout << " Geometry relaxation thresholds are reached within " << istep << " steps." << std::endl; - std::cout << " ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~" << std::endl; + ofs_running << " Largest force per step (eV/Angstrom):" << format_relax_history(force_hist); } - - if (inp.relax_nmax == 0) + if (!stress_hist.empty()) { - std::cout << "-----------------------------------------------" << std::endl; - std::cout << " relax_nmax = 0, DRY RUN TEST SUCCEEDS :)" << std::endl; - std::cout << "-----------------------------------------------" << std::endl; + ofs_running << " Largest stress per step (kbar):" << format_relax_history(stress_hist); } + ofs_running << " Relaxation is not converged after reaching relax_nmax!" << std::endl; +} + +void Relax_Driver::print_dry_run_message() const +{ + std::cout << "-----------------------------------------------" << std::endl; + std::cout << " relax_nmax = 0, DRY RUN TEST SUCCEEDS :)" << std::endl; + std::cout << "-----------------------------------------------" << std::endl; } diff --git a/source/source_relax/relax_driver.h b/source/source_relax/relax_driver.h index 8636c98fc76..be4b99ca478 100644 --- a/source/source_relax/relax_driver.h +++ b/source/source_relax/relax_driver.h @@ -9,6 +9,7 @@ #include "source_base/matrix.h" #include #include +#include /** * @brief Driver class for geometry relaxation calculations. @@ -46,6 +47,13 @@ class Relax_Driver /// Old relaxation optimizer (IonCellOptimizer class) IonCellOptimizer rl_old; + /// Cached global output directory (read once from PARAM at relax_driver entry) + std::string out_dir_; + /// Cached DeePKS setorb flag (read once from PARAM at relax_driver entry) + bool deepks_setorb_ = false; + /// Cached MPI rank (read once at relax_driver entry) + int my_rank_ = 0; + /** * @brief Initialize the relaxation optimizer. * @@ -102,7 +110,7 @@ class Relax_Driver * @param etot Total energy in Ry corresponding to this structure. * @param stress Stress matrix (3x3) in Ry/Bohr^3 corresponding to this structure. */ - void stru_out(const int istep, UnitCell& ucell, const Input_para& inp, const double etot, const ModuleBase::matrix& stress); + void stru_out(const int istep, UnitCell& ucell, const Input_para& inp, const double etot, const ModuleBase::matrix& stress, const ModuleBase::matrix& force); /** * @brief Output JSON format results. @@ -124,8 +132,36 @@ class Relax_Driver * @param inp Input parameters for the calculation. * @param etot Total energy of the final step. * @param stress Stress tensor of the final step. + * @param force Force matrix of the final step. + * @param geometry_evaluated false when the final geometry was proposed by + * the optimizer but never evaluated; forces and stress are then + * omitted / marked N/A. + */ + void final_out(const int istep, + UnitCell& ucell, + const Input_para& inp, + const double etot, + const ModuleBase::matrix& stress, + const ModuleBase::matrix& force, + const bool geometry_evaluated, + std::ofstream& ofs_running); + + /** + * @brief Print the unified not-converged summary to the running log. + * + * Reports the relaxation method, the number of ionic steps taken and the + * per-step largest force / stress history, so ASE and users can diagnose + * why the relaxation stopped at relax_nmax. + * + * @param inp Input parameters (selects the active optimizer and relax_nmax). + * @param ofs_running Output stream for the running log. + */ + void print_not_converged_summary(const Input_para& inp, std::ofstream& ofs_running) const; + + /** + * @brief Print the dry-run success message to screen. */ - void final_out(const int istep, UnitCell& ucell, const Input_para& inp, const double etot, const ModuleBase::matrix& stress); + void print_dry_run_message() const; }; #endif diff --git a/source/source_relax/relax_history.h b/source/source_relax/relax_history.h new file mode 100644 index 00000000000..d71b6ae482e --- /dev/null +++ b/source/source_relax/relax_history.h @@ -0,0 +1,70 @@ +#ifndef RELAX_HISTORY_H +#define RELAX_HISTORY_H + +#include +#include +#include +#include + +/** + * @file relax_history.h + * @brief Shared formatting of per-step convergence history for the running log. + * + * Both relaxation paths (Relax in relax_sync.cpp and IonCellOptimizer in + * relax_nsync.cpp) record the largest force / stress of each step and print a + * summary when relaxation finishes. This free function keeps that summary + * layout identical across the two paths. + */ + +/** + * @brief Format a per-step history for the running log. + * + * Layout rules: + * - Up to max_full entries are printed in full, per_line values per line. + * - Beyond max_full, only the first and last keep entries are shown, with an + * omission marker in between, so very long runs stay readable. + */ +inline std::string format_relax_history(const std::vector& hist, + const int per_line = 5, + const int max_full = 100, + const int keep = 10) +{ + std::ostringstream out; + out << std::scientific << std::setprecision(3); + + const int n = static_cast(hist.size()); + const bool truncate = (n > max_full); + const int head = truncate ? keep : n; + + int printed = 0; + auto print_one = [&out, &printed, per_line](const double value) { + if (printed % per_line == 0) + { + out << "\n "; + } + else + { + out << " "; + } + out << value; + ++printed; + }; + + for (int i = 0; i < head; ++i) + { + print_one(hist[i]); + } + if (truncate) + { + const int omitted = n - 2 * keep; + out << "\n ... (omitted " << omitted << " step(s)) ..."; + for (int i = n - keep; i < n; ++i) + { + print_one(hist[i]); + } + } + out << "\n"; + return out.str(); +} + +#endif // RELAX_HISTORY_H diff --git a/source/source_relax/relax_nsync.cpp b/source/source_relax/relax_nsync.cpp index e0c1b241429..548090880d8 100644 --- a/source/source_relax/relax_nsync.cpp +++ b/source/source_relax/relax_nsync.cpp @@ -1,12 +1,30 @@ #include "relax_nsync.h" +#include "relax_history.h" #include "source_base/global_function.h" #include "source_base/global_variable.h" #include "source_cell/cell_tools.h" #include "source_cell/update_cell.h" +void IonCellOptimizer::print_converged_summary(const int istep, + const bool with_stress, + std::ofstream& ofs_running) const +{ + ofs_running << " Relaxation method: " << inp_->relax_method[0] << std::endl; + ofs_running << " Relaxation converged in " << istep << " step(s)." << std::endl; + if (!max_force_history_.empty()) + { + ofs_running << " Largest force per step (eV/Angstrom):" << format_relax_history(max_force_history_); + } + if (with_stress && !max_stress_history_.empty()) + { + ofs_running << " Largest stress per step (kbar):" << format_relax_history(max_stress_history_); + } + ofs_running << "\n Relaxation is converged!" << std::endl; +} + /** * @brief Initialize relaxation algorithms based on calculation type. - * + * * Allocates memory and initializes the appropriate relaxation methods: * - For "relax" calculation: only initializes Ions_Move_Methods * - For "cell-relax" calculation: initializes both Ions_Move_Methods and @@ -17,6 +35,8 @@ void IonCellOptimizer::init_relax(const int& natom, const Input_para& inp) { inp_ = &inp; + max_force_history_.clear(); + max_stress_history_.clear(); if (inp_->calculation == "relax") { @@ -66,9 +86,32 @@ bool IonCellOptimizer::relax_step(const int& istep, ucell.ionic_position_updated = false; ucell.cell_parameter_updated = false; - // Check if we've reached the maximum number of iterations + // relax_nmax == 0 is a valid dry-run mode: no relaxation step may run, and + // it must not be reported as a failed relaxation (mirrors the dry-run + // branch of Relax_Driver::final_out). Terminate immediately and silently. + if (inp_->relax_nmax == 0) + { + return true; + } + + // Check if we've reached the maximum number of iterations. if (istep == inp_->relax_nmax) { + // This step never ran cal_movement, so no force was recorded for it; + // the actual number of ionic steps taken is the history length. + const int ionic_steps = static_cast(max_force_history_.size()); + ofs_running << " Relaxation method: " << inp_->relax_method[0] << std::endl; + ofs_running << " Relaxation stopped after " << ionic_steps << " ionic step(s) (relax_nmax = " + << inp_->relax_nmax << " reached)." << std::endl; + if (!max_force_history_.empty()) + { + ofs_running << " Largest force per step (eV/Angstrom):" << format_relax_history(max_force_history_); + } + if (!max_stress_history_.empty()) + { + ofs_running << " Largest stress per step (kbar):" << format_relax_history(max_stress_history_); + } + ofs_running << " Relaxation is not converged after reaching relax_nmax!" << std::endl; return true; } @@ -86,6 +129,11 @@ bool IonCellOptimizer::relax_step(const int& istep, const bool need_atom_relax = (is_relax || is_cell_relax) && unitcell::if_atoms_can_move(ucell.atoms, ucell.ntype); const bool need_cell_relax = is_cell_relax && unitcell::if_cell_can_change(ucell.lat_axis_free); + // Track whether the atomic part has converged this step (used to decide + // which final marker to print and whether cell relaxation may proceed) + bool atom_converged = false; + bool atom_relax_performed = false; + // Atomic relaxation branch if (need_atom_relax) { @@ -104,24 +152,40 @@ bool IonCellOptimizer::relax_step(const int& istep, IMM.cal_movement(istep, force_step, force, energy, ucell, ofs_running, relax_method, criteria); ++force_step; - - // Check convergence - bool converged = IMM.get_converged(); - if (!converged) + atom_relax_performed = true; + atom_converged = IMM.get_converged(); + + const double max_force_ev = IMM.get_largest_grad() * ModuleBase::Ry_to_eV / ModuleBase::BOHR_TO_A; + max_force_history_.push_back(max_force_ev); + ofs_running << " Largest force is " << max_force_ev + << " eV/Angstrom while threshold is " << inp_->force_thr_ev << " eV/Angstrom" << std::endl; + if (!atom_converged) { + ofs_running << "\n Relaxation is not converged yet!" << std::endl; ucell.ionic_position_updated = true; return false; // not converged } else if (!is_cell_relax) { + print_converged_summary(istep, false, ofs_running); return true; // converged } + // When the lattice is fully fixed (e.g. fixed_axes = abc), there is no + // cell step to run; the atomic convergence above is the final result. + else if (!need_cell_relax) + { + print_converged_summary(istep, false, ofs_running); + return true; // converged, ions relaxed with a fixed lattice + } // Otherwise, continue to cell relaxation } else if (is_relax) { - // Relax mode but no atoms can move - nothing to do - ModuleBase::WARNING("IonCellOptimizer", "No atom is allowed to move!"); + // Relax mode but no atoms can move - nothing to do. The no-op run is + // still a valid converged relaxation, so emit the unified summary to + // give the running log an explicit success marker. + ModuleBase::WARNING("IonCellOptimizer", "No atoms are allowed to move!"); + print_converged_summary(istep, false, ofs_running); return true; } @@ -141,9 +205,18 @@ bool IonCellOptimizer::relax_step(const int& istep, LCM.cal_lattice_change(istep, stress_step, stress, energy, ucell, ofs_running, criteria); bool converged = LCM.get_converged(); - - if (!converged) + + const double max_stress_kbar = LCM.get_largest_grad(); + max_stress_history_.push_back(max_stress_kbar); + ofs_running << " Largest stress is " << max_stress_kbar + << " kbar while threshold is " << inp_->stress_thr << " kbar" << std::endl; + if (converged) + { + print_converged_summary(istep, true, ofs_running); + } + else { + ofs_running << "\n Relaxation is not converged yet!" << std::endl; // Reset force_step counter after cell change for fresh atomic relaxation force_step = 1; stress_step++; diff --git a/source/source_relax/relax_nsync.h b/source/source_relax/relax_nsync.h index d7e73f8852a..fbe3269a03f 100644 --- a/source/source_relax/relax_nsync.h +++ b/source/source_relax/relax_nsync.h @@ -6,6 +6,7 @@ #include "source_cell/unitcell.h" #include "source_io/module_parameter/input_parameter.h" #include +#include /** * @class IonCellOptimizer @@ -59,6 +60,31 @@ class IonCellOptimizer Ions_Move_Methods IMM; ///< Ionic movement methods for atom relaxation Lattice_Change_Methods LCM; ///< Lattice change methods for cell relaxation const Input_para* inp_ = nullptr; + + /// Largest force of each ionic step (eV/Angstrom), kept across cell changes. + std::vector max_force_history_; + /// Largest stress of each cell step (kbar), only filled in cell-relax. + std::vector max_stress_history_; + + /** + * @brief Print the unified convergence summary to the running log. + * + * Shared by every exit path that reports a converged relaxation, so the + * ASE interface always finds the same set of lines (issue #6546). + * + * @param istep Total ionic steps taken. + * @param with_stress Whether the per-step stress history is printed + * (true for cell-relax with a movable lattice). + * @param ofs_running Output stream for the running log. + */ + void print_converged_summary(const int istep, + const bool with_stress, + std::ofstream& ofs_running) const; + + public: + /// Read-only observers of the per-step convergence history, for the final summary. + const std::vector& get_max_force_history() const { return max_force_history_; } + const std::vector& get_max_stress_history() const { return max_stress_history_; } }; #endif \ No newline at end of file diff --git a/source/source_relax/relax_stru_io.cpp b/source/source_relax/relax_stru_io.cpp new file mode 100644 index 00000000000..8d3ec0efe0b --- /dev/null +++ b/source/source_relax/relax_stru_io.cpp @@ -0,0 +1,144 @@ +#include "relax_stru_io.h" + +#include "source_base/formatter.h" +#include "source_cell/cif_io.h" +#include "source_cell/print_cell.h" +#include "source_main/version.h" + +#include + +namespace relax_stru_io +{ + +std::string build_stru_header(const int istep, + const double etot, + const ModuleBase::matrix& stress, + const Input_para& inp, + const bool is_final, + const bool geometry_evaluated) +{ + std::time_t now = std::time(nullptr); + char time_buf[64]; + std::strftime(time_buf, sizeof(time_buf), "%Y-%m-%d %H:%M:%S", std::localtime(&now)); + + const char* final_tag = is_final ? " (FINAL)" : ""; + const double etot_ev = etot * ModuleBase::Ry_to_eV; + // Line 3 is labeled with an energy value only when that energy belongs + // to the geometry written below. When the optimizer proposed the final + // geometry but it was never evaluated, etot is stale, so print N/A to + // keep the energy label and the coordinates on the same frame. + std::string header; + if (geometry_evaluated) + { + header = FmtCore::format("# ABACUS version: %s\n# Written at %s\n# RELAX STEP %d%s, Energy: %.8f eV\n", + VERSION, + time_buf, + istep + 1, + final_tag, + etot_ev); + } + else + { + header = FmtCore::format("# ABACUS version: %s\n# Written at %s\n# RELAX STEP %d%s, Energy: N/A\n", + VERSION, + time_buf, + istep + 1, + final_tag); + } + + // stress in kbar: Ry/Bohr^3 -> kbar, always exactly 3 lines for + // downstream parsers. N/A marks uncomputed stress; no inline comment is + // appended so that every stress line has the same token count. + const double stress_transform = ModuleBase::RYDBERG_SI + / (ModuleBase::BOHR_RADIUS_SI * ModuleBase::BOHR_RADIUS_SI + * ModuleBase::BOHR_RADIUS_SI) + * 1.0e-8; + const bool stress_valid = inp.cal_stress && geometry_evaluated; + if (stress_valid) + { + for (int i = 0; i < 3; i++) + { + header += FmtCore::format("# Stress (kbar): %.6f %.6f %.6f\n", + stress(i, 0) * stress_transform, + stress(i, 1) * stress_transform, + stress(i, 2) * stress_transform); + } + } + else + { + header += "# Stress (kbar): N/A N/A N/A\n" + "# Stress (kbar): N/A N/A N/A\n" + "# Stress (kbar): N/A N/A N/A\n"; + } + + // Line 7: single NOTE line describing the state of stress/forces/geometry. + // Keeping the header at exactly 7 lines makes it easy for downstream + // parsers to skip a fixed-size comment block. + std::string note; + if (!geometry_evaluated) + { + note = "# NOTE: geometry proposed by optimizer but not evaluated; energy N/A; stress N/A; forces omitted"; + } + else if (!inp.cal_stress && !inp.cal_force) + { + note = "# NOTE: stress not computed (cal_stress=0); forces not computed (cal_force=0); per-atom f fields omitted intentionally"; + } + else if (!inp.cal_stress) + { + note = "# NOTE: stress not computed (cal_stress=0)"; + } + else if (!inp.cal_force) + { + note = "# NOTE: forces not computed (cal_force=0); per-atom f fields omitted intentionally"; + } + else + { + note = "# NOTE: stress and forces computed for this geometry"; + } + header += note + "\n"; + + return header; +} + +bool need_orbital(const Input_para& inp) +{ + bool need_orb = inp.basis_type == "pw"; + need_orb = need_orb && inp.init_wfc.substr(0, 3) == "nao"; + need_orb = need_orb || inp.basis_type == "lcao"; + need_orb = need_orb || inp.basis_type == "lcao_in_pw"; + return need_orb; +} + +void write_stru(UnitCell& ucell, + const Input_para& inp, + const std::string& filename, + const std::string& header, + const ModuleBase::matrix& force, + const bool need_orb, + const bool deepks_setorb, + const int my_rank, + const bool has_force) +{ + if (inp.out_stru == 1) + { + unitcell::print_stru_file(ucell, + ucell.atoms, + ucell.latvec, + filename, + header, + inp.nspin, + inp.calculation == "md", + inp.out_mul, + need_orb, + deepks_setorb, + my_rank, + force, + has_force); + } + else if (inp.out_stru == 2) + { + ModuleIO::CifParser::write(filename, ucell, header, "data_?", my_rank); + } +} + +} // namespace relax_stru_io diff --git a/source/source_relax/relax_stru_io.h b/source/source_relax/relax_stru_io.h new file mode 100644 index 00000000000..205da87c76a --- /dev/null +++ b/source/source_relax/relax_stru_io.h @@ -0,0 +1,79 @@ +#ifndef RELAX_STRU_IO_H +#define RELAX_STRU_IO_H + +#include "source_base/matrix.h" +#include "source_cell/unitcell.h" +#include "source_io/module_parameter/input_parameter.h" + +#include + +/** + * @brief Helpers for writing structure (STRU/CIF) files during relaxation. + * + * These free functions extract the file-writing logic shared by + * Relax_Driver::stru_out and Relax_Driver::final_out so that the two + * call sites only orchestrate when and which files to write. + */ +namespace relax_stru_io +{ + /** + * @brief Build the header comment for a structure file. + * + * Contains the ABACUS version, a timestamp, the relaxation step number, + * the total energy in eV, the 3x3 stress tensor in kbar and a trailing + * NOTE line. The header is always exactly 7 lines so that downstream + * parsers can skip a fixed-size comment block. + * + * @param istep Current (zero-based) relaxation step; printed as istep + 1. + * @param etot Total energy in Ry. + * @param stress Stress tensor (3x3) in Ry/Bohr^3. + * @param inp Input parameters (cal_force, cal_stress select the NOTE text). + * @param is_final If true, the step label is marked "(FINAL)". + * @param geometry_evaluated false when the geometry was proposed by the + * optimizer but never evaluated; the energy and stress are then + * marked N/A and the NOTE explains that forces are omitted. + * @return The formatted header string. + */ + std::string build_stru_header(const int istep, + const double etot, + const ModuleBase::matrix& stress, + const Input_para& inp, + const bool is_final, + const bool geometry_evaluated); + + /** + * @brief Whether orbital output is required in the structure file. + * + * True for lcao / lcao_in_pw bases, and for pw with nao-initialized + * wavefunctions. + */ + bool need_orbital(const Input_para& inp); + + /** + * @brief Write one structure file in STRU (out_stru == 1) or CIF + * (out_stru == 2) format. + * + * @param ucell Unit cell to write. + * @param inp Input parameters (nspin, out_mul, calculation). + * @param filename Output file path. + * @param header Header comment built by build_stru_header. + * @param force Atomic forces (used for STRU output when has_force is true). + * @param need_orb Whether to write orbitals (from need_orbital). + * @param deepks_setorb DeePKS setorb flag forwarded to the STRU writer. + * @param my_rank MPI rank of the calling process. + * @param has_force true to emit per-atom f fields; forwarded to + * print_stru_file. Callers must pass false when the forces do not + * belong to the geometry being written (e.g. early relax exit). + */ + void write_stru(UnitCell& ucell, + const Input_para& inp, + const std::string& filename, + const std::string& header, + const ModuleBase::matrix& force, + const bool need_orb, + const bool deepks_setorb, + const int my_rank, + const bool has_force); +} // namespace relax_stru_io + +#endif diff --git a/source/source_relax/relax_sync.cpp b/source/source_relax/relax_sync.cpp index 72e47cbb2cb..b2e79c5a931 100644 --- a/source/source_relax/relax_sync.cpp +++ b/source/source_relax/relax_sync.cpp @@ -1,5 +1,6 @@ #include "relax_sync.h" +#include "relax_history.h" #include "source_base/matrix3.h" #include "source_base/parallel_common.h" @@ -17,6 +18,9 @@ void Relax::init_relax(const int nat_in, const Input_para& inp) inp_ = &inp; + max_force_history_.clear(); + max_stress_history_.clear(); + // set some initial conditions / constants nat = nat_in; istep = 0; @@ -111,22 +115,40 @@ bool Relax::setup_gradient(const UnitCell& ucell, const ModuleBase::matrix& forc // indicating whether force & stress are converged bool force_converged = true; - double max_grad = 0.0; //========================================= // set gradient for ions degrees of freedom //========================================= + const double max_grad = setup_ion_gradient(ucell, force, force_converged, ofs_running); + + //========================================= + // set gradient for cell degrees of freedom + //========================================= + + if (if_cell_moves) + { + setup_cell_gradient(ucell, stress, force_converged, ofs_running); + } + + print_gradient_summary(force_converged, ofs_running); + + return force_converged; +} + +double Relax::setup_ion_gradient(const UnitCell& ucell, const ModuleBase::matrix& force, + bool& force_converged, std::ofstream& ofs_running) +{ grad_ion.zero_out(); ModuleBase::matrix force_eva = force * (ModuleBase::Ry_to_eV / ModuleBase::BOHR_TO_A); // convert to eV/Angstrom + double max_grad = 0.0; int iat = 0; for (int it = 0; it < ucell.ntype; it++) { Atom* atom = &ucell.atoms[it]; for (int ia = 0; ia < ucell.atoms[it].na; ia++) { - double force2 = 0.0; if (atom->mbl[ia].x == 1) { grad_ion(iat, 0) = force_eva(iat, 0); @@ -178,104 +200,117 @@ bool Relax::setup_gradient(const UnitCell& ucell, const ModuleBase::matrix& forc } - ofs_running << "\n Largest force is " << max_grad << + max_force_history_.push_back(max_grad); + ofs_running << "\n Largest force is " << max_grad << " eV/Angstrom while threshold is " << inp_->force_thr_ev << " eV/Angstrom" << std::endl; - //========================================= - // set gradient for cell degrees of freedom - //========================================= + return max_grad; +} - if (if_cell_moves) +void Relax::setup_cell_gradient(const UnitCell& ucell, const ModuleBase::matrix& stress, + bool& force_converged, std::ofstream& ofs_running) +{ + grad_cell.zero_out(); + ModuleBase::matrix stress_ev = stress * (ucell.omega * ModuleBase::Ry_to_eV); + + if (inp_->fixed_axes == "shape") + { + double pressure = (stress_ev(0, 0) + stress_ev(1, 1) + stress_ev(2, 2)) / 3.0; + stress_ev.zero_out(); + stress_ev(0, 0) = pressure; // apply constraints + stress_ev(1, 1) = pressure; + stress_ev(2, 2) = pressure; + } + else if (inp_->fixed_axes == "volume") + { + double pressure = (stress_ev(0, 0) + stress_ev(1, 1) + stress_ev(2, 2)) / 3.0; + stress_ev(0, 0) -= pressure; + stress_ev(1, 1) -= pressure; + stress_ev(2, 2) -= pressure; + } + else if (inp_->fixed_axes != "None") { - grad_cell.zero_out(); - ModuleBase::matrix stress_ev = stress * (ucell.omega * ModuleBase::Ry_to_eV); + // Note stress is given in the directions of lattice vectors + // So we need to first convert to Cartesian and then apply the constraint + ModuleBase::matrix stress_cart = ucell.latvec.to_matrix() * stress_ev; - if (inp_->fixed_axes == "shape") + if (ucell.lat_axis_free[0] == 0) { - double pressure = (stress_ev(0, 0) + stress_ev(1, 1) + stress_ev(2, 2)) / 3.0; - stress_ev.zero_out(); - stress_ev(0, 0) = pressure; // apply constraints - stress_ev(1, 1) = pressure; - stress_ev(2, 2) = pressure; + stress_cart(0, 0) = 0; + stress_cart(0, 1) = 0; + stress_cart(0, 2) = 0; } - else if (inp_->fixed_axes == "volume") + if (ucell.lat_axis_free[1] == 0) { - double pressure = (stress_ev(0, 0) + stress_ev(1, 1) + stress_ev(2, 2)) / 3.0; - stress_ev(0, 0) -= pressure; - stress_ev(1, 1) -= pressure; - stress_ev(2, 2) -= pressure; + stress_cart(1, 0) = 0; + stress_cart(1, 1) = 0; + stress_cart(1, 2) = 0; } - else if (inp_->fixed_axes != "None") + if (ucell.lat_axis_free[2] == 0) { - // Note stress is given in the directions of lattice vectors - // So we need to first convert to Cartesian and then apply the constraint - ModuleBase::matrix stress_cart = ucell.latvec.to_matrix() * stress_ev; - - if (ucell.lat_axis_free[0] == 0) - { - stress_cart(0, 0) = 0; - stress_cart(0, 1) = 0; - stress_cart(0, 2) = 0; - } - if (ucell.lat_axis_free[1] == 0) - { - stress_cart(1, 0) = 0; - stress_cart(1, 1) = 0; - stress_cart(1, 2) = 0; - } - if (ucell.lat_axis_free[2] == 0) - { - stress_cart(2, 0) = 0; - stress_cart(2, 1) = 0; - stress_cart(2, 2) = 0; - } - - stress_ev = ucell.GT.to_matrix() * stress_cart; + stress_cart(2, 0) = 0; + stress_cart(2, 1) = 0; + stress_cart(2, 2) = 0; } - for (int i = 0; i < 3; i++) + stress_ev = ucell.GT.to_matrix() * stress_cart; + } + + for (int i = 0; i < 3; i++) + { + for (int j = 0; j < 3; j++) { - for (int j = 0; j < 3; j++) - { - grad_cell(i, j) = stress_ev(i, j); // apply constraints - } + grad_cell(i, j) = stress_ev(i, j); // apply constraints } + } - double largest_grad = 0.0; - double stress_ii_max = 0.0; + double largest_grad = 0.0; - for (int i = 0; i < 3; i++) + for (int i = 0; i < 3; i++) + { + for (int j = 0; j < 3; j++) { - for (int j = 0; j < 3; j++) + double grad = grad_cell(i, j) / (ucell.omega * ModuleBase::Ry_to_eV); + if (largest_grad < std::abs(grad)) { - double grad = grad_cell(i, j) / (ucell.omega * ModuleBase::Ry_to_eV); - if (largest_grad < std::abs(grad)) - { - largest_grad = std::abs(grad); - } + largest_grad = std::abs(grad); } } + } - double unit_transform = ModuleBase::RYDBERG_SI / pow(ModuleBase::BOHR_RADIUS_SI, 3) * 1.0e-8; - largest_grad = largest_grad * unit_transform; - - if (largest_grad > inp_->stress_thr) - { - force_converged = false; - } + double unit_transform = ModuleBase::RYDBERG_SI / pow(ModuleBase::BOHR_RADIUS_SI, 3) * 1.0e-8; + largest_grad = largest_grad * unit_transform; - ofs_running << " Largest stress is " << largest_grad << " kbar while threshold is " << inp_->stress_thr << " kbar" << std::endl; + if (largest_grad > inp_->stress_thr) + { + force_converged = false; } + max_stress_history_.push_back(largest_grad); + ofs_running << " Largest stress is " << largest_grad << " kbar while threshold is " + << inp_->stress_thr << " kbar" << std::endl; +} + +void Relax::print_gradient_summary(bool force_converged, std::ofstream& ofs_running) +{ if (force_converged) { + // setup_gradient runs before istep is incremented at the end of + // relax_step, so the step that just converged is istep + 1. + const int converged_step = istep + 1; + const std::string method = inp_->relax_method.empty() ? "unknown" : inp_->relax_method[0]; + ofs_running << " Relaxation method: " << method << std::endl; + ofs_running << " Relaxation converged in " << converged_step << " step(s)." << std::endl; + ofs_running << " Largest force per step (eV/Angstrom):" << format_relax_history(max_force_history_); + if (if_cell_moves) + { + ofs_running << " Largest stress per step (kbar):" << format_relax_history(max_stress_history_); + } ofs_running << "\n Relaxation is converged!" << std::endl; } else { ofs_running << "\n Relaxation is not converged yet!" << std::endl; } - - return force_converged; } void Relax::calculate_gamma() @@ -517,77 +552,103 @@ void Relax::move_cell_ions(UnitCell& ucell, const bool is_new_dir, std::ofstream if (if_cell_moves) { - // imo matrix3 class is not a very clever way to store 3*3 matrix ... - ModuleBase::Matrix3 sr_dr_cell; - auto cp_mat_to_mat3 = [&sr_dr_cell, this]() -> void { - sr_dr_cell.e11 = search_dr_cell(0, 0); - sr_dr_cell.e12 = search_dr_cell(0, 1); - sr_dr_cell.e13 = search_dr_cell(0, 2); - sr_dr_cell.e21 = search_dr_cell(1, 0); - sr_dr_cell.e22 = search_dr_cell(1, 1); - sr_dr_cell.e23 = search_dr_cell(1, 2); - sr_dr_cell.e31 = search_dr_cell(2, 0); - sr_dr_cell.e32 = search_dr_cell(2, 1); - sr_dr_cell.e33 = search_dr_cell(2, 2); - }; - cp_mat_to_mat3(); + update_lattice(ucell, fac, is_new_dir); + } - if (ModuleSymmetry::Symmetry::symm_flag && ucell.symm.nrotk > 0) - { - search_dr_cell = sr_dr_cell.Transpose().to_matrix(); - ucell.symm.symmetrize_mat3(search_dr_cell, ucell.lat); - cp_mat_to_mat3(); - sr_dr_cell = sr_dr_cell.Transpose(); - } + // ================================================================= + // Step 2 & 3 : update direct & Cartesian atomic positions + // ================================================================= - // The logic here is as follows: a line search is a continuation - // in the new direction; but ucell.latvec now is already - // different from when the current CG step starts; - // as a result, we need to save latvec at the beginning of - // each CG step - if (is_new_dir) - { - latvec_save = ucell.latvec; - } + update_ion_positions(ucell, fac, ofs_running); - ModuleBase::Matrix3 move_cell = latvec_save * sr_dr_cell; + // ================================================================= + // Step 4 : update G,GT and other stuff + // ================================================================= - // should be close to 0, but set again to avoid numerical issues - if (ucell.lat_axis_free[0] == 0) - { - move_cell.e11 = 0; - move_cell.e12 = 0; - move_cell.e13 = 0; - } - if (ucell.lat_axis_free[1] == 0) - { - move_cell.e21 = 0; - move_cell.e22 = 0; - move_cell.e23 = 0; - } - if (ucell.lat_axis_free[2] == 0) - { - move_cell.e31 = 0; - move_cell.e32 = 0; - move_cell.e33 = 0; - } - ucell.latvec += move_cell * (step_size * fac * fac_stress); + // ================================================================= + // Step 6 : prepare something for next SCF + // ================================================================= + // I have a strong feeling that this part should be + // at the beginning of the next step (namely 'beforescf'), + // but before we have a better organized Esolver + // I do not want to change it + if (if_cell_moves) + { + update_reciprocal_cell(ucell, ofs_running); + } +} - if (inp_->fixed_axes == "volume") - { - double omega_new = std::abs(ucell.latvec.Det()) * pow(ucell.lat0, 3); - ucell.latvec *= pow(ucell.omega / omega_new, 1.0 / 3.0); - } - if (inp_->fixed_ibrav) - { - unitcell::remake_cell(ucell.lat); - } +void Relax::update_lattice(UnitCell& ucell, double fac, bool is_new_dir) +{ + // imo matrix3 class is not a very clever way to store 3*3 matrix ... + ModuleBase::Matrix3 sr_dr_cell; + auto cp_mat_to_mat3 = [&sr_dr_cell, this]() -> void { + sr_dr_cell.e11 = search_dr_cell(0, 0); + sr_dr_cell.e12 = search_dr_cell(0, 1); + sr_dr_cell.e13 = search_dr_cell(0, 2); + sr_dr_cell.e21 = search_dr_cell(1, 0); + sr_dr_cell.e22 = search_dr_cell(1, 1); + sr_dr_cell.e23 = search_dr_cell(1, 2); + sr_dr_cell.e31 = search_dr_cell(2, 0); + sr_dr_cell.e32 = search_dr_cell(2, 1); + sr_dr_cell.e33 = search_dr_cell(2, 2); + }; + cp_mat_to_mat3(); + + if (ModuleSymmetry::Symmetry::symm_flag && ucell.symm.nrotk > 0) + { + search_dr_cell = sr_dr_cell.Transpose().to_matrix(); + ucell.symm.symmetrize_mat3(search_dr_cell, ucell.lat); + cp_mat_to_mat3(); + sr_dr_cell = sr_dr_cell.Transpose(); } - // ================================================================= - // Step 2 & 3 : update direct & Cartesian atomic positions - // ================================================================= + // The logic here is as follows: a line search is a continuation + // in the new direction; but ucell.latvec now is already + // different from when the current CG step starts; + // as a result, we need to save latvec at the beginning of + // each CG step + if (is_new_dir) + { + latvec_save = ucell.latvec; + } + + ModuleBase::Matrix3 move_cell = latvec_save * sr_dr_cell; + + // should be close to 0, but set again to avoid numerical issues + if (ucell.lat_axis_free[0] == 0) + { + move_cell.e11 = 0; + move_cell.e12 = 0; + move_cell.e13 = 0; + } + if (ucell.lat_axis_free[1] == 0) + { + move_cell.e21 = 0; + move_cell.e22 = 0; + move_cell.e23 = 0; + } + if (ucell.lat_axis_free[2] == 0) + { + move_cell.e31 = 0; + move_cell.e32 = 0; + move_cell.e33 = 0; + } + ucell.latvec += move_cell * (step_size * fac * fac_stress); + if (inp_->fixed_axes == "volume") + { + double omega_new = std::abs(ucell.latvec.Det()) * pow(ucell.lat0, 3); + ucell.latvec *= pow(ucell.omega / omega_new, 1.0 / 3.0); + } + if (inp_->fixed_ibrav) + { + unitcell::remake_cell(ucell.lat); + } +} + +void Relax::update_ion_positions(UnitCell& ucell, double fac, std::ofstream& ofs_running) +{ // Calculating displacement in Cartesian coordinate (in Angstrom) std::vector move_ion(nat * 3, 0.0); @@ -631,65 +692,52 @@ void Relax::move_cell_ions(UnitCell& ucell, const bool is_new_dir, std::ofstream // Print the structure file. unitcell::print_tau(ucell.atoms,ucell.Coordinate,ucell.ntype,ucell.lat0,ofs_running); +} - // ================================================================= - // Step 4 : update G,GT and other stuff - // ================================================================= - - if (if_cell_moves) - { - ucell.a1.x = ucell.latvec.e11; - ucell.a1.y = ucell.latvec.e12; - ucell.a1.z = ucell.latvec.e13; - ucell.a2.x = ucell.latvec.e21; - ucell.a2.y = ucell.latvec.e22; - ucell.a2.z = ucell.latvec.e23; - ucell.a3.x = ucell.latvec.e31; - ucell.a3.y = ucell.latvec.e32; - ucell.a3.z = ucell.latvec.e33; +void Relax::update_reciprocal_cell(UnitCell& ucell, std::ofstream& ofs_running) +{ + ucell.a1.x = ucell.latvec.e11; + ucell.a1.y = ucell.latvec.e12; + ucell.a1.z = ucell.latvec.e13; + ucell.a2.x = ucell.latvec.e21; + ucell.a2.y = ucell.latvec.e22; + ucell.a2.z = ucell.latvec.e23; + ucell.a3.x = ucell.latvec.e31; + ucell.a3.y = ucell.latvec.e32; + ucell.a3.z = ucell.latvec.e33; #ifdef __MPI - // distribute lattice vectors. - Parallel_Common::bcast_double(ucell.latvec.e11); - Parallel_Common::bcast_double(ucell.latvec.e12); - Parallel_Common::bcast_double(ucell.latvec.e13); - Parallel_Common::bcast_double(ucell.latvec.e21); - Parallel_Common::bcast_double(ucell.latvec.e22); - Parallel_Common::bcast_double(ucell.latvec.e23); - Parallel_Common::bcast_double(ucell.latvec.e31); - Parallel_Common::bcast_double(ucell.latvec.e32); - Parallel_Common::bcast_double(ucell.latvec.e33); - - // distribute lattice vectors. - Parallel_Common::bcast_double(ucell.a1.x); - Parallel_Common::bcast_double(ucell.a1.y); - Parallel_Common::bcast_double(ucell.a1.z); - Parallel_Common::bcast_double(ucell.a2.x); - Parallel_Common::bcast_double(ucell.a2.y); - Parallel_Common::bcast_double(ucell.a2.z); - Parallel_Common::bcast_double(ucell.a3.x); - Parallel_Common::bcast_double(ucell.a3.y); - Parallel_Common::bcast_double(ucell.a3.z); + // distribute lattice vectors. + Parallel_Common::bcast_double(ucell.latvec.e11); + Parallel_Common::bcast_double(ucell.latvec.e12); + Parallel_Common::bcast_double(ucell.latvec.e13); + Parallel_Common::bcast_double(ucell.latvec.e21); + Parallel_Common::bcast_double(ucell.latvec.e22); + Parallel_Common::bcast_double(ucell.latvec.e23); + Parallel_Common::bcast_double(ucell.latvec.e31); + Parallel_Common::bcast_double(ucell.latvec.e32); + Parallel_Common::bcast_double(ucell.latvec.e33); + + // distribute lattice vectors. + Parallel_Common::bcast_double(ucell.a1.x); + Parallel_Common::bcast_double(ucell.a1.y); + Parallel_Common::bcast_double(ucell.a1.z); + Parallel_Common::bcast_double(ucell.a2.x); + Parallel_Common::bcast_double(ucell.a2.y); + Parallel_Common::bcast_double(ucell.a2.z); + Parallel_Common::bcast_double(ucell.a3.x); + Parallel_Common::bcast_double(ucell.a3.y); + Parallel_Common::bcast_double(ucell.a3.z); #endif - ucell.omega = std::abs(ucell.latvec.Det()) * ucell.lat0 * ucell.lat0 * ucell.lat0; + ucell.omega = std::abs(ucell.latvec.Det()) * ucell.lat0 * ucell.lat0 * ucell.lat0; - ucell.GT = ucell.latvec.Inverse(); - ucell.G = ucell.GT.Transpose(); - ucell.GGT = ucell.G * ucell.GT; - ucell.invGGT = ucell.GGT.Inverse(); - } + ucell.GT = ucell.latvec.Inverse(); + ucell.G = ucell.GT.Transpose(); + ucell.GGT = ucell.G * ucell.GT; + ucell.invGGT = ucell.GGT.Inverse(); - // ================================================================= - // Step 6 : prepare something for next SCF - // ================================================================= - // I have a strong feeling that this part should be - // at the beginning of the next step (namely 'beforescf'), - // but before we have a better organized Esolver - // I do not want to change it - if (if_cell_moves) - { - unitcell::setup_cell_after_vc(ucell, ofs_running, inp_->nspin); - ModuleBase::GlobalFunc::DONE(ofs_running, "SETUP UNITCELL"); - } + // prepare something for next SCF + unitcell::setup_cell_after_vc(ucell, ofs_running, inp_->nspin); + ModuleBase::GlobalFunc::DONE(ofs_running, "SETUP UNITCELL"); } diff --git a/source/source_relax/relax_sync.h b/source/source_relax/relax_sync.h index ef79ec11703..ebb2e6dc3f2 100644 --- a/source/source_relax/relax_sync.h +++ b/source/source_relax/relax_sync.h @@ -9,6 +9,7 @@ #include "source_cell/unitcell.h" #include "source_io/module_parameter/input_parameter.h" #include +#include class Relax { @@ -34,6 +35,26 @@ class Relax // based on threshold in force & stress bool setup_gradient(const UnitCell& ucell, const ModuleBase::matrix& force, const ModuleBase::matrix& stress, std::ofstream& ofs_running); + // Compute the ionic gradient from the force (in eV/Angstrom), honoring + // per-atom move flags, and return the largest component. Records the value + // in the force history and the running log, prints the optional out_level + // diagnostics, and updates force_converged. + double setup_ion_gradient(const UnitCell& ucell, const ModuleBase::matrix& force, + bool& force_converged, std::ofstream& ofs_running); + + // Compute the cell gradient from the stress, applying the fixed_axes + // constraints (shape / volume / per-axis), and update force_converged. + // Records the largest stress in the history and the running log. Only runs + // when if_cell_moves is true. + void setup_cell_gradient(const UnitCell& ucell, const ModuleBase::matrix& stress, + bool& force_converged, std::ofstream& ofs_running); + + // Print the converged / not-converged summary to the running log. On + // convergence, reports the method, the converged step (istep + 1, since + // setup_gradient runs before istep is incremented), and the per-step + // largest force (and stress when the cell moves). + void print_gradient_summary(bool force_converged, std::ofstream& ofs_running); + // check whether previous line search is done bool check_line_search(); @@ -46,6 +67,24 @@ class Relax // move ions and lattice vectors void move_cell_ions(UnitCell& ucell, const bool is_new_dir, std::ofstream& ofs_running); + // Step 1 of move_cell_ions: update latvec along the cell search + // direction, honoring per-axis free flags, the volume constraint and + // fixed_ibrav. Saves latvec at the start of each CG step. Only runs when + // if_cell_moves is true. + void update_lattice(UnitCell& ucell, double fac, bool is_new_dir); + + // Steps 2 & 3 of move_cell_ions: compute the ionic displacement along the + // ion search direction (Cartesian -> direct via the OLD GT), apply the + // per-atom move flags and symmetry, then update taud/tau and print the + // structure. + void update_ion_positions(UnitCell& ucell, double fac, std::ofstream& ofs_running); + + // Steps 4 & 6 of move_cell_ions: refresh a1/a2/a3, omega and the + // reciprocal lattice (G/GT/GGT) from the new latvec, broadcast them under + // MPI, and re-setup the cell for the next SCF. Only runs when + // if_cell_moves is true. + void update_reciprocal_cell(UnitCell& ucell, std::ofstream& ofs_running); + int nat = 0; // number of atoms bool ltrial = false; // if last step is trial step @@ -105,6 +144,16 @@ class Relax ModuleBase::Matrix3 latvec_save; Line_Search ls; const Input_para* inp_ = nullptr; + + /// Largest force of each step (eV/Angstrom), includes trial and line-search steps. + std::vector max_force_history_; + /// Largest stress of each step (kbar), only filled in cell-relax. + std::vector max_stress_history_; + + public: + /// Read-only observers of the per-step convergence history, for the final summary. + const std::vector& get_max_force_history() const { return max_force_history_; } + const std::vector& get_max_stress_history() const { return max_stress_history_; } }; #endif \ No newline at end of file diff --git a/source/source_relax/socket_handlers.cpp b/source/source_relax/socket_handlers.cpp index dc333ced087..94fa504beb7 100644 --- a/source/source_relax/socket_handlers.cpp +++ b/source/source_relax/socket_handlers.cpp @@ -5,6 +5,9 @@ #include "source_cell/unitcell.h" #include "source_esolver/esolver.h" #include "source_io/module_parameter/input_parameter.h" +#ifdef __JSON +#include "source_io/module_json/output_info.h" +#endif #include #include @@ -443,6 +446,10 @@ void handle_posdata(IpiSocket& socket, PosdataPayload payload = read_posdata(socket, *context.ucell, context.state); bcast_posdata(payload); check_posdata_geometry(context, payload, context); +#ifdef __JSON + // Each POSDATA frame is one electronic-structure output step. + Json::init_output_array_obj(); +#endif run_esolver_for_positions(*context.ucell, context.esolver, payload.positions, context.istep); ComputedFrame computed; diff --git a/source/source_relax/test/CMakeLists.txt b/source/source_relax/test/CMakeLists.txt deleted file mode 100644 index 22376932aa3..00000000000 --- a/source/source_relax/test/CMakeLists.txt +++ /dev/null @@ -1,151 +0,0 @@ -abacus_disable_feature_definitions(__MPI) -abacus_disable_feature_definitions(__LCAO) -abacus_disable_feature_definitions(__MLALGO) -abacus_disable_feature_definitions(__CUDA) -abacus_disable_feature_definitions(__ROCM) - -install(DIRECTORY support DESTINATION ${CMAKE_CURRENT_BINARY_DIR}) - - -AddTest( - TARGET MODULE_RELAX_socket_ipi_test - SOURCES test_socket_ipi.cpp ../socket_ipi.cpp -) - -AddTest( - TARGET MODULE_RELAX_socket_frame_test - SOURCES test_socket_frame.cpp ../socket_frame.cpp ../socket_frame_utils.cpp -) - -AddTest( - TARGET MODULE_RELAX_socket_driver_test - LIBS MPI::MPI_CXX base device - KEEP_FEATURE_DEFINITIONS __MPI - SOURCES test_socket_driver.cpp - ../socket_driver.cpp - ../socket_utils.cpp - ../socket_handlers.cpp - ../socket_frame_utils.cpp - ../socket_frame.cpp - ../socket_ipi.cpp - ../../source_cell/update_cell.cpp - ../../source_cell/bcast_cell.cpp -) -set_tests_properties(MODULE_RELAX_socket_driver_test PROPERTIES TIMEOUT 15) - -AddTest( - TARGET MODULE_RELAX_relax_new_line_search - LIBS parameter - SOURCES line_search_test.cpp ../line_search.cpp ../../source_base/global_variable.cpp ../../source_base/global_file.cpp ../../source_base/global_function.cpp ../../source_base/timer.cpp ../../source_base/tool_quit.cpp -) - -AddTest( - TARGET MODULE_RELAX_relax_new_relax - SOURCES relax_test.cpp ../relax_sync.cpp ../line_search.cpp ../relax_data.cpp ../../source_base/tool_quit.cpp ../../source_base/global_variable.cpp ../../source_base/global_file.cpp ../../source_base/timer.cpp - ../../source_base/matrix3.cpp ../../source_base/intarray.cpp ../../source_base/tool_title.cpp - ../../source_base/global_function.cpp ../../source_base/matrix.cpp - ../../source_base/complexarray.cpp ../../source_base/realarray.cpp - ../../source_base/module_external/blas_connector_base.cpp ../../source_base/module_external/blas_connector_vector.cpp ../../source_base/module_external/blas_connector_matrix.cpp - ../../source_cell/update_cell.cpp ../../source_cell/print_cell.cpp ../../source_cell/bcast_cell.cpp ../../source_base/output.cpp - LIBS parameter -) - -list(APPEND cell_source_files - ../../source_cell/update_cell.cpp - ../../source_cell/bcast_cell.cpp - ../../source_cell/print_cell.cpp -) -AddTest( - TARGET MODULE_RELAX_lat_change_method_test - LIBS parameter base device - SOURCES lat_change_method_test.cpp ../lat_change_method.cpp ../lattice_change_basic.cpp ../relax_data.cpp mock_remake_cell.cpp -) - -AddTest( - TARGET MODULE_RELAX_lattice_change_basic_test - LIBS parameter base device - SOURCES lattice_change_basic_test.cpp ../lattice_change_basic.cpp ../relax_data.cpp mock_remake_cell.cpp -) - -AddTest( - TARGET MODULE_RELAX_lattice_change_cg_test - LIBS parameter base device - SOURCES lattice_change_cg_test.cpp - ../lattice_change_cg.cpp - ../lattice_change_basic.cpp - ../cg_base.cpp - ../relax_data.cpp - mock_remake_cell.cpp - -) - -AddTest( - TARGET MODULE_RELAX_bfgs_basic_test - LIBS parameter base device symmetry - SOURCES bfgs_basic_test.cpp ../bfgs_basic.cpp ../relax_data.cpp ../ions_move_basic.cpp - - ${cell_source_files} -) - - -AddTest( - TARGET MODULE_RELAX_bfgs_test - LIBS parameter base device - SOURCES bfgs_test.cpp ../ions_move_bfgs2.cpp ../ions_move_basic.cpp ../matrix_methods.cpp ../relax_data.cpp ${cell_source_files} -) - -AddTest( - TARGET MODULE_RELAX_ions_move_basic_test - LIBS parameter base device - SOURCES ions_move_basic_test.cpp ../ions_move_basic.cpp ../relax_data.cpp ${cell_source_files} -) - -AddTest( - TARGET MODULE_RELAX_ions_move_bfgs_test - LIBS parameter base device - SOURCES ions_move_bfgs_test.cpp - ../ions_move_bfgs.cpp - ../ions_move_basic.cpp - ../bfgs_basic.cpp - ../relax_data.cpp - - ${cell_source_files} -) - -AddTest( - TARGET MODULE_RELAX_ions_move_methods_test - LIBS parameter base device - SOURCES ions_move_methods_test.cpp - ../ions_move_methods.cpp - ../ions_move_bfgs.cpp - ../ions_move_bfgs2.cpp - ../ions_move_cg.cpp - ../ions_move_sd.cpp - ../ions_move_lbfgs.cpp - ../ions_move_basic.cpp - ../bfgs_basic.cpp - ../cg_base.cpp - ../matrix_methods.cpp - ../relax_data.cpp - - ../../source_cell/bcast_cell.cpp - ../../source_cell/print_cell.cpp -) - -AddTest( - TARGET MODULE_RELAX_ions_move_cg_test - LIBS parameter base device - SOURCES ions_move_cg_test.cpp - ../ions_move_cg.cpp - ../cg_base.cpp - ../ions_move_basic.cpp - ../relax_data.cpp - - ${cell_source_files} -) - -AddTest( - TARGET MODULE_RELAX_ions_move_sd_test - LIBS parameter base device - SOURCES ions_move_sd_test.cpp ../ions_move_sd.cpp ../ions_move_basic.cpp ../relax_data.cpp ${cell_source_files} -) diff --git a/source/source_relax/test/bfgs_basic_test.cpp b/source/source_relax/test/bfgs_basic_test.cpp deleted file mode 100644 index 81efd5d1b7a..00000000000 --- a/source/source_relax/test/bfgs_basic_test.cpp +++ /dev/null @@ -1,358 +0,0 @@ -#include "source_relax/ions_move_basic.h" -#include "source_relax/relax_data.h" -#include "gmock/gmock.h" -#include "source_io/module_parameter/parameter.h" -#include "gtest/gtest.h" -#define private public -#define protected public -#include "source_relax/bfgs_basic.h" -#undef private -#undef protected -/************************************************ - * unit tests of class BFGS_Basic - ***********************************************/ - -class BFGSBasicTest : public ::testing::Test -{ - public: - int test_relax_method = 0; - - protected: - void SetUp() override - { - // Initialize variables before each test - } - - void TearDown() override - { - // Clean up after each test - } - - BFGS_Basic bfgs; -}; - -// Test whether the allocate_basic() function can correctly allocate memory space -TEST_F(BFGSBasicTest, TestAllocate) -{ - Ions_Move_Basic::dim = 4; - bfgs.allocate_basic(); - - // Check if allocated vectors are not empty - EXPECT_EQ(bfgs.pos.size(), 4U); - EXPECT_EQ(bfgs.pos_p.size(), 4U); - EXPECT_EQ(bfgs.grad.size(), 4U); - EXPECT_EQ(bfgs.grad_p.size(), 4U); - EXPECT_EQ(bfgs.move.size(), 4U); - EXPECT_EQ(bfgs.move_p.size(), 4U); -} - -// Test if a dimension less than or equal to 0 results in an assertion error -TEST_F(BFGSBasicTest, TestAllocateWithZeroDimension) -{ - Ions_Move_Basic::dim = 0; - ASSERT_DEATH(bfgs.allocate_basic(), ""); -} - -// Test function update_inverse_hessian() assert death -TEST_F(BFGSBasicTest, UpdateInverseHessianDeath) -{ - Ions_Move_Basic::dim = 0; - double lat0 = 1.0; - std::ofstream ofs("test_log_update_inverse_hessian_death.log"); - ASSERT_DEATH(bfgs.update_inverse_hessian(lat0, ofs), ""); - ofs.close(); - std::remove("test_log_update_inverse_hessian_death.log"); -} - -// Test function update_inverse_hessian() when sdoty = 0 -TEST_F(BFGSBasicTest, UpdateInverseHessianCase1) -{ - Ions_Move_Basic::dim = 3; - double lat0 = 1.0; - bfgs.allocate_basic(); - - std::ofstream ofs("test_log_update_inverse_hessian_case1.log"); - bfgs.update_inverse_hessian(lat0, ofs); - ofs.close(); - - std::string expected_output - = " WARINIG: unexpected behaviour in update_inverse_hessian\n Resetting bfgs history \n"; - std::ifstream ifs("test_log_update_inverse_hessian_case1.log"); - std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); - ifs.close(); - std::remove("test_log_update_inverse_hessian_case1.log"); - - EXPECT_EQ(expected_output, output); -} - -// Test function update_inverse_hessian() -TEST_F(BFGSBasicTest, UpdateInverseHessianCase2) -{ - Ions_Move_Basic::dim = 3; - double lat0 = 1.0; - bfgs.allocate_basic(); - bfgs.pos[0] = 2.0; - bfgs.grad[0] = 2.0; - - std::ofstream ofs("test_log_update_inverse_hessian_case2.log"); - bfgs.update_inverse_hessian(lat0, ofs); - ofs.close(); - - EXPECT_DOUBLE_EQ(bfgs.inv_hess(0, 0), 0.5); - EXPECT_DOUBLE_EQ(bfgs.inv_hess(0, 1), 0.0); - EXPECT_DOUBLE_EQ(bfgs.inv_hess(0, 2), 0.0); - EXPECT_DOUBLE_EQ(bfgs.inv_hess(1, 0), 0.0); - EXPECT_DOUBLE_EQ(bfgs.inv_hess(1, 1), 0.0); - EXPECT_DOUBLE_EQ(bfgs.inv_hess(1, 2), 0.0); - EXPECT_DOUBLE_EQ(bfgs.inv_hess(2, 0), 0.0); - EXPECT_DOUBLE_EQ(bfgs.inv_hess(2, 1), 0.0); - EXPECT_DOUBLE_EQ(bfgs.inv_hess(2, 2), 0.0); - std::remove("test_log_update_inverse_hessian_case2.log"); -} - -// Test function check_wolfe_conditions() -TEST_F(BFGSBasicTest, CheckWolfeConditions) -{ - Ions_Move_Basic::dim = 3; - test_relax_method = 1; - bfgs.allocate_basic(); - bfgs.pos[0] = 2.0; - bfgs.grad[0] = 2.0; - bfgs.move[0] = 1.0; - std::vector etot_info = {10.0, 0.0}; - - std::ofstream ofs("test_log_check_wolfe_conditions.log"); - bfgs.check_wolfe_conditions(ofs, etot_info); - ofs.close(); - - std::string expected_output - = " etot - etot_p = 10\n relax_bfgs_w1 * dot_p = -0\n " - " dot = 0\n relax_bfgs_w2 * dot_p = -0\n " - " relax_bfgs_w1 = -1\n relax_bfgs_w2 = -1\n " - "wolfe1 = 0\n wolfe2 = 0\n etot - etot_p = 10\n " - " relax_bfgs_w1 * dot_p = -0\n wolfe1 = 0\n " - " wolfe2 = 0\n wolfe condition satisfied = 0\n"; - - std::ifstream ifs("test_log_check_wolfe_conditions.log"); - std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); - ifs.close(); - std::remove("test_log_check_wolfe_conditions.log"); - - EXPECT_EQ(bfgs.wolfe_flag, false); - EXPECT_EQ(expected_output, output); -} - -// Test function reset_hessian() -TEST_F(BFGSBasicTest, ResetHessian) -{ - Ions_Move_Basic::dim = 3; - bfgs.allocate_basic(); - - bfgs.reset_hessian(); - - EXPECT_DOUBLE_EQ(bfgs.inv_hess(0, 0), 1.0); - EXPECT_DOUBLE_EQ(bfgs.inv_hess(0, 1), 0.0); - EXPECT_DOUBLE_EQ(bfgs.inv_hess(0, 2), 0.0); - EXPECT_DOUBLE_EQ(bfgs.inv_hess(1, 0), 0.0); - EXPECT_DOUBLE_EQ(bfgs.inv_hess(1, 1), 1.0); - EXPECT_DOUBLE_EQ(bfgs.inv_hess(1, 2), 0.0); - EXPECT_DOUBLE_EQ(bfgs.inv_hess(2, 0), 0.0); - EXPECT_DOUBLE_EQ(bfgs.inv_hess(2, 1), 0.0); - EXPECT_DOUBLE_EQ(bfgs.inv_hess(2, 2), 1.0); -} - -// Test function save_bfgs() -TEST_F(BFGSBasicTest, SaveBfgs) -{ - Ions_Move_Basic::dim = 2; - bfgs.save_flag = false; - bfgs.allocate_basic(); - bfgs.pos[0] = 1.0; - bfgs.pos[1] = 2.0; - bfgs.grad[0] = 3.0; - bfgs.grad[1] = 4.0; - bfgs.move[0] = 5.0; - bfgs.move[1] = 6.0; - - bfgs.save_bfgs(); - - EXPECT_EQ(bfgs.save_flag, true); - EXPECT_DOUBLE_EQ(bfgs.pos[0], 1.0); - EXPECT_DOUBLE_EQ(bfgs.pos[1], 2.0); - EXPECT_DOUBLE_EQ(bfgs.grad[0], 3.0); - EXPECT_DOUBLE_EQ(bfgs.grad[1], 4.0); - EXPECT_DOUBLE_EQ(bfgs.move[0], 5.0); - EXPECT_DOUBLE_EQ(bfgs.move[1], 6.0); -} - -// Test function new_step() when update_iter == 1 -TEST_F(BFGSBasicTest, NewStepCase1) -{ - Ions_Move_Basic::dim = 2; - int update_iter = 0; - Ions_Move_Basic::largest_grad = 0.0; - Ions_Move_Basic::relax_bfgs_init = 0.3; - Ions_Move_Basic::best_xxx = -0.4; - bfgs.bfgs_ndim = 1; - bfgs.allocate_basic(); - bfgs.grad[0] = 1.0; - bfgs.grad[1] = 2.0; - bfgs.inv_hess(0, 0) = -3.0; - bfgs.inv_hess(0, 1) = -4.0; - bfgs.inv_hess(1, 0) = -5.0; - bfgs.inv_hess(1, 1) = -6.0; - - double lat0 = 1.0; - std::ofstream ofs("test_log.log"); - std::vector etot_info(2, 0.0); - bfgs.new_step(lat0, update_iter, ofs, etot_info, test_relax_method); - - EXPECT_EQ(update_iter, 1); - EXPECT_EQ(bfgs.tr_min_hit, false); - EXPECT_DOUBLE_EQ(Ions_Move_Basic::relax_bfgs_init, 0.2); - EXPECT_DOUBLE_EQ(Ions_Move_Basic::best_xxx, 0.4); - EXPECT_DOUBLE_EQ(Ions_Move_Basic::trust_radius, 0.2); - EXPECT_DOUBLE_EQ(bfgs.move[0], -1.0); - EXPECT_DOUBLE_EQ(bfgs.move[1], -2.0); - EXPECT_DOUBLE_EQ(bfgs.inv_hess(0, 0), 1.0); - EXPECT_DOUBLE_EQ(bfgs.inv_hess(0, 1), 0.0); - EXPECT_DOUBLE_EQ(bfgs.inv_hess(1, 0), 0.0); - EXPECT_DOUBLE_EQ(bfgs.inv_hess(1, 1), 1.0); -} - -// Test function new_step() when update_iter > 1 -TEST_F(BFGSBasicTest, NewStepCase2) -{ - Ions_Move_Basic::dim = 2; - int update_iter = 2; - Ions_Move_Basic::largest_grad = 0.0; - Ions_Move_Basic::relax_bfgs_init = 0.3; - Ions_Move_Basic::best_xxx = -0.4; - bfgs.bfgs_ndim = 1; - bfgs.allocate_basic(); - bfgs.grad[0] = 1.0; - bfgs.grad[1] = 2.0; - bfgs.inv_hess(0, 0) = -3.0; - bfgs.inv_hess(0, 1) = -4.0; - bfgs.inv_hess(1, 0) = -5.0; - bfgs.inv_hess(1, 1) = -6.0; - - double lat0 = 1.0; - std::ofstream ofs("test_log.log"); - std::vector etot_info(2, 0.0); - bfgs.new_step(lat0, update_iter, ofs, etot_info, test_relax_method); - - EXPECT_EQ(update_iter, 3); - EXPECT_DOUBLE_EQ(Ions_Move_Basic::trust_radius, -1.0); - EXPECT_DOUBLE_EQ(bfgs.move[0], -1.0); - EXPECT_DOUBLE_EQ(bfgs.move[1], -2.0); - EXPECT_DOUBLE_EQ(bfgs.inv_hess(0, 0), 1.0); - EXPECT_DOUBLE_EQ(bfgs.inv_hess(0, 1), 0.0); - EXPECT_DOUBLE_EQ(bfgs.inv_hess(1, 0), 0.0); - EXPECT_DOUBLE_EQ(bfgs.inv_hess(1, 1), 1.0); -} - -// Test function new_step() when bfgs_ndim > 1 -TEST_F(BFGSBasicTest, NewStepWarningQuit) -{ - Ions_Move_Basic::dim = 2; - int update_iter = 0; - bfgs.bfgs_ndim = 2; - bfgs.allocate_basic(); - double lat0 = 1.0; - std::ofstream ofs("test_log.log"); - std::vector etot_info(2, 0.0); - - testing::internal::CaptureStdout(); - EXPECT_EXIT(bfgs.new_step(lat0, update_iter, ofs, etot_info, test_relax_method), ::testing::ExitedWithCode(1), ""); - std::string output = testing::internal::GetCapturedStdout(); - EXPECT_THAT(output, testing::HasSubstr("bfgs_ndim > 1 not implemented yet")); -} - -// Test function compute_trust_radius() case 1 -TEST_F(BFGSBasicTest, ComputeTrustRadiusCase1) -{ - Ions_Move_Basic::dim = 2; - bfgs.allocate_basic(); - bfgs.grad_p[0] = 1.0; - bfgs.move_p[1] = 2.0; - bfgs.inv_hess(0, 0) = -3.0; - bfgs.inv_hess(0, 1) = -4.0; - bfgs.inv_hess(1, 0) = -5.0; - bfgs.inv_hess(1, 1) = -6.0; - bfgs.wolfe_flag = true; - bfgs.relax_bfgs_w1 = 1.0; - std::vector etot_info = {0.0, 0.0, 0.0}; - - std::ofstream ofs("test_log.log"); - bfgs.compute_trust_radius(ofs, etot_info, test_relax_method); - - EXPECT_EQ(bfgs.tr_min_hit, false); - EXPECT_DOUBLE_EQ(Ions_Move_Basic::trust_radius, -1.0); - EXPECT_DOUBLE_EQ(bfgs.move[0], 0.0); - EXPECT_DOUBLE_EQ(bfgs.move[1], 0.0); - EXPECT_DOUBLE_EQ(bfgs.inv_hess(0, 0), -3.0); - EXPECT_DOUBLE_EQ(bfgs.inv_hess(0, 1), -4.0); - EXPECT_DOUBLE_EQ(bfgs.inv_hess(1, 0), -5.0); - EXPECT_DOUBLE_EQ(bfgs.inv_hess(1, 1), -6.0); -} - -// Test function compute_trust_radius() case 2 -TEST_F(BFGSBasicTest, ComputeTrustRadiusCase2) -{ - Ions_Move_Basic::dim = 2; - Ions_Move_Basic::trust_radius_old = 0.0; - Ions_Move_Basic::relax_bfgs_rmin = 100.0; - test_relax_method = 1; - bfgs.allocate_basic(); - bfgs.grad_p[0] = 1.0; - bfgs.move[1] = 2.0; - bfgs.move_p[0] = 2.0; - bfgs.inv_hess(0, 0) = -3.0; - bfgs.inv_hess(0, 1) = -4.0; - bfgs.inv_hess(1, 0) = -5.0; - bfgs.inv_hess(1, 1) = -6.0; - bfgs.wolfe_flag = false; - bfgs.relax_bfgs_w1 = 1.0; - bfgs.tr_min_hit = false; - std::vector etot_info = {0.0, 0.0, 0.0}; - - std::ofstream ofs("test_log.log"); - bfgs.compute_trust_radius(ofs, etot_info, test_relax_method); - - EXPECT_EQ(bfgs.tr_min_hit, true); - EXPECT_DOUBLE_EQ(Ions_Move_Basic::trust_radius, 100.0); - EXPECT_DOUBLE_EQ(bfgs.move[0], 0.0); - EXPECT_DOUBLE_EQ(bfgs.move[1], 0.0); - EXPECT_DOUBLE_EQ(bfgs.inv_hess(0, 0), 1.0); - EXPECT_DOUBLE_EQ(bfgs.inv_hess(0, 1), 0.0); - EXPECT_DOUBLE_EQ(bfgs.inv_hess(1, 0), 0.0); - EXPECT_DOUBLE_EQ(bfgs.inv_hess(1, 1), 1.0); -} - -// Test function compute_trust_radius() warning_quit -TEST_F(BFGSBasicTest, ComputeTrustRadiusWarningQuit) -{ - Ions_Move_Basic::dim = 2; - Ions_Move_Basic::trust_radius_old = 0.0; - Ions_Move_Basic::relax_bfgs_rmin = 100.0; - test_relax_method = 1; - bfgs.allocate_basic(); - bfgs.grad_p[0] = 1.0; - bfgs.move[1] = 2.0; - bfgs.move_p[0] = 2.0; - bfgs.inv_hess(0, 0) = -3.0; - bfgs.inv_hess(0, 1) = -4.0; - bfgs.inv_hess(1, 0) = -5.0; - bfgs.inv_hess(1, 1) = -6.0; - bfgs.wolfe_flag = false; - bfgs.relax_bfgs_w1 = 1.0; - bfgs.tr_min_hit = true; - std::vector etot_info = {0.0, 0.0, 0.0}; - - std::ofstream ofs("test_log.log"); - testing::internal::CaptureStdout(); - EXPECT_EXIT(bfgs.compute_trust_radius(ofs, etot_info, test_relax_method), ::testing::ExitedWithCode(1), ""); - std::string output = testing::internal::GetCapturedStdout(); - EXPECT_THAT(output, testing::HasSubstr("bfgs history already reset at previous step, we got trapped!")); -} \ No newline at end of file diff --git a/source/source_relax/test/bfgs_test.cpp b/source/source_relax/test/bfgs_test.cpp deleted file mode 100644 index 79003212858..00000000000 --- a/source/source_relax/test/bfgs_test.cpp +++ /dev/null @@ -1,379 +0,0 @@ -// source/source_relax/test/bfgs_test.cpp -#include "gtest/gtest.h" -#include "gmock/gmock.h" -#include "for_test.h" - -#include "source_relax/ions_move_bfgs2.h" - -#include "source_io/module_parameter/parameter.h" - -#include "source_relax/ions_move_basic.h" // for Ions_Move_Basic static members -#include "source_relax/relax_data.h" -#include - -/************************************************ - * unit tests for Ions_Move_BFGS2 (no MockUnitCell) - ***********************************************/ - -class BFGSTest : public ::testing::Test { -protected: - Ions_Move_BFGS2 bfgs; - void SetUp() override - { - // Initialize variables before each test - } - - void TearDown() override - { - // nothing global to clean here - } - - // Friendship is not inherited: a TEST_F body lives in a class derived from - // this fixture, so the internals of Ions_Move_BFGS2 (which friends the - // fixture itself) are reached through these forwarders. They all carry a - // get_ / call_ prefix because several tests declare locals named pos, - // force, size, steplength, dpos and maxstep. - const std::vector>& get_H() const - { - return bfgs.H; - } - const std::vector>& get_pos() const - { - return bfgs.pos; - } - const std::vector& get_pos0() const - { - return bfgs.pos0; - } - const std::vector>& get_pos_taud() const - { - return bfgs.pos_taud; - } - const std::vector& get_pos_taud0() const - { - return bfgs.pos_taud0; - } - const std::vector>& get_force() const - { - return bfgs.force; - } - const std::vector& get_force0() const - { - return bfgs.force0; - } - const std::vector& get_steplength() const - { - return bfgs.steplength; - } - const std::vector>& get_dpos() const - { - return bfgs.dpos; - } - int get_size() const - { - return bfgs.size; - } - double get_alpha() const - { - return bfgs.alpha; - } - double get_maxstep() const - { - return bfgs.maxstep; - } - bool get_sign() const - { - return bfgs.sign; - } - double get_largest_grad() const - { - return bfgs.largest_grad; - } - bool get_is_initialized() const - { - return bfgs.is_initialized; - } - - void set_sign(const bool value) - { - bfgs.sign = value; - } - void set_is_initialized(const bool value) - { - bfgs.is_initialized = value; - } - void set_force0(const std::vector& value) - { - bfgs.force0 = value; - } - void set_pos_taud_x(const int i, const double value) - { - bfgs.pos_taud[i].x = value; - } - - void call_DetermineStep(std::vector& steplength, - std::vector>& dpos, - double& maxstep) - { - bfgs.DetermineStep(steplength, dpos, maxstep); - } - void call_Update(std::vector& pos, std::vector& force, UnitCell& ucell) - { - bfgs.Update(pos, force, bfgs.H, ucell); - } - void call_GetPos(UnitCell& ucell) - { - bfgs.GetPos(ucell, bfgs.pos); - } - void call_GetPostaud(UnitCell& ucell) - { - bfgs.GetPostaud(ucell, bfgs.pos_taud); - } - void call_CalculateLargestGrad(const ModuleBase::matrix& force, UnitCell& ucell) - { - bfgs.CalculateLargestGrad(force, ucell); - } -}; - -// Test whether the allocate() function can correctly allocate memory space -TEST_F(BFGSTest, TestAllocate) -{ - int size = 2; - bfgs.allocate(size); - - // Check if allocated arrays are not empty - EXPECT_FALSE(get_H().empty()); - EXPECT_FALSE(get_pos().empty()); - EXPECT_FALSE(get_pos0().empty()); - EXPECT_FALSE(get_pos_taud().empty()); - EXPECT_FALSE(get_pos_taud0().empty()); - EXPECT_FALSE(get_force().empty()); - EXPECT_FALSE(get_force0().empty()); - EXPECT_FALSE(get_steplength().empty()); - EXPECT_FALSE(get_dpos().empty()); - EXPECT_EQ(get_size(), size); - EXPECT_EQ(get_alpha(), 70); - EXPECT_EQ(get_maxstep(), PARAM.inp.relax_bfgs_rmax); - EXPECT_TRUE(get_sign()); - EXPECT_EQ(get_largest_grad(), 0.0); -} - -// Test that relax_step will auto-initialize if not already initialized -TEST_F(BFGSTest, RelaxStepAutoInitialize) -{ - set_is_initialized(false); - - UnitCell ucell; - ucell.nat = 2; - ucell.ntype = 1; - ucell.atoms = new Atom[ucell.ntype]; - ucell.atoms[0].na = 2; - ucell.atoms[0].tau = std::vector>(2); - ucell.atoms[0].taud = std::vector>(2); - ucell.atoms[0].mbl = std::vector>(2, {1, 1, 1}); - ucell.atoms[0].tau[0].x = 0.0; ucell.atoms[0].tau[0].y = 0.0; ucell.atoms[0].tau[0].z = 0.0; - ucell.atoms[0].tau[1].x = 1.0; ucell.atoms[0].tau[1].y = 0.0; ucell.atoms[0].tau[1].z = 0.0; - ucell.lat0 = 1.0; - - ModuleBase::matrix force(2, 3); - force(0, 0) = 0.1; force(0, 1) = 0.0; force(0, 2) = 0.0; - force(1, 0) = -0.1; force(1, 1) = 0.0; force(1, 2) = 0.0; - - std::ofstream ofs_running; - - // Before relax_step, is_initialized should be false - EXPECT_FALSE(get_is_initialized()); - bfgs.relax_step(force, ucell, ofs_running); - // After relax_step, is_initialized should be true - EXPECT_TRUE(get_is_initialized()); -} - -// Test if a dimension less than or equal to 0 results in an assertion error -TEST_F(BFGSTest, TestAllocateWithZeroDimension) -{ - int size = 0; - ASSERT_DEATH(bfgs.allocate(size), ""); -} - -// Test DetermineStep scaling -TEST_F(BFGSTest, DetermineStepScaling) -{ - int size = 2; - bfgs.allocate(size); - - std::vector steplength = {1.0, 0.1}; - std::vector> dpos = { - ModuleBase::Vector3(1.0, 1.0, 1.0), - ModuleBase::Vector3(0.1, 0.1, 0.1) - }; - double maxstep = 0.5; - call_DetermineStep(steplength, dpos, maxstep); - - // first atom scaled down to maxstep - EXPECT_NEAR(dpos[0][0], 0.5, 1e-12); - EXPECT_NEAR(dpos[0][1], 0.5, 1e-12); - EXPECT_NEAR(dpos[0][2], 0.5, 1e-12); - - // second atom unchanged (small) - EXPECT_NEAR(dpos[1][0], 0.05, 1e-12); - EXPECT_NEAR(dpos[1][1], 0.05, 1e-12); - EXPECT_NEAR(dpos[1][2], 0.05, 1e-12); -} - -TEST_F(BFGSTest, UpdateUsesAbsoluteDisplacementThreshold) -{ - UnitCell ucell; - ucell.ntype = 1; - ucell.nat = 1; - ucell.lat0 = 1.0; - ucell.latvec.Identity(); - ucell.atoms = new Atom[ucell.ntype]; - ucell.atoms[0].na = 1; - ucell.atoms[0].mbl = std::vector>(1, {1, 1, 1}); - ucell.iat2it = new int[ucell.nat]; - ucell.iat2ia = new int[ucell.nat]; - ucell.iat2it[0] = 0; - ucell.iat2ia[0] = 0; - - bfgs.allocate(ucell.nat); - set_sign(false); - set_pos_taud_x(0, -1.0e-6); - set_force0({0.0, 0.0, 0.0}); - - std::vector pos = {0.0, 0.0, 0.0}; - std::vector force = {1.0, 0.0, 0.0}; - const double initial_h00 = get_H()[0][0]; - - call_Update(pos, force, ucell); - - EXPECT_NE(get_H()[0][0], initial_h00); -} - -// Test GetPos and GetPostaud without creating extra helper class -TEST_F(BFGSTest, GetPosAndPostaud) -{ - // prepare UnitCell with 1 type and 2 atoms - UnitCell ucell; - ucell.ntype = 1; - ucell.nat = 2; - ucell.lat0 = 2.0; - - // allocate atoms array - ucell.atoms = new Atom[ucell.ntype]; - ucell.atoms[0].na = 2; - ucell.atoms[0].tau = std::vector>(2); - ucell.atoms[0].taud = std::vector>(2); - ucell.atoms[0].mbl = std::vector>(2, {1, 1, 1}); - - // set coordinates - ucell.atoms[0].tau[0].x = 0.0; ucell.atoms[0].tau[0].y = 0.0; ucell.atoms[0].tau[0].z = 0.0; - ucell.atoms[0].tau[1].x = 1.0; ucell.atoms[0].tau[1].y = 0.0; ucell.atoms[0].tau[1].z = 0.0; - - // allocate mapping arrays - ucell.iat2it = new int[ucell.nat]; - ucell.iat2ia = new int[ucell.nat]; - int k = 0; - for (int it = 0; it < ucell.ntype; ++it) { - for (int ia = 0; ia < ucell.atoms[it].na; ++ia) { - ucell.iat2it[k] = it; - ucell.iat2ia[k] = ia; - ++k; - } - } - - // allocate bfgs arrays and call getters - bfgs.allocate(ucell.nat); - call_GetPos(ucell); - call_GetPostaud(ucell); - - // pos is tau * BOHR_TO_A * lat0 - EXPECT_DOUBLE_EQ(get_pos()[0][0], ucell.atoms[0].tau[0].x * ModuleBase::BOHR_TO_A * ucell.lat0); - EXPECT_DOUBLE_EQ(get_pos_taud()[1][2], ucell.atoms[0].taud[1].z); -} - -// Test CalculateLargestGrad (uses ModuleBase::matrix) -TEST_F(BFGSTest, CalculateLargestGrad) -{ - // UnitCell with 1 type and 2 atoms - UnitCell ucell; - ucell.ntype = 1; - ucell.nat = 2; - ucell.lat0 = 2.0; - - ucell.atoms = new Atom[ucell.ntype]; - ucell.atoms[0].na = 2; - ucell.atoms[0].mbl = std::vector>(2, {1, 1, 1}); - - // mapping arrays - ucell.iat2it = new int[ucell.nat]; - ucell.iat2ia = new int[ucell.nat]; - int k = 0; - for (int it = 0; it < ucell.ntype; ++it) { - for (int ia = 0; ia < ucell.atoms[it].na; ++ia) { - ucell.iat2it[k] = it; - ucell.iat2ia[k] = ia; - ++k; - } - } - - // build force matrix: 2 atoms x 3 components - ModuleBase::matrix force(2, 3); - force(0, 0) = -2.0; // this yields grad component = -(-2.0)*lat0 = 4.0 -> divided by lat0 => 2.0 - force(0, 1) = 0.0; - force(0, 2) = 1.0; - force(1, 0) = 3.0; // this yields abs = 6.0 -> divided by lat0 => 3.0 (this should be largest) - force(1, 1) = -1.0; - force(1, 2) = 0.0; - - bfgs.allocate(ucell.nat); - call_CalculateLargestGrad(force, ucell); - - // expected largest_grad = 3.0 (see calculation above) - EXPECT_NEAR(get_largest_grad(), 6.0, 1e-12); -} - -// Test relax_step basic functionality -TEST_F(BFGSTest, RelaxStepBasic) -{ - // Setup UnitCell with 1 type, 2 atoms - UnitCell ucell; - ucell.ntype = 1; - ucell.nat = 2; - ucell.lat0 = 1.0; - ucell.atoms = new Atom[ucell.ntype]; - ucell.atoms[0].na = 2; - ucell.atoms[0].tau = std::vector>(2); - ucell.atoms[0].taud = std::vector>(2); - ucell.atoms[0].mbl = std::vector>(2, {1, 1, 1}); - ucell.iat2it = new int[ucell.nat]; - ucell.iat2ia = new int[ucell.nat]; - int k = 0; - for (int it = 0; it < ucell.ntype; ++it) { - for (int ia = 0; ia < ucell.atoms[it].na; ++ia) { - ucell.iat2it[k] = it; - ucell.iat2ia[k] = ia; - ++k; - } - } - // Set initial positions - ucell.atoms[0].tau[0].x = 0.0; ucell.atoms[0].tau[0].y = 0.0; ucell.atoms[0].tau[0].z = 0.0; - ucell.atoms[0].tau[1].x = 1.0; ucell.atoms[0].tau[1].y = 0.0; ucell.atoms[0].tau[1].z = 0.0; - // Setup force matrix - ModuleBase::matrix force(2, 3); - force(0, 0) = 0.1; force(0, 1) = 0.0; force(0, 2) = 0.0; - force(1, 0) = -0.1; force(1, 1) = 0.0; force(1, 2) = 0.0; - // Allocate and call relax_step - bfgs.allocate(ucell.nat); - - std::ofstream ofs_running; - - bfgs.relax_step(force, ucell, ofs_running); - // Check that ionic_position_updated is true - EXPECT_TRUE(ucell.ionic_position_updated); - // Check that force values are set (converted units) - EXPECT_NEAR(get_force()[0][0], 0.1 * ModuleBase::Ry_to_eV / ModuleBase::BOHR_TO_A, 1e-12); - EXPECT_NEAR(get_force()[1][0], -0.1 * ModuleBase::Ry_to_eV / ModuleBase::BOHR_TO_A, 1e-12); - // Check that positions are updated (not equal to initial) - EXPECT_NEAR(get_pos()[0][0], 0.0, 1e-12); - EXPECT_NE(get_pos()[1][0], 1.0); -} diff --git a/source/source_relax/test/ions_move_basic_test.cpp b/source/source_relax/test/ions_move_basic_test.cpp deleted file mode 100644 index 75971752683..00000000000 --- a/source/source_relax/test/ions_move_basic_test.cpp +++ /dev/null @@ -1,301 +0,0 @@ -#include "gmock/gmock.h" -#include "gtest/gtest.h" -#include "source_relax/ions_move_basic.h" -#include "source_relax/relax_data.h" -#include "for_test.h" - -/************************************************ - * unit tests of namespace Ions_Move_Basic - ***********************************************/ - -// Define a fixture for the tests -class IonsMoveBasicTest : public ::testing::Test -{ - protected: - UnitCell ucell; - double move[6], pos[6], grad[6]; - ModuleBase::matrix force; - - virtual void SetUp() - { - // Initialize variables before each test - force.create(ucell.nat, 3); - force(0, 0) = 1.0; - force(0, 1) = 2.0; - force(0, 2) = 3.0; - force(1, 0) = 4.0; - force(1, 1) = 5.0; - force(1, 2) = 6.0; - } - - virtual void TearDown() - { - // Clean up after each test - } -}; - -// Test the setup_gradient() function -TEST_F(IonsMoveBasicTest, SetupGradient) -{ - // Call the function being tested - Ions_Move_Basic::dim = 6; - std::ofstream ofs("test_setup_gradient.log"); - Ions_Move_Basic::setup_gradient(ucell, force, pos, grad, ofs); - ofs.close(); - std::remove("test_setup_gradient.log"); - - // Check that the expected positions and gradients were generated - EXPECT_DOUBLE_EQ(pos[0], 0.0); - EXPECT_DOUBLE_EQ(pos[1], 10.0); - EXPECT_DOUBLE_EQ(pos[2], 20.0); - EXPECT_DOUBLE_EQ(pos[3], 30.0); - EXPECT_DOUBLE_EQ(pos[4], 40.0); - EXPECT_DOUBLE_EQ(pos[5], 50.0); - EXPECT_DOUBLE_EQ(grad[0], -10.0); - EXPECT_DOUBLE_EQ(grad[1], -20.0); - EXPECT_DOUBLE_EQ(grad[2], -30.0); - EXPECT_DOUBLE_EQ(grad[3], -40.0); - EXPECT_DOUBLE_EQ(grad[4], -50.0); - EXPECT_DOUBLE_EQ(grad[5], -60.0); -} - -// Test the move_atoms() function -TEST_F(IonsMoveBasicTest, MoveAtoms) -{ - // Initialize data - Ions_Move_Basic::dim = 6; - const int test_relax_method = 1; - for (int i = 0; i < Ions_Move_Basic::dim; ++i) - { - pos[i] = 0.0; - move[i] = i; - } - - // Call the function being tested - std::ofstream ofs("test_move_atoms.log"); - Ions_Move_Basic::move_atoms(ucell, move, pos, ofs, test_relax_method); - ofs.close(); - - // Check the results - std::ifstream ifs("test_move_atoms.log"); - std::string expected_output = "\n movement of ions (unit is Bohr) : \n Atom x y " - " z\n move_1 0 1 2\n " - "move_2 3 4 5\n"; - std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); - ifs.close(); - std::remove("test_move_atoms.log"); - - EXPECT_THAT(output , ::testing::HasSubstr(expected_output)); - EXPECT_DOUBLE_EQ(pos[0], 0.0); - EXPECT_DOUBLE_EQ(pos[1], 1.0); - EXPECT_DOUBLE_EQ(pos[2], 2.0); - EXPECT_DOUBLE_EQ(pos[3], 3.0); - EXPECT_DOUBLE_EQ(pos[4], 4.0); - EXPECT_DOUBLE_EQ(pos[5], 5.0); -} - -// Test the check_converged() function case 1 -TEST_F(IonsMoveBasicTest, CheckConvergedCase1) -{ - // Initialize data - Ions_Move_Basic::dim = 6; - int update_iter = 1; - const int test_relax_method = 1; - const std::string out_level = "ie"; - const double force_thr = -1; // Input_para default; this test never set it - const double force_thr_ev = -1; // Input_para default; this test never set it - std::vector etot_info(2, 0.0); - for (int i = 0; i < Ions_Move_Basic::dim; ++i) - { - grad[i] = 0.0; - } - - // Call the function being tested - std::ofstream ofs("test_check_converged_case1.log"); - testing::internal::CaptureStdout(); - bool converged = Ions_Move_Basic::check_converged(ucell, grad, update_iter, ofs, etot_info, force_thr, force_thr_ev, out_level, test_relax_method); - std::string std_outout = testing::internal::GetCapturedStdout(); - ofs.close(); - - // Check the results - std::ifstream ifs("test_check_converged_case1.log"); - std::string ofs_output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); - ifs.close(); - std::remove("test_check_converged_case1.log"); - - std::string expected_ofs - = " old total energy (ry) = 0\n new total energy (ry) = 0\n " - " energy difference (ry) = 0\n largest gradient (ry/bohr) = 0\n\n" - " Largest force is 0 eV/Angstrom while threshold is -1 eV/Angstrom\n largest force is 0, no " - "movement is possible.\n it may converged, otherwise no movement of atom is allowed.\n"; - std::string expected_std = " ETOT DIFF (eV) : 0\n LARGEST GRAD (eV/Angstrom) : 0\n"; - - EXPECT_THAT(ofs_output , ::testing::HasSubstr(expected_ofs)); - EXPECT_EQ(expected_std, std_outout); - EXPECT_EQ(update_iter, 1); - EXPECT_EQ(converged, true); - EXPECT_DOUBLE_EQ(Ions_Move_Basic::largest_grad, 0.0); -} - -// Test the check_converged() function case 2 -TEST_F(IonsMoveBasicTest, CheckConvergedCase2) -{ - // Initialize data - Ions_Move_Basic::dim = 6; - int update_iter = 1; - std::vector etot_info(2, 0.0); - const int test_relax_method = 1; - const std::string out_level = "ie"; - const double force_thr = 1.0; - const double force_thr_ev = -1; // Input_para default; this test never set it - grad[0] = 1.0; - - // Call the function being tested - std::ofstream ofs("test_check_converged_case2.log"); - testing::internal::CaptureStdout(); - bool converged = Ions_Move_Basic::check_converged(ucell, grad, update_iter, ofs, etot_info, force_thr, force_thr_ev, out_level, test_relax_method); - std::string std_outout = testing::internal::GetCapturedStdout(); - ofs.close(); - - // Check the results - std::ifstream ifs("test_check_converged_case2.log"); - std::string ofs_output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); - ifs.close(); - std::remove("test_check_converged_case2.log"); - - std::string expected_ofs - = " old total energy (ry) = 0\n new total energy (ry) = 0\n " - " energy difference (ry) = 0\n largest gradient (ry/bohr) = 0.1\n\n" - " Largest force is 2.57111 eV/Angstrom while threshold is -1 eV/Angstrom\n\n Ion relaxation is " - "converged!\n\n Energy difference (Ry) = 0\n"; - std::string expected_std = " ETOT DIFF (eV) : 0\n LARGEST GRAD (eV/Angstrom) : 2.57111\n"; - - EXPECT_THAT(ofs_output , ::testing::HasSubstr(expected_ofs)); - EXPECT_EQ(expected_std, std_outout); - EXPECT_EQ(update_iter, 2); - EXPECT_EQ(converged, true); - EXPECT_DOUBLE_EQ(Ions_Move_Basic::largest_grad, 0.1); -} - -// Test the check_converged() function case 3 -TEST_F(IonsMoveBasicTest, CheckConvergedCase3) -{ - // Initialize data - Ions_Move_Basic::dim = 6; - int update_iter = 1; - std::vector etot_info = {1.0, 0.0}; - const int test_relax_method = 1; - const std::string out_level = "ie"; - const double force_thr = 1.0; - const double force_thr_ev = -1; // Input_para default; this test never set it - grad[0] = 1.0; - - // Call the function being tested - std::ofstream ofs("test_check_converged_case3.log"); - testing::internal::CaptureStdout(); - bool converged = Ions_Move_Basic::check_converged(ucell, grad, update_iter, ofs, etot_info, force_thr, force_thr_ev, out_level, test_relax_method); - std::string std_outout = testing::internal::GetCapturedStdout(); - ofs.close(); - - // Check the results - std::ifstream ifs("test_check_converged_case3.log"); - std::string ofs_output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); - ifs.close(); - std::remove("test_check_converged_case3.log"); - - std::string expected_ofs - = " old total energy (ry) = 0\n new total energy (ry) = 1\n " - " energy difference (ry) = 1\n largest gradient (ry/bohr) = 0.1\n\n" - " Largest force is 2.57111 eV/Angstrom while threshold is -1 eV/Angstrom\n\n Ion relaxation is not " - "converged yet (threshold is 25.7111)\n"; - std::string expected_std = " ETOT DIFF (eV) : 13.6057\n LARGEST GRAD (eV/Angstrom) : 2.57111\n"; - - EXPECT_THAT(ofs_output , ::testing::HasSubstr(expected_ofs)); - EXPECT_EQ(expected_std, std_outout); - EXPECT_EQ(update_iter, 1); - EXPECT_EQ(converged, false); - EXPECT_DOUBLE_EQ(Ions_Move_Basic::largest_grad, 0.1); -} - -// Test the terminate() function when converged -TEST_F(IonsMoveBasicTest, TerminateConverged) -{ - // Initialize data - const bool converged = true; - const int istep = 2; - const int update_iter = 5; - - // Call the function being tested - std::ofstream ofs("test_terminate_converged.log"); - Ions_Move_Basic::terminate(converged, update_iter, ucell, istep, ofs); - ofs.close(); - - // Check the results - std::ifstream ifs("test_terminate_converged.log"); - std::string ofs_output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); - ifs.close(); - std::remove("test_terminate_converged.log"); - - std::string expected_ofs = " end of geometry optimization\n istep = 2\n " - " update iteration = 5\n"; - - EXPECT_THAT(ofs_output , ::testing::HasSubstr(expected_ofs)); -} - -// Test the terminate() function when not converged -TEST_F(IonsMoveBasicTest, TerminateNotConverged) -{ - // Initialize data - const bool converged = false; - const int istep = 10; - const int update_iter = 0; - - // Call the function being tested - std::ofstream ofs("test_terminate_not_converged.log"); - Ions_Move_Basic::terminate(converged, update_iter, ucell, istep, ofs); - ofs.close(); - - // Check the results - std::ifstream ifs("test_terminate_not_converged.log"); - std::string ofs_output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); - ifs.close(); - std::remove("test_terminate_not_converged.log"); - - std::string expected_ofs = " the maximum number of steps has been reached.\n end of geometry optimization.\n"; - - EXPECT_THAT(ofs_output , ::testing::HasSubstr(expected_ofs)); -} - -// Test the setup_etot() function -TEST_F(IonsMoveBasicTest, SetupEtot) -{ - // Initialize data - const int istep = 2; - std::vector etot_info = {2.0, 1.0}; - double energy_in = 3.0; - - // Call the function being tested - std::ofstream ofs("/dev/null"); - Ions_Move_Basic::setup_etot(energy_in, istep, etot_info); - ofs.close(); - - // Check the results - EXPECT_DOUBLE_EQ(etot_info[1], 2.0); - EXPECT_DOUBLE_EQ(etot_info[0], 3.0); - EXPECT_DOUBLE_EQ(etot_info[0] - etot_info[1], 1.0); -} - -// Test the dot_func() function -TEST_F(IonsMoveBasicTest, DotFunc) -{ - // Initialize data - double dim_in = 3; - double a[3] = {1.0, 2.0, 3.0}; - double b[3] = {1.0, 2.0, 3.0}; - - // Call the function being tested - double result = Ions_Move_Basic::dot_func(a, b, dim_in); - - // Check the results - EXPECT_DOUBLE_EQ(result, 14.0); -} diff --git a/source/source_relax/test/ions_move_bfgs_test.cpp b/source/source_relax/test/ions_move_bfgs_test.cpp deleted file mode 100644 index 1726b97adcb..00000000000 --- a/source/source_relax/test/ions_move_bfgs_test.cpp +++ /dev/null @@ -1,465 +0,0 @@ -#include "source_relax/relax_criteria.h" -#include "for_test.h" -#include "gtest/gtest.h" -#include "gmock/gmock.h" -#define private public -#define protected public -#include "source_io/module_parameter/parameter.h" -#include "source_relax/ions_move_basic.h" -#include "source_relax/ions_move_bfgs.h" -#undef private -#undef protected - -/************************************************ - * unit tests of class Ions_Move_BFGS - ***********************************************/ - -// Define a fixture for the tests -class IonsMoveBFGSTest : public ::testing::Test -{ - public: - Relax_Criteria criteria; - - protected: - Ions_Move_BFGS bfgs; - int update_iter; - - virtual void SetUp() - { - // Initialize variables before each test - Ions_Move_Basic::dim = 6; - update_iter = 0; - } - - virtual void TearDown() - { - // Clean up after each test - } -}; - -// Test the allocate() function case 1 -TEST_F(IonsMoveBFGSTest, AllocateCase1) -{ - // Initilize data - bfgs.init_done = true; - bfgs.save_flag = true; - - // Call the function being tested - bfgs.allocate(); - - // Check that the expected results - EXPECT_EQ(bfgs.init_done, true); - EXPECT_EQ(bfgs.save_flag, true); -} - -// Test the allocate() function case 2 -TEST_F(IonsMoveBFGSTest, AllocateCase2) -{ - // Initilize data - bfgs.init_done = false; - bfgs.save_flag = true; - - // Call the function being tested - bfgs.allocate(); - - // Check that the expected results - EXPECT_EQ(bfgs.init_done, true); - EXPECT_EQ(bfgs.save_flag, false); -} - -// Test the start() function case 1 -TEST_F(IonsMoveBFGSTest, StartCase1) -{ - // Initilize data - UnitCell ucell; - ModuleBase::matrix force(2, 3); - double energy_in = 0.0; - const int istep = 1; - bfgs.init_done = false; - bfgs.save_flag = true; - std::vector etot_info(2, 0.0); - - // Call the function being tested - bfgs.allocate(); - std::ofstream ofs("test_start_case1.log"); - bfgs.start(ucell, force, energy_in, istep, update_iter, ofs, etot_info, criteria); - ofs.close(); - - // Check the results - std::ifstream ifs("test_start_case1.log"); - std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); - ifs.close(); - std::remove("test_start_case1.log"); - - EXPECT_THAT(output, testing::HasSubstr("update iteration")); -} - -// Test the start() function case 2 -TEST_F(IonsMoveBFGSTest, StartCase2) -{ - // Initilize data - UnitCell ucell; - // Initialize UnitCell with 2 atoms - ucell.ntype = 1; - ucell.nat = 2; - ucell.atoms = new Atom[ucell.ntype]; - ucell.atoms[0].na = 2; - ucell.atoms[0].tau = std::vector>(2); - ucell.atoms[0].taud = std::vector>(2); - ucell.atoms[0].mbl = std::vector>(2, {1, 1, 1}); - ucell.atoms[0].tau[0].x = 0.0; ucell.atoms[0].tau[0].y = 0.0; ucell.atoms[0].tau[0].z = 0.0; - ucell.atoms[0].tau[1].x = 1.0; ucell.atoms[0].tau[1].y = 0.0; ucell.atoms[0].tau[1].z = 0.0; - ucell.lat0 = 1.0; - ucell.set_atom_flag = true; - - // Initialize PARAM - criteria.force_thr = 1.0e-3; - criteria.force_thr_ev = criteria.force_thr * 13.6058 / 0.529177; - criteria.test_relax_method = 1; - criteria.out_level = "ie"; - - // Initialize istep - const int istep = 1; - - ModuleBase::matrix force(2, 3); - force(0, 0) = 10.0; - double energy_in = 0.0; - bfgs.init_done = false; - bfgs.save_flag = true; - std::vector etot_info(2, 0.0); - - // Call the function being tested - bfgs.allocate(); - std::ofstream ofs("test_start_case2.log"); - bfgs.start(ucell, force, energy_in, istep, update_iter, ofs, etot_info, criteria); - ofs.close(); - - // Check the results - std::ifstream ifs("test_start_case2.log"); - std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); - ifs.close(); - std::remove("test_start_case2.log"); - - EXPECT_THAT(output, testing::HasSubstr("Ion relaxation is not converged yet")); - - // Clean up - delete[] ucell.atoms; -} - -// Test the restart_bfgs() function case 1 -TEST_F(IonsMoveBFGSTest, RestartBfgsCase1) -{ - // Initilize data - bfgs.init_done = false; - criteria.test_relax_method = 1; - double lat0 = 1.0; - bfgs.allocate(); - bfgs.save_flag = true; - for (int i = 0; i < Ions_Move_Basic::dim; ++i) - { - bfgs.move_p[i] = 1.0; - bfgs.pos[i] = 1.0; - bfgs.pos_p[i] = i; - } - - // Call the function being tested - std::ofstream ofs("test_restart_bfgs_case1.log"); - bfgs.restart_bfgs(lat0, update_iter, ofs, criteria.test_relax_method); - ofs.close(); - - // Check the results - std::ifstream ifs("test_restart_bfgs_case1.log"); - std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); - ifs.close(); - std::remove("test_restart_bfgs_case1.log"); - - std::string expected_output = " trust_radius_old (bohr) = 2.44949\n"; - - EXPECT_THAT(output, testing::HasSubstr(expected_output)); - EXPECT_NEAR(Ions_Move_Basic::trust_radius_old, 2.4494897427831779, 1e-12); - EXPECT_DOUBLE_EQ(bfgs.move_p[0], 0.0); - EXPECT_DOUBLE_EQ(bfgs.move_p[1], 0.0); - EXPECT_DOUBLE_EQ(bfgs.move_p[2], 0.0); - EXPECT_NEAR(bfgs.move_p[3], -0.40824829046386307, 1e-12); - EXPECT_NEAR(bfgs.move_p[4], -0.81649658092772615, 1e-12); - EXPECT_NEAR(bfgs.move_p[5], -1.2247448713915892, 1e-12); -} - -// Test the restart_bfgs() function case 2 -TEST_F(IonsMoveBFGSTest, RestartBfgsCase2) -{ - // Initilize data - bfgs.init_done = false; - bfgs.allocate(); - criteria.test_relax_method = 1; - double lat0 = 1.0; - for (int i = 0; i < Ions_Move_Basic::dim; ++i) - { - bfgs.move_p[i] = 1.0; - bfgs.pos[i] = i; - bfgs.pos_p[i] = i; - } - - // Call the function being tested - std::ofstream ofs("test_restart_bfgs_case2.log"); - bfgs.restart_bfgs(lat0, update_iter, ofs, criteria.test_relax_method); - ofs.close(); - std::remove("test_restart_bfgs_case2.log"); - - // Check the results - EXPECT_DOUBLE_EQ(update_iter, 0.0); - EXPECT_DOUBLE_EQ(bfgs.tr_min_hit, false); - for (int i = 0; i < Ions_Move_Basic::dim; ++i) - { - EXPECT_DOUBLE_EQ(bfgs.pos_p[i], 0.0); - EXPECT_DOUBLE_EQ(bfgs.grad_p[i], 0.0); - EXPECT_DOUBLE_EQ(bfgs.move_p[i], 0.0); - for (int j = 0; j < Ions_Move_Basic::dim; ++j) - { - if (i == j) - { - EXPECT_DOUBLE_EQ(bfgs.inv_hess(i, j), 1.0); - } - else - { - EXPECT_DOUBLE_EQ(bfgs.inv_hess(i, j), 0.0); - } - } - } -} - -// Test the bfgs_routine() function case 1 -TEST_F(IonsMoveBFGSTest, BfgsRoutineCase1) -{ - // Initilize data - bfgs.init_done = false; - bfgs.allocate(); - bfgs.tr_min_hit = false; - criteria.test_relax_method = 1; - criteria.out_level = "ie"; - double lat0 = 1.0; - const int istep = 1; - std::vector etot_info = {1.0, 0.9, 0.1}; - Ions_Move_Basic::relax_bfgs_rmin = 1.0; - for (int i = 0; i < Ions_Move_Basic::dim; ++i) - { - bfgs.move_p[i] = 0.0; - bfgs.grad_p[i] = i; - bfgs.pos_p[i] = i; - } - - // Call the function being tested - std::ofstream ofs("test_bfgs_routine_case1.log"); - testing::internal::CaptureStdout(); - bfgs.bfgs_routine(lat0, istep, update_iter, ofs, etot_info, criteria.out_level, criteria.test_relax_method); - std::string std_outout = testing::internal::GetCapturedStdout(); - ofs.close(); - - // Check the results - std::ifstream ifs("test_bfgs_routine_case1.log"); - std::string ofs_output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); - ifs.close(); - std::remove("test_bfgs_routine_case1.log"); - - std::string expected_ofs - = " dE0s = 0\n den = 0.1\n " - " interpolated trust radius = 0\ntrust_radius = 0\nrelax_bfgs_rmin = 1\nrelax_bfgs_rmax = -1\n " - "trust_radius < relax_bfgs_rmin, reset bfgs history.\n"; - std::string expected_std = " BFGS TRUST (Bohr) : 1\n"; - - EXPECT_THAT(ofs_output, ::testing::HasSubstr(expected_ofs)); - EXPECT_EQ(expected_std, std_outout); - - EXPECT_DOUBLE_EQ(Ions_Move_Basic::trust_radius, 1.0); - EXPECT_DOUBLE_EQ(etot_info[0], 0.9); - EXPECT_DOUBLE_EQ(bfgs.tr_min_hit, true); - EXPECT_NEAR(bfgs.move[0], 0.0, 1e-12); - EXPECT_NEAR(bfgs.move[1], -0.13483997249264842, 1e-12); - EXPECT_NEAR(bfgs.move[2], -0.26967994498529685, 1e-12); - EXPECT_NEAR(bfgs.move[3], -0.40451991747794525, 1e-12); - EXPECT_NEAR(bfgs.move[4], -0.5393598899705937, 1e-12); - EXPECT_NEAR(bfgs.move[5], -0.67419986246324215, 1e-12); - for (int i = 0; i < Ions_Move_Basic::dim; ++i) - { - EXPECT_DOUBLE_EQ(bfgs.pos[i], i); - EXPECT_DOUBLE_EQ(bfgs.grad[i], i); - for (int j = 0; j < Ions_Move_Basic::dim; ++j) - { - if (i == j) - EXPECT_DOUBLE_EQ(bfgs.inv_hess(i, j), 1.0); - else - EXPECT_DOUBLE_EQ(bfgs.inv_hess(i, j), 0.0); - } - } -} - -// Test the bfgs_routine() function case 2 -TEST_F(IonsMoveBFGSTest, BfgsRoutineCase2) -{ - // Initilize data - bfgs.init_done = false; - bfgs.allocate(); - bfgs.tr_min_hit = false; - criteria.test_relax_method = 0; - criteria.out_level = "none"; - double lat0 = 1.0; - const int istep = 1; - std::vector etot_info = {1.0, 0.9, 0.1}; - Ions_Move_Basic::relax_bfgs_rmin = -1.0; - for (int i = 0; i < Ions_Move_Basic::dim; ++i) - { - bfgs.move_p[i] = i; - bfgs.grad_p[i] = i; - bfgs.pos_p[i] = i; - } - - // Call the function being tested - std::ofstream ofs("test_bfgs_routine_case2.log"); - testing::internal::CaptureStdout(); - bfgs.bfgs_routine(lat0, istep, update_iter, ofs, etot_info, criteria.out_level, criteria.test_relax_method); - std::string std_outout = testing::internal::GetCapturedStdout(); - ofs.close(); - - // Check the results - std::ifstream ifs("test_bfgs_routine_case2.log"); - std::string ofs_output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); - ifs.close(); - std::remove("test_bfgs_routine_case2.log"); - - std::string expected_ofs = " quadratic interpolation is impossible.\n"; - std::string expected_std = ""; - - EXPECT_THAT(ofs_output, ::testing::HasSubstr(expected_ofs)); - EXPECT_EQ(expected_std, std_outout); - - EXPECT_DOUBLE_EQ(Ions_Move_Basic::trust_radius, -0.5); - EXPECT_DOUBLE_EQ(etot_info[0], 0.9); - EXPECT_DOUBLE_EQ(bfgs.tr_min_hit, false); - EXPECT_NEAR(bfgs.move[0], 0.0, 1e-12); - EXPECT_NEAR(bfgs.move[1], 0.067419986246324212, 1e-12); - EXPECT_NEAR(bfgs.move[2], 0.13483997249264842, 1e-12); - EXPECT_NEAR(bfgs.move[3], 0.20225995873897262, 1e-12); - EXPECT_NEAR(bfgs.move[4], 0.26967994498529685, 1e-12); - EXPECT_NEAR(bfgs.move[5], 0.33709993123162107, 1e-12); - for (int i = 0; i < Ions_Move_Basic::dim; ++i) - { - EXPECT_DOUBLE_EQ(bfgs.pos[i], i); - EXPECT_DOUBLE_EQ(bfgs.grad[i], i); - for (int j = 0; j < Ions_Move_Basic::dim; ++j) - { - EXPECT_DOUBLE_EQ(bfgs.inv_hess(i, j), 0.0); - } - } -} - -// Test the bfgs_routine() function case 3 -TEST_F(IonsMoveBFGSTest, BfgsRoutineCase3) -{ - // Initilize data - double lat0 = 1.0; - const int istep = 1; - std::vector etot_info = {0.9, 1.0, -0.1}; - update_iter = 0; - Ions_Move_Basic::largest_grad = 0.0; - Ions_Move_Basic::relax_bfgs_init = 0.3; - Ions_Move_Basic::best_xxx = -0.4; - bfgs.init_done = false; - bfgs.allocate(); - bfgs.bfgs_ndim = 1; - bfgs.grad[0] = 1.0; - bfgs.grad[1] = 2.0; - bfgs.inv_hess(0, 0) = -3.0; - bfgs.inv_hess(0, 1) = -4.0; - bfgs.inv_hess(1, 0) = -5.0; - bfgs.inv_hess(1, 1) = -6.0; - - // Call the function being tested - std::ofstream ofs("test_bfgs_routine_case3.log"); - bfgs.bfgs_routine(lat0, istep, update_iter, ofs, etot_info, criteria.out_level, criteria.test_relax_method); - ofs.close(); - - // Check the results - std::ifstream ifs("test_bfgs_routine_case3.log"); - std::string ofs_output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); - ifs.close(); - std::remove("test_bfgs_routine_case3.log"); - - std::string expected_ofs = " check the norm of new move 410 (Bohr)\n Uphill move : resetting bfgs history\n"; - - EXPECT_THAT(ofs_output, ::testing::HasSubstr(expected_ofs)); - EXPECT_DOUBLE_EQ(Ions_Move_Basic::trust_radius, 0.2); - EXPECT_DOUBLE_EQ(etot_info[0], 0.9); - EXPECT_DOUBLE_EQ(bfgs.tr_min_hit, false); - EXPECT_NEAR(bfgs.move[0], -0.089442719099991588, 1e-12); - EXPECT_NEAR(bfgs.move[1], -0.17888543819998318, 1e-12); - EXPECT_DOUBLE_EQ(bfgs.move[2], 0.0); - EXPECT_DOUBLE_EQ(bfgs.move[3], 0.0); - EXPECT_DOUBLE_EQ(bfgs.move[4], 0.0); - EXPECT_DOUBLE_EQ(bfgs.move[5], 0.0); - EXPECT_DOUBLE_EQ(bfgs.pos[0], 0.0); - EXPECT_DOUBLE_EQ(bfgs.pos[1], 0.0); - EXPECT_DOUBLE_EQ(bfgs.pos[2], 0.0); - EXPECT_DOUBLE_EQ(bfgs.pos[3], 0.0); - EXPECT_DOUBLE_EQ(bfgs.pos[4], 0.0); - EXPECT_DOUBLE_EQ(bfgs.pos[5], 0.0); - EXPECT_DOUBLE_EQ(bfgs.grad[0], 1.0); - EXPECT_DOUBLE_EQ(bfgs.grad[1], 2.0); - EXPECT_DOUBLE_EQ(bfgs.grad[2], 0.0); - EXPECT_DOUBLE_EQ(bfgs.grad[3], 0.0); - EXPECT_DOUBLE_EQ(bfgs.grad[4], 0.0); - EXPECT_DOUBLE_EQ(bfgs.grad[5], 0.0); -} - -// Test the bfgs_routine() function warning quit 1 -TEST_F(IonsMoveBFGSTest, BfgsRoutineWarningQuit1) -{ - // Initilize data - bfgs.init_done = false; - bfgs.allocate(); - bfgs.tr_min_hit = true; - criteria.test_relax_method = 1; - criteria.out_level = "ie"; - double lat0 = 1.0; - const int istep = 1; - std::vector etot_info = {1.0, 0.9, 0.1}; - Ions_Move_Basic::relax_bfgs_rmin = 1.0; - for (int i = 0; i < Ions_Move_Basic::dim; ++i) - { - bfgs.move_p[i] = 0.0; - bfgs.grad_p[i] = i; - bfgs.pos_p[i] = i; - } - - // Check the results - std::ofstream ofs("test_bfgs_routine_warning_quit1.log"); - testing::internal::CaptureStdout(); - EXPECT_EXIT(bfgs.bfgs_routine(lat0, istep, update_iter, ofs, etot_info, criteria.out_level, criteria.test_relax_method), ::testing::ExitedWithCode(1), ""); - std::string output = testing::internal::GetCapturedStdout(); - ofs.close(); - std::remove("test_bfgs_routine_warning_quit1.log"); - EXPECT_THAT(output, testing::HasSubstr("trust radius is too small! Break down.")); -} - -// Test the bfgs_routine() function warning quit 2 -TEST_F(IonsMoveBFGSTest, BfgsRoutineWarningQuit2) -{ - // Initilize data - bfgs.init_done = false; - bfgs.allocate(); - bfgs.tr_min_hit = false; - criteria.test_relax_method = 1; - criteria.out_level = "ie"; - double lat0 = 1.0; - const int istep = 1; - std::vector etot_info = {1.0, 0.9, 0.1}; - Ions_Move_Basic::relax_bfgs_rmin = 1.0; - - // Check the results - std::ofstream ofs("test_bfgs_routine_warning_quit2.log"); - testing::internal::CaptureStdout(); - EXPECT_EXIT(bfgs.bfgs_routine(lat0, istep, update_iter, ofs, etot_info, criteria.out_level, criteria.test_relax_method), ::testing::ExitedWithCode(1), ""); - std::string output = testing::internal::GetCapturedStdout(); - ofs.close(); - std::remove("test_bfgs_routine_warning_quit2.log"); - EXPECT_THAT(output, testing::HasSubstr("BFGS: move-length unreasonably short")); -} diff --git a/source/source_relax/test/ions_move_cg_test.cpp b/source/source_relax/test/ions_move_cg_test.cpp deleted file mode 100644 index f311ea605f7..00000000000 --- a/source/source_relax/test/ions_move_cg_test.cpp +++ /dev/null @@ -1,650 +0,0 @@ -#include "source_relax/relax_criteria.h" -#include -#include "for_test.h" -#include "gtest/gtest.h" -#include "gmock/gmock.h" -#include "source_relax/ions_move_basic.h" -#include "source_relax/ions_move_cg.h" - -/************************************************ - * unit tests of class Ions_Move_CG - ***********************************************/ - -class IonsMoveCGTest : public ::testing::Test -{ - public: - Relax_Criteria criteria; - - protected: - void SetUp() override - { - // Initialize variables before each test - Ions_Move_Basic::dim = 6; - update_iter = 5; - im_cg.allocate(Ions_Move_Basic::dim); - criteria.force_thr = 0.001; - - // ban the 'cout' - // mohan add 2025-05-02 - std::cout.rdbuf(NULL); - } - - void TearDown() override - { - // Clean up after each test - } - void setupucell(UnitCell& ucell) - { - for (int it = 0; it < ucell.ntype; it++) - { - Atom* atom = &ucell.atoms[it]; - atom->label="test"; - for (int ia = 0; ia < atom->na; ia++) - { - atom->mag[ia]= 1; - for (int ik = 0; ik < 3; ++ik) - { - atom->tau[ia][ik] = 1; - atom->mbl[ia][ik] = 1; - atom->vel[ia][ik] = 1; - } - } - } - ucell.lat.GT.Zero(); - } - Ions_Move_CG im_cg; - int update_iter; - - // Friendship is not inherited: a TEST_F body lives in a class derived from - // this fixture, so the internals of Ions_Move_CG (which friends the fixture - // itself) are reached through these forwarders. - const std::vector& get_pos0() const - { - return im_cg.pos0; - } - const std::vector& get_grad0() const - { - return im_cg.grad0; - } - const std::vector& get_cg_grad0() const - { - return im_cg.cg_grad0; - } - const std::vector& get_move0() const - { - return im_cg.move0; - } - void set_move0(const int i, const double value) - { - im_cg.move0[i] = value; - } -}; - -// Test whether the allocate() function can correctly allocate memory space -TEST_F(IonsMoveCGTest, TestAllocate) -{ - const int dim = 4; - im_cg.allocate(dim); - - // Check if allocated vectors are not empty - EXPECT_EQ(get_pos0().size(), 4U); - EXPECT_EQ(get_grad0().size(), 4U); - EXPECT_EQ(get_cg_grad0().size(), 4U); - EXPECT_EQ(get_move0().size(), 4U); -} - -// Test if a dimension less than or equal to 0 results in an assertion error -TEST_F(IonsMoveCGTest, TestAllocateWithZeroDimension) -{ - const int dim = 0; - ASSERT_DEATH(im_cg.allocate(dim), ""); -} - -// Check that the arrays are correctly initialized to 0 -TEST_F(IonsMoveCGTest, TestAllocateAndInitialize) -{ - const int dim = 3; - im_cg.allocate(dim); - - // Check that the arrays are correctly initialized to 0 - EXPECT_DOUBLE_EQ(0.0, get_pos0()[0]); - EXPECT_DOUBLE_EQ(0.0, get_grad0()[1]); - EXPECT_DOUBLE_EQ(0.0, get_cg_grad0()[2]); - EXPECT_DOUBLE_EQ(0.0, get_move0()[0]); -} - -// Test function start() when converged -TEST_F(IonsMoveCGTest, TestStartConverged) -{ - // setup data - const int istep = 1; - UnitCell ucell; - setupucell(ucell); - ModuleBase::matrix force(2, 3); - double etot = 0.0; - std::vector etot_info(2, 0.0); - std::vector relax_method = {"cg", "1"}; - - // call function - std::ofstream ofs("TestStartConverged.log"); - im_cg.start(ucell, force, etot, istep, update_iter, ofs, etot_info, relax_method, criteria); - ofs.close(); - - // Check output - std::string expected_output = "\n Largest force is 0 eV/Angstrom while threshold is -1 eV/Angstrom\n" - " largest force is 0, no movement is possible.\n it may converged, otherwise no " - "movement of atom is allowed.\n end of geometry optimization\n " - " istep = 1\n update iteration = 5\n"; - std::ifstream ifs("TestStartConverged.log"); - std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); - ifs.close(); - std::remove("TestStartConverged.log"); - - - std::regex pattern(R"(==> .*::.*\t[\d\.]+ GB\t\d+ s\n )"); - output = std::regex_replace(output, pattern, ""); - EXPECT_THAT(output, testing::HasSubstr(expected_output)); - EXPECT_EQ(update_iter, 5); - EXPECT_DOUBLE_EQ(Ions_Move_Basic::largest_grad, 0.0); -} - -// Test function start() sd branch -TEST_F(IonsMoveCGTest, TestStartSd) -{ - // setup data - const int istep = 1; - std::vector relax_method = {"cg_bfgs", "1"}; - Ions_Move_CG::RELAX_CG_THR = 100.0; - UnitCell ucell; - setupucell(ucell); - ModuleBase::matrix force(2, 3); - force(0, 0) = 0.01; - double etot = 0.0; - std::vector etot_info(2, 0.0); - - // call function - std::ofstream ofs("TestStartSd.log"); - im_cg.start(ucell, force, etot, istep, update_iter, ofs, etot_info, relax_method, criteria); - ofs.close(); - - // Check output - std::string expected_output = "\n Largest force is 0.257111 eV/Angstrom while threshold is -1 eV/Angstrom\n\n" - " Ion relaxation is not converged yet (threshold is 0.0257111)\n"; - std::ifstream ifs("TestStartSd.log"); - std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); - ifs.close(); - std::remove("TestStartSd.log"); // mohan - - EXPECT_THAT(output, testing::HasSubstr(expected_output)); - EXPECT_EQ(update_iter, 5); - EXPECT_EQ(relax_method[0], "bfgs"); - EXPECT_DOUBLE_EQ(Ions_Move_Basic::largest_grad, 0.01); - EXPECT_DOUBLE_EQ(Ions_Move_Basic::best_xxx, -1.0); - EXPECT_DOUBLE_EQ(Ions_Move_Basic::relax_bfgs_init, 1.0); -} - -// Test function start() trial branch with goto -TEST_F(IonsMoveCGTest, TestStartTrialGoto) -{ - // setup data - const int istep = 1; - Ions_Move_CG::RELAX_CG_THR = 100.0; - UnitCell ucell; - setupucell(ucell); - ModuleBase::matrix force(2, 3); - force(0, 0) = 0.1; - double etot = 0.0; - std::vector etot_info(2, 0.0); - std::vector relax_method = {"cg_bfgs", "1"}; - - // call function - set_move0(0, 1.0); - std::ofstream ofs1("TestStartTrialGoto_temp1.log"); - im_cg.start(ucell, force, etot, istep, update_iter, ofs1, etot_info, relax_method, criteria); - ofs1.close(); - std::remove("TestStartTrialGoto_temp1.log"); - int istep_2 = 2; - set_move0(0, 10.0); - force(0, 0) = 0.001; - relax_method = {"cg_bfgs", "1"}; - std::ofstream ofs("TestStartTrialGoto.log"); - im_cg.start(ucell, force, etot, istep_2, update_iter, ofs, etot_info, relax_method, criteria); - ofs.close(); - - // Check output - std::string expected_output = "\n Largest force is 0.0257111 eV/Angstrom while threshold is -1 eV/Angstrom\n\n" - " Ion relaxation is not converged yet (threshold is 0.0257111)\n"; - std::ifstream ifs("TestStartTrialGoto.log"); - std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); - ifs.close(); - std::remove("TestStartTrialGoto.log"); - - EXPECT_THAT(output, testing::HasSubstr(expected_output)); - EXPECT_EQ(update_iter, 5); - EXPECT_EQ(relax_method[0], "bfgs"); - EXPECT_DOUBLE_EQ(Ions_Move_Basic::largest_grad, 0.001); - EXPECT_DOUBLE_EQ(Ions_Move_Basic::best_xxx, 10.0); - EXPECT_DOUBLE_EQ(Ions_Move_Basic::relax_bfgs_init, 10.0); -} - -// Test function start() trial branch without goto -TEST_F(IonsMoveCGTest, TestStartTrial) -{ - // setup data - const int istep = 1; - UnitCell ucell; - setupucell(ucell); - ModuleBase::matrix force(2, 3); - force(0, 0) = 0.01; - double etot = 0.0; - std::vector etot_info(2, 0.0); - std::vector relax_method = {"cg_bfgs", "1"}; - - // call function - set_move0(0, 1.0); - std::ofstream ofs1("TestStartTrial_temp1.log"); - im_cg.start(ucell, force, etot, istep, update_iter, ofs1, etot_info, relax_method, criteria); - ofs1.close(); - std::remove("TestStartTrial_temp1.log"); - int istep_2 = 2; - set_move0(0, 10.0); - std::ofstream ofs("TestStartTrial.log"); - im_cg.start(ucell, force, etot, istep_2, update_iter, ofs, etot_info, relax_method, criteria); - ofs.close(); - - // Check output - std::string expected_output = "\n Largest force is 0.257111 eV/Angstrom while threshold is -1 eV/Angstrom\n\n" - " Ion relaxation is not converged yet (threshold is 0.0257111)\n"; - std::ifstream ifs("TestStartTrial.log"); - std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); - ifs.close(); - std::remove("TestStartTrial.log"); - - EXPECT_THAT(output, testing::HasSubstr(expected_output)); - EXPECT_EQ(update_iter, 5); - EXPECT_EQ(relax_method[0], "bfgs"); - EXPECT_DOUBLE_EQ(Ions_Move_Basic::largest_grad, 0.01); - EXPECT_DOUBLE_EQ(Ions_Move_Basic::best_xxx, 10.0); - EXPECT_DOUBLE_EQ(Ions_Move_Basic::relax_bfgs_init, 70.0); -} - -// Test function start() no trial branch with goto case 1 -TEST_F(IonsMoveCGTest, TestStartNoTrialGotoCase1) -{ - // setup data - const int istep = 1; - Ions_Move_CG::RELAX_CG_THR = 100.0; - UnitCell ucell; - setupucell(ucell); - ModuleBase::matrix force(2, 3); - force(0, 0) = 0.1; - double etot = 0.0; - std::vector etot_info(2, 0.0); - std::vector relax_method = {"cg_bfgs", "1"}; - - // call function - set_move0(0, 1.0); - std::ofstream ofs1("TestStartNoTrialGotoCase1_temp1.log"); - im_cg.start(ucell, force, etot, istep, update_iter, ofs1, etot_info, relax_method, criteria); - ofs1.close(); - std::remove("TestStartNoTrialGotoCase1_temp1.log"); - int istep_2 = 2; - std::ofstream ofs2("TestStartNoTrialGotoCase1_temp2.log"); - im_cg.start(ucell, force, etot, istep_2, update_iter, ofs2, etot_info, relax_method, criteria); - ofs2.close(); - std::remove("TestStartNoTrialGotoCase1_temp2.log"); - set_move0(0, 1.0); - force(0, 0) = 0.001; - relax_method = {"cg_bfgs", "1"}; - std::ofstream ofs("TestStartNoTrialGotoCase1.log"); - im_cg.start(ucell, force, etot, istep_2, update_iter, ofs, etot_info, relax_method, criteria); - ofs.close(); - - // Check output - std::string expected_output = "\n Largest force is 0.0257111 eV/Angstrom while threshold is -1 eV/Angstrom\n\n" - " Ion relaxation is not converged yet (threshold is 0.0257111)\n"; - std::ifstream ifs("TestStartNoTrialGotoCase1.log"); - std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); - ifs.close(); - std::remove("TestStartNoTrialGotoCase1.log"); - - EXPECT_THAT(output, testing::HasSubstr(expected_output)); - EXPECT_EQ(update_iter, 5); - EXPECT_EQ(relax_method[0], "bfgs"); - EXPECT_DOUBLE_EQ(Ions_Move_Basic::largest_grad, 0.001); - EXPECT_DOUBLE_EQ(Ions_Move_Basic::best_xxx, 490.0); - EXPECT_DOUBLE_EQ(Ions_Move_Basic::relax_bfgs_init, 490.0); -} - -// Test function start() no trial branch with goto case 2 -TEST_F(IonsMoveCGTest, TestStartNoTrialGotoCase2) -{ - // setup data - const int istep = 1; - Ions_Move_CG::RELAX_CG_THR = 100.0; - UnitCell ucell; - setupucell(ucell); - ModuleBase::matrix force(2, 3); - force(0, 0) = 0.01; - double etot = 0.0; - std::vector etot_info(2, 0.0); - std::vector relax_method = {"cg_bfgs", "1"}; - Ions_Move_Basic::best_xxx = 1.0; - Ions_Move_Basic::relax_bfgs_init = 1.0; - - // call function - set_move0(0, 1.0); - std::ofstream ofs1("TestStartNoTrialGotoCase2_temp1.log"); - im_cg.start(ucell, force, etot, istep, update_iter, ofs1, etot_info, relax_method, criteria); - ofs1.close(); - std::remove("TestStartNoTrialGotoCase2_temp1.log"); - int istep_2 = 2; - set_move0(0, 10.0); - std::ofstream ofs2("TestStartNoTrialGotoCase2_temp2.log"); - im_cg.start(ucell, force, etot, istep_2, update_iter, ofs2, etot_info, relax_method, criteria); - ofs2.close(); - std::remove("TestStartNoTrialGotoCase2_temp2.log"); - relax_method = {"cg_bfgs", "1"}; - std::ofstream ofs("TestStartNoTrialGotoCase2.log"); - im_cg.start(ucell, force, etot, istep_2, update_iter, ofs, etot_info, relax_method, criteria); - ofs.close(); - - // Check output - std::string expected_output = "\n Largest force is 0.257111 eV/Angstrom while threshold is -1 eV/Angstrom\n\n" - " Ion relaxation is not converged yet (threshold is 0.0257111)\n"; - std::ifstream ifs("TestStartNoTrialGotoCase2.log"); - std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); - ifs.close(); - std::remove("TestStartNoTrialGotoCase2.log"); - - EXPECT_THAT(output, testing::HasSubstr(expected_output)); - EXPECT_EQ(update_iter, 5); - EXPECT_EQ(relax_method[0], "bfgs"); - EXPECT_DOUBLE_EQ(Ions_Move_Basic::largest_grad, 0.01); - EXPECT_DOUBLE_EQ(Ions_Move_Basic::best_xxx, 70.0); - EXPECT_DOUBLE_EQ(Ions_Move_Basic::relax_bfgs_init, 70.0); -} - -// Test function start() no trial branch without goto -TEST_F(IonsMoveCGTest, TestStartNoTrial) -{ - // setup data - const int istep = 1; - Ions_Move_CG::RELAX_CG_THR = 100.0; - UnitCell ucell; - setupucell(ucell); - ModuleBase::matrix force(2, 3); - force(0, 0) = 0.01; - double etot = 0.0; - std::vector etot_info(2, 0.0); - std::vector relax_method = {"cg_bfgs", "1"}; - Ions_Move_Basic::best_xxx = 1.0; - Ions_Move_Basic::relax_bfgs_init = 1.0; - - // call function - set_move0(0, 1.0); - std::ofstream ofs1("TestStartNoTrial_temp1.log"); - im_cg.start(ucell, force, etot, istep, update_iter, ofs1, etot_info, relax_method, criteria); - ofs1.close(); - std::remove("TestStartNoTrial_temp1.log"); - int istep_2 = 2; - set_move0(0, 1.0); - force(0, 0) = 0.001; - std::ofstream ofs2("TestStartNoTrial_temp2.log"); - im_cg.start(ucell, force, etot, istep_2, update_iter, ofs2, etot_info, relax_method, criteria); - ofs2.close(); - std::remove("TestStartNoTrial_temp2.log"); - std::ofstream ofs("TestStartNoTrial.log"); - im_cg.start(ucell, force, etot, istep_2, update_iter, ofs, etot_info, relax_method, criteria); - ofs.close(); - - // Check output - std::string expected_output = "\n Largest force is 0.0257111 eV/Angstrom while threshold is -1 eV/Angstrom\n\n" - " Ion relaxation is not converged yet (threshold is 0.0257111)\n"; - std::ifstream ifs("TestStartNoTrial.log"); - std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); - ifs.close(); - std::remove("TestStartNoTrial.log"); - - EXPECT_THAT(output, testing::HasSubstr(expected_output)); - EXPECT_EQ(update_iter, 5); - EXPECT_EQ(relax_method[0], "bfgs"); - EXPECT_DOUBLE_EQ(Ions_Move_Basic::largest_grad, 0.001); - EXPECT_DOUBLE_EQ(Ions_Move_Basic::best_xxx, 1.0); - EXPECT_NEAR(Ions_Move_Basic::relax_bfgs_init, 1.2345679012345678, 1e-12); -} - -// Test function setup_cg_grad() when ncggrad is multiple of 10000 -TEST_F(IonsMoveCGTest, SetupCgGradNcggradIsMultipleOf10000) -{ - double grad[6] = {1.0, 2.0, 3.0, 4.0, 5.0, 6.0}; - double grad0[6] = {1.0, 2.0, 3.0, 4.0, 5.0, 6.0}; - double cggrad[6] = {9.0, 8.0, 7.0, 6.0, 5.0, 4.0}; - double cggrad0[6] = {1.0, 2.0, 3.0, 4.0, 5.0, 6.0}; - int ncggrad = 50000; // multiple of 10000 - int flag = 0; - - im_cg.setup_cg_grad(Ions_Move_Basic::dim, grad, grad0, cggrad, cggrad0, ncggrad, flag); - - EXPECT_DOUBLE_EQ(cggrad[0], grad[0]); - EXPECT_DOUBLE_EQ(cggrad[1], grad[1]); - EXPECT_DOUBLE_EQ(cggrad[2], grad[2]); - EXPECT_DOUBLE_EQ(cggrad[3], grad[3]); - EXPECT_DOUBLE_EQ(cggrad[4], grad[4]); - EXPECT_DOUBLE_EQ(cggrad[5], grad[5]); -} - -// Test function setup_cg_grad() when ncggrad is not multiple of 10000, gamma1 < 0.5 -TEST_F(IonsMoveCGTest, SetupCgGradNcggradIsNotMultipleOf10000Case1) -{ - double grad[6] = {1.0, 0.0, 0.0, 0.0, 0.0, 0.0}; - double grad0[6] = {4.0, 0.0, 0.0, 0.0, 0.0, 0.0}; - double cggrad[6] = {4.0, 0.0, 0.0, 0.0, 0.0, 0.0}; - double cggrad0[6] = {4.0, 0.0, 0.0, 0.0, 0.0, 0.0}; - int ncggrad = 100; - int flag = 0; - - im_cg.setup_cg_grad(Ions_Move_Basic::dim, grad, grad0, cggrad, cggrad0, ncggrad, flag); - - EXPECT_DOUBLE_EQ(cggrad[0], 1.25); - EXPECT_DOUBLE_EQ(cggrad[1], 0.0); - EXPECT_DOUBLE_EQ(cggrad[2], 0.0); - EXPECT_DOUBLE_EQ(cggrad[3], 0.0); - EXPECT_DOUBLE_EQ(cggrad[4], 0.0); - EXPECT_DOUBLE_EQ(cggrad[5], 0.0); -} - -// Test function setup_cg_grad() when ncggrad is not multiple of 10000, gamma1 >= 0.5 -TEST_F(IonsMoveCGTest, SetupCgGradNcggradIsNotMultipleOf10000Case2) -{ - double grad[6] = {1.0, 2.0, 3.0, 4.0, 5.0, 6.0}; - double grad0[6] = {1.0, 2.0, 3.0, 4.0, 5.0, 6.0}; - double cggrad[6] = {9.0, 8.0, 7.0, 6.0, 5.0, 4.0}; - double cggrad0[6] = {1.0, 2.0, 3.0, 4.0, 5.0, 6.0}; - int ncggrad = 100; - int flag = 0; - - im_cg.setup_cg_grad(Ions_Move_Basic::dim, grad, grad0, cggrad, cggrad0, ncggrad, flag); - - EXPECT_DOUBLE_EQ(cggrad[0], grad[0]); - EXPECT_DOUBLE_EQ(cggrad[1], grad[1]); - EXPECT_DOUBLE_EQ(cggrad[2], grad[2]); - EXPECT_DOUBLE_EQ(cggrad[3], grad[3]); - EXPECT_DOUBLE_EQ(cggrad[4], grad[4]); - EXPECT_DOUBLE_EQ(cggrad[5], grad[5]); -} - -// Test function third_order() case 1 -TEST_F(IonsMoveCGTest, ThirdOrderCase1) -{ - double e0 = 1.0; - double e1 = 1.0; - double fa = 10.0; - double fb = -9.99; - double x = 1.0; - double bestX = -1.0; // arbitrary initial value - - im_cg.third_order(e0, e1, fa, fb, x, bestX); - - EXPECT_DOUBLE_EQ(bestX, x * fb / (fa - fb)); -} - -// Test function third_order() case 2 -TEST_F(IonsMoveCGTest, ThirdOrderCase2) -{ - double e0 = 1.0; - double e1 = 1.0; - double fa = -10.0; - double fb = 9.9; - double x = 1.0; - double bestX = -1.0; // arbitrary initial value - - im_cg.third_order(e0, e1, fa, fb, x, bestX); - - EXPECT_DOUBLE_EQ(bestX, x * fb / (fa - fb)); -} - -// Test function third_order() case 3 -TEST_F(IonsMoveCGTest, ThirdOrderCase3) -{ - double e0 = 1.0; - double e1 = 1.0; - double fa = 10.0; - double fb = -10.1; - double x = 1.0; - double bestX = -1.0; // arbitrary initial value - - im_cg.third_order(e0, e1, fa, fb, x, bestX); - - EXPECT_DOUBLE_EQ(bestX, x * fb / (fa - fb)); -} - -// Test function Brent() case 1 -TEST_F(IonsMoveCGTest, BrentCase1) -{ - double fa = 2.0; - double fb = 1.0; - double fc = 1.0; - double xa = -3.0; - double xb = 2.0; - double xc = 1.0; - double best_x = 0.0; - double xpt = 0.0; - - im_cg.Brent(fa, fb, fc, xa, xb, xc, best_x, xpt); - - EXPECT_DOUBLE_EQ(fa, 2.0); - EXPECT_DOUBLE_EQ(fb, 1.0); - EXPECT_DOUBLE_EQ(fc, 1.0); - EXPECT_DOUBLE_EQ(xa, -3.0); - EXPECT_DOUBLE_EQ(xb, 1.0); - EXPECT_DOUBLE_EQ(xc, 4.0); - EXPECT_DOUBLE_EQ(best_x, 4.0); - EXPECT_DOUBLE_EQ(xpt, 4.0); -} - -// Test function Brent() case 2 -TEST_F(IonsMoveCGTest, BrentCase2) -{ - double fa = -2.0; - double fb = 3.0; - double fc = -4.0; - double xa = 1.0; - double xb = 2.0; - double xc = 3.0; - double best_x = 0.0; - double xpt = 0.0; - - im_cg.Brent(fa, fb, fc, xa, xb, xc, best_x, xpt); - - EXPECT_DOUBLE_EQ(fa, -4.0); - EXPECT_DOUBLE_EQ(fb, 3.0); - EXPECT_DOUBLE_EQ(fc, -4.0); - EXPECT_DOUBLE_EQ(xa, 3.0); - EXPECT_DOUBLE_EQ(xb, 2.0); - EXPECT_NEAR(xc, 1.2046663545568725, 1e-12); - EXPECT_NEAR(best_x, 1.2046663545568725, 1e-12); - EXPECT_NEAR(xpt, 1.2046663545568725, 1e-12); -} - -// Test function Brent() case 3 -TEST_F(IonsMoveCGTest, BrentCase3) -{ - double fa = 1.0; - double fb = -3.0; - double fc = -4.0; - double xa = 3.0; - double xb = 2.0; - double xc = 1.0; - double best_x = 0.0; - double xpt = 0.0; - - im_cg.Brent(fa, fb, fc, xa, xb, xc, best_x, xpt); - - EXPECT_DOUBLE_EQ(fa, 1.0); - EXPECT_DOUBLE_EQ(fb, -4.0); - EXPECT_DOUBLE_EQ(fc, -4.0); - EXPECT_DOUBLE_EQ(xa, 3.0); - EXPECT_DOUBLE_EQ(xb, 1.0); - EXPECT_NEAR(xc, 2.8081429669660172, 1e-12); - EXPECT_NEAR(best_x, 2.8081429669660172, 1e-12); - EXPECT_NEAR(xpt, 2.8081429669660172, 1e-12); -} - -// Test function Brent() case 4 -TEST_F(IonsMoveCGTest, BrentCase4) -{ - double fa = 2.0; - double fb = -3.0; - double fc = 4.0; - double xa = 0.0; - double xb = 2.0; - double xc = 1.0; - double best_x = 0.0; - double xpt = 0.0; - - im_cg.Brent(fa, fb, fc, xa, xb, xc, best_x, xpt); - - EXPECT_DOUBLE_EQ(fa, 4.0); - EXPECT_DOUBLE_EQ(fb, -3.0); - EXPECT_DOUBLE_EQ(fc, 4.0); - EXPECT_DOUBLE_EQ(xa, 1.0); - EXPECT_DOUBLE_EQ(xb, 2.0); - EXPECT_DOUBLE_EQ(xc, 2.0); - EXPECT_DOUBLE_EQ(best_x, 2.0); - EXPECT_DOUBLE_EQ(xpt, 2.0); -} - -// Test function f_cal() -TEST_F(IonsMoveCGTest, Fcal) -{ - Ions_Move_Basic::dim = 9; - double g0[9] = {1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0}; - double g1[9] = {1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0}; - double f_value = 0.0; - - im_cg.f_cal(Ions_Move_Basic::dim, g0, g1, f_value); - - EXPECT_DOUBLE_EQ(f_value, 3.0); -} - -// Test function setup_move() -TEST_F(IonsMoveCGTest, SetupMove) -{ - Ions_Move_Basic::dim = 9; - double trust_radius = 1.0; - double cg_gradn[9] = {1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0}; - double move[9] = {0.0}; - - im_cg.setup_move(Ions_Move_Basic::dim, move, cg_gradn, trust_radius); - - EXPECT_DOUBLE_EQ(move[0], -1.0); - EXPECT_DOUBLE_EQ(move[1], -1.0); - EXPECT_DOUBLE_EQ(move[2], -1.0); - EXPECT_DOUBLE_EQ(move[3], -1.0); - EXPECT_DOUBLE_EQ(move[4], -1.0); - EXPECT_DOUBLE_EQ(move[5], -1.0); - EXPECT_DOUBLE_EQ(move[6], -1.0); - EXPECT_DOUBLE_EQ(move[7], -1.0); - EXPECT_DOUBLE_EQ(move[8], -1.0); -} diff --git a/source/source_relax/test/ions_move_methods_test.cpp b/source/source_relax/test/ions_move_methods_test.cpp deleted file mode 100644 index 9c396eb28f8..00000000000 --- a/source/source_relax/test/ions_move_methods_test.cpp +++ /dev/null @@ -1,296 +0,0 @@ -#include "source_relax/relax_criteria.h" -#include "for_test.h" -#include "gmock/gmock.h" -#include "gtest/gtest.h" -#include -#include -#include -#include -#include -#include -#define private public -#define protected public -#include "source_relax/ions_move_methods.h" -#undef protected -#undef private -/************************************************ - * unit tests of class Ions_Move_Methods - ***********************************************/ - -/** - * - Tested Functions: - * - Ions_Move_Methods::allocate() - * - Ions_Move_Methods::cal_movement() - * - Ions_Move_Methods::get_converged() - * - Ions_Move_Methods::get_ediff() - * - Ions_Move_Methods::get_largest_grad() - * - Ions_Move_Methods::get_trust_radius() - * - Ions_Move_Methods::get_update_iter() - */ - - // Mock the remake_cell function from update_cell.h -namespace unitcell -{ - // Track if remake_cell was called and with what lattice - static bool remake_cell_called = false; - static std::string remake_cell_latName; - static ModuleBase::Matrix3 remake_cell_latvec; - - void remake_cell(Lattice& lat) - { - remake_cell_called = true; - remake_cell_latName = lat.latName; - remake_cell_latvec = lat.latvec; - - // Mock implementation: enforce simple cubic structure for "sc" - if (lat.latName == "sc") - { - double celldm = std::sqrt(lat.latvec.e11 * lat.latvec.e11 + - lat.latvec.e12 * lat.latvec.e12 + - lat.latvec.e13 * lat.latvec.e13); - lat.latvec.Zero(); - lat.latvec.e11 = celldm; - lat.latvec.e22 = celldm; - lat.latvec.e33 = celldm; - } - // Mock implementation: enforce FCC structure for "fcc" - else if (lat.latName == "fcc") - { - double celldm = std::sqrt(lat.latvec.e11 * lat.latvec.e11 + - lat.latvec.e12 * lat.latvec.e12 + - lat.latvec.e13 * lat.latvec.e13) / std::sqrt(2.0); - lat.latvec.e11 = -celldm; - lat.latvec.e12 = 0.0; - lat.latvec.e13 = celldm; - lat.latvec.e21 = 0.0; - lat.latvec.e22 = celldm; - lat.latvec.e23 = celldm; - lat.latvec.e31 = -celldm; - lat.latvec.e32 = celldm; - lat.latvec.e33 = 0.0; - } - } - - void update_pos_tau(const Lattice&, const double*, const int, const int, Atom*) - { - } - - // Helper function to reset mock state - void reset_remake_cell_mock() - { - remake_cell_called = false; - remake_cell_latName = ""; - remake_cell_latvec.Zero(); - } - - // Helper function to check if remake_cell was called - bool was_remake_cell_called() - { - return remake_cell_called; - } -} - -// Define a fixture for the tests -class IonsMoveMethodsTest : public ::testing::Test -{ - public: - Relax_Criteria criteria; - - protected: - Ions_Move_Methods imm; - const int natom = 2; - - virtual void SetUp() - { - // Initialize variables before each test - } - - virtual void TearDown() - { - // Clean up after each test - } -}; - -// Test the allocate() function -TEST_F(IonsMoveMethodsTest, Allocate) -{ - imm.allocate(natom, "bfgs", "1"); - EXPECT_EQ(Ions_Move_Basic::dim, 6); - - imm.allocate(natom, "sd", "1"); - EXPECT_EQ(Ions_Move_Basic::dim, 6); - - imm.allocate(natom, "cg", "1"); - EXPECT_EQ(Ions_Move_Basic::dim, 6); - - imm.allocate(natom, "cg_bfgs", "1"); - EXPECT_EQ(Ions_Move_Basic::dim, 6); -} - -// Test the allocate() function warning quit -TEST_F(IonsMoveMethodsTest, AllocateWarningQuit) -{ - GlobalV::ofs_warning.open("log"); - imm.allocate(natom, "none", "1"); - GlobalV::ofs_warning.close(); - - std::ifstream ifs("log"); - std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); - EXPECT_THAT(output, testing::HasSubstr("the parameter relax_method is not correct.")); - ifs.close(); - std::remove("log"); -} - -// Test the cal_movement() function -TEST_F(IonsMoveMethodsTest, CalMovement) -{ - const int istep = 0; - const int force_step = 1; - ModuleBase::matrix f(natom, 3); - f(0, 0) = 0.1; - f(1, 1) = -0.1; - const double etot = 0.0; - UnitCell ucell; - std::ofstream ofs; - - std::vector relax_method; - - relax_method = {"bfgs", "1"}; - imm.allocate(natom, "bfgs", "1"); - imm.cal_movement(istep, force_step, f, etot, ucell, ofs, relax_method, criteria); - - relax_method = {"sd", "1"}; - imm.allocate(natom, "sd", "1"); - imm.cal_movement(istep, force_step, f, etot, ucell, ofs, relax_method, criteria); - - relax_method = {"cg", "1"}; - imm.allocate(natom, "cg", "1"); - imm.cal_movement(istep, force_step, f, etot, ucell, ofs, relax_method, criteria); - - relax_method = {"cg_bfgs", "1"}; - imm.allocate(natom, "cg_bfgs", "1"); - imm.cal_movement(istep, force_step, f, etot, ucell, ofs, relax_method, criteria); -} - -// Test the cal_movement() function warning quit -TEST_F(IonsMoveMethodsTest, CalMovementWarningQuit) -{ - const int istep = 0; - const int force_step = 1; - const ModuleBase::matrix f(3, 3); - const double etot = 0.0; - UnitCell ucell; - std::ofstream ofs; - std::vector relax_method = {"none", "1"}; - imm.allocate(natom, "none", "1"); - - GlobalV::ofs_warning.open("log"); - imm.cal_movement(istep, force_step, f, etot, ucell, ofs, relax_method, criteria); - GlobalV::ofs_warning.close(); - - std::ifstream ifs("log"); - std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); - EXPECT_THAT(output, testing::HasSubstr("the parameter relax_method is not correct.")); - ifs.close(); - std::remove("log"); -} - - - -// Test the get_ediff() function -TEST_F(IonsMoveMethodsTest, GetEdiff) -{ - // etot_info_ is initialized to {0.0, 0.0, 0.0} by default - EXPECT_DOUBLE_EQ(imm.get_ediff(), 0.0); -} - -// Test the get_largest_grad() function -TEST_F(IonsMoveMethodsTest, GetLargestGrad) -{ - Ions_Move_Basic::largest_grad = 2.0; - - EXPECT_DOUBLE_EQ(imm.get_largest_grad(), 2.0); -} - -// Test the get_trust_radius() function -TEST_F(IonsMoveMethodsTest, GetTrustRadius) -{ - Ions_Move_Basic::trust_radius = 3.0; - - EXPECT_DOUBLE_EQ(imm.get_trust_radius(), 3.0); -} - -// Test the get_update_iter() function -TEST_F(IonsMoveMethodsTest, GetUpdateIter) -{ - imm.update_iter_ = 4; - - EXPECT_EQ(imm.get_update_iter(), 4); -} - -TEST_F(IonsMoveMethodsTest, ResetAfterCellChange) -{ - const std::string log_file = "reset_after_cell_change.log"; - std::ofstream ofs(log_file); - - imm.allocate(natom, "bfgs", "2"); - imm.converged_ = true; - imm.update_iter_ = 4; - imm.etot_info_ = {-1.0, -2.0}; - imm.bfgs.first_step = false; - imm.bfgs.save_flag = true; - imm.bfgs.tr_min_hit = true; - std::fill(imm.bfgs.pos.begin(), imm.bfgs.pos.end(), 1.0); - std::fill(imm.bfgs.pos_p.begin(), imm.bfgs.pos_p.end(), 2.0); - std::fill(imm.bfgs.grad.begin(), imm.bfgs.grad.end(), 3.0); - std::fill(imm.bfgs.grad_p.begin(), imm.bfgs.grad_p.end(), 4.0); - std::fill(imm.bfgs.move.begin(), imm.bfgs.move.end(), 5.0); - std::fill(imm.bfgs.move_p.begin(), imm.bfgs.move_p.end(), 6.0); - Ions_Move_Basic::trust_radius = 0.3; - Ions_Move_Basic::trust_radius_old = 0.2; - - imm.reset_after_cell_change({"bfgs", "2"}, ofs); - - EXPECT_FALSE(imm.converged_); - EXPECT_EQ(imm.update_iter_, 0); - EXPECT_THAT(imm.etot_info_, testing::Each(0.0)); - EXPECT_TRUE(imm.bfgs.first_step); - EXPECT_FALSE(imm.bfgs.save_flag); - EXPECT_FALSE(imm.bfgs.tr_min_hit); - EXPECT_THAT(imm.bfgs.pos, testing::Each(0.0)); - EXPECT_THAT(imm.bfgs.pos_p, testing::Each(0.0)); - EXPECT_THAT(imm.bfgs.grad, testing::Each(0.0)); - EXPECT_THAT(imm.bfgs.grad_p, testing::Each(0.0)); - EXPECT_THAT(imm.bfgs.move, testing::Each(0.0)); - EXPECT_THAT(imm.bfgs.move_p, testing::Each(0.0)); - for (int i = 0; i < Ions_Move_Basic::dim; ++i) - { - for (int j = 0; j < Ions_Move_Basic::dim; ++j) - { - EXPECT_DOUBLE_EQ(imm.bfgs.inv_hess(i, j), i == j ? 1.0 : 0.0); - } - } - EXPECT_DOUBLE_EQ(Ions_Move_Basic::trust_radius, 0.0); - EXPECT_DOUBLE_EQ(Ions_Move_Basic::trust_radius_old, 0.0); - - imm.allocate(natom, "bfgs", "1"); - imm.converged_ = true; - imm.update_iter_ = 3; - imm.etot_info_ = {-3.0, -4.0}; - ASSERT_TRUE(imm.bfgs_trad.is_initialized); - - imm.reset_after_cell_change({"bfgs", "1"}, ofs); - ofs.close(); - - EXPECT_FALSE(imm.converged_); - EXPECT_EQ(imm.update_iter_, 0); - EXPECT_THAT(imm.etot_info_, testing::Each(0.0)); - EXPECT_FALSE(imm.bfgs_trad.is_initialized); - - std::ifstream ifs(log_file); - const std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); - EXPECT_THAT(output, testing::HasSubstr("Reset ionic BFGS history after cell change.")); - EXPECT_THAT(output, testing::HasSubstr("Reset traditional ionic BFGS history after cell change.")); - ifs.close(); - std::remove(log_file.c_str()); -} diff --git a/source/source_relax/test/ions_move_sd_test.cpp b/source/source_relax/test/ions_move_sd_test.cpp deleted file mode 100644 index f31d3eeb125..00000000000 --- a/source/source_relax/test/ions_move_sd_test.cpp +++ /dev/null @@ -1,222 +0,0 @@ -#include "source_relax/relax_criteria.h" -#include -#include "for_test.h" -#include "gmock/gmock.h" -#include "gtest/gtest.h" -#include "source_relax/ions_move_basic.h" -#include "source_relax/ions_move_sd.h" - -/************************************************ - * unit tests of class Ions_Move_SD - ***********************************************/ - -class IonsMoveSDTest : public ::testing::Test -{ - public: - Relax_Criteria criteria; - - protected: - void SetUp() override - { - // Initialize variables before each test - Ions_Move_Basic::dim = 6; - update_iter = 5; - im_sd.allocate(); - criteria.force_thr = 0.001; - } - - void TearDown() override - { - // Clean up after each test - } - - Ions_Move_SD im_sd; - int update_iter; -}; - -// Test whether the allocate() function can correctly allocate memory space -TEST_F(IonsMoveSDTest, TestAllocate) -{ - Ions_Move_Basic::dim = 4; - im_sd.allocate(); - - // Check if allocated vectors are not empty - EXPECT_EQ(im_sd.get_grad_saved().size(), 4U); - EXPECT_EQ(im_sd.get_pos_saved().size(), 4U); -} - -// Test if a dimension less than or equal to 0 results in an assertion error -TEST_F(IonsMoveSDTest, TestAllocateWithZeroDimension) -{ - Ions_Move_Basic::dim = 0; - ASSERT_DEATH(im_sd.allocate(), ""); -} - -// Check that the arrays are correctly initialized to 0 -TEST_F(IonsMoveSDTest, TestAllocateAndInitialize) -{ - Ions_Move_Basic::dim = 4; - im_sd.allocate(); - - // Check that the arrays are correctly initialized to 0 - EXPECT_DOUBLE_EQ(im_sd.get_grad_saved()[0], 0.0); - EXPECT_DOUBLE_EQ(im_sd.get_pos_saved()[1], 0.0); - EXPECT_DOUBLE_EQ(im_sd.get_grad_saved()[2], 0.0); - EXPECT_DOUBLE_EQ(im_sd.get_pos_saved()[3], 0.0); -} - -// Test function start() when converged -TEST_F(IonsMoveSDTest, TestStartConverged) -{ - // setup data - const int istep = 1; - UnitCell ucell; - ModuleBase::matrix force(2, 3); - double etot = 0.0; - std::vector etot_info(2, 0.0); - - // call function - std::ofstream ofs("test_sd_start_converged.log"); - im_sd.start(ucell, force, etot, istep, update_iter, ofs, etot_info, criteria); - ofs.close(); - - // Check output - std::string expected_output = "\n Largest force is 0 eV/Angstrom while threshold is -1 eV/Angstrom\n" - " largest force is 0, no movement is possible.\n it may converged, otherwise no " - "movement of atom is allowed.\n end of geometry optimization\n " - " istep = 1\n update iteration = 5\n"; - std::ifstream ifs("test_sd_start_converged.log"); - std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); - ifs.close(); - std::remove("test_sd_start_converged.log"); - - std::regex pattern(R"(==> .*::.*\t[\d\.]+ GB\t\d+ s\n )"); - output = std::regex_replace(output, pattern, ""); - EXPECT_THAT(output, testing::HasSubstr(expected_output)); - EXPECT_EQ(update_iter, 5); - EXPECT_DOUBLE_EQ(Ions_Move_Basic::largest_grad, 0.0); - EXPECT_DOUBLE_EQ(im_sd.get_energy_saved(), 0.0); - EXPECT_DOUBLE_EQ(im_sd.get_pos_saved()[0], 0.0); - EXPECT_DOUBLE_EQ(im_sd.get_pos_saved()[1], 10.0); - EXPECT_DOUBLE_EQ(im_sd.get_pos_saved()[2], 20.0); - EXPECT_DOUBLE_EQ(im_sd.get_pos_saved()[3], 30.0); - EXPECT_DOUBLE_EQ(im_sd.get_pos_saved()[4], 40.0); - EXPECT_DOUBLE_EQ(im_sd.get_pos_saved()[5], 50.0); -} - -// Test function start() when nor converged -TEST_F(IonsMoveSDTest, TestStartNotConverged) -{ - // setup data - const int istep = 1; - UnitCell ucell; - ModuleBase::matrix force(2, 3); - force(0, 0) = 1.0; - double etot = 0.0; - std::vector etot_info(2, 0.0); - for (int it = 0; it < ucell.ntype; it++) - { - Atom* atom = &ucell.atoms[it]; - for (int ia = 0; ia < atom->na; ia++) - { - for (int ik = 0; ik < 3; ++ik) - { - atom->tau[ia][ik] = (ik + 1)/3; - atom->mbl[ia][ik] = 1; - } - } - } - - // call function - std::ofstream ofs("test_sd_start_not_converged.log"); - im_sd.start(ucell, force, etot, istep, update_iter, ofs, etot_info, criteria); - ofs.close(); - - // Check output - std::string expected_output = "\n Largest force is 25.7111 eV/Angstrom while threshold is -1 eV/Angstrom\n\n" - " Ion relaxation is not converged yet (threshold is 0.0257111)\n"; - std::ifstream ifs("test_sd_start_not_converged.log"); - std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); - ifs.close(); - std::remove("test_sd_start_not_converged.log"); - - EXPECT_THAT(output, testing::HasSubstr(expected_output)); - EXPECT_EQ(update_iter, 6); - EXPECT_DOUBLE_EQ(Ions_Move_Basic::largest_grad, 1.0); - EXPECT_DOUBLE_EQ(im_sd.get_energy_saved(), 0.0); - EXPECT_DOUBLE_EQ(im_sd.get_pos_saved()[0], -1.0); - EXPECT_DOUBLE_EQ(im_sd.get_pos_saved()[1], 0.0); - EXPECT_DOUBLE_EQ(im_sd.get_pos_saved()[2], 10.0); - EXPECT_DOUBLE_EQ(im_sd.get_pos_saved()[3], 0.0); - EXPECT_DOUBLE_EQ(im_sd.get_pos_saved()[4], 0.0); - EXPECT_DOUBLE_EQ(im_sd.get_pos_saved()[5], 10.0); - EXPECT_DOUBLE_EQ(im_sd.get_grad_saved()[0], -1.0); - EXPECT_DOUBLE_EQ(im_sd.get_grad_saved()[1], 0.0); - EXPECT_DOUBLE_EQ(im_sd.get_grad_saved()[2], 0.0); - EXPECT_DOUBLE_EQ(im_sd.get_grad_saved()[3], 0.0); - EXPECT_DOUBLE_EQ(im_sd.get_grad_saved()[4], 0.0); - EXPECT_DOUBLE_EQ(im_sd.get_grad_saved()[5], 0.0); -} - -// Test function cal_tradius_sd() case 1 -TEST_F(IonsMoveSDTest, CalTradiusSdCase1) -{ - // setup data - const int istep = 1; - criteria.out_level = "ie"; - std::vector etot_info(2, 0.0); - - // call function - testing::internal::CaptureStdout(); - ions_move_sd::cal_tradius_sd(istep, etot_info, criteria.out_level); - std::string std_outout = testing::internal::GetCapturedStdout(); - - // Check the results - std::string expected_std = " SD RADIUS (Bohr) : -1\n"; - EXPECT_EQ(expected_std, std_outout); - EXPECT_EQ(Ions_Move_Basic::trust_radius, -1.0); -} - -// Test function cal_tradius_sd() case 2 -TEST_F(IonsMoveSDTest, CalTradiusSdCase2) -{ - // setup data - const int istep = 2; - std::vector etot_info = {0.0, 1.0}; - criteria.out_level = "m"; - - // call function - ions_move_sd::cal_tradius_sd(istep, etot_info, criteria.out_level); - - // Check the results - EXPECT_EQ(Ions_Move_Basic::trust_radius, -1.0); -} - -// Test function cal_tradius_sd() case 3 -TEST_F(IonsMoveSDTest, CalTradiusSdCase3) -{ - // setup data - const int istep = 2; - std::vector etot_info = {1.0, 0.0}; - criteria.out_level = "m"; - - // call function - ions_move_sd::cal_tradius_sd(istep, etot_info, criteria.out_level); - - // Check the results - EXPECT_EQ(Ions_Move_Basic::trust_radius, -0.5); -} - -// Test function cal_tradius_sd() warning quit -TEST_F(IonsMoveSDTest, CalTradiusWraningQuit) -{ - // setup data - const int istep = 0; - std::vector etot_info(2, 0.0); - - // Check the results - testing::internal::CaptureStdout(); - EXPECT_EXIT(ions_move_sd::cal_tradius_sd(istep, etot_info, criteria.out_level), ::testing::ExitedWithCode(1), ""); - std::string output = testing::internal::GetCapturedStdout(); - EXPECT_THAT(output, testing::HasSubstr("istep < 1!")); -} diff --git a/source/source_relax/test/lattice_change_basic_test.cpp b/source/source_relax/test/lattice_change_basic_test.cpp deleted file mode 100644 index fb1e1193a71..00000000000 --- a/source/source_relax/test/lattice_change_basic_test.cpp +++ /dev/null @@ -1,1073 +0,0 @@ -#include -#include "source_relax/lattice_change_basic.h" -#include "source_relax/relax_data.h" -#include "mock_remake_cell.h" - -#include "for_test.h" -#include "gtest/gtest.h" -#include "gmock/gmock.h" -#include "source_io/module_parameter/parameter.h" - -/************************************************ - * unit tests of namespace Lattice_Change_Basic - ***********************************************/ - -// Define a fixture for the tests -class LatticeChangeBasicTest : public ::testing::Test -{ - public: - bool fixed_ibrav = false; - double stress_thr = 10.0; - - protected: - ModuleBase::matrix stress; - UnitCell ucell; - double lat[9], grad[9], move[9]; - - virtual void SetUp() - { - // Initialize variables before each test - stress.create(3, 3); - // Reset mock state before each test - unitcell::reset_remake_cell_mock(); - // Reset fixed_ibrav to default - fixed_ibrav = false; - } - - virtual void TearDown() - { - // Clean up after each test - unitcell::reset_remake_cell_mock(); - fixed_ibrav = false; - } -}; - -// Test the setup_gradient function with fixed_axes is volume -TEST_F(LatticeChangeBasicTest, SetupGradientVolume) -{ - // Initialize variables - ucell.lat_axis_free[0] = 1; - ucell.lat_axis_free[1] = 1; - ucell.lat_axis_free[2] = 1; - stress(0, 0) = 1.0; - stress(0, 1) = 2.0; - stress(0, 2) = 3.0; - stress(1, 0) = 4.0; - stress(1, 1) = 5.0; - stress(1, 2) = 6.0; - stress(2, 0) = 7.0; - stress(2, 1) = 8.0; - stress(2, 2) = 9.0; - - Lattice_Change_Basic::fixed_axes = "volume"; - - // Call setup_gradient method - Lattice_Change_Basic::setup_gradient(ucell, lat, grad, stress); - - // Check expected values for stress - EXPECT_DOUBLE_EQ(stress(0, 0), -4.0); - EXPECT_DOUBLE_EQ(stress(1, 1), 0.0); - EXPECT_DOUBLE_EQ(stress(2, 2), 4.0); - - // Check expected values for lat - EXPECT_DOUBLE_EQ(lat[0], 10.0); - EXPECT_DOUBLE_EQ(lat[1], 0.0); - EXPECT_DOUBLE_EQ(lat[2], 0.0); - EXPECT_DOUBLE_EQ(lat[3], 0.0); - EXPECT_DOUBLE_EQ(lat[4], 10.0); - EXPECT_DOUBLE_EQ(lat[5], 0.0); - EXPECT_DOUBLE_EQ(lat[6], 0.0); - EXPECT_DOUBLE_EQ(lat[7], 0.0); - EXPECT_DOUBLE_EQ(lat[8], 10.0); - - // Check expected values for grad - EXPECT_DOUBLE_EQ(grad[0], 40.0); - EXPECT_DOUBLE_EQ(grad[1], -20.0); - EXPECT_DOUBLE_EQ(grad[2], -30.0); - EXPECT_DOUBLE_EQ(grad[3], -40.0); - EXPECT_DOUBLE_EQ(grad[4], 0.0); - EXPECT_DOUBLE_EQ(grad[5], -60.0); - EXPECT_DOUBLE_EQ(grad[6], -70.0); - EXPECT_DOUBLE_EQ(grad[7], -80.0); - EXPECT_DOUBLE_EQ(grad[8], -40.0); -} - -// Test the setup_gradient function with fixed_axes is not volume -TEST_F(LatticeChangeBasicTest, SetupGradientNone) -{ - // Initialize variables - ucell.lat_axis_free[0] = 1; - ucell.lat_axis_free[1] = 1; - ucell.lat_axis_free[2] = 1; - stress(0, 0) = 1.0; - stress(0, 1) = 2.0; - stress(0, 2) = 3.0; - stress(1, 0) = 4.0; - stress(1, 1) = 5.0; - stress(1, 2) = 6.0; - stress(2, 0) = 7.0; - stress(2, 1) = 8.0; - stress(2, 2) = 9.0; - - Lattice_Change_Basic::fixed_axes = "None"; - - // Call setup_gradient method - Lattice_Change_Basic::setup_gradient(ucell, lat, grad, stress); - - // Check expected values for grad - EXPECT_DOUBLE_EQ(grad[0], -10.0); - EXPECT_DOUBLE_EQ(grad[1], -20.0); - EXPECT_DOUBLE_EQ(grad[2], -30.0); - EXPECT_DOUBLE_EQ(grad[3], -40.0); - EXPECT_DOUBLE_EQ(grad[4], -50.0); - EXPECT_DOUBLE_EQ(grad[5], -60.0); - EXPECT_DOUBLE_EQ(grad[6], -70.0); - EXPECT_DOUBLE_EQ(grad[7], -80.0); - EXPECT_DOUBLE_EQ(grad[8], -90.0); -} - -TEST_F(LatticeChangeBasicTest, ChangeLattice) -{ - // Initialize variables - ucell.lat_axis_free[0] = 1; - ucell.lat_axis_free[1] = 1; - ucell.lat_axis_free[2] = 1; - lat[0] = 1.0; - lat[1] = 0.0; - lat[2] = 0.0; - lat[3] = 0.0; - lat[4] = 2.0; - lat[5] = 0.0; - lat[6] = 0.0; - lat[7] = 0.0; - lat[8] = 3.0; - - move[0] = 1.0; - move[1] = 0.0; - move[2] = 0.0; - move[3] = 0.0; - move[4] = 2.0; - move[5] = 0.0; - move[6] = 0.0; - move[7] = 0.0; - move[8] = 3.0; - - // Call change_lattice method - Lattice_Change_Basic::change_lattice(ucell, move, lat, fixed_ibrav); - - // Check expected values for ucell after lattice change - EXPECT_DOUBLE_EQ(ucell.latvec.e11, 0.2); - EXPECT_DOUBLE_EQ(ucell.latvec.e12, 0.0); - EXPECT_DOUBLE_EQ(ucell.latvec.e13, 0.0); - EXPECT_DOUBLE_EQ(ucell.latvec.e21, 0.0); - EXPECT_DOUBLE_EQ(ucell.latvec.e22, 0.4); - EXPECT_DOUBLE_EQ(ucell.latvec.e23, 0.0); - EXPECT_DOUBLE_EQ(ucell.latvec.e31, 0.0); - EXPECT_DOUBLE_EQ(ucell.latvec.e32, 0.0); - EXPECT_DOUBLE_EQ(ucell.latvec.e33, 0.6); - - EXPECT_DOUBLE_EQ(ucell.a1.x, 0.2); - EXPECT_DOUBLE_EQ(ucell.a1.y, 0.0); - EXPECT_DOUBLE_EQ(ucell.a1.z, 0.0); - EXPECT_DOUBLE_EQ(ucell.a2.x, 0.0); - EXPECT_DOUBLE_EQ(ucell.a2.y, 0.4); - EXPECT_DOUBLE_EQ(ucell.a2.z, 0.0); - EXPECT_DOUBLE_EQ(ucell.a3.x, 0.0); - EXPECT_DOUBLE_EQ(ucell.a3.y, 0.0); - EXPECT_DOUBLE_EQ(ucell.a3.z, 0.6); - - EXPECT_DOUBLE_EQ(ucell.omega, 48.0); - - EXPECT_DOUBLE_EQ(ucell.GT.e11, 5.0); - EXPECT_DOUBLE_EQ(ucell.GT.e12, 0.0); - EXPECT_DOUBLE_EQ(ucell.GT.e13, 0.0); - EXPECT_DOUBLE_EQ(ucell.GT.e21, 0.0); - EXPECT_DOUBLE_EQ(ucell.GT.e22, 2.5); - EXPECT_DOUBLE_EQ(ucell.GT.e23, 0.0); - EXPECT_DOUBLE_EQ(ucell.GT.e31, 0.0); - EXPECT_DOUBLE_EQ(ucell.GT.e32, 0.0); - EXPECT_NEAR(ucell.GT.e33, 1.666666666666667, 1e-12); - - EXPECT_DOUBLE_EQ(ucell.G.e11, 5.0); - EXPECT_DOUBLE_EQ(ucell.G.e12, 0.0); - EXPECT_DOUBLE_EQ(ucell.G.e13, 0.0); - EXPECT_DOUBLE_EQ(ucell.G.e21, 0.0); - EXPECT_DOUBLE_EQ(ucell.G.e22, 2.5); - EXPECT_DOUBLE_EQ(ucell.G.e23, 0.0); - EXPECT_DOUBLE_EQ(ucell.G.e31, 0.0); - EXPECT_DOUBLE_EQ(ucell.G.e32, 0.0); - EXPECT_NEAR(ucell.G.e33, 1.666666666666667, 1e-12); - - EXPECT_DOUBLE_EQ(ucell.GGT.e11, 25.0); - EXPECT_DOUBLE_EQ(ucell.GGT.e12, 0.0); - EXPECT_DOUBLE_EQ(ucell.GGT.e13, 0.0); - EXPECT_DOUBLE_EQ(ucell.GGT.e21, 0.0); - EXPECT_DOUBLE_EQ(ucell.GGT.e22, 6.25); - EXPECT_DOUBLE_EQ(ucell.GGT.e23, 0.0); - EXPECT_DOUBLE_EQ(ucell.GGT.e31, 0.0); - EXPECT_DOUBLE_EQ(ucell.GGT.e32, 0.0); - EXPECT_NEAR(ucell.GGT.e33, 2.7777777777777786, 1e-12); - - EXPECT_DOUBLE_EQ(ucell.invGGT.e11, 0.04); - EXPECT_DOUBLE_EQ(ucell.invGGT.e12, 0.0); - EXPECT_DOUBLE_EQ(ucell.invGGT.e13, 0.0); - EXPECT_DOUBLE_EQ(ucell.invGGT.e21, 0.0); - EXPECT_DOUBLE_EQ(ucell.invGGT.e22, 0.16); - EXPECT_DOUBLE_EQ(ucell.invGGT.e23, 0.0); - EXPECT_DOUBLE_EQ(ucell.invGGT.e31, 0.0); - EXPECT_DOUBLE_EQ(ucell.invGGT.e32, 0.0); - EXPECT_DOUBLE_EQ(ucell.invGGT.e33, 0.36); -} - -// Test for check_converged when ucell.lat_axis_free[0] == 1 && ucell.lat_axis_free[1] == 1 && ucell.lat_axis_free[2] == 1, but not converged -TEST_F(LatticeChangeBasicTest, CheckConvergedCase1) -{ - // Set up test data - Lattice_Change_Basic::update_iter = 0; - stress_thr = 10.0; - std::ofstream ofs("test_check_converged_case1.log"); - ucell.lat_axis_free[0] = 1; - ucell.lat_axis_free[1] = 1; - ucell.lat_axis_free[2] = 1; - stress(0, 0) = 1.0; - stress(0, 1) = 2.0; - stress(0, 2) = 3.0; - stress(1, 0) = 4.0; - stress(1, 1) = 5.0; - stress(1, 2) = 6.0; - stress(2, 0) = 7.0; - stress(2, 1) = 8.0; - stress(2, 2) = 9.0; - - // Call the function under test - bool converged = Lattice_Change_Basic::check_converged(ucell, stress, grad, ofs, stress_thr); - ofs.close(); - - // Check the results - std::ifstream ifs("test_check_converged_case1.log"); - std::string expected_output = "\n Geometry relaxation is not converged because threshold is 10 kbar\n"; - std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); - EXPECT_EQ(output, expected_output); - EXPECT_EQ(Lattice_Change_Basic::update_iter, 0); - EXPECT_NEAR(Lattice_Change_Basic::largest_grad, 1323947.0517790401, 1e-12); - EXPECT_FALSE(converged); - - ifs.close(); - std::remove("test_check_converged_case1.log"); -} - -// Test for check_converged when ucell.lat_axis_free[0] == 1 && ucell.lat_axis_free[1] == 1 && ucell.lat_axis_free[2] == 1 && largest_grad == 0 -TEST_F(LatticeChangeBasicTest, CheckConvergedCase2) -{ - // Set up test data - Lattice_Change_Basic::update_iter = 0; - stress_thr = 10.0; - std::ofstream ofs("test_check_converged_case2.log"); - ucell.lat_axis_free[0] = 1; - ucell.lat_axis_free[1] = 1; - ucell.lat_axis_free[2] = 1; - stress(0, 0) = 0.0; - stress(0, 1) = 0.0; - stress(0, 2) = 0.0; - stress(1, 0) = 0.0; - stress(1, 1) = 0.0; - stress(1, 2) = 0.0; - stress(2, 0) = 0.0; - stress(2, 1) = 0.0; - stress(2, 2) = 0.0; - - // Call the function under test - bool converged = Lattice_Change_Basic::check_converged(ucell, stress, grad, ofs, stress_thr); - ofs.close(); - - // Check the results - std::ifstream ifs("test_check_converged_case2.log"); - std::string expected_output = " Largest stress is 0, movement is impossible.\n"; - std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); - EXPECT_EQ(output, expected_output); - EXPECT_EQ(Lattice_Change_Basic::update_iter, 0); - EXPECT_DOUBLE_EQ(Lattice_Change_Basic::largest_grad, 0.0); - EXPECT_TRUE(converged); - - ifs.close(); - std::remove("test_check_converged_case2.log"); -} - -// Test for check_converged when ucell.lat_axis_free[0] == 1 && ucell.lat_axis_free[1] == 1 && ucell.lat_axis_free[2] == 1, and converged -TEST_F(LatticeChangeBasicTest, CheckConvergedCase3) -{ - // Set up test data - Lattice_Change_Basic::update_iter = 0; - stress_thr = 10.0; - std::ofstream ofs("test_check_converged_case3.log"); - ucell.lat_axis_free[0] = 1; - ucell.lat_axis_free[1] = 1; - ucell.lat_axis_free[2] = 1; - stress(0, 0) = 0.000001; - stress(0, 1) = 0.0; - stress(0, 2) = 0.0; - stress(1, 0) = 0.0; - stress(1, 1) = 0.0; - stress(1, 2) = 0.0; - stress(2, 0) = 0.0; - stress(2, 1) = 0.0; - stress(2, 2) = 0.0; - - // Call the function under test - bool converged = Lattice_Change_Basic::check_converged(ucell, stress, grad, ofs, stress_thr); - ofs.close(); - - // Check the results - std::ifstream ifs("test_check_converged_case3.log"); - std::string expected_output = "\n Geometry relaxation is converged!\n\n Largest stress is 0.147105 kbar while threshold is 10 kbar\n"; - std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); - EXPECT_EQ(output, expected_output); - EXPECT_EQ(Lattice_Change_Basic::update_iter, 1); - EXPECT_NEAR(Lattice_Change_Basic::largest_grad, 0.14710522797544887, 1e-12); - EXPECT_TRUE(converged); - - ifs.close(); - std::remove("test_check_converged_case3.log"); -} - -// Test for check_converged when ucell.lat_axis_free != 1, but not converged -TEST_F(LatticeChangeBasicTest, CheckConvergedCase4) -{ - // Set up test data - Lattice_Change_Basic::update_iter = 0; - stress_thr = 10.0; - std::ofstream ofs("test_check_converged_case4.log"); - ucell.lat_axis_free[0] = 0; - ucell.lat_axis_free[1] = 0; - ucell.lat_axis_free[2] = 0; - grad[0] = 1.0; - grad[1] = 1.0; - grad[2] = 1.0; - grad[3] = 1.0; - grad[4] = 1.0; - grad[5] = 1.0; - grad[6] = 1.0; - grad[7] = 1.0; - grad[8] = 1.0; - - // Call the function under test - bool converged = Lattice_Change_Basic::check_converged(ucell, stress, grad, ofs, stress_thr); - ofs.close(); - - // Check the results - std::ifstream ifs("test_check_converged_case4.log"); - std::string expected_output = "\n Geometry relaxation is not converged because threshold is 10 kbar\n"; - std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); - EXPECT_EQ(output, expected_output); - EXPECT_EQ(Lattice_Change_Basic::update_iter, 0); - EXPECT_NEAR(Lattice_Change_Basic::largest_grad, 147105.22797544891, 1e-12); - EXPECT_FALSE(converged); - - ifs.close(); - std::remove("test_check_converged_case4.log"); -} - -// Test for check_converged when ucell.lat_axis_free != 1, and largest_grad == 0 -TEST_F(LatticeChangeBasicTest, CheckConvergedCase5) -{ - // Set up test data - Lattice_Change_Basic::update_iter = 0; - stress_thr = 10.0; - std::ofstream ofs("test_check_converged_case5.log"); - ucell.lat_axis_free[0] = 0; - ucell.lat_axis_free[1] = 0; - ucell.lat_axis_free[2] = 0; - grad[0] = 0.0; - grad[1] = 0.0; - grad[2] = 0.0; - grad[3] = 0.0; - grad[4] = 0.0; - grad[5] = 0.0; - grad[6] = 0.0; - grad[7] = 0.0; - grad[8] = 0.0; - - // Call the function under test - bool converged = Lattice_Change_Basic::check_converged(ucell, stress, grad, ofs, stress_thr); - ofs.close(); - - // Check the results - std::ifstream ifs("test_check_converged_case5.log"); - std::string expected_output = " Largest stress is 0, movement is impossible.\n"; - std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); - EXPECT_EQ(output, expected_output); - EXPECT_EQ(Lattice_Change_Basic::update_iter, 0); - EXPECT_DOUBLE_EQ(Lattice_Change_Basic::largest_grad, 0.0); - EXPECT_TRUE(converged); - - ifs.close(); - std::remove("test_check_converged_case5.log"); -} - -// Test for check_converged when ucell.lat_axis_free != 1, and converged -TEST_F(LatticeChangeBasicTest, CheckConvergedCase6) -{ - // Set up test data - Lattice_Change_Basic::update_iter = 0; - stress_thr = 10.0; - std::ofstream ofs("test_check_converged_case6.log"); - ucell.lat_axis_free[0] = 0; - ucell.lat_axis_free[1] = 0; - ucell.lat_axis_free[2] = 0; - grad[0] = 0.000001; - grad[1] = 0.0; - grad[2] = 0.0; - grad[3] = 0.0; - grad[4] = 0.0; - grad[5] = 0.0; - grad[6] = 0.0; - grad[7] = 0.0; - grad[8] = 0.0; - - // Call the function under test - bool converged = Lattice_Change_Basic::check_converged(ucell, stress, grad, ofs, stress_thr); - ofs.close(); - - // Check the results - std::ifstream ifs("test_check_converged_case6.log"); - std::string expected_output = "\n Geometry relaxation is converged!\n\n Largest stress is 0.147105 kbar while threshold is 10 kbar\n"; - std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); - EXPECT_EQ(output, expected_output); - EXPECT_EQ(Lattice_Change_Basic::update_iter, 1); - EXPECT_NEAR(Lattice_Change_Basic::largest_grad, 0.14710522797544887, 1e-12); - EXPECT_TRUE(converged); - - ifs.close(); - std::remove("test_check_converged_case6.log"); -} - -TEST_F(LatticeChangeBasicTest, TerminateConverged) -{ - Lattice_Change_Basic::stress_step = 5; - Lattice_Change_Basic::update_iter = 10; - - std::string expected_output = " end of lattice optimization\n stress_step = 5\n " - " update iteration = 10\n"; - - std::ofstream ofs("test_terminate_converged.log"); - Lattice_Change_Basic::terminate(true, ofs); - ofs.close(); - - std::ifstream ifs("test_terminate_converged.log"); - std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); - - EXPECT_EQ(expected_output, output); - ifs.close(); - std::remove("test_terminate_converged.log"); -} - -TEST_F(LatticeChangeBasicTest, TerminateNotConverged) -{ - std::string expected_output = " the maximum number of steps has been reached.\n end of lattice optimization.\n"; - - std::ofstream ofs("test_terminate_not_converged.log"); - Lattice_Change_Basic::terminate(false, ofs); - ofs.close(); - - std::ifstream ifs("test_terminate_not_converged.log"); - std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); - - EXPECT_EQ(expected_output, output); - ifs.close(); - std::remove("test_terminate_not_converged.log"); -} - - -TEST_F(LatticeChangeBasicTest, SetupEtot) -{ - Lattice_Change_Basic::stress_step = 2; - double energy_in = 80.0; - std::vector etot_info = {100.0, 90.0}; - - Lattice_Change_Basic::setup_etot(energy_in, etot_info); - - EXPECT_DOUBLE_EQ(100.0, etot_info[1]); - EXPECT_DOUBLE_EQ(80.0, etot_info[0]); - EXPECT_DOUBLE_EQ(-20.0, etot_info[0] - etot_info[1]); -} - -// ============================================================================ -// NEW TESTS FOR SHAPE CONSTRAINT, VOLUME RESCALING, AND FIXED_IBRAV -// ============================================================================ - -// Test the setup_gradient function with fixed_axes = "shape" -TEST_F(LatticeChangeBasicTest, SetupGradientShape) -{ - // Initialize variables - ucell.lat_axis_free[0] = 1; - ucell.lat_axis_free[1] = 1; - ucell.lat_axis_free[2] = 1; - - // Non-isotropic stress tensor - stress(0, 0) = 1.0; - stress(0, 1) = 2.0; - stress(0, 2) = 3.0; - stress(1, 0) = 4.0; - stress(1, 1) = 5.0; - stress(1, 2) = 6.0; - stress(2, 0) = 7.0; - stress(2, 1) = 8.0; - stress(2, 2) = 9.0; - - Lattice_Change_Basic::fixed_axes = "shape"; - - // Call setup_gradient method - Lattice_Change_Basic::setup_gradient(ucell, lat, grad, stress); - - // Check that stress becomes isotropic (only diagonal, all equal) - // Average pressure = (1 + 5 + 9) / 3 = 5 - EXPECT_DOUBLE_EQ(stress(0, 0), 5.0); - EXPECT_DOUBLE_EQ(stress(1, 1), 5.0); - EXPECT_DOUBLE_EQ(stress(2, 2), 5.0); - - // Off-diagonal elements should be zero - EXPECT_DOUBLE_EQ(stress(0, 1), 0.0); - EXPECT_DOUBLE_EQ(stress(0, 2), 0.0); - EXPECT_DOUBLE_EQ(stress(1, 0), 0.0); - EXPECT_DOUBLE_EQ(stress(1, 2), 0.0); - EXPECT_DOUBLE_EQ(stress(2, 0), 0.0); - EXPECT_DOUBLE_EQ(stress(2, 1), 0.0); -} - -// Test volume constraint rescaling in change_lattice -TEST_F(LatticeChangeBasicTest, ChangeLatticeVolumeRescaling) -{ - // Initialize variables - ucell.lat_axis_free[0] = 1; - ucell.lat_axis_free[1] = 1; - ucell.lat_axis_free[2] = 1; - ucell.lat0 = 10.0; - - // Set initial lattice (cubic, volume = 1000) - ucell.latvec.e11 = 1.0; - ucell.latvec.e12 = 0.0; - ucell.latvec.e13 = 0.0; - ucell.latvec.e21 = 0.0; - ucell.latvec.e22 = 1.0; - ucell.latvec.e23 = 0.0; - ucell.latvec.e31 = 0.0; - ucell.latvec.e32 = 0.0; - ucell.latvec.e33 = 1.0; - - ucell.omega = 1000.0; // Initial volume - - lat[0] = 10.0; - lat[1] = 0.0; - lat[2] = 0.0; - lat[3] = 0.0; - lat[4] = 10.0; - lat[5] = 0.0; - lat[6] = 0.0; - lat[7] = 0.0; - lat[8] = 10.0; - - // Apply a move that would change volume (expand by 10%) - move[0] = 1.0; - move[1] = 0.0; - move[2] = 0.0; - move[3] = 0.0; - move[4] = 1.0; - move[5] = 0.0; - move[6] = 0.0; - move[7] = 0.0; - move[8] = 1.0; - - Lattice_Change_Basic::fixed_axes = "volume"; - - // Call change_lattice method - Lattice_Change_Basic::change_lattice(ucell, move, lat, fixed_ibrav); - - // Check that volume is preserved (should still be 1000) - EXPECT_NEAR(ucell.omega, 1000.0, 1e-8); - - // Check that lattice vectors were rescaled uniformly - double expected_scale = std::pow(1000.0 / 1331.0, 1.0/3.0); // (old_vol / new_vol)^(1/3) - EXPECT_NEAR(ucell.latvec.e11, 1.1 * expected_scale, 1e-10); - EXPECT_NEAR(ucell.latvec.e22, 1.1 * expected_scale, 1e-10); - EXPECT_NEAR(ucell.latvec.e33, 1.1 * expected_scale, 1e-10); -} - -// Test volume constraint with non-cubic cell -TEST_F(LatticeChangeBasicTest, ChangeLatticeVolumeRescalingNonCubic) -{ - // Initialize variables - ucell.lat_axis_free[0] = 1; - ucell.lat_axis_free[1] = 1; - ucell.lat_axis_free[2] = 1; - ucell.lat0 = 10.0; - - // Set initial lattice (non-cubic, volume = 1200) - ucell.latvec.e11 = 1.0; - ucell.latvec.e12 = 0.0; - ucell.latvec.e13 = 0.0; - ucell.latvec.e21 = 0.0; - ucell.latvec.e22 = 1.2; - ucell.latvec.e23 = 0.0; - ucell.latvec.e31 = 0.0; - ucell.latvec.e32 = 0.0; - ucell.latvec.e33 = 1.0; - - ucell.omega = 1200.0; // Initial volume - - lat[0] = 10.0; - lat[1] = 0.0; - lat[2] = 0.0; - lat[3] = 0.0; - lat[4] = 12.0; - lat[5] = 0.0; - lat[6] = 0.0; - lat[7] = 0.0; - lat[8] = 10.0; - - // Apply a move that would change volume - move[0] = 0.5; - move[1] = 0.0; - move[2] = 0.0; - move[3] = 0.0; - move[4] = -0.5; - move[5] = 0.0; - move[6] = 0.0; - move[7] = 0.0; - move[8] = 0.3; - - Lattice_Change_Basic::fixed_axes = "volume"; - - // Call change_lattice method - Lattice_Change_Basic::change_lattice(ucell, move, lat, fixed_ibrav); - - // Check that volume is preserved - EXPECT_NEAR(ucell.omega, 1200.0, 1e-8); -} - -// Test change_lattice without volume constraint (should change volume) -TEST_F(LatticeChangeBasicTest, ChangeLatticeNoVolumeConstraint) -{ - // Initialize variables - ucell.lat_axis_free[0] = 1; - ucell.lat_axis_free[1] = 1; - ucell.lat_axis_free[2] = 1; - ucell.lat0 = 10.0; - - // Set initial lattice - ucell.latvec.e11 = 1.0; - ucell.latvec.e12 = 0.0; - ucell.latvec.e13 = 0.0; - ucell.latvec.e21 = 0.0; - ucell.latvec.e22 = 1.0; - ucell.latvec.e23 = 0.0; - ucell.latvec.e31 = 0.0; - ucell.latvec.e32 = 0.0; - ucell.latvec.e33 = 1.0; - - ucell.omega = 1000.0; // Initial volume - - lat[0] = 10.0; - lat[1] = 0.0; - lat[2] = 0.0; - lat[3] = 0.0; - lat[4] = 10.0; - lat[5] = 0.0; - lat[6] = 0.0; - lat[7] = 0.0; - lat[8] = 10.0; - - // Apply a move that changes volume - move[0] = 1.0; - move[1] = 0.0; - move[2] = 0.0; - move[3] = 0.0; - move[4] = 1.0; - move[5] = 0.0; - move[6] = 0.0; - move[7] = 0.0; - move[8] = 1.0; - - Lattice_Change_Basic::fixed_axes = "None"; - - // Call change_lattice method - Lattice_Change_Basic::change_lattice(ucell, move, lat, fixed_ibrav); - - // Check that volume DID change (should be 1331) - EXPECT_NEAR(ucell.omega, 1331.0, 1e-8); - - // Check lattice vectors - EXPECT_DOUBLE_EQ(ucell.latvec.e11, 1.1); - EXPECT_DOUBLE_EQ(ucell.latvec.e22, 1.1); - EXPECT_DOUBLE_EQ(ucell.latvec.e33, 1.1); -} - -// Test fixed_ibrav with simple cubic lattice -TEST_F(LatticeChangeBasicTest, ChangeLatticeFixedIbravSimpleCubic) -{ - // Initialize variables - ucell.lat_axis_free[0] = 1; - ucell.lat_axis_free[1] = 1; - ucell.lat_axis_free[2] = 1; - ucell.lat0 = 10.0; - ucell.latName = "sc"; - - // Set initial lattice (slightly distorted cubic) - ucell.latvec.e11 = 1.0; - ucell.latvec.e12 = 0.01; // Small distortion - ucell.latvec.e13 = 0.0; - ucell.latvec.e21 = 0.0; - ucell.latvec.e22 = 1.0; - ucell.latvec.e23 = 0.0; - ucell.latvec.e31 = 0.0; - ucell.latvec.e32 = 0.0; - ucell.latvec.e33 = 1.0; - - lat[0] = 10.0; - lat[1] = 0.1; - lat[2] = 0.0; - lat[3] = 0.0; - lat[4] = 10.0; - lat[5] = 0.0; - lat[6] = 0.0; - lat[7] = 0.0; - lat[8] = 10.0; - - // Apply a move - move[0] = 0.1; - move[1] = 0.0; - move[2] = 0.0; - move[3] = 0.0; - move[4] = 0.1; - move[5] = 0.0; - move[6] = 0.0; - move[7] = 0.0; - move[8] = 0.1; - - fixed_ibrav = true; - Lattice_Change_Basic::fixed_axes = "None"; - - // Verify remake_cell was not called yet - EXPECT_FALSE(unitcell::was_remake_cell_called()); - - // Call change_lattice method - Lattice_Change_Basic::change_lattice(ucell, move, lat, fixed_ibrav); - - // Verify remake_cell was called - EXPECT_TRUE(unitcell::was_remake_cell_called()); - - // Check that lattice is now perfect cubic (all diagonal, equal) - // This is enforced by the mock remake_cell function - EXPECT_NEAR(ucell.latvec.e11, ucell.latvec.e22, 1e-10); - EXPECT_NEAR(ucell.latvec.e22, ucell.latvec.e33, 1e-10); - EXPECT_NEAR(ucell.latvec.e12, 0.0, 1e-10); - EXPECT_NEAR(ucell.latvec.e13, 0.0, 1e-10); - EXPECT_NEAR(ucell.latvec.e21, 0.0, 1e-10); - EXPECT_NEAR(ucell.latvec.e23, 0.0, 1e-10); - EXPECT_NEAR(ucell.latvec.e31, 0.0, 1e-10); - EXPECT_NEAR(ucell.latvec.e32, 0.0, 1e-10); - - // Reset for other tests - fixed_ibrav = false; -} - -// Test fixed_ibrav with FCC lattice -TEST_F(LatticeChangeBasicTest, ChangeLatticeFixedIbravFCC) -{ - // Initialize variables - ucell.lat_axis_free[0] = 1; - ucell.lat_axis_free[1] = 1; - ucell.lat_axis_free[2] = 1; - ucell.lat0 = 10.0; - ucell.latName = "fcc"; - - // Set initial lattice (slightly distorted FCC) - double celldm = 1.0; - ucell.latvec.e11 = -celldm + 0.01; // Small distortion - ucell.latvec.e12 = 0.0; - ucell.latvec.e13 = celldm; - ucell.latvec.e21 = 0.0; - ucell.latvec.e22 = celldm; - ucell.latvec.e23 = celldm; - ucell.latvec.e31 = -celldm; - ucell.latvec.e32 = celldm; - ucell.latvec.e33 = 0.0; - - lat[0] = (-celldm + 0.01) * ucell.lat0; - lat[1] = 0.0; - lat[2] = celldm * ucell.lat0; - lat[3] = 0.0; - lat[4] = celldm * ucell.lat0; - lat[5] = celldm * ucell.lat0; - lat[6] = -celldm * ucell.lat0; - lat[7] = celldm * ucell.lat0; - lat[8] = 0.0; - - // Apply a small move - for (int i = 0; i < 9; i++) move[i] = 0.01 * ucell.lat0; - - fixed_ibrav = true; - Lattice_Change_Basic::fixed_axes = "None"; - - // Verify remake_cell was not called yet - EXPECT_FALSE(unitcell::was_remake_cell_called()); - - // Call change_lattice method - Lattice_Change_Basic::change_lattice(ucell, move, lat, fixed_ibrav); - - // Verify remake_cell was called - EXPECT_TRUE(unitcell::was_remake_cell_called()); - - // Check that lattice maintains FCC structure - // For FCC: a1 = (-a, 0, a), a2 = (0, a, a), a3 = (-a, a, 0) - // All should have same magnitude - double mag1 = std::sqrt(ucell.latvec.e11*ucell.latvec.e11 + - ucell.latvec.e12*ucell.latvec.e12 + - ucell.latvec.e13*ucell.latvec.e13); - double mag2 = std::sqrt(ucell.latvec.e21*ucell.latvec.e21 + - ucell.latvec.e22*ucell.latvec.e22 + - ucell.latvec.e23*ucell.latvec.e23); - double mag3 = std::sqrt(ucell.latvec.e31*ucell.latvec.e31 + - ucell.latvec.e32*ucell.latvec.e32 + - ucell.latvec.e33*ucell.latvec.e33); - - EXPECT_NEAR(mag1, mag2, 1e-10); - EXPECT_NEAR(mag2, mag3, 1e-10); - - // Check FCC structure: e11 should be negative, e13 positive, e12 = 0 - EXPECT_LT(ucell.latvec.e11, 0.0); - EXPECT_GT(ucell.latvec.e13, 0.0); - EXPECT_NEAR(ucell.latvec.e12, 0.0, 1e-10); - - // Reset for other tests - fixed_ibrav = false; -} - -// Test combination of fixed_axes = "volume" and fixed_ibrav -TEST_F(LatticeChangeBasicTest, ChangeLatticeVolumeAndIbrav) -{ - // Initialize variables - ucell.lat_axis_free[0] = 1; - ucell.lat_axis_free[1] = 1; - ucell.lat_axis_free[2] = 1; - ucell.lat0 = 10.0; - ucell.latName = "sc"; - - // Set initial lattice (cubic) - ucell.latvec.e11 = 1.0; - ucell.latvec.e12 = 0.0; - ucell.latvec.e13 = 0.0; - ucell.latvec.e21 = 0.0; - ucell.latvec.e22 = 1.0; - ucell.latvec.e23 = 0.0; - ucell.latvec.e31 = 0.0; - ucell.latvec.e32 = 0.0; - ucell.latvec.e33 = 1.0; - - ucell.omega = 1000.0; // Initial volume - - lat[0] = 10.0; - lat[1] = 0.0; - lat[2] = 0.0; - lat[3] = 0.0; - lat[4] = 10.0; - lat[5] = 0.0; - lat[6] = 0.0; - lat[7] = 0.0; - lat[8] = 10.0; - - // Apply a move that would change volume and break symmetry - move[0] = 1.0; - move[1] = 0.1; // Try to break symmetry - move[2] = 0.0; - move[3] = 0.0; - move[4] = 0.8; - move[5] = 0.0; - move[6] = 0.0; - move[7] = 0.0; - move[8] = 1.2; - - fixed_ibrav = true; - Lattice_Change_Basic::fixed_axes = "volume"; - - // Verify remake_cell was not called yet - EXPECT_FALSE(unitcell::was_remake_cell_called()); - - // Call change_lattice method - Lattice_Change_Basic::change_lattice(ucell, move, lat, fixed_ibrav); - - // Verify remake_cell was called (should be called before volume rescaling) - EXPECT_TRUE(unitcell::was_remake_cell_called()); - - // Check that volume is preserved - EXPECT_NEAR(ucell.omega, 1000.0, 1e-8); - - // Check that lattice is cubic (all diagonal, equal) - EXPECT_NEAR(ucell.latvec.e11, ucell.latvec.e22, 1e-10); - EXPECT_NEAR(ucell.latvec.e22, ucell.latvec.e33, 1e-10); - EXPECT_NEAR(ucell.latvec.e12, 0.0, 1e-10); - EXPECT_NEAR(ucell.latvec.e13, 0.0, 1e-10); - EXPECT_NEAR(ucell.latvec.e21, 0.0, 1e-10); - EXPECT_NEAR(ucell.latvec.e23, 0.0, 1e-10); - EXPECT_NEAR(ucell.latvec.e31, 0.0, 1e-10); - EXPECT_NEAR(ucell.latvec.e32, 0.0, 1e-10); - - // Reset for other tests - fixed_ibrav = false; -} - -// Test axis constraint with fixed_axes = "a" -TEST_F(LatticeChangeBasicTest, SetupGradientAxisA) -{ - // Initialize variables - ucell.lat_axis_free[0] = 0; // First axis fixed - ucell.lat_axis_free[1] = 1; - ucell.lat_axis_free[2] = 1; - - stress(0, 0) = 1.0; - stress(0, 1) = 2.0; - stress(0, 2) = 3.0; - stress(1, 0) = 4.0; - stress(1, 1) = 5.0; - stress(1, 2) = 6.0; - stress(2, 0) = 7.0; - stress(2, 1) = 8.0; - stress(2, 2) = 9.0; - - Lattice_Change_Basic::fixed_axes = "a"; - - // Call setup_gradient method - Lattice_Change_Basic::setup_gradient(ucell, lat, grad, stress); - - // Check that gradient for first lattice vector is zero - EXPECT_DOUBLE_EQ(grad[0], 0.0); - EXPECT_DOUBLE_EQ(grad[1], 0.0); - EXPECT_DOUBLE_EQ(grad[2], 0.0); - - // Check that gradients for other vectors are non-zero - EXPECT_NE(grad[3], 0.0); - EXPECT_NE(grad[4], 0.0); - EXPECT_NE(grad[5], 0.0); - EXPECT_NE(grad[6], 0.0); - EXPECT_NE(grad[7], 0.0); - EXPECT_NE(grad[8], 0.0); -} - -// Test that fixed axis doesn't move in change_lattice -TEST_F(LatticeChangeBasicTest, ChangeLatticeFixedAxisA) -{ - // Initialize variables - ucell.lat_axis_free[0] = 0; // First axis fixed - ucell.lat_axis_free[1] = 1; - ucell.lat_axis_free[2] = 1; - ucell.lat0 = 10.0; - - // Set initial lattice - ucell.latvec.e11 = 1.0; - ucell.latvec.e12 = 0.1; - ucell.latvec.e13 = 0.2; - ucell.latvec.e21 = 0.0; - ucell.latvec.e22 = 1.0; - ucell.latvec.e23 = 0.0; - ucell.latvec.e31 = 0.0; - ucell.latvec.e32 = 0.0; - ucell.latvec.e33 = 1.0; - - // Save initial first lattice vector - double initial_e11 = ucell.latvec.e11; - double initial_e12 = ucell.latvec.e12; - double initial_e13 = ucell.latvec.e13; - - lat[0] = 10.0; - lat[1] = 1.0; - lat[2] = 2.0; - lat[3] = 0.0; - lat[4] = 10.0; - lat[5] = 0.0; - lat[6] = 0.0; - lat[7] = 0.0; - lat[8] = 10.0; - - // Try to move all lattice vectors - move[0] = 1.0; - move[1] = 0.5; - move[2] = 0.3; - move[3] = 0.5; - move[4] = 1.0; - move[5] = 0.0; - move[6] = 0.0; - move[7] = 0.0; - move[8] = 1.0; - - Lattice_Change_Basic::fixed_axes = "a"; - - // Call change_lattice method - Lattice_Change_Basic::change_lattice(ucell, move, lat, fixed_ibrav); - - // Check that first lattice vector didn't change - EXPECT_DOUBLE_EQ(ucell.latvec.e11, initial_e11); - EXPECT_DOUBLE_EQ(ucell.latvec.e12, initial_e12); - EXPECT_DOUBLE_EQ(ucell.latvec.e13, initial_e13); - - // Check that other lattice vectors did change - EXPECT_NE(ucell.latvec.e21, 0.0); - EXPECT_NE(ucell.latvec.e22, 1.0); - EXPECT_NE(ucell.latvec.e33, 1.0); -} - -// Test that remake_cell is NOT called when fixed_ibrav = false -TEST_F(LatticeChangeBasicTest, ChangeLatticeNoFixedIbrav) -{ - // Initialize variables - ucell.lat_axis_free[0] = 1; - ucell.lat_axis_free[1] = 1; - ucell.lat_axis_free[2] = 1; - ucell.lat0 = 10.0; - ucell.latName = "sc"; - - // Set initial lattice - ucell.latvec.e11 = 1.0; - ucell.latvec.e12 = 0.01; // Small distortion - ucell.latvec.e13 = 0.0; - ucell.latvec.e21 = 0.0; - ucell.latvec.e22 = 1.0; - ucell.latvec.e23 = 0.0; - ucell.latvec.e31 = 0.0; - ucell.latvec.e32 = 0.0; - ucell.latvec.e33 = 1.0; - - lat[0] = 10.0; - lat[1] = 0.1; - lat[2] = 0.0; - lat[3] = 0.0; - lat[4] = 10.0; - lat[5] = 0.0; - lat[6] = 0.0; - lat[7] = 0.0; - lat[8] = 10.0; - - // Apply a move - move[0] = 0.1; - move[1] = 0.0; - move[2] = 0.0; - move[3] = 0.0; - move[4] = 0.1; - move[5] = 0.0; - move[6] = 0.0; - move[7] = 0.0; - move[8] = 0.1; - - fixed_ibrav = false; // Explicitly set to false - Lattice_Change_Basic::fixed_axes = "None"; - - // Verify remake_cell was not called yet - EXPECT_FALSE(unitcell::was_remake_cell_called()); - - // Call change_lattice method - Lattice_Change_Basic::change_lattice(ucell, move, lat, fixed_ibrav); - - // Verify remake_cell was NOT called - EXPECT_FALSE(unitcell::was_remake_cell_called()); - - // Check that distortion remains (e12 should still be non-zero) - EXPECT_NE(ucell.latvec.e12, 0.0); -} diff --git a/source/source_relax/test/lattice_change_cg_test.cpp b/source/source_relax/test/lattice_change_cg_test.cpp deleted file mode 100644 index e782082faca..00000000000 --- a/source/source_relax/test/lattice_change_cg_test.cpp +++ /dev/null @@ -1,576 +0,0 @@ -#include "source_relax/relax_criteria.h" -#include "for_test.h" -#include "gtest/gtest.h" -#include "mock_remake_cell.h" -#include "source_relax/lattice_change_basic.h" -#include "source_relax/lattice_change_cg.h" - -/************************************************ - * unit tests of class Lattice_Change_CG - ***********************************************/ - -class LatticeChangeCGTest : public ::testing::Test -{ - public: - Relax_Criteria criteria; - - protected: - void SetUp() override - { - // Initialize variables before each test - Lattice_Change_Basic::dim = 9; - Lattice_Change_Basic::stress_step = 1; - Lattice_Change_Basic::update_iter = 5; - lc_cg.allocate(); - etot_info.resize(2, 0.0); - } - - void TearDown() override - { - // Clean up after each test - } - - Lattice_Change_CG lc_cg; - std::vector etot_info; - - // Friendship is not inherited: a TEST_F body lives in a class derived from - // this fixture, so the internals of Lattice_Change_CG (which friends the - // fixture itself) are reached through these forwarders. - const std::vector& get_lat0() const - { - return lc_cg.lat0; - } - const std::vector& get_grad0() const - { - return lc_cg.grad0; - } - const std::vector& get_cg_grad0() const - { - return lc_cg.cg_grad0; - } - const std::vector& get_move0() const - { - return lc_cg.move0; - } - void set_move0(const int i, const double value) - { - lc_cg.move0[i] = value; - } -}; - -// Test whether the allocate() function can correctly allocate memory space -TEST_F(LatticeChangeCGTest, TestAllocate) -{ - Lattice_Change_Basic::dim = 4; - lc_cg.allocate(); - - // Check if allocated vectors are not empty - EXPECT_EQ(get_lat0().size(), 4U); - EXPECT_EQ(get_grad0().size(), 4U); - EXPECT_EQ(get_cg_grad0().size(), 4U); - EXPECT_EQ(get_move0().size(), 4U); -} - -// Test if a dimension less than or equal to 0 results in an assertion error -TEST_F(LatticeChangeCGTest, TestAllocateWithZeroDimension) -{ - Lattice_Change_Basic::dim = 0; - ASSERT_DEATH(lc_cg.allocate(), ""); -} - -// Check that the arrays are correctly initialized to 0 -TEST_F(LatticeChangeCGTest, TestAllocateAndInitialize) -{ - Lattice_Change_Basic::dim = 3; - lc_cg.allocate(); - - // Check that the arrays are correctly initialized to 0 - EXPECT_DOUBLE_EQ(0.0, get_lat0()[0]); - EXPECT_DOUBLE_EQ(0.0, get_grad0()[1]); - EXPECT_DOUBLE_EQ(0.0, get_cg_grad0()[2]); - EXPECT_DOUBLE_EQ(0.0, get_move0()[0]); -} - -// Test function start() when converged -TEST_F(LatticeChangeCGTest, TestStartConverged) -{ - // setup data - UnitCell ucell; - ucell.lat_axis_free[0] = 1; - ucell.lat_axis_free[1] = 1; - ucell.lat_axis_free[2] = 1; - ModuleBase::matrix stress(3, 3); - double etot = 0.0; - - // call function - std::ofstream ofs("test_lc_cg_start_converged.log"); - lc_cg.start(ucell, stress, etot, ofs, etot_info, criteria); - ofs.close(); - - // Check output - std::string expected_output - = " Largest stress is 0, movement is impossible.\n end of lattice optimization\n stress_step = 1\n " - " update iteration = 5\n"; - std::ifstream ifs("test_lc_cg_start_converged.log"); - std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); - - EXPECT_EQ(expected_output, output); - ifs.close(); - std::remove("test_lc_cg_start_converged.log"); -} - -// Test function start() sd branch -TEST_F(LatticeChangeCGTest, TestStartSd) -{ - // setup data - UnitCell ucell; - ucell.lat_axis_free[0] = 1; - ucell.lat_axis_free[1] = 1; - ucell.lat_axis_free[2] = 1; - ModuleBase::matrix stress(3, 3); - stress(0, 0) = 0.01; - double etot = 0.0; - - // call function - std::ofstream ofs("test_lc_cg_start_sd.log"); - lc_cg.start(ucell, stress, etot, ofs, etot_info, criteria); - ofs.close(); - - // Check output - std::string expected_output = "\n Geometry relaxation is not converged because threshold is 0.5 kbar\n"; - std::ifstream ifs("test_lc_cg_start_sd.log"); - std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); - - EXPECT_EQ(expected_output, output); - EXPECT_DOUBLE_EQ(Lattice_Change_Basic::lattice_change_ini, 0.01); - ifs.close(); - std::remove("test_lc_cg_start_sd.log"); -} - -// Test function start() trial branch with goto -TEST_F(LatticeChangeCGTest, TestStartTrialGoto) -{ - // setup data - UnitCell ucell; - ucell.lat_axis_free[0] = 1; - ucell.lat_axis_free[1] = 1; - ucell.lat_axis_free[2] = 1; - ModuleBase::matrix stress(3, 3); - stress(0, 1) = 0.01; - double etot = 0.0; - - // call function - set_move0(0, 1.0); - std::ofstream ofs1("test_lc_cg_start_trial_goto_temp1.log"); - lc_cg.start(ucell, stress, etot, ofs1, etot_info, criteria); - ofs1.close(); - std::remove("test_lc_cg_start_trial_goto_temp1.log"); - Lattice_Change_Basic::stress_step = 2; - set_move0(0, 10.0); - std::ofstream ofs("test_lc_cg_start_trial_goto.log"); - lc_cg.start(ucell, stress, etot, ofs, etot_info, criteria); - ofs.close(); - - // Check output - std::string expected_output = "\n Geometry relaxation is not converged because threshold is 0.5 kbar\n"; - std::ifstream ifs("test_lc_cg_start_trial_goto.log"); - std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); - - EXPECT_EQ(expected_output, output); - EXPECT_NEAR(Lattice_Change_Basic::lattice_change_ini, 10.000004999998749, 1e-12); - ifs.close(); - std::remove("test_lc_cg_start_trial_goto.log"); -} - -// Test function start() trial branch without goto -TEST_F(LatticeChangeCGTest, TestStartTrial) -{ - // setup data - UnitCell ucell; - ucell.lat_axis_free[0] = 1; - ucell.lat_axis_free[1] = 1; - ucell.lat_axis_free[2] = 1; - ModuleBase::matrix stress(3, 3); - stress(0, 1) = 0.01; - double etot = 0.0; - - // call function - std::ofstream ofs1("test_lc_cg_start_trial_temp1.log"); - lc_cg.start(ucell, stress, etot, ofs1, etot_info, criteria); - ofs1.close(); - std::remove("test_lc_cg_start_trial_temp1.log"); - Lattice_Change_Basic::stress_step = 2; - std::ofstream ofs("test_lc_cg_start_trial.log"); - lc_cg.start(ucell, stress, etot, ofs, etot_info, criteria); - ofs.close(); - - // Check output - std::string expected_output = "\n Geometry relaxation is not converged because threshold is 0.5 kbar\n"; - std::ifstream ifs("test_lc_cg_start_trial.log"); - std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); - - EXPECT_EQ(expected_output, output); - EXPECT_NEAR(Lattice_Change_Basic::lattice_change_ini, 70.000034999991243, 1e-12); - ifs.close(); - std::remove("test_lc_cg_start_trial.log"); -} - -// Test function start() no trial branch with goto case 1 -TEST_F(LatticeChangeCGTest, TestStartNoTrialGotoCase1) -{ - // setup data - UnitCell ucell; - ucell.lat_axis_free[0] = 1; - ucell.lat_axis_free[1] = 1; - ucell.lat_axis_free[2] = 1; - ModuleBase::matrix stress(3, 3); - stress(0, 1) = 0.01; - double etot = 0.0; - - // call function - std::ofstream ofs1("test_lc_cg_start_notrial_goto_case1_temp1.log"); - lc_cg.start(ucell, stress, etot, ofs1, etot_info, criteria); - ofs1.close(); - std::remove("test_lc_cg_start_notrial_goto_case1_temp1.log"); - Lattice_Change_Basic::stress_step = 2; - std::ofstream ofs2("test_lc_cg_start_notrial_goto_case1_temp2.log"); - lc_cg.start(ucell, stress, etot, ofs2, etot_info, criteria); - ofs2.close(); - std::remove("test_lc_cg_start_notrial_goto_case1_temp2.log"); - std::ofstream ofs("test_lc_cg_start_notrial_goto_case1.log"); - lc_cg.start(ucell, stress, etot, ofs, etot_info, criteria); - ofs.close(); - - // Check output - std::string expected_output = "\n Geometry relaxation is not converged because threshold is 0.5 kbar\n"; - std::ifstream ifs("test_lc_cg_start_notrial_goto_case1.log"); - std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); - - EXPECT_EQ(expected_output, output); - EXPECT_NEAR(Lattice_Change_Basic::lattice_change_ini, 490.00024499993867, 1e-12); - ifs.close(); - std::remove("test_lc_cg_start_notrial_goto_case1.log"); -} - -// Test function start() no trial branch with goto case 2 -TEST_F(LatticeChangeCGTest, TestStartNoTrialGotoCase2) -{ - // setup data - UnitCell ucell; - ucell.lat_axis_free[0] = 1; - ucell.lat_axis_free[1] = 1; - ucell.lat_axis_free[2] = 1; - ModuleBase::matrix stress(3, 3); - stress(0, 1) = 0.01; - double etot = 0.0; - - // call function - set_move0(0, 0.1); - std::ofstream ofs1("test_lc_cg_start_notrial_goto_case2_temp1.log"); - lc_cg.start(ucell, stress, etot, ofs1, etot_info, criteria); - ofs1.close(); - std::remove("test_lc_cg_start_notrial_goto_case2_temp1.log"); - Lattice_Change_Basic::stress_step = 2; - std::ofstream ofs2("test_lc_cg_start_notrial_goto_case2_temp2.log"); - lc_cg.start(ucell, stress, etot, ofs2, etot_info, criteria); - ofs2.close(); - std::remove("test_lc_cg_start_notrial_goto_case2_temp2.log"); - std::ofstream ofs("test_lc_cg_start_notrial_goto_case2.log"); - set_move0(0, 0.1); - stress(0, 1) = 0.0001; - lc_cg.start(ucell, stress, etot, ofs, etot_info, criteria); - ofs.close(); - - // Check output - std::string expected_output = "\n Geometry relaxation is not converged because threshold is 0.5 kbar\n"; - std::ifstream ifs("test_lc_cg_start_notrial_goto_case2.log"); - std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); - - EXPECT_EQ(expected_output, output); - EXPECT_NEAR(Lattice_Change_Basic::lattice_change_ini, 3430.0017149995706, 1e-12); - ifs.close(); - std::remove("test_lc_cg_start_notrial_goto_case2.log"); -} - -// Test function start() no trial branch without goto -TEST_F(LatticeChangeCGTest, TestStartNoTrial) -{ - // setup data - UnitCell ucell; - ucell.lat_axis_free[0] = 1; - ucell.lat_axis_free[1] = 1; - ucell.lat_axis_free[2] = 1; - ModuleBase::matrix stress(3, 3); - stress(0, 1) = 0.01; - double etot = 0.0; - - // call function - set_move0(0, 1.0); - std::ofstream ofs1("test_lc_cg_start_notrial_temp1.log"); - lc_cg.start(ucell, stress, etot, ofs1, etot_info, criteria); - ofs1.close(); - std::remove("test_lc_cg_start_notrial_temp1.log"); - Lattice_Change_Basic::stress_step = 2; - set_move0(0, 10.0); - std::ofstream ofs2("test_lc_cg_start_notrial_temp2.log"); - lc_cg.start(ucell, stress, etot, ofs2, etot_info, criteria); - ofs2.close(); - std::remove("test_lc_cg_start_notrial_temp2.log"); - std::ofstream ofs("test_lc_cg_start_notrial.log"); - lc_cg.start(ucell, stress, etot, ofs, etot_info, criteria); - ofs.close(); - - // Check output - std::string expected_output = "\n Geometry relaxation is not converged because threshold is 0.5 kbar\n"; - std::ifstream ifs("test_lc_cg_start_notrial.log"); - std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); - - EXPECT_EQ(expected_output, output); - EXPECT_NEAR(Lattice_Change_Basic::lattice_change_ini, 96040.106328872833, 1e-12); - ifs.close(); - std::remove("test_lc_cg_start_notrial.log"); -} - -// Test function setup_cg_grad() when ncggrad is multiple of 10000 -TEST_F(LatticeChangeCGTest, SetupCgGradNcggradIsMultipleOf10000) -{ - double grad[9] = {1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0}; - double grad0[9] = {1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0}; - double cggrad[9] = {9.0, 8.0, 7.0, 6.0, 5.0, 4.0, 3.0, 2.0, 1.0}; - double cggrad0[9] = {1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0}; - int ncggrad = 50000; // multiple of 10000 - int flag = 0; - - lc_cg.setup_cg_grad(Lattice_Change_Basic::dim, grad, grad0, cggrad, cggrad0, ncggrad, flag); - - EXPECT_DOUBLE_EQ(cggrad[0], grad[0]); - EXPECT_DOUBLE_EQ(cggrad[1], grad[1]); - EXPECT_DOUBLE_EQ(cggrad[2], grad[2]); - EXPECT_DOUBLE_EQ(cggrad[3], grad[3]); - EXPECT_DOUBLE_EQ(cggrad[4], grad[4]); - EXPECT_DOUBLE_EQ(cggrad[5], grad[5]); - EXPECT_DOUBLE_EQ(cggrad[6], grad[6]); - EXPECT_DOUBLE_EQ(cggrad[7], grad[7]); - EXPECT_DOUBLE_EQ(cggrad[8], grad[8]); -} - -// Test function setup_cg_grad() when ncggrad is not multiple of 10000, gamma1 < 0.5 -TEST_F(LatticeChangeCGTest, SetupCgGradNcggradIsNotMultipleOf10000Case1) -{ - double grad[9] = {1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0}; - double grad0[9] = {4.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0}; - double cggrad[9] = {4.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0}; - double cggrad0[9] = {4.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0}; - int ncggrad = 100; - int flag = 0; - - lc_cg.setup_cg_grad(Lattice_Change_Basic::dim, grad, grad0, cggrad, cggrad0, ncggrad, flag); - - EXPECT_DOUBLE_EQ(cggrad[0], 1.25); - EXPECT_DOUBLE_EQ(cggrad[1], 0.0); - EXPECT_DOUBLE_EQ(cggrad[2], 0.0); - EXPECT_DOUBLE_EQ(cggrad[3], 0.0); - EXPECT_DOUBLE_EQ(cggrad[4], 0.0); - EXPECT_DOUBLE_EQ(cggrad[5], 0.0); - EXPECT_DOUBLE_EQ(cggrad[6], 0.0); - EXPECT_DOUBLE_EQ(cggrad[7], 0.0); - EXPECT_DOUBLE_EQ(cggrad[8], 0.0); -} - -// Test function setup_cg_grad() when ncggrad is not multiple of 10000, gamma1 >= 0.5 -TEST_F(LatticeChangeCGTest, SetupCgGradNcggradIsNotMultipleOf10000Case2) -{ - double grad[9] = {1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0}; - double grad0[9] = {1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0}; - double cggrad[9] = {9.0, 8.0, 7.0, 6.0, 5.0, 4.0, 3.0, 2.0, 1.0}; - double cggrad0[9] = {1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0}; - int ncggrad = 100; - int flag = 0; - - lc_cg.setup_cg_grad(Lattice_Change_Basic::dim, grad, grad0, cggrad, cggrad0, ncggrad, flag); - - EXPECT_DOUBLE_EQ(cggrad[0], grad[0]); - EXPECT_DOUBLE_EQ(cggrad[1], grad[1]); - EXPECT_DOUBLE_EQ(cggrad[2], grad[2]); - EXPECT_DOUBLE_EQ(cggrad[3], grad[3]); - EXPECT_DOUBLE_EQ(cggrad[4], grad[4]); - EXPECT_DOUBLE_EQ(cggrad[5], grad[5]); - EXPECT_DOUBLE_EQ(cggrad[6], grad[6]); - EXPECT_DOUBLE_EQ(cggrad[7], grad[7]); - EXPECT_DOUBLE_EQ(cggrad[8], grad[8]); -} - -// Test function third_order() case 1 -TEST_F(LatticeChangeCGTest, ThirdOrderCase1) -{ - double e0 = 1.0; - double e1 = 1.0; - double fa = 10.0; - double fb = -9.99; - double x = 1.0; - double bestX = -1.0; // arbitrary initial value - - lc_cg.third_order(e0, e1, fa, fb, x, bestX); - - EXPECT_DOUBLE_EQ(bestX, x * fb / (fa - fb)); -} - -// Test function third_order() case 2 -TEST_F(LatticeChangeCGTest, ThirdOrderCase2) -{ - double e0 = 1.0; - double e1 = 1.0; - double fa = -10.0; - double fb = 9.9; - double x = 1.0; - double bestX = -1.0; // arbitrary initial value - - lc_cg.third_order(e0, e1, fa, fb, x, bestX); - - EXPECT_DOUBLE_EQ(bestX, x * fb / (fa - fb)); -} - -// Test function third_order() case 3 -TEST_F(LatticeChangeCGTest, ThirdOrderCase3) -{ - double e0 = 1.0; - double e1 = 1.0; - double fa = 10.0; - double fb = -10.1; - double x = 1.0; - double bestX = -1.0; // arbitrary initial value - - lc_cg.third_order(e0, e1, fa, fb, x, bestX); - - EXPECT_DOUBLE_EQ(bestX, x * fb / (fa - fb)); -} - -// Test function Brent() case 1 -TEST_F(LatticeChangeCGTest, BrentCase1) -{ - double fa = 2.0; - double fb = 1.0; - double fc = 1.0; - double xa = -3.0; - double xb = 2.0; - double xc = 1.0; - double best_x = 0.0; - double xpt = 0.0; - - lc_cg.Brent(fa, fb, fc, xa, xb, xc, best_x, xpt); - - EXPECT_DOUBLE_EQ(fa, 2.0); - EXPECT_DOUBLE_EQ(fb, 1.0); - EXPECT_DOUBLE_EQ(fc, 1.0); - EXPECT_DOUBLE_EQ(xa, -3.0); - EXPECT_DOUBLE_EQ(xb, 1.0); - EXPECT_DOUBLE_EQ(xc, 4.0); - EXPECT_DOUBLE_EQ(best_x, 4.0); - EXPECT_DOUBLE_EQ(xpt, 4.0); -} - -// Test function Brent() case 2 -TEST_F(LatticeChangeCGTest, BrentCase2) -{ - double fa = -2.0; - double fb = 3.0; - double fc = -4.0; - double xa = 1.0; - double xb = 2.0; - double xc = 3.0; - double best_x = 0.0; - double xpt = 0.0; - - lc_cg.Brent(fa, fb, fc, xa, xb, xc, best_x, xpt); - - EXPECT_DOUBLE_EQ(fa, -4.0); - EXPECT_DOUBLE_EQ(fb, 3.0); - EXPECT_DOUBLE_EQ(fc, -4.0); - EXPECT_DOUBLE_EQ(xa, 3.0); - EXPECT_DOUBLE_EQ(xb, 2.0); - EXPECT_NEAR(xc, 1.2046663545568725, 1e-12); - EXPECT_NEAR(best_x, 1.2046663545568725, 1e-12); - EXPECT_NEAR(xpt, 1.2046663545568725, 1e-12); -} - -// Test function Brent() case 3 -TEST_F(LatticeChangeCGTest, BrentCase3) -{ - double fa = 1.0; - double fb = -3.0; - double fc = -4.0; - double xa = 3.0; - double xb = 2.0; - double xc = 1.0; - double best_x = 0.0; - double xpt = 0.0; - - lc_cg.Brent(fa, fb, fc, xa, xb, xc, best_x, xpt); - - EXPECT_DOUBLE_EQ(fa, 1.0); - EXPECT_DOUBLE_EQ(fb, -4.0); - EXPECT_DOUBLE_EQ(fc, -4.0); - EXPECT_DOUBLE_EQ(xa, 3.0); - EXPECT_DOUBLE_EQ(xb, 1.0); - EXPECT_NEAR(xc, 2.8081429669660172, 1e-12); - EXPECT_NEAR(best_x, 2.8081429669660172, 1e-12); - EXPECT_NEAR(xpt, 2.8081429669660172, 1e-12); -} - -// Test function Brent() case 4 -TEST_F(LatticeChangeCGTest, BrentCase4) -{ - double fa = 2.0; - double fb = -3.0; - double fc = 4.0; - double xa = 0.0; - double xb = 2.0; - double xc = 1.0; - double best_x = 0.0; - double xpt = 0.0; - - lc_cg.Brent(fa, fb, fc, xa, xb, xc, best_x, xpt); - - EXPECT_DOUBLE_EQ(fa, 4.0); - EXPECT_DOUBLE_EQ(fb, -3.0); - EXPECT_DOUBLE_EQ(fc, 4.0); - EXPECT_DOUBLE_EQ(xa, 1.0); - EXPECT_DOUBLE_EQ(xb, 2.0); - EXPECT_DOUBLE_EQ(xc, 2.0); - EXPECT_DOUBLE_EQ(best_x, 2.0); - EXPECT_DOUBLE_EQ(xpt, 2.0); -} - -// Test function f_cal() -TEST_F(LatticeChangeCGTest, Fcal) -{ - Lattice_Change_Basic::dim = 9; - double g0[9] = {1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0}; - double g1[9] = {1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0}; - double f_value = 0.0; - - lc_cg.f_cal(Lattice_Change_Basic::dim, g0, g1, f_value); - - EXPECT_DOUBLE_EQ(f_value, 3.0); -} - -// Test function setup_move() -TEST_F(LatticeChangeCGTest, SetupMove) -{ - Lattice_Change_Basic::dim = 9; - double trust_radius = 1.0; - double cg_gradn[9] = {1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0}; - double move[9] = {0.0}; - - lc_cg.setup_move(Lattice_Change_Basic::dim, move, cg_gradn, trust_radius); - - EXPECT_DOUBLE_EQ(move[0], -1.0); - EXPECT_DOUBLE_EQ(move[1], -1.0); - EXPECT_DOUBLE_EQ(move[2], -1.0); - EXPECT_DOUBLE_EQ(move[3], -1.0); - EXPECT_DOUBLE_EQ(move[4], -1.0); - EXPECT_DOUBLE_EQ(move[5], -1.0); - EXPECT_DOUBLE_EQ(move[6], -1.0); - EXPECT_DOUBLE_EQ(move[7], -1.0); - EXPECT_DOUBLE_EQ(move[8], -1.0); -} diff --git a/source/source_relax/test/relax_test.cpp b/source/source_relax/test/relax_test.cpp deleted file mode 100644 index a5562b09ba1..00000000000 --- a/source/source_relax/test/relax_test.cpp +++ /dev/null @@ -1,352 +0,0 @@ -#include "gtest/gtest.h" -#include -#include "../relax_sync.h" -#include "source_cell/unitcell.h" -#include "source_io/module_parameter/input_parameter.h" -#include "relax_test.h" -#include - - -class Test_SETGRAD : public testing::Test -{ - protected: - Relax rl; - std::vector result; - Input_para inp; - UnitCell ucell; - std::ofstream ofs; - - void SetUp() - { - inp.force_thr = 0.001; - inp.calculation = "cell-relax"; - - ModuleBase::matrix force_in, stress_in; - int nat = 3; - - force_in.create(nat,3); - stress_in.create(3,3); - - force_in(0,0) = 0.1; force_in(0,1) = 0.1; force_in(0,2)= 0.1; - force_in(1,0) = 0; force_in(1,1) = 0.1; force_in(1,2)= 0.1; - force_in(2,0) = 0; force_in(2,1) = 0; force_in(2,2)= 0.1; - - stress_in(0,0) = 1; stress_in(0,1) = 1; stress_in(0,2)= 1; - stress_in(1,0) = 0; stress_in(1,1) = 1; stress_in(1,2)= 1; - stress_in(2,0) = 0; stress_in(2,1) = 0; stress_in(2,2)= 1; - - ucell.ntype = 1; - ucell.nat = nat; - ucell.atoms = new Atom[1]; - ucell.atoms[0].na = nat; - ucell.omega = 1.0; - ucell.lat0 = 1.0; - - ucell.iat2it = new int[nat]; - ucell.iat2ia = new int[nat]; - ucell.atoms[0].mbl.resize(nat); - ucell.atoms[0].taud.resize(nat); - ucell.atoms[0].tau.resize(nat); - ucell.atoms[0].dis.resize(nat); - ucell.atoms[0].mag.resize(nat); - ucell.atoms[0].vel.resize(nat); - - ucell.iat2it[0] = 0; - ucell.iat2it[1] = 0; - ucell.iat2it[2] = 0; - - ucell.iat2ia[0] = 0; - ucell.iat2ia[1] = 1; - ucell.iat2ia[2] = 2; - - ucell.atoms[0].mbl[0].x = 0; - ucell.atoms[0].mbl[0].y = 0; - ucell.atoms[0].mbl[0].z = 1; - - ucell.atoms[0].mbl[1].x = 0; - ucell.atoms[0].mbl[1].y = 1; - ucell.atoms[0].mbl[1].z = 0; - - ucell.atoms[0].mbl[2].x = 1; - ucell.atoms[0].mbl[2].y = 0; - ucell.atoms[0].mbl[2].z = 0; - - ucell.atoms[0].taud[0] = 0.0; - ucell.atoms[0].taud[1] = 0.0; - ucell.atoms[0].taud[2] = 0.0; - - ucell.atoms[0].tau.resize(nat); - - ucell.lat_axis_free[0] = 1; - ucell.lat_axis_free[1] = 1; - ucell.lat_axis_free[2] = 1; - - rl.init_relax(nat, inp); - rl.relax_step(ucell,force_in,stress_in,0.0, ofs); - - for(int i=0;i<3;i++) - { - result.push_back(ucell.atoms[0].taud[i].x); - result.push_back(ucell.atoms[0].taud[i].y); - result.push_back(ucell.atoms[0].taud[i].z); - } - push_result(); - - //reset lattice vector - ucell.latvec.Identity(); - inp.fixed_axes = "shape"; - rl.init_relax(nat, inp); - rl.relax_step(ucell,force_in,stress_in,0.0, ofs); - push_result(); - - //reset lattice vector - ucell.latvec.Identity(); - inp.fixed_axes = "volume"; - rl.init_relax(nat, inp); - rl.relax_step(ucell,force_in,stress_in,0.0, ofs); - push_result(); - - //reset lattice vector - ucell.latvec.Identity(); - inp.fixed_axes = "a"; //anything other than "None" - inp.fixed_ibrav = true; - ucell.latName = "sc"; - ucell.lat_axis_free[0] = 0; - ucell.lat_axis_free[1] = 0; - ucell.lat_axis_free[2] = 0; - rl.init_relax(nat, inp); - rl.relax_step(ucell,force_in,stress_in,0.0, ofs); - - push_result(); - } - - void push_result() - { - result.push_back(ucell.latvec.e11); - result.push_back(ucell.latvec.e12); - result.push_back(ucell.latvec.e13); - result.push_back(ucell.latvec.e21); - result.push_back(ucell.latvec.e22); - result.push_back(ucell.latvec.e23); - result.push_back(ucell.latvec.e31); - result.push_back(ucell.latvec.e32); - result.push_back(ucell.latvec.e33); - } - -}; - -TEST_F(Test_SETGRAD, relax_new) -{ - std::vector result_ref = - { - 0, 0, 0.24293434145, - 0, 0.242934341453, 0, - 0, 0, 0, - //paramter for taud - 1.2267616333,0.2267616333,0.22676163333, - 0, 1.2267616333 ,0.2267616333, - 0, 0, 1.22676163333, - // paramter for fisrt time after relaxation - 1.3677603495, 0, 0, - 0, 1.36776034956, 0, - 0, 0, 1.36776034956, - // paramter for second time after relaxation - 1.3677603495 ,0.3633367476,0.36333674766, - 0, 1.3677603495 ,0.36333674766, - 0, 0, 1.3677603495 , - // paramter for third time after relaxation - 1,0,0,0,1,0,0,0,1 - // paramter for fourth time after relaxation - }; - for(int i=0;i result; - Input_para inp; - UnitCell ucell; - std::ofstream ofs; - - void SetUp() - { - std::ifstream force_file("./support/force.txt"); - std::ifstream stress_file("./support/stress.txt"); - std::ifstream energy_file("./support/stress.txt"); - - int nstep = 3; - int nat = 5; - double energy; - inp.calculation = "cell-relax"; - inp.force_thr = 0.001; - inp.fixed_axes = "a"; - inp.stress_thr = 0.01; - inp.fixed_ibrav = false; - - this->setup_cell(); - - ModuleBase::matrix force_in, stress_in; - force_in.create(nat,3); - stress_in.create(3,3); - - rl.init_relax(nat, inp); - - for(int istep=0;istep> force_in(i,j); - } - } - for(int i=0;i<3;i++) - { - for(int j=0;j<3;j++) - { - stress_file >> stress_in(i,j); - } - } - - energy_file >> energy; - - inp.fixed_ibrav = false; - rl.relax_step(ucell,force_in,stress_in,energy, ofs); - result.push_back(ucell.atoms[0].taud[0].x); - result.push_back(ucell.atoms[0].taud[0].y); - result.push_back(ucell.atoms[0].taud[0].z); - result.push_back(ucell.atoms[1].taud[0].x); - result.push_back(ucell.atoms[1].taud[0].y); - result.push_back(ucell.atoms[1].taud[0].z); - result.push_back(ucell.atoms[2].taud[0].x); - result.push_back(ucell.atoms[2].taud[0].y); - result.push_back(ucell.atoms[2].taud[0].z); - result.push_back(ucell.atoms[2].taud[1].x); - result.push_back(ucell.atoms[2].taud[1].y); - result.push_back(ucell.atoms[2].taud[1].z); - result.push_back(ucell.atoms[2].taud[2].x); - result.push_back(ucell.atoms[2].taud[2].y); - result.push_back(ucell.atoms[2].taud[2].z); - result.push_back(ucell.latvec.e11); - result.push_back(ucell.latvec.e12); - result.push_back(ucell.latvec.e13); - result.push_back(ucell.latvec.e21); - result.push_back(ucell.latvec.e22); - result.push_back(ucell.latvec.e23); - result.push_back(ucell.latvec.e31); - result.push_back(ucell.latvec.e32); - result.push_back(ucell.latvec.e33); - } - } - - void setup_cell() - { - int ntype = 3, nat = 5; - ucell.ntype = ntype; - ucell.nat = nat; - - ucell.omega = 452.590903143121; - ucell.lat0 = 1.8897259886; - ucell.iat2it = new int[nat]; - ucell.iat2ia = new int[nat]; - ucell.iat2it[0] = 0; - ucell.iat2it[1] = 1; - ucell.iat2it[2] = 2; - ucell.iat2it[3] = 2; - ucell.iat2it[4] = 2; - - ucell.iat2ia[0] = 0; - ucell.iat2ia[1] = 0; - ucell.iat2ia[2] = 0; - ucell.iat2ia[3] = 1; - ucell.iat2ia[4] = 2; - - ucell.atoms = new Atom[ntype]; - ucell.atoms[0].na = 1; - ucell.atoms[1].na = 1; - ucell.atoms[2].na = 3; - - for(int i=0;i result_ref= - { - 0.5000000586,0.4999998876,0.009364595811, - 0.9999999281,1.901333279e-07,0.5476035454, - 3.782097706e-07,0.4999999285,0.4770375874, - 0.4999997826,2.072311863e-07,0.477037871, - 0.9999998523,0.9999997866,0.9349574003, - // paramter for taud after first relaxation - 4.006349654,-1.93128788e-07,5.793863639e-07, - -1.93128788e-07,4.006354579,-3.86257576e-07, - 6.757962549e-07,-4.505308366e-07,3.966870038, - // paramter for latvec after first relaxation - 0.5000000566,0.4999998916,0.009177239183, - 0.9999999308,1.832935626e-07,0.5467689737, - 3.647769323e-07,0.4999999311,0.4771204124, - 0.4999997903,1.998879545e-07,0.4771206859, - 0.9999998574,0.9999997943,0.9358136888, - // paramter for taud after second relaxation - 3.999761277,-1.656764727e-07,4.97029418e-07, - -1.656764727e-07,3.999765501,-3.313529453e-07, - 5.797351131e-07,-3.864900754e-07,4.010925071, - // paramter for latvec after second relaxation - 0.500000082,0.4999999574,0.01057784352, - 0.9999999149,1.939640249e-07,0.5455830599, - 3.795967781e-07,0.4999998795,0.4765373919, - 0.4999998037,2.756298268e-07,0.4765374602, - 0.9999998196,0.9999996936,0.9367652445, - // paramter for taud after third relaxation - 4.017733155,-1.420363309e-07,2.637046077e-07, - -1.420364243e-07,4.017735987,3.126225134e-07, - 3.479123171e-07,2.578467568e-07,4.011674933 - // paramter for latvec after third relaxation - }; - for(int i=0;i(ifs)), std::istreambuf_iterator()); + ifs.close(); + std::remove("test_log_update_inverse_hessian_case1.log"); + + EXPECT_EQ(expected_output, output); +} + +// Test function update_inverse_hessian() +TEST_F(BFGSBasicTest, UpdateInverseHessianCase2) +{ + Ions_Move_Basic::dim = 3; + double lat0 = 1.0; + bfgs.allocate_basic_for_testing(); + bfgs.get_pos()[0] = 2.0; + bfgs.get_grad()[0] = 2.0; + + std::ofstream ofs("test_log_update_inverse_hessian_case2.log"); + bfgs.update_inverse_hessian_for_testing(lat0, ofs); + ofs.close(); + + EXPECT_DOUBLE_EQ(bfgs.get_inv_hess()(0, 0), 0.5); + EXPECT_DOUBLE_EQ(bfgs.get_inv_hess()(0, 1), 0.0); + EXPECT_DOUBLE_EQ(bfgs.get_inv_hess()(0, 2), 0.0); + EXPECT_DOUBLE_EQ(bfgs.get_inv_hess()(1, 0), 0.0); + EXPECT_DOUBLE_EQ(bfgs.get_inv_hess()(1, 1), 0.0); + EXPECT_DOUBLE_EQ(bfgs.get_inv_hess()(1, 2), 0.0); + EXPECT_DOUBLE_EQ(bfgs.get_inv_hess()(2, 0), 0.0); + EXPECT_DOUBLE_EQ(bfgs.get_inv_hess()(2, 1), 0.0); + EXPECT_DOUBLE_EQ(bfgs.get_inv_hess()(2, 2), 0.0); + std::remove("test_log_update_inverse_hessian_case2.log"); +} + +// Test function check_wolfe_conditions() +TEST_F(BFGSBasicTest, CheckWolfeConditions) +{ + Ions_Move_Basic::dim = 3; + test_relax_method = 1; + bfgs.allocate_basic_for_testing(); + bfgs.get_pos()[0] = 2.0; + bfgs.get_grad()[0] = 2.0; + bfgs.get_move()[0] = 1.0; + std::vector etot_info = {10.0, 0.0}; + + std::ofstream ofs("test_log_check_wolfe_conditions.log"); + bfgs.check_wolfe_conditions_for_testing(ofs, etot_info); + ofs.close(); + + std::string expected_output + = " etot - etot_p = 10\n relax_bfgs_w1 * dot_p = -0\n " + " dot = 0\n relax_bfgs_w2 * dot_p = -0\n " + " relax_bfgs_w1 = -1\n relax_bfgs_w2 = -1\n " + "wolfe1 = 0\n wolfe2 = 0\n etot - etot_p = 10\n " + " relax_bfgs_w1 * dot_p = -0\n wolfe1 = 0\n " + " wolfe2 = 0\n wolfe condition satisfied = 0\n"; + + std::ifstream ifs("test_log_check_wolfe_conditions.log"); + std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + ifs.close(); + std::remove("test_log_check_wolfe_conditions.log"); + + EXPECT_EQ(bfgs.get_wolfe_flag(), false); + EXPECT_EQ(expected_output, output); +} + +// Test function reset_hessian() +TEST_F(BFGSBasicTest, ResetHessian) +{ + Ions_Move_Basic::dim = 3; + bfgs.allocate_basic_for_testing(); + + bfgs.reset_hessian_for_testing(); + + EXPECT_DOUBLE_EQ(bfgs.get_inv_hess()(0, 0), 1.0); + EXPECT_DOUBLE_EQ(bfgs.get_inv_hess()(0, 1), 0.0); + EXPECT_DOUBLE_EQ(bfgs.get_inv_hess()(0, 2), 0.0); + EXPECT_DOUBLE_EQ(bfgs.get_inv_hess()(1, 0), 0.0); + EXPECT_DOUBLE_EQ(bfgs.get_inv_hess()(1, 1), 1.0); + EXPECT_DOUBLE_EQ(bfgs.get_inv_hess()(1, 2), 0.0); + EXPECT_DOUBLE_EQ(bfgs.get_inv_hess()(2, 0), 0.0); + EXPECT_DOUBLE_EQ(bfgs.get_inv_hess()(2, 1), 0.0); + EXPECT_DOUBLE_EQ(bfgs.get_inv_hess()(2, 2), 1.0); +} + +// Test function save_bfgs() +TEST_F(BFGSBasicTest, SaveBfgs) +{ + Ions_Move_Basic::dim = 2; + bfgs.get_save_flag() = false; + bfgs.allocate_basic_for_testing(); + bfgs.get_pos()[0] = 1.0; + bfgs.get_pos()[1] = 2.0; + bfgs.get_grad()[0] = 3.0; + bfgs.get_grad()[1] = 4.0; + bfgs.get_move()[0] = 5.0; + bfgs.get_move()[1] = 6.0; + + bfgs.save_bfgs_for_testing(); + + EXPECT_EQ(bfgs.get_save_flag(), true); + EXPECT_DOUBLE_EQ(bfgs.get_pos()[0], 1.0); + EXPECT_DOUBLE_EQ(bfgs.get_pos()[1], 2.0); + EXPECT_DOUBLE_EQ(bfgs.get_grad()[0], 3.0); + EXPECT_DOUBLE_EQ(bfgs.get_grad()[1], 4.0); + EXPECT_DOUBLE_EQ(bfgs.get_move()[0], 5.0); + EXPECT_DOUBLE_EQ(bfgs.get_move()[1], 6.0); +} + +// Test function new_step() when update_iter == 1 +TEST_F(BFGSBasicTest, NewStepCase1) +{ + Ions_Move_Basic::dim = 2; + int update_iter = 0; + Ions_Move_Basic::largest_grad = 0.0; + Ions_Move_Basic::relax_bfgs_init = 0.3; + Ions_Move_Basic::best_xxx = -0.4; + bfgs.get_bfgs_ndim() = 1; + bfgs.allocate_basic_for_testing(); + bfgs.get_grad()[0] = 1.0; + bfgs.get_grad()[1] = 2.0; + bfgs.get_inv_hess()(0, 0) = -3.0; + bfgs.get_inv_hess()(0, 1) = -4.0; + bfgs.get_inv_hess()(1, 0) = -5.0; + bfgs.get_inv_hess()(1, 1) = -6.0; + + double lat0 = 1.0; + std::ofstream ofs("test_log.log"); + std::vector etot_info(2, 0.0); + bfgs.new_step_for_testing(lat0, update_iter, ofs, etot_info, test_relax_method); + + EXPECT_EQ(update_iter, 1); + EXPECT_EQ(bfgs.get_tr_min_hit(), false); + EXPECT_DOUBLE_EQ(Ions_Move_Basic::relax_bfgs_init, 0.2); + EXPECT_DOUBLE_EQ(Ions_Move_Basic::best_xxx, 0.4); + EXPECT_DOUBLE_EQ(Ions_Move_Basic::trust_radius, 0.2); + EXPECT_DOUBLE_EQ(bfgs.get_move()[0], -1.0); + EXPECT_DOUBLE_EQ(bfgs.get_move()[1], -2.0); + EXPECT_DOUBLE_EQ(bfgs.get_inv_hess()(0, 0), 1.0); + EXPECT_DOUBLE_EQ(bfgs.get_inv_hess()(0, 1), 0.0); + EXPECT_DOUBLE_EQ(bfgs.get_inv_hess()(1, 0), 0.0); + EXPECT_DOUBLE_EQ(bfgs.get_inv_hess()(1, 1), 1.0); +} + +// Test function new_step() when update_iter > 1 +TEST_F(BFGSBasicTest, NewStepCase2) +{ + Ions_Move_Basic::dim = 2; + int update_iter = 2; + Ions_Move_Basic::largest_grad = 0.0; + Ions_Move_Basic::relax_bfgs_init = 0.3; + Ions_Move_Basic::best_xxx = -0.4; + bfgs.get_bfgs_ndim() = 1; + bfgs.allocate_basic_for_testing(); + bfgs.get_grad()[0] = 1.0; + bfgs.get_grad()[1] = 2.0; + bfgs.get_inv_hess()(0, 0) = -3.0; + bfgs.get_inv_hess()(0, 1) = -4.0; + bfgs.get_inv_hess()(1, 0) = -5.0; + bfgs.get_inv_hess()(1, 1) = -6.0; + + double lat0 = 1.0; + std::ofstream ofs("test_log.log"); + std::vector etot_info(2, 0.0); + bfgs.new_step_for_testing(lat0, update_iter, ofs, etot_info, test_relax_method); + + EXPECT_EQ(update_iter, 3); + EXPECT_DOUBLE_EQ(Ions_Move_Basic::trust_radius, -1.0); + EXPECT_DOUBLE_EQ(bfgs.get_move()[0], -1.0); + EXPECT_DOUBLE_EQ(bfgs.get_move()[1], -2.0); + EXPECT_DOUBLE_EQ(bfgs.get_inv_hess()(0, 0), 1.0); + EXPECT_DOUBLE_EQ(bfgs.get_inv_hess()(0, 1), 0.0); + EXPECT_DOUBLE_EQ(bfgs.get_inv_hess()(1, 0), 0.0); + EXPECT_DOUBLE_EQ(bfgs.get_inv_hess()(1, 1), 1.0); +} + +// Test function new_step() when bfgs_ndim > 1 +TEST_F(BFGSBasicTest, NewStepWarningQuit) +{ + Ions_Move_Basic::dim = 2; + int update_iter = 0; + bfgs.get_bfgs_ndim() = 2; + bfgs.allocate_basic_for_testing(); + double lat0 = 1.0; + std::ofstream ofs("test_log.log"); + std::vector etot_info(2, 0.0); + + testing::internal::CaptureStdout(); + EXPECT_EXIT(bfgs.new_step_for_testing(lat0, update_iter, ofs, etot_info, test_relax_method), ::testing::ExitedWithCode(1), ""); + std::string output = testing::internal::GetCapturedStdout(); + EXPECT_THAT(output, testing::HasSubstr("bfgs_ndim > 1 not implemented yet")); +} + +// Test function compute_trust_radius() case 1 +TEST_F(BFGSBasicTest, ComputeTrustRadiusCase1) +{ + Ions_Move_Basic::dim = 2; + bfgs.allocate_basic_for_testing(); + bfgs.get_grad_p()[0] = 1.0; + bfgs.get_move_p()[1] = 2.0; + bfgs.get_inv_hess()(0, 0) = -3.0; + bfgs.get_inv_hess()(0, 1) = -4.0; + bfgs.get_inv_hess()(1, 0) = -5.0; + bfgs.get_inv_hess()(1, 1) = -6.0; + bfgs.get_wolfe_flag() = true; + bfgs.relax_bfgs_w1 = 1.0; + std::vector etot_info = {0.0, 0.0, 0.0}; + + std::ofstream ofs("test_log.log"); + bfgs.compute_trust_radius_for_testing(ofs, etot_info, test_relax_method); + + EXPECT_EQ(bfgs.get_tr_min_hit(), false); + EXPECT_DOUBLE_EQ(Ions_Move_Basic::trust_radius, -1.0); + EXPECT_DOUBLE_EQ(bfgs.get_move()[0], 0.0); + EXPECT_DOUBLE_EQ(bfgs.get_move()[1], 0.0); + EXPECT_DOUBLE_EQ(bfgs.get_inv_hess()(0, 0), -3.0); + EXPECT_DOUBLE_EQ(bfgs.get_inv_hess()(0, 1), -4.0); + EXPECT_DOUBLE_EQ(bfgs.get_inv_hess()(1, 0), -5.0); + EXPECT_DOUBLE_EQ(bfgs.get_inv_hess()(1, 1), -6.0); +} + +// Test function compute_trust_radius() case 2 +TEST_F(BFGSBasicTest, ComputeTrustRadiusCase2) +{ + Ions_Move_Basic::dim = 2; + Ions_Move_Basic::trust_radius_old = 0.0; + Ions_Move_Basic::relax_bfgs_rmin = 100.0; + test_relax_method = 1; + bfgs.allocate_basic_for_testing(); + bfgs.get_grad_p()[0] = 1.0; + bfgs.get_move()[1] = 2.0; + bfgs.get_move_p()[0] = 2.0; + bfgs.get_inv_hess()(0, 0) = -3.0; + bfgs.get_inv_hess()(0, 1) = -4.0; + bfgs.get_inv_hess()(1, 0) = -5.0; + bfgs.get_inv_hess()(1, 1) = -6.0; + bfgs.get_wolfe_flag() = false; + bfgs.relax_bfgs_w1 = 1.0; + bfgs.get_tr_min_hit() = false; + std::vector etot_info = {0.0, 0.0, 0.0}; + + std::ofstream ofs("test_log.log"); + bfgs.compute_trust_radius_for_testing(ofs, etot_info, test_relax_method); + + EXPECT_EQ(bfgs.get_tr_min_hit(), true); + EXPECT_DOUBLE_EQ(Ions_Move_Basic::trust_radius, 100.0); + EXPECT_DOUBLE_EQ(bfgs.get_move()[0], 0.0); + EXPECT_DOUBLE_EQ(bfgs.get_move()[1], 0.0); + EXPECT_DOUBLE_EQ(bfgs.get_inv_hess()(0, 0), 1.0); + EXPECT_DOUBLE_EQ(bfgs.get_inv_hess()(0, 1), 0.0); + EXPECT_DOUBLE_EQ(bfgs.get_inv_hess()(1, 0), 0.0); + EXPECT_DOUBLE_EQ(bfgs.get_inv_hess()(1, 1), 1.0); +} + +// Test function compute_trust_radius() warning_quit +TEST_F(BFGSBasicTest, ComputeTrustRadiusWarningQuit) +{ + Ions_Move_Basic::dim = 2; + Ions_Move_Basic::trust_radius_old = 0.0; + Ions_Move_Basic::relax_bfgs_rmin = 100.0; + test_relax_method = 1; + bfgs.allocate_basic_for_testing(); + bfgs.get_grad_p()[0] = 1.0; + bfgs.get_move()[1] = 2.0; + bfgs.get_move_p()[0] = 2.0; + bfgs.get_inv_hess()(0, 0) = -3.0; + bfgs.get_inv_hess()(0, 1) = -4.0; + bfgs.get_inv_hess()(1, 0) = -5.0; + bfgs.get_inv_hess()(1, 1) = -6.0; + bfgs.get_wolfe_flag() = false; + bfgs.relax_bfgs_w1 = 1.0; + bfgs.get_tr_min_hit() = true; + std::vector etot_info = {0.0, 0.0, 0.0}; + + std::ofstream ofs("test_log.log"); + testing::internal::CaptureStdout(); + EXPECT_EXIT(bfgs.compute_trust_radius_for_testing(ofs, etot_info, test_relax_method), ::testing::ExitedWithCode(1), ""); + std::string output = testing::internal::GetCapturedStdout(); + EXPECT_THAT(output, testing::HasSubstr("bfgs history already reset at previous step, we got trapped!")); +} \ No newline at end of file diff --git a/source/source_relax/unittests/test_cg_base.cpp b/source/source_relax/unittests/test_cg_base.cpp new file mode 100644 index 00000000000..72223537d09 --- /dev/null +++ b/source/source_relax/unittests/test_cg_base.cpp @@ -0,0 +1,409 @@ +#include "../cg_base.h" + +#include + +#include + +// Unit tests for CG_Base. normalize() is protected, so it is exercised +// through a test subclass that re-exposes it (instead of a +// #define protected public access hack). +class TestableCG : public CG_Base +{ + public: + using CG_Base::normalize; +}; + +TEST(CGBase, SetupCgGradRestartOnCounter) +{ + // ncggrad % 10000 == 0 triggers the restart branch: cg_grad = grad. + const int dim = 3; + double grad[dim] = {1.0, 2.0, 3.0}; + double grad0[dim] = {9.0, 9.0, 9.0}; + double cg_grad[dim] = {0.0, 0.0, 0.0}; + double cg_grad0[dim] = {7.0, 8.0, 9.0}; + int ncggrad = 0; + int flag = 0; + + TestableCG cg; + cg.setup_cg_grad(dim, grad, grad0, cg_grad, cg_grad0, ncggrad, flag); + + for (int i = 0; i < dim; ++i) + { + EXPECT_DOUBLE_EQ(cg_grad[i], grad[i]); + } +} + +TEST(CGBase, SetupCgGradRestartOnFlagTwo) +{ + // flag == 2 forces the restart branch regardless of ncggrad. + const int dim = 2; + double grad[dim] = {4.0, 5.0}; + double grad0[dim] = {1.0, 1.0}; + double cg_grad[dim] = {0.0, 0.0}; + double cg_grad0[dim] = {2.0, 2.0}; + int ncggrad = 5; // not a restart multiple + int flag = 2; + + TestableCG cg; + cg.setup_cg_grad(dim, grad, grad0, cg_grad, cg_grad0, ncggrad, flag); + + for (int i = 0; i < dim; ++i) + { + EXPECT_DOUBLE_EQ(cg_grad[i], grad[i]); + } +} + +TEST(CGBase, SetupCgGradFRBranch) +{ + // gamma1 = gg/gp_gp < 0.5 selects the Fletcher-Reeves coefficient. + // grad0 = {1,0}: gp_gp = 1. grad = {0.5,0}: gg = 0.25 -> gamma1 = 0.25. + const int dim = 2; + double grad[dim] = {0.5, 0.0}; + double grad0[dim] = {1.0, 0.0}; + double cg_grad[dim] = {0.0, 0.0}; + double cg_grad0[dim] = {2.0, 0.0}; + int ncggrad = 1; + int flag = 0; + + TestableCG cg; + cg.setup_cg_grad(dim, grad, grad0, cg_grad, cg_grad0, ncggrad, flag); + + const double gamma = 0.25; + EXPECT_DOUBLE_EQ(cg_grad[0], grad[0] + gamma * cg_grad0[0]); + EXPECT_DOUBLE_EQ(cg_grad[1], grad[1] + gamma * cg_grad0[1]); +} + +TEST(CGBase, SetupCgGradPRPBranch) +{ + // gamma1 = gg/gp_gp >= 0.5 selects the Polak-Ribiere coefficient + // gamma2 = (gg - g_gp)/gp_gp. + // grad0 = {1,0}: gp_gp = 1. grad = {1,0}: gg = 1, g_gp = 1 + // -> gamma1 = 1 (>= 0.5), gamma2 = 0. + const int dim = 2; + double grad[dim] = {1.0, 0.0}; + double grad0[dim] = {1.0, 0.0}; + double cg_grad[dim] = {0.0, 0.0}; + double cg_grad0[dim] = {3.0, 0.0}; + int ncggrad = 1; + int flag = 0; + + TestableCG cg; + cg.setup_cg_grad(dim, grad, grad0, cg_grad, cg_grad0, ncggrad, flag); + + const double gamma = 0.0; + EXPECT_DOUBLE_EQ(cg_grad[0], grad[0] + gamma * cg_grad0[0]); + EXPECT_DOUBLE_EQ(cg_grad[1], grad[1] + gamma * cg_grad0[1]); +} + +TEST(CGBase, FCalProjectsNormalized) +{ + // f_value = (g0 . g1) / |g0|. + const int dim = 2; + double g0[dim] = {3.0, 4.0}; // |g0| = 5 + double g1[dim] = {1.0, 0.0}; + double f_value = 0.0; + + TestableCG cg; + cg.f_cal(dim, g0, g1, f_value); + + EXPECT_DOUBLE_EQ(f_value, 0.6); // 3/5 +} + +TEST(CGBase, SetupMoveNegativeScaled) +{ + const int dim = 3; + double move[dim] = {0.0, 0.0, 0.0}; + double cg_gradn[dim] = {1.0, 0.0, -2.0}; + double trust_radius = 0.5; + + TestableCG cg; + cg.setup_move(dim, move, cg_gradn, trust_radius); + + EXPECT_DOUBLE_EQ(move[0], -0.5); + EXPECT_DOUBLE_EQ(move[1], 0.0); + EXPECT_DOUBLE_EQ(move[2], 1.0); +} + +TEST(CGBase, NormalizeUnitVector) +{ + const int dim = 2; + double cg_grad[dim] = {3.0, 4.0}; + double cg_gradn[dim] = {0.0, 0.0}; + + TestableCG cg; + cg.normalize(dim, cg_gradn, cg_grad); + + EXPECT_DOUBLE_EQ(cg_gradn[0], 0.6); + EXPECT_DOUBLE_EQ(cg_gradn[1], 0.8); + const double norm = std::sqrt(cg_gradn[0] * cg_gradn[0] + cg_gradn[1] * cg_gradn[1]); + EXPECT_DOUBLE_EQ(norm, 1.0); +} + +TEST(CGBase, NormalizeZeroVectorSafe) +{ + // A zero input must not divide by zero; cg_gradn is left untouched. + const int dim = 2; + double cg_grad[dim] = {0.0, 0.0}; + double cg_gradn[dim] = {1.0, 1.0}; + + TestableCG cg; + cg.normalize(dim, cg_gradn, cg_grad); + + EXPECT_DOUBLE_EQ(cg_gradn[0], 1.0); + EXPECT_DOUBLE_EQ(cg_gradn[1], 1.0); +} + +TEST(CGBase, BrentSameSignBranch) +{ + // fa*fb > 0: linear extrapolation, dmove = (xc*fa - xa*fc)/(fa - fc), + // best_x = dmove - xpt, then xpt = xc = dmove, xb/fb shift to xc/fc. + TestableCG cg; + double fa = 1.0; + double fb = 2.0; + double fc = 0.5; + double xa = 0.0; + double xb = 1.0; + double xc = 0.5; + double best_x = 0.0; + double xpt = 0.0; + + cg.Brent(fa, fb, fc, xa, xb, xc, best_x, xpt); + + const double dmove = (0.5 * 1.0 - 0.0 * 0.5) / (1.0 - 0.5); // = 1.0 + EXPECT_DOUBLE_EQ(best_x, dmove); // xpt was 0 + EXPECT_DOUBLE_EQ(xpt, dmove); + EXPECT_DOUBLE_EQ(xc, dmove); + EXPECT_DOUBLE_EQ(xb, 0.5); // old xc + EXPECT_DOUBLE_EQ(fb, 0.5); // old fc +} + +TEST(CGBase, BrentOppositeSignQuadratic) +{ + // fa*fb < 0 fits a quadratic through (xa,fa),(xb,fb),(xc,fc) and picks the + // stationary point with the lower integrated value. Only sanity-check the + // invariant: xpt and xc move to the returned displacement + old xpt. + TestableCG cg; + double fa = 1.0; + double fb = -1.0; + double fc = 0.5; + double xa = 0.0; + double xb = 1.0; + double xc = 0.5; + double best_x = 0.0; + double xpt = 0.0; + + cg.Brent(fa, fb, fc, xa, xb, xc, best_x, xpt); + + EXPECT_DOUBLE_EQ(xpt, best_x); // xpt = dmove = best_x + old xpt(0) + EXPECT_DOUBLE_EQ(xc, xpt); +} + +TEST(CGBase, ThirdOrderFallbackDmoveh) +{ + // When |k3/k1| < 0.01 the cubic term is negligible and the linear + // estimate dmoveh = x*fb/(fa - fb) is used. Choose fa ~ -fb so the cubic + // coefficient stays tiny. + TestableCG cg; + double best_x = 0.0; + const double x = 1.0; + const double fa = -1.0; + const double fb = 1.0; + // e0, e1 picked so that e1 - e0 = (fa+fb)*x/2 = 0 makes k3 = 0 exactly. + const double e0 = 0.0; + const double e1 = 0.0; + + cg.third_order(e0, e1, fa, fb, x, best_x); + + const double dmoveh = x * fb / (fa - fb); // = -0.5 + EXPECT_DOUBLE_EQ(best_x, dmoveh); +} + +TEST(CGBase, ThirdOrderCubicFit) +{ + // Choose inputs that take the genuine cubic-fit branch: k3 != 0, k3 > 0, + // a non-negative discriminant, and |k3/k1| >= 0.01 so neither fallback + // fires. With x=1, fa=-1, fb=1, e1-e0=-0.5: + // k3 = 3*((fb+fa)*x - 2*(e1-e0)) / x^3 = 3 + // k2 = (fb-fa)/x - k3*x = -1 + // k1 = fa = -1 + TestableCG cg; + double best_x = 0.0; + const double x = 1.0; + const double fa = -1.0; + const double fb = 1.0; + const double e0 = 0.0; + const double e1 = -0.5; + + cg.third_order(e0, e1, fa, fb, x, best_x); + + // best_x must be one of the two stationary points minus x, whichever gives + // the lower cubic energy. + const double k3 = 3.0; + const double k2 = -1.0; + const double k1 = -1.0; + const double disc = std::sqrt(1.0 - 4.0 * k1 * k3 / (k2 * k2)); // sqrt(13) + const double dmove1 = -k2 * (1.0 - disc) / (2.0 * k3); + const double dmove2 = -k2 * (1.0 + disc) / (2.0 * k3); + const double ecal1 = k3 * dmove1 * dmove1 * dmove1 / 3.0 + k2 * dmove1 * dmove1 / 2.0 + k1 * dmove1; + const double ecal2 = k3 * dmove2 * dmove2 * dmove2 / 3.0 + k2 * dmove2 * dmove2 / 2.0 + k1 * dmove2; + const double expected = (ecal2 > ecal1) ? (dmove1 - x) : (dmove2 - x); + + EXPECT_TRUE(std::isfinite(best_x)); + EXPECT_DOUBLE_EQ(best_x, expected); +} + +// --- Cases merged from test_ions_move_cg.cpp / test_lattice_change_cg.cpp --- +// These exercise the CG_Base line-search helpers directly (they were +// previously tested redundantly through the Ions_Move_CG and +// Lattice_Change_CG subclasses). References are pre-verified numerical +// outputs of the algorithm. + +TEST(CGBase, BrentSameSignMovesToXcCase) +{ + // fa, fb, fc all > 0 with fa*fb > 0: pure linear extrapolation moves the + // bracket to xc. + TestableCG cg; + double fa = 2.0; + double fb = 1.0; + double fc = 1.0; + double xa = -3.0; + double xb = 2.0; + double xc = 1.0; + double best_x = 0.0; + double xpt = 0.0; + + cg.Brent(fa, fb, fc, xa, xb, xc, best_x, xpt); + + EXPECT_DOUBLE_EQ(fa, 2.0); + EXPECT_DOUBLE_EQ(xb, 1.0); + EXPECT_DOUBLE_EQ(xc, 4.0); + EXPECT_DOUBLE_EQ(best_x, 4.0); + EXPECT_DOUBLE_EQ(xpt, 4.0); +} + +TEST(CGBase, BrentQuadraticCaseA) +{ + TestableCG cg; + double fa = -2.0; + double fb = 3.0; + double fc = -4.0; + double xa = 1.0; + double xb = 2.0; + double xc = 3.0; + double best_x = 0.0; + double xpt = 0.0; + + cg.Brent(fa, fb, fc, xa, xb, xc, best_x, xpt); + + EXPECT_DOUBLE_EQ(fa, -4.0); + EXPECT_DOUBLE_EQ(xa, 3.0); + EXPECT_DOUBLE_EQ(xb, 2.0); + EXPECT_NEAR(xc, 1.2046663545568725, 1e-12); + EXPECT_NEAR(best_x, 1.2046663545568725, 1e-12); + EXPECT_NEAR(xpt, 1.2046663545568725, 1e-12); +} + +TEST(CGBase, BrentQuadraticCaseB) +{ + TestableCG cg; + double fa = 1.0; + double fb = -3.0; + double fc = -4.0; + double xa = 3.0; + double xb = 2.0; + double xc = 1.0; + double best_x = 0.0; + double xpt = 0.0; + + cg.Brent(fa, fb, fc, xa, xb, xc, best_x, xpt); + + EXPECT_DOUBLE_EQ(fa, 1.0); + EXPECT_DOUBLE_EQ(xa, 3.0); + EXPECT_DOUBLE_EQ(xb, 1.0); + EXPECT_NEAR(xc, 2.8081429669660172, 1e-12); + EXPECT_NEAR(best_x, 2.8081429669660172, 1e-12); + EXPECT_NEAR(xpt, 2.8081429669660172, 1e-12); +} + +TEST(CGBase, BrentQuadraticCaseC) +{ + TestableCG cg; + double fa = 2.0; + double fb = -3.0; + double fc = 4.0; + double xa = 0.0; + double xb = 2.0; + double xc = 1.0; + double best_x = 0.0; + double xpt = 0.0; + + cg.Brent(fa, fb, fc, xa, xb, xc, best_x, xpt); + + EXPECT_DOUBLE_EQ(fa, 4.0); + EXPECT_DOUBLE_EQ(xa, 1.0); + EXPECT_DOUBLE_EQ(xb, 2.0); + EXPECT_DOUBLE_EQ(xc, 2.0); + EXPECT_DOUBLE_EQ(best_x, 2.0); + EXPECT_DOUBLE_EQ(xpt, 2.0); +} + +TEST(CGBase, ThirdOrderLinearFallbackPositiveFa) +{ + // |k3/k1| small: linear estimate dmoveh = x*fb/(fa-fb) is returned. + TestableCG cg; + double best_x = -1.0; + + cg.third_order(1.0, 1.0, 10.0, -9.99, 1.0, best_x); + + EXPECT_DOUBLE_EQ(best_x, 1.0 * -9.99 / (10.0 - -9.99)); +} + +TEST(CGBase, ThirdOrderLinearFallbackNegativeFa) +{ + TestableCG cg; + double best_x = -1.0; + + cg.third_order(1.0, 1.0, -10.0, 9.9, 1.0, best_x); + + EXPECT_DOUBLE_EQ(best_x, 1.0 * 9.9 / (-10.0 - 9.9)); +} + +TEST(CGBase, ThirdOrderLinearFallbackMixedSign) +{ + TestableCG cg; + double best_x = -1.0; + + cg.third_order(1.0, 1.0, 10.0, -10.1, 1.0, best_x); + + EXPECT_DOUBLE_EQ(best_x, 1.0 * -10.1 / (10.0 - -10.1)); +} + +TEST(CGBase, FCalUniformNine) +{ + // g0 = g1 = all-ones in 9 dims: f_value = (9)/sqrt(9) = 3. + TestableCG cg; + const int dim = 9; + double g0[dim] = {1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0}; + double g1[dim] = {1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0}; + double f_value = 0.0; + + cg.f_cal(dim, g0, g1, f_value); + + EXPECT_DOUBLE_EQ(f_value, 3.0); +} + +TEST(CGBase, SetupMoveUniformNine) +{ + TestableCG cg; + const int dim = 9; + double trust_radius = 1.0; + double cg_gradn[dim] = {1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0}; + double move[dim] = {0.0}; + + cg.setup_move(dim, move, cg_gradn, trust_radius); + + for (int i = 0; i < dim; ++i) + { + EXPECT_DOUBLE_EQ(move[i], -1.0); + } +} diff --git a/source/source_relax/unittests/test_ions_move_basic.cpp b/source/source_relax/unittests/test_ions_move_basic.cpp new file mode 100644 index 00000000000..a82e80672da --- /dev/null +++ b/source/source_relax/unittests/test_ions_move_basic.cpp @@ -0,0 +1,297 @@ +#include "gmock/gmock.h" +#include "gtest/gtest.h" +#include "source_relax/ions_move_basic.h" +#include "source_relax/relax_data.h" +#include "for_test.h" + +/************************************************ + * unit tests of namespace Ions_Move_Basic + ***********************************************/ + +// Define a fixture for the tests +class IonsMoveBasicTest : public ::testing::Test +{ + protected: + UnitCell ucell; + double move[6], pos[6], grad[6]; + ModuleBase::matrix force; + + virtual void SetUp() + { + // Initialize variables before each test + force.create(ucell.nat, 3); + force(0, 0) = 1.0; + force(0, 1) = 2.0; + force(0, 2) = 3.0; + force(1, 0) = 4.0; + force(1, 1) = 5.0; + force(1, 2) = 6.0; + } + + virtual void TearDown() + { + // Clean up after each test + } +}; + +// Test the setup_gradient() function +TEST_F(IonsMoveBasicTest, SetupGradient) +{ + // Call the function being tested + Ions_Move_Basic::dim = 6; + std::ofstream ofs("test_setup_gradient.log"); + Ions_Move_Basic::setup_gradient(ucell, force, pos, grad, ofs); + ofs.close(); + std::remove("test_setup_gradient.log"); + + // Check that the expected positions and gradients were generated + EXPECT_DOUBLE_EQ(pos[0], 0.0); + EXPECT_DOUBLE_EQ(pos[1], 10.0); + EXPECT_DOUBLE_EQ(pos[2], 20.0); + EXPECT_DOUBLE_EQ(pos[3], 30.0); + EXPECT_DOUBLE_EQ(pos[4], 40.0); + EXPECT_DOUBLE_EQ(pos[5], 50.0); + EXPECT_DOUBLE_EQ(grad[0], -10.0); + EXPECT_DOUBLE_EQ(grad[1], -20.0); + EXPECT_DOUBLE_EQ(grad[2], -30.0); + EXPECT_DOUBLE_EQ(grad[3], -40.0); + EXPECT_DOUBLE_EQ(grad[4], -50.0); + EXPECT_DOUBLE_EQ(grad[5], -60.0); +} + +// Test the move_atoms() function +TEST_F(IonsMoveBasicTest, MoveAtoms) +{ + // Initialize data + Ions_Move_Basic::dim = 6; + const int test_relax_method = 1; + for (int i = 0; i < Ions_Move_Basic::dim; ++i) + { + pos[i] = 0.0; + move[i] = i; + } + + // Call the function being tested + std::ofstream ofs("test_move_atoms.log"); + Ions_Move_Basic::move_atoms(ucell, move, pos, ofs, test_relax_method); + ofs.close(); + + // Check the results + std::ifstream ifs("test_move_atoms.log"); + std::string expected_output = "\n movement of ions (unit is Bohr) : \n Atom x y " + " z\n move_1 0 1 2\n " + "move_2 3 4 5\n"; + std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + ifs.close(); + std::remove("test_move_atoms.log"); + + EXPECT_THAT(output , ::testing::HasSubstr(expected_output)); + EXPECT_DOUBLE_EQ(pos[0], 0.0); + EXPECT_DOUBLE_EQ(pos[1], 1.0); + EXPECT_DOUBLE_EQ(pos[2], 2.0); + EXPECT_DOUBLE_EQ(pos[3], 3.0); + EXPECT_DOUBLE_EQ(pos[4], 4.0); + EXPECT_DOUBLE_EQ(pos[5], 5.0); +} + +// Test the check_converged() function case 1 +TEST_F(IonsMoveBasicTest, CheckConvergedCase1) +{ + // Initialize data + Ions_Move_Basic::dim = 6; + int update_iter = 1; + const int test_relax_method = 1; + const std::string out_level = "ie"; + const double force_thr = -1; // Input_para default; this test never set it + const double force_thr_ev = -1; // Input_para default; this test never set it + std::vector etot_info(2, 0.0); + for (int i = 0; i < Ions_Move_Basic::dim; ++i) + { + grad[i] = 0.0; + } + + // Call the function being tested + std::ofstream ofs("test_check_converged_case1.log"); + testing::internal::CaptureStdout(); + bool converged = Ions_Move_Basic::check_converged(ucell, grad, update_iter, ofs, etot_info, force_thr, force_thr_ev, out_level, test_relax_method); + std::string std_outout = testing::internal::GetCapturedStdout(); + ofs.close(); + + // Check the results + std::ifstream ifs("test_check_converged_case1.log"); + std::string ofs_output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + ifs.close(); + std::remove("test_check_converged_case1.log"); + + std::string expected_ofs + = " old total energy (ry) = 0\n new total energy (ry) = 0\n " + " energy difference (ry) = 0\n largest gradient (ry/bohr) = 0\n\n" + " largest force is 0, no " + "movement is possible.\n it may converged, otherwise no movement of atom is allowed.\n"; + std::string expected_std = " ETOT DIFF (eV) : 0\n LARGEST GRAD (eV/Angstrom) : 0\n"; + + EXPECT_THAT(ofs_output , ::testing::HasSubstr(expected_ofs)); + EXPECT_EQ(expected_std, std_outout); + EXPECT_EQ(update_iter, 1); + EXPECT_EQ(converged, true); + EXPECT_DOUBLE_EQ(Ions_Move_Basic::largest_grad, 0.0); +} + +// Test the check_converged() function case 2 +TEST_F(IonsMoveBasicTest, CheckConvergedCase2) +{ + // Initialize data + Ions_Move_Basic::dim = 6; + int update_iter = 1; + std::vector etot_info(2, 0.0); + const int test_relax_method = 1; + const std::string out_level = "ie"; + const double force_thr = 1.0; + const double force_thr_ev = -1; // Input_para default; this test never set it + grad[0] = 1.0; + + // Call the function being tested + std::ofstream ofs("test_check_converged_case2.log"); + testing::internal::CaptureStdout(); + bool converged = Ions_Move_Basic::check_converged(ucell, grad, update_iter, ofs, etot_info, force_thr, force_thr_ev, out_level, test_relax_method); + std::string std_outout = testing::internal::GetCapturedStdout(); + ofs.close(); + + // Check the results + std::ifstream ifs("test_check_converged_case2.log"); + std::string ofs_output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + ifs.close(); + std::remove("test_check_converged_case2.log"); + + std::string expected_ofs + = " old total energy (ry) = 0\n new total energy (ry) = 0\n " + " energy difference (ry) = 0\n largest gradient (ry/bohr) = 0.1\n"; + std::string expected_std = " ETOT DIFF (eV) : 0\n LARGEST GRAD (eV/Angstrom) : 2.57111\n"; + + EXPECT_THAT(ofs_output , ::testing::HasSubstr(expected_ofs)); + EXPECT_EQ(expected_std, std_outout); + EXPECT_EQ(update_iter, 2); + EXPECT_EQ(converged, true); + EXPECT_DOUBLE_EQ(Ions_Move_Basic::largest_grad, 0.1); +} + +// Test the check_converged() function case 3 +TEST_F(IonsMoveBasicTest, CheckConvergedCase3) +{ + // Initialize data + Ions_Move_Basic::dim = 6; + int update_iter = 1; + std::vector etot_info = {1.0, 0.0}; + const int test_relax_method = 1; + const std::string out_level = "ie"; + const double force_thr = 1.0; + const double force_thr_ev = -1; // Input_para default; this test never set it + grad[0] = 1.0; + + // Call the function being tested + std::ofstream ofs("test_check_converged_case3.log"); + testing::internal::CaptureStdout(); + bool converged = Ions_Move_Basic::check_converged(ucell, grad, update_iter, ofs, etot_info, force_thr, force_thr_ev, out_level, test_relax_method); + std::string std_outout = testing::internal::GetCapturedStdout(); + ofs.close(); + + // Check the results + std::ifstream ifs("test_check_converged_case3.log"); + std::string ofs_output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + ifs.close(); + std::remove("test_check_converged_case3.log"); + + std::string expected_ofs + = " old total energy (ry) = 0\n new total energy (ry) = 1\n " + " energy difference (ry) = 1\n largest gradient (ry/bohr) = 0.1\n"; + std::string expected_std = " ETOT DIFF (eV) : 13.6057\n LARGEST GRAD (eV/Angstrom) : 2.57111\n"; + + EXPECT_THAT(ofs_output , ::testing::HasSubstr(expected_ofs)); + EXPECT_EQ(expected_std, std_outout); + EXPECT_EQ(update_iter, 1); + EXPECT_EQ(converged, false); + EXPECT_DOUBLE_EQ(Ions_Move_Basic::largest_grad, 0.1); +} + +// Test the terminate() function when converged +TEST_F(IonsMoveBasicTest, TerminateConverged) +{ + // Initialize data + const bool converged = true; + const int istep = 2; + const int update_iter = 5; + + // Call the function being tested + std::ofstream ofs("test_terminate_converged.log"); + Ions_Move_Basic::terminate(converged, update_iter, ucell, istep, ofs); + ofs.close(); + + // Check the results + std::ifstream ifs("test_terminate_converged.log"); + std::string ofs_output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + ifs.close(); + std::remove("test_terminate_converged.log"); + + std::string expected_ofs = " end of geometry optimization\n istep = 2\n " + " update iteration = 5\n"; + + EXPECT_THAT(ofs_output , ::testing::HasSubstr(expected_ofs)); +} + +// Test the terminate() function when not converged +TEST_F(IonsMoveBasicTest, TerminateNotConverged) +{ + // Initialize data + const bool converged = false; + const int istep = 10; + const int update_iter = 0; + + // Call the function being tested + std::ofstream ofs("test_terminate_not_converged.log"); + Ions_Move_Basic::terminate(converged, update_iter, ucell, istep, ofs); + ofs.close(); + + // Check the results + std::ifstream ifs("test_terminate_not_converged.log"); + std::string ofs_output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + ifs.close(); + std::remove("test_terminate_not_converged.log"); + + std::string expected_ofs = " the maximum number of steps has been reached.\n Relaxation is not converged after reaching relax_nmax!\n end of geometry optimization.\n"; + + EXPECT_THAT(ofs_output , ::testing::HasSubstr(expected_ofs)); +} + +// Test the setup_etot() function +TEST_F(IonsMoveBasicTest, SetupEtot) +{ + // Initialize data + const int istep = 2; + std::vector etot_info = {2.0, 1.0}; + double energy_in = 3.0; + + // Call the function being tested + std::ofstream ofs("/dev/null"); + Ions_Move_Basic::setup_etot(energy_in, istep, etot_info); + ofs.close(); + + // Check the results + EXPECT_DOUBLE_EQ(etot_info[1], 2.0); + EXPECT_DOUBLE_EQ(etot_info[0], 3.0); + EXPECT_DOUBLE_EQ(etot_info[0] - etot_info[1], 1.0); +} + +// Test the dot_func() function +TEST_F(IonsMoveBasicTest, DotFunc) +{ + // Initialize data + double dim_in = 3; + double a[3] = {1.0, 2.0, 3.0}; + double b[3] = {1.0, 2.0, 3.0}; + + // Call the function being tested + double result = Ions_Move_Basic::dot_func(a, b, dim_in); + + // Check the results + EXPECT_DOUBLE_EQ(result, 14.0); +} diff --git a/source/source_relax/unittests/test_ions_move_bfgs.cpp b/source/source_relax/unittests/test_ions_move_bfgs.cpp new file mode 100644 index 00000000000..cd634686a6c --- /dev/null +++ b/source/source_relax/unittests/test_ions_move_bfgs.cpp @@ -0,0 +1,464 @@ +#include "source_relax/relax_criteria.h" +#include "for_test.h" +#include "gtest/gtest.h" +#include "gmock/gmock.h" +#include "source_io/module_parameter/parameter.h" +#include "source_relax/ions_move_basic.h" +#include "source_relax/ions_move_bfgs.h" + +/************************************************ + * unit tests of class Ions_Move_BFGS + ***********************************************/ + +// Define a fixture for the tests +class IonsMoveBFGSTest : public ::testing::Test +{ + public: + Relax_Criteria criteria; + + protected: + Ions_Move_BFGS bfgs; + int update_iter; + + virtual void SetUp() + { + // Initialize variables before each test + Ions_Move_Basic::dim = 6; + update_iter = 0; + } + + virtual void TearDown() + { + // Clean up after each test + } +}; + +// Test the allocate() function case 1 +TEST_F(IonsMoveBFGSTest, AllocateCase1) +{ + // Initilize data + bfgs.get_init_done() = true; + bfgs.get_save_flag() = true; + + // Call the function being tested + bfgs.allocate(); + + // Check that the expected results + EXPECT_EQ(bfgs.get_init_done(), true); + EXPECT_EQ(bfgs.get_save_flag(), true); +} + +// Test the allocate() function case 2 +TEST_F(IonsMoveBFGSTest, AllocateCase2) +{ + // Initilize data + bfgs.get_init_done() = false; + bfgs.get_save_flag() = true; + + // Call the function being tested + bfgs.allocate(); + + // Check that the expected results + EXPECT_EQ(bfgs.get_init_done(), true); + EXPECT_EQ(bfgs.get_save_flag(), false); +} + +// Test the start() function case 1 +TEST_F(IonsMoveBFGSTest, StartCase1) +{ + // Initilize data + UnitCell ucell; + ModuleBase::matrix force(2, 3); + double energy_in = 0.0; + const int istep = 1; + bfgs.get_init_done() = false; + bfgs.get_save_flag() = true; + std::vector etot_info(2, 0.0); + + // Call the function being tested + bfgs.allocate(); + std::ofstream ofs("test_start_case1.log"); + bfgs.start(ucell, force, energy_in, istep, update_iter, ofs, etot_info, criteria); + ofs.close(); + + // Check the results + std::ifstream ifs("test_start_case1.log"); + std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + ifs.close(); + std::remove("test_start_case1.log"); + + EXPECT_THAT(output, testing::HasSubstr("update iteration")); +} + +// Test the start() function case 2 +TEST_F(IonsMoveBFGSTest, StartCase2) +{ + // Initilize data + UnitCell ucell; + // Initialize UnitCell with 2 atoms + ucell.ntype = 1; + ucell.nat = 2; + ucell.atoms = new Atom[ucell.ntype]; + ucell.atoms[0].na = 2; + ucell.atoms[0].tau = std::vector>(2); + ucell.atoms[0].taud = std::vector>(2); + ucell.atoms[0].mbl = std::vector>(2, {1, 1, 1}); + ucell.atoms[0].tau[0].x = 0.0; ucell.atoms[0].tau[0].y = 0.0; ucell.atoms[0].tau[0].z = 0.0; + ucell.atoms[0].tau[1].x = 1.0; ucell.atoms[0].tau[1].y = 0.0; ucell.atoms[0].tau[1].z = 0.0; + ucell.lat0 = 1.0; + ucell.set_atom_flag = true; + + // Initialize PARAM + criteria.force_thr = 1.0e-3; + criteria.force_thr_ev = criteria.force_thr * 13.6058 / 0.529177; + criteria.test_relax_method = 1; + criteria.out_level = "ie"; + + // Initialize istep + const int istep = 1; + + ModuleBase::matrix force(2, 3); + force(0, 0) = 10.0; + double energy_in = 0.0; + bfgs.get_init_done() = false; + bfgs.get_save_flag() = true; + std::vector etot_info(2, 0.0); + + // Call the function being tested + bfgs.allocate(); + std::ofstream ofs("test_start_case2.log"); + bfgs.start(ucell, force, energy_in, istep, update_iter, ofs, etot_info, criteria); + ofs.close(); + + // Check the results + std::ifstream ifs("test_start_case2.log"); + std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + ifs.close(); + std::remove("test_start_case2.log"); + + // Reporting moved to IonCellOptimizer::relax_step; check_converged no longer + // prints convergence info to the log. Assert state instead. + EXPECT_THAT(output, testing::Not(testing::HasSubstr("Ion relaxation is not converged yet"))); + EXPECT_DOUBLE_EQ(Ions_Move_Basic::largest_grad, 10.0); + + // Clean up + delete[] ucell.atoms; +} + +// Test the restart_bfgs() function case 1 +TEST_F(IonsMoveBFGSTest, RestartBfgsCase1) +{ + // Initilize data + bfgs.get_init_done() = false; + criteria.test_relax_method = 1; + double lat0 = 1.0; + bfgs.allocate(); + bfgs.get_save_flag() = true; + for (int i = 0; i < Ions_Move_Basic::dim; ++i) + { + bfgs.get_move_p()[i] = 1.0; + bfgs.get_pos()[i] = 1.0; + bfgs.get_pos_p()[i] = i; + } + + // Call the function being tested + std::ofstream ofs("test_restart_bfgs_case1.log"); + bfgs.restart_bfgs_for_testing(lat0, update_iter, ofs, criteria.test_relax_method); + ofs.close(); + + // Check the results + std::ifstream ifs("test_restart_bfgs_case1.log"); + std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + ifs.close(); + std::remove("test_restart_bfgs_case1.log"); + + std::string expected_output = " trust_radius_old (bohr) = 2.44949\n"; + + EXPECT_THAT(output, testing::HasSubstr(expected_output)); + EXPECT_NEAR(Ions_Move_Basic::trust_radius_old, 2.4494897427831779, 1e-12); + EXPECT_DOUBLE_EQ(bfgs.get_move_p()[0], 0.0); + EXPECT_DOUBLE_EQ(bfgs.get_move_p()[1], 0.0); + EXPECT_DOUBLE_EQ(bfgs.get_move_p()[2], 0.0); + EXPECT_NEAR(bfgs.get_move_p()[3], -0.40824829046386307, 1e-12); + EXPECT_NEAR(bfgs.get_move_p()[4], -0.81649658092772615, 1e-12); + EXPECT_NEAR(bfgs.get_move_p()[5], -1.2247448713915892, 1e-12); +} + +// Test the restart_bfgs() function case 2 +TEST_F(IonsMoveBFGSTest, RestartBfgsCase2) +{ + // Initilize data + bfgs.get_init_done() = false; + bfgs.allocate(); + criteria.test_relax_method = 1; + double lat0 = 1.0; + for (int i = 0; i < Ions_Move_Basic::dim; ++i) + { + bfgs.get_move_p()[i] = 1.0; + bfgs.get_pos()[i] = i; + bfgs.get_pos_p()[i] = i; + } + + // Call the function being tested + std::ofstream ofs("test_restart_bfgs_case2.log"); + bfgs.restart_bfgs_for_testing(lat0, update_iter, ofs, criteria.test_relax_method); + ofs.close(); + std::remove("test_restart_bfgs_case2.log"); + + // Check the results + EXPECT_DOUBLE_EQ(update_iter, 0.0); + EXPECT_DOUBLE_EQ(bfgs.get_tr_min_hit(), false); + for (int i = 0; i < Ions_Move_Basic::dim; ++i) + { + EXPECT_DOUBLE_EQ(bfgs.get_pos_p()[i], 0.0); + EXPECT_DOUBLE_EQ(bfgs.get_grad_p()[i], 0.0); + EXPECT_DOUBLE_EQ(bfgs.get_move_p()[i], 0.0); + for (int j = 0; j < Ions_Move_Basic::dim; ++j) + { + if (i == j) + { + EXPECT_DOUBLE_EQ(bfgs.get_inv_hess()(i, j), 1.0); + } + else + { + EXPECT_DOUBLE_EQ(bfgs.get_inv_hess()(i, j), 0.0); + } + } + } +} + +// Test the bfgs_routine() function case 1 +TEST_F(IonsMoveBFGSTest, BfgsRoutineCase1) +{ + // Initilize data + bfgs.get_init_done() = false; + bfgs.allocate(); + bfgs.get_tr_min_hit() = false; + criteria.test_relax_method = 1; + criteria.out_level = "ie"; + double lat0 = 1.0; + const int istep = 1; + std::vector etot_info = {1.0, 0.9, 0.1}; + Ions_Move_Basic::relax_bfgs_rmin = 1.0; + for (int i = 0; i < Ions_Move_Basic::dim; ++i) + { + bfgs.get_move_p()[i] = 0.0; + bfgs.get_grad_p()[i] = i; + bfgs.get_pos_p()[i] = i; + } + + // Call the function being tested + std::ofstream ofs("test_bfgs_routine_case1.log"); + testing::internal::CaptureStdout(); + bfgs.bfgs_routine_for_testing(lat0, istep, update_iter, ofs, etot_info, criteria.out_level, criteria.test_relax_method); + std::string std_outout = testing::internal::GetCapturedStdout(); + ofs.close(); + + // Check the results + std::ifstream ifs("test_bfgs_routine_case1.log"); + std::string ofs_output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + ifs.close(); + std::remove("test_bfgs_routine_case1.log"); + + std::string expected_ofs + = " dE0s = 0\n den = 0.1\n " + " interpolated trust radius = 0\ntrust_radius = 0\nrelax_bfgs_rmin = 1\nrelax_bfgs_rmax = -1\n " + "trust_radius < relax_bfgs_rmin, reset bfgs history.\n"; + std::string expected_std = " BFGS TRUST (Bohr) : 1\n"; + + EXPECT_THAT(ofs_output, ::testing::HasSubstr(expected_ofs)); + EXPECT_EQ(expected_std, std_outout); + + EXPECT_DOUBLE_EQ(Ions_Move_Basic::trust_radius, 1.0); + EXPECT_DOUBLE_EQ(etot_info[0], 0.9); + EXPECT_DOUBLE_EQ(bfgs.get_tr_min_hit(), true); + EXPECT_NEAR(bfgs.get_move()[0], 0.0, 1e-12); + EXPECT_NEAR(bfgs.get_move()[1], -0.13483997249264842, 1e-12); + EXPECT_NEAR(bfgs.get_move()[2], -0.26967994498529685, 1e-12); + EXPECT_NEAR(bfgs.get_move()[3], -0.40451991747794525, 1e-12); + EXPECT_NEAR(bfgs.get_move()[4], -0.5393598899705937, 1e-12); + EXPECT_NEAR(bfgs.get_move()[5], -0.67419986246324215, 1e-12); + for (int i = 0; i < Ions_Move_Basic::dim; ++i) + { + EXPECT_DOUBLE_EQ(bfgs.get_pos()[i], i); + EXPECT_DOUBLE_EQ(bfgs.get_grad()[i], i); + for (int j = 0; j < Ions_Move_Basic::dim; ++j) + { + if (i == j) + EXPECT_DOUBLE_EQ(bfgs.get_inv_hess()(i, j), 1.0); + else + EXPECT_DOUBLE_EQ(bfgs.get_inv_hess()(i, j), 0.0); + } + } +} + +// Test the bfgs_routine() function case 2 +TEST_F(IonsMoveBFGSTest, BfgsRoutineCase2) +{ + // Initilize data + bfgs.get_init_done() = false; + bfgs.allocate(); + bfgs.get_tr_min_hit() = false; + criteria.test_relax_method = 0; + criteria.out_level = "none"; + double lat0 = 1.0; + const int istep = 1; + std::vector etot_info = {1.0, 0.9, 0.1}; + Ions_Move_Basic::relax_bfgs_rmin = -1.0; + for (int i = 0; i < Ions_Move_Basic::dim; ++i) + { + bfgs.get_move_p()[i] = i; + bfgs.get_grad_p()[i] = i; + bfgs.get_pos_p()[i] = i; + } + + // Call the function being tested + std::ofstream ofs("test_bfgs_routine_case2.log"); + testing::internal::CaptureStdout(); + bfgs.bfgs_routine_for_testing(lat0, istep, update_iter, ofs, etot_info, criteria.out_level, criteria.test_relax_method); + std::string std_outout = testing::internal::GetCapturedStdout(); + ofs.close(); + + // Check the results + std::ifstream ifs("test_bfgs_routine_case2.log"); + std::string ofs_output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + ifs.close(); + std::remove("test_bfgs_routine_case2.log"); + + std::string expected_ofs = " quadratic interpolation is impossible.\n"; + std::string expected_std = ""; + + EXPECT_THAT(ofs_output, ::testing::HasSubstr(expected_ofs)); + EXPECT_EQ(expected_std, std_outout); + + EXPECT_DOUBLE_EQ(Ions_Move_Basic::trust_radius, -0.5); + EXPECT_DOUBLE_EQ(etot_info[0], 0.9); + EXPECT_DOUBLE_EQ(bfgs.get_tr_min_hit(), false); + EXPECT_NEAR(bfgs.get_move()[0], 0.0, 1e-12); + EXPECT_NEAR(bfgs.get_move()[1], 0.067419986246324212, 1e-12); + EXPECT_NEAR(bfgs.get_move()[2], 0.13483997249264842, 1e-12); + EXPECT_NEAR(bfgs.get_move()[3], 0.20225995873897262, 1e-12); + EXPECT_NEAR(bfgs.get_move()[4], 0.26967994498529685, 1e-12); + EXPECT_NEAR(bfgs.get_move()[5], 0.33709993123162107, 1e-12); + for (int i = 0; i < Ions_Move_Basic::dim; ++i) + { + EXPECT_DOUBLE_EQ(bfgs.get_pos()[i], i); + EXPECT_DOUBLE_EQ(bfgs.get_grad()[i], i); + for (int j = 0; j < Ions_Move_Basic::dim; ++j) + { + EXPECT_DOUBLE_EQ(bfgs.get_inv_hess()(i, j), 0.0); + } + } +} + +// Test the bfgs_routine() function case 3 +TEST_F(IonsMoveBFGSTest, BfgsRoutineCase3) +{ + // Initilize data + double lat0 = 1.0; + const int istep = 1; + std::vector etot_info = {0.9, 1.0, -0.1}; + update_iter = 0; + Ions_Move_Basic::largest_grad = 0.0; + Ions_Move_Basic::relax_bfgs_init = 0.3; + Ions_Move_Basic::best_xxx = -0.4; + bfgs.get_init_done() = false; + bfgs.allocate(); + bfgs.get_bfgs_ndim() = 1; + bfgs.get_grad()[0] = 1.0; + bfgs.get_grad()[1] = 2.0; + bfgs.get_inv_hess()(0, 0) = -3.0; + bfgs.get_inv_hess()(0, 1) = -4.0; + bfgs.get_inv_hess()(1, 0) = -5.0; + bfgs.get_inv_hess()(1, 1) = -6.0; + + // Call the function being tested + std::ofstream ofs("test_bfgs_routine_case3.log"); + bfgs.bfgs_routine_for_testing(lat0, istep, update_iter, ofs, etot_info, criteria.out_level, criteria.test_relax_method); + ofs.close(); + + // Check the results + std::ifstream ifs("test_bfgs_routine_case3.log"); + std::string ofs_output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + ifs.close(); + std::remove("test_bfgs_routine_case3.log"); + + std::string expected_ofs = " check the norm of new move 410 (Bohr)\n Uphill move : resetting bfgs history\n"; + + EXPECT_THAT(ofs_output, ::testing::HasSubstr(expected_ofs)); + EXPECT_DOUBLE_EQ(Ions_Move_Basic::trust_radius, 0.2); + EXPECT_DOUBLE_EQ(etot_info[0], 0.9); + EXPECT_DOUBLE_EQ(bfgs.get_tr_min_hit(), false); + EXPECT_NEAR(bfgs.get_move()[0], -0.089442719099991588, 1e-12); + EXPECT_NEAR(bfgs.get_move()[1], -0.17888543819998318, 1e-12); + EXPECT_DOUBLE_EQ(bfgs.get_move()[2], 0.0); + EXPECT_DOUBLE_EQ(bfgs.get_move()[3], 0.0); + EXPECT_DOUBLE_EQ(bfgs.get_move()[4], 0.0); + EXPECT_DOUBLE_EQ(bfgs.get_move()[5], 0.0); + EXPECT_DOUBLE_EQ(bfgs.get_pos()[0], 0.0); + EXPECT_DOUBLE_EQ(bfgs.get_pos()[1], 0.0); + EXPECT_DOUBLE_EQ(bfgs.get_pos()[2], 0.0); + EXPECT_DOUBLE_EQ(bfgs.get_pos()[3], 0.0); + EXPECT_DOUBLE_EQ(bfgs.get_pos()[4], 0.0); + EXPECT_DOUBLE_EQ(bfgs.get_pos()[5], 0.0); + EXPECT_DOUBLE_EQ(bfgs.get_grad()[0], 1.0); + EXPECT_DOUBLE_EQ(bfgs.get_grad()[1], 2.0); + EXPECT_DOUBLE_EQ(bfgs.get_grad()[2], 0.0); + EXPECT_DOUBLE_EQ(bfgs.get_grad()[3], 0.0); + EXPECT_DOUBLE_EQ(bfgs.get_grad()[4], 0.0); + EXPECT_DOUBLE_EQ(bfgs.get_grad()[5], 0.0); +} + +// Test the bfgs_routine() function warning quit 1 +TEST_F(IonsMoveBFGSTest, BfgsRoutineWarningQuit1) +{ + // Initilize data + bfgs.get_init_done() = false; + bfgs.allocate(); + bfgs.get_tr_min_hit() = true; + criteria.test_relax_method = 1; + criteria.out_level = "ie"; + double lat0 = 1.0; + const int istep = 1; + std::vector etot_info = {1.0, 0.9, 0.1}; + Ions_Move_Basic::relax_bfgs_rmin = 1.0; + for (int i = 0; i < Ions_Move_Basic::dim; ++i) + { + bfgs.get_move_p()[i] = 0.0; + bfgs.get_grad_p()[i] = i; + bfgs.get_pos_p()[i] = i; + } + + // Check the results + std::ofstream ofs("test_bfgs_routine_warning_quit1.log"); + testing::internal::CaptureStdout(); + EXPECT_EXIT(bfgs.bfgs_routine_for_testing(lat0, istep, update_iter, ofs, etot_info, criteria.out_level, criteria.test_relax_method), ::testing::ExitedWithCode(1), ""); + std::string output = testing::internal::GetCapturedStdout(); + ofs.close(); + std::remove("test_bfgs_routine_warning_quit1.log"); + EXPECT_THAT(output, testing::HasSubstr("trust radius is too small! Break down.")); +} + +// Test the bfgs_routine() function warning quit 2 +TEST_F(IonsMoveBFGSTest, BfgsRoutineWarningQuit2) +{ + // Initilize data + bfgs.get_init_done() = false; + bfgs.allocate(); + bfgs.get_tr_min_hit() = false; + criteria.test_relax_method = 1; + criteria.out_level = "ie"; + double lat0 = 1.0; + const int istep = 1; + std::vector etot_info = {1.0, 0.9, 0.1}; + Ions_Move_Basic::relax_bfgs_rmin = 1.0; + + // Check the results + std::ofstream ofs("test_bfgs_routine_warning_quit2.log"); + testing::internal::CaptureStdout(); + EXPECT_EXIT(bfgs.bfgs_routine_for_testing(lat0, istep, update_iter, ofs, etot_info, criteria.out_level, criteria.test_relax_method), ::testing::ExitedWithCode(1), ""); + std::string output = testing::internal::GetCapturedStdout(); + ofs.close(); + std::remove("test_bfgs_routine_warning_quit2.log"); + EXPECT_THAT(output, testing::HasSubstr("BFGS: move-length unreasonably short")); +} diff --git a/source/source_relax/unittests/test_ions_move_bfgs2.cpp b/source/source_relax/unittests/test_ions_move_bfgs2.cpp new file mode 100644 index 00000000000..76dad4a9f8b --- /dev/null +++ b/source/source_relax/unittests/test_ions_move_bfgs2.cpp @@ -0,0 +1,377 @@ +// source/source_relax/unittests/test_ions_move_bfgs2.cpp +#include "gtest/gtest.h" +#include "gmock/gmock.h" +#include "for_test.h" + +#include "source_relax/ions_move_bfgs2.h" + +#include "source_io/module_parameter/parameter.h" + +#include "source_relax/ions_move_basic.h" // for Ions_Move_Basic static members +#include "source_relax/relax_data.h" +#include + +/************************************************ + * unit tests for Ions_Move_BFGS2 (no MockUnitCell) + ***********************************************/ + +class BFGSTest : public ::testing::Test { +protected: + Ions_Move_BFGS2 bfgs; + void SetUp() override + { + // Initialize variables before each test + } + + void TearDown() override + { + // nothing global to clean here + } + + // Friendship is not inherited: a TEST_F body lives in a class derived from + // this fixture, so the internals of Ions_Move_BFGS2 (which friends the + // fixture itself) are reached through these forwarders. They all carry a + // get_ / call_ prefix because several tests declare locals named pos, + // force, size, steplength, dpos and maxstep. + const std::vector>& get_H() const + { + return bfgs.H; + } + const std::vector>& get_pos() const + { + return bfgs.pos; + } + const std::vector& get_pos0() const + { + return bfgs.pos0; + } + const std::vector>& get_pos_taud() const + { + return bfgs.pos_taud; + } + const std::vector& get_pos_taud0() const + { + return bfgs.pos_taud0; + } + const std::vector>& get_force() const + { + return bfgs.force; + } + const std::vector& get_force0() const + { + return bfgs.force0; + } + const std::vector& get_steplength() const + { + return bfgs.steplength; + } + const std::vector>& get_dpos() const + { + return bfgs.dpos; + } + int get_size() const + { + return bfgs.size; + } + double get_alpha() const + { + return bfgs.alpha; + } + double get_maxstep() const + { + return bfgs.maxstep; + } + bool get_sign() const + { + return bfgs.sign; + } + bool get_is_initialized() const + { + return bfgs.is_initialized; + } + + void set_sign(const bool value) + { + bfgs.sign = value; + } + void set_is_initialized(const bool value) + { + bfgs.is_initialized = value; + } + void set_force0(const std::vector& value) + { + bfgs.force0 = value; + } + void set_pos_taud_x(const int i, const double value) + { + bfgs.pos_taud[i].x = value; + } + + void call_DetermineStep(std::vector& steplength, + std::vector>& dpos, + double& maxstep) + { + bfgs.DetermineStep(steplength, dpos, maxstep); + } + void call_Update(std::vector& pos, std::vector& force, UnitCell& ucell) + { + bfgs.Update(pos, force, bfgs.H, ucell); + } + void call_GetPos(UnitCell& ucell) + { + bfgs.GetPos(ucell, bfgs.pos); + } + void call_GetPostaud(UnitCell& ucell) + { + bfgs.GetPostaud(ucell, bfgs.pos_taud); + } + void call_CalculateLargestGrad(const ModuleBase::matrix& force, UnitCell& ucell) + { + bfgs.CalculateLargestGrad(force, ucell); + } +}; + +// Test whether the allocate() function can correctly allocate memory space +TEST_F(BFGSTest, TestAllocate) +{ + int size = 2; + bfgs.allocate(size); + + // Check if allocated arrays are not empty + EXPECT_FALSE(get_H().empty()); + EXPECT_FALSE(get_pos().empty()); + EXPECT_FALSE(get_pos0().empty()); + EXPECT_FALSE(get_pos_taud().empty()); + EXPECT_FALSE(get_pos_taud0().empty()); + EXPECT_FALSE(get_force().empty()); + EXPECT_FALSE(get_force0().empty()); + EXPECT_FALSE(get_steplength().empty()); + EXPECT_FALSE(get_dpos().empty()); + EXPECT_EQ(get_size(), size); + EXPECT_EQ(get_alpha(), 70); + EXPECT_EQ(get_maxstep(), PARAM.inp.relax_bfgs_rmax); + EXPECT_TRUE(get_sign()); + EXPECT_EQ(Ions_Move_Basic::largest_grad, 0.0); +} + +// Test that relax_step will auto-initialize if not already initialized +TEST_F(BFGSTest, RelaxStepAutoInitialize) +{ + set_is_initialized(false); + + UnitCell ucell; + ucell.nat = 2; + ucell.ntype = 1; + ucell.atoms = new Atom[ucell.ntype]; + ucell.atoms[0].na = 2; + ucell.atoms[0].tau = std::vector>(2); + ucell.atoms[0].taud = std::vector>(2); + ucell.atoms[0].mbl = std::vector>(2, {1, 1, 1}); + ucell.atoms[0].tau[0].x = 0.0; ucell.atoms[0].tau[0].y = 0.0; ucell.atoms[0].tau[0].z = 0.0; + ucell.atoms[0].tau[1].x = 1.0; ucell.atoms[0].tau[1].y = 0.0; ucell.atoms[0].tau[1].z = 0.0; + ucell.lat0 = 1.0; + + ModuleBase::matrix force(2, 3); + force(0, 0) = 0.1; force(0, 1) = 0.0; force(0, 2) = 0.0; + force(1, 0) = -0.1; force(1, 1) = 0.0; force(1, 2) = 0.0; + + std::ofstream ofs_running; + + // Before relax_step, is_initialized should be false + EXPECT_FALSE(get_is_initialized()); + bfgs.relax_step(force, ucell, ofs_running); + // After relax_step, is_initialized should be true + EXPECT_TRUE(get_is_initialized()); +} + +// Test if a dimension less than or equal to 0 results in an assertion error +TEST_F(BFGSTest, TestAllocateWithZeroDimension) +{ + int size = 0; + ASSERT_DEATH(bfgs.allocate(size), ""); +} + +// Test DetermineStep scaling +TEST_F(BFGSTest, DetermineStepScaling) +{ + int size = 2; + bfgs.allocate(size); + + std::vector steplength = {1.0, 0.1}; + std::vector> dpos = { + ModuleBase::Vector3(1.0, 1.0, 1.0), + ModuleBase::Vector3(0.1, 0.1, 0.1) + }; + double maxstep = 0.5; + call_DetermineStep(steplength, dpos, maxstep); + + // first atom scaled down to maxstep + EXPECT_NEAR(dpos[0][0], 0.5, 1e-12); + EXPECT_NEAR(dpos[0][1], 0.5, 1e-12); + EXPECT_NEAR(dpos[0][2], 0.5, 1e-12); + + // second atom unchanged (small) + EXPECT_NEAR(dpos[1][0], 0.05, 1e-12); + EXPECT_NEAR(dpos[1][1], 0.05, 1e-12); + EXPECT_NEAR(dpos[1][2], 0.05, 1e-12); +} + +TEST_F(BFGSTest, UpdateUsesAbsoluteDisplacementThreshold) +{ + UnitCell ucell; + ucell.ntype = 1; + ucell.nat = 1; + ucell.lat0 = 1.0; + ucell.latvec.Identity(); + ucell.atoms = new Atom[ucell.ntype]; + ucell.atoms[0].na = 1; + ucell.atoms[0].mbl = std::vector>(1, {1, 1, 1}); + ucell.iat2it = new int[ucell.nat]; + ucell.iat2ia = new int[ucell.nat]; + ucell.iat2it[0] = 0; + ucell.iat2ia[0] = 0; + + bfgs.allocate(ucell.nat); + set_sign(false); + set_pos_taud_x(0, -1.0e-6); + set_force0({0.0, 0.0, 0.0}); + + std::vector pos = {0.0, 0.0, 0.0}; + std::vector force = {1.0, 0.0, 0.0}; + const double initial_h00 = get_H()[0][0]; + + call_Update(pos, force, ucell); + + EXPECT_NE(get_H()[0][0], initial_h00); +} + +// Test GetPos and GetPostaud without creating extra helper class +TEST_F(BFGSTest, GetPosAndPostaud) +{ + // prepare UnitCell with 1 type and 2 atoms + UnitCell ucell; + ucell.ntype = 1; + ucell.nat = 2; + ucell.lat0 = 2.0; + + // allocate atoms array + ucell.atoms = new Atom[ucell.ntype]; + ucell.atoms[0].na = 2; + ucell.atoms[0].tau = std::vector>(2); + ucell.atoms[0].taud = std::vector>(2); + ucell.atoms[0].mbl = std::vector>(2, {1, 1, 1}); + + // set coordinates + ucell.atoms[0].tau[0].x = 0.0; ucell.atoms[0].tau[0].y = 0.0; ucell.atoms[0].tau[0].z = 0.0; + ucell.atoms[0].tau[1].x = 1.0; ucell.atoms[0].tau[1].y = 0.0; ucell.atoms[0].tau[1].z = 0.0; + + // allocate mapping arrays + ucell.iat2it = new int[ucell.nat]; + ucell.iat2ia = new int[ucell.nat]; + int k = 0; + for (int it = 0; it < ucell.ntype; ++it) { + for (int ia = 0; ia < ucell.atoms[it].na; ++ia) { + ucell.iat2it[k] = it; + ucell.iat2ia[k] = ia; + ++k; + } + } + + // allocate bfgs arrays and call getters + bfgs.allocate(ucell.nat); + call_GetPos(ucell); + call_GetPostaud(ucell); + + // pos is tau * BOHR_TO_A * lat0 + EXPECT_DOUBLE_EQ(get_pos()[0][0], ucell.atoms[0].tau[0].x * ModuleBase::BOHR_TO_A * ucell.lat0); + EXPECT_DOUBLE_EQ(get_pos_taud()[1][2], ucell.atoms[0].taud[1].z); +} + +// Test CalculateLargestGrad (uses ModuleBase::matrix) +TEST_F(BFGSTest, CalculateLargestGrad) +{ + // UnitCell with 1 type and 2 atoms + UnitCell ucell; + ucell.ntype = 1; + ucell.nat = 2; + ucell.lat0 = 2.0; + + ucell.atoms = new Atom[ucell.ntype]; + ucell.atoms[0].na = 2; + ucell.atoms[0].mbl = std::vector>(2, {1, 1, 1}); + + // mapping arrays + ucell.iat2it = new int[ucell.nat]; + ucell.iat2ia = new int[ucell.nat]; + int k = 0; + for (int it = 0; it < ucell.ntype; ++it) { + for (int ia = 0; ia < ucell.atoms[it].na; ++ia) { + ucell.iat2it[k] = it; + ucell.iat2ia[k] = ia; + ++k; + } + } + + // build force matrix: 2 atoms x 3 components + ModuleBase::matrix force(2, 3); + force(0, 0) = -2.0; // this yields grad component = -(-2.0)*lat0 = 4.0 -> divided by lat0 => 2.0 + force(0, 1) = 0.0; + force(0, 2) = 1.0; + force(1, 0) = 3.0; // this yields abs = 6.0 -> divided by lat0 => 3.0 (this should be largest) + force(1, 1) = -1.0; + force(1, 2) = 0.0; + + bfgs.allocate(ucell.nat); + call_CalculateLargestGrad(force, ucell); + + // The computed value is stored in Ions_Move_Basic::largest_grad (Ry/Bohr after /lat0). + // grad component = -force * lat0; after dividing by lat0 this gives back force in Ry/Bohr. + // largest = max(|3.0|, |2.0|) = 3.0; lat0=2.0 => grad before /lat0 = 6.0, after /lat0 = 3.0 + EXPECT_NEAR(Ions_Move_Basic::largest_grad, 3.0, 1e-12); +} + +// Test relax_step basic functionality +TEST_F(BFGSTest, RelaxStepBasic) +{ + // Setup UnitCell with 1 type, 2 atoms + UnitCell ucell; + ucell.ntype = 1; + ucell.nat = 2; + ucell.lat0 = 1.0; + ucell.atoms = new Atom[ucell.ntype]; + ucell.atoms[0].na = 2; + ucell.atoms[0].tau = std::vector>(2); + ucell.atoms[0].taud = std::vector>(2); + ucell.atoms[0].mbl = std::vector>(2, {1, 1, 1}); + ucell.iat2it = new int[ucell.nat]; + ucell.iat2ia = new int[ucell.nat]; + int k = 0; + for (int it = 0; it < ucell.ntype; ++it) { + for (int ia = 0; ia < ucell.atoms[it].na; ++ia) { + ucell.iat2it[k] = it; + ucell.iat2ia[k] = ia; + ++k; + } + } + // Set initial positions + ucell.atoms[0].tau[0].x = 0.0; ucell.atoms[0].tau[0].y = 0.0; ucell.atoms[0].tau[0].z = 0.0; + ucell.atoms[0].tau[1].x = 1.0; ucell.atoms[0].tau[1].y = 0.0; ucell.atoms[0].tau[1].z = 0.0; + // Setup force matrix + ModuleBase::matrix force(2, 3); + force(0, 0) = 0.1; force(0, 1) = 0.0; force(0, 2) = 0.0; + force(1, 0) = -0.1; force(1, 1) = 0.0; force(1, 2) = 0.0; + // Allocate and call relax_step + bfgs.allocate(ucell.nat); + + std::ofstream ofs_running; + + bfgs.relax_step(force, ucell, ofs_running); + // Check that ionic_position_updated is true + EXPECT_TRUE(ucell.ionic_position_updated); + // Check that force values are set (converted units) + EXPECT_NEAR(get_force()[0][0], 0.1 * ModuleBase::Ry_to_eV / ModuleBase::BOHR_TO_A, 1e-12); + EXPECT_NEAR(get_force()[1][0], -0.1 * ModuleBase::Ry_to_eV / ModuleBase::BOHR_TO_A, 1e-12); + // Check that positions are updated (not equal to initial) + EXPECT_NEAR(get_pos()[0][0], 0.0, 1e-12); + EXPECT_NE(get_pos()[1][0], 1.0); +} diff --git a/source/source_relax/unittests/test_ions_move_cg.cpp b/source/source_relax/unittests/test_ions_move_cg.cpp new file mode 100644 index 00000000000..8603be86b55 --- /dev/null +++ b/source/source_relax/unittests/test_ions_move_cg.cpp @@ -0,0 +1,417 @@ +#include "source_relax/relax_criteria.h" +#include +#include "for_test.h" +#include "gtest/gtest.h" +#include "gmock/gmock.h" +#include "source_relax/ions_move_basic.h" +#include "source_relax/ions_move_cg.h" + +/************************************************ + * unit tests of class Ions_Move_CG + ***********************************************/ + +class IonsMoveCGTest : public ::testing::Test +{ + public: + Relax_Criteria criteria; + + protected: + void SetUp() override + { + // Initialize variables before each test + Ions_Move_Basic::dim = 6; + update_iter = 5; + im_cg.allocate(Ions_Move_Basic::dim); + criteria.force_thr = 0.001; + + // ban the 'cout' + // mohan add 2025-05-02 + std::cout.rdbuf(NULL); + } + + void TearDown() override + { + // Clean up after each test + } + void setupucell(UnitCell& ucell) + { + for (int it = 0; it < ucell.ntype; it++) + { + Atom* atom = &ucell.atoms[it]; + atom->label="test"; + for (int ia = 0; ia < atom->na; ia++) + { + atom->mag[ia]= 1; + for (int ik = 0; ik < 3; ++ik) + { + atom->tau[ia][ik] = 1; + atom->mbl[ia][ik] = 1; + atom->vel[ia][ik] = 1; + } + } + } + ucell.lat.GT.Zero(); + } + Ions_Move_CG im_cg; + int update_iter; + + // Friendship is not inherited: a TEST_F body lives in a class derived from + // this fixture, so the internals of Ions_Move_CG (which friends the fixture + // itself) are reached through these forwarders. + const std::vector& get_pos0() const + { + return im_cg.pos0; + } + const std::vector& get_grad0() const + { + return im_cg.grad0; + } + const std::vector& get_cg_grad0() const + { + return im_cg.cg_grad0; + } + const std::vector& get_move0() const + { + return im_cg.move0; + } + void set_move0(const int i, const double value) + { + im_cg.move0[i] = value; + } +}; + +// Test whether the allocate() function can correctly allocate memory space +TEST_F(IonsMoveCGTest, TestAllocate) +{ + const int dim = 4; + im_cg.allocate(dim); + + // Check if allocated vectors are not empty + EXPECT_EQ(get_pos0().size(), 4U); + EXPECT_EQ(get_grad0().size(), 4U); + EXPECT_EQ(get_cg_grad0().size(), 4U); + EXPECT_EQ(get_move0().size(), 4U); +} + +// Test if a dimension less than or equal to 0 results in an assertion error +TEST_F(IonsMoveCGTest, TestAllocateWithZeroDimension) +{ + const int dim = 0; + ASSERT_DEATH(im_cg.allocate(dim), ""); +} + +// Check that the arrays are correctly initialized to 0 +TEST_F(IonsMoveCGTest, TestAllocateAndInitialize) +{ + const int dim = 3; + im_cg.allocate(dim); + + // Check that the arrays are correctly initialized to 0 + EXPECT_DOUBLE_EQ(0.0, get_pos0()[0]); + EXPECT_DOUBLE_EQ(0.0, get_grad0()[1]); + EXPECT_DOUBLE_EQ(0.0, get_cg_grad0()[2]); + EXPECT_DOUBLE_EQ(0.0, get_move0()[0]); +} + +// Test function start() when converged +TEST_F(IonsMoveCGTest, TestStartConverged) +{ + // setup data + const int istep = 1; + UnitCell ucell; + setupucell(ucell); + ModuleBase::matrix force(2, 3); + double etot = 0.0; + std::vector etot_info(2, 0.0); + std::vector relax_method = {"cg", "1"}; + + // call function + std::ofstream ofs("TestStartConverged.log"); + im_cg.start(ucell, force, etot, istep, update_iter, ofs, etot_info, relax_method, criteria); + ofs.close(); + + // Check output + // Reporting moved to IonCellOptimizer::relax_step; check_converged no longer + // prints "Largest force is ...". The zero-force branch message and terminate + // output remain. + std::string expected_output = " largest force is 0, no movement is possible.\n it may converged, otherwise no " + "movement of atom is allowed.\n end of geometry optimization\n " + " istep = 1\n update iteration = 5\n"; + std::ifstream ifs("TestStartConverged.log"); + std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + ifs.close(); + std::remove("TestStartConverged.log"); + + + std::regex pattern(R"(==> .*::.*\t[\d\.]+ GB\t\d+ s\n )"); + output = std::regex_replace(output, pattern, ""); + EXPECT_THAT(output, testing::HasSubstr(expected_output)); + EXPECT_EQ(update_iter, 5); + EXPECT_DOUBLE_EQ(Ions_Move_Basic::largest_grad, 0.0); +} + +// Test function start() sd branch +TEST_F(IonsMoveCGTest, TestStartSd) +{ + // setup data + const int istep = 1; + std::vector relax_method = {"cg_bfgs", "1"}; + Ions_Move_CG::RELAX_CG_THR = 100.0; + UnitCell ucell; + setupucell(ucell); + ModuleBase::matrix force(2, 3); + force(0, 0) = 0.01; + double etot = 0.0; + std::vector etot_info(2, 0.0); + + // call function + std::ofstream ofs("TestStartSd.log"); + im_cg.start(ucell, force, etot, istep, update_iter, ofs, etot_info, relax_method, criteria); + ofs.close(); + + // Check output: reporting moved to IonCellOptimizer::relax_step; + // check_converged prints nothing here anymore. + std::string expected_output = ""; + std::ifstream ifs("TestStartSd.log"); + std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + ifs.close(); + std::remove("TestStartSd.log"); // mohan + + EXPECT_THAT(output, testing::HasSubstr(expected_output)); + EXPECT_EQ(update_iter, 5); + EXPECT_EQ(relax_method[0], "bfgs"); + EXPECT_DOUBLE_EQ(Ions_Move_Basic::largest_grad, 0.01); + EXPECT_DOUBLE_EQ(Ions_Move_Basic::best_xxx, -1.0); + EXPECT_DOUBLE_EQ(Ions_Move_Basic::relax_bfgs_init, 1.0); +} + +// Test function start() trial branch with goto +TEST_F(IonsMoveCGTest, TestStartTrialGoto) +{ + // setup data + const int istep = 1; + Ions_Move_CG::RELAX_CG_THR = 100.0; + UnitCell ucell; + setupucell(ucell); + ModuleBase::matrix force(2, 3); + force(0, 0) = 0.1; + double etot = 0.0; + std::vector etot_info(2, 0.0); + std::vector relax_method = {"cg_bfgs", "1"}; + + // call function + set_move0(0, 1.0); + std::ofstream ofs1("TestStartTrialGoto_temp1.log"); + im_cg.start(ucell, force, etot, istep, update_iter, ofs1, etot_info, relax_method, criteria); + ofs1.close(); + std::remove("TestStartTrialGoto_temp1.log"); + int istep_2 = 2; + set_move0(0, 10.0); + force(0, 0) = 0.001; + relax_method = {"cg_bfgs", "1"}; + std::ofstream ofs("TestStartTrialGoto.log"); + im_cg.start(ucell, force, etot, istep_2, update_iter, ofs, etot_info, relax_method, criteria); + ofs.close(); + + // Check output: reporting moved to IonCellOptimizer::relax_step; + // check_converged prints nothing here anymore. + std::string expected_output = ""; + std::ifstream ifs("TestStartTrialGoto.log"); + std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + ifs.close(); + std::remove("TestStartTrialGoto.log"); + + EXPECT_THAT(output, testing::HasSubstr(expected_output)); + EXPECT_EQ(update_iter, 5); + EXPECT_EQ(relax_method[0], "bfgs"); + EXPECT_DOUBLE_EQ(Ions_Move_Basic::largest_grad, 0.001); + EXPECT_DOUBLE_EQ(Ions_Move_Basic::best_xxx, 10.0); + EXPECT_DOUBLE_EQ(Ions_Move_Basic::relax_bfgs_init, 10.0); +} + +// Test function start() trial branch without goto +TEST_F(IonsMoveCGTest, TestStartTrial) +{ + // setup data + const int istep = 1; + UnitCell ucell; + setupucell(ucell); + ModuleBase::matrix force(2, 3); + force(0, 0) = 0.01; + double etot = 0.0; + std::vector etot_info(2, 0.0); + std::vector relax_method = {"cg_bfgs", "1"}; + + // call function + set_move0(0, 1.0); + std::ofstream ofs1("TestStartTrial_temp1.log"); + im_cg.start(ucell, force, etot, istep, update_iter, ofs1, etot_info, relax_method, criteria); + ofs1.close(); + std::remove("TestStartTrial_temp1.log"); + int istep_2 = 2; + set_move0(0, 10.0); + std::ofstream ofs("TestStartTrial.log"); + im_cg.start(ucell, force, etot, istep_2, update_iter, ofs, etot_info, relax_method, criteria); + ofs.close(); + + // Check output: reporting moved to IonCellOptimizer::relax_step; + // check_converged prints nothing here anymore. + std::string expected_output = ""; + std::ifstream ifs("TestStartTrial.log"); + std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + ifs.close(); + std::remove("TestStartTrial.log"); + + EXPECT_THAT(output, testing::HasSubstr(expected_output)); + EXPECT_EQ(update_iter, 5); + EXPECT_EQ(relax_method[0], "bfgs"); + EXPECT_DOUBLE_EQ(Ions_Move_Basic::largest_grad, 0.01); + EXPECT_DOUBLE_EQ(Ions_Move_Basic::best_xxx, 10.0); + EXPECT_DOUBLE_EQ(Ions_Move_Basic::relax_bfgs_init, 70.0); +} + +// Test function start() no trial branch with goto case 1 +TEST_F(IonsMoveCGTest, TestStartNoTrialGotoCase1) +{ + // setup data + const int istep = 1; + Ions_Move_CG::RELAX_CG_THR = 100.0; + UnitCell ucell; + setupucell(ucell); + ModuleBase::matrix force(2, 3); + force(0, 0) = 0.1; + double etot = 0.0; + std::vector etot_info(2, 0.0); + std::vector relax_method = {"cg_bfgs", "1"}; + + // call function + set_move0(0, 1.0); + std::ofstream ofs1("TestStartNoTrialGotoCase1_temp1.log"); + im_cg.start(ucell, force, etot, istep, update_iter, ofs1, etot_info, relax_method, criteria); + ofs1.close(); + std::remove("TestStartNoTrialGotoCase1_temp1.log"); + int istep_2 = 2; + std::ofstream ofs2("TestStartNoTrialGotoCase1_temp2.log"); + im_cg.start(ucell, force, etot, istep_2, update_iter, ofs2, etot_info, relax_method, criteria); + ofs2.close(); + std::remove("TestStartNoTrialGotoCase1_temp2.log"); + set_move0(0, 1.0); + force(0, 0) = 0.001; + relax_method = {"cg_bfgs", "1"}; + std::ofstream ofs("TestStartNoTrialGotoCase1.log"); + im_cg.start(ucell, force, etot, istep_2, update_iter, ofs, etot_info, relax_method, criteria); + ofs.close(); + + // Check output: reporting moved to IonCellOptimizer::relax_step; + // check_converged prints nothing here anymore. + std::string expected_output = ""; + std::ifstream ifs("TestStartNoTrialGotoCase1.log"); + std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + ifs.close(); + std::remove("TestStartNoTrialGotoCase1.log"); + + EXPECT_THAT(output, testing::HasSubstr(expected_output)); + EXPECT_EQ(update_iter, 5); + EXPECT_EQ(relax_method[0], "bfgs"); + EXPECT_DOUBLE_EQ(Ions_Move_Basic::largest_grad, 0.001); + EXPECT_DOUBLE_EQ(Ions_Move_Basic::best_xxx, 490.0); + EXPECT_DOUBLE_EQ(Ions_Move_Basic::relax_bfgs_init, 490.0); +} + +// Test function start() no trial branch with goto case 2 +TEST_F(IonsMoveCGTest, TestStartNoTrialGotoCase2) +{ + // setup data + const int istep = 1; + Ions_Move_CG::RELAX_CG_THR = 100.0; + UnitCell ucell; + setupucell(ucell); + ModuleBase::matrix force(2, 3); + force(0, 0) = 0.01; + double etot = 0.0; + std::vector etot_info(2, 0.0); + std::vector relax_method = {"cg_bfgs", "1"}; + Ions_Move_Basic::best_xxx = 1.0; + Ions_Move_Basic::relax_bfgs_init = 1.0; + + // call function + set_move0(0, 1.0); + std::ofstream ofs1("TestStartNoTrialGotoCase2_temp1.log"); + im_cg.start(ucell, force, etot, istep, update_iter, ofs1, etot_info, relax_method, criteria); + ofs1.close(); + std::remove("TestStartNoTrialGotoCase2_temp1.log"); + int istep_2 = 2; + set_move0(0, 10.0); + std::ofstream ofs2("TestStartNoTrialGotoCase2_temp2.log"); + im_cg.start(ucell, force, etot, istep_2, update_iter, ofs2, etot_info, relax_method, criteria); + ofs2.close(); + std::remove("TestStartNoTrialGotoCase2_temp2.log"); + relax_method = {"cg_bfgs", "1"}; + std::ofstream ofs("TestStartNoTrialGotoCase2.log"); + im_cg.start(ucell, force, etot, istep_2, update_iter, ofs, etot_info, relax_method, criteria); + ofs.close(); + + // Check output: reporting moved to IonCellOptimizer::relax_step; + // check_converged prints nothing here anymore. + std::string expected_output = ""; + std::ifstream ifs("TestStartNoTrialGotoCase2.log"); + std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + ifs.close(); + std::remove("TestStartNoTrialGotoCase2.log"); + + EXPECT_THAT(output, testing::HasSubstr(expected_output)); + EXPECT_EQ(update_iter, 5); + EXPECT_EQ(relax_method[0], "bfgs"); + EXPECT_DOUBLE_EQ(Ions_Move_Basic::largest_grad, 0.01); + EXPECT_DOUBLE_EQ(Ions_Move_Basic::best_xxx, 70.0); + EXPECT_DOUBLE_EQ(Ions_Move_Basic::relax_bfgs_init, 70.0); +} + +// Test function start() no trial branch without goto +TEST_F(IonsMoveCGTest, TestStartNoTrial) +{ + // setup data + const int istep = 1; + Ions_Move_CG::RELAX_CG_THR = 100.0; + UnitCell ucell; + setupucell(ucell); + ModuleBase::matrix force(2, 3); + force(0, 0) = 0.01; + double etot = 0.0; + std::vector etot_info(2, 0.0); + std::vector relax_method = {"cg_bfgs", "1"}; + Ions_Move_Basic::best_xxx = 1.0; + Ions_Move_Basic::relax_bfgs_init = 1.0; + + // call function + set_move0(0, 1.0); + std::ofstream ofs1("TestStartNoTrial_temp1.log"); + im_cg.start(ucell, force, etot, istep, update_iter, ofs1, etot_info, relax_method, criteria); + ofs1.close(); + std::remove("TestStartNoTrial_temp1.log"); + int istep_2 = 2; + set_move0(0, 1.0); + force(0, 0) = 0.001; + std::ofstream ofs2("TestStartNoTrial_temp2.log"); + im_cg.start(ucell, force, etot, istep_2, update_iter, ofs2, etot_info, relax_method, criteria); + ofs2.close(); + std::remove("TestStartNoTrial_temp2.log"); + std::ofstream ofs("TestStartNoTrial.log"); + im_cg.start(ucell, force, etot, istep_2, update_iter, ofs, etot_info, relax_method, criteria); + ofs.close(); + + // Check output: reporting moved to IonCellOptimizer::relax_step; + // check_converged prints nothing here anymore. + std::string expected_output = ""; + std::ifstream ifs("TestStartNoTrial.log"); + std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + ifs.close(); + std::remove("TestStartNoTrial.log"); + + EXPECT_THAT(output, testing::HasSubstr(expected_output)); + EXPECT_EQ(update_iter, 5); + EXPECT_EQ(relax_method[0], "bfgs"); + EXPECT_DOUBLE_EQ(Ions_Move_Basic::largest_grad, 0.001); + EXPECT_DOUBLE_EQ(Ions_Move_Basic::best_xxx, 1.0); + EXPECT_NEAR(Ions_Move_Basic::relax_bfgs_init, 1.2345679012345678, 1e-12); +} diff --git a/source/source_relax/unittests/test_ions_move_lbfgs.cpp b/source/source_relax/unittests/test_ions_move_lbfgs.cpp new file mode 100644 index 00000000000..e7f882719b7 --- /dev/null +++ b/source/source_relax/unittests/test_ions_move_lbfgs.cpp @@ -0,0 +1,167 @@ +#include +#include +#include + +#include "gmock/gmock.h" +#include "gtest/gtest.h" +#include "source_cell/unitcell.h" +#include "source_relax/ions_move_basic.h" +#include "source_relax/ions_move_lbfgs.h" +#include "source_relax/relax_criteria.h" + +/************************************************ + * unit tests of class Ions_Move_LBFGS + ***********************************************/ + +/** + * - Tested Functions: + * - Ions_Move_LBFGS::relax_step() converged / not-converged branches + * - Ions_Move_LBFGS::relax_step() fills Ions_Move_Basic::largest_grad + * (the value reported by IonCellOptimizer::relax_step in the running log) + */ + +namespace unitcell +{ +// Stub: the real update_pos_tau performs lattice-vector algebra that needs a +// fully built UnitCell; these tests only care about the convergence decision. +void update_pos_tau(const Lattice&, const double*, const int, const int, Atom*) +{ +} +} // namespace unitcell + +UnitCell::UnitCell() +{ + Coordinate = "Direct"; + latName = "none"; + lat0 = 1.0; + latvec.Identity(); + + ntype = 1; + nat = 1; + iat2it = nullptr; + iat2ia = nullptr; + iwt2iat = nullptr; + iwt2iw = nullptr; + itia2iat.create(1, 1); + + atoms = new Atom[ntype]; + set_atom_flag = true; + atoms[0].label = "Si"; +} +UnitCell::~UnitCell() +{ +} +Magnetism::Magnetism() +{ +} +Magnetism::~Magnetism() +{ +} +Atom::Atom() +{ + na = 1; + tau.resize(na); + dis.resize(na); + mag.resize(na); + mbl.resize(na); + vel.resize(na); + taud.resize(na); + mbl[0] = {1, 1, 1}; +} +Atom::~Atom() +{ +} +#ifdef __MPI +void Atom::bcast_atom() +{ +} +void Atom::bcast_atom2() +{ +} +#endif +Atom_pseudo::Atom_pseudo() +{ +} +Atom_pseudo::~Atom_pseudo() +{ +} +pseudo::pseudo() +{ +} +pseudo::~pseudo() +{ +} +SepPot::SepPot() +{ +} +SepPot::~SepPot() +{ +} +Sep_Cell::Sep_Cell() noexcept +{ +} +Sep_Cell::~Sep_Cell() noexcept +{ +} +int ModuleSymmetry::Symmetry::symm_flag = 0; +void ModuleSymmetry::Symmetry::symmetrize_mat3(ModuleBase::matrix& sigma, const Lattice& lat) const +{ +} +void ModuleSymmetry::Symmetry::symmetrize_vec3_nat(double* v) const +{ +} + +// Friend of Parameter; the only sanctioned way to write PARAM.input in tests. +// Not needed anymore: allocate() takes relax_bfgs_rmax and out_level explicitly. +class IonsMoveLBFGSTest : public ::testing::Test +{ + protected: + Ions_Move_LBFGS lbfgs; + const int natom = 1; + + void SetUp() override + { + lbfgs.allocate(natom, 0.2, "m"); // keep stdout quiet + } +}; + +// A force below force_thr_ev must be reported as converged, with +// Ions_Move_Basic::largest_grad filled for the running-log report. +TEST_F(IonsMoveLBFGSTest, RelaxStepConverged) +{ + UnitCell ucell; + ModuleBase::matrix force(natom, 3); + force(0, 0) = 1.0e-4; // Ry/Bohr, ~0.005 eV/Angstrom + const double etot = 0.0; + std::ofstream ofs("lbfgs_converged.log"); + Relax_Criteria criteria; + criteria.force_thr_ev = 1.0; // eV/Angstrom + + const bool converged = lbfgs.relax_step(force, ucell, etot, ofs, criteria); + ofs.close(); + std::remove("lbfgs_converged.log"); + + EXPECT_TRUE(converged); + EXPECT_NEAR(Ions_Move_Basic::largest_grad, 1.0e-4, 1e-12); // Ry/Bohr, lat0 = 1 + EXPECT_LT(Ions_Move_Basic::largest_grad * ModuleBase::Ry_to_eV / ModuleBase::BOHR_TO_A, criteria.force_thr_ev); +} + +// A force above force_thr_ev must be reported as not converged. +TEST_F(IonsMoveLBFGSTest, RelaxStepNotConverged) +{ + UnitCell ucell; + ModuleBase::matrix force(natom, 3); + force(0, 0) = 0.1; // Ry/Bohr, ~51.4 eV/Angstrom + const double etot = 0.0; + std::ofstream ofs("lbfgs_not_converged.log"); + Relax_Criteria criteria; + criteria.force_thr_ev = 1.0; // eV/Angstrom + + const bool converged = lbfgs.relax_step(force, ucell, etot, ofs, criteria); + ofs.close(); + std::remove("lbfgs_not_converged.log"); + + EXPECT_FALSE(converged); + EXPECT_NEAR(Ions_Move_Basic::largest_grad, 0.1, 1e-12); + EXPECT_GT(Ions_Move_Basic::largest_grad * ModuleBase::Ry_to_eV / ModuleBase::BOHR_TO_A, criteria.force_thr_ev); +} diff --git a/source/source_relax/unittests/test_ions_move_methods.cpp b/source/source_relax/unittests/test_ions_move_methods.cpp new file mode 100644 index 00000000000..2504e018706 --- /dev/null +++ b/source/source_relax/unittests/test_ions_move_methods.cpp @@ -0,0 +1,293 @@ +#include "source_relax/relax_criteria.h" +#include "for_test.h" +#include "gmock/gmock.h" +#include "gtest/gtest.h" +#include +#include +#include +#include +#include +#include +#include "source_relax/ions_move_methods.h" +/************************************************ + * unit tests of class Ions_Move_Methods + ***********************************************/ + +/** + * - Tested Functions: + * - Ions_Move_Methods::allocate() + * - Ions_Move_Methods::cal_movement() + * - Ions_Move_Methods::get_converged() + * - Ions_Move_Methods::get_ediff() + * - Ions_Move_Methods::get_largest_grad() + * - Ions_Move_Methods::get_trust_radius() + * - Ions_Move_Methods::get_update_iter() + */ + + // Mock the remake_cell function from update_cell.h +namespace unitcell +{ + // Track if remake_cell was called and with what lattice + static bool remake_cell_called = false; + static std::string remake_cell_latName; + static ModuleBase::Matrix3 remake_cell_latvec; + + void remake_cell(Lattice& lat) + { + remake_cell_called = true; + remake_cell_latName = lat.latName; + remake_cell_latvec = lat.latvec; + + // Mock implementation: enforce simple cubic structure for "sc" + if (lat.latName == "sc") + { + double celldm = std::sqrt(lat.latvec.e11 * lat.latvec.e11 + + lat.latvec.e12 * lat.latvec.e12 + + lat.latvec.e13 * lat.latvec.e13); + lat.latvec.Zero(); + lat.latvec.e11 = celldm; + lat.latvec.e22 = celldm; + lat.latvec.e33 = celldm; + } + // Mock implementation: enforce FCC structure for "fcc" + else if (lat.latName == "fcc") + { + double celldm = std::sqrt(lat.latvec.e11 * lat.latvec.e11 + + lat.latvec.e12 * lat.latvec.e12 + + lat.latvec.e13 * lat.latvec.e13) / std::sqrt(2.0); + lat.latvec.e11 = -celldm; + lat.latvec.e12 = 0.0; + lat.latvec.e13 = celldm; + lat.latvec.e21 = 0.0; + lat.latvec.e22 = celldm; + lat.latvec.e23 = celldm; + lat.latvec.e31 = -celldm; + lat.latvec.e32 = celldm; + lat.latvec.e33 = 0.0; + } + } + + void update_pos_tau(const Lattice&, const double*, const int, const int, Atom*) + { + } + + // Helper function to reset mock state + void reset_remake_cell_mock() + { + remake_cell_called = false; + remake_cell_latName = ""; + remake_cell_latvec.Zero(); + } + + // Helper function to check if remake_cell was called + bool was_remake_cell_called() + { + return remake_cell_called; + } +} + +// Define a fixture for the tests +class IonsMoveMethodsTest : public ::testing::Test +{ + public: + Relax_Criteria criteria; + + protected: + Ions_Move_Methods imm; + const int natom = 2; + + virtual void SetUp() + { + // Initialize variables before each test + } + + virtual void TearDown() + { + // Clean up after each test + } +}; + +// Test the allocate() function +TEST_F(IonsMoveMethodsTest, Allocate) +{ + imm.allocate(natom, "bfgs", "1"); + EXPECT_EQ(Ions_Move_Basic::dim, 6); + + imm.allocate(natom, "sd", "1"); + EXPECT_EQ(Ions_Move_Basic::dim, 6); + + imm.allocate(natom, "cg", "1"); + EXPECT_EQ(Ions_Move_Basic::dim, 6); + + imm.allocate(natom, "cg_bfgs", "1"); + EXPECT_EQ(Ions_Move_Basic::dim, 6); +} + +// Test the allocate() function warning quit +TEST_F(IonsMoveMethodsTest, AllocateWarningQuit) +{ + GlobalV::ofs_warning.open("log"); + imm.allocate(natom, "none", "1"); + GlobalV::ofs_warning.close(); + + std::ifstream ifs("log"); + std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + EXPECT_THAT(output, testing::HasSubstr("the parameter relax_method is not correct.")); + ifs.close(); + std::remove("log"); +} + +// Test the cal_movement() function +TEST_F(IonsMoveMethodsTest, CalMovement) +{ + const int istep = 0; + const int force_step = 1; + ModuleBase::matrix f(natom, 3); + f(0, 0) = 0.1; + f(1, 1) = -0.1; + const double etot = 0.0; + UnitCell ucell; + std::ofstream ofs; + + std::vector relax_method; + + relax_method = {"bfgs", "1"}; + imm.allocate(natom, "bfgs", "1"); + imm.cal_movement(istep, force_step, f, etot, ucell, ofs, relax_method, criteria); + + relax_method = {"sd", "1"}; + imm.allocate(natom, "sd", "1"); + imm.cal_movement(istep, force_step, f, etot, ucell, ofs, relax_method, criteria); + + relax_method = {"cg", "1"}; + imm.allocate(natom, "cg", "1"); + imm.cal_movement(istep, force_step, f, etot, ucell, ofs, relax_method, criteria); + + relax_method = {"cg_bfgs", "1"}; + imm.allocate(natom, "cg_bfgs", "1"); + imm.cal_movement(istep, force_step, f, etot, ucell, ofs, relax_method, criteria); +} + +// Test the cal_movement() function warning quit +TEST_F(IonsMoveMethodsTest, CalMovementWarningQuit) +{ + const int istep = 0; + const int force_step = 1; + const ModuleBase::matrix f(3, 3); + const double etot = 0.0; + UnitCell ucell; + std::ofstream ofs; + std::vector relax_method = {"none", "1"}; + imm.allocate(natom, "none", "1"); + + GlobalV::ofs_warning.open("log"); + imm.cal_movement(istep, force_step, f, etot, ucell, ofs, relax_method, criteria); + GlobalV::ofs_warning.close(); + + std::ifstream ifs("log"); + std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + EXPECT_THAT(output, testing::HasSubstr("the parameter relax_method is not correct.")); + ifs.close(); + std::remove("log"); +} + + + +// Test the get_ediff() function +TEST_F(IonsMoveMethodsTest, GetEdiff) +{ + // etot_info_ is initialized to {0.0, 0.0, 0.0} by default + EXPECT_DOUBLE_EQ(imm.get_ediff(), 0.0); +} + +// Test the get_largest_grad() function +TEST_F(IonsMoveMethodsTest, GetLargestGrad) +{ + Ions_Move_Basic::largest_grad = 2.0; + + EXPECT_DOUBLE_EQ(imm.get_largest_grad(), 2.0); +} + +// Test the get_trust_radius() function +TEST_F(IonsMoveMethodsTest, GetTrustRadius) +{ + Ions_Move_Basic::trust_radius = 3.0; + + EXPECT_DOUBLE_EQ(imm.get_trust_radius(), 3.0); +} + +// Test the get_update_iter() function +TEST_F(IonsMoveMethodsTest, GetUpdateIter) +{ + imm.set_update_iter(4); + + EXPECT_EQ(imm.get_update_iter(), 4); +} + +TEST_F(IonsMoveMethodsTest, ResetAfterCellChange) +{ + const std::string log_file = "reset_after_cell_change.log"; + std::ofstream ofs(log_file); + + imm.allocate(natom, "bfgs", "2"); + imm.set_converged(true); + imm.set_update_iter(4); + imm.get_etot_info() = {-1.0, -2.0}; + Ions_Move_BFGS& bfgs = imm.get_bfgs(); + bfgs.get_first_step() = false; + bfgs.get_save_flag() = true; + bfgs.get_tr_min_hit() = true; + std::fill(bfgs.get_pos().begin(), bfgs.get_pos().end(), 1.0); + std::fill(bfgs.get_pos_p().begin(), bfgs.get_pos_p().end(), 2.0); + std::fill(bfgs.get_grad().begin(), bfgs.get_grad().end(), 3.0); + std::fill(bfgs.get_grad_p().begin(), bfgs.get_grad_p().end(), 4.0); + std::fill(bfgs.get_move().begin(), bfgs.get_move().end(), 5.0); + std::fill(bfgs.get_move_p().begin(), bfgs.get_move_p().end(), 6.0); + Ions_Move_Basic::trust_radius = 0.3; + Ions_Move_Basic::trust_radius_old = 0.2; + + imm.reset_after_cell_change({"bfgs", "2"}, ofs); + + EXPECT_FALSE(imm.get_converged()); + EXPECT_EQ(imm.get_update_iter(), 0); + EXPECT_THAT(imm.get_etot_info(), testing::Each(0.0)); + EXPECT_TRUE(bfgs.get_first_step()); + EXPECT_FALSE(bfgs.get_save_flag()); + EXPECT_FALSE(bfgs.get_tr_min_hit()); + EXPECT_THAT(bfgs.get_pos(), testing::Each(0.0)); + EXPECT_THAT(bfgs.get_pos_p(), testing::Each(0.0)); + EXPECT_THAT(bfgs.get_grad(), testing::Each(0.0)); + EXPECT_THAT(bfgs.get_grad_p(), testing::Each(0.0)); + EXPECT_THAT(bfgs.get_move(), testing::Each(0.0)); + EXPECT_THAT(bfgs.get_move_p(), testing::Each(0.0)); + for (int i = 0; i < Ions_Move_Basic::dim; ++i) + { + for (int j = 0; j < Ions_Move_Basic::dim; ++j) + { + EXPECT_DOUBLE_EQ(bfgs.get_inv_hess()(i, j), i == j ? 1.0 : 0.0); + } + } + EXPECT_DOUBLE_EQ(Ions_Move_Basic::trust_radius, 0.0); + EXPECT_DOUBLE_EQ(Ions_Move_Basic::trust_radius_old, 0.0); + + imm.allocate(natom, "bfgs", "1"); + imm.set_converged(true); + imm.set_update_iter(3); + imm.get_etot_info() = {-3.0, -4.0}; + ASSERT_TRUE(imm.get_bfgs_trad().get_is_initialized()); + + imm.reset_after_cell_change({"bfgs", "1"}, ofs); + ofs.close(); + + EXPECT_FALSE(imm.get_converged()); + EXPECT_EQ(imm.get_update_iter(), 0); + EXPECT_THAT(imm.get_etot_info(), testing::Each(0.0)); + EXPECT_FALSE(imm.get_bfgs_trad().get_is_initialized()); + + std::ifstream ifs(log_file); + const std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + EXPECT_THAT(output, testing::HasSubstr("Reset ionic BFGS history after cell change.")); + EXPECT_THAT(output, testing::HasSubstr("Reset traditional ionic BFGS history after cell change.")); + ifs.close(); + std::remove(log_file.c_str()); +} diff --git a/source/source_relax/unittests/test_ions_move_sd.cpp b/source/source_relax/unittests/test_ions_move_sd.cpp new file mode 100644 index 00000000000..507b3b216a9 --- /dev/null +++ b/source/source_relax/unittests/test_ions_move_sd.cpp @@ -0,0 +1,222 @@ +#include "source_relax/relax_criteria.h" +#include +#include "for_test.h" +#include "gmock/gmock.h" +#include "gtest/gtest.h" +#include "source_relax/ions_move_basic.h" +#include "source_relax/ions_move_sd.h" + +/************************************************ + * unit tests of class Ions_Move_SD + ***********************************************/ + +class IonsMoveSDTest : public ::testing::Test +{ + public: + Relax_Criteria criteria; + + protected: + void SetUp() override + { + // Initialize variables before each test + Ions_Move_Basic::dim = 6; + update_iter = 5; + im_sd.allocate(); + criteria.force_thr = 0.001; + } + + void TearDown() override + { + // Clean up after each test + } + + Ions_Move_SD im_sd; + int update_iter; +}; + +// Test whether the allocate() function can correctly allocate memory space +TEST_F(IonsMoveSDTest, TestAllocate) +{ + Ions_Move_Basic::dim = 4; + im_sd.allocate(); + + // Check if allocated vectors are not empty + EXPECT_EQ(im_sd.get_grad_saved().size(), 4U); + EXPECT_EQ(im_sd.get_pos_saved().size(), 4U); +} + +// Test if a dimension less than or equal to 0 results in an assertion error +TEST_F(IonsMoveSDTest, TestAllocateWithZeroDimension) +{ + Ions_Move_Basic::dim = 0; + ASSERT_DEATH(im_sd.allocate(), ""); +} + +// Check that the arrays are correctly initialized to 0 +TEST_F(IonsMoveSDTest, TestAllocateAndInitialize) +{ + Ions_Move_Basic::dim = 4; + im_sd.allocate(); + + // Check that the arrays are correctly initialized to 0 + EXPECT_DOUBLE_EQ(im_sd.get_grad_saved()[0], 0.0); + EXPECT_DOUBLE_EQ(im_sd.get_pos_saved()[1], 0.0); + EXPECT_DOUBLE_EQ(im_sd.get_grad_saved()[2], 0.0); + EXPECT_DOUBLE_EQ(im_sd.get_pos_saved()[3], 0.0); +} + +// Test function start() when converged +TEST_F(IonsMoveSDTest, TestStartConverged) +{ + // setup data + const int istep = 1; + UnitCell ucell; + ModuleBase::matrix force(2, 3); + double etot = 0.0; + std::vector etot_info(2, 0.0); + + // call function + std::ofstream ofs("test_sd_start_converged.log"); + im_sd.start(ucell, force, etot, istep, update_iter, ofs, etot_info, criteria); + ofs.close(); + + // Check output: reporting moved to IonCellOptimizer::relax_step; + // check_converged no longer prints "Largest force is ...". + std::string expected_output = " largest force is 0, no movement is possible.\n it may converged, otherwise no " + "movement of atom is allowed.\n end of geometry optimization\n " + " istep = 1\n update iteration = 5\n"; + std::ifstream ifs("test_sd_start_converged.log"); + std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + ifs.close(); + std::remove("test_sd_start_converged.log"); + + std::regex pattern(R"(==> .*::.*\t[\d\.]+ GB\t\d+ s\n )"); + output = std::regex_replace(output, pattern, ""); + EXPECT_THAT(output, testing::HasSubstr(expected_output)); + EXPECT_EQ(update_iter, 5); + EXPECT_DOUBLE_EQ(Ions_Move_Basic::largest_grad, 0.0); + EXPECT_DOUBLE_EQ(im_sd.get_energy_saved(), 0.0); + EXPECT_DOUBLE_EQ(im_sd.get_pos_saved()[0], 0.0); + EXPECT_DOUBLE_EQ(im_sd.get_pos_saved()[1], 10.0); + EXPECT_DOUBLE_EQ(im_sd.get_pos_saved()[2], 20.0); + EXPECT_DOUBLE_EQ(im_sd.get_pos_saved()[3], 30.0); + EXPECT_DOUBLE_EQ(im_sd.get_pos_saved()[4], 40.0); + EXPECT_DOUBLE_EQ(im_sd.get_pos_saved()[5], 50.0); +} + +// Test function start() when nor converged +TEST_F(IonsMoveSDTest, TestStartNotConverged) +{ + // setup data + const int istep = 1; + UnitCell ucell; + ModuleBase::matrix force(2, 3); + force(0, 0) = 1.0; + double etot = 0.0; + std::vector etot_info(2, 0.0); + for (int it = 0; it < ucell.ntype; it++) + { + Atom* atom = &ucell.atoms[it]; + for (int ia = 0; ia < atom->na; ia++) + { + for (int ik = 0; ik < 3; ++ik) + { + atom->tau[ia][ik] = (ik + 1)/3; + atom->mbl[ia][ik] = 1; + } + } + } + + // call function + std::ofstream ofs("test_sd_start_not_converged.log"); + im_sd.start(ucell, force, etot, istep, update_iter, ofs, etot_info, criteria); + ofs.close(); + + // Check output: reporting moved to IonCellOptimizer::relax_step; + // check_converged prints nothing here anymore. + std::string expected_output = ""; + std::ifstream ifs("test_sd_start_not_converged.log"); + std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + ifs.close(); + std::remove("test_sd_start_not_converged.log"); + + EXPECT_THAT(output, testing::HasSubstr(expected_output)); + EXPECT_EQ(update_iter, 6); + EXPECT_DOUBLE_EQ(Ions_Move_Basic::largest_grad, 1.0); + EXPECT_DOUBLE_EQ(im_sd.get_energy_saved(), 0.0); + EXPECT_DOUBLE_EQ(im_sd.get_pos_saved()[0], -1.0); + EXPECT_DOUBLE_EQ(im_sd.get_pos_saved()[1], 0.0); + EXPECT_DOUBLE_EQ(im_sd.get_pos_saved()[2], 10.0); + EXPECT_DOUBLE_EQ(im_sd.get_pos_saved()[3], 0.0); + EXPECT_DOUBLE_EQ(im_sd.get_pos_saved()[4], 0.0); + EXPECT_DOUBLE_EQ(im_sd.get_pos_saved()[5], 10.0); + EXPECT_DOUBLE_EQ(im_sd.get_grad_saved()[0], -1.0); + EXPECT_DOUBLE_EQ(im_sd.get_grad_saved()[1], 0.0); + EXPECT_DOUBLE_EQ(im_sd.get_grad_saved()[2], 0.0); + EXPECT_DOUBLE_EQ(im_sd.get_grad_saved()[3], 0.0); + EXPECT_DOUBLE_EQ(im_sd.get_grad_saved()[4], 0.0); + EXPECT_DOUBLE_EQ(im_sd.get_grad_saved()[5], 0.0); +} + +// Test function cal_tradius_sd() case 1 +TEST_F(IonsMoveSDTest, CalTradiusSdCase1) +{ + // setup data + const int istep = 1; + criteria.out_level = "ie"; + std::vector etot_info(2, 0.0); + + // call function + testing::internal::CaptureStdout(); + ions_move_sd::cal_tradius_sd(istep, etot_info, criteria.out_level); + std::string std_outout = testing::internal::GetCapturedStdout(); + + // Check the results + std::string expected_std = " SD RADIUS (Bohr) : -1\n"; + EXPECT_EQ(expected_std, std_outout); + EXPECT_EQ(Ions_Move_Basic::trust_radius, -1.0); +} + +// Test function cal_tradius_sd() case 2 +TEST_F(IonsMoveSDTest, CalTradiusSdCase2) +{ + // setup data + const int istep = 2; + std::vector etot_info = {0.0, 1.0}; + criteria.out_level = "m"; + + // call function + ions_move_sd::cal_tradius_sd(istep, etot_info, criteria.out_level); + + // Check the results + EXPECT_EQ(Ions_Move_Basic::trust_radius, -1.0); +} + +// Test function cal_tradius_sd() case 3 +TEST_F(IonsMoveSDTest, CalTradiusSdCase3) +{ + // setup data + const int istep = 2; + std::vector etot_info = {1.0, 0.0}; + criteria.out_level = "m"; + + // call function + ions_move_sd::cal_tradius_sd(istep, etot_info, criteria.out_level); + + // Check the results + EXPECT_EQ(Ions_Move_Basic::trust_radius, -0.5); +} + +// Test function cal_tradius_sd() warning quit +TEST_F(IonsMoveSDTest, CalTradiusWraningQuit) +{ + // setup data + const int istep = 0; + std::vector etot_info(2, 0.0); + + // Check the results + testing::internal::CaptureStdout(); + EXPECT_EXIT(ions_move_sd::cal_tradius_sd(istep, etot_info, criteria.out_level), ::testing::ExitedWithCode(1), ""); + std::string output = testing::internal::GetCapturedStdout(); + EXPECT_THAT(output, testing::HasSubstr("istep < 1!")); +} diff --git a/source/source_relax/test/lat_change_method_test.cpp b/source/source_relax/unittests/test_lat_change_method.cpp similarity index 100% rename from source/source_relax/test/lat_change_method_test.cpp rename to source/source_relax/unittests/test_lat_change_method.cpp diff --git a/source/source_relax/unittests/test_lattice_change_basic.cpp b/source/source_relax/unittests/test_lattice_change_basic.cpp new file mode 100644 index 00000000000..a77254da56d --- /dev/null +++ b/source/source_relax/unittests/test_lattice_change_basic.cpp @@ -0,0 +1,1074 @@ +#include +#include "source_relax/lattice_change_basic.h" +#include "source_relax/relax_data.h" +#include "mock_remake_cell.h" + +#include "for_test.h" +#include "gtest/gtest.h" +#include "gmock/gmock.h" +#include "source_io/module_parameter/parameter.h" + +/************************************************ + * unit tests of namespace Lattice_Change_Basic + ***********************************************/ + +// Define a fixture for the tests +class LatticeChangeBasicTest : public ::testing::Test +{ + public: + bool fixed_ibrav = false; + double stress_thr = 10.0; + + protected: + ModuleBase::matrix stress; + UnitCell ucell; + double lat[9], grad[9], move[9]; + + virtual void SetUp() + { + // Initialize variables before each test + stress.create(3, 3); + // Reset mock state before each test + unitcell::reset_remake_cell_mock(); + // Reset fixed_ibrav to default + fixed_ibrav = false; + } + + virtual void TearDown() + { + // Clean up after each test + unitcell::reset_remake_cell_mock(); + fixed_ibrav = false; + } +}; + +// Test the setup_gradient function with fixed_axes is volume +TEST_F(LatticeChangeBasicTest, SetupGradientVolume) +{ + // Initialize variables + ucell.lat_axis_free[0] = 1; + ucell.lat_axis_free[1] = 1; + ucell.lat_axis_free[2] = 1; + stress(0, 0) = 1.0; + stress(0, 1) = 2.0; + stress(0, 2) = 3.0; + stress(1, 0) = 4.0; + stress(1, 1) = 5.0; + stress(1, 2) = 6.0; + stress(2, 0) = 7.0; + stress(2, 1) = 8.0; + stress(2, 2) = 9.0; + + Lattice_Change_Basic::fixed_axes = "volume"; + + // Call setup_gradient method + Lattice_Change_Basic::setup_gradient(ucell, lat, grad, stress); + + // Check expected values for stress + EXPECT_DOUBLE_EQ(stress(0, 0), -4.0); + EXPECT_DOUBLE_EQ(stress(1, 1), 0.0); + EXPECT_DOUBLE_EQ(stress(2, 2), 4.0); + + // Check expected values for lat + EXPECT_DOUBLE_EQ(lat[0], 10.0); + EXPECT_DOUBLE_EQ(lat[1], 0.0); + EXPECT_DOUBLE_EQ(lat[2], 0.0); + EXPECT_DOUBLE_EQ(lat[3], 0.0); + EXPECT_DOUBLE_EQ(lat[4], 10.0); + EXPECT_DOUBLE_EQ(lat[5], 0.0); + EXPECT_DOUBLE_EQ(lat[6], 0.0); + EXPECT_DOUBLE_EQ(lat[7], 0.0); + EXPECT_DOUBLE_EQ(lat[8], 10.0); + + // Check expected values for grad + EXPECT_DOUBLE_EQ(grad[0], 40.0); + EXPECT_DOUBLE_EQ(grad[1], -20.0); + EXPECT_DOUBLE_EQ(grad[2], -30.0); + EXPECT_DOUBLE_EQ(grad[3], -40.0); + EXPECT_DOUBLE_EQ(grad[4], 0.0); + EXPECT_DOUBLE_EQ(grad[5], -60.0); + EXPECT_DOUBLE_EQ(grad[6], -70.0); + EXPECT_DOUBLE_EQ(grad[7], -80.0); + EXPECT_DOUBLE_EQ(grad[8], -40.0); +} + +// Test the setup_gradient function with fixed_axes is not volume +TEST_F(LatticeChangeBasicTest, SetupGradientNone) +{ + // Initialize variables + ucell.lat_axis_free[0] = 1; + ucell.lat_axis_free[1] = 1; + ucell.lat_axis_free[2] = 1; + stress(0, 0) = 1.0; + stress(0, 1) = 2.0; + stress(0, 2) = 3.0; + stress(1, 0) = 4.0; + stress(1, 1) = 5.0; + stress(1, 2) = 6.0; + stress(2, 0) = 7.0; + stress(2, 1) = 8.0; + stress(2, 2) = 9.0; + + Lattice_Change_Basic::fixed_axes = "None"; + + // Call setup_gradient method + Lattice_Change_Basic::setup_gradient(ucell, lat, grad, stress); + + // Check expected values for grad + EXPECT_DOUBLE_EQ(grad[0], -10.0); + EXPECT_DOUBLE_EQ(grad[1], -20.0); + EXPECT_DOUBLE_EQ(grad[2], -30.0); + EXPECT_DOUBLE_EQ(grad[3], -40.0); + EXPECT_DOUBLE_EQ(grad[4], -50.0); + EXPECT_DOUBLE_EQ(grad[5], -60.0); + EXPECT_DOUBLE_EQ(grad[6], -70.0); + EXPECT_DOUBLE_EQ(grad[7], -80.0); + EXPECT_DOUBLE_EQ(grad[8], -90.0); +} + +TEST_F(LatticeChangeBasicTest, ChangeLattice) +{ + // Initialize variables + ucell.lat_axis_free[0] = 1; + ucell.lat_axis_free[1] = 1; + ucell.lat_axis_free[2] = 1; + lat[0] = 1.0; + lat[1] = 0.0; + lat[2] = 0.0; + lat[3] = 0.0; + lat[4] = 2.0; + lat[5] = 0.0; + lat[6] = 0.0; + lat[7] = 0.0; + lat[8] = 3.0; + + move[0] = 1.0; + move[1] = 0.0; + move[2] = 0.0; + move[3] = 0.0; + move[4] = 2.0; + move[5] = 0.0; + move[6] = 0.0; + move[7] = 0.0; + move[8] = 3.0; + + // Call change_lattice method + Lattice_Change_Basic::change_lattice(ucell, move, lat, fixed_ibrav); + + // Check expected values for ucell after lattice change + EXPECT_DOUBLE_EQ(ucell.latvec.e11, 0.2); + EXPECT_DOUBLE_EQ(ucell.latvec.e12, 0.0); + EXPECT_DOUBLE_EQ(ucell.latvec.e13, 0.0); + EXPECT_DOUBLE_EQ(ucell.latvec.e21, 0.0); + EXPECT_DOUBLE_EQ(ucell.latvec.e22, 0.4); + EXPECT_DOUBLE_EQ(ucell.latvec.e23, 0.0); + EXPECT_DOUBLE_EQ(ucell.latvec.e31, 0.0); + EXPECT_DOUBLE_EQ(ucell.latvec.e32, 0.0); + EXPECT_DOUBLE_EQ(ucell.latvec.e33, 0.6); + + EXPECT_DOUBLE_EQ(ucell.a1.x, 0.2); + EXPECT_DOUBLE_EQ(ucell.a1.y, 0.0); + EXPECT_DOUBLE_EQ(ucell.a1.z, 0.0); + EXPECT_DOUBLE_EQ(ucell.a2.x, 0.0); + EXPECT_DOUBLE_EQ(ucell.a2.y, 0.4); + EXPECT_DOUBLE_EQ(ucell.a2.z, 0.0); + EXPECT_DOUBLE_EQ(ucell.a3.x, 0.0); + EXPECT_DOUBLE_EQ(ucell.a3.y, 0.0); + EXPECT_DOUBLE_EQ(ucell.a3.z, 0.6); + + EXPECT_DOUBLE_EQ(ucell.omega, 48.0); + + EXPECT_DOUBLE_EQ(ucell.GT.e11, 5.0); + EXPECT_DOUBLE_EQ(ucell.GT.e12, 0.0); + EXPECT_DOUBLE_EQ(ucell.GT.e13, 0.0); + EXPECT_DOUBLE_EQ(ucell.GT.e21, 0.0); + EXPECT_DOUBLE_EQ(ucell.GT.e22, 2.5); + EXPECT_DOUBLE_EQ(ucell.GT.e23, 0.0); + EXPECT_DOUBLE_EQ(ucell.GT.e31, 0.0); + EXPECT_DOUBLE_EQ(ucell.GT.e32, 0.0); + EXPECT_NEAR(ucell.GT.e33, 1.666666666666667, 1e-12); + + EXPECT_DOUBLE_EQ(ucell.G.e11, 5.0); + EXPECT_DOUBLE_EQ(ucell.G.e12, 0.0); + EXPECT_DOUBLE_EQ(ucell.G.e13, 0.0); + EXPECT_DOUBLE_EQ(ucell.G.e21, 0.0); + EXPECT_DOUBLE_EQ(ucell.G.e22, 2.5); + EXPECT_DOUBLE_EQ(ucell.G.e23, 0.0); + EXPECT_DOUBLE_EQ(ucell.G.e31, 0.0); + EXPECT_DOUBLE_EQ(ucell.G.e32, 0.0); + EXPECT_NEAR(ucell.G.e33, 1.666666666666667, 1e-12); + + EXPECT_DOUBLE_EQ(ucell.GGT.e11, 25.0); + EXPECT_DOUBLE_EQ(ucell.GGT.e12, 0.0); + EXPECT_DOUBLE_EQ(ucell.GGT.e13, 0.0); + EXPECT_DOUBLE_EQ(ucell.GGT.e21, 0.0); + EXPECT_DOUBLE_EQ(ucell.GGT.e22, 6.25); + EXPECT_DOUBLE_EQ(ucell.GGT.e23, 0.0); + EXPECT_DOUBLE_EQ(ucell.GGT.e31, 0.0); + EXPECT_DOUBLE_EQ(ucell.GGT.e32, 0.0); + EXPECT_NEAR(ucell.GGT.e33, 2.7777777777777786, 1e-12); + + EXPECT_DOUBLE_EQ(ucell.invGGT.e11, 0.04); + EXPECT_DOUBLE_EQ(ucell.invGGT.e12, 0.0); + EXPECT_DOUBLE_EQ(ucell.invGGT.e13, 0.0); + EXPECT_DOUBLE_EQ(ucell.invGGT.e21, 0.0); + EXPECT_DOUBLE_EQ(ucell.invGGT.e22, 0.16); + EXPECT_DOUBLE_EQ(ucell.invGGT.e23, 0.0); + EXPECT_DOUBLE_EQ(ucell.invGGT.e31, 0.0); + EXPECT_DOUBLE_EQ(ucell.invGGT.e32, 0.0); + EXPECT_DOUBLE_EQ(ucell.invGGT.e33, 0.36); +} + +// Test for check_converged when ucell.lat_axis_free[0] == 1 && ucell.lat_axis_free[1] == 1 && ucell.lat_axis_free[2] == 1, but not converged +TEST_F(LatticeChangeBasicTest, CheckConvergedCase1) +{ + // Set up test data + Lattice_Change_Basic::update_iter = 0; + stress_thr = 10.0; + std::ofstream ofs("test_check_converged_case1.log"); + ucell.lat_axis_free[0] = 1; + ucell.lat_axis_free[1] = 1; + ucell.lat_axis_free[2] = 1; + stress(0, 0) = 1.0; + stress(0, 1) = 2.0; + stress(0, 2) = 3.0; + stress(1, 0) = 4.0; + stress(1, 1) = 5.0; + stress(1, 2) = 6.0; + stress(2, 0) = 7.0; + stress(2, 1) = 8.0; + stress(2, 2) = 9.0; + + // Call the function under test + bool converged = Lattice_Change_Basic::check_converged(ucell, stress, grad, ofs, stress_thr); + ofs.close(); + + // Check the results + std::ifstream ifs("test_check_converged_case1.log"); + std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + // Reporting moved to IonCellOptimizer::relax_step; nothing printed here now. + EXPECT_EQ(output, ""); + EXPECT_EQ(Lattice_Change_Basic::update_iter, 0); + EXPECT_NEAR(Lattice_Change_Basic::largest_grad, 1323947.0517790401, 1e-12); + EXPECT_FALSE(converged); + + ifs.close(); + std::remove("test_check_converged_case1.log"); +} + +// Test for check_converged when ucell.lat_axis_free[0] == 1 && ucell.lat_axis_free[1] == 1 && ucell.lat_axis_free[2] == 1 && largest_grad == 0 +TEST_F(LatticeChangeBasicTest, CheckConvergedCase2) +{ + // Set up test data + Lattice_Change_Basic::update_iter = 0; + stress_thr = 10.0; + std::ofstream ofs("test_check_converged_case2.log"); + ucell.lat_axis_free[0] = 1; + ucell.lat_axis_free[1] = 1; + ucell.lat_axis_free[2] = 1; + stress(0, 0) = 0.0; + stress(0, 1) = 0.0; + stress(0, 2) = 0.0; + stress(1, 0) = 0.0; + stress(1, 1) = 0.0; + stress(1, 2) = 0.0; + stress(2, 0) = 0.0; + stress(2, 1) = 0.0; + stress(2, 2) = 0.0; + + // Call the function under test + bool converged = Lattice_Change_Basic::check_converged(ucell, stress, grad, ofs, stress_thr); + ofs.close(); + + // Check the results + std::ifstream ifs("test_check_converged_case2.log"); + std::string expected_output = " Largest stress is 0, movement is impossible.\n"; + std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + EXPECT_EQ(output, expected_output); + EXPECT_EQ(Lattice_Change_Basic::update_iter, 0); + EXPECT_DOUBLE_EQ(Lattice_Change_Basic::largest_grad, 0.0); + EXPECT_TRUE(converged); + + ifs.close(); + std::remove("test_check_converged_case2.log"); +} + +// Test for check_converged when ucell.lat_axis_free[0] == 1 && ucell.lat_axis_free[1] == 1 && ucell.lat_axis_free[2] == 1, and converged +TEST_F(LatticeChangeBasicTest, CheckConvergedCase3) +{ + // Set up test data + Lattice_Change_Basic::update_iter = 0; + stress_thr = 10.0; + std::ofstream ofs("test_check_converged_case3.log"); + ucell.lat_axis_free[0] = 1; + ucell.lat_axis_free[1] = 1; + ucell.lat_axis_free[2] = 1; + stress(0, 0) = 0.000001; + stress(0, 1) = 0.0; + stress(0, 2) = 0.0; + stress(1, 0) = 0.0; + stress(1, 1) = 0.0; + stress(1, 2) = 0.0; + stress(2, 0) = 0.0; + stress(2, 1) = 0.0; + stress(2, 2) = 0.0; + + // Call the function under test + bool converged = Lattice_Change_Basic::check_converged(ucell, stress, grad, ofs, stress_thr); + ofs.close(); + + // Check the results + std::ifstream ifs("test_check_converged_case3.log"); + std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + // Convergence/stress reporting moved to IonCellOptimizer::relax_step; + // check_converged itself now emits nothing on this branch. + EXPECT_EQ(output, ""); + EXPECT_EQ(Lattice_Change_Basic::update_iter, 1); + EXPECT_NEAR(Lattice_Change_Basic::largest_grad, 0.14710522797544887, 1e-12); + EXPECT_TRUE(converged); + + ifs.close(); + std::remove("test_check_converged_case3.log"); +} + +// Test for check_converged when ucell.lat_axis_free != 1, but not converged +TEST_F(LatticeChangeBasicTest, CheckConvergedCase4) +{ + // Set up test data + Lattice_Change_Basic::update_iter = 0; + stress_thr = 10.0; + std::ofstream ofs("test_check_converged_case4.log"); + ucell.lat_axis_free[0] = 0; + ucell.lat_axis_free[1] = 0; + ucell.lat_axis_free[2] = 0; + grad[0] = 1.0; + grad[1] = 1.0; + grad[2] = 1.0; + grad[3] = 1.0; + grad[4] = 1.0; + grad[5] = 1.0; + grad[6] = 1.0; + grad[7] = 1.0; + grad[8] = 1.0; + + // Call the function under test + bool converged = Lattice_Change_Basic::check_converged(ucell, stress, grad, ofs, stress_thr); + ofs.close(); + + // Check the results + std::ifstream ifs("test_check_converged_case4.log"); + std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + // Reporting moved to IonCellOptimizer::relax_step; nothing printed here now. + EXPECT_EQ(output, ""); + EXPECT_EQ(Lattice_Change_Basic::update_iter, 0); + EXPECT_NEAR(Lattice_Change_Basic::largest_grad, 147105.22797544891, 1e-12); + EXPECT_FALSE(converged); + + ifs.close(); + std::remove("test_check_converged_case4.log"); +} + +// Test for check_converged when ucell.lat_axis_free != 1, and largest_grad == 0 +TEST_F(LatticeChangeBasicTest, CheckConvergedCase5) +{ + // Set up test data + Lattice_Change_Basic::update_iter = 0; + stress_thr = 10.0; + std::ofstream ofs("test_check_converged_case5.log"); + ucell.lat_axis_free[0] = 0; + ucell.lat_axis_free[1] = 0; + ucell.lat_axis_free[2] = 0; + grad[0] = 0.0; + grad[1] = 0.0; + grad[2] = 0.0; + grad[3] = 0.0; + grad[4] = 0.0; + grad[5] = 0.0; + grad[6] = 0.0; + grad[7] = 0.0; + grad[8] = 0.0; + + // Call the function under test + bool converged = Lattice_Change_Basic::check_converged(ucell, stress, grad, ofs, stress_thr); + ofs.close(); + + // Check the results + std::ifstream ifs("test_check_converged_case5.log"); + std::string expected_output = " Largest stress is 0, movement is impossible.\n"; + std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + EXPECT_EQ(output, expected_output); + EXPECT_EQ(Lattice_Change_Basic::update_iter, 0); + EXPECT_DOUBLE_EQ(Lattice_Change_Basic::largest_grad, 0.0); + EXPECT_TRUE(converged); + + ifs.close(); + std::remove("test_check_converged_case5.log"); +} + +// Test for check_converged when ucell.lat_axis_free != 1, and converged +TEST_F(LatticeChangeBasicTest, CheckConvergedCase6) +{ + // Set up test data + Lattice_Change_Basic::update_iter = 0; + stress_thr = 10.0; + std::ofstream ofs("test_check_converged_case6.log"); + ucell.lat_axis_free[0] = 0; + ucell.lat_axis_free[1] = 0; + ucell.lat_axis_free[2] = 0; + grad[0] = 0.000001; + grad[1] = 0.0; + grad[2] = 0.0; + grad[3] = 0.0; + grad[4] = 0.0; + grad[5] = 0.0; + grad[6] = 0.0; + grad[7] = 0.0; + grad[8] = 0.0; + + // Call the function under test + bool converged = Lattice_Change_Basic::check_converged(ucell, stress, grad, ofs, stress_thr); + ofs.close(); + + // Check the results + std::ifstream ifs("test_check_converged_case6.log"); + std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + // Reporting moved to IonCellOptimizer::relax_step; nothing printed here now. + EXPECT_EQ(output, ""); + EXPECT_EQ(Lattice_Change_Basic::update_iter, 1); + EXPECT_NEAR(Lattice_Change_Basic::largest_grad, 0.14710522797544887, 1e-12); + EXPECT_TRUE(converged); + + ifs.close(); + std::remove("test_check_converged_case6.log"); +} + +TEST_F(LatticeChangeBasicTest, TerminateConverged) +{ + Lattice_Change_Basic::stress_step = 5; + Lattice_Change_Basic::update_iter = 10; + + std::string expected_output = " end of lattice optimization\n stress_step = 5\n " + " update iteration = 10\n"; + + std::ofstream ofs("test_terminate_converged.log"); + Lattice_Change_Basic::terminate(true, ofs); + ofs.close(); + + std::ifstream ifs("test_terminate_converged.log"); + std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + + EXPECT_EQ(expected_output, output); + ifs.close(); + std::remove("test_terminate_converged.log"); +} + +TEST_F(LatticeChangeBasicTest, TerminateNotConverged) +{ + std::string expected_output = " the maximum number of steps has been reached.\n end of lattice optimization.\n"; + + std::ofstream ofs("test_terminate_not_converged.log"); + Lattice_Change_Basic::terminate(false, ofs); + ofs.close(); + + std::ifstream ifs("test_terminate_not_converged.log"); + std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + + EXPECT_EQ(expected_output, output); + ifs.close(); + std::remove("test_terminate_not_converged.log"); +} + + +TEST_F(LatticeChangeBasicTest, SetupEtot) +{ + Lattice_Change_Basic::stress_step = 2; + double energy_in = 80.0; + std::vector etot_info = {100.0, 90.0}; + + Lattice_Change_Basic::setup_etot(energy_in, etot_info); + + EXPECT_DOUBLE_EQ(100.0, etot_info[1]); + EXPECT_DOUBLE_EQ(80.0, etot_info[0]); + EXPECT_DOUBLE_EQ(-20.0, etot_info[0] - etot_info[1]); +} + +// ============================================================================ +// NEW TESTS FOR SHAPE CONSTRAINT, VOLUME RESCALING, AND FIXED_IBRAV +// ============================================================================ + +// Test the setup_gradient function with fixed_axes = "shape" +TEST_F(LatticeChangeBasicTest, SetupGradientShape) +{ + // Initialize variables + ucell.lat_axis_free[0] = 1; + ucell.lat_axis_free[1] = 1; + ucell.lat_axis_free[2] = 1; + + // Non-isotropic stress tensor + stress(0, 0) = 1.0; + stress(0, 1) = 2.0; + stress(0, 2) = 3.0; + stress(1, 0) = 4.0; + stress(1, 1) = 5.0; + stress(1, 2) = 6.0; + stress(2, 0) = 7.0; + stress(2, 1) = 8.0; + stress(2, 2) = 9.0; + + Lattice_Change_Basic::fixed_axes = "shape"; + + // Call setup_gradient method + Lattice_Change_Basic::setup_gradient(ucell, lat, grad, stress); + + // Check that stress becomes isotropic (only diagonal, all equal) + // Average pressure = (1 + 5 + 9) / 3 = 5 + EXPECT_DOUBLE_EQ(stress(0, 0), 5.0); + EXPECT_DOUBLE_EQ(stress(1, 1), 5.0); + EXPECT_DOUBLE_EQ(stress(2, 2), 5.0); + + // Off-diagonal elements should be zero + EXPECT_DOUBLE_EQ(stress(0, 1), 0.0); + EXPECT_DOUBLE_EQ(stress(0, 2), 0.0); + EXPECT_DOUBLE_EQ(stress(1, 0), 0.0); + EXPECT_DOUBLE_EQ(stress(1, 2), 0.0); + EXPECT_DOUBLE_EQ(stress(2, 0), 0.0); + EXPECT_DOUBLE_EQ(stress(2, 1), 0.0); +} + +// Test volume constraint rescaling in change_lattice +TEST_F(LatticeChangeBasicTest, ChangeLatticeVolumeRescaling) +{ + // Initialize variables + ucell.lat_axis_free[0] = 1; + ucell.lat_axis_free[1] = 1; + ucell.lat_axis_free[2] = 1; + ucell.lat0 = 10.0; + + // Set initial lattice (cubic, volume = 1000) + ucell.latvec.e11 = 1.0; + ucell.latvec.e12 = 0.0; + ucell.latvec.e13 = 0.0; + ucell.latvec.e21 = 0.0; + ucell.latvec.e22 = 1.0; + ucell.latvec.e23 = 0.0; + ucell.latvec.e31 = 0.0; + ucell.latvec.e32 = 0.0; + ucell.latvec.e33 = 1.0; + + ucell.omega = 1000.0; // Initial volume + + lat[0] = 10.0; + lat[1] = 0.0; + lat[2] = 0.0; + lat[3] = 0.0; + lat[4] = 10.0; + lat[5] = 0.0; + lat[6] = 0.0; + lat[7] = 0.0; + lat[8] = 10.0; + + // Apply a move that would change volume (expand by 10%) + move[0] = 1.0; + move[1] = 0.0; + move[2] = 0.0; + move[3] = 0.0; + move[4] = 1.0; + move[5] = 0.0; + move[6] = 0.0; + move[7] = 0.0; + move[8] = 1.0; + + Lattice_Change_Basic::fixed_axes = "volume"; + + // Call change_lattice method + Lattice_Change_Basic::change_lattice(ucell, move, lat, fixed_ibrav); + + // Check that volume is preserved (should still be 1000) + EXPECT_NEAR(ucell.omega, 1000.0, 1e-8); + + // Check that lattice vectors were rescaled uniformly + double expected_scale = std::pow(1000.0 / 1331.0, 1.0/3.0); // (old_vol / new_vol)^(1/3) + EXPECT_NEAR(ucell.latvec.e11, 1.1 * expected_scale, 1e-10); + EXPECT_NEAR(ucell.latvec.e22, 1.1 * expected_scale, 1e-10); + EXPECT_NEAR(ucell.latvec.e33, 1.1 * expected_scale, 1e-10); +} + +// Test volume constraint with non-cubic cell +TEST_F(LatticeChangeBasicTest, ChangeLatticeVolumeRescalingNonCubic) +{ + // Initialize variables + ucell.lat_axis_free[0] = 1; + ucell.lat_axis_free[1] = 1; + ucell.lat_axis_free[2] = 1; + ucell.lat0 = 10.0; + + // Set initial lattice (non-cubic, volume = 1200) + ucell.latvec.e11 = 1.0; + ucell.latvec.e12 = 0.0; + ucell.latvec.e13 = 0.0; + ucell.latvec.e21 = 0.0; + ucell.latvec.e22 = 1.2; + ucell.latvec.e23 = 0.0; + ucell.latvec.e31 = 0.0; + ucell.latvec.e32 = 0.0; + ucell.latvec.e33 = 1.0; + + ucell.omega = 1200.0; // Initial volume + + lat[0] = 10.0; + lat[1] = 0.0; + lat[2] = 0.0; + lat[3] = 0.0; + lat[4] = 12.0; + lat[5] = 0.0; + lat[6] = 0.0; + lat[7] = 0.0; + lat[8] = 10.0; + + // Apply a move that would change volume + move[0] = 0.5; + move[1] = 0.0; + move[2] = 0.0; + move[3] = 0.0; + move[4] = -0.5; + move[5] = 0.0; + move[6] = 0.0; + move[7] = 0.0; + move[8] = 0.3; + + Lattice_Change_Basic::fixed_axes = "volume"; + + // Call change_lattice method + Lattice_Change_Basic::change_lattice(ucell, move, lat, fixed_ibrav); + + // Check that volume is preserved + EXPECT_NEAR(ucell.omega, 1200.0, 1e-8); +} + +// Test change_lattice without volume constraint (should change volume) +TEST_F(LatticeChangeBasicTest, ChangeLatticeNoVolumeConstraint) +{ + // Initialize variables + ucell.lat_axis_free[0] = 1; + ucell.lat_axis_free[1] = 1; + ucell.lat_axis_free[2] = 1; + ucell.lat0 = 10.0; + + // Set initial lattice + ucell.latvec.e11 = 1.0; + ucell.latvec.e12 = 0.0; + ucell.latvec.e13 = 0.0; + ucell.latvec.e21 = 0.0; + ucell.latvec.e22 = 1.0; + ucell.latvec.e23 = 0.0; + ucell.latvec.e31 = 0.0; + ucell.latvec.e32 = 0.0; + ucell.latvec.e33 = 1.0; + + ucell.omega = 1000.0; // Initial volume + + lat[0] = 10.0; + lat[1] = 0.0; + lat[2] = 0.0; + lat[3] = 0.0; + lat[4] = 10.0; + lat[5] = 0.0; + lat[6] = 0.0; + lat[7] = 0.0; + lat[8] = 10.0; + + // Apply a move that changes volume + move[0] = 1.0; + move[1] = 0.0; + move[2] = 0.0; + move[3] = 0.0; + move[4] = 1.0; + move[5] = 0.0; + move[6] = 0.0; + move[7] = 0.0; + move[8] = 1.0; + + Lattice_Change_Basic::fixed_axes = "None"; + + // Call change_lattice method + Lattice_Change_Basic::change_lattice(ucell, move, lat, fixed_ibrav); + + // Check that volume DID change (should be 1331) + EXPECT_NEAR(ucell.omega, 1331.0, 1e-8); + + // Check lattice vectors + EXPECT_DOUBLE_EQ(ucell.latvec.e11, 1.1); + EXPECT_DOUBLE_EQ(ucell.latvec.e22, 1.1); + EXPECT_DOUBLE_EQ(ucell.latvec.e33, 1.1); +} + +// Test fixed_ibrav with simple cubic lattice +TEST_F(LatticeChangeBasicTest, ChangeLatticeFixedIbravSimpleCubic) +{ + // Initialize variables + ucell.lat_axis_free[0] = 1; + ucell.lat_axis_free[1] = 1; + ucell.lat_axis_free[2] = 1; + ucell.lat0 = 10.0; + ucell.latName = "sc"; + + // Set initial lattice (slightly distorted cubic) + ucell.latvec.e11 = 1.0; + ucell.latvec.e12 = 0.01; // Small distortion + ucell.latvec.e13 = 0.0; + ucell.latvec.e21 = 0.0; + ucell.latvec.e22 = 1.0; + ucell.latvec.e23 = 0.0; + ucell.latvec.e31 = 0.0; + ucell.latvec.e32 = 0.0; + ucell.latvec.e33 = 1.0; + + lat[0] = 10.0; + lat[1] = 0.1; + lat[2] = 0.0; + lat[3] = 0.0; + lat[4] = 10.0; + lat[5] = 0.0; + lat[6] = 0.0; + lat[7] = 0.0; + lat[8] = 10.0; + + // Apply a move + move[0] = 0.1; + move[1] = 0.0; + move[2] = 0.0; + move[3] = 0.0; + move[4] = 0.1; + move[5] = 0.0; + move[6] = 0.0; + move[7] = 0.0; + move[8] = 0.1; + + fixed_ibrav = true; + Lattice_Change_Basic::fixed_axes = "None"; + + // Verify remake_cell was not called yet + EXPECT_FALSE(unitcell::was_remake_cell_called()); + + // Call change_lattice method + Lattice_Change_Basic::change_lattice(ucell, move, lat, fixed_ibrav); + + // Verify remake_cell was called + EXPECT_TRUE(unitcell::was_remake_cell_called()); + + // Check that lattice is now perfect cubic (all diagonal, equal) + // This is enforced by the mock remake_cell function + EXPECT_NEAR(ucell.latvec.e11, ucell.latvec.e22, 1e-10); + EXPECT_NEAR(ucell.latvec.e22, ucell.latvec.e33, 1e-10); + EXPECT_NEAR(ucell.latvec.e12, 0.0, 1e-10); + EXPECT_NEAR(ucell.latvec.e13, 0.0, 1e-10); + EXPECT_NEAR(ucell.latvec.e21, 0.0, 1e-10); + EXPECT_NEAR(ucell.latvec.e23, 0.0, 1e-10); + EXPECT_NEAR(ucell.latvec.e31, 0.0, 1e-10); + EXPECT_NEAR(ucell.latvec.e32, 0.0, 1e-10); + + // Reset for other tests + fixed_ibrav = false; +} + +// Test fixed_ibrav with FCC lattice +TEST_F(LatticeChangeBasicTest, ChangeLatticeFixedIbravFCC) +{ + // Initialize variables + ucell.lat_axis_free[0] = 1; + ucell.lat_axis_free[1] = 1; + ucell.lat_axis_free[2] = 1; + ucell.lat0 = 10.0; + ucell.latName = "fcc"; + + // Set initial lattice (slightly distorted FCC) + double celldm = 1.0; + ucell.latvec.e11 = -celldm + 0.01; // Small distortion + ucell.latvec.e12 = 0.0; + ucell.latvec.e13 = celldm; + ucell.latvec.e21 = 0.0; + ucell.latvec.e22 = celldm; + ucell.latvec.e23 = celldm; + ucell.latvec.e31 = -celldm; + ucell.latvec.e32 = celldm; + ucell.latvec.e33 = 0.0; + + lat[0] = (-celldm + 0.01) * ucell.lat0; + lat[1] = 0.0; + lat[2] = celldm * ucell.lat0; + lat[3] = 0.0; + lat[4] = celldm * ucell.lat0; + lat[5] = celldm * ucell.lat0; + lat[6] = -celldm * ucell.lat0; + lat[7] = celldm * ucell.lat0; + lat[8] = 0.0; + + // Apply a small move + for (int i = 0; i < 9; i++) move[i] = 0.01 * ucell.lat0; + + fixed_ibrav = true; + Lattice_Change_Basic::fixed_axes = "None"; + + // Verify remake_cell was not called yet + EXPECT_FALSE(unitcell::was_remake_cell_called()); + + // Call change_lattice method + Lattice_Change_Basic::change_lattice(ucell, move, lat, fixed_ibrav); + + // Verify remake_cell was called + EXPECT_TRUE(unitcell::was_remake_cell_called()); + + // Check that lattice maintains FCC structure + // For FCC: a1 = (-a, 0, a), a2 = (0, a, a), a3 = (-a, a, 0) + // All should have same magnitude + double mag1 = std::sqrt(ucell.latvec.e11*ucell.latvec.e11 + + ucell.latvec.e12*ucell.latvec.e12 + + ucell.latvec.e13*ucell.latvec.e13); + double mag2 = std::sqrt(ucell.latvec.e21*ucell.latvec.e21 + + ucell.latvec.e22*ucell.latvec.e22 + + ucell.latvec.e23*ucell.latvec.e23); + double mag3 = std::sqrt(ucell.latvec.e31*ucell.latvec.e31 + + ucell.latvec.e32*ucell.latvec.e32 + + ucell.latvec.e33*ucell.latvec.e33); + + EXPECT_NEAR(mag1, mag2, 1e-10); + EXPECT_NEAR(mag2, mag3, 1e-10); + + // Check FCC structure: e11 should be negative, e13 positive, e12 = 0 + EXPECT_LT(ucell.latvec.e11, 0.0); + EXPECT_GT(ucell.latvec.e13, 0.0); + EXPECT_NEAR(ucell.latvec.e12, 0.0, 1e-10); + + // Reset for other tests + fixed_ibrav = false; +} + +// Test combination of fixed_axes = "volume" and fixed_ibrav +TEST_F(LatticeChangeBasicTest, ChangeLatticeVolumeAndIbrav) +{ + // Initialize variables + ucell.lat_axis_free[0] = 1; + ucell.lat_axis_free[1] = 1; + ucell.lat_axis_free[2] = 1; + ucell.lat0 = 10.0; + ucell.latName = "sc"; + + // Set initial lattice (cubic) + ucell.latvec.e11 = 1.0; + ucell.latvec.e12 = 0.0; + ucell.latvec.e13 = 0.0; + ucell.latvec.e21 = 0.0; + ucell.latvec.e22 = 1.0; + ucell.latvec.e23 = 0.0; + ucell.latvec.e31 = 0.0; + ucell.latvec.e32 = 0.0; + ucell.latvec.e33 = 1.0; + + ucell.omega = 1000.0; // Initial volume + + lat[0] = 10.0; + lat[1] = 0.0; + lat[2] = 0.0; + lat[3] = 0.0; + lat[4] = 10.0; + lat[5] = 0.0; + lat[6] = 0.0; + lat[7] = 0.0; + lat[8] = 10.0; + + // Apply a move that would change volume and break symmetry + move[0] = 1.0; + move[1] = 0.1; // Try to break symmetry + move[2] = 0.0; + move[3] = 0.0; + move[4] = 0.8; + move[5] = 0.0; + move[6] = 0.0; + move[7] = 0.0; + move[8] = 1.2; + + fixed_ibrav = true; + Lattice_Change_Basic::fixed_axes = "volume"; + + // Verify remake_cell was not called yet + EXPECT_FALSE(unitcell::was_remake_cell_called()); + + // Call change_lattice method + Lattice_Change_Basic::change_lattice(ucell, move, lat, fixed_ibrav); + + // Verify remake_cell was called (should be called before volume rescaling) + EXPECT_TRUE(unitcell::was_remake_cell_called()); + + // Check that volume is preserved + EXPECT_NEAR(ucell.omega, 1000.0, 1e-8); + + // Check that lattice is cubic (all diagonal, equal) + EXPECT_NEAR(ucell.latvec.e11, ucell.latvec.e22, 1e-10); + EXPECT_NEAR(ucell.latvec.e22, ucell.latvec.e33, 1e-10); + EXPECT_NEAR(ucell.latvec.e12, 0.0, 1e-10); + EXPECT_NEAR(ucell.latvec.e13, 0.0, 1e-10); + EXPECT_NEAR(ucell.latvec.e21, 0.0, 1e-10); + EXPECT_NEAR(ucell.latvec.e23, 0.0, 1e-10); + EXPECT_NEAR(ucell.latvec.e31, 0.0, 1e-10); + EXPECT_NEAR(ucell.latvec.e32, 0.0, 1e-10); + + // Reset for other tests + fixed_ibrav = false; +} + +// Test axis constraint with fixed_axes = "a" +TEST_F(LatticeChangeBasicTest, SetupGradientAxisA) +{ + // Initialize variables + ucell.lat_axis_free[0] = 0; // First axis fixed + ucell.lat_axis_free[1] = 1; + ucell.lat_axis_free[2] = 1; + + stress(0, 0) = 1.0; + stress(0, 1) = 2.0; + stress(0, 2) = 3.0; + stress(1, 0) = 4.0; + stress(1, 1) = 5.0; + stress(1, 2) = 6.0; + stress(2, 0) = 7.0; + stress(2, 1) = 8.0; + stress(2, 2) = 9.0; + + Lattice_Change_Basic::fixed_axes = "a"; + + // Call setup_gradient method + Lattice_Change_Basic::setup_gradient(ucell, lat, grad, stress); + + // Check that gradient for first lattice vector is zero + EXPECT_DOUBLE_EQ(grad[0], 0.0); + EXPECT_DOUBLE_EQ(grad[1], 0.0); + EXPECT_DOUBLE_EQ(grad[2], 0.0); + + // Check that gradients for other vectors are non-zero + EXPECT_NE(grad[3], 0.0); + EXPECT_NE(grad[4], 0.0); + EXPECT_NE(grad[5], 0.0); + EXPECT_NE(grad[6], 0.0); + EXPECT_NE(grad[7], 0.0); + EXPECT_NE(grad[8], 0.0); +} + +// Test that fixed axis doesn't move in change_lattice +TEST_F(LatticeChangeBasicTest, ChangeLatticeFixedAxisA) +{ + // Initialize variables + ucell.lat_axis_free[0] = 0; // First axis fixed + ucell.lat_axis_free[1] = 1; + ucell.lat_axis_free[2] = 1; + ucell.lat0 = 10.0; + + // Set initial lattice + ucell.latvec.e11 = 1.0; + ucell.latvec.e12 = 0.1; + ucell.latvec.e13 = 0.2; + ucell.latvec.e21 = 0.0; + ucell.latvec.e22 = 1.0; + ucell.latvec.e23 = 0.0; + ucell.latvec.e31 = 0.0; + ucell.latvec.e32 = 0.0; + ucell.latvec.e33 = 1.0; + + // Save initial first lattice vector + double initial_e11 = ucell.latvec.e11; + double initial_e12 = ucell.latvec.e12; + double initial_e13 = ucell.latvec.e13; + + lat[0] = 10.0; + lat[1] = 1.0; + lat[2] = 2.0; + lat[3] = 0.0; + lat[4] = 10.0; + lat[5] = 0.0; + lat[6] = 0.0; + lat[7] = 0.0; + lat[8] = 10.0; + + // Try to move all lattice vectors + move[0] = 1.0; + move[1] = 0.5; + move[2] = 0.3; + move[3] = 0.5; + move[4] = 1.0; + move[5] = 0.0; + move[6] = 0.0; + move[7] = 0.0; + move[8] = 1.0; + + Lattice_Change_Basic::fixed_axes = "a"; + + // Call change_lattice method + Lattice_Change_Basic::change_lattice(ucell, move, lat, fixed_ibrav); + + // Check that first lattice vector didn't change + EXPECT_DOUBLE_EQ(ucell.latvec.e11, initial_e11); + EXPECT_DOUBLE_EQ(ucell.latvec.e12, initial_e12); + EXPECT_DOUBLE_EQ(ucell.latvec.e13, initial_e13); + + // Check that other lattice vectors did change + EXPECT_NE(ucell.latvec.e21, 0.0); + EXPECT_NE(ucell.latvec.e22, 1.0); + EXPECT_NE(ucell.latvec.e33, 1.0); +} + +// Test that remake_cell is NOT called when fixed_ibrav = false +TEST_F(LatticeChangeBasicTest, ChangeLatticeNoFixedIbrav) +{ + // Initialize variables + ucell.lat_axis_free[0] = 1; + ucell.lat_axis_free[1] = 1; + ucell.lat_axis_free[2] = 1; + ucell.lat0 = 10.0; + ucell.latName = "sc"; + + // Set initial lattice + ucell.latvec.e11 = 1.0; + ucell.latvec.e12 = 0.01; // Small distortion + ucell.latvec.e13 = 0.0; + ucell.latvec.e21 = 0.0; + ucell.latvec.e22 = 1.0; + ucell.latvec.e23 = 0.0; + ucell.latvec.e31 = 0.0; + ucell.latvec.e32 = 0.0; + ucell.latvec.e33 = 1.0; + + lat[0] = 10.0; + lat[1] = 0.1; + lat[2] = 0.0; + lat[3] = 0.0; + lat[4] = 10.0; + lat[5] = 0.0; + lat[6] = 0.0; + lat[7] = 0.0; + lat[8] = 10.0; + + // Apply a move + move[0] = 0.1; + move[1] = 0.0; + move[2] = 0.0; + move[3] = 0.0; + move[4] = 0.1; + move[5] = 0.0; + move[6] = 0.0; + move[7] = 0.0; + move[8] = 0.1; + + fixed_ibrav = false; // Explicitly set to false + Lattice_Change_Basic::fixed_axes = "None"; + + // Verify remake_cell was not called yet + EXPECT_FALSE(unitcell::was_remake_cell_called()); + + // Call change_lattice method + Lattice_Change_Basic::change_lattice(ucell, move, lat, fixed_ibrav); + + // Verify remake_cell was NOT called + EXPECT_FALSE(unitcell::was_remake_cell_called()); + + // Check that distortion remains (e12 should still be non-zero) + EXPECT_NE(ucell.latvec.e12, 0.0); +} diff --git a/source/source_relax/unittests/test_lattice_change_cg.cpp b/source/source_relax/unittests/test_lattice_change_cg.cpp new file mode 100644 index 00000000000..fc073052cb9 --- /dev/null +++ b/source/source_relax/unittests/test_lattice_change_cg.cpp @@ -0,0 +1,339 @@ +#include "source_relax/relax_criteria.h" +#include "for_test.h" +#include "gtest/gtest.h" +#include "mock_remake_cell.h" +#include "source_relax/lattice_change_basic.h" +#include "source_relax/lattice_change_cg.h" + +/************************************************ + * unit tests of class Lattice_Change_CG + ***********************************************/ + +class LatticeChangeCGTest : public ::testing::Test +{ + public: + Relax_Criteria criteria; + + protected: + void SetUp() override + { + // Initialize variables before each test + Lattice_Change_Basic::dim = 9; + Lattice_Change_Basic::stress_step = 1; + Lattice_Change_Basic::update_iter = 5; + lc_cg.allocate(); + etot_info.resize(2, 0.0); + } + + void TearDown() override + { + // Clean up after each test + } + + Lattice_Change_CG lc_cg; + std::vector etot_info; + + // Friendship is not inherited: a TEST_F body lives in a class derived from + // this fixture, so the internals of Lattice_Change_CG (which friends the + // fixture itself) are reached through these forwarders. + const std::vector& get_lat0() const + { + return lc_cg.lat0; + } + const std::vector& get_grad0() const + { + return lc_cg.grad0; + } + const std::vector& get_cg_grad0() const + { + return lc_cg.cg_grad0; + } + const std::vector& get_move0() const + { + return lc_cg.move0; + } + void set_move0(const int i, const double value) + { + lc_cg.move0[i] = value; + } +}; + +// Test whether the allocate() function can correctly allocate memory space +TEST_F(LatticeChangeCGTest, TestAllocate) +{ + Lattice_Change_Basic::dim = 4; + lc_cg.allocate(); + + // Check if allocated vectors are not empty + EXPECT_EQ(get_lat0().size(), 4U); + EXPECT_EQ(get_grad0().size(), 4U); + EXPECT_EQ(get_cg_grad0().size(), 4U); + EXPECT_EQ(get_move0().size(), 4U); +} + +// Test if a dimension less than or equal to 0 results in an assertion error +TEST_F(LatticeChangeCGTest, TestAllocateWithZeroDimension) +{ + Lattice_Change_Basic::dim = 0; + ASSERT_DEATH(lc_cg.allocate(), ""); +} + +// Check that the arrays are correctly initialized to 0 +TEST_F(LatticeChangeCGTest, TestAllocateAndInitialize) +{ + Lattice_Change_Basic::dim = 3; + lc_cg.allocate(); + + // Check that the arrays are correctly initialized to 0 + EXPECT_DOUBLE_EQ(0.0, get_lat0()[0]); + EXPECT_DOUBLE_EQ(0.0, get_grad0()[1]); + EXPECT_DOUBLE_EQ(0.0, get_cg_grad0()[2]); + EXPECT_DOUBLE_EQ(0.0, get_move0()[0]); +} + +// Test function start() when converged +TEST_F(LatticeChangeCGTest, TestStartConverged) +{ + // setup data + UnitCell ucell; + ucell.lat_axis_free[0] = 1; + ucell.lat_axis_free[1] = 1; + ucell.lat_axis_free[2] = 1; + ModuleBase::matrix stress(3, 3); + double etot = 0.0; + + // call function + std::ofstream ofs("test_lc_cg_start_converged.log"); + lc_cg.start(ucell, stress, etot, ofs, etot_info, criteria); + ofs.close(); + + // Check output + std::string expected_output + = " Largest stress is 0, movement is impossible.\n end of lattice optimization\n stress_step = 1\n " + " update iteration = 5\n"; + std::ifstream ifs("test_lc_cg_start_converged.log"); + std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + + EXPECT_EQ(expected_output, output); + ifs.close(); + std::remove("test_lc_cg_start_converged.log"); +} + +// Test function start() sd branch +TEST_F(LatticeChangeCGTest, TestStartSd) +{ + // setup data + UnitCell ucell; + ucell.lat_axis_free[0] = 1; + ucell.lat_axis_free[1] = 1; + ucell.lat_axis_free[2] = 1; + ModuleBase::matrix stress(3, 3); + stress(0, 0) = 0.01; + double etot = 0.0; + + // call function + std::ofstream ofs("test_lc_cg_start_sd.log"); + lc_cg.start(ucell, stress, etot, ofs, etot_info, criteria); + ofs.close(); + + // Check output: reporting moved to IonCellOptimizer::relax_step; + // check_converged prints nothing here anymore. + std::string expected_output = ""; + std::ifstream ifs("test_lc_cg_start_sd.log"); + std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + + EXPECT_EQ(expected_output, output); + EXPECT_DOUBLE_EQ(Lattice_Change_Basic::lattice_change_ini, 0.01); + ifs.close(); + std::remove("test_lc_cg_start_sd.log"); +} + +// Test function start() trial branch with goto +TEST_F(LatticeChangeCGTest, TestStartTrialGoto) +{ + // setup data + UnitCell ucell; + ucell.lat_axis_free[0] = 1; + ucell.lat_axis_free[1] = 1; + ucell.lat_axis_free[2] = 1; + ModuleBase::matrix stress(3, 3); + stress(0, 1) = 0.01; + double etot = 0.0; + + // call function + set_move0(0, 1.0); + std::ofstream ofs1("test_lc_cg_start_trial_goto_temp1.log"); + lc_cg.start(ucell, stress, etot, ofs1, etot_info, criteria); + ofs1.close(); + std::remove("test_lc_cg_start_trial_goto_temp1.log"); + Lattice_Change_Basic::stress_step = 2; + set_move0(0, 10.0); + std::ofstream ofs("test_lc_cg_start_trial_goto.log"); + lc_cg.start(ucell, stress, etot, ofs, etot_info, criteria); + ofs.close(); + + // Check output: reporting moved to IonCellOptimizer::relax_step; + // check_converged prints nothing here anymore. + std::string expected_output = ""; + std::ifstream ifs("test_lc_cg_start_trial_goto.log"); + std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + + EXPECT_EQ(expected_output, output); + EXPECT_NEAR(Lattice_Change_Basic::lattice_change_ini, 10.000004999998749, 1e-12); + ifs.close(); + std::remove("test_lc_cg_start_trial_goto.log"); +} + +// Test function start() trial branch without goto +TEST_F(LatticeChangeCGTest, TestStartTrial) +{ + // setup data + UnitCell ucell; + ucell.lat_axis_free[0] = 1; + ucell.lat_axis_free[1] = 1; + ucell.lat_axis_free[2] = 1; + ModuleBase::matrix stress(3, 3); + stress(0, 1) = 0.01; + double etot = 0.0; + + // call function + std::ofstream ofs1("test_lc_cg_start_trial_temp1.log"); + lc_cg.start(ucell, stress, etot, ofs1, etot_info, criteria); + ofs1.close(); + std::remove("test_lc_cg_start_trial_temp1.log"); + Lattice_Change_Basic::stress_step = 2; + std::ofstream ofs("test_lc_cg_start_trial.log"); + lc_cg.start(ucell, stress, etot, ofs, etot_info, criteria); + ofs.close(); + + // Check output: reporting moved to IonCellOptimizer::relax_step; + // check_converged prints nothing here anymore. + std::string expected_output = ""; + std::ifstream ifs("test_lc_cg_start_trial.log"); + std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + + EXPECT_EQ(expected_output, output); + EXPECT_NEAR(Lattice_Change_Basic::lattice_change_ini, 70.000034999991243, 1e-12); + ifs.close(); + std::remove("test_lc_cg_start_trial.log"); +} + +// Test function start() no trial branch with goto case 1 +TEST_F(LatticeChangeCGTest, TestStartNoTrialGotoCase1) +{ + // setup data + UnitCell ucell; + ucell.lat_axis_free[0] = 1; + ucell.lat_axis_free[1] = 1; + ucell.lat_axis_free[2] = 1; + ModuleBase::matrix stress(3, 3); + stress(0, 1) = 0.01; + double etot = 0.0; + + // call function + std::ofstream ofs1("test_lc_cg_start_notrial_goto_case1_temp1.log"); + lc_cg.start(ucell, stress, etot, ofs1, etot_info, criteria); + ofs1.close(); + std::remove("test_lc_cg_start_notrial_goto_case1_temp1.log"); + Lattice_Change_Basic::stress_step = 2; + std::ofstream ofs2("test_lc_cg_start_notrial_goto_case1_temp2.log"); + lc_cg.start(ucell, stress, etot, ofs2, etot_info, criteria); + ofs2.close(); + std::remove("test_lc_cg_start_notrial_goto_case1_temp2.log"); + std::ofstream ofs("test_lc_cg_start_notrial_goto_case1.log"); + lc_cg.start(ucell, stress, etot, ofs, etot_info, criteria); + ofs.close(); + + // Check output: reporting moved to IonCellOptimizer::relax_step; + // check_converged prints nothing here anymore. + std::string expected_output = ""; + std::ifstream ifs("test_lc_cg_start_notrial_goto_case1.log"); + std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + + EXPECT_EQ(expected_output, output); + EXPECT_NEAR(Lattice_Change_Basic::lattice_change_ini, 490.00024499993867, 1e-12); + ifs.close(); + std::remove("test_lc_cg_start_notrial_goto_case1.log"); +} + +// Test function start() no trial branch with goto case 2 +TEST_F(LatticeChangeCGTest, TestStartNoTrialGotoCase2) +{ + // setup data + UnitCell ucell; + ucell.lat_axis_free[0] = 1; + ucell.lat_axis_free[1] = 1; + ucell.lat_axis_free[2] = 1; + ModuleBase::matrix stress(3, 3); + stress(0, 1) = 0.01; + double etot = 0.0; + + // call function + set_move0(0, 0.1); + std::ofstream ofs1("test_lc_cg_start_notrial_goto_case2_temp1.log"); + lc_cg.start(ucell, stress, etot, ofs1, etot_info, criteria); + ofs1.close(); + std::remove("test_lc_cg_start_notrial_goto_case2_temp1.log"); + Lattice_Change_Basic::stress_step = 2; + std::ofstream ofs2("test_lc_cg_start_notrial_goto_case2_temp2.log"); + lc_cg.start(ucell, stress, etot, ofs2, etot_info, criteria); + ofs2.close(); + std::remove("test_lc_cg_start_notrial_goto_case2_temp2.log"); + std::ofstream ofs("test_lc_cg_start_notrial_goto_case2.log"); + set_move0(0, 0.1); + stress(0, 1) = 0.0001; + lc_cg.start(ucell, stress, etot, ofs, etot_info, criteria); + ofs.close(); + + // Check output: reporting moved to IonCellOptimizer::relax_step; + // check_converged prints nothing here anymore. + std::string expected_output = ""; + std::ifstream ifs("test_lc_cg_start_notrial_goto_case2.log"); + std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + + EXPECT_EQ(expected_output, output); + EXPECT_NEAR(Lattice_Change_Basic::lattice_change_ini, 3430.0017149995706, 1e-12); + ifs.close(); + std::remove("test_lc_cg_start_notrial_goto_case2.log"); +} + +// Test function start() no trial branch without goto +TEST_F(LatticeChangeCGTest, TestStartNoTrial) +{ + // setup data + UnitCell ucell; + ucell.lat_axis_free[0] = 1; + ucell.lat_axis_free[1] = 1; + ucell.lat_axis_free[2] = 1; + ModuleBase::matrix stress(3, 3); + stress(0, 1) = 0.01; + double etot = 0.0; + + // call function + set_move0(0, 1.0); + std::ofstream ofs1("test_lc_cg_start_notrial_temp1.log"); + lc_cg.start(ucell, stress, etot, ofs1, etot_info, criteria); + ofs1.close(); + std::remove("test_lc_cg_start_notrial_temp1.log"); + Lattice_Change_Basic::stress_step = 2; + set_move0(0, 10.0); + std::ofstream ofs2("test_lc_cg_start_notrial_temp2.log"); + lc_cg.start(ucell, stress, etot, ofs2, etot_info, criteria); + ofs2.close(); + std::remove("test_lc_cg_start_notrial_temp2.log"); + std::ofstream ofs("test_lc_cg_start_notrial.log"); + lc_cg.start(ucell, stress, etot, ofs, etot_info, criteria); + ofs.close(); + + // Check output: reporting moved to IonCellOptimizer::relax_step; + // check_converged prints nothing here anymore. + std::string expected_output = ""; + std::ifstream ifs("test_lc_cg_start_notrial.log"); + std::string output((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + + EXPECT_EQ(expected_output, output); + EXPECT_NEAR(Lattice_Change_Basic::lattice_change_ini, 96040.106328872833, 1e-12); + ifs.close(); + std::remove("test_lc_cg_start_notrial.log"); +} + diff --git a/source/source_relax/test/line_search_test.cpp b/source/source_relax/unittests/test_line_search.cpp similarity index 100% rename from source/source_relax/test/line_search_test.cpp rename to source/source_relax/unittests/test_line_search.cpp diff --git a/source/source_relax/unittests/test_matrix_methods.cpp b/source/source_relax/unittests/test_matrix_methods.cpp new file mode 100644 index 00000000000..c40d86b32ef --- /dev/null +++ b/source/source_relax/unittests/test_matrix_methods.cpp @@ -0,0 +1,186 @@ +#include "../matrix_methods.h" + +#include + +#include + +// Unit tests for the free matrix/vector helpers in matrix_methods.cpp. +// All functions are pure, so the tests assert exact values with hand-picked +// inputs. Two behaviors worth pinning down: DotInMAndV1 vs DotInMAndV2 differ +// in row/column orientation, and MPlus actually divides by its scalar. + +TEST(MatrixMethods, ReshapeMToVFlattens) +{ + std::vector> matrix; + matrix.push_back(ModuleBase::Vector3(1.0, 2.0, 3.0)); + matrix.push_back(ModuleBase::Vector3(4.0, 5.0, 6.0)); + + std::vector result = ReshapeMToV(matrix); + + ASSERT_EQ(result.size(), 6); + EXPECT_DOUBLE_EQ(result[0], 1.0); + EXPECT_DOUBLE_EQ(result[1], 2.0); + EXPECT_DOUBLE_EQ(result[2], 3.0); + EXPECT_DOUBLE_EQ(result[3], 4.0); + EXPECT_DOUBLE_EQ(result[4], 5.0); + EXPECT_DOUBLE_EQ(result[5], 6.0); +} + +TEST(MatrixMethods, ReshapeVToMGroupsByThree) +{ + std::vector vec = {1.0, 2.0, 3.0, 4.0, 5.0, 6.0}; + + std::vector> result = ReshapeVToM(vec); + + ASSERT_EQ(result.size(), 2); + EXPECT_DOUBLE_EQ(result[0].x, 1.0); + EXPECT_DOUBLE_EQ(result[0].y, 2.0); + EXPECT_DOUBLE_EQ(result[0].z, 3.0); + EXPECT_DOUBLE_EQ(result[1].x, 4.0); + EXPECT_DOUBLE_EQ(result[1].y, 5.0); + EXPECT_DOUBLE_EQ(result[1].z, 6.0); +} + +TEST(MatrixMethods, ReshapeRoundTrip) +{ + std::vector> matrix; + matrix.push_back(ModuleBase::Vector3(1.5, -2.5, 3.5)); + matrix.push_back(ModuleBase::Vector3(-4.5, 5.5, -6.5)); + + std::vector flat = ReshapeMToV(matrix); + std::vector> result = ReshapeVToM(flat); + + ASSERT_EQ(result.size(), matrix.size()); + for (size_t i = 0; i < matrix.size(); ++i) + { + EXPECT_DOUBLE_EQ(result[i].x, matrix[i].x); + EXPECT_DOUBLE_EQ(result[i].y, matrix[i].y); + EXPECT_DOUBLE_EQ(result[i].z, matrix[i].z); + } +} + +TEST(MatrixMethods, MAddMElementwise) +{ + std::vector> a = {{1.0, 2.0}, {3.0, 4.0}}; + std::vector> b = {{5.0, 6.0}, {7.0, 8.0}}; + + std::vector> result = MAddM(a, b); + + EXPECT_DOUBLE_EQ(result[0][0], 6.0); + EXPECT_DOUBLE_EQ(result[0][1], 8.0); + EXPECT_DOUBLE_EQ(result[1][0], 10.0); + EXPECT_DOUBLE_EQ(result[1][1], 12.0); +} + +TEST(MatrixMethods, MSubMElementwise) +{ + std::vector> a = {{5.0, 7.0}, {9.0, 11.0}}; + std::vector> b = {{1.0, 2.0}, {3.0, 4.0}}; + + std::vector> result = MSubM(a, b); + + EXPECT_DOUBLE_EQ(result[0][0], 4.0); + EXPECT_DOUBLE_EQ(result[0][1], 5.0); + EXPECT_DOUBLE_EQ(result[1][0], 6.0); + EXPECT_DOUBLE_EQ(result[1][1], 7.0); +} + +TEST(MatrixMethods, VAddVElementwise) +{ + std::vector a = {1.0, 2.0, 3.0}; + std::vector b = {4.0, 5.0, 6.0}; + + std::vector result = VAddV(a, b); + + ASSERT_EQ(result.size(), 3); + EXPECT_DOUBLE_EQ(result[0], 5.0); + EXPECT_DOUBLE_EQ(result[1], 7.0); + EXPECT_DOUBLE_EQ(result[2], 9.0); +} + +TEST(MatrixMethods, VSubVElementwise) +{ + std::vector a = {5.0, 7.0, 9.0}; + std::vector b = {1.0, 2.0, 3.0}; + + std::vector result = VSubV(a, b); + + ASSERT_EQ(result.size(), 3); + EXPECT_DOUBLE_EQ(result[0], 4.0); + EXPECT_DOUBLE_EQ(result[1], 5.0); + EXPECT_DOUBLE_EQ(result[2], 6.0); +} + +TEST(MatrixMethods, DotInMAndV1RowMajor) +{ + // result[i] = sum_j matrix[i][j] * vec[j] + std::vector> matrix = {{1.0, 2.0}, {3.0, 4.0}}; + std::vector vec = {5.0, 6.0}; + + std::vector result = DotInMAndV1(matrix, vec); + + ASSERT_EQ(result.size(), 2); + EXPECT_DOUBLE_EQ(result[0], 17.0); // 1*5 + 2*6 + EXPECT_DOUBLE_EQ(result[1], 39.0); // 3*5 + 4*6 +} + +TEST(MatrixMethods, DotInMAndV2ColumnMajor) +{ + // result[i] = sum_j matrix[j][i] * vec[j] + std::vector> matrix = {{1.0, 2.0}, {3.0, 4.0}}; + std::vector vec = {5.0, 6.0}; + + std::vector result = DotInMAndV2(matrix, vec); + + ASSERT_EQ(result.size(), 2); + EXPECT_DOUBLE_EQ(result[0], 23.0); // 1*5 + 3*6 + EXPECT_DOUBLE_EQ(result[1], 34.0); // 2*5 + 4*6 +} + +TEST(MatrixMethods, DotInVAndVScalar) +{ + std::vector vec1 = {1.0, 2.0, 3.0}; + std::vector vec2 = {4.0, 5.0, 6.0}; + + double result = DotInVAndV(vec1, vec2); + + EXPECT_DOUBLE_EQ(result, 32.0); // 4 + 10 + 18 +} + +TEST(MatrixMethods, OuterVAndVProducesMatrix) +{ + std::vector a = {1.0, 2.0}; + std::vector b = {3.0, 4.0}; + + std::vector> result = OuterVAndV(a, b); + + EXPECT_DOUBLE_EQ(result[0][0], 3.0); + EXPECT_DOUBLE_EQ(result[0][1], 4.0); + EXPECT_DOUBLE_EQ(result[1][0], 6.0); + EXPECT_DOUBLE_EQ(result[1][1], 8.0); +} + +TEST(MatrixMethods, MPlusDividesByScalar) +{ + // Despite the name, MPlus divides every element by the scalar. + std::vector> a = {{2.0, 4.0}, {6.0, 8.0}}; + + std::vector> result = MPlus(a, 2.0); + + EXPECT_DOUBLE_EQ(result[0][0], 1.0); + EXPECT_DOUBLE_EQ(result[0][1], 2.0); + EXPECT_DOUBLE_EQ(result[1][0], 3.0); + EXPECT_DOUBLE_EQ(result[1][1], 4.0); +} + +TEST(MatrixMethods, DotInVAndFloatScales) +{ + std::vector vec = {1.0, 2.0, 3.0}; + + std::vector result = DotInVAndFloat(vec, 3.0); + + ASSERT_EQ(result.size(), 3); + EXPECT_DOUBLE_EQ(result[0], 3.0); + EXPECT_DOUBLE_EQ(result[1], 6.0); + EXPECT_DOUBLE_EQ(result[2], 9.0); +} diff --git a/source/source_relax/unittests/test_relax_history.cpp b/source/source_relax/unittests/test_relax_history.cpp new file mode 100644 index 00000000000..c00b13ffaae --- /dev/null +++ b/source/source_relax/unittests/test_relax_history.cpp @@ -0,0 +1,44 @@ +#include "../relax_history.h" + +#include "gmock/gmock.h" +#include "gtest/gtest.h" + +using testing::HasSubstr; +using testing::Not; + +TEST(RelaxHistory, EmptyHistory) +{ + const std::vector hist; + EXPECT_EQ(format_relax_history(hist), "\n"); +} + +TEST(RelaxHistory, FewStepsAllShown) +{ + const std::vector hist = {1.0, 2.0, 3.0}; + const std::string out = format_relax_history(hist); + EXPECT_THAT(out, HasSubstr("1.000e+00")); + EXPECT_THAT(out, HasSubstr("2.000e+00")); + EXPECT_THAT(out, HasSubstr("3.000e+00")); + EXPECT_THAT(out, Not(HasSubstr("omitted"))); +} + +TEST(RelaxHistory, LineBreakEveryFive) +{ + const std::vector hist(7, 1.0); + const std::string out = format_relax_history(hist); + // 7 values with per_line = 5 -> two leading "\n " line starts + EXPECT_THAT(out, HasSubstr("\n 1.000e+00 1.000e+00 1.000e+00 1.000e+00 1.000e+00\n 1.000e+00")); +} + +TEST(RelaxHistory, LongHistoryTruncated) +{ + std::vector hist(150, 0.0); + hist.front() = 1.0; // first value kept + hist.back() = 2.0; // last value kept + hist[75] = 9.0; // middle value must be omitted + const std::string out = format_relax_history(hist); + EXPECT_THAT(out, HasSubstr("... (omitted 130 step(s)) ...")); + EXPECT_THAT(out, HasSubstr("1.000e+00")); + EXPECT_THAT(out, HasSubstr("2.000e+00")); + EXPECT_THAT(out, Not(HasSubstr("9.000e+00"))); +} diff --git a/source/source_relax/unittests/test_relax_nsync.cpp b/source/source_relax/unittests/test_relax_nsync.cpp new file mode 100644 index 00000000000..fffd6fdd16f --- /dev/null +++ b/source/source_relax/unittests/test_relax_nsync.cpp @@ -0,0 +1,503 @@ +#include +#include +#include +#include + +#include "gmock/gmock.h" +#include "gtest/gtest.h" +#include "source_base/global_variable.h" +#include "source_cell/unitcell.h" +#include "source_io/module_parameter/parameter.h" +#include "source_relax/lattice_change_basic.h" +#include "source_relax/relax_nsync.h" + +/************************************************ + * unit tests of class IonCellOptimizer (relax_nsync.cpp) + ***********************************************/ + +/** + * The running-log convergence reporting is centralized in + * IonCellOptimizer::relax_step. These tests lock the unified format shared by + * every relax method (sd/cg/bfgs/bfgs_trad/lbfgs) and by cell-relax: + * Largest force is ... eV/Angstrom while threshold is ... eV/Angstrom + * Largest stress is ... kbar while threshold is ... kbar + * Relaxation is converged! + * Relaxation is not converged yet! + * Relaxation is not converged after reaching relax_nmax! + * The markers are what the ASE interface greps for (issue #6546). + */ + +namespace unitcell +{ +// Stub: the real update_pos_tau performs lattice-vector algebra that needs a +// fully built UnitCell; these tests only care about the log output. +void update_pos_tau(const Lattice&, const double*, const int, const int, Atom*) +{ +} +// Stub: the real setup_cell_after_vc recomputes reciprocal lattice etc. +void setup_cell_after_vc(UnitCell&, std::ofstream&, const int) +{ +} +} // namespace unitcell + +UnitCell::UnitCell() +{ + Coordinate = "Direct"; + latName = "none"; + lat0 = 1.0; + latvec.Identity(); + + ntype = 1; + nat = 1; + iat2it = nullptr; + iat2ia = nullptr; + iwt2iat = nullptr; + iwt2iw = nullptr; + itia2iat.create(1, 1); + + atoms = new Atom[ntype]; + set_atom_flag = true; + atoms[0].label = "Si"; +} +UnitCell::~UnitCell() +{ +} +Magnetism::Magnetism() +{ +} +Magnetism::~Magnetism() +{ +} +Atom::Atom() +{ + na = 1; + tau.resize(na); + dis.resize(na); + mag.resize(na); + mbl.resize(na); + vel.resize(na); + taud.resize(na); + mbl[0] = {1, 1, 1}; +} +Atom::~Atom() +{ +} +#ifdef __MPI +void Atom::bcast_atom() +{ +} +void Atom::bcast_atom2() +{ +} +#endif +Atom_pseudo::Atom_pseudo() +{ +} +Atom_pseudo::~Atom_pseudo() +{ +} +pseudo::pseudo() +{ +} +pseudo::~pseudo() +{ +} +SepPot::SepPot() +{ +} +SepPot::~SepPot() +{ +} +Sep_Cell::Sep_Cell() noexcept +{ +} +Sep_Cell::~Sep_Cell() noexcept +{ +} +int ModuleSymmetry::Symmetry::symm_flag = 0; +void ModuleSymmetry::Symmetry::symmetrize_mat3(ModuleBase::matrix& sigma, const Lattice& lat) const +{ +} +void ModuleSymmetry::Symmetry::symmetrize_vec3_nat(double* v) const +{ +} + +// Friend of Parameter; the only sanctioned way to write PARAM.input in tests. +class TestParameters +{ + public: + static void set_relax(const std::string& out_level, + const double relax_bfgs_rmax, + const double relax_bfgs_rmin, + const double relax_bfgs_init, + const std::string& fixed_axes) + { + PARAM.input.out_level = out_level; + PARAM.input.relax_bfgs_rmax = relax_bfgs_rmax; + PARAM.input.relax_bfgs_rmin = relax_bfgs_rmin; + PARAM.input.relax_bfgs_init = relax_bfgs_init; + PARAM.input.fixed_axes = fixed_axes; + } +}; + +class IonCellOptimizerTest : public ::testing::Test +{ + protected: + IonCellOptimizer optimizer; + Input_para inp; + const int natom = 1; + const double etot = 0.0; + const std::string log_file = "relax_nsync_test.log"; + const std::string warning_file = "relax_nsync_test_warning.log"; + + void SetUp() override + { + // keep stdout quiet (out_level "m") and give bfgs sane trust radii + TestParameters::set_relax("m", 0.2, 1.0e-5, 0.5, "None"); + + inp.calculation = "relax"; + inp.relax_method = {"lbfgs", "1"}; + inp.relax_nmax = 50; + inp.force_thr = 0.1; // Ry/Bohr + inp.force_thr_ev = 0.1 * 13.6058 / 0.529177; // eV/Angstrom + inp.stress_thr = 0.5; // kbar + inp.cal_force = 1; + inp.cal_stress = 1; + inp.out_level = "m"; + inp.test_relax_method = 0; + inp.fixed_ibrav = false; + inp.nspin = 1; + } + + void TearDown() override + { + std::remove(log_file.c_str()); + std::remove(warning_file.c_str()); + } + + // UnitCell is non-copyable (unique_ptr member), so fill it in place. + void setup_ucell(UnitCell& ucell) + { + ucell.lat_axis_free[0] = 1; + ucell.lat_axis_free[1] = 1; + ucell.lat_axis_free[2] = 1; + // The mock UnitCell leaves iat2it/iat2ia null; the L-BFGS update loop + // dereferences them from the second iteration on, so give the single + // atom a valid type/index. Ownership passes to UnitCell's internal + // Statistics member, whose destructor releases them -- do NOT delete + // them here. + ucell.iat2it = new int[natom]; + ucell.iat2ia = new int[natom]; + ucell.iat2it[0] = 0; + ucell.iat2ia[0] = 0; + } + + std::string read_log() + { + std::ifstream ifs(log_file); + std::string content((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + ifs.close(); + return content; + } + + std::string read_warning_log() + { + std::ifstream ifs(warning_file); + std::string content((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + ifs.close(); + return content; + } +}; + +// relax + small force: the log must carry the force line and the ASE marker. +TEST_F(IonCellOptimizerTest, RelaxConverged) +{ + UnitCell ucell; + setup_ucell(ucell); + optimizer.init_relax(natom, inp); + ModuleBase::matrix force(natom, 3); + force(0, 0) = 1.0e-4; // well below force_thr + ModuleBase::matrix stress(3, 3); + int force_step = 1; + int stress_step = 1; + std::ofstream ofs(log_file); + + const bool done = optimizer.relax_step(1, etot, ucell, force, stress, force_step, stress_step, ofs); + ofs.close(); + + EXPECT_TRUE(done); + const std::string log = read_log(); + EXPECT_THAT(log, testing::HasSubstr(" Largest force is ")); + EXPECT_THAT(log, testing::HasSubstr(" eV/Angstrom while threshold is ")); + EXPECT_THAT(log, testing::HasSubstr(" Relaxation method: lbfgs")); + EXPECT_THAT(log, testing::HasSubstr(" Relaxation converged in 1 step(s).")); + EXPECT_THAT(log, testing::HasSubstr(" Largest force per step (eV/Angstrom):")); + EXPECT_THAT(log, testing::HasSubstr("\n Relaxation is converged!")); + EXPECT_THAT(log, testing::Not(testing::HasSubstr("not converged"))); +} + +// relax + large force: not converged, and no ASE converged marker may appear. +TEST_F(IonCellOptimizerTest, RelaxNotConverged) +{ + UnitCell ucell; + setup_ucell(ucell); + optimizer.init_relax(natom, inp); + ModuleBase::matrix force(natom, 3); + force(0, 0) = 1.0; // above force_thr + ModuleBase::matrix stress(3, 3); + int force_step = 1; + int stress_step = 1; + std::ofstream ofs(log_file); + + const bool done = optimizer.relax_step(1, etot, ucell, force, stress, force_step, stress_step, ofs); + ofs.close(); + + EXPECT_FALSE(done); + const std::string log = read_log(); + EXPECT_THAT(log, testing::HasSubstr(" Largest force is ")); + EXPECT_THAT(log, testing::HasSubstr("\n Relaxation is not converged yet!")); + EXPECT_THAT(log, testing::Not(testing::HasSubstr("Relaxation is converged!"))); +} + +// Reaching relax_nmax must terminate with an explicit not-converged marker. +TEST_F(IonCellOptimizerTest, RelaxNmaxReached) +{ + UnitCell ucell; + setup_ucell(ucell); + optimizer.init_relax(natom, inp); + ModuleBase::matrix force(natom, 3); + ModuleBase::matrix stress(3, 3); + int force_step = 1; + int stress_step = 1; + std::ofstream ofs(log_file); + + const bool done = optimizer.relax_step(inp.relax_nmax, etot, ucell, force, stress, force_step, stress_step, ofs); + ofs.close(); + + EXPECT_TRUE(done); + // Key invariant: when relax_nmax terminates the loop, cal_movement is + // skipped and the geometry is NOT touched. The force/stress captured by + // the driver before relax_step are still consistent with ucell, so the + // driver's geometry_evaluated=true is honest. + EXPECT_FALSE(ucell.ionic_position_updated); + const std::string log = read_log(); + EXPECT_THAT(log, testing::HasSubstr(" Relaxation method: lbfgs")); + EXPECT_THAT(log, testing::HasSubstr(" ionic step(s) (relax_nmax = 50 reached).")); + EXPECT_THAT(log, testing::HasSubstr(" Relaxation is not converged after reaching relax_nmax!")); + EXPECT_THAT(log, testing::Not(testing::HasSubstr("Relaxation is converged!"))); +} + +// cell-relax + small force + small stress: both force and stress lines are +// reported and the converged marker is printed once the cell step converges. +TEST_F(IonCellOptimizerTest, CellRelaxConverged) +{ + inp.calculation = "cell-relax"; + UnitCell ucell; + setup_ucell(ucell); + optimizer.init_relax(natom, inp); + ModuleBase::matrix force(natom, 3); + force(0, 0) = 1.0e-4; + ModuleBase::matrix stress(3, 3); // zero stress, below stress_thr + int force_step = 1; + int stress_step = 1; + std::ofstream ofs(log_file); + + const bool done = optimizer.relax_step(1, etot, ucell, force, stress, force_step, stress_step, ofs); + ofs.close(); + + EXPECT_TRUE(done); + const std::string log = read_log(); + EXPECT_THAT(log, testing::HasSubstr(" Largest force is ")); + EXPECT_THAT(log, testing::HasSubstr(" Largest stress is ")); + EXPECT_THAT(log, testing::HasSubstr(" kbar while threshold is ")); + EXPECT_THAT(log, testing::HasSubstr(" Relaxation method: lbfgs")); + EXPECT_THAT(log, testing::HasSubstr(" Relaxation converged in 1 step(s).")); + EXPECT_THAT(log, testing::HasSubstr(" Largest force per step (eV/Angstrom):")); + EXPECT_THAT(log, testing::HasSubstr(" Largest stress per step (kbar):")); + EXPECT_THAT(log, testing::HasSubstr("\n Relaxation is converged!")); +} + +// cell-relax + small force + large stress: the ionic part converged but the +// cell part did not, so the log shows not-converged and no ASE marker. +TEST_F(IonCellOptimizerTest, CellRelaxNotConverged) +{ + inp.calculation = "cell-relax"; + UnitCell ucell; + setup_ucell(ucell); + optimizer.init_relax(natom, inp); + ModuleBase::matrix force(natom, 3); + force(0, 0) = 1.0e-4; + ModuleBase::matrix stress(3, 3); + stress(0, 0) = 1.0; // ~2.9e5 kbar, above stress_thr + int force_step = 1; + int stress_step = 1; + std::ofstream ofs(log_file); + + const bool done = optimizer.relax_step(1, etot, ucell, force, stress, force_step, stress_step, ofs); + ofs.close(); + + EXPECT_FALSE(done); + const std::string log = read_log(); + EXPECT_THAT(log, testing::HasSubstr(" Largest stress is ")); + EXPECT_THAT(log, testing::HasSubstr("\n Relaxation is not converged yet!")); + EXPECT_THAT(log, testing::Not(testing::HasSubstr("Relaxation is converged!"))); +} + +// Decision: the force history survives a cell change (reset_after_cell_change +// resets force_step and the ionic optimizer, but max_force_history_ must keep +// every ionic step from the whole run). After one unconverged cell step, a +// second ionic step must append, not restart, the history. +TEST_F(IonCellOptimizerTest, CellRelaxForceHistoryNotClearedAcrossCellChange) +{ + inp.calculation = "cell-relax"; + UnitCell ucell; + setup_ucell(ucell); + optimizer.init_relax(natom, inp); + + std::ofstream ofs(log_file); + int force_step = 1; + int stress_step = 1; + + // Step 1: force converged but stress not -> triggers cell change, which + // resets force_step to 1 and calls IMM.reset_after_cell_change. + ModuleBase::matrix force(natom, 3); + force(0, 0) = 1.0e-4; + ModuleBase::matrix stress(3, 3); + stress(0, 0) = 1.0; // ~2.9e5 kbar, above stress_thr -> cell change + const bool done1 = optimizer.relax_step(1, etot, ucell, force, stress, force_step, stress_step, ofs); + EXPECT_FALSE(done1); + EXPECT_EQ(force_step, 1); // reset by the cell change + ASSERT_EQ(optimizer.get_max_force_history().size(), 1u); + + // Step 2: another ionic step after the reset. The history must accumulate. + const bool done2 = optimizer.relax_step(2, etot, ucell, force, stress, force_step, stress_step, ofs); + ofs.close(); + EXPECT_FALSE(done2); + EXPECT_EQ(optimizer.get_max_force_history().size(), 2u); +} + +// Regression test for the review on PR #8067: relax_nmax = 0 is a valid +// dry-run mode. The not-converged branch must stay silent so the running log +// never contains "Relaxation is not converged after reaching relax_nmax!". +TEST_F(IonCellOptimizerTest, DryRunRelaxNmaxZero) +{ + inp.calculation = "relax"; + inp.relax_nmax = 0; + UnitCell ucell; + setup_ucell(ucell); + optimizer.init_relax(natom, inp); + ModuleBase::matrix force(natom, 3); + force(0, 0) = 1.0; // would be above threshold, but no step should run + ModuleBase::matrix stress(3, 3); + int force_step = 1; + int stress_step = 1; + std::ofstream ofs(log_file); + + // istep == relax_nmax == 0 hits the former not-converged branch. + const bool done = optimizer.relax_step(0, etot, ucell, force, stress, force_step, stress_step, ofs); + ofs.close(); + + EXPECT_TRUE(done); + const std::string log = read_log(); + EXPECT_THAT(log, testing::Not(testing::HasSubstr("not converged"))); + EXPECT_THAT(log, testing::Not(testing::HasSubstr("relax_nmax = 0 reached"))); +} + +// cell-relax + fixed_axes = abc: the lattice is fully fixed but the atoms can +// still move. Once the forces converge there is no cell step to run, and the +// log must still carry the unified converged summary (PR #8067 review). +TEST_F(IonCellOptimizerTest, CellRelaxFixedLatticeConverged) +{ + inp.calculation = "cell-relax"; + UnitCell ucell; + setup_ucell(ucell); + // fixed_axes = abc: no lattice vector may change. + ucell.lat_axis_free[0] = 0; + ucell.lat_axis_free[1] = 0; + ucell.lat_axis_free[2] = 0; + optimizer.init_relax(natom, inp); + ModuleBase::matrix force(natom, 3); + force(0, 0) = 1.0e-4; // below force_thr + ModuleBase::matrix stress(3, 3); + int force_step = 1; + int stress_step = 1; + std::ofstream ofs(log_file); + + const bool done = optimizer.relax_step(1, etot, ucell, force, stress, force_step, stress_step, ofs); + ofs.close(); + + EXPECT_TRUE(done); + const std::string log = read_log(); + EXPECT_THAT(log, testing::HasSubstr(" Largest force is ")); + EXPECT_THAT(log, testing::HasSubstr(" Relaxation method: lbfgs")); + EXPECT_THAT(log, testing::HasSubstr(" Relaxation converged in 1 step(s).")); + EXPECT_THAT(log, testing::HasSubstr(" Largest force per step (eV/Angstrom):")); + EXPECT_THAT(log, testing::HasSubstr("\n Relaxation is converged!")); + // No cell step ran, so no stress history line and no not-converged marker. + EXPECT_THAT(log, testing::Not(testing::HasSubstr(" Largest stress per step (kbar):"))); + EXPECT_THAT(log, testing::Not(testing::HasSubstr("not converged"))); +} + +// Regression test for the no-atom-can-move branch of relax_step: with every +// atom fixed (mbl = 0), relax mode is a valid no-op relaxation. The running +// log must still carry the unified converged summary so that downstream tools +// (e.g. the ASE interface) find an explicit success marker. +TEST_F(IonCellOptimizerTest, RelaxAllAtomsFixedConverged) +{ + inp.calculation = "relax"; + UnitCell ucell; + setup_ucell(ucell); + // Fix the only atom in all directions: no atoms are allowed to move. + ucell.atoms[0].mbl[0] = {0, 0, 0}; + optimizer.init_relax(natom, inp); + ModuleBase::matrix force(natom, 3); + ModuleBase::matrix stress(3, 3); + int force_step = 1; + int stress_step = 1; + std::ofstream ofs(log_file); + + // Redirect warnings to a file so the no-op warning can be inspected. + GlobalV::ofs_warning.open(warning_file); + const bool done = optimizer.relax_step(1, etot, ucell, force, stress, force_step, stress_step, ofs); + GlobalV::ofs_warning.close(); + ofs.close(); + + EXPECT_TRUE(done); + const std::string log = read_log(); + EXPECT_THAT(log, testing::HasSubstr(" Relaxation method: lbfgs")); + EXPECT_THAT(log, testing::HasSubstr(" Relaxation converged in 1 step(s).")); + EXPECT_THAT(log, testing::HasSubstr("\n Relaxation is converged!")); + // No ionic step ran, so no force history and no not-converged marker. + EXPECT_THAT(log, testing::Not(testing::HasSubstr(" Largest force per step (eV/Angstrom):"))); + EXPECT_THAT(log, testing::Not(testing::HasSubstr("not converged"))); + // The no-op branch must warn that no atoms are allowed to move. + const std::string warning_log = read_warning_log(); + EXPECT_THAT(warning_log, testing::HasSubstr("No atoms are allowed to move!")); +} + +// Same fixed-lattice setup, but the forces are above threshold: relaxation is +// genuinely in progress, so the log must not print the converged marker. +TEST_F(IonCellOptimizerTest, CellRelaxFixedLatticeNotConverged) +{ + inp.calculation = "cell-relax"; + UnitCell ucell; + setup_ucell(ucell); + ucell.lat_axis_free[0] = 0; + ucell.lat_axis_free[1] = 0; + ucell.lat_axis_free[2] = 0; + optimizer.init_relax(natom, inp); + ModuleBase::matrix force(natom, 3); + force(0, 0) = 1.0; // above force_thr + ModuleBase::matrix stress(3, 3); + int force_step = 1; + int stress_step = 1; + std::ofstream ofs(log_file); + + const bool done = optimizer.relax_step(1, etot, ucell, force, stress, force_step, stress_step, ofs); + ofs.close(); + + EXPECT_FALSE(done); + const std::string log = read_log(); + EXPECT_THAT(log, testing::HasSubstr("\n Relaxation is not converged yet!")); + EXPECT_THAT(log, testing::Not(testing::HasSubstr("Relaxation is converged!"))); +} diff --git a/source/source_relax/unittests/test_relax_stru_io.cpp b/source/source_relax/unittests/test_relax_stru_io.cpp new file mode 100644 index 00000000000..1d339a23c4c --- /dev/null +++ b/source/source_relax/unittests/test_relax_stru_io.cpp @@ -0,0 +1,258 @@ +#include "../relax_stru_io.h" + +#include "gmock/gmock.h" +#include "gtest/gtest.h" + +#include +#include +#include + +using testing::HasSubstr; +using testing::Not; + +namespace +{ + +ModuleBase::matrix identity_stress() +{ + ModuleBase::matrix stress(3, 3); + for (int i = 0; i < 3; i++) + { + for (int j = 0; j < 3; j++) + { + stress(i, j) = (i == j) ? 1.0 : 0.0; + } + } + return stress; +} + +Input_para fully_evaluated_inp() +{ + Input_para inp; + inp.cal_force = true; + inp.cal_stress = true; + return inp; +} + +} // namespace + +TEST(RelaxStruIO, HeaderContainsVersionAndStep) +{ + const Input_para inp = fully_evaluated_inp(); + const std::string header = relax_stru_io::build_stru_header(0, 1.0, identity_stress(), inp, false, true); + EXPECT_THAT(header, HasSubstr("# ABACUS version:")); + EXPECT_THAT(header, HasSubstr("# RELAX STEP 1,")); + EXPECT_THAT(header, Not(HasSubstr("(FINAL)"))); +} + +TEST(RelaxStruIO, HeaderFinalStepLabel) +{ + const Input_para inp = fully_evaluated_inp(); + const std::string header = relax_stru_io::build_stru_header(4, 1.0, identity_stress(), inp, true, true); + EXPECT_THAT(header, HasSubstr("# RELAX STEP 5 (FINAL),")); +} + +TEST(RelaxStruIO, HeaderConvertsEnergyToEv) +{ + // etot in Ry, printed in eV via ModuleBase::Ry_to_eV + const Input_para inp = fully_evaluated_inp(); + const std::string header = relax_stru_io::build_stru_header(0, 2.0, identity_stress(), inp, false, true); + char expected[64]; + std::snprintf(expected, sizeof(expected), "%.8f eV", 2.0 * ModuleBase::Ry_to_eV); + EXPECT_THAT(header, HasSubstr(expected)); +} + +TEST(RelaxStruIO, HeaderWritesThreeStressRowsInKbar) +{ + ModuleBase::matrix stress(3, 3); + for (int i = 0; i < 3; i++) + { + for (int j = 0; j < 3; j++) + { + stress(i, j) = 0.0; + } + } + stress(0, 0) = 1.0; + stress(1, 1) = 2.0; + stress(2, 2) = 3.0; + + const Input_para inp = fully_evaluated_inp(); + const std::string header = relax_stru_io::build_stru_header(0, 1.0, stress, inp, false, true); + // three "# Stress (kbar):" lines + int count = 0; + size_t pos = 0; + while ((pos = header.find("# Stress (kbar):", pos)) != std::string::npos) + { + count++; + pos += 1; + } + EXPECT_EQ(count, 3); +} + +// cal_force / cal_stress header combinations ------------------------------- + +// Fixture providing a populated stress matrix and the helpers used by the +// header behavior tests. These tests check only build_stru_header output; +// driver-loop control flow is covered by the relax sync/nsync suites. +class HeaderBehaviorTest : public testing::Test +{ +protected: + Input_para inp; + ModuleBase::matrix stress; + const double etot = -10.5; // Ry + + void SetUp() override + { + stress.create(3, 3); + stress(0, 0) = 1.0; + stress(0, 1) = 0.1; + stress(0, 2) = 0.2; + stress(1, 0) = 0.1; + stress(1, 1) = 1.0; + stress(1, 2) = 0.3; + stress(2, 0) = 0.2; + stress(2, 1) = 0.3; + stress(2, 2) = 1.0; + } + + static std::vector split_lines(const std::string& s) + { + std::vector lines; + std::istringstream iss(s); + std::string line; + while (std::getline(iss, line)) + { + lines.push_back(line); + } + return lines; + } + + static int count_substr(const std::string& s, const std::string& sub) + { + int count = 0; + size_t pos = 0; + while ((pos = s.find(sub, pos)) != std::string::npos) + { + ++count; + pos += 1; + } + return count; + } +}; + +TEST_F(HeaderBehaviorTest, CalForceStressTrue) +{ + inp.cal_force = true; + inp.cal_stress = true; + + const std::string header = relax_stru_io::build_stru_header(0, etot, stress, inp, false, true); + const auto lines = split_lines(header); + + ASSERT_EQ(lines.size(), 7u); + EXPECT_THAT(lines[0], testing::HasSubstr("# ABACUS version:")); + EXPECT_THAT(lines[1], testing::HasSubstr("# Written at")); + EXPECT_THAT(lines[2], testing::HasSubstr("# RELAX STEP 1, Energy:")); + EXPECT_THAT(lines[3], testing::HasSubstr("# Stress (kbar): ")); + EXPECT_THAT(lines[4], testing::HasSubstr("# Stress (kbar): ")); + EXPECT_THAT(lines[5], testing::HasSubstr("# Stress (kbar): ")); + EXPECT_THAT(lines[6], testing::HasSubstr("# NOTE: stress and forces computed for this geometry")); + EXPECT_THAT(header, testing::Not(testing::HasSubstr("N/A"))); +} + +TEST_F(HeaderBehaviorTest, CalStressFalse) +{ + inp.cal_force = true; + inp.cal_stress = false; + + const std::string header = relax_stru_io::build_stru_header(0, etot, stress, inp, false, true); + const auto lines = split_lines(header); + + ASSERT_EQ(lines.size(), 7u); + // Stress lines are pure N/A, no inline reason comment + EXPECT_EQ(count_substr(header, "# Stress (kbar): N/A N/A N/A"), 3); + EXPECT_THAT(lines[3], testing::Not(testing::HasSubstr("cal_stress"))); + EXPECT_THAT(lines[6], testing::HasSubstr("# NOTE: stress not computed (cal_stress=0)")); +} + +TEST_F(HeaderBehaviorTest, CalForceFalse) +{ + inp.cal_force = false; + inp.cal_stress = true; + + const std::string header = relax_stru_io::build_stru_header(0, etot, stress, inp, false, true); + const auto lines = split_lines(header); + + ASSERT_EQ(lines.size(), 7u); + EXPECT_THAT(header, testing::Not(testing::HasSubstr("N/A"))); + EXPECT_THAT(lines[6], testing::HasSubstr("# NOTE: forces not computed (cal_force=0); per-atom f fields omitted intentionally")); +} + +TEST_F(HeaderBehaviorTest, CalForceStressBothFalse) +{ + inp.cal_force = false; + inp.cal_stress = false; + + const std::string header = relax_stru_io::build_stru_header(0, etot, stress, inp, false, true); + const auto lines = split_lines(header); + + ASSERT_EQ(lines.size(), 7u); + EXPECT_EQ(count_substr(header, "# Stress (kbar): N/A N/A N/A"), 3); + EXPECT_THAT(lines[6], testing::HasSubstr("# NOTE: stress not computed (cal_stress=0); forces not computed (cal_force=0)")); +} + +TEST_F(HeaderBehaviorTest, FinalStep) +{ + inp.cal_force = true; + inp.cal_stress = true; + + const std::string header = relax_stru_io::build_stru_header(4, etot, stress, inp, true, true); + const auto lines = split_lines(header); + + ASSERT_EQ(lines.size(), 7u); + EXPECT_THAT(lines[2], testing::HasSubstr("# RELAX STEP 5 (FINAL), Energy:")); +} + +TEST_F(HeaderBehaviorTest, GeometryNotEvaluated) +{ + inp.cal_force = true; + inp.cal_stress = true; + + const std::string header = relax_stru_io::build_stru_header(4, etot, stress, inp, true, false); + const auto lines = split_lines(header); + + ASSERT_EQ(lines.size(), 7u); + EXPECT_THAT(lines[2], testing::HasSubstr("Energy: N/A")); + EXPECT_EQ(count_substr(header, "# Stress (kbar): N/A N/A N/A"), 3); + EXPECT_THAT(lines[6], testing::HasSubstr("# NOTE: geometry proposed by optimizer but not evaluated")); + EXPECT_THAT(lines[6], testing::HasSubstr("energy N/A")); + EXPECT_THAT(lines[6], testing::HasSubstr("stress N/A")); + EXPECT_THAT(lines[6], testing::HasSubstr("forces omitted")); + EXPECT_THAT(lines[6], testing::Not(testing::HasSubstr("energy above belongs to the last evaluated geometry"))); +} + +// need_orbital ------------------------------------------------------------ + +TEST(RelaxStruIO, NeedOrbitalForLcao) +{ + Input_para inp; + inp.basis_type = "lcao"; + EXPECT_TRUE(relax_stru_io::need_orbital(inp)); +} + +TEST(RelaxStruIO, NeedOrbitalForLcaoInPw) +{ + Input_para inp; + inp.basis_type = "lcao_in_pw"; + EXPECT_TRUE(relax_stru_io::need_orbital(inp)); +} + +TEST(RelaxStruIO, NeedOrbitalForPwOnlyWithNaoWfc) +{ + Input_para inp; + inp.basis_type = "pw"; + inp.init_wfc = "atomic"; + EXPECT_FALSE(relax_stru_io::need_orbital(inp)); + + inp.init_wfc = "nao"; + EXPECT_TRUE(relax_stru_io::need_orbital(inp)); +} diff --git a/source/source_relax/unittests/test_relax_sync.cpp b/source/source_relax/unittests/test_relax_sync.cpp new file mode 100644 index 00000000000..ce176d42345 --- /dev/null +++ b/source/source_relax/unittests/test_relax_sync.cpp @@ -0,0 +1,893 @@ +#include "gtest/gtest.h" +#include "gmock/gmock.h" +#include +#include "../relax_sync.h" +#include "source_cell/unitcell.h" +#include "source_io/module_parameter/input_parameter.h" +#include "relax_test.h" +#include + +// Fixed mock SCF outputs (forces, stresses, energies) feeding relax_step, +// previously read from support/*.txt. Inlining them keeps the test +// independent of the working directory. Values are the first 3 steps of the +// original files, so result_ref is unchanged. +namespace { +constexpr int kRelaxSteps = 3; +constexpr int kRelaxNat = 5; +constexpr int kRelaxNforce = kRelaxSteps * kRelaxNat * 3; // 45 +constexpr int kRelaxNstress = kRelaxSteps * 3 * 3; // 27 + +const double kRelaxForces[kRelaxNforce] = { + 4.56e-08, -8.74e-08, 4.0671950e-03, + -5.59e-08, 1.479e-07, 1.81171376e-02, + 2.942e-07, -5.56e-08, -1.7979905e-03, + -1.691e-07, 1.612e-07, -1.7977699e-03, + -1.149e-07, -1.660e-07, -1.85885722e-02, + 6.73e-08, 5.90e-08, 4.8608620e-03, + -1.632e-07, -1.833e-07, -2.24081615e-02, + 4.93e-08, 1.246e-07, -1.8938742e-03, + 9.84e-08, 4.34e-08, -1.8939930e-03, + -5.19e-08, -4.38e-08, 2.13351667e-02, + 8.25e-08, 2.095e-07, 4.5297162e-03, + -5.19e-08, 3.57e-08, -3.6659292e-03, + 4.95e-08, -1.654e-07, -1.8863808e-03, + 4.14e-08, 2.440e-07, -1.8870376e-03, + -1.215e-07, -3.239e-07, 2.9096315e-03, +}; + +const double kRelaxStresses[kRelaxNstress] = { + 4.79987e-05, -2.0e-10, 6.0e-10, + -2.0e-10, 4.80038e-05, -4.0e-10, + 6.0e-10, -4.0e-10, -2.751687e-04, + 1.040725e-04, 2.0e-10, -1.6e-09, + 2.0e-10, 1.040682e-04, -7.0e-10, + -1.6e-09, -7.0e-10, 1.982021e-04, + 8.00586e-05, 1.0e-10, -1.0e-09, + 1.0e-10, 8.00526e-05, 2.8e-09, + -1.0e-09, 2.8e-09, -6.4010e-06, +}; + +const double kRelaxEnergies[kRelaxSteps] = { + 4.79987e-05, + -2.0e-10, + 6.0e-10, +}; +} // namespace + +class Test_SETGRAD : public testing::Test +{ + protected: + Relax rl; + std::vector result; + Input_para inp; + UnitCell ucell; + std::ofstream ofs; + + void SetUp() + { + inp.force_thr = 0.001; + inp.calculation = "cell-relax"; + + ModuleBase::matrix force_in, stress_in; + int nat = 3; + + force_in.create(nat,3); + stress_in.create(3,3); + + force_in(0,0) = 0.1; force_in(0,1) = 0.1; force_in(0,2)= 0.1; + force_in(1,0) = 0; force_in(1,1) = 0.1; force_in(1,2)= 0.1; + force_in(2,0) = 0; force_in(2,1) = 0; force_in(2,2)= 0.1; + + stress_in(0,0) = 1; stress_in(0,1) = 1; stress_in(0,2)= 1; + stress_in(1,0) = 0; stress_in(1,1) = 1; stress_in(1,2)= 1; + stress_in(2,0) = 0; stress_in(2,1) = 0; stress_in(2,2)= 1; + + ucell.ntype = 1; + ucell.nat = nat; + ucell.atoms = new Atom[1]; + ucell.atoms[0].na = nat; + ucell.omega = 1.0; + ucell.lat0 = 1.0; + + ucell.iat2it = new int[nat]; + ucell.iat2ia = new int[nat]; + ucell.atoms[0].mbl.resize(nat); + ucell.atoms[0].taud.resize(nat); + ucell.atoms[0].tau.resize(nat); + ucell.atoms[0].dis.resize(nat); + ucell.atoms[0].mag.resize(nat); + ucell.atoms[0].vel.resize(nat); + + ucell.iat2it[0] = 0; + ucell.iat2it[1] = 0; + ucell.iat2it[2] = 0; + + ucell.iat2ia[0] = 0; + ucell.iat2ia[1] = 1; + ucell.iat2ia[2] = 2; + + ucell.atoms[0].mbl[0].x = 0; + ucell.atoms[0].mbl[0].y = 0; + ucell.atoms[0].mbl[0].z = 1; + + ucell.atoms[0].mbl[1].x = 0; + ucell.atoms[0].mbl[1].y = 1; + ucell.atoms[0].mbl[1].z = 0; + + ucell.atoms[0].mbl[2].x = 1; + ucell.atoms[0].mbl[2].y = 0; + ucell.atoms[0].mbl[2].z = 0; + + ucell.atoms[0].taud[0] = 0.0; + ucell.atoms[0].taud[1] = 0.0; + ucell.atoms[0].taud[2] = 0.0; + + ucell.atoms[0].tau.resize(nat); + + ucell.lat_axis_free[0] = 1; + ucell.lat_axis_free[1] = 1; + ucell.lat_axis_free[2] = 1; + + rl.init_relax(nat, inp); + rl.relax_step(ucell,force_in,stress_in,0.0, ofs); + + for(int i=0;i<3;i++) + { + result.push_back(ucell.atoms[0].taud[i].x); + result.push_back(ucell.atoms[0].taud[i].y); + result.push_back(ucell.atoms[0].taud[i].z); + } + push_result(); + + //reset lattice vector + ucell.latvec.Identity(); + inp.fixed_axes = "shape"; + rl.init_relax(nat, inp); + rl.relax_step(ucell,force_in,stress_in,0.0, ofs); + push_result(); + + //reset lattice vector + ucell.latvec.Identity(); + inp.fixed_axes = "volume"; + rl.init_relax(nat, inp); + rl.relax_step(ucell,force_in,stress_in,0.0, ofs); + push_result(); + + //reset lattice vector + ucell.latvec.Identity(); + inp.fixed_axes = "a"; //anything other than "None" + inp.fixed_ibrav = true; + ucell.latName = "sc"; + ucell.lat_axis_free[0] = 0; + ucell.lat_axis_free[1] = 0; + ucell.lat_axis_free[2] = 0; + rl.init_relax(nat, inp); + rl.relax_step(ucell,force_in,stress_in,0.0, ofs); + + push_result(); + } + + void push_result() + { + result.push_back(ucell.latvec.e11); + result.push_back(ucell.latvec.e12); + result.push_back(ucell.latvec.e13); + result.push_back(ucell.latvec.e21); + result.push_back(ucell.latvec.e22); + result.push_back(ucell.latvec.e23); + result.push_back(ucell.latvec.e31); + result.push_back(ucell.latvec.e32); + result.push_back(ucell.latvec.e33); + } + +}; + +TEST_F(Test_SETGRAD, relax_new) +{ + std::vector result_ref = + { + 0, 0, 0.24293434145, + 0, 0.242934341453, 0, + 0, 0, 0, + //paramter for taud + 1.2267616333,0.2267616333,0.22676163333, + 0, 1.2267616333 ,0.2267616333, + 0, 0, 1.22676163333, + // paramter for fisrt time after relaxation + 1.3677603495, 0, 0, + 0, 1.36776034956, 0, + 0, 0, 1.36776034956, + // paramter for second time after relaxation + 1.3677603495 ,0.3633367476,0.36333674766, + 0, 1.3677603495 ,0.36333674766, + 0, 0, 1.3677603495 , + // paramter for third time after relaxation + 1,0,0,0,1,0,0,0,1 + // paramter for fourth time after relaxation + }; + for(int i=0;i result; + Input_para inp; + UnitCell ucell; + std::ofstream ofs; + + void SetUp() + { + int nstep = 3; + int nat = 5; + inp.calculation = "cell-relax"; + inp.force_thr = 0.001; + inp.fixed_axes = "a"; + inp.stress_thr = 0.01; + inp.fixed_ibrav = false; + + this->setup_cell(); + + ModuleBase::matrix force_in, stress_in; + force_in.create(nat,3); + stress_in.create(3,3); + + rl.init_relax(nat, inp); + + for(int istep=0;istep result_ref= + { + 0.5000000586,0.4999998876,0.009364595811, + 0.9999999281,1.901333279e-07,0.5476035454, + 3.782097706e-07,0.4999999285,0.4770375874, + 0.4999997826,2.072311863e-07,0.477037871, + 0.9999998523,0.9999997866,0.9349574003, + // paramter for taud after first relaxation + 4.006349654,-1.93128788e-07,5.793863639e-07, + -1.93128788e-07,4.006354579,-3.86257576e-07, + 6.757962549e-07,-4.505308366e-07,3.966870038, + // paramter for latvec after first relaxation + 0.5000000566,0.4999998916,0.009177239183, + 0.9999999308,1.832935626e-07,0.5467689737, + 3.647769323e-07,0.4999999311,0.4771204124, + 0.4999997903,1.998879545e-07,0.4771206859, + 0.9999998574,0.9999997943,0.9358136888, + // paramter for taud after second relaxation + 3.999761277,-1.656764727e-07,4.97029418e-07, + -1.656764727e-07,3.999765501,-3.313529453e-07, + 5.797351131e-07,-3.864900754e-07,4.010925071, + // paramter for latvec after second relaxation + 0.500000082,0.4999999574,0.01057784352, + 0.9999999149,1.939640249e-07,0.5455830599, + 3.795967781e-07,0.4999998795,0.4765373919, + 0.4999998037,2.756298268e-07,0.4765374602, + 0.9999998196,0.9999996936,0.9367652445, + // paramter for taud after third relaxation + 4.017733155,-1.420363309e-07,2.637046077e-07, + -1.420364243e-07,4.017735987,3.126225134e-07, + 3.479123171e-07,2.578467568e-07,4.011674933 + // paramter for latvec after third relaxation + }; + for(int i=0;i converged immediately. + ModuleBase::matrix force_in(nat, 3); + ModuleBase::matrix stress_in(3, 3); + force_in(0, 0) = 1.0e-4; + + Relax rl; + rl.init_relax(nat, inp); + + std::ofstream ofs("./running_relax_sync_test.log"); + const bool done = rl.relax_step(ucell, force_in, stress_in, 0.0, ofs); + ofs.close(); + + EXPECT_TRUE(done); + + std::ifstream ifs("./running_relax_sync_test.log"); + const std::string log((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + ifs.close(); + std::remove("./running_relax_sync_test.log"); + + EXPECT_THAT(log, testing::HasSubstr(" Relaxation method: cg")); + EXPECT_THAT(log, testing::HasSubstr(" Relaxation converged in 1 step(s).")); + EXPECT_THAT(log, testing::HasSubstr(" Largest force per step (eV/Angstrom):")); + EXPECT_THAT(log, testing::HasSubstr("\n Relaxation is converged!")); + + delete[] ucell.atoms; +} + +// First step above threshold (not converged), second step below (converged). +// Locks the converged_step = istep+1 accounting and the per-step history. +TEST(RelaxSyncSummary, TwoStepConvergence) +{ + const int nat = 1; + Input_para inp; + inp.calculation = "relax"; + inp.relax_method = {"cg", "2"}; + inp.force_thr = 0.001; // force_thr_eva ~ 0.0257 eV/Angstrom + inp.force_thr_ev = inp.force_thr * 13.6058 / 0.529177; + + UnitCell ucell; + ucell.ntype = 1; + ucell.nat = nat; + ucell.atoms = new Atom[1]; + ucell.atoms[0].na = nat; + ucell.atoms[0].label = "Si"; + ucell.omega = 1.0; + ucell.lat0 = 1.0; + ucell.iat2it = new int[nat]; + ucell.iat2ia = new int[nat]; + ucell.iat2it[0] = 0; + ucell.iat2ia[0] = 0; + ucell.atoms[0].mbl.resize(nat); + ucell.atoms[0].taud.resize(nat); + ucell.atoms[0].tau.resize(nat); + ucell.atoms[0].dis.resize(nat); + ucell.atoms[0].mag.resize(nat); + ucell.atoms[0].vel.resize(nat); + ucell.atoms[0].mbl[0] = {1, 1, 1}; + ucell.atoms[0].taud[0] = {0.0, 0.0, 0.0}; + ucell.latvec.Identity(); + + ModuleBase::matrix stress_in(3, 3); + + Relax rl; + rl.init_relax(nat, inp); + + // Step 1: force above threshold -> not converged. The magnitude must stay + // small enough that the CG step keeps the atom inside the unit cell + // (latvec is the identity), otherwise ABACUS aborts with "Movement of + // atom is larger than the cell length". + ModuleBase::matrix force_big(nat, 3); + force_big(0, 0) = 0.01; // ~0.257 eV/Angstrom, above threshold + std::ofstream ofs("./running_relax_sync_test.log"); + const bool done1 = rl.relax_step(ucell, force_big, stress_in, 0.0, ofs); + EXPECT_FALSE(done1); + + // Step 2: small force -> converged. + ModuleBase::matrix force_small(nat, 3); + force_small(0, 0) = 1.0e-4; + const bool done2 = rl.relax_step(ucell, force_small, stress_in, 0.0, ofs); + ofs.close(); + EXPECT_TRUE(done2); + + std::ifstream ifs("./running_relax_sync_test.log"); + const std::string log((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + ifs.close(); + std::remove("./running_relax_sync_test.log"); + + EXPECT_THAT(log, testing::HasSubstr("\n Relaxation is not converged yet!")); + EXPECT_THAT(log, testing::HasSubstr(" Relaxation converged in 2 step(s).")); + EXPECT_THAT(log, testing::HasSubstr("\n Relaxation is converged!")); + + // Each relax_step pushes exactly one force entry. + ASSERT_EQ(rl.get_max_force_history().size(), 2u); + + delete[] ucell.atoms; +} + +// cell-relax converged: the summary must also carry the stress history line. +TEST(RelaxSyncSummary, CellRelaxConvergedPrintsStressHistory) +{ + const int nat = 1; + Input_para inp; + inp.calculation = "cell-relax"; + inp.relax_method = {"cg", "2"}; + inp.force_thr = 0.001; + inp.force_thr_ev = inp.force_thr * 13.6058 / 0.529177; + inp.stress_thr = 0.5; // kbar + + UnitCell ucell; + ucell.ntype = 1; + ucell.nat = nat; + ucell.atoms = new Atom[1]; + ucell.atoms[0].na = nat; + ucell.atoms[0].label = "Si"; + ucell.omega = 1.0; + ucell.lat0 = 1.0; + ucell.iat2it = new int[nat]; + ucell.iat2ia = new int[nat]; + ucell.iat2it[0] = 0; + ucell.iat2ia[0] = 0; + ucell.atoms[0].mbl.resize(nat); + ucell.atoms[0].taud.resize(nat); + ucell.atoms[0].tau.resize(nat); + ucell.atoms[0].dis.resize(nat); + ucell.atoms[0].mag.resize(nat); + ucell.atoms[0].vel.resize(nat); + ucell.atoms[0].mbl[0] = {1, 1, 1}; + ucell.atoms[0].taud[0] = {0.0, 0.0, 0.0}; + ucell.latvec.Identity(); + ucell.lat_axis_free[0] = 1; + ucell.lat_axis_free[1] = 1; + ucell.lat_axis_free[2] = 1; + + ModuleBase::matrix force_in(nat, 3); + ModuleBase::matrix stress_in(3, 3); + force_in(0, 0) = 1.0e-4; // force converged + // zero stress -> below stress_thr -> cell converged too + + Relax rl; + rl.init_relax(nat, inp); + + std::ofstream ofs("./running_relax_sync_test.log"); + const bool done = rl.relax_step(ucell, force_in, stress_in, 0.0, ofs); + ofs.close(); + EXPECT_TRUE(done); + + std::ifstream ifs("./running_relax_sync_test.log"); + const std::string log((std::istreambuf_iterator(ifs)), std::istreambuf_iterator()); + ifs.close(); + std::remove("./running_relax_sync_test.log"); + + EXPECT_THAT(log, testing::HasSubstr(" Largest stress is ")); + EXPECT_THAT(log, testing::HasSubstr(" Largest stress per step (kbar):")); + EXPECT_THAT(log, testing::HasSubstr("\n Relaxation is converged!")); + + delete[] ucell.atoms; +} + +// --------------------------------------------------------------------------- +// Behavior tests locking move_cell_ions branches before its refactor. +// All drive the public relax_step and assert only externally observable +// UnitCell state (latvec / taud / omega), so the private-method split must +// keep these numbers bit-stable. +// --------------------------------------------------------------------------- + +namespace +{ +// Build a minimal 2-type, 3-atom cell with the given per-atom move flags. +// Atom 0 is type 0 (1 atom), atoms 1-2 are type 1 (2 atoms). taud start at 0. +// UnitCell is non-copyable, so the caller supplies the object to fill. +void make_two_type_cell(UnitCell& ucell, const int mbl_flat[9]) +{ + ucell.ntype = 2; + ucell.nat = 3; + ucell.omega = 8.0; // 2x2x2 cell + ucell.lat0 = 1.0; + ucell.atoms = new Atom[2]; + ucell.atoms[0].na = 1; + ucell.atoms[1].na = 2; + for (int t = 0; t < 2; t++) + { + const int na = ucell.atoms[t].na; + ucell.atoms[t].label = "X"; + ucell.atoms[t].mbl.resize(na); + ucell.atoms[t].taud.resize(na); + ucell.atoms[t].tau.resize(na); + ucell.atoms[t].dis.resize(na); + ucell.atoms[t].mag.resize(na); + ucell.atoms[t].vel.resize(na); + } + // iat: 0 -> (type0, ia0); 1 -> (type1, ia0); 2 -> (type1, ia1) + ucell.iat2it = new int[3]; + ucell.iat2ia = new int[3]; + ucell.iat2it[0] = 0; ucell.iat2it[1] = 1; ucell.iat2it[2] = 1; + ucell.iat2ia[0] = 0; ucell.iat2ia[1] = 0; ucell.iat2ia[2] = 1; + + // mbl flags, laid out per atom + ucell.atoms[0].mbl[0] = {mbl_flat[0], mbl_flat[1], mbl_flat[2]}; + ucell.atoms[1].mbl[0] = {mbl_flat[3], mbl_flat[4], mbl_flat[5]}; + ucell.atoms[1].mbl[1] = {mbl_flat[6], mbl_flat[7], mbl_flat[8]}; + + ucell.atoms[0].taud[0] = {0.0, 0.0, 0.0}; + ucell.atoms[1].taud[0] = {0.0, 0.0, 0.0}; + ucell.atoms[1].taud[1] = {0.0, 0.0, 0.0}; + + ucell.latvec.Identity(); + ucell.latvec.e11 = 2.0; + ucell.latvec.e22 = 2.0; + ucell.latvec.e33 = 2.0; + ucell.lat_axis_free[0] = 1; + ucell.lat_axis_free[1] = 1; + ucell.lat_axis_free[2] = 1; +} + +void free_two_type_cell(UnitCell& ucell) +{ + delete[] ucell.atoms; + // iat2it / iat2ia are owned by the mock; release explicitly. + delete[] ucell.iat2it; + delete[] ucell.iat2ia; + ucell.iat2it = nullptr; + ucell.iat2ia = nullptr; +} +} // namespace + +// fixed_axes="shape": only the hydrostatic part of the stress survives, so the +// lattice scales isotropically (off-diagonals stay zero, diagonal scales equal). +TEST(RelaxSyncCellMove, FixedShapeIsotropicScale) +{ + const int mbl[9] = {1,1,1, 1,1,1, 1,1,1}; + UnitCell ucell; + make_two_type_cell(ucell, mbl); + + Input_para inp; + inp.calculation = "cell-relax"; + inp.force_thr = 0.0; // force never converged -> take the CG move + inp.force_thr_ev = 0.0; + inp.stress_thr = 0.0; // stress never converged either + inp.fixed_axes = "shape"; + inp.fixed_ibrav = false; + + ModuleBase::matrix force_in(3, 3); + force_in(0, 0) = 0.01; // nonzero -> not converged -> performs the move + ModuleBase::matrix stress_in(3, 3); + stress_in(0, 0) = 1.0; stress_in(1, 1) = 2.0; stress_in(2, 2) = 3.0; + stress_in(0, 1) = 0.5; // anisotropic + off-diagonal parts must be dropped + + Relax rl; + rl.init_relax(3, inp); + std::ofstream ofs("./running_relax_sync_shape.log"); + rl.relax_step(ucell, force_in, stress_in, 0.0, ofs); + ofs.close(); + std::remove("./running_relax_sync_shape.log"); + + // Off-diagonal lattice components unchanged (remain zero). + EXPECT_NEAR(ucell.latvec.e12, 0.0, 1e-12); + EXPECT_NEAR(ucell.latvec.e13, 0.0, 1e-12); + EXPECT_NEAR(ucell.latvec.e21, 0.0, 1e-12); + EXPECT_NEAR(ucell.latvec.e23, 0.0, 1e-12); + EXPECT_NEAR(ucell.latvec.e31, 0.0, 1e-12); + EXPECT_NEAR(ucell.latvec.e32, 0.0, 1e-12); + // Isotropic scaling: all three diagonals shifted by the same amount. + EXPECT_NEAR(ucell.latvec.e11 - 2.0, ucell.latvec.e22 - 2.0, 1e-12); + EXPECT_NEAR(ucell.latvec.e22 - 2.0, ucell.latvec.e33 - 2.0, 1e-12); + + free_two_type_cell(ucell); +} + +// --------------------------------------------------------------------------- +// Max-step termination on the Relax (simultaneous) path. +// +// Unlike IonCellOptimizer::relax_step (which checks istep == relax_nmax and +// returns true without moving), Relax::relax_step has no max-step guard. +// When the driver loop hits relax_nmax, the last relax_step still calls +// move_cell_ions, so the geometry IS updated and the force/stress in the +// driver become stale. The driver then sets geometry_evaluated=false. +// +// This test locks that observable behavior: relax_step moves taud and returns +// false (not converged). +// --------------------------------------------------------------------------- +TEST(RelaxSyncMaxStep, GeometryMovedAndNotConverged) +{ + const int nat = 1; + Input_para inp; + inp.calculation = "relax"; + inp.relax_method = {"cg", "2"}; + inp.force_thr = 0.0; // never converged -> always take the move + inp.force_thr_ev = 0.0; + inp.fixed_axes = "None"; + inp.fixed_ibrav = false; + + UnitCell ucell; + ucell.ntype = 1; + ucell.nat = nat; + ucell.atoms = new Atom[1]; + ucell.atoms[0].na = nat; + ucell.atoms[0].label = "Si"; + ucell.omega = 1.0; + ucell.lat0 = 1.0; + ucell.iat2it = new int[nat]; + ucell.iat2ia = new int[nat]; + ucell.iat2it[0] = 0; + ucell.iat2ia[0] = 0; + ucell.atoms[0].mbl.resize(nat); + ucell.atoms[0].taud.resize(nat); + ucell.atoms[0].tau.resize(nat); + ucell.atoms[0].dis.resize(nat); + ucell.atoms[0].mag.resize(nat); + ucell.atoms[0].vel.resize(nat); + ucell.atoms[0].mbl[0] = {1, 1, 1}; + ucell.atoms[0].taud[0] = {0.0, 0.0, 0.0}; + ucell.latvec.Identity(); + + ModuleBase::matrix force_in(nat, 3); + ModuleBase::matrix stress_in(3, 3); + force_in(0, 0) = 0.01; // nonzero -> not converged -> move + + Relax rl; + rl.init_relax(nat, inp); + std::ofstream ofs("./running_relax_sync_maxstep.log"); + const bool done = rl.relax_step(ucell, force_in, stress_in, 0.0, ofs); + ofs.close(); + std::remove("./running_relax_sync_maxstep.log"); + + EXPECT_FALSE(done); + // The geometry was moved: taud is no longer the initial value. + EXPECT_NE(ucell.atoms[0].taud[0].x, 0.0); + EXPECT_TRUE(ucell.ionic_position_updated); + + delete[] ucell.atoms; +} + +// fixed_axes="volume": the cell may change shape but its volume is re-scaled +// back to the original omega after each move (relax_sync.cpp Step 1). +TEST(RelaxSyncCellMove, FixedVolumePreservesOmega) +{ + const int mbl[9] = {1,1,1, 1,1,1, 1,1,1}; + UnitCell ucell; + make_two_type_cell(ucell, mbl); + const double omega0 = ucell.omega; + + Input_para inp; + inp.calculation = "cell-relax"; + inp.force_thr = 0.0; + inp.force_thr_ev = 0.0; + inp.stress_thr = 0.0; + inp.fixed_axes = "volume"; + inp.fixed_ibrav = false; + + ModuleBase::matrix force_in(3, 3); + force_in(0, 0) = 0.01; // nonzero -> not converged + ModuleBase::matrix stress_in(3, 3); + stress_in(0, 0) = 1.0; stress_in(1, 1) = -0.5; stress_in(0, 1) = 0.3; + + Relax rl; + rl.init_relax(3, inp); + std::ofstream ofs("./running_relax_sync_vol.log"); + rl.relax_step(ucell, force_in, stress_in, 0.0, ofs); + ofs.close(); + std::remove("./running_relax_sync_vol.log"); + + const double omega_new = std::abs(ucell.latvec.Det()) * ucell.lat0 * ucell.lat0 * ucell.lat0; + EXPECT_NEAR(omega_new, omega0, 1e-8); + + free_two_type_cell(ucell); +} + +// lat_axis_free = {1,0,0}: only the first lattice row may move; rows 2 and 3 +// must remain exactly at their initial values (relax_sync.cpp 572-589). +TEST(RelaxSyncCellMove, PartialAxisFreeKeepsLockedRows) +{ + const int mbl[9] = {1,1,1, 1,1,1, 1,1,1}; + UnitCell ucell; + make_two_type_cell(ucell, mbl); + ucell.lat_axis_free[0] = 1; + ucell.lat_axis_free[1] = 0; // lock row 2 + ucell.lat_axis_free[2] = 0; // lock row 3 + + Input_para inp; + inp.calculation = "cell-relax"; + inp.force_thr = 0.0; + inp.force_thr_ev = 0.0; + inp.stress_thr = 0.0; + inp.fixed_axes = "None"; + inp.fixed_ibrav = false; + + ModuleBase::matrix force_in(3, 3); + force_in(0, 0) = 0.01; // nonzero -> not converged + ModuleBase::matrix stress_in(3, 3); + stress_in(0, 0) = 1.0; stress_in(1, 1) = 1.0; stress_in(2, 2) = 1.0; + stress_in(0, 1) = 0.2; stress_in(1, 0) = 0.2; + + Relax rl; + rl.init_relax(3, inp); + std::ofstream ofs("./running_relax_sync_axis.log"); + rl.relax_step(ucell, force_in, stress_in, 0.0, ofs); + ofs.close(); + std::remove("./running_relax_sync_axis.log"); + + // Locked rows unchanged. + EXPECT_NEAR(ucell.latvec.e21, 0.0, 1e-12); + EXPECT_NEAR(ucell.latvec.e22, 2.0, 1e-12); + EXPECT_NEAR(ucell.latvec.e23, 0.0, 1e-12); + EXPECT_NEAR(ucell.latvec.e31, 0.0, 1e-12); + EXPECT_NEAR(ucell.latvec.e32, 0.0, 1e-12); + EXPECT_NEAR(ucell.latvec.e33, 2.0, 1e-12); + + free_two_type_cell(ucell); +} + +// Mixed mbl flags across two atom types: each atom's direct-coordinate +// displacement must be zero along the constrained (mbl==0) directions and +// nonzero only along free ones (relax_sync.cpp Step 2&3, iat2it/iat2ia + mbl). +TEST(RelaxSyncCellMove, MixedMblMovesOnlyFreeComponents) +{ + // atom0: free x only; atom1(type1,ia0): free y only; atom2(type1,ia1): free z only + const int mbl[9] = {1,0,0, 0,1,0, 0,0,1}; + UnitCell ucell; + make_two_type_cell(ucell, mbl); + + Input_para inp; + inp.calculation = "relax"; // ions only, keep lattice fixed for clarity + inp.force_thr = 0.0; + inp.force_thr_ev = 0.0; + inp.fixed_axes = "None"; + inp.fixed_ibrav = false; + + // Uniform force on every atom/component; mbl must mask the movement. + ModuleBase::matrix force_in(3, 3); + for (int i = 0; i < 3; i++) + { + for (int j = 0; j < 3; j++) + { + force_in(i, j) = 0.05; + } + } + ModuleBase::matrix stress_in(3, 3); + + Relax rl; + rl.init_relax(3, inp); + std::ofstream ofs("./running_relax_sync_mbl.log"); + rl.relax_step(ucell, force_in, stress_in, 0.0, ofs); + ofs.close(); + std::remove("./running_relax_sync_mbl.log"); + + // atom0: only x moved + EXPECT_NE(ucell.atoms[0].taud[0].x, 0.0); + EXPECT_NEAR(ucell.atoms[0].taud[0].y, 0.0, 1e-12); + EXPECT_NEAR(ucell.atoms[0].taud[0].z, 0.0, 1e-12); + // atom1: only y moved + EXPECT_NEAR(ucell.atoms[1].taud[0].x, 0.0, 1e-12); + EXPECT_NE(ucell.atoms[1].taud[0].y, 0.0); + EXPECT_NEAR(ucell.atoms[1].taud[0].z, 0.0, 1e-12); + // atom2: only z moved + EXPECT_NEAR(ucell.atoms[1].taud[1].x, 0.0, 1e-12); + EXPECT_NEAR(ucell.atoms[1].taud[1].y, 0.0, 1e-12); + EXPECT_NE(ucell.atoms[1].taud[1].z, 0.0); + + free_two_type_cell(ucell); +} \ No newline at end of file diff --git a/source/source_relax/test/test_socket_driver.cpp b/source/source_relax/unittests/test_socket_driver.cpp similarity index 100% rename from source/source_relax/test/test_socket_driver.cpp rename to source/source_relax/unittests/test_socket_driver.cpp diff --git a/source/source_relax/test/test_socket_frame.cpp b/source/source_relax/unittests/test_socket_frame.cpp similarity index 100% rename from source/source_relax/test/test_socket_frame.cpp rename to source/source_relax/unittests/test_socket_frame.cpp diff --git a/source/source_relax/test/test_socket_ipi.cpp b/source/source_relax/unittests/test_socket_ipi.cpp similarity index 100% rename from source/source_relax/test/test_socket_ipi.cpp rename to source/source_relax/unittests/test_socket_ipi.cpp diff --git a/tests/01_PW/021_PW_kspace3/INPUT b/tests/01_PW/021_PW_kspace3/INPUT deleted file mode 100644 index 892a68ed177..00000000000 --- a/tests/01_PW/021_PW_kspace3/INPUT +++ /dev/null @@ -1,16 +0,0 @@ -INPUT_PARAMETERS -suffix autotest -calculation scf -init_wfc random - -symmetry 1 -ecutwfc 5 -scf_thr 1e-08 -scf_nmax 100 -basis_type pw - -smearing_method gaussian -smearing_sigma 0.020000 - -kspacing 0.5 1.0 1.0 -pseudo_dir ../../PP_ORB diff --git a/tests/01_PW/021_PW_kspace3/README b/tests/01_PW/021_PW_kspace3/README deleted file mode 100644 index edeef55dc95..00000000000 --- a/tests/01_PW/021_PW_kspace3/README +++ /dev/null @@ -1 +0,0 @@ -test kspacing with 3 input values diff --git a/tests/01_PW/021_PW_kspace3/STRU b/tests/01_PW/021_PW_kspace3/STRU deleted file mode 100644 index eacf7d675a4..00000000000 --- a/tests/01_PW/021_PW_kspace3/STRU +++ /dev/null @@ -1,28 +0,0 @@ -ATOMIC_SPECIES -Mo 1.00 Mo_ONCV_PBE-1.0.upf -S 1.00 S_ONCV_PBE-1.0.upf - -LATTICE_CONSTANT -1.88972613 - -LATTICE_VECTORS -3.1903157234 0.0 0.0 --1.5951578617 2.7628944626 0.0 -0.0 0.0 14.8790035248 - -ATOMIC_POSITIONS -Cartesian # Cartesian(Unit is LATTICE_CONSTANT) - -Mo -0.0 -2 -0.029027168000 1.784347725000 3.608837876000 0 0 0 -1.660723717000 1.002844149000 11.140101332000 0 0 0 -S -0.0 -4 -1.561446854000 1.061209471000 5.316422735000 0 0 0 -0.082013669000 1.868596936000 12.638587707000 0 0 0 -1.592155903000 0.839435772000 2.071206131000 0 0 0 --0.012459422000 1.814452282000 9.540172847000 0 0 0 - diff --git a/tests/01_PW/021_PW_kspace3/result.ref b/tests/01_PW/021_PW_kspace3/result.ref deleted file mode 100644 index a81d8ad1a3d..00000000000 --- a/tests/01_PW/021_PW_kspace3/result.ref +++ /dev/null @@ -1,6 +0,0 @@ -etotref -4337.996541332115 -etotperatomref -722.9994235554 -pointgroupref C_1 -spacegroupref C_1 -nksibzref 4 -totaltimeref 1.03 diff --git a/tests/01_PW/031_PW_15_CF_CS/INPUT b/tests/01_PW/031_PW_15_CF_CS/INPUT deleted file mode 100644 index 470e9aa7374..00000000000 --- a/tests/01_PW/031_PW_15_CF_CS/INPUT +++ /dev/null @@ -1,33 +0,0 @@ -INPUT_PARAMETERS -#Parameters (General) -suffix autotest -pseudo_dir ../../PP_ORB -gamma_only 0 -calculation scf -symmetry 1 -relax_nmax 1 -out_level ie -smearing_method gaussian -smearing_sigma 0.02 - -#Parameters (3.PW) -ecutwfc 5 -scf_thr 1e-9 -scf_nmax 100 - -#Parameters (LCAO) -basis_type pw -ks_solver cg -chg_extrap second-order -pw_diag_thr 0.00001 - -cal_force 1 -#test_force 1 -cal_stress 1 -#test_stress 1 - -mixing_type broyden -mixing_beta 0.4 -mixing_gg0 1.5 - -pw_seed 1 diff --git a/tests/01_PW/031_PW_15_CF_CS/KPT b/tests/01_PW/031_PW_15_CF_CS/KPT deleted file mode 100644 index 1776ebb617b..00000000000 --- a/tests/01_PW/031_PW_15_CF_CS/KPT +++ /dev/null @@ -1,4 +0,0 @@ -K_POINTS -0 -Gamma -2 1 1 0 0 0 diff --git a/tests/01_PW/031_PW_15_CF_CS/README b/tests/01_PW/031_PW_15_CF_CS/README deleted file mode 100644 index ad914a9e031..00000000000 --- a/tests/01_PW/031_PW_15_CF_CS/README +++ /dev/null @@ -1 +0,0 @@ -calculate stress and force diff --git a/tests/01_PW/031_PW_15_CF_CS/STRU b/tests/01_PW/031_PW_15_CF_CS/STRU deleted file mode 100644 index b03baadd25e..00000000000 --- a/tests/01_PW/031_PW_15_CF_CS/STRU +++ /dev/null @@ -1,23 +0,0 @@ -ATOMIC_SPECIES -As 1 As_dojo.upf upf201 -Ga 1 Ga_dojo.upf upf201 - -LATTICE_CONSTANT -1 // add lattice constant, 10.58 ang - -LATTICE_VECTORS -5.33 5.33 0.0 -0.0 5.33 5.33 -5.33 0.0 5.33 -ATOMIC_POSITIONS -Direct //Cartesian or Direct coordinate. - -As -0 -1 -0.300000 0.3300000 0.27000000 0 0 0 - -Ga //Element Label -0 -1 //number of atom -0.00000 0.00000 0.000000 0 0 0 diff --git a/tests/01_PW/031_PW_15_CF_CS/result.ref b/tests/01_PW/031_PW_15_CF_CS/result.ref deleted file mode 100644 index cffe7c4fbee..00000000000 --- a/tests/01_PW/031_PW_15_CF_CS/result.ref +++ /dev/null @@ -1,8 +0,0 @@ -etotref -3497.9130098229479700 -etotperatomref -1748.9565049115 -totalforceref 99.345678 -totalstressref 32344.551605 -pointgroupref C_1 -spacegroupref C_1 -nksibzref 2 -totaltimeref 0.67 diff --git a/tests/01_PW/033_PW_CF_CS_S1_smallg/INPUT b/tests/01_PW/033_PW_CF_CS_S1_smallg/INPUT deleted file mode 100644 index 0275db7bb09..00000000000 --- a/tests/01_PW/033_PW_CF_CS_S1_smallg/INPUT +++ /dev/null @@ -1,27 +0,0 @@ -INPUT_PARAMETERS -#Parameters (1.General) -suffix autotest -calculation scf - -nbands 6 -symmetry 1 - -#Parameters (2.Iteration) -ecutwfc 20 -scf_thr 1e-9 -scf_nmax 100 - -cal_stress 1 -cal_force 1 - -#Parameters (3.Basis) -basis_type pw - -#Parameters (4.Smearing) -smearing_method gauss -smearing_sigma 0.002 - -#Parameters (5.Mixing) -mixing_type plain -mixing_beta 0.7 -pseudo_dir ../../PP_ORB diff --git a/tests/01_PW/033_PW_CF_CS_S1_smallg/README b/tests/01_PW/033_PW_CF_CS_S1_smallg/README deleted file mode 100644 index da5ebf6f425..00000000000 --- a/tests/01_PW/033_PW_CF_CS_S1_smallg/README +++ /dev/null @@ -1 +0,0 @@ -test energies, forces, stresses when system is large (small-g components) diff --git a/tests/01_PW/033_PW_CF_CS_S1_smallg/STRU b/tests/01_PW/033_PW_CF_CS_S1_smallg/STRU deleted file mode 100644 index 8eca0dff931..00000000000 --- a/tests/01_PW/033_PW_CF_CS_S1_smallg/STRU +++ /dev/null @@ -1,26 +0,0 @@ -ATOMIC_SPECIES -H 1.000 H.pz-vbc.UPF -O 1.000 O.pz-mt.UPF - - -LATTICE_CONSTANT -10 // add lattice constant, 10.58 ang - -LATTICE_VECTORS -1.0 0.0 0.0 -0.0 1.0 0.0 -0.0 0.0 1.0 - -ATOMIC_POSITIONS -Direct //Cartesian or Direct coordinate. - -H // element type -0 // magnetism -2 // number of atoms -0.57155 0.05539 0.000 1 1 1 -0.42845 0.05539 0.000 1 1 1 - -O // Element type -0 // magnetism -1 //number of atoms -0.500 0.000 0.000 1 1 1 diff --git a/tests/01_PW/033_PW_CF_CS_S1_smallg/result.ref b/tests/01_PW/033_PW_CF_CS_S1_smallg/result.ref deleted file mode 100644 index 3d005031616..00000000000 --- a/tests/01_PW/033_PW_CF_CS_S1_smallg/result.ref +++ /dev/null @@ -1,8 +0,0 @@ -etotref -376.5302709539294597 -etotperatomref -125.5100903180 -totalforceref 987.469003 -totalstressref 2048.626494 -pointgroupref C_2v -spacegroupref C_2v -nksibzref 1 -totaltimeref 0.85 diff --git a/tests/01_PW/034_PW_CF_CS_S2_smallg/INPUT b/tests/01_PW/034_PW_CF_CS_S2_smallg/INPUT deleted file mode 100644 index 0a8c37dd6ca..00000000000 --- a/tests/01_PW/034_PW_CF_CS_S2_smallg/INPUT +++ /dev/null @@ -1,28 +0,0 @@ -INPUT_PARAMETERS -#Parameters (1.General) -suffix autotest -calculation scf - -nbands 6 -symmetry 1 -nspin 2 - -#Parameters (2.Iteration) -ecutwfc 20 -scf_thr 1e-9 -scf_nmax 100 - -cal_stress 1 -cal_force 1 - -#Parameters (3.Basis) -basis_type pw - -#Parameters (4.Smearing) -smearing_method gauss -smearing_sigma 0.002 - -#Parameters (5.Mixing) -mixing_type plain -mixing_beta 0.7 -pseudo_dir ../../PP_ORB diff --git a/tests/01_PW/034_PW_CF_CS_S2_smallg/README b/tests/01_PW/034_PW_CF_CS_S2_smallg/README deleted file mode 100644 index 0129061674a..00000000000 --- a/tests/01_PW/034_PW_CF_CS_S2_smallg/README +++ /dev/null @@ -1 +0,0 @@ -Func:PW/nspin=2,small-g; Sys:H2O; Ref:energy,force,stress \ No newline at end of file diff --git a/tests/01_PW/034_PW_CF_CS_S2_smallg/STRU b/tests/01_PW/034_PW_CF_CS_S2_smallg/STRU deleted file mode 100644 index c3d7a7f4419..00000000000 --- a/tests/01_PW/034_PW_CF_CS_S2_smallg/STRU +++ /dev/null @@ -1,26 +0,0 @@ -ATOMIC_SPECIES -H 1.000 H.pz-vbc.UPF -O 1.000 O.pz-mt.UPF - - -LATTICE_CONSTANT -10 // add lattice constant, 10.58 ang - -LATTICE_VECTORS -1.0 0.0 0.0 -0.0 1.0 0.0 -0.0 0.0 1.0 - -ATOMIC_POSITIONS -Direct //Cartesian or Direct coordinate. - -H // element type -1 // magnetism -2 // number of atoms -0.57155 0.05539 0.000 1 1 1 -0.42845 0.05539 0.000 1 1 1 - -O // Element type -1 // magnetism -1 //number of atoms -0.500 0.000 0.000 1 1 1 diff --git a/tests/01_PW/034_PW_CF_CS_S2_smallg/result.ref b/tests/01_PW/034_PW_CF_CS_S2_smallg/result.ref deleted file mode 100644 index ae97180fbbd..00000000000 --- a/tests/01_PW/034_PW_CF_CS_S2_smallg/result.ref +++ /dev/null @@ -1,8 +0,0 @@ -etotref -376.5302709547709696 -etotperatomref -125.5100903183 -totalforceref 987.469181 -totalstressref 2048.626205 -pointgroupref C_2v -spacegroupref C_2v -nksibzref 1 -totaltimeref 1.68 diff --git a/tests/01_PW/035_PW_15_SO/STRU b/tests/01_PW/035_PW_15_SO/STRU deleted file mode 100644 index 901aa2771d4..00000000000 --- a/tests/01_PW/035_PW_15_SO/STRU +++ /dev/null @@ -1,23 +0,0 @@ -ATOMIC_SPECIES -As 1 As_ONCV_PBE_FR-1.1.upf upf201 -Ga 1 Ga_ONCV_PBE_FR-1.0.upf.txt upf201 - -LATTICE_CONSTANT -1 // add lattice constant, 10.58 ang - -LATTICE_VECTORS -5.34197 5.34197 0.0 -0.0 5.34197 5.34197 -5.34197 0.0 5.34197 -ATOMIC_POSITIONS -Direct //Cartesian or Direct coordinate. - -As -0 -1 -0.2500000 0.2500000 0.25000000 0 0 0 - -Ga //Element Label -0 -1 //number of atom -0.00000 0.00000 0.000000 0 0 0 diff --git a/tests/01_PW/037_PW_FM/INPUT b/tests/01_PW/037_PW_FM/INPUT deleted file mode 100644 index 90eb264db5b..00000000000 --- a/tests/01_PW/037_PW_FM/INPUT +++ /dev/null @@ -1,31 +0,0 @@ -INPUT_PARAMETERS -suffix autotest - -init_wfc random -#nbands 40 - -calculation scf -nspin 2 -ecutwfc 15 -scf_thr 1.0e-8 -scf_nmax 50 -out_chg 0 - -#init_chg file -#out_dos 1 -#dos_sigma 0.05 -#out_band 1 - -smearing_method gaussian -smearing_sigma 0.07 - -mixing_type broyden -mixing_beta 0.2 - -ks_solver cg -basis_type pw -symmetry 1 -pseudo_dir ../../PP_ORB -orbital_dir ../../PP_ORB - -pw_seed 1 diff --git a/tests/01_PW/037_PW_FM/KPT b/tests/01_PW/037_PW_FM/KPT deleted file mode 100644 index db314324451..00000000000 --- a/tests/01_PW/037_PW_FM/KPT +++ /dev/null @@ -1,4 +0,0 @@ -K_POINTS -0 -Gamma -1 1 1 0 0 0 diff --git a/tests/01_PW/037_PW_FM/README b/tests/01_PW/037_PW_FM/README deleted file mode 100644 index ac29b48cae4..00000000000 --- a/tests/01_PW/037_PW_FM/README +++ /dev/null @@ -1 +0,0 @@ -Fe Ferromagnetic, nspin 2, init magnet: up + up diff --git a/tests/01_PW/037_PW_FM/STRU b/tests/01_PW/037_PW_FM/STRU deleted file mode 100644 index 4ceed6ebbdb..00000000000 --- a/tests/01_PW/037_PW_FM/STRU +++ /dev/null @@ -1,22 +0,0 @@ -ATOMIC_SPECIES -Fe 1.000 Fe_ONCV_PBE-1.0.upf - -NUMERICAL_ORBITAL -Fe_gga_9au_100Ry_4s2p2d1f.orb - -LATTICE_CONSTANT -6 - -LATTICE_VECTORS -1.00 0.00 0.00 -0.00 1.00 0.00 -0.00 0.00 1.00 -ATOMIC_POSITIONS -Direct - -Fe -5.0 -2 -0.00 0.00 0.00 1 1 1 -0.50 0.50 0.50 1 1 1 - diff --git a/tests/01_PW/037_PW_FM/result.ref b/tests/01_PW/037_PW_FM/result.ref deleted file mode 100644 index 533bdd98806..00000000000 --- a/tests/01_PW/037_PW_FM/result.ref +++ /dev/null @@ -1,7 +0,0 @@ -etotref -5866.197297354694 -etotperatomref -2933.0986486773 -pointgroupref O_h -spacegroupref O_h -nksibzref 1 -totaltimeref 28.9098 -0.84 diff --git a/tests/01_PW/038_PW_NC_GGA_GRAD2/INPUT b/tests/01_PW/038_PW_NC_GGA_GRAD2/INPUT new file mode 100644 index 00000000000..0cd42b8d630 --- /dev/null +++ b/tests/01_PW/038_PW_NC_GGA_GRAD2/INPUT @@ -0,0 +1,38 @@ +INPUT_PARAMETERS + +init_wfc random +# PBE noncollinear GGA gradient regression +basis_type pw +calculation scf +noncolin 1 +nspin 4 +dft_functional pbe +gga_grad 2 +symmetry 0 +cal_force 1 +cal_stress 1 + +# plane wave and scf +ecutwfc 20 +pw_seed 1 +scf_thr 1.0e-10 +scf_nmax 150 +out_chg 0 + +# smearing method +smearing_method gaussian +smearing_sigma 0.02 + +# charge mixing method +mixing_type broyden +mixing_beta 0.2 +mixing_ndim 10 + +# diagonalizaiton +ks_solver dav_subspace +diago_smooth_ethr true +pw_diag_ndim 2 + +# directories +pseudo_dir ../../PP_ORB +suffix autotest diff --git a/tests/01_PW/038_PW_NC_GGA_GRAD2/KPT b/tests/01_PW/038_PW_NC_GGA_GRAD2/KPT new file mode 100644 index 00000000000..b5b3bdb1ae2 --- /dev/null +++ b/tests/01_PW/038_PW_NC_GGA_GRAD2/KPT @@ -0,0 +1,4 @@ +K_POINTS +0 +Gamma +1 1 2 0 0 0 diff --git a/tests/01_PW/038_PW_NC_GGA_GRAD2/README b/tests/01_PW/038_PW_NC_GGA_GRAD2/README new file mode 100644 index 00000000000..93236854c77 --- /dev/null +++ b/tests/01_PW/038_PW_NC_GGA_GRAD2/README @@ -0,0 +1,16 @@ +PBE noncollinear Fe2 regression for gga_grad=2, including forces and stress. + +Derived from 038_PW_NC with different initial spin directions on the two atoms, +explicit PBE/gga_grad settings, and scf_thr=1e-10. +The initial moments point in different directions and decay towards zero during +SCF. This case exercises the noncollinear path including its weak-moment limit; +it is not a regression for a stable magnetic ground state. +The pseudopotential is reused from tests/PP_ORB; no numerical orbitals are needed. + +From tests/01_PW, run: + OMP_NUM_THREADS=1 bash ../integrate/Autotest.sh -a /path/to/abacus -n 4 -r '^038_PW_NC_GGA_GRAD2$' + +result.ref is generated with one MPI rank using the same command with -n 1 -g. +The reference covers energy per atom, the force summary, and the stress summary. +Default Autotest tolerances apply; no case-specific threshold overrides are used. +This SCF regression complements the XC module's density/stress derivative tests. diff --git a/tests/01_PW/038_PW_NC_GGA_GRAD2/STRU b/tests/01_PW/038_PW_NC_GGA_GRAD2/STRU new file mode 100644 index 00000000000..29ad1c21031 --- /dev/null +++ b/tests/01_PW/038_PW_NC_GGA_GRAD2/STRU @@ -0,0 +1,18 @@ +ATOMIC_SPECIES +Fe 1.000 Fe_ONCV_PBE-1.0.upf + +LATTICE_CONSTANT +6 + +LATTICE_VECTORS +1.00 0.00 0.00 +0.00 1.00 0.00 +0.00 0.00 1.00 +ATOMIC_POSITIONS +Direct + +Fe +1.0 +2 +0.00 0.00 0.00 1 1 1 angle1 90 angle2 0 +0.51 0.52 0.53 1 1 1 angle1 70 angle2 60 diff --git a/tests/01_PW/038_PW_NC_GGA_GRAD2/result.ref b/tests/01_PW/038_PW_NC_GGA_GRAD2/result.ref new file mode 100644 index 00000000000..b21347b5a09 --- /dev/null +++ b/tests/01_PW/038_PW_NC_GGA_GRAD2/result.ref @@ -0,0 +1,5 @@ +etotref -6178.4040535332069339 +etotperatomref -3089.2020267666 +totalforceref 18.267150 +totalstressref 73803.586958 +totaltimeref 1.47 diff --git a/tests/01_PW/039_PW_FD_smear/INPUT b/tests/01_PW/039_PW_FD_smear/INPUT deleted file mode 100644 index 31337aed87c..00000000000 --- a/tests/01_PW/039_PW_FD_smear/INPUT +++ /dev/null @@ -1,27 +0,0 @@ -INPUT_PARAMETERS -#Parameters (1.General) -suffix autotest -calculation scf -init_wfc random - -nbands 35 -symmetry 1 -pseudo_dir ../../PP_ORB - -#Parameters (2.Iteration) -ecutwfc 20 -scf_thr 1e-8 -scf_nmax 100 - - -#Parameters (3.Basis) -basis_type pw - -#Parameters (4.Smearing) -smearing_method fd -smearing_sigma 0.002 - -#Parameters (5.Mixing) -mixing_type broyden -mixing_beta 0.6 -mixing_gg0 2 diff --git a/tests/01_PW/039_PW_FD_smear/README b/tests/01_PW/039_PW_FD_smear/README deleted file mode 100644 index 714b3435a6b..00000000000 --- a/tests/01_PW/039_PW_FD_smear/README +++ /dev/null @@ -1 +0,0 @@ -test smearing method: fd for (110) surface of Al diff --git a/tests/01_PW/039_PW_FD_smear/result.ref b/tests/01_PW/039_PW_FD_smear/result.ref deleted file mode 100644 index 5d64430424a..00000000000 --- a/tests/01_PW/039_PW_FD_smear/result.ref +++ /dev/null @@ -1,6 +0,0 @@ -etotref -9154.924406411264 -etotperatomref -1830.9848812823 -pointgroupref C_2v -spacegroupref D_2h -nksibzref 1 -totaltimeref 1.89 diff --git a/tests/01_PW/040_PW_FX_smear/INPUT b/tests/01_PW/040_PW_FX_smear/INPUT deleted file mode 100644 index dbb738bf2e9..00000000000 --- a/tests/01_PW/040_PW_FX_smear/INPUT +++ /dev/null @@ -1,25 +0,0 @@ -INPUT_PARAMETERS -#Parameters (1.General) -suffix autotest -calculation scf - -nbands 6 -symmetry 1 -pseudo_dir ../../PP_ORB - -#Parameters (2.Iteration) -ecutwfc 20 -scf_thr 1e-8 -scf_nmax 100 - - -#Parameters (3.Basis) -basis_type pw - -#Parameters (4.Smearing) -smearing_method fixed -smearing_sigma 0.002 - -#Parameters (5.Mixing) -mixing_type plain -mixing_beta 0.7 diff --git a/tests/01_PW/040_PW_FX_smear/README b/tests/01_PW/040_PW_FX_smear/README deleted file mode 100644 index ac5fc700793..00000000000 --- a/tests/01_PW/040_PW_FX_smear/README +++ /dev/null @@ -1 +0,0 @@ -test smearing method: fixed smearing for GaN (zinc-blend) diff --git a/tests/01_PW/040_PW_FX_smear/STRU b/tests/01_PW/040_PW_FX_smear/STRU deleted file mode 100644 index 9b42a124cad..00000000000 --- a/tests/01_PW/040_PW_FX_smear/STRU +++ /dev/null @@ -1,19 +0,0 @@ -ATOMIC_SPECIES -Si 14 Si.pz-vbc.UPF - -LATTICE_CONSTANT -10.2 // add lattice constant - -LATTICE_VECTORS -0.0 0.5 0.5 -0.5 0.0 0.5 -0.5 0.5 0.0 - -ATOMIC_POSITIONS -Direct - -Si // Element type -0.0 // magnetism -2 -0.00 0.00 0.00 1 1 1 -0.25 0.25 0.25 1 1 1 diff --git a/tests/01_PW/040_PW_FX_smear/result.ref b/tests/01_PW/040_PW_FX_smear/result.ref deleted file mode 100644 index 2cd72d532e0..00000000000 --- a/tests/01_PW/040_PW_FX_smear/result.ref +++ /dev/null @@ -1,6 +0,0 @@ -etotref -212.9934339557329 -etotperatomref -106.4967169779 -pointgroupref T_d -spacegroupref O_h -nksibzref 3 -totaltimeref diff --git a/tests/01_PW/041_PW_GA_smear/INPUT b/tests/01_PW/041_PW_GA_smear/INPUT deleted file mode 100644 index 3d3237a7d90..00000000000 --- a/tests/01_PW/041_PW_GA_smear/INPUT +++ /dev/null @@ -1,27 +0,0 @@ -INPUT_PARAMETERS -#Parameters (1.General) -suffix autotest -calculation scf -init_wfc random - -nbands 30 -symmetry 1 -pseudo_dir ../../PP_ORB - -#Parameters (2.Iteration) -ecutwfc 20 -scf_thr 1e-8 -scf_nmax 100 - - -#Parameters (3.Basis) -basis_type pw - -#Parameters (4.Smearing) -smearing_method gauss -smearing_sigma 0.002 - -#Parameters (5.Mixing) -mixing_type broyden -mixing_beta 0.3 -mixing_gg0 1.5 diff --git a/tests/01_PW/041_PW_GA_smear/README b/tests/01_PW/041_PW_GA_smear/README deleted file mode 100644 index 2cbe4af77fb..00000000000 --- a/tests/01_PW/041_PW_GA_smear/README +++ /dev/null @@ -1 +0,0 @@ -test smearing method: gauss for (110) surface of fcc Al diff --git a/tests/01_PW/041_PW_GA_smear/STRU b/tests/01_PW/041_PW_GA_smear/STRU deleted file mode 100644 index 6cbe32e800d..00000000000 --- a/tests/01_PW/041_PW_GA_smear/STRU +++ /dev/null @@ -1,22 +0,0 @@ -ATOMIC_SPECIES -Al 27 Al_ONCV_PBE-1.0.upf - -LATTICE_CONSTANT -1.00 - -LATTICE_VECTORS -5.41176 0.00000 0.00 -0.00000 7.65339 0.00 -0.00000 0.00000 34.0 - -ATOMIC_POSITIONS -Cartesian - -Al -0.0 -5 -0.00000 0.000000 0.00000 1 1 1 -2.70588 3.826695 2.70588 1 1 1 -0.00000 0.000000 5.41176 1 1 1 -2.70588 3.826695 8.11764 1 1 1 -0.00000 0.000000 10.82352 1 1 1 diff --git a/tests/01_PW/041_PW_GA_smear/result.ref b/tests/01_PW/041_PW_GA_smear/result.ref deleted file mode 100644 index fe2af220f7f..00000000000 --- a/tests/01_PW/041_PW_GA_smear/result.ref +++ /dev/null @@ -1,6 +0,0 @@ -etotref -9154.889008165452 -etotperatomref -1830.9778016331 -pointgroupref C_2v -spacegroupref D_2h -nksibzref 1 -totaltimeref 2.55 diff --git a/tests/01_PW/042_PW_M2_smear/INPUT b/tests/01_PW/042_PW_M2_smear/INPUT deleted file mode 100644 index 49d336afaa8..00000000000 --- a/tests/01_PW/042_PW_M2_smear/INPUT +++ /dev/null @@ -1,26 +0,0 @@ -INPUT_PARAMETERS -#Parameters (1.General) -suffix autotest -calculation scf -init_wfc random - -nbands 8 -symmetry 1 -pseudo_dir ../../PP_ORB - -#Parameters (2.Iteration) -ecutwfc 20 -scf_thr 1e-8 -scf_nmax 100 - - -#Parameters (3.Basis) -basis_type pw - -#Parameters (4.Smearing) -smearing_method mp2 -smearing_sigma 0.002 - -#Parameters (5.Mixing) -mixing_type plain -mixing_beta 0.7 diff --git a/tests/01_PW/042_PW_M2_smear/README b/tests/01_PW/042_PW_M2_smear/README deleted file mode 100644 index fc388d06a8e..00000000000 --- a/tests/01_PW/042_PW_M2_smear/README +++ /dev/null @@ -1 +0,0 @@ -test smearing method: mp2 for fcc Al diff --git a/tests/01_PW/042_PW_M2_smear/STRU b/tests/01_PW/042_PW_M2_smear/STRU deleted file mode 100644 index 8b010fa85ad..00000000000 --- a/tests/01_PW/042_PW_M2_smear/STRU +++ /dev/null @@ -1,18 +0,0 @@ -ATOMIC_SPECIES -Al 27 Al_ONCV_PBE-1.0.upf - -LATTICE_CONSTANT -10.2 // add lattice constant - -LATTICE_VECTORS -0.0 0.5 0.5 -0.5 0.0 0.5 -0.5 0.5 0.0 - -ATOMIC_POSITIONS -Direct - -Al // Element type -0.0 // magnetism -1 -0.00 0.00 0.00 1 1 1 diff --git a/tests/01_PW/042_PW_M2_smear/result.ref b/tests/01_PW/042_PW_M2_smear/result.ref deleted file mode 100644 index 8aa76f0e9ad..00000000000 --- a/tests/01_PW/042_PW_M2_smear/result.ref +++ /dev/null @@ -1,6 +0,0 @@ -etotref -1824.760079857581 -etotperatomref -1824.7600798576 -pointgroupref O_h -spacegroupref O_h -nksibzref 3 -totaltimeref 0.12612 diff --git a/tests/01_PW/043_PW_MP_smear/INPUT b/tests/01_PW/043_PW_MP_smear/INPUT deleted file mode 100644 index 862a4ad7f2f..00000000000 --- a/tests/01_PW/043_PW_MP_smear/INPUT +++ /dev/null @@ -1,26 +0,0 @@ -INPUT_PARAMETERS -#Parameters (1.General) -suffix autotest -calculation scf -init_wfc random - -nbands 12 -symmetry 1 -pseudo_dir ../../PP_ORB - -#Parameters (2.Iteration) -ecutwfc 20 -scf_thr 1e-8 -scf_nmax 100 - - -#Parameters (3.Basis) -basis_type pw - -#Parameters (4.Smearing) -smearing_method mp -smearing_sigma 0.002 - -#Parameters (5.Mixing) -mixing_type plain -mixing_beta 0.5 diff --git a/tests/01_PW/043_PW_MP_smear/README b/tests/01_PW/043_PW_MP_smear/README deleted file mode 100644 index 66295923426..00000000000 --- a/tests/01_PW/043_PW_MP_smear/README +++ /dev/null @@ -1 +0,0 @@ -test smearing method: mp for bcc Fe diff --git a/tests/01_PW/043_PW_MP_smear/STRU b/tests/01_PW/043_PW_MP_smear/STRU deleted file mode 100644 index 5fbb1a3b879..00000000000 --- a/tests/01_PW/043_PW_MP_smear/STRU +++ /dev/null @@ -1,18 +0,0 @@ -ATOMIC_SPECIES -Fe 56 Fe_ONCV_PBE-1.0.upf - -LATTICE_CONSTANT -10.2 // add lattice constant - -LATTICE_VECTORS --0.5 0.5 0.5 -0.5 -0.5 0.5 -0.5 0.5 -0.5 - -ATOMIC_POSITIONS -Direct - -Fe // Element type -0.0 // magnetism -1 -0.00 0.00 0.00 1 1 1 diff --git a/tests/01_PW/043_PW_MP_smear/result.ref b/tests/01_PW/043_PW_MP_smear/result.ref deleted file mode 100644 index 40717e5d5eb..00000000000 --- a/tests/01_PW/043_PW_MP_smear/result.ref +++ /dev/null @@ -1,6 +0,0 @@ -etotref -3076.795495833718 -etotperatomref -3076.7954958337 -pointgroupref O_h -spacegroupref O_h -nksibzref 3 -totaltimeref 0.32529 diff --git a/tests/01_PW/044_PW_MV_smear/INPUT b/tests/01_PW/044_PW_MV_smear/INPUT deleted file mode 100644 index 0602a022f55..00000000000 --- a/tests/01_PW/044_PW_MV_smear/INPUT +++ /dev/null @@ -1,26 +0,0 @@ -INPUT_PARAMETERS -#Parameters (1.General) -suffix autotest -calculation scf -init_wfc random - -nbands 6 -symmetry 1 -pseudo_dir ../../PP_ORB - -#Parameters (2.Iteration) -ecutwfc 20 -scf_thr 1e-8 -scf_nmax 100 - - -#Parameters (3.Basis) -basis_type pw - -#Parameters (4.Smearing) -smearing_method mv -smearing_sigma 0.002 - -#Parameters (5.Mixing) -mixing_type plain -mixing_beta 0.5 diff --git a/tests/01_PW/044_PW_MV_smear/README b/tests/01_PW/044_PW_MV_smear/README deleted file mode 100644 index f37ce648005..00000000000 --- a/tests/01_PW/044_PW_MV_smear/README +++ /dev/null @@ -1 +0,0 @@ -test smearing method: mv for hcp Be diff --git a/tests/01_PW/044_PW_MV_smear/STRU b/tests/01_PW/044_PW_MV_smear/STRU deleted file mode 100644 index 1a57eff484c..00000000000 --- a/tests/01_PW/044_PW_MV_smear/STRU +++ /dev/null @@ -1,19 +0,0 @@ -ATOMIC_SPECIES -Be 9 Be_ONCV_PBE-1.0.upf - -LATTICE_CONSTANT -4.30 // add lattice constant - -LATTICE_VECTORS -1.0 0.0 0.0 --0.5 0.866 0.0 -0.0 0.0 1.568 - -ATOMIC_POSITIONS -Direct - -Be // Element type -0.0 // magnetism -2 -0.000 0.000 0.00 1 1 1 -0.667 0.333 0.50 1 1 1 diff --git a/tests/01_PW/044_PW_MV_smear/result.ref b/tests/01_PW/044_PW_MV_smear/result.ref deleted file mode 100644 index a90f6f0111f..00000000000 --- a/tests/01_PW/044_PW_MV_smear/result.ref +++ /dev/null @@ -1,6 +0,0 @@ -etotref -712.6316223439621 -etotperatomref -356.3158111720 -pointgroupref C_1h -spacegroupref C_2h -nksibzref 8 -totaltimeref 0.15855 diff --git a/tests/01_PW/052_PW_OB/INPUT b/tests/01_PW/052_PW_OB/INPUT deleted file mode 100644 index 830c99509f6..00000000000 --- a/tests/01_PW/052_PW_OB/INPUT +++ /dev/null @@ -1,34 +0,0 @@ -INPUT_PARAMETERS -#Parameters (1.General) -suffix autotest -calculation nscf - -nbands 6 -symmetry 0 -pseudo_dir ../../PP_ORB -read_file_dir ./ -kpar 2 - -#Parameters (2.Iteration) -ecutwfc 20 -scf_thr 1e-8 -scf_nmax 100 - -init_chg file -init_wfc random -out_band 1 -pw_diag_thr 0.00001 - -#Parameters (3.Basis) -basis_type pw - -#Parameters (4.Smearing) -smearing_method gauss -smearing_sigma 0.002 - -#Parameters (5.Mixing) -mixing_type plain -mixing_beta 0.7 - - -pw_seed 1 diff --git a/tests/01_PW/052_PW_OB/KPT b/tests/01_PW/052_PW_OB/KPT deleted file mode 100644 index b86bf0dff35..00000000000 --- a/tests/01_PW/052_PW_OB/KPT +++ /dev/null @@ -1,5 +0,0 @@ -K_POINTS -2 -Line -0.5 0.5 0.5 5 -0.0 0.0 0.0 1 diff --git a/tests/01_PW/052_PW_OB/README b/tests/01_PW/052_PW_OB/README deleted file mode 100644 index def433f1347..00000000000 --- a/tests/01_PW/052_PW_OB/README +++ /dev/null @@ -1 +0,0 @@ -output bands in pw basis diff --git a/tests/01_PW/052_PW_OB/STRU b/tests/01_PW/052_PW_OB/STRU deleted file mode 100644 index 9b42a124cad..00000000000 --- a/tests/01_PW/052_PW_OB/STRU +++ /dev/null @@ -1,19 +0,0 @@ -ATOMIC_SPECIES -Si 14 Si.pz-vbc.UPF - -LATTICE_CONSTANT -10.2 // add lattice constant - -LATTICE_VECTORS -0.0 0.5 0.5 -0.5 0.0 0.5 -0.5 0.5 0.0 - -ATOMIC_POSITIONS -Direct - -Si // Element type -0.0 // magnetism -2 -0.00 0.00 0.00 1 1 1 -0.25 0.25 0.25 1 1 1 diff --git a/tests/01_PW/052_PW_OB/band.txt.ref b/tests/01_PW/052_PW_OB/band.txt.ref deleted file mode 100644 index 0a1088ceb5e..00000000000 --- a/tests/01_PW/052_PW_OB/band.txt.ref +++ /dev/null @@ -1,6 +0,0 @@ - 1 0.00000000 -3.38704889 -0.79801666 5.06489026 5.06490253 7.84115900 9.60595365 - 2 0.17320508 -3.86198803 -0.07518005 5.13653307 5.13653200 7.91811409 9.68496369 - 3 0.34641016 -4.63079483 1.43421001 5.35290770 5.35285262 8.15538060 9.81497711 - 4 0.51961524 -5.25812025 3.25095973 5.69541240 5.69542353 8.51883375 9.65336577 - 5 0.69282032 -5.65197852 5.13708720 6.08467575 6.08469139 8.86796173 9.13705931 - 6 0.86602540 -5.78586742 6.28881704 6.28877545 6.28878152 8.83653074 8.83650706 diff --git a/tests/01_PW/052_PW_OB/result.ref b/tests/01_PW/052_PW_OB/result.ref deleted file mode 100644 index 6cf8149fd04..00000000000 --- a/tests/01_PW/052_PW_OB/result.ref +++ /dev/null @@ -1,4 +0,0 @@ -etotref -206.9727001566756 -etotperatomref -103.4863500783 -CompareBand_pass 0 -totaltimeref 0.23 diff --git a/tests/01_PW/053_PW_OD/INPUT b/tests/01_PW/053_PW_OD/INPUT deleted file mode 100644 index af164a1f41a..00000000000 --- a/tests/01_PW/053_PW_OD/INPUT +++ /dev/null @@ -1,35 +0,0 @@ -INPUT_PARAMETERS -#Parameters (1.General) -suffix autotest -calculation nscf - -nbands 6 -symmetry 0 -pseudo_dir ../../PP_ORB -read_file_dir ./ - -#Parameters (2.Iteration) -ecutwfc 20 -scf_thr 1e-8 -scf_nmax 100 - -init_chg file -out_dos 1 -pw_diag_thr 0.00001 - -#Parameters (3.Basis) -basis_type pw - -#Parameters (4.Smearing) -smearing_method gauss -smearing_sigma 0.002 - -#Parameters (5.Mixing) -mixing_type plain -mixing_beta 0.7 - -#dos_emin_ev -15 #minimal range for dos -#dos_emax_ev 15 #maximal range for dos -dos_edelta_ev 0.01 #delta energy for dos -#dos_scale 0.01 #scale dos range by -dos_sigma 0.05 #gauss b coefficeinet(default=0.07) diff --git a/tests/01_PW/053_PW_OD/KPT b/tests/01_PW/053_PW_OD/KPT deleted file mode 100644 index b86bf0dff35..00000000000 --- a/tests/01_PW/053_PW_OD/KPT +++ /dev/null @@ -1,5 +0,0 @@ -K_POINTS -2 -Line -0.5 0.5 0.5 5 -0.0 0.0 0.0 1 diff --git a/tests/01_PW/053_PW_OD/README b/tests/01_PW/053_PW_OD/README deleted file mode 100644 index 24d0d8a2f24..00000000000 --- a/tests/01_PW/053_PW_OD/README +++ /dev/null @@ -1 +0,0 @@ -output DOS in pw basis diff --git a/tests/01_PW/053_PW_OD/STRU b/tests/01_PW/053_PW_OD/STRU deleted file mode 100644 index 9b42a124cad..00000000000 --- a/tests/01_PW/053_PW_OD/STRU +++ /dev/null @@ -1,19 +0,0 @@ -ATOMIC_SPECIES -Si 14 Si.pz-vbc.UPF - -LATTICE_CONSTANT -10.2 // add lattice constant - -LATTICE_VECTORS -0.0 0.5 0.5 -0.5 0.0 0.5 -0.5 0.5 0.0 - -ATOMIC_POSITIONS -Direct - -Si // Element type -0.0 // magnetism -2 -0.00 0.00 0.00 1 1 1 -0.25 0.25 0.25 1 1 1 diff --git a/tests/01_PW/053_PW_OD/chg.cube b/tests/01_PW/053_PW_OD/chg.cube deleted file mode 100644 index d91fd221e30..00000000000 --- a/tests/01_PW/053_PW_OD/chg.cube +++ /dev/null @@ -1,2312 +0,0 @@ -STEP: 0 Cubefile created from ABACUS. Inner loop is z, followed by y and x -1 (nspin) 0.479503 (fermi energy, in Ry) -2 0.0 0.0 0.0 -24 0.212500 0.212500 0.000000 -24 0.212500 0.000000 0.212500 -24 0.000000 0.212500 0.212500 -14 14 0.000000 0.000000 0.000000 -14 14 2.550001 2.550001 2.550001 - 9.03211615896e-04 4.78306293438e-03 1.79849681520e-02 3.86813628609e-02 5.67646475266e-02 6.31142666785e-02 - 5.82846684414e-02 4.88652050826e-02 4.00749900281e-02 3.35320462758e-02 2.89819100992e-02 2.61347637909e-02 - 2.51358351357e-02 2.61347637909e-02 2.89819100992e-02 3.35320462758e-02 4.00749900281e-02 4.88652050826e-02 - 5.82846684414e-02 6.31142666785e-02 5.67646475266e-02 3.86813628609e-02 1.79849681520e-02 4.78306293438e-03 - 4.78306293438e-03 1.49395756337e-02 3.47728130512e-02 5.65857611912e-02 6.91691506706e-02 6.87390140452e-02 - 6.04031976834e-02 5.07662838782e-02 4.30069852462e-02 3.73919347545e-02 3.34978085182e-02 3.13851597832e-02 - 3.13851597832e-02 3.34978085182e-02 3.73919347545e-02 4.30069852462e-02 5.07662838782e-02 6.04031976834e-02 - 6.87390140452e-02 6.91691506706e-02 5.65857611912e-02 3.47728130512e-02 1.49395756337e-02 4.78306293438e-03 - 1.79849681520e-02 3.47728130512e-02 5.63481382706e-02 7.22242608078e-02 7.59717702126e-02 7.00265784056e-02 - 6.08204348290e-02 5.26365510379e-02 4.64352697346e-02 4.17937584090e-02 3.86408229568e-02 3.74820931920e-02 - 3.86408229568e-02 4.17937584090e-02 4.64352697346e-02 5.26365510379e-02 6.08204348290e-02 7.00265784056e-02 - 7.59717702126e-02 7.22242608078e-02 5.63481382706e-02 3.47728130512e-02 1.79849681520e-02 1.20127916963e-02 - 3.86813628609e-02 5.65857611912e-02 7.22242608078e-02 7.85611057280e-02 7.55510577091e-02 6.81380649556e-02 - 6.07372364008e-02 5.48042821337e-02 4.99875697106e-02 4.60708732894e-02 4.37078082031e-02 4.37078082031e-02 - 4.60708732894e-02 4.99875697106e-02 5.48042821337e-02 6.07372364008e-02 6.81380649556e-02 7.55510577091e-02 - 7.85611057280e-02 7.22242608078e-02 5.65857611912e-02 3.86813628609e-02 2.72980449809e-02 2.72980449809e-02 - 5.67646475266e-02 6.91691506706e-02 7.59717702126e-02 7.55510577091e-02 7.08330821436e-02 6.54257706676e-02 - 6.08204348290e-02 5.66502458441e-02 5.24827367699e-02 4.90171582760e-02 4.76311652618e-02 4.90171582760e-02 - 5.24827367699e-02 5.66502458441e-02 6.08204348290e-02 6.54257706676e-02 7.08330821436e-02 7.55510577091e-02 - 7.59717702126e-02 6.91691506706e-02 5.67646475266e-02 4.49731492966e-02 4.01965567029e-02 4.49731492966e-02 - 6.31142666785e-02 6.87390140452e-02 7.00265784056e-02 6.81380649556e-02 6.54257706676e-02 6.30211041418e-02 - 6.04031976834e-02 5.67802908281e-02 5.26323100532e-02 4.97519001270e-02 4.97519001270e-02 5.26323100532e-02 - 5.67802908281e-02 6.04031976834e-02 6.30211041418e-02 6.54257706676e-02 6.81380649556e-02 7.00265784056e-02 - 6.87390140452e-02 6.31142666785e-02 5.52660749302e-02 4.96150823327e-02 4.96150823327e-02 5.52660749302e-02 - 5.82846684414e-02 6.04031976834e-02 6.08204348290e-02 6.07372364008e-02 6.08204348290e-02 6.04031976834e-02 - 5.82846684414e-02 5.43728424707e-02 5.03920982744e-02 4.87050863807e-02 5.03920982744e-02 5.43728424707e-02 - 5.82846684414e-02 6.04031976834e-02 6.08204348290e-02 6.07372364008e-02 6.08204348290e-02 6.04031976834e-02 - 5.82846684414e-02 5.43728424707e-02 5.03920982744e-02 4.87050863807e-02 5.03920982744e-02 5.43728424707e-02 - 4.88652050826e-02 5.07662838782e-02 5.26365510379e-02 5.48042821337e-02 5.66502458441e-02 5.67802908281e-02 - 5.43728424707e-02 5.04488982123e-02 4.74726445019e-02 4.74726445019e-02 5.04488982123e-02 5.43728424707e-02 - 5.67802908281e-02 5.66502458441e-02 5.48042821337e-02 5.26365510379e-02 5.07662838782e-02 4.88652050826e-02 - 4.64972219765e-02 4.39611245421e-02 4.22555839350e-02 4.22555839350e-02 4.39611245421e-02 4.64972219765e-02 - 4.00749900281e-02 4.30069852462e-02 4.64352697346e-02 4.99875697106e-02 5.24827367699e-02 5.26323100532e-02 - 5.03920982744e-02 4.74726445019e-02 4.61549122721e-02 4.74726445019e-02 5.03920982744e-02 5.26323100532e-02 - 5.24827367699e-02 4.99875697106e-02 4.64352697346e-02 4.30069852462e-02 4.00749900281e-02 3.74959847737e-02 - 3.52270678377e-02 3.35736913190e-02 3.29591336478e-02 3.35736913190e-02 3.52270678377e-02 3.74959847737e-02 - 3.35320462758e-02 3.73919347545e-02 4.17937584090e-02 4.60708732894e-02 4.90171582760e-02 4.97519001270e-02 - 4.87050863807e-02 4.74726445019e-02 4.74726445019e-02 4.87050863807e-02 4.97519001270e-02 4.90171582760e-02 - 4.60708732894e-02 4.17937584090e-02 3.73919347545e-02 3.35320462758e-02 3.03107984462e-02 2.76901755249e-02 - 2.57739478448e-02 2.47523565945e-02 2.47523565945e-02 2.57739478448e-02 2.76901755249e-02 3.03107984462e-02 - 2.89819100992e-02 3.34978085182e-02 3.86408229568e-02 4.37078082031e-02 4.76311652618e-02 4.97519001270e-02 - 5.03920982744e-02 5.04488982123e-02 5.03920982744e-02 4.97519001270e-02 4.76311652618e-02 4.37078082031e-02 - 3.86408229568e-02 3.34978085182e-02 2.89819100992e-02 2.52525868671e-02 2.22527540382e-02 2.00031512771e-02 - 1.86064391176e-02 1.81349712987e-02 1.86064391176e-02 2.00031512771e-02 2.22527540382e-02 2.52525868671e-02 - 2.61347637909e-02 3.13851597832e-02 3.74820931920e-02 4.37078082031e-02 4.90171582760e-02 5.26323100532e-02 - 5.43728424707e-02 5.43728424707e-02 5.26323100532e-02 4.90171582760e-02 4.37078082031e-02 3.74820931920e-02 - 3.13851597832e-02 2.61347637909e-02 2.19036737434e-02 1.85732113773e-02 1.60537467662e-02 1.43606388158e-02 - 1.35182636086e-02 1.35182636086e-02 1.43606388158e-02 1.60537467662e-02 1.85732113773e-02 2.19036737434e-02 - 2.51358351357e-02 3.13851597832e-02 3.86408229568e-02 4.60708732894e-02 5.24827367699e-02 5.67802908281e-02 - 5.82846684414e-02 5.67802908281e-02 5.24827367699e-02 4.60708732894e-02 3.86408229568e-02 3.13851597832e-02 - 2.51358351357e-02 2.01594924058e-02 1.63414636372e-02 1.35012692014e-02 1.15426690136e-02 1.04118462818e-02 - 1.00461401111e-02 1.04118462818e-02 1.15426690136e-02 1.35012692014e-02 1.63414636372e-02 2.01594924058e-02 - 2.61347637909e-02 3.34978085182e-02 4.17937584090e-02 4.99875697106e-02 5.66502458441e-02 6.04031976834e-02 - 6.04031976834e-02 5.66502458441e-02 4.99875697106e-02 4.17937584090e-02 3.34978085182e-02 2.61347637909e-02 - 2.01594924058e-02 1.55749641354e-02 1.22047713863e-02 9.87827853279e-03 8.45428906874e-03 7.79360764654e-03 - 7.79360764654e-03 8.45428906874e-03 9.87827853279e-03 1.22047713863e-02 1.55749641354e-02 2.01594924058e-02 - 2.89819100992e-02 3.73919347545e-02 4.64352697346e-02 5.48042821337e-02 6.08204348290e-02 6.30211041418e-02 - 6.08204348290e-02 5.48042821337e-02 4.64352697346e-02 3.73919347545e-02 2.89819100992e-02 2.19036737434e-02 - 1.63414636372e-02 1.22047713863e-02 9.31927456850e-03 7.49319081292e-03 6.52118028642e-03 6.22225650652e-03 - 6.52118028642e-03 7.49319081292e-03 9.31927456850e-03 1.22047713863e-02 1.63414636372e-02 2.19036737434e-02 - 3.35320462758e-02 4.30069852462e-02 5.26365510379e-02 6.07372364008e-02 6.54257706676e-02 6.54257706676e-02 - 6.07372364008e-02 5.26365510379e-02 4.30069852462e-02 3.35320462758e-02 2.52525868671e-02 1.85732113773e-02 - 1.35012692014e-02 9.87827853279e-03 7.49319081292e-03 6.10931604239e-03 5.49418812711e-03 5.49418812711e-03 - 6.10931604239e-03 7.49319081292e-03 9.87827853279e-03 1.35012692014e-02 1.85732113773e-02 2.52525868671e-02 - 4.00749900281e-02 5.07662838782e-02 6.08204348290e-02 6.81380649556e-02 7.08330821436e-02 6.81380649556e-02 - 6.08204348290e-02 5.07662838782e-02 4.00749900281e-02 3.03107984462e-02 2.22527540382e-02 1.60537467662e-02 - 1.15426690136e-02 8.45428906874e-03 6.52118028642e-03 5.49418812711e-03 5.17760394975e-03 5.49418812711e-03 - 6.52118028642e-03 8.45428906874e-03 1.15426690136e-02 1.60537467662e-02 2.22527540382e-02 3.03107984462e-02 - 4.88652050826e-02 6.04031976834e-02 7.00265784056e-02 7.55510577091e-02 7.55510577091e-02 7.00265784056e-02 - 6.04031976834e-02 4.88652050826e-02 3.74959847737e-02 2.76901755249e-02 2.00031512771e-02 1.43606388158e-02 - 1.04118462818e-02 7.79360764654e-03 6.22225650652e-03 5.49418812711e-03 5.49418812711e-03 6.22225650652e-03 - 7.79360764654e-03 1.04118462818e-02 1.43606388158e-02 2.00031512771e-02 2.76901755249e-02 3.74959847737e-02 - 5.82846684414e-02 6.87390140452e-02 7.59717702126e-02 7.85611057280e-02 7.59717702126e-02 6.87390140452e-02 - 5.82846684414e-02 4.64972219765e-02 3.52270678377e-02 2.57739478448e-02 1.86064391176e-02 1.35182636086e-02 - 1.00461401111e-02 7.79360764654e-03 6.52118028642e-03 6.10931604239e-03 6.52118028642e-03 7.79360764654e-03 - 1.00461401111e-02 1.35182636086e-02 1.86064391176e-02 2.57739478448e-02 3.52270678377e-02 4.64972219765e-02 - 6.31142666785e-02 6.91691506706e-02 7.22242608078e-02 7.22242608078e-02 6.91691506706e-02 6.31142666785e-02 - 5.43728424707e-02 4.39611245421e-02 3.35736913190e-02 2.47523565945e-02 1.81349712987e-02 1.35182636086e-02 - 1.04118462818e-02 8.45428906874e-03 7.49319081292e-03 7.49319081292e-03 8.45428906874e-03 1.04118462818e-02 - 1.35182636086e-02 1.81349712987e-02 2.47523565945e-02 3.35736913190e-02 4.39611245421e-02 5.43728424707e-02 - 5.67646475266e-02 5.65857611912e-02 5.63481382706e-02 5.65857611912e-02 5.67646475266e-02 5.52660749302e-02 - 5.03920982744e-02 4.22555839350e-02 3.29591336478e-02 2.47523565945e-02 1.86064391176e-02 1.43606388158e-02 - 1.15426690136e-02 9.87827853279e-03 9.31927456850e-03 9.87827853279e-03 1.15426690136e-02 1.43606388158e-02 - 1.86064391176e-02 2.47523565945e-02 3.29591336478e-02 4.22555839350e-02 5.03920982744e-02 5.52660749302e-02 - 3.86813628609e-02 3.47728130512e-02 3.47728130512e-02 3.86813628609e-02 4.49731492966e-02 4.96150823327e-02 - 4.87050863807e-02 4.22555839350e-02 3.35736913190e-02 2.57739478448e-02 2.00031512771e-02 1.60537467662e-02 - 1.35012692014e-02 1.22047713863e-02 1.22047713863e-02 1.35012692014e-02 1.60537467662e-02 2.00031512771e-02 - 2.57739478448e-02 3.35736913190e-02 4.22555839350e-02 4.87050863807e-02 4.96150823327e-02 4.49731492966e-02 - 1.79849681520e-02 1.49395756337e-02 1.79849681520e-02 2.72980449809e-02 4.01965567029e-02 4.96150823327e-02 - 5.03920982744e-02 4.39611245421e-02 3.52270678377e-02 2.76901755249e-02 2.22527540382e-02 1.85732113773e-02 - 1.63414636372e-02 1.55749641354e-02 1.63414636372e-02 1.85732113773e-02 2.22527540382e-02 2.76901755249e-02 - 3.52270678377e-02 4.39611245421e-02 5.03920982744e-02 4.96150823327e-02 4.01965567029e-02 2.72980449809e-02 - 4.78306293438e-03 4.78306293438e-03 1.20127916963e-02 2.72980449809e-02 4.49731492966e-02 5.52660749302e-02 - 5.43728424707e-02 4.64972219765e-02 3.74959847737e-02 3.03107984462e-02 2.52525868671e-02 2.19036737434e-02 - 2.01594924058e-02 2.01594924058e-02 2.19036737434e-02 2.52525868671e-02 3.03107984462e-02 3.74959847737e-02 - 4.64972219765e-02 5.43728424707e-02 5.52660749302e-02 4.49731492966e-02 2.72980449809e-02 1.20127916963e-02 - 4.78306293438e-03 1.49395756337e-02 3.47728130512e-02 5.65857611912e-02 6.91691506706e-02 6.87390140452e-02 - 6.04031976834e-02 5.07662838782e-02 4.30069852462e-02 3.73919347545e-02 3.34978085182e-02 3.13851597832e-02 - 3.13851597832e-02 3.34978085182e-02 3.73919347545e-02 4.30069852462e-02 5.07662838782e-02 6.04031976834e-02 - 6.87390140452e-02 6.91691506706e-02 5.65857611912e-02 3.47728130512e-02 1.49395756337e-02 4.78306293438e-03 - 1.49395756337e-02 3.23216192243e-02 5.54677921285e-02 7.32259633424e-02 7.81603362307e-02 7.24709839797e-02 - 6.30211041418e-02 5.45216826827e-02 4.80826222932e-02 4.32645707778e-02 3.99801891127e-02 3.87690158276e-02 - 3.99801891127e-02 4.32645707778e-02 4.80826222932e-02 5.45216826827e-02 6.30211041418e-02 7.24709839797e-02 - 7.81603362307e-02 7.32259633424e-02 5.54677921285e-02 3.23216192243e-02 1.49395756337e-02 8.92458323932e-03 - 3.47728130512e-02 5.54677921285e-02 7.45491088161e-02 8.33624452951e-02 8.11535443280e-02 7.34279203403e-02 - 6.54257706676e-02 5.89796865623e-02 5.37274523246e-02 4.94024317434e-02 4.67604720778e-02 4.67604720778e-02 - 4.94024317434e-02 5.37274523246e-02 5.89796865623e-02 6.54257706676e-02 7.34279203403e-02 8.11535443280e-02 - 8.33624452951e-02 7.45491088161e-02 5.54677921285e-02 3.47728130512e-02 2.21086562145e-02 2.21086562145e-02 - 5.65857611912e-02 7.32259633424e-02 8.33624452951e-02 8.43291015351e-02 7.94813267109e-02 7.34093289667e-02 - 6.81380649556e-02 6.32743536159e-02 5.82637609313e-02 5.39959847166e-02 5.22694162731e-02 5.39959847166e-02 - 5.82637609313e-02 6.32743536159e-02 6.81380649556e-02 7.34093289667e-02 7.94813267109e-02 8.43291015351e-02 - 8.33624452951e-02 7.32259633424e-02 5.65857611912e-02 4.14983823737e-02 3.55340946595e-02 4.14983823737e-02 - 6.91691506706e-02 7.81603362307e-02 8.11535443280e-02 7.94813267109e-02 7.63593661757e-02 7.34279203403e-02 - 7.00265784056e-02 6.50724984809e-02 5.92671896593e-02 5.51986766687e-02 5.51986766687e-02 5.92671896593e-02 - 6.50724984809e-02 7.00265784056e-02 7.34279203403e-02 7.63593661757e-02 7.94813267109e-02 8.11535443280e-02 - 7.81603362307e-02 6.91691506706e-02 5.74779615393e-02 4.93152444260e-02 4.93152444260e-02 5.74779615393e-02 - 6.87390140452e-02 7.24709839797e-02 7.34279203403e-02 7.34093289667e-02 7.34279203403e-02 7.24709839797e-02 - 6.87390140452e-02 6.22085036765e-02 5.57017474383e-02 5.29690593685e-02 5.57017474383e-02 6.22085036765e-02 - 6.87390140452e-02 7.24709839797e-02 7.34279203403e-02 7.34093289667e-02 7.34279203403e-02 7.24709839797e-02 - 6.87390140452e-02 6.22085036765e-02 5.57017474383e-02 5.29690593685e-02 5.57017474383e-02 6.22085036765e-02 - 6.04031976834e-02 6.30211041418e-02 6.54257706676e-02 6.81380649556e-02 7.00265784056e-02 6.87390140452e-02 - 6.31142666785e-02 5.52660749302e-02 4.96150823327e-02 4.96150823327e-02 5.52660749302e-02 6.31142666785e-02 - 6.87390140452e-02 7.00265784056e-02 6.81380649556e-02 6.54257706676e-02 6.30211041418e-02 6.04031976834e-02 - 5.67802908281e-02 5.26323100532e-02 4.97519001270e-02 4.97519001270e-02 5.26323100532e-02 5.67802908281e-02 - 5.07662838782e-02 5.45216826827e-02 5.89796865623e-02 6.32743536159e-02 6.50724984809e-02 6.22085036765e-02 - 5.52660749302e-02 4.80686289281e-02 4.50421513586e-02 4.80686289281e-02 5.52660749302e-02 6.22085036765e-02 - 6.50724984809e-02 6.32743536159e-02 5.89796865623e-02 5.45216826827e-02 5.07662838782e-02 4.74317203940e-02 - 4.42738472380e-02 4.17752511891e-02 4.08013474350e-02 4.17752511891e-02 4.42738472380e-02 4.74317203940e-02 - 4.30069852462e-02 4.80826222932e-02 5.37274523246e-02 5.82637609313e-02 5.92671896593e-02 5.57017474383e-02 - 4.96150823327e-02 4.50421513586e-02 4.50421513586e-02 4.96150823327e-02 5.57017474383e-02 5.92671896593e-02 - 5.82637609313e-02 5.37274523246e-02 4.80826222932e-02 4.30069852462e-02 3.88905177841e-02 3.55533814714e-02 - 3.30014183614e-02 3.15640784817e-02 3.15640784817e-02 3.30014183614e-02 3.55533814714e-02 3.88905177841e-02 - 3.73919347545e-02 4.32645707778e-02 4.94024317434e-02 5.39959847166e-02 5.51986766687e-02 5.29690593685e-02 - 4.96150823327e-02 4.80686289281e-02 4.96150823327e-02 5.29690593685e-02 5.51986766687e-02 5.39959847166e-02 - 4.94024317434e-02 4.32645707778e-02 3.73919347545e-02 3.25477771477e-02 2.87349610469e-02 2.58559673304e-02 - 2.40042344862e-02 2.33592944544e-02 2.40042344862e-02 2.58559673304e-02 2.87349610469e-02 3.25477771477e-02 - 3.34978085182e-02 3.99801891127e-02 4.67604720778e-02 5.22694162731e-02 5.51986766687e-02 5.57017474383e-02 - 5.52660749302e-02 5.52660749302e-02 5.57017474383e-02 5.51986766687e-02 5.22694162731e-02 4.67604720778e-02 - 3.99801891127e-02 3.34978085182e-02 2.81262227272e-02 2.39287446557e-02 2.07503567214e-02 1.85584096087e-02 - 1.74298422700e-02 1.74298422700e-02 1.85584096087e-02 2.07503567214e-02 2.39287446557e-02 2.81262227272e-02 - 3.13851597832e-02 3.87690158276e-02 4.67604720778e-02 5.39959847166e-02 5.92671896593e-02 6.22085036765e-02 - 6.31142666785e-02 6.22085036765e-02 5.92671896593e-02 5.39959847166e-02 4.67604720778e-02 3.87690158276e-02 - 3.13851597832e-02 2.53584894772e-02 2.07351318935e-02 1.72789698058e-02 1.48305747723e-02 1.33587578257e-02 - 1.28681196083e-02 1.33587578257e-02 1.48305747723e-02 1.72789698058e-02 2.07351318935e-02 2.53584894772e-02 - 3.13851597832e-02 3.99801891127e-02 4.94024317434e-02 5.82637609313e-02 6.50724984809e-02 6.87390140452e-02 - 6.87390140452e-02 6.50724984809e-02 5.82637609313e-02 4.94024317434e-02 3.99801891127e-02 3.13851597832e-02 - 2.43885514961e-02 1.90731200357e-02 1.51716517813e-02 1.24193280193e-02 1.06646265060e-02 9.81666041939e-03 - 9.81666041939e-03 1.06646265060e-02 1.24193280193e-02 1.51716517813e-02 1.90731200357e-02 2.43885514961e-02 - 3.34978085182e-02 4.32645707778e-02 5.37274523246e-02 6.32743536159e-02 7.00265784056e-02 7.24709839797e-02 - 7.00265784056e-02 6.32743536159e-02 5.37274523246e-02 4.32645707778e-02 3.34978085182e-02 2.53584894772e-02 - 1.90731200357e-02 1.44498895771e-02 1.12010383875e-02 9.09041352545e-03 7.92776657338e-03 7.56111056333e-03 - 7.92776657338e-03 9.09041352545e-03 1.12010383875e-02 1.44498895771e-02 1.90731200357e-02 2.53584894772e-02 - 3.73919347545e-02 4.80826222932e-02 5.89796865623e-02 6.81380649556e-02 7.34279203403e-02 7.34279203403e-02 - 6.81380649556e-02 5.89796865623e-02 4.80826222932e-02 3.73919347545e-02 2.81262227272e-02 2.07351318935e-02 - 1.51716517813e-02 1.12010383875e-02 8.56724324800e-03 7.02039368822e-03 6.32497949646e-03 6.32497949646e-03 - 7.02039368822e-03 8.56724324800e-03 1.12010383875e-02 1.51716517813e-02 2.07351318935e-02 2.81262227272e-02 - 4.30069852462e-02 5.45216826827e-02 6.54257706676e-02 7.34093289667e-02 7.63593661757e-02 7.34093289667e-02 - 6.54257706676e-02 5.45216826827e-02 4.30069852462e-02 3.25477771477e-02 2.39287446557e-02 1.72789698058e-02 - 1.24193280193e-02 9.09041352545e-03 7.02039368822e-03 5.93416429324e-03 5.60286211346e-03 5.93416429324e-03 - 7.02039368822e-03 9.09041352545e-03 1.24193280193e-02 1.72789698058e-02 2.39287446557e-02 3.25477771477e-02 - 5.07662838782e-02 6.30211041418e-02 7.34279203403e-02 7.94813267109e-02 7.94813267109e-02 7.34279203403e-02 - 6.30211041418e-02 5.07662838782e-02 3.88905177841e-02 2.87349610469e-02 2.07503567214e-02 1.48305747723e-02 - 1.06646265060e-02 7.92776657338e-03 6.32497949646e-03 5.60286211346e-03 5.60286211346e-03 6.32497949646e-03 - 7.92776657338e-03 1.06646265060e-02 1.48305747723e-02 2.07503567214e-02 2.87349610469e-02 3.88905177841e-02 - 6.04031976834e-02 7.24709839797e-02 8.11535443280e-02 8.43291015351e-02 8.11535443280e-02 7.24709839797e-02 - 6.04031976834e-02 4.74317203940e-02 3.55533814714e-02 2.58559673304e-02 1.85584096087e-02 1.33587578257e-02 - 9.81666041939e-03 7.56111056333e-03 6.32497949646e-03 5.93416429324e-03 6.32497949646e-03 7.56111056333e-03 - 9.81666041939e-03 1.33587578257e-02 1.85584096087e-02 2.58559673304e-02 3.55533814714e-02 4.74317203940e-02 - 6.87390140452e-02 7.81603362307e-02 8.33624452951e-02 8.33624452951e-02 7.81603362307e-02 6.87390140452e-02 - 5.67802908281e-02 4.42738472380e-02 3.30014183614e-02 2.40042344862e-02 1.74298422700e-02 1.28681196083e-02 - 9.81666041939e-03 7.92776657338e-03 7.02039368822e-03 7.02039368822e-03 7.92776657338e-03 9.81666041939e-03 - 1.28681196083e-02 1.74298422700e-02 2.40042344862e-02 3.30014183614e-02 4.42738472380e-02 5.67802908281e-02 - 6.91691506706e-02 7.32259633424e-02 7.45491088161e-02 7.32259633424e-02 6.91691506706e-02 6.22085036765e-02 - 5.26323100532e-02 4.17752511891e-02 3.15640784817e-02 2.33592944544e-02 1.74298422700e-02 1.33587578257e-02 - 1.06646265060e-02 9.09041352545e-03 8.56724324800e-03 9.09041352545e-03 1.06646265060e-02 1.33587578257e-02 - 1.74298422700e-02 2.33592944544e-02 3.15640784817e-02 4.17752511891e-02 5.26323100532e-02 6.22085036765e-02 - 5.65857611912e-02 5.54677921285e-02 5.54677921285e-02 5.65857611912e-02 5.74779615393e-02 5.57017474383e-02 - 4.97519001270e-02 4.08013474350e-02 3.15640784817e-02 2.40042344862e-02 1.85584096087e-02 1.48305747723e-02 - 1.24193280193e-02 1.12010383875e-02 1.12010383875e-02 1.24193280193e-02 1.48305747723e-02 1.85584096087e-02 - 2.40042344862e-02 3.15640784817e-02 4.08013474350e-02 4.97519001270e-02 5.57017474383e-02 5.74779615393e-02 - 3.47728130512e-02 3.23216192243e-02 3.47728130512e-02 4.14983823737e-02 4.93152444260e-02 5.29690593685e-02 - 4.97519001270e-02 4.17752511891e-02 3.30014183614e-02 2.58559673304e-02 2.07503567214e-02 1.72789698058e-02 - 1.51716517813e-02 1.44498895771e-02 1.51716517813e-02 1.72789698058e-02 2.07503567214e-02 2.58559673304e-02 - 3.30014183614e-02 4.17752511891e-02 4.97519001270e-02 5.29690593685e-02 4.93152444260e-02 4.14983823737e-02 - 1.49395756337e-02 1.49395756337e-02 2.21086562145e-02 3.55340946595e-02 4.93152444260e-02 5.57017474383e-02 - 5.26323100532e-02 4.42738472380e-02 3.55533814714e-02 2.87349610469e-02 2.39287446557e-02 2.07351318935e-02 - 1.90731200357e-02 1.90731200357e-02 2.07351318935e-02 2.39287446557e-02 2.87349610469e-02 3.55533814714e-02 - 4.42738472380e-02 5.26323100532e-02 5.57017474383e-02 4.93152444260e-02 3.55340946595e-02 2.21086562145e-02 - 4.78306293438e-03 8.92458323932e-03 2.21086562145e-02 4.14983823737e-02 5.74779615393e-02 6.22085036765e-02 - 5.67802908281e-02 4.74317203940e-02 3.88905177841e-02 3.25477771477e-02 2.81262227272e-02 2.53584894772e-02 - 2.43885514961e-02 2.53584894772e-02 2.81262227272e-02 3.25477771477e-02 3.88905177841e-02 4.74317203940e-02 - 5.67802908281e-02 6.22085036765e-02 5.74779615393e-02 4.14983823737e-02 2.21086562145e-02 8.92458323932e-03 - 1.79849681520e-02 3.47728130512e-02 5.63481382706e-02 7.22242608078e-02 7.59717702126e-02 7.00265784056e-02 - 6.08204348290e-02 5.26365510379e-02 4.64352697346e-02 4.17937584090e-02 3.86408229568e-02 3.74820931920e-02 - 3.86408229568e-02 4.17937584090e-02 4.64352697346e-02 5.26365510379e-02 6.08204348290e-02 7.00265784056e-02 - 7.59717702126e-02 7.22242608078e-02 5.63481382706e-02 3.47728130512e-02 1.79849681520e-02 1.20127916963e-02 - 3.47728130512e-02 5.54677921285e-02 7.45491088161e-02 8.33624452951e-02 8.11535443280e-02 7.34279203403e-02 - 6.54257706676e-02 5.89796865623e-02 5.37274523246e-02 4.94024317434e-02 4.67604720778e-02 4.67604720778e-02 - 4.94024317434e-02 5.37274523246e-02 5.89796865623e-02 6.54257706676e-02 7.34279203403e-02 8.11535443280e-02 - 8.33624452951e-02 7.45491088161e-02 5.54677921285e-02 3.47728130512e-02 2.21086562145e-02 2.21086562145e-02 - 5.63481382706e-02 7.45491088161e-02 8.59799490006e-02 8.75278859551e-02 8.26693353815e-02 7.63593661757e-02 - 7.08330821436e-02 6.56955874028e-02 6.03520680545e-02 5.57693346902e-02 5.39094660610e-02 5.57693346902e-02 - 6.03520680545e-02 6.56955874028e-02 7.08330821436e-02 7.63593661757e-02 8.26693353815e-02 8.75278859551e-02 - 8.59799490006e-02 7.45491088161e-02 5.63481382706e-02 4.01124125220e-02 3.37527793613e-02 4.01124125220e-02 - 7.22242608078e-02 8.33624452951e-02 8.75278859551e-02 8.60847359405e-02 8.27498054349e-02 7.94813267109e-02 - 7.55510577091e-02 6.97193973966e-02 6.28462967289e-02 5.80220562298e-02 5.80220562298e-02 6.28462967289e-02 - 6.97193973966e-02 7.55510577091e-02 7.94813267109e-02 8.27498054349e-02 8.60847359405e-02 8.75278859551e-02 - 8.33624452951e-02 7.22242608078e-02 5.81927448576e-02 4.85514460951e-02 4.85514460951e-02 5.81927448576e-02 - 7.59717702126e-02 8.11535443280e-02 8.26693353815e-02 8.27498054349e-02 8.26693353815e-02 8.11535443280e-02 - 7.59717702126e-02 6.72242960853e-02 5.86480861577e-02 5.50728972421e-02 5.86480861577e-02 6.72242960853e-02 - 7.59717702126e-02 8.11535443280e-02 8.26693353815e-02 8.27498054349e-02 8.26693353815e-02 8.11535443280e-02 - 7.59717702126e-02 6.72242960853e-02 5.86480861577e-02 5.50728972421e-02 5.86480861577e-02 6.72242960853e-02 - 7.00265784056e-02 7.34279203403e-02 7.63593661757e-02 7.94813267109e-02 8.11535443280e-02 7.81603362307e-02 - 6.91691506706e-02 5.74779615393e-02 4.93152444260e-02 4.93152444260e-02 5.74779615393e-02 6.91691506706e-02 - 7.81603362307e-02 8.11535443280e-02 7.94813267109e-02 7.63593661757e-02 7.34279203403e-02 7.00265784056e-02 - 6.50724984809e-02 5.92671896593e-02 5.51986766687e-02 5.51986766687e-02 5.92671896593e-02 6.50724984809e-02 - 6.08204348290e-02 6.54257706676e-02 7.08330821436e-02 7.55510577091e-02 7.59717702126e-02 6.91691506706e-02 - 5.67646475266e-02 4.49731492966e-02 4.01965567029e-02 4.49731492966e-02 5.67646475266e-02 6.91691506706e-02 - 7.59717702126e-02 7.55510577091e-02 7.08330821436e-02 6.54257706676e-02 6.08204348290e-02 5.66502458441e-02 - 5.24827367699e-02 4.90171582760e-02 4.76311652618e-02 4.90171582760e-02 5.24827367699e-02 5.66502458441e-02 - 5.26365510379e-02 5.89796865623e-02 6.56955874028e-02 6.97193973966e-02 6.72242960853e-02 5.74779615393e-02 - 4.49731492966e-02 3.64979381134e-02 3.64979381134e-02 4.49731492966e-02 5.74779615393e-02 6.72242960853e-02 - 6.97193973966e-02 6.56955874028e-02 5.89796865623e-02 5.26365510379e-02 4.75580604876e-02 4.34476578087e-02 - 4.01880472274e-02 3.82730721923e-02 3.82730721923e-02 4.01880472274e-02 4.34476578087e-02 4.75580604876e-02 - 4.64352697346e-02 5.37274523246e-02 6.03520680545e-02 6.28462967289e-02 5.86480861577e-02 4.93152444260e-02 - 4.01965567029e-02 3.64979381134e-02 4.01965567029e-02 4.93152444260e-02 5.86480861577e-02 6.28462967289e-02 - 6.03520680545e-02 5.37274523246e-02 4.64352697346e-02 4.03211904768e-02 3.56148556138e-02 3.20717125323e-02 - 2.97324869747e-02 2.88969153857e-02 2.97324869747e-02 3.20717125323e-02 3.56148556138e-02 4.03211904768e-02 - 4.17937584090e-02 4.94024317434e-02 5.57693346902e-02 5.80220562298e-02 5.50728972421e-02 4.93152444260e-02 - 4.49731492966e-02 4.49731492966e-02 4.93152444260e-02 5.50728972421e-02 5.80220562298e-02 5.57693346902e-02 - 4.94024317434e-02 4.17937584090e-02 3.50622322206e-02 2.98227515673e-02 2.59124227944e-02 2.31896632144e-02 - 2.17534458677e-02 2.17534458677e-02 2.31896632144e-02 2.59124227944e-02 2.98227515673e-02 3.50622322206e-02 - 3.86408229568e-02 4.67604720778e-02 5.39094660610e-02 5.80220562298e-02 5.86480861577e-02 5.74779615393e-02 - 5.67646475266e-02 5.74779615393e-02 5.86480861577e-02 5.80220562298e-02 5.39094660610e-02 4.67604720778e-02 - 3.86408229568e-02 3.14092023629e-02 2.57560150434e-02 2.15543530804e-02 1.85555754828e-02 1.67052485052e-02 - 1.60740086611e-02 1.67052485052e-02 1.85555754828e-02 2.15543530804e-02 2.57560150434e-02 3.14092023629e-02 - 3.74820931920e-02 4.67604720778e-02 5.57693346902e-02 6.28462967289e-02 6.72242960853e-02 6.91691506706e-02 - 6.91691506706e-02 6.72242960853e-02 6.28462967289e-02 5.57693346902e-02 4.67604720778e-02 3.74820931920e-02 - 2.94353177762e-02 2.32096903098e-02 1.86453765526e-02 1.53937416744e-02 1.32551887233e-02 1.21839358632e-02 - 1.21839358632e-02 1.32551887233e-02 1.53937416744e-02 1.86453765526e-02 2.32096903098e-02 2.94353177762e-02 - 3.86408229568e-02 4.94024317434e-02 6.03520680545e-02 6.97193973966e-02 7.59717702126e-02 7.81603362307e-02 - 7.59717702126e-02 6.97193973966e-02 6.03520680545e-02 4.94024317434e-02 3.86408229568e-02 2.94353177762e-02 - 2.23208692653e-02 1.71372885777e-02 1.34835937042e-02 1.10438280884e-02 9.64234738867e-03 9.18607976908e-03 - 9.64234738867e-03 1.10438280884e-02 1.34835937042e-02 1.71372885777e-02 2.23208692653e-02 2.94353177762e-02 - 4.17937584090e-02 5.37274523246e-02 6.56955874028e-02 7.55510577091e-02 8.11535443280e-02 8.11535443280e-02 - 7.55510577091e-02 6.56955874028e-02 5.37274523246e-02 4.17937584090e-02 3.14092023629e-02 2.32096903098e-02 - 1.71372885777e-02 1.28345023601e-02 9.94413016678e-03 8.19475376688e-03 7.38199424752e-03 7.38199424752e-03 - 8.19475376688e-03 9.94413016678e-03 1.28345023601e-02 1.71372885777e-02 2.32096903098e-02 3.14092023629e-02 - 4.64352697346e-02 5.89796865623e-02 7.08330821436e-02 7.94813267109e-02 8.26693353815e-02 7.94813267109e-02 - 7.08330821436e-02 5.89796865623e-02 4.64352697346e-02 3.50622322206e-02 2.57560150434e-02 1.86453765526e-02 - 1.34835937042e-02 9.94413016678e-03 7.72360398793e-03 6.54341818442e-03 6.17987618622e-03 6.54341818442e-03 - 7.72360398793e-03 9.94413016678e-03 1.34835937042e-02 1.86453765526e-02 2.57560150434e-02 3.50622322206e-02 - 5.26365510379e-02 6.54257706676e-02 7.63593661757e-02 8.27498054349e-02 8.27498054349e-02 7.63593661757e-02 - 6.54257706676e-02 5.26365510379e-02 4.03211904768e-02 2.98227515673e-02 2.15543530804e-02 1.53937416744e-02 - 1.10438280884e-02 8.19475376688e-03 6.54341818442e-03 5.80894063737e-03 5.80894063737e-03 6.54341818442e-03 - 8.19475376688e-03 1.10438280884e-02 1.53937416744e-02 2.15543530804e-02 2.98227515673e-02 4.03211904768e-02 - 6.08204348290e-02 7.34279203403e-02 8.26693353815e-02 8.60847359405e-02 8.26693353815e-02 7.34279203403e-02 - 6.08204348290e-02 4.75580604876e-02 3.56148556138e-02 2.59124227944e-02 1.85555754828e-02 1.32551887233e-02 - 9.64234738867e-03 7.38199424752e-03 6.17987618622e-03 5.80894063737e-03 6.17987618622e-03 7.38199424752e-03 - 9.64234738867e-03 1.32551887233e-02 1.85555754828e-02 2.59124227944e-02 3.56148556138e-02 4.75580604876e-02 - 7.00265784056e-02 8.11535443280e-02 8.75278859551e-02 8.75278859551e-02 8.11535443280e-02 7.00265784056e-02 - 5.66502458441e-02 4.34476578087e-02 3.20717125323e-02 2.31896632144e-02 1.67052485052e-02 1.21839358632e-02 - 9.18607976908e-03 7.38199424752e-03 6.54341818442e-03 6.54341818442e-03 7.38199424752e-03 9.18607976908e-03 - 1.21839358632e-02 1.67052485052e-02 2.31896632144e-02 3.20717125323e-02 4.34476578087e-02 5.66502458441e-02 - 7.59717702126e-02 8.33624452951e-02 8.59799490006e-02 8.33624452951e-02 7.59717702126e-02 6.50724984809e-02 - 5.24827367699e-02 4.01880472274e-02 2.97324869747e-02 2.17534458677e-02 1.60740086611e-02 1.21839358632e-02 - 9.64234738867e-03 8.19475376688e-03 7.72360398793e-03 8.19475376688e-03 9.64234738867e-03 1.21839358632e-02 - 1.60740086611e-02 2.17534458677e-02 2.97324869747e-02 4.01880472274e-02 5.24827367699e-02 6.50724984809e-02 - 7.22242608078e-02 7.45491088161e-02 7.45491088161e-02 7.22242608078e-02 6.72242960853e-02 5.92671896593e-02 - 4.90171582760e-02 3.82730721923e-02 2.88969153857e-02 2.17534458677e-02 1.67052485052e-02 1.32551887233e-02 - 1.10438280884e-02 9.94413016678e-03 9.94413016678e-03 1.10438280884e-02 1.32551887233e-02 1.67052485052e-02 - 2.17534458677e-02 2.88969153857e-02 3.82730721923e-02 4.90171582760e-02 5.92671896593e-02 6.72242960853e-02 - 5.63481382706e-02 5.54677921285e-02 5.63481382706e-02 5.81927448576e-02 5.86480861577e-02 5.51986766687e-02 - 4.76311652618e-02 3.82730721923e-02 2.97324869747e-02 2.31896632144e-02 1.85555754828e-02 1.53937416744e-02 - 1.34835937042e-02 1.28345023601e-02 1.34835937042e-02 1.53937416744e-02 1.85555754828e-02 2.31896632144e-02 - 2.97324869747e-02 3.82730721923e-02 4.76311652618e-02 5.51986766687e-02 5.86480861577e-02 5.81927448576e-02 - 3.47728130512e-02 3.47728130512e-02 4.01124125220e-02 4.85514460951e-02 5.50728972421e-02 5.51986766687e-02 - 4.90171582760e-02 4.01880472274e-02 3.20717125323e-02 2.59124227944e-02 2.15543530804e-02 1.86453765526e-02 - 1.71372885777e-02 1.71372885777e-02 1.86453765526e-02 2.15543530804e-02 2.59124227944e-02 3.20717125323e-02 - 4.01880472274e-02 4.90171582760e-02 5.51986766687e-02 5.50728972421e-02 4.85514460951e-02 4.01124125220e-02 - 1.79849681520e-02 2.21086562145e-02 3.37527793613e-02 4.85514460951e-02 5.86480861577e-02 5.92671896593e-02 - 5.24827367699e-02 4.34476578087e-02 3.56148556138e-02 2.98227515673e-02 2.57560150434e-02 2.32096903098e-02 - 2.23208692653e-02 2.32096903098e-02 2.57560150434e-02 2.98227515673e-02 3.56148556138e-02 4.34476578087e-02 - 5.24827367699e-02 5.92671896593e-02 5.86480861577e-02 4.85514460951e-02 3.37527793613e-02 2.21086562145e-02 - 1.20127916963e-02 2.21086562145e-02 4.01124125220e-02 5.81927448576e-02 6.72242960853e-02 6.50724984809e-02 - 5.66502458441e-02 4.75580604876e-02 4.03211904768e-02 3.50622322206e-02 3.14092023629e-02 2.94353177762e-02 - 2.94353177762e-02 3.14092023629e-02 3.50622322206e-02 4.03211904768e-02 4.75580604876e-02 5.66502458441e-02 - 6.50724984809e-02 6.72242960853e-02 5.81927448576e-02 4.01124125220e-02 2.21086562145e-02 1.20127916963e-02 - 3.86813628609e-02 5.65857611912e-02 7.22242608078e-02 7.85611057280e-02 7.55510577091e-02 6.81380649556e-02 - 6.07372364008e-02 5.48042821337e-02 4.99875697106e-02 4.60708732894e-02 4.37078082031e-02 4.37078082031e-02 - 4.60708732894e-02 4.99875697106e-02 5.48042821337e-02 6.07372364008e-02 6.81380649556e-02 7.55510577091e-02 - 7.85611057280e-02 7.22242608078e-02 5.65857611912e-02 3.86813628609e-02 2.72980449809e-02 2.72980449809e-02 - 5.65857611912e-02 7.32259633424e-02 8.33624452951e-02 8.43291015351e-02 7.94813267109e-02 7.34093289667e-02 - 6.81380649556e-02 6.32743536159e-02 5.82637609313e-02 5.39959847166e-02 5.22694162731e-02 5.39959847166e-02 - 5.82637609313e-02 6.32743536159e-02 6.81380649556e-02 7.34093289667e-02 7.94813267109e-02 8.43291015351e-02 - 8.33624452951e-02 7.32259633424e-02 5.65857611912e-02 4.14983823737e-02 3.55340946595e-02 4.14983823737e-02 - 7.22242608078e-02 8.33624452951e-02 8.75278859551e-02 8.60847359405e-02 8.27498054349e-02 7.94813267109e-02 - 7.55510577091e-02 6.97193973966e-02 6.28462967289e-02 5.80220562298e-02 5.80220562298e-02 6.28462967289e-02 - 6.97193973966e-02 7.55510577091e-02 7.94813267109e-02 8.27498054349e-02 8.60847359405e-02 8.75278859551e-02 - 8.33624452951e-02 7.22242608078e-02 5.81927448576e-02 4.85514460951e-02 4.85514460951e-02 5.81927448576e-02 - 7.85611057280e-02 8.43291015351e-02 8.60847359405e-02 8.62128629277e-02 8.60847359405e-02 8.43291015351e-02 - 7.85611057280e-02 6.89438173057e-02 5.95705865689e-02 5.56741558656e-02 5.95705865689e-02 6.89438173057e-02 - 7.85611057280e-02 8.43291015351e-02 8.60847359405e-02 8.62128629277e-02 8.60847359405e-02 8.43291015351e-02 - 7.85611057280e-02 6.89438173057e-02 5.95705865689e-02 5.56741558656e-02 5.95705865689e-02 6.89438173057e-02 - 7.55510577091e-02 7.94813267109e-02 8.27498054349e-02 8.60847359405e-02 8.75278859551e-02 8.33624452951e-02 - 7.22242608078e-02 5.81927448576e-02 4.85514460951e-02 4.85514460951e-02 5.81927448576e-02 7.22242608078e-02 - 8.33624452951e-02 8.75278859551e-02 8.60847359405e-02 8.27498054349e-02 7.94813267109e-02 7.55510577091e-02 - 6.97193973966e-02 6.28462967289e-02 5.80220562298e-02 5.80220562298e-02 6.28462967289e-02 6.97193973966e-02 - 6.81380649556e-02 7.34093289667e-02 7.94813267109e-02 8.43291015351e-02 8.33624452951e-02 7.32259633424e-02 - 5.65857611912e-02 4.14983823737e-02 3.55340946595e-02 4.14983823737e-02 5.65857611912e-02 7.32259633424e-02 - 8.33624452951e-02 8.43291015351e-02 7.94813267109e-02 7.34093289667e-02 6.81380649556e-02 6.32743536159e-02 - 5.82637609313e-02 5.39959847166e-02 5.22694162731e-02 5.39959847166e-02 5.82637609313e-02 6.32743536159e-02 - 6.07372364008e-02 6.81380649556e-02 7.55510577091e-02 7.85611057280e-02 7.22242608078e-02 5.65857611912e-02 - 3.86813628609e-02 2.72980449809e-02 2.72980449809e-02 3.86813628609e-02 5.65857611912e-02 7.22242608078e-02 - 7.85611057280e-02 7.55510577091e-02 6.81380649556e-02 6.07372364008e-02 5.48042821337e-02 4.99875697106e-02 - 4.60708732894e-02 4.37078082031e-02 4.37078082031e-02 4.60708732894e-02 4.99875697106e-02 5.48042821337e-02 - 5.48042821337e-02 6.32743536159e-02 6.97193973966e-02 6.89438173057e-02 5.81927448576e-02 4.14983823737e-02 - 2.72980449809e-02 2.19326842858e-02 2.72980449809e-02 4.14983823737e-02 5.81927448576e-02 6.89438173057e-02 - 6.97193973966e-02 6.32743536159e-02 5.48042821337e-02 4.74868469137e-02 4.19230722756e-02 3.77513071960e-02 - 3.49472650912e-02 3.39275143754e-02 3.49472650912e-02 3.77513071960e-02 4.19230722756e-02 4.74868469137e-02 - 4.99875697106e-02 5.82637609313e-02 6.28462967289e-02 5.95705865689e-02 4.85514460951e-02 3.55340946595e-02 - 2.72980449809e-02 2.72980449809e-02 3.55340946595e-02 4.85514460951e-02 5.95705865689e-02 6.28462967289e-02 - 5.82637609313e-02 4.99875697106e-02 4.19230722756e-02 3.55984989831e-02 3.09544991478e-02 2.77206217649e-02 - 2.59885128162e-02 2.59885128162e-02 2.77206217649e-02 3.09544991478e-02 3.55984989831e-02 4.19230722756e-02 - 4.60708732894e-02 5.39959847166e-02 5.80220562298e-02 5.56741558656e-02 4.85514460951e-02 4.14983823737e-02 - 3.86813628609e-02 4.14983823737e-02 4.85514460951e-02 5.56741558656e-02 5.80220562298e-02 5.39959847166e-02 - 4.60708732894e-02 3.77513071960e-02 3.09544991478e-02 2.59309064966e-02 2.23681861845e-02 2.01439779904e-02 - 1.93739445237e-02 2.01439779904e-02 2.23681861845e-02 2.59309064966e-02 3.09544991478e-02 3.77513071960e-02 - 4.37078082031e-02 5.22694162731e-02 5.80220562298e-02 5.95705865689e-02 5.81927448576e-02 5.65857611912e-02 - 5.65857611912e-02 5.81927448576e-02 5.95705865689e-02 5.80220562298e-02 5.22694162731e-02 4.37078082031e-02 - 3.49472650912e-02 2.77206217649e-02 2.23681861845e-02 1.85605049655e-02 1.60207976494e-02 1.47188141408e-02 - 1.47188141408e-02 1.60207976494e-02 1.85605049655e-02 2.23681861845e-02 2.77206217649e-02 3.49472650912e-02 - 4.37078082031e-02 5.39959847166e-02 6.28462967289e-02 6.89438173057e-02 7.22242608078e-02 7.32259633424e-02 - 7.22242608078e-02 6.89438173057e-02 6.28462967289e-02 5.39959847166e-02 4.37078082031e-02 3.39275143754e-02 - 2.59885128162e-02 2.01439779904e-02 1.60207976494e-02 1.32196503024e-02 1.15565994691e-02 1.10006986257e-02 - 1.15565994691e-02 1.32196503024e-02 1.60207976494e-02 2.01439779904e-02 2.59885128162e-02 3.39275143754e-02 - 4.60708732894e-02 5.82637609313e-02 6.97193973966e-02 7.85611057280e-02 8.33624452951e-02 8.33624452951e-02 - 7.85611057280e-02 6.97193973966e-02 5.82637609313e-02 4.60708732894e-02 3.49472650912e-02 2.59885128162e-02 - 1.93739445237e-02 1.47188141408e-02 1.15565994691e-02 9.57706447476e-03 8.62132731210e-03 8.62132731210e-03 - 9.57706447476e-03 1.15565994691e-02 1.47188141408e-02 1.93739445237e-02 2.59885128162e-02 3.49472650912e-02 - 4.99875697106e-02 6.32743536159e-02 7.55510577091e-02 8.43291015351e-02 8.75278859551e-02 8.43291015351e-02 - 7.55510577091e-02 6.32743536159e-02 4.99875697106e-02 3.77513071960e-02 2.77206217649e-02 2.01439779904e-02 - 1.47188141408e-02 1.10006986257e-02 8.62132731210e-03 7.31454489847e-03 6.90130049286e-03 7.31454489847e-03 - 8.62132731210e-03 1.10006986257e-02 1.47188141408e-02 2.01439779904e-02 2.77206217649e-02 3.77513071960e-02 - 5.48042821337e-02 6.81380649556e-02 7.94813267109e-02 8.60847359405e-02 8.60847359405e-02 7.94813267109e-02 - 6.81380649556e-02 5.48042821337e-02 4.19230722756e-02 3.09544991478e-02 2.23681861845e-02 1.60207976494e-02 - 1.15565994691e-02 8.62132731210e-03 6.90130049286e-03 6.12647535151e-03 6.12647535151e-03 6.90130049286e-03 - 8.62132731210e-03 1.15565994691e-02 1.60207976494e-02 2.23681861845e-02 3.09544991478e-02 4.19230722756e-02 - 6.07372364008e-02 7.34093289667e-02 8.27498054349e-02 8.62128629277e-02 8.27498054349e-02 7.34093289667e-02 - 6.07372364008e-02 4.74868469137e-02 3.55984989831e-02 2.59309064966e-02 1.85605049655e-02 1.32196503024e-02 - 9.57706447476e-03 7.31454489847e-03 6.12647535151e-03 5.76375313750e-03 6.12647535151e-03 7.31454489847e-03 - 9.57706447476e-03 1.32196503024e-02 1.85605049655e-02 2.59309064966e-02 3.55984989831e-02 4.74868469137e-02 - 6.81380649556e-02 7.94813267109e-02 8.60847359405e-02 8.60847359405e-02 7.94813267109e-02 6.81380649556e-02 - 5.48042821337e-02 4.19230722756e-02 3.09544991478e-02 2.23681861845e-02 1.60207976494e-02 1.15565994691e-02 - 8.62132731210e-03 6.90130049286e-03 6.12647535151e-03 6.12647535151e-03 6.90130049286e-03 8.62132731210e-03 - 1.15565994691e-02 1.60207976494e-02 2.23681861845e-02 3.09544991478e-02 4.19230722756e-02 5.48042821337e-02 - 7.55510577091e-02 8.43291015351e-02 8.75278859551e-02 8.43291015351e-02 7.55510577091e-02 6.32743536159e-02 - 4.99875697106e-02 3.77513071960e-02 2.77206217649e-02 2.01439779904e-02 1.47188141408e-02 1.10006986257e-02 - 8.62132731210e-03 7.31454489847e-03 6.90130049286e-03 7.31454489847e-03 8.62132731210e-03 1.10006986257e-02 - 1.47188141408e-02 2.01439779904e-02 2.77206217649e-02 3.77513071960e-02 4.99875697106e-02 6.32743536159e-02 - 7.85611057280e-02 8.33624452951e-02 8.33624452951e-02 7.85611057280e-02 6.97193973966e-02 5.82637609313e-02 - 4.60708732894e-02 3.49472650912e-02 2.59885128162e-02 1.93739445237e-02 1.47188141408e-02 1.15565994691e-02 - 9.57706447476e-03 8.62132731210e-03 8.62132731210e-03 9.57706447476e-03 1.15565994691e-02 1.47188141408e-02 - 1.93739445237e-02 2.59885128162e-02 3.49472650912e-02 4.60708732894e-02 5.82637609313e-02 6.97193973966e-02 - 7.22242608078e-02 7.32259633424e-02 7.22242608078e-02 6.89438173057e-02 6.28462967289e-02 5.39959847166e-02 - 4.37078082031e-02 3.39275143754e-02 2.59885128162e-02 2.01439779904e-02 1.60207976494e-02 1.32196503024e-02 - 1.15565994691e-02 1.10006986257e-02 1.15565994691e-02 1.32196503024e-02 1.60207976494e-02 2.01439779904e-02 - 2.59885128162e-02 3.39275143754e-02 4.37078082031e-02 5.39959847166e-02 6.28462967289e-02 6.89438173057e-02 - 5.65857611912e-02 5.65857611912e-02 5.81927448576e-02 5.95705865689e-02 5.80220562298e-02 5.22694162731e-02 - 4.37078082031e-02 3.49472650912e-02 2.77206217649e-02 2.23681861845e-02 1.85605049655e-02 1.60207976494e-02 - 1.47188141408e-02 1.47188141408e-02 1.60207976494e-02 1.85605049655e-02 2.23681861845e-02 2.77206217649e-02 - 3.49472650912e-02 4.37078082031e-02 5.22694162731e-02 5.80220562298e-02 5.95705865689e-02 5.81927448576e-02 - 3.86813628609e-02 4.14983823737e-02 4.85514460951e-02 5.56741558656e-02 5.80220562298e-02 5.39959847166e-02 - 4.60708732894e-02 3.77513071960e-02 3.09544991478e-02 2.59309064966e-02 2.23681861845e-02 2.01439779904e-02 - 1.93739445237e-02 2.01439779904e-02 2.23681861845e-02 2.59309064966e-02 3.09544991478e-02 3.77513071960e-02 - 4.60708732894e-02 5.39959847166e-02 5.80220562298e-02 5.56741558656e-02 4.85514460951e-02 4.14983823737e-02 - 2.72980449809e-02 3.55340946595e-02 4.85514460951e-02 5.95705865689e-02 6.28462967289e-02 5.82637609313e-02 - 4.99875697106e-02 4.19230722756e-02 3.55984989831e-02 3.09544991478e-02 2.77206217649e-02 2.59885128162e-02 - 2.59885128162e-02 2.77206217649e-02 3.09544991478e-02 3.55984989831e-02 4.19230722756e-02 4.99875697106e-02 - 5.82637609313e-02 6.28462967289e-02 5.95705865689e-02 4.85514460951e-02 3.55340946595e-02 2.72980449809e-02 - 2.72980449809e-02 4.14983823737e-02 5.81927448576e-02 6.89438173057e-02 6.97193973966e-02 6.32743536159e-02 - 5.48042821337e-02 4.74868469137e-02 4.19230722756e-02 3.77513071960e-02 3.49472650912e-02 3.39275143754e-02 - 3.49472650912e-02 3.77513071960e-02 4.19230722756e-02 4.74868469137e-02 5.48042821337e-02 6.32743536159e-02 - 6.97193973966e-02 6.89438173057e-02 5.81927448576e-02 4.14983823737e-02 2.72980449809e-02 2.19326842858e-02 - 5.67646475266e-02 6.91691506706e-02 7.59717702126e-02 7.55510577091e-02 7.08330821436e-02 6.54257706676e-02 - 6.08204348290e-02 5.66502458441e-02 5.24827367699e-02 4.90171582760e-02 4.76311652618e-02 4.90171582760e-02 - 5.24827367699e-02 5.66502458441e-02 6.08204348290e-02 6.54257706676e-02 7.08330821436e-02 7.55510577091e-02 - 7.59717702126e-02 6.91691506706e-02 5.67646475266e-02 4.49731492966e-02 4.01965567029e-02 4.49731492966e-02 - 6.91691506706e-02 7.81603362307e-02 8.11535443280e-02 7.94813267109e-02 7.63593661757e-02 7.34279203403e-02 - 7.00265784056e-02 6.50724984809e-02 5.92671896593e-02 5.51986766687e-02 5.51986766687e-02 5.92671896593e-02 - 6.50724984809e-02 7.00265784056e-02 7.34279203403e-02 7.63593661757e-02 7.94813267109e-02 8.11535443280e-02 - 7.81603362307e-02 6.91691506706e-02 5.74779615393e-02 4.93152444260e-02 4.93152444260e-02 5.74779615393e-02 - 7.59717702126e-02 8.11535443280e-02 8.26693353815e-02 8.27498054349e-02 8.26693353815e-02 8.11535443280e-02 - 7.59717702126e-02 6.72242960853e-02 5.86480861577e-02 5.50728972421e-02 5.86480861577e-02 6.72242960853e-02 - 7.59717702126e-02 8.11535443280e-02 8.26693353815e-02 8.27498054349e-02 8.26693353815e-02 8.11535443280e-02 - 7.59717702126e-02 6.72242960853e-02 5.86480861577e-02 5.50728972421e-02 5.86480861577e-02 6.72242960853e-02 - 7.55510577091e-02 7.94813267109e-02 8.27498054349e-02 8.60847359405e-02 8.75278859551e-02 8.33624452951e-02 - 7.22242608078e-02 5.81927448576e-02 4.85514460951e-02 4.85514460951e-02 5.81927448576e-02 7.22242608078e-02 - 8.33624452951e-02 8.75278859551e-02 8.60847359405e-02 8.27498054349e-02 7.94813267109e-02 7.55510577091e-02 - 6.97193973966e-02 6.28462967289e-02 5.80220562298e-02 5.80220562298e-02 6.28462967289e-02 6.97193973966e-02 - 7.08330821436e-02 7.63593661757e-02 8.26693353815e-02 8.75278859551e-02 8.59799490006e-02 7.45491088161e-02 - 5.63481382706e-02 4.01124125220e-02 3.37527793613e-02 4.01124125220e-02 5.63481382706e-02 7.45491088161e-02 - 8.59799490006e-02 8.75278859551e-02 8.26693353815e-02 7.63593661757e-02 7.08330821436e-02 6.56955874028e-02 - 6.03520680545e-02 5.57693346902e-02 5.39094660610e-02 5.57693346902e-02 6.03520680545e-02 6.56955874028e-02 - 6.54257706676e-02 7.34279203403e-02 8.11535443280e-02 8.33624452951e-02 7.45491088161e-02 5.54677921285e-02 - 3.47728130512e-02 2.21086562145e-02 2.21086562145e-02 3.47728130512e-02 5.54677921285e-02 7.45491088161e-02 - 8.33624452951e-02 8.11535443280e-02 7.34279203403e-02 6.54257706676e-02 5.89796865623e-02 5.37274523246e-02 - 4.94024317434e-02 4.67604720778e-02 4.67604720778e-02 4.94024317434e-02 5.37274523246e-02 5.89796865623e-02 - 6.08204348290e-02 7.00265784056e-02 7.59717702126e-02 7.22242608078e-02 5.63481382706e-02 3.47728130512e-02 - 1.79849681520e-02 1.20127916963e-02 1.79849681520e-02 3.47728130512e-02 5.63481382706e-02 7.22242608078e-02 - 7.59717702126e-02 7.00265784056e-02 6.08204348290e-02 5.26365510379e-02 4.64352697346e-02 4.17937584090e-02 - 3.86408229568e-02 3.74820931920e-02 3.86408229568e-02 4.17937584090e-02 4.64352697346e-02 5.26365510379e-02 - 5.66502458441e-02 6.50724984809e-02 6.72242960853e-02 5.81927448576e-02 4.01124125220e-02 2.21086562145e-02 - 1.20127916963e-02 1.20127916963e-02 2.21086562145e-02 4.01124125220e-02 5.81927448576e-02 6.72242960853e-02 - 6.50724984809e-02 5.66502458441e-02 4.75580604876e-02 4.03211904768e-02 3.50622322206e-02 3.14092023629e-02 - 2.94353177762e-02 2.94353177762e-02 3.14092023629e-02 3.50622322206e-02 4.03211904768e-02 4.75580604876e-02 - 5.24827367699e-02 5.92671896593e-02 5.86480861577e-02 4.85514460951e-02 3.37527793613e-02 2.21086562145e-02 - 1.79849681520e-02 2.21086562145e-02 3.37527793613e-02 4.85514460951e-02 5.86480861577e-02 5.92671896593e-02 - 5.24827367699e-02 4.34476578087e-02 3.56148556138e-02 2.98227515673e-02 2.57560150434e-02 2.32096903098e-02 - 2.23208692653e-02 2.32096903098e-02 2.57560150434e-02 2.98227515673e-02 3.56148556138e-02 4.34476578087e-02 - 4.90171582760e-02 5.51986766687e-02 5.50728972421e-02 4.85514460951e-02 4.01124125220e-02 3.47728130512e-02 - 3.47728130512e-02 4.01124125220e-02 4.85514460951e-02 5.50728972421e-02 5.51986766687e-02 4.90171582760e-02 - 4.01880472274e-02 3.20717125323e-02 2.59124227944e-02 2.15543530804e-02 1.86453765526e-02 1.71372885777e-02 - 1.71372885777e-02 1.86453765526e-02 2.15543530804e-02 2.59124227944e-02 3.20717125323e-02 4.01880472274e-02 - 4.76311652618e-02 5.51986766687e-02 5.86480861577e-02 5.81927448576e-02 5.63481382706e-02 5.54677921285e-02 - 5.63481382706e-02 5.81927448576e-02 5.86480861577e-02 5.51986766687e-02 4.76311652618e-02 3.82730721923e-02 - 2.97324869747e-02 2.31896632144e-02 1.85555754828e-02 1.53937416744e-02 1.34835937042e-02 1.28345023601e-02 - 1.34835937042e-02 1.53937416744e-02 1.85555754828e-02 2.31896632144e-02 2.97324869747e-02 3.82730721923e-02 - 4.90171582760e-02 5.92671896593e-02 6.72242960853e-02 7.22242608078e-02 7.45491088161e-02 7.45491088161e-02 - 7.22242608078e-02 6.72242960853e-02 5.92671896593e-02 4.90171582760e-02 3.82730721923e-02 2.88969153857e-02 - 2.17534458677e-02 1.67052485052e-02 1.32551887233e-02 1.10438280884e-02 9.94413016678e-03 9.94413016678e-03 - 1.10438280884e-02 1.32551887233e-02 1.67052485052e-02 2.17534458677e-02 2.88969153857e-02 3.82730721923e-02 - 5.24827367699e-02 6.50724984809e-02 7.59717702126e-02 8.33624452951e-02 8.59799490006e-02 8.33624452951e-02 - 7.59717702126e-02 6.50724984809e-02 5.24827367699e-02 4.01880472274e-02 2.97324869747e-02 2.17534458677e-02 - 1.60740086611e-02 1.21839358632e-02 9.64234738867e-03 8.19475376688e-03 7.72360398793e-03 8.19475376688e-03 - 9.64234738867e-03 1.21839358632e-02 1.60740086611e-02 2.17534458677e-02 2.97324869747e-02 4.01880472274e-02 - 5.66502458441e-02 7.00265784056e-02 8.11535443280e-02 8.75278859551e-02 8.75278859551e-02 8.11535443280e-02 - 7.00265784056e-02 5.66502458441e-02 4.34476578087e-02 3.20717125323e-02 2.31896632144e-02 1.67052485052e-02 - 1.21839358632e-02 9.18607976908e-03 7.38199424752e-03 6.54341818442e-03 6.54341818442e-03 7.38199424752e-03 - 9.18607976908e-03 1.21839358632e-02 1.67052485052e-02 2.31896632144e-02 3.20717125323e-02 4.34476578087e-02 - 6.08204348290e-02 7.34279203403e-02 8.26693353815e-02 8.60847359405e-02 8.26693353815e-02 7.34279203403e-02 - 6.08204348290e-02 4.75580604876e-02 3.56148556138e-02 2.59124227944e-02 1.85555754828e-02 1.32551887233e-02 - 9.64234738867e-03 7.38199424752e-03 6.17987618622e-03 5.80894063737e-03 6.17987618622e-03 7.38199424752e-03 - 9.64234738867e-03 1.32551887233e-02 1.85555754828e-02 2.59124227944e-02 3.56148556138e-02 4.75580604876e-02 - 6.54257706676e-02 7.63593661757e-02 8.27498054349e-02 8.27498054349e-02 7.63593661757e-02 6.54257706676e-02 - 5.26365510379e-02 4.03211904768e-02 2.98227515673e-02 2.15543530804e-02 1.53937416744e-02 1.10438280884e-02 - 8.19475376688e-03 6.54341818442e-03 5.80894063737e-03 5.80894063737e-03 6.54341818442e-03 8.19475376688e-03 - 1.10438280884e-02 1.53937416744e-02 2.15543530804e-02 2.98227515673e-02 4.03211904768e-02 5.26365510379e-02 - 7.08330821436e-02 7.94813267109e-02 8.26693353815e-02 7.94813267109e-02 7.08330821436e-02 5.89796865623e-02 - 4.64352697346e-02 3.50622322206e-02 2.57560150434e-02 1.86453765526e-02 1.34835937042e-02 9.94413016678e-03 - 7.72360398793e-03 6.54341818442e-03 6.17987618622e-03 6.54341818442e-03 7.72360398793e-03 9.94413016678e-03 - 1.34835937042e-02 1.86453765526e-02 2.57560150434e-02 3.50622322206e-02 4.64352697346e-02 5.89796865623e-02 - 7.55510577091e-02 8.11535443280e-02 8.11535443280e-02 7.55510577091e-02 6.56955874028e-02 5.37274523246e-02 - 4.17937584090e-02 3.14092023629e-02 2.32096903098e-02 1.71372885777e-02 1.28345023601e-02 9.94413016678e-03 - 8.19475376688e-03 7.38199424752e-03 7.38199424752e-03 8.19475376688e-03 9.94413016678e-03 1.28345023601e-02 - 1.71372885777e-02 2.32096903098e-02 3.14092023629e-02 4.17937584090e-02 5.37274523246e-02 6.56955874028e-02 - 7.59717702126e-02 7.81603362307e-02 7.59717702126e-02 6.97193973966e-02 6.03520680545e-02 4.94024317434e-02 - 3.86408229568e-02 2.94353177762e-02 2.23208692653e-02 1.71372885777e-02 1.34835937042e-02 1.10438280884e-02 - 9.64234738867e-03 9.18607976908e-03 9.64234738867e-03 1.10438280884e-02 1.34835937042e-02 1.71372885777e-02 - 2.23208692653e-02 2.94353177762e-02 3.86408229568e-02 4.94024317434e-02 6.03520680545e-02 6.97193973966e-02 - 6.91691506706e-02 6.91691506706e-02 6.72242960853e-02 6.28462967289e-02 5.57693346902e-02 4.67604720778e-02 - 3.74820931920e-02 2.94353177762e-02 2.32096903098e-02 1.86453765526e-02 1.53937416744e-02 1.32551887233e-02 - 1.21839358632e-02 1.21839358632e-02 1.32551887233e-02 1.53937416744e-02 1.86453765526e-02 2.32096903098e-02 - 2.94353177762e-02 3.74820931920e-02 4.67604720778e-02 5.57693346902e-02 6.28462967289e-02 6.72242960853e-02 - 5.67646475266e-02 5.74779615393e-02 5.86480861577e-02 5.80220562298e-02 5.39094660610e-02 4.67604720778e-02 - 3.86408229568e-02 3.14092023629e-02 2.57560150434e-02 2.15543530804e-02 1.85555754828e-02 1.67052485052e-02 - 1.60740086611e-02 1.67052485052e-02 1.85555754828e-02 2.15543530804e-02 2.57560150434e-02 3.14092023629e-02 - 3.86408229568e-02 4.67604720778e-02 5.39094660610e-02 5.80220562298e-02 5.86480861577e-02 5.74779615393e-02 - 4.49731492966e-02 4.93152444260e-02 5.50728972421e-02 5.80220562298e-02 5.57693346902e-02 4.94024317434e-02 - 4.17937584090e-02 3.50622322206e-02 2.98227515673e-02 2.59124227944e-02 2.31896632144e-02 2.17534458677e-02 - 2.17534458677e-02 2.31896632144e-02 2.59124227944e-02 2.98227515673e-02 3.50622322206e-02 4.17937584090e-02 - 4.94024317434e-02 5.57693346902e-02 5.80220562298e-02 5.50728972421e-02 4.93152444260e-02 4.49731492966e-02 - 4.01965567029e-02 4.93152444260e-02 5.86480861577e-02 6.28462967289e-02 6.03520680545e-02 5.37274523246e-02 - 4.64352697346e-02 4.03211904768e-02 3.56148556138e-02 3.20717125323e-02 2.97324869747e-02 2.88969153857e-02 - 2.97324869747e-02 3.20717125323e-02 3.56148556138e-02 4.03211904768e-02 4.64352697346e-02 5.37274523246e-02 - 6.03520680545e-02 6.28462967289e-02 5.86480861577e-02 4.93152444260e-02 4.01965567029e-02 3.64979381134e-02 - 4.49731492966e-02 5.74779615393e-02 6.72242960853e-02 6.97193973966e-02 6.56955874028e-02 5.89796865623e-02 - 5.26365510379e-02 4.75580604876e-02 4.34476578087e-02 4.01880472274e-02 3.82730721923e-02 3.82730721923e-02 - 4.01880472274e-02 4.34476578087e-02 4.75580604876e-02 5.26365510379e-02 5.89796865623e-02 6.56955874028e-02 - 6.97193973966e-02 6.72242960853e-02 5.74779615393e-02 4.49731492966e-02 3.64979381134e-02 3.64979381134e-02 - 6.31142666785e-02 6.87390140452e-02 7.00265784056e-02 6.81380649556e-02 6.54257706676e-02 6.30211041418e-02 - 6.04031976834e-02 5.67802908281e-02 5.26323100532e-02 4.97519001270e-02 4.97519001270e-02 5.26323100532e-02 - 5.67802908281e-02 6.04031976834e-02 6.30211041418e-02 6.54257706676e-02 6.81380649556e-02 7.00265784056e-02 - 6.87390140452e-02 6.31142666785e-02 5.52660749302e-02 4.96150823327e-02 4.96150823327e-02 5.52660749302e-02 - 6.87390140452e-02 7.24709839797e-02 7.34279203403e-02 7.34093289667e-02 7.34279203403e-02 7.24709839797e-02 - 6.87390140452e-02 6.22085036765e-02 5.57017474383e-02 5.29690593685e-02 5.57017474383e-02 6.22085036765e-02 - 6.87390140452e-02 7.24709839797e-02 7.34279203403e-02 7.34093289667e-02 7.34279203403e-02 7.24709839797e-02 - 6.87390140452e-02 6.22085036765e-02 5.57017474383e-02 5.29690593685e-02 5.57017474383e-02 6.22085036765e-02 - 7.00265784056e-02 7.34279203403e-02 7.63593661757e-02 7.94813267109e-02 8.11535443280e-02 7.81603362307e-02 - 6.91691506706e-02 5.74779615393e-02 4.93152444260e-02 4.93152444260e-02 5.74779615393e-02 6.91691506706e-02 - 7.81603362307e-02 8.11535443280e-02 7.94813267109e-02 7.63593661757e-02 7.34279203403e-02 7.00265784056e-02 - 6.50724984809e-02 5.92671896593e-02 5.51986766687e-02 5.51986766687e-02 5.92671896593e-02 6.50724984809e-02 - 6.81380649556e-02 7.34093289667e-02 7.94813267109e-02 8.43291015351e-02 8.33624452951e-02 7.32259633424e-02 - 5.65857611912e-02 4.14983823737e-02 3.55340946595e-02 4.14983823737e-02 5.65857611912e-02 7.32259633424e-02 - 8.33624452951e-02 8.43291015351e-02 7.94813267109e-02 7.34093289667e-02 6.81380649556e-02 6.32743536159e-02 - 5.82637609313e-02 5.39959847166e-02 5.22694162731e-02 5.39959847166e-02 5.82637609313e-02 6.32743536159e-02 - 6.54257706676e-02 7.34279203403e-02 8.11535443280e-02 8.33624452951e-02 7.45491088161e-02 5.54677921285e-02 - 3.47728130512e-02 2.21086562145e-02 2.21086562145e-02 3.47728130512e-02 5.54677921285e-02 7.45491088161e-02 - 8.33624452951e-02 8.11535443280e-02 7.34279203403e-02 6.54257706676e-02 5.89796865623e-02 5.37274523246e-02 - 4.94024317434e-02 4.67604720778e-02 4.67604720778e-02 4.94024317434e-02 5.37274523246e-02 5.89796865623e-02 - 6.30211041418e-02 7.24709839797e-02 7.81603362307e-02 7.32259633424e-02 5.54677921285e-02 3.23216192243e-02 - 1.49395756337e-02 8.92458323932e-03 1.49395756337e-02 3.23216192243e-02 5.54677921285e-02 7.32259633424e-02 - 7.81603362307e-02 7.24709839797e-02 6.30211041418e-02 5.45216826827e-02 4.80826222932e-02 4.32645707778e-02 - 3.99801891127e-02 3.87690158276e-02 3.99801891127e-02 4.32645707778e-02 4.80826222932e-02 5.45216826827e-02 - 6.04031976834e-02 6.87390140452e-02 6.91691506706e-02 5.65857611912e-02 3.47728130512e-02 1.49395756337e-02 - 4.78306293438e-03 4.78306293438e-03 1.49395756337e-02 3.47728130512e-02 5.65857611912e-02 6.91691506706e-02 - 6.87390140452e-02 6.04031976834e-02 5.07662838782e-02 4.30069852462e-02 3.73919347545e-02 3.34978085182e-02 - 3.13851597832e-02 3.13851597832e-02 3.34978085182e-02 3.73919347545e-02 4.30069852462e-02 5.07662838782e-02 - 5.67802908281e-02 6.22085036765e-02 5.74779615393e-02 4.14983823737e-02 2.21086562145e-02 8.92458323932e-03 - 4.78306293438e-03 8.92458323932e-03 2.21086562145e-02 4.14983823737e-02 5.74779615393e-02 6.22085036765e-02 - 5.67802908281e-02 4.74317203940e-02 3.88905177841e-02 3.25477771477e-02 2.81262227272e-02 2.53584894772e-02 - 2.43885514961e-02 2.53584894772e-02 2.81262227272e-02 3.25477771477e-02 3.88905177841e-02 4.74317203940e-02 - 5.26323100532e-02 5.57017474383e-02 4.93152444260e-02 3.55340946595e-02 2.21086562145e-02 1.49395756337e-02 - 1.49395756337e-02 2.21086562145e-02 3.55340946595e-02 4.93152444260e-02 5.57017474383e-02 5.26323100532e-02 - 4.42738472380e-02 3.55533814714e-02 2.87349610469e-02 2.39287446557e-02 2.07351318935e-02 1.90731200357e-02 - 1.90731200357e-02 2.07351318935e-02 2.39287446557e-02 2.87349610469e-02 3.55533814714e-02 4.42738472380e-02 - 4.97519001270e-02 5.29690593685e-02 4.93152444260e-02 4.14983823737e-02 3.47728130512e-02 3.23216192243e-02 - 3.47728130512e-02 4.14983823737e-02 4.93152444260e-02 5.29690593685e-02 4.97519001270e-02 4.17752511891e-02 - 3.30014183614e-02 2.58559673304e-02 2.07503567214e-02 1.72789698058e-02 1.51716517813e-02 1.44498895771e-02 - 1.51716517813e-02 1.72789698058e-02 2.07503567214e-02 2.58559673304e-02 3.30014183614e-02 4.17752511891e-02 - 4.97519001270e-02 5.57017474383e-02 5.74779615393e-02 5.65857611912e-02 5.54677921285e-02 5.54677921285e-02 - 5.65857611912e-02 5.74779615393e-02 5.57017474383e-02 4.97519001270e-02 4.08013474350e-02 3.15640784817e-02 - 2.40042344862e-02 1.85584096087e-02 1.48305747723e-02 1.24193280193e-02 1.12010383875e-02 1.12010383875e-02 - 1.24193280193e-02 1.48305747723e-02 1.85584096087e-02 2.40042344862e-02 3.15640784817e-02 4.08013474350e-02 - 5.26323100532e-02 6.22085036765e-02 6.91691506706e-02 7.32259633424e-02 7.45491088161e-02 7.32259633424e-02 - 6.91691506706e-02 6.22085036765e-02 5.26323100532e-02 4.17752511891e-02 3.15640784817e-02 2.33592944544e-02 - 1.74298422700e-02 1.33587578257e-02 1.06646265060e-02 9.09041352545e-03 8.56724324800e-03 9.09041352545e-03 - 1.06646265060e-02 1.33587578257e-02 1.74298422700e-02 2.33592944544e-02 3.15640784817e-02 4.17752511891e-02 - 5.67802908281e-02 6.87390140452e-02 7.81603362307e-02 8.33624452951e-02 8.33624452951e-02 7.81603362307e-02 - 6.87390140452e-02 5.67802908281e-02 4.42738472380e-02 3.30014183614e-02 2.40042344862e-02 1.74298422700e-02 - 1.28681196083e-02 9.81666041939e-03 7.92776657338e-03 7.02039368822e-03 7.02039368822e-03 7.92776657338e-03 - 9.81666041939e-03 1.28681196083e-02 1.74298422700e-02 2.40042344862e-02 3.30014183614e-02 4.42738472380e-02 - 6.04031976834e-02 7.24709839797e-02 8.11535443280e-02 8.43291015351e-02 8.11535443280e-02 7.24709839797e-02 - 6.04031976834e-02 4.74317203940e-02 3.55533814714e-02 2.58559673304e-02 1.85584096087e-02 1.33587578257e-02 - 9.81666041939e-03 7.56111056333e-03 6.32497949646e-03 5.93416429324e-03 6.32497949646e-03 7.56111056333e-03 - 9.81666041939e-03 1.33587578257e-02 1.85584096087e-02 2.58559673304e-02 3.55533814714e-02 4.74317203940e-02 - 6.30211041418e-02 7.34279203403e-02 7.94813267109e-02 7.94813267109e-02 7.34279203403e-02 6.30211041418e-02 - 5.07662838782e-02 3.88905177841e-02 2.87349610469e-02 2.07503567214e-02 1.48305747723e-02 1.06646265060e-02 - 7.92776657338e-03 6.32497949646e-03 5.60286211346e-03 5.60286211346e-03 6.32497949646e-03 7.92776657338e-03 - 1.06646265060e-02 1.48305747723e-02 2.07503567214e-02 2.87349610469e-02 3.88905177841e-02 5.07662838782e-02 - 6.54257706676e-02 7.34093289667e-02 7.63593661757e-02 7.34093289667e-02 6.54257706676e-02 5.45216826827e-02 - 4.30069852462e-02 3.25477771477e-02 2.39287446557e-02 1.72789698058e-02 1.24193280193e-02 9.09041352545e-03 - 7.02039368822e-03 5.93416429324e-03 5.60286211346e-03 5.93416429324e-03 7.02039368822e-03 9.09041352545e-03 - 1.24193280193e-02 1.72789698058e-02 2.39287446557e-02 3.25477771477e-02 4.30069852462e-02 5.45216826827e-02 - 6.81380649556e-02 7.34279203403e-02 7.34279203403e-02 6.81380649556e-02 5.89796865623e-02 4.80826222932e-02 - 3.73919347545e-02 2.81262227272e-02 2.07351318935e-02 1.51716517813e-02 1.12010383875e-02 8.56724324800e-03 - 7.02039368822e-03 6.32497949646e-03 6.32497949646e-03 7.02039368822e-03 8.56724324800e-03 1.12010383875e-02 - 1.51716517813e-02 2.07351318935e-02 2.81262227272e-02 3.73919347545e-02 4.80826222932e-02 5.89796865623e-02 - 7.00265784056e-02 7.24709839797e-02 7.00265784056e-02 6.32743536159e-02 5.37274523246e-02 4.32645707778e-02 - 3.34978085182e-02 2.53584894772e-02 1.90731200357e-02 1.44498895771e-02 1.12010383875e-02 9.09041352545e-03 - 7.92776657338e-03 7.56111056333e-03 7.92776657338e-03 9.09041352545e-03 1.12010383875e-02 1.44498895771e-02 - 1.90731200357e-02 2.53584894772e-02 3.34978085182e-02 4.32645707778e-02 5.37274523246e-02 6.32743536159e-02 - 6.87390140452e-02 6.87390140452e-02 6.50724984809e-02 5.82637609313e-02 4.94024317434e-02 3.99801891127e-02 - 3.13851597832e-02 2.43885514961e-02 1.90731200357e-02 1.51716517813e-02 1.24193280193e-02 1.06646265060e-02 - 9.81666041939e-03 9.81666041939e-03 1.06646265060e-02 1.24193280193e-02 1.51716517813e-02 1.90731200357e-02 - 2.43885514961e-02 3.13851597832e-02 3.99801891127e-02 4.94024317434e-02 5.82637609313e-02 6.50724984809e-02 - 6.31142666785e-02 6.22085036765e-02 5.92671896593e-02 5.39959847166e-02 4.67604720778e-02 3.87690158276e-02 - 3.13851597832e-02 2.53584894772e-02 2.07351318935e-02 1.72789698058e-02 1.48305747723e-02 1.33587578257e-02 - 1.28681196083e-02 1.33587578257e-02 1.48305747723e-02 1.72789698058e-02 2.07351318935e-02 2.53584894772e-02 - 3.13851597832e-02 3.87690158276e-02 4.67604720778e-02 5.39959847166e-02 5.92671896593e-02 6.22085036765e-02 - 5.52660749302e-02 5.57017474383e-02 5.51986766687e-02 5.22694162731e-02 4.67604720778e-02 3.99801891127e-02 - 3.34978085182e-02 2.81262227272e-02 2.39287446557e-02 2.07503567214e-02 1.85584096087e-02 1.74298422700e-02 - 1.74298422700e-02 1.85584096087e-02 2.07503567214e-02 2.39287446557e-02 2.81262227272e-02 3.34978085182e-02 - 3.99801891127e-02 4.67604720778e-02 5.22694162731e-02 5.51986766687e-02 5.57017474383e-02 5.52660749302e-02 - 4.96150823327e-02 5.29690593685e-02 5.51986766687e-02 5.39959847166e-02 4.94024317434e-02 4.32645707778e-02 - 3.73919347545e-02 3.25477771477e-02 2.87349610469e-02 2.58559673304e-02 2.40042344862e-02 2.33592944544e-02 - 2.40042344862e-02 2.58559673304e-02 2.87349610469e-02 3.25477771477e-02 3.73919347545e-02 4.32645707778e-02 - 4.94024317434e-02 5.39959847166e-02 5.51986766687e-02 5.29690593685e-02 4.96150823327e-02 4.80686289281e-02 - 4.96150823327e-02 5.57017474383e-02 5.92671896593e-02 5.82637609313e-02 5.37274523246e-02 4.80826222932e-02 - 4.30069852462e-02 3.88905177841e-02 3.55533814714e-02 3.30014183614e-02 3.15640784817e-02 3.15640784817e-02 - 3.30014183614e-02 3.55533814714e-02 3.88905177841e-02 4.30069852462e-02 4.80826222932e-02 5.37274523246e-02 - 5.82637609313e-02 5.92671896593e-02 5.57017474383e-02 4.96150823327e-02 4.50421513586e-02 4.50421513586e-02 - 5.52660749302e-02 6.22085036765e-02 6.50724984809e-02 6.32743536159e-02 5.89796865623e-02 5.45216826827e-02 - 5.07662838782e-02 4.74317203940e-02 4.42738472380e-02 4.17752511891e-02 4.08013474350e-02 4.17752511891e-02 - 4.42738472380e-02 4.74317203940e-02 5.07662838782e-02 5.45216826827e-02 5.89796865623e-02 6.32743536159e-02 - 6.50724984809e-02 6.22085036765e-02 5.52660749302e-02 4.80686289281e-02 4.50421513586e-02 4.80686289281e-02 - 5.82846684414e-02 6.04031976834e-02 6.08204348290e-02 6.07372364008e-02 6.08204348290e-02 6.04031976834e-02 - 5.82846684414e-02 5.43728424707e-02 5.03920982744e-02 4.87050863807e-02 5.03920982744e-02 5.43728424707e-02 - 5.82846684414e-02 6.04031976834e-02 6.08204348290e-02 6.07372364008e-02 6.08204348290e-02 6.04031976834e-02 - 5.82846684414e-02 5.43728424707e-02 5.03920982744e-02 4.87050863807e-02 5.03920982744e-02 5.43728424707e-02 - 6.04031976834e-02 6.30211041418e-02 6.54257706676e-02 6.81380649556e-02 7.00265784056e-02 6.87390140452e-02 - 6.31142666785e-02 5.52660749302e-02 4.96150823327e-02 4.96150823327e-02 5.52660749302e-02 6.31142666785e-02 - 6.87390140452e-02 7.00265784056e-02 6.81380649556e-02 6.54257706676e-02 6.30211041418e-02 6.04031976834e-02 - 5.67802908281e-02 5.26323100532e-02 4.97519001270e-02 4.97519001270e-02 5.26323100532e-02 5.67802908281e-02 - 6.08204348290e-02 6.54257706676e-02 7.08330821436e-02 7.55510577091e-02 7.59717702126e-02 6.91691506706e-02 - 5.67646475266e-02 4.49731492966e-02 4.01965567029e-02 4.49731492966e-02 5.67646475266e-02 6.91691506706e-02 - 7.59717702126e-02 7.55510577091e-02 7.08330821436e-02 6.54257706676e-02 6.08204348290e-02 5.66502458441e-02 - 5.24827367699e-02 4.90171582760e-02 4.76311652618e-02 4.90171582760e-02 5.24827367699e-02 5.66502458441e-02 - 6.07372364008e-02 6.81380649556e-02 7.55510577091e-02 7.85611057280e-02 7.22242608078e-02 5.65857611912e-02 - 3.86813628609e-02 2.72980449809e-02 2.72980449809e-02 3.86813628609e-02 5.65857611912e-02 7.22242608078e-02 - 7.85611057280e-02 7.55510577091e-02 6.81380649556e-02 6.07372364008e-02 5.48042821337e-02 4.99875697106e-02 - 4.60708732894e-02 4.37078082031e-02 4.37078082031e-02 4.60708732894e-02 4.99875697106e-02 5.48042821337e-02 - 6.08204348290e-02 7.00265784056e-02 7.59717702126e-02 7.22242608078e-02 5.63481382706e-02 3.47728130512e-02 - 1.79849681520e-02 1.20127916963e-02 1.79849681520e-02 3.47728130512e-02 5.63481382706e-02 7.22242608078e-02 - 7.59717702126e-02 7.00265784056e-02 6.08204348290e-02 5.26365510379e-02 4.64352697346e-02 4.17937584090e-02 - 3.86408229568e-02 3.74820931920e-02 3.86408229568e-02 4.17937584090e-02 4.64352697346e-02 5.26365510379e-02 - 6.04031976834e-02 6.87390140452e-02 6.91691506706e-02 5.65857611912e-02 3.47728130512e-02 1.49395756337e-02 - 4.78306293438e-03 4.78306293438e-03 1.49395756337e-02 3.47728130512e-02 5.65857611912e-02 6.91691506706e-02 - 6.87390140452e-02 6.04031976834e-02 5.07662838782e-02 4.30069852462e-02 3.73919347545e-02 3.34978085182e-02 - 3.13851597832e-02 3.13851597832e-02 3.34978085182e-02 3.73919347545e-02 4.30069852462e-02 5.07662838782e-02 - 5.82846684414e-02 6.31142666785e-02 5.67646475266e-02 3.86813628609e-02 1.79849681520e-02 4.78306293438e-03 - 9.03211615896e-04 4.78306293438e-03 1.79849681520e-02 3.86813628609e-02 5.67646475266e-02 6.31142666785e-02 - 5.82846684414e-02 4.88652050826e-02 4.00749900281e-02 3.35320462758e-02 2.89819100992e-02 2.61347637909e-02 - 2.51358351357e-02 2.61347637909e-02 2.89819100992e-02 3.35320462758e-02 4.00749900281e-02 4.88652050826e-02 - 5.43728424707e-02 5.52660749302e-02 4.49731492966e-02 2.72980449809e-02 1.20127916963e-02 4.78306293438e-03 - 4.78306293438e-03 1.20127916963e-02 2.72980449809e-02 4.49731492966e-02 5.52660749302e-02 5.43728424707e-02 - 4.64972219765e-02 3.74959847737e-02 3.03107984462e-02 2.52525868671e-02 2.19036737434e-02 2.01594924058e-02 - 2.01594924058e-02 2.19036737434e-02 2.52525868671e-02 3.03107984462e-02 3.74959847737e-02 4.64972219765e-02 - 5.03920982744e-02 4.96150823327e-02 4.01965567029e-02 2.72980449809e-02 1.79849681520e-02 1.49395756337e-02 - 1.79849681520e-02 2.72980449809e-02 4.01965567029e-02 4.96150823327e-02 5.03920982744e-02 4.39611245421e-02 - 3.52270678377e-02 2.76901755249e-02 2.22527540382e-02 1.85732113773e-02 1.63414636372e-02 1.55749641354e-02 - 1.63414636372e-02 1.85732113773e-02 2.22527540382e-02 2.76901755249e-02 3.52270678377e-02 4.39611245421e-02 - 4.87050863807e-02 4.96150823327e-02 4.49731492966e-02 3.86813628609e-02 3.47728130512e-02 3.47728130512e-02 - 3.86813628609e-02 4.49731492966e-02 4.96150823327e-02 4.87050863807e-02 4.22555839350e-02 3.35736913190e-02 - 2.57739478448e-02 2.00031512771e-02 1.60537467662e-02 1.35012692014e-02 1.22047713863e-02 1.22047713863e-02 - 1.35012692014e-02 1.60537467662e-02 2.00031512771e-02 2.57739478448e-02 3.35736913190e-02 4.22555839350e-02 - 5.03920982744e-02 5.52660749302e-02 5.67646475266e-02 5.65857611912e-02 5.63481382706e-02 5.65857611912e-02 - 5.67646475266e-02 5.52660749302e-02 5.03920982744e-02 4.22555839350e-02 3.29591336478e-02 2.47523565945e-02 - 1.86064391176e-02 1.43606388158e-02 1.15426690136e-02 9.87827853279e-03 9.31927456850e-03 9.87827853279e-03 - 1.15426690136e-02 1.43606388158e-02 1.86064391176e-02 2.47523565945e-02 3.29591336478e-02 4.22555839350e-02 - 5.43728424707e-02 6.31142666785e-02 6.91691506706e-02 7.22242608078e-02 7.22242608078e-02 6.91691506706e-02 - 6.31142666785e-02 5.43728424707e-02 4.39611245421e-02 3.35736913190e-02 2.47523565945e-02 1.81349712987e-02 - 1.35182636086e-02 1.04118462818e-02 8.45428906874e-03 7.49319081292e-03 7.49319081292e-03 8.45428906874e-03 - 1.04118462818e-02 1.35182636086e-02 1.81349712987e-02 2.47523565945e-02 3.35736913190e-02 4.39611245421e-02 - 5.82846684414e-02 6.87390140452e-02 7.59717702126e-02 7.85611057280e-02 7.59717702126e-02 6.87390140452e-02 - 5.82846684414e-02 4.64972219765e-02 3.52270678377e-02 2.57739478448e-02 1.86064391176e-02 1.35182636086e-02 - 1.00461401111e-02 7.79360764654e-03 6.52118028642e-03 6.10931604239e-03 6.52118028642e-03 7.79360764654e-03 - 1.00461401111e-02 1.35182636086e-02 1.86064391176e-02 2.57739478448e-02 3.52270678377e-02 4.64972219765e-02 - 6.04031976834e-02 7.00265784056e-02 7.55510577091e-02 7.55510577091e-02 7.00265784056e-02 6.04031976834e-02 - 4.88652050826e-02 3.74959847737e-02 2.76901755249e-02 2.00031512771e-02 1.43606388158e-02 1.04118462818e-02 - 7.79360764654e-03 6.22225650652e-03 5.49418812711e-03 5.49418812711e-03 6.22225650652e-03 7.79360764654e-03 - 1.04118462818e-02 1.43606388158e-02 2.00031512771e-02 2.76901755249e-02 3.74959847737e-02 4.88652050826e-02 - 6.08204348290e-02 6.81380649556e-02 7.08330821436e-02 6.81380649556e-02 6.08204348290e-02 5.07662838782e-02 - 4.00749900281e-02 3.03107984462e-02 2.22527540382e-02 1.60537467662e-02 1.15426690136e-02 8.45428906874e-03 - 6.52118028642e-03 5.49418812711e-03 5.17760394975e-03 5.49418812711e-03 6.52118028642e-03 8.45428906874e-03 - 1.15426690136e-02 1.60537467662e-02 2.22527540382e-02 3.03107984462e-02 4.00749900281e-02 5.07662838782e-02 - 6.07372364008e-02 6.54257706676e-02 6.54257706676e-02 6.07372364008e-02 5.26365510379e-02 4.30069852462e-02 - 3.35320462758e-02 2.52525868671e-02 1.85732113773e-02 1.35012692014e-02 9.87827853279e-03 7.49319081292e-03 - 6.10931604239e-03 5.49418812711e-03 5.49418812711e-03 6.10931604239e-03 7.49319081292e-03 9.87827853279e-03 - 1.35012692014e-02 1.85732113773e-02 2.52525868671e-02 3.35320462758e-02 4.30069852462e-02 5.26365510379e-02 - 6.08204348290e-02 6.30211041418e-02 6.08204348290e-02 5.48042821337e-02 4.64352697346e-02 3.73919347545e-02 - 2.89819100992e-02 2.19036737434e-02 1.63414636372e-02 1.22047713863e-02 9.31927456850e-03 7.49319081292e-03 - 6.52118028642e-03 6.22225650652e-03 6.52118028642e-03 7.49319081292e-03 9.31927456850e-03 1.22047713863e-02 - 1.63414636372e-02 2.19036737434e-02 2.89819100992e-02 3.73919347545e-02 4.64352697346e-02 5.48042821337e-02 - 6.04031976834e-02 6.04031976834e-02 5.66502458441e-02 4.99875697106e-02 4.17937584090e-02 3.34978085182e-02 - 2.61347637909e-02 2.01594924058e-02 1.55749641354e-02 1.22047713863e-02 9.87827853279e-03 8.45428906874e-03 - 7.79360764654e-03 7.79360764654e-03 8.45428906874e-03 9.87827853279e-03 1.22047713863e-02 1.55749641354e-02 - 2.01594924058e-02 2.61347637909e-02 3.34978085182e-02 4.17937584090e-02 4.99875697106e-02 5.66502458441e-02 - 5.82846684414e-02 5.67802908281e-02 5.24827367699e-02 4.60708732894e-02 3.86408229568e-02 3.13851597832e-02 - 2.51358351357e-02 2.01594924058e-02 1.63414636372e-02 1.35012692014e-02 1.15426690136e-02 1.04118462818e-02 - 1.00461401111e-02 1.04118462818e-02 1.15426690136e-02 1.35012692014e-02 1.63414636372e-02 2.01594924058e-02 - 2.51358351357e-02 3.13851597832e-02 3.86408229568e-02 4.60708732894e-02 5.24827367699e-02 5.67802908281e-02 - 5.43728424707e-02 5.26323100532e-02 4.90171582760e-02 4.37078082031e-02 3.74820931920e-02 3.13851597832e-02 - 2.61347637909e-02 2.19036737434e-02 1.85732113773e-02 1.60537467662e-02 1.43606388158e-02 1.35182636086e-02 - 1.35182636086e-02 1.43606388158e-02 1.60537467662e-02 1.85732113773e-02 2.19036737434e-02 2.61347637909e-02 - 3.13851597832e-02 3.74820931920e-02 4.37078082031e-02 4.90171582760e-02 5.26323100532e-02 5.43728424707e-02 - 5.03920982744e-02 4.97519001270e-02 4.76311652618e-02 4.37078082031e-02 3.86408229568e-02 3.34978085182e-02 - 2.89819100992e-02 2.52525868671e-02 2.22527540382e-02 2.00031512771e-02 1.86064391176e-02 1.81349712987e-02 - 1.86064391176e-02 2.00031512771e-02 2.22527540382e-02 2.52525868671e-02 2.89819100992e-02 3.34978085182e-02 - 3.86408229568e-02 4.37078082031e-02 4.76311652618e-02 4.97519001270e-02 5.03920982744e-02 5.04488982123e-02 - 4.87050863807e-02 4.97519001270e-02 4.90171582760e-02 4.60708732894e-02 4.17937584090e-02 3.73919347545e-02 - 3.35320462758e-02 3.03107984462e-02 2.76901755249e-02 2.57739478448e-02 2.47523565945e-02 2.47523565945e-02 - 2.57739478448e-02 2.76901755249e-02 3.03107984462e-02 3.35320462758e-02 3.73919347545e-02 4.17937584090e-02 - 4.60708732894e-02 4.90171582760e-02 4.97519001270e-02 4.87050863807e-02 4.74726445019e-02 4.74726445019e-02 - 5.03920982744e-02 5.26323100532e-02 5.24827367699e-02 4.99875697106e-02 4.64352697346e-02 4.30069852462e-02 - 4.00749900281e-02 3.74959847737e-02 3.52270678377e-02 3.35736913190e-02 3.29591336478e-02 3.35736913190e-02 - 3.52270678377e-02 3.74959847737e-02 4.00749900281e-02 4.30069852462e-02 4.64352697346e-02 4.99875697106e-02 - 5.24827367699e-02 5.26323100532e-02 5.03920982744e-02 4.74726445019e-02 4.61549122721e-02 4.74726445019e-02 - 5.43728424707e-02 5.67802908281e-02 5.66502458441e-02 5.48042821337e-02 5.26365510379e-02 5.07662838782e-02 - 4.88652050826e-02 4.64972219765e-02 4.39611245421e-02 4.22555839350e-02 4.22555839350e-02 4.39611245421e-02 - 4.64972219765e-02 4.88652050826e-02 5.07662838782e-02 5.26365510379e-02 5.48042821337e-02 5.66502458441e-02 - 5.67802908281e-02 5.43728424707e-02 5.04488982123e-02 4.74726445019e-02 4.74726445019e-02 5.04488982123e-02 - 4.88652050826e-02 5.07662838782e-02 5.26365510379e-02 5.48042821337e-02 5.66502458441e-02 5.67802908281e-02 - 5.43728424707e-02 5.04488982123e-02 4.74726445019e-02 4.74726445019e-02 5.04488982123e-02 5.43728424707e-02 - 5.67802908281e-02 5.66502458441e-02 5.48042821337e-02 5.26365510379e-02 5.07662838782e-02 4.88652050826e-02 - 4.64972219765e-02 4.39611245421e-02 4.22555839350e-02 4.22555839350e-02 4.39611245421e-02 4.64972219765e-02 - 5.07662838782e-02 5.45216826827e-02 5.89796865623e-02 6.32743536159e-02 6.50724984809e-02 6.22085036765e-02 - 5.52660749302e-02 4.80686289281e-02 4.50421513586e-02 4.80686289281e-02 5.52660749302e-02 6.22085036765e-02 - 6.50724984809e-02 6.32743536159e-02 5.89796865623e-02 5.45216826827e-02 5.07662838782e-02 4.74317203940e-02 - 4.42738472380e-02 4.17752511891e-02 4.08013474350e-02 4.17752511891e-02 4.42738472380e-02 4.74317203940e-02 - 5.26365510379e-02 5.89796865623e-02 6.56955874028e-02 6.97193973966e-02 6.72242960853e-02 5.74779615393e-02 - 4.49731492966e-02 3.64979381134e-02 3.64979381134e-02 4.49731492966e-02 5.74779615393e-02 6.72242960853e-02 - 6.97193973966e-02 6.56955874028e-02 5.89796865623e-02 5.26365510379e-02 4.75580604876e-02 4.34476578087e-02 - 4.01880472274e-02 3.82730721923e-02 3.82730721923e-02 4.01880472274e-02 4.34476578087e-02 4.75580604876e-02 - 5.48042821337e-02 6.32743536159e-02 6.97193973966e-02 6.89438173057e-02 5.81927448576e-02 4.14983823737e-02 - 2.72980449809e-02 2.19326842858e-02 2.72980449809e-02 4.14983823737e-02 5.81927448576e-02 6.89438173057e-02 - 6.97193973966e-02 6.32743536159e-02 5.48042821337e-02 4.74868469137e-02 4.19230722756e-02 3.77513071960e-02 - 3.49472650912e-02 3.39275143754e-02 3.49472650912e-02 3.77513071960e-02 4.19230722756e-02 4.74868469137e-02 - 5.66502458441e-02 6.50724984809e-02 6.72242960853e-02 5.81927448576e-02 4.01124125220e-02 2.21086562145e-02 - 1.20127916963e-02 1.20127916963e-02 2.21086562145e-02 4.01124125220e-02 5.81927448576e-02 6.72242960853e-02 - 6.50724984809e-02 5.66502458441e-02 4.75580604876e-02 4.03211904768e-02 3.50622322206e-02 3.14092023629e-02 - 2.94353177762e-02 2.94353177762e-02 3.14092023629e-02 3.50622322206e-02 4.03211904768e-02 4.75580604876e-02 - 5.67802908281e-02 6.22085036765e-02 5.74779615393e-02 4.14983823737e-02 2.21086562145e-02 8.92458323932e-03 - 4.78306293438e-03 8.92458323932e-03 2.21086562145e-02 4.14983823737e-02 5.74779615393e-02 6.22085036765e-02 - 5.67802908281e-02 4.74317203940e-02 3.88905177841e-02 3.25477771477e-02 2.81262227272e-02 2.53584894772e-02 - 2.43885514961e-02 2.53584894772e-02 2.81262227272e-02 3.25477771477e-02 3.88905177841e-02 4.74317203940e-02 - 5.43728424707e-02 5.52660749302e-02 4.49731492966e-02 2.72980449809e-02 1.20127916963e-02 4.78306293438e-03 - 4.78306293438e-03 1.20127916963e-02 2.72980449809e-02 4.49731492966e-02 5.52660749302e-02 5.43728424707e-02 - 4.64972219765e-02 3.74959847737e-02 3.03107984462e-02 2.52525868671e-02 2.19036737434e-02 2.01594924058e-02 - 2.01594924058e-02 2.19036737434e-02 2.52525868671e-02 3.03107984462e-02 3.74959847737e-02 4.64972219765e-02 - 5.04488982123e-02 4.80686289281e-02 3.64979381134e-02 2.19326842858e-02 1.20127916963e-02 8.92458323932e-03 - 1.20127916963e-02 2.19326842858e-02 3.64979381134e-02 4.80686289281e-02 5.04488982123e-02 4.46898454449e-02 - 3.60143597485e-02 2.83448586606e-02 2.27884625410e-02 1.90357893277e-02 1.67618733320e-02 1.59804507535e-02 - 1.67618733320e-02 1.90357893277e-02 2.27884625410e-02 2.83448586606e-02 3.60143597485e-02 4.46898454449e-02 - 4.74726445019e-02 4.50421513586e-02 3.64979381134e-02 2.72980449809e-02 2.21086562145e-02 2.21086562145e-02 - 2.72980449809e-02 3.64979381134e-02 4.50421513586e-02 4.74726445019e-02 4.28256060156e-02 3.46378859300e-02 - 2.67482412330e-02 2.07965618706e-02 1.67247696023e-02 1.41011278955e-02 1.27676986907e-02 1.27676986907e-02 - 1.41011278955e-02 1.67247696023e-02 2.07965618706e-02 2.67482412330e-02 3.46378859300e-02 4.28256060156e-02 - 4.74726445019e-02 4.80686289281e-02 4.49731492966e-02 4.14983823737e-02 4.01124125220e-02 4.14983823737e-02 - 4.49731492966e-02 4.80686289281e-02 4.74726445019e-02 4.20317983678e-02 3.37861689888e-02 2.57020582842e-02 - 1.94109424364e-02 1.50351544907e-02 1.21379949101e-02 1.04248538239e-02 9.84753974693e-03 1.04248538239e-02 - 1.21379949101e-02 1.50351544907e-02 1.94109424364e-02 2.57020582842e-02 3.37861689888e-02 4.20317983678e-02 - 5.04488982123e-02 5.52660749302e-02 5.74779615393e-02 5.81927448576e-02 5.81927448576e-02 5.74779615393e-02 - 5.52660749302e-02 5.04488982123e-02 4.28256060156e-02 3.37861689888e-02 2.53397129065e-02 1.87149718700e-02 - 1.40306458540e-02 1.08721475193e-02 8.87112379151e-03 7.87995370977e-03 7.87995370977e-03 8.87112379151e-03 - 1.08721475193e-02 1.40306458540e-02 1.87149718700e-02 2.53397129065e-02 3.37861689888e-02 4.28256060156e-02 - 5.43728424707e-02 6.22085036765e-02 6.72242960853e-02 6.89438173057e-02 6.72242960853e-02 6.22085036765e-02 - 5.43728424707e-02 4.46898454449e-02 3.46378859300e-02 2.57020582842e-02 1.87149718700e-02 1.37029642554e-02 - 1.02670781727e-02 8.01216613758e-03 6.71488304626e-03 6.28902224701e-03 6.71488304626e-03 8.01216613758e-03 - 1.02670781727e-02 1.37029642554e-02 1.87149718700e-02 2.57020582842e-02 3.46378859300e-02 4.46898454449e-02 - 5.67802908281e-02 6.50724984809e-02 6.97193973966e-02 6.97193973966e-02 6.50724984809e-02 5.67802908281e-02 - 4.64972219765e-02 3.60143597485e-02 2.67482412330e-02 1.94109424364e-02 1.40306458540e-02 1.02670781727e-02 - 7.74242036277e-03 6.18966814651e-03 5.45146116297e-03 5.45146116297e-03 6.18966814651e-03 7.74242036277e-03 - 1.02670781727e-02 1.40306458540e-02 1.94109424364e-02 2.67482412330e-02 3.60143597485e-02 4.64972219765e-02 - 5.66502458441e-02 6.32743536159e-02 6.56955874028e-02 6.32743536159e-02 5.66502458441e-02 4.74317203940e-02 - 3.74959847737e-02 2.83448586606e-02 2.07965618706e-02 1.50351544907e-02 1.08721475193e-02 8.01216613758e-03 - 6.18966814651e-03 5.19548069751e-03 4.88266428153e-03 5.19548069751e-03 6.18966814651e-03 8.01216613758e-03 - 1.08721475193e-02 1.50351544907e-02 2.07965618706e-02 2.83448586606e-02 3.74959847737e-02 4.74317203940e-02 - 5.48042821337e-02 5.89796865623e-02 5.89796865623e-02 5.48042821337e-02 4.75580604876e-02 3.88905177841e-02 - 3.03107984462e-02 2.27884625410e-02 1.67247696023e-02 1.21379949101e-02 8.87112379151e-03 6.71488304626e-03 - 5.45146116297e-03 4.88266428153e-03 4.88266428153e-03 5.45146116297e-03 6.71488304626e-03 8.87112379151e-03 - 1.21379949101e-02 1.67247696023e-02 2.27884625410e-02 3.03107984462e-02 3.88905177841e-02 4.75580604876e-02 - 5.26365510379e-02 5.45216826827e-02 5.26365510379e-02 4.74868469137e-02 4.03211904768e-02 3.25477771477e-02 - 2.52525868671e-02 1.90357893277e-02 1.41011278955e-02 1.04248538239e-02 7.87995370977e-03 6.28902224701e-03 - 5.45146116297e-03 5.19548069751e-03 5.45146116297e-03 6.28902224701e-03 7.87995370977e-03 1.04248538239e-02 - 1.41011278955e-02 1.90357893277e-02 2.52525868671e-02 3.25477771477e-02 4.03211904768e-02 4.74868469137e-02 - 5.07662838782e-02 5.07662838782e-02 4.75580604876e-02 4.19230722756e-02 3.50622322206e-02 2.81262227272e-02 - 2.19036737434e-02 1.67618733320e-02 1.27676986907e-02 9.84753974693e-03 7.87995370977e-03 6.71488304626e-03 - 6.18966814651e-03 6.18966814651e-03 6.71488304626e-03 7.87995370977e-03 9.84753974693e-03 1.27676986907e-02 - 1.67618733320e-02 2.19036737434e-02 2.81262227272e-02 3.50622322206e-02 4.19230722756e-02 4.75580604876e-02 - 4.88652050826e-02 4.74317203940e-02 4.34476578087e-02 3.77513071960e-02 3.14092023629e-02 2.53584894772e-02 - 2.01594924058e-02 1.59804507535e-02 1.27676986907e-02 1.04248538239e-02 8.87112379151e-03 8.01216613758e-03 - 7.74242036277e-03 8.01216613758e-03 8.87112379151e-03 1.04248538239e-02 1.27676986907e-02 1.59804507535e-02 - 2.01594924058e-02 2.53584894772e-02 3.14092023629e-02 3.77513071960e-02 4.34476578087e-02 4.74317203940e-02 - 4.64972219765e-02 4.42738472380e-02 4.01880472274e-02 3.49472650912e-02 2.94353177762e-02 2.43885514961e-02 - 2.01594924058e-02 1.67618733320e-02 1.41011278955e-02 1.21379949101e-02 1.08721475193e-02 1.02670781727e-02 - 1.02670781727e-02 1.08721475193e-02 1.21379949101e-02 1.41011278955e-02 1.67618733320e-02 2.01594924058e-02 - 2.43885514961e-02 2.94353177762e-02 3.49472650912e-02 4.01880472274e-02 4.42738472380e-02 4.64972219765e-02 - 4.39611245421e-02 4.17752511891e-02 3.82730721923e-02 3.39275143754e-02 2.94353177762e-02 2.53584894772e-02 - 2.19036737434e-02 1.90357893277e-02 1.67247696023e-02 1.50351544907e-02 1.40306458540e-02 1.37029642554e-02 - 1.40306458540e-02 1.50351544907e-02 1.67247696023e-02 1.90357893277e-02 2.19036737434e-02 2.53584894772e-02 - 2.94353177762e-02 3.39275143754e-02 3.82730721923e-02 4.17752511891e-02 4.39611245421e-02 4.46898454449e-02 - 4.22555839350e-02 4.08013474350e-02 3.82730721923e-02 3.49472650912e-02 3.14092023629e-02 2.81262227272e-02 - 2.52525868671e-02 2.27884625410e-02 2.07965618706e-02 1.94109424364e-02 1.87149718700e-02 1.87149718700e-02 - 1.94109424364e-02 2.07965618706e-02 2.27884625410e-02 2.52525868671e-02 2.81262227272e-02 3.14092023629e-02 - 3.49472650912e-02 3.82730721923e-02 4.08013474350e-02 4.22555839350e-02 4.28256060156e-02 4.28256060156e-02 - 4.22555839350e-02 4.17752511891e-02 4.01880472274e-02 3.77513071960e-02 3.50622322206e-02 3.25477771477e-02 - 3.03107984462e-02 2.83448586606e-02 2.67482412330e-02 2.57020582842e-02 2.53397129065e-02 2.57020582842e-02 - 2.67482412330e-02 2.83448586606e-02 3.03107984462e-02 3.25477771477e-02 3.50622322206e-02 3.77513071960e-02 - 4.01880472274e-02 4.17752511891e-02 4.22555839350e-02 4.20317983678e-02 4.18365826600e-02 4.20317983678e-02 - 4.39611245421e-02 4.42738472380e-02 4.34476578087e-02 4.19230722756e-02 4.03211904768e-02 3.88905177841e-02 - 3.74959847737e-02 3.60143597485e-02 3.46378859300e-02 3.37861689888e-02 3.37861689888e-02 3.46378859300e-02 - 3.60143597485e-02 3.74959847737e-02 3.88905177841e-02 4.03211904768e-02 4.19230722756e-02 4.34476578087e-02 - 4.42738472380e-02 4.39611245421e-02 4.28256060156e-02 4.18365826600e-02 4.18365826600e-02 4.28256060156e-02 - 4.64972219765e-02 4.74317203940e-02 4.75580604876e-02 4.74868469137e-02 4.75580604876e-02 4.74317203940e-02 - 4.64972219765e-02 4.46898454449e-02 4.28256060156e-02 4.20317983678e-02 4.28256060156e-02 4.46898454449e-02 - 4.64972219765e-02 4.74317203940e-02 4.75580604876e-02 4.74868469137e-02 4.75580604876e-02 4.74317203940e-02 - 4.64972219765e-02 4.46898454449e-02 4.28256060156e-02 4.20317983678e-02 4.28256060156e-02 4.46898454449e-02 - 4.00749900281e-02 4.30069852462e-02 4.64352697346e-02 4.99875697106e-02 5.24827367699e-02 5.26323100532e-02 - 5.03920982744e-02 4.74726445019e-02 4.61549122721e-02 4.74726445019e-02 5.03920982744e-02 5.26323100532e-02 - 5.24827367699e-02 4.99875697106e-02 4.64352697346e-02 4.30069852462e-02 4.00749900281e-02 3.74959847737e-02 - 3.52270678377e-02 3.35736913190e-02 3.29591336478e-02 3.35736913190e-02 3.52270678377e-02 3.74959847737e-02 - 4.30069852462e-02 4.80826222932e-02 5.37274523246e-02 5.82637609313e-02 5.92671896593e-02 5.57017474383e-02 - 4.96150823327e-02 4.50421513586e-02 4.50421513586e-02 4.96150823327e-02 5.57017474383e-02 5.92671896593e-02 - 5.82637609313e-02 5.37274523246e-02 4.80826222932e-02 4.30069852462e-02 3.88905177841e-02 3.55533814714e-02 - 3.30014183614e-02 3.15640784817e-02 3.15640784817e-02 3.30014183614e-02 3.55533814714e-02 3.88905177841e-02 - 4.64352697346e-02 5.37274523246e-02 6.03520680545e-02 6.28462967289e-02 5.86480861577e-02 4.93152444260e-02 - 4.01965567029e-02 3.64979381134e-02 4.01965567029e-02 4.93152444260e-02 5.86480861577e-02 6.28462967289e-02 - 6.03520680545e-02 5.37274523246e-02 4.64352697346e-02 4.03211904768e-02 3.56148556138e-02 3.20717125323e-02 - 2.97324869747e-02 2.88969153857e-02 2.97324869747e-02 3.20717125323e-02 3.56148556138e-02 4.03211904768e-02 - 4.99875697106e-02 5.82637609313e-02 6.28462967289e-02 5.95705865689e-02 4.85514460951e-02 3.55340946595e-02 - 2.72980449809e-02 2.72980449809e-02 3.55340946595e-02 4.85514460951e-02 5.95705865689e-02 6.28462967289e-02 - 5.82637609313e-02 4.99875697106e-02 4.19230722756e-02 3.55984989831e-02 3.09544991478e-02 2.77206217649e-02 - 2.59885128162e-02 2.59885128162e-02 2.77206217649e-02 3.09544991478e-02 3.55984989831e-02 4.19230722756e-02 - 5.24827367699e-02 5.92671896593e-02 5.86480861577e-02 4.85514460951e-02 3.37527793613e-02 2.21086562145e-02 - 1.79849681520e-02 2.21086562145e-02 3.37527793613e-02 4.85514460951e-02 5.86480861577e-02 5.92671896593e-02 - 5.24827367699e-02 4.34476578087e-02 3.56148556138e-02 2.98227515673e-02 2.57560150434e-02 2.32096903098e-02 - 2.23208692653e-02 2.32096903098e-02 2.57560150434e-02 2.98227515673e-02 3.56148556138e-02 4.34476578087e-02 - 5.26323100532e-02 5.57017474383e-02 4.93152444260e-02 3.55340946595e-02 2.21086562145e-02 1.49395756337e-02 - 1.49395756337e-02 2.21086562145e-02 3.55340946595e-02 4.93152444260e-02 5.57017474383e-02 5.26323100532e-02 - 4.42738472380e-02 3.55533814714e-02 2.87349610469e-02 2.39287446557e-02 2.07351318935e-02 1.90731200357e-02 - 1.90731200357e-02 2.07351318935e-02 2.39287446557e-02 2.87349610469e-02 3.55533814714e-02 4.42738472380e-02 - 5.03920982744e-02 4.96150823327e-02 4.01965567029e-02 2.72980449809e-02 1.79849681520e-02 1.49395756337e-02 - 1.79849681520e-02 2.72980449809e-02 4.01965567029e-02 4.96150823327e-02 5.03920982744e-02 4.39611245421e-02 - 3.52270678377e-02 2.76901755249e-02 2.22527540382e-02 1.85732113773e-02 1.63414636372e-02 1.55749641354e-02 - 1.63414636372e-02 1.85732113773e-02 2.22527540382e-02 2.76901755249e-02 3.52270678377e-02 4.39611245421e-02 - 4.74726445019e-02 4.50421513586e-02 3.64979381134e-02 2.72980449809e-02 2.21086562145e-02 2.21086562145e-02 - 2.72980449809e-02 3.64979381134e-02 4.50421513586e-02 4.74726445019e-02 4.28256060156e-02 3.46378859300e-02 - 2.67482412330e-02 2.07965618706e-02 1.67247696023e-02 1.41011278955e-02 1.27676986907e-02 1.27676986907e-02 - 1.41011278955e-02 1.67247696023e-02 2.07965618706e-02 2.67482412330e-02 3.46378859300e-02 4.28256060156e-02 - 4.61549122721e-02 4.50421513586e-02 4.01965567029e-02 3.55340946595e-02 3.37527793613e-02 3.55340946595e-02 - 4.01965567029e-02 4.50421513586e-02 4.61549122721e-02 4.18365826600e-02 3.40527023771e-02 2.60389015581e-02 - 1.96971752186e-02 1.52731342228e-02 1.23490348915e-02 1.06212947837e-02 1.00387761995e-02 1.06212947837e-02 - 1.23490348915e-02 1.52731342228e-02 1.96971752186e-02 2.60389015581e-02 3.40527023771e-02 4.18365826600e-02 - 4.74726445019e-02 4.96150823327e-02 4.93152444260e-02 4.85514460951e-02 4.85514460951e-02 4.93152444260e-02 - 4.96150823327e-02 4.74726445019e-02 4.18365826600e-02 3.38082178713e-02 2.56667001275e-02 1.90473786081e-02 - 1.43139225142e-02 1.11222486043e-02 9.10176004354e-03 8.09940143038e-03 8.09940143038e-03 9.10176004354e-03 - 1.11222486043e-02 1.43139225142e-02 1.90473786081e-02 2.56667001275e-02 3.38082178713e-02 4.18365826600e-02 - 5.03920982744e-02 5.57017474383e-02 5.86480861577e-02 5.95705865689e-02 5.86480861577e-02 5.57017474383e-02 - 5.03920982744e-02 4.28256060156e-02 3.40527023771e-02 2.56667001275e-02 1.88392765173e-02 1.38569882819e-02 - 1.04242770349e-02 8.16346419979e-03 6.85502201162e-03 6.42332605072e-03 6.85502201162e-03 8.16346419979e-03 - 1.04242770349e-02 1.38569882819e-02 1.88392765173e-02 2.56667001275e-02 3.40527023771e-02 4.28256060156e-02 - 5.26323100532e-02 5.92671896593e-02 6.28462967289e-02 6.28462967289e-02 5.92671896593e-02 5.26323100532e-02 - 4.39611245421e-02 3.46378859300e-02 2.60389015581e-02 1.90473786081e-02 1.38569882819e-02 1.02031841081e-02 - 7.72923129933e-03 6.18550458410e-03 5.44128685160e-03 5.44128685160e-03 6.18550458410e-03 7.72923129933e-03 - 1.02031841081e-02 1.38569882819e-02 1.90473786081e-02 2.60389015581e-02 3.46378859300e-02 4.39611245421e-02 - 5.24827367699e-02 5.82637609313e-02 6.03520680545e-02 5.82637609313e-02 5.24827367699e-02 4.42738472380e-02 - 3.52270678377e-02 2.67482412330e-02 1.96971752186e-02 1.43139225142e-02 1.04242770349e-02 7.72923129933e-03 - 5.97825202186e-03 5.00022810356e-03 4.68737681852e-03 5.00022810356e-03 5.97825202186e-03 7.72923129933e-03 - 1.04242770349e-02 1.43139225142e-02 1.96971752186e-02 2.67482412330e-02 3.52270678377e-02 4.42738472380e-02 - 4.99875697106e-02 5.37274523246e-02 5.37274523246e-02 4.99875697106e-02 4.34476578087e-02 3.55533814714e-02 - 2.76901755249e-02 2.07965618706e-02 1.52731342228e-02 1.11222486043e-02 8.16346419979e-03 6.18550458410e-03 - 5.00022810356e-03 4.45430428341e-03 4.45430428341e-03 5.00022810356e-03 6.18550458410e-03 8.16346419979e-03 - 1.11222486043e-02 1.52731342228e-02 2.07965618706e-02 2.76901755249e-02 3.55533814714e-02 4.34476578087e-02 - 4.64352697346e-02 4.80826222932e-02 4.64352697346e-02 4.19230722756e-02 3.56148556138e-02 2.87349610469e-02 - 2.22527540382e-02 1.67247696023e-02 1.23490348915e-02 9.10176004354e-03 6.85502201162e-03 5.44128685160e-03 - 4.68737681852e-03 4.45430428341e-03 4.68737681852e-03 5.44128685160e-03 6.85502201162e-03 9.10176004354e-03 - 1.23490348915e-02 1.67247696023e-02 2.22527540382e-02 2.87349610469e-02 3.56148556138e-02 4.19230722756e-02 - 4.30069852462e-02 4.30069852462e-02 4.03211904768e-02 3.55984989831e-02 2.98227515673e-02 2.39287446557e-02 - 1.85732113773e-02 1.41011278955e-02 1.06212947837e-02 8.09940143038e-03 6.42332605072e-03 5.44128685160e-03 - 5.00022810356e-03 5.00022810356e-03 5.44128685160e-03 6.42332605072e-03 8.09940143038e-03 1.06212947837e-02 - 1.41011278955e-02 1.85732113773e-02 2.39287446557e-02 2.98227515673e-02 3.55984989831e-02 4.03211904768e-02 - 4.00749900281e-02 3.88905177841e-02 3.56148556138e-02 3.09544991478e-02 2.57560150434e-02 2.07351318935e-02 - 1.63414636372e-02 1.27676986907e-02 1.00387761995e-02 8.09940143038e-03 6.85502201162e-03 6.18550458410e-03 - 5.97825202186e-03 6.18550458410e-03 6.85502201162e-03 8.09940143038e-03 1.00387761995e-02 1.27676986907e-02 - 1.63414636372e-02 2.07351318935e-02 2.57560150434e-02 3.09544991478e-02 3.56148556138e-02 3.88905177841e-02 - 3.74959847737e-02 3.55533814714e-02 3.20717125323e-02 2.77206217649e-02 2.32096903098e-02 1.90731200357e-02 - 1.55749641354e-02 1.27676986907e-02 1.06212947837e-02 9.10176004354e-03 8.16346419979e-03 7.72923129933e-03 - 7.72923129933e-03 8.16346419979e-03 9.10176004354e-03 1.06212947837e-02 1.27676986907e-02 1.55749641354e-02 - 1.90731200357e-02 2.32096903098e-02 2.77206217649e-02 3.20717125323e-02 3.55533814714e-02 3.74959847737e-02 - 3.52270678377e-02 3.30014183614e-02 2.97324869747e-02 2.59885128162e-02 2.23208692653e-02 1.90731200357e-02 - 1.63414636372e-02 1.41011278955e-02 1.23490348915e-02 1.11222486043e-02 1.04242770349e-02 1.02031841081e-02 - 1.04242770349e-02 1.11222486043e-02 1.23490348915e-02 1.41011278955e-02 1.63414636372e-02 1.90731200357e-02 - 2.23208692653e-02 2.59885128162e-02 2.97324869747e-02 3.30014183614e-02 3.52270678377e-02 3.60143597485e-02 - 3.35736913190e-02 3.15640784817e-02 2.88969153857e-02 2.59885128162e-02 2.32096903098e-02 2.07351318935e-02 - 1.85732113773e-02 1.67247696023e-02 1.52731342228e-02 1.43139225142e-02 1.38569882819e-02 1.38569882819e-02 - 1.43139225142e-02 1.52731342228e-02 1.67247696023e-02 1.85732113773e-02 2.07351318935e-02 2.32096903098e-02 - 2.59885128162e-02 2.88969153857e-02 3.15640784817e-02 3.35736913190e-02 3.46378859300e-02 3.46378859300e-02 - 3.29591336478e-02 3.15640784817e-02 2.97324869747e-02 2.77206217649e-02 2.57560150434e-02 2.39287446557e-02 - 2.22527540382e-02 2.07965618706e-02 1.96971752186e-02 1.90473786081e-02 1.88392765173e-02 1.90473786081e-02 - 1.96971752186e-02 2.07965618706e-02 2.22527540382e-02 2.39287446557e-02 2.57560150434e-02 2.77206217649e-02 - 2.97324869747e-02 3.15640784817e-02 3.29591336478e-02 3.37861689888e-02 3.40527023771e-02 3.37861689888e-02 - 3.35736913190e-02 3.30014183614e-02 3.20717125323e-02 3.09544991478e-02 2.98227515673e-02 2.87349610469e-02 - 2.76901755249e-02 2.67482412330e-02 2.60389015581e-02 2.56667001275e-02 2.56667001275e-02 2.60389015581e-02 - 2.67482412330e-02 2.76901755249e-02 2.87349610469e-02 2.98227515673e-02 3.09544991478e-02 3.20717125323e-02 - 3.30014183614e-02 3.35736913190e-02 3.37861689888e-02 3.38082178713e-02 3.38082178713e-02 3.37861689888e-02 - 3.52270678377e-02 3.55533814714e-02 3.56148556138e-02 3.55984989831e-02 3.56148556138e-02 3.55533814714e-02 - 3.52270678377e-02 3.46378859300e-02 3.40527023771e-02 3.38082178713e-02 3.40527023771e-02 3.46378859300e-02 - 3.52270678377e-02 3.55533814714e-02 3.56148556138e-02 3.55984989831e-02 3.56148556138e-02 3.55533814714e-02 - 3.52270678377e-02 3.46378859300e-02 3.40527023771e-02 3.38082178713e-02 3.40527023771e-02 3.46378859300e-02 - 3.74959847737e-02 3.88905177841e-02 4.03211904768e-02 4.19230722756e-02 4.34476578087e-02 4.42738472380e-02 - 4.39611245421e-02 4.28256060156e-02 4.18365826600e-02 4.18365826600e-02 4.28256060156e-02 4.39611245421e-02 - 4.42738472380e-02 4.34476578087e-02 4.19230722756e-02 4.03211904768e-02 3.88905177841e-02 3.74959847737e-02 - 3.60143597485e-02 3.46378859300e-02 3.37861689888e-02 3.37861689888e-02 3.46378859300e-02 3.60143597485e-02 - 3.35320462758e-02 3.73919347545e-02 4.17937584090e-02 4.60708732894e-02 4.90171582760e-02 4.97519001270e-02 - 4.87050863807e-02 4.74726445019e-02 4.74726445019e-02 4.87050863807e-02 4.97519001270e-02 4.90171582760e-02 - 4.60708732894e-02 4.17937584090e-02 3.73919347545e-02 3.35320462758e-02 3.03107984462e-02 2.76901755249e-02 - 2.57739478448e-02 2.47523565945e-02 2.47523565945e-02 2.57739478448e-02 2.76901755249e-02 3.03107984462e-02 - 3.73919347545e-02 4.32645707778e-02 4.94024317434e-02 5.39959847166e-02 5.51986766687e-02 5.29690593685e-02 - 4.96150823327e-02 4.80686289281e-02 4.96150823327e-02 5.29690593685e-02 5.51986766687e-02 5.39959847166e-02 - 4.94024317434e-02 4.32645707778e-02 3.73919347545e-02 3.25477771477e-02 2.87349610469e-02 2.58559673304e-02 - 2.40042344862e-02 2.33592944544e-02 2.40042344862e-02 2.58559673304e-02 2.87349610469e-02 3.25477771477e-02 - 4.17937584090e-02 4.94024317434e-02 5.57693346902e-02 5.80220562298e-02 5.50728972421e-02 4.93152444260e-02 - 4.49731492966e-02 4.49731492966e-02 4.93152444260e-02 5.50728972421e-02 5.80220562298e-02 5.57693346902e-02 - 4.94024317434e-02 4.17937584090e-02 3.50622322206e-02 2.98227515673e-02 2.59124227944e-02 2.31896632144e-02 - 2.17534458677e-02 2.17534458677e-02 2.31896632144e-02 2.59124227944e-02 2.98227515673e-02 3.50622322206e-02 - 4.60708732894e-02 5.39959847166e-02 5.80220562298e-02 5.56741558656e-02 4.85514460951e-02 4.14983823737e-02 - 3.86813628609e-02 4.14983823737e-02 4.85514460951e-02 5.56741558656e-02 5.80220562298e-02 5.39959847166e-02 - 4.60708732894e-02 3.77513071960e-02 3.09544991478e-02 2.59309064966e-02 2.23681861845e-02 2.01439779904e-02 - 1.93739445237e-02 2.01439779904e-02 2.23681861845e-02 2.59309064966e-02 3.09544991478e-02 3.77513071960e-02 - 4.90171582760e-02 5.51986766687e-02 5.50728972421e-02 4.85514460951e-02 4.01124125220e-02 3.47728130512e-02 - 3.47728130512e-02 4.01124125220e-02 4.85514460951e-02 5.50728972421e-02 5.51986766687e-02 4.90171582760e-02 - 4.01880472274e-02 3.20717125323e-02 2.59124227944e-02 2.15543530804e-02 1.86453765526e-02 1.71372885777e-02 - 1.71372885777e-02 1.86453765526e-02 2.15543530804e-02 2.59124227944e-02 3.20717125323e-02 4.01880472274e-02 - 4.97519001270e-02 5.29690593685e-02 4.93152444260e-02 4.14983823737e-02 3.47728130512e-02 3.23216192243e-02 - 3.47728130512e-02 4.14983823737e-02 4.93152444260e-02 5.29690593685e-02 4.97519001270e-02 4.17752511891e-02 - 3.30014183614e-02 2.58559673304e-02 2.07503567214e-02 1.72789698058e-02 1.51716517813e-02 1.44498895771e-02 - 1.51716517813e-02 1.72789698058e-02 2.07503567214e-02 2.58559673304e-02 3.30014183614e-02 4.17752511891e-02 - 4.87050863807e-02 4.96150823327e-02 4.49731492966e-02 3.86813628609e-02 3.47728130512e-02 3.47728130512e-02 - 3.86813628609e-02 4.49731492966e-02 4.96150823327e-02 4.87050863807e-02 4.22555839350e-02 3.35736913190e-02 - 2.57739478448e-02 2.00031512771e-02 1.60537467662e-02 1.35012692014e-02 1.22047713863e-02 1.22047713863e-02 - 1.35012692014e-02 1.60537467662e-02 2.00031512771e-02 2.57739478448e-02 3.35736913190e-02 4.22555839350e-02 - 4.74726445019e-02 4.80686289281e-02 4.49731492966e-02 4.14983823737e-02 4.01124125220e-02 4.14983823737e-02 - 4.49731492966e-02 4.80686289281e-02 4.74726445019e-02 4.20317983678e-02 3.37861689888e-02 2.57020582842e-02 - 1.94109424364e-02 1.50351544907e-02 1.21379949101e-02 1.04248538239e-02 9.84753974693e-03 1.04248538239e-02 - 1.21379949101e-02 1.50351544907e-02 1.94109424364e-02 2.57020582842e-02 3.37861689888e-02 4.20317983678e-02 - 4.74726445019e-02 4.96150823327e-02 4.93152444260e-02 4.85514460951e-02 4.85514460951e-02 4.93152444260e-02 - 4.96150823327e-02 4.74726445019e-02 4.18365826600e-02 3.38082178713e-02 2.56667001275e-02 1.90473786081e-02 - 1.43139225142e-02 1.11222486043e-02 9.10176004354e-03 8.09940143038e-03 8.09940143038e-03 9.10176004354e-03 - 1.11222486043e-02 1.43139225142e-02 1.90473786081e-02 2.56667001275e-02 3.38082178713e-02 4.18365826600e-02 - 4.87050863807e-02 5.29690593685e-02 5.50728972421e-02 5.56741558656e-02 5.50728972421e-02 5.29690593685e-02 - 4.87050863807e-02 4.20317983678e-02 3.38082178713e-02 2.56591838074e-02 1.88953143126e-02 1.39177867633e-02 - 1.04809874105e-02 8.21703388740e-03 6.90595653158e-03 6.47303924297e-03 6.90595653158e-03 8.21703388740e-03 - 1.04809874105e-02 1.39177867633e-02 1.88953143126e-02 2.56591838074e-02 3.38082178713e-02 4.20317983678e-02 - 4.97519001270e-02 5.51986766687e-02 5.80220562298e-02 5.80220562298e-02 5.51986766687e-02 4.97519001270e-02 - 4.22555839350e-02 3.37861689888e-02 2.56667001275e-02 1.88953143126e-02 1.37978854292e-02 1.01850325403e-02 - 7.72761616008e-03 6.18689298800e-03 5.44101481998e-03 5.44101481998e-03 6.18689298800e-03 7.72761616008e-03 - 1.01850325403e-02 1.37978854292e-02 1.88953143126e-02 2.56667001275e-02 3.37861689888e-02 4.22555839350e-02 - 4.90171582760e-02 5.39959847166e-02 5.57693346902e-02 5.39959847166e-02 4.90171582760e-02 4.17752511891e-02 - 3.35736913190e-02 2.57020582842e-02 1.90473786081e-02 1.39177867633e-02 1.01850325403e-02 7.57424593471e-03 - 5.85659906617e-03 4.88536934411e-03 4.57233815477e-03 4.88536934411e-03 5.85659906617e-03 7.57424593471e-03 - 1.01850325403e-02 1.39177867633e-02 1.90473786081e-02 2.57020582842e-02 3.35736913190e-02 4.17752511891e-02 - 4.60708732894e-02 4.94024317434e-02 4.94024317434e-02 4.60708732894e-02 4.01880472274e-02 3.30014183614e-02 - 2.57739478448e-02 1.94109424364e-02 1.43139225142e-02 1.04809874105e-02 7.72761616008e-03 5.85659906617e-03 - 4.71220524742e-03 4.17601417174e-03 4.17601417174e-03 4.71220524742e-03 5.85659906617e-03 7.72761616008e-03 - 1.04809874105e-02 1.43139225142e-02 1.94109424364e-02 2.57739478448e-02 3.30014183614e-02 4.01880472274e-02 - 4.17937584090e-02 4.32645707778e-02 4.17937584090e-02 3.77513071960e-02 3.20717125323e-02 2.58559673304e-02 - 2.00031512771e-02 1.50351544907e-02 1.11222486043e-02 8.21703388740e-03 6.18689298800e-03 4.88536934411e-03 - 4.17601417174e-03 3.95332068977e-03 4.17601417174e-03 4.88536934411e-03 6.18689298800e-03 8.21703388740e-03 - 1.11222486043e-02 1.50351544907e-02 2.00031512771e-02 2.58559673304e-02 3.20717125323e-02 3.77513071960e-02 - 3.73919347545e-02 3.73919347545e-02 3.50622322206e-02 3.09544991478e-02 2.59124227944e-02 2.07503567214e-02 - 1.60537467662e-02 1.21379949101e-02 9.10176004354e-03 6.90595653158e-03 5.44101481998e-03 4.57233815477e-03 - 4.17601417174e-03 4.17601417174e-03 4.57233815477e-03 5.44101481998e-03 6.90595653158e-03 9.10176004354e-03 - 1.21379949101e-02 1.60537467662e-02 2.07503567214e-02 2.59124227944e-02 3.09544991478e-02 3.50622322206e-02 - 3.35320462758e-02 3.25477771477e-02 2.98227515673e-02 2.59309064966e-02 2.15543530804e-02 1.72789698058e-02 - 1.35012692014e-02 1.04248538239e-02 8.09940143038e-03 6.47303924297e-03 5.44101481998e-03 4.88536934411e-03 - 4.71220524742e-03 4.88536934411e-03 5.44101481998e-03 6.47303924297e-03 8.09940143038e-03 1.04248538239e-02 - 1.35012692014e-02 1.72789698058e-02 2.15543530804e-02 2.59309064966e-02 2.98227515673e-02 3.25477771477e-02 - 3.03107984462e-02 2.87349610469e-02 2.59124227944e-02 2.23681861845e-02 1.86453765526e-02 1.51716517813e-02 - 1.22047713863e-02 9.84753974693e-03 8.09940143038e-03 6.90595653158e-03 6.18689298800e-03 5.85659906617e-03 - 5.85659906617e-03 6.18689298800e-03 6.90595653158e-03 8.09940143038e-03 9.84753974693e-03 1.22047713863e-02 - 1.51716517813e-02 1.86453765526e-02 2.23681861845e-02 2.59124227944e-02 2.87349610469e-02 3.03107984462e-02 - 2.76901755249e-02 2.58559673304e-02 2.31896632144e-02 2.01439779904e-02 1.71372885777e-02 1.44498895771e-02 - 1.22047713863e-02 1.04248538239e-02 9.10176004354e-03 8.21703388740e-03 7.72761616008e-03 7.57424593471e-03 - 7.72761616008e-03 8.21703388740e-03 9.10176004354e-03 1.04248538239e-02 1.22047713863e-02 1.44498895771e-02 - 1.71372885777e-02 2.01439779904e-02 2.31896632144e-02 2.58559673304e-02 2.76901755249e-02 2.83448586606e-02 - 2.57739478448e-02 2.40042344862e-02 2.17534458677e-02 1.93739445237e-02 1.71372885777e-02 1.51716517813e-02 - 1.35012692014e-02 1.21379949101e-02 1.11222486043e-02 1.04809874105e-02 1.01850325403e-02 1.01850325403e-02 - 1.04809874105e-02 1.11222486043e-02 1.21379949101e-02 1.35012692014e-02 1.51716517813e-02 1.71372885777e-02 - 1.93739445237e-02 2.17534458677e-02 2.40042344862e-02 2.57739478448e-02 2.67482412330e-02 2.67482412330e-02 - 2.47523565945e-02 2.33592944544e-02 2.17534458677e-02 2.01439779904e-02 1.86453765526e-02 1.72789698058e-02 - 1.60537467662e-02 1.50351544907e-02 1.43139225142e-02 1.39177867633e-02 1.37978854292e-02 1.39177867633e-02 - 1.43139225142e-02 1.50351544907e-02 1.60537467662e-02 1.72789698058e-02 1.86453765526e-02 2.01439779904e-02 - 2.17534458677e-02 2.33592944544e-02 2.47523565945e-02 2.57020582842e-02 2.60389015581e-02 2.57020582842e-02 - 2.47523565945e-02 2.40042344862e-02 2.31896632144e-02 2.23681861845e-02 2.15543530804e-02 2.07503567214e-02 - 2.00031512771e-02 1.94109424364e-02 1.90473786081e-02 1.88953143126e-02 1.88953143126e-02 1.90473786081e-02 - 1.94109424364e-02 2.00031512771e-02 2.07503567214e-02 2.15543530804e-02 2.23681861845e-02 2.31896632144e-02 - 2.40042344862e-02 2.47523565945e-02 2.53397129065e-02 2.56667001275e-02 2.56667001275e-02 2.53397129065e-02 - 2.57739478448e-02 2.58559673304e-02 2.59124227944e-02 2.59309064966e-02 2.59124227944e-02 2.58559673304e-02 - 2.57739478448e-02 2.57020582842e-02 2.56667001275e-02 2.56591838074e-02 2.56667001275e-02 2.57020582842e-02 - 2.57739478448e-02 2.58559673304e-02 2.59124227944e-02 2.59309064966e-02 2.59124227944e-02 2.58559673304e-02 - 2.57739478448e-02 2.57020582842e-02 2.56667001275e-02 2.56591838074e-02 2.56667001275e-02 2.57020582842e-02 - 2.76901755249e-02 2.87349610469e-02 2.98227515673e-02 3.09544991478e-02 3.20717125323e-02 3.30014183614e-02 - 3.35736913190e-02 3.37861689888e-02 3.38082178713e-02 3.38082178713e-02 3.37861689888e-02 3.35736913190e-02 - 3.30014183614e-02 3.20717125323e-02 3.09544991478e-02 2.98227515673e-02 2.87349610469e-02 2.76901755249e-02 - 2.67482412330e-02 2.60389015581e-02 2.56667001275e-02 2.56667001275e-02 2.60389015581e-02 2.67482412330e-02 - 3.03107984462e-02 3.25477771477e-02 3.50622322206e-02 3.77513071960e-02 4.01880472274e-02 4.17752511891e-02 - 4.22555839350e-02 4.20317983678e-02 4.18365826600e-02 4.20317983678e-02 4.22555839350e-02 4.17752511891e-02 - 4.01880472274e-02 3.77513071960e-02 3.50622322206e-02 3.25477771477e-02 3.03107984462e-02 2.83448586606e-02 - 2.67482412330e-02 2.57020582842e-02 2.53397129065e-02 2.57020582842e-02 2.67482412330e-02 2.83448586606e-02 - 2.89819100992e-02 3.34978085182e-02 3.86408229568e-02 4.37078082031e-02 4.76311652618e-02 4.97519001270e-02 - 5.03920982744e-02 5.04488982123e-02 5.03920982744e-02 4.97519001270e-02 4.76311652618e-02 4.37078082031e-02 - 3.86408229568e-02 3.34978085182e-02 2.89819100992e-02 2.52525868671e-02 2.22527540382e-02 2.00031512771e-02 - 1.86064391176e-02 1.81349712987e-02 1.86064391176e-02 2.00031512771e-02 2.22527540382e-02 2.52525868671e-02 - 3.34978085182e-02 3.99801891127e-02 4.67604720778e-02 5.22694162731e-02 5.51986766687e-02 5.57017474383e-02 - 5.52660749302e-02 5.52660749302e-02 5.57017474383e-02 5.51986766687e-02 5.22694162731e-02 4.67604720778e-02 - 3.99801891127e-02 3.34978085182e-02 2.81262227272e-02 2.39287446557e-02 2.07503567214e-02 1.85584096087e-02 - 1.74298422700e-02 1.74298422700e-02 1.85584096087e-02 2.07503567214e-02 2.39287446557e-02 2.81262227272e-02 - 3.86408229568e-02 4.67604720778e-02 5.39094660610e-02 5.80220562298e-02 5.86480861577e-02 5.74779615393e-02 - 5.67646475266e-02 5.74779615393e-02 5.86480861577e-02 5.80220562298e-02 5.39094660610e-02 4.67604720778e-02 - 3.86408229568e-02 3.14092023629e-02 2.57560150434e-02 2.15543530804e-02 1.85555754828e-02 1.67052485052e-02 - 1.60740086611e-02 1.67052485052e-02 1.85555754828e-02 2.15543530804e-02 2.57560150434e-02 3.14092023629e-02 - 4.37078082031e-02 5.22694162731e-02 5.80220562298e-02 5.95705865689e-02 5.81927448576e-02 5.65857611912e-02 - 5.65857611912e-02 5.81927448576e-02 5.95705865689e-02 5.80220562298e-02 5.22694162731e-02 4.37078082031e-02 - 3.49472650912e-02 2.77206217649e-02 2.23681861845e-02 1.85605049655e-02 1.60207976494e-02 1.47188141408e-02 - 1.47188141408e-02 1.60207976494e-02 1.85605049655e-02 2.23681861845e-02 2.77206217649e-02 3.49472650912e-02 - 4.76311652618e-02 5.51986766687e-02 5.86480861577e-02 5.81927448576e-02 5.63481382706e-02 5.54677921285e-02 - 5.63481382706e-02 5.81927448576e-02 5.86480861577e-02 5.51986766687e-02 4.76311652618e-02 3.82730721923e-02 - 2.97324869747e-02 2.31896632144e-02 1.85555754828e-02 1.53937416744e-02 1.34835937042e-02 1.28345023601e-02 - 1.34835937042e-02 1.53937416744e-02 1.85555754828e-02 2.31896632144e-02 2.97324869747e-02 3.82730721923e-02 - 4.97519001270e-02 5.57017474383e-02 5.74779615393e-02 5.65857611912e-02 5.54677921285e-02 5.54677921285e-02 - 5.65857611912e-02 5.74779615393e-02 5.57017474383e-02 4.97519001270e-02 4.08013474350e-02 3.15640784817e-02 - 2.40042344862e-02 1.85584096087e-02 1.48305747723e-02 1.24193280193e-02 1.12010383875e-02 1.12010383875e-02 - 1.24193280193e-02 1.48305747723e-02 1.85584096087e-02 2.40042344862e-02 3.15640784817e-02 4.08013474350e-02 - 5.03920982744e-02 5.52660749302e-02 5.67646475266e-02 5.65857611912e-02 5.63481382706e-02 5.65857611912e-02 - 5.67646475266e-02 5.52660749302e-02 5.03920982744e-02 4.22555839350e-02 3.29591336478e-02 2.47523565945e-02 - 1.86064391176e-02 1.43606388158e-02 1.15426690136e-02 9.87827853279e-03 9.31927456850e-03 9.87827853279e-03 - 1.15426690136e-02 1.43606388158e-02 1.86064391176e-02 2.47523565945e-02 3.29591336478e-02 4.22555839350e-02 - 5.04488982123e-02 5.52660749302e-02 5.74779615393e-02 5.81927448576e-02 5.81927448576e-02 5.74779615393e-02 - 5.52660749302e-02 5.04488982123e-02 4.28256060156e-02 3.37861689888e-02 2.53397129065e-02 1.87149718700e-02 - 1.40306458540e-02 1.08721475193e-02 8.87112379151e-03 7.87995370977e-03 7.87995370977e-03 8.87112379151e-03 - 1.08721475193e-02 1.40306458540e-02 1.87149718700e-02 2.53397129065e-02 3.37861689888e-02 4.28256060156e-02 - 5.03920982744e-02 5.57017474383e-02 5.86480861577e-02 5.95705865689e-02 5.86480861577e-02 5.57017474383e-02 - 5.03920982744e-02 4.28256060156e-02 3.40527023771e-02 2.56667001275e-02 1.88392765173e-02 1.38569882819e-02 - 1.04242770349e-02 8.16346419979e-03 6.85502201162e-03 6.42332605072e-03 6.85502201162e-03 8.16346419979e-03 - 1.04242770349e-02 1.38569882819e-02 1.88392765173e-02 2.56667001275e-02 3.40527023771e-02 4.28256060156e-02 - 4.97519001270e-02 5.51986766687e-02 5.80220562298e-02 5.80220562298e-02 5.51986766687e-02 4.97519001270e-02 - 4.22555839350e-02 3.37861689888e-02 2.56667001275e-02 1.88953143126e-02 1.37978854292e-02 1.01850325403e-02 - 7.72761616008e-03 6.18689298800e-03 5.44101481998e-03 5.44101481998e-03 6.18689298800e-03 7.72761616008e-03 - 1.01850325403e-02 1.37978854292e-02 1.88953143126e-02 2.56667001275e-02 3.37861689888e-02 4.22555839350e-02 - 4.76311652618e-02 5.22694162731e-02 5.39094660610e-02 5.22694162731e-02 4.76311652618e-02 4.08013474350e-02 - 3.29591336478e-02 2.53397129065e-02 1.88392765173e-02 1.37978854292e-02 1.01130635710e-02 7.52528842907e-03 - 5.81607896972e-03 4.84662495208e-03 4.53364885369e-03 4.84662495208e-03 5.81607896972e-03 7.52528842907e-03 - 1.01130635710e-02 1.37978854292e-02 1.88392765173e-02 2.53397129065e-02 3.29591336478e-02 4.08013474350e-02 - 4.37078082031e-02 4.67604720778e-02 4.67604720778e-02 4.37078082031e-02 3.82730721923e-02 3.15640784817e-02 - 2.47523565945e-02 1.87149718700e-02 1.38569882819e-02 1.01850325403e-02 7.52528842907e-03 5.69805609664e-03 - 4.56877489732e-03 4.03593630205e-03 4.03593630205e-03 4.56877489732e-03 5.69805609664e-03 7.52528842907e-03 - 1.01850325403e-02 1.38569882819e-02 1.87149718700e-02 2.47523565945e-02 3.15640784817e-02 3.82730721923e-02 - 3.86408229568e-02 3.99801891127e-02 3.86408229568e-02 3.49472650912e-02 2.97324869747e-02 2.40042344862e-02 - 1.86064391176e-02 1.40306458540e-02 1.04242770349e-02 7.72761616008e-03 5.81607896972e-03 4.56877489732e-03 - 3.87834628559e-03 3.65964058851e-03 3.87834628559e-03 4.56877489732e-03 5.81607896972e-03 7.72761616008e-03 - 1.04242770349e-02 1.40306458540e-02 1.86064391176e-02 2.40042344862e-02 2.97324869747e-02 3.49472650912e-02 - 3.34978085182e-02 3.34978085182e-02 3.14092023629e-02 2.77206217649e-02 2.31896632144e-02 1.85584096087e-02 - 1.43606388158e-02 1.08721475193e-02 8.16346419979e-03 6.18689298800e-03 4.84662495208e-03 4.03593630205e-03 - 3.65964058851e-03 3.65964058851e-03 4.03593630205e-03 4.84662495208e-03 6.18689298800e-03 8.16346419979e-03 - 1.08721475193e-02 1.43606388158e-02 1.85584096087e-02 2.31896632144e-02 2.77206217649e-02 3.14092023629e-02 - 2.89819100992e-02 2.81262227272e-02 2.57560150434e-02 2.23681861845e-02 1.85555754828e-02 1.48305747723e-02 - 1.15426690136e-02 8.87112379151e-03 6.85502201162e-03 5.44101481998e-03 4.53364885369e-03 4.03593630205e-03 - 3.87834628559e-03 4.03593630205e-03 4.53364885369e-03 5.44101481998e-03 6.85502201162e-03 8.87112379151e-03 - 1.15426690136e-02 1.48305747723e-02 1.85555754828e-02 2.23681861845e-02 2.57560150434e-02 2.81262227272e-02 - 2.52525868671e-02 2.39287446557e-02 2.15543530804e-02 1.85605049655e-02 1.53937416744e-02 1.24193280193e-02 - 9.87827853279e-03 7.87995370977e-03 6.42332605072e-03 5.44101481998e-03 4.84662495208e-03 4.56877489732e-03 - 4.56877489732e-03 4.84662495208e-03 5.44101481998e-03 6.42332605072e-03 7.87995370977e-03 9.87827853279e-03 - 1.24193280193e-02 1.53937416744e-02 1.85605049655e-02 2.15543530804e-02 2.39287446557e-02 2.52525868671e-02 - 2.22527540382e-02 2.07503567214e-02 1.85555754828e-02 1.60207976494e-02 1.34835937042e-02 1.12010383875e-02 - 9.31927456850e-03 7.87995370977e-03 6.85502201162e-03 6.18689298800e-03 5.81607896972e-03 5.69805609664e-03 - 5.81607896972e-03 6.18689298800e-03 6.85502201162e-03 7.87995370977e-03 9.31927456850e-03 1.12010383875e-02 - 1.34835937042e-02 1.60207976494e-02 1.85555754828e-02 2.07503567214e-02 2.22527540382e-02 2.27884625410e-02 - 2.00031512771e-02 1.85584096087e-02 1.67052485052e-02 1.47188141408e-02 1.28345023601e-02 1.12010383875e-02 - 9.87827853279e-03 8.87112379151e-03 8.16346419979e-03 7.72761616008e-03 7.52528842907e-03 7.52528842907e-03 - 7.72761616008e-03 8.16346419979e-03 8.87112379151e-03 9.87827853279e-03 1.12010383875e-02 1.28345023601e-02 - 1.47188141408e-02 1.67052485052e-02 1.85584096087e-02 2.00031512771e-02 2.07965618706e-02 2.07965618706e-02 - 1.86064391176e-02 1.74298422700e-02 1.60740086611e-02 1.47188141408e-02 1.34835937042e-02 1.24193280193e-02 - 1.15426690136e-02 1.08721475193e-02 1.04242770349e-02 1.01850325403e-02 1.01130635710e-02 1.01850325403e-02 - 1.04242770349e-02 1.08721475193e-02 1.15426690136e-02 1.24193280193e-02 1.34835937042e-02 1.47188141408e-02 - 1.60740086611e-02 1.74298422700e-02 1.86064391176e-02 1.94109424364e-02 1.96971752186e-02 1.94109424364e-02 - 1.81349712987e-02 1.74298422700e-02 1.67052485052e-02 1.60207976494e-02 1.53937416744e-02 1.48305747723e-02 - 1.43606388158e-02 1.40306458540e-02 1.38569882819e-02 1.37978854292e-02 1.37978854292e-02 1.38569882819e-02 - 1.40306458540e-02 1.43606388158e-02 1.48305747723e-02 1.53937416744e-02 1.60207976494e-02 1.67052485052e-02 - 1.74298422700e-02 1.81349712987e-02 1.87149718700e-02 1.90473786081e-02 1.90473786081e-02 1.87149718700e-02 - 1.86064391176e-02 1.85584096087e-02 1.85555754828e-02 1.85605049655e-02 1.85555754828e-02 1.85584096087e-02 - 1.86064391176e-02 1.87149718700e-02 1.88392765173e-02 1.88953143126e-02 1.88392765173e-02 1.87149718700e-02 - 1.86064391176e-02 1.85584096087e-02 1.85555754828e-02 1.85605049655e-02 1.85555754828e-02 1.85584096087e-02 - 1.86064391176e-02 1.87149718700e-02 1.88392765173e-02 1.88953143126e-02 1.88392765173e-02 1.87149718700e-02 - 2.00031512771e-02 2.07503567214e-02 2.15543530804e-02 2.23681861845e-02 2.31896632144e-02 2.40042344862e-02 - 2.47523565945e-02 2.53397129065e-02 2.56667001275e-02 2.56667001275e-02 2.53397129065e-02 2.47523565945e-02 - 2.40042344862e-02 2.31896632144e-02 2.23681861845e-02 2.15543530804e-02 2.07503567214e-02 2.00031512771e-02 - 1.94109424364e-02 1.90473786081e-02 1.88953143126e-02 1.88953143126e-02 1.90473786081e-02 1.94109424364e-02 - 2.22527540382e-02 2.39287446557e-02 2.57560150434e-02 2.77206217649e-02 2.97324869747e-02 3.15640784817e-02 - 3.29591336478e-02 3.37861689888e-02 3.40527023771e-02 3.37861689888e-02 3.29591336478e-02 3.15640784817e-02 - 2.97324869747e-02 2.77206217649e-02 2.57560150434e-02 2.39287446557e-02 2.22527540382e-02 2.07965618706e-02 - 1.96971752186e-02 1.90473786081e-02 1.88392765173e-02 1.90473786081e-02 1.96971752186e-02 2.07965618706e-02 - 2.52525868671e-02 2.81262227272e-02 3.14092023629e-02 3.49472650912e-02 3.82730721923e-02 4.08013474350e-02 - 4.22555839350e-02 4.28256060156e-02 4.28256060156e-02 4.22555839350e-02 4.08013474350e-02 3.82730721923e-02 - 3.49472650912e-02 3.14092023629e-02 2.81262227272e-02 2.52525868671e-02 2.27884625410e-02 2.07965618706e-02 - 1.94109424364e-02 1.87149718700e-02 1.87149718700e-02 1.94109424364e-02 2.07965618706e-02 2.27884625410e-02 - 2.61347637909e-02 3.13851597832e-02 3.74820931920e-02 4.37078082031e-02 4.90171582760e-02 5.26323100532e-02 - 5.43728424707e-02 5.43728424707e-02 5.26323100532e-02 4.90171582760e-02 4.37078082031e-02 3.74820931920e-02 - 3.13851597832e-02 2.61347637909e-02 2.19036737434e-02 1.85732113773e-02 1.60537467662e-02 1.43606388158e-02 - 1.35182636086e-02 1.35182636086e-02 1.43606388158e-02 1.60537467662e-02 1.85732113773e-02 2.19036737434e-02 - 3.13851597832e-02 3.87690158276e-02 4.67604720778e-02 5.39959847166e-02 5.92671896593e-02 6.22085036765e-02 - 6.31142666785e-02 6.22085036765e-02 5.92671896593e-02 5.39959847166e-02 4.67604720778e-02 3.87690158276e-02 - 3.13851597832e-02 2.53584894772e-02 2.07351318935e-02 1.72789698058e-02 1.48305747723e-02 1.33587578257e-02 - 1.28681196083e-02 1.33587578257e-02 1.48305747723e-02 1.72789698058e-02 2.07351318935e-02 2.53584894772e-02 - 3.74820931920e-02 4.67604720778e-02 5.57693346902e-02 6.28462967289e-02 6.72242960853e-02 6.91691506706e-02 - 6.91691506706e-02 6.72242960853e-02 6.28462967289e-02 5.57693346902e-02 4.67604720778e-02 3.74820931920e-02 - 2.94353177762e-02 2.32096903098e-02 1.86453765526e-02 1.53937416744e-02 1.32551887233e-02 1.21839358632e-02 - 1.21839358632e-02 1.32551887233e-02 1.53937416744e-02 1.86453765526e-02 2.32096903098e-02 2.94353177762e-02 - 4.37078082031e-02 5.39959847166e-02 6.28462967289e-02 6.89438173057e-02 7.22242608078e-02 7.32259633424e-02 - 7.22242608078e-02 6.89438173057e-02 6.28462967289e-02 5.39959847166e-02 4.37078082031e-02 3.39275143754e-02 - 2.59885128162e-02 2.01439779904e-02 1.60207976494e-02 1.32196503024e-02 1.15565994691e-02 1.10006986257e-02 - 1.15565994691e-02 1.32196503024e-02 1.60207976494e-02 2.01439779904e-02 2.59885128162e-02 3.39275143754e-02 - 4.90171582760e-02 5.92671896593e-02 6.72242960853e-02 7.22242608078e-02 7.45491088161e-02 7.45491088161e-02 - 7.22242608078e-02 6.72242960853e-02 5.92671896593e-02 4.90171582760e-02 3.82730721923e-02 2.88969153857e-02 - 2.17534458677e-02 1.67052485052e-02 1.32551887233e-02 1.10438280884e-02 9.94413016678e-03 9.94413016678e-03 - 1.10438280884e-02 1.32551887233e-02 1.67052485052e-02 2.17534458677e-02 2.88969153857e-02 3.82730721923e-02 - 5.26323100532e-02 6.22085036765e-02 6.91691506706e-02 7.32259633424e-02 7.45491088161e-02 7.32259633424e-02 - 6.91691506706e-02 6.22085036765e-02 5.26323100532e-02 4.17752511891e-02 3.15640784817e-02 2.33592944544e-02 - 1.74298422700e-02 1.33587578257e-02 1.06646265060e-02 9.09041352545e-03 8.56724324800e-03 9.09041352545e-03 - 1.06646265060e-02 1.33587578257e-02 1.74298422700e-02 2.33592944544e-02 3.15640784817e-02 4.17752511891e-02 - 5.43728424707e-02 6.31142666785e-02 6.91691506706e-02 7.22242608078e-02 7.22242608078e-02 6.91691506706e-02 - 6.31142666785e-02 5.43728424707e-02 4.39611245421e-02 3.35736913190e-02 2.47523565945e-02 1.81349712987e-02 - 1.35182636086e-02 1.04118462818e-02 8.45428906874e-03 7.49319081292e-03 7.49319081292e-03 8.45428906874e-03 - 1.04118462818e-02 1.35182636086e-02 1.81349712987e-02 2.47523565945e-02 3.35736913190e-02 4.39611245421e-02 - 5.43728424707e-02 6.22085036765e-02 6.72242960853e-02 6.89438173057e-02 6.72242960853e-02 6.22085036765e-02 - 5.43728424707e-02 4.46898454449e-02 3.46378859300e-02 2.57020582842e-02 1.87149718700e-02 1.37029642554e-02 - 1.02670781727e-02 8.01216613758e-03 6.71488304626e-03 6.28902224701e-03 6.71488304626e-03 8.01216613758e-03 - 1.02670781727e-02 1.37029642554e-02 1.87149718700e-02 2.57020582842e-02 3.46378859300e-02 4.46898454449e-02 - 5.26323100532e-02 5.92671896593e-02 6.28462967289e-02 6.28462967289e-02 5.92671896593e-02 5.26323100532e-02 - 4.39611245421e-02 3.46378859300e-02 2.60389015581e-02 1.90473786081e-02 1.38569882819e-02 1.02031841081e-02 - 7.72923129933e-03 6.18550458410e-03 5.44128685160e-03 5.44128685160e-03 6.18550458410e-03 7.72923129933e-03 - 1.02031841081e-02 1.38569882819e-02 1.90473786081e-02 2.60389015581e-02 3.46378859300e-02 4.39611245421e-02 - 4.90171582760e-02 5.39959847166e-02 5.57693346902e-02 5.39959847166e-02 4.90171582760e-02 4.17752511891e-02 - 3.35736913190e-02 2.57020582842e-02 1.90473786081e-02 1.39177867633e-02 1.01850325403e-02 7.57424593471e-03 - 5.85659906617e-03 4.88536934411e-03 4.57233815477e-03 4.88536934411e-03 5.85659906617e-03 7.57424593471e-03 - 1.01850325403e-02 1.39177867633e-02 1.90473786081e-02 2.57020582842e-02 3.35736913190e-02 4.17752511891e-02 - 4.37078082031e-02 4.67604720778e-02 4.67604720778e-02 4.37078082031e-02 3.82730721923e-02 3.15640784817e-02 - 2.47523565945e-02 1.87149718700e-02 1.38569882819e-02 1.01850325403e-02 7.52528842907e-03 5.69805609664e-03 - 4.56877489732e-03 4.03593630205e-03 4.03593630205e-03 4.56877489732e-03 5.69805609664e-03 7.52528842907e-03 - 1.01850325403e-02 1.38569882819e-02 1.87149718700e-02 2.47523565945e-02 3.15640784817e-02 3.82730721923e-02 - 3.74820931920e-02 3.87690158276e-02 3.74820931920e-02 3.39275143754e-02 2.88969153857e-02 2.33592944544e-02 - 1.81349712987e-02 1.37029642554e-02 1.02031841081e-02 7.57424593471e-03 5.69805609664e-03 4.46542762661e-03 - 3.77975659219e-03 3.56204422580e-03 3.77975659219e-03 4.46542762661e-03 5.69805609664e-03 7.57424593471e-03 - 1.02031841081e-02 1.37029642554e-02 1.81349712987e-02 2.33592944544e-02 2.88969153857e-02 3.39275143754e-02 - 3.13851597832e-02 3.13851597832e-02 2.94353177762e-02 2.59885128162e-02 2.17534458677e-02 1.74298422700e-02 - 1.35182636086e-02 1.02670781727e-02 7.72923129933e-03 5.85659906617e-03 4.56877489732e-03 3.77975659219e-03 - 3.41039424489e-03 3.41039424489e-03 3.77975659219e-03 4.56877489732e-03 5.85659906617e-03 7.72923129933e-03 - 1.02670781727e-02 1.35182636086e-02 1.74298422700e-02 2.17534458677e-02 2.59885128162e-02 2.94353177762e-02 - 2.61347637909e-02 2.53584894772e-02 2.32096903098e-02 2.01439779904e-02 1.67052485052e-02 1.33587578257e-02 - 1.04118462818e-02 8.01216613758e-03 6.18550458410e-03 4.88536934411e-03 4.03593630205e-03 3.56204422580e-03 - 3.41039424489e-03 3.56204422580e-03 4.03593630205e-03 4.88536934411e-03 6.18550458410e-03 8.01216613758e-03 - 1.04118462818e-02 1.33587578257e-02 1.67052485052e-02 2.01439779904e-02 2.32096903098e-02 2.53584894772e-02 - 2.19036737434e-02 2.07351318935e-02 1.86453765526e-02 1.60207976494e-02 1.32551887233e-02 1.06646265060e-02 - 8.45428906874e-03 6.71488304626e-03 5.44128685160e-03 4.57233815477e-03 4.03593630205e-03 3.77975659219e-03 - 3.77975659219e-03 4.03593630205e-03 4.57233815477e-03 5.44128685160e-03 6.71488304626e-03 8.45428906874e-03 - 1.06646265060e-02 1.32551887233e-02 1.60207976494e-02 1.86453765526e-02 2.07351318935e-02 2.19036737434e-02 - 1.85732113773e-02 1.72789698058e-02 1.53937416744e-02 1.32196503024e-02 1.10438280884e-02 9.09041352545e-03 - 7.49319081292e-03 6.28902224701e-03 5.44128685160e-03 4.88536934411e-03 4.56877489732e-03 4.46542762661e-03 - 4.56877489732e-03 4.88536934411e-03 5.44128685160e-03 6.28902224701e-03 7.49319081292e-03 9.09041352545e-03 - 1.10438280884e-02 1.32196503024e-02 1.53937416744e-02 1.72789698058e-02 1.85732113773e-02 1.90357893277e-02 - 1.60537467662e-02 1.48305747723e-02 1.32551887233e-02 1.15565994691e-02 9.94413016678e-03 8.56724324800e-03 - 7.49319081292e-03 6.71488304626e-03 6.18550458410e-03 5.85659906617e-03 5.69805609664e-03 5.69805609664e-03 - 5.85659906617e-03 6.18550458410e-03 6.71488304626e-03 7.49319081292e-03 8.56724324800e-03 9.94413016678e-03 - 1.15565994691e-02 1.32551887233e-02 1.48305747723e-02 1.60537467662e-02 1.67247696023e-02 1.67247696023e-02 - 1.43606388158e-02 1.33587578257e-02 1.21839358632e-02 1.10006986257e-02 9.94413016678e-03 9.09041352545e-03 - 8.45428906874e-03 8.01216613758e-03 7.72923129933e-03 7.57424593471e-03 7.52528842907e-03 7.57424593471e-03 - 7.72923129933e-03 8.01216613758e-03 8.45428906874e-03 9.09041352545e-03 9.94413016678e-03 1.10006986257e-02 - 1.21839358632e-02 1.33587578257e-02 1.43606388158e-02 1.50351544907e-02 1.52731342228e-02 1.50351544907e-02 - 1.35182636086e-02 1.28681196083e-02 1.21839358632e-02 1.15565994691e-02 1.10438280884e-02 1.06646265060e-02 - 1.04118462818e-02 1.02670781727e-02 1.02031841081e-02 1.01850325403e-02 1.01850325403e-02 1.02031841081e-02 - 1.02670781727e-02 1.04118462818e-02 1.06646265060e-02 1.10438280884e-02 1.15565994691e-02 1.21839358632e-02 - 1.28681196083e-02 1.35182636086e-02 1.40306458540e-02 1.43139225142e-02 1.43139225142e-02 1.40306458540e-02 - 1.35182636086e-02 1.33587578257e-02 1.32551887233e-02 1.32196503024e-02 1.32551887233e-02 1.33587578257e-02 - 1.35182636086e-02 1.37029642554e-02 1.38569882819e-02 1.39177867633e-02 1.38569882819e-02 1.37029642554e-02 - 1.35182636086e-02 1.33587578257e-02 1.32551887233e-02 1.32196503024e-02 1.32551887233e-02 1.33587578257e-02 - 1.35182636086e-02 1.37029642554e-02 1.38569882819e-02 1.39177867633e-02 1.38569882819e-02 1.37029642554e-02 - 1.43606388158e-02 1.48305747723e-02 1.53937416744e-02 1.60207976494e-02 1.67052485052e-02 1.74298422700e-02 - 1.81349712987e-02 1.87149718700e-02 1.90473786081e-02 1.90473786081e-02 1.87149718700e-02 1.81349712987e-02 - 1.74298422700e-02 1.67052485052e-02 1.60207976494e-02 1.53937416744e-02 1.48305747723e-02 1.43606388158e-02 - 1.40306458540e-02 1.38569882819e-02 1.37978854292e-02 1.37978854292e-02 1.38569882819e-02 1.40306458540e-02 - 1.60537467662e-02 1.72789698058e-02 1.86453765526e-02 2.01439779904e-02 2.17534458677e-02 2.33592944544e-02 - 2.47523565945e-02 2.57020582842e-02 2.60389015581e-02 2.57020582842e-02 2.47523565945e-02 2.33592944544e-02 - 2.17534458677e-02 2.01439779904e-02 1.86453765526e-02 1.72789698058e-02 1.60537467662e-02 1.50351544907e-02 - 1.43139225142e-02 1.39177867633e-02 1.37978854292e-02 1.39177867633e-02 1.43139225142e-02 1.50351544907e-02 - 1.85732113773e-02 2.07351318935e-02 2.32096903098e-02 2.59885128162e-02 2.88969153857e-02 3.15640784817e-02 - 3.35736913190e-02 3.46378859300e-02 3.46378859300e-02 3.35736913190e-02 3.15640784817e-02 2.88969153857e-02 - 2.59885128162e-02 2.32096903098e-02 2.07351318935e-02 1.85732113773e-02 1.67247696023e-02 1.52731342228e-02 - 1.43139225142e-02 1.38569882819e-02 1.38569882819e-02 1.43139225142e-02 1.52731342228e-02 1.67247696023e-02 - 2.19036737434e-02 2.53584894772e-02 2.94353177762e-02 3.39275143754e-02 3.82730721923e-02 4.17752511891e-02 - 4.39611245421e-02 4.46898454449e-02 4.39611245421e-02 4.17752511891e-02 3.82730721923e-02 3.39275143754e-02 - 2.94353177762e-02 2.53584894772e-02 2.19036737434e-02 1.90357893277e-02 1.67247696023e-02 1.50351544907e-02 - 1.40306458540e-02 1.37029642554e-02 1.40306458540e-02 1.50351544907e-02 1.67247696023e-02 1.90357893277e-02 - 2.51358351357e-02 3.13851597832e-02 3.86408229568e-02 4.60708732894e-02 5.24827367699e-02 5.67802908281e-02 - 5.82846684414e-02 5.67802908281e-02 5.24827367699e-02 4.60708732894e-02 3.86408229568e-02 3.13851597832e-02 - 2.51358351357e-02 2.01594924058e-02 1.63414636372e-02 1.35012692014e-02 1.15426690136e-02 1.04118462818e-02 - 1.00461401111e-02 1.04118462818e-02 1.15426690136e-02 1.35012692014e-02 1.63414636372e-02 2.01594924058e-02 - 3.13851597832e-02 3.99801891127e-02 4.94024317434e-02 5.82637609313e-02 6.50724984809e-02 6.87390140452e-02 - 6.87390140452e-02 6.50724984809e-02 5.82637609313e-02 4.94024317434e-02 3.99801891127e-02 3.13851597832e-02 - 2.43885514961e-02 1.90731200357e-02 1.51716517813e-02 1.24193280193e-02 1.06646265060e-02 9.81666041939e-03 - 9.81666041939e-03 1.06646265060e-02 1.24193280193e-02 1.51716517813e-02 1.90731200357e-02 2.43885514961e-02 - 3.86408229568e-02 4.94024317434e-02 6.03520680545e-02 6.97193973966e-02 7.59717702126e-02 7.81603362307e-02 - 7.59717702126e-02 6.97193973966e-02 6.03520680545e-02 4.94024317434e-02 3.86408229568e-02 2.94353177762e-02 - 2.23208692653e-02 1.71372885777e-02 1.34835937042e-02 1.10438280884e-02 9.64234738867e-03 9.18607976908e-03 - 9.64234738867e-03 1.10438280884e-02 1.34835937042e-02 1.71372885777e-02 2.23208692653e-02 2.94353177762e-02 - 4.60708732894e-02 5.82637609313e-02 6.97193973966e-02 7.85611057280e-02 8.33624452951e-02 8.33624452951e-02 - 7.85611057280e-02 6.97193973966e-02 5.82637609313e-02 4.60708732894e-02 3.49472650912e-02 2.59885128162e-02 - 1.93739445237e-02 1.47188141408e-02 1.15565994691e-02 9.57706447476e-03 8.62132731210e-03 8.62132731210e-03 - 9.57706447476e-03 1.15565994691e-02 1.47188141408e-02 1.93739445237e-02 2.59885128162e-02 3.49472650912e-02 - 5.24827367699e-02 6.50724984809e-02 7.59717702126e-02 8.33624452951e-02 8.59799490006e-02 8.33624452951e-02 - 7.59717702126e-02 6.50724984809e-02 5.24827367699e-02 4.01880472274e-02 2.97324869747e-02 2.17534458677e-02 - 1.60740086611e-02 1.21839358632e-02 9.64234738867e-03 8.19475376688e-03 7.72360398793e-03 8.19475376688e-03 - 9.64234738867e-03 1.21839358632e-02 1.60740086611e-02 2.17534458677e-02 2.97324869747e-02 4.01880472274e-02 - 5.67802908281e-02 6.87390140452e-02 7.81603362307e-02 8.33624452951e-02 8.33624452951e-02 7.81603362307e-02 - 6.87390140452e-02 5.67802908281e-02 4.42738472380e-02 3.30014183614e-02 2.40042344862e-02 1.74298422700e-02 - 1.28681196083e-02 9.81666041939e-03 7.92776657338e-03 7.02039368822e-03 7.02039368822e-03 7.92776657338e-03 - 9.81666041939e-03 1.28681196083e-02 1.74298422700e-02 2.40042344862e-02 3.30014183614e-02 4.42738472380e-02 - 5.82846684414e-02 6.87390140452e-02 7.59717702126e-02 7.85611057280e-02 7.59717702126e-02 6.87390140452e-02 - 5.82846684414e-02 4.64972219765e-02 3.52270678377e-02 2.57739478448e-02 1.86064391176e-02 1.35182636086e-02 - 1.00461401111e-02 7.79360764654e-03 6.52118028642e-03 6.10931604239e-03 6.52118028642e-03 7.79360764654e-03 - 1.00461401111e-02 1.35182636086e-02 1.86064391176e-02 2.57739478448e-02 3.52270678377e-02 4.64972219765e-02 - 5.67802908281e-02 6.50724984809e-02 6.97193973966e-02 6.97193973966e-02 6.50724984809e-02 5.67802908281e-02 - 4.64972219765e-02 3.60143597485e-02 2.67482412330e-02 1.94109424364e-02 1.40306458540e-02 1.02670781727e-02 - 7.74242036277e-03 6.18966814651e-03 5.45146116297e-03 5.45146116297e-03 6.18966814651e-03 7.74242036277e-03 - 1.02670781727e-02 1.40306458540e-02 1.94109424364e-02 2.67482412330e-02 3.60143597485e-02 4.64972219765e-02 - 5.24827367699e-02 5.82637609313e-02 6.03520680545e-02 5.82637609313e-02 5.24827367699e-02 4.42738472380e-02 - 3.52270678377e-02 2.67482412330e-02 1.96971752186e-02 1.43139225142e-02 1.04242770349e-02 7.72923129933e-03 - 5.97825202186e-03 5.00022810356e-03 4.68737681852e-03 5.00022810356e-03 5.97825202186e-03 7.72923129933e-03 - 1.04242770349e-02 1.43139225142e-02 1.96971752186e-02 2.67482412330e-02 3.52270678377e-02 4.42738472380e-02 - 4.60708732894e-02 4.94024317434e-02 4.94024317434e-02 4.60708732894e-02 4.01880472274e-02 3.30014183614e-02 - 2.57739478448e-02 1.94109424364e-02 1.43139225142e-02 1.04809874105e-02 7.72761616008e-03 5.85659906617e-03 - 4.71220524742e-03 4.17601417174e-03 4.17601417174e-03 4.71220524742e-03 5.85659906617e-03 7.72761616008e-03 - 1.04809874105e-02 1.43139225142e-02 1.94109424364e-02 2.57739478448e-02 3.30014183614e-02 4.01880472274e-02 - 3.86408229568e-02 3.99801891127e-02 3.86408229568e-02 3.49472650912e-02 2.97324869747e-02 2.40042344862e-02 - 1.86064391176e-02 1.40306458540e-02 1.04242770349e-02 7.72761616008e-03 5.81607896972e-03 4.56877489732e-03 - 3.87834628559e-03 3.65964058851e-03 3.87834628559e-03 4.56877489732e-03 5.81607896972e-03 7.72761616008e-03 - 1.04242770349e-02 1.40306458540e-02 1.86064391176e-02 2.40042344862e-02 2.97324869747e-02 3.49472650912e-02 - 3.13851597832e-02 3.13851597832e-02 2.94353177762e-02 2.59885128162e-02 2.17534458677e-02 1.74298422700e-02 - 1.35182636086e-02 1.02670781727e-02 7.72923129933e-03 5.85659906617e-03 4.56877489732e-03 3.77975659219e-03 - 3.41039424489e-03 3.41039424489e-03 3.77975659219e-03 4.56877489732e-03 5.85659906617e-03 7.72923129933e-03 - 1.02670781727e-02 1.35182636086e-02 1.74298422700e-02 2.17534458677e-02 2.59885128162e-02 2.94353177762e-02 - 2.51358351357e-02 2.43885514961e-02 2.23208692653e-02 1.93739445237e-02 1.60740086611e-02 1.28681196083e-02 - 1.00461401111e-02 7.74242036277e-03 5.97825202186e-03 4.71220524742e-03 3.87834628559e-03 3.41039424489e-03 - 3.26020314087e-03 3.41039424489e-03 3.87834628559e-03 4.71220524742e-03 5.97825202186e-03 7.74242036277e-03 - 1.00461401111e-02 1.28681196083e-02 1.60740086611e-02 1.93739445237e-02 2.23208692653e-02 2.43885514961e-02 - 2.01594924058e-02 1.90731200357e-02 1.71372885777e-02 1.47188141408e-02 1.21839358632e-02 9.81666041939e-03 - 7.79360764654e-03 6.18966814651e-03 5.00022810356e-03 4.17601417174e-03 3.65964058851e-03 3.41039424489e-03 - 3.41039424489e-03 3.65964058851e-03 4.17601417174e-03 5.00022810356e-03 6.18966814651e-03 7.79360764654e-03 - 9.81666041939e-03 1.21839358632e-02 1.47188141408e-02 1.71372885777e-02 1.90731200357e-02 2.01594924058e-02 - 1.63414636372e-02 1.51716517813e-02 1.34835937042e-02 1.15565994691e-02 9.64234738867e-03 7.92776657338e-03 - 6.52118028642e-03 5.45146116297e-03 4.68737681852e-03 4.17601417174e-03 3.87834628559e-03 3.77975659219e-03 - 3.87834628559e-03 4.17601417174e-03 4.68737681852e-03 5.45146116297e-03 6.52118028642e-03 7.92776657338e-03 - 9.64234738867e-03 1.15565994691e-02 1.34835937042e-02 1.51716517813e-02 1.63414636372e-02 1.67618733320e-02 - 1.35012692014e-02 1.24193280193e-02 1.10438280884e-02 9.57706447476e-03 8.19475376688e-03 7.02039368822e-03 - 6.10931604239e-03 5.45146116297e-03 5.00022810356e-03 4.71220524742e-03 4.56877489732e-03 4.56877489732e-03 - 4.71220524742e-03 5.00022810356e-03 5.45146116297e-03 6.10931604239e-03 7.02039368822e-03 8.19475376688e-03 - 9.57706447476e-03 1.10438280884e-02 1.24193280193e-02 1.35012692014e-02 1.41011278955e-02 1.41011278955e-02 - 1.15426690136e-02 1.06646265060e-02 9.64234738867e-03 8.62132731210e-03 7.72360398793e-03 7.02039368822e-03 - 6.52118028642e-03 6.18966814651e-03 5.97825202186e-03 5.85659906617e-03 5.81607896972e-03 5.85659906617e-03 - 5.97825202186e-03 6.18966814651e-03 6.52118028642e-03 7.02039368822e-03 7.72360398793e-03 8.62132731210e-03 - 9.64234738867e-03 1.06646265060e-02 1.15426690136e-02 1.21379949101e-02 1.23490348915e-02 1.21379949101e-02 - 1.04118462818e-02 9.81666041939e-03 9.18607976908e-03 8.62132731210e-03 8.19475376688e-03 7.92776657338e-03 - 7.79360764654e-03 7.74242036277e-03 7.72923129933e-03 7.72761616008e-03 7.72761616008e-03 7.72923129933e-03 - 7.74242036277e-03 7.79360764654e-03 7.92776657338e-03 8.19475376688e-03 8.62132731210e-03 9.18607976908e-03 - 9.81666041939e-03 1.04118462818e-02 1.08721475193e-02 1.11222486043e-02 1.11222486043e-02 1.08721475193e-02 - 1.00461401111e-02 9.81666041939e-03 9.64234738867e-03 9.57706447476e-03 9.64234738867e-03 9.81666041939e-03 - 1.00461401111e-02 1.02670781727e-02 1.04242770349e-02 1.04809874105e-02 1.04242770349e-02 1.02670781727e-02 - 1.00461401111e-02 9.81666041939e-03 9.64234738867e-03 9.57706447476e-03 9.64234738867e-03 9.81666041939e-03 - 1.00461401111e-02 1.02670781727e-02 1.04242770349e-02 1.04809874105e-02 1.04242770349e-02 1.02670781727e-02 - 1.04118462818e-02 1.06646265060e-02 1.10438280884e-02 1.15565994691e-02 1.21839358632e-02 1.28681196083e-02 - 1.35182636086e-02 1.40306458540e-02 1.43139225142e-02 1.43139225142e-02 1.40306458540e-02 1.35182636086e-02 - 1.28681196083e-02 1.21839358632e-02 1.15565994691e-02 1.10438280884e-02 1.06646265060e-02 1.04118462818e-02 - 1.02670781727e-02 1.02031841081e-02 1.01850325403e-02 1.01850325403e-02 1.02031841081e-02 1.02670781727e-02 - 1.15426690136e-02 1.24193280193e-02 1.34835937042e-02 1.47188141408e-02 1.60740086611e-02 1.74298422700e-02 - 1.86064391176e-02 1.94109424364e-02 1.96971752186e-02 1.94109424364e-02 1.86064391176e-02 1.74298422700e-02 - 1.60740086611e-02 1.47188141408e-02 1.34835937042e-02 1.24193280193e-02 1.15426690136e-02 1.08721475193e-02 - 1.04242770349e-02 1.01850325403e-02 1.01130635710e-02 1.01850325403e-02 1.04242770349e-02 1.08721475193e-02 - 1.35012692014e-02 1.51716517813e-02 1.71372885777e-02 1.93739445237e-02 2.17534458677e-02 2.40042344862e-02 - 2.57739478448e-02 2.67482412330e-02 2.67482412330e-02 2.57739478448e-02 2.40042344862e-02 2.17534458677e-02 - 1.93739445237e-02 1.71372885777e-02 1.51716517813e-02 1.35012692014e-02 1.21379949101e-02 1.11222486043e-02 - 1.04809874105e-02 1.01850325403e-02 1.01850325403e-02 1.04809874105e-02 1.11222486043e-02 1.21379949101e-02 - 1.63414636372e-02 1.90731200357e-02 2.23208692653e-02 2.59885128162e-02 2.97324869747e-02 3.30014183614e-02 - 3.52270678377e-02 3.60143597485e-02 3.52270678377e-02 3.30014183614e-02 2.97324869747e-02 2.59885128162e-02 - 2.23208692653e-02 1.90731200357e-02 1.63414636372e-02 1.41011278955e-02 1.23490348915e-02 1.11222486043e-02 - 1.04242770349e-02 1.02031841081e-02 1.04242770349e-02 1.11222486043e-02 1.23490348915e-02 1.41011278955e-02 - 2.01594924058e-02 2.43885514961e-02 2.94353177762e-02 3.49472650912e-02 4.01880472274e-02 4.42738472380e-02 - 4.64972219765e-02 4.64972219765e-02 4.42738472380e-02 4.01880472274e-02 3.49472650912e-02 2.94353177762e-02 - 2.43885514961e-02 2.01594924058e-02 1.67618733320e-02 1.41011278955e-02 1.21379949101e-02 1.08721475193e-02 - 1.02670781727e-02 1.02670781727e-02 1.08721475193e-02 1.21379949101e-02 1.41011278955e-02 1.67618733320e-02 - 2.61347637909e-02 3.34978085182e-02 4.17937584090e-02 4.99875697106e-02 5.66502458441e-02 6.04031976834e-02 - 6.04031976834e-02 5.66502458441e-02 4.99875697106e-02 4.17937584090e-02 3.34978085182e-02 2.61347637909e-02 - 2.01594924058e-02 1.55749641354e-02 1.22047713863e-02 9.87827853279e-03 8.45428906874e-03 7.79360764654e-03 - 7.79360764654e-03 8.45428906874e-03 9.87827853279e-03 1.22047713863e-02 1.55749641354e-02 2.01594924058e-02 - 3.34978085182e-02 4.32645707778e-02 5.37274523246e-02 6.32743536159e-02 7.00265784056e-02 7.24709839797e-02 - 7.00265784056e-02 6.32743536159e-02 5.37274523246e-02 4.32645707778e-02 3.34978085182e-02 2.53584894772e-02 - 1.90731200357e-02 1.44498895771e-02 1.12010383875e-02 9.09041352545e-03 7.92776657338e-03 7.56111056333e-03 - 7.92776657338e-03 9.09041352545e-03 1.12010383875e-02 1.44498895771e-02 1.90731200357e-02 2.53584894772e-02 - 4.17937584090e-02 5.37274523246e-02 6.56955874028e-02 7.55510577091e-02 8.11535443280e-02 8.11535443280e-02 - 7.55510577091e-02 6.56955874028e-02 5.37274523246e-02 4.17937584090e-02 3.14092023629e-02 2.32096903098e-02 - 1.71372885777e-02 1.28345023601e-02 9.94413016678e-03 8.19475376688e-03 7.38199424752e-03 7.38199424752e-03 - 8.19475376688e-03 9.94413016678e-03 1.28345023601e-02 1.71372885777e-02 2.32096903098e-02 3.14092023629e-02 - 4.99875697106e-02 6.32743536159e-02 7.55510577091e-02 8.43291015351e-02 8.75278859551e-02 8.43291015351e-02 - 7.55510577091e-02 6.32743536159e-02 4.99875697106e-02 3.77513071960e-02 2.77206217649e-02 2.01439779904e-02 - 1.47188141408e-02 1.10006986257e-02 8.62132731210e-03 7.31454489847e-03 6.90130049286e-03 7.31454489847e-03 - 8.62132731210e-03 1.10006986257e-02 1.47188141408e-02 2.01439779904e-02 2.77206217649e-02 3.77513071960e-02 - 5.66502458441e-02 7.00265784056e-02 8.11535443280e-02 8.75278859551e-02 8.75278859551e-02 8.11535443280e-02 - 7.00265784056e-02 5.66502458441e-02 4.34476578087e-02 3.20717125323e-02 2.31896632144e-02 1.67052485052e-02 - 1.21839358632e-02 9.18607976908e-03 7.38199424752e-03 6.54341818442e-03 6.54341818442e-03 7.38199424752e-03 - 9.18607976908e-03 1.21839358632e-02 1.67052485052e-02 2.31896632144e-02 3.20717125323e-02 4.34476578087e-02 - 6.04031976834e-02 7.24709839797e-02 8.11535443280e-02 8.43291015351e-02 8.11535443280e-02 7.24709839797e-02 - 6.04031976834e-02 4.74317203940e-02 3.55533814714e-02 2.58559673304e-02 1.85584096087e-02 1.33587578257e-02 - 9.81666041939e-03 7.56111056333e-03 6.32497949646e-03 5.93416429324e-03 6.32497949646e-03 7.56111056333e-03 - 9.81666041939e-03 1.33587578257e-02 1.85584096087e-02 2.58559673304e-02 3.55533814714e-02 4.74317203940e-02 - 6.04031976834e-02 7.00265784056e-02 7.55510577091e-02 7.55510577091e-02 7.00265784056e-02 6.04031976834e-02 - 4.88652050826e-02 3.74959847737e-02 2.76901755249e-02 2.00031512771e-02 1.43606388158e-02 1.04118462818e-02 - 7.79360764654e-03 6.22225650652e-03 5.49418812711e-03 5.49418812711e-03 6.22225650652e-03 7.79360764654e-03 - 1.04118462818e-02 1.43606388158e-02 2.00031512771e-02 2.76901755249e-02 3.74959847737e-02 4.88652050826e-02 - 5.66502458441e-02 6.32743536159e-02 6.56955874028e-02 6.32743536159e-02 5.66502458441e-02 4.74317203940e-02 - 3.74959847737e-02 2.83448586606e-02 2.07965618706e-02 1.50351544907e-02 1.08721475193e-02 8.01216613758e-03 - 6.18966814651e-03 5.19548069751e-03 4.88266428153e-03 5.19548069751e-03 6.18966814651e-03 8.01216613758e-03 - 1.08721475193e-02 1.50351544907e-02 2.07965618706e-02 2.83448586606e-02 3.74959847737e-02 4.74317203940e-02 - 4.99875697106e-02 5.37274523246e-02 5.37274523246e-02 4.99875697106e-02 4.34476578087e-02 3.55533814714e-02 - 2.76901755249e-02 2.07965618706e-02 1.52731342228e-02 1.11222486043e-02 8.16346419979e-03 6.18550458410e-03 - 5.00022810356e-03 4.45430428341e-03 4.45430428341e-03 5.00022810356e-03 6.18550458410e-03 8.16346419979e-03 - 1.11222486043e-02 1.52731342228e-02 2.07965618706e-02 2.76901755249e-02 3.55533814714e-02 4.34476578087e-02 - 4.17937584090e-02 4.32645707778e-02 4.17937584090e-02 3.77513071960e-02 3.20717125323e-02 2.58559673304e-02 - 2.00031512771e-02 1.50351544907e-02 1.11222486043e-02 8.21703388740e-03 6.18689298800e-03 4.88536934411e-03 - 4.17601417174e-03 3.95332068977e-03 4.17601417174e-03 4.88536934411e-03 6.18689298800e-03 8.21703388740e-03 - 1.11222486043e-02 1.50351544907e-02 2.00031512771e-02 2.58559673304e-02 3.20717125323e-02 3.77513071960e-02 - 3.34978085182e-02 3.34978085182e-02 3.14092023629e-02 2.77206217649e-02 2.31896632144e-02 1.85584096087e-02 - 1.43606388158e-02 1.08721475193e-02 8.16346419979e-03 6.18689298800e-03 4.84662495208e-03 4.03593630205e-03 - 3.65964058851e-03 3.65964058851e-03 4.03593630205e-03 4.84662495208e-03 6.18689298800e-03 8.16346419979e-03 - 1.08721475193e-02 1.43606388158e-02 1.85584096087e-02 2.31896632144e-02 2.77206217649e-02 3.14092023629e-02 - 2.61347637909e-02 2.53584894772e-02 2.32096903098e-02 2.01439779904e-02 1.67052485052e-02 1.33587578257e-02 - 1.04118462818e-02 8.01216613758e-03 6.18550458410e-03 4.88536934411e-03 4.03593630205e-03 3.56204422580e-03 - 3.41039424489e-03 3.56204422580e-03 4.03593630205e-03 4.88536934411e-03 6.18550458410e-03 8.01216613758e-03 - 1.04118462818e-02 1.33587578257e-02 1.67052485052e-02 2.01439779904e-02 2.32096903098e-02 2.53584894772e-02 - 2.01594924058e-02 1.90731200357e-02 1.71372885777e-02 1.47188141408e-02 1.21839358632e-02 9.81666041939e-03 - 7.79360764654e-03 6.18966814651e-03 5.00022810356e-03 4.17601417174e-03 3.65964058851e-03 3.41039424489e-03 - 3.41039424489e-03 3.65964058851e-03 4.17601417174e-03 5.00022810356e-03 6.18966814651e-03 7.79360764654e-03 - 9.81666041939e-03 1.21839358632e-02 1.47188141408e-02 1.71372885777e-02 1.90731200357e-02 2.01594924058e-02 - 1.55749641354e-02 1.44498895771e-02 1.28345023601e-02 1.10006986257e-02 9.18607976908e-03 7.56111056333e-03 - 6.22225650652e-03 5.19548069751e-03 4.45430428341e-03 3.95332068977e-03 3.65964058851e-03 3.56204422580e-03 - 3.65964058851e-03 3.95332068977e-03 4.45430428341e-03 5.19548069751e-03 6.22225650652e-03 7.56111056333e-03 - 9.18607976908e-03 1.10006986257e-02 1.28345023601e-02 1.44498895771e-02 1.55749641354e-02 1.59804507535e-02 - 1.22047713863e-02 1.12010383875e-02 9.94413016678e-03 8.62132731210e-03 7.38199424752e-03 6.32497949646e-03 - 5.49418812711e-03 4.88266428153e-03 4.45430428341e-03 4.17601417174e-03 4.03593630205e-03 4.03593630205e-03 - 4.17601417174e-03 4.45430428341e-03 4.88266428153e-03 5.49418812711e-03 6.32497949646e-03 7.38199424752e-03 - 8.62132731210e-03 9.94413016678e-03 1.12010383875e-02 1.22047713863e-02 1.27676986907e-02 1.27676986907e-02 - 9.87827853279e-03 9.09041352545e-03 8.19475376688e-03 7.31454489847e-03 6.54341818442e-03 5.93416429324e-03 - 5.49418812711e-03 5.19548069751e-03 5.00022810356e-03 4.88536934411e-03 4.84662495208e-03 4.88536934411e-03 - 5.00022810356e-03 5.19548069751e-03 5.49418812711e-03 5.93416429324e-03 6.54341818442e-03 7.31454489847e-03 - 8.19475376688e-03 9.09041352545e-03 9.87827853279e-03 1.04248538239e-02 1.06212947837e-02 1.04248538239e-02 - 8.45428906874e-03 7.92776657338e-03 7.38199424752e-03 6.90130049286e-03 6.54341818442e-03 6.32497949646e-03 - 6.22225650652e-03 6.18966814651e-03 6.18550458410e-03 6.18689298800e-03 6.18689298800e-03 6.18550458410e-03 - 6.18966814651e-03 6.22225650652e-03 6.32497949646e-03 6.54341818442e-03 6.90130049286e-03 7.38199424752e-03 - 7.92776657338e-03 8.45428906874e-03 8.87112379151e-03 9.10176004354e-03 9.10176004354e-03 8.87112379151e-03 - 7.79360764654e-03 7.56111056333e-03 7.38199424752e-03 7.31454489847e-03 7.38199424752e-03 7.56111056333e-03 - 7.79360764654e-03 8.01216613758e-03 8.16346419979e-03 8.21703388740e-03 8.16346419979e-03 8.01216613758e-03 - 7.79360764654e-03 7.56111056333e-03 7.38199424752e-03 7.31454489847e-03 7.38199424752e-03 7.56111056333e-03 - 7.79360764654e-03 8.01216613758e-03 8.16346419979e-03 8.21703388740e-03 8.16346419979e-03 8.01216613758e-03 - 7.79360764654e-03 7.92776657338e-03 8.19475376688e-03 8.62132731210e-03 9.18607976908e-03 9.81666041939e-03 - 1.04118462818e-02 1.08721475193e-02 1.11222486043e-02 1.11222486043e-02 1.08721475193e-02 1.04118462818e-02 - 9.81666041939e-03 9.18607976908e-03 8.62132731210e-03 8.19475376688e-03 7.92776657338e-03 7.79360764654e-03 - 7.74242036277e-03 7.72923129933e-03 7.72761616008e-03 7.72761616008e-03 7.72923129933e-03 7.74242036277e-03 - 8.45428906874e-03 9.09041352545e-03 9.94413016678e-03 1.10006986257e-02 1.21839358632e-02 1.33587578257e-02 - 1.43606388158e-02 1.50351544907e-02 1.52731342228e-02 1.50351544907e-02 1.43606388158e-02 1.33587578257e-02 - 1.21839358632e-02 1.10006986257e-02 9.94413016678e-03 9.09041352545e-03 8.45428906874e-03 8.01216613758e-03 - 7.72923129933e-03 7.57424593471e-03 7.52528842907e-03 7.57424593471e-03 7.72923129933e-03 8.01216613758e-03 - 9.87827853279e-03 1.12010383875e-02 1.28345023601e-02 1.47188141408e-02 1.67052485052e-02 1.85584096087e-02 - 2.00031512771e-02 2.07965618706e-02 2.07965618706e-02 2.00031512771e-02 1.85584096087e-02 1.67052485052e-02 - 1.47188141408e-02 1.28345023601e-02 1.12010383875e-02 9.87827853279e-03 8.87112379151e-03 8.16346419979e-03 - 7.72761616008e-03 7.52528842907e-03 7.52528842907e-03 7.72761616008e-03 8.16346419979e-03 8.87112379151e-03 - 1.22047713863e-02 1.44498895771e-02 1.71372885777e-02 2.01439779904e-02 2.31896632144e-02 2.58559673304e-02 - 2.76901755249e-02 2.83448586606e-02 2.76901755249e-02 2.58559673304e-02 2.31896632144e-02 2.01439779904e-02 - 1.71372885777e-02 1.44498895771e-02 1.22047713863e-02 1.04248538239e-02 9.10176004354e-03 8.21703388740e-03 - 7.72761616008e-03 7.57424593471e-03 7.72761616008e-03 8.21703388740e-03 9.10176004354e-03 1.04248538239e-02 - 1.55749641354e-02 1.90731200357e-02 2.32096903098e-02 2.77206217649e-02 3.20717125323e-02 3.55533814714e-02 - 3.74959847737e-02 3.74959847737e-02 3.55533814714e-02 3.20717125323e-02 2.77206217649e-02 2.32096903098e-02 - 1.90731200357e-02 1.55749641354e-02 1.27676986907e-02 1.06212947837e-02 9.10176004354e-03 8.16346419979e-03 - 7.72923129933e-03 7.72923129933e-03 8.16346419979e-03 9.10176004354e-03 1.06212947837e-02 1.27676986907e-02 - 2.01594924058e-02 2.53584894772e-02 3.14092023629e-02 3.77513071960e-02 4.34476578087e-02 4.74317203940e-02 - 4.88652050826e-02 4.74317203940e-02 4.34476578087e-02 3.77513071960e-02 3.14092023629e-02 2.53584894772e-02 - 2.01594924058e-02 1.59804507535e-02 1.27676986907e-02 1.04248538239e-02 8.87112379151e-03 8.01216613758e-03 - 7.74242036277e-03 8.01216613758e-03 8.87112379151e-03 1.04248538239e-02 1.27676986907e-02 1.59804507535e-02 - 2.89819100992e-02 3.73919347545e-02 4.64352697346e-02 5.48042821337e-02 6.08204348290e-02 6.30211041418e-02 - 6.08204348290e-02 5.48042821337e-02 4.64352697346e-02 3.73919347545e-02 2.89819100992e-02 2.19036737434e-02 - 1.63414636372e-02 1.22047713863e-02 9.31927456850e-03 7.49319081292e-03 6.52118028642e-03 6.22225650652e-03 - 6.52118028642e-03 7.49319081292e-03 9.31927456850e-03 1.22047713863e-02 1.63414636372e-02 2.19036737434e-02 - 3.73919347545e-02 4.80826222932e-02 5.89796865623e-02 6.81380649556e-02 7.34279203403e-02 7.34279203403e-02 - 6.81380649556e-02 5.89796865623e-02 4.80826222932e-02 3.73919347545e-02 2.81262227272e-02 2.07351318935e-02 - 1.51716517813e-02 1.12010383875e-02 8.56724324800e-03 7.02039368822e-03 6.32497949646e-03 6.32497949646e-03 - 7.02039368822e-03 8.56724324800e-03 1.12010383875e-02 1.51716517813e-02 2.07351318935e-02 2.81262227272e-02 - 4.64352697346e-02 5.89796865623e-02 7.08330821436e-02 7.94813267109e-02 8.26693353815e-02 7.94813267109e-02 - 7.08330821436e-02 5.89796865623e-02 4.64352697346e-02 3.50622322206e-02 2.57560150434e-02 1.86453765526e-02 - 1.34835937042e-02 9.94413016678e-03 7.72360398793e-03 6.54341818442e-03 6.17987618622e-03 6.54341818442e-03 - 7.72360398793e-03 9.94413016678e-03 1.34835937042e-02 1.86453765526e-02 2.57560150434e-02 3.50622322206e-02 - 5.48042821337e-02 6.81380649556e-02 7.94813267109e-02 8.60847359405e-02 8.60847359405e-02 7.94813267109e-02 - 6.81380649556e-02 5.48042821337e-02 4.19230722756e-02 3.09544991478e-02 2.23681861845e-02 1.60207976494e-02 - 1.15565994691e-02 8.62132731210e-03 6.90130049286e-03 6.12647535151e-03 6.12647535151e-03 6.90130049286e-03 - 8.62132731210e-03 1.15565994691e-02 1.60207976494e-02 2.23681861845e-02 3.09544991478e-02 4.19230722756e-02 - 6.08204348290e-02 7.34279203403e-02 8.26693353815e-02 8.60847359405e-02 8.26693353815e-02 7.34279203403e-02 - 6.08204348290e-02 4.75580604876e-02 3.56148556138e-02 2.59124227944e-02 1.85555754828e-02 1.32551887233e-02 - 9.64234738867e-03 7.38199424752e-03 6.17987618622e-03 5.80894063737e-03 6.17987618622e-03 7.38199424752e-03 - 9.64234738867e-03 1.32551887233e-02 1.85555754828e-02 2.59124227944e-02 3.56148556138e-02 4.75580604876e-02 - 6.30211041418e-02 7.34279203403e-02 7.94813267109e-02 7.94813267109e-02 7.34279203403e-02 6.30211041418e-02 - 5.07662838782e-02 3.88905177841e-02 2.87349610469e-02 2.07503567214e-02 1.48305747723e-02 1.06646265060e-02 - 7.92776657338e-03 6.32497949646e-03 5.60286211346e-03 5.60286211346e-03 6.32497949646e-03 7.92776657338e-03 - 1.06646265060e-02 1.48305747723e-02 2.07503567214e-02 2.87349610469e-02 3.88905177841e-02 5.07662838782e-02 - 6.08204348290e-02 6.81380649556e-02 7.08330821436e-02 6.81380649556e-02 6.08204348290e-02 5.07662838782e-02 - 4.00749900281e-02 3.03107984462e-02 2.22527540382e-02 1.60537467662e-02 1.15426690136e-02 8.45428906874e-03 - 6.52118028642e-03 5.49418812711e-03 5.17760394975e-03 5.49418812711e-03 6.52118028642e-03 8.45428906874e-03 - 1.15426690136e-02 1.60537467662e-02 2.22527540382e-02 3.03107984462e-02 4.00749900281e-02 5.07662838782e-02 - 5.48042821337e-02 5.89796865623e-02 5.89796865623e-02 5.48042821337e-02 4.75580604876e-02 3.88905177841e-02 - 3.03107984462e-02 2.27884625410e-02 1.67247696023e-02 1.21379949101e-02 8.87112379151e-03 6.71488304626e-03 - 5.45146116297e-03 4.88266428153e-03 4.88266428153e-03 5.45146116297e-03 6.71488304626e-03 8.87112379151e-03 - 1.21379949101e-02 1.67247696023e-02 2.27884625410e-02 3.03107984462e-02 3.88905177841e-02 4.75580604876e-02 - 4.64352697346e-02 4.80826222932e-02 4.64352697346e-02 4.19230722756e-02 3.56148556138e-02 2.87349610469e-02 - 2.22527540382e-02 1.67247696023e-02 1.23490348915e-02 9.10176004354e-03 6.85502201162e-03 5.44128685160e-03 - 4.68737681852e-03 4.45430428341e-03 4.68737681852e-03 5.44128685160e-03 6.85502201162e-03 9.10176004354e-03 - 1.23490348915e-02 1.67247696023e-02 2.22527540382e-02 2.87349610469e-02 3.56148556138e-02 4.19230722756e-02 - 3.73919347545e-02 3.73919347545e-02 3.50622322206e-02 3.09544991478e-02 2.59124227944e-02 2.07503567214e-02 - 1.60537467662e-02 1.21379949101e-02 9.10176004354e-03 6.90595653158e-03 5.44101481998e-03 4.57233815477e-03 - 4.17601417174e-03 4.17601417174e-03 4.57233815477e-03 5.44101481998e-03 6.90595653158e-03 9.10176004354e-03 - 1.21379949101e-02 1.60537467662e-02 2.07503567214e-02 2.59124227944e-02 3.09544991478e-02 3.50622322206e-02 - 2.89819100992e-02 2.81262227272e-02 2.57560150434e-02 2.23681861845e-02 1.85555754828e-02 1.48305747723e-02 - 1.15426690136e-02 8.87112379151e-03 6.85502201162e-03 5.44101481998e-03 4.53364885369e-03 4.03593630205e-03 - 3.87834628559e-03 4.03593630205e-03 4.53364885369e-03 5.44101481998e-03 6.85502201162e-03 8.87112379151e-03 - 1.15426690136e-02 1.48305747723e-02 1.85555754828e-02 2.23681861845e-02 2.57560150434e-02 2.81262227272e-02 - 2.19036737434e-02 2.07351318935e-02 1.86453765526e-02 1.60207976494e-02 1.32551887233e-02 1.06646265060e-02 - 8.45428906874e-03 6.71488304626e-03 5.44128685160e-03 4.57233815477e-03 4.03593630205e-03 3.77975659219e-03 - 3.77975659219e-03 4.03593630205e-03 4.57233815477e-03 5.44128685160e-03 6.71488304626e-03 8.45428906874e-03 - 1.06646265060e-02 1.32551887233e-02 1.60207976494e-02 1.86453765526e-02 2.07351318935e-02 2.19036737434e-02 - 1.63414636372e-02 1.51716517813e-02 1.34835937042e-02 1.15565994691e-02 9.64234738867e-03 7.92776657338e-03 - 6.52118028642e-03 5.45146116297e-03 4.68737681852e-03 4.17601417174e-03 3.87834628559e-03 3.77975659219e-03 - 3.87834628559e-03 4.17601417174e-03 4.68737681852e-03 5.45146116297e-03 6.52118028642e-03 7.92776657338e-03 - 9.64234738867e-03 1.15565994691e-02 1.34835937042e-02 1.51716517813e-02 1.63414636372e-02 1.67618733320e-02 - 1.22047713863e-02 1.12010383875e-02 9.94413016678e-03 8.62132731210e-03 7.38199424752e-03 6.32497949646e-03 - 5.49418812711e-03 4.88266428153e-03 4.45430428341e-03 4.17601417174e-03 4.03593630205e-03 4.03593630205e-03 - 4.17601417174e-03 4.45430428341e-03 4.88266428153e-03 5.49418812711e-03 6.32497949646e-03 7.38199424752e-03 - 8.62132731210e-03 9.94413016678e-03 1.12010383875e-02 1.22047713863e-02 1.27676986907e-02 1.27676986907e-02 - 9.31927456850e-03 8.56724324800e-03 7.72360398793e-03 6.90130049286e-03 6.17987618622e-03 5.60286211346e-03 - 5.17760394975e-03 4.88266428153e-03 4.68737681852e-03 4.57233815477e-03 4.53364885369e-03 4.57233815477e-03 - 4.68737681852e-03 4.88266428153e-03 5.17760394975e-03 5.60286211346e-03 6.17987618622e-03 6.90130049286e-03 - 7.72360398793e-03 8.56724324800e-03 9.31927456850e-03 9.84753974693e-03 1.00387761995e-02 9.84753974693e-03 - 7.49319081292e-03 7.02039368822e-03 6.54341818442e-03 6.12647535151e-03 5.80894063737e-03 5.60286211346e-03 - 5.49418812711e-03 5.45146116297e-03 5.44128685160e-03 5.44101481998e-03 5.44101481998e-03 5.44128685160e-03 - 5.45146116297e-03 5.49418812711e-03 5.60286211346e-03 5.80894063737e-03 6.12647535151e-03 6.54341818442e-03 - 7.02039368822e-03 7.49319081292e-03 7.87995370977e-03 8.09940143038e-03 8.09940143038e-03 7.87995370977e-03 - 6.52118028642e-03 6.32497949646e-03 6.17987618622e-03 6.12647535151e-03 6.17987618622e-03 6.32497949646e-03 - 6.52118028642e-03 6.71488304626e-03 6.85502201162e-03 6.90595653158e-03 6.85502201162e-03 6.71488304626e-03 - 6.52118028642e-03 6.32497949646e-03 6.17987618622e-03 6.12647535151e-03 6.17987618622e-03 6.32497949646e-03 - 6.52118028642e-03 6.71488304626e-03 6.85502201162e-03 6.90595653158e-03 6.85502201162e-03 6.71488304626e-03 - 6.22225650652e-03 6.32497949646e-03 6.54341818442e-03 6.90130049286e-03 7.38199424752e-03 7.92776657338e-03 - 8.45428906874e-03 8.87112379151e-03 9.10176004354e-03 9.10176004354e-03 8.87112379151e-03 8.45428906874e-03 - 7.92776657338e-03 7.38199424752e-03 6.90130049286e-03 6.54341818442e-03 6.32497949646e-03 6.22225650652e-03 - 6.18966814651e-03 6.18550458410e-03 6.18689298800e-03 6.18689298800e-03 6.18550458410e-03 6.18966814651e-03 - 6.52118028642e-03 7.02039368822e-03 7.72360398793e-03 8.62132731210e-03 9.64234738867e-03 1.06646265060e-02 - 1.15426690136e-02 1.21379949101e-02 1.23490348915e-02 1.21379949101e-02 1.15426690136e-02 1.06646265060e-02 - 9.64234738867e-03 8.62132731210e-03 7.72360398793e-03 7.02039368822e-03 6.52118028642e-03 6.18966814651e-03 - 5.97825202186e-03 5.85659906617e-03 5.81607896972e-03 5.85659906617e-03 5.97825202186e-03 6.18966814651e-03 - 7.49319081292e-03 8.56724324800e-03 9.94413016678e-03 1.15565994691e-02 1.32551887233e-02 1.48305747723e-02 - 1.60537467662e-02 1.67247696023e-02 1.67247696023e-02 1.60537467662e-02 1.48305747723e-02 1.32551887233e-02 - 1.15565994691e-02 9.94413016678e-03 8.56724324800e-03 7.49319081292e-03 6.71488304626e-03 6.18550458410e-03 - 5.85659906617e-03 5.69805609664e-03 5.69805609664e-03 5.85659906617e-03 6.18550458410e-03 6.71488304626e-03 - 9.31927456850e-03 1.12010383875e-02 1.34835937042e-02 1.60207976494e-02 1.85555754828e-02 2.07503567214e-02 - 2.22527540382e-02 2.27884625410e-02 2.22527540382e-02 2.07503567214e-02 1.85555754828e-02 1.60207976494e-02 - 1.34835937042e-02 1.12010383875e-02 9.31927456850e-03 7.87995370977e-03 6.85502201162e-03 6.18689298800e-03 - 5.81607896972e-03 5.69805609664e-03 5.81607896972e-03 6.18689298800e-03 6.85502201162e-03 7.87995370977e-03 - 1.22047713863e-02 1.51716517813e-02 1.86453765526e-02 2.23681861845e-02 2.59124227944e-02 2.87349610469e-02 - 3.03107984462e-02 3.03107984462e-02 2.87349610469e-02 2.59124227944e-02 2.23681861845e-02 1.86453765526e-02 - 1.51716517813e-02 1.22047713863e-02 9.84753974693e-03 8.09940143038e-03 6.90595653158e-03 6.18689298800e-03 - 5.85659906617e-03 5.85659906617e-03 6.18689298800e-03 6.90595653158e-03 8.09940143038e-03 9.84753974693e-03 - 1.63414636372e-02 2.07351318935e-02 2.57560150434e-02 3.09544991478e-02 3.56148556138e-02 3.88905177841e-02 - 4.00749900281e-02 3.88905177841e-02 3.56148556138e-02 3.09544991478e-02 2.57560150434e-02 2.07351318935e-02 - 1.63414636372e-02 1.27676986907e-02 1.00387761995e-02 8.09940143038e-03 6.85502201162e-03 6.18550458410e-03 - 5.97825202186e-03 6.18550458410e-03 6.85502201162e-03 8.09940143038e-03 1.00387761995e-02 1.27676986907e-02 - 2.19036737434e-02 2.81262227272e-02 3.50622322206e-02 4.19230722756e-02 4.75580604876e-02 5.07662838782e-02 - 5.07662838782e-02 4.75580604876e-02 4.19230722756e-02 3.50622322206e-02 2.81262227272e-02 2.19036737434e-02 - 1.67618733320e-02 1.27676986907e-02 9.84753974693e-03 7.87995370977e-03 6.71488304626e-03 6.18966814651e-03 - 6.18966814651e-03 6.71488304626e-03 7.87995370977e-03 9.84753974693e-03 1.27676986907e-02 1.67618733320e-02 - 3.35320462758e-02 4.30069852462e-02 5.26365510379e-02 6.07372364008e-02 6.54257706676e-02 6.54257706676e-02 - 6.07372364008e-02 5.26365510379e-02 4.30069852462e-02 3.35320462758e-02 2.52525868671e-02 1.85732113773e-02 - 1.35012692014e-02 9.87827853279e-03 7.49319081292e-03 6.10931604239e-03 5.49418812711e-03 5.49418812711e-03 - 6.10931604239e-03 7.49319081292e-03 9.87827853279e-03 1.35012692014e-02 1.85732113773e-02 2.52525868671e-02 - 4.30069852462e-02 5.45216826827e-02 6.54257706676e-02 7.34093289667e-02 7.63593661757e-02 7.34093289667e-02 - 6.54257706676e-02 5.45216826827e-02 4.30069852462e-02 3.25477771477e-02 2.39287446557e-02 1.72789698058e-02 - 1.24193280193e-02 9.09041352545e-03 7.02039368822e-03 5.93416429324e-03 5.60286211346e-03 5.93416429324e-03 - 7.02039368822e-03 9.09041352545e-03 1.24193280193e-02 1.72789698058e-02 2.39287446557e-02 3.25477771477e-02 - 5.26365510379e-02 6.54257706676e-02 7.63593661757e-02 8.27498054349e-02 8.27498054349e-02 7.63593661757e-02 - 6.54257706676e-02 5.26365510379e-02 4.03211904768e-02 2.98227515673e-02 2.15543530804e-02 1.53937416744e-02 - 1.10438280884e-02 8.19475376688e-03 6.54341818442e-03 5.80894063737e-03 5.80894063737e-03 6.54341818442e-03 - 8.19475376688e-03 1.10438280884e-02 1.53937416744e-02 2.15543530804e-02 2.98227515673e-02 4.03211904768e-02 - 6.07372364008e-02 7.34093289667e-02 8.27498054349e-02 8.62128629277e-02 8.27498054349e-02 7.34093289667e-02 - 6.07372364008e-02 4.74868469137e-02 3.55984989831e-02 2.59309064966e-02 1.85605049655e-02 1.32196503024e-02 - 9.57706447476e-03 7.31454489847e-03 6.12647535151e-03 5.76375313750e-03 6.12647535151e-03 7.31454489847e-03 - 9.57706447476e-03 1.32196503024e-02 1.85605049655e-02 2.59309064966e-02 3.55984989831e-02 4.74868469137e-02 - 6.54257706676e-02 7.63593661757e-02 8.27498054349e-02 8.27498054349e-02 7.63593661757e-02 6.54257706676e-02 - 5.26365510379e-02 4.03211904768e-02 2.98227515673e-02 2.15543530804e-02 1.53937416744e-02 1.10438280884e-02 - 8.19475376688e-03 6.54341818442e-03 5.80894063737e-03 5.80894063737e-03 6.54341818442e-03 8.19475376688e-03 - 1.10438280884e-02 1.53937416744e-02 2.15543530804e-02 2.98227515673e-02 4.03211904768e-02 5.26365510379e-02 - 6.54257706676e-02 7.34093289667e-02 7.63593661757e-02 7.34093289667e-02 6.54257706676e-02 5.45216826827e-02 - 4.30069852462e-02 3.25477771477e-02 2.39287446557e-02 1.72789698058e-02 1.24193280193e-02 9.09041352545e-03 - 7.02039368822e-03 5.93416429324e-03 5.60286211346e-03 5.93416429324e-03 7.02039368822e-03 9.09041352545e-03 - 1.24193280193e-02 1.72789698058e-02 2.39287446557e-02 3.25477771477e-02 4.30069852462e-02 5.45216826827e-02 - 6.07372364008e-02 6.54257706676e-02 6.54257706676e-02 6.07372364008e-02 5.26365510379e-02 4.30069852462e-02 - 3.35320462758e-02 2.52525868671e-02 1.85732113773e-02 1.35012692014e-02 9.87827853279e-03 7.49319081292e-03 - 6.10931604239e-03 5.49418812711e-03 5.49418812711e-03 6.10931604239e-03 7.49319081292e-03 9.87827853279e-03 - 1.35012692014e-02 1.85732113773e-02 2.52525868671e-02 3.35320462758e-02 4.30069852462e-02 5.26365510379e-02 - 5.26365510379e-02 5.45216826827e-02 5.26365510379e-02 4.74868469137e-02 4.03211904768e-02 3.25477771477e-02 - 2.52525868671e-02 1.90357893277e-02 1.41011278955e-02 1.04248538239e-02 7.87995370977e-03 6.28902224701e-03 - 5.45146116297e-03 5.19548069751e-03 5.45146116297e-03 6.28902224701e-03 7.87995370977e-03 1.04248538239e-02 - 1.41011278955e-02 1.90357893277e-02 2.52525868671e-02 3.25477771477e-02 4.03211904768e-02 4.74868469137e-02 - 4.30069852462e-02 4.30069852462e-02 4.03211904768e-02 3.55984989831e-02 2.98227515673e-02 2.39287446557e-02 - 1.85732113773e-02 1.41011278955e-02 1.06212947837e-02 8.09940143038e-03 6.42332605072e-03 5.44128685160e-03 - 5.00022810356e-03 5.00022810356e-03 5.44128685160e-03 6.42332605072e-03 8.09940143038e-03 1.06212947837e-02 - 1.41011278955e-02 1.85732113773e-02 2.39287446557e-02 2.98227515673e-02 3.55984989831e-02 4.03211904768e-02 - 3.35320462758e-02 3.25477771477e-02 2.98227515673e-02 2.59309064966e-02 2.15543530804e-02 1.72789698058e-02 - 1.35012692014e-02 1.04248538239e-02 8.09940143038e-03 6.47303924297e-03 5.44101481998e-03 4.88536934411e-03 - 4.71220524742e-03 4.88536934411e-03 5.44101481998e-03 6.47303924297e-03 8.09940143038e-03 1.04248538239e-02 - 1.35012692014e-02 1.72789698058e-02 2.15543530804e-02 2.59309064966e-02 2.98227515673e-02 3.25477771477e-02 - 2.52525868671e-02 2.39287446557e-02 2.15543530804e-02 1.85605049655e-02 1.53937416744e-02 1.24193280193e-02 - 9.87827853279e-03 7.87995370977e-03 6.42332605072e-03 5.44101481998e-03 4.84662495208e-03 4.56877489732e-03 - 4.56877489732e-03 4.84662495208e-03 5.44101481998e-03 6.42332605072e-03 7.87995370977e-03 9.87827853279e-03 - 1.24193280193e-02 1.53937416744e-02 1.85605049655e-02 2.15543530804e-02 2.39287446557e-02 2.52525868671e-02 - 1.85732113773e-02 1.72789698058e-02 1.53937416744e-02 1.32196503024e-02 1.10438280884e-02 9.09041352545e-03 - 7.49319081292e-03 6.28902224701e-03 5.44128685160e-03 4.88536934411e-03 4.56877489732e-03 4.46542762661e-03 - 4.56877489732e-03 4.88536934411e-03 5.44128685160e-03 6.28902224701e-03 7.49319081292e-03 9.09041352545e-03 - 1.10438280884e-02 1.32196503024e-02 1.53937416744e-02 1.72789698058e-02 1.85732113773e-02 1.90357893277e-02 - 1.35012692014e-02 1.24193280193e-02 1.10438280884e-02 9.57706447476e-03 8.19475376688e-03 7.02039368822e-03 - 6.10931604239e-03 5.45146116297e-03 5.00022810356e-03 4.71220524742e-03 4.56877489732e-03 4.56877489732e-03 - 4.71220524742e-03 5.00022810356e-03 5.45146116297e-03 6.10931604239e-03 7.02039368822e-03 8.19475376688e-03 - 9.57706447476e-03 1.10438280884e-02 1.24193280193e-02 1.35012692014e-02 1.41011278955e-02 1.41011278955e-02 - 9.87827853279e-03 9.09041352545e-03 8.19475376688e-03 7.31454489847e-03 6.54341818442e-03 5.93416429324e-03 - 5.49418812711e-03 5.19548069751e-03 5.00022810356e-03 4.88536934411e-03 4.84662495208e-03 4.88536934411e-03 - 5.00022810356e-03 5.19548069751e-03 5.49418812711e-03 5.93416429324e-03 6.54341818442e-03 7.31454489847e-03 - 8.19475376688e-03 9.09041352545e-03 9.87827853279e-03 1.04248538239e-02 1.06212947837e-02 1.04248538239e-02 - 7.49319081292e-03 7.02039368822e-03 6.54341818442e-03 6.12647535151e-03 5.80894063737e-03 5.60286211346e-03 - 5.49418812711e-03 5.45146116297e-03 5.44128685160e-03 5.44101481998e-03 5.44101481998e-03 5.44128685160e-03 - 5.45146116297e-03 5.49418812711e-03 5.60286211346e-03 5.80894063737e-03 6.12647535151e-03 6.54341818442e-03 - 7.02039368822e-03 7.49319081292e-03 7.87995370977e-03 8.09940143038e-03 8.09940143038e-03 7.87995370977e-03 - 6.10931604239e-03 5.93416429324e-03 5.80894063737e-03 5.76375313750e-03 5.80894063737e-03 5.93416429324e-03 - 6.10931604239e-03 6.28902224701e-03 6.42332605072e-03 6.47303924297e-03 6.42332605072e-03 6.28902224701e-03 - 6.10931604239e-03 5.93416429324e-03 5.80894063737e-03 5.76375313750e-03 5.80894063737e-03 5.93416429324e-03 - 6.10931604239e-03 6.28902224701e-03 6.42332605072e-03 6.47303924297e-03 6.42332605072e-03 6.28902224701e-03 - 5.49418812711e-03 5.60286211346e-03 5.80894063737e-03 6.12647535151e-03 6.54341818442e-03 7.02039368822e-03 - 7.49319081292e-03 7.87995370977e-03 8.09940143038e-03 8.09940143038e-03 7.87995370977e-03 7.49319081292e-03 - 7.02039368822e-03 6.54341818442e-03 6.12647535151e-03 5.80894063737e-03 5.60286211346e-03 5.49418812711e-03 - 5.45146116297e-03 5.44128685160e-03 5.44101481998e-03 5.44101481998e-03 5.44128685160e-03 5.45146116297e-03 - 5.49418812711e-03 5.93416429324e-03 6.54341818442e-03 7.31454489847e-03 8.19475376688e-03 9.09041352545e-03 - 9.87827853279e-03 1.04248538239e-02 1.06212947837e-02 1.04248538239e-02 9.87827853279e-03 9.09041352545e-03 - 8.19475376688e-03 7.31454489847e-03 6.54341818442e-03 5.93416429324e-03 5.49418812711e-03 5.19548069751e-03 - 5.00022810356e-03 4.88536934411e-03 4.84662495208e-03 4.88536934411e-03 5.00022810356e-03 5.19548069751e-03 - 6.10931604239e-03 7.02039368822e-03 8.19475376688e-03 9.57706447476e-03 1.10438280884e-02 1.24193280193e-02 - 1.35012692014e-02 1.41011278955e-02 1.41011278955e-02 1.35012692014e-02 1.24193280193e-02 1.10438280884e-02 - 9.57706447476e-03 8.19475376688e-03 7.02039368822e-03 6.10931604239e-03 5.45146116297e-03 5.00022810356e-03 - 4.71220524742e-03 4.56877489732e-03 4.56877489732e-03 4.71220524742e-03 5.00022810356e-03 5.45146116297e-03 - 7.49319081292e-03 9.09041352545e-03 1.10438280884e-02 1.32196503024e-02 1.53937416744e-02 1.72789698058e-02 - 1.85732113773e-02 1.90357893277e-02 1.85732113773e-02 1.72789698058e-02 1.53937416744e-02 1.32196503024e-02 - 1.10438280884e-02 9.09041352545e-03 7.49319081292e-03 6.28902224701e-03 5.44128685160e-03 4.88536934411e-03 - 4.56877489732e-03 4.46542762661e-03 4.56877489732e-03 4.88536934411e-03 5.44128685160e-03 6.28902224701e-03 - 9.87827853279e-03 1.24193280193e-02 1.53937416744e-02 1.85605049655e-02 2.15543530804e-02 2.39287446557e-02 - 2.52525868671e-02 2.52525868671e-02 2.39287446557e-02 2.15543530804e-02 1.85605049655e-02 1.53937416744e-02 - 1.24193280193e-02 9.87827853279e-03 7.87995370977e-03 6.42332605072e-03 5.44101481998e-03 4.84662495208e-03 - 4.56877489732e-03 4.56877489732e-03 4.84662495208e-03 5.44101481998e-03 6.42332605072e-03 7.87995370977e-03 - 1.35012692014e-02 1.72789698058e-02 2.15543530804e-02 2.59309064966e-02 2.98227515673e-02 3.25477771477e-02 - 3.35320462758e-02 3.25477771477e-02 2.98227515673e-02 2.59309064966e-02 2.15543530804e-02 1.72789698058e-02 - 1.35012692014e-02 1.04248538239e-02 8.09940143038e-03 6.47303924297e-03 5.44101481998e-03 4.88536934411e-03 - 4.71220524742e-03 4.88536934411e-03 5.44101481998e-03 6.47303924297e-03 8.09940143038e-03 1.04248538239e-02 - 1.85732113773e-02 2.39287446557e-02 2.98227515673e-02 3.55984989831e-02 4.03211904768e-02 4.30069852462e-02 - 4.30069852462e-02 4.03211904768e-02 3.55984989831e-02 2.98227515673e-02 2.39287446557e-02 1.85732113773e-02 - 1.41011278955e-02 1.06212947837e-02 8.09940143038e-03 6.42332605072e-03 5.44128685160e-03 5.00022810356e-03 - 5.00022810356e-03 5.44128685160e-03 6.42332605072e-03 8.09940143038e-03 1.06212947837e-02 1.41011278955e-02 - 2.52525868671e-02 3.25477771477e-02 4.03211904768e-02 4.74868469137e-02 5.26365510379e-02 5.45216826827e-02 - 5.26365510379e-02 4.74868469137e-02 4.03211904768e-02 3.25477771477e-02 2.52525868671e-02 1.90357893277e-02 - 1.41011278955e-02 1.04248538239e-02 7.87995370977e-03 6.28902224701e-03 5.45146116297e-03 5.19548069751e-03 - 5.45146116297e-03 6.28902224701e-03 7.87995370977e-03 1.04248538239e-02 1.41011278955e-02 1.90357893277e-02 - 4.00749900281e-02 5.07662838782e-02 6.08204348290e-02 6.81380649556e-02 7.08330821436e-02 6.81380649556e-02 - 6.08204348290e-02 5.07662838782e-02 4.00749900281e-02 3.03107984462e-02 2.22527540382e-02 1.60537467662e-02 - 1.15426690136e-02 8.45428906874e-03 6.52118028642e-03 5.49418812711e-03 5.17760394975e-03 5.49418812711e-03 - 6.52118028642e-03 8.45428906874e-03 1.15426690136e-02 1.60537467662e-02 2.22527540382e-02 3.03107984462e-02 - 5.07662838782e-02 6.30211041418e-02 7.34279203403e-02 7.94813267109e-02 7.94813267109e-02 7.34279203403e-02 - 6.30211041418e-02 5.07662838782e-02 3.88905177841e-02 2.87349610469e-02 2.07503567214e-02 1.48305747723e-02 - 1.06646265060e-02 7.92776657338e-03 6.32497949646e-03 5.60286211346e-03 5.60286211346e-03 6.32497949646e-03 - 7.92776657338e-03 1.06646265060e-02 1.48305747723e-02 2.07503567214e-02 2.87349610469e-02 3.88905177841e-02 - 6.08204348290e-02 7.34279203403e-02 8.26693353815e-02 8.60847359405e-02 8.26693353815e-02 7.34279203403e-02 - 6.08204348290e-02 4.75580604876e-02 3.56148556138e-02 2.59124227944e-02 1.85555754828e-02 1.32551887233e-02 - 9.64234738867e-03 7.38199424752e-03 6.17987618622e-03 5.80894063737e-03 6.17987618622e-03 7.38199424752e-03 - 9.64234738867e-03 1.32551887233e-02 1.85555754828e-02 2.59124227944e-02 3.56148556138e-02 4.75580604876e-02 - 6.81380649556e-02 7.94813267109e-02 8.60847359405e-02 8.60847359405e-02 7.94813267109e-02 6.81380649556e-02 - 5.48042821337e-02 4.19230722756e-02 3.09544991478e-02 2.23681861845e-02 1.60207976494e-02 1.15565994691e-02 - 8.62132731210e-03 6.90130049286e-03 6.12647535151e-03 6.12647535151e-03 6.90130049286e-03 8.62132731210e-03 - 1.15565994691e-02 1.60207976494e-02 2.23681861845e-02 3.09544991478e-02 4.19230722756e-02 5.48042821337e-02 - 7.08330821436e-02 7.94813267109e-02 8.26693353815e-02 7.94813267109e-02 7.08330821436e-02 5.89796865623e-02 - 4.64352697346e-02 3.50622322206e-02 2.57560150434e-02 1.86453765526e-02 1.34835937042e-02 9.94413016678e-03 - 7.72360398793e-03 6.54341818442e-03 6.17987618622e-03 6.54341818442e-03 7.72360398793e-03 9.94413016678e-03 - 1.34835937042e-02 1.86453765526e-02 2.57560150434e-02 3.50622322206e-02 4.64352697346e-02 5.89796865623e-02 - 6.81380649556e-02 7.34279203403e-02 7.34279203403e-02 6.81380649556e-02 5.89796865623e-02 4.80826222932e-02 - 3.73919347545e-02 2.81262227272e-02 2.07351318935e-02 1.51716517813e-02 1.12010383875e-02 8.56724324800e-03 - 7.02039368822e-03 6.32497949646e-03 6.32497949646e-03 7.02039368822e-03 8.56724324800e-03 1.12010383875e-02 - 1.51716517813e-02 2.07351318935e-02 2.81262227272e-02 3.73919347545e-02 4.80826222932e-02 5.89796865623e-02 - 6.08204348290e-02 6.30211041418e-02 6.08204348290e-02 5.48042821337e-02 4.64352697346e-02 3.73919347545e-02 - 2.89819100992e-02 2.19036737434e-02 1.63414636372e-02 1.22047713863e-02 9.31927456850e-03 7.49319081292e-03 - 6.52118028642e-03 6.22225650652e-03 6.52118028642e-03 7.49319081292e-03 9.31927456850e-03 1.22047713863e-02 - 1.63414636372e-02 2.19036737434e-02 2.89819100992e-02 3.73919347545e-02 4.64352697346e-02 5.48042821337e-02 - 5.07662838782e-02 5.07662838782e-02 4.75580604876e-02 4.19230722756e-02 3.50622322206e-02 2.81262227272e-02 - 2.19036737434e-02 1.67618733320e-02 1.27676986907e-02 9.84753974693e-03 7.87995370977e-03 6.71488304626e-03 - 6.18966814651e-03 6.18966814651e-03 6.71488304626e-03 7.87995370977e-03 9.84753974693e-03 1.27676986907e-02 - 1.67618733320e-02 2.19036737434e-02 2.81262227272e-02 3.50622322206e-02 4.19230722756e-02 4.75580604876e-02 - 4.00749900281e-02 3.88905177841e-02 3.56148556138e-02 3.09544991478e-02 2.57560150434e-02 2.07351318935e-02 - 1.63414636372e-02 1.27676986907e-02 1.00387761995e-02 8.09940143038e-03 6.85502201162e-03 6.18550458410e-03 - 5.97825202186e-03 6.18550458410e-03 6.85502201162e-03 8.09940143038e-03 1.00387761995e-02 1.27676986907e-02 - 1.63414636372e-02 2.07351318935e-02 2.57560150434e-02 3.09544991478e-02 3.56148556138e-02 3.88905177841e-02 - 3.03107984462e-02 2.87349610469e-02 2.59124227944e-02 2.23681861845e-02 1.86453765526e-02 1.51716517813e-02 - 1.22047713863e-02 9.84753974693e-03 8.09940143038e-03 6.90595653158e-03 6.18689298800e-03 5.85659906617e-03 - 5.85659906617e-03 6.18689298800e-03 6.90595653158e-03 8.09940143038e-03 9.84753974693e-03 1.22047713863e-02 - 1.51716517813e-02 1.86453765526e-02 2.23681861845e-02 2.59124227944e-02 2.87349610469e-02 3.03107984462e-02 - 2.22527540382e-02 2.07503567214e-02 1.85555754828e-02 1.60207976494e-02 1.34835937042e-02 1.12010383875e-02 - 9.31927456850e-03 7.87995370977e-03 6.85502201162e-03 6.18689298800e-03 5.81607896972e-03 5.69805609664e-03 - 5.81607896972e-03 6.18689298800e-03 6.85502201162e-03 7.87995370977e-03 9.31927456850e-03 1.12010383875e-02 - 1.34835937042e-02 1.60207976494e-02 1.85555754828e-02 2.07503567214e-02 2.22527540382e-02 2.27884625410e-02 - 1.60537467662e-02 1.48305747723e-02 1.32551887233e-02 1.15565994691e-02 9.94413016678e-03 8.56724324800e-03 - 7.49319081292e-03 6.71488304626e-03 6.18550458410e-03 5.85659906617e-03 5.69805609664e-03 5.69805609664e-03 - 5.85659906617e-03 6.18550458410e-03 6.71488304626e-03 7.49319081292e-03 8.56724324800e-03 9.94413016678e-03 - 1.15565994691e-02 1.32551887233e-02 1.48305747723e-02 1.60537467662e-02 1.67247696023e-02 1.67247696023e-02 - 1.15426690136e-02 1.06646265060e-02 9.64234738867e-03 8.62132731210e-03 7.72360398793e-03 7.02039368822e-03 - 6.52118028642e-03 6.18966814651e-03 5.97825202186e-03 5.85659906617e-03 5.81607896972e-03 5.85659906617e-03 - 5.97825202186e-03 6.18966814651e-03 6.52118028642e-03 7.02039368822e-03 7.72360398793e-03 8.62132731210e-03 - 9.64234738867e-03 1.06646265060e-02 1.15426690136e-02 1.21379949101e-02 1.23490348915e-02 1.21379949101e-02 - 8.45428906874e-03 7.92776657338e-03 7.38199424752e-03 6.90130049286e-03 6.54341818442e-03 6.32497949646e-03 - 6.22225650652e-03 6.18966814651e-03 6.18550458410e-03 6.18689298800e-03 6.18689298800e-03 6.18550458410e-03 - 6.18966814651e-03 6.22225650652e-03 6.32497949646e-03 6.54341818442e-03 6.90130049286e-03 7.38199424752e-03 - 7.92776657338e-03 8.45428906874e-03 8.87112379151e-03 9.10176004354e-03 9.10176004354e-03 8.87112379151e-03 - 6.52118028642e-03 6.32497949646e-03 6.17987618622e-03 6.12647535151e-03 6.17987618622e-03 6.32497949646e-03 - 6.52118028642e-03 6.71488304626e-03 6.85502201162e-03 6.90595653158e-03 6.85502201162e-03 6.71488304626e-03 - 6.52118028642e-03 6.32497949646e-03 6.17987618622e-03 6.12647535151e-03 6.17987618622e-03 6.32497949646e-03 - 6.52118028642e-03 6.71488304626e-03 6.85502201162e-03 6.90595653158e-03 6.85502201162e-03 6.71488304626e-03 - 5.49418812711e-03 5.60286211346e-03 5.80894063737e-03 6.12647535151e-03 6.54341818442e-03 7.02039368822e-03 - 7.49319081292e-03 7.87995370977e-03 8.09940143038e-03 8.09940143038e-03 7.87995370977e-03 7.49319081292e-03 - 7.02039368822e-03 6.54341818442e-03 6.12647535151e-03 5.80894063737e-03 5.60286211346e-03 5.49418812711e-03 - 5.45146116297e-03 5.44128685160e-03 5.44101481998e-03 5.44101481998e-03 5.44128685160e-03 5.45146116297e-03 - 5.17760394975e-03 5.60286211346e-03 6.17987618622e-03 6.90130049286e-03 7.72360398793e-03 8.56724324800e-03 - 9.31927456850e-03 9.84753974693e-03 1.00387761995e-02 9.84753974693e-03 9.31927456850e-03 8.56724324800e-03 - 7.72360398793e-03 6.90130049286e-03 6.17987618622e-03 5.60286211346e-03 5.17760394975e-03 4.88266428153e-03 - 4.68737681852e-03 4.57233815477e-03 4.53364885369e-03 4.57233815477e-03 4.68737681852e-03 4.88266428153e-03 - 5.49418812711e-03 6.32497949646e-03 7.38199424752e-03 8.62132731210e-03 9.94413016678e-03 1.12010383875e-02 - 1.22047713863e-02 1.27676986907e-02 1.27676986907e-02 1.22047713863e-02 1.12010383875e-02 9.94413016678e-03 - 8.62132731210e-03 7.38199424752e-03 6.32497949646e-03 5.49418812711e-03 4.88266428153e-03 4.45430428341e-03 - 4.17601417174e-03 4.03593630205e-03 4.03593630205e-03 4.17601417174e-03 4.45430428341e-03 4.88266428153e-03 - 6.52118028642e-03 7.92776657338e-03 9.64234738867e-03 1.15565994691e-02 1.34835937042e-02 1.51716517813e-02 - 1.63414636372e-02 1.67618733320e-02 1.63414636372e-02 1.51716517813e-02 1.34835937042e-02 1.15565994691e-02 - 9.64234738867e-03 7.92776657338e-03 6.52118028642e-03 5.45146116297e-03 4.68737681852e-03 4.17601417174e-03 - 3.87834628559e-03 3.77975659219e-03 3.87834628559e-03 4.17601417174e-03 4.68737681852e-03 5.45146116297e-03 - 8.45428906874e-03 1.06646265060e-02 1.32551887233e-02 1.60207976494e-02 1.86453765526e-02 2.07351318935e-02 - 2.19036737434e-02 2.19036737434e-02 2.07351318935e-02 1.86453765526e-02 1.60207976494e-02 1.32551887233e-02 - 1.06646265060e-02 8.45428906874e-03 6.71488304626e-03 5.44128685160e-03 4.57233815477e-03 4.03593630205e-03 - 3.77975659219e-03 3.77975659219e-03 4.03593630205e-03 4.57233815477e-03 5.44128685160e-03 6.71488304626e-03 - 1.15426690136e-02 1.48305747723e-02 1.85555754828e-02 2.23681861845e-02 2.57560150434e-02 2.81262227272e-02 - 2.89819100992e-02 2.81262227272e-02 2.57560150434e-02 2.23681861845e-02 1.85555754828e-02 1.48305747723e-02 - 1.15426690136e-02 8.87112379151e-03 6.85502201162e-03 5.44101481998e-03 4.53364885369e-03 4.03593630205e-03 - 3.87834628559e-03 4.03593630205e-03 4.53364885369e-03 5.44101481998e-03 6.85502201162e-03 8.87112379151e-03 - 1.60537467662e-02 2.07503567214e-02 2.59124227944e-02 3.09544991478e-02 3.50622322206e-02 3.73919347545e-02 - 3.73919347545e-02 3.50622322206e-02 3.09544991478e-02 2.59124227944e-02 2.07503567214e-02 1.60537467662e-02 - 1.21379949101e-02 9.10176004354e-03 6.90595653158e-03 5.44101481998e-03 4.57233815477e-03 4.17601417174e-03 - 4.17601417174e-03 4.57233815477e-03 5.44101481998e-03 6.90595653158e-03 9.10176004354e-03 1.21379949101e-02 - 2.22527540382e-02 2.87349610469e-02 3.56148556138e-02 4.19230722756e-02 4.64352697346e-02 4.80826222932e-02 - 4.64352697346e-02 4.19230722756e-02 3.56148556138e-02 2.87349610469e-02 2.22527540382e-02 1.67247696023e-02 - 1.23490348915e-02 9.10176004354e-03 6.85502201162e-03 5.44128685160e-03 4.68737681852e-03 4.45430428341e-03 - 4.68737681852e-03 5.44128685160e-03 6.85502201162e-03 9.10176004354e-03 1.23490348915e-02 1.67247696023e-02 - 3.03107984462e-02 3.88905177841e-02 4.75580604876e-02 5.48042821337e-02 5.89796865623e-02 5.89796865623e-02 - 5.48042821337e-02 4.75580604876e-02 3.88905177841e-02 3.03107984462e-02 2.27884625410e-02 1.67247696023e-02 - 1.21379949101e-02 8.87112379151e-03 6.71488304626e-03 5.45146116297e-03 4.88266428153e-03 4.88266428153e-03 - 5.45146116297e-03 6.71488304626e-03 8.87112379151e-03 1.21379949101e-02 1.67247696023e-02 2.27884625410e-02 - 4.88652050826e-02 6.04031976834e-02 7.00265784056e-02 7.55510577091e-02 7.55510577091e-02 7.00265784056e-02 - 6.04031976834e-02 4.88652050826e-02 3.74959847737e-02 2.76901755249e-02 2.00031512771e-02 1.43606388158e-02 - 1.04118462818e-02 7.79360764654e-03 6.22225650652e-03 5.49418812711e-03 5.49418812711e-03 6.22225650652e-03 - 7.79360764654e-03 1.04118462818e-02 1.43606388158e-02 2.00031512771e-02 2.76901755249e-02 3.74959847737e-02 - 6.04031976834e-02 7.24709839797e-02 8.11535443280e-02 8.43291015351e-02 8.11535443280e-02 7.24709839797e-02 - 6.04031976834e-02 4.74317203940e-02 3.55533814714e-02 2.58559673304e-02 1.85584096087e-02 1.33587578257e-02 - 9.81666041939e-03 7.56111056333e-03 6.32497949646e-03 5.93416429324e-03 6.32497949646e-03 7.56111056333e-03 - 9.81666041939e-03 1.33587578257e-02 1.85584096087e-02 2.58559673304e-02 3.55533814714e-02 4.74317203940e-02 - 7.00265784056e-02 8.11535443280e-02 8.75278859551e-02 8.75278859551e-02 8.11535443280e-02 7.00265784056e-02 - 5.66502458441e-02 4.34476578087e-02 3.20717125323e-02 2.31896632144e-02 1.67052485052e-02 1.21839358632e-02 - 9.18607976908e-03 7.38199424752e-03 6.54341818442e-03 6.54341818442e-03 7.38199424752e-03 9.18607976908e-03 - 1.21839358632e-02 1.67052485052e-02 2.31896632144e-02 3.20717125323e-02 4.34476578087e-02 5.66502458441e-02 - 7.55510577091e-02 8.43291015351e-02 8.75278859551e-02 8.43291015351e-02 7.55510577091e-02 6.32743536159e-02 - 4.99875697106e-02 3.77513071960e-02 2.77206217649e-02 2.01439779904e-02 1.47188141408e-02 1.10006986257e-02 - 8.62132731210e-03 7.31454489847e-03 6.90130049286e-03 7.31454489847e-03 8.62132731210e-03 1.10006986257e-02 - 1.47188141408e-02 2.01439779904e-02 2.77206217649e-02 3.77513071960e-02 4.99875697106e-02 6.32743536159e-02 - 7.55510577091e-02 8.11535443280e-02 8.11535443280e-02 7.55510577091e-02 6.56955874028e-02 5.37274523246e-02 - 4.17937584090e-02 3.14092023629e-02 2.32096903098e-02 1.71372885777e-02 1.28345023601e-02 9.94413016678e-03 - 8.19475376688e-03 7.38199424752e-03 7.38199424752e-03 8.19475376688e-03 9.94413016678e-03 1.28345023601e-02 - 1.71372885777e-02 2.32096903098e-02 3.14092023629e-02 4.17937584090e-02 5.37274523246e-02 6.56955874028e-02 - 7.00265784056e-02 7.24709839797e-02 7.00265784056e-02 6.32743536159e-02 5.37274523246e-02 4.32645707778e-02 - 3.34978085182e-02 2.53584894772e-02 1.90731200357e-02 1.44498895771e-02 1.12010383875e-02 9.09041352545e-03 - 7.92776657338e-03 7.56111056333e-03 7.92776657338e-03 9.09041352545e-03 1.12010383875e-02 1.44498895771e-02 - 1.90731200357e-02 2.53584894772e-02 3.34978085182e-02 4.32645707778e-02 5.37274523246e-02 6.32743536159e-02 - 6.04031976834e-02 6.04031976834e-02 5.66502458441e-02 4.99875697106e-02 4.17937584090e-02 3.34978085182e-02 - 2.61347637909e-02 2.01594924058e-02 1.55749641354e-02 1.22047713863e-02 9.87827853279e-03 8.45428906874e-03 - 7.79360764654e-03 7.79360764654e-03 8.45428906874e-03 9.87827853279e-03 1.22047713863e-02 1.55749641354e-02 - 2.01594924058e-02 2.61347637909e-02 3.34978085182e-02 4.17937584090e-02 4.99875697106e-02 5.66502458441e-02 - 4.88652050826e-02 4.74317203940e-02 4.34476578087e-02 3.77513071960e-02 3.14092023629e-02 2.53584894772e-02 - 2.01594924058e-02 1.59804507535e-02 1.27676986907e-02 1.04248538239e-02 8.87112379151e-03 8.01216613758e-03 - 7.74242036277e-03 8.01216613758e-03 8.87112379151e-03 1.04248538239e-02 1.27676986907e-02 1.59804507535e-02 - 2.01594924058e-02 2.53584894772e-02 3.14092023629e-02 3.77513071960e-02 4.34476578087e-02 4.74317203940e-02 - 3.74959847737e-02 3.55533814714e-02 3.20717125323e-02 2.77206217649e-02 2.32096903098e-02 1.90731200357e-02 - 1.55749641354e-02 1.27676986907e-02 1.06212947837e-02 9.10176004354e-03 8.16346419979e-03 7.72923129933e-03 - 7.72923129933e-03 8.16346419979e-03 9.10176004354e-03 1.06212947837e-02 1.27676986907e-02 1.55749641354e-02 - 1.90731200357e-02 2.32096903098e-02 2.77206217649e-02 3.20717125323e-02 3.55533814714e-02 3.74959847737e-02 - 2.76901755249e-02 2.58559673304e-02 2.31896632144e-02 2.01439779904e-02 1.71372885777e-02 1.44498895771e-02 - 1.22047713863e-02 1.04248538239e-02 9.10176004354e-03 8.21703388740e-03 7.72761616008e-03 7.57424593471e-03 - 7.72761616008e-03 8.21703388740e-03 9.10176004354e-03 1.04248538239e-02 1.22047713863e-02 1.44498895771e-02 - 1.71372885777e-02 2.01439779904e-02 2.31896632144e-02 2.58559673304e-02 2.76901755249e-02 2.83448586606e-02 - 2.00031512771e-02 1.85584096087e-02 1.67052485052e-02 1.47188141408e-02 1.28345023601e-02 1.12010383875e-02 - 9.87827853279e-03 8.87112379151e-03 8.16346419979e-03 7.72761616008e-03 7.52528842907e-03 7.52528842907e-03 - 7.72761616008e-03 8.16346419979e-03 8.87112379151e-03 9.87827853279e-03 1.12010383875e-02 1.28345023601e-02 - 1.47188141408e-02 1.67052485052e-02 1.85584096087e-02 2.00031512771e-02 2.07965618706e-02 2.07965618706e-02 - 1.43606388158e-02 1.33587578257e-02 1.21839358632e-02 1.10006986257e-02 9.94413016678e-03 9.09041352545e-03 - 8.45428906874e-03 8.01216613758e-03 7.72923129933e-03 7.57424593471e-03 7.52528842907e-03 7.57424593471e-03 - 7.72923129933e-03 8.01216613758e-03 8.45428906874e-03 9.09041352545e-03 9.94413016678e-03 1.10006986257e-02 - 1.21839358632e-02 1.33587578257e-02 1.43606388158e-02 1.50351544907e-02 1.52731342228e-02 1.50351544907e-02 - 1.04118462818e-02 9.81666041939e-03 9.18607976908e-03 8.62132731210e-03 8.19475376688e-03 7.92776657338e-03 - 7.79360764654e-03 7.74242036277e-03 7.72923129933e-03 7.72761616008e-03 7.72761616008e-03 7.72923129933e-03 - 7.74242036277e-03 7.79360764654e-03 7.92776657338e-03 8.19475376688e-03 8.62132731210e-03 9.18607976908e-03 - 9.81666041939e-03 1.04118462818e-02 1.08721475193e-02 1.11222486043e-02 1.11222486043e-02 1.08721475193e-02 - 7.79360764654e-03 7.56111056333e-03 7.38199424752e-03 7.31454489847e-03 7.38199424752e-03 7.56111056333e-03 - 7.79360764654e-03 8.01216613758e-03 8.16346419979e-03 8.21703388740e-03 8.16346419979e-03 8.01216613758e-03 - 7.79360764654e-03 7.56111056333e-03 7.38199424752e-03 7.31454489847e-03 7.38199424752e-03 7.56111056333e-03 - 7.79360764654e-03 8.01216613758e-03 8.16346419979e-03 8.21703388740e-03 8.16346419979e-03 8.01216613758e-03 - 6.22225650652e-03 6.32497949646e-03 6.54341818442e-03 6.90130049286e-03 7.38199424752e-03 7.92776657338e-03 - 8.45428906874e-03 8.87112379151e-03 9.10176004354e-03 9.10176004354e-03 8.87112379151e-03 8.45428906874e-03 - 7.92776657338e-03 7.38199424752e-03 6.90130049286e-03 6.54341818442e-03 6.32497949646e-03 6.22225650652e-03 - 6.18966814651e-03 6.18550458410e-03 6.18689298800e-03 6.18689298800e-03 6.18550458410e-03 6.18966814651e-03 - 5.49418812711e-03 5.93416429324e-03 6.54341818442e-03 7.31454489847e-03 8.19475376688e-03 9.09041352545e-03 - 9.87827853279e-03 1.04248538239e-02 1.06212947837e-02 1.04248538239e-02 9.87827853279e-03 9.09041352545e-03 - 8.19475376688e-03 7.31454489847e-03 6.54341818442e-03 5.93416429324e-03 5.49418812711e-03 5.19548069751e-03 - 5.00022810356e-03 4.88536934411e-03 4.84662495208e-03 4.88536934411e-03 5.00022810356e-03 5.19548069751e-03 - 5.49418812711e-03 6.32497949646e-03 7.38199424752e-03 8.62132731210e-03 9.94413016678e-03 1.12010383875e-02 - 1.22047713863e-02 1.27676986907e-02 1.27676986907e-02 1.22047713863e-02 1.12010383875e-02 9.94413016678e-03 - 8.62132731210e-03 7.38199424752e-03 6.32497949646e-03 5.49418812711e-03 4.88266428153e-03 4.45430428341e-03 - 4.17601417174e-03 4.03593630205e-03 4.03593630205e-03 4.17601417174e-03 4.45430428341e-03 4.88266428153e-03 - 6.22225650652e-03 7.56111056333e-03 9.18607976908e-03 1.10006986257e-02 1.28345023601e-02 1.44498895771e-02 - 1.55749641354e-02 1.59804507535e-02 1.55749641354e-02 1.44498895771e-02 1.28345023601e-02 1.10006986257e-02 - 9.18607976908e-03 7.56111056333e-03 6.22225650652e-03 5.19548069751e-03 4.45430428341e-03 3.95332068977e-03 - 3.65964058851e-03 3.56204422580e-03 3.65964058851e-03 3.95332068977e-03 4.45430428341e-03 5.19548069751e-03 - 7.79360764654e-03 9.81666041939e-03 1.21839358632e-02 1.47188141408e-02 1.71372885777e-02 1.90731200357e-02 - 2.01594924058e-02 2.01594924058e-02 1.90731200357e-02 1.71372885777e-02 1.47188141408e-02 1.21839358632e-02 - 9.81666041939e-03 7.79360764654e-03 6.18966814651e-03 5.00022810356e-03 4.17601417174e-03 3.65964058851e-03 - 3.41039424489e-03 3.41039424489e-03 3.65964058851e-03 4.17601417174e-03 5.00022810356e-03 6.18966814651e-03 - 1.04118462818e-02 1.33587578257e-02 1.67052485052e-02 2.01439779904e-02 2.32096903098e-02 2.53584894772e-02 - 2.61347637909e-02 2.53584894772e-02 2.32096903098e-02 2.01439779904e-02 1.67052485052e-02 1.33587578257e-02 - 1.04118462818e-02 8.01216613758e-03 6.18550458410e-03 4.88536934411e-03 4.03593630205e-03 3.56204422580e-03 - 3.41039424489e-03 3.56204422580e-03 4.03593630205e-03 4.88536934411e-03 6.18550458410e-03 8.01216613758e-03 - 1.43606388158e-02 1.85584096087e-02 2.31896632144e-02 2.77206217649e-02 3.14092023629e-02 3.34978085182e-02 - 3.34978085182e-02 3.14092023629e-02 2.77206217649e-02 2.31896632144e-02 1.85584096087e-02 1.43606388158e-02 - 1.08721475193e-02 8.16346419979e-03 6.18689298800e-03 4.84662495208e-03 4.03593630205e-03 3.65964058851e-03 - 3.65964058851e-03 4.03593630205e-03 4.84662495208e-03 6.18689298800e-03 8.16346419979e-03 1.08721475193e-02 - 2.00031512771e-02 2.58559673304e-02 3.20717125323e-02 3.77513071960e-02 4.17937584090e-02 4.32645707778e-02 - 4.17937584090e-02 3.77513071960e-02 3.20717125323e-02 2.58559673304e-02 2.00031512771e-02 1.50351544907e-02 - 1.11222486043e-02 8.21703388740e-03 6.18689298800e-03 4.88536934411e-03 4.17601417174e-03 3.95332068977e-03 - 4.17601417174e-03 4.88536934411e-03 6.18689298800e-03 8.21703388740e-03 1.11222486043e-02 1.50351544907e-02 - 2.76901755249e-02 3.55533814714e-02 4.34476578087e-02 4.99875697106e-02 5.37274523246e-02 5.37274523246e-02 - 4.99875697106e-02 4.34476578087e-02 3.55533814714e-02 2.76901755249e-02 2.07965618706e-02 1.52731342228e-02 - 1.11222486043e-02 8.16346419979e-03 6.18550458410e-03 5.00022810356e-03 4.45430428341e-03 4.45430428341e-03 - 5.00022810356e-03 6.18550458410e-03 8.16346419979e-03 1.11222486043e-02 1.52731342228e-02 2.07965618706e-02 - 3.74959847737e-02 4.74317203940e-02 5.66502458441e-02 6.32743536159e-02 6.56955874028e-02 6.32743536159e-02 - 5.66502458441e-02 4.74317203940e-02 3.74959847737e-02 2.83448586606e-02 2.07965618706e-02 1.50351544907e-02 - 1.08721475193e-02 8.01216613758e-03 6.18966814651e-03 5.19548069751e-03 4.88266428153e-03 5.19548069751e-03 - 6.18966814651e-03 8.01216613758e-03 1.08721475193e-02 1.50351544907e-02 2.07965618706e-02 2.83448586606e-02 - 5.82846684414e-02 6.87390140452e-02 7.59717702126e-02 7.85611057280e-02 7.59717702126e-02 6.87390140452e-02 - 5.82846684414e-02 4.64972219765e-02 3.52270678377e-02 2.57739478448e-02 1.86064391176e-02 1.35182636086e-02 - 1.00461401111e-02 7.79360764654e-03 6.52118028642e-03 6.10931604239e-03 6.52118028642e-03 7.79360764654e-03 - 1.00461401111e-02 1.35182636086e-02 1.86064391176e-02 2.57739478448e-02 3.52270678377e-02 4.64972219765e-02 - 6.87390140452e-02 7.81603362307e-02 8.33624452951e-02 8.33624452951e-02 7.81603362307e-02 6.87390140452e-02 - 5.67802908281e-02 4.42738472380e-02 3.30014183614e-02 2.40042344862e-02 1.74298422700e-02 1.28681196083e-02 - 9.81666041939e-03 7.92776657338e-03 7.02039368822e-03 7.02039368822e-03 7.92776657338e-03 9.81666041939e-03 - 1.28681196083e-02 1.74298422700e-02 2.40042344862e-02 3.30014183614e-02 4.42738472380e-02 5.67802908281e-02 - 7.59717702126e-02 8.33624452951e-02 8.59799490006e-02 8.33624452951e-02 7.59717702126e-02 6.50724984809e-02 - 5.24827367699e-02 4.01880472274e-02 2.97324869747e-02 2.17534458677e-02 1.60740086611e-02 1.21839358632e-02 - 9.64234738867e-03 8.19475376688e-03 7.72360398793e-03 8.19475376688e-03 9.64234738867e-03 1.21839358632e-02 - 1.60740086611e-02 2.17534458677e-02 2.97324869747e-02 4.01880472274e-02 5.24827367699e-02 6.50724984809e-02 - 7.85611057280e-02 8.33624452951e-02 8.33624452951e-02 7.85611057280e-02 6.97193973966e-02 5.82637609313e-02 - 4.60708732894e-02 3.49472650912e-02 2.59885128162e-02 1.93739445237e-02 1.47188141408e-02 1.15565994691e-02 - 9.57706447476e-03 8.62132731210e-03 8.62132731210e-03 9.57706447476e-03 1.15565994691e-02 1.47188141408e-02 - 1.93739445237e-02 2.59885128162e-02 3.49472650912e-02 4.60708732894e-02 5.82637609313e-02 6.97193973966e-02 - 7.59717702126e-02 7.81603362307e-02 7.59717702126e-02 6.97193973966e-02 6.03520680545e-02 4.94024317434e-02 - 3.86408229568e-02 2.94353177762e-02 2.23208692653e-02 1.71372885777e-02 1.34835937042e-02 1.10438280884e-02 - 9.64234738867e-03 9.18607976908e-03 9.64234738867e-03 1.10438280884e-02 1.34835937042e-02 1.71372885777e-02 - 2.23208692653e-02 2.94353177762e-02 3.86408229568e-02 4.94024317434e-02 6.03520680545e-02 6.97193973966e-02 - 6.87390140452e-02 6.87390140452e-02 6.50724984809e-02 5.82637609313e-02 4.94024317434e-02 3.99801891127e-02 - 3.13851597832e-02 2.43885514961e-02 1.90731200357e-02 1.51716517813e-02 1.24193280193e-02 1.06646265060e-02 - 9.81666041939e-03 9.81666041939e-03 1.06646265060e-02 1.24193280193e-02 1.51716517813e-02 1.90731200357e-02 - 2.43885514961e-02 3.13851597832e-02 3.99801891127e-02 4.94024317434e-02 5.82637609313e-02 6.50724984809e-02 - 5.82846684414e-02 5.67802908281e-02 5.24827367699e-02 4.60708732894e-02 3.86408229568e-02 3.13851597832e-02 - 2.51358351357e-02 2.01594924058e-02 1.63414636372e-02 1.35012692014e-02 1.15426690136e-02 1.04118462818e-02 - 1.00461401111e-02 1.04118462818e-02 1.15426690136e-02 1.35012692014e-02 1.63414636372e-02 2.01594924058e-02 - 2.51358351357e-02 3.13851597832e-02 3.86408229568e-02 4.60708732894e-02 5.24827367699e-02 5.67802908281e-02 - 4.64972219765e-02 4.42738472380e-02 4.01880472274e-02 3.49472650912e-02 2.94353177762e-02 2.43885514961e-02 - 2.01594924058e-02 1.67618733320e-02 1.41011278955e-02 1.21379949101e-02 1.08721475193e-02 1.02670781727e-02 - 1.02670781727e-02 1.08721475193e-02 1.21379949101e-02 1.41011278955e-02 1.67618733320e-02 2.01594924058e-02 - 2.43885514961e-02 2.94353177762e-02 3.49472650912e-02 4.01880472274e-02 4.42738472380e-02 4.64972219765e-02 - 3.52270678377e-02 3.30014183614e-02 2.97324869747e-02 2.59885128162e-02 2.23208692653e-02 1.90731200357e-02 - 1.63414636372e-02 1.41011278955e-02 1.23490348915e-02 1.11222486043e-02 1.04242770349e-02 1.02031841081e-02 - 1.04242770349e-02 1.11222486043e-02 1.23490348915e-02 1.41011278955e-02 1.63414636372e-02 1.90731200357e-02 - 2.23208692653e-02 2.59885128162e-02 2.97324869747e-02 3.30014183614e-02 3.52270678377e-02 3.60143597485e-02 - 2.57739478448e-02 2.40042344862e-02 2.17534458677e-02 1.93739445237e-02 1.71372885777e-02 1.51716517813e-02 - 1.35012692014e-02 1.21379949101e-02 1.11222486043e-02 1.04809874105e-02 1.01850325403e-02 1.01850325403e-02 - 1.04809874105e-02 1.11222486043e-02 1.21379949101e-02 1.35012692014e-02 1.51716517813e-02 1.71372885777e-02 - 1.93739445237e-02 2.17534458677e-02 2.40042344862e-02 2.57739478448e-02 2.67482412330e-02 2.67482412330e-02 - 1.86064391176e-02 1.74298422700e-02 1.60740086611e-02 1.47188141408e-02 1.34835937042e-02 1.24193280193e-02 - 1.15426690136e-02 1.08721475193e-02 1.04242770349e-02 1.01850325403e-02 1.01130635710e-02 1.01850325403e-02 - 1.04242770349e-02 1.08721475193e-02 1.15426690136e-02 1.24193280193e-02 1.34835937042e-02 1.47188141408e-02 - 1.60740086611e-02 1.74298422700e-02 1.86064391176e-02 1.94109424364e-02 1.96971752186e-02 1.94109424364e-02 - 1.35182636086e-02 1.28681196083e-02 1.21839358632e-02 1.15565994691e-02 1.10438280884e-02 1.06646265060e-02 - 1.04118462818e-02 1.02670781727e-02 1.02031841081e-02 1.01850325403e-02 1.01850325403e-02 1.02031841081e-02 - 1.02670781727e-02 1.04118462818e-02 1.06646265060e-02 1.10438280884e-02 1.15565994691e-02 1.21839358632e-02 - 1.28681196083e-02 1.35182636086e-02 1.40306458540e-02 1.43139225142e-02 1.43139225142e-02 1.40306458540e-02 - 1.00461401111e-02 9.81666041939e-03 9.64234738867e-03 9.57706447476e-03 9.64234738867e-03 9.81666041939e-03 - 1.00461401111e-02 1.02670781727e-02 1.04242770349e-02 1.04809874105e-02 1.04242770349e-02 1.02670781727e-02 - 1.00461401111e-02 9.81666041939e-03 9.64234738867e-03 9.57706447476e-03 9.64234738867e-03 9.81666041939e-03 - 1.00461401111e-02 1.02670781727e-02 1.04242770349e-02 1.04809874105e-02 1.04242770349e-02 1.02670781727e-02 - 7.79360764654e-03 7.92776657338e-03 8.19475376688e-03 8.62132731210e-03 9.18607976908e-03 9.81666041939e-03 - 1.04118462818e-02 1.08721475193e-02 1.11222486043e-02 1.11222486043e-02 1.08721475193e-02 1.04118462818e-02 - 9.81666041939e-03 9.18607976908e-03 8.62132731210e-03 8.19475376688e-03 7.92776657338e-03 7.79360764654e-03 - 7.74242036277e-03 7.72923129933e-03 7.72761616008e-03 7.72761616008e-03 7.72923129933e-03 7.74242036277e-03 - 6.52118028642e-03 7.02039368822e-03 7.72360398793e-03 8.62132731210e-03 9.64234738867e-03 1.06646265060e-02 - 1.15426690136e-02 1.21379949101e-02 1.23490348915e-02 1.21379949101e-02 1.15426690136e-02 1.06646265060e-02 - 9.64234738867e-03 8.62132731210e-03 7.72360398793e-03 7.02039368822e-03 6.52118028642e-03 6.18966814651e-03 - 5.97825202186e-03 5.85659906617e-03 5.81607896972e-03 5.85659906617e-03 5.97825202186e-03 6.18966814651e-03 - 6.10931604239e-03 7.02039368822e-03 8.19475376688e-03 9.57706447476e-03 1.10438280884e-02 1.24193280193e-02 - 1.35012692014e-02 1.41011278955e-02 1.41011278955e-02 1.35012692014e-02 1.24193280193e-02 1.10438280884e-02 - 9.57706447476e-03 8.19475376688e-03 7.02039368822e-03 6.10931604239e-03 5.45146116297e-03 5.00022810356e-03 - 4.71220524742e-03 4.56877489732e-03 4.56877489732e-03 4.71220524742e-03 5.00022810356e-03 5.45146116297e-03 - 6.52118028642e-03 7.92776657338e-03 9.64234738867e-03 1.15565994691e-02 1.34835937042e-02 1.51716517813e-02 - 1.63414636372e-02 1.67618733320e-02 1.63414636372e-02 1.51716517813e-02 1.34835937042e-02 1.15565994691e-02 - 9.64234738867e-03 7.92776657338e-03 6.52118028642e-03 5.45146116297e-03 4.68737681852e-03 4.17601417174e-03 - 3.87834628559e-03 3.77975659219e-03 3.87834628559e-03 4.17601417174e-03 4.68737681852e-03 5.45146116297e-03 - 7.79360764654e-03 9.81666041939e-03 1.21839358632e-02 1.47188141408e-02 1.71372885777e-02 1.90731200357e-02 - 2.01594924058e-02 2.01594924058e-02 1.90731200357e-02 1.71372885777e-02 1.47188141408e-02 1.21839358632e-02 - 9.81666041939e-03 7.79360764654e-03 6.18966814651e-03 5.00022810356e-03 4.17601417174e-03 3.65964058851e-03 - 3.41039424489e-03 3.41039424489e-03 3.65964058851e-03 4.17601417174e-03 5.00022810356e-03 6.18966814651e-03 - 1.00461401111e-02 1.28681196083e-02 1.60740086611e-02 1.93739445237e-02 2.23208692653e-02 2.43885514961e-02 - 2.51358351357e-02 2.43885514961e-02 2.23208692653e-02 1.93739445237e-02 1.60740086611e-02 1.28681196083e-02 - 1.00461401111e-02 7.74242036277e-03 5.97825202186e-03 4.71220524742e-03 3.87834628559e-03 3.41039424489e-03 - 3.26020314086e-03 3.41039424489e-03 3.87834628559e-03 4.71220524742e-03 5.97825202186e-03 7.74242036277e-03 - 1.35182636086e-02 1.74298422700e-02 2.17534458677e-02 2.59885128162e-02 2.94353177762e-02 3.13851597832e-02 - 3.13851597832e-02 2.94353177762e-02 2.59885128162e-02 2.17534458677e-02 1.74298422700e-02 1.35182636086e-02 - 1.02670781727e-02 7.72923129933e-03 5.85659906617e-03 4.56877489732e-03 3.77975659219e-03 3.41039424489e-03 - 3.41039424489e-03 3.77975659219e-03 4.56877489732e-03 5.85659906617e-03 7.72923129933e-03 1.02670781727e-02 - 1.86064391176e-02 2.40042344862e-02 2.97324869747e-02 3.49472650912e-02 3.86408229568e-02 3.99801891127e-02 - 3.86408229568e-02 3.49472650912e-02 2.97324869747e-02 2.40042344862e-02 1.86064391176e-02 1.40306458540e-02 - 1.04242770349e-02 7.72761616008e-03 5.81607896972e-03 4.56877489732e-03 3.87834628559e-03 3.65964058851e-03 - 3.87834628559e-03 4.56877489732e-03 5.81607896972e-03 7.72761616008e-03 1.04242770349e-02 1.40306458540e-02 - 2.57739478448e-02 3.30014183614e-02 4.01880472274e-02 4.60708732894e-02 4.94024317434e-02 4.94024317434e-02 - 4.60708732894e-02 4.01880472274e-02 3.30014183614e-02 2.57739478448e-02 1.94109424364e-02 1.43139225142e-02 - 1.04809874105e-02 7.72761616008e-03 5.85659906617e-03 4.71220524742e-03 4.17601417174e-03 4.17601417174e-03 - 4.71220524742e-03 5.85659906617e-03 7.72761616008e-03 1.04809874105e-02 1.43139225142e-02 1.94109424364e-02 - 3.52270678377e-02 4.42738472380e-02 5.24827367699e-02 5.82637609313e-02 6.03520680545e-02 5.82637609313e-02 - 5.24827367699e-02 4.42738472380e-02 3.52270678377e-02 2.67482412330e-02 1.96971752186e-02 1.43139225142e-02 - 1.04242770349e-02 7.72923129933e-03 5.97825202186e-03 5.00022810356e-03 4.68737681852e-03 5.00022810356e-03 - 5.97825202186e-03 7.72923129933e-03 1.04242770349e-02 1.43139225142e-02 1.96971752186e-02 2.67482412330e-02 - 4.64972219765e-02 5.67802908281e-02 6.50724984809e-02 6.97193973966e-02 6.97193973966e-02 6.50724984809e-02 - 5.67802908281e-02 4.64972219765e-02 3.60143597485e-02 2.67482412330e-02 1.94109424364e-02 1.40306458540e-02 - 1.02670781727e-02 7.74242036277e-03 6.18966814651e-03 5.45146116297e-03 5.45146116297e-03 6.18966814651e-03 - 7.74242036277e-03 1.02670781727e-02 1.40306458540e-02 1.94109424364e-02 2.67482412330e-02 3.60143597485e-02 - 6.31142666785e-02 6.91691506706e-02 7.22242608078e-02 7.22242608078e-02 6.91691506706e-02 6.31142666785e-02 - 5.43728424707e-02 4.39611245421e-02 3.35736913190e-02 2.47523565945e-02 1.81349712987e-02 1.35182636086e-02 - 1.04118462818e-02 8.45428906874e-03 7.49319081292e-03 7.49319081292e-03 8.45428906874e-03 1.04118462818e-02 - 1.35182636086e-02 1.81349712987e-02 2.47523565945e-02 3.35736913190e-02 4.39611245421e-02 5.43728424707e-02 - 6.91691506706e-02 7.32259633424e-02 7.45491088161e-02 7.32259633424e-02 6.91691506706e-02 6.22085036765e-02 - 5.26323100532e-02 4.17752511891e-02 3.15640784817e-02 2.33592944544e-02 1.74298422700e-02 1.33587578257e-02 - 1.06646265060e-02 9.09041352545e-03 8.56724324800e-03 9.09041352545e-03 1.06646265060e-02 1.33587578257e-02 - 1.74298422700e-02 2.33592944544e-02 3.15640784817e-02 4.17752511891e-02 5.26323100532e-02 6.22085036765e-02 - 7.22242608078e-02 7.45491088161e-02 7.45491088161e-02 7.22242608078e-02 6.72242960853e-02 5.92671896593e-02 - 4.90171582760e-02 3.82730721923e-02 2.88969153857e-02 2.17534458677e-02 1.67052485052e-02 1.32551887233e-02 - 1.10438280884e-02 9.94413016678e-03 9.94413016678e-03 1.10438280884e-02 1.32551887233e-02 1.67052485052e-02 - 2.17534458677e-02 2.88969153857e-02 3.82730721923e-02 4.90171582760e-02 5.92671896593e-02 6.72242960853e-02 - 7.22242608078e-02 7.32259633424e-02 7.22242608078e-02 6.89438173057e-02 6.28462967289e-02 5.39959847166e-02 - 4.37078082031e-02 3.39275143754e-02 2.59885128162e-02 2.01439779904e-02 1.60207976494e-02 1.32196503024e-02 - 1.15565994691e-02 1.10006986257e-02 1.15565994691e-02 1.32196503024e-02 1.60207976494e-02 2.01439779904e-02 - 2.59885128162e-02 3.39275143754e-02 4.37078082031e-02 5.39959847166e-02 6.28462967289e-02 6.89438173057e-02 - 6.91691506706e-02 6.91691506706e-02 6.72242960853e-02 6.28462967289e-02 5.57693346902e-02 4.67604720778e-02 - 3.74820931920e-02 2.94353177762e-02 2.32096903098e-02 1.86453765526e-02 1.53937416744e-02 1.32551887233e-02 - 1.21839358632e-02 1.21839358632e-02 1.32551887233e-02 1.53937416744e-02 1.86453765526e-02 2.32096903098e-02 - 2.94353177762e-02 3.74820931920e-02 4.67604720778e-02 5.57693346902e-02 6.28462967289e-02 6.72242960853e-02 - 6.31142666785e-02 6.22085036765e-02 5.92671896593e-02 5.39959847166e-02 4.67604720778e-02 3.87690158276e-02 - 3.13851597832e-02 2.53584894772e-02 2.07351318935e-02 1.72789698058e-02 1.48305747723e-02 1.33587578257e-02 - 1.28681196083e-02 1.33587578257e-02 1.48305747723e-02 1.72789698058e-02 2.07351318935e-02 2.53584894772e-02 - 3.13851597832e-02 3.87690158276e-02 4.67604720778e-02 5.39959847166e-02 5.92671896593e-02 6.22085036765e-02 - 5.43728424707e-02 5.26323100532e-02 4.90171582760e-02 4.37078082031e-02 3.74820931920e-02 3.13851597832e-02 - 2.61347637909e-02 2.19036737434e-02 1.85732113773e-02 1.60537467662e-02 1.43606388158e-02 1.35182636086e-02 - 1.35182636086e-02 1.43606388158e-02 1.60537467662e-02 1.85732113773e-02 2.19036737434e-02 2.61347637909e-02 - 3.13851597832e-02 3.74820931920e-02 4.37078082031e-02 4.90171582760e-02 5.26323100532e-02 5.43728424707e-02 - 4.39611245421e-02 4.17752511891e-02 3.82730721923e-02 3.39275143754e-02 2.94353177762e-02 2.53584894772e-02 - 2.19036737434e-02 1.90357893277e-02 1.67247696023e-02 1.50351544907e-02 1.40306458540e-02 1.37029642554e-02 - 1.40306458540e-02 1.50351544907e-02 1.67247696023e-02 1.90357893277e-02 2.19036737434e-02 2.53584894772e-02 - 2.94353177762e-02 3.39275143754e-02 3.82730721923e-02 4.17752511891e-02 4.39611245421e-02 4.46898454449e-02 - 3.35736913190e-02 3.15640784817e-02 2.88969153857e-02 2.59885128162e-02 2.32096903098e-02 2.07351318935e-02 - 1.85732113773e-02 1.67247696023e-02 1.52731342228e-02 1.43139225142e-02 1.38569882819e-02 1.38569882819e-02 - 1.43139225142e-02 1.52731342228e-02 1.67247696023e-02 1.85732113773e-02 2.07351318935e-02 2.32096903098e-02 - 2.59885128162e-02 2.88969153857e-02 3.15640784817e-02 3.35736913190e-02 3.46378859300e-02 3.46378859300e-02 - 2.47523565945e-02 2.33592944544e-02 2.17534458677e-02 2.01439779904e-02 1.86453765526e-02 1.72789698058e-02 - 1.60537467662e-02 1.50351544907e-02 1.43139225142e-02 1.39177867633e-02 1.37978854292e-02 1.39177867633e-02 - 1.43139225142e-02 1.50351544907e-02 1.60537467662e-02 1.72789698058e-02 1.86453765526e-02 2.01439779904e-02 - 2.17534458677e-02 2.33592944544e-02 2.47523565945e-02 2.57020582842e-02 2.60389015581e-02 2.57020582842e-02 - 1.81349712987e-02 1.74298422700e-02 1.67052485052e-02 1.60207976494e-02 1.53937416744e-02 1.48305747723e-02 - 1.43606388158e-02 1.40306458540e-02 1.38569882819e-02 1.37978854292e-02 1.37978854292e-02 1.38569882819e-02 - 1.40306458540e-02 1.43606388158e-02 1.48305747723e-02 1.53937416744e-02 1.60207976494e-02 1.67052485052e-02 - 1.74298422700e-02 1.81349712987e-02 1.87149718700e-02 1.90473786081e-02 1.90473786081e-02 1.87149718700e-02 - 1.35182636086e-02 1.33587578257e-02 1.32551887233e-02 1.32196503024e-02 1.32551887233e-02 1.33587578257e-02 - 1.35182636086e-02 1.37029642554e-02 1.38569882819e-02 1.39177867633e-02 1.38569882819e-02 1.37029642554e-02 - 1.35182636086e-02 1.33587578257e-02 1.32551887233e-02 1.32196503024e-02 1.32551887233e-02 1.33587578257e-02 - 1.35182636086e-02 1.37029642554e-02 1.38569882819e-02 1.39177867633e-02 1.38569882819e-02 1.37029642554e-02 - 1.04118462818e-02 1.06646265060e-02 1.10438280884e-02 1.15565994691e-02 1.21839358632e-02 1.28681196083e-02 - 1.35182636086e-02 1.40306458540e-02 1.43139225142e-02 1.43139225142e-02 1.40306458540e-02 1.35182636086e-02 - 1.28681196083e-02 1.21839358632e-02 1.15565994691e-02 1.10438280884e-02 1.06646265060e-02 1.04118462818e-02 - 1.02670781727e-02 1.02031841081e-02 1.01850325403e-02 1.01850325403e-02 1.02031841081e-02 1.02670781727e-02 - 8.45428906874e-03 9.09041352545e-03 9.94413016678e-03 1.10006986257e-02 1.21839358632e-02 1.33587578257e-02 - 1.43606388158e-02 1.50351544907e-02 1.52731342228e-02 1.50351544907e-02 1.43606388158e-02 1.33587578257e-02 - 1.21839358632e-02 1.10006986257e-02 9.94413016678e-03 9.09041352545e-03 8.45428906874e-03 8.01216613758e-03 - 7.72923129933e-03 7.57424593471e-03 7.52528842907e-03 7.57424593471e-03 7.72923129933e-03 8.01216613758e-03 - 7.49319081292e-03 8.56724324800e-03 9.94413016678e-03 1.15565994691e-02 1.32551887233e-02 1.48305747723e-02 - 1.60537467662e-02 1.67247696023e-02 1.67247696023e-02 1.60537467662e-02 1.48305747723e-02 1.32551887233e-02 - 1.15565994691e-02 9.94413016678e-03 8.56724324800e-03 7.49319081292e-03 6.71488304626e-03 6.18550458410e-03 - 5.85659906617e-03 5.69805609664e-03 5.69805609664e-03 5.85659906617e-03 6.18550458410e-03 6.71488304626e-03 - 7.49319081292e-03 9.09041352545e-03 1.10438280884e-02 1.32196503024e-02 1.53937416744e-02 1.72789698058e-02 - 1.85732113773e-02 1.90357893277e-02 1.85732113773e-02 1.72789698058e-02 1.53937416744e-02 1.32196503024e-02 - 1.10438280884e-02 9.09041352545e-03 7.49319081292e-03 6.28902224701e-03 5.44128685160e-03 4.88536934411e-03 - 4.56877489732e-03 4.46542762661e-03 4.56877489732e-03 4.88536934411e-03 5.44128685160e-03 6.28902224701e-03 - 8.45428906874e-03 1.06646265060e-02 1.32551887233e-02 1.60207976494e-02 1.86453765526e-02 2.07351318935e-02 - 2.19036737434e-02 2.19036737434e-02 2.07351318935e-02 1.86453765526e-02 1.60207976494e-02 1.32551887233e-02 - 1.06646265060e-02 8.45428906874e-03 6.71488304626e-03 5.44128685160e-03 4.57233815477e-03 4.03593630205e-03 - 3.77975659219e-03 3.77975659219e-03 4.03593630205e-03 4.57233815477e-03 5.44128685160e-03 6.71488304626e-03 - 1.04118462818e-02 1.33587578257e-02 1.67052485052e-02 2.01439779904e-02 2.32096903098e-02 2.53584894772e-02 - 2.61347637909e-02 2.53584894772e-02 2.32096903098e-02 2.01439779904e-02 1.67052485052e-02 1.33587578257e-02 - 1.04118462818e-02 8.01216613758e-03 6.18550458410e-03 4.88536934411e-03 4.03593630205e-03 3.56204422580e-03 - 3.41039424489e-03 3.56204422580e-03 4.03593630205e-03 4.88536934411e-03 6.18550458410e-03 8.01216613758e-03 - 1.35182636086e-02 1.74298422700e-02 2.17534458677e-02 2.59885128162e-02 2.94353177762e-02 3.13851597832e-02 - 3.13851597832e-02 2.94353177762e-02 2.59885128162e-02 2.17534458677e-02 1.74298422700e-02 1.35182636086e-02 - 1.02670781727e-02 7.72923129933e-03 5.85659906617e-03 4.56877489732e-03 3.77975659219e-03 3.41039424489e-03 - 3.41039424489e-03 3.77975659219e-03 4.56877489732e-03 5.85659906617e-03 7.72923129933e-03 1.02670781727e-02 - 1.81349712987e-02 2.33592944544e-02 2.88969153857e-02 3.39275143754e-02 3.74820931920e-02 3.87690158276e-02 - 3.74820931920e-02 3.39275143754e-02 2.88969153857e-02 2.33592944544e-02 1.81349712987e-02 1.37029642554e-02 - 1.02031841081e-02 7.57424593471e-03 5.69805609664e-03 4.46542762661e-03 3.77975659219e-03 3.56204422580e-03 - 3.77975659219e-03 4.46542762661e-03 5.69805609664e-03 7.57424593471e-03 1.02031841081e-02 1.37029642554e-02 - 2.47523565945e-02 3.15640784817e-02 3.82730721923e-02 4.37078082031e-02 4.67604720778e-02 4.67604720778e-02 - 4.37078082031e-02 3.82730721923e-02 3.15640784817e-02 2.47523565945e-02 1.87149718700e-02 1.38569882819e-02 - 1.01850325403e-02 7.52528842907e-03 5.69805609664e-03 4.56877489732e-03 4.03593630205e-03 4.03593630205e-03 - 4.56877489732e-03 5.69805609664e-03 7.52528842907e-03 1.01850325403e-02 1.38569882819e-02 1.87149718700e-02 - 3.35736913190e-02 4.17752511891e-02 4.90171582760e-02 5.39959847166e-02 5.57693346902e-02 5.39959847166e-02 - 4.90171582760e-02 4.17752511891e-02 3.35736913190e-02 2.57020582842e-02 1.90473786081e-02 1.39177867633e-02 - 1.01850325403e-02 7.57424593471e-03 5.85659906617e-03 4.88536934411e-03 4.57233815477e-03 4.88536934411e-03 - 5.85659906617e-03 7.57424593471e-03 1.01850325403e-02 1.39177867633e-02 1.90473786081e-02 2.57020582842e-02 - 4.39611245421e-02 5.26323100532e-02 5.92671896593e-02 6.28462967289e-02 6.28462967289e-02 5.92671896593e-02 - 5.26323100532e-02 4.39611245421e-02 3.46378859300e-02 2.60389015581e-02 1.90473786081e-02 1.38569882819e-02 - 1.02031841081e-02 7.72923129933e-03 6.18550458410e-03 5.44128685160e-03 5.44128685160e-03 6.18550458410e-03 - 7.72923129933e-03 1.02031841081e-02 1.38569882819e-02 1.90473786081e-02 2.60389015581e-02 3.46378859300e-02 - 5.43728424707e-02 6.22085036765e-02 6.72242960853e-02 6.89438173057e-02 6.72242960853e-02 6.22085036765e-02 - 5.43728424707e-02 4.46898454449e-02 3.46378859300e-02 2.57020582842e-02 1.87149718700e-02 1.37029642554e-02 - 1.02670781727e-02 8.01216613758e-03 6.71488304626e-03 6.28902224701e-03 6.71488304626e-03 8.01216613758e-03 - 1.02670781727e-02 1.37029642554e-02 1.87149718700e-02 2.57020582842e-02 3.46378859300e-02 4.46898454449e-02 - 5.67646475266e-02 5.65857611912e-02 5.63481382706e-02 5.65857611912e-02 5.67646475266e-02 5.52660749302e-02 - 5.03920982744e-02 4.22555839350e-02 3.29591336478e-02 2.47523565945e-02 1.86064391176e-02 1.43606388158e-02 - 1.15426690136e-02 9.87827853279e-03 9.31927456850e-03 9.87827853279e-03 1.15426690136e-02 1.43606388158e-02 - 1.86064391176e-02 2.47523565945e-02 3.29591336478e-02 4.22555839350e-02 5.03920982744e-02 5.52660749302e-02 - 5.65857611912e-02 5.54677921285e-02 5.54677921285e-02 5.65857611912e-02 5.74779615393e-02 5.57017474383e-02 - 4.97519001270e-02 4.08013474350e-02 3.15640784817e-02 2.40042344862e-02 1.85584096087e-02 1.48305747723e-02 - 1.24193280193e-02 1.12010383875e-02 1.12010383875e-02 1.24193280193e-02 1.48305747723e-02 1.85584096087e-02 - 2.40042344862e-02 3.15640784817e-02 4.08013474350e-02 4.97519001270e-02 5.57017474383e-02 5.74779615393e-02 - 5.63481382706e-02 5.54677921285e-02 5.63481382706e-02 5.81927448576e-02 5.86480861577e-02 5.51986766687e-02 - 4.76311652618e-02 3.82730721923e-02 2.97324869747e-02 2.31896632144e-02 1.85555754828e-02 1.53937416744e-02 - 1.34835937042e-02 1.28345023601e-02 1.34835937042e-02 1.53937416744e-02 1.85555754828e-02 2.31896632144e-02 - 2.97324869747e-02 3.82730721923e-02 4.76311652618e-02 5.51986766687e-02 5.86480861577e-02 5.81927448576e-02 - 5.65857611912e-02 5.65857611912e-02 5.81927448576e-02 5.95705865689e-02 5.80220562298e-02 5.22694162731e-02 - 4.37078082031e-02 3.49472650912e-02 2.77206217649e-02 2.23681861845e-02 1.85605049655e-02 1.60207976494e-02 - 1.47188141408e-02 1.47188141408e-02 1.60207976494e-02 1.85605049655e-02 2.23681861845e-02 2.77206217649e-02 - 3.49472650912e-02 4.37078082031e-02 5.22694162731e-02 5.80220562298e-02 5.95705865689e-02 5.81927448576e-02 - 5.67646475266e-02 5.74779615393e-02 5.86480861577e-02 5.80220562298e-02 5.39094660610e-02 4.67604720778e-02 - 3.86408229568e-02 3.14092023629e-02 2.57560150434e-02 2.15543530804e-02 1.85555754828e-02 1.67052485052e-02 - 1.60740086611e-02 1.67052485052e-02 1.85555754828e-02 2.15543530804e-02 2.57560150434e-02 3.14092023629e-02 - 3.86408229568e-02 4.67604720778e-02 5.39094660610e-02 5.80220562298e-02 5.86480861577e-02 5.74779615393e-02 - 5.52660749302e-02 5.57017474383e-02 5.51986766687e-02 5.22694162731e-02 4.67604720778e-02 3.99801891127e-02 - 3.34978085182e-02 2.81262227272e-02 2.39287446557e-02 2.07503567214e-02 1.85584096087e-02 1.74298422700e-02 - 1.74298422700e-02 1.85584096087e-02 2.07503567214e-02 2.39287446557e-02 2.81262227272e-02 3.34978085182e-02 - 3.99801891127e-02 4.67604720778e-02 5.22694162731e-02 5.51986766687e-02 5.57017474383e-02 5.52660749302e-02 - 5.03920982744e-02 4.97519001270e-02 4.76311652618e-02 4.37078082031e-02 3.86408229568e-02 3.34978085182e-02 - 2.89819100992e-02 2.52525868671e-02 2.22527540382e-02 2.00031512771e-02 1.86064391176e-02 1.81349712987e-02 - 1.86064391176e-02 2.00031512771e-02 2.22527540382e-02 2.52525868671e-02 2.89819100992e-02 3.34978085182e-02 - 3.86408229568e-02 4.37078082031e-02 4.76311652618e-02 4.97519001270e-02 5.03920982744e-02 5.04488982123e-02 - 4.22555839350e-02 4.08013474350e-02 3.82730721923e-02 3.49472650912e-02 3.14092023629e-02 2.81262227272e-02 - 2.52525868671e-02 2.27884625410e-02 2.07965618706e-02 1.94109424364e-02 1.87149718700e-02 1.87149718700e-02 - 1.94109424364e-02 2.07965618706e-02 2.27884625410e-02 2.52525868671e-02 2.81262227272e-02 3.14092023629e-02 - 3.49472650912e-02 3.82730721923e-02 4.08013474350e-02 4.22555839350e-02 4.28256060156e-02 4.28256060156e-02 - 3.29591336478e-02 3.15640784817e-02 2.97324869747e-02 2.77206217649e-02 2.57560150434e-02 2.39287446557e-02 - 2.22527540382e-02 2.07965618706e-02 1.96971752186e-02 1.90473786081e-02 1.88392765173e-02 1.90473786081e-02 - 1.96971752186e-02 2.07965618706e-02 2.22527540382e-02 2.39287446557e-02 2.57560150434e-02 2.77206217649e-02 - 2.97324869747e-02 3.15640784817e-02 3.29591336478e-02 3.37861689888e-02 3.40527023771e-02 3.37861689888e-02 - 2.47523565945e-02 2.40042344862e-02 2.31896632144e-02 2.23681861845e-02 2.15543530804e-02 2.07503567214e-02 - 2.00031512771e-02 1.94109424364e-02 1.90473786081e-02 1.88953143126e-02 1.88953143126e-02 1.90473786081e-02 - 1.94109424364e-02 2.00031512771e-02 2.07503567214e-02 2.15543530804e-02 2.23681861845e-02 2.31896632144e-02 - 2.40042344862e-02 2.47523565945e-02 2.53397129065e-02 2.56667001275e-02 2.56667001275e-02 2.53397129065e-02 - 1.86064391176e-02 1.85584096087e-02 1.85555754828e-02 1.85605049655e-02 1.85555754828e-02 1.85584096087e-02 - 1.86064391176e-02 1.87149718700e-02 1.88392765173e-02 1.88953143126e-02 1.88392765173e-02 1.87149718700e-02 - 1.86064391176e-02 1.85584096087e-02 1.85555754828e-02 1.85605049655e-02 1.85555754828e-02 1.85584096087e-02 - 1.86064391176e-02 1.87149718700e-02 1.88392765173e-02 1.88953143126e-02 1.88392765173e-02 1.87149718700e-02 - 1.43606388158e-02 1.48305747723e-02 1.53937416744e-02 1.60207976494e-02 1.67052485052e-02 1.74298422700e-02 - 1.81349712987e-02 1.87149718700e-02 1.90473786081e-02 1.90473786081e-02 1.87149718700e-02 1.81349712987e-02 - 1.74298422700e-02 1.67052485052e-02 1.60207976494e-02 1.53937416744e-02 1.48305747723e-02 1.43606388158e-02 - 1.40306458540e-02 1.38569882819e-02 1.37978854292e-02 1.37978854292e-02 1.38569882819e-02 1.40306458540e-02 - 1.15426690136e-02 1.24193280193e-02 1.34835937042e-02 1.47188141408e-02 1.60740086611e-02 1.74298422700e-02 - 1.86064391176e-02 1.94109424364e-02 1.96971752186e-02 1.94109424364e-02 1.86064391176e-02 1.74298422700e-02 - 1.60740086611e-02 1.47188141408e-02 1.34835937042e-02 1.24193280193e-02 1.15426690136e-02 1.08721475193e-02 - 1.04242770349e-02 1.01850325403e-02 1.01130635710e-02 1.01850325403e-02 1.04242770349e-02 1.08721475193e-02 - 9.87827853279e-03 1.12010383875e-02 1.28345023601e-02 1.47188141408e-02 1.67052485052e-02 1.85584096087e-02 - 2.00031512771e-02 2.07965618706e-02 2.07965618706e-02 2.00031512771e-02 1.85584096087e-02 1.67052485052e-02 - 1.47188141408e-02 1.28345023601e-02 1.12010383875e-02 9.87827853279e-03 8.87112379151e-03 8.16346419979e-03 - 7.72761616008e-03 7.52528842907e-03 7.52528842907e-03 7.72761616008e-03 8.16346419979e-03 8.87112379151e-03 - 9.31927456850e-03 1.12010383875e-02 1.34835937042e-02 1.60207976494e-02 1.85555754828e-02 2.07503567214e-02 - 2.22527540382e-02 2.27884625410e-02 2.22527540382e-02 2.07503567214e-02 1.85555754828e-02 1.60207976494e-02 - 1.34835937042e-02 1.12010383875e-02 9.31927456850e-03 7.87995370977e-03 6.85502201162e-03 6.18689298800e-03 - 5.81607896972e-03 5.69805609664e-03 5.81607896972e-03 6.18689298800e-03 6.85502201162e-03 7.87995370977e-03 - 9.87827853279e-03 1.24193280193e-02 1.53937416744e-02 1.85605049655e-02 2.15543530804e-02 2.39287446557e-02 - 2.52525868671e-02 2.52525868671e-02 2.39287446557e-02 2.15543530804e-02 1.85605049655e-02 1.53937416744e-02 - 1.24193280193e-02 9.87827853279e-03 7.87995370977e-03 6.42332605072e-03 5.44101481998e-03 4.84662495208e-03 - 4.56877489732e-03 4.56877489732e-03 4.84662495208e-03 5.44101481998e-03 6.42332605072e-03 7.87995370977e-03 - 1.15426690136e-02 1.48305747723e-02 1.85555754828e-02 2.23681861845e-02 2.57560150434e-02 2.81262227272e-02 - 2.89819100992e-02 2.81262227272e-02 2.57560150434e-02 2.23681861845e-02 1.85555754828e-02 1.48305747723e-02 - 1.15426690136e-02 8.87112379151e-03 6.85502201162e-03 5.44101481998e-03 4.53364885369e-03 4.03593630205e-03 - 3.87834628559e-03 4.03593630205e-03 4.53364885369e-03 5.44101481998e-03 6.85502201162e-03 8.87112379151e-03 - 1.43606388158e-02 1.85584096087e-02 2.31896632144e-02 2.77206217649e-02 3.14092023629e-02 3.34978085182e-02 - 3.34978085182e-02 3.14092023629e-02 2.77206217649e-02 2.31896632144e-02 1.85584096087e-02 1.43606388158e-02 - 1.08721475193e-02 8.16346419979e-03 6.18689298800e-03 4.84662495208e-03 4.03593630205e-03 3.65964058851e-03 - 3.65964058851e-03 4.03593630205e-03 4.84662495208e-03 6.18689298800e-03 8.16346419979e-03 1.08721475193e-02 - 1.86064391176e-02 2.40042344862e-02 2.97324869747e-02 3.49472650912e-02 3.86408229568e-02 3.99801891127e-02 - 3.86408229568e-02 3.49472650912e-02 2.97324869747e-02 2.40042344862e-02 1.86064391176e-02 1.40306458540e-02 - 1.04242770349e-02 7.72761616008e-03 5.81607896972e-03 4.56877489732e-03 3.87834628559e-03 3.65964058851e-03 - 3.87834628559e-03 4.56877489732e-03 5.81607896972e-03 7.72761616008e-03 1.04242770349e-02 1.40306458540e-02 - 2.47523565945e-02 3.15640784817e-02 3.82730721923e-02 4.37078082031e-02 4.67604720778e-02 4.67604720778e-02 - 4.37078082031e-02 3.82730721923e-02 3.15640784817e-02 2.47523565945e-02 1.87149718700e-02 1.38569882819e-02 - 1.01850325403e-02 7.52528842907e-03 5.69805609664e-03 4.56877489732e-03 4.03593630205e-03 4.03593630205e-03 - 4.56877489732e-03 5.69805609664e-03 7.52528842907e-03 1.01850325403e-02 1.38569882819e-02 1.87149718700e-02 - 3.29591336478e-02 4.08013474350e-02 4.76311652618e-02 5.22694162731e-02 5.39094660610e-02 5.22694162731e-02 - 4.76311652618e-02 4.08013474350e-02 3.29591336478e-02 2.53397129065e-02 1.88392765173e-02 1.37978854292e-02 - 1.01130635710e-02 7.52528842907e-03 5.81607896972e-03 4.84662495208e-03 4.53364885369e-03 4.84662495208e-03 - 5.81607896972e-03 7.52528842907e-03 1.01130635710e-02 1.37978854292e-02 1.88392765173e-02 2.53397129065e-02 - 4.22555839350e-02 4.97519001270e-02 5.51986766687e-02 5.80220562298e-02 5.80220562298e-02 5.51986766687e-02 - 4.97519001270e-02 4.22555839350e-02 3.37861689888e-02 2.56667001275e-02 1.88953143126e-02 1.37978854292e-02 - 1.01850325403e-02 7.72761616008e-03 6.18689298800e-03 5.44101481998e-03 5.44101481998e-03 6.18689298800e-03 - 7.72761616008e-03 1.01850325403e-02 1.37978854292e-02 1.88953143126e-02 2.56667001275e-02 3.37861689888e-02 - 5.03920982744e-02 5.57017474383e-02 5.86480861577e-02 5.95705865689e-02 5.86480861577e-02 5.57017474383e-02 - 5.03920982744e-02 4.28256060156e-02 3.40527023771e-02 2.56667001275e-02 1.88392765173e-02 1.38569882819e-02 - 1.04242770349e-02 8.16346419979e-03 6.85502201162e-03 6.42332605072e-03 6.85502201162e-03 8.16346419979e-03 - 1.04242770349e-02 1.38569882819e-02 1.88392765173e-02 2.56667001275e-02 3.40527023771e-02 4.28256060156e-02 - 5.52660749302e-02 5.74779615393e-02 5.81927448576e-02 5.81927448576e-02 5.74779615393e-02 5.52660749302e-02 - 5.04488982123e-02 4.28256060156e-02 3.37861689888e-02 2.53397129065e-02 1.87149718700e-02 1.40306458540e-02 - 1.08721475193e-02 8.87112379151e-03 7.87995370977e-03 7.87995370977e-03 8.87112379151e-03 1.08721475193e-02 - 1.40306458540e-02 1.87149718700e-02 2.53397129065e-02 3.37861689888e-02 4.28256060156e-02 5.04488982123e-02 - 3.86813628609e-02 3.47728130512e-02 3.47728130512e-02 3.86813628609e-02 4.49731492966e-02 4.96150823327e-02 - 4.87050863807e-02 4.22555839350e-02 3.35736913190e-02 2.57739478448e-02 2.00031512771e-02 1.60537467662e-02 - 1.35012692014e-02 1.22047713863e-02 1.22047713863e-02 1.35012692014e-02 1.60537467662e-02 2.00031512771e-02 - 2.57739478448e-02 3.35736913190e-02 4.22555839350e-02 4.87050863807e-02 4.96150823327e-02 4.49731492966e-02 - 3.47728130512e-02 3.23216192243e-02 3.47728130512e-02 4.14983823737e-02 4.93152444260e-02 5.29690593685e-02 - 4.97519001270e-02 4.17752511891e-02 3.30014183614e-02 2.58559673304e-02 2.07503567214e-02 1.72789698058e-02 - 1.51716517813e-02 1.44498895771e-02 1.51716517813e-02 1.72789698058e-02 2.07503567214e-02 2.58559673304e-02 - 3.30014183614e-02 4.17752511891e-02 4.97519001270e-02 5.29690593685e-02 4.93152444260e-02 4.14983823737e-02 - 3.47728130512e-02 3.47728130512e-02 4.01124125220e-02 4.85514460951e-02 5.50728972421e-02 5.51986766687e-02 - 4.90171582760e-02 4.01880472274e-02 3.20717125323e-02 2.59124227944e-02 2.15543530804e-02 1.86453765526e-02 - 1.71372885777e-02 1.71372885777e-02 1.86453765526e-02 2.15543530804e-02 2.59124227944e-02 3.20717125323e-02 - 4.01880472274e-02 4.90171582760e-02 5.51986766687e-02 5.50728972421e-02 4.85514460951e-02 4.01124125220e-02 - 3.86813628609e-02 4.14983823737e-02 4.85514460951e-02 5.56741558656e-02 5.80220562298e-02 5.39959847166e-02 - 4.60708732894e-02 3.77513071960e-02 3.09544991478e-02 2.59309064966e-02 2.23681861845e-02 2.01439779904e-02 - 1.93739445237e-02 2.01439779904e-02 2.23681861845e-02 2.59309064966e-02 3.09544991478e-02 3.77513071960e-02 - 4.60708732894e-02 5.39959847166e-02 5.80220562298e-02 5.56741558656e-02 4.85514460951e-02 4.14983823737e-02 - 4.49731492966e-02 4.93152444260e-02 5.50728972421e-02 5.80220562298e-02 5.57693346902e-02 4.94024317434e-02 - 4.17937584090e-02 3.50622322206e-02 2.98227515673e-02 2.59124227944e-02 2.31896632144e-02 2.17534458677e-02 - 2.17534458677e-02 2.31896632144e-02 2.59124227944e-02 2.98227515673e-02 3.50622322206e-02 4.17937584090e-02 - 4.94024317434e-02 5.57693346902e-02 5.80220562298e-02 5.50728972421e-02 4.93152444260e-02 4.49731492966e-02 - 4.96150823327e-02 5.29690593685e-02 5.51986766687e-02 5.39959847166e-02 4.94024317434e-02 4.32645707778e-02 - 3.73919347545e-02 3.25477771477e-02 2.87349610469e-02 2.58559673304e-02 2.40042344862e-02 2.33592944544e-02 - 2.40042344862e-02 2.58559673304e-02 2.87349610469e-02 3.25477771477e-02 3.73919347545e-02 4.32645707778e-02 - 4.94024317434e-02 5.39959847166e-02 5.51986766687e-02 5.29690593685e-02 4.96150823327e-02 4.80686289281e-02 - 4.87050863807e-02 4.97519001270e-02 4.90171582760e-02 4.60708732894e-02 4.17937584090e-02 3.73919347545e-02 - 3.35320462758e-02 3.03107984462e-02 2.76901755249e-02 2.57739478448e-02 2.47523565945e-02 2.47523565945e-02 - 2.57739478448e-02 2.76901755249e-02 3.03107984462e-02 3.35320462758e-02 3.73919347545e-02 4.17937584090e-02 - 4.60708732894e-02 4.90171582760e-02 4.97519001270e-02 4.87050863807e-02 4.74726445019e-02 4.74726445019e-02 - 4.22555839350e-02 4.17752511891e-02 4.01880472274e-02 3.77513071960e-02 3.50622322206e-02 3.25477771477e-02 - 3.03107984462e-02 2.83448586606e-02 2.67482412330e-02 2.57020582842e-02 2.53397129065e-02 2.57020582842e-02 - 2.67482412330e-02 2.83448586606e-02 3.03107984462e-02 3.25477771477e-02 3.50622322206e-02 3.77513071960e-02 - 4.01880472274e-02 4.17752511891e-02 4.22555839350e-02 4.20317983678e-02 4.18365826600e-02 4.20317983678e-02 - 3.35736913190e-02 3.30014183614e-02 3.20717125323e-02 3.09544991478e-02 2.98227515673e-02 2.87349610469e-02 - 2.76901755249e-02 2.67482412330e-02 2.60389015581e-02 2.56667001275e-02 2.56667001275e-02 2.60389015581e-02 - 2.67482412330e-02 2.76901755249e-02 2.87349610469e-02 2.98227515673e-02 3.09544991478e-02 3.20717125323e-02 - 3.30014183614e-02 3.35736913190e-02 3.37861689888e-02 3.38082178713e-02 3.38082178713e-02 3.37861689888e-02 - 2.57739478448e-02 2.58559673304e-02 2.59124227944e-02 2.59309064966e-02 2.59124227944e-02 2.58559673304e-02 - 2.57739478448e-02 2.57020582842e-02 2.56667001275e-02 2.56591838074e-02 2.56667001275e-02 2.57020582842e-02 - 2.57739478448e-02 2.58559673304e-02 2.59124227944e-02 2.59309064966e-02 2.59124227944e-02 2.58559673304e-02 - 2.57739478448e-02 2.57020582842e-02 2.56667001275e-02 2.56591838074e-02 2.56667001275e-02 2.57020582842e-02 - 2.00031512771e-02 2.07503567214e-02 2.15543530804e-02 2.23681861845e-02 2.31896632144e-02 2.40042344862e-02 - 2.47523565945e-02 2.53397129065e-02 2.56667001275e-02 2.56667001275e-02 2.53397129065e-02 2.47523565945e-02 - 2.40042344862e-02 2.31896632144e-02 2.23681861845e-02 2.15543530804e-02 2.07503567214e-02 2.00031512771e-02 - 1.94109424364e-02 1.90473786081e-02 1.88953143126e-02 1.88953143126e-02 1.90473786081e-02 1.94109424364e-02 - 1.60537467662e-02 1.72789698058e-02 1.86453765526e-02 2.01439779904e-02 2.17534458677e-02 2.33592944544e-02 - 2.47523565945e-02 2.57020582842e-02 2.60389015581e-02 2.57020582842e-02 2.47523565945e-02 2.33592944544e-02 - 2.17534458677e-02 2.01439779904e-02 1.86453765526e-02 1.72789698058e-02 1.60537467662e-02 1.50351544907e-02 - 1.43139225142e-02 1.39177867633e-02 1.37978854292e-02 1.39177867633e-02 1.43139225142e-02 1.50351544907e-02 - 1.35012692014e-02 1.51716517813e-02 1.71372885777e-02 1.93739445237e-02 2.17534458677e-02 2.40042344862e-02 - 2.57739478448e-02 2.67482412330e-02 2.67482412330e-02 2.57739478448e-02 2.40042344862e-02 2.17534458677e-02 - 1.93739445237e-02 1.71372885777e-02 1.51716517813e-02 1.35012692014e-02 1.21379949101e-02 1.11222486043e-02 - 1.04809874105e-02 1.01850325403e-02 1.01850325403e-02 1.04809874105e-02 1.11222486043e-02 1.21379949101e-02 - 1.22047713863e-02 1.44498895771e-02 1.71372885777e-02 2.01439779904e-02 2.31896632144e-02 2.58559673304e-02 - 2.76901755249e-02 2.83448586606e-02 2.76901755249e-02 2.58559673304e-02 2.31896632144e-02 2.01439779904e-02 - 1.71372885777e-02 1.44498895771e-02 1.22047713863e-02 1.04248538239e-02 9.10176004354e-03 8.21703388740e-03 - 7.72761616008e-03 7.57424593471e-03 7.72761616008e-03 8.21703388740e-03 9.10176004354e-03 1.04248538239e-02 - 1.22047713863e-02 1.51716517813e-02 1.86453765526e-02 2.23681861845e-02 2.59124227944e-02 2.87349610469e-02 - 3.03107984462e-02 3.03107984462e-02 2.87349610469e-02 2.59124227944e-02 2.23681861845e-02 1.86453765526e-02 - 1.51716517813e-02 1.22047713863e-02 9.84753974693e-03 8.09940143038e-03 6.90595653158e-03 6.18689298800e-03 - 5.85659906617e-03 5.85659906617e-03 6.18689298800e-03 6.90595653158e-03 8.09940143038e-03 9.84753974693e-03 - 1.35012692014e-02 1.72789698058e-02 2.15543530804e-02 2.59309064966e-02 2.98227515673e-02 3.25477771477e-02 - 3.35320462758e-02 3.25477771477e-02 2.98227515673e-02 2.59309064966e-02 2.15543530804e-02 1.72789698058e-02 - 1.35012692014e-02 1.04248538239e-02 8.09940143038e-03 6.47303924297e-03 5.44101481998e-03 4.88536934411e-03 - 4.71220524742e-03 4.88536934411e-03 5.44101481998e-03 6.47303924297e-03 8.09940143038e-03 1.04248538239e-02 - 1.60537467662e-02 2.07503567214e-02 2.59124227944e-02 3.09544991478e-02 3.50622322206e-02 3.73919347545e-02 - 3.73919347545e-02 3.50622322206e-02 3.09544991478e-02 2.59124227944e-02 2.07503567214e-02 1.60537467662e-02 - 1.21379949101e-02 9.10176004354e-03 6.90595653158e-03 5.44101481998e-03 4.57233815477e-03 4.17601417174e-03 - 4.17601417174e-03 4.57233815477e-03 5.44101481998e-03 6.90595653158e-03 9.10176004354e-03 1.21379949101e-02 - 2.00031512771e-02 2.58559673304e-02 3.20717125323e-02 3.77513071960e-02 4.17937584090e-02 4.32645707778e-02 - 4.17937584090e-02 3.77513071960e-02 3.20717125323e-02 2.58559673304e-02 2.00031512771e-02 1.50351544907e-02 - 1.11222486043e-02 8.21703388740e-03 6.18689298800e-03 4.88536934411e-03 4.17601417174e-03 3.95332068977e-03 - 4.17601417174e-03 4.88536934411e-03 6.18689298800e-03 8.21703388740e-03 1.11222486043e-02 1.50351544907e-02 - 2.57739478448e-02 3.30014183614e-02 4.01880472274e-02 4.60708732894e-02 4.94024317434e-02 4.94024317434e-02 - 4.60708732894e-02 4.01880472274e-02 3.30014183614e-02 2.57739478448e-02 1.94109424364e-02 1.43139225142e-02 - 1.04809874105e-02 7.72761616008e-03 5.85659906617e-03 4.71220524742e-03 4.17601417174e-03 4.17601417174e-03 - 4.71220524742e-03 5.85659906617e-03 7.72761616008e-03 1.04809874105e-02 1.43139225142e-02 1.94109424364e-02 - 3.35736913190e-02 4.17752511891e-02 4.90171582760e-02 5.39959847166e-02 5.57693346902e-02 5.39959847166e-02 - 4.90171582760e-02 4.17752511891e-02 3.35736913190e-02 2.57020582842e-02 1.90473786081e-02 1.39177867633e-02 - 1.01850325403e-02 7.57424593471e-03 5.85659906617e-03 4.88536934411e-03 4.57233815477e-03 4.88536934411e-03 - 5.85659906617e-03 7.57424593471e-03 1.01850325403e-02 1.39177867633e-02 1.90473786081e-02 2.57020582842e-02 - 4.22555839350e-02 4.97519001270e-02 5.51986766687e-02 5.80220562298e-02 5.80220562298e-02 5.51986766687e-02 - 4.97519001270e-02 4.22555839350e-02 3.37861689888e-02 2.56667001275e-02 1.88953143126e-02 1.37978854292e-02 - 1.01850325403e-02 7.72761616008e-03 6.18689298800e-03 5.44101481998e-03 5.44101481998e-03 6.18689298800e-03 - 7.72761616008e-03 1.01850325403e-02 1.37978854292e-02 1.88953143126e-02 2.56667001275e-02 3.37861689888e-02 - 4.87050863807e-02 5.29690593685e-02 5.50728972421e-02 5.56741558656e-02 5.50728972421e-02 5.29690593685e-02 - 4.87050863807e-02 4.20317983678e-02 3.38082178713e-02 2.56591838074e-02 1.88953143126e-02 1.39177867633e-02 - 1.04809874105e-02 8.21703388740e-03 6.90595653158e-03 6.47303924297e-03 6.90595653158e-03 8.21703388740e-03 - 1.04809874105e-02 1.39177867633e-02 1.88953143126e-02 2.56591838074e-02 3.38082178713e-02 4.20317983678e-02 - 4.96150823327e-02 4.93152444260e-02 4.85514460951e-02 4.85514460951e-02 4.93152444260e-02 4.96150823327e-02 - 4.74726445019e-02 4.18365826600e-02 3.38082178713e-02 2.56667001275e-02 1.90473786081e-02 1.43139225142e-02 - 1.11222486043e-02 9.10176004354e-03 8.09940143038e-03 8.09940143038e-03 9.10176004354e-03 1.11222486043e-02 - 1.43139225142e-02 1.90473786081e-02 2.56667001275e-02 3.38082178713e-02 4.18365826600e-02 4.74726445019e-02 - 4.49731492966e-02 4.14983823737e-02 4.01124125220e-02 4.14983823737e-02 4.49731492966e-02 4.80686289281e-02 - 4.74726445019e-02 4.20317983678e-02 3.37861689888e-02 2.57020582842e-02 1.94109424364e-02 1.50351544907e-02 - 1.21379949101e-02 1.04248538239e-02 9.84753974693e-03 1.04248538239e-02 1.21379949101e-02 1.50351544907e-02 - 1.94109424364e-02 2.57020582842e-02 3.37861689888e-02 4.20317983678e-02 4.74726445019e-02 4.80686289281e-02 - 1.79849681520e-02 1.49395756337e-02 1.79849681520e-02 2.72980449809e-02 4.01965567029e-02 4.96150823327e-02 - 5.03920982744e-02 4.39611245421e-02 3.52270678377e-02 2.76901755249e-02 2.22527540382e-02 1.85732113773e-02 - 1.63414636372e-02 1.55749641354e-02 1.63414636372e-02 1.85732113773e-02 2.22527540382e-02 2.76901755249e-02 - 3.52270678377e-02 4.39611245421e-02 5.03920982744e-02 4.96150823327e-02 4.01965567029e-02 2.72980449809e-02 - 1.49395756337e-02 1.49395756337e-02 2.21086562145e-02 3.55340946595e-02 4.93152444260e-02 5.57017474383e-02 - 5.26323100532e-02 4.42738472380e-02 3.55533814714e-02 2.87349610469e-02 2.39287446557e-02 2.07351318935e-02 - 1.90731200357e-02 1.90731200357e-02 2.07351318935e-02 2.39287446557e-02 2.87349610469e-02 3.55533814714e-02 - 4.42738472380e-02 5.26323100532e-02 5.57017474383e-02 4.93152444260e-02 3.55340946595e-02 2.21086562145e-02 - 1.79849681520e-02 2.21086562145e-02 3.37527793613e-02 4.85514460951e-02 5.86480861577e-02 5.92671896593e-02 - 5.24827367699e-02 4.34476578087e-02 3.56148556138e-02 2.98227515673e-02 2.57560150434e-02 2.32096903098e-02 - 2.23208692653e-02 2.32096903098e-02 2.57560150434e-02 2.98227515673e-02 3.56148556138e-02 4.34476578087e-02 - 5.24827367699e-02 5.92671896593e-02 5.86480861577e-02 4.85514460951e-02 3.37527793613e-02 2.21086562145e-02 - 2.72980449809e-02 3.55340946595e-02 4.85514460951e-02 5.95705865689e-02 6.28462967289e-02 5.82637609313e-02 - 4.99875697106e-02 4.19230722756e-02 3.55984989831e-02 3.09544991478e-02 2.77206217649e-02 2.59885128162e-02 - 2.59885128162e-02 2.77206217649e-02 3.09544991478e-02 3.55984989831e-02 4.19230722756e-02 4.99875697106e-02 - 5.82637609313e-02 6.28462967289e-02 5.95705865689e-02 4.85514460951e-02 3.55340946595e-02 2.72980449809e-02 - 4.01965567029e-02 4.93152444260e-02 5.86480861577e-02 6.28462967289e-02 6.03520680545e-02 5.37274523246e-02 - 4.64352697346e-02 4.03211904768e-02 3.56148556138e-02 3.20717125323e-02 2.97324869747e-02 2.88969153857e-02 - 2.97324869747e-02 3.20717125323e-02 3.56148556138e-02 4.03211904768e-02 4.64352697346e-02 5.37274523246e-02 - 6.03520680545e-02 6.28462967289e-02 5.86480861577e-02 4.93152444260e-02 4.01965567029e-02 3.64979381134e-02 - 4.96150823327e-02 5.57017474383e-02 5.92671896593e-02 5.82637609313e-02 5.37274523246e-02 4.80826222932e-02 - 4.30069852462e-02 3.88905177841e-02 3.55533814714e-02 3.30014183614e-02 3.15640784817e-02 3.15640784817e-02 - 3.30014183614e-02 3.55533814714e-02 3.88905177841e-02 4.30069852462e-02 4.80826222932e-02 5.37274523246e-02 - 5.82637609313e-02 5.92671896593e-02 5.57017474383e-02 4.96150823327e-02 4.50421513586e-02 4.50421513586e-02 - 5.03920982744e-02 5.26323100532e-02 5.24827367699e-02 4.99875697106e-02 4.64352697346e-02 4.30069852462e-02 - 4.00749900281e-02 3.74959847737e-02 3.52270678377e-02 3.35736913190e-02 3.29591336478e-02 3.35736913190e-02 - 3.52270678377e-02 3.74959847737e-02 4.00749900281e-02 4.30069852462e-02 4.64352697346e-02 4.99875697106e-02 - 5.24827367699e-02 5.26323100532e-02 5.03920982744e-02 4.74726445019e-02 4.61549122721e-02 4.74726445019e-02 - 4.39611245421e-02 4.42738472380e-02 4.34476578087e-02 4.19230722756e-02 4.03211904768e-02 3.88905177841e-02 - 3.74959847737e-02 3.60143597485e-02 3.46378859300e-02 3.37861689888e-02 3.37861689888e-02 3.46378859300e-02 - 3.60143597485e-02 3.74959847737e-02 3.88905177841e-02 4.03211904768e-02 4.19230722756e-02 4.34476578087e-02 - 4.42738472380e-02 4.39611245421e-02 4.28256060156e-02 4.18365826600e-02 4.18365826600e-02 4.28256060156e-02 - 3.52270678377e-02 3.55533814714e-02 3.56148556138e-02 3.55984989831e-02 3.56148556138e-02 3.55533814714e-02 - 3.52270678377e-02 3.46378859300e-02 3.40527023771e-02 3.38082178713e-02 3.40527023771e-02 3.46378859300e-02 - 3.52270678377e-02 3.55533814714e-02 3.56148556138e-02 3.55984989831e-02 3.56148556138e-02 3.55533814714e-02 - 3.52270678377e-02 3.46378859300e-02 3.40527023771e-02 3.38082178713e-02 3.40527023771e-02 3.46378859300e-02 - 2.76901755249e-02 2.87349610469e-02 2.98227515673e-02 3.09544991478e-02 3.20717125323e-02 3.30014183614e-02 - 3.35736913190e-02 3.37861689888e-02 3.38082178713e-02 3.38082178713e-02 3.37861689888e-02 3.35736913190e-02 - 3.30014183614e-02 3.20717125323e-02 3.09544991478e-02 2.98227515673e-02 2.87349610469e-02 2.76901755249e-02 - 2.67482412330e-02 2.60389015581e-02 2.56667001275e-02 2.56667001275e-02 2.60389015581e-02 2.67482412330e-02 - 2.22527540382e-02 2.39287446557e-02 2.57560150434e-02 2.77206217649e-02 2.97324869747e-02 3.15640784817e-02 - 3.29591336478e-02 3.37861689888e-02 3.40527023771e-02 3.37861689888e-02 3.29591336478e-02 3.15640784817e-02 - 2.97324869747e-02 2.77206217649e-02 2.57560150434e-02 2.39287446557e-02 2.22527540382e-02 2.07965618706e-02 - 1.96971752186e-02 1.90473786081e-02 1.88392765173e-02 1.90473786081e-02 1.96971752186e-02 2.07965618706e-02 - 1.85732113773e-02 2.07351318935e-02 2.32096903098e-02 2.59885128162e-02 2.88969153857e-02 3.15640784817e-02 - 3.35736913190e-02 3.46378859300e-02 3.46378859300e-02 3.35736913190e-02 3.15640784817e-02 2.88969153857e-02 - 2.59885128162e-02 2.32096903098e-02 2.07351318935e-02 1.85732113773e-02 1.67247696023e-02 1.52731342228e-02 - 1.43139225142e-02 1.38569882819e-02 1.38569882819e-02 1.43139225142e-02 1.52731342228e-02 1.67247696023e-02 - 1.63414636372e-02 1.90731200357e-02 2.23208692653e-02 2.59885128162e-02 2.97324869747e-02 3.30014183614e-02 - 3.52270678377e-02 3.60143597485e-02 3.52270678377e-02 3.30014183614e-02 2.97324869747e-02 2.59885128162e-02 - 2.23208692653e-02 1.90731200357e-02 1.63414636372e-02 1.41011278955e-02 1.23490348915e-02 1.11222486043e-02 - 1.04242770349e-02 1.02031841081e-02 1.04242770349e-02 1.11222486043e-02 1.23490348915e-02 1.41011278955e-02 - 1.55749641354e-02 1.90731200357e-02 2.32096903098e-02 2.77206217649e-02 3.20717125323e-02 3.55533814714e-02 - 3.74959847737e-02 3.74959847737e-02 3.55533814714e-02 3.20717125323e-02 2.77206217649e-02 2.32096903098e-02 - 1.90731200357e-02 1.55749641354e-02 1.27676986907e-02 1.06212947837e-02 9.10176004354e-03 8.16346419979e-03 - 7.72923129933e-03 7.72923129933e-03 8.16346419979e-03 9.10176004354e-03 1.06212947837e-02 1.27676986907e-02 - 1.63414636372e-02 2.07351318935e-02 2.57560150434e-02 3.09544991478e-02 3.56148556138e-02 3.88905177841e-02 - 4.00749900281e-02 3.88905177841e-02 3.56148556138e-02 3.09544991478e-02 2.57560150434e-02 2.07351318935e-02 - 1.63414636372e-02 1.27676986907e-02 1.00387761995e-02 8.09940143038e-03 6.85502201162e-03 6.18550458410e-03 - 5.97825202186e-03 6.18550458410e-03 6.85502201162e-03 8.09940143038e-03 1.00387761995e-02 1.27676986907e-02 - 1.85732113773e-02 2.39287446557e-02 2.98227515673e-02 3.55984989831e-02 4.03211904768e-02 4.30069852462e-02 - 4.30069852462e-02 4.03211904768e-02 3.55984989831e-02 2.98227515673e-02 2.39287446557e-02 1.85732113773e-02 - 1.41011278955e-02 1.06212947837e-02 8.09940143038e-03 6.42332605072e-03 5.44128685160e-03 5.00022810356e-03 - 5.00022810356e-03 5.44128685160e-03 6.42332605072e-03 8.09940143038e-03 1.06212947837e-02 1.41011278955e-02 - 2.22527540382e-02 2.87349610469e-02 3.56148556138e-02 4.19230722756e-02 4.64352697346e-02 4.80826222932e-02 - 4.64352697346e-02 4.19230722756e-02 3.56148556138e-02 2.87349610469e-02 2.22527540382e-02 1.67247696023e-02 - 1.23490348915e-02 9.10176004354e-03 6.85502201162e-03 5.44128685160e-03 4.68737681852e-03 4.45430428341e-03 - 4.68737681852e-03 5.44128685160e-03 6.85502201162e-03 9.10176004354e-03 1.23490348915e-02 1.67247696023e-02 - 2.76901755249e-02 3.55533814714e-02 4.34476578087e-02 4.99875697106e-02 5.37274523246e-02 5.37274523246e-02 - 4.99875697106e-02 4.34476578087e-02 3.55533814714e-02 2.76901755249e-02 2.07965618706e-02 1.52731342228e-02 - 1.11222486043e-02 8.16346419979e-03 6.18550458410e-03 5.00022810356e-03 4.45430428341e-03 4.45430428341e-03 - 5.00022810356e-03 6.18550458410e-03 8.16346419979e-03 1.11222486043e-02 1.52731342228e-02 2.07965618706e-02 - 3.52270678377e-02 4.42738472380e-02 5.24827367699e-02 5.82637609313e-02 6.03520680545e-02 5.82637609313e-02 - 5.24827367699e-02 4.42738472380e-02 3.52270678377e-02 2.67482412330e-02 1.96971752186e-02 1.43139225142e-02 - 1.04242770349e-02 7.72923129933e-03 5.97825202186e-03 5.00022810356e-03 4.68737681852e-03 5.00022810356e-03 - 5.97825202186e-03 7.72923129933e-03 1.04242770349e-02 1.43139225142e-02 1.96971752186e-02 2.67482412330e-02 - 4.39611245421e-02 5.26323100532e-02 5.92671896593e-02 6.28462967289e-02 6.28462967289e-02 5.92671896593e-02 - 5.26323100532e-02 4.39611245421e-02 3.46378859300e-02 2.60389015581e-02 1.90473786081e-02 1.38569882819e-02 - 1.02031841081e-02 7.72923129933e-03 6.18550458410e-03 5.44128685160e-03 5.44128685160e-03 6.18550458410e-03 - 7.72923129933e-03 1.02031841081e-02 1.38569882819e-02 1.90473786081e-02 2.60389015581e-02 3.46378859300e-02 - 5.03920982744e-02 5.57017474383e-02 5.86480861577e-02 5.95705865689e-02 5.86480861577e-02 5.57017474383e-02 - 5.03920982744e-02 4.28256060156e-02 3.40527023771e-02 2.56667001275e-02 1.88392765173e-02 1.38569882819e-02 - 1.04242770349e-02 8.16346419979e-03 6.85502201162e-03 6.42332605072e-03 6.85502201162e-03 8.16346419979e-03 - 1.04242770349e-02 1.38569882819e-02 1.88392765173e-02 2.56667001275e-02 3.40527023771e-02 4.28256060156e-02 - 4.96150823327e-02 4.93152444260e-02 4.85514460951e-02 4.85514460951e-02 4.93152444260e-02 4.96150823327e-02 - 4.74726445019e-02 4.18365826600e-02 3.38082178713e-02 2.56667001275e-02 1.90473786081e-02 1.43139225142e-02 - 1.11222486043e-02 9.10176004354e-03 8.09940143038e-03 8.09940143038e-03 9.10176004354e-03 1.11222486043e-02 - 1.43139225142e-02 1.90473786081e-02 2.56667001275e-02 3.38082178713e-02 4.18365826600e-02 4.74726445019e-02 - 4.01965567029e-02 3.55340946595e-02 3.37527793613e-02 3.55340946595e-02 4.01965567029e-02 4.50421513586e-02 - 4.61549122721e-02 4.18365826600e-02 3.40527023771e-02 2.60389015581e-02 1.96971752186e-02 1.52731342228e-02 - 1.23490348915e-02 1.06212947837e-02 1.00387761995e-02 1.06212947837e-02 1.23490348915e-02 1.52731342228e-02 - 1.96971752186e-02 2.60389015581e-02 3.40527023771e-02 4.18365826600e-02 4.61549122721e-02 4.50421513586e-02 - 2.72980449809e-02 2.21086562145e-02 2.21086562145e-02 2.72980449809e-02 3.64979381134e-02 4.50421513586e-02 - 4.74726445019e-02 4.28256060156e-02 3.46378859300e-02 2.67482412330e-02 2.07965618706e-02 1.67247696023e-02 - 1.41011278955e-02 1.27676986907e-02 1.27676986907e-02 1.41011278955e-02 1.67247696023e-02 2.07965618706e-02 - 2.67482412330e-02 3.46378859300e-02 4.28256060156e-02 4.74726445019e-02 4.50421513586e-02 3.64979381134e-02 - 4.78306293438e-03 4.78306293438e-03 1.20127916963e-02 2.72980449809e-02 4.49731492966e-02 5.52660749302e-02 - 5.43728424707e-02 4.64972219765e-02 3.74959847737e-02 3.03107984462e-02 2.52525868671e-02 2.19036737434e-02 - 2.01594924058e-02 2.01594924058e-02 2.19036737434e-02 2.52525868671e-02 3.03107984462e-02 3.74959847737e-02 - 4.64972219765e-02 5.43728424707e-02 5.52660749302e-02 4.49731492966e-02 2.72980449809e-02 1.20127916963e-02 - 4.78306293438e-03 8.92458323932e-03 2.21086562145e-02 4.14983823737e-02 5.74779615393e-02 6.22085036765e-02 - 5.67802908281e-02 4.74317203940e-02 3.88905177841e-02 3.25477771477e-02 2.81262227272e-02 2.53584894772e-02 - 2.43885514961e-02 2.53584894772e-02 2.81262227272e-02 3.25477771477e-02 3.88905177841e-02 4.74317203940e-02 - 5.67802908281e-02 6.22085036765e-02 5.74779615393e-02 4.14983823737e-02 2.21086562145e-02 8.92458323932e-03 - 1.20127916963e-02 2.21086562145e-02 4.01124125220e-02 5.81927448576e-02 6.72242960853e-02 6.50724984809e-02 - 5.66502458441e-02 4.75580604876e-02 4.03211904768e-02 3.50622322206e-02 3.14092023629e-02 2.94353177762e-02 - 2.94353177762e-02 3.14092023629e-02 3.50622322206e-02 4.03211904768e-02 4.75580604876e-02 5.66502458441e-02 - 6.50724984809e-02 6.72242960853e-02 5.81927448576e-02 4.01124125220e-02 2.21086562145e-02 1.20127916963e-02 - 2.72980449809e-02 4.14983823737e-02 5.81927448576e-02 6.89438173057e-02 6.97193973966e-02 6.32743536159e-02 - 5.48042821337e-02 4.74868469137e-02 4.19230722756e-02 3.77513071960e-02 3.49472650912e-02 3.39275143754e-02 - 3.49472650912e-02 3.77513071960e-02 4.19230722756e-02 4.74868469137e-02 5.48042821337e-02 6.32743536159e-02 - 6.97193973966e-02 6.89438173057e-02 5.81927448576e-02 4.14983823737e-02 2.72980449809e-02 2.19326842858e-02 - 4.49731492966e-02 5.74779615393e-02 6.72242960853e-02 6.97193973966e-02 6.56955874028e-02 5.89796865623e-02 - 5.26365510379e-02 4.75580604876e-02 4.34476578087e-02 4.01880472274e-02 3.82730721923e-02 3.82730721923e-02 - 4.01880472274e-02 4.34476578087e-02 4.75580604876e-02 5.26365510379e-02 5.89796865623e-02 6.56955874028e-02 - 6.97193973966e-02 6.72242960853e-02 5.74779615393e-02 4.49731492966e-02 3.64979381134e-02 3.64979381134e-02 - 5.52660749302e-02 6.22085036765e-02 6.50724984809e-02 6.32743536159e-02 5.89796865623e-02 5.45216826827e-02 - 5.07662838782e-02 4.74317203940e-02 4.42738472380e-02 4.17752511891e-02 4.08013474350e-02 4.17752511891e-02 - 4.42738472380e-02 4.74317203940e-02 5.07662838782e-02 5.45216826827e-02 5.89796865623e-02 6.32743536159e-02 - 6.50724984809e-02 6.22085036765e-02 5.52660749302e-02 4.80686289281e-02 4.50421513586e-02 4.80686289281e-02 - 5.43728424707e-02 5.67802908281e-02 5.66502458441e-02 5.48042821337e-02 5.26365510379e-02 5.07662838782e-02 - 4.88652050826e-02 4.64972219765e-02 4.39611245421e-02 4.22555839350e-02 4.22555839350e-02 4.39611245421e-02 - 4.64972219765e-02 4.88652050826e-02 5.07662838782e-02 5.26365510379e-02 5.48042821337e-02 5.66502458441e-02 - 5.67802908281e-02 5.43728424707e-02 5.04488982123e-02 4.74726445019e-02 4.74726445019e-02 5.04488982123e-02 - 4.64972219765e-02 4.74317203940e-02 4.75580604876e-02 4.74868469137e-02 4.75580604876e-02 4.74317203940e-02 - 4.64972219765e-02 4.46898454449e-02 4.28256060156e-02 4.20317983678e-02 4.28256060156e-02 4.46898454449e-02 - 4.64972219765e-02 4.74317203940e-02 4.75580604876e-02 4.74868469137e-02 4.75580604876e-02 4.74317203940e-02 - 4.64972219765e-02 4.46898454449e-02 4.28256060156e-02 4.20317983678e-02 4.28256060156e-02 4.46898454449e-02 - 3.74959847737e-02 3.88905177841e-02 4.03211904768e-02 4.19230722756e-02 4.34476578087e-02 4.42738472380e-02 - 4.39611245421e-02 4.28256060156e-02 4.18365826600e-02 4.18365826600e-02 4.28256060156e-02 4.39611245421e-02 - 4.42738472380e-02 4.34476578087e-02 4.19230722756e-02 4.03211904768e-02 3.88905177841e-02 3.74959847737e-02 - 3.60143597485e-02 3.46378859300e-02 3.37861689888e-02 3.37861689888e-02 3.46378859300e-02 3.60143597485e-02 - 3.03107984462e-02 3.25477771477e-02 3.50622322206e-02 3.77513071960e-02 4.01880472274e-02 4.17752511891e-02 - 4.22555839350e-02 4.20317983678e-02 4.18365826600e-02 4.20317983678e-02 4.22555839350e-02 4.17752511891e-02 - 4.01880472274e-02 3.77513071960e-02 3.50622322206e-02 3.25477771477e-02 3.03107984462e-02 2.83448586606e-02 - 2.67482412330e-02 2.57020582842e-02 2.53397129065e-02 2.57020582842e-02 2.67482412330e-02 2.83448586606e-02 - 2.52525868671e-02 2.81262227272e-02 3.14092023629e-02 3.49472650912e-02 3.82730721923e-02 4.08013474350e-02 - 4.22555839350e-02 4.28256060156e-02 4.28256060156e-02 4.22555839350e-02 4.08013474350e-02 3.82730721923e-02 - 3.49472650912e-02 3.14092023629e-02 2.81262227272e-02 2.52525868671e-02 2.27884625410e-02 2.07965618706e-02 - 1.94109424364e-02 1.87149718700e-02 1.87149718700e-02 1.94109424364e-02 2.07965618706e-02 2.27884625410e-02 - 2.19036737434e-02 2.53584894772e-02 2.94353177762e-02 3.39275143754e-02 3.82730721923e-02 4.17752511891e-02 - 4.39611245421e-02 4.46898454449e-02 4.39611245421e-02 4.17752511891e-02 3.82730721923e-02 3.39275143754e-02 - 2.94353177762e-02 2.53584894772e-02 2.19036737434e-02 1.90357893277e-02 1.67247696023e-02 1.50351544907e-02 - 1.40306458540e-02 1.37029642554e-02 1.40306458540e-02 1.50351544907e-02 1.67247696023e-02 1.90357893277e-02 - 2.01594924058e-02 2.43885514961e-02 2.94353177762e-02 3.49472650912e-02 4.01880472274e-02 4.42738472380e-02 - 4.64972219765e-02 4.64972219765e-02 4.42738472380e-02 4.01880472274e-02 3.49472650912e-02 2.94353177762e-02 - 2.43885514961e-02 2.01594924058e-02 1.67618733320e-02 1.41011278955e-02 1.21379949101e-02 1.08721475193e-02 - 1.02670781727e-02 1.02670781727e-02 1.08721475193e-02 1.21379949101e-02 1.41011278955e-02 1.67618733320e-02 - 2.01594924058e-02 2.53584894772e-02 3.14092023629e-02 3.77513071960e-02 4.34476578087e-02 4.74317203940e-02 - 4.88652050826e-02 4.74317203940e-02 4.34476578087e-02 3.77513071960e-02 3.14092023629e-02 2.53584894772e-02 - 2.01594924058e-02 1.59804507535e-02 1.27676986907e-02 1.04248538239e-02 8.87112379151e-03 8.01216613758e-03 - 7.74242036277e-03 8.01216613758e-03 8.87112379151e-03 1.04248538239e-02 1.27676986907e-02 1.59804507535e-02 - 2.19036737434e-02 2.81262227272e-02 3.50622322206e-02 4.19230722756e-02 4.75580604876e-02 5.07662838782e-02 - 5.07662838782e-02 4.75580604876e-02 4.19230722756e-02 3.50622322206e-02 2.81262227272e-02 2.19036737434e-02 - 1.67618733320e-02 1.27676986907e-02 9.84753974693e-03 7.87995370977e-03 6.71488304626e-03 6.18966814651e-03 - 6.18966814651e-03 6.71488304626e-03 7.87995370977e-03 9.84753974693e-03 1.27676986907e-02 1.67618733320e-02 - 2.52525868671e-02 3.25477771477e-02 4.03211904768e-02 4.74868469137e-02 5.26365510379e-02 5.45216826827e-02 - 5.26365510379e-02 4.74868469137e-02 4.03211904768e-02 3.25477771477e-02 2.52525868671e-02 1.90357893277e-02 - 1.41011278955e-02 1.04248538239e-02 7.87995370977e-03 6.28902224701e-03 5.45146116297e-03 5.19548069751e-03 - 5.45146116297e-03 6.28902224701e-03 7.87995370977e-03 1.04248538239e-02 1.41011278955e-02 1.90357893277e-02 - 3.03107984462e-02 3.88905177841e-02 4.75580604876e-02 5.48042821337e-02 5.89796865623e-02 5.89796865623e-02 - 5.48042821337e-02 4.75580604876e-02 3.88905177841e-02 3.03107984462e-02 2.27884625410e-02 1.67247696023e-02 - 1.21379949101e-02 8.87112379151e-03 6.71488304626e-03 5.45146116297e-03 4.88266428153e-03 4.88266428153e-03 - 5.45146116297e-03 6.71488304626e-03 8.87112379151e-03 1.21379949101e-02 1.67247696023e-02 2.27884625410e-02 - 3.74959847737e-02 4.74317203940e-02 5.66502458441e-02 6.32743536159e-02 6.56955874028e-02 6.32743536159e-02 - 5.66502458441e-02 4.74317203940e-02 3.74959847737e-02 2.83448586606e-02 2.07965618706e-02 1.50351544907e-02 - 1.08721475193e-02 8.01216613758e-03 6.18966814651e-03 5.19548069751e-03 4.88266428153e-03 5.19548069751e-03 - 6.18966814651e-03 8.01216613758e-03 1.08721475193e-02 1.50351544907e-02 2.07965618706e-02 2.83448586606e-02 - 4.64972219765e-02 5.67802908281e-02 6.50724984809e-02 6.97193973966e-02 6.97193973966e-02 6.50724984809e-02 - 5.67802908281e-02 4.64972219765e-02 3.60143597485e-02 2.67482412330e-02 1.94109424364e-02 1.40306458540e-02 - 1.02670781727e-02 7.74242036277e-03 6.18966814651e-03 5.45146116297e-03 5.45146116297e-03 6.18966814651e-03 - 7.74242036277e-03 1.02670781727e-02 1.40306458540e-02 1.94109424364e-02 2.67482412330e-02 3.60143597485e-02 - 5.43728424707e-02 6.22085036765e-02 6.72242960853e-02 6.89438173057e-02 6.72242960853e-02 6.22085036765e-02 - 5.43728424707e-02 4.46898454449e-02 3.46378859300e-02 2.57020582842e-02 1.87149718700e-02 1.37029642554e-02 - 1.02670781727e-02 8.01216613758e-03 6.71488304626e-03 6.28902224701e-03 6.71488304626e-03 8.01216613758e-03 - 1.02670781727e-02 1.37029642554e-02 1.87149718700e-02 2.57020582842e-02 3.46378859300e-02 4.46898454449e-02 - 5.52660749302e-02 5.74779615393e-02 5.81927448576e-02 5.81927448576e-02 5.74779615393e-02 5.52660749302e-02 - 5.04488982123e-02 4.28256060156e-02 3.37861689888e-02 2.53397129065e-02 1.87149718700e-02 1.40306458540e-02 - 1.08721475193e-02 8.87112379151e-03 7.87995370977e-03 7.87995370977e-03 8.87112379151e-03 1.08721475193e-02 - 1.40306458540e-02 1.87149718700e-02 2.53397129065e-02 3.37861689888e-02 4.28256060156e-02 5.04488982123e-02 - 4.49731492966e-02 4.14983823737e-02 4.01124125220e-02 4.14983823737e-02 4.49731492966e-02 4.80686289281e-02 - 4.74726445019e-02 4.20317983678e-02 3.37861689888e-02 2.57020582842e-02 1.94109424364e-02 1.50351544907e-02 - 1.21379949101e-02 1.04248538239e-02 9.84753974693e-03 1.04248538239e-02 1.21379949101e-02 1.50351544907e-02 - 1.94109424364e-02 2.57020582842e-02 3.37861689888e-02 4.20317983678e-02 4.74726445019e-02 4.80686289281e-02 - 2.72980449809e-02 2.21086562145e-02 2.21086562145e-02 2.72980449809e-02 3.64979381134e-02 4.50421513586e-02 - 4.74726445019e-02 4.28256060156e-02 3.46378859300e-02 2.67482412330e-02 2.07965618706e-02 1.67247696023e-02 - 1.41011278955e-02 1.27676986907e-02 1.27676986907e-02 1.41011278955e-02 1.67247696023e-02 2.07965618706e-02 - 2.67482412330e-02 3.46378859300e-02 4.28256060156e-02 4.74726445019e-02 4.50421513586e-02 3.64979381134e-02 - 1.20127916963e-02 8.92458323932e-03 1.20127916963e-02 2.19326842858e-02 3.64979381134e-02 4.80686289281e-02 - 5.04488982123e-02 4.46898454449e-02 3.60143597485e-02 2.83448586606e-02 2.27884625410e-02 1.90357893277e-02 - 1.67618733320e-02 1.59804507535e-02 1.67618733320e-02 1.90357893277e-02 2.27884625410e-02 2.83448586606e-02 - 3.60143597485e-02 4.46898454449e-02 5.04488982123e-02 4.80686289281e-02 3.64979381134e-02 2.19326842858e-02 diff --git a/tests/01_PW/053_PW_OD/result.ref b/tests/01_PW/053_PW_OD/result.ref deleted file mode 100644 index e8d2437f128..00000000000 --- a/tests/01_PW/053_PW_OD/result.ref +++ /dev/null @@ -1,4 +0,0 @@ -etotref -206.9727320695149 -etotperatomref -103.4863660348 -totaldosref 12 -totaltimeref 0.19 diff --git a/tests/01_PW/055_PW_OW/INPUT b/tests/01_PW/055_PW_OW/INPUT deleted file mode 100644 index 62a57372e35..00000000000 --- a/tests/01_PW/055_PW_OW/INPUT +++ /dev/null @@ -1,18 +0,0 @@ -INPUT_PARAMETERS -basis_type pw -nspin 2 -calculation scf -init_wfc random -out_wfc_pw 1 - -nbands 6 -symmetry 1 - -pseudo_dir ../../PP_ORB -suffix autotest - -ecutwfc 20 -scf_thr 1e-9 -scf_nmax 100 - -pw_seed 1 diff --git a/tests/01_PW/055_PW_OW/README b/tests/01_PW/055_PW_OW/README deleted file mode 100644 index 75c6683eb49..00000000000 --- a/tests/01_PW/055_PW_OW/README +++ /dev/null @@ -1 +0,0 @@ -test the output of wfc_pw, and verify the maximum absolute coefficient diff --git a/tests/01_PW/055_PW_OW/STRU b/tests/01_PW/055_PW_OW/STRU deleted file mode 100644 index e2502200744..00000000000 --- a/tests/01_PW/055_PW_OW/STRU +++ /dev/null @@ -1,19 +0,0 @@ -ATOMIC_SPECIES -Si 14 Si_ONCV_PBE-1.0.upf upf201 - -LATTICE_CONSTANT -10.2 // add lattice constant - -LATTICE_VECTORS -0.0 0.5 0.5 -0.5 0.0 0.5 -0.5 0.5 0.0 - -ATOMIC_POSITIONS -Direct - -Si // Element type -1 // magnetism -2 -0.00 0.00 0.00 1 1 1 -0.25 0.25 0.25 1 1 1 diff --git a/tests/01_PW/055_PW_OW/result.ref b/tests/01_PW/055_PW_OW/result.ref deleted file mode 100644 index 8a380abb696..00000000000 --- a/tests/01_PW/055_PW_OW/result.ref +++ /dev/null @@ -1,12 +0,0 @@ -etotref -211.8032844431862 -etotperatomref -105.9016422216 -Max_wfc_1 0.9526 -Max_wfc_2 0.3328 -Max_wfc_3 0.3249 -Max_wfc_4 0.3627 -Max_wfc_5 0.3194 -Max_wfc_6 0.3987 -pointgroupref T_d -spacegroupref O_h -nksibzref 3 -totaltimeref 0.67 diff --git a/tests/01_PW/057_PW_SO_IW/STRU b/tests/01_PW/057_PW_SO_IW/STRU deleted file mode 100644 index 901aa2771d4..00000000000 --- a/tests/01_PW/057_PW_SO_IW/STRU +++ /dev/null @@ -1,23 +0,0 @@ -ATOMIC_SPECIES -As 1 As_ONCV_PBE_FR-1.1.upf upf201 -Ga 1 Ga_ONCV_PBE_FR-1.0.upf.txt upf201 - -LATTICE_CONSTANT -1 // add lattice constant, 10.58 ang - -LATTICE_VECTORS -5.34197 5.34197 0.0 -0.0 5.34197 5.34197 -5.34197 0.0 5.34197 -ATOMIC_POSITIONS -Direct //Cartesian or Direct coordinate. - -As -0 -1 -0.2500000 0.2500000 0.25000000 0 0 0 - -Ga //Element Label -0 -1 //number of atom -0.00000 0.00000 0.000000 0 0 0 diff --git a/tests/01_PW/063_PW_CR/README b/tests/01_PW/063_PW_CR/README deleted file mode 100644 index c8630b33c92..00000000000 --- a/tests/01_PW/063_PW_CR/README +++ /dev/null @@ -1 +0,0 @@ -test cell-relax for atom can't move diff --git a/tests/01_PW/063_PW_CR/result.ref b/tests/01_PW/063_PW_CR/result.ref deleted file mode 100644 index 1edd56c59ac..00000000000 --- a/tests/01_PW/063_PW_CR/result.ref +++ /dev/null @@ -1,5 +0,0 @@ -etotref -211.8247254821208969 -etotperatomref -105.9123627411 -totalforceref 0.000000 -totalstressref 342.188349 -totaltimeref 1.87 diff --git a/tests/01_PW/064_PW_CR_fix_a/INPUT b/tests/01_PW/064_PW_CR_fix_a/INPUT deleted file mode 100644 index e35038e00b3..00000000000 --- a/tests/01_PW/064_PW_CR_fix_a/INPUT +++ /dev/null @@ -1,29 +0,0 @@ -INPUT_PARAMETERS -#Parameters (General) -suffix autotest -pseudo_dir ../../PP_ORB -init_wfc random -pw_seed 1 - -nbands 8 -calculation cell-relax -symmetry 0 - -#Parameters (Accuracy) -ecutwfc 20 -scf_nmax 20 - -basis_type pw -relax_nmax 2 -relax_method cg 1 - -cal_stress 1 -stress_thr 1e-6 -cal_force 1 -force_thr_ev 1.0e-3 - -mixing_type broyden -mixing_beta 0.7 -mixing_gg0 1.5 - -fixed_axes a diff --git a/tests/01_PW/064_PW_CR_fix_a/README b/tests/01_PW/064_PW_CR_fix_a/README deleted file mode 100644 index da228def894..00000000000 --- a/tests/01_PW/064_PW_CR_fix_a/README +++ /dev/null @@ -1 +0,0 @@ -test cell-relax for the fixed a axe diff --git a/tests/01_PW/064_PW_CR_fix_a/result.ref b/tests/01_PW/064_PW_CR_fix_a/result.ref deleted file mode 100644 index 8db47b0fc9f..00000000000 --- a/tests/01_PW/064_PW_CR_fix_a/result.ref +++ /dev/null @@ -1,5 +0,0 @@ -etotref -211.8234658590234574 -etotperatomref -105.9117329295 -totalforceref 0.058886 -totalstressref 348.169515 -totaltimeref 1.63 diff --git a/tests/01_PW/065_PW_CR_fix_ab/INPUT b/tests/01_PW/065_PW_CR_fix_ab/INPUT deleted file mode 100644 index 6908aae7065..00000000000 --- a/tests/01_PW/065_PW_CR_fix_ab/INPUT +++ /dev/null @@ -1,29 +0,0 @@ -INPUT_PARAMETERS -#Parameters (General) -suffix autotest -pseudo_dir ../../PP_ORB -pw_seed 1 -init_wfc random - -nbands 8 -calculation cell-relax -symmetry 0 - -#Parameters (Accuracy) -ecutwfc 20 -scf_nmax 20 - -basis_type pw -relax_nmax 2 -relax_method cg 1 - -cal_stress 1 -stress_thr 1e-6 -cal_force 1 -force_thr_ev 1.0e-3 - -mixing_type broyden -mixing_beta 0.7 -mixing_gg0 1.5 - -fixed_axes ab diff --git a/tests/01_PW/065_PW_CR_fix_ab/README b/tests/01_PW/065_PW_CR_fix_ab/README deleted file mode 100644 index eb8408d0da9..00000000000 --- a/tests/01_PW/065_PW_CR_fix_ab/README +++ /dev/null @@ -1 +0,0 @@ -test cell-relax for the fixed ab axe diff --git a/tests/01_PW/065_PW_CR_fix_ab/result.ref b/tests/01_PW/065_PW_CR_fix_ab/result.ref deleted file mode 100644 index 9581f93b7a6..00000000000 --- a/tests/01_PW/065_PW_CR_fix_ab/result.ref +++ /dev/null @@ -1,5 +0,0 @@ -etotref -211.8218194786840058 -etotperatomref -105.9109097393 -totalforceref 0.083326 -totalstressref 352.065828 -totaltimeref 1.66 diff --git a/tests/01_PW/066_PW_CR_fix_abc/README b/tests/01_PW/066_PW_CR_fix_abc/README deleted file mode 100644 index a0f176b9a13..00000000000 --- a/tests/01_PW/066_PW_CR_fix_abc/README +++ /dev/null @@ -1 +0,0 @@ -test cell-relax for the fixed abc axe diff --git a/tests/01_PW/066_PW_CR_fix_abc/STRU b/tests/01_PW/066_PW_CR_fix_abc/STRU deleted file mode 100644 index be2226fd835..00000000000 --- a/tests/01_PW/066_PW_CR_fix_abc/STRU +++ /dev/null @@ -1,18 +0,0 @@ -ATOMIC_SPECIES -Si 1.000 Si_ONCV_PBE-1.0.upf #Element, Mass, Pseudopotential - -LATTICE_CONSTANT -10.21 #Lattice constant - -LATTICE_VECTORS -0.0 0.5 0.5 #Lattice vector 1 -0.5 0.0 0.5 #Lattice vector 2 -0.5 0.5 0.0 #Lattice vector 3 - -ATOMIC_POSITIONS -Cartesian #Cartesian(Unit is LATTICE_CONSTANT) -Si #Name of element -0.0 #Magnetic for this element. -2 #Number of atoms -0.00 0.00 0.00 0 0 0 #x,y,z, move_x, move_y, move_z -0.25 0.25 0.25 0 0 0 diff --git a/tests/01_PW/066_PW_CR_fix_abc/result.ref b/tests/01_PW/066_PW_CR_fix_abc/result.ref deleted file mode 100644 index aa7c43b6c7f..00000000000 --- a/tests/01_PW/066_PW_CR_fix_abc/result.ref +++ /dev/null @@ -1,5 +0,0 @@ -etotref -211.8179023689464486 -etotperatomref -105.9089511845 -totalforceref 0.000000 -totalstressref 350.991183 -totaltimeref 0.61 diff --git a/tests/01_PW/067_PW_CR_fix_ac/README b/tests/01_PW/067_PW_CR_fix_ac/README deleted file mode 100644 index c36595c4f11..00000000000 --- a/tests/01_PW/067_PW_CR_fix_ac/README +++ /dev/null @@ -1 +0,0 @@ -test cell-relax for the fixed ac axe diff --git a/tests/01_PW/067_PW_CR_fix_ac/STRU b/tests/01_PW/067_PW_CR_fix_ac/STRU deleted file mode 100644 index be2226fd835..00000000000 --- a/tests/01_PW/067_PW_CR_fix_ac/STRU +++ /dev/null @@ -1,18 +0,0 @@ -ATOMIC_SPECIES -Si 1.000 Si_ONCV_PBE-1.0.upf #Element, Mass, Pseudopotential - -LATTICE_CONSTANT -10.21 #Lattice constant - -LATTICE_VECTORS -0.0 0.5 0.5 #Lattice vector 1 -0.5 0.0 0.5 #Lattice vector 2 -0.5 0.5 0.0 #Lattice vector 3 - -ATOMIC_POSITIONS -Cartesian #Cartesian(Unit is LATTICE_CONSTANT) -Si #Name of element -0.0 #Magnetic for this element. -2 #Number of atoms -0.00 0.00 0.00 0 0 0 #x,y,z, move_x, move_y, move_z -0.25 0.25 0.25 0 0 0 diff --git a/tests/01_PW/067_PW_CR_fix_ac/result.ref b/tests/01_PW/067_PW_CR_fix_ac/result.ref deleted file mode 100644 index 21f1bc6b534..00000000000 --- a/tests/01_PW/067_PW_CR_fix_ac/result.ref +++ /dev/null @@ -1,5 +0,0 @@ -etotref -211.8218194443972493 -etotperatomref -105.9109097222 -totalforceref 0.083578 -totalstressref 352.046235 -totaltimeref 1.61 diff --git a/tests/01_PW/068_PW_CR_fix_b/INPUT b/tests/01_PW/068_PW_CR_fix_b/INPUT deleted file mode 100644 index e8aee35633c..00000000000 --- a/tests/01_PW/068_PW_CR_fix_b/INPUT +++ /dev/null @@ -1,29 +0,0 @@ -INPUT_PARAMETERS -#Parameters (General) -suffix autotest -pseudo_dir ../../PP_ORB -pw_seed 1 -init_wfc random - -nbands 8 -calculation cell-relax -symmetry 0 - -#Parameters (Accuracy) -ecutwfc 20 -scf_nmax 20 - -basis_type pw -relax_nmax 2 -relax_method cg 1 - -cal_stress 1 -stress_thr 1e-6 -cal_force 1 -force_thr_ev 1.0e-3 - -mixing_type broyden -mixing_beta 0.7 -mixing_gg0 1.5 - -fixed_axes b diff --git a/tests/01_PW/068_PW_CR_fix_b/README b/tests/01_PW/068_PW_CR_fix_b/README deleted file mode 100644 index 9eca62f718e..00000000000 --- a/tests/01_PW/068_PW_CR_fix_b/README +++ /dev/null @@ -1 +0,0 @@ -test cell-relax for the fixed b axe diff --git a/tests/01_PW/068_PW_CR_fix_b/STRU b/tests/01_PW/068_PW_CR_fix_b/STRU deleted file mode 100644 index be2226fd835..00000000000 --- a/tests/01_PW/068_PW_CR_fix_b/STRU +++ /dev/null @@ -1,18 +0,0 @@ -ATOMIC_SPECIES -Si 1.000 Si_ONCV_PBE-1.0.upf #Element, Mass, Pseudopotential - -LATTICE_CONSTANT -10.21 #Lattice constant - -LATTICE_VECTORS -0.0 0.5 0.5 #Lattice vector 1 -0.5 0.0 0.5 #Lattice vector 2 -0.5 0.5 0.0 #Lattice vector 3 - -ATOMIC_POSITIONS -Cartesian #Cartesian(Unit is LATTICE_CONSTANT) -Si #Name of element -0.0 #Magnetic for this element. -2 #Number of atoms -0.00 0.00 0.00 0 0 0 #x,y,z, move_x, move_y, move_z -0.25 0.25 0.25 0 0 0 diff --git a/tests/01_PW/068_PW_CR_fix_b/result.ref b/tests/01_PW/068_PW_CR_fix_b/result.ref deleted file mode 100644 index 7b8b8a23e33..00000000000 --- a/tests/01_PW/068_PW_CR_fix_b/result.ref +++ /dev/null @@ -1,5 +0,0 @@ -etotref -211.8234658682507643 -etotperatomref -105.9117329341 -totalforceref 0.059004 -totalstressref 348.162455 -totaltimeref 1.61 diff --git a/tests/01_PW/069_PW_CR_fix_bc/INPUT b/tests/01_PW/069_PW_CR_fix_bc/INPUT deleted file mode 100644 index 7af4dcfb864..00000000000 --- a/tests/01_PW/069_PW_CR_fix_bc/INPUT +++ /dev/null @@ -1,29 +0,0 @@ -INPUT_PARAMETERS -#Parameters (General) -suffix autotest -pseudo_dir ../../PP_ORB -pw_seed 1 -init_wfc random - -nbands 8 -calculation cell-relax -symmetry 0 - -#Parameters (Accuracy) -ecutwfc 20 -scf_nmax 20 - -basis_type pw -relax_nmax 2 -relax_method cg 1 - -cal_stress 1 -stress_thr 1e-6 -cal_force 1 -force_thr_ev 1.0e-3 - -mixing_type broyden -mixing_beta 0.7 -mixing_gg0 1.5 - -fixed_axes bc diff --git a/tests/01_PW/069_PW_CR_fix_bc/README b/tests/01_PW/069_PW_CR_fix_bc/README deleted file mode 100644 index 67ed8c1f037..00000000000 --- a/tests/01_PW/069_PW_CR_fix_bc/README +++ /dev/null @@ -1 +0,0 @@ -test cell-relax for the fixed bc axe diff --git a/tests/01_PW/069_PW_CR_fix_bc/STRU b/tests/01_PW/069_PW_CR_fix_bc/STRU deleted file mode 100644 index be2226fd835..00000000000 --- a/tests/01_PW/069_PW_CR_fix_bc/STRU +++ /dev/null @@ -1,18 +0,0 @@ -ATOMIC_SPECIES -Si 1.000 Si_ONCV_PBE-1.0.upf #Element, Mass, Pseudopotential - -LATTICE_CONSTANT -10.21 #Lattice constant - -LATTICE_VECTORS -0.0 0.5 0.5 #Lattice vector 1 -0.5 0.0 0.5 #Lattice vector 2 -0.5 0.5 0.0 #Lattice vector 3 - -ATOMIC_POSITIONS -Cartesian #Cartesian(Unit is LATTICE_CONSTANT) -Si #Name of element -0.0 #Magnetic for this element. -2 #Number of atoms -0.00 0.00 0.00 0 0 0 #x,y,z, move_x, move_y, move_z -0.25 0.25 0.25 0 0 0 diff --git a/tests/01_PW/069_PW_CR_fix_bc/result.ref b/tests/01_PW/069_PW_CR_fix_bc/result.ref deleted file mode 100644 index fc151d6ab2f..00000000000 --- a/tests/01_PW/069_PW_CR_fix_bc/result.ref +++ /dev/null @@ -1,5 +0,0 @@ -etotref -211.8218194597949946 -etotperatomref -105.9109097299 -totalforceref 0.083428 -totalstressref 352.069308 -totaltimeref 1.65 diff --git a/tests/01_PW/070_PW_CR_fix_c/INPUT b/tests/01_PW/070_PW_CR_fix_c/INPUT deleted file mode 100644 index e9f0e44fe8e..00000000000 --- a/tests/01_PW/070_PW_CR_fix_c/INPUT +++ /dev/null @@ -1,29 +0,0 @@ -INPUT_PARAMETERS -#Parameters (General) -suffix autotest -pseudo_dir ../../PP_ORB -pw_seed 1 -init_wfc random - -nbands 8 -calculation cell-relax -symmetry 0 - -#Parameters (Accuracy) -ecutwfc 20 -scf_nmax 20 - -basis_type pw -relax_nmax 2 -relax_method cg 1 - -cal_stress 1 -stress_thr 1e-6 -cal_force 1 -force_thr_ev 1.0e-3 - -mixing_type broyden -mixing_beta 0.7 -mixing_gg0 1.5 - -fixed_axes c diff --git a/tests/01_PW/070_PW_CR_fix_c/README b/tests/01_PW/070_PW_CR_fix_c/README deleted file mode 100644 index d6f569267a5..00000000000 --- a/tests/01_PW/070_PW_CR_fix_c/README +++ /dev/null @@ -1 +0,0 @@ -test cell-relax for the fixed c axe diff --git a/tests/01_PW/070_PW_CR_fix_c/STRU b/tests/01_PW/070_PW_CR_fix_c/STRU deleted file mode 100644 index be2226fd835..00000000000 --- a/tests/01_PW/070_PW_CR_fix_c/STRU +++ /dev/null @@ -1,18 +0,0 @@ -ATOMIC_SPECIES -Si 1.000 Si_ONCV_PBE-1.0.upf #Element, Mass, Pseudopotential - -LATTICE_CONSTANT -10.21 #Lattice constant - -LATTICE_VECTORS -0.0 0.5 0.5 #Lattice vector 1 -0.5 0.0 0.5 #Lattice vector 2 -0.5 0.5 0.0 #Lattice vector 3 - -ATOMIC_POSITIONS -Cartesian #Cartesian(Unit is LATTICE_CONSTANT) -Si #Name of element -0.0 #Magnetic for this element. -2 #Number of atoms -0.00 0.00 0.00 0 0 0 #x,y,z, move_x, move_y, move_z -0.25 0.25 0.25 0 0 0 diff --git a/tests/01_PW/070_PW_CR_fix_c/result.ref b/tests/01_PW/070_PW_CR_fix_c/result.ref deleted file mode 100644 index 53415565f55..00000000000 --- a/tests/01_PW/070_PW_CR_fix_c/result.ref +++ /dev/null @@ -1,5 +0,0 @@ -etotref -211.8234658446594381 -etotperatomref -105.9117329223 -totalforceref 0.058676 -totalstressref 348.189070 -totaltimeref 1.62 diff --git a/tests/01_PW/071_PW_CR_move/README b/tests/01_PW/071_PW_CR_move/README deleted file mode 100644 index 03b79d7a60d..00000000000 --- a/tests/01_PW/071_PW_CR_move/README +++ /dev/null @@ -1 +0,0 @@ -test cell-relax for atom can move diff --git a/tests/01_PW/071_PW_CR_move/result.ref b/tests/01_PW/071_PW_CR_move/result.ref deleted file mode 100644 index 17a84a44585..00000000000 --- a/tests/01_PW/071_PW_CR_move/result.ref +++ /dev/null @@ -1,5 +0,0 @@ -etotref -211.8247283707780468 -etotperatomref -105.9123641854 -totalforceref 0.000000 -totalstressref 342.186876 -totaltimeref 1.09 diff --git a/tests/01_PW/075_PW_CHG_BINARY/KPT b/tests/01_PW/075_PW_CHG_BINARY/KPT deleted file mode 100644 index f5f7f4ec34c..00000000000 --- a/tests/01_PW/075_PW_CHG_BINARY/KPT +++ /dev/null @@ -1,4 +0,0 @@ -K_POINTS -0 -Gamma -2 2 2 0 0 0 diff --git a/tests/01_PW/083_PW_sol_H2/INPUT b/tests/01_PW/083_PW_sol_H2/INPUT deleted file mode 100644 index 8c5be69eedf..00000000000 --- a/tests/01_PW/083_PW_sol_H2/INPUT +++ /dev/null @@ -1,21 +0,0 @@ -INPUT_PARAMETERS -#Parameters (General) -suffix autotest -pseudo_dir ../../PP_ORB -init_wfc random -pw_seed 1 - -nbands 2 -calculation scf -basis_type pw - -#Parameters (Accuracy) -ecutwfc 20 -scf_nmax 50 - -#Parameters (Solvation Model) -imp_sol 1 -eb_k 80 -tau 0.000010798 -sigma_k 0.6 -nc_k 0.00037 diff --git a/tests/01_PW/083_PW_sol_H2/README b/tests/01_PW/083_PW_sol_H2/README deleted file mode 100644 index 204638e64dc..00000000000 --- a/tests/01_PW/083_PW_sol_H2/README +++ /dev/null @@ -1,2 +0,0 @@ -test inplicit solvation model correction for H2 - diff --git a/tests/01_PW/083_PW_sol_H2/result.ref b/tests/01_PW/083_PW_sol_H2/result.ref deleted file mode 100644 index 1171796606c..00000000000 --- a/tests/01_PW/083_PW_sol_H2/result.ref +++ /dev/null @@ -1,5 +0,0 @@ -etotref -31.89428054070554 -etotperatomref -15.9471402704 -esolelref -0.8065636756 -esolcavref 0.0238598134 -totaltimeref 6.93 diff --git a/tests/01_PW/084_PW_sol_H2O/INPUT b/tests/01_PW/084_PW_sol_H2O/INPUT deleted file mode 100644 index 5ab63f1fda6..00000000000 --- a/tests/01_PW/084_PW_sol_H2O/INPUT +++ /dev/null @@ -1,22 +0,0 @@ -INPUT_PARAMETERS - -calculation scf -basis_type pw -init_wfc random -pw_seed 1 -cal_force 1 -nbands 20 -ecutwfc 20 -scf_nmax 50 - -#Parameters (Solvation Model) -imp_sol 1 -eb_k 80 -tau 0.000010798 -sigma_k 0.6 -nc_k 0.00037 - - -#Parameters (General) -suffix autotest -pseudo_dir ../../PP_ORB diff --git a/tests/01_PW/084_PW_sol_H2O/README b/tests/01_PW/084_PW_sol_H2O/README deleted file mode 100644 index 1d2532d0a68..00000000000 --- a/tests/01_PW/084_PW_sol_H2O/README +++ /dev/null @@ -1,2 +0,0 @@ -test inplicit solvation model correction for H2O - diff --git a/tests/01_PW/084_PW_sol_H2O/result.ref b/tests/01_PW/084_PW_sol_H2O/result.ref deleted file mode 100644 index 5cb7a594a7c..00000000000 --- a/tests/01_PW/084_PW_sol_H2O/result.ref +++ /dev/null @@ -1,6 +0,0 @@ -etotref -442.8368758671084038 -etotperatomref -147.6122919557 -totalforceref 16.736445 -esolelref -0.9245824197 -esolcavref 0.0325548859 -totaltimeref 4.46 diff --git a/tests/01_PW/095_PW_NPT/KPT b/tests/01_PW/095_PW_NPT/KPT deleted file mode 100644 index f5f7f4ec34c..00000000000 --- a/tests/01_PW/095_PW_NPT/KPT +++ /dev/null @@ -1,4 +0,0 @@ -K_POINTS -0 -Gamma -2 2 2 0 0 0 diff --git a/tests/01_PW/096_PW_NVT/KPT b/tests/01_PW/096_PW_NVT/KPT deleted file mode 100644 index f5f7f4ec34c..00000000000 --- a/tests/01_PW/096_PW_NVT/KPT +++ /dev/null @@ -1,4 +0,0 @@ -K_POINTS -0 -Gamma -2 2 2 0 0 0 diff --git a/tests/01_PW/097_PW_PBE0/INPUT b/tests/01_PW/097_PW_PBE0/INPUT deleted file mode 100644 index 590755ff1b4..00000000000 --- a/tests/01_PW/097_PW_PBE0/INPUT +++ /dev/null @@ -1,24 +0,0 @@ -INPUT_PARAMETERS -#Parameters (System) -suffix autotest -init_wfc random - -nbands 2 -stru_file STRU -kpoint_file KPT -pseudo_dir ../../PP_ORB -pw_seed 1 -calculation scf -#Parameters (PW) -ecutwfc 10.0 # Rydberg -#Parameters (electronic) -basis_type pw -scf_thr 1e-3 -smearing_method fixed - -cal_force 0 -cal_stress 1 - -dft_functional pbe0 -symmetry -1 -exx_gamma_extrapolation false diff --git a/tests/01_PW/097_PW_PBE0/README b/tests/01_PW/097_PW_PBE0/README deleted file mode 100644 index d42d5ef29af..00000000000 --- a/tests/01_PW/097_PW_PBE0/README +++ /dev/null @@ -1 +0,0 @@ -hybrid functionals for pw basis. note that this calculation is not physically correct, only for testing purpose. diff --git a/tests/01_PW/097_PW_PBE0/STRU b/tests/01_PW/097_PW_PBE0/STRU deleted file mode 100644 index c48d5b0745c..00000000000 --- a/tests/01_PW/097_PW_PBE0/STRU +++ /dev/null @@ -1,24 +0,0 @@ -ATOMIC_SPECIES -H 1 H_ONCV_PBE-1.0.upf #Pseudopotentials are downloaded from http://www.quantum-simulation.org/potentials/sg15_oncv/upf/ - -NUMERICAL_ORBITAL -orb_Si.dat - -LATTICE_CONSTANT -1.889766 - -LATTICE_VECTORS -0.0 2.708337 2.708337 -2.708337 0.0 2.708337 -2.708337 2.708337 0.0 - - - -ATOMIC_POSITIONS -Direct - -H -0 -2 -0.125 0.125 0.125 0 0 0 -0.875 0.875 0.875 0 0 0 diff --git a/tests/01_PW/097_PW_PBE0/result.ref b/tests/01_PW/097_PW_PBE0/result.ref deleted file mode 100644 index a5359e71240..00000000000 --- a/tests/01_PW/097_PW_PBE0/result.ref +++ /dev/null @@ -1,4 +0,0 @@ -etotref -30.5936375896797834 -etotperatomref -15.2968187948 -totalstressref 935.890087 -totaltimeref 12.91 diff --git a/tests/01_PW/097_PW_PBE0_AFM/STRU b/tests/01_PW/097_PW_PBE0_AFM/STRU index 41c2d006328..c343c44eba8 100644 --- a/tests/01_PW/097_PW_PBE0_AFM/STRU +++ b/tests/01_PW/097_PW_PBE0_AFM/STRU @@ -1,7 +1,6 @@ ATOMIC_SPECIES Fe 55.845 Fe_ONCV_PBE-1.0.upf - LATTICE_CONSTANT 1.889725989 diff --git a/tests/01_PW/097_PW_PBE0_FM/INPUT b/tests/01_PW/097_PW_PBE0_FM/INPUT deleted file mode 100644 index 5e1067de10d..00000000000 --- a/tests/01_PW/097_PW_PBE0_FM/INPUT +++ /dev/null @@ -1,31 +0,0 @@ -INPUT_PARAMETERS -#Parameters (System) -suffix autotest -calculation scf -nspin 2 -pseudo_dir ../../PP_ORB -orbital_dir ../../PP_ORB - -#Parameters (PW) -ecutwfc 30 -basis_type pw - -#Parameters (SCF) -scf_thr 1.0e-7 -scf_nmax 100 -smearing_method gaussian -smearing_sigma 0.015 -mixing_type pulay -mixing_ndim 8 -mixing_beta 0.2 - -#Parameters (Hybrid Functional) -dft_functional pbe0 - -#Parameters (Other) -symmetry -1 -init_wfc random -pw_seed 1 -device cpu -exx_thr_type energy -exx_ene_thr 5e-2 diff --git a/tests/01_PW/097_PW_PBE0_FM/KPT b/tests/01_PW/097_PW_PBE0_FM/KPT deleted file mode 100644 index e583e6e418a..00000000000 --- a/tests/01_PW/097_PW_PBE0_FM/KPT +++ /dev/null @@ -1,7 +0,0 @@ -K_POINTS -0 -Gamma -1 1 1 0 0 0 -### If you are running an energy calculation, please make sure your final energy is -### converged with respect to the k-point settings, unless you set a loose k-point -### mesh on purpose. diff --git a/tests/01_PW/097_PW_PBE0_FM/README b/tests/01_PW/097_PW_PBE0_FM/README deleted file mode 100644 index 99bca880b54..00000000000 --- a/tests/01_PW/097_PW_PBE0_FM/README +++ /dev/null @@ -1,3 +0,0 @@ -Fe Ferromagnetic with nspin=2 hybrid functional (PBE0), PW basis, init magnet: up + up - - diff --git a/tests/01_PW/097_PW_PBE0_FM/STRU b/tests/01_PW/097_PW_PBE0_FM/STRU deleted file mode 100644 index 34fa5b31d03..00000000000 --- a/tests/01_PW/097_PW_PBE0_FM/STRU +++ /dev/null @@ -1,19 +0,0 @@ -ATOMIC_SPECIES -Fe 55.845 Fe_ONCV_PBE-1.0.upf - - -LATTICE_CONSTANT -1.889725989 - -LATTICE_VECTORS - 2.8328152400000000 0.0000000000000000 0.0000000000000000 - 0.0000000000000000 2.8328152400000000 0.0000000000000000 - 0.0000000000000000 0.0000000000000000 2.8328152400000000 -ATOMIC_POSITIONS -Direct - -Fe -1.0 -2 -0.0 0.0 0.0 mag 5 -0.5 0.5 0.5 mag 5 diff --git a/tests/01_PW/097_PW_PBE0_FM/result.ref b/tests/01_PW/097_PW_PBE0_FM/result.ref deleted file mode 100644 index 5b7663ffcdf..00000000000 --- a/tests/01_PW/097_PW_PBE0_FM/result.ref +++ /dev/null @@ -1,3 +0,0 @@ -etotref -6382.460958449486 -etotperatomref -3191.2304792247 -totaltimeref 1.59 \ No newline at end of file diff --git a/tests/01_PW/209_PW_DFTHALF/KPT b/tests/01_PW/209_PW_DFTHALF/KPT deleted file mode 100644 index f5f7f4ec34c..00000000000 --- a/tests/01_PW/209_PW_DFTHALF/KPT +++ /dev/null @@ -1,4 +0,0 @@ -K_POINTS -0 -Gamma -2 2 2 0 0 0 diff --git a/tests/01_PW/212_PW_USPP_BPCG/KPT b/tests/01_PW/212_PW_USPP_BPCG/KPT deleted file mode 100644 index f5f7f4ec34c..00000000000 --- a/tests/01_PW/212_PW_USPP_BPCG/KPT +++ /dev/null @@ -1,4 +0,0 @@ -K_POINTS -0 -Gamma -2 2 2 0 0 0 diff --git a/tests/01_PW/214_PW_SCAN_NLCC/INPUT b/tests/01_PW/214_PW_SCAN_NLCC/INPUT new file mode 100644 index 00000000000..d3436594883 --- /dev/null +++ b/tests/01_PW/214_PW_SCAN_NLCC/INPUT @@ -0,0 +1,32 @@ +INPUT_PARAMETERS +#Parameters (1.General) +suffix autotest +calculation scf +kpar 2 +init_wfc random + +pseudo_dir ../../PP_ORB + +#Parameters (2.Iteration) +ecutwfc 50 +scf_thr 1e-9 + +dft_functional scan + +#Parameters (3.Basis) +basis_type pw + +#Parameters (4.Smearing) +smearing_method gauss +smearing_sigma 0.002 + +#Parameters (5.Mixing) +mixing_type broyden +mixing_beta 0.7 + +cal_force 1 +cal_stress 1 + +mixing_tau 1 + +pw_seed 1 diff --git a/tests/01_PW/035_PW_15_SO/KPT b/tests/01_PW/214_PW_SCAN_NLCC/KPT similarity index 100% rename from tests/01_PW/035_PW_15_SO/KPT rename to tests/01_PW/214_PW_SCAN_NLCC/KPT diff --git a/tests/01_PW/214_PW_SCAN_NLCC/README b/tests/01_PW/214_PW_SCAN_NLCC/README new file mode 100644 index 00000000000..2f7bdb30754 --- /dev/null +++ b/tests/01_PW/214_PW_SCAN_NLCC/README @@ -0,0 +1 @@ +scan functional test in PW base with an NLCC pseudopotential : system Si2, force and stress are tested diff --git a/tests/01_PW/214_PW_SCAN_NLCC/STRU b/tests/01_PW/214_PW_SCAN_NLCC/STRU new file mode 100644 index 00000000000..fdaa0601867 --- /dev/null +++ b/tests/01_PW/214_PW_SCAN_NLCC/STRU @@ -0,0 +1,19 @@ +ATOMIC_SPECIES +Si 28.0855 Si_dojo_nsoc.upf upf201 + +LATTICE_CONSTANT +10.2 // add lattice constant + +LATTICE_VECTORS +0.0 0.5 0.5 +0.5 0.0 0.5 +0.5 0.5 0.0 + +ATOMIC_POSITIONS +Direct + +Si // Element type +0.0 // magnetism +2 +0.00 0.00 0.00 1 1 1 +0.25 0.25 0.25 1 1 1 diff --git a/tests/01_PW/214_PW_SCAN_NLCC/result.ref b/tests/01_PW/214_PW_SCAN_NLCC/result.ref new file mode 100644 index 00000000000..029c1b2866e --- /dev/null +++ b/tests/01_PW/214_PW_SCAN_NLCC/result.ref @@ -0,0 +1,5 @@ +etotref -221.8397468418885410 +etotperatomref -110.9198734209 +totalforceref 0.000000 +totalstressref 864.422802 +totaltimeref 0.97 diff --git a/tests/01_PW/801_PW_LT_sc/INPUT b/tests/01_PW/801_PW_LT_sc/INPUT deleted file mode 100644 index 2f959670c41..00000000000 --- a/tests/01_PW/801_PW_LT_sc/INPUT +++ /dev/null @@ -1,20 +0,0 @@ -INPUT_PARAMETERS -#Parameters (System) -suffix autotest - -nbands 4 -stru_file STRU -kpoint_file KPT -pseudo_dir ../../PP_ORB -calculation scf -latname sc -#Parameters (PW) -ecutwfc 25.0 # Rydberg -#Parameters (electronic) -basis_type pw -scf_thr 1e-10 - -cal_force 1 -cal_stress 1 - -pw_seed 1 diff --git a/tests/01_PW/801_PW_LT_sc/README b/tests/01_PW/801_PW_LT_sc/README deleted file mode 100644 index d4f6c9624a3..00000000000 --- a/tests/01_PW/801_PW_LT_sc/README +++ /dev/null @@ -1 +0,0 @@ -sc (simple cubic) for pw base diff --git a/tests/01_PW/801_PW_LT_sc/result.ref b/tests/01_PW/801_PW_LT_sc/result.ref deleted file mode 100644 index f4b76acd0da..00000000000 --- a/tests/01_PW/801_PW_LT_sc/result.ref +++ /dev/null @@ -1,5 +0,0 @@ -etotref -30.3983392986648404 -etotperatomref -15.1991696493 -totalforceref 6.695464 -totalstressref 32.286157 -totaltimeref diff --git a/tests/01_PW/802_PW_LT_fcc/INPUT b/tests/01_PW/802_PW_LT_fcc/INPUT deleted file mode 100644 index 9ab93774437..00000000000 --- a/tests/01_PW/802_PW_LT_fcc/INPUT +++ /dev/null @@ -1,18 +0,0 @@ -INPUT_PARAMETERS -#Parameters (System) -suffix autotest - -nbands 4 -stru_file STRU -kpoint_file KPT -pseudo_dir ../../PP_ORB -calculation scf -latname fcc -#Parameters (PW) -ecutwfc 25.0 # Rydberg -#Parameters (electronic) -basis_type pw -scf_thr 1e-10 - -cal_force 1 -cal_stress 1 diff --git a/tests/01_PW/802_PW_LT_fcc/README b/tests/01_PW/802_PW_LT_fcc/README deleted file mode 100644 index 28c543ea502..00000000000 --- a/tests/01_PW/802_PW_LT_fcc/README +++ /dev/null @@ -1 +0,0 @@ -fcc (face centered cubic) for pw base diff --git a/tests/01_PW/802_PW_LT_fcc/STRU b/tests/01_PW/802_PW_LT_fcc/STRU deleted file mode 100644 index 7aa5b84baec..00000000000 --- a/tests/01_PW/802_PW_LT_fcc/STRU +++ /dev/null @@ -1,16 +0,0 @@ -#This is the atom file containing all the information -#about the lattice structure. - -ATOMIC_SPECIES -H 1.0008 H.pz-vbc.UPF - -LATTICE_CONSTANT -10.0 #Lattice constant - -ATOMIC_POSITIONS -Cartesian #Cartesian(Unit is LATTICE_CONSTANT) -H #Name of element -0.0 #Magnetic for this element. -2 #Number of atoms -0.00 0.00 -0.0661400 0 0 0 #x,y,z, move_x, move_y, move_z -0.00 0.00 0.0661400 0 0 0 #x,y,z, move_x, move_y, move_z diff --git a/tests/01_PW/802_PW_LT_fcc/result.ref b/tests/01_PW/802_PW_LT_fcc/result.ref deleted file mode 100644 index 2a2dd994663..00000000000 --- a/tests/01_PW/802_PW_LT_fcc/result.ref +++ /dev/null @@ -1,5 +0,0 @@ -etotref -31.6719740726179921 -etotperatomref -15.8359870363 -totalforceref 10.852600 -totalstressref 298.152661 -totaltimeref 0.38 diff --git a/tests/01_PW/803_PW_LT_bcc/INPUT b/tests/01_PW/803_PW_LT_bcc/INPUT deleted file mode 100644 index 82d0093cc32..00000000000 --- a/tests/01_PW/803_PW_LT_bcc/INPUT +++ /dev/null @@ -1,18 +0,0 @@ -INPUT_PARAMETERS -#Parameters (System) -suffix autotest - -nbands 4 -stru_file STRU -kpoint_file KPT -pseudo_dir ../../PP_ORB -calculation scf -latname bcc -#Parameters (PW) -ecutwfc 25.0 # Rydberg -#Parameters (electronic) -basis_type pw -scf_thr 1e-10 - -cal_force 1 -cal_stress 1 diff --git a/tests/01_PW/803_PW_LT_bcc/README b/tests/01_PW/803_PW_LT_bcc/README deleted file mode 100644 index f9f6e72e397..00000000000 --- a/tests/01_PW/803_PW_LT_bcc/README +++ /dev/null @@ -1 +0,0 @@ -bcc (body centered cubic) for pw base diff --git a/tests/01_PW/803_PW_LT_bcc/STRU b/tests/01_PW/803_PW_LT_bcc/STRU deleted file mode 100644 index 7aa5b84baec..00000000000 --- a/tests/01_PW/803_PW_LT_bcc/STRU +++ /dev/null @@ -1,16 +0,0 @@ -#This is the atom file containing all the information -#about the lattice structure. - -ATOMIC_SPECIES -H 1.0008 H.pz-vbc.UPF - -LATTICE_CONSTANT -10.0 #Lattice constant - -ATOMIC_POSITIONS -Cartesian #Cartesian(Unit is LATTICE_CONSTANT) -H #Name of element -0.0 #Magnetic for this element. -2 #Number of atoms -0.00 0.00 -0.0661400 0 0 0 #x,y,z, move_x, move_y, move_z -0.00 0.00 0.0661400 0 0 0 #x,y,z, move_x, move_y, move_z diff --git a/tests/01_PW/803_PW_LT_bcc/result.ref b/tests/01_PW/803_PW_LT_bcc/result.ref deleted file mode 100644 index 815ac3a9b19..00000000000 --- a/tests/01_PW/803_PW_LT_bcc/result.ref +++ /dev/null @@ -1,5 +0,0 @@ -etotref -30.6144125394732107 -etotperatomref -15.3072062697 -totalforceref 7.577494 -totalstressref 84.029412 -totaltimeref 0.30 diff --git a/tests/01_PW/804_PW_LT_hex/INPUT b/tests/01_PW/804_PW_LT_hex/INPUT deleted file mode 100644 index 6cac5f33258..00000000000 --- a/tests/01_PW/804_PW_LT_hex/INPUT +++ /dev/null @@ -1,19 +0,0 @@ -INPUT_PARAMETERS -#Parameters (System) -suffix autotest - -nbands 4 -stru_file STRU -kpoint_file KPT -pseudo_dir ../../PP_ORB -pw_seed 1 -calculation scf -latname hexagonal -#Parameters (PW) -ecutwfc 25.0 # Rydberg -#Parameters (electronic) -basis_type pw -scf_thr 1e-10 - -cal_force 1 -cal_stress 1 diff --git a/tests/01_PW/804_PW_LT_hex/README b/tests/01_PW/804_PW_LT_hex/README deleted file mode 100644 index 6e3fd9f5e73..00000000000 --- a/tests/01_PW/804_PW_LT_hex/README +++ /dev/null @@ -1 +0,0 @@ -hexagonal for pw base diff --git a/tests/01_PW/804_PW_LT_hex/STRU b/tests/01_PW/804_PW_LT_hex/STRU deleted file mode 100644 index 7773153dbed..00000000000 --- a/tests/01_PW/804_PW_LT_hex/STRU +++ /dev/null @@ -1,19 +0,0 @@ -#This is the atom file containing all the information -#about the lattice structure. - -ATOMIC_SPECIES -H 1.0008 H.pz-vbc.UPF - -LATTICE_CONSTANT -10.0 #Lattice constant - -LATTICE_PARAMETERS -2.0 - -ATOMIC_POSITIONS -Cartesian #Cartesian(Unit is LATTICE_CONSTANT) -H #Name of element -0.0 #Magnetic for this element. -2 #Number of atoms -0.00 0.00 -0.0661400 0 0 0 #x,y,z, move_x, move_y, move_z -0.00 0.00 0.0661400 0 0 0 #x,y,z, move_x, move_y, move_z diff --git a/tests/01_PW/804_PW_LT_hex/result.ref b/tests/01_PW/804_PW_LT_hex/result.ref deleted file mode 100644 index 1bc0312b179..00000000000 --- a/tests/01_PW/804_PW_LT_hex/result.ref +++ /dev/null @@ -1,5 +0,0 @@ -etotref -30.3807460309569350 -etotperatomref -15.1903730155 -totalforceref 6.625336 -totalstressref 20.693367 -totaltimeref 0.48 diff --git a/tests/01_PW/805_PW_LT_trigonal/INPUT b/tests/01_PW/805_PW_LT_trigonal/INPUT deleted file mode 100644 index 8d7d7a80152..00000000000 --- a/tests/01_PW/805_PW_LT_trigonal/INPUT +++ /dev/null @@ -1,19 +0,0 @@ -INPUT_PARAMETERS -#Parameters (System) -suffix autotest - -nbands 4 -stru_file STRU -kpoint_file KPT -pseudo_dir ../../PP_ORB -pw_seed 1 -calculation scf -latname trigonal -#Parameters (PW) -ecutwfc 25.0 # Rydberg -#Parameters (electronic) -basis_type pw -scf_thr 1e-10 - -cal_force 1 -cal_stress 1 diff --git a/tests/01_PW/805_PW_LT_trigonal/README b/tests/01_PW/805_PW_LT_trigonal/README deleted file mode 100644 index f1bf4ec0935..00000000000 --- a/tests/01_PW/805_PW_LT_trigonal/README +++ /dev/null @@ -1 +0,0 @@ -trigonal for pw base diff --git a/tests/01_PW/811_PW_LT_bco/INPUT b/tests/01_PW/811_PW_LT_bco/INPUT deleted file mode 100644 index c2d1e174da4..00000000000 --- a/tests/01_PW/811_PW_LT_bco/INPUT +++ /dev/null @@ -1,19 +0,0 @@ -INPUT_PARAMETERS -#Parameters (System) -suffix autotest - -nbands 4 -stru_file STRU -kpoint_file KPT -pseudo_dir ../../PP_ORB -pw_seed 1 -calculation scf -latname bco -#Parameters (PW) -ecutwfc 25.0 # Rydberg -#Parameters (electronic) -basis_type pw -scf_thr 1e-10 - -cal_force 1 -cal_stress 1 diff --git a/tests/01_PW/811_PW_LT_bco/README b/tests/01_PW/811_PW_LT_bco/README deleted file mode 100644 index b37557eeab7..00000000000 --- a/tests/01_PW/811_PW_LT_bco/README +++ /dev/null @@ -1 +0,0 @@ -bco (body centered orthorhombic) for pw base diff --git a/tests/01_PW/811_PW_LT_bco/STRU b/tests/01_PW/811_PW_LT_bco/STRU deleted file mode 100644 index 837ba8e4911..00000000000 --- a/tests/01_PW/811_PW_LT_bco/STRU +++ /dev/null @@ -1,19 +0,0 @@ -#This is the atom file containing all the information -#about the lattice structure. - -ATOMIC_SPECIES -H 1.0008 H.pz-vbc.UPF - -LATTICE_CONSTANT -10.0 #Lattice constant - -LATTICE_PARAMETERS -1.5 2.0 - -ATOMIC_POSITIONS -Cartesian #Cartesian(Unit is LATTICE_CONSTANT) -H #Name of element -0.0 #Magnetic for this element. -2 #Number of atoms -0.00 0.00 -0.0661400 0 0 0 #x,y,z, move_x, move_y, move_z -0.00 0.00 0.0661400 0 0 0 #x,y,z, move_x, move_y, move_z diff --git a/tests/01_PW/811_PW_LT_bco/result.ref b/tests/01_PW/811_PW_LT_bco/result.ref deleted file mode 100644 index 53c778cb0c6..00000000000 --- a/tests/01_PW/811_PW_LT_bco/result.ref +++ /dev/null @@ -1,5 +0,0 @@ -etotref -30.3664499595127850 -etotperatomref -15.1832249798 -totalforceref 6.543818 -totalstressref 21.509776 -totaltimeref 0.62 diff --git a/tests/01_PW/CASES_CPU.txt b/tests/01_PW/CASES_CPU.txt index b2892c3d44f..e145798111b 100644 --- a/tests/01_PW/CASES_CPU.txt +++ b/tests/01_PW/CASES_CPU.txt @@ -22,7 +22,6 @@ scf_out_chg_tau 018_PW_LPS8 019_PW_Coulomb 020_PW_kspace -021_PW_kspace3 022_PW_CG 023_PW_DA 024_PW_DS @@ -32,20 +31,12 @@ scf_out_chg_tau 028_PW_PINT_UKS 029_PW_15_CF_CS_S1_smallg 030_PW_15_CF_CS_S2_smallg -031_PW_15_CF_CS 032_PW_15_CF_CS_bspline -033_PW_CF_CS_S1_smallg -034_PW_CF_CS_S2_smallg -035_PW_15_SO +scf_spin4 036_PW_AF -037_PW_FM 038_PW_NC -039_PW_FD_smear -040_PW_FX_smear -041_PW_GA_smear -042_PW_M2_smear -043_PW_MP_smear -044_PW_MV_smear +038_PW_NC_GGA_GRAD2 +scf_Al100_kspace3 045_PW_BD_chgmix 046_PW_KK_chgmix 047_PW_PK_chgmix @@ -53,28 +44,21 @@ scf_out_chg_tau 049_PW_PU_chgmix 050_PW_CHG_mismatch 051_PW_OBOD_MemSaver -052_PW_OB -053_PW_OD -055_PW_OW +scf_out_band_dos_wf 056_PW_IW -057_PW_SO_IW +scf_in_wf_spin4 058_PW_RE_MB 059_PW_RE_MB_traj 060_PW_RE_MG 061_PW_RE_NEW 062_PW_RE_PINT_RKS -063_PW_CR -064_PW_CR_fix_a -065_PW_CR_fix_ab -066_PW_CR_fix_abc -067_PW_CR_fix_ac -068_PW_CR_fix_b -069_PW_CR_fix_bc -070_PW_CR_fix_c -071_PW_CR_move +relax_cell +relax_cell_fix_abc +relax_cell_fix_ac +relax_cell_move 073_PW_SY 074_PW_SY_LiRH -075_PW_CHG_BINARY +scf_in_chgdat 076_PW_elec_add 077_PW_elec_minus 078_PW_S2_elec_add @@ -82,8 +66,8 @@ scf_out_chg_tau 080_PW_dipole 081_PW_efield 082_PW_gatefield -083_PW_sol_H2 -084_PW_sol_H2O +scf_sol_H2 +scf_sol_H2O 085_PW_get_pchg 086_PW_get_wf 087_PW_get_pchg_kpar_bndpar @@ -94,11 +78,13 @@ scf_out_chg_tau 092_PW_CR_VDW3 093_PW_MSST 094_PW_MSST2 -095_PW_NPT -096_PW_NVT -097_PW_PBE0 +#095_PW_NPT # disabled: uses ref_cell_factor=1.05 which is currently blocked +md_nvt 097_PW_PBE0_AFM -097_PW_PBE0_FM +scf_pbe0_spin1 +scf_pbe0_spin2 +scf_pbe0_spin2_kpar +scf_pbe0_afm_kpar 099_PW_15_SO_avg 100_PW_DJ_SO 101_PW_W90 @@ -113,26 +99,27 @@ scf_out_chg_tau 207_PW_SCANL 207_PW_skip 208_PW_CG_float -209_PW_DFTHALF +scf_half 210_PW_kspace_shift 211_PW_BPCG_KB_OCP_CHG -212_PW_USPP_BPCG -213_PW_USPP_pchg_wfc -801_PW_LT_sc -802_PW_LT_fcc -803_PW_LT_bcc -804_PW_LT_hex -805_PW_LT_trigonal +scf_uspp_bpcg +scf_uspp_pchg_wfc +214_PW_SCAN_NLCC +scf_brav_sc_smear_fd +scf_brav_fcc_smear_ga +scf_brav_bcc_smear_m2 +scf_brav_hex_smear_mp +scf_brav_trig_smear_fx 806_PW_LT_st 807_PW_LT_bct 808_PW_LT_so 809_PW_LT_baco 810_PW_LT_fco -811_PW_LT_bco +scf_brav_bco_smear_mv 812_PW_LT_sm 813_PW_LT_bacm 814_PW_LT_triclinic -815_PW_DFTU_S2_Z -816_PW_DFTU_S4_XY +scf_dftu_spin2 +scf_dftu_spin4 scf_deltaspin2 scf_deltaspin4 diff --git a/tests/01_PW/CASES_GPU.txt b/tests/01_PW/CASES_GPU.txt index 80e9078f9c8..f4dc0654527 100644 --- a/tests/01_PW/CASES_GPU.txt +++ b/tests/01_PW/CASES_GPU.txt @@ -22,7 +22,6 @@ scf_out_elf 018_PW_LPS8 019_PW_Coulomb 020_PW_kspace -021_PW_kspace3 022_PW_CG 023_PW_DA 024_PW_DS @@ -32,20 +31,11 @@ scf_out_elf 028_PW_PINT_UKS 029_PW_15_CF_CS_S1_smallg 030_PW_15_CF_CS_S2_smallg -031_PW_15_CF_CS 032_PW_15_CF_CS_bspline -033_PW_CF_CS_S1_smallg -034_PW_CF_CS_S2_smallg -035_PW_15_SO +scf_spin4 036_PW_AF -037_PW_FM #038_PW_NC -039_PW_FD_smear -040_PW_FX_smear -041_PW_GA_smear -042_PW_M2_smear -043_PW_MP_smear -044_PW_MV_smear +scf_Al100_kspace3 045_PW_BD_chgmix 046_PW_KK_chgmix 047_PW_PK_chgmix @@ -53,28 +43,22 @@ scf_out_elf 049_PW_PU_chgmix 050_PW_CHG_mismatch 051_PW_OBOD_MemSaver -#052_PW_OB -053_PW_OD -#055_PW_OW +scf_out_band_dos_wf 056_PW_IW -057_PW_SO_IW +scf_in_wf_spin4 058_PW_RE_MB 059_PW_RE_MB_traj 060_PW_RE_MG #061_PW_RE_NEW #062_PW_RE_PINT_RKS #063_PW_CR -#064_PW_CR_fix_a -#065_PW_CR_fix_ab -066_PW_CR_fix_abc -#067_PW_CR_fix_ac -#068_PW_CR_fix_b -#069_PW_CR_fix_bc -#070_PW_CR_fix_c -#071_PW_CR_move +relax_cell +relax_cell_fix_abc +relax_cell_fix_ac +relax_cell_move 073_PW_SY 074_PW_SY_LiRH -075_PW_CHG_BINARY +scf_in_chgdat 076_PW_elec_add 077_PW_elec_minus 078_PW_S2_elec_add @@ -82,8 +66,8 @@ scf_out_elf 080_PW_dipole 081_PW_efield 082_PW_gatefield -083_PW_sol_H2 -084_PW_sol_H2O +scf_sol_H2 +scf_sol_H2O 085_PW_get_pchg #086_PW_get_wf #087_PW_get_pchg_kpar_bndpar @@ -94,11 +78,10 @@ scf_out_elf 092_PW_CR_VDW3 #093_PW_MSST #094_PW_MSST2 -095_PW_NPT -#096_PW_NVT -#097_PW_PBE0 -097_PW_PBE0_AFM -097_PW_PBE0_FM +#md_npt # disabled: uses ref_cell_factor=1.05 which is currently blocked +#md_nvt +scf_pbe0_spin1 +scf_pbe0_spin2 099_PW_15_SO_avg #100_PW_DJ_SO #101_PW_W90 @@ -112,23 +95,23 @@ scf_out_elf #206_PW_SCAN_S2 #207_PW_skip #208_PW_CG_float -209_PW_DFTHALF +scf_half 210_PW_kspace_shift -801_PW_LT_sc -802_PW_LT_fcc -#803_PW_LT_bcc -804_PW_LT_hex -805_PW_LT_trigonal +scf_brav_sc_smear_fd +scf_brav_fcc_smear_ga +#scf_brav_bcc_smear_m2 +scf_brav_hex_smear_mp +scf_brav_trig_smear_fx 806_PW_LT_st 807_PW_LT_bct 808_PW_LT_so 809_PW_LT_baco 810_PW_LT_fco -811_PW_LT_bco +scf_brav_bco_smear_mv 812_PW_LT_sm 813_PW_LT_bacm 814_PW_LT_triclinic -815_PW_DFTU_S2_Z -816_PW_DFTU_S4_XY +scf_dftu_spin2 +scf_dftu_spin4 scf_deltaspin2 scf_deltaspin4 diff --git a/tests/01_PW/CMakeLists.txt b/tests/01_PW/CMakeLists.txt index 63fe4962347..77ceb9a9857 100644 --- a/tests/01_PW/CMakeLists.txt +++ b/tests/01_PW/CMakeLists.txt @@ -10,7 +10,7 @@ if(ENABLE_ASAN) else() add_test( NAME 01_PW - COMMAND ${BASH} ../integrate/Autotest.sh -a ${ABACUS_BIN_PATH} -n 4 + COMMAND ${BASH} ../integrate/Autotest.sh -a ${ABACUS_BIN_PATH} -n 4 -j auto WORKING_DIRECTORY ${ABACUS_TEST_DIR}/01_PW ) endif() diff --git a/tests/01_PW/095_PW_NPT/INPUT b/tests/01_PW/md_npt/INPUT similarity index 100% rename from tests/01_PW/095_PW_NPT/INPUT rename to tests/01_PW/md_npt/INPUT diff --git a/tests/01_PW/040_PW_FX_smear/KPT b/tests/01_PW/md_npt/KPT similarity index 100% rename from tests/01_PW/040_PW_FX_smear/KPT rename to tests/01_PW/md_npt/KPT diff --git a/tests/01_PW/095_PW_NPT/README b/tests/01_PW/md_npt/README similarity index 100% rename from tests/01_PW/095_PW_NPT/README rename to tests/01_PW/md_npt/README diff --git a/tests/01_PW/095_PW_NPT/STRU b/tests/01_PW/md_npt/STRU similarity index 100% rename from tests/01_PW/095_PW_NPT/STRU rename to tests/01_PW/md_npt/STRU diff --git a/tests/01_PW/095_PW_NPT/result.ref b/tests/01_PW/md_npt/result.ref similarity index 100% rename from tests/01_PW/095_PW_NPT/result.ref rename to tests/01_PW/md_npt/result.ref diff --git a/tests/01_PW/096_PW_NVT/INPUT b/tests/01_PW/md_nvt/INPUT similarity index 100% rename from tests/01_PW/096_PW_NVT/INPUT rename to tests/01_PW/md_nvt/INPUT diff --git a/tests/01_PW/042_PW_M2_smear/KPT b/tests/01_PW/md_nvt/KPT similarity index 100% rename from tests/01_PW/042_PW_M2_smear/KPT rename to tests/01_PW/md_nvt/KPT diff --git a/tests/01_PW/096_PW_NVT/README b/tests/01_PW/md_nvt/README similarity index 100% rename from tests/01_PW/096_PW_NVT/README rename to tests/01_PW/md_nvt/README diff --git a/tests/01_PW/096_PW_NVT/Restart_md.txt b/tests/01_PW/md_nvt/Restart_md.txt similarity index 100% rename from tests/01_PW/096_PW_NVT/Restart_md.txt rename to tests/01_PW/md_nvt/Restart_md.txt diff --git a/tests/01_PW/096_PW_NVT/STRU_MD_2 b/tests/01_PW/md_nvt/STRU_MD_2 similarity index 100% rename from tests/01_PW/096_PW_NVT/STRU_MD_2 rename to tests/01_PW/md_nvt/STRU_MD_2 diff --git a/tests/01_PW/096_PW_NVT/result.ref b/tests/01_PW/md_nvt/result.ref similarity index 100% rename from tests/01_PW/096_PW_NVT/result.ref rename to tests/01_PW/md_nvt/result.ref diff --git a/tests/01_PW/063_PW_CR/INPUT b/tests/01_PW/relax_cell/INPUT similarity index 100% rename from tests/01_PW/063_PW_CR/INPUT rename to tests/01_PW/relax_cell/INPUT diff --git a/tests/01_PW/043_PW_MP_smear/KPT b/tests/01_PW/relax_cell/KPT similarity index 100% rename from tests/01_PW/043_PW_MP_smear/KPT rename to tests/01_PW/relax_cell/KPT diff --git a/tests/01_PW/relax_cell/README b/tests/01_PW/relax_cell/README new file mode 100644 index 00000000000..2a9148cfa64 --- /dev/null +++ b/tests/01_PW/relax_cell/README @@ -0,0 +1 @@ +test cell-relax baseline (no fixed_axes, atoms fixed) diff --git a/tests/01_PW/063_PW_CR/STRU b/tests/01_PW/relax_cell/STRU similarity index 100% rename from tests/01_PW/063_PW_CR/STRU rename to tests/01_PW/relax_cell/STRU diff --git a/tests/01_PW/relax_cell/result.ref b/tests/01_PW/relax_cell/result.ref new file mode 100644 index 00000000000..aafd94134c1 --- /dev/null +++ b/tests/01_PW/relax_cell/result.ref @@ -0,0 +1,5 @@ +etotref -211.8247254813327345 +etotperatomref -105.9123627407 +totalforceref 0.000000 +totalstressref 342.187740 +totaltimeref 0.87 diff --git a/tests/01_PW/066_PW_CR_fix_abc/INPUT b/tests/01_PW/relax_cell_fix_abc/INPUT similarity index 100% rename from tests/01_PW/066_PW_CR_fix_abc/INPUT rename to tests/01_PW/relax_cell_fix_abc/INPUT diff --git a/tests/01_PW/044_PW_MV_smear/KPT b/tests/01_PW/relax_cell_fix_abc/KPT similarity index 100% rename from tests/01_PW/044_PW_MV_smear/KPT rename to tests/01_PW/relax_cell_fix_abc/KPT diff --git a/tests/01_PW/relax_cell_fix_abc/README b/tests/01_PW/relax_cell_fix_abc/README new file mode 100644 index 00000000000..71a1cc97d81 --- /dev/null +++ b/tests/01_PW/relax_cell_fix_abc/README @@ -0,0 +1 @@ +test cell-relax with fixed_axes abc (all lattice fixed) diff --git a/tests/01_PW/064_PW_CR_fix_a/STRU b/tests/01_PW/relax_cell_fix_abc/STRU similarity index 100% rename from tests/01_PW/064_PW_CR_fix_a/STRU rename to tests/01_PW/relax_cell_fix_abc/STRU diff --git a/tests/01_PW/relax_cell_fix_abc/result.ref b/tests/01_PW/relax_cell_fix_abc/result.ref new file mode 100644 index 00000000000..c7a5c0c0603 --- /dev/null +++ b/tests/01_PW/relax_cell_fix_abc/result.ref @@ -0,0 +1,5 @@ +etotref -211.8179023689466476 +etotperatomref -105.9089511845 +totalforceref 0.000000 +totalstressref 350.991183 +totaltimeref 0.34 diff --git a/tests/01_PW/067_PW_CR_fix_ac/INPUT b/tests/01_PW/relax_cell_fix_ac/INPUT similarity index 100% rename from tests/01_PW/067_PW_CR_fix_ac/INPUT rename to tests/01_PW/relax_cell_fix_ac/INPUT diff --git a/tests/01_PW/055_PW_OW/KPT b/tests/01_PW/relax_cell_fix_ac/KPT similarity index 100% rename from tests/01_PW/055_PW_OW/KPT rename to tests/01_PW/relax_cell_fix_ac/KPT diff --git a/tests/01_PW/relax_cell_fix_ac/README b/tests/01_PW/relax_cell_fix_ac/README new file mode 100644 index 00000000000..ecfe949b7e7 --- /dev/null +++ b/tests/01_PW/relax_cell_fix_ac/README @@ -0,0 +1 @@ +test cell-relax with fixed_axes ac (partially fixed) diff --git a/tests/01_PW/065_PW_CR_fix_ab/STRU b/tests/01_PW/relax_cell_fix_ac/STRU similarity index 100% rename from tests/01_PW/065_PW_CR_fix_ab/STRU rename to tests/01_PW/relax_cell_fix_ac/STRU diff --git a/tests/01_PW/relax_cell_fix_ac/result.ref b/tests/01_PW/relax_cell_fix_ac/result.ref new file mode 100644 index 00000000000..d5c3dad2858 --- /dev/null +++ b/tests/01_PW/relax_cell_fix_ac/result.ref @@ -0,0 +1,5 @@ +etotref -211.8218194444026778 +etotperatomref -105.9109097222 +totalforceref 0.083578 +totalstressref 352.046235 +totaltimeref 0.83 diff --git a/tests/01_PW/071_PW_CR_move/INPUT b/tests/01_PW/relax_cell_move/INPUT similarity index 100% rename from tests/01_PW/071_PW_CR_move/INPUT rename to tests/01_PW/relax_cell_move/INPUT diff --git a/tests/01_PW/063_PW_CR/KPT b/tests/01_PW/relax_cell_move/KPT similarity index 100% rename from tests/01_PW/063_PW_CR/KPT rename to tests/01_PW/relax_cell_move/KPT diff --git a/tests/01_PW/relax_cell_move/README b/tests/01_PW/relax_cell_move/README new file mode 100644 index 00000000000..0da02b56ecc --- /dev/null +++ b/tests/01_PW/relax_cell_move/README @@ -0,0 +1 @@ +test cell-relax with movable atoms diff --git a/tests/01_PW/071_PW_CR_move/STRU b/tests/01_PW/relax_cell_move/STRU similarity index 100% rename from tests/01_PW/071_PW_CR_move/STRU rename to tests/01_PW/relax_cell_move/STRU diff --git a/tests/01_PW/relax_cell_move/result.ref b/tests/01_PW/relax_cell_move/result.ref new file mode 100644 index 00000000000..7ebc3481037 --- /dev/null +++ b/tests/01_PW/relax_cell_move/result.ref @@ -0,0 +1,5 @@ +etotref -211.8247283714259765 +etotperatomref -105.9123641857 +totalforceref 0.000000 +totalstressref 342.186675 +totaltimeref 0.56 diff --git a/tests/01_PW/scf_Al100_kspace3/INPUT b/tests/01_PW/scf_Al100_kspace3/INPUT new file mode 100644 index 00000000000..aa50061cdf2 --- /dev/null +++ b/tests/01_PW/scf_Al100_kspace3/INPUT @@ -0,0 +1,32 @@ +INPUT_PARAMETERS +#Parameters (1.General) +suffix autotest +calculation scf +init_wfc random + +nbands 35 +symmetry 1 +pseudo_dir ../../PP_ORB + +#Parameters (2.Iteration) +ecutwfc 20 +scf_thr 1e-8 +scf_nmax 100 + + +#Parameters (3.Basis) +basis_type pw + +#Parameters (4.Smearing) +smearing_method fd +smearing_sigma 0.002 + +#Parameters (5.Mixing) +mixing_type broyden +mixing_beta 0.6 +mixing_gg0 2 + +kspacing 0.5 1.0 1.0 + +cal_force 1 +cal_stress 1 diff --git a/tests/08_EXX/09_KP_HSE_comp/KPT b/tests/01_PW/scf_Al100_kspace3/KPT similarity index 100% rename from tests/08_EXX/09_KP_HSE_comp/KPT rename to tests/01_PW/scf_Al100_kspace3/KPT diff --git a/tests/01_PW/scf_Al100_kspace3/README b/tests/01_PW/scf_Al100_kspace3/README new file mode 100644 index 00000000000..142daa2b921 --- /dev/null +++ b/tests/01_PW/scf_Al100_kspace3/README @@ -0,0 +1,2 @@ +test smearing method: fd for (110) surface of Al +test kspacing with 3 input values diff --git a/tests/01_PW/039_PW_FD_smear/STRU b/tests/01_PW/scf_Al100_kspace3/STRU similarity index 100% rename from tests/01_PW/039_PW_FD_smear/STRU rename to tests/01_PW/scf_Al100_kspace3/STRU diff --git a/tests/01_PW/scf_Al100_kspace3/result.ref b/tests/01_PW/scf_Al100_kspace3/result.ref new file mode 100644 index 00000000000..501a28c8da0 --- /dev/null +++ b/tests/01_PW/scf_Al100_kspace3/result.ref @@ -0,0 +1,8 @@ +etotref -9136.7962400229062041 +etotperatomref -1827.3592480046 +totalforceref 2.897260 +totalstressref 16547.816920 +pointgroupref C_2v +spacegroupref D_2h +nksibzref 2 +totaltimeref 1.57 diff --git a/tests/01_PW/scf_Al100_kspace3/threshold b/tests/01_PW/scf_Al100_kspace3/threshold new file mode 100644 index 00000000000..83b85193ba2 --- /dev/null +++ b/tests/01_PW/scf_Al100_kspace3/threshold @@ -0,0 +1,2 @@ +force_threshold 0.001 +stress_threshold 0.1 diff --git a/tests/01_PW/scf_brav_bcc_smear_m2/INPUT b/tests/01_PW/scf_brav_bcc_smear_m2/INPUT new file mode 100644 index 00000000000..41bbad6988e --- /dev/null +++ b/tests/01_PW/scf_brav_bcc_smear_m2/INPUT @@ -0,0 +1,21 @@ +INPUT_PARAMETERS +#Parameters (System) +suffix autotest + +stru_file STRU +kpoint_file KPT +pseudo_dir ../../PP_ORB +calculation scf +latname bcc +#Parameters (PW) +ecutwfc 25.0 # Rydberg +#Parameters (electronic) +basis_type pw +scf_thr 1e-10 + +cal_force 1 +cal_stress 1 + +#Parameters (Smearing) +smearing_method mp2 +smearing_sigma 0.002 diff --git a/tests/01_PW/034_PW_CF_CS_S2_smallg/KPT b/tests/01_PW/scf_brav_bcc_smear_m2/KPT similarity index 100% rename from tests/01_PW/034_PW_CF_CS_S2_smallg/KPT rename to tests/01_PW/scf_brav_bcc_smear_m2/KPT diff --git a/tests/01_PW/scf_brav_bcc_smear_m2/README b/tests/01_PW/scf_brav_bcc_smear_m2/README new file mode 100644 index 00000000000..c93d3125c65 --- /dev/null +++ b/tests/01_PW/scf_brav_bcc_smear_m2/README @@ -0,0 +1,2 @@ +bcc (body centered cubic) for pw base +test smearing method: mp2 diff --git a/tests/01_PW/scf_brav_bcc_smear_m2/STRU b/tests/01_PW/scf_brav_bcc_smear_m2/STRU new file mode 100644 index 00000000000..a3de357c51f --- /dev/null +++ b/tests/01_PW/scf_brav_bcc_smear_m2/STRU @@ -0,0 +1,15 @@ +#This is the atom file containing all the information +#about the lattice structure. + +ATOMIC_SPECIES +Li 6.941 Li-s.PD04.PBE.UPF + +LATTICE_CONSTANT +10.0 #Lattice constant + +ATOMIC_POSITIONS +Cartesian #Cartesian(Unit is LATTICE_CONSTANT) +Li #Name of element +0.0 #Magnetic for this element. +1 #Number of atoms +0.00 0.00 0.00 0 0 0 #x,y,z, move_x, move_y, move_z diff --git a/tests/01_PW/scf_brav_bcc_smear_m2/result.ref b/tests/01_PW/scf_brav_bcc_smear_m2/result.ref new file mode 100644 index 00000000000..354184ebfeb --- /dev/null +++ b/tests/01_PW/scf_brav_bcc_smear_m2/result.ref @@ -0,0 +1,5 @@ +etotref -188.8664055532999839 +etotperatomref -188.8664055533 +totalforceref 0.000000 +totalstressref 1120.277193 +totaltimeref 0.37 diff --git a/tests/01_PW/scf_brav_bco_smear_mv/INPUT b/tests/01_PW/scf_brav_bco_smear_mv/INPUT new file mode 100644 index 00000000000..e66f53bae35 --- /dev/null +++ b/tests/01_PW/scf_brav_bco_smear_mv/INPUT @@ -0,0 +1,22 @@ +INPUT_PARAMETERS +#Parameters (System) +suffix autotest + +stru_file STRU +kpoint_file KPT +pseudo_dir ../../PP_ORB +pw_seed 1 +calculation scf +latname bco +#Parameters (PW) +ecutwfc 10.0 # Rydberg +#Parameters (electronic) +basis_type pw +scf_thr 1e-10 + +cal_force 1 +cal_stress 1 + +#Parameters (Smearing) +smearing_method mv +smearing_sigma 0.002 diff --git a/tests/01_PW/064_PW_CR_fix_a/KPT b/tests/01_PW/scf_brav_bco_smear_mv/KPT similarity index 100% rename from tests/01_PW/064_PW_CR_fix_a/KPT rename to tests/01_PW/scf_brav_bco_smear_mv/KPT diff --git a/tests/01_PW/scf_brav_bco_smear_mv/README b/tests/01_PW/scf_brav_bco_smear_mv/README new file mode 100644 index 00000000000..8407ac7ac54 --- /dev/null +++ b/tests/01_PW/scf_brav_bco_smear_mv/README @@ -0,0 +1,2 @@ +bco (body centered orthorhombic) for pw base +test smearing method: mv diff --git a/tests/01_PW/scf_brav_bco_smear_mv/STRU b/tests/01_PW/scf_brav_bco_smear_mv/STRU new file mode 100644 index 00000000000..b4893c039d2 --- /dev/null +++ b/tests/01_PW/scf_brav_bco_smear_mv/STRU @@ -0,0 +1,19 @@ +#This is the atom file containing all the information +#about the lattice structure. + +ATOMIC_SPECIES +Be 9.012182 Be_ONCV_PBE-1.0.upf + +LATTICE_CONSTANT +4.30 + +LATTICE_PARAMETERS +1.5 2.0 + +ATOMIC_POSITIONS +Direct +Be +0.0 +2 +0.00 0.00 0.00 1 1 1 +0.50 0.50 0.00 1 1 1 diff --git a/tests/01_PW/scf_brav_bco_smear_mv/result.ref b/tests/01_PW/scf_brav_bco_smear_mv/result.ref new file mode 100644 index 00000000000..34f26c42609 --- /dev/null +++ b/tests/01_PW/scf_brav_bco_smear_mv/result.ref @@ -0,0 +1,5 @@ +etotref -613.5130951758653737 +etotperatomref -306.7565475879 +totalforceref 0.000000 +totalstressref 25345.496277 +totaltimeref 0.38 diff --git a/tests/01_PW/scf_brav_fcc_smear_ga/INPUT b/tests/01_PW/scf_brav_fcc_smear_ga/INPUT new file mode 100644 index 00000000000..ee04cc96e4a --- /dev/null +++ b/tests/01_PW/scf_brav_fcc_smear_ga/INPUT @@ -0,0 +1,23 @@ +INPUT_PARAMETERS +#Parameters (System) +suffix autotest + +stru_file STRU +kpoint_file KPT +pseudo_dir ../../PP_ORB +calculation scf +latname fcc +nspin 2 +nbands 12 +#Parameters (PW) +ecutwfc 25.0 # Rydberg +#Parameters (electronic) +basis_type pw +scf_thr 1e-10 + +cal_force 1 +cal_stress 1 + +#Parameters (Smearing) +smearing_method gauss +smearing_sigma 0.002 diff --git a/tests/01_PW/065_PW_CR_fix_ab/KPT b/tests/01_PW/scf_brav_fcc_smear_ga/KPT similarity index 100% rename from tests/01_PW/065_PW_CR_fix_ab/KPT rename to tests/01_PW/scf_brav_fcc_smear_ga/KPT diff --git a/tests/01_PW/scf_brav_fcc_smear_ga/README b/tests/01_PW/scf_brav_fcc_smear_ga/README new file mode 100644 index 00000000000..76e16335d55 --- /dev/null +++ b/tests/01_PW/scf_brav_fcc_smear_ga/README @@ -0,0 +1,3 @@ +fcc (face centered cubic) for pw base +test smearing method: gauss +test fcc Fe with FM spin (nspin=2, init_mag=2) diff --git a/tests/01_PW/scf_brav_fcc_smear_ga/STRU b/tests/01_PW/scf_brav_fcc_smear_ga/STRU new file mode 100644 index 00000000000..57d033235d3 --- /dev/null +++ b/tests/01_PW/scf_brav_fcc_smear_ga/STRU @@ -0,0 +1,15 @@ +#This is the atom file containing all the information +#about the lattice structure. + +ATOMIC_SPECIES +Fe 55.845 Fe_ONCV_PBE-1.0.upf + +LATTICE_CONSTANT +3.6 + +ATOMIC_POSITIONS +Direct +Fe +2.0 +1 +0.00 0.00 0.00 1 1 1 diff --git a/tests/01_PW/scf_brav_fcc_smear_ga/result.ref b/tests/01_PW/scf_brav_fcc_smear_ga/result.ref new file mode 100644 index 00000000000..4e9e1da44de --- /dev/null +++ b/tests/01_PW/scf_brav_fcc_smear_ga/result.ref @@ -0,0 +1,5 @@ +etotref -2975.2444908250290609 +etotperatomref -2975.2444908250 +totalforceref 0.000000 +totalstressref 175430.680782 +totaltimeref 0.41 diff --git a/tests/01_PW/scf_brav_fcc_smear_ga/threshold b/tests/01_PW/scf_brav_fcc_smear_ga/threshold new file mode 100644 index 00000000000..e76ca35cc88 --- /dev/null +++ b/tests/01_PW/scf_brav_fcc_smear_ga/threshold @@ -0,0 +1 @@ +stress_threshold 1.0 diff --git a/tests/01_PW/scf_brav_hex_smear_mp/INPUT b/tests/01_PW/scf_brav_hex_smear_mp/INPUT new file mode 100644 index 00000000000..baa45cc85c8 --- /dev/null +++ b/tests/01_PW/scf_brav_hex_smear_mp/INPUT @@ -0,0 +1,22 @@ +INPUT_PARAMETERS +#Parameters (System) +suffix autotest + +stru_file STRU +kpoint_file KPT +pseudo_dir ../../PP_ORB +pw_seed 1 +calculation scf +latname hexagonal +#Parameters (PW) +ecutwfc 25.0 # Rydberg +#Parameters (electronic) +basis_type pw +scf_thr 1e-10 + +cal_force 1 +cal_stress 1 + +#Parameters (Smearing) +smearing_method mp +smearing_sigma 0.002 diff --git a/tests/01_PW/066_PW_CR_fix_abc/KPT b/tests/01_PW/scf_brav_hex_smear_mp/KPT similarity index 100% rename from tests/01_PW/066_PW_CR_fix_abc/KPT rename to tests/01_PW/scf_brav_hex_smear_mp/KPT diff --git a/tests/01_PW/scf_brav_hex_smear_mp/README b/tests/01_PW/scf_brav_hex_smear_mp/README new file mode 100644 index 00000000000..bfc7269c868 --- /dev/null +++ b/tests/01_PW/scf_brav_hex_smear_mp/README @@ -0,0 +1,2 @@ +hexagonal for pw base +test smearing method: mp diff --git a/tests/01_PW/scf_brav_hex_smear_mp/STRU b/tests/01_PW/scf_brav_hex_smear_mp/STRU new file mode 100644 index 00000000000..16862323e4e --- /dev/null +++ b/tests/01_PW/scf_brav_hex_smear_mp/STRU @@ -0,0 +1,19 @@ +#This is the atom file containing all the information +#about the lattice structure. + +ATOMIC_SPECIES +Mg 24.305 Mg_ONCV_PBE-1.0.upf + +LATTICE_CONSTANT +6.2 #Lattice constant + +LATTICE_PARAMETERS +1.633 + +ATOMIC_POSITIONS +Direct #Direct(Unit is LATTICE_CONSTANT) +Mg #Name of element +0.0 #Magnetic for this element. +2 #Number of atoms +0.00 0.00 0.00 0 0 0 +0.3333333333 0.6666666667 0.5 0 0 0 diff --git a/tests/01_PW/scf_brav_hex_smear_mp/result.ref b/tests/01_PW/scf_brav_hex_smear_mp/result.ref new file mode 100644 index 00000000000..525c33a2b09 --- /dev/null +++ b/tests/01_PW/scf_brav_hex_smear_mp/result.ref @@ -0,0 +1,5 @@ +etotref -2939.2333452796997335 +etotperatomref -1469.6166726398 +totalforceref 0.000000 +totalstressref 6958.443379 +totaltimeref 0.55 diff --git a/tests/01_PW/scf_brav_sc_smear_fd/INPUT b/tests/01_PW/scf_brav_sc_smear_fd/INPUT new file mode 100644 index 00000000000..fcd9adf09f4 --- /dev/null +++ b/tests/01_PW/scf_brav_sc_smear_fd/INPUT @@ -0,0 +1,20 @@ +INPUT_PARAMETERS +#Parameters (System) +suffix autotest + +nbands 4 +stru_file STRU +kpoint_file KPT +pseudo_dir ../../PP_ORB +calculation scf +latname sc +ecutwfc 25.0 # Rydberg +basis_type pw +scf_thr 1e-10 + +cal_force 1 +cal_stress 1 +pw_seed 1 + +smearing_method fd +smearing_sigma 0.002 diff --git a/tests/01_PW/039_PW_FD_smear/KPT b/tests/01_PW/scf_brav_sc_smear_fd/KPT similarity index 100% rename from tests/01_PW/039_PW_FD_smear/KPT rename to tests/01_PW/scf_brav_sc_smear_fd/KPT diff --git a/tests/01_PW/scf_brav_sc_smear_fd/README b/tests/01_PW/scf_brav_sc_smear_fd/README new file mode 100644 index 00000000000..391fd53d8e3 --- /dev/null +++ b/tests/01_PW/scf_brav_sc_smear_fd/README @@ -0,0 +1,2 @@ +Bravais Lattice: sc (simple cubic) for pw base +Fermi-Dirac (FD) smearing diff --git a/tests/01_PW/801_PW_LT_sc/STRU b/tests/01_PW/scf_brav_sc_smear_fd/STRU similarity index 100% rename from tests/01_PW/801_PW_LT_sc/STRU rename to tests/01_PW/scf_brav_sc_smear_fd/STRU diff --git a/tests/01_PW/scf_brav_sc_smear_fd/result.ref b/tests/01_PW/scf_brav_sc_smear_fd/result.ref new file mode 100644 index 00000000000..e40e805d2cc --- /dev/null +++ b/tests/01_PW/scf_brav_sc_smear_fd/result.ref @@ -0,0 +1,5 @@ +etotref -30.3983392990711145 +etotperatomref -15.1991696495 +totalforceref 6.695502 +totalstressref 32.286307 +totaltimeref 0.22 diff --git a/tests/01_PW/scf_brav_trig_smear_fx/INPUT b/tests/01_PW/scf_brav_trig_smear_fx/INPUT new file mode 100644 index 00000000000..103a2bb946e --- /dev/null +++ b/tests/01_PW/scf_brav_trig_smear_fx/INPUT @@ -0,0 +1,22 @@ +INPUT_PARAMETERS +#Parameters (System) +suffix autotest + +nbands 4 +stru_file STRU +kpoint_file KPT +pseudo_dir ../../PP_ORB +pw_seed 1 +calculation scf +latname trigonal +#Parameters (PW) +ecutwfc 25.0 # Rydberg +#Parameters (electronic) +basis_type pw +scf_thr 1e-10 + +smearing_method fixed +smearing_sigma 0.002 + +cal_force 1 +cal_stress 1 diff --git a/tests/01_PW/041_PW_GA_smear/KPT b/tests/01_PW/scf_brav_trig_smear_fx/KPT similarity index 100% rename from tests/01_PW/041_PW_GA_smear/KPT rename to tests/01_PW/scf_brav_trig_smear_fx/KPT diff --git a/tests/01_PW/scf_brav_trig_smear_fx/README b/tests/01_PW/scf_brav_trig_smear_fx/README new file mode 100644 index 00000000000..0ebf6df5412 --- /dev/null +++ b/tests/01_PW/scf_brav_trig_smear_fx/README @@ -0,0 +1,2 @@ +trigonal for pw base +test smearing method: fixed diff --git a/tests/01_PW/805_PW_LT_trigonal/STRU b/tests/01_PW/scf_brav_trig_smear_fx/STRU similarity index 100% rename from tests/01_PW/805_PW_LT_trigonal/STRU rename to tests/01_PW/scf_brav_trig_smear_fx/STRU diff --git a/tests/01_PW/805_PW_LT_trigonal/result.ref b/tests/01_PW/scf_brav_trig_smear_fx/result.ref similarity index 100% rename from tests/01_PW/805_PW_LT_trigonal/result.ref rename to tests/01_PW/scf_brav_trig_smear_fx/result.ref diff --git a/tests/01_PW/815_PW_DFTU_S2_Z/INPUT b/tests/01_PW/scf_dftu_spin2/INPUT similarity index 100% rename from tests/01_PW/815_PW_DFTU_S2_Z/INPUT rename to tests/01_PW/scf_dftu_spin2/INPUT diff --git a/tests/01_PW/815_PW_DFTU_S2_Z/KPT b/tests/01_PW/scf_dftu_spin2/KPT similarity index 100% rename from tests/01_PW/815_PW_DFTU_S2_Z/KPT rename to tests/01_PW/scf_dftu_spin2/KPT diff --git a/tests/01_PW/815_PW_DFTU_S2_Z/README b/tests/01_PW/scf_dftu_spin2/README similarity index 100% rename from tests/01_PW/815_PW_DFTU_S2_Z/README rename to tests/01_PW/scf_dftu_spin2/README diff --git a/tests/01_PW/815_PW_DFTU_S2_Z/STRU b/tests/01_PW/scf_dftu_spin2/STRU similarity index 100% rename from tests/01_PW/815_PW_DFTU_S2_Z/STRU rename to tests/01_PW/scf_dftu_spin2/STRU diff --git a/tests/01_PW/815_PW_DFTU_S2_Z/result.ref b/tests/01_PW/scf_dftu_spin2/result.ref similarity index 100% rename from tests/01_PW/815_PW_DFTU_S2_Z/result.ref rename to tests/01_PW/scf_dftu_spin2/result.ref diff --git a/tests/01_PW/816_PW_DFTU_S4_XY/INPUT b/tests/01_PW/scf_dftu_spin4/INPUT similarity index 100% rename from tests/01_PW/816_PW_DFTU_S4_XY/INPUT rename to tests/01_PW/scf_dftu_spin4/INPUT diff --git a/tests/01_PW/816_PW_DFTU_S4_XY/KPT b/tests/01_PW/scf_dftu_spin4/KPT similarity index 100% rename from tests/01_PW/816_PW_DFTU_S4_XY/KPT rename to tests/01_PW/scf_dftu_spin4/KPT diff --git a/tests/01_PW/816_PW_DFTU_S4_XY/README b/tests/01_PW/scf_dftu_spin4/README similarity index 100% rename from tests/01_PW/816_PW_DFTU_S4_XY/README rename to tests/01_PW/scf_dftu_spin4/README diff --git a/tests/01_PW/816_PW_DFTU_S4_XY/STRU b/tests/01_PW/scf_dftu_spin4/STRU similarity index 100% rename from tests/01_PW/816_PW_DFTU_S4_XY/STRU rename to tests/01_PW/scf_dftu_spin4/STRU diff --git a/tests/01_PW/816_PW_DFTU_S4_XY/result.ref b/tests/01_PW/scf_dftu_spin4/result.ref similarity index 100% rename from tests/01_PW/816_PW_DFTU_S4_XY/result.ref rename to tests/01_PW/scf_dftu_spin4/result.ref diff --git a/tests/01_PW/209_PW_DFTHALF/INPUT b/tests/01_PW/scf_half/INPUT similarity index 100% rename from tests/01_PW/209_PW_DFTHALF/INPUT rename to tests/01_PW/scf_half/INPUT diff --git a/tests/01_PW/067_PW_CR_fix_ac/KPT b/tests/01_PW/scf_half/KPT similarity index 100% rename from tests/01_PW/067_PW_CR_fix_ac/KPT rename to tests/01_PW/scf_half/KPT diff --git a/tests/01_PW/209_PW_DFTHALF/README b/tests/01_PW/scf_half/README similarity index 100% rename from tests/01_PW/209_PW_DFTHALF/README rename to tests/01_PW/scf_half/README diff --git a/tests/01_PW/209_PW_DFTHALF/STRU b/tests/01_PW/scf_half/STRU similarity index 100% rename from tests/01_PW/209_PW_DFTHALF/STRU rename to tests/01_PW/scf_half/STRU diff --git a/tests/01_PW/209_PW_DFTHALF/result.ref b/tests/01_PW/scf_half/result.ref similarity index 100% rename from tests/01_PW/209_PW_DFTHALF/result.ref rename to tests/01_PW/scf_half/result.ref diff --git a/tests/01_PW/075_PW_CHG_BINARY/INPUT b/tests/01_PW/scf_in_chgdat/INPUT similarity index 100% rename from tests/01_PW/075_PW_CHG_BINARY/INPUT rename to tests/01_PW/scf_in_chgdat/INPUT diff --git a/tests/01_PW/068_PW_CR_fix_b/KPT b/tests/01_PW/scf_in_chgdat/KPT similarity index 100% rename from tests/01_PW/068_PW_CR_fix_b/KPT rename to tests/01_PW/scf_in_chgdat/KPT diff --git a/tests/01_PW/075_PW_CHG_BINARY/README b/tests/01_PW/scf_in_chgdat/README similarity index 100% rename from tests/01_PW/075_PW_CHG_BINARY/README rename to tests/01_PW/scf_in_chgdat/README diff --git a/tests/01_PW/075_PW_CHG_BINARY/STRU b/tests/01_PW/scf_in_chgdat/STRU similarity index 100% rename from tests/01_PW/075_PW_CHG_BINARY/STRU rename to tests/01_PW/scf_in_chgdat/STRU diff --git a/tests/01_PW/075_PW_CHG_BINARY/autotest-CHARGE-DENSITY.restart b/tests/01_PW/scf_in_chgdat/autotest-CHARGE-DENSITY.restart similarity index 100% rename from tests/01_PW/075_PW_CHG_BINARY/autotest-CHARGE-DENSITY.restart rename to tests/01_PW/scf_in_chgdat/autotest-CHARGE-DENSITY.restart diff --git a/tests/01_PW/075_PW_CHG_BINARY/result.ref b/tests/01_PW/scf_in_chgdat/result.ref similarity index 100% rename from tests/01_PW/075_PW_CHG_BINARY/result.ref rename to tests/01_PW/scf_in_chgdat/result.ref diff --git a/tests/01_PW/057_PW_SO_IW/INPUT b/tests/01_PW/scf_in_wf_spin4/INPUT similarity index 100% rename from tests/01_PW/057_PW_SO_IW/INPUT rename to tests/01_PW/scf_in_wf_spin4/INPUT diff --git a/tests/01_PW/057_PW_SO_IW/KPT b/tests/01_PW/scf_in_wf_spin4/KPT similarity index 100% rename from tests/01_PW/057_PW_SO_IW/KPT rename to tests/01_PW/scf_in_wf_spin4/KPT diff --git a/tests/01_PW/057_PW_SO_IW/README b/tests/01_PW/scf_in_wf_spin4/README similarity index 100% rename from tests/01_PW/057_PW_SO_IW/README rename to tests/01_PW/scf_in_wf_spin4/README diff --git a/tests/01_PW/scf_in_wf_spin4/STRU b/tests/01_PW/scf_in_wf_spin4/STRU new file mode 100644 index 00000000000..71a2da3c729 --- /dev/null +++ b/tests/01_PW/scf_in_wf_spin4/STRU @@ -0,0 +1,23 @@ +ATOMIC_SPECIES +As 1 As_ONCV_PBE_FR-1.1.upf upf201 +Ga 1 Ga_ONCV_PBE_FR-1.0.upf.txt upf201 + +LATTICE_CONSTANT +1 // add lattice constant, 10.58 ang + +LATTICE_VECTORS +5.34197 5.34197 0.0 +0.0 5.34197 5.34197 +5.34197 0.0 5.34197 +ATOMIC_POSITIONS +Direct //Cartesian or Direct coordinate. + +As +0 +1 +0.2500000 0.2500000 0.25000000 0 0 0 mag 1.7320508 + +Ga //Element Label +0 +1 //number of atom +0.00000 0.00000 0.000000 0 0 0 mag 1.7320508 diff --git a/tests/01_PW/057_PW_SO_IW/eig.txt b/tests/01_PW/scf_in_wf_spin4/eig.txt similarity index 100% rename from tests/01_PW/057_PW_SO_IW/eig.txt rename to tests/01_PW/scf_in_wf_spin4/eig.txt diff --git a/tests/01_PW/057_PW_SO_IW/result.ref b/tests/01_PW/scf_in_wf_spin4/result.ref similarity index 100% rename from tests/01_PW/057_PW_SO_IW/result.ref rename to tests/01_PW/scf_in_wf_spin4/result.ref diff --git a/tests/01_PW/057_PW_SO_IW/wfk1s4_pw.dat b/tests/01_PW/scf_in_wf_spin4/wfk1s4_pw.dat similarity index 100% rename from tests/01_PW/057_PW_SO_IW/wfk1s4_pw.dat rename to tests/01_PW/scf_in_wf_spin4/wfk1s4_pw.dat diff --git a/tests/01_PW/scf_out_band_dos_wf/INPUT b/tests/01_PW/scf_out_band_dos_wf/INPUT new file mode 100644 index 00000000000..2248c2d5ea1 --- /dev/null +++ b/tests/01_PW/scf_out_band_dos_wf/INPUT @@ -0,0 +1,39 @@ +INPUT_PARAMETERS +#Parameters (1.General) +suffix autotest +calculation scf +cal_force 1 +cal_stress 1 + +nbands 6 +nspin 1 +symmetry 1 +pseudo_dir ../../PP_ORB + +#Parameters (2.Iteration) +ecutwfc 20 +scf_thr 1e-8 +scf_nmax 100 + +init_chg atomic +init_wfc random +out_band 1 +out_dos 1 +out_wfc_pw 1 +pw_diag_thr 0.00001 + +#Parameters (3.Basis) +basis_type pw + +#Parameters (4.Smearing) +smearing_method gauss +smearing_sigma 0.002 + +#Parameters (5.Mixing) +mixing_type plain +mixing_beta 0.7 + +dos_edelta_ev 0.01 +dos_sigma 0.05 + +pw_seed 1 diff --git a/tests/08_EXX/03_KP_S4_HSE/KPT b/tests/01_PW/scf_out_band_dos_wf/KPT similarity index 100% rename from tests/08_EXX/03_KP_S4_HSE/KPT rename to tests/01_PW/scf_out_band_dos_wf/KPT diff --git a/tests/01_PW/scf_out_band_dos_wf/README b/tests/01_PW/scf_out_band_dos_wf/README new file mode 100644 index 00000000000..0c25bd6ab79 --- /dev/null +++ b/tests/01_PW/scf_out_band_dos_wf/README @@ -0,0 +1 @@ +test the output of band, dos and wave function in pw basis diff --git a/tests/01_PW/scf_out_band_dos_wf/STRU b/tests/01_PW/scf_out_band_dos_wf/STRU new file mode 100644 index 00000000000..660cfceb4c5 --- /dev/null +++ b/tests/01_PW/scf_out_band_dos_wf/STRU @@ -0,0 +1,19 @@ +ATOMIC_SPECIES +Si 14 Si.pz-vbc.UPF + +LATTICE_CONSTANT +10.2 // add lattice constant + +LATTICE_VECTORS +0.0 0.5 0.5 +0.5 0.0 0.5 +0.5 0.5 0.0 + +ATOMIC_POSITIONS +Direct + +Si // Element type +0.0 // magnetism +2 +0.00 0.00 0.00 1 1 1 +0.26 0.25 0.25 1 1 1 diff --git a/tests/01_PW/scf_out_band_dos_wf/band.txt.ref b/tests/01_PW/scf_out_band_dos_wf/band.txt.ref new file mode 100644 index 00000000000..c9091976f05 --- /dev/null +++ b/tests/01_PW/scf_out_band_dos_wf/band.txt.ref @@ -0,0 +1,2 @@ + 1 0.00000000 -5.36802595 6.70427084 6.85743256 7.06566100 9.07047012 9.18721704 + 2 0.86602540 -2.86959821 -0.48990148 5.56030011 5.71333408 8.21423976 9.78665199 diff --git a/tests/01_PW/052_PW_OB/chg.cube b/tests/01_PW/scf_out_band_dos_wf/chg.cube similarity index 100% rename from tests/01_PW/052_PW_OB/chg.cube rename to tests/01_PW/scf_out_band_dos_wf/chg.cube diff --git a/tests/01_PW/scf_out_band_dos_wf/result.ref b/tests/01_PW/scf_out_band_dos_wf/result.ref new file mode 100644 index 00000000000..68646a858af --- /dev/null +++ b/tests/01_PW/scf_out_band_dos_wf/result.ref @@ -0,0 +1,16 @@ +etotref -205.6688343917421093 +etotperatomref -102.8344171959 +totalforceref 2.063434 +totalstressref 1184.321602 +totaldosref 12.000000 +CompareBand_pass 0 +Max_wfc_1 0.9183 +Max_wfc_2 0.5000 +Max_wfc_3 0.3992 +Max_wfc_4 0.4681 +Max_wfc_5 0.5574 +Max_wfc_6 0.4435 +pointgroupref C_1h +spacegroupref C_2h +nksibzref 2 +totaltimeref 0.37 diff --git a/tests/01_PW/scf_out_chg_tau/chg.cube.ref b/tests/01_PW/scf_out_chg_tau/chg.cube.ref index f1fb5a544f6..30f03c206c2 100644 --- a/tests/01_PW/scf_out_chg_tau/chg.cube.ref +++ b/tests/01_PW/scf_out_chg_tau/chg.cube.ref @@ -6,179 +6,179 @@ Ionic_Step 1 Cubefile created from ABACUS. Inner loop is z, followed by y and x 24 0.212500 0.212500 0.000000 14 4.000000 0.000000 0.000000 0.000000 14 4.000000 2.550000 2.550000 2.550000 - 6.306e-04 4.526e-03 1.770e-02 3.853e-02 5.728e-02 6.466e-02 + 6.291e-04 4.524e-03 1.770e-02 3.852e-02 5.728e-02 6.466e-02 6.053e-02 5.116e-02 4.192e-02 3.468e-02 2.936e-02 2.589e-02 2.464e-02 2.589e-02 2.936e-02 3.468e-02 4.192e-02 5.116e-02 - 6.053e-02 6.466e-02 5.728e-02 3.853e-02 1.770e-02 4.526e-03 - 4.526e-03 1.527e-02 3.604e-02 5.903e-02 7.277e-02 7.300e-02 - 6.459e-02 5.434e-02 4.565e-02 3.895e-02 3.401e-02 3.122e-02 - 3.122e-02 3.401e-02 3.895e-02 4.565e-02 5.434e-02 6.459e-02 - 7.300e-02 7.277e-02 5.903e-02 3.604e-02 1.527e-02 4.526e-03 - 1.770e-02 3.604e-02 5.962e-02 7.727e-02 8.198e-02 7.603e-02 - 6.613e-02 5.689e-02 4.939e-02 4.330e-02 3.892e-02 3.728e-02 - 3.892e-02 4.330e-02 4.939e-02 5.689e-02 6.613e-02 7.603e-02 - 8.198e-02 7.727e-02 5.962e-02 3.604e-02 1.770e-02 1.115e-02 - 3.853e-02 5.903e-02 7.727e-02 8.528e-02 8.268e-02 7.478e-02 - 6.639e-02 5.911e-02 5.260e-02 4.693e-02 4.341e-02 4.341e-02 - 4.693e-02 5.260e-02 5.911e-02 6.639e-02 7.478e-02 8.268e-02 - 8.528e-02 7.727e-02 5.903e-02 3.853e-02 2.556e-02 2.556e-02 - 5.728e-02 7.277e-02 8.198e-02 8.268e-02 7.798e-02 7.190e-02 - 6.613e-02 6.025e-02 5.398e-02 4.868e-02 4.655e-02 4.868e-02 - 5.398e-02 6.025e-02 6.613e-02 7.190e-02 7.798e-02 8.268e-02 - 8.198e-02 7.277e-02 5.728e-02 4.294e-02 3.719e-02 4.294e-02 - 6.466e-02 7.300e-02 7.603e-02 7.478e-02 7.190e-02 6.870e-02 - 6.459e-02 5.881e-02 5.237e-02 4.798e-02 4.798e-02 5.237e-02 - 5.881e-02 6.459e-02 6.870e-02 7.190e-02 7.478e-02 7.603e-02 - 7.300e-02 6.466e-02 5.397e-02 4.650e-02 4.650e-02 5.397e-02 - 6.053e-02 6.459e-02 6.613e-02 6.639e-02 6.613e-02 6.459e-02 + 6.053e-02 6.466e-02 5.728e-02 3.852e-02 1.770e-02 4.524e-03 + 4.524e-03 1.526e-02 3.604e-02 5.903e-02 7.277e-02 7.300e-02 + 6.460e-02 5.434e-02 4.565e-02 3.895e-02 3.401e-02 3.122e-02 + 3.122e-02 3.401e-02 3.895e-02 4.565e-02 5.434e-02 6.460e-02 + 7.300e-02 7.277e-02 5.903e-02 3.604e-02 1.526e-02 4.524e-03 + 1.770e-02 3.604e-02 5.962e-02 7.728e-02 8.198e-02 7.603e-02 + 6.613e-02 5.690e-02 4.939e-02 4.330e-02 3.892e-02 3.727e-02 + 3.892e-02 4.330e-02 4.939e-02 5.690e-02 6.613e-02 7.603e-02 + 8.198e-02 7.728e-02 5.962e-02 3.604e-02 1.770e-02 1.114e-02 + 3.852e-02 5.903e-02 7.728e-02 8.528e-02 8.269e-02 7.478e-02 + 6.639e-02 5.911e-02 5.260e-02 4.693e-02 4.340e-02 4.340e-02 + 4.693e-02 5.260e-02 5.911e-02 6.639e-02 7.478e-02 8.269e-02 + 8.528e-02 7.728e-02 5.903e-02 3.852e-02 2.555e-02 2.555e-02 + 5.728e-02 7.277e-02 8.198e-02 8.269e-02 7.798e-02 7.190e-02 + 6.613e-02 6.025e-02 5.398e-02 4.867e-02 4.655e-02 4.867e-02 + 5.398e-02 6.025e-02 6.613e-02 7.190e-02 7.798e-02 8.269e-02 + 8.198e-02 7.277e-02 5.728e-02 4.294e-02 3.718e-02 4.294e-02 + 6.466e-02 7.300e-02 7.603e-02 7.478e-02 7.190e-02 6.871e-02 + 6.460e-02 5.881e-02 5.237e-02 4.798e-02 4.798e-02 5.237e-02 + 5.881e-02 6.460e-02 6.871e-02 7.190e-02 7.478e-02 7.603e-02 + 7.300e-02 6.466e-02 5.396e-02 4.650e-02 4.650e-02 5.396e-02 + 6.053e-02 6.460e-02 6.613e-02 6.639e-02 6.613e-02 6.460e-02 6.053e-02 5.421e-02 4.821e-02 4.573e-02 4.821e-02 5.421e-02 - 6.053e-02 6.459e-02 6.613e-02 6.639e-02 6.613e-02 6.459e-02 + 6.053e-02 6.460e-02 6.613e-02 6.639e-02 6.613e-02 6.460e-02 6.053e-02 5.421e-02 4.821e-02 4.573e-02 4.821e-02 5.421e-02 - 5.116e-02 5.434e-02 5.689e-02 5.911e-02 6.025e-02 5.881e-02 + 5.116e-02 5.434e-02 5.690e-02 5.911e-02 6.025e-02 5.881e-02 5.421e-02 4.814e-02 4.385e-02 4.385e-02 4.814e-02 5.421e-02 - 5.881e-02 6.025e-02 5.911e-02 5.689e-02 5.434e-02 5.116e-02 - 4.707e-02 4.283e-02 4.006e-02 4.006e-02 4.283e-02 4.707e-02 + 5.881e-02 6.025e-02 5.911e-02 5.690e-02 5.434e-02 5.116e-02 + 4.707e-02 4.283e-02 4.005e-02 4.005e-02 4.283e-02 4.707e-02 4.192e-02 4.565e-02 4.939e-02 5.260e-02 5.398e-02 5.237e-02 - 4.821e-02 4.385e-02 4.201e-02 4.385e-02 4.821e-02 5.237e-02 + 4.821e-02 4.385e-02 4.200e-02 4.385e-02 4.821e-02 5.237e-02 5.398e-02 5.260e-02 4.939e-02 4.565e-02 4.192e-02 3.820e-02 3.469e-02 3.206e-02 3.107e-02 3.206e-02 3.469e-02 3.820e-02 - 3.468e-02 3.895e-02 4.330e-02 4.693e-02 4.868e-02 4.798e-02 - 4.573e-02 4.385e-02 4.385e-02 4.573e-02 4.798e-02 4.868e-02 + 3.468e-02 3.895e-02 4.330e-02 4.693e-02 4.867e-02 4.798e-02 + 4.573e-02 4.385e-02 4.385e-02 4.573e-02 4.798e-02 4.867e-02 4.693e-02 4.330e-02 3.895e-02 3.468e-02 3.074e-02 2.728e-02 2.460e-02 2.313e-02 2.313e-02 2.460e-02 2.728e-02 3.074e-02 - 2.936e-02 3.401e-02 3.892e-02 4.341e-02 4.655e-02 4.798e-02 - 4.821e-02 4.814e-02 4.821e-02 4.798e-02 4.655e-02 4.341e-02 + 2.936e-02 3.401e-02 3.892e-02 4.340e-02 4.655e-02 4.798e-02 + 4.821e-02 4.814e-02 4.821e-02 4.798e-02 4.655e-02 4.340e-02 3.892e-02 3.401e-02 2.936e-02 2.524e-02 2.171e-02 1.893e-02 1.714e-02 1.653e-02 1.714e-02 1.893e-02 2.171e-02 2.524e-02 - 2.589e-02 3.122e-02 3.728e-02 4.341e-02 4.868e-02 5.237e-02 - 5.421e-02 5.421e-02 5.237e-02 4.868e-02 4.341e-02 3.728e-02 + 2.589e-02 3.122e-02 3.727e-02 4.340e-02 4.867e-02 5.237e-02 + 5.421e-02 5.421e-02 5.237e-02 4.867e-02 4.340e-02 3.727e-02 3.122e-02 2.589e-02 2.145e-02 1.781e-02 1.495e-02 1.296e-02 1.196e-02 1.196e-02 1.296e-02 1.495e-02 1.781e-02 2.145e-02 2.464e-02 3.122e-02 3.892e-02 4.693e-02 5.398e-02 5.881e-02 6.053e-02 5.881e-02 5.398e-02 4.693e-02 3.892e-02 3.122e-02 2.464e-02 1.942e-02 1.538e-02 1.232e-02 1.016e-02 8.892e-03 - 8.479e-03 8.892e-03 1.016e-02 1.232e-02 1.538e-02 1.942e-02 - 2.589e-02 3.401e-02 4.330e-02 5.260e-02 6.025e-02 6.459e-02 - 6.459e-02 6.025e-02 5.260e-02 4.330e-02 3.401e-02 2.589e-02 + 8.478e-03 8.892e-03 1.016e-02 1.232e-02 1.538e-02 1.942e-02 + 2.589e-02 3.401e-02 4.330e-02 5.260e-02 6.025e-02 6.460e-02 + 6.460e-02 6.025e-02 5.260e-02 4.330e-02 3.401e-02 2.589e-02 1.942e-02 1.453e-02 1.096e-02 8.487e-03 6.959e-03 6.247e-03 6.247e-03 6.959e-03 8.487e-03 1.096e-02 1.453e-02 1.942e-02 - 2.936e-02 3.895e-02 4.939e-02 5.911e-02 6.613e-02 6.870e-02 + 2.936e-02 3.895e-02 4.939e-02 5.911e-02 6.613e-02 6.871e-02 6.613e-02 5.911e-02 4.939e-02 3.895e-02 2.936e-02 2.145e-02 1.538e-02 1.096e-02 7.921e-03 6.008e-03 4.997e-03 4.689e-03 4.997e-03 6.008e-03 7.921e-03 1.096e-02 1.538e-02 2.145e-02 - 3.468e-02 4.565e-02 5.689e-02 6.639e-02 7.190e-02 7.190e-02 - 6.639e-02 5.689e-02 4.565e-02 3.468e-02 2.524e-02 1.781e-02 + 3.468e-02 4.565e-02 5.690e-02 6.639e-02 7.190e-02 7.190e-02 + 6.639e-02 5.690e-02 4.565e-02 3.468e-02 2.524e-02 1.781e-02 1.232e-02 8.487e-03 6.008e-03 4.595e-03 3.979e-03 3.979e-03 4.595e-03 6.008e-03 8.487e-03 1.232e-02 1.781e-02 2.524e-02 4.192e-02 5.434e-02 6.613e-02 7.478e-02 7.798e-02 7.478e-02 6.613e-02 5.434e-02 4.192e-02 3.074e-02 2.171e-02 1.495e-02 1.016e-02 6.959e-03 4.997e-03 3.979e-03 3.671e-03 3.979e-03 4.997e-03 6.959e-03 1.016e-02 1.495e-02 2.171e-02 3.074e-02 - 5.116e-02 6.459e-02 7.603e-02 8.268e-02 8.268e-02 7.603e-02 - 6.459e-02 5.116e-02 3.820e-02 2.728e-02 1.893e-02 1.296e-02 + 5.116e-02 6.460e-02 7.603e-02 8.269e-02 8.269e-02 7.603e-02 + 6.460e-02 5.116e-02 3.820e-02 2.728e-02 1.893e-02 1.296e-02 8.892e-03 6.247e-03 4.689e-03 3.979e-03 3.979e-03 4.689e-03 6.247e-03 8.892e-03 1.296e-02 1.893e-02 2.728e-02 3.820e-02 6.053e-02 7.300e-02 8.198e-02 8.528e-02 8.198e-02 7.300e-02 6.053e-02 4.707e-02 3.469e-02 2.460e-02 1.714e-02 1.196e-02 - 8.479e-03 6.247e-03 4.997e-03 4.595e-03 4.997e-03 6.247e-03 - 8.479e-03 1.196e-02 1.714e-02 2.460e-02 3.469e-02 4.707e-02 - 6.466e-02 7.277e-02 7.727e-02 7.727e-02 7.277e-02 6.466e-02 + 8.478e-03 6.247e-03 4.997e-03 4.595e-03 4.997e-03 6.247e-03 + 8.478e-03 1.196e-02 1.714e-02 2.460e-02 3.469e-02 4.707e-02 + 6.466e-02 7.277e-02 7.728e-02 7.728e-02 7.277e-02 6.466e-02 5.421e-02 4.283e-02 3.206e-02 2.313e-02 1.653e-02 1.196e-02 8.892e-03 6.959e-03 6.008e-03 6.008e-03 6.959e-03 8.892e-03 1.196e-02 1.653e-02 2.313e-02 3.206e-02 4.283e-02 5.421e-02 - 5.728e-02 5.903e-02 5.962e-02 5.903e-02 5.728e-02 5.397e-02 - 4.821e-02 4.006e-02 3.107e-02 2.313e-02 1.714e-02 1.296e-02 + 5.728e-02 5.903e-02 5.962e-02 5.903e-02 5.728e-02 5.396e-02 + 4.821e-02 4.005e-02 3.107e-02 2.313e-02 1.714e-02 1.296e-02 1.016e-02 8.487e-03 7.921e-03 8.487e-03 1.016e-02 1.296e-02 - 1.714e-02 2.313e-02 3.107e-02 4.006e-02 4.821e-02 5.397e-02 - 3.853e-02 3.604e-02 3.604e-02 3.853e-02 4.294e-02 4.650e-02 - 4.573e-02 4.006e-02 3.206e-02 2.460e-02 1.893e-02 1.495e-02 + 1.714e-02 2.313e-02 3.107e-02 4.005e-02 4.821e-02 5.396e-02 + 3.852e-02 3.604e-02 3.604e-02 3.852e-02 4.294e-02 4.650e-02 + 4.573e-02 4.005e-02 3.206e-02 2.460e-02 1.893e-02 1.495e-02 1.232e-02 1.096e-02 1.096e-02 1.232e-02 1.495e-02 1.893e-02 - 2.460e-02 3.206e-02 4.006e-02 4.573e-02 4.650e-02 4.294e-02 - 1.770e-02 1.527e-02 1.770e-02 2.556e-02 3.719e-02 4.650e-02 + 2.460e-02 3.206e-02 4.005e-02 4.573e-02 4.650e-02 4.294e-02 + 1.770e-02 1.526e-02 1.770e-02 2.555e-02 3.718e-02 4.650e-02 4.821e-02 4.283e-02 3.469e-02 2.728e-02 2.171e-02 1.781e-02 1.538e-02 1.453e-02 1.538e-02 1.781e-02 2.171e-02 2.728e-02 - 3.469e-02 4.283e-02 4.821e-02 4.650e-02 3.719e-02 2.556e-02 - 4.526e-03 4.526e-03 1.115e-02 2.556e-02 4.294e-02 5.397e-02 + 3.469e-02 4.283e-02 4.821e-02 4.650e-02 3.718e-02 2.555e-02 + 4.524e-03 4.524e-03 1.114e-02 2.555e-02 4.294e-02 5.396e-02 5.421e-02 4.707e-02 3.820e-02 3.074e-02 2.524e-02 2.145e-02 1.942e-02 1.942e-02 2.145e-02 2.524e-02 3.074e-02 3.820e-02 - 4.707e-02 5.421e-02 5.397e-02 4.294e-02 2.556e-02 1.115e-02 - 4.526e-03 1.527e-02 3.604e-02 5.903e-02 7.277e-02 7.300e-02 - 6.459e-02 5.434e-02 4.565e-02 3.895e-02 3.401e-02 3.122e-02 - 3.122e-02 3.401e-02 3.895e-02 4.565e-02 5.434e-02 6.459e-02 - 7.300e-02 7.277e-02 5.903e-02 3.604e-02 1.527e-02 4.526e-03 - 1.527e-02 3.431e-02 5.950e-02 7.896e-02 8.473e-02 7.892e-02 - 6.870e-02 5.911e-02 5.133e-02 4.501e-02 4.045e-02 3.872e-02 - 4.045e-02 4.501e-02 5.133e-02 5.911e-02 6.870e-02 7.892e-02 - 8.473e-02 7.896e-02 5.950e-02 3.431e-02 1.527e-02 8.601e-03 - 3.604e-02 5.950e-02 8.117e-02 9.146e-02 8.941e-02 8.099e-02 - 7.190e-02 6.404e-02 5.700e-02 5.077e-02 4.685e-02 4.685e-02 - 5.077e-02 5.700e-02 6.404e-02 7.190e-02 8.099e-02 8.941e-02 - 9.146e-02 8.117e-02 5.950e-02 3.604e-02 2.162e-02 2.162e-02 - 5.903e-02 7.896e-02 9.146e-02 9.329e-02 8.818e-02 8.128e-02 - 7.478e-02 6.809e-02 6.074e-02 5.436e-02 5.176e-02 5.436e-02 - 6.074e-02 6.809e-02 7.478e-02 8.128e-02 8.818e-02 9.329e-02 - 9.146e-02 7.896e-02 5.903e-02 4.108e-02 3.399e-02 4.108e-02 - 7.277e-02 8.473e-02 8.941e-02 8.818e-02 8.475e-02 8.099e-02 - 7.603e-02 6.863e-02 6.009e-02 5.416e-02 5.416e-02 6.009e-02 - 6.863e-02 7.603e-02 8.099e-02 8.475e-02 8.818e-02 8.941e-02 - 8.473e-02 7.277e-02 5.775e-02 4.740e-02 4.740e-02 5.775e-02 - 7.300e-02 7.892e-02 8.099e-02 8.128e-02 8.099e-02 7.892e-02 + 4.707e-02 5.421e-02 5.396e-02 4.294e-02 2.555e-02 1.114e-02 + 4.524e-03 1.526e-02 3.604e-02 5.903e-02 7.277e-02 7.300e-02 + 6.460e-02 5.434e-02 4.565e-02 3.895e-02 3.401e-02 3.122e-02 + 3.122e-02 3.401e-02 3.895e-02 4.565e-02 5.434e-02 6.460e-02 + 7.300e-02 7.277e-02 5.903e-02 3.604e-02 1.526e-02 4.524e-03 + 1.526e-02 3.431e-02 5.950e-02 7.897e-02 8.474e-02 7.892e-02 + 6.871e-02 5.911e-02 5.133e-02 4.501e-02 4.045e-02 3.872e-02 + 4.045e-02 4.501e-02 5.133e-02 5.911e-02 6.871e-02 7.892e-02 + 8.474e-02 7.897e-02 5.950e-02 3.431e-02 1.526e-02 8.599e-03 + 3.604e-02 5.950e-02 8.118e-02 9.146e-02 8.941e-02 8.100e-02 + 7.190e-02 6.405e-02 5.700e-02 5.077e-02 4.685e-02 4.685e-02 + 5.077e-02 5.700e-02 6.405e-02 7.190e-02 8.100e-02 8.941e-02 + 9.146e-02 8.118e-02 5.950e-02 3.604e-02 2.162e-02 2.162e-02 + 5.903e-02 7.897e-02 9.146e-02 9.329e-02 8.818e-02 8.129e-02 + 7.478e-02 6.809e-02 6.074e-02 5.435e-02 5.176e-02 5.435e-02 + 6.074e-02 6.809e-02 7.478e-02 8.129e-02 8.818e-02 9.329e-02 + 9.146e-02 7.897e-02 5.903e-02 4.108e-02 3.399e-02 4.108e-02 + 7.277e-02 8.474e-02 8.941e-02 8.818e-02 8.476e-02 8.100e-02 + 7.603e-02 6.864e-02 6.009e-02 5.416e-02 5.416e-02 6.009e-02 + 6.864e-02 7.603e-02 8.100e-02 8.476e-02 8.818e-02 8.941e-02 + 8.474e-02 7.277e-02 5.775e-02 4.740e-02 4.740e-02 5.775e-02 + 7.300e-02 7.892e-02 8.100e-02 8.129e-02 8.100e-02 7.892e-02 7.300e-02 6.353e-02 5.442e-02 5.065e-02 5.442e-02 6.353e-02 - 7.300e-02 7.892e-02 8.099e-02 8.128e-02 8.099e-02 7.892e-02 + 7.300e-02 7.892e-02 8.100e-02 8.129e-02 8.100e-02 7.892e-02 7.300e-02 6.353e-02 5.442e-02 5.065e-02 5.442e-02 6.353e-02 - 6.459e-02 6.870e-02 7.190e-02 7.478e-02 7.603e-02 7.300e-02 - 6.466e-02 5.397e-02 4.650e-02 4.650e-02 5.397e-02 6.466e-02 - 7.300e-02 7.603e-02 7.478e-02 7.190e-02 6.870e-02 6.459e-02 + 6.460e-02 6.871e-02 7.190e-02 7.478e-02 7.603e-02 7.300e-02 + 6.466e-02 5.396e-02 4.650e-02 4.650e-02 5.396e-02 6.466e-02 + 7.300e-02 7.603e-02 7.478e-02 7.190e-02 6.871e-02 6.460e-02 5.881e-02 5.237e-02 4.798e-02 4.798e-02 5.237e-02 5.881e-02 - 5.434e-02 5.911e-02 6.404e-02 6.809e-02 6.863e-02 6.353e-02 - 5.397e-02 4.471e-02 4.091e-02 4.471e-02 5.397e-02 6.353e-02 - 6.863e-02 6.809e-02 6.404e-02 5.911e-02 5.434e-02 4.953e-02 + 5.434e-02 5.911e-02 6.405e-02 6.809e-02 6.864e-02 6.353e-02 + 5.396e-02 4.470e-02 4.090e-02 4.470e-02 5.396e-02 6.353e-02 + 6.864e-02 6.809e-02 6.405e-02 5.911e-02 5.434e-02 4.953e-02 4.466e-02 4.073e-02 3.920e-02 4.073e-02 4.466e-02 4.953e-02 4.565e-02 5.133e-02 5.700e-02 6.074e-02 6.009e-02 5.442e-02 - 4.650e-02 4.091e-02 4.091e-02 4.650e-02 5.442e-02 6.009e-02 + 4.650e-02 4.090e-02 4.090e-02 4.650e-02 5.442e-02 6.009e-02 6.074e-02 5.700e-02 5.133e-02 4.565e-02 4.056e-02 3.604e-02 3.238e-02 3.025e-02 3.025e-02 3.238e-02 3.604e-02 4.056e-02 - 3.895e-02 4.501e-02 5.077e-02 5.436e-02 5.416e-02 5.065e-02 - 4.650e-02 4.471e-02 4.650e-02 5.065e-02 5.416e-02 5.436e-02 - 5.077e-02 4.501e-02 3.895e-02 3.354e-02 2.897e-02 2.531e-02 - 2.286e-02 2.199e-02 2.286e-02 2.531e-02 2.897e-02 3.354e-02 + 3.895e-02 4.501e-02 5.077e-02 5.435e-02 5.416e-02 5.065e-02 + 4.650e-02 4.470e-02 4.650e-02 5.065e-02 5.416e-02 5.435e-02 + 5.077e-02 4.501e-02 3.895e-02 3.355e-02 2.897e-02 2.531e-02 + 2.286e-02 2.199e-02 2.286e-02 2.531e-02 2.897e-02 3.355e-02 3.401e-02 4.045e-02 4.685e-02 5.176e-02 5.416e-02 5.442e-02 - 5.397e-02 5.397e-02 5.442e-02 5.416e-02 5.176e-02 4.685e-02 - 4.045e-02 3.401e-02 2.838e-02 2.375e-02 2.007e-02 1.744e-02 - 1.606e-02 1.606e-02 1.744e-02 2.007e-02 2.375e-02 2.838e-02 - 3.122e-02 3.872e-02 4.685e-02 5.436e-02 6.009e-02 6.353e-02 - 6.466e-02 6.353e-02 6.009e-02 5.436e-02 4.685e-02 3.872e-02 - 3.122e-02 2.502e-02 2.015e-02 1.639e-02 1.366e-02 1.198e-02 - 1.141e-02 1.198e-02 1.366e-02 1.639e-02 2.015e-02 2.502e-02 - 3.122e-02 4.045e-02 5.077e-02 6.074e-02 6.863e-02 7.300e-02 - 7.300e-02 6.863e-02 6.074e-02 5.077e-02 4.045e-02 3.122e-02 + 5.396e-02 5.396e-02 5.442e-02 5.416e-02 5.176e-02 4.685e-02 + 4.045e-02 3.401e-02 2.839e-02 2.375e-02 2.007e-02 1.744e-02 + 1.606e-02 1.606e-02 1.744e-02 2.007e-02 2.375e-02 2.839e-02 + 3.122e-02 3.872e-02 4.685e-02 5.435e-02 6.009e-02 6.353e-02 + 6.466e-02 6.353e-02 6.009e-02 5.435e-02 4.685e-02 3.872e-02 + 3.122e-02 2.502e-02 2.015e-02 1.640e-02 1.366e-02 1.198e-02 + 1.141e-02 1.198e-02 1.366e-02 1.640e-02 2.015e-02 2.502e-02 + 3.122e-02 4.045e-02 5.077e-02 6.074e-02 6.864e-02 7.300e-02 + 7.300e-02 6.864e-02 6.074e-02 5.077e-02 4.045e-02 3.122e-02 2.381e-02 1.822e-02 1.411e-02 1.116e-02 9.254e-03 8.324e-03 8.324e-03 9.254e-03 1.116e-02 1.411e-02 1.822e-02 2.381e-02 3.401e-02 4.501e-02 5.700e-02 6.809e-02 7.603e-02 7.892e-02 7.603e-02 6.809e-02 5.700e-02 4.501e-02 3.401e-02 2.502e-02 1.822e-02 1.331e-02 9.883e-03 7.650e-03 6.418e-03 6.030e-03 6.418e-03 7.650e-03 9.883e-03 1.331e-02 1.822e-02 2.502e-02 - 3.895e-02 5.133e-02 6.404e-02 7.478e-02 8.099e-02 8.099e-02 - 7.478e-02 6.404e-02 5.133e-02 3.895e-02 2.838e-02 2.015e-02 + 3.895e-02 5.133e-02 6.405e-02 7.478e-02 8.100e-02 8.100e-02 + 7.478e-02 6.405e-02 5.133e-02 3.895e-02 2.839e-02 2.015e-02 1.411e-02 9.883e-03 7.115e-03 5.505e-03 4.789e-03 4.789e-03 - 5.505e-03 7.115e-03 9.883e-03 1.411e-02 2.015e-02 2.838e-02 - 4.565e-02 5.911e-02 7.190e-02 8.128e-02 8.475e-02 8.128e-02 - 7.190e-02 5.911e-02 4.565e-02 3.354e-02 2.375e-02 1.639e-02 + 5.505e-03 7.115e-03 9.883e-03 1.411e-02 2.015e-02 2.839e-02 + 4.565e-02 5.911e-02 7.190e-02 8.129e-02 8.476e-02 8.129e-02 + 7.190e-02 5.911e-02 4.565e-02 3.355e-02 2.375e-02 1.640e-02 1.116e-02 7.650e-03 5.505e-03 4.400e-03 4.068e-03 4.400e-03 - 5.505e-03 7.650e-03 1.116e-02 1.639e-02 2.375e-02 3.354e-02 - 5.434e-02 6.870e-02 8.099e-02 8.818e-02 8.818e-02 8.099e-02 - 6.870e-02 5.434e-02 4.056e-02 2.897e-02 2.007e-02 1.366e-02 + 5.505e-03 7.650e-03 1.116e-02 1.640e-02 2.375e-02 3.355e-02 + 5.434e-02 6.871e-02 8.100e-02 8.818e-02 8.818e-02 8.100e-02 + 6.871e-02 5.434e-02 4.056e-02 2.897e-02 2.007e-02 1.366e-02 9.254e-03 6.418e-03 4.789e-03 4.068e-03 4.068e-03 4.789e-03 6.418e-03 9.254e-03 1.366e-02 2.007e-02 2.897e-02 4.056e-02 - 6.459e-02 7.892e-02 8.941e-02 9.329e-02 8.941e-02 7.892e-02 - 6.459e-02 4.953e-02 3.604e-02 2.531e-02 1.744e-02 1.198e-02 + 6.460e-02 7.892e-02 8.941e-02 9.329e-02 8.941e-02 7.892e-02 + 6.460e-02 4.953e-02 3.604e-02 2.531e-02 1.744e-02 1.198e-02 8.324e-03 6.030e-03 4.789e-03 4.400e-03 4.789e-03 6.030e-03 8.324e-03 1.198e-02 1.744e-02 2.531e-02 3.604e-02 4.953e-02 - 7.300e-02 8.473e-02 9.146e-02 9.146e-02 8.473e-02 7.300e-02 + 7.300e-02 8.474e-02 9.146e-02 9.146e-02 8.474e-02 7.300e-02 5.881e-02 4.466e-02 3.238e-02 2.286e-02 1.606e-02 1.141e-02 8.324e-03 6.418e-03 5.505e-03 5.505e-03 6.418e-03 8.324e-03 1.141e-02 1.606e-02 2.286e-02 3.238e-02 4.466e-02 5.881e-02 - 7.277e-02 7.896e-02 8.117e-02 7.896e-02 7.277e-02 6.353e-02 + 7.277e-02 7.897e-02 8.118e-02 7.897e-02 7.277e-02 6.353e-02 5.237e-02 4.073e-02 3.025e-02 2.199e-02 1.606e-02 1.198e-02 9.254e-03 7.650e-03 7.115e-03 7.650e-03 9.254e-03 1.198e-02 1.606e-02 2.199e-02 3.025e-02 4.073e-02 5.237e-02 6.353e-02 @@ -187,454 +187,454 @@ Ionic_Step 1 Cubefile created from ABACUS. Inner loop is z, followed by y and x 1.116e-02 9.883e-03 9.883e-03 1.116e-02 1.366e-02 1.744e-02 2.286e-02 3.025e-02 3.920e-02 4.798e-02 5.442e-02 5.775e-02 3.604e-02 3.431e-02 3.604e-02 4.108e-02 4.740e-02 5.065e-02 - 4.798e-02 4.073e-02 3.238e-02 2.531e-02 2.007e-02 1.639e-02 - 1.411e-02 1.331e-02 1.411e-02 1.639e-02 2.007e-02 2.531e-02 + 4.798e-02 4.073e-02 3.238e-02 2.531e-02 2.007e-02 1.640e-02 + 1.411e-02 1.331e-02 1.411e-02 1.640e-02 2.007e-02 2.531e-02 3.238e-02 4.073e-02 4.798e-02 5.065e-02 4.740e-02 4.108e-02 - 1.527e-02 1.527e-02 2.162e-02 3.399e-02 4.740e-02 5.442e-02 + 1.526e-02 1.526e-02 2.162e-02 3.399e-02 4.740e-02 5.442e-02 5.237e-02 4.466e-02 3.604e-02 2.897e-02 2.375e-02 2.015e-02 1.822e-02 1.822e-02 2.015e-02 2.375e-02 2.897e-02 3.604e-02 4.466e-02 5.237e-02 5.442e-02 4.740e-02 3.399e-02 2.162e-02 - 4.526e-03 8.601e-03 2.162e-02 4.108e-02 5.775e-02 6.353e-02 - 5.881e-02 4.953e-02 4.056e-02 3.354e-02 2.838e-02 2.502e-02 - 2.381e-02 2.502e-02 2.838e-02 3.354e-02 4.056e-02 4.953e-02 - 5.881e-02 6.353e-02 5.775e-02 4.108e-02 2.162e-02 8.601e-03 - 1.770e-02 3.604e-02 5.962e-02 7.727e-02 8.198e-02 7.603e-02 - 6.613e-02 5.689e-02 4.939e-02 4.330e-02 3.892e-02 3.728e-02 - 3.892e-02 4.330e-02 4.939e-02 5.689e-02 6.613e-02 7.603e-02 - 8.198e-02 7.727e-02 5.962e-02 3.604e-02 1.770e-02 1.115e-02 - 3.604e-02 5.950e-02 8.117e-02 9.146e-02 8.941e-02 8.099e-02 - 7.190e-02 6.404e-02 5.700e-02 5.077e-02 4.685e-02 4.685e-02 - 5.077e-02 5.700e-02 6.404e-02 7.190e-02 8.099e-02 8.941e-02 - 9.146e-02 8.117e-02 5.950e-02 3.604e-02 2.162e-02 2.162e-02 - 5.962e-02 8.117e-02 9.491e-02 9.719e-02 9.194e-02 8.475e-02 + 4.524e-03 8.599e-03 2.162e-02 4.108e-02 5.775e-02 6.353e-02 + 5.881e-02 4.953e-02 4.056e-02 3.355e-02 2.839e-02 2.502e-02 + 2.381e-02 2.502e-02 2.839e-02 3.355e-02 4.056e-02 4.953e-02 + 5.881e-02 6.353e-02 5.775e-02 4.108e-02 2.162e-02 8.599e-03 + 1.770e-02 3.604e-02 5.962e-02 7.728e-02 8.198e-02 7.603e-02 + 6.613e-02 5.690e-02 4.939e-02 4.330e-02 3.892e-02 3.727e-02 + 3.892e-02 4.330e-02 4.939e-02 5.690e-02 6.613e-02 7.603e-02 + 8.198e-02 7.728e-02 5.962e-02 3.604e-02 1.770e-02 1.114e-02 + 3.604e-02 5.950e-02 8.118e-02 9.146e-02 8.941e-02 8.100e-02 + 7.190e-02 6.405e-02 5.700e-02 5.077e-02 4.685e-02 4.685e-02 + 5.077e-02 5.700e-02 6.405e-02 7.190e-02 8.100e-02 8.941e-02 + 9.146e-02 8.118e-02 5.950e-02 3.604e-02 2.162e-02 2.162e-02 + 5.962e-02 8.118e-02 9.492e-02 9.719e-02 9.195e-02 8.476e-02 7.798e-02 7.099e-02 6.323e-02 5.643e-02 5.366e-02 5.643e-02 - 6.323e-02 7.099e-02 7.798e-02 8.475e-02 9.194e-02 9.719e-02 - 9.491e-02 8.117e-02 5.962e-02 4.041e-02 3.287e-02 4.041e-02 - 7.727e-02 9.146e-02 9.719e-02 9.600e-02 9.227e-02 8.818e-02 - 8.268e-02 7.430e-02 6.446e-02 5.759e-02 5.759e-02 6.446e-02 - 7.430e-02 8.268e-02 8.818e-02 9.227e-02 9.600e-02 9.719e-02 - 9.146e-02 7.727e-02 5.968e-02 4.765e-02 4.765e-02 5.968e-02 - 8.198e-02 8.941e-02 9.194e-02 9.227e-02 9.194e-02 8.941e-02 - 8.198e-02 7.000e-02 5.845e-02 5.366e-02 5.845e-02 7.000e-02 - 8.198e-02 8.941e-02 9.194e-02 9.227e-02 9.194e-02 8.941e-02 - 8.198e-02 7.000e-02 5.845e-02 5.366e-02 5.845e-02 7.000e-02 - 7.603e-02 8.099e-02 8.475e-02 8.818e-02 8.941e-02 8.473e-02 + 6.323e-02 7.099e-02 7.798e-02 8.476e-02 9.195e-02 9.719e-02 + 9.492e-02 8.118e-02 5.962e-02 4.040e-02 3.287e-02 4.040e-02 + 7.728e-02 9.146e-02 9.719e-02 9.601e-02 9.227e-02 8.818e-02 + 8.269e-02 7.430e-02 6.446e-02 5.759e-02 5.759e-02 6.446e-02 + 7.430e-02 8.269e-02 8.818e-02 9.227e-02 9.601e-02 9.719e-02 + 9.146e-02 7.728e-02 5.968e-02 4.765e-02 4.765e-02 5.968e-02 + 8.198e-02 8.941e-02 9.195e-02 9.227e-02 9.195e-02 8.941e-02 + 8.198e-02 7.000e-02 5.844e-02 5.366e-02 5.844e-02 7.000e-02 + 8.198e-02 8.941e-02 9.195e-02 9.227e-02 9.195e-02 8.941e-02 + 8.198e-02 7.000e-02 5.844e-02 5.366e-02 5.844e-02 7.000e-02 + 7.603e-02 8.100e-02 8.476e-02 8.818e-02 8.941e-02 8.474e-02 7.277e-02 5.775e-02 4.740e-02 4.740e-02 5.775e-02 7.277e-02 - 8.473e-02 8.941e-02 8.818e-02 8.475e-02 8.099e-02 7.603e-02 - 6.863e-02 6.009e-02 5.416e-02 5.416e-02 6.009e-02 6.863e-02 - 6.613e-02 7.190e-02 7.798e-02 8.268e-02 8.198e-02 7.277e-02 - 5.728e-02 4.294e-02 3.719e-02 4.294e-02 5.728e-02 7.277e-02 - 8.198e-02 8.268e-02 7.798e-02 7.190e-02 6.613e-02 6.025e-02 - 5.398e-02 4.868e-02 4.655e-02 4.868e-02 5.398e-02 6.025e-02 - 5.689e-02 6.404e-02 7.099e-02 7.430e-02 7.000e-02 5.775e-02 + 8.474e-02 8.941e-02 8.818e-02 8.476e-02 8.100e-02 7.603e-02 + 6.864e-02 6.009e-02 5.416e-02 5.416e-02 6.009e-02 6.864e-02 + 6.613e-02 7.190e-02 7.798e-02 8.269e-02 8.198e-02 7.277e-02 + 5.728e-02 4.294e-02 3.718e-02 4.294e-02 5.728e-02 7.277e-02 + 8.198e-02 8.269e-02 7.798e-02 7.190e-02 6.613e-02 6.025e-02 + 5.398e-02 4.867e-02 4.655e-02 4.867e-02 5.398e-02 6.025e-02 + 5.690e-02 6.405e-02 7.099e-02 7.430e-02 7.000e-02 5.775e-02 4.294e-02 3.313e-02 3.313e-02 4.294e-02 5.775e-02 7.000e-02 - 7.430e-02 7.099e-02 6.404e-02 5.689e-02 5.061e-02 4.502e-02 - 4.029e-02 3.742e-02 3.742e-02 4.029e-02 4.502e-02 5.061e-02 - 4.939e-02 5.700e-02 6.323e-02 6.446e-02 5.845e-02 4.740e-02 - 3.719e-02 3.313e-02 3.719e-02 4.740e-02 5.845e-02 6.446e-02 + 7.430e-02 7.099e-02 6.405e-02 5.690e-02 5.062e-02 4.502e-02 + 4.029e-02 3.742e-02 3.742e-02 4.029e-02 4.502e-02 5.062e-02 + 4.939e-02 5.700e-02 6.323e-02 6.446e-02 5.844e-02 4.740e-02 + 3.718e-02 3.313e-02 3.718e-02 4.740e-02 5.844e-02 6.446e-02 6.323e-02 5.700e-02 4.939e-02 4.252e-02 3.681e-02 3.221e-02 2.903e-02 2.787e-02 2.903e-02 3.221e-02 3.681e-02 4.252e-02 4.330e-02 5.077e-02 5.643e-02 5.759e-02 5.366e-02 4.740e-02 4.294e-02 4.294e-02 4.740e-02 5.366e-02 5.759e-02 5.643e-02 5.077e-02 4.330e-02 3.625e-02 3.042e-02 2.582e-02 2.247e-02 2.066e-02 2.066e-02 2.247e-02 2.582e-02 3.042e-02 3.625e-02 - 3.892e-02 4.685e-02 5.366e-02 5.759e-02 5.845e-02 5.775e-02 - 5.728e-02 5.775e-02 5.845e-02 5.759e-02 5.366e-02 4.685e-02 + 3.892e-02 4.685e-02 5.366e-02 5.759e-02 5.844e-02 5.775e-02 + 5.728e-02 5.775e-02 5.844e-02 5.759e-02 5.366e-02 4.685e-02 3.892e-02 3.163e-02 2.570e-02 2.110e-02 1.770e-02 1.554e-02 1.480e-02 1.554e-02 1.770e-02 2.110e-02 2.570e-02 3.163e-02 - 3.728e-02 4.685e-02 5.643e-02 6.446e-02 7.000e-02 7.277e-02 - 7.277e-02 7.000e-02 6.446e-02 5.643e-02 4.685e-02 3.728e-02 + 3.727e-02 4.685e-02 5.643e-02 6.446e-02 7.000e-02 7.277e-02 + 7.277e-02 7.000e-02 6.446e-02 5.643e-02 4.685e-02 3.727e-02 2.904e-02 2.263e-02 1.784e-02 1.435e-02 1.200e-02 1.081e-02 1.081e-02 1.200e-02 1.435e-02 1.784e-02 2.263e-02 2.904e-02 - 3.892e-02 5.077e-02 6.323e-02 7.430e-02 8.198e-02 8.473e-02 + 3.892e-02 5.077e-02 6.323e-02 7.430e-02 8.198e-02 8.474e-02 8.198e-02 7.430e-02 6.323e-02 5.077e-02 3.892e-02 2.904e-02 2.153e-02 1.612e-02 1.228e-02 9.698e-03 8.199e-03 7.708e-03 8.199e-03 9.698e-03 1.228e-02 1.612e-02 2.153e-02 2.904e-02 - 4.330e-02 5.700e-02 7.099e-02 8.268e-02 8.941e-02 8.941e-02 - 8.268e-02 7.099e-02 5.700e-02 4.330e-02 3.163e-02 2.263e-02 + 4.330e-02 5.700e-02 7.099e-02 8.269e-02 8.941e-02 8.941e-02 + 8.269e-02 7.099e-02 5.700e-02 4.330e-02 3.163e-02 2.263e-02 1.612e-02 1.157e-02 8.541e-03 6.706e-03 5.856e-03 5.856e-03 6.706e-03 8.541e-03 1.157e-02 1.612e-02 2.263e-02 3.163e-02 - 4.939e-02 6.404e-02 7.798e-02 8.818e-02 9.194e-02 8.818e-02 - 7.798e-02 6.404e-02 4.939e-02 3.625e-02 2.570e-02 1.784e-02 + 4.939e-02 6.405e-02 7.798e-02 8.818e-02 9.195e-02 8.818e-02 + 7.798e-02 6.405e-02 4.939e-02 3.625e-02 2.570e-02 1.784e-02 1.228e-02 8.541e-03 6.220e-03 5.001e-03 4.629e-03 5.001e-03 6.220e-03 8.541e-03 1.228e-02 1.784e-02 2.570e-02 3.625e-02 - 5.689e-02 7.190e-02 8.475e-02 9.227e-02 9.227e-02 8.475e-02 - 7.190e-02 5.689e-02 4.252e-02 3.042e-02 2.110e-02 1.435e-02 + 5.690e-02 7.190e-02 8.476e-02 9.227e-02 9.227e-02 8.476e-02 + 7.190e-02 5.690e-02 4.252e-02 3.042e-02 2.110e-02 1.435e-02 9.698e-03 6.706e-03 5.001e-03 4.255e-03 4.255e-03 5.001e-03 6.706e-03 9.698e-03 1.435e-02 2.110e-02 3.042e-02 4.252e-02 - 6.613e-02 8.099e-02 9.194e-02 9.600e-02 9.194e-02 8.099e-02 - 6.613e-02 5.061e-02 3.681e-02 2.582e-02 1.770e-02 1.200e-02 + 6.613e-02 8.100e-02 9.195e-02 9.601e-02 9.195e-02 8.100e-02 + 6.613e-02 5.062e-02 3.681e-02 2.582e-02 1.770e-02 1.200e-02 8.199e-03 5.856e-03 4.629e-03 4.255e-03 4.629e-03 5.856e-03 - 8.199e-03 1.200e-02 1.770e-02 2.582e-02 3.681e-02 5.061e-02 + 8.199e-03 1.200e-02 1.770e-02 2.582e-02 3.681e-02 5.062e-02 7.603e-02 8.941e-02 9.719e-02 9.719e-02 8.941e-02 7.603e-02 6.025e-02 4.502e-02 3.221e-02 2.247e-02 1.554e-02 1.081e-02 7.708e-03 5.856e-03 5.001e-03 5.001e-03 5.856e-03 7.708e-03 1.081e-02 1.554e-02 2.247e-02 3.221e-02 4.502e-02 6.025e-02 - 8.198e-02 9.146e-02 9.491e-02 9.146e-02 8.198e-02 6.863e-02 + 8.198e-02 9.146e-02 9.492e-02 9.146e-02 8.198e-02 6.864e-02 5.398e-02 4.029e-02 2.903e-02 2.066e-02 1.480e-02 1.081e-02 8.199e-03 6.706e-03 6.220e-03 6.706e-03 8.199e-03 1.081e-02 - 1.480e-02 2.066e-02 2.903e-02 4.029e-02 5.398e-02 6.863e-02 - 7.727e-02 8.117e-02 8.117e-02 7.727e-02 7.000e-02 6.009e-02 - 4.868e-02 3.742e-02 2.787e-02 2.066e-02 1.554e-02 1.200e-02 + 1.480e-02 2.066e-02 2.903e-02 4.029e-02 5.398e-02 6.864e-02 + 7.728e-02 8.118e-02 8.118e-02 7.728e-02 7.000e-02 6.009e-02 + 4.867e-02 3.742e-02 2.787e-02 2.066e-02 1.554e-02 1.200e-02 9.698e-03 8.541e-03 8.541e-03 9.698e-03 1.200e-02 1.554e-02 - 2.066e-02 2.787e-02 3.742e-02 4.868e-02 6.009e-02 7.000e-02 - 5.962e-02 5.950e-02 5.962e-02 5.968e-02 5.845e-02 5.416e-02 + 2.066e-02 2.787e-02 3.742e-02 4.867e-02 6.009e-02 7.000e-02 + 5.962e-02 5.950e-02 5.962e-02 5.968e-02 5.844e-02 5.416e-02 4.655e-02 3.742e-02 2.903e-02 2.247e-02 1.770e-02 1.435e-02 1.228e-02 1.157e-02 1.228e-02 1.435e-02 1.770e-02 2.247e-02 - 2.903e-02 3.742e-02 4.655e-02 5.416e-02 5.845e-02 5.968e-02 - 3.604e-02 3.604e-02 4.041e-02 4.765e-02 5.366e-02 5.416e-02 - 4.868e-02 4.029e-02 3.221e-02 2.582e-02 2.110e-02 1.784e-02 + 2.903e-02 3.742e-02 4.655e-02 5.416e-02 5.844e-02 5.968e-02 + 3.604e-02 3.604e-02 4.040e-02 4.765e-02 5.366e-02 5.416e-02 + 4.867e-02 4.029e-02 3.221e-02 2.582e-02 2.110e-02 1.784e-02 1.612e-02 1.612e-02 1.784e-02 2.110e-02 2.582e-02 3.221e-02 - 4.029e-02 4.868e-02 5.416e-02 5.366e-02 4.765e-02 4.041e-02 - 1.770e-02 2.162e-02 3.287e-02 4.765e-02 5.845e-02 6.009e-02 + 4.029e-02 4.867e-02 5.416e-02 5.366e-02 4.765e-02 4.040e-02 + 1.770e-02 2.162e-02 3.287e-02 4.765e-02 5.844e-02 6.009e-02 5.398e-02 4.502e-02 3.681e-02 3.042e-02 2.570e-02 2.263e-02 2.153e-02 2.263e-02 2.570e-02 3.042e-02 3.681e-02 4.502e-02 - 5.398e-02 6.009e-02 5.845e-02 4.765e-02 3.287e-02 2.162e-02 - 1.115e-02 2.162e-02 4.041e-02 5.968e-02 7.000e-02 6.863e-02 - 6.025e-02 5.061e-02 4.252e-02 3.625e-02 3.163e-02 2.904e-02 - 2.904e-02 3.163e-02 3.625e-02 4.252e-02 5.061e-02 6.025e-02 - 6.863e-02 7.000e-02 5.968e-02 4.041e-02 2.162e-02 1.115e-02 - 3.853e-02 5.903e-02 7.727e-02 8.528e-02 8.268e-02 7.478e-02 - 6.639e-02 5.911e-02 5.260e-02 4.693e-02 4.341e-02 4.341e-02 - 4.693e-02 5.260e-02 5.911e-02 6.639e-02 7.478e-02 8.268e-02 - 8.528e-02 7.727e-02 5.903e-02 3.853e-02 2.556e-02 2.556e-02 - 5.903e-02 7.896e-02 9.146e-02 9.329e-02 8.818e-02 8.128e-02 - 7.478e-02 6.809e-02 6.074e-02 5.436e-02 5.176e-02 5.436e-02 - 6.074e-02 6.809e-02 7.478e-02 8.128e-02 8.818e-02 9.329e-02 - 9.146e-02 7.896e-02 5.903e-02 4.108e-02 3.399e-02 4.108e-02 - 7.727e-02 9.146e-02 9.719e-02 9.600e-02 9.227e-02 8.818e-02 - 8.268e-02 7.430e-02 6.446e-02 5.759e-02 5.759e-02 6.446e-02 - 7.430e-02 8.268e-02 8.818e-02 9.227e-02 9.600e-02 9.719e-02 - 9.146e-02 7.727e-02 5.968e-02 4.765e-02 4.765e-02 5.968e-02 - 8.528e-02 9.329e-02 9.600e-02 9.634e-02 9.600e-02 9.329e-02 - 8.528e-02 7.233e-02 5.986e-02 5.469e-02 5.986e-02 7.233e-02 - 8.528e-02 9.329e-02 9.600e-02 9.634e-02 9.600e-02 9.329e-02 - 8.528e-02 7.233e-02 5.986e-02 5.469e-02 5.986e-02 7.233e-02 - 8.268e-02 8.818e-02 9.227e-02 9.600e-02 9.719e-02 9.146e-02 - 7.727e-02 5.968e-02 4.765e-02 4.765e-02 5.968e-02 7.727e-02 - 9.146e-02 9.719e-02 9.600e-02 9.227e-02 8.818e-02 8.268e-02 + 5.398e-02 6.009e-02 5.844e-02 4.765e-02 3.287e-02 2.162e-02 + 1.114e-02 2.162e-02 4.040e-02 5.968e-02 7.000e-02 6.864e-02 + 6.025e-02 5.062e-02 4.252e-02 3.625e-02 3.163e-02 2.904e-02 + 2.904e-02 3.163e-02 3.625e-02 4.252e-02 5.062e-02 6.025e-02 + 6.864e-02 7.000e-02 5.968e-02 4.040e-02 2.162e-02 1.114e-02 + 3.852e-02 5.903e-02 7.728e-02 8.528e-02 8.269e-02 7.478e-02 + 6.639e-02 5.911e-02 5.260e-02 4.693e-02 4.340e-02 4.340e-02 + 4.693e-02 5.260e-02 5.911e-02 6.639e-02 7.478e-02 8.269e-02 + 8.528e-02 7.728e-02 5.903e-02 3.852e-02 2.555e-02 2.555e-02 + 5.903e-02 7.897e-02 9.146e-02 9.329e-02 8.818e-02 8.129e-02 + 7.478e-02 6.809e-02 6.074e-02 5.435e-02 5.176e-02 5.435e-02 + 6.074e-02 6.809e-02 7.478e-02 8.129e-02 8.818e-02 9.329e-02 + 9.146e-02 7.897e-02 5.903e-02 4.108e-02 3.399e-02 4.108e-02 + 7.728e-02 9.146e-02 9.719e-02 9.601e-02 9.227e-02 8.818e-02 + 8.269e-02 7.430e-02 6.446e-02 5.759e-02 5.759e-02 6.446e-02 + 7.430e-02 8.269e-02 8.818e-02 9.227e-02 9.601e-02 9.719e-02 + 9.146e-02 7.728e-02 5.968e-02 4.765e-02 4.765e-02 5.968e-02 + 8.528e-02 9.329e-02 9.601e-02 9.635e-02 9.601e-02 9.329e-02 + 8.528e-02 7.233e-02 5.985e-02 5.469e-02 5.985e-02 7.233e-02 + 8.528e-02 9.329e-02 9.601e-02 9.635e-02 9.601e-02 9.329e-02 + 8.528e-02 7.233e-02 5.985e-02 5.469e-02 5.985e-02 7.233e-02 + 8.269e-02 8.818e-02 9.227e-02 9.601e-02 9.719e-02 9.146e-02 + 7.728e-02 5.968e-02 4.765e-02 4.765e-02 5.968e-02 7.728e-02 + 9.146e-02 9.719e-02 9.601e-02 9.227e-02 8.818e-02 8.269e-02 7.430e-02 6.446e-02 5.759e-02 5.759e-02 6.446e-02 7.430e-02 - 7.478e-02 8.128e-02 8.818e-02 9.329e-02 9.146e-02 7.896e-02 - 5.903e-02 4.108e-02 3.399e-02 4.108e-02 5.903e-02 7.896e-02 - 9.146e-02 9.329e-02 8.818e-02 8.128e-02 7.478e-02 6.809e-02 - 6.074e-02 5.436e-02 5.176e-02 5.436e-02 6.074e-02 6.809e-02 - 6.639e-02 7.478e-02 8.268e-02 8.528e-02 7.727e-02 5.903e-02 - 3.853e-02 2.556e-02 2.556e-02 3.853e-02 5.903e-02 7.727e-02 - 8.528e-02 8.268e-02 7.478e-02 6.639e-02 5.911e-02 5.260e-02 - 4.693e-02 4.341e-02 4.341e-02 4.693e-02 5.260e-02 5.911e-02 + 7.478e-02 8.129e-02 8.818e-02 9.329e-02 9.146e-02 7.897e-02 + 5.903e-02 4.108e-02 3.399e-02 4.108e-02 5.903e-02 7.897e-02 + 9.146e-02 9.329e-02 8.818e-02 8.129e-02 7.478e-02 6.809e-02 + 6.074e-02 5.435e-02 5.176e-02 5.435e-02 6.074e-02 6.809e-02 + 6.639e-02 7.478e-02 8.269e-02 8.528e-02 7.728e-02 5.903e-02 + 3.852e-02 2.555e-02 2.555e-02 3.852e-02 5.903e-02 7.728e-02 + 8.528e-02 8.269e-02 7.478e-02 6.639e-02 5.911e-02 5.260e-02 + 4.693e-02 4.340e-02 4.340e-02 4.693e-02 5.260e-02 5.911e-02 5.911e-02 6.809e-02 7.430e-02 7.233e-02 5.968e-02 4.108e-02 - 2.556e-02 1.973e-02 2.556e-02 4.108e-02 5.968e-02 7.233e-02 + 2.555e-02 1.973e-02 2.555e-02 4.108e-02 5.968e-02 7.233e-02 7.430e-02 6.809e-02 5.911e-02 5.086e-02 4.411e-02 3.864e-02 3.477e-02 3.333e-02 3.477e-02 3.864e-02 4.411e-02 5.086e-02 - 5.260e-02 6.074e-02 6.446e-02 5.986e-02 4.765e-02 3.399e-02 - 2.556e-02 2.556e-02 3.399e-02 4.765e-02 5.986e-02 6.446e-02 + 5.260e-02 6.074e-02 6.446e-02 5.985e-02 4.765e-02 3.399e-02 + 2.555e-02 2.555e-02 3.399e-02 4.765e-02 5.985e-02 6.446e-02 6.074e-02 5.260e-02 4.411e-02 3.705e-02 3.153e-02 2.748e-02 2.524e-02 2.524e-02 2.748e-02 3.153e-02 3.705e-02 4.411e-02 - 4.693e-02 5.436e-02 5.759e-02 5.469e-02 4.765e-02 4.108e-02 - 3.853e-02 4.108e-02 4.765e-02 5.469e-02 5.759e-02 5.436e-02 + 4.693e-02 5.435e-02 5.759e-02 5.469e-02 4.765e-02 4.108e-02 + 3.852e-02 4.108e-02 4.765e-02 5.469e-02 5.759e-02 5.435e-02 4.693e-02 3.864e-02 3.153e-02 2.600e-02 2.191e-02 1.926e-02 1.833e-02 1.926e-02 2.191e-02 2.600e-02 3.153e-02 3.864e-02 - 4.341e-02 5.176e-02 5.759e-02 5.986e-02 5.968e-02 5.903e-02 - 5.903e-02 5.968e-02 5.986e-02 5.759e-02 5.176e-02 4.341e-02 + 4.340e-02 5.176e-02 5.759e-02 5.985e-02 5.968e-02 5.903e-02 + 5.903e-02 5.968e-02 5.985e-02 5.759e-02 5.176e-02 4.340e-02 3.477e-02 2.748e-02 2.191e-02 1.780e-02 1.498e-02 1.351e-02 1.351e-02 1.498e-02 1.780e-02 2.191e-02 2.748e-02 3.477e-02 - 4.341e-02 5.436e-02 6.446e-02 7.233e-02 7.727e-02 7.896e-02 - 7.727e-02 7.233e-02 6.446e-02 5.436e-02 4.341e-02 3.333e-02 + 4.340e-02 5.435e-02 6.446e-02 7.233e-02 7.728e-02 7.897e-02 + 7.728e-02 7.233e-02 6.446e-02 5.435e-02 4.340e-02 3.333e-02 2.524e-02 1.926e-02 1.498e-02 1.201e-02 1.022e-02 9.620e-03 1.022e-02 1.201e-02 1.498e-02 1.926e-02 2.524e-02 3.333e-02 4.693e-02 6.074e-02 7.430e-02 8.528e-02 9.146e-02 9.146e-02 8.528e-02 7.430e-02 6.074e-02 4.693e-02 3.477e-02 2.524e-02 - 1.833e-02 1.351e-02 1.022e-02 8.149e-03 7.143e-03 7.143e-03 - 8.149e-03 1.022e-02 1.351e-02 1.833e-02 2.524e-02 3.477e-02 - 5.260e-02 6.809e-02 8.268e-02 9.329e-02 9.719e-02 9.329e-02 - 8.268e-02 6.809e-02 5.260e-02 3.864e-02 2.748e-02 1.926e-02 + 1.833e-02 1.351e-02 1.022e-02 8.150e-03 7.143e-03 7.143e-03 + 8.150e-03 1.022e-02 1.351e-02 1.833e-02 2.524e-02 3.477e-02 + 5.260e-02 6.809e-02 8.269e-02 9.329e-02 9.719e-02 9.329e-02 + 8.269e-02 6.809e-02 5.260e-02 3.864e-02 2.748e-02 1.926e-02 1.351e-02 9.620e-03 7.143e-03 5.788e-03 5.361e-03 5.788e-03 7.143e-03 9.620e-03 1.351e-02 1.926e-02 2.748e-02 3.864e-02 - 5.911e-02 7.478e-02 8.818e-02 9.600e-02 9.600e-02 8.818e-02 + 5.911e-02 7.478e-02 8.818e-02 9.601e-02 9.601e-02 8.818e-02 7.478e-02 5.911e-02 4.411e-02 3.153e-02 2.191e-02 1.498e-02 1.022e-02 7.143e-03 5.361e-03 4.569e-03 4.569e-03 5.361e-03 7.143e-03 1.022e-02 1.498e-02 2.191e-02 3.153e-02 4.411e-02 - 6.639e-02 8.128e-02 9.227e-02 9.634e-02 9.227e-02 8.128e-02 + 6.639e-02 8.129e-02 9.227e-02 9.635e-02 9.227e-02 8.129e-02 6.639e-02 5.086e-02 3.705e-02 2.600e-02 1.780e-02 1.201e-02 - 8.149e-03 5.788e-03 4.569e-03 4.201e-03 4.569e-03 5.788e-03 - 8.149e-03 1.201e-02 1.780e-02 2.600e-02 3.705e-02 5.086e-02 - 7.478e-02 8.818e-02 9.600e-02 9.600e-02 8.818e-02 7.478e-02 + 8.150e-03 5.788e-03 4.569e-03 4.201e-03 4.569e-03 5.788e-03 + 8.150e-03 1.201e-02 1.780e-02 2.600e-02 3.705e-02 5.086e-02 + 7.478e-02 8.818e-02 9.601e-02 9.601e-02 8.818e-02 7.478e-02 5.911e-02 4.411e-02 3.153e-02 2.191e-02 1.498e-02 1.022e-02 7.143e-03 5.361e-03 4.569e-03 4.569e-03 5.361e-03 7.143e-03 1.022e-02 1.498e-02 2.191e-02 3.153e-02 4.411e-02 5.911e-02 - 8.268e-02 9.329e-02 9.719e-02 9.329e-02 8.268e-02 6.809e-02 + 8.269e-02 9.329e-02 9.719e-02 9.329e-02 8.269e-02 6.809e-02 5.260e-02 3.864e-02 2.748e-02 1.926e-02 1.351e-02 9.620e-03 7.143e-03 5.788e-03 5.361e-03 5.788e-03 7.143e-03 9.620e-03 1.351e-02 1.926e-02 2.748e-02 3.864e-02 5.260e-02 6.809e-02 8.528e-02 9.146e-02 9.146e-02 8.528e-02 7.430e-02 6.074e-02 4.693e-02 3.477e-02 2.524e-02 1.833e-02 1.351e-02 1.022e-02 - 8.149e-03 7.143e-03 7.143e-03 8.149e-03 1.022e-02 1.351e-02 + 8.150e-03 7.143e-03 7.143e-03 8.150e-03 1.022e-02 1.351e-02 1.833e-02 2.524e-02 3.477e-02 4.693e-02 6.074e-02 7.430e-02 - 7.727e-02 7.896e-02 7.727e-02 7.233e-02 6.446e-02 5.436e-02 - 4.341e-02 3.333e-02 2.524e-02 1.926e-02 1.498e-02 1.201e-02 + 7.728e-02 7.897e-02 7.728e-02 7.233e-02 6.446e-02 5.435e-02 + 4.340e-02 3.333e-02 2.524e-02 1.926e-02 1.498e-02 1.201e-02 1.022e-02 9.620e-03 1.022e-02 1.201e-02 1.498e-02 1.926e-02 - 2.524e-02 3.333e-02 4.341e-02 5.436e-02 6.446e-02 7.233e-02 - 5.903e-02 5.903e-02 5.968e-02 5.986e-02 5.759e-02 5.176e-02 - 4.341e-02 3.477e-02 2.748e-02 2.191e-02 1.780e-02 1.498e-02 + 2.524e-02 3.333e-02 4.340e-02 5.435e-02 6.446e-02 7.233e-02 + 5.903e-02 5.903e-02 5.968e-02 5.985e-02 5.759e-02 5.176e-02 + 4.340e-02 3.477e-02 2.748e-02 2.191e-02 1.780e-02 1.498e-02 1.351e-02 1.351e-02 1.498e-02 1.780e-02 2.191e-02 2.748e-02 - 3.477e-02 4.341e-02 5.176e-02 5.759e-02 5.986e-02 5.968e-02 - 3.853e-02 4.108e-02 4.765e-02 5.469e-02 5.759e-02 5.436e-02 + 3.477e-02 4.340e-02 5.176e-02 5.759e-02 5.985e-02 5.968e-02 + 3.852e-02 4.108e-02 4.765e-02 5.469e-02 5.759e-02 5.435e-02 4.693e-02 3.864e-02 3.153e-02 2.600e-02 2.191e-02 1.926e-02 1.833e-02 1.926e-02 2.191e-02 2.600e-02 3.153e-02 3.864e-02 - 4.693e-02 5.436e-02 5.759e-02 5.469e-02 4.765e-02 4.108e-02 - 2.556e-02 3.399e-02 4.765e-02 5.986e-02 6.446e-02 6.074e-02 + 4.693e-02 5.435e-02 5.759e-02 5.469e-02 4.765e-02 4.108e-02 + 2.555e-02 3.399e-02 4.765e-02 5.985e-02 6.446e-02 6.074e-02 5.260e-02 4.411e-02 3.705e-02 3.153e-02 2.748e-02 2.524e-02 2.524e-02 2.748e-02 3.153e-02 3.705e-02 4.411e-02 5.260e-02 - 6.074e-02 6.446e-02 5.986e-02 4.765e-02 3.399e-02 2.556e-02 - 2.556e-02 4.108e-02 5.968e-02 7.233e-02 7.430e-02 6.809e-02 + 6.074e-02 6.446e-02 5.985e-02 4.765e-02 3.399e-02 2.555e-02 + 2.555e-02 4.108e-02 5.968e-02 7.233e-02 7.430e-02 6.809e-02 5.911e-02 5.086e-02 4.411e-02 3.864e-02 3.477e-02 3.333e-02 3.477e-02 3.864e-02 4.411e-02 5.086e-02 5.911e-02 6.809e-02 - 7.430e-02 7.233e-02 5.968e-02 4.108e-02 2.556e-02 1.973e-02 - 5.728e-02 7.277e-02 8.198e-02 8.268e-02 7.798e-02 7.190e-02 - 6.613e-02 6.025e-02 5.398e-02 4.868e-02 4.655e-02 4.868e-02 - 5.398e-02 6.025e-02 6.613e-02 7.190e-02 7.798e-02 8.268e-02 - 8.198e-02 7.277e-02 5.728e-02 4.294e-02 3.719e-02 4.294e-02 - 7.277e-02 8.473e-02 8.941e-02 8.818e-02 8.475e-02 8.099e-02 - 7.603e-02 6.863e-02 6.009e-02 5.416e-02 5.416e-02 6.009e-02 - 6.863e-02 7.603e-02 8.099e-02 8.475e-02 8.818e-02 8.941e-02 - 8.473e-02 7.277e-02 5.775e-02 4.740e-02 4.740e-02 5.775e-02 - 8.198e-02 8.941e-02 9.194e-02 9.227e-02 9.194e-02 8.941e-02 - 8.198e-02 7.000e-02 5.845e-02 5.366e-02 5.845e-02 7.000e-02 - 8.198e-02 8.941e-02 9.194e-02 9.227e-02 9.194e-02 8.941e-02 - 8.198e-02 7.000e-02 5.845e-02 5.366e-02 5.845e-02 7.000e-02 - 8.268e-02 8.818e-02 9.227e-02 9.600e-02 9.719e-02 9.146e-02 - 7.727e-02 5.968e-02 4.765e-02 4.765e-02 5.968e-02 7.727e-02 - 9.146e-02 9.719e-02 9.600e-02 9.227e-02 8.818e-02 8.268e-02 + 7.430e-02 7.233e-02 5.968e-02 4.108e-02 2.555e-02 1.973e-02 + 5.728e-02 7.277e-02 8.198e-02 8.269e-02 7.798e-02 7.190e-02 + 6.613e-02 6.025e-02 5.398e-02 4.867e-02 4.655e-02 4.867e-02 + 5.398e-02 6.025e-02 6.613e-02 7.190e-02 7.798e-02 8.269e-02 + 8.198e-02 7.277e-02 5.728e-02 4.294e-02 3.718e-02 4.294e-02 + 7.277e-02 8.474e-02 8.941e-02 8.818e-02 8.476e-02 8.100e-02 + 7.603e-02 6.864e-02 6.009e-02 5.416e-02 5.416e-02 6.009e-02 + 6.864e-02 7.603e-02 8.100e-02 8.476e-02 8.818e-02 8.941e-02 + 8.474e-02 7.277e-02 5.775e-02 4.740e-02 4.740e-02 5.775e-02 + 8.198e-02 8.941e-02 9.195e-02 9.227e-02 9.195e-02 8.941e-02 + 8.198e-02 7.000e-02 5.844e-02 5.366e-02 5.844e-02 7.000e-02 + 8.198e-02 8.941e-02 9.195e-02 9.227e-02 9.195e-02 8.941e-02 + 8.198e-02 7.000e-02 5.844e-02 5.366e-02 5.844e-02 7.000e-02 + 8.269e-02 8.818e-02 9.227e-02 9.601e-02 9.719e-02 9.146e-02 + 7.728e-02 5.968e-02 4.765e-02 4.765e-02 5.968e-02 7.728e-02 + 9.146e-02 9.719e-02 9.601e-02 9.227e-02 8.818e-02 8.269e-02 7.430e-02 6.446e-02 5.759e-02 5.759e-02 6.446e-02 7.430e-02 - 7.798e-02 8.475e-02 9.194e-02 9.719e-02 9.491e-02 8.117e-02 - 5.962e-02 4.041e-02 3.287e-02 4.041e-02 5.962e-02 8.117e-02 - 9.491e-02 9.719e-02 9.194e-02 8.475e-02 7.798e-02 7.099e-02 + 7.798e-02 8.476e-02 9.195e-02 9.719e-02 9.492e-02 8.118e-02 + 5.962e-02 4.040e-02 3.287e-02 4.040e-02 5.962e-02 8.118e-02 + 9.492e-02 9.719e-02 9.195e-02 8.476e-02 7.798e-02 7.099e-02 6.323e-02 5.643e-02 5.366e-02 5.643e-02 6.323e-02 7.099e-02 - 7.190e-02 8.099e-02 8.941e-02 9.146e-02 8.117e-02 5.950e-02 - 3.604e-02 2.162e-02 2.162e-02 3.604e-02 5.950e-02 8.117e-02 - 9.146e-02 8.941e-02 8.099e-02 7.190e-02 6.404e-02 5.700e-02 - 5.077e-02 4.685e-02 4.685e-02 5.077e-02 5.700e-02 6.404e-02 - 6.613e-02 7.603e-02 8.198e-02 7.727e-02 5.962e-02 3.604e-02 - 1.770e-02 1.115e-02 1.770e-02 3.604e-02 5.962e-02 7.727e-02 - 8.198e-02 7.603e-02 6.613e-02 5.689e-02 4.939e-02 4.330e-02 - 3.892e-02 3.728e-02 3.892e-02 4.330e-02 4.939e-02 5.689e-02 - 6.025e-02 6.863e-02 7.000e-02 5.968e-02 4.041e-02 2.162e-02 - 1.115e-02 1.115e-02 2.162e-02 4.041e-02 5.968e-02 7.000e-02 - 6.863e-02 6.025e-02 5.061e-02 4.252e-02 3.625e-02 3.163e-02 - 2.904e-02 2.904e-02 3.163e-02 3.625e-02 4.252e-02 5.061e-02 - 5.398e-02 6.009e-02 5.845e-02 4.765e-02 3.287e-02 2.162e-02 - 1.770e-02 2.162e-02 3.287e-02 4.765e-02 5.845e-02 6.009e-02 + 7.190e-02 8.100e-02 8.941e-02 9.146e-02 8.118e-02 5.950e-02 + 3.604e-02 2.162e-02 2.162e-02 3.604e-02 5.950e-02 8.118e-02 + 9.146e-02 8.941e-02 8.100e-02 7.190e-02 6.405e-02 5.700e-02 + 5.077e-02 4.685e-02 4.685e-02 5.077e-02 5.700e-02 6.405e-02 + 6.613e-02 7.603e-02 8.198e-02 7.728e-02 5.962e-02 3.604e-02 + 1.770e-02 1.114e-02 1.770e-02 3.604e-02 5.962e-02 7.728e-02 + 8.198e-02 7.603e-02 6.613e-02 5.690e-02 4.939e-02 4.330e-02 + 3.892e-02 3.727e-02 3.892e-02 4.330e-02 4.939e-02 5.690e-02 + 6.025e-02 6.864e-02 7.000e-02 5.968e-02 4.040e-02 2.162e-02 + 1.114e-02 1.114e-02 2.162e-02 4.040e-02 5.968e-02 7.000e-02 + 6.864e-02 6.025e-02 5.062e-02 4.252e-02 3.625e-02 3.163e-02 + 2.904e-02 2.904e-02 3.163e-02 3.625e-02 4.252e-02 5.062e-02 + 5.398e-02 6.009e-02 5.844e-02 4.765e-02 3.287e-02 2.162e-02 + 1.770e-02 2.162e-02 3.287e-02 4.765e-02 5.844e-02 6.009e-02 5.398e-02 4.502e-02 3.681e-02 3.042e-02 2.570e-02 2.263e-02 2.153e-02 2.263e-02 2.570e-02 3.042e-02 3.681e-02 4.502e-02 - 4.868e-02 5.416e-02 5.366e-02 4.765e-02 4.041e-02 3.604e-02 - 3.604e-02 4.041e-02 4.765e-02 5.366e-02 5.416e-02 4.868e-02 + 4.867e-02 5.416e-02 5.366e-02 4.765e-02 4.040e-02 3.604e-02 + 3.604e-02 4.040e-02 4.765e-02 5.366e-02 5.416e-02 4.867e-02 4.029e-02 3.221e-02 2.582e-02 2.110e-02 1.784e-02 1.612e-02 1.612e-02 1.784e-02 2.110e-02 2.582e-02 3.221e-02 4.029e-02 - 4.655e-02 5.416e-02 5.845e-02 5.968e-02 5.962e-02 5.950e-02 - 5.962e-02 5.968e-02 5.845e-02 5.416e-02 4.655e-02 3.742e-02 + 4.655e-02 5.416e-02 5.844e-02 5.968e-02 5.962e-02 5.950e-02 + 5.962e-02 5.968e-02 5.844e-02 5.416e-02 4.655e-02 3.742e-02 2.903e-02 2.247e-02 1.770e-02 1.435e-02 1.228e-02 1.157e-02 1.228e-02 1.435e-02 1.770e-02 2.247e-02 2.903e-02 3.742e-02 - 4.868e-02 6.009e-02 7.000e-02 7.727e-02 8.117e-02 8.117e-02 - 7.727e-02 7.000e-02 6.009e-02 4.868e-02 3.742e-02 2.787e-02 + 4.867e-02 6.009e-02 7.000e-02 7.728e-02 8.118e-02 8.118e-02 + 7.728e-02 7.000e-02 6.009e-02 4.867e-02 3.742e-02 2.787e-02 2.066e-02 1.554e-02 1.200e-02 9.698e-03 8.541e-03 8.541e-03 9.698e-03 1.200e-02 1.554e-02 2.066e-02 2.787e-02 3.742e-02 - 5.398e-02 6.863e-02 8.198e-02 9.146e-02 9.491e-02 9.146e-02 - 8.198e-02 6.863e-02 5.398e-02 4.029e-02 2.903e-02 2.066e-02 + 5.398e-02 6.864e-02 8.198e-02 9.146e-02 9.492e-02 9.146e-02 + 8.198e-02 6.864e-02 5.398e-02 4.029e-02 2.903e-02 2.066e-02 1.480e-02 1.081e-02 8.199e-03 6.706e-03 6.220e-03 6.706e-03 8.199e-03 1.081e-02 1.480e-02 2.066e-02 2.903e-02 4.029e-02 6.025e-02 7.603e-02 8.941e-02 9.719e-02 9.719e-02 8.941e-02 7.603e-02 6.025e-02 4.502e-02 3.221e-02 2.247e-02 1.554e-02 1.081e-02 7.708e-03 5.856e-03 5.001e-03 5.001e-03 5.856e-03 7.708e-03 1.081e-02 1.554e-02 2.247e-02 3.221e-02 4.502e-02 - 6.613e-02 8.099e-02 9.194e-02 9.600e-02 9.194e-02 8.099e-02 - 6.613e-02 5.061e-02 3.681e-02 2.582e-02 1.770e-02 1.200e-02 + 6.613e-02 8.100e-02 9.195e-02 9.601e-02 9.195e-02 8.100e-02 + 6.613e-02 5.062e-02 3.681e-02 2.582e-02 1.770e-02 1.200e-02 8.199e-03 5.856e-03 4.629e-03 4.255e-03 4.629e-03 5.856e-03 - 8.199e-03 1.200e-02 1.770e-02 2.582e-02 3.681e-02 5.061e-02 - 7.190e-02 8.475e-02 9.227e-02 9.227e-02 8.475e-02 7.190e-02 - 5.689e-02 4.252e-02 3.042e-02 2.110e-02 1.435e-02 9.698e-03 + 8.199e-03 1.200e-02 1.770e-02 2.582e-02 3.681e-02 5.062e-02 + 7.190e-02 8.476e-02 9.227e-02 9.227e-02 8.476e-02 7.190e-02 + 5.690e-02 4.252e-02 3.042e-02 2.110e-02 1.435e-02 9.698e-03 6.706e-03 5.001e-03 4.255e-03 4.255e-03 5.001e-03 6.706e-03 - 9.698e-03 1.435e-02 2.110e-02 3.042e-02 4.252e-02 5.689e-02 - 7.798e-02 8.818e-02 9.194e-02 8.818e-02 7.798e-02 6.404e-02 + 9.698e-03 1.435e-02 2.110e-02 3.042e-02 4.252e-02 5.690e-02 + 7.798e-02 8.818e-02 9.195e-02 8.818e-02 7.798e-02 6.405e-02 4.939e-02 3.625e-02 2.570e-02 1.784e-02 1.228e-02 8.541e-03 6.220e-03 5.001e-03 4.629e-03 5.001e-03 6.220e-03 8.541e-03 - 1.228e-02 1.784e-02 2.570e-02 3.625e-02 4.939e-02 6.404e-02 - 8.268e-02 8.941e-02 8.941e-02 8.268e-02 7.099e-02 5.700e-02 + 1.228e-02 1.784e-02 2.570e-02 3.625e-02 4.939e-02 6.405e-02 + 8.269e-02 8.941e-02 8.941e-02 8.269e-02 7.099e-02 5.700e-02 4.330e-02 3.163e-02 2.263e-02 1.612e-02 1.157e-02 8.541e-03 6.706e-03 5.856e-03 5.856e-03 6.706e-03 8.541e-03 1.157e-02 1.612e-02 2.263e-02 3.163e-02 4.330e-02 5.700e-02 7.099e-02 - 8.198e-02 8.473e-02 8.198e-02 7.430e-02 6.323e-02 5.077e-02 + 8.198e-02 8.474e-02 8.198e-02 7.430e-02 6.323e-02 5.077e-02 3.892e-02 2.904e-02 2.153e-02 1.612e-02 1.228e-02 9.698e-03 8.199e-03 7.708e-03 8.199e-03 9.698e-03 1.228e-02 1.612e-02 2.153e-02 2.904e-02 3.892e-02 5.077e-02 6.323e-02 7.430e-02 7.277e-02 7.277e-02 7.000e-02 6.446e-02 5.643e-02 4.685e-02 - 3.728e-02 2.904e-02 2.263e-02 1.784e-02 1.435e-02 1.200e-02 + 3.727e-02 2.904e-02 2.263e-02 1.784e-02 1.435e-02 1.200e-02 1.081e-02 1.081e-02 1.200e-02 1.435e-02 1.784e-02 2.263e-02 - 2.904e-02 3.728e-02 4.685e-02 5.643e-02 6.446e-02 7.000e-02 - 5.728e-02 5.775e-02 5.845e-02 5.759e-02 5.366e-02 4.685e-02 + 2.904e-02 3.727e-02 4.685e-02 5.643e-02 6.446e-02 7.000e-02 + 5.728e-02 5.775e-02 5.844e-02 5.759e-02 5.366e-02 4.685e-02 3.892e-02 3.163e-02 2.570e-02 2.110e-02 1.770e-02 1.554e-02 1.480e-02 1.554e-02 1.770e-02 2.110e-02 2.570e-02 3.163e-02 - 3.892e-02 4.685e-02 5.366e-02 5.759e-02 5.845e-02 5.775e-02 + 3.892e-02 4.685e-02 5.366e-02 5.759e-02 5.844e-02 5.775e-02 4.294e-02 4.740e-02 5.366e-02 5.759e-02 5.643e-02 5.077e-02 4.330e-02 3.625e-02 3.042e-02 2.582e-02 2.247e-02 2.066e-02 2.066e-02 2.247e-02 2.582e-02 3.042e-02 3.625e-02 4.330e-02 5.077e-02 5.643e-02 5.759e-02 5.366e-02 4.740e-02 4.294e-02 - 3.719e-02 4.740e-02 5.845e-02 6.446e-02 6.323e-02 5.700e-02 + 3.718e-02 4.740e-02 5.844e-02 6.446e-02 6.323e-02 5.700e-02 4.939e-02 4.252e-02 3.681e-02 3.221e-02 2.903e-02 2.787e-02 2.903e-02 3.221e-02 3.681e-02 4.252e-02 4.939e-02 5.700e-02 - 6.323e-02 6.446e-02 5.845e-02 4.740e-02 3.719e-02 3.313e-02 - 4.294e-02 5.775e-02 7.000e-02 7.430e-02 7.099e-02 6.404e-02 - 5.689e-02 5.061e-02 4.502e-02 4.029e-02 3.742e-02 3.742e-02 - 4.029e-02 4.502e-02 5.061e-02 5.689e-02 6.404e-02 7.099e-02 + 6.323e-02 6.446e-02 5.844e-02 4.740e-02 3.718e-02 3.313e-02 + 4.294e-02 5.775e-02 7.000e-02 7.430e-02 7.099e-02 6.405e-02 + 5.690e-02 5.062e-02 4.502e-02 4.029e-02 3.742e-02 3.742e-02 + 4.029e-02 4.502e-02 5.062e-02 5.690e-02 6.405e-02 7.099e-02 7.430e-02 7.000e-02 5.775e-02 4.294e-02 3.313e-02 3.313e-02 - 6.466e-02 7.300e-02 7.603e-02 7.478e-02 7.190e-02 6.870e-02 - 6.459e-02 5.881e-02 5.237e-02 4.798e-02 4.798e-02 5.237e-02 - 5.881e-02 6.459e-02 6.870e-02 7.190e-02 7.478e-02 7.603e-02 - 7.300e-02 6.466e-02 5.397e-02 4.650e-02 4.650e-02 5.397e-02 - 7.300e-02 7.892e-02 8.099e-02 8.128e-02 8.099e-02 7.892e-02 + 6.466e-02 7.300e-02 7.603e-02 7.478e-02 7.190e-02 6.871e-02 + 6.460e-02 5.881e-02 5.237e-02 4.798e-02 4.798e-02 5.237e-02 + 5.881e-02 6.460e-02 6.871e-02 7.190e-02 7.478e-02 7.603e-02 + 7.300e-02 6.466e-02 5.396e-02 4.650e-02 4.650e-02 5.396e-02 + 7.300e-02 7.892e-02 8.100e-02 8.129e-02 8.100e-02 7.892e-02 7.300e-02 6.353e-02 5.442e-02 5.065e-02 5.442e-02 6.353e-02 - 7.300e-02 7.892e-02 8.099e-02 8.128e-02 8.099e-02 7.892e-02 + 7.300e-02 7.892e-02 8.100e-02 8.129e-02 8.100e-02 7.892e-02 7.300e-02 6.353e-02 5.442e-02 5.065e-02 5.442e-02 6.353e-02 - 7.603e-02 8.099e-02 8.475e-02 8.818e-02 8.941e-02 8.473e-02 + 7.603e-02 8.100e-02 8.476e-02 8.818e-02 8.941e-02 8.474e-02 7.277e-02 5.775e-02 4.740e-02 4.740e-02 5.775e-02 7.277e-02 - 8.473e-02 8.941e-02 8.818e-02 8.475e-02 8.099e-02 7.603e-02 - 6.863e-02 6.009e-02 5.416e-02 5.416e-02 6.009e-02 6.863e-02 - 7.478e-02 8.128e-02 8.818e-02 9.329e-02 9.146e-02 7.896e-02 - 5.903e-02 4.108e-02 3.399e-02 4.108e-02 5.903e-02 7.896e-02 - 9.146e-02 9.329e-02 8.818e-02 8.128e-02 7.478e-02 6.809e-02 - 6.074e-02 5.436e-02 5.176e-02 5.436e-02 6.074e-02 6.809e-02 - 7.190e-02 8.099e-02 8.941e-02 9.146e-02 8.117e-02 5.950e-02 - 3.604e-02 2.162e-02 2.162e-02 3.604e-02 5.950e-02 8.117e-02 - 9.146e-02 8.941e-02 8.099e-02 7.190e-02 6.404e-02 5.700e-02 - 5.077e-02 4.685e-02 4.685e-02 5.077e-02 5.700e-02 6.404e-02 - 6.870e-02 7.892e-02 8.473e-02 7.896e-02 5.950e-02 3.431e-02 - 1.527e-02 8.601e-03 1.527e-02 3.431e-02 5.950e-02 7.896e-02 - 8.473e-02 7.892e-02 6.870e-02 5.911e-02 5.133e-02 4.501e-02 + 8.474e-02 8.941e-02 8.818e-02 8.476e-02 8.100e-02 7.603e-02 + 6.864e-02 6.009e-02 5.416e-02 5.416e-02 6.009e-02 6.864e-02 + 7.478e-02 8.129e-02 8.818e-02 9.329e-02 9.146e-02 7.897e-02 + 5.903e-02 4.108e-02 3.399e-02 4.108e-02 5.903e-02 7.897e-02 + 9.146e-02 9.329e-02 8.818e-02 8.129e-02 7.478e-02 6.809e-02 + 6.074e-02 5.435e-02 5.176e-02 5.435e-02 6.074e-02 6.809e-02 + 7.190e-02 8.100e-02 8.941e-02 9.146e-02 8.118e-02 5.950e-02 + 3.604e-02 2.162e-02 2.162e-02 3.604e-02 5.950e-02 8.118e-02 + 9.146e-02 8.941e-02 8.100e-02 7.190e-02 6.405e-02 5.700e-02 + 5.077e-02 4.685e-02 4.685e-02 5.077e-02 5.700e-02 6.405e-02 + 6.871e-02 7.892e-02 8.474e-02 7.897e-02 5.950e-02 3.431e-02 + 1.526e-02 8.599e-03 1.526e-02 3.431e-02 5.950e-02 7.897e-02 + 8.474e-02 7.892e-02 6.871e-02 5.911e-02 5.133e-02 4.501e-02 4.045e-02 3.872e-02 4.045e-02 4.501e-02 5.133e-02 5.911e-02 - 6.459e-02 7.300e-02 7.277e-02 5.903e-02 3.604e-02 1.527e-02 - 4.526e-03 4.526e-03 1.527e-02 3.604e-02 5.903e-02 7.277e-02 - 7.300e-02 6.459e-02 5.434e-02 4.565e-02 3.895e-02 3.401e-02 + 6.460e-02 7.300e-02 7.277e-02 5.903e-02 3.604e-02 1.526e-02 + 4.524e-03 4.524e-03 1.526e-02 3.604e-02 5.903e-02 7.277e-02 + 7.300e-02 6.460e-02 5.434e-02 4.565e-02 3.895e-02 3.401e-02 3.122e-02 3.122e-02 3.401e-02 3.895e-02 4.565e-02 5.434e-02 - 5.881e-02 6.353e-02 5.775e-02 4.108e-02 2.162e-02 8.601e-03 - 4.526e-03 8.601e-03 2.162e-02 4.108e-02 5.775e-02 6.353e-02 - 5.881e-02 4.953e-02 4.056e-02 3.354e-02 2.838e-02 2.502e-02 - 2.381e-02 2.502e-02 2.838e-02 3.354e-02 4.056e-02 4.953e-02 - 5.237e-02 5.442e-02 4.740e-02 3.399e-02 2.162e-02 1.527e-02 - 1.527e-02 2.162e-02 3.399e-02 4.740e-02 5.442e-02 5.237e-02 + 5.881e-02 6.353e-02 5.775e-02 4.108e-02 2.162e-02 8.599e-03 + 4.524e-03 8.599e-03 2.162e-02 4.108e-02 5.775e-02 6.353e-02 + 5.881e-02 4.953e-02 4.056e-02 3.355e-02 2.839e-02 2.502e-02 + 2.381e-02 2.502e-02 2.839e-02 3.355e-02 4.056e-02 4.953e-02 + 5.237e-02 5.442e-02 4.740e-02 3.399e-02 2.162e-02 1.526e-02 + 1.526e-02 2.162e-02 3.399e-02 4.740e-02 5.442e-02 5.237e-02 4.466e-02 3.604e-02 2.897e-02 2.375e-02 2.015e-02 1.822e-02 1.822e-02 2.015e-02 2.375e-02 2.897e-02 3.604e-02 4.466e-02 4.798e-02 5.065e-02 4.740e-02 4.108e-02 3.604e-02 3.431e-02 3.604e-02 4.108e-02 4.740e-02 5.065e-02 4.798e-02 4.073e-02 - 3.238e-02 2.531e-02 2.007e-02 1.639e-02 1.411e-02 1.331e-02 - 1.411e-02 1.639e-02 2.007e-02 2.531e-02 3.238e-02 4.073e-02 + 3.238e-02 2.531e-02 2.007e-02 1.640e-02 1.411e-02 1.331e-02 + 1.411e-02 1.640e-02 2.007e-02 2.531e-02 3.238e-02 4.073e-02 4.798e-02 5.442e-02 5.775e-02 5.903e-02 5.950e-02 5.950e-02 5.903e-02 5.775e-02 5.442e-02 4.798e-02 3.920e-02 3.025e-02 2.286e-02 1.744e-02 1.366e-02 1.116e-02 9.883e-03 9.883e-03 1.116e-02 1.366e-02 1.744e-02 2.286e-02 3.025e-02 3.920e-02 - 5.237e-02 6.353e-02 7.277e-02 7.896e-02 8.117e-02 7.896e-02 + 5.237e-02 6.353e-02 7.277e-02 7.897e-02 8.118e-02 7.897e-02 7.277e-02 6.353e-02 5.237e-02 4.073e-02 3.025e-02 2.199e-02 1.606e-02 1.198e-02 9.254e-03 7.650e-03 7.115e-03 7.650e-03 9.254e-03 1.198e-02 1.606e-02 2.199e-02 3.025e-02 4.073e-02 - 5.881e-02 7.300e-02 8.473e-02 9.146e-02 9.146e-02 8.473e-02 + 5.881e-02 7.300e-02 8.474e-02 9.146e-02 9.146e-02 8.474e-02 7.300e-02 5.881e-02 4.466e-02 3.238e-02 2.286e-02 1.606e-02 1.141e-02 8.324e-03 6.418e-03 5.505e-03 5.505e-03 6.418e-03 8.324e-03 1.141e-02 1.606e-02 2.286e-02 3.238e-02 4.466e-02 - 6.459e-02 7.892e-02 8.941e-02 9.329e-02 8.941e-02 7.892e-02 - 6.459e-02 4.953e-02 3.604e-02 2.531e-02 1.744e-02 1.198e-02 + 6.460e-02 7.892e-02 8.941e-02 9.329e-02 8.941e-02 7.892e-02 + 6.460e-02 4.953e-02 3.604e-02 2.531e-02 1.744e-02 1.198e-02 8.324e-03 6.030e-03 4.789e-03 4.400e-03 4.789e-03 6.030e-03 8.324e-03 1.198e-02 1.744e-02 2.531e-02 3.604e-02 4.953e-02 - 6.870e-02 8.099e-02 8.818e-02 8.818e-02 8.099e-02 6.870e-02 + 6.871e-02 8.100e-02 8.818e-02 8.818e-02 8.100e-02 6.871e-02 5.434e-02 4.056e-02 2.897e-02 2.007e-02 1.366e-02 9.254e-03 6.418e-03 4.789e-03 4.068e-03 4.068e-03 4.789e-03 6.418e-03 9.254e-03 1.366e-02 2.007e-02 2.897e-02 4.056e-02 5.434e-02 - 7.190e-02 8.128e-02 8.475e-02 8.128e-02 7.190e-02 5.911e-02 - 4.565e-02 3.354e-02 2.375e-02 1.639e-02 1.116e-02 7.650e-03 + 7.190e-02 8.129e-02 8.476e-02 8.129e-02 7.190e-02 5.911e-02 + 4.565e-02 3.355e-02 2.375e-02 1.640e-02 1.116e-02 7.650e-03 5.505e-03 4.400e-03 4.068e-03 4.400e-03 5.505e-03 7.650e-03 - 1.116e-02 1.639e-02 2.375e-02 3.354e-02 4.565e-02 5.911e-02 - 7.478e-02 8.099e-02 8.099e-02 7.478e-02 6.404e-02 5.133e-02 - 3.895e-02 2.838e-02 2.015e-02 1.411e-02 9.883e-03 7.115e-03 + 1.116e-02 1.640e-02 2.375e-02 3.355e-02 4.565e-02 5.911e-02 + 7.478e-02 8.100e-02 8.100e-02 7.478e-02 6.405e-02 5.133e-02 + 3.895e-02 2.839e-02 2.015e-02 1.411e-02 9.883e-03 7.115e-03 5.505e-03 4.789e-03 4.789e-03 5.505e-03 7.115e-03 9.883e-03 - 1.411e-02 2.015e-02 2.838e-02 3.895e-02 5.133e-02 6.404e-02 + 1.411e-02 2.015e-02 2.839e-02 3.895e-02 5.133e-02 6.405e-02 7.603e-02 7.892e-02 7.603e-02 6.809e-02 5.700e-02 4.501e-02 3.401e-02 2.502e-02 1.822e-02 1.331e-02 9.883e-03 7.650e-03 6.418e-03 6.030e-03 6.418e-03 7.650e-03 9.883e-03 1.331e-02 1.822e-02 2.502e-02 3.401e-02 4.501e-02 5.700e-02 6.809e-02 - 7.300e-02 7.300e-02 6.863e-02 6.074e-02 5.077e-02 4.045e-02 + 7.300e-02 7.300e-02 6.864e-02 6.074e-02 5.077e-02 4.045e-02 3.122e-02 2.381e-02 1.822e-02 1.411e-02 1.116e-02 9.254e-03 8.324e-03 8.324e-03 9.254e-03 1.116e-02 1.411e-02 1.822e-02 - 2.381e-02 3.122e-02 4.045e-02 5.077e-02 6.074e-02 6.863e-02 - 6.466e-02 6.353e-02 6.009e-02 5.436e-02 4.685e-02 3.872e-02 - 3.122e-02 2.502e-02 2.015e-02 1.639e-02 1.366e-02 1.198e-02 - 1.141e-02 1.198e-02 1.366e-02 1.639e-02 2.015e-02 2.502e-02 - 3.122e-02 3.872e-02 4.685e-02 5.436e-02 6.009e-02 6.353e-02 - 5.397e-02 5.442e-02 5.416e-02 5.176e-02 4.685e-02 4.045e-02 - 3.401e-02 2.838e-02 2.375e-02 2.007e-02 1.744e-02 1.606e-02 - 1.606e-02 1.744e-02 2.007e-02 2.375e-02 2.838e-02 3.401e-02 - 4.045e-02 4.685e-02 5.176e-02 5.416e-02 5.442e-02 5.397e-02 - 4.650e-02 5.065e-02 5.416e-02 5.436e-02 5.077e-02 4.501e-02 - 3.895e-02 3.354e-02 2.897e-02 2.531e-02 2.286e-02 2.199e-02 - 2.286e-02 2.531e-02 2.897e-02 3.354e-02 3.895e-02 4.501e-02 - 5.077e-02 5.436e-02 5.416e-02 5.065e-02 4.650e-02 4.471e-02 + 2.381e-02 3.122e-02 4.045e-02 5.077e-02 6.074e-02 6.864e-02 + 6.466e-02 6.353e-02 6.009e-02 5.435e-02 4.685e-02 3.872e-02 + 3.122e-02 2.502e-02 2.015e-02 1.640e-02 1.366e-02 1.198e-02 + 1.141e-02 1.198e-02 1.366e-02 1.640e-02 2.015e-02 2.502e-02 + 3.122e-02 3.872e-02 4.685e-02 5.435e-02 6.009e-02 6.353e-02 + 5.396e-02 5.442e-02 5.416e-02 5.176e-02 4.685e-02 4.045e-02 + 3.401e-02 2.839e-02 2.375e-02 2.007e-02 1.744e-02 1.606e-02 + 1.606e-02 1.744e-02 2.007e-02 2.375e-02 2.839e-02 3.401e-02 + 4.045e-02 4.685e-02 5.176e-02 5.416e-02 5.442e-02 5.396e-02 + 4.650e-02 5.065e-02 5.416e-02 5.435e-02 5.077e-02 4.501e-02 + 3.895e-02 3.355e-02 2.897e-02 2.531e-02 2.286e-02 2.199e-02 + 2.286e-02 2.531e-02 2.897e-02 3.355e-02 3.895e-02 4.501e-02 + 5.077e-02 5.435e-02 5.416e-02 5.065e-02 4.650e-02 4.470e-02 4.650e-02 5.442e-02 6.009e-02 6.074e-02 5.700e-02 5.133e-02 4.565e-02 4.056e-02 3.604e-02 3.238e-02 3.025e-02 3.025e-02 3.238e-02 3.604e-02 4.056e-02 4.565e-02 5.133e-02 5.700e-02 - 6.074e-02 6.009e-02 5.442e-02 4.650e-02 4.091e-02 4.091e-02 - 5.397e-02 6.353e-02 6.863e-02 6.809e-02 6.404e-02 5.911e-02 + 6.074e-02 6.009e-02 5.442e-02 4.650e-02 4.090e-02 4.090e-02 + 5.396e-02 6.353e-02 6.864e-02 6.809e-02 6.405e-02 5.911e-02 5.434e-02 4.953e-02 4.466e-02 4.073e-02 3.920e-02 4.073e-02 - 4.466e-02 4.953e-02 5.434e-02 5.911e-02 6.404e-02 6.809e-02 - 6.863e-02 6.353e-02 5.397e-02 4.471e-02 4.091e-02 4.471e-02 - 6.053e-02 6.459e-02 6.613e-02 6.639e-02 6.613e-02 6.459e-02 + 4.466e-02 4.953e-02 5.434e-02 5.911e-02 6.405e-02 6.809e-02 + 6.864e-02 6.353e-02 5.396e-02 4.470e-02 4.090e-02 4.470e-02 + 6.053e-02 6.460e-02 6.613e-02 6.639e-02 6.613e-02 6.460e-02 6.053e-02 5.421e-02 4.821e-02 4.573e-02 4.821e-02 5.421e-02 - 6.053e-02 6.459e-02 6.613e-02 6.639e-02 6.613e-02 6.459e-02 + 6.053e-02 6.460e-02 6.613e-02 6.639e-02 6.613e-02 6.460e-02 6.053e-02 5.421e-02 4.821e-02 4.573e-02 4.821e-02 5.421e-02 - 6.459e-02 6.870e-02 7.190e-02 7.478e-02 7.603e-02 7.300e-02 - 6.466e-02 5.397e-02 4.650e-02 4.650e-02 5.397e-02 6.466e-02 - 7.300e-02 7.603e-02 7.478e-02 7.190e-02 6.870e-02 6.459e-02 + 6.460e-02 6.871e-02 7.190e-02 7.478e-02 7.603e-02 7.300e-02 + 6.466e-02 5.396e-02 4.650e-02 4.650e-02 5.396e-02 6.466e-02 + 7.300e-02 7.603e-02 7.478e-02 7.190e-02 6.871e-02 6.460e-02 5.881e-02 5.237e-02 4.798e-02 4.798e-02 5.237e-02 5.881e-02 - 6.613e-02 7.190e-02 7.798e-02 8.268e-02 8.198e-02 7.277e-02 - 5.728e-02 4.294e-02 3.719e-02 4.294e-02 5.728e-02 7.277e-02 - 8.198e-02 8.268e-02 7.798e-02 7.190e-02 6.613e-02 6.025e-02 - 5.398e-02 4.868e-02 4.655e-02 4.868e-02 5.398e-02 6.025e-02 - 6.639e-02 7.478e-02 8.268e-02 8.528e-02 7.727e-02 5.903e-02 - 3.853e-02 2.556e-02 2.556e-02 3.853e-02 5.903e-02 7.727e-02 - 8.528e-02 8.268e-02 7.478e-02 6.639e-02 5.911e-02 5.260e-02 - 4.693e-02 4.341e-02 4.341e-02 4.693e-02 5.260e-02 5.911e-02 - 6.613e-02 7.603e-02 8.198e-02 7.727e-02 5.962e-02 3.604e-02 - 1.770e-02 1.115e-02 1.770e-02 3.604e-02 5.962e-02 7.727e-02 - 8.198e-02 7.603e-02 6.613e-02 5.689e-02 4.939e-02 4.330e-02 - 3.892e-02 3.728e-02 3.892e-02 4.330e-02 4.939e-02 5.689e-02 - 6.459e-02 7.300e-02 7.277e-02 5.903e-02 3.604e-02 1.527e-02 - 4.526e-03 4.526e-03 1.527e-02 3.604e-02 5.903e-02 7.277e-02 - 7.300e-02 6.459e-02 5.434e-02 4.565e-02 3.895e-02 3.401e-02 + 6.613e-02 7.190e-02 7.798e-02 8.269e-02 8.198e-02 7.277e-02 + 5.728e-02 4.294e-02 3.718e-02 4.294e-02 5.728e-02 7.277e-02 + 8.198e-02 8.269e-02 7.798e-02 7.190e-02 6.613e-02 6.025e-02 + 5.398e-02 4.867e-02 4.655e-02 4.867e-02 5.398e-02 6.025e-02 + 6.639e-02 7.478e-02 8.269e-02 8.528e-02 7.728e-02 5.903e-02 + 3.852e-02 2.555e-02 2.555e-02 3.852e-02 5.903e-02 7.728e-02 + 8.528e-02 8.269e-02 7.478e-02 6.639e-02 5.911e-02 5.260e-02 + 4.693e-02 4.340e-02 4.340e-02 4.693e-02 5.260e-02 5.911e-02 + 6.613e-02 7.603e-02 8.198e-02 7.728e-02 5.962e-02 3.604e-02 + 1.770e-02 1.114e-02 1.770e-02 3.604e-02 5.962e-02 7.728e-02 + 8.198e-02 7.603e-02 6.613e-02 5.690e-02 4.939e-02 4.330e-02 + 3.892e-02 3.727e-02 3.892e-02 4.330e-02 4.939e-02 5.690e-02 + 6.460e-02 7.300e-02 7.277e-02 5.903e-02 3.604e-02 1.526e-02 + 4.524e-03 4.524e-03 1.526e-02 3.604e-02 5.903e-02 7.277e-02 + 7.300e-02 6.460e-02 5.434e-02 4.565e-02 3.895e-02 3.401e-02 3.122e-02 3.122e-02 3.401e-02 3.895e-02 4.565e-02 5.434e-02 - 6.053e-02 6.466e-02 5.728e-02 3.853e-02 1.770e-02 4.526e-03 - 6.306e-04 4.526e-03 1.770e-02 3.853e-02 5.728e-02 6.466e-02 + 6.053e-02 6.466e-02 5.728e-02 3.852e-02 1.770e-02 4.524e-03 + 6.291e-04 4.524e-03 1.770e-02 3.852e-02 5.728e-02 6.466e-02 6.053e-02 5.116e-02 4.192e-02 3.468e-02 2.936e-02 2.589e-02 2.464e-02 2.589e-02 2.936e-02 3.468e-02 4.192e-02 5.116e-02 - 5.421e-02 5.397e-02 4.294e-02 2.556e-02 1.115e-02 4.526e-03 - 4.526e-03 1.115e-02 2.556e-02 4.294e-02 5.397e-02 5.421e-02 + 5.421e-02 5.396e-02 4.294e-02 2.555e-02 1.114e-02 4.524e-03 + 4.524e-03 1.114e-02 2.555e-02 4.294e-02 5.396e-02 5.421e-02 4.707e-02 3.820e-02 3.074e-02 2.524e-02 2.145e-02 1.942e-02 1.942e-02 2.145e-02 2.524e-02 3.074e-02 3.820e-02 4.707e-02 - 4.821e-02 4.650e-02 3.719e-02 2.556e-02 1.770e-02 1.527e-02 - 1.770e-02 2.556e-02 3.719e-02 4.650e-02 4.821e-02 4.283e-02 + 4.821e-02 4.650e-02 3.718e-02 2.555e-02 1.770e-02 1.526e-02 + 1.770e-02 2.555e-02 3.718e-02 4.650e-02 4.821e-02 4.283e-02 3.469e-02 2.728e-02 2.171e-02 1.781e-02 1.538e-02 1.453e-02 1.538e-02 1.781e-02 2.171e-02 2.728e-02 3.469e-02 4.283e-02 - 4.573e-02 4.650e-02 4.294e-02 3.853e-02 3.604e-02 3.604e-02 - 3.853e-02 4.294e-02 4.650e-02 4.573e-02 4.006e-02 3.206e-02 + 4.573e-02 4.650e-02 4.294e-02 3.852e-02 3.604e-02 3.604e-02 + 3.852e-02 4.294e-02 4.650e-02 4.573e-02 4.005e-02 3.206e-02 2.460e-02 1.893e-02 1.495e-02 1.232e-02 1.096e-02 1.096e-02 - 1.232e-02 1.495e-02 1.893e-02 2.460e-02 3.206e-02 4.006e-02 - 4.821e-02 5.397e-02 5.728e-02 5.903e-02 5.962e-02 5.903e-02 - 5.728e-02 5.397e-02 4.821e-02 4.006e-02 3.107e-02 2.313e-02 + 1.232e-02 1.495e-02 1.893e-02 2.460e-02 3.206e-02 4.005e-02 + 4.821e-02 5.396e-02 5.728e-02 5.903e-02 5.962e-02 5.903e-02 + 5.728e-02 5.396e-02 4.821e-02 4.005e-02 3.107e-02 2.313e-02 1.714e-02 1.296e-02 1.016e-02 8.487e-03 7.921e-03 8.487e-03 - 1.016e-02 1.296e-02 1.714e-02 2.313e-02 3.107e-02 4.006e-02 - 5.421e-02 6.466e-02 7.277e-02 7.727e-02 7.727e-02 7.277e-02 + 1.016e-02 1.296e-02 1.714e-02 2.313e-02 3.107e-02 4.005e-02 + 5.421e-02 6.466e-02 7.277e-02 7.728e-02 7.728e-02 7.277e-02 6.466e-02 5.421e-02 4.283e-02 3.206e-02 2.313e-02 1.653e-02 1.196e-02 8.892e-03 6.959e-03 6.008e-03 6.008e-03 6.959e-03 8.892e-03 1.196e-02 1.653e-02 2.313e-02 3.206e-02 4.283e-02 6.053e-02 7.300e-02 8.198e-02 8.528e-02 8.198e-02 7.300e-02 6.053e-02 4.707e-02 3.469e-02 2.460e-02 1.714e-02 1.196e-02 - 8.479e-03 6.247e-03 4.997e-03 4.595e-03 4.997e-03 6.247e-03 - 8.479e-03 1.196e-02 1.714e-02 2.460e-02 3.469e-02 4.707e-02 - 6.459e-02 7.603e-02 8.268e-02 8.268e-02 7.603e-02 6.459e-02 + 8.478e-03 6.247e-03 4.997e-03 4.595e-03 4.997e-03 6.247e-03 + 8.478e-03 1.196e-02 1.714e-02 2.460e-02 3.469e-02 4.707e-02 + 6.460e-02 7.603e-02 8.269e-02 8.269e-02 7.603e-02 6.460e-02 5.116e-02 3.820e-02 2.728e-02 1.893e-02 1.296e-02 8.892e-03 6.247e-03 4.689e-03 3.979e-03 3.979e-03 4.689e-03 6.247e-03 8.892e-03 1.296e-02 1.893e-02 2.728e-02 3.820e-02 5.116e-02 @@ -642,113 +642,113 @@ Ionic_Step 1 Cubefile created from ABACUS. Inner loop is z, followed by y and x 4.192e-02 3.074e-02 2.171e-02 1.495e-02 1.016e-02 6.959e-03 4.997e-03 3.979e-03 3.671e-03 3.979e-03 4.997e-03 6.959e-03 1.016e-02 1.495e-02 2.171e-02 3.074e-02 4.192e-02 5.434e-02 - 6.639e-02 7.190e-02 7.190e-02 6.639e-02 5.689e-02 4.565e-02 + 6.639e-02 7.190e-02 7.190e-02 6.639e-02 5.690e-02 4.565e-02 3.468e-02 2.524e-02 1.781e-02 1.232e-02 8.487e-03 6.008e-03 4.595e-03 3.979e-03 3.979e-03 4.595e-03 6.008e-03 8.487e-03 - 1.232e-02 1.781e-02 2.524e-02 3.468e-02 4.565e-02 5.689e-02 - 6.613e-02 6.870e-02 6.613e-02 5.911e-02 4.939e-02 3.895e-02 + 1.232e-02 1.781e-02 2.524e-02 3.468e-02 4.565e-02 5.690e-02 + 6.613e-02 6.871e-02 6.613e-02 5.911e-02 4.939e-02 3.895e-02 2.936e-02 2.145e-02 1.538e-02 1.096e-02 7.921e-03 6.008e-03 4.997e-03 4.689e-03 4.997e-03 6.008e-03 7.921e-03 1.096e-02 1.538e-02 2.145e-02 2.936e-02 3.895e-02 4.939e-02 5.911e-02 - 6.459e-02 6.459e-02 6.025e-02 5.260e-02 4.330e-02 3.401e-02 + 6.460e-02 6.460e-02 6.025e-02 5.260e-02 4.330e-02 3.401e-02 2.589e-02 1.942e-02 1.453e-02 1.096e-02 8.487e-03 6.959e-03 6.247e-03 6.247e-03 6.959e-03 8.487e-03 1.096e-02 1.453e-02 1.942e-02 2.589e-02 3.401e-02 4.330e-02 5.260e-02 6.025e-02 6.053e-02 5.881e-02 5.398e-02 4.693e-02 3.892e-02 3.122e-02 2.464e-02 1.942e-02 1.538e-02 1.232e-02 1.016e-02 8.892e-03 - 8.479e-03 8.892e-03 1.016e-02 1.232e-02 1.538e-02 1.942e-02 + 8.478e-03 8.892e-03 1.016e-02 1.232e-02 1.538e-02 1.942e-02 2.464e-02 3.122e-02 3.892e-02 4.693e-02 5.398e-02 5.881e-02 - 5.421e-02 5.237e-02 4.868e-02 4.341e-02 3.728e-02 3.122e-02 + 5.421e-02 5.237e-02 4.867e-02 4.340e-02 3.727e-02 3.122e-02 2.589e-02 2.145e-02 1.781e-02 1.495e-02 1.296e-02 1.196e-02 1.196e-02 1.296e-02 1.495e-02 1.781e-02 2.145e-02 2.589e-02 - 3.122e-02 3.728e-02 4.341e-02 4.868e-02 5.237e-02 5.421e-02 - 4.821e-02 4.798e-02 4.655e-02 4.341e-02 3.892e-02 3.401e-02 + 3.122e-02 3.727e-02 4.340e-02 4.867e-02 5.237e-02 5.421e-02 + 4.821e-02 4.798e-02 4.655e-02 4.340e-02 3.892e-02 3.401e-02 2.936e-02 2.524e-02 2.171e-02 1.893e-02 1.714e-02 1.653e-02 1.714e-02 1.893e-02 2.171e-02 2.524e-02 2.936e-02 3.401e-02 - 3.892e-02 4.341e-02 4.655e-02 4.798e-02 4.821e-02 4.814e-02 - 4.573e-02 4.798e-02 4.868e-02 4.693e-02 4.330e-02 3.895e-02 + 3.892e-02 4.340e-02 4.655e-02 4.798e-02 4.821e-02 4.814e-02 + 4.573e-02 4.798e-02 4.867e-02 4.693e-02 4.330e-02 3.895e-02 3.468e-02 3.074e-02 2.728e-02 2.460e-02 2.313e-02 2.313e-02 2.460e-02 2.728e-02 3.074e-02 3.468e-02 3.895e-02 4.330e-02 - 4.693e-02 4.868e-02 4.798e-02 4.573e-02 4.385e-02 4.385e-02 + 4.693e-02 4.867e-02 4.798e-02 4.573e-02 4.385e-02 4.385e-02 4.821e-02 5.237e-02 5.398e-02 5.260e-02 4.939e-02 4.565e-02 4.192e-02 3.820e-02 3.469e-02 3.206e-02 3.107e-02 3.206e-02 3.469e-02 3.820e-02 4.192e-02 4.565e-02 4.939e-02 5.260e-02 - 5.398e-02 5.237e-02 4.821e-02 4.385e-02 4.201e-02 4.385e-02 - 5.421e-02 5.881e-02 6.025e-02 5.911e-02 5.689e-02 5.434e-02 - 5.116e-02 4.707e-02 4.283e-02 4.006e-02 4.006e-02 4.283e-02 - 4.707e-02 5.116e-02 5.434e-02 5.689e-02 5.911e-02 6.025e-02 + 5.398e-02 5.237e-02 4.821e-02 4.385e-02 4.200e-02 4.385e-02 + 5.421e-02 5.881e-02 6.025e-02 5.911e-02 5.690e-02 5.434e-02 + 5.116e-02 4.707e-02 4.283e-02 4.005e-02 4.005e-02 4.283e-02 + 4.707e-02 5.116e-02 5.434e-02 5.690e-02 5.911e-02 6.025e-02 5.881e-02 5.421e-02 4.814e-02 4.385e-02 4.385e-02 4.814e-02 - 5.116e-02 5.434e-02 5.689e-02 5.911e-02 6.025e-02 5.881e-02 + 5.116e-02 5.434e-02 5.690e-02 5.911e-02 6.025e-02 5.881e-02 5.421e-02 4.814e-02 4.385e-02 4.385e-02 4.814e-02 5.421e-02 - 5.881e-02 6.025e-02 5.911e-02 5.689e-02 5.434e-02 5.116e-02 - 4.707e-02 4.283e-02 4.006e-02 4.006e-02 4.283e-02 4.707e-02 - 5.434e-02 5.911e-02 6.404e-02 6.809e-02 6.863e-02 6.353e-02 - 5.397e-02 4.471e-02 4.091e-02 4.471e-02 5.397e-02 6.353e-02 - 6.863e-02 6.809e-02 6.404e-02 5.911e-02 5.434e-02 4.953e-02 + 5.881e-02 6.025e-02 5.911e-02 5.690e-02 5.434e-02 5.116e-02 + 4.707e-02 4.283e-02 4.005e-02 4.005e-02 4.283e-02 4.707e-02 + 5.434e-02 5.911e-02 6.405e-02 6.809e-02 6.864e-02 6.353e-02 + 5.396e-02 4.470e-02 4.090e-02 4.470e-02 5.396e-02 6.353e-02 + 6.864e-02 6.809e-02 6.405e-02 5.911e-02 5.434e-02 4.953e-02 4.466e-02 4.073e-02 3.920e-02 4.073e-02 4.466e-02 4.953e-02 - 5.689e-02 6.404e-02 7.099e-02 7.430e-02 7.000e-02 5.775e-02 + 5.690e-02 6.405e-02 7.099e-02 7.430e-02 7.000e-02 5.775e-02 4.294e-02 3.313e-02 3.313e-02 4.294e-02 5.775e-02 7.000e-02 - 7.430e-02 7.099e-02 6.404e-02 5.689e-02 5.061e-02 4.502e-02 - 4.029e-02 3.742e-02 3.742e-02 4.029e-02 4.502e-02 5.061e-02 + 7.430e-02 7.099e-02 6.405e-02 5.690e-02 5.062e-02 4.502e-02 + 4.029e-02 3.742e-02 3.742e-02 4.029e-02 4.502e-02 5.062e-02 5.911e-02 6.809e-02 7.430e-02 7.233e-02 5.968e-02 4.108e-02 - 2.556e-02 1.973e-02 2.556e-02 4.108e-02 5.968e-02 7.233e-02 + 2.555e-02 1.973e-02 2.555e-02 4.108e-02 5.968e-02 7.233e-02 7.430e-02 6.809e-02 5.911e-02 5.086e-02 4.411e-02 3.864e-02 3.477e-02 3.333e-02 3.477e-02 3.864e-02 4.411e-02 5.086e-02 - 6.025e-02 6.863e-02 7.000e-02 5.968e-02 4.041e-02 2.162e-02 - 1.115e-02 1.115e-02 2.162e-02 4.041e-02 5.968e-02 7.000e-02 - 6.863e-02 6.025e-02 5.061e-02 4.252e-02 3.625e-02 3.163e-02 - 2.904e-02 2.904e-02 3.163e-02 3.625e-02 4.252e-02 5.061e-02 - 5.881e-02 6.353e-02 5.775e-02 4.108e-02 2.162e-02 8.601e-03 - 4.526e-03 8.601e-03 2.162e-02 4.108e-02 5.775e-02 6.353e-02 - 5.881e-02 4.953e-02 4.056e-02 3.354e-02 2.838e-02 2.502e-02 - 2.381e-02 2.502e-02 2.838e-02 3.354e-02 4.056e-02 4.953e-02 - 5.421e-02 5.397e-02 4.294e-02 2.556e-02 1.115e-02 4.526e-03 - 4.526e-03 1.115e-02 2.556e-02 4.294e-02 5.397e-02 5.421e-02 + 6.025e-02 6.864e-02 7.000e-02 5.968e-02 4.040e-02 2.162e-02 + 1.114e-02 1.114e-02 2.162e-02 4.040e-02 5.968e-02 7.000e-02 + 6.864e-02 6.025e-02 5.062e-02 4.252e-02 3.625e-02 3.163e-02 + 2.904e-02 2.904e-02 3.163e-02 3.625e-02 4.252e-02 5.062e-02 + 5.881e-02 6.353e-02 5.775e-02 4.108e-02 2.162e-02 8.599e-03 + 4.524e-03 8.599e-03 2.162e-02 4.108e-02 5.775e-02 6.353e-02 + 5.881e-02 4.953e-02 4.056e-02 3.355e-02 2.839e-02 2.502e-02 + 2.381e-02 2.502e-02 2.839e-02 3.355e-02 4.056e-02 4.953e-02 + 5.421e-02 5.396e-02 4.294e-02 2.555e-02 1.114e-02 4.524e-03 + 4.524e-03 1.114e-02 2.555e-02 4.294e-02 5.396e-02 5.421e-02 4.707e-02 3.820e-02 3.074e-02 2.524e-02 2.145e-02 1.942e-02 1.942e-02 2.145e-02 2.524e-02 3.074e-02 3.820e-02 4.707e-02 - 4.814e-02 4.471e-02 3.313e-02 1.973e-02 1.115e-02 8.601e-03 - 1.115e-02 1.973e-02 3.313e-02 4.471e-02 4.814e-02 4.355e-02 + 4.814e-02 4.470e-02 3.313e-02 1.973e-02 1.114e-02 8.599e-03 + 1.114e-02 1.973e-02 3.313e-02 4.470e-02 4.814e-02 4.355e-02 3.552e-02 2.799e-02 2.230e-02 1.832e-02 1.584e-02 1.498e-02 1.584e-02 1.832e-02 2.230e-02 2.799e-02 3.552e-02 4.355e-02 - 4.385e-02 4.091e-02 3.313e-02 2.556e-02 2.162e-02 2.162e-02 - 2.556e-02 3.313e-02 4.091e-02 4.385e-02 4.032e-02 3.303e-02 + 4.385e-02 4.090e-02 3.313e-02 2.555e-02 2.162e-02 2.162e-02 + 2.555e-02 3.313e-02 4.090e-02 4.385e-02 4.032e-02 3.303e-02 2.558e-02 1.975e-02 1.566e-02 1.297e-02 1.158e-02 1.158e-02 1.297e-02 1.566e-02 1.975e-02 2.558e-02 3.303e-02 4.032e-02 - 4.385e-02 4.471e-02 4.294e-02 4.108e-02 4.041e-02 4.108e-02 - 4.294e-02 4.471e-02 4.385e-02 3.901e-02 3.151e-02 2.393e-02 + 4.385e-02 4.470e-02 4.294e-02 4.108e-02 4.040e-02 4.108e-02 + 4.294e-02 4.470e-02 4.385e-02 3.901e-02 3.151e-02 2.393e-02 1.789e-02 1.363e-02 1.078e-02 9.073e-03 8.494e-03 9.073e-03 1.078e-02 1.363e-02 1.789e-02 2.393e-02 3.151e-02 3.901e-02 - 4.814e-02 5.397e-02 5.775e-02 5.968e-02 5.968e-02 5.775e-02 - 5.397e-02 4.814e-02 4.032e-02 3.151e-02 2.335e-02 1.693e-02 + 4.814e-02 5.396e-02 5.775e-02 5.968e-02 5.968e-02 5.775e-02 + 5.396e-02 4.814e-02 4.032e-02 3.151e-02 2.335e-02 1.693e-02 1.238e-02 9.326e-03 7.387e-03 6.424e-03 6.424e-03 7.387e-03 9.326e-03 1.238e-02 1.693e-02 2.335e-02 3.151e-02 4.032e-02 5.421e-02 6.353e-02 7.000e-02 7.233e-02 7.000e-02 6.353e-02 5.421e-02 4.355e-02 3.303e-02 2.393e-02 1.693e-02 1.197e-02 8.621e-03 6.445e-03 5.200e-03 4.792e-03 5.200e-03 6.445e-03 8.621e-03 1.197e-02 1.693e-02 2.393e-02 3.303e-02 4.355e-02 - 5.881e-02 6.863e-02 7.430e-02 7.430e-02 6.863e-02 5.881e-02 + 5.881e-02 6.864e-02 7.430e-02 7.430e-02 6.864e-02 5.881e-02 4.707e-02 3.552e-02 2.558e-02 1.789e-02 1.238e-02 8.621e-03 - 6.148e-03 4.651e-03 3.948e-03 3.948e-03 4.651e-03 6.148e-03 + 6.147e-03 4.651e-03 3.948e-03 3.948e-03 4.651e-03 6.147e-03 8.621e-03 1.238e-02 1.789e-02 2.558e-02 3.552e-02 4.707e-02 6.025e-02 6.809e-02 7.099e-02 6.809e-02 6.025e-02 4.953e-02 3.820e-02 2.799e-02 1.975e-02 1.363e-02 9.326e-03 6.445e-03 4.651e-03 3.695e-03 3.399e-03 3.695e-03 4.651e-03 6.445e-03 9.326e-03 1.363e-02 1.975e-02 2.799e-02 3.820e-02 4.953e-02 - 5.911e-02 6.404e-02 6.404e-02 5.911e-02 5.061e-02 4.056e-02 + 5.911e-02 6.405e-02 6.405e-02 5.911e-02 5.062e-02 4.056e-02 3.074e-02 2.230e-02 1.566e-02 1.078e-02 7.387e-03 5.200e-03 3.948e-03 3.399e-03 3.399e-03 3.948e-03 5.200e-03 7.387e-03 - 1.078e-02 1.566e-02 2.230e-02 3.074e-02 4.056e-02 5.061e-02 - 5.689e-02 5.911e-02 5.689e-02 5.086e-02 4.252e-02 3.354e-02 + 1.078e-02 1.566e-02 2.230e-02 3.074e-02 4.056e-02 5.062e-02 + 5.690e-02 5.911e-02 5.690e-02 5.086e-02 4.252e-02 3.355e-02 2.524e-02 1.832e-02 1.297e-02 9.073e-03 6.424e-03 4.792e-03 3.948e-03 3.695e-03 3.948e-03 4.792e-03 6.424e-03 9.073e-03 - 1.297e-02 1.832e-02 2.524e-02 3.354e-02 4.252e-02 5.086e-02 - 5.434e-02 5.434e-02 5.061e-02 4.411e-02 3.625e-02 2.838e-02 + 1.297e-02 1.832e-02 2.524e-02 3.355e-02 4.252e-02 5.086e-02 + 5.434e-02 5.434e-02 5.062e-02 4.411e-02 3.625e-02 2.839e-02 2.145e-02 1.584e-02 1.158e-02 8.494e-03 6.424e-03 5.200e-03 4.651e-03 4.651e-03 5.200e-03 6.424e-03 8.494e-03 1.158e-02 - 1.584e-02 2.145e-02 2.838e-02 3.625e-02 4.411e-02 5.061e-02 + 1.584e-02 2.145e-02 2.839e-02 3.625e-02 4.411e-02 5.062e-02 5.116e-02 4.953e-02 4.502e-02 3.864e-02 3.163e-02 2.502e-02 1.942e-02 1.498e-02 1.158e-02 9.073e-03 7.387e-03 6.445e-03 - 6.148e-03 6.445e-03 7.387e-03 9.073e-03 1.158e-02 1.498e-02 + 6.147e-03 6.445e-03 7.387e-03 9.073e-03 1.158e-02 1.498e-02 1.942e-02 2.502e-02 3.163e-02 3.864e-02 4.502e-02 4.953e-02 4.707e-02 4.466e-02 4.029e-02 3.477e-02 2.904e-02 2.381e-02 1.942e-02 1.584e-02 1.297e-02 1.078e-02 9.326e-03 8.621e-03 @@ -758,104 +758,104 @@ Ionic_Step 1 Cubefile created from ABACUS. Inner loop is z, followed by y and x 2.145e-02 1.832e-02 1.566e-02 1.363e-02 1.238e-02 1.197e-02 1.238e-02 1.363e-02 1.566e-02 1.832e-02 2.145e-02 2.502e-02 2.904e-02 3.333e-02 3.742e-02 4.073e-02 4.283e-02 4.355e-02 - 4.006e-02 3.920e-02 3.742e-02 3.477e-02 3.163e-02 2.838e-02 + 4.005e-02 3.920e-02 3.742e-02 3.477e-02 3.163e-02 2.839e-02 2.524e-02 2.230e-02 1.975e-02 1.789e-02 1.693e-02 1.693e-02 - 1.789e-02 1.975e-02 2.230e-02 2.524e-02 2.838e-02 3.163e-02 - 3.477e-02 3.742e-02 3.920e-02 4.006e-02 4.032e-02 4.032e-02 - 4.006e-02 4.073e-02 4.029e-02 3.864e-02 3.625e-02 3.354e-02 + 1.789e-02 1.975e-02 2.230e-02 2.524e-02 2.839e-02 3.163e-02 + 3.477e-02 3.742e-02 3.920e-02 4.005e-02 4.032e-02 4.032e-02 + 4.005e-02 4.073e-02 4.029e-02 3.864e-02 3.625e-02 3.355e-02 3.074e-02 2.799e-02 2.558e-02 2.393e-02 2.335e-02 2.393e-02 - 2.558e-02 2.799e-02 3.074e-02 3.354e-02 3.625e-02 3.864e-02 - 4.029e-02 4.073e-02 4.006e-02 3.901e-02 3.852e-02 3.901e-02 + 2.558e-02 2.799e-02 3.074e-02 3.355e-02 3.625e-02 3.864e-02 + 4.029e-02 4.073e-02 4.005e-02 3.901e-02 3.852e-02 3.901e-02 4.283e-02 4.466e-02 4.502e-02 4.411e-02 4.252e-02 4.056e-02 3.820e-02 3.552e-02 3.303e-02 3.151e-02 3.151e-02 3.303e-02 3.552e-02 3.820e-02 4.056e-02 4.252e-02 4.411e-02 4.502e-02 4.466e-02 4.283e-02 4.032e-02 3.852e-02 3.852e-02 4.032e-02 - 4.707e-02 4.953e-02 5.061e-02 5.086e-02 5.061e-02 4.953e-02 + 4.707e-02 4.953e-02 5.062e-02 5.086e-02 5.062e-02 4.953e-02 4.707e-02 4.355e-02 4.032e-02 3.901e-02 4.032e-02 4.355e-02 - 4.707e-02 4.953e-02 5.061e-02 5.086e-02 5.061e-02 4.953e-02 + 4.707e-02 4.953e-02 5.062e-02 5.086e-02 5.062e-02 4.953e-02 4.707e-02 4.355e-02 4.032e-02 3.901e-02 4.032e-02 4.355e-02 4.192e-02 4.565e-02 4.939e-02 5.260e-02 5.398e-02 5.237e-02 - 4.821e-02 4.385e-02 4.201e-02 4.385e-02 4.821e-02 5.237e-02 + 4.821e-02 4.385e-02 4.200e-02 4.385e-02 4.821e-02 5.237e-02 5.398e-02 5.260e-02 4.939e-02 4.565e-02 4.192e-02 3.820e-02 3.469e-02 3.206e-02 3.107e-02 3.206e-02 3.469e-02 3.820e-02 4.565e-02 5.133e-02 5.700e-02 6.074e-02 6.009e-02 5.442e-02 - 4.650e-02 4.091e-02 4.091e-02 4.650e-02 5.442e-02 6.009e-02 + 4.650e-02 4.090e-02 4.090e-02 4.650e-02 5.442e-02 6.009e-02 6.074e-02 5.700e-02 5.133e-02 4.565e-02 4.056e-02 3.604e-02 3.238e-02 3.025e-02 3.025e-02 3.238e-02 3.604e-02 4.056e-02 - 4.939e-02 5.700e-02 6.323e-02 6.446e-02 5.845e-02 4.740e-02 - 3.719e-02 3.313e-02 3.719e-02 4.740e-02 5.845e-02 6.446e-02 + 4.939e-02 5.700e-02 6.323e-02 6.446e-02 5.844e-02 4.740e-02 + 3.718e-02 3.313e-02 3.718e-02 4.740e-02 5.844e-02 6.446e-02 6.323e-02 5.700e-02 4.939e-02 4.252e-02 3.681e-02 3.221e-02 2.903e-02 2.787e-02 2.903e-02 3.221e-02 3.681e-02 4.252e-02 - 5.260e-02 6.074e-02 6.446e-02 5.986e-02 4.765e-02 3.399e-02 - 2.556e-02 2.556e-02 3.399e-02 4.765e-02 5.986e-02 6.446e-02 + 5.260e-02 6.074e-02 6.446e-02 5.985e-02 4.765e-02 3.399e-02 + 2.555e-02 2.555e-02 3.399e-02 4.765e-02 5.985e-02 6.446e-02 6.074e-02 5.260e-02 4.411e-02 3.705e-02 3.153e-02 2.748e-02 2.524e-02 2.524e-02 2.748e-02 3.153e-02 3.705e-02 4.411e-02 - 5.398e-02 6.009e-02 5.845e-02 4.765e-02 3.287e-02 2.162e-02 - 1.770e-02 2.162e-02 3.287e-02 4.765e-02 5.845e-02 6.009e-02 + 5.398e-02 6.009e-02 5.844e-02 4.765e-02 3.287e-02 2.162e-02 + 1.770e-02 2.162e-02 3.287e-02 4.765e-02 5.844e-02 6.009e-02 5.398e-02 4.502e-02 3.681e-02 3.042e-02 2.570e-02 2.263e-02 2.153e-02 2.263e-02 2.570e-02 3.042e-02 3.681e-02 4.502e-02 - 5.237e-02 5.442e-02 4.740e-02 3.399e-02 2.162e-02 1.527e-02 - 1.527e-02 2.162e-02 3.399e-02 4.740e-02 5.442e-02 5.237e-02 + 5.237e-02 5.442e-02 4.740e-02 3.399e-02 2.162e-02 1.526e-02 + 1.526e-02 2.162e-02 3.399e-02 4.740e-02 5.442e-02 5.237e-02 4.466e-02 3.604e-02 2.897e-02 2.375e-02 2.015e-02 1.822e-02 1.822e-02 2.015e-02 2.375e-02 2.897e-02 3.604e-02 4.466e-02 - 4.821e-02 4.650e-02 3.719e-02 2.556e-02 1.770e-02 1.527e-02 - 1.770e-02 2.556e-02 3.719e-02 4.650e-02 4.821e-02 4.283e-02 + 4.821e-02 4.650e-02 3.718e-02 2.555e-02 1.770e-02 1.526e-02 + 1.770e-02 2.555e-02 3.718e-02 4.650e-02 4.821e-02 4.283e-02 3.469e-02 2.728e-02 2.171e-02 1.781e-02 1.538e-02 1.453e-02 1.538e-02 1.781e-02 2.171e-02 2.728e-02 3.469e-02 4.283e-02 - 4.385e-02 4.091e-02 3.313e-02 2.556e-02 2.162e-02 2.162e-02 - 2.556e-02 3.313e-02 4.091e-02 4.385e-02 4.032e-02 3.303e-02 + 4.385e-02 4.090e-02 3.313e-02 2.555e-02 2.162e-02 2.162e-02 + 2.555e-02 3.313e-02 4.090e-02 4.385e-02 4.032e-02 3.303e-02 2.558e-02 1.975e-02 1.566e-02 1.297e-02 1.158e-02 1.158e-02 1.297e-02 1.566e-02 1.975e-02 2.558e-02 3.303e-02 4.032e-02 - 4.201e-02 4.091e-02 3.719e-02 3.399e-02 3.287e-02 3.399e-02 - 3.719e-02 4.091e-02 4.201e-02 3.852e-02 3.165e-02 2.422e-02 - 1.816e-02 1.386e-02 1.100e-02 9.285e-03 8.703e-03 9.285e-03 + 4.200e-02 4.090e-02 3.718e-02 3.399e-02 3.287e-02 3.399e-02 + 3.718e-02 4.090e-02 4.200e-02 3.852e-02 3.165e-02 2.422e-02 + 1.816e-02 1.386e-02 1.100e-02 9.285e-03 8.704e-03 9.285e-03 1.100e-02 1.386e-02 1.816e-02 2.422e-02 3.165e-02 3.852e-02 4.385e-02 4.650e-02 4.740e-02 4.765e-02 4.765e-02 4.740e-02 - 4.650e-02 4.385e-02 3.852e-02 3.112e-02 2.348e-02 1.717e-02 - 1.262e-02 9.563e-03 7.626e-03 6.662e-03 6.662e-03 7.626e-03 - 9.563e-03 1.262e-02 1.717e-02 2.348e-02 3.112e-02 3.852e-02 - 4.821e-02 5.442e-02 5.845e-02 5.986e-02 5.845e-02 5.442e-02 - 4.821e-02 4.032e-02 3.165e-02 2.348e-02 1.684e-02 1.201e-02 - 8.723e-03 6.580e-03 5.346e-03 4.940e-03 5.346e-03 6.580e-03 - 8.723e-03 1.201e-02 1.684e-02 2.348e-02 3.165e-02 4.032e-02 + 4.650e-02 4.385e-02 3.852e-02 3.112e-02 2.348e-02 1.716e-02 + 1.262e-02 9.562e-03 7.626e-03 6.662e-03 6.662e-03 7.626e-03 + 9.562e-03 1.262e-02 1.716e-02 2.348e-02 3.112e-02 3.852e-02 + 4.821e-02 5.442e-02 5.844e-02 5.985e-02 5.844e-02 5.442e-02 + 4.821e-02 4.032e-02 3.165e-02 2.348e-02 1.683e-02 1.201e-02 + 8.723e-03 6.580e-03 5.346e-03 4.939e-03 5.346e-03 6.580e-03 + 8.723e-03 1.201e-02 1.683e-02 2.348e-02 3.165e-02 4.032e-02 5.237e-02 6.009e-02 6.446e-02 6.446e-02 6.009e-02 5.237e-02 - 4.283e-02 3.303e-02 2.422e-02 1.717e-02 1.201e-02 8.453e-03 - 6.091e-03 4.638e-03 3.945e-03 3.945e-03 4.638e-03 6.091e-03 - 8.453e-03 1.201e-02 1.717e-02 2.422e-02 3.303e-02 4.283e-02 + 4.283e-02 3.303e-02 2.422e-02 1.716e-02 1.201e-02 8.453e-03 + 6.090e-03 4.638e-03 3.944e-03 3.944e-03 4.638e-03 6.090e-03 + 8.453e-03 1.201e-02 1.716e-02 2.422e-02 3.303e-02 4.283e-02 5.398e-02 6.074e-02 6.323e-02 6.074e-02 5.398e-02 4.466e-02 - 3.469e-02 2.558e-02 1.816e-02 1.262e-02 8.723e-03 6.091e-03 - 4.420e-03 3.506e-03 3.218e-03 3.506e-03 4.420e-03 6.091e-03 + 3.469e-02 2.558e-02 1.816e-02 1.262e-02 8.723e-03 6.090e-03 + 4.420e-03 3.506e-03 3.218e-03 3.506e-03 4.420e-03 6.090e-03 8.723e-03 1.262e-02 1.816e-02 2.558e-02 3.469e-02 4.466e-02 5.260e-02 5.700e-02 5.700e-02 5.260e-02 4.502e-02 3.604e-02 - 2.728e-02 1.975e-02 1.386e-02 9.563e-03 6.580e-03 4.638e-03 + 2.728e-02 1.975e-02 1.386e-02 9.562e-03 6.580e-03 4.638e-03 3.506e-03 2.998e-03 2.998e-03 3.506e-03 4.638e-03 6.580e-03 - 9.563e-03 1.386e-02 1.975e-02 2.728e-02 3.604e-02 4.502e-02 + 9.562e-03 1.386e-02 1.975e-02 2.728e-02 3.604e-02 4.502e-02 4.939e-02 5.133e-02 4.939e-02 4.411e-02 3.681e-02 2.897e-02 - 2.171e-02 1.566e-02 1.100e-02 7.626e-03 5.346e-03 3.945e-03 - 3.218e-03 2.998e-03 3.218e-03 3.945e-03 5.346e-03 7.626e-03 + 2.171e-02 1.566e-02 1.100e-02 7.626e-03 5.346e-03 3.944e-03 + 3.218e-03 2.998e-03 3.218e-03 3.944e-03 5.346e-03 7.626e-03 1.100e-02 1.566e-02 2.171e-02 2.897e-02 3.681e-02 4.411e-02 4.565e-02 4.565e-02 4.252e-02 3.705e-02 3.042e-02 2.375e-02 - 1.781e-02 1.297e-02 9.285e-03 6.662e-03 4.940e-03 3.945e-03 - 3.506e-03 3.506e-03 3.945e-03 4.940e-03 6.662e-03 9.285e-03 + 1.781e-02 1.297e-02 9.285e-03 6.662e-03 4.939e-03 3.944e-03 + 3.506e-03 3.506e-03 3.944e-03 4.939e-03 6.662e-03 9.285e-03 1.297e-02 1.781e-02 2.375e-02 3.042e-02 3.705e-02 4.252e-02 4.192e-02 4.056e-02 3.681e-02 3.153e-02 2.570e-02 2.015e-02 - 1.538e-02 1.158e-02 8.703e-03 6.662e-03 5.346e-03 4.638e-03 - 4.420e-03 4.638e-03 5.346e-03 6.662e-03 8.703e-03 1.158e-02 + 1.538e-02 1.158e-02 8.704e-03 6.662e-03 5.346e-03 4.638e-03 + 4.420e-03 4.638e-03 5.346e-03 6.662e-03 8.704e-03 1.158e-02 1.538e-02 2.015e-02 2.570e-02 3.153e-02 3.681e-02 4.056e-02 3.820e-02 3.604e-02 3.221e-02 2.748e-02 2.263e-02 1.822e-02 - 1.453e-02 1.158e-02 9.285e-03 7.626e-03 6.580e-03 6.091e-03 - 6.091e-03 6.580e-03 7.626e-03 9.285e-03 1.158e-02 1.453e-02 + 1.453e-02 1.158e-02 9.285e-03 7.626e-03 6.580e-03 6.090e-03 + 6.090e-03 6.580e-03 7.626e-03 9.285e-03 1.158e-02 1.453e-02 1.822e-02 2.263e-02 2.748e-02 3.221e-02 3.604e-02 3.820e-02 3.469e-02 3.238e-02 2.903e-02 2.524e-02 2.153e-02 1.822e-02 - 1.538e-02 1.297e-02 1.100e-02 9.563e-03 8.723e-03 8.453e-03 - 8.723e-03 9.563e-03 1.100e-02 1.297e-02 1.538e-02 1.822e-02 + 1.538e-02 1.297e-02 1.100e-02 9.562e-03 8.723e-03 8.453e-03 + 8.723e-03 9.562e-03 1.100e-02 1.297e-02 1.538e-02 1.822e-02 2.153e-02 2.524e-02 2.903e-02 3.238e-02 3.469e-02 3.552e-02 3.206e-02 3.025e-02 2.787e-02 2.524e-02 2.263e-02 2.015e-02 1.781e-02 1.566e-02 1.386e-02 1.262e-02 1.201e-02 1.201e-02 1.262e-02 1.386e-02 1.566e-02 1.781e-02 2.015e-02 2.263e-02 2.524e-02 2.787e-02 3.025e-02 3.206e-02 3.303e-02 3.303e-02 3.107e-02 3.025e-02 2.903e-02 2.748e-02 2.570e-02 2.375e-02 - 2.171e-02 1.975e-02 1.816e-02 1.717e-02 1.684e-02 1.717e-02 + 2.171e-02 1.975e-02 1.816e-02 1.716e-02 1.683e-02 1.716e-02 1.816e-02 1.975e-02 2.171e-02 2.375e-02 2.570e-02 2.748e-02 2.903e-02 3.025e-02 3.107e-02 3.151e-02 3.165e-02 3.151e-02 3.206e-02 3.238e-02 3.221e-02 3.153e-02 3.042e-02 2.897e-02 @@ -870,88 +870,88 @@ Ionic_Step 1 Cubefile created from ABACUS. Inner loop is z, followed by y and x 4.283e-02 4.032e-02 3.852e-02 3.852e-02 4.032e-02 4.283e-02 4.466e-02 4.502e-02 4.411e-02 4.252e-02 4.056e-02 3.820e-02 3.552e-02 3.303e-02 3.151e-02 3.151e-02 3.303e-02 3.552e-02 - 3.468e-02 3.895e-02 4.330e-02 4.693e-02 4.868e-02 4.798e-02 - 4.573e-02 4.385e-02 4.385e-02 4.573e-02 4.798e-02 4.868e-02 + 3.468e-02 3.895e-02 4.330e-02 4.693e-02 4.867e-02 4.798e-02 + 4.573e-02 4.385e-02 4.385e-02 4.573e-02 4.798e-02 4.867e-02 4.693e-02 4.330e-02 3.895e-02 3.468e-02 3.074e-02 2.728e-02 2.460e-02 2.313e-02 2.313e-02 2.460e-02 2.728e-02 3.074e-02 - 3.895e-02 4.501e-02 5.077e-02 5.436e-02 5.416e-02 5.065e-02 - 4.650e-02 4.471e-02 4.650e-02 5.065e-02 5.416e-02 5.436e-02 - 5.077e-02 4.501e-02 3.895e-02 3.354e-02 2.897e-02 2.531e-02 - 2.286e-02 2.199e-02 2.286e-02 2.531e-02 2.897e-02 3.354e-02 + 3.895e-02 4.501e-02 5.077e-02 5.435e-02 5.416e-02 5.065e-02 + 4.650e-02 4.470e-02 4.650e-02 5.065e-02 5.416e-02 5.435e-02 + 5.077e-02 4.501e-02 3.895e-02 3.355e-02 2.897e-02 2.531e-02 + 2.286e-02 2.199e-02 2.286e-02 2.531e-02 2.897e-02 3.355e-02 4.330e-02 5.077e-02 5.643e-02 5.759e-02 5.366e-02 4.740e-02 4.294e-02 4.294e-02 4.740e-02 5.366e-02 5.759e-02 5.643e-02 5.077e-02 4.330e-02 3.625e-02 3.042e-02 2.582e-02 2.247e-02 2.066e-02 2.066e-02 2.247e-02 2.582e-02 3.042e-02 3.625e-02 - 4.693e-02 5.436e-02 5.759e-02 5.469e-02 4.765e-02 4.108e-02 - 3.853e-02 4.108e-02 4.765e-02 5.469e-02 5.759e-02 5.436e-02 + 4.693e-02 5.435e-02 5.759e-02 5.469e-02 4.765e-02 4.108e-02 + 3.852e-02 4.108e-02 4.765e-02 5.469e-02 5.759e-02 5.435e-02 4.693e-02 3.864e-02 3.153e-02 2.600e-02 2.191e-02 1.926e-02 1.833e-02 1.926e-02 2.191e-02 2.600e-02 3.153e-02 3.864e-02 - 4.868e-02 5.416e-02 5.366e-02 4.765e-02 4.041e-02 3.604e-02 - 3.604e-02 4.041e-02 4.765e-02 5.366e-02 5.416e-02 4.868e-02 + 4.867e-02 5.416e-02 5.366e-02 4.765e-02 4.040e-02 3.604e-02 + 3.604e-02 4.040e-02 4.765e-02 5.366e-02 5.416e-02 4.867e-02 4.029e-02 3.221e-02 2.582e-02 2.110e-02 1.784e-02 1.612e-02 1.612e-02 1.784e-02 2.110e-02 2.582e-02 3.221e-02 4.029e-02 4.798e-02 5.065e-02 4.740e-02 4.108e-02 3.604e-02 3.431e-02 3.604e-02 4.108e-02 4.740e-02 5.065e-02 4.798e-02 4.073e-02 - 3.238e-02 2.531e-02 2.007e-02 1.639e-02 1.411e-02 1.331e-02 - 1.411e-02 1.639e-02 2.007e-02 2.531e-02 3.238e-02 4.073e-02 - 4.573e-02 4.650e-02 4.294e-02 3.853e-02 3.604e-02 3.604e-02 - 3.853e-02 4.294e-02 4.650e-02 4.573e-02 4.006e-02 3.206e-02 + 3.238e-02 2.531e-02 2.007e-02 1.640e-02 1.411e-02 1.331e-02 + 1.411e-02 1.640e-02 2.007e-02 2.531e-02 3.238e-02 4.073e-02 + 4.573e-02 4.650e-02 4.294e-02 3.852e-02 3.604e-02 3.604e-02 + 3.852e-02 4.294e-02 4.650e-02 4.573e-02 4.005e-02 3.206e-02 2.460e-02 1.893e-02 1.495e-02 1.232e-02 1.096e-02 1.096e-02 - 1.232e-02 1.495e-02 1.893e-02 2.460e-02 3.206e-02 4.006e-02 - 4.385e-02 4.471e-02 4.294e-02 4.108e-02 4.041e-02 4.108e-02 - 4.294e-02 4.471e-02 4.385e-02 3.901e-02 3.151e-02 2.393e-02 + 1.232e-02 1.495e-02 1.893e-02 2.460e-02 3.206e-02 4.005e-02 + 4.385e-02 4.470e-02 4.294e-02 4.108e-02 4.040e-02 4.108e-02 + 4.294e-02 4.470e-02 4.385e-02 3.901e-02 3.151e-02 2.393e-02 1.789e-02 1.363e-02 1.078e-02 9.073e-03 8.494e-03 9.073e-03 1.078e-02 1.363e-02 1.789e-02 2.393e-02 3.151e-02 3.901e-02 4.385e-02 4.650e-02 4.740e-02 4.765e-02 4.765e-02 4.740e-02 - 4.650e-02 4.385e-02 3.852e-02 3.112e-02 2.348e-02 1.717e-02 - 1.262e-02 9.563e-03 7.626e-03 6.662e-03 6.662e-03 7.626e-03 - 9.563e-03 1.262e-02 1.717e-02 2.348e-02 3.112e-02 3.852e-02 + 4.650e-02 4.385e-02 3.852e-02 3.112e-02 2.348e-02 1.716e-02 + 1.262e-02 9.562e-03 7.626e-03 6.662e-03 6.662e-03 7.626e-03 + 9.562e-03 1.262e-02 1.716e-02 2.348e-02 3.112e-02 3.852e-02 4.573e-02 5.065e-02 5.366e-02 5.469e-02 5.366e-02 5.065e-02 4.573e-02 3.901e-02 3.112e-02 2.333e-02 1.682e-02 1.203e-02 - 8.760e-03 6.629e-03 5.400e-03 4.995e-03 5.400e-03 6.629e-03 + 8.760e-03 6.628e-03 5.400e-03 4.995e-03 5.400e-03 6.628e-03 8.760e-03 1.203e-02 1.682e-02 2.333e-02 3.112e-02 3.901e-02 4.798e-02 5.416e-02 5.759e-02 5.759e-02 5.416e-02 4.798e-02 - 4.006e-02 3.151e-02 2.348e-02 1.682e-02 1.185e-02 8.378e-03 - 6.065e-03 4.635e-03 3.948e-03 3.948e-03 4.635e-03 6.065e-03 - 8.378e-03 1.185e-02 1.682e-02 2.348e-02 3.151e-02 4.006e-02 - 4.868e-02 5.436e-02 5.643e-02 5.436e-02 4.868e-02 4.073e-02 - 3.206e-02 2.393e-02 1.717e-02 1.203e-02 8.378e-03 5.886e-03 + 4.005e-02 3.151e-02 2.348e-02 1.682e-02 1.185e-02 8.378e-03 + 6.064e-03 4.635e-03 3.948e-03 3.948e-03 4.635e-03 6.064e-03 + 8.378e-03 1.185e-02 1.682e-02 2.348e-02 3.151e-02 4.005e-02 + 4.867e-02 5.435e-02 5.643e-02 5.435e-02 4.867e-02 4.073e-02 + 3.206e-02 2.393e-02 1.716e-02 1.203e-02 8.378e-03 5.886e-03 4.284e-03 3.394e-03 3.110e-03 3.394e-03 4.284e-03 5.886e-03 - 8.378e-03 1.203e-02 1.717e-02 2.393e-02 3.206e-02 4.073e-02 + 8.378e-03 1.203e-02 1.716e-02 2.393e-02 3.206e-02 4.073e-02 4.693e-02 5.077e-02 5.077e-02 4.693e-02 4.029e-02 3.238e-02 - 2.460e-02 1.789e-02 1.262e-02 8.760e-03 6.065e-03 4.284e-03 - 3.224e-03 2.739e-03 2.739e-03 3.224e-03 4.284e-03 6.065e-03 + 2.460e-02 1.789e-02 1.262e-02 8.760e-03 6.064e-03 4.284e-03 + 3.224e-03 2.739e-03 2.739e-03 3.224e-03 4.284e-03 6.064e-03 8.760e-03 1.262e-02 1.789e-02 2.460e-02 3.238e-02 4.029e-02 4.330e-02 4.501e-02 4.330e-02 3.864e-02 3.221e-02 2.531e-02 - 1.893e-02 1.363e-02 9.563e-03 6.629e-03 4.635e-03 3.394e-03 - 2.739e-03 2.539e-03 2.739e-03 3.394e-03 4.635e-03 6.629e-03 - 9.563e-03 1.363e-02 1.893e-02 2.531e-02 3.221e-02 3.864e-02 + 1.893e-02 1.363e-02 9.562e-03 6.628e-03 4.635e-03 3.394e-03 + 2.739e-03 2.539e-03 2.739e-03 3.394e-03 4.635e-03 6.628e-03 + 9.562e-03 1.363e-02 1.893e-02 2.531e-02 3.221e-02 3.864e-02 3.895e-02 3.895e-02 3.625e-02 3.153e-02 2.582e-02 2.007e-02 1.495e-02 1.078e-02 7.626e-03 5.400e-03 3.948e-03 3.110e-03 2.739e-03 2.739e-03 3.110e-03 3.948e-03 5.400e-03 7.626e-03 1.078e-02 1.495e-02 2.007e-02 2.582e-02 3.153e-02 3.625e-02 - 3.468e-02 3.354e-02 3.042e-02 2.600e-02 2.110e-02 1.639e-02 + 3.468e-02 3.355e-02 3.042e-02 2.600e-02 2.110e-02 1.640e-02 1.232e-02 9.073e-03 6.662e-03 4.995e-03 3.948e-03 3.394e-03 3.224e-03 3.394e-03 3.948e-03 4.995e-03 6.662e-03 9.073e-03 - 1.232e-02 1.639e-02 2.110e-02 2.600e-02 3.042e-02 3.354e-02 + 1.232e-02 1.640e-02 2.110e-02 2.600e-02 3.042e-02 3.355e-02 3.074e-02 2.897e-02 2.582e-02 2.191e-02 1.784e-02 1.411e-02 1.096e-02 8.494e-03 6.662e-03 5.400e-03 4.635e-03 4.284e-03 4.284e-03 4.635e-03 5.400e-03 6.662e-03 8.494e-03 1.096e-02 1.411e-02 1.784e-02 2.191e-02 2.582e-02 2.897e-02 3.074e-02 2.728e-02 2.531e-02 2.247e-02 1.926e-02 1.612e-02 1.331e-02 - 1.096e-02 9.073e-03 7.626e-03 6.629e-03 6.065e-03 5.886e-03 - 6.065e-03 6.629e-03 7.626e-03 9.073e-03 1.096e-02 1.331e-02 + 1.096e-02 9.073e-03 7.626e-03 6.628e-03 6.064e-03 5.886e-03 + 6.064e-03 6.628e-03 7.626e-03 9.073e-03 1.096e-02 1.331e-02 1.612e-02 1.926e-02 2.247e-02 2.531e-02 2.728e-02 2.799e-02 2.460e-02 2.286e-02 2.066e-02 1.833e-02 1.612e-02 1.411e-02 - 1.232e-02 1.078e-02 9.563e-03 8.760e-03 8.378e-03 8.378e-03 - 8.760e-03 9.563e-03 1.078e-02 1.232e-02 1.411e-02 1.612e-02 + 1.232e-02 1.078e-02 9.562e-03 8.760e-03 8.378e-03 8.378e-03 + 8.760e-03 9.562e-03 1.078e-02 1.232e-02 1.411e-02 1.612e-02 1.833e-02 2.066e-02 2.286e-02 2.460e-02 2.558e-02 2.558e-02 - 2.313e-02 2.199e-02 2.066e-02 1.926e-02 1.784e-02 1.639e-02 + 2.313e-02 2.199e-02 2.066e-02 1.926e-02 1.784e-02 1.640e-02 1.495e-02 1.363e-02 1.262e-02 1.203e-02 1.185e-02 1.203e-02 - 1.262e-02 1.363e-02 1.495e-02 1.639e-02 1.784e-02 1.926e-02 + 1.262e-02 1.363e-02 1.495e-02 1.640e-02 1.784e-02 1.926e-02 2.066e-02 2.199e-02 2.313e-02 2.393e-02 2.422e-02 2.393e-02 2.313e-02 2.286e-02 2.247e-02 2.191e-02 2.110e-02 2.007e-02 - 1.893e-02 1.789e-02 1.717e-02 1.682e-02 1.682e-02 1.717e-02 + 1.893e-02 1.789e-02 1.716e-02 1.682e-02 1.682e-02 1.716e-02 1.789e-02 1.893e-02 2.007e-02 2.110e-02 2.191e-02 2.247e-02 2.286e-02 2.313e-02 2.335e-02 2.348e-02 2.348e-02 2.335e-02 2.460e-02 2.531e-02 2.582e-02 2.600e-02 2.582e-02 2.531e-02 @@ -962,81 +962,81 @@ Ionic_Step 1 Cubefile created from ABACUS. Inner loop is z, followed by y and x 3.206e-02 3.151e-02 3.112e-02 3.112e-02 3.151e-02 3.206e-02 3.238e-02 3.221e-02 3.153e-02 3.042e-02 2.897e-02 2.728e-02 2.558e-02 2.422e-02 2.348e-02 2.348e-02 2.422e-02 2.558e-02 - 3.074e-02 3.354e-02 3.625e-02 3.864e-02 4.029e-02 4.073e-02 - 4.006e-02 3.901e-02 3.852e-02 3.901e-02 4.006e-02 4.073e-02 - 4.029e-02 3.864e-02 3.625e-02 3.354e-02 3.074e-02 2.799e-02 + 3.074e-02 3.355e-02 3.625e-02 3.864e-02 4.029e-02 4.073e-02 + 4.005e-02 3.901e-02 3.852e-02 3.901e-02 4.005e-02 4.073e-02 + 4.029e-02 3.864e-02 3.625e-02 3.355e-02 3.074e-02 2.799e-02 2.558e-02 2.393e-02 2.335e-02 2.393e-02 2.558e-02 2.799e-02 - 2.936e-02 3.401e-02 3.892e-02 4.341e-02 4.655e-02 4.798e-02 - 4.821e-02 4.814e-02 4.821e-02 4.798e-02 4.655e-02 4.341e-02 + 2.936e-02 3.401e-02 3.892e-02 4.340e-02 4.655e-02 4.798e-02 + 4.821e-02 4.814e-02 4.821e-02 4.798e-02 4.655e-02 4.340e-02 3.892e-02 3.401e-02 2.936e-02 2.524e-02 2.171e-02 1.893e-02 1.714e-02 1.653e-02 1.714e-02 1.893e-02 2.171e-02 2.524e-02 3.401e-02 4.045e-02 4.685e-02 5.176e-02 5.416e-02 5.442e-02 - 5.397e-02 5.397e-02 5.442e-02 5.416e-02 5.176e-02 4.685e-02 - 4.045e-02 3.401e-02 2.838e-02 2.375e-02 2.007e-02 1.744e-02 - 1.606e-02 1.606e-02 1.744e-02 2.007e-02 2.375e-02 2.838e-02 - 3.892e-02 4.685e-02 5.366e-02 5.759e-02 5.845e-02 5.775e-02 - 5.728e-02 5.775e-02 5.845e-02 5.759e-02 5.366e-02 4.685e-02 + 5.396e-02 5.396e-02 5.442e-02 5.416e-02 5.176e-02 4.685e-02 + 4.045e-02 3.401e-02 2.839e-02 2.375e-02 2.007e-02 1.744e-02 + 1.606e-02 1.606e-02 1.744e-02 2.007e-02 2.375e-02 2.839e-02 + 3.892e-02 4.685e-02 5.366e-02 5.759e-02 5.844e-02 5.775e-02 + 5.728e-02 5.775e-02 5.844e-02 5.759e-02 5.366e-02 4.685e-02 3.892e-02 3.163e-02 2.570e-02 2.110e-02 1.770e-02 1.554e-02 1.480e-02 1.554e-02 1.770e-02 2.110e-02 2.570e-02 3.163e-02 - 4.341e-02 5.176e-02 5.759e-02 5.986e-02 5.968e-02 5.903e-02 - 5.903e-02 5.968e-02 5.986e-02 5.759e-02 5.176e-02 4.341e-02 + 4.340e-02 5.176e-02 5.759e-02 5.985e-02 5.968e-02 5.903e-02 + 5.903e-02 5.968e-02 5.985e-02 5.759e-02 5.176e-02 4.340e-02 3.477e-02 2.748e-02 2.191e-02 1.780e-02 1.498e-02 1.351e-02 1.351e-02 1.498e-02 1.780e-02 2.191e-02 2.748e-02 3.477e-02 - 4.655e-02 5.416e-02 5.845e-02 5.968e-02 5.962e-02 5.950e-02 - 5.962e-02 5.968e-02 5.845e-02 5.416e-02 4.655e-02 3.742e-02 + 4.655e-02 5.416e-02 5.844e-02 5.968e-02 5.962e-02 5.950e-02 + 5.962e-02 5.968e-02 5.844e-02 5.416e-02 4.655e-02 3.742e-02 2.903e-02 2.247e-02 1.770e-02 1.435e-02 1.228e-02 1.157e-02 1.228e-02 1.435e-02 1.770e-02 2.247e-02 2.903e-02 3.742e-02 4.798e-02 5.442e-02 5.775e-02 5.903e-02 5.950e-02 5.950e-02 5.903e-02 5.775e-02 5.442e-02 4.798e-02 3.920e-02 3.025e-02 2.286e-02 1.744e-02 1.366e-02 1.116e-02 9.883e-03 9.883e-03 1.116e-02 1.366e-02 1.744e-02 2.286e-02 3.025e-02 3.920e-02 - 4.821e-02 5.397e-02 5.728e-02 5.903e-02 5.962e-02 5.903e-02 - 5.728e-02 5.397e-02 4.821e-02 4.006e-02 3.107e-02 2.313e-02 + 4.821e-02 5.396e-02 5.728e-02 5.903e-02 5.962e-02 5.903e-02 + 5.728e-02 5.396e-02 4.821e-02 4.005e-02 3.107e-02 2.313e-02 1.714e-02 1.296e-02 1.016e-02 8.487e-03 7.921e-03 8.487e-03 - 1.016e-02 1.296e-02 1.714e-02 2.313e-02 3.107e-02 4.006e-02 - 4.814e-02 5.397e-02 5.775e-02 5.968e-02 5.968e-02 5.775e-02 - 5.397e-02 4.814e-02 4.032e-02 3.151e-02 2.335e-02 1.693e-02 + 1.016e-02 1.296e-02 1.714e-02 2.313e-02 3.107e-02 4.005e-02 + 4.814e-02 5.396e-02 5.775e-02 5.968e-02 5.968e-02 5.775e-02 + 5.396e-02 4.814e-02 4.032e-02 3.151e-02 2.335e-02 1.693e-02 1.238e-02 9.326e-03 7.387e-03 6.424e-03 6.424e-03 7.387e-03 9.326e-03 1.238e-02 1.693e-02 2.335e-02 3.151e-02 4.032e-02 - 4.821e-02 5.442e-02 5.845e-02 5.986e-02 5.845e-02 5.442e-02 - 4.821e-02 4.032e-02 3.165e-02 2.348e-02 1.684e-02 1.201e-02 - 8.723e-03 6.580e-03 5.346e-03 4.940e-03 5.346e-03 6.580e-03 - 8.723e-03 1.201e-02 1.684e-02 2.348e-02 3.165e-02 4.032e-02 + 4.821e-02 5.442e-02 5.844e-02 5.985e-02 5.844e-02 5.442e-02 + 4.821e-02 4.032e-02 3.165e-02 2.348e-02 1.683e-02 1.201e-02 + 8.723e-03 6.580e-03 5.346e-03 4.939e-03 5.346e-03 6.580e-03 + 8.723e-03 1.201e-02 1.683e-02 2.348e-02 3.165e-02 4.032e-02 4.798e-02 5.416e-02 5.759e-02 5.759e-02 5.416e-02 4.798e-02 - 4.006e-02 3.151e-02 2.348e-02 1.682e-02 1.185e-02 8.378e-03 - 6.065e-03 4.635e-03 3.948e-03 3.948e-03 4.635e-03 6.065e-03 - 8.378e-03 1.185e-02 1.682e-02 2.348e-02 3.151e-02 4.006e-02 + 4.005e-02 3.151e-02 2.348e-02 1.682e-02 1.185e-02 8.378e-03 + 6.064e-03 4.635e-03 3.948e-03 3.948e-03 4.635e-03 6.064e-03 + 8.378e-03 1.185e-02 1.682e-02 2.348e-02 3.151e-02 4.005e-02 4.655e-02 5.176e-02 5.366e-02 5.176e-02 4.655e-02 3.920e-02 - 3.107e-02 2.335e-02 1.684e-02 1.185e-02 8.270e-03 5.820e-03 + 3.107e-02 2.335e-02 1.683e-02 1.185e-02 8.270e-03 5.820e-03 4.238e-03 3.356e-03 3.074e-03 3.356e-03 4.238e-03 5.820e-03 - 8.270e-03 1.185e-02 1.684e-02 2.335e-02 3.107e-02 3.920e-02 - 4.341e-02 4.685e-02 4.685e-02 4.341e-02 3.742e-02 3.025e-02 + 8.270e-03 1.185e-02 1.683e-02 2.335e-02 3.107e-02 3.920e-02 + 4.340e-02 4.685e-02 4.685e-02 4.340e-02 3.742e-02 3.025e-02 2.313e-02 1.693e-02 1.201e-02 8.378e-03 5.820e-03 4.112e-03 3.084e-03 2.610e-03 2.610e-03 3.084e-03 4.112e-03 5.820e-03 8.378e-03 1.201e-02 1.693e-02 2.313e-02 3.025e-02 3.742e-02 3.892e-02 4.045e-02 3.892e-02 3.477e-02 2.903e-02 2.286e-02 - 1.714e-02 1.238e-02 8.723e-03 6.065e-03 4.238e-03 3.084e-03 - 2.467e-03 2.276e-03 2.467e-03 3.084e-03 4.238e-03 6.065e-03 + 1.714e-02 1.238e-02 8.723e-03 6.064e-03 4.238e-03 3.084e-03 + 2.467e-03 2.276e-03 2.467e-03 3.084e-03 4.238e-03 6.064e-03 8.723e-03 1.238e-02 1.714e-02 2.286e-02 2.903e-02 3.477e-02 3.401e-02 3.401e-02 3.163e-02 2.748e-02 2.247e-02 1.744e-02 1.296e-02 9.326e-03 6.580e-03 4.635e-03 3.356e-03 2.610e-03 2.276e-03 2.276e-03 2.610e-03 3.356e-03 4.635e-03 6.580e-03 9.326e-03 1.296e-02 1.744e-02 2.247e-02 2.748e-02 3.163e-02 - 2.936e-02 2.838e-02 2.570e-02 2.191e-02 1.770e-02 1.366e-02 + 2.936e-02 2.839e-02 2.570e-02 2.191e-02 1.770e-02 1.366e-02 1.016e-02 7.387e-03 5.346e-03 3.948e-03 3.074e-03 2.610e-03 2.467e-03 2.610e-03 3.074e-03 3.948e-03 5.346e-03 7.387e-03 - 1.016e-02 1.366e-02 1.770e-02 2.191e-02 2.570e-02 2.838e-02 + 1.016e-02 1.366e-02 1.770e-02 2.191e-02 2.570e-02 2.839e-02 2.524e-02 2.375e-02 2.110e-02 1.780e-02 1.435e-02 1.116e-02 - 8.487e-03 6.424e-03 4.940e-03 3.948e-03 3.356e-03 3.084e-03 - 3.084e-03 3.356e-03 3.948e-03 4.940e-03 6.424e-03 8.487e-03 + 8.487e-03 6.424e-03 4.939e-03 3.948e-03 3.356e-03 3.084e-03 + 3.084e-03 3.356e-03 3.948e-03 4.939e-03 6.424e-03 8.487e-03 1.116e-02 1.435e-02 1.780e-02 2.110e-02 2.375e-02 2.524e-02 2.171e-02 2.007e-02 1.770e-02 1.498e-02 1.228e-02 9.883e-03 7.921e-03 6.424e-03 5.346e-03 4.635e-03 4.238e-03 4.112e-03 4.238e-03 4.635e-03 5.346e-03 6.424e-03 7.921e-03 9.883e-03 1.228e-02 1.498e-02 1.770e-02 2.007e-02 2.171e-02 2.230e-02 1.893e-02 1.744e-02 1.554e-02 1.351e-02 1.157e-02 9.883e-03 - 8.487e-03 7.387e-03 6.580e-03 6.065e-03 5.820e-03 5.820e-03 - 6.065e-03 6.580e-03 7.387e-03 8.487e-03 9.883e-03 1.157e-02 + 8.487e-03 7.387e-03 6.580e-03 6.064e-03 5.820e-03 5.820e-03 + 6.064e-03 6.580e-03 7.387e-03 8.487e-03 9.883e-03 1.157e-02 1.351e-02 1.554e-02 1.744e-02 1.893e-02 1.975e-02 1.975e-02 1.714e-02 1.606e-02 1.480e-02 1.351e-02 1.228e-02 1.116e-02 1.016e-02 9.326e-03 8.723e-03 8.378e-03 8.270e-03 8.378e-03 @@ -1045,48 +1045,48 @@ Ionic_Step 1 Cubefile created from ABACUS. Inner loop is z, followed by y and x 1.653e-02 1.606e-02 1.554e-02 1.498e-02 1.435e-02 1.366e-02 1.296e-02 1.238e-02 1.201e-02 1.185e-02 1.185e-02 1.201e-02 1.238e-02 1.296e-02 1.366e-02 1.435e-02 1.498e-02 1.554e-02 - 1.606e-02 1.653e-02 1.693e-02 1.717e-02 1.717e-02 1.693e-02 + 1.606e-02 1.653e-02 1.693e-02 1.716e-02 1.716e-02 1.693e-02 1.714e-02 1.744e-02 1.770e-02 1.780e-02 1.770e-02 1.744e-02 - 1.714e-02 1.693e-02 1.684e-02 1.682e-02 1.684e-02 1.693e-02 + 1.714e-02 1.693e-02 1.683e-02 1.682e-02 1.683e-02 1.693e-02 1.714e-02 1.744e-02 1.770e-02 1.780e-02 1.770e-02 1.744e-02 - 1.714e-02 1.693e-02 1.684e-02 1.682e-02 1.684e-02 1.693e-02 + 1.714e-02 1.693e-02 1.683e-02 1.682e-02 1.683e-02 1.693e-02 1.893e-02 2.007e-02 2.110e-02 2.191e-02 2.247e-02 2.286e-02 2.313e-02 2.335e-02 2.348e-02 2.348e-02 2.335e-02 2.313e-02 2.286e-02 2.247e-02 2.191e-02 2.110e-02 2.007e-02 1.893e-02 - 1.789e-02 1.717e-02 1.682e-02 1.682e-02 1.717e-02 1.789e-02 + 1.789e-02 1.716e-02 1.682e-02 1.682e-02 1.716e-02 1.789e-02 2.171e-02 2.375e-02 2.570e-02 2.748e-02 2.903e-02 3.025e-02 3.107e-02 3.151e-02 3.165e-02 3.151e-02 3.107e-02 3.025e-02 2.903e-02 2.748e-02 2.570e-02 2.375e-02 2.171e-02 1.975e-02 - 1.816e-02 1.717e-02 1.684e-02 1.717e-02 1.816e-02 1.975e-02 - 2.524e-02 2.838e-02 3.163e-02 3.477e-02 3.742e-02 3.920e-02 - 4.006e-02 4.032e-02 4.032e-02 4.006e-02 3.920e-02 3.742e-02 - 3.477e-02 3.163e-02 2.838e-02 2.524e-02 2.230e-02 1.975e-02 + 1.816e-02 1.716e-02 1.683e-02 1.716e-02 1.816e-02 1.975e-02 + 2.524e-02 2.839e-02 3.163e-02 3.477e-02 3.742e-02 3.920e-02 + 4.005e-02 4.032e-02 4.032e-02 4.005e-02 3.920e-02 3.742e-02 + 3.477e-02 3.163e-02 2.839e-02 2.524e-02 2.230e-02 1.975e-02 1.789e-02 1.693e-02 1.693e-02 1.789e-02 1.975e-02 2.230e-02 - 2.589e-02 3.122e-02 3.728e-02 4.341e-02 4.868e-02 5.237e-02 - 5.421e-02 5.421e-02 5.237e-02 4.868e-02 4.341e-02 3.728e-02 + 2.589e-02 3.122e-02 3.727e-02 4.340e-02 4.867e-02 5.237e-02 + 5.421e-02 5.421e-02 5.237e-02 4.867e-02 4.340e-02 3.727e-02 3.122e-02 2.589e-02 2.145e-02 1.781e-02 1.495e-02 1.296e-02 1.196e-02 1.196e-02 1.296e-02 1.495e-02 1.781e-02 2.145e-02 - 3.122e-02 3.872e-02 4.685e-02 5.436e-02 6.009e-02 6.353e-02 - 6.466e-02 6.353e-02 6.009e-02 5.436e-02 4.685e-02 3.872e-02 - 3.122e-02 2.502e-02 2.015e-02 1.639e-02 1.366e-02 1.198e-02 - 1.141e-02 1.198e-02 1.366e-02 1.639e-02 2.015e-02 2.502e-02 - 3.728e-02 4.685e-02 5.643e-02 6.446e-02 7.000e-02 7.277e-02 - 7.277e-02 7.000e-02 6.446e-02 5.643e-02 4.685e-02 3.728e-02 + 3.122e-02 3.872e-02 4.685e-02 5.435e-02 6.009e-02 6.353e-02 + 6.466e-02 6.353e-02 6.009e-02 5.435e-02 4.685e-02 3.872e-02 + 3.122e-02 2.502e-02 2.015e-02 1.640e-02 1.366e-02 1.198e-02 + 1.141e-02 1.198e-02 1.366e-02 1.640e-02 2.015e-02 2.502e-02 + 3.727e-02 4.685e-02 5.643e-02 6.446e-02 7.000e-02 7.277e-02 + 7.277e-02 7.000e-02 6.446e-02 5.643e-02 4.685e-02 3.727e-02 2.904e-02 2.263e-02 1.784e-02 1.435e-02 1.200e-02 1.081e-02 1.081e-02 1.200e-02 1.435e-02 1.784e-02 2.263e-02 2.904e-02 - 4.341e-02 5.436e-02 6.446e-02 7.233e-02 7.727e-02 7.896e-02 - 7.727e-02 7.233e-02 6.446e-02 5.436e-02 4.341e-02 3.333e-02 + 4.340e-02 5.435e-02 6.446e-02 7.233e-02 7.728e-02 7.897e-02 + 7.728e-02 7.233e-02 6.446e-02 5.435e-02 4.340e-02 3.333e-02 2.524e-02 1.926e-02 1.498e-02 1.201e-02 1.022e-02 9.620e-03 1.022e-02 1.201e-02 1.498e-02 1.926e-02 2.524e-02 3.333e-02 - 4.868e-02 6.009e-02 7.000e-02 7.727e-02 8.117e-02 8.117e-02 - 7.727e-02 7.000e-02 6.009e-02 4.868e-02 3.742e-02 2.787e-02 + 4.867e-02 6.009e-02 7.000e-02 7.728e-02 8.118e-02 8.118e-02 + 7.728e-02 7.000e-02 6.009e-02 4.867e-02 3.742e-02 2.787e-02 2.066e-02 1.554e-02 1.200e-02 9.698e-03 8.541e-03 8.541e-03 9.698e-03 1.200e-02 1.554e-02 2.066e-02 2.787e-02 3.742e-02 - 5.237e-02 6.353e-02 7.277e-02 7.896e-02 8.117e-02 7.896e-02 + 5.237e-02 6.353e-02 7.277e-02 7.897e-02 8.118e-02 7.897e-02 7.277e-02 6.353e-02 5.237e-02 4.073e-02 3.025e-02 2.199e-02 1.606e-02 1.198e-02 9.254e-03 7.650e-03 7.115e-03 7.650e-03 9.254e-03 1.198e-02 1.606e-02 2.199e-02 3.025e-02 4.073e-02 - 5.421e-02 6.466e-02 7.277e-02 7.727e-02 7.727e-02 7.277e-02 + 5.421e-02 6.466e-02 7.277e-02 7.728e-02 7.728e-02 7.277e-02 6.466e-02 5.421e-02 4.283e-02 3.206e-02 2.313e-02 1.653e-02 1.196e-02 8.892e-03 6.959e-03 6.008e-03 6.008e-03 6.959e-03 8.892e-03 1.196e-02 1.653e-02 2.313e-02 3.206e-02 4.283e-02 @@ -1095,44 +1095,44 @@ Ionic_Step 1 Cubefile created from ABACUS. Inner loop is z, followed by y and x 8.621e-03 6.445e-03 5.200e-03 4.792e-03 5.200e-03 6.445e-03 8.621e-03 1.197e-02 1.693e-02 2.393e-02 3.303e-02 4.355e-02 5.237e-02 6.009e-02 6.446e-02 6.446e-02 6.009e-02 5.237e-02 - 4.283e-02 3.303e-02 2.422e-02 1.717e-02 1.201e-02 8.453e-03 - 6.091e-03 4.638e-03 3.945e-03 3.945e-03 4.638e-03 6.091e-03 - 8.453e-03 1.201e-02 1.717e-02 2.422e-02 3.303e-02 4.283e-02 - 4.868e-02 5.436e-02 5.643e-02 5.436e-02 4.868e-02 4.073e-02 - 3.206e-02 2.393e-02 1.717e-02 1.203e-02 8.378e-03 5.886e-03 + 4.283e-02 3.303e-02 2.422e-02 1.716e-02 1.201e-02 8.453e-03 + 6.090e-03 4.638e-03 3.944e-03 3.944e-03 4.638e-03 6.090e-03 + 8.453e-03 1.201e-02 1.716e-02 2.422e-02 3.303e-02 4.283e-02 + 4.867e-02 5.435e-02 5.643e-02 5.435e-02 4.867e-02 4.073e-02 + 3.206e-02 2.393e-02 1.716e-02 1.203e-02 8.378e-03 5.886e-03 4.284e-03 3.394e-03 3.110e-03 3.394e-03 4.284e-03 5.886e-03 - 8.378e-03 1.203e-02 1.717e-02 2.393e-02 3.206e-02 4.073e-02 - 4.341e-02 4.685e-02 4.685e-02 4.341e-02 3.742e-02 3.025e-02 + 8.378e-03 1.203e-02 1.716e-02 2.393e-02 3.206e-02 4.073e-02 + 4.340e-02 4.685e-02 4.685e-02 4.340e-02 3.742e-02 3.025e-02 2.313e-02 1.693e-02 1.201e-02 8.378e-03 5.820e-03 4.112e-03 3.084e-03 2.610e-03 2.610e-03 3.084e-03 4.112e-03 5.820e-03 8.378e-03 1.201e-02 1.693e-02 2.313e-02 3.025e-02 3.742e-02 - 3.728e-02 3.872e-02 3.728e-02 3.333e-02 2.787e-02 2.199e-02 + 3.727e-02 3.872e-02 3.727e-02 3.333e-02 2.787e-02 2.199e-02 1.653e-02 1.197e-02 8.453e-03 5.886e-03 4.112e-03 2.984e-03 2.378e-03 2.190e-03 2.378e-03 2.984e-03 4.112e-03 5.886e-03 8.453e-03 1.197e-02 1.653e-02 2.199e-02 2.787e-02 3.333e-02 3.122e-02 3.122e-02 2.904e-02 2.524e-02 2.066e-02 1.606e-02 - 1.196e-02 8.621e-03 6.091e-03 4.284e-03 3.084e-03 2.378e-03 - 2.060e-03 2.060e-03 2.378e-03 3.084e-03 4.284e-03 6.091e-03 + 1.196e-02 8.621e-03 6.090e-03 4.284e-03 3.084e-03 2.378e-03 + 2.060e-03 2.060e-03 2.378e-03 3.084e-03 4.284e-03 6.090e-03 8.621e-03 1.196e-02 1.606e-02 2.066e-02 2.524e-02 2.904e-02 2.589e-02 2.502e-02 2.263e-02 1.926e-02 1.554e-02 1.198e-02 8.892e-03 6.445e-03 4.638e-03 3.394e-03 2.610e-03 2.190e-03 2.060e-03 2.190e-03 2.610e-03 3.394e-03 4.638e-03 6.445e-03 8.892e-03 1.198e-02 1.554e-02 1.926e-02 2.263e-02 2.502e-02 2.145e-02 2.015e-02 1.784e-02 1.498e-02 1.200e-02 9.254e-03 - 6.959e-03 5.200e-03 3.945e-03 3.110e-03 2.610e-03 2.378e-03 - 2.378e-03 2.610e-03 3.110e-03 3.945e-03 5.200e-03 6.959e-03 + 6.959e-03 5.200e-03 3.944e-03 3.110e-03 2.610e-03 2.378e-03 + 2.378e-03 2.610e-03 3.110e-03 3.944e-03 5.200e-03 6.959e-03 9.254e-03 1.200e-02 1.498e-02 1.784e-02 2.015e-02 2.145e-02 - 1.781e-02 1.639e-02 1.435e-02 1.201e-02 9.698e-03 7.650e-03 - 6.008e-03 4.792e-03 3.945e-03 3.394e-03 3.084e-03 2.984e-03 - 3.084e-03 3.394e-03 3.945e-03 4.792e-03 6.008e-03 7.650e-03 - 9.698e-03 1.201e-02 1.435e-02 1.639e-02 1.781e-02 1.832e-02 + 1.781e-02 1.640e-02 1.435e-02 1.201e-02 9.698e-03 7.650e-03 + 6.008e-03 4.792e-03 3.944e-03 3.394e-03 3.084e-03 2.984e-03 + 3.084e-03 3.394e-03 3.944e-03 4.792e-03 6.008e-03 7.650e-03 + 9.698e-03 1.201e-02 1.435e-02 1.640e-02 1.781e-02 1.832e-02 1.495e-02 1.366e-02 1.200e-02 1.022e-02 8.541e-03 7.115e-03 6.008e-03 5.200e-03 4.638e-03 4.284e-03 4.112e-03 4.112e-03 4.284e-03 4.638e-03 5.200e-03 6.008e-03 7.115e-03 8.541e-03 1.022e-02 1.200e-02 1.366e-02 1.495e-02 1.566e-02 1.566e-02 1.296e-02 1.198e-02 1.081e-02 9.620e-03 8.541e-03 7.650e-03 - 6.959e-03 6.445e-03 6.091e-03 5.886e-03 5.820e-03 5.886e-03 - 6.091e-03 6.445e-03 6.959e-03 7.650e-03 8.541e-03 9.620e-03 + 6.959e-03 6.445e-03 6.090e-03 5.886e-03 5.820e-03 5.886e-03 + 6.090e-03 6.445e-03 6.959e-03 7.650e-03 8.541e-03 9.620e-03 1.081e-02 1.198e-02 1.296e-02 1.363e-02 1.386e-02 1.363e-02 1.196e-02 1.141e-02 1.081e-02 1.022e-02 9.698e-03 9.254e-03 8.892e-03 8.621e-03 8.453e-03 8.378e-03 8.378e-03 8.453e-03 @@ -1143,12 +1143,12 @@ Ionic_Step 1 Cubefile created from ABACUS. Inner loop is z, followed by y and x 1.196e-02 1.198e-02 1.200e-02 1.201e-02 1.200e-02 1.198e-02 1.196e-02 1.197e-02 1.201e-02 1.203e-02 1.201e-02 1.197e-02 1.296e-02 1.366e-02 1.435e-02 1.498e-02 1.554e-02 1.606e-02 - 1.653e-02 1.693e-02 1.717e-02 1.717e-02 1.693e-02 1.653e-02 + 1.653e-02 1.693e-02 1.716e-02 1.716e-02 1.693e-02 1.653e-02 1.606e-02 1.554e-02 1.498e-02 1.435e-02 1.366e-02 1.296e-02 1.238e-02 1.201e-02 1.185e-02 1.185e-02 1.201e-02 1.238e-02 - 1.495e-02 1.639e-02 1.784e-02 1.926e-02 2.066e-02 2.199e-02 + 1.495e-02 1.640e-02 1.784e-02 1.926e-02 2.066e-02 2.199e-02 2.313e-02 2.393e-02 2.422e-02 2.393e-02 2.313e-02 2.199e-02 - 2.066e-02 1.926e-02 1.784e-02 1.639e-02 1.495e-02 1.363e-02 + 2.066e-02 1.926e-02 1.784e-02 1.640e-02 1.495e-02 1.363e-02 1.262e-02 1.203e-02 1.185e-02 1.203e-02 1.262e-02 1.363e-02 1.781e-02 2.015e-02 2.263e-02 2.524e-02 2.787e-02 3.025e-02 3.206e-02 3.303e-02 3.303e-02 3.206e-02 3.025e-02 2.787e-02 @@ -1161,55 +1161,55 @@ Ionic_Step 1 Cubefile created from ABACUS. Inner loop is z, followed by y and x 2.464e-02 3.122e-02 3.892e-02 4.693e-02 5.398e-02 5.881e-02 6.053e-02 5.881e-02 5.398e-02 4.693e-02 3.892e-02 3.122e-02 2.464e-02 1.942e-02 1.538e-02 1.232e-02 1.016e-02 8.892e-03 - 8.479e-03 8.892e-03 1.016e-02 1.232e-02 1.538e-02 1.942e-02 - 3.122e-02 4.045e-02 5.077e-02 6.074e-02 6.863e-02 7.300e-02 - 7.300e-02 6.863e-02 6.074e-02 5.077e-02 4.045e-02 3.122e-02 + 8.478e-03 8.892e-03 1.016e-02 1.232e-02 1.538e-02 1.942e-02 + 3.122e-02 4.045e-02 5.077e-02 6.074e-02 6.864e-02 7.300e-02 + 7.300e-02 6.864e-02 6.074e-02 5.077e-02 4.045e-02 3.122e-02 2.381e-02 1.822e-02 1.411e-02 1.116e-02 9.254e-03 8.324e-03 8.324e-03 9.254e-03 1.116e-02 1.411e-02 1.822e-02 2.381e-02 - 3.892e-02 5.077e-02 6.323e-02 7.430e-02 8.198e-02 8.473e-02 + 3.892e-02 5.077e-02 6.323e-02 7.430e-02 8.198e-02 8.474e-02 8.198e-02 7.430e-02 6.323e-02 5.077e-02 3.892e-02 2.904e-02 2.153e-02 1.612e-02 1.228e-02 9.698e-03 8.199e-03 7.708e-03 8.199e-03 9.698e-03 1.228e-02 1.612e-02 2.153e-02 2.904e-02 4.693e-02 6.074e-02 7.430e-02 8.528e-02 9.146e-02 9.146e-02 8.528e-02 7.430e-02 6.074e-02 4.693e-02 3.477e-02 2.524e-02 - 1.833e-02 1.351e-02 1.022e-02 8.149e-03 7.143e-03 7.143e-03 - 8.149e-03 1.022e-02 1.351e-02 1.833e-02 2.524e-02 3.477e-02 - 5.398e-02 6.863e-02 8.198e-02 9.146e-02 9.491e-02 9.146e-02 - 8.198e-02 6.863e-02 5.398e-02 4.029e-02 2.903e-02 2.066e-02 + 1.833e-02 1.351e-02 1.022e-02 8.150e-03 7.143e-03 7.143e-03 + 8.150e-03 1.022e-02 1.351e-02 1.833e-02 2.524e-02 3.477e-02 + 5.398e-02 6.864e-02 8.198e-02 9.146e-02 9.492e-02 9.146e-02 + 8.198e-02 6.864e-02 5.398e-02 4.029e-02 2.903e-02 2.066e-02 1.480e-02 1.081e-02 8.199e-03 6.706e-03 6.220e-03 6.706e-03 8.199e-03 1.081e-02 1.480e-02 2.066e-02 2.903e-02 4.029e-02 - 5.881e-02 7.300e-02 8.473e-02 9.146e-02 9.146e-02 8.473e-02 + 5.881e-02 7.300e-02 8.474e-02 9.146e-02 9.146e-02 8.474e-02 7.300e-02 5.881e-02 4.466e-02 3.238e-02 2.286e-02 1.606e-02 1.141e-02 8.324e-03 6.418e-03 5.505e-03 5.505e-03 6.418e-03 8.324e-03 1.141e-02 1.606e-02 2.286e-02 3.238e-02 4.466e-02 6.053e-02 7.300e-02 8.198e-02 8.528e-02 8.198e-02 7.300e-02 6.053e-02 4.707e-02 3.469e-02 2.460e-02 1.714e-02 1.196e-02 - 8.479e-03 6.247e-03 4.997e-03 4.595e-03 4.997e-03 6.247e-03 - 8.479e-03 1.196e-02 1.714e-02 2.460e-02 3.469e-02 4.707e-02 - 5.881e-02 6.863e-02 7.430e-02 7.430e-02 6.863e-02 5.881e-02 + 8.478e-03 6.247e-03 4.997e-03 4.595e-03 4.997e-03 6.247e-03 + 8.478e-03 1.196e-02 1.714e-02 2.460e-02 3.469e-02 4.707e-02 + 5.881e-02 6.864e-02 7.430e-02 7.430e-02 6.864e-02 5.881e-02 4.707e-02 3.552e-02 2.558e-02 1.789e-02 1.238e-02 8.621e-03 - 6.148e-03 4.651e-03 3.948e-03 3.948e-03 4.651e-03 6.148e-03 + 6.147e-03 4.651e-03 3.948e-03 3.948e-03 4.651e-03 6.147e-03 8.621e-03 1.238e-02 1.789e-02 2.558e-02 3.552e-02 4.707e-02 5.398e-02 6.074e-02 6.323e-02 6.074e-02 5.398e-02 4.466e-02 - 3.469e-02 2.558e-02 1.816e-02 1.262e-02 8.723e-03 6.091e-03 - 4.420e-03 3.506e-03 3.218e-03 3.506e-03 4.420e-03 6.091e-03 + 3.469e-02 2.558e-02 1.816e-02 1.262e-02 8.723e-03 6.090e-03 + 4.420e-03 3.506e-03 3.218e-03 3.506e-03 4.420e-03 6.090e-03 8.723e-03 1.262e-02 1.816e-02 2.558e-02 3.469e-02 4.466e-02 4.693e-02 5.077e-02 5.077e-02 4.693e-02 4.029e-02 3.238e-02 - 2.460e-02 1.789e-02 1.262e-02 8.760e-03 6.065e-03 4.284e-03 - 3.224e-03 2.739e-03 2.739e-03 3.224e-03 4.284e-03 6.065e-03 + 2.460e-02 1.789e-02 1.262e-02 8.760e-03 6.064e-03 4.284e-03 + 3.224e-03 2.739e-03 2.739e-03 3.224e-03 4.284e-03 6.064e-03 8.760e-03 1.262e-02 1.789e-02 2.460e-02 3.238e-02 4.029e-02 3.892e-02 4.045e-02 3.892e-02 3.477e-02 2.903e-02 2.286e-02 - 1.714e-02 1.238e-02 8.723e-03 6.065e-03 4.238e-03 3.084e-03 - 2.467e-03 2.276e-03 2.467e-03 3.084e-03 4.238e-03 6.065e-03 + 1.714e-02 1.238e-02 8.723e-03 6.064e-03 4.238e-03 3.084e-03 + 2.467e-03 2.276e-03 2.467e-03 3.084e-03 4.238e-03 6.064e-03 8.723e-03 1.238e-02 1.714e-02 2.286e-02 2.903e-02 3.477e-02 3.122e-02 3.122e-02 2.904e-02 2.524e-02 2.066e-02 1.606e-02 - 1.196e-02 8.621e-03 6.091e-03 4.284e-03 3.084e-03 2.378e-03 - 2.060e-03 2.060e-03 2.378e-03 3.084e-03 4.284e-03 6.091e-03 + 1.196e-02 8.621e-03 6.090e-03 4.284e-03 3.084e-03 2.378e-03 + 2.060e-03 2.060e-03 2.378e-03 3.084e-03 4.284e-03 6.090e-03 8.621e-03 1.196e-02 1.606e-02 2.066e-02 2.524e-02 2.904e-02 2.464e-02 2.381e-02 2.153e-02 1.833e-02 1.480e-02 1.141e-02 - 8.479e-03 6.148e-03 4.420e-03 3.224e-03 2.467e-03 2.060e-03 - 1.933e-03 2.060e-03 2.467e-03 3.224e-03 4.420e-03 6.148e-03 - 8.479e-03 1.141e-02 1.480e-02 1.833e-02 2.153e-02 2.381e-02 + 8.478e-03 6.147e-03 4.420e-03 3.224e-03 2.467e-03 2.060e-03 + 1.933e-03 2.060e-03 2.467e-03 3.224e-03 4.420e-03 6.147e-03 + 8.478e-03 1.141e-02 1.480e-02 1.833e-02 2.153e-02 2.381e-02 1.942e-02 1.822e-02 1.612e-02 1.351e-02 1.081e-02 8.324e-03 6.247e-03 4.651e-03 3.506e-03 2.739e-03 2.276e-03 2.060e-03 2.060e-03 2.276e-03 2.739e-03 3.506e-03 4.651e-03 6.247e-03 @@ -1218,22 +1218,22 @@ Ionic_Step 1 Cubefile created from ABACUS. Inner loop is z, followed by y and x 4.997e-03 3.948e-03 3.218e-03 2.739e-03 2.467e-03 2.378e-03 2.467e-03 2.739e-03 3.218e-03 3.948e-03 4.997e-03 6.418e-03 8.199e-03 1.022e-02 1.228e-02 1.411e-02 1.538e-02 1.584e-02 - 1.232e-02 1.116e-02 9.698e-03 8.149e-03 6.706e-03 5.505e-03 + 1.232e-02 1.116e-02 9.698e-03 8.150e-03 6.706e-03 5.505e-03 4.595e-03 3.948e-03 3.506e-03 3.224e-03 3.084e-03 3.084e-03 3.224e-03 3.506e-03 3.948e-03 4.595e-03 5.505e-03 6.706e-03 - 8.149e-03 9.698e-03 1.116e-02 1.232e-02 1.297e-02 1.297e-02 + 8.150e-03 9.698e-03 1.116e-02 1.232e-02 1.297e-02 1.297e-02 1.016e-02 9.254e-03 8.199e-03 7.143e-03 6.220e-03 5.505e-03 4.997e-03 4.651e-03 4.420e-03 4.284e-03 4.238e-03 4.284e-03 4.420e-03 4.651e-03 4.997e-03 5.505e-03 6.220e-03 7.143e-03 8.199e-03 9.254e-03 1.016e-02 1.078e-02 1.100e-02 1.078e-02 8.892e-03 8.324e-03 7.708e-03 7.143e-03 6.706e-03 6.418e-03 - 6.247e-03 6.148e-03 6.091e-03 6.065e-03 6.065e-03 6.091e-03 - 6.148e-03 6.247e-03 6.418e-03 6.706e-03 7.143e-03 7.708e-03 - 8.324e-03 8.892e-03 9.326e-03 9.563e-03 9.563e-03 9.326e-03 - 8.479e-03 8.324e-03 8.199e-03 8.149e-03 8.199e-03 8.324e-03 - 8.479e-03 8.621e-03 8.723e-03 8.760e-03 8.723e-03 8.621e-03 - 8.479e-03 8.324e-03 8.199e-03 8.149e-03 8.199e-03 8.324e-03 - 8.479e-03 8.621e-03 8.723e-03 8.760e-03 8.723e-03 8.621e-03 + 6.247e-03 6.147e-03 6.090e-03 6.064e-03 6.064e-03 6.090e-03 + 6.147e-03 6.247e-03 6.418e-03 6.706e-03 7.143e-03 7.708e-03 + 8.324e-03 8.892e-03 9.326e-03 9.562e-03 9.562e-03 9.326e-03 + 8.478e-03 8.324e-03 8.199e-03 8.150e-03 8.199e-03 8.324e-03 + 8.478e-03 8.621e-03 8.723e-03 8.760e-03 8.723e-03 8.621e-03 + 8.478e-03 8.324e-03 8.199e-03 8.150e-03 8.199e-03 8.324e-03 + 8.478e-03 8.621e-03 8.723e-03 8.760e-03 8.723e-03 8.621e-03 8.892e-03 9.254e-03 9.698e-03 1.022e-02 1.081e-02 1.141e-02 1.196e-02 1.238e-02 1.262e-02 1.262e-02 1.238e-02 1.196e-02 1.141e-02 1.081e-02 1.022e-02 9.698e-03 9.254e-03 8.892e-03 @@ -1244,41 +1244,41 @@ Ionic_Step 1 Cubefile created from ABACUS. Inner loop is z, followed by y and x 8.723e-03 8.378e-03 8.270e-03 8.378e-03 8.723e-03 9.326e-03 1.232e-02 1.411e-02 1.612e-02 1.833e-02 2.066e-02 2.286e-02 2.460e-02 2.558e-02 2.558e-02 2.460e-02 2.286e-02 2.066e-02 - 1.833e-02 1.612e-02 1.411e-02 1.232e-02 1.078e-02 9.563e-03 - 8.760e-03 8.378e-03 8.378e-03 8.760e-03 9.563e-03 1.078e-02 + 1.833e-02 1.612e-02 1.411e-02 1.232e-02 1.078e-02 9.562e-03 + 8.760e-03 8.378e-03 8.378e-03 8.760e-03 9.562e-03 1.078e-02 1.538e-02 1.822e-02 2.153e-02 2.524e-02 2.903e-02 3.238e-02 3.469e-02 3.552e-02 3.469e-02 3.238e-02 2.903e-02 2.524e-02 - 2.153e-02 1.822e-02 1.538e-02 1.297e-02 1.100e-02 9.563e-03 - 8.723e-03 8.453e-03 8.723e-03 9.563e-03 1.100e-02 1.297e-02 + 2.153e-02 1.822e-02 1.538e-02 1.297e-02 1.100e-02 9.562e-03 + 8.723e-03 8.453e-03 8.723e-03 9.562e-03 1.100e-02 1.297e-02 1.942e-02 2.381e-02 2.904e-02 3.477e-02 4.029e-02 4.466e-02 4.707e-02 4.707e-02 4.466e-02 4.029e-02 3.477e-02 2.904e-02 2.381e-02 1.942e-02 1.584e-02 1.297e-02 1.078e-02 9.326e-03 8.621e-03 8.621e-03 9.326e-03 1.078e-02 1.297e-02 1.584e-02 - 2.589e-02 3.401e-02 4.330e-02 5.260e-02 6.025e-02 6.459e-02 - 6.459e-02 6.025e-02 5.260e-02 4.330e-02 3.401e-02 2.589e-02 + 2.589e-02 3.401e-02 4.330e-02 5.260e-02 6.025e-02 6.460e-02 + 6.460e-02 6.025e-02 5.260e-02 4.330e-02 3.401e-02 2.589e-02 1.942e-02 1.453e-02 1.096e-02 8.487e-03 6.959e-03 6.247e-03 6.247e-03 6.959e-03 8.487e-03 1.096e-02 1.453e-02 1.942e-02 3.401e-02 4.501e-02 5.700e-02 6.809e-02 7.603e-02 7.892e-02 7.603e-02 6.809e-02 5.700e-02 4.501e-02 3.401e-02 2.502e-02 1.822e-02 1.331e-02 9.883e-03 7.650e-03 6.418e-03 6.030e-03 6.418e-03 7.650e-03 9.883e-03 1.331e-02 1.822e-02 2.502e-02 - 4.330e-02 5.700e-02 7.099e-02 8.268e-02 8.941e-02 8.941e-02 - 8.268e-02 7.099e-02 5.700e-02 4.330e-02 3.163e-02 2.263e-02 + 4.330e-02 5.700e-02 7.099e-02 8.269e-02 8.941e-02 8.941e-02 + 8.269e-02 7.099e-02 5.700e-02 4.330e-02 3.163e-02 2.263e-02 1.612e-02 1.157e-02 8.541e-03 6.706e-03 5.856e-03 5.856e-03 6.706e-03 8.541e-03 1.157e-02 1.612e-02 2.263e-02 3.163e-02 - 5.260e-02 6.809e-02 8.268e-02 9.329e-02 9.719e-02 9.329e-02 - 8.268e-02 6.809e-02 5.260e-02 3.864e-02 2.748e-02 1.926e-02 + 5.260e-02 6.809e-02 8.269e-02 9.329e-02 9.719e-02 9.329e-02 + 8.269e-02 6.809e-02 5.260e-02 3.864e-02 2.748e-02 1.926e-02 1.351e-02 9.620e-03 7.143e-03 5.788e-03 5.361e-03 5.788e-03 7.143e-03 9.620e-03 1.351e-02 1.926e-02 2.748e-02 3.864e-02 6.025e-02 7.603e-02 8.941e-02 9.719e-02 9.719e-02 8.941e-02 7.603e-02 6.025e-02 4.502e-02 3.221e-02 2.247e-02 1.554e-02 1.081e-02 7.708e-03 5.856e-03 5.001e-03 5.001e-03 5.856e-03 7.708e-03 1.081e-02 1.554e-02 2.247e-02 3.221e-02 4.502e-02 - 6.459e-02 7.892e-02 8.941e-02 9.329e-02 8.941e-02 7.892e-02 - 6.459e-02 4.953e-02 3.604e-02 2.531e-02 1.744e-02 1.198e-02 + 6.460e-02 7.892e-02 8.941e-02 9.329e-02 8.941e-02 7.892e-02 + 6.460e-02 4.953e-02 3.604e-02 2.531e-02 1.744e-02 1.198e-02 8.324e-03 6.030e-03 4.789e-03 4.400e-03 4.789e-03 6.030e-03 8.324e-03 1.198e-02 1.744e-02 2.531e-02 3.604e-02 4.953e-02 - 6.459e-02 7.603e-02 8.268e-02 8.268e-02 7.603e-02 6.459e-02 + 6.460e-02 7.603e-02 8.269e-02 8.269e-02 7.603e-02 6.460e-02 5.116e-02 3.820e-02 2.728e-02 1.893e-02 1.296e-02 8.892e-03 6.247e-03 4.689e-03 3.979e-03 3.979e-03 4.689e-03 6.247e-03 8.892e-03 1.296e-02 1.893e-02 2.728e-02 3.820e-02 5.116e-02 @@ -1287,13 +1287,13 @@ Ionic_Step 1 Cubefile created from ABACUS. Inner loop is z, followed by y and x 4.651e-03 3.695e-03 3.399e-03 3.695e-03 4.651e-03 6.445e-03 9.326e-03 1.363e-02 1.975e-02 2.799e-02 3.820e-02 4.953e-02 5.260e-02 5.700e-02 5.700e-02 5.260e-02 4.502e-02 3.604e-02 - 2.728e-02 1.975e-02 1.386e-02 9.563e-03 6.580e-03 4.638e-03 + 2.728e-02 1.975e-02 1.386e-02 9.562e-03 6.580e-03 4.638e-03 3.506e-03 2.998e-03 2.998e-03 3.506e-03 4.638e-03 6.580e-03 - 9.563e-03 1.386e-02 1.975e-02 2.728e-02 3.604e-02 4.502e-02 + 9.562e-03 1.386e-02 1.975e-02 2.728e-02 3.604e-02 4.502e-02 4.330e-02 4.501e-02 4.330e-02 3.864e-02 3.221e-02 2.531e-02 - 1.893e-02 1.363e-02 9.563e-03 6.629e-03 4.635e-03 3.394e-03 - 2.739e-03 2.539e-03 2.739e-03 3.394e-03 4.635e-03 6.629e-03 - 9.563e-03 1.363e-02 1.893e-02 2.531e-02 3.221e-02 3.864e-02 + 1.893e-02 1.363e-02 9.562e-03 6.628e-03 4.635e-03 3.394e-03 + 2.739e-03 2.539e-03 2.739e-03 3.394e-03 4.635e-03 6.628e-03 + 9.562e-03 1.363e-02 1.893e-02 2.531e-02 3.221e-02 3.864e-02 3.401e-02 3.401e-02 3.163e-02 2.748e-02 2.247e-02 1.744e-02 1.296e-02 9.326e-03 6.580e-03 4.635e-03 3.356e-03 2.610e-03 2.276e-03 2.276e-03 2.610e-03 3.356e-03 4.635e-03 6.580e-03 @@ -1323,54 +1323,54 @@ Ionic_Step 1 Cubefile created from ABACUS. Inner loop is z, followed by y and x 4.651e-03 4.689e-03 4.789e-03 5.001e-03 5.361e-03 5.856e-03 6.418e-03 6.959e-03 7.387e-03 7.626e-03 7.626e-03 7.387e-03 6.247e-03 6.030e-03 5.856e-03 5.788e-03 5.856e-03 6.030e-03 - 6.247e-03 6.445e-03 6.580e-03 6.629e-03 6.580e-03 6.445e-03 + 6.247e-03 6.445e-03 6.580e-03 6.628e-03 6.580e-03 6.445e-03 6.247e-03 6.030e-03 5.856e-03 5.788e-03 5.856e-03 6.030e-03 - 6.247e-03 6.445e-03 6.580e-03 6.629e-03 6.580e-03 6.445e-03 + 6.247e-03 6.445e-03 6.580e-03 6.628e-03 6.580e-03 6.445e-03 6.247e-03 6.418e-03 6.706e-03 7.143e-03 7.708e-03 8.324e-03 - 8.892e-03 9.326e-03 9.563e-03 9.563e-03 9.326e-03 8.892e-03 + 8.892e-03 9.326e-03 9.562e-03 9.562e-03 9.326e-03 8.892e-03 8.324e-03 7.708e-03 7.143e-03 6.706e-03 6.418e-03 6.247e-03 - 6.148e-03 6.091e-03 6.065e-03 6.065e-03 6.091e-03 6.148e-03 + 6.147e-03 6.090e-03 6.064e-03 6.064e-03 6.090e-03 6.147e-03 6.959e-03 7.650e-03 8.541e-03 9.620e-03 1.081e-02 1.198e-02 1.296e-02 1.363e-02 1.386e-02 1.363e-02 1.296e-02 1.198e-02 1.081e-02 9.620e-03 8.541e-03 7.650e-03 6.959e-03 6.445e-03 - 6.091e-03 5.886e-03 5.820e-03 5.886e-03 6.091e-03 6.445e-03 + 6.090e-03 5.886e-03 5.820e-03 5.886e-03 6.090e-03 6.445e-03 8.487e-03 9.883e-03 1.157e-02 1.351e-02 1.554e-02 1.744e-02 1.893e-02 1.975e-02 1.975e-02 1.893e-02 1.744e-02 1.554e-02 1.351e-02 1.157e-02 9.883e-03 8.487e-03 7.387e-03 6.580e-03 - 6.065e-03 5.820e-03 5.820e-03 6.065e-03 6.580e-03 7.387e-03 + 6.064e-03 5.820e-03 5.820e-03 6.064e-03 6.580e-03 7.387e-03 1.096e-02 1.331e-02 1.612e-02 1.926e-02 2.247e-02 2.531e-02 2.728e-02 2.799e-02 2.728e-02 2.531e-02 2.247e-02 1.926e-02 - 1.612e-02 1.331e-02 1.096e-02 9.073e-03 7.626e-03 6.629e-03 - 6.065e-03 5.886e-03 6.065e-03 6.629e-03 7.626e-03 9.073e-03 + 1.612e-02 1.331e-02 1.096e-02 9.073e-03 7.626e-03 6.628e-03 + 6.064e-03 5.886e-03 6.064e-03 6.628e-03 7.626e-03 9.073e-03 1.453e-02 1.822e-02 2.263e-02 2.748e-02 3.221e-02 3.604e-02 3.820e-02 3.820e-02 3.604e-02 3.221e-02 2.748e-02 2.263e-02 1.822e-02 1.453e-02 1.158e-02 9.285e-03 7.626e-03 6.580e-03 - 6.091e-03 6.091e-03 6.580e-03 7.626e-03 9.285e-03 1.158e-02 + 6.090e-03 6.090e-03 6.580e-03 7.626e-03 9.285e-03 1.158e-02 1.942e-02 2.502e-02 3.163e-02 3.864e-02 4.502e-02 4.953e-02 5.116e-02 4.953e-02 4.502e-02 3.864e-02 3.163e-02 2.502e-02 1.942e-02 1.498e-02 1.158e-02 9.073e-03 7.387e-03 6.445e-03 - 6.148e-03 6.445e-03 7.387e-03 9.073e-03 1.158e-02 1.498e-02 - 2.936e-02 3.895e-02 4.939e-02 5.911e-02 6.613e-02 6.870e-02 + 6.147e-03 6.445e-03 7.387e-03 9.073e-03 1.158e-02 1.498e-02 + 2.936e-02 3.895e-02 4.939e-02 5.911e-02 6.613e-02 6.871e-02 6.613e-02 5.911e-02 4.939e-02 3.895e-02 2.936e-02 2.145e-02 1.538e-02 1.096e-02 7.921e-03 6.008e-03 4.997e-03 4.689e-03 4.997e-03 6.008e-03 7.921e-03 1.096e-02 1.538e-02 2.145e-02 - 3.895e-02 5.133e-02 6.404e-02 7.478e-02 8.099e-02 8.099e-02 - 7.478e-02 6.404e-02 5.133e-02 3.895e-02 2.838e-02 2.015e-02 + 3.895e-02 5.133e-02 6.405e-02 7.478e-02 8.100e-02 8.100e-02 + 7.478e-02 6.405e-02 5.133e-02 3.895e-02 2.839e-02 2.015e-02 1.411e-02 9.883e-03 7.115e-03 5.505e-03 4.789e-03 4.789e-03 - 5.505e-03 7.115e-03 9.883e-03 1.411e-02 2.015e-02 2.838e-02 - 4.939e-02 6.404e-02 7.798e-02 8.818e-02 9.194e-02 8.818e-02 - 7.798e-02 6.404e-02 4.939e-02 3.625e-02 2.570e-02 1.784e-02 + 5.505e-03 7.115e-03 9.883e-03 1.411e-02 2.015e-02 2.839e-02 + 4.939e-02 6.405e-02 7.798e-02 8.818e-02 9.195e-02 8.818e-02 + 7.798e-02 6.405e-02 4.939e-02 3.625e-02 2.570e-02 1.784e-02 1.228e-02 8.541e-03 6.220e-03 5.001e-03 4.629e-03 5.001e-03 6.220e-03 8.541e-03 1.228e-02 1.784e-02 2.570e-02 3.625e-02 - 5.911e-02 7.478e-02 8.818e-02 9.600e-02 9.600e-02 8.818e-02 + 5.911e-02 7.478e-02 8.818e-02 9.601e-02 9.601e-02 8.818e-02 7.478e-02 5.911e-02 4.411e-02 3.153e-02 2.191e-02 1.498e-02 1.022e-02 7.143e-03 5.361e-03 4.569e-03 4.569e-03 5.361e-03 7.143e-03 1.022e-02 1.498e-02 2.191e-02 3.153e-02 4.411e-02 - 6.613e-02 8.099e-02 9.194e-02 9.600e-02 9.194e-02 8.099e-02 - 6.613e-02 5.061e-02 3.681e-02 2.582e-02 1.770e-02 1.200e-02 + 6.613e-02 8.100e-02 9.195e-02 9.601e-02 9.195e-02 8.100e-02 + 6.613e-02 5.062e-02 3.681e-02 2.582e-02 1.770e-02 1.200e-02 8.199e-03 5.856e-03 4.629e-03 4.255e-03 4.629e-03 5.856e-03 - 8.199e-03 1.200e-02 1.770e-02 2.582e-02 3.681e-02 5.061e-02 - 6.870e-02 8.099e-02 8.818e-02 8.818e-02 8.099e-02 6.870e-02 + 8.199e-03 1.200e-02 1.770e-02 2.582e-02 3.681e-02 5.062e-02 + 6.871e-02 8.100e-02 8.818e-02 8.818e-02 8.100e-02 6.871e-02 5.434e-02 4.056e-02 2.897e-02 2.007e-02 1.366e-02 9.254e-03 6.418e-03 4.789e-03 4.068e-03 4.068e-03 4.789e-03 6.418e-03 9.254e-03 1.366e-02 2.007e-02 2.897e-02 4.056e-02 5.434e-02 @@ -1378,25 +1378,25 @@ Ionic_Step 1 Cubefile created from ABACUS. Inner loop is z, followed by y and x 4.192e-02 3.074e-02 2.171e-02 1.495e-02 1.016e-02 6.959e-03 4.997e-03 3.979e-03 3.671e-03 3.979e-03 4.997e-03 6.959e-03 1.016e-02 1.495e-02 2.171e-02 3.074e-02 4.192e-02 5.434e-02 - 5.911e-02 6.404e-02 6.404e-02 5.911e-02 5.061e-02 4.056e-02 + 5.911e-02 6.405e-02 6.405e-02 5.911e-02 5.062e-02 4.056e-02 3.074e-02 2.230e-02 1.566e-02 1.078e-02 7.387e-03 5.200e-03 3.948e-03 3.399e-03 3.399e-03 3.948e-03 5.200e-03 7.387e-03 - 1.078e-02 1.566e-02 2.230e-02 3.074e-02 4.056e-02 5.061e-02 + 1.078e-02 1.566e-02 2.230e-02 3.074e-02 4.056e-02 5.062e-02 4.939e-02 5.133e-02 4.939e-02 4.411e-02 3.681e-02 2.897e-02 - 2.171e-02 1.566e-02 1.100e-02 7.626e-03 5.346e-03 3.945e-03 - 3.218e-03 2.998e-03 3.218e-03 3.945e-03 5.346e-03 7.626e-03 + 2.171e-02 1.566e-02 1.100e-02 7.626e-03 5.346e-03 3.944e-03 + 3.218e-03 2.998e-03 3.218e-03 3.944e-03 5.346e-03 7.626e-03 1.100e-02 1.566e-02 2.171e-02 2.897e-02 3.681e-02 4.411e-02 3.895e-02 3.895e-02 3.625e-02 3.153e-02 2.582e-02 2.007e-02 1.495e-02 1.078e-02 7.626e-03 5.400e-03 3.948e-03 3.110e-03 2.739e-03 2.739e-03 3.110e-03 3.948e-03 5.400e-03 7.626e-03 1.078e-02 1.495e-02 2.007e-02 2.582e-02 3.153e-02 3.625e-02 - 2.936e-02 2.838e-02 2.570e-02 2.191e-02 1.770e-02 1.366e-02 + 2.936e-02 2.839e-02 2.570e-02 2.191e-02 1.770e-02 1.366e-02 1.016e-02 7.387e-03 5.346e-03 3.948e-03 3.074e-03 2.610e-03 2.467e-03 2.610e-03 3.074e-03 3.948e-03 5.346e-03 7.387e-03 - 1.016e-02 1.366e-02 1.770e-02 2.191e-02 2.570e-02 2.838e-02 + 1.016e-02 1.366e-02 1.770e-02 2.191e-02 2.570e-02 2.839e-02 2.145e-02 2.015e-02 1.784e-02 1.498e-02 1.200e-02 9.254e-03 - 6.959e-03 5.200e-03 3.945e-03 3.110e-03 2.610e-03 2.378e-03 - 2.378e-03 2.610e-03 3.110e-03 3.945e-03 5.200e-03 6.959e-03 + 6.959e-03 5.200e-03 3.944e-03 3.110e-03 2.610e-03 2.378e-03 + 2.378e-03 2.610e-03 3.110e-03 3.944e-03 5.200e-03 6.959e-03 9.254e-03 1.200e-02 1.498e-02 1.784e-02 2.015e-02 2.145e-02 1.538e-02 1.411e-02 1.228e-02 1.022e-02 8.199e-03 6.418e-03 4.997e-03 3.948e-03 3.218e-03 2.739e-03 2.467e-03 2.378e-03 @@ -1409,9 +1409,9 @@ Ionic_Step 1 Cubefile created from ABACUS. Inner loop is z, followed by y and x 7.921e-03 7.115e-03 6.220e-03 5.361e-03 4.629e-03 4.068e-03 3.671e-03 3.399e-03 3.218e-03 3.110e-03 3.074e-03 3.110e-03 3.218e-03 3.399e-03 3.671e-03 4.068e-03 4.629e-03 5.361e-03 - 6.220e-03 7.115e-03 7.921e-03 8.494e-03 8.703e-03 8.494e-03 + 6.220e-03 7.115e-03 7.921e-03 8.494e-03 8.704e-03 8.494e-03 6.008e-03 5.505e-03 5.001e-03 4.569e-03 4.255e-03 4.068e-03 - 3.979e-03 3.948e-03 3.945e-03 3.948e-03 3.948e-03 3.945e-03 + 3.979e-03 3.948e-03 3.944e-03 3.948e-03 3.948e-03 3.944e-03 3.948e-03 3.979e-03 4.068e-03 4.255e-03 4.569e-03 5.001e-03 5.505e-03 6.008e-03 6.424e-03 6.662e-03 6.662e-03 6.424e-03 4.997e-03 4.789e-03 4.629e-03 4.569e-03 4.629e-03 4.789e-03 @@ -1440,143 +1440,143 @@ Ionic_Step 1 Cubefile created from ABACUS. Inner loop is z, followed by y and x 4.284e-03 4.284e-03 4.635e-03 5.400e-03 6.662e-03 8.494e-03 1.538e-02 2.015e-02 2.570e-02 3.153e-02 3.681e-02 4.056e-02 4.192e-02 4.056e-02 3.681e-02 3.153e-02 2.570e-02 2.015e-02 - 1.538e-02 1.158e-02 8.703e-03 6.662e-03 5.346e-03 4.638e-03 - 4.420e-03 4.638e-03 5.346e-03 6.662e-03 8.703e-03 1.158e-02 - 2.145e-02 2.838e-02 3.625e-02 4.411e-02 5.061e-02 5.434e-02 - 5.434e-02 5.061e-02 4.411e-02 3.625e-02 2.838e-02 2.145e-02 + 1.538e-02 1.158e-02 8.704e-03 6.662e-03 5.346e-03 4.638e-03 + 4.420e-03 4.638e-03 5.346e-03 6.662e-03 8.704e-03 1.158e-02 + 2.145e-02 2.839e-02 3.625e-02 4.411e-02 5.062e-02 5.434e-02 + 5.434e-02 5.062e-02 4.411e-02 3.625e-02 2.839e-02 2.145e-02 1.584e-02 1.158e-02 8.494e-03 6.424e-03 5.200e-03 4.651e-03 4.651e-03 5.200e-03 6.424e-03 8.494e-03 1.158e-02 1.584e-02 - 3.468e-02 4.565e-02 5.689e-02 6.639e-02 7.190e-02 7.190e-02 - 6.639e-02 5.689e-02 4.565e-02 3.468e-02 2.524e-02 1.781e-02 + 3.468e-02 4.565e-02 5.690e-02 6.639e-02 7.190e-02 7.190e-02 + 6.639e-02 5.690e-02 4.565e-02 3.468e-02 2.524e-02 1.781e-02 1.232e-02 8.487e-03 6.008e-03 4.595e-03 3.979e-03 3.979e-03 4.595e-03 6.008e-03 8.487e-03 1.232e-02 1.781e-02 2.524e-02 - 4.565e-02 5.911e-02 7.190e-02 8.128e-02 8.475e-02 8.128e-02 - 7.190e-02 5.911e-02 4.565e-02 3.354e-02 2.375e-02 1.639e-02 + 4.565e-02 5.911e-02 7.190e-02 8.129e-02 8.476e-02 8.129e-02 + 7.190e-02 5.911e-02 4.565e-02 3.355e-02 2.375e-02 1.640e-02 1.116e-02 7.650e-03 5.505e-03 4.400e-03 4.068e-03 4.400e-03 - 5.505e-03 7.650e-03 1.116e-02 1.639e-02 2.375e-02 3.354e-02 - 5.689e-02 7.190e-02 8.475e-02 9.227e-02 9.227e-02 8.475e-02 - 7.190e-02 5.689e-02 4.252e-02 3.042e-02 2.110e-02 1.435e-02 + 5.505e-03 7.650e-03 1.116e-02 1.640e-02 2.375e-02 3.355e-02 + 5.690e-02 7.190e-02 8.476e-02 9.227e-02 9.227e-02 8.476e-02 + 7.190e-02 5.690e-02 4.252e-02 3.042e-02 2.110e-02 1.435e-02 9.698e-03 6.706e-03 5.001e-03 4.255e-03 4.255e-03 5.001e-03 6.706e-03 9.698e-03 1.435e-02 2.110e-02 3.042e-02 4.252e-02 - 6.639e-02 8.128e-02 9.227e-02 9.634e-02 9.227e-02 8.128e-02 + 6.639e-02 8.129e-02 9.227e-02 9.635e-02 9.227e-02 8.129e-02 6.639e-02 5.086e-02 3.705e-02 2.600e-02 1.780e-02 1.201e-02 - 8.149e-03 5.788e-03 4.569e-03 4.201e-03 4.569e-03 5.788e-03 - 8.149e-03 1.201e-02 1.780e-02 2.600e-02 3.705e-02 5.086e-02 - 7.190e-02 8.475e-02 9.227e-02 9.227e-02 8.475e-02 7.190e-02 - 5.689e-02 4.252e-02 3.042e-02 2.110e-02 1.435e-02 9.698e-03 + 8.150e-03 5.788e-03 4.569e-03 4.201e-03 4.569e-03 5.788e-03 + 8.150e-03 1.201e-02 1.780e-02 2.600e-02 3.705e-02 5.086e-02 + 7.190e-02 8.476e-02 9.227e-02 9.227e-02 8.476e-02 7.190e-02 + 5.690e-02 4.252e-02 3.042e-02 2.110e-02 1.435e-02 9.698e-03 6.706e-03 5.001e-03 4.255e-03 4.255e-03 5.001e-03 6.706e-03 - 9.698e-03 1.435e-02 2.110e-02 3.042e-02 4.252e-02 5.689e-02 - 7.190e-02 8.128e-02 8.475e-02 8.128e-02 7.190e-02 5.911e-02 - 4.565e-02 3.354e-02 2.375e-02 1.639e-02 1.116e-02 7.650e-03 + 9.698e-03 1.435e-02 2.110e-02 3.042e-02 4.252e-02 5.690e-02 + 7.190e-02 8.129e-02 8.476e-02 8.129e-02 7.190e-02 5.911e-02 + 4.565e-02 3.355e-02 2.375e-02 1.640e-02 1.116e-02 7.650e-03 5.505e-03 4.400e-03 4.068e-03 4.400e-03 5.505e-03 7.650e-03 - 1.116e-02 1.639e-02 2.375e-02 3.354e-02 4.565e-02 5.911e-02 - 6.639e-02 7.190e-02 7.190e-02 6.639e-02 5.689e-02 4.565e-02 + 1.116e-02 1.640e-02 2.375e-02 3.355e-02 4.565e-02 5.911e-02 + 6.639e-02 7.190e-02 7.190e-02 6.639e-02 5.690e-02 4.565e-02 3.468e-02 2.524e-02 1.781e-02 1.232e-02 8.487e-03 6.008e-03 4.595e-03 3.979e-03 3.979e-03 4.595e-03 6.008e-03 8.487e-03 - 1.232e-02 1.781e-02 2.524e-02 3.468e-02 4.565e-02 5.689e-02 - 5.689e-02 5.911e-02 5.689e-02 5.086e-02 4.252e-02 3.354e-02 + 1.232e-02 1.781e-02 2.524e-02 3.468e-02 4.565e-02 5.690e-02 + 5.690e-02 5.911e-02 5.690e-02 5.086e-02 4.252e-02 3.355e-02 2.524e-02 1.832e-02 1.297e-02 9.073e-03 6.424e-03 4.792e-03 3.948e-03 3.695e-03 3.948e-03 4.792e-03 6.424e-03 9.073e-03 - 1.297e-02 1.832e-02 2.524e-02 3.354e-02 4.252e-02 5.086e-02 + 1.297e-02 1.832e-02 2.524e-02 3.355e-02 4.252e-02 5.086e-02 4.565e-02 4.565e-02 4.252e-02 3.705e-02 3.042e-02 2.375e-02 - 1.781e-02 1.297e-02 9.285e-03 6.662e-03 4.940e-03 3.945e-03 - 3.506e-03 3.506e-03 3.945e-03 4.940e-03 6.662e-03 9.285e-03 + 1.781e-02 1.297e-02 9.285e-03 6.662e-03 4.939e-03 3.944e-03 + 3.506e-03 3.506e-03 3.944e-03 4.939e-03 6.662e-03 9.285e-03 1.297e-02 1.781e-02 2.375e-02 3.042e-02 3.705e-02 4.252e-02 - 3.468e-02 3.354e-02 3.042e-02 2.600e-02 2.110e-02 1.639e-02 + 3.468e-02 3.355e-02 3.042e-02 2.600e-02 2.110e-02 1.640e-02 1.232e-02 9.073e-03 6.662e-03 4.995e-03 3.948e-03 3.394e-03 3.224e-03 3.394e-03 3.948e-03 4.995e-03 6.662e-03 9.073e-03 - 1.232e-02 1.639e-02 2.110e-02 2.600e-02 3.042e-02 3.354e-02 + 1.232e-02 1.640e-02 2.110e-02 2.600e-02 3.042e-02 3.355e-02 2.524e-02 2.375e-02 2.110e-02 1.780e-02 1.435e-02 1.116e-02 - 8.487e-03 6.424e-03 4.940e-03 3.948e-03 3.356e-03 3.084e-03 - 3.084e-03 3.356e-03 3.948e-03 4.940e-03 6.424e-03 8.487e-03 + 8.487e-03 6.424e-03 4.939e-03 3.948e-03 3.356e-03 3.084e-03 + 3.084e-03 3.356e-03 3.948e-03 4.939e-03 6.424e-03 8.487e-03 1.116e-02 1.435e-02 1.780e-02 2.110e-02 2.375e-02 2.524e-02 - 1.781e-02 1.639e-02 1.435e-02 1.201e-02 9.698e-03 7.650e-03 - 6.008e-03 4.792e-03 3.945e-03 3.394e-03 3.084e-03 2.984e-03 - 3.084e-03 3.394e-03 3.945e-03 4.792e-03 6.008e-03 7.650e-03 - 9.698e-03 1.201e-02 1.435e-02 1.639e-02 1.781e-02 1.832e-02 - 1.232e-02 1.116e-02 9.698e-03 8.149e-03 6.706e-03 5.505e-03 + 1.781e-02 1.640e-02 1.435e-02 1.201e-02 9.698e-03 7.650e-03 + 6.008e-03 4.792e-03 3.944e-03 3.394e-03 3.084e-03 2.984e-03 + 3.084e-03 3.394e-03 3.944e-03 4.792e-03 6.008e-03 7.650e-03 + 9.698e-03 1.201e-02 1.435e-02 1.640e-02 1.781e-02 1.832e-02 + 1.232e-02 1.116e-02 9.698e-03 8.150e-03 6.706e-03 5.505e-03 4.595e-03 3.948e-03 3.506e-03 3.224e-03 3.084e-03 3.084e-03 3.224e-03 3.506e-03 3.948e-03 4.595e-03 5.505e-03 6.706e-03 - 8.149e-03 9.698e-03 1.116e-02 1.232e-02 1.297e-02 1.297e-02 + 8.150e-03 9.698e-03 1.116e-02 1.232e-02 1.297e-02 1.297e-02 8.487e-03 7.650e-03 6.706e-03 5.788e-03 5.001e-03 4.400e-03 3.979e-03 3.695e-03 3.506e-03 3.394e-03 3.356e-03 3.394e-03 3.506e-03 3.695e-03 3.979e-03 4.400e-03 5.001e-03 5.788e-03 6.706e-03 7.650e-03 8.487e-03 9.073e-03 9.285e-03 9.073e-03 6.008e-03 5.505e-03 5.001e-03 4.569e-03 4.255e-03 4.068e-03 - 3.979e-03 3.948e-03 3.945e-03 3.948e-03 3.948e-03 3.945e-03 + 3.979e-03 3.948e-03 3.944e-03 3.948e-03 3.948e-03 3.944e-03 3.948e-03 3.979e-03 4.068e-03 4.255e-03 4.569e-03 5.001e-03 5.505e-03 6.008e-03 6.424e-03 6.662e-03 6.662e-03 6.424e-03 4.595e-03 4.400e-03 4.255e-03 4.201e-03 4.255e-03 4.400e-03 - 4.595e-03 4.792e-03 4.940e-03 4.995e-03 4.940e-03 4.792e-03 + 4.595e-03 4.792e-03 4.939e-03 4.995e-03 4.939e-03 4.792e-03 4.595e-03 4.400e-03 4.255e-03 4.201e-03 4.255e-03 4.400e-03 - 4.595e-03 4.792e-03 4.940e-03 4.995e-03 4.940e-03 4.792e-03 + 4.595e-03 4.792e-03 4.939e-03 4.995e-03 4.939e-03 4.792e-03 3.979e-03 4.068e-03 4.255e-03 4.569e-03 5.001e-03 5.505e-03 6.008e-03 6.424e-03 6.662e-03 6.662e-03 6.424e-03 6.008e-03 5.505e-03 5.001e-03 4.569e-03 4.255e-03 4.068e-03 3.979e-03 - 3.948e-03 3.945e-03 3.948e-03 3.948e-03 3.945e-03 3.948e-03 + 3.948e-03 3.944e-03 3.948e-03 3.948e-03 3.944e-03 3.948e-03 3.979e-03 4.400e-03 5.001e-03 5.788e-03 6.706e-03 7.650e-03 8.487e-03 9.073e-03 9.285e-03 9.073e-03 8.487e-03 7.650e-03 6.706e-03 5.788e-03 5.001e-03 4.400e-03 3.979e-03 3.695e-03 3.506e-03 3.394e-03 3.356e-03 3.394e-03 3.506e-03 3.695e-03 - 4.595e-03 5.505e-03 6.706e-03 8.149e-03 9.698e-03 1.116e-02 + 4.595e-03 5.505e-03 6.706e-03 8.150e-03 9.698e-03 1.116e-02 1.232e-02 1.297e-02 1.297e-02 1.232e-02 1.116e-02 9.698e-03 - 8.149e-03 6.706e-03 5.505e-03 4.595e-03 3.948e-03 3.506e-03 + 8.150e-03 6.706e-03 5.505e-03 4.595e-03 3.948e-03 3.506e-03 3.224e-03 3.084e-03 3.084e-03 3.224e-03 3.506e-03 3.948e-03 - 6.008e-03 7.650e-03 9.698e-03 1.201e-02 1.435e-02 1.639e-02 - 1.781e-02 1.832e-02 1.781e-02 1.639e-02 1.435e-02 1.201e-02 - 9.698e-03 7.650e-03 6.008e-03 4.792e-03 3.945e-03 3.394e-03 - 3.084e-03 2.984e-03 3.084e-03 3.394e-03 3.945e-03 4.792e-03 + 6.008e-03 7.650e-03 9.698e-03 1.201e-02 1.435e-02 1.640e-02 + 1.781e-02 1.832e-02 1.781e-02 1.640e-02 1.435e-02 1.201e-02 + 9.698e-03 7.650e-03 6.008e-03 4.792e-03 3.944e-03 3.394e-03 + 3.084e-03 2.984e-03 3.084e-03 3.394e-03 3.944e-03 4.792e-03 8.487e-03 1.116e-02 1.435e-02 1.780e-02 2.110e-02 2.375e-02 2.524e-02 2.524e-02 2.375e-02 2.110e-02 1.780e-02 1.435e-02 - 1.116e-02 8.487e-03 6.424e-03 4.940e-03 3.948e-03 3.356e-03 - 3.084e-03 3.084e-03 3.356e-03 3.948e-03 4.940e-03 6.424e-03 - 1.232e-02 1.639e-02 2.110e-02 2.600e-02 3.042e-02 3.354e-02 - 3.468e-02 3.354e-02 3.042e-02 2.600e-02 2.110e-02 1.639e-02 + 1.116e-02 8.487e-03 6.424e-03 4.939e-03 3.948e-03 3.356e-03 + 3.084e-03 3.084e-03 3.356e-03 3.948e-03 4.939e-03 6.424e-03 + 1.232e-02 1.640e-02 2.110e-02 2.600e-02 3.042e-02 3.355e-02 + 3.468e-02 3.355e-02 3.042e-02 2.600e-02 2.110e-02 1.640e-02 1.232e-02 9.073e-03 6.662e-03 4.995e-03 3.948e-03 3.394e-03 3.224e-03 3.394e-03 3.948e-03 4.995e-03 6.662e-03 9.073e-03 1.781e-02 2.375e-02 3.042e-02 3.705e-02 4.252e-02 4.565e-02 4.565e-02 4.252e-02 3.705e-02 3.042e-02 2.375e-02 1.781e-02 - 1.297e-02 9.285e-03 6.662e-03 4.940e-03 3.945e-03 3.506e-03 - 3.506e-03 3.945e-03 4.940e-03 6.662e-03 9.285e-03 1.297e-02 - 2.524e-02 3.354e-02 4.252e-02 5.086e-02 5.689e-02 5.911e-02 - 5.689e-02 5.086e-02 4.252e-02 3.354e-02 2.524e-02 1.832e-02 + 1.297e-02 9.285e-03 6.662e-03 4.939e-03 3.944e-03 3.506e-03 + 3.506e-03 3.944e-03 4.939e-03 6.662e-03 9.285e-03 1.297e-02 + 2.524e-02 3.355e-02 4.252e-02 5.086e-02 5.690e-02 5.911e-02 + 5.690e-02 5.086e-02 4.252e-02 3.355e-02 2.524e-02 1.832e-02 1.297e-02 9.073e-03 6.424e-03 4.792e-03 3.948e-03 3.695e-03 3.948e-03 4.792e-03 6.424e-03 9.073e-03 1.297e-02 1.832e-02 4.192e-02 5.434e-02 6.613e-02 7.478e-02 7.798e-02 7.478e-02 6.613e-02 5.434e-02 4.192e-02 3.074e-02 2.171e-02 1.495e-02 1.016e-02 6.959e-03 4.997e-03 3.979e-03 3.671e-03 3.979e-03 4.997e-03 6.959e-03 1.016e-02 1.495e-02 2.171e-02 3.074e-02 - 5.434e-02 6.870e-02 8.099e-02 8.818e-02 8.818e-02 8.099e-02 - 6.870e-02 5.434e-02 4.056e-02 2.897e-02 2.007e-02 1.366e-02 + 5.434e-02 6.871e-02 8.100e-02 8.818e-02 8.818e-02 8.100e-02 + 6.871e-02 5.434e-02 4.056e-02 2.897e-02 2.007e-02 1.366e-02 9.254e-03 6.418e-03 4.789e-03 4.068e-03 4.068e-03 4.789e-03 6.418e-03 9.254e-03 1.366e-02 2.007e-02 2.897e-02 4.056e-02 - 6.613e-02 8.099e-02 9.194e-02 9.600e-02 9.194e-02 8.099e-02 - 6.613e-02 5.061e-02 3.681e-02 2.582e-02 1.770e-02 1.200e-02 + 6.613e-02 8.100e-02 9.195e-02 9.601e-02 9.195e-02 8.100e-02 + 6.613e-02 5.062e-02 3.681e-02 2.582e-02 1.770e-02 1.200e-02 8.199e-03 5.856e-03 4.629e-03 4.255e-03 4.629e-03 5.856e-03 - 8.199e-03 1.200e-02 1.770e-02 2.582e-02 3.681e-02 5.061e-02 - 7.478e-02 8.818e-02 9.600e-02 9.600e-02 8.818e-02 7.478e-02 + 8.199e-03 1.200e-02 1.770e-02 2.582e-02 3.681e-02 5.062e-02 + 7.478e-02 8.818e-02 9.601e-02 9.601e-02 8.818e-02 7.478e-02 5.911e-02 4.411e-02 3.153e-02 2.191e-02 1.498e-02 1.022e-02 7.143e-03 5.361e-03 4.569e-03 4.569e-03 5.361e-03 7.143e-03 1.022e-02 1.498e-02 2.191e-02 3.153e-02 4.411e-02 5.911e-02 - 7.798e-02 8.818e-02 9.194e-02 8.818e-02 7.798e-02 6.404e-02 + 7.798e-02 8.818e-02 9.195e-02 8.818e-02 7.798e-02 6.405e-02 4.939e-02 3.625e-02 2.570e-02 1.784e-02 1.228e-02 8.541e-03 6.220e-03 5.001e-03 4.629e-03 5.001e-03 6.220e-03 8.541e-03 - 1.228e-02 1.784e-02 2.570e-02 3.625e-02 4.939e-02 6.404e-02 - 7.478e-02 8.099e-02 8.099e-02 7.478e-02 6.404e-02 5.133e-02 - 3.895e-02 2.838e-02 2.015e-02 1.411e-02 9.883e-03 7.115e-03 + 1.228e-02 1.784e-02 2.570e-02 3.625e-02 4.939e-02 6.405e-02 + 7.478e-02 8.100e-02 8.100e-02 7.478e-02 6.405e-02 5.133e-02 + 3.895e-02 2.839e-02 2.015e-02 1.411e-02 9.883e-03 7.115e-03 5.505e-03 4.789e-03 4.789e-03 5.505e-03 7.115e-03 9.883e-03 - 1.411e-02 2.015e-02 2.838e-02 3.895e-02 5.133e-02 6.404e-02 - 6.613e-02 6.870e-02 6.613e-02 5.911e-02 4.939e-02 3.895e-02 + 1.411e-02 2.015e-02 2.839e-02 3.895e-02 5.133e-02 6.405e-02 + 6.613e-02 6.871e-02 6.613e-02 5.911e-02 4.939e-02 3.895e-02 2.936e-02 2.145e-02 1.538e-02 1.096e-02 7.921e-03 6.008e-03 4.997e-03 4.689e-03 4.997e-03 6.008e-03 7.921e-03 1.096e-02 1.538e-02 2.145e-02 2.936e-02 3.895e-02 4.939e-02 5.911e-02 - 5.434e-02 5.434e-02 5.061e-02 4.411e-02 3.625e-02 2.838e-02 + 5.434e-02 5.434e-02 5.062e-02 4.411e-02 3.625e-02 2.839e-02 2.145e-02 1.584e-02 1.158e-02 8.494e-03 6.424e-03 5.200e-03 4.651e-03 4.651e-03 5.200e-03 6.424e-03 8.494e-03 1.158e-02 - 1.584e-02 2.145e-02 2.838e-02 3.625e-02 4.411e-02 5.061e-02 + 1.584e-02 2.145e-02 2.839e-02 3.625e-02 4.411e-02 5.062e-02 4.192e-02 4.056e-02 3.681e-02 3.153e-02 2.570e-02 2.015e-02 - 1.538e-02 1.158e-02 8.703e-03 6.662e-03 5.346e-03 4.638e-03 - 4.420e-03 4.638e-03 5.346e-03 6.662e-03 8.703e-03 1.158e-02 + 1.538e-02 1.158e-02 8.704e-03 6.662e-03 5.346e-03 4.638e-03 + 4.420e-03 4.638e-03 5.346e-03 6.662e-03 8.704e-03 1.158e-02 1.538e-02 2.015e-02 2.570e-02 3.153e-02 3.681e-02 4.056e-02 3.074e-02 2.897e-02 2.582e-02 2.191e-02 1.784e-02 1.411e-02 1.096e-02 8.494e-03 6.662e-03 5.400e-03 4.635e-03 4.284e-03 @@ -1605,9 +1605,9 @@ Ionic_Step 1 Cubefile created from ABACUS. Inner loop is z, followed by y and x 3.979e-03 4.068e-03 4.255e-03 4.569e-03 5.001e-03 5.505e-03 6.008e-03 6.424e-03 6.662e-03 6.662e-03 6.424e-03 6.008e-03 5.505e-03 5.001e-03 4.569e-03 4.255e-03 4.068e-03 3.979e-03 - 3.948e-03 3.945e-03 3.948e-03 3.948e-03 3.945e-03 3.948e-03 + 3.948e-03 3.944e-03 3.948e-03 3.948e-03 3.944e-03 3.948e-03 3.671e-03 4.068e-03 4.629e-03 5.361e-03 6.220e-03 7.115e-03 - 7.921e-03 8.494e-03 8.703e-03 8.494e-03 7.921e-03 7.115e-03 + 7.921e-03 8.494e-03 8.704e-03 8.494e-03 7.921e-03 7.115e-03 6.220e-03 5.361e-03 4.629e-03 4.068e-03 3.671e-03 3.399e-03 3.218e-03 3.110e-03 3.074e-03 3.110e-03 3.218e-03 3.399e-03 3.979e-03 4.789e-03 5.856e-03 7.143e-03 8.541e-03 9.883e-03 @@ -1620,10 +1620,10 @@ Ionic_Step 1 Cubefile created from ABACUS. Inner loop is z, followed by y and x 2.467e-03 2.378e-03 2.467e-03 2.739e-03 3.218e-03 3.948e-03 6.959e-03 9.254e-03 1.200e-02 1.498e-02 1.784e-02 2.015e-02 2.145e-02 2.145e-02 2.015e-02 1.784e-02 1.498e-02 1.200e-02 - 9.254e-03 6.959e-03 5.200e-03 3.945e-03 3.110e-03 2.610e-03 - 2.378e-03 2.378e-03 2.610e-03 3.110e-03 3.945e-03 5.200e-03 - 1.016e-02 1.366e-02 1.770e-02 2.191e-02 2.570e-02 2.838e-02 - 2.936e-02 2.838e-02 2.570e-02 2.191e-02 1.770e-02 1.366e-02 + 9.254e-03 6.959e-03 5.200e-03 3.944e-03 3.110e-03 2.610e-03 + 2.378e-03 2.378e-03 2.610e-03 3.110e-03 3.944e-03 5.200e-03 + 1.016e-02 1.366e-02 1.770e-02 2.191e-02 2.570e-02 2.839e-02 + 2.936e-02 2.839e-02 2.570e-02 2.191e-02 1.770e-02 1.366e-02 1.016e-02 7.387e-03 5.346e-03 3.948e-03 3.074e-03 2.610e-03 2.467e-03 2.610e-03 3.074e-03 3.948e-03 5.346e-03 7.387e-03 1.495e-02 2.007e-02 2.582e-02 3.153e-02 3.625e-02 3.895e-02 @@ -1632,29 +1632,29 @@ Ionic_Step 1 Cubefile created from ABACUS. Inner loop is z, followed by y and x 2.739e-03 3.110e-03 3.948e-03 5.400e-03 7.626e-03 1.078e-02 2.171e-02 2.897e-02 3.681e-02 4.411e-02 4.939e-02 5.133e-02 4.939e-02 4.411e-02 3.681e-02 2.897e-02 2.171e-02 1.566e-02 - 1.100e-02 7.626e-03 5.346e-03 3.945e-03 3.218e-03 2.998e-03 - 3.218e-03 3.945e-03 5.346e-03 7.626e-03 1.100e-02 1.566e-02 - 3.074e-02 4.056e-02 5.061e-02 5.911e-02 6.404e-02 6.404e-02 - 5.911e-02 5.061e-02 4.056e-02 3.074e-02 2.230e-02 1.566e-02 + 1.100e-02 7.626e-03 5.346e-03 3.944e-03 3.218e-03 2.998e-03 + 3.218e-03 3.944e-03 5.346e-03 7.626e-03 1.100e-02 1.566e-02 + 3.074e-02 4.056e-02 5.062e-02 5.911e-02 6.405e-02 6.405e-02 + 5.911e-02 5.062e-02 4.056e-02 3.074e-02 2.230e-02 1.566e-02 1.078e-02 7.387e-03 5.200e-03 3.948e-03 3.399e-03 3.399e-03 3.948e-03 5.200e-03 7.387e-03 1.078e-02 1.566e-02 2.230e-02 - 5.116e-02 6.459e-02 7.603e-02 8.268e-02 8.268e-02 7.603e-02 - 6.459e-02 5.116e-02 3.820e-02 2.728e-02 1.893e-02 1.296e-02 + 5.116e-02 6.460e-02 7.603e-02 8.269e-02 8.269e-02 7.603e-02 + 6.460e-02 5.116e-02 3.820e-02 2.728e-02 1.893e-02 1.296e-02 8.892e-03 6.247e-03 4.689e-03 3.979e-03 3.979e-03 4.689e-03 6.247e-03 8.892e-03 1.296e-02 1.893e-02 2.728e-02 3.820e-02 - 6.459e-02 7.892e-02 8.941e-02 9.329e-02 8.941e-02 7.892e-02 - 6.459e-02 4.953e-02 3.604e-02 2.531e-02 1.744e-02 1.198e-02 + 6.460e-02 7.892e-02 8.941e-02 9.329e-02 8.941e-02 7.892e-02 + 6.460e-02 4.953e-02 3.604e-02 2.531e-02 1.744e-02 1.198e-02 8.324e-03 6.030e-03 4.789e-03 4.400e-03 4.789e-03 6.030e-03 8.324e-03 1.198e-02 1.744e-02 2.531e-02 3.604e-02 4.953e-02 7.603e-02 8.941e-02 9.719e-02 9.719e-02 8.941e-02 7.603e-02 6.025e-02 4.502e-02 3.221e-02 2.247e-02 1.554e-02 1.081e-02 7.708e-03 5.856e-03 5.001e-03 5.001e-03 5.856e-03 7.708e-03 1.081e-02 1.554e-02 2.247e-02 3.221e-02 4.502e-02 6.025e-02 - 8.268e-02 9.329e-02 9.719e-02 9.329e-02 8.268e-02 6.809e-02 + 8.269e-02 9.329e-02 9.719e-02 9.329e-02 8.269e-02 6.809e-02 5.260e-02 3.864e-02 2.748e-02 1.926e-02 1.351e-02 9.620e-03 7.143e-03 5.788e-03 5.361e-03 5.788e-03 7.143e-03 9.620e-03 1.351e-02 1.926e-02 2.748e-02 3.864e-02 5.260e-02 6.809e-02 - 8.268e-02 8.941e-02 8.941e-02 8.268e-02 7.099e-02 5.700e-02 + 8.269e-02 8.941e-02 8.941e-02 8.269e-02 7.099e-02 5.700e-02 4.330e-02 3.163e-02 2.263e-02 1.612e-02 1.157e-02 8.541e-03 6.706e-03 5.856e-03 5.856e-03 6.706e-03 8.541e-03 1.157e-02 1.612e-02 2.263e-02 3.163e-02 4.330e-02 5.700e-02 7.099e-02 @@ -1662,38 +1662,38 @@ Ionic_Step 1 Cubefile created from ABACUS. Inner loop is z, followed by y and x 3.401e-02 2.502e-02 1.822e-02 1.331e-02 9.883e-03 7.650e-03 6.418e-03 6.030e-03 6.418e-03 7.650e-03 9.883e-03 1.331e-02 1.822e-02 2.502e-02 3.401e-02 4.501e-02 5.700e-02 6.809e-02 - 6.459e-02 6.459e-02 6.025e-02 5.260e-02 4.330e-02 3.401e-02 + 6.460e-02 6.460e-02 6.025e-02 5.260e-02 4.330e-02 3.401e-02 2.589e-02 1.942e-02 1.453e-02 1.096e-02 8.487e-03 6.959e-03 6.247e-03 6.247e-03 6.959e-03 8.487e-03 1.096e-02 1.453e-02 1.942e-02 2.589e-02 3.401e-02 4.330e-02 5.260e-02 6.025e-02 5.116e-02 4.953e-02 4.502e-02 3.864e-02 3.163e-02 2.502e-02 1.942e-02 1.498e-02 1.158e-02 9.073e-03 7.387e-03 6.445e-03 - 6.148e-03 6.445e-03 7.387e-03 9.073e-03 1.158e-02 1.498e-02 + 6.147e-03 6.445e-03 7.387e-03 9.073e-03 1.158e-02 1.498e-02 1.942e-02 2.502e-02 3.163e-02 3.864e-02 4.502e-02 4.953e-02 3.820e-02 3.604e-02 3.221e-02 2.748e-02 2.263e-02 1.822e-02 - 1.453e-02 1.158e-02 9.285e-03 7.626e-03 6.580e-03 6.091e-03 - 6.091e-03 6.580e-03 7.626e-03 9.285e-03 1.158e-02 1.453e-02 + 1.453e-02 1.158e-02 9.285e-03 7.626e-03 6.580e-03 6.090e-03 + 6.090e-03 6.580e-03 7.626e-03 9.285e-03 1.158e-02 1.453e-02 1.822e-02 2.263e-02 2.748e-02 3.221e-02 3.604e-02 3.820e-02 2.728e-02 2.531e-02 2.247e-02 1.926e-02 1.612e-02 1.331e-02 - 1.096e-02 9.073e-03 7.626e-03 6.629e-03 6.065e-03 5.886e-03 - 6.065e-03 6.629e-03 7.626e-03 9.073e-03 1.096e-02 1.331e-02 + 1.096e-02 9.073e-03 7.626e-03 6.628e-03 6.064e-03 5.886e-03 + 6.064e-03 6.628e-03 7.626e-03 9.073e-03 1.096e-02 1.331e-02 1.612e-02 1.926e-02 2.247e-02 2.531e-02 2.728e-02 2.799e-02 1.893e-02 1.744e-02 1.554e-02 1.351e-02 1.157e-02 9.883e-03 - 8.487e-03 7.387e-03 6.580e-03 6.065e-03 5.820e-03 5.820e-03 - 6.065e-03 6.580e-03 7.387e-03 8.487e-03 9.883e-03 1.157e-02 + 8.487e-03 7.387e-03 6.580e-03 6.064e-03 5.820e-03 5.820e-03 + 6.064e-03 6.580e-03 7.387e-03 8.487e-03 9.883e-03 1.157e-02 1.351e-02 1.554e-02 1.744e-02 1.893e-02 1.975e-02 1.975e-02 1.296e-02 1.198e-02 1.081e-02 9.620e-03 8.541e-03 7.650e-03 - 6.959e-03 6.445e-03 6.091e-03 5.886e-03 5.820e-03 5.886e-03 - 6.091e-03 6.445e-03 6.959e-03 7.650e-03 8.541e-03 9.620e-03 + 6.959e-03 6.445e-03 6.090e-03 5.886e-03 5.820e-03 5.886e-03 + 6.090e-03 6.445e-03 6.959e-03 7.650e-03 8.541e-03 9.620e-03 1.081e-02 1.198e-02 1.296e-02 1.363e-02 1.386e-02 1.363e-02 8.892e-03 8.324e-03 7.708e-03 7.143e-03 6.706e-03 6.418e-03 - 6.247e-03 6.148e-03 6.091e-03 6.065e-03 6.065e-03 6.091e-03 - 6.148e-03 6.247e-03 6.418e-03 6.706e-03 7.143e-03 7.708e-03 - 8.324e-03 8.892e-03 9.326e-03 9.563e-03 9.563e-03 9.326e-03 + 6.247e-03 6.147e-03 6.090e-03 6.064e-03 6.064e-03 6.090e-03 + 6.147e-03 6.247e-03 6.418e-03 6.706e-03 7.143e-03 7.708e-03 + 8.324e-03 8.892e-03 9.326e-03 9.562e-03 9.562e-03 9.326e-03 6.247e-03 6.030e-03 5.856e-03 5.788e-03 5.856e-03 6.030e-03 - 6.247e-03 6.445e-03 6.580e-03 6.629e-03 6.580e-03 6.445e-03 + 6.247e-03 6.445e-03 6.580e-03 6.628e-03 6.580e-03 6.445e-03 6.247e-03 6.030e-03 5.856e-03 5.788e-03 5.856e-03 6.030e-03 - 6.247e-03 6.445e-03 6.580e-03 6.629e-03 6.580e-03 6.445e-03 + 6.247e-03 6.445e-03 6.580e-03 6.628e-03 6.580e-03 6.445e-03 4.689e-03 4.789e-03 5.001e-03 5.361e-03 5.856e-03 6.418e-03 6.959e-03 7.387e-03 7.626e-03 7.626e-03 7.387e-03 6.959e-03 6.418e-03 5.856e-03 5.361e-03 5.001e-03 4.789e-03 4.689e-03 @@ -1724,55 +1724,55 @@ Ionic_Step 1 Cubefile created from ABACUS. Inner loop is z, followed by y and x 2.276e-03 2.610e-03 3.356e-03 4.635e-03 6.580e-03 9.326e-03 1.893e-02 2.531e-02 3.221e-02 3.864e-02 4.330e-02 4.501e-02 4.330e-02 3.864e-02 3.221e-02 2.531e-02 1.893e-02 1.363e-02 - 9.563e-03 6.629e-03 4.635e-03 3.394e-03 2.739e-03 2.539e-03 - 2.739e-03 3.394e-03 4.635e-03 6.629e-03 9.563e-03 1.363e-02 + 9.562e-03 6.628e-03 4.635e-03 3.394e-03 2.739e-03 2.539e-03 + 2.739e-03 3.394e-03 4.635e-03 6.628e-03 9.562e-03 1.363e-02 2.728e-02 3.604e-02 4.502e-02 5.260e-02 5.700e-02 5.700e-02 5.260e-02 4.502e-02 3.604e-02 2.728e-02 1.975e-02 1.386e-02 - 9.563e-03 6.580e-03 4.638e-03 3.506e-03 2.998e-03 2.998e-03 - 3.506e-03 4.638e-03 6.580e-03 9.563e-03 1.386e-02 1.975e-02 + 9.562e-03 6.580e-03 4.638e-03 3.506e-03 2.998e-03 2.998e-03 + 3.506e-03 4.638e-03 6.580e-03 9.562e-03 1.386e-02 1.975e-02 3.820e-02 4.953e-02 6.025e-02 6.809e-02 7.099e-02 6.809e-02 6.025e-02 4.953e-02 3.820e-02 2.799e-02 1.975e-02 1.363e-02 9.326e-03 6.445e-03 4.651e-03 3.695e-03 3.399e-03 3.695e-03 4.651e-03 6.445e-03 9.326e-03 1.363e-02 1.975e-02 2.799e-02 6.053e-02 7.300e-02 8.198e-02 8.528e-02 8.198e-02 7.300e-02 6.053e-02 4.707e-02 3.469e-02 2.460e-02 1.714e-02 1.196e-02 - 8.479e-03 6.247e-03 4.997e-03 4.595e-03 4.997e-03 6.247e-03 - 8.479e-03 1.196e-02 1.714e-02 2.460e-02 3.469e-02 4.707e-02 - 7.300e-02 8.473e-02 9.146e-02 9.146e-02 8.473e-02 7.300e-02 + 8.478e-03 6.247e-03 4.997e-03 4.595e-03 4.997e-03 6.247e-03 + 8.478e-03 1.196e-02 1.714e-02 2.460e-02 3.469e-02 4.707e-02 + 7.300e-02 8.474e-02 9.146e-02 9.146e-02 8.474e-02 7.300e-02 5.881e-02 4.466e-02 3.238e-02 2.286e-02 1.606e-02 1.141e-02 8.324e-03 6.418e-03 5.505e-03 5.505e-03 6.418e-03 8.324e-03 1.141e-02 1.606e-02 2.286e-02 3.238e-02 4.466e-02 5.881e-02 - 8.198e-02 9.146e-02 9.491e-02 9.146e-02 8.198e-02 6.863e-02 + 8.198e-02 9.146e-02 9.492e-02 9.146e-02 8.198e-02 6.864e-02 5.398e-02 4.029e-02 2.903e-02 2.066e-02 1.480e-02 1.081e-02 8.199e-03 6.706e-03 6.220e-03 6.706e-03 8.199e-03 1.081e-02 - 1.480e-02 2.066e-02 2.903e-02 4.029e-02 5.398e-02 6.863e-02 + 1.480e-02 2.066e-02 2.903e-02 4.029e-02 5.398e-02 6.864e-02 8.528e-02 9.146e-02 9.146e-02 8.528e-02 7.430e-02 6.074e-02 4.693e-02 3.477e-02 2.524e-02 1.833e-02 1.351e-02 1.022e-02 - 8.149e-03 7.143e-03 7.143e-03 8.149e-03 1.022e-02 1.351e-02 + 8.150e-03 7.143e-03 7.143e-03 8.150e-03 1.022e-02 1.351e-02 1.833e-02 2.524e-02 3.477e-02 4.693e-02 6.074e-02 7.430e-02 - 8.198e-02 8.473e-02 8.198e-02 7.430e-02 6.323e-02 5.077e-02 + 8.198e-02 8.474e-02 8.198e-02 7.430e-02 6.323e-02 5.077e-02 3.892e-02 2.904e-02 2.153e-02 1.612e-02 1.228e-02 9.698e-03 8.199e-03 7.708e-03 8.199e-03 9.698e-03 1.228e-02 1.612e-02 2.153e-02 2.904e-02 3.892e-02 5.077e-02 6.323e-02 7.430e-02 - 7.300e-02 7.300e-02 6.863e-02 6.074e-02 5.077e-02 4.045e-02 + 7.300e-02 7.300e-02 6.864e-02 6.074e-02 5.077e-02 4.045e-02 3.122e-02 2.381e-02 1.822e-02 1.411e-02 1.116e-02 9.254e-03 8.324e-03 8.324e-03 9.254e-03 1.116e-02 1.411e-02 1.822e-02 - 2.381e-02 3.122e-02 4.045e-02 5.077e-02 6.074e-02 6.863e-02 + 2.381e-02 3.122e-02 4.045e-02 5.077e-02 6.074e-02 6.864e-02 6.053e-02 5.881e-02 5.398e-02 4.693e-02 3.892e-02 3.122e-02 2.464e-02 1.942e-02 1.538e-02 1.232e-02 1.016e-02 8.892e-03 - 8.479e-03 8.892e-03 1.016e-02 1.232e-02 1.538e-02 1.942e-02 + 8.478e-03 8.892e-03 1.016e-02 1.232e-02 1.538e-02 1.942e-02 2.464e-02 3.122e-02 3.892e-02 4.693e-02 5.398e-02 5.881e-02 4.707e-02 4.466e-02 4.029e-02 3.477e-02 2.904e-02 2.381e-02 1.942e-02 1.584e-02 1.297e-02 1.078e-02 9.326e-03 8.621e-03 8.621e-03 9.326e-03 1.078e-02 1.297e-02 1.584e-02 1.942e-02 2.381e-02 2.904e-02 3.477e-02 4.029e-02 4.466e-02 4.707e-02 3.469e-02 3.238e-02 2.903e-02 2.524e-02 2.153e-02 1.822e-02 - 1.538e-02 1.297e-02 1.100e-02 9.563e-03 8.723e-03 8.453e-03 - 8.723e-03 9.563e-03 1.100e-02 1.297e-02 1.538e-02 1.822e-02 + 1.538e-02 1.297e-02 1.100e-02 9.562e-03 8.723e-03 8.453e-03 + 8.723e-03 9.562e-03 1.100e-02 1.297e-02 1.538e-02 1.822e-02 2.153e-02 2.524e-02 2.903e-02 3.238e-02 3.469e-02 3.552e-02 2.460e-02 2.286e-02 2.066e-02 1.833e-02 1.612e-02 1.411e-02 - 1.232e-02 1.078e-02 9.563e-03 8.760e-03 8.378e-03 8.378e-03 - 8.760e-03 9.563e-03 1.078e-02 1.232e-02 1.411e-02 1.612e-02 + 1.232e-02 1.078e-02 9.562e-03 8.760e-03 8.378e-03 8.378e-03 + 8.760e-03 9.562e-03 1.078e-02 1.232e-02 1.411e-02 1.612e-02 1.833e-02 2.066e-02 2.286e-02 2.460e-02 2.558e-02 2.558e-02 1.714e-02 1.606e-02 1.480e-02 1.351e-02 1.228e-02 1.116e-02 1.016e-02 9.326e-03 8.723e-03 8.378e-03 8.270e-03 8.378e-03 @@ -1782,21 +1782,21 @@ Ionic_Step 1 Cubefile created from ABACUS. Inner loop is z, followed by y and x 8.892e-03 8.621e-03 8.453e-03 8.378e-03 8.378e-03 8.453e-03 8.621e-03 8.892e-03 9.254e-03 9.698e-03 1.022e-02 1.081e-02 1.141e-02 1.196e-02 1.238e-02 1.262e-02 1.262e-02 1.238e-02 - 8.479e-03 8.324e-03 8.199e-03 8.149e-03 8.199e-03 8.324e-03 - 8.479e-03 8.621e-03 8.723e-03 8.760e-03 8.723e-03 8.621e-03 - 8.479e-03 8.324e-03 8.199e-03 8.149e-03 8.199e-03 8.324e-03 - 8.479e-03 8.621e-03 8.723e-03 8.760e-03 8.723e-03 8.621e-03 + 8.478e-03 8.324e-03 8.199e-03 8.150e-03 8.199e-03 8.324e-03 + 8.478e-03 8.621e-03 8.723e-03 8.760e-03 8.723e-03 8.621e-03 + 8.478e-03 8.324e-03 8.199e-03 8.150e-03 8.199e-03 8.324e-03 + 8.478e-03 8.621e-03 8.723e-03 8.760e-03 8.723e-03 8.621e-03 6.247e-03 6.418e-03 6.706e-03 7.143e-03 7.708e-03 8.324e-03 - 8.892e-03 9.326e-03 9.563e-03 9.563e-03 9.326e-03 8.892e-03 + 8.892e-03 9.326e-03 9.562e-03 9.562e-03 9.326e-03 8.892e-03 8.324e-03 7.708e-03 7.143e-03 6.706e-03 6.418e-03 6.247e-03 - 6.148e-03 6.091e-03 6.065e-03 6.065e-03 6.091e-03 6.148e-03 + 6.147e-03 6.090e-03 6.064e-03 6.064e-03 6.090e-03 6.147e-03 4.997e-03 5.505e-03 6.220e-03 7.143e-03 8.199e-03 9.254e-03 1.016e-02 1.078e-02 1.100e-02 1.078e-02 1.016e-02 9.254e-03 8.199e-03 7.143e-03 6.220e-03 5.505e-03 4.997e-03 4.651e-03 4.420e-03 4.284e-03 4.238e-03 4.284e-03 4.420e-03 4.651e-03 - 4.595e-03 5.505e-03 6.706e-03 8.149e-03 9.698e-03 1.116e-02 + 4.595e-03 5.505e-03 6.706e-03 8.150e-03 9.698e-03 1.116e-02 1.232e-02 1.297e-02 1.297e-02 1.232e-02 1.116e-02 9.698e-03 - 8.149e-03 6.706e-03 5.505e-03 4.595e-03 3.948e-03 3.506e-03 + 8.150e-03 6.706e-03 5.505e-03 4.595e-03 3.948e-03 3.506e-03 3.224e-03 3.084e-03 3.084e-03 3.224e-03 3.506e-03 3.948e-03 4.997e-03 6.418e-03 8.199e-03 1.022e-02 1.228e-02 1.411e-02 1.538e-02 1.584e-02 1.538e-02 1.411e-02 1.228e-02 1.022e-02 @@ -1806,58 +1806,58 @@ Ionic_Step 1 Cubefile created from ABACUS. Inner loop is z, followed by y and x 1.942e-02 1.942e-02 1.822e-02 1.612e-02 1.351e-02 1.081e-02 8.324e-03 6.247e-03 4.651e-03 3.506e-03 2.739e-03 2.276e-03 2.060e-03 2.060e-03 2.276e-03 2.739e-03 3.506e-03 4.651e-03 - 8.479e-03 1.141e-02 1.480e-02 1.833e-02 2.153e-02 2.381e-02 + 8.478e-03 1.141e-02 1.480e-02 1.833e-02 2.153e-02 2.381e-02 2.464e-02 2.381e-02 2.153e-02 1.833e-02 1.480e-02 1.141e-02 - 8.479e-03 6.148e-03 4.420e-03 3.224e-03 2.467e-03 2.060e-03 - 1.933e-03 2.060e-03 2.467e-03 3.224e-03 4.420e-03 6.148e-03 + 8.478e-03 6.147e-03 4.420e-03 3.224e-03 2.467e-03 2.060e-03 + 1.933e-03 2.060e-03 2.467e-03 3.224e-03 4.420e-03 6.147e-03 1.196e-02 1.606e-02 2.066e-02 2.524e-02 2.904e-02 3.122e-02 3.122e-02 2.904e-02 2.524e-02 2.066e-02 1.606e-02 1.196e-02 - 8.621e-03 6.091e-03 4.284e-03 3.084e-03 2.378e-03 2.060e-03 - 2.060e-03 2.378e-03 3.084e-03 4.284e-03 6.091e-03 8.621e-03 + 8.621e-03 6.090e-03 4.284e-03 3.084e-03 2.378e-03 2.060e-03 + 2.060e-03 2.378e-03 3.084e-03 4.284e-03 6.090e-03 8.621e-03 1.714e-02 2.286e-02 2.903e-02 3.477e-02 3.892e-02 4.045e-02 3.892e-02 3.477e-02 2.903e-02 2.286e-02 1.714e-02 1.238e-02 - 8.723e-03 6.065e-03 4.238e-03 3.084e-03 2.467e-03 2.276e-03 - 2.467e-03 3.084e-03 4.238e-03 6.065e-03 8.723e-03 1.238e-02 + 8.723e-03 6.064e-03 4.238e-03 3.084e-03 2.467e-03 2.276e-03 + 2.467e-03 3.084e-03 4.238e-03 6.064e-03 8.723e-03 1.238e-02 2.460e-02 3.238e-02 4.029e-02 4.693e-02 5.077e-02 5.077e-02 4.693e-02 4.029e-02 3.238e-02 2.460e-02 1.789e-02 1.262e-02 - 8.760e-03 6.065e-03 4.284e-03 3.224e-03 2.739e-03 2.739e-03 - 3.224e-03 4.284e-03 6.065e-03 8.760e-03 1.262e-02 1.789e-02 + 8.760e-03 6.064e-03 4.284e-03 3.224e-03 2.739e-03 2.739e-03 + 3.224e-03 4.284e-03 6.064e-03 8.760e-03 1.262e-02 1.789e-02 3.469e-02 4.466e-02 5.398e-02 6.074e-02 6.323e-02 6.074e-02 5.398e-02 4.466e-02 3.469e-02 2.558e-02 1.816e-02 1.262e-02 - 8.723e-03 6.091e-03 4.420e-03 3.506e-03 3.218e-03 3.506e-03 - 4.420e-03 6.091e-03 8.723e-03 1.262e-02 1.816e-02 2.558e-02 - 4.707e-02 5.881e-02 6.863e-02 7.430e-02 7.430e-02 6.863e-02 + 8.723e-03 6.090e-03 4.420e-03 3.506e-03 3.218e-03 3.506e-03 + 4.420e-03 6.090e-03 8.723e-03 1.262e-02 1.816e-02 2.558e-02 + 4.707e-02 5.881e-02 6.864e-02 7.430e-02 7.430e-02 6.864e-02 5.881e-02 4.707e-02 3.552e-02 2.558e-02 1.789e-02 1.238e-02 - 8.621e-03 6.148e-03 4.651e-03 3.948e-03 3.948e-03 4.651e-03 - 6.148e-03 8.621e-03 1.238e-02 1.789e-02 2.558e-02 3.552e-02 - 6.466e-02 7.277e-02 7.727e-02 7.727e-02 7.277e-02 6.466e-02 + 8.621e-03 6.147e-03 4.651e-03 3.948e-03 3.948e-03 4.651e-03 + 6.147e-03 8.621e-03 1.238e-02 1.789e-02 2.558e-02 3.552e-02 + 6.466e-02 7.277e-02 7.728e-02 7.728e-02 7.277e-02 6.466e-02 5.421e-02 4.283e-02 3.206e-02 2.313e-02 1.653e-02 1.196e-02 8.892e-03 6.959e-03 6.008e-03 6.008e-03 6.959e-03 8.892e-03 1.196e-02 1.653e-02 2.313e-02 3.206e-02 4.283e-02 5.421e-02 - 7.277e-02 7.896e-02 8.117e-02 7.896e-02 7.277e-02 6.353e-02 + 7.277e-02 7.897e-02 8.118e-02 7.897e-02 7.277e-02 6.353e-02 5.237e-02 4.073e-02 3.025e-02 2.199e-02 1.606e-02 1.198e-02 9.254e-03 7.650e-03 7.115e-03 7.650e-03 9.254e-03 1.198e-02 1.606e-02 2.199e-02 3.025e-02 4.073e-02 5.237e-02 6.353e-02 - 7.727e-02 8.117e-02 8.117e-02 7.727e-02 7.000e-02 6.009e-02 - 4.868e-02 3.742e-02 2.787e-02 2.066e-02 1.554e-02 1.200e-02 + 7.728e-02 8.118e-02 8.118e-02 7.728e-02 7.000e-02 6.009e-02 + 4.867e-02 3.742e-02 2.787e-02 2.066e-02 1.554e-02 1.200e-02 9.698e-03 8.541e-03 8.541e-03 9.698e-03 1.200e-02 1.554e-02 - 2.066e-02 2.787e-02 3.742e-02 4.868e-02 6.009e-02 7.000e-02 - 7.727e-02 7.896e-02 7.727e-02 7.233e-02 6.446e-02 5.436e-02 - 4.341e-02 3.333e-02 2.524e-02 1.926e-02 1.498e-02 1.201e-02 + 2.066e-02 2.787e-02 3.742e-02 4.867e-02 6.009e-02 7.000e-02 + 7.728e-02 7.897e-02 7.728e-02 7.233e-02 6.446e-02 5.435e-02 + 4.340e-02 3.333e-02 2.524e-02 1.926e-02 1.498e-02 1.201e-02 1.022e-02 9.620e-03 1.022e-02 1.201e-02 1.498e-02 1.926e-02 - 2.524e-02 3.333e-02 4.341e-02 5.436e-02 6.446e-02 7.233e-02 + 2.524e-02 3.333e-02 4.340e-02 5.435e-02 6.446e-02 7.233e-02 7.277e-02 7.277e-02 7.000e-02 6.446e-02 5.643e-02 4.685e-02 - 3.728e-02 2.904e-02 2.263e-02 1.784e-02 1.435e-02 1.200e-02 + 3.727e-02 2.904e-02 2.263e-02 1.784e-02 1.435e-02 1.200e-02 1.081e-02 1.081e-02 1.200e-02 1.435e-02 1.784e-02 2.263e-02 - 2.904e-02 3.728e-02 4.685e-02 5.643e-02 6.446e-02 7.000e-02 - 6.466e-02 6.353e-02 6.009e-02 5.436e-02 4.685e-02 3.872e-02 - 3.122e-02 2.502e-02 2.015e-02 1.639e-02 1.366e-02 1.198e-02 - 1.141e-02 1.198e-02 1.366e-02 1.639e-02 2.015e-02 2.502e-02 - 3.122e-02 3.872e-02 4.685e-02 5.436e-02 6.009e-02 6.353e-02 - 5.421e-02 5.237e-02 4.868e-02 4.341e-02 3.728e-02 3.122e-02 + 2.904e-02 3.727e-02 4.685e-02 5.643e-02 6.446e-02 7.000e-02 + 6.466e-02 6.353e-02 6.009e-02 5.435e-02 4.685e-02 3.872e-02 + 3.122e-02 2.502e-02 2.015e-02 1.640e-02 1.366e-02 1.198e-02 + 1.141e-02 1.198e-02 1.366e-02 1.640e-02 2.015e-02 2.502e-02 + 3.122e-02 3.872e-02 4.685e-02 5.435e-02 6.009e-02 6.353e-02 + 5.421e-02 5.237e-02 4.867e-02 4.340e-02 3.727e-02 3.122e-02 2.589e-02 2.145e-02 1.781e-02 1.495e-02 1.296e-02 1.196e-02 1.196e-02 1.296e-02 1.495e-02 1.781e-02 2.145e-02 2.589e-02 - 3.122e-02 3.728e-02 4.341e-02 4.868e-02 5.237e-02 5.421e-02 + 3.122e-02 3.727e-02 4.340e-02 4.867e-02 5.237e-02 5.421e-02 4.283e-02 4.073e-02 3.742e-02 3.333e-02 2.904e-02 2.502e-02 2.145e-02 1.832e-02 1.566e-02 1.363e-02 1.238e-02 1.197e-02 1.238e-02 1.363e-02 1.566e-02 1.832e-02 2.145e-02 2.502e-02 @@ -1866,14 +1866,14 @@ Ionic_Step 1 Cubefile created from ABACUS. Inner loop is z, followed by y and x 1.781e-02 1.566e-02 1.386e-02 1.262e-02 1.201e-02 1.201e-02 1.262e-02 1.386e-02 1.566e-02 1.781e-02 2.015e-02 2.263e-02 2.524e-02 2.787e-02 3.025e-02 3.206e-02 3.303e-02 3.303e-02 - 2.313e-02 2.199e-02 2.066e-02 1.926e-02 1.784e-02 1.639e-02 + 2.313e-02 2.199e-02 2.066e-02 1.926e-02 1.784e-02 1.640e-02 1.495e-02 1.363e-02 1.262e-02 1.203e-02 1.185e-02 1.203e-02 - 1.262e-02 1.363e-02 1.495e-02 1.639e-02 1.784e-02 1.926e-02 + 1.262e-02 1.363e-02 1.495e-02 1.640e-02 1.784e-02 1.926e-02 2.066e-02 2.199e-02 2.313e-02 2.393e-02 2.422e-02 2.393e-02 1.653e-02 1.606e-02 1.554e-02 1.498e-02 1.435e-02 1.366e-02 1.296e-02 1.238e-02 1.201e-02 1.185e-02 1.185e-02 1.201e-02 1.238e-02 1.296e-02 1.366e-02 1.435e-02 1.498e-02 1.554e-02 - 1.606e-02 1.653e-02 1.693e-02 1.717e-02 1.717e-02 1.693e-02 + 1.606e-02 1.653e-02 1.693e-02 1.716e-02 1.716e-02 1.693e-02 1.196e-02 1.198e-02 1.200e-02 1.201e-02 1.200e-02 1.198e-02 1.196e-02 1.197e-02 1.201e-02 1.203e-02 1.201e-02 1.197e-02 1.196e-02 1.198e-02 1.200e-02 1.201e-02 1.200e-02 1.198e-02 @@ -1885,93 +1885,93 @@ Ionic_Step 1 Cubefile created from ABACUS. Inner loop is z, followed by y and x 6.959e-03 7.650e-03 8.541e-03 9.620e-03 1.081e-02 1.198e-02 1.296e-02 1.363e-02 1.386e-02 1.363e-02 1.296e-02 1.198e-02 1.081e-02 9.620e-03 8.541e-03 7.650e-03 6.959e-03 6.445e-03 - 6.091e-03 5.886e-03 5.820e-03 5.886e-03 6.091e-03 6.445e-03 + 6.090e-03 5.886e-03 5.820e-03 5.886e-03 6.090e-03 6.445e-03 6.008e-03 7.115e-03 8.541e-03 1.022e-02 1.200e-02 1.366e-02 1.495e-02 1.566e-02 1.566e-02 1.495e-02 1.366e-02 1.200e-02 1.022e-02 8.541e-03 7.115e-03 6.008e-03 5.200e-03 4.638e-03 4.284e-03 4.112e-03 4.112e-03 4.284e-03 4.638e-03 5.200e-03 - 6.008e-03 7.650e-03 9.698e-03 1.201e-02 1.435e-02 1.639e-02 - 1.781e-02 1.832e-02 1.781e-02 1.639e-02 1.435e-02 1.201e-02 - 9.698e-03 7.650e-03 6.008e-03 4.792e-03 3.945e-03 3.394e-03 - 3.084e-03 2.984e-03 3.084e-03 3.394e-03 3.945e-03 4.792e-03 + 6.008e-03 7.650e-03 9.698e-03 1.201e-02 1.435e-02 1.640e-02 + 1.781e-02 1.832e-02 1.781e-02 1.640e-02 1.435e-02 1.201e-02 + 9.698e-03 7.650e-03 6.008e-03 4.792e-03 3.944e-03 3.394e-03 + 3.084e-03 2.984e-03 3.084e-03 3.394e-03 3.944e-03 4.792e-03 6.959e-03 9.254e-03 1.200e-02 1.498e-02 1.784e-02 2.015e-02 2.145e-02 2.145e-02 2.015e-02 1.784e-02 1.498e-02 1.200e-02 - 9.254e-03 6.959e-03 5.200e-03 3.945e-03 3.110e-03 2.610e-03 - 2.378e-03 2.378e-03 2.610e-03 3.110e-03 3.945e-03 5.200e-03 + 9.254e-03 6.959e-03 5.200e-03 3.944e-03 3.110e-03 2.610e-03 + 2.378e-03 2.378e-03 2.610e-03 3.110e-03 3.944e-03 5.200e-03 8.892e-03 1.198e-02 1.554e-02 1.926e-02 2.263e-02 2.502e-02 2.589e-02 2.502e-02 2.263e-02 1.926e-02 1.554e-02 1.198e-02 8.892e-03 6.445e-03 4.638e-03 3.394e-03 2.610e-03 2.190e-03 2.060e-03 2.190e-03 2.610e-03 3.394e-03 4.638e-03 6.445e-03 1.196e-02 1.606e-02 2.066e-02 2.524e-02 2.904e-02 3.122e-02 3.122e-02 2.904e-02 2.524e-02 2.066e-02 1.606e-02 1.196e-02 - 8.621e-03 6.091e-03 4.284e-03 3.084e-03 2.378e-03 2.060e-03 - 2.060e-03 2.378e-03 3.084e-03 4.284e-03 6.091e-03 8.621e-03 - 1.653e-02 2.199e-02 2.787e-02 3.333e-02 3.728e-02 3.872e-02 - 3.728e-02 3.333e-02 2.787e-02 2.199e-02 1.653e-02 1.197e-02 + 8.621e-03 6.090e-03 4.284e-03 3.084e-03 2.378e-03 2.060e-03 + 2.060e-03 2.378e-03 3.084e-03 4.284e-03 6.090e-03 8.621e-03 + 1.653e-02 2.199e-02 2.787e-02 3.333e-02 3.727e-02 3.872e-02 + 3.727e-02 3.333e-02 2.787e-02 2.199e-02 1.653e-02 1.197e-02 8.453e-03 5.886e-03 4.112e-03 2.984e-03 2.378e-03 2.190e-03 2.378e-03 2.984e-03 4.112e-03 5.886e-03 8.453e-03 1.197e-02 - 2.313e-02 3.025e-02 3.742e-02 4.341e-02 4.685e-02 4.685e-02 - 4.341e-02 3.742e-02 3.025e-02 2.313e-02 1.693e-02 1.201e-02 + 2.313e-02 3.025e-02 3.742e-02 4.340e-02 4.685e-02 4.685e-02 + 4.340e-02 3.742e-02 3.025e-02 2.313e-02 1.693e-02 1.201e-02 8.378e-03 5.820e-03 4.112e-03 3.084e-03 2.610e-03 2.610e-03 3.084e-03 4.112e-03 5.820e-03 8.378e-03 1.201e-02 1.693e-02 - 3.206e-02 4.073e-02 4.868e-02 5.436e-02 5.643e-02 5.436e-02 - 4.868e-02 4.073e-02 3.206e-02 2.393e-02 1.717e-02 1.203e-02 + 3.206e-02 4.073e-02 4.867e-02 5.435e-02 5.643e-02 5.435e-02 + 4.867e-02 4.073e-02 3.206e-02 2.393e-02 1.716e-02 1.203e-02 8.378e-03 5.886e-03 4.284e-03 3.394e-03 3.110e-03 3.394e-03 - 4.284e-03 5.886e-03 8.378e-03 1.203e-02 1.717e-02 2.393e-02 + 4.284e-03 5.886e-03 8.378e-03 1.203e-02 1.716e-02 2.393e-02 4.283e-02 5.237e-02 6.009e-02 6.446e-02 6.446e-02 6.009e-02 - 5.237e-02 4.283e-02 3.303e-02 2.422e-02 1.717e-02 1.201e-02 - 8.453e-03 6.091e-03 4.638e-03 3.945e-03 3.945e-03 4.638e-03 - 6.091e-03 8.453e-03 1.201e-02 1.717e-02 2.422e-02 3.303e-02 + 5.237e-02 4.283e-02 3.303e-02 2.422e-02 1.716e-02 1.201e-02 + 8.453e-03 6.090e-03 4.638e-03 3.944e-03 3.944e-03 4.638e-03 + 6.090e-03 8.453e-03 1.201e-02 1.716e-02 2.422e-02 3.303e-02 5.421e-02 6.353e-02 7.000e-02 7.233e-02 7.000e-02 6.353e-02 5.421e-02 4.355e-02 3.303e-02 2.393e-02 1.693e-02 1.197e-02 8.621e-03 6.445e-03 5.200e-03 4.792e-03 5.200e-03 6.445e-03 8.621e-03 1.197e-02 1.693e-02 2.393e-02 3.303e-02 4.355e-02 - 5.728e-02 5.903e-02 5.962e-02 5.903e-02 5.728e-02 5.397e-02 - 4.821e-02 4.006e-02 3.107e-02 2.313e-02 1.714e-02 1.296e-02 + 5.728e-02 5.903e-02 5.962e-02 5.903e-02 5.728e-02 5.396e-02 + 4.821e-02 4.005e-02 3.107e-02 2.313e-02 1.714e-02 1.296e-02 1.016e-02 8.487e-03 7.921e-03 8.487e-03 1.016e-02 1.296e-02 - 1.714e-02 2.313e-02 3.107e-02 4.006e-02 4.821e-02 5.397e-02 + 1.714e-02 2.313e-02 3.107e-02 4.005e-02 4.821e-02 5.396e-02 5.903e-02 5.950e-02 5.950e-02 5.903e-02 5.775e-02 5.442e-02 4.798e-02 3.920e-02 3.025e-02 2.286e-02 1.744e-02 1.366e-02 1.116e-02 9.883e-03 9.883e-03 1.116e-02 1.366e-02 1.744e-02 2.286e-02 3.025e-02 3.920e-02 4.798e-02 5.442e-02 5.775e-02 - 5.962e-02 5.950e-02 5.962e-02 5.968e-02 5.845e-02 5.416e-02 + 5.962e-02 5.950e-02 5.962e-02 5.968e-02 5.844e-02 5.416e-02 4.655e-02 3.742e-02 2.903e-02 2.247e-02 1.770e-02 1.435e-02 1.228e-02 1.157e-02 1.228e-02 1.435e-02 1.770e-02 2.247e-02 - 2.903e-02 3.742e-02 4.655e-02 5.416e-02 5.845e-02 5.968e-02 - 5.903e-02 5.903e-02 5.968e-02 5.986e-02 5.759e-02 5.176e-02 - 4.341e-02 3.477e-02 2.748e-02 2.191e-02 1.780e-02 1.498e-02 + 2.903e-02 3.742e-02 4.655e-02 5.416e-02 5.844e-02 5.968e-02 + 5.903e-02 5.903e-02 5.968e-02 5.985e-02 5.759e-02 5.176e-02 + 4.340e-02 3.477e-02 2.748e-02 2.191e-02 1.780e-02 1.498e-02 1.351e-02 1.351e-02 1.498e-02 1.780e-02 2.191e-02 2.748e-02 - 3.477e-02 4.341e-02 5.176e-02 5.759e-02 5.986e-02 5.968e-02 - 5.728e-02 5.775e-02 5.845e-02 5.759e-02 5.366e-02 4.685e-02 + 3.477e-02 4.340e-02 5.176e-02 5.759e-02 5.985e-02 5.968e-02 + 5.728e-02 5.775e-02 5.844e-02 5.759e-02 5.366e-02 4.685e-02 3.892e-02 3.163e-02 2.570e-02 2.110e-02 1.770e-02 1.554e-02 1.480e-02 1.554e-02 1.770e-02 2.110e-02 2.570e-02 3.163e-02 - 3.892e-02 4.685e-02 5.366e-02 5.759e-02 5.845e-02 5.775e-02 - 5.397e-02 5.442e-02 5.416e-02 5.176e-02 4.685e-02 4.045e-02 - 3.401e-02 2.838e-02 2.375e-02 2.007e-02 1.744e-02 1.606e-02 - 1.606e-02 1.744e-02 2.007e-02 2.375e-02 2.838e-02 3.401e-02 - 4.045e-02 4.685e-02 5.176e-02 5.416e-02 5.442e-02 5.397e-02 - 4.821e-02 4.798e-02 4.655e-02 4.341e-02 3.892e-02 3.401e-02 + 3.892e-02 4.685e-02 5.366e-02 5.759e-02 5.844e-02 5.775e-02 + 5.396e-02 5.442e-02 5.416e-02 5.176e-02 4.685e-02 4.045e-02 + 3.401e-02 2.839e-02 2.375e-02 2.007e-02 1.744e-02 1.606e-02 + 1.606e-02 1.744e-02 2.007e-02 2.375e-02 2.839e-02 3.401e-02 + 4.045e-02 4.685e-02 5.176e-02 5.416e-02 5.442e-02 5.396e-02 + 4.821e-02 4.798e-02 4.655e-02 4.340e-02 3.892e-02 3.401e-02 2.936e-02 2.524e-02 2.171e-02 1.893e-02 1.714e-02 1.653e-02 1.714e-02 1.893e-02 2.171e-02 2.524e-02 2.936e-02 3.401e-02 - 3.892e-02 4.341e-02 4.655e-02 4.798e-02 4.821e-02 4.814e-02 - 4.006e-02 3.920e-02 3.742e-02 3.477e-02 3.163e-02 2.838e-02 + 3.892e-02 4.340e-02 4.655e-02 4.798e-02 4.821e-02 4.814e-02 + 4.005e-02 3.920e-02 3.742e-02 3.477e-02 3.163e-02 2.839e-02 2.524e-02 2.230e-02 1.975e-02 1.789e-02 1.693e-02 1.693e-02 - 1.789e-02 1.975e-02 2.230e-02 2.524e-02 2.838e-02 3.163e-02 - 3.477e-02 3.742e-02 3.920e-02 4.006e-02 4.032e-02 4.032e-02 + 1.789e-02 1.975e-02 2.230e-02 2.524e-02 2.839e-02 3.163e-02 + 3.477e-02 3.742e-02 3.920e-02 4.005e-02 4.032e-02 4.032e-02 3.107e-02 3.025e-02 2.903e-02 2.748e-02 2.570e-02 2.375e-02 - 2.171e-02 1.975e-02 1.816e-02 1.717e-02 1.684e-02 1.717e-02 + 2.171e-02 1.975e-02 1.816e-02 1.716e-02 1.683e-02 1.716e-02 1.816e-02 1.975e-02 2.171e-02 2.375e-02 2.570e-02 2.748e-02 2.903e-02 3.025e-02 3.107e-02 3.151e-02 3.165e-02 3.151e-02 2.313e-02 2.286e-02 2.247e-02 2.191e-02 2.110e-02 2.007e-02 - 1.893e-02 1.789e-02 1.717e-02 1.682e-02 1.682e-02 1.717e-02 + 1.893e-02 1.789e-02 1.716e-02 1.682e-02 1.682e-02 1.716e-02 1.789e-02 1.893e-02 2.007e-02 2.110e-02 2.191e-02 2.247e-02 2.286e-02 2.313e-02 2.335e-02 2.348e-02 2.348e-02 2.335e-02 1.714e-02 1.744e-02 1.770e-02 1.780e-02 1.770e-02 1.744e-02 - 1.714e-02 1.693e-02 1.684e-02 1.682e-02 1.684e-02 1.693e-02 + 1.714e-02 1.693e-02 1.683e-02 1.682e-02 1.683e-02 1.693e-02 1.714e-02 1.744e-02 1.770e-02 1.780e-02 1.770e-02 1.744e-02 - 1.714e-02 1.693e-02 1.684e-02 1.682e-02 1.684e-02 1.693e-02 + 1.714e-02 1.693e-02 1.683e-02 1.682e-02 1.683e-02 1.693e-02 1.296e-02 1.366e-02 1.435e-02 1.498e-02 1.554e-02 1.606e-02 - 1.653e-02 1.693e-02 1.717e-02 1.717e-02 1.693e-02 1.653e-02 + 1.653e-02 1.693e-02 1.716e-02 1.716e-02 1.693e-02 1.653e-02 1.606e-02 1.554e-02 1.498e-02 1.435e-02 1.366e-02 1.296e-02 1.238e-02 1.201e-02 1.185e-02 1.185e-02 1.201e-02 1.238e-02 1.016e-02 1.116e-02 1.228e-02 1.351e-02 1.480e-02 1.606e-02 @@ -1981,17 +1981,17 @@ Ionic_Step 1 Cubefile created from ABACUS. Inner loop is z, followed by y and x 8.487e-03 9.883e-03 1.157e-02 1.351e-02 1.554e-02 1.744e-02 1.893e-02 1.975e-02 1.975e-02 1.893e-02 1.744e-02 1.554e-02 1.351e-02 1.157e-02 9.883e-03 8.487e-03 7.387e-03 6.580e-03 - 6.065e-03 5.820e-03 5.820e-03 6.065e-03 6.580e-03 7.387e-03 + 6.064e-03 5.820e-03 5.820e-03 6.064e-03 6.580e-03 7.387e-03 7.921e-03 9.883e-03 1.228e-02 1.498e-02 1.770e-02 2.007e-02 2.171e-02 2.230e-02 2.171e-02 2.007e-02 1.770e-02 1.498e-02 1.228e-02 9.883e-03 7.921e-03 6.424e-03 5.346e-03 4.635e-03 4.238e-03 4.112e-03 4.238e-03 4.635e-03 5.346e-03 6.424e-03 8.487e-03 1.116e-02 1.435e-02 1.780e-02 2.110e-02 2.375e-02 2.524e-02 2.524e-02 2.375e-02 2.110e-02 1.780e-02 1.435e-02 - 1.116e-02 8.487e-03 6.424e-03 4.940e-03 3.948e-03 3.356e-03 - 3.084e-03 3.084e-03 3.356e-03 3.948e-03 4.940e-03 6.424e-03 - 1.016e-02 1.366e-02 1.770e-02 2.191e-02 2.570e-02 2.838e-02 - 2.936e-02 2.838e-02 2.570e-02 2.191e-02 1.770e-02 1.366e-02 + 1.116e-02 8.487e-03 6.424e-03 4.939e-03 3.948e-03 3.356e-03 + 3.084e-03 3.084e-03 3.356e-03 3.948e-03 4.939e-03 6.424e-03 + 1.016e-02 1.366e-02 1.770e-02 2.191e-02 2.570e-02 2.839e-02 + 2.936e-02 2.839e-02 2.570e-02 2.191e-02 1.770e-02 1.366e-02 1.016e-02 7.387e-03 5.346e-03 3.948e-03 3.074e-03 2.610e-03 2.467e-03 2.610e-03 3.074e-03 3.948e-03 5.346e-03 7.387e-03 1.296e-02 1.744e-02 2.247e-02 2.748e-02 3.163e-02 3.401e-02 @@ -2000,60 +2000,60 @@ Ionic_Step 1 Cubefile created from ABACUS. Inner loop is z, followed by y and x 2.276e-03 2.610e-03 3.356e-03 4.635e-03 6.580e-03 9.326e-03 1.714e-02 2.286e-02 2.903e-02 3.477e-02 3.892e-02 4.045e-02 3.892e-02 3.477e-02 2.903e-02 2.286e-02 1.714e-02 1.238e-02 - 8.723e-03 6.065e-03 4.238e-03 3.084e-03 2.467e-03 2.276e-03 - 2.467e-03 3.084e-03 4.238e-03 6.065e-03 8.723e-03 1.238e-02 - 2.313e-02 3.025e-02 3.742e-02 4.341e-02 4.685e-02 4.685e-02 - 4.341e-02 3.742e-02 3.025e-02 2.313e-02 1.693e-02 1.201e-02 + 8.723e-03 6.064e-03 4.238e-03 3.084e-03 2.467e-03 2.276e-03 + 2.467e-03 3.084e-03 4.238e-03 6.064e-03 8.723e-03 1.238e-02 + 2.313e-02 3.025e-02 3.742e-02 4.340e-02 4.685e-02 4.685e-02 + 4.340e-02 3.742e-02 3.025e-02 2.313e-02 1.693e-02 1.201e-02 8.378e-03 5.820e-03 4.112e-03 3.084e-03 2.610e-03 2.610e-03 3.084e-03 4.112e-03 5.820e-03 8.378e-03 1.201e-02 1.693e-02 3.107e-02 3.920e-02 4.655e-02 5.176e-02 5.366e-02 5.176e-02 - 4.655e-02 3.920e-02 3.107e-02 2.335e-02 1.684e-02 1.185e-02 + 4.655e-02 3.920e-02 3.107e-02 2.335e-02 1.683e-02 1.185e-02 8.270e-03 5.820e-03 4.238e-03 3.356e-03 3.074e-03 3.356e-03 - 4.238e-03 5.820e-03 8.270e-03 1.185e-02 1.684e-02 2.335e-02 - 4.006e-02 4.798e-02 5.416e-02 5.759e-02 5.759e-02 5.416e-02 - 4.798e-02 4.006e-02 3.151e-02 2.348e-02 1.682e-02 1.185e-02 - 8.378e-03 6.065e-03 4.635e-03 3.948e-03 3.948e-03 4.635e-03 - 6.065e-03 8.378e-03 1.185e-02 1.682e-02 2.348e-02 3.151e-02 - 4.821e-02 5.442e-02 5.845e-02 5.986e-02 5.845e-02 5.442e-02 - 4.821e-02 4.032e-02 3.165e-02 2.348e-02 1.684e-02 1.201e-02 - 8.723e-03 6.580e-03 5.346e-03 4.940e-03 5.346e-03 6.580e-03 - 8.723e-03 1.201e-02 1.684e-02 2.348e-02 3.165e-02 4.032e-02 - 5.397e-02 5.775e-02 5.968e-02 5.968e-02 5.775e-02 5.397e-02 + 4.238e-03 5.820e-03 8.270e-03 1.185e-02 1.683e-02 2.335e-02 + 4.005e-02 4.798e-02 5.416e-02 5.759e-02 5.759e-02 5.416e-02 + 4.798e-02 4.005e-02 3.151e-02 2.348e-02 1.682e-02 1.185e-02 + 8.378e-03 6.064e-03 4.635e-03 3.948e-03 3.948e-03 4.635e-03 + 6.064e-03 8.378e-03 1.185e-02 1.682e-02 2.348e-02 3.151e-02 + 4.821e-02 5.442e-02 5.844e-02 5.985e-02 5.844e-02 5.442e-02 + 4.821e-02 4.032e-02 3.165e-02 2.348e-02 1.683e-02 1.201e-02 + 8.723e-03 6.580e-03 5.346e-03 4.939e-03 5.346e-03 6.580e-03 + 8.723e-03 1.201e-02 1.683e-02 2.348e-02 3.165e-02 4.032e-02 + 5.396e-02 5.775e-02 5.968e-02 5.968e-02 5.775e-02 5.396e-02 4.814e-02 4.032e-02 3.151e-02 2.335e-02 1.693e-02 1.238e-02 9.326e-03 7.387e-03 6.424e-03 6.424e-03 7.387e-03 9.326e-03 1.238e-02 1.693e-02 2.335e-02 3.151e-02 4.032e-02 4.814e-02 - 3.853e-02 3.604e-02 3.604e-02 3.853e-02 4.294e-02 4.650e-02 - 4.573e-02 4.006e-02 3.206e-02 2.460e-02 1.893e-02 1.495e-02 + 3.852e-02 3.604e-02 3.604e-02 3.852e-02 4.294e-02 4.650e-02 + 4.573e-02 4.005e-02 3.206e-02 2.460e-02 1.893e-02 1.495e-02 1.232e-02 1.096e-02 1.096e-02 1.232e-02 1.495e-02 1.893e-02 - 2.460e-02 3.206e-02 4.006e-02 4.573e-02 4.650e-02 4.294e-02 + 2.460e-02 3.206e-02 4.005e-02 4.573e-02 4.650e-02 4.294e-02 3.604e-02 3.431e-02 3.604e-02 4.108e-02 4.740e-02 5.065e-02 - 4.798e-02 4.073e-02 3.238e-02 2.531e-02 2.007e-02 1.639e-02 - 1.411e-02 1.331e-02 1.411e-02 1.639e-02 2.007e-02 2.531e-02 + 4.798e-02 4.073e-02 3.238e-02 2.531e-02 2.007e-02 1.640e-02 + 1.411e-02 1.331e-02 1.411e-02 1.640e-02 2.007e-02 2.531e-02 3.238e-02 4.073e-02 4.798e-02 5.065e-02 4.740e-02 4.108e-02 - 3.604e-02 3.604e-02 4.041e-02 4.765e-02 5.366e-02 5.416e-02 - 4.868e-02 4.029e-02 3.221e-02 2.582e-02 2.110e-02 1.784e-02 + 3.604e-02 3.604e-02 4.040e-02 4.765e-02 5.366e-02 5.416e-02 + 4.867e-02 4.029e-02 3.221e-02 2.582e-02 2.110e-02 1.784e-02 1.612e-02 1.612e-02 1.784e-02 2.110e-02 2.582e-02 3.221e-02 - 4.029e-02 4.868e-02 5.416e-02 5.366e-02 4.765e-02 4.041e-02 - 3.853e-02 4.108e-02 4.765e-02 5.469e-02 5.759e-02 5.436e-02 + 4.029e-02 4.867e-02 5.416e-02 5.366e-02 4.765e-02 4.040e-02 + 3.852e-02 4.108e-02 4.765e-02 5.469e-02 5.759e-02 5.435e-02 4.693e-02 3.864e-02 3.153e-02 2.600e-02 2.191e-02 1.926e-02 1.833e-02 1.926e-02 2.191e-02 2.600e-02 3.153e-02 3.864e-02 - 4.693e-02 5.436e-02 5.759e-02 5.469e-02 4.765e-02 4.108e-02 + 4.693e-02 5.435e-02 5.759e-02 5.469e-02 4.765e-02 4.108e-02 4.294e-02 4.740e-02 5.366e-02 5.759e-02 5.643e-02 5.077e-02 4.330e-02 3.625e-02 3.042e-02 2.582e-02 2.247e-02 2.066e-02 2.066e-02 2.247e-02 2.582e-02 3.042e-02 3.625e-02 4.330e-02 5.077e-02 5.643e-02 5.759e-02 5.366e-02 4.740e-02 4.294e-02 - 4.650e-02 5.065e-02 5.416e-02 5.436e-02 5.077e-02 4.501e-02 - 3.895e-02 3.354e-02 2.897e-02 2.531e-02 2.286e-02 2.199e-02 - 2.286e-02 2.531e-02 2.897e-02 3.354e-02 3.895e-02 4.501e-02 - 5.077e-02 5.436e-02 5.416e-02 5.065e-02 4.650e-02 4.471e-02 - 4.573e-02 4.798e-02 4.868e-02 4.693e-02 4.330e-02 3.895e-02 + 4.650e-02 5.065e-02 5.416e-02 5.435e-02 5.077e-02 4.501e-02 + 3.895e-02 3.355e-02 2.897e-02 2.531e-02 2.286e-02 2.199e-02 + 2.286e-02 2.531e-02 2.897e-02 3.355e-02 3.895e-02 4.501e-02 + 5.077e-02 5.435e-02 5.416e-02 5.065e-02 4.650e-02 4.470e-02 + 4.573e-02 4.798e-02 4.867e-02 4.693e-02 4.330e-02 3.895e-02 3.468e-02 3.074e-02 2.728e-02 2.460e-02 2.313e-02 2.313e-02 2.460e-02 2.728e-02 3.074e-02 3.468e-02 3.895e-02 4.330e-02 - 4.693e-02 4.868e-02 4.798e-02 4.573e-02 4.385e-02 4.385e-02 - 4.006e-02 4.073e-02 4.029e-02 3.864e-02 3.625e-02 3.354e-02 + 4.693e-02 4.867e-02 4.798e-02 4.573e-02 4.385e-02 4.385e-02 + 4.005e-02 4.073e-02 4.029e-02 3.864e-02 3.625e-02 3.355e-02 3.074e-02 2.799e-02 2.558e-02 2.393e-02 2.335e-02 2.393e-02 - 2.558e-02 2.799e-02 3.074e-02 3.354e-02 3.625e-02 3.864e-02 - 4.029e-02 4.073e-02 4.006e-02 3.901e-02 3.852e-02 3.901e-02 + 2.558e-02 2.799e-02 3.074e-02 3.355e-02 3.625e-02 3.864e-02 + 4.029e-02 4.073e-02 4.005e-02 3.901e-02 3.852e-02 3.901e-02 3.206e-02 3.238e-02 3.221e-02 3.153e-02 3.042e-02 2.897e-02 2.728e-02 2.558e-02 2.422e-02 2.348e-02 2.348e-02 2.422e-02 2.558e-02 2.728e-02 2.897e-02 3.042e-02 3.153e-02 3.221e-02 @@ -2065,25 +2065,25 @@ Ionic_Step 1 Cubefile created from ABACUS. Inner loop is z, followed by y and x 1.893e-02 2.007e-02 2.110e-02 2.191e-02 2.247e-02 2.286e-02 2.313e-02 2.335e-02 2.348e-02 2.348e-02 2.335e-02 2.313e-02 2.286e-02 2.247e-02 2.191e-02 2.110e-02 2.007e-02 1.893e-02 - 1.789e-02 1.717e-02 1.682e-02 1.682e-02 1.717e-02 1.789e-02 - 1.495e-02 1.639e-02 1.784e-02 1.926e-02 2.066e-02 2.199e-02 + 1.789e-02 1.716e-02 1.682e-02 1.682e-02 1.716e-02 1.789e-02 + 1.495e-02 1.640e-02 1.784e-02 1.926e-02 2.066e-02 2.199e-02 2.313e-02 2.393e-02 2.422e-02 2.393e-02 2.313e-02 2.199e-02 - 2.066e-02 1.926e-02 1.784e-02 1.639e-02 1.495e-02 1.363e-02 + 2.066e-02 1.926e-02 1.784e-02 1.640e-02 1.495e-02 1.363e-02 1.262e-02 1.203e-02 1.185e-02 1.203e-02 1.262e-02 1.363e-02 1.232e-02 1.411e-02 1.612e-02 1.833e-02 2.066e-02 2.286e-02 2.460e-02 2.558e-02 2.558e-02 2.460e-02 2.286e-02 2.066e-02 - 1.833e-02 1.612e-02 1.411e-02 1.232e-02 1.078e-02 9.563e-03 - 8.760e-03 8.378e-03 8.378e-03 8.760e-03 9.563e-03 1.078e-02 + 1.833e-02 1.612e-02 1.411e-02 1.232e-02 1.078e-02 9.562e-03 + 8.760e-03 8.378e-03 8.378e-03 8.760e-03 9.562e-03 1.078e-02 1.096e-02 1.331e-02 1.612e-02 1.926e-02 2.247e-02 2.531e-02 2.728e-02 2.799e-02 2.728e-02 2.531e-02 2.247e-02 1.926e-02 - 1.612e-02 1.331e-02 1.096e-02 9.073e-03 7.626e-03 6.629e-03 - 6.065e-03 5.886e-03 6.065e-03 6.629e-03 7.626e-03 9.073e-03 + 1.612e-02 1.331e-02 1.096e-02 9.073e-03 7.626e-03 6.628e-03 + 6.064e-03 5.886e-03 6.064e-03 6.628e-03 7.626e-03 9.073e-03 1.096e-02 1.411e-02 1.784e-02 2.191e-02 2.582e-02 2.897e-02 3.074e-02 3.074e-02 2.897e-02 2.582e-02 2.191e-02 1.784e-02 1.411e-02 1.096e-02 8.494e-03 6.662e-03 5.400e-03 4.635e-03 4.284e-03 4.284e-03 4.635e-03 5.400e-03 6.662e-03 8.494e-03 - 1.232e-02 1.639e-02 2.110e-02 2.600e-02 3.042e-02 3.354e-02 - 3.468e-02 3.354e-02 3.042e-02 2.600e-02 2.110e-02 1.639e-02 + 1.232e-02 1.640e-02 2.110e-02 2.600e-02 3.042e-02 3.355e-02 + 3.468e-02 3.355e-02 3.042e-02 2.600e-02 2.110e-02 1.640e-02 1.232e-02 9.073e-03 6.662e-03 4.995e-03 3.948e-03 3.394e-03 3.224e-03 3.394e-03 3.948e-03 4.995e-03 6.662e-03 9.073e-03 1.495e-02 2.007e-02 2.582e-02 3.153e-02 3.625e-02 3.895e-02 @@ -2092,60 +2092,60 @@ Ionic_Step 1 Cubefile created from ABACUS. Inner loop is z, followed by y and x 2.739e-03 3.110e-03 3.948e-03 5.400e-03 7.626e-03 1.078e-02 1.893e-02 2.531e-02 3.221e-02 3.864e-02 4.330e-02 4.501e-02 4.330e-02 3.864e-02 3.221e-02 2.531e-02 1.893e-02 1.363e-02 - 9.563e-03 6.629e-03 4.635e-03 3.394e-03 2.739e-03 2.539e-03 - 2.739e-03 3.394e-03 4.635e-03 6.629e-03 9.563e-03 1.363e-02 + 9.562e-03 6.628e-03 4.635e-03 3.394e-03 2.739e-03 2.539e-03 + 2.739e-03 3.394e-03 4.635e-03 6.628e-03 9.562e-03 1.363e-02 2.460e-02 3.238e-02 4.029e-02 4.693e-02 5.077e-02 5.077e-02 4.693e-02 4.029e-02 3.238e-02 2.460e-02 1.789e-02 1.262e-02 - 8.760e-03 6.065e-03 4.284e-03 3.224e-03 2.739e-03 2.739e-03 - 3.224e-03 4.284e-03 6.065e-03 8.760e-03 1.262e-02 1.789e-02 - 3.206e-02 4.073e-02 4.868e-02 5.436e-02 5.643e-02 5.436e-02 - 4.868e-02 4.073e-02 3.206e-02 2.393e-02 1.717e-02 1.203e-02 + 8.760e-03 6.064e-03 4.284e-03 3.224e-03 2.739e-03 2.739e-03 + 3.224e-03 4.284e-03 6.064e-03 8.760e-03 1.262e-02 1.789e-02 + 3.206e-02 4.073e-02 4.867e-02 5.435e-02 5.643e-02 5.435e-02 + 4.867e-02 4.073e-02 3.206e-02 2.393e-02 1.716e-02 1.203e-02 8.378e-03 5.886e-03 4.284e-03 3.394e-03 3.110e-03 3.394e-03 - 4.284e-03 5.886e-03 8.378e-03 1.203e-02 1.717e-02 2.393e-02 - 4.006e-02 4.798e-02 5.416e-02 5.759e-02 5.759e-02 5.416e-02 - 4.798e-02 4.006e-02 3.151e-02 2.348e-02 1.682e-02 1.185e-02 - 8.378e-03 6.065e-03 4.635e-03 3.948e-03 3.948e-03 4.635e-03 - 6.065e-03 8.378e-03 1.185e-02 1.682e-02 2.348e-02 3.151e-02 + 4.284e-03 5.886e-03 8.378e-03 1.203e-02 1.716e-02 2.393e-02 + 4.005e-02 4.798e-02 5.416e-02 5.759e-02 5.759e-02 5.416e-02 + 4.798e-02 4.005e-02 3.151e-02 2.348e-02 1.682e-02 1.185e-02 + 8.378e-03 6.064e-03 4.635e-03 3.948e-03 3.948e-03 4.635e-03 + 6.064e-03 8.378e-03 1.185e-02 1.682e-02 2.348e-02 3.151e-02 4.573e-02 5.065e-02 5.366e-02 5.469e-02 5.366e-02 5.065e-02 4.573e-02 3.901e-02 3.112e-02 2.333e-02 1.682e-02 1.203e-02 - 8.760e-03 6.629e-03 5.400e-03 4.995e-03 5.400e-03 6.629e-03 + 8.760e-03 6.628e-03 5.400e-03 4.995e-03 5.400e-03 6.628e-03 8.760e-03 1.203e-02 1.682e-02 2.333e-02 3.112e-02 3.901e-02 4.650e-02 4.740e-02 4.765e-02 4.765e-02 4.740e-02 4.650e-02 - 4.385e-02 3.852e-02 3.112e-02 2.348e-02 1.717e-02 1.262e-02 - 9.563e-03 7.626e-03 6.662e-03 6.662e-03 7.626e-03 9.563e-03 - 1.262e-02 1.717e-02 2.348e-02 3.112e-02 3.852e-02 4.385e-02 - 4.294e-02 4.108e-02 4.041e-02 4.108e-02 4.294e-02 4.471e-02 + 4.385e-02 3.852e-02 3.112e-02 2.348e-02 1.716e-02 1.262e-02 + 9.562e-03 7.626e-03 6.662e-03 6.662e-03 7.626e-03 9.562e-03 + 1.262e-02 1.716e-02 2.348e-02 3.112e-02 3.852e-02 4.385e-02 + 4.294e-02 4.108e-02 4.040e-02 4.108e-02 4.294e-02 4.470e-02 4.385e-02 3.901e-02 3.151e-02 2.393e-02 1.789e-02 1.363e-02 1.078e-02 9.073e-03 8.494e-03 9.073e-03 1.078e-02 1.363e-02 - 1.789e-02 2.393e-02 3.151e-02 3.901e-02 4.385e-02 4.471e-02 - 1.770e-02 1.527e-02 1.770e-02 2.556e-02 3.719e-02 4.650e-02 + 1.789e-02 2.393e-02 3.151e-02 3.901e-02 4.385e-02 4.470e-02 + 1.770e-02 1.526e-02 1.770e-02 2.555e-02 3.718e-02 4.650e-02 4.821e-02 4.283e-02 3.469e-02 2.728e-02 2.171e-02 1.781e-02 1.538e-02 1.453e-02 1.538e-02 1.781e-02 2.171e-02 2.728e-02 - 3.469e-02 4.283e-02 4.821e-02 4.650e-02 3.719e-02 2.556e-02 - 1.527e-02 1.527e-02 2.162e-02 3.399e-02 4.740e-02 5.442e-02 + 3.469e-02 4.283e-02 4.821e-02 4.650e-02 3.718e-02 2.555e-02 + 1.526e-02 1.526e-02 2.162e-02 3.399e-02 4.740e-02 5.442e-02 5.237e-02 4.466e-02 3.604e-02 2.897e-02 2.375e-02 2.015e-02 1.822e-02 1.822e-02 2.015e-02 2.375e-02 2.897e-02 3.604e-02 4.466e-02 5.237e-02 5.442e-02 4.740e-02 3.399e-02 2.162e-02 - 1.770e-02 2.162e-02 3.287e-02 4.765e-02 5.845e-02 6.009e-02 + 1.770e-02 2.162e-02 3.287e-02 4.765e-02 5.844e-02 6.009e-02 5.398e-02 4.502e-02 3.681e-02 3.042e-02 2.570e-02 2.263e-02 2.153e-02 2.263e-02 2.570e-02 3.042e-02 3.681e-02 4.502e-02 - 5.398e-02 6.009e-02 5.845e-02 4.765e-02 3.287e-02 2.162e-02 - 2.556e-02 3.399e-02 4.765e-02 5.986e-02 6.446e-02 6.074e-02 + 5.398e-02 6.009e-02 5.844e-02 4.765e-02 3.287e-02 2.162e-02 + 2.555e-02 3.399e-02 4.765e-02 5.985e-02 6.446e-02 6.074e-02 5.260e-02 4.411e-02 3.705e-02 3.153e-02 2.748e-02 2.524e-02 2.524e-02 2.748e-02 3.153e-02 3.705e-02 4.411e-02 5.260e-02 - 6.074e-02 6.446e-02 5.986e-02 4.765e-02 3.399e-02 2.556e-02 - 3.719e-02 4.740e-02 5.845e-02 6.446e-02 6.323e-02 5.700e-02 + 6.074e-02 6.446e-02 5.985e-02 4.765e-02 3.399e-02 2.555e-02 + 3.718e-02 4.740e-02 5.844e-02 6.446e-02 6.323e-02 5.700e-02 4.939e-02 4.252e-02 3.681e-02 3.221e-02 2.903e-02 2.787e-02 2.903e-02 3.221e-02 3.681e-02 4.252e-02 4.939e-02 5.700e-02 - 6.323e-02 6.446e-02 5.845e-02 4.740e-02 3.719e-02 3.313e-02 + 6.323e-02 6.446e-02 5.844e-02 4.740e-02 3.718e-02 3.313e-02 4.650e-02 5.442e-02 6.009e-02 6.074e-02 5.700e-02 5.133e-02 4.565e-02 4.056e-02 3.604e-02 3.238e-02 3.025e-02 3.025e-02 3.238e-02 3.604e-02 4.056e-02 4.565e-02 5.133e-02 5.700e-02 - 6.074e-02 6.009e-02 5.442e-02 4.650e-02 4.091e-02 4.091e-02 + 6.074e-02 6.009e-02 5.442e-02 4.650e-02 4.090e-02 4.090e-02 4.821e-02 5.237e-02 5.398e-02 5.260e-02 4.939e-02 4.565e-02 4.192e-02 3.820e-02 3.469e-02 3.206e-02 3.107e-02 3.206e-02 3.469e-02 3.820e-02 4.192e-02 4.565e-02 4.939e-02 5.260e-02 - 5.398e-02 5.237e-02 4.821e-02 4.385e-02 4.201e-02 4.385e-02 + 5.398e-02 5.237e-02 4.821e-02 4.385e-02 4.200e-02 4.385e-02 4.283e-02 4.466e-02 4.502e-02 4.411e-02 4.252e-02 4.056e-02 3.820e-02 3.552e-02 3.303e-02 3.151e-02 3.151e-02 3.303e-02 3.552e-02 3.820e-02 4.056e-02 4.252e-02 4.411e-02 4.502e-02 @@ -2161,102 +2161,102 @@ Ionic_Step 1 Cubefile created from ABACUS. Inner loop is z, followed by y and x 2.171e-02 2.375e-02 2.570e-02 2.748e-02 2.903e-02 3.025e-02 3.107e-02 3.151e-02 3.165e-02 3.151e-02 3.107e-02 3.025e-02 2.903e-02 2.748e-02 2.570e-02 2.375e-02 2.171e-02 1.975e-02 - 1.816e-02 1.717e-02 1.684e-02 1.717e-02 1.816e-02 1.975e-02 + 1.816e-02 1.716e-02 1.683e-02 1.716e-02 1.816e-02 1.975e-02 1.781e-02 2.015e-02 2.263e-02 2.524e-02 2.787e-02 3.025e-02 3.206e-02 3.303e-02 3.303e-02 3.206e-02 3.025e-02 2.787e-02 2.524e-02 2.263e-02 2.015e-02 1.781e-02 1.566e-02 1.386e-02 1.262e-02 1.201e-02 1.201e-02 1.262e-02 1.386e-02 1.566e-02 1.538e-02 1.822e-02 2.153e-02 2.524e-02 2.903e-02 3.238e-02 3.469e-02 3.552e-02 3.469e-02 3.238e-02 2.903e-02 2.524e-02 - 2.153e-02 1.822e-02 1.538e-02 1.297e-02 1.100e-02 9.563e-03 - 8.723e-03 8.453e-03 8.723e-03 9.563e-03 1.100e-02 1.297e-02 + 2.153e-02 1.822e-02 1.538e-02 1.297e-02 1.100e-02 9.562e-03 + 8.723e-03 8.453e-03 8.723e-03 9.562e-03 1.100e-02 1.297e-02 1.453e-02 1.822e-02 2.263e-02 2.748e-02 3.221e-02 3.604e-02 3.820e-02 3.820e-02 3.604e-02 3.221e-02 2.748e-02 2.263e-02 1.822e-02 1.453e-02 1.158e-02 9.285e-03 7.626e-03 6.580e-03 - 6.091e-03 6.091e-03 6.580e-03 7.626e-03 9.285e-03 1.158e-02 + 6.090e-03 6.090e-03 6.580e-03 7.626e-03 9.285e-03 1.158e-02 1.538e-02 2.015e-02 2.570e-02 3.153e-02 3.681e-02 4.056e-02 4.192e-02 4.056e-02 3.681e-02 3.153e-02 2.570e-02 2.015e-02 - 1.538e-02 1.158e-02 8.703e-03 6.662e-03 5.346e-03 4.638e-03 - 4.420e-03 4.638e-03 5.346e-03 6.662e-03 8.703e-03 1.158e-02 + 1.538e-02 1.158e-02 8.704e-03 6.662e-03 5.346e-03 4.638e-03 + 4.420e-03 4.638e-03 5.346e-03 6.662e-03 8.704e-03 1.158e-02 1.781e-02 2.375e-02 3.042e-02 3.705e-02 4.252e-02 4.565e-02 4.565e-02 4.252e-02 3.705e-02 3.042e-02 2.375e-02 1.781e-02 - 1.297e-02 9.285e-03 6.662e-03 4.940e-03 3.945e-03 3.506e-03 - 3.506e-03 3.945e-03 4.940e-03 6.662e-03 9.285e-03 1.297e-02 + 1.297e-02 9.285e-03 6.662e-03 4.939e-03 3.944e-03 3.506e-03 + 3.506e-03 3.944e-03 4.939e-03 6.662e-03 9.285e-03 1.297e-02 2.171e-02 2.897e-02 3.681e-02 4.411e-02 4.939e-02 5.133e-02 4.939e-02 4.411e-02 3.681e-02 2.897e-02 2.171e-02 1.566e-02 - 1.100e-02 7.626e-03 5.346e-03 3.945e-03 3.218e-03 2.998e-03 - 3.218e-03 3.945e-03 5.346e-03 7.626e-03 1.100e-02 1.566e-02 + 1.100e-02 7.626e-03 5.346e-03 3.944e-03 3.218e-03 2.998e-03 + 3.218e-03 3.944e-03 5.346e-03 7.626e-03 1.100e-02 1.566e-02 2.728e-02 3.604e-02 4.502e-02 5.260e-02 5.700e-02 5.700e-02 5.260e-02 4.502e-02 3.604e-02 2.728e-02 1.975e-02 1.386e-02 - 9.563e-03 6.580e-03 4.638e-03 3.506e-03 2.998e-03 2.998e-03 - 3.506e-03 4.638e-03 6.580e-03 9.563e-03 1.386e-02 1.975e-02 + 9.562e-03 6.580e-03 4.638e-03 3.506e-03 2.998e-03 2.998e-03 + 3.506e-03 4.638e-03 6.580e-03 9.562e-03 1.386e-02 1.975e-02 3.469e-02 4.466e-02 5.398e-02 6.074e-02 6.323e-02 6.074e-02 5.398e-02 4.466e-02 3.469e-02 2.558e-02 1.816e-02 1.262e-02 - 8.723e-03 6.091e-03 4.420e-03 3.506e-03 3.218e-03 3.506e-03 - 4.420e-03 6.091e-03 8.723e-03 1.262e-02 1.816e-02 2.558e-02 + 8.723e-03 6.090e-03 4.420e-03 3.506e-03 3.218e-03 3.506e-03 + 4.420e-03 6.090e-03 8.723e-03 1.262e-02 1.816e-02 2.558e-02 4.283e-02 5.237e-02 6.009e-02 6.446e-02 6.446e-02 6.009e-02 - 5.237e-02 4.283e-02 3.303e-02 2.422e-02 1.717e-02 1.201e-02 - 8.453e-03 6.091e-03 4.638e-03 3.945e-03 3.945e-03 4.638e-03 - 6.091e-03 8.453e-03 1.201e-02 1.717e-02 2.422e-02 3.303e-02 - 4.821e-02 5.442e-02 5.845e-02 5.986e-02 5.845e-02 5.442e-02 - 4.821e-02 4.032e-02 3.165e-02 2.348e-02 1.684e-02 1.201e-02 - 8.723e-03 6.580e-03 5.346e-03 4.940e-03 5.346e-03 6.580e-03 - 8.723e-03 1.201e-02 1.684e-02 2.348e-02 3.165e-02 4.032e-02 + 5.237e-02 4.283e-02 3.303e-02 2.422e-02 1.716e-02 1.201e-02 + 8.453e-03 6.090e-03 4.638e-03 3.944e-03 3.944e-03 4.638e-03 + 6.090e-03 8.453e-03 1.201e-02 1.716e-02 2.422e-02 3.303e-02 + 4.821e-02 5.442e-02 5.844e-02 5.985e-02 5.844e-02 5.442e-02 + 4.821e-02 4.032e-02 3.165e-02 2.348e-02 1.683e-02 1.201e-02 + 8.723e-03 6.580e-03 5.346e-03 4.939e-03 5.346e-03 6.580e-03 + 8.723e-03 1.201e-02 1.683e-02 2.348e-02 3.165e-02 4.032e-02 4.650e-02 4.740e-02 4.765e-02 4.765e-02 4.740e-02 4.650e-02 - 4.385e-02 3.852e-02 3.112e-02 2.348e-02 1.717e-02 1.262e-02 - 9.563e-03 7.626e-03 6.662e-03 6.662e-03 7.626e-03 9.563e-03 - 1.262e-02 1.717e-02 2.348e-02 3.112e-02 3.852e-02 4.385e-02 - 3.719e-02 3.399e-02 3.287e-02 3.399e-02 3.719e-02 4.091e-02 - 4.201e-02 3.852e-02 3.165e-02 2.422e-02 1.816e-02 1.386e-02 - 1.100e-02 9.285e-03 8.703e-03 9.285e-03 1.100e-02 1.386e-02 - 1.816e-02 2.422e-02 3.165e-02 3.852e-02 4.201e-02 4.091e-02 - 2.556e-02 2.162e-02 2.162e-02 2.556e-02 3.313e-02 4.091e-02 + 4.385e-02 3.852e-02 3.112e-02 2.348e-02 1.716e-02 1.262e-02 + 9.562e-03 7.626e-03 6.662e-03 6.662e-03 7.626e-03 9.562e-03 + 1.262e-02 1.716e-02 2.348e-02 3.112e-02 3.852e-02 4.385e-02 + 3.718e-02 3.399e-02 3.287e-02 3.399e-02 3.718e-02 4.090e-02 + 4.200e-02 3.852e-02 3.165e-02 2.422e-02 1.816e-02 1.386e-02 + 1.100e-02 9.285e-03 8.704e-03 9.285e-03 1.100e-02 1.386e-02 + 1.816e-02 2.422e-02 3.165e-02 3.852e-02 4.200e-02 4.090e-02 + 2.555e-02 2.162e-02 2.162e-02 2.555e-02 3.313e-02 4.090e-02 4.385e-02 4.032e-02 3.303e-02 2.558e-02 1.975e-02 1.566e-02 1.297e-02 1.158e-02 1.158e-02 1.297e-02 1.566e-02 1.975e-02 - 2.558e-02 3.303e-02 4.032e-02 4.385e-02 4.091e-02 3.313e-02 - 4.526e-03 4.526e-03 1.115e-02 2.556e-02 4.294e-02 5.397e-02 + 2.558e-02 3.303e-02 4.032e-02 4.385e-02 4.090e-02 3.313e-02 + 4.524e-03 4.524e-03 1.114e-02 2.555e-02 4.294e-02 5.396e-02 5.421e-02 4.707e-02 3.820e-02 3.074e-02 2.524e-02 2.145e-02 1.942e-02 1.942e-02 2.145e-02 2.524e-02 3.074e-02 3.820e-02 - 4.707e-02 5.421e-02 5.397e-02 4.294e-02 2.556e-02 1.115e-02 - 4.526e-03 8.601e-03 2.162e-02 4.108e-02 5.775e-02 6.353e-02 - 5.881e-02 4.953e-02 4.056e-02 3.354e-02 2.838e-02 2.502e-02 - 2.381e-02 2.502e-02 2.838e-02 3.354e-02 4.056e-02 4.953e-02 - 5.881e-02 6.353e-02 5.775e-02 4.108e-02 2.162e-02 8.601e-03 - 1.115e-02 2.162e-02 4.041e-02 5.968e-02 7.000e-02 6.863e-02 - 6.025e-02 5.061e-02 4.252e-02 3.625e-02 3.163e-02 2.904e-02 - 2.904e-02 3.163e-02 3.625e-02 4.252e-02 5.061e-02 6.025e-02 - 6.863e-02 7.000e-02 5.968e-02 4.041e-02 2.162e-02 1.115e-02 - 2.556e-02 4.108e-02 5.968e-02 7.233e-02 7.430e-02 6.809e-02 + 4.707e-02 5.421e-02 5.396e-02 4.294e-02 2.555e-02 1.114e-02 + 4.524e-03 8.599e-03 2.162e-02 4.108e-02 5.775e-02 6.353e-02 + 5.881e-02 4.953e-02 4.056e-02 3.355e-02 2.839e-02 2.502e-02 + 2.381e-02 2.502e-02 2.839e-02 3.355e-02 4.056e-02 4.953e-02 + 5.881e-02 6.353e-02 5.775e-02 4.108e-02 2.162e-02 8.599e-03 + 1.114e-02 2.162e-02 4.040e-02 5.968e-02 7.000e-02 6.864e-02 + 6.025e-02 5.062e-02 4.252e-02 3.625e-02 3.163e-02 2.904e-02 + 2.904e-02 3.163e-02 3.625e-02 4.252e-02 5.062e-02 6.025e-02 + 6.864e-02 7.000e-02 5.968e-02 4.040e-02 2.162e-02 1.114e-02 + 2.555e-02 4.108e-02 5.968e-02 7.233e-02 7.430e-02 6.809e-02 5.911e-02 5.086e-02 4.411e-02 3.864e-02 3.477e-02 3.333e-02 3.477e-02 3.864e-02 4.411e-02 5.086e-02 5.911e-02 6.809e-02 - 7.430e-02 7.233e-02 5.968e-02 4.108e-02 2.556e-02 1.973e-02 - 4.294e-02 5.775e-02 7.000e-02 7.430e-02 7.099e-02 6.404e-02 - 5.689e-02 5.061e-02 4.502e-02 4.029e-02 3.742e-02 3.742e-02 - 4.029e-02 4.502e-02 5.061e-02 5.689e-02 6.404e-02 7.099e-02 + 7.430e-02 7.233e-02 5.968e-02 4.108e-02 2.555e-02 1.973e-02 + 4.294e-02 5.775e-02 7.000e-02 7.430e-02 7.099e-02 6.405e-02 + 5.690e-02 5.062e-02 4.502e-02 4.029e-02 3.742e-02 3.742e-02 + 4.029e-02 4.502e-02 5.062e-02 5.690e-02 6.405e-02 7.099e-02 7.430e-02 7.000e-02 5.775e-02 4.294e-02 3.313e-02 3.313e-02 - 5.397e-02 6.353e-02 6.863e-02 6.809e-02 6.404e-02 5.911e-02 + 5.396e-02 6.353e-02 6.864e-02 6.809e-02 6.405e-02 5.911e-02 5.434e-02 4.953e-02 4.466e-02 4.073e-02 3.920e-02 4.073e-02 - 4.466e-02 4.953e-02 5.434e-02 5.911e-02 6.404e-02 6.809e-02 - 6.863e-02 6.353e-02 5.397e-02 4.471e-02 4.091e-02 4.471e-02 - 5.421e-02 5.881e-02 6.025e-02 5.911e-02 5.689e-02 5.434e-02 - 5.116e-02 4.707e-02 4.283e-02 4.006e-02 4.006e-02 4.283e-02 - 4.707e-02 5.116e-02 5.434e-02 5.689e-02 5.911e-02 6.025e-02 + 4.466e-02 4.953e-02 5.434e-02 5.911e-02 6.405e-02 6.809e-02 + 6.864e-02 6.353e-02 5.396e-02 4.470e-02 4.090e-02 4.470e-02 + 5.421e-02 5.881e-02 6.025e-02 5.911e-02 5.690e-02 5.434e-02 + 5.116e-02 4.707e-02 4.283e-02 4.005e-02 4.005e-02 4.283e-02 + 4.707e-02 5.116e-02 5.434e-02 5.690e-02 5.911e-02 6.025e-02 5.881e-02 5.421e-02 4.814e-02 4.385e-02 4.385e-02 4.814e-02 - 4.707e-02 4.953e-02 5.061e-02 5.086e-02 5.061e-02 4.953e-02 + 4.707e-02 4.953e-02 5.062e-02 5.086e-02 5.062e-02 4.953e-02 4.707e-02 4.355e-02 4.032e-02 3.901e-02 4.032e-02 4.355e-02 - 4.707e-02 4.953e-02 5.061e-02 5.086e-02 5.061e-02 4.953e-02 + 4.707e-02 4.953e-02 5.062e-02 5.086e-02 5.062e-02 4.953e-02 4.707e-02 4.355e-02 4.032e-02 3.901e-02 4.032e-02 4.355e-02 3.820e-02 4.056e-02 4.252e-02 4.411e-02 4.502e-02 4.466e-02 4.283e-02 4.032e-02 3.852e-02 3.852e-02 4.032e-02 4.283e-02 4.466e-02 4.502e-02 4.411e-02 4.252e-02 4.056e-02 3.820e-02 3.552e-02 3.303e-02 3.151e-02 3.151e-02 3.303e-02 3.552e-02 - 3.074e-02 3.354e-02 3.625e-02 3.864e-02 4.029e-02 4.073e-02 - 4.006e-02 3.901e-02 3.852e-02 3.901e-02 4.006e-02 4.073e-02 - 4.029e-02 3.864e-02 3.625e-02 3.354e-02 3.074e-02 2.799e-02 + 3.074e-02 3.355e-02 3.625e-02 3.864e-02 4.029e-02 4.073e-02 + 4.005e-02 3.901e-02 3.852e-02 3.901e-02 4.005e-02 4.073e-02 + 4.029e-02 3.864e-02 3.625e-02 3.355e-02 3.074e-02 2.799e-02 2.558e-02 2.393e-02 2.335e-02 2.393e-02 2.558e-02 2.799e-02 - 2.524e-02 2.838e-02 3.163e-02 3.477e-02 3.742e-02 3.920e-02 - 4.006e-02 4.032e-02 4.032e-02 4.006e-02 3.920e-02 3.742e-02 - 3.477e-02 3.163e-02 2.838e-02 2.524e-02 2.230e-02 1.975e-02 + 2.524e-02 2.839e-02 3.163e-02 3.477e-02 3.742e-02 3.920e-02 + 4.005e-02 4.032e-02 4.032e-02 4.005e-02 3.920e-02 3.742e-02 + 3.477e-02 3.163e-02 2.839e-02 2.524e-02 2.230e-02 1.975e-02 1.789e-02 1.693e-02 1.693e-02 1.789e-02 1.975e-02 2.230e-02 2.145e-02 2.502e-02 2.904e-02 3.333e-02 3.742e-02 4.073e-02 4.283e-02 4.355e-02 4.283e-02 4.073e-02 3.742e-02 3.333e-02 @@ -2269,44 +2269,44 @@ Ionic_Step 1 Cubefile created from ABACUS. Inner loop is z, followed by y and x 1.942e-02 2.502e-02 3.163e-02 3.864e-02 4.502e-02 4.953e-02 5.116e-02 4.953e-02 4.502e-02 3.864e-02 3.163e-02 2.502e-02 1.942e-02 1.498e-02 1.158e-02 9.073e-03 7.387e-03 6.445e-03 - 6.148e-03 6.445e-03 7.387e-03 9.073e-03 1.158e-02 1.498e-02 - 2.145e-02 2.838e-02 3.625e-02 4.411e-02 5.061e-02 5.434e-02 - 5.434e-02 5.061e-02 4.411e-02 3.625e-02 2.838e-02 2.145e-02 + 6.147e-03 6.445e-03 7.387e-03 9.073e-03 1.158e-02 1.498e-02 + 2.145e-02 2.839e-02 3.625e-02 4.411e-02 5.062e-02 5.434e-02 + 5.434e-02 5.062e-02 4.411e-02 3.625e-02 2.839e-02 2.145e-02 1.584e-02 1.158e-02 8.494e-03 6.424e-03 5.200e-03 4.651e-03 4.651e-03 5.200e-03 6.424e-03 8.494e-03 1.158e-02 1.584e-02 - 2.524e-02 3.354e-02 4.252e-02 5.086e-02 5.689e-02 5.911e-02 - 5.689e-02 5.086e-02 4.252e-02 3.354e-02 2.524e-02 1.832e-02 + 2.524e-02 3.355e-02 4.252e-02 5.086e-02 5.690e-02 5.911e-02 + 5.690e-02 5.086e-02 4.252e-02 3.355e-02 2.524e-02 1.832e-02 1.297e-02 9.073e-03 6.424e-03 4.792e-03 3.948e-03 3.695e-03 3.948e-03 4.792e-03 6.424e-03 9.073e-03 1.297e-02 1.832e-02 - 3.074e-02 4.056e-02 5.061e-02 5.911e-02 6.404e-02 6.404e-02 - 5.911e-02 5.061e-02 4.056e-02 3.074e-02 2.230e-02 1.566e-02 + 3.074e-02 4.056e-02 5.062e-02 5.911e-02 6.405e-02 6.405e-02 + 5.911e-02 5.062e-02 4.056e-02 3.074e-02 2.230e-02 1.566e-02 1.078e-02 7.387e-03 5.200e-03 3.948e-03 3.399e-03 3.399e-03 3.948e-03 5.200e-03 7.387e-03 1.078e-02 1.566e-02 2.230e-02 3.820e-02 4.953e-02 6.025e-02 6.809e-02 7.099e-02 6.809e-02 6.025e-02 4.953e-02 3.820e-02 2.799e-02 1.975e-02 1.363e-02 9.326e-03 6.445e-03 4.651e-03 3.695e-03 3.399e-03 3.695e-03 4.651e-03 6.445e-03 9.326e-03 1.363e-02 1.975e-02 2.799e-02 - 4.707e-02 5.881e-02 6.863e-02 7.430e-02 7.430e-02 6.863e-02 + 4.707e-02 5.881e-02 6.864e-02 7.430e-02 7.430e-02 6.864e-02 5.881e-02 4.707e-02 3.552e-02 2.558e-02 1.789e-02 1.238e-02 - 8.621e-03 6.148e-03 4.651e-03 3.948e-03 3.948e-03 4.651e-03 - 6.148e-03 8.621e-03 1.238e-02 1.789e-02 2.558e-02 3.552e-02 + 8.621e-03 6.147e-03 4.651e-03 3.948e-03 3.948e-03 4.651e-03 + 6.147e-03 8.621e-03 1.238e-02 1.789e-02 2.558e-02 3.552e-02 5.421e-02 6.353e-02 7.000e-02 7.233e-02 7.000e-02 6.353e-02 5.421e-02 4.355e-02 3.303e-02 2.393e-02 1.693e-02 1.197e-02 8.621e-03 6.445e-03 5.200e-03 4.792e-03 5.200e-03 6.445e-03 8.621e-03 1.197e-02 1.693e-02 2.393e-02 3.303e-02 4.355e-02 - 5.397e-02 5.775e-02 5.968e-02 5.968e-02 5.775e-02 5.397e-02 + 5.396e-02 5.775e-02 5.968e-02 5.968e-02 5.775e-02 5.396e-02 4.814e-02 4.032e-02 3.151e-02 2.335e-02 1.693e-02 1.238e-02 9.326e-03 7.387e-03 6.424e-03 6.424e-03 7.387e-03 9.326e-03 1.238e-02 1.693e-02 2.335e-02 3.151e-02 4.032e-02 4.814e-02 - 4.294e-02 4.108e-02 4.041e-02 4.108e-02 4.294e-02 4.471e-02 + 4.294e-02 4.108e-02 4.040e-02 4.108e-02 4.294e-02 4.470e-02 4.385e-02 3.901e-02 3.151e-02 2.393e-02 1.789e-02 1.363e-02 1.078e-02 9.073e-03 8.494e-03 9.073e-03 1.078e-02 1.363e-02 - 1.789e-02 2.393e-02 3.151e-02 3.901e-02 4.385e-02 4.471e-02 - 2.556e-02 2.162e-02 2.162e-02 2.556e-02 3.313e-02 4.091e-02 + 1.789e-02 2.393e-02 3.151e-02 3.901e-02 4.385e-02 4.470e-02 + 2.555e-02 2.162e-02 2.162e-02 2.555e-02 3.313e-02 4.090e-02 4.385e-02 4.032e-02 3.303e-02 2.558e-02 1.975e-02 1.566e-02 1.297e-02 1.158e-02 1.158e-02 1.297e-02 1.566e-02 1.975e-02 - 2.558e-02 3.303e-02 4.032e-02 4.385e-02 4.091e-02 3.313e-02 - 1.115e-02 8.601e-03 1.115e-02 1.973e-02 3.313e-02 4.471e-02 + 2.558e-02 3.303e-02 4.032e-02 4.385e-02 4.090e-02 3.313e-02 + 1.114e-02 8.599e-03 1.114e-02 1.973e-02 3.313e-02 4.470e-02 4.814e-02 4.355e-02 3.552e-02 2.799e-02 2.230e-02 1.832e-02 1.584e-02 1.498e-02 1.584e-02 1.832e-02 2.230e-02 2.799e-02 - 3.552e-02 4.355e-02 4.814e-02 4.471e-02 3.313e-02 1.973e-02 + 3.552e-02 4.355e-02 4.814e-02 4.470e-02 3.313e-02 1.973e-02 diff --git a/tests/01_PW/scf_out_chg_tau/tau.cube.ref b/tests/01_PW/scf_out_chg_tau/tau.cube.ref index 3a0219a8985..adaf19d4091 100644 --- a/tests/01_PW/scf_out_chg_tau/tau.cube.ref +++ b/tests/01_PW/scf_out_chg_tau/tau.cube.ref @@ -4,2309 +4,2309 @@ Ionic_Step 1 Cubefile created from ABACUS. Inner loop is z, followed by y and x 24 0.000000 0.212500 0.212500 24 0.212500 0.000000 0.212500 24 0.212500 0.212500 0.000000 - 14 4 0.000000 0.000000 0.000000 - 14 4 2.550000 2.550000 2.550000 - 9.35850333253e-02 1.05271425072e-01 1.14984029929e-01 9.46618817553e-02 6.13624406409e-02 4.15726567626e-02 - 3.48683159917e-02 3.06058831029e-02 2.62414819209e-02 2.33118163733e-02 2.21386056500e-02 2.19611077293e-02 - 2.19947599998e-02 2.19611077293e-02 2.21386056500e-02 2.33118163733e-02 2.62414819209e-02 3.06058831029e-02 - 3.48683159917e-02 4.15726567626e-02 6.13624406410e-02 9.46618817554e-02 1.14984029929e-01 1.05271425072e-01 - 1.05271425072e-01 1.14284409879e-01 1.02255317231e-01 6.87950359263e-02 4.23899857519e-02 3.31551669202e-02 - 3.03789244340e-02 2.75584598461e-02 2.55791633527e-02 2.52445347088e-02 2.59181147281e-02 2.65858530209e-02 - 2.65858530209e-02 2.59181147281e-02 2.52445347088e-02 2.55791633527e-02 2.75584598461e-02 3.03789244340e-02 - 3.31551669202e-02 4.23899857520e-02 6.87950359263e-02 1.02255317231e-01 1.14284409879e-01 1.05271425072e-01 - 1.14984029929e-01 1.02255317231e-01 7.14595557376e-02 4.26148361403e-02 3.05273339684e-02 2.81921950446e-02 - 2.69081613820e-02 2.62352804996e-02 2.71142998602e-02 2.89473471924e-02 3.06728014251e-02 3.13656133728e-02 - 3.06728014251e-02 2.89473471924e-02 2.71142998602e-02 2.62352804996e-02 2.69081613820e-02 2.81921950446e-02 - 3.05273339684e-02 4.26148361403e-02 7.14595557377e-02 1.02255317231e-01 1.14984029929e-01 1.15476799016e-01 - 9.46618817553e-02 6.87950359263e-02 4.26148361403e-02 2.92890570903e-02 2.61164989591e-02 2.52955690008e-02 - 2.54927185045e-02 2.74769895380e-02 3.04773481040e-02 3.33647463124e-02 3.51262182185e-02 3.51262182185e-02 - 3.33647463124e-02 3.04773481040e-02 2.74769895380e-02 2.54927185045e-02 2.52955690008e-02 2.61164989591e-02 - 2.92890570903e-02 4.26148361403e-02 6.87950359264e-02 9.46618817554e-02 1.07504612680e-01 1.07504612679e-01 - 6.13624406409e-02 4.23899857519e-02 3.05273339684e-02 2.61164989591e-02 2.44579386769e-02 2.45139707404e-02 - 2.69081613820e-02 3.06792139923e-02 3.45751625707e-02 3.75326171916e-02 3.86293456568e-02 3.75326171916e-02 - 3.45751625707e-02 3.06792139923e-02 2.69081613820e-02 2.45139707404e-02 2.44579386769e-02 2.61164989591e-02 - 3.05273339684e-02 4.23899857520e-02 6.13624406410e-02 7.81193802830e-02 8.44119137021e-02 7.81193802830e-02 - 4.15726567626e-02 3.31551669202e-02 2.81921950446e-02 2.52955690008e-02 2.45139707404e-02 2.65224863971e-02 - 3.03789244340e-02 3.48628450776e-02 3.89798518706e-02 4.14583134664e-02 4.14583134664e-02 3.89798518705e-02 - 3.48628450776e-02 3.03789244340e-02 2.65224863971e-02 2.45139707404e-02 2.52955690008e-02 2.81921950446e-02 - 3.31551669202e-02 4.15726567627e-02 5.18158087398e-02 5.87902882083e-02 5.87902882083e-02 5.18158087397e-02 - 3.48683159917e-02 3.03789244340e-02 2.69081613820e-02 2.54927185045e-02 2.69081613820e-02 3.03789244340e-02 - 3.48683159917e-02 3.97328533271e-02 4.36605690195e-02 4.51578187598e-02 4.36605690195e-02 3.97328533271e-02 - 3.48683159917e-02 3.03789244340e-02 2.69081613820e-02 2.54927185045e-02 2.69081613820e-02 3.03789244340e-02 - 3.48683159917e-02 3.97328533271e-02 4.36605690195e-02 4.51578187598e-02 4.36605690195e-02 3.97328533270e-02 - 3.06058831029e-02 2.75584598461e-02 2.62352804996e-02 2.74769895380e-02 3.06792139923e-02 3.48628450776e-02 - 3.97328533271e-02 4.45490987358e-02 4.76134439312e-02 4.76134439312e-02 4.45490987358e-02 3.97328533271e-02 - 3.48628450776e-02 3.06792139923e-02 2.74769895380e-02 2.62352804996e-02 2.75584598461e-02 3.06058831029e-02 - 3.38925854809e-02 3.62673468746e-02 3.73540502296e-02 3.73540502296e-02 3.62673468745e-02 3.38925854809e-02 - 2.62414819209e-02 2.55791633527e-02 2.71142998602e-02 3.04773481040e-02 3.45751625707e-02 3.89798518706e-02 - 4.36605690195e-02 4.76134439312e-02 4.91904413241e-02 4.76134439312e-02 4.36605690195e-02 3.89798518706e-02 - 3.45751625707e-02 3.04773481040e-02 2.71142998602e-02 2.55791633527e-02 2.62414819209e-02 2.81653274579e-02 - 2.99301385615e-02 3.07822057975e-02 3.09609358381e-02 3.07822057975e-02 2.99301385615e-02 2.81653274579e-02 - 2.33118163733e-02 2.52445347087e-02 2.89473471924e-02 3.33647463124e-02 3.75326171916e-02 4.14583134665e-02 - 4.51578187598e-02 4.76134439313e-02 4.76134439313e-02 4.51578187598e-02 4.14583134665e-02 3.75326171916e-02 - 3.33647463124e-02 2.89473471924e-02 2.52445347088e-02 2.33118163733e-02 2.31387869200e-02 2.37301976795e-02 - 2.40967233573e-02 2.40904203272e-02 2.40904203272e-02 2.40967233573e-02 2.37301976795e-02 2.31387869200e-02 - 2.21386056500e-02 2.59181147280e-02 3.06728014251e-02 3.51262182185e-02 3.86293456568e-02 4.14583134665e-02 - 4.36605690195e-02 4.45490987358e-02 4.36605690195e-02 4.14583134665e-02 3.86293456568e-02 3.51262182185e-02 - 3.06728014251e-02 2.59181147281e-02 2.21386056500e-02 1.99975947255e-02 1.91106081253e-02 1.86355176327e-02 - 1.81617046076e-02 1.79266526907e-02 1.81617046076e-02 1.86355176327e-02 1.91106081253e-02 1.99975947255e-02 - 2.19611077293e-02 2.65858530209e-02 3.13656133727e-02 3.51262182185e-02 3.75326171916e-02 3.89798518706e-02 - 3.97328533271e-02 3.97328533271e-02 3.89798518706e-02 3.75326171916e-02 3.51262182185e-02 3.13656133727e-02 - 2.65858530209e-02 2.19611077293e-02 1.85132763207e-02 1.63906429721e-02 1.50739196267e-02 1.40806383265e-02 - 1.34421324248e-02 1.34421324248e-02 1.40806383265e-02 1.50739196267e-02 1.63906429721e-02 1.85132763207e-02 - 2.19947599998e-02 2.65858530209e-02 3.06728014251e-02 3.33647463124e-02 3.45751625707e-02 3.48628450776e-02 - 3.48683159917e-02 3.48628450776e-02 3.45751625707e-02 3.33647463124e-02 3.06728014251e-02 2.65858530209e-02 - 2.19947599998e-02 1.79741436760e-02 1.50424663757e-02 1.30485807277e-02 1.16095381953e-02 1.06051613742e-02 - 1.02248872526e-02 1.06051613742e-02 1.16095381953e-02 1.30485807277e-02 1.50424663757e-02 1.79741436760e-02 - 2.19611077293e-02 2.59181147281e-02 2.89473471924e-02 3.04773481040e-02 3.06792139923e-02 3.03789244340e-02 - 3.03789244340e-02 3.06792139923e-02 3.04773481040e-02 2.89473471924e-02 2.59181147281e-02 2.19611077293e-02 - 1.79741436760e-02 1.46288647205e-02 1.21045128275e-02 1.02401100733e-02 8.89920278515e-03 8.15758815907e-03 - 8.15758815907e-03 8.89920278515e-03 1.02401100733e-02 1.21045128275e-02 1.46288647205e-02 1.79741436760e-02 - 2.21386056500e-02 2.52445347088e-02 2.71142998602e-02 2.74769895380e-02 2.69081613821e-02 2.65224863971e-02 - 2.69081613821e-02 2.74769895380e-02 2.71142998602e-02 2.52445347088e-02 2.21386056500e-02 1.85132763207e-02 - 1.50424663757e-02 1.21045128275e-02 9.78219490488e-03 8.04455907974e-03 6.92803651719e-03 6.53729803366e-03 - 6.92803651720e-03 8.04455907974e-03 9.78219490488e-03 1.21045128275e-02 1.50424663757e-02 1.85132763207e-02 - 2.33118163733e-02 2.55791633528e-02 2.62352804996e-02 2.54927185046e-02 2.45139707405e-02 2.45139707405e-02 - 2.54927185046e-02 2.62352804996e-02 2.55791633528e-02 2.33118163733e-02 1.99975947255e-02 1.63906429721e-02 - 1.30485807277e-02 1.02401100733e-02 8.04455907974e-03 6.50338402317e-03 5.69411798450e-03 5.69411798450e-03 - 6.50338402317e-03 8.04455907974e-03 1.02401100733e-02 1.30485807277e-02 1.63906429721e-02 1.99975947255e-02 - 2.62414819209e-02 2.75584598462e-02 2.69081613821e-02 2.52955690009e-02 2.44579386769e-02 2.52955690009e-02 - 2.69081613821e-02 2.75584598462e-02 2.62414819209e-02 2.31387869200e-02 1.91106081253e-02 1.50739196267e-02 - 1.16095381953e-02 8.89920278516e-03 6.92803651720e-03 5.69411798450e-03 5.26652136817e-03 5.69411798450e-03 - 6.92803651720e-03 8.89920278516e-03 1.16095381953e-02 1.50739196267e-02 1.91106081254e-02 2.31387869200e-02 - 3.06058831030e-02 3.03789244340e-02 2.81921950447e-02 2.61164989592e-02 2.61164989591e-02 2.81921950447e-02 - 3.03789244340e-02 3.06058831030e-02 2.81653274579e-02 2.37301976795e-02 1.86355176327e-02 1.40806383265e-02 - 1.06051613742e-02 8.15758815908e-03 6.53729803367e-03 5.69411798450e-03 5.69411798450e-03 6.53729803367e-03 - 8.15758815908e-03 1.06051613742e-02 1.40806383265e-02 1.86355176327e-02 2.37301976795e-02 2.81653274579e-02 - 3.48683159917e-02 3.31551669202e-02 3.05273339685e-02 2.92890570903e-02 3.05273339685e-02 3.31551669202e-02 - 3.48683159917e-02 3.38925854809e-02 2.99301385615e-02 2.40967233573e-02 1.81617046076e-02 1.34421324248e-02 - 1.02248872526e-02 8.15758815908e-03 6.92803651720e-03 6.50338402317e-03 6.92803651720e-03 8.15758815908e-03 - 1.02248872526e-02 1.34421324248e-02 1.81617046076e-02 2.40967233573e-02 2.99301385615e-02 3.38925854810e-02 - 4.15726567627e-02 4.23899857520e-02 4.26148361404e-02 4.26148361404e-02 4.23899857520e-02 4.15726567627e-02 - 3.97328533271e-02 3.62673468746e-02 3.07822057975e-02 2.40904203272e-02 1.79266526907e-02 1.34421324248e-02 - 1.06051613742e-02 8.89920278516e-03 8.04455907974e-03 8.04455907974e-03 8.89920278516e-03 1.06051613742e-02 - 1.34421324248e-02 1.79266526907e-02 2.40904203272e-02 3.07822057975e-02 3.62673468746e-02 3.97328533271e-02 - 6.13624406410e-02 6.87950359264e-02 7.14595557378e-02 6.87950359264e-02 6.13624406410e-02 5.18158087398e-02 - 4.36605690195e-02 3.73540502297e-02 3.09609358381e-02 2.40904203272e-02 1.81617046076e-02 1.40806383265e-02 - 1.16095381953e-02 1.02401100733e-02 9.78219490488e-03 1.02401100733e-02 1.16095381953e-02 1.40806383265e-02 - 1.81617046076e-02 2.40904203272e-02 3.09609358381e-02 3.73540502297e-02 4.36605690195e-02 5.18158087398e-02 - 9.46618817555e-02 1.02255317232e-01 1.02255317232e-01 9.46618817555e-02 7.81193802831e-02 5.87902882084e-02 - 4.51578187598e-02 3.73540502296e-02 3.07822057975e-02 2.40967233573e-02 1.86355176327e-02 1.50739196267e-02 - 1.30485807277e-02 1.21045128275e-02 1.21045128275e-02 1.30485807277e-02 1.50739196267e-02 1.86355176327e-02 - 2.40967233573e-02 3.07822057975e-02 3.73540502297e-02 4.51578187598e-02 5.87902882084e-02 7.81193802831e-02 - 1.14984029929e-01 1.14284409879e-01 1.14984029929e-01 1.07504612680e-01 8.44119137022e-02 5.87902882084e-02 - 4.36605690195e-02 3.62673468746e-02 2.99301385615e-02 2.37301976795e-02 1.91106081253e-02 1.63906429721e-02 - 1.50424663757e-02 1.46288647205e-02 1.50424663757e-02 1.63906429721e-02 1.91106081253e-02 2.37301976795e-02 - 2.99301385615e-02 3.62673468746e-02 4.36605690195e-02 5.87902882084e-02 8.44119137023e-02 1.07504612680e-01 - 1.05271425072e-01 1.05271425072e-01 1.15476799016e-01 1.07504612680e-01 7.81193802830e-02 5.18158087398e-02 - 3.97328533271e-02 3.38925854809e-02 2.81653274579e-02 2.31387869200e-02 1.99975947255e-02 1.85132763207e-02 - 1.79741436761e-02 1.79741436761e-02 1.85132763207e-02 1.99975947255e-02 2.31387869200e-02 2.81653274579e-02 - 3.38925854809e-02 3.97328533271e-02 5.18158087398e-02 7.81193802831e-02 1.07504612680e-01 1.15476799016e-01 - 1.05271425072e-01 1.14284409879e-01 1.02255317231e-01 6.87950359263e-02 4.23899857520e-02 3.31551669202e-02 - 3.03789244340e-02 2.75584598461e-02 2.55791633527e-02 2.52445347088e-02 2.59181147281e-02 2.65858530209e-02 - 2.65858530209e-02 2.59181147281e-02 2.52445347088e-02 2.55791633527e-02 2.75584598461e-02 3.03789244340e-02 - 3.31551669202e-02 4.23899857520e-02 6.87950359264e-02 1.02255317231e-01 1.14284409879e-01 1.05271425072e-01 - 1.14284409879e-01 1.05382836884e-01 7.44578962493e-02 4.30330806506e-02 2.95749315132e-02 2.73937018060e-02 - 2.65224863971e-02 2.61140936360e-02 2.72051400328e-02 2.92643382604e-02 3.12134113093e-02 3.20070729423e-02 - 3.12134113093e-02 2.92643382604e-02 2.72051400328e-02 2.61140936360e-02 2.65224863971e-02 2.73937018060e-02 - 2.95749315132e-02 4.30330806506e-02 7.44578962494e-02 1.05382836884e-01 1.14284409879e-01 1.12133375345e-01 - 1.02255317231e-01 7.44578962493e-02 4.31931066266e-02 2.67216511960e-02 2.35389825848e-02 2.36141349499e-02 - 2.45139707404e-02 2.70513354759e-02 3.05548189106e-02 3.39942859271e-02 3.61935553818e-02 3.61935553818e-02 - 3.39942859271e-02 3.05548189106e-02 2.70513354759e-02 2.45139707404e-02 2.36141349499e-02 2.35389825848e-02 - 2.67216511960e-02 4.31931066266e-02 7.44578962494e-02 1.02255317231e-01 1.13479602981e-01 1.13479602981e-01 - 6.87950359263e-02 4.30330806506e-02 2.67216511960e-02 2.17746576295e-02 2.10846196163e-02 2.20787139947e-02 - 2.52955690008e-02 2.97974895143e-02 3.46096179590e-02 3.86294987685e-02 4.02266025417e-02 3.86294987685e-02 - 3.46096179590e-02 2.97974895143e-02 2.52955690008e-02 2.20787139947e-02 2.10846196163e-02 2.17746576295e-02 - 2.67216511960e-02 4.30330806506e-02 6.87950359263e-02 9.00454509227e-02 9.74781930219e-02 9.00454509227e-02 - 4.23899857519e-02 2.95749315132e-02 2.35389825848e-02 2.10846196163e-02 2.09116171502e-02 2.36141349499e-02 - 2.81921950446e-02 3.39196724265e-02 4.00842207789e-02 4.42637285519e-02 4.42637285519e-02 4.00842207789e-02 - 3.39196724265e-02 2.81921950446e-02 2.36141349499e-02 2.09116171502e-02 2.10846196163e-02 2.35389825848e-02 - 2.95749315132e-02 4.23899857520e-02 5.87809730550e-02 6.98255714909e-02 6.98255714909e-02 5.87809730550e-02 - 3.31551669201e-02 2.73937018060e-02 2.36141349499e-02 2.20787139947e-02 2.36141349499e-02 2.73937018060e-02 - 3.31551669202e-02 4.11388919901e-02 4.86717503238e-02 5.17382028009e-02 4.86717503238e-02 4.11388919901e-02 - 3.31551669202e-02 2.73937018060e-02 2.36141349499e-02 2.20787139947e-02 2.36141349500e-02 2.73937018060e-02 - 3.31551669202e-02 4.11388919901e-02 4.86717503238e-02 5.17382028008e-02 4.86717503238e-02 4.11388919901e-02 - 3.03789244340e-02 2.65224863971e-02 2.45139707404e-02 2.52955690008e-02 2.81921950446e-02 3.31551669202e-02 - 4.15726567626e-02 5.18158087398e-02 5.87902882083e-02 5.87902882083e-02 5.18158087398e-02 4.15726567626e-02 - 3.31551669202e-02 2.81921950447e-02 2.52955690008e-02 2.45139707405e-02 2.65224863971e-02 3.03789244340e-02 - 3.48628450776e-02 3.89798518705e-02 4.14583134664e-02 4.14583134664e-02 3.89798518705e-02 3.48628450776e-02 - 2.75584598461e-02 2.61140936360e-02 2.70513354759e-02 2.97974895143e-02 3.39196724265e-02 4.11388919902e-02 - 5.18158087398e-02 6.17008685349e-02 6.56428313091e-02 6.17008685349e-02 5.18158087398e-02 4.11388919902e-02 - 3.39196724265e-02 2.97974895143e-02 2.70513354759e-02 2.61140936360e-02 2.75584598461e-02 3.04051143293e-02 - 3.31963581644e-02 3.49769025117e-02 3.55406204185e-02 3.49769025117e-02 3.31963581643e-02 3.04051143293e-02 - 2.55791633527e-02 2.72051400328e-02 3.05548189106e-02 3.46096179590e-02 4.00842207790e-02 4.86717503239e-02 - 5.87902882084e-02 6.56428313091e-02 6.56428313091e-02 5.87902882084e-02 4.86717503239e-02 4.00842207790e-02 - 3.46096179591e-02 3.05548189106e-02 2.72051400328e-02 2.55791633527e-02 2.59618688149e-02 2.73091501295e-02 - 2.84217494631e-02 2.88778389824e-02 2.88778389824e-02 2.84217494631e-02 2.73091501295e-02 2.59618688149e-02 - 2.52445347087e-02 2.92643382604e-02 3.39942859271e-02 3.86294987685e-02 4.42637285519e-02 5.17382028009e-02 - 5.87902882084e-02 6.17008685349e-02 5.87902882084e-02 5.17382028009e-02 4.42637285519e-02 3.86294987685e-02 - 3.39942859271e-02 2.92643382604e-02 2.52445347088e-02 2.30253665657e-02 2.24043982374e-02 2.23752455712e-02 - 2.22713307355e-02 2.21709602700e-02 2.22713307355e-02 2.23752455712e-02 2.24043982374e-02 2.30253665656e-02 - 2.59181147280e-02 3.12134113093e-02 3.61935553818e-02 4.02266025417e-02 4.42637285519e-02 4.86717503239e-02 - 5.18158087398e-02 5.18158087398e-02 4.86717503239e-02 4.42637285519e-02 4.02266025417e-02 3.61935553818e-02 - 3.12134113093e-02 2.59181147281e-02 2.17948830411e-02 1.93871115228e-02 1.81179302207e-02 1.72611853896e-02 - 1.66950035376e-02 1.66950035376e-02 1.72611853896e-02 1.81179302207e-02 1.93871115228e-02 2.17948830411e-02 - 2.65858530209e-02 3.20070729422e-02 3.61935553818e-02 3.86294987685e-02 4.00842207790e-02 4.11388919902e-02 - 4.15726567627e-02 4.11388919902e-02 4.00842207790e-02 3.86294987685e-02 3.61935553818e-02 3.20070729422e-02 - 2.65858530209e-02 2.15195877577e-02 1.78888288560e-02 1.56224022211e-02 1.41149739580e-02 1.30618774824e-02 - 1.26511415702e-02 1.30618774824e-02 1.41149739580e-02 1.56224022211e-02 1.78888288560e-02 2.15195877577e-02 - 2.65858530209e-02 3.12134113093e-02 3.39942859271e-02 3.46096179590e-02 3.39196724265e-02 3.31551669202e-02 - 3.31551669202e-02 3.39196724265e-02 3.46096179590e-02 3.39942859271e-02 3.12134113093e-02 2.65858530209e-02 - 2.15038028103e-02 1.72812015630e-02 1.43188164440e-02 1.22934163580e-02 1.08500532301e-02 1.00210602336e-02 - 1.00210602336e-02 1.08500532301e-02 1.22934163580e-02 1.43188164440e-02 1.72812015630e-02 2.15038028103e-02 - 2.59181147281e-02 2.92643382604e-02 3.05548189106e-02 2.97974895143e-02 2.81921950447e-02 2.73937018061e-02 - 2.81921950447e-02 2.97974895143e-02 3.05548189106e-02 2.92643382604e-02 2.59181147281e-02 2.15195877577e-02 - 1.72812015630e-02 1.38961516944e-02 1.14031750108e-02 9.57619631350e-03 8.36360112567e-03 7.92498387887e-03 - 8.36360112567e-03 9.57619631350e-03 1.14031750108e-02 1.38961516944e-02 1.72812015630e-02 2.15195877577e-02 - 2.52445347088e-02 2.72051400328e-02 2.70513354759e-02 2.52955690009e-02 2.36141349500e-02 2.36141349500e-02 - 2.52955690009e-02 2.70513354759e-02 2.72051400328e-02 2.52445347088e-02 2.17948830411e-02 1.78888288560e-02 - 1.43188164440e-02 1.14031750108e-02 9.13724506496e-03 7.50451749096e-03 6.62087909721e-03 6.62087909721e-03 - 7.50451749096e-03 9.13724506496e-03 1.14031750108e-02 1.43188164440e-02 1.78888288560e-02 2.17948830411e-02 - 2.55791633528e-02 2.61140936361e-02 2.45139707405e-02 2.20787139948e-02 2.09116171503e-02 2.20787139947e-02 - 2.45139707405e-02 2.61140936361e-02 2.55791633528e-02 2.30253665657e-02 1.93871115228e-02 1.56224022211e-02 - 1.22934163580e-02 9.57619631351e-03 7.50451749097e-03 6.16385997204e-03 5.69328932590e-03 6.16385997204e-03 - 7.50451749097e-03 9.57619631351e-03 1.22934163580e-02 1.56224022211e-02 1.93871115228e-02 2.30253665657e-02 - 2.75584598462e-02 2.65224863971e-02 2.36141349500e-02 2.10846196163e-02 2.10846196163e-02 2.36141349500e-02 - 2.65224863971e-02 2.75584598462e-02 2.59618688149e-02 2.24043982374e-02 1.81179302207e-02 1.41149739580e-02 - 1.08500532301e-02 8.36360112567e-03 6.62087909721e-03 5.69328932590e-03 5.69328932590e-03 6.62087909721e-03 - 8.36360112567e-03 1.08500532301e-02 1.41149739580e-02 1.81179302207e-02 2.24043982374e-02 2.59618688149e-02 - 3.03789244340e-02 2.73937018061e-02 2.35389825848e-02 2.17746576296e-02 2.35389825848e-02 2.73937018061e-02 - 3.03789244340e-02 3.04051143294e-02 2.73091501295e-02 2.23752455712e-02 1.72611853896e-02 1.30618774824e-02 - 1.00210602336e-02 7.92498387888e-03 6.62087909721e-03 6.16385997204e-03 6.62087909721e-03 7.92498387888e-03 - 1.00210602336e-02 1.30618774824e-02 1.72611853896e-02 2.23752455712e-02 2.73091501295e-02 3.04051143294e-02 - 3.31551669202e-02 2.95749315132e-02 2.67216511960e-02 2.67216511960e-02 2.95749315132e-02 3.31551669202e-02 - 3.48628450776e-02 3.31963581644e-02 2.84217494631e-02 2.22713307355e-02 1.66950035376e-02 1.26511415702e-02 - 1.00210602336e-02 8.36360112567e-03 7.50451749096e-03 7.50451749096e-03 8.36360112567e-03 1.00210602336e-02 - 1.26511415702e-02 1.66950035376e-02 2.22713307355e-02 2.84217494631e-02 3.31963581644e-02 3.48628450776e-02 - 4.23899857520e-02 4.30330806507e-02 4.31931066267e-02 4.30330806507e-02 4.23899857520e-02 4.11388919902e-02 - 3.89798518706e-02 3.49769025118e-02 2.88778389824e-02 2.21709602700e-02 1.66950035376e-02 1.30618774824e-02 - 1.08500532301e-02 9.57619631350e-03 9.13724506496e-03 9.57619631350e-03 1.08500532301e-02 1.30618774824e-02 - 1.66950035376e-02 2.21709602700e-02 2.88778389824e-02 3.49769025118e-02 3.89798518706e-02 4.11388919902e-02 - 6.87950359264e-02 7.44578962495e-02 7.44578962495e-02 6.87950359264e-02 5.87809730551e-02 4.86717503239e-02 - 4.14583134664e-02 3.55406204185e-02 2.88778389824e-02 2.22713307355e-02 1.72611853896e-02 1.41149739580e-02 - 1.22934163580e-02 1.14031750108e-02 1.14031750108e-02 1.22934163580e-02 1.41149739580e-02 1.72611853896e-02 - 2.22713307355e-02 2.88778389824e-02 3.55406204185e-02 4.14583134665e-02 4.86717503239e-02 5.87809730551e-02 - 1.02255317232e-01 1.05382836884e-01 1.02255317232e-01 9.00454509228e-02 6.98255714910e-02 5.17382028009e-02 - 4.14583134664e-02 3.49769025117e-02 2.84217494631e-02 2.23752455712e-02 1.81179302207e-02 1.56224022211e-02 - 1.43188164440e-02 1.38961516945e-02 1.43188164440e-02 1.56224022211e-02 1.81179302207e-02 2.23752455712e-02 - 2.84217494631e-02 3.49769025118e-02 4.14583134664e-02 5.17382028009e-02 6.98255714910e-02 9.00454509228e-02 - 1.14284409879e-01 1.14284409879e-01 1.13479602981e-01 9.74781930220e-02 6.98255714910e-02 4.86717503239e-02 - 3.89798518706e-02 3.31963581644e-02 2.73091501295e-02 2.24043982374e-02 1.93871115228e-02 1.78888288560e-02 - 1.72812015630e-02 1.72812015630e-02 1.78888288560e-02 1.93871115228e-02 2.24043982374e-02 2.73091501295e-02 - 3.31963581644e-02 3.89798518706e-02 4.86717503239e-02 6.98255714910e-02 9.74781930221e-02 1.13479602981e-01 - 1.05271425072e-01 1.12133375345e-01 1.13479602981e-01 9.00454509227e-02 5.87809730550e-02 4.11388919901e-02 - 3.48628450776e-02 3.04051143293e-02 2.59618688149e-02 2.30253665657e-02 2.17948830411e-02 2.15195877577e-02 - 2.15038028103e-02 2.15195877577e-02 2.17948830411e-02 2.30253665657e-02 2.59618688149e-02 3.04051143293e-02 - 3.48628450776e-02 4.11388919902e-02 5.87809730550e-02 9.00454509228e-02 1.13479602981e-01 1.12133375345e-01 - 1.14984029929e-01 1.02255317231e-01 7.14595557377e-02 4.26148361403e-02 3.05273339684e-02 2.81921950446e-02 - 2.69081613820e-02 2.62352804996e-02 2.71142998602e-02 2.89473471924e-02 3.06728014251e-02 3.13656133728e-02 - 3.06728014251e-02 2.89473471924e-02 2.71142998602e-02 2.62352804996e-02 2.69081613820e-02 2.81921950447e-02 - 3.05273339684e-02 4.26148361403e-02 7.14595557377e-02 1.02255317231e-01 1.14984029929e-01 1.15476799016e-01 - 1.02255317231e-01 7.44578962493e-02 4.31931066266e-02 2.67216511960e-02 2.35389825848e-02 2.36141349499e-02 - 2.45139707404e-02 2.70513354759e-02 3.05548189106e-02 3.39942859271e-02 3.61935553818e-02 3.61935553818e-02 - 3.39942859271e-02 3.05548189106e-02 2.70513354759e-02 2.45139707404e-02 2.36141349500e-02 2.35389825848e-02 - 2.67216511960e-02 4.31931066266e-02 7.44578962494e-02 1.02255317231e-01 1.13479602981e-01 1.13479602981e-01 - 7.14595557376e-02 4.31931066266e-02 2.50661805908e-02 1.98434779695e-02 1.95329704288e-02 2.09116171502e-02 - 2.44579386769e-02 2.92436594720e-02 3.44216532954e-02 3.88811262511e-02 4.06883718959e-02 3.88811262512e-02 - 3.44216532954e-02 2.92436594720e-02 2.44579386769e-02 2.09116171502e-02 1.95329704289e-02 1.98434779696e-02 - 2.50661805908e-02 4.31931066266e-02 7.14595557377e-02 9.40538638177e-02 1.01688733219e-01 9.40538638177e-02 - 4.26148361403e-02 2.67216511960e-02 1.98434779695e-02 1.76763940774e-02 1.79304117733e-02 2.10846196163e-02 - 2.61164989591e-02 3.26920605346e-02 4.03062291682e-02 4.56987320892e-02 4.56987320892e-02 4.03062291682e-02 - 3.26920605346e-02 2.61164989591e-02 2.10846196163e-02 1.79304117733e-02 1.76763940774e-02 1.98434779696e-02 - 2.67216511960e-02 4.26148361403e-02 6.30064198497e-02 7.65071402185e-02 7.65071402185e-02 6.30064198496e-02 - 3.05273339684e-02 2.35389825848e-02 1.95329704288e-02 1.79304117733e-02 1.95329704288e-02 2.35389825848e-02 - 3.05273339684e-02 4.15214050146e-02 5.24371687843e-02 5.69437593911e-02 5.24371687843e-02 4.15214050146e-02 - 3.05273339684e-02 2.35389825848e-02 1.95329704289e-02 1.79304117733e-02 1.95329704289e-02 2.35389825848e-02 - 3.05273339684e-02 4.15214050146e-02 5.24371687843e-02 5.69437593911e-02 5.24371687843e-02 4.15214050146e-02 - 2.81921950446e-02 2.36141349499e-02 2.09116171502e-02 2.10846196163e-02 2.35389825848e-02 2.95749315132e-02 - 4.23899857520e-02 5.87809730550e-02 6.98255714909e-02 6.98255714909e-02 5.87809730550e-02 4.23899857520e-02 - 2.95749315132e-02 2.35389825848e-02 2.10846196163e-02 2.09116171502e-02 2.36141349500e-02 2.81921950446e-02 - 3.39196724265e-02 4.00842207789e-02 4.42637285519e-02 4.42637285519e-02 4.00842207789e-02 3.39196724265e-02 - 2.69081613820e-02 2.45139707404e-02 2.44579386769e-02 2.61164989591e-02 3.05273339684e-02 4.23899857520e-02 - 6.13624406410e-02 7.81193802830e-02 8.44119137022e-02 7.81193802831e-02 6.13624406410e-02 4.23899857520e-02 - 3.05273339685e-02 2.61164989591e-02 2.44579386769e-02 2.45139707405e-02 2.69081613820e-02 3.06792139923e-02 - 3.45751625707e-02 3.75326171916e-02 3.86293456568e-02 3.75326171916e-02 3.45751625707e-02 3.06792139923e-02 - 2.62352804996e-02 2.70513354759e-02 2.92436594720e-02 3.26920605347e-02 4.15214050146e-02 5.87809730550e-02 - 7.81193802831e-02 8.97664201881e-02 8.97664201881e-02 7.81193802831e-02 5.87809730551e-02 4.15214050147e-02 - 3.26920605347e-02 2.92436594720e-02 2.70513354759e-02 2.62352804996e-02 2.73903662614e-02 2.95890354456e-02 - 3.16057035862e-02 3.27029610793e-02 3.27029610793e-02 3.16057035862e-02 2.95890354456e-02 2.73903662614e-02 - 2.71142998602e-02 3.05548189106e-02 3.44216532954e-02 4.03062291683e-02 5.24371687844e-02 6.98255714910e-02 - 8.44119137022e-02 8.97664201881e-02 8.44119137023e-02 6.98255714910e-02 5.24371687844e-02 4.03062291683e-02 - 3.44216532954e-02 3.05548189106e-02 2.71142998602e-02 2.51931467428e-02 2.49964037842e-02 2.55726183949e-02 - 2.60492747623e-02 2.61902387743e-02 2.60492747623e-02 2.55726183949e-02 2.49964037842e-02 2.51931467427e-02 - 2.89473471924e-02 3.39942859271e-02 3.88811262512e-02 4.56987320892e-02 5.69437593912e-02 6.98255714910e-02 - 7.81193802831e-02 7.81193802831e-02 6.98255714910e-02 5.69437593912e-02 4.56987320893e-02 3.88811262512e-02 - 3.39942859271e-02 2.89473471924e-02 2.46286629032e-02 2.20777720561e-02 2.09446754967e-02 2.03920879638e-02 - 2.00583037371e-02 2.00583037371e-02 2.03920879638e-02 2.09446754967e-02 2.20777720561e-02 2.46286629032e-02 - 3.06728014251e-02 3.61935553818e-02 4.06883718959e-02 4.56987320892e-02 5.24371687844e-02 5.87809730551e-02 - 6.13624406410e-02 5.87809730551e-02 5.24371687844e-02 4.56987320893e-02 4.06883718959e-02 3.61935553818e-02 - 3.06728014251e-02 2.49902463304e-02 2.07196781889e-02 1.81502966987e-02 1.66276099946e-02 1.56489970468e-02 - 1.52718999852e-02 1.56489970468e-02 1.66276099946e-02 1.81502966987e-02 2.07196781889e-02 2.49902463304e-02 - 3.13656133727e-02 3.61935553818e-02 3.88811262512e-02 4.03062291683e-02 4.15214050147e-02 4.23899857520e-02 - 4.23899857520e-02 4.15214050147e-02 4.03062291683e-02 3.88811262512e-02 3.61935553818e-02 3.13656133727e-02 - 2.54054430258e-02 2.02040886880e-02 1.66389339973e-02 1.43894839458e-02 1.28937939562e-02 1.20439356208e-02 - 1.20439356208e-02 1.28937939562e-02 1.43894839458e-02 1.66389339973e-02 2.02040886880e-02 2.54054430258e-02 - 3.06728014251e-02 3.39942859271e-02 3.44216532954e-02 3.26920605347e-02 3.05273339685e-02 2.95749315132e-02 - 3.05273339685e-02 3.26920605347e-02 3.44216532954e-02 3.39942859271e-02 3.06728014251e-02 2.54054430258e-02 - 2.00653973118e-02 1.59287126341e-02 1.31234388541e-02 1.12137577955e-02 9.96234500644e-03 9.50248837487e-03 - 9.96234500643e-03 1.12137577955e-02 1.31234388541e-02 1.59287126341e-02 2.00653973118e-02 2.54054430258e-02 - 2.89473471924e-02 3.05548189106e-02 2.92436594720e-02 2.61164989591e-02 2.35389825848e-02 2.35389825848e-02 - 2.61164989591e-02 2.92436594720e-02 3.05548189106e-02 2.89473471924e-02 2.49902463304e-02 2.02040886880e-02 - 1.59287126341e-02 1.26855285494e-02 1.03422447642e-02 8.68876908730e-03 7.77912772161e-03 7.77912772161e-03 - 8.68876908730e-03 1.03422447642e-02 1.26855285494e-02 1.59287126341e-02 2.02040886880e-02 2.49902463304e-02 - 2.71142998602e-02 2.70513354759e-02 2.44579386769e-02 2.10846196163e-02 1.95329704289e-02 2.10846196163e-02 - 2.44579386769e-02 2.70513354759e-02 2.71142998602e-02 2.46286629032e-02 2.07196781889e-02 1.66389339973e-02 - 1.31234388541e-02 1.03422447642e-02 8.23785813084e-03 6.85668012358e-03 6.36555591034e-03 6.85668012358e-03 - 8.23785813084e-03 1.03422447642e-02 1.31234388541e-02 1.66389339973e-02 2.07196781889e-02 2.46286629032e-02 - 2.62352804996e-02 2.45139707405e-02 2.09116171503e-02 1.79304117734e-02 1.79304117734e-02 2.09116171503e-02 - 2.45139707405e-02 2.62352804996e-02 2.51931467428e-02 2.20777720561e-02 1.81502966987e-02 1.43894839458e-02 - 1.12137577956e-02 8.68876908731e-03 6.85668012359e-03 5.86550216408e-03 5.86550216408e-03 6.85668012359e-03 - 8.68876908731e-03 1.12137577956e-02 1.43894839458e-02 1.81502966987e-02 2.20777720561e-02 2.51931467428e-02 - 2.69081613821e-02 2.36141349500e-02 1.95329704289e-02 1.76763940775e-02 1.95329704289e-02 2.36141349500e-02 - 2.69081613821e-02 2.73903662614e-02 2.49964037843e-02 2.09446754967e-02 1.66276099946e-02 1.28937939562e-02 - 9.96234500644e-03 7.77912772162e-03 6.36555591034e-03 5.86550216408e-03 6.36555591034e-03 7.77912772162e-03 - 9.96234500644e-03 1.28937939562e-02 1.66276099946e-02 2.09446754967e-02 2.49964037843e-02 2.73903662614e-02 - 2.81921950447e-02 2.35389825848e-02 1.98434779696e-02 1.98434779696e-02 2.35389825848e-02 2.81921950447e-02 - 3.06792139923e-02 2.95890354456e-02 2.55726183949e-02 2.03920879638e-02 1.56489970468e-02 1.20439356208e-02 - 9.50248837488e-03 7.77912772162e-03 6.85668012358e-03 6.85668012358e-03 7.77912772161e-03 9.50248837488e-03 - 1.20439356208e-02 1.56489970468e-02 2.03920879638e-02 2.55726183949e-02 2.95890354456e-02 3.06792139923e-02 - 3.05273339685e-02 2.67216511960e-02 2.50661805908e-02 2.67216511960e-02 3.05273339685e-02 3.39196724265e-02 - 3.45751625707e-02 3.16057035862e-02 2.60492747623e-02 2.00583037371e-02 1.52718999852e-02 1.20439356208e-02 - 9.96234500644e-03 8.68876908730e-03 8.23785813083e-03 8.68876908731e-03 9.96234500644e-03 1.20439356208e-02 - 1.52718999852e-02 2.00583037371e-02 2.60492747623e-02 3.16057035862e-02 3.45751625707e-02 3.39196724265e-02 - 4.26148361404e-02 4.31931066267e-02 4.31931066267e-02 4.26148361404e-02 4.15214050147e-02 4.00842207790e-02 - 3.75326171916e-02 3.27029610793e-02 2.61902387743e-02 2.00583037371e-02 1.56489970468e-02 1.28937939562e-02 - 1.12137577955e-02 1.03422447642e-02 1.03422447642e-02 1.12137577955e-02 1.28937939562e-02 1.56489970468e-02 - 2.00583037371e-02 2.61902387743e-02 3.27029610793e-02 3.75326171916e-02 4.00842207790e-02 4.15214050147e-02 - 7.14595557378e-02 7.44578962495e-02 7.14595557377e-02 6.30064198497e-02 5.24371687844e-02 4.42637285519e-02 - 3.86293456568e-02 3.27029610793e-02 2.60492747623e-02 2.03920879638e-02 1.66276099946e-02 1.43894839458e-02 - 1.31234388541e-02 1.26855285494e-02 1.31234388541e-02 1.43894839458e-02 1.66276099946e-02 2.03920879638e-02 - 2.60492747623e-02 3.27029610793e-02 3.86293456568e-02 4.42637285519e-02 5.24371687844e-02 6.30064198498e-02 - 1.02255317232e-01 1.02255317232e-01 9.40538638178e-02 7.65071402186e-02 5.69437593911e-02 4.42637285519e-02 - 3.75326171916e-02 3.16057035862e-02 2.55726183949e-02 2.09446754967e-02 1.81502966987e-02 1.66389339974e-02 - 1.59287126341e-02 1.59287126341e-02 1.66389339974e-02 1.81502966987e-02 2.09446754967e-02 2.55726183949e-02 - 3.16057035862e-02 3.75326171916e-02 4.42637285519e-02 5.69437593912e-02 7.65071402186e-02 9.40538638178e-02 - 1.14984029929e-01 1.13479602981e-01 1.01688733220e-01 7.65071402186e-02 5.24371687843e-02 4.00842207790e-02 - 3.45751625707e-02 2.95890354456e-02 2.49964037842e-02 2.20777720561e-02 2.07196781889e-02 2.02040886880e-02 - 2.00653973119e-02 2.02040886880e-02 2.07196781889e-02 2.20777720561e-02 2.49964037842e-02 2.95890354456e-02 - 3.45751625707e-02 4.00842207790e-02 5.24371687844e-02 7.65071402186e-02 1.01688733220e-01 1.13479602981e-01 - 1.15476799016e-01 1.13479602981e-01 9.40538638177e-02 6.30064198497e-02 4.15214050146e-02 3.39196724265e-02 - 3.06792139923e-02 2.73903662614e-02 2.51931467427e-02 2.46286629032e-02 2.49902463304e-02 2.54054430258e-02 - 2.54054430258e-02 2.49902463304e-02 2.46286629032e-02 2.51931467428e-02 2.73903662614e-02 3.06792139923e-02 - 3.39196724265e-02 4.15214050147e-02 6.30064198497e-02 9.40538638178e-02 1.13479602981e-01 1.15476799016e-01 - 9.46618817554e-02 6.87950359263e-02 4.26148361403e-02 2.92890570903e-02 2.61164989591e-02 2.52955690008e-02 - 2.54927185045e-02 2.74769895380e-02 3.04773481039e-02 3.33647463123e-02 3.51262182185e-02 3.51262182185e-02 - 3.33647463124e-02 3.04773481040e-02 2.74769895380e-02 2.54927185045e-02 2.52955690009e-02 2.61164989591e-02 - 2.92890570903e-02 4.26148361403e-02 6.87950359264e-02 9.46618817554e-02 1.07504612680e-01 1.07504612680e-01 - 6.87950359263e-02 4.30330806506e-02 2.67216511960e-02 2.17746576295e-02 2.10846196163e-02 2.20787139947e-02 - 2.52955690008e-02 2.97974895143e-02 3.46096179590e-02 3.86294987685e-02 4.02266025417e-02 3.86294987685e-02 - 3.46096179590e-02 2.97974895143e-02 2.52955690008e-02 2.20787139947e-02 2.10846196163e-02 2.17746576295e-02 - 2.67216511960e-02 4.30330806506e-02 6.87950359263e-02 9.00454509227e-02 9.74781930220e-02 9.00454509227e-02 - 4.26148361403e-02 2.67216511960e-02 1.98434779695e-02 1.76763940774e-02 1.79304117733e-02 2.10846196163e-02 - 2.61164989591e-02 3.26920605346e-02 4.03062291682e-02 4.56987320892e-02 4.56987320892e-02 4.03062291683e-02 - 3.26920605347e-02 2.61164989591e-02 2.10846196163e-02 1.79304117733e-02 1.76763940774e-02 1.98434779696e-02 - 2.67216511960e-02 4.26148361403e-02 6.30064198497e-02 7.65071402185e-02 7.65071402185e-02 6.30064198496e-02 - 2.92890570903e-02 2.17746576295e-02 1.76763940774e-02 1.60510791347e-02 1.76763940774e-02 2.17746576295e-02 - 2.92890570903e-02 4.15474983449e-02 5.38553081705e-02 5.89461275929e-02 5.38553081705e-02 4.15474983449e-02 - 2.92890570903e-02 2.17746576295e-02 1.76763940774e-02 1.60510791348e-02 1.76763940774e-02 2.17746576295e-02 - 2.92890570903e-02 4.15474983449e-02 5.38553081705e-02 5.89461275929e-02 5.38553081705e-02 4.15474983449e-02 - 2.61164989591e-02 2.10846196163e-02 1.79304117733e-02 1.76763940774e-02 1.98434779695e-02 2.67216511960e-02 - 4.26148361403e-02 6.30064198496e-02 7.65071402185e-02 7.65071402185e-02 6.30064198497e-02 4.26148361403e-02 - 2.67216511960e-02 1.98434779696e-02 1.76763940774e-02 1.79304117733e-02 2.10846196163e-02 2.61164989591e-02 - 3.26920605347e-02 4.03062291683e-02 4.56987320892e-02 4.56987320892e-02 4.03062291682e-02 3.26920605346e-02 - 2.52955690008e-02 2.20787139947e-02 2.10846196163e-02 2.17746576295e-02 2.67216511960e-02 4.30330806506e-02 - 6.87950359263e-02 9.00454509227e-02 9.74781930220e-02 9.00454509227e-02 6.87950359263e-02 4.30330806506e-02 - 2.67216511960e-02 2.17746576295e-02 2.10846196163e-02 2.20787139947e-02 2.52955690008e-02 2.97974895143e-02 - 3.46096179590e-02 3.86294987685e-02 4.02266025417e-02 3.86294987685e-02 3.46096179590e-02 2.97974895143e-02 - 2.54927185045e-02 2.52955690008e-02 2.61164989591e-02 2.92890570903e-02 4.26148361403e-02 6.87950359263e-02 - 9.46618817554e-02 1.07504612680e-01 1.07504612680e-01 9.46618817554e-02 6.87950359264e-02 4.26148361403e-02 - 2.92890570903e-02 2.61164989591e-02 2.52955690009e-02 2.54927185045e-02 2.74769895380e-02 3.04773481040e-02 - 3.33647463124e-02 3.51262182185e-02 3.51262182185e-02 3.33647463123e-02 3.04773481039e-02 2.74769895380e-02 - 2.74769895380e-02 2.97974895143e-02 3.26920605347e-02 4.15474983449e-02 6.30064198497e-02 9.00454509228e-02 - 1.07504612680e-01 1.12225405021e-01 1.07504612680e-01 9.00454509228e-02 6.30064198497e-02 4.15474983450e-02 - 3.26920605347e-02 2.97974895143e-02 2.74769895380e-02 2.61756575962e-02 2.65400648390e-02 2.78013937268e-02 - 2.89590113766e-02 2.93979324046e-02 2.89590113766e-02 2.78013937268e-02 2.65400648390e-02 2.61756575962e-02 - 3.04773481040e-02 3.46096179590e-02 4.03062291683e-02 5.38553081705e-02 7.65071402186e-02 9.74781930221e-02 - 1.07504612680e-01 1.07504612680e-01 9.74781930221e-02 7.65071402186e-02 5.38553081706e-02 4.03062291683e-02 - 3.46096179590e-02 3.04773481040e-02 2.65400648390e-02 2.40839675381e-02 2.32030440049e-02 2.30939042850e-02 - 2.31231583381e-02 2.31231583381e-02 2.30939042850e-02 2.32030440049e-02 2.40839675381e-02 2.65400648390e-02 - 3.33647463123e-02 3.86294987685e-02 4.56987320892e-02 5.89461275929e-02 7.65071402186e-02 9.00454509228e-02 - 9.46618817555e-02 9.00454509228e-02 7.65071402186e-02 5.89461275929e-02 4.56987320892e-02 3.86294987685e-02 - 3.33647463124e-02 2.78013937268e-02 2.32030440049e-02 2.03983415996e-02 1.89263764236e-02 1.81430425205e-02 - 1.78657504093e-02 1.81430425204e-02 1.89263764236e-02 2.03983415996e-02 2.32030440049e-02 2.78013937268e-02 - 3.51262182184e-02 4.02266025417e-02 4.56987320892e-02 5.38553081706e-02 6.30064198497e-02 6.87950359264e-02 - 6.87950359264e-02 6.30064198498e-02 5.38553081706e-02 4.56987320892e-02 4.02266025417e-02 3.51262182185e-02 - 2.89590113766e-02 2.30939042849e-02 1.89263764236e-02 1.64062528191e-02 1.48932100517e-02 1.40901390285e-02 - 1.40901390285e-02 1.48932100517e-02 1.64062528191e-02 1.89263764236e-02 2.30939042849e-02 2.89590113766e-02 - 3.51262182185e-02 3.86294987685e-02 4.03062291683e-02 4.15474983450e-02 4.26148361404e-02 4.30330806506e-02 - 4.26148361404e-02 4.15474983450e-02 4.03062291683e-02 3.86294987685e-02 3.51262182185e-02 2.93979324046e-02 - 2.31231583381e-02 1.81430425204e-02 1.48932100517e-02 1.28580577518e-02 1.16029685725e-02 1.11481007476e-02 - 1.16029685725e-02 1.28580577518e-02 1.48932100517e-02 1.81430425204e-02 2.31231583381e-02 2.93979324046e-02 - 3.33647463124e-02 3.46096179590e-02 3.26920605347e-02 2.92890570903e-02 2.67216511960e-02 2.67216511960e-02 - 2.92890570903e-02 3.26920605347e-02 3.46096179591e-02 3.33647463124e-02 2.89590113766e-02 2.31231583381e-02 - 1.78657504093e-02 1.40901390285e-02 1.16029685725e-02 9.96167567605e-03 9.07031679536e-03 9.07031679536e-03 - 9.96167567605e-03 1.16029685725e-02 1.40901390285e-02 1.78657504093e-02 2.31231583381e-02 2.89590113766e-02 - 3.04773481040e-02 2.97974895143e-02 2.61164989591e-02 2.17746576295e-02 1.98434779696e-02 2.17746576295e-02 - 2.61164989591e-02 2.97974895143e-02 3.04773481040e-02 2.78013937268e-02 2.30939042850e-02 1.81430425204e-02 - 1.40901390285e-02 1.11481007476e-02 9.07031679536e-03 7.73176989509e-03 7.25301575416e-03 7.73176989509e-03 - 9.07031679536e-03 1.11481007476e-02 1.40901390285e-02 1.81430425204e-02 2.30939042849e-02 2.78013937268e-02 - 2.74769895380e-02 2.52955690009e-02 2.10846196163e-02 1.76763940775e-02 1.76763940775e-02 2.10846196163e-02 - 2.52955690009e-02 2.74769895380e-02 2.65400648390e-02 2.32030440049e-02 1.89263764236e-02 1.48932100517e-02 - 1.16029685725e-02 9.07031679537e-03 7.25301575416e-03 6.26631005461e-03 6.26631005461e-03 7.25301575416e-03 - 9.07031679537e-03 1.16029685725e-02 1.48932100517e-02 1.89263764236e-02 2.32030440049e-02 2.65400648390e-02 - 2.54927185046e-02 2.20787139947e-02 1.79304117734e-02 1.60510791348e-02 1.79304117734e-02 2.20787139947e-02 - 2.54927185046e-02 2.61756575963e-02 2.40839675381e-02 2.03983415996e-02 1.64062528191e-02 1.28580577518e-02 - 9.96167567605e-03 7.73176989510e-03 6.26631005461e-03 5.74651192180e-03 6.26631005461e-03 7.73176989510e-03 - 9.96167567605e-03 1.28580577518e-02 1.64062528191e-02 2.03983415996e-02 2.40839675381e-02 2.61756575963e-02 - 2.52955690009e-02 2.10846196163e-02 1.76763940775e-02 1.76763940775e-02 2.10846196163e-02 2.52955690009e-02 - 2.74769895380e-02 2.65400648390e-02 2.32030440049e-02 1.89263764236e-02 1.48932100517e-02 1.16029685725e-02 - 9.07031679537e-03 7.25301575416e-03 6.26631005461e-03 6.26631005461e-03 7.25301575416e-03 9.07031679537e-03 - 1.16029685725e-02 1.48932100517e-02 1.89263764236e-02 2.32030440049e-02 2.65400648390e-02 2.74769895380e-02 - 2.61164989591e-02 2.17746576295e-02 1.98434779696e-02 2.17746576295e-02 2.61164989591e-02 2.97974895143e-02 - 3.04773481040e-02 2.78013937268e-02 2.30939042850e-02 1.81430425204e-02 1.40901390285e-02 1.11481007476e-02 - 9.07031679536e-03 7.73176989509e-03 7.25301575416e-03 7.73176989509e-03 9.07031679536e-03 1.11481007476e-02 - 1.40901390285e-02 1.81430425204e-02 2.30939042850e-02 2.78013937268e-02 3.04773481040e-02 2.97974895143e-02 - 2.92890570903e-02 2.67216511960e-02 2.67216511960e-02 2.92890570903e-02 3.26920605347e-02 3.46096179590e-02 - 3.33647463124e-02 2.89590113766e-02 2.31231583381e-02 1.78657504093e-02 1.40901390285e-02 1.16029685725e-02 - 9.96167567605e-03 9.07031679536e-03 9.07031679536e-03 9.96167567605e-03 1.16029685725e-02 1.40901390285e-02 - 1.78657504093e-02 2.31231583381e-02 2.89590113766e-02 3.33647463124e-02 3.46096179591e-02 3.26920605347e-02 - 4.26148361404e-02 4.30330806506e-02 4.26148361404e-02 4.15474983450e-02 4.03062291683e-02 3.86294987685e-02 - 3.51262182185e-02 2.93979324046e-02 2.31231583381e-02 1.81430425204e-02 1.48932100517e-02 1.28580577518e-02 - 1.16029685725e-02 1.11481007476e-02 1.16029685725e-02 1.28580577518e-02 1.48932100517e-02 1.81430425204e-02 - 2.31231583381e-02 2.93979324046e-02 3.51262182185e-02 3.86294987685e-02 4.03062291683e-02 4.15474983450e-02 - 6.87950359264e-02 6.87950359264e-02 6.30064198497e-02 5.38553081706e-02 4.56987320892e-02 4.02266025417e-02 - 3.51262182184e-02 2.89590113766e-02 2.30939042849e-02 1.89263764236e-02 1.64062528191e-02 1.48932100517e-02 - 1.40901390285e-02 1.40901390285e-02 1.48932100517e-02 1.64062528191e-02 1.89263764236e-02 2.30939042849e-02 - 2.89590113766e-02 3.51262182185e-02 4.02266025417e-02 4.56987320892e-02 5.38553081706e-02 6.30064198498e-02 - 9.46618817555e-02 9.00454509228e-02 7.65071402186e-02 5.89461275929e-02 4.56987320892e-02 3.86294987685e-02 - 3.33647463124e-02 2.78013937268e-02 2.32030440049e-02 2.03983415996e-02 1.89263764236e-02 1.81430425204e-02 - 1.78657504093e-02 1.81430425205e-02 1.89263764236e-02 2.03983415996e-02 2.32030440049e-02 2.78013937268e-02 - 3.33647463124e-02 3.86294987685e-02 4.56987320892e-02 5.89461275929e-02 7.65071402186e-02 9.00454509228e-02 - 1.07504612680e-01 9.74781930221e-02 7.65071402186e-02 5.38553081705e-02 4.03062291683e-02 3.46096179590e-02 - 3.04773481040e-02 2.65400648390e-02 2.40839675381e-02 2.32030440049e-02 2.30939042850e-02 2.31231583381e-02 - 2.31231583381e-02 2.30939042850e-02 2.32030440049e-02 2.40839675381e-02 2.65400648390e-02 3.04773481040e-02 - 3.46096179590e-02 4.03062291683e-02 5.38553081706e-02 7.65071402186e-02 9.74781930221e-02 1.07504612680e-01 - 1.07504612680e-01 9.00454509228e-02 6.30064198497e-02 4.15474983449e-02 3.26920605347e-02 2.97974895143e-02 - 2.74769895380e-02 2.61756575962e-02 2.65400648390e-02 2.78013937268e-02 2.89590113766e-02 2.93979324046e-02 - 2.89590113766e-02 2.78013937268e-02 2.65400648390e-02 2.61756575962e-02 2.74769895380e-02 2.97974895143e-02 - 3.26920605347e-02 4.15474983450e-02 6.30064198497e-02 9.00454509228e-02 1.07504612680e-01 1.12225405021e-01 - 6.13624406410e-02 4.23899857520e-02 3.05273339684e-02 2.61164989591e-02 2.44579386769e-02 2.45139707404e-02 - 2.69081613820e-02 3.06792139923e-02 3.45751625707e-02 3.75326171916e-02 3.86293456568e-02 3.75326171916e-02 - 3.45751625707e-02 3.06792139923e-02 2.69081613820e-02 2.45139707405e-02 2.44579386769e-02 2.61164989591e-02 - 3.05273339685e-02 4.23899857520e-02 6.13624406410e-02 7.81193802831e-02 8.44119137022e-02 7.81193802830e-02 - 4.23899857520e-02 2.95749315132e-02 2.35389825848e-02 2.10846196163e-02 2.09116171502e-02 2.36141349499e-02 - 2.81921950446e-02 3.39196724265e-02 4.00842207789e-02 4.42637285519e-02 4.42637285519e-02 4.00842207789e-02 - 3.39196724265e-02 2.81921950446e-02 2.36141349500e-02 2.09116171502e-02 2.10846196163e-02 2.35389825848e-02 - 2.95749315132e-02 4.23899857520e-02 5.87809730550e-02 6.98255714909e-02 6.98255714909e-02 5.87809730550e-02 - 3.05273339684e-02 2.35389825848e-02 1.95329704288e-02 1.79304117733e-02 1.95329704288e-02 2.35389825848e-02 - 3.05273339684e-02 4.15214050146e-02 5.24371687843e-02 5.69437593911e-02 5.24371687843e-02 4.15214050146e-02 - 3.05273339684e-02 2.35389825848e-02 1.95329704289e-02 1.79304117733e-02 1.95329704289e-02 2.35389825848e-02 - 3.05273339684e-02 4.15214050146e-02 5.24371687843e-02 5.69437593911e-02 5.24371687843e-02 4.15214050146e-02 - 2.61164989591e-02 2.10846196163e-02 1.79304117733e-02 1.76763940774e-02 1.98434779695e-02 2.67216511960e-02 - 4.26148361403e-02 6.30064198496e-02 7.65071402185e-02 7.65071402185e-02 6.30064198497e-02 4.26148361403e-02 - 2.67216511960e-02 1.98434779696e-02 1.76763940774e-02 1.79304117733e-02 2.10846196163e-02 2.61164989591e-02 - 3.26920605346e-02 4.03062291682e-02 4.56987320892e-02 4.56987320892e-02 4.03062291682e-02 3.26920605346e-02 - 2.44579386769e-02 2.09116171502e-02 1.95329704288e-02 1.98434779695e-02 2.50661805908e-02 4.31931066266e-02 - 7.14595557376e-02 9.40538638177e-02 1.01688733219e-01 9.40538638177e-02 7.14595557377e-02 4.31931066266e-02 - 2.50661805908e-02 1.98434779696e-02 1.95329704289e-02 2.09116171502e-02 2.44579386769e-02 2.92436594720e-02 - 3.44216532954e-02 3.88811262512e-02 4.06883718959e-02 3.88811262511e-02 3.44216532954e-02 2.92436594720e-02 - 2.45139707404e-02 2.36141349499e-02 2.35389825848e-02 2.67216511960e-02 4.31931066266e-02 7.44578962493e-02 - 1.02255317231e-01 1.13479602981e-01 1.13479602981e-01 1.02255317231e-01 7.44578962494e-02 4.31931066266e-02 - 2.67216511960e-02 2.35389825848e-02 2.36141349500e-02 2.45139707404e-02 2.70513354759e-02 3.05548189106e-02 - 3.39942859271e-02 3.61935553818e-02 3.61935553818e-02 3.39942859271e-02 3.05548189106e-02 2.70513354759e-02 - 2.69081613820e-02 2.81921950446e-02 3.05273339684e-02 4.26148361403e-02 7.14595557377e-02 1.02255317231e-01 - 1.14984029929e-01 1.15476799016e-01 1.14984029929e-01 1.02255317231e-01 7.14595557377e-02 4.26148361403e-02 - 3.05273339684e-02 2.81921950447e-02 2.69081613820e-02 2.62352804996e-02 2.71142998602e-02 2.89473471924e-02 - 3.06728014251e-02 3.13656133728e-02 3.06728014251e-02 2.89473471924e-02 2.71142998602e-02 2.62352804996e-02 - 3.06792139923e-02 3.39196724265e-02 4.15214050146e-02 6.30064198497e-02 9.40538638177e-02 1.13479602981e-01 - 1.15476799016e-01 1.15476799016e-01 1.13479602981e-01 9.40538638178e-02 6.30064198497e-02 4.15214050147e-02 - 3.39196724265e-02 3.06792139923e-02 2.73903662614e-02 2.51931467428e-02 2.46286629032e-02 2.49902463304e-02 - 2.54054430258e-02 2.54054430258e-02 2.49902463304e-02 2.46286629032e-02 2.51931467427e-02 2.73903662614e-02 - 3.45751625707e-02 4.00842207790e-02 5.24371687843e-02 7.65071402186e-02 1.01688733220e-01 1.13479602981e-01 - 1.14984029929e-01 1.13479602981e-01 1.01688733220e-01 7.65071402186e-02 5.24371687844e-02 4.00842207790e-02 - 3.45751625707e-02 2.95890354456e-02 2.49964037842e-02 2.20777720561e-02 2.07196781889e-02 2.02040886880e-02 - 2.00653973119e-02 2.02040886880e-02 2.07196781889e-02 2.20777720561e-02 2.49964037842e-02 2.95890354456e-02 - 3.75326171916e-02 4.42637285519e-02 5.69437593911e-02 7.65071402186e-02 9.40538638178e-02 1.02255317232e-01 - 1.02255317232e-01 9.40538638178e-02 7.65071402186e-02 5.69437593912e-02 4.42637285519e-02 3.75326171916e-02 - 3.16057035862e-02 2.55726183949e-02 2.09446754967e-02 1.81502966987e-02 1.66389339974e-02 1.59287126341e-02 - 1.59287126341e-02 1.66389339974e-02 1.81502966987e-02 2.09446754967e-02 2.55726183949e-02 3.16057035862e-02 - 3.86293456568e-02 4.42637285519e-02 5.24371687844e-02 6.30064198497e-02 7.14595557377e-02 7.44578962495e-02 - 7.14595557378e-02 6.30064198498e-02 5.24371687844e-02 4.42637285519e-02 3.86293456568e-02 3.27029610793e-02 - 2.60492747623e-02 2.03920879638e-02 1.66276099946e-02 1.43894839458e-02 1.31234388541e-02 1.26855285494e-02 - 1.31234388541e-02 1.43894839458e-02 1.66276099946e-02 2.03920879638e-02 2.60492747623e-02 3.27029610793e-02 - 3.75326171916e-02 4.00842207790e-02 4.15214050147e-02 4.26148361404e-02 4.31931066267e-02 4.31931066267e-02 - 4.26148361404e-02 4.15214050147e-02 4.00842207790e-02 3.75326171916e-02 3.27029610793e-02 2.61902387743e-02 - 2.00583037371e-02 1.56489970468e-02 1.28937939562e-02 1.12137577955e-02 1.03422447642e-02 1.03422447642e-02 - 1.12137577955e-02 1.28937939562e-02 1.56489970468e-02 2.00583037371e-02 2.61902387743e-02 3.27029610793e-02 - 3.45751625707e-02 3.39196724265e-02 3.05273339685e-02 2.67216511960e-02 2.50661805908e-02 2.67216511960e-02 - 3.05273339685e-02 3.39196724265e-02 3.45751625707e-02 3.16057035862e-02 2.60492747623e-02 2.00583037371e-02 - 1.52718999852e-02 1.20439356208e-02 9.96234500644e-03 8.68876908731e-03 8.23785813083e-03 8.68876908730e-03 - 9.96234500644e-03 1.20439356208e-02 1.52718999852e-02 2.00583037371e-02 2.60492747623e-02 3.16057035862e-02 - 3.06792139923e-02 2.81921950447e-02 2.35389825848e-02 1.98434779696e-02 1.98434779696e-02 2.35389825848e-02 - 2.81921950447e-02 3.06792139923e-02 2.95890354456e-02 2.55726183949e-02 2.03920879638e-02 1.56489970468e-02 - 1.20439356208e-02 9.50248837488e-03 7.77912772161e-03 6.85668012358e-03 6.85668012358e-03 7.77912772162e-03 - 9.50248837488e-03 1.20439356208e-02 1.56489970468e-02 2.03920879638e-02 2.55726183949e-02 2.95890354456e-02 - 2.69081613821e-02 2.36141349500e-02 1.95329704289e-02 1.76763940775e-02 1.95329704289e-02 2.36141349500e-02 - 2.69081613821e-02 2.73903662614e-02 2.49964037843e-02 2.09446754967e-02 1.66276099946e-02 1.28937939562e-02 - 9.96234500644e-03 7.77912772162e-03 6.36555591034e-03 5.86550216408e-03 6.36555591034e-03 7.77912772162e-03 - 9.96234500644e-03 1.28937939562e-02 1.66276099946e-02 2.09446754967e-02 2.49964037843e-02 2.73903662614e-02 - 2.45139707405e-02 2.09116171503e-02 1.79304117734e-02 1.79304117734e-02 2.09116171503e-02 2.45139707405e-02 - 2.62352804996e-02 2.51931467428e-02 2.20777720561e-02 1.81502966987e-02 1.43894839458e-02 1.12137577956e-02 - 8.68876908731e-03 6.85668012359e-03 5.86550216408e-03 5.86550216408e-03 6.85668012359e-03 8.68876908731e-03 - 1.12137577956e-02 1.43894839458e-02 1.81502966987e-02 2.20777720561e-02 2.51931467428e-02 2.62352804996e-02 - 2.44579386769e-02 2.10846196163e-02 1.95329704289e-02 2.10846196163e-02 2.44579386769e-02 2.70513354759e-02 - 2.71142998602e-02 2.46286629032e-02 2.07196781889e-02 1.66389339973e-02 1.31234388541e-02 1.03422447642e-02 - 8.23785813084e-03 6.85668012358e-03 6.36555591034e-03 6.85668012358e-03 8.23785813084e-03 1.03422447642e-02 - 1.31234388541e-02 1.66389339973e-02 2.07196781889e-02 2.46286629032e-02 2.71142998602e-02 2.70513354759e-02 - 2.61164989591e-02 2.35389825848e-02 2.35389825848e-02 2.61164989591e-02 2.92436594720e-02 3.05548189106e-02 - 2.89473471924e-02 2.49902463304e-02 2.02040886880e-02 1.59287126341e-02 1.26855285494e-02 1.03422447642e-02 - 8.68876908730e-03 7.77912772161e-03 7.77912772161e-03 8.68876908730e-03 1.03422447642e-02 1.26855285494e-02 - 1.59287126341e-02 2.02040886880e-02 2.49902463304e-02 2.89473471924e-02 3.05548189106e-02 2.92436594720e-02 - 3.05273339685e-02 2.95749315132e-02 3.05273339685e-02 3.26920605347e-02 3.44216532954e-02 3.39942859271e-02 - 3.06728014251e-02 2.54054430258e-02 2.00653973118e-02 1.59287126341e-02 1.31234388541e-02 1.12137577955e-02 - 9.96234500643e-03 9.50248837487e-03 9.96234500644e-03 1.12137577955e-02 1.31234388541e-02 1.59287126341e-02 - 2.00653973118e-02 2.54054430258e-02 3.06728014251e-02 3.39942859271e-02 3.44216532954e-02 3.26920605347e-02 - 4.23899857520e-02 4.23899857520e-02 4.15214050147e-02 4.03062291683e-02 3.88811262512e-02 3.61935553818e-02 - 3.13656133727e-02 2.54054430258e-02 2.02040886880e-02 1.66389339973e-02 1.43894839458e-02 1.28937939562e-02 - 1.20439356208e-02 1.20439356208e-02 1.28937939562e-02 1.43894839458e-02 1.66389339973e-02 2.02040886880e-02 - 2.54054430258e-02 3.13656133727e-02 3.61935553818e-02 3.88811262512e-02 4.03062291683e-02 4.15214050147e-02 - 6.13624406410e-02 5.87809730551e-02 5.24371687844e-02 4.56987320892e-02 4.06883718959e-02 3.61935553818e-02 - 3.06728014251e-02 2.49902463304e-02 2.07196781889e-02 1.81502966987e-02 1.66276099946e-02 1.56489970468e-02 - 1.52718999852e-02 1.56489970468e-02 1.66276099946e-02 1.81502966987e-02 2.07196781889e-02 2.49902463304e-02 - 3.06728014251e-02 3.61935553818e-02 4.06883718959e-02 4.56987320893e-02 5.24371687844e-02 5.87809730551e-02 - 7.81193802831e-02 6.98255714910e-02 5.69437593912e-02 4.56987320892e-02 3.88811262512e-02 3.39942859271e-02 - 2.89473471924e-02 2.46286629032e-02 2.20777720561e-02 2.09446754967e-02 2.03920879638e-02 2.00583037371e-02 - 2.00583037371e-02 2.03920879638e-02 2.09446754967e-02 2.20777720561e-02 2.46286629032e-02 2.89473471924e-02 - 3.39942859271e-02 3.88811262512e-02 4.56987320893e-02 5.69437593912e-02 6.98255714910e-02 7.81193802831e-02 - 8.44119137022e-02 6.98255714910e-02 5.24371687844e-02 4.03062291683e-02 3.44216532954e-02 3.05548189106e-02 - 2.71142998602e-02 2.51931467427e-02 2.49964037842e-02 2.55726183949e-02 2.60492747623e-02 2.61902387743e-02 - 2.60492747623e-02 2.55726183949e-02 2.49964037842e-02 2.51931467428e-02 2.71142998602e-02 3.05548189106e-02 - 3.44216532954e-02 4.03062291683e-02 5.24371687844e-02 6.98255714910e-02 8.44119137023e-02 8.97664201881e-02 - 7.81193802831e-02 5.87809730550e-02 4.15214050146e-02 3.26920605347e-02 2.92436594720e-02 2.70513354759e-02 - 2.62352804996e-02 2.73903662614e-02 2.95890354456e-02 3.16057035862e-02 3.27029610793e-02 3.27029610793e-02 - 3.16057035862e-02 2.95890354456e-02 2.73903662614e-02 2.62352804996e-02 2.70513354759e-02 2.92436594720e-02 - 3.26920605347e-02 4.15214050147e-02 5.87809730551e-02 7.81193802831e-02 8.97664201881e-02 8.97664201881e-02 - 4.15726567626e-02 3.31551669202e-02 2.81921950446e-02 2.52955690008e-02 2.45139707404e-02 2.65224863971e-02 - 3.03789244340e-02 3.48628450776e-02 3.89798518705e-02 4.14583134664e-02 4.14583134664e-02 3.89798518705e-02 - 3.48628450776e-02 3.03789244340e-02 2.65224863971e-02 2.45139707405e-02 2.52955690009e-02 2.81921950447e-02 - 3.31551669202e-02 4.15726567626e-02 5.18158087398e-02 5.87902882083e-02 5.87902882083e-02 5.18158087398e-02 - 3.31551669202e-02 2.73937018060e-02 2.36141349499e-02 2.20787139947e-02 2.36141349499e-02 2.73937018060e-02 - 3.31551669201e-02 4.11388919901e-02 4.86717503238e-02 5.17382028008e-02 4.86717503238e-02 4.11388919901e-02 - 3.31551669202e-02 2.73937018060e-02 2.36141349500e-02 2.20787139947e-02 2.36141349499e-02 2.73937018060e-02 - 3.31551669202e-02 4.11388919901e-02 4.86717503238e-02 5.17382028009e-02 4.86717503238e-02 4.11388919901e-02 - 2.81921950446e-02 2.36141349499e-02 2.09116171502e-02 2.10846196163e-02 2.35389825848e-02 2.95749315132e-02 - 4.23899857519e-02 5.87809730550e-02 6.98255714909e-02 6.98255714909e-02 5.87809730550e-02 4.23899857520e-02 - 2.95749315132e-02 2.35389825848e-02 2.10846196163e-02 2.09116171502e-02 2.36141349499e-02 2.81921950446e-02 - 3.39196724265e-02 4.00842207789e-02 4.42637285519e-02 4.42637285519e-02 4.00842207789e-02 3.39196724265e-02 - 2.52955690008e-02 2.20787139947e-02 2.10846196163e-02 2.17746576295e-02 2.67216511960e-02 4.30330806506e-02 - 6.87950359263e-02 9.00454509227e-02 9.74781930219e-02 9.00454509227e-02 6.87950359263e-02 4.30330806506e-02 - 2.67216511960e-02 2.17746576295e-02 2.10846196163e-02 2.20787139947e-02 2.52955690008e-02 2.97974895143e-02 - 3.46096179590e-02 3.86294987685e-02 4.02266025417e-02 3.86294987685e-02 3.46096179590e-02 2.97974895143e-02 - 2.45139707404e-02 2.36141349499e-02 2.35389825848e-02 2.67216511960e-02 4.31931066266e-02 7.44578962493e-02 - 1.02255317231e-01 1.13479602981e-01 1.13479602981e-01 1.02255317231e-01 7.44578962494e-02 4.31931066266e-02 - 2.67216511960e-02 2.35389825848e-02 2.36141349499e-02 2.45139707404e-02 2.70513354759e-02 3.05548189106e-02 - 3.39942859271e-02 3.61935553818e-02 3.61935553818e-02 3.39942859271e-02 3.05548189106e-02 2.70513354759e-02 - 2.65224863971e-02 2.73937018060e-02 2.95749315132e-02 4.30330806506e-02 7.44578962493e-02 1.05382836884e-01 - 1.14284409879e-01 1.12133375345e-01 1.14284409879e-01 1.05382836884e-01 7.44578962494e-02 4.30330806506e-02 - 2.95749315132e-02 2.73937018060e-02 2.65224863971e-02 2.61140936360e-02 2.72051400328e-02 2.92643382604e-02 - 3.12134113093e-02 3.20070729423e-02 3.12134113093e-02 2.92643382604e-02 2.72051400328e-02 2.61140936360e-02 - 3.03789244340e-02 3.31551669202e-02 4.23899857520e-02 6.87950359263e-02 1.02255317231e-01 1.14284409879e-01 - 1.05271425072e-01 1.05271425072e-01 1.14284409879e-01 1.02255317231e-01 6.87950359264e-02 4.23899857520e-02 - 3.31551669202e-02 3.03789244340e-02 2.75584598461e-02 2.55791633527e-02 2.52445347088e-02 2.59181147281e-02 - 2.65858530209e-02 2.65858530209e-02 2.59181147281e-02 2.52445347088e-02 2.55791633527e-02 2.75584598461e-02 - 3.48628450776e-02 4.11388919901e-02 5.87809730550e-02 9.00454509227e-02 1.13479602981e-01 1.12133375345e-01 - 1.05271425072e-01 1.12133375345e-01 1.13479602981e-01 9.00454509228e-02 5.87809730550e-02 4.11388919902e-02 - 3.48628450776e-02 3.04051143293e-02 2.59618688149e-02 2.30253665657e-02 2.17948830411e-02 2.15195877577e-02 - 2.15038028103e-02 2.15195877577e-02 2.17948830411e-02 2.30253665657e-02 2.59618688149e-02 3.04051143293e-02 - 3.89798518706e-02 4.86717503239e-02 6.98255714910e-02 9.74781930220e-02 1.13479602981e-01 1.14284409879e-01 - 1.14284409879e-01 1.13479602981e-01 9.74781930221e-02 6.98255714910e-02 4.86717503239e-02 3.89798518706e-02 - 3.31963581644e-02 2.73091501295e-02 2.24043982374e-02 1.93871115228e-02 1.78888288560e-02 1.72812015630e-02 - 1.72812015630e-02 1.78888288560e-02 1.93871115228e-02 2.24043982374e-02 2.73091501295e-02 3.31963581644e-02 - 4.14583134664e-02 5.17382028009e-02 6.98255714910e-02 9.00454509228e-02 1.02255317232e-01 1.05382836884e-01 - 1.02255317232e-01 9.00454509228e-02 6.98255714910e-02 5.17382028009e-02 4.14583134665e-02 3.49769025118e-02 - 2.84217494631e-02 2.23752455712e-02 1.81179302207e-02 1.56224022211e-02 1.43188164440e-02 1.38961516945e-02 - 1.43188164440e-02 1.56224022211e-02 1.81179302207e-02 2.23752455712e-02 2.84217494631e-02 3.49769025117e-02 - 4.14583134664e-02 4.86717503239e-02 5.87809730551e-02 6.87950359264e-02 7.44578962495e-02 7.44578962495e-02 - 6.87950359264e-02 5.87809730551e-02 4.86717503239e-02 4.14583134665e-02 3.55406204185e-02 2.88778389824e-02 - 2.22713307355e-02 1.72611853896e-02 1.41149739580e-02 1.22934163580e-02 1.14031750108e-02 1.14031750108e-02 - 1.22934163580e-02 1.41149739580e-02 1.72611853896e-02 2.22713307355e-02 2.88778389824e-02 3.55406204185e-02 - 3.89798518706e-02 4.11388919902e-02 4.23899857520e-02 4.30330806507e-02 4.31931066267e-02 4.30330806507e-02 - 4.23899857520e-02 4.11388919902e-02 3.89798518706e-02 3.49769025118e-02 2.88778389824e-02 2.21709602700e-02 - 1.66950035376e-02 1.30618774824e-02 1.08500532301e-02 9.57619631350e-03 9.13724506496e-03 9.57619631350e-03 - 1.08500532301e-02 1.30618774824e-02 1.66950035376e-02 2.21709602700e-02 2.88778389824e-02 3.49769025118e-02 - 3.48628450776e-02 3.31551669202e-02 2.95749315132e-02 2.67216511960e-02 2.67216511960e-02 2.95749315132e-02 - 3.31551669202e-02 3.48628450776e-02 3.31963581644e-02 2.84217494631e-02 2.22713307355e-02 1.66950035376e-02 - 1.26511415702e-02 1.00210602336e-02 8.36360112567e-03 7.50451749096e-03 7.50451749096e-03 8.36360112567e-03 - 1.00210602336e-02 1.26511415702e-02 1.66950035376e-02 2.22713307355e-02 2.84217494631e-02 3.31963581644e-02 - 3.03789244340e-02 2.73937018061e-02 2.35389825848e-02 2.17746576296e-02 2.35389825848e-02 2.73937018061e-02 - 3.03789244340e-02 3.04051143294e-02 2.73091501295e-02 2.23752455712e-02 1.72611853896e-02 1.30618774824e-02 - 1.00210602336e-02 7.92498387888e-03 6.62087909721e-03 6.16385997204e-03 6.62087909721e-03 7.92498387888e-03 - 1.00210602336e-02 1.30618774824e-02 1.72611853896e-02 2.23752455712e-02 2.73091501295e-02 3.04051143294e-02 - 2.65224863971e-02 2.36141349500e-02 2.10846196163e-02 2.10846196163e-02 2.36141349500e-02 2.65224863971e-02 - 2.75584598462e-02 2.59618688149e-02 2.24043982374e-02 1.81179302207e-02 1.41149739580e-02 1.08500532301e-02 - 8.36360112567e-03 6.62087909721e-03 5.69328932590e-03 5.69328932590e-03 6.62087909721e-03 8.36360112567e-03 - 1.08500532301e-02 1.41149739580e-02 1.81179302207e-02 2.24043982374e-02 2.59618688149e-02 2.75584598462e-02 - 2.45139707405e-02 2.20787139947e-02 2.09116171503e-02 2.20787139948e-02 2.45139707405e-02 2.61140936361e-02 - 2.55791633528e-02 2.30253665657e-02 1.93871115228e-02 1.56224022211e-02 1.22934163580e-02 9.57619631351e-03 - 7.50451749097e-03 6.16385997204e-03 5.69328932590e-03 6.16385997204e-03 7.50451749097e-03 9.57619631351e-03 - 1.22934163580e-02 1.56224022211e-02 1.93871115228e-02 2.30253665657e-02 2.55791633528e-02 2.61140936361e-02 - 2.52955690009e-02 2.36141349500e-02 2.36141349500e-02 2.52955690009e-02 2.70513354759e-02 2.72051400328e-02 - 2.52445347088e-02 2.17948830411e-02 1.78888288560e-02 1.43188164440e-02 1.14031750108e-02 9.13724506496e-03 - 7.50451749096e-03 6.62087909721e-03 6.62087909721e-03 7.50451749096e-03 9.13724506496e-03 1.14031750108e-02 - 1.43188164440e-02 1.78888288560e-02 2.17948830411e-02 2.52445347088e-02 2.72051400328e-02 2.70513354759e-02 - 2.81921950447e-02 2.73937018061e-02 2.81921950447e-02 2.97974895143e-02 3.05548189106e-02 2.92643382604e-02 - 2.59181147281e-02 2.15195877577e-02 1.72812015630e-02 1.38961516944e-02 1.14031750108e-02 9.57619631350e-03 - 8.36360112567e-03 7.92498387887e-03 8.36360112567e-03 9.57619631350e-03 1.14031750108e-02 1.38961516944e-02 - 1.72812015630e-02 2.15195877577e-02 2.59181147281e-02 2.92643382604e-02 3.05548189106e-02 2.97974895143e-02 - 3.31551669202e-02 3.31551669202e-02 3.39196724265e-02 3.46096179590e-02 3.39942859271e-02 3.12134113093e-02 - 2.65858530209e-02 2.15038028103e-02 1.72812015630e-02 1.43188164440e-02 1.22934163580e-02 1.08500532301e-02 - 1.00210602336e-02 1.00210602336e-02 1.08500532301e-02 1.22934163580e-02 1.43188164440e-02 1.72812015630e-02 - 2.15038028103e-02 2.65858530209e-02 3.12134113093e-02 3.39942859271e-02 3.46096179590e-02 3.39196724265e-02 - 4.15726567627e-02 4.11388919902e-02 4.00842207790e-02 3.86294987685e-02 3.61935553818e-02 3.20070729422e-02 - 2.65858530209e-02 2.15195877577e-02 1.78888288560e-02 1.56224022211e-02 1.41149739580e-02 1.30618774824e-02 - 1.26511415702e-02 1.30618774824e-02 1.41149739580e-02 1.56224022211e-02 1.78888288560e-02 2.15195877577e-02 - 2.65858530209e-02 3.20070729422e-02 3.61935553818e-02 3.86294987685e-02 4.00842207790e-02 4.11388919902e-02 - 5.18158087398e-02 4.86717503239e-02 4.42637285519e-02 4.02266025417e-02 3.61935553818e-02 3.12134113093e-02 - 2.59181147280e-02 2.17948830411e-02 1.93871115228e-02 1.81179302207e-02 1.72611853896e-02 1.66950035376e-02 - 1.66950035376e-02 1.72611853896e-02 1.81179302207e-02 1.93871115228e-02 2.17948830411e-02 2.59181147281e-02 - 3.12134113093e-02 3.61935553818e-02 4.02266025417e-02 4.42637285519e-02 4.86717503239e-02 5.18158087398e-02 - 5.87902882084e-02 5.17382028009e-02 4.42637285519e-02 3.86294987685e-02 3.39942859271e-02 2.92643382604e-02 - 2.52445347087e-02 2.30253665656e-02 2.24043982374e-02 2.23752455712e-02 2.22713307355e-02 2.21709602700e-02 - 2.22713307355e-02 2.23752455712e-02 2.24043982374e-02 2.30253665657e-02 2.52445347088e-02 2.92643382604e-02 - 3.39942859271e-02 3.86294987685e-02 4.42637285519e-02 5.17382028009e-02 5.87902882084e-02 6.17008685349e-02 - 5.87902882084e-02 4.86717503239e-02 4.00842207790e-02 3.46096179590e-02 3.05548189106e-02 2.72051400328e-02 - 2.55791633527e-02 2.59618688149e-02 2.73091501295e-02 2.84217494631e-02 2.88778389824e-02 2.88778389824e-02 - 2.84217494631e-02 2.73091501295e-02 2.59618688149e-02 2.55791633527e-02 2.72051400328e-02 3.05548189106e-02 - 3.46096179591e-02 4.00842207790e-02 4.86717503239e-02 5.87902882084e-02 6.56428313091e-02 6.56428313091e-02 - 5.18158087398e-02 4.11388919902e-02 3.39196724265e-02 2.97974895143e-02 2.70513354759e-02 2.61140936360e-02 - 2.75584598461e-02 3.04051143293e-02 3.31963581643e-02 3.49769025117e-02 3.55406204185e-02 3.49769025117e-02 - 3.31963581644e-02 3.04051143293e-02 2.75584598461e-02 2.61140936360e-02 2.70513354759e-02 2.97974895143e-02 - 3.39196724265e-02 4.11388919902e-02 5.18158087398e-02 6.17008685349e-02 6.56428313091e-02 6.17008685349e-02 - 3.48683159917e-02 3.03789244340e-02 2.69081613820e-02 2.54927185045e-02 2.69081613820e-02 3.03789244340e-02 - 3.48683159917e-02 3.97328533270e-02 4.36605690195e-02 4.51578187598e-02 4.36605690195e-02 3.97328533271e-02 - 3.48683159917e-02 3.03789244340e-02 2.69081613820e-02 2.54927185045e-02 2.69081613820e-02 3.03789244340e-02 - 3.48683159917e-02 3.97328533271e-02 4.36605690195e-02 4.51578187598e-02 4.36605690195e-02 3.97328533271e-02 - 3.03789244340e-02 2.65224863971e-02 2.45139707404e-02 2.52955690008e-02 2.81921950446e-02 3.31551669202e-02 - 4.15726567626e-02 5.18158087397e-02 5.87902882083e-02 5.87902882083e-02 5.18158087398e-02 4.15726567627e-02 - 3.31551669202e-02 2.81921950447e-02 2.52955690008e-02 2.45139707404e-02 2.65224863971e-02 3.03789244340e-02 - 3.48628450776e-02 3.89798518705e-02 4.14583134664e-02 4.14583134664e-02 3.89798518706e-02 3.48628450776e-02 - 2.69081613820e-02 2.45139707404e-02 2.44579386769e-02 2.61164989591e-02 3.05273339684e-02 4.23899857519e-02 - 6.13624406409e-02 7.81193802830e-02 8.44119137021e-02 7.81193802830e-02 6.13624406410e-02 4.23899857520e-02 - 3.05273339684e-02 2.61164989591e-02 2.44579386769e-02 2.45139707404e-02 2.69081613820e-02 3.06792139923e-02 - 3.45751625707e-02 3.75326171916e-02 3.86293456568e-02 3.75326171916e-02 3.45751625707e-02 3.06792139923e-02 - 2.54927185045e-02 2.52955690008e-02 2.61164989591e-02 2.92890570903e-02 4.26148361403e-02 6.87950359263e-02 - 9.46618817553e-02 1.07504612679e-01 1.07504612680e-01 9.46618817554e-02 6.87950359264e-02 4.26148361403e-02 - 2.92890570903e-02 2.61164989591e-02 2.52955690008e-02 2.54927185045e-02 2.74769895380e-02 3.04773481040e-02 - 3.33647463124e-02 3.51262182185e-02 3.51262182185e-02 3.33647463124e-02 3.04773481040e-02 2.74769895380e-02 - 2.69081613820e-02 2.81921950446e-02 3.05273339684e-02 4.26148361403e-02 7.14595557376e-02 1.02255317231e-01 - 1.14984029929e-01 1.15476799016e-01 1.14984029929e-01 1.02255317231e-01 7.14595557377e-02 4.26148361403e-02 - 3.05273339684e-02 2.81921950446e-02 2.69081613820e-02 2.62352804996e-02 2.71142998602e-02 2.89473471924e-02 - 3.06728014251e-02 3.13656133728e-02 3.06728014251e-02 2.89473471924e-02 2.71142998602e-02 2.62352804996e-02 - 3.03789244340e-02 3.31551669202e-02 4.23899857519e-02 6.87950359263e-02 1.02255317231e-01 1.14284409879e-01 - 1.05271425072e-01 1.05271425072e-01 1.14284409879e-01 1.02255317231e-01 6.87950359263e-02 4.23899857520e-02 - 3.31551669202e-02 3.03789244340e-02 2.75584598461e-02 2.55791633527e-02 2.52445347088e-02 2.59181147281e-02 - 2.65858530209e-02 2.65858530209e-02 2.59181147281e-02 2.52445347088e-02 2.55791633527e-02 2.75584598461e-02 - 3.48683159917e-02 4.15726567626e-02 6.13624406409e-02 9.46618817553e-02 1.14984029929e-01 1.05271425072e-01 - 9.35850333253e-02 1.05271425072e-01 1.14984029929e-01 9.46618817554e-02 6.13624406410e-02 4.15726567626e-02 - 3.48683159917e-02 3.06058831029e-02 2.62414819209e-02 2.33118163733e-02 2.21386056500e-02 2.19611077293e-02 - 2.19947599998e-02 2.19611077293e-02 2.21386056500e-02 2.33118163733e-02 2.62414819209e-02 3.06058831029e-02 - 3.97328533271e-02 5.18158087398e-02 7.81193802830e-02 1.07504612680e-01 1.15476799016e-01 1.05271425072e-01 - 1.05271425072e-01 1.15476799016e-01 1.07504612680e-01 7.81193802831e-02 5.18158087398e-02 3.97328533271e-02 - 3.38925854809e-02 2.81653274579e-02 2.31387869200e-02 1.99975947255e-02 1.85132763207e-02 1.79741436761e-02 - 1.79741436761e-02 1.85132763207e-02 1.99975947255e-02 2.31387869200e-02 2.81653274579e-02 3.38925854809e-02 - 4.36605690195e-02 5.87902882084e-02 8.44119137022e-02 1.07504612680e-01 1.14984029929e-01 1.14284409879e-01 - 1.14984029929e-01 1.07504612680e-01 8.44119137023e-02 5.87902882084e-02 4.36605690195e-02 3.62673468746e-02 - 2.99301385615e-02 2.37301976795e-02 1.91106081253e-02 1.63906429721e-02 1.50424663757e-02 1.46288647205e-02 - 1.50424663757e-02 1.63906429721e-02 1.91106081253e-02 2.37301976795e-02 2.99301385615e-02 3.62673468746e-02 - 4.51578187598e-02 5.87902882084e-02 7.81193802831e-02 9.46618817555e-02 1.02255317232e-01 1.02255317232e-01 - 9.46618817555e-02 7.81193802831e-02 5.87902882084e-02 4.51578187598e-02 3.73540502297e-02 3.07822057975e-02 - 2.40967233573e-02 1.86355176327e-02 1.50739196267e-02 1.30485807277e-02 1.21045128275e-02 1.21045128275e-02 - 1.30485807277e-02 1.50739196267e-02 1.86355176327e-02 2.40967233573e-02 3.07822057975e-02 3.73540502296e-02 - 4.36605690195e-02 5.18158087398e-02 6.13624406410e-02 6.87950359264e-02 7.14595557378e-02 6.87950359264e-02 - 6.13624406410e-02 5.18158087398e-02 4.36605690195e-02 3.73540502297e-02 3.09609358381e-02 2.40904203272e-02 - 1.81617046076e-02 1.40806383265e-02 1.16095381953e-02 1.02401100733e-02 9.78219490488e-03 1.02401100733e-02 - 1.16095381953e-02 1.40806383265e-02 1.81617046076e-02 2.40904203272e-02 3.09609358381e-02 3.73540502297e-02 - 3.97328533271e-02 4.15726567627e-02 4.23899857520e-02 4.26148361404e-02 4.26148361404e-02 4.23899857520e-02 - 4.15726567627e-02 3.97328533271e-02 3.62673468746e-02 3.07822057975e-02 2.40904203272e-02 1.79266526907e-02 - 1.34421324248e-02 1.06051613742e-02 8.89920278516e-03 8.04455907974e-03 8.04455907974e-03 8.89920278516e-03 - 1.06051613742e-02 1.34421324248e-02 1.79266526907e-02 2.40904203272e-02 3.07822057975e-02 3.62673468746e-02 - 3.48683159917e-02 3.31551669202e-02 3.05273339685e-02 2.92890570903e-02 3.05273339685e-02 3.31551669202e-02 - 3.48683159917e-02 3.38925854810e-02 2.99301385615e-02 2.40967233573e-02 1.81617046076e-02 1.34421324248e-02 - 1.02248872526e-02 8.15758815908e-03 6.92803651720e-03 6.50338402317e-03 6.92803651720e-03 8.15758815908e-03 - 1.02248872526e-02 1.34421324248e-02 1.81617046076e-02 2.40967233573e-02 2.99301385615e-02 3.38925854809e-02 - 3.03789244340e-02 2.81921950447e-02 2.61164989591e-02 2.61164989592e-02 2.81921950447e-02 3.03789244340e-02 - 3.06058831030e-02 2.81653274579e-02 2.37301976795e-02 1.86355176327e-02 1.40806383265e-02 1.06051613742e-02 - 8.15758815908e-03 6.53729803367e-03 5.69411798450e-03 5.69411798450e-03 6.53729803367e-03 8.15758815908e-03 - 1.06051613742e-02 1.40806383265e-02 1.86355176327e-02 2.37301976795e-02 2.81653274579e-02 3.06058831030e-02 - 2.69081613821e-02 2.52955690009e-02 2.44579386769e-02 2.52955690009e-02 2.69081613821e-02 2.75584598462e-02 - 2.62414819209e-02 2.31387869200e-02 1.91106081254e-02 1.50739196267e-02 1.16095381953e-02 8.89920278516e-03 - 6.92803651720e-03 5.69411798450e-03 5.26652136817e-03 5.69411798450e-03 6.92803651720e-03 8.89920278516e-03 - 1.16095381953e-02 1.50739196267e-02 1.91106081253e-02 2.31387869200e-02 2.62414819209e-02 2.75584598462e-02 - 2.54927185046e-02 2.45139707405e-02 2.45139707405e-02 2.54927185046e-02 2.62352804996e-02 2.55791633528e-02 - 2.33118163733e-02 1.99975947255e-02 1.63906429721e-02 1.30485807277e-02 1.02401100733e-02 8.04455907974e-03 - 6.50338402317e-03 5.69411798450e-03 5.69411798450e-03 6.50338402317e-03 8.04455907974e-03 1.02401100733e-02 - 1.30485807277e-02 1.63906429721e-02 1.99975947255e-02 2.33118163733e-02 2.55791633528e-02 2.62352804996e-02 - 2.69081613821e-02 2.65224863971e-02 2.69081613821e-02 2.74769895380e-02 2.71142998602e-02 2.52445347088e-02 - 2.21386056500e-02 1.85132763207e-02 1.50424663757e-02 1.21045128275e-02 9.78219490488e-03 8.04455907974e-03 - 6.92803651720e-03 6.53729803366e-03 6.92803651720e-03 8.04455907974e-03 9.78219490488e-03 1.21045128275e-02 - 1.50424663757e-02 1.85132763207e-02 2.21386056500e-02 2.52445347088e-02 2.71142998602e-02 2.74769895380e-02 - 3.03789244340e-02 3.03789244340e-02 3.06792139923e-02 3.04773481040e-02 2.89473471924e-02 2.59181147281e-02 - 2.19611077293e-02 1.79741436760e-02 1.46288647205e-02 1.21045128275e-02 1.02401100733e-02 8.89920278515e-03 - 8.15758815907e-03 8.15758815907e-03 8.89920278515e-03 1.02401100733e-02 1.21045128275e-02 1.46288647205e-02 - 1.79741436760e-02 2.19611077293e-02 2.59181147281e-02 2.89473471924e-02 3.04773481040e-02 3.06792139923e-02 - 3.48683159917e-02 3.48628450777e-02 3.45751625707e-02 3.33647463124e-02 3.06728014251e-02 2.65858530209e-02 - 2.19947599998e-02 1.79741436760e-02 1.50424663757e-02 1.30485807277e-02 1.16095381953e-02 1.06051613742e-02 - 1.02248872526e-02 1.06051613742e-02 1.16095381953e-02 1.30485807277e-02 1.50424663757e-02 1.79741436760e-02 - 2.19947599998e-02 2.65858530209e-02 3.06728014251e-02 3.33647463124e-02 3.45751625707e-02 3.48628450776e-02 - 3.97328533271e-02 3.89798518706e-02 3.75326171916e-02 3.51262182185e-02 3.13656133727e-02 2.65858530209e-02 - 2.19611077293e-02 1.85132763207e-02 1.63906429721e-02 1.50739196267e-02 1.40806383265e-02 1.34421324248e-02 - 1.34421324248e-02 1.40806383265e-02 1.50739196267e-02 1.63906429721e-02 1.85132763207e-02 2.19611077293e-02 - 2.65858530209e-02 3.13656133727e-02 3.51262182185e-02 3.75326171916e-02 3.89798518706e-02 3.97328533271e-02 - 4.36605690195e-02 4.14583134665e-02 3.86293456568e-02 3.51262182185e-02 3.06728014251e-02 2.59181147280e-02 - 2.21386056500e-02 1.99975947255e-02 1.91106081253e-02 1.86355176327e-02 1.81617046076e-02 1.79266526907e-02 - 1.81617046076e-02 1.86355176327e-02 1.91106081253e-02 1.99975947255e-02 2.21386056500e-02 2.59181147281e-02 - 3.06728014251e-02 3.51262182185e-02 3.86293456568e-02 4.14583134665e-02 4.36605690195e-02 4.45490987358e-02 - 4.51578187598e-02 4.14583134665e-02 3.75326171916e-02 3.33647463124e-02 2.89473471924e-02 2.52445347087e-02 - 2.33118163733e-02 2.31387869200e-02 2.37301976795e-02 2.40967233573e-02 2.40904203272e-02 2.40904203272e-02 - 2.40967233573e-02 2.37301976795e-02 2.31387869200e-02 2.33118163733e-02 2.52445347088e-02 2.89473471924e-02 - 3.33647463124e-02 3.75326171916e-02 4.14583134665e-02 4.51578187598e-02 4.76134439313e-02 4.76134439313e-02 - 4.36605690195e-02 3.89798518706e-02 3.45751625707e-02 3.04773481040e-02 2.71142998602e-02 2.55791633527e-02 - 2.62414819209e-02 2.81653274579e-02 2.99301385615e-02 3.07822057975e-02 3.09609358381e-02 3.07822057975e-02 - 2.99301385615e-02 2.81653274579e-02 2.62414819209e-02 2.55791633527e-02 2.71142998602e-02 3.04773481040e-02 - 3.45751625707e-02 3.89798518706e-02 4.36605690195e-02 4.76134439312e-02 4.91904413241e-02 4.76134439312e-02 - 3.97328533271e-02 3.48628450776e-02 3.06792139923e-02 2.74769895380e-02 2.62352804996e-02 2.75584598461e-02 - 3.06058831029e-02 3.38925854809e-02 3.62673468745e-02 3.73540502296e-02 3.73540502296e-02 3.62673468746e-02 - 3.38925854809e-02 3.06058831029e-02 2.75584598461e-02 2.62352804996e-02 2.74769895380e-02 3.06792139923e-02 - 3.48628450776e-02 3.97328533271e-02 4.45490987358e-02 4.76134439312e-02 4.76134439312e-02 4.45490987358e-02 - 3.06058831029e-02 2.75584598461e-02 2.62352804996e-02 2.74769895380e-02 3.06792139923e-02 3.48628450776e-02 - 3.97328533270e-02 4.45490987357e-02 4.76134439312e-02 4.76134439312e-02 4.45490987358e-02 3.97328533271e-02 - 3.48628450776e-02 3.06792139923e-02 2.74769895380e-02 2.62352804996e-02 2.75584598461e-02 3.06058831029e-02 - 3.38925854809e-02 3.62673468746e-02 3.73540502296e-02 3.73540502296e-02 3.62673468746e-02 3.38925854809e-02 - 2.75584598461e-02 2.61140936360e-02 2.70513354759e-02 2.97974895143e-02 3.39196724265e-02 4.11388919901e-02 - 5.18158087397e-02 6.17008685348e-02 6.56428313090e-02 6.17008685349e-02 5.18158087398e-02 4.11388919902e-02 - 3.39196724265e-02 2.97974895143e-02 2.70513354759e-02 2.61140936360e-02 2.75584598461e-02 3.04051143293e-02 - 3.31963581644e-02 3.49769025117e-02 3.55406204185e-02 3.49769025118e-02 3.31963581644e-02 3.04051143293e-02 - 2.62352804996e-02 2.70513354759e-02 2.92436594720e-02 3.26920605347e-02 4.15214050146e-02 5.87809730550e-02 - 7.81193802830e-02 8.97664201879e-02 8.97664201880e-02 7.81193802830e-02 5.87809730550e-02 4.15214050146e-02 - 3.26920605347e-02 2.92436594720e-02 2.70513354759e-02 2.62352804996e-02 2.73903662614e-02 2.95890354456e-02 - 3.16057035862e-02 3.27029610793e-02 3.27029610793e-02 3.16057035862e-02 2.95890354456e-02 2.73903662614e-02 - 2.74769895380e-02 2.97974895143e-02 3.26920605346e-02 4.15474983449e-02 6.30064198496e-02 9.00454509227e-02 - 1.07504612679e-01 1.12225405020e-01 1.07504612680e-01 9.00454509227e-02 6.30064198497e-02 4.15474983449e-02 - 3.26920605347e-02 2.97974895143e-02 2.74769895380e-02 2.61756575962e-02 2.65400648390e-02 2.78013937268e-02 - 2.89590113766e-02 2.93979324046e-02 2.89590113766e-02 2.78013937268e-02 2.65400648390e-02 2.61756575962e-02 - 3.06792139923e-02 3.39196724265e-02 4.15214050146e-02 6.30064198496e-02 9.40538638176e-02 1.13479602981e-01 - 1.15476799016e-01 1.15476799016e-01 1.13479602981e-01 9.40538638177e-02 6.30064198497e-02 4.15214050146e-02 - 3.39196724265e-02 3.06792139923e-02 2.73903662614e-02 2.51931467428e-02 2.46286629032e-02 2.49902463304e-02 - 2.54054430258e-02 2.54054430258e-02 2.49902463304e-02 2.46286629032e-02 2.51931467427e-02 2.73903662614e-02 - 3.48628450776e-02 4.11388919901e-02 5.87809730550e-02 9.00454509227e-02 1.13479602981e-01 1.12133375345e-01 - 1.05271425072e-01 1.12133375345e-01 1.13479602981e-01 9.00454509227e-02 5.87809730550e-02 4.11388919901e-02 - 3.48628450776e-02 3.04051143293e-02 2.59618688149e-02 2.30253665657e-02 2.17948830411e-02 2.15195877577e-02 - 2.15038028103e-02 2.15195877577e-02 2.17948830411e-02 2.30253665657e-02 2.59618688149e-02 3.04051143293e-02 - 3.97328533270e-02 5.18158087397e-02 7.81193802830e-02 1.07504612680e-01 1.15476799016e-01 1.05271425072e-01 - 1.05271425072e-01 1.15476799016e-01 1.07504612680e-01 7.81193802831e-02 5.18158087398e-02 3.97328533271e-02 - 3.38925854809e-02 2.81653274579e-02 2.31387869200e-02 1.99975947255e-02 1.85132763207e-02 1.79741436761e-02 - 1.79741436761e-02 1.85132763207e-02 1.99975947255e-02 2.31387869200e-02 2.81653274579e-02 3.38925854809e-02 - 4.45490987358e-02 6.17008685349e-02 8.97664201880e-02 1.12225405021e-01 1.15476799016e-01 1.12133375345e-01 - 1.15476799016e-01 1.12225405021e-01 8.97664201881e-02 6.17008685349e-02 4.45490987358e-02 3.66854482599e-02 - 3.04404777485e-02 2.42075414898e-02 1.94536795471e-02 1.66439398964e-02 1.52762540169e-02 1.48656767561e-02 - 1.52762540169e-02 1.66439398964e-02 1.94536795471e-02 2.42075414898e-02 3.04404777485e-02 3.66854482599e-02 - 4.76134439312e-02 6.56428313091e-02 8.97664201881e-02 1.07504612680e-01 1.13479602981e-01 1.13479602981e-01 - 1.07504612680e-01 8.97664201881e-02 6.56428313091e-02 4.76134439312e-02 3.82423088608e-02 3.17361436403e-02 - 2.51249889113e-02 1.94290489291e-02 1.55937378378e-02 1.34169388324e-02 1.24265880360e-02 1.24265880360e-02 - 1.34169388324e-02 1.55937378378e-02 1.94290489291e-02 2.51249889113e-02 3.17361436403e-02 3.82423088608e-02 - 4.76134439312e-02 6.17008685349e-02 7.81193802831e-02 9.00454509228e-02 9.40538638178e-02 9.00454509228e-02 - 7.81193802831e-02 6.17008685349e-02 4.76134439313e-02 3.86937384314e-02 3.20848019959e-02 2.53973709283e-02 - 1.92840317465e-02 1.48492046219e-02 1.21090534530e-02 1.06109348738e-02 1.01198499891e-02 1.06109348738e-02 - 1.21090534529e-02 1.48492046219e-02 1.92840317465e-02 2.53973709283e-02 3.20848019959e-02 3.86937384314e-02 - 4.45490987358e-02 5.18158087398e-02 5.87809730551e-02 6.30064198498e-02 6.30064198498e-02 5.87809730551e-02 - 5.18158087398e-02 4.45490987358e-02 3.82423088608e-02 3.20848019959e-02 2.53797374387e-02 1.90345757835e-02 - 1.42308801751e-02 1.11331987338e-02 9.28357359923e-03 8.37400390520e-03 8.37400390520e-03 9.28357359923e-03 - 1.11331987338e-02 1.42308801751e-02 1.90345757835e-02 2.53797374387e-02 3.20848019959e-02 3.82423088608e-02 - 3.97328533271e-02 4.11388919902e-02 4.15214050147e-02 4.15474983450e-02 4.15214050147e-02 4.11388919902e-02 - 3.97328533271e-02 3.66854482599e-02 3.17361436403e-02 2.53973709283e-02 1.90345757835e-02 1.39667079820e-02 - 1.05557815436e-02 8.41997198039e-03 7.17862759620e-03 6.75449318420e-03 7.17862759620e-03 8.41997198039e-03 - 1.05557815436e-02 1.39667079820e-02 1.90345757835e-02 2.53973709283e-02 3.17361436403e-02 3.66854482599e-02 - 3.48628450776e-02 3.39196724265e-02 3.26920605347e-02 3.26920605347e-02 3.39196724265e-02 3.48628450777e-02 - 3.38925854810e-02 3.04404777485e-02 2.51249889113e-02 1.92840317465e-02 1.42308801751e-02 1.05557815436e-02 - 8.11230157085e-03 6.56138198057e-03 5.76667312114e-03 5.76667312114e-03 6.56138198057e-03 8.11230157085e-03 - 1.05557815436e-02 1.42308801751e-02 1.92840317465e-02 2.51249889113e-02 3.04404777485e-02 3.38925854809e-02 - 3.06792139923e-02 2.97974895143e-02 2.92436594720e-02 2.97974895143e-02 3.06792139923e-02 3.04051143294e-02 - 2.81653274579e-02 2.42075414899e-02 1.94290489291e-02 1.48492046219e-02 1.11331987338e-02 8.41997198039e-03 - 6.56138198057e-03 5.43239778449e-03 5.04343778102e-03 5.43239778449e-03 6.56138198057e-03 8.41997198039e-03 - 1.11331987338e-02 1.48492046219e-02 1.94290489291e-02 2.42075414899e-02 2.81653274579e-02 3.04051143294e-02 - 2.74769895380e-02 2.70513354759e-02 2.70513354759e-02 2.74769895380e-02 2.73903662614e-02 2.59618688149e-02 - 2.31387869200e-02 1.94536795471e-02 1.55937378378e-02 1.21090534530e-02 9.28357359923e-03 7.17862759620e-03 - 5.76667312114e-03 5.04343778102e-03 5.04343778102e-03 5.76667312114e-03 7.17862759620e-03 9.28357359923e-03 - 1.21090534530e-02 1.55937378378e-02 1.94536795471e-02 2.31387869200e-02 2.59618688149e-02 2.73903662614e-02 - 2.62352804996e-02 2.61140936361e-02 2.62352804996e-02 2.61756575963e-02 2.51931467428e-02 2.30253665657e-02 - 1.99975947255e-02 1.66439398964e-02 1.34169388324e-02 1.06109348738e-02 8.37400390520e-03 6.75449318420e-03 - 5.76667312114e-03 5.43239778449e-03 5.76667312114e-03 6.75449318420e-03 8.37400390520e-03 1.06109348738e-02 - 1.34169388324e-02 1.66439398964e-02 1.99975947255e-02 2.30253665657e-02 2.51931467428e-02 2.61756575963e-02 - 2.75584598462e-02 2.75584598462e-02 2.73903662614e-02 2.65400648390e-02 2.46286629032e-02 2.17948830411e-02 - 1.85132763207e-02 1.52762540169e-02 1.24265880360e-02 1.01198499891e-02 8.37400390520e-03 7.17862759619e-03 - 6.56138198056e-03 6.56138198056e-03 7.17862759619e-03 8.37400390520e-03 1.01198499891e-02 1.24265880360e-02 - 1.52762540169e-02 1.85132763207e-02 2.17948830411e-02 2.46286629032e-02 2.65400648390e-02 2.73903662614e-02 - 3.06058831030e-02 3.04051143294e-02 2.95890354456e-02 2.78013937268e-02 2.49902463304e-02 2.15195877577e-02 - 1.79741436760e-02 1.48656767560e-02 1.24265880360e-02 1.06109348738e-02 9.28357359922e-03 8.41997198038e-03 - 8.11230157084e-03 8.41997198038e-03 9.28357359922e-03 1.06109348738e-02 1.24265880360e-02 1.48656767560e-02 - 1.79741436760e-02 2.15195877577e-02 2.49902463304e-02 2.78013937268e-02 2.95890354456e-02 3.04051143294e-02 - 3.38925854810e-02 3.31963581644e-02 3.16057035862e-02 2.89590113766e-02 2.54054430258e-02 2.15038028103e-02 - 1.79741436760e-02 1.52762540168e-02 1.34169388324e-02 1.21090534529e-02 1.11331987338e-02 1.05557815436e-02 - 1.05557815436e-02 1.11331987338e-02 1.21090534529e-02 1.34169388324e-02 1.52762540169e-02 1.79741436760e-02 - 2.15038028103e-02 2.54054430258e-02 2.89590113766e-02 3.16057035862e-02 3.31963581644e-02 3.38925854810e-02 - 3.62673468746e-02 3.49769025118e-02 3.27029610793e-02 2.93979324046e-02 2.54054430258e-02 2.15195877577e-02 - 1.85132763207e-02 1.66439398964e-02 1.55937378378e-02 1.48492046219e-02 1.42308801751e-02 1.39667079820e-02 - 1.42308801751e-02 1.48492046219e-02 1.55937378378e-02 1.66439398964e-02 1.85132763207e-02 2.15195877577e-02 - 2.54054430258e-02 2.93979324045e-02 3.27029610793e-02 3.49769025118e-02 3.62673468746e-02 3.66854482599e-02 - 3.73540502297e-02 3.55406204185e-02 3.27029610793e-02 2.89590113766e-02 2.49902463304e-02 2.17948830411e-02 - 1.99975947255e-02 1.94536795471e-02 1.94290489291e-02 1.92840317465e-02 1.90345757835e-02 1.90345757835e-02 - 1.92840317465e-02 1.94290489291e-02 1.94536795471e-02 1.99975947255e-02 2.17948830411e-02 2.49902463304e-02 - 2.89590113766e-02 3.27029610793e-02 3.55406204185e-02 3.73540502297e-02 3.82423088608e-02 3.82423088608e-02 - 3.73540502297e-02 3.49769025118e-02 3.16057035862e-02 2.78013937268e-02 2.46286629032e-02 2.30253665656e-02 - 2.31387869200e-02 2.42075414898e-02 2.51249889113e-02 2.53973709283e-02 2.53797374387e-02 2.53973709283e-02 - 2.51249889113e-02 2.42075414898e-02 2.31387869200e-02 2.30253665657e-02 2.46286629032e-02 2.78013937268e-02 - 3.16057035862e-02 3.49769025118e-02 3.73540502297e-02 3.86937384314e-02 3.91205047681e-02 3.86937384314e-02 - 3.62673468746e-02 3.31963581644e-02 2.95890354456e-02 2.65400648390e-02 2.51931467427e-02 2.59618688149e-02 - 2.81653274579e-02 3.04404777485e-02 3.17361436402e-02 3.20848019958e-02 3.20848019958e-02 3.17361436402e-02 - 3.04404777485e-02 2.81653274579e-02 2.59618688149e-02 2.51931467428e-02 2.65400648390e-02 2.95890354456e-02 - 3.31963581644e-02 3.62673468746e-02 3.82423088608e-02 3.91205047681e-02 3.91205047681e-02 3.82423088608e-02 - 3.38925854809e-02 3.04051143293e-02 2.73903662614e-02 2.61756575962e-02 2.73903662614e-02 3.04051143293e-02 - 3.38925854809e-02 3.66854482599e-02 3.82423088607e-02 3.86937384314e-02 3.82423088607e-02 3.66854482599e-02 - 3.38925854809e-02 3.04051143293e-02 2.73903662614e-02 2.61756575962e-02 2.73903662614e-02 3.04051143293e-02 - 3.38925854809e-02 3.66854482599e-02 3.82423088608e-02 3.86937384314e-02 3.82423088608e-02 3.66854482599e-02 - 2.62414819209e-02 2.55791633527e-02 2.71142998602e-02 3.04773481040e-02 3.45751625707e-02 3.89798518705e-02 - 4.36605690195e-02 4.76134439312e-02 4.91904413240e-02 4.76134439312e-02 4.36605690195e-02 3.89798518706e-02 - 3.45751625707e-02 3.04773481040e-02 2.71142998602e-02 2.55791633527e-02 2.62414819209e-02 2.81653274579e-02 - 2.99301385615e-02 3.07822057975e-02 3.09609358381e-02 3.07822057975e-02 2.99301385615e-02 2.81653274579e-02 - 2.55791633527e-02 2.72051400328e-02 3.05548189106e-02 3.46096179590e-02 4.00842207789e-02 4.86717503238e-02 - 5.87902882083e-02 6.56428313090e-02 6.56428313090e-02 5.87902882083e-02 4.86717503239e-02 4.00842207790e-02 - 3.46096179590e-02 3.05548189106e-02 2.72051400328e-02 2.55791633527e-02 2.59618688149e-02 2.73091501295e-02 - 2.84217494631e-02 2.88778389824e-02 2.88778389824e-02 2.84217494631e-02 2.73091501295e-02 2.59618688149e-02 - 2.71142998602e-02 3.05548189106e-02 3.44216532954e-02 4.03062291682e-02 5.24371687843e-02 6.98255714909e-02 - 8.44119137021e-02 8.97664201879e-02 8.44119137022e-02 6.98255714909e-02 5.24371687843e-02 4.03062291683e-02 - 3.44216532954e-02 3.05548189106e-02 2.71142998602e-02 2.51931467428e-02 2.49964037842e-02 2.55726183949e-02 - 2.60492747623e-02 2.61902387743e-02 2.60492747623e-02 2.55726183949e-02 2.49964037842e-02 2.51931467428e-02 - 3.04773481040e-02 3.46096179590e-02 4.03062291682e-02 5.38553081705e-02 7.65071402185e-02 9.74781930219e-02 - 1.07504612679e-01 1.07504612679e-01 9.74781930219e-02 7.65071402185e-02 5.38553081705e-02 4.03062291683e-02 - 3.46096179590e-02 3.04773481040e-02 2.65400648390e-02 2.40839675381e-02 2.32030440049e-02 2.30939042850e-02 - 2.31231583381e-02 2.31231583381e-02 2.30939042850e-02 2.32030440049e-02 2.40839675381e-02 2.65400648390e-02 - 3.45751625707e-02 4.00842207789e-02 5.24371687843e-02 7.65071402185e-02 1.01688733219e-01 1.13479602981e-01 - 1.14984029929e-01 1.13479602981e-01 1.01688733220e-01 7.65071402185e-02 5.24371687843e-02 4.00842207789e-02 - 3.45751625707e-02 2.95890354456e-02 2.49964037843e-02 2.20777720561e-02 2.07196781889e-02 2.02040886880e-02 - 2.00653973119e-02 2.02040886880e-02 2.07196781889e-02 2.20777720561e-02 2.49964037842e-02 2.95890354456e-02 - 3.89798518705e-02 4.86717503238e-02 6.98255714909e-02 9.74781930219e-02 1.13479602981e-01 1.14284409879e-01 - 1.14284409879e-01 1.13479602981e-01 9.74781930220e-02 6.98255714909e-02 4.86717503238e-02 3.89798518706e-02 - 3.31963581644e-02 2.73091501295e-02 2.24043982374e-02 1.93871115228e-02 1.78888288560e-02 1.72812015630e-02 - 1.72812015630e-02 1.78888288560e-02 1.93871115228e-02 2.24043982374e-02 2.73091501295e-02 3.31963581644e-02 - 4.36605690195e-02 5.87902882083e-02 8.44119137021e-02 1.07504612680e-01 1.14984029929e-01 1.14284409879e-01 - 1.14984029929e-01 1.07504612680e-01 8.44119137022e-02 5.87902882083e-02 4.36605690195e-02 3.62673468746e-02 - 2.99301385615e-02 2.37301976795e-02 1.91106081253e-02 1.63906429721e-02 1.50424663757e-02 1.46288647205e-02 - 1.50424663757e-02 1.63906429721e-02 1.91106081253e-02 2.37301976795e-02 2.99301385615e-02 3.62673468746e-02 - 4.76134439312e-02 6.56428313091e-02 8.97664201880e-02 1.07504612680e-01 1.13479602981e-01 1.13479602981e-01 - 1.07504612680e-01 8.97664201881e-02 6.56428313091e-02 4.76134439312e-02 3.82423088608e-02 3.17361436403e-02 - 2.51249889113e-02 1.94290489291e-02 1.55937378378e-02 1.34169388324e-02 1.24265880360e-02 1.24265880360e-02 - 1.34169388324e-02 1.55937378378e-02 1.94290489291e-02 2.51249889113e-02 3.17361436403e-02 3.82423088608e-02 - 4.91904413241e-02 6.56428313091e-02 8.44119137022e-02 9.74781930220e-02 1.01688733220e-01 9.74781930220e-02 - 8.44119137022e-02 6.56428313091e-02 4.91904413241e-02 3.91205047681e-02 3.24128421387e-02 2.58521940319e-02 - 1.97044449599e-02 1.51354763629e-02 1.22800512627e-02 1.07225418333e-02 1.02153626600e-02 1.07225418333e-02 - 1.22800512627e-02 1.51354763629e-02 1.97044449599e-02 2.58521940319e-02 3.24128421387e-02 3.91205047681e-02 - 4.76134439312e-02 5.87902882084e-02 6.98255714910e-02 7.65071402186e-02 7.65071402186e-02 6.98255714910e-02 - 5.87902882084e-02 4.76134439313e-02 3.91205047681e-02 3.25042138914e-02 2.59777844449e-02 1.96802339930e-02 - 1.47279181689e-02 1.14481144732e-02 9.47290433111e-03 8.50665152157e-03 8.50665152157e-03 9.47290433111e-03 - 1.14481144732e-02 1.47279181688e-02 1.96802339930e-02 2.59777844449e-02 3.25042138914e-02 3.91205047681e-02 - 4.36605690195e-02 4.86717503239e-02 5.24371687844e-02 5.38553081706e-02 5.24371687844e-02 4.86717503239e-02 - 4.36605690195e-02 3.82423088608e-02 3.24128421387e-02 2.59777844449e-02 1.96031535283e-02 1.44210111226e-02 - 1.08699825973e-02 8.62540500906e-03 7.31933010297e-03 6.87426464911e-03 7.31933010297e-03 8.62540500906e-03 - 1.08699825973e-02 1.44210111226e-02 1.96031535282e-02 2.59777844449e-02 3.24128421387e-02 3.82423088608e-02 - 3.89798518706e-02 4.00842207790e-02 4.03062291683e-02 4.03062291683e-02 4.00842207790e-02 3.89798518706e-02 - 3.62673468746e-02 3.17361436403e-02 2.58521940319e-02 1.96802339930e-02 1.44210111226e-02 1.06518186545e-02 - 8.18383164207e-03 6.62684043802e-03 5.82746138702e-03 5.82746138702e-03 6.62684043802e-03 8.18383164206e-03 - 1.06518186545e-02 1.44210111226e-02 1.96802339930e-02 2.58521940319e-02 3.17361436403e-02 3.62673468746e-02 - 3.45751625707e-02 3.46096179591e-02 3.44216532954e-02 3.46096179591e-02 3.45751625707e-02 3.31963581644e-02 - 2.99301385615e-02 2.51249889113e-02 1.97044449599e-02 1.47279181689e-02 1.08699825973e-02 8.18383164207e-03 - 6.39967919814e-03 5.32068131556e-03 4.94632325183e-03 5.32068131556e-03 6.39967919814e-03 8.18383164206e-03 - 1.08699825973e-02 1.47279181689e-02 1.97044449599e-02 2.51249889113e-02 2.99301385615e-02 3.31963581644e-02 - 3.04773481040e-02 3.05548189106e-02 3.05548189106e-02 3.04773481040e-02 2.95890354456e-02 2.73091501295e-02 - 2.37301976795e-02 1.94290489291e-02 1.51354763629e-02 1.14481144732e-02 8.62540500906e-03 6.62684043802e-03 - 5.32068131556e-03 4.65175947716e-03 4.65175947716e-03 5.32068131556e-03 6.62684043802e-03 8.62540500906e-03 - 1.14481144732e-02 1.51354763629e-02 1.94290489291e-02 2.37301976795e-02 2.73091501295e-02 2.95890354456e-02 - 2.71142998602e-02 2.72051400328e-02 2.71142998602e-02 2.65400648390e-02 2.49964037843e-02 2.24043982374e-02 - 1.91106081254e-02 1.55937378378e-02 1.22800512627e-02 9.47290433111e-03 7.31933010297e-03 5.82746138702e-03 - 4.94632325183e-03 4.65175947716e-03 4.94632325183e-03 5.82746138702e-03 7.31933010297e-03 9.47290433111e-03 - 1.22800512627e-02 1.55937378378e-02 1.91106081253e-02 2.24043982374e-02 2.49964037843e-02 2.65400648390e-02 - 2.55791633528e-02 2.55791633528e-02 2.51931467428e-02 2.40839675381e-02 2.20777720561e-02 1.93871115228e-02 - 1.63906429721e-02 1.34169388324e-02 1.07225418333e-02 8.50665152157e-03 6.87426464911e-03 5.82746138702e-03 - 5.32068131556e-03 5.32068131556e-03 5.82746138702e-03 6.87426464911e-03 8.50665152157e-03 1.07225418333e-02 - 1.34169388324e-02 1.63906429721e-02 1.93871115228e-02 2.20777720561e-02 2.40839675381e-02 2.51931467428e-02 - 2.62414819209e-02 2.59618688149e-02 2.49964037843e-02 2.32030440049e-02 2.07196781889e-02 1.78888288560e-02 - 1.50424663757e-02 1.24265880360e-02 1.02153626600e-02 8.50665152157e-03 7.31933010296e-03 6.62684043801e-03 - 6.39967919814e-03 6.62684043801e-03 7.31933010296e-03 8.50665152157e-03 1.02153626600e-02 1.24265880360e-02 - 1.50424663757e-02 1.78888288560e-02 2.07196781889e-02 2.32030440049e-02 2.49964037843e-02 2.59618688149e-02 - 2.81653274579e-02 2.73091501295e-02 2.55726183949e-02 2.30939042850e-02 2.02040886880e-02 1.72812015630e-02 - 1.46288647205e-02 1.24265880360e-02 1.07225418333e-02 9.47290433110e-03 8.62540500905e-03 8.18383164206e-03 - 8.18383164206e-03 8.62540500905e-03 9.47290433110e-03 1.07225418333e-02 1.24265880360e-02 1.46288647205e-02 - 1.72812015630e-02 2.02040886880e-02 2.30939042849e-02 2.55726183949e-02 2.73091501295e-02 2.81653274579e-02 - 2.99301385615e-02 2.84217494631e-02 2.60492747623e-02 2.31231583381e-02 2.00653973118e-02 1.72812015630e-02 - 1.50424663757e-02 1.34169388324e-02 1.22800512627e-02 1.14481144732e-02 1.08699825973e-02 1.06518186545e-02 - 1.08699825973e-02 1.14481144732e-02 1.22800512627e-02 1.34169388324e-02 1.50424663757e-02 1.72812015630e-02 - 2.00653973118e-02 2.31231583381e-02 2.60492747623e-02 2.84217494631e-02 2.99301385615e-02 3.04404777485e-02 - 3.07822057975e-02 2.88778389824e-02 2.61902387743e-02 2.31231583381e-02 2.02040886880e-02 1.78888288560e-02 - 1.63906429721e-02 1.55937378378e-02 1.51354763629e-02 1.47279181688e-02 1.44210111226e-02 1.44210111226e-02 - 1.47279181688e-02 1.51354763629e-02 1.55937378378e-02 1.63906429721e-02 1.78888288560e-02 2.02040886880e-02 - 2.31231583381e-02 2.61902387743e-02 2.88778389824e-02 3.07822057975e-02 3.17361436403e-02 3.17361436403e-02 - 3.09609358381e-02 2.88778389824e-02 2.60492747623e-02 2.30939042849e-02 2.07196781889e-02 1.93871115227e-02 - 1.91106081253e-02 1.94290489291e-02 1.97044449599e-02 1.96802339929e-02 1.96031535282e-02 1.96802339930e-02 - 1.97044449599e-02 1.94290489291e-02 1.91106081253e-02 1.93871115228e-02 2.07196781889e-02 2.30939042849e-02 - 2.60492747622e-02 2.88778389824e-02 3.09609358381e-02 3.20848019959e-02 3.24128421387e-02 3.20848019959e-02 - 3.07822057975e-02 2.84217494631e-02 2.55726183949e-02 2.32030440049e-02 2.20777720560e-02 2.24043982374e-02 - 2.37301976795e-02 2.51249889113e-02 2.58521940319e-02 2.59777844449e-02 2.59777844449e-02 2.58521940319e-02 - 2.51249889113e-02 2.37301976795e-02 2.24043982374e-02 2.20777720561e-02 2.32030440049e-02 2.55726183949e-02 - 2.84217494631e-02 3.07822057975e-02 3.20848019959e-02 3.25042138914e-02 3.25042138914e-02 3.20848019959e-02 - 2.99301385615e-02 2.73091501295e-02 2.49964037842e-02 2.40839675381e-02 2.49964037842e-02 2.73091501295e-02 - 2.99301385615e-02 3.17361436402e-02 3.24128421387e-02 3.25042138914e-02 3.24128421387e-02 3.17361436402e-02 - 2.99301385615e-02 2.73091501295e-02 2.49964037842e-02 2.40839675381e-02 2.49964037842e-02 2.73091501295e-02 - 2.99301385615e-02 3.17361436402e-02 3.24128421387e-02 3.25042138914e-02 3.24128421387e-02 3.17361436403e-02 - 2.81653274579e-02 2.59618688149e-02 2.51931467427e-02 2.65400648390e-02 2.95890354456e-02 3.31963581643e-02 - 3.62673468745e-02 3.82423088607e-02 3.91205047681e-02 3.91205047681e-02 3.82423088608e-02 3.62673468746e-02 - 3.31963581644e-02 2.95890354456e-02 2.65400648390e-02 2.51931467428e-02 2.59618688149e-02 2.81653274579e-02 - 3.04404777485e-02 3.17361436402e-02 3.20848019959e-02 3.20848019959e-02 3.17361436403e-02 3.04404777485e-02 - 2.33118163733e-02 2.52445347088e-02 2.89473471924e-02 3.33647463123e-02 3.75326171916e-02 4.14583134664e-02 - 4.51578187598e-02 4.76134439312e-02 4.76134439312e-02 4.51578187598e-02 4.14583134664e-02 3.75326171916e-02 - 3.33647463123e-02 2.89473471924e-02 2.52445347088e-02 2.33118163733e-02 2.31387869200e-02 2.37301976795e-02 - 2.40967233573e-02 2.40904203272e-02 2.40904203272e-02 2.40967233573e-02 2.37301976795e-02 2.31387869200e-02 - 2.52445347088e-02 2.92643382604e-02 3.39942859271e-02 3.86294987685e-02 4.42637285519e-02 5.17382028008e-02 - 5.87902882083e-02 6.17008685348e-02 5.87902882083e-02 5.17382028008e-02 4.42637285519e-02 3.86294987685e-02 - 3.39942859271e-02 2.92643382604e-02 2.52445347088e-02 2.30253665657e-02 2.24043982374e-02 2.23752455712e-02 - 2.22713307355e-02 2.21709602700e-02 2.22713307355e-02 2.23752455712e-02 2.24043982374e-02 2.30253665657e-02 - 2.89473471924e-02 3.39942859271e-02 3.88811262511e-02 4.56987320892e-02 5.69437593911e-02 6.98255714909e-02 - 7.81193802830e-02 7.81193802830e-02 6.98255714909e-02 5.69437593911e-02 4.56987320892e-02 3.88811262511e-02 - 3.39942859271e-02 2.89473471924e-02 2.46286629032e-02 2.20777720561e-02 2.09446754967e-02 2.03920879638e-02 - 2.00583037371e-02 2.00583037371e-02 2.03920879638e-02 2.09446754967e-02 2.20777720561e-02 2.46286629032e-02 - 3.33647463123e-02 3.86294987685e-02 4.56987320892e-02 5.89461275928e-02 7.65071402184e-02 9.00454509226e-02 - 9.46618817553e-02 9.00454509227e-02 7.65071402185e-02 5.89461275929e-02 4.56987320892e-02 3.86294987685e-02 - 3.33647463124e-02 2.78013937268e-02 2.32030440049e-02 2.03983415996e-02 1.89263764236e-02 1.81430425204e-02 - 1.78657504093e-02 1.81430425204e-02 1.89263764236e-02 2.03983415996e-02 2.32030440049e-02 2.78013937268e-02 - 3.75326171916e-02 4.42637285519e-02 5.69437593911e-02 7.65071402184e-02 9.40538638176e-02 1.02255317231e-01 - 1.02255317231e-01 9.40538638177e-02 7.65071402185e-02 5.69437593911e-02 4.42637285519e-02 3.75326171916e-02 - 3.16057035862e-02 2.55726183949e-02 2.09446754967e-02 1.81502966987e-02 1.66389339973e-02 1.59287126341e-02 - 1.59287126341e-02 1.66389339974e-02 1.81502966987e-02 2.09446754967e-02 2.55726183949e-02 3.16057035862e-02 - 4.14583134664e-02 5.17382028008e-02 6.98255714909e-02 9.00454509227e-02 1.02255317231e-01 1.05382836884e-01 - 1.02255317231e-01 9.00454509227e-02 6.98255714909e-02 5.17382028009e-02 4.14583134664e-02 3.49769025118e-02 - 2.84217494631e-02 2.23752455712e-02 1.81179302207e-02 1.56224022211e-02 1.43188164440e-02 1.38961516945e-02 - 1.43188164440e-02 1.56224022211e-02 1.81179302207e-02 2.23752455712e-02 2.84217494631e-02 3.49769025117e-02 - 4.51578187598e-02 5.87902882083e-02 7.81193802830e-02 9.46618817553e-02 1.02255317231e-01 1.02255317231e-01 - 9.46618817554e-02 7.81193802830e-02 5.87902882083e-02 4.51578187598e-02 3.73540502297e-02 3.07822057975e-02 - 2.40967233573e-02 1.86355176327e-02 1.50739196267e-02 1.30485807277e-02 1.21045128275e-02 1.21045128275e-02 - 1.30485807277e-02 1.50739196267e-02 1.86355176327e-02 2.40967233573e-02 3.07822057975e-02 3.73540502296e-02 - 4.76134439312e-02 6.17008685349e-02 7.81193802831e-02 9.00454509227e-02 9.40538638177e-02 9.00454509227e-02 - 7.81193802831e-02 6.17008685349e-02 4.76134439312e-02 3.86937384314e-02 3.20848019959e-02 2.53973709283e-02 - 1.92840317465e-02 1.48492046219e-02 1.21090534529e-02 1.06109348738e-02 1.01198499891e-02 1.06109348738e-02 - 1.21090534529e-02 1.48492046219e-02 1.92840317465e-02 2.53973709283e-02 3.20848019959e-02 3.86937384314e-02 - 4.76134439312e-02 5.87902882084e-02 6.98255714910e-02 7.65071402186e-02 7.65071402186e-02 6.98255714910e-02 - 5.87902882084e-02 4.76134439312e-02 3.91205047681e-02 3.25042138914e-02 2.59777844449e-02 1.96802339930e-02 - 1.47279181688e-02 1.14481144732e-02 9.47290433111e-03 8.50665152157e-03 8.50665152157e-03 9.47290433111e-03 - 1.14481144732e-02 1.47279181688e-02 1.96802339930e-02 2.59777844449e-02 3.25042138914e-02 3.91205047681e-02 - 4.51578187598e-02 5.17382028009e-02 5.69437593912e-02 5.89461275929e-02 5.69437593912e-02 5.17382028009e-02 - 4.51578187598e-02 3.86937384314e-02 3.25042138914e-02 2.60947919991e-02 1.97871610956e-02 1.45986463266e-02 - 1.09990704821e-02 8.70256016587e-03 7.36052161959e-03 6.90308343843e-03 7.36052161959e-03 8.70256016587e-03 - 1.09990704821e-02 1.45986463266e-02 1.97871610956e-02 2.60947919991e-02 3.25042138914e-02 3.86937384314e-02 - 4.14583134665e-02 4.42637285519e-02 4.56987320893e-02 4.56987320893e-02 4.42637285519e-02 4.14583134665e-02 - 3.73540502297e-02 3.20848019959e-02 2.59777844449e-02 1.97871610956e-02 1.45320680305e-02 1.07546834176e-02 - 8.26409670066e-03 6.67439308889e-03 5.85076317272e-03 5.85076317272e-03 6.67439308889e-03 8.26409670065e-03 - 1.07546834176e-02 1.45320680305e-02 1.97871610956e-02 2.59777844449e-02 3.20848019959e-02 3.73540502297e-02 - 3.75326171916e-02 3.86294987685e-02 3.88811262512e-02 3.86294987685e-02 3.75326171917e-02 3.49769025118e-02 - 3.07822057975e-02 2.53973709283e-02 1.96802339930e-02 1.45986463266e-02 1.07546834176e-02 8.12161313175e-03 - 6.36714430780e-03 5.28636096575e-03 4.90588959931e-03 5.28636096575e-03 6.36714430780e-03 8.12161313175e-03 - 1.07546834176e-02 1.45986463266e-02 1.96802339930e-02 2.53973709283e-02 3.07822057975e-02 3.49769025118e-02 - 3.33647463124e-02 3.39942859272e-02 3.39942859272e-02 3.33647463124e-02 3.16057035862e-02 2.84217494631e-02 - 2.40967233573e-02 1.92840317465e-02 1.47279181689e-02 1.09990704821e-02 8.26409670066e-03 6.36714430780e-03 - 5.11303460195e-03 4.45367296569e-03 4.45367296569e-03 5.11303460195e-03 6.36714430780e-03 8.26409670065e-03 - 1.09990704821e-02 1.47279181689e-02 1.92840317465e-02 2.40967233573e-02 2.84217494631e-02 3.16057035862e-02 - 2.89473471924e-02 2.92643382604e-02 2.89473471924e-02 2.78013937268e-02 2.55726183949e-02 2.23752455712e-02 - 1.86355176327e-02 1.48492046219e-02 1.14481144732e-02 8.70256016587e-03 6.67439308889e-03 5.28636096575e-03 - 4.45367296569e-03 4.16945913194e-03 4.45367296569e-03 5.28636096575e-03 6.67439308889e-03 8.70256016587e-03 - 1.14481144732e-02 1.48492046219e-02 1.86355176327e-02 2.23752455712e-02 2.55726183949e-02 2.78013937268e-02 - 2.52445347088e-02 2.52445347088e-02 2.46286629032e-02 2.32030440050e-02 2.09446754967e-02 1.81179302207e-02 - 1.50739196267e-02 1.21090534530e-02 9.47290433111e-03 7.36052161959e-03 5.85076317272e-03 4.90588959931e-03 - 4.45367296569e-03 4.45367296569e-03 4.90588959931e-03 5.85076317272e-03 7.36052161959e-03 9.47290433111e-03 - 1.21090534529e-02 1.50739196267e-02 1.81179302207e-02 2.09446754967e-02 2.32030440049e-02 2.46286629032e-02 - 2.33118163733e-02 2.30253665657e-02 2.20777720561e-02 2.03983415996e-02 1.81502966987e-02 1.56224022211e-02 - 1.30485807277e-02 1.06109348738e-02 8.50665152157e-03 6.90308343843e-03 5.85076317272e-03 5.28636096575e-03 - 5.11303460195e-03 5.28636096575e-03 5.85076317272e-03 6.90308343843e-03 8.50665152157e-03 1.06109348738e-02 - 1.30485807277e-02 1.56224022211e-02 1.81502966987e-02 2.03983415996e-02 2.20777720561e-02 2.30253665657e-02 - 2.31387869200e-02 2.24043982374e-02 2.09446754967e-02 1.89263764236e-02 1.66389339973e-02 1.43188164440e-02 - 1.21045128275e-02 1.01198499891e-02 8.50665152157e-03 7.36052161958e-03 6.67439308888e-03 6.36714430779e-03 - 6.36714430779e-03 6.67439308888e-03 7.36052161958e-03 8.50665152156e-03 1.01198499891e-02 1.21045128275e-02 - 1.43188164440e-02 1.66389339973e-02 1.89263764236e-02 2.09446754967e-02 2.24043982374e-02 2.31387869200e-02 - 2.37301976795e-02 2.23752455712e-02 2.03920879638e-02 1.81430425204e-02 1.59287126341e-02 1.38961516944e-02 - 1.21045128275e-02 1.06109348738e-02 9.47290433110e-03 8.70256016586e-03 8.26409670065e-03 8.12161313174e-03 - 8.26409670065e-03 8.70256016586e-03 9.47290433110e-03 1.06109348738e-02 1.21045128275e-02 1.38961516944e-02 - 1.59287126341e-02 1.81430425204e-02 2.03920879638e-02 2.23752455712e-02 2.37301976795e-02 2.42075414899e-02 - 2.40967233573e-02 2.22713307355e-02 2.00583037371e-02 1.78657504093e-02 1.59287126341e-02 1.43188164440e-02 - 1.30485807277e-02 1.21090534529e-02 1.14481144732e-02 1.09990704821e-02 1.07546834176e-02 1.07546834176e-02 - 1.09990704821e-02 1.14481144732e-02 1.21090534529e-02 1.30485807277e-02 1.43188164440e-02 1.59287126341e-02 - 1.78657504093e-02 2.00583037371e-02 2.22713307355e-02 2.40967233573e-02 2.51249889113e-02 2.51249889113e-02 - 2.40904203272e-02 2.21709602700e-02 2.00583037371e-02 1.81430425204e-02 1.66389339973e-02 1.56224022211e-02 - 1.50739196267e-02 1.48492046219e-02 1.47279181688e-02 1.45986463266e-02 1.45320680305e-02 1.45986463266e-02 - 1.47279181688e-02 1.48492046219e-02 1.50739196267e-02 1.56224022211e-02 1.66389339973e-02 1.81430425204e-02 - 2.00583037371e-02 2.21709602700e-02 2.40904203272e-02 2.53973709283e-02 2.58521940319e-02 2.53973709283e-02 - 2.40904203272e-02 2.22713307355e-02 2.03920879638e-02 1.89263764236e-02 1.81502966987e-02 1.81179302207e-02 - 1.86355176327e-02 1.92840317465e-02 1.96802339929e-02 1.97871610956e-02 1.97871610956e-02 1.96802339930e-02 - 1.92840317465e-02 1.86355176327e-02 1.81179302207e-02 1.81502966987e-02 1.89263764236e-02 2.03920879638e-02 - 2.22713307355e-02 2.40904203272e-02 2.53797374387e-02 2.59777844449e-02 2.59777844449e-02 2.53797374387e-02 - 2.40967233573e-02 2.23752455712e-02 2.09446754967e-02 2.03983415996e-02 2.09446754967e-02 2.23752455712e-02 - 2.40967233573e-02 2.53973709283e-02 2.59777844449e-02 2.60947919991e-02 2.59777844449e-02 2.53973709283e-02 - 2.40967233573e-02 2.23752455712e-02 2.09446754967e-02 2.03983415996e-02 2.09446754967e-02 2.23752455712e-02 - 2.40967233573e-02 2.53973709283e-02 2.59777844449e-02 2.60947919991e-02 2.59777844449e-02 2.53973709283e-02 - 2.37301976795e-02 2.24043982374e-02 2.20777720561e-02 2.32030440049e-02 2.55726183949e-02 2.84217494631e-02 - 3.07822057975e-02 3.20848019958e-02 3.25042138914e-02 3.25042138914e-02 3.20848019958e-02 3.07822057975e-02 - 2.84217494631e-02 2.55726183949e-02 2.32030440049e-02 2.20777720561e-02 2.24043982374e-02 2.37301976795e-02 - 2.51249889113e-02 2.58521940319e-02 2.59777844449e-02 2.59777844449e-02 2.58521940319e-02 2.51249889113e-02 - 2.31387869200e-02 2.30253665656e-02 2.46286629032e-02 2.78013937268e-02 3.16057035862e-02 3.49769025117e-02 - 3.73540502296e-02 3.86937384314e-02 3.91205047681e-02 3.86937384314e-02 3.73540502296e-02 3.49769025117e-02 - 3.16057035862e-02 2.78013937268e-02 2.46286629032e-02 2.30253665656e-02 2.31387869200e-02 2.42075414898e-02 - 2.51249889113e-02 2.53973709283e-02 2.53797374387e-02 2.53973709283e-02 2.51249889113e-02 2.42075414898e-02 - 2.21386056500e-02 2.59181147281e-02 3.06728014251e-02 3.51262182184e-02 3.86293456568e-02 4.14583134664e-02 - 4.36605690195e-02 4.45490987357e-02 4.36605690195e-02 4.14583134664e-02 3.86293456568e-02 3.51262182184e-02 - 3.06728014251e-02 2.59181147281e-02 2.21386056500e-02 1.99975947255e-02 1.91106081253e-02 1.86355176327e-02 - 1.81617046076e-02 1.79266526907e-02 1.81617046076e-02 1.86355176327e-02 1.91106081253e-02 1.99975947255e-02 - 2.59181147281e-02 3.12134113093e-02 3.61935553818e-02 4.02266025417e-02 4.42637285519e-02 4.86717503238e-02 - 5.18158087397e-02 5.18158087397e-02 4.86717503238e-02 4.42637285519e-02 4.02266025417e-02 3.61935553818e-02 - 3.12134113093e-02 2.59181147281e-02 2.17948830411e-02 1.93871115228e-02 1.81179302207e-02 1.72611853896e-02 - 1.66950035376e-02 1.66950035376e-02 1.72611853896e-02 1.81179302207e-02 1.93871115228e-02 2.17948830411e-02 - 3.06728014251e-02 3.61935553818e-02 4.06883718959e-02 4.56987320892e-02 5.24371687843e-02 5.87809730549e-02 - 6.13624406409e-02 5.87809730550e-02 5.24371687843e-02 4.56987320892e-02 4.06883718959e-02 3.61935553818e-02 - 3.06728014251e-02 2.49902463304e-02 2.07196781889e-02 1.81502966987e-02 1.66276099946e-02 1.56489970468e-02 - 1.52718999852e-02 1.56489970468e-02 1.66276099946e-02 1.81502966987e-02 2.07196781889e-02 2.49902463304e-02 - 3.51262182184e-02 4.02266025417e-02 4.56987320892e-02 5.38553081705e-02 6.30064198496e-02 6.87950359263e-02 - 6.87950359263e-02 6.30064198496e-02 5.38553081705e-02 4.56987320892e-02 4.02266025417e-02 3.51262182185e-02 - 2.89590113766e-02 2.30939042849e-02 1.89263764236e-02 1.64062528191e-02 1.48932100517e-02 1.40901390285e-02 - 1.40901390285e-02 1.48932100517e-02 1.64062528191e-02 1.89263764236e-02 2.30939042849e-02 2.89590113766e-02 - 3.86293456568e-02 4.42637285519e-02 5.24371687843e-02 6.30064198496e-02 7.14595557376e-02 7.44578962493e-02 - 7.14595557376e-02 6.30064198496e-02 5.24371687843e-02 4.42637285519e-02 3.86293456568e-02 3.27029610793e-02 - 2.60492747623e-02 2.03920879638e-02 1.66276099946e-02 1.43894839458e-02 1.31234388541e-02 1.26855285494e-02 - 1.31234388541e-02 1.43894839458e-02 1.66276099946e-02 2.03920879638e-02 2.60492747623e-02 3.27029610793e-02 - 4.14583134664e-02 4.86717503238e-02 5.87809730550e-02 6.87950359263e-02 7.44578962493e-02 7.44578962493e-02 - 6.87950359263e-02 5.87809730550e-02 4.86717503238e-02 4.14583134664e-02 3.55406204185e-02 2.88778389824e-02 - 2.22713307355e-02 1.72611853896e-02 1.41149739580e-02 1.22934163580e-02 1.14031750108e-02 1.14031750108e-02 - 1.22934163580e-02 1.41149739580e-02 1.72611853896e-02 2.22713307355e-02 2.88778389824e-02 3.55406204185e-02 - 4.36605690195e-02 5.18158087397e-02 6.13624406409e-02 6.87950359263e-02 7.14595557376e-02 6.87950359263e-02 - 6.13624406409e-02 5.18158087398e-02 4.36605690195e-02 3.73540502297e-02 3.09609358381e-02 2.40904203272e-02 - 1.81617046076e-02 1.40806383265e-02 1.16095381953e-02 1.02401100733e-02 9.78219490488e-03 1.02401100733e-02 - 1.16095381953e-02 1.40806383265e-02 1.81617046076e-02 2.40904203272e-02 3.09609358381e-02 3.73540502296e-02 - 4.45490987358e-02 5.18158087398e-02 5.87809730550e-02 6.30064198497e-02 6.30064198497e-02 5.87809730550e-02 - 5.18158087398e-02 4.45490987358e-02 3.82423088608e-02 3.20848019959e-02 2.53797374387e-02 1.90345757835e-02 - 1.42308801751e-02 1.11331987338e-02 9.28357359923e-03 8.37400390520e-03 8.37400390520e-03 9.28357359923e-03 - 1.11331987338e-02 1.42308801751e-02 1.90345757835e-02 2.53797374387e-02 3.20848019959e-02 3.82423088608e-02 - 4.36605690195e-02 4.86717503239e-02 5.24371687844e-02 5.38553081705e-02 5.24371687844e-02 4.86717503239e-02 - 4.36605690195e-02 3.82423088608e-02 3.24128421387e-02 2.59777844449e-02 1.96031535282e-02 1.44210111226e-02 - 1.08699825973e-02 8.62540500906e-03 7.31933010296e-03 6.87426464911e-03 7.31933010296e-03 8.62540500906e-03 - 1.08699825973e-02 1.44210111226e-02 1.96031535282e-02 2.59777844449e-02 3.24128421387e-02 3.82423088608e-02 - 4.14583134665e-02 4.42637285519e-02 4.56987320892e-02 4.56987320892e-02 4.42637285519e-02 4.14583134665e-02 - 3.73540502297e-02 3.20848019959e-02 2.59777844449e-02 1.97871610956e-02 1.45320680305e-02 1.07546834176e-02 - 8.26409670065e-03 6.67439308889e-03 5.85076317272e-03 5.85076317272e-03 6.67439308889e-03 8.26409670065e-03 - 1.07546834176e-02 1.45320680305e-02 1.97871610956e-02 2.59777844449e-02 3.20848019959e-02 3.73540502297e-02 - 3.86293456568e-02 4.02266025417e-02 4.06883718959e-02 4.02266025417e-02 3.86293456568e-02 3.55406204185e-02 - 3.09609358382e-02 2.53797374387e-02 1.96031535283e-02 1.45320680305e-02 1.07233722050e-02 8.11874591477e-03 - 6.36996915701e-03 5.27978060753e-03 4.89318523684e-03 5.27978060753e-03 6.36996915701e-03 8.11874591477e-03 - 1.07233722050e-02 1.45320680305e-02 1.96031535283e-02 2.53797374387e-02 3.09609358381e-02 3.55406204185e-02 - 3.51262182185e-02 3.61935553818e-02 3.61935553818e-02 3.51262182185e-02 3.27029610793e-02 2.88778389824e-02 - 2.40904203272e-02 1.90345757835e-02 1.44210111226e-02 1.07546834176e-02 8.11874591477e-03 6.28719817635e-03 - 5.04614391795e-03 4.37599558137e-03 4.37599558137e-03 5.04614391795e-03 6.28719817635e-03 8.11874591477e-03 - 1.07546834176e-02 1.44210111226e-02 1.90345757835e-02 2.40904203272e-02 2.88778389824e-02 3.27029610793e-02 - 3.06728014251e-02 3.12134113093e-02 3.06728014251e-02 2.89590113767e-02 2.60492747623e-02 2.22713307355e-02 - 1.81617046076e-02 1.42308801751e-02 1.08699825973e-02 8.26409670065e-03 6.36996915701e-03 5.04614391795e-03 - 4.21584246473e-03 3.92281290681e-03 4.21584246473e-03 5.04614391795e-03 6.36996915701e-03 8.26409670065e-03 - 1.08699825973e-02 1.42308801751e-02 1.81617046076e-02 2.22713307355e-02 2.60492747623e-02 2.89590113767e-02 - 2.59181147281e-02 2.59181147281e-02 2.49902463304e-02 2.30939042850e-02 2.03920879638e-02 1.72611853896e-02 - 1.40806383265e-02 1.11331987338e-02 8.62540500906e-03 6.67439308889e-03 5.27978060753e-03 4.37599558137e-03 - 3.92281290681e-03 3.92281290681e-03 4.37599558137e-03 5.27978060753e-03 6.67439308888e-03 8.62540500905e-03 - 1.11331987338e-02 1.40806383265e-02 1.72611853896e-02 2.03920879638e-02 2.30939042850e-02 2.49902463304e-02 - 2.21386056500e-02 2.17948830411e-02 2.07196781889e-02 1.89263764236e-02 1.66276099946e-02 1.41149739580e-02 - 1.16095381953e-02 9.28357359923e-03 7.31933010297e-03 5.85076317272e-03 4.89318523683e-03 4.37599558137e-03 - 4.21584246472e-03 4.37599558137e-03 4.89318523683e-03 5.85076317272e-03 7.31933010296e-03 9.28357359923e-03 - 1.16095381953e-02 1.41149739580e-02 1.66276099946e-02 1.89263764236e-02 2.07196781889e-02 2.17948830411e-02 - 1.99975947255e-02 1.93871115228e-02 1.81502966987e-02 1.64062528191e-02 1.43894839458e-02 1.22934163580e-02 - 1.02401100733e-02 8.37400390520e-03 6.87426464911e-03 5.85076317272e-03 5.27978060753e-03 5.04614391795e-03 - 5.04614391795e-03 5.27978060753e-03 5.85076317272e-03 6.87426464911e-03 8.37400390520e-03 1.02401100733e-02 - 1.22934163580e-02 1.43894839458e-02 1.64062528191e-02 1.81502966987e-02 1.93871115228e-02 1.99975947255e-02 - 1.91106081254e-02 1.81179302207e-02 1.66276099946e-02 1.48932100517e-02 1.31234388541e-02 1.14031750108e-02 - 9.78219490488e-03 8.37400390520e-03 7.31933010296e-03 6.67439308888e-03 6.36996915701e-03 6.28719817634e-03 - 6.36996915701e-03 6.67439308888e-03 7.31933010296e-03 8.37400390520e-03 9.78219490488e-03 1.14031750108e-02 - 1.31234388541e-02 1.48932100517e-02 1.66276099946e-02 1.81179302207e-02 1.91106081253e-02 1.94536795471e-02 - 1.86355176327e-02 1.72611853896e-02 1.56489970468e-02 1.40901390285e-02 1.26855285494e-02 1.14031750108e-02 - 1.02401100733e-02 9.28357359922e-03 8.62540500905e-03 8.26409670065e-03 8.11874591476e-03 8.11874591476e-03 - 8.26409670065e-03 8.62540500905e-03 9.28357359922e-03 1.02401100733e-02 1.14031750108e-02 1.26855285494e-02 - 1.40901390285e-02 1.56489970468e-02 1.72611853896e-02 1.86355176327e-02 1.94290489291e-02 1.94290489291e-02 - 1.81617046076e-02 1.66950035376e-02 1.52718999852e-02 1.40901390285e-02 1.31234388541e-02 1.22934163580e-02 - 1.16095381953e-02 1.11331987338e-02 1.08699825973e-02 1.07546834176e-02 1.07233722050e-02 1.07546834176e-02 - 1.08699825973e-02 1.11331987338e-02 1.16095381953e-02 1.22934163580e-02 1.31234388541e-02 1.40901390285e-02 - 1.52718999852e-02 1.66950035376e-02 1.81617046076e-02 1.92840317465e-02 1.97044449599e-02 1.92840317465e-02 - 1.79266526907e-02 1.66950035376e-02 1.56489970468e-02 1.48932100517e-02 1.43894839458e-02 1.41149739580e-02 - 1.40806383265e-02 1.42308801751e-02 1.44210111226e-02 1.45320680305e-02 1.45320680305e-02 1.44210111226e-02 - 1.42308801751e-02 1.40806383265e-02 1.41149739580e-02 1.43894839458e-02 1.48932100517e-02 1.56489970468e-02 - 1.66950035376e-02 1.79266526907e-02 1.90345757835e-02 1.96802339930e-02 1.96802339930e-02 1.90345757835e-02 - 1.81617046076e-02 1.72611853896e-02 1.66276099946e-02 1.64062528191e-02 1.66276099946e-02 1.72611853896e-02 - 1.81617046076e-02 1.90345757835e-02 1.96031535282e-02 1.97871610956e-02 1.96031535282e-02 1.90345757835e-02 - 1.81617046076e-02 1.72611853896e-02 1.66276099946e-02 1.64062528191e-02 1.66276099946e-02 1.72611853896e-02 - 1.81617046076e-02 1.90345757835e-02 1.96031535282e-02 1.97871610956e-02 1.96031535282e-02 1.90345757835e-02 - 1.86355176327e-02 1.81179302207e-02 1.81502966987e-02 1.89263764236e-02 2.03920879637e-02 2.22713307355e-02 - 2.40904203272e-02 2.53797374387e-02 2.59777844449e-02 2.59777844449e-02 2.53797374387e-02 2.40904203272e-02 - 2.22713307355e-02 2.03920879638e-02 1.89263764236e-02 1.81502966987e-02 1.81179302207e-02 1.86355176327e-02 - 1.92840317465e-02 1.96802339930e-02 1.97871610956e-02 1.97871610956e-02 1.96802339930e-02 1.92840317465e-02 - 1.91106081253e-02 1.93871115228e-02 2.07196781889e-02 2.30939042849e-02 2.60492747622e-02 2.88778389824e-02 - 3.09609358381e-02 3.20848019958e-02 3.24128421387e-02 3.20848019958e-02 3.09609358381e-02 2.88778389824e-02 - 2.60492747623e-02 2.30939042849e-02 2.07196781889e-02 1.93871115228e-02 1.91106081253e-02 1.94290489291e-02 - 1.97044449599e-02 1.96802339930e-02 1.96031535282e-02 1.96802339930e-02 1.97044449599e-02 1.94290489291e-02 - 1.99975947255e-02 2.17948830411e-02 2.49902463304e-02 2.89590113766e-02 3.27029610793e-02 3.55406204184e-02 - 3.73540502296e-02 3.82423088607e-02 3.82423088607e-02 3.73540502296e-02 3.55406204185e-02 3.27029610793e-02 - 2.89590113766e-02 2.49902463304e-02 2.17948830411e-02 1.99975947255e-02 1.94536795471e-02 1.94290489291e-02 - 1.92840317465e-02 1.90345757835e-02 1.90345757835e-02 1.92840317465e-02 1.94290489291e-02 1.94536795471e-02 - 2.19611077293e-02 2.65858530209e-02 3.13656133727e-02 3.51262182184e-02 3.75326171916e-02 3.89798518705e-02 - 3.97328533270e-02 3.97328533270e-02 3.89798518705e-02 3.75326171916e-02 3.51262182184e-02 3.13656133727e-02 - 2.65858530209e-02 2.19611077293e-02 1.85132763207e-02 1.63906429721e-02 1.50739196267e-02 1.40806383265e-02 - 1.34421324248e-02 1.34421324248e-02 1.40806383265e-02 1.50739196267e-02 1.63906429721e-02 1.85132763207e-02 - 2.65858530209e-02 3.20070729422e-02 3.61935553818e-02 3.86294987685e-02 4.00842207789e-02 4.11388919901e-02 - 4.15726567626e-02 4.11388919901e-02 4.00842207789e-02 3.86294987685e-02 3.61935553818e-02 3.20070729422e-02 - 2.65858530209e-02 2.15195877577e-02 1.78888288560e-02 1.56224022211e-02 1.41149739580e-02 1.30618774824e-02 - 1.26511415702e-02 1.30618774824e-02 1.41149739580e-02 1.56224022211e-02 1.78888288560e-02 2.15195877577e-02 - 3.13656133727e-02 3.61935553818e-02 3.88811262511e-02 4.03062291682e-02 4.15214050146e-02 4.23899857519e-02 - 4.23899857519e-02 4.15214050146e-02 4.03062291682e-02 3.88811262511e-02 3.61935553818e-02 3.13656133727e-02 - 2.54054430258e-02 2.02040886880e-02 1.66389339973e-02 1.43894839458e-02 1.28937939562e-02 1.20439356208e-02 - 1.20439356208e-02 1.28937939562e-02 1.43894839458e-02 1.66389339973e-02 2.02040886880e-02 2.54054430258e-02 - 3.51262182184e-02 3.86294987685e-02 4.03062291682e-02 4.15474983448e-02 4.26148361403e-02 4.30330806505e-02 - 4.26148361403e-02 4.15474983449e-02 4.03062291682e-02 3.86294987685e-02 3.51262182184e-02 2.93979324046e-02 - 2.31231583381e-02 1.81430425204e-02 1.48932100517e-02 1.28580577518e-02 1.16029685725e-02 1.11481007476e-02 - 1.16029685725e-02 1.28580577518e-02 1.48932100517e-02 1.81430425204e-02 2.31231583381e-02 2.93979324046e-02 - 3.75326171916e-02 4.00842207789e-02 4.15214050146e-02 4.26148361403e-02 4.31931066266e-02 4.31931066266e-02 - 4.26148361403e-02 4.15214050146e-02 4.00842207789e-02 3.75326171916e-02 3.27029610793e-02 2.61902387743e-02 - 2.00583037371e-02 1.56489970468e-02 1.28937939562e-02 1.12137577955e-02 1.03422447642e-02 1.03422447642e-02 - 1.12137577955e-02 1.28937939562e-02 1.56489970468e-02 2.00583037371e-02 2.61902387743e-02 3.27029610793e-02 - 3.89798518705e-02 4.11388919901e-02 4.23899857519e-02 4.30330806506e-02 4.31931066266e-02 4.30330806506e-02 - 4.23899857520e-02 4.11388919901e-02 3.89798518706e-02 3.49769025118e-02 2.88778389824e-02 2.21709602700e-02 - 1.66950035376e-02 1.30618774824e-02 1.08500532301e-02 9.57619631350e-03 9.13724506496e-03 9.57619631350e-03 - 1.08500532301e-02 1.30618774824e-02 1.66950035376e-02 2.21709602700e-02 2.88778389824e-02 3.49769025117e-02 - 3.97328533270e-02 4.15726567626e-02 4.23899857520e-02 4.26148361403e-02 4.26148361403e-02 4.23899857520e-02 - 4.15726567626e-02 3.97328533271e-02 3.62673468746e-02 3.07822057975e-02 2.40904203272e-02 1.79266526907e-02 - 1.34421324248e-02 1.06051613742e-02 8.89920278516e-03 8.04455907974e-03 8.04455907974e-03 8.89920278516e-03 - 1.06051613742e-02 1.34421324248e-02 1.79266526907e-02 2.40904203272e-02 3.07822057975e-02 3.62673468746e-02 - 3.97328533271e-02 4.11388919902e-02 4.15214050146e-02 4.15474983449e-02 4.15214050146e-02 4.11388919902e-02 - 3.97328533271e-02 3.66854482599e-02 3.17361436403e-02 2.53973709283e-02 1.90345757835e-02 1.39667079820e-02 - 1.05557815436e-02 8.41997198039e-03 7.17862759619e-03 6.75449318420e-03 7.17862759619e-03 8.41997198039e-03 - 1.05557815436e-02 1.39667079820e-02 1.90345757835e-02 2.53973709283e-02 3.17361436402e-02 3.66854482599e-02 - 3.89798518706e-02 4.00842207790e-02 4.03062291683e-02 4.03062291683e-02 4.00842207790e-02 3.89798518706e-02 - 3.62673468746e-02 3.17361436403e-02 2.58521940319e-02 1.96802339930e-02 1.44210111226e-02 1.06518186545e-02 - 8.18383164206e-03 6.62684043802e-03 5.82746138702e-03 5.82746138702e-03 6.62684043801e-03 8.18383164206e-03 - 1.06518186545e-02 1.44210111226e-02 1.96802339930e-02 2.58521940319e-02 3.17361436403e-02 3.62673468746e-02 - 3.75326171916e-02 3.86294987685e-02 3.88811262512e-02 3.86294987685e-02 3.75326171916e-02 3.49769025118e-02 - 3.07822057975e-02 2.53973709283e-02 1.96802339930e-02 1.45986463266e-02 1.07546834176e-02 8.12161313175e-03 - 6.36714430780e-03 5.28636096575e-03 4.90588959931e-03 5.28636096575e-03 6.36714430780e-03 8.12161313175e-03 - 1.07546834176e-02 1.45986463266e-02 1.96802339930e-02 2.53973709283e-02 3.07822057975e-02 3.49769025118e-02 - 3.51262182185e-02 3.61935553818e-02 3.61935553818e-02 3.51262182185e-02 3.27029610793e-02 2.88778389824e-02 - 2.40904203272e-02 1.90345757835e-02 1.44210111226e-02 1.07546834176e-02 8.11874591477e-03 6.28719817635e-03 - 5.04614391795e-03 4.37599558137e-03 4.37599558137e-03 5.04614391795e-03 6.28719817635e-03 8.11874591477e-03 - 1.07546834176e-02 1.44210111226e-02 1.90345757835e-02 2.40904203272e-02 2.88778389824e-02 3.27029610793e-02 - 3.13656133728e-02 3.20070729423e-02 3.13656133728e-02 2.93979324046e-02 2.61902387743e-02 2.21709602700e-02 - 1.79266526907e-02 1.39667079820e-02 1.06518186545e-02 8.12161313175e-03 6.28719817635e-03 4.98513742725e-03 - 4.14918646508e-03 3.84964089400e-03 4.14918646508e-03 4.98513742725e-03 6.28719817635e-03 8.12161313174e-03 - 1.06518186545e-02 1.39667079820e-02 1.79266526907e-02 2.21709602700e-02 2.61902387743e-02 2.93979324046e-02 - 2.65858530209e-02 2.65858530209e-02 2.54054430258e-02 2.31231583381e-02 2.00583037371e-02 1.66950035376e-02 - 1.34421324248e-02 1.05557815436e-02 8.18383164206e-03 6.36714430779e-03 5.04614391795e-03 4.14918646508e-03 - 3.67768921534e-03 3.67768921534e-03 4.14918646508e-03 5.04614391795e-03 6.36714430779e-03 8.18383164206e-03 - 1.05557815436e-02 1.34421324248e-02 1.66950035376e-02 2.00583037371e-02 2.31231583381e-02 2.54054430258e-02 - 2.19611077293e-02 2.15195877577e-02 2.02040886880e-02 1.81430425205e-02 1.56489970468e-02 1.30618774824e-02 - 1.06051613742e-02 8.41997198039e-03 6.62684043802e-03 5.28636096575e-03 4.37599558137e-03 3.84964089400e-03 - 3.67768921534e-03 3.84964089400e-03 4.37599558137e-03 5.28636096575e-03 6.62684043801e-03 8.41997198039e-03 - 1.06051613742e-02 1.30618774824e-02 1.56489970468e-02 1.81430425205e-02 2.02040886880e-02 2.15195877577e-02 - 1.85132763207e-02 1.78888288560e-02 1.66389339974e-02 1.48932100517e-02 1.28937939562e-02 1.08500532301e-02 - 8.89920278516e-03 7.17862759619e-03 5.82746138702e-03 4.90588959931e-03 4.37599558137e-03 4.14918646508e-03 - 4.14918646508e-03 4.37599558137e-03 4.90588959931e-03 5.82746138702e-03 7.17862759619e-03 8.89920278516e-03 - 1.08500532301e-02 1.28937939562e-02 1.48932100517e-02 1.66389339974e-02 1.78888288560e-02 1.85132763207e-02 - 1.63906429721e-02 1.56224022211e-02 1.43894839458e-02 1.28580577518e-02 1.12137577955e-02 9.57619631350e-03 - 8.04455907974e-03 6.75449318420e-03 5.82746138702e-03 5.28636096575e-03 5.04614391795e-03 4.98513742725e-03 - 5.04614391795e-03 5.28636096575e-03 5.82746138702e-03 6.75449318420e-03 8.04455907974e-03 9.57619631350e-03 - 1.12137577955e-02 1.28580577518e-02 1.43894839458e-02 1.56224022211e-02 1.63906429721e-02 1.66439398964e-02 - 1.50739196267e-02 1.41149739580e-02 1.28937939562e-02 1.16029685725e-02 1.03422447642e-02 9.13724506496e-03 - 8.04455907974e-03 7.17862759619e-03 6.62684043801e-03 6.36714430779e-03 6.28719817634e-03 6.28719817634e-03 - 6.36714430779e-03 6.62684043801e-03 7.17862759619e-03 8.04455907974e-03 9.13724506496e-03 1.03422447642e-02 - 1.16029685725e-02 1.28937939562e-02 1.41149739580e-02 1.50739196267e-02 1.55937378378e-02 1.55937378378e-02 - 1.40806383265e-02 1.30618774824e-02 1.20439356208e-02 1.11481007476e-02 1.03422447642e-02 9.57619631350e-03 - 8.89920278515e-03 8.41997198038e-03 8.18383164206e-03 8.12161313174e-03 8.11874591476e-03 8.12161313174e-03 - 8.18383164206e-03 8.41997198038e-03 8.89920278515e-03 9.57619631350e-03 1.03422447642e-02 1.11481007476e-02 - 1.20439356208e-02 1.30618774824e-02 1.40806383265e-02 1.48492046219e-02 1.51354763629e-02 1.48492046219e-02 - 1.34421324248e-02 1.26511415702e-02 1.20439356208e-02 1.16029685725e-02 1.12137577955e-02 1.08500532301e-02 - 1.06051613742e-02 1.05557815436e-02 1.06518186545e-02 1.07546834176e-02 1.07546834175e-02 1.06518186545e-02 - 1.05557815436e-02 1.06051613742e-02 1.08500532301e-02 1.12137577955e-02 1.16029685725e-02 1.20439356208e-02 - 1.26511415702e-02 1.34421324248e-02 1.42308801751e-02 1.47279181688e-02 1.47279181688e-02 1.42308801751e-02 - 1.34421324248e-02 1.30618774824e-02 1.28937939562e-02 1.28580577518e-02 1.28937939562e-02 1.30618774824e-02 - 1.34421324248e-02 1.39667079820e-02 1.44210111226e-02 1.45986463266e-02 1.44210111226e-02 1.39667079820e-02 - 1.34421324248e-02 1.30618774824e-02 1.28937939562e-02 1.28580577518e-02 1.28937939562e-02 1.30618774824e-02 - 1.34421324248e-02 1.39667079820e-02 1.44210111226e-02 1.45986463266e-02 1.44210111226e-02 1.39667079820e-02 - 1.40806383265e-02 1.41149739580e-02 1.43894839458e-02 1.48932100517e-02 1.56489970468e-02 1.66950035376e-02 - 1.79266526907e-02 1.90345757835e-02 1.96802339929e-02 1.96802339929e-02 1.90345757835e-02 1.79266526907e-02 - 1.66950035376e-02 1.56489970468e-02 1.48932100517e-02 1.43894839458e-02 1.41149739580e-02 1.40806383265e-02 - 1.42308801751e-02 1.44210111226e-02 1.45320680305e-02 1.45320680305e-02 1.44210111226e-02 1.42308801751e-02 - 1.50739196267e-02 1.56224022211e-02 1.66389339973e-02 1.81430425204e-02 2.00583037371e-02 2.21709602700e-02 - 2.40904203272e-02 2.53973709283e-02 2.58521940319e-02 2.53973709283e-02 2.40904203272e-02 2.21709602700e-02 - 2.00583037371e-02 1.81430425204e-02 1.66389339973e-02 1.56224022211e-02 1.50739196267e-02 1.48492046219e-02 - 1.47279181688e-02 1.45986463266e-02 1.45320680305e-02 1.45986463266e-02 1.47279181688e-02 1.48492046219e-02 - 1.63906429721e-02 1.78888288560e-02 2.02040886880e-02 2.31231583381e-02 2.61902387743e-02 2.88778389824e-02 - 3.07822057975e-02 3.17361436402e-02 3.17361436402e-02 3.07822057975e-02 2.88778389824e-02 2.61902387743e-02 - 2.31231583381e-02 2.02040886880e-02 1.78888288560e-02 1.63906429721e-02 1.55937378378e-02 1.51354763629e-02 - 1.47279181688e-02 1.44210111226e-02 1.44210111226e-02 1.47279181688e-02 1.51354763629e-02 1.55937378378e-02 - 1.85132763207e-02 2.15195877577e-02 2.54054430258e-02 2.93979324045e-02 3.27029610793e-02 3.49769025117e-02 - 3.62673468745e-02 3.66854482599e-02 3.62673468745e-02 3.49769025117e-02 3.27029610793e-02 2.93979324046e-02 - 2.54054430258e-02 2.15195877577e-02 1.85132763207e-02 1.66439398964e-02 1.55937378378e-02 1.48492046219e-02 - 1.42308801751e-02 1.39667079820e-02 1.42308801751e-02 1.48492046219e-02 1.55937378378e-02 1.66439398964e-02 - 2.19947599998e-02 2.65858530209e-02 3.06728014251e-02 3.33647463123e-02 3.45751625707e-02 3.48628450776e-02 - 3.48683159917e-02 3.48628450776e-02 3.45751625707e-02 3.33647463123e-02 3.06728014251e-02 2.65858530209e-02 - 2.19947599998e-02 1.79741436761e-02 1.50424663757e-02 1.30485807277e-02 1.16095381953e-02 1.06051613742e-02 - 1.02248872526e-02 1.06051613742e-02 1.16095381953e-02 1.30485807277e-02 1.50424663757e-02 1.79741436761e-02 - 2.65858530209e-02 3.12134113093e-02 3.39942859271e-02 3.46096179590e-02 3.39196724264e-02 3.31551669201e-02 - 3.31551669202e-02 3.39196724265e-02 3.46096179590e-02 3.39942859271e-02 3.12134113093e-02 2.65858530209e-02 - 2.15038028103e-02 1.72812015630e-02 1.43188164440e-02 1.22934163580e-02 1.08500532301e-02 1.00210602336e-02 - 1.00210602336e-02 1.08500532301e-02 1.22934163580e-02 1.43188164440e-02 1.72812015630e-02 2.15038028103e-02 - 3.06728014251e-02 3.39942859271e-02 3.44216532953e-02 3.26920605346e-02 3.05273339684e-02 2.95749315132e-02 - 3.05273339684e-02 3.26920605346e-02 3.44216532954e-02 3.39942859271e-02 3.06728014251e-02 2.54054430258e-02 - 2.00653973118e-02 1.59287126341e-02 1.31234388541e-02 1.12137577955e-02 9.96234500644e-03 9.50248837487e-03 - 9.96234500644e-03 1.12137577955e-02 1.31234388541e-02 1.59287126341e-02 2.00653973118e-02 2.54054430258e-02 - 3.33647463123e-02 3.46096179590e-02 3.26920605346e-02 2.92890570902e-02 2.67216511959e-02 2.67216511959e-02 - 2.92890570902e-02 3.26920605346e-02 3.46096179590e-02 3.33647463123e-02 2.89590113766e-02 2.31231583381e-02 - 1.78657504093e-02 1.40901390285e-02 1.16029685725e-02 9.96167567605e-03 9.07031679536e-03 9.07031679536e-03 - 9.96167567605e-03 1.16029685725e-02 1.40901390285e-02 1.78657504093e-02 2.31231583381e-02 2.89590113766e-02 - 3.45751625707e-02 3.39196724265e-02 3.05273339684e-02 2.67216511959e-02 2.50661805907e-02 2.67216511959e-02 - 3.05273339684e-02 3.39196724265e-02 3.45751625707e-02 3.16057035862e-02 2.60492747623e-02 2.00583037371e-02 - 1.52718999852e-02 1.20439356208e-02 9.96234500644e-03 8.68876908730e-03 8.23785813083e-03 8.68876908730e-03 - 9.96234500643e-03 1.20439356208e-02 1.52718999852e-02 2.00583037371e-02 2.60492747623e-02 3.16057035862e-02 - 3.48628450776e-02 3.31551669202e-02 2.95749315132e-02 2.67216511960e-02 2.67216511960e-02 2.95749315132e-02 - 3.31551669202e-02 3.48628450776e-02 3.31963581644e-02 2.84217494631e-02 2.22713307355e-02 1.66950035376e-02 - 1.26511415702e-02 1.00210602336e-02 8.36360112567e-03 7.50451749096e-03 7.50451749096e-03 8.36360112567e-03 - 1.00210602336e-02 1.26511415702e-02 1.66950035376e-02 2.22713307355e-02 2.84217494631e-02 3.31963581644e-02 - 3.48683159917e-02 3.31551669202e-02 3.05273339684e-02 2.92890570903e-02 3.05273339684e-02 3.31551669202e-02 - 3.48683159917e-02 3.38925854809e-02 2.99301385615e-02 2.40967233573e-02 1.81617046076e-02 1.34421324248e-02 - 1.02248872526e-02 8.15758815908e-03 6.92803651720e-03 6.50338402317e-03 6.92803651720e-03 8.15758815907e-03 - 1.02248872526e-02 1.34421324248e-02 1.81617046076e-02 2.40967233573e-02 2.99301385615e-02 3.38925854809e-02 - 3.48628450776e-02 3.39196724265e-02 3.26920605347e-02 3.26920605347e-02 3.39196724265e-02 3.48628450776e-02 - 3.38925854809e-02 3.04404777485e-02 2.51249889113e-02 1.92840317465e-02 1.42308801751e-02 1.05557815436e-02 - 8.11230157085e-03 6.56138198057e-03 5.76667312114e-03 5.76667312114e-03 6.56138198057e-03 8.11230157084e-03 - 1.05557815436e-02 1.42308801751e-02 1.92840317465e-02 2.51249889113e-02 3.04404777485e-02 3.38925854809e-02 - 3.45751625707e-02 3.46096179590e-02 3.44216532954e-02 3.46096179590e-02 3.45751625707e-02 3.31963581644e-02 - 2.99301385615e-02 2.51249889113e-02 1.97044449599e-02 1.47279181688e-02 1.08699825973e-02 8.18383164206e-03 - 6.39967919814e-03 5.32068131556e-03 4.94632325183e-03 5.32068131556e-03 6.39967919814e-03 8.18383164206e-03 - 1.08699825973e-02 1.47279181688e-02 1.97044449599e-02 2.51249889113e-02 2.99301385615e-02 3.31963581644e-02 - 3.33647463124e-02 3.39942859271e-02 3.39942859271e-02 3.33647463124e-02 3.16057035862e-02 2.84217494631e-02 - 2.40967233573e-02 1.92840317465e-02 1.47279181688e-02 1.09990704821e-02 8.26409670065e-03 6.36714430780e-03 - 5.11303460195e-03 4.45367296569e-03 4.45367296569e-03 5.11303460195e-03 6.36714430779e-03 8.26409670065e-03 - 1.09990704821e-02 1.47279181688e-02 1.92840317465e-02 2.40967233573e-02 2.84217494631e-02 3.16057035862e-02 - 3.06728014251e-02 3.12134113093e-02 3.06728014251e-02 2.89590113766e-02 2.60492747623e-02 2.22713307355e-02 - 1.81617046076e-02 1.42308801751e-02 1.08699825973e-02 8.26409670065e-03 6.36996915701e-03 5.04614391795e-03 - 4.21584246473e-03 3.92281290681e-03 4.21584246473e-03 5.04614391795e-03 6.36996915701e-03 8.26409670065e-03 - 1.08699825973e-02 1.42308801751e-02 1.81617046076e-02 2.22713307355e-02 2.60492747623e-02 2.89590113767e-02 - 2.65858530209e-02 2.65858530209e-02 2.54054430258e-02 2.31231583381e-02 2.00583037371e-02 1.66950035376e-02 - 1.34421324248e-02 1.05557815436e-02 8.18383164206e-03 6.36714430779e-03 5.04614391795e-03 4.14918646508e-03 - 3.67768921534e-03 3.67768921534e-03 4.14918646508e-03 5.04614391795e-03 6.36714430779e-03 8.18383164206e-03 - 1.05557815436e-02 1.34421324248e-02 1.66950035376e-02 2.00583037371e-02 2.31231583381e-02 2.54054430258e-02 - 2.19947599998e-02 2.15038028103e-02 2.00653973119e-02 1.78657504093e-02 1.52718999852e-02 1.26511415702e-02 - 1.02248872526e-02 8.11230157084e-03 6.39967919814e-03 5.11303460195e-03 4.21584246472e-03 3.67768921534e-03 - 3.49726793106e-03 3.67768921534e-03 4.21584246472e-03 5.11303460195e-03 6.39967919814e-03 8.11230157084e-03 - 1.02248872526e-02 1.26511415702e-02 1.52718999852e-02 1.78657504093e-02 2.00653973119e-02 2.15038028103e-02 - 1.79741436761e-02 1.72812015630e-02 1.59287126341e-02 1.40901390285e-02 1.20439356208e-02 1.00210602336e-02 - 8.15758815908e-03 6.56138198057e-03 5.32068131556e-03 4.45367296569e-03 3.92281290681e-03 3.67768921534e-03 - 3.67768921534e-03 3.92281290681e-03 4.45367296569e-03 5.32068131556e-03 6.56138198057e-03 8.15758815907e-03 - 1.00210602336e-02 1.20439356208e-02 1.40901390285e-02 1.59287126341e-02 1.72812015630e-02 1.79741436761e-02 - 1.50424663757e-02 1.43188164440e-02 1.31234388541e-02 1.16029685725e-02 9.96234500644e-03 8.36360112567e-03 - 6.92803651720e-03 5.76667312114e-03 4.94632325183e-03 4.45367296569e-03 4.21584246472e-03 4.14918646508e-03 - 4.21584246473e-03 4.45367296569e-03 4.94632325183e-03 5.76667312114e-03 6.92803651720e-03 8.36360112567e-03 - 9.96234500644e-03 1.16029685725e-02 1.31234388541e-02 1.43188164440e-02 1.50424663757e-02 1.52762540169e-02 - 1.30485807277e-02 1.22934163580e-02 1.12137577955e-02 9.96167567605e-03 8.68876908730e-03 7.50451749096e-03 - 6.50338402317e-03 5.76667312114e-03 5.32068131556e-03 5.11303460194e-03 5.04614391795e-03 5.04614391795e-03 - 5.11303460195e-03 5.32068131556e-03 5.76667312114e-03 6.50338402317e-03 7.50451749096e-03 8.68876908730e-03 - 9.96167567605e-03 1.12137577955e-02 1.22934163580e-02 1.30485807277e-02 1.34169388324e-02 1.34169388324e-02 - 1.16095381953e-02 1.08500532301e-02 9.96234500644e-03 9.07031679536e-03 8.23785813083e-03 7.50451749096e-03 - 6.92803651719e-03 6.56138198056e-03 6.39967919814e-03 6.36714430779e-03 6.36996915701e-03 6.36714430779e-03 - 6.39967919814e-03 6.56138198056e-03 6.92803651719e-03 7.50451749096e-03 8.23785813083e-03 9.07031679536e-03 - 9.96234500644e-03 1.08500532301e-02 1.16095381953e-02 1.21090534529e-02 1.22800512627e-02 1.21090534529e-02 - 1.06051613742e-02 1.00210602336e-02 9.50248837488e-03 9.07031679536e-03 8.68876908730e-03 8.36360112567e-03 - 8.15758815907e-03 8.11230157084e-03 8.18383164206e-03 8.26409670065e-03 8.26409670065e-03 8.18383164206e-03 - 8.11230157084e-03 8.15758815907e-03 8.36360112566e-03 8.68876908730e-03 9.07031679536e-03 9.50248837488e-03 - 1.00210602336e-02 1.06051613742e-02 1.11331987338e-02 1.14481144732e-02 1.14481144732e-02 1.11331987338e-02 - 1.02248872526e-02 1.00210602336e-02 9.96234500644e-03 9.96167567605e-03 9.96234500643e-03 1.00210602336e-02 - 1.02248872526e-02 1.05557815436e-02 1.08699825973e-02 1.09990704821e-02 1.08699825973e-02 1.05557815436e-02 - 1.02248872526e-02 1.00210602336e-02 9.96234500643e-03 9.96167567604e-03 9.96234500644e-03 1.00210602336e-02 - 1.02248872526e-02 1.05557815436e-02 1.08699825973e-02 1.09990704821e-02 1.08699825973e-02 1.05557815436e-02 - 1.06051613742e-02 1.08500532301e-02 1.12137577955e-02 1.16029685725e-02 1.20439356208e-02 1.26511415702e-02 - 1.34421324248e-02 1.42308801751e-02 1.47279181688e-02 1.47279181688e-02 1.42308801751e-02 1.34421324248e-02 - 1.26511415702e-02 1.20439356208e-02 1.16029685725e-02 1.12137577955e-02 1.08500532301e-02 1.06051613742e-02 - 1.05557815436e-02 1.06518186545e-02 1.07546834176e-02 1.07546834176e-02 1.06518186545e-02 1.05557815436e-02 - 1.16095381953e-02 1.22934163580e-02 1.31234388541e-02 1.40901390285e-02 1.52718999852e-02 1.66950035376e-02 - 1.81617046076e-02 1.92840317465e-02 1.97044449599e-02 1.92840317465e-02 1.81617046076e-02 1.66950035376e-02 - 1.52718999852e-02 1.40901390285e-02 1.31234388541e-02 1.22934163580e-02 1.16095381953e-02 1.11331987338e-02 - 1.08699825973e-02 1.07546834176e-02 1.07233722050e-02 1.07546834176e-02 1.08699825973e-02 1.11331987338e-02 - 1.30485807277e-02 1.43188164440e-02 1.59287126341e-02 1.78657504093e-02 2.00583037371e-02 2.22713307355e-02 - 2.40967233573e-02 2.51249889113e-02 2.51249889113e-02 2.40967233573e-02 2.22713307355e-02 2.00583037371e-02 - 1.78657504093e-02 1.59287126341e-02 1.43188164440e-02 1.30485807277e-02 1.21090534529e-02 1.14481144732e-02 - 1.09990704821e-02 1.07546834176e-02 1.07546834176e-02 1.09990704821e-02 1.14481144732e-02 1.21090534529e-02 - 1.50424663757e-02 1.72812015630e-02 2.00653973118e-02 2.31231583381e-02 2.60492747622e-02 2.84217494631e-02 - 2.99301385615e-02 3.04404777485e-02 2.99301385615e-02 2.84217494631e-02 2.60492747622e-02 2.31231583381e-02 - 2.00653973118e-02 1.72812015630e-02 1.50424663757e-02 1.34169388324e-02 1.22800512627e-02 1.14481144732e-02 - 1.08699825973e-02 1.06518186545e-02 1.08699825973e-02 1.14481144732e-02 1.22800512627e-02 1.34169388324e-02 - 1.79741436761e-02 2.15038028103e-02 2.54054430258e-02 2.89590113766e-02 3.16057035862e-02 3.31963581643e-02 - 3.38925854809e-02 3.38925854809e-02 3.31963581644e-02 3.16057035862e-02 2.89590113766e-02 2.54054430258e-02 - 2.15038028103e-02 1.79741436761e-02 1.52762540169e-02 1.34169388324e-02 1.21090534529e-02 1.11331987338e-02 - 1.05557815436e-02 1.05557815436e-02 1.11331987338e-02 1.21090534529e-02 1.34169388324e-02 1.52762540169e-02 - 2.19611077293e-02 2.59181147281e-02 2.89473471924e-02 3.04773481039e-02 3.06792139923e-02 3.03789244340e-02 - 3.03789244340e-02 3.06792139923e-02 3.04773481040e-02 2.89473471924e-02 2.59181147281e-02 2.19611077293e-02 - 1.79741436761e-02 1.46288647205e-02 1.21045128275e-02 1.02401100733e-02 8.89920278516e-03 8.15758815908e-03 - 8.15758815907e-03 8.89920278515e-03 1.02401100733e-02 1.21045128275e-02 1.46288647205e-02 1.79741436761e-02 - 2.59181147281e-02 2.92643382604e-02 3.05548189106e-02 2.97974895143e-02 2.81921950446e-02 2.73937018060e-02 - 2.81921950446e-02 2.97974895143e-02 3.05548189106e-02 2.92643382604e-02 2.59181147281e-02 2.15195877577e-02 - 1.72812015630e-02 1.38961516944e-02 1.14031750108e-02 9.57619631350e-03 8.36360112567e-03 7.92498387888e-03 - 8.36360112567e-03 9.57619631350e-03 1.14031750108e-02 1.38961516944e-02 1.72812015630e-02 2.15195877577e-02 - 2.89473471924e-02 3.05548189106e-02 2.92436594720e-02 2.61164989591e-02 2.35389825848e-02 2.35389825848e-02 - 2.61164989591e-02 2.92436594720e-02 3.05548189106e-02 2.89473471924e-02 2.49902463304e-02 2.02040886880e-02 - 1.59287126341e-02 1.26855285494e-02 1.03422447642e-02 8.68876908730e-03 7.77912772161e-03 7.77912772161e-03 - 8.68876908730e-03 1.03422447642e-02 1.26855285494e-02 1.59287126341e-02 2.02040886880e-02 2.49902463304e-02 - 3.04773481040e-02 2.97974895143e-02 2.61164989591e-02 2.17746576295e-02 1.98434779695e-02 2.17746576295e-02 - 2.61164989591e-02 2.97974895143e-02 3.04773481040e-02 2.78013937268e-02 2.30939042849e-02 1.81430425204e-02 - 1.40901390285e-02 1.11481007476e-02 9.07031679536e-03 7.73176989509e-03 7.25301575416e-03 7.73176989509e-03 - 9.07031679536e-03 1.11481007476e-02 1.40901390285e-02 1.81430425204e-02 2.30939042849e-02 2.78013937268e-02 - 3.06792139923e-02 2.81921950446e-02 2.35389825848e-02 1.98434779695e-02 1.98434779695e-02 2.35389825848e-02 - 2.81921950446e-02 3.06792139923e-02 2.95890354456e-02 2.55726183949e-02 2.03920879638e-02 1.56489970468e-02 - 1.20439356208e-02 9.50248837488e-03 7.77912772161e-03 6.85668012358e-03 6.85668012358e-03 7.77912772161e-03 - 9.50248837487e-03 1.20439356208e-02 1.56489970468e-02 2.03920879638e-02 2.55726183949e-02 2.95890354456e-02 - 3.03789244340e-02 2.73937018060e-02 2.35389825848e-02 2.17746576295e-02 2.35389825848e-02 2.73937018060e-02 - 3.03789244340e-02 3.04051143293e-02 2.73091501295e-02 2.23752455712e-02 1.72611853896e-02 1.30618774824e-02 - 1.00210602336e-02 7.92498387888e-03 6.62087909721e-03 6.16385997204e-03 6.62087909720e-03 7.92498387887e-03 - 1.00210602336e-02 1.30618774824e-02 1.72611853896e-02 2.23752455712e-02 2.73091501295e-02 3.04051143293e-02 - 3.03789244340e-02 2.81921950446e-02 2.61164989591e-02 2.61164989591e-02 2.81921950446e-02 3.03789244340e-02 - 3.06058831029e-02 2.81653274579e-02 2.37301976795e-02 1.86355176327e-02 1.40806383265e-02 1.06051613742e-02 - 8.15758815908e-03 6.53729803367e-03 5.69411798450e-03 5.69411798450e-03 6.53729803366e-03 8.15758815908e-03 - 1.06051613742e-02 1.40806383265e-02 1.86355176327e-02 2.37301976795e-02 2.81653274579e-02 3.06058831029e-02 - 3.06792139923e-02 2.97974895143e-02 2.92436594720e-02 2.97974895143e-02 3.06792139923e-02 3.04051143293e-02 - 2.81653274579e-02 2.42075414898e-02 1.94290489291e-02 1.48492046219e-02 1.11331987338e-02 8.41997198039e-03 - 6.56138198057e-03 5.43239778449e-03 5.04343778102e-03 5.43239778449e-03 6.56138198057e-03 8.41997198039e-03 - 1.11331987338e-02 1.48492046219e-02 1.94290489291e-02 2.42075414898e-02 2.81653274579e-02 3.04051143293e-02 - 3.04773481040e-02 3.05548189106e-02 3.05548189106e-02 3.04773481040e-02 2.95890354456e-02 2.73091501295e-02 - 2.37301976795e-02 1.94290489291e-02 1.51354763629e-02 1.14481144732e-02 8.62540500906e-03 6.62684043802e-03 - 5.32068131556e-03 4.65175947716e-03 4.65175947716e-03 5.32068131556e-03 6.62684043802e-03 8.62540500906e-03 - 1.14481144732e-02 1.51354763629e-02 1.94290489291e-02 2.37301976795e-02 2.73091501295e-02 2.95890354456e-02 - 2.89473471924e-02 2.92643382604e-02 2.89473471924e-02 2.78013937268e-02 2.55726183949e-02 2.23752455712e-02 - 1.86355176327e-02 1.48492046219e-02 1.14481144732e-02 8.70256016587e-03 6.67439308889e-03 5.28636096575e-03 - 4.45367296569e-03 4.16945913194e-03 4.45367296569e-03 5.28636096575e-03 6.67439308889e-03 8.70256016587e-03 - 1.14481144732e-02 1.48492046219e-02 1.86355176327e-02 2.23752455712e-02 2.55726183949e-02 2.78013937268e-02 - 2.59181147281e-02 2.59181147281e-02 2.49902463304e-02 2.30939042850e-02 2.03920879638e-02 1.72611853896e-02 - 1.40806383265e-02 1.11331987338e-02 8.62540500906e-03 6.67439308888e-03 5.27978060753e-03 4.37599558137e-03 - 3.92281290681e-03 3.92281290681e-03 4.37599558137e-03 5.27978060753e-03 6.67439308888e-03 8.62540500905e-03 - 1.11331987338e-02 1.40806383265e-02 1.72611853896e-02 2.03920879638e-02 2.30939042850e-02 2.49902463304e-02 - 2.19611077293e-02 2.15195877577e-02 2.02040886880e-02 1.81430425204e-02 1.56489970468e-02 1.30618774824e-02 - 1.06051613742e-02 8.41997198039e-03 6.62684043801e-03 5.28636096575e-03 4.37599558137e-03 3.84964089400e-03 - 3.67768921534e-03 3.84964089400e-03 4.37599558137e-03 5.28636096575e-03 6.62684043801e-03 8.41997198039e-03 - 1.06051613742e-02 1.30618774824e-02 1.56489970468e-02 1.81430425205e-02 2.02040886880e-02 2.15195877577e-02 - 1.79741436761e-02 1.72812015630e-02 1.59287126341e-02 1.40901390285e-02 1.20439356208e-02 1.00210602336e-02 - 8.15758815908e-03 6.56138198057e-03 5.32068131556e-03 4.45367296569e-03 3.92281290681e-03 3.67768921534e-03 - 3.67768921534e-03 3.92281290681e-03 4.45367296569e-03 5.32068131556e-03 6.56138198057e-03 8.15758815907e-03 - 1.00210602336e-02 1.20439356208e-02 1.40901390285e-02 1.59287126341e-02 1.72812015630e-02 1.79741436761e-02 - 1.46288647205e-02 1.38961516945e-02 1.26855285494e-02 1.11481007476e-02 9.50248837487e-03 7.92498387888e-03 - 6.53729803366e-03 5.43239778449e-03 4.65175947716e-03 4.16945913194e-03 3.92281290681e-03 3.84964089400e-03 - 3.92281290681e-03 4.16945913194e-03 4.65175947716e-03 5.43239778449e-03 6.53729803366e-03 7.92498387888e-03 - 9.50248837487e-03 1.11481007476e-02 1.26855285494e-02 1.38961516945e-02 1.46288647205e-02 1.48656767561e-02 - 1.21045128275e-02 1.14031750108e-02 1.03422447642e-02 9.07031679536e-03 7.77912772161e-03 6.62087909721e-03 - 5.69411798450e-03 5.04343778102e-03 4.65175947716e-03 4.45367296569e-03 4.37599558137e-03 4.37599558137e-03 - 4.45367296569e-03 4.65175947716e-03 5.04343778102e-03 5.69411798450e-03 6.62087909721e-03 7.77912772161e-03 - 9.07031679536e-03 1.03422447642e-02 1.14031750108e-02 1.21045128275e-02 1.24265880360e-02 1.24265880360e-02 - 1.02401100733e-02 9.57619631350e-03 8.68876908730e-03 7.73176989509e-03 6.85668012358e-03 6.16385997204e-03 - 5.69411798450e-03 5.43239778448e-03 5.32068131556e-03 5.28636096575e-03 5.27978060753e-03 5.28636096575e-03 - 5.32068131556e-03 5.43239778449e-03 5.69411798450e-03 6.16385997204e-03 6.85668012358e-03 7.73176989509e-03 - 8.68876908730e-03 9.57619631350e-03 1.02401100733e-02 1.06109348738e-02 1.07225418333e-02 1.06109348738e-02 - 8.89920278516e-03 8.36360112567e-03 7.77912772161e-03 7.25301575416e-03 6.85668012358e-03 6.62087909720e-03 - 6.53729803366e-03 6.56138198056e-03 6.62684043801e-03 6.67439308888e-03 6.67439308888e-03 6.62684043801e-03 - 6.56138198056e-03 6.53729803366e-03 6.62087909720e-03 6.85668012358e-03 7.25301575416e-03 7.77912772161e-03 - 8.36360112567e-03 8.89920278516e-03 9.28357359923e-03 9.47290433111e-03 9.47290433111e-03 9.28357359923e-03 - 8.15758815908e-03 7.92498387888e-03 7.77912772161e-03 7.73176989509e-03 7.77912772161e-03 7.92498387887e-03 - 8.15758815907e-03 8.41997198038e-03 8.62540500905e-03 8.70256016586e-03 8.62540500905e-03 8.41997198038e-03 - 8.15758815907e-03 7.92498387887e-03 7.77912772161e-03 7.73176989509e-03 7.77912772161e-03 7.92498387888e-03 - 8.15758815908e-03 8.41997198039e-03 8.62540500905e-03 8.70256016587e-03 8.62540500906e-03 8.41997198039e-03 - 8.15758815908e-03 8.36360112567e-03 8.68876908730e-03 9.07031679536e-03 9.50248837487e-03 1.00210602336e-02 - 1.06051613742e-02 1.11331987338e-02 1.14481144732e-02 1.14481144732e-02 1.11331987338e-02 1.06051613742e-02 - 1.00210602336e-02 9.50248837487e-03 9.07031679536e-03 8.68876908730e-03 8.36360112567e-03 8.15758815908e-03 - 8.11230157085e-03 8.18383164206e-03 8.26409670065e-03 8.26409670065e-03 8.18383164206e-03 8.11230157085e-03 - 8.89920278516e-03 9.57619631350e-03 1.03422447642e-02 1.11481007476e-02 1.20439356208e-02 1.30618774824e-02 - 1.40806383265e-02 1.48492046219e-02 1.51354763629e-02 1.48492046219e-02 1.40806383265e-02 1.30618774824e-02 - 1.20439356208e-02 1.11481007476e-02 1.03422447642e-02 9.57619631350e-03 8.89920278516e-03 8.41997198039e-03 - 8.18383164206e-03 8.12161313175e-03 8.11874591477e-03 8.12161313174e-03 8.18383164206e-03 8.41997198039e-03 - 1.02401100733e-02 1.14031750108e-02 1.26855285494e-02 1.40901390285e-02 1.56489970468e-02 1.72611853896e-02 - 1.86355176327e-02 1.94290489291e-02 1.94290489291e-02 1.86355176327e-02 1.72611853896e-02 1.56489970468e-02 - 1.40901390285e-02 1.26855285494e-02 1.14031750108e-02 1.02401100733e-02 9.28357359923e-03 8.62540500906e-03 - 8.26409670065e-03 8.11874591476e-03 8.11874591476e-03 8.26409670065e-03 8.62540500906e-03 9.28357359923e-03 - 1.21045128275e-02 1.38961516944e-02 1.59287126341e-02 1.81430425204e-02 2.03920879638e-02 2.23752455712e-02 - 2.37301976795e-02 2.42075414898e-02 2.37301976795e-02 2.23752455712e-02 2.03920879638e-02 1.81430425204e-02 - 1.59287126341e-02 1.38961516944e-02 1.21045128275e-02 1.06109348738e-02 9.47290433111e-03 8.70256016587e-03 - 8.26409670065e-03 8.12161313174e-03 8.26409670065e-03 8.70256016586e-03 9.47290433111e-03 1.06109348738e-02 - 1.46288647205e-02 1.72812015630e-02 2.02040886880e-02 2.30939042849e-02 2.55726183949e-02 2.73091501295e-02 - 2.81653274579e-02 2.81653274579e-02 2.73091501295e-02 2.55726183949e-02 2.30939042849e-02 2.02040886880e-02 - 1.72812015630e-02 1.46288647205e-02 1.24265880360e-02 1.07225418333e-02 9.47290433111e-03 8.62540500905e-03 - 8.18383164206e-03 8.18383164206e-03 8.62540500905e-03 9.47290433111e-03 1.07225418333e-02 1.24265880360e-02 - 1.79741436761e-02 2.15195877577e-02 2.49902463304e-02 2.78013937268e-02 2.95890354456e-02 3.04051143293e-02 - 3.06058831029e-02 3.04051143293e-02 2.95890354456e-02 2.78013937268e-02 2.49902463304e-02 2.15195877577e-02 - 1.79741436761e-02 1.48656767560e-02 1.24265880360e-02 1.06109348738e-02 9.28357359923e-03 8.41997198039e-03 - 8.11230157084e-03 8.41997198039e-03 9.28357359923e-03 1.06109348738e-02 1.24265880360e-02 1.48656767560e-02 - 2.21386056500e-02 2.52445347087e-02 2.71142998602e-02 2.74769895380e-02 2.69081613820e-02 2.65224863971e-02 - 2.69081613820e-02 2.74769895380e-02 2.71142998602e-02 2.52445347087e-02 2.21386056500e-02 1.85132763207e-02 - 1.50424663757e-02 1.21045128275e-02 9.78219490488e-03 8.04455907974e-03 6.92803651720e-03 6.53729803366e-03 - 6.92803651720e-03 8.04455907974e-03 9.78219490488e-03 1.21045128275e-02 1.50424663757e-02 1.85132763207e-02 - 2.52445347087e-02 2.72051400328e-02 2.70513354759e-02 2.52955690008e-02 2.36141349499e-02 2.36141349499e-02 - 2.52955690008e-02 2.70513354759e-02 2.72051400328e-02 2.52445347087e-02 2.17948830411e-02 1.78888288560e-02 - 1.43188164440e-02 1.14031750108e-02 9.13724506496e-03 7.50451749096e-03 6.62087909721e-03 6.62087909721e-03 - 7.50451749096e-03 9.13724506495e-03 1.14031750108e-02 1.43188164440e-02 1.78888288560e-02 2.17948830411e-02 - 2.71142998602e-02 2.70513354759e-02 2.44579386769e-02 2.10846196163e-02 1.95329704288e-02 2.10846196163e-02 - 2.44579386769e-02 2.70513354759e-02 2.71142998602e-02 2.46286629032e-02 2.07196781889e-02 1.66389339973e-02 - 1.31234388541e-02 1.03422447642e-02 8.23785813083e-03 6.85668012358e-03 6.36555591034e-03 6.85668012358e-03 - 8.23785813083e-03 1.03422447642e-02 1.31234388541e-02 1.66389339973e-02 2.07196781889e-02 2.46286629032e-02 - 2.74769895380e-02 2.52955690008e-02 2.10846196163e-02 1.76763940774e-02 1.76763940774e-02 2.10846196163e-02 - 2.52955690008e-02 2.74769895380e-02 2.65400648390e-02 2.32030440049e-02 1.89263764236e-02 1.48932100517e-02 - 1.16029685725e-02 9.07031679536e-03 7.25301575416e-03 6.26631005461e-03 6.26631005461e-03 7.25301575415e-03 - 9.07031679536e-03 1.16029685725e-02 1.48932100517e-02 1.89263764236e-02 2.32030440049e-02 2.65400648390e-02 - 2.69081613820e-02 2.36141349499e-02 1.95329704288e-02 1.76763940774e-02 1.95329704288e-02 2.36141349499e-02 - 2.69081613820e-02 2.73903662614e-02 2.49964037842e-02 2.09446754967e-02 1.66276099946e-02 1.28937939562e-02 - 9.96234500644e-03 7.77912772161e-03 6.36555591034e-03 5.86550216407e-03 6.36555591033e-03 7.77912772161e-03 - 9.96234500644e-03 1.28937939562e-02 1.66276099946e-02 2.09446754967e-02 2.49964037842e-02 2.73903662614e-02 - 2.65224863971e-02 2.36141349499e-02 2.10846196163e-02 2.10846196163e-02 2.36141349499e-02 2.65224863971e-02 - 2.75584598461e-02 2.59618688149e-02 2.24043982374e-02 1.81179302207e-02 1.41149739580e-02 1.08500532301e-02 - 8.36360112567e-03 6.62087909721e-03 5.69328932590e-03 5.69328932590e-03 6.62087909721e-03 8.36360112567e-03 - 1.08500532301e-02 1.41149739580e-02 1.81179302207e-02 2.24043982374e-02 2.59618688149e-02 2.75584598461e-02 - 2.69081613820e-02 2.52955690008e-02 2.44579386769e-02 2.52955690008e-02 2.69081613820e-02 2.75584598461e-02 - 2.62414819209e-02 2.31387869200e-02 1.91106081253e-02 1.50739196267e-02 1.16095381953e-02 8.89920278516e-03 - 6.92803651720e-03 5.69411798450e-03 5.26652136817e-03 5.69411798450e-03 6.92803651720e-03 8.89920278516e-03 - 1.16095381953e-02 1.50739196267e-02 1.91106081253e-02 2.31387869200e-02 2.62414819209e-02 2.75584598461e-02 - 2.74769895380e-02 2.70513354759e-02 2.70513354759e-02 2.74769895380e-02 2.73903662614e-02 2.59618688149e-02 - 2.31387869200e-02 1.94536795471e-02 1.55937378378e-02 1.21090534529e-02 9.28357359923e-03 7.17862759619e-03 - 5.76667312114e-03 5.04343778102e-03 5.04343778102e-03 5.76667312114e-03 7.17862759619e-03 9.28357359923e-03 - 1.21090534529e-02 1.55937378378e-02 1.94536795471e-02 2.31387869200e-02 2.59618688149e-02 2.73903662614e-02 - 2.71142998602e-02 2.72051400328e-02 2.71142998602e-02 2.65400648390e-02 2.49964037842e-02 2.24043982374e-02 - 1.91106081253e-02 1.55937378378e-02 1.22800512627e-02 9.47290433111e-03 7.31933010296e-03 5.82746138702e-03 - 4.94632325183e-03 4.65175947716e-03 4.94632325183e-03 5.82746138702e-03 7.31933010296e-03 9.47290433111e-03 - 1.22800512627e-02 1.55937378378e-02 1.91106081253e-02 2.24043982374e-02 2.49964037842e-02 2.65400648390e-02 - 2.52445347088e-02 2.52445347088e-02 2.46286629032e-02 2.32030440049e-02 2.09446754967e-02 1.81179302207e-02 - 1.50739196267e-02 1.21090534529e-02 9.47290433111e-03 7.36052161959e-03 5.85076317272e-03 4.90588959931e-03 - 4.45367296569e-03 4.45367296569e-03 4.90588959931e-03 5.85076317272e-03 7.36052161959e-03 9.47290433111e-03 - 1.21090534529e-02 1.50739196267e-02 1.81179302207e-02 2.09446754967e-02 2.32030440049e-02 2.46286629032e-02 - 2.21386056500e-02 2.17948830411e-02 2.07196781889e-02 1.89263764236e-02 1.66276099946e-02 1.41149739580e-02 - 1.16095381953e-02 9.28357359923e-03 7.31933010296e-03 5.85076317272e-03 4.89318523683e-03 4.37599558137e-03 - 4.21584246472e-03 4.37599558137e-03 4.89318523683e-03 5.85076317272e-03 7.31933010296e-03 9.28357359923e-03 - 1.16095381953e-02 1.41149739580e-02 1.66276099946e-02 1.89263764236e-02 2.07196781889e-02 2.17948830411e-02 - 1.85132763207e-02 1.78888288560e-02 1.66389339973e-02 1.48932100517e-02 1.28937939562e-02 1.08500532301e-02 - 8.89920278516e-03 7.17862759619e-03 5.82746138702e-03 4.90588959931e-03 4.37599558137e-03 4.14918646508e-03 - 4.14918646508e-03 4.37599558137e-03 4.90588959931e-03 5.82746138702e-03 7.17862759619e-03 8.89920278516e-03 - 1.08500532301e-02 1.28937939562e-02 1.48932100517e-02 1.66389339973e-02 1.78888288560e-02 1.85132763207e-02 - 1.50424663757e-02 1.43188164440e-02 1.31234388541e-02 1.16029685725e-02 9.96234500644e-03 8.36360112567e-03 - 6.92803651720e-03 5.76667312114e-03 4.94632325183e-03 4.45367296569e-03 4.21584246472e-03 4.14918646508e-03 - 4.21584246473e-03 4.45367296569e-03 4.94632325183e-03 5.76667312114e-03 6.92803651720e-03 8.36360112567e-03 - 9.96234500643e-03 1.16029685725e-02 1.31234388541e-02 1.43188164440e-02 1.50424663757e-02 1.52762540169e-02 - 1.21045128275e-02 1.14031750108e-02 1.03422447642e-02 9.07031679536e-03 7.77912772161e-03 6.62087909721e-03 - 5.69411798450e-03 5.04343778102e-03 4.65175947716e-03 4.45367296569e-03 4.37599558137e-03 4.37599558137e-03 - 4.45367296569e-03 4.65175947716e-03 5.04343778102e-03 5.69411798450e-03 6.62087909721e-03 7.77912772161e-03 - 9.07031679536e-03 1.03422447642e-02 1.14031750108e-02 1.21045128275e-02 1.24265880360e-02 1.24265880360e-02 - 9.78219490488e-03 9.13724506495e-03 8.23785813083e-03 7.25301575416e-03 6.36555591034e-03 5.69328932590e-03 - 5.26652136817e-03 5.04343778102e-03 4.94632325183e-03 4.90588959931e-03 4.89318523683e-03 4.90588959931e-03 - 4.94632325183e-03 5.04343778102e-03 5.26652136817e-03 5.69328932590e-03 6.36555591034e-03 7.25301575416e-03 - 8.23785813083e-03 9.13724506495e-03 9.78219490488e-03 1.01198499891e-02 1.02153626600e-02 1.01198499891e-02 - 8.04455907974e-03 7.50451749096e-03 6.85668012358e-03 6.26631005461e-03 5.86550216407e-03 5.69328932590e-03 - 5.69411798450e-03 5.76667312114e-03 5.82746138702e-03 5.85076317272e-03 5.85076317272e-03 5.82746138702e-03 - 5.76667312114e-03 5.69411798450e-03 5.69328932590e-03 5.86550216407e-03 6.26631005461e-03 6.85668012358e-03 - 7.50451749096e-03 8.04455907974e-03 8.37400390520e-03 8.50665152156e-03 8.50665152157e-03 8.37400390520e-03 - 6.92803651720e-03 6.62087909721e-03 6.36555591034e-03 6.26631005461e-03 6.36555591033e-03 6.62087909720e-03 - 6.92803651719e-03 7.17862759619e-03 7.31933010296e-03 7.36052161958e-03 7.31933010296e-03 7.17862759619e-03 - 6.92803651719e-03 6.62087909720e-03 6.36555591033e-03 6.26631005461e-03 6.36555591034e-03 6.62087909721e-03 - 6.92803651720e-03 7.17862759619e-03 7.31933010296e-03 7.36052161959e-03 7.31933010296e-03 7.17862759619e-03 - 6.53729803367e-03 6.62087909721e-03 6.85668012358e-03 7.25301575416e-03 7.77912772161e-03 8.36360112566e-03 - 8.89920278515e-03 9.28357359922e-03 9.47290433110e-03 9.47290433110e-03 9.28357359922e-03 8.89920278515e-03 - 8.36360112566e-03 7.77912772161e-03 7.25301575415e-03 6.85668012358e-03 6.62087909721e-03 6.53729803366e-03 - 6.56138198057e-03 6.62684043801e-03 6.67439308888e-03 6.67439308889e-03 6.62684043802e-03 6.56138198057e-03 - 6.92803651720e-03 7.50451749096e-03 8.23785813083e-03 9.07031679536e-03 9.96234500643e-03 1.08500532301e-02 - 1.16095381953e-02 1.21090534529e-02 1.22800512627e-02 1.21090534529e-02 1.16095381953e-02 1.08500532301e-02 - 9.96234500643e-03 9.07031679536e-03 8.23785813083e-03 7.50451749096e-03 6.92803651720e-03 6.56138198057e-03 - 6.39967919814e-03 6.36714430779e-03 6.36996915701e-03 6.36714430779e-03 6.39967919814e-03 6.56138198057e-03 - 8.04455907974e-03 9.13724506496e-03 1.03422447642e-02 1.16029685725e-02 1.28937939562e-02 1.41149739580e-02 - 1.50739196267e-02 1.55937378378e-02 1.55937378378e-02 1.50739196267e-02 1.41149739580e-02 1.28937939562e-02 - 1.16029685725e-02 1.03422447642e-02 9.13724506496e-03 8.04455907974e-03 7.17862759619e-03 6.62684043801e-03 - 6.36714430779e-03 6.28719817635e-03 6.28719817635e-03 6.36714430779e-03 6.62684043802e-03 7.17862759619e-03 - 9.78219490488e-03 1.14031750108e-02 1.31234388541e-02 1.48932100517e-02 1.66276099946e-02 1.81179302207e-02 - 1.91106081253e-02 1.94536795471e-02 1.91106081253e-02 1.81179302207e-02 1.66276099946e-02 1.48932100517e-02 - 1.31234388541e-02 1.14031750108e-02 9.78219490488e-03 8.37400390520e-03 7.31933010296e-03 6.67439308888e-03 - 6.36996915701e-03 6.28719817635e-03 6.36996915701e-03 6.67439308888e-03 7.31933010296e-03 8.37400390520e-03 - 1.21045128275e-02 1.43188164440e-02 1.66389339973e-02 1.89263764236e-02 2.09446754967e-02 2.24043982374e-02 - 2.31387869200e-02 2.31387869200e-02 2.24043982374e-02 2.09446754967e-02 1.89263764236e-02 1.66389339973e-02 - 1.43188164440e-02 1.21045128275e-02 1.01198499891e-02 8.50665152157e-03 7.36052161959e-03 6.67439308888e-03 - 6.36714430779e-03 6.36714430779e-03 6.67439308888e-03 7.36052161959e-03 8.50665152157e-03 1.01198499891e-02 - 1.50424663757e-02 1.78888288560e-02 2.07196781889e-02 2.32030440049e-02 2.49964037842e-02 2.59618688149e-02 - 2.62414819209e-02 2.59618688149e-02 2.49964037842e-02 2.32030440049e-02 2.07196781889e-02 1.78888288560e-02 - 1.50424663757e-02 1.24265880360e-02 1.02153626600e-02 8.50665152157e-03 7.31933010296e-03 6.62684043801e-03 - 6.39967919814e-03 6.62684043801e-03 7.31933010296e-03 8.50665152157e-03 1.02153626600e-02 1.24265880360e-02 - 1.85132763207e-02 2.17948830411e-02 2.46286629032e-02 2.65400648390e-02 2.73903662614e-02 2.75584598461e-02 - 2.75584598461e-02 2.73903662614e-02 2.65400648390e-02 2.46286629032e-02 2.17948830411e-02 1.85132763207e-02 - 1.52762540169e-02 1.24265880360e-02 1.01198499891e-02 8.37400390520e-03 7.17862759619e-03 6.56138198057e-03 - 6.56138198057e-03 7.17862759619e-03 8.37400390520e-03 1.01198499891e-02 1.24265880360e-02 1.52762540169e-02 - 2.33118163733e-02 2.55791633527e-02 2.62352804996e-02 2.54927185045e-02 2.45139707404e-02 2.45139707404e-02 - 2.54927185045e-02 2.62352804996e-02 2.55791633527e-02 2.33118163733e-02 1.99975947255e-02 1.63906429721e-02 - 1.30485807277e-02 1.02401100733e-02 8.04455907974e-03 6.50338402317e-03 5.69411798450e-03 5.69411798450e-03 - 6.50338402317e-03 8.04455907974e-03 1.02401100733e-02 1.30485807277e-02 1.63906429721e-02 1.99975947255e-02 - 2.55791633527e-02 2.61140936360e-02 2.45139707404e-02 2.20787139947e-02 2.09116171502e-02 2.20787139947e-02 - 2.45139707404e-02 2.61140936360e-02 2.55791633527e-02 2.30253665656e-02 1.93871115228e-02 1.56224022211e-02 - 1.22934163580e-02 9.57619631350e-03 7.50451749096e-03 6.16385997204e-03 5.69328932590e-03 6.16385997204e-03 - 7.50451749096e-03 9.57619631350e-03 1.22934163580e-02 1.56224022211e-02 1.93871115228e-02 2.30253665656e-02 - 2.62352804996e-02 2.45139707404e-02 2.09116171502e-02 1.79304117733e-02 1.79304117733e-02 2.09116171502e-02 - 2.45139707404e-02 2.62352804996e-02 2.51931467427e-02 2.20777720561e-02 1.81502966987e-02 1.43894839458e-02 - 1.12137577955e-02 8.68876908730e-03 6.85668012358e-03 5.86550216407e-03 5.86550216407e-03 6.85668012358e-03 - 8.68876908730e-03 1.12137577955e-02 1.43894839458e-02 1.81502966987e-02 2.20777720561e-02 2.51931467427e-02 - 2.54927185045e-02 2.20787139947e-02 1.79304117733e-02 1.60510791347e-02 1.79304117733e-02 2.20787139947e-02 - 2.54927185045e-02 2.61756575962e-02 2.40839675381e-02 2.03983415996e-02 1.64062528191e-02 1.28580577518e-02 - 9.96167567605e-03 7.73176989509e-03 6.26631005461e-03 5.74651192180e-03 6.26631005461e-03 7.73176989509e-03 - 9.96167567605e-03 1.28580577518e-02 1.64062528191e-02 2.03983415996e-02 2.40839675381e-02 2.61756575962e-02 - 2.45139707404e-02 2.09116171502e-02 1.79304117733e-02 1.79304117733e-02 2.09116171502e-02 2.45139707404e-02 - 2.62352804996e-02 2.51931467427e-02 2.20777720561e-02 1.81502966987e-02 1.43894839458e-02 1.12137577955e-02 - 8.68876908730e-03 6.85668012358e-03 5.86550216407e-03 5.86550216407e-03 6.85668012358e-03 8.68876908730e-03 - 1.12137577955e-02 1.43894839458e-02 1.81502966987e-02 2.20777720561e-02 2.51931467427e-02 2.62352804996e-02 - 2.45139707404e-02 2.20787139947e-02 2.09116171502e-02 2.20787139947e-02 2.45139707404e-02 2.61140936360e-02 - 2.55791633527e-02 2.30253665656e-02 1.93871115228e-02 1.56224022211e-02 1.22934163580e-02 9.57619631350e-03 - 7.50451749096e-03 6.16385997204e-03 5.69328932590e-03 6.16385997204e-03 7.50451749096e-03 9.57619631350e-03 - 1.22934163580e-02 1.56224022211e-02 1.93871115228e-02 2.30253665656e-02 2.55791633527e-02 2.61140936360e-02 - 2.54927185045e-02 2.45139707404e-02 2.45139707404e-02 2.54927185045e-02 2.62352804996e-02 2.55791633527e-02 - 2.33118163733e-02 1.99975947255e-02 1.63906429721e-02 1.30485807277e-02 1.02401100733e-02 8.04455907974e-03 - 6.50338402317e-03 5.69411798450e-03 5.69411798450e-03 6.50338402317e-03 8.04455907974e-03 1.02401100733e-02 - 1.30485807277e-02 1.63906429721e-02 1.99975947255e-02 2.33118163733e-02 2.55791633527e-02 2.62352804996e-02 - 2.62352804996e-02 2.61140936360e-02 2.62352804996e-02 2.61756575962e-02 2.51931467427e-02 2.30253665656e-02 - 1.99975947255e-02 1.66439398964e-02 1.34169388324e-02 1.06109348738e-02 8.37400390520e-03 6.75449318420e-03 - 5.76667312114e-03 5.43239778449e-03 5.76667312114e-03 6.75449318420e-03 8.37400390520e-03 1.06109348738e-02 - 1.34169388324e-02 1.66439398964e-02 1.99975947255e-02 2.30253665656e-02 2.51931467428e-02 2.61756575962e-02 - 2.55791633527e-02 2.55791633527e-02 2.51931467427e-02 2.40839675381e-02 2.20777720561e-02 1.93871115228e-02 - 1.63906429721e-02 1.34169388324e-02 1.07225418333e-02 8.50665152157e-03 6.87426464911e-03 5.82746138702e-03 - 5.32068131556e-03 5.32068131556e-03 5.82746138702e-03 6.87426464911e-03 8.50665152157e-03 1.07225418333e-02 - 1.34169388324e-02 1.63906429721e-02 1.93871115228e-02 2.20777720561e-02 2.40839675381e-02 2.51931467428e-02 - 2.33118163733e-02 2.30253665656e-02 2.20777720561e-02 2.03983415996e-02 1.81502966987e-02 1.56224022211e-02 - 1.30485807277e-02 1.06109348738e-02 8.50665152157e-03 6.90308343843e-03 5.85076317272e-03 5.28636096575e-03 - 5.11303460195e-03 5.28636096575e-03 5.85076317272e-03 6.90308343843e-03 8.50665152157e-03 1.06109348738e-02 - 1.30485807277e-02 1.56224022211e-02 1.81502966987e-02 2.03983415996e-02 2.20777720561e-02 2.30253665656e-02 - 1.99975947255e-02 1.93871115228e-02 1.81502966987e-02 1.64062528191e-02 1.43894839458e-02 1.22934163580e-02 - 1.02401100733e-02 8.37400390520e-03 6.87426464911e-03 5.85076317272e-03 5.27978060753e-03 5.04614391795e-03 - 5.04614391795e-03 5.27978060753e-03 5.85076317272e-03 6.87426464911e-03 8.37400390520e-03 1.02401100733e-02 - 1.22934163580e-02 1.43894839458e-02 1.64062528191e-02 1.81502966987e-02 1.93871115228e-02 1.99975947255e-02 - 1.63906429721e-02 1.56224022211e-02 1.43894839458e-02 1.28580577518e-02 1.12137577955e-02 9.57619631350e-03 - 8.04455907974e-03 6.75449318420e-03 5.82746138702e-03 5.28636096575e-03 5.04614391795e-03 4.98513742725e-03 - 5.04614391795e-03 5.28636096575e-03 5.82746138702e-03 6.75449318420e-03 8.04455907974e-03 9.57619631350e-03 - 1.12137577955e-02 1.28580577518e-02 1.43894839458e-02 1.56224022211e-02 1.63906429721e-02 1.66439398964e-02 - 1.30485807277e-02 1.22934163580e-02 1.12137577955e-02 9.96167567604e-03 8.68876908730e-03 7.50451749096e-03 - 6.50338402317e-03 5.76667312114e-03 5.32068131556e-03 5.11303460195e-03 5.04614391795e-03 5.04614391795e-03 - 5.11303460195e-03 5.32068131556e-03 5.76667312114e-03 6.50338402317e-03 7.50451749096e-03 8.68876908730e-03 - 9.96167567604e-03 1.12137577955e-02 1.22934163580e-02 1.30485807277e-02 1.34169388324e-02 1.34169388324e-02 - 1.02401100733e-02 9.57619631350e-03 8.68876908730e-03 7.73176989509e-03 6.85668012358e-03 6.16385997204e-03 - 5.69411798450e-03 5.43239778449e-03 5.32068131556e-03 5.28636096575e-03 5.27978060753e-03 5.28636096575e-03 - 5.32068131556e-03 5.43239778449e-03 5.69411798450e-03 6.16385997204e-03 6.85668012358e-03 7.73176989509e-03 - 8.68876908730e-03 9.57619631350e-03 1.02401100733e-02 1.06109348738e-02 1.07225418333e-02 1.06109348738e-02 - 8.04455907974e-03 7.50451749096e-03 6.85668012358e-03 6.26631005461e-03 5.86550216407e-03 5.69328932590e-03 - 5.69411798450e-03 5.76667312114e-03 5.82746138702e-03 5.85076317272e-03 5.85076317272e-03 5.82746138702e-03 - 5.76667312114e-03 5.69411798450e-03 5.69328932590e-03 5.86550216407e-03 6.26631005461e-03 6.85668012358e-03 - 7.50451749096e-03 8.04455907974e-03 8.37400390520e-03 8.50665152156e-03 8.50665152156e-03 8.37400390520e-03 - 6.50338402317e-03 6.16385997204e-03 5.86550216407e-03 5.74651192180e-03 5.86550216407e-03 6.16385997204e-03 - 6.50338402317e-03 6.75449318420e-03 6.87426464911e-03 6.90308343843e-03 6.87426464911e-03 6.75449318420e-03 - 6.50338402317e-03 6.16385997204e-03 5.86550216407e-03 5.74651192180e-03 5.86550216407e-03 6.16385997204e-03 - 6.50338402317e-03 6.75449318420e-03 6.87426464911e-03 6.90308343843e-03 6.87426464911e-03 6.75449318420e-03 - 5.69411798450e-03 5.69328932590e-03 5.86550216407e-03 6.26631005461e-03 6.85668012358e-03 7.50451749096e-03 - 8.04455907974e-03 8.37400390520e-03 8.50665152156e-03 8.50665152156e-03 8.37400390520e-03 8.04455907974e-03 - 7.50451749096e-03 6.85668012358e-03 6.26631005461e-03 5.86550216407e-03 5.69328932590e-03 5.69411798450e-03 - 5.76667312114e-03 5.82746138702e-03 5.85076317272e-03 5.85076317272e-03 5.82746138702e-03 5.76667312114e-03 - 5.69411798450e-03 6.16385997204e-03 6.85668012358e-03 7.73176989509e-03 8.68876908730e-03 9.57619631350e-03 - 1.02401100733e-02 1.06109348738e-02 1.07225418333e-02 1.06109348738e-02 1.02401100733e-02 9.57619631350e-03 - 8.68876908730e-03 7.73176989509e-03 6.85668012358e-03 6.16385997204e-03 5.69411798450e-03 5.43239778449e-03 - 5.32068131556e-03 5.28636096575e-03 5.27978060753e-03 5.28636096575e-03 5.32068131556e-03 5.43239778449e-03 - 6.50338402317e-03 7.50451749096e-03 8.68876908730e-03 9.96167567604e-03 1.12137577955e-02 1.22934163580e-02 - 1.30485807277e-02 1.34169388324e-02 1.34169388324e-02 1.30485807277e-02 1.22934163580e-02 1.12137577955e-02 - 9.96167567604e-03 8.68876908730e-03 7.50451749096e-03 6.50338402317e-03 5.76667312114e-03 5.32068131556e-03 - 5.11303460195e-03 5.04614391795e-03 5.04614391795e-03 5.11303460195e-03 5.32068131556e-03 5.76667312114e-03 - 8.04455907974e-03 9.57619631350e-03 1.12137577955e-02 1.28580577518e-02 1.43894839458e-02 1.56224022211e-02 - 1.63906429721e-02 1.66439398964e-02 1.63906429721e-02 1.56224022211e-02 1.43894839458e-02 1.28580577518e-02 - 1.12137577955e-02 9.57619631350e-03 8.04455907974e-03 6.75449318420e-03 5.82746138702e-03 5.28636096575e-03 - 5.04614391795e-03 4.98513742725e-03 5.04614391795e-03 5.28636096575e-03 5.82746138702e-03 6.75449318420e-03 - 1.02401100733e-02 1.22934163580e-02 1.43894839458e-02 1.64062528191e-02 1.81502966987e-02 1.93871115228e-02 - 1.99975947255e-02 1.99975947255e-02 1.93871115228e-02 1.81502966987e-02 1.64062528191e-02 1.43894839458e-02 - 1.22934163580e-02 1.02401100733e-02 8.37400390520e-03 6.87426464911e-03 5.85076317272e-03 5.27978060753e-03 - 5.04614391795e-03 5.04614391795e-03 5.27978060753e-03 5.85076317272e-03 6.87426464911e-03 8.37400390520e-03 - 1.30485807277e-02 1.56224022211e-02 1.81502966987e-02 2.03983415996e-02 2.20777720561e-02 2.30253665656e-02 - 2.33118163733e-02 2.30253665656e-02 2.20777720561e-02 2.03983415996e-02 1.81502966987e-02 1.56224022211e-02 - 1.30485807277e-02 1.06109348738e-02 8.50665152157e-03 6.90308343843e-03 5.85076317272e-03 5.28636096575e-03 - 5.11303460195e-03 5.28636096575e-03 5.85076317272e-03 6.90308343843e-03 8.50665152157e-03 1.06109348738e-02 - 1.63906429721e-02 1.93871115228e-02 2.20777720561e-02 2.40839675381e-02 2.51931467427e-02 2.55791633527e-02 - 2.55791633527e-02 2.51931467428e-02 2.40839675381e-02 2.20777720561e-02 1.93871115228e-02 1.63906429721e-02 - 1.34169388324e-02 1.07225418333e-02 8.50665152157e-03 6.87426464911e-03 5.82746138702e-03 5.32068131556e-03 - 5.32068131556e-03 5.82746138702e-03 6.87426464911e-03 8.50665152157e-03 1.07225418333e-02 1.34169388324e-02 - 1.99975947255e-02 2.30253665656e-02 2.51931467427e-02 2.61756575962e-02 2.62352804996e-02 2.61140936360e-02 - 2.62352804996e-02 2.61756575962e-02 2.51931467428e-02 2.30253665656e-02 1.99975947255e-02 1.66439398964e-02 - 1.34169388324e-02 1.06109348738e-02 8.37400390520e-03 6.75449318420e-03 5.76667312114e-03 5.43239778449e-03 - 5.76667312114e-03 6.75449318420e-03 8.37400390520e-03 1.06109348738e-02 1.34169388324e-02 1.66439398964e-02 - 2.62414819209e-02 2.75584598461e-02 2.69081613820e-02 2.52955690008e-02 2.44579386769e-02 2.52955690008e-02 - 2.69081613820e-02 2.75584598461e-02 2.62414819209e-02 2.31387869200e-02 1.91106081253e-02 1.50739196267e-02 - 1.16095381953e-02 8.89920278516e-03 6.92803651720e-03 5.69411798450e-03 5.26652136817e-03 5.69411798450e-03 - 6.92803651720e-03 8.89920278516e-03 1.16095381953e-02 1.50739196267e-02 1.91106081253e-02 2.31387869200e-02 - 2.75584598461e-02 2.65224863971e-02 2.36141349499e-02 2.10846196163e-02 2.10846196163e-02 2.36141349499e-02 - 2.65224863971e-02 2.75584598461e-02 2.59618688149e-02 2.24043982374e-02 1.81179302207e-02 1.41149739580e-02 - 1.08500532301e-02 8.36360112567e-03 6.62087909721e-03 5.69328932590e-03 5.69328932590e-03 6.62087909721e-03 - 8.36360112567e-03 1.08500532301e-02 1.41149739580e-02 1.81179302207e-02 2.24043982374e-02 2.59618688149e-02 - 2.69081613820e-02 2.36141349499e-02 1.95329704288e-02 1.76763940774e-02 1.95329704288e-02 2.36141349499e-02 - 2.69081613820e-02 2.73903662614e-02 2.49964037842e-02 2.09446754967e-02 1.66276099946e-02 1.28937939562e-02 - 9.96234500644e-03 7.77912772161e-03 6.36555591033e-03 5.86550216407e-03 6.36555591034e-03 7.77912772161e-03 - 9.96234500644e-03 1.28937939562e-02 1.66276099946e-02 2.09446754967e-02 2.49964037842e-02 2.73903662614e-02 - 2.52955690008e-02 2.10846196163e-02 1.76763940774e-02 1.76763940774e-02 2.10846196163e-02 2.52955690008e-02 - 2.74769895380e-02 2.65400648390e-02 2.32030440049e-02 1.89263764236e-02 1.48932100517e-02 1.16029685725e-02 - 9.07031679536e-03 7.25301575415e-03 6.26631005461e-03 6.26631005461e-03 7.25301575416e-03 9.07031679536e-03 - 1.16029685725e-02 1.48932100517e-02 1.89263764236e-02 2.32030440049e-02 2.65400648390e-02 2.74769895380e-02 - 2.44579386769e-02 2.10846196163e-02 1.95329704288e-02 2.10846196163e-02 2.44579386769e-02 2.70513354759e-02 - 2.71142998602e-02 2.46286629032e-02 2.07196781889e-02 1.66389339973e-02 1.31234388541e-02 1.03422447642e-02 - 8.23785813083e-03 6.85668012358e-03 6.36555591034e-03 6.85668012358e-03 8.23785813083e-03 1.03422447642e-02 - 1.31234388541e-02 1.66389339973e-02 2.07196781889e-02 2.46286629032e-02 2.71142998602e-02 2.70513354759e-02 - 2.52955690008e-02 2.36141349499e-02 2.36141349499e-02 2.52955690008e-02 2.70513354759e-02 2.72051400328e-02 - 2.52445347087e-02 2.17948830411e-02 1.78888288560e-02 1.43188164440e-02 1.14031750108e-02 9.13724506495e-03 - 7.50451749096e-03 6.62087909721e-03 6.62087909721e-03 7.50451749096e-03 9.13724506496e-03 1.14031750108e-02 - 1.43188164440e-02 1.78888288560e-02 2.17948830411e-02 2.52445347087e-02 2.72051400328e-02 2.70513354759e-02 - 2.69081613820e-02 2.65224863971e-02 2.69081613820e-02 2.74769895380e-02 2.71142998602e-02 2.52445347087e-02 - 2.21386056500e-02 1.85132763207e-02 1.50424663757e-02 1.21045128275e-02 9.78219490488e-03 8.04455907974e-03 - 6.92803651720e-03 6.53729803366e-03 6.92803651720e-03 8.04455907974e-03 9.78219490488e-03 1.21045128275e-02 - 1.50424663757e-02 1.85132763207e-02 2.21386056500e-02 2.52445347088e-02 2.71142998602e-02 2.74769895380e-02 - 2.75584598461e-02 2.75584598461e-02 2.73903662614e-02 2.65400648390e-02 2.46286629032e-02 2.17948830411e-02 - 1.85132763207e-02 1.52762540169e-02 1.24265880360e-02 1.01198499891e-02 8.37400390520e-03 7.17862759619e-03 - 6.56138198057e-03 6.56138198057e-03 7.17862759619e-03 8.37400390520e-03 1.01198499891e-02 1.24265880360e-02 - 1.52762540169e-02 1.85132763207e-02 2.17948830411e-02 2.46286629032e-02 2.65400648390e-02 2.73903662614e-02 - 2.62414819209e-02 2.59618688149e-02 2.49964037842e-02 2.32030440049e-02 2.07196781889e-02 1.78888288560e-02 - 1.50424663757e-02 1.24265880360e-02 1.02153626600e-02 8.50665152157e-03 7.31933010296e-03 6.62684043801e-03 - 6.39967919814e-03 6.62684043801e-03 7.31933010296e-03 8.50665152157e-03 1.02153626600e-02 1.24265880360e-02 - 1.50424663757e-02 1.78888288560e-02 2.07196781889e-02 2.32030440049e-02 2.49964037842e-02 2.59618688149e-02 - 2.31387869200e-02 2.24043982374e-02 2.09446754967e-02 1.89263764236e-02 1.66389339973e-02 1.43188164440e-02 - 1.21045128275e-02 1.01198499891e-02 8.50665152157e-03 7.36052161959e-03 6.67439308888e-03 6.36714430779e-03 - 6.36714430779e-03 6.67439308888e-03 7.36052161959e-03 8.50665152157e-03 1.01198499891e-02 1.21045128275e-02 - 1.43188164440e-02 1.66389339973e-02 1.89263764236e-02 2.09446754967e-02 2.24043982374e-02 2.31387869200e-02 - 1.91106081253e-02 1.81179302207e-02 1.66276099946e-02 1.48932100517e-02 1.31234388541e-02 1.14031750108e-02 - 9.78219490488e-03 8.37400390520e-03 7.31933010296e-03 6.67439308888e-03 6.36996915701e-03 6.28719817635e-03 - 6.36996915701e-03 6.67439308888e-03 7.31933010296e-03 8.37400390520e-03 9.78219490488e-03 1.14031750108e-02 - 1.31234388541e-02 1.48932100517e-02 1.66276099946e-02 1.81179302207e-02 1.91106081253e-02 1.94536795471e-02 - 1.50739196267e-02 1.41149739580e-02 1.28937939562e-02 1.16029685725e-02 1.03422447642e-02 9.13724506496e-03 - 8.04455907974e-03 7.17862759619e-03 6.62684043802e-03 6.36714430779e-03 6.28719817635e-03 6.28719817635e-03 - 6.36714430779e-03 6.62684043801e-03 7.17862759619e-03 8.04455907974e-03 9.13724506495e-03 1.03422447642e-02 - 1.16029685725e-02 1.28937939562e-02 1.41149739580e-02 1.50739196267e-02 1.55937378378e-02 1.55937378378e-02 - 1.16095381953e-02 1.08500532301e-02 9.96234500643e-03 9.07031679536e-03 8.23785813083e-03 7.50451749096e-03 - 6.92803651720e-03 6.56138198057e-03 6.39967919814e-03 6.36714430779e-03 6.36996915701e-03 6.36714430779e-03 - 6.39967919814e-03 6.56138198057e-03 6.92803651720e-03 7.50451749096e-03 8.23785813083e-03 9.07031679536e-03 - 9.96234500643e-03 1.08500532301e-02 1.16095381953e-02 1.21090534529e-02 1.22800512627e-02 1.21090534529e-02 - 8.89920278515e-03 8.36360112566e-03 7.77912772161e-03 7.25301575416e-03 6.85668012358e-03 6.62087909721e-03 - 6.53729803367e-03 6.56138198057e-03 6.62684043802e-03 6.67439308889e-03 6.67439308888e-03 6.62684043801e-03 - 6.56138198057e-03 6.53729803366e-03 6.62087909721e-03 6.85668012358e-03 7.25301575415e-03 7.77912772161e-03 - 8.36360112566e-03 8.89920278515e-03 9.28357359922e-03 9.47290433110e-03 9.47290433110e-03 9.28357359922e-03 - 6.92803651719e-03 6.62087909720e-03 6.36555591033e-03 6.26631005461e-03 6.36555591034e-03 6.62087909721e-03 - 6.92803651720e-03 7.17862759619e-03 7.31933010296e-03 7.36052161959e-03 7.31933010296e-03 7.17862759619e-03 - 6.92803651720e-03 6.62087909721e-03 6.36555591034e-03 6.26631005461e-03 6.36555591033e-03 6.62087909720e-03 - 6.92803651719e-03 7.17862759619e-03 7.31933010296e-03 7.36052161958e-03 7.31933010296e-03 7.17862759619e-03 - 5.69411798450e-03 5.69328932590e-03 5.86550216407e-03 6.26631005461e-03 6.85668012358e-03 7.50451749096e-03 - 8.04455907974e-03 8.37400390520e-03 8.50665152157e-03 8.50665152157e-03 8.37400390520e-03 8.04455907974e-03 - 7.50451749096e-03 6.85668012358e-03 6.26631005461e-03 5.86550216407e-03 5.69328932590e-03 5.69411798450e-03 - 5.76667312114e-03 5.82746138702e-03 5.85076317272e-03 5.85076317272e-03 5.82746138702e-03 5.76667312114e-03 - 5.26652136817e-03 5.69328932590e-03 6.36555591034e-03 7.25301575416e-03 8.23785813083e-03 9.13724506495e-03 - 9.78219490488e-03 1.01198499891e-02 1.02153626600e-02 1.01198499891e-02 9.78219490488e-03 9.13724506495e-03 - 8.23785813083e-03 7.25301575416e-03 6.36555591034e-03 5.69328932590e-03 5.26652136817e-03 5.04343778102e-03 - 4.94632325183e-03 4.90588959931e-03 4.89318523683e-03 4.90588959931e-03 4.94632325183e-03 5.04343778102e-03 - 5.69411798450e-03 6.62087909721e-03 7.77912772161e-03 9.07031679536e-03 1.03422447642e-02 1.14031750108e-02 - 1.21045128275e-02 1.24265880360e-02 1.24265880360e-02 1.21045128275e-02 1.14031750108e-02 1.03422447642e-02 - 9.07031679536e-03 7.77912772161e-03 6.62087909721e-03 5.69411798450e-03 5.04343778102e-03 4.65175947716e-03 - 4.45367296569e-03 4.37599558137e-03 4.37599558137e-03 4.45367296569e-03 4.65175947716e-03 5.04343778102e-03 - 6.92803651720e-03 8.36360112567e-03 9.96234500644e-03 1.16029685725e-02 1.31234388541e-02 1.43188164440e-02 - 1.50424663757e-02 1.52762540169e-02 1.50424663757e-02 1.43188164440e-02 1.31234388541e-02 1.16029685725e-02 - 9.96234500643e-03 8.36360112567e-03 6.92803651720e-03 5.76667312114e-03 4.94632325183e-03 4.45367296569e-03 - 4.21584246473e-03 4.14918646508e-03 4.21584246472e-03 4.45367296569e-03 4.94632325183e-03 5.76667312114e-03 - 8.89920278516e-03 1.08500532301e-02 1.28937939562e-02 1.48932100517e-02 1.66389339973e-02 1.78888288560e-02 - 1.85132763207e-02 1.85132763207e-02 1.78888288560e-02 1.66389339973e-02 1.48932100517e-02 1.28937939562e-02 - 1.08500532301e-02 8.89920278516e-03 7.17862759619e-03 5.82746138702e-03 4.90588959931e-03 4.37599558137e-03 - 4.14918646508e-03 4.14918646508e-03 4.37599558137e-03 4.90588959931e-03 5.82746138702e-03 7.17862759619e-03 - 1.16095381953e-02 1.41149739580e-02 1.66276099946e-02 1.89263764236e-02 2.07196781889e-02 2.17948830411e-02 - 2.21386056500e-02 2.17948830411e-02 2.07196781889e-02 1.89263764236e-02 1.66276099946e-02 1.41149739580e-02 - 1.16095381953e-02 9.28357359923e-03 7.31933010296e-03 5.85076317272e-03 4.89318523683e-03 4.37599558137e-03 - 4.21584246472e-03 4.37599558137e-03 4.89318523683e-03 5.85076317272e-03 7.31933010296e-03 9.28357359923e-03 - 1.50739196267e-02 1.81179302207e-02 2.09446754967e-02 2.32030440049e-02 2.46286629032e-02 2.52445347088e-02 - 2.52445347088e-02 2.46286629032e-02 2.32030440049e-02 2.09446754967e-02 1.81179302207e-02 1.50739196267e-02 - 1.21090534529e-02 9.47290433111e-03 7.36052161959e-03 5.85076317272e-03 4.90588959931e-03 4.45367296569e-03 - 4.45367296569e-03 4.90588959931e-03 5.85076317272e-03 7.36052161959e-03 9.47290433111e-03 1.21090534529e-02 - 1.91106081253e-02 2.24043982374e-02 2.49964037842e-02 2.65400648390e-02 2.71142998602e-02 2.72051400328e-02 - 2.71142998602e-02 2.65400648390e-02 2.49964037842e-02 2.24043982374e-02 1.91106081253e-02 1.55937378378e-02 - 1.22800512627e-02 9.47290433111e-03 7.31933010296e-03 5.82746138702e-03 4.94632325183e-03 4.65175947716e-03 - 4.94632325183e-03 5.82746138702e-03 7.31933010296e-03 9.47290433111e-03 1.22800512627e-02 1.55937378378e-02 - 2.31387869200e-02 2.59618688149e-02 2.73903662614e-02 2.74769895380e-02 2.70513354759e-02 2.70513354759e-02 - 2.74769895380e-02 2.73903662614e-02 2.59618688149e-02 2.31387869200e-02 1.94536795471e-02 1.55937378378e-02 - 1.21090534529e-02 9.28357359923e-03 7.17862759619e-03 5.76667312114e-03 5.04343778102e-03 5.04343778102e-03 - 5.76667312114e-03 7.17862759619e-03 9.28357359923e-03 1.21090534529e-02 1.55937378378e-02 1.94536795471e-02 - 3.06058831029e-02 3.03789244340e-02 2.81921950446e-02 2.61164989591e-02 2.61164989591e-02 2.81921950446e-02 - 3.03789244340e-02 3.06058831029e-02 2.81653274579e-02 2.37301976795e-02 1.86355176327e-02 1.40806383265e-02 - 1.06051613742e-02 8.15758815908e-03 6.53729803366e-03 5.69411798450e-03 5.69411798450e-03 6.53729803367e-03 - 8.15758815908e-03 1.06051613742e-02 1.40806383265e-02 1.86355176327e-02 2.37301976795e-02 2.81653274579e-02 - 3.03789244340e-02 2.73937018060e-02 2.35389825848e-02 2.17746576295e-02 2.35389825848e-02 2.73937018060e-02 - 3.03789244340e-02 3.04051143293e-02 2.73091501295e-02 2.23752455712e-02 1.72611853896e-02 1.30618774824e-02 - 1.00210602336e-02 7.92498387887e-03 6.62087909720e-03 6.16385997204e-03 6.62087909721e-03 7.92498387888e-03 - 1.00210602336e-02 1.30618774824e-02 1.72611853896e-02 2.23752455712e-02 2.73091501295e-02 3.04051143293e-02 - 2.81921950446e-02 2.35389825848e-02 1.98434779695e-02 1.98434779695e-02 2.35389825848e-02 2.81921950446e-02 - 3.06792139923e-02 2.95890354456e-02 2.55726183949e-02 2.03920879638e-02 1.56489970468e-02 1.20439356208e-02 - 9.50248837487e-03 7.77912772161e-03 6.85668012358e-03 6.85668012358e-03 7.77912772161e-03 9.50248837488e-03 - 1.20439356208e-02 1.56489970468e-02 2.03920879638e-02 2.55726183949e-02 2.95890354456e-02 3.06792139923e-02 - 2.61164989591e-02 2.17746576295e-02 1.98434779695e-02 2.17746576295e-02 2.61164989591e-02 2.97974895143e-02 - 3.04773481040e-02 2.78013937268e-02 2.30939042849e-02 1.81430425204e-02 1.40901390285e-02 1.11481007476e-02 - 9.07031679536e-03 7.73176989509e-03 7.25301575416e-03 7.73176989509e-03 9.07031679536e-03 1.11481007476e-02 - 1.40901390285e-02 1.81430425204e-02 2.30939042849e-02 2.78013937268e-02 3.04773481040e-02 2.97974895143e-02 - 2.61164989591e-02 2.35389825848e-02 2.35389825848e-02 2.61164989591e-02 2.92436594720e-02 3.05548189106e-02 - 2.89473471924e-02 2.49902463304e-02 2.02040886880e-02 1.59287126341e-02 1.26855285494e-02 1.03422447642e-02 - 8.68876908730e-03 7.77912772161e-03 7.77912772161e-03 8.68876908730e-03 1.03422447642e-02 1.26855285494e-02 - 1.59287126341e-02 2.02040886880e-02 2.49902463304e-02 2.89473471924e-02 3.05548189106e-02 2.92436594720e-02 - 2.81921950446e-02 2.73937018060e-02 2.81921950446e-02 2.97974895143e-02 3.05548189106e-02 2.92643382604e-02 - 2.59181147281e-02 2.15195877577e-02 1.72812015630e-02 1.38961516944e-02 1.14031750108e-02 9.57619631350e-03 - 8.36360112567e-03 7.92498387888e-03 8.36360112567e-03 9.57619631350e-03 1.14031750108e-02 1.38961516944e-02 - 1.72812015630e-02 2.15195877577e-02 2.59181147281e-02 2.92643382604e-02 3.05548189106e-02 2.97974895143e-02 - 3.03789244340e-02 3.03789244340e-02 3.06792139923e-02 3.04773481039e-02 2.89473471924e-02 2.59181147280e-02 - 2.19611077293e-02 1.79741436761e-02 1.46288647205e-02 1.21045128275e-02 1.02401100733e-02 8.89920278515e-03 - 8.15758815907e-03 8.15758815907e-03 8.89920278516e-03 1.02401100733e-02 1.21045128275e-02 1.46288647205e-02 - 1.79741436761e-02 2.19611077293e-02 2.59181147281e-02 2.89473471924e-02 3.04773481040e-02 3.06792139923e-02 - 3.06058831029e-02 3.04051143293e-02 2.95890354456e-02 2.78013937268e-02 2.49902463304e-02 2.15195877577e-02 - 1.79741436761e-02 1.48656767560e-02 1.24265880360e-02 1.06109348738e-02 9.28357359923e-03 8.41997198039e-03 - 8.11230157084e-03 8.41997198039e-03 9.28357359923e-03 1.06109348738e-02 1.24265880360e-02 1.48656767560e-02 - 1.79741436761e-02 2.15195877577e-02 2.49902463304e-02 2.78013937268e-02 2.95890354456e-02 3.04051143293e-02 - 2.81653274579e-02 2.73091501295e-02 2.55726183949e-02 2.30939042849e-02 2.02040886880e-02 1.72812015630e-02 - 1.46288647205e-02 1.24265880360e-02 1.07225418333e-02 9.47290433111e-03 8.62540500905e-03 8.18383164206e-03 - 8.18383164206e-03 8.62540500905e-03 9.47290433111e-03 1.07225418333e-02 1.24265880360e-02 1.46288647205e-02 - 1.72812015630e-02 2.02040886880e-02 2.30939042849e-02 2.55726183949e-02 2.73091501295e-02 2.81653274579e-02 - 2.37301976795e-02 2.23752455712e-02 2.03920879638e-02 1.81430425204e-02 1.59287126341e-02 1.38961516944e-02 - 1.21045128275e-02 1.06109348738e-02 9.47290433111e-03 8.70256016586e-03 8.26409670065e-03 8.12161313174e-03 - 8.26409670065e-03 8.70256016587e-03 9.47290433111e-03 1.06109348738e-02 1.21045128275e-02 1.38961516944e-02 - 1.59287126341e-02 1.81430425204e-02 2.03920879638e-02 2.23752455712e-02 2.37301976795e-02 2.42075414898e-02 - 1.86355176327e-02 1.72611853896e-02 1.56489970468e-02 1.40901390285e-02 1.26855285494e-02 1.14031750108e-02 - 1.02401100733e-02 9.28357359923e-03 8.62540500906e-03 8.26409670065e-03 8.11874591476e-03 8.11874591476e-03 - 8.26409670065e-03 8.62540500906e-03 9.28357359923e-03 1.02401100733e-02 1.14031750108e-02 1.26855285494e-02 - 1.40901390285e-02 1.56489970468e-02 1.72611853896e-02 1.86355176327e-02 1.94290489291e-02 1.94290489291e-02 - 1.40806383265e-02 1.30618774824e-02 1.20439356208e-02 1.11481007476e-02 1.03422447642e-02 9.57619631350e-03 - 8.89920278516e-03 8.41997198039e-03 8.18383164206e-03 8.12161313174e-03 8.11874591477e-03 8.12161313175e-03 - 8.18383164206e-03 8.41997198039e-03 8.89920278516e-03 9.57619631350e-03 1.03422447642e-02 1.11481007476e-02 - 1.20439356208e-02 1.30618774824e-02 1.40806383265e-02 1.48492046219e-02 1.51354763629e-02 1.48492046219e-02 - 1.06051613742e-02 1.00210602336e-02 9.50248837487e-03 9.07031679536e-03 8.68876908730e-03 8.36360112567e-03 - 8.15758815908e-03 8.11230157085e-03 8.18383164206e-03 8.26409670065e-03 8.26409670065e-03 8.18383164206e-03 - 8.11230157085e-03 8.15758815908e-03 8.36360112567e-03 8.68876908730e-03 9.07031679536e-03 9.50248837487e-03 - 1.00210602336e-02 1.06051613742e-02 1.11331987338e-02 1.14481144732e-02 1.14481144732e-02 1.11331987338e-02 - 8.15758815907e-03 7.92498387887e-03 7.77912772161e-03 7.73176989509e-03 7.77912772161e-03 7.92498387888e-03 - 8.15758815908e-03 8.41997198039e-03 8.62540500906e-03 8.70256016587e-03 8.62540500905e-03 8.41997198039e-03 - 8.15758815908e-03 7.92498387888e-03 7.77912772161e-03 7.73176989509e-03 7.77912772161e-03 7.92498387887e-03 - 8.15758815907e-03 8.41997198038e-03 8.62540500905e-03 8.70256016586e-03 8.62540500905e-03 8.41997198038e-03 - 6.53729803366e-03 6.62087909720e-03 6.85668012358e-03 7.25301575416e-03 7.77912772161e-03 8.36360112567e-03 - 8.89920278516e-03 9.28357359923e-03 9.47290433111e-03 9.47290433111e-03 9.28357359923e-03 8.89920278516e-03 - 8.36360112567e-03 7.77912772161e-03 7.25301575416e-03 6.85668012358e-03 6.62087909720e-03 6.53729803366e-03 - 6.56138198056e-03 6.62684043801e-03 6.67439308888e-03 6.67439308888e-03 6.62684043801e-03 6.56138198056e-03 - 5.69411798450e-03 6.16385997204e-03 6.85668012358e-03 7.73176989509e-03 8.68876908730e-03 9.57619631350e-03 - 1.02401100733e-02 1.06109348738e-02 1.07225418333e-02 1.06109348738e-02 1.02401100733e-02 9.57619631350e-03 - 8.68876908730e-03 7.73176989509e-03 6.85668012358e-03 6.16385997204e-03 5.69411798450e-03 5.43239778449e-03 - 5.32068131556e-03 5.28636096575e-03 5.27978060753e-03 5.28636096575e-03 5.32068131556e-03 5.43239778448e-03 - 5.69411798450e-03 6.62087909721e-03 7.77912772161e-03 9.07031679536e-03 1.03422447642e-02 1.14031750108e-02 - 1.21045128275e-02 1.24265880360e-02 1.24265880360e-02 1.21045128275e-02 1.14031750108e-02 1.03422447642e-02 - 9.07031679536e-03 7.77912772161e-03 6.62087909721e-03 5.69411798450e-03 5.04343778102e-03 4.65175947716e-03 - 4.45367296569e-03 4.37599558137e-03 4.37599558137e-03 4.45367296569e-03 4.65175947716e-03 5.04343778102e-03 - 6.53729803366e-03 7.92498387888e-03 9.50248837487e-03 1.11481007476e-02 1.26855285494e-02 1.38961516945e-02 - 1.46288647205e-02 1.48656767561e-02 1.46288647205e-02 1.38961516945e-02 1.26855285494e-02 1.11481007476e-02 - 9.50248837487e-03 7.92498387888e-03 6.53729803366e-03 5.43239778449e-03 4.65175947716e-03 4.16945913194e-03 - 3.92281290681e-03 3.84964089400e-03 3.92281290681e-03 4.16945913194e-03 4.65175947716e-03 5.43239778449e-03 - 8.15758815908e-03 1.00210602336e-02 1.20439356208e-02 1.40901390285e-02 1.59287126341e-02 1.72812015630e-02 - 1.79741436761e-02 1.79741436761e-02 1.72812015630e-02 1.59287126341e-02 1.40901390285e-02 1.20439356208e-02 - 1.00210602336e-02 8.15758815907e-03 6.56138198057e-03 5.32068131556e-03 4.45367296569e-03 3.92281290681e-03 - 3.67768921534e-03 3.67768921534e-03 3.92281290681e-03 4.45367296569e-03 5.32068131556e-03 6.56138198057e-03 - 1.06051613742e-02 1.30618774824e-02 1.56489970468e-02 1.81430425204e-02 2.02040886880e-02 2.15195877577e-02 - 2.19611077293e-02 2.15195877577e-02 2.02040886880e-02 1.81430425205e-02 1.56489970468e-02 1.30618774824e-02 - 1.06051613742e-02 8.41997198039e-03 6.62684043801e-03 5.28636096575e-03 4.37599558137e-03 3.84964089400e-03 - 3.67768921534e-03 3.84964089400e-03 4.37599558137e-03 5.28636096575e-03 6.62684043801e-03 8.41997198039e-03 - 1.40806383265e-02 1.72611853896e-02 2.03920879638e-02 2.30939042850e-02 2.49902463304e-02 2.59181147281e-02 - 2.59181147281e-02 2.49902463304e-02 2.30939042850e-02 2.03920879638e-02 1.72611853896e-02 1.40806383265e-02 - 1.11331987338e-02 8.62540500906e-03 6.67439308888e-03 5.27978060753e-03 4.37599558137e-03 3.92281290681e-03 - 3.92281290681e-03 4.37599558137e-03 5.27978060753e-03 6.67439308888e-03 8.62540500906e-03 1.11331987338e-02 - 1.86355176327e-02 2.23752455712e-02 2.55726183949e-02 2.78013937268e-02 2.89473471924e-02 2.92643382604e-02 - 2.89473471924e-02 2.78013937268e-02 2.55726183949e-02 2.23752455712e-02 1.86355176327e-02 1.48492046219e-02 - 1.14481144732e-02 8.70256016587e-03 6.67439308889e-03 5.28636096575e-03 4.45367296569e-03 4.16945913194e-03 - 4.45367296569e-03 5.28636096575e-03 6.67439308889e-03 8.70256016587e-03 1.14481144732e-02 1.48492046219e-02 - 2.37301976795e-02 2.73091501295e-02 2.95890354456e-02 3.04773481040e-02 3.05548189106e-02 3.05548189106e-02 - 3.04773481040e-02 2.95890354456e-02 2.73091501295e-02 2.37301976795e-02 1.94290489291e-02 1.51354763629e-02 - 1.14481144732e-02 8.62540500906e-03 6.62684043802e-03 5.32068131556e-03 4.65175947716e-03 4.65175947716e-03 - 5.32068131556e-03 6.62684043802e-03 8.62540500906e-03 1.14481144732e-02 1.51354763629e-02 1.94290489291e-02 - 2.81653274579e-02 3.04051143293e-02 3.06792139923e-02 2.97974895143e-02 2.92436594720e-02 2.97974895143e-02 - 3.06792139923e-02 3.04051143293e-02 2.81653274579e-02 2.42075414898e-02 1.94290489291e-02 1.48492046219e-02 - 1.11331987338e-02 8.41997198039e-03 6.56138198057e-03 5.43239778449e-03 5.04343778102e-03 5.43239778449e-03 - 6.56138198057e-03 8.41997198039e-03 1.11331987338e-02 1.48492046219e-02 1.94290489291e-02 2.42075414898e-02 - 3.48683159917e-02 3.31551669202e-02 3.05273339684e-02 2.92890570903e-02 3.05273339684e-02 3.31551669202e-02 - 3.48683159917e-02 3.38925854809e-02 2.99301385615e-02 2.40967233573e-02 1.81617046076e-02 1.34421324248e-02 - 1.02248872526e-02 8.15758815907e-03 6.92803651720e-03 6.50338402317e-03 6.92803651720e-03 8.15758815908e-03 - 1.02248872526e-02 1.34421324248e-02 1.81617046076e-02 2.40967233573e-02 2.99301385615e-02 3.38925854809e-02 - 3.31551669202e-02 2.95749315132e-02 2.67216511960e-02 2.67216511960e-02 2.95749315132e-02 3.31551669202e-02 - 3.48628450776e-02 3.31963581644e-02 2.84217494631e-02 2.22713307355e-02 1.66950035376e-02 1.26511415702e-02 - 1.00210602336e-02 8.36360112567e-03 7.50451749096e-03 7.50451749096e-03 8.36360112567e-03 1.00210602336e-02 - 1.26511415702e-02 1.66950035376e-02 2.22713307355e-02 2.84217494631e-02 3.31963581644e-02 3.48628450776e-02 - 3.05273339684e-02 2.67216511959e-02 2.50661805907e-02 2.67216511959e-02 3.05273339684e-02 3.39196724265e-02 - 3.45751625707e-02 3.16057035862e-02 2.60492747623e-02 2.00583037371e-02 1.52718999852e-02 1.20439356208e-02 - 9.96234500643e-03 8.68876908730e-03 8.23785813083e-03 8.68876908730e-03 9.96234500644e-03 1.20439356208e-02 - 1.52718999852e-02 2.00583037371e-02 2.60492747623e-02 3.16057035862e-02 3.45751625707e-02 3.39196724265e-02 - 2.92890570902e-02 2.67216511959e-02 2.67216511959e-02 2.92890570902e-02 3.26920605346e-02 3.46096179590e-02 - 3.33647463123e-02 2.89590113766e-02 2.31231583381e-02 1.78657504093e-02 1.40901390285e-02 1.16029685725e-02 - 9.96167567605e-03 9.07031679536e-03 9.07031679536e-03 9.96167567605e-03 1.16029685725e-02 1.40901390285e-02 - 1.78657504093e-02 2.31231583381e-02 2.89590113766e-02 3.33647463123e-02 3.46096179590e-02 3.26920605346e-02 - 3.05273339684e-02 2.95749315132e-02 3.05273339684e-02 3.26920605346e-02 3.44216532953e-02 3.39942859271e-02 - 3.06728014251e-02 2.54054430258e-02 2.00653973118e-02 1.59287126341e-02 1.31234388541e-02 1.12137577955e-02 - 9.96234500644e-03 9.50248837487e-03 9.96234500644e-03 1.12137577955e-02 1.31234388541e-02 1.59287126341e-02 - 2.00653973118e-02 2.54054430258e-02 3.06728014251e-02 3.39942859271e-02 3.44216532954e-02 3.26920605346e-02 - 3.31551669202e-02 3.31551669201e-02 3.39196724264e-02 3.46096179590e-02 3.39942859271e-02 3.12134113093e-02 - 2.65858530209e-02 2.15038028103e-02 1.72812015630e-02 1.43188164440e-02 1.22934163580e-02 1.08500532301e-02 - 1.00210602336e-02 1.00210602336e-02 1.08500532301e-02 1.22934163580e-02 1.43188164440e-02 1.72812015630e-02 - 2.15038028103e-02 2.65858530209e-02 3.12134113093e-02 3.39942859271e-02 3.46096179590e-02 3.39196724265e-02 - 3.48683159917e-02 3.48628450776e-02 3.45751625707e-02 3.33647463123e-02 3.06728014251e-02 2.65858530209e-02 - 2.19947599998e-02 1.79741436761e-02 1.50424663757e-02 1.30485807277e-02 1.16095381953e-02 1.06051613742e-02 - 1.02248872526e-02 1.06051613742e-02 1.16095381953e-02 1.30485807277e-02 1.50424663757e-02 1.79741436761e-02 - 2.19947599998e-02 2.65858530209e-02 3.06728014251e-02 3.33647463123e-02 3.45751625707e-02 3.48628450776e-02 - 3.38925854809e-02 3.31963581643e-02 3.16057035862e-02 2.89590113766e-02 2.54054430258e-02 2.15038028103e-02 - 1.79741436761e-02 1.52762540169e-02 1.34169388324e-02 1.21090534529e-02 1.11331987338e-02 1.05557815436e-02 - 1.05557815436e-02 1.11331987338e-02 1.21090534529e-02 1.34169388324e-02 1.52762540169e-02 1.79741436761e-02 - 2.15038028103e-02 2.54054430258e-02 2.89590113766e-02 3.16057035862e-02 3.31963581644e-02 3.38925854809e-02 - 2.99301385615e-02 2.84217494631e-02 2.60492747622e-02 2.31231583381e-02 2.00653973118e-02 1.72812015630e-02 - 1.50424663757e-02 1.34169388324e-02 1.22800512627e-02 1.14481144732e-02 1.08699825973e-02 1.06518186545e-02 - 1.08699825973e-02 1.14481144732e-02 1.22800512627e-02 1.34169388324e-02 1.50424663757e-02 1.72812015630e-02 - 2.00653973118e-02 2.31231583381e-02 2.60492747622e-02 2.84217494631e-02 2.99301385615e-02 3.04404777485e-02 - 2.40967233573e-02 2.22713307355e-02 2.00583037371e-02 1.78657504093e-02 1.59287126341e-02 1.43188164440e-02 - 1.30485807277e-02 1.21090534529e-02 1.14481144732e-02 1.09990704821e-02 1.07546834176e-02 1.07546834176e-02 - 1.09990704821e-02 1.14481144732e-02 1.21090534529e-02 1.30485807277e-02 1.43188164440e-02 1.59287126341e-02 - 1.78657504093e-02 2.00583037371e-02 2.22713307355e-02 2.40967233573e-02 2.51249889113e-02 2.51249889113e-02 - 1.81617046076e-02 1.66950035376e-02 1.52718999852e-02 1.40901390285e-02 1.31234388541e-02 1.22934163580e-02 - 1.16095381953e-02 1.11331987338e-02 1.08699825973e-02 1.07546834176e-02 1.07233722050e-02 1.07546834176e-02 - 1.08699825973e-02 1.11331987338e-02 1.16095381953e-02 1.22934163580e-02 1.31234388541e-02 1.40901390285e-02 - 1.52718999852e-02 1.66950035376e-02 1.81617046076e-02 1.92840317465e-02 1.97044449599e-02 1.92840317465e-02 - 1.34421324248e-02 1.26511415702e-02 1.20439356208e-02 1.16029685725e-02 1.12137577955e-02 1.08500532301e-02 - 1.06051613742e-02 1.05557815436e-02 1.06518186545e-02 1.07546834176e-02 1.07546834176e-02 1.06518186545e-02 - 1.05557815436e-02 1.06051613742e-02 1.08500532301e-02 1.12137577955e-02 1.16029685725e-02 1.20439356208e-02 - 1.26511415702e-02 1.34421324248e-02 1.42308801751e-02 1.47279181688e-02 1.47279181688e-02 1.42308801751e-02 - 1.02248872526e-02 1.00210602336e-02 9.96234500643e-03 9.96167567605e-03 9.96234500644e-03 1.00210602336e-02 - 1.02248872526e-02 1.05557815436e-02 1.08699825973e-02 1.09990704821e-02 1.08699825973e-02 1.05557815436e-02 - 1.02248872526e-02 1.00210602336e-02 9.96234500644e-03 9.96167567604e-03 9.96234500643e-03 1.00210602336e-02 - 1.02248872526e-02 1.05557815436e-02 1.08699825973e-02 1.09990704821e-02 1.08699825973e-02 1.05557815436e-02 - 8.15758815907e-03 8.36360112567e-03 8.68876908730e-03 9.07031679536e-03 9.50248837488e-03 1.00210602336e-02 - 1.06051613742e-02 1.11331987338e-02 1.14481144732e-02 1.14481144732e-02 1.11331987338e-02 1.06051613742e-02 - 1.00210602336e-02 9.50248837488e-03 9.07031679536e-03 8.68876908730e-03 8.36360112566e-03 8.15758815907e-03 - 8.11230157084e-03 8.18383164206e-03 8.26409670065e-03 8.26409670065e-03 8.18383164206e-03 8.11230157084e-03 - 6.92803651719e-03 7.50451749096e-03 8.23785813083e-03 9.07031679536e-03 9.96234500644e-03 1.08500532301e-02 - 1.16095381953e-02 1.21090534529e-02 1.22800512627e-02 1.21090534529e-02 1.16095381953e-02 1.08500532301e-02 - 9.96234500644e-03 9.07031679536e-03 8.23785813083e-03 7.50451749096e-03 6.92803651719e-03 6.56138198056e-03 - 6.39967919814e-03 6.36714430779e-03 6.36996915701e-03 6.36714430779e-03 6.39967919814e-03 6.56138198056e-03 - 6.50338402317e-03 7.50451749096e-03 8.68876908731e-03 9.96167567605e-03 1.12137577955e-02 1.22934163580e-02 - 1.30485807277e-02 1.34169388324e-02 1.34169388324e-02 1.30485807277e-02 1.22934163580e-02 1.12137577955e-02 - 9.96167567605e-03 8.68876908730e-03 7.50451749096e-03 6.50338402317e-03 5.76667312114e-03 5.32068131556e-03 - 5.11303460195e-03 5.04614391795e-03 5.04614391795e-03 5.11303460194e-03 5.32068131556e-03 5.76667312114e-03 - 6.92803651720e-03 8.36360112567e-03 9.96234500644e-03 1.16029685725e-02 1.31234388541e-02 1.43188164440e-02 - 1.50424663757e-02 1.52762540169e-02 1.50424663757e-02 1.43188164440e-02 1.31234388541e-02 1.16029685725e-02 - 9.96234500644e-03 8.36360112567e-03 6.92803651720e-03 5.76667312114e-03 4.94632325183e-03 4.45367296569e-03 - 4.21584246473e-03 4.14918646508e-03 4.21584246472e-03 4.45367296569e-03 4.94632325183e-03 5.76667312114e-03 - 8.15758815908e-03 1.00210602336e-02 1.20439356208e-02 1.40901390285e-02 1.59287126341e-02 1.72812015630e-02 - 1.79741436761e-02 1.79741436761e-02 1.72812015630e-02 1.59287126341e-02 1.40901390285e-02 1.20439356208e-02 - 1.00210602336e-02 8.15758815907e-03 6.56138198057e-03 5.32068131556e-03 4.45367296569e-03 3.92281290681e-03 - 3.67768921534e-03 3.67768921534e-03 3.92281290681e-03 4.45367296569e-03 5.32068131556e-03 6.56138198057e-03 - 1.02248872526e-02 1.26511415702e-02 1.52718999852e-02 1.78657504093e-02 2.00653973119e-02 2.15038028103e-02 - 2.19947599998e-02 2.15038028103e-02 2.00653973119e-02 1.78657504093e-02 1.52718999852e-02 1.26511415702e-02 - 1.02248872526e-02 8.11230157084e-03 6.39967919814e-03 5.11303460195e-03 4.21584246472e-03 3.67768921534e-03 - 3.49726793106e-03 3.67768921534e-03 4.21584246472e-03 5.11303460195e-03 6.39967919814e-03 8.11230157084e-03 - 1.34421324248e-02 1.66950035376e-02 2.00583037371e-02 2.31231583381e-02 2.54054430258e-02 2.65858530209e-02 - 2.65858530209e-02 2.54054430258e-02 2.31231583381e-02 2.00583037371e-02 1.66950035376e-02 1.34421324248e-02 - 1.05557815436e-02 8.18383164206e-03 6.36714430779e-03 5.04614391795e-03 4.14918646508e-03 3.67768921534e-03 - 3.67768921534e-03 4.14918646508e-03 5.04614391795e-03 6.36714430779e-03 8.18383164206e-03 1.05557815436e-02 - 1.81617046076e-02 2.22713307355e-02 2.60492747623e-02 2.89590113767e-02 3.06728014251e-02 3.12134113093e-02 - 3.06728014251e-02 2.89590113767e-02 2.60492747623e-02 2.22713307355e-02 1.81617046076e-02 1.42308801751e-02 - 1.08699825973e-02 8.26409670065e-03 6.36996915701e-03 5.04614391795e-03 4.21584246473e-03 3.92281290681e-03 - 4.21584246473e-03 5.04614391795e-03 6.36996915701e-03 8.26409670065e-03 1.08699825973e-02 1.42308801751e-02 - 2.40967233573e-02 2.84217494631e-02 3.16057035862e-02 3.33647463124e-02 3.39942859271e-02 3.39942859271e-02 - 3.33647463124e-02 3.16057035862e-02 2.84217494631e-02 2.40967233573e-02 1.92840317465e-02 1.47279181688e-02 - 1.09990704821e-02 8.26409670065e-03 6.36714430779e-03 5.11303460195e-03 4.45367296569e-03 4.45367296569e-03 - 5.11303460195e-03 6.36714430780e-03 8.26409670065e-03 1.09990704821e-02 1.47279181688e-02 1.92840317465e-02 - 2.99301385615e-02 3.31963581644e-02 3.45751625707e-02 3.46096179590e-02 3.44216532954e-02 3.46096179590e-02 - 3.45751625707e-02 3.31963581644e-02 2.99301385615e-02 2.51249889113e-02 1.97044449599e-02 1.47279181688e-02 - 1.08699825973e-02 8.18383164206e-03 6.39967919814e-03 5.32068131556e-03 4.94632325183e-03 5.32068131556e-03 - 6.39967919814e-03 8.18383164206e-03 1.08699825973e-02 1.47279181688e-02 1.97044449599e-02 2.51249889113e-02 - 3.38925854809e-02 3.48628450776e-02 3.39196724265e-02 3.26920605347e-02 3.26920605347e-02 3.39196724265e-02 - 3.48628450776e-02 3.38925854809e-02 3.04404777485e-02 2.51249889113e-02 1.92840317465e-02 1.42308801751e-02 - 1.05557815436e-02 8.11230157084e-03 6.56138198057e-03 5.76667312114e-03 5.76667312114e-03 6.56138198057e-03 - 8.11230157085e-03 1.05557815436e-02 1.42308801751e-02 1.92840317465e-02 2.51249889113e-02 3.04404777485e-02 - 4.15726567626e-02 4.23899857520e-02 4.26148361403e-02 4.26148361403e-02 4.23899857520e-02 4.15726567626e-02 - 3.97328533270e-02 3.62673468746e-02 3.07822057975e-02 2.40904203272e-02 1.79266526907e-02 1.34421324248e-02 - 1.06051613742e-02 8.89920278516e-03 8.04455907974e-03 8.04455907974e-03 8.89920278516e-03 1.06051613742e-02 - 1.34421324248e-02 1.79266526907e-02 2.40904203272e-02 3.07822057975e-02 3.62673468746e-02 3.97328533271e-02 - 4.23899857520e-02 4.30330806506e-02 4.31931066266e-02 4.30330806506e-02 4.23899857519e-02 4.11388919901e-02 - 3.89798518705e-02 3.49769025117e-02 2.88778389824e-02 2.21709602700e-02 1.66950035376e-02 1.30618774824e-02 - 1.08500532301e-02 9.57619631350e-03 9.13724506496e-03 9.57619631350e-03 1.08500532301e-02 1.30618774824e-02 - 1.66950035376e-02 2.21709602700e-02 2.88778389824e-02 3.49769025118e-02 3.89798518706e-02 4.11388919901e-02 - 4.26148361403e-02 4.31931066266e-02 4.31931066266e-02 4.26148361403e-02 4.15214050146e-02 4.00842207789e-02 - 3.75326171916e-02 3.27029610793e-02 2.61902387743e-02 2.00583037371e-02 1.56489970468e-02 1.28937939562e-02 - 1.12137577955e-02 1.03422447642e-02 1.03422447642e-02 1.12137577955e-02 1.28937939562e-02 1.56489970468e-02 - 2.00583037371e-02 2.61902387743e-02 3.27029610793e-02 3.75326171916e-02 4.00842207789e-02 4.15214050146e-02 - 4.26148361403e-02 4.30330806505e-02 4.26148361403e-02 4.15474983448e-02 4.03062291682e-02 3.86294987685e-02 - 3.51262182184e-02 2.93979324046e-02 2.31231583381e-02 1.81430425204e-02 1.48932100517e-02 1.28580577518e-02 - 1.16029685725e-02 1.11481007476e-02 1.16029685725e-02 1.28580577518e-02 1.48932100517e-02 1.81430425204e-02 - 2.31231583381e-02 2.93979324046e-02 3.51262182184e-02 3.86294987685e-02 4.03062291682e-02 4.15474983449e-02 - 4.23899857519e-02 4.23899857519e-02 4.15214050146e-02 4.03062291682e-02 3.88811262511e-02 3.61935553818e-02 - 3.13656133727e-02 2.54054430258e-02 2.02040886880e-02 1.66389339973e-02 1.43894839458e-02 1.28937939562e-02 - 1.20439356208e-02 1.20439356208e-02 1.28937939562e-02 1.43894839458e-02 1.66389339973e-02 2.02040886880e-02 - 2.54054430258e-02 3.13656133727e-02 3.61935553818e-02 3.88811262511e-02 4.03062291682e-02 4.15214050146e-02 - 4.15726567626e-02 4.11388919901e-02 4.00842207789e-02 3.86294987685e-02 3.61935553818e-02 3.20070729422e-02 - 2.65858530209e-02 2.15195877577e-02 1.78888288560e-02 1.56224022211e-02 1.41149739580e-02 1.30618774824e-02 - 1.26511415702e-02 1.30618774824e-02 1.41149739580e-02 1.56224022211e-02 1.78888288560e-02 2.15195877577e-02 - 2.65858530209e-02 3.20070729422e-02 3.61935553818e-02 3.86294987685e-02 4.00842207789e-02 4.11388919901e-02 - 3.97328533270e-02 3.89798518705e-02 3.75326171916e-02 3.51262182184e-02 3.13656133727e-02 2.65858530209e-02 - 2.19611077293e-02 1.85132763207e-02 1.63906429721e-02 1.50739196267e-02 1.40806383265e-02 1.34421324248e-02 - 1.34421324248e-02 1.40806383265e-02 1.50739196267e-02 1.63906429721e-02 1.85132763207e-02 2.19611077293e-02 - 2.65858530209e-02 3.13656133727e-02 3.51262182184e-02 3.75326171916e-02 3.89798518705e-02 3.97328533270e-02 - 3.62673468745e-02 3.49769025117e-02 3.27029610793e-02 2.93979324045e-02 2.54054430258e-02 2.15195877577e-02 - 1.85132763207e-02 1.66439398964e-02 1.55937378378e-02 1.48492046219e-02 1.42308801751e-02 1.39667079820e-02 - 1.42308801751e-02 1.48492046219e-02 1.55937378378e-02 1.66439398964e-02 1.85132763207e-02 2.15195877577e-02 - 2.54054430258e-02 2.93979324046e-02 3.27029610793e-02 3.49769025117e-02 3.62673468745e-02 3.66854482599e-02 - 3.07822057975e-02 2.88778389824e-02 2.61902387743e-02 2.31231583381e-02 2.02040886880e-02 1.78888288560e-02 - 1.63906429721e-02 1.55937378378e-02 1.51354763629e-02 1.47279181688e-02 1.44210111226e-02 1.44210111226e-02 - 1.47279181688e-02 1.51354763629e-02 1.55937378378e-02 1.63906429721e-02 1.78888288560e-02 2.02040886880e-02 - 2.31231583381e-02 2.61902387743e-02 2.88778389824e-02 3.07822057975e-02 3.17361436402e-02 3.17361436402e-02 - 2.40904203272e-02 2.21709602700e-02 2.00583037371e-02 1.81430425204e-02 1.66389339973e-02 1.56224022211e-02 - 1.50739196267e-02 1.48492046219e-02 1.47279181688e-02 1.45986463266e-02 1.45320680305e-02 1.45986463266e-02 - 1.47279181688e-02 1.48492046219e-02 1.50739196267e-02 1.56224022211e-02 1.66389339973e-02 1.81430425204e-02 - 2.00583037371e-02 2.21709602700e-02 2.40904203272e-02 2.53973709283e-02 2.58521940319e-02 2.53973709283e-02 - 1.79266526907e-02 1.66950035376e-02 1.56489970468e-02 1.48932100517e-02 1.43894839458e-02 1.41149739580e-02 - 1.40806383265e-02 1.42308801751e-02 1.44210111226e-02 1.45320680305e-02 1.45320680305e-02 1.44210111226e-02 - 1.42308801751e-02 1.40806383265e-02 1.41149739580e-02 1.43894839458e-02 1.48932100517e-02 1.56489970468e-02 - 1.66950035376e-02 1.79266526907e-02 1.90345757835e-02 1.96802339929e-02 1.96802339929e-02 1.90345757835e-02 - 1.34421324248e-02 1.30618774824e-02 1.28937939562e-02 1.28580577518e-02 1.28937939562e-02 1.30618774824e-02 - 1.34421324248e-02 1.39667079820e-02 1.44210111226e-02 1.45986463266e-02 1.44210111226e-02 1.39667079820e-02 - 1.34421324248e-02 1.30618774824e-02 1.28937939562e-02 1.28580577518e-02 1.28937939562e-02 1.30618774824e-02 - 1.34421324248e-02 1.39667079820e-02 1.44210111226e-02 1.45986463266e-02 1.44210111226e-02 1.39667079820e-02 - 1.06051613742e-02 1.08500532301e-02 1.12137577955e-02 1.16029685725e-02 1.20439356208e-02 1.26511415702e-02 - 1.34421324248e-02 1.42308801751e-02 1.47279181688e-02 1.47279181688e-02 1.42308801751e-02 1.34421324248e-02 - 1.26511415702e-02 1.20439356208e-02 1.16029685725e-02 1.12137577955e-02 1.08500532301e-02 1.06051613742e-02 - 1.05557815436e-02 1.06518186545e-02 1.07546834175e-02 1.07546834176e-02 1.06518186545e-02 1.05557815436e-02 - 8.89920278515e-03 9.57619631350e-03 1.03422447642e-02 1.11481007476e-02 1.20439356208e-02 1.30618774824e-02 - 1.40806383265e-02 1.48492046219e-02 1.51354763629e-02 1.48492046219e-02 1.40806383265e-02 1.30618774824e-02 - 1.20439356208e-02 1.11481007476e-02 1.03422447642e-02 9.57619631350e-03 8.89920278515e-03 8.41997198038e-03 - 8.18383164206e-03 8.12161313174e-03 8.11874591476e-03 8.12161313174e-03 8.18383164206e-03 8.41997198038e-03 - 8.04455907974e-03 9.13724506496e-03 1.03422447642e-02 1.16029685725e-02 1.28937939562e-02 1.41149739580e-02 - 1.50739196267e-02 1.55937378378e-02 1.55937378378e-02 1.50739196267e-02 1.41149739580e-02 1.28937939562e-02 - 1.16029685725e-02 1.03422447642e-02 9.13724506496e-03 8.04455907974e-03 7.17862759619e-03 6.62684043801e-03 - 6.36714430779e-03 6.28719817634e-03 6.28719817634e-03 6.36714430779e-03 6.62684043801e-03 7.17862759619e-03 - 8.04455907974e-03 9.57619631350e-03 1.12137577955e-02 1.28580577518e-02 1.43894839458e-02 1.56224022211e-02 - 1.63906429721e-02 1.66439398964e-02 1.63906429721e-02 1.56224022211e-02 1.43894839458e-02 1.28580577518e-02 - 1.12137577955e-02 9.57619631350e-03 8.04455907974e-03 6.75449318420e-03 5.82746138702e-03 5.28636096575e-03 - 5.04614391795e-03 4.98513742725e-03 5.04614391795e-03 5.28636096575e-03 5.82746138702e-03 6.75449318420e-03 - 8.89920278516e-03 1.08500532301e-02 1.28937939562e-02 1.48932100517e-02 1.66389339974e-02 1.78888288560e-02 - 1.85132763207e-02 1.85132763207e-02 1.78888288560e-02 1.66389339974e-02 1.48932100517e-02 1.28937939562e-02 - 1.08500532301e-02 8.89920278516e-03 7.17862759619e-03 5.82746138702e-03 4.90588959931e-03 4.37599558137e-03 - 4.14918646508e-03 4.14918646508e-03 4.37599558137e-03 4.90588959931e-03 5.82746138702e-03 7.17862759619e-03 - 1.06051613742e-02 1.30618774824e-02 1.56489970468e-02 1.81430425205e-02 2.02040886880e-02 2.15195877577e-02 - 2.19611077293e-02 2.15195877577e-02 2.02040886880e-02 1.81430425205e-02 1.56489970468e-02 1.30618774824e-02 - 1.06051613742e-02 8.41997198039e-03 6.62684043801e-03 5.28636096575e-03 4.37599558137e-03 3.84964089400e-03 - 3.67768921534e-03 3.84964089400e-03 4.37599558137e-03 5.28636096575e-03 6.62684043802e-03 8.41997198039e-03 - 1.34421324248e-02 1.66950035376e-02 2.00583037371e-02 2.31231583381e-02 2.54054430258e-02 2.65858530209e-02 - 2.65858530209e-02 2.54054430258e-02 2.31231583381e-02 2.00583037371e-02 1.66950035376e-02 1.34421324248e-02 - 1.05557815436e-02 8.18383164206e-03 6.36714430779e-03 5.04614391795e-03 4.14918646508e-03 3.67768921534e-03 - 3.67768921534e-03 4.14918646508e-03 5.04614391795e-03 6.36714430779e-03 8.18383164206e-03 1.05557815436e-02 - 1.79266526907e-02 2.21709602700e-02 2.61902387743e-02 2.93979324046e-02 3.13656133728e-02 3.20070729423e-02 - 3.13656133728e-02 2.93979324046e-02 2.61902387743e-02 2.21709602700e-02 1.79266526907e-02 1.39667079820e-02 - 1.06518186545e-02 8.12161313174e-03 6.28719817635e-03 4.98513742725e-03 4.14918646508e-03 3.84964089400e-03 - 4.14918646508e-03 4.98513742725e-03 6.28719817635e-03 8.12161313175e-03 1.06518186545e-02 1.39667079820e-02 - 2.40904203272e-02 2.88778389824e-02 3.27029610793e-02 3.51262182185e-02 3.61935553818e-02 3.61935553818e-02 - 3.51262182185e-02 3.27029610793e-02 2.88778389824e-02 2.40904203272e-02 1.90345757835e-02 1.44210111226e-02 - 1.07546834176e-02 8.11874591477e-03 6.28719817635e-03 5.04614391795e-03 4.37599558137e-03 4.37599558137e-03 - 5.04614391795e-03 6.28719817635e-03 8.11874591477e-03 1.07546834176e-02 1.44210111226e-02 1.90345757835e-02 - 3.07822057975e-02 3.49769025118e-02 3.75326171916e-02 3.86294987685e-02 3.88811262512e-02 3.86294987685e-02 - 3.75326171916e-02 3.49769025118e-02 3.07822057975e-02 2.53973709283e-02 1.96802339930e-02 1.45986463266e-02 - 1.07546834176e-02 8.12161313175e-03 6.36714430780e-03 5.28636096575e-03 4.90588959931e-03 5.28636096575e-03 - 6.36714430780e-03 8.12161313175e-03 1.07546834176e-02 1.45986463266e-02 1.96802339930e-02 2.53973709283e-02 - 3.62673468746e-02 3.89798518706e-02 4.00842207790e-02 4.03062291683e-02 4.03062291683e-02 4.00842207790e-02 - 3.89798518706e-02 3.62673468746e-02 3.17361436403e-02 2.58521940319e-02 1.96802339930e-02 1.44210111226e-02 - 1.06518186545e-02 8.18383164206e-03 6.62684043801e-03 5.82746138702e-03 5.82746138702e-03 6.62684043802e-03 - 8.18383164206e-03 1.06518186545e-02 1.44210111226e-02 1.96802339930e-02 2.58521940319e-02 3.17361436403e-02 - 3.97328533271e-02 4.11388919902e-02 4.15214050146e-02 4.15474983449e-02 4.15214050146e-02 4.11388919902e-02 - 3.97328533271e-02 3.66854482599e-02 3.17361436402e-02 2.53973709283e-02 1.90345757835e-02 1.39667079820e-02 - 1.05557815436e-02 8.41997198039e-03 7.17862759619e-03 6.75449318420e-03 7.17862759619e-03 8.41997198039e-03 - 1.05557815436e-02 1.39667079820e-02 1.90345757835e-02 2.53973709283e-02 3.17361436403e-02 3.66854482599e-02 - 6.13624406409e-02 6.87950359263e-02 7.14595557376e-02 6.87950359263e-02 6.13624406409e-02 5.18158087397e-02 - 4.36605690195e-02 3.73540502296e-02 3.09609358381e-02 2.40904203272e-02 1.81617046076e-02 1.40806383265e-02 - 1.16095381953e-02 1.02401100733e-02 9.78219490488e-03 1.02401100733e-02 1.16095381953e-02 1.40806383265e-02 - 1.81617046076e-02 2.40904203272e-02 3.09609358381e-02 3.73540502297e-02 4.36605690195e-02 5.18158087398e-02 - 6.87950359263e-02 7.44578962493e-02 7.44578962493e-02 6.87950359263e-02 5.87809730550e-02 4.86717503238e-02 - 4.14583134664e-02 3.55406204185e-02 2.88778389824e-02 2.22713307355e-02 1.72611853896e-02 1.41149739580e-02 - 1.22934163580e-02 1.14031750108e-02 1.14031750108e-02 1.22934163580e-02 1.41149739580e-02 1.72611853896e-02 - 2.22713307355e-02 2.88778389824e-02 3.55406204185e-02 4.14583134664e-02 4.86717503238e-02 5.87809730550e-02 - 7.14595557376e-02 7.44578962493e-02 7.14595557376e-02 6.30064198496e-02 5.24371687843e-02 4.42637285519e-02 - 3.86293456568e-02 3.27029610793e-02 2.60492747623e-02 2.03920879638e-02 1.66276099946e-02 1.43894839458e-02 - 1.31234388541e-02 1.26855285494e-02 1.31234388541e-02 1.43894839458e-02 1.66276099946e-02 2.03920879638e-02 - 2.60492747623e-02 3.27029610793e-02 3.86293456568e-02 4.42637285519e-02 5.24371687843e-02 6.30064198496e-02 - 6.87950359263e-02 6.87950359263e-02 6.30064198496e-02 5.38553081705e-02 4.56987320892e-02 4.02266025417e-02 - 3.51262182184e-02 2.89590113766e-02 2.30939042849e-02 1.89263764236e-02 1.64062528191e-02 1.48932100517e-02 - 1.40901390285e-02 1.40901390285e-02 1.48932100517e-02 1.64062528191e-02 1.89263764236e-02 2.30939042849e-02 - 2.89590113766e-02 3.51262182185e-02 4.02266025417e-02 4.56987320892e-02 5.38553081705e-02 6.30064198496e-02 - 6.13624406409e-02 5.87809730549e-02 5.24371687843e-02 4.56987320892e-02 4.06883718959e-02 3.61935553818e-02 - 3.06728014251e-02 2.49902463304e-02 2.07196781889e-02 1.81502966987e-02 1.66276099946e-02 1.56489970468e-02 - 1.52718999852e-02 1.56489970468e-02 1.66276099946e-02 1.81502966987e-02 2.07196781889e-02 2.49902463304e-02 - 3.06728014251e-02 3.61935553818e-02 4.06883718959e-02 4.56987320892e-02 5.24371687843e-02 5.87809730550e-02 - 5.18158087397e-02 4.86717503238e-02 4.42637285519e-02 4.02266025417e-02 3.61935553818e-02 3.12134113093e-02 - 2.59181147281e-02 2.17948830411e-02 1.93871115228e-02 1.81179302207e-02 1.72611853896e-02 1.66950035376e-02 - 1.66950035376e-02 1.72611853896e-02 1.81179302207e-02 1.93871115228e-02 2.17948830411e-02 2.59181147281e-02 - 3.12134113093e-02 3.61935553818e-02 4.02266025417e-02 4.42637285519e-02 4.86717503238e-02 5.18158087397e-02 - 4.36605690195e-02 4.14583134664e-02 3.86293456568e-02 3.51262182184e-02 3.06728014251e-02 2.59181147281e-02 - 2.21386056500e-02 1.99975947255e-02 1.91106081253e-02 1.86355176327e-02 1.81617046076e-02 1.79266526907e-02 - 1.81617046076e-02 1.86355176327e-02 1.91106081253e-02 1.99975947255e-02 2.21386056500e-02 2.59181147281e-02 - 3.06728014251e-02 3.51262182184e-02 3.86293456568e-02 4.14583134664e-02 4.36605690195e-02 4.45490987357e-02 - 3.73540502296e-02 3.55406204184e-02 3.27029610793e-02 2.89590113766e-02 2.49902463304e-02 2.17948830411e-02 - 1.99975947255e-02 1.94536795471e-02 1.94290489291e-02 1.92840317465e-02 1.90345757835e-02 1.90345757835e-02 - 1.92840317465e-02 1.94290489291e-02 1.94536795471e-02 1.99975947255e-02 2.17948830411e-02 2.49902463304e-02 - 2.89590113766e-02 3.27029610793e-02 3.55406204185e-02 3.73540502296e-02 3.82423088607e-02 3.82423088607e-02 - 3.09609358381e-02 2.88778389824e-02 2.60492747622e-02 2.30939042849e-02 2.07196781889e-02 1.93871115228e-02 - 1.91106081253e-02 1.94290489291e-02 1.97044449599e-02 1.96802339930e-02 1.96031535282e-02 1.96802339930e-02 - 1.97044449599e-02 1.94290489291e-02 1.91106081253e-02 1.93871115228e-02 2.07196781889e-02 2.30939042849e-02 - 2.60492747623e-02 2.88778389824e-02 3.09609358381e-02 3.20848019958e-02 3.24128421387e-02 3.20848019958e-02 - 2.40904203272e-02 2.22713307355e-02 2.03920879637e-02 1.89263764236e-02 1.81502966987e-02 1.81179302207e-02 - 1.86355176327e-02 1.92840317465e-02 1.96802339930e-02 1.97871610956e-02 1.97871610956e-02 1.96802339930e-02 - 1.92840317465e-02 1.86355176327e-02 1.81179302207e-02 1.81502966987e-02 1.89263764236e-02 2.03920879638e-02 - 2.22713307355e-02 2.40904203272e-02 2.53797374387e-02 2.59777844449e-02 2.59777844449e-02 2.53797374387e-02 - 1.81617046076e-02 1.72611853896e-02 1.66276099946e-02 1.64062528191e-02 1.66276099946e-02 1.72611853896e-02 - 1.81617046076e-02 1.90345757835e-02 1.96031535282e-02 1.97871610956e-02 1.96031535282e-02 1.90345757835e-02 - 1.81617046076e-02 1.72611853896e-02 1.66276099946e-02 1.64062528191e-02 1.66276099946e-02 1.72611853896e-02 - 1.81617046076e-02 1.90345757835e-02 1.96031535282e-02 1.97871610956e-02 1.96031535282e-02 1.90345757835e-02 - 1.40806383265e-02 1.41149739580e-02 1.43894839458e-02 1.48932100517e-02 1.56489970468e-02 1.66950035376e-02 - 1.79266526907e-02 1.90345757835e-02 1.96802339930e-02 1.96802339930e-02 1.90345757835e-02 1.79266526907e-02 - 1.66950035376e-02 1.56489970468e-02 1.48932100517e-02 1.43894839458e-02 1.41149739580e-02 1.40806383265e-02 - 1.42308801751e-02 1.44210111226e-02 1.45320680305e-02 1.45320680305e-02 1.44210111226e-02 1.42308801751e-02 - 1.16095381953e-02 1.22934163580e-02 1.31234388541e-02 1.40901390285e-02 1.52718999852e-02 1.66950035376e-02 - 1.81617046076e-02 1.92840317465e-02 1.97044449599e-02 1.92840317465e-02 1.81617046076e-02 1.66950035376e-02 - 1.52718999852e-02 1.40901390285e-02 1.31234388541e-02 1.22934163580e-02 1.16095381953e-02 1.11331987338e-02 - 1.08699825973e-02 1.07546834176e-02 1.07233722050e-02 1.07546834176e-02 1.08699825973e-02 1.11331987338e-02 - 1.02401100733e-02 1.14031750108e-02 1.26855285494e-02 1.40901390285e-02 1.56489970468e-02 1.72611853896e-02 - 1.86355176327e-02 1.94290489291e-02 1.94290489291e-02 1.86355176327e-02 1.72611853896e-02 1.56489970468e-02 - 1.40901390285e-02 1.26855285494e-02 1.14031750108e-02 1.02401100733e-02 9.28357359922e-03 8.62540500905e-03 - 8.26409670065e-03 8.11874591476e-03 8.11874591476e-03 8.26409670065e-03 8.62540500905e-03 9.28357359922e-03 - 9.78219490488e-03 1.14031750108e-02 1.31234388541e-02 1.48932100517e-02 1.66276099946e-02 1.81179302207e-02 - 1.91106081254e-02 1.94536795471e-02 1.91106081254e-02 1.81179302207e-02 1.66276099946e-02 1.48932100517e-02 - 1.31234388541e-02 1.14031750108e-02 9.78219490488e-03 8.37400390520e-03 7.31933010296e-03 6.67439308888e-03 - 6.36996915701e-03 6.28719817634e-03 6.36996915701e-03 6.67439308888e-03 7.31933010296e-03 8.37400390520e-03 - 1.02401100733e-02 1.22934163580e-02 1.43894839458e-02 1.64062528191e-02 1.81502966987e-02 1.93871115228e-02 - 1.99975947255e-02 1.99975947255e-02 1.93871115228e-02 1.81502966987e-02 1.64062528191e-02 1.43894839458e-02 - 1.22934163580e-02 1.02401100733e-02 8.37400390520e-03 6.87426464911e-03 5.85076317272e-03 5.27978060753e-03 - 5.04614391795e-03 5.04614391795e-03 5.27978060753e-03 5.85076317272e-03 6.87426464911e-03 8.37400390520e-03 - 1.16095381953e-02 1.41149739580e-02 1.66276099946e-02 1.89263764236e-02 2.07196781889e-02 2.17948830411e-02 - 2.21386056500e-02 2.17948830411e-02 2.07196781889e-02 1.89263764236e-02 1.66276099946e-02 1.41149739580e-02 - 1.16095381953e-02 9.28357359923e-03 7.31933010296e-03 5.85076317272e-03 4.89318523683e-03 4.37599558137e-03 - 4.21584246472e-03 4.37599558137e-03 4.89318523683e-03 5.85076317272e-03 7.31933010297e-03 9.28357359923e-03 - 1.40806383265e-02 1.72611853896e-02 2.03920879638e-02 2.30939042850e-02 2.49902463304e-02 2.59181147281e-02 - 2.59181147281e-02 2.49902463304e-02 2.30939042850e-02 2.03920879638e-02 1.72611853896e-02 1.40806383265e-02 - 1.11331987338e-02 8.62540500905e-03 6.67439308888e-03 5.27978060753e-03 4.37599558137e-03 3.92281290681e-03 - 3.92281290681e-03 4.37599558137e-03 5.27978060753e-03 6.67439308889e-03 8.62540500906e-03 1.11331987338e-02 - 1.81617046076e-02 2.22713307355e-02 2.60492747623e-02 2.89590113767e-02 3.06728014251e-02 3.12134113093e-02 - 3.06728014251e-02 2.89590113767e-02 2.60492747623e-02 2.22713307355e-02 1.81617046076e-02 1.42308801751e-02 - 1.08699825973e-02 8.26409670065e-03 6.36996915701e-03 5.04614391795e-03 4.21584246473e-03 3.92281290681e-03 - 4.21584246473e-03 5.04614391795e-03 6.36996915701e-03 8.26409670066e-03 1.08699825973e-02 1.42308801751e-02 - 2.40904203272e-02 2.88778389824e-02 3.27029610793e-02 3.51262182185e-02 3.61935553818e-02 3.61935553818e-02 - 3.51262182185e-02 3.27029610793e-02 2.88778389824e-02 2.40904203272e-02 1.90345757835e-02 1.44210111226e-02 - 1.07546834176e-02 8.11874591477e-03 6.28719817635e-03 5.04614391795e-03 4.37599558137e-03 4.37599558137e-03 - 5.04614391795e-03 6.28719817635e-03 8.11874591477e-03 1.07546834176e-02 1.44210111226e-02 1.90345757835e-02 - 3.09609358382e-02 3.55406204185e-02 3.86293456568e-02 4.02266025417e-02 4.06883718959e-02 4.02266025417e-02 - 3.86293456568e-02 3.55406204185e-02 3.09609358381e-02 2.53797374387e-02 1.96031535283e-02 1.45320680305e-02 - 1.07233722050e-02 8.11874591477e-03 6.36996915701e-03 5.27978060753e-03 4.89318523684e-03 5.27978060753e-03 - 6.36996915701e-03 8.11874591477e-03 1.07233722050e-02 1.45320680305e-02 1.96031535283e-02 2.53797374387e-02 - 3.73540502297e-02 4.14583134665e-02 4.42637285519e-02 4.56987320892e-02 4.56987320892e-02 4.42637285519e-02 - 4.14583134665e-02 3.73540502297e-02 3.20848019959e-02 2.59777844449e-02 1.97871610956e-02 1.45320680305e-02 - 1.07546834176e-02 8.26409670065e-03 6.67439308889e-03 5.85076317272e-03 5.85076317272e-03 6.67439308889e-03 - 8.26409670066e-03 1.07546834176e-02 1.45320680305e-02 1.97871610956e-02 2.59777844449e-02 3.20848019959e-02 - 4.36605690195e-02 4.86717503239e-02 5.24371687844e-02 5.38553081705e-02 5.24371687844e-02 4.86717503239e-02 - 4.36605690195e-02 3.82423088608e-02 3.24128421387e-02 2.59777844449e-02 1.96031535282e-02 1.44210111226e-02 - 1.08699825973e-02 8.62540500906e-03 7.31933010296e-03 6.87426464911e-03 7.31933010296e-03 8.62540500906e-03 - 1.08699825973e-02 1.44210111226e-02 1.96031535282e-02 2.59777844449e-02 3.24128421387e-02 3.82423088608e-02 - 5.18158087398e-02 5.87809730550e-02 6.30064198497e-02 6.30064198497e-02 5.87809730550e-02 5.18158087398e-02 - 4.45490987358e-02 3.82423088608e-02 3.20848019959e-02 2.53797374387e-02 1.90345757835e-02 1.42308801751e-02 - 1.11331987338e-02 9.28357359923e-03 8.37400390520e-03 8.37400390520e-03 9.28357359923e-03 1.11331987338e-02 - 1.42308801751e-02 1.90345757835e-02 2.53797374387e-02 3.20848019959e-02 3.82423088608e-02 4.45490987358e-02 - 9.46618817554e-02 1.02255317231e-01 1.02255317231e-01 9.46618817553e-02 7.81193802830e-02 5.87902882083e-02 - 4.51578187598e-02 3.73540502296e-02 3.07822057975e-02 2.40967233573e-02 1.86355176327e-02 1.50739196267e-02 - 1.30485807277e-02 1.21045128275e-02 1.21045128275e-02 1.30485807277e-02 1.50739196267e-02 1.86355176327e-02 - 2.40967233573e-02 3.07822057975e-02 3.73540502297e-02 4.51578187598e-02 5.87902882083e-02 7.81193802830e-02 - 1.02255317231e-01 1.05382836884e-01 1.02255317231e-01 9.00454509227e-02 6.98255714909e-02 5.17382028008e-02 - 4.14583134664e-02 3.49769025117e-02 2.84217494631e-02 2.23752455712e-02 1.81179302207e-02 1.56224022211e-02 - 1.43188164440e-02 1.38961516945e-02 1.43188164440e-02 1.56224022211e-02 1.81179302207e-02 2.23752455712e-02 - 2.84217494631e-02 3.49769025118e-02 4.14583134664e-02 5.17382028009e-02 6.98255714909e-02 9.00454509227e-02 - 1.02255317231e-01 1.02255317231e-01 9.40538638176e-02 7.65071402184e-02 5.69437593911e-02 4.42637285519e-02 - 3.75326171916e-02 3.16057035862e-02 2.55726183949e-02 2.09446754967e-02 1.81502966987e-02 1.66389339974e-02 - 1.59287126341e-02 1.59287126341e-02 1.66389339973e-02 1.81502966987e-02 2.09446754967e-02 2.55726183949e-02 - 3.16057035862e-02 3.75326171916e-02 4.42637285519e-02 5.69437593911e-02 7.65071402185e-02 9.40538638177e-02 - 9.46618817553e-02 9.00454509226e-02 7.65071402184e-02 5.89461275928e-02 4.56987320892e-02 3.86294987685e-02 - 3.33647463123e-02 2.78013937268e-02 2.32030440049e-02 2.03983415996e-02 1.89263764236e-02 1.81430425204e-02 - 1.78657504093e-02 1.81430425204e-02 1.89263764236e-02 2.03983415996e-02 2.32030440049e-02 2.78013937268e-02 - 3.33647463124e-02 3.86294987685e-02 4.56987320892e-02 5.89461275929e-02 7.65071402185e-02 9.00454509227e-02 - 7.81193802830e-02 6.98255714909e-02 5.69437593911e-02 4.56987320892e-02 3.88811262511e-02 3.39942859271e-02 - 2.89473471924e-02 2.46286629032e-02 2.20777720561e-02 2.09446754967e-02 2.03920879638e-02 2.00583037371e-02 - 2.00583037371e-02 2.03920879638e-02 2.09446754967e-02 2.20777720561e-02 2.46286629032e-02 2.89473471924e-02 - 3.39942859271e-02 3.88811262511e-02 4.56987320892e-02 5.69437593911e-02 6.98255714909e-02 7.81193802830e-02 - 5.87902882083e-02 5.17382028008e-02 4.42637285519e-02 3.86294987685e-02 3.39942859271e-02 2.92643382604e-02 - 2.52445347088e-02 2.30253665657e-02 2.24043982374e-02 2.23752455712e-02 2.22713307355e-02 2.21709602700e-02 - 2.22713307355e-02 2.23752455712e-02 2.24043982374e-02 2.30253665657e-02 2.52445347088e-02 2.92643382604e-02 - 3.39942859271e-02 3.86294987685e-02 4.42637285519e-02 5.17382028008e-02 5.87902882083e-02 6.17008685348e-02 - 4.51578187598e-02 4.14583134664e-02 3.75326171916e-02 3.33647463123e-02 2.89473471924e-02 2.52445347088e-02 - 2.33118163733e-02 2.31387869200e-02 2.37301976795e-02 2.40967233573e-02 2.40904203272e-02 2.40904203272e-02 - 2.40967233573e-02 2.37301976795e-02 2.31387869200e-02 2.33118163733e-02 2.52445347088e-02 2.89473471924e-02 - 3.33647463124e-02 3.75326171916e-02 4.14583134664e-02 4.51578187598e-02 4.76134439312e-02 4.76134439312e-02 - 3.73540502296e-02 3.49769025117e-02 3.16057035862e-02 2.78013937268e-02 2.46286629032e-02 2.30253665656e-02 - 2.31387869200e-02 2.42075414898e-02 2.51249889113e-02 2.53973709283e-02 2.53797374387e-02 2.53973709283e-02 - 2.51249889113e-02 2.42075414898e-02 2.31387869200e-02 2.30253665656e-02 2.46286629032e-02 2.78013937268e-02 - 3.16057035862e-02 3.49769025117e-02 3.73540502296e-02 3.86937384314e-02 3.91205047681e-02 3.86937384314e-02 - 3.07822057975e-02 2.84217494631e-02 2.55726183949e-02 2.32030440049e-02 2.20777720561e-02 2.24043982374e-02 - 2.37301976795e-02 2.51249889113e-02 2.58521940319e-02 2.59777844449e-02 2.59777844449e-02 2.58521940319e-02 - 2.51249889113e-02 2.37301976795e-02 2.24043982374e-02 2.20777720561e-02 2.32030440049e-02 2.55726183949e-02 - 2.84217494631e-02 3.07822057975e-02 3.20848019958e-02 3.25042138914e-02 3.25042138914e-02 3.20848019958e-02 - 2.40967233573e-02 2.23752455712e-02 2.09446754967e-02 2.03983415996e-02 2.09446754967e-02 2.23752455712e-02 - 2.40967233573e-02 2.53973709283e-02 2.59777844449e-02 2.60947919991e-02 2.59777844449e-02 2.53973709283e-02 - 2.40967233573e-02 2.23752455712e-02 2.09446754967e-02 2.03983415996e-02 2.09446754967e-02 2.23752455712e-02 - 2.40967233573e-02 2.53973709283e-02 2.59777844449e-02 2.60947919991e-02 2.59777844449e-02 2.53973709283e-02 - 1.86355176327e-02 1.81179302207e-02 1.81502966987e-02 1.89263764236e-02 2.03920879638e-02 2.22713307355e-02 - 2.40904203272e-02 2.53797374387e-02 2.59777844449e-02 2.59777844449e-02 2.53797374387e-02 2.40904203272e-02 - 2.22713307355e-02 2.03920879638e-02 1.89263764236e-02 1.81502966987e-02 1.81179302207e-02 1.86355176327e-02 - 1.92840317465e-02 1.96802339930e-02 1.97871610956e-02 1.97871610956e-02 1.96802339929e-02 1.92840317465e-02 - 1.50739196267e-02 1.56224022211e-02 1.66389339973e-02 1.81430425204e-02 2.00583037371e-02 2.21709602700e-02 - 2.40904203272e-02 2.53973709283e-02 2.58521940319e-02 2.53973709283e-02 2.40904203272e-02 2.21709602700e-02 - 2.00583037371e-02 1.81430425204e-02 1.66389339973e-02 1.56224022211e-02 1.50739196267e-02 1.48492046219e-02 - 1.47279181688e-02 1.45986463266e-02 1.45320680305e-02 1.45986463266e-02 1.47279181688e-02 1.48492046219e-02 - 1.30485807277e-02 1.43188164440e-02 1.59287126341e-02 1.78657504093e-02 2.00583037371e-02 2.22713307355e-02 - 2.40967233573e-02 2.51249889113e-02 2.51249889113e-02 2.40967233573e-02 2.22713307355e-02 2.00583037371e-02 - 1.78657504093e-02 1.59287126341e-02 1.43188164440e-02 1.30485807277e-02 1.21090534529e-02 1.14481144732e-02 - 1.09990704821e-02 1.07546834176e-02 1.07546834176e-02 1.09990704821e-02 1.14481144732e-02 1.21090534529e-02 - 1.21045128275e-02 1.38961516944e-02 1.59287126341e-02 1.81430425204e-02 2.03920879638e-02 2.23752455712e-02 - 2.37301976795e-02 2.42075414899e-02 2.37301976795e-02 2.23752455712e-02 2.03920879638e-02 1.81430425204e-02 - 1.59287126341e-02 1.38961516944e-02 1.21045128275e-02 1.06109348738e-02 9.47290433110e-03 8.70256016586e-03 - 8.26409670065e-03 8.12161313174e-03 8.26409670065e-03 8.70256016586e-03 9.47290433110e-03 1.06109348738e-02 - 1.21045128275e-02 1.43188164440e-02 1.66389339973e-02 1.89263764236e-02 2.09446754967e-02 2.24043982374e-02 - 2.31387869200e-02 2.31387869200e-02 2.24043982374e-02 2.09446754967e-02 1.89263764236e-02 1.66389339973e-02 - 1.43188164440e-02 1.21045128275e-02 1.01198499891e-02 8.50665152156e-03 7.36052161958e-03 6.67439308888e-03 - 6.36714430779e-03 6.36714430779e-03 6.67439308888e-03 7.36052161959e-03 8.50665152157e-03 1.01198499891e-02 - 1.30485807277e-02 1.56224022211e-02 1.81502966987e-02 2.03983415996e-02 2.20777720561e-02 2.30253665657e-02 - 2.33118163733e-02 2.30253665657e-02 2.20777720561e-02 2.03983415996e-02 1.81502966987e-02 1.56224022211e-02 - 1.30485807277e-02 1.06109348738e-02 8.50665152157e-03 6.90308343843e-03 5.85076317272e-03 5.28636096575e-03 - 5.11303460195e-03 5.28636096575e-03 5.85076317272e-03 6.90308343843e-03 8.50665152157e-03 1.06109348738e-02 - 1.50739196267e-02 1.81179302207e-02 2.09446754967e-02 2.32030440050e-02 2.46286629032e-02 2.52445347088e-02 - 2.52445347088e-02 2.46286629032e-02 2.32030440049e-02 2.09446754967e-02 1.81179302207e-02 1.50739196267e-02 - 1.21090534529e-02 9.47290433111e-03 7.36052161959e-03 5.85076317272e-03 4.90588959931e-03 4.45367296569e-03 - 4.45367296569e-03 4.90588959931e-03 5.85076317272e-03 7.36052161959e-03 9.47290433111e-03 1.21090534530e-02 - 1.86355176327e-02 2.23752455712e-02 2.55726183949e-02 2.78013937268e-02 2.89473471924e-02 2.92643382604e-02 - 2.89473471924e-02 2.78013937268e-02 2.55726183949e-02 2.23752455712e-02 1.86355176327e-02 1.48492046219e-02 - 1.14481144732e-02 8.70256016587e-03 6.67439308889e-03 5.28636096575e-03 4.45367296569e-03 4.16945913194e-03 - 4.45367296569e-03 5.28636096575e-03 6.67439308889e-03 8.70256016587e-03 1.14481144732e-02 1.48492046219e-02 - 2.40967233573e-02 2.84217494631e-02 3.16057035862e-02 3.33647463124e-02 3.39942859272e-02 3.39942859272e-02 - 3.33647463124e-02 3.16057035862e-02 2.84217494631e-02 2.40967233573e-02 1.92840317465e-02 1.47279181689e-02 - 1.09990704821e-02 8.26409670065e-03 6.36714430780e-03 5.11303460195e-03 4.45367296569e-03 4.45367296569e-03 - 5.11303460195e-03 6.36714430780e-03 8.26409670066e-03 1.09990704821e-02 1.47279181689e-02 1.92840317465e-02 - 3.07822057975e-02 3.49769025118e-02 3.75326171917e-02 3.86294987685e-02 3.88811262512e-02 3.86294987685e-02 - 3.75326171916e-02 3.49769025118e-02 3.07822057975e-02 2.53973709283e-02 1.96802339930e-02 1.45986463266e-02 - 1.07546834176e-02 8.12161313175e-03 6.36714430780e-03 5.28636096575e-03 4.90588959931e-03 5.28636096575e-03 - 6.36714430780e-03 8.12161313175e-03 1.07546834176e-02 1.45986463266e-02 1.96802339930e-02 2.53973709283e-02 - 3.73540502297e-02 4.14583134665e-02 4.42637285519e-02 4.56987320893e-02 4.56987320893e-02 4.42637285519e-02 - 4.14583134665e-02 3.73540502297e-02 3.20848019959e-02 2.59777844449e-02 1.97871610956e-02 1.45320680305e-02 - 1.07546834176e-02 8.26409670065e-03 6.67439308889e-03 5.85076317272e-03 5.85076317272e-03 6.67439308889e-03 - 8.26409670066e-03 1.07546834176e-02 1.45320680305e-02 1.97871610956e-02 2.59777844449e-02 3.20848019959e-02 - 4.51578187598e-02 5.17382028009e-02 5.69437593912e-02 5.89461275929e-02 5.69437593912e-02 5.17382028009e-02 - 4.51578187598e-02 3.86937384314e-02 3.25042138914e-02 2.60947919991e-02 1.97871610956e-02 1.45986463266e-02 - 1.09990704821e-02 8.70256016587e-03 7.36052161959e-03 6.90308343843e-03 7.36052161959e-03 8.70256016587e-03 - 1.09990704821e-02 1.45986463266e-02 1.97871610956e-02 2.60947919991e-02 3.25042138914e-02 3.86937384314e-02 - 5.87902882084e-02 6.98255714910e-02 7.65071402186e-02 7.65071402186e-02 6.98255714910e-02 5.87902882084e-02 - 4.76134439312e-02 3.91205047681e-02 3.25042138914e-02 2.59777844449e-02 1.96802339930e-02 1.47279181688e-02 - 1.14481144732e-02 9.47290433111e-03 8.50665152157e-03 8.50665152157e-03 9.47290433111e-03 1.14481144732e-02 - 1.47279181688e-02 1.96802339930e-02 2.59777844449e-02 3.25042138914e-02 3.91205047681e-02 4.76134439313e-02 - 7.81193802831e-02 9.00454509227e-02 9.40538638177e-02 9.00454509227e-02 7.81193802831e-02 6.17008685349e-02 - 4.76134439312e-02 3.86937384314e-02 3.20848019959e-02 2.53973709283e-02 1.92840317465e-02 1.48492046219e-02 - 1.21090534529e-02 1.06109348738e-02 1.01198499891e-02 1.06109348738e-02 1.21090534529e-02 1.48492046219e-02 - 1.92840317465e-02 2.53973709283e-02 3.20848019959e-02 3.86937384314e-02 4.76134439312e-02 6.17008685349e-02 - 1.14984029929e-01 1.14284409879e-01 1.14984029929e-01 1.07504612680e-01 8.44119137021e-02 5.87902882083e-02 - 4.36605690195e-02 3.62673468746e-02 2.99301385615e-02 2.37301976795e-02 1.91106081253e-02 1.63906429721e-02 - 1.50424663757e-02 1.46288647205e-02 1.50424663757e-02 1.63906429721e-02 1.91106081253e-02 2.37301976795e-02 - 2.99301385615e-02 3.62673468746e-02 4.36605690195e-02 5.87902882083e-02 8.44119137022e-02 1.07504612680e-01 - 1.14284409879e-01 1.14284409879e-01 1.13479602981e-01 9.74781930219e-02 6.98255714909e-02 4.86717503238e-02 - 3.89798518705e-02 3.31963581644e-02 2.73091501295e-02 2.24043982374e-02 1.93871115228e-02 1.78888288560e-02 - 1.72812015630e-02 1.72812015630e-02 1.78888288560e-02 1.93871115228e-02 2.24043982374e-02 2.73091501295e-02 - 3.31963581644e-02 3.89798518706e-02 4.86717503238e-02 6.98255714909e-02 9.74781930220e-02 1.13479602981e-01 - 1.14984029929e-01 1.13479602981e-01 1.01688733219e-01 7.65071402185e-02 5.24371687843e-02 4.00842207789e-02 - 3.45751625707e-02 2.95890354456e-02 2.49964037842e-02 2.20777720561e-02 2.07196781889e-02 2.02040886880e-02 - 2.00653973119e-02 2.02040886880e-02 2.07196781889e-02 2.20777720561e-02 2.49964037843e-02 2.95890354456e-02 - 3.45751625707e-02 4.00842207789e-02 5.24371687843e-02 7.65071402185e-02 1.01688733220e-01 1.13479602981e-01 - 1.07504612679e-01 9.74781930219e-02 7.65071402185e-02 5.38553081705e-02 4.03062291682e-02 3.46096179590e-02 - 3.04773481040e-02 2.65400648390e-02 2.40839675381e-02 2.32030440049e-02 2.30939042850e-02 2.31231583381e-02 - 2.31231583381e-02 2.30939042850e-02 2.32030440049e-02 2.40839675381e-02 2.65400648390e-02 3.04773481040e-02 - 3.46096179590e-02 4.03062291683e-02 5.38553081705e-02 7.65071402185e-02 9.74781930219e-02 1.07504612679e-01 - 8.44119137021e-02 6.98255714909e-02 5.24371687843e-02 4.03062291682e-02 3.44216532954e-02 3.05548189106e-02 - 2.71142998602e-02 2.51931467428e-02 2.49964037842e-02 2.55726183949e-02 2.60492747623e-02 2.61902387743e-02 - 2.60492747623e-02 2.55726183949e-02 2.49964037842e-02 2.51931467428e-02 2.71142998602e-02 3.05548189106e-02 - 3.44216532954e-02 4.03062291683e-02 5.24371687843e-02 6.98255714909e-02 8.44119137022e-02 8.97664201879e-02 - 5.87902882083e-02 4.86717503238e-02 4.00842207789e-02 3.46096179590e-02 3.05548189106e-02 2.72051400328e-02 - 2.55791633527e-02 2.59618688149e-02 2.73091501295e-02 2.84217494631e-02 2.88778389824e-02 2.88778389824e-02 - 2.84217494631e-02 2.73091501295e-02 2.59618688149e-02 2.55791633527e-02 2.72051400328e-02 3.05548189106e-02 - 3.46096179590e-02 4.00842207790e-02 4.86717503239e-02 5.87902882083e-02 6.56428313090e-02 6.56428313090e-02 - 4.36605690195e-02 3.89798518705e-02 3.45751625707e-02 3.04773481040e-02 2.71142998602e-02 2.55791633527e-02 - 2.62414819209e-02 2.81653274579e-02 2.99301385615e-02 3.07822057975e-02 3.09609358381e-02 3.07822057975e-02 - 2.99301385615e-02 2.81653274579e-02 2.62414819209e-02 2.55791633527e-02 2.71142998602e-02 3.04773481040e-02 - 3.45751625707e-02 3.89798518706e-02 4.36605690195e-02 4.76134439312e-02 4.91904413240e-02 4.76134439312e-02 - 3.62673468745e-02 3.31963581643e-02 2.95890354456e-02 2.65400648390e-02 2.51931467427e-02 2.59618688149e-02 - 2.81653274579e-02 3.04404777485e-02 3.17361436403e-02 3.20848019959e-02 3.20848019959e-02 3.17361436402e-02 - 3.04404777485e-02 2.81653274579e-02 2.59618688149e-02 2.51931467428e-02 2.65400648390e-02 2.95890354456e-02 - 3.31963581644e-02 3.62673468746e-02 3.82423088608e-02 3.91205047681e-02 3.91205047681e-02 3.82423088607e-02 - 2.99301385615e-02 2.73091501295e-02 2.49964037842e-02 2.40839675381e-02 2.49964037842e-02 2.73091501295e-02 - 2.99301385615e-02 3.17361436403e-02 3.24128421387e-02 3.25042138914e-02 3.24128421387e-02 3.17361436402e-02 - 2.99301385615e-02 2.73091501295e-02 2.49964037842e-02 2.40839675381e-02 2.49964037842e-02 2.73091501295e-02 - 2.99301385615e-02 3.17361436402e-02 3.24128421387e-02 3.25042138914e-02 3.24128421387e-02 3.17361436402e-02 - 2.37301976795e-02 2.24043982374e-02 2.20777720560e-02 2.32030440049e-02 2.55726183949e-02 2.84217494631e-02 - 3.07822057975e-02 3.20848019959e-02 3.25042138914e-02 3.25042138914e-02 3.20848019959e-02 3.07822057975e-02 - 2.84217494631e-02 2.55726183949e-02 2.32030440049e-02 2.20777720561e-02 2.24043982374e-02 2.37301976795e-02 - 2.51249889113e-02 2.58521940319e-02 2.59777844449e-02 2.59777844449e-02 2.58521940319e-02 2.51249889113e-02 - 1.91106081253e-02 1.93871115227e-02 2.07196781889e-02 2.30939042849e-02 2.60492747622e-02 2.88778389824e-02 - 3.09609358381e-02 3.20848019959e-02 3.24128421387e-02 3.20848019959e-02 3.09609358381e-02 2.88778389824e-02 - 2.60492747622e-02 2.30939042849e-02 2.07196781889e-02 1.93871115228e-02 1.91106081253e-02 1.94290489291e-02 - 1.97044449599e-02 1.96802339930e-02 1.96031535282e-02 1.96802339929e-02 1.97044449599e-02 1.94290489291e-02 - 1.63906429721e-02 1.78888288560e-02 2.02040886880e-02 2.31231583381e-02 2.61902387743e-02 2.88778389824e-02 - 3.07822057975e-02 3.17361436403e-02 3.17361436403e-02 3.07822057975e-02 2.88778389824e-02 2.61902387743e-02 - 2.31231583381e-02 2.02040886880e-02 1.78888288560e-02 1.63906429721e-02 1.55937378378e-02 1.51354763629e-02 - 1.47279181688e-02 1.44210111226e-02 1.44210111226e-02 1.47279181688e-02 1.51354763629e-02 1.55937378378e-02 - 1.50424663757e-02 1.72812015630e-02 2.00653973118e-02 2.31231583381e-02 2.60492747623e-02 2.84217494631e-02 - 2.99301385615e-02 3.04404777485e-02 2.99301385615e-02 2.84217494631e-02 2.60492747623e-02 2.31231583381e-02 - 2.00653973118e-02 1.72812015630e-02 1.50424663757e-02 1.34169388324e-02 1.22800512627e-02 1.14481144732e-02 - 1.08699825973e-02 1.06518186545e-02 1.08699825973e-02 1.14481144732e-02 1.22800512627e-02 1.34169388324e-02 - 1.46288647205e-02 1.72812015630e-02 2.02040886880e-02 2.30939042849e-02 2.55726183949e-02 2.73091501295e-02 - 2.81653274579e-02 2.81653274579e-02 2.73091501295e-02 2.55726183949e-02 2.30939042849e-02 2.02040886880e-02 - 1.72812015630e-02 1.46288647205e-02 1.24265880360e-02 1.07225418333e-02 9.47290433110e-03 8.62540500905e-03 - 8.18383164206e-03 8.18383164206e-03 8.62540500905e-03 9.47290433110e-03 1.07225418333e-02 1.24265880360e-02 - 1.50424663757e-02 1.78888288560e-02 2.07196781889e-02 2.32030440049e-02 2.49964037843e-02 2.59618688149e-02 - 2.62414819209e-02 2.59618688149e-02 2.49964037843e-02 2.32030440049e-02 2.07196781889e-02 1.78888288560e-02 - 1.50424663757e-02 1.24265880360e-02 1.02153626600e-02 8.50665152156e-03 7.31933010296e-03 6.62684043801e-03 - 6.39967919814e-03 6.62684043801e-03 7.31933010296e-03 8.50665152157e-03 1.02153626600e-02 1.24265880360e-02 - 1.63906429721e-02 1.93871115228e-02 2.20777720561e-02 2.40839675381e-02 2.51931467428e-02 2.55791633528e-02 - 2.55791633528e-02 2.51931467428e-02 2.40839675381e-02 2.20777720561e-02 1.93871115228e-02 1.63906429721e-02 - 1.34169388324e-02 1.07225418333e-02 8.50665152157e-03 6.87426464911e-03 5.82746138702e-03 5.32068131556e-03 - 5.32068131556e-03 5.82746138702e-03 6.87426464911e-03 8.50665152157e-03 1.07225418333e-02 1.34169388324e-02 - 1.91106081254e-02 2.24043982374e-02 2.49964037843e-02 2.65400648390e-02 2.71142998602e-02 2.72051400328e-02 - 2.71142998602e-02 2.65400648390e-02 2.49964037843e-02 2.24043982374e-02 1.91106081253e-02 1.55937378378e-02 - 1.22800512627e-02 9.47290433111e-03 7.31933010297e-03 5.82746138702e-03 4.94632325183e-03 4.65175947716e-03 - 4.94632325183e-03 5.82746138702e-03 7.31933010297e-03 9.47290433111e-03 1.22800512627e-02 1.55937378378e-02 - 2.37301976795e-02 2.73091501295e-02 2.95890354456e-02 3.04773481040e-02 3.05548189106e-02 3.05548189106e-02 - 3.04773481040e-02 2.95890354456e-02 2.73091501295e-02 2.37301976795e-02 1.94290489291e-02 1.51354763629e-02 - 1.14481144732e-02 8.62540500906e-03 6.62684043802e-03 5.32068131556e-03 4.65175947716e-03 4.65175947716e-03 - 5.32068131556e-03 6.62684043802e-03 8.62540500906e-03 1.14481144732e-02 1.51354763629e-02 1.94290489291e-02 - 2.99301385615e-02 3.31963581644e-02 3.45751625707e-02 3.46096179591e-02 3.44216532954e-02 3.46096179591e-02 - 3.45751625707e-02 3.31963581644e-02 2.99301385615e-02 2.51249889113e-02 1.97044449599e-02 1.47279181689e-02 - 1.08699825973e-02 8.18383164206e-03 6.39967919814e-03 5.32068131556e-03 4.94632325183e-03 5.32068131556e-03 - 6.39967919814e-03 8.18383164207e-03 1.08699825973e-02 1.47279181689e-02 1.97044449599e-02 2.51249889113e-02 - 3.62673468746e-02 3.89798518706e-02 4.00842207790e-02 4.03062291683e-02 4.03062291683e-02 4.00842207790e-02 - 3.89798518706e-02 3.62673468746e-02 3.17361436403e-02 2.58521940319e-02 1.96802339930e-02 1.44210111226e-02 - 1.06518186545e-02 8.18383164206e-03 6.62684043802e-03 5.82746138702e-03 5.82746138702e-03 6.62684043802e-03 - 8.18383164207e-03 1.06518186545e-02 1.44210111226e-02 1.96802339930e-02 2.58521940319e-02 3.17361436403e-02 - 4.36605690195e-02 4.86717503239e-02 5.24371687844e-02 5.38553081706e-02 5.24371687844e-02 4.86717503239e-02 - 4.36605690195e-02 3.82423088608e-02 3.24128421387e-02 2.59777844449e-02 1.96031535282e-02 1.44210111226e-02 - 1.08699825973e-02 8.62540500906e-03 7.31933010297e-03 6.87426464911e-03 7.31933010297e-03 8.62540500906e-03 - 1.08699825973e-02 1.44210111226e-02 1.96031535283e-02 2.59777844449e-02 3.24128421387e-02 3.82423088608e-02 - 5.87902882084e-02 6.98255714910e-02 7.65071402186e-02 7.65071402186e-02 6.98255714910e-02 5.87902882084e-02 - 4.76134439312e-02 3.91205047681e-02 3.25042138914e-02 2.59777844449e-02 1.96802339930e-02 1.47279181688e-02 - 1.14481144732e-02 9.47290433111e-03 8.50665152157e-03 8.50665152157e-03 9.47290433111e-03 1.14481144732e-02 - 1.47279181689e-02 1.96802339930e-02 2.59777844449e-02 3.25042138914e-02 3.91205047681e-02 4.76134439313e-02 - 8.44119137022e-02 9.74781930220e-02 1.01688733220e-01 9.74781930220e-02 8.44119137022e-02 6.56428313091e-02 - 4.91904413241e-02 3.91205047681e-02 3.24128421387e-02 2.58521940319e-02 1.97044449599e-02 1.51354763629e-02 - 1.22800512627e-02 1.07225418333e-02 1.02153626600e-02 1.07225418333e-02 1.22800512627e-02 1.51354763629e-02 - 1.97044449599e-02 2.58521940319e-02 3.24128421387e-02 3.91205047681e-02 4.91904413241e-02 6.56428313091e-02 - 1.07504612680e-01 1.13479602981e-01 1.13479602981e-01 1.07504612680e-01 8.97664201880e-02 6.56428313091e-02 - 4.76134439312e-02 3.82423088608e-02 3.17361436403e-02 2.51249889113e-02 1.94290489291e-02 1.55937378378e-02 - 1.34169388324e-02 1.24265880360e-02 1.24265880360e-02 1.34169388324e-02 1.55937378378e-02 1.94290489291e-02 - 2.51249889113e-02 3.17361436403e-02 3.82423088608e-02 4.76134439312e-02 6.56428313091e-02 8.97664201881e-02 - 1.05271425072e-01 1.05271425072e-01 1.15476799016e-01 1.07504612680e-01 7.81193802830e-02 5.18158087397e-02 - 3.97328533270e-02 3.38925854809e-02 2.81653274579e-02 2.31387869200e-02 1.99975947255e-02 1.85132763207e-02 - 1.79741436761e-02 1.79741436761e-02 1.85132763207e-02 1.99975947255e-02 2.31387869200e-02 2.81653274579e-02 - 3.38925854809e-02 3.97328533271e-02 5.18158087398e-02 7.81193802831e-02 1.07504612680e-01 1.15476799016e-01 - 1.05271425072e-01 1.12133375345e-01 1.13479602981e-01 9.00454509227e-02 5.87809730550e-02 4.11388919901e-02 - 3.48628450776e-02 3.04051143293e-02 2.59618688149e-02 2.30253665657e-02 2.17948830411e-02 2.15195877577e-02 - 2.15038028103e-02 2.15195877577e-02 2.17948830411e-02 2.30253665657e-02 2.59618688149e-02 3.04051143293e-02 - 3.48628450776e-02 4.11388919901e-02 5.87809730550e-02 9.00454509227e-02 1.13479602981e-01 1.12133375345e-01 - 1.15476799016e-01 1.13479602981e-01 9.40538638176e-02 6.30064198496e-02 4.15214050146e-02 3.39196724265e-02 - 3.06792139923e-02 2.73903662614e-02 2.51931467427e-02 2.46286629032e-02 2.49902463304e-02 2.54054430258e-02 - 2.54054430258e-02 2.49902463304e-02 2.46286629032e-02 2.51931467428e-02 2.73903662614e-02 3.06792139923e-02 - 3.39196724265e-02 4.15214050146e-02 6.30064198497e-02 9.40538638177e-02 1.13479602981e-01 1.15476799016e-01 - 1.07504612679e-01 9.00454509227e-02 6.30064198496e-02 4.15474983449e-02 3.26920605346e-02 2.97974895143e-02 - 2.74769895380e-02 2.61756575962e-02 2.65400648390e-02 2.78013937268e-02 2.89590113766e-02 2.93979324046e-02 - 2.89590113766e-02 2.78013937268e-02 2.65400648390e-02 2.61756575962e-02 2.74769895380e-02 2.97974895143e-02 - 3.26920605347e-02 4.15474983449e-02 6.30064198497e-02 9.00454509227e-02 1.07504612680e-01 1.12225405020e-01 - 7.81193802830e-02 5.87809730550e-02 4.15214050146e-02 3.26920605347e-02 2.92436594720e-02 2.70513354759e-02 - 2.62352804996e-02 2.73903662614e-02 2.95890354456e-02 3.16057035862e-02 3.27029610793e-02 3.27029610793e-02 - 3.16057035862e-02 2.95890354456e-02 2.73903662614e-02 2.62352804996e-02 2.70513354759e-02 2.92436594720e-02 - 3.26920605347e-02 4.15214050146e-02 5.87809730550e-02 7.81193802830e-02 8.97664201880e-02 8.97664201879e-02 - 5.18158087397e-02 4.11388919901e-02 3.39196724265e-02 2.97974895143e-02 2.70513354759e-02 2.61140936360e-02 - 2.75584598461e-02 3.04051143293e-02 3.31963581644e-02 3.49769025118e-02 3.55406204185e-02 3.49769025117e-02 - 3.31963581644e-02 3.04051143293e-02 2.75584598461e-02 2.61140936360e-02 2.70513354759e-02 2.97974895143e-02 - 3.39196724265e-02 4.11388919902e-02 5.18158087398e-02 6.17008685349e-02 6.56428313090e-02 6.17008685348e-02 - 3.97328533270e-02 3.48628450776e-02 3.06792139923e-02 2.74769895380e-02 2.62352804996e-02 2.75584598461e-02 - 3.06058831029e-02 3.38925854809e-02 3.62673468746e-02 3.73540502296e-02 3.73540502296e-02 3.62673468746e-02 - 3.38925854809e-02 3.06058831029e-02 2.75584598461e-02 2.62352804996e-02 2.74769895380e-02 3.06792139923e-02 - 3.48628450776e-02 3.97328533271e-02 4.45490987358e-02 4.76134439312e-02 4.76134439312e-02 4.45490987357e-02 - 3.38925854809e-02 3.04051143293e-02 2.73903662614e-02 2.61756575962e-02 2.73903662614e-02 3.04051143293e-02 - 3.38925854809e-02 3.66854482599e-02 3.82423088608e-02 3.86937384314e-02 3.82423088608e-02 3.66854482599e-02 - 3.38925854809e-02 3.04051143293e-02 2.73903662614e-02 2.61756575962e-02 2.73903662614e-02 3.04051143293e-02 - 3.38925854809e-02 3.66854482599e-02 3.82423088607e-02 3.86937384314e-02 3.82423088607e-02 3.66854482599e-02 - 2.81653274579e-02 2.59618688149e-02 2.51931467427e-02 2.65400648390e-02 2.95890354456e-02 3.31963581644e-02 - 3.62673468746e-02 3.82423088608e-02 3.91205047681e-02 3.91205047681e-02 3.82423088608e-02 3.62673468746e-02 - 3.31963581644e-02 2.95890354456e-02 2.65400648390e-02 2.51931467428e-02 2.59618688149e-02 2.81653274579e-02 - 3.04404777485e-02 3.17361436402e-02 3.20848019958e-02 3.20848019958e-02 3.17361436402e-02 3.04404777485e-02 - 2.31387869200e-02 2.30253665656e-02 2.46286629032e-02 2.78013937268e-02 3.16057035862e-02 3.49769025118e-02 - 3.73540502297e-02 3.86937384314e-02 3.91205047681e-02 3.86937384314e-02 3.73540502297e-02 3.49769025118e-02 - 3.16057035862e-02 2.78013937268e-02 2.46286629032e-02 2.30253665657e-02 2.31387869200e-02 2.42075414898e-02 - 2.51249889113e-02 2.53973709283e-02 2.53797374387e-02 2.53973709283e-02 2.51249889113e-02 2.42075414898e-02 - 1.99975947255e-02 2.17948830411e-02 2.49902463304e-02 2.89590113766e-02 3.27029610793e-02 3.55406204185e-02 - 3.73540502297e-02 3.82423088608e-02 3.82423088608e-02 3.73540502297e-02 3.55406204185e-02 3.27029610793e-02 - 2.89590113766e-02 2.49902463304e-02 2.17948830411e-02 1.99975947255e-02 1.94536795471e-02 1.94290489291e-02 - 1.92840317465e-02 1.90345757835e-02 1.90345757835e-02 1.92840317465e-02 1.94290489291e-02 1.94536795471e-02 - 1.85132763207e-02 2.15195877577e-02 2.54054430258e-02 2.93979324046e-02 3.27029610793e-02 3.49769025118e-02 - 3.62673468746e-02 3.66854482599e-02 3.62673468746e-02 3.49769025118e-02 3.27029610793e-02 2.93979324046e-02 - 2.54054430258e-02 2.15195877577e-02 1.85132763207e-02 1.66439398964e-02 1.55937378378e-02 1.48492046219e-02 - 1.42308801751e-02 1.39667079820e-02 1.42308801751e-02 1.48492046219e-02 1.55937378378e-02 1.66439398964e-02 - 1.79741436760e-02 2.15038028103e-02 2.54054430258e-02 2.89590113766e-02 3.16057035862e-02 3.31963581644e-02 - 3.38925854810e-02 3.38925854810e-02 3.31963581644e-02 3.16057035862e-02 2.89590113766e-02 2.54054430258e-02 - 2.15038028103e-02 1.79741436760e-02 1.52762540169e-02 1.34169388324e-02 1.21090534529e-02 1.11331987338e-02 - 1.05557815436e-02 1.05557815436e-02 1.11331987338e-02 1.21090534529e-02 1.34169388324e-02 1.52762540168e-02 - 1.79741436760e-02 2.15195877577e-02 2.49902463304e-02 2.78013937268e-02 2.95890354456e-02 3.04051143294e-02 - 3.06058831030e-02 3.04051143294e-02 2.95890354456e-02 2.78013937268e-02 2.49902463304e-02 2.15195877577e-02 - 1.79741436760e-02 1.48656767560e-02 1.24265880360e-02 1.06109348738e-02 9.28357359922e-03 8.41997198038e-03 - 8.11230157084e-03 8.41997198038e-03 9.28357359923e-03 1.06109348738e-02 1.24265880360e-02 1.48656767560e-02 - 1.85132763207e-02 2.17948830411e-02 2.46286629032e-02 2.65400648390e-02 2.73903662614e-02 2.75584598462e-02 - 2.75584598462e-02 2.73903662614e-02 2.65400648390e-02 2.46286629032e-02 2.17948830411e-02 1.85132763207e-02 - 1.52762540169e-02 1.24265880360e-02 1.01198499891e-02 8.37400390520e-03 7.17862759619e-03 6.56138198056e-03 - 6.56138198056e-03 7.17862759619e-03 8.37400390520e-03 1.01198499891e-02 1.24265880360e-02 1.52762540169e-02 - 1.99975947255e-02 2.30253665657e-02 2.51931467428e-02 2.61756575963e-02 2.62352804996e-02 2.61140936361e-02 - 2.62352804996e-02 2.61756575963e-02 2.51931467428e-02 2.30253665657e-02 1.99975947255e-02 1.66439398964e-02 - 1.34169388324e-02 1.06109348738e-02 8.37400390520e-03 6.75449318420e-03 5.76667312114e-03 5.43239778449e-03 - 5.76667312114e-03 6.75449318420e-03 8.37400390520e-03 1.06109348738e-02 1.34169388324e-02 1.66439398964e-02 - 2.31387869200e-02 2.59618688149e-02 2.73903662614e-02 2.74769895380e-02 2.70513354759e-02 2.70513354759e-02 - 2.74769895380e-02 2.73903662614e-02 2.59618688149e-02 2.31387869200e-02 1.94536795471e-02 1.55937378378e-02 - 1.21090534530e-02 9.28357359923e-03 7.17862759620e-03 5.76667312114e-03 5.04343778102e-03 5.04343778102e-03 - 5.76667312114e-03 7.17862759620e-03 9.28357359923e-03 1.21090534530e-02 1.55937378378e-02 1.94536795471e-02 - 2.81653274579e-02 3.04051143294e-02 3.06792139923e-02 2.97974895143e-02 2.92436594720e-02 2.97974895143e-02 - 3.06792139923e-02 3.04051143294e-02 2.81653274579e-02 2.42075414899e-02 1.94290489291e-02 1.48492046219e-02 - 1.11331987338e-02 8.41997198039e-03 6.56138198057e-03 5.43239778449e-03 5.04343778102e-03 5.43239778449e-03 - 6.56138198057e-03 8.41997198039e-03 1.11331987338e-02 1.48492046219e-02 1.94290489291e-02 2.42075414899e-02 - 3.38925854810e-02 3.48628450777e-02 3.39196724265e-02 3.26920605347e-02 3.26920605347e-02 3.39196724265e-02 - 3.48628450776e-02 3.38925854809e-02 3.04404777485e-02 2.51249889113e-02 1.92840317465e-02 1.42308801751e-02 - 1.05557815436e-02 8.11230157085e-03 6.56138198057e-03 5.76667312114e-03 5.76667312114e-03 6.56138198057e-03 - 8.11230157085e-03 1.05557815436e-02 1.42308801751e-02 1.92840317465e-02 2.51249889113e-02 3.04404777485e-02 - 3.97328533271e-02 4.11388919902e-02 4.15214050147e-02 4.15474983450e-02 4.15214050147e-02 4.11388919902e-02 - 3.97328533271e-02 3.66854482599e-02 3.17361436403e-02 2.53973709283e-02 1.90345757835e-02 1.39667079820e-02 - 1.05557815436e-02 8.41997198039e-03 7.17862759620e-03 6.75449318420e-03 7.17862759620e-03 8.41997198039e-03 - 1.05557815436e-02 1.39667079820e-02 1.90345757835e-02 2.53973709283e-02 3.17361436403e-02 3.66854482599e-02 - 5.18158087398e-02 5.87809730551e-02 6.30064198498e-02 6.30064198498e-02 5.87809730551e-02 5.18158087398e-02 - 4.45490987358e-02 3.82423088608e-02 3.20848019959e-02 2.53797374387e-02 1.90345757835e-02 1.42308801751e-02 - 1.11331987338e-02 9.28357359923e-03 8.37400390520e-03 8.37400390520e-03 9.28357359923e-03 1.11331987338e-02 - 1.42308801751e-02 1.90345757835e-02 2.53797374387e-02 3.20848019959e-02 3.82423088608e-02 4.45490987358e-02 - 7.81193802831e-02 9.00454509228e-02 9.40538638178e-02 9.00454509228e-02 7.81193802831e-02 6.17008685349e-02 - 4.76134439312e-02 3.86937384314e-02 3.20848019959e-02 2.53973709283e-02 1.92840317465e-02 1.48492046219e-02 - 1.21090534529e-02 1.06109348738e-02 1.01198499891e-02 1.06109348738e-02 1.21090534530e-02 1.48492046219e-02 - 1.92840317465e-02 2.53973709283e-02 3.20848019959e-02 3.86937384314e-02 4.76134439313e-02 6.17008685349e-02 - 1.07504612680e-01 1.13479602981e-01 1.13479602981e-01 1.07504612680e-01 8.97664201881e-02 6.56428313091e-02 - 4.76134439312e-02 3.82423088608e-02 3.17361436403e-02 2.51249889113e-02 1.94290489291e-02 1.55937378378e-02 - 1.34169388324e-02 1.24265880360e-02 1.24265880360e-02 1.34169388324e-02 1.55937378378e-02 1.94290489291e-02 - 2.51249889113e-02 3.17361436403e-02 3.82423088608e-02 4.76134439312e-02 6.56428313091e-02 8.97664201881e-02 - 1.15476799016e-01 1.12133375345e-01 1.15476799016e-01 1.12225405021e-01 8.97664201880e-02 6.17008685349e-02 - 4.45490987358e-02 3.66854482599e-02 3.04404777485e-02 2.42075414898e-02 1.94536795471e-02 1.66439398964e-02 - 1.52762540169e-02 1.48656767561e-02 1.52762540169e-02 1.66439398964e-02 1.94536795471e-02 2.42075414898e-02 - 3.04404777485e-02 3.66854482599e-02 4.45490987358e-02 6.17008685349e-02 8.97664201881e-02 1.12225405021e-01 + 14 4.000000 0.000000 0.000000 0.000000 + 14 4.000000 2.550000 2.550000 2.550000 + 9.35837065899e-02 1.05270975780e-01 1.14984832163e-01 9.46627422162e-02 6.13627808209e-02 4.15728345981e-02 + 3.48685420831e-02 3.06060385055e-02 2.62414574798e-02 2.33116271285e-02 2.21383854932e-02 2.19609803409e-02 + 2.19946944680e-02 2.19609803409e-02 2.21383854933e-02 2.33116271285e-02 2.62414574798e-02 3.06060385056e-02 + 3.48685420831e-02 4.15728345981e-02 6.13627808209e-02 9.46627422163e-02 1.14984832163e-01 1.05270975780e-01 + 1.05270975780e-01 1.14285203523e-01 1.02256358961e-01 6.87952783262e-02 4.23899114590e-02 3.31552867935e-02 + 3.03790382414e-02 2.75583653887e-02 2.55789017267e-02 2.52442663181e-02 2.59180287237e-02 2.65859681267e-02 + 2.65859681267e-02 2.59180287237e-02 2.52442663181e-02 2.55789017267e-02 2.75583653887e-02 3.03790382414e-02 + 3.31552867935e-02 4.23899114590e-02 6.87952783262e-02 1.02256358961e-01 1.14285203523e-01 1.05270975780e-01 + 1.14984832163e-01 1.02256358961e-01 7.14597362941e-02 4.26144257223e-02 3.05271423180e-02 2.81921518485e-02 + 2.69079511827e-02 2.62349275771e-02 2.71139620581e-02 2.89472257765e-02 3.06730045455e-02 3.13659743854e-02 + 3.06730045455e-02 2.89472257765e-02 2.71139620581e-02 2.62349275771e-02 2.69079511827e-02 2.81921518485e-02 + 3.05271423180e-02 4.26144257223e-02 7.14597362941e-02 1.02256358961e-01 1.14984832163e-01 1.15477102967e-01 + 9.46627422161e-02 6.87952783261e-02 4.26144257223e-02 2.92886940372e-02 2.61163074370e-02 2.52952680324e-02 + 2.54923302762e-02 2.74766558578e-02 3.04771749334e-02 3.33648425996e-02 3.51265406940e-02 3.51265406940e-02 + 3.33648425996e-02 3.04771749334e-02 2.74766558578e-02 2.54923302762e-02 2.52952680324e-02 2.61163074370e-02 + 2.92886940372e-02 4.26144257223e-02 6.87952783262e-02 9.46627422163e-02 1.07505424132e-01 1.07505424132e-01 + 6.13627808208e-02 4.23899114590e-02 3.05271423180e-02 2.61163074370e-02 2.44576079909e-02 2.45136072590e-02 + 2.69079511826e-02 3.06791433434e-02 3.45751721122e-02 3.75326780270e-02 3.86294226543e-02 3.75326780270e-02 + 3.45751721123e-02 3.06791433435e-02 2.69079511826e-02 2.45136072589e-02 2.44576079909e-02 2.61163074370e-02 + 3.05271423180e-02 4.23899114590e-02 6.13627808209e-02 7.81199920260e-02 8.44125475991e-02 7.81199920260e-02 + 4.15728345981e-02 3.31552867935e-02 2.81921518485e-02 2.52952680324e-02 2.45136072590e-02 2.65223600849e-02 + 3.03790382413e-02 3.48629669819e-02 3.89797931477e-02 4.14580711769e-02 4.14580711769e-02 3.89797931478e-02 + 3.48629669819e-02 3.03790382414e-02 2.65223600849e-02 2.45136072589e-02 2.52952680324e-02 2.81921518485e-02 + 3.31552867935e-02 4.15728345981e-02 5.18159940627e-02 5.87904239654e-02 5.87904239654e-02 5.18159940627e-02 + 3.48685420831e-02 3.03790382414e-02 2.69079511827e-02 2.54923302762e-02 2.69079511826e-02 3.03790382413e-02 + 3.48685420831e-02 3.97328731661e-02 4.36602554331e-02 4.51573443066e-02 4.36602554332e-02 3.97328731662e-02 + 3.48685420831e-02 3.03790382414e-02 2.69079511826e-02 2.54923302762e-02 2.69079511826e-02 3.03790382414e-02 + 3.48685420831e-02 3.97328731662e-02 4.36602554332e-02 4.51573443066e-02 4.36602554331e-02 3.97328731661e-02 + 3.06060385055e-02 2.75583653887e-02 2.62349275771e-02 2.74766558578e-02 3.06791433434e-02 3.48629669819e-02 + 3.97328731661e-02 4.45488243286e-02 4.76129372810e-02 4.76129372810e-02 4.45488243286e-02 3.97328731662e-02 + 3.48629669819e-02 3.06791433435e-02 2.74766558578e-02 2.62349275771e-02 2.75583653887e-02 3.06060385056e-02 + 3.38926655663e-02 3.62670291685e-02 3.73533616181e-02 3.73533616181e-02 3.62670291685e-02 3.38926655663e-02 + 2.62414574798e-02 2.55789017267e-02 2.71139620581e-02 3.04771749334e-02 3.45751721122e-02 3.89797931477e-02 + 4.36602554331e-02 4.76129372810e-02 4.91898793330e-02 4.76129372810e-02 4.36602554332e-02 3.89797931478e-02 + 3.45751721123e-02 3.04771749334e-02 2.71139620581e-02 2.55789017267e-02 2.62414574798e-02 2.81654297894e-02 + 2.99300705585e-02 3.07818020073e-02 3.09603631544e-02 3.07818020073e-02 2.99300705585e-02 2.81654297894e-02 + 2.33116271285e-02 2.52442663181e-02 2.89472257765e-02 3.33648425996e-02 3.75326780270e-02 4.14580711769e-02 + 4.51573443066e-02 4.76129372810e-02 4.76129372810e-02 4.51573443066e-02 4.14580711769e-02 3.75326780271e-02 + 3.33648425996e-02 2.89472257765e-02 2.52442663181e-02 2.33116271285e-02 2.31387782150e-02 2.37302554132e-02 + 2.40966608398e-02 2.40902041017e-02 2.40902041017e-02 2.40966608398e-02 2.37302554132e-02 2.31387782150e-02 + 2.21383854932e-02 2.59180287237e-02 3.06730045455e-02 3.51265406939e-02 3.86294226543e-02 4.14580711769e-02 + 4.36602554331e-02 4.45488243286e-02 4.36602554332e-02 4.14580711769e-02 3.86294226544e-02 3.51265406940e-02 + 3.06730045456e-02 2.59180287237e-02 2.21383854932e-02 1.99974571615e-02 1.91106183945e-02 1.86355831047e-02 + 1.81617249718e-02 1.79266369346e-02 1.81617249718e-02 1.86355831047e-02 1.91106183945e-02 1.99974571616e-02 + 2.19609803409e-02 2.65859681267e-02 3.13659743854e-02 3.51265406940e-02 3.75326780270e-02 3.89797931478e-02 + 3.97328731661e-02 3.97328731661e-02 3.89797931478e-02 3.75326780270e-02 3.51265406940e-02 3.13659743854e-02 + 2.65859681267e-02 2.19609803409e-02 1.85131064743e-02 1.63905844408e-02 1.50739753351e-02 1.40807237278e-02 + 1.34421984781e-02 1.34421984781e-02 1.40807237278e-02 1.50739753351e-02 1.63905844408e-02 1.85131064743e-02 + 2.19946944680e-02 2.65859681267e-02 3.06730045455e-02 3.33648425996e-02 3.45751721123e-02 3.48629669819e-02 + 3.48685420831e-02 3.48629669819e-02 3.45751721123e-02 3.33648425996e-02 3.06730045455e-02 2.65859681267e-02 + 2.19946944681e-02 1.79740052840e-02 1.50423980149e-02 1.30486211596e-02 1.16096263393e-02 1.06052341643e-02 + 1.02249435892e-02 1.06052341643e-02 1.16096263393e-02 1.30486211596e-02 1.50423980149e-02 1.79740052840e-02 + 2.19609803409e-02 2.59180287237e-02 2.89472257765e-02 3.04771749334e-02 3.06791433435e-02 3.03790382414e-02 + 3.03790382414e-02 3.06791433435e-02 3.04771749334e-02 2.89472257765e-02 2.59180287237e-02 2.19609803409e-02 + 1.79740052840e-02 1.46288065643e-02 1.21045641480e-02 1.02402112845e-02 8.89928200875e-03 8.15763089847e-03 + 8.15763089847e-03 8.89928200875e-03 1.02402112845e-02 1.21045641480e-02 1.46288065643e-02 1.79740052840e-02 + 2.21383854932e-02 2.52442663181e-02 2.71139620581e-02 2.74766558578e-02 2.69079511827e-02 2.65223600849e-02 + 2.69079511826e-02 2.74766558578e-02 2.71139620581e-02 2.52442663181e-02 2.21383854932e-02 1.85131064743e-02 + 1.50423980149e-02 1.21045641480e-02 9.78230850882e-03 8.04465702446e-03 6.92808975974e-03 6.53733030468e-03 + 6.92808975973e-03 8.04465702445e-03 9.78230850882e-03 1.21045641480e-02 1.50423980149e-02 1.85131064743e-02 + 2.33116271285e-02 2.55789017267e-02 2.62349275771e-02 2.54923302762e-02 2.45136072589e-02 2.45136072589e-02 + 2.54923302762e-02 2.62349275771e-02 2.55789017267e-02 2.33116271285e-02 1.99974571615e-02 1.63905844408e-02 + 1.30486211596e-02 1.02402112845e-02 8.04465702446e-03 6.50344737139e-03 5.69415565947e-03 5.69415565947e-03 + 6.50344737139e-03 8.04465702445e-03 1.02402112845e-02 1.30486211596e-02 1.63905844408e-02 1.99974571616e-02 + 2.62414574798e-02 2.75583653887e-02 2.69079511826e-02 2.52952680324e-02 2.44576079909e-02 2.52952680324e-02 + 2.69079511827e-02 2.75583653887e-02 2.62414574798e-02 2.31387782150e-02 1.91106183945e-02 1.50739753351e-02 + 1.16096263393e-02 8.89928200875e-03 6.92808975973e-03 5.69415565947e-03 5.26655497605e-03 5.69415565947e-03 + 6.92808975974e-03 8.89928200875e-03 1.16096263393e-02 1.50739753351e-02 1.91106183945e-02 2.31387782150e-02 + 3.06060385056e-02 3.03790382414e-02 2.81921518485e-02 2.61163074370e-02 2.61163074370e-02 2.81921518485e-02 + 3.03790382414e-02 3.06060385055e-02 2.81654297894e-02 2.37302554132e-02 1.86355831047e-02 1.40807237278e-02 + 1.06052341643e-02 8.15763089846e-03 6.53733030468e-03 5.69415565947e-03 5.69415565947e-03 6.53733030468e-03 + 8.15763089847e-03 1.06052341643e-02 1.40807237278e-02 1.86355831047e-02 2.37302554132e-02 2.81654297894e-02 + 3.48685420831e-02 3.31552867935e-02 3.05271423180e-02 2.92886940372e-02 3.05271423180e-02 3.31552867935e-02 + 3.48685420831e-02 3.38926655663e-02 2.99300705585e-02 2.40966608398e-02 1.81617249718e-02 1.34421984781e-02 + 1.02249435892e-02 8.15763089846e-03 6.92808975973e-03 6.50344737139e-03 6.92808975974e-03 8.15763089847e-03 + 1.02249435892e-02 1.34421984781e-02 1.81617249718e-02 2.40966608398e-02 2.99300705585e-02 3.38926655663e-02 + 4.15728345981e-02 4.23899114590e-02 4.26144257223e-02 4.26144257223e-02 4.23899114590e-02 4.15728345981e-02 + 3.97328731662e-02 3.62670291685e-02 3.07818020073e-02 2.40902041017e-02 1.79266369346e-02 1.34421984781e-02 + 1.06052341643e-02 8.89928200875e-03 8.04465702445e-03 8.04465702446e-03 8.89928200875e-03 1.06052341643e-02 + 1.34421984781e-02 1.79266369346e-02 2.40902041017e-02 3.07818020073e-02 3.62670291685e-02 3.97328731661e-02 + 6.13627808209e-02 6.87952783262e-02 7.14597362941e-02 6.87952783262e-02 6.13627808209e-02 5.18159940627e-02 + 4.36602554332e-02 3.73533616181e-02 3.09603631544e-02 2.40902041017e-02 1.81617249718e-02 1.40807237278e-02 + 1.16096263393e-02 1.02402112845e-02 9.78230850882e-03 1.02402112845e-02 1.16096263393e-02 1.40807237278e-02 + 1.81617249718e-02 2.40902041017e-02 3.09603631544e-02 3.73533616181e-02 4.36602554332e-02 5.18159940627e-02 + 9.46627422162e-02 1.02256358961e-01 1.02256358961e-01 9.46627422163e-02 7.81199920260e-02 5.87904239655e-02 + 4.51573443066e-02 3.73533616181e-02 3.07818020073e-02 2.40966608398e-02 1.86355831047e-02 1.50739753351e-02 + 1.30486211596e-02 1.21045641480e-02 1.21045641480e-02 1.30486211596e-02 1.50739753351e-02 1.86355831047e-02 + 2.40966608398e-02 3.07818020073e-02 3.73533616181e-02 4.51573443066e-02 5.87904239655e-02 7.81199920260e-02 + 1.14984832163e-01 1.14285203523e-01 1.14984832163e-01 1.07505424132e-01 8.44125475991e-02 5.87904239654e-02 + 4.36602554332e-02 3.62670291685e-02 2.99300705585e-02 2.37302554132e-02 1.91106183945e-02 1.63905844408e-02 + 1.50423980149e-02 1.46288065643e-02 1.50423980149e-02 1.63905844408e-02 1.91106183945e-02 2.37302554132e-02 + 2.99300705585e-02 3.62670291685e-02 4.36602554332e-02 5.87904239655e-02 8.44125475992e-02 1.07505424132e-01 + 1.05270975780e-01 1.05270975780e-01 1.15477102967e-01 1.07505424132e-01 7.81199920260e-02 5.18159940627e-02 + 3.97328731661e-02 3.38926655663e-02 2.81654297894e-02 2.31387782150e-02 1.99974571616e-02 1.85131064743e-02 + 1.79740052840e-02 1.79740052840e-02 1.85131064743e-02 1.99974571616e-02 2.31387782150e-02 2.81654297894e-02 + 3.38926655663e-02 3.97328731662e-02 5.18159940627e-02 7.81199920261e-02 1.07505424132e-01 1.15477102967e-01 + 1.05270975780e-01 1.14285203523e-01 1.02256358961e-01 6.87952783262e-02 4.23899114590e-02 3.31552867935e-02 + 3.03790382414e-02 2.75583653887e-02 2.55789017267e-02 2.52442663181e-02 2.59180287237e-02 2.65859681267e-02 + 2.65859681267e-02 2.59180287237e-02 2.52442663181e-02 2.55789017267e-02 2.75583653887e-02 3.03790382414e-02 + 3.31552867935e-02 4.23899114590e-02 6.87952783262e-02 1.02256358961e-01 1.14285203523e-01 1.05270975780e-01 + 1.14285203523e-01 1.05383989813e-01 7.44580546054e-02 4.30325583083e-02 2.95747487953e-02 2.73937743101e-02 + 2.65223600849e-02 2.61137483222e-02 2.72047829307e-02 2.92642577664e-02 3.12137668085e-02 3.20076435874e-02 + 3.12137668085e-02 2.92642577664e-02 2.72047829307e-02 2.61137483222e-02 2.65223600849e-02 2.73937743101e-02 + 2.95747487953e-02 4.30325583083e-02 7.44580546055e-02 1.05383989813e-01 1.14285203523e-01 1.12133532869e-01 + 1.02256358961e-01 7.44580546054e-02 4.31923684876e-02 2.67211542468e-02 2.35389487572e-02 2.36140048007e-02 + 2.45136072590e-02 2.70509290066e-02 3.05546433144e-02 3.39946385756e-02 3.61943858423e-02 3.61943858423e-02 + 3.39946385756e-02 3.05546433144e-02 2.70509290066e-02 2.45136072589e-02 2.36140048007e-02 2.35389487571e-02 + 2.67211542468e-02 4.31923684875e-02 7.44580546055e-02 1.02256358961e-01 1.13480556613e-01 1.13480556613e-01 + 6.87952783262e-02 4.30325583083e-02 2.67211542468e-02 2.17745697719e-02 2.10845164455e-02 2.20784329209e-02 + 2.52952680324e-02 2.97972897260e-02 3.46097554135e-02 3.86301228830e-02 4.02274592299e-02 3.86301228830e-02 + 3.46097554135e-02 2.97972897261e-02 2.52952680324e-02 2.20784329209e-02 2.10845164455e-02 2.17745697719e-02 + 2.67211542468e-02 4.30325583083e-02 6.87952783262e-02 9.00462898609e-02 9.74791227834e-02 9.00462898608e-02 + 4.23899114590e-02 2.95747487953e-02 2.35389487571e-02 2.10845164455e-02 2.09113970026e-02 2.36140048007e-02 + 2.81921518484e-02 3.39197128099e-02 4.00844862497e-02 4.42642004890e-02 4.42642004891e-02 4.00844862497e-02 + 3.39197128100e-02 2.81921518485e-02 2.36140048007e-02 2.09113970026e-02 2.10845164455e-02 2.35389487571e-02 + 2.95747487953e-02 4.23899114590e-02 5.87813204091e-02 6.98261770600e-02 6.98261770600e-02 5.87813204090e-02 + 3.31552867934e-02 2.73937743101e-02 2.36140048007e-02 2.20784329209e-02 2.36140048007e-02 2.73937743101e-02 + 3.31552867934e-02 4.11390439154e-02 4.86719445336e-02 5.17384028835e-02 4.86719445337e-02 4.11390439155e-02 + 3.31552867935e-02 2.73937743101e-02 2.36140048007e-02 2.20784329209e-02 2.36140048007e-02 2.73937743101e-02 + 3.31552867935e-02 4.11390439155e-02 4.86719445337e-02 5.17384028836e-02 4.86719445336e-02 4.11390439155e-02 + 3.03790382414e-02 2.65223600849e-02 2.45136072590e-02 2.52952680324e-02 2.81921518484e-02 3.31552867934e-02 + 4.15728345980e-02 5.18159940626e-02 5.87904239654e-02 5.87904239654e-02 5.18159940627e-02 4.15728345981e-02 + 3.31552867935e-02 2.81921518485e-02 2.52952680324e-02 2.45136072589e-02 2.65223600849e-02 3.03790382414e-02 + 3.48629669819e-02 3.89797931478e-02 4.14580711769e-02 4.14580711769e-02 3.89797931477e-02 3.48629669819e-02 + 2.75583653887e-02 2.61137483222e-02 2.70509290066e-02 2.97972897260e-02 3.39197128099e-02 4.11390439154e-02 + 5.18159940626e-02 6.17010347926e-02 6.56429700804e-02 6.17010347927e-02 5.18159940627e-02 4.11390439155e-02 + 3.39197128099e-02 2.97972897260e-02 2.70509290066e-02 2.61137483222e-02 2.75583653887e-02 3.04051608500e-02 + 3.31962688978e-02 3.49765243617e-02 3.55400935242e-02 3.49765243617e-02 3.31962688978e-02 3.04051608500e-02 + 2.55789017267e-02 2.72047829307e-02 3.05546433144e-02 3.46097554135e-02 4.00844862496e-02 4.86719445336e-02 + 5.87904239654e-02 6.56429700804e-02 6.56429700805e-02 5.87904239655e-02 4.86719445337e-02 4.00844862497e-02 + 3.46097554135e-02 3.05546433144e-02 2.72047829307e-02 2.55789017267e-02 2.59617964748e-02 2.73091130234e-02 + 2.84215335265e-02 2.88774182856e-02 2.88774182856e-02 2.84215335265e-02 2.73091130234e-02 2.59617964748e-02 + 2.52442663181e-02 2.92642577664e-02 3.39946385756e-02 3.86301228830e-02 4.42642004890e-02 5.17384028835e-02 + 5.87904239654e-02 6.17010347926e-02 5.87904239655e-02 5.17384028836e-02 4.42642004891e-02 3.86301228831e-02 + 3.39946385756e-02 2.92642577664e-02 2.52442663181e-02 2.30251854758e-02 2.24043544384e-02 2.23752062804e-02 + 2.22711776717e-02 2.21707375456e-02 2.22711776717e-02 2.23752062804e-02 2.24043544384e-02 2.30251854758e-02 + 2.59180287237e-02 3.12137668085e-02 3.61943858422e-02 4.02274592299e-02 4.42642004890e-02 4.86719445337e-02 + 5.18159940627e-02 5.18159940627e-02 4.86719445337e-02 4.42642004891e-02 4.02274592300e-02 3.61943858423e-02 + 3.12137668086e-02 2.59180287237e-02 2.17946809311e-02 1.93870166210e-02 1.81179478705e-02 1.72612102704e-02 + 1.66949823920e-02 1.66949823920e-02 1.72612102705e-02 1.81179478705e-02 1.93870166210e-02 2.17946809311e-02 + 2.65859681267e-02 3.20076435873e-02 3.61943858423e-02 3.86301228830e-02 4.00844862497e-02 4.11390439155e-02 + 4.15728345981e-02 4.11390439155e-02 4.00844862497e-02 3.86301228830e-02 3.61943858423e-02 3.20076435874e-02 + 2.65859681267e-02 2.15194519516e-02 1.78887124933e-02 1.56224103699e-02 1.41150555464e-02 1.30619533075e-02 + 1.26512009150e-02 1.30619533075e-02 1.41150555464e-02 1.56224103699e-02 1.78887124933e-02 2.15194519516e-02 + 2.65859681267e-02 3.12137668085e-02 3.39946385756e-02 3.46097554135e-02 3.39197128100e-02 3.31552867935e-02 + 3.31552867935e-02 3.39197128099e-02 3.46097554135e-02 3.39946385756e-02 3.12137668086e-02 2.65859681267e-02 + 2.15037112468e-02 1.72810954144e-02 1.43188156133e-02 1.22935034476e-02 1.08501521193e-02 1.00211358334e-02 + 1.00211358334e-02 1.08501521193e-02 1.22935034476e-02 1.43188156133e-02 1.72810954144e-02 2.15037112467e-02 + 2.59180287237e-02 2.92642577664e-02 3.05546433144e-02 2.97972897261e-02 2.81921518485e-02 2.73937743101e-02 + 2.81921518485e-02 2.97972897260e-02 3.05546433144e-02 2.92642577664e-02 2.59180287237e-02 2.15194519516e-02 + 1.72810954144e-02 1.38961510916e-02 1.14032634249e-02 9.57630048387e-03 8.36367529240e-03 7.92503968743e-03 + 8.36367529240e-03 9.57630048387e-03 1.14032634249e-02 1.38961510916e-02 1.72810954144e-02 2.15194519516e-02 + 2.52442663181e-02 2.72047829307e-02 2.70509290066e-02 2.52952680324e-02 2.36140048007e-02 2.36140048007e-02 + 2.52952680324e-02 2.70509290066e-02 2.72047829307e-02 2.52442663181e-02 2.17946809311e-02 1.78887124933e-02 + 1.43188156133e-02 1.14032634249e-02 9.13735254995e-03 7.50459533106e-03 6.62092791970e-03 6.62092791970e-03 + 7.50459533106e-03 9.13735254995e-03 1.14032634249e-02 1.43188156133e-02 1.78887124933e-02 2.17946809311e-02 + 2.55789017267e-02 2.61137483222e-02 2.45136072590e-02 2.20784329209e-02 2.09113970026e-02 2.20784329209e-02 + 2.45136072589e-02 2.61137483222e-02 2.55789017267e-02 2.30251854758e-02 1.93870166210e-02 1.56224103699e-02 + 1.22935034476e-02 9.57630048387e-03 7.50459533106e-03 6.16390944839e-03 5.69332826895e-03 6.16390944840e-03 + 7.50459533106e-03 9.57630048387e-03 1.22935034476e-02 1.56224103699e-02 1.93870166210e-02 2.30251854758e-02 + 2.75583653887e-02 2.65223600849e-02 2.36140048007e-02 2.10845164455e-02 2.10845164455e-02 2.36140048007e-02 + 2.65223600849e-02 2.75583653887e-02 2.59617964748e-02 2.24043544384e-02 1.81179478705e-02 1.41150555464e-02 + 1.08501521193e-02 8.36367529240e-03 6.62092791970e-03 5.69332826895e-03 5.69332826895e-03 6.62092791970e-03 + 8.36367529240e-03 1.08501521193e-02 1.41150555464e-02 1.81179478705e-02 2.24043544384e-02 2.59617964748e-02 + 3.03790382414e-02 2.73937743101e-02 2.35389487571e-02 2.17745697719e-02 2.35389487571e-02 2.73937743101e-02 + 3.03790382414e-02 3.04051608500e-02 2.73091130234e-02 2.23752062804e-02 1.72612102704e-02 1.30619533075e-02 + 1.00211358334e-02 7.92503968743e-03 6.62092791970e-03 6.16390944840e-03 6.62092791971e-03 7.92503968743e-03 + 1.00211358334e-02 1.30619533075e-02 1.72612102705e-02 2.23752062804e-02 2.73091130234e-02 3.04051608500e-02 + 3.31552867935e-02 2.95747487953e-02 2.67211542468e-02 2.67211542468e-02 2.95747487953e-02 3.31552867935e-02 + 3.48629669819e-02 3.31962688978e-02 2.84215335265e-02 2.22711776717e-02 1.66949823920e-02 1.26512009150e-02 + 1.00211358334e-02 8.36367529240e-03 7.50459533106e-03 7.50459533106e-03 8.36367529240e-03 1.00211358334e-02 + 1.26512009150e-02 1.66949823920e-02 2.22711776717e-02 2.84215335265e-02 3.31962688978e-02 3.48629669819e-02 + 4.23899114590e-02 4.30325583083e-02 4.31923684876e-02 4.30325583083e-02 4.23899114590e-02 4.11390439155e-02 + 3.89797931478e-02 3.49765243617e-02 2.88774182856e-02 2.21707375456e-02 1.66949823920e-02 1.30619533075e-02 + 1.08501521193e-02 9.57630048387e-03 9.13735254995e-03 9.57630048387e-03 1.08501521193e-02 1.30619533075e-02 + 1.66949823920e-02 2.21707375456e-02 2.88774182856e-02 3.49765243617e-02 3.89797931478e-02 4.11390439155e-02 + 6.87952783262e-02 7.44580546055e-02 7.44580546055e-02 6.87952783262e-02 5.87813204091e-02 4.86719445337e-02 + 4.14580711769e-02 3.55400935242e-02 2.88774182856e-02 2.22711776717e-02 1.72612102705e-02 1.41150555464e-02 + 1.22935034476e-02 1.14032634249e-02 1.14032634249e-02 1.22935034476e-02 1.41150555464e-02 1.72612102705e-02 + 2.22711776717e-02 2.88774182856e-02 3.55400935242e-02 4.14580711769e-02 4.86719445337e-02 5.87813204091e-02 + 1.02256358961e-01 1.05383989813e-01 1.02256358961e-01 9.00462898609e-02 6.98261770600e-02 5.17384028836e-02 + 4.14580711769e-02 3.49765243617e-02 2.84215335265e-02 2.23752062804e-02 1.81179478705e-02 1.56224103699e-02 + 1.43188156133e-02 1.38961510916e-02 1.43188156133e-02 1.56224103699e-02 1.81179478705e-02 2.23752062804e-02 + 2.84215335265e-02 3.49765243617e-02 4.14580711769e-02 5.17384028836e-02 6.98261770601e-02 9.00462898609e-02 + 1.14285203523e-01 1.14285203523e-01 1.13480556613e-01 9.74791227833e-02 6.98261770600e-02 4.86719445337e-02 + 3.89797931477e-02 3.31962688978e-02 2.73091130234e-02 2.24043544384e-02 1.93870166210e-02 1.78887124933e-02 + 1.72810954144e-02 1.72810954144e-02 1.78887124933e-02 1.93870166210e-02 2.24043544384e-02 2.73091130234e-02 + 3.31962688978e-02 3.89797931478e-02 4.86719445337e-02 6.98261770601e-02 9.74791227834e-02 1.13480556613e-01 + 1.05270975780e-01 1.12133532869e-01 1.13480556613e-01 9.00462898608e-02 5.87813204090e-02 4.11390439155e-02 + 3.48629669819e-02 3.04051608500e-02 2.59617964748e-02 2.30251854758e-02 2.17946809311e-02 2.15194519516e-02 + 2.15037112468e-02 2.15194519516e-02 2.17946809311e-02 2.30251854758e-02 2.59617964748e-02 3.04051608500e-02 + 3.48629669819e-02 4.11390439155e-02 5.87813204091e-02 9.00462898609e-02 1.13480556613e-01 1.12133532870e-01 + 1.14984832163e-01 1.02256358961e-01 7.14597362941e-02 4.26144257223e-02 3.05271423180e-02 2.81921518485e-02 + 2.69079511827e-02 2.62349275771e-02 2.71139620581e-02 2.89472257765e-02 3.06730045455e-02 3.13659743854e-02 + 3.06730045455e-02 2.89472257765e-02 2.71139620581e-02 2.62349275771e-02 2.69079511826e-02 2.81921518485e-02 + 3.05271423180e-02 4.26144257223e-02 7.14597362942e-02 1.02256358961e-01 1.14984832163e-01 1.15477102967e-01 + 1.02256358961e-01 7.44580546054e-02 4.31923684876e-02 2.67211542468e-02 2.35389487571e-02 2.36140048007e-02 + 2.45136072590e-02 2.70509290066e-02 3.05546433144e-02 3.39946385756e-02 3.61943858423e-02 3.61943858423e-02 + 3.39946385756e-02 3.05546433144e-02 2.70509290066e-02 2.45136072589e-02 2.36140048007e-02 2.35389487571e-02 + 2.67211542468e-02 4.31923684875e-02 7.44580546055e-02 1.02256358961e-01 1.13480556613e-01 1.13480556613e-01 + 7.14597362941e-02 4.31923684876e-02 2.50655517889e-02 1.98434626957e-02 1.95330017723e-02 2.09113970026e-02 + 2.44576079909e-02 2.92434029240e-02 3.44218462468e-02 3.88820048627e-02 4.06895832708e-02 3.88820048628e-02 + 3.44218462469e-02 2.92434029240e-02 2.44576079909e-02 2.09113970026e-02 1.95330017723e-02 1.98434626957e-02 + 2.50655517888e-02 4.31923684875e-02 7.14597362942e-02 9.40547983347e-02 1.01689797158e-01 9.40547983346e-02 + 4.26144257223e-02 2.67211542468e-02 1.98434626957e-02 1.76765157693e-02 1.79303755335e-02 2.10845164455e-02 + 2.61163074370e-02 3.26919751551e-02 4.03066875222e-02 4.56997211400e-02 4.56997211400e-02 4.03066875223e-02 + 3.26919751552e-02 2.61163074370e-02 2.10845164455e-02 1.79303755335e-02 1.76765157693e-02 1.98434626957e-02 + 2.67211542468e-02 4.26144257223e-02 6.30067497702e-02 7.65079897136e-02 7.65079897136e-02 6.30067497701e-02 + 3.05271423180e-02 2.35389487571e-02 1.95330017723e-02 1.79303755335e-02 1.95330017723e-02 2.35389487571e-02 + 3.05271423180e-02 4.15214590889e-02 5.24377104593e-02 5.69445178737e-02 5.24377104593e-02 4.15214590890e-02 + 3.05271423180e-02 2.35389487571e-02 1.95330017723e-02 1.79303755334e-02 1.95330017723e-02 2.35389487571e-02 + 3.05271423180e-02 4.15214590889e-02 5.24377104594e-02 5.69445178737e-02 5.24377104593e-02 4.15214590889e-02 + 2.81921518485e-02 2.36140048007e-02 2.09113970026e-02 2.10845164455e-02 2.35389487571e-02 2.95747487953e-02 + 4.23899114589e-02 5.87813204090e-02 6.98261770600e-02 6.98261770600e-02 5.87813204091e-02 4.23899114590e-02 + 2.95747487953e-02 2.35389487571e-02 2.10845164455e-02 2.09113970026e-02 2.36140048007e-02 2.81921518485e-02 + 3.39197128100e-02 4.00844862497e-02 4.42642004891e-02 4.42642004890e-02 4.00844862497e-02 3.39197128099e-02 + 2.69079511827e-02 2.45136072590e-02 2.44576079909e-02 2.61163074370e-02 3.05271423180e-02 4.23899114589e-02 + 6.13627808208e-02 7.81199920259e-02 8.44125475991e-02 7.81199920260e-02 6.13627808209e-02 4.23899114590e-02 + 3.05271423180e-02 2.61163074370e-02 2.44576079908e-02 2.45136072589e-02 2.69079511826e-02 3.06791433435e-02 + 3.45751721123e-02 3.75326780270e-02 3.86294226544e-02 3.75326780270e-02 3.45751721122e-02 3.06791433435e-02 + 2.62349275771e-02 2.70509290066e-02 2.92434029240e-02 3.26919751551e-02 4.15214590888e-02 5.87813204089e-02 + 7.81199920259e-02 8.97670392509e-02 8.97670392509e-02 7.81199920261e-02 5.87813204091e-02 4.15214590889e-02 + 3.26919751551e-02 2.92434029240e-02 2.70509290066e-02 2.62349275771e-02 2.73901656280e-02 2.95889063006e-02 + 3.16055353689e-02 3.27027225515e-02 3.27027225515e-02 3.16055353689e-02 2.95889063006e-02 2.73901656280e-02 + 2.71139620581e-02 3.05546433144e-02 3.44218462468e-02 4.03066875222e-02 5.24377104592e-02 6.98261770599e-02 + 8.44125475990e-02 8.97670392509e-02 8.44125475992e-02 6.98261770601e-02 5.24377104594e-02 4.03066875223e-02 + 3.44218462469e-02 3.05546433144e-02 2.71139620581e-02 2.51928884334e-02 2.49962561202e-02 2.55724702285e-02 + 2.60490352762e-02 2.61899431515e-02 2.60490352762e-02 2.55724702285e-02 2.49962561202e-02 2.51928884334e-02 + 2.89472257765e-02 3.39946385756e-02 3.88820048627e-02 4.56997211400e-02 5.69445178736e-02 6.98261770600e-02 + 7.81199920260e-02 7.81199920260e-02 6.98261770601e-02 5.69445178738e-02 4.56997211401e-02 3.88820048628e-02 + 3.39946385756e-02 2.89472257765e-02 2.46284218885e-02 2.20776282425e-02 2.09446150107e-02 2.03920068264e-02 + 2.00581600358e-02 2.00581600358e-02 2.03920068264e-02 2.09446150107e-02 2.20776282425e-02 2.46284218885e-02 + 3.06730045455e-02 3.61943858422e-02 4.06895832708e-02 4.56997211400e-02 5.24377104593e-02 5.87813204090e-02 + 6.13627808209e-02 5.87813204091e-02 5.24377104594e-02 4.56997211401e-02 4.06895832709e-02 3.61943858423e-02 + 3.06730045455e-02 2.49901046905e-02 2.07195371247e-02 1.81502745082e-02 1.66276495780e-02 1.56490177739e-02 + 1.52718979502e-02 1.56490177739e-02 1.66276495780e-02 1.81502745082e-02 2.07195371247e-02 2.49901046905e-02 + 3.13659743854e-02 3.61943858423e-02 3.88820048628e-02 4.03066875223e-02 4.15214590889e-02 4.23899114590e-02 + 4.23899114590e-02 4.15214590889e-02 4.03066875223e-02 3.88820048628e-02 3.61943858423e-02 3.13659743854e-02 + 2.54054025139e-02 2.02039597509e-02 1.66389072281e-02 1.43895601245e-02 1.28938940154e-02 1.20440164119e-02 + 1.20440164119e-02 1.28938940154e-02 1.43895601245e-02 1.66389072281e-02 2.02039597509e-02 2.54054025138e-02 + 3.06730045455e-02 3.39946385756e-02 3.44218462469e-02 3.26919751552e-02 3.05271423180e-02 2.95747487953e-02 + 3.05271423180e-02 3.26919751551e-02 3.44218462469e-02 3.39946385756e-02 3.06730045455e-02 2.54054025139e-02 + 2.00652756289e-02 1.59286730225e-02 1.31235074234e-02 1.12138690779e-02 9.96244247360e-03 9.50257178746e-03 + 9.96244247360e-03 1.12138690779e-02 1.31235074234e-02 1.59286730225e-02 2.00652756288e-02 2.54054025138e-02 + 2.89472257765e-02 3.05546433144e-02 2.92434029240e-02 2.61163074370e-02 2.35389487571e-02 2.35389487571e-02 + 2.61163074370e-02 2.92434029240e-02 3.05546433144e-02 2.89472257765e-02 2.49901046905e-02 2.02039597509e-02 + 1.59286730225e-02 1.26855892424e-02 1.03423481784e-02 8.68885911334e-03 7.77919418380e-03 7.77919418380e-03 + 8.68885911334e-03 1.03423481784e-02 1.26855892423e-02 1.59286730225e-02 2.02039597509e-02 2.49901046905e-02 + 2.71139620581e-02 2.70509290066e-02 2.44576079909e-02 2.10845164455e-02 1.95330017723e-02 2.10845164455e-02 + 2.44576079908e-02 2.70509290066e-02 2.71139620581e-02 2.46284218885e-02 2.07195371247e-02 1.66389072281e-02 + 1.31235074234e-02 1.03423481784e-02 8.23794392376e-03 6.85673718666e-03 6.36560105343e-03 6.85673718666e-03 + 8.23794392376e-03 1.03423481784e-02 1.31235074234e-02 1.66389072281e-02 2.07195371247e-02 2.46284218885e-02 + 2.62349275771e-02 2.45136072590e-02 2.09113970026e-02 1.79303755335e-02 1.79303755335e-02 2.09113970026e-02 + 2.45136072589e-02 2.62349275771e-02 2.51928884334e-02 2.20776282425e-02 1.81502745082e-02 1.43895601245e-02 + 1.12138690779e-02 8.68885911334e-03 6.85673718666e-03 5.86554131392e-03 5.86554131392e-03 6.85673718666e-03 + 8.68885911334e-03 1.12138690779e-02 1.43895601245e-02 1.81502745082e-02 2.20776282425e-02 2.51928884334e-02 + 2.69079511826e-02 2.36140048007e-02 1.95330017723e-02 1.76765157693e-02 1.95330017723e-02 2.36140048007e-02 + 2.69079511827e-02 2.73901656281e-02 2.49962561202e-02 2.09446150107e-02 1.66276495780e-02 1.28938940154e-02 + 9.96244247360e-03 7.77919418380e-03 6.36560105343e-03 5.86554131392e-03 6.36560105343e-03 7.77919418380e-03 + 9.96244247360e-03 1.28938940154e-02 1.66276495780e-02 2.09446150107e-02 2.49962561202e-02 2.73901656280e-02 + 2.81921518485e-02 2.35389487571e-02 1.98434626957e-02 1.98434626957e-02 2.35389487571e-02 2.81921518485e-02 + 3.06791433435e-02 2.95889063006e-02 2.55724702285e-02 2.03920068264e-02 1.56490177739e-02 1.20440164119e-02 + 9.50257178746e-03 7.77919418380e-03 6.85673718666e-03 6.85673718666e-03 7.77919418380e-03 9.50257178746e-03 + 1.20440164119e-02 1.56490177739e-02 2.03920068264e-02 2.55724702285e-02 2.95889063006e-02 3.06791433435e-02 + 3.05271423180e-02 2.67211542468e-02 2.50655517888e-02 2.67211542468e-02 3.05271423180e-02 3.39197128100e-02 + 3.45751721123e-02 3.16055353689e-02 2.60490352762e-02 2.00581600358e-02 1.52718979502e-02 1.20440164119e-02 + 9.96244247360e-03 8.68885911334e-03 8.23794392376e-03 8.68885911334e-03 9.96244247360e-03 1.20440164119e-02 + 1.52718979502e-02 2.00581600358e-02 2.60490352763e-02 3.16055353689e-02 3.45751721123e-02 3.39197128099e-02 + 4.26144257223e-02 4.31923684875e-02 4.31923684876e-02 4.26144257223e-02 4.15214590889e-02 4.00844862497e-02 + 3.75326780270e-02 3.27027225515e-02 2.61899431515e-02 2.00581600358e-02 1.56490177739e-02 1.28938940154e-02 + 1.12138690779e-02 1.03423481784e-02 1.03423481784e-02 1.12138690779e-02 1.28938940154e-02 1.56490177739e-02 + 2.00581600358e-02 2.61899431515e-02 3.27027225516e-02 3.75326780271e-02 4.00844862497e-02 4.15214590889e-02 + 7.14597362942e-02 7.44580546055e-02 7.14597362942e-02 6.30067497702e-02 5.24377104593e-02 4.42642004891e-02 + 3.86294226543e-02 3.27027225515e-02 2.60490352762e-02 2.03920068264e-02 1.66276495780e-02 1.43895601245e-02 + 1.31235074234e-02 1.26855892424e-02 1.31235074234e-02 1.43895601245e-02 1.66276495780e-02 2.03920068264e-02 + 2.60490352763e-02 3.27027225516e-02 3.86294226544e-02 4.42642004891e-02 5.24377104594e-02 6.30067497702e-02 + 1.02256358961e-01 1.02256358961e-01 9.40547983346e-02 7.65079897136e-02 5.69445178737e-02 4.42642004890e-02 + 3.75326780270e-02 3.16055353689e-02 2.55724702285e-02 2.09446150107e-02 1.81502745082e-02 1.66389072281e-02 + 1.59286730225e-02 1.59286730225e-02 1.66389072281e-02 1.81502745082e-02 2.09446150107e-02 2.55724702285e-02 + 3.16055353689e-02 3.75326780271e-02 4.42642004891e-02 5.69445178738e-02 7.65079897137e-02 9.40547983347e-02 + 1.14984832163e-01 1.13480556613e-01 1.01689797158e-01 7.65079897135e-02 5.24377104593e-02 4.00844862497e-02 + 3.45751721122e-02 2.95889063006e-02 2.49962561202e-02 2.20776282425e-02 2.07195371247e-02 2.02039597509e-02 + 2.00652756289e-02 2.02039597509e-02 2.07195371247e-02 2.20776282425e-02 2.49962561202e-02 2.95889063006e-02 + 3.45751721123e-02 4.00844862497e-02 5.24377104594e-02 7.65079897137e-02 1.01689797158e-01 1.13480556613e-01 + 1.15477102967e-01 1.13480556613e-01 9.40547983345e-02 6.30067497701e-02 4.15214590889e-02 3.39197128099e-02 + 3.06791433435e-02 2.73901656280e-02 2.51928884334e-02 2.46284218885e-02 2.49901046905e-02 2.54054025138e-02 + 2.54054025138e-02 2.49901046905e-02 2.46284218885e-02 2.51928884334e-02 2.73901656280e-02 3.06791433435e-02 + 3.39197128099e-02 4.15214590889e-02 6.30067497702e-02 9.40547983347e-02 1.13480556613e-01 1.15477102967e-01 + 9.46627422161e-02 6.87952783261e-02 4.26144257223e-02 2.92886940371e-02 2.61163074370e-02 2.52952680324e-02 + 2.54923302762e-02 2.74766558578e-02 3.04771749334e-02 3.33648425996e-02 3.51265406940e-02 3.51265406940e-02 + 3.33648425996e-02 3.04771749334e-02 2.74766558578e-02 2.54923302762e-02 2.52952680324e-02 2.61163074370e-02 + 2.92886940371e-02 4.26144257223e-02 6.87952783262e-02 9.46627422163e-02 1.07505424132e-01 1.07505424132e-01 + 6.87952783261e-02 4.30325583083e-02 2.67211542468e-02 2.17745697719e-02 2.10845164455e-02 2.20784329209e-02 + 2.52952680324e-02 2.97972897261e-02 3.46097554135e-02 3.86301228830e-02 4.02274592299e-02 3.86301228830e-02 + 3.46097554135e-02 2.97972897261e-02 2.52952680324e-02 2.20784329209e-02 2.10845164455e-02 2.17745697719e-02 + 2.67211542468e-02 4.30325583083e-02 6.87952783262e-02 9.00462898609e-02 9.74791227833e-02 9.00462898608e-02 + 4.26144257223e-02 2.67211542468e-02 1.98434626957e-02 1.76765157693e-02 1.79303755335e-02 2.10845164455e-02 + 2.61163074370e-02 3.26919751551e-02 4.03066875223e-02 4.56997211400e-02 4.56997211401e-02 4.03066875223e-02 + 3.26919751552e-02 2.61163074370e-02 2.10845164455e-02 1.79303755334e-02 1.76765157693e-02 1.98434626957e-02 + 2.67211542468e-02 4.26144257223e-02 6.30067497702e-02 7.65079897136e-02 7.65079897136e-02 6.30067497701e-02 + 2.92886940371e-02 2.17745697719e-02 1.76765157693e-02 1.60511803073e-02 1.76765157693e-02 2.17745697719e-02 + 2.92886940371e-02 4.15474648474e-02 5.38559646463e-02 5.89470993596e-02 5.38559646464e-02 4.15474648475e-02 + 2.92886940372e-02 2.17745697719e-02 1.76765157693e-02 1.60511803073e-02 1.76765157693e-02 2.17745697719e-02 + 2.92886940372e-02 4.15474648475e-02 5.38559646464e-02 5.89470993596e-02 5.38559646463e-02 4.15474648474e-02 + 2.61163074370e-02 2.10845164455e-02 1.79303755335e-02 1.76765157693e-02 1.98434626957e-02 2.67211542468e-02 + 4.26144257223e-02 6.30067497701e-02 7.65079897136e-02 7.65079897136e-02 6.30067497702e-02 4.26144257223e-02 + 2.67211542468e-02 1.98434626957e-02 1.76765157693e-02 1.79303755334e-02 2.10845164455e-02 2.61163074370e-02 + 3.26919751552e-02 4.03066875223e-02 4.56997211401e-02 4.56997211400e-02 4.03066875223e-02 3.26919751551e-02 + 2.52952680324e-02 2.20784329209e-02 2.10845164455e-02 2.17745697719e-02 2.67211542468e-02 4.30325583083e-02 + 6.87952783261e-02 9.00462898608e-02 9.74791227833e-02 9.00462898609e-02 6.87952783262e-02 4.30325583083e-02 + 2.67211542468e-02 2.17745697719e-02 2.10845164455e-02 2.20784329209e-02 2.52952680324e-02 2.97972897261e-02 + 3.46097554135e-02 3.86301228830e-02 4.02274592299e-02 3.86301228830e-02 3.46097554135e-02 2.97972897261e-02 + 2.54923302762e-02 2.52952680324e-02 2.61163074370e-02 2.92886940371e-02 4.26144257223e-02 6.87952783261e-02 + 9.46627422161e-02 1.07505424132e-01 1.07505424132e-01 9.46627422163e-02 6.87952783262e-02 4.26144257223e-02 + 2.92886940371e-02 2.61163074370e-02 2.52952680324e-02 2.54923302762e-02 2.74766558578e-02 3.04771749334e-02 + 3.33648425996e-02 3.51265406940e-02 3.51265406940e-02 3.33648425996e-02 3.04771749334e-02 2.74766558578e-02 + 2.74766558578e-02 2.97972897261e-02 3.26919751551e-02 4.15474648474e-02 6.30067497701e-02 9.00462898607e-02 + 1.07505424132e-01 1.12226003113e-01 1.07505424132e-01 9.00462898609e-02 6.30067497702e-02 4.15474648475e-02 + 3.26919751551e-02 2.97972897260e-02 2.74766558578e-02 2.61753235783e-02 2.65397992490e-02 2.78012163806e-02 + 2.89589200657e-02 2.93978765598e-02 2.89589200656e-02 2.78012163806e-02 2.65397992490e-02 2.61753235783e-02 + 3.04771749334e-02 3.46097554135e-02 4.03066875222e-02 5.38559646463e-02 7.65079897135e-02 9.74791227833e-02 + 1.07505424132e-01 1.07505424132e-01 9.74791227835e-02 7.65079897137e-02 5.38559646464e-02 4.03066875223e-02 + 3.46097554135e-02 3.04771749334e-02 2.65397992490e-02 2.40837535674e-02 2.32028796117e-02 2.30937481116e-02 + 2.31229887194e-02 2.31229887194e-02 2.30937481115e-02 2.32028796117e-02 2.40837535674e-02 2.65397992490e-02 + 3.33648425996e-02 3.86301228830e-02 4.56997211400e-02 5.89470993595e-02 7.65079897136e-02 9.00462898608e-02 + 9.46627422163e-02 9.00462898609e-02 7.65079897137e-02 5.89470993597e-02 4.56997211401e-02 3.86301228831e-02 + 3.33648425996e-02 2.78012163806e-02 2.32028796117e-02 2.03982633679e-02 1.89263306496e-02 1.81429746336e-02 + 1.78656633258e-02 1.81429746336e-02 1.89263306496e-02 2.03982633679e-02 2.32028796117e-02 2.78012163806e-02 + 3.51265406940e-02 4.02274592299e-02 4.56997211400e-02 5.38559646464e-02 6.30067497702e-02 6.87952783262e-02 + 6.87952783262e-02 6.30067497702e-02 5.38559646464e-02 4.56997211401e-02 4.02274592300e-02 3.51265406940e-02 + 2.89589200657e-02 2.30937481116e-02 1.89263306496e-02 1.64062970608e-02 1.48932680012e-02 1.40901757966e-02 + 1.40901757966e-02 1.48932680012e-02 1.64062970608e-02 1.89263306496e-02 2.30937481115e-02 2.89589200656e-02 + 3.51265406940e-02 3.86301228830e-02 4.03066875223e-02 4.15474648475e-02 4.26144257223e-02 4.30325583083e-02 + 4.26144257223e-02 4.15474648475e-02 4.03066875223e-02 3.86301228830e-02 3.51265406940e-02 2.93978765599e-02 + 2.31229887194e-02 1.81429746336e-02 1.48932680012e-02 1.28581680779e-02 1.16030717026e-02 1.11481924117e-02 + 1.16030717026e-02 1.28581680779e-02 1.48932680012e-02 1.81429746336e-02 2.31229887194e-02 2.93978765598e-02 + 3.33648425996e-02 3.46097554135e-02 3.26919751552e-02 2.92886940372e-02 2.67211542468e-02 2.67211542468e-02 + 2.92886940371e-02 3.26919751551e-02 3.46097554135e-02 3.33648425996e-02 2.89589200657e-02 2.31229887194e-02 + 1.78656633258e-02 1.40901757966e-02 1.16030717026e-02 9.96178172025e-03 9.07040710948e-03 9.07040710948e-03 + 9.96178172025e-03 1.16030717026e-02 1.40901757966e-02 1.78656633258e-02 2.31229887194e-02 2.89589200657e-02 + 3.04771749334e-02 2.97972897261e-02 2.61163074370e-02 2.17745697719e-02 1.98434626957e-02 2.17745697719e-02 + 2.61163074370e-02 2.97972897260e-02 3.04771749334e-02 2.78012163806e-02 2.30937481116e-02 1.81429746336e-02 + 1.40901757966e-02 1.11481924117e-02 9.07040710948e-03 7.73184010469e-03 7.25307658556e-03 7.73184010469e-03 + 9.07040710948e-03 1.11481924117e-02 1.40901757966e-02 1.81429746336e-02 2.30937481115e-02 2.78012163806e-02 + 2.74766558578e-02 2.52952680324e-02 2.10845164455e-02 1.76765157693e-02 1.76765157693e-02 2.10845164455e-02 + 2.52952680324e-02 2.74766558578e-02 2.65397992490e-02 2.32028796117e-02 1.89263306496e-02 1.48932680012e-02 + 1.16030717026e-02 9.07040710948e-03 7.25307658556e-03 6.26635360835e-03 6.26635360835e-03 7.25307658556e-03 + 9.07040710948e-03 1.16030717026e-02 1.48932680012e-02 1.89263306496e-02 2.32028796117e-02 2.65397992490e-02 + 2.54923302762e-02 2.20784329209e-02 1.79303755335e-02 1.60511803073e-02 1.79303755335e-02 2.20784329209e-02 + 2.54923302762e-02 2.61753235783e-02 2.40837535674e-02 2.03982633679e-02 1.64062970608e-02 1.28581680779e-02 + 9.96178172025e-03 7.73184010469e-03 6.26635360835e-03 5.74654712650e-03 6.26635360835e-03 7.73184010469e-03 + 9.96178172025e-03 1.28581680779e-02 1.64062970608e-02 2.03982633679e-02 2.40837535674e-02 2.61753235783e-02 + 2.52952680324e-02 2.10845164455e-02 1.76765157693e-02 1.76765157693e-02 2.10845164455e-02 2.52952680324e-02 + 2.74766558578e-02 2.65397992490e-02 2.32028796117e-02 1.89263306496e-02 1.48932680012e-02 1.16030717026e-02 + 9.07040710948e-03 7.25307658556e-03 6.26635360835e-03 6.26635360835e-03 7.25307658556e-03 9.07040710948e-03 + 1.16030717026e-02 1.48932680012e-02 1.89263306496e-02 2.32028796117e-02 2.65397992490e-02 2.74766558578e-02 + 2.61163074370e-02 2.17745697719e-02 1.98434626957e-02 2.17745697719e-02 2.61163074370e-02 2.97972897261e-02 + 3.04771749334e-02 2.78012163806e-02 2.30937481115e-02 1.81429746336e-02 1.40901757966e-02 1.11481924117e-02 + 9.07040710948e-03 7.73184010469e-03 7.25307658556e-03 7.73184010469e-03 9.07040710948e-03 1.11481924117e-02 + 1.40901757966e-02 1.81429746336e-02 2.30937481116e-02 2.78012163806e-02 3.04771749334e-02 2.97972897260e-02 + 2.92886940371e-02 2.67211542468e-02 2.67211542468e-02 2.92886940372e-02 3.26919751552e-02 3.46097554135e-02 + 3.33648425996e-02 2.89589200657e-02 2.31229887194e-02 1.78656633258e-02 1.40901757966e-02 1.16030717026e-02 + 9.96178172025e-03 9.07040710948e-03 9.07040710948e-03 9.96178172025e-03 1.16030717026e-02 1.40901757966e-02 + 1.78656633258e-02 2.31229887194e-02 2.89589200657e-02 3.33648425996e-02 3.46097554135e-02 3.26919751551e-02 + 4.26144257223e-02 4.30325583083e-02 4.26144257223e-02 4.15474648475e-02 4.03066875223e-02 3.86301228830e-02 + 3.51265406940e-02 2.93978765598e-02 2.31229887194e-02 1.81429746336e-02 1.48932680012e-02 1.28581680779e-02 + 1.16030717026e-02 1.11481924117e-02 1.16030717026e-02 1.28581680779e-02 1.48932680012e-02 1.81429746336e-02 + 2.31229887194e-02 2.93978765599e-02 3.51265406940e-02 3.86301228830e-02 4.03066875223e-02 4.15474648475e-02 + 6.87952783262e-02 6.87952783262e-02 6.30067497702e-02 5.38559646464e-02 4.56997211400e-02 4.02274592299e-02 + 3.51265406940e-02 2.89589200656e-02 2.30937481115e-02 1.89263306496e-02 1.64062970608e-02 1.48932680012e-02 + 1.40901757966e-02 1.40901757966e-02 1.48932680012e-02 1.64062970608e-02 1.89263306496e-02 2.30937481116e-02 + 2.89589200657e-02 3.51265406940e-02 4.02274592300e-02 4.56997211401e-02 5.38559646464e-02 6.30067497702e-02 + 9.46627422163e-02 9.00462898608e-02 7.65079897136e-02 5.89470993595e-02 4.56997211400e-02 3.86301228830e-02 + 3.33648425996e-02 2.78012163806e-02 2.32028796117e-02 2.03982633679e-02 1.89263306496e-02 1.81429746336e-02 + 1.78656633258e-02 1.81429746336e-02 1.89263306496e-02 2.03982633679e-02 2.32028796117e-02 2.78012163806e-02 + 3.33648425996e-02 3.86301228831e-02 4.56997211401e-02 5.89470993597e-02 7.65079897137e-02 9.00462898609e-02 + 1.07505424132e-01 9.74791227833e-02 7.65079897135e-02 5.38559646463e-02 4.03066875222e-02 3.46097554135e-02 + 3.04771749334e-02 2.65397992490e-02 2.40837535674e-02 2.32028796117e-02 2.30937481115e-02 2.31229887194e-02 + 2.31229887194e-02 2.30937481116e-02 2.32028796117e-02 2.40837535674e-02 2.65397992490e-02 3.04771749334e-02 + 3.46097554135e-02 4.03066875223e-02 5.38559646464e-02 7.65079897137e-02 9.74791227835e-02 1.07505424132e-01 + 1.07505424132e-01 9.00462898607e-02 6.30067497701e-02 4.15474648474e-02 3.26919751551e-02 2.97972897261e-02 + 2.74766558578e-02 2.61753235783e-02 2.65397992490e-02 2.78012163806e-02 2.89589200656e-02 2.93978765598e-02 + 2.89589200657e-02 2.78012163806e-02 2.65397992490e-02 2.61753235783e-02 2.74766558578e-02 2.97972897260e-02 + 3.26919751551e-02 4.15474648475e-02 6.30067497702e-02 9.00462898609e-02 1.07505424132e-01 1.12226003113e-01 + 6.13627808208e-02 4.23899114589e-02 3.05271423180e-02 2.61163074370e-02 2.44576079909e-02 2.45136072590e-02 + 2.69079511827e-02 3.06791433435e-02 3.45751721122e-02 3.75326780270e-02 3.86294226544e-02 3.75326780270e-02 + 3.45751721123e-02 3.06791433435e-02 2.69079511826e-02 2.45136072589e-02 2.44576079908e-02 2.61163074370e-02 + 3.05271423180e-02 4.23899114590e-02 6.13627808209e-02 7.81199920260e-02 8.44125475991e-02 7.81199920259e-02 + 4.23899114589e-02 2.95747487953e-02 2.35389487571e-02 2.10845164455e-02 2.09113970026e-02 2.36140048007e-02 + 2.81921518485e-02 3.39197128099e-02 4.00844862497e-02 4.42642004890e-02 4.42642004891e-02 4.00844862497e-02 + 3.39197128100e-02 2.81921518485e-02 2.36140048007e-02 2.09113970026e-02 2.10845164455e-02 2.35389487571e-02 + 2.95747487953e-02 4.23899114590e-02 5.87813204091e-02 6.98261770600e-02 6.98261770600e-02 5.87813204090e-02 + 3.05271423180e-02 2.35389487571e-02 1.95330017723e-02 1.79303755335e-02 1.95330017723e-02 2.35389487571e-02 + 3.05271423180e-02 4.15214590889e-02 5.24377104593e-02 5.69445178737e-02 5.24377104594e-02 4.15214590889e-02 + 3.05271423180e-02 2.35389487571e-02 1.95330017723e-02 1.79303755334e-02 1.95330017723e-02 2.35389487571e-02 + 3.05271423180e-02 4.15214590890e-02 5.24377104593e-02 5.69445178737e-02 5.24377104593e-02 4.15214590889e-02 + 2.61163074370e-02 2.10845164455e-02 1.79303755335e-02 1.76765157693e-02 1.98434626957e-02 2.67211542468e-02 + 4.26144257223e-02 6.30067497701e-02 7.65079897136e-02 7.65079897136e-02 6.30067497702e-02 4.26144257223e-02 + 2.67211542468e-02 1.98434626957e-02 1.76765157693e-02 1.79303755335e-02 2.10845164455e-02 2.61163074370e-02 + 3.26919751552e-02 4.03066875223e-02 4.56997211400e-02 4.56997211400e-02 4.03066875222e-02 3.26919751551e-02 + 2.44576079909e-02 2.09113970026e-02 1.95330017723e-02 1.98434626957e-02 2.50655517889e-02 4.31923684876e-02 + 7.14597362941e-02 9.40547983346e-02 1.01689797158e-01 9.40547983347e-02 7.14597362942e-02 4.31923684875e-02 + 2.50655517888e-02 1.98434626957e-02 1.95330017723e-02 2.09113970026e-02 2.44576079909e-02 2.92434029240e-02 + 3.44218462469e-02 3.88820048628e-02 4.06895832708e-02 3.88820048627e-02 3.44218462468e-02 2.92434029240e-02 + 2.45136072590e-02 2.36140048007e-02 2.35389487571e-02 2.67211542468e-02 4.31923684876e-02 7.44580546054e-02 + 1.02256358961e-01 1.13480556613e-01 1.13480556613e-01 1.02256358961e-01 7.44580546055e-02 4.31923684875e-02 + 2.67211542468e-02 2.35389487571e-02 2.36140048007e-02 2.45136072589e-02 2.70509290066e-02 3.05546433144e-02 + 3.39946385756e-02 3.61943858423e-02 3.61943858423e-02 3.39946385756e-02 3.05546433144e-02 2.70509290066e-02 + 2.69079511827e-02 2.81921518485e-02 3.05271423180e-02 4.26144257223e-02 7.14597362941e-02 1.02256358961e-01 + 1.14984832163e-01 1.15477102967e-01 1.14984832163e-01 1.02256358961e-01 7.14597362942e-02 4.26144257223e-02 + 3.05271423180e-02 2.81921518485e-02 2.69079511826e-02 2.62349275771e-02 2.71139620581e-02 2.89472257765e-02 + 3.06730045455e-02 3.13659743854e-02 3.06730045455e-02 2.89472257765e-02 2.71139620581e-02 2.62349275771e-02 + 3.06791433435e-02 3.39197128099e-02 4.15214590889e-02 6.30067497701e-02 9.40547983345e-02 1.13480556613e-01 + 1.15477102967e-01 1.15477102967e-01 1.13480556613e-01 9.40547983347e-02 6.30067497702e-02 4.15214590889e-02 + 3.39197128099e-02 3.06791433435e-02 2.73901656280e-02 2.51928884334e-02 2.46284218885e-02 2.49901046905e-02 + 2.54054025138e-02 2.54054025138e-02 2.49901046905e-02 2.46284218885e-02 2.51928884334e-02 2.73901656280e-02 + 3.45751721122e-02 4.00844862497e-02 5.24377104593e-02 7.65079897135e-02 1.01689797158e-01 1.13480556613e-01 + 1.14984832163e-01 1.13480556613e-01 1.01689797158e-01 7.65079897137e-02 5.24377104594e-02 4.00844862497e-02 + 3.45751721123e-02 2.95889063006e-02 2.49962561202e-02 2.20776282425e-02 2.07195371247e-02 2.02039597509e-02 + 2.00652756289e-02 2.02039597509e-02 2.07195371247e-02 2.20776282425e-02 2.49962561202e-02 2.95889063006e-02 + 3.75326780270e-02 4.42642004890e-02 5.69445178737e-02 7.65079897136e-02 9.40547983346e-02 1.02256358961e-01 + 1.02256358961e-01 9.40547983347e-02 7.65079897137e-02 5.69445178738e-02 4.42642004891e-02 3.75326780271e-02 + 3.16055353689e-02 2.55724702285e-02 2.09446150107e-02 1.81502745082e-02 1.66389072281e-02 1.59286730225e-02 + 1.59286730225e-02 1.66389072281e-02 1.81502745082e-02 2.09446150107e-02 2.55724702285e-02 3.16055353689e-02 + 3.86294226543e-02 4.42642004891e-02 5.24377104593e-02 6.30067497702e-02 7.14597362942e-02 7.44580546055e-02 + 7.14597362942e-02 6.30067497702e-02 5.24377104594e-02 4.42642004891e-02 3.86294226544e-02 3.27027225516e-02 + 2.60490352763e-02 2.03920068264e-02 1.66276495780e-02 1.43895601245e-02 1.31235074234e-02 1.26855892424e-02 + 1.31235074234e-02 1.43895601245e-02 1.66276495780e-02 2.03920068264e-02 2.60490352762e-02 3.27027225515e-02 + 3.75326780270e-02 4.00844862497e-02 4.15214590889e-02 4.26144257223e-02 4.31923684876e-02 4.31923684875e-02 + 4.26144257223e-02 4.15214590889e-02 4.00844862497e-02 3.75326780271e-02 3.27027225516e-02 2.61899431515e-02 + 2.00581600358e-02 1.56490177739e-02 1.28938940154e-02 1.12138690779e-02 1.03423481784e-02 1.03423481784e-02 + 1.12138690779e-02 1.28938940154e-02 1.56490177739e-02 2.00581600358e-02 2.61899431515e-02 3.27027225515e-02 + 3.45751721123e-02 3.39197128100e-02 3.05271423180e-02 2.67211542468e-02 2.50655517888e-02 2.67211542468e-02 + 3.05271423180e-02 3.39197128099e-02 3.45751721123e-02 3.16055353689e-02 2.60490352763e-02 2.00581600358e-02 + 1.52718979502e-02 1.20440164119e-02 9.96244247360e-03 8.68885911334e-03 8.23794392376e-03 8.68885911334e-03 + 9.96244247360e-03 1.20440164119e-02 1.52718979502e-02 2.00581600358e-02 2.60490352762e-02 3.16055353689e-02 + 3.06791433435e-02 2.81921518485e-02 2.35389487571e-02 1.98434626957e-02 1.98434626957e-02 2.35389487571e-02 + 2.81921518485e-02 3.06791433435e-02 2.95889063006e-02 2.55724702285e-02 2.03920068264e-02 1.56490177739e-02 + 1.20440164119e-02 9.50257178746e-03 7.77919418380e-03 6.85673718666e-03 6.85673718666e-03 7.77919418380e-03 + 9.50257178746e-03 1.20440164119e-02 1.56490177739e-02 2.03920068264e-02 2.55724702285e-02 2.95889063006e-02 + 2.69079511827e-02 2.36140048007e-02 1.95330017723e-02 1.76765157693e-02 1.95330017723e-02 2.36140048007e-02 + 2.69079511826e-02 2.73901656280e-02 2.49962561202e-02 2.09446150107e-02 1.66276495780e-02 1.28938940154e-02 + 9.96244247360e-03 7.77919418380e-03 6.36560105343e-03 5.86554131392e-03 6.36560105343e-03 7.77919418380e-03 + 9.96244247360e-03 1.28938940154e-02 1.66276495780e-02 2.09446150107e-02 2.49962561202e-02 2.73901656281e-02 + 2.45136072589e-02 2.09113970026e-02 1.79303755335e-02 1.79303755335e-02 2.09113970026e-02 2.45136072590e-02 + 2.62349275771e-02 2.51928884334e-02 2.20776282425e-02 1.81502745082e-02 1.43895601245e-02 1.12138690779e-02 + 8.68885911334e-03 6.85673718666e-03 5.86554131392e-03 5.86554131392e-03 6.85673718666e-03 8.68885911334e-03 + 1.12138690779e-02 1.43895601245e-02 1.81502745082e-02 2.20776282425e-02 2.51928884334e-02 2.62349275771e-02 + 2.44576079908e-02 2.10845164455e-02 1.95330017723e-02 2.10845164455e-02 2.44576079909e-02 2.70509290066e-02 + 2.71139620581e-02 2.46284218885e-02 2.07195371247e-02 1.66389072281e-02 1.31235074234e-02 1.03423481784e-02 + 8.23794392376e-03 6.85673718666e-03 6.36560105343e-03 6.85673718666e-03 8.23794392376e-03 1.03423481784e-02 + 1.31235074234e-02 1.66389072281e-02 2.07195371247e-02 2.46284218885e-02 2.71139620581e-02 2.70509290066e-02 + 2.61163074370e-02 2.35389487571e-02 2.35389487571e-02 2.61163074370e-02 2.92434029240e-02 3.05546433144e-02 + 2.89472257765e-02 2.49901046905e-02 2.02039597509e-02 1.59286730225e-02 1.26855892423e-02 1.03423481784e-02 + 8.68885911334e-03 7.77919418380e-03 7.77919418380e-03 8.68885911334e-03 1.03423481784e-02 1.26855892424e-02 + 1.59286730225e-02 2.02039597509e-02 2.49901046905e-02 2.89472257765e-02 3.05546433144e-02 2.92434029240e-02 + 3.05271423180e-02 2.95747487953e-02 3.05271423180e-02 3.26919751552e-02 3.44218462469e-02 3.39946385756e-02 + 3.06730045455e-02 2.54054025138e-02 2.00652756288e-02 1.59286730225e-02 1.31235074234e-02 1.12138690779e-02 + 9.96244247360e-03 9.50257178746e-03 9.96244247360e-03 1.12138690779e-02 1.31235074234e-02 1.59286730225e-02 + 2.00652756289e-02 2.54054025139e-02 3.06730045455e-02 3.39946385756e-02 3.44218462469e-02 3.26919751551e-02 + 4.23899114590e-02 4.23899114590e-02 4.15214590889e-02 4.03066875223e-02 3.88820048628e-02 3.61943858423e-02 + 3.13659743854e-02 2.54054025138e-02 2.02039597509e-02 1.66389072281e-02 1.43895601245e-02 1.28938940154e-02 + 1.20440164119e-02 1.20440164119e-02 1.28938940154e-02 1.43895601245e-02 1.66389072281e-02 2.02039597509e-02 + 2.54054025139e-02 3.13659743854e-02 3.61943858423e-02 3.88820048628e-02 4.03066875223e-02 4.15214590889e-02 + 6.13627808209e-02 5.87813204090e-02 5.24377104593e-02 4.56997211400e-02 4.06895832708e-02 3.61943858422e-02 + 3.06730045455e-02 2.49901046905e-02 2.07195371247e-02 1.81502745082e-02 1.66276495780e-02 1.56490177739e-02 + 1.52718979502e-02 1.56490177739e-02 1.66276495780e-02 1.81502745082e-02 2.07195371247e-02 2.49901046905e-02 + 3.06730045455e-02 3.61943858423e-02 4.06895832709e-02 4.56997211401e-02 5.24377104594e-02 5.87813204091e-02 + 7.81199920260e-02 6.98261770600e-02 5.69445178736e-02 4.56997211400e-02 3.88820048627e-02 3.39946385756e-02 + 2.89472257765e-02 2.46284218885e-02 2.20776282425e-02 2.09446150107e-02 2.03920068264e-02 2.00581600358e-02 + 2.00581600358e-02 2.03920068264e-02 2.09446150107e-02 2.20776282425e-02 2.46284218885e-02 2.89472257765e-02 + 3.39946385756e-02 3.88820048628e-02 4.56997211401e-02 5.69445178738e-02 6.98261770601e-02 7.81199920260e-02 + 8.44125475990e-02 6.98261770599e-02 5.24377104592e-02 4.03066875222e-02 3.44218462468e-02 3.05546433144e-02 + 2.71139620581e-02 2.51928884334e-02 2.49962561202e-02 2.55724702285e-02 2.60490352762e-02 2.61899431515e-02 + 2.60490352762e-02 2.55724702285e-02 2.49962561202e-02 2.51928884334e-02 2.71139620581e-02 3.05546433144e-02 + 3.44218462469e-02 4.03066875223e-02 5.24377104594e-02 6.98261770601e-02 8.44125475992e-02 8.97670392509e-02 + 7.81199920259e-02 5.87813204089e-02 4.15214590888e-02 3.26919751551e-02 2.92434029240e-02 2.70509290066e-02 + 2.62349275771e-02 2.73901656280e-02 2.95889063006e-02 3.16055353689e-02 3.27027225515e-02 3.27027225515e-02 + 3.16055353689e-02 2.95889063006e-02 2.73901656280e-02 2.62349275771e-02 2.70509290066e-02 2.92434029240e-02 + 3.26919751551e-02 4.15214590889e-02 5.87813204091e-02 7.81199920261e-02 8.97670392509e-02 8.97670392509e-02 + 4.15728345980e-02 3.31552867934e-02 2.81921518484e-02 2.52952680324e-02 2.45136072590e-02 2.65223600849e-02 + 3.03790382414e-02 3.48629669819e-02 3.89797931477e-02 4.14580711769e-02 4.14580711769e-02 3.89797931478e-02 + 3.48629669819e-02 3.03790382414e-02 2.65223600849e-02 2.45136072589e-02 2.52952680324e-02 2.81921518485e-02 + 3.31552867935e-02 4.15728345981e-02 5.18159940627e-02 5.87904239654e-02 5.87904239654e-02 5.18159940626e-02 + 3.31552867934e-02 2.73937743101e-02 2.36140048007e-02 2.20784329209e-02 2.36140048007e-02 2.73937743101e-02 + 3.31552867934e-02 4.11390439155e-02 4.86719445336e-02 5.17384028836e-02 4.86719445337e-02 4.11390439155e-02 + 3.31552867935e-02 2.73937743101e-02 2.36140048007e-02 2.20784329209e-02 2.36140048007e-02 2.73937743101e-02 + 3.31552867935e-02 4.11390439155e-02 4.86719445337e-02 5.17384028835e-02 4.86719445336e-02 4.11390439154e-02 + 2.81921518484e-02 2.36140048007e-02 2.09113970026e-02 2.10845164455e-02 2.35389487571e-02 2.95747487953e-02 + 4.23899114590e-02 5.87813204090e-02 6.98261770600e-02 6.98261770600e-02 5.87813204091e-02 4.23899114590e-02 + 2.95747487953e-02 2.35389487571e-02 2.10845164455e-02 2.09113970026e-02 2.36140048007e-02 2.81921518485e-02 + 3.39197128100e-02 4.00844862497e-02 4.42642004891e-02 4.42642004890e-02 4.00844862497e-02 3.39197128099e-02 + 2.52952680324e-02 2.20784329209e-02 2.10845164455e-02 2.17745697719e-02 2.67211542468e-02 4.30325583083e-02 + 6.87952783262e-02 9.00462898608e-02 9.74791227834e-02 9.00462898609e-02 6.87952783262e-02 4.30325583083e-02 + 2.67211542468e-02 2.17745697719e-02 2.10845164455e-02 2.20784329209e-02 2.52952680324e-02 2.97972897261e-02 + 3.46097554135e-02 3.86301228830e-02 4.02274592299e-02 3.86301228830e-02 3.46097554135e-02 2.97972897260e-02 + 2.45136072590e-02 2.36140048007e-02 2.35389487572e-02 2.67211542468e-02 4.31923684876e-02 7.44580546054e-02 + 1.02256358961e-01 1.13480556613e-01 1.13480556613e-01 1.02256358961e-01 7.44580546055e-02 4.31923684875e-02 + 2.67211542468e-02 2.35389487571e-02 2.36140048007e-02 2.45136072589e-02 2.70509290066e-02 3.05546433144e-02 + 3.39946385756e-02 3.61943858423e-02 3.61943858423e-02 3.39946385756e-02 3.05546433144e-02 2.70509290066e-02 + 2.65223600849e-02 2.73937743101e-02 2.95747487953e-02 4.30325583083e-02 7.44580546054e-02 1.05383989813e-01 + 1.14285203523e-01 1.12133532869e-01 1.14285203523e-01 1.05383989813e-01 7.44580546055e-02 4.30325583083e-02 + 2.95747487953e-02 2.73937743101e-02 2.65223600849e-02 2.61137483222e-02 2.72047829307e-02 2.92642577664e-02 + 3.12137668085e-02 3.20076435874e-02 3.12137668085e-02 2.92642577664e-02 2.72047829307e-02 2.61137483222e-02 + 3.03790382414e-02 3.31552867935e-02 4.23899114590e-02 6.87952783262e-02 1.02256358961e-01 1.14285203523e-01 + 1.05270975780e-01 1.05270975780e-01 1.14285203523e-01 1.02256358961e-01 6.87952783262e-02 4.23899114590e-02 + 3.31552867935e-02 3.03790382414e-02 2.75583653887e-02 2.55789017267e-02 2.52442663181e-02 2.59180287237e-02 + 2.65859681267e-02 2.65859681267e-02 2.59180287237e-02 2.52442663181e-02 2.55789017267e-02 2.75583653887e-02 + 3.48629669819e-02 4.11390439155e-02 5.87813204090e-02 9.00462898608e-02 1.13480556613e-01 1.12133532869e-01 + 1.05270975780e-01 1.12133532870e-01 1.13480556613e-01 9.00462898609e-02 5.87813204091e-02 4.11390439155e-02 + 3.48629669819e-02 3.04051608500e-02 2.59617964748e-02 2.30251854758e-02 2.17946809311e-02 2.15194519516e-02 + 2.15037112468e-02 2.15194519516e-02 2.17946809311e-02 2.30251854758e-02 2.59617964748e-02 3.04051608500e-02 + 3.89797931477e-02 4.86719445337e-02 6.98261770600e-02 9.74791227833e-02 1.13480556613e-01 1.14285203523e-01 + 1.14285203523e-01 1.13480556613e-01 9.74791227834e-02 6.98261770601e-02 4.86719445337e-02 3.89797931478e-02 + 3.31962688978e-02 2.73091130234e-02 2.24043544384e-02 1.93870166210e-02 1.78887124933e-02 1.72810954144e-02 + 1.72810954144e-02 1.78887124933e-02 1.93870166210e-02 2.24043544384e-02 2.73091130234e-02 3.31962688978e-02 + 4.14580711769e-02 5.17384028836e-02 6.98261770600e-02 9.00462898609e-02 1.02256358961e-01 1.05383989813e-01 + 1.02256358961e-01 9.00462898609e-02 6.98261770601e-02 5.17384028836e-02 4.14580711769e-02 3.49765243617e-02 + 2.84215335265e-02 2.23752062804e-02 1.81179478705e-02 1.56224103699e-02 1.43188156133e-02 1.38961510916e-02 + 1.43188156133e-02 1.56224103699e-02 1.81179478705e-02 2.23752062804e-02 2.84215335265e-02 3.49765243617e-02 + 4.14580711769e-02 4.86719445337e-02 5.87813204091e-02 6.87952783262e-02 7.44580546055e-02 7.44580546055e-02 + 6.87952783262e-02 5.87813204091e-02 4.86719445337e-02 4.14580711769e-02 3.55400935242e-02 2.88774182856e-02 + 2.22711776717e-02 1.72612102705e-02 1.41150555464e-02 1.22935034476e-02 1.14032634249e-02 1.14032634249e-02 + 1.22935034476e-02 1.41150555464e-02 1.72612102705e-02 2.22711776717e-02 2.88774182856e-02 3.55400935242e-02 + 3.89797931478e-02 4.11390439155e-02 4.23899114590e-02 4.30325583083e-02 4.31923684876e-02 4.30325583083e-02 + 4.23899114590e-02 4.11390439155e-02 3.89797931478e-02 3.49765243617e-02 2.88774182856e-02 2.21707375456e-02 + 1.66949823920e-02 1.30619533075e-02 1.08501521193e-02 9.57630048387e-03 9.13735254995e-03 9.57630048387e-03 + 1.08501521193e-02 1.30619533075e-02 1.66949823920e-02 2.21707375456e-02 2.88774182856e-02 3.49765243617e-02 + 3.48629669819e-02 3.31552867935e-02 2.95747487953e-02 2.67211542468e-02 2.67211542468e-02 2.95747487953e-02 + 3.31552867935e-02 3.48629669819e-02 3.31962688978e-02 2.84215335265e-02 2.22711776717e-02 1.66949823920e-02 + 1.26512009150e-02 1.00211358334e-02 8.36367529240e-03 7.50459533106e-03 7.50459533106e-03 8.36367529240e-03 + 1.00211358334e-02 1.26512009150e-02 1.66949823920e-02 2.22711776717e-02 2.84215335265e-02 3.31962688978e-02 + 3.03790382414e-02 2.73937743101e-02 2.35389487571e-02 2.17745697719e-02 2.35389487571e-02 2.73937743101e-02 + 3.03790382414e-02 3.04051608500e-02 2.73091130234e-02 2.23752062804e-02 1.72612102705e-02 1.30619533075e-02 + 1.00211358334e-02 7.92503968743e-03 6.62092791971e-03 6.16390944840e-03 6.62092791970e-03 7.92503968743e-03 + 1.00211358334e-02 1.30619533075e-02 1.72612102704e-02 2.23752062804e-02 2.73091130234e-02 3.04051608500e-02 + 2.65223600849e-02 2.36140048007e-02 2.10845164455e-02 2.10845164455e-02 2.36140048007e-02 2.65223600849e-02 + 2.75583653887e-02 2.59617964748e-02 2.24043544384e-02 1.81179478705e-02 1.41150555464e-02 1.08501521193e-02 + 8.36367529240e-03 6.62092791970e-03 5.69332826895e-03 5.69332826895e-03 6.62092791970e-03 8.36367529240e-03 + 1.08501521193e-02 1.41150555464e-02 1.81179478705e-02 2.24043544384e-02 2.59617964748e-02 2.75583653887e-02 + 2.45136072589e-02 2.20784329209e-02 2.09113970026e-02 2.20784329209e-02 2.45136072590e-02 2.61137483222e-02 + 2.55789017267e-02 2.30251854758e-02 1.93870166210e-02 1.56224103699e-02 1.22935034476e-02 9.57630048387e-03 + 7.50459533106e-03 6.16390944840e-03 5.69332826895e-03 6.16390944839e-03 7.50459533106e-03 9.57630048387e-03 + 1.22935034476e-02 1.56224103699e-02 1.93870166210e-02 2.30251854758e-02 2.55789017267e-02 2.61137483222e-02 + 2.52952680324e-02 2.36140048007e-02 2.36140048007e-02 2.52952680324e-02 2.70509290066e-02 2.72047829307e-02 + 2.52442663181e-02 2.17946809311e-02 1.78887124933e-02 1.43188156133e-02 1.14032634249e-02 9.13735254995e-03 + 7.50459533106e-03 6.62092791970e-03 6.62092791970e-03 7.50459533106e-03 9.13735254995e-03 1.14032634249e-02 + 1.43188156133e-02 1.78887124933e-02 2.17946809311e-02 2.52442663181e-02 2.72047829307e-02 2.70509290066e-02 + 2.81921518485e-02 2.73937743101e-02 2.81921518485e-02 2.97972897261e-02 3.05546433144e-02 2.92642577664e-02 + 2.59180287237e-02 2.15194519516e-02 1.72810954144e-02 1.38961510916e-02 1.14032634249e-02 9.57630048387e-03 + 8.36367529240e-03 7.92503968743e-03 8.36367529240e-03 9.57630048387e-03 1.14032634249e-02 1.38961510916e-02 + 1.72810954144e-02 2.15194519516e-02 2.59180287237e-02 2.92642577664e-02 3.05546433144e-02 2.97972897260e-02 + 3.31552867935e-02 3.31552867935e-02 3.39197128100e-02 3.46097554135e-02 3.39946385756e-02 3.12137668085e-02 + 2.65859681267e-02 2.15037112467e-02 1.72810954144e-02 1.43188156133e-02 1.22935034476e-02 1.08501521193e-02 + 1.00211358334e-02 1.00211358334e-02 1.08501521193e-02 1.22935034476e-02 1.43188156133e-02 1.72810954144e-02 + 2.15037112468e-02 2.65859681267e-02 3.12137668086e-02 3.39946385756e-02 3.46097554135e-02 3.39197128099e-02 + 4.15728345981e-02 4.11390439155e-02 4.00844862497e-02 3.86301228830e-02 3.61943858423e-02 3.20076435873e-02 + 2.65859681267e-02 2.15194519516e-02 1.78887124933e-02 1.56224103699e-02 1.41150555464e-02 1.30619533075e-02 + 1.26512009150e-02 1.30619533075e-02 1.41150555464e-02 1.56224103699e-02 1.78887124933e-02 2.15194519516e-02 + 2.65859681267e-02 3.20076435874e-02 3.61943858423e-02 3.86301228830e-02 4.00844862497e-02 4.11390439155e-02 + 5.18159940627e-02 4.86719445337e-02 4.42642004890e-02 4.02274592299e-02 3.61943858422e-02 3.12137668085e-02 + 2.59180287237e-02 2.17946809311e-02 1.93870166210e-02 1.81179478705e-02 1.72612102705e-02 1.66949823920e-02 + 1.66949823920e-02 1.72612102704e-02 1.81179478705e-02 1.93870166210e-02 2.17946809311e-02 2.59180287237e-02 + 3.12137668086e-02 3.61943858423e-02 4.02274592300e-02 4.42642004891e-02 4.86719445337e-02 5.18159940627e-02 + 5.87904239654e-02 5.17384028835e-02 4.42642004890e-02 3.86301228830e-02 3.39946385756e-02 2.92642577664e-02 + 2.52442663181e-02 2.30251854758e-02 2.24043544384e-02 2.23752062804e-02 2.22711776717e-02 2.21707375456e-02 + 2.22711776717e-02 2.23752062804e-02 2.24043544384e-02 2.30251854758e-02 2.52442663181e-02 2.92642577664e-02 + 3.39946385756e-02 3.86301228831e-02 4.42642004891e-02 5.17384028836e-02 5.87904239655e-02 6.17010347926e-02 + 5.87904239654e-02 4.86719445336e-02 4.00844862496e-02 3.46097554135e-02 3.05546433144e-02 2.72047829307e-02 + 2.55789017267e-02 2.59617964748e-02 2.73091130234e-02 2.84215335265e-02 2.88774182856e-02 2.88774182856e-02 + 2.84215335265e-02 2.73091130234e-02 2.59617964748e-02 2.55789017267e-02 2.72047829307e-02 3.05546433144e-02 + 3.46097554135e-02 4.00844862497e-02 4.86719445337e-02 5.87904239655e-02 6.56429700805e-02 6.56429700804e-02 + 5.18159940626e-02 4.11390439154e-02 3.39197128099e-02 2.97972897260e-02 2.70509290066e-02 2.61137483222e-02 + 2.75583653887e-02 3.04051608500e-02 3.31962688978e-02 3.49765243617e-02 3.55400935242e-02 3.49765243617e-02 + 3.31962688978e-02 3.04051608500e-02 2.75583653887e-02 2.61137483222e-02 2.70509290066e-02 2.97972897260e-02 + 3.39197128099e-02 4.11390439155e-02 5.18159940627e-02 6.17010347927e-02 6.56429700804e-02 6.17010347926e-02 + 3.48685420831e-02 3.03790382413e-02 2.69079511826e-02 2.54923302762e-02 2.69079511827e-02 3.03790382414e-02 + 3.48685420831e-02 3.97328731661e-02 4.36602554331e-02 4.51573443066e-02 4.36602554332e-02 3.97328731662e-02 + 3.48685420831e-02 3.03790382414e-02 2.69079511826e-02 2.54923302762e-02 2.69079511826e-02 3.03790382414e-02 + 3.48685420831e-02 3.97328731662e-02 4.36602554332e-02 4.51573443066e-02 4.36602554331e-02 3.97328731661e-02 + 3.03790382413e-02 2.65223600849e-02 2.45136072590e-02 2.52952680324e-02 2.81921518485e-02 3.31552867935e-02 + 4.15728345981e-02 5.18159940627e-02 5.87904239654e-02 5.87904239654e-02 5.18159940627e-02 4.15728345981e-02 + 3.31552867935e-02 2.81921518485e-02 2.52952680324e-02 2.45136072589e-02 2.65223600849e-02 3.03790382414e-02 + 3.48629669819e-02 3.89797931478e-02 4.14580711769e-02 4.14580711769e-02 3.89797931477e-02 3.48629669819e-02 + 2.69079511826e-02 2.45136072590e-02 2.44576079909e-02 2.61163074370e-02 3.05271423180e-02 4.23899114590e-02 + 6.13627808208e-02 7.81199920260e-02 8.44125475991e-02 7.81199920260e-02 6.13627808209e-02 4.23899114590e-02 + 3.05271423180e-02 2.61163074370e-02 2.44576079909e-02 2.45136072589e-02 2.69079511826e-02 3.06791433435e-02 + 3.45751721123e-02 3.75326780270e-02 3.86294226543e-02 3.75326780270e-02 3.45751721122e-02 3.06791433434e-02 + 2.54923302762e-02 2.52952680324e-02 2.61163074370e-02 2.92886940372e-02 4.26144257223e-02 6.87952783261e-02 + 9.46627422161e-02 1.07505424132e-01 1.07505424132e-01 9.46627422163e-02 6.87952783262e-02 4.26144257223e-02 + 2.92886940372e-02 2.61163074370e-02 2.52952680324e-02 2.54923302762e-02 2.74766558578e-02 3.04771749334e-02 + 3.33648425996e-02 3.51265406940e-02 3.51265406940e-02 3.33648425996e-02 3.04771749334e-02 2.74766558578e-02 + 2.69079511827e-02 2.81921518485e-02 3.05271423180e-02 4.26144257223e-02 7.14597362941e-02 1.02256358961e-01 + 1.14984832163e-01 1.15477102967e-01 1.14984832163e-01 1.02256358961e-01 7.14597362941e-02 4.26144257223e-02 + 3.05271423180e-02 2.81921518485e-02 2.69079511827e-02 2.62349275771e-02 2.71139620581e-02 2.89472257765e-02 + 3.06730045455e-02 3.13659743854e-02 3.06730045455e-02 2.89472257765e-02 2.71139620581e-02 2.62349275771e-02 + 3.03790382414e-02 3.31552867935e-02 4.23899114590e-02 6.87952783262e-02 1.02256358961e-01 1.14285203523e-01 + 1.05270975780e-01 1.05270975780e-01 1.14285203523e-01 1.02256358961e-01 6.87952783262e-02 4.23899114590e-02 + 3.31552867935e-02 3.03790382414e-02 2.75583653887e-02 2.55789017267e-02 2.52442663181e-02 2.59180287237e-02 + 2.65859681267e-02 2.65859681267e-02 2.59180287237e-02 2.52442663181e-02 2.55789017267e-02 2.75583653887e-02 + 3.48685420831e-02 4.15728345981e-02 6.13627808209e-02 9.46627422162e-02 1.14984832163e-01 1.05270975780e-01 + 9.35837065899e-02 1.05270975780e-01 1.14984832163e-01 9.46627422163e-02 6.13627808209e-02 4.15728345981e-02 + 3.48685420831e-02 3.06060385056e-02 2.62414574798e-02 2.33116271285e-02 2.21383854933e-02 2.19609803409e-02 + 2.19946944680e-02 2.19609803409e-02 2.21383854932e-02 2.33116271285e-02 2.62414574798e-02 3.06060385055e-02 + 3.97328731661e-02 5.18159940627e-02 7.81199920260e-02 1.07505424132e-01 1.15477102967e-01 1.05270975780e-01 + 1.05270975780e-01 1.15477102967e-01 1.07505424132e-01 7.81199920261e-02 5.18159940627e-02 3.97328731662e-02 + 3.38926655663e-02 2.81654297894e-02 2.31387782150e-02 1.99974571616e-02 1.85131064743e-02 1.79740052840e-02 + 1.79740052840e-02 1.85131064743e-02 1.99974571616e-02 2.31387782150e-02 2.81654297894e-02 3.38926655663e-02 + 4.36602554332e-02 5.87904239654e-02 8.44125475991e-02 1.07505424132e-01 1.14984832163e-01 1.14285203523e-01 + 1.14984832163e-01 1.07505424132e-01 8.44125475992e-02 5.87904239655e-02 4.36602554332e-02 3.62670291685e-02 + 2.99300705585e-02 2.37302554132e-02 1.91106183945e-02 1.63905844408e-02 1.50423980149e-02 1.46288065643e-02 + 1.50423980149e-02 1.63905844408e-02 1.91106183945e-02 2.37302554132e-02 2.99300705585e-02 3.62670291685e-02 + 4.51573443066e-02 5.87904239655e-02 7.81199920260e-02 9.46627422163e-02 1.02256358961e-01 1.02256358961e-01 + 9.46627422162e-02 7.81199920260e-02 5.87904239655e-02 4.51573443066e-02 3.73533616181e-02 3.07818020073e-02 + 2.40966608398e-02 1.86355831047e-02 1.50739753351e-02 1.30486211596e-02 1.21045641480e-02 1.21045641480e-02 + 1.30486211596e-02 1.50739753351e-02 1.86355831047e-02 2.40966608398e-02 3.07818020073e-02 3.73533616181e-02 + 4.36602554332e-02 5.18159940627e-02 6.13627808209e-02 6.87952783262e-02 7.14597362941e-02 6.87952783262e-02 + 6.13627808209e-02 5.18159940627e-02 4.36602554332e-02 3.73533616181e-02 3.09603631544e-02 2.40902041017e-02 + 1.81617249718e-02 1.40807237278e-02 1.16096263393e-02 1.02402112845e-02 9.78230850882e-03 1.02402112845e-02 + 1.16096263393e-02 1.40807237278e-02 1.81617249718e-02 2.40902041017e-02 3.09603631544e-02 3.73533616181e-02 + 3.97328731662e-02 4.15728345981e-02 4.23899114590e-02 4.26144257223e-02 4.26144257223e-02 4.23899114590e-02 + 4.15728345981e-02 3.97328731661e-02 3.62670291685e-02 3.07818020073e-02 2.40902041017e-02 1.79266369346e-02 + 1.34421984781e-02 1.06052341643e-02 8.89928200875e-03 8.04465702446e-03 8.04465702445e-03 8.89928200875e-03 + 1.06052341643e-02 1.34421984781e-02 1.79266369346e-02 2.40902041017e-02 3.07818020073e-02 3.62670291685e-02 + 3.48685420831e-02 3.31552867935e-02 3.05271423180e-02 2.92886940372e-02 3.05271423180e-02 3.31552867935e-02 + 3.48685420831e-02 3.38926655663e-02 2.99300705585e-02 2.40966608398e-02 1.81617249718e-02 1.34421984781e-02 + 1.02249435892e-02 8.15763089847e-03 6.92808975974e-03 6.50344737139e-03 6.92808975973e-03 8.15763089846e-03 + 1.02249435892e-02 1.34421984781e-02 1.81617249718e-02 2.40966608398e-02 2.99300705585e-02 3.38926655663e-02 + 3.03790382414e-02 2.81921518485e-02 2.61163074370e-02 2.61163074370e-02 2.81921518485e-02 3.03790382414e-02 + 3.06060385056e-02 2.81654297894e-02 2.37302554132e-02 1.86355831047e-02 1.40807237278e-02 1.06052341643e-02 + 8.15763089847e-03 6.53733030468e-03 5.69415565947e-03 5.69415565947e-03 6.53733030468e-03 8.15763089846e-03 + 1.06052341643e-02 1.40807237278e-02 1.86355831047e-02 2.37302554132e-02 2.81654297894e-02 3.06060385055e-02 + 2.69079511827e-02 2.52952680324e-02 2.44576079909e-02 2.52952680324e-02 2.69079511826e-02 2.75583653887e-02 + 2.62414574798e-02 2.31387782150e-02 1.91106183945e-02 1.50739753351e-02 1.16096263393e-02 8.89928200875e-03 + 6.92808975974e-03 5.69415565947e-03 5.26655497605e-03 5.69415565947e-03 6.92808975973e-03 8.89928200875e-03 + 1.16096263393e-02 1.50739753351e-02 1.91106183945e-02 2.31387782150e-02 2.62414574798e-02 2.75583653887e-02 + 2.54923302762e-02 2.45136072590e-02 2.45136072589e-02 2.54923302762e-02 2.62349275771e-02 2.55789017267e-02 + 2.33116271285e-02 1.99974571616e-02 1.63905844408e-02 1.30486211596e-02 1.02402112845e-02 8.04465702445e-03 + 6.50344737139e-03 5.69415565947e-03 5.69415565947e-03 6.50344737139e-03 8.04465702446e-03 1.02402112845e-02 + 1.30486211596e-02 1.63905844408e-02 1.99974571615e-02 2.33116271285e-02 2.55789017267e-02 2.62349275771e-02 + 2.69079511826e-02 2.65223600849e-02 2.69079511827e-02 2.74766558578e-02 2.71139620581e-02 2.52442663181e-02 + 2.21383854932e-02 1.85131064743e-02 1.50423980149e-02 1.21045641480e-02 9.78230850882e-03 8.04465702445e-03 + 6.92808975973e-03 6.53733030468e-03 6.92808975974e-03 8.04465702446e-03 9.78230850882e-03 1.21045641480e-02 + 1.50423980149e-02 1.85131064743e-02 2.21383854932e-02 2.52442663181e-02 2.71139620581e-02 2.74766558578e-02 + 3.03790382414e-02 3.03790382414e-02 3.06791433435e-02 3.04771749334e-02 2.89472257765e-02 2.59180287237e-02 + 2.19609803409e-02 1.79740052840e-02 1.46288065643e-02 1.21045641480e-02 1.02402112845e-02 8.89928200875e-03 + 8.15763089847e-03 8.15763089847e-03 8.89928200875e-03 1.02402112845e-02 1.21045641480e-02 1.46288065643e-02 + 1.79740052840e-02 2.19609803409e-02 2.59180287237e-02 2.89472257765e-02 3.04771749334e-02 3.06791433435e-02 + 3.48685420831e-02 3.48629669819e-02 3.45751721123e-02 3.33648425996e-02 3.06730045455e-02 2.65859681267e-02 + 2.19946944680e-02 1.79740052840e-02 1.50423980149e-02 1.30486211596e-02 1.16096263393e-02 1.06052341643e-02 + 1.02249435892e-02 1.06052341643e-02 1.16096263393e-02 1.30486211596e-02 1.50423980149e-02 1.79740052840e-02 + 2.19946944681e-02 2.65859681267e-02 3.06730045455e-02 3.33648425996e-02 3.45751721123e-02 3.48629669819e-02 + 3.97328731662e-02 3.89797931478e-02 3.75326780270e-02 3.51265406940e-02 3.13659743854e-02 2.65859681267e-02 + 2.19609803409e-02 1.85131064743e-02 1.63905844408e-02 1.50739753351e-02 1.40807237278e-02 1.34421984781e-02 + 1.34421984781e-02 1.40807237278e-02 1.50739753351e-02 1.63905844408e-02 1.85131064743e-02 2.19609803409e-02 + 2.65859681267e-02 3.13659743854e-02 3.51265406940e-02 3.75326780270e-02 3.89797931478e-02 3.97328731661e-02 + 4.36602554331e-02 4.14580711769e-02 3.86294226543e-02 3.51265406939e-02 3.06730045455e-02 2.59180287237e-02 + 2.21383854932e-02 1.99974571616e-02 1.91106183945e-02 1.86355831047e-02 1.81617249718e-02 1.79266369346e-02 + 1.81617249718e-02 1.86355831047e-02 1.91106183945e-02 1.99974571615e-02 2.21383854932e-02 2.59180287237e-02 + 3.06730045456e-02 3.51265406940e-02 3.86294226544e-02 4.14580711769e-02 4.36602554332e-02 4.45488243286e-02 + 4.51573443066e-02 4.14580711769e-02 3.75326780270e-02 3.33648425996e-02 2.89472257765e-02 2.52442663181e-02 + 2.33116271285e-02 2.31387782150e-02 2.37302554132e-02 2.40966608398e-02 2.40902041017e-02 2.40902041017e-02 + 2.40966608398e-02 2.37302554132e-02 2.31387782150e-02 2.33116271285e-02 2.52442663181e-02 2.89472257765e-02 + 3.33648425996e-02 3.75326780271e-02 4.14580711769e-02 4.51573443066e-02 4.76129372810e-02 4.76129372810e-02 + 4.36602554331e-02 3.89797931477e-02 3.45751721122e-02 3.04771749334e-02 2.71139620581e-02 2.55789017267e-02 + 2.62414574798e-02 2.81654297894e-02 2.99300705585e-02 3.07818020073e-02 3.09603631544e-02 3.07818020073e-02 + 2.99300705585e-02 2.81654297894e-02 2.62414574798e-02 2.55789017267e-02 2.71139620581e-02 3.04771749334e-02 + 3.45751721123e-02 3.89797931478e-02 4.36602554332e-02 4.76129372810e-02 4.91898793330e-02 4.76129372810e-02 + 3.97328731661e-02 3.48629669819e-02 3.06791433434e-02 2.74766558578e-02 2.62349275771e-02 2.75583653887e-02 + 3.06060385055e-02 3.38926655663e-02 3.62670291685e-02 3.73533616181e-02 3.73533616181e-02 3.62670291685e-02 + 3.38926655663e-02 3.06060385056e-02 2.75583653887e-02 2.62349275771e-02 2.74766558578e-02 3.06791433435e-02 + 3.48629669819e-02 3.97328731662e-02 4.45488243286e-02 4.76129372810e-02 4.76129372810e-02 4.45488243286e-02 + 3.06060385055e-02 2.75583653887e-02 2.62349275771e-02 2.74766558578e-02 3.06791433435e-02 3.48629669819e-02 + 3.97328731661e-02 4.45488243286e-02 4.76129372810e-02 4.76129372810e-02 4.45488243286e-02 3.97328731661e-02 + 3.48629669819e-02 3.06791433435e-02 2.74766558578e-02 2.62349275771e-02 2.75583653887e-02 3.06060385056e-02 + 3.38926655663e-02 3.62670291685e-02 3.73533616181e-02 3.73533616181e-02 3.62670291685e-02 3.38926655662e-02 + 2.75583653887e-02 2.61137483222e-02 2.70509290066e-02 2.97972897261e-02 3.39197128099e-02 4.11390439155e-02 + 5.18159940626e-02 6.17010347926e-02 6.56429700805e-02 6.17010347927e-02 5.18159940627e-02 4.11390439155e-02 + 3.39197128100e-02 2.97972897261e-02 2.70509290066e-02 2.61137483222e-02 2.75583653887e-02 3.04051608500e-02 + 3.31962688978e-02 3.49765243617e-02 3.55400935242e-02 3.49765243617e-02 3.31962688978e-02 3.04051608500e-02 + 2.62349275771e-02 2.70509290066e-02 2.92434029241e-02 3.26919751552e-02 4.15214590889e-02 5.87813204090e-02 + 7.81199920259e-02 8.97670392509e-02 8.97670392509e-02 7.81199920260e-02 5.87813204091e-02 4.15214590889e-02 + 3.26919751552e-02 2.92434029240e-02 2.70509290066e-02 2.62349275771e-02 2.73901656281e-02 2.95889063006e-02 + 3.16055353689e-02 3.27027225515e-02 3.27027225515e-02 3.16055353689e-02 2.95889063006e-02 2.73901656280e-02 + 2.74766558578e-02 2.97972897261e-02 3.26919751552e-02 4.15474648474e-02 6.30067497700e-02 9.00462898607e-02 + 1.07505424132e-01 1.12226003113e-01 1.07505424132e-01 9.00462898609e-02 6.30067497702e-02 4.15474648475e-02 + 3.26919751552e-02 2.97972897261e-02 2.74766558578e-02 2.61753235783e-02 2.65397992490e-02 2.78012163806e-02 + 2.89589200657e-02 2.93978765598e-02 2.89589200656e-02 2.78012163806e-02 2.65397992490e-02 2.61753235783e-02 + 3.06791433435e-02 3.39197128099e-02 4.15214590889e-02 6.30067497701e-02 9.40547983345e-02 1.13480556613e-01 + 1.15477102966e-01 1.15477102967e-01 1.13480556613e-01 9.40547983346e-02 6.30067497702e-02 4.15214590889e-02 + 3.39197128100e-02 3.06791433435e-02 2.73901656281e-02 2.51928884334e-02 2.46284218885e-02 2.49901046905e-02 + 2.54054025138e-02 2.54054025138e-02 2.49901046905e-02 2.46284218885e-02 2.51928884334e-02 2.73901656280e-02 + 3.48629669819e-02 4.11390439155e-02 5.87813204090e-02 9.00462898607e-02 1.13480556613e-01 1.12133532869e-01 + 1.05270975780e-01 1.12133532869e-01 1.13480556613e-01 9.00462898609e-02 5.87813204091e-02 4.11390439155e-02 + 3.48629669819e-02 3.04051608500e-02 2.59617964748e-02 2.30251854758e-02 2.17946809311e-02 2.15194519516e-02 + 2.15037112468e-02 2.15194519516e-02 2.17946809311e-02 2.30251854758e-02 2.59617964748e-02 3.04051608500e-02 + 3.97328731661e-02 5.18159940627e-02 7.81199920259e-02 1.07505424132e-01 1.15477102966e-01 1.05270975780e-01 + 1.05270975780e-01 1.15477102967e-01 1.07505424132e-01 7.81199920260e-02 5.18159940627e-02 3.97328731662e-02 + 3.38926655663e-02 2.81654297894e-02 2.31387782150e-02 1.99974571616e-02 1.85131064743e-02 1.79740052840e-02 + 1.79740052840e-02 1.85131064743e-02 1.99974571616e-02 2.31387782150e-02 2.81654297894e-02 3.38926655663e-02 + 4.45488243286e-02 6.17010347926e-02 8.97670392509e-02 1.12226003113e-01 1.15477102967e-01 1.12133532869e-01 + 1.15477102967e-01 1.12226003113e-01 8.97670392509e-02 6.17010347927e-02 4.45488243287e-02 3.66852084288e-02 + 3.04405035973e-02 2.42076546579e-02 1.94536912926e-02 1.66438543464e-02 1.52761582425e-02 1.48655958108e-02 + 1.52761582425e-02 1.66438543464e-02 1.94536912926e-02 2.42076546579e-02 3.04405035973e-02 3.66852084288e-02 + 4.76129372810e-02 6.56429700805e-02 8.97670392509e-02 1.07505424132e-01 1.13480556613e-01 1.13480556613e-01 + 1.07505424132e-01 8.97670392509e-02 6.56429700805e-02 4.76129372810e-02 3.82416345368e-02 3.17358439118e-02 + 2.51250422144e-02 1.94291633463e-02 1.55937775675e-02 1.34169422858e-02 1.24266085814e-02 1.24266085814e-02 + 1.34169422858e-02 1.55937775675e-02 1.94291633463e-02 2.51250422144e-02 3.17358439118e-02 3.82416345368e-02 + 4.76129372810e-02 6.17010347927e-02 7.81199920260e-02 9.00462898609e-02 9.40547983346e-02 9.00462898608e-02 + 7.81199920260e-02 6.17010347927e-02 4.76129372810e-02 3.86928693264e-02 3.20842088246e-02 2.53972544857e-02 + 1.92841419762e-02 1.48493143122e-02 1.21091269495e-02 1.06110221336e-02 1.01199573737e-02 1.06110221336e-02 + 1.21091269494e-02 1.48493143122e-02 1.92841419762e-02 2.53972544857e-02 3.20842088246e-02 3.86928693264e-02 + 4.45488243286e-02 5.18159940627e-02 5.87813204091e-02 6.30067497702e-02 6.30067497701e-02 5.87813204090e-02 + 5.18159940627e-02 4.45488243286e-02 3.82416345368e-02 3.20842088246e-02 2.53795329069e-02 1.90346417875e-02 + 1.42309947893e-02 1.11332793734e-02 9.28365371522e-03 8.37411495641e-03 8.37411495641e-03 9.28365371521e-03 + 1.11332793734e-02 1.42309947893e-02 1.90346417875e-02 2.53795329069e-02 3.20842088246e-02 3.82416345368e-02 + 3.97328731661e-02 4.11390439155e-02 4.15214590889e-02 4.15474648475e-02 4.15214590889e-02 4.11390439155e-02 + 3.97328731661e-02 3.66852084288e-02 3.17358439118e-02 2.53972544857e-02 1.90346417875e-02 1.39668146374e-02 + 1.05558438840e-02 8.42000879273e-03 7.17868687881e-03 6.75457206647e-03 7.17868687881e-03 8.42000879273e-03 + 1.05558438840e-02 1.39668146374e-02 1.90346417875e-02 2.53972544857e-02 3.17358439118e-02 3.66852084288e-02 + 3.48629669819e-02 3.39197128100e-02 3.26919751552e-02 3.26919751552e-02 3.39197128099e-02 3.48629669819e-02 + 3.38926655663e-02 3.04405035973e-02 2.51250422144e-02 1.92841419762e-02 1.42309947893e-02 1.05558438840e-02 + 8.11231618988e-03 6.56139720408e-03 5.76671059723e-03 5.76671059723e-03 6.56139720408e-03 8.11231618987e-03 + 1.05558438840e-02 1.42309947893e-02 1.92841419762e-02 2.51250422144e-02 3.04405035973e-02 3.38926655663e-02 + 3.06791433435e-02 2.97972897261e-02 2.92434029240e-02 2.97972897261e-02 3.06791433435e-02 3.04051608500e-02 + 2.81654297894e-02 2.42076546579e-02 1.94291633463e-02 1.48493143123e-02 1.11332793734e-02 8.42000879273e-03 + 6.56139720408e-03 5.43241849957e-03 5.04346463958e-03 5.43241849957e-03 6.56139720408e-03 8.42000879273e-03 + 1.11332793734e-02 1.48493143122e-02 1.94291633463e-02 2.42076546578e-02 2.81654297894e-02 3.04051608500e-02 + 2.74766558578e-02 2.70509290066e-02 2.70509290066e-02 2.74766558578e-02 2.73901656280e-02 2.59617964748e-02 + 2.31387782150e-02 1.94536912926e-02 1.55937775675e-02 1.21091269495e-02 9.28365371522e-03 7.17868687881e-03 + 5.76671059723e-03 5.04346463958e-03 5.04346463958e-03 5.76671059723e-03 7.17868687881e-03 9.28365371521e-03 + 1.21091269494e-02 1.55937775675e-02 1.94536912925e-02 2.31387782150e-02 2.59617964748e-02 2.73901656280e-02 + 2.62349275771e-02 2.61137483222e-02 2.62349275771e-02 2.61753235783e-02 2.51928884334e-02 2.30251854758e-02 + 1.99974571616e-02 1.66438543464e-02 1.34169422858e-02 1.06110221336e-02 8.37411495641e-03 6.75457206647e-03 + 5.76671059723e-03 5.43241849957e-03 5.76671059723e-03 6.75457206648e-03 8.37411495641e-03 1.06110221336e-02 + 1.34169422858e-02 1.66438543464e-02 1.99974571616e-02 2.30251854758e-02 2.51928884334e-02 2.61753235783e-02 + 2.75583653887e-02 2.75583653887e-02 2.73901656280e-02 2.65397992490e-02 2.46284218885e-02 2.17946809311e-02 + 1.85131064743e-02 1.52761582425e-02 1.24266085814e-02 1.01199573737e-02 8.37411495641e-03 7.17868687881e-03 + 6.56139720408e-03 6.56139720408e-03 7.17868687881e-03 8.37411495641e-03 1.01199573737e-02 1.24266085814e-02 + 1.52761582425e-02 1.85131064743e-02 2.17946809311e-02 2.46284218885e-02 2.65397992490e-02 2.73901656280e-02 + 3.06060385056e-02 3.04051608500e-02 2.95889063006e-02 2.78012163806e-02 2.49901046905e-02 2.15194519516e-02 + 1.79740052840e-02 1.48655958108e-02 1.24266085814e-02 1.06110221336e-02 9.28365371522e-03 8.42000879273e-03 + 8.11231618987e-03 8.42000879273e-03 9.28365371522e-03 1.06110221336e-02 1.24266085814e-02 1.48655958108e-02 + 1.79740052840e-02 2.15194519516e-02 2.49901046905e-02 2.78012163806e-02 2.95889063006e-02 3.04051608500e-02 + 3.38926655663e-02 3.31962688978e-02 3.16055353689e-02 2.89589200656e-02 2.54054025138e-02 2.15037112467e-02 + 1.79740052840e-02 1.52761582425e-02 1.34169422858e-02 1.21091269494e-02 1.11332793734e-02 1.05558438840e-02 + 1.05558438840e-02 1.11332793734e-02 1.21091269494e-02 1.34169422858e-02 1.52761582425e-02 1.79740052840e-02 + 2.15037112468e-02 2.54054025138e-02 2.89589200657e-02 3.16055353689e-02 3.31962688978e-02 3.38926655663e-02 + 3.62670291685e-02 3.49765243617e-02 3.27027225515e-02 2.93978765598e-02 2.54054025138e-02 2.15194519516e-02 + 1.85131064743e-02 1.66438543464e-02 1.55937775675e-02 1.48493143122e-02 1.42309947893e-02 1.39668146374e-02 + 1.42309947893e-02 1.48493143122e-02 1.55937775675e-02 1.66438543464e-02 1.85131064743e-02 2.15194519516e-02 + 2.54054025139e-02 2.93978765599e-02 3.27027225515e-02 3.49765243617e-02 3.62670291685e-02 3.66852084288e-02 + 3.73533616181e-02 3.55400935242e-02 3.27027225515e-02 2.89589200656e-02 2.49901046905e-02 2.17946809311e-02 + 1.99974571616e-02 1.94536912926e-02 1.94291633463e-02 1.92841419762e-02 1.90346417875e-02 1.90346417874e-02 + 1.92841419762e-02 1.94291633463e-02 1.94536912925e-02 1.99974571615e-02 2.17946809311e-02 2.49901046905e-02 + 2.89589200657e-02 3.27027225515e-02 3.55400935242e-02 3.73533616181e-02 3.82416345368e-02 3.82416345368e-02 + 3.73533616181e-02 3.49765243617e-02 3.16055353689e-02 2.78012163806e-02 2.46284218885e-02 2.30251854758e-02 + 2.31387782150e-02 2.42076546579e-02 2.51250422144e-02 2.53972544857e-02 2.53795329069e-02 2.53972544857e-02 + 2.51250422144e-02 2.42076546578e-02 2.31387782150e-02 2.30251854758e-02 2.46284218885e-02 2.78012163806e-02 + 3.16055353689e-02 3.49765243617e-02 3.73533616181e-02 3.86928693264e-02 3.91195978595e-02 3.86928693264e-02 + 3.62670291685e-02 3.31962688978e-02 2.95889063006e-02 2.65397992490e-02 2.51928884334e-02 2.59617964748e-02 + 2.81654297894e-02 3.04405035973e-02 3.17358439118e-02 3.20842088246e-02 3.20842088246e-02 3.17358439118e-02 + 3.04405035973e-02 2.81654297894e-02 2.59617964748e-02 2.51928884334e-02 2.65397992490e-02 2.95889063006e-02 + 3.31962688978e-02 3.62670291685e-02 3.82416345368e-02 3.91195978595e-02 3.91195978595e-02 3.82416345368e-02 + 3.38926655662e-02 3.04051608500e-02 2.73901656280e-02 2.61753235783e-02 2.73901656280e-02 3.04051608500e-02 + 3.38926655663e-02 3.66852084288e-02 3.82416345368e-02 3.86928693264e-02 3.82416345368e-02 3.66852084288e-02 + 3.38926655663e-02 3.04051608500e-02 2.73901656280e-02 2.61753235783e-02 2.73901656280e-02 3.04051608500e-02 + 3.38926655663e-02 3.66852084288e-02 3.82416345368e-02 3.86928693264e-02 3.82416345368e-02 3.66852084288e-02 + 2.62414574798e-02 2.55789017267e-02 2.71139620581e-02 3.04771749334e-02 3.45751721123e-02 3.89797931477e-02 + 4.36602554331e-02 4.76129372810e-02 4.91898793330e-02 4.76129372810e-02 4.36602554332e-02 3.89797931478e-02 + 3.45751721123e-02 3.04771749334e-02 2.71139620581e-02 2.55789017267e-02 2.62414574798e-02 2.81654297894e-02 + 2.99300705585e-02 3.07818020073e-02 3.09603631544e-02 3.07818020073e-02 2.99300705585e-02 2.81654297894e-02 + 2.55789017267e-02 2.72047829307e-02 3.05546433144e-02 3.46097554135e-02 4.00844862497e-02 4.86719445336e-02 + 5.87904239654e-02 6.56429700804e-02 6.56429700805e-02 5.87904239654e-02 4.86719445337e-02 4.00844862497e-02 + 3.46097554135e-02 3.05546433144e-02 2.72047829307e-02 2.55789017267e-02 2.59617964748e-02 2.73091130234e-02 + 2.84215335265e-02 2.88774182856e-02 2.88774182856e-02 2.84215335265e-02 2.73091130234e-02 2.59617964748e-02 + 2.71139620581e-02 3.05546433144e-02 3.44218462469e-02 4.03066875222e-02 5.24377104592e-02 6.98261770599e-02 + 8.44125475990e-02 8.97670392509e-02 8.44125475991e-02 6.98261770600e-02 5.24377104593e-02 4.03066875223e-02 + 3.44218462469e-02 3.05546433144e-02 2.71139620581e-02 2.51928884334e-02 2.49962561202e-02 2.55724702285e-02 + 2.60490352762e-02 2.61899431515e-02 2.60490352762e-02 2.55724702285e-02 2.49962561202e-02 2.51928884334e-02 + 3.04771749334e-02 3.46097554135e-02 4.03066875222e-02 5.38559646463e-02 7.65079897134e-02 9.74791227832e-02 + 1.07505424132e-01 1.07505424132e-01 9.74791227833e-02 7.65079897136e-02 5.38559646464e-02 4.03066875223e-02 + 3.46097554135e-02 3.04771749334e-02 2.65397992490e-02 2.40837535674e-02 2.32028796117e-02 2.30937481116e-02 + 2.31229887194e-02 2.31229887194e-02 2.30937481115e-02 2.32028796117e-02 2.40837535674e-02 2.65397992490e-02 + 3.45751721122e-02 4.00844862497e-02 5.24377104592e-02 7.65079897134e-02 1.01689797158e-01 1.13480556613e-01 + 1.14984832163e-01 1.13480556613e-01 1.01689797158e-01 7.65079897136e-02 5.24377104593e-02 4.00844862497e-02 + 3.45751721123e-02 2.95889063006e-02 2.49962561202e-02 2.20776282425e-02 2.07195371247e-02 2.02039597509e-02 + 2.00652756289e-02 2.02039597509e-02 2.07195371247e-02 2.20776282425e-02 2.49962561202e-02 2.95889063006e-02 + 3.89797931477e-02 4.86719445336e-02 6.98261770599e-02 9.74791227832e-02 1.13480556613e-01 1.14285203523e-01 + 1.14285203523e-01 1.13480556613e-01 9.74791227833e-02 6.98261770600e-02 4.86719445337e-02 3.89797931478e-02 + 3.31962688978e-02 2.73091130234e-02 2.24043544384e-02 1.93870166210e-02 1.78887124933e-02 1.72810954144e-02 + 1.72810954144e-02 1.78887124933e-02 1.93870166210e-02 2.24043544384e-02 2.73091130234e-02 3.31962688978e-02 + 4.36602554331e-02 5.87904239654e-02 8.44125475990e-02 1.07505424132e-01 1.14984832163e-01 1.14285203523e-01 + 1.14984832163e-01 1.07505424132e-01 8.44125475991e-02 5.87904239654e-02 4.36602554332e-02 3.62670291685e-02 + 2.99300705585e-02 2.37302554132e-02 1.91106183945e-02 1.63905844408e-02 1.50423980149e-02 1.46288065643e-02 + 1.50423980149e-02 1.63905844408e-02 1.91106183945e-02 2.37302554132e-02 2.99300705585e-02 3.62670291685e-02 + 4.76129372810e-02 6.56429700805e-02 8.97670392509e-02 1.07505424132e-01 1.13480556613e-01 1.13480556613e-01 + 1.07505424132e-01 8.97670392509e-02 6.56429700805e-02 4.76129372810e-02 3.82416345368e-02 3.17358439118e-02 + 2.51250422144e-02 1.94291633463e-02 1.55937775675e-02 1.34169422858e-02 1.24266085814e-02 1.24266085814e-02 + 1.34169422858e-02 1.55937775675e-02 1.94291633463e-02 2.51250422144e-02 3.17358439118e-02 3.82416345368e-02 + 4.91898793330e-02 6.56429700805e-02 8.44125475991e-02 9.74791227833e-02 1.01689797158e-01 9.74791227833e-02 + 8.44125475991e-02 6.56429700805e-02 4.91898793330e-02 3.91195978595e-02 3.24122722065e-02 2.58521406843e-02 + 1.97046043739e-02 1.51356003261e-02 1.22801184033e-02 1.07226215343e-02 1.02154655471e-02 1.07226215343e-02 + 1.22801184033e-02 1.51356003261e-02 1.97046043739e-02 2.58521406843e-02 3.24122722065e-02 3.91195978595e-02 + 4.76129372810e-02 5.87904239654e-02 6.98261770600e-02 7.65079897136e-02 7.65079897136e-02 6.98261770600e-02 + 5.87904239654e-02 4.76129372810e-02 3.91195978595e-02 3.25035303326e-02 2.59776327244e-02 1.96803910193e-02 + 1.47280857174e-02 1.14482068791e-02 9.47298406613e-03 8.50676845052e-03 8.50676845052e-03 9.47298406612e-03 + 1.14482068790e-02 1.47280857174e-02 1.96803910193e-02 2.59776327244e-02 3.25035303326e-02 3.91195978595e-02 + 4.36602554332e-02 4.86719445337e-02 5.24377104593e-02 5.38559646464e-02 5.24377104593e-02 4.86719445337e-02 + 4.36602554332e-02 3.82416345368e-02 3.24122722065e-02 2.59776327244e-02 1.96033035629e-02 1.44211937670e-02 + 1.08700745212e-02 8.62544633520e-03 7.31939690082e-03 6.87435772715e-03 7.31939690081e-03 8.62544633519e-03 + 1.08700745212e-02 1.44211937670e-02 1.96033035629e-02 2.59776327244e-02 3.24122722064e-02 3.82416345368e-02 + 3.89797931478e-02 4.00844862497e-02 4.03066875223e-02 4.03066875223e-02 4.00844862497e-02 3.89797931478e-02 + 3.62670291685e-02 3.17358439118e-02 2.58521406843e-02 1.96803910193e-02 1.44211937670e-02 1.06519075585e-02 + 8.18384117181e-03 6.62684941428e-03 5.82750449515e-03 5.82750449515e-03 6.62684941427e-03 8.18384117181e-03 + 1.06519075585e-02 1.44211937670e-02 1.96803910193e-02 2.58521406843e-02 3.17358439118e-02 3.62670291685e-02 + 3.45751721123e-02 3.46097554135e-02 3.44218462469e-02 3.46097554135e-02 3.45751721123e-02 3.31962688978e-02 + 2.99300705586e-02 2.51250422144e-02 1.97046043739e-02 1.47280857174e-02 1.08700745212e-02 8.18384117181e-03 + 6.39966515662e-03 5.32069114083e-03 4.94634941966e-03 5.32069114083e-03 6.39966515661e-03 8.18384117181e-03 + 1.08700745212e-02 1.47280857174e-02 1.97046043738e-02 2.51250422144e-02 2.99300705585e-02 3.31962688978e-02 + 3.04771749334e-02 3.05546433144e-02 3.05546433144e-02 3.04771749334e-02 2.95889063006e-02 2.73091130234e-02 + 2.37302554132e-02 1.94291633463e-02 1.51356003261e-02 1.14482068791e-02 8.62544633520e-03 6.62684941428e-03 + 5.32069114083e-03 4.65178002992e-03 4.65178002992e-03 5.32069114083e-03 6.62684941427e-03 8.62544633519e-03 + 1.14482068790e-02 1.51356003261e-02 1.94291633463e-02 2.37302554132e-02 2.73091130234e-02 2.95889063006e-02 + 2.71139620581e-02 2.72047829307e-02 2.71139620581e-02 2.65397992490e-02 2.49962561202e-02 2.24043544384e-02 + 1.91106183945e-02 1.55937775675e-02 1.22801184033e-02 9.47298406613e-03 7.31939690082e-03 5.82750449515e-03 + 4.94634941966e-03 4.65178002992e-03 4.94634941966e-03 5.82750449515e-03 7.31939690081e-03 9.47298406612e-03 + 1.22801184032e-02 1.55937775675e-02 1.91106183945e-02 2.24043544384e-02 2.49962561202e-02 2.65397992490e-02 + 2.55789017267e-02 2.55789017267e-02 2.51928884334e-02 2.40837535674e-02 2.20776282425e-02 1.93870166210e-02 + 1.63905844408e-02 1.34169422858e-02 1.07226215343e-02 8.50676845052e-03 6.87435772715e-03 5.82750449515e-03 + 5.32069114083e-03 5.32069114083e-03 5.82750449515e-03 6.87435772715e-03 8.50676845052e-03 1.07226215343e-02 + 1.34169422858e-02 1.63905844408e-02 1.93870166210e-02 2.20776282425e-02 2.40837535674e-02 2.51928884334e-02 + 2.62414574798e-02 2.59617964748e-02 2.49962561202e-02 2.32028796117e-02 2.07195371247e-02 1.78887124933e-02 + 1.50423980149e-02 1.24266085814e-02 1.02154655471e-02 8.50676845052e-03 7.31939690081e-03 6.62684941427e-03 + 6.39966515661e-03 6.62684941428e-03 7.31939690082e-03 8.50676845052e-03 1.02154655471e-02 1.24266085814e-02 + 1.50423980149e-02 1.78887124933e-02 2.07195371247e-02 2.32028796117e-02 2.49962561202e-02 2.59617964748e-02 + 2.81654297894e-02 2.73091130234e-02 2.55724702285e-02 2.30937481115e-02 2.02039597509e-02 1.72810954144e-02 + 1.46288065643e-02 1.24266085814e-02 1.07226215343e-02 9.47298406612e-03 8.62544633519e-03 8.18384117181e-03 + 8.18384117181e-03 8.62544633519e-03 9.47298406612e-03 1.07226215343e-02 1.24266085814e-02 1.46288065643e-02 + 1.72810954144e-02 2.02039597509e-02 2.30937481116e-02 2.55724702285e-02 2.73091130234e-02 2.81654297894e-02 + 2.99300705585e-02 2.84215335265e-02 2.60490352762e-02 2.31229887194e-02 2.00652756288e-02 1.72810954144e-02 + 1.50423980149e-02 1.34169422858e-02 1.22801184033e-02 1.14482068790e-02 1.08700745212e-02 1.06519075585e-02 + 1.08700745212e-02 1.14482068790e-02 1.22801184032e-02 1.34169422858e-02 1.50423980149e-02 1.72810954144e-02 + 2.00652756289e-02 2.31229887194e-02 2.60490352762e-02 2.84215335265e-02 2.99300705585e-02 3.04405035973e-02 + 3.07818020073e-02 2.88774182856e-02 2.61899431515e-02 2.31229887194e-02 2.02039597509e-02 1.78887124933e-02 + 1.63905844408e-02 1.55937775675e-02 1.51356003261e-02 1.47280857174e-02 1.44211937670e-02 1.44211937670e-02 + 1.47280857174e-02 1.51356003261e-02 1.55937775675e-02 1.63905844408e-02 1.78887124933e-02 2.02039597509e-02 + 2.31229887194e-02 2.61899431515e-02 2.88774182856e-02 3.07818020073e-02 3.17358439118e-02 3.17358439118e-02 + 3.09603631544e-02 2.88774182856e-02 2.60490352762e-02 2.30937481115e-02 2.07195371247e-02 1.93870166210e-02 + 1.91106183945e-02 1.94291633463e-02 1.97046043739e-02 1.96803910193e-02 1.96033035629e-02 1.96803910193e-02 + 1.97046043738e-02 1.94291633463e-02 1.91106183945e-02 1.93870166210e-02 2.07195371247e-02 2.30937481116e-02 + 2.60490352762e-02 2.88774182856e-02 3.09603631544e-02 3.20842088246e-02 3.24122722065e-02 3.20842088246e-02 + 3.07818020073e-02 2.84215335265e-02 2.55724702285e-02 2.32028796117e-02 2.20776282425e-02 2.24043544384e-02 + 2.37302554132e-02 2.51250422144e-02 2.58521406843e-02 2.59776327244e-02 2.59776327244e-02 2.58521406843e-02 + 2.51250422144e-02 2.37302554132e-02 2.24043544384e-02 2.20776282425e-02 2.32028796117e-02 2.55724702285e-02 + 2.84215335265e-02 3.07818020073e-02 3.20842088246e-02 3.25035303326e-02 3.25035303326e-02 3.20842088246e-02 + 2.99300705585e-02 2.73091130234e-02 2.49962561202e-02 2.40837535673e-02 2.49962561202e-02 2.73091130234e-02 + 2.99300705585e-02 3.17358439118e-02 3.24122722065e-02 3.25035303326e-02 3.24122722064e-02 3.17358439118e-02 + 2.99300705585e-02 2.73091130234e-02 2.49962561202e-02 2.40837535673e-02 2.49962561202e-02 2.73091130234e-02 + 2.99300705585e-02 3.17358439118e-02 3.24122722064e-02 3.25035303326e-02 3.24122722065e-02 3.17358439118e-02 + 2.81654297894e-02 2.59617964748e-02 2.51928884334e-02 2.65397992490e-02 2.95889063006e-02 3.31962688978e-02 + 3.62670291685e-02 3.82416345368e-02 3.91195978595e-02 3.91195978594e-02 3.82416345368e-02 3.62670291685e-02 + 3.31962688978e-02 2.95889063006e-02 2.65397992490e-02 2.51928884334e-02 2.59617964748e-02 2.81654297894e-02 + 3.04405035973e-02 3.17358439118e-02 3.20842088246e-02 3.20842088246e-02 3.17358439118e-02 3.04405035973e-02 + 2.33116271285e-02 2.52442663181e-02 2.89472257765e-02 3.33648425996e-02 3.75326780270e-02 4.14580711769e-02 + 4.51573443066e-02 4.76129372810e-02 4.76129372810e-02 4.51573443066e-02 4.14580711769e-02 3.75326780270e-02 + 3.33648425996e-02 2.89472257765e-02 2.52442663181e-02 2.33116271285e-02 2.31387782150e-02 2.37302554132e-02 + 2.40966608398e-02 2.40902041017e-02 2.40902041017e-02 2.40966608398e-02 2.37302554132e-02 2.31387782150e-02 + 2.52442663181e-02 2.92642577664e-02 3.39946385756e-02 3.86301228830e-02 4.42642004890e-02 5.17384028835e-02 + 5.87904239654e-02 6.17010347926e-02 5.87904239654e-02 5.17384028836e-02 4.42642004891e-02 3.86301228830e-02 + 3.39946385756e-02 2.92642577664e-02 2.52442663181e-02 2.30251854758e-02 2.24043544384e-02 2.23752062804e-02 + 2.22711776717e-02 2.21707375456e-02 2.22711776717e-02 2.23752062804e-02 2.24043544384e-02 2.30251854758e-02 + 2.89472257765e-02 3.39946385756e-02 3.88820048627e-02 4.56997211399e-02 5.69445178736e-02 6.98261770599e-02 + 7.81199920259e-02 7.81199920260e-02 6.98261770600e-02 5.69445178737e-02 4.56997211400e-02 3.88820048628e-02 + 3.39946385756e-02 2.89472257765e-02 2.46284218885e-02 2.20776282425e-02 2.09446150107e-02 2.03920068264e-02 + 2.00581600358e-02 2.00581600358e-02 2.03920068264e-02 2.09446150107e-02 2.20776282425e-02 2.46284218885e-02 + 3.33648425996e-02 3.86301228830e-02 4.56997211399e-02 5.89470993594e-02 7.65079897135e-02 9.00462898607e-02 + 9.46627422162e-02 9.00462898608e-02 7.65079897136e-02 5.89470993595e-02 4.56997211400e-02 3.86301228830e-02 + 3.33648425996e-02 2.78012163806e-02 2.32028796117e-02 2.03982633679e-02 1.89263306496e-02 1.81429746336e-02 + 1.78656633258e-02 1.81429746336e-02 1.89263306496e-02 2.03982633679e-02 2.32028796117e-02 2.78012163806e-02 + 3.75326780270e-02 4.42642004890e-02 5.69445178736e-02 7.65079897135e-02 9.40547983345e-02 1.02256358961e-01 + 1.02256358961e-01 9.40547983346e-02 7.65079897135e-02 5.69445178737e-02 4.42642004890e-02 3.75326780270e-02 + 3.16055353689e-02 2.55724702285e-02 2.09446150107e-02 1.81502745082e-02 1.66389072281e-02 1.59286730225e-02 + 1.59286730225e-02 1.66389072281e-02 1.81502745082e-02 2.09446150107e-02 2.55724702285e-02 3.16055353689e-02 + 4.14580711769e-02 5.17384028835e-02 6.98261770599e-02 9.00462898607e-02 1.02256358961e-01 1.05383989813e-01 + 1.02256358961e-01 9.00462898608e-02 6.98261770600e-02 5.17384028835e-02 4.14580711769e-02 3.49765243617e-02 + 2.84215335265e-02 2.23752062804e-02 1.81179478705e-02 1.56224103699e-02 1.43188156133e-02 1.38961510916e-02 + 1.43188156133e-02 1.56224103699e-02 1.81179478705e-02 2.23752062804e-02 2.84215335265e-02 3.49765243617e-02 + 4.51573443066e-02 5.87904239654e-02 7.81199920259e-02 9.46627422161e-02 1.02256358961e-01 1.02256358961e-01 + 9.46627422161e-02 7.81199920259e-02 5.87904239654e-02 4.51573443066e-02 3.73533616181e-02 3.07818020073e-02 + 2.40966608398e-02 1.86355831047e-02 1.50739753351e-02 1.30486211596e-02 1.21045641480e-02 1.21045641480e-02 + 1.30486211596e-02 1.50739753351e-02 1.86355831047e-02 2.40966608398e-02 3.07818020073e-02 3.73533616181e-02 + 4.76129372810e-02 6.17010347926e-02 7.81199920260e-02 9.00462898608e-02 9.40547983346e-02 9.00462898608e-02 + 7.81199920259e-02 6.17010347926e-02 4.76129372810e-02 3.86928693264e-02 3.20842088246e-02 2.53972544857e-02 + 1.92841419762e-02 1.48493143122e-02 1.21091269495e-02 1.06110221336e-02 1.01199573737e-02 1.06110221336e-02 + 1.21091269494e-02 1.48493143122e-02 1.92841419762e-02 2.53972544857e-02 3.20842088246e-02 3.86928693264e-02 + 4.76129372810e-02 5.87904239654e-02 6.98261770600e-02 7.65079897135e-02 7.65079897135e-02 6.98261770599e-02 + 5.87904239654e-02 4.76129372810e-02 3.91195978595e-02 3.25035303326e-02 2.59776327244e-02 1.96803910193e-02 + 1.47280857174e-02 1.14482068790e-02 9.47298406613e-03 8.50676845052e-03 8.50676845052e-03 9.47298406612e-03 + 1.14482068790e-02 1.47280857174e-02 1.96803910193e-02 2.59776327244e-02 3.25035303326e-02 3.91195978595e-02 + 4.51573443066e-02 5.17384028836e-02 5.69445178737e-02 5.89470993595e-02 5.69445178737e-02 5.17384028835e-02 + 4.51573443066e-02 3.86928693264e-02 3.25035303326e-02 2.60946322497e-02 1.97873550487e-02 1.45988685534e-02 + 1.09991800129e-02 8.70260576991e-03 7.36059222257e-03 6.90318266868e-03 7.36059222257e-03 8.70260576990e-03 + 1.09991800129e-02 1.45988685534e-02 1.97873550487e-02 2.60946322497e-02 3.25035303326e-02 3.86928693264e-02 + 4.14580711769e-02 4.42642004891e-02 4.56997211400e-02 4.56997211400e-02 4.42642004890e-02 4.14580711769e-02 + 3.73533616181e-02 3.20842088246e-02 2.59776327244e-02 1.97873550487e-02 1.45323142804e-02 1.07548084489e-02 + 8.26411545329e-03 6.67440432828e-03 5.85081359877e-03 5.85081359876e-03 6.67440432828e-03 8.26411545328e-03 + 1.07548084489e-02 1.45323142804e-02 1.97873550487e-02 2.59776327244e-02 3.20842088246e-02 3.73533616181e-02 + 3.75326780270e-02 3.86301228830e-02 3.88820048628e-02 3.86301228830e-02 3.75326780270e-02 3.49765243617e-02 + 3.07818020073e-02 2.53972544857e-02 1.96803910193e-02 1.45988685534e-02 1.07548084489e-02 8.12162338295e-03 + 6.36712108007e-03 5.28637047081e-03 4.90592224711e-03 5.28637047081e-03 6.36712108007e-03 8.12162338295e-03 + 1.07548084489e-02 1.45988685534e-02 1.96803910193e-02 2.53972544857e-02 3.07818020073e-02 3.49765243617e-02 + 3.33648425996e-02 3.39946385756e-02 3.39946385756e-02 3.33648425996e-02 3.16055353689e-02 2.84215335265e-02 + 2.40966608398e-02 1.92841419762e-02 1.47280857174e-02 1.09991800129e-02 8.26411545329e-03 6.36712108007e-03 + 5.11302992589e-03 4.45369745770e-03 4.45369745770e-03 5.11302992589e-03 6.36712108007e-03 8.26411545328e-03 + 1.09991800129e-02 1.47280857174e-02 1.92841419762e-02 2.40966608398e-02 2.84215335265e-02 3.16055353689e-02 + 2.89472257765e-02 2.92642577664e-02 2.89472257765e-02 2.78012163806e-02 2.55724702285e-02 2.23752062804e-02 + 1.86355831047e-02 1.48493143122e-02 1.14482068790e-02 8.70260576991e-03 6.67440432828e-03 5.28637047081e-03 + 4.45369745770e-03 4.16949135529e-03 4.45369745770e-03 5.28637047081e-03 6.67440432828e-03 8.70260576990e-03 + 1.14482068790e-02 1.48493143122e-02 1.86355831047e-02 2.23752062804e-02 2.55724702285e-02 2.78012163806e-02 + 2.52442663181e-02 2.52442663181e-02 2.46284218885e-02 2.32028796117e-02 2.09446150107e-02 1.81179478705e-02 + 1.50739753351e-02 1.21091269494e-02 9.47298406612e-03 7.36059222257e-03 5.85081359877e-03 4.90592224711e-03 + 4.45369745770e-03 4.45369745770e-03 4.90592224711e-03 5.85081359877e-03 7.36059222257e-03 9.47298406612e-03 + 1.21091269494e-02 1.50739753351e-02 1.81179478705e-02 2.09446150107e-02 2.32028796117e-02 2.46284218885e-02 + 2.33116271285e-02 2.30251854758e-02 2.20776282425e-02 2.03982633679e-02 1.81502745082e-02 1.56224103699e-02 + 1.30486211596e-02 1.06110221336e-02 8.50676845052e-03 6.90318266868e-03 5.85081359877e-03 5.28637047081e-03 + 5.11302992589e-03 5.28637047081e-03 5.85081359877e-03 6.90318266868e-03 8.50676845052e-03 1.06110221336e-02 + 1.30486211596e-02 1.56224103699e-02 1.81502745082e-02 2.03982633679e-02 2.20776282425e-02 2.30251854758e-02 + 2.31387782150e-02 2.24043544384e-02 2.09446150107e-02 1.89263306496e-02 1.66389072281e-02 1.43188156133e-02 + 1.21045641480e-02 1.01199573737e-02 8.50676845052e-03 7.36059222257e-03 6.67440432828e-03 6.36712108007e-03 + 6.36712108007e-03 6.67440432828e-03 7.36059222257e-03 8.50676845052e-03 1.01199573737e-02 1.21045641480e-02 + 1.43188156133e-02 1.66389072281e-02 1.89263306496e-02 2.09446150107e-02 2.24043544384e-02 2.31387782150e-02 + 2.37302554132e-02 2.23752062804e-02 2.03920068264e-02 1.81429746336e-02 1.59286730224e-02 1.38961510916e-02 + 1.21045641480e-02 1.06110221336e-02 9.47298406612e-03 8.70260576990e-03 8.26411545329e-03 8.12162338295e-03 + 8.26411545329e-03 8.70260576990e-03 9.47298406612e-03 1.06110221336e-02 1.21045641480e-02 1.38961510916e-02 + 1.59286730225e-02 1.81429746336e-02 2.03920068264e-02 2.23752062804e-02 2.37302554132e-02 2.42076546579e-02 + 2.40966608398e-02 2.22711776717e-02 2.00581600358e-02 1.78656633258e-02 1.59286730225e-02 1.43188156133e-02 + 1.30486211596e-02 1.21091269494e-02 1.14482068790e-02 1.09991800129e-02 1.07548084489e-02 1.07548084489e-02 + 1.09991800129e-02 1.14482068790e-02 1.21091269494e-02 1.30486211596e-02 1.43188156133e-02 1.59286730225e-02 + 1.78656633258e-02 2.00581600358e-02 2.22711776717e-02 2.40966608398e-02 2.51250422144e-02 2.51250422144e-02 + 2.40902041017e-02 2.21707375455e-02 2.00581600358e-02 1.81429746336e-02 1.66389072281e-02 1.56224103699e-02 + 1.50739753351e-02 1.48493143122e-02 1.47280857174e-02 1.45988685534e-02 1.45323142804e-02 1.45988685534e-02 + 1.47280857174e-02 1.48493143122e-02 1.50739753351e-02 1.56224103699e-02 1.66389072281e-02 1.81429746336e-02 + 2.00581600358e-02 2.21707375456e-02 2.40902041017e-02 2.53972544857e-02 2.58521406843e-02 2.53972544857e-02 + 2.40902041017e-02 2.22711776717e-02 2.03920068264e-02 1.89263306496e-02 1.81502745082e-02 1.81179478705e-02 + 1.86355831047e-02 1.92841419762e-02 1.96803910193e-02 1.97873550487e-02 1.97873550487e-02 1.96803910193e-02 + 1.92841419762e-02 1.86355831047e-02 1.81179478705e-02 1.81502745082e-02 1.89263306496e-02 2.03920068264e-02 + 2.22711776717e-02 2.40902041017e-02 2.53795329069e-02 2.59776327244e-02 2.59776327244e-02 2.53795329069e-02 + 2.40966608398e-02 2.23752062804e-02 2.09446150107e-02 2.03982633679e-02 2.09446150107e-02 2.23752062804e-02 + 2.40966608398e-02 2.53972544857e-02 2.59776327244e-02 2.60946322497e-02 2.59776327244e-02 2.53972544857e-02 + 2.40966608398e-02 2.23752062804e-02 2.09446150107e-02 2.03982633679e-02 2.09446150107e-02 2.23752062804e-02 + 2.40966608398e-02 2.53972544857e-02 2.59776327244e-02 2.60946322497e-02 2.59776327244e-02 2.53972544857e-02 + 2.37302554132e-02 2.24043544384e-02 2.20776282425e-02 2.32028796117e-02 2.55724702285e-02 2.84215335265e-02 + 3.07818020073e-02 3.20842088246e-02 3.25035303326e-02 3.25035303326e-02 3.20842088246e-02 3.07818020073e-02 + 2.84215335265e-02 2.55724702285e-02 2.32028796117e-02 2.20776282425e-02 2.24043544384e-02 2.37302554132e-02 + 2.51250422144e-02 2.58521406843e-02 2.59776327244e-02 2.59776327244e-02 2.58521406843e-02 2.51250422144e-02 + 2.31387782150e-02 2.30251854758e-02 2.46284218885e-02 2.78012163806e-02 3.16055353689e-02 3.49765243617e-02 + 3.73533616181e-02 3.86928693264e-02 3.91195978594e-02 3.86928693264e-02 3.73533616181e-02 3.49765243617e-02 + 3.16055353689e-02 2.78012163806e-02 2.46284218885e-02 2.30251854758e-02 2.31387782150e-02 2.42076546579e-02 + 2.51250422144e-02 2.53972544857e-02 2.53795329069e-02 2.53972544857e-02 2.51250422144e-02 2.42076546579e-02 + 2.21383854932e-02 2.59180287237e-02 3.06730045455e-02 3.51265406939e-02 3.86294226543e-02 4.14580711769e-02 + 4.36602554331e-02 4.45488243286e-02 4.36602554331e-02 4.14580711769e-02 3.86294226543e-02 3.51265406940e-02 + 3.06730045455e-02 2.59180287237e-02 2.21383854933e-02 1.99974571616e-02 1.91106183945e-02 1.86355831047e-02 + 1.81617249718e-02 1.79266369346e-02 1.81617249718e-02 1.86355831047e-02 1.91106183945e-02 1.99974571616e-02 + 2.59180287237e-02 3.12137668085e-02 3.61943858422e-02 4.02274592299e-02 4.42642004890e-02 4.86719445336e-02 + 5.18159940627e-02 5.18159940627e-02 4.86719445336e-02 4.42642004890e-02 4.02274592299e-02 3.61943858423e-02 + 3.12137668085e-02 2.59180287237e-02 2.17946809311e-02 1.93870166210e-02 1.81179478705e-02 1.72612102705e-02 + 1.66949823920e-02 1.66949823920e-02 1.72612102705e-02 1.81179478705e-02 1.93870166210e-02 2.17946809311e-02 + 3.06730045455e-02 3.61943858422e-02 4.06895832707e-02 4.56997211399e-02 5.24377104592e-02 5.87813204090e-02 + 6.13627808209e-02 5.87813204090e-02 5.24377104593e-02 4.56997211400e-02 4.06895832708e-02 3.61943858423e-02 + 3.06730045455e-02 2.49901046905e-02 2.07195371247e-02 1.81502745082e-02 1.66276495780e-02 1.56490177739e-02 + 1.52718979502e-02 1.56490177739e-02 1.66276495780e-02 1.81502745082e-02 2.07195371247e-02 2.49901046905e-02 + 3.51265406939e-02 4.02274592298e-02 4.56997211399e-02 5.38559646463e-02 6.30067497701e-02 6.87952783262e-02 + 6.87952783262e-02 6.30067497701e-02 5.38559646463e-02 4.56997211400e-02 4.02274592299e-02 3.51265406940e-02 + 2.89589200657e-02 2.30937481116e-02 1.89263306496e-02 1.64062970608e-02 1.48932680012e-02 1.40901757966e-02 + 1.40901757966e-02 1.48932680012e-02 1.64062970608e-02 1.89263306496e-02 2.30937481115e-02 2.89589200656e-02 + 3.86294226543e-02 4.42642004890e-02 5.24377104592e-02 6.30067497701e-02 7.14597362941e-02 7.44580546054e-02 + 7.14597362941e-02 6.30067497701e-02 5.24377104593e-02 4.42642004890e-02 3.86294226543e-02 3.27027225515e-02 + 2.60490352762e-02 2.03920068264e-02 1.66276495780e-02 1.43895601245e-02 1.31235074234e-02 1.26855892424e-02 + 1.31235074234e-02 1.43895601245e-02 1.66276495780e-02 2.03920068263e-02 2.60490352762e-02 3.27027225515e-02 + 4.14580711769e-02 4.86719445336e-02 5.87813204090e-02 6.87952783262e-02 7.44580546054e-02 7.44580546054e-02 + 6.87952783262e-02 5.87813204090e-02 4.86719445336e-02 4.14580711769e-02 3.55400935242e-02 2.88774182856e-02 + 2.22711776717e-02 1.72612102705e-02 1.41150555464e-02 1.22935034476e-02 1.14032634249e-02 1.14032634249e-02 + 1.22935034476e-02 1.41150555464e-02 1.72612102704e-02 2.22711776717e-02 2.88774182856e-02 3.55400935242e-02 + 4.36602554331e-02 5.18159940626e-02 6.13627808208e-02 6.87952783262e-02 7.14597362941e-02 6.87952783261e-02 + 6.13627808208e-02 5.18159940626e-02 4.36602554331e-02 3.73533616181e-02 3.09603631544e-02 2.40902041017e-02 + 1.81617249718e-02 1.40807237278e-02 1.16096263393e-02 1.02402112845e-02 9.78230850882e-03 1.02402112845e-02 + 1.16096263393e-02 1.40807237278e-02 1.81617249718e-02 2.40902041017e-02 3.09603631544e-02 3.73533616181e-02 + 4.45488243286e-02 5.18159940627e-02 5.87813204090e-02 6.30067497701e-02 6.30067497701e-02 5.87813204090e-02 + 5.18159940626e-02 4.45488243286e-02 3.82416345368e-02 3.20842088246e-02 2.53795329070e-02 1.90346417875e-02 + 1.42309947893e-02 1.11332793734e-02 9.28365371522e-03 8.37411495641e-03 8.37411495641e-03 9.28365371522e-03 + 1.11332793734e-02 1.42309947893e-02 1.90346417875e-02 2.53795329070e-02 3.20842088246e-02 3.82416345368e-02 + 4.36602554331e-02 4.86719445336e-02 5.24377104593e-02 5.38559646463e-02 5.24377104592e-02 4.86719445336e-02 + 4.36602554331e-02 3.82416345368e-02 3.24122722065e-02 2.59776327244e-02 1.96033035629e-02 1.44211937670e-02 + 1.08700745212e-02 8.62544633520e-03 7.31939690082e-03 6.87435772716e-03 7.31939690082e-03 8.62544633519e-03 + 1.08700745212e-02 1.44211937670e-02 1.96033035629e-02 2.59776327244e-02 3.24122722065e-02 3.82416345368e-02 + 4.14580711769e-02 4.42642004890e-02 4.56997211400e-02 4.56997211400e-02 4.42642004890e-02 4.14580711769e-02 + 3.73533616181e-02 3.20842088246e-02 2.59776327244e-02 1.97873550487e-02 1.45323142804e-02 1.07548084489e-02 + 8.26411545329e-03 6.67440432828e-03 5.85081359877e-03 5.85081359877e-03 6.67440432828e-03 8.26411545328e-03 + 1.07548084489e-02 1.45323142804e-02 1.97873550487e-02 2.59776327244e-02 3.20842088246e-02 3.73533616181e-02 + 3.86294226544e-02 4.02274592299e-02 4.06895832708e-02 4.02274592299e-02 3.86294226544e-02 3.55400935242e-02 + 3.09603631544e-02 2.53795329070e-02 1.96033035629e-02 1.45323142804e-02 1.07235160838e-02 8.11876205861e-03 + 6.36994587767e-03 5.27979197806e-03 4.89322201917e-03 5.27979197805e-03 6.36994587766e-03 8.11876205861e-03 + 1.07235160838e-02 1.45323142804e-02 1.96033035629e-02 2.53795329069e-02 3.09603631544e-02 3.55400935242e-02 + 3.51265406940e-02 3.61943858423e-02 3.61943858423e-02 3.51265406940e-02 3.27027225515e-02 2.88774182856e-02 + 2.40902041017e-02 1.90346417875e-02 1.44211937670e-02 1.07548084489e-02 8.11876205861e-03 6.28716491126e-03 + 5.04613651753e-03 4.37602725379e-03 4.37602725379e-03 5.04613651753e-03 6.28716491126e-03 8.11876205861e-03 + 1.07548084489e-02 1.44211937670e-02 1.90346417875e-02 2.40902041017e-02 2.88774182856e-02 3.27027225515e-02 + 3.06730045455e-02 3.12137668085e-02 3.06730045455e-02 2.89589200657e-02 2.60490352763e-02 2.22711776717e-02 + 1.81617249718e-02 1.42309947893e-02 1.08700745212e-02 8.26411545329e-03 6.36994587767e-03 5.04613651753e-03 + 4.21587397243e-03 3.92286252630e-03 4.21587397243e-03 5.04613651753e-03 6.36994587766e-03 8.26411545328e-03 + 1.08700745212e-02 1.42309947893e-02 1.81617249718e-02 2.22711776717e-02 2.60490352762e-02 2.89589200657e-02 + 2.59180287237e-02 2.59180287237e-02 2.49901046905e-02 2.30937481116e-02 2.03920068264e-02 1.72612102705e-02 + 1.40807237278e-02 1.11332793734e-02 8.62544633520e-03 6.67440432828e-03 5.27979197806e-03 4.37602725379e-03 + 3.92286252630e-03 3.92286252630e-03 4.37602725379e-03 5.27979197805e-03 6.67440432828e-03 8.62544633519e-03 + 1.11332793734e-02 1.40807237278e-02 1.72612102704e-02 2.03920068264e-02 2.30937481116e-02 2.49901046905e-02 + 2.21383854932e-02 2.17946809311e-02 2.07195371247e-02 1.89263306496e-02 1.66276495780e-02 1.41150555464e-02 + 1.16096263393e-02 9.28365371522e-03 7.31939690082e-03 5.85081359877e-03 4.89322201917e-03 4.37602725379e-03 + 4.21587397243e-03 4.37602725379e-03 4.89322201917e-03 5.85081359876e-03 7.31939690081e-03 9.28365371521e-03 + 1.16096263393e-02 1.41150555464e-02 1.66276495780e-02 1.89263306496e-02 2.07195371247e-02 2.17946809311e-02 + 1.99974571616e-02 1.93870166210e-02 1.81502745082e-02 1.64062970608e-02 1.43895601245e-02 1.22935034476e-02 + 1.02402112845e-02 8.37411495641e-03 6.87435772716e-03 5.85081359877e-03 5.27979197805e-03 5.04613651753e-03 + 5.04613651753e-03 5.27979197805e-03 5.85081359877e-03 6.87435772715e-03 8.37411495641e-03 1.02402112845e-02 + 1.22935034476e-02 1.43895601245e-02 1.64062970608e-02 1.81502745082e-02 1.93870166210e-02 1.99974571616e-02 + 1.91106183945e-02 1.81179478705e-02 1.66276495780e-02 1.48932680012e-02 1.31235074234e-02 1.14032634249e-02 + 9.78230850882e-03 8.37411495641e-03 7.31939690082e-03 6.67440432828e-03 6.36994587767e-03 6.28716491126e-03 + 6.36994587767e-03 6.67440432828e-03 7.31939690082e-03 8.37411495641e-03 9.78230850881e-03 1.14032634249e-02 + 1.31235074234e-02 1.48932680012e-02 1.66276495780e-02 1.81179478705e-02 1.91106183945e-02 1.94536912925e-02 + 1.86355831047e-02 1.72612102704e-02 1.56490177739e-02 1.40901757966e-02 1.26855892423e-02 1.14032634249e-02 + 1.02402112845e-02 9.28365371522e-03 8.62544633519e-03 8.26411545328e-03 8.11876205861e-03 8.11876205861e-03 + 8.26411545329e-03 8.62544633519e-03 9.28365371522e-03 1.02402112845e-02 1.14032634249e-02 1.26855892423e-02 + 1.40901757966e-02 1.56490177739e-02 1.72612102705e-02 1.86355831047e-02 1.94291633463e-02 1.94291633463e-02 + 1.81617249718e-02 1.66949823920e-02 1.52718979502e-02 1.40901757966e-02 1.31235074234e-02 1.22935034476e-02 + 1.16096263393e-02 1.11332793734e-02 1.08700745212e-02 1.07548084489e-02 1.07235160838e-02 1.07548084489e-02 + 1.08700745212e-02 1.11332793734e-02 1.16096263393e-02 1.22935034476e-02 1.31235074234e-02 1.40901757966e-02 + 1.52718979502e-02 1.66949823920e-02 1.81617249718e-02 1.92841419762e-02 1.97046043738e-02 1.92841419762e-02 + 1.79266369346e-02 1.66949823920e-02 1.56490177739e-02 1.48932680012e-02 1.43895601245e-02 1.41150555464e-02 + 1.40807237278e-02 1.42309947893e-02 1.44211937670e-02 1.45323142804e-02 1.45323142804e-02 1.44211937670e-02 + 1.42309947893e-02 1.40807237278e-02 1.41150555464e-02 1.43895601245e-02 1.48932680012e-02 1.56490177739e-02 + 1.66949823920e-02 1.79266369346e-02 1.90346417875e-02 1.96803910193e-02 1.96803910193e-02 1.90346417875e-02 + 1.81617249718e-02 1.72612102705e-02 1.66276495780e-02 1.64062970608e-02 1.66276495780e-02 1.72612102704e-02 + 1.81617249718e-02 1.90346417875e-02 1.96033035629e-02 1.97873550487e-02 1.96033035629e-02 1.90346417875e-02 + 1.81617249718e-02 1.72612102704e-02 1.66276495780e-02 1.64062970608e-02 1.66276495780e-02 1.72612102705e-02 + 1.81617249718e-02 1.90346417874e-02 1.96033035629e-02 1.97873550487e-02 1.96033035629e-02 1.90346417875e-02 + 1.86355831047e-02 1.81179478705e-02 1.81502745082e-02 1.89263306496e-02 2.03920068264e-02 2.22711776717e-02 + 2.40902041017e-02 2.53795329069e-02 2.59776327244e-02 2.59776327244e-02 2.53795329069e-02 2.40902041017e-02 + 2.22711776717e-02 2.03920068264e-02 1.89263306496e-02 1.81502745082e-02 1.81179478705e-02 1.86355831047e-02 + 1.92841419762e-02 1.96803910193e-02 1.97873550487e-02 1.97873550487e-02 1.96803910193e-02 1.92841419762e-02 + 1.91106183945e-02 1.93870166210e-02 2.07195371247e-02 2.30937481115e-02 2.60490352762e-02 2.88774182856e-02 + 3.09603631544e-02 3.20842088246e-02 3.24122722065e-02 3.20842088246e-02 3.09603631544e-02 2.88774182856e-02 + 2.60490352762e-02 2.30937481115e-02 2.07195371247e-02 1.93870166210e-02 1.91106183945e-02 1.94291633463e-02 + 1.97046043738e-02 1.96803910193e-02 1.96033035629e-02 1.96803910193e-02 1.97046043739e-02 1.94291633463e-02 + 1.99974571616e-02 2.17946809311e-02 2.49901046905e-02 2.89589200656e-02 3.27027225515e-02 3.55400935242e-02 + 3.73533616181e-02 3.82416345368e-02 3.82416345368e-02 3.73533616181e-02 3.55400935242e-02 3.27027225515e-02 + 2.89589200657e-02 2.49901046905e-02 2.17946809311e-02 1.99974571616e-02 1.94536912926e-02 1.94291633463e-02 + 1.92841419762e-02 1.90346417874e-02 1.90346417875e-02 1.92841419762e-02 1.94291633463e-02 1.94536912926e-02 + 2.19609803409e-02 2.65859681267e-02 3.13659743853e-02 3.51265406939e-02 3.75326780270e-02 3.89797931477e-02 + 3.97328731661e-02 3.97328731661e-02 3.89797931477e-02 3.75326780270e-02 3.51265406940e-02 3.13659743854e-02 + 2.65859681267e-02 2.19609803409e-02 1.85131064743e-02 1.63905844408e-02 1.50739753351e-02 1.40807237278e-02 + 1.34421984781e-02 1.34421984781e-02 1.40807237278e-02 1.50739753351e-02 1.63905844408e-02 1.85131064743e-02 + 2.65859681267e-02 3.20076435873e-02 3.61943858422e-02 3.86301228830e-02 4.00844862497e-02 4.11390439155e-02 + 4.15728345981e-02 4.11390439155e-02 4.00844862497e-02 3.86301228830e-02 3.61943858423e-02 3.20076435873e-02 + 2.65859681267e-02 2.15194519516e-02 1.78887124933e-02 1.56224103699e-02 1.41150555464e-02 1.30619533075e-02 + 1.26512009150e-02 1.30619533075e-02 1.41150555464e-02 1.56224103699e-02 1.78887124933e-02 2.15194519516e-02 + 3.13659743853e-02 3.61943858422e-02 3.88820048627e-02 4.03066875223e-02 4.15214590889e-02 4.23899114590e-02 + 4.23899114590e-02 4.15214590889e-02 4.03066875223e-02 3.88820048627e-02 3.61943858423e-02 3.13659743854e-02 + 2.54054025138e-02 2.02039597509e-02 1.66389072281e-02 1.43895601245e-02 1.28938940154e-02 1.20440164119e-02 + 1.20440164119e-02 1.28938940154e-02 1.43895601245e-02 1.66389072281e-02 2.02039597509e-02 2.54054025138e-02 + 3.51265406939e-02 3.86301228830e-02 4.03066875223e-02 4.15474648475e-02 4.26144257223e-02 4.30325583083e-02 + 4.26144257223e-02 4.15474648474e-02 4.03066875222e-02 3.86301228830e-02 3.51265406940e-02 2.93978765598e-02 + 2.31229887194e-02 1.81429746336e-02 1.48932680013e-02 1.28581680779e-02 1.16030717026e-02 1.11481924118e-02 + 1.16030717026e-02 1.28581680779e-02 1.48932680012e-02 1.81429746336e-02 2.31229887194e-02 2.93978765598e-02 + 3.75326780270e-02 4.00844862497e-02 4.15214590889e-02 4.26144257223e-02 4.31923684876e-02 4.31923684876e-02 + 4.26144257223e-02 4.15214590889e-02 4.00844862497e-02 3.75326780270e-02 3.27027225515e-02 2.61899431515e-02 + 2.00581600358e-02 1.56490177739e-02 1.28938940154e-02 1.12138690779e-02 1.03423481784e-02 1.03423481784e-02 + 1.12138690779e-02 1.28938940154e-02 1.56490177739e-02 2.00581600358e-02 2.61899431515e-02 3.27027225515e-02 + 3.89797931477e-02 4.11390439155e-02 4.23899114590e-02 4.30325583083e-02 4.31923684876e-02 4.30325583083e-02 + 4.23899114590e-02 4.11390439154e-02 3.89797931477e-02 3.49765243617e-02 2.88774182856e-02 2.21707375456e-02 + 1.66949823920e-02 1.30619533075e-02 1.08501521193e-02 9.57630048387e-03 9.13735254995e-03 9.57630048387e-03 + 1.08501521193e-02 1.30619533075e-02 1.66949823920e-02 2.21707375456e-02 2.88774182856e-02 3.49765243617e-02 + 3.97328731661e-02 4.15728345981e-02 4.23899114590e-02 4.26144257223e-02 4.26144257223e-02 4.23899114589e-02 + 4.15728345981e-02 3.97328731661e-02 3.62670291685e-02 3.07818020073e-02 2.40902041017e-02 1.79266369346e-02 + 1.34421984781e-02 1.06052341643e-02 8.89928200876e-03 8.04465702446e-03 8.04465702446e-03 8.89928200876e-03 + 1.06052341643e-02 1.34421984781e-02 1.79266369346e-02 2.40902041017e-02 3.07818020073e-02 3.62670291685e-02 + 3.97328731661e-02 4.11390439155e-02 4.15214590889e-02 4.15474648474e-02 4.15214590889e-02 4.11390439154e-02 + 3.97328731661e-02 3.66852084288e-02 3.17358439118e-02 2.53972544857e-02 1.90346417875e-02 1.39668146374e-02 + 1.05558438840e-02 8.42000879273e-03 7.17868687882e-03 6.75457206648e-03 7.17868687882e-03 8.42000879273e-03 + 1.05558438840e-02 1.39668146374e-02 1.90346417875e-02 2.53972544857e-02 3.17358439118e-02 3.66852084288e-02 + 3.89797931477e-02 4.00844862497e-02 4.03066875222e-02 4.03066875222e-02 4.00844862497e-02 3.89797931477e-02 + 3.62670291685e-02 3.17358439118e-02 2.58521406843e-02 1.96803910193e-02 1.44211937670e-02 1.06519075585e-02 + 8.18384117181e-03 6.62684941428e-03 5.82750449515e-03 5.82750449515e-03 6.62684941427e-03 8.18384117181e-03 + 1.06519075585e-02 1.44211937670e-02 1.96803910193e-02 2.58521406843e-02 3.17358439118e-02 3.62670291685e-02 + 3.75326780270e-02 3.86301228830e-02 3.88820048627e-02 3.86301228830e-02 3.75326780270e-02 3.49765243617e-02 + 3.07818020073e-02 2.53972544857e-02 1.96803910193e-02 1.45988685534e-02 1.07548084489e-02 8.12162338295e-03 + 6.36712108007e-03 5.28637047081e-03 4.90592224711e-03 5.28637047081e-03 6.36712108007e-03 8.12162338294e-03 + 1.07548084489e-02 1.45988685534e-02 1.96803910193e-02 2.53972544857e-02 3.07818020073e-02 3.49765243617e-02 + 3.51265406940e-02 3.61943858423e-02 3.61943858423e-02 3.51265406940e-02 3.27027225515e-02 2.88774182856e-02 + 2.40902041017e-02 1.90346417875e-02 1.44211937670e-02 1.07548084489e-02 8.11876205862e-03 6.28716491126e-03 + 5.04613651753e-03 4.37602725379e-03 4.37602725379e-03 5.04613651753e-03 6.28716491126e-03 8.11876205861e-03 + 1.07548084489e-02 1.44211937670e-02 1.90346417875e-02 2.40902041017e-02 2.88774182856e-02 3.27027225515e-02 + 3.13659743854e-02 3.20076435874e-02 3.13659743854e-02 2.93978765598e-02 2.61899431515e-02 2.21707375456e-02 + 1.79266369346e-02 1.39668146374e-02 1.06519075585e-02 8.12162338295e-03 6.28716491126e-03 4.98512607635e-03 + 4.14922255853e-03 3.84969900803e-03 4.14922255853e-03 4.98512607635e-03 6.28716491126e-03 8.12162338295e-03 + 1.06519075585e-02 1.39668146374e-02 1.79266369346e-02 2.21707375456e-02 2.61899431515e-02 2.93978765598e-02 + 2.65859681267e-02 2.65859681267e-02 2.54054025138e-02 2.31229887194e-02 2.00581600358e-02 1.66949823920e-02 + 1.34421984781e-02 1.05558438840e-02 8.18384117181e-03 6.36712108007e-03 5.04613651754e-03 4.14922255853e-03 + 3.67775927312e-03 3.67775927312e-03 4.14922255853e-03 5.04613651753e-03 6.36712108007e-03 8.18384117181e-03 + 1.05558438840e-02 1.34421984781e-02 1.66949823920e-02 2.00581600358e-02 2.31229887194e-02 2.54054025138e-02 + 2.19609803409e-02 2.15194519516e-02 2.02039597509e-02 1.81429746336e-02 1.56490177739e-02 1.30619533075e-02 + 1.06052341643e-02 8.42000879273e-03 6.62684941428e-03 5.28637047082e-03 4.37602725379e-03 3.84969900803e-03 + 3.67775927312e-03 3.84969900803e-03 4.37602725379e-03 5.28637047081e-03 6.62684941427e-03 8.42000879273e-03 + 1.06052341643e-02 1.30619533075e-02 1.56490177739e-02 1.81429746336e-02 2.02039597509e-02 2.15194519516e-02 + 1.85131064743e-02 1.78887124933e-02 1.66389072281e-02 1.48932680012e-02 1.28938940154e-02 1.08501521193e-02 + 8.89928200876e-03 7.17868687882e-03 5.82750449516e-03 4.90592224711e-03 4.37602725379e-03 4.14922255853e-03 + 4.14922255853e-03 4.37602725379e-03 4.90592224711e-03 5.82750449515e-03 7.17868687881e-03 8.89928200875e-03 + 1.08501521193e-02 1.28938940154e-02 1.48932680012e-02 1.66389072281e-02 1.78887124933e-02 1.85131064743e-02 + 1.63905844408e-02 1.56224103699e-02 1.43895601245e-02 1.28581680779e-02 1.12138690779e-02 9.57630048387e-03 + 8.04465702446e-03 6.75457206648e-03 5.82750449515e-03 5.28637047081e-03 5.04613651753e-03 4.98512607635e-03 + 5.04613651753e-03 5.28637047081e-03 5.82750449515e-03 6.75457206647e-03 8.04465702446e-03 9.57630048387e-03 + 1.12138690779e-02 1.28581680779e-02 1.43895601245e-02 1.56224103699e-02 1.63905844408e-02 1.66438543464e-02 + 1.50739753351e-02 1.41150555464e-02 1.28938940154e-02 1.16030717026e-02 1.03423481784e-02 9.13735254995e-03 + 8.04465702446e-03 7.17868687882e-03 6.62684941428e-03 6.36712108007e-03 6.28716491126e-03 6.28716491126e-03 + 6.36712108007e-03 6.62684941427e-03 7.17868687881e-03 8.04465702446e-03 9.13735254995e-03 1.03423481784e-02 + 1.16030717026e-02 1.28938940154e-02 1.41150555464e-02 1.50739753351e-02 1.55937775675e-02 1.55937775675e-02 + 1.40807237278e-02 1.30619533075e-02 1.20440164119e-02 1.11481924117e-02 1.03423481784e-02 9.57630048387e-03 + 8.89928200876e-03 8.42000879273e-03 8.18384117181e-03 8.12162338295e-03 8.11876205861e-03 8.12162338295e-03 + 8.18384117181e-03 8.42000879273e-03 8.89928200875e-03 9.57630048387e-03 1.03423481784e-02 1.11481924117e-02 + 1.20440164119e-02 1.30619533075e-02 1.40807237278e-02 1.48493143122e-02 1.51356003261e-02 1.48493143122e-02 + 1.34421984781e-02 1.26512009150e-02 1.20440164119e-02 1.16030717026e-02 1.12138690779e-02 1.08501521193e-02 + 1.06052341643e-02 1.05558438840e-02 1.06519075585e-02 1.07548084489e-02 1.07548084489e-02 1.06519075585e-02 + 1.05558438840e-02 1.06052341643e-02 1.08501521193e-02 1.12138690779e-02 1.16030717026e-02 1.20440164119e-02 + 1.26512009150e-02 1.34421984781e-02 1.42309947893e-02 1.47280857174e-02 1.47280857174e-02 1.42309947893e-02 + 1.34421984781e-02 1.30619533075e-02 1.28938940154e-02 1.28581680779e-02 1.28938940154e-02 1.30619533075e-02 + 1.34421984781e-02 1.39668146374e-02 1.44211937670e-02 1.45988685534e-02 1.44211937670e-02 1.39668146374e-02 + 1.34421984781e-02 1.30619533075e-02 1.28938940154e-02 1.28581680779e-02 1.28938940154e-02 1.30619533075e-02 + 1.34421984781e-02 1.39668146374e-02 1.44211937670e-02 1.45988685534e-02 1.44211937670e-02 1.39668146374e-02 + 1.40807237278e-02 1.41150555464e-02 1.43895601245e-02 1.48932680012e-02 1.56490177739e-02 1.66949823920e-02 + 1.79266369346e-02 1.90346417875e-02 1.96803910193e-02 1.96803910193e-02 1.90346417875e-02 1.79266369346e-02 + 1.66949823920e-02 1.56490177739e-02 1.48932680012e-02 1.43895601245e-02 1.41150555464e-02 1.40807237278e-02 + 1.42309947893e-02 1.44211937670e-02 1.45323142804e-02 1.45323142804e-02 1.44211937670e-02 1.42309947893e-02 + 1.50739753351e-02 1.56224103699e-02 1.66389072281e-02 1.81429746336e-02 2.00581600358e-02 2.21707375456e-02 + 2.40902041017e-02 2.53972544857e-02 2.58521406843e-02 2.53972544857e-02 2.40902041017e-02 2.21707375455e-02 + 2.00581600358e-02 1.81429746336e-02 1.66389072281e-02 1.56224103699e-02 1.50739753351e-02 1.48493143122e-02 + 1.47280857174e-02 1.45988685534e-02 1.45323142804e-02 1.45988685534e-02 1.47280857174e-02 1.48493143123e-02 + 1.63905844408e-02 1.78887124933e-02 2.02039597509e-02 2.31229887194e-02 2.61899431515e-02 2.88774182856e-02 + 3.07818020073e-02 3.17358439118e-02 3.17358439118e-02 3.07818020073e-02 2.88774182856e-02 2.61899431515e-02 + 2.31229887194e-02 2.02039597509e-02 1.78887124933e-02 1.63905844408e-02 1.55937775675e-02 1.51356003261e-02 + 1.47280857174e-02 1.44211937670e-02 1.44211937670e-02 1.47280857174e-02 1.51356003261e-02 1.55937775675e-02 + 1.85131064743e-02 2.15194519516e-02 2.54054025138e-02 2.93978765598e-02 3.27027225515e-02 3.49765243617e-02 + 3.62670291685e-02 3.66852084288e-02 3.62670291685e-02 3.49765243617e-02 3.27027225515e-02 2.93978765598e-02 + 2.54054025138e-02 2.15194519516e-02 1.85131064743e-02 1.66438543464e-02 1.55937775675e-02 1.48493143122e-02 + 1.42309947893e-02 1.39668146374e-02 1.42309947893e-02 1.48493143123e-02 1.55937775675e-02 1.66438543464e-02 + 2.19946944680e-02 2.65859681267e-02 3.06730045455e-02 3.33648425996e-02 3.45751721123e-02 3.48629669819e-02 + 3.48685420831e-02 3.48629669819e-02 3.45751721122e-02 3.33648425996e-02 3.06730045455e-02 2.65859681267e-02 + 2.19946944680e-02 1.79740052840e-02 1.50423980149e-02 1.30486211596e-02 1.16096263393e-02 1.06052341643e-02 + 1.02249435892e-02 1.06052341643e-02 1.16096263393e-02 1.30486211596e-02 1.50423980149e-02 1.79740052840e-02 + 2.65859681267e-02 3.12137668085e-02 3.39946385756e-02 3.46097554135e-02 3.39197128099e-02 3.31552867935e-02 + 3.31552867935e-02 3.39197128099e-02 3.46097554135e-02 3.39946385756e-02 3.12137668085e-02 2.65859681267e-02 + 2.15037112467e-02 1.72810954144e-02 1.43188156133e-02 1.22935034476e-02 1.08501521193e-02 1.00211358334e-02 + 1.00211358334e-02 1.08501521193e-02 1.22935034476e-02 1.43188156133e-02 1.72810954144e-02 2.15037112467e-02 + 3.06730045455e-02 3.39946385756e-02 3.44218462469e-02 3.26919751552e-02 3.05271423180e-02 2.95747487953e-02 + 3.05271423180e-02 3.26919751551e-02 3.44218462468e-02 3.39946385756e-02 3.06730045455e-02 2.54054025138e-02 + 2.00652756289e-02 1.59286730225e-02 1.31235074234e-02 1.12138690779e-02 9.96244247361e-03 9.50257178747e-03 + 9.96244247361e-03 1.12138690779e-02 1.31235074234e-02 1.59286730224e-02 2.00652756288e-02 2.54054025138e-02 + 3.33648425996e-02 3.46097554135e-02 3.26919751552e-02 2.92886940372e-02 2.67211542468e-02 2.67211542468e-02 + 2.92886940372e-02 3.26919751551e-02 3.46097554135e-02 3.33648425996e-02 2.89589200656e-02 2.31229887194e-02 + 1.78656633258e-02 1.40901757966e-02 1.16030717026e-02 9.96178172025e-03 9.07040710948e-03 9.07040710948e-03 + 9.96178172025e-03 1.16030717026e-02 1.40901757966e-02 1.78656633258e-02 2.31229887194e-02 2.89589200656e-02 + 3.45751721123e-02 3.39197128099e-02 3.05271423180e-02 2.67211542468e-02 2.50655517889e-02 2.67211542468e-02 + 3.05271423180e-02 3.39197128099e-02 3.45751721122e-02 3.16055353689e-02 2.60490352762e-02 2.00581600358e-02 + 1.52718979502e-02 1.20440164119e-02 9.96244247361e-03 8.68885911334e-03 8.23794392377e-03 8.68885911335e-03 + 9.96244247361e-03 1.20440164119e-02 1.52718979502e-02 2.00581600358e-02 2.60490352762e-02 3.16055353689e-02 + 3.48629669819e-02 3.31552867935e-02 2.95747487953e-02 2.67211542468e-02 2.67211542468e-02 2.95747487953e-02 + 3.31552867935e-02 3.48629669819e-02 3.31962688978e-02 2.84215335265e-02 2.22711776717e-02 1.66949823920e-02 + 1.26512009150e-02 1.00211358334e-02 8.36367529241e-03 7.50459533107e-03 7.50459533107e-03 8.36367529241e-03 + 1.00211358334e-02 1.26512009150e-02 1.66949823920e-02 2.22711776717e-02 2.84215335265e-02 3.31962688978e-02 + 3.48685420831e-02 3.31552867935e-02 3.05271423180e-02 2.92886940371e-02 3.05271423180e-02 3.31552867934e-02 + 3.48685420831e-02 3.38926655662e-02 2.99300705585e-02 2.40966608398e-02 1.81617249718e-02 1.34421984781e-02 + 1.02249435892e-02 8.15763089847e-03 6.92808975974e-03 6.50344737139e-03 6.92808975974e-03 8.15763089847e-03 + 1.02249435892e-02 1.34421984781e-02 1.81617249718e-02 2.40966608398e-02 2.99300705585e-02 3.38926655663e-02 + 3.48629669819e-02 3.39197128099e-02 3.26919751551e-02 3.26919751551e-02 3.39197128099e-02 3.48629669819e-02 + 3.38926655663e-02 3.04405035973e-02 2.51250422144e-02 1.92841419762e-02 1.42309947893e-02 1.05558438840e-02 + 8.11231618988e-03 6.56139720409e-03 5.76671059723e-03 5.76671059723e-03 6.56139720408e-03 8.11231618988e-03 + 1.05558438840e-02 1.42309947893e-02 1.92841419762e-02 2.51250422144e-02 3.04405035973e-02 3.38926655663e-02 + 3.45751721122e-02 3.46097554135e-02 3.44218462468e-02 3.46097554135e-02 3.45751721122e-02 3.31962688978e-02 + 2.99300705585e-02 2.51250422144e-02 1.97046043739e-02 1.47280857174e-02 1.08700745212e-02 8.18384117181e-03 + 6.39966515662e-03 5.32069114083e-03 4.94634941966e-03 5.32069114083e-03 6.39966515661e-03 8.18384117180e-03 + 1.08700745212e-02 1.47280857174e-02 1.97046043739e-02 2.51250422144e-02 2.99300705585e-02 3.31962688978e-02 + 3.33648425996e-02 3.39946385756e-02 3.39946385756e-02 3.33648425996e-02 3.16055353689e-02 2.84215335265e-02 + 2.40966608398e-02 1.92841419762e-02 1.47280857174e-02 1.09991800130e-02 8.26411545329e-03 6.36712108007e-03 + 5.11302992590e-03 4.45369745770e-03 4.45369745770e-03 5.11302992589e-03 6.36712108006e-03 8.26411545328e-03 + 1.09991800129e-02 1.47280857174e-02 1.92841419762e-02 2.40966608398e-02 2.84215335265e-02 3.16055353689e-02 + 3.06730045455e-02 3.12137668085e-02 3.06730045455e-02 2.89589200657e-02 2.60490352762e-02 2.22711776717e-02 + 1.81617249718e-02 1.42309947893e-02 1.08700745212e-02 8.26411545329e-03 6.36994587767e-03 5.04613651754e-03 + 4.21587397244e-03 3.92286252630e-03 4.21587397243e-03 5.04613651753e-03 6.36994587766e-03 8.26411545328e-03 + 1.08700745212e-02 1.42309947893e-02 1.81617249718e-02 2.22711776717e-02 2.60490352762e-02 2.89589200657e-02 + 2.65859681267e-02 2.65859681267e-02 2.54054025138e-02 2.31229887194e-02 2.00581600358e-02 1.66949823920e-02 + 1.34421984781e-02 1.05558438840e-02 8.18384117181e-03 6.36712108007e-03 5.04613651754e-03 4.14922255853e-03 + 3.67775927312e-03 3.67775927312e-03 4.14922255853e-03 5.04613651753e-03 6.36712108007e-03 8.18384117181e-03 + 1.05558438840e-02 1.34421984781e-02 1.66949823920e-02 2.00581600358e-02 2.31229887194e-02 2.54054025138e-02 + 2.19946944681e-02 2.15037112468e-02 2.00652756289e-02 1.78656633258e-02 1.52718979502e-02 1.26512009150e-02 + 1.02249435892e-02 8.11231618988e-03 6.39966515662e-03 5.11302992590e-03 4.21587397244e-03 3.67775927312e-03 + 3.49735469022e-03 3.67775927312e-03 4.21587397243e-03 5.11302992589e-03 6.39966515661e-03 8.11231618988e-03 + 1.02249435892e-02 1.26512009150e-02 1.52718979502e-02 1.78656633258e-02 2.00652756289e-02 2.15037112468e-02 + 1.79740052840e-02 1.72810954144e-02 1.59286730225e-02 1.40901757966e-02 1.20440164119e-02 1.00211358334e-02 + 8.15763089847e-03 6.56139720409e-03 5.32069114083e-03 4.45369745770e-03 3.92286252630e-03 3.67775927312e-03 + 3.67775927312e-03 3.92286252630e-03 4.45369745770e-03 5.32069114083e-03 6.56139720408e-03 8.15763089847e-03 + 1.00211358334e-02 1.20440164119e-02 1.40901757966e-02 1.59286730225e-02 1.72810954144e-02 1.79740052840e-02 + 1.50423980149e-02 1.43188156133e-02 1.31235074234e-02 1.16030717026e-02 9.96244247360e-03 8.36367529241e-03 + 6.92808975974e-03 5.76671059724e-03 4.94634941966e-03 4.45369745770e-03 4.21587397243e-03 4.14922255853e-03 + 4.21587397243e-03 4.45369745770e-03 4.94634941966e-03 5.76671059723e-03 6.92808975974e-03 8.36367529240e-03 + 9.96244247360e-03 1.16030717026e-02 1.31235074234e-02 1.43188156133e-02 1.50423980149e-02 1.52761582425e-02 + 1.30486211596e-02 1.22935034476e-02 1.12138690779e-02 9.96178172025e-03 8.68885911334e-03 7.50459533107e-03 + 6.50344737139e-03 5.76671059724e-03 5.32069114083e-03 5.11302992589e-03 5.04613651753e-03 5.04613651753e-03 + 5.11302992589e-03 5.32069114083e-03 5.76671059723e-03 6.50344737139e-03 7.50459533107e-03 8.68885911334e-03 + 9.96178172025e-03 1.12138690779e-02 1.22935034476e-02 1.30486211596e-02 1.34169422858e-02 1.34169422858e-02 + 1.16096263393e-02 1.08501521193e-02 9.96244247360e-03 9.07040710948e-03 8.23794392377e-03 7.50459533107e-03 + 6.92808975974e-03 6.56139720408e-03 6.39966515661e-03 6.36712108007e-03 6.36994587767e-03 6.36712108007e-03 + 6.39966515661e-03 6.56139720408e-03 6.92808975974e-03 7.50459533107e-03 8.23794392377e-03 9.07040710948e-03 + 9.96244247360e-03 1.08501521193e-02 1.16096263393e-02 1.21091269494e-02 1.22801184032e-02 1.21091269494e-02 + 1.06052341643e-02 1.00211358334e-02 9.50257178747e-03 9.07040710948e-03 8.68885911335e-03 8.36367529241e-03 + 8.15763089847e-03 8.11231618988e-03 8.18384117181e-03 8.26411545328e-03 8.26411545329e-03 8.18384117181e-03 + 8.11231618988e-03 8.15763089847e-03 8.36367529241e-03 8.68885911334e-03 9.07040710948e-03 9.50257178747e-03 + 1.00211358334e-02 1.06052341643e-02 1.11332793734e-02 1.14482068790e-02 1.14482068790e-02 1.11332793734e-02 + 1.02249435892e-02 1.00211358334e-02 9.96244247361e-03 9.96178172025e-03 9.96244247361e-03 1.00211358334e-02 + 1.02249435892e-02 1.05558438840e-02 1.08700745212e-02 1.09991800129e-02 1.08700745212e-02 1.05558438840e-02 + 1.02249435892e-02 1.00211358334e-02 9.96244247360e-03 9.96178172025e-03 9.96244247360e-03 1.00211358334e-02 + 1.02249435892e-02 1.05558438840e-02 1.08700745212e-02 1.09991800129e-02 1.08700745212e-02 1.05558438840e-02 + 1.06052341643e-02 1.08501521193e-02 1.12138690779e-02 1.16030717026e-02 1.20440164119e-02 1.26512009150e-02 + 1.34421984781e-02 1.42309947893e-02 1.47280857174e-02 1.47280857174e-02 1.42309947893e-02 1.34421984781e-02 + 1.26512009150e-02 1.20440164119e-02 1.16030717026e-02 1.12138690779e-02 1.08501521193e-02 1.06052341643e-02 + 1.05558438840e-02 1.06519075585e-02 1.07548084489e-02 1.07548084489e-02 1.06519075585e-02 1.05558438840e-02 + 1.16096263393e-02 1.22935034476e-02 1.31235074234e-02 1.40901757966e-02 1.52718979502e-02 1.66949823920e-02 + 1.81617249718e-02 1.92841419762e-02 1.97046043739e-02 1.92841419762e-02 1.81617249718e-02 1.66949823920e-02 + 1.52718979502e-02 1.40901757966e-02 1.31235074234e-02 1.22935034476e-02 1.16096263393e-02 1.11332793734e-02 + 1.08700745212e-02 1.07548084489e-02 1.07235160838e-02 1.07548084489e-02 1.08700745212e-02 1.11332793734e-02 + 1.30486211596e-02 1.43188156133e-02 1.59286730224e-02 1.78656633258e-02 2.00581600358e-02 2.22711776717e-02 + 2.40966608398e-02 2.51250422144e-02 2.51250422144e-02 2.40966608398e-02 2.22711776717e-02 2.00581600358e-02 + 1.78656633258e-02 1.59286730225e-02 1.43188156133e-02 1.30486211596e-02 1.21091269495e-02 1.14482068790e-02 + 1.09991800129e-02 1.07548084489e-02 1.07548084489e-02 1.09991800130e-02 1.14482068791e-02 1.21091269495e-02 + 1.50423980149e-02 1.72810954144e-02 2.00652756288e-02 2.31229887194e-02 2.60490352762e-02 2.84215335265e-02 + 2.99300705585e-02 3.04405035973e-02 2.99300705585e-02 2.84215335265e-02 2.60490352762e-02 2.31229887194e-02 + 2.00652756288e-02 1.72810954144e-02 1.50423980149e-02 1.34169422858e-02 1.22801184033e-02 1.14482068790e-02 + 1.08700745212e-02 1.06519075585e-02 1.08700745212e-02 1.14482068791e-02 1.22801184033e-02 1.34169422858e-02 + 1.79740052840e-02 2.15037112467e-02 2.54054025138e-02 2.89589200656e-02 3.16055353689e-02 3.31962688978e-02 + 3.38926655663e-02 3.38926655663e-02 3.31962688978e-02 3.16055353689e-02 2.89589200656e-02 2.54054025138e-02 + 2.15037112467e-02 1.79740052840e-02 1.52761582425e-02 1.34169422858e-02 1.21091269495e-02 1.11332793734e-02 + 1.05558438840e-02 1.05558438840e-02 1.11332793734e-02 1.21091269495e-02 1.34169422858e-02 1.52761582425e-02 + 2.19609803409e-02 2.59180287237e-02 2.89472257765e-02 3.04771749334e-02 3.06791433435e-02 3.03790382414e-02 + 3.03790382414e-02 3.06791433435e-02 3.04771749334e-02 2.89472257765e-02 2.59180287237e-02 2.19609803409e-02 + 1.79740052840e-02 1.46288065643e-02 1.21045641480e-02 1.02402112845e-02 8.89928200876e-03 8.15763089847e-03 + 8.15763089848e-03 8.89928200876e-03 1.02402112845e-02 1.21045641480e-02 1.46288065643e-02 1.79740052840e-02 + 2.59180287237e-02 2.92642577664e-02 3.05546433144e-02 2.97972897261e-02 2.81921518485e-02 2.73937743101e-02 + 2.81921518485e-02 2.97972897261e-02 3.05546433144e-02 2.92642577664e-02 2.59180287237e-02 2.15194519516e-02 + 1.72810954144e-02 1.38961510916e-02 1.14032634249e-02 9.57630048387e-03 8.36367529241e-03 7.92503968744e-03 + 8.36367529241e-03 9.57630048387e-03 1.14032634249e-02 1.38961510916e-02 1.72810954144e-02 2.15194519516e-02 + 2.89472257765e-02 3.05546433144e-02 2.92434029241e-02 2.61163074370e-02 2.35389487571e-02 2.35389487571e-02 + 2.61163074370e-02 2.92434029240e-02 3.05546433144e-02 2.89472257765e-02 2.49901046905e-02 2.02039597509e-02 + 1.59286730225e-02 1.26855892424e-02 1.03423481784e-02 8.68885911334e-03 7.77919418380e-03 7.77919418380e-03 + 8.68885911335e-03 1.03423481784e-02 1.26855892423e-02 1.59286730225e-02 2.02039597509e-02 2.49901046905e-02 + 3.04771749334e-02 2.97972897261e-02 2.61163074370e-02 2.17745697719e-02 1.98434626957e-02 2.17745697719e-02 + 2.61163074370e-02 2.97972897261e-02 3.04771749334e-02 2.78012163806e-02 2.30937481115e-02 1.81429746336e-02 + 1.40901757966e-02 1.11481924118e-02 9.07040710948e-03 7.73184010469e-03 7.25307658557e-03 7.73184010469e-03 + 9.07040710948e-03 1.11481924117e-02 1.40901757966e-02 1.81429746336e-02 2.30937481115e-02 2.78012163806e-02 + 3.06791433435e-02 2.81921518485e-02 2.35389487571e-02 1.98434626957e-02 1.98434626957e-02 2.35389487571e-02 + 2.81921518485e-02 3.06791433435e-02 2.95889063006e-02 2.55724702285e-02 2.03920068264e-02 1.56490177739e-02 + 1.20440164119e-02 9.50257178747e-03 7.77919418380e-03 6.85673718666e-03 6.85673718666e-03 7.77919418380e-03 + 9.50257178747e-03 1.20440164119e-02 1.56490177739e-02 2.03920068264e-02 2.55724702285e-02 2.95889063006e-02 + 3.03790382414e-02 2.73937743101e-02 2.35389487571e-02 2.17745697719e-02 2.35389487571e-02 2.73937743101e-02 + 3.03790382413e-02 3.04051608500e-02 2.73091130234e-02 2.23752062804e-02 1.72612102705e-02 1.30619533075e-02 + 1.00211358334e-02 7.92503968744e-03 6.62092791971e-03 6.16390944840e-03 6.62092791971e-03 7.92503968744e-03 + 1.00211358334e-02 1.30619533075e-02 1.72612102705e-02 2.23752062804e-02 2.73091130234e-02 3.04051608500e-02 + 3.03790382414e-02 2.81921518485e-02 2.61163074370e-02 2.61163074370e-02 2.81921518485e-02 3.03790382413e-02 + 3.06060385055e-02 2.81654297894e-02 2.37302554132e-02 1.86355831047e-02 1.40807237278e-02 1.06052341643e-02 + 8.15763089847e-03 6.53733030469e-03 5.69415565947e-03 5.69415565947e-03 6.53733030468e-03 8.15763089847e-03 + 1.06052341643e-02 1.40807237278e-02 1.86355831047e-02 2.37302554132e-02 2.81654297894e-02 3.06060385055e-02 + 3.06791433435e-02 2.97972897260e-02 2.92434029240e-02 2.97972897261e-02 3.06791433435e-02 3.04051608500e-02 + 2.81654297894e-02 2.42076546579e-02 1.94291633463e-02 1.48493143123e-02 1.11332793734e-02 8.42000879274e-03 + 6.56139720409e-03 5.43241849957e-03 5.04346463959e-03 5.43241849957e-03 6.56139720408e-03 8.42000879273e-03 + 1.11332793734e-02 1.48493143122e-02 1.94291633463e-02 2.42076546579e-02 2.81654297894e-02 3.04051608500e-02 + 3.04771749334e-02 3.05546433144e-02 3.05546433144e-02 3.04771749334e-02 2.95889063006e-02 2.73091130234e-02 + 2.37302554132e-02 1.94291633463e-02 1.51356003261e-02 1.14482068791e-02 8.62544633520e-03 6.62684941428e-03 + 5.32069114083e-03 4.65178002992e-03 4.65178002992e-03 5.32069114083e-03 6.62684941427e-03 8.62544633519e-03 + 1.14482068790e-02 1.51356003261e-02 1.94291633463e-02 2.37302554132e-02 2.73091130234e-02 2.95889063006e-02 + 2.89472257765e-02 2.92642577664e-02 2.89472257765e-02 2.78012163806e-02 2.55724702285e-02 2.23752062804e-02 + 1.86355831047e-02 1.48493143123e-02 1.14482068791e-02 8.70260576991e-03 6.67440432829e-03 5.28637047082e-03 + 4.45369745770e-03 4.16949135529e-03 4.45369745770e-03 5.28637047081e-03 6.67440432828e-03 8.70260576990e-03 + 1.14482068790e-02 1.48493143123e-02 1.86355831048e-02 2.23752062804e-02 2.55724702285e-02 2.78012163806e-02 + 2.59180287237e-02 2.59180287237e-02 2.49901046905e-02 2.30937481116e-02 2.03920068264e-02 1.72612102705e-02 + 1.40807237278e-02 1.11332793734e-02 8.62544633520e-03 6.67440432829e-03 5.27979197806e-03 4.37602725380e-03 + 3.92286252630e-03 3.92286252630e-03 4.37602725379e-03 5.27979197805e-03 6.67440432828e-03 8.62544633519e-03 + 1.11332793734e-02 1.40807237278e-02 1.72612102705e-02 2.03920068264e-02 2.30937481115e-02 2.49901046905e-02 + 2.19609803409e-02 2.15194519516e-02 2.02039597509e-02 1.81429746336e-02 1.56490177739e-02 1.30619533075e-02 + 1.06052341643e-02 8.42000879274e-03 6.62684941428e-03 5.28637047082e-03 4.37602725380e-03 3.84969900804e-03 + 3.67775927312e-03 3.84969900803e-03 4.37602725379e-03 5.28637047081e-03 6.62684941427e-03 8.42000879273e-03 + 1.06052341643e-02 1.30619533075e-02 1.56490177739e-02 1.81429746336e-02 2.02039597509e-02 2.15194519516e-02 + 1.79740052840e-02 1.72810954144e-02 1.59286730225e-02 1.40901757966e-02 1.20440164119e-02 1.00211358334e-02 + 8.15763089847e-03 6.56139720409e-03 5.32069114083e-03 4.45369745770e-03 3.92286252630e-03 3.67775927312e-03 + 3.67775927312e-03 3.92286252630e-03 4.45369745770e-03 5.32069114083e-03 6.56139720408e-03 8.15763089847e-03 + 1.00211358334e-02 1.20440164119e-02 1.40901757966e-02 1.59286730225e-02 1.72810954144e-02 1.79740052840e-02 + 1.46288065643e-02 1.38961510916e-02 1.26855892424e-02 1.11481924117e-02 9.50257178747e-03 7.92503968744e-03 + 6.53733030469e-03 5.43241849957e-03 4.65178002992e-03 4.16949135529e-03 3.92286252630e-03 3.84969900803e-03 + 3.92286252630e-03 4.16949135529e-03 4.65178002992e-03 5.43241849957e-03 6.53733030468e-03 7.92503968743e-03 + 9.50257178747e-03 1.11481924117e-02 1.26855892423e-02 1.38961510916e-02 1.46288065643e-02 1.48655958108e-02 + 1.21045641480e-02 1.14032634249e-02 1.03423481784e-02 9.07040710948e-03 7.77919418380e-03 6.62092791971e-03 + 5.69415565947e-03 5.04346463959e-03 4.65178002992e-03 4.45369745770e-03 4.37602725379e-03 4.37602725379e-03 + 4.45369745770e-03 4.65178002992e-03 5.04346463958e-03 5.69415565947e-03 6.62092791971e-03 7.77919418380e-03 + 9.07040710948e-03 1.03423481784e-02 1.14032634249e-02 1.21045641480e-02 1.24266085814e-02 1.24266085814e-02 + 1.02402112845e-02 9.57630048387e-03 8.68885911334e-03 7.73184010469e-03 6.85673718666e-03 6.16390944840e-03 + 5.69415565947e-03 5.43241849957e-03 5.32069114083e-03 5.28637047081e-03 5.27979197805e-03 5.28637047081e-03 + 5.32069114083e-03 5.43241849957e-03 5.69415565947e-03 6.16390944840e-03 6.85673718666e-03 7.73184010469e-03 + 8.68885911334e-03 9.57630048386e-03 1.02402112845e-02 1.06110221336e-02 1.07226215343e-02 1.06110221336e-02 + 8.89928200875e-03 8.36367529240e-03 7.77919418380e-03 7.25307658557e-03 6.85673718666e-03 6.62092791971e-03 + 6.53733030468e-03 6.56139720408e-03 6.62684941427e-03 6.67440432828e-03 6.67440432828e-03 6.62684941427e-03 + 6.56139720408e-03 6.53733030468e-03 6.62092791971e-03 6.85673718666e-03 7.25307658557e-03 7.77919418380e-03 + 8.36367529240e-03 8.89928200875e-03 9.28365371521e-03 9.47298406612e-03 9.47298406612e-03 9.28365371522e-03 + 8.15763089847e-03 7.92503968744e-03 7.77919418380e-03 7.73184010469e-03 7.77919418380e-03 7.92503968744e-03 + 8.15763089847e-03 8.42000879273e-03 8.62544633519e-03 8.70260576990e-03 8.62544633519e-03 8.42000879273e-03 + 8.15763089847e-03 7.92503968743e-03 7.77919418380e-03 7.73184010469e-03 7.77919418380e-03 7.92503968744e-03 + 8.15763089847e-03 8.42000879273e-03 8.62544633519e-03 8.70260576990e-03 8.62544633520e-03 8.42000879273e-03 + 8.15763089847e-03 8.36367529241e-03 8.68885911335e-03 9.07040710948e-03 9.50257178747e-03 1.00211358334e-02 + 1.06052341643e-02 1.11332793734e-02 1.14482068790e-02 1.14482068790e-02 1.11332793734e-02 1.06052341643e-02 + 1.00211358334e-02 9.50257178747e-03 9.07040710948e-03 8.68885911334e-03 8.36367529241e-03 8.15763089847e-03 + 8.11231618988e-03 8.18384117181e-03 8.26411545329e-03 8.26411545329e-03 8.18384117181e-03 8.11231618988e-03 + 8.89928200876e-03 9.57630048387e-03 1.03423481784e-02 1.11481924117e-02 1.20440164119e-02 1.30619533075e-02 + 1.40807237278e-02 1.48493143122e-02 1.51356003261e-02 1.48493143122e-02 1.40807237278e-02 1.30619533075e-02 + 1.20440164119e-02 1.11481924117e-02 1.03423481784e-02 9.57630048387e-03 8.89928200876e-03 8.42000879274e-03 + 8.18384117181e-03 8.12162338295e-03 8.11876205861e-03 8.12162338296e-03 8.18384117182e-03 8.42000879274e-03 + 1.02402112845e-02 1.14032634249e-02 1.26855892423e-02 1.40901757966e-02 1.56490177739e-02 1.72612102704e-02 + 1.86355831047e-02 1.94291633463e-02 1.94291633463e-02 1.86355831047e-02 1.72612102704e-02 1.56490177739e-02 + 1.40901757966e-02 1.26855892423e-02 1.14032634249e-02 1.02402112845e-02 9.28365371522e-03 8.62544633520e-03 + 8.26411545329e-03 8.11876205861e-03 8.11876205862e-03 8.26411545329e-03 8.62544633520e-03 9.28365371522e-03 + 1.21045641480e-02 1.38961510916e-02 1.59286730224e-02 1.81429746336e-02 2.03920068264e-02 2.23752062804e-02 + 2.37302554132e-02 2.42076546579e-02 2.37302554132e-02 2.23752062804e-02 2.03920068263e-02 1.81429746336e-02 + 1.59286730224e-02 1.38961510916e-02 1.21045641480e-02 1.06110221336e-02 9.47298406612e-03 8.70260576991e-03 + 8.26411545329e-03 8.12162338295e-03 8.26411545329e-03 8.70260576991e-03 9.47298406613e-03 1.06110221336e-02 + 1.46288065643e-02 1.72810954144e-02 2.02039597509e-02 2.30937481115e-02 2.55724702285e-02 2.73091130234e-02 + 2.81654297894e-02 2.81654297894e-02 2.73091130234e-02 2.55724702285e-02 2.30937481115e-02 2.02039597509e-02 + 1.72810954144e-02 1.46288065643e-02 1.24266085814e-02 1.07226215343e-02 9.47298406613e-03 8.62544633520e-03 + 8.18384117181e-03 8.18384117181e-03 8.62544633520e-03 9.47298406613e-03 1.07226215343e-02 1.24266085814e-02 + 1.79740052840e-02 2.15194519516e-02 2.49901046905e-02 2.78012163806e-02 2.95889063006e-02 3.04051608500e-02 + 3.06060385055e-02 3.04051608500e-02 2.95889063006e-02 2.78012163806e-02 2.49901046905e-02 2.15194519516e-02 + 1.79740052840e-02 1.48655958108e-02 1.24266085814e-02 1.06110221336e-02 9.28365371522e-03 8.42000879274e-03 + 8.11231618988e-03 8.42000879274e-03 9.28365371522e-03 1.06110221336e-02 1.24266085814e-02 1.48655958108e-02 + 2.21383854932e-02 2.52442663181e-02 2.71139620581e-02 2.74766558578e-02 2.69079511827e-02 2.65223600849e-02 + 2.69079511827e-02 2.74766558578e-02 2.71139620581e-02 2.52442663181e-02 2.21383854932e-02 1.85131064743e-02 + 1.50423980149e-02 1.21045641480e-02 9.78230850882e-03 8.04465702446e-03 6.92808975974e-03 6.53733030468e-03 + 6.92808975974e-03 8.04465702446e-03 9.78230850882e-03 1.21045641480e-02 1.50423980149e-02 1.85131064743e-02 + 2.52442663181e-02 2.72047829307e-02 2.70509290066e-02 2.52952680324e-02 2.36140048007e-02 2.36140048007e-02 + 2.52952680324e-02 2.70509290066e-02 2.72047829307e-02 2.52442663181e-02 2.17946809311e-02 1.78887124933e-02 + 1.43188156133e-02 1.14032634249e-02 9.13735254995e-03 7.50459533106e-03 6.62092791971e-03 6.62092791971e-03 + 7.50459533107e-03 9.13735254995e-03 1.14032634249e-02 1.43188156133e-02 1.78887124933e-02 2.17946809311e-02 + 2.71139620581e-02 2.70509290066e-02 2.44576079909e-02 2.10845164455e-02 1.95330017723e-02 2.10845164455e-02 + 2.44576079909e-02 2.70509290066e-02 2.71139620581e-02 2.46284218885e-02 2.07195371247e-02 1.66389072281e-02 + 1.31235074234e-02 1.03423481784e-02 8.23794392376e-03 6.85673718666e-03 6.36560105343e-03 6.85673718666e-03 + 8.23794392377e-03 1.03423481784e-02 1.31235074234e-02 1.66389072281e-02 2.07195371247e-02 2.46284218885e-02 + 2.74766558578e-02 2.52952680324e-02 2.10845164455e-02 1.76765157693e-02 1.76765157693e-02 2.10845164455e-02 + 2.52952680324e-02 2.74766558578e-02 2.65397992490e-02 2.32028796117e-02 1.89263306496e-02 1.48932680012e-02 + 1.16030717026e-02 9.07040710948e-03 7.25307658557e-03 6.26635360836e-03 6.26635360836e-03 7.25307658557e-03 + 9.07040710948e-03 1.16030717026e-02 1.48932680012e-02 1.89263306496e-02 2.32028796117e-02 2.65397992490e-02 + 2.69079511827e-02 2.36140048007e-02 1.95330017723e-02 1.76765157693e-02 1.95330017723e-02 2.36140048007e-02 + 2.69079511827e-02 2.73901656280e-02 2.49962561202e-02 2.09446150107e-02 1.66276495780e-02 1.28938940154e-02 + 9.96244247361e-03 7.77919418380e-03 6.36560105343e-03 5.86554131393e-03 6.36560105343e-03 7.77919418380e-03 + 9.96244247361e-03 1.28938940154e-02 1.66276495780e-02 2.09446150107e-02 2.49962561202e-02 2.73901656280e-02 + 2.65223600849e-02 2.36140048007e-02 2.10845164455e-02 2.10845164455e-02 2.36140048007e-02 2.65223600849e-02 + 2.75583653887e-02 2.59617964748e-02 2.24043544384e-02 1.81179478705e-02 1.41150555464e-02 1.08501521193e-02 + 8.36367529241e-03 6.62092791971e-03 5.69332826896e-03 5.69332826896e-03 6.62092791971e-03 8.36367529241e-03 + 1.08501521193e-02 1.41150555464e-02 1.81179478705e-02 2.24043544384e-02 2.59617964748e-02 2.75583653887e-02 + 2.69079511827e-02 2.52952680324e-02 2.44576079909e-02 2.52952680324e-02 2.69079511827e-02 2.75583653887e-02 + 2.62414574798e-02 2.31387782150e-02 1.91106183945e-02 1.50739753351e-02 1.16096263393e-02 8.89928200876e-03 + 6.92808975974e-03 5.69415565947e-03 5.26655497606e-03 5.69415565947e-03 6.92808975974e-03 8.89928200875e-03 + 1.16096263393e-02 1.50739753351e-02 1.91106183945e-02 2.31387782150e-02 2.62414574798e-02 2.75583653887e-02 + 2.74766558578e-02 2.70509290066e-02 2.70509290066e-02 2.74766558578e-02 2.73901656281e-02 2.59617964748e-02 + 2.31387782150e-02 1.94536912926e-02 1.55937775675e-02 1.21091269495e-02 9.28365371522e-03 7.17868687882e-03 + 5.76671059723e-03 5.04346463959e-03 5.04346463959e-03 5.76671059723e-03 7.17868687881e-03 9.28365371522e-03 + 1.21091269494e-02 1.55937775675e-02 1.94536912925e-02 2.31387782150e-02 2.59617964748e-02 2.73901656280e-02 + 2.71139620581e-02 2.72047829307e-02 2.71139620581e-02 2.65397992490e-02 2.49962561202e-02 2.24043544384e-02 + 1.91106183945e-02 1.55937775675e-02 1.22801184033e-02 9.47298406613e-03 7.31939690082e-03 5.82750449515e-03 + 4.94634941966e-03 4.65178002993e-03 4.94634941966e-03 5.82750449515e-03 7.31939690081e-03 9.47298406612e-03 + 1.22801184032e-02 1.55937775675e-02 1.91106183945e-02 2.24043544384e-02 2.49962561202e-02 2.65397992490e-02 + 2.52442663181e-02 2.52442663181e-02 2.46284218885e-02 2.32028796117e-02 2.09446150107e-02 1.81179478705e-02 + 1.50739753351e-02 1.21091269495e-02 9.47298406613e-03 7.36059222258e-03 5.85081359877e-03 4.90592224711e-03 + 4.45369745771e-03 4.45369745771e-03 4.90592224711e-03 5.85081359877e-03 7.36059222257e-03 9.47298406612e-03 + 1.21091269494e-02 1.50739753351e-02 1.81179478705e-02 2.09446150107e-02 2.32028796117e-02 2.46284218885e-02 + 2.21383854932e-02 2.17946809311e-02 2.07195371247e-02 1.89263306496e-02 1.66276495780e-02 1.41150555464e-02 + 1.16096263393e-02 9.28365371522e-03 7.31939690082e-03 5.85081359877e-03 4.89322201917e-03 4.37602725380e-03 + 4.21587397244e-03 4.37602725379e-03 4.89322201917e-03 5.85081359876e-03 7.31939690081e-03 9.28365371521e-03 + 1.16096263393e-02 1.41150555464e-02 1.66276495780e-02 1.89263306496e-02 2.07195371247e-02 2.17946809311e-02 + 1.85131064743e-02 1.78887124933e-02 1.66389072281e-02 1.48932680012e-02 1.28938940154e-02 1.08501521193e-02 + 8.89928200876e-03 7.17868687882e-03 5.82750449515e-03 4.90592224711e-03 4.37602725380e-03 4.14922255853e-03 + 4.14922255853e-03 4.37602725379e-03 4.90592224711e-03 5.82750449515e-03 7.17868687881e-03 8.89928200875e-03 + 1.08501521193e-02 1.28938940154e-02 1.48932680012e-02 1.66389072281e-02 1.78887124933e-02 1.85131064743e-02 + 1.50423980149e-02 1.43188156133e-02 1.31235074234e-02 1.16030717026e-02 9.96244247361e-03 8.36367529241e-03 + 6.92808975974e-03 5.76671059723e-03 4.94634941966e-03 4.45369745770e-03 4.21587397244e-03 4.14922255853e-03 + 4.21587397243e-03 4.45369745770e-03 4.94634941966e-03 5.76671059723e-03 6.92808975973e-03 8.36367529240e-03 + 9.96244247360e-03 1.16030717026e-02 1.31235074234e-02 1.43188156133e-02 1.50423980149e-02 1.52761582425e-02 + 1.21045641480e-02 1.14032634249e-02 1.03423481784e-02 9.07040710948e-03 7.77919418380e-03 6.62092791971e-03 + 5.69415565947e-03 5.04346463959e-03 4.65178002992e-03 4.45369745770e-03 4.37602725379e-03 4.37602725379e-03 + 4.45369745770e-03 4.65178002992e-03 5.04346463958e-03 5.69415565947e-03 6.62092791971e-03 7.77919418380e-03 + 9.07040710948e-03 1.03423481784e-02 1.14032634249e-02 1.21045641480e-02 1.24266085814e-02 1.24266085814e-02 + 9.78230850881e-03 9.13735254995e-03 8.23794392376e-03 7.25307658557e-03 6.36560105343e-03 5.69332826896e-03 + 5.26655497606e-03 5.04346463959e-03 4.94634941966e-03 4.90592224711e-03 4.89322201917e-03 4.90592224711e-03 + 4.94634941966e-03 5.04346463958e-03 5.26655497605e-03 5.69332826895e-03 6.36560105343e-03 7.25307658557e-03 + 8.23794392376e-03 9.13735254994e-03 9.78230850881e-03 1.01199573737e-02 1.02154655471e-02 1.01199573737e-02 + 8.04465702445e-03 7.50459533106e-03 6.85673718666e-03 6.26635360836e-03 5.86554131393e-03 5.69332826896e-03 + 5.69415565947e-03 5.76671059723e-03 5.82750449515e-03 5.85081359876e-03 5.85081359876e-03 5.82750449515e-03 + 5.76671059723e-03 5.69415565947e-03 5.69332826895e-03 5.86554131392e-03 6.26635360836e-03 6.85673718666e-03 + 7.50459533106e-03 8.04465702445e-03 8.37411495641e-03 8.50676845051e-03 8.50676845051e-03 8.37411495641e-03 + 6.92808975974e-03 6.62092791971e-03 6.36560105343e-03 6.26635360836e-03 6.36560105343e-03 6.62092791971e-03 + 6.92808975974e-03 7.17868687881e-03 7.31939690081e-03 7.36059222257e-03 7.31939690081e-03 7.17868687881e-03 + 6.92808975973e-03 6.62092791970e-03 6.36560105343e-03 6.26635360836e-03 6.36560105343e-03 6.62092791971e-03 + 6.92808975974e-03 7.17868687881e-03 7.31939690081e-03 7.36059222257e-03 7.31939690081e-03 7.17868687881e-03 + 6.53733030468e-03 6.62092791971e-03 6.85673718666e-03 7.25307658557e-03 7.77919418380e-03 8.36367529240e-03 + 8.89928200875e-03 9.28365371521e-03 9.47298406612e-03 9.47298406612e-03 9.28365371521e-03 8.89928200875e-03 + 8.36367529240e-03 7.77919418380e-03 7.25307658557e-03 6.85673718666e-03 6.62092791971e-03 6.53733030468e-03 + 6.56139720408e-03 6.62684941428e-03 6.67440432828e-03 6.67440432828e-03 6.62684941428e-03 6.56139720409e-03 + 6.92808975974e-03 7.50459533107e-03 8.23794392377e-03 9.07040710948e-03 9.96244247361e-03 1.08501521193e-02 + 1.16096263393e-02 1.21091269494e-02 1.22801184032e-02 1.21091269494e-02 1.16096263393e-02 1.08501521193e-02 + 9.96244247360e-03 9.07040710948e-03 8.23794392376e-03 7.50459533106e-03 6.92808975974e-03 6.56139720409e-03 + 6.39966515662e-03 6.36712108007e-03 6.36994587767e-03 6.36712108007e-03 6.39966515662e-03 6.56139720409e-03 + 8.04465702446e-03 9.13735254995e-03 1.03423481784e-02 1.16030717026e-02 1.28938940154e-02 1.41150555464e-02 + 1.50739753351e-02 1.55937775675e-02 1.55937775675e-02 1.50739753351e-02 1.41150555464e-02 1.28938940154e-02 + 1.16030717026e-02 1.03423481784e-02 9.13735254995e-03 8.04465702445e-03 7.17868687881e-03 6.62684941428e-03 + 6.36712108007e-03 6.28716491126e-03 6.28716491127e-03 6.36712108007e-03 6.62684941428e-03 7.17868687882e-03 + 9.78230850881e-03 1.14032634249e-02 1.31235074234e-02 1.48932680012e-02 1.66276495780e-02 1.81179478705e-02 + 1.91106183945e-02 1.94536912925e-02 1.91106183945e-02 1.81179478705e-02 1.66276495780e-02 1.48932680012e-02 + 1.31235074234e-02 1.14032634249e-02 9.78230850881e-03 8.37411495641e-03 7.31939690082e-03 6.67440432828e-03 + 6.36994587767e-03 6.28716491126e-03 6.36994587767e-03 6.67440432829e-03 7.31939690082e-03 8.37411495641e-03 + 1.21045641480e-02 1.43188156133e-02 1.66389072281e-02 1.89263306496e-02 2.09446150107e-02 2.24043544384e-02 + 2.31387782150e-02 2.31387782150e-02 2.24043544384e-02 2.09446150107e-02 1.89263306496e-02 1.66389072280e-02 + 1.43188156133e-02 1.21045641480e-02 1.01199573737e-02 8.50676845052e-03 7.36059222257e-03 6.67440432828e-03 + 6.36712108007e-03 6.36712108007e-03 6.67440432828e-03 7.36059222257e-03 8.50676845052e-03 1.01199573737e-02 + 1.50423980149e-02 1.78887124933e-02 2.07195371247e-02 2.32028796117e-02 2.49962561202e-02 2.59617964748e-02 + 2.62414574798e-02 2.59617964748e-02 2.49962561202e-02 2.32028796117e-02 2.07195371247e-02 1.78887124933e-02 + 1.50423980149e-02 1.24266085814e-02 1.02154655471e-02 8.50676845052e-03 7.31939690082e-03 6.62684941428e-03 + 6.39966515662e-03 6.62684941428e-03 7.31939690082e-03 8.50676845052e-03 1.02154655471e-02 1.24266085814e-02 + 1.85131064743e-02 2.17946809311e-02 2.46284218885e-02 2.65397992490e-02 2.73901656281e-02 2.75583653887e-02 + 2.75583653887e-02 2.73901656280e-02 2.65397992490e-02 2.46284218885e-02 2.17946809311e-02 1.85131064742e-02 + 1.52761582425e-02 1.24266085814e-02 1.01199573737e-02 8.37411495641e-03 7.17868687882e-03 6.56139720409e-03 + 6.56139720409e-03 7.17868687882e-03 8.37411495641e-03 1.01199573737e-02 1.24266085814e-02 1.52761582425e-02 + 2.33116271286e-02 2.55789017267e-02 2.62349275771e-02 2.54923302763e-02 2.45136072590e-02 2.45136072590e-02 + 2.54923302762e-02 2.62349275771e-02 2.55789017267e-02 2.33116271285e-02 1.99974571615e-02 1.63905844408e-02 + 1.30486211596e-02 1.02402112845e-02 8.04465702446e-03 6.50344737139e-03 5.69415565947e-03 5.69415565947e-03 + 6.50344737139e-03 8.04465702446e-03 1.02402112845e-02 1.30486211596e-02 1.63905844408e-02 1.99974571616e-02 + 2.55789017267e-02 2.61137483222e-02 2.45136072590e-02 2.20784329209e-02 2.09113970026e-02 2.20784329209e-02 + 2.45136072590e-02 2.61137483222e-02 2.55789017267e-02 2.30251854758e-02 1.93870166210e-02 1.56224103699e-02 + 1.22935034476e-02 9.57630048387e-03 7.50459533106e-03 6.16390944840e-03 5.69332826896e-03 6.16390944840e-03 + 7.50459533107e-03 9.57630048387e-03 1.22935034476e-02 1.56224103699e-02 1.93870166210e-02 2.30251854758e-02 + 2.62349275771e-02 2.45136072590e-02 2.09113970026e-02 1.79303755335e-02 1.79303755335e-02 2.09113970026e-02 + 2.45136072590e-02 2.62349275771e-02 2.51928884334e-02 2.20776282425e-02 1.81502745082e-02 1.43895601245e-02 + 1.12138690779e-02 8.68885911334e-03 6.85673718666e-03 5.86554131392e-03 5.86554131393e-03 6.85673718666e-03 + 8.68885911335e-03 1.12138690779e-02 1.43895601245e-02 1.81502745082e-02 2.20776282425e-02 2.51928884334e-02 + 2.54923302763e-02 2.20784329209e-02 1.79303755335e-02 1.60511803073e-02 1.79303755335e-02 2.20784329209e-02 + 2.54923302763e-02 2.61753235783e-02 2.40837535674e-02 2.03982633679e-02 1.64062970608e-02 1.28581680779e-02 + 9.96178172026e-03 7.73184010469e-03 6.26635360836e-03 5.74654712650e-03 6.26635360836e-03 7.73184010469e-03 + 9.96178172026e-03 1.28581680779e-02 1.64062970608e-02 2.03982633679e-02 2.40837535674e-02 2.61753235783e-02 + 2.45136072590e-02 2.09113970026e-02 1.79303755335e-02 1.79303755335e-02 2.09113970026e-02 2.45136072590e-02 + 2.62349275771e-02 2.51928884334e-02 2.20776282425e-02 1.81502745082e-02 1.43895601245e-02 1.12138690779e-02 + 8.68885911335e-03 6.85673718666e-03 5.86554131393e-03 5.86554131392e-03 6.85673718666e-03 8.68885911334e-03 + 1.12138690779e-02 1.43895601245e-02 1.81502745082e-02 2.20776282425e-02 2.51928884334e-02 2.62349275771e-02 + 2.45136072590e-02 2.20784329209e-02 2.09113970026e-02 2.20784329209e-02 2.45136072590e-02 2.61137483222e-02 + 2.55789017267e-02 2.30251854758e-02 1.93870166210e-02 1.56224103699e-02 1.22935034476e-02 9.57630048387e-03 + 7.50459533107e-03 6.16390944840e-03 5.69332826896e-03 6.16390944840e-03 7.50459533106e-03 9.57630048387e-03 + 1.22935034476e-02 1.56224103699e-02 1.93870166210e-02 2.30251854758e-02 2.55789017267e-02 2.61137483222e-02 + 2.54923302762e-02 2.45136072590e-02 2.45136072590e-02 2.54923302763e-02 2.62349275771e-02 2.55789017267e-02 + 2.33116271286e-02 1.99974571616e-02 1.63905844408e-02 1.30486211596e-02 1.02402112845e-02 8.04465702446e-03 + 6.50344737139e-03 5.69415565947e-03 5.69415565947e-03 6.50344737139e-03 8.04465702446e-03 1.02402112845e-02 + 1.30486211596e-02 1.63905844408e-02 1.99974571615e-02 2.33116271285e-02 2.55789017267e-02 2.62349275771e-02 + 2.62349275771e-02 2.61137483222e-02 2.62349275771e-02 2.61753235783e-02 2.51928884334e-02 2.30251854758e-02 + 1.99974571616e-02 1.66438543464e-02 1.34169422858e-02 1.06110221336e-02 8.37411495641e-03 6.75457206648e-03 + 5.76671059723e-03 5.43241849957e-03 5.76671059723e-03 6.75457206648e-03 8.37411495641e-03 1.06110221336e-02 + 1.34169422858e-02 1.66438543464e-02 1.99974571615e-02 2.30251854758e-02 2.51928884334e-02 2.61753235783e-02 + 2.55789017267e-02 2.55789017267e-02 2.51928884334e-02 2.40837535674e-02 2.20776282425e-02 1.93870166210e-02 + 1.63905844408e-02 1.34169422858e-02 1.07226215343e-02 8.50676845052e-03 6.87435772716e-03 5.82750449515e-03 + 5.32069114083e-03 5.32069114083e-03 5.82750449515e-03 6.87435772715e-03 8.50676845052e-03 1.07226215343e-02 + 1.34169422858e-02 1.63905844408e-02 1.93870166210e-02 2.20776282425e-02 2.40837535673e-02 2.51928884334e-02 + 2.33116271285e-02 2.30251854758e-02 2.20776282425e-02 2.03982633679e-02 1.81502745082e-02 1.56224103699e-02 + 1.30486211596e-02 1.06110221336e-02 8.50676845052e-03 6.90318266869e-03 5.85081359877e-03 5.28637047082e-03 + 5.11302992590e-03 5.28637047082e-03 5.85081359877e-03 6.90318266868e-03 8.50676845051e-03 1.06110221336e-02 + 1.30486211596e-02 1.56224103699e-02 1.81502745082e-02 2.03982633679e-02 2.20776282425e-02 2.30251854758e-02 + 1.99974571616e-02 1.93870166210e-02 1.81502745082e-02 1.64062970608e-02 1.43895601245e-02 1.22935034476e-02 + 1.02402112845e-02 8.37411495641e-03 6.87435772715e-03 5.85081359877e-03 5.27979197806e-03 5.04613651754e-03 + 5.04613651754e-03 5.27979197806e-03 5.85081359877e-03 6.87435772715e-03 8.37411495641e-03 1.02402112845e-02 + 1.22935034476e-02 1.43895601245e-02 1.64062970608e-02 1.81502745082e-02 1.93870166210e-02 1.99974571616e-02 + 1.63905844408e-02 1.56224103699e-02 1.43895601245e-02 1.28581680779e-02 1.12138690779e-02 9.57630048387e-03 + 8.04465702446e-03 6.75457206647e-03 5.82750449515e-03 5.28637047082e-03 5.04613651754e-03 4.98512607635e-03 + 5.04613651753e-03 5.28637047081e-03 5.82750449515e-03 6.75457206647e-03 8.04465702445e-03 9.57630048386e-03 + 1.12138690779e-02 1.28581680779e-02 1.43895601245e-02 1.56224103699e-02 1.63905844408e-02 1.66438543464e-02 + 1.30486211596e-02 1.22935034476e-02 1.12138690779e-02 9.96178172025e-03 8.68885911335e-03 7.50459533107e-03 + 6.50344737139e-03 5.76671059723e-03 5.32069114083e-03 5.11302992590e-03 5.04613651753e-03 5.04613651753e-03 + 5.11302992589e-03 5.32069114083e-03 5.76671059723e-03 6.50344737138e-03 7.50459533106e-03 8.68885911334e-03 + 9.96178172025e-03 1.12138690779e-02 1.22935034476e-02 1.30486211596e-02 1.34169422858e-02 1.34169422858e-02 + 1.02402112845e-02 9.57630048387e-03 8.68885911334e-03 7.73184010469e-03 6.85673718666e-03 6.16390944840e-03 + 5.69415565947e-03 5.43241849957e-03 5.32069114083e-03 5.28637047081e-03 5.27979197805e-03 5.28637047081e-03 + 5.32069114083e-03 5.43241849957e-03 5.69415565947e-03 6.16390944840e-03 6.85673718666e-03 7.73184010469e-03 + 8.68885911334e-03 9.57630048386e-03 1.02402112845e-02 1.06110221336e-02 1.07226215343e-02 1.06110221336e-02 + 8.04465702445e-03 7.50459533106e-03 6.85673718666e-03 6.26635360836e-03 5.86554131393e-03 5.69332826896e-03 + 5.69415565947e-03 5.76671059723e-03 5.82750449515e-03 5.85081359876e-03 5.85081359876e-03 5.82750449515e-03 + 5.76671059723e-03 5.69415565947e-03 5.69332826895e-03 5.86554131392e-03 6.26635360836e-03 6.85673718666e-03 + 7.50459533106e-03 8.04465702445e-03 8.37411495641e-03 8.50676845051e-03 8.50676845051e-03 8.37411495641e-03 + 6.50344737139e-03 6.16390944840e-03 5.86554131393e-03 5.74654712650e-03 5.86554131393e-03 6.16390944840e-03 + 6.50344737139e-03 6.75457206647e-03 6.87435772715e-03 6.90318266868e-03 6.87435772715e-03 6.75457206647e-03 + 6.50344737138e-03 6.16390944840e-03 5.86554131392e-03 5.74654712650e-03 5.86554131392e-03 6.16390944840e-03 + 6.50344737138e-03 6.75457206647e-03 6.87435772715e-03 6.90318266868e-03 6.87435772715e-03 6.75457206647e-03 + 5.69415565947e-03 5.69332826896e-03 5.86554131393e-03 6.26635360836e-03 6.85673718666e-03 7.50459533106e-03 + 8.04465702445e-03 8.37411495641e-03 8.50676845051e-03 8.50676845051e-03 8.37411495641e-03 8.04465702445e-03 + 7.50459533106e-03 6.85673718666e-03 6.26635360836e-03 5.86554131392e-03 5.69332826895e-03 5.69415565947e-03 + 5.76671059723e-03 5.82750449515e-03 5.85081359876e-03 5.85081359876e-03 5.82750449515e-03 5.76671059723e-03 + 5.69415565947e-03 6.16390944840e-03 6.85673718666e-03 7.73184010469e-03 8.68885911334e-03 9.57630048387e-03 + 1.02402112845e-02 1.06110221336e-02 1.07226215343e-02 1.06110221336e-02 1.02402112845e-02 9.57630048386e-03 + 8.68885911334e-03 7.73184010469e-03 6.85673718666e-03 6.16390944840e-03 5.69415565947e-03 5.43241849957e-03 + 5.32069114083e-03 5.28637047081e-03 5.27979197805e-03 5.28637047081e-03 5.32069114083e-03 5.43241849957e-03 + 6.50344737139e-03 7.50459533107e-03 8.68885911335e-03 9.96178172025e-03 1.12138690779e-02 1.22935034476e-02 + 1.30486211596e-02 1.34169422858e-02 1.34169422858e-02 1.30486211596e-02 1.22935034476e-02 1.12138690779e-02 + 9.96178172025e-03 8.68885911334e-03 7.50459533106e-03 6.50344737138e-03 5.76671059723e-03 5.32069114083e-03 + 5.11302992589e-03 5.04613651753e-03 5.04613651753e-03 5.11302992590e-03 5.32069114083e-03 5.76671059723e-03 + 8.04465702446e-03 9.57630048387e-03 1.12138690779e-02 1.28581680779e-02 1.43895601245e-02 1.56224103699e-02 + 1.63905844408e-02 1.66438543464e-02 1.63905844408e-02 1.56224103699e-02 1.43895601245e-02 1.28581680779e-02 + 1.12138690779e-02 9.57630048386e-03 8.04465702445e-03 6.75457206647e-03 5.82750449515e-03 5.28637047081e-03 + 5.04613651753e-03 4.98512607635e-03 5.04613651754e-03 5.28637047082e-03 5.82750449515e-03 6.75457206647e-03 + 1.02402112845e-02 1.22935034476e-02 1.43895601245e-02 1.64062970608e-02 1.81502745082e-02 1.93870166210e-02 + 1.99974571616e-02 1.99974571616e-02 1.93870166210e-02 1.81502745082e-02 1.64062970608e-02 1.43895601245e-02 + 1.22935034476e-02 1.02402112845e-02 8.37411495641e-03 6.87435772715e-03 5.85081359877e-03 5.27979197806e-03 + 5.04613651754e-03 5.04613651754e-03 5.27979197806e-03 5.85081359877e-03 6.87435772715e-03 8.37411495641e-03 + 1.30486211596e-02 1.56224103699e-02 1.81502745082e-02 2.03982633679e-02 2.20776282425e-02 2.30251854758e-02 + 2.33116271285e-02 2.30251854758e-02 2.20776282425e-02 2.03982633679e-02 1.81502745082e-02 1.56224103699e-02 + 1.30486211596e-02 1.06110221336e-02 8.50676845051e-03 6.90318266868e-03 5.85081359877e-03 5.28637047082e-03 + 5.11302992590e-03 5.28637047082e-03 5.85081359877e-03 6.90318266869e-03 8.50676845052e-03 1.06110221336e-02 + 1.63905844408e-02 1.93870166210e-02 2.20776282425e-02 2.40837535674e-02 2.51928884334e-02 2.55789017267e-02 + 2.55789017267e-02 2.51928884334e-02 2.40837535673e-02 2.20776282425e-02 1.93870166210e-02 1.63905844408e-02 + 1.34169422858e-02 1.07226215343e-02 8.50676845052e-03 6.87435772715e-03 5.82750449515e-03 5.32069114083e-03 + 5.32069114083e-03 5.82750449515e-03 6.87435772716e-03 8.50676845052e-03 1.07226215343e-02 1.34169422858e-02 + 1.99974571616e-02 2.30251854758e-02 2.51928884334e-02 2.61753235783e-02 2.62349275771e-02 2.61137483222e-02 + 2.62349275771e-02 2.61753235783e-02 2.51928884334e-02 2.30251854758e-02 1.99974571615e-02 1.66438543464e-02 + 1.34169422858e-02 1.06110221336e-02 8.37411495641e-03 6.75457206648e-03 5.76671059723e-03 5.43241849957e-03 + 5.76671059723e-03 6.75457206648e-03 8.37411495641e-03 1.06110221336e-02 1.34169422858e-02 1.66438543464e-02 + 2.62414574798e-02 2.75583653887e-02 2.69079511827e-02 2.52952680324e-02 2.44576079909e-02 2.52952680324e-02 + 2.69079511827e-02 2.75583653887e-02 2.62414574798e-02 2.31387782150e-02 1.91106183945e-02 1.50739753351e-02 + 1.16096263393e-02 8.89928200875e-03 6.92808975974e-03 5.69415565947e-03 5.26655497606e-03 5.69415565947e-03 + 6.92808975974e-03 8.89928200876e-03 1.16096263393e-02 1.50739753351e-02 1.91106183945e-02 2.31387782150e-02 + 2.75583653887e-02 2.65223600849e-02 2.36140048007e-02 2.10845164455e-02 2.10845164455e-02 2.36140048007e-02 + 2.65223600849e-02 2.75583653887e-02 2.59617964748e-02 2.24043544384e-02 1.81179478705e-02 1.41150555464e-02 + 1.08501521193e-02 8.36367529241e-03 6.62092791971e-03 5.69332826896e-03 5.69332826896e-03 6.62092791971e-03 + 8.36367529241e-03 1.08501521193e-02 1.41150555464e-02 1.81179478705e-02 2.24043544384e-02 2.59617964748e-02 + 2.69079511827e-02 2.36140048007e-02 1.95330017723e-02 1.76765157693e-02 1.95330017723e-02 2.36140048007e-02 + 2.69079511827e-02 2.73901656280e-02 2.49962561202e-02 2.09446150107e-02 1.66276495780e-02 1.28938940154e-02 + 9.96244247361e-03 7.77919418380e-03 6.36560105343e-03 5.86554131393e-03 6.36560105343e-03 7.77919418380e-03 + 9.96244247361e-03 1.28938940154e-02 1.66276495780e-02 2.09446150107e-02 2.49962561202e-02 2.73901656280e-02 + 2.52952680324e-02 2.10845164455e-02 1.76765157693e-02 1.76765157693e-02 2.10845164455e-02 2.52952680324e-02 + 2.74766558578e-02 2.65397992490e-02 2.32028796117e-02 1.89263306496e-02 1.48932680012e-02 1.16030717026e-02 + 9.07040710948e-03 7.25307658557e-03 6.26635360836e-03 6.26635360836e-03 7.25307658557e-03 9.07040710948e-03 + 1.16030717026e-02 1.48932680012e-02 1.89263306496e-02 2.32028796117e-02 2.65397992490e-02 2.74766558578e-02 + 2.44576079909e-02 2.10845164455e-02 1.95330017723e-02 2.10845164455e-02 2.44576079909e-02 2.70509290066e-02 + 2.71139620581e-02 2.46284218885e-02 2.07195371247e-02 1.66389072281e-02 1.31235074234e-02 1.03423481784e-02 + 8.23794392377e-03 6.85673718666e-03 6.36560105343e-03 6.85673718666e-03 8.23794392376e-03 1.03423481784e-02 + 1.31235074234e-02 1.66389072281e-02 2.07195371247e-02 2.46284218885e-02 2.71139620581e-02 2.70509290066e-02 + 2.52952680324e-02 2.36140048007e-02 2.36140048007e-02 2.52952680324e-02 2.70509290066e-02 2.72047829307e-02 + 2.52442663181e-02 2.17946809311e-02 1.78887124933e-02 1.43188156133e-02 1.14032634249e-02 9.13735254995e-03 + 7.50459533107e-03 6.62092791971e-03 6.62092791971e-03 7.50459533106e-03 9.13735254995e-03 1.14032634249e-02 + 1.43188156133e-02 1.78887124933e-02 2.17946809311e-02 2.52442663181e-02 2.72047829307e-02 2.70509290066e-02 + 2.69079511827e-02 2.65223600849e-02 2.69079511827e-02 2.74766558578e-02 2.71139620581e-02 2.52442663181e-02 + 2.21383854932e-02 1.85131064743e-02 1.50423980149e-02 1.21045641480e-02 9.78230850882e-03 8.04465702446e-03 + 6.92808975974e-03 6.53733030468e-03 6.92808975974e-03 8.04465702446e-03 9.78230850882e-03 1.21045641480e-02 + 1.50423980149e-02 1.85131064743e-02 2.21383854932e-02 2.52442663181e-02 2.71139620581e-02 2.74766558578e-02 + 2.75583653887e-02 2.75583653887e-02 2.73901656281e-02 2.65397992490e-02 2.46284218885e-02 2.17946809311e-02 + 1.85131064743e-02 1.52761582425e-02 1.24266085814e-02 1.01199573737e-02 8.37411495641e-03 7.17868687882e-03 + 6.56139720409e-03 6.56139720409e-03 7.17868687882e-03 8.37411495641e-03 1.01199573737e-02 1.24266085814e-02 + 1.52761582425e-02 1.85131064742e-02 2.17946809311e-02 2.46284218885e-02 2.65397992490e-02 2.73901656280e-02 + 2.62414574798e-02 2.59617964748e-02 2.49962561202e-02 2.32028796117e-02 2.07195371247e-02 1.78887124933e-02 + 1.50423980149e-02 1.24266085814e-02 1.02154655471e-02 8.50676845052e-03 7.31939690082e-03 6.62684941428e-03 + 6.39966515662e-03 6.62684941428e-03 7.31939690082e-03 8.50676845052e-03 1.02154655471e-02 1.24266085814e-02 + 1.50423980149e-02 1.78887124933e-02 2.07195371247e-02 2.32028796117e-02 2.49962561202e-02 2.59617964748e-02 + 2.31387782150e-02 2.24043544384e-02 2.09446150107e-02 1.89263306496e-02 1.66389072281e-02 1.43188156133e-02 + 1.21045641480e-02 1.01199573737e-02 8.50676845052e-03 7.36059222257e-03 6.67440432828e-03 6.36712108007e-03 + 6.36712108007e-03 6.67440432828e-03 7.36059222257e-03 8.50676845052e-03 1.01199573737e-02 1.21045641480e-02 + 1.43188156133e-02 1.66389072280e-02 1.89263306496e-02 2.09446150107e-02 2.24043544384e-02 2.31387782150e-02 + 1.91106183945e-02 1.81179478705e-02 1.66276495780e-02 1.48932680012e-02 1.31235074234e-02 1.14032634249e-02 + 9.78230850881e-03 8.37411495641e-03 7.31939690082e-03 6.67440432829e-03 6.36994587767e-03 6.28716491126e-03 + 6.36994587767e-03 6.67440432828e-03 7.31939690082e-03 8.37411495641e-03 9.78230850881e-03 1.14032634249e-02 + 1.31235074234e-02 1.48932680012e-02 1.66276495780e-02 1.81179478705e-02 1.91106183945e-02 1.94536912925e-02 + 1.50739753351e-02 1.41150555464e-02 1.28938940154e-02 1.16030717026e-02 1.03423481784e-02 9.13735254995e-03 + 8.04465702446e-03 7.17868687882e-03 6.62684941428e-03 6.36712108007e-03 6.28716491127e-03 6.28716491126e-03 + 6.36712108007e-03 6.62684941428e-03 7.17868687881e-03 8.04465702445e-03 9.13735254995e-03 1.03423481784e-02 + 1.16030717026e-02 1.28938940154e-02 1.41150555464e-02 1.50739753351e-02 1.55937775675e-02 1.55937775675e-02 + 1.16096263393e-02 1.08501521193e-02 9.96244247361e-03 9.07040710948e-03 8.23794392377e-03 7.50459533107e-03 + 6.92808975974e-03 6.56139720409e-03 6.39966515662e-03 6.36712108007e-03 6.36994587767e-03 6.36712108007e-03 + 6.39966515662e-03 6.56139720409e-03 6.92808975974e-03 7.50459533106e-03 8.23794392376e-03 9.07040710948e-03 + 9.96244247360e-03 1.08501521193e-02 1.16096263393e-02 1.21091269494e-02 1.22801184032e-02 1.21091269494e-02 + 8.89928200875e-03 8.36367529240e-03 7.77919418380e-03 7.25307658557e-03 6.85673718666e-03 6.62092791971e-03 + 6.53733030468e-03 6.56139720409e-03 6.62684941428e-03 6.67440432828e-03 6.67440432828e-03 6.62684941428e-03 + 6.56139720408e-03 6.53733030468e-03 6.62092791971e-03 6.85673718666e-03 7.25307658557e-03 7.77919418380e-03 + 8.36367529240e-03 8.89928200875e-03 9.28365371521e-03 9.47298406612e-03 9.47298406612e-03 9.28365371521e-03 + 6.92808975974e-03 6.62092791971e-03 6.36560105343e-03 6.26635360836e-03 6.36560105343e-03 6.62092791971e-03 + 6.92808975974e-03 7.17868687881e-03 7.31939690081e-03 7.36059222257e-03 7.31939690081e-03 7.17868687881e-03 + 6.92808975974e-03 6.62092791971e-03 6.36560105343e-03 6.26635360836e-03 6.36560105343e-03 6.62092791970e-03 + 6.92808975973e-03 7.17868687881e-03 7.31939690081e-03 7.36059222257e-03 7.31939690081e-03 7.17868687881e-03 + 5.69415565947e-03 5.69332826896e-03 5.86554131393e-03 6.26635360836e-03 6.85673718666e-03 7.50459533106e-03 + 8.04465702445e-03 8.37411495641e-03 8.50676845051e-03 8.50676845051e-03 8.37411495641e-03 8.04465702445e-03 + 7.50459533106e-03 6.85673718666e-03 6.26635360836e-03 5.86554131392e-03 5.69332826895e-03 5.69415565947e-03 + 5.76671059723e-03 5.82750449515e-03 5.85081359876e-03 5.85081359876e-03 5.82750449515e-03 5.76671059723e-03 + 5.26655497606e-03 5.69332826896e-03 6.36560105343e-03 7.25307658557e-03 8.23794392376e-03 9.13735254995e-03 + 9.78230850881e-03 1.01199573737e-02 1.02154655471e-02 1.01199573737e-02 9.78230850881e-03 9.13735254994e-03 + 8.23794392376e-03 7.25307658557e-03 6.36560105343e-03 5.69332826895e-03 5.26655497605e-03 5.04346463958e-03 + 4.94634941966e-03 4.90592224711e-03 4.89322201917e-03 4.90592224711e-03 4.94634941966e-03 5.04346463959e-03 + 5.69415565947e-03 6.62092791971e-03 7.77919418380e-03 9.07040710948e-03 1.03423481784e-02 1.14032634249e-02 + 1.21045641480e-02 1.24266085814e-02 1.24266085814e-02 1.21045641480e-02 1.14032634249e-02 1.03423481784e-02 + 9.07040710948e-03 7.77919418380e-03 6.62092791971e-03 5.69415565947e-03 5.04346463958e-03 4.65178002992e-03 + 4.45369745770e-03 4.37602725379e-03 4.37602725379e-03 4.45369745770e-03 4.65178002992e-03 5.04346463959e-03 + 6.92808975974e-03 8.36367529241e-03 9.96244247361e-03 1.16030717026e-02 1.31235074234e-02 1.43188156133e-02 + 1.50423980149e-02 1.52761582425e-02 1.50423980149e-02 1.43188156133e-02 1.31235074234e-02 1.16030717026e-02 + 9.96244247360e-03 8.36367529240e-03 6.92808975973e-03 5.76671059723e-03 4.94634941966e-03 4.45369745770e-03 + 4.21587397243e-03 4.14922255853e-03 4.21587397244e-03 4.45369745770e-03 4.94634941966e-03 5.76671059723e-03 + 8.89928200876e-03 1.08501521193e-02 1.28938940154e-02 1.48932680012e-02 1.66389072281e-02 1.78887124933e-02 + 1.85131064743e-02 1.85131064743e-02 1.78887124933e-02 1.66389072281e-02 1.48932680012e-02 1.28938940154e-02 + 1.08501521193e-02 8.89928200875e-03 7.17868687881e-03 5.82750449515e-03 4.90592224711e-03 4.37602725379e-03 + 4.14922255853e-03 4.14922255853e-03 4.37602725380e-03 4.90592224711e-03 5.82750449515e-03 7.17868687882e-03 + 1.16096263393e-02 1.41150555464e-02 1.66276495780e-02 1.89263306496e-02 2.07195371247e-02 2.17946809311e-02 + 2.21383854932e-02 2.17946809311e-02 2.07195371247e-02 1.89263306496e-02 1.66276495780e-02 1.41150555464e-02 + 1.16096263393e-02 9.28365371521e-03 7.31939690081e-03 5.85081359876e-03 4.89322201917e-03 4.37602725379e-03 + 4.21587397244e-03 4.37602725380e-03 4.89322201917e-03 5.85081359877e-03 7.31939690082e-03 9.28365371522e-03 + 1.50739753351e-02 1.81179478705e-02 2.09446150107e-02 2.32028796117e-02 2.46284218885e-02 2.52442663181e-02 + 2.52442663181e-02 2.46284218885e-02 2.32028796117e-02 2.09446150107e-02 1.81179478705e-02 1.50739753351e-02 + 1.21091269494e-02 9.47298406612e-03 7.36059222257e-03 5.85081359877e-03 4.90592224711e-03 4.45369745771e-03 + 4.45369745771e-03 4.90592224711e-03 5.85081359877e-03 7.36059222258e-03 9.47298406613e-03 1.21091269495e-02 + 1.91106183945e-02 2.24043544384e-02 2.49962561202e-02 2.65397992490e-02 2.71139620581e-02 2.72047829307e-02 + 2.71139620581e-02 2.65397992490e-02 2.49962561202e-02 2.24043544384e-02 1.91106183945e-02 1.55937775675e-02 + 1.22801184032e-02 9.47298406612e-03 7.31939690081e-03 5.82750449515e-03 4.94634941966e-03 4.65178002993e-03 + 4.94634941966e-03 5.82750449515e-03 7.31939690082e-03 9.47298406613e-03 1.22801184033e-02 1.55937775675e-02 + 2.31387782150e-02 2.59617964748e-02 2.73901656281e-02 2.74766558578e-02 2.70509290066e-02 2.70509290066e-02 + 2.74766558578e-02 2.73901656280e-02 2.59617964748e-02 2.31387782150e-02 1.94536912925e-02 1.55937775675e-02 + 1.21091269494e-02 9.28365371522e-03 7.17868687881e-03 5.76671059723e-03 5.04346463959e-03 5.04346463959e-03 + 5.76671059723e-03 7.17868687882e-03 9.28365371522e-03 1.21091269495e-02 1.55937775675e-02 1.94536912926e-02 + 3.06060385055e-02 3.03790382413e-02 2.81921518485e-02 2.61163074370e-02 2.61163074370e-02 2.81921518485e-02 + 3.03790382414e-02 3.06060385055e-02 2.81654297894e-02 2.37302554132e-02 1.86355831047e-02 1.40807237278e-02 + 1.06052341643e-02 8.15763089847e-03 6.53733030468e-03 5.69415565947e-03 5.69415565947e-03 6.53733030469e-03 + 8.15763089847e-03 1.06052341643e-02 1.40807237278e-02 1.86355831047e-02 2.37302554132e-02 2.81654297894e-02 + 3.03790382413e-02 2.73937743101e-02 2.35389487571e-02 2.17745697719e-02 2.35389487571e-02 2.73937743101e-02 + 3.03790382414e-02 3.04051608500e-02 2.73091130234e-02 2.23752062804e-02 1.72612102705e-02 1.30619533075e-02 + 1.00211358334e-02 7.92503968744e-03 6.62092791971e-03 6.16390944840e-03 6.62092791971e-03 7.92503968744e-03 + 1.00211358334e-02 1.30619533075e-02 1.72612102705e-02 2.23752062804e-02 2.73091130234e-02 3.04051608500e-02 + 2.81921518485e-02 2.35389487571e-02 1.98434626957e-02 1.98434626957e-02 2.35389487571e-02 2.81921518485e-02 + 3.06791433435e-02 2.95889063006e-02 2.55724702285e-02 2.03920068264e-02 1.56490177739e-02 1.20440164119e-02 + 9.50257178747e-03 7.77919418380e-03 6.85673718666e-03 6.85673718666e-03 7.77919418380e-03 9.50257178747e-03 + 1.20440164119e-02 1.56490177739e-02 2.03920068264e-02 2.55724702285e-02 2.95889063006e-02 3.06791433435e-02 + 2.61163074370e-02 2.17745697719e-02 1.98434626957e-02 2.17745697719e-02 2.61163074370e-02 2.97972897261e-02 + 3.04771749334e-02 2.78012163806e-02 2.30937481115e-02 1.81429746336e-02 1.40901757966e-02 1.11481924118e-02 + 9.07040710948e-03 7.73184010469e-03 7.25307658557e-03 7.73184010469e-03 9.07040710948e-03 1.11481924118e-02 + 1.40901757966e-02 1.81429746336e-02 2.30937481115e-02 2.78012163806e-02 3.04771749334e-02 2.97972897261e-02 + 2.61163074370e-02 2.35389487571e-02 2.35389487571e-02 2.61163074370e-02 2.92434029241e-02 3.05546433144e-02 + 2.89472257765e-02 2.49901046905e-02 2.02039597509e-02 1.59286730225e-02 1.26855892423e-02 1.03423481784e-02 + 8.68885911335e-03 7.77919418380e-03 7.77919418380e-03 8.68885911334e-03 1.03423481784e-02 1.26855892424e-02 + 1.59286730225e-02 2.02039597509e-02 2.49901046905e-02 2.89472257765e-02 3.05546433144e-02 2.92434029240e-02 + 2.81921518485e-02 2.73937743101e-02 2.81921518485e-02 2.97972897261e-02 3.05546433144e-02 2.92642577664e-02 + 2.59180287237e-02 2.15194519516e-02 1.72810954144e-02 1.38961510916e-02 1.14032634249e-02 9.57630048387e-03 + 8.36367529241e-03 7.92503968744e-03 8.36367529241e-03 9.57630048387e-03 1.14032634249e-02 1.38961510916e-02 + 1.72810954144e-02 2.15194519516e-02 2.59180287237e-02 2.92642577664e-02 3.05546433144e-02 2.97972897261e-02 + 3.03790382414e-02 3.03790382414e-02 3.06791433435e-02 3.04771749334e-02 2.89472257765e-02 2.59180287237e-02 + 2.19609803409e-02 1.79740052840e-02 1.46288065643e-02 1.21045641480e-02 1.02402112845e-02 8.89928200876e-03 + 8.15763089848e-03 8.15763089847e-03 8.89928200876e-03 1.02402112845e-02 1.21045641480e-02 1.46288065643e-02 + 1.79740052840e-02 2.19609803409e-02 2.59180287237e-02 2.89472257765e-02 3.04771749334e-02 3.06791433435e-02 + 3.06060385055e-02 3.04051608500e-02 2.95889063006e-02 2.78012163806e-02 2.49901046905e-02 2.15194519516e-02 + 1.79740052840e-02 1.48655958108e-02 1.24266085814e-02 1.06110221336e-02 9.28365371522e-03 8.42000879274e-03 + 8.11231618988e-03 8.42000879274e-03 9.28365371522e-03 1.06110221336e-02 1.24266085814e-02 1.48655958108e-02 + 1.79740052840e-02 2.15194519516e-02 2.49901046905e-02 2.78012163806e-02 2.95889063006e-02 3.04051608500e-02 + 2.81654297894e-02 2.73091130234e-02 2.55724702285e-02 2.30937481115e-02 2.02039597509e-02 1.72810954144e-02 + 1.46288065643e-02 1.24266085814e-02 1.07226215343e-02 9.47298406613e-03 8.62544633520e-03 8.18384117181e-03 + 8.18384117181e-03 8.62544633520e-03 9.47298406613e-03 1.07226215343e-02 1.24266085814e-02 1.46288065643e-02 + 1.72810954144e-02 2.02039597509e-02 2.30937481115e-02 2.55724702285e-02 2.73091130234e-02 2.81654297894e-02 + 2.37302554132e-02 2.23752062804e-02 2.03920068264e-02 1.81429746336e-02 1.59286730224e-02 1.38961510916e-02 + 1.21045641480e-02 1.06110221336e-02 9.47298406613e-03 8.70260576991e-03 8.26411545329e-03 8.12162338295e-03 + 8.26411545329e-03 8.70260576991e-03 9.47298406612e-03 1.06110221336e-02 1.21045641480e-02 1.38961510916e-02 + 1.59286730224e-02 1.81429746336e-02 2.03920068263e-02 2.23752062804e-02 2.37302554132e-02 2.42076546579e-02 + 1.86355831047e-02 1.72612102704e-02 1.56490177739e-02 1.40901757966e-02 1.26855892423e-02 1.14032634249e-02 + 1.02402112845e-02 9.28365371522e-03 8.62544633520e-03 8.26411545329e-03 8.11876205862e-03 8.11876205861e-03 + 8.26411545329e-03 8.62544633520e-03 9.28365371522e-03 1.02402112845e-02 1.14032634249e-02 1.26855892423e-02 + 1.40901757966e-02 1.56490177739e-02 1.72612102704e-02 1.86355831047e-02 1.94291633463e-02 1.94291633463e-02 + 1.40807237278e-02 1.30619533075e-02 1.20440164119e-02 1.11481924117e-02 1.03423481784e-02 9.57630048387e-03 + 8.89928200876e-03 8.42000879274e-03 8.18384117182e-03 8.12162338296e-03 8.11876205861e-03 8.12162338295e-03 + 8.18384117181e-03 8.42000879274e-03 8.89928200876e-03 9.57630048387e-03 1.03423481784e-02 1.11481924117e-02 + 1.20440164119e-02 1.30619533075e-02 1.40807237278e-02 1.48493143122e-02 1.51356003261e-02 1.48493143122e-02 + 1.06052341643e-02 1.00211358334e-02 9.50257178747e-03 9.07040710948e-03 8.68885911335e-03 8.36367529241e-03 + 8.15763089847e-03 8.11231618988e-03 8.18384117181e-03 8.26411545329e-03 8.26411545329e-03 8.18384117181e-03 + 8.11231618988e-03 8.15763089847e-03 8.36367529241e-03 8.68885911334e-03 9.07040710948e-03 9.50257178747e-03 + 1.00211358334e-02 1.06052341643e-02 1.11332793734e-02 1.14482068790e-02 1.14482068790e-02 1.11332793734e-02 + 8.15763089847e-03 7.92503968744e-03 7.77919418380e-03 7.73184010469e-03 7.77919418380e-03 7.92503968744e-03 + 8.15763089847e-03 8.42000879273e-03 8.62544633520e-03 8.70260576990e-03 8.62544633519e-03 8.42000879273e-03 + 8.15763089847e-03 7.92503968744e-03 7.77919418380e-03 7.73184010469e-03 7.77919418380e-03 7.92503968743e-03 + 8.15763089847e-03 8.42000879273e-03 8.62544633519e-03 8.70260576990e-03 8.62544633519e-03 8.42000879273e-03 + 6.53733030468e-03 6.62092791971e-03 6.85673718666e-03 7.25307658557e-03 7.77919418380e-03 8.36367529240e-03 + 8.89928200875e-03 9.28365371522e-03 9.47298406612e-03 9.47298406612e-03 9.28365371521e-03 8.89928200875e-03 + 8.36367529240e-03 7.77919418380e-03 7.25307658557e-03 6.85673718666e-03 6.62092791971e-03 6.53733030468e-03 + 6.56139720408e-03 6.62684941427e-03 6.67440432828e-03 6.67440432828e-03 6.62684941427e-03 6.56139720408e-03 + 5.69415565947e-03 6.16390944840e-03 6.85673718666e-03 7.73184010469e-03 8.68885911334e-03 9.57630048387e-03 + 1.02402112845e-02 1.06110221336e-02 1.07226215343e-02 1.06110221336e-02 1.02402112845e-02 9.57630048386e-03 + 8.68885911334e-03 7.73184010469e-03 6.85673718666e-03 6.16390944840e-03 5.69415565947e-03 5.43241849957e-03 + 5.32069114083e-03 5.28637047081e-03 5.27979197805e-03 5.28637047081e-03 5.32069114083e-03 5.43241849957e-03 + 5.69415565947e-03 6.62092791971e-03 7.77919418380e-03 9.07040710948e-03 1.03423481784e-02 1.14032634249e-02 + 1.21045641480e-02 1.24266085814e-02 1.24266085814e-02 1.21045641480e-02 1.14032634249e-02 1.03423481784e-02 + 9.07040710948e-03 7.77919418380e-03 6.62092791971e-03 5.69415565947e-03 5.04346463958e-03 4.65178002992e-03 + 4.45369745770e-03 4.37602725379e-03 4.37602725379e-03 4.45369745770e-03 4.65178002992e-03 5.04346463959e-03 + 6.53733030469e-03 7.92503968744e-03 9.50257178747e-03 1.11481924117e-02 1.26855892424e-02 1.38961510916e-02 + 1.46288065643e-02 1.48655958108e-02 1.46288065643e-02 1.38961510916e-02 1.26855892423e-02 1.11481924117e-02 + 9.50257178747e-03 7.92503968743e-03 6.53733030468e-03 5.43241849957e-03 4.65178002992e-03 4.16949135529e-03 + 3.92286252630e-03 3.84969900803e-03 3.92286252630e-03 4.16949135529e-03 4.65178002992e-03 5.43241849957e-03 + 8.15763089847e-03 1.00211358334e-02 1.20440164119e-02 1.40901757966e-02 1.59286730225e-02 1.72810954144e-02 + 1.79740052840e-02 1.79740052840e-02 1.72810954144e-02 1.59286730225e-02 1.40901757966e-02 1.20440164119e-02 + 1.00211358334e-02 8.15763089847e-03 6.56139720408e-03 5.32069114083e-03 4.45369745770e-03 3.92286252630e-03 + 3.67775927312e-03 3.67775927312e-03 3.92286252630e-03 4.45369745770e-03 5.32069114083e-03 6.56139720409e-03 + 1.06052341643e-02 1.30619533075e-02 1.56490177739e-02 1.81429746336e-02 2.02039597509e-02 2.15194519516e-02 + 2.19609803409e-02 2.15194519516e-02 2.02039597509e-02 1.81429746336e-02 1.56490177739e-02 1.30619533075e-02 + 1.06052341643e-02 8.42000879273e-03 6.62684941427e-03 5.28637047081e-03 4.37602725379e-03 3.84969900803e-03 + 3.67775927312e-03 3.84969900804e-03 4.37602725380e-03 5.28637047082e-03 6.62684941428e-03 8.42000879274e-03 + 1.40807237278e-02 1.72612102705e-02 2.03920068264e-02 2.30937481116e-02 2.49901046905e-02 2.59180287237e-02 + 2.59180287237e-02 2.49901046905e-02 2.30937481115e-02 2.03920068264e-02 1.72612102705e-02 1.40807237278e-02 + 1.11332793734e-02 8.62544633519e-03 6.67440432828e-03 5.27979197805e-03 4.37602725379e-03 3.92286252630e-03 + 3.92286252630e-03 4.37602725380e-03 5.27979197806e-03 6.67440432829e-03 8.62544633520e-03 1.11332793734e-02 + 1.86355831047e-02 2.23752062804e-02 2.55724702285e-02 2.78012163806e-02 2.89472257765e-02 2.92642577664e-02 + 2.89472257765e-02 2.78012163806e-02 2.55724702285e-02 2.23752062804e-02 1.86355831048e-02 1.48493143123e-02 + 1.14482068790e-02 8.70260576990e-03 6.67440432828e-03 5.28637047081e-03 4.45369745770e-03 4.16949135529e-03 + 4.45369745770e-03 5.28637047082e-03 6.67440432829e-03 8.70260576991e-03 1.14482068791e-02 1.48493143123e-02 + 2.37302554132e-02 2.73091130234e-02 2.95889063006e-02 3.04771749334e-02 3.05546433144e-02 3.05546433144e-02 + 3.04771749334e-02 2.95889063006e-02 2.73091130234e-02 2.37302554132e-02 1.94291633463e-02 1.51356003261e-02 + 1.14482068790e-02 8.62544633519e-03 6.62684941427e-03 5.32069114083e-03 4.65178002992e-03 4.65178002992e-03 + 5.32069114083e-03 6.62684941428e-03 8.62544633520e-03 1.14482068791e-02 1.51356003261e-02 1.94291633463e-02 + 2.81654297894e-02 3.04051608500e-02 3.06791433435e-02 2.97972897261e-02 2.92434029240e-02 2.97972897260e-02 + 3.06791433435e-02 3.04051608500e-02 2.81654297894e-02 2.42076546579e-02 1.94291633463e-02 1.48493143122e-02 + 1.11332793734e-02 8.42000879273e-03 6.56139720408e-03 5.43241849957e-03 5.04346463959e-03 5.43241849957e-03 + 6.56139720409e-03 8.42000879274e-03 1.11332793734e-02 1.48493143123e-02 1.94291633463e-02 2.42076546579e-02 + 3.48685420831e-02 3.31552867934e-02 3.05271423180e-02 2.92886940371e-02 3.05271423180e-02 3.31552867935e-02 + 3.48685420831e-02 3.38926655663e-02 2.99300705585e-02 2.40966608398e-02 1.81617249718e-02 1.34421984781e-02 + 1.02249435892e-02 8.15763089847e-03 6.92808975974e-03 6.50344737139e-03 6.92808975974e-03 8.15763089847e-03 + 1.02249435892e-02 1.34421984781e-02 1.81617249718e-02 2.40966608398e-02 2.99300705585e-02 3.38926655662e-02 + 3.31552867935e-02 2.95747487953e-02 2.67211542468e-02 2.67211542468e-02 2.95747487953e-02 3.31552867935e-02 + 3.48629669819e-02 3.31962688978e-02 2.84215335265e-02 2.22711776717e-02 1.66949823920e-02 1.26512009150e-02 + 1.00211358334e-02 8.36367529241e-03 7.50459533107e-03 7.50459533107e-03 8.36367529241e-03 1.00211358334e-02 + 1.26512009150e-02 1.66949823920e-02 2.22711776717e-02 2.84215335265e-02 3.31962688978e-02 3.48629669819e-02 + 3.05271423180e-02 2.67211542468e-02 2.50655517889e-02 2.67211542468e-02 3.05271423180e-02 3.39197128099e-02 + 3.45751721123e-02 3.16055353689e-02 2.60490352762e-02 2.00581600358e-02 1.52718979502e-02 1.20440164119e-02 + 9.96244247361e-03 8.68885911335e-03 8.23794392377e-03 8.68885911334e-03 9.96244247361e-03 1.20440164119e-02 + 1.52718979502e-02 2.00581600358e-02 2.60490352762e-02 3.16055353689e-02 3.45751721122e-02 3.39197128099e-02 + 2.92886940372e-02 2.67211542468e-02 2.67211542468e-02 2.92886940372e-02 3.26919751552e-02 3.46097554135e-02 + 3.33648425996e-02 2.89589200656e-02 2.31229887194e-02 1.78656633258e-02 1.40901757966e-02 1.16030717026e-02 + 9.96178172025e-03 9.07040710948e-03 9.07040710948e-03 9.96178172025e-03 1.16030717026e-02 1.40901757966e-02 + 1.78656633258e-02 2.31229887194e-02 2.89589200656e-02 3.33648425996e-02 3.46097554135e-02 3.26919751551e-02 + 3.05271423180e-02 2.95747487953e-02 3.05271423180e-02 3.26919751552e-02 3.44218462469e-02 3.39946385756e-02 + 3.06730045455e-02 2.54054025138e-02 2.00652756288e-02 1.59286730224e-02 1.31235074234e-02 1.12138690779e-02 + 9.96244247361e-03 9.50257178747e-03 9.96244247361e-03 1.12138690779e-02 1.31235074234e-02 1.59286730225e-02 + 2.00652756289e-02 2.54054025138e-02 3.06730045455e-02 3.39946385756e-02 3.44218462468e-02 3.26919751551e-02 + 3.31552867935e-02 3.31552867935e-02 3.39197128099e-02 3.46097554135e-02 3.39946385756e-02 3.12137668085e-02 + 2.65859681267e-02 2.15037112467e-02 1.72810954144e-02 1.43188156133e-02 1.22935034476e-02 1.08501521193e-02 + 1.00211358334e-02 1.00211358334e-02 1.08501521193e-02 1.22935034476e-02 1.43188156133e-02 1.72810954144e-02 + 2.15037112467e-02 2.65859681267e-02 3.12137668085e-02 3.39946385756e-02 3.46097554135e-02 3.39197128099e-02 + 3.48685420831e-02 3.48629669819e-02 3.45751721123e-02 3.33648425996e-02 3.06730045455e-02 2.65859681267e-02 + 2.19946944680e-02 1.79740052840e-02 1.50423980149e-02 1.30486211596e-02 1.16096263393e-02 1.06052341643e-02 + 1.02249435892e-02 1.06052341643e-02 1.16096263393e-02 1.30486211596e-02 1.50423980149e-02 1.79740052840e-02 + 2.19946944680e-02 2.65859681267e-02 3.06730045455e-02 3.33648425996e-02 3.45751721122e-02 3.48629669819e-02 + 3.38926655663e-02 3.31962688978e-02 3.16055353689e-02 2.89589200656e-02 2.54054025138e-02 2.15037112467e-02 + 1.79740052840e-02 1.52761582425e-02 1.34169422858e-02 1.21091269495e-02 1.11332793734e-02 1.05558438840e-02 + 1.05558438840e-02 1.11332793734e-02 1.21091269495e-02 1.34169422858e-02 1.52761582425e-02 1.79740052840e-02 + 2.15037112467e-02 2.54054025138e-02 2.89589200656e-02 3.16055353689e-02 3.31962688978e-02 3.38926655663e-02 + 2.99300705585e-02 2.84215335265e-02 2.60490352762e-02 2.31229887194e-02 2.00652756288e-02 1.72810954144e-02 + 1.50423980149e-02 1.34169422858e-02 1.22801184033e-02 1.14482068791e-02 1.08700745212e-02 1.06519075585e-02 + 1.08700745212e-02 1.14482068790e-02 1.22801184033e-02 1.34169422858e-02 1.50423980149e-02 1.72810954144e-02 + 2.00652756288e-02 2.31229887194e-02 2.60490352762e-02 2.84215335265e-02 2.99300705585e-02 3.04405035973e-02 + 2.40966608398e-02 2.22711776717e-02 2.00581600358e-02 1.78656633258e-02 1.59286730224e-02 1.43188156133e-02 + 1.30486211596e-02 1.21091269495e-02 1.14482068791e-02 1.09991800130e-02 1.07548084489e-02 1.07548084489e-02 + 1.09991800129e-02 1.14482068790e-02 1.21091269495e-02 1.30486211596e-02 1.43188156133e-02 1.59286730225e-02 + 1.78656633258e-02 2.00581600358e-02 2.22711776717e-02 2.40966608398e-02 2.51250422144e-02 2.51250422144e-02 + 1.81617249718e-02 1.66949823920e-02 1.52718979502e-02 1.40901757966e-02 1.31235074234e-02 1.22935034476e-02 + 1.16096263393e-02 1.11332793734e-02 1.08700745212e-02 1.07548084489e-02 1.07235160838e-02 1.07548084489e-02 + 1.08700745212e-02 1.11332793734e-02 1.16096263393e-02 1.22935034476e-02 1.31235074234e-02 1.40901757966e-02 + 1.52718979502e-02 1.66949823920e-02 1.81617249718e-02 1.92841419762e-02 1.97046043739e-02 1.92841419762e-02 + 1.34421984781e-02 1.26512009150e-02 1.20440164119e-02 1.16030717026e-02 1.12138690779e-02 1.08501521193e-02 + 1.06052341643e-02 1.05558438840e-02 1.06519075585e-02 1.07548084489e-02 1.07548084489e-02 1.06519075585e-02 + 1.05558438840e-02 1.06052341643e-02 1.08501521193e-02 1.12138690779e-02 1.16030717026e-02 1.20440164119e-02 + 1.26512009150e-02 1.34421984781e-02 1.42309947893e-02 1.47280857174e-02 1.47280857174e-02 1.42309947893e-02 + 1.02249435892e-02 1.00211358334e-02 9.96244247361e-03 9.96178172025e-03 9.96244247361e-03 1.00211358334e-02 + 1.02249435892e-02 1.05558438840e-02 1.08700745212e-02 1.09991800129e-02 1.08700745212e-02 1.05558438840e-02 + 1.02249435892e-02 1.00211358334e-02 9.96244247360e-03 9.96178172025e-03 9.96244247360e-03 1.00211358334e-02 + 1.02249435892e-02 1.05558438840e-02 1.08700745212e-02 1.09991800129e-02 1.08700745212e-02 1.05558438840e-02 + 8.15763089847e-03 8.36367529241e-03 8.68885911335e-03 9.07040710948e-03 9.50257178747e-03 1.00211358334e-02 + 1.06052341643e-02 1.11332793734e-02 1.14482068790e-02 1.14482068790e-02 1.11332793734e-02 1.06052341643e-02 + 1.00211358334e-02 9.50257178747e-03 9.07040710948e-03 8.68885911334e-03 8.36367529240e-03 8.15763089847e-03 + 8.11231618988e-03 8.18384117181e-03 8.26411545329e-03 8.26411545328e-03 8.18384117181e-03 8.11231618988e-03 + 6.92808975974e-03 7.50459533107e-03 8.23794392377e-03 9.07040710948e-03 9.96244247360e-03 1.08501521193e-02 + 1.16096263393e-02 1.21091269494e-02 1.22801184032e-02 1.21091269494e-02 1.16096263393e-02 1.08501521193e-02 + 9.96244247360e-03 9.07040710948e-03 8.23794392377e-03 7.50459533107e-03 6.92808975974e-03 6.56139720408e-03 + 6.39966515661e-03 6.36712108007e-03 6.36994587767e-03 6.36712108007e-03 6.39966515661e-03 6.56139720408e-03 + 6.50344737139e-03 7.50459533107e-03 8.68885911334e-03 9.96178172025e-03 1.12138690779e-02 1.22935034476e-02 + 1.30486211596e-02 1.34169422858e-02 1.34169422858e-02 1.30486211596e-02 1.22935034476e-02 1.12138690779e-02 + 9.96178172025e-03 8.68885911334e-03 7.50459533107e-03 6.50344737139e-03 5.76671059723e-03 5.32069114083e-03 + 5.11302992589e-03 5.04613651753e-03 5.04613651753e-03 5.11302992589e-03 5.32069114083e-03 5.76671059724e-03 + 6.92808975974e-03 8.36367529241e-03 9.96244247360e-03 1.16030717026e-02 1.31235074234e-02 1.43188156133e-02 + 1.50423980149e-02 1.52761582425e-02 1.50423980149e-02 1.43188156133e-02 1.31235074234e-02 1.16030717026e-02 + 9.96244247360e-03 8.36367529240e-03 6.92808975974e-03 5.76671059723e-03 4.94634941966e-03 4.45369745770e-03 + 4.21587397243e-03 4.14922255853e-03 4.21587397243e-03 4.45369745770e-03 4.94634941966e-03 5.76671059724e-03 + 8.15763089847e-03 1.00211358334e-02 1.20440164119e-02 1.40901757966e-02 1.59286730225e-02 1.72810954144e-02 + 1.79740052840e-02 1.79740052840e-02 1.72810954144e-02 1.59286730225e-02 1.40901757966e-02 1.20440164119e-02 + 1.00211358334e-02 8.15763089847e-03 6.56139720408e-03 5.32069114083e-03 4.45369745770e-03 3.92286252630e-03 + 3.67775927312e-03 3.67775927312e-03 3.92286252630e-03 4.45369745770e-03 5.32069114083e-03 6.56139720409e-03 + 1.02249435892e-02 1.26512009150e-02 1.52718979502e-02 1.78656633258e-02 2.00652756289e-02 2.15037112468e-02 + 2.19946944681e-02 2.15037112468e-02 2.00652756289e-02 1.78656633258e-02 1.52718979502e-02 1.26512009150e-02 + 1.02249435892e-02 8.11231618988e-03 6.39966515661e-03 5.11302992589e-03 4.21587397243e-03 3.67775927312e-03 + 3.49735469022e-03 3.67775927312e-03 4.21587397244e-03 5.11302992590e-03 6.39966515662e-03 8.11231618988e-03 + 1.34421984781e-02 1.66949823920e-02 2.00581600358e-02 2.31229887194e-02 2.54054025138e-02 2.65859681267e-02 + 2.65859681267e-02 2.54054025138e-02 2.31229887194e-02 2.00581600358e-02 1.66949823920e-02 1.34421984781e-02 + 1.05558438840e-02 8.18384117181e-03 6.36712108007e-03 5.04613651753e-03 4.14922255853e-03 3.67775927312e-03 + 3.67775927312e-03 4.14922255853e-03 5.04613651754e-03 6.36712108007e-03 8.18384117181e-03 1.05558438840e-02 + 1.81617249718e-02 2.22711776717e-02 2.60490352762e-02 2.89589200657e-02 3.06730045455e-02 3.12137668085e-02 + 3.06730045455e-02 2.89589200657e-02 2.60490352762e-02 2.22711776717e-02 1.81617249718e-02 1.42309947893e-02 + 1.08700745212e-02 8.26411545328e-03 6.36994587766e-03 5.04613651753e-03 4.21587397243e-03 3.92286252630e-03 + 4.21587397244e-03 5.04613651754e-03 6.36994587767e-03 8.26411545329e-03 1.08700745212e-02 1.42309947893e-02 + 2.40966608398e-02 2.84215335265e-02 3.16055353689e-02 3.33648425996e-02 3.39946385756e-02 3.39946385756e-02 + 3.33648425996e-02 3.16055353689e-02 2.84215335265e-02 2.40966608398e-02 1.92841419762e-02 1.47280857174e-02 + 1.09991800129e-02 8.26411545328e-03 6.36712108006e-03 5.11302992589e-03 4.45369745770e-03 4.45369745770e-03 + 5.11302992590e-03 6.36712108007e-03 8.26411545329e-03 1.09991800130e-02 1.47280857174e-02 1.92841419762e-02 + 2.99300705585e-02 3.31962688978e-02 3.45751721122e-02 3.46097554135e-02 3.44218462468e-02 3.46097554135e-02 + 3.45751721122e-02 3.31962688978e-02 2.99300705585e-02 2.51250422144e-02 1.97046043739e-02 1.47280857174e-02 + 1.08700745212e-02 8.18384117180e-03 6.39966515661e-03 5.32069114083e-03 4.94634941966e-03 5.32069114083e-03 + 6.39966515662e-03 8.18384117181e-03 1.08700745212e-02 1.47280857174e-02 1.97046043739e-02 2.51250422144e-02 + 3.38926655663e-02 3.48629669819e-02 3.39197128099e-02 3.26919751551e-02 3.26919751551e-02 3.39197128099e-02 + 3.48629669819e-02 3.38926655663e-02 3.04405035973e-02 2.51250422144e-02 1.92841419762e-02 1.42309947893e-02 + 1.05558438840e-02 8.11231618988e-03 6.56139720408e-03 5.76671059723e-03 5.76671059723e-03 6.56139720409e-03 + 8.11231618988e-03 1.05558438840e-02 1.42309947893e-02 1.92841419762e-02 2.51250422144e-02 3.04405035973e-02 + 4.15728345981e-02 4.23899114589e-02 4.26144257223e-02 4.26144257223e-02 4.23899114590e-02 4.15728345981e-02 + 3.97328731661e-02 3.62670291685e-02 3.07818020073e-02 2.40902041017e-02 1.79266369346e-02 1.34421984781e-02 + 1.06052341643e-02 8.89928200876e-03 8.04465702446e-03 8.04465702446e-03 8.89928200876e-03 1.06052341643e-02 + 1.34421984781e-02 1.79266369346e-02 2.40902041017e-02 3.07818020073e-02 3.62670291685e-02 3.97328731661e-02 + 4.23899114590e-02 4.30325583083e-02 4.31923684876e-02 4.30325583083e-02 4.23899114590e-02 4.11390439155e-02 + 3.89797931477e-02 3.49765243617e-02 2.88774182856e-02 2.21707375456e-02 1.66949823920e-02 1.30619533075e-02 + 1.08501521193e-02 9.57630048387e-03 9.13735254995e-03 9.57630048387e-03 1.08501521193e-02 1.30619533075e-02 + 1.66949823920e-02 2.21707375456e-02 2.88774182856e-02 3.49765243617e-02 3.89797931477e-02 4.11390439154e-02 + 4.26144257223e-02 4.31923684876e-02 4.31923684876e-02 4.26144257223e-02 4.15214590889e-02 4.00844862497e-02 + 3.75326780270e-02 3.27027225515e-02 2.61899431515e-02 2.00581600358e-02 1.56490177739e-02 1.28938940154e-02 + 1.12138690779e-02 1.03423481784e-02 1.03423481784e-02 1.12138690779e-02 1.28938940154e-02 1.56490177739e-02 + 2.00581600358e-02 2.61899431515e-02 3.27027225515e-02 3.75326780270e-02 4.00844862497e-02 4.15214590889e-02 + 4.26144257223e-02 4.30325583083e-02 4.26144257223e-02 4.15474648475e-02 4.03066875223e-02 3.86301228830e-02 + 3.51265406939e-02 2.93978765598e-02 2.31229887194e-02 1.81429746336e-02 1.48932680012e-02 1.28581680779e-02 + 1.16030717026e-02 1.11481924118e-02 1.16030717026e-02 1.28581680779e-02 1.48932680013e-02 1.81429746336e-02 + 2.31229887194e-02 2.93978765598e-02 3.51265406940e-02 3.86301228830e-02 4.03066875222e-02 4.15474648474e-02 + 4.23899114590e-02 4.23899114590e-02 4.15214590889e-02 4.03066875223e-02 3.88820048627e-02 3.61943858422e-02 + 3.13659743853e-02 2.54054025138e-02 2.02039597509e-02 1.66389072281e-02 1.43895601245e-02 1.28938940154e-02 + 1.20440164119e-02 1.20440164119e-02 1.28938940154e-02 1.43895601245e-02 1.66389072281e-02 2.02039597509e-02 + 2.54054025138e-02 3.13659743854e-02 3.61943858423e-02 3.88820048627e-02 4.03066875223e-02 4.15214590889e-02 + 4.15728345981e-02 4.11390439155e-02 4.00844862497e-02 3.86301228830e-02 3.61943858422e-02 3.20076435873e-02 + 2.65859681267e-02 2.15194519516e-02 1.78887124933e-02 1.56224103699e-02 1.41150555464e-02 1.30619533075e-02 + 1.26512009150e-02 1.30619533075e-02 1.41150555464e-02 1.56224103699e-02 1.78887124933e-02 2.15194519516e-02 + 2.65859681267e-02 3.20076435873e-02 3.61943858423e-02 3.86301228830e-02 4.00844862497e-02 4.11390439155e-02 + 3.97328731661e-02 3.89797931477e-02 3.75326780270e-02 3.51265406939e-02 3.13659743853e-02 2.65859681267e-02 + 2.19609803409e-02 1.85131064743e-02 1.63905844408e-02 1.50739753351e-02 1.40807237278e-02 1.34421984781e-02 + 1.34421984781e-02 1.40807237278e-02 1.50739753351e-02 1.63905844408e-02 1.85131064743e-02 2.19609803409e-02 + 2.65859681267e-02 3.13659743854e-02 3.51265406940e-02 3.75326780270e-02 3.89797931477e-02 3.97328731661e-02 + 3.62670291685e-02 3.49765243617e-02 3.27027225515e-02 2.93978765598e-02 2.54054025138e-02 2.15194519516e-02 + 1.85131064743e-02 1.66438543464e-02 1.55937775675e-02 1.48493143123e-02 1.42309947893e-02 1.39668146374e-02 + 1.42309947893e-02 1.48493143122e-02 1.55937775675e-02 1.66438543464e-02 1.85131064743e-02 2.15194519516e-02 + 2.54054025138e-02 2.93978765598e-02 3.27027225515e-02 3.49765243617e-02 3.62670291685e-02 3.66852084288e-02 + 3.07818020073e-02 2.88774182856e-02 2.61899431515e-02 2.31229887194e-02 2.02039597509e-02 1.78887124933e-02 + 1.63905844408e-02 1.55937775675e-02 1.51356003261e-02 1.47280857174e-02 1.44211937670e-02 1.44211937670e-02 + 1.47280857174e-02 1.51356003261e-02 1.55937775675e-02 1.63905844408e-02 1.78887124933e-02 2.02039597509e-02 + 2.31229887194e-02 2.61899431515e-02 2.88774182856e-02 3.07818020073e-02 3.17358439118e-02 3.17358439118e-02 + 2.40902041017e-02 2.21707375456e-02 2.00581600358e-02 1.81429746336e-02 1.66389072281e-02 1.56224103699e-02 + 1.50739753351e-02 1.48493143123e-02 1.47280857174e-02 1.45988685534e-02 1.45323142804e-02 1.45988685534e-02 + 1.47280857174e-02 1.48493143122e-02 1.50739753351e-02 1.56224103699e-02 1.66389072281e-02 1.81429746336e-02 + 2.00581600358e-02 2.21707375455e-02 2.40902041017e-02 2.53972544857e-02 2.58521406843e-02 2.53972544857e-02 + 1.79266369346e-02 1.66949823920e-02 1.56490177739e-02 1.48932680012e-02 1.43895601245e-02 1.41150555464e-02 + 1.40807237278e-02 1.42309947893e-02 1.44211937670e-02 1.45323142804e-02 1.45323142804e-02 1.44211937670e-02 + 1.42309947893e-02 1.40807237278e-02 1.41150555464e-02 1.43895601245e-02 1.48932680012e-02 1.56490177739e-02 + 1.66949823920e-02 1.79266369346e-02 1.90346417875e-02 1.96803910193e-02 1.96803910193e-02 1.90346417875e-02 + 1.34421984781e-02 1.30619533075e-02 1.28938940154e-02 1.28581680779e-02 1.28938940154e-02 1.30619533075e-02 + 1.34421984781e-02 1.39668146374e-02 1.44211937670e-02 1.45988685534e-02 1.44211937670e-02 1.39668146374e-02 + 1.34421984781e-02 1.30619533075e-02 1.28938940154e-02 1.28581680779e-02 1.28938940154e-02 1.30619533075e-02 + 1.34421984781e-02 1.39668146374e-02 1.44211937670e-02 1.45988685534e-02 1.44211937670e-02 1.39668146374e-02 + 1.06052341643e-02 1.08501521193e-02 1.12138690779e-02 1.16030717026e-02 1.20440164119e-02 1.26512009150e-02 + 1.34421984781e-02 1.42309947893e-02 1.47280857174e-02 1.47280857174e-02 1.42309947893e-02 1.34421984781e-02 + 1.26512009150e-02 1.20440164119e-02 1.16030717026e-02 1.12138690779e-02 1.08501521193e-02 1.06052341643e-02 + 1.05558438840e-02 1.06519075585e-02 1.07548084489e-02 1.07548084489e-02 1.06519075585e-02 1.05558438840e-02 + 8.89928200876e-03 9.57630048387e-03 1.03423481784e-02 1.11481924117e-02 1.20440164119e-02 1.30619533075e-02 + 1.40807237278e-02 1.48493143122e-02 1.51356003261e-02 1.48493143122e-02 1.40807237278e-02 1.30619533075e-02 + 1.20440164119e-02 1.11481924117e-02 1.03423481784e-02 9.57630048387e-03 8.89928200875e-03 8.42000879273e-03 + 8.18384117181e-03 8.12162338295e-03 8.11876205861e-03 8.12162338295e-03 8.18384117181e-03 8.42000879273e-03 + 8.04465702446e-03 9.13735254995e-03 1.03423481784e-02 1.16030717026e-02 1.28938940154e-02 1.41150555464e-02 + 1.50739753351e-02 1.55937775675e-02 1.55937775675e-02 1.50739753351e-02 1.41150555464e-02 1.28938940154e-02 + 1.16030717026e-02 1.03423481784e-02 9.13735254995e-03 8.04465702446e-03 7.17868687881e-03 6.62684941427e-03 + 6.36712108007e-03 6.28716491126e-03 6.28716491126e-03 6.36712108007e-03 6.62684941428e-03 7.17868687882e-03 + 8.04465702446e-03 9.57630048387e-03 1.12138690779e-02 1.28581680779e-02 1.43895601245e-02 1.56224103699e-02 + 1.63905844408e-02 1.66438543464e-02 1.63905844408e-02 1.56224103699e-02 1.43895601245e-02 1.28581680779e-02 + 1.12138690779e-02 9.57630048387e-03 8.04465702446e-03 6.75457206647e-03 5.82750449515e-03 5.28637047081e-03 + 5.04613651753e-03 4.98512607635e-03 5.04613651753e-03 5.28637047081e-03 5.82750449515e-03 6.75457206648e-03 + 8.89928200876e-03 1.08501521193e-02 1.28938940154e-02 1.48932680012e-02 1.66389072281e-02 1.78887124933e-02 + 1.85131064743e-02 1.85131064743e-02 1.78887124933e-02 1.66389072281e-02 1.48932680012e-02 1.28938940154e-02 + 1.08501521193e-02 8.89928200875e-03 7.17868687881e-03 5.82750449515e-03 4.90592224711e-03 4.37602725379e-03 + 4.14922255853e-03 4.14922255853e-03 4.37602725379e-03 4.90592224711e-03 5.82750449516e-03 7.17868687882e-03 + 1.06052341643e-02 1.30619533075e-02 1.56490177739e-02 1.81429746336e-02 2.02039597509e-02 2.15194519516e-02 + 2.19609803409e-02 2.15194519516e-02 2.02039597509e-02 1.81429746336e-02 1.56490177739e-02 1.30619533075e-02 + 1.06052341643e-02 8.42000879273e-03 6.62684941427e-03 5.28637047081e-03 4.37602725379e-03 3.84969900803e-03 + 3.67775927312e-03 3.84969900803e-03 4.37602725379e-03 5.28637047082e-03 6.62684941428e-03 8.42000879273e-03 + 1.34421984781e-02 1.66949823920e-02 2.00581600358e-02 2.31229887194e-02 2.54054025138e-02 2.65859681267e-02 + 2.65859681267e-02 2.54054025138e-02 2.31229887194e-02 2.00581600358e-02 1.66949823920e-02 1.34421984781e-02 + 1.05558438840e-02 8.18384117181e-03 6.36712108007e-03 5.04613651753e-03 4.14922255853e-03 3.67775927312e-03 + 3.67775927312e-03 4.14922255853e-03 5.04613651754e-03 6.36712108007e-03 8.18384117181e-03 1.05558438840e-02 + 1.79266369346e-02 2.21707375456e-02 2.61899431515e-02 2.93978765598e-02 3.13659743854e-02 3.20076435874e-02 + 3.13659743854e-02 2.93978765598e-02 2.61899431515e-02 2.21707375456e-02 1.79266369346e-02 1.39668146374e-02 + 1.06519075585e-02 8.12162338295e-03 6.28716491126e-03 4.98512607635e-03 4.14922255853e-03 3.84969900803e-03 + 4.14922255853e-03 4.98512607635e-03 6.28716491126e-03 8.12162338295e-03 1.06519075585e-02 1.39668146374e-02 + 2.40902041017e-02 2.88774182856e-02 3.27027225515e-02 3.51265406940e-02 3.61943858423e-02 3.61943858423e-02 + 3.51265406940e-02 3.27027225515e-02 2.88774182856e-02 2.40902041017e-02 1.90346417875e-02 1.44211937670e-02 + 1.07548084489e-02 8.11876205861e-03 6.28716491126e-03 5.04613651753e-03 4.37602725379e-03 4.37602725379e-03 + 5.04613651753e-03 6.28716491126e-03 8.11876205862e-03 1.07548084489e-02 1.44211937670e-02 1.90346417875e-02 + 3.07818020073e-02 3.49765243617e-02 3.75326780270e-02 3.86301228830e-02 3.88820048627e-02 3.86301228830e-02 + 3.75326780270e-02 3.49765243617e-02 3.07818020073e-02 2.53972544857e-02 1.96803910193e-02 1.45988685534e-02 + 1.07548084489e-02 8.12162338294e-03 6.36712108007e-03 5.28637047081e-03 4.90592224711e-03 5.28637047081e-03 + 6.36712108007e-03 8.12162338295e-03 1.07548084489e-02 1.45988685534e-02 1.96803910193e-02 2.53972544857e-02 + 3.62670291685e-02 3.89797931477e-02 4.00844862497e-02 4.03066875222e-02 4.03066875222e-02 4.00844862497e-02 + 3.89797931477e-02 3.62670291685e-02 3.17358439118e-02 2.58521406843e-02 1.96803910193e-02 1.44211937670e-02 + 1.06519075585e-02 8.18384117181e-03 6.62684941427e-03 5.82750449515e-03 5.82750449515e-03 6.62684941428e-03 + 8.18384117181e-03 1.06519075585e-02 1.44211937670e-02 1.96803910193e-02 2.58521406843e-02 3.17358439118e-02 + 3.97328731661e-02 4.11390439154e-02 4.15214590889e-02 4.15474648474e-02 4.15214590889e-02 4.11390439155e-02 + 3.97328731661e-02 3.66852084288e-02 3.17358439118e-02 2.53972544857e-02 1.90346417875e-02 1.39668146374e-02 + 1.05558438840e-02 8.42000879273e-03 7.17868687882e-03 6.75457206648e-03 7.17868687882e-03 8.42000879273e-03 + 1.05558438840e-02 1.39668146374e-02 1.90346417875e-02 2.53972544857e-02 3.17358439118e-02 3.66852084288e-02 + 6.13627808208e-02 6.87952783261e-02 7.14597362941e-02 6.87952783262e-02 6.13627808208e-02 5.18159940626e-02 + 4.36602554331e-02 3.73533616181e-02 3.09603631544e-02 2.40902041017e-02 1.81617249718e-02 1.40807237278e-02 + 1.16096263393e-02 1.02402112845e-02 9.78230850882e-03 1.02402112845e-02 1.16096263393e-02 1.40807237278e-02 + 1.81617249718e-02 2.40902041017e-02 3.09603631544e-02 3.73533616181e-02 4.36602554331e-02 5.18159940626e-02 + 6.87952783262e-02 7.44580546054e-02 7.44580546054e-02 6.87952783262e-02 5.87813204090e-02 4.86719445336e-02 + 4.14580711769e-02 3.55400935242e-02 2.88774182856e-02 2.22711776717e-02 1.72612102704e-02 1.41150555464e-02 + 1.22935034476e-02 1.14032634249e-02 1.14032634249e-02 1.22935034476e-02 1.41150555464e-02 1.72612102705e-02 + 2.22711776717e-02 2.88774182856e-02 3.55400935242e-02 4.14580711769e-02 4.86719445336e-02 5.87813204090e-02 + 7.14597362941e-02 7.44580546054e-02 7.14597362941e-02 6.30067497701e-02 5.24377104592e-02 4.42642004890e-02 + 3.86294226543e-02 3.27027225515e-02 2.60490352762e-02 2.03920068263e-02 1.66276495780e-02 1.43895601245e-02 + 1.31235074234e-02 1.26855892424e-02 1.31235074234e-02 1.43895601245e-02 1.66276495780e-02 2.03920068264e-02 + 2.60490352762e-02 3.27027225515e-02 3.86294226543e-02 4.42642004890e-02 5.24377104593e-02 6.30067497701e-02 + 6.87952783262e-02 6.87952783262e-02 6.30067497701e-02 5.38559646463e-02 4.56997211399e-02 4.02274592298e-02 + 3.51265406939e-02 2.89589200656e-02 2.30937481115e-02 1.89263306496e-02 1.64062970608e-02 1.48932680012e-02 + 1.40901757966e-02 1.40901757966e-02 1.48932680012e-02 1.64062970608e-02 1.89263306496e-02 2.30937481116e-02 + 2.89589200657e-02 3.51265406940e-02 4.02274592299e-02 4.56997211400e-02 5.38559646463e-02 6.30067497701e-02 + 6.13627808209e-02 5.87813204090e-02 5.24377104592e-02 4.56997211399e-02 4.06895832707e-02 3.61943858422e-02 + 3.06730045455e-02 2.49901046905e-02 2.07195371247e-02 1.81502745082e-02 1.66276495780e-02 1.56490177739e-02 + 1.52718979502e-02 1.56490177739e-02 1.66276495780e-02 1.81502745082e-02 2.07195371247e-02 2.49901046905e-02 + 3.06730045455e-02 3.61943858423e-02 4.06895832708e-02 4.56997211400e-02 5.24377104593e-02 5.87813204090e-02 + 5.18159940627e-02 4.86719445336e-02 4.42642004890e-02 4.02274592299e-02 3.61943858422e-02 3.12137668085e-02 + 2.59180287237e-02 2.17946809311e-02 1.93870166210e-02 1.81179478705e-02 1.72612102705e-02 1.66949823920e-02 + 1.66949823920e-02 1.72612102705e-02 1.81179478705e-02 1.93870166210e-02 2.17946809311e-02 2.59180287237e-02 + 3.12137668085e-02 3.61943858423e-02 4.02274592299e-02 4.42642004890e-02 4.86719445336e-02 5.18159940627e-02 + 4.36602554331e-02 4.14580711769e-02 3.86294226543e-02 3.51265406939e-02 3.06730045455e-02 2.59180287237e-02 + 2.21383854932e-02 1.99974571616e-02 1.91106183945e-02 1.86355831047e-02 1.81617249718e-02 1.79266369346e-02 + 1.81617249718e-02 1.86355831047e-02 1.91106183945e-02 1.99974571616e-02 2.21383854933e-02 2.59180287237e-02 + 3.06730045455e-02 3.51265406940e-02 3.86294226543e-02 4.14580711769e-02 4.36602554331e-02 4.45488243286e-02 + 3.73533616181e-02 3.55400935242e-02 3.27027225515e-02 2.89589200656e-02 2.49901046905e-02 2.17946809311e-02 + 1.99974571616e-02 1.94536912926e-02 1.94291633463e-02 1.92841419762e-02 1.90346417875e-02 1.90346417874e-02 + 1.92841419762e-02 1.94291633463e-02 1.94536912926e-02 1.99974571616e-02 2.17946809311e-02 2.49901046905e-02 + 2.89589200657e-02 3.27027225515e-02 3.55400935242e-02 3.73533616181e-02 3.82416345368e-02 3.82416345368e-02 + 3.09603631544e-02 2.88774182856e-02 2.60490352762e-02 2.30937481115e-02 2.07195371247e-02 1.93870166210e-02 + 1.91106183945e-02 1.94291633463e-02 1.97046043739e-02 1.96803910193e-02 1.96033035629e-02 1.96803910193e-02 + 1.97046043738e-02 1.94291633463e-02 1.91106183945e-02 1.93870166210e-02 2.07195371247e-02 2.30937481115e-02 + 2.60490352762e-02 2.88774182856e-02 3.09603631544e-02 3.20842088246e-02 3.24122722065e-02 3.20842088246e-02 + 2.40902041017e-02 2.22711776717e-02 2.03920068264e-02 1.89263306496e-02 1.81502745082e-02 1.81179478705e-02 + 1.86355831047e-02 1.92841419762e-02 1.96803910193e-02 1.97873550487e-02 1.97873550487e-02 1.96803910193e-02 + 1.92841419762e-02 1.86355831047e-02 1.81179478705e-02 1.81502745082e-02 1.89263306496e-02 2.03920068264e-02 + 2.22711776717e-02 2.40902041017e-02 2.53795329069e-02 2.59776327244e-02 2.59776327244e-02 2.53795329069e-02 + 1.81617249718e-02 1.72612102704e-02 1.66276495780e-02 1.64062970608e-02 1.66276495780e-02 1.72612102705e-02 + 1.81617249718e-02 1.90346417875e-02 1.96033035629e-02 1.97873550487e-02 1.96033035629e-02 1.90346417874e-02 + 1.81617249718e-02 1.72612102705e-02 1.66276495780e-02 1.64062970608e-02 1.66276495780e-02 1.72612102704e-02 + 1.81617249718e-02 1.90346417875e-02 1.96033035629e-02 1.97873550487e-02 1.96033035629e-02 1.90346417875e-02 + 1.40807237278e-02 1.41150555464e-02 1.43895601245e-02 1.48932680012e-02 1.56490177739e-02 1.66949823920e-02 + 1.79266369346e-02 1.90346417875e-02 1.96803910193e-02 1.96803910193e-02 1.90346417875e-02 1.79266369346e-02 + 1.66949823920e-02 1.56490177739e-02 1.48932680012e-02 1.43895601245e-02 1.41150555464e-02 1.40807237278e-02 + 1.42309947893e-02 1.44211937670e-02 1.45323142804e-02 1.45323142804e-02 1.44211937670e-02 1.42309947893e-02 + 1.16096263393e-02 1.22935034476e-02 1.31235074234e-02 1.40901757966e-02 1.52718979502e-02 1.66949823920e-02 + 1.81617249718e-02 1.92841419762e-02 1.97046043738e-02 1.92841419762e-02 1.81617249718e-02 1.66949823920e-02 + 1.52718979502e-02 1.40901757966e-02 1.31235074234e-02 1.22935034476e-02 1.16096263393e-02 1.11332793734e-02 + 1.08700745212e-02 1.07548084489e-02 1.07235160838e-02 1.07548084489e-02 1.08700745212e-02 1.11332793734e-02 + 1.02402112845e-02 1.14032634249e-02 1.26855892423e-02 1.40901757966e-02 1.56490177739e-02 1.72612102704e-02 + 1.86355831047e-02 1.94291633463e-02 1.94291633463e-02 1.86355831047e-02 1.72612102705e-02 1.56490177739e-02 + 1.40901757966e-02 1.26855892423e-02 1.14032634249e-02 1.02402112845e-02 9.28365371522e-03 8.62544633519e-03 + 8.26411545329e-03 8.11876205861e-03 8.11876205861e-03 8.26411545328e-03 8.62544633519e-03 9.28365371522e-03 + 9.78230850882e-03 1.14032634249e-02 1.31235074234e-02 1.48932680012e-02 1.66276495780e-02 1.81179478705e-02 + 1.91106183945e-02 1.94536912925e-02 1.91106183945e-02 1.81179478705e-02 1.66276495780e-02 1.48932680012e-02 + 1.31235074234e-02 1.14032634249e-02 9.78230850881e-03 8.37411495641e-03 7.31939690082e-03 6.67440432828e-03 + 6.36994587767e-03 6.28716491126e-03 6.36994587767e-03 6.67440432828e-03 7.31939690082e-03 8.37411495641e-03 + 1.02402112845e-02 1.22935034476e-02 1.43895601245e-02 1.64062970608e-02 1.81502745082e-02 1.93870166210e-02 + 1.99974571616e-02 1.99974571616e-02 1.93870166210e-02 1.81502745082e-02 1.64062970608e-02 1.43895601245e-02 + 1.22935034476e-02 1.02402112845e-02 8.37411495641e-03 6.87435772715e-03 5.85081359877e-03 5.27979197805e-03 + 5.04613651753e-03 5.04613651753e-03 5.27979197805e-03 5.85081359877e-03 6.87435772716e-03 8.37411495641e-03 + 1.16096263393e-02 1.41150555464e-02 1.66276495780e-02 1.89263306496e-02 2.07195371247e-02 2.17946809311e-02 + 2.21383854932e-02 2.17946809311e-02 2.07195371247e-02 1.89263306496e-02 1.66276495780e-02 1.41150555464e-02 + 1.16096263393e-02 9.28365371521e-03 7.31939690081e-03 5.85081359876e-03 4.89322201917e-03 4.37602725379e-03 + 4.21587397243e-03 4.37602725379e-03 4.89322201917e-03 5.85081359877e-03 7.31939690082e-03 9.28365371522e-03 + 1.40807237278e-02 1.72612102705e-02 2.03920068264e-02 2.30937481116e-02 2.49901046905e-02 2.59180287237e-02 + 2.59180287237e-02 2.49901046905e-02 2.30937481116e-02 2.03920068264e-02 1.72612102704e-02 1.40807237278e-02 + 1.11332793734e-02 8.62544633519e-03 6.67440432828e-03 5.27979197805e-03 4.37602725379e-03 3.92286252630e-03 + 3.92286252630e-03 4.37602725379e-03 5.27979197806e-03 6.67440432828e-03 8.62544633520e-03 1.11332793734e-02 + 1.81617249718e-02 2.22711776717e-02 2.60490352763e-02 2.89589200657e-02 3.06730045455e-02 3.12137668085e-02 + 3.06730045455e-02 2.89589200657e-02 2.60490352762e-02 2.22711776717e-02 1.81617249718e-02 1.42309947893e-02 + 1.08700745212e-02 8.26411545328e-03 6.36994587766e-03 5.04613651753e-03 4.21587397243e-03 3.92286252630e-03 + 4.21587397243e-03 5.04613651753e-03 6.36994587767e-03 8.26411545329e-03 1.08700745212e-02 1.42309947893e-02 + 2.40902041017e-02 2.88774182856e-02 3.27027225515e-02 3.51265406940e-02 3.61943858423e-02 3.61943858423e-02 + 3.51265406940e-02 3.27027225515e-02 2.88774182856e-02 2.40902041017e-02 1.90346417875e-02 1.44211937670e-02 + 1.07548084489e-02 8.11876205861e-03 6.28716491126e-03 5.04613651753e-03 4.37602725379e-03 4.37602725379e-03 + 5.04613651753e-03 6.28716491126e-03 8.11876205861e-03 1.07548084489e-02 1.44211937670e-02 1.90346417875e-02 + 3.09603631544e-02 3.55400935242e-02 3.86294226544e-02 4.02274592299e-02 4.06895832708e-02 4.02274592299e-02 + 3.86294226544e-02 3.55400935242e-02 3.09603631544e-02 2.53795329069e-02 1.96033035629e-02 1.45323142804e-02 + 1.07235160838e-02 8.11876205861e-03 6.36994587766e-03 5.27979197805e-03 4.89322201917e-03 5.27979197806e-03 + 6.36994587767e-03 8.11876205861e-03 1.07235160838e-02 1.45323142804e-02 1.96033035629e-02 2.53795329070e-02 + 3.73533616181e-02 4.14580711769e-02 4.42642004890e-02 4.56997211400e-02 4.56997211400e-02 4.42642004890e-02 + 4.14580711769e-02 3.73533616181e-02 3.20842088246e-02 2.59776327244e-02 1.97873550487e-02 1.45323142804e-02 + 1.07548084489e-02 8.26411545328e-03 6.67440432828e-03 5.85081359877e-03 5.85081359877e-03 6.67440432828e-03 + 8.26411545329e-03 1.07548084489e-02 1.45323142804e-02 1.97873550487e-02 2.59776327244e-02 3.20842088246e-02 + 4.36602554331e-02 4.86719445336e-02 5.24377104592e-02 5.38559646463e-02 5.24377104593e-02 4.86719445336e-02 + 4.36602554331e-02 3.82416345368e-02 3.24122722065e-02 2.59776327244e-02 1.96033035629e-02 1.44211937670e-02 + 1.08700745212e-02 8.62544633519e-03 7.31939690082e-03 6.87435772716e-03 7.31939690082e-03 8.62544633520e-03 + 1.08700745212e-02 1.44211937670e-02 1.96033035629e-02 2.59776327244e-02 3.24122722065e-02 3.82416345368e-02 + 5.18159940626e-02 5.87813204090e-02 6.30067497701e-02 6.30067497701e-02 5.87813204090e-02 5.18159940627e-02 + 4.45488243286e-02 3.82416345368e-02 3.20842088246e-02 2.53795329070e-02 1.90346417875e-02 1.42309947893e-02 + 1.11332793734e-02 9.28365371522e-03 8.37411495641e-03 8.37411495641e-03 9.28365371522e-03 1.11332793734e-02 + 1.42309947893e-02 1.90346417875e-02 2.53795329070e-02 3.20842088246e-02 3.82416345368e-02 4.45488243286e-02 + 9.46627422161e-02 1.02256358961e-01 1.02256358961e-01 9.46627422161e-02 7.81199920259e-02 5.87904239654e-02 + 4.51573443066e-02 3.73533616181e-02 3.07818020073e-02 2.40966608398e-02 1.86355831047e-02 1.50739753351e-02 + 1.30486211596e-02 1.21045641480e-02 1.21045641480e-02 1.30486211596e-02 1.50739753351e-02 1.86355831047e-02 + 2.40966608398e-02 3.07818020073e-02 3.73533616181e-02 4.51573443066e-02 5.87904239654e-02 7.81199920259e-02 + 1.02256358961e-01 1.05383989813e-01 1.02256358961e-01 9.00462898607e-02 6.98261770599e-02 5.17384028835e-02 + 4.14580711769e-02 3.49765243617e-02 2.84215335265e-02 2.23752062804e-02 1.81179478705e-02 1.56224103699e-02 + 1.43188156133e-02 1.38961510916e-02 1.43188156133e-02 1.56224103699e-02 1.81179478705e-02 2.23752062804e-02 + 2.84215335265e-02 3.49765243617e-02 4.14580711769e-02 5.17384028835e-02 6.98261770600e-02 9.00462898608e-02 + 1.02256358961e-01 1.02256358961e-01 9.40547983345e-02 7.65079897135e-02 5.69445178736e-02 4.42642004890e-02 + 3.75326780270e-02 3.16055353689e-02 2.55724702285e-02 2.09446150107e-02 1.81502745082e-02 1.66389072281e-02 + 1.59286730225e-02 1.59286730225e-02 1.66389072281e-02 1.81502745082e-02 2.09446150107e-02 2.55724702285e-02 + 3.16055353689e-02 3.75326780270e-02 4.42642004890e-02 5.69445178737e-02 7.65079897135e-02 9.40547983346e-02 + 9.46627422162e-02 9.00462898607e-02 7.65079897135e-02 5.89470993594e-02 4.56997211399e-02 3.86301228830e-02 + 3.33648425996e-02 2.78012163806e-02 2.32028796117e-02 2.03982633679e-02 1.89263306496e-02 1.81429746336e-02 + 1.78656633258e-02 1.81429746336e-02 1.89263306496e-02 2.03982633679e-02 2.32028796117e-02 2.78012163806e-02 + 3.33648425996e-02 3.86301228830e-02 4.56997211400e-02 5.89470993595e-02 7.65079897136e-02 9.00462898608e-02 + 7.81199920259e-02 6.98261770599e-02 5.69445178736e-02 4.56997211399e-02 3.88820048627e-02 3.39946385756e-02 + 2.89472257765e-02 2.46284218885e-02 2.20776282425e-02 2.09446150107e-02 2.03920068264e-02 2.00581600358e-02 + 2.00581600358e-02 2.03920068264e-02 2.09446150107e-02 2.20776282425e-02 2.46284218885e-02 2.89472257765e-02 + 3.39946385756e-02 3.88820048628e-02 4.56997211400e-02 5.69445178737e-02 6.98261770600e-02 7.81199920260e-02 + 5.87904239654e-02 5.17384028835e-02 4.42642004890e-02 3.86301228830e-02 3.39946385756e-02 2.92642577664e-02 + 2.52442663181e-02 2.30251854758e-02 2.24043544384e-02 2.23752062804e-02 2.22711776717e-02 2.21707375456e-02 + 2.22711776717e-02 2.23752062804e-02 2.24043544384e-02 2.30251854758e-02 2.52442663181e-02 2.92642577664e-02 + 3.39946385756e-02 3.86301228830e-02 4.42642004891e-02 5.17384028836e-02 5.87904239654e-02 6.17010347926e-02 + 4.51573443066e-02 4.14580711769e-02 3.75326780270e-02 3.33648425996e-02 2.89472257765e-02 2.52442663181e-02 + 2.33116271285e-02 2.31387782150e-02 2.37302554132e-02 2.40966608398e-02 2.40902041017e-02 2.40902041017e-02 + 2.40966608398e-02 2.37302554132e-02 2.31387782150e-02 2.33116271285e-02 2.52442663181e-02 2.89472257765e-02 + 3.33648425996e-02 3.75326780270e-02 4.14580711769e-02 4.51573443066e-02 4.76129372810e-02 4.76129372810e-02 + 3.73533616181e-02 3.49765243617e-02 3.16055353689e-02 2.78012163806e-02 2.46284218885e-02 2.30251854758e-02 + 2.31387782150e-02 2.42076546579e-02 2.51250422144e-02 2.53972544857e-02 2.53795329069e-02 2.53972544857e-02 + 2.51250422144e-02 2.42076546579e-02 2.31387782150e-02 2.30251854758e-02 2.46284218885e-02 2.78012163806e-02 + 3.16055353689e-02 3.49765243617e-02 3.73533616181e-02 3.86928693264e-02 3.91195978594e-02 3.86928693264e-02 + 3.07818020073e-02 2.84215335265e-02 2.55724702285e-02 2.32028796117e-02 2.20776282425e-02 2.24043544384e-02 + 2.37302554132e-02 2.51250422144e-02 2.58521406843e-02 2.59776327244e-02 2.59776327244e-02 2.58521406843e-02 + 2.51250422144e-02 2.37302554132e-02 2.24043544384e-02 2.20776282425e-02 2.32028796117e-02 2.55724702285e-02 + 2.84215335265e-02 3.07818020073e-02 3.20842088246e-02 3.25035303326e-02 3.25035303326e-02 3.20842088246e-02 + 2.40966608398e-02 2.23752062804e-02 2.09446150107e-02 2.03982633679e-02 2.09446150107e-02 2.23752062804e-02 + 2.40966608398e-02 2.53972544857e-02 2.59776327244e-02 2.60946322497e-02 2.59776327244e-02 2.53972544857e-02 + 2.40966608398e-02 2.23752062804e-02 2.09446150107e-02 2.03982633679e-02 2.09446150107e-02 2.23752062804e-02 + 2.40966608398e-02 2.53972544857e-02 2.59776327244e-02 2.60946322497e-02 2.59776327244e-02 2.53972544857e-02 + 1.86355831047e-02 1.81179478705e-02 1.81502745082e-02 1.89263306496e-02 2.03920068264e-02 2.22711776717e-02 + 2.40902041017e-02 2.53795329069e-02 2.59776327244e-02 2.59776327244e-02 2.53795329069e-02 2.40902041017e-02 + 2.22711776717e-02 2.03920068264e-02 1.89263306496e-02 1.81502745082e-02 1.81179478705e-02 1.86355831047e-02 + 1.92841419762e-02 1.96803910193e-02 1.97873550487e-02 1.97873550487e-02 1.96803910193e-02 1.92841419762e-02 + 1.50739753351e-02 1.56224103699e-02 1.66389072281e-02 1.81429746336e-02 2.00581600358e-02 2.21707375455e-02 + 2.40902041017e-02 2.53972544857e-02 2.58521406843e-02 2.53972544857e-02 2.40902041017e-02 2.21707375456e-02 + 2.00581600358e-02 1.81429746336e-02 1.66389072281e-02 1.56224103699e-02 1.50739753351e-02 1.48493143122e-02 + 1.47280857174e-02 1.45988685534e-02 1.45323142804e-02 1.45988685534e-02 1.47280857174e-02 1.48493143122e-02 + 1.30486211596e-02 1.43188156133e-02 1.59286730225e-02 1.78656633258e-02 2.00581600358e-02 2.22711776717e-02 + 2.40966608398e-02 2.51250422144e-02 2.51250422144e-02 2.40966608398e-02 2.22711776717e-02 2.00581600358e-02 + 1.78656633258e-02 1.59286730225e-02 1.43188156133e-02 1.30486211596e-02 1.21091269494e-02 1.14482068790e-02 + 1.09991800129e-02 1.07548084489e-02 1.07548084489e-02 1.09991800129e-02 1.14482068790e-02 1.21091269494e-02 + 1.21045641480e-02 1.38961510916e-02 1.59286730224e-02 1.81429746336e-02 2.03920068264e-02 2.23752062804e-02 + 2.37302554132e-02 2.42076546579e-02 2.37302554132e-02 2.23752062804e-02 2.03920068264e-02 1.81429746336e-02 + 1.59286730225e-02 1.38961510916e-02 1.21045641480e-02 1.06110221336e-02 9.47298406612e-03 8.70260576990e-03 + 8.26411545329e-03 8.12162338295e-03 8.26411545329e-03 8.70260576990e-03 9.47298406612e-03 1.06110221336e-02 + 1.21045641480e-02 1.43188156133e-02 1.66389072281e-02 1.89263306496e-02 2.09446150107e-02 2.24043544384e-02 + 2.31387782150e-02 2.31387782150e-02 2.24043544384e-02 2.09446150107e-02 1.89263306496e-02 1.66389072281e-02 + 1.43188156133e-02 1.21045641480e-02 1.01199573737e-02 8.50676845052e-03 7.36059222257e-03 6.67440432828e-03 + 6.36712108007e-03 6.36712108007e-03 6.67440432828e-03 7.36059222257e-03 8.50676845052e-03 1.01199573737e-02 + 1.30486211596e-02 1.56224103699e-02 1.81502745082e-02 2.03982633679e-02 2.20776282425e-02 2.30251854758e-02 + 2.33116271285e-02 2.30251854758e-02 2.20776282425e-02 2.03982633679e-02 1.81502745082e-02 1.56224103699e-02 + 1.30486211596e-02 1.06110221336e-02 8.50676845052e-03 6.90318266868e-03 5.85081359877e-03 5.28637047081e-03 + 5.11302992589e-03 5.28637047081e-03 5.85081359877e-03 6.90318266868e-03 8.50676845052e-03 1.06110221336e-02 + 1.50739753351e-02 1.81179478705e-02 2.09446150107e-02 2.32028796117e-02 2.46284218885e-02 2.52442663181e-02 + 2.52442663181e-02 2.46284218885e-02 2.32028796117e-02 2.09446150107e-02 1.81179478705e-02 1.50739753351e-02 + 1.21091269494e-02 9.47298406612e-03 7.36059222257e-03 5.85081359877e-03 4.90592224711e-03 4.45369745770e-03 + 4.45369745770e-03 4.90592224711e-03 5.85081359877e-03 7.36059222257e-03 9.47298406612e-03 1.21091269494e-02 + 1.86355831047e-02 2.23752062804e-02 2.55724702285e-02 2.78012163806e-02 2.89472257765e-02 2.92642577664e-02 + 2.89472257765e-02 2.78012163806e-02 2.55724702285e-02 2.23752062804e-02 1.86355831047e-02 1.48493143122e-02 + 1.14482068790e-02 8.70260576990e-03 6.67440432828e-03 5.28637047081e-03 4.45369745770e-03 4.16949135529e-03 + 4.45369745770e-03 5.28637047081e-03 6.67440432828e-03 8.70260576991e-03 1.14482068790e-02 1.48493143122e-02 + 2.40966608398e-02 2.84215335265e-02 3.16055353689e-02 3.33648425996e-02 3.39946385756e-02 3.39946385756e-02 + 3.33648425996e-02 3.16055353689e-02 2.84215335265e-02 2.40966608398e-02 1.92841419762e-02 1.47280857174e-02 + 1.09991800129e-02 8.26411545328e-03 6.36712108007e-03 5.11302992589e-03 4.45369745770e-03 4.45369745770e-03 + 5.11302992589e-03 6.36712108007e-03 8.26411545329e-03 1.09991800129e-02 1.47280857174e-02 1.92841419762e-02 + 3.07818020073e-02 3.49765243617e-02 3.75326780270e-02 3.86301228830e-02 3.88820048628e-02 3.86301228830e-02 + 3.75326780270e-02 3.49765243617e-02 3.07818020073e-02 2.53972544857e-02 1.96803910193e-02 1.45988685534e-02 + 1.07548084489e-02 8.12162338295e-03 6.36712108007e-03 5.28637047081e-03 4.90592224711e-03 5.28637047081e-03 + 6.36712108007e-03 8.12162338295e-03 1.07548084489e-02 1.45988685534e-02 1.96803910193e-02 2.53972544857e-02 + 3.73533616181e-02 4.14580711769e-02 4.42642004890e-02 4.56997211400e-02 4.56997211400e-02 4.42642004891e-02 + 4.14580711769e-02 3.73533616181e-02 3.20842088246e-02 2.59776327244e-02 1.97873550487e-02 1.45323142804e-02 + 1.07548084489e-02 8.26411545328e-03 6.67440432828e-03 5.85081359876e-03 5.85081359877e-03 6.67440432828e-03 + 8.26411545329e-03 1.07548084489e-02 1.45323142804e-02 1.97873550487e-02 2.59776327244e-02 3.20842088246e-02 + 4.51573443066e-02 5.17384028835e-02 5.69445178737e-02 5.89470993595e-02 5.69445178737e-02 5.17384028836e-02 + 4.51573443066e-02 3.86928693264e-02 3.25035303326e-02 2.60946322497e-02 1.97873550487e-02 1.45988685534e-02 + 1.09991800129e-02 8.70260576990e-03 7.36059222257e-03 6.90318266868e-03 7.36059222257e-03 8.70260576991e-03 + 1.09991800129e-02 1.45988685534e-02 1.97873550487e-02 2.60946322497e-02 3.25035303326e-02 3.86928693264e-02 + 5.87904239654e-02 6.98261770599e-02 7.65079897135e-02 7.65079897135e-02 6.98261770600e-02 5.87904239654e-02 + 4.76129372810e-02 3.91195978595e-02 3.25035303326e-02 2.59776327244e-02 1.96803910193e-02 1.47280857174e-02 + 1.14482068790e-02 9.47298406612e-03 8.50676845052e-03 8.50676845052e-03 9.47298406613e-03 1.14482068790e-02 + 1.47280857174e-02 1.96803910193e-02 2.59776327244e-02 3.25035303326e-02 3.91195978595e-02 4.76129372810e-02 + 7.81199920259e-02 9.00462898608e-02 9.40547983346e-02 9.00462898608e-02 7.81199920260e-02 6.17010347926e-02 + 4.76129372810e-02 3.86928693264e-02 3.20842088246e-02 2.53972544857e-02 1.92841419762e-02 1.48493143122e-02 + 1.21091269494e-02 1.06110221336e-02 1.01199573737e-02 1.06110221336e-02 1.21091269495e-02 1.48493143122e-02 + 1.92841419762e-02 2.53972544857e-02 3.20842088246e-02 3.86928693264e-02 4.76129372810e-02 6.17010347926e-02 + 1.14984832163e-01 1.14285203523e-01 1.14984832163e-01 1.07505424132e-01 8.44125475990e-02 5.87904239654e-02 + 4.36602554331e-02 3.62670291685e-02 2.99300705585e-02 2.37302554132e-02 1.91106183945e-02 1.63905844408e-02 + 1.50423980149e-02 1.46288065643e-02 1.50423980149e-02 1.63905844408e-02 1.91106183945e-02 2.37302554132e-02 + 2.99300705585e-02 3.62670291685e-02 4.36602554332e-02 5.87904239654e-02 8.44125475991e-02 1.07505424132e-01 + 1.14285203523e-01 1.14285203523e-01 1.13480556613e-01 9.74791227832e-02 6.98261770599e-02 4.86719445336e-02 + 3.89797931477e-02 3.31962688978e-02 2.73091130234e-02 2.24043544384e-02 1.93870166210e-02 1.78887124933e-02 + 1.72810954144e-02 1.72810954144e-02 1.78887124933e-02 1.93870166210e-02 2.24043544384e-02 2.73091130234e-02 + 3.31962688978e-02 3.89797931478e-02 4.86719445337e-02 6.98261770600e-02 9.74791227833e-02 1.13480556613e-01 + 1.14984832163e-01 1.13480556613e-01 1.01689797158e-01 7.65079897134e-02 5.24377104592e-02 4.00844862497e-02 + 3.45751721122e-02 2.95889063006e-02 2.49962561202e-02 2.20776282425e-02 2.07195371247e-02 2.02039597509e-02 + 2.00652756289e-02 2.02039597509e-02 2.07195371247e-02 2.20776282425e-02 2.49962561202e-02 2.95889063006e-02 + 3.45751721123e-02 4.00844862497e-02 5.24377104593e-02 7.65079897136e-02 1.01689797158e-01 1.13480556613e-01 + 1.07505424132e-01 9.74791227832e-02 7.65079897134e-02 5.38559646463e-02 4.03066875222e-02 3.46097554135e-02 + 3.04771749334e-02 2.65397992490e-02 2.40837535674e-02 2.32028796117e-02 2.30937481115e-02 2.31229887194e-02 + 2.31229887194e-02 2.30937481116e-02 2.32028796117e-02 2.40837535674e-02 2.65397992490e-02 3.04771749334e-02 + 3.46097554135e-02 4.03066875223e-02 5.38559646464e-02 7.65079897136e-02 9.74791227833e-02 1.07505424132e-01 + 8.44125475990e-02 6.98261770599e-02 5.24377104592e-02 4.03066875222e-02 3.44218462469e-02 3.05546433144e-02 + 2.71139620581e-02 2.51928884334e-02 2.49962561202e-02 2.55724702285e-02 2.60490352762e-02 2.61899431515e-02 + 2.60490352762e-02 2.55724702285e-02 2.49962561202e-02 2.51928884334e-02 2.71139620581e-02 3.05546433144e-02 + 3.44218462469e-02 4.03066875223e-02 5.24377104593e-02 6.98261770600e-02 8.44125475991e-02 8.97670392509e-02 + 5.87904239654e-02 4.86719445336e-02 4.00844862497e-02 3.46097554135e-02 3.05546433144e-02 2.72047829307e-02 + 2.55789017267e-02 2.59617964748e-02 2.73091130234e-02 2.84215335265e-02 2.88774182856e-02 2.88774182856e-02 + 2.84215335265e-02 2.73091130234e-02 2.59617964748e-02 2.55789017267e-02 2.72047829307e-02 3.05546433144e-02 + 3.46097554135e-02 4.00844862497e-02 4.86719445337e-02 5.87904239654e-02 6.56429700805e-02 6.56429700804e-02 + 4.36602554331e-02 3.89797931477e-02 3.45751721123e-02 3.04771749334e-02 2.71139620581e-02 2.55789017267e-02 + 2.62414574798e-02 2.81654297894e-02 2.99300705585e-02 3.07818020073e-02 3.09603631544e-02 3.07818020073e-02 + 2.99300705585e-02 2.81654297894e-02 2.62414574798e-02 2.55789017267e-02 2.71139620581e-02 3.04771749334e-02 + 3.45751721123e-02 3.89797931478e-02 4.36602554332e-02 4.76129372810e-02 4.91898793330e-02 4.76129372810e-02 + 3.62670291685e-02 3.31962688978e-02 2.95889063006e-02 2.65397992490e-02 2.51928884334e-02 2.59617964748e-02 + 2.81654297894e-02 3.04405035973e-02 3.17358439118e-02 3.20842088246e-02 3.20842088246e-02 3.17358439118e-02 + 3.04405035973e-02 2.81654297894e-02 2.59617964748e-02 2.51928884334e-02 2.65397992490e-02 2.95889063006e-02 + 3.31962688978e-02 3.62670291685e-02 3.82416345368e-02 3.91195978594e-02 3.91195978595e-02 3.82416345368e-02 + 2.99300705585e-02 2.73091130234e-02 2.49962561202e-02 2.40837535673e-02 2.49962561202e-02 2.73091130234e-02 + 2.99300705585e-02 3.17358439118e-02 3.24122722065e-02 3.25035303326e-02 3.24122722064e-02 3.17358439118e-02 + 2.99300705585e-02 2.73091130234e-02 2.49962561202e-02 2.40837535673e-02 2.49962561202e-02 2.73091130234e-02 + 2.99300705585e-02 3.17358439118e-02 3.24122722064e-02 3.25035303326e-02 3.24122722065e-02 3.17358439118e-02 + 2.37302554132e-02 2.24043544384e-02 2.20776282425e-02 2.32028796117e-02 2.55724702285e-02 2.84215335265e-02 + 3.07818020073e-02 3.20842088246e-02 3.25035303326e-02 3.25035303326e-02 3.20842088246e-02 3.07818020073e-02 + 2.84215335265e-02 2.55724702285e-02 2.32028796117e-02 2.20776282425e-02 2.24043544384e-02 2.37302554132e-02 + 2.51250422144e-02 2.58521406843e-02 2.59776327244e-02 2.59776327244e-02 2.58521406843e-02 2.51250422144e-02 + 1.91106183945e-02 1.93870166210e-02 2.07195371247e-02 2.30937481115e-02 2.60490352762e-02 2.88774182856e-02 + 3.09603631544e-02 3.20842088246e-02 3.24122722065e-02 3.20842088246e-02 3.09603631544e-02 2.88774182856e-02 + 2.60490352762e-02 2.30937481116e-02 2.07195371247e-02 1.93870166210e-02 1.91106183945e-02 1.94291633463e-02 + 1.97046043738e-02 1.96803910193e-02 1.96033035629e-02 1.96803910193e-02 1.97046043739e-02 1.94291633463e-02 + 1.63905844408e-02 1.78887124933e-02 2.02039597509e-02 2.31229887194e-02 2.61899431515e-02 2.88774182856e-02 + 3.07818020073e-02 3.17358439118e-02 3.17358439118e-02 3.07818020073e-02 2.88774182856e-02 2.61899431515e-02 + 2.31229887194e-02 2.02039597509e-02 1.78887124933e-02 1.63905844408e-02 1.55937775675e-02 1.51356003261e-02 + 1.47280857174e-02 1.44211937670e-02 1.44211937670e-02 1.47280857174e-02 1.51356003261e-02 1.55937775675e-02 + 1.50423980149e-02 1.72810954144e-02 2.00652756288e-02 2.31229887194e-02 2.60490352762e-02 2.84215335265e-02 + 2.99300705585e-02 3.04405035973e-02 2.99300705585e-02 2.84215335265e-02 2.60490352762e-02 2.31229887194e-02 + 2.00652756289e-02 1.72810954144e-02 1.50423980149e-02 1.34169422858e-02 1.22801184032e-02 1.14482068790e-02 + 1.08700745212e-02 1.06519075585e-02 1.08700745212e-02 1.14482068790e-02 1.22801184033e-02 1.34169422858e-02 + 1.46288065643e-02 1.72810954144e-02 2.02039597509e-02 2.30937481115e-02 2.55724702285e-02 2.73091130234e-02 + 2.81654297894e-02 2.81654297894e-02 2.73091130234e-02 2.55724702285e-02 2.30937481116e-02 2.02039597509e-02 + 1.72810954144e-02 1.46288065643e-02 1.24266085814e-02 1.07226215343e-02 9.47298406612e-03 8.62544633519e-03 + 8.18384117181e-03 8.18384117181e-03 8.62544633519e-03 9.47298406612e-03 1.07226215343e-02 1.24266085814e-02 + 1.50423980149e-02 1.78887124933e-02 2.07195371247e-02 2.32028796117e-02 2.49962561202e-02 2.59617964748e-02 + 2.62414574798e-02 2.59617964748e-02 2.49962561202e-02 2.32028796117e-02 2.07195371247e-02 1.78887124933e-02 + 1.50423980149e-02 1.24266085814e-02 1.02154655471e-02 8.50676845052e-03 7.31939690082e-03 6.62684941428e-03 + 6.39966515661e-03 6.62684941427e-03 7.31939690081e-03 8.50676845052e-03 1.02154655471e-02 1.24266085814e-02 + 1.63905844408e-02 1.93870166210e-02 2.20776282425e-02 2.40837535674e-02 2.51928884334e-02 2.55789017267e-02 + 2.55789017267e-02 2.51928884334e-02 2.40837535674e-02 2.20776282425e-02 1.93870166210e-02 1.63905844408e-02 + 1.34169422858e-02 1.07226215343e-02 8.50676845052e-03 6.87435772715e-03 5.82750449515e-03 5.32069114083e-03 + 5.32069114083e-03 5.82750449515e-03 6.87435772715e-03 8.50676845052e-03 1.07226215343e-02 1.34169422858e-02 + 1.91106183945e-02 2.24043544384e-02 2.49962561202e-02 2.65397992490e-02 2.71139620581e-02 2.72047829307e-02 + 2.71139620581e-02 2.65397992490e-02 2.49962561202e-02 2.24043544384e-02 1.91106183945e-02 1.55937775675e-02 + 1.22801184032e-02 9.47298406612e-03 7.31939690081e-03 5.82750449515e-03 4.94634941966e-03 4.65178002992e-03 + 4.94634941966e-03 5.82750449515e-03 7.31939690082e-03 9.47298406613e-03 1.22801184033e-02 1.55937775675e-02 + 2.37302554132e-02 2.73091130234e-02 2.95889063006e-02 3.04771749334e-02 3.05546433144e-02 3.05546433144e-02 + 3.04771749334e-02 2.95889063006e-02 2.73091130234e-02 2.37302554132e-02 1.94291633463e-02 1.51356003261e-02 + 1.14482068790e-02 8.62544633519e-03 6.62684941427e-03 5.32069114083e-03 4.65178002992e-03 4.65178002992e-03 + 5.32069114083e-03 6.62684941428e-03 8.62544633520e-03 1.14482068791e-02 1.51356003261e-02 1.94291633463e-02 + 2.99300705586e-02 3.31962688978e-02 3.45751721123e-02 3.46097554135e-02 3.44218462469e-02 3.46097554135e-02 + 3.45751721123e-02 3.31962688978e-02 2.99300705585e-02 2.51250422144e-02 1.97046043738e-02 1.47280857174e-02 + 1.08700745212e-02 8.18384117181e-03 6.39966515661e-03 5.32069114083e-03 4.94634941966e-03 5.32069114083e-03 + 6.39966515662e-03 8.18384117181e-03 1.08700745212e-02 1.47280857174e-02 1.97046043739e-02 2.51250422144e-02 + 3.62670291685e-02 3.89797931478e-02 4.00844862497e-02 4.03066875223e-02 4.03066875223e-02 4.00844862497e-02 + 3.89797931478e-02 3.62670291685e-02 3.17358439118e-02 2.58521406843e-02 1.96803910193e-02 1.44211937670e-02 + 1.06519075585e-02 8.18384117181e-03 6.62684941427e-03 5.82750449515e-03 5.82750449515e-03 6.62684941428e-03 + 8.18384117181e-03 1.06519075585e-02 1.44211937670e-02 1.96803910193e-02 2.58521406843e-02 3.17358439118e-02 + 4.36602554332e-02 4.86719445337e-02 5.24377104593e-02 5.38559646464e-02 5.24377104593e-02 4.86719445337e-02 + 4.36602554332e-02 3.82416345368e-02 3.24122722064e-02 2.59776327244e-02 1.96033035629e-02 1.44211937670e-02 + 1.08700745212e-02 8.62544633519e-03 7.31939690081e-03 6.87435772715e-03 7.31939690082e-03 8.62544633520e-03 + 1.08700745212e-02 1.44211937670e-02 1.96033035629e-02 2.59776327244e-02 3.24122722065e-02 3.82416345368e-02 + 5.87904239654e-02 6.98261770600e-02 7.65079897136e-02 7.65079897136e-02 6.98261770600e-02 5.87904239654e-02 + 4.76129372810e-02 3.91195978595e-02 3.25035303326e-02 2.59776327244e-02 1.96803910193e-02 1.47280857174e-02 + 1.14482068790e-02 9.47298406612e-03 8.50676845052e-03 8.50676845052e-03 9.47298406613e-03 1.14482068791e-02 + 1.47280857174e-02 1.96803910193e-02 2.59776327244e-02 3.25035303326e-02 3.91195978595e-02 4.76129372810e-02 + 8.44125475991e-02 9.74791227833e-02 1.01689797158e-01 9.74791227833e-02 8.44125475991e-02 6.56429700805e-02 + 4.91898793330e-02 3.91195978595e-02 3.24122722065e-02 2.58521406843e-02 1.97046043739e-02 1.51356003261e-02 + 1.22801184033e-02 1.07226215343e-02 1.02154655471e-02 1.07226215343e-02 1.22801184033e-02 1.51356003261e-02 + 1.97046043739e-02 2.58521406843e-02 3.24122722065e-02 3.91195978595e-02 4.91898793330e-02 6.56429700805e-02 + 1.07505424132e-01 1.13480556613e-01 1.13480556613e-01 1.07505424132e-01 8.97670392509e-02 6.56429700805e-02 + 4.76129372810e-02 3.82416345368e-02 3.17358439118e-02 2.51250422144e-02 1.94291633463e-02 1.55937775675e-02 + 1.34169422858e-02 1.24266085814e-02 1.24266085814e-02 1.34169422858e-02 1.55937775675e-02 1.94291633463e-02 + 2.51250422144e-02 3.17358439118e-02 3.82416345368e-02 4.76129372810e-02 6.56429700805e-02 8.97670392509e-02 + 1.05270975780e-01 1.05270975780e-01 1.15477102966e-01 1.07505424132e-01 7.81199920259e-02 5.18159940627e-02 + 3.97328731661e-02 3.38926655663e-02 2.81654297894e-02 2.31387782150e-02 1.99974571616e-02 1.85131064743e-02 + 1.79740052840e-02 1.79740052840e-02 1.85131064743e-02 1.99974571616e-02 2.31387782150e-02 2.81654297894e-02 + 3.38926655663e-02 3.97328731662e-02 5.18159940627e-02 7.81199920260e-02 1.07505424132e-01 1.15477102967e-01 + 1.05270975780e-01 1.12133532869e-01 1.13480556613e-01 9.00462898607e-02 5.87813204090e-02 4.11390439155e-02 + 3.48629669819e-02 3.04051608500e-02 2.59617964748e-02 2.30251854758e-02 2.17946809311e-02 2.15194519516e-02 + 2.15037112468e-02 2.15194519516e-02 2.17946809311e-02 2.30251854758e-02 2.59617964748e-02 3.04051608500e-02 + 3.48629669819e-02 4.11390439155e-02 5.87813204091e-02 9.00462898609e-02 1.13480556613e-01 1.12133532869e-01 + 1.15477102966e-01 1.13480556613e-01 9.40547983345e-02 6.30067497701e-02 4.15214590889e-02 3.39197128099e-02 + 3.06791433435e-02 2.73901656280e-02 2.51928884334e-02 2.46284218885e-02 2.49901046905e-02 2.54054025138e-02 + 2.54054025138e-02 2.49901046905e-02 2.46284218885e-02 2.51928884334e-02 2.73901656281e-02 3.06791433435e-02 + 3.39197128100e-02 4.15214590889e-02 6.30067497702e-02 9.40547983346e-02 1.13480556613e-01 1.15477102967e-01 + 1.07505424132e-01 9.00462898607e-02 6.30067497700e-02 4.15474648474e-02 3.26919751552e-02 2.97972897261e-02 + 2.74766558578e-02 2.61753235783e-02 2.65397992490e-02 2.78012163806e-02 2.89589200656e-02 2.93978765598e-02 + 2.89589200657e-02 2.78012163806e-02 2.65397992490e-02 2.61753235783e-02 2.74766558578e-02 2.97972897261e-02 + 3.26919751552e-02 4.15474648475e-02 6.30067497702e-02 9.00462898609e-02 1.07505424132e-01 1.12226003113e-01 + 7.81199920259e-02 5.87813204090e-02 4.15214590889e-02 3.26919751552e-02 2.92434029241e-02 2.70509290066e-02 + 2.62349275771e-02 2.73901656280e-02 2.95889063006e-02 3.16055353689e-02 3.27027225515e-02 3.27027225515e-02 + 3.16055353689e-02 2.95889063006e-02 2.73901656281e-02 2.62349275771e-02 2.70509290066e-02 2.92434029240e-02 + 3.26919751552e-02 4.15214590889e-02 5.87813204091e-02 7.81199920260e-02 8.97670392509e-02 8.97670392509e-02 + 5.18159940626e-02 4.11390439155e-02 3.39197128099e-02 2.97972897261e-02 2.70509290066e-02 2.61137483222e-02 + 2.75583653887e-02 3.04051608500e-02 3.31962688978e-02 3.49765243617e-02 3.55400935242e-02 3.49765243617e-02 + 3.31962688978e-02 3.04051608500e-02 2.75583653887e-02 2.61137483222e-02 2.70509290066e-02 2.97972897261e-02 + 3.39197128100e-02 4.11390439155e-02 5.18159940627e-02 6.17010347927e-02 6.56429700805e-02 6.17010347926e-02 + 3.97328731661e-02 3.48629669819e-02 3.06791433435e-02 2.74766558578e-02 2.62349275771e-02 2.75583653887e-02 + 3.06060385055e-02 3.38926655662e-02 3.62670291685e-02 3.73533616181e-02 3.73533616181e-02 3.62670291685e-02 + 3.38926655663e-02 3.06060385056e-02 2.75583653887e-02 2.62349275771e-02 2.74766558578e-02 3.06791433435e-02 + 3.48629669819e-02 3.97328731661e-02 4.45488243286e-02 4.76129372810e-02 4.76129372810e-02 4.45488243286e-02 + 3.38926655663e-02 3.04051608500e-02 2.73901656280e-02 2.61753235783e-02 2.73901656280e-02 3.04051608500e-02 + 3.38926655662e-02 3.66852084288e-02 3.82416345368e-02 3.86928693264e-02 3.82416345368e-02 3.66852084288e-02 + 3.38926655663e-02 3.04051608500e-02 2.73901656280e-02 2.61753235783e-02 2.73901656280e-02 3.04051608500e-02 + 3.38926655663e-02 3.66852084288e-02 3.82416345368e-02 3.86928693264e-02 3.82416345368e-02 3.66852084288e-02 + 2.81654297894e-02 2.59617964748e-02 2.51928884334e-02 2.65397992490e-02 2.95889063006e-02 3.31962688978e-02 + 3.62670291685e-02 3.82416345368e-02 3.91195978595e-02 3.91195978595e-02 3.82416345368e-02 3.62670291685e-02 + 3.31962688978e-02 2.95889063006e-02 2.65397992490e-02 2.51928884334e-02 2.59617964748e-02 2.81654297894e-02 + 3.04405035973e-02 3.17358439118e-02 3.20842088246e-02 3.20842088246e-02 3.17358439118e-02 3.04405035973e-02 + 2.31387782150e-02 2.30251854758e-02 2.46284218885e-02 2.78012163806e-02 3.16055353689e-02 3.49765243617e-02 + 3.73533616181e-02 3.86928693264e-02 3.91195978595e-02 3.86928693264e-02 3.73533616181e-02 3.49765243617e-02 + 3.16055353689e-02 2.78012163806e-02 2.46284218885e-02 2.30251854758e-02 2.31387782150e-02 2.42076546578e-02 + 2.51250422144e-02 2.53972544857e-02 2.53795329069e-02 2.53972544857e-02 2.51250422144e-02 2.42076546579e-02 + 1.99974571616e-02 2.17946809311e-02 2.49901046905e-02 2.89589200656e-02 3.27027225515e-02 3.55400935242e-02 + 3.73533616181e-02 3.82416345368e-02 3.82416345368e-02 3.73533616181e-02 3.55400935242e-02 3.27027225515e-02 + 2.89589200657e-02 2.49901046905e-02 2.17946809311e-02 1.99974571615e-02 1.94536912925e-02 1.94291633463e-02 + 1.92841419762e-02 1.90346417874e-02 1.90346417875e-02 1.92841419762e-02 1.94291633463e-02 1.94536912926e-02 + 1.85131064743e-02 2.15194519516e-02 2.54054025138e-02 2.93978765598e-02 3.27027225515e-02 3.49765243617e-02 + 3.62670291685e-02 3.66852084288e-02 3.62670291685e-02 3.49765243617e-02 3.27027225515e-02 2.93978765599e-02 + 2.54054025139e-02 2.15194519516e-02 1.85131064743e-02 1.66438543464e-02 1.55937775675e-02 1.48493143122e-02 + 1.42309947893e-02 1.39668146374e-02 1.42309947893e-02 1.48493143122e-02 1.55937775675e-02 1.66438543464e-02 + 1.79740052840e-02 2.15037112467e-02 2.54054025138e-02 2.89589200656e-02 3.16055353689e-02 3.31962688978e-02 + 3.38926655663e-02 3.38926655663e-02 3.31962688978e-02 3.16055353689e-02 2.89589200657e-02 2.54054025138e-02 + 2.15037112468e-02 1.79740052840e-02 1.52761582425e-02 1.34169422858e-02 1.21091269494e-02 1.11332793734e-02 + 1.05558438840e-02 1.05558438840e-02 1.11332793734e-02 1.21091269494e-02 1.34169422858e-02 1.52761582425e-02 + 1.79740052840e-02 2.15194519516e-02 2.49901046905e-02 2.78012163806e-02 2.95889063006e-02 3.04051608500e-02 + 3.06060385056e-02 3.04051608500e-02 2.95889063006e-02 2.78012163806e-02 2.49901046905e-02 2.15194519516e-02 + 1.79740052840e-02 1.48655958108e-02 1.24266085814e-02 1.06110221336e-02 9.28365371522e-03 8.42000879273e-03 + 8.11231618987e-03 8.42000879273e-03 9.28365371522e-03 1.06110221336e-02 1.24266085814e-02 1.48655958108e-02 + 1.85131064743e-02 2.17946809311e-02 2.46284218885e-02 2.65397992490e-02 2.73901656280e-02 2.75583653887e-02 + 2.75583653887e-02 2.73901656280e-02 2.65397992490e-02 2.46284218885e-02 2.17946809311e-02 1.85131064743e-02 + 1.52761582425e-02 1.24266085814e-02 1.01199573737e-02 8.37411495641e-03 7.17868687881e-03 6.56139720408e-03 + 6.56139720408e-03 7.17868687881e-03 8.37411495641e-03 1.01199573737e-02 1.24266085814e-02 1.52761582425e-02 + 1.99974571616e-02 2.30251854758e-02 2.51928884334e-02 2.61753235783e-02 2.62349275771e-02 2.61137483222e-02 + 2.62349275771e-02 2.61753235783e-02 2.51928884334e-02 2.30251854758e-02 1.99974571616e-02 1.66438543464e-02 + 1.34169422858e-02 1.06110221336e-02 8.37411495641e-03 6.75457206648e-03 5.76671059723e-03 5.43241849957e-03 + 5.76671059723e-03 6.75457206647e-03 8.37411495641e-03 1.06110221336e-02 1.34169422858e-02 1.66438543464e-02 + 2.31387782150e-02 2.59617964748e-02 2.73901656280e-02 2.74766558578e-02 2.70509290066e-02 2.70509290066e-02 + 2.74766558578e-02 2.73901656280e-02 2.59617964748e-02 2.31387782150e-02 1.94536912925e-02 1.55937775675e-02 + 1.21091269494e-02 9.28365371521e-03 7.17868687881e-03 5.76671059723e-03 5.04346463958e-03 5.04346463958e-03 + 5.76671059723e-03 7.17868687881e-03 9.28365371522e-03 1.21091269495e-02 1.55937775675e-02 1.94536912926e-02 + 2.81654297894e-02 3.04051608500e-02 3.06791433435e-02 2.97972897261e-02 2.92434029240e-02 2.97972897261e-02 + 3.06791433435e-02 3.04051608500e-02 2.81654297894e-02 2.42076546578e-02 1.94291633463e-02 1.48493143122e-02 + 1.11332793734e-02 8.42000879273e-03 6.56139720408e-03 5.43241849957e-03 5.04346463958e-03 5.43241849957e-03 + 6.56139720408e-03 8.42000879273e-03 1.11332793734e-02 1.48493143123e-02 1.94291633463e-02 2.42076546579e-02 + 3.38926655663e-02 3.48629669819e-02 3.39197128099e-02 3.26919751552e-02 3.26919751552e-02 3.39197128100e-02 + 3.48629669819e-02 3.38926655663e-02 3.04405035973e-02 2.51250422144e-02 1.92841419762e-02 1.42309947893e-02 + 1.05558438840e-02 8.11231618987e-03 6.56139720408e-03 5.76671059723e-03 5.76671059723e-03 6.56139720408e-03 + 8.11231618988e-03 1.05558438840e-02 1.42309947893e-02 1.92841419762e-02 2.51250422144e-02 3.04405035973e-02 + 3.97328731661e-02 4.11390439155e-02 4.15214590889e-02 4.15474648475e-02 4.15214590889e-02 4.11390439155e-02 + 3.97328731661e-02 3.66852084288e-02 3.17358439118e-02 2.53972544857e-02 1.90346417875e-02 1.39668146374e-02 + 1.05558438840e-02 8.42000879273e-03 7.17868687881e-03 6.75457206647e-03 7.17868687881e-03 8.42000879273e-03 + 1.05558438840e-02 1.39668146374e-02 1.90346417875e-02 2.53972544857e-02 3.17358439118e-02 3.66852084288e-02 + 5.18159940627e-02 5.87813204090e-02 6.30067497701e-02 6.30067497702e-02 5.87813204091e-02 5.18159940627e-02 + 4.45488243286e-02 3.82416345368e-02 3.20842088246e-02 2.53795329069e-02 1.90346417875e-02 1.42309947893e-02 + 1.11332793734e-02 9.28365371521e-03 8.37411495641e-03 8.37411495641e-03 9.28365371522e-03 1.11332793734e-02 + 1.42309947893e-02 1.90346417875e-02 2.53795329069e-02 3.20842088246e-02 3.82416345368e-02 4.45488243286e-02 + 7.81199920260e-02 9.00462898608e-02 9.40547983346e-02 9.00462898609e-02 7.81199920260e-02 6.17010347927e-02 + 4.76129372810e-02 3.86928693264e-02 3.20842088246e-02 2.53972544857e-02 1.92841419762e-02 1.48493143122e-02 + 1.21091269494e-02 1.06110221336e-02 1.01199573737e-02 1.06110221336e-02 1.21091269495e-02 1.48493143122e-02 + 1.92841419762e-02 2.53972544857e-02 3.20842088246e-02 3.86928693264e-02 4.76129372810e-02 6.17010347927e-02 + 1.07505424132e-01 1.13480556613e-01 1.13480556613e-01 1.07505424132e-01 8.97670392509e-02 6.56429700805e-02 + 4.76129372810e-02 3.82416345368e-02 3.17358439118e-02 2.51250422144e-02 1.94291633463e-02 1.55937775675e-02 + 1.34169422858e-02 1.24266085814e-02 1.24266085814e-02 1.34169422858e-02 1.55937775675e-02 1.94291633463e-02 + 2.51250422144e-02 3.17358439118e-02 3.82416345368e-02 4.76129372810e-02 6.56429700805e-02 8.97670392509e-02 + 1.15477102967e-01 1.12133532869e-01 1.15477102967e-01 1.12226003113e-01 8.97670392509e-02 6.17010347926e-02 + 4.45488243286e-02 3.66852084288e-02 3.04405035973e-02 2.42076546579e-02 1.94536912926e-02 1.66438543464e-02 + 1.52761582425e-02 1.48655958108e-02 1.52761582425e-02 1.66438543464e-02 1.94536912926e-02 2.42076546579e-02 + 3.04405035973e-02 3.66852084288e-02 4.45488243287e-02 6.17010347927e-02 8.97670392509e-02 1.12226003113e-01 diff --git a/tests/01_PW/scf_out_elf/refelftot.cube b/tests/01_PW/scf_out_elf/elftot.cube.ref similarity index 100% rename from tests/01_PW/scf_out_elf/refelftot.cube rename to tests/01_PW/scf_out_elf/elftot.cube.ref diff --git a/tests/01_PW/scf_pbe0_afm_kpar/INPUT b/tests/01_PW/scf_pbe0_afm_kpar/INPUT new file mode 100644 index 00000000000..7090fe68377 --- /dev/null +++ b/tests/01_PW/scf_pbe0_afm_kpar/INPUT @@ -0,0 +1,32 @@ +INPUT_PARAMETERS +#Parameters (System) +suffix autotest +calculation scf +nspin 2 +pseudo_dir ../../PP_ORB +orbital_dir ../../PP_ORB + +#Parameters (PW) +ecutwfc 12 +basis_type pw + +#Parameters (SCF) +scf_thr 1.0e-9 +scf_nmax 100 +smearing_method fixed +mixing_type pulay +mixing_ndim 8 +mixing_beta 0.2 + +#Parameters (Hybrid Functional) +dft_functional pbe0 + +#Parameters (Other) +symmetry -1 +init_wfc random +pw_seed 1 +device cpu +exx_thr_type energy +exx_ene_thr 5e-2 + +kpar 2 diff --git a/tests/01_PW/scf_pbe0_afm_kpar/KPT b/tests/01_PW/scf_pbe0_afm_kpar/KPT new file mode 100644 index 00000000000..2bce8546f16 --- /dev/null +++ b/tests/01_PW/scf_pbe0_afm_kpar/KPT @@ -0,0 +1,4 @@ +K_POINTS +0 +Gamma +1 3 1 0 0 0 diff --git a/tests/01_PW/scf_pbe0_afm_kpar/README b/tests/01_PW/scf_pbe0_afm_kpar/README new file mode 100644 index 00000000000..f4fe093ec2d --- /dev/null +++ b/tests/01_PW/scf_pbe0_afm_kpar/README @@ -0,0 +1 @@ +PW PBE0 AFM Fe with uneven k-point pools. diff --git a/tests/01_PW/scf_pbe0_afm_kpar/STRU b/tests/01_PW/scf_pbe0_afm_kpar/STRU new file mode 100644 index 00000000000..d4074b48f18 --- /dev/null +++ b/tests/01_PW/scf_pbe0_afm_kpar/STRU @@ -0,0 +1,18 @@ +ATOMIC_SPECIES +Fe 55.845 Fe_ONCV_PBE-1.0.upf + +LATTICE_CONSTANT +1.889725989 + +LATTICE_VECTORS + 3.5000000000000000 0.0000000000000000 0.0000000000000000 + 0.0000000000000000 3.5000000000000000 0.0000000000000000 + 0.0000000000000000 0.0000000000000000 3.5000000000000000 +ATOMIC_POSITIONS +Direct + +Fe +1.0 +2 +0.0 0.0 0.0 mag 5 +0.5 0.5 0.5 mag -5 diff --git a/tests/01_PW/scf_pbe0_afm_kpar/magnetism.ref b/tests/01_PW/scf_pbe0_afm_kpar/magnetism.ref new file mode 100644 index 00000000000..8bde63c052d --- /dev/null +++ b/tests/01_PW/scf_pbe0_afm_kpar/magnetism.ref @@ -0,0 +1 @@ +0 6.53481 diff --git a/tests/01_PW/scf_pbe0_afm_kpar/result.ref b/tests/01_PW/scf_pbe0_afm_kpar/result.ref new file mode 100644 index 00000000000..27fde65b550 --- /dev/null +++ b/tests/01_PW/scf_pbe0_afm_kpar/result.ref @@ -0,0 +1,4 @@ +etotref -5596.07517080915 +etotperatomref -2798.0375854046 +CompareMagnetism_pass 0 +totaltimeref 5.24 diff --git a/tests/01_PW/scf_pbe0_afm_kpar/threshold b/tests/01_PW/scf_pbe0_afm_kpar/threshold new file mode 100644 index 00000000000..8fcfcebb260 --- /dev/null +++ b/tests/01_PW/scf_pbe0_afm_kpar/threshold @@ -0,0 +1 @@ +threshold 0.0001 diff --git a/tests/01_PW/scf_pbe0_spin1/INPUT b/tests/01_PW/scf_pbe0_spin1/INPUT new file mode 100644 index 00000000000..bb90e226d1d --- /dev/null +++ b/tests/01_PW/scf_pbe0_spin1/INPUT @@ -0,0 +1,24 @@ +INPUT_PARAMETERS +#Parameters (System) +suffix autotest +init_wfc random + +nbands 2 +stru_file STRU +kpoint_file KPT +pseudo_dir ../../PP_ORB +pw_seed 1 +calculation scf +#Parameters (PW) +ecutwfc 10.0 # Rydberg +#Parameters (electronic) +basis_type pw +scf_thr 1e-7 +smearing_method fixed + +cal_force 1 + +dft_functional pbe0 +symmetry -1 +exx_gamma_extra false + diff --git a/tests/01_PW/083_PW_sol_H2/KPT b/tests/01_PW/scf_pbe0_spin1/KPT similarity index 100% rename from tests/01_PW/083_PW_sol_H2/KPT rename to tests/01_PW/scf_pbe0_spin1/KPT diff --git a/tests/01_PW/scf_pbe0_spin1/README b/tests/01_PW/scf_pbe0_spin1/README new file mode 100644 index 00000000000..ebb5ae4b0ca --- /dev/null +++ b/tests/01_PW/scf_pbe0_spin1/README @@ -0,0 +1 @@ +PW basis, hybrid functional PBE0, nspin=1, forces and stress are calculated. diff --git a/tests/01_PW/scf_pbe0_spin1/STRU b/tests/01_PW/scf_pbe0_spin1/STRU new file mode 100644 index 00000000000..c35003c65ba --- /dev/null +++ b/tests/01_PW/scf_pbe0_spin1/STRU @@ -0,0 +1,19 @@ +ATOMIC_SPECIES +H 1 H_ONCV_PBE-1.0.upf + +LATTICE_CONSTANT +1.889766 + +LATTICE_VECTORS +0.0 2.708337 2.708337 +2.708337 0.0 2.708337 +2.708337 2.708337 0.0 + +ATOMIC_POSITIONS +Direct + +H +0 +2 +0.125 0.125 0.125 0 0 0 +0.675 0.835 0.875 0 0 0 diff --git a/tests/01_PW/scf_pbe0_spin1/result.ref b/tests/01_PW/scf_pbe0_spin1/result.ref new file mode 100644 index 00000000000..93f52b89850 --- /dev/null +++ b/tests/01_PW/scf_pbe0_spin1/result.ref @@ -0,0 +1,4 @@ +etotref -30.7341719031018634 +etotperatomref -15.3670859516 +totalforceref 2.591254 +totaltimeref 0.10 diff --git a/tests/01_PW/scf_pbe0_spin2/INPUT b/tests/01_PW/scf_pbe0_spin2/INPUT new file mode 100644 index 00000000000..e3533263f15 --- /dev/null +++ b/tests/01_PW/scf_pbe0_spin2/INPUT @@ -0,0 +1,35 @@ +INPUT_PARAMETERS +#Parameters (System) +suffix autotest +calculation scf +nspin 2 +pseudo_dir ../../PP_ORB +orbital_dir ../../PP_ORB + +#Parameters (PW) +ecutwfc 12 +basis_type pw + +#Parameters (SCF) +scf_thr 1.0e-7 +scf_nmax 100 +smearing_method gaussian +smearing_sigma 0.015 +mixing_type pulay +mixing_ndim 8 +mixing_beta 0.2 + +#Parameters (Hybrid Functional) +dft_functional pbe0 + +#Parameters (Other) +symmetry -1 +init_wfc random +pw_seed 1 +device cpu +exx_thr_type energy +exx_ene_thr 5e-2 + +cal_force 1 +#cal_stress 1 # not implement yet +out_elf 1 diff --git a/tests/01_PW/scf_pbe0_spin2/KPT b/tests/01_PW/scf_pbe0_spin2/KPT new file mode 100644 index 00000000000..7ec0096d1b4 --- /dev/null +++ b/tests/01_PW/scf_pbe0_spin2/KPT @@ -0,0 +1,7 @@ +K_POINTS +0 +Gamma +1 2 1 0 0 0 +### If you are running an energy calculation, please make sure your final energy is +### converged with respect to the k-point settings, unless you set a loose k-point +### mesh on purpose. diff --git a/tests/01_PW/scf_pbe0_spin2/README b/tests/01_PW/scf_pbe0_spin2/README new file mode 100644 index 00000000000..6a826eb6da0 --- /dev/null +++ b/tests/01_PW/scf_pbe0_spin2/README @@ -0,0 +1,3 @@ +PW basis, FM Fe nspin=2 hybrid functional (PBE0), force calcualted but stress is not implemented yet + + diff --git a/tests/01_PW/scf_pbe0_spin2/STRU b/tests/01_PW/scf_pbe0_spin2/STRU new file mode 100644 index 00000000000..bfdb2812ead --- /dev/null +++ b/tests/01_PW/scf_pbe0_spin2/STRU @@ -0,0 +1,18 @@ +ATOMIC_SPECIES +Fe 55.845 Fe_ONCV_PBE-1.0.upf + +LATTICE_CONSTANT +1.889725989 + +LATTICE_VECTORS + 2.8328152400000000 0.0000000000000000 0.0000000000000000 + 0.0000000000000000 2.8328152400000000 0.0000000000000000 + 0.0000000000000000 0.0000000000000000 2.8328152400000000 +ATOMIC_POSITIONS +Direct + +Fe +1.0 +2 +0.0 0.0 0.03 mag 5 +0.25 0.61 0.5 mag 5 diff --git a/tests/01_PW/scf_pbe0_spin2/elftot.cube.ref b/tests/01_PW/scf_pbe0_spin2/elftot.cube.ref new file mode 100644 index 00000000000..481ef00d168 --- /dev/null +++ b/tests/01_PW/scf_pbe0_spin2/elftot.cube.ref @@ -0,0 +1,296 @@ +Ionic_Step 1 Cubefile created from ABACUS. Inner loop is z, followed by y and x +2 # number of spin directions +2 0.0 0.0 0.0 +12 0.446104 0.000000 0.000000 +12 0.000000 0.446104 0.000000 +12 0.000000 0.000000 0.446104 + 26 16.000000 0.000000 0.000000 0.160597 + 26 16.000000 1.338311 3.265479 2.676622 + 5.573e-01 5.848e-01 6.431e-01 5.587e-01 3.202e-01 3.220e-01 + 2.597e-01 3.226e-01 2.422e-01 3.971e-01 6.133e-01 6.251e-01 + 5.977e-01 6.117e-01 6.507e-01 5.753e-01 3.343e-01 2.748e-01 + 1.915e-01 2.634e-01 2.184e-01 3.734e-01 5.989e-01 6.395e-01 + 6.461e-01 6.390e-01 6.283e-01 5.082e-01 2.647e-01 2.030e-01 + 1.052e-01 1.466e-01 1.785e-01 3.086e-01 5.684e-01 6.554e-01 + 5.853e-01 5.554e-01 4.764e-01 3.208e-01 1.753e-01 1.594e-01 + 1.564e-01 1.459e-01 1.288e-01 1.946e-01 4.187e-01 5.609e-01 + 3.485e-01 2.969e-01 2.392e-01 2.243e-01 1.798e-01 1.873e-01 + 2.381e-01 2.042e-01 1.423e-01 1.539e-01 2.168e-01 3.046e-01 + 2.190e-01 1.709e-01 1.408e-01 1.470e-01 1.987e-01 3.090e-01 + 3.777e-01 3.270e-01 2.008e-01 1.179e-01 1.153e-01 1.748e-01 + 1.162e-01 1.013e-01 7.108e-02 1.426e-01 3.234e-01 4.677e-01 + 5.226e-01 4.932e-01 3.597e-01 1.643e-01 7.372e-02 8.748e-02 + 9.294e-02 1.226e-01 1.406e-01 2.370e-01 4.079e-01 5.107e-01 + 5.589e-01 5.676e-01 4.788e-01 2.693e-01 1.510e-01 1.317e-01 + 2.534e-01 2.786e-01 2.540e-01 2.942e-01 4.082e-01 4.402e-01 + 4.570e-01 5.012e-01 4.755e-01 2.913e-01 1.868e-01 2.195e-01 + 5.631e-01 5.561e-01 4.519e-01 3.267e-01 3.500e-01 3.314e-01 + 2.801e-01 3.394e-01 3.566e-01 2.112e-01 2.883e-01 4.839e-01 + 6.479e-01 6.496e-01 6.077e-01 4.328e-01 3.191e-01 2.992e-01 + 1.671e-01 2.132e-01 2.570e-01 2.606e-01 5.206e-01 6.388e-01 + 6.018e-01 6.204e-01 6.448e-01 5.221e-01 3.221e-01 3.505e-01 + 1.664e-01 2.159e-01 2.639e-01 3.840e-01 6.201e-01 6.506e-01 + 6.161e-01 6.396e-01 6.755e-01 5.539e-01 3.023e-01 3.367e-01 + 2.137e-01 2.731e-01 2.262e-01 3.538e-01 6.236e-01 6.661e-01 + 6.348e-01 6.448e-01 6.426e-01 5.013e-01 2.351e-01 2.245e-01 + 1.594e-01 2.080e-01 1.529e-01 2.708e-01 5.482e-01 6.406e-01 + 6.376e-01 6.265e-01 5.618e-01 3.737e-01 1.804e-01 1.876e-01 + 1.310e-01 1.607e-01 1.437e-01 1.928e-01 4.439e-01 6.015e-01 + 5.534e-01 5.071e-01 3.776e-01 2.581e-01 2.085e-01 2.054e-01 + 2.179e-01 2.072e-01 1.763e-01 1.653e-01 2.909e-01 4.750e-01 + 3.355e-01 2.763e-01 2.285e-01 2.461e-01 2.338e-01 2.937e-01 + 3.846e-01 3.347e-01 2.097e-01 1.673e-01 1.920e-01 2.687e-01 + 2.164e-01 2.026e-01 1.503e-01 1.733e-01 3.278e-01 4.971e-01 + 5.656e-01 5.193e-01 3.432e-01 1.471e-01 1.123e-01 1.978e-01 + 1.203e-01 1.642e-01 1.023e-01 2.420e-01 5.175e-01 6.318e-01 + 6.533e-01 6.383e-01 5.307e-01 2.481e-01 8.597e-02 1.335e-01 + 1.173e-01 2.177e-01 1.843e-01 3.644e-01 6.165e-01 6.703e-01 + 6.636e-01 6.838e-01 6.471e-01 3.939e-01 1.568e-01 1.981e-01 + 2.207e-01 3.124e-01 2.782e-01 3.754e-01 6.144e-01 6.426e-01 + 6.069e-01 6.594e-01 6.687e-01 4.090e-01 1.881e-01 2.443e-01 + 4.832e-01 5.174e-01 4.184e-01 3.091e-01 4.891e-01 5.609e-01 + 5.060e-01 5.667e-01 5.521e-01 2.321e-01 2.334e-01 4.554e-01 + 6.103e-01 6.305e-01 6.001e-01 3.778e-01 3.462e-01 4.470e-01 + 3.678e-01 4.011e-01 3.224e-01 1.913e-01 4.802e-01 6.368e-01 + 6.202e-01 6.453e-01 6.776e-01 5.243e-01 3.273e-01 3.784e-01 + 2.327e-01 2.568e-01 2.457e-01 3.303e-01 6.392e-01 6.820e-01 + 6.590e-01 6.597e-01 6.593e-01 4.937e-01 2.391e-01 2.451e-01 + 1.522e-01 1.750e-01 1.819e-01 2.945e-01 6.121e-01 6.915e-01 + 6.655e-01 6.602e-01 6.118e-01 4.007e-01 1.423e-01 1.483e-01 + 1.299e-01 1.399e-01 1.016e-01 1.893e-01 5.015e-01 6.465e-01 + 6.004e-01 5.816e-01 4.631e-01 2.448e-01 1.195e-01 1.663e-01 + 1.947e-01 1.888e-01 1.107e-01 1.151e-01 3.155e-01 5.254e-01 + 4.510e-01 4.006e-01 2.752e-01 2.214e-01 2.139e-01 2.146e-01 + 2.685e-01 2.565e-01 1.903e-01 1.499e-01 1.933e-01 3.421e-01 + 2.610e-01 2.307e-01 2.162e-01 2.443e-01 2.828e-01 3.900e-01 + 4.877e-01 4.537e-01 3.022e-01 1.866e-01 1.741e-01 2.116e-01 + 1.441e-01 1.948e-01 1.697e-01 2.536e-01 4.552e-01 5.822e-01 + 6.233e-01 6.039e-01 4.791e-01 2.396e-01 1.365e-01 1.899e-01 + 1.040e-01 2.179e-01 1.764e-01 3.698e-01 5.910e-01 6.453e-01 + 6.402e-01 6.404e-01 5.801e-01 3.487e-01 1.428e-01 1.960e-01 + 1.768e-01 3.435e-01 2.468e-01 4.457e-01 6.471e-01 6.585e-01 + 6.176e-01 6.456e-01 6.431e-01 4.475e-01 1.994e-01 3.011e-01 + 1.796e-01 3.476e-01 3.213e-01 4.503e-01 6.610e-01 6.673e-01 + 6.124e-01 6.576e-01 6.823e-01 4.853e-01 2.381e-01 2.871e-01 + 3.035e-01 3.945e-01 3.679e-01 3.532e-01 5.868e-01 6.553e-01 + 6.182e-01 6.490e-01 6.321e-01 3.229e-01 2.163e-01 3.575e-01 + 4.704e-01 5.043e-01 4.957e-01 3.218e-01 4.028e-01 5.630e-01 + 5.508e-01 5.448e-01 4.093e-01 1.749e-01 3.676e-01 5.422e-01 + 5.885e-01 6.010e-01 6.255e-01 4.419e-01 2.889e-01 3.735e-01 + 3.218e-01 3.005e-01 2.202e-01 2.449e-01 5.890e-01 6.616e-01 + 5.527e-01 5.152e-01 4.542e-01 3.222e-01 1.912e-01 1.625e-01 + 1.227e-01 1.288e-01 1.661e-01 2.276e-01 4.302e-01 5.515e-01 + 5.516e-01 5.232e-01 4.204e-01 2.346e-01 9.450e-02 8.714e-02 + 9.631e-02 9.267e-02 8.510e-02 1.383e-01 3.359e-01 5.025e-01 + 4.261e-01 4.012e-01 2.816e-01 1.574e-01 1.056e-01 1.295e-01 + 1.927e-01 1.714e-01 9.823e-02 9.835e-02 1.917e-01 3.390e-01 + 2.736e-01 2.402e-01 1.790e-01 1.964e-01 2.047e-01 2.114e-01 + 2.696e-01 2.657e-01 2.001e-01 1.536e-01 1.381e-01 2.000e-01 + 1.793e-01 1.716e-01 1.607e-01 2.295e-01 3.514e-01 4.570e-01 + 5.184e-01 5.126e-01 4.083e-01 2.254e-01 1.306e-01 1.484e-01 + 9.556e-02 1.704e-01 1.858e-01 3.671e-01 5.657e-01 6.247e-01 + 6.374e-01 6.491e-01 5.956e-01 3.731e-01 1.655e-01 1.621e-01 + 1.159e-01 2.691e-01 2.646e-01 4.919e-01 6.394e-01 6.393e-01 + 6.135e-01 6.411e-01 6.321e-01 4.637e-01 2.266e-01 2.613e-01 + 2.227e-01 3.581e-01 2.720e-01 4.890e-01 6.381e-01 6.145e-01 + 5.566e-01 6.012e-01 6.321e-01 4.916e-01 2.454e-01 3.255e-01 + 1.528e-01 3.506e-01 3.003e-01 4.520e-01 6.287e-01 6.274e-01 + 5.765e-01 6.189e-01 6.492e-01 5.030e-01 2.680e-01 3.043e-01 + 1.390e-01 2.593e-01 3.226e-01 3.917e-01 5.911e-01 6.507e-01 + 6.401e-01 6.513e-01 6.223e-01 4.042e-01 2.376e-01 2.492e-01 + 2.360e-01 2.734e-01 3.401e-01 3.262e-01 4.556e-01 5.876e-01 + 6.177e-01 5.866e-01 4.506e-01 2.278e-01 2.651e-01 3.212e-01 + 4.174e-01 3.972e-01 4.083e-01 3.332e-01 2.875e-01 3.558e-01 + 3.752e-01 3.226e-01 2.270e-01 2.122e-01 3.972e-01 4.689e-01 + 3.435e-01 2.948e-01 2.152e-01 1.958e-01 1.675e-01 1.179e-01 + 1.090e-01 1.113e-01 1.493e-01 1.878e-01 2.016e-01 2.842e-01 + 3.001e-01 2.838e-01 2.153e-01 1.416e-01 7.603e-02 5.136e-02 + 6.285e-02 6.452e-02 7.413e-02 1.219e-01 1.747e-01 2.452e-01 + 1.943e-01 1.929e-01 1.521e-01 1.084e-01 7.328e-02 7.324e-02 + 1.121e-01 1.063e-01 7.611e-02 8.801e-02 1.129e-01 1.538e-01 + 1.365e-01 1.170e-01 8.128e-02 1.058e-01 1.515e-01 1.758e-01 + 1.989e-01 2.042e-01 1.842e-01 1.173e-01 6.170e-02 9.103e-02 + 1.012e-01 8.608e-02 6.472e-02 1.741e-01 3.700e-01 4.411e-01 + 4.488e-01 4.641e-01 4.250e-01 2.248e-01 6.024e-02 6.134e-02 + 8.163e-02 1.303e-01 1.448e-01 4.009e-01 6.103e-01 6.294e-01 + 6.169e-01 6.492e-01 6.347e-01 4.167e-01 1.410e-01 1.176e-01 + 1.181e-01 2.541e-01 2.611e-01 5.413e-01 6.835e-01 6.672e-01 + 6.359e-01 6.763e-01 6.847e-01 5.209e-01 2.385e-01 2.440e-01 + 1.558e-01 2.596e-01 2.257e-01 5.041e-01 6.685e-01 6.542e-01 + 6.100e-01 6.526e-01 6.755e-01 5.269e-01 2.337e-01 2.325e-01 + 1.152e-01 2.310e-01 1.635e-01 3.799e-01 6.116e-01 6.443e-01 + 6.196e-01 6.473e-01 6.490e-01 4.708e-01 1.836e-01 1.771e-01 + 9.627e-02 1.912e-01 1.856e-01 2.894e-01 5.301e-01 6.333e-01 + 6.508e-01 6.445e-01 5.751e-01 3.433e-01 1.609e-01 1.650e-01 + 1.429e-01 1.730e-01 2.336e-01 2.725e-01 4.003e-01 5.470e-01 + 6.091e-01 5.595e-01 3.901e-01 2.105e-01 1.914e-01 1.817e-01 + 2.586e-01 2.256e-01 2.223e-01 2.643e-01 2.516e-01 2.968e-01 + 3.548e-01 2.891e-01 1.894e-01 1.937e-01 2.184e-01 2.352e-01 + 4.151e-01 3.867e-01 2.786e-01 1.809e-01 1.208e-01 1.016e-01 + 1.019e-01 1.038e-01 1.120e-01 1.319e-01 1.942e-01 3.335e-01 + 2.380e-01 2.845e-01 3.420e-01 2.097e-01 7.777e-02 5.339e-02 + 5.657e-02 5.975e-02 7.219e-02 1.403e-01 2.497e-01 2.412e-01 + 1.374e-01 1.601e-01 1.714e-01 1.041e-01 5.986e-02 5.779e-02 + 7.079e-02 7.240e-02 7.121e-02 8.578e-02 1.170e-01 1.329e-01 + 1.094e-01 7.300e-02 3.877e-02 6.945e-02 9.825e-02 1.011e-01 + 1.079e-01 1.138e-01 1.246e-01 9.428e-02 3.371e-02 5.213e-02 + 8.188e-02 4.259e-02 3.528e-02 1.302e-01 2.527e-01 2.726e-01 + 2.546e-01 2.729e-01 2.805e-01 1.698e-01 3.973e-02 3.403e-02 + 7.518e-02 6.972e-02 7.856e-02 2.793e-01 4.989e-01 5.167e-01 + 4.889e-01 5.209e-01 5.092e-01 2.872e-01 7.994e-02 6.465e-02 + 5.997e-02 1.103e-01 1.514e-01 4.375e-01 6.305e-01 6.332e-01 + 6.089e-01 6.337e-01 6.203e-01 4.126e-01 1.494e-01 1.096e-01 + 6.914e-02 1.383e-01 1.454e-01 4.243e-01 6.477e-01 6.755e-01 + 6.571e-01 6.679e-01 6.441e-01 4.355e-01 1.548e-01 1.246e-01 + 7.486e-02 1.184e-01 7.258e-02 2.667e-01 5.762e-01 6.669e-01 + 6.693e-01 6.615e-01 6.006e-01 3.375e-01 8.689e-02 8.445e-02 + 1.158e-01 1.312e-01 5.744e-02 1.384e-01 4.277e-01 6.058e-01 + 6.465e-01 6.081e-01 4.612e-01 1.786e-01 5.562e-02 9.433e-02 + 2.552e-01 2.167e-01 1.372e-01 1.293e-01 2.569e-01 4.540e-01 + 5.406e-01 4.592e-01 2.459e-01 1.010e-01 9.817e-02 1.920e-01 + 4.270e-01 3.347e-01 2.228e-01 1.732e-01 1.483e-01 2.105e-01 + 2.681e-01 2.045e-01 1.168e-01 1.122e-01 1.610e-01 2.957e-01 + 2.587e-01 2.612e-01 1.620e-01 7.865e-02 7.886e-02 1.152e-01 + 1.205e-01 1.096e-01 8.072e-02 5.955e-02 9.146e-02 1.911e-01 + 1.016e-01 1.248e-01 1.604e-01 1.446e-01 1.175e-01 1.154e-01 + 1.031e-01 9.174e-02 1.035e-01 1.390e-01 1.425e-01 1.076e-01 + 5.259e-02 6.592e-02 1.043e-01 1.134e-01 1.012e-01 9.469e-02 + 8.716e-02 8.578e-02 9.640e-02 1.208e-01 1.182e-01 7.032e-02 + 6.678e-02 4.754e-02 4.283e-02 8.650e-02 7.325e-02 5.298e-02 + 5.730e-02 5.866e-02 6.909e-02 9.208e-02 5.767e-02 4.653e-02 + 9.455e-02 3.968e-02 4.835e-02 1.290e-01 1.062e-01 7.318e-02 + 6.555e-02 7.361e-02 9.938e-02 1.162e-01 5.525e-02 4.253e-02 + 6.565e-02 3.901e-02 5.105e-02 1.295e-01 2.011e-01 2.135e-01 + 2.048e-01 2.062e-01 1.873e-01 1.118e-01 4.983e-02 4.207e-02 + 3.540e-02 3.860e-02 6.341e-02 1.814e-01 3.354e-01 4.060e-01 + 4.166e-01 3.842e-01 2.888e-01 1.401e-01 6.539e-02 4.379e-02 + 5.987e-02 8.136e-02 9.224e-02 2.072e-01 4.079e-01 5.319e-01 + 5.606e-01 5.039e-01 3.534e-01 1.561e-01 8.223e-02 8.268e-02 + 8.239e-02 1.061e-01 7.141e-02 1.442e-01 3.735e-01 5.553e-01 + 6.020e-01 5.302e-01 3.366e-01 1.168e-01 5.907e-02 9.697e-02 + 7.276e-02 6.270e-02 4.775e-02 8.796e-02 2.598e-01 4.686e-01 + 5.409e-01 4.476e-01 2.283e-01 7.155e-02 4.414e-02 6.321e-02 + 1.160e-01 8.274e-02 8.149e-02 9.507e-02 1.515e-01 2.947e-01 + 3.735e-01 2.739e-01 1.194e-01 6.498e-02 6.779e-02 8.553e-02 + 2.318e-01 1.899e-01 1.311e-01 9.609e-02 9.672e-02 1.477e-01 + 1.734e-01 1.292e-01 8.059e-02 6.533e-02 7.938e-02 1.661e-01 + 4.716e-01 4.879e-01 2.758e-01 9.334e-02 6.222e-02 1.024e-01 + 1.578e-01 1.035e-01 4.697e-02 4.047e-02 9.554e-02 2.912e-01 + 2.605e-01 2.597e-01 1.756e-01 1.285e-01 1.170e-01 1.281e-01 + 1.551e-01 1.189e-01 9.772e-02 8.328e-02 7.580e-02 1.474e-01 + 6.357e-02 8.507e-02 1.126e-01 8.271e-02 6.632e-02 8.840e-02 + 9.800e-02 9.401e-02 7.539e-02 6.089e-02 6.624e-02 6.466e-02 + 2.840e-02 3.792e-02 4.457e-02 3.427e-02 2.775e-02 3.374e-02 + 4.385e-02 3.690e-02 3.007e-02 3.676e-02 4.119e-02 3.044e-02 + 4.652e-02 3.392e-02 2.810e-02 4.220e-02 3.493e-02 2.109e-02 + 1.660e-02 1.693e-02 2.741e-02 4.417e-02 3.200e-02 2.895e-02 + 6.210e-02 4.008e-02 3.764e-02 5.764e-02 5.762e-02 5.150e-02 + 4.765e-02 4.127e-02 3.704e-02 3.960e-02 3.751e-02 4.375e-02 + 6.441e-02 4.938e-02 5.419e-02 7.046e-02 8.983e-02 1.460e-01 + 1.842e-01 1.254e-01 5.069e-02 3.429e-02 5.128e-02 5.358e-02 + 1.070e-01 8.677e-02 7.905e-02 8.102e-02 1.305e-01 2.710e-01 + 3.541e-01 2.400e-01 7.683e-02 3.396e-02 5.690e-02 7.602e-02 + 1.076e-01 9.277e-02 6.400e-02 6.712e-02 1.514e-01 3.293e-01 + 4.120e-01 2.979e-01 1.026e-01 2.960e-02 3.478e-02 6.922e-02 + 1.432e-01 9.907e-02 4.967e-02 5.067e-02 1.202e-01 2.719e-01 + 3.428e-01 2.559e-01 9.234e-02 2.501e-02 2.523e-02 6.981e-02 + 2.851e-01 2.409e-01 1.251e-01 5.602e-02 6.540e-02 1.551e-01 + 2.192e-01 1.575e-01 5.370e-02 2.306e-02 5.163e-02 1.915e-01 + 4.276e-01 4.541e-01 2.807e-01 7.143e-02 4.263e-02 9.198e-02 + 1.324e-01 9.198e-02 3.358e-02 2.313e-02 9.972e-02 3.371e-01 + 3.675e-01 3.529e-01 1.916e-01 9.033e-02 7.782e-02 8.899e-02 + 1.180e-01 9.261e-02 6.415e-02 3.441e-02 5.392e-02 1.942e-01 + 2.300e-01 2.198e-01 1.271e-01 8.005e-02 6.862e-02 5.462e-02 + 7.290e-02 6.255e-02 5.973e-02 3.954e-02 4.105e-02 1.172e-01 + 7.429e-02 7.748e-02 6.862e-02 6.410e-02 4.931e-02 3.707e-02 + 5.630e-02 5.276e-02 4.202e-02 3.870e-02 3.702e-02 5.257e-02 + 1.865e-02 2.218e-02 3.455e-02 4.989e-02 3.637e-02 2.940e-02 + 5.379e-02 4.219e-02 2.925e-02 4.280e-02 3.709e-02 2.186e-02 + 2.749e-02 2.647e-02 2.518e-02 3.112e-02 3.392e-02 3.387e-02 + 3.184e-02 2.697e-02 3.129e-02 3.972e-02 2.987e-02 2.388e-02 + 8.335e-02 7.707e-02 5.081e-02 4.999e-02 8.199e-02 7.518e-02 + 4.610e-02 4.505e-02 5.861e-02 5.169e-02 4.976e-02 6.948e-02 + 1.268e-01 1.299e-01 1.149e-01 1.205e-01 1.424e-01 1.753e-01 + 1.838e-01 1.364e-01 8.139e-02 7.694e-02 9.982e-02 1.106e-01 + 1.668e-01 1.403e-01 8.486e-02 8.823e-02 1.749e-01 3.773e-01 + 4.621e-01 3.234e-01 1.263e-01 5.798e-02 5.542e-02 8.572e-02 + 1.205e-01 1.107e-01 5.309e-02 5.155e-02 1.952e-01 4.627e-01 + 4.840e-01 4.019e-01 1.879e-01 3.652e-02 2.543e-02 6.524e-02 + 1.758e-01 1.572e-01 7.125e-02 4.879e-02 1.199e-01 2.759e-01 + 3.344e-01 3.004e-01 1.214e-01 2.796e-02 2.929e-02 9.387e-02 + 2.922e-01 2.764e-01 1.441e-01 5.911e-02 5.120e-02 9.653e-02 + 1.780e-01 1.334e-01 4.269e-02 2.147e-02 4.855e-02 1.718e-01 + 3.836e-01 3.722e-01 2.075e-01 7.865e-02 5.023e-02 7.098e-02 + 1.060e-01 7.885e-02 3.732e-02 2.321e-02 6.134e-02 2.236e-01 + 5.739e-01 5.744e-01 4.740e-01 2.675e-01 1.103e-01 9.551e-02 + 8.207e-02 8.804e-02 8.312e-02 1.047e-01 2.777e-01 4.765e-01 + 4.656e-01 4.690e-01 4.116e-01 2.650e-01 1.047e-01 5.053e-02 + 4.042e-02 4.367e-02 6.594e-02 1.098e-01 2.549e-01 3.994e-01 + 2.714e-01 2.768e-01 2.645e-01 1.826e-01 7.475e-02 3.494e-02 + 4.696e-02 4.271e-02 4.301e-02 8.537e-02 1.869e-01 2.566e-01 + 9.450e-02 9.755e-02 1.198e-01 1.361e-01 8.101e-02 3.380e-02 + 8.251e-02 6.268e-02 4.111e-02 9.425e-02 1.260e-01 1.098e-01 + 5.084e-02 4.802e-02 6.278e-02 1.048e-01 5.916e-02 3.301e-02 + 7.043e-02 5.502e-02 4.041e-02 9.339e-02 8.499e-02 5.674e-02 + 8.482e-02 8.187e-02 8.218e-02 9.071e-02 8.362e-02 6.503e-02 + 5.404e-02 5.713e-02 7.981e-02 1.017e-01 8.830e-02 8.100e-02 + 8.332e-02 1.021e-01 1.202e-01 1.445e-01 1.636e-01 1.131e-01 + 7.714e-02 7.939e-02 1.274e-01 1.681e-01 1.268e-01 8.921e-02 + 1.245e-01 1.271e-01 9.628e-02 8.138e-02 1.326e-01 2.121e-01 + 2.199e-01 1.823e-01 1.266e-01 7.954e-02 7.731e-02 1.037e-01 + 1.296e-01 1.350e-01 9.907e-02 7.922e-02 1.150e-01 2.054e-01 + 2.676e-01 2.344e-01 1.094e-01 5.916e-02 6.951e-02 9.816e-02 + 2.888e-01 2.768e-01 1.611e-01 1.032e-01 1.041e-01 9.620e-02 + 1.350e-01 1.271e-01 8.038e-02 7.046e-02 7.908e-02 1.675e-01 + 4.829e-01 4.671e-01 2.950e-01 1.176e-01 5.665e-02 5.526e-02 + 7.626e-02 7.144e-02 5.246e-02 5.147e-02 1.265e-01 3.253e-01 + 5.812e-01 5.749e-01 4.291e-01 1.850e-01 6.474e-02 7.727e-02 + 8.175e-02 8.397e-02 5.923e-02 6.572e-02 2.201e-01 4.551e-01 + 6.641e-01 6.732e-01 6.654e-01 5.047e-01 1.777e-01 1.154e-01 + 9.467e-02 9.895e-02 1.101e-01 2.871e-01 5.901e-01 6.692e-01 + 6.334e-01 6.411e-01 6.615e-01 5.548e-01 2.405e-01 1.240e-01 + 7.590e-02 9.364e-02 1.346e-01 3.130e-01 5.921e-01 6.563e-01 + 5.450e-01 5.511e-01 5.759e-01 4.462e-01 1.565e-01 7.739e-02 + 7.610e-02 7.611e-02 8.059e-02 2.201e-01 5.069e-01 5.803e-01 + 3.401e-01 3.497e-01 3.607e-01 2.367e-01 6.988e-02 3.650e-02 + 7.555e-02 5.483e-02 3.513e-02 1.269e-01 3.217e-01 3.729e-01 + 1.338e-01 1.319e-01 1.324e-01 1.049e-01 4.063e-02 3.769e-02 + 9.497e-02 6.292e-02 2.468e-02 7.601e-02 1.538e-01 1.457e-01 + 8.936e-02 7.623e-02 6.560e-02 7.324e-02 9.302e-02 1.231e-01 + 1.292e-01 1.141e-01 8.018e-02 7.258e-02 8.428e-02 8.800e-02 + 4.955e-02 4.558e-02 5.071e-02 1.257e-01 3.020e-01 2.731e-01 + 2.022e-01 1.929e-01 2.311e-01 1.714e-01 7.512e-02 5.123e-02 + 9.187e-02 8.875e-02 1.009e-01 1.553e-01 2.678e-01 3.540e-01 + 3.708e-01 3.370e-01 2.557e-01 1.580e-01 1.164e-01 1.027e-01 + 1.791e-01 1.706e-01 1.332e-01 1.564e-01 2.116e-01 3.114e-01 + 4.431e-01 3.893e-01 2.066e-01 1.342e-01 1.050e-01 1.280e-01 + 4.601e-01 4.295e-01 2.674e-01 1.541e-01 1.559e-01 2.199e-01 + 3.063e-01 2.713e-01 1.548e-01 9.629e-02 1.393e-01 3.115e-01 + 6.196e-01 6.041e-01 4.614e-01 1.963e-01 6.974e-02 1.198e-01 + 1.590e-01 1.376e-01 7.068e-02 8.901e-02 2.981e-01 5.268e-01 + 6.639e-01 6.643e-01 5.996e-01 3.413e-01 7.898e-02 8.198e-02 + 9.961e-02 8.221e-02 5.985e-02 1.691e-01 4.950e-01 6.344e-01 + 6.130e-01 6.292e-01 6.668e-01 5.645e-01 2.662e-01 2.033e-01 + 1.915e-01 2.036e-01 1.825e-01 3.981e-01 6.417e-01 6.667e-01 + 6.357e-01 6.451e-01 6.900e-01 6.231e-01 3.488e-01 2.428e-01 + 1.734e-01 2.255e-01 2.176e-01 4.182e-01 6.558e-01 6.868e-01 + 6.423e-01 6.403e-01 6.614e-01 5.629e-01 2.636e-01 1.663e-01 + 1.076e-01 1.349e-01 1.538e-01 3.386e-01 6.215e-01 6.793e-01 + 5.319e-01 5.250e-01 5.034e-01 3.319e-01 1.052e-01 7.555e-02 + 9.453e-02 7.571e-02 5.958e-02 1.794e-01 4.570e-01 5.498e-01 + 2.642e-01 2.496e-01 2.176e-01 1.458e-01 7.501e-02 8.554e-02 + 1.267e-01 9.971e-02 5.278e-02 9.622e-02 2.137e-01 2.626e-01 + 1.434e-01 1.123e-01 9.146e-02 1.011e-01 1.137e-01 1.501e-01 + 1.728e-01 1.469e-01 1.015e-01 8.791e-02 9.878e-02 1.255e-01 + 6.879e-02 5.056e-02 4.898e-02 1.015e-01 1.778e-01 2.431e-01 + 2.715e-01 2.381e-01 1.780e-01 1.259e-01 6.939e-02 5.474e-02 + 8.725e-02 8.716e-02 1.199e-01 1.656e-01 1.966e-01 2.711e-01 + 3.405e-01 3.144e-01 2.217e-01 1.709e-01 1.566e-01 1.071e-01 + 2.392e-01 2.293e-01 2.085e-01 2.448e-01 2.248e-01 2.302e-01 + 2.885e-01 2.780e-01 2.228e-01 2.112e-01 1.694e-01 1.795e-01 + 5.520e-01 5.233e-01 3.962e-01 2.915e-01 2.534e-01 1.975e-01 + 1.729e-01 1.846e-01 2.098e-01 1.873e-01 2.535e-01 4.356e-01 + 6.497e-01 6.378e-01 5.549e-01 3.534e-01 2.064e-01 2.073e-01 + 1.152e-01 1.353e-01 1.735e-01 2.160e-01 4.575e-01 6.085e-01 + 6.338e-01 6.389e-01 6.251e-01 4.533e-01 1.886e-01 2.008e-01 + 1.329e-01 1.433e-01 1.529e-01 3.173e-01 5.923e-01 6.548e-01 diff --git a/tests/01_PW/scf_pbe0_spin2/result.ref b/tests/01_PW/scf_pbe0_spin2/result.ref new file mode 100644 index 00000000000..ee5a18fef04 --- /dev/null +++ b/tests/01_PW/scf_pbe0_spin2/result.ref @@ -0,0 +1,5 @@ +etotref -5531.0865843768351624 +etotperatomref -2765.5432921884 +totalforceref 52.920378 +ComparePot1_pass 0 +totaltimeref 1.94 diff --git a/tests/01_PW/097_PW_PBE0_FM/threshold b/tests/01_PW/scf_pbe0_spin2/threshold similarity index 100% rename from tests/01_PW/097_PW_PBE0_FM/threshold rename to tests/01_PW/scf_pbe0_spin2/threshold diff --git a/tests/01_PW/scf_pbe0_spin2_kpar/INPUT b/tests/01_PW/scf_pbe0_spin2_kpar/INPUT new file mode 100644 index 00000000000..c2304794b06 --- /dev/null +++ b/tests/01_PW/scf_pbe0_spin2_kpar/INPUT @@ -0,0 +1,35 @@ +INPUT_PARAMETERS +#Parameters (System) +suffix autotest +calculation scf +nspin 2 +pseudo_dir ../../PP_ORB +orbital_dir ../../PP_ORB + +#Parameters (PW) +ecutwfc 12 +basis_type pw + +#Parameters (SCF) +scf_thr 1.0e-7 +scf_nmax 100 +smearing_method gaussian +smearing_sigma 0.015 +mixing_type pulay +mixing_ndim 8 +mixing_beta 0.2 + +#Parameters (Hybrid Functional) +dft_functional pbe0 + +#Parameters (Other) +symmetry -1 +init_wfc random +pw_seed 1 +device cpu +exx_thr_type energy +exx_ene_thr 5e-2 + +cal_force 1 +#cal_stress 1 # not implement yet +kpar 2 diff --git a/tests/01_PW/scf_pbe0_spin2_kpar/KPT b/tests/01_PW/scf_pbe0_spin2_kpar/KPT new file mode 100644 index 00000000000..2bce8546f16 --- /dev/null +++ b/tests/01_PW/scf_pbe0_spin2_kpar/KPT @@ -0,0 +1,4 @@ +K_POINTS +0 +Gamma +1 3 1 0 0 0 diff --git a/tests/01_PW/scf_pbe0_spin2_kpar/README b/tests/01_PW/scf_pbe0_spin2_kpar/README new file mode 100644 index 00000000000..ddddc4e4217 --- /dev/null +++ b/tests/01_PW/scf_pbe0_spin2_kpar/README @@ -0,0 +1 @@ +PW PBE0 spin-polarized Fe with uneven k-point pools. diff --git a/tests/01_PW/scf_pbe0_spin2_kpar/STRU b/tests/01_PW/scf_pbe0_spin2_kpar/STRU new file mode 100644 index 00000000000..5077b295590 --- /dev/null +++ b/tests/01_PW/scf_pbe0_spin2_kpar/STRU @@ -0,0 +1,18 @@ +ATOMIC_SPECIES +Fe 55.845 Fe_ONCV_PBE-1.0.upf + +LATTICE_CONSTANT +1.889725989 + +LATTICE_VECTORS + 3.0000000000000000 0.0000000000000000 0.0000000000000000 + 0.0000000000000000 3.0000000000000000 0.0000000000000000 + 0.0000000000000000 0.0000000000000000 3.0000000000000000 +ATOMIC_POSITIONS +Direct + +Fe +1.0 +2 +0.0 0.0 0.03 mag 5 +0.25 0.61 0.5 mag 5 diff --git a/tests/01_PW/scf_pbe0_spin2_kpar/magnetism.ref b/tests/01_PW/scf_pbe0_spin2_kpar/magnetism.ref new file mode 100644 index 00000000000..b64017437c3 --- /dev/null +++ b/tests/01_PW/scf_pbe0_spin2_kpar/magnetism.ref @@ -0,0 +1 @@ +0.666667 0.699096 diff --git a/tests/01_PW/scf_pbe0_spin2_kpar/result.ref b/tests/01_PW/scf_pbe0_spin2_kpar/result.ref new file mode 100644 index 00000000000..b1e8dc81cd6 --- /dev/null +++ b/tests/01_PW/scf_pbe0_spin2_kpar/result.ref @@ -0,0 +1,5 @@ +etotref -5540.4733663613642420 +etotperatomref -2770.2366831807 +CompareMagnetism_pass 0 +totalforceref 87.323908 +totaltimeref 2.38 diff --git a/tests/01_PW/scf_sol_H2/INPUT b/tests/01_PW/scf_sol_H2/INPUT new file mode 100644 index 00000000000..a5ac2b323e2 --- /dev/null +++ b/tests/01_PW/scf_sol_H2/INPUT @@ -0,0 +1,23 @@ +INPUT_PARAMETERS +#Parameters (General) +suffix autotest +pseudo_dir ../../PP_ORB +init_wfc random +pw_seed 1 + +nbands 2 +calculation scf +basis_type pw + +#Parameters (Accuracy) +ecutwfc 20 +scf_nmax 50 +cal_force 1 +#cal_stress 1 # implicit solvation stress is not implemented yet + +#Parameters (Solvation Model) +imp_sol 1 +eb_k 80 +tau 0.000010798 +sigma_k 0.6 +nc_k 0.00037 diff --git a/tests/01_PW/084_PW_sol_H2O/KPT b/tests/01_PW/scf_sol_H2/KPT similarity index 100% rename from tests/01_PW/084_PW_sol_H2O/KPT rename to tests/01_PW/scf_sol_H2/KPT diff --git a/tests/01_PW/scf_sol_H2/README b/tests/01_PW/scf_sol_H2/README new file mode 100644 index 00000000000..940ff4033fc --- /dev/null +++ b/tests/01_PW/scf_sol_H2/README @@ -0,0 +1 @@ +test implicit solvation model correction and force for H2 diff --git a/tests/01_PW/083_PW_sol_H2/STRU b/tests/01_PW/scf_sol_H2/STRU similarity index 100% rename from tests/01_PW/083_PW_sol_H2/STRU rename to tests/01_PW/scf_sol_H2/STRU diff --git a/tests/01_PW/scf_sol_H2/result.ref b/tests/01_PW/scf_sol_H2/result.ref new file mode 100644 index 00000000000..55c5474e2e7 --- /dev/null +++ b/tests/01_PW/scf_sol_H2/result.ref @@ -0,0 +1,6 @@ +etotref -31.8942805407092997 +etotperatomref -15.9471402704 +totalforceref 2.675344 +esolelref -0.8065636756 +esolcavref 0.0238598134 +totaltimeref 1.44 diff --git a/tests/01_PW/scf_sol_H2O/INPUT b/tests/01_PW/scf_sol_H2O/INPUT new file mode 100644 index 00000000000..342da6f2bf1 --- /dev/null +++ b/tests/01_PW/scf_sol_H2O/INPUT @@ -0,0 +1,23 @@ +INPUT_PARAMETERS + +calculation scf +basis_type pw +init_wfc random +pw_seed 1 +cal_force 1 +#cal_stress 1 # implicit solvation stress is not implemented yet +nbands 20 +ecutwfc 20 +scf_nmax 50 + +#Parameters (Solvation Model) +imp_sol 1 +eb_k 80 +tau 0.000010798 +sigma_k 0.6 +nc_k 0.00037 + + +#Parameters (General) +suffix autotest +pseudo_dir ../../PP_ORB diff --git a/tests/01_PW/097_PW_PBE0/KPT b/tests/01_PW/scf_sol_H2O/KPT similarity index 100% rename from tests/01_PW/097_PW_PBE0/KPT rename to tests/01_PW/scf_sol_H2O/KPT diff --git a/tests/01_PW/scf_sol_H2O/README b/tests/01_PW/scf_sol_H2O/README new file mode 100644 index 00000000000..2cd1ec4e712 --- /dev/null +++ b/tests/01_PW/scf_sol_H2O/README @@ -0,0 +1 @@ +test implicit solvation model correction and force for H2O diff --git a/tests/01_PW/084_PW_sol_H2O/STRU b/tests/01_PW/scf_sol_H2O/STRU similarity index 100% rename from tests/01_PW/084_PW_sol_H2O/STRU rename to tests/01_PW/scf_sol_H2O/STRU diff --git a/tests/01_PW/scf_sol_H2O/result.ref b/tests/01_PW/scf_sol_H2O/result.ref new file mode 100644 index 00000000000..6497a667c78 --- /dev/null +++ b/tests/01_PW/scf_sol_H2O/result.ref @@ -0,0 +1,6 @@ +etotref -442.8368758671149408 +etotperatomref -147.6122919557 +totalforceref 16.736445 +esolelref -0.9245824197 +esolcavref 0.0325548859 +totaltimeref 2.64 diff --git a/tests/01_PW/035_PW_15_SO/INPUT b/tests/01_PW/scf_spin4/INPUT similarity index 100% rename from tests/01_PW/035_PW_15_SO/INPUT rename to tests/01_PW/scf_spin4/INPUT diff --git a/tests/08_EXX/07_KP_CR_HSE/KPT b/tests/01_PW/scf_spin4/KPT similarity index 100% rename from tests/08_EXX/07_KP_CR_HSE/KPT rename to tests/01_PW/scf_spin4/KPT diff --git a/tests/01_PW/035_PW_15_SO/README b/tests/01_PW/scf_spin4/README similarity index 100% rename from tests/01_PW/035_PW_15_SO/README rename to tests/01_PW/scf_spin4/README diff --git a/tests/01_PW/scf_spin4/STRU b/tests/01_PW/scf_spin4/STRU new file mode 100644 index 00000000000..71a2da3c729 --- /dev/null +++ b/tests/01_PW/scf_spin4/STRU @@ -0,0 +1,23 @@ +ATOMIC_SPECIES +As 1 As_ONCV_PBE_FR-1.1.upf upf201 +Ga 1 Ga_ONCV_PBE_FR-1.0.upf.txt upf201 + +LATTICE_CONSTANT +1 // add lattice constant, 10.58 ang + +LATTICE_VECTORS +5.34197 5.34197 0.0 +0.0 5.34197 5.34197 +5.34197 0.0 5.34197 +ATOMIC_POSITIONS +Direct //Cartesian or Direct coordinate. + +As +0 +1 +0.2500000 0.2500000 0.25000000 0 0 0 mag 1.7320508 + +Ga //Element Label +0 +1 //number of atom +0.00000 0.00000 0.000000 0 0 0 mag 1.7320508 diff --git a/tests/01_PW/035_PW_15_SO/result.ref b/tests/01_PW/scf_spin4/result.ref similarity index 100% rename from tests/01_PW/035_PW_15_SO/result.ref rename to tests/01_PW/scf_spin4/result.ref diff --git a/tests/01_PW/212_PW_USPP_BPCG/INPUT b/tests/01_PW/scf_uspp_bpcg/INPUT similarity index 100% rename from tests/01_PW/212_PW_USPP_BPCG/INPUT rename to tests/01_PW/scf_uspp_bpcg/INPUT diff --git a/tests/01_PW/069_PW_CR_fix_bc/KPT b/tests/01_PW/scf_uspp_bpcg/KPT similarity index 100% rename from tests/01_PW/069_PW_CR_fix_bc/KPT rename to tests/01_PW/scf_uspp_bpcg/KPT diff --git a/tests/01_PW/212_PW_USPP_BPCG/README b/tests/01_PW/scf_uspp_bpcg/README similarity index 100% rename from tests/01_PW/212_PW_USPP_BPCG/README rename to tests/01_PW/scf_uspp_bpcg/README diff --git a/tests/01_PW/212_PW_USPP_BPCG/STRU b/tests/01_PW/scf_uspp_bpcg/STRU similarity index 100% rename from tests/01_PW/212_PW_USPP_BPCG/STRU rename to tests/01_PW/scf_uspp_bpcg/STRU diff --git a/tests/01_PW/212_PW_USPP_BPCG/result.ref b/tests/01_PW/scf_uspp_bpcg/result.ref similarity index 100% rename from tests/01_PW/212_PW_USPP_BPCG/result.ref rename to tests/01_PW/scf_uspp_bpcg/result.ref diff --git a/tests/01_PW/213_PW_USPP_pchg_wfc/INPUT b/tests/01_PW/scf_uspp_pchg_wfc/INPUT similarity index 100% rename from tests/01_PW/213_PW_USPP_pchg_wfc/INPUT rename to tests/01_PW/scf_uspp_pchg_wfc/INPUT diff --git a/tests/01_PW/213_PW_USPP_pchg_wfc/KPT b/tests/01_PW/scf_uspp_pchg_wfc/KPT similarity index 100% rename from tests/01_PW/213_PW_USPP_pchg_wfc/KPT rename to tests/01_PW/scf_uspp_pchg_wfc/KPT diff --git a/tests/01_PW/213_PW_USPP_pchg_wfc/README b/tests/01_PW/scf_uspp_pchg_wfc/README similarity index 100% rename from tests/01_PW/213_PW_USPP_pchg_wfc/README rename to tests/01_PW/scf_uspp_pchg_wfc/README diff --git a/tests/01_PW/213_PW_USPP_pchg_wfc/STRU b/tests/01_PW/scf_uspp_pchg_wfc/STRU similarity index 100% rename from tests/01_PW/213_PW_USPP_pchg_wfc/STRU rename to tests/01_PW/scf_uspp_pchg_wfc/STRU diff --git a/tests/01_PW/213_PW_USPP_pchg_wfc/result.ref b/tests/01_PW/scf_uspp_pchg_wfc/result.ref similarity index 100% rename from tests/01_PW/213_PW_USPP_pchg_wfc/result.ref rename to tests/01_PW/scf_uspp_pchg_wfc/result.ref diff --git a/tests/02_NAO_Gamma/CASES_CPU.txt b/tests/02_NAO_Gamma/CASES_CPU.txt index 5600c73b844..de87db6ecd4 100644 --- a/tests/02_NAO_Gamma/CASES_CPU.txt +++ b/tests/02_NAO_Gamma/CASES_CPU.txt @@ -8,7 +8,7 @@ md_msst md_msst2 md_out_hk_syns md_out_hk_spin2 -relax_bfgs2 +relax_lbfgs relax_cell relax_old_cg relax_out_hk diff --git a/tests/02_NAO_Gamma/CASES_GPU.txt b/tests/02_NAO_Gamma/CASES_GPU.txt index f5dde8f0b0b..40111029340 100644 --- a/tests/02_NAO_Gamma/CASES_GPU.txt +++ b/tests/02_NAO_Gamma/CASES_GPU.txt @@ -8,7 +8,7 @@ md_msst md_msst2 md_out_hk_syns md_out_hk_spin2 -relax_bfgs2 +relax_lbfgs relax_cell relax_old_cg relax_out_hk diff --git a/tests/02_NAO_Gamma/CMakeLists.txt b/tests/02_NAO_Gamma/CMakeLists.txt index e6a0472a501..a316c4adbc1 100644 --- a/tests/02_NAO_Gamma/CMakeLists.txt +++ b/tests/02_NAO_Gamma/CMakeLists.txt @@ -10,7 +10,7 @@ if(ENABLE_ASAN) else() add_test( NAME 02_NAO_Gamma - COMMAND ${BASH} ../integrate/Autotest.sh -a ${ABACUS_BIN_PATH} -n 4 + COMMAND ${BASH} ../integrate/Autotest.sh -a ${ABACUS_BIN_PATH} -n 4 -j auto WORKING_DIRECTORY ${ABACUS_TEST_DIR}/02_NAO_Gamma ) endif() diff --git a/tests/02_NAO_Gamma/relax_bfgs2/INPUT b/tests/02_NAO_Gamma/relax_bfgs2/INPUT deleted file mode 100644 index 94e0a5fb71b..00000000000 --- a/tests/02_NAO_Gamma/relax_bfgs2/INPUT +++ /dev/null @@ -1,33 +0,0 @@ -INPUT_PARAMETERS -#Parameters (1.General) -suffix autotest -calculation relax - -nbands 8 -symmetry 1 -pseudo_dir ../../PP_ORB -orbital_dir ../../PP_ORB - -#Parameters (2.Iteration) -ecutwfc 20 -scf_thr 1e-9 -scf_nmax 100 - -#Parameters (3.Relaxation) -relax_nmax 2 -cal_force 1 -force_thr_ev 0.01 -relax_method bfgs 2 - -#Parameters (4.Basis) -basis_type lcao - -#Parameters (5.Smearing) -smearing_method gauss -smearing_sigma 0.002 - -#Parameters (6.Mixing) -mixing_type broyden -mixing_beta 0.5 - -gamma_only 1 diff --git a/tests/02_NAO_Gamma/relax_bfgs2/README b/tests/02_NAO_Gamma/relax_bfgs2/README deleted file mode 100644 index feced63f83e..00000000000 --- a/tests/02_NAO_Gamma/relax_bfgs2/README +++ /dev/null @@ -1 +0,0 @@ -test BFGS method with diamond Si, symmetry=on in lcao base set diff --git a/tests/02_NAO_Gamma/relax_bfgs2/result.ref b/tests/02_NAO_Gamma/relax_bfgs2/result.ref deleted file mode 100644 index 72bd54b01fa..00000000000 --- a/tests/02_NAO_Gamma/relax_bfgs2/result.ref +++ /dev/null @@ -1,7 +0,0 @@ -etotref -196.2863059190713 -etotperatomref -98.1431529595 -totalforceref 17.063797 -pointgroupref C_3v -spacegroupref D_3d -nksibzref 1 -totaltimeref 1.1139 diff --git a/tests/02_NAO_Gamma/relax_lbfgs/INPUT b/tests/02_NAO_Gamma/relax_lbfgs/INPUT new file mode 100644 index 00000000000..1d6b37d78b8 --- /dev/null +++ b/tests/02_NAO_Gamma/relax_lbfgs/INPUT @@ -0,0 +1,33 @@ +INPUT_PARAMETERS +#Parameters (1.General) +suffix autotest +calculation relax + +nbands 8 +symmetry 1 +pseudo_dir ../../PP_ORB +orbital_dir ../../PP_ORB + +#Parameters (2.Iteration) +ecutwfc 20 +scf_thr 1e-9 +scf_nmax 100 + +#Parameters (3.Relaxation) +relax_nmax 2 +cal_force 1 +force_thr_ev 0.01 +relax_method lbfgs + +#Parameters (4.Basis) +basis_type lcao + +#Parameters (5.Smearing) +smearing_method gauss +smearing_sigma 0.002 + +#Parameters (6.Mixing) +mixing_type broyden +mixing_beta 0.5 + +gamma_only 1 diff --git a/tests/01_PW/801_PW_LT_sc/KPT b/tests/02_NAO_Gamma/relax_lbfgs/KPT similarity index 100% rename from tests/01_PW/801_PW_LT_sc/KPT rename to tests/02_NAO_Gamma/relax_lbfgs/KPT diff --git a/tests/02_NAO_Gamma/relax_lbfgs/README b/tests/02_NAO_Gamma/relax_lbfgs/README new file mode 100644 index 00000000000..83bf57f9db4 --- /dev/null +++ b/tests/02_NAO_Gamma/relax_lbfgs/README @@ -0,0 +1 @@ +test L-BFGS method with diamond Si, symmetry=on in lcao base set diff --git a/tests/02_NAO_Gamma/relax_bfgs2/STRU b/tests/02_NAO_Gamma/relax_lbfgs/STRU similarity index 100% rename from tests/02_NAO_Gamma/relax_bfgs2/STRU rename to tests/02_NAO_Gamma/relax_lbfgs/STRU diff --git a/tests/02_NAO_Gamma/relax_lbfgs/result.ref b/tests/02_NAO_Gamma/relax_lbfgs/result.ref new file mode 100644 index 00000000000..60c9f3b71d8 --- /dev/null +++ b/tests/02_NAO_Gamma/relax_lbfgs/result.ref @@ -0,0 +1,7 @@ +etotref -196.6218662921184 +etotperatomref -98.3109331461 +totalforceref 0.089652 +pointgroupref C_3v +spacegroupref D_3d +nksibzref 1 +totaltimeref 1.46 diff --git a/tests/02_NAO_Gamma/relax_out_hk_spin2/INPUT b/tests/02_NAO_Gamma/relax_out_hk_spin2/INPUT index a18590c76e8..a9497b2548b 100644 --- a/tests/02_NAO_Gamma/relax_out_hk_spin2/INPUT +++ b/tests/02_NAO_Gamma/relax_out_hk_spin2/INPUT @@ -5,7 +5,7 @@ calculation relax nspin 2 nbands 8 -symmetry 1 +symmetry 0 pseudo_dir ../../PP_ORB orbital_dir ../../PP_ORB @@ -33,3 +33,4 @@ mixing_beta 0.5 gamma_only 1 out_mat_hs 1 5 out_app_flag 1 +out_stru 1 diff --git a/tests/02_NAO_Gamma/relax_out_hk_spin2/STRU b/tests/02_NAO_Gamma/relax_out_hk_spin2/STRU index 086df89083b..83ba175970e 100644 --- a/tests/02_NAO_Gamma/relax_out_hk_spin2/STRU +++ b/tests/02_NAO_Gamma/relax_out_hk_spin2/STRU @@ -18,5 +18,5 @@ Direct Si 0.0 2 -0.00 0.00 0.00 1 1 1 -0.251 0.251 0.251 1 1 1 +0.00 0.00 0.00 1 1 1 mag 0.8333 +0.251 0.251 0.251 1 1 1 mag -0.8333 diff --git a/tests/02_NAO_Gamma/relax_out_hk_spin2/hks1_nao.txt.ref b/tests/02_NAO_Gamma/relax_out_hk_spin2/hks1_nao.txt.ref index 49626df883b..2806b6fce43 100644 --- a/tests/02_NAO_Gamma/relax_out_hk_spin2/hks1_nao.txt.ref +++ b/tests/02_NAO_Gamma/relax_out_hk_spin2/hks1_nao.txt.ref @@ -6,21 +6,21 @@ # columns 8 #------------------------------------------------------------------------ Row 1 - -1.59432e-01 9.24250e-04 -9.24250e-04 -9.24250e-04 -4.72133e-01 -4.06675e-04 4.06675e-04 4.06675e-04 + -1.46465e-01 -5.54985e-02 5.54985e-02 5.54985e-02 -4.63965e-01 2.87499e-03 -2.87499e-03 -2.87499e-03 Row 2 - 1.32251e+00 -9.21838e-04 -9.21838e-04 1.98459e-04 -7.16261e-02 -7.43902e-05 -7.43902e-05 + 1.33713e+00 5.61998e-03 5.61998e-03 -1.82286e-02 -7.47451e-02 -7.87879e-03 -7.87879e-03 Row 3 - 1.32251e+00 9.21838e-04 -1.98459e-04 -7.43902e-05 -7.16261e-02 7.43902e-05 + 1.33713e+00 -5.61998e-03 1.82286e-02 -7.87879e-03 -7.47451e-02 7.87879e-03 Row 4 - 1.32251e+00 -1.98459e-04 -7.43902e-05 7.43902e-05 -7.16261e-02 + 1.33713e+00 1.82286e-02 -7.87879e-03 7.87879e-03 -7.47451e-02 Row 5 - -1.59427e-01 -1.31465e-03 1.31465e-03 1.31465e-03 + -1.45396e-01 -2.16756e-02 2.16756e-02 2.16756e-02 Row 6 - 1.32243e+00 -9.14026e-04 -9.14026e-04 + 1.33895e+00 7.95019e-03 7.95019e-03 Row 7 - 1.32243e+00 9.14026e-04 + 1.33895e+00 -7.95019e-03 Row 8 - 1.32243e+00 + 1.33895e+00 #------------------------------------------------------------------------ # ionic step 2 # filename OUT.autotest/hks1_nao.txt @@ -29,18 +29,18 @@ Row 8 # columns 8 #------------------------------------------------------------------------ Row 1 - -1.46777e-01 -2.23363e-03 2.23363e-03 2.23363e-03 -4.61821e-01 -7.94405e-03 7.94405e-03 7.94405e-03 + -1.28577e-01 4.12179e-03 -4.12179e-03 -4.12179e-03 -4.44719e-01 6.36353e-03 -6.36353e-03 -6.36353e-03 Row 2 - 1.33207e+00 3.84212e-03 3.84212e-03 7.79856e-03 -7.60611e-02 1.44855e-02 1.44855e-02 + 1.34664e+00 7.56506e-04 7.56506e-04 -6.23871e-03 -1.06299e-01 -3.22716e-04 -3.22716e-04 Row 3 - 1.33207e+00 -3.84212e-03 -7.79856e-03 1.44855e-02 -7.60611e-02 -1.44855e-02 + 1.34664e+00 -7.56506e-04 6.23871e-03 -3.22716e-04 -1.06299e-01 3.22716e-04 Row 4 - 1.33207e+00 -7.79856e-03 1.44855e-02 -1.44855e-02 -7.60611e-02 + 1.34664e+00 6.23871e-03 -3.22716e-04 3.22716e-04 -1.06299e-01 Row 5 - -1.46781e-01 2.16502e-03 -2.16502e-03 -2.16502e-03 + -1.28227e-01 -4.00283e-03 4.00283e-03 4.00283e-03 Row 6 - 1.33292e+00 3.21564e-03 3.21564e-03 + 1.34698e+00 -1.55864e-04 -1.55864e-04 Row 7 - 1.33292e+00 -3.21564e-03 + 1.34698e+00 1.55864e-04 Row 8 - 1.33292e+00 + 1.34698e+00 diff --git a/tests/02_NAO_Gamma/relax_out_hk_spin2/hks2_nao.txt.ref b/tests/02_NAO_Gamma/relax_out_hk_spin2/hks2_nao.txt.ref index 3040a28b9a5..2d5a4a8f4b0 100644 --- a/tests/02_NAO_Gamma/relax_out_hk_spin2/hks2_nao.txt.ref +++ b/tests/02_NAO_Gamma/relax_out_hk_spin2/hks2_nao.txt.ref @@ -6,21 +6,21 @@ # columns 8 #------------------------------------------------------------------------ Row 1 - -1.59432e-01 9.24250e-04 -9.24250e-04 -9.24250e-04 -4.72133e-01 -4.06675e-04 4.06675e-04 4.06675e-04 + -1.45415e-01 2.13857e-02 -2.13857e-02 -2.13857e-02 -4.63929e-01 1.80193e-02 -1.80193e-02 -1.80193e-02 Row 2 - 1.32251e+00 -9.21838e-04 -9.21838e-04 1.98459e-04 -7.16261e-02 -7.43902e-05 -7.43902e-05 + 1.33904e+00 7.87864e-03 7.87864e-03 -3.03839e-03 -7.47346e-02 -7.89126e-03 -7.89126e-03 Row 3 - 1.32251e+00 9.21838e-04 -1.98459e-04 -7.43902e-05 -7.16261e-02 7.43902e-05 + 1.33904e+00 -7.87864e-03 3.03839e-03 -7.89126e-03 -7.47346e-02 7.89126e-03 Row 4 - 1.32251e+00 -1.98459e-04 -7.43902e-05 7.43902e-05 -7.16261e-02 + 1.33904e+00 3.03839e-03 -7.89126e-03 7.89126e-03 -7.47346e-02 Row 5 - -1.59427e-01 -1.31465e-03 1.31465e-03 1.31465e-03 + -1.46410e-01 5.51595e-02 -5.51595e-02 -5.51595e-02 Row 6 - 1.32243e+00 -9.14026e-04 -9.14026e-04 + 1.33712e+00 5.59176e-03 5.59176e-03 Row 7 - 1.32243e+00 9.14026e-04 + 1.33712e+00 -5.59176e-03 Row 8 - 1.32243e+00 + 1.33712e+00 #------------------------------------------------------------------------ # ionic step 2 # filename OUT.autotest/hks2_nao.txt @@ -29,18 +29,18 @@ Row 8 # columns 8 #------------------------------------------------------------------------ Row 1 - -1.46777e-01 -2.23363e-03 2.23363e-03 2.23363e-03 -4.61821e-01 -7.94405e-03 7.94405e-03 7.94405e-03 + -1.28577e-01 4.12178e-03 -4.12178e-03 -4.12178e-03 -4.44719e-01 6.36352e-03 -6.36352e-03 -6.36352e-03 Row 2 - 1.33207e+00 3.84212e-03 3.84212e-03 7.79856e-03 -7.60611e-02 1.44855e-02 1.44855e-02 + 1.34664e+00 7.56507e-04 7.56507e-04 -6.23871e-03 -1.06299e-01 -3.22715e-04 -3.22715e-04 Row 3 - 1.33207e+00 -3.84212e-03 -7.79856e-03 1.44855e-02 -7.60611e-02 -1.44855e-02 + 1.34664e+00 -7.56507e-04 6.23871e-03 -3.22715e-04 -1.06299e-01 3.22715e-04 Row 4 - 1.33207e+00 -7.79856e-03 1.44855e-02 -1.44855e-02 -7.60611e-02 + 1.34664e+00 6.23871e-03 -3.22715e-04 3.22715e-04 -1.06299e-01 Row 5 - -1.46781e-01 2.16502e-03 -2.16502e-03 -2.16502e-03 + -1.28227e-01 -4.00285e-03 4.00285e-03 4.00285e-03 Row 6 - 1.33292e+00 3.21564e-03 3.21564e-03 + 1.34698e+00 -1.55864e-04 -1.55864e-04 Row 7 - 1.33292e+00 -3.21564e-03 + 1.34698e+00 1.55864e-04 Row 8 - 1.33292e+00 + 1.34698e+00 diff --git a/tests/02_NAO_Gamma/relax_out_hk_spin2/result.ref b/tests/02_NAO_Gamma/relax_out_hk_spin2/result.ref index 9b76a4a19f7..a678d300c3a 100644 --- a/tests/02_NAO_Gamma/relax_out_hk_spin2/result.ref +++ b/tests/02_NAO_Gamma/relax_out_hk_spin2/result.ref @@ -1,10 +1,7 @@ -etotref -183.772932286494 -etotperatomref -91.8864661432 -totalforceref 23.971608 +etotref -186.6797421072863 +etotperatomref -93.3398710536 +totalforceref 21.160350 CompareH1_pass 0 CompareH2_pass 0 CompareS_pass 0 -pointgroupref C_3v -spacegroupref D_3d -nksibzref 1 -totaltimeref 0.54 +totaltimeref 0.36 diff --git a/tests/02_NAO_Gamma/relax_out_hk_spin2/sk_nao.txt.ref b/tests/02_NAO_Gamma/relax_out_hk_spin2/sk_nao.txt.ref index 59f326ca1ad..2030d7d0ce7 100644 --- a/tests/02_NAO_Gamma/relax_out_hk_spin2/sk_nao.txt.ref +++ b/tests/02_NAO_Gamma/relax_out_hk_spin2/sk_nao.txt.ref @@ -6,7 +6,7 @@ # columns 8 #------------------------------------------------------------------------ Row 1 - 1.15061e+00 -6.50521e-19 0.00000e+00 0.00000e+00 8.22869e-01 9.81584e-04 -9.81584e-04 -9.81584e-04 + 1.15061e+00 1.31188e-17 0.00000e+00 0.00000e+00 8.22869e-01 9.81584e-04 -9.81584e-04 -9.81584e-04 Row 2 1.16565e+00 0.00000e+00 0.00000e+00 -9.81584e-04 -3.19683e-01 -3.40116e-03 -3.40116e-03 Row 3 @@ -14,7 +14,7 @@ Row 3 Row 4 1.16565e+00 9.81584e-04 -3.40116e-03 3.40116e-03 -3.19683e-01 Row 5 - 1.15061e+00 -6.50521e-19 0.00000e+00 0.00000e+00 + 1.15061e+00 1.31188e-17 0.00000e+00 0.00000e+00 Row 6 1.16565e+00 0.00000e+00 0.00000e+00 Row 7 @@ -29,18 +29,18 @@ Row 8 # columns 8 #------------------------------------------------------------------------ Row 1 - 1.15061e+00 -6.50521e-19 1.39379e-20 1.39379e-20 8.23962e-01 -1.82457e-02 1.82457e-02 1.82457e-02 + 1.15061e+00 1.31188e-17 0.00000e+00 0.00000e+00 8.26037e-01 1.61416e-02 -1.61416e-02 -1.61416e-02 Row 2 - 1.16565e+00 -1.26565e-19 -1.26565e-19 1.82457e-02 -3.15301e-01 5.22531e-02 5.22531e-02 + 1.16565e+00 0.00000e+00 0.00000e+00 -1.61416e-02 -2.96923e-01 -1.19879e-01 -1.19879e-01 Row 3 - 1.16565e+00 8.05970e-19 -1.82457e-02 5.22531e-02 -3.15301e-01 -5.22531e-02 + 1.16565e+00 0.00000e+00 1.61416e-02 -1.19879e-01 -2.96923e-01 1.19879e-01 Row 4 - 1.16565e+00 -1.82457e-02 5.22531e-02 -5.22531e-02 -3.15301e-01 + 1.16565e+00 1.61416e-02 -1.19879e-01 1.19879e-01 -2.96923e-01 Row 5 - 1.15061e+00 -6.50521e-19 -2.02733e-20 -2.02733e-20 + 1.15061e+00 1.31188e-17 -6.97944e-18 -3.50999e-18 Row 6 - 1.16565e+00 1.84095e-19 1.84095e-19 + 1.16565e+00 3.68189e-19 -6.57070e-18 Row 7 - 1.16565e+00 -1.17232e-18 + 1.16565e+00 -5.44590e-18 Row 8 1.16565e+00 diff --git a/tests/03_NAO_multik/99_SCF_DPSIN/STRU b/tests/03_NAO_multik/99_SCF_DPSIN/STRU index ae2b187b97b..2597e7ff17f 100644 --- a/tests/03_NAO_multik/99_SCF_DPSIN/STRU +++ b/tests/03_NAO_multik/99_SCF_DPSIN/STRU @@ -17,5 +17,5 @@ Direct Fe 1.0 2 -0.0 0.0 0.0 mag 5 angle1 90 angle2 0 +0.0 0.0 0.0 mag 5 angle1 90 angle2 0 0.5 0.5 0.5 mag 5 angle1 90 angle2 0 diff --git a/tests/03_NAO_multik/CASES_CPU.txt b/tests/03_NAO_multik/CASES_CPU.txt index d6cf9890bda..bd57f5d9d02 100644 --- a/tests/03_NAO_multik/CASES_CPU.txt +++ b/tests/03_NAO_multik/CASES_CPU.txt @@ -3,6 +3,7 @@ scf_pp_upf201 scf_pp_sg15 scf_pp_gth scf_angle_spin4 +scf_gga_grad2_spin4 scf_u_spin1 scf_u_spin2 scf_u_spin4 diff --git a/tests/03_NAO_multik/CMakeLists.txt b/tests/03_NAO_multik/CMakeLists.txt index e563143a4f6..30edcad4ef4 100644 --- a/tests/03_NAO_multik/CMakeLists.txt +++ b/tests/03_NAO_multik/CMakeLists.txt @@ -10,7 +10,7 @@ if(ENABLE_ASAN) else() add_test( NAME 03_NAO_multik - COMMAND ${BASH} ../integrate/Autotest.sh -a ${ABACUS_BIN_PATH} -n 4 + COMMAND ${BASH} ../integrate/Autotest.sh -a ${ABACUS_BIN_PATH} -n 4 -j auto WORKING_DIRECTORY ${ABACUS_TEST_DIR}/03_NAO_multik ) endif() @@ -18,7 +18,7 @@ endif() if(ENABLE_CNPY) add_test( NAME 03_NAO_multik_npz - COMMAND ${BASH} ../integrate/Autotest.sh -a ${ABACUS_BIN_PATH} -n 4 -f CASES_CNPY.txt + COMMAND ${BASH} ../integrate/Autotest.sh -a ${ABACUS_BIN_PATH} -n 4 -j auto -f CASES_CNPY.txt WORKING_DIRECTORY ${ABACUS_TEST_DIR}/03_NAO_multik ) endif() diff --git a/tests/03_NAO_multik/nscf_out_hsr_tr_rr/rr.csr.ref b/tests/03_NAO_multik/nscf_out_hsr_tr_rr/rr.csr.ref deleted file mode 100644 index fdcbb21d7fc..00000000000 --- a/tests/03_NAO_multik/nscf_out_hsr_tr_rr/rr.csr.ref +++ /dev/null @@ -1,120 +0,0 @@ -STEP: 0 -Matrix Dimension of r(R): 6 -Matrix number of r(R): 9 --1 0 0 -11 - -5.4355234064887357e-04 -2.1473170091570490e-03 5.1333474787468233e-03 2.3603825256042605e-03 5.4355234240473422e-04 2.1473170195584670e-03 1.7961611271515325e-07 5.1333474715539017e-03 1.7956089589418311e-07 1.4507105770573264e-10 2.3603825227658699e-03 - 0 2 4 5 1 1 4 1 2 4 1 - 0 4 5 7 7 10 11 -19 - 1.5888453060005491e-03 4.9323612435011127e-02 8.1388085979022338e-03 -2.2447620453835165e-02 -9.4716661059746198e-03 5.0174062294694130e-04 5.0232329754066486e-03 4.9168492671987422e-05 -1.8581691608995785e-04 -8.4518803570646753e-06 3.0979588859972336e-05 1.4268223952885980e-02 1.8581691608996815e-04 -7.0477306448630032e-04 -3.0979585346186800e-05 3.4284954432330173e-03 -8.4518802809157595e-06 3.0979592375406482e-05 3.0979588859972369e-05 - 0 1 2 4 5 1 1 2 4 5 3 1 2 4 5 1 2 4 5 - 0 5 6 10 11 15 19 -21 - 1.0452929644737452e-03 4.9323612429221730e-02 6.5810207848271771e-03 1.5846643248884507e-03 -1.8357922204779761e-02 -6.4500807746596658e-03 1.0452929644737452e-03 6.5810207848271372e-03 4.9168492671987422e-05 8.4518803570646601e-06 -1.8581691608995785e-04 1.5846643240028787e-03 8.4518802809157747e-06 3.0979588859972336e-05 -3.0979592375406671e-05 1.8357922204779782e-02 1.8581691608996815e-04 3.0979585346186658e-05 -7.0477306448630032e-04 6.4500807746596719e-03 3.0979588859972369e-05 - 0 1 2 3 4 5 1 1 2 3 4 1 2 3 4 1 2 3 4 1 5 - 0 6 7 11 15 19 21 --1 1 0 -6 - -2.3092907241019366e-05 5.6690476139248052e-05 -6.8949593514089024e-05 2.3092904543560498e-05 5.6690490396901744e-05 -6.8949608245366054e-05 - 2 4 5 1 1 1 - 0 3 3 4 4 5 6 -7 - 1.3538019843891611e-01 2.0108687062611521e-04 -5.5028589403184538e-04 5.3802677665700438e-04 1.5490105884147449e-04 4.1238669227238920e-04 -4.2464581012085351e-04 - 1 2 4 5 1 1 1 - 0 4 4 5 5 6 7 -9 - 6.7690099222684819e-02 8.8996981692547931e-05 1.2259117374840981e-05 -2.4066815025887816e-04 2.4066815025887816e-04 8.8996981692517492e-05 1.2259117848464283e-05 2.4066815025887764e-04 -2.4066815025887764e-04 - 1 2 3 4 5 1 1 1 1 - 0 5 5 6 7 8 9 -0 -1 0 -19 - 5.0174062294694130e-04 4.9323612435011127e-02 1.5888453060005491e-03 8.1388085979022338e-03 -9.4716661059746198e-03 -2.2447620453835165e-02 5.0232329754066486e-03 4.9168492671987422e-05 -8.4518803570646448e-06 -1.8581691608995785e-04 3.0979588859972336e-05 3.4284954432330173e-03 -8.4518802809157595e-06 3.0979588859972302e-05 3.0979592375406590e-05 1.4268223952885980e-02 1.8581691608996815e-04 -3.0979585346186658e-05 -7.0477306448630032e-04 - 0 0 1 2 4 5 0 2 4 5 3 0 2 4 5 0 2 4 5 - 0 1 6 10 11 15 19 -11 - 5.4355234240473422e-04 -5.4355234064887357e-04 -2.1473170091570490e-03 2.3603825256042605e-03 5.1333474787468233e-03 2.1473170195584670e-03 1.7961611271507193e-07 2.3603825227658699e-03 5.1333474715539017e-03 1.7956089589418311e-07 1.4507105759731243e-10 - 0 1 2 4 5 0 5 0 0 2 5 - 0 1 5 7 7 8 11 -21 - 1.0452929644737452e-03 4.9323612429221730e-02 1.0452929644737452e-03 6.5810207848271771e-03 1.5846643248884507e-03 -6.4500807746596658e-03 -1.8357922204779761e-02 6.5810207848271372e-03 4.9168492671987422e-05 8.4518803570646601e-06 -1.8581691608995785e-04 1.5846643240028787e-03 8.4518802809157747e-06 3.0979588859972336e-05 -3.0979592375406671e-05 6.4500807746596719e-03 3.0979588859972302e-05 1.8357922204779782e-02 1.8581691608996815e-04 3.0979585346186658e-05 -7.0477306448630032e-04 - 0 0 1 2 3 4 5 0 2 3 5 0 2 3 5 0 4 0 2 3 5 - 0 1 7 11 15 17 21 -0 0 -1 -28 - 5.0174062294694130e-04 4.2201824921407663e-05 1.5377364662592626e-04 3.3127570471413757e-05 -3.9981148122774390e-05 1.5888453060005491e-03 7.1945994080535258e-05 2.5754040489733176e-04 -7.3814082924723113e-05 6.6960505273362480e-05 4.2201825777109796e-05 7.1945993224820928e-05 4.9168492671987422e-05 1.8581691608995785e-04 -8.4518803570646482e-06 -1.5377363497737592e-04 -2.5754041654589403e-04 -1.8581691608996815e-04 -7.0477306448630032e-04 3.0979585346186631e-05 3.3127567851668109e-05 -7.3814085544471865e-05 -8.4518802809157866e-06 -3.0979592375406665e-05 3.0979588859972288e-05 -3.9981145094151316e-05 6.6960508301988658e-05 3.0979588859972383e-05 - 0 2 3 4 5 1 2 3 4 5 0 1 2 3 4 0 1 2 3 4 0 1 2 3 4 0 1 5 - 0 5 10 15 20 25 28 -28 - 1.5888453060005491e-03 7.1945994080535258e-05 2.5754040489733176e-04 6.6960505273362480e-05 -7.3814082924723113e-05 5.0174062294694130e-04 4.2201824921407663e-05 1.5377364662592626e-04 -3.9981148122774390e-05 3.3127570471413750e-05 7.1945993224820928e-05 4.2201825777109796e-05 4.9168492671987422e-05 1.8581691608995785e-04 -8.4518803570646719e-06 -2.5754041654589403e-04 -1.5377363497737592e-04 -1.8581691608996815e-04 -7.0477306448630032e-04 3.0979585346186726e-05 6.6960508301988658e-05 -3.9981145094151316e-05 3.0979588859972288e-05 -7.3814085544471851e-05 3.3127567851668122e-05 -8.4518802809157612e-06 -3.0979592375406570e-05 3.0979588859972383e-05 - 0 2 3 4 5 1 2 3 4 5 0 1 2 3 5 0 1 2 3 5 0 1 4 0 1 2 3 5 - 0 5 10 15 20 23 28 -19 - 1.2641580499511582e-07 7.4801010380547432e-07 -1.5874475623656715e-06 1.5874475623656715e-06 1.2641580499511582e-07 7.4801010380547432e-07 1.5874475623656715e-06 -1.5874475623656715e-06 -1.2641251381570050e-07 -1.2641251381570050e-07 -1.7961611271509904e-07 7.4805449706160819e-07 7.4805449706160819e-07 -1.7956089589418311e-07 1.4507105759731243e-10 1.5874359138109447e-06 -1.5874359138109447e-06 -1.5874359138109447e-06 1.5874359138109447e-06 - 2 3 4 5 2 3 4 5 0 1 3 0 1 2 3 0 1 0 1 - 0 4 8 11 15 17 19 -0 0 0 -22 - 2.3999984392235278e+00 3.6107291831173324e-02 5.1482523098694510e-01 3.1693465467544379e-01 -4.3964257418491415e-01 3.6107291835551378e-02 7.5999950575502657e+00 7.5808368502949131e-01 -8.3992721124076430e-01 7.1721929173129384e-01 5.1482523103614264e-01 7.5808368498029377e-01 4.9999997481678466e+00 -6.8174263394568979e-01 4.9999993048414364e+00 3.1693465463238718e-01 -8.3992721128382075e-01 -6.8174263053531514e-01 4.9999993048414364e+00 -4.3964257413808383e-01 7.1721929177812416e-01 4.9999993048414364e+00 - 0 1 2 4 5 0 1 2 4 5 0 1 2 4 3 0 1 2 4 0 1 5 - 0 5 10 14 15 19 22 -22 - 7.5999950575502657e+00 3.6107291835551378e-02 7.5808368502949131e-01 7.1721929173129384e-01 -8.3992721124076430e-01 3.6107291831173324e-02 2.3999984392235278e+00 5.1482523098694510e-01 -4.3964257418491415e-01 3.1693465467544379e-01 7.5808368498029377e-01 5.1482523103614264e-01 4.9999997481678466e+00 -6.8174263394568979e-01 4.9999993048414364e+00 7.1721929177812416e-01 -4.3964257413808383e-01 4.9999993048414364e+00 -8.3992721128382075e-01 3.1693465463238718e-01 -6.8174263053531514e-01 4.9999993048414364e+00 - 0 1 2 4 5 0 1 2 4 5 0 1 2 5 3 0 1 4 0 1 2 5 - 0 5 10 14 15 18 22 -26 - 4.9999967483868968e+00 3.6107291833362351e-02 6.3645445800821820e-01 1.2270791950947033e-01 5.7843093295810399e-01 -5.7843093295810399e-01 3.6107291833362351e-02 4.9999967483868968e+00 6.3645445800821820e-01 1.2270791950947033e-01 -5.7843093295810399e-01 5.7843093295810399e-01 6.3645445800821820e-01 6.3645445800821820e-01 4.9999997481678466e+00 6.8174263394568979e-01 1.2270791950569672e-01 1.2270791950569672e-01 6.8174263053531514e-01 4.9999993048414364e+00 5.7843093295810399e-01 -5.7843093295810399e-01 4.9999993048414364e+00 -5.7843093295810399e-01 5.7843093295810399e-01 4.9999993048414364e+00 - 0 1 2 3 4 5 0 1 2 3 4 5 0 1 2 3 0 1 2 3 0 1 4 0 1 5 - 0 6 12 16 20 23 26 -0 0 1 -28 - 5.0174062294694130e-04 4.2201824921407663e-05 -1.5377364662592626e-04 3.3127570471413757e-05 -3.9981148122774390e-05 1.5888453060005491e-03 7.1945994080535258e-05 -2.5754040489733176e-04 -7.3814082924723113e-05 6.6960505273362480e-05 4.2201825777109796e-05 7.1945993224820928e-05 4.9168492671987422e-05 -1.8581691608995785e-04 -8.4518803570646482e-06 1.5377363497737592e-04 2.5754041654589403e-04 1.8581691608996815e-04 -7.0477306448630032e-04 -3.0979585346186631e-05 3.3127567851668109e-05 -7.3814085544471865e-05 -8.4518802809157866e-06 3.0979592375406665e-05 3.0979588859972288e-05 -3.9981145094151316e-05 6.6960508301988658e-05 3.0979588859972383e-05 - 0 2 3 4 5 1 2 3 4 5 0 1 2 3 4 0 1 2 3 4 0 1 2 3 4 0 1 5 - 0 5 10 15 20 25 28 -28 - 1.5888453060005491e-03 7.1945994080535258e-05 -2.5754040489733176e-04 6.6960505273362480e-05 -7.3814082924723113e-05 5.0174062294694130e-04 4.2201824921407663e-05 -1.5377364662592626e-04 -3.9981148122774390e-05 3.3127570471413750e-05 7.1945993224820928e-05 4.2201825777109796e-05 4.9168492671987422e-05 -1.8581691608995785e-04 -8.4518803570646719e-06 2.5754041654589403e-04 1.5377363497737592e-04 1.8581691608996815e-04 -7.0477306448630032e-04 -3.0979585346186726e-05 6.6960508301988658e-05 -3.9981145094151316e-05 3.0979588859972288e-05 -7.3814085544471851e-05 3.3127567851668122e-05 -8.4518802809157612e-06 3.0979592375406570e-05 3.0979588859972383e-05 - 0 2 3 4 5 1 2 3 4 5 0 1 2 3 5 0 1 2 3 5 0 1 4 0 1 2 3 5 - 0 5 10 15 20 23 28 -24 - 2.0905859280695603e-03 1.1402140319694779e-04 -4.1056604141945252e-04 1.0852910095850255e-04 -1.0852910095850255e-04 2.0905859280695603e-03 1.1402140319694779e-04 -4.1056604141945252e-04 -1.0852910095850255e-04 1.0852910095850255e-04 1.1427423151574642e-04 1.1427423151574642e-04 9.8336985054881158e-05 -3.7181344829263080e-04 4.1206210602033155e-04 4.1206210602033155e-04 3.7145427128404212e-04 -1.4095462740436582e-03 1.0535421748232903e-04 -1.0535421748232903e-04 6.1959174206158919e-05 -1.0535421748232903e-04 1.0535421748232903e-04 6.1959174206159109e-05 - 0 2 3 4 5 1 2 3 4 5 0 1 2 3 0 1 2 3 0 1 4 0 1 5 - 0 5 10 14 18 21 24 -0 1 0 -19 - 5.0174062294694130e-04 4.9323612423432341e-02 5.0232329717521222e-03 3.4284954433447105e-03 1.4268223955724356e-02 1.5888453060005491e-03 8.1388085942476259e-03 4.9168492671987422e-05 -8.4518803570646448e-06 1.8581691608995785e-04 3.0979588859972336e-05 -9.4716661060863273e-03 -8.4518802809157595e-06 3.0979588859972302e-05 -3.0979592375406590e-05 -2.2447620456673582e-02 -1.8581691608996815e-04 3.0979585346186658e-05 -7.0477306448630032e-04 - 0 1 2 4 5 1 1 2 4 5 3 1 2 4 5 1 2 4 5 - 0 5 6 10 11 15 19 -15 - 2.6341382695963637e-03 9.8647224853099402e-02 1.5309358578811404e-02 1.5260544074923591e-02 4.1849191888306346e-02 1.5470335865427562e-03 1.1014724550095807e-02 9.8336985054881158e-05 3.7181344829263080e-04 6.1959174206158987e-05 -1.0539779026553474e-02 6.1959174206158960e-05 -3.1582496938005661e-02 -3.7145427128404212e-04 -1.4095462740436582e-03 - 0 1 2 4 5 1 1 2 5 3 1 4 1 2 5 - 0 5 6 9 10 12 15 -21 - 1.0452929644737452e-03 4.9323612429221730e-02 6.5810207848271771e-03 1.5846643248884507e-03 6.4500807746596658e-03 1.8357922204779761e-02 1.0452929644737452e-03 6.5810207848271372e-03 4.9168492671987422e-05 8.4518803570646601e-06 1.8581691608995785e-04 1.5846643240028787e-03 8.4518802809157747e-06 3.0979588859972336e-05 3.0979592375406671e-05 -6.4500807746596719e-03 3.0979588859972302e-05 -1.8357922204779782e-02 -1.8581691608996815e-04 -3.0979585346186658e-05 -7.0477306448630032e-04 - 0 1 2 3 4 5 1 1 2 3 5 1 2 3 5 1 4 1 2 3 5 - 0 6 7 11 15 17 21 -1 -1 0 -7 - 1.3538019843891611e-01 2.0108687062611521e-04 5.3802677665700438e-04 -5.5028589403184538e-04 1.5490105884147449e-04 -4.2464581012085351e-04 4.1238669227238920e-04 - 0 2 4 5 0 0 0 - 0 0 4 5 5 6 7 -6 - -2.3092907241019366e-05 -6.8949593514089024e-05 5.6690476139247998e-05 2.3092904543560498e-05 -6.8949608245366054e-05 5.6690490396901799e-05 - 2 4 5 0 0 0 - 0 0 3 4 4 5 6 -9 - 6.7690099222684819e-02 8.8996981692547931e-05 1.2259117374840981e-05 2.4066815025887816e-04 -2.4066815025887816e-04 8.8996981692517492e-05 1.2259117848464283e-05 -2.4066815025887764e-04 2.4066815025887764e-04 - 0 2 3 4 5 0 0 0 0 - 0 0 5 6 7 8 9 -1 0 0 -15 - 1.5470335865427562e-03 9.8647224853099402e-02 2.6341382695963637e-03 1.5309358578811404e-02 4.1849191888306346e-02 1.5260544074923591e-02 1.1014724550095807e-02 9.8336985054881158e-05 3.7181344829263085e-04 6.1959174206158987e-05 -3.1582496938005661e-02 -3.7145427128404212e-04 -1.4095462740436582e-03 -1.0539779026553474e-02 6.1959174206159176e-05 - 0 0 1 2 4 5 0 2 4 3 0 2 4 0 5 - 0 1 6 9 10 13 15 -19 - 1.5888453060005491e-03 4.9323612423432341e-02 5.0174062294694130e-04 5.0232329717521222e-03 1.4268223955724356e-02 3.4284954433447105e-03 8.1388085942476259e-03 4.9168492671987422e-05 1.8581691608995785e-04 -8.4518803570646753e-06 3.0979588859972336e-05 -2.2447620456673582e-02 -1.8581691608996815e-04 -7.0477306448630032e-04 3.0979585346186800e-05 -9.4716661060863273e-03 -8.4518802809157595e-06 -3.0979592375406482e-05 3.0979588859972369e-05 - 0 0 1 2 4 5 0 2 4 5 3 0 2 4 5 0 2 4 5 - 0 1 6 10 11 15 19 -21 - 1.0452929644737452e-03 4.9323612429221730e-02 1.0452929644737452e-03 6.5810207848271771e-03 1.5846643248884507e-03 1.8357922204779761e-02 6.4500807746596658e-03 6.5810207848271372e-03 4.9168492671987422e-05 8.4518803570646601e-06 1.8581691608995785e-04 1.5846643240028787e-03 8.4518802809157747e-06 3.0979588859972336e-05 3.0979592375406671e-05 -1.8357922204779782e-02 -1.8581691608996815e-04 -3.0979585346186658e-05 -7.0477306448630032e-04 -6.4500807746596719e-03 3.0979588859972369e-05 - 0 0 1 2 3 4 5 0 2 3 4 0 2 3 4 0 2 3 4 0 5 - 0 1 7 11 15 19 21 diff --git a/tests/03_NAO_multik/nscf_out_hsr_tr_rr/rr_nao.txt.ref b/tests/03_NAO_multik/nscf_out_hsr_tr_rr/rr_nao.txt.ref new file mode 100644 index 00000000000..db8fe0b5461 --- /dev/null +++ b/tests/03_NAO_multik/nscf_out_hsr_tr_rr/rr_nao.txt.ref @@ -0,0 +1,120 @@ +STEP: 0 +Matrix Dimension of r(R): 6 +Matrix number of r(R): 9 +-1 0 0 +11 + -5.43552341e-04 -2.14731701e-03 5.13334748e-03 2.36038253e-03 5.43552342e-04 2.14731702e-03 1.79616113e-07 5.13334747e-03 1.79560896e-07 1.45071058e-10 2.36038252e-03 + 0 2 4 5 1 1 4 1 2 4 1 + 0 4 5 7 7 10 11 +19 + 1.58884531e-03 4.93236124e-02 8.13880860e-03 -2.24476205e-02 -9.47166611e-03 5.01740623e-04 5.02323298e-03 4.91684927e-05 -1.85816916e-04 -8.45188036e-06 3.09795889e-05 1.42682240e-02 1.85816916e-04 -7.04773064e-04 -3.09795853e-05 3.42849544e-03 -8.45188028e-06 3.09795924e-05 3.09795889e-05 + 0 1 2 4 5 1 1 2 4 5 3 1 2 4 5 1 2 4 5 + 0 5 6 10 11 15 19 +21 + 1.04529296e-03 4.93236124e-02 6.58102078e-03 1.58466432e-03 -1.83579222e-02 -6.45008077e-03 1.04529296e-03 6.58102078e-03 4.91684927e-05 8.45188036e-06 -1.85816916e-04 1.58466432e-03 8.45188028e-06 3.09795889e-05 -3.09795924e-05 1.83579222e-02 1.85816916e-04 3.09795853e-05 -7.04773064e-04 6.45008077e-03 3.09795889e-05 + 0 1 2 3 4 5 1 1 2 3 4 1 2 3 4 1 2 3 4 1 5 + 0 6 7 11 15 19 21 +-1 1 0 +6 + -2.30929072e-05 5.66904761e-05 -6.89495935e-05 2.30929045e-05 5.66904904e-05 -6.89496082e-05 + 2 4 5 1 1 1 + 0 3 3 4 4 5 6 +7 + 1.35380198e-01 2.01086871e-04 -5.50285894e-04 5.38026777e-04 1.54901059e-04 4.12386692e-04 -4.24645810e-04 + 1 2 4 5 1 1 1 + 0 4 4 5 5 6 7 +9 + 6.76900992e-02 8.89969817e-05 1.22591174e-05 -2.40668150e-04 2.40668150e-04 8.89969817e-05 1.22591178e-05 2.40668150e-04 -2.40668150e-04 + 1 2 3 4 5 1 1 1 1 + 0 5 5 6 7 8 9 +0 -1 0 +19 + 5.01740623e-04 4.93236124e-02 1.58884531e-03 8.13880860e-03 -9.47166611e-03 -2.24476205e-02 5.02323298e-03 4.91684927e-05 -8.45188036e-06 -1.85816916e-04 3.09795889e-05 3.42849544e-03 -8.45188028e-06 3.09795889e-05 3.09795924e-05 1.42682240e-02 1.85816916e-04 -3.09795853e-05 -7.04773064e-04 + 0 0 1 2 4 5 0 2 4 5 3 0 2 4 5 0 2 4 5 + 0 1 6 10 11 15 19 +11 + 5.43552342e-04 -5.43552341e-04 -2.14731701e-03 2.36038253e-03 5.13334748e-03 2.14731702e-03 1.79616113e-07 2.36038252e-03 5.13334747e-03 1.79560896e-07 1.45071058e-10 + 0 1 2 4 5 0 5 0 0 2 5 + 0 1 5 7 7 8 11 +21 + 1.04529296e-03 4.93236124e-02 1.04529296e-03 6.58102078e-03 1.58466432e-03 -6.45008077e-03 -1.83579222e-02 6.58102078e-03 4.91684927e-05 8.45188036e-06 -1.85816916e-04 1.58466432e-03 8.45188028e-06 3.09795889e-05 -3.09795924e-05 6.45008077e-03 3.09795889e-05 1.83579222e-02 1.85816916e-04 3.09795853e-05 -7.04773064e-04 + 0 0 1 2 3 4 5 0 2 3 5 0 2 3 5 0 4 0 2 3 5 + 0 1 7 11 15 17 21 +0 0 -1 +28 + 5.01740623e-04 4.22018249e-05 1.53773647e-04 3.31275705e-05 -3.99811481e-05 1.58884531e-03 7.19459941e-05 2.57540405e-04 -7.38140829e-05 6.69605053e-05 4.22018258e-05 7.19459932e-05 4.91684927e-05 1.85816916e-04 -8.45188036e-06 -1.53773635e-04 -2.57540417e-04 -1.85816916e-04 -7.04773064e-04 3.09795853e-05 3.31275679e-05 -7.38140855e-05 -8.45188028e-06 -3.09795924e-05 3.09795889e-05 -3.99811451e-05 6.69605083e-05 3.09795889e-05 + 0 2 3 4 5 1 2 3 4 5 0 1 2 3 4 0 1 2 3 4 0 1 2 3 4 0 1 5 + 0 5 10 15 20 25 28 +28 + 1.58884531e-03 7.19459941e-05 2.57540405e-04 6.69605053e-05 -7.38140829e-05 5.01740623e-04 4.22018249e-05 1.53773647e-04 -3.99811481e-05 3.31275705e-05 7.19459932e-05 4.22018258e-05 4.91684927e-05 1.85816916e-04 -8.45188036e-06 -2.57540417e-04 -1.53773635e-04 -1.85816916e-04 -7.04773064e-04 3.09795853e-05 6.69605083e-05 -3.99811451e-05 3.09795889e-05 -7.38140855e-05 3.31275679e-05 -8.45188028e-06 -3.09795924e-05 3.09795889e-05 + 0 2 3 4 5 1 2 3 4 5 0 1 2 3 5 0 1 2 3 5 0 1 4 0 1 2 3 5 + 0 5 10 15 20 23 28 +19 + 1.26415805e-07 7.48010104e-07 -1.58744756e-06 1.58744756e-06 1.26415805e-07 7.48010104e-07 1.58744756e-06 -1.58744756e-06 -1.26412514e-07 -1.26412514e-07 -1.79616113e-07 7.48054497e-07 7.48054497e-07 -1.79560896e-07 1.45071058e-10 1.58743591e-06 -1.58743591e-06 -1.58743591e-06 1.58743591e-06 + 2 3 4 5 2 3 4 5 0 1 3 0 1 2 3 0 1 0 1 + 0 4 8 11 15 17 19 +0 0 0 +22 + 2.39999844e+00 3.61072918e-02 5.14825231e-01 3.16934655e-01 -4.39642574e-01 3.61072918e-02 7.59999506e+00 7.58083685e-01 -8.39927211e-01 7.17219292e-01 5.14825231e-01 7.58083685e-01 4.99999975e+00 -6.81742634e-01 4.99999930e+00 3.16934655e-01 -8.39927211e-01 -6.81742631e-01 4.99999930e+00 -4.39642574e-01 7.17219292e-01 4.99999930e+00 + 0 1 2 4 5 0 1 2 4 5 0 1 2 4 3 0 1 2 4 0 1 5 + 0 5 10 14 15 19 22 +22 + 7.59999506e+00 3.61072918e-02 7.58083685e-01 7.17219292e-01 -8.39927211e-01 3.61072918e-02 2.39999844e+00 5.14825231e-01 -4.39642574e-01 3.16934655e-01 7.58083685e-01 5.14825231e-01 4.99999975e+00 -6.81742634e-01 4.99999930e+00 7.17219292e-01 -4.39642574e-01 4.99999930e+00 -8.39927211e-01 3.16934655e-01 -6.81742631e-01 4.99999930e+00 + 0 1 2 4 5 0 1 2 4 5 0 1 2 5 3 0 1 4 0 1 2 5 + 0 5 10 14 15 18 22 +26 + 4.99999675e+00 3.61072918e-02 6.36454458e-01 1.22707920e-01 5.78430933e-01 -5.78430933e-01 3.61072918e-02 4.99999675e+00 6.36454458e-01 1.22707920e-01 -5.78430933e-01 5.78430933e-01 6.36454458e-01 6.36454458e-01 4.99999975e+00 6.81742634e-01 1.22707920e-01 1.22707920e-01 6.81742631e-01 4.99999930e+00 5.78430933e-01 -5.78430933e-01 4.99999930e+00 -5.78430933e-01 5.78430933e-01 4.99999930e+00 + 0 1 2 3 4 5 0 1 2 3 4 5 0 1 2 3 0 1 2 3 0 1 4 0 1 5 + 0 6 12 16 20 23 26 +0 0 1 +28 + 5.01740623e-04 4.22018249e-05 -1.53773647e-04 3.31275705e-05 -3.99811481e-05 1.58884531e-03 7.19459941e-05 -2.57540405e-04 -7.38140829e-05 6.69605053e-05 4.22018258e-05 7.19459932e-05 4.91684927e-05 -1.85816916e-04 -8.45188036e-06 1.53773635e-04 2.57540417e-04 1.85816916e-04 -7.04773064e-04 -3.09795853e-05 3.31275679e-05 -7.38140855e-05 -8.45188028e-06 3.09795924e-05 3.09795889e-05 -3.99811451e-05 6.69605083e-05 3.09795889e-05 + 0 2 3 4 5 1 2 3 4 5 0 1 2 3 4 0 1 2 3 4 0 1 2 3 4 0 1 5 + 0 5 10 15 20 25 28 +28 + 1.58884531e-03 7.19459941e-05 -2.57540405e-04 6.69605053e-05 -7.38140829e-05 5.01740623e-04 4.22018249e-05 -1.53773647e-04 -3.99811481e-05 3.31275705e-05 7.19459932e-05 4.22018258e-05 4.91684927e-05 -1.85816916e-04 -8.45188036e-06 2.57540417e-04 1.53773635e-04 1.85816916e-04 -7.04773064e-04 -3.09795853e-05 6.69605083e-05 -3.99811451e-05 3.09795889e-05 -7.38140855e-05 3.31275679e-05 -8.45188028e-06 3.09795924e-05 3.09795889e-05 + 0 2 3 4 5 1 2 3 4 5 0 1 2 3 5 0 1 2 3 5 0 1 4 0 1 2 3 5 + 0 5 10 15 20 23 28 +24 + 2.09058593e-03 1.14021403e-04 -4.10566041e-04 1.08529101e-04 -1.08529101e-04 2.09058593e-03 1.14021403e-04 -4.10566041e-04 -1.08529101e-04 1.08529101e-04 1.14274232e-04 1.14274232e-04 9.83369851e-05 -3.71813448e-04 4.12062106e-04 4.12062106e-04 3.71454271e-04 -1.40954627e-03 1.05354217e-04 -1.05354217e-04 6.19591742e-05 -1.05354217e-04 1.05354217e-04 6.19591742e-05 + 0 2 3 4 5 1 2 3 4 5 0 1 2 3 0 1 2 3 0 1 4 0 1 5 + 0 5 10 14 18 21 24 +0 1 0 +19 + 5.01740623e-04 4.93236124e-02 5.02323297e-03 3.42849544e-03 1.42682240e-02 1.58884531e-03 8.13880859e-03 4.91684927e-05 -8.45188036e-06 1.85816916e-04 3.09795889e-05 -9.47166611e-03 -8.45188028e-06 3.09795889e-05 -3.09795924e-05 -2.24476205e-02 -1.85816916e-04 3.09795853e-05 -7.04773064e-04 + 0 1 2 4 5 1 1 2 4 5 3 1 2 4 5 1 2 4 5 + 0 5 6 10 11 15 19 +15 + 2.63413827e-03 9.86472249e-02 1.53093586e-02 1.52605441e-02 4.18491919e-02 1.54703359e-03 1.10147246e-02 9.83369851e-05 3.71813448e-04 6.19591742e-05 -1.05397790e-02 6.19591742e-05 -3.15824969e-02 -3.71454271e-04 -1.40954627e-03 + 0 1 2 4 5 1 1 2 5 3 1 4 1 2 5 + 0 5 6 9 10 12 15 +21 + 1.04529296e-03 4.93236124e-02 6.58102078e-03 1.58466432e-03 6.45008077e-03 1.83579222e-02 1.04529296e-03 6.58102078e-03 4.91684927e-05 8.45188036e-06 1.85816916e-04 1.58466432e-03 8.45188028e-06 3.09795889e-05 3.09795924e-05 -6.45008077e-03 3.09795889e-05 -1.83579222e-02 -1.85816916e-04 -3.09795853e-05 -7.04773064e-04 + 0 1 2 3 4 5 1 1 2 3 5 1 2 3 5 1 4 1 2 3 5 + 0 6 7 11 15 17 21 +1 -1 0 +7 + 1.35380198e-01 2.01086871e-04 5.38026777e-04 -5.50285894e-04 1.54901059e-04 -4.24645810e-04 4.12386692e-04 + 0 2 4 5 0 0 0 + 0 0 4 5 5 6 7 +6 + -2.30929072e-05 -6.89495935e-05 5.66904761e-05 2.30929045e-05 -6.89496082e-05 5.66904904e-05 + 2 4 5 0 0 0 + 0 0 3 4 4 5 6 +9 + 6.76900992e-02 8.89969817e-05 1.22591174e-05 2.40668150e-04 -2.40668150e-04 8.89969817e-05 1.22591178e-05 -2.40668150e-04 2.40668150e-04 + 0 2 3 4 5 0 0 0 0 + 0 0 5 6 7 8 9 +1 0 0 +15 + 1.54703359e-03 9.86472249e-02 2.63413827e-03 1.53093586e-02 4.18491919e-02 1.52605441e-02 1.10147246e-02 9.83369851e-05 3.71813448e-04 6.19591742e-05 -3.15824969e-02 -3.71454271e-04 -1.40954627e-03 -1.05397790e-02 6.19591742e-05 + 0 0 1 2 4 5 0 2 4 3 0 2 4 0 5 + 0 1 6 9 10 13 15 +19 + 1.58884531e-03 4.93236124e-02 5.01740623e-04 5.02323297e-03 1.42682240e-02 3.42849544e-03 8.13880859e-03 4.91684927e-05 1.85816916e-04 -8.45188036e-06 3.09795889e-05 -2.24476205e-02 -1.85816916e-04 -7.04773064e-04 3.09795853e-05 -9.47166611e-03 -8.45188028e-06 -3.09795924e-05 3.09795889e-05 + 0 0 1 2 4 5 0 2 4 5 3 0 2 4 5 0 2 4 5 + 0 1 6 10 11 15 19 +21 + 1.04529296e-03 4.93236124e-02 1.04529296e-03 6.58102078e-03 1.58466432e-03 1.83579222e-02 6.45008077e-03 6.58102078e-03 4.91684927e-05 8.45188036e-06 1.85816916e-04 1.58466432e-03 8.45188028e-06 3.09795889e-05 3.09795924e-05 -1.83579222e-02 -1.85816916e-04 -3.09795853e-05 -7.04773064e-04 -6.45008077e-03 3.09795889e-05 + 0 0 1 2 3 4 5 0 2 3 4 0 2 3 4 0 2 3 4 0 5 + 0 1 7 11 15 19 21 diff --git a/tests/03_NAO_multik/nscf_out_hsr_tr_rr/trs1_nao.csr.ref b/tests/03_NAO_multik/nscf_out_hsr_tr_rr/tr_nao.csr.ref similarity index 100% rename from tests/03_NAO_multik/nscf_out_hsr_tr_rr/trs1_nao.csr.ref rename to tests/03_NAO_multik/nscf_out_hsr_tr_rr/tr_nao.csr.ref diff --git a/tests/03_NAO_multik/scf_deltaspin2/result.ref b/tests/03_NAO_multik/scf_deltaspin2/result.ref index fc858235e90..ea1d55e48c8 100644 --- a/tests/03_NAO_multik/scf_deltaspin2/result.ref +++ b/tests/03_NAO_multik/scf_deltaspin2/result.ref @@ -1,4 +1,5 @@ -etotref -6762.435776188675 -etotperatomref -3381.217888094338 -totalforceref 63.230574 -totalstressref 2916.957427 +etotref -6789.4013179079911424 +etotperatomref -3394.7006589540 +totalforceref 67.888296 +totalstressref 6225.500296 +totaltimeref 2.43 diff --git a/tests/03_NAO_multik/scf_gga_grad2_spin4/INPUT b/tests/03_NAO_multik/scf_gga_grad2_spin4/INPUT new file mode 100644 index 00000000000..3e2a0afcc54 --- /dev/null +++ b/tests/03_NAO_multik/scf_gga_grad2_spin4/INPUT @@ -0,0 +1,35 @@ +INPUT_PARAMETERS +#Parameters (1.General) +suffix autotest +calculation scf + +symmetry 0 +pseudo_dir ../../PP_ORB +orbital_dir ../../PP_ORB +gamma_only 0 + +#Parameters (2.Iteration) +ecutwfc 15 +scf_thr 1e-11 +scf_nmax 100 + +noncolin 1 +nspin 4 +dft_functional pbe +gga_grad 2 +#Parameters (3.Basis) +basis_type lcao +ks_solver scalapack_gvx + +#Parameters (4.Smearing) +smearing_method gauss +smearing_sigma 0.02 + +#Parameters (5.Mixing) +mixing_type broyden +mixing_beta 0.7 +mixing_ndim 15 +mixing_gg0 1.0 + +cal_force 1 +cal_stress 1 diff --git a/tests/08_EXX/06_KP_MD_HSE/KPT b/tests/03_NAO_multik/scf_gga_grad2_spin4/KPT similarity index 100% rename from tests/08_EXX/06_KP_MD_HSE/KPT rename to tests/03_NAO_multik/scf_gga_grad2_spin4/KPT diff --git a/tests/03_NAO_multik/scf_gga_grad2_spin4/README b/tests/03_NAO_multik/scf_gga_grad2_spin4/README new file mode 100644 index 00000000000..82b3862c15d --- /dev/null +++ b/tests/03_NAO_multik/scf_gga_grad2_spin4/README @@ -0,0 +1,9 @@ +PBE noncollinear LCAO regression for gga_grad=2, including forces and stress. +Derived from scf_angle_spin4; pseudopotential and orbitals reuse tests/PP_ORB. +Uses scalapack_gvx and scf_thr=1e-11 for parallel reference comparisons. + +From tests/03_NAO_multik, run: + OMP_NUM_THREADS=1 bash ../integrate/Autotest.sh -a /path/to/abacus -n 4 -r '^scf_gga_grad2_spin4$' + +Generate result.ref with the same command using -n 1 -g. Default energy, +force and stress tolerances apply; no threshold overrides are used. diff --git a/tests/03_NAO_multik/scf_gga_grad2_spin4/STRU b/tests/03_NAO_multik/scf_gga_grad2_spin4/STRU new file mode 100644 index 00000000000..1df1e78433d --- /dev/null +++ b/tests/03_NAO_multik/scf_gga_grad2_spin4/STRU @@ -0,0 +1,22 @@ +ATOMIC_SPECIES +Fe 1.000 Fe_ONCV_PBE-1.0.upf + +NUMERICAL_ORBITAL +Fe_gga_6au_100Ry_4s2p2d1f.orb + +LATTICE_CONSTANT +7 + +LATTICE_VECTORS +1.00 0.00 0.00 +0.00 1.00 0.00 +0.00 0.00 1.00 + +ATOMIC_POSITIONS +Direct + +Fe +0.0 +2 +0.00 0.00 0.00 1 1 1 mag 1.0 angle1 90 angle2 0 +0.50 0.50 0.50 1 1 1 mag 1.0 angle1 90 angle2 180 diff --git a/tests/03_NAO_multik/scf_gga_grad2_spin4/result.ref b/tests/03_NAO_multik/scf_gga_grad2_spin4/result.ref new file mode 100644 index 00000000000..edccf99a269 --- /dev/null +++ b/tests/03_NAO_multik/scf_gga_grad2_spin4/result.ref @@ -0,0 +1,5 @@ +etotref -6267.5006966712917347 +etotperatomref -3133.7503483356 +totalforceref 0.000000 +totalstressref 3941.107233 +totaltimeref 5.57 diff --git a/tests/03_NAO_multik/scf_out_chg_pot1/chg.cube.ref b/tests/03_NAO_multik/scf_out_chg_pot1/chg.cube.ref index 49b6c6e43da..b2ecfc2f87e 100644 --- a/tests/03_NAO_multik/scf_out_chg_pot1/chg.cube.ref +++ b/tests/03_NAO_multik/scf_out_chg_pot1/chg.cube.ref @@ -6,15 +6,15 @@ Ionic_Step 1 Cubefile created from ABACUS. Inner loop is z, followed by y and x 8 0.000000 0.000000 0.662500 33 15.000000 1.325000 1.325000 1.325000 31 13.000000 0.000000 0.000000 0.000000 - 5.58354e-03 2.18221e+00 2.07562e-01 1.19126e-01 7.67323e-02 6.02807e-02 + 5.58353e-03 2.18221e+00 2.07562e-01 1.19126e-01 7.67323e-02 6.02807e-02 1.57466e-01 2.19572e+00 2.18221e+00 1.16536e+00 2.37646e-01 1.64234e-01 1.16112e-01 6.83896e-02 1.27907e-01 1.06722e+00 2.07562e-01 2.37646e-01 2.28880e-01 1.63372e-01 1.13633e-01 5.78449e-02 7.06533e-02 1.39007e-01 - 1.19126e-01 1.64234e-01 1.63372e-01 1.32380e-01 8.36268e-02 4.41108e-02 + 1.19126e-01 1.64234e-01 1.63372e-01 1.32380e-01 8.36269e-02 4.41108e-02 4.11865e-02 6.42572e-02 - 7.67323e-02 1.16112e-01 1.13633e-01 8.36268e-02 4.37873e-02 3.35748e-02 + 7.67323e-02 1.16112e-01 1.13633e-01 8.36269e-02 4.37874e-02 3.35748e-02 3.83631e-02 4.88584e-02 6.02807e-02 6.83896e-02 5.78449e-02 4.41108e-02 3.35748e-02 2.63647e-02 2.80467e-02 4.29578e-02 @@ -42,7 +42,7 @@ Ionic_Step 1 Cubefile created from ABACUS. Inner loop is z, followed by y and x 7.06533e-02 1.39007e-01 2.37646e-01 1.27736e+00 2.40739e+00 1.23255e+00 1.64801e-01 8.40243e-02 7.82972e-02 1.51620e-01 - 2.28880e-01 2.40739e+00 5.81649e-03 2.39320e+00 1.94498e-01 8.51576e-02 + 2.28880e-01 2.40739e+00 5.81645e-03 2.39320e+00 1.94498e-01 8.51576e-02 6.66000e-02 1.16541e-01 1.63372e-01 1.23255e+00 2.39320e+00 1.23378e+00 1.32731e-01 7.61344e-02 5.52126e-02 8.72644e-02 @@ -54,38 +54,38 @@ Ionic_Step 1 Cubefile created from ABACUS. Inner loop is z, followed by y and x 3.36094e-02 5.74526e-02 1.39007e-01 1.51620e-01 1.16541e-01 8.72644e-02 6.16640e-02 3.85916e-02 5.74526e-02 1.00802e-01 - 1.19126e-01 1.64234e-01 1.63372e-01 1.32380e-01 8.36268e-02 4.41108e-02 + 1.19126e-01 1.64234e-01 1.63372e-01 1.32380e-01 8.36269e-02 4.41108e-02 4.11865e-02 6.42572e-02 1.64234e-01 4.92662e-01 1.23255e+00 4.81993e-01 1.21578e-01 6.74730e-02 5.09510e-02 9.05022e-02 1.63372e-01 1.23255e+00 2.39320e+00 1.23378e+00 1.32731e-01 7.61344e-02 5.52126e-02 8.72644e-02 - 1.32380e-01 4.81993e-01 1.23378e+00 4.58434e-01 1.19299e-01 6.83304e-02 + 1.32380e-01 4.81993e-01 1.23378e+00 4.58433e-01 1.19299e-01 6.83304e-02 4.94441e-02 7.37458e-02 - 8.36268e-02 1.21578e-01 1.32731e-01 1.19299e-01 8.49603e-02 4.35048e-02 + 8.36269e-02 1.21578e-01 1.32731e-01 1.19299e-01 8.49603e-02 4.35048e-02 3.27200e-02 4.27881e-02 4.41108e-02 6.74730e-02 7.61344e-02 6.83304e-02 4.35048e-02 3.15330e-02 3.07987e-02 3.15675e-02 4.11865e-02 5.09510e-02 5.52126e-02 4.94441e-02 3.27200e-02 3.07987e-02 - 2.59564e-02 3.28004e-02 + 2.59563e-02 3.28004e-02 6.42572e-02 9.05022e-02 8.72644e-02 7.37458e-02 4.27881e-02 3.15675e-02 3.28004e-02 4.60122e-02 - 7.67323e-02 1.16112e-01 1.13633e-01 8.36268e-02 4.37873e-02 3.35748e-02 + 7.67323e-02 1.16112e-01 1.13633e-01 8.36269e-02 4.37874e-02 3.35748e-02 3.83631e-02 4.88584e-02 1.16112e-01 1.61484e-01 1.64801e-01 1.21578e-01 6.95388e-02 3.99685e-02 4.24335e-02 6.00999e-02 1.13633e-01 1.64801e-01 1.94498e-01 1.32731e-01 8.30022e-02 4.47298e-02 4.13448e-02 6.16640e-02 - 8.36268e-02 1.21578e-01 1.32731e-01 1.19299e-01 8.49603e-02 4.35048e-02 + 8.36269e-02 1.21578e-01 1.32731e-01 1.19299e-01 8.49603e-02 4.35048e-02 3.27200e-02 4.27881e-02 - 4.37873e-02 6.95388e-02 8.30022e-02 8.49603e-02 5.74135e-02 3.62021e-02 + 4.37874e-02 6.95388e-02 8.30022e-02 8.49603e-02 5.74134e-02 3.62021e-02 2.68643e-02 2.91967e-02 3.35748e-02 3.99685e-02 4.47298e-02 4.35048e-02 3.62021e-02 2.35001e-02 1.96139e-02 2.57121e-02 3.83631e-02 4.24335e-02 4.13448e-02 3.27200e-02 2.68643e-02 1.96139e-02 2.50516e-02 2.95263e-02 4.88584e-02 6.00999e-02 6.16640e-02 4.27881e-02 2.91967e-02 2.57121e-02 - 2.95263e-02 3.70504e-02 + 2.95263e-02 3.70505e-02 6.02807e-02 6.83896e-02 5.78449e-02 4.41108e-02 3.35748e-02 2.63647e-02 2.80467e-02 4.29578e-02 6.83896e-02 9.29643e-02 8.40243e-02 6.74730e-02 3.99685e-02 2.95603e-02 @@ -97,8 +97,8 @@ Ionic_Step 1 Cubefile created from ABACUS. Inner loop is z, followed by y and x 3.35748e-02 3.99685e-02 4.47298e-02 4.35048e-02 3.62021e-02 2.35001e-02 1.96139e-02 2.57121e-02 2.63647e-02 2.95603e-02 3.48676e-02 3.15330e-02 2.35001e-02 1.97954e-02 - 1.99588e-02 2.25182e-02 - 2.80467e-02 3.21705e-02 3.36248e-02 3.07987e-02 1.96139e-02 1.99588e-02 + 1.99589e-02 2.25182e-02 + 2.80467e-02 3.21705e-02 3.36248e-02 3.07987e-02 1.96139e-02 1.99589e-02 1.89056e-02 2.32253e-02 4.29578e-02 4.65705e-02 3.85916e-02 3.15675e-02 2.57121e-02 2.25182e-02 2.32253e-02 3.44643e-02 @@ -109,12 +109,12 @@ Ionic_Step 1 Cubefile created from ABACUS. Inner loop is z, followed by y and x 7.06533e-02 7.82972e-02 6.66000e-02 5.52126e-02 4.13448e-02 3.36248e-02 3.36094e-02 5.74526e-02 4.11865e-02 5.09510e-02 5.52126e-02 4.94441e-02 3.27200e-02 3.07987e-02 - 2.59564e-02 3.28004e-02 + 2.59563e-02 3.28004e-02 3.83631e-02 4.24335e-02 4.13448e-02 3.27200e-02 2.68643e-02 1.96139e-02 2.50516e-02 2.95263e-02 - 2.80467e-02 3.21705e-02 3.36248e-02 3.07987e-02 1.96139e-02 1.99588e-02 + 2.80467e-02 3.21705e-02 3.36248e-02 3.07987e-02 1.96139e-02 1.99589e-02 1.89056e-02 2.32253e-02 - 4.92095e-02 4.55573e-02 3.36094e-02 2.59564e-02 2.50516e-02 1.89056e-02 + 4.92095e-02 4.55573e-02 3.36094e-02 2.59563e-02 2.50516e-02 1.89056e-02 2.72609e-02 3.99728e-02 1.04952e-01 9.43573e-02 5.74526e-02 3.28004e-02 2.95263e-02 2.32253e-02 3.99728e-02 7.71535e-02 @@ -127,10 +127,10 @@ Ionic_Step 1 Cubefile created from ABACUS. Inner loop is z, followed by y and x 6.42572e-02 9.05022e-02 8.72644e-02 7.37458e-02 4.27881e-02 3.15675e-02 3.28004e-02 4.60122e-02 4.88584e-02 6.00999e-02 6.16640e-02 4.27881e-02 2.91967e-02 2.57121e-02 - 2.95263e-02 3.70504e-02 + 2.95263e-02 3.70505e-02 4.29578e-02 4.65705e-02 3.85916e-02 3.15675e-02 2.57121e-02 2.25182e-02 2.32253e-02 3.44643e-02 1.04952e-01 9.43573e-02 5.74526e-02 3.28004e-02 2.95263e-02 2.32253e-02 3.99728e-02 7.71535e-02 - 1.04625e+00 4.94732e-01 1.00802e-01 4.60122e-02 3.70504e-02 3.44643e-02 + 1.04625e+00 4.94732e-01 1.00802e-01 4.60122e-02 3.70505e-02 3.44643e-02 7.71535e-02 4.57945e-01 diff --git a/tests/03_NAO_multik/scf_out_chg_tau/chg.cube.ref b/tests/03_NAO_multik/scf_out_chg_tau/chg.cube.ref index 1ae0be21f2a..83a0ac324a7 100644 --- a/tests/03_NAO_multik/scf_out_chg_tau/chg.cube.ref +++ b/tests/03_NAO_multik/scf_out_chg_tau/chg.cube.ref @@ -6,594 +6,594 @@ Ionic_Step 1 Cubefile created from ABACUS. Inner loop is z, followed by y and x 15 0.340000 0.340000 0.000000 14 4.000000 0.000000 0.000000 0.000000 14 4.000000 2.550000 2.550000 2.550000 - 2.511e-03 1.824e-02 5.185e-02 6.777e-02 6.109e-02 4.496e-02 - 3.013e-02 2.140e-02 2.140e-02 3.013e-02 4.496e-02 6.109e-02 + 2.512e-03 1.824e-02 5.185e-02 6.777e-02 6.109e-02 4.496e-02 + 3.013e-02 2.139e-02 2.139e-02 3.013e-02 4.496e-02 6.109e-02 6.777e-02 5.185e-02 1.824e-02 - 1.824e-02 6.194e-02 9.277e-02 9.268e-02 7.503e-02 5.407e-02 - 3.750e-02 3.084e-02 3.750e-02 5.407e-02 7.503e-02 9.268e-02 + 1.824e-02 6.194e-02 9.277e-02 9.268e-02 7.503e-02 5.406e-02 + 3.749e-02 3.083e-02 3.749e-02 5.406e-02 7.503e-02 9.268e-02 9.277e-02 6.194e-02 1.824e-02 - 5.185e-02 9.277e-02 1.077e-01 9.861e-02 7.874e-02 5.687e-02 - 4.149e-02 4.149e-02 5.687e-02 7.874e-02 9.861e-02 1.077e-01 + 5.185e-02 9.277e-02 1.077e-01 9.860e-02 7.874e-02 5.686e-02 + 4.148e-02 4.148e-02 5.686e-02 7.874e-02 9.860e-02 1.077e-01 9.277e-02 5.185e-02 2.392e-02 - 6.777e-02 9.268e-02 9.861e-02 9.038e-02 7.202e-02 5.019e-02 - 3.965e-02 5.019e-02 7.202e-02 9.038e-02 9.861e-02 9.268e-02 + 6.777e-02 9.268e-02 9.860e-02 9.038e-02 7.202e-02 5.018e-02 + 3.964e-02 5.018e-02 7.202e-02 9.038e-02 9.860e-02 9.268e-02 6.777e-02 4.007e-02 4.007e-02 - 6.109e-02 7.503e-02 7.874e-02 7.202e-02 5.340e-02 3.517e-02 - 3.517e-02 5.340e-02 7.202e-02 7.874e-02 7.503e-02 6.109e-02 - 4.205e-02 3.244e-02 4.205e-02 - 4.496e-02 5.407e-02 5.687e-02 5.019e-02 3.517e-02 2.639e-02 - 3.517e-02 5.019e-02 5.687e-02 5.407e-02 4.496e-02 3.255e-02 + 6.109e-02 7.503e-02 7.874e-02 7.202e-02 5.339e-02 3.516e-02 + 3.516e-02 5.339e-02 7.202e-02 7.874e-02 7.503e-02 6.109e-02 + 4.206e-02 3.245e-02 4.206e-02 + 4.496e-02 5.406e-02 5.686e-02 5.018e-02 3.516e-02 2.637e-02 + 3.516e-02 5.018e-02 5.686e-02 5.406e-02 4.496e-02 3.255e-02 2.347e-02 2.347e-02 3.255e-02 - 3.013e-02 3.750e-02 4.149e-02 3.965e-02 3.517e-02 3.517e-02 - 3.965e-02 4.149e-02 3.750e-02 3.013e-02 2.182e-02 1.523e-02 - 1.264e-02 1.523e-02 2.182e-02 - 2.140e-02 3.084e-02 4.149e-02 5.019e-02 5.340e-02 5.019e-02 - 4.149e-02 3.084e-02 2.140e-02 1.445e-02 9.636e-03 7.208e-03 - 7.208e-03 9.636e-03 1.445e-02 - 2.140e-02 3.750e-02 5.687e-02 7.202e-02 7.202e-02 5.687e-02 - 3.750e-02 2.140e-02 1.179e-02 6.907e-03 4.700e-03 4.099e-03 - 4.700e-03 6.907e-03 1.179e-02 - 3.013e-02 5.407e-02 7.874e-02 9.038e-02 7.874e-02 5.407e-02 - 3.013e-02 1.445e-02 6.907e-03 4.025e-03 3.161e-03 3.161e-03 - 4.025e-03 6.907e-03 1.445e-02 - 4.496e-02 7.503e-02 9.861e-02 9.861e-02 7.503e-02 4.496e-02 - 2.182e-02 9.636e-03 4.700e-03 3.161e-03 2.848e-03 3.161e-03 - 4.700e-03 9.636e-03 2.182e-02 + 3.013e-02 3.749e-02 4.148e-02 3.964e-02 3.516e-02 3.516e-02 + 3.964e-02 4.148e-02 3.749e-02 3.013e-02 2.183e-02 1.523e-02 + 1.265e-02 1.523e-02 2.183e-02 + 2.139e-02 3.083e-02 4.148e-02 5.018e-02 5.339e-02 5.018e-02 + 4.148e-02 3.083e-02 2.139e-02 1.445e-02 9.638e-03 7.210e-03 + 7.210e-03 9.638e-03 1.445e-02 + 2.139e-02 3.749e-02 5.686e-02 7.202e-02 7.202e-02 5.686e-02 + 3.749e-02 2.139e-02 1.179e-02 6.909e-03 4.702e-03 4.101e-03 + 4.702e-03 6.909e-03 1.179e-02 + 3.013e-02 5.406e-02 7.874e-02 9.038e-02 7.874e-02 5.406e-02 + 3.013e-02 1.445e-02 6.909e-03 4.027e-03 3.163e-03 3.163e-03 + 4.027e-03 6.909e-03 1.445e-02 + 4.496e-02 7.503e-02 9.860e-02 9.860e-02 7.503e-02 4.496e-02 + 2.183e-02 9.638e-03 4.702e-03 3.163e-03 2.850e-03 3.163e-03 + 4.702e-03 9.638e-03 2.183e-02 6.109e-02 9.268e-02 1.077e-01 9.268e-02 6.109e-02 3.255e-02 - 1.523e-02 7.208e-03 4.099e-03 3.161e-03 3.161e-03 4.099e-03 - 7.208e-03 1.523e-02 3.255e-02 - 6.777e-02 9.277e-02 9.277e-02 6.777e-02 4.205e-02 2.347e-02 - 1.264e-02 7.208e-03 4.700e-03 4.025e-03 4.700e-03 7.208e-03 - 1.264e-02 2.347e-02 4.205e-02 - 5.185e-02 6.194e-02 5.185e-02 4.007e-02 3.244e-02 2.347e-02 - 1.523e-02 9.636e-03 6.907e-03 6.907e-03 9.636e-03 1.523e-02 - 2.347e-02 3.244e-02 4.007e-02 - 1.824e-02 1.824e-02 2.392e-02 4.007e-02 4.205e-02 3.255e-02 - 2.182e-02 1.445e-02 1.179e-02 1.445e-02 2.182e-02 3.255e-02 - 4.205e-02 4.007e-02 2.392e-02 - 1.824e-02 6.194e-02 9.277e-02 9.268e-02 7.503e-02 5.407e-02 - 3.750e-02 3.084e-02 3.750e-02 5.407e-02 7.503e-02 9.268e-02 + 1.523e-02 7.210e-03 4.101e-03 3.163e-03 3.163e-03 4.101e-03 + 7.210e-03 1.523e-02 3.255e-02 + 6.777e-02 9.277e-02 9.277e-02 6.777e-02 4.206e-02 2.347e-02 + 1.265e-02 7.210e-03 4.702e-03 4.027e-03 4.702e-03 7.210e-03 + 1.265e-02 2.347e-02 4.206e-02 + 5.185e-02 6.194e-02 5.185e-02 4.007e-02 3.245e-02 2.347e-02 + 1.523e-02 9.638e-03 6.909e-03 6.909e-03 9.638e-03 1.523e-02 + 2.347e-02 3.245e-02 4.007e-02 + 1.824e-02 1.824e-02 2.392e-02 4.007e-02 4.206e-02 3.255e-02 + 2.183e-02 1.445e-02 1.179e-02 1.445e-02 2.183e-02 3.255e-02 + 4.206e-02 4.007e-02 2.392e-02 + 1.824e-02 6.194e-02 9.277e-02 9.268e-02 7.503e-02 5.406e-02 + 3.749e-02 3.083e-02 3.749e-02 5.406e-02 7.503e-02 9.268e-02 9.277e-02 6.194e-02 1.824e-02 - 6.194e-02 1.075e-01 1.242e-01 1.128e-01 9.101e-02 6.610e-02 - 4.772e-02 4.772e-02 6.610e-02 9.101e-02 1.128e-01 1.242e-01 + 6.194e-02 1.075e-01 1.242e-01 1.128e-01 9.101e-02 6.609e-02 + 4.771e-02 4.771e-02 6.609e-02 9.101e-02 1.128e-01 1.242e-01 1.075e-01 6.194e-02 2.857e-02 - 9.277e-02 1.242e-01 1.311e-01 1.188e-01 9.562e-02 6.682e-02 - 5.234e-02 6.682e-02 9.562e-02 1.188e-01 1.311e-01 1.242e-01 + 9.277e-02 1.242e-01 1.311e-01 1.188e-01 9.562e-02 6.681e-02 + 5.233e-02 6.681e-02 9.562e-02 1.188e-01 1.311e-01 1.242e-01 9.277e-02 5.272e-02 5.272e-02 - 9.268e-02 1.128e-01 1.188e-01 1.093e-01 8.147e-02 5.147e-02 - 5.147e-02 8.147e-02 1.093e-01 1.188e-01 1.128e-01 9.268e-02 - 6.359e-02 4.810e-02 6.359e-02 - 7.503e-02 9.101e-02 9.562e-02 8.147e-02 4.916e-02 2.974e-02 - 4.916e-02 8.147e-02 9.562e-02 9.101e-02 7.503e-02 5.455e-02 + 9.268e-02 1.128e-01 1.188e-01 1.093e-01 8.146e-02 5.146e-02 + 5.146e-02 8.146e-02 1.093e-01 1.188e-01 1.128e-01 9.268e-02 + 6.360e-02 4.810e-02 6.360e-02 + 7.503e-02 9.101e-02 9.562e-02 8.146e-02 4.915e-02 2.973e-02 + 4.915e-02 8.146e-02 9.562e-02 9.101e-02 7.503e-02 5.455e-02 3.923e-02 3.923e-02 5.455e-02 - 5.407e-02 6.610e-02 6.682e-02 5.147e-02 2.974e-02 2.974e-02 - 5.147e-02 6.682e-02 6.610e-02 5.407e-02 3.967e-02 2.757e-02 + 5.406e-02 6.609e-02 6.681e-02 5.146e-02 2.973e-02 2.973e-02 + 5.146e-02 6.681e-02 6.609e-02 5.406e-02 3.967e-02 2.757e-02 2.272e-02 2.757e-02 3.967e-02 - 3.750e-02 4.772e-02 5.234e-02 5.147e-02 4.916e-02 5.147e-02 - 5.234e-02 4.772e-02 3.750e-02 2.653e-02 1.795e-02 1.312e-02 - 1.312e-02 1.795e-02 2.653e-02 - 3.084e-02 4.772e-02 6.682e-02 8.147e-02 8.147e-02 6.682e-02 - 4.772e-02 3.084e-02 1.895e-02 1.184e-02 7.950e-03 6.755e-03 - 7.950e-03 1.184e-02 1.895e-02 - 3.750e-02 6.610e-02 9.562e-02 1.093e-01 9.562e-02 6.610e-02 - 3.750e-02 1.895e-02 9.768e-03 5.832e-03 4.402e-03 4.402e-03 - 5.832e-03 9.768e-03 1.895e-02 - 5.407e-02 9.101e-02 1.188e-01 1.188e-01 9.101e-02 5.407e-02 - 2.653e-02 1.184e-02 5.832e-03 3.833e-03 3.410e-03 3.833e-03 - 5.832e-03 1.184e-02 2.653e-02 + 3.749e-02 4.771e-02 5.233e-02 5.146e-02 4.915e-02 5.146e-02 + 5.233e-02 4.771e-02 3.749e-02 2.653e-02 1.795e-02 1.313e-02 + 1.313e-02 1.795e-02 2.653e-02 + 3.083e-02 4.771e-02 6.681e-02 8.146e-02 8.146e-02 6.681e-02 + 4.771e-02 3.083e-02 1.895e-02 1.184e-02 7.953e-03 6.758e-03 + 7.953e-03 1.184e-02 1.895e-02 + 3.749e-02 6.609e-02 9.562e-02 1.093e-01 9.562e-02 6.609e-02 + 3.749e-02 1.895e-02 9.768e-03 5.833e-03 4.404e-03 4.404e-03 + 5.833e-03 9.768e-03 1.895e-02 + 5.406e-02 9.101e-02 1.188e-01 1.188e-01 9.101e-02 5.406e-02 + 2.653e-02 1.184e-02 5.833e-03 3.835e-03 3.412e-03 3.835e-03 + 5.833e-03 1.184e-02 2.653e-02 7.503e-02 1.128e-01 1.311e-01 1.128e-01 7.503e-02 3.967e-02 - 1.795e-02 7.950e-03 4.402e-03 3.410e-03 3.410e-03 4.402e-03 - 7.950e-03 1.795e-02 3.967e-02 + 1.795e-02 7.953e-03 4.404e-03 3.412e-03 3.412e-03 4.404e-03 + 7.953e-03 1.795e-02 3.967e-02 9.268e-02 1.242e-01 1.242e-01 9.268e-02 5.455e-02 2.757e-02 - 1.312e-02 6.755e-03 4.402e-03 3.833e-03 4.402e-03 6.755e-03 - 1.312e-02 2.757e-02 5.455e-02 - 9.277e-02 1.075e-01 9.277e-02 6.359e-02 3.923e-02 2.272e-02 - 1.312e-02 7.950e-03 5.832e-03 5.832e-03 7.950e-03 1.312e-02 - 2.272e-02 3.923e-02 6.359e-02 + 1.313e-02 6.758e-03 4.404e-03 3.835e-03 4.404e-03 6.758e-03 + 1.313e-02 2.757e-02 5.455e-02 + 9.277e-02 1.075e-01 9.277e-02 6.360e-02 3.923e-02 2.272e-02 + 1.313e-02 7.953e-03 5.833e-03 5.833e-03 7.953e-03 1.313e-02 + 2.272e-02 3.923e-02 6.360e-02 6.194e-02 6.194e-02 5.272e-02 4.810e-02 3.923e-02 2.757e-02 1.795e-02 1.184e-02 9.768e-03 1.184e-02 1.795e-02 2.757e-02 3.923e-02 4.810e-02 5.272e-02 - 1.824e-02 2.857e-02 5.272e-02 6.359e-02 5.455e-02 3.967e-02 + 1.824e-02 2.857e-02 5.272e-02 6.360e-02 5.455e-02 3.967e-02 2.653e-02 1.895e-02 1.895e-02 2.653e-02 3.967e-02 5.455e-02 - 6.359e-02 5.272e-02 2.857e-02 - 5.185e-02 9.277e-02 1.077e-01 9.861e-02 7.874e-02 5.687e-02 - 4.149e-02 4.149e-02 5.687e-02 7.874e-02 9.861e-02 1.077e-01 + 6.360e-02 5.272e-02 2.857e-02 + 5.185e-02 9.277e-02 1.077e-01 9.860e-02 7.874e-02 5.686e-02 + 4.148e-02 4.148e-02 5.686e-02 7.874e-02 9.860e-02 1.077e-01 9.277e-02 5.185e-02 2.392e-02 - 9.277e-02 1.242e-01 1.311e-01 1.188e-01 9.562e-02 6.682e-02 - 5.234e-02 6.682e-02 9.562e-02 1.188e-01 1.311e-01 1.242e-01 + 9.277e-02 1.242e-01 1.311e-01 1.188e-01 9.562e-02 6.681e-02 + 5.233e-02 6.681e-02 9.562e-02 1.188e-01 1.311e-01 1.242e-01 9.277e-02 5.272e-02 5.272e-02 - 1.077e-01 1.311e-01 1.384e-01 1.269e-01 9.487e-02 5.924e-02 - 5.924e-02 9.487e-02 1.269e-01 1.384e-01 1.311e-01 1.077e-01 - 7.321e-02 5.525e-02 7.321e-02 - 9.861e-02 1.188e-01 1.269e-01 1.094e-01 6.638e-02 3.678e-02 - 6.638e-02 1.094e-01 1.269e-01 1.188e-01 9.861e-02 7.273e-02 + 1.077e-01 1.311e-01 1.384e-01 1.269e-01 9.487e-02 5.923e-02 + 5.923e-02 9.487e-02 1.269e-01 1.384e-01 1.311e-01 1.077e-01 + 7.322e-02 5.525e-02 7.322e-02 + 9.860e-02 1.188e-01 1.269e-01 1.094e-01 6.638e-02 3.677e-02 + 6.638e-02 1.094e-01 1.269e-01 1.188e-01 9.860e-02 7.273e-02 5.204e-02 5.204e-02 7.273e-02 - 7.874e-02 9.562e-02 9.487e-02 6.638e-02 2.424e-02 2.424e-02 + 7.874e-02 9.562e-02 9.487e-02 6.638e-02 2.423e-02 2.423e-02 6.638e-02 9.487e-02 9.562e-02 7.874e-02 5.829e-02 4.066e-02 3.364e-02 4.066e-02 5.829e-02 - 5.687e-02 6.682e-02 5.924e-02 3.678e-02 2.424e-02 3.678e-02 - 5.924e-02 6.682e-02 5.687e-02 4.213e-02 2.859e-02 2.065e-02 - 2.065e-02 2.859e-02 4.213e-02 - 4.149e-02 5.234e-02 5.924e-02 6.638e-02 6.638e-02 5.924e-02 - 5.234e-02 4.149e-02 2.898e-02 1.892e-02 1.265e-02 1.050e-02 - 1.265e-02 1.892e-02 2.898e-02 - 4.149e-02 6.682e-02 9.487e-02 1.094e-01 9.487e-02 6.682e-02 - 4.149e-02 2.401e-02 1.370e-02 8.411e-03 6.183e-03 6.183e-03 - 8.411e-03 1.370e-02 2.401e-02 - 5.687e-02 9.562e-02 1.269e-01 1.269e-01 9.562e-02 5.687e-02 - 2.898e-02 1.370e-02 7.112e-03 4.615e-03 4.017e-03 4.615e-03 - 7.112e-03 1.370e-02 2.898e-02 + 5.686e-02 6.681e-02 5.923e-02 3.677e-02 2.423e-02 3.677e-02 + 5.923e-02 6.681e-02 5.686e-02 4.213e-02 2.859e-02 2.066e-02 + 2.066e-02 2.859e-02 4.213e-02 + 4.148e-02 5.233e-02 5.923e-02 6.638e-02 6.638e-02 5.923e-02 + 5.233e-02 4.148e-02 2.897e-02 1.892e-02 1.266e-02 1.050e-02 + 1.266e-02 1.892e-02 2.897e-02 + 4.148e-02 6.681e-02 9.487e-02 1.094e-01 9.487e-02 6.681e-02 + 4.148e-02 2.401e-02 1.370e-02 8.413e-03 6.186e-03 6.186e-03 + 8.413e-03 1.370e-02 2.401e-02 + 5.686e-02 9.562e-02 1.269e-01 1.269e-01 9.562e-02 5.686e-02 + 2.897e-02 1.370e-02 7.113e-03 4.616e-03 4.019e-03 4.616e-03 + 7.113e-03 1.370e-02 2.897e-02 7.874e-02 1.188e-01 1.384e-01 1.188e-01 7.874e-02 4.213e-02 - 1.892e-02 8.411e-03 4.615e-03 3.559e-03 3.559e-03 4.615e-03 - 8.411e-03 1.892e-02 4.213e-02 - 9.861e-02 1.311e-01 1.311e-01 9.861e-02 5.829e-02 2.859e-02 - 1.265e-02 6.183e-03 4.017e-03 3.559e-03 4.017e-03 6.183e-03 - 1.265e-02 2.859e-02 5.829e-02 - 1.077e-01 1.242e-01 1.077e-01 7.273e-02 4.066e-02 2.065e-02 - 1.050e-02 6.183e-03 4.615e-03 4.615e-03 6.183e-03 1.050e-02 - 2.065e-02 4.066e-02 7.273e-02 - 9.277e-02 9.277e-02 7.321e-02 5.204e-02 3.364e-02 2.065e-02 - 1.265e-02 8.411e-03 7.112e-03 8.411e-03 1.265e-02 2.065e-02 - 3.364e-02 5.204e-02 7.321e-02 + 1.892e-02 8.413e-03 4.616e-03 3.561e-03 3.561e-03 4.616e-03 + 8.413e-03 1.892e-02 4.213e-02 + 9.860e-02 1.311e-01 1.311e-01 9.860e-02 5.829e-02 2.859e-02 + 1.266e-02 6.186e-03 4.019e-03 3.561e-03 4.019e-03 6.186e-03 + 1.266e-02 2.859e-02 5.829e-02 + 1.077e-01 1.242e-01 1.077e-01 7.273e-02 4.066e-02 2.066e-02 + 1.050e-02 6.186e-03 4.616e-03 4.616e-03 6.186e-03 1.050e-02 + 2.066e-02 4.066e-02 7.273e-02 + 9.277e-02 9.277e-02 7.322e-02 5.204e-02 3.364e-02 2.066e-02 + 1.266e-02 8.413e-03 7.113e-03 8.413e-03 1.266e-02 2.066e-02 + 3.364e-02 5.204e-02 7.322e-02 5.185e-02 5.272e-02 5.525e-02 5.204e-02 4.066e-02 2.859e-02 1.892e-02 1.370e-02 1.370e-02 1.892e-02 2.859e-02 4.066e-02 5.204e-02 5.525e-02 5.272e-02 - 2.392e-02 5.272e-02 7.321e-02 7.273e-02 5.829e-02 4.213e-02 - 2.898e-02 2.401e-02 2.898e-02 4.213e-02 5.829e-02 7.273e-02 - 7.321e-02 5.272e-02 2.392e-02 - 6.777e-02 9.268e-02 9.861e-02 9.038e-02 7.202e-02 5.019e-02 - 3.965e-02 5.019e-02 7.202e-02 9.038e-02 9.861e-02 9.268e-02 + 2.392e-02 5.272e-02 7.322e-02 7.273e-02 5.829e-02 4.213e-02 + 2.897e-02 2.401e-02 2.897e-02 4.213e-02 5.829e-02 7.273e-02 + 7.322e-02 5.272e-02 2.392e-02 + 6.777e-02 9.268e-02 9.860e-02 9.038e-02 7.202e-02 5.018e-02 + 3.964e-02 5.018e-02 7.202e-02 9.038e-02 9.860e-02 9.268e-02 6.777e-02 4.007e-02 4.007e-02 - 9.268e-02 1.128e-01 1.188e-01 1.093e-01 8.147e-02 5.147e-02 - 5.147e-02 8.147e-02 1.093e-01 1.188e-01 1.128e-01 9.268e-02 - 6.359e-02 4.810e-02 6.359e-02 - 9.861e-02 1.188e-01 1.269e-01 1.094e-01 6.638e-02 3.678e-02 - 6.638e-02 1.094e-01 1.269e-01 1.188e-01 9.861e-02 7.273e-02 + 9.268e-02 1.128e-01 1.188e-01 1.093e-01 8.146e-02 5.146e-02 + 5.146e-02 8.146e-02 1.093e-01 1.188e-01 1.128e-01 9.268e-02 + 6.360e-02 4.810e-02 6.360e-02 + 9.860e-02 1.188e-01 1.269e-01 1.094e-01 6.638e-02 3.677e-02 + 6.638e-02 1.094e-01 1.269e-01 1.188e-01 9.860e-02 7.273e-02 5.204e-02 5.204e-02 7.273e-02 - 9.038e-02 1.093e-01 1.094e-01 7.723e-02 2.834e-02 2.834e-02 - 7.723e-02 1.094e-01 1.093e-01 9.038e-02 6.717e-02 4.690e-02 + 9.038e-02 1.093e-01 1.094e-01 7.724e-02 2.834e-02 2.834e-02 + 7.724e-02 1.094e-01 1.093e-01 9.038e-02 6.717e-02 4.690e-02 3.841e-02 4.690e-02 6.717e-02 - 7.202e-02 8.147e-02 6.638e-02 2.834e-02 6.079e-03 2.834e-02 - 6.638e-02 8.147e-02 7.202e-02 5.348e-02 3.637e-02 2.620e-02 + 7.202e-02 8.146e-02 6.638e-02 2.834e-02 6.078e-03 2.834e-02 + 6.638e-02 8.146e-02 7.202e-02 5.348e-02 3.637e-02 2.620e-02 2.620e-02 3.637e-02 5.348e-02 - 5.019e-02 5.147e-02 3.678e-02 2.834e-02 2.834e-02 3.678e-02 - 5.147e-02 5.019e-02 3.817e-02 2.583e-02 1.720e-02 1.407e-02 + 5.018e-02 5.146e-02 3.677e-02 2.834e-02 2.834e-02 3.677e-02 + 5.146e-02 5.018e-02 3.817e-02 2.583e-02 1.720e-02 1.407e-02 1.720e-02 2.583e-02 3.817e-02 - 3.965e-02 5.147e-02 6.638e-02 7.723e-02 6.638e-02 5.147e-02 - 3.965e-02 2.780e-02 1.791e-02 1.127e-02 8.102e-03 8.102e-03 - 1.127e-02 1.791e-02 2.780e-02 - 5.019e-02 8.147e-02 1.094e-01 1.094e-01 8.147e-02 5.019e-02 - 2.780e-02 1.479e-02 8.316e-03 5.415e-03 4.623e-03 5.415e-03 - 8.316e-03 1.479e-02 2.780e-02 + 3.964e-02 5.146e-02 6.638e-02 7.724e-02 6.638e-02 5.146e-02 + 3.964e-02 2.780e-02 1.790e-02 1.127e-02 8.105e-03 8.105e-03 + 1.127e-02 1.790e-02 2.780e-02 + 5.018e-02 8.146e-02 1.094e-01 1.094e-01 8.146e-02 5.018e-02 + 2.780e-02 1.479e-02 8.317e-03 5.417e-03 4.626e-03 5.417e-03 + 8.317e-03 1.479e-02 2.780e-02 7.202e-02 1.093e-01 1.269e-01 1.093e-01 7.202e-02 3.817e-02 - 1.791e-02 8.316e-03 4.687e-03 3.581e-03 3.581e-03 4.687e-03 - 8.316e-03 1.791e-02 3.817e-02 + 1.790e-02 8.317e-03 4.689e-03 3.583e-03 3.583e-03 4.689e-03 + 8.317e-03 1.790e-02 3.817e-02 9.038e-02 1.188e-01 1.188e-01 9.038e-02 5.348e-02 2.583e-02 - 1.127e-02 5.415e-03 3.581e-03 3.198e-03 3.581e-03 5.415e-03 + 1.127e-02 5.417e-03 3.583e-03 3.200e-03 3.583e-03 5.417e-03 1.127e-02 2.583e-02 5.348e-02 - 9.861e-02 1.128e-01 9.861e-02 6.717e-02 3.637e-02 1.720e-02 - 8.102e-03 4.623e-03 3.581e-03 3.581e-03 4.623e-03 8.102e-03 + 9.860e-02 1.128e-01 9.860e-02 6.717e-02 3.637e-02 1.720e-02 + 8.105e-03 4.626e-03 3.583e-03 3.583e-03 4.626e-03 8.105e-03 1.720e-02 3.637e-02 6.717e-02 9.268e-02 9.268e-02 7.273e-02 4.690e-02 2.620e-02 1.407e-02 - 8.102e-03 5.415e-03 4.687e-03 5.415e-03 8.102e-03 1.407e-02 + 8.105e-03 5.417e-03 4.689e-03 5.417e-03 8.105e-03 1.407e-02 2.620e-02 4.690e-02 7.273e-02 - 6.777e-02 6.359e-02 5.204e-02 3.841e-02 2.620e-02 1.720e-02 - 1.127e-02 8.316e-03 8.316e-03 1.127e-02 1.720e-02 2.620e-02 - 3.841e-02 5.204e-02 6.359e-02 + 6.777e-02 6.360e-02 5.204e-02 3.841e-02 2.620e-02 1.720e-02 + 1.127e-02 8.317e-03 8.317e-03 1.127e-02 1.720e-02 2.620e-02 + 3.841e-02 5.204e-02 6.360e-02 4.007e-02 4.810e-02 5.204e-02 4.690e-02 3.637e-02 2.583e-02 - 1.791e-02 1.479e-02 1.791e-02 2.583e-02 3.637e-02 4.690e-02 + 1.790e-02 1.479e-02 1.790e-02 2.583e-02 3.637e-02 4.690e-02 5.204e-02 4.810e-02 4.007e-02 - 4.007e-02 6.359e-02 7.273e-02 6.717e-02 5.348e-02 3.817e-02 + 4.007e-02 6.360e-02 7.273e-02 6.717e-02 5.348e-02 3.817e-02 2.780e-02 2.780e-02 3.817e-02 5.348e-02 6.717e-02 7.273e-02 - 6.359e-02 4.007e-02 2.575e-02 - 6.109e-02 7.503e-02 7.874e-02 7.202e-02 5.340e-02 3.517e-02 - 3.517e-02 5.340e-02 7.202e-02 7.874e-02 7.503e-02 6.109e-02 - 4.205e-02 3.244e-02 4.205e-02 - 7.503e-02 9.101e-02 9.562e-02 8.147e-02 4.916e-02 2.974e-02 - 4.916e-02 8.147e-02 9.562e-02 9.101e-02 7.503e-02 5.455e-02 + 6.360e-02 4.007e-02 2.576e-02 + 6.109e-02 7.503e-02 7.874e-02 7.202e-02 5.339e-02 3.516e-02 + 3.516e-02 5.339e-02 7.202e-02 7.874e-02 7.503e-02 6.109e-02 + 4.206e-02 3.245e-02 4.206e-02 + 7.503e-02 9.101e-02 9.562e-02 8.146e-02 4.915e-02 2.973e-02 + 4.915e-02 8.146e-02 9.562e-02 9.101e-02 7.503e-02 5.455e-02 3.923e-02 3.923e-02 5.455e-02 - 7.874e-02 9.562e-02 9.487e-02 6.638e-02 2.424e-02 2.424e-02 + 7.874e-02 9.562e-02 9.487e-02 6.638e-02 2.423e-02 2.423e-02 6.638e-02 9.487e-02 9.562e-02 7.874e-02 5.829e-02 4.066e-02 3.364e-02 4.066e-02 5.829e-02 - 7.202e-02 8.147e-02 6.638e-02 2.834e-02 6.079e-03 2.834e-02 - 6.638e-02 8.147e-02 7.202e-02 5.348e-02 3.637e-02 2.620e-02 + 7.202e-02 8.146e-02 6.638e-02 2.834e-02 6.078e-03 2.834e-02 + 6.638e-02 8.146e-02 7.202e-02 5.348e-02 3.637e-02 2.620e-02 2.620e-02 3.637e-02 5.348e-02 - 5.340e-02 4.916e-02 2.424e-02 6.079e-03 6.079e-03 2.424e-02 - 4.916e-02 5.340e-02 4.254e-02 2.919e-02 1.938e-02 1.586e-02 - 1.938e-02 2.919e-02 4.254e-02 - 3.517e-02 2.974e-02 2.424e-02 2.834e-02 2.424e-02 2.974e-02 - 3.517e-02 2.987e-02 2.075e-02 1.340e-02 9.462e-03 9.462e-03 + 5.339e-02 4.915e-02 2.423e-02 6.078e-03 6.078e-03 2.423e-02 + 4.915e-02 5.339e-02 4.253e-02 2.919e-02 1.938e-02 1.586e-02 + 1.938e-02 2.919e-02 4.253e-02 + 3.516e-02 2.973e-02 2.423e-02 2.834e-02 2.423e-02 2.973e-02 + 3.516e-02 2.987e-02 2.075e-02 1.340e-02 9.466e-03 9.466e-03 1.340e-02 2.075e-02 2.987e-02 - 3.517e-02 4.916e-02 6.638e-02 6.638e-02 4.916e-02 3.517e-02 - 2.402e-02 1.520e-02 9.366e-03 6.160e-03 5.124e-03 6.160e-03 - 9.366e-03 1.520e-02 2.402e-02 - 5.340e-02 8.147e-02 9.487e-02 8.147e-02 5.340e-02 2.987e-02 - 1.520e-02 7.945e-03 4.685e-03 3.516e-03 3.516e-03 4.685e-03 - 7.945e-03 1.520e-02 2.987e-02 - 7.202e-02 9.562e-02 9.562e-02 7.202e-02 4.254e-02 2.075e-02 - 9.366e-03 4.685e-03 3.128e-03 2.806e-03 3.128e-03 4.685e-03 - 9.366e-03 2.075e-02 4.254e-02 + 3.516e-02 4.915e-02 6.638e-02 6.638e-02 4.915e-02 3.516e-02 + 2.401e-02 1.520e-02 9.366e-03 6.163e-03 5.128e-03 6.163e-03 + 9.366e-03 1.520e-02 2.401e-02 + 5.339e-02 8.146e-02 9.487e-02 8.146e-02 5.339e-02 2.987e-02 + 1.520e-02 7.944e-03 4.687e-03 3.518e-03 3.518e-03 4.687e-03 + 7.944e-03 1.520e-02 2.987e-02 + 7.202e-02 9.562e-02 9.562e-02 7.202e-02 4.253e-02 2.075e-02 + 9.366e-03 4.687e-03 3.130e-03 2.808e-03 3.130e-03 4.687e-03 + 9.366e-03 2.075e-02 4.253e-02 7.874e-02 9.101e-02 7.874e-02 5.348e-02 2.919e-02 1.340e-02 - 6.160e-03 3.516e-03 2.806e-03 2.806e-03 3.516e-03 6.160e-03 + 6.163e-03 3.518e-03 2.808e-03 2.808e-03 3.518e-03 6.163e-03 1.340e-02 2.919e-02 5.348e-02 - 7.503e-02 7.503e-02 5.829e-02 3.637e-02 1.938e-02 9.462e-03 - 5.124e-03 3.516e-03 3.128e-03 3.516e-03 5.124e-03 9.462e-03 + 7.503e-02 7.503e-02 5.829e-02 3.637e-02 1.938e-02 9.466e-03 + 5.128e-03 3.518e-03 3.130e-03 3.518e-03 5.128e-03 9.466e-03 1.938e-02 3.637e-02 5.829e-02 - 6.109e-02 5.455e-02 4.066e-02 2.620e-02 1.586e-02 9.462e-03 - 6.160e-03 4.685e-03 4.685e-03 6.160e-03 9.462e-03 1.586e-02 + 6.109e-02 5.455e-02 4.066e-02 2.620e-02 1.586e-02 9.466e-03 + 6.163e-03 4.687e-03 4.687e-03 6.163e-03 9.466e-03 1.586e-02 2.620e-02 4.066e-02 5.455e-02 - 4.205e-02 3.923e-02 3.364e-02 2.620e-02 1.938e-02 1.340e-02 - 9.366e-03 7.945e-03 9.366e-03 1.340e-02 1.938e-02 2.620e-02 - 3.364e-02 3.923e-02 4.205e-02 - 3.244e-02 3.923e-02 4.066e-02 3.637e-02 2.919e-02 2.075e-02 + 4.206e-02 3.923e-02 3.364e-02 2.620e-02 1.938e-02 1.340e-02 + 9.366e-03 7.944e-03 9.366e-03 1.340e-02 1.938e-02 2.620e-02 + 3.364e-02 3.923e-02 4.206e-02 + 3.245e-02 3.923e-02 4.066e-02 3.637e-02 2.919e-02 2.075e-02 1.520e-02 1.520e-02 2.075e-02 2.919e-02 3.637e-02 4.066e-02 - 3.923e-02 3.244e-02 2.820e-02 - 4.205e-02 5.455e-02 5.829e-02 5.348e-02 4.254e-02 2.987e-02 - 2.402e-02 2.987e-02 4.254e-02 5.348e-02 5.829e-02 5.455e-02 - 4.205e-02 2.820e-02 2.820e-02 - 4.496e-02 5.407e-02 5.687e-02 5.019e-02 3.517e-02 2.639e-02 - 3.517e-02 5.019e-02 5.687e-02 5.407e-02 4.496e-02 3.255e-02 + 3.923e-02 3.245e-02 2.820e-02 + 4.206e-02 5.455e-02 5.829e-02 5.348e-02 4.253e-02 2.987e-02 + 2.401e-02 2.987e-02 4.253e-02 5.348e-02 5.829e-02 5.455e-02 + 4.206e-02 2.820e-02 2.820e-02 + 4.496e-02 5.406e-02 5.686e-02 5.018e-02 3.516e-02 2.637e-02 + 3.516e-02 5.018e-02 5.686e-02 5.406e-02 4.496e-02 3.255e-02 2.347e-02 2.347e-02 3.255e-02 - 5.407e-02 6.610e-02 6.682e-02 5.147e-02 2.974e-02 2.974e-02 - 5.147e-02 6.682e-02 6.610e-02 5.407e-02 3.967e-02 2.757e-02 + 5.406e-02 6.609e-02 6.681e-02 5.146e-02 2.973e-02 2.973e-02 + 5.146e-02 6.681e-02 6.609e-02 5.406e-02 3.967e-02 2.757e-02 2.272e-02 2.757e-02 3.967e-02 - 5.687e-02 6.682e-02 5.924e-02 3.678e-02 2.424e-02 3.678e-02 - 5.924e-02 6.682e-02 5.687e-02 4.213e-02 2.859e-02 2.065e-02 - 2.065e-02 2.859e-02 4.213e-02 - 5.019e-02 5.147e-02 3.678e-02 2.834e-02 2.834e-02 3.678e-02 - 5.147e-02 5.019e-02 3.817e-02 2.583e-02 1.720e-02 1.407e-02 + 5.686e-02 6.681e-02 5.923e-02 3.677e-02 2.423e-02 3.677e-02 + 5.923e-02 6.681e-02 5.686e-02 4.213e-02 2.859e-02 2.066e-02 + 2.066e-02 2.859e-02 4.213e-02 + 5.018e-02 5.146e-02 3.677e-02 2.834e-02 2.834e-02 3.677e-02 + 5.146e-02 5.018e-02 3.817e-02 2.583e-02 1.720e-02 1.407e-02 1.720e-02 2.583e-02 3.817e-02 - 3.517e-02 2.974e-02 2.424e-02 2.834e-02 2.424e-02 2.974e-02 - 3.517e-02 2.987e-02 2.075e-02 1.340e-02 9.462e-03 9.462e-03 + 3.516e-02 2.973e-02 2.423e-02 2.834e-02 2.423e-02 2.973e-02 + 3.516e-02 2.987e-02 2.075e-02 1.340e-02 9.466e-03 9.466e-03 1.340e-02 2.075e-02 2.987e-02 - 2.639e-02 2.974e-02 3.678e-02 3.678e-02 2.974e-02 2.639e-02 - 2.189e-02 1.506e-02 9.746e-03 6.238e-03 5.103e-03 6.238e-03 - 9.746e-03 1.506e-02 2.189e-02 - 3.517e-02 5.147e-02 5.924e-02 5.147e-02 3.517e-02 2.189e-02 - 1.272e-02 7.339e-03 4.549e-03 3.386e-03 3.386e-03 4.549e-03 - 7.339e-03 1.272e-02 2.189e-02 - 5.019e-02 6.682e-02 6.682e-02 5.019e-02 2.987e-02 1.506e-02 - 7.339e-03 3.980e-03 2.711e-03 2.404e-03 2.711e-03 3.980e-03 - 7.339e-03 1.506e-02 2.987e-02 - 5.687e-02 6.610e-02 5.687e-02 3.817e-02 2.075e-02 9.746e-03 - 4.549e-03 2.711e-03 2.197e-03 2.197e-03 2.711e-03 4.549e-03 - 9.746e-03 2.075e-02 3.817e-02 - 5.407e-02 5.407e-02 4.213e-02 2.583e-02 1.340e-02 6.238e-03 - 3.386e-03 2.404e-03 2.197e-03 2.404e-03 3.386e-03 6.238e-03 + 2.637e-02 2.973e-02 3.677e-02 3.677e-02 2.973e-02 2.637e-02 + 2.188e-02 1.506e-02 9.747e-03 6.241e-03 5.106e-03 6.241e-03 + 9.747e-03 1.506e-02 2.188e-02 + 3.516e-02 5.146e-02 5.923e-02 5.146e-02 3.516e-02 2.188e-02 + 1.271e-02 7.338e-03 4.550e-03 3.388e-03 3.388e-03 4.550e-03 + 7.338e-03 1.271e-02 2.188e-02 + 5.018e-02 6.681e-02 6.681e-02 5.018e-02 2.987e-02 1.506e-02 + 7.338e-03 3.981e-03 2.713e-03 2.406e-03 2.713e-03 3.981e-03 + 7.338e-03 1.506e-02 2.987e-02 + 5.686e-02 6.609e-02 5.686e-02 3.817e-02 2.075e-02 9.747e-03 + 4.550e-03 2.713e-03 2.198e-03 2.198e-03 2.713e-03 4.550e-03 + 9.747e-03 2.075e-02 3.817e-02 + 5.406e-02 5.406e-02 4.213e-02 2.583e-02 1.340e-02 6.241e-03 + 3.388e-03 2.406e-03 2.198e-03 2.406e-03 3.388e-03 6.241e-03 1.340e-02 2.583e-02 4.213e-02 - 4.496e-02 3.967e-02 2.859e-02 1.720e-02 9.462e-03 5.103e-03 - 3.386e-03 2.711e-03 2.711e-03 3.386e-03 5.103e-03 9.462e-03 + 4.496e-02 3.967e-02 2.859e-02 1.720e-02 9.466e-03 5.106e-03 + 3.388e-03 2.713e-03 2.713e-03 3.388e-03 5.106e-03 9.466e-03 1.720e-02 2.859e-02 3.967e-02 - 3.255e-02 2.757e-02 2.065e-02 1.407e-02 9.462e-03 6.238e-03 - 4.549e-03 3.980e-03 4.549e-03 6.238e-03 9.462e-03 1.407e-02 - 2.065e-02 2.757e-02 3.255e-02 - 2.347e-02 2.272e-02 2.065e-02 1.720e-02 1.340e-02 9.746e-03 - 7.339e-03 7.339e-03 9.746e-03 1.340e-02 1.720e-02 2.065e-02 + 3.255e-02 2.757e-02 2.066e-02 1.407e-02 9.466e-03 6.241e-03 + 4.550e-03 3.981e-03 4.550e-03 6.241e-03 9.466e-03 1.407e-02 + 2.066e-02 2.757e-02 3.255e-02 + 2.347e-02 2.272e-02 2.066e-02 1.720e-02 1.340e-02 9.747e-03 + 7.338e-03 7.338e-03 9.747e-03 1.340e-02 1.720e-02 2.066e-02 2.272e-02 2.347e-02 2.327e-02 2.347e-02 2.757e-02 2.859e-02 2.583e-02 2.075e-02 1.506e-02 - 1.272e-02 1.506e-02 2.075e-02 2.583e-02 2.859e-02 2.757e-02 + 1.271e-02 1.506e-02 2.075e-02 2.583e-02 2.859e-02 2.757e-02 2.347e-02 1.916e-02 1.916e-02 - 3.255e-02 3.967e-02 4.213e-02 3.817e-02 2.987e-02 2.189e-02 - 2.189e-02 2.987e-02 3.817e-02 4.213e-02 3.967e-02 3.255e-02 + 3.255e-02 3.967e-02 4.213e-02 3.817e-02 2.987e-02 2.188e-02 + 2.188e-02 2.987e-02 3.817e-02 4.213e-02 3.967e-02 3.255e-02 2.327e-02 1.916e-02 2.327e-02 - 3.013e-02 3.750e-02 4.149e-02 3.965e-02 3.517e-02 3.517e-02 - 3.965e-02 4.149e-02 3.750e-02 3.013e-02 2.182e-02 1.523e-02 - 1.264e-02 1.523e-02 2.182e-02 - 3.750e-02 4.772e-02 5.234e-02 5.147e-02 4.916e-02 5.147e-02 - 5.234e-02 4.772e-02 3.750e-02 2.653e-02 1.795e-02 1.312e-02 - 1.312e-02 1.795e-02 2.653e-02 - 4.149e-02 5.234e-02 5.924e-02 6.638e-02 6.638e-02 5.924e-02 - 5.234e-02 4.149e-02 2.898e-02 1.892e-02 1.265e-02 1.050e-02 - 1.265e-02 1.892e-02 2.898e-02 - 3.965e-02 5.147e-02 6.638e-02 7.723e-02 6.638e-02 5.147e-02 - 3.965e-02 2.780e-02 1.791e-02 1.127e-02 8.102e-03 8.102e-03 - 1.127e-02 1.791e-02 2.780e-02 - 3.517e-02 4.916e-02 6.638e-02 6.638e-02 4.916e-02 3.517e-02 - 2.402e-02 1.520e-02 9.366e-03 6.160e-03 5.124e-03 6.160e-03 - 9.366e-03 1.520e-02 2.402e-02 - 3.517e-02 5.147e-02 5.924e-02 5.147e-02 3.517e-02 2.189e-02 - 1.272e-02 7.339e-03 4.549e-03 3.386e-03 3.386e-03 4.549e-03 - 7.339e-03 1.272e-02 2.189e-02 - 3.965e-02 5.234e-02 5.234e-02 3.965e-02 2.402e-02 1.272e-02 - 6.592e-03 3.710e-03 2.530e-03 2.234e-03 2.530e-03 3.710e-03 - 6.592e-03 1.272e-02 2.402e-02 - 4.149e-02 4.772e-02 4.149e-02 2.780e-02 1.520e-02 7.339e-03 - 3.710e-03 2.300e-03 1.872e-03 1.872e-03 2.300e-03 3.710e-03 - 7.339e-03 1.520e-02 2.780e-02 - 3.750e-02 3.750e-02 2.898e-02 1.791e-02 9.366e-03 4.549e-03 - 2.530e-03 1.872e-03 1.724e-03 1.872e-03 2.530e-03 4.549e-03 - 9.366e-03 1.791e-02 2.898e-02 - 3.013e-02 2.653e-02 1.892e-02 1.127e-02 6.160e-03 3.386e-03 - 2.234e-03 1.872e-03 1.872e-03 2.234e-03 3.386e-03 6.160e-03 + 3.013e-02 3.749e-02 4.148e-02 3.964e-02 3.516e-02 3.516e-02 + 3.964e-02 4.148e-02 3.749e-02 3.013e-02 2.183e-02 1.523e-02 + 1.265e-02 1.523e-02 2.183e-02 + 3.749e-02 4.771e-02 5.233e-02 5.146e-02 4.915e-02 5.146e-02 + 5.233e-02 4.771e-02 3.749e-02 2.653e-02 1.795e-02 1.313e-02 + 1.313e-02 1.795e-02 2.653e-02 + 4.148e-02 5.233e-02 5.923e-02 6.638e-02 6.638e-02 5.923e-02 + 5.233e-02 4.148e-02 2.897e-02 1.892e-02 1.266e-02 1.050e-02 + 1.266e-02 1.892e-02 2.897e-02 + 3.964e-02 5.146e-02 6.638e-02 7.724e-02 6.638e-02 5.146e-02 + 3.964e-02 2.780e-02 1.790e-02 1.127e-02 8.105e-03 8.105e-03 + 1.127e-02 1.790e-02 2.780e-02 + 3.516e-02 4.915e-02 6.638e-02 6.638e-02 4.915e-02 3.516e-02 + 2.401e-02 1.520e-02 9.366e-03 6.163e-03 5.128e-03 6.163e-03 + 9.366e-03 1.520e-02 2.401e-02 + 3.516e-02 5.146e-02 5.923e-02 5.146e-02 3.516e-02 2.188e-02 + 1.271e-02 7.338e-03 4.550e-03 3.388e-03 3.388e-03 4.550e-03 + 7.338e-03 1.271e-02 2.188e-02 + 3.964e-02 5.233e-02 5.233e-02 3.964e-02 2.401e-02 1.271e-02 + 6.590e-03 3.711e-03 2.532e-03 2.236e-03 2.532e-03 3.711e-03 + 6.590e-03 1.271e-02 2.401e-02 + 4.148e-02 4.771e-02 4.148e-02 2.780e-02 1.520e-02 7.338e-03 + 3.711e-03 2.301e-03 1.874e-03 1.874e-03 2.301e-03 3.711e-03 + 7.338e-03 1.520e-02 2.780e-02 + 3.749e-02 3.749e-02 2.897e-02 1.790e-02 9.366e-03 4.550e-03 + 2.532e-03 1.874e-03 1.725e-03 1.874e-03 2.532e-03 4.550e-03 + 9.366e-03 1.790e-02 2.897e-02 + 3.013e-02 2.653e-02 1.892e-02 1.127e-02 6.163e-03 3.388e-03 + 2.236e-03 1.874e-03 1.874e-03 2.236e-03 3.388e-03 6.163e-03 1.127e-02 1.892e-02 2.653e-02 - 2.182e-02 1.795e-02 1.265e-02 8.102e-03 5.124e-03 3.386e-03 - 2.530e-03 2.300e-03 2.530e-03 3.386e-03 5.124e-03 8.102e-03 - 1.265e-02 1.795e-02 2.182e-02 - 1.523e-02 1.312e-02 1.050e-02 8.102e-03 6.160e-03 4.549e-03 - 3.710e-03 3.710e-03 4.549e-03 6.160e-03 8.102e-03 1.050e-02 - 1.312e-02 1.523e-02 1.600e-02 - 1.264e-02 1.312e-02 1.265e-02 1.127e-02 9.366e-03 7.339e-03 - 6.592e-03 7.339e-03 9.366e-03 1.127e-02 1.265e-02 1.312e-02 - 1.264e-02 1.184e-02 1.184e-02 - 1.523e-02 1.795e-02 1.892e-02 1.791e-02 1.520e-02 1.272e-02 - 1.272e-02 1.520e-02 1.791e-02 1.892e-02 1.795e-02 1.523e-02 + 2.183e-02 1.795e-02 1.266e-02 8.105e-03 5.128e-03 3.388e-03 + 2.532e-03 2.301e-03 2.532e-03 3.388e-03 5.128e-03 8.105e-03 + 1.266e-02 1.795e-02 2.183e-02 + 1.523e-02 1.313e-02 1.050e-02 8.105e-03 6.163e-03 4.550e-03 + 3.711e-03 3.711e-03 4.550e-03 6.163e-03 8.105e-03 1.050e-02 + 1.313e-02 1.523e-02 1.600e-02 + 1.265e-02 1.313e-02 1.266e-02 1.127e-02 9.366e-03 7.338e-03 + 6.590e-03 7.338e-03 9.366e-03 1.127e-02 1.266e-02 1.313e-02 + 1.265e-02 1.184e-02 1.184e-02 + 1.523e-02 1.795e-02 1.892e-02 1.790e-02 1.520e-02 1.271e-02 + 1.271e-02 1.520e-02 1.790e-02 1.892e-02 1.795e-02 1.523e-02 1.184e-02 1.044e-02 1.184e-02 - 2.182e-02 2.653e-02 2.898e-02 2.780e-02 2.402e-02 2.189e-02 - 2.402e-02 2.780e-02 2.898e-02 2.653e-02 2.182e-02 1.600e-02 + 2.183e-02 2.653e-02 2.897e-02 2.780e-02 2.401e-02 2.188e-02 + 2.401e-02 2.780e-02 2.897e-02 2.653e-02 2.183e-02 1.600e-02 1.184e-02 1.184e-02 1.600e-02 - 2.140e-02 3.084e-02 4.149e-02 5.019e-02 5.340e-02 5.019e-02 - 4.149e-02 3.084e-02 2.140e-02 1.445e-02 9.636e-03 7.208e-03 - 7.208e-03 9.636e-03 1.445e-02 - 3.084e-02 4.772e-02 6.682e-02 8.147e-02 8.147e-02 6.682e-02 - 4.772e-02 3.084e-02 1.895e-02 1.184e-02 7.950e-03 6.755e-03 - 7.950e-03 1.184e-02 1.895e-02 - 4.149e-02 6.682e-02 9.487e-02 1.094e-01 9.487e-02 6.682e-02 - 4.149e-02 2.401e-02 1.370e-02 8.411e-03 6.183e-03 6.183e-03 - 8.411e-03 1.370e-02 2.401e-02 - 5.019e-02 8.147e-02 1.094e-01 1.094e-01 8.147e-02 5.019e-02 - 2.780e-02 1.479e-02 8.316e-03 5.415e-03 4.623e-03 5.415e-03 - 8.316e-03 1.479e-02 2.780e-02 - 5.340e-02 8.147e-02 9.487e-02 8.147e-02 5.340e-02 2.987e-02 - 1.520e-02 7.945e-03 4.685e-03 3.516e-03 3.516e-03 4.685e-03 - 7.945e-03 1.520e-02 2.987e-02 - 5.019e-02 6.682e-02 6.682e-02 5.019e-02 2.987e-02 1.506e-02 - 7.339e-03 3.980e-03 2.711e-03 2.404e-03 2.711e-03 3.980e-03 - 7.339e-03 1.506e-02 2.987e-02 - 4.149e-02 4.772e-02 4.149e-02 2.780e-02 1.520e-02 7.339e-03 - 3.710e-03 2.300e-03 1.872e-03 1.872e-03 2.300e-03 3.710e-03 - 7.339e-03 1.520e-02 2.780e-02 - 3.084e-02 3.084e-02 2.401e-02 1.479e-02 7.945e-03 3.980e-03 - 2.300e-03 1.716e-03 1.594e-03 1.716e-03 2.300e-03 3.980e-03 - 7.945e-03 1.479e-02 2.401e-02 - 2.140e-02 1.895e-02 1.370e-02 8.316e-03 4.685e-03 2.711e-03 - 1.872e-03 1.594e-03 1.594e-03 1.872e-03 2.711e-03 4.685e-03 - 8.316e-03 1.370e-02 1.895e-02 - 1.445e-02 1.184e-02 8.411e-03 5.415e-03 3.516e-03 2.404e-03 - 1.872e-03 1.716e-03 1.872e-03 2.404e-03 3.516e-03 5.415e-03 - 8.411e-03 1.184e-02 1.445e-02 - 9.636e-03 7.950e-03 6.183e-03 4.623e-03 3.516e-03 2.711e-03 - 2.300e-03 2.300e-03 2.711e-03 3.516e-03 4.623e-03 6.183e-03 - 7.950e-03 9.636e-03 1.022e-02 - 7.208e-03 6.755e-03 6.183e-03 5.415e-03 4.685e-03 3.980e-03 - 3.710e-03 3.980e-03 4.685e-03 5.415e-03 6.183e-03 6.755e-03 - 7.208e-03 7.335e-03 7.335e-03 - 7.208e-03 7.950e-03 8.411e-03 8.316e-03 7.945e-03 7.339e-03 - 7.339e-03 7.945e-03 8.316e-03 8.411e-03 7.950e-03 7.208e-03 - 6.264e-03 5.856e-03 6.264e-03 - 9.636e-03 1.184e-02 1.370e-02 1.479e-02 1.520e-02 1.506e-02 - 1.520e-02 1.479e-02 1.370e-02 1.184e-02 9.636e-03 7.335e-03 - 5.856e-03 5.856e-03 7.335e-03 + 2.139e-02 3.083e-02 4.148e-02 5.018e-02 5.339e-02 5.018e-02 + 4.148e-02 3.083e-02 2.139e-02 1.445e-02 9.638e-03 7.210e-03 + 7.210e-03 9.638e-03 1.445e-02 + 3.083e-02 4.771e-02 6.681e-02 8.146e-02 8.146e-02 6.681e-02 + 4.771e-02 3.083e-02 1.895e-02 1.184e-02 7.953e-03 6.758e-03 + 7.953e-03 1.184e-02 1.895e-02 + 4.148e-02 6.681e-02 9.487e-02 1.094e-01 9.487e-02 6.681e-02 + 4.148e-02 2.401e-02 1.370e-02 8.413e-03 6.186e-03 6.186e-03 + 8.413e-03 1.370e-02 2.401e-02 + 5.018e-02 8.146e-02 1.094e-01 1.094e-01 8.146e-02 5.018e-02 + 2.780e-02 1.479e-02 8.317e-03 5.417e-03 4.626e-03 5.417e-03 + 8.317e-03 1.479e-02 2.780e-02 + 5.339e-02 8.146e-02 9.487e-02 8.146e-02 5.339e-02 2.987e-02 + 1.520e-02 7.944e-03 4.687e-03 3.518e-03 3.518e-03 4.687e-03 + 7.944e-03 1.520e-02 2.987e-02 + 5.018e-02 6.681e-02 6.681e-02 5.018e-02 2.987e-02 1.506e-02 + 7.338e-03 3.981e-03 2.713e-03 2.406e-03 2.713e-03 3.981e-03 + 7.338e-03 1.506e-02 2.987e-02 + 4.148e-02 4.771e-02 4.148e-02 2.780e-02 1.520e-02 7.338e-03 + 3.711e-03 2.301e-03 1.874e-03 1.874e-03 2.301e-03 3.711e-03 + 7.338e-03 1.520e-02 2.780e-02 + 3.083e-02 3.083e-02 2.401e-02 1.479e-02 7.944e-03 3.981e-03 + 2.301e-03 1.717e-03 1.596e-03 1.717e-03 2.301e-03 3.981e-03 + 7.944e-03 1.479e-02 2.401e-02 + 2.139e-02 1.895e-02 1.370e-02 8.317e-03 4.687e-03 2.713e-03 + 1.874e-03 1.596e-03 1.596e-03 1.874e-03 2.713e-03 4.687e-03 + 8.317e-03 1.370e-02 1.895e-02 + 1.445e-02 1.184e-02 8.413e-03 5.417e-03 3.518e-03 2.406e-03 + 1.874e-03 1.717e-03 1.874e-03 2.406e-03 3.518e-03 5.417e-03 + 8.413e-03 1.184e-02 1.445e-02 + 9.638e-03 7.953e-03 6.186e-03 4.626e-03 3.518e-03 2.713e-03 + 2.301e-03 2.301e-03 2.713e-03 3.518e-03 4.626e-03 6.186e-03 + 7.953e-03 9.638e-03 1.022e-02 + 7.210e-03 6.758e-03 6.186e-03 5.417e-03 4.687e-03 3.981e-03 + 3.711e-03 3.981e-03 4.687e-03 5.417e-03 6.186e-03 6.758e-03 + 7.210e-03 7.338e-03 7.338e-03 + 7.210e-03 7.953e-03 8.413e-03 8.317e-03 7.944e-03 7.338e-03 + 7.338e-03 7.944e-03 8.317e-03 8.413e-03 7.953e-03 7.210e-03 + 6.267e-03 5.859e-03 6.267e-03 + 9.638e-03 1.184e-02 1.370e-02 1.479e-02 1.520e-02 1.506e-02 + 1.520e-02 1.479e-02 1.370e-02 1.184e-02 9.638e-03 7.338e-03 + 5.859e-03 5.859e-03 7.338e-03 1.445e-02 1.895e-02 2.401e-02 2.780e-02 2.987e-02 2.987e-02 - 2.780e-02 2.401e-02 1.895e-02 1.445e-02 1.022e-02 7.335e-03 - 6.264e-03 7.335e-03 1.022e-02 - 2.140e-02 3.750e-02 5.687e-02 7.202e-02 7.202e-02 5.687e-02 - 3.750e-02 2.140e-02 1.179e-02 6.907e-03 4.700e-03 4.099e-03 - 4.700e-03 6.907e-03 1.179e-02 - 3.750e-02 6.610e-02 9.562e-02 1.093e-01 9.562e-02 6.610e-02 - 3.750e-02 1.895e-02 9.768e-03 5.832e-03 4.402e-03 4.402e-03 - 5.832e-03 9.768e-03 1.895e-02 - 5.687e-02 9.562e-02 1.269e-01 1.269e-01 9.562e-02 5.687e-02 - 2.898e-02 1.370e-02 7.112e-03 4.615e-03 4.017e-03 4.615e-03 - 7.112e-03 1.370e-02 2.898e-02 + 2.780e-02 2.401e-02 1.895e-02 1.445e-02 1.022e-02 7.338e-03 + 6.267e-03 7.338e-03 1.022e-02 + 2.139e-02 3.749e-02 5.686e-02 7.202e-02 7.202e-02 5.686e-02 + 3.749e-02 2.139e-02 1.179e-02 6.909e-03 4.702e-03 4.101e-03 + 4.702e-03 6.909e-03 1.179e-02 + 3.749e-02 6.609e-02 9.562e-02 1.093e-01 9.562e-02 6.609e-02 + 3.749e-02 1.895e-02 9.768e-03 5.833e-03 4.404e-03 4.404e-03 + 5.833e-03 9.768e-03 1.895e-02 + 5.686e-02 9.562e-02 1.269e-01 1.269e-01 9.562e-02 5.686e-02 + 2.897e-02 1.370e-02 7.113e-03 4.616e-03 4.019e-03 4.616e-03 + 7.113e-03 1.370e-02 2.897e-02 7.202e-02 1.093e-01 1.269e-01 1.093e-01 7.202e-02 3.817e-02 - 1.791e-02 8.316e-03 4.687e-03 3.581e-03 3.581e-03 4.687e-03 - 8.316e-03 1.791e-02 3.817e-02 - 7.202e-02 9.562e-02 9.562e-02 7.202e-02 4.254e-02 2.075e-02 - 9.366e-03 4.685e-03 3.128e-03 2.806e-03 3.128e-03 4.685e-03 - 9.366e-03 2.075e-02 4.254e-02 - 5.687e-02 6.610e-02 5.687e-02 3.817e-02 2.075e-02 9.746e-03 - 4.549e-03 2.711e-03 2.197e-03 2.197e-03 2.711e-03 4.549e-03 - 9.746e-03 2.075e-02 3.817e-02 - 3.750e-02 3.750e-02 2.898e-02 1.791e-02 9.366e-03 4.549e-03 - 2.530e-03 1.872e-03 1.724e-03 1.872e-03 2.530e-03 4.549e-03 - 9.366e-03 1.791e-02 2.898e-02 - 2.140e-02 1.895e-02 1.370e-02 8.316e-03 4.685e-03 2.711e-03 - 1.872e-03 1.594e-03 1.594e-03 1.872e-03 2.711e-03 4.685e-03 - 8.316e-03 1.370e-02 1.895e-02 - 1.179e-02 9.768e-03 7.112e-03 4.687e-03 3.128e-03 2.197e-03 - 1.724e-03 1.594e-03 1.724e-03 2.197e-03 3.128e-03 4.687e-03 - 7.112e-03 9.768e-03 1.179e-02 - 6.907e-03 5.832e-03 4.615e-03 3.581e-03 2.806e-03 2.197e-03 - 1.872e-03 1.872e-03 2.197e-03 2.806e-03 3.581e-03 4.615e-03 - 5.832e-03 6.907e-03 7.107e-03 - 4.700e-03 4.402e-03 4.017e-03 3.581e-03 3.128e-03 2.711e-03 - 2.530e-03 2.711e-03 3.128e-03 3.581e-03 4.017e-03 4.402e-03 - 4.700e-03 4.859e-03 4.859e-03 - 4.099e-03 4.402e-03 4.615e-03 4.687e-03 4.685e-03 4.549e-03 - 4.549e-03 4.685e-03 4.687e-03 4.615e-03 4.402e-03 4.099e-03 - 3.778e-03 3.600e-03 3.778e-03 - 4.700e-03 5.832e-03 7.112e-03 8.316e-03 9.366e-03 9.746e-03 - 9.366e-03 8.316e-03 7.112e-03 5.832e-03 4.700e-03 3.778e-03 - 3.212e-03 3.212e-03 3.778e-03 - 6.907e-03 9.768e-03 1.370e-02 1.791e-02 2.075e-02 2.075e-02 - 1.791e-02 1.370e-02 9.768e-03 6.907e-03 4.859e-03 3.600e-03 - 3.212e-03 3.600e-03 4.859e-03 - 1.179e-02 1.895e-02 2.898e-02 3.817e-02 4.254e-02 3.817e-02 - 2.898e-02 1.895e-02 1.179e-02 7.107e-03 4.859e-03 3.778e-03 - 3.778e-03 4.859e-03 7.107e-03 - 3.013e-02 5.407e-02 7.874e-02 9.038e-02 7.874e-02 5.407e-02 - 3.013e-02 1.445e-02 6.907e-03 4.025e-03 3.161e-03 3.161e-03 - 4.025e-03 6.907e-03 1.445e-02 - 5.407e-02 9.101e-02 1.188e-01 1.188e-01 9.101e-02 5.407e-02 - 2.653e-02 1.184e-02 5.832e-03 3.833e-03 3.410e-03 3.833e-03 - 5.832e-03 1.184e-02 2.653e-02 + 1.790e-02 8.317e-03 4.689e-03 3.583e-03 3.583e-03 4.689e-03 + 8.317e-03 1.790e-02 3.817e-02 + 7.202e-02 9.562e-02 9.562e-02 7.202e-02 4.253e-02 2.075e-02 + 9.366e-03 4.687e-03 3.130e-03 2.808e-03 3.130e-03 4.687e-03 + 9.366e-03 2.075e-02 4.253e-02 + 5.686e-02 6.609e-02 5.686e-02 3.817e-02 2.075e-02 9.747e-03 + 4.550e-03 2.713e-03 2.198e-03 2.198e-03 2.713e-03 4.550e-03 + 9.747e-03 2.075e-02 3.817e-02 + 3.749e-02 3.749e-02 2.897e-02 1.790e-02 9.366e-03 4.550e-03 + 2.532e-03 1.874e-03 1.725e-03 1.874e-03 2.532e-03 4.550e-03 + 9.366e-03 1.790e-02 2.897e-02 + 2.139e-02 1.895e-02 1.370e-02 8.317e-03 4.687e-03 2.713e-03 + 1.874e-03 1.596e-03 1.596e-03 1.874e-03 2.713e-03 4.687e-03 + 8.317e-03 1.370e-02 1.895e-02 + 1.179e-02 9.768e-03 7.113e-03 4.689e-03 3.130e-03 2.198e-03 + 1.725e-03 1.596e-03 1.725e-03 2.198e-03 3.130e-03 4.689e-03 + 7.113e-03 9.768e-03 1.179e-02 + 6.909e-03 5.833e-03 4.616e-03 3.583e-03 2.808e-03 2.198e-03 + 1.874e-03 1.874e-03 2.198e-03 2.808e-03 3.583e-03 4.616e-03 + 5.833e-03 6.909e-03 7.108e-03 + 4.702e-03 4.404e-03 4.019e-03 3.583e-03 3.130e-03 2.713e-03 + 2.532e-03 2.713e-03 3.130e-03 3.583e-03 4.019e-03 4.404e-03 + 4.702e-03 4.861e-03 4.861e-03 + 4.101e-03 4.404e-03 4.616e-03 4.689e-03 4.687e-03 4.550e-03 + 4.550e-03 4.687e-03 4.689e-03 4.616e-03 4.404e-03 4.101e-03 + 3.780e-03 3.603e-03 3.780e-03 + 4.702e-03 5.833e-03 7.113e-03 8.317e-03 9.366e-03 9.747e-03 + 9.366e-03 8.317e-03 7.113e-03 5.833e-03 4.702e-03 3.780e-03 + 3.214e-03 3.214e-03 3.780e-03 + 6.909e-03 9.768e-03 1.370e-02 1.790e-02 2.075e-02 2.075e-02 + 1.790e-02 1.370e-02 9.768e-03 6.909e-03 4.861e-03 3.603e-03 + 3.214e-03 3.603e-03 4.861e-03 + 1.179e-02 1.895e-02 2.897e-02 3.817e-02 4.253e-02 3.817e-02 + 2.897e-02 1.895e-02 1.179e-02 7.108e-03 4.861e-03 3.780e-03 + 3.780e-03 4.861e-03 7.108e-03 + 3.013e-02 5.406e-02 7.874e-02 9.038e-02 7.874e-02 5.406e-02 + 3.013e-02 1.445e-02 6.909e-03 4.027e-03 3.163e-03 3.163e-03 + 4.027e-03 6.909e-03 1.445e-02 + 5.406e-02 9.101e-02 1.188e-01 1.188e-01 9.101e-02 5.406e-02 + 2.653e-02 1.184e-02 5.833e-03 3.835e-03 3.412e-03 3.835e-03 + 5.833e-03 1.184e-02 2.653e-02 7.874e-02 1.188e-01 1.384e-01 1.188e-01 7.874e-02 4.213e-02 - 1.892e-02 8.411e-03 4.615e-03 3.559e-03 3.559e-03 4.615e-03 - 8.411e-03 1.892e-02 4.213e-02 + 1.892e-02 8.413e-03 4.616e-03 3.561e-03 3.561e-03 4.616e-03 + 8.413e-03 1.892e-02 4.213e-02 9.038e-02 1.188e-01 1.188e-01 9.038e-02 5.348e-02 2.583e-02 - 1.127e-02 5.415e-03 3.581e-03 3.198e-03 3.581e-03 5.415e-03 + 1.127e-02 5.417e-03 3.583e-03 3.200e-03 3.583e-03 5.417e-03 1.127e-02 2.583e-02 5.348e-02 7.874e-02 9.101e-02 7.874e-02 5.348e-02 2.919e-02 1.340e-02 - 6.160e-03 3.516e-03 2.806e-03 2.806e-03 3.516e-03 6.160e-03 + 6.163e-03 3.518e-03 2.808e-03 2.808e-03 3.518e-03 6.163e-03 1.340e-02 2.919e-02 5.348e-02 - 5.407e-02 5.407e-02 4.213e-02 2.583e-02 1.340e-02 6.238e-03 - 3.386e-03 2.404e-03 2.197e-03 2.404e-03 3.386e-03 6.238e-03 + 5.406e-02 5.406e-02 4.213e-02 2.583e-02 1.340e-02 6.241e-03 + 3.388e-03 2.406e-03 2.198e-03 2.406e-03 3.388e-03 6.241e-03 1.340e-02 2.583e-02 4.213e-02 - 3.013e-02 2.653e-02 1.892e-02 1.127e-02 6.160e-03 3.386e-03 - 2.234e-03 1.872e-03 1.872e-03 2.234e-03 3.386e-03 6.160e-03 + 3.013e-02 2.653e-02 1.892e-02 1.127e-02 6.163e-03 3.388e-03 + 2.236e-03 1.874e-03 1.874e-03 2.236e-03 3.388e-03 6.163e-03 1.127e-02 1.892e-02 2.653e-02 - 1.445e-02 1.184e-02 8.411e-03 5.415e-03 3.516e-03 2.404e-03 - 1.872e-03 1.716e-03 1.872e-03 2.404e-03 3.516e-03 5.415e-03 - 8.411e-03 1.184e-02 1.445e-02 - 6.907e-03 5.832e-03 4.615e-03 3.581e-03 2.806e-03 2.197e-03 - 1.872e-03 1.872e-03 2.197e-03 2.806e-03 3.581e-03 4.615e-03 - 5.832e-03 6.907e-03 7.107e-03 - 4.025e-03 3.833e-03 3.559e-03 3.198e-03 2.806e-03 2.404e-03 - 2.234e-03 2.404e-03 2.806e-03 3.198e-03 3.559e-03 3.833e-03 - 4.025e-03 4.050e-03 4.050e-03 - 3.161e-03 3.410e-03 3.559e-03 3.581e-03 3.516e-03 3.386e-03 - 3.386e-03 3.516e-03 3.581e-03 3.559e-03 3.410e-03 3.161e-03 - 2.901e-03 2.741e-03 2.901e-03 - 3.161e-03 3.833e-03 4.615e-03 5.415e-03 6.160e-03 6.238e-03 - 6.160e-03 5.415e-03 4.615e-03 3.833e-03 3.161e-03 2.611e-03 - 2.246e-03 2.246e-03 2.611e-03 - 4.025e-03 5.832e-03 8.411e-03 1.127e-02 1.340e-02 1.340e-02 - 1.127e-02 8.411e-03 5.832e-03 4.025e-03 2.901e-03 2.246e-03 - 2.029e-03 2.246e-03 2.901e-03 - 6.907e-03 1.184e-02 1.892e-02 2.583e-02 2.919e-02 2.583e-02 - 1.892e-02 1.184e-02 6.907e-03 4.050e-03 2.741e-03 2.246e-03 - 2.246e-03 2.741e-03 4.050e-03 + 1.445e-02 1.184e-02 8.413e-03 5.417e-03 3.518e-03 2.406e-03 + 1.874e-03 1.717e-03 1.874e-03 2.406e-03 3.518e-03 5.417e-03 + 8.413e-03 1.184e-02 1.445e-02 + 6.909e-03 5.833e-03 4.616e-03 3.583e-03 2.808e-03 2.198e-03 + 1.874e-03 1.874e-03 2.198e-03 2.808e-03 3.583e-03 4.616e-03 + 5.833e-03 6.909e-03 7.108e-03 + 4.027e-03 3.835e-03 3.561e-03 3.200e-03 2.808e-03 2.406e-03 + 2.236e-03 2.406e-03 2.808e-03 3.200e-03 3.561e-03 3.835e-03 + 4.027e-03 4.051e-03 4.051e-03 + 3.163e-03 3.412e-03 3.561e-03 3.583e-03 3.518e-03 3.388e-03 + 3.388e-03 3.518e-03 3.583e-03 3.561e-03 3.412e-03 3.163e-03 + 2.903e-03 2.743e-03 2.903e-03 + 3.163e-03 3.835e-03 4.616e-03 5.417e-03 6.163e-03 6.241e-03 + 6.163e-03 5.417e-03 4.616e-03 3.835e-03 3.163e-03 2.613e-03 + 2.248e-03 2.248e-03 2.613e-03 + 4.027e-03 5.833e-03 8.413e-03 1.127e-02 1.340e-02 1.340e-02 + 1.127e-02 8.413e-03 5.833e-03 4.027e-03 2.903e-03 2.248e-03 + 2.031e-03 2.248e-03 2.903e-03 + 6.909e-03 1.184e-02 1.892e-02 2.583e-02 2.919e-02 2.583e-02 + 1.892e-02 1.184e-02 6.909e-03 4.051e-03 2.743e-03 2.248e-03 + 2.248e-03 2.743e-03 4.051e-03 1.445e-02 2.653e-02 4.213e-02 5.348e-02 5.348e-02 4.213e-02 - 2.653e-02 1.445e-02 7.107e-03 4.050e-03 2.901e-03 2.611e-03 - 2.901e-03 4.050e-03 7.107e-03 - 4.496e-02 7.503e-02 9.861e-02 9.861e-02 7.503e-02 4.496e-02 - 2.182e-02 9.636e-03 4.700e-03 3.161e-03 2.848e-03 3.161e-03 - 4.700e-03 9.636e-03 2.182e-02 + 2.653e-02 1.445e-02 7.108e-03 4.051e-03 2.903e-03 2.613e-03 + 2.903e-03 4.051e-03 7.108e-03 + 4.496e-02 7.503e-02 9.860e-02 9.860e-02 7.503e-02 4.496e-02 + 2.183e-02 9.638e-03 4.702e-03 3.163e-03 2.850e-03 3.163e-03 + 4.702e-03 9.638e-03 2.183e-02 7.503e-02 1.128e-01 1.311e-01 1.128e-01 7.503e-02 3.967e-02 - 1.795e-02 7.950e-03 4.402e-03 3.410e-03 3.410e-03 4.402e-03 - 7.950e-03 1.795e-02 3.967e-02 - 9.861e-02 1.311e-01 1.311e-01 9.861e-02 5.829e-02 2.859e-02 - 1.265e-02 6.183e-03 4.017e-03 3.559e-03 4.017e-03 6.183e-03 - 1.265e-02 2.859e-02 5.829e-02 - 9.861e-02 1.128e-01 9.861e-02 6.717e-02 3.637e-02 1.720e-02 - 8.102e-03 4.623e-03 3.581e-03 3.581e-03 4.623e-03 8.102e-03 + 1.795e-02 7.953e-03 4.404e-03 3.412e-03 3.412e-03 4.404e-03 + 7.953e-03 1.795e-02 3.967e-02 + 9.860e-02 1.311e-01 1.311e-01 9.860e-02 5.829e-02 2.859e-02 + 1.266e-02 6.186e-03 4.019e-03 3.561e-03 4.019e-03 6.186e-03 + 1.266e-02 2.859e-02 5.829e-02 + 9.860e-02 1.128e-01 9.860e-02 6.717e-02 3.637e-02 1.720e-02 + 8.105e-03 4.626e-03 3.583e-03 3.583e-03 4.626e-03 8.105e-03 1.720e-02 3.637e-02 6.717e-02 - 7.503e-02 7.503e-02 5.829e-02 3.637e-02 1.938e-02 9.462e-03 - 5.124e-03 3.516e-03 3.128e-03 3.516e-03 5.124e-03 9.462e-03 + 7.503e-02 7.503e-02 5.829e-02 3.637e-02 1.938e-02 9.466e-03 + 5.128e-03 3.518e-03 3.130e-03 3.518e-03 5.128e-03 9.466e-03 1.938e-02 3.637e-02 5.829e-02 - 4.496e-02 3.967e-02 2.859e-02 1.720e-02 9.462e-03 5.103e-03 - 3.386e-03 2.711e-03 2.711e-03 3.386e-03 5.103e-03 9.462e-03 + 4.496e-02 3.967e-02 2.859e-02 1.720e-02 9.466e-03 5.106e-03 + 3.388e-03 2.713e-03 2.713e-03 3.388e-03 5.106e-03 9.466e-03 1.720e-02 2.859e-02 3.967e-02 - 2.182e-02 1.795e-02 1.265e-02 8.102e-03 5.124e-03 3.386e-03 - 2.530e-03 2.300e-03 2.530e-03 3.386e-03 5.124e-03 8.102e-03 - 1.265e-02 1.795e-02 2.182e-02 - 9.636e-03 7.950e-03 6.183e-03 4.623e-03 3.516e-03 2.711e-03 - 2.300e-03 2.300e-03 2.711e-03 3.516e-03 4.623e-03 6.183e-03 - 7.950e-03 9.636e-03 1.022e-02 - 4.700e-03 4.402e-03 4.017e-03 3.581e-03 3.128e-03 2.711e-03 - 2.530e-03 2.711e-03 3.128e-03 3.581e-03 4.017e-03 4.402e-03 - 4.700e-03 4.859e-03 4.859e-03 - 3.161e-03 3.410e-03 3.559e-03 3.581e-03 3.516e-03 3.386e-03 - 3.386e-03 3.516e-03 3.581e-03 3.559e-03 3.410e-03 3.161e-03 - 2.901e-03 2.741e-03 2.901e-03 - 2.848e-03 3.410e-03 4.017e-03 4.623e-03 5.124e-03 5.103e-03 - 5.124e-03 4.623e-03 4.017e-03 3.410e-03 2.848e-03 2.365e-03 - 2.016e-03 2.016e-03 2.365e-03 - 3.161e-03 4.402e-03 6.183e-03 8.102e-03 9.462e-03 9.462e-03 - 8.102e-03 6.183e-03 4.402e-03 3.161e-03 2.365e-03 1.853e-03 - 1.702e-03 1.853e-03 2.365e-03 - 4.700e-03 7.950e-03 1.265e-02 1.720e-02 1.938e-02 1.720e-02 - 1.265e-02 7.950e-03 4.700e-03 2.901e-03 2.016e-03 1.702e-03 - 1.702e-03 2.016e-03 2.901e-03 - 9.636e-03 1.795e-02 2.859e-02 3.637e-02 3.637e-02 2.859e-02 - 1.795e-02 9.636e-03 4.859e-03 2.741e-03 2.016e-03 1.853e-03 - 2.016e-03 2.741e-03 4.859e-03 - 2.182e-02 3.967e-02 5.829e-02 6.717e-02 5.829e-02 3.967e-02 - 2.182e-02 1.022e-02 4.859e-03 2.901e-03 2.365e-03 2.365e-03 - 2.901e-03 4.859e-03 1.022e-02 + 2.183e-02 1.795e-02 1.266e-02 8.105e-03 5.128e-03 3.388e-03 + 2.532e-03 2.301e-03 2.532e-03 3.388e-03 5.128e-03 8.105e-03 + 1.266e-02 1.795e-02 2.183e-02 + 9.638e-03 7.953e-03 6.186e-03 4.626e-03 3.518e-03 2.713e-03 + 2.301e-03 2.301e-03 2.713e-03 3.518e-03 4.626e-03 6.186e-03 + 7.953e-03 9.638e-03 1.022e-02 + 4.702e-03 4.404e-03 4.019e-03 3.583e-03 3.130e-03 2.713e-03 + 2.532e-03 2.713e-03 3.130e-03 3.583e-03 4.019e-03 4.404e-03 + 4.702e-03 4.861e-03 4.861e-03 + 3.163e-03 3.412e-03 3.561e-03 3.583e-03 3.518e-03 3.388e-03 + 3.388e-03 3.518e-03 3.583e-03 3.561e-03 3.412e-03 3.163e-03 + 2.903e-03 2.743e-03 2.903e-03 + 2.850e-03 3.412e-03 4.019e-03 4.626e-03 5.128e-03 5.106e-03 + 5.128e-03 4.626e-03 4.019e-03 3.412e-03 2.850e-03 2.367e-03 + 2.017e-03 2.017e-03 2.367e-03 + 3.163e-03 4.404e-03 6.186e-03 8.105e-03 9.466e-03 9.466e-03 + 8.105e-03 6.186e-03 4.404e-03 3.163e-03 2.367e-03 1.855e-03 + 1.703e-03 1.855e-03 2.367e-03 + 4.702e-03 7.953e-03 1.266e-02 1.720e-02 1.938e-02 1.720e-02 + 1.266e-02 7.953e-03 4.702e-03 2.903e-03 2.017e-03 1.703e-03 + 1.703e-03 2.017e-03 2.903e-03 + 9.638e-03 1.795e-02 2.859e-02 3.637e-02 3.637e-02 2.859e-02 + 1.795e-02 9.638e-03 4.861e-03 2.743e-03 2.017e-03 1.855e-03 + 2.017e-03 2.743e-03 4.861e-03 + 2.183e-02 3.967e-02 5.829e-02 6.717e-02 5.829e-02 3.967e-02 + 2.183e-02 1.022e-02 4.861e-03 2.903e-03 2.367e-03 2.367e-03 + 2.903e-03 4.861e-03 1.022e-02 6.109e-02 9.268e-02 1.077e-01 9.268e-02 6.109e-02 3.255e-02 - 1.523e-02 7.208e-03 4.099e-03 3.161e-03 3.161e-03 4.099e-03 - 7.208e-03 1.523e-02 3.255e-02 + 1.523e-02 7.210e-03 4.101e-03 3.163e-03 3.163e-03 4.101e-03 + 7.210e-03 1.523e-02 3.255e-02 9.268e-02 1.242e-01 1.242e-01 9.268e-02 5.455e-02 2.757e-02 - 1.312e-02 6.755e-03 4.402e-03 3.833e-03 4.402e-03 6.755e-03 - 1.312e-02 2.757e-02 5.455e-02 - 1.077e-01 1.242e-01 1.077e-01 7.273e-02 4.066e-02 2.065e-02 - 1.050e-02 6.183e-03 4.615e-03 4.615e-03 6.183e-03 1.050e-02 - 2.065e-02 4.066e-02 7.273e-02 + 1.313e-02 6.758e-03 4.404e-03 3.835e-03 4.404e-03 6.758e-03 + 1.313e-02 2.757e-02 5.455e-02 + 1.077e-01 1.242e-01 1.077e-01 7.273e-02 4.066e-02 2.066e-02 + 1.050e-02 6.186e-03 4.616e-03 4.616e-03 6.186e-03 1.050e-02 + 2.066e-02 4.066e-02 7.273e-02 9.268e-02 9.268e-02 7.273e-02 4.690e-02 2.620e-02 1.407e-02 - 8.102e-03 5.415e-03 4.687e-03 5.415e-03 8.102e-03 1.407e-02 + 8.105e-03 5.417e-03 4.689e-03 5.417e-03 8.105e-03 1.407e-02 2.620e-02 4.690e-02 7.273e-02 - 6.109e-02 5.455e-02 4.066e-02 2.620e-02 1.586e-02 9.462e-03 - 6.160e-03 4.685e-03 4.685e-03 6.160e-03 9.462e-03 1.586e-02 + 6.109e-02 5.455e-02 4.066e-02 2.620e-02 1.586e-02 9.466e-03 + 6.163e-03 4.687e-03 4.687e-03 6.163e-03 9.466e-03 1.586e-02 2.620e-02 4.066e-02 5.455e-02 - 3.255e-02 2.757e-02 2.065e-02 1.407e-02 9.462e-03 6.238e-03 - 4.549e-03 3.980e-03 4.549e-03 6.238e-03 9.462e-03 1.407e-02 - 2.065e-02 2.757e-02 3.255e-02 - 1.523e-02 1.312e-02 1.050e-02 8.102e-03 6.160e-03 4.549e-03 - 3.710e-03 3.710e-03 4.549e-03 6.160e-03 8.102e-03 1.050e-02 - 1.312e-02 1.523e-02 1.600e-02 - 7.208e-03 6.755e-03 6.183e-03 5.415e-03 4.685e-03 3.980e-03 - 3.710e-03 3.980e-03 4.685e-03 5.415e-03 6.183e-03 6.755e-03 - 7.208e-03 7.335e-03 7.335e-03 - 4.099e-03 4.402e-03 4.615e-03 4.687e-03 4.685e-03 4.549e-03 - 4.549e-03 4.685e-03 4.687e-03 4.615e-03 4.402e-03 4.099e-03 - 3.778e-03 3.600e-03 3.778e-03 - 3.161e-03 3.833e-03 4.615e-03 5.415e-03 6.160e-03 6.238e-03 - 6.160e-03 5.415e-03 4.615e-03 3.833e-03 3.161e-03 2.611e-03 - 2.246e-03 2.246e-03 2.611e-03 - 3.161e-03 4.402e-03 6.183e-03 8.102e-03 9.462e-03 9.462e-03 - 8.102e-03 6.183e-03 4.402e-03 3.161e-03 2.365e-03 1.853e-03 - 1.702e-03 1.853e-03 2.365e-03 - 4.099e-03 6.755e-03 1.050e-02 1.407e-02 1.586e-02 1.407e-02 - 1.050e-02 6.755e-03 4.099e-03 2.611e-03 1.853e-03 1.578e-03 - 1.578e-03 1.853e-03 2.611e-03 - 7.208e-03 1.312e-02 2.065e-02 2.620e-02 2.620e-02 2.065e-02 - 1.312e-02 7.208e-03 3.778e-03 2.246e-03 1.702e-03 1.578e-03 - 1.702e-03 2.246e-03 3.778e-03 + 3.255e-02 2.757e-02 2.066e-02 1.407e-02 9.466e-03 6.241e-03 + 4.550e-03 3.981e-03 4.550e-03 6.241e-03 9.466e-03 1.407e-02 + 2.066e-02 2.757e-02 3.255e-02 + 1.523e-02 1.313e-02 1.050e-02 8.105e-03 6.163e-03 4.550e-03 + 3.711e-03 3.711e-03 4.550e-03 6.163e-03 8.105e-03 1.050e-02 + 1.313e-02 1.523e-02 1.600e-02 + 7.210e-03 6.758e-03 6.186e-03 5.417e-03 4.687e-03 3.981e-03 + 3.711e-03 3.981e-03 4.687e-03 5.417e-03 6.186e-03 6.758e-03 + 7.210e-03 7.338e-03 7.338e-03 + 4.101e-03 4.404e-03 4.616e-03 4.689e-03 4.687e-03 4.550e-03 + 4.550e-03 4.687e-03 4.689e-03 4.616e-03 4.404e-03 4.101e-03 + 3.780e-03 3.603e-03 3.780e-03 + 3.163e-03 3.835e-03 4.616e-03 5.417e-03 6.163e-03 6.241e-03 + 6.163e-03 5.417e-03 4.616e-03 3.835e-03 3.163e-03 2.613e-03 + 2.248e-03 2.248e-03 2.613e-03 + 3.163e-03 4.404e-03 6.186e-03 8.105e-03 9.466e-03 9.466e-03 + 8.105e-03 6.186e-03 4.404e-03 3.163e-03 2.367e-03 1.855e-03 + 1.703e-03 1.855e-03 2.367e-03 + 4.101e-03 6.758e-03 1.050e-02 1.407e-02 1.586e-02 1.407e-02 + 1.050e-02 6.758e-03 4.101e-03 2.613e-03 1.855e-03 1.579e-03 + 1.579e-03 1.855e-03 2.613e-03 + 7.210e-03 1.313e-02 2.066e-02 2.620e-02 2.620e-02 2.066e-02 + 1.313e-02 7.210e-03 3.780e-03 2.248e-03 1.703e-03 1.579e-03 + 1.703e-03 2.248e-03 3.780e-03 1.523e-02 2.757e-02 4.066e-02 4.690e-02 4.066e-02 2.757e-02 - 1.523e-02 7.335e-03 3.600e-03 2.246e-03 1.853e-03 1.853e-03 - 2.246e-03 3.600e-03 7.335e-03 + 1.523e-02 7.338e-03 3.603e-03 2.248e-03 1.855e-03 1.855e-03 + 2.248e-03 3.603e-03 7.338e-03 3.255e-02 5.455e-02 7.273e-02 7.273e-02 5.455e-02 3.255e-02 - 1.600e-02 7.335e-03 3.778e-03 2.611e-03 2.365e-03 2.611e-03 - 3.778e-03 7.335e-03 1.600e-02 - 6.777e-02 9.277e-02 9.277e-02 6.777e-02 4.205e-02 2.347e-02 - 1.264e-02 7.208e-03 4.700e-03 4.025e-03 4.700e-03 7.208e-03 - 1.264e-02 2.347e-02 4.205e-02 - 9.277e-02 1.075e-01 9.277e-02 6.359e-02 3.923e-02 2.272e-02 - 1.312e-02 7.950e-03 5.832e-03 5.832e-03 7.950e-03 1.312e-02 - 2.272e-02 3.923e-02 6.359e-02 - 9.277e-02 9.277e-02 7.321e-02 5.204e-02 3.364e-02 2.065e-02 - 1.265e-02 8.411e-03 7.112e-03 8.411e-03 1.265e-02 2.065e-02 - 3.364e-02 5.204e-02 7.321e-02 - 6.777e-02 6.359e-02 5.204e-02 3.841e-02 2.620e-02 1.720e-02 - 1.127e-02 8.316e-03 8.316e-03 1.127e-02 1.720e-02 2.620e-02 - 3.841e-02 5.204e-02 6.359e-02 - 4.205e-02 3.923e-02 3.364e-02 2.620e-02 1.938e-02 1.340e-02 - 9.366e-03 7.945e-03 9.366e-03 1.340e-02 1.938e-02 2.620e-02 - 3.364e-02 3.923e-02 4.205e-02 - 2.347e-02 2.272e-02 2.065e-02 1.720e-02 1.340e-02 9.746e-03 - 7.339e-03 7.339e-03 9.746e-03 1.340e-02 1.720e-02 2.065e-02 + 1.600e-02 7.338e-03 3.780e-03 2.613e-03 2.367e-03 2.613e-03 + 3.780e-03 7.338e-03 1.600e-02 + 6.777e-02 9.277e-02 9.277e-02 6.777e-02 4.206e-02 2.347e-02 + 1.265e-02 7.210e-03 4.702e-03 4.027e-03 4.702e-03 7.210e-03 + 1.265e-02 2.347e-02 4.206e-02 + 9.277e-02 1.075e-01 9.277e-02 6.360e-02 3.923e-02 2.272e-02 + 1.313e-02 7.953e-03 5.833e-03 5.833e-03 7.953e-03 1.313e-02 + 2.272e-02 3.923e-02 6.360e-02 + 9.277e-02 9.277e-02 7.322e-02 5.204e-02 3.364e-02 2.066e-02 + 1.266e-02 8.413e-03 7.113e-03 8.413e-03 1.266e-02 2.066e-02 + 3.364e-02 5.204e-02 7.322e-02 + 6.777e-02 6.360e-02 5.204e-02 3.841e-02 2.620e-02 1.720e-02 + 1.127e-02 8.317e-03 8.317e-03 1.127e-02 1.720e-02 2.620e-02 + 3.841e-02 5.204e-02 6.360e-02 + 4.206e-02 3.923e-02 3.364e-02 2.620e-02 1.938e-02 1.340e-02 + 9.366e-03 7.944e-03 9.366e-03 1.340e-02 1.938e-02 2.620e-02 + 3.364e-02 3.923e-02 4.206e-02 + 2.347e-02 2.272e-02 2.066e-02 1.720e-02 1.340e-02 9.747e-03 + 7.338e-03 7.338e-03 9.747e-03 1.340e-02 1.720e-02 2.066e-02 2.272e-02 2.347e-02 2.327e-02 - 1.264e-02 1.312e-02 1.265e-02 1.127e-02 9.366e-03 7.339e-03 - 6.592e-03 7.339e-03 9.366e-03 1.127e-02 1.265e-02 1.312e-02 - 1.264e-02 1.184e-02 1.184e-02 - 7.208e-03 7.950e-03 8.411e-03 8.316e-03 7.945e-03 7.339e-03 - 7.339e-03 7.945e-03 8.316e-03 8.411e-03 7.950e-03 7.208e-03 - 6.264e-03 5.856e-03 6.264e-03 - 4.700e-03 5.832e-03 7.112e-03 8.316e-03 9.366e-03 9.746e-03 - 9.366e-03 8.316e-03 7.112e-03 5.832e-03 4.700e-03 3.778e-03 - 3.212e-03 3.212e-03 3.778e-03 - 4.025e-03 5.832e-03 8.411e-03 1.127e-02 1.340e-02 1.340e-02 - 1.127e-02 8.411e-03 5.832e-03 4.025e-03 2.901e-03 2.246e-03 - 2.029e-03 2.246e-03 2.901e-03 - 4.700e-03 7.950e-03 1.265e-02 1.720e-02 1.938e-02 1.720e-02 - 1.265e-02 7.950e-03 4.700e-03 2.901e-03 2.016e-03 1.702e-03 - 1.702e-03 2.016e-03 2.901e-03 - 7.208e-03 1.312e-02 2.065e-02 2.620e-02 2.620e-02 2.065e-02 - 1.312e-02 7.208e-03 3.778e-03 2.246e-03 1.702e-03 1.578e-03 - 1.702e-03 2.246e-03 3.778e-03 - 1.264e-02 2.272e-02 3.364e-02 3.841e-02 3.364e-02 2.272e-02 - 1.264e-02 6.264e-03 3.212e-03 2.029e-03 1.702e-03 1.702e-03 - 2.029e-03 3.212e-03 6.264e-03 + 1.265e-02 1.313e-02 1.266e-02 1.127e-02 9.366e-03 7.338e-03 + 6.590e-03 7.338e-03 9.366e-03 1.127e-02 1.266e-02 1.313e-02 + 1.265e-02 1.184e-02 1.184e-02 + 7.210e-03 7.953e-03 8.413e-03 8.317e-03 7.944e-03 7.338e-03 + 7.338e-03 7.944e-03 8.317e-03 8.413e-03 7.953e-03 7.210e-03 + 6.267e-03 5.859e-03 6.267e-03 + 4.702e-03 5.833e-03 7.113e-03 8.317e-03 9.366e-03 9.747e-03 + 9.366e-03 8.317e-03 7.113e-03 5.833e-03 4.702e-03 3.780e-03 + 3.214e-03 3.214e-03 3.780e-03 + 4.027e-03 5.833e-03 8.413e-03 1.127e-02 1.340e-02 1.340e-02 + 1.127e-02 8.413e-03 5.833e-03 4.027e-03 2.903e-03 2.248e-03 + 2.031e-03 2.248e-03 2.903e-03 + 4.702e-03 7.953e-03 1.266e-02 1.720e-02 1.938e-02 1.720e-02 + 1.266e-02 7.953e-03 4.702e-03 2.903e-03 2.017e-03 1.703e-03 + 1.703e-03 2.017e-03 2.903e-03 + 7.210e-03 1.313e-02 2.066e-02 2.620e-02 2.620e-02 2.066e-02 + 1.313e-02 7.210e-03 3.780e-03 2.248e-03 1.703e-03 1.579e-03 + 1.703e-03 2.248e-03 3.780e-03 + 1.265e-02 2.272e-02 3.364e-02 3.841e-02 3.364e-02 2.272e-02 + 1.265e-02 6.267e-03 3.214e-03 2.031e-03 1.703e-03 1.703e-03 + 2.031e-03 3.214e-03 6.267e-03 2.347e-02 3.923e-02 5.204e-02 5.204e-02 3.923e-02 2.347e-02 - 1.184e-02 5.856e-03 3.212e-03 2.246e-03 2.016e-03 2.246e-03 - 3.212e-03 5.856e-03 1.184e-02 - 4.205e-02 6.359e-02 7.321e-02 6.359e-02 4.205e-02 2.327e-02 - 1.184e-02 6.264e-03 3.778e-03 2.901e-03 2.901e-03 3.778e-03 - 6.264e-03 1.184e-02 2.327e-02 - 5.185e-02 6.194e-02 5.185e-02 4.007e-02 3.244e-02 2.347e-02 - 1.523e-02 9.636e-03 6.907e-03 6.907e-03 9.636e-03 1.523e-02 - 2.347e-02 3.244e-02 4.007e-02 + 1.184e-02 5.859e-03 3.214e-03 2.248e-03 2.017e-03 2.248e-03 + 3.214e-03 5.859e-03 1.184e-02 + 4.206e-02 6.360e-02 7.322e-02 6.360e-02 4.206e-02 2.327e-02 + 1.184e-02 6.267e-03 3.780e-03 2.903e-03 2.903e-03 3.780e-03 + 6.267e-03 1.184e-02 2.327e-02 + 5.185e-02 6.194e-02 5.185e-02 4.007e-02 3.245e-02 2.347e-02 + 1.523e-02 9.638e-03 6.909e-03 6.909e-03 9.638e-03 1.523e-02 + 2.347e-02 3.245e-02 4.007e-02 6.194e-02 6.194e-02 5.272e-02 4.810e-02 3.923e-02 2.757e-02 1.795e-02 1.184e-02 9.768e-03 1.184e-02 1.795e-02 2.757e-02 3.923e-02 4.810e-02 5.272e-02 @@ -601,83 +601,83 @@ Ionic_Step 1 Cubefile created from ABACUS. Inner loop is z, followed by y and x 1.892e-02 1.370e-02 1.370e-02 1.892e-02 2.859e-02 4.066e-02 5.204e-02 5.525e-02 5.272e-02 4.007e-02 4.810e-02 5.204e-02 4.690e-02 3.637e-02 2.583e-02 - 1.791e-02 1.479e-02 1.791e-02 2.583e-02 3.637e-02 4.690e-02 + 1.790e-02 1.479e-02 1.790e-02 2.583e-02 3.637e-02 4.690e-02 5.204e-02 4.810e-02 4.007e-02 - 3.244e-02 3.923e-02 4.066e-02 3.637e-02 2.919e-02 2.075e-02 + 3.245e-02 3.923e-02 4.066e-02 3.637e-02 2.919e-02 2.075e-02 1.520e-02 1.520e-02 2.075e-02 2.919e-02 3.637e-02 4.066e-02 - 3.923e-02 3.244e-02 2.820e-02 + 3.923e-02 3.245e-02 2.820e-02 2.347e-02 2.757e-02 2.859e-02 2.583e-02 2.075e-02 1.506e-02 - 1.272e-02 1.506e-02 2.075e-02 2.583e-02 2.859e-02 2.757e-02 + 1.271e-02 1.506e-02 2.075e-02 2.583e-02 2.859e-02 2.757e-02 2.347e-02 1.916e-02 1.916e-02 - 1.523e-02 1.795e-02 1.892e-02 1.791e-02 1.520e-02 1.272e-02 - 1.272e-02 1.520e-02 1.791e-02 1.892e-02 1.795e-02 1.523e-02 + 1.523e-02 1.795e-02 1.892e-02 1.790e-02 1.520e-02 1.271e-02 + 1.271e-02 1.520e-02 1.790e-02 1.892e-02 1.795e-02 1.523e-02 1.184e-02 1.044e-02 1.184e-02 - 9.636e-03 1.184e-02 1.370e-02 1.479e-02 1.520e-02 1.506e-02 - 1.520e-02 1.479e-02 1.370e-02 1.184e-02 9.636e-03 7.335e-03 - 5.856e-03 5.856e-03 7.335e-03 - 6.907e-03 9.768e-03 1.370e-02 1.791e-02 2.075e-02 2.075e-02 - 1.791e-02 1.370e-02 9.768e-03 6.907e-03 4.859e-03 3.600e-03 - 3.212e-03 3.600e-03 4.859e-03 - 6.907e-03 1.184e-02 1.892e-02 2.583e-02 2.919e-02 2.583e-02 - 1.892e-02 1.184e-02 6.907e-03 4.050e-03 2.741e-03 2.246e-03 - 2.246e-03 2.741e-03 4.050e-03 - 9.636e-03 1.795e-02 2.859e-02 3.637e-02 3.637e-02 2.859e-02 - 1.795e-02 9.636e-03 4.859e-03 2.741e-03 2.016e-03 1.853e-03 - 2.016e-03 2.741e-03 4.859e-03 + 9.638e-03 1.184e-02 1.370e-02 1.479e-02 1.520e-02 1.506e-02 + 1.520e-02 1.479e-02 1.370e-02 1.184e-02 9.638e-03 7.338e-03 + 5.859e-03 5.859e-03 7.338e-03 + 6.909e-03 9.768e-03 1.370e-02 1.790e-02 2.075e-02 2.075e-02 + 1.790e-02 1.370e-02 9.768e-03 6.909e-03 4.861e-03 3.603e-03 + 3.214e-03 3.603e-03 4.861e-03 + 6.909e-03 1.184e-02 1.892e-02 2.583e-02 2.919e-02 2.583e-02 + 1.892e-02 1.184e-02 6.909e-03 4.051e-03 2.743e-03 2.248e-03 + 2.248e-03 2.743e-03 4.051e-03 + 9.638e-03 1.795e-02 2.859e-02 3.637e-02 3.637e-02 2.859e-02 + 1.795e-02 9.638e-03 4.861e-03 2.743e-03 2.017e-03 1.855e-03 + 2.017e-03 2.743e-03 4.861e-03 1.523e-02 2.757e-02 4.066e-02 4.690e-02 4.066e-02 2.757e-02 - 1.523e-02 7.335e-03 3.600e-03 2.246e-03 1.853e-03 1.853e-03 - 2.246e-03 3.600e-03 7.335e-03 + 1.523e-02 7.338e-03 3.603e-03 2.248e-03 1.855e-03 1.855e-03 + 2.248e-03 3.603e-03 7.338e-03 2.347e-02 3.923e-02 5.204e-02 5.204e-02 3.923e-02 2.347e-02 - 1.184e-02 5.856e-03 3.212e-03 2.246e-03 2.016e-03 2.246e-03 - 3.212e-03 5.856e-03 1.184e-02 - 3.244e-02 4.810e-02 5.525e-02 4.810e-02 3.244e-02 1.916e-02 - 1.044e-02 5.856e-03 3.600e-03 2.741e-03 2.741e-03 3.600e-03 - 5.856e-03 1.044e-02 1.916e-02 + 1.184e-02 5.859e-03 3.214e-03 2.248e-03 2.017e-03 2.248e-03 + 3.214e-03 5.859e-03 1.184e-02 + 3.245e-02 4.810e-02 5.525e-02 4.810e-02 3.245e-02 1.916e-02 + 1.044e-02 5.859e-03 3.603e-03 2.743e-03 2.743e-03 3.603e-03 + 5.859e-03 1.044e-02 1.916e-02 4.007e-02 5.272e-02 5.272e-02 4.007e-02 2.820e-02 1.916e-02 - 1.184e-02 7.335e-03 4.859e-03 4.050e-03 4.859e-03 7.335e-03 + 1.184e-02 7.338e-03 4.861e-03 4.051e-03 4.861e-03 7.338e-03 1.184e-02 1.916e-02 2.820e-02 - 1.824e-02 1.824e-02 2.392e-02 4.007e-02 4.205e-02 3.255e-02 - 2.182e-02 1.445e-02 1.179e-02 1.445e-02 2.182e-02 3.255e-02 - 4.205e-02 4.007e-02 2.392e-02 - 1.824e-02 2.857e-02 5.272e-02 6.359e-02 5.455e-02 3.967e-02 + 1.824e-02 1.824e-02 2.392e-02 4.007e-02 4.206e-02 3.255e-02 + 2.183e-02 1.445e-02 1.179e-02 1.445e-02 2.183e-02 3.255e-02 + 4.206e-02 4.007e-02 2.392e-02 + 1.824e-02 2.857e-02 5.272e-02 6.360e-02 5.455e-02 3.967e-02 2.653e-02 1.895e-02 1.895e-02 2.653e-02 3.967e-02 5.455e-02 - 6.359e-02 5.272e-02 2.857e-02 - 2.392e-02 5.272e-02 7.321e-02 7.273e-02 5.829e-02 4.213e-02 - 2.898e-02 2.401e-02 2.898e-02 4.213e-02 5.829e-02 7.273e-02 - 7.321e-02 5.272e-02 2.392e-02 - 4.007e-02 6.359e-02 7.273e-02 6.717e-02 5.348e-02 3.817e-02 + 6.360e-02 5.272e-02 2.857e-02 + 2.392e-02 5.272e-02 7.322e-02 7.273e-02 5.829e-02 4.213e-02 + 2.897e-02 2.401e-02 2.897e-02 4.213e-02 5.829e-02 7.273e-02 + 7.322e-02 5.272e-02 2.392e-02 + 4.007e-02 6.360e-02 7.273e-02 6.717e-02 5.348e-02 3.817e-02 2.780e-02 2.780e-02 3.817e-02 5.348e-02 6.717e-02 7.273e-02 - 6.359e-02 4.007e-02 2.575e-02 - 4.205e-02 5.455e-02 5.829e-02 5.348e-02 4.254e-02 2.987e-02 - 2.402e-02 2.987e-02 4.254e-02 5.348e-02 5.829e-02 5.455e-02 - 4.205e-02 2.820e-02 2.820e-02 - 3.255e-02 3.967e-02 4.213e-02 3.817e-02 2.987e-02 2.189e-02 - 2.189e-02 2.987e-02 3.817e-02 4.213e-02 3.967e-02 3.255e-02 + 6.360e-02 4.007e-02 2.576e-02 + 4.206e-02 5.455e-02 5.829e-02 5.348e-02 4.253e-02 2.987e-02 + 2.401e-02 2.987e-02 4.253e-02 5.348e-02 5.829e-02 5.455e-02 + 4.206e-02 2.820e-02 2.820e-02 + 3.255e-02 3.967e-02 4.213e-02 3.817e-02 2.987e-02 2.188e-02 + 2.188e-02 2.987e-02 3.817e-02 4.213e-02 3.967e-02 3.255e-02 2.327e-02 1.916e-02 2.327e-02 - 2.182e-02 2.653e-02 2.898e-02 2.780e-02 2.402e-02 2.189e-02 - 2.402e-02 2.780e-02 2.898e-02 2.653e-02 2.182e-02 1.600e-02 + 2.183e-02 2.653e-02 2.897e-02 2.780e-02 2.401e-02 2.188e-02 + 2.401e-02 2.780e-02 2.897e-02 2.653e-02 2.183e-02 1.600e-02 1.184e-02 1.184e-02 1.600e-02 1.445e-02 1.895e-02 2.401e-02 2.780e-02 2.987e-02 2.987e-02 - 2.780e-02 2.401e-02 1.895e-02 1.445e-02 1.022e-02 7.335e-03 - 6.264e-03 7.335e-03 1.022e-02 - 1.179e-02 1.895e-02 2.898e-02 3.817e-02 4.254e-02 3.817e-02 - 2.898e-02 1.895e-02 1.179e-02 7.107e-03 4.859e-03 3.778e-03 - 3.778e-03 4.859e-03 7.107e-03 + 2.780e-02 2.401e-02 1.895e-02 1.445e-02 1.022e-02 7.338e-03 + 6.267e-03 7.338e-03 1.022e-02 + 1.179e-02 1.895e-02 2.897e-02 3.817e-02 4.253e-02 3.817e-02 + 2.897e-02 1.895e-02 1.179e-02 7.108e-03 4.861e-03 3.780e-03 + 3.780e-03 4.861e-03 7.108e-03 1.445e-02 2.653e-02 4.213e-02 5.348e-02 5.348e-02 4.213e-02 - 2.653e-02 1.445e-02 7.107e-03 4.050e-03 2.901e-03 2.611e-03 - 2.901e-03 4.050e-03 7.107e-03 - 2.182e-02 3.967e-02 5.829e-02 6.717e-02 5.829e-02 3.967e-02 - 2.182e-02 1.022e-02 4.859e-03 2.901e-03 2.365e-03 2.365e-03 - 2.901e-03 4.859e-03 1.022e-02 + 2.653e-02 1.445e-02 7.108e-03 4.051e-03 2.903e-03 2.613e-03 + 2.903e-03 4.051e-03 7.108e-03 + 2.183e-02 3.967e-02 5.829e-02 6.717e-02 5.829e-02 3.967e-02 + 2.183e-02 1.022e-02 4.861e-03 2.903e-03 2.367e-03 2.367e-03 + 2.903e-03 4.861e-03 1.022e-02 3.255e-02 5.455e-02 7.273e-02 7.273e-02 5.455e-02 3.255e-02 - 1.600e-02 7.335e-03 3.778e-03 2.611e-03 2.365e-03 2.611e-03 - 3.778e-03 7.335e-03 1.600e-02 - 4.205e-02 6.359e-02 7.321e-02 6.359e-02 4.205e-02 2.327e-02 - 1.184e-02 6.264e-03 3.778e-03 2.901e-03 2.901e-03 3.778e-03 - 6.264e-03 1.184e-02 2.327e-02 + 1.600e-02 7.338e-03 3.780e-03 2.613e-03 2.367e-03 2.613e-03 + 3.780e-03 7.338e-03 1.600e-02 + 4.206e-02 6.360e-02 7.322e-02 6.360e-02 4.206e-02 2.327e-02 + 1.184e-02 6.267e-03 3.780e-03 2.903e-03 2.903e-03 3.780e-03 + 6.267e-03 1.184e-02 2.327e-02 4.007e-02 5.272e-02 5.272e-02 4.007e-02 2.820e-02 1.916e-02 - 1.184e-02 7.335e-03 4.859e-03 4.050e-03 4.859e-03 7.335e-03 + 1.184e-02 7.338e-03 4.861e-03 4.051e-03 4.861e-03 7.338e-03 1.184e-02 1.916e-02 2.820e-02 - 2.392e-02 2.857e-02 2.392e-02 2.575e-02 2.820e-02 2.327e-02 - 1.600e-02 1.022e-02 7.107e-03 7.107e-03 1.022e-02 1.600e-02 - 2.327e-02 2.820e-02 2.575e-02 + 2.392e-02 2.857e-02 2.392e-02 2.576e-02 2.820e-02 2.327e-02 + 1.600e-02 1.022e-02 7.108e-03 7.108e-03 1.022e-02 1.600e-02 + 2.327e-02 2.820e-02 2.576e-02 diff --git a/tests/03_NAO_multik/scf_out_chg_tau/tau.cube.ref b/tests/03_NAO_multik/scf_out_chg_tau/tau.cube.ref index 45966ff097b..41ca199fac5 100644 --- a/tests/03_NAO_multik/scf_out_chg_tau/tau.cube.ref +++ b/tests/03_NAO_multik/scf_out_chg_tau/tau.cube.ref @@ -6,678 +6,678 @@ Ionic_Step 1 Cubefile created from ABACUS. Inner loop is z, followed by y and x 15 0.340000 0.340000 0.000000 14 4.000000 0.000000 0.000000 0.000000 14 4.000000 2.550000 2.550000 2.550000 - 1.80281085967e-01 1.82422828268e-01 1.16467735388e-01 6.52691932418e-02 4.77193978469e-02 3.72920735613e-02 - 3.18085146238e-02 2.99172055199e-02 2.99172055199e-02 3.18085146238e-02 3.72920735613e-02 4.77193978469e-02 - 6.52691932418e-02 1.16467735388e-01 1.82422828268e-01 - 1.82422828268e-01 1.02568469446e-01 5.03836760403e-02 3.50294268557e-02 3.42073290623e-02 3.78194924587e-02 - 4.02222959331e-02 4.34116802001e-02 4.02222959331e-02 3.78194924587e-02 3.42073290623e-02 3.50294268557e-02 - 5.03836760403e-02 1.02568469446e-01 1.82422828268e-01 - 1.16467735388e-01 5.03836760403e-02 2.76155885440e-02 2.56381222091e-02 3.16578904846e-02 4.30811551462e-02 - 5.01938086050e-02 5.01938086050e-02 4.30811551462e-02 3.16578904846e-02 2.56381222091e-02 2.76155885440e-02 - 5.03836760403e-02 1.16467735388e-01 1.75978291339e-01 - 6.52691932418e-02 3.50294268557e-02 2.56381222091e-02 2.77650059484e-02 3.92607806531e-02 5.15953878901e-02 - 6.40797263891e-02 5.15953878901e-02 3.92607806531e-02 2.77650059484e-02 2.56381222091e-02 3.50294268557e-02 - 6.52691932418e-02 9.95387143197e-02 9.95387143197e-02 - 4.77193978469e-02 3.42073290623e-02 3.16578904846e-02 3.92607806531e-02 5.77610358000e-02 7.23281111668e-02 - 7.23281111668e-02 5.77610358000e-02 3.92607806531e-02 3.16578904846e-02 3.42073290623e-02 4.77193978469e-02 - 5.61335799375e-02 6.46788103129e-02 5.61335799375e-02 - 3.72920735613e-02 3.78194924587e-02 4.30811551462e-02 5.15953878901e-02 7.23281111668e-02 8.49232065105e-02 - 7.23281111668e-02 5.15953878901e-02 4.30811551462e-02 3.78194924587e-02 3.72920735613e-02 3.85875039457e-02 - 3.97809938382e-02 3.97809938382e-02 3.85875039457e-02 - 3.18085146238e-02 4.02222959331e-02 5.01938086050e-02 6.40797263891e-02 7.23281111668e-02 7.23281111668e-02 - 6.40797263891e-02 5.01938086050e-02 4.02222959331e-02 3.18085146238e-02 2.77237360128e-02 2.42768838976e-02 - 2.38593064853e-02 2.42768838976e-02 2.77237360128e-02 - 2.99172055199e-02 4.34116802001e-02 5.01938086050e-02 5.15953878901e-02 5.77610358000e-02 5.15953878901e-02 - 5.01938086050e-02 4.34116802001e-02 2.99172055199e-02 2.21844351566e-02 1.56599181080e-02 1.25503014842e-02 - 1.25503014842e-02 1.56599181080e-02 2.21844351566e-02 - 2.99172055199e-02 4.02222959331e-02 4.30811551462e-02 3.92607806531e-02 3.92607806531e-02 4.30811551462e-02 - 4.02222959331e-02 2.99172055199e-02 1.95071574201e-02 1.13646063453e-02 6.70218401668e-03 4.59069441377e-03 - 6.70218401668e-03 1.13646063453e-02 1.95071574201e-02 - 3.18085146238e-02 3.78194924587e-02 3.16578904846e-02 2.77650059484e-02 3.16578904846e-02 3.78194924587e-02 - 3.18085146238e-02 2.21844351566e-02 1.13646063453e-02 5.11954914432e-03 1.60658688370e-03 1.60658688370e-03 - 5.11954914432e-03 1.13646063453e-02 2.21844351566e-02 - 3.72920735613e-02 3.42073290623e-02 2.56381222091e-02 2.56381222091e-02 3.42073290623e-02 3.72920735613e-02 - 2.77237360128e-02 1.56599181080e-02 6.70218401668e-03 1.60658688370e-03 2.99179150647e-04 1.60658688370e-03 - 6.70218401668e-03 1.56599181080e-02 2.77237360128e-02 - 4.77193978469e-02 3.50294268557e-02 2.76155885440e-02 3.50294268557e-02 4.77193978469e-02 3.85875039457e-02 - 2.42768838976e-02 1.25503014842e-02 4.59069441377e-03 1.60658688370e-03 1.60658688370e-03 4.59069441377e-03 - 1.25503014842e-02 2.42768838976e-02 3.85875039457e-02 - 6.52691932418e-02 5.03836760403e-02 5.03836760403e-02 6.52691932418e-02 5.61335799375e-02 3.97809938382e-02 - 2.38593064853e-02 1.25503014842e-02 6.70218401668e-03 5.11954914432e-03 6.70218401668e-03 1.25503014842e-02 - 2.38593064853e-02 3.97809938382e-02 5.61335799375e-02 - 1.16467735388e-01 1.02568469446e-01 1.16467735388e-01 9.95387143197e-02 6.46788103129e-02 3.97809938382e-02 - 2.42768838976e-02 1.56599181080e-02 1.13646063453e-02 1.13646063453e-02 1.56599181080e-02 2.42768838976e-02 - 3.97809938382e-02 6.46788103129e-02 9.95387143197e-02 - 1.82422828268e-01 1.82422828268e-01 1.75978291339e-01 9.95387143197e-02 5.61335799375e-02 3.85875039457e-02 - 2.77237360128e-02 2.21844351566e-02 1.95071574201e-02 2.21844351566e-02 2.77237360128e-02 3.85875039457e-02 - 5.61335799375e-02 9.95387143197e-02 1.75978291339e-01 - 1.82422828268e-01 1.02568469446e-01 5.03836760403e-02 3.50294268557e-02 3.42073290623e-02 3.78194924587e-02 - 4.02222959331e-02 4.34116802001e-02 4.02222959331e-02 3.78194924587e-02 3.42073290623e-02 3.50294268557e-02 - 5.03836760403e-02 1.02568469446e-01 1.82422828268e-01 - 1.02568469446e-01 3.75098528213e-02 1.55091198600e-02 1.76415461442e-02 2.80289195955e-02 4.32649438817e-02 - 5.27225154807e-02 5.27225154807e-02 4.32649438817e-02 2.80289195955e-02 1.76415461442e-02 1.55091198600e-02 - 3.75098528213e-02 1.02568469446e-01 1.83280188132e-01 - 5.03836760403e-02 1.55091198600e-02 7.66046977902e-03 1.45740503756e-02 3.18433510883e-02 5.41698533623e-02 - 7.22755322173e-02 5.41698533623e-02 3.18433510883e-02 1.45740503756e-02 7.66046977902e-03 1.55091198600e-02 - 5.03836760403e-02 1.13539754126e-01 1.13539754126e-01 - 3.50294268557e-02 1.76415461442e-02 1.45740503756e-02 2.47958446021e-02 5.21992902722e-02 8.90053073193e-02 - 8.90053073193e-02 5.21992902722e-02 2.47958446021e-02 1.45740503756e-02 1.76415461442e-02 3.50294268557e-02 - 5.92235143217e-02 8.15533129102e-02 5.92235143217e-02 - 3.42073290623e-02 2.80289195955e-02 3.18433510883e-02 5.21992902722e-02 9.62989068912e-02 1.28883911425e-01 - 9.62989068912e-02 5.21992902722e-02 3.18433510883e-02 2.80289195955e-02 3.42073290623e-02 4.46676835001e-02 - 5.23936115453e-02 5.23936115453e-02 4.46676835001e-02 - 3.78194924587e-02 4.32649438817e-02 5.41698533623e-02 8.90053073193e-02 1.28883911425e-01 1.28883911425e-01 - 8.90053073193e-02 5.41698533623e-02 4.32649438817e-02 3.78194924587e-02 3.60022233749e-02 3.49506556165e-02 - 3.70994631101e-02 3.49506556165e-02 3.60022233749e-02 - 4.02222959331e-02 5.27225154807e-02 7.22755322173e-02 8.90053073193e-02 9.62989068912e-02 8.90053073193e-02 - 7.22755322173e-02 5.27225154807e-02 4.02222959331e-02 3.08604739677e-02 2.43879073207e-02 2.13888507874e-02 - 2.13888507874e-02 2.43879073207e-02 3.08604739677e-02 - 4.34116802001e-02 5.27225154807e-02 5.41698533623e-02 5.21992902722e-02 5.21992902722e-02 5.41698533623e-02 - 5.27225154807e-02 4.34116802001e-02 2.79666745937e-02 1.78991060313e-02 1.23746382387e-02 9.66467351917e-03 - 1.23746382387e-02 1.78991060313e-02 2.79666745937e-02 - 4.02222959331e-02 4.32649438817e-02 3.18433510883e-02 2.47958446021e-02 3.18433510883e-02 4.32649438817e-02 - 4.02222959331e-02 2.79666745937e-02 1.52963446205e-02 7.92513025055e-03 3.86875558844e-03 3.86875558844e-03 - 7.92513025055e-03 1.52963446205e-02 2.79666745937e-02 - 3.78194924587e-02 2.80289195955e-02 1.45740503756e-02 1.45740503756e-02 2.80289195955e-02 3.78194924587e-02 - 3.08604739677e-02 1.78991060313e-02 7.92513025055e-03 2.15869962360e-03 6.20007891972e-04 2.15869962360e-03 - 7.92513025055e-03 1.78991060313e-02 3.08604739677e-02 - 3.42073290623e-02 1.76415461442e-02 7.66046977902e-03 1.76415461442e-02 3.42073290623e-02 3.60022233749e-02 - 2.43879073207e-02 1.23746382387e-02 3.86875558844e-03 6.20007891972e-04 6.20007891972e-04 3.86875558844e-03 - 1.23746382387e-02 2.43879073207e-02 3.60022233749e-02 - 3.50294268557e-02 1.55091198600e-02 1.55091198600e-02 3.50294268557e-02 4.46676835001e-02 3.49506556165e-02 - 2.13888507874e-02 9.66467351917e-03 3.86875558844e-03 2.15869962360e-03 3.86875558844e-03 9.66467351917e-03 - 2.13888507874e-02 3.49506556165e-02 4.46676835001e-02 - 5.03836760403e-02 3.75098528213e-02 5.03836760403e-02 5.92235143217e-02 5.23936115453e-02 3.70994631101e-02 - 2.13888507874e-02 1.23746382387e-02 7.92513025055e-03 7.92513025055e-03 1.23746382387e-02 2.13888507874e-02 - 3.70994631101e-02 5.23936115453e-02 5.92235143217e-02 - 1.02568469446e-01 1.02568469446e-01 1.13539754126e-01 8.15533129102e-02 5.23936115453e-02 3.49506556165e-02 - 2.43879073207e-02 1.78991060313e-02 1.52963446205e-02 1.78991060313e-02 2.43879073207e-02 3.49506556165e-02 - 5.23936115453e-02 8.15533129102e-02 1.13539754126e-01 - 1.82422828268e-01 1.83280188132e-01 1.13539754126e-01 5.92235143217e-02 4.46676835001e-02 3.60022233749e-02 - 3.08604739677e-02 2.79666745937e-02 2.79666745937e-02 3.08604739677e-02 3.60022233749e-02 4.46676835001e-02 - 5.92235143217e-02 1.13539754126e-01 1.83280188132e-01 - 1.16467735388e-01 5.03836760403e-02 2.76155885440e-02 2.56381222091e-02 3.16578904846e-02 4.30811551462e-02 - 5.01938086050e-02 5.01938086050e-02 4.30811551462e-02 3.16578904846e-02 2.56381222091e-02 2.76155885440e-02 - 5.03836760403e-02 1.16467735388e-01 1.75978291339e-01 - 5.03836760403e-02 1.55091198600e-02 7.66046977902e-03 1.45740503756e-02 3.18433510883e-02 5.41698533623e-02 - 7.22755322173e-02 5.41698533623e-02 3.18433510883e-02 1.45740503756e-02 7.66046977902e-03 1.55091198600e-02 - 5.03836760403e-02 1.13539754126e-01 1.13539754126e-01 - 2.76155885440e-02 7.66046977902e-03 5.88407505951e-04 1.22283197111e-02 4.60292144819e-02 9.32262743675e-02 - 9.32262743675e-02 4.60292144819e-02 1.22283197111e-02 5.88407505951e-04 7.66046977902e-03 2.76155885440e-02 - 6.09424067177e-02 8.53804463649e-02 6.09424067177e-02 - 2.56381222091e-02 1.45740503756e-02 1.22283197111e-02 3.36135559891e-02 9.58701888529e-02 1.55893621190e-01 - 9.58701888529e-02 3.36135559891e-02 1.22283197111e-02 1.45740503756e-02 2.56381222091e-02 4.40351969320e-02 - 5.84769754663e-02 5.84769754663e-02 4.40351969320e-02 - 3.16578904846e-02 3.18433510883e-02 4.60292144819e-02 9.58701888529e-02 1.78937466321e-01 1.78937466321e-01 - 9.58701888529e-02 4.60292144819e-02 3.18433510883e-02 3.16578904846e-02 3.72212996136e-02 4.27231306029e-02 - 4.57138189196e-02 4.27231306029e-02 3.72212996136e-02 - 4.30811551462e-02 5.41698533623e-02 9.32262743675e-02 1.55893621190e-01 1.78937466321e-01 1.55893621190e-01 - 9.32262743675e-02 5.41698533623e-02 4.30811551462e-02 3.54179538479e-02 3.30817407316e-02 3.00019977593e-02 - 3.00019977593e-02 3.30817407316e-02 3.54179538479e-02 - 5.01938086050e-02 7.22755322173e-02 9.32262743675e-02 9.58701888529e-02 9.58701888529e-02 9.32262743675e-02 - 7.22755322173e-02 5.01938086050e-02 3.49314010620e-02 2.55149658819e-02 1.86401066199e-02 1.63183576357e-02 - 1.86401066199e-02 2.55149658819e-02 3.49314010620e-02 - 5.01938086050e-02 5.41698533623e-02 4.60292144819e-02 3.36135559891e-02 4.60292144819e-02 5.41698533623e-02 - 5.01938086050e-02 3.55073021064e-02 2.15005775831e-02 1.24898667953e-02 8.27412039851e-03 8.27412039851e-03 - 1.24898667953e-02 2.15005775831e-02 3.55073021064e-02 - 4.30811551462e-02 3.18433510883e-02 1.22283197111e-02 1.22283197111e-02 3.18433510883e-02 4.30811551462e-02 - 3.49314010620e-02 2.15005775831e-02 9.79121310155e-03 4.02268913133e-03 2.15067339004e-03 4.02268913133e-03 - 9.79121310155e-03 2.15005775831e-02 3.49314010620e-02 - 3.16578904846e-02 1.45740503756e-02 5.88407505951e-04 1.45740503756e-02 3.16578904846e-02 3.54179538479e-02 - 2.55149658819e-02 1.24898667953e-02 4.02268913133e-03 5.43091896583e-04 5.43091896583e-04 4.02268913133e-03 - 1.24898667953e-02 2.55149658819e-02 3.54179538479e-02 - 2.56381222091e-02 7.66046977902e-03 7.66046977902e-03 2.56381222091e-02 3.72212996136e-02 3.30817407316e-02 - 1.86401066199e-02 8.27412039851e-03 2.15067339004e-03 5.43091896583e-04 2.15067339004e-03 8.27412039851e-03 - 1.86401066199e-02 3.30817407316e-02 3.72212996136e-02 - 2.76155885440e-02 1.55091198600e-02 2.76155885440e-02 4.40351969320e-02 4.27231306029e-02 3.00019977593e-02 - 1.63183576357e-02 8.27412039851e-03 4.02268913133e-03 4.02268913133e-03 8.27412039851e-03 1.63183576357e-02 - 3.00019977593e-02 4.27231306029e-02 4.40351969320e-02 - 5.03836760403e-02 5.03836760403e-02 6.09424067177e-02 5.84769754663e-02 4.57138189196e-02 3.00019977593e-02 - 1.86401066199e-02 1.24898667953e-02 9.79121310155e-03 1.24898667953e-02 1.86401066199e-02 3.00019977593e-02 - 4.57138189196e-02 5.84769754663e-02 6.09424067177e-02 - 1.16467735388e-01 1.13539754126e-01 8.53804463649e-02 5.84769754663e-02 4.27231306029e-02 3.30817407316e-02 - 2.55149658819e-02 2.15005775831e-02 2.15005775831e-02 2.55149658819e-02 3.30817407316e-02 4.27231306029e-02 - 5.84769754663e-02 8.53804463649e-02 1.13539754126e-01 - 1.75978291339e-01 1.13539754126e-01 6.09424067177e-02 4.40351969320e-02 3.72212996136e-02 3.54179538479e-02 - 3.49314010620e-02 3.55073021064e-02 3.49314010620e-02 3.54179538479e-02 3.72212996136e-02 4.40351969320e-02 - 6.09424067177e-02 1.13539754126e-01 1.75978291339e-01 - 6.52691932418e-02 3.50294268557e-02 2.56381222091e-02 2.77650059484e-02 3.92607806531e-02 5.15953878901e-02 - 6.40797263891e-02 5.15953878901e-02 3.92607806531e-02 2.77650059484e-02 2.56381222091e-02 3.50294268557e-02 - 6.52691932418e-02 9.95387143197e-02 9.95387143197e-02 - 3.50294268557e-02 1.76415461442e-02 1.45740503756e-02 2.47958446021e-02 5.21992902722e-02 8.90053073193e-02 - 8.90053073193e-02 5.21992902722e-02 2.47958446021e-02 1.45740503756e-02 1.76415461442e-02 3.50294268557e-02 - 5.92235143217e-02 8.15533129102e-02 5.92235143217e-02 - 2.56381222091e-02 1.45740503756e-02 1.22283197111e-02 3.36135559891e-02 9.58701888529e-02 1.55893621190e-01 - 9.58701888529e-02 3.36135559891e-02 1.22283197111e-02 1.45740503756e-02 2.56381222091e-02 4.40351969320e-02 - 5.84769754663e-02 5.84769754663e-02 4.40351969320e-02 - 2.77650059484e-02 2.47958446021e-02 3.36135559891e-02 6.76894902970e-02 1.65125441372e-01 1.65125441372e-01 - 6.76894902970e-02 3.36135559891e-02 2.47958446021e-02 2.77650059484e-02 3.74646524133e-02 4.48502710858e-02 - 5.02634504938e-02 4.48502710858e-02 3.74646524133e-02 - 3.92607806531e-02 5.21992902722e-02 9.58701888529e-02 1.65125441372e-01 1.94666393607e-01 1.65125441372e-01 - 9.58701888529e-02 5.21992902722e-02 3.92607806531e-02 3.70588268986e-02 3.66436362392e-02 3.62873182074e-02 - 3.62873182074e-02 3.66436362392e-02 3.70588268986e-02 - 5.15953878901e-02 8.90053073193e-02 1.55893621190e-01 1.65125441372e-01 1.65125441372e-01 1.55893621190e-01 - 8.90053073193e-02 5.15953878901e-02 3.88199146511e-02 3.00331874093e-02 2.47929678075e-02 2.24956267167e-02 - 2.47929678075e-02 3.00331874093e-02 3.88199146511e-02 - 6.40797263891e-02 8.90053073193e-02 9.58701888529e-02 6.76894902970e-02 9.58701888529e-02 8.90053073193e-02 - 6.40797263891e-02 4.06623304521e-02 2.56772132883e-02 1.73142648442e-02 1.28060436295e-02 1.28060436295e-02 - 1.73142648442e-02 2.56772132883e-02 4.06623304521e-02 - 5.15953878901e-02 5.21992902722e-02 3.36135559891e-02 3.36135559891e-02 5.21992902722e-02 5.15953878901e-02 - 4.06623304521e-02 2.60569281693e-02 1.40094656991e-02 7.35387417276e-03 5.53145518161e-03 7.35387417276e-03 - 1.40094656991e-02 2.60569281693e-02 4.06623304521e-02 - 3.92607806531e-02 2.47958446021e-02 1.22283197111e-02 2.47958446021e-02 3.92607806531e-02 3.88199146511e-02 - 2.56772132883e-02 1.40094656991e-02 5.18295256070e-03 1.64102393395e-03 1.64102393395e-03 5.18295256070e-03 - 1.40094656991e-02 2.56772132883e-02 3.88199146511e-02 - 2.77650059484e-02 1.45740503756e-02 1.45740503756e-02 2.77650059484e-02 3.70588268986e-02 3.00331874093e-02 - 1.73142648442e-02 7.35387417276e-03 1.64102393395e-03 1.47520974601e-04 1.64102393395e-03 7.35387417276e-03 - 1.73142648442e-02 3.00331874093e-02 3.70588268986e-02 - 2.56381222091e-02 1.76415461442e-02 2.56381222091e-02 3.74646524133e-02 3.66436362392e-02 2.47929678075e-02 - 1.28060436295e-02 5.53145518161e-03 1.64102393395e-03 1.64102393395e-03 5.53145518161e-03 1.28060436295e-02 - 2.47929678075e-02 3.66436362392e-02 3.74646524133e-02 - 3.50294268557e-02 3.50294268557e-02 4.40351969320e-02 4.48502710858e-02 3.62873182074e-02 2.24956267167e-02 - 1.28060436295e-02 7.35387417276e-03 5.18295256070e-03 7.35387417276e-03 1.28060436295e-02 2.24956267167e-02 - 3.62873182074e-02 4.48502710858e-02 4.40351969320e-02 - 6.52691932418e-02 5.92235143217e-02 5.84769754663e-02 5.02634504938e-02 3.62873182074e-02 2.47929678075e-02 - 1.73142648442e-02 1.40094656991e-02 1.40094656991e-02 1.73142648442e-02 2.47929678075e-02 3.62873182074e-02 - 5.02634504938e-02 5.84769754663e-02 5.92235143217e-02 - 9.95387143197e-02 8.15533129102e-02 5.84769754663e-02 4.48502710858e-02 3.66436362392e-02 3.00331874093e-02 - 2.56772132883e-02 2.60569281693e-02 2.56772132883e-02 3.00331874093e-02 3.66436362392e-02 4.48502710858e-02 - 5.84769754663e-02 8.15533129102e-02 9.95387143197e-02 - 9.95387143197e-02 5.92235143217e-02 4.40351969320e-02 3.74646524133e-02 3.70588268986e-02 3.88199146511e-02 - 4.06623304521e-02 4.06623304521e-02 3.88199146511e-02 3.70588268986e-02 3.74646524133e-02 4.40351969320e-02 - 5.92235143217e-02 9.95387143197e-02 1.21879341630e-01 - 4.77193978469e-02 3.42073290623e-02 3.16578904846e-02 3.92607806531e-02 5.77610358000e-02 7.23281111668e-02 - 7.23281111668e-02 5.77610358000e-02 3.92607806531e-02 3.16578904846e-02 3.42073290623e-02 4.77193978469e-02 - 5.61335799375e-02 6.46788103129e-02 5.61335799375e-02 - 3.42073290623e-02 2.80289195955e-02 3.18433510883e-02 5.21992902722e-02 9.62989068912e-02 1.28883911425e-01 - 9.62989068912e-02 5.21992902722e-02 3.18433510883e-02 2.80289195955e-02 3.42073290623e-02 4.46676835001e-02 - 5.23936115453e-02 5.23936115453e-02 4.46676835001e-02 - 3.16578904846e-02 3.18433510883e-02 4.60292144819e-02 9.58701888529e-02 1.78937466321e-01 1.78937466321e-01 - 9.58701888529e-02 4.60292144819e-02 3.18433510883e-02 3.16578904846e-02 3.72212996136e-02 4.27231306029e-02 - 4.57138189196e-02 4.27231306029e-02 3.72212996136e-02 - 3.92607806531e-02 5.21992902722e-02 9.58701888529e-02 1.65125441372e-01 1.94666393607e-01 1.65125441372e-01 - 9.58701888529e-02 5.21992902722e-02 3.92607806531e-02 3.70588268986e-02 3.66436362392e-02 3.62873182074e-02 - 3.62873182074e-02 3.66436362392e-02 3.70588268986e-02 - 5.77610358000e-02 9.62989068912e-02 1.78937466321e-01 1.94666393607e-01 1.94666393607e-01 1.78937466321e-01 - 9.62989068912e-02 5.77610358000e-02 4.31689270135e-02 3.18895470196e-02 2.71142914004e-02 2.45905634624e-02 - 2.71142914004e-02 3.18895470196e-02 4.31689270135e-02 - 7.23281111668e-02 1.28883911425e-01 1.78937466321e-01 1.65125441372e-01 1.78937466321e-01 1.28883911425e-01 - 7.23281111668e-02 4.49581105718e-02 2.94734728104e-02 2.09853466063e-02 1.67257046016e-02 1.67257046016e-02 - 2.09853466063e-02 2.94734728104e-02 4.49581105718e-02 - 7.23281111668e-02 9.62989068912e-02 9.58701888529e-02 9.58701888529e-02 9.62989068912e-02 7.23281111668e-02 - 4.78620662790e-02 2.78871820412e-02 1.64170481704e-02 1.00194370256e-02 8.33783116227e-03 1.00194370256e-02 - 1.64170481704e-02 2.78871820412e-02 4.78620662790e-02 - 5.77610358000e-02 5.21992902722e-02 4.60292144819e-02 5.21992902722e-02 5.77610358000e-02 4.49581105718e-02 - 2.78871820412e-02 1.43763201712e-02 7.08929107284e-03 3.65259578883e-03 3.65259578883e-03 7.08929107284e-03 - 1.43763201712e-02 2.78871820412e-02 4.49581105718e-02 - 3.92607806531e-02 3.18433510883e-02 3.18433510883e-02 3.92607806531e-02 4.31689270135e-02 2.94734728104e-02 - 1.64170481704e-02 7.08929107284e-03 2.06279931440e-03 7.49011933950e-04 2.06279931440e-03 7.08929107284e-03 - 1.64170481704e-02 2.94734728104e-02 4.31689270135e-02 - 3.16578904846e-02 2.80289195955e-02 3.16578904846e-02 3.70588268986e-02 3.18895470196e-02 2.09853466063e-02 - 1.00194370256e-02 3.65259578883e-03 7.49011933950e-04 7.49011933950e-04 3.65259578883e-03 1.00194370256e-02 - 2.09853466063e-02 3.18895470196e-02 3.70588268986e-02 - 3.42073290623e-02 3.42073290623e-02 3.72212996136e-02 3.66436362392e-02 2.71142914004e-02 1.67257046016e-02 - 8.33783116227e-03 3.65259578883e-03 2.06279931440e-03 3.65259578883e-03 8.33783116227e-03 1.67257046016e-02 - 2.71142914004e-02 3.66436362392e-02 3.72212996136e-02 - 4.77193978469e-02 4.46676835001e-02 4.27231306029e-02 3.62873182074e-02 2.45905634624e-02 1.67257046016e-02 - 1.00194370256e-02 7.08929107284e-03 7.08929107284e-03 1.00194370256e-02 1.67257046016e-02 2.45905634624e-02 - 3.62873182074e-02 4.27231306029e-02 4.46676835001e-02 - 5.61335799375e-02 5.23936115453e-02 4.57138189196e-02 3.62873182074e-02 2.71142914004e-02 2.09853466063e-02 - 1.64170481704e-02 1.43763201712e-02 1.64170481704e-02 2.09853466063e-02 2.71142914004e-02 3.62873182074e-02 - 4.57138189196e-02 5.23936115453e-02 5.61335799375e-02 - 6.46788103129e-02 5.23936115453e-02 4.27231306029e-02 3.66436362392e-02 3.18895470196e-02 2.94734728104e-02 - 2.78871820412e-02 2.78871820412e-02 2.94734728104e-02 3.18895470196e-02 3.66436362392e-02 4.27231306029e-02 - 5.23936115453e-02 6.46788103129e-02 7.10707876951e-02 - 5.61335799375e-02 4.46676835001e-02 3.72212996136e-02 3.70588268986e-02 4.31689270135e-02 4.49581105718e-02 - 4.78620662790e-02 4.49581105718e-02 4.31689270135e-02 3.70588268986e-02 3.72212996136e-02 4.46676835001e-02 - 5.61335799375e-02 7.10707876951e-02 7.10707876951e-02 - 3.72920735613e-02 3.78194924587e-02 4.30811551462e-02 5.15953878901e-02 7.23281111668e-02 8.49232065105e-02 - 7.23281111668e-02 5.15953878901e-02 4.30811551462e-02 3.78194924587e-02 3.72920735613e-02 3.85875039457e-02 - 3.97809938382e-02 3.97809938382e-02 3.85875039457e-02 - 3.78194924587e-02 4.32649438817e-02 5.41698533623e-02 8.90053073193e-02 1.28883911425e-01 1.28883911425e-01 - 8.90053073193e-02 5.41698533623e-02 4.32649438817e-02 3.78194924587e-02 3.60022233749e-02 3.49506556165e-02 - 3.70994631101e-02 3.49506556165e-02 3.60022233749e-02 - 4.30811551462e-02 5.41698533623e-02 9.32262743675e-02 1.55893621190e-01 1.78937466321e-01 1.55893621190e-01 - 9.32262743675e-02 5.41698533623e-02 4.30811551462e-02 3.54179538479e-02 3.30817407316e-02 3.00019977593e-02 - 3.00019977593e-02 3.30817407316e-02 3.54179538479e-02 - 5.15953878901e-02 8.90053073193e-02 1.55893621190e-01 1.65125441372e-01 1.65125441372e-01 1.55893621190e-01 - 8.90053073193e-02 5.15953878901e-02 3.88199146511e-02 3.00331874093e-02 2.47929678075e-02 2.24956267167e-02 - 2.47929678075e-02 3.00331874093e-02 3.88199146511e-02 - 7.23281111668e-02 1.28883911425e-01 1.78937466321e-01 1.65125441372e-01 1.78937466321e-01 1.28883911425e-01 - 7.23281111668e-02 4.49581105718e-02 2.94734728104e-02 2.09853466063e-02 1.67257046016e-02 1.67257046016e-02 - 2.09853466063e-02 2.94734728104e-02 4.49581105718e-02 - 8.49232065105e-02 1.28883911425e-01 1.55893621190e-01 1.55893621190e-01 1.28883911425e-01 8.49232065105e-02 - 5.02965815685e-02 3.02094126272e-02 1.68972398574e-02 1.09153312429e-02 9.07395147277e-03 1.09153312429e-02 - 1.68972398574e-02 3.02094126272e-02 5.02965815685e-02 - 7.23281111668e-02 8.90053073193e-02 9.32262743675e-02 8.90053073193e-02 7.23281111668e-02 5.02965815685e-02 - 2.97058195093e-02 1.56734539515e-02 8.96019182257e-03 5.26506660326e-03 5.26506660326e-03 8.96019182257e-03 - 1.56734539515e-02 2.97058195093e-02 5.02965815685e-02 - 5.15953878901e-02 5.41698533623e-02 5.41698533623e-02 5.15953878901e-02 4.49581105718e-02 3.02094126272e-02 - 1.56734539515e-02 7.32490142478e-03 3.29271151918e-03 2.09418534593e-03 3.29271151918e-03 7.32490142478e-03 - 1.56734539515e-02 3.02094126272e-02 4.49581105718e-02 - 4.30811551462e-02 4.32649438817e-02 4.30811551462e-02 3.88199146511e-02 2.94734728104e-02 1.68972398574e-02 - 8.96019182257e-03 3.29271151918e-03 1.38542142842e-03 1.38542142842e-03 3.29271151918e-03 8.96019182257e-03 - 1.68972398574e-02 2.94734728104e-02 3.88199146511e-02 - 3.78194924587e-02 3.78194924587e-02 3.54179538479e-02 3.00331874093e-02 2.09853466063e-02 1.09153312429e-02 - 5.26506660326e-03 2.09418534593e-03 1.38542142842e-03 2.09418534593e-03 5.26506660326e-03 1.09153312429e-02 - 2.09853466063e-02 3.00331874093e-02 3.54179538479e-02 - 3.72920735613e-02 3.60022233749e-02 3.30817407316e-02 2.47929678075e-02 1.67257046016e-02 9.07395147277e-03 - 5.26506660326e-03 3.29271151918e-03 3.29271151918e-03 5.26506660326e-03 9.07395147277e-03 1.67257046016e-02 - 2.47929678075e-02 3.30817407316e-02 3.60022233749e-02 - 3.85875039457e-02 3.49506556165e-02 3.00019977593e-02 2.24956267167e-02 1.67257046016e-02 1.09153312429e-02 - 8.96019182257e-03 7.32490142478e-03 8.96019182257e-03 1.09153312429e-02 1.67257046016e-02 2.24956267167e-02 - 3.00019977593e-02 3.49506556165e-02 3.85875039457e-02 - 3.97809938382e-02 3.70994631101e-02 3.00019977593e-02 2.47929678075e-02 2.09853466063e-02 1.68972398574e-02 - 1.56734539515e-02 1.56734539515e-02 1.68972398574e-02 2.09853466063e-02 2.47929678075e-02 3.00019977593e-02 - 3.70994631101e-02 3.97809938382e-02 4.24295300017e-02 - 3.97809938382e-02 3.49506556165e-02 3.30817407316e-02 3.00331874093e-02 2.94734728104e-02 3.02094126272e-02 - 2.97058195093e-02 3.02094126272e-02 2.94734728104e-02 3.00331874093e-02 3.30817407316e-02 3.49506556165e-02 - 3.97809938382e-02 4.45611403771e-02 4.45611403771e-02 - 3.85875039457e-02 3.60022233749e-02 3.54179538479e-02 3.88199146511e-02 4.49581105718e-02 5.02965815685e-02 - 5.02965815685e-02 4.49581105718e-02 3.88199146511e-02 3.54179538479e-02 3.60022233749e-02 3.85875039457e-02 - 4.24295300017e-02 4.45611403771e-02 4.24295300017e-02 - 3.18085146238e-02 4.02222959331e-02 5.01938086050e-02 6.40797263891e-02 7.23281111668e-02 7.23281111668e-02 - 6.40797263891e-02 5.01938086050e-02 4.02222959331e-02 3.18085146238e-02 2.77237360128e-02 2.42768838976e-02 - 2.38593064853e-02 2.42768838976e-02 2.77237360128e-02 - 4.02222959331e-02 5.27225154807e-02 7.22755322173e-02 8.90053073193e-02 9.62989068912e-02 8.90053073193e-02 - 7.22755322173e-02 5.27225154807e-02 4.02222959331e-02 3.08604739677e-02 2.43879073207e-02 2.13888507874e-02 - 2.13888507874e-02 2.43879073207e-02 3.08604739677e-02 - 5.01938086050e-02 7.22755322173e-02 9.32262743675e-02 9.58701888529e-02 9.58701888529e-02 9.32262743675e-02 - 7.22755322173e-02 5.01938086050e-02 3.49314010620e-02 2.55149658819e-02 1.86401066199e-02 1.63183576357e-02 - 1.86401066199e-02 2.55149658819e-02 3.49314010620e-02 - 6.40797263891e-02 8.90053073193e-02 9.58701888529e-02 6.76894902970e-02 9.58701888529e-02 8.90053073193e-02 - 6.40797263891e-02 4.06623304521e-02 2.56772132883e-02 1.73142648442e-02 1.28060436295e-02 1.28060436295e-02 - 1.73142648442e-02 2.56772132883e-02 4.06623304521e-02 - 7.23281111668e-02 9.62989068912e-02 9.58701888529e-02 9.58701888529e-02 9.62989068912e-02 7.23281111668e-02 - 4.78620662790e-02 2.78871820412e-02 1.64170481704e-02 1.00194370256e-02 8.33783116227e-03 1.00194370256e-02 - 1.64170481704e-02 2.78871820412e-02 4.78620662790e-02 - 7.23281111668e-02 8.90053073193e-02 9.32262743675e-02 8.90053073193e-02 7.23281111668e-02 5.02965815685e-02 - 2.97058195093e-02 1.56734539515e-02 8.96019182257e-03 5.26506660326e-03 5.26506660326e-03 8.96019182257e-03 - 1.56734539515e-02 2.97058195093e-02 5.02965815685e-02 - 6.40797263891e-02 7.22755322173e-02 7.22755322173e-02 6.40797263891e-02 4.78620662790e-02 2.97058195093e-02 - 1.53666013085e-02 7.66605713530e-03 3.60269726266e-03 2.68920014434e-03 3.60269726266e-03 7.66605713530e-03 - 1.53666013085e-02 2.97058195093e-02 4.78620662790e-02 - 5.01938086050e-02 5.27225154807e-02 5.01938086050e-02 4.06623304521e-02 2.78871820412e-02 1.56734539515e-02 - 7.66605713530e-03 2.63663327441e-03 1.11118867375e-03 1.11118867375e-03 2.63663327441e-03 7.66605713530e-03 - 1.56734539515e-02 2.78871820412e-02 4.06623304521e-02 - 4.02222959331e-02 4.02222959331e-02 3.49314010620e-02 2.56772132883e-02 1.64170481704e-02 8.96019182257e-03 - 3.60269726266e-03 1.11118867375e-03 5.28120831661e-04 1.11118867375e-03 3.60269726266e-03 8.96019182257e-03 - 1.64170481704e-02 2.56772132883e-02 3.49314010620e-02 - 3.18085146238e-02 3.08604739677e-02 2.55149658819e-02 1.73142648442e-02 1.00194370256e-02 5.26506660326e-03 - 2.68920014434e-03 1.11118867375e-03 1.11118867375e-03 2.68920014434e-03 5.26506660326e-03 1.00194370256e-02 - 1.73142648442e-02 2.55149658819e-02 3.08604739677e-02 - 2.77237360128e-02 2.43879073207e-02 1.86401066199e-02 1.28060436295e-02 8.33783116227e-03 5.26506660326e-03 - 3.60269726266e-03 2.63663327441e-03 3.60269726266e-03 5.26506660326e-03 8.33783116227e-03 1.28060436295e-02 - 1.86401066199e-02 2.43879073207e-02 2.77237360128e-02 - 2.42768838976e-02 2.13888507874e-02 1.63183576357e-02 1.28060436295e-02 1.00194370256e-02 8.96019182257e-03 - 7.66605713530e-03 7.66605713530e-03 8.96019182257e-03 1.00194370256e-02 1.28060436295e-02 1.63183576357e-02 - 2.13888507874e-02 2.42768838976e-02 2.61642583587e-02 - 2.38593064853e-02 2.13888507874e-02 1.86401066199e-02 1.73142648442e-02 1.64170481704e-02 1.56734539515e-02 - 1.53666013085e-02 1.56734539515e-02 1.64170481704e-02 1.73142648442e-02 1.86401066199e-02 2.13888507874e-02 - 2.38593064853e-02 2.49875482035e-02 2.49875482035e-02 - 2.42768838976e-02 2.43879073207e-02 2.55149658819e-02 2.56772132883e-02 2.78871820412e-02 2.97058195093e-02 - 2.97058195093e-02 2.78871820412e-02 2.56772132883e-02 2.55149658819e-02 2.43879073207e-02 2.42768838976e-02 - 2.49875482035e-02 2.65236093943e-02 2.49875482035e-02 - 2.77237360128e-02 3.08604739677e-02 3.49314010620e-02 4.06623304521e-02 4.78620662790e-02 5.02965815685e-02 - 4.78620662790e-02 4.06623304521e-02 3.49314010620e-02 3.08604739677e-02 2.77237360128e-02 2.61642583587e-02 - 2.49875482035e-02 2.49875482035e-02 2.61642583587e-02 - 2.99172055199e-02 4.34116802001e-02 5.01938086050e-02 5.15953878901e-02 5.77610358000e-02 5.15953878901e-02 - 5.01938086050e-02 4.34116802001e-02 2.99172055199e-02 2.21844351566e-02 1.56599181080e-02 1.25503014842e-02 - 1.25503014842e-02 1.56599181080e-02 2.21844351566e-02 - 4.34116802001e-02 5.27225154807e-02 5.41698533623e-02 5.21992902722e-02 5.21992902722e-02 5.41698533623e-02 - 5.27225154807e-02 4.34116802001e-02 2.79666745937e-02 1.78991060313e-02 1.23746382387e-02 9.66467351917e-03 - 1.23746382387e-02 1.78991060313e-02 2.79666745937e-02 - 5.01938086050e-02 5.41698533623e-02 4.60292144819e-02 3.36135559891e-02 4.60292144819e-02 5.41698533623e-02 - 5.01938086050e-02 3.55073021064e-02 2.15005775831e-02 1.24898667953e-02 8.27412039851e-03 8.27412039851e-03 - 1.24898667953e-02 2.15005775831e-02 3.55073021064e-02 - 5.15953878901e-02 5.21992902722e-02 3.36135559891e-02 3.36135559891e-02 5.21992902722e-02 5.15953878901e-02 - 4.06623304521e-02 2.60569281693e-02 1.40094656991e-02 7.35387417276e-03 5.53145518161e-03 7.35387417276e-03 - 1.40094656991e-02 2.60569281693e-02 4.06623304521e-02 - 5.77610358000e-02 5.21992902722e-02 4.60292144819e-02 5.21992902722e-02 5.77610358000e-02 4.49581105718e-02 - 2.78871820412e-02 1.43763201712e-02 7.08929107284e-03 3.65259578883e-03 3.65259578883e-03 7.08929107284e-03 - 1.43763201712e-02 2.78871820412e-02 4.49581105718e-02 - 5.15953878901e-02 5.41698533623e-02 5.41698533623e-02 5.15953878901e-02 4.49581105718e-02 3.02094126272e-02 - 1.56734539515e-02 7.32490142478e-03 3.29271151918e-03 2.09418534593e-03 3.29271151918e-03 7.32490142478e-03 - 1.56734539515e-02 3.02094126272e-02 4.49581105718e-02 - 5.01938086050e-02 5.27225154807e-02 5.01938086050e-02 4.06623304521e-02 2.78871820412e-02 1.56734539515e-02 - 7.66605713530e-03 2.63663327441e-03 1.11118867375e-03 1.11118867375e-03 2.63663327441e-03 7.66605713530e-03 - 1.56734539515e-02 2.78871820412e-02 4.06623304521e-02 - 4.34116802001e-02 4.34116802001e-02 3.55073021064e-02 2.60569281693e-02 1.43763201712e-02 7.32490142479e-03 - 2.63663327441e-03 9.44908409994e-04 3.22484228196e-04 9.44908409994e-04 2.63663327441e-03 7.32490142479e-03 - 1.43763201712e-02 2.60569281693e-02 3.55073021064e-02 - 2.99172055199e-02 2.79666745937e-02 2.15005775831e-02 1.40094656991e-02 7.08929107284e-03 3.29271151918e-03 - 1.11118867375e-03 3.22484228196e-04 3.22484228196e-04 1.11118867375e-03 3.29271151918e-03 7.08929107284e-03 - 1.40094656991e-02 2.15005775831e-02 2.79666745937e-02 - 2.21844351566e-02 1.78991060313e-02 1.24898667953e-02 7.35387417276e-03 3.65259578883e-03 2.09418534593e-03 - 1.11118867375e-03 9.44908409994e-04 1.11118867375e-03 2.09418534593e-03 3.65259578883e-03 7.35387417276e-03 - 1.24898667953e-02 1.78991060313e-02 2.21844351566e-02 - 1.56599181080e-02 1.23746382387e-02 8.27412039851e-03 5.53145518161e-03 3.65259578883e-03 3.29271151918e-03 - 2.63663327441e-03 2.63663327441e-03 3.29271151918e-03 3.65259578883e-03 5.53145518161e-03 8.27412039851e-03 - 1.23746382387e-02 1.56599181080e-02 1.79880069490e-02 - 1.25503014842e-02 9.66467351917e-03 8.27412039851e-03 7.35387417276e-03 7.08929107284e-03 7.32490142478e-03 - 7.66605713530e-03 7.32490142478e-03 7.08929107284e-03 7.35387417276e-03 8.27412039851e-03 9.66467351917e-03 - 1.25503014842e-02 1.49627569580e-02 1.49627569580e-02 - 1.25503014842e-02 1.23746382387e-02 1.24898667953e-02 1.40094656991e-02 1.43763201712e-02 1.56734539515e-02 - 1.56734539515e-02 1.43763201712e-02 1.40094656991e-02 1.24898667953e-02 1.23746382387e-02 1.25503014842e-02 - 1.36231130391e-02 1.31818719408e-02 1.36231130391e-02 - 1.56599181080e-02 1.78991060313e-02 2.15005775831e-02 2.60569281693e-02 2.78871820412e-02 3.02094126272e-02 - 2.78871820412e-02 2.60569281693e-02 2.15005775831e-02 1.78991060313e-02 1.56599181080e-02 1.49627569580e-02 - 1.31818719408e-02 1.31818719408e-02 1.49627569580e-02 - 2.21844351566e-02 2.79666745937e-02 3.55073021064e-02 4.06623304521e-02 4.49581105718e-02 4.49581105718e-02 - 4.06623304521e-02 3.55073021064e-02 2.79666745937e-02 2.21844351566e-02 1.79880069490e-02 1.49627569580e-02 - 1.36231130391e-02 1.49627569580e-02 1.79880069490e-02 - 2.99172055199e-02 4.02222959331e-02 4.30811551462e-02 3.92607806531e-02 3.92607806531e-02 4.30811551462e-02 - 4.02222959331e-02 2.99172055199e-02 1.95071574201e-02 1.13646063453e-02 6.70218401668e-03 4.59069441377e-03 - 6.70218401668e-03 1.13646063453e-02 1.95071574201e-02 - 4.02222959331e-02 4.32649438817e-02 3.18433510883e-02 2.47958446021e-02 3.18433510883e-02 4.32649438817e-02 - 4.02222959331e-02 2.79666745937e-02 1.52963446205e-02 7.92513025055e-03 3.86875558844e-03 3.86875558844e-03 - 7.92513025055e-03 1.52963446205e-02 2.79666745937e-02 - 4.30811551462e-02 3.18433510883e-02 1.22283197111e-02 1.22283197111e-02 3.18433510883e-02 4.30811551462e-02 - 3.49314010620e-02 2.15005775831e-02 9.79121310155e-03 4.02268913133e-03 2.15067339004e-03 4.02268913133e-03 - 9.79121310155e-03 2.15005775831e-02 3.49314010620e-02 - 3.92607806531e-02 2.47958446021e-02 1.22283197111e-02 2.47958446021e-02 3.92607806531e-02 3.88199146511e-02 - 2.56772132883e-02 1.40094656991e-02 5.18295256070e-03 1.64102393395e-03 1.64102393395e-03 5.18295256070e-03 - 1.40094656991e-02 2.56772132883e-02 3.88199146511e-02 - 3.92607806531e-02 3.18433510883e-02 3.18433510883e-02 3.92607806531e-02 4.31689270135e-02 2.94734728104e-02 - 1.64170481704e-02 7.08929107284e-03 2.06279931440e-03 7.49011933950e-04 2.06279931440e-03 7.08929107284e-03 - 1.64170481704e-02 2.94734728104e-02 4.31689270135e-02 - 4.30811551462e-02 4.32649438817e-02 4.30811551462e-02 3.88199146511e-02 2.94734728104e-02 1.68972398574e-02 - 8.96019182257e-03 3.29271151918e-03 1.38542142842e-03 1.38542142842e-03 3.29271151918e-03 8.96019182257e-03 - 1.68972398574e-02 2.94734728104e-02 3.88199146511e-02 - 4.02222959331e-02 4.02222959331e-02 3.49314010620e-02 2.56772132883e-02 1.64170481704e-02 8.96019182257e-03 - 3.60269726266e-03 1.11118867375e-03 5.28120831661e-04 1.11118867375e-03 3.60269726266e-03 8.96019182257e-03 - 1.64170481704e-02 2.56772132883e-02 3.49314010620e-02 - 2.99172055199e-02 2.79666745937e-02 2.15005775831e-02 1.40094656991e-02 7.08929107284e-03 3.29271151918e-03 - 1.11118867375e-03 3.22484228196e-04 3.22484228196e-04 1.11118867375e-03 3.29271151918e-03 7.08929107284e-03 - 1.40094656991e-02 2.15005775831e-02 2.79666745937e-02 - 1.95071574201e-02 1.52963446205e-02 9.79121310155e-03 5.18295256070e-03 2.06279931440e-03 1.38542142842e-03 - 5.28120831661e-04 3.22484228196e-04 5.28120831661e-04 1.38542142842e-03 2.06279931440e-03 5.18295256070e-03 - 9.79121310155e-03 1.52963446205e-02 1.95071574201e-02 - 1.13646063453e-02 7.92513025055e-03 4.02268913133e-03 1.64102393395e-03 7.49011933950e-04 1.38542142842e-03 - 1.11118867375e-03 1.11118867375e-03 1.38542142842e-03 7.49011933950e-04 1.64102393395e-03 4.02268913133e-03 - 7.92513025055e-03 1.13646063453e-02 1.40786159428e-02 - 6.70218401668e-03 3.86875558844e-03 2.15067339004e-03 1.64102393395e-03 2.06279931440e-03 3.29271151918e-03 - 3.60269726266e-03 3.29271151918e-03 2.06279931440e-03 1.64102393395e-03 2.15067339004e-03 3.86875558844e-03 - 6.70218401668e-03 8.64523106293e-03 8.64523106293e-03 - 4.59069441377e-03 3.86875558844e-03 4.02268913133e-03 5.18295256070e-03 7.08929107284e-03 8.96019182257e-03 - 8.96019182257e-03 7.08929107284e-03 5.18295256070e-03 4.02268913133e-03 3.86875558844e-03 4.59069441377e-03 - 6.75138246715e-03 6.84444382539e-03 6.75138246715e-03 - 6.70218401668e-03 7.92513025055e-03 9.79121310155e-03 1.40094656991e-02 1.64170481704e-02 1.68972398574e-02 - 1.64170481704e-02 1.40094656991e-02 9.79121310155e-03 7.92513025055e-03 6.70218401668e-03 6.75138246715e-03 - 5.80050554995e-03 5.80050554995e-03 6.75138246715e-03 - 1.13646063453e-02 1.52963446205e-02 2.15005775831e-02 2.56772132883e-02 2.94734728104e-02 2.94734728104e-02 - 2.56772132883e-02 2.15005775831e-02 1.52963446205e-02 1.13646063453e-02 8.64523106293e-03 6.84444382539e-03 - 5.80050554995e-03 6.84444382539e-03 8.64523106293e-03 - 1.95071574201e-02 2.79666745937e-02 3.49314010620e-02 3.88199146511e-02 4.31689270135e-02 3.88199146511e-02 - 3.49314010620e-02 2.79666745937e-02 1.95071574201e-02 1.40786159428e-02 8.64523106293e-03 6.75138246715e-03 - 6.75138246715e-03 8.64523106293e-03 1.40786159428e-02 - 3.18085146238e-02 3.78194924587e-02 3.16578904846e-02 2.77650059484e-02 3.16578904846e-02 3.78194924587e-02 - 3.18085146238e-02 2.21844351566e-02 1.13646063453e-02 5.11954914432e-03 1.60658688370e-03 1.60658688370e-03 - 5.11954914432e-03 1.13646063453e-02 2.21844351566e-02 - 3.78194924587e-02 2.80289195955e-02 1.45740503756e-02 1.45740503756e-02 2.80289195955e-02 3.78194924587e-02 - 3.08604739677e-02 1.78991060313e-02 7.92513025055e-03 2.15869962360e-03 6.20007891972e-04 2.15869962360e-03 - 7.92513025055e-03 1.78991060313e-02 3.08604739677e-02 - 3.16578904846e-02 1.45740503756e-02 5.88407505951e-04 1.45740503756e-02 3.16578904846e-02 3.54179538479e-02 - 2.55149658819e-02 1.24898667953e-02 4.02268913133e-03 5.43091896583e-04 5.43091896583e-04 4.02268913133e-03 - 1.24898667953e-02 2.55149658819e-02 3.54179538479e-02 - 2.77650059484e-02 1.45740503756e-02 1.45740503756e-02 2.77650059484e-02 3.70588268986e-02 3.00331874093e-02 - 1.73142648442e-02 7.35387417276e-03 1.64102393395e-03 1.47520974601e-04 1.64102393395e-03 7.35387417276e-03 - 1.73142648442e-02 3.00331874093e-02 3.70588268986e-02 - 3.16578904846e-02 2.80289195955e-02 3.16578904846e-02 3.70588268986e-02 3.18895470196e-02 2.09853466063e-02 - 1.00194370256e-02 3.65259578883e-03 7.49011933950e-04 7.49011933950e-04 3.65259578883e-03 1.00194370256e-02 - 2.09853466063e-02 3.18895470196e-02 3.70588268986e-02 - 3.78194924587e-02 3.78194924587e-02 3.54179538479e-02 3.00331874093e-02 2.09853466063e-02 1.09153312429e-02 - 5.26506660326e-03 2.09418534593e-03 1.38542142842e-03 2.09418534593e-03 5.26506660326e-03 1.09153312429e-02 - 2.09853466063e-02 3.00331874093e-02 3.54179538479e-02 - 3.18085146238e-02 3.08604739677e-02 2.55149658819e-02 1.73142648442e-02 1.00194370256e-02 5.26506660326e-03 - 2.68920014434e-03 1.11118867375e-03 1.11118867375e-03 2.68920014434e-03 5.26506660326e-03 1.00194370256e-02 - 1.73142648442e-02 2.55149658819e-02 3.08604739677e-02 - 2.21844351566e-02 1.78991060313e-02 1.24898667953e-02 7.35387417276e-03 3.65259578883e-03 2.09418534593e-03 - 1.11118867375e-03 9.44908409994e-04 1.11118867375e-03 2.09418534593e-03 3.65259578883e-03 7.35387417276e-03 - 1.24898667953e-02 1.78991060313e-02 2.21844351566e-02 - 1.13646063453e-02 7.92513025055e-03 4.02268913133e-03 1.64102393395e-03 7.49011933950e-04 1.38542142842e-03 - 1.11118867375e-03 1.11118867375e-03 1.38542142842e-03 7.49011933950e-04 1.64102393395e-03 4.02268913133e-03 - 7.92513025055e-03 1.13646063453e-02 1.40786159428e-02 - 5.11954914432e-03 2.15869962360e-03 5.43091896583e-04 1.47520974601e-04 7.49011933950e-04 2.09418534593e-03 - 2.68920014434e-03 2.09418534593e-03 7.49011933950e-04 1.47520974601e-04 5.43091896583e-04 2.15869962360e-03 - 5.11954914432e-03 7.01813743887e-03 7.01813743887e-03 - 1.60658688370e-03 6.20007891972e-04 5.43091896583e-04 1.64102393395e-03 3.65259578883e-03 5.26506660326e-03 - 5.26506660326e-03 3.65259578883e-03 1.64102393395e-03 5.43091896583e-04 6.20007891972e-04 1.60658688370e-03 - 3.53006767470e-03 4.09661395937e-03 3.53006767470e-03 - 1.60658688370e-03 2.15869962360e-03 4.02268913133e-03 7.35387417276e-03 1.00194370256e-02 1.09153312429e-02 - 1.00194370256e-02 7.35387417276e-03 4.02268913133e-03 2.15869962360e-03 1.60658688370e-03 2.37456788896e-03 - 2.19087193687e-03 2.19087193687e-03 2.37456788896e-03 - 5.11954914432e-03 7.92513025055e-03 1.24898667953e-02 1.73142648442e-02 2.09853466063e-02 2.09853466063e-02 - 1.73142648442e-02 1.24898667953e-02 7.92513025055e-03 5.11954914432e-03 3.53006767470e-03 2.19087193687e-03 - 2.26732755529e-03 2.19087193687e-03 3.53006767470e-03 - 1.13646063453e-02 1.78991060313e-02 2.55149658819e-02 3.00331874093e-02 3.18895470196e-02 3.00331874093e-02 - 2.55149658819e-02 1.78991060313e-02 1.13646063453e-02 7.01813743887e-03 4.09661395937e-03 2.19087193687e-03 - 2.19087193687e-03 4.09661395937e-03 7.01813743887e-03 - 2.21844351566e-02 3.08604739677e-02 3.54179538479e-02 3.70588268986e-02 3.70588268986e-02 3.54179538479e-02 - 3.08604739677e-02 2.21844351566e-02 1.40786159428e-02 7.01813743887e-03 3.53006767470e-03 2.37456788896e-03 - 3.53006767470e-03 7.01813743887e-03 1.40786159428e-02 - 3.72920735613e-02 3.42073290623e-02 2.56381222091e-02 2.56381222091e-02 3.42073290623e-02 3.72920735613e-02 - 2.77237360128e-02 1.56599181080e-02 6.70218401668e-03 1.60658688370e-03 2.99179150647e-04 1.60658688370e-03 - 6.70218401668e-03 1.56599181080e-02 2.77237360128e-02 - 3.42073290623e-02 1.76415461442e-02 7.66046977902e-03 1.76415461442e-02 3.42073290623e-02 3.60022233749e-02 - 2.43879073207e-02 1.23746382387e-02 3.86875558844e-03 6.20007891972e-04 6.20007891972e-04 3.86875558844e-03 - 1.23746382387e-02 2.43879073207e-02 3.60022233749e-02 - 2.56381222091e-02 7.66046977902e-03 7.66046977902e-03 2.56381222091e-02 3.72212996136e-02 3.30817407316e-02 - 1.86401066199e-02 8.27412039851e-03 2.15067339004e-03 5.43091896583e-04 2.15067339004e-03 8.27412039851e-03 - 1.86401066199e-02 3.30817407316e-02 3.72212996136e-02 - 2.56381222091e-02 1.76415461442e-02 2.56381222091e-02 3.74646524133e-02 3.66436362392e-02 2.47929678075e-02 - 1.28060436295e-02 5.53145518161e-03 1.64102393395e-03 1.64102393395e-03 5.53145518161e-03 1.28060436295e-02 - 2.47929678075e-02 3.66436362392e-02 3.74646524133e-02 - 3.42073290623e-02 3.42073290623e-02 3.72212996136e-02 3.66436362392e-02 2.71142914004e-02 1.67257046016e-02 - 8.33783116227e-03 3.65259578883e-03 2.06279931440e-03 3.65259578883e-03 8.33783116227e-03 1.67257046016e-02 - 2.71142914004e-02 3.66436362392e-02 3.72212996136e-02 - 3.72920735613e-02 3.60022233749e-02 3.30817407316e-02 2.47929678075e-02 1.67257046016e-02 9.07395147277e-03 - 5.26506660326e-03 3.29271151918e-03 3.29271151918e-03 5.26506660326e-03 9.07395147277e-03 1.67257046016e-02 - 2.47929678075e-02 3.30817407316e-02 3.60022233749e-02 - 2.77237360128e-02 2.43879073207e-02 1.86401066199e-02 1.28060436295e-02 8.33783116227e-03 5.26506660326e-03 - 3.60269726266e-03 2.63663327441e-03 3.60269726266e-03 5.26506660326e-03 8.33783116227e-03 1.28060436295e-02 - 1.86401066199e-02 2.43879073207e-02 2.77237360128e-02 - 1.56599181080e-02 1.23746382387e-02 8.27412039851e-03 5.53145518161e-03 3.65259578883e-03 3.29271151918e-03 - 2.63663327441e-03 2.63663327441e-03 3.29271151918e-03 3.65259578883e-03 5.53145518161e-03 8.27412039851e-03 - 1.23746382387e-02 1.56599181080e-02 1.79880069490e-02 - 6.70218401668e-03 3.86875558844e-03 2.15067339004e-03 1.64102393395e-03 2.06279931440e-03 3.29271151918e-03 - 3.60269726266e-03 3.29271151918e-03 2.06279931440e-03 1.64102393395e-03 2.15067339004e-03 3.86875558844e-03 - 6.70218401668e-03 8.64523106293e-03 8.64523106293e-03 - 1.60658688370e-03 6.20007891972e-04 5.43091896583e-04 1.64102393395e-03 3.65259578883e-03 5.26506660326e-03 - 5.26506660326e-03 3.65259578883e-03 1.64102393395e-03 5.43091896583e-04 6.20007891972e-04 1.60658688370e-03 - 3.53006767470e-03 4.09661395937e-03 3.53006767470e-03 - 2.99179150647e-04 6.20007891972e-04 2.15067339004e-03 5.53145518161e-03 8.33783116227e-03 9.07395147277e-03 - 8.33783116227e-03 5.53145518161e-03 2.15067339004e-03 6.20007891972e-04 2.99179150647e-04 1.09356208174e-03 - 1.38503345990e-03 1.38503345990e-03 1.09356208174e-03 - 1.60658688370e-03 3.86875558844e-03 8.27412039851e-03 1.28060436295e-02 1.67257046016e-02 1.67257046016e-02 - 1.28060436295e-02 8.27412039851e-03 3.86875558844e-03 1.60658688370e-03 1.09356208174e-03 7.14663240551e-04 - 7.65589594291e-04 7.14663240551e-04 1.09356208174e-03 - 6.70218401668e-03 1.23746382387e-02 1.86401066199e-02 2.47929678075e-02 2.71142914004e-02 2.47929678075e-02 - 1.86401066199e-02 1.23746382387e-02 6.70218401668e-03 3.53006767470e-03 1.38503345990e-03 7.65589594291e-04 - 7.65589594291e-04 1.38503345990e-03 3.53006767470e-03 - 1.56599181080e-02 2.43879073207e-02 3.30817407316e-02 3.66436362392e-02 3.66436362392e-02 3.30817407316e-02 - 2.43879073207e-02 1.56599181080e-02 8.64523106293e-03 4.09661395937e-03 1.38503345990e-03 7.14663240551e-04 - 1.38503345990e-03 4.09661395937e-03 8.64523106293e-03 - 2.77237360128e-02 3.60022233749e-02 3.72212996136e-02 3.74646524133e-02 3.72212996136e-02 3.60022233749e-02 - 2.77237360128e-02 1.79880069490e-02 8.64523106293e-03 3.53006767470e-03 1.09356208174e-03 1.09356208174e-03 - 3.53006767470e-03 8.64523106293e-03 1.79880069490e-02 - 4.77193978469e-02 3.50294268557e-02 2.76155885440e-02 3.50294268557e-02 4.77193978469e-02 3.85875039457e-02 - 2.42768838976e-02 1.25503014842e-02 4.59069441377e-03 1.60658688370e-03 1.60658688370e-03 4.59069441377e-03 - 1.25503014842e-02 2.42768838976e-02 3.85875039457e-02 - 3.50294268557e-02 1.55091198600e-02 1.55091198600e-02 3.50294268557e-02 4.46676835001e-02 3.49506556165e-02 - 2.13888507874e-02 9.66467351917e-03 3.86875558844e-03 2.15869962360e-03 3.86875558844e-03 9.66467351917e-03 - 2.13888507874e-02 3.49506556165e-02 4.46676835001e-02 - 2.76155885440e-02 1.55091198600e-02 2.76155885440e-02 4.40351969320e-02 4.27231306029e-02 3.00019977593e-02 - 1.63183576357e-02 8.27412039851e-03 4.02268913133e-03 4.02268913133e-03 8.27412039851e-03 1.63183576357e-02 - 3.00019977593e-02 4.27231306029e-02 4.40351969320e-02 - 3.50294268557e-02 3.50294268557e-02 4.40351969320e-02 4.48502710858e-02 3.62873182074e-02 2.24956267167e-02 - 1.28060436295e-02 7.35387417276e-03 5.18295256070e-03 7.35387417276e-03 1.28060436295e-02 2.24956267167e-02 - 3.62873182074e-02 4.48502710858e-02 4.40351969320e-02 - 4.77193978469e-02 4.46676835001e-02 4.27231306029e-02 3.62873182074e-02 2.45905634624e-02 1.67257046016e-02 - 1.00194370256e-02 7.08929107284e-03 7.08929107284e-03 1.00194370256e-02 1.67257046016e-02 2.45905634624e-02 - 3.62873182074e-02 4.27231306029e-02 4.46676835001e-02 - 3.85875039457e-02 3.49506556165e-02 3.00019977593e-02 2.24956267167e-02 1.67257046016e-02 1.09153312429e-02 - 8.96019182257e-03 7.32490142478e-03 8.96019182257e-03 1.09153312429e-02 1.67257046016e-02 2.24956267167e-02 - 3.00019977593e-02 3.49506556165e-02 3.85875039457e-02 - 2.42768838976e-02 2.13888507874e-02 1.63183576357e-02 1.28060436295e-02 1.00194370256e-02 8.96019182257e-03 - 7.66605713530e-03 7.66605713530e-03 8.96019182257e-03 1.00194370256e-02 1.28060436295e-02 1.63183576357e-02 - 2.13888507874e-02 2.42768838976e-02 2.61642583587e-02 - 1.25503014842e-02 9.66467351917e-03 8.27412039851e-03 7.35387417276e-03 7.08929107284e-03 7.32490142478e-03 - 7.66605713530e-03 7.32490142478e-03 7.08929107284e-03 7.35387417276e-03 8.27412039851e-03 9.66467351917e-03 - 1.25503014842e-02 1.49627569580e-02 1.49627569580e-02 - 4.59069441377e-03 3.86875558844e-03 4.02268913133e-03 5.18295256070e-03 7.08929107284e-03 8.96019182257e-03 - 8.96019182257e-03 7.08929107284e-03 5.18295256070e-03 4.02268913133e-03 3.86875558844e-03 4.59069441377e-03 - 6.75138246715e-03 6.84444382539e-03 6.75138246715e-03 - 1.60658688370e-03 2.15869962360e-03 4.02268913133e-03 7.35387417276e-03 1.00194370256e-02 1.09153312429e-02 - 1.00194370256e-02 7.35387417276e-03 4.02268913133e-03 2.15869962360e-03 1.60658688370e-03 2.37456788896e-03 - 2.19087193687e-03 2.19087193687e-03 2.37456788896e-03 - 1.60658688370e-03 3.86875558844e-03 8.27412039851e-03 1.28060436295e-02 1.67257046016e-02 1.67257046016e-02 - 1.28060436295e-02 8.27412039851e-03 3.86875558844e-03 1.60658688370e-03 1.09356208174e-03 7.14663240551e-04 - 7.65589594291e-04 7.14663240551e-04 1.09356208174e-03 - 4.59069441377e-03 9.66467351917e-03 1.63183576357e-02 2.24956267167e-02 2.45905634624e-02 2.24956267167e-02 - 1.63183576357e-02 9.66467351917e-03 4.59069441377e-03 2.37456788896e-03 7.14663240551e-04 2.93067202991e-04 - 2.93067202991e-04 7.14663240551e-04 2.37456788896e-03 - 1.25503014842e-02 2.13888507874e-02 3.00019977593e-02 3.62873182074e-02 3.62873182074e-02 3.00019977593e-02 - 2.13888507874e-02 1.25503014842e-02 6.75138246715e-03 2.19087193687e-03 7.65589594291e-04 2.93067202991e-04 - 7.65589594291e-04 2.19087193687e-03 6.75138246715e-03 - 2.42768838976e-02 3.49506556165e-02 4.27231306029e-02 4.48502710858e-02 4.27231306029e-02 3.49506556165e-02 - 2.42768838976e-02 1.49627569580e-02 6.84444382539e-03 2.19087193687e-03 7.14663240551e-04 7.14663240551e-04 - 2.19087193687e-03 6.84444382539e-03 1.49627569580e-02 - 3.85875039457e-02 4.46676835001e-02 4.40351969320e-02 4.40351969320e-02 4.46676835001e-02 3.85875039457e-02 - 2.61642583587e-02 1.49627569580e-02 6.75138246715e-03 2.37456788896e-03 1.09356208174e-03 2.37456788896e-03 - 6.75138246715e-03 1.49627569580e-02 2.61642583587e-02 - 6.52691932418e-02 5.03836760403e-02 5.03836760403e-02 6.52691932418e-02 5.61335799375e-02 3.97809938382e-02 - 2.38593064853e-02 1.25503014842e-02 6.70218401668e-03 5.11954914432e-03 6.70218401668e-03 1.25503014842e-02 - 2.38593064853e-02 3.97809938382e-02 5.61335799375e-02 - 5.03836760403e-02 3.75098528213e-02 5.03836760403e-02 5.92235143217e-02 5.23936115453e-02 3.70994631101e-02 - 2.13888507874e-02 1.23746382387e-02 7.92513025055e-03 7.92513025055e-03 1.23746382387e-02 2.13888507874e-02 - 3.70994631101e-02 5.23936115453e-02 5.92235143217e-02 - 5.03836760403e-02 5.03836760403e-02 6.09424067177e-02 5.84769754663e-02 4.57138189196e-02 3.00019977593e-02 - 1.86401066199e-02 1.24898667953e-02 9.79121310155e-03 1.24898667953e-02 1.86401066199e-02 3.00019977593e-02 - 4.57138189196e-02 5.84769754663e-02 6.09424067177e-02 - 6.52691932418e-02 5.92235143217e-02 5.84769754663e-02 5.02634504938e-02 3.62873182074e-02 2.47929678075e-02 - 1.73142648442e-02 1.40094656991e-02 1.40094656991e-02 1.73142648442e-02 2.47929678075e-02 3.62873182074e-02 - 5.02634504938e-02 5.84769754663e-02 5.92235143217e-02 - 5.61335799375e-02 5.23936115453e-02 4.57138189196e-02 3.62873182074e-02 2.71142914004e-02 2.09853466063e-02 - 1.64170481704e-02 1.43763201712e-02 1.64170481704e-02 2.09853466063e-02 2.71142914004e-02 3.62873182074e-02 - 4.57138189196e-02 5.23936115453e-02 5.61335799375e-02 - 3.97809938382e-02 3.70994631101e-02 3.00019977593e-02 2.47929678075e-02 2.09853466063e-02 1.68972398574e-02 - 1.56734539515e-02 1.56734539515e-02 1.68972398574e-02 2.09853466063e-02 2.47929678075e-02 3.00019977593e-02 - 3.70994631101e-02 3.97809938382e-02 4.24295300017e-02 - 2.38593064853e-02 2.13888507874e-02 1.86401066199e-02 1.73142648442e-02 1.64170481704e-02 1.56734539515e-02 - 1.53666013085e-02 1.56734539515e-02 1.64170481704e-02 1.73142648442e-02 1.86401066199e-02 2.13888507874e-02 - 2.38593064853e-02 2.49875482035e-02 2.49875482035e-02 - 1.25503014842e-02 1.23746382387e-02 1.24898667953e-02 1.40094656991e-02 1.43763201712e-02 1.56734539515e-02 - 1.56734539515e-02 1.43763201712e-02 1.40094656991e-02 1.24898667953e-02 1.23746382387e-02 1.25503014842e-02 - 1.36231130391e-02 1.31818719408e-02 1.36231130391e-02 - 6.70218401668e-03 7.92513025055e-03 9.79121310155e-03 1.40094656991e-02 1.64170481704e-02 1.68972398574e-02 - 1.64170481704e-02 1.40094656991e-02 9.79121310155e-03 7.92513025055e-03 6.70218401668e-03 6.75138246715e-03 - 5.80050554995e-03 5.80050554995e-03 6.75138246715e-03 - 5.11954914432e-03 7.92513025055e-03 1.24898667953e-02 1.73142648442e-02 2.09853466063e-02 2.09853466063e-02 - 1.73142648442e-02 1.24898667953e-02 7.92513025055e-03 5.11954914432e-03 3.53006767470e-03 2.19087193687e-03 - 2.26732755529e-03 2.19087193687e-03 3.53006767470e-03 - 6.70218401668e-03 1.23746382387e-02 1.86401066199e-02 2.47929678075e-02 2.71142914004e-02 2.47929678075e-02 - 1.86401066199e-02 1.23746382387e-02 6.70218401668e-03 3.53006767470e-03 1.38503345990e-03 7.65589594291e-04 - 7.65589594291e-04 1.38503345990e-03 3.53006767470e-03 - 1.25503014842e-02 2.13888507874e-02 3.00019977593e-02 3.62873182074e-02 3.62873182074e-02 3.00019977593e-02 - 2.13888507874e-02 1.25503014842e-02 6.75138246715e-03 2.19087193687e-03 7.65589594291e-04 2.93067202991e-04 - 7.65589594291e-04 2.19087193687e-03 6.75138246715e-03 - 2.38593064853e-02 3.70994631101e-02 4.57138189196e-02 5.02634504938e-02 4.57138189196e-02 3.70994631101e-02 - 2.38593064853e-02 1.36231130391e-02 5.80050554995e-03 2.26732755529e-03 7.65589594291e-04 7.65589594291e-04 - 2.26732755529e-03 5.80050554995e-03 1.36231130391e-02 - 3.97809938382e-02 5.23936115453e-02 5.84769754663e-02 5.84769754663e-02 5.23936115453e-02 3.97809938382e-02 - 2.49875482035e-02 1.31818719408e-02 5.80050554995e-03 2.19087193687e-03 1.38503345990e-03 2.19087193687e-03 - 5.80050554995e-03 1.31818719408e-02 2.49875482035e-02 - 5.61335799375e-02 5.92235143217e-02 6.09424067177e-02 5.92235143217e-02 5.61335799375e-02 4.24295300017e-02 - 2.49875482035e-02 1.36231130391e-02 6.75138246715e-03 3.53006767470e-03 3.53006767470e-03 6.75138246715e-03 - 1.36231130391e-02 2.49875482035e-02 4.24295300017e-02 - 1.16467735388e-01 1.02568469446e-01 1.16467735388e-01 9.95387143197e-02 6.46788103129e-02 3.97809938382e-02 - 2.42768838976e-02 1.56599181080e-02 1.13646063453e-02 1.13646063453e-02 1.56599181080e-02 2.42768838976e-02 - 3.97809938382e-02 6.46788103129e-02 9.95387143197e-02 - 1.02568469446e-01 1.02568469446e-01 1.13539754126e-01 8.15533129102e-02 5.23936115453e-02 3.49506556165e-02 - 2.43879073207e-02 1.78991060313e-02 1.52963446205e-02 1.78991060313e-02 2.43879073207e-02 3.49506556165e-02 - 5.23936115453e-02 8.15533129102e-02 1.13539754126e-01 - 1.16467735388e-01 1.13539754126e-01 8.53804463649e-02 5.84769754663e-02 4.27231306029e-02 3.30817407316e-02 - 2.55149658819e-02 2.15005775831e-02 2.15005775831e-02 2.55149658819e-02 3.30817407316e-02 4.27231306029e-02 - 5.84769754663e-02 8.53804463649e-02 1.13539754126e-01 - 9.95387143197e-02 8.15533129102e-02 5.84769754663e-02 4.48502710858e-02 3.66436362392e-02 3.00331874093e-02 - 2.56772132883e-02 2.60569281693e-02 2.56772132883e-02 3.00331874093e-02 3.66436362392e-02 4.48502710858e-02 - 5.84769754663e-02 8.15533129102e-02 9.95387143197e-02 - 6.46788103129e-02 5.23936115453e-02 4.27231306029e-02 3.66436362392e-02 3.18895470196e-02 2.94734728104e-02 - 2.78871820412e-02 2.78871820412e-02 2.94734728104e-02 3.18895470196e-02 3.66436362392e-02 4.27231306029e-02 - 5.23936115453e-02 6.46788103129e-02 7.10707876951e-02 - 3.97809938382e-02 3.49506556165e-02 3.30817407316e-02 3.00331874093e-02 2.94734728104e-02 3.02094126272e-02 - 2.97058195093e-02 3.02094126272e-02 2.94734728104e-02 3.00331874093e-02 3.30817407316e-02 3.49506556165e-02 - 3.97809938382e-02 4.45611403771e-02 4.45611403771e-02 - 2.42768838976e-02 2.43879073207e-02 2.55149658819e-02 2.56772132883e-02 2.78871820412e-02 2.97058195093e-02 - 2.97058195093e-02 2.78871820412e-02 2.56772132883e-02 2.55149658819e-02 2.43879073207e-02 2.42768838976e-02 - 2.49875482035e-02 2.65236093943e-02 2.49875482035e-02 - 1.56599181080e-02 1.78991060313e-02 2.15005775831e-02 2.60569281693e-02 2.78871820412e-02 3.02094126272e-02 - 2.78871820412e-02 2.60569281693e-02 2.15005775831e-02 1.78991060313e-02 1.56599181080e-02 1.49627569580e-02 - 1.31818719408e-02 1.31818719408e-02 1.49627569580e-02 - 1.13646063453e-02 1.52963446205e-02 2.15005775831e-02 2.56772132883e-02 2.94734728104e-02 2.94734728104e-02 - 2.56772132883e-02 2.15005775831e-02 1.52963446205e-02 1.13646063453e-02 8.64523106293e-03 6.84444382539e-03 - 5.80050554995e-03 6.84444382539e-03 8.64523106293e-03 - 1.13646063453e-02 1.78991060313e-02 2.55149658819e-02 3.00331874093e-02 3.18895470196e-02 3.00331874093e-02 - 2.55149658819e-02 1.78991060313e-02 1.13646063453e-02 7.01813743887e-03 4.09661395937e-03 2.19087193687e-03 - 2.19087193687e-03 4.09661395937e-03 7.01813743887e-03 - 1.56599181080e-02 2.43879073207e-02 3.30817407316e-02 3.66436362392e-02 3.66436362392e-02 3.30817407316e-02 - 2.43879073207e-02 1.56599181080e-02 8.64523106293e-03 4.09661395937e-03 1.38503345990e-03 7.14663240551e-04 - 1.38503345990e-03 4.09661395937e-03 8.64523106293e-03 - 2.42768838976e-02 3.49506556165e-02 4.27231306029e-02 4.48502710858e-02 4.27231306029e-02 3.49506556165e-02 - 2.42768838976e-02 1.49627569580e-02 6.84444382539e-03 2.19087193687e-03 7.14663240551e-04 7.14663240551e-04 - 2.19087193687e-03 6.84444382539e-03 1.49627569580e-02 - 3.97809938382e-02 5.23936115453e-02 5.84769754663e-02 5.84769754663e-02 5.23936115453e-02 3.97809938382e-02 - 2.49875482035e-02 1.31818719408e-02 5.80050554995e-03 2.19087193687e-03 1.38503345990e-03 2.19087193687e-03 - 5.80050554995e-03 1.31818719408e-02 2.49875482035e-02 - 6.46788103129e-02 8.15533129102e-02 8.53804463649e-02 8.15533129102e-02 6.46788103129e-02 4.45611403771e-02 - 2.65236093943e-02 1.31818719408e-02 6.84444382539e-03 4.09661395937e-03 4.09661395937e-03 6.84444382539e-03 - 1.31818719408e-02 2.65236093943e-02 4.45611403771e-02 - 9.95387143197e-02 1.13539754126e-01 1.13539754126e-01 9.95387143197e-02 7.10707876951e-02 4.45611403771e-02 - 2.49875482035e-02 1.49627569580e-02 8.64523106293e-03 7.01813743887e-03 8.64523106293e-03 1.49627569580e-02 - 2.49875482035e-02 4.45611403771e-02 7.10707876951e-02 - 1.82422828268e-01 1.82422828268e-01 1.75978291339e-01 9.95387143197e-02 5.61335799375e-02 3.85875039457e-02 - 2.77237360128e-02 2.21844351566e-02 1.95071574201e-02 2.21844351566e-02 2.77237360128e-02 3.85875039457e-02 - 5.61335799375e-02 9.95387143197e-02 1.75978291339e-01 - 1.82422828268e-01 1.83280188132e-01 1.13539754126e-01 5.92235143217e-02 4.46676835001e-02 3.60022233749e-02 - 3.08604739677e-02 2.79666745937e-02 2.79666745937e-02 3.08604739677e-02 3.60022233749e-02 4.46676835001e-02 - 5.92235143217e-02 1.13539754126e-01 1.83280188132e-01 - 1.75978291339e-01 1.13539754126e-01 6.09424067177e-02 4.40351969320e-02 3.72212996136e-02 3.54179538479e-02 - 3.49314010620e-02 3.55073021064e-02 3.49314010620e-02 3.54179538479e-02 3.72212996136e-02 4.40351969320e-02 - 6.09424067177e-02 1.13539754126e-01 1.75978291339e-01 - 9.95387143197e-02 5.92235143217e-02 4.40351969320e-02 3.74646524133e-02 3.70588268986e-02 3.88199146511e-02 - 4.06623304521e-02 4.06623304521e-02 3.88199146511e-02 3.70588268986e-02 3.74646524133e-02 4.40351969320e-02 - 5.92235143217e-02 9.95387143197e-02 1.21879341630e-01 - 5.61335799375e-02 4.46676835001e-02 3.72212996136e-02 3.70588268986e-02 4.31689270135e-02 4.49581105718e-02 - 4.78620662790e-02 4.49581105718e-02 4.31689270135e-02 3.70588268986e-02 3.72212996136e-02 4.46676835001e-02 - 5.61335799375e-02 7.10707876951e-02 7.10707876951e-02 - 3.85875039457e-02 3.60022233749e-02 3.54179538479e-02 3.88199146511e-02 4.49581105718e-02 5.02965815685e-02 - 5.02965815685e-02 4.49581105718e-02 3.88199146511e-02 3.54179538479e-02 3.60022233749e-02 3.85875039457e-02 - 4.24295300017e-02 4.45611403771e-02 4.24295300017e-02 - 2.77237360128e-02 3.08604739677e-02 3.49314010620e-02 4.06623304521e-02 4.78620662790e-02 5.02965815685e-02 - 4.78620662790e-02 4.06623304521e-02 3.49314010620e-02 3.08604739677e-02 2.77237360128e-02 2.61642583587e-02 - 2.49875482035e-02 2.49875482035e-02 2.61642583587e-02 - 2.21844351566e-02 2.79666745937e-02 3.55073021064e-02 4.06623304521e-02 4.49581105718e-02 4.49581105718e-02 - 4.06623304521e-02 3.55073021064e-02 2.79666745937e-02 2.21844351566e-02 1.79880069490e-02 1.49627569580e-02 - 1.36231130391e-02 1.49627569580e-02 1.79880069490e-02 - 1.95071574201e-02 2.79666745937e-02 3.49314010620e-02 3.88199146511e-02 4.31689270135e-02 3.88199146511e-02 - 3.49314010620e-02 2.79666745937e-02 1.95071574201e-02 1.40786159428e-02 8.64523106293e-03 6.75138246715e-03 - 6.75138246715e-03 8.64523106293e-03 1.40786159428e-02 - 2.21844351566e-02 3.08604739677e-02 3.54179538479e-02 3.70588268986e-02 3.70588268986e-02 3.54179538479e-02 - 3.08604739677e-02 2.21844351566e-02 1.40786159428e-02 7.01813743887e-03 3.53006767470e-03 2.37456788896e-03 - 3.53006767470e-03 7.01813743887e-03 1.40786159428e-02 - 2.77237360128e-02 3.60022233749e-02 3.72212996136e-02 3.74646524133e-02 3.72212996136e-02 3.60022233749e-02 - 2.77237360128e-02 1.79880069490e-02 8.64523106293e-03 3.53006767470e-03 1.09356208174e-03 1.09356208174e-03 - 3.53006767470e-03 8.64523106293e-03 1.79880069490e-02 - 3.85875039457e-02 4.46676835001e-02 4.40351969320e-02 4.40351969320e-02 4.46676835001e-02 3.85875039457e-02 - 2.61642583587e-02 1.49627569580e-02 6.75138246715e-03 2.37456788896e-03 1.09356208174e-03 2.37456788896e-03 - 6.75138246715e-03 1.49627569580e-02 2.61642583587e-02 - 5.61335799375e-02 5.92235143217e-02 6.09424067177e-02 5.92235143217e-02 5.61335799375e-02 4.24295300017e-02 - 2.49875482035e-02 1.36231130391e-02 6.75138246715e-03 3.53006767470e-03 3.53006767470e-03 6.75138246715e-03 - 1.36231130391e-02 2.49875482035e-02 4.24295300017e-02 - 9.95387143197e-02 1.13539754126e-01 1.13539754126e-01 9.95387143197e-02 7.10707876951e-02 4.45611403771e-02 - 2.49875482035e-02 1.49627569580e-02 8.64523106293e-03 7.01813743887e-03 8.64523106293e-03 1.49627569580e-02 - 2.49875482035e-02 4.45611403771e-02 7.10707876951e-02 - 1.75978291339e-01 1.83280188132e-01 1.75978291339e-01 1.21879341630e-01 7.10707876951e-02 4.24295300017e-02 - 2.61642583587e-02 1.79880069490e-02 1.40786159428e-02 1.40786159428e-02 1.79880069490e-02 2.61642583587e-02 - 4.24295300017e-02 7.10707876951e-02 1.21879341630e-01 + 1.80405346865e-01 1.82505675344e-01 1.16509120681e-01 6.52918175640e-02 4.77333687537e-02 3.72988193747e-02 + 3.18111513620e-02 2.99175972602e-02 2.99175972602e-02 3.18111513620e-02 3.72988193747e-02 4.77333687537e-02 + 6.52918175640e-02 1.16509120681e-01 1.82505675344e-01 + 1.82505675344e-01 1.02615488905e-01 5.04046005791e-02 3.50377196433e-02 3.42116560673e-02 3.78205596100e-02 + 4.02211086967e-02 4.34096597635e-02 4.02211086967e-02 3.78205596100e-02 3.42116560673e-02 3.50377196433e-02 + 5.04046005791e-02 1.02615488905e-01 1.82505675344e-01 + 1.16509120681e-01 5.04046005791e-02 2.76230644258e-02 2.56404718355e-02 3.16546517690e-02 4.30758146535e-02 + 5.01876788897e-02 5.01876788897e-02 4.30758146535e-02 3.16546517690e-02 2.56404718355e-02 2.76230644258e-02 + 5.04046005791e-02 1.16509120681e-01 1.76044347723e-01 + 6.52918175640e-02 3.50377196433e-02 2.56404718355e-02 2.77593740924e-02 3.92509115207e-02 5.15825069379e-02 + 6.40667422061e-02 5.15825069379e-02 3.92509115207e-02 2.77593740924e-02 2.56404718355e-02 3.50377196433e-02 + 6.52918175640e-02 9.95704232240e-02 9.95704232240e-02 + 4.77333687537e-02 3.42116560673e-02 3.16546517690e-02 3.92509115207e-02 5.77463557561e-02 7.23072529491e-02 + 7.23072529491e-02 5.77463557561e-02 3.92509115207e-02 3.16546517690e-02 3.42116560673e-02 4.77333687537e-02 + 5.61479157062e-02 6.46951081959e-02 5.61479157062e-02 + 3.72988193747e-02 3.78205596100e-02 4.30758146535e-02 5.15825069379e-02 7.23072529491e-02 8.48962030063e-02 + 7.23072529491e-02 5.15825069379e-02 4.30758146535e-02 3.78205596100e-02 3.72988193747e-02 3.85963986077e-02 + 3.97912967555e-02 3.97912967555e-02 3.85963986077e-02 + 3.18111513620e-02 4.02211086967e-02 5.01876788897e-02 6.40667422061e-02 7.23072529491e-02 7.23072529491e-02 + 6.40667422061e-02 5.01876788897e-02 4.02211086967e-02 3.18111513620e-02 2.77285509468e-02 2.42831208041e-02 + 2.38655503495e-02 2.42831208041e-02 2.77285509468e-02 + 2.99175972602e-02 4.34096597635e-02 5.01876788897e-02 5.15825069379e-02 5.77463557561e-02 5.15825069379e-02 + 5.01876788897e-02 4.34096597635e-02 2.99175972602e-02 2.21871186712e-02 1.56634321975e-02 1.25541732198e-02 + 1.25541732198e-02 1.56634321975e-02 2.21871186712e-02 + 2.99175972602e-02 4.02211086967e-02 4.30758146535e-02 3.92509115207e-02 3.92509115207e-02 4.30758146535e-02 + 4.02211086967e-02 2.99175972602e-02 1.95082540137e-02 1.13664842271e-02 6.70391326813e-03 4.59226684865e-03 + 6.70391326813e-03 1.13664842271e-02 1.95082540137e-02 + 3.18111513620e-02 3.78205596100e-02 3.16546517690e-02 2.77593740924e-02 3.16546517690e-02 3.78205596100e-02 + 3.18111513620e-02 2.21871186712e-02 1.13664842271e-02 5.12121388847e-03 1.60728955735e-03 1.60728955735e-03 + 5.12121388847e-03 1.13664842271e-02 2.21871186712e-02 + 3.72988193747e-02 3.42116560673e-02 2.56404718355e-02 2.56404718355e-02 3.42116560673e-02 3.72988193747e-02 + 2.77285509468e-02 1.56634321975e-02 6.70391326813e-03 1.60728955735e-03 2.99424480887e-04 1.60728955735e-03 + 6.70391326813e-03 1.56634321975e-02 2.77285509468e-02 + 4.77333687537e-02 3.50377196433e-02 2.76230644258e-02 3.50377196433e-02 4.77333687537e-02 3.85963986077e-02 + 2.42831208041e-02 1.25541732198e-02 4.59226684865e-03 1.60728955735e-03 1.60728955735e-03 4.59226684865e-03 + 1.25541732198e-02 2.42831208041e-02 3.85963986077e-02 + 6.52918175640e-02 5.04046005791e-02 5.04046005791e-02 6.52918175640e-02 5.61479157062e-02 3.97912967555e-02 + 2.38655503495e-02 1.25541732198e-02 6.70391326813e-03 5.12121388847e-03 6.70391326813e-03 1.25541732198e-02 + 2.38655503495e-02 3.97912967555e-02 5.61479157062e-02 + 1.16509120681e-01 1.02615488905e-01 1.16509120681e-01 9.95704232240e-02 6.46951081959e-02 3.97912967555e-02 + 2.42831208041e-02 1.56634321975e-02 1.13664842271e-02 1.13664842271e-02 1.56634321975e-02 2.42831208041e-02 + 3.97912967555e-02 6.46951081959e-02 9.95704232240e-02 + 1.82505675344e-01 1.82505675344e-01 1.76044347723e-01 9.95704232240e-02 5.61479157062e-02 3.85963986077e-02 + 2.77285509468e-02 2.21871186712e-02 1.95082540137e-02 2.21871186712e-02 2.77285509468e-02 3.85963986077e-02 + 5.61479157062e-02 9.95704232240e-02 1.76044347723e-01 + 1.82505675344e-01 1.02615488905e-01 5.04046005791e-02 3.50377196433e-02 3.42116560673e-02 3.78205596100e-02 + 4.02211086967e-02 4.34096597635e-02 4.02211086967e-02 3.78205596100e-02 3.42116560673e-02 3.50377196433e-02 + 5.04046005791e-02 1.02615488905e-01 1.82505675344e-01 + 1.02615488905e-01 3.75414545463e-02 1.55211868757e-02 1.76462182571e-02 2.80273245508e-02 4.32610311780e-02 + 5.27172717907e-02 5.27172717907e-02 4.32610311780e-02 2.80273245508e-02 1.76462182571e-02 1.55211868757e-02 + 3.75414545463e-02 1.02615488905e-01 1.83341145810e-01 + 5.04046005791e-02 1.55211868757e-02 7.66628780847e-03 1.45718308971e-02 3.18367211069e-02 5.41592621661e-02 + 7.22668367439e-02 5.41592621661e-02 3.18367211069e-02 1.45718308971e-02 7.66628780847e-03 1.55211868757e-02 + 5.04046005791e-02 1.13574350192e-01 1.13574350192e-01 + 3.50377196433e-02 1.76462182571e-02 1.45718308971e-02 2.47874578675e-02 5.21829490033e-02 8.89831281823e-02 + 8.89831281823e-02 5.21829490033e-02 2.47874578675e-02 1.45718308971e-02 1.76462182571e-02 3.50377196433e-02 + 5.92333053982e-02 8.15738219549e-02 5.92333053982e-02 + 3.42116560673e-02 2.80273245508e-02 3.18367211069e-02 5.21829490033e-02 9.62641003175e-02 1.28838821866e-01 + 9.62641003175e-02 5.21829490033e-02 3.18367211069e-02 2.80273245508e-02 3.42116560673e-02 4.46752712878e-02 + 5.24033937689e-02 5.24033937689e-02 4.46752712878e-02 + 3.78205596100e-02 4.32610311780e-02 5.41592621661e-02 8.89831281823e-02 1.28838821866e-01 1.28838821866e-01 + 8.89831281823e-02 5.41592621661e-02 4.32610311780e-02 3.78205596100e-02 3.60058531033e-02 3.49564497515e-02 + 3.71064909955e-02 3.49564497515e-02 3.60058531033e-02 + 4.02211086967e-02 5.27172717907e-02 7.22668367439e-02 8.89831281823e-02 9.62641003175e-02 8.89831281823e-02 + 7.22668367439e-02 5.27172717907e-02 4.02211086967e-02 3.08620524079e-02 2.43906523476e-02 2.13924434679e-02 + 2.13924434679e-02 2.43906523476e-02 3.08620524079e-02 + 4.34096597635e-02 5.27172717907e-02 5.41592621661e-02 5.21829490033e-02 5.21829490033e-02 5.41592621661e-02 + 5.27172717907e-02 4.34096597635e-02 2.79664526175e-02 1.78999426265e-02 1.23767071708e-02 9.66580695579e-03 + 1.23767071708e-02 1.78999426265e-02 2.79664526175e-02 + 4.02211086967e-02 4.32610311780e-02 3.18367211069e-02 2.47874578675e-02 3.18367211069e-02 4.32610311780e-02 + 4.02211086967e-02 2.79664526175e-02 1.52960780652e-02 7.92572136635e-03 3.86919526274e-03 3.86919526274e-03 + 7.92572136635e-03 1.52960780652e-02 2.79664526175e-02 + 3.78205596100e-02 2.80273245508e-02 1.45718308971e-02 1.45718308971e-02 2.80273245508e-02 3.78205596100e-02 + 3.08620524079e-02 1.78999426265e-02 7.92572136635e-03 2.15903301226e-03 6.20357753249e-04 2.15903301226e-03 + 7.92572136635e-03 1.78999426265e-02 3.08620524079e-02 + 3.42116560673e-02 1.76462182571e-02 7.66628780847e-03 1.76462182571e-02 3.42116560673e-02 3.60058531033e-02 + 2.43906523476e-02 1.23767071708e-02 3.86919526274e-03 6.20357753249e-04 6.20357753249e-04 3.86919526274e-03 + 1.23767071708e-02 2.43906523476e-02 3.60058531033e-02 + 3.50377196433e-02 1.55211868757e-02 1.55211868757e-02 3.50377196433e-02 4.46752712878e-02 3.49564497515e-02 + 2.13924434679e-02 9.66580695579e-03 3.86919526274e-03 2.15903301226e-03 3.86919526274e-03 9.66580695579e-03 + 2.13924434679e-02 3.49564497515e-02 4.46752712878e-02 + 5.04046005791e-02 3.75414545463e-02 5.04046005791e-02 5.92333053982e-02 5.24033937689e-02 3.71064909955e-02 + 2.13924434679e-02 1.23767071708e-02 7.92572136635e-03 7.92572136635e-03 1.23767071708e-02 2.13924434679e-02 + 3.71064909955e-02 5.24033937689e-02 5.92333053982e-02 + 1.02615488905e-01 1.02615488905e-01 1.13574350192e-01 8.15738219549e-02 5.24033937689e-02 3.49564497515e-02 + 2.43906523476e-02 1.78999426265e-02 1.52960780652e-02 1.78999426265e-02 2.43906523476e-02 3.49564497515e-02 + 5.24033937689e-02 8.15738219549e-02 1.13574350192e-01 + 1.82505675344e-01 1.83341145810e-01 1.13574350192e-01 5.92333053982e-02 4.46752712878e-02 3.60058531033e-02 + 3.08620524079e-02 2.79664526175e-02 2.79664526175e-02 3.08620524079e-02 3.60058531033e-02 4.46752712878e-02 + 5.92333053982e-02 1.13574350192e-01 1.83341145810e-01 + 1.16509120681e-01 5.04046005791e-02 2.76230644258e-02 2.56404718355e-02 3.16546517690e-02 4.30758146535e-02 + 5.01876788897e-02 5.01876788897e-02 4.30758146535e-02 3.16546517690e-02 2.56404718355e-02 2.76230644258e-02 + 5.04046005791e-02 1.16509120681e-01 1.76044347723e-01 + 5.04046005791e-02 1.55211868757e-02 7.66628780847e-03 1.45718308971e-02 3.18367211069e-02 5.41592621661e-02 + 7.22668367439e-02 5.41592621661e-02 3.18367211069e-02 1.45718308971e-02 7.66628780847e-03 1.55211868757e-02 + 5.04046005791e-02 1.13574350192e-01 1.13574350192e-01 + 2.76230644258e-02 7.66628780847e-03 5.85084407671e-04 1.22185262764e-02 4.60124070413e-02 9.32039362341e-02 + 9.32039362341e-02 4.60124070413e-02 1.22185262764e-02 5.85084407671e-04 7.66628780847e-03 2.76230644258e-02 + 6.09545305143e-02 8.53986300889e-02 6.09545305143e-02 + 2.56404718355e-02 1.45718308971e-02 1.22185262764e-02 3.35905473089e-02 9.58263801583e-02 1.55844281554e-01 + 9.58263801583e-02 3.35905473089e-02 1.22185262764e-02 1.45718308971e-02 2.56404718355e-02 4.40402072259e-02 + 5.84841221586e-02 5.84841221586e-02 4.40402072259e-02 + 3.16546517690e-02 3.18367211069e-02 4.60124070413e-02 9.58263801583e-02 1.78863090053e-01 1.78863090053e-01 + 9.58263801583e-02 4.60124070413e-02 3.18367211069e-02 3.16546517690e-02 3.72214469881e-02 4.27264005082e-02 + 4.57183984058e-02 4.27264005082e-02 3.72214469881e-02 + 4.30758146535e-02 5.41592621661e-02 9.32039362341e-02 1.55844281554e-01 1.78863090053e-01 1.55844281554e-01 + 9.32039362341e-02 5.41592621661e-02 4.30758146535e-02 3.54156347447e-02 3.30827615217e-02 3.00038356153e-02 + 3.00038356153e-02 3.30827615217e-02 3.54156347447e-02 + 5.01876788897e-02 7.22668367439e-02 9.32039362341e-02 9.58263801583e-02 9.58263801583e-02 9.32039362341e-02 + 7.22668367439e-02 5.01876788897e-02 3.49273913093e-02 2.55137286753e-02 1.86408380774e-02 1.63178959213e-02 + 1.86408380774e-02 2.55137286753e-02 3.49273913093e-02 + 5.01876788897e-02 5.41592621661e-02 4.60124070413e-02 3.35905473089e-02 4.60124070413e-02 5.41592621661e-02 + 5.01876788897e-02 3.55022456672e-02 2.14983233332e-02 1.24891306807e-02 8.27428669466e-03 8.27428669466e-03 + 1.24891306807e-02 2.14983233332e-02 3.55022456672e-02 + 4.30758146535e-02 3.18367211069e-02 1.22185262764e-02 1.22185262764e-02 3.18367211069e-02 4.30758146535e-02 + 3.49273913093e-02 2.14983233332e-02 9.78995536003e-03 4.02221833331e-03 2.15070114075e-03 4.02221833331e-03 + 9.78995536003e-03 2.14983233332e-02 3.49273913093e-02 + 3.16546517690e-02 1.45718308971e-02 5.85084407671e-04 1.45718308971e-02 3.16546517690e-02 3.54156347447e-02 + 2.55137286753e-02 1.24891306807e-02 4.02221833331e-03 5.43062651502e-04 5.43062651502e-04 4.02221833331e-03 + 1.24891306807e-02 2.55137286753e-02 3.54156347447e-02 + 2.56404718355e-02 7.66628780847e-03 7.66628780847e-03 2.56404718355e-02 3.72214469881e-02 3.30827615217e-02 + 1.86408380774e-02 8.27428669466e-03 2.15070114075e-03 5.43062651502e-04 2.15070114075e-03 8.27428669466e-03 + 1.86408380774e-02 3.30827615217e-02 3.72214469881e-02 + 2.76230644258e-02 1.55211868757e-02 2.76230644258e-02 4.40402072259e-02 4.27264005082e-02 3.00038356153e-02 + 1.63178959213e-02 8.27428669466e-03 4.02221833331e-03 4.02221833331e-03 8.27428669466e-03 1.63178959213e-02 + 3.00038356153e-02 4.27264005082e-02 4.40402072259e-02 + 5.04046005791e-02 5.04046005791e-02 6.09545305143e-02 5.84841221586e-02 4.57183984058e-02 3.00038356153e-02 + 1.86408380774e-02 1.24891306807e-02 9.78995536003e-03 1.24891306807e-02 1.86408380774e-02 3.00038356153e-02 + 4.57183984058e-02 5.84841221586e-02 6.09545305143e-02 + 1.16509120681e-01 1.13574350192e-01 8.53986300889e-02 5.84841221586e-02 4.27264005082e-02 3.30827615217e-02 + 2.55137286753e-02 2.14983233332e-02 2.14983233332e-02 2.55137286753e-02 3.30827615217e-02 4.27264005082e-02 + 5.84841221586e-02 8.53986300889e-02 1.13574350192e-01 + 1.76044347723e-01 1.13574350192e-01 6.09545305143e-02 4.40402072259e-02 3.72214469881e-02 3.54156347447e-02 + 3.49273913093e-02 3.55022456672e-02 3.49273913093e-02 3.54156347447e-02 3.72214469881e-02 4.40402072259e-02 + 6.09545305143e-02 1.13574350192e-01 1.76044347723e-01 + 6.52918175640e-02 3.50377196433e-02 2.56404718355e-02 2.77593740924e-02 3.92509115207e-02 5.15825069379e-02 + 6.40667422061e-02 5.15825069379e-02 3.92509115207e-02 2.77593740924e-02 2.56404718355e-02 3.50377196433e-02 + 6.52918175640e-02 9.95704232240e-02 9.95704232240e-02 + 3.50377196433e-02 1.76462182571e-02 1.45718308971e-02 2.47874578675e-02 5.21829490033e-02 8.89831281823e-02 + 8.89831281823e-02 5.21829490033e-02 2.47874578675e-02 1.45718308971e-02 1.76462182571e-02 3.50377196433e-02 + 5.92333053982e-02 8.15738219549e-02 5.92333053982e-02 + 2.56404718355e-02 1.45718308971e-02 1.22185262764e-02 3.35905473089e-02 9.58263801583e-02 1.55844281554e-01 + 9.58263801583e-02 3.35905473089e-02 1.22185262764e-02 1.45718308971e-02 2.56404718355e-02 4.40402072259e-02 + 5.84841221586e-02 5.84841221586e-02 4.40402072259e-02 + 2.77593740924e-02 2.47874578675e-02 3.35905473089e-02 6.76329451587e-02 1.65042874919e-01 1.65042874919e-01 + 6.76329451587e-02 3.35905473089e-02 2.47874578675e-02 2.77593740924e-02 3.74638848543e-02 4.48521019830e-02 + 5.02668272913e-02 4.48521019830e-02 3.74638848543e-02 + 3.92509115207e-02 5.21829490033e-02 9.58263801583e-02 1.65042874919e-01 1.94546783494e-01 1.65042874919e-01 + 9.58263801583e-02 5.21829490033e-02 3.92509115207e-02 3.70537599473e-02 3.66424673274e-02 3.62877199395e-02 + 3.62877199395e-02 3.66424673274e-02 3.70537599473e-02 + 5.15825069379e-02 8.89831281823e-02 1.55844281554e-01 1.65042874919e-01 1.65042874919e-01 1.55844281554e-01 + 8.89831281823e-02 5.15825069379e-02 3.88117173565e-02 3.00285261604e-02 2.47911036584e-02 2.24936844433e-02 + 2.47911036584e-02 3.00285261604e-02 3.88117173565e-02 + 6.40667422061e-02 8.89831281823e-02 9.58263801583e-02 6.76329451587e-02 9.58263801583e-02 8.89831281823e-02 + 6.40667422061e-02 4.06521018269e-02 2.56714935438e-02 1.73117518289e-02 1.28051588719e-02 1.28051588719e-02 + 1.73117518289e-02 2.56714935438e-02 4.06521018269e-02 + 5.15825069379e-02 5.21829490033e-02 3.35905473089e-02 3.35905473089e-02 5.21829490033e-02 5.15825069379e-02 + 4.06521018269e-02 2.60511627292e-02 1.40065686484e-02 7.35300020881e-03 5.53109046735e-03 7.35300020881e-03 + 1.40065686484e-02 2.60511627292e-02 4.06521018269e-02 + 3.92509115207e-02 2.47874578675e-02 1.22185262764e-02 2.47874578675e-02 3.92509115207e-02 3.88117173565e-02 + 2.56714935438e-02 1.40065686484e-02 5.18167581107e-03 1.64060165739e-03 1.64060165739e-03 5.18167581107e-03 + 1.40065686484e-02 2.56714935438e-02 3.88117173565e-02 + 2.77593740924e-02 1.45718308971e-02 1.45718308971e-02 2.77593740924e-02 3.70537599473e-02 3.00285261604e-02 + 1.73117518289e-02 7.35300020881e-03 1.64060165739e-03 1.47427756958e-04 1.64060165739e-03 7.35300020881e-03 + 1.73117518289e-02 3.00285261604e-02 3.70537599473e-02 + 2.56404718355e-02 1.76462182571e-02 2.56404718355e-02 3.74638848543e-02 3.66424673274e-02 2.47911036584e-02 + 1.28051588719e-02 5.53109046735e-03 1.64060165739e-03 1.64060165739e-03 5.53109046735e-03 1.28051588719e-02 + 2.47911036584e-02 3.66424673274e-02 3.74638848543e-02 + 3.50377196433e-02 3.50377196433e-02 4.40402072259e-02 4.48521019830e-02 3.62877199395e-02 2.24936844433e-02 + 1.28051588719e-02 7.35300020881e-03 5.18167581107e-03 7.35300020881e-03 1.28051588719e-02 2.24936844433e-02 + 3.62877199395e-02 4.48521019830e-02 4.40402072259e-02 + 6.52918175640e-02 5.92333053982e-02 5.84841221586e-02 5.02668272913e-02 3.62877199395e-02 2.47911036584e-02 + 1.73117518289e-02 1.40065686484e-02 1.40065686484e-02 1.73117518289e-02 2.47911036584e-02 3.62877199395e-02 + 5.02668272913e-02 5.84841221586e-02 5.92333053982e-02 + 9.95704232240e-02 8.15738219549e-02 5.84841221586e-02 4.48521019830e-02 3.66424673274e-02 3.00285261604e-02 + 2.56714935438e-02 2.60511627292e-02 2.56714935438e-02 3.00285261604e-02 3.66424673274e-02 4.48521019830e-02 + 5.84841221586e-02 8.15738219549e-02 9.95704232240e-02 + 9.95704232240e-02 5.92333053982e-02 4.40402072259e-02 3.74638848543e-02 3.70537599473e-02 3.88117173565e-02 + 4.06521018269e-02 4.06521018269e-02 3.88117173565e-02 3.70537599473e-02 3.74638848543e-02 4.40402072259e-02 + 5.92333053982e-02 9.95704232240e-02 1.21925608243e-01 + 4.77333687537e-02 3.42116560673e-02 3.16546517690e-02 3.92509115207e-02 5.77463557561e-02 7.23072529491e-02 + 7.23072529491e-02 5.77463557561e-02 3.92509115207e-02 3.16546517690e-02 3.42116560673e-02 4.77333687537e-02 + 5.61479157062e-02 6.46951081959e-02 5.61479157062e-02 + 3.42116560673e-02 2.80273245508e-02 3.18367211069e-02 5.21829490033e-02 9.62641003175e-02 1.28838821866e-01 + 9.62641003175e-02 5.21829490033e-02 3.18367211069e-02 2.80273245508e-02 3.42116560673e-02 4.46752712878e-02 + 5.24033937689e-02 5.24033937689e-02 4.46752712878e-02 + 3.16546517690e-02 3.18367211069e-02 4.60124070413e-02 9.58263801583e-02 1.78863090053e-01 1.78863090053e-01 + 9.58263801583e-02 4.60124070413e-02 3.18367211069e-02 3.16546517690e-02 3.72214469881e-02 4.27264005082e-02 + 4.57183984058e-02 4.27264005082e-02 3.72214469881e-02 + 3.92509115207e-02 5.21829490033e-02 9.58263801583e-02 1.65042874919e-01 1.94546783494e-01 1.65042874919e-01 + 9.58263801583e-02 5.21829490033e-02 3.92509115207e-02 3.70537599473e-02 3.66424673274e-02 3.62877199395e-02 + 3.62877199395e-02 3.66424673274e-02 3.70537599473e-02 + 5.77463557561e-02 9.62641003175e-02 1.78863090053e-01 1.94546783494e-01 1.94546783494e-01 1.78863090053e-01 + 9.62641003175e-02 5.77463557561e-02 4.31600648738e-02 3.18835940632e-02 2.71116363186e-02 2.45878312145e-02 + 2.71116363186e-02 3.18835940632e-02 4.31600648738e-02 + 7.23072529491e-02 1.28838821866e-01 1.78863090053e-01 1.65042874919e-01 1.78863090053e-01 1.28838821866e-01 + 7.23072529491e-02 4.49452510339e-02 2.94652595219e-02 2.09809548480e-02 1.67238267603e-02 1.67238267603e-02 + 2.09809548480e-02 2.94652595219e-02 4.49452510339e-02 + 7.23072529491e-02 9.62641003175e-02 9.58263801583e-02 9.58263801583e-02 9.62641003175e-02 7.23072529491e-02 + 4.78489705763e-02 2.78780404497e-02 1.64122236650e-02 1.00173301250e-02 8.33655930782e-03 1.00173301250e-02 + 1.64122236650e-02 2.78780404497e-02 4.78489705763e-02 + 5.77463557561e-02 5.21829490033e-02 4.60124070413e-02 5.21829490033e-02 5.77463557561e-02 4.49452510339e-02 + 2.78780404497e-02 1.43711514656e-02 7.08705078074e-03 3.65189274881e-03 3.65189274881e-03 7.08705078074e-03 + 1.43711514656e-02 2.78780404497e-02 4.49452510339e-02 + 3.92509115207e-02 3.18367211069e-02 3.18367211069e-02 3.92509115207e-02 4.31600648738e-02 2.94652595219e-02 + 1.64122236650e-02 7.08705078074e-03 2.06207427592e-03 7.48839342246e-04 2.06207427592e-03 7.08705078074e-03 + 1.64122236650e-02 2.94652595219e-02 4.31600648738e-02 + 3.16546517690e-02 2.80273245508e-02 3.16546517690e-02 3.70537599473e-02 3.18835940632e-02 2.09809548480e-02 + 1.00173301250e-02 3.65189274881e-03 7.48839342246e-04 7.48839342246e-04 3.65189274881e-03 1.00173301250e-02 + 2.09809548480e-02 3.18835940632e-02 3.70537599473e-02 + 3.42116560673e-02 3.42116560673e-02 3.72214469881e-02 3.66424673274e-02 2.71116363186e-02 1.67238267603e-02 + 8.33655930782e-03 3.65189274881e-03 2.06207427592e-03 3.65189274881e-03 8.33655930782e-03 1.67238267603e-02 + 2.71116363186e-02 3.66424673274e-02 3.72214469881e-02 + 4.77333687537e-02 4.46752712878e-02 4.27264005082e-02 3.62877199395e-02 2.45878312145e-02 1.67238267603e-02 + 1.00173301250e-02 7.08705078074e-03 7.08705078074e-03 1.00173301250e-02 1.67238267603e-02 2.45878312145e-02 + 3.62877199395e-02 4.27264005082e-02 4.46752712878e-02 + 5.61479157062e-02 5.24033937689e-02 4.57183984058e-02 3.62877199395e-02 2.71116363186e-02 2.09809548480e-02 + 1.64122236650e-02 1.43711514656e-02 1.64122236650e-02 2.09809548480e-02 2.71116363186e-02 3.62877199395e-02 + 4.57183984058e-02 5.24033937689e-02 5.61479157062e-02 + 6.46951081959e-02 5.24033937689e-02 4.27264005082e-02 3.66424673274e-02 3.18835940632e-02 2.94652595219e-02 + 2.78780404497e-02 2.78780404497e-02 2.94652595219e-02 3.18835940632e-02 3.66424673274e-02 4.27264005082e-02 + 5.24033937689e-02 6.46951081959e-02 7.10938675575e-02 + 5.61479157062e-02 4.46752712878e-02 3.72214469881e-02 3.70537599473e-02 4.31600648738e-02 4.49452510339e-02 + 4.78489705763e-02 4.49452510339e-02 4.31600648738e-02 3.70537599473e-02 3.72214469881e-02 4.46752712878e-02 + 5.61479157062e-02 7.10938675575e-02 7.10938675575e-02 + 3.72988193747e-02 3.78205596100e-02 4.30758146535e-02 5.15825069379e-02 7.23072529491e-02 8.48962030063e-02 + 7.23072529491e-02 5.15825069379e-02 4.30758146535e-02 3.78205596100e-02 3.72988193747e-02 3.85963986077e-02 + 3.97912967555e-02 3.97912967555e-02 3.85963986077e-02 + 3.78205596100e-02 4.32610311780e-02 5.41592621661e-02 8.89831281823e-02 1.28838821866e-01 1.28838821866e-01 + 8.89831281823e-02 5.41592621661e-02 4.32610311780e-02 3.78205596100e-02 3.60058531033e-02 3.49564497515e-02 + 3.71064909955e-02 3.49564497515e-02 3.60058531033e-02 + 4.30758146535e-02 5.41592621661e-02 9.32039362341e-02 1.55844281554e-01 1.78863090053e-01 1.55844281554e-01 + 9.32039362341e-02 5.41592621661e-02 4.30758146535e-02 3.54156347447e-02 3.30827615217e-02 3.00038356153e-02 + 3.00038356153e-02 3.30827615217e-02 3.54156347447e-02 + 5.15825069379e-02 8.89831281823e-02 1.55844281554e-01 1.65042874919e-01 1.65042874919e-01 1.55844281554e-01 + 8.89831281823e-02 5.15825069379e-02 3.88117173565e-02 3.00285261604e-02 2.47911036584e-02 2.24936844433e-02 + 2.47911036584e-02 3.00285261604e-02 3.88117173565e-02 + 7.23072529491e-02 1.28838821866e-01 1.78863090053e-01 1.65042874919e-01 1.78863090053e-01 1.28838821866e-01 + 7.23072529491e-02 4.49452510339e-02 2.94652595219e-02 2.09809548480e-02 1.67238267603e-02 1.67238267603e-02 + 2.09809548480e-02 2.94652595219e-02 4.49452510339e-02 + 8.48962030063e-02 1.28838821866e-01 1.55844281554e-01 1.55844281554e-01 1.28838821866e-01 8.48962030063e-02 + 5.02796892510e-02 3.01994514047e-02 1.68913524358e-02 1.09125507012e-02 9.07231580734e-03 1.09125507012e-02 + 1.68913524358e-02 3.01994514047e-02 5.02796892510e-02 + 7.23072529491e-02 8.89831281823e-02 9.32039362341e-02 8.89831281823e-02 7.23072529491e-02 5.02796892510e-02 + 2.96935773932e-02 1.56673079604e-02 8.95764366269e-03 5.26425072315e-03 5.26425072315e-03 8.95764366269e-03 + 1.56673079604e-02 2.96935773932e-02 5.02796892510e-02 + 5.15825069379e-02 5.41592621661e-02 5.41592621661e-02 5.15825069379e-02 4.49452510339e-02 3.01994514047e-02 + 1.56673079604e-02 7.32181384452e-03 3.29204038761e-03 2.09399363094e-03 3.29204038761e-03 7.32181384452e-03 + 1.56673079604e-02 3.01994514047e-02 4.49452510339e-02 + 4.30758146535e-02 4.32610311780e-02 4.30758146535e-02 3.88117173565e-02 2.94652595219e-02 1.68913524358e-02 + 8.95764366269e-03 3.29204038761e-03 1.38575808115e-03 1.38575808115e-03 3.29204038761e-03 8.95764366269e-03 + 1.68913524358e-02 2.94652595219e-02 3.88117173565e-02 + 3.78205596100e-02 3.78205596100e-02 3.54156347447e-02 3.00285261604e-02 2.09809548480e-02 1.09125507012e-02 + 5.26425072315e-03 2.09399363094e-03 1.38575808115e-03 2.09399363094e-03 5.26425072315e-03 1.09125507012e-02 + 2.09809548480e-02 3.00285261604e-02 3.54156347447e-02 + 3.72988193747e-02 3.60058531033e-02 3.30827615217e-02 2.47911036584e-02 1.67238267603e-02 9.07231580734e-03 + 5.26425072315e-03 3.29204038761e-03 3.29204038761e-03 5.26425072315e-03 9.07231580734e-03 1.67238267603e-02 + 2.47911036584e-02 3.30827615217e-02 3.60058531033e-02 + 3.85963986077e-02 3.49564497515e-02 3.00038356153e-02 2.24936844433e-02 1.67238267603e-02 1.09125507012e-02 + 8.95764366269e-03 7.32181384452e-03 8.95764366269e-03 1.09125507012e-02 1.67238267603e-02 2.24936844433e-02 + 3.00038356153e-02 3.49564497515e-02 3.85963986077e-02 + 3.97912967555e-02 3.71064909955e-02 3.00038356153e-02 2.47911036584e-02 2.09809548480e-02 1.68913524358e-02 + 1.56673079604e-02 1.56673079604e-02 1.68913524358e-02 2.09809548480e-02 2.47911036584e-02 3.00038356153e-02 + 3.71064909955e-02 3.97912967555e-02 4.24418182390e-02 + 3.97912967555e-02 3.49564497515e-02 3.30827615217e-02 3.00285261604e-02 2.94652595219e-02 3.01994514047e-02 + 2.96935773932e-02 3.01994514047e-02 2.94652595219e-02 3.00285261604e-02 3.30827615217e-02 3.49564497515e-02 + 3.97912967555e-02 4.45753367440e-02 4.45753367440e-02 + 3.85963986077e-02 3.60058531033e-02 3.54156347447e-02 3.88117173565e-02 4.49452510339e-02 5.02796892510e-02 + 5.02796892510e-02 4.49452510339e-02 3.88117173565e-02 3.54156347447e-02 3.60058531033e-02 3.85963986077e-02 + 4.24418182390e-02 4.45753367440e-02 4.24418182390e-02 + 3.18111513620e-02 4.02211086967e-02 5.01876788897e-02 6.40667422061e-02 7.23072529491e-02 7.23072529491e-02 + 6.40667422061e-02 5.01876788897e-02 4.02211086967e-02 3.18111513620e-02 2.77285509468e-02 2.42831208041e-02 + 2.38655503495e-02 2.42831208041e-02 2.77285509468e-02 + 4.02211086967e-02 5.27172717907e-02 7.22668367439e-02 8.89831281823e-02 9.62641003175e-02 8.89831281823e-02 + 7.22668367439e-02 5.27172717907e-02 4.02211086967e-02 3.08620524079e-02 2.43906523476e-02 2.13924434679e-02 + 2.13924434679e-02 2.43906523476e-02 3.08620524079e-02 + 5.01876788897e-02 7.22668367439e-02 9.32039362341e-02 9.58263801583e-02 9.58263801583e-02 9.32039362341e-02 + 7.22668367439e-02 5.01876788897e-02 3.49273913093e-02 2.55137286753e-02 1.86408380774e-02 1.63178959213e-02 + 1.86408380774e-02 2.55137286753e-02 3.49273913093e-02 + 6.40667422061e-02 8.89831281823e-02 9.58263801583e-02 6.76329451587e-02 9.58263801583e-02 8.89831281823e-02 + 6.40667422061e-02 4.06521018269e-02 2.56714935438e-02 1.73117518289e-02 1.28051588719e-02 1.28051588719e-02 + 1.73117518289e-02 2.56714935438e-02 4.06521018269e-02 + 7.23072529491e-02 9.62641003175e-02 9.58263801583e-02 9.58263801583e-02 9.62641003175e-02 7.23072529491e-02 + 4.78489705763e-02 2.78780404497e-02 1.64122236650e-02 1.00173301250e-02 8.33655930782e-03 1.00173301250e-02 + 1.64122236650e-02 2.78780404497e-02 4.78489705763e-02 + 7.23072529491e-02 8.89831281823e-02 9.32039362341e-02 8.89831281823e-02 7.23072529491e-02 5.02796892510e-02 + 2.96935773932e-02 1.56673079604e-02 8.95764366269e-03 5.26425072315e-03 5.26425072315e-03 8.95764366269e-03 + 1.56673079604e-02 2.96935773932e-02 5.02796892510e-02 + 6.40667422061e-02 7.22668367439e-02 7.22668367439e-02 6.40667422061e-02 4.78489705763e-02 2.96935773932e-02 + 1.53593248116e-02 7.66263937498e-03 3.60185781299e-03 2.68892841248e-03 3.60185781299e-03 7.66263937498e-03 + 1.53593248116e-02 2.96935773932e-02 4.78489705763e-02 + 5.01876788897e-02 5.27172717907e-02 5.01876788897e-02 4.06521018269e-02 2.78780404497e-02 1.56673079604e-02 + 7.66263937498e-03 2.63545476236e-03 1.11117793737e-03 1.11117793737e-03 2.63545476236e-03 7.66263937498e-03 + 1.56673079604e-02 2.78780404497e-02 4.06521018269e-02 + 4.02211086967e-02 4.02211086967e-02 3.49273913093e-02 2.56714935438e-02 1.64122236650e-02 8.95764366269e-03 + 3.60185781299e-03 1.11117793737e-03 5.28711121037e-04 1.11117793737e-03 3.60185781299e-03 8.95764366269e-03 + 1.64122236650e-02 2.56714935438e-02 3.49273913093e-02 + 3.18111513620e-02 3.08620524079e-02 2.55137286753e-02 1.73117518289e-02 1.00173301250e-02 5.26425072315e-03 + 2.68892841248e-03 1.11117793737e-03 1.11117793737e-03 2.68892841248e-03 5.26425072315e-03 1.00173301250e-02 + 1.73117518289e-02 2.55137286753e-02 3.08620524079e-02 + 2.77285509468e-02 2.43906523476e-02 1.86408380774e-02 1.28051588719e-02 8.33655930782e-03 5.26425072315e-03 + 3.60185781299e-03 2.63545476236e-03 3.60185781299e-03 5.26425072315e-03 8.33655930782e-03 1.28051588719e-02 + 1.86408380774e-02 2.43906523476e-02 2.77285509468e-02 + 2.42831208041e-02 2.13924434679e-02 1.63178959213e-02 1.28051588719e-02 1.00173301250e-02 8.95764366269e-03 + 7.66263937498e-03 7.66263937498e-03 8.95764366269e-03 1.00173301250e-02 1.28051588719e-02 1.63178959213e-02 + 2.13924434679e-02 2.42831208041e-02 2.61711989421e-02 + 2.38655503495e-02 2.13924434679e-02 1.86408380774e-02 1.73117518289e-02 1.64122236650e-02 1.56673079604e-02 + 1.53593248116e-02 1.56673079604e-02 1.64122236650e-02 1.73117518289e-02 1.86408380774e-02 2.13924434679e-02 + 2.38655503495e-02 2.49947177389e-02 2.49947177389e-02 + 2.42831208041e-02 2.43906523476e-02 2.55137286753e-02 2.56714935438e-02 2.78780404497e-02 2.96935773932e-02 + 2.96935773932e-02 2.78780404497e-02 2.56714935438e-02 2.55137286753e-02 2.43906523476e-02 2.42831208041e-02 + 2.49947177389e-02 2.65333628877e-02 2.49947177389e-02 + 2.77285509468e-02 3.08620524079e-02 3.49273913093e-02 4.06521018269e-02 4.78489705763e-02 5.02796892510e-02 + 4.78489705763e-02 4.06521018269e-02 3.49273913093e-02 3.08620524079e-02 2.77285509468e-02 2.61711989421e-02 + 2.49947177389e-02 2.49947177389e-02 2.61711989421e-02 + 2.99175972602e-02 4.34096597635e-02 5.01876788897e-02 5.15825069379e-02 5.77463557561e-02 5.15825069379e-02 + 5.01876788897e-02 4.34096597635e-02 2.99175972602e-02 2.21871186712e-02 1.56634321975e-02 1.25541732198e-02 + 1.25541732198e-02 1.56634321975e-02 2.21871186712e-02 + 4.34096597635e-02 5.27172717907e-02 5.41592621661e-02 5.21829490033e-02 5.21829490033e-02 5.41592621661e-02 + 5.27172717907e-02 4.34096597635e-02 2.79664526175e-02 1.78999426265e-02 1.23767071708e-02 9.66580695579e-03 + 1.23767071708e-02 1.78999426265e-02 2.79664526175e-02 + 5.01876788897e-02 5.41592621661e-02 4.60124070413e-02 3.35905473089e-02 4.60124070413e-02 5.41592621661e-02 + 5.01876788897e-02 3.55022456672e-02 2.14983233332e-02 1.24891306807e-02 8.27428669466e-03 8.27428669466e-03 + 1.24891306807e-02 2.14983233332e-02 3.55022456672e-02 + 5.15825069379e-02 5.21829490033e-02 3.35905473089e-02 3.35905473089e-02 5.21829490033e-02 5.15825069379e-02 + 4.06521018269e-02 2.60511627292e-02 1.40065686484e-02 7.35300020881e-03 5.53109046735e-03 7.35300020881e-03 + 1.40065686484e-02 2.60511627292e-02 4.06521018269e-02 + 5.77463557561e-02 5.21829490033e-02 4.60124070413e-02 5.21829490033e-02 5.77463557561e-02 4.49452510339e-02 + 2.78780404497e-02 1.43711514656e-02 7.08705078074e-03 3.65189274881e-03 3.65189274881e-03 7.08705078074e-03 + 1.43711514656e-02 2.78780404497e-02 4.49452510339e-02 + 5.15825069379e-02 5.41592621661e-02 5.41592621661e-02 5.15825069379e-02 4.49452510339e-02 3.01994514047e-02 + 1.56673079604e-02 7.32181384452e-03 3.29204038761e-03 2.09399363094e-03 3.29204038761e-03 7.32181384452e-03 + 1.56673079604e-02 3.01994514047e-02 4.49452510339e-02 + 5.01876788897e-02 5.27172717907e-02 5.01876788897e-02 4.06521018269e-02 2.78780404497e-02 1.56673079604e-02 + 7.66263937498e-03 2.63545476236e-03 1.11117793737e-03 1.11117793737e-03 2.63545476236e-03 7.66263937498e-03 + 1.56673079604e-02 2.78780404497e-02 4.06521018269e-02 + 4.34096597635e-02 4.34096597635e-02 3.55022456672e-02 2.60511627292e-02 1.43711514656e-02 7.32181384452e-03 + 2.63545476236e-03 9.45253549219e-04 3.23056152168e-04 9.45253549219e-04 2.63545476236e-03 7.32181384452e-03 + 1.43711514656e-02 2.60511627292e-02 3.55022456672e-02 + 2.99175972602e-02 2.79664526175e-02 2.14983233332e-02 1.40065686484e-02 7.08705078074e-03 3.29204038761e-03 + 1.11117793737e-03 3.23056152168e-04 3.23056152168e-04 1.11117793737e-03 3.29204038761e-03 7.08705078074e-03 + 1.40065686484e-02 2.14983233332e-02 2.79664526175e-02 + 2.21871186712e-02 1.78999426265e-02 1.24891306807e-02 7.35300020881e-03 3.65189274881e-03 2.09399363094e-03 + 1.11117793737e-03 9.45253549219e-04 1.11117793737e-03 2.09399363094e-03 3.65189274881e-03 7.35300020881e-03 + 1.24891306807e-02 1.78999426265e-02 2.21871186712e-02 + 1.56634321975e-02 1.23767071708e-02 8.27428669466e-03 5.53109046735e-03 3.65189274881e-03 3.29204038761e-03 + 2.63545476236e-03 2.63545476236e-03 3.29204038761e-03 3.65189274881e-03 5.53109046735e-03 8.27428669466e-03 + 1.23767071708e-02 1.56634321975e-02 1.79922853063e-02 + 1.25541732198e-02 9.66580695579e-03 8.27428669466e-03 7.35300020881e-03 7.08705078074e-03 7.32181384452e-03 + 7.66263937498e-03 7.32181384452e-03 7.08705078074e-03 7.35300020881e-03 8.27428669466e-03 9.66580695579e-03 + 1.25541732198e-02 1.49682875172e-02 1.49682875172e-02 + 1.25541732198e-02 1.23767071708e-02 1.24891306807e-02 1.40065686484e-02 1.43711514656e-02 1.56673079604e-02 + 1.56673079604e-02 1.43711514656e-02 1.40065686484e-02 1.24891306807e-02 1.23767071708e-02 1.25541732198e-02 + 1.36280371710e-02 1.31863090803e-02 1.36280371710e-02 + 1.56634321975e-02 1.78999426265e-02 2.14983233332e-02 2.60511627292e-02 2.78780404497e-02 3.01994514047e-02 + 2.78780404497e-02 2.60511627292e-02 2.14983233332e-02 1.78999426265e-02 1.56634321975e-02 1.49682875172e-02 + 1.31863090803e-02 1.31863090803e-02 1.49682875172e-02 + 2.21871186712e-02 2.79664526175e-02 3.55022456672e-02 4.06521018269e-02 4.49452510339e-02 4.49452510339e-02 + 4.06521018269e-02 3.55022456672e-02 2.79664526175e-02 2.21871186712e-02 1.79922853063e-02 1.49682875172e-02 + 1.36280371710e-02 1.49682875172e-02 1.79922853063e-02 + 2.99175972602e-02 4.02211086967e-02 4.30758146535e-02 3.92509115207e-02 3.92509115207e-02 4.30758146535e-02 + 4.02211086967e-02 2.99175972602e-02 1.95082540137e-02 1.13664842271e-02 6.70391326813e-03 4.59226684865e-03 + 6.70391326813e-03 1.13664842271e-02 1.95082540137e-02 + 4.02211086967e-02 4.32610311780e-02 3.18367211069e-02 2.47874578675e-02 3.18367211069e-02 4.32610311780e-02 + 4.02211086967e-02 2.79664526175e-02 1.52960780652e-02 7.92572136635e-03 3.86919526274e-03 3.86919526274e-03 + 7.92572136635e-03 1.52960780652e-02 2.79664526175e-02 + 4.30758146535e-02 3.18367211069e-02 1.22185262764e-02 1.22185262764e-02 3.18367211069e-02 4.30758146535e-02 + 3.49273913093e-02 2.14983233332e-02 9.78995536003e-03 4.02221833331e-03 2.15070114075e-03 4.02221833331e-03 + 9.78995536003e-03 2.14983233332e-02 3.49273913093e-02 + 3.92509115207e-02 2.47874578675e-02 1.22185262764e-02 2.47874578675e-02 3.92509115207e-02 3.88117173565e-02 + 2.56714935438e-02 1.40065686484e-02 5.18167581107e-03 1.64060165739e-03 1.64060165739e-03 5.18167581107e-03 + 1.40065686484e-02 2.56714935438e-02 3.88117173565e-02 + 3.92509115207e-02 3.18367211069e-02 3.18367211069e-02 3.92509115207e-02 4.31600648738e-02 2.94652595219e-02 + 1.64122236650e-02 7.08705078074e-03 2.06207427592e-03 7.48839342246e-04 2.06207427592e-03 7.08705078074e-03 + 1.64122236650e-02 2.94652595219e-02 4.31600648738e-02 + 4.30758146535e-02 4.32610311780e-02 4.30758146535e-02 3.88117173565e-02 2.94652595219e-02 1.68913524358e-02 + 8.95764366269e-03 3.29204038761e-03 1.38575808115e-03 1.38575808115e-03 3.29204038761e-03 8.95764366269e-03 + 1.68913524358e-02 2.94652595219e-02 3.88117173565e-02 + 4.02211086967e-02 4.02211086967e-02 3.49273913093e-02 2.56714935438e-02 1.64122236650e-02 8.95764366269e-03 + 3.60185781299e-03 1.11117793737e-03 5.28711121037e-04 1.11117793737e-03 3.60185781299e-03 8.95764366269e-03 + 1.64122236650e-02 2.56714935438e-02 3.49273913093e-02 + 2.99175972602e-02 2.79664526175e-02 2.14983233332e-02 1.40065686484e-02 7.08705078074e-03 3.29204038761e-03 + 1.11117793737e-03 3.23056152168e-04 3.23056152168e-04 1.11117793737e-03 3.29204038761e-03 7.08705078074e-03 + 1.40065686484e-02 2.14983233332e-02 2.79664526175e-02 + 1.95082540137e-02 1.52960780652e-02 9.78995536003e-03 5.18167581107e-03 2.06207427592e-03 1.38575808115e-03 + 5.28711121037e-04 3.23056152168e-04 5.28711121037e-04 1.38575808115e-03 2.06207427592e-03 5.18167581107e-03 + 9.78995536003e-03 1.52960780652e-02 1.95082540137e-02 + 1.13664842271e-02 7.92572136635e-03 4.02221833331e-03 1.64060165739e-03 7.48839342246e-04 1.38575808115e-03 + 1.11117793737e-03 1.11117793737e-03 1.38575808115e-03 7.48839342246e-04 1.64060165739e-03 4.02221833331e-03 + 7.92572136635e-03 1.13664842271e-02 1.40817702606e-02 + 6.70391326813e-03 3.86919526274e-03 2.15070114075e-03 1.64060165739e-03 2.06207427592e-03 3.29204038761e-03 + 3.60185781299e-03 3.29204038761e-03 2.06207427592e-03 1.64060165739e-03 2.15070114075e-03 3.86919526274e-03 + 6.70391326813e-03 8.64797657421e-03 8.64797657421e-03 + 4.59226684865e-03 3.86919526274e-03 4.02221833331e-03 5.18167581107e-03 7.08705078074e-03 8.95764366269e-03 + 8.95764366269e-03 7.08705078074e-03 5.18167581107e-03 4.02221833331e-03 3.86919526274e-03 4.59226684865e-03 + 6.75414542788e-03 6.84674742673e-03 6.75414542788e-03 + 6.70391326813e-03 7.92572136635e-03 9.78995536003e-03 1.40065686484e-02 1.64122236650e-02 1.68913524358e-02 + 1.64122236650e-02 1.40065686484e-02 9.78995536003e-03 7.92572136635e-03 6.70391326813e-03 6.75414542788e-03 + 5.80240507057e-03 5.80240507057e-03 6.75414542788e-03 + 1.13664842271e-02 1.52960780652e-02 2.14983233332e-02 2.56714935438e-02 2.94652595219e-02 2.94652595219e-02 + 2.56714935438e-02 2.14983233332e-02 1.52960780652e-02 1.13664842271e-02 8.64797657421e-03 6.84674742673e-03 + 5.80240507057e-03 6.84674742673e-03 8.64797657421e-03 + 1.95082540137e-02 2.79664526175e-02 3.49273913093e-02 3.88117173565e-02 4.31600648738e-02 3.88117173565e-02 + 3.49273913093e-02 2.79664526175e-02 1.95082540137e-02 1.40817702606e-02 8.64797657421e-03 6.75414542788e-03 + 6.75414542788e-03 8.64797657421e-03 1.40817702606e-02 + 3.18111513620e-02 3.78205596100e-02 3.16546517690e-02 2.77593740924e-02 3.16546517690e-02 3.78205596100e-02 + 3.18111513620e-02 2.21871186712e-02 1.13664842271e-02 5.12121388847e-03 1.60728955735e-03 1.60728955735e-03 + 5.12121388847e-03 1.13664842271e-02 2.21871186712e-02 + 3.78205596100e-02 2.80273245508e-02 1.45718308971e-02 1.45718308971e-02 2.80273245508e-02 3.78205596100e-02 + 3.08620524079e-02 1.78999426265e-02 7.92572136635e-03 2.15903301226e-03 6.20357753249e-04 2.15903301226e-03 + 7.92572136635e-03 1.78999426265e-02 3.08620524079e-02 + 3.16546517690e-02 1.45718308971e-02 5.85084407671e-04 1.45718308971e-02 3.16546517690e-02 3.54156347447e-02 + 2.55137286753e-02 1.24891306807e-02 4.02221833331e-03 5.43062651502e-04 5.43062651502e-04 4.02221833331e-03 + 1.24891306807e-02 2.55137286753e-02 3.54156347447e-02 + 2.77593740924e-02 1.45718308971e-02 1.45718308971e-02 2.77593740924e-02 3.70537599473e-02 3.00285261604e-02 + 1.73117518289e-02 7.35300020881e-03 1.64060165739e-03 1.47427756958e-04 1.64060165739e-03 7.35300020881e-03 + 1.73117518289e-02 3.00285261604e-02 3.70537599473e-02 + 3.16546517690e-02 2.80273245508e-02 3.16546517690e-02 3.70537599473e-02 3.18835940632e-02 2.09809548480e-02 + 1.00173301250e-02 3.65189274881e-03 7.48839342246e-04 7.48839342246e-04 3.65189274881e-03 1.00173301250e-02 + 2.09809548480e-02 3.18835940632e-02 3.70537599473e-02 + 3.78205596100e-02 3.78205596100e-02 3.54156347447e-02 3.00285261604e-02 2.09809548480e-02 1.09125507012e-02 + 5.26425072315e-03 2.09399363094e-03 1.38575808115e-03 2.09399363094e-03 5.26425072315e-03 1.09125507012e-02 + 2.09809548480e-02 3.00285261604e-02 3.54156347447e-02 + 3.18111513620e-02 3.08620524079e-02 2.55137286753e-02 1.73117518289e-02 1.00173301250e-02 5.26425072315e-03 + 2.68892841248e-03 1.11117793737e-03 1.11117793737e-03 2.68892841248e-03 5.26425072315e-03 1.00173301250e-02 + 1.73117518289e-02 2.55137286753e-02 3.08620524079e-02 + 2.21871186712e-02 1.78999426265e-02 1.24891306807e-02 7.35300020881e-03 3.65189274881e-03 2.09399363094e-03 + 1.11117793737e-03 9.45253549219e-04 1.11117793737e-03 2.09399363094e-03 3.65189274881e-03 7.35300020881e-03 + 1.24891306807e-02 1.78999426265e-02 2.21871186712e-02 + 1.13664842271e-02 7.92572136635e-03 4.02221833331e-03 1.64060165739e-03 7.48839342246e-04 1.38575808115e-03 + 1.11117793737e-03 1.11117793737e-03 1.38575808115e-03 7.48839342246e-04 1.64060165739e-03 4.02221833331e-03 + 7.92572136635e-03 1.13664842271e-02 1.40817702606e-02 + 5.12121388847e-03 2.15903301226e-03 5.43062651502e-04 1.47427756958e-04 7.48839342246e-04 2.09399363094e-03 + 2.68892841248e-03 2.09399363094e-03 7.48839342246e-04 1.47427756958e-04 5.43062651502e-04 2.15903301226e-03 + 5.12121388847e-03 7.02061778975e-03 7.02061778975e-03 + 1.60728955735e-03 6.20357753249e-04 5.43062651502e-04 1.64060165739e-03 3.65189274881e-03 5.26425072315e-03 + 5.26425072315e-03 3.65189274881e-03 1.64060165739e-03 5.43062651502e-04 6.20357753249e-04 1.60728955735e-03 + 3.53131094457e-03 4.09811851128e-03 3.53131094457e-03 + 1.60728955735e-03 2.15903301226e-03 4.02221833331e-03 7.35300020881e-03 1.00173301250e-02 1.09125507012e-02 + 1.00173301250e-02 7.35300020881e-03 4.02221833331e-03 2.15903301226e-03 1.60728955735e-03 2.37585679780e-03 + 2.19136307302e-03 2.19136307302e-03 2.37585679780e-03 + 5.12121388847e-03 7.92572136635e-03 1.24891306807e-02 1.73117518289e-02 2.09809548480e-02 2.09809548480e-02 + 1.73117518289e-02 1.24891306807e-02 7.92572136635e-03 5.12121388847e-03 3.53131094457e-03 2.19136307302e-03 + 2.26805038404e-03 2.19136307302e-03 3.53131094457e-03 + 1.13664842271e-02 1.78999426265e-02 2.55137286753e-02 3.00285261604e-02 3.18835940632e-02 3.00285261604e-02 + 2.55137286753e-02 1.78999426265e-02 1.13664842271e-02 7.02061778975e-03 4.09811851128e-03 2.19136307302e-03 + 2.19136307302e-03 4.09811851128e-03 7.02061778975e-03 + 2.21871186712e-02 3.08620524079e-02 3.54156347447e-02 3.70537599473e-02 3.70537599473e-02 3.54156347447e-02 + 3.08620524079e-02 2.21871186712e-02 1.40817702606e-02 7.02061778975e-03 3.53131094457e-03 2.37585679780e-03 + 3.53131094457e-03 7.02061778975e-03 1.40817702606e-02 + 3.72988193747e-02 3.42116560673e-02 2.56404718355e-02 2.56404718355e-02 3.42116560673e-02 3.72988193747e-02 + 2.77285509468e-02 1.56634321975e-02 6.70391326813e-03 1.60728955735e-03 2.99424480887e-04 1.60728955735e-03 + 6.70391326813e-03 1.56634321975e-02 2.77285509468e-02 + 3.42116560673e-02 1.76462182571e-02 7.66628780847e-03 1.76462182571e-02 3.42116560673e-02 3.60058531033e-02 + 2.43906523476e-02 1.23767071708e-02 3.86919526274e-03 6.20357753249e-04 6.20357753249e-04 3.86919526274e-03 + 1.23767071708e-02 2.43906523476e-02 3.60058531033e-02 + 2.56404718355e-02 7.66628780847e-03 7.66628780847e-03 2.56404718355e-02 3.72214469881e-02 3.30827615217e-02 + 1.86408380774e-02 8.27428669466e-03 2.15070114075e-03 5.43062651502e-04 2.15070114075e-03 8.27428669466e-03 + 1.86408380774e-02 3.30827615217e-02 3.72214469881e-02 + 2.56404718355e-02 1.76462182571e-02 2.56404718355e-02 3.74638848543e-02 3.66424673274e-02 2.47911036584e-02 + 1.28051588719e-02 5.53109046735e-03 1.64060165739e-03 1.64060165739e-03 5.53109046735e-03 1.28051588719e-02 + 2.47911036584e-02 3.66424673274e-02 3.74638848543e-02 + 3.42116560673e-02 3.42116560673e-02 3.72214469881e-02 3.66424673274e-02 2.71116363186e-02 1.67238267603e-02 + 8.33655930782e-03 3.65189274881e-03 2.06207427592e-03 3.65189274881e-03 8.33655930782e-03 1.67238267603e-02 + 2.71116363186e-02 3.66424673274e-02 3.72214469881e-02 + 3.72988193747e-02 3.60058531033e-02 3.30827615217e-02 2.47911036584e-02 1.67238267603e-02 9.07231580734e-03 + 5.26425072315e-03 3.29204038761e-03 3.29204038761e-03 5.26425072315e-03 9.07231580734e-03 1.67238267603e-02 + 2.47911036584e-02 3.30827615217e-02 3.60058531033e-02 + 2.77285509468e-02 2.43906523476e-02 1.86408380774e-02 1.28051588719e-02 8.33655930782e-03 5.26425072315e-03 + 3.60185781299e-03 2.63545476236e-03 3.60185781299e-03 5.26425072315e-03 8.33655930782e-03 1.28051588719e-02 + 1.86408380774e-02 2.43906523476e-02 2.77285509468e-02 + 1.56634321975e-02 1.23767071708e-02 8.27428669466e-03 5.53109046735e-03 3.65189274881e-03 3.29204038761e-03 + 2.63545476236e-03 2.63545476236e-03 3.29204038761e-03 3.65189274881e-03 5.53109046735e-03 8.27428669466e-03 + 1.23767071708e-02 1.56634321975e-02 1.79922853063e-02 + 6.70391326813e-03 3.86919526274e-03 2.15070114075e-03 1.64060165739e-03 2.06207427592e-03 3.29204038761e-03 + 3.60185781299e-03 3.29204038761e-03 2.06207427592e-03 1.64060165739e-03 2.15070114075e-03 3.86919526274e-03 + 6.70391326813e-03 8.64797657421e-03 8.64797657421e-03 + 1.60728955735e-03 6.20357753249e-04 5.43062651502e-04 1.64060165739e-03 3.65189274881e-03 5.26425072315e-03 + 5.26425072315e-03 3.65189274881e-03 1.64060165739e-03 5.43062651502e-04 6.20357753249e-04 1.60728955735e-03 + 3.53131094457e-03 4.09811851128e-03 3.53131094457e-03 + 2.99424480887e-04 6.20357753249e-04 2.15070114075e-03 5.53109046735e-03 8.33655930782e-03 9.07231580734e-03 + 8.33655930782e-03 5.53109046735e-03 2.15070114075e-03 6.20357753249e-04 2.99424480887e-04 1.09406336109e-03 + 1.38543835531e-03 1.38543835531e-03 1.09406336109e-03 + 1.60728955735e-03 3.86919526274e-03 8.27428669466e-03 1.28051588719e-02 1.67238267603e-02 1.67238267603e-02 + 1.28051588719e-02 8.27428669466e-03 3.86919526274e-03 1.60728955735e-03 1.09406336109e-03 7.14379378004e-04 + 7.65566866990e-04 7.14379378004e-04 1.09406336109e-03 + 6.70391326813e-03 1.23767071708e-02 1.86408380774e-02 2.47911036584e-02 2.71116363186e-02 2.47911036584e-02 + 1.86408380774e-02 1.23767071708e-02 6.70391326813e-03 3.53131094457e-03 1.38543835531e-03 7.65566866990e-04 + 7.65566866990e-04 1.38543835531e-03 3.53131094457e-03 + 1.56634321975e-02 2.43906523476e-02 3.30827615217e-02 3.66424673274e-02 3.66424673274e-02 3.30827615217e-02 + 2.43906523476e-02 1.56634321975e-02 8.64797657421e-03 4.09811851128e-03 1.38543835531e-03 7.14379378004e-04 + 1.38543835531e-03 4.09811851128e-03 8.64797657421e-03 + 2.77285509468e-02 3.60058531033e-02 3.72214469881e-02 3.74638848543e-02 3.72214469881e-02 3.60058531033e-02 + 2.77285509468e-02 1.79922853063e-02 8.64797657421e-03 3.53131094457e-03 1.09406336109e-03 1.09406336109e-03 + 3.53131094457e-03 8.64797657421e-03 1.79922853063e-02 + 4.77333687537e-02 3.50377196433e-02 2.76230644258e-02 3.50377196433e-02 4.77333687537e-02 3.85963986077e-02 + 2.42831208041e-02 1.25541732198e-02 4.59226684865e-03 1.60728955735e-03 1.60728955735e-03 4.59226684865e-03 + 1.25541732198e-02 2.42831208041e-02 3.85963986077e-02 + 3.50377196433e-02 1.55211868757e-02 1.55211868757e-02 3.50377196433e-02 4.46752712878e-02 3.49564497515e-02 + 2.13924434679e-02 9.66580695579e-03 3.86919526274e-03 2.15903301226e-03 3.86919526274e-03 9.66580695579e-03 + 2.13924434679e-02 3.49564497515e-02 4.46752712878e-02 + 2.76230644258e-02 1.55211868757e-02 2.76230644258e-02 4.40402072259e-02 4.27264005082e-02 3.00038356153e-02 + 1.63178959213e-02 8.27428669466e-03 4.02221833331e-03 4.02221833331e-03 8.27428669466e-03 1.63178959213e-02 + 3.00038356153e-02 4.27264005082e-02 4.40402072259e-02 + 3.50377196433e-02 3.50377196433e-02 4.40402072259e-02 4.48521019830e-02 3.62877199395e-02 2.24936844433e-02 + 1.28051588719e-02 7.35300020881e-03 5.18167581107e-03 7.35300020881e-03 1.28051588719e-02 2.24936844433e-02 + 3.62877199395e-02 4.48521019830e-02 4.40402072259e-02 + 4.77333687537e-02 4.46752712878e-02 4.27264005082e-02 3.62877199395e-02 2.45878312145e-02 1.67238267603e-02 + 1.00173301250e-02 7.08705078074e-03 7.08705078074e-03 1.00173301250e-02 1.67238267603e-02 2.45878312145e-02 + 3.62877199395e-02 4.27264005082e-02 4.46752712878e-02 + 3.85963986077e-02 3.49564497515e-02 3.00038356153e-02 2.24936844433e-02 1.67238267603e-02 1.09125507012e-02 + 8.95764366269e-03 7.32181384452e-03 8.95764366269e-03 1.09125507012e-02 1.67238267603e-02 2.24936844433e-02 + 3.00038356153e-02 3.49564497515e-02 3.85963986077e-02 + 2.42831208041e-02 2.13924434679e-02 1.63178959213e-02 1.28051588719e-02 1.00173301250e-02 8.95764366269e-03 + 7.66263937498e-03 7.66263937498e-03 8.95764366269e-03 1.00173301250e-02 1.28051588719e-02 1.63178959213e-02 + 2.13924434679e-02 2.42831208041e-02 2.61711989421e-02 + 1.25541732198e-02 9.66580695579e-03 8.27428669466e-03 7.35300020881e-03 7.08705078074e-03 7.32181384452e-03 + 7.66263937498e-03 7.32181384452e-03 7.08705078074e-03 7.35300020881e-03 8.27428669466e-03 9.66580695579e-03 + 1.25541732198e-02 1.49682875172e-02 1.49682875172e-02 + 4.59226684865e-03 3.86919526274e-03 4.02221833331e-03 5.18167581107e-03 7.08705078074e-03 8.95764366269e-03 + 8.95764366269e-03 7.08705078074e-03 5.18167581107e-03 4.02221833331e-03 3.86919526274e-03 4.59226684865e-03 + 6.75414542788e-03 6.84674742673e-03 6.75414542788e-03 + 1.60728955735e-03 2.15903301226e-03 4.02221833331e-03 7.35300020881e-03 1.00173301250e-02 1.09125507012e-02 + 1.00173301250e-02 7.35300020881e-03 4.02221833331e-03 2.15903301226e-03 1.60728955735e-03 2.37585679780e-03 + 2.19136307302e-03 2.19136307302e-03 2.37585679780e-03 + 1.60728955735e-03 3.86919526274e-03 8.27428669466e-03 1.28051588719e-02 1.67238267603e-02 1.67238267603e-02 + 1.28051588719e-02 8.27428669466e-03 3.86919526274e-03 1.60728955735e-03 1.09406336109e-03 7.14379378004e-04 + 7.65566866990e-04 7.14379378004e-04 1.09406336109e-03 + 4.59226684865e-03 9.66580695579e-03 1.63178959213e-02 2.24936844433e-02 2.45878312145e-02 2.24936844433e-02 + 1.63178959213e-02 9.66580695579e-03 4.59226684865e-03 2.37585679780e-03 7.14379378004e-04 2.92634334870e-04 + 2.92634334870e-04 7.14379378004e-04 2.37585679780e-03 + 1.25541732198e-02 2.13924434679e-02 3.00038356153e-02 3.62877199395e-02 3.62877199395e-02 3.00038356153e-02 + 2.13924434679e-02 1.25541732198e-02 6.75414542788e-03 2.19136307302e-03 7.65566866990e-04 2.92634334870e-04 + 7.65566866990e-04 2.19136307302e-03 6.75414542788e-03 + 2.42831208041e-02 3.49564497515e-02 4.27264005082e-02 4.48521019830e-02 4.27264005082e-02 3.49564497515e-02 + 2.42831208041e-02 1.49682875172e-02 6.84674742673e-03 2.19136307302e-03 7.14379378004e-04 7.14379378004e-04 + 2.19136307302e-03 6.84674742673e-03 1.49682875172e-02 + 3.85963986077e-02 4.46752712878e-02 4.40402072259e-02 4.40402072259e-02 4.46752712878e-02 3.85963986077e-02 + 2.61711989421e-02 1.49682875172e-02 6.75414542788e-03 2.37585679780e-03 1.09406336109e-03 2.37585679780e-03 + 6.75414542788e-03 1.49682875172e-02 2.61711989421e-02 + 6.52918175640e-02 5.04046005791e-02 5.04046005791e-02 6.52918175640e-02 5.61479157062e-02 3.97912967555e-02 + 2.38655503495e-02 1.25541732198e-02 6.70391326813e-03 5.12121388847e-03 6.70391326813e-03 1.25541732198e-02 + 2.38655503495e-02 3.97912967555e-02 5.61479157062e-02 + 5.04046005791e-02 3.75414545463e-02 5.04046005791e-02 5.92333053982e-02 5.24033937689e-02 3.71064909955e-02 + 2.13924434679e-02 1.23767071708e-02 7.92572136635e-03 7.92572136635e-03 1.23767071708e-02 2.13924434679e-02 + 3.71064909955e-02 5.24033937689e-02 5.92333053982e-02 + 5.04046005791e-02 5.04046005791e-02 6.09545305143e-02 5.84841221586e-02 4.57183984058e-02 3.00038356153e-02 + 1.86408380774e-02 1.24891306807e-02 9.78995536003e-03 1.24891306807e-02 1.86408380774e-02 3.00038356153e-02 + 4.57183984058e-02 5.84841221586e-02 6.09545305143e-02 + 6.52918175640e-02 5.92333053982e-02 5.84841221586e-02 5.02668272913e-02 3.62877199395e-02 2.47911036584e-02 + 1.73117518289e-02 1.40065686484e-02 1.40065686484e-02 1.73117518289e-02 2.47911036584e-02 3.62877199395e-02 + 5.02668272913e-02 5.84841221586e-02 5.92333053982e-02 + 5.61479157062e-02 5.24033937689e-02 4.57183984058e-02 3.62877199395e-02 2.71116363186e-02 2.09809548480e-02 + 1.64122236650e-02 1.43711514656e-02 1.64122236650e-02 2.09809548480e-02 2.71116363186e-02 3.62877199395e-02 + 4.57183984058e-02 5.24033937689e-02 5.61479157062e-02 + 3.97912967555e-02 3.71064909955e-02 3.00038356153e-02 2.47911036584e-02 2.09809548480e-02 1.68913524358e-02 + 1.56673079604e-02 1.56673079604e-02 1.68913524358e-02 2.09809548480e-02 2.47911036584e-02 3.00038356153e-02 + 3.71064909955e-02 3.97912967555e-02 4.24418182390e-02 + 2.38655503495e-02 2.13924434679e-02 1.86408380774e-02 1.73117518289e-02 1.64122236650e-02 1.56673079604e-02 + 1.53593248116e-02 1.56673079604e-02 1.64122236650e-02 1.73117518289e-02 1.86408380774e-02 2.13924434679e-02 + 2.38655503495e-02 2.49947177389e-02 2.49947177389e-02 + 1.25541732198e-02 1.23767071708e-02 1.24891306807e-02 1.40065686484e-02 1.43711514656e-02 1.56673079604e-02 + 1.56673079604e-02 1.43711514656e-02 1.40065686484e-02 1.24891306807e-02 1.23767071708e-02 1.25541732198e-02 + 1.36280371710e-02 1.31863090803e-02 1.36280371710e-02 + 6.70391326813e-03 7.92572136635e-03 9.78995536003e-03 1.40065686484e-02 1.64122236650e-02 1.68913524358e-02 + 1.64122236650e-02 1.40065686484e-02 9.78995536003e-03 7.92572136635e-03 6.70391326813e-03 6.75414542788e-03 + 5.80240507057e-03 5.80240507057e-03 6.75414542788e-03 + 5.12121388847e-03 7.92572136635e-03 1.24891306807e-02 1.73117518289e-02 2.09809548480e-02 2.09809548480e-02 + 1.73117518289e-02 1.24891306807e-02 7.92572136635e-03 5.12121388847e-03 3.53131094457e-03 2.19136307302e-03 + 2.26805038404e-03 2.19136307302e-03 3.53131094457e-03 + 6.70391326813e-03 1.23767071708e-02 1.86408380774e-02 2.47911036584e-02 2.71116363186e-02 2.47911036584e-02 + 1.86408380774e-02 1.23767071708e-02 6.70391326813e-03 3.53131094457e-03 1.38543835531e-03 7.65566866990e-04 + 7.65566866990e-04 1.38543835531e-03 3.53131094457e-03 + 1.25541732198e-02 2.13924434679e-02 3.00038356153e-02 3.62877199395e-02 3.62877199395e-02 3.00038356153e-02 + 2.13924434679e-02 1.25541732198e-02 6.75414542788e-03 2.19136307302e-03 7.65566866990e-04 2.92634334870e-04 + 7.65566866990e-04 2.19136307302e-03 6.75414542788e-03 + 2.38655503495e-02 3.71064909955e-02 4.57183984058e-02 5.02668272913e-02 4.57183984058e-02 3.71064909955e-02 + 2.38655503495e-02 1.36280371710e-02 5.80240507057e-03 2.26805038404e-03 7.65566866990e-04 7.65566866990e-04 + 2.26805038404e-03 5.80240507057e-03 1.36280371710e-02 + 3.97912967555e-02 5.24033937689e-02 5.84841221586e-02 5.84841221586e-02 5.24033937689e-02 3.97912967555e-02 + 2.49947177389e-02 1.31863090803e-02 5.80240507057e-03 2.19136307302e-03 1.38543835531e-03 2.19136307302e-03 + 5.80240507057e-03 1.31863090803e-02 2.49947177389e-02 + 5.61479157062e-02 5.92333053982e-02 6.09545305143e-02 5.92333053982e-02 5.61479157062e-02 4.24418182390e-02 + 2.49947177389e-02 1.36280371710e-02 6.75414542788e-03 3.53131094457e-03 3.53131094457e-03 6.75414542788e-03 + 1.36280371710e-02 2.49947177389e-02 4.24418182390e-02 + 1.16509120681e-01 1.02615488905e-01 1.16509120681e-01 9.95704232240e-02 6.46951081959e-02 3.97912967555e-02 + 2.42831208041e-02 1.56634321975e-02 1.13664842271e-02 1.13664842271e-02 1.56634321975e-02 2.42831208041e-02 + 3.97912967555e-02 6.46951081959e-02 9.95704232240e-02 + 1.02615488905e-01 1.02615488905e-01 1.13574350192e-01 8.15738219549e-02 5.24033937689e-02 3.49564497515e-02 + 2.43906523476e-02 1.78999426265e-02 1.52960780652e-02 1.78999426265e-02 2.43906523476e-02 3.49564497515e-02 + 5.24033937689e-02 8.15738219549e-02 1.13574350192e-01 + 1.16509120681e-01 1.13574350192e-01 8.53986300889e-02 5.84841221586e-02 4.27264005082e-02 3.30827615217e-02 + 2.55137286753e-02 2.14983233332e-02 2.14983233332e-02 2.55137286753e-02 3.30827615217e-02 4.27264005082e-02 + 5.84841221586e-02 8.53986300889e-02 1.13574350192e-01 + 9.95704232240e-02 8.15738219549e-02 5.84841221586e-02 4.48521019830e-02 3.66424673274e-02 3.00285261604e-02 + 2.56714935438e-02 2.60511627292e-02 2.56714935438e-02 3.00285261604e-02 3.66424673274e-02 4.48521019830e-02 + 5.84841221586e-02 8.15738219549e-02 9.95704232240e-02 + 6.46951081959e-02 5.24033937689e-02 4.27264005082e-02 3.66424673274e-02 3.18835940632e-02 2.94652595219e-02 + 2.78780404497e-02 2.78780404497e-02 2.94652595219e-02 3.18835940632e-02 3.66424673274e-02 4.27264005082e-02 + 5.24033937689e-02 6.46951081959e-02 7.10938675575e-02 + 3.97912967555e-02 3.49564497515e-02 3.30827615217e-02 3.00285261604e-02 2.94652595219e-02 3.01994514047e-02 + 2.96935773932e-02 3.01994514047e-02 2.94652595219e-02 3.00285261604e-02 3.30827615217e-02 3.49564497515e-02 + 3.97912967555e-02 4.45753367440e-02 4.45753367440e-02 + 2.42831208041e-02 2.43906523476e-02 2.55137286753e-02 2.56714935438e-02 2.78780404497e-02 2.96935773932e-02 + 2.96935773932e-02 2.78780404497e-02 2.56714935438e-02 2.55137286753e-02 2.43906523476e-02 2.42831208041e-02 + 2.49947177389e-02 2.65333628877e-02 2.49947177389e-02 + 1.56634321975e-02 1.78999426265e-02 2.14983233332e-02 2.60511627292e-02 2.78780404497e-02 3.01994514047e-02 + 2.78780404497e-02 2.60511627292e-02 2.14983233332e-02 1.78999426265e-02 1.56634321975e-02 1.49682875172e-02 + 1.31863090803e-02 1.31863090803e-02 1.49682875172e-02 + 1.13664842271e-02 1.52960780652e-02 2.14983233332e-02 2.56714935438e-02 2.94652595219e-02 2.94652595219e-02 + 2.56714935438e-02 2.14983233332e-02 1.52960780652e-02 1.13664842271e-02 8.64797657421e-03 6.84674742673e-03 + 5.80240507057e-03 6.84674742673e-03 8.64797657421e-03 + 1.13664842271e-02 1.78999426265e-02 2.55137286753e-02 3.00285261604e-02 3.18835940632e-02 3.00285261604e-02 + 2.55137286753e-02 1.78999426265e-02 1.13664842271e-02 7.02061778975e-03 4.09811851128e-03 2.19136307302e-03 + 2.19136307302e-03 4.09811851128e-03 7.02061778975e-03 + 1.56634321975e-02 2.43906523476e-02 3.30827615217e-02 3.66424673274e-02 3.66424673274e-02 3.30827615217e-02 + 2.43906523476e-02 1.56634321975e-02 8.64797657421e-03 4.09811851128e-03 1.38543835531e-03 7.14379378004e-04 + 1.38543835531e-03 4.09811851128e-03 8.64797657421e-03 + 2.42831208041e-02 3.49564497515e-02 4.27264005082e-02 4.48521019830e-02 4.27264005082e-02 3.49564497515e-02 + 2.42831208041e-02 1.49682875172e-02 6.84674742673e-03 2.19136307302e-03 7.14379378004e-04 7.14379378004e-04 + 2.19136307302e-03 6.84674742673e-03 1.49682875172e-02 + 3.97912967555e-02 5.24033937689e-02 5.84841221586e-02 5.84841221586e-02 5.24033937689e-02 3.97912967555e-02 + 2.49947177389e-02 1.31863090803e-02 5.80240507057e-03 2.19136307302e-03 1.38543835531e-03 2.19136307302e-03 + 5.80240507057e-03 1.31863090803e-02 2.49947177389e-02 + 6.46951081959e-02 8.15738219549e-02 8.53986300889e-02 8.15738219549e-02 6.46951081959e-02 4.45753367440e-02 + 2.65333628877e-02 1.31863090803e-02 6.84674742673e-03 4.09811851128e-03 4.09811851128e-03 6.84674742673e-03 + 1.31863090803e-02 2.65333628877e-02 4.45753367440e-02 + 9.95704232240e-02 1.13574350192e-01 1.13574350192e-01 9.95704232240e-02 7.10938675575e-02 4.45753367440e-02 + 2.49947177389e-02 1.49682875172e-02 8.64797657421e-03 7.02061778975e-03 8.64797657421e-03 1.49682875171e-02 + 2.49947177389e-02 4.45753367440e-02 7.10938675575e-02 + 1.82505675344e-01 1.82505675344e-01 1.76044347723e-01 9.95704232240e-02 5.61479157062e-02 3.85963986077e-02 + 2.77285509468e-02 2.21871186712e-02 1.95082540137e-02 2.21871186712e-02 2.77285509468e-02 3.85963986077e-02 + 5.61479157062e-02 9.95704232240e-02 1.76044347723e-01 + 1.82505675344e-01 1.83341145810e-01 1.13574350192e-01 5.92333053982e-02 4.46752712878e-02 3.60058531033e-02 + 3.08620524079e-02 2.79664526175e-02 2.79664526175e-02 3.08620524079e-02 3.60058531033e-02 4.46752712878e-02 + 5.92333053982e-02 1.13574350192e-01 1.83341145810e-01 + 1.76044347723e-01 1.13574350192e-01 6.09545305143e-02 4.40402072259e-02 3.72214469881e-02 3.54156347447e-02 + 3.49273913093e-02 3.55022456672e-02 3.49273913093e-02 3.54156347447e-02 3.72214469881e-02 4.40402072259e-02 + 6.09545305143e-02 1.13574350192e-01 1.76044347723e-01 + 9.95704232240e-02 5.92333053982e-02 4.40402072259e-02 3.74638848543e-02 3.70537599473e-02 3.88117173565e-02 + 4.06521018269e-02 4.06521018269e-02 3.88117173565e-02 3.70537599473e-02 3.74638848543e-02 4.40402072259e-02 + 5.92333053982e-02 9.95704232240e-02 1.21925608243e-01 + 5.61479157062e-02 4.46752712878e-02 3.72214469881e-02 3.70537599473e-02 4.31600648738e-02 4.49452510339e-02 + 4.78489705763e-02 4.49452510339e-02 4.31600648738e-02 3.70537599473e-02 3.72214469881e-02 4.46752712878e-02 + 5.61479157062e-02 7.10938675575e-02 7.10938675575e-02 + 3.85963986077e-02 3.60058531033e-02 3.54156347447e-02 3.88117173565e-02 4.49452510339e-02 5.02796892510e-02 + 5.02796892510e-02 4.49452510339e-02 3.88117173565e-02 3.54156347447e-02 3.60058531033e-02 3.85963986077e-02 + 4.24418182390e-02 4.45753367440e-02 4.24418182390e-02 + 2.77285509468e-02 3.08620524079e-02 3.49273913093e-02 4.06521018269e-02 4.78489705763e-02 5.02796892510e-02 + 4.78489705763e-02 4.06521018269e-02 3.49273913093e-02 3.08620524079e-02 2.77285509468e-02 2.61711989421e-02 + 2.49947177389e-02 2.49947177389e-02 2.61711989421e-02 + 2.21871186712e-02 2.79664526175e-02 3.55022456672e-02 4.06521018269e-02 4.49452510339e-02 4.49452510339e-02 + 4.06521018269e-02 3.55022456672e-02 2.79664526175e-02 2.21871186712e-02 1.79922853063e-02 1.49682875172e-02 + 1.36280371710e-02 1.49682875172e-02 1.79922853063e-02 + 1.95082540137e-02 2.79664526175e-02 3.49273913093e-02 3.88117173565e-02 4.31600648738e-02 3.88117173565e-02 + 3.49273913093e-02 2.79664526175e-02 1.95082540137e-02 1.40817702606e-02 8.64797657421e-03 6.75414542788e-03 + 6.75414542788e-03 8.64797657421e-03 1.40817702606e-02 + 2.21871186712e-02 3.08620524079e-02 3.54156347447e-02 3.70537599473e-02 3.70537599473e-02 3.54156347447e-02 + 3.08620524079e-02 2.21871186712e-02 1.40817702606e-02 7.02061778975e-03 3.53131094457e-03 2.37585679780e-03 + 3.53131094457e-03 7.02061778975e-03 1.40817702606e-02 + 2.77285509468e-02 3.60058531033e-02 3.72214469881e-02 3.74638848543e-02 3.72214469881e-02 3.60058531033e-02 + 2.77285509468e-02 1.79922853063e-02 8.64797657421e-03 3.53131094457e-03 1.09406336109e-03 1.09406336109e-03 + 3.53131094457e-03 8.64797657421e-03 1.79922853063e-02 + 3.85963986077e-02 4.46752712878e-02 4.40402072259e-02 4.40402072259e-02 4.46752712878e-02 3.85963986077e-02 + 2.61711989421e-02 1.49682875172e-02 6.75414542788e-03 2.37585679780e-03 1.09406336109e-03 2.37585679780e-03 + 6.75414542788e-03 1.49682875172e-02 2.61711989421e-02 + 5.61479157062e-02 5.92333053982e-02 6.09545305143e-02 5.92333053982e-02 5.61479157062e-02 4.24418182390e-02 + 2.49947177389e-02 1.36280371710e-02 6.75414542788e-03 3.53131094457e-03 3.53131094457e-03 6.75414542788e-03 + 1.36280371710e-02 2.49947177389e-02 4.24418182390e-02 + 9.95704232240e-02 1.13574350192e-01 1.13574350192e-01 9.95704232240e-02 7.10938675575e-02 4.45753367440e-02 + 2.49947177389e-02 1.49682875172e-02 8.64797657421e-03 7.02061778975e-03 8.64797657421e-03 1.49682875171e-02 + 2.49947177389e-02 4.45753367440e-02 7.10938675575e-02 + 1.76044347723e-01 1.83341145810e-01 1.76044347723e-01 1.21925608243e-01 7.10938675575e-02 4.24418182390e-02 + 2.61711989421e-02 1.79922853063e-02 1.40817702606e-02 1.40817702606e-02 1.79922853063e-02 2.61711989421e-02 + 4.24418182390e-02 7.10938675575e-02 1.21925608243e-01 diff --git a/tests/03_NAO_multik/scf_out_dh_t/trs1_nao.csr.ref b/tests/03_NAO_multik/scf_out_dh_t/tr_nao.csr.ref similarity index 100% rename from tests/03_NAO_multik/scf_out_dh_t/trs1_nao.csr.ref rename to tests/03_NAO_multik/scf_out_dh_t/tr_nao.csr.ref diff --git a/tests/03_NAO_multik/scf_out_dos_spin4/INPUT b/tests/03_NAO_multik/scf_out_dos_spin4/INPUT index b57f2e84774..11b9c7932ef 100644 --- a/tests/03_NAO_multik/scf_out_dos_spin4/INPUT +++ b/tests/03_NAO_multik/scf_out_dos_spin4/INPUT @@ -1,4 +1,4 @@ -INPUT_PARAMETERS +INPUT_PARAMETERS #Parameters (General) suffix autotest pseudo_dir ../../PP_ORB @@ -12,7 +12,7 @@ smearing_method gaussian smearing_sigma 0.001 #Parameters (3.PW) ecutwfc 10 -scf_thr 1e-6 +scf_thr 1e-8 scf_nmax 100 #noncolin 1 @@ -24,7 +24,7 @@ cal_stress 1 #Parameters (LCAO) basis_type lcao ks_solver scalapack_gvx -pw_diag_thr 0.00001 +pw_diag_thr 1e-5 mixing_type pulay mixing_beta 0.7 diff --git a/tests/03_NAO_multik/scf_out_dos_spin4/STRU b/tests/03_NAO_multik/scf_out_dos_spin4/STRU index 79211e7719f..887de781273 100644 --- a/tests/03_NAO_multik/scf_out_dos_spin4/STRU +++ b/tests/03_NAO_multik/scf_out_dos_spin4/STRU @@ -19,9 +19,9 @@ Direct //Cartesian or Direct coordinate. As 0 1 -0.2600000 0.2400000 0.245000000 0 0 0 +0.2600000 0.2400000 0.245000000 0 0 0 mag 1.7320508 -Ga //Element Label +Ga //Element Label 0 -1 //number of atom -0.00000 0.00000 0.000000 0 0 0 +1 //number of atom +0.00000 0.00000 0.000000 0 0 0 mag 1.7320508 diff --git a/tests/03_NAO_multik/scf_out_dos_spin4/result.ref b/tests/03_NAO_multik/scf_out_dos_spin4/result.ref index e54b51d3064..8f9915deded 100644 --- a/tests/03_NAO_multik/scf_out_dos_spin4/result.ref +++ b/tests/03_NAO_multik/scf_out_dos_spin4/result.ref @@ -1,6 +1,6 @@ -etotref -1964.0663947982770878 -etotperatomref -982.0331973991 -totalforceref 0.162158 -totalstressref 1877.059089 -totaldosref 38 -totaltimeref 16.23 +etotref -1964.0663947983236994 +etotperatomref -982.0331973992 +totalforceref 0.162676 +totalstressref 1877.059085 +totaldosref 38.000000 +totaltimeref 5.26 diff --git a/tests/03_NAO_multik/scf_out_elf/refelftot.cube b/tests/03_NAO_multik/scf_out_elf/elftot.cube.ref similarity index 100% rename from tests/03_NAO_multik/scf_out_elf/refelftot.cube rename to tests/03_NAO_multik/scf_out_elf/elftot.cube.ref diff --git a/tests/03_NAO_multik/scf_out_hsr_spin4/rr.csr.ref b/tests/03_NAO_multik/scf_out_hsr_spin4/rr_nao.txt.ref similarity index 100% rename from tests/03_NAO_multik/scf_out_hsr_spin4/rr.csr.ref rename to tests/03_NAO_multik/scf_out_hsr_spin4/rr_nao.txt.ref diff --git a/tests/04_FF/CMakeLists.txt b/tests/04_FF/CMakeLists.txt index e40cc5079a9..23900804d14 100644 --- a/tests/04_FF/CMakeLists.txt +++ b/tests/04_FF/CMakeLists.txt @@ -10,7 +10,7 @@ if(ENABLE_ASAN) else() add_test( NAME 04_FF - COMMAND ${BASH} ../integrate/Autotest.sh -a ${ABACUS_BIN_PATH} -n 4 + COMMAND ${BASH} ../integrate/Autotest.sh -a ${ABACUS_BIN_PATH} -n 4 -j auto WORKING_DIRECTORY ${ABACUS_TEST_DIR}/04_FF ) endif() diff --git a/tests/05_rtTDDFT/12_NO_re_TDDFT/restart/Restart_td.txt b/tests/05_rtTDDFT/12_NO_re_TDDFT/restart/Restart_td.txt index 4bcfc2a96fa..81128e41c60 100644 --- a/tests/05_rtTDDFT/12_NO_re_TDDFT/restart/Restart_td.txt +++ b/tests/05_rtTDDFT/12_NO_re_TDDFT/restart/Restart_td.txt @@ -1,3 +1,6 @@ -5 -0 0 0 -0 0 0 \ No newline at end of file +# Hartree atomic units; step indices start at 0. +# gauge: 0=length, 1=velocity, 2=hybrid; source: 0=field, 1=file +# step left_step(-1=none) gauge dt source +5 -1 0 2.0670686668280682 0 +0 0 0 # A_left: x y z +0 0 0 # A_right: x y z diff --git a/tests/05_rtTDDFT/19_NO_SUPERSINE_TDDFT/vector_pot.txt.ref b/tests/05_rtTDDFT/19_NO_SUPERSINE_TDDFT/vector_pot.txt.ref index 90a39b97af9..563b1e1dcd9 100644 --- a/tests/05_rtTDDFT/19_NO_SUPERSINE_TDDFT/vector_pot.txt.ref +++ b/tests/05_rtTDDFT/19_NO_SUPERSINE_TDDFT/vector_pot.txt.ref @@ -1,4 +1,4 @@ -#istep A_x A_y A_z +#istep A_x A_y A_z # propagation values, Hartree atomic units 1 0.000000e+00 0.000000e+00 0.000000e+00 2 4.096138e-03 -2.078722e-02 0.000000e+00 3 4.375154e-03 -5.368872e-02 0.000000e+00 diff --git a/tests/05_rtTDDFT/20_NO_MIXED_EFIELD_TDDFT/vector_pot.txt.ref b/tests/05_rtTDDFT/20_NO_MIXED_EFIELD_TDDFT/vector_pot.txt.ref index c863f6c0a6a..fe7ac60c596 100644 --- a/tests/05_rtTDDFT/20_NO_MIXED_EFIELD_TDDFT/vector_pot.txt.ref +++ b/tests/05_rtTDDFT/20_NO_MIXED_EFIELD_TDDFT/vector_pot.txt.ref @@ -1,4 +1,4 @@ -#istep A_x A_y A_z +#istep A_x A_y A_z # propagation values, Hartree atomic units 1 -4.438787e-04 -1.062812e-04 -3.472515e-05 2 -1.236916e-03 -5.069838e-05 -3.925135e-04 3 -3.258834e-03 -6.966863e-04 -9.687700e-04 diff --git a/tests/05_rtTDDFT/21_PW_len_TDDFT/INPUT b/tests/05_rtTDDFT/21_PW_len_TDDFT/INPUT new file mode 100644 index 00000000000..f4940135f26 --- /dev/null +++ b/tests/05_rtTDDFT/21_PW_len_TDDFT/INPUT @@ -0,0 +1,51 @@ +INPUT_PARAMETERS + +suffix autotest +pseudo_dir ../../PP_ORB +esolver_type tddft +basis_type pw +calculation md +device cpu +precision double +ecutwfc 20 +ecutrho 80 +nbands 6 +nspin 1 +kpar 2 +symmetry 0 +init_wfc atomic+random +pw_seed 1 +ks_solver cg +pw_diag_thr 1e-14 +scf_thr 1e-12 +scf_nmax 100 +mixing_type broyden +mixing_beta 0.5 +mixing_gg0 0 +smearing_method gauss +smearing_sigma 0.002 +md_type nve +init_vel 1 +md_dt 0.01 +md_nstep 9 +cal_force 0 +cal_stress 0 +td_stype 0 +td_vext 1 +td_vext_dire 1 +td_ttype 0 +td_tstart 0 +td_tend 100 +td_gauss_freq 2 +td_gauss_phase 0 +td_gauss_sigma 0.04 +td_gauss_t0 4 +td_gauss_amp 0.001 +lin_solver bicgstab +lin_precond kinetic +lin_thr 1e-14 +lin_maxiter 100 +td_cn_init 0 +lin_reconstruct 0 +out_current 1 +out_level ie diff --git a/tests/05_rtTDDFT/21_PW_len_TDDFT/KPT b/tests/05_rtTDDFT/21_PW_len_TDDFT/KPT new file mode 100644 index 00000000000..23f1466d0f8 --- /dev/null +++ b/tests/05_rtTDDFT/21_PW_len_TDDFT/KPT @@ -0,0 +1,5 @@ +K_POINTS +2 +Direct +0.17 0.11 0.07 0.5 +-0.17 -0.11 -0.07 0.5 diff --git a/tests/05_rtTDDFT/21_PW_len_TDDFT/README b/tests/05_rtTDDFT/21_PW_len_TDDFT/README new file mode 100644 index 00000000000..69b906ef37a --- /dev/null +++ b/tests/05_rtTDDFT/21_PW_len_TDDFT/README @@ -0,0 +1 @@ +Length-gauge PW RT-TDDFT: self-consistent propagation and nonlocal current. diff --git a/tests/05_rtTDDFT/21_PW_len_TDDFT/STRU b/tests/05_rtTDDFT/21_PW_len_TDDFT/STRU new file mode 100644 index 00000000000..e059e5d6062 --- /dev/null +++ b/tests/05_rtTDDFT/21_PW_len_TDDFT/STRU @@ -0,0 +1,18 @@ +ATOMIC_SPECIES +Si 28 Si.pz-n-nc.UPF upf201 + +LATTICE_CONSTANT +10.2 + +LATTICE_VECTORS +0.0 0.5 0.5 +0.5 0.0 0.5 +0.5 0.5 0.0 + +ATOMIC_POSITIONS +Direct +Si +0.0 +2 +0.00 0.00 0.00 0 0 0 v 0 0 0 +0.25 0.25 0.25 0 0 0 v 0 0 0 diff --git a/tests/05_rtTDDFT/21_PW_len_TDDFT/current_tot.txt.ref b/tests/05_rtTDDFT/21_PW_len_TDDFT/current_tot.txt.ref new file mode 100644 index 00000000000..26fef0c8591 --- /dev/null +++ b/tests/05_rtTDDFT/21_PW_len_TDDFT/current_tot.txt.ref @@ -0,0 +1,9 @@ +1 -8.7372533313726819e-13 1.7628266998490296e-14 -1.8749696788461911e-13 +2 -5.1131148800069390e-09 9.3768918085396477e-12 -6.5784959639751502e-12 +3 -1.9381978870963455e-08 3.3429211065289635e-11 -2.7056414902621067e-11 +4 -3.9203858190765775e-08 7.0456786011446578e-11 -5.0922105945962130e-11 +5 -5.9267906815224327e-08 1.2505092797541114e-10 -6.8173851992747819e-11 +6 -7.4450138263918426e-08 2.0667322060979868e-10 -7.4987113370199227e-11 +7 -8.1044871563319266e-08 3.1929273005902725e-10 -7.4715103247141183e-11 +8 -7.6863270621976298e-08 4.5319603618971026e-10 -7.9047871820757897e-11 +9 -6.1419514746217234e-08 5.8561397596435925e-10 -1.0917389591425692e-10 diff --git a/tests/05_rtTDDFT/21_PW_len_TDDFT/result.ref b/tests/05_rtTDDFT/21_PW_len_TDDFT/result.ref new file mode 100644 index 00000000000..1cf2fbe01ca --- /dev/null +++ b/tests/05_rtTDDFT/21_PW_len_TDDFT/result.ref @@ -0,0 +1,4 @@ +etotref -251.7584934168609 +etotperatomref -125.8792467084 +CompareCurrent_pass 0 +totaltimeref 0.96 diff --git a/tests/05_rtTDDFT/22_PW_vel_TDDFT/INPUT b/tests/05_rtTDDFT/22_PW_vel_TDDFT/INPUT new file mode 100644 index 00000000000..de0591c7624 --- /dev/null +++ b/tests/05_rtTDDFT/22_PW_vel_TDDFT/INPUT @@ -0,0 +1,51 @@ +INPUT_PARAMETERS + +suffix autotest +pseudo_dir ../../PP_ORB +esolver_type tddft +basis_type pw +calculation md +device cpu +precision double +ecutwfc 20 +ecutrho 80 +nbands 6 +nspin 1 +kpar 2 +symmetry 0 +init_wfc atomic+random +pw_seed 1 +ks_solver cg +pw_diag_thr 1e-14 +scf_thr 1e-12 +scf_nmax 100 +mixing_type broyden +mixing_beta 0.5 +mixing_gg0 0 +smearing_method gauss +smearing_sigma 0.002 +md_type nve +init_vel 1 +md_dt 0.01 +md_nstep 9 +cal_force 0 +cal_stress 0 +td_stype 1 +td_vext 1 +td_vext_dire 1 +td_ttype 0 +td_tstart 0 +td_tend 100 +td_gauss_freq 2 +td_gauss_phase 0 +td_gauss_sigma 0.04 +td_gauss_t0 4 +td_gauss_amp 0.001 +lin_solver bicgstab +lin_precond kinetic +lin_thr 1e-14 +lin_maxiter 100 +td_cn_init 0 +lin_reconstruct 0 +out_current 1 +out_level ie diff --git a/tests/05_rtTDDFT/22_PW_vel_TDDFT/KPT b/tests/05_rtTDDFT/22_PW_vel_TDDFT/KPT new file mode 100644 index 00000000000..23f1466d0f8 --- /dev/null +++ b/tests/05_rtTDDFT/22_PW_vel_TDDFT/KPT @@ -0,0 +1,5 @@ +K_POINTS +2 +Direct +0.17 0.11 0.07 0.5 +-0.17 -0.11 -0.07 0.5 diff --git a/tests/05_rtTDDFT/22_PW_vel_TDDFT/README b/tests/05_rtTDDFT/22_PW_vel_TDDFT/README new file mode 100644 index 00000000000..c5075f6fffa --- /dev/null +++ b/tests/05_rtTDDFT/22_PW_vel_TDDFT/README @@ -0,0 +1 @@ +Velocity-gauge PW RT-TDDFT: vector-potential coupling and nonlocal current. diff --git a/tests/05_rtTDDFT/22_PW_vel_TDDFT/STRU b/tests/05_rtTDDFT/22_PW_vel_TDDFT/STRU new file mode 100644 index 00000000000..e059e5d6062 --- /dev/null +++ b/tests/05_rtTDDFT/22_PW_vel_TDDFT/STRU @@ -0,0 +1,18 @@ +ATOMIC_SPECIES +Si 28 Si.pz-n-nc.UPF upf201 + +LATTICE_CONSTANT +10.2 + +LATTICE_VECTORS +0.0 0.5 0.5 +0.5 0.0 0.5 +0.5 0.5 0.0 + +ATOMIC_POSITIONS +Direct +Si +0.0 +2 +0.00 0.00 0.00 0 0 0 v 0 0 0 +0.25 0.25 0.25 0 0 0 v 0 0 0 diff --git a/tests/05_rtTDDFT/22_PW_vel_TDDFT/current_tot.txt.ref b/tests/05_rtTDDFT/22_PW_vel_TDDFT/current_tot.txt.ref new file mode 100644 index 00000000000..3da56c70b31 --- /dev/null +++ b/tests/05_rtTDDFT/22_PW_vel_TDDFT/current_tot.txt.ref @@ -0,0 +1,9 @@ +1 1.0605063946060377e-12 -4.2509106434665706e-13 1.5087168526571182e-12 +2 1.3751229920540302e-07 1.6621406925090305e-11 -6.2358083121770329e-12 +3 3.0372208642324491e-07 6.4089539038915301e-11 -2.3316792632040393e-12 +4 4.8810917648953070e-07 1.6195288859878264e-10 2.7263499438301492e-11 +5 6.7777087857869071e-07 3.2484685380946662e-10 9.8423686153289534e-11 +6 8.5563526468913026e-07 5.6892549437259944e-10 2.3151902321144712e-10 +7 1.0040847301693847e-06 9.1229061982594634e-10 4.4941574231629083e-10 +8 1.1100465588291771e-06 1.3739801542497154e-09 7.7301888872485590e-10 +9 1.1668032707079688e-06 1.9781422929124190e-09 1.2187167552518205e-09 diff --git a/tests/05_rtTDDFT/22_PW_vel_TDDFT/result.ref b/tests/05_rtTDDFT/22_PW_vel_TDDFT/result.ref new file mode 100644 index 00000000000..6fbfd086bad --- /dev/null +++ b/tests/05_rtTDDFT/22_PW_vel_TDDFT/result.ref @@ -0,0 +1,4 @@ +etotref -251.7684286794544 +etotperatomref -125.8842143397 +CompareCurrent_pass 0 +totaltimeref 0.95 diff --git a/tests/05_rtTDDFT/23_PW_spin_TDDFT/INPUT b/tests/05_rtTDDFT/23_PW_spin_TDDFT/INPUT new file mode 100644 index 00000000000..27d9f7ecd36 --- /dev/null +++ b/tests/05_rtTDDFT/23_PW_spin_TDDFT/INPUT @@ -0,0 +1,52 @@ +INPUT_PARAMETERS + +suffix autotest +pseudo_dir ../../PP_ORB +esolver_type tddft +basis_type pw +calculation md +device cpu +precision double +ecutwfc 20 +ecutrho 80 +nbands 6 +nspin 2 +kpar 2 +symmetry 0 +init_wfc atomic+random +pw_seed 1 +ks_solver cg +pw_diag_thr 1e-14 +scf_thr 1e-12 +scf_nmax 100 +mixing_type broyden +mixing_beta 0.5 +mixing_gg0 0 +smearing_method gauss +smearing_sigma 0.002 +md_type nve +init_vel 1 +md_dt 0.01 +md_nstep 9 +cal_force 0 +cal_stress 0 +td_stype 1 +td_vext 1 +td_vext_dire 1 +td_ttype 0 +td_tstart 0 +td_tend 100 +td_gauss_freq 2 +td_gauss_phase 0 +td_gauss_sigma 0.04 +td_gauss_t0 4 +td_gauss_amp 0.001 +lin_solver bicgstab +lin_precond kinetic +lin_thr 1e-14 +lin_maxiter 100 +td_cn_init 0 +lin_reconstruct 0 +out_current 1 +out_level ie +nupdown 2 diff --git a/tests/05_rtTDDFT/23_PW_spin_TDDFT/KPT b/tests/05_rtTDDFT/23_PW_spin_TDDFT/KPT new file mode 100644 index 00000000000..23f1466d0f8 --- /dev/null +++ b/tests/05_rtTDDFT/23_PW_spin_TDDFT/KPT @@ -0,0 +1,5 @@ +K_POINTS +2 +Direct +0.17 0.11 0.07 0.5 +-0.17 -0.11 -0.07 0.5 diff --git a/tests/05_rtTDDFT/23_PW_spin_TDDFT/README b/tests/05_rtTDDFT/23_PW_spin_TDDFT/README new file mode 100644 index 00000000000..1149585cac2 --- /dev/null +++ b/tests/05_rtTDDFT/23_PW_spin_TDDFT/README @@ -0,0 +1 @@ +Collinear spin-polarized PW RT-TDDFT: energy and current in the velocity gauge. diff --git a/tests/05_rtTDDFT/23_PW_spin_TDDFT/STRU b/tests/05_rtTDDFT/23_PW_spin_TDDFT/STRU new file mode 100644 index 00000000000..e059e5d6062 --- /dev/null +++ b/tests/05_rtTDDFT/23_PW_spin_TDDFT/STRU @@ -0,0 +1,18 @@ +ATOMIC_SPECIES +Si 28 Si.pz-n-nc.UPF upf201 + +LATTICE_CONSTANT +10.2 + +LATTICE_VECTORS +0.0 0.5 0.5 +0.5 0.0 0.5 +0.5 0.5 0.0 + +ATOMIC_POSITIONS +Direct +Si +0.0 +2 +0.00 0.00 0.00 0 0 0 v 0 0 0 +0.25 0.25 0.25 0 0 0 v 0 0 0 diff --git a/tests/05_rtTDDFT/23_PW_spin_TDDFT/current_tot.txt.ref b/tests/05_rtTDDFT/23_PW_spin_TDDFT/current_tot.txt.ref new file mode 100644 index 00000000000..69dafa2d46a --- /dev/null +++ b/tests/05_rtTDDFT/23_PW_spin_TDDFT/current_tot.txt.ref @@ -0,0 +1,9 @@ +1 1.5025535501917469e-12 -8.8904842960136839e-13 1.1617423411051124e-12 +2 1.3797666509972984e-07 5.8049036859975030e-11 -1.6103222081221969e-11 +3 3.0486346906799782e-07 1.5631405417288439e-10 -6.1436137939717484e-11 +4 4.9018109890949085e-07 2.9946478049434098e-10 -1.4569551218743284e-10 +5 6.8117146315149763e-07 4.7250332772220869e-10 -2.6397355494475357e-10 +6 8.6108570707769328e-07 6.5023539402896511e-10 -4.0485208008971540e-10 +7 1.0125550888168468e-06 8.0823915159476320e-10 -5.6077585677627105e-10 +8 1.1224997542301526e-06 9.3482755759430696e-10 -7.3456197539074472e-10 +9 1.1841395591464872e-06 1.0386878203305518e-09 -9.3862300152470329e-10 diff --git a/tests/05_rtTDDFT/23_PW_spin_TDDFT/result.ref b/tests/05_rtTDDFT/23_PW_spin_TDDFT/result.ref new file mode 100644 index 00000000000..f3418be7773 --- /dev/null +++ b/tests/05_rtTDDFT/23_PW_spin_TDDFT/result.ref @@ -0,0 +1,4 @@ +etotref -249.7637335161314 +etotperatomref -124.8818667581 +CompareCurrent_pass 0 +totaltimeref 1.95 diff --git a/tests/05_rtTDDFT/24_PW_cgs_TDDFT/INPUT b/tests/05_rtTDDFT/24_PW_cgs_TDDFT/INPUT new file mode 100644 index 00000000000..d68f39c5199 --- /dev/null +++ b/tests/05_rtTDDFT/24_PW_cgs_TDDFT/INPUT @@ -0,0 +1,52 @@ +INPUT_PARAMETERS + +suffix autotest +pseudo_dir ../../PP_ORB +esolver_type tddft +basis_type pw +calculation md +device cpu +precision double +ecutwfc 20 +ecutrho 80 +nbands 6 +nspin 1 +kpar 2 +symmetry 0 +init_wfc atomic+random +pw_seed 1 +ks_solver cg +pw_diag_thr 1e-14 +scf_thr 1e-12 +scf_nmax 100 +mixing_type broyden +mixing_beta 0.5 +mixing_gg0 0 +smearing_method gauss +smearing_sigma 0.002 +md_type nve +init_vel 1 +md_dt 0.01 +md_nstep 9 +cal_force 0 +cal_stress 0 +td_stype 1 +td_vext 1 +td_vext_dire 1 +td_ttype 0 +td_tstart 0 +td_tend 100 +td_gauss_freq 2 +td_gauss_phase 0 +td_gauss_sigma 0.04 +td_gauss_t0 4 +td_gauss_amp 0.001 +lin_solver cgs +td_orthonormal lowdin +lin_precond kinetic +lin_thr 1e-14 +lin_maxiter 100 +td_cn_init 0 +lin_reconstruct 0 +out_current 1 +out_level ie diff --git a/tests/05_rtTDDFT/24_PW_cgs_TDDFT/KPT b/tests/05_rtTDDFT/24_PW_cgs_TDDFT/KPT new file mode 100644 index 00000000000..23f1466d0f8 --- /dev/null +++ b/tests/05_rtTDDFT/24_PW_cgs_TDDFT/KPT @@ -0,0 +1,5 @@ +K_POINTS +2 +Direct +0.17 0.11 0.07 0.5 +-0.17 -0.11 -0.07 0.5 diff --git a/tests/05_rtTDDFT/24_PW_cgs_TDDFT/README b/tests/05_rtTDDFT/24_PW_cgs_TDDFT/README new file mode 100644 index 00000000000..52939c4e083 --- /dev/null +++ b/tests/05_rtTDDFT/24_PW_cgs_TDDFT/README @@ -0,0 +1 @@ +Velocity-gauge PW RT-TDDFT with CGS and kinetic preconditioning: energy and current. diff --git a/tests/05_rtTDDFT/24_PW_cgs_TDDFT/STRU b/tests/05_rtTDDFT/24_PW_cgs_TDDFT/STRU new file mode 100644 index 00000000000..e059e5d6062 --- /dev/null +++ b/tests/05_rtTDDFT/24_PW_cgs_TDDFT/STRU @@ -0,0 +1,18 @@ +ATOMIC_SPECIES +Si 28 Si.pz-n-nc.UPF upf201 + +LATTICE_CONSTANT +10.2 + +LATTICE_VECTORS +0.0 0.5 0.5 +0.5 0.0 0.5 +0.5 0.5 0.0 + +ATOMIC_POSITIONS +Direct +Si +0.0 +2 +0.00 0.00 0.00 0 0 0 v 0 0 0 +0.25 0.25 0.25 0 0 0 v 0 0 0 diff --git a/tests/05_rtTDDFT/24_PW_cgs_TDDFT/current_tot.txt.ref b/tests/05_rtTDDFT/24_PW_cgs_TDDFT/current_tot.txt.ref new file mode 100644 index 00000000000..7a893ef3a7c --- /dev/null +++ b/tests/05_rtTDDFT/24_PW_cgs_TDDFT/current_tot.txt.ref @@ -0,0 +1,9 @@ +1 1.0605063946060377e-12 -4.2509106434665706e-13 1.5087168526571182e-12 +2 1.3751229921042474e-07 1.6621411528317294e-11 -6.2358041274252230e-12 +3 3.0372208643266064e-07 6.4089539875865666e-11 -2.3316796816792203e-12 +4 4.8810917649507548e-07 1.6195288755259469e-10 2.7263502786102940e-11 +5 6.7777087859469741e-07 3.2484685255404112e-10 9.8423691593466895e-11 +6 8.5563526471643574e-07 5.6892549479107466e-10 2.3151902614077338e-10 +7 1.0040847302052688e-06 9.1229062129060941e-10 4.4941574399019156e-10 +8 1.1100465588729078e-06 1.3739801550866658e-09 7.7301888788790556e-10 +9 1.1668032707619521e-06 1.9781422910292807e-09 1.2187167552518205e-09 diff --git a/tests/05_rtTDDFT/24_PW_cgs_TDDFT/result.ref b/tests/05_rtTDDFT/24_PW_cgs_TDDFT/result.ref new file mode 100644 index 00000000000..72bef06f26f --- /dev/null +++ b/tests/05_rtTDDFT/24_PW_cgs_TDDFT/result.ref @@ -0,0 +1,4 @@ +etotref -251.7684286794546 +etotperatomref -125.8842143397 +CompareCurrent_pass 0 +totaltimeref 0.96 diff --git a/tests/05_rtTDDFT/25_PW_noprec_TDDFT/INPUT b/tests/05_rtTDDFT/25_PW_noprec_TDDFT/INPUT new file mode 100644 index 00000000000..986c0efc74d --- /dev/null +++ b/tests/05_rtTDDFT/25_PW_noprec_TDDFT/INPUT @@ -0,0 +1,51 @@ +INPUT_PARAMETERS + +suffix autotest +pseudo_dir ../../PP_ORB +esolver_type tddft +basis_type pw +calculation md +device cpu +precision double +ecutwfc 20 +ecutrho 80 +nbands 6 +nspin 1 +kpar 2 +symmetry 0 +init_wfc atomic+random +pw_seed 1 +ks_solver cg +pw_diag_thr 1e-14 +scf_thr 1e-12 +scf_nmax 100 +mixing_type broyden +mixing_beta 0.5 +mixing_gg0 0 +smearing_method gauss +smearing_sigma 0.002 +md_type nve +init_vel 1 +md_dt 0.01 +md_nstep 9 +cal_force 0 +cal_stress 0 +td_stype 1 +td_vext 1 +td_vext_dire 1 +td_ttype 0 +td_tstart 0 +td_tend 100 +td_gauss_freq 2 +td_gauss_phase 0 +td_gauss_sigma 0.04 +td_gauss_t0 4 +td_gauss_amp 0.001 +lin_solver bicgstab +lin_precond none +lin_thr 1e-14 +lin_maxiter 100 +td_cn_init 0 +lin_reconstruct 0 +out_current 1 +out_level ie diff --git a/tests/05_rtTDDFT/25_PW_noprec_TDDFT/KPT b/tests/05_rtTDDFT/25_PW_noprec_TDDFT/KPT new file mode 100644 index 00000000000..23f1466d0f8 --- /dev/null +++ b/tests/05_rtTDDFT/25_PW_noprec_TDDFT/KPT @@ -0,0 +1,5 @@ +K_POINTS +2 +Direct +0.17 0.11 0.07 0.5 +-0.17 -0.11 -0.07 0.5 diff --git a/tests/05_rtTDDFT/25_PW_noprec_TDDFT/README b/tests/05_rtTDDFT/25_PW_noprec_TDDFT/README new file mode 100644 index 00000000000..6e973ef58ce --- /dev/null +++ b/tests/05_rtTDDFT/25_PW_noprec_TDDFT/README @@ -0,0 +1 @@ +Velocity-gauge PW RT-TDDFT with BiCGSTAB and no preconditioning: energy and current. diff --git a/tests/05_rtTDDFT/25_PW_noprec_TDDFT/STRU b/tests/05_rtTDDFT/25_PW_noprec_TDDFT/STRU new file mode 100644 index 00000000000..e059e5d6062 --- /dev/null +++ b/tests/05_rtTDDFT/25_PW_noprec_TDDFT/STRU @@ -0,0 +1,18 @@ +ATOMIC_SPECIES +Si 28 Si.pz-n-nc.UPF upf201 + +LATTICE_CONSTANT +10.2 + +LATTICE_VECTORS +0.0 0.5 0.5 +0.5 0.0 0.5 +0.5 0.5 0.0 + +ATOMIC_POSITIONS +Direct +Si +0.0 +2 +0.00 0.00 0.00 0 0 0 v 0 0 0 +0.25 0.25 0.25 0 0 0 v 0 0 0 diff --git a/tests/05_rtTDDFT/25_PW_noprec_TDDFT/current_tot.txt.ref b/tests/05_rtTDDFT/25_PW_noprec_TDDFT/current_tot.txt.ref new file mode 100644 index 00000000000..d43e525c342 --- /dev/null +++ b/tests/05_rtTDDFT/25_PW_noprec_TDDFT/current_tot.txt.ref @@ -0,0 +1,9 @@ +1 1.0605063946060377e-12 -4.2509106434665706e-13 1.5087168526571182e-12 +2 1.3751229918102684e-07 1.6621418223920190e-11 -6.2358032904748610e-12 +3 3.0372208635775354e-07 6.4089552430121092e-11 -2.3316704752252389e-12 +4 4.8810917641807606e-07 1.6195290303617638e-10 2.7263507389329932e-11 +5 6.7777087850033127e-07 3.2484685694803053e-10 9.8423689082615799e-11 +6 8.5563526460271511e-07 5.6892548244605687e-10 2.3151901358651796e-10 +7 1.0040847300863172e-06 9.1229059388048510e-10 4.4941572390338285e-10 +8 1.1100465587529101e-06 1.3739801153315235e-09 7.7301886277939466e-10 +9 1.1668032706442560e-06 1.9781422383014078e-09 1.2187167200999052e-09 diff --git a/tests/05_rtTDDFT/25_PW_noprec_TDDFT/result.ref b/tests/05_rtTDDFT/25_PW_noprec_TDDFT/result.ref new file mode 100644 index 00000000000..44d87fd6970 --- /dev/null +++ b/tests/05_rtTDDFT/25_PW_noprec_TDDFT/result.ref @@ -0,0 +1,4 @@ +etotref -251.7684286794546 +etotperatomref -125.8842143397 +CompareCurrent_pass 0 +totaltimeref 1.20 diff --git a/tests/05_rtTDDFT/26_PW_single_TDDFT/INPUT b/tests/05_rtTDDFT/26_PW_single_TDDFT/INPUT new file mode 100644 index 00000000000..ee2e0eff51a --- /dev/null +++ b/tests/05_rtTDDFT/26_PW_single_TDDFT/INPUT @@ -0,0 +1,53 @@ +INPUT_PARAMETERS + +suffix autotest +pseudo_dir ../../PP_ORB +esolver_type tddft +basis_type pw +calculation md +device cpu +precision single +ecutwfc 8 +ecutrho 32 +nbands 2 +nspin 1 +kpar 2 +symmetry 0 +init_wfc atomic +pw_seed 1 +ks_solver cg +pw_diag_thr 1e-8 +scf_thr 1e-8 +scf_nmax 100 +mixing_type broyden +mixing_beta 0.5 +mixing_gg0 0 +smearing_method gauss +smearing_sigma 0.002 +md_type nve +init_vel 1 +md_dt 0.01 +md_nstep 9 +cal_force 0 +cal_stress 0 +td_stype 1 +td_vext 1 +td_vext_dire 1 +td_ttype 0 +td_tstart 0 +td_tend 100 +td_gauss_freq 2 +td_gauss_phase 0 +td_gauss_sigma 0.04 +td_gauss_t0 4 +td_gauss_amp 0.1 +lin_solver bicgstab +td_orthonormal newton_schulz +td_out_stat 1 +lin_precond kinetic +lin_thr 1e-7 +lin_maxiter 100 +td_cn_init 0 +lin_reconstruct 0 +out_current 1 +out_level ie diff --git a/tests/08_EXX/12_KP_OXC/KPT b/tests/05_rtTDDFT/26_PW_single_TDDFT/KPT similarity index 100% rename from tests/08_EXX/12_KP_OXC/KPT rename to tests/05_rtTDDFT/26_PW_single_TDDFT/KPT diff --git a/tests/05_rtTDDFT/26_PW_single_TDDFT/README b/tests/05_rtTDDFT/26_PW_single_TDDFT/README new file mode 100644 index 00000000000..140375e8bdc --- /dev/null +++ b/tests/05_rtTDDFT/26_PW_single_TDDFT/README @@ -0,0 +1 @@ +Single-precision PW RT-TDDFT for H2 under an electric field: energy and current with kpar=2. diff --git a/tests/05_rtTDDFT/26_PW_single_TDDFT/STRU b/tests/05_rtTDDFT/26_PW_single_TDDFT/STRU new file mode 100644 index 00000000000..a8ca5da26fc --- /dev/null +++ b/tests/05_rtTDDFT/26_PW_single_TDDFT/STRU @@ -0,0 +1,18 @@ +ATOMIC_SPECIES +H 1 H.pz-vbc.UPF + +LATTICE_CONSTANT +40 + +LATTICE_VECTORS +1 0 0 +0 1 0 +0 0 1 + +ATOMIC_POSITIONS +Cartesian_au +H +0.0 +2 +19.3 20 20 0 0 0 v 0 0 0 +20.7 20 20 0 0 0 v 0 0 0 diff --git a/tests/05_rtTDDFT/26_PW_single_TDDFT/current_tot.txt.ref b/tests/05_rtTDDFT/26_PW_single_TDDFT/current_tot.txt.ref new file mode 100644 index 00000000000..b768122dc76 --- /dev/null +++ b/tests/05_rtTDDFT/26_PW_single_TDDFT/current_tot.txt.ref @@ -0,0 +1,9 @@ +1 -5.6024873629212379e-13 -3.0559021979570391e-13 2.8529001383503785e-13 +2 1.5215795428957790e-08 2.4738255888223650e-13 3.4182434660579021e-13 +3 3.3523581805638968e-08 1.4115357771515847e-12 4.5536907578025425e-13 +4 5.2988645620644091e-08 8.7311491370201108e-13 4.9864823381540191e-13 +5 7.1118054620455948e-08 1.0040821507573129e-12 3.9423753150913400e-13 +6 8.5259031038731338e-08 4.3655745685100556e-14 4.2547299017314801e-13 +7 9.3154805654194207e-08 0.0000000000000000e+00 4.4994408199272583e-13 +8 9.3486822152044622e-08 4.6566128730773927e-13 3.6639136169469564e-13 +9 8.5802006651647392e-08 1.1350493878126144e-12 3.1320723792305215e-13 diff --git a/tests/05_rtTDDFT/26_PW_single_TDDFT/result.ref b/tests/05_rtTDDFT/26_PW_single_TDDFT/result.ref new file mode 100644 index 00000000000..bf589c77c5e --- /dev/null +++ b/tests/05_rtTDDFT/26_PW_single_TDDFT/result.ref @@ -0,0 +1,4 @@ +etotref -28.58908359094873 +etotperatomref -14.2945417955 +CompareCurrent_pass 0 +totaltimeref 4.05 diff --git a/tests/05_rtTDDFT/26_PW_single_TDDFT/threshold b/tests/05_rtTDDFT/26_PW_single_TDDFT/threshold new file mode 100644 index 00000000000..8fcfcebb260 --- /dev/null +++ b/tests/05_rtTDDFT/26_PW_single_TDDFT/threshold @@ -0,0 +1 @@ +threshold 0.0001 diff --git a/tests/05_rtTDDFT/27_PW_gmres_recycle_TDDFT/INPUT b/tests/05_rtTDDFT/27_PW_gmres_recycle_TDDFT/INPUT new file mode 100644 index 00000000000..6b12e559ef0 --- /dev/null +++ b/tests/05_rtTDDFT/27_PW_gmres_recycle_TDDFT/INPUT @@ -0,0 +1,52 @@ +INPUT_PARAMETERS + +suffix autotest +pseudo_dir ../../PP_ORB +esolver_type tddft +basis_type pw +calculation md +device cpu +precision double +ecutwfc 20 +ecutrho 80 +nbands 6 +nspin 1 +kpar 2 +symmetry 0 +init_wfc atomic +pw_seed 1 +ks_solver cg +pw_diag_thr 1e-15 +scf_thr 1e-13 +scf_nmax 100 +mixing_type broyden +mixing_beta 0.5 +mixing_gg0 0 +smearing_method gauss +smearing_sigma 0.002 +md_type nve +init_vel 1 +md_dt 0.01 +md_nstep 11 +cal_force 0 +cal_stress 0 +td_stype 1 +td_vext 1 +td_vext_dire 1 +td_ttype 0 +td_tstart 0 +td_tend 100 +td_gauss_freq 2 +td_gauss_phase 0 +td_gauss_sigma 0.04 +td_gauss_t0 4 +td_gauss_amp 0.01 +lin_solver gmres +lin_precond kinetic_recycle +lin_thr 1e-11 +lin_maxiter 100 +out_current 1 +out_level ie +lin_gmres_restart 20 +td_cn_init 1 +lin_reconstruct 1 diff --git a/tests/05_rtTDDFT/27_PW_gmres_recycle_TDDFT/KPT b/tests/05_rtTDDFT/27_PW_gmres_recycle_TDDFT/KPT new file mode 100644 index 00000000000..03dfec74177 --- /dev/null +++ b/tests/05_rtTDDFT/27_PW_gmres_recycle_TDDFT/KPT @@ -0,0 +1,6 @@ +K_POINTS +3 +Direct +0.0 0.0 0.0 0.5 +0.17 0.11 0.07 0.25 +-0.17 -0.11 -0.07 0.25 diff --git a/tests/05_rtTDDFT/27_PW_gmres_recycle_TDDFT/README b/tests/05_rtTDDFT/27_PW_gmres_recycle_TDDFT/README new file mode 100644 index 00000000000..448c8ab9a8a --- /dev/null +++ b/tests/05_rtTDDFT/27_PW_gmres_recycle_TDDFT/README @@ -0,0 +1 @@ +Double-precision PW RT-TDDFT with GMRES(20), history preconditioning, CN initialization and residual reconstruction. Tests energy and current with three k points distributed over two pools (kpar=2). diff --git a/tests/05_rtTDDFT/27_PW_gmres_recycle_TDDFT/STRU b/tests/05_rtTDDFT/27_PW_gmres_recycle_TDDFT/STRU new file mode 100644 index 00000000000..e059e5d6062 --- /dev/null +++ b/tests/05_rtTDDFT/27_PW_gmres_recycle_TDDFT/STRU @@ -0,0 +1,18 @@ +ATOMIC_SPECIES +Si 28 Si.pz-n-nc.UPF upf201 + +LATTICE_CONSTANT +10.2 + +LATTICE_VECTORS +0.0 0.5 0.5 +0.5 0.0 0.5 +0.5 0.5 0.0 + +ATOMIC_POSITIONS +Direct +Si +0.0 +2 +0.00 0.00 0.00 0 0 0 v 0 0 0 +0.25 0.25 0.25 0 0 0 v 0 0 0 diff --git a/tests/05_rtTDDFT/27_PW_gmres_recycle_TDDFT/current_tot.txt.ref b/tests/05_rtTDDFT/27_PW_gmres_recycle_TDDFT/current_tot.txt.ref new file mode 100644 index 00000000000..4f4e4f67551 --- /dev/null +++ b/tests/05_rtTDDFT/27_PW_gmres_recycle_TDDFT/current_tot.txt.ref @@ -0,0 +1,11 @@ +1 4.2936108047451097e-13 -2.4637098567017647e-13 1.1469044666133506e-13 +2 1.3684983720511565e-06 8.1369010321866002e-11 -3.9066490646652952e-11 +3 3.0190089257789310e-06 3.1699090575680301e-10 -1.7795759597298859e-11 +4 4.8444606284476297e-06 8.1008643537411144e-10 1.3715835422850556e-10 +5 6.7137769288199204e-06 1.6376879822670157e-09 5.0720485712728539e-10 +6 8.4530928015672038e-06 2.8846612733306468e-09 1.1949993113429527e-09 +7 9.8830552413079795e-06 4.6476825547231378e-09 2.3164995244103979e-09 +8 1.0870499501461829e-05 7.0273996237572805e-09 3.9779609187124104e-09 +9 1.1344957566042592e-05 1.0147693859726738e-08 6.2631648888650860e-09 +10 1.1298352661174024e-05 1.4154858985090144e-08 9.2238949751129842e-09 +11 1.0783503747898042e-05 1.9160638454416877e-08 1.2861428365591574e-08 diff --git a/tests/05_rtTDDFT/27_PW_gmres_recycle_TDDFT/result.ref b/tests/05_rtTDDFT/27_PW_gmres_recycle_TDDFT/result.ref new file mode 100644 index 00000000000..067e455a734 --- /dev/null +++ b/tests/05_rtTDDFT/27_PW_gmres_recycle_TDDFT/result.ref @@ -0,0 +1,4 @@ +etotref -249.0536998379458 +etotperatomref -124.5268499190 +CompareCurrent_pass 0 +totaltimeref 1.33 diff --git a/tests/05_rtTDDFT/28_PW_gmres_subspace_TDDFT/INPUT b/tests/05_rtTDDFT/28_PW_gmres_subspace_TDDFT/INPUT new file mode 100644 index 00000000000..01877521657 --- /dev/null +++ b/tests/05_rtTDDFT/28_PW_gmres_subspace_TDDFT/INPUT @@ -0,0 +1,52 @@ +INPUT_PARAMETERS + +suffix autotest +pseudo_dir ../../PP_ORB +esolver_type tddft +basis_type pw +calculation md +device cpu +precision single +ecutwfc 8 +ecutrho 32 +nbands 2 +nspin 1 +kpar 2 +symmetry 0 +init_wfc atomic +pw_seed 1 +ks_solver cg +pw_diag_thr 1e-8 +scf_thr 1e-8 +scf_nmax 100 +mixing_type broyden +mixing_beta 0.5 +mixing_gg0 0 +smearing_method gauss +smearing_sigma 0.002 +md_type nve +init_vel 1 +md_dt 0.01 +md_nstep 9 +cal_force 0 +cal_stress 0 +td_stype 1 +td_vext 1 +td_vext_dire 1 +td_ttype 0 +td_tstart 0 +td_tend 100 +td_gauss_freq 2 +td_gauss_phase 0 +td_gauss_sigma 0.04 +td_gauss_t0 4 +td_gauss_amp 0.1 +lin_solver gmres +lin_precond kinetic_subspace +lin_thr 1e-6 +lin_maxiter 100 +out_current 1 +out_level ie +lin_gmres_restart 20 +td_cn_init 1 +lin_reconstruct 1 diff --git a/tests/08_EXX/14_NO_TDDFT_PBE0/KPT b/tests/05_rtTDDFT/28_PW_gmres_subspace_TDDFT/KPT similarity index 100% rename from tests/08_EXX/14_NO_TDDFT_PBE0/KPT rename to tests/05_rtTDDFT/28_PW_gmres_subspace_TDDFT/KPT diff --git a/tests/05_rtTDDFT/28_PW_gmres_subspace_TDDFT/README b/tests/05_rtTDDFT/28_PW_gmres_subspace_TDDFT/README new file mode 100644 index 00000000000..ea4ac160044 --- /dev/null +++ b/tests/05_rtTDDFT/28_PW_gmres_subspace_TDDFT/README @@ -0,0 +1 @@ +Single-precision PW RT-TDDFT with GMRES(20), coarse-space preconditioning and CN initialization. Tests energy and current with kpar=2 and residual reconstruction subject to independent residual checks. diff --git a/tests/05_rtTDDFT/28_PW_gmres_subspace_TDDFT/STRU b/tests/05_rtTDDFT/28_PW_gmres_subspace_TDDFT/STRU new file mode 100644 index 00000000000..a8ca5da26fc --- /dev/null +++ b/tests/05_rtTDDFT/28_PW_gmres_subspace_TDDFT/STRU @@ -0,0 +1,18 @@ +ATOMIC_SPECIES +H 1 H.pz-vbc.UPF + +LATTICE_CONSTANT +40 + +LATTICE_VECTORS +1 0 0 +0 1 0 +0 0 1 + +ATOMIC_POSITIONS +Cartesian_au +H +0.0 +2 +19.3 20 20 0 0 0 v 0 0 0 +20.7 20 20 0 0 0 v 0 0 0 diff --git a/tests/05_rtTDDFT/28_PW_gmres_subspace_TDDFT/current_tot.txt.ref b/tests/05_rtTDDFT/28_PW_gmres_subspace_TDDFT/current_tot.txt.ref new file mode 100644 index 00000000000..93a436a452f --- /dev/null +++ b/tests/05_rtTDDFT/28_PW_gmres_subspace_TDDFT/current_tot.txt.ref @@ -0,0 +1,9 @@ +1 -5.6024873629212379e-13 -3.0559021979570391e-13 2.8529001383503785e-13 +2 1.5216937754303218e-08 1.8917489796876906e-13 3.3454128356424919e-13 +3 3.3524207537993790e-08 1.0622898116707801e-12 4.0079584096019969e-13 +4 5.2989307732786984e-08 1.3242242857813835e-12 3.7205438729870367e-13 +5 7.1119036874733864e-08 7.5669959187507626e-13 3.7364245031312751e-13 +6 8.5259147454053163e-08 -1.3096723705530166e-13 4.5739056986349169e-13 +7 9.3145310529507696e-08 -1.6007106751203536e-13 3.9358027947855590e-13 +8 9.3477523478213694e-08 1.4551915228366852e-13 3.0836133646516830e-13 +9 8.5794068581890316e-08 1.3242242857813835e-12 2.4052937419583032e-13 diff --git a/tests/05_rtTDDFT/28_PW_gmres_subspace_TDDFT/result.ref b/tests/05_rtTDDFT/28_PW_gmres_subspace_TDDFT/result.ref new file mode 100644 index 00000000000..99001e33e8f --- /dev/null +++ b/tests/05_rtTDDFT/28_PW_gmres_subspace_TDDFT/result.ref @@ -0,0 +1,4 @@ +etotref -28.58910155231029 +etotperatomref -14.2945507762 +CompareCurrent_pass 0 +totaltimeref 3.27 diff --git a/tests/05_rtTDDFT/28_PW_gmres_subspace_TDDFT/threshold b/tests/05_rtTDDFT/28_PW_gmres_subspace_TDDFT/threshold new file mode 100644 index 00000000000..8fcfcebb260 --- /dev/null +++ b/tests/05_rtTDDFT/28_PW_gmres_subspace_TDDFT/threshold @@ -0,0 +1 @@ +threshold 0.0001 diff --git a/tests/05_rtTDDFT/CASES_CPU.txt b/tests/05_rtTDDFT/CASES_CPU.txt index f01c2b2353b..949a7e13a1a 100644 --- a/tests/05_rtTDDFT/CASES_CPU.txt +++ b/tests/05_rtTDDFT/CASES_CPU.txt @@ -18,3 +18,11 @@ 18_NO_hyb_TDDFT 19_NO_SUPERSINE_TDDFT 20_NO_MIXED_EFIELD_TDDFT +21_PW_len_TDDFT +22_PW_vel_TDDFT +23_PW_spin_TDDFT +24_PW_cgs_TDDFT +25_PW_noprec_TDDFT +26_PW_single_TDDFT +27_PW_gmres_recycle_TDDFT +28_PW_gmres_subspace_TDDFT diff --git a/tests/05_rtTDDFT/CMakeLists.txt b/tests/05_rtTDDFT/CMakeLists.txt index f0fbd317f1d..0c07dab375a 100644 --- a/tests/05_rtTDDFT/CMakeLists.txt +++ b/tests/05_rtTDDFT/CMakeLists.txt @@ -10,7 +10,7 @@ if(ENABLE_ASAN) else() add_test( NAME 05_rtTDDFT - COMMAND ${BASH} ../integrate/Autotest.sh -a ${ABACUS_BIN_PATH} -n 4 + COMMAND ${BASH} ../integrate/Autotest.sh -a ${ABACUS_BIN_PATH} -n 4 -j auto WORKING_DIRECTORY ${ABACUS_TEST_DIR}/05_rtTDDFT ) endif() diff --git a/tests/06_SDFT/17_PW_NLKG_100/refOnsager.txt b/tests/06_SDFT/17_PW_NLKG_100/refOnsager.txt index 0c095c441f0..d4a7f76e18c 100644 --- a/tests/06_SDFT/17_PW_NLKG_100/refOnsager.txt +++ b/tests/06_SDFT/17_PW_NLKG_100/refOnsager.txt @@ -1,1001 +1,1001 @@ ## w(eV) sigma(Sm^-1) kappa(W(mK)^-1) L12/e(Am^-1) L22/e^2(Wm^-1) - 0.01 205270 390.731 -4.17284e+06 1.21843e+08 - 0.03 205268 390.725 -4.17282e+06 1.21842e+08 - 0.05 205264 390.713 -4.17279e+06 1.21841e+08 - 0.07 205259 390.695 -4.17273e+06 1.21839e+08 - 0.09 205251 390.671 -4.17265e+06 1.21837e+08 - 0.11 205242 390.641 -4.17256e+06 1.21834e+08 - 0.13 205231 390.605 -4.17244e+06 1.21831e+08 - 0.15 205219 390.563 -4.17231e+06 1.21826e+08 - 0.17 205204 390.514 -4.17216e+06 1.21822e+08 - 0.19 205188 390.46 -4.17199e+06 1.21816e+08 - 0.21 205170 390.4 -4.1718e+06 1.21811e+08 - 0.23 205150 390.334 -4.17159e+06 1.21804e+08 - 0.25 205128 390.262 -4.17136e+06 1.21797e+08 - 0.27 205104 390.184 -4.17112e+06 1.21789e+08 - 0.29 205079 390.1 -4.17085e+06 1.21781e+08 - 0.31 205052 390.01 -4.17057e+06 1.21772e+08 - 0.33 205023 389.914 -4.17027e+06 1.21763e+08 - 0.35 204992 389.812 -4.16995e+06 1.21753e+08 - 0.37 204960 389.704 -4.16961e+06 1.21742e+08 - 0.39 204925 389.59 -4.16925e+06 1.21731e+08 - 0.41 204889 389.471 -4.16887e+06 1.21719e+08 - 0.43 204852 389.345 -4.16848e+06 1.21707e+08 - 0.45 204812 389.213 -4.16806e+06 1.21694e+08 - 0.47 204771 389.076 -4.16763e+06 1.21681e+08 - 0.49 204727 388.932 -4.16718e+06 1.21667e+08 - 0.51 204682 388.783 -4.16671e+06 1.21652e+08 - 0.53 204636 388.627 -4.16622e+06 1.21637e+08 - 0.55 204587 388.466 -4.16572e+06 1.21621e+08 - 0.57 204537 388.299 -4.16519e+06 1.21605e+08 - 0.59 204485 388.126 -4.16465e+06 1.21588e+08 - 0.61 204432 387.947 -4.16409e+06 1.2157e+08 - 0.63 204376 387.762 -4.16351e+06 1.21552e+08 - 0.65 204319 387.572 -4.16292e+06 1.21533e+08 - 0.67 204261 387.375 -4.1623e+06 1.21514e+08 - 0.69 204200 387.173 -4.16167e+06 1.21494e+08 - 0.71 204138 386.965 -4.16102e+06 1.21474e+08 - 0.73 204074 386.751 -4.16035e+06 1.21453e+08 - 0.75 204008 386.532 -4.15967e+06 1.21431e+08 - 0.77 203941 386.306 -4.15896e+06 1.21409e+08 - 0.79 203872 386.075 -4.15824e+06 1.21387e+08 - 0.81 203802 385.838 -4.15751e+06 1.21364e+08 - 0.83 203729 385.596 -4.15675e+06 1.2134e+08 - 0.85 203655 385.348 -4.15598e+06 1.21316e+08 - 0.87 203580 385.094 -4.15518e+06 1.21291e+08 - 0.89 203503 384.834 -4.15438e+06 1.21265e+08 - 0.91 203424 384.568 -4.15355e+06 1.21239e+08 - 0.93 203343 384.297 -4.15271e+06 1.21213e+08 - 0.95 203261 384.021 -4.15185e+06 1.21186e+08 - 0.97 203177 383.739 -4.15097e+06 1.21158e+08 - 0.99 203092 383.451 -4.15008e+06 1.2113e+08 - 1.01 203005 383.157 -4.14917e+06 1.21101e+08 - 1.03 202917 382.858 -4.14824e+06 1.21072e+08 - 1.05 202827 382.554 -4.14729e+06 1.21042e+08 - 1.07 202735 382.244 -4.14633e+06 1.21012e+08 - 1.09 202642 381.928 -4.14535e+06 1.20981e+08 - 1.11 202547 381.607 -4.14435e+06 1.20949e+08 - 1.13 202451 381.28 -4.14334e+06 1.20917e+08 - 1.15 202353 380.948 -4.14231e+06 1.20885e+08 - 1.17 202253 380.611 -4.14127e+06 1.20851e+08 - 1.19 202152 380.268 -4.1402e+06 1.20818e+08 - 1.21 202050 379.92 -4.13913e+06 1.20784e+08 - 1.23 201946 379.566 -4.13803e+06 1.20749e+08 - 1.25 201840 379.207 -4.13692e+06 1.20714e+08 - 1.27 201733 378.843 -4.13579e+06 1.20678e+08 - 1.29 201625 378.474 -4.13464e+06 1.20641e+08 - 1.31 201515 378.099 -4.13348e+06 1.20605e+08 - 1.33 201403 377.719 -4.13231e+06 1.20567e+08 - 1.35 201290 377.334 -4.13111e+06 1.20529e+08 - 1.37 201176 376.943 -4.1299e+06 1.20491e+08 - 1.39 201060 376.548 -4.12867e+06 1.20452e+08 - 1.41 200942 376.147 -4.12743e+06 1.20412e+08 - 1.43 200824 375.741 -4.12617e+06 1.20372e+08 - 1.45 200703 375.33 -4.1249e+06 1.20332e+08 - 1.47 200582 374.914 -4.1236e+06 1.20291e+08 - 1.49 200459 374.493 -4.12229e+06 1.20249e+08 - 1.51 200334 374.067 -4.12097e+06 1.20207e+08 - 1.53 200208 373.636 -4.11963e+06 1.20164e+08 - 1.55 200081 373.2 -4.11827e+06 1.20121e+08 - 1.57 199952 372.759 -4.1169e+06 1.20077e+08 - 1.59 199822 372.314 -4.11551e+06 1.20033e+08 - 1.61 199691 371.863 -4.1141e+06 1.19988e+08 - 1.63 199558 371.408 -4.11268e+06 1.19942e+08 - 1.65 199423 370.948 -4.11124e+06 1.19897e+08 - 1.67 199288 370.483 -4.10978e+06 1.1985e+08 - 1.69 199151 370.013 -4.10831e+06 1.19803e+08 - 1.71 199013 369.538 -4.10682e+06 1.19756e+08 - 1.73 198873 369.059 -4.10532e+06 1.19708e+08 - 1.75 198732 368.576 -4.1038e+06 1.19659e+08 - 1.77 198589 368.087 -4.10226e+06 1.1961e+08 - 1.79 198446 367.594 -4.1007e+06 1.19561e+08 - 1.81 198301 367.097 -4.09913e+06 1.1951e+08 - 1.83 198154 366.595 -4.09754e+06 1.1946e+08 - 1.85 198006 366.089 -4.09594e+06 1.19409e+08 - 1.87 197857 365.578 -4.09432e+06 1.19357e+08 - 1.89 197707 365.063 -4.09268e+06 1.19305e+08 - 1.91 197555 364.543 -4.09102e+06 1.19252e+08 - 1.93 197402 364.019 -4.08935e+06 1.19199e+08 - 1.95 197248 363.491 -4.08766e+06 1.19145e+08 - 1.97 197093 362.959 -4.08595e+06 1.1909e+08 - 1.99 196936 362.422 -4.08422e+06 1.19035e+08 - 2.01 196777 361.881 -4.08248e+06 1.1898e+08 - 2.03 196618 361.337 -4.08072e+06 1.18924e+08 - 2.05 196457 360.788 -4.07894e+06 1.18867e+08 - 2.07 196295 360.234 -4.07714e+06 1.1881e+08 - 2.09 196132 359.677 -4.07533e+06 1.18753e+08 - 2.11 195967 359.116 -4.0735e+06 1.18694e+08 - 2.13 195801 358.551 -4.07164e+06 1.18636e+08 - 2.15 195634 357.982 -4.06978e+06 1.18576e+08 - 2.17 195465 357.41 -4.06789e+06 1.18516e+08 - 2.19 195295 356.833 -4.06598e+06 1.18456e+08 - 2.21 195124 356.253 -4.06406e+06 1.18395e+08 - 2.23 194952 355.668 -4.06211e+06 1.18333e+08 - 2.25 194778 355.08 -4.06015e+06 1.18271e+08 - 2.27 194603 354.489 -4.05816e+06 1.18209e+08 - 2.29 194427 353.894 -4.05616e+06 1.18145e+08 - 2.31 194250 353.295 -4.05414e+06 1.18082e+08 - 2.33 194071 352.693 -4.0521e+06 1.18017e+08 - 2.35 193891 352.087 -4.05004e+06 1.17952e+08 - 2.37 193709 351.478 -4.04796e+06 1.17887e+08 - 2.39 193527 350.865 -4.04585e+06 1.17821e+08 - 2.41 193343 350.249 -4.04373e+06 1.17754e+08 - 2.43 193157 349.63 -4.04159e+06 1.17687e+08 - 2.45 192971 349.007 -4.03943e+06 1.17619e+08 - 2.47 192783 348.381 -4.03724e+06 1.1755e+08 - 2.49 192594 347.752 -4.03503e+06 1.17481e+08 - 2.51 192404 347.12 -4.03281e+06 1.17412e+08 - 2.53 192212 346.484 -4.03056e+06 1.17342e+08 - 2.55 192019 345.846 -4.02829e+06 1.17271e+08 - 2.57 191824 345.204 -4.02599e+06 1.17199e+08 - 2.59 191629 344.56 -4.02368e+06 1.17127e+08 - 2.61 191432 343.912 -4.02134e+06 1.17055e+08 - 2.63 191234 343.262 -4.01898e+06 1.16981e+08 - 2.65 191034 342.608 -4.0166e+06 1.16907e+08 - 2.67 190833 341.952 -4.01419e+06 1.16833e+08 - 2.69 190631 341.293 -4.01177e+06 1.16758e+08 - 2.71 190427 340.632 -4.00931e+06 1.16682e+08 - 2.73 190222 339.967 -4.00684e+06 1.16606e+08 - 2.75 190016 339.3 -4.00434e+06 1.16529e+08 - 2.77 189809 338.63 -4.00181e+06 1.16451e+08 - 2.79 189600 337.958 -3.99927e+06 1.16373e+08 - 2.81 189390 337.283 -3.99669e+06 1.16294e+08 - 2.83 189178 336.606 -3.9941e+06 1.16215e+08 - 2.85 188965 335.926 -3.99147e+06 1.16134e+08 - 2.87 188751 335.244 -3.98883e+06 1.16054e+08 - 2.89 188535 334.559 -3.98615e+06 1.15972e+08 - 2.91 188318 333.872 -3.98346e+06 1.1589e+08 - 2.93 188099 333.183 -3.98073e+06 1.15807e+08 - 2.95 187880 332.491 -3.97798e+06 1.15724e+08 - 2.97 187658 331.797 -3.97521e+06 1.1564e+08 - 2.99 187436 331.101 -3.97241e+06 1.15555e+08 - 3.01 187212 330.403 -3.96958e+06 1.15469e+08 - 3.03 186987 329.703 -3.96672e+06 1.15383e+08 - 3.05 186760 329.001 -3.96384e+06 1.15297e+08 - 3.07 186532 328.296 -3.96093e+06 1.15209e+08 - 3.09 186302 327.59 -3.95799e+06 1.15121e+08 - 3.11 186071 326.882 -3.95503e+06 1.15032e+08 - 3.13 185839 326.171 -3.95204e+06 1.14943e+08 - 3.15 185605 325.459 -3.94902e+06 1.14853e+08 - 3.17 185370 324.745 -3.94597e+06 1.14762e+08 - 3.19 185133 324.03 -3.94289e+06 1.1467e+08 - 3.21 184895 323.312 -3.93979e+06 1.14578e+08 - 3.23 184655 322.593 -3.93666e+06 1.14485e+08 - 3.25 184414 321.872 -3.9335e+06 1.14392e+08 - 3.27 184172 321.149 -3.93031e+06 1.14298e+08 - 3.29 183928 320.425 -3.92709e+06 1.14203e+08 - 3.31 183683 319.699 -3.92384e+06 1.14107e+08 - 3.33 183436 318.972 -3.92056e+06 1.14011e+08 - 3.35 183188 318.243 -3.91726e+06 1.13914e+08 - 3.37 182938 317.512 -3.91392e+06 1.13816e+08 - 3.39 182687 316.781 -3.91056e+06 1.13718e+08 - 3.41 182435 316.047 -3.90716e+06 1.13619e+08 - 3.43 182181 315.313 -3.90374e+06 1.13519e+08 - 3.45 181925 314.577 -3.90028e+06 1.13418e+08 - 3.47 181668 313.839 -3.8968e+06 1.13317e+08 - 3.49 181410 313.101 -3.89328e+06 1.13215e+08 - 3.51 181150 312.361 -3.88974e+06 1.13113e+08 - 3.53 180889 311.62 -3.88616e+06 1.1301e+08 - 3.55 180626 310.878 -3.88256e+06 1.12906e+08 - 3.57 180362 310.134 -3.87892e+06 1.12801e+08 - 3.59 180096 309.39 -3.87526e+06 1.12696e+08 - 3.61 179829 308.644 -3.87156e+06 1.1259e+08 - 3.63 179561 307.897 -3.86784e+06 1.12483e+08 - 3.65 179291 307.15 -3.86408e+06 1.12376e+08 - 3.67 179019 306.401 -3.86029e+06 1.12268e+08 - 3.69 178747 305.651 -3.85648e+06 1.12159e+08 - 3.71 178472 304.9 -3.85263e+06 1.1205e+08 - 3.73 178197 304.149 -3.84875e+06 1.11939e+08 - 3.75 177919 303.396 -3.84484e+06 1.11829e+08 - 3.77 177641 302.643 -3.84091e+06 1.11717e+08 - 3.79 177361 301.889 -3.83694e+06 1.11605e+08 - 3.81 177079 301.134 -3.83294e+06 1.11492e+08 - 3.83 176796 300.378 -3.82891e+06 1.11379e+08 - 3.85 176512 299.622 -3.82485e+06 1.11265e+08 - 3.87 176227 298.864 -3.82076e+06 1.1115e+08 - 3.89 175940 298.106 -3.81665e+06 1.11035e+08 - 3.91 175651 297.348 -3.8125e+06 1.10919e+08 - 3.93 175361 296.589 -3.80832e+06 1.10802e+08 - 3.95 175070 295.829 -3.80412e+06 1.10685e+08 - 3.97 174778 295.068 -3.79988e+06 1.10567e+08 - 3.99 174484 294.307 -3.79561e+06 1.10448e+08 - 4.01 174189 293.546 -3.79132e+06 1.10329e+08 - 4.03 173892 292.784 -3.787e+06 1.10209e+08 - 4.05 173594 292.022 -3.78265e+06 1.10088e+08 - 4.07 173295 291.259 -3.77827e+06 1.09967e+08 - 4.09 172995 290.495 -3.77386e+06 1.09846e+08 - 4.11 172693 289.732 -3.76942e+06 1.09723e+08 - 4.13 172390 288.967 -3.76496e+06 1.096e+08 - 4.15 172086 288.203 -3.76046e+06 1.09477e+08 - 4.17 171780 287.438 -3.75595e+06 1.09353e+08 - 4.19 171473 286.673 -3.7514e+06 1.09228e+08 - 4.21 171165 285.908 -3.74682e+06 1.09103e+08 - 4.23 170856 285.142 -3.74222e+06 1.08977e+08 - 4.25 170546 284.376 -3.7376e+06 1.08851e+08 - 4.27 170234 283.61 -3.73294e+06 1.08724e+08 - 4.29 169921 282.844 -3.72826e+06 1.08597e+08 - 4.31 169608 282.078 -3.72356e+06 1.08469e+08 - 4.33 169293 281.311 -3.71883e+06 1.0834e+08 - 4.35 168976 280.545 -3.71407e+06 1.08211e+08 - 4.37 168659 279.778 -3.70929e+06 1.08082e+08 - 4.39 168341 279.011 -3.70449e+06 1.07952e+08 - 4.41 168022 278.244 -3.69966e+06 1.07821e+08 - 4.43 167701 277.478 -3.6948e+06 1.0769e+08 - 4.45 167380 276.711 -3.68992e+06 1.07559e+08 - 4.47 167057 275.944 -3.68502e+06 1.07427e+08 - 4.49 166734 275.178 -3.6801e+06 1.07294e+08 - 4.51 166410 274.411 -3.67515e+06 1.07161e+08 - 4.53 166084 273.645 -3.67018e+06 1.07028e+08 - 4.55 165758 272.878 -3.66519e+06 1.06894e+08 - 4.57 165431 272.112 -3.66018e+06 1.0676e+08 - 4.59 165103 271.346 -3.65515e+06 1.06625e+08 - 4.61 164774 270.581 -3.65009e+06 1.0649e+08 - 4.63 164444 269.815 -3.64502e+06 1.06355e+08 - 4.65 164113 269.05 -3.63992e+06 1.06219e+08 - 4.67 163782 268.285 -3.6348e+06 1.06082e+08 - 4.69 163449 267.52 -3.62967e+06 1.05946e+08 - 4.71 163116 266.756 -3.62451e+06 1.05809e+08 - 4.73 162783 265.992 -3.61934e+06 1.05671e+08 - 4.75 162448 265.229 -3.61415e+06 1.05533e+08 - 4.77 162113 264.465 -3.60894e+06 1.05395e+08 - 4.79 161777 263.703 -3.60371e+06 1.05257e+08 - 4.81 161440 262.94 -3.59846e+06 1.05118e+08 - 4.83 161103 262.178 -3.5932e+06 1.04979e+08 - 4.85 160765 261.417 -3.58792e+06 1.04839e+08 - 4.87 160427 260.656 -3.58263e+06 1.04699e+08 - 4.89 160088 259.896 -3.57731e+06 1.04559e+08 - 4.91 159748 259.136 -3.57199e+06 1.04419e+08 - 4.93 159408 258.377 -3.56664e+06 1.04278e+08 - 4.95 159067 257.618 -3.56129e+06 1.04137e+08 - 4.97 158726 256.86 -3.55591e+06 1.03996e+08 - 4.99 158384 256.103 -3.55053e+06 1.03854e+08 - 5.01 158042 255.346 -3.54513e+06 1.03712e+08 - 5.03 157699 254.59 -3.53971e+06 1.0357e+08 - 5.05 157356 253.835 -3.53428e+06 1.03428e+08 - 5.07 157012 253.08 -3.52884e+06 1.03285e+08 - 5.09 156669 252.327 -3.52339e+06 1.03143e+08 - 5.11 156324 251.574 -3.51793e+06 1.03e+08 - 5.13 155980 250.821 -3.51245e+06 1.02856e+08 - 5.15 155635 250.07 -3.50696e+06 1.02713e+08 - 5.17 155290 249.32 -3.50146e+06 1.02569e+08 - 5.19 154944 248.57 -3.49595e+06 1.02426e+08 - 5.21 154598 247.821 -3.49043e+06 1.02281e+08 - 5.23 154252 247.073 -3.48489e+06 1.02137e+08 - 5.25 153906 246.326 -3.47935e+06 1.01993e+08 - 5.27 153559 245.58 -3.4738e+06 1.01848e+08 - 5.29 153213 244.835 -3.46824e+06 1.01704e+08 - 5.31 152866 244.091 -3.46267e+06 1.01559e+08 - 5.33 152519 243.348 -3.45709e+06 1.01414e+08 - 5.35 152171 242.605 -3.4515e+06 1.01268e+08 - 5.37 151824 241.864 -3.4459e+06 1.01123e+08 - 5.39 151476 241.124 -3.44029e+06 1.00977e+08 - 5.41 151129 240.385 -3.43468e+06 1.00832e+08 - 5.43 150781 239.648 -3.42906e+06 1.00686e+08 - 5.45 150433 238.911 -3.42343e+06 1.0054e+08 - 5.47 150085 238.175 -3.41779e+06 1.00394e+08 - 5.49 149737 237.441 -3.41215e+06 1.00248e+08 - 5.51 149389 236.707 -3.4065e+06 1.00102e+08 - 5.53 149041 235.975 -3.40084e+06 9.99553e+07 - 5.55 148693 235.244 -3.39517e+06 9.98088e+07 - 5.57 148345 234.514 -3.3895e+06 9.96622e+07 - 5.59 147997 233.786 -3.38382e+06 9.95154e+07 - 5.61 147649 233.059 -3.37814e+06 9.93686e+07 - 5.63 147301 232.333 -3.37245e+06 9.92217e+07 - 5.65 146953 231.608 -3.36675e+06 9.90747e+07 - 5.67 146605 230.884 -3.36105e+06 9.89275e+07 - 5.69 146257 230.162 -3.35534e+06 9.87803e+07 - 5.71 145909 229.441 -3.34963e+06 9.8633e+07 - 5.73 145561 228.722 -3.34391e+06 9.84856e+07 - 5.75 145213 228.003 -3.33819e+06 9.83381e+07 - 5.77 144865 227.286 -3.33246e+06 9.81906e+07 - 5.79 144518 226.571 -3.32672e+06 9.80429e+07 - 5.81 144170 225.857 -3.32098e+06 9.78951e+07 - 5.83 143823 225.144 -3.31523e+06 9.77473e+07 - 5.85 143476 224.433 -3.30948e+06 9.75994e+07 - 5.87 143128 223.723 -3.30373e+06 9.74513e+07 - 5.89 142781 223.014 -3.29797e+06 9.73032e+07 - 5.91 142434 222.307 -3.2922e+06 9.7155e+07 - 5.93 142087 221.601 -3.28643e+06 9.70067e+07 - 5.95 141740 220.897 -3.28065e+06 9.68583e+07 - 5.97 141394 220.194 -3.27487e+06 9.67099e+07 - 5.99 141047 219.493 -3.26908e+06 9.65613e+07 - 6.01 140701 218.793 -3.26329e+06 9.64126e+07 - 6.03 140355 218.094 -3.25749e+06 9.62639e+07 - 6.05 140008 217.398 -3.25169e+06 9.6115e+07 - 6.07 139662 216.702 -3.24588e+06 9.59661e+07 - 6.09 139316 216.008 -3.24007e+06 9.5817e+07 - 6.11 138971 215.316 -3.23425e+06 9.56679e+07 - 6.13 138625 214.625 -3.22843e+06 9.55186e+07 - 6.15 138279 213.935 -3.2226e+06 9.53692e+07 - 6.17 137934 213.247 -3.21676e+06 9.52197e+07 - 6.19 137589 212.561 -3.21092e+06 9.50701e+07 - 6.21 137243 211.876 -3.20507e+06 9.49204e+07 - 6.23 136898 211.193 -3.19922e+06 9.47705e+07 - 6.25 136553 210.511 -3.19336e+06 9.46205e+07 - 6.27 136208 209.831 -3.1875e+06 9.44704e+07 - 6.29 135864 209.152 -3.18163e+06 9.43202e+07 - 6.31 135519 208.475 -3.17575e+06 9.41698e+07 - 6.33 135174 207.799 -3.16986e+06 9.40192e+07 - 6.35 134830 207.125 -3.16397e+06 9.38686e+07 - 6.37 134485 206.452 -3.15807e+06 9.37177e+07 - 6.39 134141 205.781 -3.15217e+06 9.35667e+07 - 6.41 133797 205.112 -3.14625e+06 9.34156e+07 - 6.43 133452 204.444 -3.14033e+06 9.32643e+07 - 6.45 133108 203.778 -3.13441e+06 9.31128e+07 - 6.47 132764 203.113 -3.12847e+06 9.29611e+07 - 6.49 132420 202.45 -3.12253e+06 9.28092e+07 - 6.51 132076 201.788 -3.11657e+06 9.26572e+07 - 6.53 131732 201.128 -3.11061e+06 9.2505e+07 - 6.55 131388 200.469 -3.10465e+06 9.23525e+07 - 6.57 131044 199.812 -3.09867e+06 9.21999e+07 - 6.59 130700 199.156 -3.09268e+06 9.2047e+07 - 6.61 130356 198.502 -3.08669e+06 9.1894e+07 - 6.63 130012 197.849 -3.08068e+06 9.17407e+07 - 6.65 129668 197.198 -3.07467e+06 9.15872e+07 - 6.67 129324 196.549 -3.06864e+06 9.14334e+07 - 6.69 128980 195.901 -3.06261e+06 9.12794e+07 - 6.71 128636 195.254 -3.05657e+06 9.11252e+07 - 6.73 128292 194.609 -3.05051e+06 9.09707e+07 - 6.75 127948 193.966 -3.04445e+06 9.08159e+07 - 6.77 127603 193.324 -3.03837e+06 9.06609e+07 - 6.79 127259 192.683 -3.03228e+06 9.05056e+07 - 6.81 126915 192.044 -3.02619e+06 9.035e+07 - 6.83 126570 191.407 -3.02008e+06 9.01942e+07 - 6.85 126226 190.771 -3.01396e+06 9.0038e+07 - 6.87 125881 190.136 -3.00782e+06 8.98816e+07 - 6.89 125536 189.503 -3.00168e+06 8.97248e+07 - 6.91 125191 188.872 -2.99552e+06 8.95678e+07 - 6.93 124846 188.242 -2.98935e+06 8.94104e+07 - 6.95 124501 187.613 -2.98317e+06 8.92527e+07 - 6.97 124156 186.986 -2.97697e+06 8.90946e+07 - 6.99 123810 186.36 -2.97076e+06 8.89362e+07 - 7.01 123465 185.735 -2.96454e+06 8.87775e+07 - 7.03 123119 185.112 -2.95831e+06 8.86184e+07 - 7.05 122773 184.491 -2.95206e+06 8.8459e+07 - 7.07 122427 183.871 -2.94579e+06 8.82992e+07 - 7.09 122081 183.252 -2.93951e+06 8.8139e+07 - 7.11 121734 182.635 -2.93322e+06 8.79785e+07 - 7.13 121387 182.019 -2.92692e+06 8.78175e+07 - 7.15 121040 181.404 -2.92059e+06 8.76562e+07 - 7.17 120693 180.791 -2.91426e+06 8.74945e+07 - 7.19 120346 180.179 -2.90791e+06 8.73324e+07 - 7.21 119998 179.569 -2.90154e+06 8.71698e+07 - 7.23 119650 178.96 -2.89516e+06 8.70069e+07 - 7.25 119302 178.352 -2.88876e+06 8.68435e+07 - 7.27 118954 177.746 -2.88235e+06 8.66797e+07 - 7.29 118605 177.14 -2.87592e+06 8.65155e+07 - 7.31 118257 176.537 -2.86947e+06 8.63509e+07 - 7.33 117908 175.934 -2.86301e+06 8.61858e+07 - 7.35 117558 175.333 -2.85653e+06 8.60202e+07 - 7.37 117209 174.734 -2.85004e+06 8.58542e+07 - 7.39 116859 174.135 -2.84353e+06 8.56878e+07 - 7.41 116509 173.538 -2.837e+06 8.55209e+07 - 7.43 116158 172.942 -2.83045e+06 8.53535e+07 - 7.45 115807 172.348 -2.82389e+06 8.51856e+07 - 7.47 115456 171.754 -2.81731e+06 8.50173e+07 - 7.49 115105 171.162 -2.81071e+06 8.48485e+07 - 7.51 114754 170.572 -2.8041e+06 8.46792e+07 - 7.53 114402 169.982 -2.79747e+06 8.45094e+07 - 7.55 114049 169.394 -2.79082e+06 8.43391e+07 - 7.57 113697 168.807 -2.78415e+06 8.41684e+07 - 7.59 113344 168.221 -2.77747e+06 8.39971e+07 - 7.61 112991 167.637 -2.77077e+06 8.38253e+07 - 7.63 112638 167.054 -2.76405e+06 8.3653e+07 - 7.65 112284 166.472 -2.75731e+06 8.34802e+07 - 7.67 111930 165.891 -2.75055e+06 8.33069e+07 - 7.69 111576 165.312 -2.74378e+06 8.31331e+07 - 7.71 111221 164.733 -2.73699e+06 8.29587e+07 - 7.73 110867 164.156 -2.73018e+06 8.27839e+07 - 7.75 110511 163.58 -2.72335e+06 8.26085e+07 - 7.77 110156 163.006 -2.71651e+06 8.24326e+07 - 7.79 109800 162.432 -2.70965e+06 8.22561e+07 - 7.81 109444 161.86 -2.70277e+06 8.20792e+07 - 7.83 109088 161.289 -2.69587e+06 8.19017e+07 - 7.85 108731 160.719 -2.68895e+06 8.17237e+07 - 7.87 108375 160.151 -2.68202e+06 8.15451e+07 - 7.89 108018 159.583 -2.67507e+06 8.1366e+07 - 7.91 107660 159.017 -2.6681e+06 8.11864e+07 - 7.93 107303 158.452 -2.66111e+06 8.10063e+07 - 7.95 106945 157.888 -2.65411e+06 8.08256e+07 - 7.97 106587 157.326 -2.64708e+06 8.06444e+07 - 7.99 106228 156.764 -2.64005e+06 8.04627e+07 - 8.01 105870 156.204 -2.63299e+06 8.02804e+07 - 8.03 105511 155.645 -2.62592e+06 8.00976e+07 - 8.05 105152 155.087 -2.61883e+06 7.99143e+07 - 8.07 104792 154.531 -2.61172e+06 7.97305e+07 - 8.09 104433 153.975 -2.6046e+06 7.95461e+07 - 8.11 104073 153.421 -2.59746e+06 7.93612e+07 - 8.13 103713 152.868 -2.5903e+06 7.91758e+07 - 8.15 103353 152.316 -2.58313e+06 7.89899e+07 - 8.17 102993 151.765 -2.57594e+06 7.88034e+07 - 8.19 102633 151.216 -2.56874e+06 7.86165e+07 - 8.21 102272 150.667 -2.56152e+06 7.8429e+07 - 8.23 101912 150.12 -2.55428e+06 7.82411e+07 - 8.25 101551 149.574 -2.54703e+06 7.80526e+07 - 8.27 101190 149.03 -2.53977e+06 7.78637e+07 - 8.29 100829 148.486 -2.53249e+06 7.76742e+07 - 8.31 100468 147.944 -2.52519e+06 7.74843e+07 - 8.33 100106 147.403 -2.51789e+06 7.72939e+07 - 8.35 99745.2 146.863 -2.51056e+06 7.7103e+07 - 8.37 99383.8 146.324 -2.50323e+06 7.69116e+07 - 8.39 99022.4 145.787 -2.49588e+06 7.67198e+07 - 8.41 98661 145.251 -2.48851e+06 7.65275e+07 - 8.43 98299.5 144.716 -2.48114e+06 7.63348e+07 - 8.45 97938 144.182 -2.47375e+06 7.61416e+07 - 8.47 97576.6 143.65 -2.46635e+06 7.5948e+07 - 8.49 97215.1 143.119 -2.45894e+06 7.57539e+07 - 8.51 96853.7 142.589 -2.45152e+06 7.55594e+07 - 8.53 96492.3 142.06 -2.44408e+06 7.53645e+07 - 8.55 96131 141.532 -2.43664e+06 7.51692e+07 - 8.57 95769.7 141.006 -2.42918e+06 7.49735e+07 - 8.59 95408.6 140.481 -2.42171e+06 7.47773e+07 - 8.61 95047.5 139.958 -2.41424e+06 7.45808e+07 - 8.63 94686.6 139.435 -2.40675e+06 7.4384e+07 - 8.65 94325.8 138.914 -2.39925e+06 7.41867e+07 - 8.67 93965.2 138.394 -2.39175e+06 7.39891e+07 - 8.69 93604.7 137.876 -2.38424e+06 7.37911e+07 - 8.71 93244.4 137.359 -2.37672e+06 7.35928e+07 - 8.73 92884.4 136.843 -2.36919e+06 7.33941e+07 - 8.75 92524.5 136.328 -2.36166e+06 7.31951e+07 - 8.77 92164.9 135.815 -2.35411e+06 7.29958e+07 - 8.79 91805.5 135.303 -2.34657e+06 7.27963e+07 - 8.81 91446.4 134.793 -2.33901e+06 7.25964e+07 - 8.83 91087.6 134.284 -2.33145e+06 7.23962e+07 - 8.85 90729.1 133.776 -2.32389e+06 7.21957e+07 - 8.87 90370.9 133.269 -2.31632e+06 7.1995e+07 - 8.89 90013.1 132.764 -2.30874e+06 7.17941e+07 - 8.91 89655.6 132.261 -2.30117e+06 7.15929e+07 - 8.93 89298.5 131.758 -2.29359e+06 7.13914e+07 - 8.95 88941.8 131.257 -2.286e+06 7.11898e+07 - 8.97 88585.5 130.758 -2.27842e+06 7.09879e+07 - 8.99 88229.6 130.26 -2.27083e+06 7.07859e+07 - 9.01 87874.1 129.763 -2.26324e+06 7.05836e+07 - 9.03 87519.2 129.268 -2.25565e+06 7.03812e+07 - 9.05 87164.7 128.774 -2.24806e+06 7.01787e+07 - 9.07 86810.7 128.282 -2.24047e+06 6.9976e+07 - 9.09 86457.3 127.791 -2.23288e+06 6.97731e+07 - 9.11 86104.4 127.302 -2.22529e+06 6.95702e+07 - 9.13 85752 126.814 -2.2177e+06 6.93671e+07 - 9.15 85400.2 126.327 -2.21011e+06 6.91639e+07 - 9.17 85049 125.842 -2.20253e+06 6.89607e+07 - 9.19 84698.4 125.359 -2.19495e+06 6.87574e+07 - 9.21 84348.5 124.877 -2.18737e+06 6.8554e+07 - 9.23 83999.2 124.396 -2.1798e+06 6.83506e+07 - 9.25 83650.6 123.917 -2.17223e+06 6.81472e+07 - 9.27 83302.6 123.44 -2.16466e+06 6.79437e+07 - 9.29 82955.4 122.964 -2.1571e+06 6.77403e+07 - 9.31 82608.8 122.49 -2.14955e+06 6.75368e+07 - 9.33 82263.1 122.017 -2.142e+06 6.73334e+07 - 9.35 81918 121.546 -2.13446e+06 6.713e+07 - 9.37 81573.8 121.076 -2.12693e+06 6.69267e+07 - 9.39 81230.3 120.608 -2.1194e+06 6.67234e+07 - 9.41 80887.7 120.141 -2.11188e+06 6.65202e+07 - 9.43 80545.9 119.677 -2.10438e+06 6.63171e+07 - 9.45 80204.9 119.213 -2.09688e+06 6.61141e+07 - 9.47 79864.8 118.752 -2.08939e+06 6.59113e+07 - 9.49 79525.5 118.292 -2.08191e+06 6.57085e+07 - 9.51 79187.2 117.833 -2.07444e+06 6.55059e+07 - 9.53 78849.8 117.376 -2.06698e+06 6.53035e+07 - 9.55 78513.3 116.921 -2.05953e+06 6.51012e+07 - 9.57 78177.8 116.468 -2.0521e+06 6.48992e+07 - 9.59 77843.2 116.016 -2.04468e+06 6.46973e+07 - 9.61 77509.6 115.566 -2.03727e+06 6.44957e+07 - 9.63 77177 115.117 -2.02988e+06 6.42943e+07 - 9.65 76845.5 114.67 -2.0225e+06 6.40931e+07 - 9.67 76514.9 114.225 -2.01513e+06 6.38922e+07 - 9.69 76185.4 113.781 -2.00778e+06 6.36915e+07 - 9.71 75857 113.34 -2.00044e+06 6.34912e+07 - 9.73 75529.6 112.9 -1.99312e+06 6.32911e+07 - 9.75 75203.4 112.461 -1.98582e+06 6.30913e+07 - 9.77 74878.2 112.025 -1.97853e+06 6.28919e+07 - 9.79 74554.2 111.59 -1.97127e+06 6.26928e+07 - 9.81 74231.3 111.156 -1.96401e+06 6.24941e+07 - 9.83 73909.6 110.725 -1.95678e+06 6.22957e+07 - 9.85 73589 110.295 -1.94957e+06 6.20977e+07 - 9.87 73269.6 109.867 -1.94237e+06 6.19001e+07 - 9.89 72951.4 109.441 -1.93519e+06 6.17029e+07 - 9.91 72634.4 109.016 -1.92804e+06 6.15061e+07 - 9.93 72318.6 108.593 -1.9209e+06 6.13097e+07 - 9.95 72004.1 108.172 -1.91379e+06 6.11138e+07 - 9.97 71690.8 107.753 -1.90669e+06 6.09183e+07 - 9.99 71378.8 107.336 -1.89962e+06 6.07233e+07 - 10.01 71068 106.92 -1.89257e+06 6.05287e+07 - 10.03 70758.6 106.506 -1.88554e+06 6.03347e+07 - 10.05 70450.4 106.094 -1.87854e+06 6.01411e+07 - 10.07 70143.5 105.684 -1.87155e+06 5.99481e+07 - 10.09 69838 105.275 -1.8646e+06 5.97556e+07 - 10.11 69533.8 104.868 -1.85766e+06 5.95636e+07 - 10.13 69230.9 104.463 -1.85075e+06 5.93722e+07 - 10.15 68929.4 104.06 -1.84386e+06 5.91813e+07 - 10.17 68629.3 103.659 -1.837e+06 5.8991e+07 - 10.19 68330.5 103.259 -1.83017e+06 5.88013e+07 - 10.21 68033.1 102.862 -1.82336e+06 5.86121e+07 - 10.23 67737.2 102.466 -1.81657e+06 5.84236e+07 - 10.25 67442.6 102.072 -1.80981e+06 5.82357e+07 - 10.27 67149.4 101.679 -1.80308e+06 5.80484e+07 - 10.29 66857.7 101.289 -1.79638e+06 5.78617e+07 - 10.31 66567.4 100.9 -1.7897e+06 5.76757e+07 - 10.33 66278.5 100.513 -1.78305e+06 5.74903e+07 - 10.35 65991.1 100.128 -1.77643e+06 5.73056e+07 - 10.37 65705.1 99.745 -1.76984e+06 5.71215e+07 - 10.39 65420.6 99.3637 -1.76327e+06 5.69382e+07 - 10.41 65137.6 98.9841 -1.75674e+06 5.67555e+07 - 10.43 64856.1 98.6064 -1.75023e+06 5.65735e+07 - 10.45 64576 98.2305 -1.74375e+06 5.63923e+07 - 10.47 64297.5 97.8564 -1.7373e+06 5.62117e+07 - 10.49 64020.4 97.4842 -1.73088e+06 5.60319e+07 - 10.51 63744.9 97.1137 -1.7245e+06 5.58528e+07 - 10.53 63470.8 96.7451 -1.71814e+06 5.56744e+07 - 10.55 63198.3 96.3784 -1.71181e+06 5.54968e+07 - 10.57 62927.3 96.0135 -1.70551e+06 5.532e+07 - 10.59 62657.9 95.6503 -1.69925e+06 5.51439e+07 - 10.61 62390 95.2891 -1.69301e+06 5.49685e+07 - 10.63 62123.6 94.9296 -1.68681e+06 5.4794e+07 - 10.65 61858.7 94.572 -1.68064e+06 5.46202e+07 - 10.67 61595.4 94.2162 -1.6745e+06 5.44473e+07 - 10.69 61333.7 93.8622 -1.66839e+06 5.42751e+07 - 10.71 61073.5 93.51 -1.66232e+06 5.41038e+07 - 10.73 60814.9 93.1597 -1.65627e+06 5.39332e+07 - 10.75 60557.9 92.8112 -1.65026e+06 5.37635e+07 - 10.77 60302.4 92.4645 -1.64428e+06 5.35946e+07 - 10.79 60048.5 92.1196 -1.63834e+06 5.34265e+07 - 10.81 59796.1 91.7765 -1.63243e+06 5.32592e+07 - 10.83 59545.3 91.4353 -1.62655e+06 5.30928e+07 - 10.85 59296.1 91.0958 -1.6207e+06 5.29273e+07 - 10.87 59048.5 90.7582 -1.61489e+06 5.27626e+07 - 10.89 58802.5 90.4223 -1.60911e+06 5.25987e+07 - 10.91 58558 90.0883 -1.60337e+06 5.24357e+07 - 10.93 58315.2 89.7561 -1.59765e+06 5.22736e+07 - 10.95 58073.9 89.4256 -1.59198e+06 5.21123e+07 - 10.97 57834.2 89.097 -1.58633e+06 5.19519e+07 - 10.99 57596.1 88.7701 -1.58073e+06 5.17924e+07 - 11.01 57359.6 88.4451 -1.57515e+06 5.16338e+07 - 11.03 57124.7 88.1218 -1.56961e+06 5.14761e+07 - 11.05 56891.3 87.8003 -1.5641e+06 5.13192e+07 - 11.07 56659.6 87.4806 -1.55863e+06 5.11632e+07 - 11.09 56429.4 87.1627 -1.55319e+06 5.10082e+07 - 11.11 56200.8 86.8465 -1.54779e+06 5.0854e+07 - 11.13 55973.8 86.5321 -1.54242e+06 5.07007e+07 - 11.15 55748.4 86.2195 -1.53709e+06 5.05483e+07 - 11.17 55524.6 85.9087 -1.53179e+06 5.03968e+07 - 11.19 55302.3 85.5996 -1.52653e+06 5.02463e+07 - 11.21 55081.7 85.2922 -1.5213e+06 5.00966e+07 - 11.23 54862.6 84.9866 -1.5161e+06 4.99479e+07 - 11.25 54645.1 84.6828 -1.51094e+06 4.98001e+07 - 11.27 54429.2 84.3807 -1.50582e+06 4.96531e+07 - 11.29 54214.8 84.0803 -1.50073e+06 4.95071e+07 - 11.31 54002 83.7817 -1.49568e+06 4.93621e+07 - 11.33 53790.8 83.4848 -1.49066e+06 4.92179e+07 - 11.35 53581.1 83.1896 -1.48567e+06 4.90746e+07 - 11.37 53373.1 82.8962 -1.48072e+06 4.89323e+07 - 11.39 53166.5 82.6045 -1.4758e+06 4.87909e+07 - 11.41 52961.6 82.3145 -1.47092e+06 4.86504e+07 - 11.43 52758.2 82.0262 -1.46608e+06 4.85109e+07 - 11.45 52556.3 81.7396 -1.46127e+06 4.83723e+07 - 11.47 52356 81.4547 -1.45649e+06 4.82345e+07 - 11.49 52157.2 81.1715 -1.45175e+06 4.80978e+07 - 11.51 51960 80.89 -1.44704e+06 4.79619e+07 - 11.53 51764.3 80.6101 -1.44237e+06 4.78269e+07 - 11.55 51570.1 80.332 -1.43773e+06 4.76929e+07 - 11.57 51377.5 80.0555 -1.43313e+06 4.75598e+07 - 11.59 51186.3 79.7808 -1.42856e+06 4.74276e+07 - 11.61 50996.7 79.5076 -1.42403e+06 4.72964e+07 - 11.63 50808.6 79.2362 -1.41953e+06 4.71661e+07 - 11.65 50622.1 78.9664 -1.41506e+06 4.70366e+07 - 11.67 50437 78.6982 -1.41063e+06 4.69081e+07 - 11.69 50253.4 78.4317 -1.40624e+06 4.67805e+07 - 11.71 50071.3 78.1668 -1.40187e+06 4.66539e+07 - 11.73 49890.7 77.9036 -1.39754e+06 4.65281e+07 - 11.75 49711.6 77.642 -1.39325e+06 4.64033e+07 - 11.77 49534 77.3821 -1.38899e+06 4.62793e+07 - 11.79 49357.8 77.1237 -1.38476e+06 4.61563e+07 - 11.81 49183.1 76.867 -1.38057e+06 4.60342e+07 - 11.83 49009.9 76.6119 -1.37641e+06 4.5913e+07 - 11.85 48838.1 76.3583 -1.37228e+06 4.57927e+07 - 11.87 48667.8 76.1064 -1.36819e+06 4.56732e+07 - 11.89 48498.9 75.8561 -1.36413e+06 4.55547e+07 - 11.91 48331.4 75.6074 -1.3601e+06 4.54371e+07 - 11.93 48165.4 75.3602 -1.3561e+06 4.53204e+07 - 11.95 48000.8 75.1146 -1.35214e+06 4.52045e+07 - 11.97 47837.6 74.8706 -1.34821e+06 4.50896e+07 - 11.99 47675.8 74.6282 -1.34432e+06 4.49755e+07 - 12.01 47515.4 74.3873 -1.34045e+06 4.48623e+07 - 12.03 47356.4 74.148 -1.33662e+06 4.475e+07 - 12.05 47198.8 73.9103 -1.33282e+06 4.46386e+07 - 12.07 47042.6 73.674 -1.32906e+06 4.4528e+07 - 12.09 46887.8 73.4394 -1.32532e+06 4.44183e+07 - 12.11 46734.3 73.2062 -1.32162e+06 4.43095e+07 - 12.13 46582.2 72.9746 -1.31795e+06 4.42015e+07 - 12.15 46431.5 72.7445 -1.31431e+06 4.40944e+07 - 12.17 46282.1 72.5159 -1.3107e+06 4.39882e+07 - 12.19 46134.1 72.2888 -1.30712e+06 4.38828e+07 - 12.21 45987.3 72.0633 -1.30357e+06 4.37782e+07 - 12.23 45842 71.8392 -1.30006e+06 4.36745e+07 - 12.25 45697.9 71.6166 -1.29657e+06 4.35716e+07 - 12.27 45555.1 71.3955 -1.29312e+06 4.34696e+07 - 12.29 45413.7 71.1759 -1.28969e+06 4.33684e+07 - 12.31 45273.5 70.9578 -1.2863e+06 4.3268e+07 - 12.33 45134.6 70.7411 -1.28294e+06 4.31684e+07 - 12.35 44997 70.5259 -1.2796e+06 4.30697e+07 - 12.37 44860.7 70.3122 -1.2763e+06 4.29718e+07 - 12.39 44725.7 70.0999 -1.27302e+06 4.28747e+07 - 12.41 44591.9 69.8891 -1.26978e+06 4.27783e+07 - 12.43 44459.4 69.6797 -1.26656e+06 4.26828e+07 - 12.45 44328.1 69.4717 -1.26337e+06 4.25881e+07 - 12.47 44198 69.2652 -1.26022e+06 4.24942e+07 - 12.49 44069.2 69.06 -1.25709e+06 4.2401e+07 - 12.51 43941.6 68.8563 -1.25399e+06 4.23087e+07 - 12.53 43815.2 68.654 -1.25091e+06 4.22171e+07 - 12.55 43690 68.4531 -1.24787e+06 4.21263e+07 - 12.57 43566 68.2537 -1.24485e+06 4.20363e+07 - 12.59 43443.2 68.0556 -1.24186e+06 4.1947e+07 - 12.61 43321.5 67.8588 -1.2389e+06 4.18585e+07 - 12.63 43201.1 67.6635 -1.23597e+06 4.17707e+07 - 12.65 43081.8 67.4696 -1.23306e+06 4.16837e+07 - 12.67 42963.6 67.277 -1.23019e+06 4.15974e+07 - 12.69 42846.7 67.0857 -1.22733e+06 4.15119e+07 - 12.71 42730.8 66.8959 -1.22451e+06 4.14271e+07 - 12.73 42616.1 66.7073 -1.22171e+06 4.1343e+07 - 12.75 42502.5 66.5201 -1.21893e+06 4.12596e+07 - 12.77 42390 66.3343 -1.21619e+06 4.1177e+07 - 12.79 42278.6 66.1498 -1.21347e+06 4.1095e+07 - 12.81 42168.4 65.9666 -1.21077e+06 4.10138e+07 - 12.83 42059.2 65.7847 -1.2081e+06 4.09333e+07 - 12.85 41951.1 65.6042 -1.20546e+06 4.08535e+07 - 12.87 41844.1 65.4249 -1.20284e+06 4.07743e+07 - 12.89 41738.1 65.247 -1.20024e+06 4.06959e+07 - 12.91 41633.2 65.0703 -1.19767e+06 4.06181e+07 - 12.93 41529.4 64.8949 -1.19513e+06 4.0541e+07 - 12.95 41426.6 64.7209 -1.19261e+06 4.04646e+07 - 12.97 41324.8 64.548 -1.19011e+06 4.03888e+07 - 12.99 41224 64.3765 -1.18764e+06 4.03137e+07 - 13.01 41124.3 64.2062 -1.18519e+06 4.02392e+07 - 13.03 41025.6 64.0372 -1.18276e+06 4.01654e+07 - 13.05 40927.9 63.8694 -1.18036e+06 4.00922e+07 - 13.07 40831.2 63.7029 -1.17798e+06 4.00197e+07 - 13.09 40735.4 63.5376 -1.17563e+06 3.99478e+07 - 13.11 40640.7 63.3736 -1.17329e+06 3.98765e+07 - 13.13 40546.9 63.2108 -1.17098e+06 3.98058e+07 - 13.15 40454.1 63.0492 -1.1687e+06 3.97358e+07 - 13.17 40362.2 62.8888 -1.16643e+06 3.96663e+07 - 13.19 40271.3 62.7296 -1.16419e+06 3.95975e+07 - 13.21 40181.3 62.5716 -1.16196e+06 3.95293e+07 - 13.23 40092.3 62.4148 -1.15976e+06 3.94616e+07 - 13.25 40004.2 62.2592 -1.15758e+06 3.93945e+07 - 13.27 39916.9 62.1048 -1.15543e+06 3.93281e+07 - 13.29 39830.6 61.9516 -1.15329e+06 3.92621e+07 - 13.31 39745.2 61.7995 -1.15117e+06 3.91968e+07 - 13.33 39660.7 61.6487 -1.14908e+06 3.9132e+07 - 13.35 39577.1 61.4989 -1.147e+06 3.90678e+07 - 13.37 39494.3 61.3504 -1.14495e+06 3.90041e+07 - 13.39 39412.5 61.2029 -1.14291e+06 3.8941e+07 - 13.41 39331.4 61.0566 -1.1409e+06 3.88785e+07 - 13.43 39251.3 60.9115 -1.1389e+06 3.88164e+07 - 13.45 39171.9 60.7675 -1.13693e+06 3.87549e+07 - 13.47 39093.4 60.6246 -1.13497e+06 3.8694e+07 - 13.49 39015.8 60.4828 -1.13303e+06 3.86335e+07 - 13.51 38939 60.3421 -1.13112e+06 3.85736e+07 - 13.53 38862.9 60.2026 -1.12922e+06 3.85141e+07 - 13.55 38787.7 60.0641 -1.12734e+06 3.84552e+07 - 13.57 38713.3 59.9268 -1.12547e+06 3.83968e+07 - 13.59 38639.7 59.7905 -1.12363e+06 3.83389e+07 - 13.61 38566.9 59.6553 -1.1218e+06 3.82814e+07 - 13.63 38494.8 59.5212 -1.11999e+06 3.82245e+07 - 13.65 38423.5 59.3881 -1.1182e+06 3.8168e+07 - 13.67 38353 59.2562 -1.11643e+06 3.8112e+07 - 13.69 38283.2 59.1253 -1.11467e+06 3.80565e+07 - 13.71 38214.2 58.9954 -1.11294e+06 3.80015e+07 - 13.73 38146 58.8666 -1.11121e+06 3.79469e+07 - 13.75 38078.4 58.7388 -1.10951e+06 3.78927e+07 - 13.77 38011.6 58.6121 -1.10782e+06 3.78391e+07 - 13.79 37945.5 58.4864 -1.10615e+06 3.77858e+07 - 13.81 37880.2 58.3617 -1.10449e+06 3.7733e+07 - 13.83 37815.5 58.2381 -1.10285e+06 3.76806e+07 - 13.85 37751.5 58.1154 -1.10123e+06 3.76287e+07 - 13.87 37688.3 57.9938 -1.09962e+06 3.75772e+07 - 13.89 37625.7 57.8732 -1.09803e+06 3.75261e+07 - 13.91 37563.8 57.7535 -1.09645e+06 3.74755e+07 - 13.93 37502.6 57.6349 -1.09489e+06 3.74252e+07 - 13.95 37442 57.5172 -1.09334e+06 3.73754e+07 - 13.97 37382.1 57.4006 -1.09181e+06 3.73259e+07 - 13.99 37322.9 57.2849 -1.09029e+06 3.72769e+07 - 14.01 37264.3 57.1702 -1.08879e+06 3.72283e+07 - 14.03 37206.3 57.0564 -1.0873e+06 3.718e+07 - 14.05 37149 56.9436 -1.08583e+06 3.71321e+07 - 14.07 37092.3 56.8318 -1.08437e+06 3.70846e+07 - 14.09 37036.2 56.7209 -1.08292e+06 3.70375e+07 - 14.11 36980.7 56.6109 -1.08149e+06 3.69908e+07 - 14.13 36925.9 56.5019 -1.08007e+06 3.69445e+07 - 14.15 36871.6 56.3938 -1.07867e+06 3.68985e+07 - 14.17 36818 56.2867 -1.07728e+06 3.68528e+07 - 14.19 36764.9 56.1805 -1.0759e+06 3.68075e+07 - 14.21 36712.5 56.0752 -1.07453e+06 3.67626e+07 - 14.23 36660.6 55.9708 -1.07318e+06 3.6718e+07 - 14.25 36609.2 55.8673 -1.07184e+06 3.66738e+07 - 14.27 36558.5 55.7647 -1.07052e+06 3.66299e+07 - 14.29 36508.3 55.663 -1.0692e+06 3.65864e+07 - 14.31 36458.6 55.5622 -1.0679e+06 3.65432e+07 - 14.33 36409.5 55.4623 -1.06661e+06 3.65003e+07 - 14.35 36361 55.3633 -1.06533e+06 3.64577e+07 - 14.37 36313 55.2652 -1.06407e+06 3.64155e+07 - 14.39 36265.5 55.1679 -1.06281e+06 3.63736e+07 - 14.41 36218.5 55.0715 -1.06157e+06 3.6332e+07 - 14.43 36172.1 54.976 -1.06034e+06 3.62907e+07 - 14.45 36126.2 54.8813 -1.05912e+06 3.62497e+07 - 14.47 36080.8 54.7874 -1.05792e+06 3.6209e+07 - 14.49 36035.8 54.6945 -1.05672e+06 3.61687e+07 - 14.51 35991.4 54.6023 -1.05553e+06 3.61286e+07 - 14.53 35947.5 54.511 -1.05436e+06 3.60888e+07 - 14.55 35904.1 54.4206 -1.05319e+06 3.60493e+07 - 14.57 35861.1 54.3309 -1.05204e+06 3.60101e+07 - 14.59 35818.7 54.2421 -1.0509e+06 3.59712e+07 - 14.61 35776.7 54.1541 -1.04977e+06 3.59326e+07 - 14.63 35735.2 54.0669 -1.04864e+06 3.58942e+07 - 14.65 35694.1 53.9806 -1.04753e+06 3.58562e+07 - 14.67 35653.5 53.895 -1.04643e+06 3.58184e+07 - 14.69 35613.3 53.8102 -1.04534e+06 3.57808e+07 - 14.71 35573.6 53.7263 -1.04426e+06 3.57436e+07 - 14.73 35534.4 53.6431 -1.04318e+06 3.57066e+07 - 14.75 35495.5 53.5607 -1.04212e+06 3.56698e+07 - 14.77 35457.1 53.4791 -1.04107e+06 3.56333e+07 - 14.79 35419.2 53.3982 -1.04002e+06 3.55971e+07 - 14.81 35381.6 53.3182 -1.03899e+06 3.55611e+07 - 14.83 35344.5 53.2389 -1.03796e+06 3.55254e+07 - 14.85 35307.8 53.1604 -1.03695e+06 3.54899e+07 - 14.87 35271.5 53.0826 -1.03594e+06 3.54547e+07 - 14.89 35235.6 53.0056 -1.03494e+06 3.54197e+07 - 14.91 35200.1 52.9293 -1.03395e+06 3.53849e+07 - 14.93 35165 52.8538 -1.03297e+06 3.53504e+07 - 14.95 35130.3 52.779 -1.032e+06 3.53161e+07 - 14.97 35096 52.705 -1.03103e+06 3.52821e+07 - 14.99 35062.1 52.6317 -1.03008e+06 3.52482e+07 - 15.01 35028.5 52.5591 -1.02913e+06 3.52146e+07 - 15.03 34995.3 52.4873 -1.02819e+06 3.51812e+07 - 15.05 34962.5 52.4161 -1.02726e+06 3.51481e+07 - 15.07 34930.1 52.3457 -1.02633e+06 3.51151e+07 - 15.09 34898 52.276 -1.02542e+06 3.50824e+07 - 15.11 34866.2 52.207 -1.02451e+06 3.50499e+07 - 15.13 34834.9 52.1387 -1.02361e+06 3.50175e+07 - 15.15 34803.9 52.071 -1.02272e+06 3.49854e+07 - 15.17 34773.2 52.0041 -1.02183e+06 3.49535e+07 - 15.19 34742.8 51.9379 -1.02095e+06 3.49218e+07 - 15.21 34712.8 51.8723 -1.02008e+06 3.48903e+07 - 15.23 34683.2 51.8075 -1.01922e+06 3.4859e+07 - 15.25 34653.9 51.7433 -1.01836e+06 3.48279e+07 - 15.27 34624.9 51.6797 -1.01751e+06 3.4797e+07 - 15.29 34596.2 51.6169 -1.01667e+06 3.47663e+07 - 15.31 34567.8 51.5547 -1.01583e+06 3.47358e+07 - 15.33 34539.8 51.4931 -1.015e+06 3.47054e+07 - 15.35 34512 51.4322 -1.01418e+06 3.46753e+07 - 15.37 34484.6 51.372 -1.01336e+06 3.46453e+07 - 15.39 34457.5 51.3124 -1.01256e+06 3.46155e+07 - 15.41 34430.7 51.2534 -1.01175e+06 3.45859e+07 - 15.43 34404.2 51.1951 -1.01096e+06 3.45565e+07 - 15.45 34378 51.1374 -1.01017e+06 3.45272e+07 - 15.47 34352 51.0803 -1.00938e+06 3.44981e+07 - 15.49 34326.4 51.0238 -1.0086e+06 3.44692e+07 - 15.51 34301 50.968 -1.00783e+06 3.44404e+07 - 15.53 34276 50.9127 -1.00707e+06 3.44118e+07 - 15.55 34251.2 50.8581 -1.00631e+06 3.43834e+07 - 15.57 34226.6 50.8041 -1.00555e+06 3.43551e+07 - 15.59 34202.4 50.7506 -1.0048e+06 3.4327e+07 - 15.61 34178.4 50.6978 -1.00406e+06 3.4299e+07 - 15.63 34154.7 50.6455 -1.00332e+06 3.42712e+07 - 15.65 34131.3 50.5939 -1.00259e+06 3.42436e+07 - 15.67 34108.1 50.5428 -1.00187e+06 3.42161e+07 - 15.69 34085.1 50.4922 -1.00114e+06 3.41887e+07 - 15.71 34062.5 50.4423 -1.00043e+06 3.41615e+07 - 15.73 34040 50.3929 -999718 3.41345e+07 - 15.75 34017.9 50.3441 -999013 3.41076e+07 - 15.77 33995.9 50.2958 -998313 3.40808e+07 - 15.79 33974.2 50.2481 -997618 3.40541e+07 - 15.81 33952.8 50.2009 -996928 3.40276e+07 - 15.83 33931.6 50.1542 -996243 3.40013e+07 - 15.85 33910.6 50.1081 -995563 3.3975e+07 - 15.87 33889.9 50.0626 -994887 3.39489e+07 - 15.89 33869.4 50.0175 -994217 3.3923e+07 - 15.91 33849.1 49.973 -993550 3.38971e+07 - 15.93 33829 49.9289 -992889 3.38714e+07 - 15.95 33809.2 49.8854 -992232 3.38458e+07 - 15.97 33789.5 49.8424 -991579 3.38203e+07 - 15.99 33770.1 49.7999 -990931 3.3795e+07 - 16.01 33750.9 49.7579 -990287 3.37697e+07 - 16.03 33732 49.7164 -989648 3.37446e+07 - 16.05 33713.2 49.6754 -989012 3.37196e+07 - 16.07 33694.6 49.6348 -988381 3.36947e+07 - 16.09 33676.3 49.5947 -987754 3.36699e+07 - 16.11 33658.1 49.5551 -987131 3.36453e+07 - 16.13 33640.1 49.516 -986512 3.36207e+07 - 16.15 33622.4 49.4773 -985897 3.35962e+07 - 16.17 33604.8 49.439 -985286 3.35718e+07 - 16.19 33587.4 49.4012 -984678 3.35476e+07 - 16.21 33570.2 49.3639 -984074 3.35234e+07 - 16.23 33553.2 49.327 -983474 3.34994e+07 - 16.25 33536.4 49.2905 -982878 3.34754e+07 - 16.27 33519.8 49.2544 -982285 3.34515e+07 - 16.29 33503.4 49.2188 -981696 3.34277e+07 - 16.31 33487.1 49.1836 -981110 3.3404e+07 - 16.33 33471 49.1488 -980528 3.33804e+07 - 16.35 33455.1 49.1143 -979949 3.33569e+07 - 16.37 33439.3 49.0803 -979373 3.33335e+07 - 16.39 33423.8 49.0467 -978801 3.33101e+07 - 16.41 33408.4 49.0135 -978232 3.32868e+07 - 16.43 33393.1 48.9806 -977666 3.32636e+07 - 16.45 33378.1 48.9481 -977103 3.32405e+07 - 16.47 33363.2 48.916 -976544 3.32175e+07 - 16.49 33348.4 48.8842 -975987 3.31945e+07 - 16.51 33333.8 48.8528 -975433 3.31716e+07 - 16.53 33319.4 48.8218 -974882 3.31488e+07 - 16.55 33305.1 48.7911 -974334 3.3126e+07 - 16.57 33291 48.7607 -973789 3.31034e+07 - 16.59 33277.1 48.7307 -973247 3.30807e+07 - 16.61 33263.2 48.701 -972708 3.30582e+07 - 16.63 33249.6 48.6716 -972171 3.30357e+07 - 16.65 33236 48.6425 -971637 3.30133e+07 - 16.67 33222.7 48.6138 -971105 3.29909e+07 - 16.69 33209.4 48.5853 -970576 3.29686e+07 - 16.71 33196.3 48.5571 -970050 3.29463e+07 - 16.73 33183.4 48.5293 -969525 3.29241e+07 - 16.75 33170.6 48.5017 -969004 3.2902e+07 - 16.77 33157.9 48.4743 -968485 3.28799e+07 - 16.79 33145.3 48.4473 -967968 3.28578e+07 - 16.81 33132.9 48.4205 -967453 3.28358e+07 - 16.83 33120.6 48.394 -966941 3.28139e+07 - 16.85 33108.5 48.3677 -966431 3.2792e+07 - 16.87 33096.4 48.3417 -965923 3.27701e+07 - 16.89 33084.5 48.3159 -965417 3.27483e+07 - 16.91 33072.7 48.2903 -964914 3.27265e+07 - 16.93 33061.1 48.265 -964412 3.27048e+07 - 16.95 33049.5 48.2398 -963913 3.26831e+07 - 16.97 33038.1 48.2149 -963415 3.26614e+07 - 16.99 33026.8 48.1902 -962920 3.26398e+07 - 17.01 33015.6 48.1657 -962426 3.26182e+07 - 17.03 33004.6 48.1414 -961934 3.25966e+07 - 17.05 32993.6 48.1172 -961444 3.25751e+07 - 17.07 32982.8 48.0933 -960956 3.25536e+07 - 17.09 32972 48.0695 -960470 3.25321e+07 - 17.11 32961.4 48.0459 -959985 3.25106e+07 - 17.13 32950.9 48.0224 -959502 3.24892e+07 - 17.15 32940.4 47.9991 -959021 3.24678e+07 - 17.17 32930.1 47.9759 -958541 3.24464e+07 - 17.19 32919.9 47.9529 -958063 3.24251e+07 - 17.21 32909.8 47.93 -957587 3.24038e+07 - 17.23 32899.8 47.9072 -957112 3.23824e+07 - 17.25 32889.9 47.8845 -956639 3.23611e+07 - 17.27 32880 47.8619 -956167 3.23399e+07 - 17.29 32870.3 47.8395 -955696 3.23186e+07 - 17.31 32860.7 47.8171 -955227 3.22973e+07 - 17.33 32851.1 47.7949 -954759 3.22761e+07 - 17.35 32841.7 47.7727 -954293 3.22549e+07 - 17.37 32832.3 47.7506 -953828 3.22337e+07 - 17.39 32823 47.7285 -953364 3.22125e+07 - 17.41 32813.8 47.7065 -952902 3.21913e+07 - 17.43 32804.7 47.6846 -952440 3.21701e+07 - 17.45 32795.7 47.6627 -951980 3.21489e+07 - 17.47 32786.8 47.6409 -951521 3.21277e+07 - 17.49 32777.9 47.6191 -951064 3.21066e+07 - 17.51 32769.1 47.5973 -950607 3.20854e+07 - 17.53 32760.4 47.5755 -950151 3.20642e+07 - 17.55 32751.8 47.5538 -949697 3.20431e+07 - 17.57 32743.2 47.532 -949243 3.20219e+07 - 17.59 32734.7 47.5103 -948791 3.20007e+07 - 17.61 32726.3 47.4886 -948340 3.19796e+07 - 17.63 32718 47.4668 -947889 3.19584e+07 - 17.65 32709.7 47.445 -947439 3.19372e+07 - 17.67 32701.5 47.4232 -946991 3.19161e+07 - 17.69 32693.4 47.4014 -946543 3.18949e+07 - 17.71 32685.3 47.3795 -946096 3.18737e+07 - 17.73 32677.3 47.3576 -945650 3.18525e+07 - 17.75 32669.4 47.3357 -945205 3.18313e+07 - 17.77 32661.5 47.3136 -944760 3.18101e+07 - 17.79 32653.7 47.2916 -944317 3.17889e+07 - 17.81 32645.9 47.2694 -943874 3.17677e+07 - 17.83 32638.2 47.2472 -943431 3.17464e+07 - 17.85 32630.6 47.2249 -942990 3.17252e+07 - 17.87 32623 47.2024 -942549 3.17039e+07 - 17.89 32615.4 47.1799 -942109 3.16826e+07 - 17.91 32607.9 47.1573 -941669 3.16614e+07 - 17.93 32600.5 47.1346 -941230 3.16401e+07 - 17.95 32593.1 47.1118 -940792 3.16187e+07 - 17.97 32585.8 47.0889 -940354 3.15974e+07 - 17.99 32578.5 47.0658 -939917 3.15761e+07 - 18.01 32571.3 47.0426 -939480 3.15547e+07 - 18.03 32564.1 47.0193 -939044 3.15333e+07 - 18.05 32556.9 46.9959 -938608 3.15119e+07 - 18.07 32549.8 46.9723 -938173 3.14905e+07 - 18.09 32542.7 46.9485 -937738 3.1469e+07 - 18.11 32535.7 46.9246 -937304 3.14476e+07 - 18.13 32528.7 46.9005 -936870 3.14261e+07 - 18.15 32521.8 46.8763 -936436 3.14046e+07 - 18.17 32514.9 46.8519 -936003 3.13831e+07 - 18.19 32508 46.8273 -935570 3.13615e+07 - 18.21 32501.1 46.8025 -935138 3.13399e+07 - 18.23 32494.3 46.7776 -934706 3.13184e+07 - 18.25 32487.6 46.7524 -934274 3.12967e+07 - 18.27 32480.8 46.7271 -933842 3.12751e+07 - 18.29 32474.1 46.7016 -933411 3.12534e+07 - 18.31 32467.4 46.6758 -932980 3.12317e+07 - 18.33 32460.8 46.6499 -932549 3.121e+07 - 18.35 32454.1 46.6237 -932119 3.11883e+07 - 18.37 32447.5 46.5974 -931688 3.11665e+07 - 18.39 32440.9 46.5708 -931258 3.11447e+07 - 18.41 32434.4 46.544 -930828 3.11229e+07 - 18.43 32427.9 46.517 -930398 3.1101e+07 - 18.45 32421.3 46.4897 -929968 3.10791e+07 - 18.47 32414.9 46.4622 -929538 3.10572e+07 - 18.49 32408.4 46.4345 -929109 3.10353e+07 - 18.51 32401.9 46.4066 -928679 3.10133e+07 - 18.53 32395.5 46.3784 -928250 3.09913e+07 - 18.55 32389.1 46.3499 -927820 3.09693e+07 - 18.57 32382.7 46.3213 -927391 3.09472e+07 - 18.59 32376.3 46.2923 -926962 3.09251e+07 - 18.61 32369.9 46.2632 -926532 3.0903e+07 - 18.63 32363.6 46.2337 -926103 3.08808e+07 - 18.65 32357.2 46.204 -925674 3.08586e+07 - 18.67 32350.9 46.1741 -925244 3.08364e+07 - 18.69 32344.6 46.1439 -924815 3.08142e+07 - 18.71 32338.3 46.1134 -924385 3.07919e+07 - 18.73 32332 46.0827 -923956 3.07696e+07 - 18.75 32325.7 46.0517 -923526 3.07472e+07 - 18.77 32319.4 46.0205 -923096 3.07248e+07 - 18.79 32313.1 45.9889 -922666 3.07024e+07 - 18.81 32306.8 45.9572 -922236 3.06799e+07 - 18.83 32300.5 45.9251 -921806 3.06575e+07 - 18.85 32294.3 45.8928 -921375 3.06349e+07 - 18.87 32288 45.8602 -920944 3.06124e+07 - 18.89 32281.7 45.8273 -920514 3.05898e+07 - 18.91 32275.5 45.7942 -920082 3.05671e+07 - 18.93 32269.2 45.7608 -919651 3.05445e+07 - 18.95 32263 45.7272 -919220 3.05218e+07 - 18.97 32256.7 45.6932 -918788 3.0499e+07 - 18.99 32250.4 45.659 -918356 3.04763e+07 - 19.01 32244.2 45.6245 -917923 3.04535e+07 - 19.03 32237.9 45.5898 -917491 3.04306e+07 - 19.05 32231.7 45.5548 -917058 3.04077e+07 - 19.07 32225.4 45.5195 -916625 3.03848e+07 - 19.09 32219.1 45.484 -916191 3.03619e+07 - 19.11 32212.8 45.4482 -915757 3.03389e+07 - 19.13 32206.6 45.4121 -915323 3.03158e+07 - 19.15 32200.3 45.3757 -914888 3.02928e+07 - 19.17 32194 45.3391 -914453 3.02697e+07 - 19.19 32187.7 45.3023 -914018 3.02465e+07 - 19.21 32181.4 45.2652 -913582 3.02233e+07 - 19.23 32175 45.2278 -913146 3.02001e+07 - 19.25 32168.7 45.1902 -912709 3.01769e+07 - 19.27 32162.4 45.1523 -912272 3.01536e+07 - 19.29 32156 45.1141 -911834 3.01302e+07 - 19.31 32149.7 45.0757 -911396 3.01069e+07 - 19.33 32143.3 45.0371 -910958 3.00835e+07 - 19.35 32137 44.9982 -910519 3.006e+07 - 19.37 32130.6 44.9591 -910079 3.00366e+07 - 19.39 32124.2 44.9198 -909640 3.0013e+07 - 19.41 32117.8 44.8802 -909199 2.99895e+07 - 19.43 32111.4 44.8403 -908758 2.99659e+07 - 19.45 32104.9 44.8003 -908317 2.99422e+07 - 19.47 32098.5 44.76 -907875 2.99186e+07 - 19.49 32092 44.7195 -907432 2.98949e+07 - 19.51 32085.6 44.6788 -906989 2.98711e+07 - 19.53 32079.1 44.6378 -906545 2.98473e+07 - 19.55 32072.6 44.5967 -906101 2.98235e+07 - 19.57 32066.1 44.5553 -905656 2.97997e+07 - 19.59 32059.6 44.5137 -905210 2.97758e+07 - 19.61 32053 44.4719 -904764 2.97518e+07 - 19.63 32046.5 44.4299 -904318 2.97279e+07 - 19.65 32039.9 44.3878 -903870 2.97039e+07 - 19.67 32033.3 44.3454 -903422 2.96798e+07 - 19.69 32026.7 44.3028 -902974 2.96557e+07 - 19.71 32020.1 44.2601 -902525 2.96316e+07 - 19.73 32013.4 44.2171 -902075 2.96075e+07 - 19.75 32006.8 44.174 -901624 2.95833e+07 - 19.77 32000.1 44.1308 -901173 2.9559e+07 - 19.79 31993.4 44.0873 -900721 2.95348e+07 - 19.81 31986.7 44.0437 -900268 2.95105e+07 - 19.83 31980 43.9999 -899815 2.94861e+07 - 19.85 31973.3 43.956 -899361 2.94618e+07 - 19.87 31966.5 43.912 -898907 2.94374e+07 - 19.89 31959.8 43.8677 -898451 2.94129e+07 - 19.91 31953 43.8234 -897995 2.93885e+07 - 19.93 31946.2 43.7789 -897538 2.93639e+07 - 19.95 31939.3 43.7343 -897081 2.93394e+07 - 19.97 31932.5 43.6895 -896623 2.93148e+07 - 19.99 31925.6 43.6446 -896164 2.92902e+07 + 0.01 247031 445.994 -5.04971e+06 1.45474e+08 + 0.03 247029 445.987 -5.04969e+06 1.45474e+08 + 0.05 247025 445.974 -5.04964e+06 1.45472e+08 + 0.07 247019 445.954 -5.04958e+06 1.4547e+08 + 0.09 247011 445.927 -5.04949e+06 1.45467e+08 + 0.11 247000 445.893 -5.04938e+06 1.45464e+08 + 0.13 246988 445.853 -5.04925e+06 1.4546e+08 + 0.15 246974 445.806 -5.04909e+06 1.45455e+08 + 0.17 246958 445.753 -5.04892e+06 1.4545e+08 + 0.19 246939 445.693 -5.04872e+06 1.45444e+08 + 0.21 246919 445.626 -5.0485e+06 1.45437e+08 + 0.23 246897 445.552 -5.04826e+06 1.45429e+08 + 0.25 246872 445.472 -5.048e+06 1.45421e+08 + 0.27 246846 445.385 -5.04771e+06 1.45412e+08 + 0.29 246818 445.292 -5.04741e+06 1.45403e+08 + 0.31 246787 445.192 -5.04708e+06 1.45393e+08 + 0.33 246755 445.085 -5.04673e+06 1.45382e+08 + 0.35 246720 444.972 -5.04636e+06 1.4537e+08 + 0.37 246684 444.852 -5.04596e+06 1.45358e+08 + 0.39 246646 444.725 -5.04555e+06 1.45345e+08 + 0.41 246605 444.592 -5.04511e+06 1.45331e+08 + 0.43 246563 444.452 -5.04465e+06 1.45317e+08 + 0.45 246518 444.305 -5.04417e+06 1.45302e+08 + 0.47 246472 444.152 -5.04367e+06 1.45287e+08 + 0.49 246424 443.992 -5.04315e+06 1.4527e+08 + 0.51 246373 443.826 -5.04261e+06 1.45254e+08 + 0.53 246321 443.653 -5.04204e+06 1.45236e+08 + 0.55 246267 443.474 -5.04145e+06 1.45218e+08 + 0.57 246211 443.288 -5.04085e+06 1.45199e+08 + 0.59 246152 443.096 -5.04022e+06 1.45179e+08 + 0.61 246092 442.897 -5.03957e+06 1.45159e+08 + 0.63 246030 442.691 -5.0389e+06 1.45138e+08 + 0.65 245966 442.479 -5.0382e+06 1.45116e+08 + 0.67 245901 442.26 -5.03749e+06 1.45094e+08 + 0.69 245833 442.035 -5.03676e+06 1.45071e+08 + 0.71 245763 441.804 -5.036e+06 1.45047e+08 + 0.73 245692 441.566 -5.03523e+06 1.45023e+08 + 0.75 245618 441.322 -5.03443e+06 1.44998e+08 + 0.77 245543 441.071 -5.03361e+06 1.44972e+08 + 0.79 245465 440.814 -5.03277e+06 1.44946e+08 + 0.81 245386 440.55 -5.03191e+06 1.44919e+08 + 0.83 245305 440.28 -5.03103e+06 1.44892e+08 + 0.85 245222 440.004 -5.03013e+06 1.44863e+08 + 0.87 245138 439.721 -5.02921e+06 1.44835e+08 + 0.89 245051 439.432 -5.02827e+06 1.44805e+08 + 0.91 244963 439.137 -5.02731e+06 1.44775e+08 + 0.93 244873 438.836 -5.02633e+06 1.44744e+08 + 0.95 244781 438.528 -5.02532e+06 1.44712e+08 + 0.97 244687 438.214 -5.0243e+06 1.4468e+08 + 0.99 244591 437.893 -5.02326e+06 1.44647e+08 + 1.01 244494 437.567 -5.0222e+06 1.44614e+08 + 1.03 244394 437.234 -5.02111e+06 1.4458e+08 + 1.05 244293 436.895 -5.02001e+06 1.44545e+08 + 1.07 244190 436.55 -5.01888e+06 1.44509e+08 + 1.09 244086 436.199 -5.01774e+06 1.44473e+08 + 1.11 243979 435.841 -5.01658e+06 1.44437e+08 + 1.13 243871 435.478 -5.01539e+06 1.44399e+08 + 1.15 243762 435.109 -5.01419e+06 1.44361e+08 + 1.17 243650 434.733 -5.01297e+06 1.44322e+08 + 1.19 243537 434.352 -5.01172e+06 1.44283e+08 + 1.21 243422 433.964 -5.01046e+06 1.44243e+08 + 1.23 243305 433.571 -5.00918e+06 1.44202e+08 + 1.25 243186 433.171 -5.00787e+06 1.44161e+08 + 1.27 243066 432.766 -5.00655e+06 1.44119e+08 + 1.29 242944 432.354 -5.00521e+06 1.44077e+08 + 1.31 242821 431.937 -5.00385e+06 1.44033e+08 + 1.33 242696 431.514 -5.00246e+06 1.4399e+08 + 1.35 242569 431.086 -5.00106e+06 1.43945e+08 + 1.37 242440 430.651 -4.99964e+06 1.439e+08 + 1.39 242310 430.211 -4.9982e+06 1.43854e+08 + 1.41 242178 429.765 -4.99674e+06 1.43808e+08 + 1.43 242045 429.313 -4.99525e+06 1.43761e+08 + 1.45 241910 428.855 -4.99375e+06 1.43713e+08 + 1.47 241773 428.392 -4.99223e+06 1.43664e+08 + 1.49 241635 427.924 -4.99069e+06 1.43615e+08 + 1.51 241495 427.449 -4.98913e+06 1.43566e+08 + 1.53 241353 426.97 -4.98755e+06 1.43515e+08 + 1.55 241210 426.484 -4.98595e+06 1.43465e+08 + 1.57 241065 425.993 -4.98433e+06 1.43413e+08 + 1.59 240919 425.497 -4.98269e+06 1.43361e+08 + 1.61 240771 424.996 -4.98103e+06 1.43308e+08 + 1.63 240621 424.489 -4.97935e+06 1.43254e+08 + 1.65 240470 423.976 -4.97765e+06 1.432e+08 + 1.67 240318 423.458 -4.97593e+06 1.43145e+08 + 1.69 240163 422.935 -4.97419e+06 1.4309e+08 + 1.71 240007 422.407 -4.97243e+06 1.43033e+08 + 1.73 239850 421.874 -4.97065e+06 1.42977e+08 + 1.75 239691 421.335 -4.96884e+06 1.42919e+08 + 1.77 239531 420.791 -4.96702e+06 1.42861e+08 + 1.79 239369 420.243 -4.96518e+06 1.42802e+08 + 1.81 239205 419.689 -4.96331e+06 1.42743e+08 + 1.83 239040 419.13 -4.96143e+06 1.42683e+08 + 1.85 238874 418.566 -4.95952e+06 1.42622e+08 + 1.87 238705 417.997 -4.9576e+06 1.42561e+08 + 1.89 238536 417.423 -4.95565e+06 1.42499e+08 + 1.91 238365 416.845 -4.95368e+06 1.42436e+08 + 1.93 238192 416.261 -4.95169e+06 1.42373e+08 + 1.95 238018 415.673 -4.94968e+06 1.42308e+08 + 1.97 237842 415.08 -4.94765e+06 1.42244e+08 + 1.99 237665 414.482 -4.94559e+06 1.42178e+08 + 2.01 237486 413.88 -4.94351e+06 1.42112e+08 + 2.03 237306 413.273 -4.94141e+06 1.42045e+08 + 2.05 237124 412.661 -4.93929e+06 1.41978e+08 + 2.07 236940 412.045 -4.93715e+06 1.4191e+08 + 2.09 236756 411.425 -4.93498e+06 1.41841e+08 + 2.11 236569 410.8 -4.9328e+06 1.41772e+08 + 2.13 236381 410.17 -4.93058e+06 1.41702e+08 + 2.15 236192 409.536 -4.92835e+06 1.41631e+08 + 2.17 236001 408.898 -4.92609e+06 1.41559e+08 + 2.19 235809 408.256 -4.92381e+06 1.41487e+08 + 2.21 235615 407.609 -4.92151e+06 1.41414e+08 + 2.23 235419 406.958 -4.91918e+06 1.4134e+08 + 2.25 235222 406.303 -4.91683e+06 1.41266e+08 + 2.27 235024 405.644 -4.91445e+06 1.41191e+08 + 2.29 234824 404.981 -4.91205e+06 1.41115e+08 + 2.31 234622 404.313 -4.90963e+06 1.41039e+08 + 2.33 234419 403.642 -4.90718e+06 1.40961e+08 + 2.35 234215 402.967 -4.9047e+06 1.40884e+08 + 2.37 234009 402.288 -4.9022e+06 1.40805e+08 + 2.39 233801 401.605 -4.89968e+06 1.40726e+08 + 2.41 233592 400.918 -4.89712e+06 1.40646e+08 + 2.43 233381 400.228 -4.89455e+06 1.40565e+08 + 2.45 233169 399.533 -4.89194e+06 1.40483e+08 + 2.47 232955 398.836 -4.88932e+06 1.40401e+08 + 2.49 232739 398.134 -4.88666e+06 1.40318e+08 + 2.51 232522 397.429 -4.88398e+06 1.40234e+08 + 2.53 232304 396.721 -4.88127e+06 1.4015e+08 + 2.55 232084 396.008 -4.87853e+06 1.40064e+08 + 2.57 231862 395.293 -4.87576e+06 1.39978e+08 + 2.59 231639 394.574 -4.87297e+06 1.39891e+08 + 2.61 231414 393.852 -4.87015e+06 1.39804e+08 + 2.63 231187 393.126 -4.8673e+06 1.39716e+08 + 2.65 230959 392.398 -4.86442e+06 1.39626e+08 + 2.67 230729 391.666 -4.86152e+06 1.39537e+08 + 2.69 230498 390.93 -4.85858e+06 1.39446e+08 + 2.71 230265 390.192 -4.85561e+06 1.39354e+08 + 2.73 230031 389.451 -4.85262e+06 1.39262e+08 + 2.75 229794 388.706 -4.84959e+06 1.39169e+08 + 2.77 229556 387.959 -4.84654e+06 1.39075e+08 + 2.79 229317 387.209 -4.84345e+06 1.38981e+08 + 2.81 229076 386.455 -4.84034e+06 1.38885e+08 + 2.83 228833 385.699 -4.83719e+06 1.38789e+08 + 2.85 228588 384.94 -4.83401e+06 1.38692e+08 + 2.87 228342 384.179 -4.8308e+06 1.38594e+08 + 2.89 228094 383.414 -4.82755e+06 1.38496e+08 + 2.91 227845 382.647 -4.82428e+06 1.38396e+08 + 2.93 227593 381.877 -4.82097e+06 1.38296e+08 + 2.95 227340 381.105 -4.81763e+06 1.38195e+08 + 2.97 227086 380.33 -4.81426e+06 1.38093e+08 + 2.99 226829 379.552 -4.81085e+06 1.3799e+08 + 3.01 226571 378.773 -4.80741e+06 1.37886e+08 + 3.03 226311 377.99 -4.80394e+06 1.37782e+08 + 3.05 226050 377.206 -4.80043e+06 1.37677e+08 + 3.07 225786 376.418 -4.79689e+06 1.3757e+08 + 3.09 225521 375.629 -4.79332e+06 1.37463e+08 + 3.11 225254 374.837 -4.78971e+06 1.37356e+08 + 3.13 224985 374.044 -4.78606e+06 1.37247e+08 + 3.15 224715 373.247 -4.78238e+06 1.37137e+08 + 3.17 224443 372.449 -4.77867e+06 1.37027e+08 + 3.19 224169 371.649 -4.77492e+06 1.36916e+08 + 3.21 223893 370.846 -4.77113e+06 1.36803e+08 + 3.23 223615 370.042 -4.7673e+06 1.3669e+08 + 3.25 223336 369.236 -4.76345e+06 1.36576e+08 + 3.27 223054 368.427 -4.75955e+06 1.36462e+08 + 3.29 222771 367.617 -4.75562e+06 1.36346e+08 + 3.31 222486 366.805 -4.75165e+06 1.36229e+08 + 3.33 222200 365.991 -4.74764e+06 1.36112e+08 + 3.35 221911 365.175 -4.7436e+06 1.35994e+08 + 3.37 221620 364.357 -4.73952e+06 1.35875e+08 + 3.39 221328 363.538 -4.7354e+06 1.35755e+08 + 3.41 221034 362.717 -4.73124e+06 1.35634e+08 + 3.43 220738 361.894 -4.72705e+06 1.35512e+08 + 3.45 220440 361.07 -4.72282e+06 1.35389e+08 + 3.47 220140 360.244 -4.71855e+06 1.35266e+08 + 3.49 219839 359.417 -4.71424e+06 1.35141e+08 + 3.51 219535 358.588 -4.7099e+06 1.35016e+08 + 3.53 219230 357.757 -4.70551e+06 1.34889e+08 + 3.55 218923 356.925 -4.70109e+06 1.34762e+08 + 3.57 218614 356.092 -4.69663e+06 1.34634e+08 + 3.59 218303 355.257 -4.69213e+06 1.34505e+08 + 3.61 217990 354.421 -4.68759e+06 1.34376e+08 + 3.63 217676 353.584 -4.68302e+06 1.34245e+08 + 3.65 217359 352.745 -4.6784e+06 1.34114e+08 + 3.67 217041 351.905 -4.67374e+06 1.33981e+08 + 3.69 216720 351.064 -4.66905e+06 1.33848e+08 + 3.71 216398 350.221 -4.66432e+06 1.33714e+08 + 3.73 216074 349.378 -4.65955e+06 1.33579e+08 + 3.75 215749 348.533 -4.65474e+06 1.33443e+08 + 3.77 215421 347.687 -4.64989e+06 1.33306e+08 + 3.79 215091 346.84 -4.645e+06 1.33168e+08 + 3.81 214760 345.993 -4.64008e+06 1.3303e+08 + 3.83 214427 345.144 -4.63511e+06 1.3289e+08 + 3.85 214092 344.294 -4.63011e+06 1.3275e+08 + 3.87 213755 343.443 -4.62507e+06 1.32609e+08 + 3.89 213416 342.591 -4.61999e+06 1.32467e+08 + 3.91 213076 341.738 -4.61487e+06 1.32324e+08 + 3.93 212733 340.885 -4.60971e+06 1.32181e+08 + 3.95 212389 340.031 -4.60451e+06 1.32036e+08 + 3.97 212043 339.175 -4.59928e+06 1.31891e+08 + 3.99 211696 338.319 -4.59401e+06 1.31745e+08 + 4.01 211346 337.463 -4.5887e+06 1.31598e+08 + 4.03 210995 336.605 -4.58336e+06 1.3145e+08 + 4.05 210642 335.747 -4.57797e+06 1.31301e+08 + 4.07 210287 334.888 -4.57255e+06 1.31152e+08 + 4.09 209931 334.029 -4.56709e+06 1.31001e+08 + 4.11 209573 333.169 -4.5616e+06 1.3085e+08 + 4.13 209213 332.308 -4.55607e+06 1.30699e+08 + 4.15 208851 331.447 -4.5505e+06 1.30546e+08 + 4.17 208488 330.586 -4.54489e+06 1.30393e+08 + 4.19 208123 329.723 -4.53925e+06 1.30238e+08 + 4.21 207757 328.861 -4.53358e+06 1.30083e+08 + 4.23 207389 327.998 -4.52787e+06 1.29928e+08 + 4.25 207019 327.134 -4.52212e+06 1.29771e+08 + 4.27 206648 326.27 -4.51634e+06 1.29614e+08 + 4.29 206275 325.406 -4.51052e+06 1.29456e+08 + 4.31 205900 324.541 -4.50467e+06 1.29297e+08 + 4.33 205524 323.677 -4.49879e+06 1.29138e+08 + 4.35 205147 322.811 -4.49287e+06 1.28978e+08 + 4.37 204768 321.946 -4.48692e+06 1.28817e+08 + 4.39 204387 321.08 -4.48093e+06 1.28655e+08 + 4.41 204005 320.214 -4.47491e+06 1.28493e+08 + 4.43 203622 319.348 -4.46886e+06 1.2833e+08 + 4.45 203237 318.482 -4.46278e+06 1.28167e+08 + 4.47 202851 317.616 -4.45667e+06 1.28002e+08 + 4.49 202463 316.749 -4.45052e+06 1.27837e+08 + 4.51 202074 315.883 -4.44435e+06 1.27672e+08 + 4.53 201684 315.017 -4.43814e+06 1.27506e+08 + 4.55 201292 314.15 -4.4319e+06 1.27339e+08 + 4.57 200899 313.284 -4.42564e+06 1.27171e+08 + 4.59 200505 312.417 -4.41934e+06 1.27003e+08 + 4.61 200109 311.551 -4.41301e+06 1.26834e+08 + 4.63 199713 310.684 -4.40666e+06 1.26665e+08 + 4.65 199315 309.818 -4.40028e+06 1.26495e+08 + 4.67 198916 308.952 -4.39386e+06 1.26324e+08 + 4.69 198515 308.086 -4.38743e+06 1.26153e+08 + 4.71 198114 307.221 -4.38096e+06 1.25982e+08 + 4.73 197711 306.355 -4.37447e+06 1.25809e+08 + 4.75 197308 305.49 -4.36795e+06 1.25636e+08 + 4.77 196903 304.625 -4.3614e+06 1.25463e+08 + 4.79 196497 303.761 -4.35483e+06 1.25289e+08 + 4.81 196090 302.896 -4.34824e+06 1.25115e+08 + 4.83 195682 302.032 -4.34162e+06 1.2494e+08 + 4.85 195274 301.169 -4.33497e+06 1.24764e+08 + 4.87 194864 300.306 -4.3283e+06 1.24589e+08 + 4.89 194453 299.443 -4.32161e+06 1.24412e+08 + 4.91 194042 298.581 -4.31489e+06 1.24235e+08 + 4.93 193629 297.719 -4.30815e+06 1.24058e+08 + 4.95 193216 296.858 -4.30139e+06 1.2388e+08 + 4.97 192802 295.997 -4.29461e+06 1.23702e+08 + 4.99 192387 295.137 -4.2878e+06 1.23523e+08 + 5.01 191971 294.278 -4.28098e+06 1.23344e+08 + 5.03 191555 293.419 -4.27413e+06 1.23164e+08 + 5.05 191138 292.56 -4.26726e+06 1.22984e+08 + 5.07 190720 291.703 -4.26037e+06 1.22804e+08 + 5.09 190301 290.846 -4.25347e+06 1.22623e+08 + 5.11 189882 289.99 -4.24654e+06 1.22442e+08 + 5.13 189462 289.134 -4.23959e+06 1.2226e+08 + 5.15 189041 288.279 -4.23263e+06 1.22078e+08 + 5.17 188620 287.425 -4.22564e+06 1.21895e+08 + 5.19 188198 286.572 -4.21864e+06 1.21713e+08 + 5.21 187776 285.72 -4.21162e+06 1.21529e+08 + 5.23 187353 284.869 -4.20459e+06 1.21346e+08 + 5.25 186930 284.018 -4.19753e+06 1.21162e+08 + 5.27 186506 283.168 -4.19046e+06 1.20978e+08 + 5.29 186081 282.32 -4.18338e+06 1.20793e+08 + 5.31 185657 281.472 -4.17628e+06 1.20608e+08 + 5.33 185231 280.625 -4.16916e+06 1.20423e+08 + 5.35 184806 279.779 -4.16202e+06 1.20238e+08 + 5.37 184379 278.934 -4.15487e+06 1.20052e+08 + 5.39 183953 278.091 -4.14771e+06 1.19865e+08 + 5.41 183526 277.248 -4.14053e+06 1.19679e+08 + 5.43 183099 276.406 -4.13334e+06 1.19492e+08 + 5.45 182671 275.566 -4.12613e+06 1.19305e+08 + 5.47 182243 274.727 -4.11891e+06 1.19118e+08 + 5.49 181815 273.888 -4.11167e+06 1.1893e+08 + 5.51 181386 273.051 -4.10442e+06 1.18742e+08 + 5.53 180958 272.215 -4.09716e+06 1.18554e+08 + 5.55 180528 271.381 -4.08988e+06 1.18365e+08 + 5.57 180099 270.547 -4.0826e+06 1.18176e+08 + 5.59 179670 269.715 -4.07529e+06 1.17987e+08 + 5.61 179240 268.884 -4.06798e+06 1.17798e+08 + 5.63 178810 268.054 -4.06065e+06 1.17608e+08 + 5.65 178379 267.226 -4.05331e+06 1.17418e+08 + 5.67 177949 266.398 -4.04596e+06 1.17228e+08 + 5.69 177518 265.573 -4.0386e+06 1.17038e+08 + 5.71 177088 264.748 -4.03122e+06 1.16847e+08 + 5.73 176657 263.925 -4.02384e+06 1.16656e+08 + 5.75 176225 263.103 -4.01644e+06 1.16465e+08 + 5.77 175794 262.283 -4.00903e+06 1.16274e+08 + 5.79 175363 261.464 -4.00161e+06 1.16082e+08 + 5.81 174931 260.646 -3.99417e+06 1.1589e+08 + 5.83 174500 259.83 -3.98673e+06 1.15698e+08 + 5.85 174068 259.015 -3.97927e+06 1.15505e+08 + 5.87 173636 258.202 -3.97181e+06 1.15313e+08 + 5.89 173204 257.39 -3.96433e+06 1.1512e+08 + 5.91 172772 256.58 -3.95684e+06 1.14926e+08 + 5.93 172339 255.771 -3.94934e+06 1.14733e+08 + 5.95 171907 254.964 -3.94182e+06 1.14539e+08 + 5.97 171475 254.158 -3.9343e+06 1.14345e+08 + 5.99 171042 253.354 -3.92677e+06 1.14151e+08 + 6.01 170609 252.551 -3.91922e+06 1.13957e+08 + 6.03 170177 251.75 -3.91166e+06 1.13762e+08 + 6.05 169744 250.95 -3.90409e+06 1.13567e+08 + 6.07 169311 250.152 -3.89651e+06 1.13372e+08 + 6.09 168878 249.355 -3.88892e+06 1.13176e+08 + 6.11 168445 248.56 -3.88132e+06 1.1298e+08 + 6.13 168012 247.767 -3.8737e+06 1.12784e+08 + 6.15 167578 246.975 -3.86608e+06 1.12588e+08 + 6.17 167145 246.185 -3.85844e+06 1.12391e+08 + 6.19 166711 245.396 -3.85079e+06 1.12195e+08 + 6.21 166278 244.609 -3.84313e+06 1.11997e+08 + 6.23 165844 243.824 -3.83545e+06 1.118e+08 + 6.25 165410 243.04 -3.82777e+06 1.11602e+08 + 6.27 164977 242.257 -3.82007e+06 1.11404e+08 + 6.29 164543 241.477 -3.81236e+06 1.11206e+08 + 6.31 164109 240.698 -3.80464e+06 1.11007e+08 + 6.33 163674 239.92 -3.7969e+06 1.10808e+08 + 6.35 163240 239.145 -3.78915e+06 1.10609e+08 + 6.37 162806 238.37 -3.78139e+06 1.1041e+08 + 6.39 162371 237.598 -3.77362e+06 1.1021e+08 + 6.41 161936 236.827 -3.76583e+06 1.1001e+08 + 6.43 161501 236.058 -3.75803e+06 1.09809e+08 + 6.45 161066 235.29 -3.75022e+06 1.09608e+08 + 6.47 160631 234.524 -3.74239e+06 1.09407e+08 + 6.49 160196 233.76 -3.73455e+06 1.09206e+08 + 6.51 159760 232.997 -3.72669e+06 1.09004e+08 + 6.53 159325 232.236 -3.71882e+06 1.08802e+08 + 6.55 158889 231.476 -3.71094e+06 1.08599e+08 + 6.57 158453 230.718 -3.70304e+06 1.08396e+08 + 6.59 158017 229.962 -3.69513e+06 1.08193e+08 + 6.61 157580 229.207 -3.6872e+06 1.0799e+08 + 6.63 157144 228.454 -3.67925e+06 1.07786e+08 + 6.65 156707 227.702 -3.67129e+06 1.07581e+08 + 6.67 156270 226.952 -3.66332e+06 1.07376e+08 + 6.69 155832 226.204 -3.65533e+06 1.07171e+08 + 6.71 155395 225.457 -3.64732e+06 1.06966e+08 + 6.73 154957 224.712 -3.6393e+06 1.0676e+08 + 6.75 154519 223.969 -3.63126e+06 1.06553e+08 + 6.77 154081 223.227 -3.62321e+06 1.06347e+08 + 6.79 153642 222.486 -3.61513e+06 1.06139e+08 + 6.81 153203 221.748 -3.60705e+06 1.05932e+08 + 6.83 152764 221.01 -3.59894e+06 1.05724e+08 + 6.85 152325 220.275 -3.59082e+06 1.05515e+08 + 6.87 151885 219.541 -3.58268e+06 1.05306e+08 + 6.89 151445 218.808 -3.57452e+06 1.05097e+08 + 6.91 151005 218.077 -3.56634e+06 1.04887e+08 + 6.93 150564 217.347 -3.55815e+06 1.04676e+08 + 6.95 150123 216.619 -3.54994e+06 1.04466e+08 + 6.97 149682 215.893 -3.5417e+06 1.04254e+08 + 6.99 149240 215.168 -3.53346e+06 1.04043e+08 + 7.01 148798 214.444 -3.52519e+06 1.0383e+08 + 7.03 148356 213.722 -3.5169e+06 1.03618e+08 + 7.05 147913 213.002 -3.50859e+06 1.03404e+08 + 7.07 147470 212.283 -3.50027e+06 1.03191e+08 + 7.09 147027 211.565 -3.49192e+06 1.02976e+08 + 7.11 146583 210.849 -3.48356e+06 1.02761e+08 + 7.13 146139 210.134 -3.47517e+06 1.02546e+08 + 7.15 145694 209.421 -3.46677e+06 1.0233e+08 + 7.17 145249 208.71 -3.45834e+06 1.02114e+08 + 7.19 144804 207.999 -3.4499e+06 1.01897e+08 + 7.21 144358 207.29 -3.44144e+06 1.0168e+08 + 7.23 143912 206.583 -3.43295e+06 1.01461e+08 + 7.25 143465 205.877 -3.42444e+06 1.01243e+08 + 7.27 143018 205.172 -3.41592e+06 1.01024e+08 + 7.29 142571 204.469 -3.40737e+06 1.00804e+08 + 7.31 142123 203.767 -3.3988e+06 1.00584e+08 + 7.33 141675 203.067 -3.39021e+06 1.00363e+08 + 7.35 141227 202.367 -3.3816e+06 1.00142e+08 + 7.37 140778 201.67 -3.37297e+06 9.99196e+07 + 7.39 140328 200.973 -3.36432e+06 9.9697e+07 + 7.41 139878 200.278 -3.35564e+06 9.94738e+07 + 7.43 139428 199.585 -3.34695e+06 9.925e+07 + 7.45 138977 198.892 -3.33823e+06 9.90257e+07 + 7.47 138526 198.201 -3.32949e+06 9.88007e+07 + 7.49 138075 197.511 -3.32073e+06 9.85751e+07 + 7.51 137623 196.823 -3.31195e+06 9.83489e+07 + 7.53 137170 196.136 -3.30314e+06 9.81221e+07 + 7.55 136717 195.45 -3.29432e+06 9.78947e+07 + 7.57 136264 194.766 -3.28547e+06 9.76667e+07 + 7.59 135810 194.083 -3.2766e+06 9.7438e+07 + 7.61 135356 193.401 -3.26771e+06 9.72087e+07 + 7.63 134902 192.72 -3.25879e+06 9.69788e+07 + 7.65 134447 192.041 -3.24986e+06 9.67483e+07 + 7.67 133992 191.363 -3.2409e+06 9.65171e+07 + 7.69 133536 190.686 -3.23193e+06 9.62853e+07 + 7.71 133080 190.01 -3.22293e+06 9.60529e+07 + 7.73 132624 189.336 -3.21391e+06 9.58199e+07 + 7.75 132167 188.663 -3.20486e+06 9.55862e+07 + 7.77 131709 187.991 -3.1958e+06 9.53518e+07 + 7.79 131252 187.321 -3.18672e+06 9.51169e+07 + 7.81 130794 186.651 -3.17761e+06 9.48813e+07 + 7.83 130336 185.983 -3.16848e+06 9.4645e+07 + 7.85 129877 185.316 -3.15933e+06 9.44081e+07 + 7.87 129418 184.651 -3.15016e+06 9.41706e+07 + 7.89 128959 183.987 -3.14097e+06 9.39325e+07 + 7.91 128499 183.323 -3.13176e+06 9.36937e+07 + 7.93 128039 182.662 -3.12253e+06 9.34543e+07 + 7.95 127579 182.001 -3.11328e+06 9.32142e+07 + 7.97 127118 181.341 -3.10401e+06 9.29735e+07 + 7.99 126657 180.683 -3.09472e+06 9.27322e+07 + 8.01 126196 180.026 -3.08541e+06 9.24903e+07 + 8.03 125735 179.37 -3.07607e+06 9.22477e+07 + 8.05 125273 178.716 -3.06672e+06 9.20045e+07 + 8.07 124811 178.062 -3.05735e+06 9.17607e+07 + 8.09 124349 177.41 -3.04797e+06 9.15163e+07 + 8.11 123887 176.759 -3.03856e+06 9.12712e+07 + 8.13 123424 176.11 -3.02913e+06 9.10256e+07 + 8.15 122962 175.461 -3.01969e+06 9.07793e+07 + 8.17 122499 174.814 -3.01022e+06 9.05324e+07 + 8.19 122036 174.168 -3.00074e+06 9.0285e+07 + 8.21 121572 173.523 -2.99125e+06 9.00369e+07 + 8.23 121109 172.88 -2.98173e+06 8.97882e+07 + 8.25 120646 172.238 -2.9722e+06 8.9539e+07 + 8.27 120182 171.596 -2.96265e+06 8.92892e+07 + 8.29 119718 170.957 -2.95309e+06 8.90388e+07 + 8.31 119255 170.318 -2.94351e+06 8.87878e+07 + 8.33 118791 169.681 -2.93391e+06 8.85363e+07 + 8.35 118327 169.045 -2.9243e+06 8.82842e+07 + 8.37 117863 168.41 -2.91467e+06 8.80316e+07 + 8.39 117399 167.776 -2.90503e+06 8.77784e+07 + 8.41 116935 167.144 -2.89537e+06 8.75247e+07 + 8.43 116472 166.513 -2.8857e+06 8.72705e+07 + 8.45 116008 165.884 -2.87602e+06 8.70158e+07 + 8.47 115544 165.255 -2.86632e+06 8.67605e+07 + 8.49 115080 164.628 -2.85662e+06 8.65047e+07 + 8.51 114617 164.002 -2.84689e+06 8.62485e+07 + 8.53 114154 163.378 -2.83716e+06 8.59917e+07 + 8.55 113690 162.755 -2.82742e+06 8.57345e+07 + 8.57 113227 162.133 -2.81766e+06 8.54768e+07 + 8.59 112764 161.512 -2.80789e+06 8.52187e+07 + 8.61 112301 160.893 -2.79812e+06 8.49601e+07 + 8.63 111839 160.275 -2.78833e+06 8.4701e+07 + 8.65 111377 159.659 -2.77854e+06 8.44415e+07 + 8.67 110915 159.043 -2.76873e+06 8.41817e+07 + 8.69 110453 158.43 -2.75892e+06 8.39214e+07 + 8.71 109992 157.817 -2.7491e+06 8.36607e+07 + 8.73 109530 157.206 -2.73927e+06 8.33996e+07 + 8.75 109070 156.597 -2.72944e+06 8.31381e+07 + 8.77 108609 155.988 -2.71959e+06 8.28763e+07 + 8.79 108149 155.382 -2.70975e+06 8.26141e+07 + 8.81 107690 154.776 -2.69989e+06 8.23516e+07 + 8.83 107230 154.172 -2.69004e+06 8.20888e+07 + 8.85 106772 153.57 -2.68017e+06 8.18256e+07 + 8.87 106313 152.969 -2.67031e+06 8.15621e+07 + 8.89 105856 152.369 -2.66044e+06 8.12984e+07 + 8.91 105398 151.771 -2.65057e+06 8.10344e+07 + 8.93 104942 151.174 -2.64069e+06 8.07701e+07 + 8.95 104485 150.579 -2.63081e+06 8.05055e+07 + 8.97 104030 149.986 -2.62094e+06 8.02407e+07 + 8.99 103575 149.394 -2.61106e+06 7.99757e+07 + 9.01 103120 148.803 -2.60118e+06 7.97105e+07 + 9.03 102667 148.214 -2.5913e+06 7.94451e+07 + 9.05 102213 147.627 -2.58142e+06 7.91795e+07 + 9.07 101761 147.041 -2.57155e+06 7.89137e+07 + 9.09 101309 146.456 -2.56167e+06 7.86478e+07 + 9.11 100858 145.874 -2.5518e+06 7.83818e+07 + 9.13 100408 145.293 -2.54194e+06 7.81156e+07 + 9.15 99958.8 144.713 -2.53207e+06 7.78493e+07 + 9.17 99510.2 144.135 -2.52221e+06 7.75829e+07 + 9.19 99062.4 143.559 -2.51236e+06 7.73165e+07 + 9.21 98615.4 142.984 -2.50251e+06 7.70499e+07 + 9.23 98169.3 142.411 -2.49266e+06 7.67834e+07 + 9.25 97724.1 141.84 -2.48283e+06 7.65168e+07 + 9.27 97279.8 141.27 -2.473e+06 7.62501e+07 + 9.29 96836.4 140.702 -2.46317e+06 7.59835e+07 + 9.31 96394 140.136 -2.45336e+06 7.57169e+07 + 9.33 95952.5 139.572 -2.44356e+06 7.54503e+07 + 9.35 95512 139.009 -2.43376e+06 7.51838e+07 + 9.37 95072.6 138.448 -2.42397e+06 7.49173e+07 + 9.39 94634.2 137.888 -2.4142e+06 7.46509e+07 + 9.41 94196.8 137.331 -2.40444e+06 7.43846e+07 + 9.43 93760.5 136.775 -2.39468e+06 7.41184e+07 + 9.45 93325.3 136.221 -2.38495e+06 7.38523e+07 + 9.47 92891.2 135.669 -2.37522e+06 7.35863e+07 + 9.49 92458.3 135.118 -2.36551e+06 7.33206e+07 + 9.51 92026.5 134.57 -2.35581e+06 7.30549e+07 + 9.53 91595.9 134.023 -2.34612e+06 7.27895e+07 + 9.55 91166.5 133.478 -2.33645e+06 7.25243e+07 + 9.57 90738.3 132.935 -2.3268e+06 7.22593e+07 + 9.59 90311.4 132.394 -2.31716e+06 7.19945e+07 + 9.61 89885.7 131.855 -2.30754e+06 7.173e+07 + 9.63 89461.2 131.317 -2.29794e+06 7.14658e+07 + 9.65 89038.1 130.782 -2.28835e+06 7.12018e+07 + 9.67 88616.3 130.248 -2.27879e+06 7.09382e+07 + 9.69 88195.8 129.716 -2.26924e+06 7.06749e+07 + 9.71 87776.7 129.186 -2.25971e+06 7.04119e+07 + 9.73 87358.9 128.659 -2.2502e+06 7.01492e+07 + 9.75 86942.5 128.133 -2.24072e+06 6.98869e+07 + 9.77 86527.5 127.609 -2.23125e+06 6.9625e+07 + 9.79 86114 127.087 -2.22181e+06 6.93635e+07 + 9.81 85701.9 126.567 -2.21238e+06 6.91024e+07 + 9.83 85291.2 126.049 -2.20298e+06 6.88417e+07 + 9.85 84882 125.532 -2.19361e+06 6.85815e+07 + 9.87 84474.3 125.018 -2.18426e+06 6.83217e+07 + 9.89 84068 124.506 -2.17493e+06 6.80624e+07 + 9.91 83663.3 123.996 -2.16562e+06 6.78036e+07 + 9.93 83260.2 123.488 -2.15635e+06 6.75453e+07 + 9.95 82858.6 122.982 -2.14709e+06 6.72875e+07 + 9.97 82458.5 122.478 -2.13787e+06 6.70303e+07 + 9.99 82060.1 121.976 -2.12867e+06 6.67736e+07 + 10.01 81663.2 121.476 -2.1195e+06 6.65174e+07 + 10.03 81267.9 120.979 -2.11035e+06 6.62618e+07 + 10.05 80874.3 120.483 -2.10123e+06 6.60069e+07 + 10.07 80482.2 119.989 -2.09215e+06 6.57525e+07 + 10.09 80091.9 119.497 -2.08309e+06 6.54987e+07 + 10.11 79703.2 119.008 -2.07406e+06 6.52456e+07 + 10.13 79316.1 118.521 -2.06506e+06 6.49932e+07 + 10.15 78930.8 118.035 -2.05609e+06 6.47413e+07 + 10.17 78547.2 117.552 -2.04715e+06 6.44902e+07 + 10.19 78165.2 117.071 -2.03824e+06 6.42397e+07 + 10.21 77785 116.592 -2.02937e+06 6.399e+07 + 10.23 77406.5 116.115 -2.02052e+06 6.3741e+07 + 10.25 77029.8 115.64 -2.01171e+06 6.34926e+07 + 10.27 76654.8 115.168 -2.00293e+06 6.32451e+07 + 10.29 76281.6 114.697 -1.99418e+06 6.29982e+07 + 10.31 75910.2 114.229 -1.98547e+06 6.27522e+07 + 10.33 75540.5 113.763 -1.97679e+06 6.25069e+07 + 10.35 75172.6 113.299 -1.96815e+06 6.22624e+07 + 10.37 74806.6 112.837 -1.95953e+06 6.20187e+07 + 10.39 74442.3 112.377 -1.95096e+06 6.17758e+07 + 10.41 74079.9 111.92 -1.94242e+06 6.15337e+07 + 10.43 73719.3 111.464 -1.93391e+06 6.12925e+07 + 10.45 73360.6 111.011 -1.92544e+06 6.10521e+07 + 10.47 73003.7 110.56 -1.91701e+06 6.08126e+07 + 10.49 72648.6 110.111 -1.90861e+06 6.05739e+07 + 10.51 72295.4 109.664 -1.90025e+06 6.03361e+07 + 10.53 71944.1 109.22 -1.89193e+06 6.00991e+07 + 10.55 71594.6 108.777 -1.88364e+06 5.98631e+07 + 10.57 71247 108.337 -1.87539e+06 5.9628e+07 + 10.59 70901.3 107.899 -1.86718e+06 5.93938e+07 + 10.61 70557.5 107.463 -1.85901e+06 5.91605e+07 + 10.63 70215.6 107.029 -1.85088e+06 5.89281e+07 + 10.65 69875.6 106.598 -1.84278e+06 5.86967e+07 + 10.67 69537.5 106.169 -1.83472e+06 5.84662e+07 + 10.69 69201.3 105.742 -1.82671e+06 5.82367e+07 + 10.71 68867 105.317 -1.81873e+06 5.80082e+07 + 10.73 68534.6 104.894 -1.81079e+06 5.77806e+07 + 10.75 68204.2 104.473 -1.80289e+06 5.7554e+07 + 10.77 67875.7 104.055 -1.79503e+06 5.73284e+07 + 10.79 67549.1 103.639 -1.78721e+06 5.71038e+07 + 10.81 67224.4 103.225 -1.77943e+06 5.68802e+07 + 10.83 66901.7 102.813 -1.77169e+06 5.66575e+07 + 10.85 66580.9 102.403 -1.76399e+06 5.6436e+07 + 10.87 66262 101.996 -1.75633e+06 5.62154e+07 + 10.89 65945.1 101.59 -1.74871e+06 5.59958e+07 + 10.91 65630.1 101.187 -1.74114e+06 5.57773e+07 + 10.93 65317.1 100.786 -1.7336e+06 5.55599e+07 + 10.95 65006 100.388 -1.72611e+06 5.53434e+07 + 10.97 64696.8 99.9911 -1.71866e+06 5.51281e+07 + 10.99 64389.6 99.5967 -1.71124e+06 5.49137e+07 + 11.01 64084.3 99.2044 -1.70387e+06 5.47005e+07 + 11.03 63781 98.8144 -1.69655e+06 5.44883e+07 + 11.05 63479.6 98.4265 -1.68926e+06 5.42772e+07 + 11.07 63180.2 98.0407 -1.68202e+06 5.40671e+07 + 11.09 62882.7 97.6572 -1.67481e+06 5.38581e+07 + 11.11 62587.1 97.2757 -1.66765e+06 5.36503e+07 + 11.13 62293.5 96.8965 -1.66053e+06 5.34435e+07 + 11.15 62001.8 96.5194 -1.65346e+06 5.32377e+07 + 11.17 61712.1 96.1444 -1.64642e+06 5.30331e+07 + 11.19 61424.3 95.7716 -1.63943e+06 5.28296e+07 + 11.21 61138.4 95.4009 -1.63248e+06 5.26272e+07 + 11.23 60854.4 95.0324 -1.62557e+06 5.24258e+07 + 11.25 60572.4 94.666 -1.61871e+06 5.22256e+07 + 11.27 60292.3 94.3017 -1.61189e+06 5.20265e+07 + 11.29 60014.1 93.9395 -1.60511e+06 5.18285e+07 + 11.31 59737.9 93.5795 -1.59837e+06 5.16316e+07 + 11.33 59463.5 93.2216 -1.59167e+06 5.14358e+07 + 11.35 59191.1 92.8659 -1.58502e+06 5.12411e+07 + 11.37 58920.6 92.5122 -1.57841e+06 5.10475e+07 + 11.39 58652 92.1606 -1.57184e+06 5.08551e+07 + 11.41 58385.2 91.8112 -1.56531e+06 5.06638e+07 + 11.43 58120.4 91.4638 -1.55883e+06 5.04735e+07 + 11.45 57857.5 91.1186 -1.55239e+06 5.02845e+07 + 11.47 57596.4 90.7754 -1.54599e+06 5.00965e+07 + 11.49 57337.2 90.4343 -1.53963e+06 4.99096e+07 + 11.51 57079.9 90.0953 -1.53332e+06 4.97239e+07 + 11.53 56824.5 89.7584 -1.52704e+06 4.95393e+07 + 11.55 56571 89.4235 -1.52081e+06 4.93558e+07 + 11.57 56319.3 89.0907 -1.51462e+06 4.91734e+07 + 11.59 56069.4 88.76 -1.50848e+06 4.89922e+07 + 11.61 55821.4 88.4313 -1.50237e+06 4.8812e+07 + 11.63 55575.3 88.1047 -1.49631e+06 4.8633e+07 + 11.65 55330.9 87.7801 -1.49029e+06 4.84551e+07 + 11.67 55088.5 87.4576 -1.48431e+06 4.82783e+07 + 11.69 54847.8 87.1371 -1.47837e+06 4.81026e+07 + 11.71 54608.9 86.8186 -1.47247e+06 4.79281e+07 + 11.73 54371.9 86.5022 -1.46661e+06 4.77546e+07 + 11.75 54136.7 86.1878 -1.4608e+06 4.75823e+07 + 11.77 53903.3 85.8754 -1.45503e+06 4.74111e+07 + 11.79 53671.6 85.565 -1.44929e+06 4.7241e+07 + 11.81 53441.8 85.2565 -1.4436e+06 4.70719e+07 + 11.83 53213.7 84.9501 -1.43795e+06 4.6904e+07 + 11.85 52987.4 84.6457 -1.43234e+06 4.67372e+07 + 11.87 52762.8 84.3433 -1.42677e+06 4.65715e+07 + 11.89 52540 84.0428 -1.42124e+06 4.64069e+07 + 11.91 52319 83.7443 -1.41575e+06 4.62434e+07 + 11.93 52099.7 83.4478 -1.4103e+06 4.6081e+07 + 11.95 51882.1 83.1533 -1.40489e+06 4.59196e+07 + 11.97 51666.3 82.8607 -1.39952e+06 4.57594e+07 + 11.99 51452.1 82.57 -1.39419e+06 4.56002e+07 + 12.01 51239.7 82.2813 -1.3889e+06 4.54421e+07 + 12.03 51029 81.9945 -1.38365e+06 4.52851e+07 + 12.05 50819.9 81.7097 -1.37844e+06 4.51291e+07 + 12.07 50612.6 81.4268 -1.37326e+06 4.49742e+07 + 12.09 50406.9 81.1458 -1.36813e+06 4.48204e+07 + 12.11 50202.9 80.8667 -1.36303e+06 4.46677e+07 + 12.13 50000.5 80.5895 -1.35798e+06 4.4516e+07 + 12.15 49799.8 80.3142 -1.35296e+06 4.43654e+07 + 12.17 49600.8 80.0408 -1.34798e+06 4.42158e+07 + 12.19 49403.3 79.7692 -1.34303e+06 4.40672e+07 + 12.21 49207.5 79.4996 -1.33813e+06 4.39197e+07 + 12.23 49013.3 79.2318 -1.33326e+06 4.37733e+07 + 12.25 48820.7 78.9659 -1.32843e+06 4.36278e+07 + 12.27 48629.8 78.7019 -1.32364e+06 4.34834e+07 + 12.29 48440.4 78.4397 -1.31889e+06 4.33401e+07 + 12.31 48252.5 78.1793 -1.31417e+06 4.31977e+07 + 12.33 48066.3 77.9208 -1.30949e+06 4.30564e+07 + 12.35 47881.6 77.6641 -1.30484e+06 4.29161e+07 + 12.37 47698.4 77.4092 -1.30023e+06 4.27768e+07 + 12.39 47516.8 77.1562 -1.29566e+06 4.26384e+07 + 12.41 47336.8 76.905 -1.29112e+06 4.25011e+07 + 12.43 47158.2 76.6555 -1.28662e+06 4.23648e+07 + 12.45 46981.2 76.4079 -1.28216e+06 4.22295e+07 + 12.47 46805.7 76.1621 -1.27773e+06 4.20951e+07 + 12.49 46631.7 75.918 -1.27333e+06 4.19618e+07 + 12.51 46459.1 75.6757 -1.26897e+06 4.18294e+07 + 12.53 46288.1 75.4352 -1.26465e+06 4.16979e+07 + 12.55 46118.5 75.1964 -1.26036e+06 4.15675e+07 + 12.57 45950.3 74.9594 -1.2561e+06 4.14379e+07 + 12.59 45783.7 74.7242 -1.25188e+06 4.13094e+07 + 12.61 45618.4 74.4907 -1.24769e+06 4.11818e+07 + 12.63 45454.6 74.2589 -1.24354e+06 4.10551e+07 + 12.65 45292.2 74.0288 -1.23942e+06 4.09294e+07 + 12.67 45131.2 73.8005 -1.23533e+06 4.08046e+07 + 12.69 44971.7 73.5739 -1.23127e+06 4.06807e+07 + 12.71 44813.5 73.3489 -1.22725e+06 4.05577e+07 + 12.73 44656.7 73.1257 -1.22326e+06 4.04357e+07 + 12.75 44501.3 72.9042 -1.2193e+06 4.03145e+07 + 12.77 44347.2 72.6843 -1.21538e+06 4.01943e+07 + 12.79 44194.5 72.4662 -1.21149e+06 4.0075e+07 + 12.81 44043.1 72.2497 -1.20763e+06 3.99565e+07 + 12.83 43893.1 72.0348 -1.2038e+06 3.98389e+07 + 12.85 43744.4 71.8216 -1.2e+06 3.97223e+07 + 12.87 43597 71.6101 -1.19623e+06 3.96065e+07 + 12.89 43451 71.4002 -1.1925e+06 3.94915e+07 + 12.91 43306.2 71.1919 -1.18879e+06 3.93774e+07 + 12.93 43162.7 70.9852 -1.18512e+06 3.92642e+07 + 12.95 43020.5 70.7802 -1.18147e+06 3.91519e+07 + 12.97 42879.6 70.5768 -1.17786e+06 3.90403e+07 + 12.99 42739.9 70.3749 -1.17427e+06 3.89297e+07 + 13.01 42601.5 70.1747 -1.17072e+06 3.88198e+07 + 13.03 42464.3 69.9761 -1.16719e+06 3.87108e+07 + 13.05 42328.4 69.779 -1.16369e+06 3.86026e+07 + 13.07 42193.7 69.5835 -1.16023e+06 3.84952e+07 + 13.09 42060.2 69.3896 -1.15679e+06 3.83887e+07 + 13.11 41927.9 69.1972 -1.15338e+06 3.82829e+07 + 13.13 41796.8 69.0064 -1.14999e+06 3.81779e+07 + 13.15 41666.9 68.8171 -1.14664e+06 3.80738e+07 + 13.17 41538.2 68.6294 -1.14331e+06 3.79704e+07 + 13.19 41410.6 68.4431 -1.14001e+06 3.78678e+07 + 13.21 41284.2 68.2584 -1.13674e+06 3.7766e+07 + 13.23 41159 68.0753 -1.1335e+06 3.76649e+07 + 13.25 41034.8 67.8936 -1.13028e+06 3.75646e+07 + 13.27 40911.9 67.7134 -1.12709e+06 3.74651e+07 + 13.29 40790 67.5347 -1.12393e+06 3.73663e+07 + 13.31 40669.3 67.3575 -1.12079e+06 3.72683e+07 + 13.33 40549.7 67.1817 -1.11768e+06 3.7171e+07 + 13.35 40431.1 67.0075 -1.11459e+06 3.70744e+07 + 13.37 40313.7 66.8347 -1.11153e+06 3.69786e+07 + 13.39 40197.3 66.6633 -1.1085e+06 3.68835e+07 + 13.41 40082 66.4934 -1.10549e+06 3.67891e+07 + 13.43 39967.8 66.3249 -1.1025e+06 3.66954e+07 + 13.45 39854.6 66.1579 -1.09954e+06 3.66024e+07 + 13.47 39742.5 65.9922 -1.09661e+06 3.65102e+07 + 13.49 39631.4 65.828 -1.0937e+06 3.64186e+07 + 13.51 39521.3 65.6652 -1.09081e+06 3.63277e+07 + 13.53 39412.3 65.5038 -1.08795e+06 3.62375e+07 + 13.55 39304.2 65.3438 -1.08511e+06 3.6148e+07 + 13.57 39197.2 65.1852 -1.0823e+06 3.60592e+07 + 13.59 39091.2 65.0279 -1.07951e+06 3.5971e+07 + 13.61 38986.1 64.872 -1.07674e+06 3.58835e+07 + 13.63 38882 64.7175 -1.07399e+06 3.57966e+07 + 13.65 38778.9 64.5644 -1.07127e+06 3.57104e+07 + 13.67 38676.8 64.4125 -1.06857e+06 3.56248e+07 + 13.69 38575.6 64.2621 -1.0659e+06 3.55399e+07 + 13.71 38475.3 64.1129 -1.06324e+06 3.54556e+07 + 13.73 38376 63.9651 -1.06061e+06 3.53719e+07 + 13.75 38277.6 63.8186 -1.058e+06 3.52889e+07 + 13.77 38180.1 63.6734 -1.05541e+06 3.52065e+07 + 13.79 38083.6 63.5295 -1.05284e+06 3.51247e+07 + 13.81 37987.9 63.3869 -1.05029e+06 3.50435e+07 + 13.83 37893.2 63.2456 -1.04777e+06 3.49629e+07 + 13.85 37799.3 63.1055 -1.04526e+06 3.48829e+07 + 13.87 37706.3 62.9668 -1.04278e+06 3.48034e+07 + 13.89 37614.2 62.8293 -1.04032e+06 3.47246e+07 + 13.91 37523 62.693 -1.03787e+06 3.46464e+07 + 13.93 37432.6 62.558 -1.03545e+06 3.45687e+07 + 13.95 37343.1 62.4243 -1.03305e+06 3.44916e+07 + 13.97 37254.4 62.2918 -1.03067e+06 3.44151e+07 + 13.99 37166.5 62.1605 -1.0283e+06 3.43391e+07 + 14.01 37079.5 62.0304 -1.02596e+06 3.42637e+07 + 14.03 36993.2 61.9016 -1.02363e+06 3.41889e+07 + 14.05 36907.8 61.7739 -1.02133e+06 3.41146e+07 + 14.07 36823.3 61.6475 -1.01904e+06 3.40408e+07 + 14.09 36739.5 61.5222 -1.01677e+06 3.39676e+07 + 14.11 36656.4 61.3982 -1.01452e+06 3.38949e+07 + 14.13 36574.2 61.2753 -1.01229e+06 3.38227e+07 + 14.15 36492.8 61.1535 -1.01008e+06 3.37511e+07 + 14.17 36412.1 61.033 -1.00788e+06 3.36799e+07 + 14.19 36332.2 60.9136 -1.00571e+06 3.36093e+07 + 14.21 36253 60.7953 -1.00355e+06 3.35392e+07 + 14.23 36174.6 60.6782 -1.0014e+06 3.34696e+07 + 14.25 36096.9 60.5622 -999280 3.34005e+07 + 14.27 36020 60.4473 -997173 3.33319e+07 + 14.29 35943.8 60.3336 -995082 3.32638e+07 + 14.31 35868.3 60.2209 -993009 3.31962e+07 + 14.33 35793.5 60.1094 -990952 3.3129e+07 + 14.35 35719.5 59.9989 -988911 3.30624e+07 + 14.37 35646.1 59.8896 -986887 3.29962e+07 + 14.39 35573.4 59.7813 -984879 3.29304e+07 + 14.41 35501.4 59.6741 -982888 3.28652e+07 + 14.43 35430.1 59.568 -980912 3.28004e+07 + 14.45 35359.5 59.4629 -978952 3.2736e+07 + 14.47 35289.5 59.3589 -977007 3.26721e+07 + 14.49 35220.2 59.2559 -975078 3.26087e+07 + 14.51 35151.6 59.154 -973164 3.25456e+07 + 14.53 35083.6 59.0531 -971266 3.24831e+07 + 14.55 35016.2 58.9532 -969382 3.24209e+07 + 14.57 34949.5 58.8544 -967513 3.23592e+07 + 14.59 34883.4 58.7565 -965659 3.22979e+07 + 14.61 34818 58.6597 -963819 3.22371e+07 + 14.63 34753.1 58.5638 -961994 3.21767e+07 + 14.65 34688.9 58.4689 -960183 3.21166e+07 + 14.67 34625.3 58.3751 -958386 3.2057e+07 + 14.69 34562.2 58.2822 -956603 3.19978e+07 + 14.71 34499.8 58.1902 -954834 3.1939e+07 + 14.73 34437.9 58.0992 -953079 3.18806e+07 + 14.75 34376.7 58.0092 -951337 3.18226e+07 + 14.77 34316 57.9201 -949609 3.17649e+07 + 14.79 34255.9 57.832 -947893 3.17077e+07 + 14.81 34196.3 57.7447 -946191 3.16509e+07 + 14.83 34137.3 57.6585 -944502 3.15944e+07 + 14.85 34078.9 57.5731 -942826 3.15383e+07 + 14.87 34021 57.4886 -941163 3.14825e+07 + 14.89 33963.6 57.405 -939512 3.14272e+07 + 14.91 33906.8 57.3224 -937873 3.13722e+07 + 14.93 33850.5 57.2406 -936247 3.13175e+07 + 14.95 33794.8 57.1597 -934633 3.12633e+07 + 14.97 33739.5 57.0797 -933032 3.12093e+07 + 14.99 33684.8 57.0005 -931442 3.11558e+07 + 15.01 33630.6 56.9222 -929864 3.11025e+07 + 15.03 33576.8 56.8448 -928297 3.10496e+07 + 15.05 33523.6 56.7682 -926742 3.09971e+07 + 15.07 33470.9 56.6924 -925199 3.09449e+07 + 15.09 33418.7 56.6175 -923667 3.0893e+07 + 15.11 33366.9 56.5433 -922146 3.08414e+07 + 15.13 33315.6 56.47 -920636 3.07902e+07 + 15.15 33264.8 56.3976 -919137 3.07393e+07 + 15.17 33214.5 56.3259 -917649 3.06887e+07 + 15.19 33164.6 56.255 -916171 3.06384e+07 + 15.21 33115.2 56.1849 -914704 3.05884e+07 + 15.23 33066.2 56.1156 -913248 3.05387e+07 + 15.25 33017.7 56.047 -911802 3.04894e+07 + 15.27 32969.6 55.9793 -910366 3.04403e+07 + 15.29 32922 55.9122 -908940 3.03915e+07 + 15.31 32874.8 55.846 -907524 3.0343e+07 + 15.33 32828 55.7804 -906118 3.02949e+07 + 15.35 32781.7 55.7156 -904721 3.0247e+07 + 15.37 32735.7 55.6516 -903335 3.01993e+07 + 15.39 32690.2 55.5883 -901958 3.0152e+07 + 15.41 32645.1 55.5256 -900590 3.01049e+07 + 15.43 32600.4 55.4637 -899231 3.00582e+07 + 15.45 32556 55.4025 -897882 3.00116e+07 + 15.47 32512.1 55.342 -896542 2.99654e+07 + 15.49 32468.6 55.2822 -895210 2.99194e+07 + 15.51 32425.4 55.223 -893888 2.98737e+07 + 15.53 32382.7 55.1645 -892574 2.98282e+07 + 15.55 32340.3 55.1067 -891269 2.9783e+07 + 15.57 32298.3 55.0495 -889972 2.9738e+07 + 15.59 32256.6 54.993 -888684 2.96933e+07 + 15.61 32215.3 54.9372 -887404 2.96488e+07 + 15.63 32174.4 54.8819 -886132 2.96045e+07 + 15.65 32133.8 54.8273 -884869 2.95605e+07 + 15.67 32093.6 54.7733 -883613 2.95168e+07 + 15.69 32053.7 54.7199 -882365 2.94732e+07 + 15.71 32014.2 54.6671 -881125 2.94299e+07 + 15.73 31975 54.615 -879893 2.93868e+07 + 15.75 31936.1 54.5634 -878668 2.9344e+07 + 15.77 31897.6 54.5124 -877450 2.93013e+07 + 15.79 31859.4 54.4619 -876240 2.92589e+07 + 15.81 31821.5 54.412 -875038 2.92167e+07 + 15.83 31783.9 54.3627 -873842 2.91746e+07 + 15.85 31746.7 54.314 -872653 2.91328e+07 + 15.87 31709.7 54.2657 -871472 2.90912e+07 + 15.89 31673.1 54.2181 -870297 2.90498e+07 + 15.91 31636.7 54.1709 -869129 2.90086e+07 + 15.93 31600.6 54.1243 -867968 2.89676e+07 + 15.95 31564.9 54.0781 -866813 2.89268e+07 + 15.97 31529.4 54.0325 -865664 2.88861e+07 + 15.99 31494.2 53.9874 -864522 2.88457e+07 + 16.01 31459.3 53.9427 -863387 2.88054e+07 + 16.03 31424.7 53.8986 -862257 2.87653e+07 + 16.05 31390.3 53.8549 -861134 2.87254e+07 + 16.07 31356.2 53.8117 -860016 2.86856e+07 + 16.09 31322.4 53.7689 -858905 2.86461e+07 + 16.11 31288.8 53.7266 -857799 2.86067e+07 + 16.13 31255.5 53.6847 -856699 2.85674e+07 + 16.15 31222.4 53.6433 -855604 2.85283e+07 + 16.17 31189.6 53.6022 -854516 2.84894e+07 + 16.19 31157.1 53.5616 -853432 2.84506e+07 + 16.21 31124.7 53.5214 -852354 2.8412e+07 + 16.23 31092.7 53.4816 -851281 2.83735e+07 + 16.25 31060.8 53.4422 -850214 2.83352e+07 + 16.27 31029.2 53.4032 -849151 2.8297e+07 + 16.29 30997.8 53.3646 -848093 2.8259e+07 + 16.31 30966.7 53.3263 -847041 2.82211e+07 + 16.33 30935.7 53.2884 -845993 2.81833e+07 + 16.35 30905 53.2508 -844950 2.81457e+07 + 16.37 30874.5 53.2136 -843911 2.81082e+07 + 16.39 30844.2 53.1767 -842878 2.80709e+07 + 16.41 30814.1 53.1401 -841848 2.80336e+07 + 16.43 30784.2 53.1038 -840823 2.79965e+07 + 16.45 30754.5 53.0679 -839803 2.79595e+07 + 16.47 30725 53.0323 -838787 2.79226e+07 + 16.49 30695.7 52.9969 -837774 2.78858e+07 + 16.51 30666.6 52.9618 -836766 2.78492e+07 + 16.53 30637.6 52.927 -835762 2.78126e+07 + 16.55 30608.9 52.8925 -834762 2.77762e+07 + 16.57 30580.3 52.8582 -833766 2.77398e+07 + 16.59 30551.9 52.8242 -832774 2.77036e+07 + 16.61 30523.7 52.7904 -831785 2.76675e+07 + 16.63 30495.7 52.7568 -830800 2.76314e+07 + 16.65 30467.8 52.7235 -829819 2.75955e+07 + 16.67 30440.1 52.6904 -828841 2.75597e+07 + 16.69 30412.5 52.6575 -827866 2.75239e+07 + 16.71 30385.1 52.6247 -826895 2.74882e+07 + 16.73 30357.9 52.5922 -825927 2.74526e+07 + 16.75 30330.8 52.5598 -824962 2.74172e+07 + 16.77 30303.8 52.5276 -824001 2.73817e+07 + 16.79 30277 52.4956 -823042 2.73464e+07 + 16.81 30250.3 52.4637 -822087 2.73111e+07 + 16.83 30223.8 52.432 -821134 2.7276e+07 + 16.85 30197.4 52.4004 -820185 2.72409e+07 + 16.87 30171.2 52.3689 -819238 2.72058e+07 + 16.89 30145 52.3376 -818294 2.71709e+07 + 16.91 30119 52.3063 -817353 2.7136e+07 + 16.93 30093.1 52.2751 -816414 2.71011e+07 + 16.95 30067.4 52.2441 -815478 2.70664e+07 + 16.97 30041.7 52.2131 -814544 2.70316e+07 + 16.99 30016.2 52.1822 -813613 2.6997e+07 + 17.01 29990.7 52.1513 -812684 2.69624e+07 + 17.03 29965.4 52.1205 -811758 2.69279e+07 + 17.05 29940.2 52.0897 -810834 2.68934e+07 + 17.07 29915.1 52.059 -809912 2.6859e+07 + 17.09 29890.1 52.0283 -808992 2.68246e+07 + 17.11 29865.2 51.9976 -808074 2.67903e+07 + 17.13 29840.3 51.9669 -807159 2.6756e+07 + 17.15 29815.6 51.9363 -806245 2.67218e+07 + 17.17 29791 51.9056 -805334 2.66876e+07 + 17.19 29766.4 51.8749 -804424 2.66534e+07 + 17.21 29741.9 51.8441 -803516 2.66194e+07 + 17.23 29717.5 51.8133 -802610 2.65853e+07 + 17.25 29693.2 51.7825 -801706 2.65513e+07 + 17.27 29669 51.7516 -800804 2.65173e+07 + 17.29 29644.8 51.7207 -799903 2.64834e+07 + 17.31 29620.7 51.6897 -799004 2.64495e+07 + 17.33 29596.6 51.6586 -798107 2.64156e+07 + 17.35 29572.7 51.6274 -797211 2.63818e+07 + 17.37 29548.7 51.5961 -796316 2.63479e+07 + 17.39 29524.9 51.5647 -795423 2.63142e+07 + 17.41 29501.1 51.5332 -794532 2.62804e+07 + 17.43 29477.3 51.5015 -793642 2.62467e+07 + 17.45 29453.6 51.4698 -792753 2.6213e+07 + 17.47 29430 51.4378 -791866 2.61794e+07 + 17.49 29406.4 51.4058 -790980 2.61457e+07 + 17.51 29382.8 51.3735 -790095 2.61121e+07 + 17.53 29359.3 51.3411 -789211 2.60786e+07 + 17.55 29335.8 51.3086 -788329 2.6045e+07 + 17.57 29312.4 51.2758 -787447 2.60115e+07 + 17.59 29289 51.2428 -786567 2.59779e+07 + 17.61 29265.6 51.2097 -785688 2.59445e+07 + 17.63 29242.3 51.1763 -784810 2.5911e+07 + 17.65 29218.9 51.1427 -783934 2.58775e+07 + 17.67 29195.6 51.1089 -783058 2.58441e+07 + 17.69 29172.4 51.0749 -782183 2.58107e+07 + 17.71 29149.1 51.0406 -781309 2.57773e+07 + 17.73 29125.9 51.006 -780436 2.57439e+07 + 17.75 29102.7 50.9712 -779564 2.57106e+07 + 17.77 29079.5 50.9361 -778693 2.56772e+07 + 17.79 29056.3 50.9008 -777823 2.56439e+07 + 17.81 29033.1 50.8652 -776953 2.56106e+07 + 17.83 29009.9 50.8293 -776085 2.55773e+07 + 17.85 28986.8 50.793 -775217 2.5544e+07 + 17.87 28963.6 50.7565 -774350 2.55108e+07 + 17.89 28940.4 50.7197 -773484 2.54775e+07 + 17.91 28917.3 50.6826 -772619 2.54443e+07 + 17.93 28894.1 50.6451 -771754 2.54111e+07 + 17.95 28871 50.6073 -770891 2.53779e+07 + 17.97 28847.8 50.5692 -770028 2.53447e+07 + 17.99 28824.6 50.5307 -769165 2.53115e+07 + 18.01 28801.4 50.4919 -768304 2.52784e+07 + 18.03 28778.2 50.4527 -767443 2.52453e+07 + 18.05 28755 50.4131 -766582 2.52121e+07 + 18.07 28731.8 50.3732 -765723 2.5179e+07 + 18.09 28708.6 50.3329 -764864 2.51459e+07 + 18.11 28685.3 50.2922 -764006 2.51129e+07 + 18.13 28662.1 50.2512 -763148 2.50798e+07 + 18.15 28638.8 50.2097 -762291 2.50467e+07 + 18.17 28615.5 50.1679 -761435 2.50137e+07 + 18.19 28592.1 50.1256 -760579 2.49807e+07 + 18.21 28568.8 50.083 -759724 2.49477e+07 + 18.23 28545.4 50.0399 -758870 2.49147e+07 + 18.25 28521.9 49.9964 -758016 2.48818e+07 + 18.27 28498.5 49.9525 -757163 2.48488e+07 + 18.29 28475 49.9081 -756310 2.48159e+07 + 18.31 28451.5 49.8634 -755458 2.4783e+07 + 18.33 28428 49.8181 -754607 2.47501e+07 + 18.35 28404.4 49.7725 -753756 2.47172e+07 + 18.37 28380.8 49.7264 -752906 2.46843e+07 + 18.39 28357.2 49.6799 -752056 2.46515e+07 + 18.41 28333.5 49.6329 -751208 2.46187e+07 + 18.43 28309.8 49.5854 -750359 2.45859e+07 + 18.45 28286 49.5375 -749512 2.45531e+07 + 18.47 28262.2 49.4891 -748665 2.45203e+07 + 18.49 28238.4 49.4403 -747818 2.44876e+07 + 18.51 28214.5 49.391 -746972 2.44549e+07 + 18.53 28190.6 49.3413 -746127 2.44222e+07 + 18.55 28166.6 49.291 -745283 2.43895e+07 + 18.57 28142.6 49.2403 -744439 2.43568e+07 + 18.59 28118.6 49.1891 -743596 2.43242e+07 + 18.61 28094.5 49.1375 -742753 2.42916e+07 + 18.63 28070.3 49.0853 -741911 2.4259e+07 + 18.65 28046.1 49.0327 -741070 2.42265e+07 + 18.67 28021.9 48.9796 -740230 2.4194e+07 + 18.69 27997.6 48.926 -739390 2.41615e+07 + 18.71 27973.3 48.872 -738551 2.4129e+07 + 18.73 27948.9 48.8174 -737712 2.40966e+07 + 18.75 27924.4 48.7624 -736875 2.40641e+07 + 18.77 27900 48.7068 -736038 2.40318e+07 + 18.79 27875.4 48.6508 -735201 2.39994e+07 + 18.81 27850.8 48.5943 -734366 2.39671e+07 + 18.83 27826.2 48.5374 -733531 2.39348e+07 + 18.85 27801.5 48.4799 -732697 2.39025e+07 + 18.87 27776.8 48.4219 -731864 2.38703e+07 + 18.89 27752 48.3635 -731031 2.38381e+07 + 18.91 27727.2 48.3046 -730200 2.3806e+07 + 18.93 27702.3 48.2452 -729369 2.37738e+07 + 18.95 27677.3 48.1853 -728539 2.37417e+07 + 18.97 27652.4 48.125 -727710 2.37097e+07 + 18.99 27627.3 48.0641 -726881 2.36777e+07 + 19.01 27602.2 48.0028 -726054 2.36457e+07 + 19.03 27577.1 47.941 -725227 2.36137e+07 + 19.05 27551.9 47.8788 -724401 2.35818e+07 + 19.07 27526.7 47.816 -723577 2.355e+07 + 19.09 27501.4 47.7528 -722753 2.35181e+07 + 19.11 27476 47.6892 -721930 2.34863e+07 + 19.13 27450.6 47.6251 -721108 2.34546e+07 + 19.15 27425.2 47.5605 -720286 2.34229e+07 + 19.17 27399.7 47.4954 -719466 2.33912e+07 + 19.19 27374.2 47.43 -718647 2.33596e+07 + 19.21 27348.6 47.364 -717829 2.3328e+07 + 19.23 27323 47.2976 -717012 2.32965e+07 + 19.25 27297.3 47.2308 -716195 2.3265e+07 + 19.27 27271.6 47.1635 -715380 2.32336e+07 + 19.29 27245.8 47.0958 -714566 2.32022e+07 + 19.31 27220 47.0277 -713753 2.31708e+07 + 19.33 27194.1 46.9592 -712941 2.31395e+07 + 19.35 27168.3 46.8902 -712130 2.31083e+07 + 19.37 27142.3 46.8208 -711320 2.30771e+07 + 19.39 27116.3 46.751 -710512 2.30459e+07 + 19.41 27090.3 46.6808 -709704 2.30148e+07 + 19.43 27064.3 46.6102 -708898 2.29837e+07 + 19.45 27038.2 46.5392 -708092 2.29527e+07 + 19.47 27012 46.4678 -707288 2.29218e+07 + 19.49 26985.9 46.396 -706485 2.28909e+07 + 19.51 26959.7 46.3239 -705684 2.286e+07 + 19.53 26933.4 46.2513 -704883 2.28292e+07 + 19.55 26907.2 46.1784 -704084 2.27985e+07 + 19.57 26880.8 46.1052 -703286 2.27678e+07 + 19.59 26854.5 46.0316 -702489 2.27371e+07 + 19.61 26828.1 45.9576 -701694 2.27066e+07 + 19.63 26801.7 45.8833 -700899 2.2676e+07 + 19.65 26775.3 45.8086 -700106 2.26456e+07 + 19.67 26748.9 45.7337 -699315 2.26151e+07 + 19.69 26722.4 45.6584 -698524 2.25848e+07 + 19.71 26695.9 45.5827 -697735 2.25545e+07 + 19.73 26669.4 45.5068 -696948 2.25242e+07 + 19.75 26642.8 45.4306 -696161 2.24941e+07 + 19.77 26616.2 45.3541 -695377 2.24639e+07 + 19.79 26589.7 45.2772 -694593 2.24339e+07 + 19.81 26563.1 45.2001 -693811 2.24038e+07 + 19.83 26536.4 45.1228 -693030 2.23739e+07 + 19.85 26509.8 45.0451 -692251 2.2344e+07 + 19.87 26483.1 44.9672 -691473 2.23142e+07 + 19.89 26456.5 44.889 -690696 2.22844e+07 + 19.91 26429.8 44.8106 -689921 2.22547e+07 + 19.93 26403.1 44.732 -689148 2.2225e+07 + 19.95 26376.4 44.6531 -688376 2.21954e+07 + 19.97 26349.7 44.574 -687605 2.21659e+07 + 19.99 26323 44.4947 -686836 2.21365e+07 diff --git a/tests/06_SDFT/23_PW_SNLKG_10D10S/refOnsager.txt b/tests/06_SDFT/23_PW_SNLKG_10D10S/refOnsager.txt index 35f7ed2337b..3ca63d195e4 100644 --- a/tests/06_SDFT/23_PW_SNLKG_10D10S/refOnsager.txt +++ b/tests/06_SDFT/23_PW_SNLKG_10D10S/refOnsager.txt @@ -1,1001 +1,1001 @@ ## w(eV) sigma(Sm^-1) kappa(W(mK)^-1) L12/e(Am^-1) L22/e^2(Wm^-1) - 0.01 353423 617.166 -7.0094e+06 1.97482e+08 - 0.03 353420 617.159 -7.00936e+06 1.97481e+08 - 0.05 353413 617.143 -7.00927e+06 1.97479e+08 - 0.07 353403 617.12 -7.00914e+06 1.97476e+08 - 0.09 353390 617.089 -7.00897e+06 1.97471e+08 - 0.11 353373 617.051 -7.00875e+06 1.97465e+08 - 0.13 353353 617.005 -7.00849e+06 1.97458e+08 - 0.15 353330 616.951 -7.00819e+06 1.9745e+08 - 0.17 353304 616.889 -7.00784e+06 1.97441e+08 - 0.19 353274 616.82 -7.00745e+06 1.97431e+08 - 0.21 353241 616.743 -7.00702e+06 1.9742e+08 - 0.23 353205 616.659 -7.00654e+06 1.97407e+08 - 0.25 353165 616.567 -7.00603e+06 1.97393e+08 - 0.27 353122 616.467 -7.00546e+06 1.97378e+08 - 0.29 353075 616.359 -7.00486e+06 1.97362e+08 - 0.31 353026 616.244 -7.00421e+06 1.97345e+08 - 0.33 352973 616.121 -7.00352e+06 1.97327e+08 - 0.35 352917 615.991 -7.00278e+06 1.97308e+08 - 0.37 352857 615.853 -7.002e+06 1.97287e+08 - 0.39 352794 615.707 -7.00118e+06 1.97265e+08 - 0.41 352728 615.554 -7.00031e+06 1.97242e+08 - 0.43 352659 615.393 -6.9994e+06 1.97218e+08 - 0.45 352586 615.224 -6.99845e+06 1.97193e+08 - 0.47 352510 615.048 -6.99746e+06 1.97167e+08 - 0.49 352431 614.864 -6.99642e+06 1.9714e+08 - 0.51 352348 614.673 -6.99533e+06 1.97111e+08 - 0.53 352262 614.474 -6.99421e+06 1.97081e+08 - 0.55 352173 614.267 -6.99304e+06 1.97051e+08 - 0.57 352081 614.053 -6.99183e+06 1.97019e+08 - 0.59 351985 613.831 -6.99057e+06 1.96985e+08 - 0.61 351886 613.602 -6.98927e+06 1.96951e+08 - 0.63 351784 613.365 -6.98793e+06 1.96916e+08 - 0.65 351678 613.121 -6.98654e+06 1.96879e+08 - 0.67 351570 612.869 -6.98512e+06 1.96841e+08 - 0.69 351458 612.61 -6.98364e+06 1.96803e+08 - 0.71 351342 612.343 -6.98213e+06 1.96763e+08 - 0.73 351224 612.068 -6.98057e+06 1.96722e+08 - 0.75 351102 611.787 -6.97897e+06 1.96679e+08 - 0.77 350977 611.497 -6.97732e+06 1.96636e+08 - 0.79 350848 611.201 -6.97563e+06 1.96591e+08 - 0.81 350717 610.896 -6.9739e+06 1.96546e+08 - 0.83 350582 610.585 -6.97212e+06 1.96499e+08 - 0.85 350444 610.266 -6.97031e+06 1.96451e+08 - 0.87 350303 609.939 -6.96844e+06 1.96402e+08 - 0.89 350158 609.606 -6.96654e+06 1.96352e+08 - 0.91 350010 609.264 -6.96459e+06 1.963e+08 - 0.93 349859 608.916 -6.96259e+06 1.96248e+08 - 0.95 349705 608.56 -6.96056e+06 1.96194e+08 - 0.97 349547 608.197 -6.95848e+06 1.96139e+08 - 0.99 349387 607.826 -6.95636e+06 1.96083e+08 - 1.01 349223 607.448 -6.95419e+06 1.96026e+08 - 1.03 349055 607.063 -6.95198e+06 1.95968e+08 - 1.05 348885 606.671 -6.94973e+06 1.95909e+08 - 1.07 348712 606.271 -6.94743e+06 1.95848e+08 - 1.09 348535 605.864 -6.94509e+06 1.95787e+08 - 1.11 348355 605.45 -6.94271e+06 1.95724e+08 - 1.13 348172 605.029 -6.94028e+06 1.9566e+08 - 1.15 347985 604.6 -6.93781e+06 1.95595e+08 - 1.17 347796 604.165 -6.9353e+06 1.95529e+08 - 1.19 347603 603.722 -6.93274e+06 1.95462e+08 - 1.21 347407 603.272 -6.93014e+06 1.95393e+08 - 1.23 347208 602.815 -6.92749e+06 1.95323e+08 - 1.25 347006 602.351 -6.92481e+06 1.95253e+08 - 1.27 346800 601.879 -6.92207e+06 1.95181e+08 - 1.29 346592 601.401 -6.9193e+06 1.95108e+08 - 1.31 346380 600.916 -6.91648e+06 1.95034e+08 - 1.33 346165 600.423 -6.91362e+06 1.94959e+08 - 1.35 345947 599.924 -6.91072e+06 1.94882e+08 - 1.37 345726 599.418 -6.90777e+06 1.94805e+08 - 1.39 345502 598.904 -6.90477e+06 1.94726e+08 - 1.41 345275 598.384 -6.90174e+06 1.94646e+08 - 1.43 345044 597.857 -6.89866e+06 1.94565e+08 - 1.45 344810 597.323 -6.89554e+06 1.94483e+08 - 1.47 344574 596.782 -6.89237e+06 1.944e+08 - 1.49 344334 596.234 -6.88916e+06 1.94316e+08 - 1.51 344091 595.68 -6.88591e+06 1.9423e+08 - 1.53 343845 595.118 -6.88261e+06 1.94144e+08 - 1.55 343596 594.55 -6.87927e+06 1.94056e+08 - 1.57 343343 593.975 -6.87589e+06 1.93967e+08 - 1.59 343088 593.393 -6.87246e+06 1.93877e+08 - 1.61 342830 592.805 -6.86899e+06 1.93786e+08 - 1.63 342568 592.21 -6.86547e+06 1.93694e+08 - 1.65 342304 591.608 -6.86192e+06 1.936e+08 - 1.67 342036 591 -6.85831e+06 1.93506e+08 - 1.69 341765 590.385 -6.85467e+06 1.9341e+08 - 1.71 341492 589.763 -6.85098e+06 1.93314e+08 - 1.73 341215 589.135 -6.84725e+06 1.93216e+08 - 1.75 340935 588.5 -6.84347e+06 1.93117e+08 - 1.77 340653 587.859 -6.83965e+06 1.93016e+08 - 1.79 340367 587.211 -6.83579e+06 1.92915e+08 - 1.81 340078 586.557 -6.83188e+06 1.92813e+08 - 1.83 339786 585.896 -6.82793e+06 1.92709e+08 - 1.85 339491 585.229 -6.82394e+06 1.92605e+08 - 1.87 339194 584.556 -6.8199e+06 1.92499e+08 - 1.89 338893 583.876 -6.81582e+06 1.92392e+08 - 1.91 338589 583.19 -6.81169e+06 1.92284e+08 - 1.93 338282 582.497 -6.80752e+06 1.92175e+08 - 1.95 337973 581.798 -6.80331e+06 1.92064e+08 - 1.97 337660 581.093 -6.79906e+06 1.91953e+08 - 1.99 337344 580.382 -6.79476e+06 1.9184e+08 - 2.01 337026 579.665 -6.79041e+06 1.91727e+08 - 2.03 336704 578.941 -6.78603e+06 1.91612e+08 - 2.05 336380 578.211 -6.7816e+06 1.91496e+08 - 2.07 336053 577.476 -6.77712e+06 1.91379e+08 - 2.09 335722 576.734 -6.77261e+06 1.91261e+08 - 2.11 335389 575.986 -6.76804e+06 1.91141e+08 - 2.13 335053 575.231 -6.76344e+06 1.91021e+08 - 2.15 334714 574.471 -6.75879e+06 1.909e+08 - 2.17 334372 573.705 -6.7541e+06 1.90777e+08 - 2.19 334028 572.933 -6.74936e+06 1.90653e+08 - 2.21 333680 572.156 -6.74458e+06 1.90528e+08 - 2.23 333330 571.372 -6.73976e+06 1.90402e+08 - 2.25 332976 570.582 -6.73489e+06 1.90275e+08 - 2.27 332620 569.787 -6.72998e+06 1.90147e+08 - 2.29 332261 568.986 -6.72503e+06 1.90017e+08 - 2.31 331899 568.179 -6.72003e+06 1.89887e+08 - 2.33 331535 567.366 -6.71499e+06 1.89755e+08 - 2.35 331167 566.548 -6.70991e+06 1.89623e+08 - 2.37 330797 565.724 -6.70478e+06 1.89489e+08 - 2.39 330424 564.894 -6.69961e+06 1.89354e+08 - 2.41 330048 564.059 -6.69439e+06 1.89218e+08 - 2.43 329669 563.218 -6.68913e+06 1.8908e+08 - 2.45 329288 562.371 -6.68383e+06 1.88942e+08 - 2.47 328904 561.519 -6.67849e+06 1.88803e+08 - 2.49 328517 560.662 -6.6731e+06 1.88662e+08 - 2.51 328127 559.799 -6.66766e+06 1.88521e+08 - 2.53 327735 558.931 -6.66219e+06 1.88378e+08 - 2.55 327340 558.057 -6.65666e+06 1.88234e+08 - 2.57 326942 557.178 -6.6511e+06 1.88089e+08 - 2.59 326541 556.294 -6.64549e+06 1.87943e+08 - 2.61 326138 555.405 -6.63984e+06 1.87796e+08 - 2.63 325732 554.51 -6.63415e+06 1.87647e+08 - 2.65 325323 553.61 -6.62841e+06 1.87498e+08 - 2.67 324912 552.705 -6.62263e+06 1.87347e+08 - 2.69 324498 551.794 -6.6168e+06 1.87196e+08 - 2.71 324081 550.879 -6.61093e+06 1.87043e+08 - 2.73 323662 549.958 -6.60502e+06 1.86889e+08 - 2.75 323240 549.033 -6.59907e+06 1.86734e+08 - 2.77 322815 548.102 -6.59307e+06 1.86578e+08 - 2.79 322388 547.166 -6.58702e+06 1.86421e+08 - 2.81 321958 546.226 -6.58094e+06 1.86262e+08 - 2.83 321525 545.28 -6.57481e+06 1.86103e+08 - 2.85 321090 544.33 -6.56864e+06 1.85942e+08 - 2.87 320652 543.375 -6.56242e+06 1.85781e+08 - 2.89 320212 542.415 -6.55616e+06 1.85618e+08 - 2.91 319769 541.45 -6.54986e+06 1.85454e+08 - 2.93 319324 540.481 -6.54351e+06 1.85289e+08 - 2.95 318876 539.507 -6.53712e+06 1.85123e+08 - 2.97 318426 538.528 -6.53069e+06 1.84956e+08 - 2.99 317973 537.545 -6.52421e+06 1.84788e+08 - 3.01 317517 536.557 -6.51769e+06 1.84618e+08 - 3.03 317059 535.564 -6.51113e+06 1.84448e+08 - 3.05 316598 534.567 -6.50452e+06 1.84276e+08 - 3.07 316135 533.565 -6.49787e+06 1.84104e+08 - 3.09 315670 532.559 -6.49118e+06 1.8393e+08 - 3.11 315202 531.549 -6.48445e+06 1.83755e+08 - 3.13 314731 530.534 -6.47767e+06 1.83579e+08 - 3.15 314259 529.515 -6.47085e+06 1.83402e+08 - 3.17 313783 528.491 -6.46398e+06 1.83224e+08 - 3.19 313305 527.463 -6.45707e+06 1.83045e+08 - 3.21 312825 526.431 -6.45012e+06 1.82865e+08 - 3.23 312343 525.395 -6.44313e+06 1.82683e+08 - 3.25 311858 524.355 -6.43609e+06 1.82501e+08 - 3.27 311370 523.31 -6.42901e+06 1.82317e+08 - 3.29 310881 522.262 -6.42189e+06 1.82133e+08 - 3.31 310388 521.209 -6.41472e+06 1.81947e+08 - 3.33 309894 520.152 -6.40751e+06 1.8176e+08 - 3.35 309397 519.092 -6.40026e+06 1.81572e+08 - 3.37 308898 518.027 -6.39296e+06 1.81383e+08 - 3.39 308396 516.958 -6.38562e+06 1.81193e+08 - 3.41 307892 515.886 -6.37824e+06 1.81002e+08 - 3.43 307386 514.81 -6.37082e+06 1.80809e+08 - 3.45 306878 513.73 -6.36335e+06 1.80616e+08 - 3.47 306367 512.646 -6.35585e+06 1.80422e+08 - 3.49 305854 511.558 -6.34829e+06 1.80226e+08 - 3.51 305339 510.467 -6.3407e+06 1.8003e+08 - 3.53 304821 509.372 -6.33306e+06 1.79832e+08 - 3.55 304302 508.274 -6.32538e+06 1.79633e+08 - 3.57 303780 507.172 -6.31766e+06 1.79433e+08 - 3.59 303255 506.066 -6.3099e+06 1.79232e+08 - 3.61 302729 504.957 -6.30209e+06 1.7903e+08 - 3.63 302200 503.845 -6.29424e+06 1.78827e+08 - 3.65 301670 502.729 -6.28635e+06 1.78623e+08 - 3.67 301137 501.609 -6.27842e+06 1.78418e+08 - 3.69 300601 500.486 -6.27044e+06 1.78212e+08 - 3.71 300064 499.36 -6.26242e+06 1.78004e+08 - 3.73 299525 498.231 -6.25436e+06 1.77796e+08 - 3.75 298983 497.098 -6.24626e+06 1.77586e+08 - 3.77 298439 495.963 -6.23812e+06 1.77376e+08 - 3.79 297894 494.824 -6.22993e+06 1.77164e+08 - 3.81 297346 493.682 -6.2217e+06 1.76951e+08 - 3.83 296796 492.536 -6.21343e+06 1.76738e+08 - 3.85 296244 491.388 -6.20512e+06 1.76523e+08 - 3.87 295690 490.237 -6.19676e+06 1.76307e+08 - 3.89 295133 489.083 -6.18837e+06 1.7609e+08 - 3.91 294575 487.925 -6.17993e+06 1.75872e+08 - 3.93 294015 486.765 -6.17145e+06 1.75653e+08 - 3.95 293453 485.602 -6.16293e+06 1.75433e+08 - 3.97 292888 484.436 -6.15437e+06 1.75212e+08 - 3.99 292322 483.268 -6.14576e+06 1.74989e+08 - 4.01 291754 482.096 -6.13712e+06 1.74766e+08 - 4.03 291184 480.922 -6.12843e+06 1.74542e+08 - 4.05 290612 479.745 -6.1197e+06 1.74316e+08 - 4.07 290037 478.566 -6.11093e+06 1.7409e+08 - 4.09 289461 477.384 -6.10212e+06 1.73862e+08 - 4.11 288883 476.199 -6.09327e+06 1.73634e+08 - 4.13 288304 475.012 -6.08438e+06 1.73404e+08 - 4.15 287722 473.822 -6.07544e+06 1.73174e+08 - 4.17 287138 472.63 -6.06647e+06 1.72942e+08 - 4.19 286553 471.436 -6.05745e+06 1.72709e+08 - 4.21 285965 470.239 -6.0484e+06 1.72475e+08 - 4.23 285376 469.039 -6.0393e+06 1.72241e+08 - 4.25 284785 467.838 -6.03016e+06 1.72005e+08 - 4.27 284192 466.634 -6.02099e+06 1.71768e+08 - 4.29 283597 465.428 -6.01177e+06 1.7153e+08 - 4.31 283001 464.219 -6.00251e+06 1.71291e+08 - 4.33 282403 463.009 -5.99321e+06 1.71051e+08 - 4.35 281802 461.796 -5.98387e+06 1.7081e+08 - 4.37 281201 460.581 -5.97449e+06 1.70568e+08 - 4.39 280597 459.364 -5.96507e+06 1.70325e+08 - 4.41 279992 458.146 -5.95561e+06 1.70081e+08 - 4.43 279385 456.925 -5.94611e+06 1.69836e+08 - 4.45 278776 455.702 -5.93657e+06 1.6959e+08 - 4.47 278165 454.477 -5.92699e+06 1.69343e+08 - 4.49 277553 453.251 -5.91737e+06 1.69095e+08 - 4.51 276939 452.022 -5.90771e+06 1.68846e+08 - 4.53 276324 450.792 -5.89802e+06 1.68595e+08 - 4.55 275707 449.56 -5.88828e+06 1.68344e+08 - 4.57 275088 448.326 -5.8785e+06 1.68092e+08 - 4.59 274467 447.091 -5.86869e+06 1.67839e+08 - 4.61 273845 445.853 -5.85883e+06 1.67585e+08 - 4.63 273222 444.615 -5.84894e+06 1.67329e+08 - 4.65 272597 443.374 -5.83901e+06 1.67073e+08 - 4.67 271970 442.132 -5.82904e+06 1.66816e+08 - 4.69 271341 440.889 -5.81903e+06 1.66558e+08 - 4.71 270712 439.644 -5.80898e+06 1.66299e+08 - 4.73 270080 438.398 -5.79889e+06 1.66038e+08 - 4.75 269447 437.15 -5.78876e+06 1.65777e+08 - 4.77 268813 435.901 -5.7786e+06 1.65515e+08 - 4.79 268177 434.65 -5.7684e+06 1.65252e+08 - 4.81 267539 433.399 -5.75816e+06 1.64988e+08 - 4.83 266900 432.146 -5.74788e+06 1.64723e+08 - 4.85 266260 430.891 -5.73757e+06 1.64457e+08 - 4.87 265618 429.636 -5.72721e+06 1.6419e+08 - 4.89 264975 428.379 -5.71682e+06 1.63922e+08 - 4.91 264330 427.121 -5.7064e+06 1.63653e+08 - 4.93 263684 425.862 -5.69593e+06 1.63383e+08 - 4.95 263037 424.602 -5.68543e+06 1.63112e+08 - 4.97 262388 423.341 -5.67489e+06 1.6284e+08 - 4.99 261738 422.079 -5.66431e+06 1.62567e+08 - 5.01 261086 420.816 -5.6537e+06 1.62293e+08 - 5.03 260433 419.552 -5.64305e+06 1.62018e+08 - 5.05 259779 418.288 -5.63236e+06 1.61743e+08 - 5.07 259124 417.022 -5.62164e+06 1.61466e+08 - 5.09 258467 415.755 -5.61088e+06 1.61188e+08 - 5.11 257809 414.488 -5.60009e+06 1.6091e+08 - 5.13 257149 413.22 -5.58926e+06 1.6063e+08 - 5.15 256489 411.951 -5.57839e+06 1.6035e+08 - 5.17 255827 410.682 -5.56749e+06 1.60069e+08 - 5.19 255164 409.412 -5.55655e+06 1.59786e+08 - 5.21 254499 408.141 -5.54557e+06 1.59503e+08 - 5.23 253834 406.87 -5.53457e+06 1.59219e+08 - 5.25 253167 405.598 -5.52352e+06 1.58934e+08 - 5.27 252499 404.325 -5.51244e+06 1.58648e+08 - 5.29 251830 403.052 -5.50133e+06 1.58361e+08 - 5.31 251160 401.779 -5.49018e+06 1.58073e+08 - 5.33 250489 400.505 -5.479e+06 1.57784e+08 - 5.35 249816 399.231 -5.46778e+06 1.57495e+08 - 5.37 249143 397.957 -5.45653e+06 1.57204e+08 - 5.39 248468 396.682 -5.44524e+06 1.56913e+08 - 5.41 247792 395.407 -5.43393e+06 1.5662e+08 - 5.43 247116 394.131 -5.42257e+06 1.56327e+08 - 5.45 246438 392.855 -5.41119e+06 1.56033e+08 - 5.47 245759 391.58 -5.39977e+06 1.55738e+08 - 5.49 245079 390.303 -5.38831e+06 1.55442e+08 - 5.51 244398 389.027 -5.37683e+06 1.55145e+08 - 5.53 243717 387.751 -5.36531e+06 1.54847e+08 - 5.55 243034 386.475 -5.35376e+06 1.54549e+08 - 5.57 242350 385.198 -5.34217e+06 1.5425e+08 - 5.59 241665 383.922 -5.33056e+06 1.53949e+08 - 5.61 240979 382.645 -5.31891e+06 1.53648e+08 - 5.63 240293 381.369 -5.30723e+06 1.53346e+08 - 5.65 239605 380.093 -5.29552e+06 1.53043e+08 - 5.67 238917 378.816 -5.28377e+06 1.5274e+08 - 5.69 238228 377.54 -5.272e+06 1.52435e+08 - 5.71 237538 376.264 -5.26019e+06 1.5213e+08 - 5.73 236847 374.989 -5.24836e+06 1.51824e+08 - 5.75 236155 373.713 -5.23649e+06 1.51517e+08 - 5.77 235462 372.438 -5.22459e+06 1.51209e+08 - 5.79 234769 371.163 -5.21266e+06 1.509e+08 - 5.81 234075 369.889 -5.2007e+06 1.5059e+08 - 5.83 233380 368.614 -5.18871e+06 1.5028e+08 - 5.85 232684 367.341 -5.17669e+06 1.49969e+08 - 5.87 231987 366.067 -5.16465e+06 1.49657e+08 - 5.89 231290 364.794 -5.15257e+06 1.49344e+08 - 5.91 230592 363.522 -5.14046e+06 1.49031e+08 - 5.93 229893 362.25 -5.12833e+06 1.48716e+08 - 5.95 229194 360.978 -5.11616e+06 1.48401e+08 - 5.97 228494 359.707 -5.10397e+06 1.48085e+08 - 5.99 227793 358.437 -5.09175e+06 1.47769e+08 - 6.01 227092 357.167 -5.0795e+06 1.47451e+08 - 6.03 226390 355.898 -5.06722e+06 1.47133e+08 - 6.05 225688 354.629 -5.05492e+06 1.46814e+08 - 6.07 224985 353.361 -5.04258e+06 1.46494e+08 - 6.09 224281 352.094 -5.03023e+06 1.46174e+08 - 6.11 223577 350.828 -5.01784e+06 1.45853e+08 - 6.13 222872 349.562 -5.00543e+06 1.45531e+08 - 6.15 222166 348.298 -4.99299e+06 1.45208e+08 - 6.17 221460 347.034 -4.98052e+06 1.44885e+08 - 6.19 220754 345.771 -4.96803e+06 1.4456e+08 - 6.21 220047 344.509 -4.95551e+06 1.44235e+08 - 6.23 219340 343.247 -4.94297e+06 1.4391e+08 - 6.25 218632 341.987 -4.9304e+06 1.43583e+08 - 6.27 217924 340.728 -4.91781e+06 1.43256e+08 - 6.29 217215 339.469 -4.90519e+06 1.42929e+08 - 6.31 216506 338.212 -4.89255e+06 1.426e+08 - 6.33 215796 336.956 -4.87988e+06 1.42271e+08 - 6.35 215086 335.7 -4.86719e+06 1.41941e+08 - 6.37 214376 334.446 -4.85447e+06 1.41611e+08 - 6.39 213665 333.193 -4.84174e+06 1.4128e+08 - 6.41 212954 331.941 -4.82897e+06 1.40948e+08 - 6.43 212243 330.691 -4.81619e+06 1.40615e+08 - 6.45 211532 329.441 -4.80338e+06 1.40282e+08 - 6.47 210820 328.193 -4.79055e+06 1.39949e+08 - 6.49 210107 326.946 -4.7777e+06 1.39614e+08 - 6.51 209395 325.7 -4.76483e+06 1.39279e+08 - 6.53 208682 324.455 -4.75193e+06 1.38943e+08 - 6.55 207969 323.212 -4.73902e+06 1.38607e+08 - 6.57 207256 321.97 -4.72608e+06 1.3827e+08 - 6.59 206543 320.73 -4.71312e+06 1.37933e+08 - 6.61 205829 319.491 -4.70014e+06 1.37594e+08 - 6.63 205116 318.253 -4.68714e+06 1.37256e+08 - 6.65 204402 317.017 -4.67412e+06 1.36916e+08 - 6.67 203688 315.782 -4.66108e+06 1.36576e+08 - 6.69 202974 314.549 -4.64802e+06 1.36236e+08 - 6.71 202260 313.317 -4.63494e+06 1.35895e+08 - 6.73 201545 312.087 -4.62185e+06 1.35553e+08 - 6.75 200831 310.858 -4.60873e+06 1.35211e+08 - 6.77 200116 309.631 -4.5956e+06 1.34868e+08 - 6.79 199402 308.406 -4.58245e+06 1.34525e+08 - 6.81 198688 307.182 -4.56928e+06 1.34181e+08 - 6.83 197973 305.96 -4.55609e+06 1.33837e+08 - 6.85 197258 304.739 -4.54289e+06 1.33492e+08 - 6.87 196544 303.52 -4.52967e+06 1.33147e+08 - 6.89 195830 302.303 -4.51643e+06 1.32801e+08 - 6.91 195115 301.087 -4.50318e+06 1.32454e+08 - 6.93 194401 299.874 -4.48991e+06 1.32107e+08 - 6.95 193687 298.662 -4.47663e+06 1.3176e+08 - 6.97 192972 297.452 -4.46333e+06 1.31412e+08 - 6.99 192258 296.244 -4.45001e+06 1.31064e+08 - 7.01 191544 295.037 -4.43668e+06 1.30715e+08 - 7.03 190831 293.832 -4.42334e+06 1.30366e+08 - 7.05 190117 292.63 -4.40998e+06 1.30016e+08 - 7.07 189404 291.429 -4.39661e+06 1.29666e+08 - 7.09 188690 290.23 -4.38323e+06 1.29315e+08 - 7.11 187977 289.033 -4.36983e+06 1.28964e+08 - 7.13 187265 287.838 -4.35642e+06 1.28613e+08 - 7.15 186552 286.645 -4.34299e+06 1.28261e+08 - 7.17 185840 285.454 -4.32956e+06 1.27909e+08 - 7.19 185128 284.265 -4.31611e+06 1.27556e+08 - 7.21 184416 283.078 -4.30266e+06 1.27203e+08 - 7.23 183705 281.893 -4.28919e+06 1.2685e+08 - 7.25 182993 280.71 -4.27571e+06 1.26496e+08 - 7.27 182283 279.53 -4.26222e+06 1.26142e+08 - 7.29 181572 278.351 -4.24872e+06 1.25787e+08 - 7.31 180862 277.175 -4.23521e+06 1.25432e+08 - 7.33 180153 276 -4.22169e+06 1.25077e+08 - 7.35 179443 274.828 -4.20816e+06 1.24722e+08 - 7.37 178734 273.658 -4.19462e+06 1.24366e+08 - 7.39 178026 272.49 -4.18108e+06 1.2401e+08 - 7.41 177318 271.325 -4.16753e+06 1.23653e+08 - 7.43 176610 270.162 -4.15396e+06 1.23296e+08 - 7.45 175903 269.001 -4.1404e+06 1.22939e+08 - 7.47 175197 267.842 -4.12682e+06 1.22582e+08 - 7.49 174491 266.685 -4.11324e+06 1.22224e+08 - 7.51 173785 265.531 -4.09965e+06 1.21866e+08 - 7.53 173080 264.379 -4.08606e+06 1.21508e+08 - 7.55 172376 263.23 -4.07246e+06 1.2115e+08 - 7.57 171672 262.083 -4.05885e+06 1.20791e+08 - 7.59 170969 260.938 -4.04524e+06 1.20433e+08 - 7.61 170266 259.796 -4.03163e+06 1.20073e+08 - 7.63 169564 258.656 -4.01801e+06 1.19714e+08 - 7.65 168863 257.518 -4.00439e+06 1.19355e+08 - 7.67 168162 256.383 -3.99076e+06 1.18995e+08 - 7.69 167462 255.25 -3.97714e+06 1.18635e+08 - 7.71 166763 254.12 -3.9635e+06 1.18275e+08 - 7.73 166064 252.993 -3.94987e+06 1.17915e+08 - 7.75 165366 251.867 -3.93624e+06 1.17555e+08 - 7.77 164669 250.745 -3.9226e+06 1.17194e+08 - 7.79 163973 249.625 -3.90896e+06 1.16834e+08 - 7.81 163277 248.507 -3.89532e+06 1.16473e+08 - 7.83 162582 247.392 -3.88168e+06 1.16112e+08 - 7.85 161888 246.28 -3.86804e+06 1.15751e+08 - 7.87 161195 245.17 -3.8544e+06 1.1539e+08 - 7.89 160502 244.062 -3.84076e+06 1.15029e+08 - 7.91 159811 242.958 -3.82712e+06 1.14667e+08 - 7.93 159120 241.856 -3.81349e+06 1.14306e+08 - 7.95 158430 240.756 -3.79985e+06 1.13945e+08 - 7.97 157741 239.66 -3.78622e+06 1.13583e+08 - 7.99 157053 238.566 -3.77259e+06 1.13222e+08 - 8.01 156366 237.474 -3.75896e+06 1.1286e+08 - 8.03 155680 236.386 -3.74534e+06 1.12499e+08 - 8.05 154994 235.3 -3.73172e+06 1.12137e+08 - 8.07 154310 234.217 -3.7181e+06 1.11775e+08 - 8.09 153627 233.136 -3.70449e+06 1.11414e+08 - 8.11 152945 232.059 -3.69088e+06 1.11052e+08 - 8.13 152263 230.984 -3.67728e+06 1.10691e+08 - 8.15 151583 229.911 -3.66368e+06 1.10329e+08 - 8.17 150904 228.842 -3.65009e+06 1.09968e+08 - 8.19 150226 227.775 -3.6365e+06 1.09606e+08 - 8.21 149548 226.712 -3.62292e+06 1.09245e+08 - 8.23 148872 225.651 -3.60935e+06 1.08884e+08 - 8.25 148198 224.593 -3.59579e+06 1.08522e+08 - 8.27 147524 223.537 -3.58223e+06 1.08161e+08 - 8.29 146851 222.485 -3.56868e+06 1.078e+08 - 8.31 146180 221.435 -3.55514e+06 1.07439e+08 - 8.33 145509 220.389 -3.54161e+06 1.07079e+08 - 8.35 144840 219.345 -3.52809e+06 1.06718e+08 - 8.37 144172 218.304 -3.51458e+06 1.06358e+08 - 8.39 143505 217.266 -3.50107e+06 1.05997e+08 - 8.41 142840 216.231 -3.48758e+06 1.05637e+08 - 8.43 142176 215.199 -3.4741e+06 1.05277e+08 - 8.45 141513 214.169 -3.46063e+06 1.04917e+08 - 8.47 140851 213.143 -3.44717e+06 1.04557e+08 - 8.49 140190 212.12 -3.43373e+06 1.04198e+08 - 8.51 139531 211.1 -3.42029e+06 1.03839e+08 - 8.53 138873 210.082 -3.40687e+06 1.0348e+08 - 8.55 138217 209.068 -3.39346e+06 1.03121e+08 - 8.57 137562 208.056 -3.38007e+06 1.02762e+08 - 8.59 136908 207.048 -3.36669e+06 1.02404e+08 - 8.61 136255 206.043 -3.35332e+06 1.02046e+08 - 8.63 135604 205.04 -3.33997e+06 1.01688e+08 - 8.65 134954 204.041 -3.32663e+06 1.01331e+08 - 8.67 134306 203.045 -3.31331e+06 1.00974e+08 - 8.69 133659 202.052 -3.3e+06 1.00617e+08 - 8.71 133014 201.061 -3.28671e+06 1.0026e+08 - 8.73 132370 200.074 -3.27344e+06 9.99041e+07 - 8.75 131727 199.09 -3.26019e+06 9.95482e+07 - 8.77 131086 198.109 -3.24695e+06 9.91926e+07 - 8.79 130447 197.131 -3.23372e+06 9.88374e+07 - 8.81 129809 196.157 -3.22052e+06 9.84825e+07 - 8.83 129173 195.185 -3.20734e+06 9.8128e+07 - 8.85 128538 194.216 -3.19417e+06 9.77739e+07 - 8.87 127904 193.251 -3.18102e+06 9.74202e+07 - 8.89 127272 192.289 -3.16789e+06 9.70669e+07 - 8.91 126642 191.329 -3.15479e+06 9.67141e+07 - 8.93 126014 190.373 -3.1417e+06 9.63616e+07 - 8.95 125386 189.42 -3.12863e+06 9.60096e+07 - 8.97 124761 188.471 -3.11558e+06 9.5658e+07 - 8.99 124137 187.524 -3.10256e+06 9.53068e+07 - 9.01 123515 186.581 -3.08956e+06 9.49562e+07 - 9.03 122895 185.64 -3.07658e+06 9.4606e+07 - 9.05 122276 184.703 -3.06362e+06 9.42563e+07 - 9.07 121659 183.769 -3.05068e+06 9.3907e+07 - 9.09 121043 182.839 -3.03777e+06 9.35583e+07 - 9.11 120429 181.911 -3.02488e+06 9.32101e+07 - 9.13 119817 180.987 -3.01201e+06 9.28625e+07 - 9.15 119207 180.065 -2.99917e+06 9.25153e+07 - 9.17 118599 179.148 -2.98636e+06 9.21687e+07 - 9.19 117992 178.233 -2.97357e+06 9.18227e+07 - 9.21 117387 177.321 -2.9608e+06 9.14772e+07 - 9.23 116784 176.413 -2.94806e+06 9.11323e+07 - 9.25 116182 175.508 -2.93535e+06 9.07879e+07 - 9.27 115582 174.606 -2.92266e+06 9.04442e+07 - 9.29 114985 173.707 -2.91e+06 9.01011e+07 - 9.31 114389 172.812 -2.89736e+06 8.97586e+07 - 9.33 113795 171.92 -2.88476e+06 8.94167e+07 - 9.35 113202 171.031 -2.87218e+06 8.90754e+07 - 9.37 112612 170.145 -2.85963e+06 8.87348e+07 - 9.39 112023 169.263 -2.84711e+06 8.83948e+07 - 9.41 111437 168.384 -2.83461e+06 8.80555e+07 - 9.43 110852 167.508 -2.82215e+06 8.77168e+07 - 9.45 110269 166.635 -2.80971e+06 8.73789e+07 - 9.47 109688 165.766 -2.79731e+06 8.70416e+07 - 9.49 109109 164.9 -2.78494e+06 8.6705e+07 - 9.51 108532 164.037 -2.77259e+06 8.63692e+07 - 9.53 107957 163.178 -2.76028e+06 8.6034e+07 - 9.55 107384 162.321 -2.748e+06 8.56996e+07 - 9.57 106813 161.469 -2.73575e+06 8.53659e+07 - 9.59 106243 160.619 -2.72353e+06 8.5033e+07 - 9.61 105676 159.773 -2.71134e+06 8.47008e+07 - 9.63 105111 158.929 -2.69919e+06 8.43694e+07 - 9.65 104548 158.09 -2.68707e+06 8.40387e+07 - 9.67 103987 157.253 -2.67498e+06 8.37088e+07 - 9.69 103428 156.42 -2.66293e+06 8.33798e+07 - 9.71 102870 155.59 -2.6509e+06 8.30515e+07 - 9.73 102315 154.764 -2.63892e+06 8.2724e+07 - 9.75 101763 153.94 -2.62697e+06 8.23974e+07 - 9.77 101212 153.12 -2.61505e+06 8.20716e+07 - 9.79 100663 152.304 -2.60317e+06 8.17466e+07 - 9.81 100116 151.49 -2.59132e+06 8.14225e+07 - 9.83 99571.6 150.68 -2.57951e+06 8.10992e+07 - 9.85 99029.1 149.874 -2.56773e+06 8.07768e+07 - 9.87 98488.8 149.07 -2.55599e+06 8.04552e+07 - 9.89 97950.6 148.27 -2.54429e+06 8.01345e+07 - 9.91 97414.5 147.473 -2.53262e+06 7.98148e+07 - 9.93 96880.5 146.68 -2.52099e+06 7.94959e+07 - 9.95 96348.7 145.89 -2.5094e+06 7.91779e+07 - 9.97 95819.1 145.103 -2.49785e+06 7.88608e+07 - 9.99 95291.7 144.319 -2.48633e+06 7.85447e+07 - 10.01 94766.4 143.539 -2.47486e+06 7.82295e+07 - 10.03 94243.3 142.762 -2.46342e+06 7.79152e+07 - 10.05 93722.4 141.989 -2.45202e+06 7.76019e+07 - 10.07 93203.7 141.218 -2.44065e+06 7.72895e+07 - 10.09 92687.3 140.452 -2.42933e+06 7.69781e+07 - 10.11 92173 139.688 -2.41805e+06 7.66677e+07 - 10.13 91661 138.928 -2.40681e+06 7.63582e+07 - 10.15 91151.2 138.171 -2.3956e+06 7.60497e+07 - 10.17 90643.6 137.417 -2.38444e+06 7.57423e+07 - 10.19 90138.3 136.667 -2.37332e+06 7.54358e+07 - 10.21 89635.3 135.92 -2.36224e+06 7.51303e+07 - 10.23 89134.5 135.176 -2.3512e+06 7.48259e+07 - 10.25 88636 134.436 -2.3402e+06 7.45225e+07 - 10.27 88139.8 133.699 -2.32925e+06 7.42201e+07 - 10.29 87645.8 132.965 -2.31833e+06 7.39187e+07 - 10.31 87154.2 132.235 -2.30746e+06 7.36184e+07 - 10.33 86664.8 131.508 -2.29663e+06 7.33192e+07 - 10.35 86177.7 130.784 -2.28584e+06 7.3021e+07 - 10.37 85692.9 130.064 -2.2751e+06 7.27239e+07 - 10.39 85210.5 129.346 -2.2644e+06 7.24278e+07 - 10.41 84730.4 128.633 -2.25374e+06 7.21329e+07 - 10.43 84252.6 127.922 -2.24313e+06 7.1839e+07 - 10.45 83777.1 127.215 -2.23256e+06 7.15463e+07 - 10.47 83303.9 126.511 -2.22203e+06 7.12546e+07 - 10.49 82833.1 125.811 -2.21155e+06 7.0964e+07 - 10.51 82364.7 125.113 -2.20111e+06 7.06746e+07 - 10.53 81898.6 124.419 -2.19072e+06 7.03863e+07 - 10.55 81434.8 123.729 -2.18037e+06 7.00991e+07 - 10.57 80973.4 123.042 -2.17006e+06 6.98131e+07 - 10.59 80514.4 122.358 -2.1598e+06 6.95281e+07 - 10.61 80057.8 121.677 -2.14959e+06 6.92444e+07 - 10.63 79603.5 120.999 -2.13942e+06 6.89618e+07 - 10.65 79151.6 120.325 -2.1293e+06 6.86804e+07 - 10.67 78702.1 119.655 -2.11923e+06 6.84001e+07 - 10.69 78254.9 118.987 -2.1092e+06 6.8121e+07 - 10.71 77810.2 118.323 -2.09922e+06 6.78431e+07 - 10.73 77367.8 117.662 -2.08928e+06 6.75663e+07 - 10.75 76927.9 117.004 -2.07939e+06 6.72908e+07 - 10.77 76490.3 116.35 -2.06955e+06 6.70164e+07 - 10.79 76055.2 115.699 -2.05975e+06 6.67433e+07 - 10.81 75622.4 115.051 -2.05e+06 6.64713e+07 - 10.83 75192.1 114.407 -2.0403e+06 6.62006e+07 - 10.85 74764.2 113.765 -2.03065e+06 6.59311e+07 - 10.87 74338.7 113.128 -2.02104e+06 6.56628e+07 - 10.89 73915.6 112.493 -2.01148e+06 6.53957e+07 - 10.91 73494.9 111.862 -2.00197e+06 6.51299e+07 - 10.93 73076.7 111.233 -1.99251e+06 6.48653e+07 - 10.95 72660.9 110.609 -1.9831e+06 6.4602e+07 - 10.97 72247.5 109.987 -1.97373e+06 6.43399e+07 - 10.99 71836.6 109.369 -1.96442e+06 6.4079e+07 - 11.01 71428 108.754 -1.95515e+06 6.38194e+07 - 11.03 71021.9 108.142 -1.94593e+06 6.35611e+07 - 11.05 70618.3 107.533 -1.93676e+06 6.3304e+07 - 11.07 70217.1 106.928 -1.92764e+06 6.30483e+07 - 11.09 69818.3 106.326 -1.91857e+06 6.27937e+07 - 11.11 69422 105.727 -1.90955e+06 6.25405e+07 - 11.13 69028.1 105.132 -1.90057e+06 6.22885e+07 - 11.15 68636.6 104.54 -1.89165e+06 6.20379e+07 - 11.17 68247.6 103.951 -1.88278e+06 6.17885e+07 - 11.19 67861 103.365 -1.87395e+06 6.15404e+07 - 11.21 67476.9 102.782 -1.86518e+06 6.12936e+07 - 11.23 67095.2 102.203 -1.85645e+06 6.10481e+07 - 11.25 66716 101.627 -1.84778e+06 6.08039e+07 - 11.27 66339.2 101.054 -1.83916e+06 6.0561e+07 - 11.29 65964.8 100.485 -1.83058e+06 6.03195e+07 - 11.31 65592.9 99.9182 -1.82206e+06 6.00792e+07 - 11.33 65223.5 99.355 -1.81359e+06 5.98403e+07 - 11.35 64856.4 98.7951 -1.80516e+06 5.96027e+07 - 11.37 64491.9 98.2384 -1.79679e+06 5.93664e+07 - 11.39 64129.8 97.6849 -1.78847e+06 5.91314e+07 - 11.41 63770.1 97.1346 -1.7802e+06 5.88978e+07 - 11.43 63412.8 96.5875 -1.77198e+06 5.86655e+07 - 11.45 63058 96.0436 -1.76381e+06 5.84345e+07 - 11.47 62705.7 95.503 -1.75569e+06 5.82049e+07 - 11.49 62355.7 94.9655 -1.74763e+06 5.79766e+07 - 11.51 62008.2 94.4312 -1.73961e+06 5.77497e+07 - 11.53 61663.2 93.9002 -1.73165e+06 5.75241e+07 - 11.55 61320.6 93.3723 -1.72373e+06 5.72998e+07 - 11.57 60980.4 92.8476 -1.71587e+06 5.70769e+07 - 11.59 60642.6 92.3261 -1.70806e+06 5.68554e+07 - 11.61 60307.3 91.8078 -1.7003e+06 5.66352e+07 - 11.63 59974.4 91.2927 -1.69259e+06 5.64163e+07 - 11.65 59643.9 90.7807 -1.68493e+06 5.61988e+07 - 11.67 59315.9 90.272 -1.67732e+06 5.59827e+07 - 11.69 58990.3 89.7664 -1.66977e+06 5.57679e+07 - 11.71 58667.1 89.264 -1.66226e+06 5.55545e+07 - 11.73 58346.3 88.7647 -1.65481e+06 5.53425e+07 - 11.75 58027.9 88.2686 -1.64741e+06 5.51318e+07 - 11.77 57711.9 87.7757 -1.64006e+06 5.49224e+07 - 11.79 57398.3 87.286 -1.63276e+06 5.47145e+07 - 11.81 57087.2 86.7994 -1.62551e+06 5.45079e+07 - 11.83 56778.4 86.3159 -1.61832e+06 5.43027e+07 - 11.85 56472 85.8356 -1.61117e+06 5.40988e+07 - 11.87 56168.1 85.3585 -1.60408e+06 5.38963e+07 - 11.89 55866.5 84.8845 -1.59704e+06 5.36952e+07 - 11.91 55567.3 84.4137 -1.59004e+06 5.34955e+07 - 11.93 55270.5 83.946 -1.5831e+06 5.32971e+07 - 11.95 54976 83.4814 -1.57622e+06 5.31001e+07 - 11.97 54684 83.02 -1.56938e+06 5.29044e+07 - 11.99 54394.3 82.5617 -1.56259e+06 5.27102e+07 - 12.01 54106.9 82.1066 -1.55586e+06 5.25173e+07 - 12.03 53822 81.6546 -1.54917e+06 5.23257e+07 - 12.05 53539.4 81.2057 -1.54254e+06 5.21356e+07 - 12.07 53259.1 80.7599 -1.53596e+06 5.19468e+07 - 12.09 52981.2 80.3173 -1.52943e+06 5.17594e+07 - 12.11 52705.6 79.8777 -1.52295e+06 5.15733e+07 - 12.13 52432.4 79.4413 -1.51652e+06 5.13887e+07 - 12.15 52161.5 79.0081 -1.51015e+06 5.12054e+07 - 12.17 51893 78.5779 -1.50382e+06 5.10234e+07 - 12.19 51626.7 78.1508 -1.49754e+06 5.08428e+07 - 12.21 51362.8 77.7269 -1.49132e+06 5.06636e+07 - 12.23 51101.2 77.306 -1.48514e+06 5.04858e+07 - 12.25 50841.9 76.8882 -1.47902e+06 5.03093e+07 - 12.27 50584.9 76.4736 -1.47295e+06 5.01342e+07 - 12.29 50330.2 76.062 -1.46692e+06 4.99605e+07 - 12.31 50077.8 75.6536 -1.46095e+06 4.97881e+07 - 12.33 49827.6 75.2482 -1.45503e+06 4.96171e+07 - 12.35 49579.8 74.8459 -1.44916e+06 4.94474e+07 - 12.37 49334.2 74.4467 -1.44333e+06 4.92791e+07 - 12.39 49090.9 74.0506 -1.43756e+06 4.91121e+07 - 12.41 48849.8 73.6576 -1.43184e+06 4.89465e+07 - 12.43 48611 73.2677 -1.42617e+06 4.87823e+07 - 12.45 48374.5 72.8808 -1.42055e+06 4.86194e+07 - 12.47 48140.2 72.497 -1.41497e+06 4.84578e+07 - 12.49 47908.1 72.1163 -1.40945e+06 4.82976e+07 - 12.51 47678.2 71.7386 -1.40398e+06 4.81388e+07 - 12.53 47450.6 71.364 -1.39855e+06 4.79812e+07 - 12.55 47225.2 70.9925 -1.39318e+06 4.78251e+07 - 12.57 47002 70.624 -1.38785e+06 4.76702e+07 - 12.59 46781 70.2586 -1.38258e+06 4.75167e+07 - 12.61 46562.1 69.8962 -1.37735e+06 4.73645e+07 - 12.63 46345.5 69.5369 -1.37217e+06 4.72137e+07 - 12.65 46131.1 69.1807 -1.36704e+06 4.70642e+07 - 12.67 45918.8 68.8274 -1.36196e+06 4.6916e+07 - 12.69 45708.7 68.4773 -1.35692e+06 4.67691e+07 - 12.71 45500.7 68.1301 -1.35194e+06 4.66236e+07 - 12.73 45294.9 67.786 -1.347e+06 4.64793e+07 - 12.75 45091.2 67.4449 -1.34211e+06 4.63364e+07 - 12.77 44889.7 67.1069 -1.33727e+06 4.61948e+07 - 12.79 44690.3 66.7719 -1.33248e+06 4.60545e+07 - 12.81 44493 66.4399 -1.32773e+06 4.59155e+07 - 12.83 44297.8 66.1109 -1.32304e+06 4.57778e+07 - 12.85 44104.7 65.785 -1.31838e+06 4.56414e+07 - 12.87 43913.7 65.462 -1.31378e+06 4.55063e+07 - 12.89 43724.8 65.1421 -1.30923e+06 4.53724e+07 - 12.91 43537.9 64.8252 -1.30472e+06 4.52399e+07 - 12.93 43353.2 64.5112 -1.30025e+06 4.51086e+07 - 12.95 43170.5 64.2003 -1.29584e+06 4.49787e+07 - 12.97 42989.8 63.8923 -1.29147e+06 4.485e+07 - 12.99 42811.2 63.5874 -1.28714e+06 4.47225e+07 - 13.01 42634.6 63.2854 -1.28287e+06 4.45964e+07 - 13.03 42460.1 62.9864 -1.27864e+06 4.44715e+07 - 13.05 42287.6 62.6904 -1.27445e+06 4.43479e+07 - 13.07 42117 62.3974 -1.27031e+06 4.42255e+07 - 13.09 41948.5 62.1073 -1.26622e+06 4.41044e+07 - 13.11 41782 61.8202 -1.26217e+06 4.39845e+07 - 13.13 41617.4 61.5361 -1.25816e+06 4.38658e+07 - 13.15 41454.9 61.2549 -1.2542e+06 4.37484e+07 - 13.17 41294.2 60.9766 -1.25029e+06 4.36323e+07 - 13.19 41135.6 60.7013 -1.24642e+06 4.35173e+07 - 13.21 40978.9 60.429 -1.2426e+06 4.34036e+07 - 13.23 40824.1 60.1596 -1.23881e+06 4.32911e+07 - 13.25 40671.3 59.8931 -1.23508e+06 4.31798e+07 - 13.27 40520.3 59.6295 -1.23138e+06 4.30698e+07 - 13.29 40371.3 59.3688 -1.22774e+06 4.29609e+07 - 13.31 40224.2 59.1111 -1.22413e+06 4.28533e+07 - 13.33 40079 58.8563 -1.22057e+06 4.27468e+07 - 13.35 39935.6 58.6043 -1.21705e+06 4.26415e+07 - 13.37 39794.2 58.3553 -1.21357e+06 4.25375e+07 - 13.39 39654.5 58.1091 -1.21014e+06 4.24346e+07 - 13.41 39516.8 57.8659 -1.20675e+06 4.23328e+07 - 13.43 39380.9 57.6255 -1.2034e+06 4.22323e+07 - 13.45 39246.8 57.3879 -1.20009e+06 4.21329e+07 - 13.47 39114.5 57.1533 -1.19682e+06 4.20347e+07 - 13.49 38984.1 56.9214 -1.1936e+06 4.19376e+07 - 13.51 38855.4 56.6925 -1.19042e+06 4.18417e+07 - 13.53 38728.6 56.4664 -1.18728e+06 4.17469e+07 - 13.55 38603.5 56.2431 -1.18418e+06 4.16533e+07 - 13.57 38480.2 56.0226 -1.18112e+06 4.15608e+07 - 13.59 38358.7 55.8049 -1.1781e+06 4.14694e+07 - 13.61 38238.9 55.5901 -1.17513e+06 4.13791e+07 - 13.63 38120.9 55.3781 -1.17219e+06 4.129e+07 - 13.65 38004.6 55.1688 -1.16929e+06 4.1202e+07 - 13.67 37890.1 54.9624 -1.16643e+06 4.11151e+07 - 13.69 37777.2 54.7587 -1.16362e+06 4.10293e+07 - 13.71 37666.1 54.5578 -1.16084e+06 4.09445e+07 - 13.73 37556.6 54.3596 -1.1581e+06 4.08609e+07 - 13.75 37448.9 54.1642 -1.1554e+06 4.07784e+07 - 13.77 37342.8 53.9715 -1.15274e+06 4.06969e+07 - 13.79 37238.3 53.7816 -1.15012e+06 4.06165e+07 - 13.81 37135.6 53.5944 -1.14753e+06 4.05371e+07 - 13.83 37034.5 53.4099 -1.14499e+06 4.04589e+07 - 13.85 36935 53.2281 -1.14248e+06 4.03816e+07 - 13.87 36837.1 53.049 -1.14001e+06 4.03055e+07 - 13.89 36740.8 52.8725 -1.13757e+06 4.02303e+07 - 13.91 36646.2 52.6988 -1.13518e+06 4.01562e+07 - 13.93 36553.1 52.5277 -1.13282e+06 4.00832e+07 - 13.95 36461.7 52.3592 -1.13049e+06 4.00111e+07 - 13.97 36371.8 52.1934 -1.12821e+06 3.99401e+07 - 13.99 36283.4 52.0302 -1.12596e+06 3.98701e+07 - 14.01 36196.7 51.8697 -1.12375e+06 3.9801e+07 - 14.03 36111.4 51.7117 -1.12157e+06 3.9733e+07 - 14.05 36027.7 51.5563 -1.11943e+06 3.9666e+07 - 14.07 35945.5 51.4035 -1.11732e+06 3.95999e+07 - 14.09 35864.8 51.2533 -1.11525e+06 3.95349e+07 - 14.11 35785.7 51.1056 -1.11321e+06 3.94708e+07 - 14.13 35708 50.9605 -1.11121e+06 3.94076e+07 - 14.15 35631.8 50.8179 -1.10924e+06 3.93455e+07 - 14.17 35557 50.6778 -1.10731e+06 3.92842e+07 - 14.19 35483.8 50.5402 -1.10541e+06 3.9224e+07 - 14.21 35412 50.4051 -1.10354e+06 3.91646e+07 - 14.23 35341.6 50.2725 -1.10171e+06 3.91062e+07 - 14.25 35272.6 50.1424 -1.09991e+06 3.90487e+07 - 14.27 35205.1 50.0147 -1.09814e+06 3.89922e+07 - 14.29 35139 49.8894 -1.09641e+06 3.89365e+07 - 14.31 35074.2 49.7666 -1.09471e+06 3.88818e+07 - 14.33 35010.9 49.6462 -1.09304e+06 3.8828e+07 - 14.35 34948.9 49.5281 -1.09141e+06 3.8775e+07 - 14.37 34888.3 49.4125 -1.0898e+06 3.8723e+07 - 14.39 34829.1 49.2992 -1.08823e+06 3.86718e+07 - 14.41 34771.2 49.1883 -1.08669e+06 3.86215e+07 - 14.43 34714.6 49.0796 -1.08518e+06 3.8572e+07 - 14.45 34659.4 48.9734 -1.0837e+06 3.85235e+07 - 14.47 34605.5 48.8694 -1.08225e+06 3.84757e+07 - 14.49 34552.9 48.7677 -1.08083e+06 3.84289e+07 - 14.51 34501.6 48.6683 -1.07944e+06 3.83828e+07 - 14.53 34451.6 48.5711 -1.07809e+06 3.83376e+07 - 14.55 34402.8 48.4762 -1.07676e+06 3.82932e+07 - 14.57 34355.3 48.3835 -1.07546e+06 3.82497e+07 - 14.59 34309.1 48.293 -1.07419e+06 3.82069e+07 - 14.61 34264.1 48.2047 -1.07295e+06 3.8165e+07 - 14.63 34220.4 48.1186 -1.07174e+06 3.81238e+07 - 14.65 34177.9 48.0346 -1.07055e+06 3.80835e+07 - 14.67 34136.6 47.9528 -1.0694e+06 3.80439e+07 - 14.69 34096.5 47.8731 -1.06827e+06 3.80051e+07 - 14.71 34057.5 47.7955 -1.06717e+06 3.79671e+07 - 14.73 34019.8 47.72 -1.0661e+06 3.79298e+07 - 14.75 33983.3 47.6466 -1.06506e+06 3.78933e+07 - 14.77 33947.9 47.5752 -1.06404e+06 3.78576e+07 - 14.79 33913.7 47.5059 -1.06305e+06 3.78226e+07 - 14.81 33880.6 47.4386 -1.06209e+06 3.77883e+07 - 14.83 33848.7 47.3733 -1.06115e+06 3.77548e+07 - 14.85 33817.8 47.31 -1.06024e+06 3.7722e+07 - 14.87 33788.1 47.2486 -1.05936e+06 3.76899e+07 - 14.89 33759.6 47.1893 -1.0585e+06 3.76585e+07 - 14.91 33732.1 47.1318 -1.05766e+06 3.76279e+07 - 14.93 33705.7 47.0763 -1.05686e+06 3.75979e+07 - 14.95 33680.3 47.0226 -1.05607e+06 3.75686e+07 - 14.97 33656.1 46.9709 -1.05532e+06 3.754e+07 - 14.99 33632.8 46.921 -1.05458e+06 3.75121e+07 - 15.01 33610.7 46.8729 -1.05387e+06 3.74848e+07 - 15.03 33589.6 46.8267 -1.05319e+06 3.74582e+07 - 15.05 33569.5 46.7822 -1.05253e+06 3.74323e+07 - 15.07 33550.4 46.7396 -1.05189e+06 3.7407e+07 - 15.09 33532.4 46.6987 -1.05127e+06 3.73824e+07 - 15.11 33515.3 46.6596 -1.05068e+06 3.73584e+07 - 15.13 33499.2 46.6222 -1.05011e+06 3.7335e+07 - 15.15 33484.2 46.5865 -1.04957e+06 3.73122e+07 - 15.17 33470 46.5526 -1.04905e+06 3.72901e+07 - 15.19 33456.9 46.5202 -1.04855e+06 3.72686e+07 - 15.21 33444.7 46.4896 -1.04807e+06 3.72477e+07 - 15.23 33433.5 46.4606 -1.04761e+06 3.72273e+07 - 15.25 33423.2 46.4332 -1.04718e+06 3.72076e+07 - 15.27 33413.8 46.4074 -1.04676e+06 3.71885e+07 - 15.29 33405.3 46.3832 -1.04637e+06 3.71699e+07 - 15.31 33397.8 46.3605 -1.046e+06 3.71519e+07 - 15.33 33391.1 46.3394 -1.04565e+06 3.71345e+07 - 15.35 33385.3 46.3199 -1.04532e+06 3.71176e+07 - 15.37 33380.5 46.3018 -1.04501e+06 3.71013e+07 - 15.39 33376.5 46.2852 -1.04472e+06 3.70855e+07 - 15.41 33373.3 46.2701 -1.04445e+06 3.70702e+07 - 15.43 33371 46.2564 -1.0442e+06 3.70555e+07 - 15.45 33369.6 46.2442 -1.04397e+06 3.70413e+07 - 15.47 33369 46.2333 -1.04376e+06 3.70277e+07 - 15.49 33369.2 46.2239 -1.04356e+06 3.70145e+07 - 15.51 33370.3 46.2159 -1.04339e+06 3.70019e+07 - 15.53 33372.1 46.2092 -1.04323e+06 3.69897e+07 - 15.55 33374.8 46.2038 -1.0431e+06 3.69781e+07 - 15.57 33378.2 46.1998 -1.04298e+06 3.69669e+07 - 15.59 33382.5 46.197 -1.04288e+06 3.69562e+07 - 15.61 33387.5 46.1955 -1.0428e+06 3.6946e+07 - 15.63 33393.3 46.1953 -1.04273e+06 3.69363e+07 - 15.65 33399.9 46.1964 -1.04268e+06 3.6927e+07 - 15.67 33407.2 46.1987 -1.04265e+06 3.69182e+07 - 15.69 33415.2 46.2021 -1.04264e+06 3.69098e+07 - 15.71 33424 46.2068 -1.04264e+06 3.69019e+07 - 15.73 33433.5 46.2126 -1.04266e+06 3.68944e+07 - 15.75 33443.8 46.2196 -1.0427e+06 3.68873e+07 - 15.77 33454.7 46.2277 -1.04275e+06 3.68807e+07 - 15.79 33466.4 46.237 -1.04282e+06 3.68745e+07 - 15.81 33478.8 46.2473 -1.0429e+06 3.68687e+07 - 15.83 33491.8 46.2587 -1.043e+06 3.68634e+07 - 15.85 33505.5 46.2712 -1.04312e+06 3.68584e+07 - 15.87 33519.9 46.2847 -1.04325e+06 3.68538e+07 - 15.89 33535 46.2993 -1.04339e+06 3.68496e+07 - 15.91 33550.7 46.3148 -1.04355e+06 3.68458e+07 - 15.93 33567.1 46.3314 -1.04373e+06 3.68424e+07 - 15.95 33584.1 46.3489 -1.04391e+06 3.68394e+07 - 15.97 33601.8 46.3674 -1.04412e+06 3.68367e+07 - 15.99 33620 46.3868 -1.04433e+06 3.68344e+07 - 16.01 33638.9 46.4072 -1.04457e+06 3.68324e+07 - 16.03 33658.5 46.4284 -1.04481e+06 3.68308e+07 - 16.05 33678.6 46.4505 -1.04507e+06 3.68296e+07 - 16.07 33699.3 46.4735 -1.04534e+06 3.68286e+07 - 16.09 33720.6 46.4974 -1.04562e+06 3.68281e+07 - 16.11 33742.5 46.5221 -1.04592e+06 3.68278e+07 - 16.13 33765 46.5476 -1.04623e+06 3.68279e+07 - 16.15 33788 46.5739 -1.04656e+06 3.68283e+07 - 16.17 33811.6 46.601 -1.04689e+06 3.6829e+07 - 16.19 33835.7 46.6289 -1.04724e+06 3.683e+07 - 16.21 33860.4 46.6575 -1.0476e+06 3.68313e+07 - 16.23 33885.7 46.6868 -1.04797e+06 3.6833e+07 - 16.25 33911.5 46.7169 -1.04835e+06 3.68349e+07 - 16.27 33937.8 46.7477 -1.04875e+06 3.68371e+07 - 16.29 33964.6 46.7792 -1.04916e+06 3.68396e+07 - 16.31 33991.9 46.8113 -1.04957e+06 3.68423e+07 - 16.33 34019.8 46.8441 -1.05e+06 3.68454e+07 - 16.35 34048.1 46.8776 -1.05044e+06 3.68487e+07 - 16.37 34076.9 46.9117 -1.05089e+06 3.68523e+07 - 16.39 34106.3 46.9463 -1.05135e+06 3.68561e+07 - 16.41 34136.1 46.9816 -1.05182e+06 3.68602e+07 - 16.43 34166.4 47.0175 -1.05231e+06 3.68645e+07 - 16.45 34197.1 47.0539 -1.0528e+06 3.68691e+07 - 16.47 34228.3 47.0909 -1.0533e+06 3.68739e+07 - 16.49 34260 47.1284 -1.05381e+06 3.6879e+07 - 16.51 34292.1 47.1665 -1.05433e+06 3.68842e+07 - 16.53 34324.7 47.205 -1.05486e+06 3.68898e+07 - 16.55 34357.7 47.244 -1.0554e+06 3.68955e+07 - 16.57 34391.1 47.2836 -1.05595e+06 3.69014e+07 - 16.59 34425 47.3235 -1.05651e+06 3.69076e+07 - 16.61 34459.2 47.364 -1.05708e+06 3.6914e+07 - 16.63 34493.9 47.4048 -1.05765e+06 3.69205e+07 - 16.65 34529 47.4461 -1.05824e+06 3.69273e+07 - 16.67 34564.5 47.4878 -1.05883e+06 3.69343e+07 - 16.69 34600.4 47.5299 -1.05943e+06 3.69414e+07 - 16.71 34636.7 47.5724 -1.06004e+06 3.69488e+07 - 16.73 34673.3 47.6152 -1.06066e+06 3.69563e+07 - 16.75 34710.3 47.6584 -1.06128e+06 3.6964e+07 - 16.77 34747.8 47.702 -1.06192e+06 3.69718e+07 - 16.79 34785.5 47.7459 -1.06256e+06 3.69799e+07 - 16.81 34823.7 47.79 -1.06321e+06 3.69881e+07 - 16.83 34862.2 47.8345 -1.06386e+06 3.69965e+07 - 16.85 34901 47.8793 -1.06452e+06 3.7005e+07 - 16.87 34940.2 47.9244 -1.06519e+06 3.70137e+07 - 16.89 34979.7 47.9697 -1.06587e+06 3.70225e+07 - 16.91 35019.6 48.0153 -1.06655e+06 3.70315e+07 - 16.93 35059.8 48.0611 -1.06724e+06 3.70406e+07 - 16.95 35100.3 48.1072 -1.06794e+06 3.70499e+07 - 16.97 35141.1 48.1535 -1.06864e+06 3.70593e+07 - 16.99 35182.2 48.1999 -1.06935e+06 3.70688e+07 - 17.01 35223.7 48.2466 -1.07007e+06 3.70784e+07 - 17.03 35265.5 48.2935 -1.07079e+06 3.70882e+07 - 17.05 35307.5 48.3405 -1.07152e+06 3.70981e+07 - 17.07 35349.9 48.3877 -1.07225e+06 3.71081e+07 - 17.09 35392.5 48.4351 -1.07299e+06 3.71183e+07 - 17.11 35435.4 48.4826 -1.07374e+06 3.71285e+07 - 17.13 35478.6 48.5302 -1.07449e+06 3.71388e+07 - 17.15 35522.1 48.5779 -1.07524e+06 3.71493e+07 - 17.17 35565.8 48.6258 -1.07601e+06 3.71599e+07 - 17.19 35609.8 48.6737 -1.07677e+06 3.71705e+07 - 17.21 35654 48.7217 -1.07754e+06 3.71812e+07 - 17.23 35698.6 48.7698 -1.07832e+06 3.71921e+07 - 17.25 35743.3 48.818 -1.0791e+06 3.7203e+07 - 17.27 35788.3 48.8663 -1.07989e+06 3.7214e+07 - 17.29 35833.6 48.9145 -1.08068e+06 3.72251e+07 - 17.31 35879.1 48.9629 -1.08147e+06 3.72363e+07 - 17.33 35924.8 49.0112 -1.08227e+06 3.72475e+07 - 17.35 35970.7 49.0596 -1.08307e+06 3.72588e+07 - 17.37 36016.9 49.108 -1.08388e+06 3.72702e+07 - 17.39 36063.3 49.1564 -1.08469e+06 3.72817e+07 - 17.41 36109.9 49.2047 -1.08551e+06 3.72932e+07 - 17.43 36156.7 49.2531 -1.08633e+06 3.73048e+07 - 17.45 36203.7 49.3014 -1.08715e+06 3.73164e+07 - 17.47 36250.9 49.3497 -1.08798e+06 3.73281e+07 - 17.49 36298.3 49.398 -1.08881e+06 3.73399e+07 - 17.51 36345.9 49.4462 -1.08965e+06 3.73517e+07 - 17.53 36393.7 49.4944 -1.09048e+06 3.73635e+07 - 17.55 36441.7 49.5425 -1.09133e+06 3.73754e+07 - 17.57 36489.8 49.5905 -1.09217e+06 3.73874e+07 - 17.59 36538.1 49.6384 -1.09302e+06 3.73993e+07 - 17.61 36586.6 49.6862 -1.09387e+06 3.74114e+07 - 17.63 36635.3 49.734 -1.09472e+06 3.74234e+07 - 17.65 36684.2 49.7816 -1.09558e+06 3.74355e+07 - 17.67 36733.2 49.8292 -1.09644e+06 3.74477e+07 - 17.69 36782.3 49.8766 -1.0973e+06 3.74598e+07 - 17.71 36831.6 49.9239 -1.09816e+06 3.7472e+07 - 17.73 36881.1 49.971 -1.09903e+06 3.74842e+07 - 17.75 36930.7 50.018 -1.0999e+06 3.74965e+07 - 17.77 36980.4 50.0649 -1.10077e+06 3.75087e+07 - 17.79 37030.3 50.1116 -1.10165e+06 3.7521e+07 - 17.81 37080.3 50.1582 -1.10252e+06 3.75333e+07 - 17.83 37130.5 50.2046 -1.1034e+06 3.75456e+07 - 17.85 37180.7 50.2508 -1.10428e+06 3.75579e+07 - 17.87 37231.1 50.2968 -1.10516e+06 3.75703e+07 - 17.89 37281.6 50.3427 -1.10605e+06 3.75826e+07 - 17.91 37332.3 50.3884 -1.10693e+06 3.7595e+07 - 17.93 37383 50.4338 -1.10782e+06 3.76073e+07 - 17.95 37433.9 50.4791 -1.10871e+06 3.76197e+07 - 17.97 37484.8 50.5242 -1.1096e+06 3.76321e+07 - 17.99 37535.9 50.569 -1.1105e+06 3.76445e+07 - 18.01 37587 50.6137 -1.11139e+06 3.76568e+07 - 18.03 37638.3 50.6581 -1.11229e+06 3.76692e+07 - 18.05 37689.6 50.7023 -1.11318e+06 3.76816e+07 - 18.07 37741.1 50.7462 -1.11408e+06 3.76939e+07 - 18.09 37792.6 50.79 -1.11498e+06 3.77063e+07 - 18.11 37844.2 50.8335 -1.11588e+06 3.77186e+07 - 18.13 37895.8 50.8767 -1.11678e+06 3.7731e+07 - 18.15 37947.6 50.9197 -1.11768e+06 3.77433e+07 - 18.17 37999.4 50.9624 -1.11859e+06 3.77556e+07 - 18.19 38051.3 51.0049 -1.11949e+06 3.77679e+07 - 18.21 38103.3 51.0471 -1.1204e+06 3.77802e+07 - 18.23 38155.3 51.0891 -1.1213e+06 3.77925e+07 - 18.25 38207.4 51.1307 -1.12221e+06 3.78048e+07 - 18.27 38259.5 51.1721 -1.12312e+06 3.7817e+07 - 18.29 38311.7 51.2133 -1.12402e+06 3.78292e+07 - 18.31 38363.9 51.2541 -1.12493e+06 3.78414e+07 - 18.33 38416.2 51.2947 -1.12584e+06 3.78536e+07 - 18.35 38468.5 51.3349 -1.12675e+06 3.78658e+07 - 18.37 38520.9 51.3749 -1.12766e+06 3.78779e+07 - 18.39 38573.3 51.4146 -1.12857e+06 3.789e+07 - 18.41 38625.7 51.454 -1.12948e+06 3.79021e+07 - 18.43 38678.1 51.4931 -1.13039e+06 3.79141e+07 - 18.45 38730.6 51.5319 -1.1313e+06 3.79261e+07 - 18.47 38783.1 51.5704 -1.13221e+06 3.79381e+07 - 18.49 38835.7 51.6086 -1.13312e+06 3.79501e+07 - 18.51 38888.2 51.6464 -1.13402e+06 3.7962e+07 - 18.53 38940.8 51.684 -1.13493e+06 3.7974e+07 - 18.55 38993.4 51.7212 -1.13584e+06 3.79858e+07 - 18.57 39046 51.7581 -1.13675e+06 3.79977e+07 - 18.59 39098.6 51.7947 -1.13766e+06 3.80095e+07 - 18.61 39151.1 51.831 -1.13857e+06 3.80212e+07 - 18.63 39203.7 51.867 -1.13948e+06 3.8033e+07 - 18.65 39256.3 51.9026 -1.14038e+06 3.80447e+07 - 18.67 39308.9 51.9379 -1.14129e+06 3.80563e+07 - 18.69 39361.5 51.9729 -1.1422e+06 3.8068e+07 - 18.71 39414.1 52.0076 -1.1431e+06 3.80795e+07 - 18.73 39466.6 52.0419 -1.14401e+06 3.80911e+07 - 18.75 39519.2 52.0759 -1.14491e+06 3.81026e+07 - 18.77 39571.7 52.1095 -1.14582e+06 3.81141e+07 - 18.79 39624.2 52.1428 -1.14672e+06 3.81255e+07 - 18.81 39676.7 52.1758 -1.14762e+06 3.81369e+07 - 18.83 39729.1 52.2085 -1.14852e+06 3.81482e+07 - 18.85 39781.5 52.2408 -1.14942e+06 3.81595e+07 - 18.87 39833.9 52.2728 -1.15032e+06 3.81708e+07 - 18.89 39886.2 52.3044 -1.15122e+06 3.8182e+07 - 18.91 39938.5 52.3357 -1.15211e+06 3.81932e+07 - 18.93 39990.8 52.3667 -1.15301e+06 3.82043e+07 - 18.95 40043 52.3973 -1.1539e+06 3.82154e+07 - 18.97 40095.1 52.4275 -1.1548e+06 3.82264e+07 - 18.99 40147.2 52.4575 -1.15569e+06 3.82374e+07 - 19.01 40199.3 52.4871 -1.15658e+06 3.82484e+07 - 19.03 40251.3 52.5163 -1.15747e+06 3.82593e+07 - 19.05 40303.2 52.5453 -1.15836e+06 3.82701e+07 - 19.07 40355.1 52.5738 -1.15924e+06 3.82809e+07 - 19.09 40406.9 52.6021 -1.16013e+06 3.82917e+07 - 19.11 40458.6 52.63 -1.16101e+06 3.83024e+07 - 19.13 40510.3 52.6575 -1.16189e+06 3.83131e+07 - 19.15 40561.9 52.6847 -1.16277e+06 3.83237e+07 - 19.17 40613.4 52.7116 -1.16365e+06 3.83343e+07 - 19.19 40664.9 52.7381 -1.16453e+06 3.83448e+07 - 19.21 40716.2 52.7643 -1.1654e+06 3.83553e+07 - 19.23 40767.5 52.7902 -1.16627e+06 3.83657e+07 - 19.25 40818.7 52.8157 -1.16715e+06 3.83761e+07 - 19.27 40869.8 52.8409 -1.16801e+06 3.83864e+07 - 19.29 40920.8 52.8658 -1.16888e+06 3.83967e+07 - 19.31 40971.7 52.8903 -1.16975e+06 3.84069e+07 - 19.33 41022.5 52.9145 -1.17061e+06 3.84171e+07 - 19.35 41073.2 52.9384 -1.17147e+06 3.84272e+07 - 19.37 41123.8 52.9619 -1.17233e+06 3.84373e+07 - 19.39 41174.2 52.9851 -1.17319e+06 3.84473e+07 - 19.41 41224.6 53.008 -1.17404e+06 3.84573e+07 - 19.43 41274.9 53.0305 -1.17489e+06 3.84672e+07 - 19.45 41325 53.0527 -1.17574e+06 3.84771e+07 - 19.47 41375.1 53.0746 -1.17659e+06 3.84869e+07 - 19.49 41425 53.0962 -1.17744e+06 3.84967e+07 - 19.51 41474.7 53.1175 -1.17828e+06 3.85064e+07 - 19.53 41524.4 53.1384 -1.17912e+06 3.85161e+07 - 19.55 41573.9 53.159 -1.17996e+06 3.85257e+07 - 19.57 41623.3 53.1793 -1.18079e+06 3.85353e+07 - 19.59 41672.6 53.1993 -1.18163e+06 3.85448e+07 - 19.61 41721.7 53.219 -1.18246e+06 3.85543e+07 - 19.63 41770.7 53.2384 -1.18329e+06 3.85637e+07 - 19.65 41819.5 53.2574 -1.18411e+06 3.8573e+07 - 19.67 41868.2 53.2762 -1.18493e+06 3.85823e+07 - 19.69 41916.7 53.2946 -1.18575e+06 3.85916e+07 - 19.71 41965.1 53.3128 -1.18657e+06 3.86008e+07 - 19.73 42013.3 53.3306 -1.18738e+06 3.861e+07 - 19.75 42061.4 53.3482 -1.18819e+06 3.86191e+07 - 19.77 42109.3 53.3654 -1.189e+06 3.86281e+07 - 19.79 42157 53.3824 -1.1898e+06 3.86371e+07 - 19.81 42204.6 53.3991 -1.19061e+06 3.86461e+07 - 19.83 42252 53.4155 -1.1914e+06 3.8655e+07 - 19.85 42299.2 53.4316 -1.1922e+06 3.86638e+07 - 19.87 42346.2 53.4474 -1.19299e+06 3.86726e+07 - 19.89 42393.1 53.4629 -1.19378e+06 3.86813e+07 - 19.91 42439.8 53.4782 -1.19457e+06 3.869e+07 - 19.93 42486.3 53.4932 -1.19535e+06 3.86986e+07 - 19.95 42532.6 53.5079 -1.19613e+06 3.87072e+07 - 19.97 42578.7 53.5223 -1.1969e+06 3.87157e+07 - 19.99 42624.6 53.5365 -1.19768e+06 3.87242e+07 + 0.01 435983 733.444 -8.82169e+06 2.47979e+08 + 0.03 435979 733.435 -8.82163e+06 2.47978e+08 + 0.05 435971 733.417 -8.82152e+06 2.47975e+08 + 0.07 435960 733.39 -8.82136e+06 2.4797e+08 + 0.09 435944 733.353 -8.82114e+06 2.47964e+08 + 0.11 435924 733.308 -8.82086e+06 2.47957e+08 + 0.13 435900 733.254 -8.82053e+06 2.47948e+08 + 0.15 435873 733.19 -8.82014e+06 2.47938e+08 + 0.17 435841 733.118 -8.81969e+06 2.47926e+08 + 0.19 435806 733.036 -8.8192e+06 2.47912e+08 + 0.21 435766 732.945 -8.81864e+06 2.47897e+08 + 0.23 435723 732.845 -8.81803e+06 2.47881e+08 + 0.25 435675 732.737 -8.81737e+06 2.47863e+08 + 0.27 435624 732.619 -8.81664e+06 2.47844e+08 + 0.29 435569 732.492 -8.81587e+06 2.47823e+08 + 0.31 435509 732.356 -8.81504e+06 2.47801e+08 + 0.33 435446 732.211 -8.81415e+06 2.47777e+08 + 0.35 435379 732.057 -8.81321e+06 2.47752e+08 + 0.37 435308 731.894 -8.81221e+06 2.47725e+08 + 0.39 435233 731.722 -8.81115e+06 2.47697e+08 + 0.41 435154 731.541 -8.81005e+06 2.47667e+08 + 0.43 435071 731.35 -8.80888e+06 2.47636e+08 + 0.45 434985 731.151 -8.80766e+06 2.47603e+08 + 0.47 434894 730.943 -8.80638e+06 2.47569e+08 + 0.49 434799 730.726 -8.80505e+06 2.47533e+08 + 0.51 434701 730.5 -8.80367e+06 2.47496e+08 + 0.53 434598 730.265 -8.80222e+06 2.47457e+08 + 0.55 434492 730.021 -8.80072e+06 2.47417e+08 + 0.57 434381 729.769 -8.79917e+06 2.47376e+08 + 0.59 434267 729.507 -8.79756e+06 2.47333e+08 + 0.61 434149 729.236 -8.7959e+06 2.47288e+08 + 0.63 434027 728.956 -8.79418e+06 2.47242e+08 + 0.65 433901 728.668 -8.7924e+06 2.47194e+08 + 0.67 433771 728.37 -8.79057e+06 2.47145e+08 + 0.69 433637 728.064 -8.78868e+06 2.47095e+08 + 0.71 433499 727.749 -8.78674e+06 2.47043e+08 + 0.73 433358 727.425 -8.78474e+06 2.46989e+08 + 0.75 433212 727.092 -8.78268e+06 2.46934e+08 + 0.77 433063 726.75 -8.78057e+06 2.46878e+08 + 0.79 432909 726.4 -8.7784e+06 2.4682e+08 + 0.81 432752 726.041 -8.77618e+06 2.4676e+08 + 0.83 432591 725.672 -8.7739e+06 2.46699e+08 + 0.85 432426 725.296 -8.77157e+06 2.46637e+08 + 0.87 432257 724.91 -8.76918e+06 2.46573e+08 + 0.89 432084 724.516 -8.76673e+06 2.46507e+08 + 0.91 431907 724.113 -8.76423e+06 2.4644e+08 + 0.93 431727 723.701 -8.76167e+06 2.46372e+08 + 0.95 431542 723.28 -8.75906e+06 2.46302e+08 + 0.97 431354 722.851 -8.75639e+06 2.4623e+08 + 0.99 431162 722.413 -8.75367e+06 2.46157e+08 + 1.01 430966 721.967 -8.75088e+06 2.46083e+08 + 1.03 430766 721.512 -8.74805e+06 2.46007e+08 + 1.05 430562 721.048 -8.74515e+06 2.4593e+08 + 1.07 430354 720.576 -8.7422e+06 2.45851e+08 + 1.09 430143 720.095 -8.7392e+06 2.4577e+08 + 1.11 429927 719.605 -8.73614e+06 2.45689e+08 + 1.13 429708 719.108 -8.73302e+06 2.45605e+08 + 1.15 429485 718.601 -8.72984e+06 2.4552e+08 + 1.17 429258 718.086 -8.72661e+06 2.45434e+08 + 1.19 429028 717.563 -8.72333e+06 2.45346e+08 + 1.21 428793 717.031 -8.71999e+06 2.45257e+08 + 1.23 428555 716.49 -8.71659e+06 2.45166e+08 + 1.25 428313 715.942 -8.71313e+06 2.45074e+08 + 1.27 428067 715.385 -8.70962e+06 2.4498e+08 + 1.29 427817 714.819 -8.70605e+06 2.44884e+08 + 1.31 427563 714.245 -8.70243e+06 2.44788e+08 + 1.33 427306 713.663 -8.69875e+06 2.44689e+08 + 1.35 427044 713.073 -8.69501e+06 2.44589e+08 + 1.37 426779 712.474 -8.69122e+06 2.44488e+08 + 1.39 426511 711.867 -8.68737e+06 2.44385e+08 + 1.41 426238 711.251 -8.68346e+06 2.44281e+08 + 1.43 425962 710.628 -8.6795e+06 2.44175e+08 + 1.45 425682 709.996 -8.67548e+06 2.44068e+08 + 1.47 425398 709.357 -8.67141e+06 2.43959e+08 + 1.49 425110 708.709 -8.66728e+06 2.43849e+08 + 1.51 424819 708.053 -8.66309e+06 2.43737e+08 + 1.53 424523 707.388 -8.65884e+06 2.43624e+08 + 1.55 424224 706.716 -8.65454e+06 2.43509e+08 + 1.57 423922 706.036 -8.65018e+06 2.43393e+08 + 1.59 423615 705.348 -8.64577e+06 2.43275e+08 + 1.61 423305 704.651 -8.64129e+06 2.43156e+08 + 1.63 422991 703.947 -8.63677e+06 2.43035e+08 + 1.65 422674 703.235 -8.63218e+06 2.42913e+08 + 1.67 422352 702.515 -8.62754e+06 2.42789e+08 + 1.69 422027 701.787 -8.62284e+06 2.42664e+08 + 1.71 421698 701.051 -8.61808e+06 2.42537e+08 + 1.73 421366 700.308 -8.61327e+06 2.42408e+08 + 1.75 421030 699.556 -8.6084e+06 2.42279e+08 + 1.77 420690 698.797 -8.60348e+06 2.42147e+08 + 1.79 420346 698.03 -8.59849e+06 2.42015e+08 + 1.81 419999 697.256 -8.59345e+06 2.4188e+08 + 1.83 419648 696.473 -8.58836e+06 2.41744e+08 + 1.85 419294 695.683 -8.5832e+06 2.41607e+08 + 1.87 418936 694.886 -8.57799e+06 2.41468e+08 + 1.89 418574 694.081 -8.57272e+06 2.41328e+08 + 1.91 418208 693.268 -8.5674e+06 2.41186e+08 + 1.93 417839 692.448 -8.56201e+06 2.41043e+08 + 1.95 417467 691.62 -8.55657e+06 2.40898e+08 + 1.97 417090 690.785 -8.55108e+06 2.40752e+08 + 1.99 416710 689.942 -8.54552e+06 2.40604e+08 + 2.01 416327 689.092 -8.53991e+06 2.40455e+08 + 2.03 415939 688.235 -8.53424e+06 2.40304e+08 + 2.05 415548 687.37 -8.52852e+06 2.40152e+08 + 2.07 415154 686.498 -8.52274e+06 2.39998e+08 + 2.09 414756 685.619 -8.5169e+06 2.39842e+08 + 2.11 414354 684.732 -8.511e+06 2.39686e+08 + 2.13 413949 683.839 -8.50504e+06 2.39527e+08 + 2.15 413541 682.938 -8.49903e+06 2.39367e+08 + 2.17 413128 682.03 -8.49296e+06 2.39206e+08 + 2.19 412713 681.114 -8.48684e+06 2.39043e+08 + 2.21 412293 680.192 -8.48065e+06 2.38879e+08 + 2.23 411870 679.263 -8.47441e+06 2.38713e+08 + 2.25 411444 678.327 -8.46811e+06 2.38546e+08 + 2.27 411014 677.383 -8.46175e+06 2.38377e+08 + 2.29 410581 676.433 -8.45534e+06 2.38206e+08 + 2.31 410144 675.476 -8.44887e+06 2.38034e+08 + 2.33 409703 674.512 -8.44234e+06 2.37861e+08 + 2.35 409259 673.541 -8.43575e+06 2.37686e+08 + 2.37 408812 672.564 -8.42911e+06 2.3751e+08 + 2.39 408361 671.579 -8.42241e+06 2.37332e+08 + 2.41 407907 670.588 -8.41565e+06 2.37152e+08 + 2.43 407449 669.59 -8.40883e+06 2.36972e+08 + 2.45 406988 668.586 -8.40196e+06 2.36789e+08 + 2.47 406523 667.575 -8.39503e+06 2.36605e+08 + 2.49 406055 666.557 -8.38804e+06 2.3642e+08 + 2.51 405583 665.533 -8.38099e+06 2.36233e+08 + 2.53 405108 664.502 -8.37388e+06 2.36044e+08 + 2.55 404630 663.465 -8.36672e+06 2.35854e+08 + 2.57 404148 662.422 -8.3595e+06 2.35663e+08 + 2.59 403663 661.372 -8.35222e+06 2.3547e+08 + 2.61 403174 660.315 -8.34489e+06 2.35276e+08 + 2.63 402683 659.253 -8.3375e+06 2.3508e+08 + 2.65 402187 658.184 -8.33004e+06 2.34882e+08 + 2.67 401689 657.109 -8.32254e+06 2.34683e+08 + 2.69 401187 656.027 -8.31497e+06 2.34483e+08 + 2.71 400681 654.94 -8.30735e+06 2.34281e+08 + 2.73 400173 653.846 -8.29966e+06 2.34077e+08 + 2.75 399661 652.746 -8.29192e+06 2.33872e+08 + 2.77 399146 651.64 -8.28413e+06 2.33666e+08 + 2.79 398627 650.529 -8.27627e+06 2.33458e+08 + 2.81 398105 649.411 -8.26836e+06 2.33248e+08 + 2.83 397580 648.287 -8.26039e+06 2.33037e+08 + 2.85 397052 647.158 -8.25236e+06 2.32825e+08 + 2.87 396520 646.022 -8.24427e+06 2.32611e+08 + 2.89 395985 644.881 -8.23613e+06 2.32395e+08 + 2.91 395447 643.734 -8.22793e+06 2.32178e+08 + 2.93 394906 642.581 -8.21967e+06 2.3196e+08 + 2.95 394361 641.423 -8.21135e+06 2.3174e+08 + 2.97 393814 640.259 -8.20298e+06 2.31518e+08 + 2.99 393263 639.089 -8.19455e+06 2.31295e+08 + 3.01 392709 637.914 -8.18606e+06 2.3107e+08 + 3.03 392151 636.733 -8.17751e+06 2.30844e+08 + 3.05 391591 635.546 -8.1689e+06 2.30617e+08 + 3.07 391027 634.355 -8.16024e+06 2.30388e+08 + 3.09 390460 633.158 -8.15152e+06 2.30157e+08 + 3.11 389890 631.955 -8.14274e+06 2.29925e+08 + 3.13 389317 630.747 -8.1339e+06 2.29692e+08 + 3.15 388741 629.534 -8.12501e+06 2.29457e+08 + 3.17 388162 628.316 -8.11606e+06 2.2922e+08 + 3.19 387579 627.092 -8.10705e+06 2.28982e+08 + 3.21 386994 625.864 -8.09798e+06 2.28743e+08 + 3.23 386405 624.63 -8.08886e+06 2.28502e+08 + 3.25 385814 623.391 -8.07968e+06 2.28259e+08 + 3.27 385219 622.147 -8.07044e+06 2.28015e+08 + 3.29 384621 620.898 -8.06114e+06 2.2777e+08 + 3.31 384020 619.644 -8.05179e+06 2.27523e+08 + 3.33 383417 618.385 -8.04238e+06 2.27275e+08 + 3.35 382810 617.122 -8.03291e+06 2.27025e+08 + 3.37 382200 615.853 -8.02338e+06 2.26773e+08 + 3.39 381587 614.58 -8.0138e+06 2.2652e+08 + 3.41 380971 613.302 -8.00416e+06 2.26266e+08 + 3.43 380353 612.019 -7.99446e+06 2.2601e+08 + 3.45 379731 610.732 -7.98471e+06 2.25753e+08 + 3.47 379107 609.44 -7.97489e+06 2.25494e+08 + 3.49 378479 608.144 -7.96502e+06 2.25234e+08 + 3.51 377849 606.843 -7.9551e+06 2.24972e+08 + 3.53 377215 605.537 -7.94511e+06 2.24708e+08 + 3.55 376579 604.227 -7.93507e+06 2.24444e+08 + 3.57 375940 602.913 -7.92498e+06 2.24177e+08 + 3.59 375298 601.594 -7.91482e+06 2.2391e+08 + 3.61 374653 600.271 -7.90461e+06 2.2364e+08 + 3.63 374005 598.944 -7.89434e+06 2.2337e+08 + 3.65 373355 597.613 -7.88401e+06 2.23098e+08 + 3.67 372701 596.277 -7.87363e+06 2.22824e+08 + 3.69 372045 594.937 -7.86319e+06 2.22549e+08 + 3.71 371386 593.593 -7.8527e+06 2.22272e+08 + 3.73 370725 592.245 -7.84215e+06 2.21994e+08 + 3.75 370060 590.894 -7.83154e+06 2.21715e+08 + 3.77 369393 589.538 -7.82087e+06 2.21434e+08 + 3.79 368723 588.178 -7.81015e+06 2.21151e+08 + 3.81 368050 586.814 -7.79937e+06 2.20867e+08 + 3.83 367375 585.447 -7.78854e+06 2.20582e+08 + 3.85 366696 584.075 -7.77765e+06 2.20295e+08 + 3.87 366016 582.7 -7.7667e+06 2.20007e+08 + 3.89 365332 581.322 -7.7557e+06 2.19717e+08 + 3.91 364646 579.939 -7.74464e+06 2.19426e+08 + 3.93 363957 578.553 -7.73352e+06 2.19133e+08 + 3.95 363266 577.164 -7.72235e+06 2.18839e+08 + 3.97 362571 575.771 -7.71113e+06 2.18544e+08 + 3.99 361875 574.374 -7.69984e+06 2.18246e+08 + 4.01 361175 572.974 -7.6885e+06 2.17948e+08 + 4.03 360473 571.571 -7.67711e+06 2.17648e+08 + 4.05 359769 570.164 -7.66566e+06 2.17347e+08 + 4.07 359062 568.754 -7.65416e+06 2.17044e+08 + 4.09 358352 567.341 -7.6426e+06 2.1674e+08 + 4.11 357640 565.924 -7.63098e+06 2.16434e+08 + 4.13 356925 564.504 -7.61931e+06 2.16127e+08 + 4.15 356208 563.082 -7.60758e+06 2.15818e+08 + 4.17 355488 561.656 -7.5958e+06 2.15508e+08 + 4.19 354766 560.227 -7.58397e+06 2.15197e+08 + 4.21 354042 558.795 -7.57208e+06 2.14884e+08 + 4.23 353315 557.36 -7.56013e+06 2.1457e+08 + 4.25 352585 555.922 -7.54813e+06 2.14254e+08 + 4.27 351853 554.481 -7.53608e+06 2.13937e+08 + 4.29 351119 553.037 -7.52397e+06 2.13619e+08 + 4.31 350382 551.591 -7.51181e+06 2.13299e+08 + 4.33 349643 550.142 -7.49959e+06 2.12977e+08 + 4.35 348901 548.69 -7.48732e+06 2.12654e+08 + 4.37 348158 547.236 -7.47499e+06 2.1233e+08 + 4.39 347411 545.779 -7.46262e+06 2.12005e+08 + 4.41 346663 544.319 -7.45018e+06 2.11678e+08 + 4.43 345912 542.857 -7.4377e+06 2.11349e+08 + 4.45 345159 541.392 -7.42516e+06 2.11019e+08 + 4.47 344404 539.925 -7.41256e+06 2.10688e+08 + 4.49 343646 538.455 -7.39992e+06 2.10356e+08 + 4.51 342886 536.984 -7.38722e+06 2.10022e+08 + 4.53 342124 535.509 -7.37447e+06 2.09686e+08 + 4.55 341360 534.033 -7.36166e+06 2.0935e+08 + 4.57 340593 532.554 -7.3488e+06 2.09011e+08 + 4.59 339825 531.073 -7.33589e+06 2.08672e+08 + 4.61 339054 529.59 -7.32293e+06 2.08331e+08 + 4.63 338281 528.105 -7.30991e+06 2.07989e+08 + 4.65 337506 526.618 -7.29685e+06 2.07645e+08 + 4.67 336728 525.129 -7.28373e+06 2.073e+08 + 4.69 335949 523.638 -7.27056e+06 2.06954e+08 + 4.71 335167 522.145 -7.25733e+06 2.06606e+08 + 4.73 334384 520.65 -7.24406e+06 2.06257e+08 + 4.75 333598 519.153 -7.23073e+06 2.05907e+08 + 4.77 332810 517.654 -7.21735e+06 2.05555e+08 + 4.79 332021 516.154 -7.20393e+06 2.05202e+08 + 4.81 331229 514.652 -7.19045e+06 2.04847e+08 + 4.83 330435 513.148 -7.17692e+06 2.04491e+08 + 4.85 329640 511.643 -7.16333e+06 2.04134e+08 + 4.87 328842 510.136 -7.1497e+06 2.03776e+08 + 4.89 328042 508.627 -7.13602e+06 2.03416e+08 + 4.91 327241 507.117 -7.12229e+06 2.03055e+08 + 4.93 326437 505.605 -7.10851e+06 2.02692e+08 + 4.95 325632 504.092 -7.09468e+06 2.02329e+08 + 4.97 324825 502.578 -7.0808e+06 2.01964e+08 + 4.99 324015 501.062 -7.06686e+06 2.01597e+08 + 5.01 323204 499.545 -7.05289e+06 2.01229e+08 + 5.03 322392 498.027 -7.03886e+06 2.0086e+08 + 5.05 321577 496.507 -7.02478e+06 2.0049e+08 + 5.07 320760 494.987 -7.01065e+06 2.00119e+08 + 5.09 319942 493.465 -6.99648e+06 1.99746e+08 + 5.11 319122 491.942 -6.98225e+06 1.99372e+08 + 5.13 318300 490.418 -6.96798e+06 1.98996e+08 + 5.15 317477 488.893 -6.95366e+06 1.98619e+08 + 5.17 316652 487.367 -6.9393e+06 1.98241e+08 + 5.19 315825 485.84 -6.92488e+06 1.97862e+08 + 5.21 314996 484.312 -6.91042e+06 1.97482e+08 + 5.23 314166 482.783 -6.89591e+06 1.971e+08 + 5.25 313334 481.253 -6.88136e+06 1.96717e+08 + 5.27 312501 479.723 -6.86676e+06 1.96333e+08 + 5.29 311665 478.192 -6.85211e+06 1.95947e+08 + 5.31 310829 476.66 -6.83741e+06 1.9556e+08 + 5.33 309990 475.128 -6.82267e+06 1.95172e+08 + 5.35 309151 473.595 -6.80789e+06 1.94783e+08 + 5.37 308309 472.061 -6.79306e+06 1.94393e+08 + 5.39 307466 470.527 -6.77818e+06 1.94001e+08 + 5.41 306622 468.992 -6.76326e+06 1.93608e+08 + 5.43 305776 467.457 -6.74829e+06 1.93214e+08 + 5.45 304928 465.921 -6.73328e+06 1.92819e+08 + 5.47 304079 464.385 -6.71822e+06 1.92422e+08 + 5.49 303229 462.849 -6.70312e+06 1.92025e+08 + 5.51 302377 461.312 -6.68798e+06 1.91626e+08 + 5.53 301524 459.775 -6.67279e+06 1.91226e+08 + 5.55 300670 458.238 -6.65756e+06 1.90825e+08 + 5.57 299814 456.7 -6.64229e+06 1.90422e+08 + 5.59 298957 455.163 -6.62697e+06 1.90019e+08 + 5.61 298098 453.625 -6.61161e+06 1.89614e+08 + 5.63 297238 452.087 -6.59621e+06 1.89208e+08 + 5.65 296377 450.549 -6.58077e+06 1.88801e+08 + 5.67 295514 449.011 -6.56528e+06 1.88393e+08 + 5.69 294651 447.473 -6.54976e+06 1.87984e+08 + 5.71 293786 445.936 -6.53419e+06 1.87574e+08 + 5.73 292920 444.398 -6.51858e+06 1.87162e+08 + 5.75 292052 442.86 -6.50293e+06 1.86749e+08 + 5.77 291184 441.323 -6.48724e+06 1.86336e+08 + 5.79 290314 439.786 -6.47151e+06 1.85921e+08 + 5.81 289443 438.249 -6.45574e+06 1.85505e+08 + 5.83 288571 436.712 -6.43993e+06 1.85088e+08 + 5.85 287698 435.175 -6.42408e+06 1.8467e+08 + 5.87 286824 433.639 -6.40819e+06 1.84251e+08 + 5.89 285949 432.104 -6.39226e+06 1.83831e+08 + 5.91 285073 430.568 -6.3763e+06 1.83409e+08 + 5.93 284195 429.034 -6.3603e+06 1.82987e+08 + 5.95 283317 427.499 -6.34425e+06 1.82564e+08 + 5.97 282438 425.965 -6.32818e+06 1.82139e+08 + 5.99 281558 424.432 -6.31206e+06 1.81714e+08 + 6.01 280676 422.9 -6.29591e+06 1.81287e+08 + 6.03 279794 421.367 -6.27972e+06 1.8086e+08 + 6.05 278911 419.836 -6.26349e+06 1.80431e+08 + 6.07 278027 418.305 -6.24723e+06 1.80002e+08 + 6.09 277142 416.775 -6.23093e+06 1.79571e+08 + 6.11 276257 415.246 -6.2146e+06 1.79139e+08 + 6.13 275370 413.718 -6.19823e+06 1.78707e+08 + 6.15 274483 412.19 -6.18183e+06 1.78273e+08 + 6.17 273595 410.664 -6.1654e+06 1.77839e+08 + 6.19 272706 409.138 -6.14893e+06 1.77404e+08 + 6.21 271816 407.613 -6.13242e+06 1.76967e+08 + 6.23 270926 406.089 -6.11588e+06 1.7653e+08 + 6.25 270035 404.566 -6.09931e+06 1.76092e+08 + 6.27 269143 403.044 -6.08271e+06 1.75653e+08 + 6.29 268250 401.524 -6.06608e+06 1.75212e+08 + 6.31 267357 400.004 -6.04941e+06 1.74771e+08 + 6.33 266464 398.485 -6.03271e+06 1.74329e+08 + 6.35 265569 396.968 -6.01598e+06 1.73887e+08 + 6.37 264674 395.452 -5.99922e+06 1.73443e+08 + 6.39 263779 393.937 -5.98243e+06 1.72998e+08 + 6.41 262882 392.423 -5.9656e+06 1.72553e+08 + 6.43 261986 390.91 -5.94875e+06 1.72107e+08 + 6.45 261089 389.399 -5.93187e+06 1.71659e+08 + 6.47 260191 387.889 -5.91496e+06 1.71211e+08 + 6.49 259293 386.381 -5.89802e+06 1.70763e+08 + 6.51 258394 384.874 -5.88106e+06 1.70313e+08 + 6.53 257495 383.368 -5.86406e+06 1.69862e+08 + 6.55 256596 381.864 -5.84704e+06 1.69411e+08 + 6.57 255696 380.362 -5.82999e+06 1.68959e+08 + 6.59 254796 378.86 -5.81291e+06 1.68506e+08 + 6.61 253895 377.361 -5.79581e+06 1.68052e+08 + 6.63 252994 375.863 -5.77868e+06 1.67598e+08 + 6.65 252093 374.367 -5.76152e+06 1.67143e+08 + 6.67 251192 372.872 -5.74434e+06 1.66687e+08 + 6.69 250290 371.379 -5.72713e+06 1.6623e+08 + 6.71 249388 369.888 -5.7099e+06 1.65773e+08 + 6.73 248486 368.398 -5.69265e+06 1.65314e+08 + 6.75 247583 366.91 -5.67537e+06 1.64855e+08 + 6.77 246680 365.424 -5.65807e+06 1.64396e+08 + 6.79 245778 363.94 -5.64075e+06 1.63936e+08 + 6.81 244875 362.458 -5.6234e+06 1.63474e+08 + 6.83 243972 360.977 -5.60603e+06 1.63013e+08 + 6.85 243068 359.499 -5.58864e+06 1.6255e+08 + 6.87 242165 358.022 -5.57123e+06 1.62087e+08 + 6.89 241262 356.548 -5.55379e+06 1.61624e+08 + 6.91 240358 355.075 -5.53634e+06 1.61159e+08 + 6.93 239455 353.604 -5.51887e+06 1.60694e+08 + 6.95 238552 352.136 -5.50137e+06 1.60229e+08 + 6.97 237648 350.669 -5.48386e+06 1.59763e+08 + 6.99 236745 349.205 -5.46633e+06 1.59296e+08 + 7.01 235842 347.742 -5.44878e+06 1.58829e+08 + 7.03 234938 346.282 -5.43121e+06 1.58361e+08 + 7.05 234035 344.824 -5.41362e+06 1.57892e+08 + 7.07 233132 343.369 -5.39602e+06 1.57423e+08 + 7.09 232230 341.915 -5.3784e+06 1.56953e+08 + 7.11 231327 340.464 -5.36077e+06 1.56483e+08 + 7.13 230425 339.015 -5.34312e+06 1.56013e+08 + 7.15 229522 337.568 -5.32545e+06 1.55541e+08 + 7.17 228620 336.124 -5.30777e+06 1.5507e+08 + 7.19 227719 334.682 -5.29007e+06 1.54598e+08 + 7.21 226817 333.242 -5.27236e+06 1.54125e+08 + 7.23 225916 331.805 -5.25464e+06 1.53652e+08 + 7.25 225015 330.37 -5.2369e+06 1.53178e+08 + 7.27 224114 328.938 -5.21915e+06 1.52704e+08 + 7.29 223214 327.508 -5.20139e+06 1.52229e+08 + 7.31 222314 326.08 -5.18361e+06 1.51755e+08 + 7.33 221415 324.655 -5.16583e+06 1.51279e+08 + 7.35 220516 323.233 -5.14803e+06 1.50803e+08 + 7.37 219617 321.813 -5.13022e+06 1.50327e+08 + 7.39 218719 320.396 -5.11241e+06 1.49851e+08 + 7.41 217822 318.982 -5.09458e+06 1.49374e+08 + 7.43 216925 317.57 -5.07674e+06 1.48897e+08 + 7.45 216028 316.161 -5.0589e+06 1.48419e+08 + 7.47 215132 314.754 -5.04105e+06 1.47941e+08 + 7.49 214237 313.35 -5.02319e+06 1.47463e+08 + 7.51 213342 311.949 -5.00532e+06 1.46984e+08 + 7.53 212447 310.551 -4.98744e+06 1.46505e+08 + 7.55 211554 309.155 -4.96956e+06 1.46026e+08 + 7.57 210661 307.763 -4.95168e+06 1.45546e+08 + 7.59 209769 306.373 -4.93378e+06 1.45067e+08 + 7.61 208877 304.985 -4.91589e+06 1.44587e+08 + 7.63 207986 303.601 -4.89799e+06 1.44106e+08 + 7.65 207096 302.22 -4.88008e+06 1.43626e+08 + 7.67 206206 300.841 -4.86217e+06 1.43145e+08 + 7.69 205318 299.466 -4.84426e+06 1.42664e+08 + 7.71 204430 298.093 -4.82634e+06 1.42183e+08 + 7.73 203543 296.724 -4.80843e+06 1.41702e+08 + 7.75 202657 295.357 -4.79051e+06 1.4122e+08 + 7.77 201771 293.994 -4.77259e+06 1.40739e+08 + 7.79 200887 292.633 -4.75467e+06 1.40257e+08 + 7.81 200003 291.275 -4.73675e+06 1.39775e+08 + 7.83 199121 289.921 -4.71883e+06 1.39293e+08 + 7.85 198239 288.57 -4.70091e+06 1.38811e+08 + 7.87 197358 287.221 -4.68299e+06 1.38329e+08 + 7.89 196479 285.876 -4.66507e+06 1.37846e+08 + 7.91 195600 284.534 -4.64716e+06 1.37364e+08 + 7.93 194722 283.195 -4.62924e+06 1.36881e+08 + 7.95 193846 281.86 -4.61134e+06 1.36399e+08 + 7.97 192970 280.527 -4.59343e+06 1.35916e+08 + 7.99 192096 279.198 -4.57553e+06 1.35434e+08 + 8.01 191222 277.872 -4.55763e+06 1.34951e+08 + 8.03 190350 276.549 -4.53974e+06 1.34469e+08 + 8.05 189479 275.229 -4.52186e+06 1.33986e+08 + 8.07 188609 273.913 -4.50398e+06 1.33503e+08 + 8.09 187740 272.6 -4.4861e+06 1.33021e+08 + 8.11 186872 271.291 -4.46824e+06 1.32538e+08 + 8.13 186006 269.984 -4.45038e+06 1.32056e+08 + 8.15 185141 268.682 -4.43253e+06 1.31574e+08 + 8.17 184277 267.382 -4.41469e+06 1.31091e+08 + 8.19 183415 266.086 -4.39685e+06 1.30609e+08 + 8.21 182554 264.793 -4.37903e+06 1.30127e+08 + 8.23 181694 263.504 -4.36121e+06 1.29645e+08 + 8.25 180835 262.218 -4.34341e+06 1.29163e+08 + 8.27 179978 260.936 -4.32562e+06 1.28682e+08 + 8.29 179122 259.657 -4.30784e+06 1.282e+08 + 8.31 178268 258.381 -4.29007e+06 1.27719e+08 + 8.33 177415 257.109 -4.27231e+06 1.27238e+08 + 8.35 176563 255.841 -4.25457e+06 1.26757e+08 + 8.37 175713 254.576 -4.23684e+06 1.26276e+08 + 8.39 174865 253.315 -4.21912e+06 1.25796e+08 + 8.41 174018 252.057 -4.20142e+06 1.25316e+08 + 8.43 173172 250.803 -4.18373e+06 1.24836e+08 + 8.45 172328 249.552 -4.16606e+06 1.24356e+08 + 8.47 171486 248.305 -4.1484e+06 1.23876e+08 + 8.49 170645 247.061 -4.13076e+06 1.23397e+08 + 8.51 169806 245.822 -4.11314e+06 1.22918e+08 + 8.53 168968 244.586 -4.09553e+06 1.2244e+08 + 8.55 168132 243.353 -4.07795e+06 1.21962e+08 + 8.57 167298 242.124 -4.06038e+06 1.21484e+08 + 8.59 166465 240.899 -4.04282e+06 1.21006e+08 + 8.61 165634 239.678 -4.02529e+06 1.20529e+08 + 8.63 164805 238.46 -4.00778e+06 1.20052e+08 + 8.65 163977 237.246 -3.99029e+06 1.19576e+08 + 8.67 163151 236.036 -3.97281e+06 1.191e+08 + 8.69 162327 234.829 -3.95536e+06 1.18625e+08 + 8.71 161505 233.627 -3.93793e+06 1.1815e+08 + 8.73 160685 232.428 -3.92053e+06 1.17675e+08 + 8.75 159866 231.232 -3.90314e+06 1.17201e+08 + 8.77 159049 230.041 -3.88578e+06 1.16727e+08 + 8.79 158235 228.854 -3.86844e+06 1.16254e+08 + 8.81 157422 227.67 -3.85113e+06 1.15781e+08 + 8.83 156610 226.49 -3.83384e+06 1.15309e+08 + 8.85 155801 225.314 -3.81657e+06 1.14837e+08 + 8.87 154994 224.142 -3.79933e+06 1.14366e+08 + 8.89 154188 222.973 -3.78212e+06 1.13895e+08 + 8.91 153385 221.809 -3.76493e+06 1.13425e+08 + 8.93 152584 220.648 -3.74777e+06 1.12956e+08 + 8.95 151784 219.492 -3.73064e+06 1.12487e+08 + 8.97 150987 218.339 -3.71353e+06 1.12018e+08 + 8.99 150191 217.19 -3.69645e+06 1.11551e+08 + 9.01 149398 216.045 -3.6794e+06 1.11083e+08 + 9.03 148607 214.904 -3.66238e+06 1.10617e+08 + 9.05 147818 213.767 -3.64539e+06 1.10151e+08 + 9.07 147031 212.634 -3.62843e+06 1.09686e+08 + 9.09 146246 211.505 -3.6115e+06 1.09222e+08 + 9.11 145463 210.38 -3.5946e+06 1.08758e+08 + 9.13 144682 209.259 -3.57773e+06 1.08295e+08 + 9.15 143903 208.142 -3.56089e+06 1.07832e+08 + 9.17 143127 207.029 -3.54409e+06 1.0737e+08 + 9.19 142353 205.92 -3.52732e+06 1.0691e+08 + 9.21 141581 204.815 -3.51058e+06 1.06449e+08 + 9.23 140811 203.713 -3.49387e+06 1.0599e+08 + 9.25 140044 202.616 -3.4772e+06 1.05531e+08 + 9.27 139278 201.523 -3.46056e+06 1.05073e+08 + 9.29 138515 200.435 -3.44396e+06 1.04616e+08 + 9.31 137755 199.35 -3.42739e+06 1.0416e+08 + 9.33 136996 198.269 -3.41086e+06 1.03704e+08 + 9.35 136240 197.192 -3.39437e+06 1.0325e+08 + 9.37 135486 196.12 -3.37791e+06 1.02796e+08 + 9.39 134735 195.051 -3.36148e+06 1.02343e+08 + 9.41 133986 193.987 -3.3451e+06 1.01891e+08 + 9.43 133239 192.926 -3.32875e+06 1.0144e+08 + 9.45 132495 191.87 -3.31244e+06 1.00989e+08 + 9.47 131753 190.818 -3.29617e+06 1.0054e+08 + 9.49 131013 189.77 -3.27994e+06 1.00091e+08 + 9.51 130276 188.726 -3.26375e+06 9.96436e+07 + 9.53 129542 187.687 -3.2476e+06 9.91969e+07 + 9.55 128810 186.651 -3.23149e+06 9.87512e+07 + 9.57 128080 185.62 -3.21541e+06 9.83064e+07 + 9.59 127353 184.593 -3.19938e+06 9.78626e+07 + 9.61 126628 183.57 -3.1834e+06 9.74198e+07 + 9.63 125906 182.551 -3.16745e+06 9.69779e+07 + 9.65 125186 181.536 -3.15154e+06 9.65371e+07 + 9.67 124469 180.525 -3.13568e+06 9.60973e+07 + 9.69 123754 179.519 -3.11986e+06 9.56585e+07 + 9.71 123042 178.517 -3.10409e+06 9.52207e+07 + 9.73 122333 177.519 -3.08835e+06 9.4784e+07 + 9.75 121626 176.525 -3.07267e+06 9.43483e+07 + 9.77 120921 175.536 -3.05702e+06 9.39137e+07 + 9.79 120220 174.55 -3.04142e+06 9.34802e+07 + 9.81 119521 173.569 -3.02587e+06 9.30477e+07 + 9.83 118824 172.592 -3.01036e+06 9.26164e+07 + 9.85 118130 171.619 -2.9949e+06 9.21861e+07 + 9.87 117439 170.651 -2.97948e+06 9.1757e+07 + 9.89 116751 169.687 -2.96411e+06 9.1329e+07 + 9.91 116065 168.727 -2.94879e+06 9.09022e+07 + 9.93 115382 167.771 -2.93352e+06 9.04765e+07 + 9.95 114701 166.819 -2.91829e+06 9.0052e+07 + 9.97 114023 165.872 -2.90311e+06 8.96286e+07 + 9.99 113348 164.929 -2.88798e+06 8.92064e+07 + 10.01 112676 163.99 -2.8729e+06 8.87854e+07 + 10.03 112007 163.056 -2.85787e+06 8.83656e+07 + 10.05 111340 162.125 -2.84288e+06 8.7947e+07 + 10.07 110676 161.199 -2.82795e+06 8.75297e+07 + 10.09 110014 160.278 -2.81306e+06 8.71135e+07 + 10.11 109356 159.36 -2.79823e+06 8.66986e+07 + 10.13 108700 158.447 -2.78345e+06 8.6285e+07 + 10.15 108047 157.538 -2.76872e+06 8.58726e+07 + 10.17 107397 156.633 -2.75403e+06 8.54614e+07 + 10.19 106750 155.733 -2.73941e+06 8.50516e+07 + 10.21 106105 154.837 -2.72483e+06 8.4643e+07 + 10.23 105463 153.945 -2.7103e+06 8.42357e+07 + 10.25 104824 153.057 -2.69583e+06 8.38297e+07 + 10.27 104188 152.174 -2.68141e+06 8.3425e+07 + 10.29 103555 151.295 -2.66704e+06 8.30217e+07 + 10.31 102925 150.42 -2.65273e+06 8.26197e+07 + 10.33 102297 149.549 -2.63847e+06 8.2219e+07 + 10.35 101673 148.683 -2.62426e+06 8.18196e+07 + 10.37 101051 147.821 -2.61011e+06 8.14216e+07 + 10.39 100432 146.964 -2.59602e+06 8.1025e+07 + 10.41 99816.5 146.11 -2.58197e+06 8.06297e+07 + 10.43 99203.5 145.261 -2.56798e+06 8.02359e+07 + 10.45 98593.5 144.417 -2.55405e+06 7.98434e+07 + 10.47 97986.3 143.576 -2.54017e+06 7.94523e+07 + 10.49 97382 142.74 -2.52635e+06 7.90626e+07 + 10.51 96780.7 141.908 -2.51259e+06 7.86743e+07 + 10.53 96182.3 141.08 -2.49888e+06 7.82874e+07 + 10.55 95586.9 140.257 -2.48523e+06 7.79019e+07 + 10.57 94994.4 139.438 -2.47163e+06 7.75179e+07 + 10.59 94404.8 138.623 -2.45809e+06 7.71353e+07 + 10.61 93818.2 137.813 -2.44461e+06 7.67542e+07 + 10.63 93234.6 137.007 -2.43118e+06 7.63745e+07 + 10.65 92653.9 136.205 -2.41782e+06 7.59963e+07 + 10.67 92076.2 135.407 -2.40451e+06 7.56195e+07 + 10.69 91501.5 134.614 -2.39126e+06 7.52442e+07 + 10.71 90929.8 133.825 -2.37806e+06 7.48704e+07 + 10.73 90361 133.04 -2.36493e+06 7.44981e+07 + 10.75 89795.2 132.26 -2.35185e+06 7.41273e+07 + 10.77 89232.5 131.483 -2.33883e+06 7.37579e+07 + 10.79 88672.7 130.712 -2.32588e+06 7.33901e+07 + 10.81 88115.9 129.944 -2.31298e+06 7.30238e+07 + 10.83 87562.2 129.181 -2.30014e+06 7.2659e+07 + 10.85 87011.4 128.421 -2.28736e+06 7.22957e+07 + 10.87 86463.6 127.667 -2.27464e+06 7.1934e+07 + 10.89 85918.9 126.916 -2.26197e+06 7.15738e+07 + 10.91 85377.2 126.17 -2.24937e+06 7.12151e+07 + 10.93 84838.5 125.428 -2.23683e+06 7.0858e+07 + 10.95 84302.8 124.69 -2.22435e+06 7.05024e+07 + 10.97 83770.1 123.957 -2.21193e+06 7.01484e+07 + 10.99 83240.5 123.228 -2.19958e+06 6.9796e+07 + 11.01 82713.9 122.503 -2.18728e+06 6.94451e+07 + 11.03 82190.3 121.782 -2.17504e+06 6.90958e+07 + 11.05 81669.8 121.066 -2.16286e+06 6.87481e+07 + 11.07 81152.3 120.354 -2.15075e+06 6.84019e+07 + 11.09 80637.8 119.646 -2.1387e+06 6.80574e+07 + 11.11 80126.4 118.942 -2.1267e+06 6.77144e+07 + 11.13 79618 118.243 -2.11477e+06 6.7373e+07 + 11.15 79112.6 117.547 -2.10291e+06 6.70333e+07 + 11.17 78610.3 116.856 -2.0911e+06 6.66951e+07 + 11.19 78111 116.17 -2.07935e+06 6.63585e+07 + 11.21 77614.7 115.487 -2.06767e+06 6.60236e+07 + 11.23 77121.5 114.809 -2.05605e+06 6.56903e+07 + 11.25 76631.3 114.135 -2.04449e+06 6.53586e+07 + 11.27 76144.2 113.465 -2.033e+06 6.50285e+07 + 11.29 75660.1 112.8 -2.02156e+06 6.47e+07 + 11.31 75179 112.139 -2.01019e+06 6.43732e+07 + 11.33 74701 111.481 -1.99888e+06 6.4048e+07 + 11.35 74226 110.828 -1.98764e+06 6.37245e+07 + 11.37 73754 110.18 -1.97646e+06 6.34025e+07 + 11.39 73285.1 109.535 -1.96534e+06 6.30823e+07 + 11.41 72819.2 108.895 -1.95428e+06 6.27637e+07 + 11.43 72356.4 108.259 -1.94328e+06 6.24467e+07 + 11.45 71896.5 107.627 -1.93235e+06 6.21314e+07 + 11.47 71439.7 106.999 -1.92149e+06 6.18177e+07 + 11.49 70986 106.376 -1.91068e+06 6.15057e+07 + 11.51 70535.2 105.756 -1.89994e+06 6.11953e+07 + 11.53 70087.5 105.141 -1.88926e+06 6.08866e+07 + 11.55 69642.8 104.53 -1.87864e+06 6.05796e+07 + 11.57 69201.1 103.923 -1.86809e+06 6.02743e+07 + 11.59 68762.4 103.32 -1.8576e+06 5.99706e+07 + 11.61 68326.7 102.722 -1.84718e+06 5.96686e+07 + 11.63 67894.1 102.127 -1.83681e+06 5.93682e+07 + 11.65 67464.4 101.537 -1.82652e+06 5.90695e+07 + 11.67 67037.8 100.951 -1.81628e+06 5.87725e+07 + 11.69 66614.1 100.369 -1.80611e+06 5.84772e+07 + 11.71 66193.4 99.7912 -1.796e+06 5.81835e+07 + 11.73 65775.8 99.2175 -1.78595e+06 5.78916e+07 + 11.75 65361.1 98.6478 -1.77597e+06 5.76013e+07 + 11.77 64949.4 98.0823 -1.76605e+06 5.73127e+07 + 11.79 64540.7 97.5209 -1.7562e+06 5.70257e+07 + 11.81 64134.9 96.9636 -1.7464e+06 5.67405e+07 + 11.83 63732.1 96.4104 -1.73667e+06 5.64569e+07 + 11.85 63332.3 95.8613 -1.72701e+06 5.6175e+07 + 11.87 62935.5 95.3163 -1.71741e+06 5.58948e+07 + 11.89 62541.6 94.7753 -1.70787e+06 5.56163e+07 + 11.91 62150.6 94.2384 -1.69839e+06 5.53394e+07 + 11.93 61762.6 93.7056 -1.68898e+06 5.50643e+07 + 11.95 61377.5 93.1768 -1.67963e+06 5.47908e+07 + 11.97 60995.4 92.6521 -1.67034e+06 5.4519e+07 + 11.99 60616.2 92.1315 -1.66112e+06 5.42489e+07 + 12.01 60239.9 91.6149 -1.65196e+06 5.39805e+07 + 12.03 59866.6 91.1023 -1.64286e+06 5.37137e+07 + 12.05 59496.1 90.5938 -1.63382e+06 5.34486e+07 + 12.07 59128.6 90.0892 -1.62485e+06 5.31853e+07 + 12.09 58763.9 89.5887 -1.61594e+06 5.29236e+07 + 12.11 58402.2 89.0922 -1.60709e+06 5.26635e+07 + 12.13 58043.3 88.5997 -1.59831e+06 5.24052e+07 + 12.15 57687.3 88.1112 -1.58959e+06 5.21485e+07 + 12.17 57334.2 87.6267 -1.58093e+06 5.18935e+07 + 12.19 56983.9 87.1461 -1.57233e+06 5.16402e+07 + 12.21 56636.5 86.6696 -1.5638e+06 5.13885e+07 + 12.23 56292 86.197 -1.55532e+06 5.11385e+07 + 12.25 55950.2 85.7283 -1.54691e+06 5.08902e+07 + 12.27 55611.4 85.2636 -1.53856e+06 5.06436e+07 + 12.29 55275.3 84.8029 -1.53027e+06 5.03986e+07 + 12.31 54942.1 84.346 -1.52205e+06 5.01553e+07 + 12.33 54611.7 83.8931 -1.51388e+06 4.99136e+07 + 12.35 54284.1 83.4442 -1.50578e+06 4.96736e+07 + 12.37 53959.3 82.9991 -1.49774e+06 4.94353e+07 + 12.39 53637.3 82.558 -1.48976e+06 4.91986e+07 + 12.41 53318 82.1207 -1.48184e+06 4.89636e+07 + 12.43 53001.6 81.6873 -1.47398e+06 4.87302e+07 + 12.45 52687.9 81.2578 -1.46619e+06 4.84985e+07 + 12.47 52376.9 80.8322 -1.45845e+06 4.82684e+07 + 12.49 52068.7 80.4104 -1.45077e+06 4.80399e+07 + 12.51 51763.2 79.9925 -1.44316e+06 4.78131e+07 + 12.53 51460.5 79.5784 -1.4356e+06 4.7588e+07 + 12.55 51160.5 79.1682 -1.42811e+06 4.73644e+07 + 12.57 50863.2 78.7617 -1.42067e+06 4.71425e+07 + 12.59 50568.6 78.3591 -1.4133e+06 4.69222e+07 + 12.61 50276.6 77.9603 -1.40598e+06 4.67036e+07 + 12.63 49987.4 77.5653 -1.39873e+06 4.64865e+07 + 12.65 49700.8 77.1741 -1.39153e+06 4.62711e+07 + 12.67 49416.9 76.7867 -1.38439e+06 4.60573e+07 + 12.69 49135.7 76.403 -1.37731e+06 4.58451e+07 + 12.71 48857.1 76.0231 -1.3703e+06 4.56345e+07 + 12.73 48581.1 75.6469 -1.36333e+06 4.54255e+07 + 12.75 48307.8 75.2745 -1.35643e+06 4.52181e+07 + 12.77 48037 74.9057 -1.34959e+06 4.50123e+07 + 12.79 47768.9 74.5407 -1.3428e+06 4.48081e+07 + 12.81 47503.4 74.1795 -1.33607e+06 4.46055e+07 + 12.83 47240.4 73.8219 -1.3294e+06 4.44045e+07 + 12.85 46980 73.4679 -1.32279e+06 4.4205e+07 + 12.87 46722.2 73.1177 -1.31624e+06 4.40071e+07 + 12.89 46466.9 72.7711 -1.30974e+06 4.38108e+07 + 12.91 46214.2 72.4282 -1.3033e+06 4.3616e+07 + 12.93 45964 72.0889 -1.29691e+06 4.34228e+07 + 12.95 45716.3 71.7532 -1.29059e+06 4.32311e+07 + 12.97 45471.1 71.4212 -1.28432e+06 4.3041e+07 + 12.99 45228.4 71.0927 -1.2781e+06 4.28525e+07 + 13.01 44988.2 70.7679 -1.27194e+06 4.26654e+07 + 13.03 44750.5 70.4466 -1.26584e+06 4.24799e+07 + 13.05 44515.2 70.1289 -1.25979e+06 4.2296e+07 + 13.07 44282.4 69.8147 -1.2538e+06 4.21135e+07 + 13.09 44052 69.5041 -1.24786e+06 4.19326e+07 + 13.11 43824 69.197 -1.24198e+06 4.17532e+07 + 13.13 43598.5 68.8934 -1.23615e+06 4.15753e+07 + 13.15 43375.3 68.5933 -1.23038e+06 4.13989e+07 + 13.17 43154.6 68.2967 -1.22466e+06 4.1224e+07 + 13.19 42936.2 68.0036 -1.219e+06 4.10506e+07 + 13.21 42720.2 67.714 -1.21339e+06 4.08787e+07 + 13.23 42506.6 67.4277 -1.20783e+06 4.07082e+07 + 13.25 42295.3 67.145 -1.20233e+06 4.05393e+07 + 13.27 42086.4 66.8656 -1.19688e+06 4.03718e+07 + 13.29 41879.7 66.5896 -1.19148e+06 4.02057e+07 + 13.31 41675.4 66.3171 -1.18613e+06 4.00411e+07 + 13.33 41473.3 66.0479 -1.18084e+06 3.9878e+07 + 13.35 41273.6 65.7821 -1.1756e+06 3.97163e+07 + 13.37 41076.1 65.5196 -1.17041e+06 3.9556e+07 + 13.39 40880.9 65.2604 -1.16527e+06 3.93972e+07 + 13.41 40687.9 65.0046 -1.16018e+06 3.92398e+07 + 13.43 40497.2 64.752 -1.15515e+06 3.90838e+07 + 13.45 40308.7 64.5028 -1.15017e+06 3.89293e+07 + 13.47 40122.4 64.2568 -1.14523e+06 3.87761e+07 + 13.49 39938.2 64.014 -1.14035e+06 3.86243e+07 + 13.51 39756.3 63.7745 -1.13551e+06 3.8474e+07 + 13.53 39576.6 63.5383 -1.13073e+06 3.8325e+07 + 13.55 39399 63.3052 -1.126e+06 3.81774e+07 + 13.57 39223.5 63.0753 -1.12131e+06 3.80311e+07 + 13.59 39050.2 62.8485 -1.11668e+06 3.78863e+07 + 13.61 38879 62.625 -1.11209e+06 3.77428e+07 + 13.63 38709.9 62.4045 -1.10755e+06 3.76006e+07 + 13.65 38542.9 62.1872 -1.10306e+06 3.74598e+07 + 13.67 38378 61.9729 -1.09862e+06 3.73203e+07 + 13.69 38215.1 61.7618 -1.09423e+06 3.71822e+07 + 13.71 38054.3 61.5537 -1.08988e+06 3.70454e+07 + 13.73 37895.6 61.3486 -1.08558e+06 3.69099e+07 + 13.75 37738.9 61.1466 -1.08133e+06 3.67757e+07 + 13.77 37584.2 60.9476 -1.07712e+06 3.66428e+07 + 13.79 37431.5 60.7516 -1.07296e+06 3.65112e+07 + 13.81 37280.7 60.5585 -1.06885e+06 3.63809e+07 + 13.83 37132 60.3684 -1.06478e+06 3.62519e+07 + 13.85 36985.2 60.1812 -1.06076e+06 3.61242e+07 + 13.87 36840.4 59.997 -1.05678e+06 3.59977e+07 + 13.89 36697.5 59.8156 -1.05285e+06 3.58725e+07 + 13.91 36556.6 59.6371 -1.04896e+06 3.57485e+07 + 13.93 36417.5 59.4615 -1.04512e+06 3.56258e+07 + 13.95 36280.4 59.2887 -1.04132e+06 3.55043e+07 + 13.97 36145.1 59.1187 -1.03756e+06 3.53841e+07 + 13.99 36011.7 58.9515 -1.03385e+06 3.52651e+07 + 14.01 35880.1 58.787 -1.03018e+06 3.51473e+07 + 14.03 35750.4 58.6253 -1.02655e+06 3.50307e+07 + 14.05 35622.6 58.4664 -1.02297e+06 3.49153e+07 + 14.07 35496.5 58.3101 -1.01943e+06 3.4801e+07 + 14.09 35372.3 58.1566 -1.01593e+06 3.4688e+07 + 14.11 35249.8 58.0057 -1.01247e+06 3.45762e+07 + 14.13 35129.1 57.8574 -1.00906e+06 3.44655e+07 + 14.15 35010.2 57.7118 -1.00569e+06 3.4356e+07 + 14.17 34893.1 57.5688 -1.00235e+06 3.42476e+07 + 14.19 34777.6 57.4284 -999060 3.41404e+07 + 14.21 34663.9 57.2905 -995808 3.40343e+07 + 14.23 34551.9 57.1551 -992596 3.39294e+07 + 14.25 34441.6 57.0223 -989423 3.38255e+07 + 14.27 34333 56.892 -986290 3.37228e+07 + 14.29 34226.1 56.7641 -983197 3.36212e+07 + 14.31 34120.8 56.6387 -980142 3.35207e+07 + 14.33 34017.2 56.5157 -977126 3.34213e+07 + 14.35 33915.2 56.3951 -974148 3.3323e+07 + 14.37 33814.8 56.2769 -971208 3.32257e+07 + 14.39 33716 56.161 -968306 3.31296e+07 + 14.41 33618.8 56.0475 -965442 3.30344e+07 + 14.43 33523.1 55.9362 -962614 3.29404e+07 + 14.45 33429.1 55.8273 -959824 3.28473e+07 + 14.47 33336.6 55.7206 -957070 3.27554e+07 + 14.49 33245.6 55.6162 -954352 3.26644e+07 + 14.51 33156.2 55.514 -951670 3.25744e+07 + 14.53 33068.2 55.414 -949024 3.24855e+07 + 14.55 32981.8 55.3161 -946413 3.23976e+07 + 14.57 32896.8 55.2204 -943838 3.23107e+07 + 14.59 32813.4 55.1268 -941297 3.22247e+07 + 14.61 32731.3 55.0354 -938791 3.21398e+07 + 14.63 32650.7 54.946 -936319 3.20558e+07 + 14.65 32571.6 54.8586 -933881 3.19728e+07 + 14.67 32493.9 54.7733 -931476 3.18907e+07 + 14.69 32417.6 54.6899 -929105 3.18096e+07 + 14.71 32342.6 54.6086 -926767 3.17294e+07 + 14.73 32269.1 54.5292 -924462 3.16502e+07 + 14.75 32196.9 54.4517 -922189 3.15719e+07 + 14.77 32126.1 54.3762 -919949 3.14945e+07 + 14.79 32056.6 54.3025 -917740 3.1418e+07 + 14.81 31988.4 54.2307 -915563 3.13424e+07 + 14.83 31921.6 54.1607 -913418 3.12677e+07 + 14.85 31856 54.0926 -911303 3.11939e+07 + 14.87 31791.8 54.0262 -909219 3.1121e+07 + 14.89 31728.8 53.9616 -907166 3.10489e+07 + 14.91 31667.1 53.8987 -905143 3.09777e+07 + 14.93 31606.7 53.8376 -903150 3.09074e+07 + 14.95 31547.5 53.7781 -901186 3.08379e+07 + 14.97 31489.5 53.7203 -899252 3.07692e+07 + 14.99 31432.7 53.6642 -897347 3.07014e+07 + 15.01 31377.1 53.6096 -895471 3.06344e+07 + 15.03 31322.7 53.5567 -893623 3.05682e+07 + 15.05 31269.5 53.5053 -891804 3.05028e+07 + 15.07 31217.5 53.4555 -890012 3.04383e+07 + 15.09 31166.6 53.4072 -888248 3.03745e+07 + 15.11 31116.9 53.3605 -886512 3.03115e+07 + 15.13 31068.2 53.3151 -884803 3.02493e+07 + 15.15 31020.7 53.2713 -883120 3.01878e+07 + 15.17 30974.3 53.2289 -881464 3.01271e+07 + 15.19 30929 53.1879 -879834 3.00672e+07 + 15.21 30884.8 53.1482 -878231 3.0008e+07 + 15.23 30841.7 53.11 -876653 2.99495e+07 + 15.25 30799.5 53.073 -875101 2.98918e+07 + 15.27 30758.5 53.0374 -873574 2.98348e+07 + 15.29 30718.5 53.0031 -872072 2.97785e+07 + 15.31 30679.4 52.97 -870594 2.97229e+07 + 15.33 30641.4 52.9382 -869141 2.96681e+07 + 15.35 30604.4 52.9076 -867712 2.96139e+07 + 15.37 30568.4 52.8783 -866307 2.95604e+07 + 15.39 30533.4 52.8501 -864926 2.95076e+07 + 15.41 30499.3 52.823 -863568 2.94554e+07 + 15.43 30466.2 52.7971 -862233 2.94039e+07 + 15.45 30434 52.7723 -860921 2.93531e+07 + 15.47 30402.7 52.7486 -859632 2.9303e+07 + 15.49 30372.3 52.726 -858364 2.92534e+07 + 15.51 30342.9 52.7044 -857119 2.92045e+07 + 15.53 30314.4 52.6838 -855896 2.91563e+07 + 15.55 30286.7 52.6642 -854695 2.91086e+07 + 15.57 30259.9 52.6456 -853514 2.90616e+07 + 15.59 30233.9 52.628 -852355 2.90152e+07 + 15.61 30208.9 52.6113 -851217 2.89694e+07 + 15.63 30184.6 52.5956 -850099 2.89241e+07 + 15.65 30161.2 52.5807 -849001 2.88795e+07 + 15.67 30138.6 52.5667 -847924 2.88354e+07 + 15.69 30116.8 52.5536 -846866 2.87919e+07 + 15.71 30095.8 52.5413 -845828 2.8749e+07 + 15.73 30075.6 52.5298 -844810 2.87066e+07 + 15.75 30056.1 52.5191 -843810 2.86648e+07 + 15.77 30037.4 52.5092 -842830 2.86235e+07 + 15.79 30019.5 52.5 -841868 2.85828e+07 + 15.81 30002.3 52.4916 -840924 2.85426e+07 + 15.83 29985.9 52.4839 -839999 2.85029e+07 + 15.85 29970.2 52.4769 -839091 2.84638e+07 + 15.87 29955.1 52.4706 -838201 2.84251e+07 + 15.89 29940.8 52.4649 -837329 2.8387e+07 + 15.91 29927.2 52.4599 -836474 2.83493e+07 + 15.93 29914.3 52.4555 -835636 2.83122e+07 + 15.95 29902 52.4517 -834815 2.82755e+07 + 15.97 29890.4 52.4485 -834010 2.82394e+07 + 15.99 29879.4 52.4458 -833222 2.82036e+07 + 16.01 29869.1 52.4437 -832449 2.81684e+07 + 16.03 29859.5 52.4421 -831693 2.81336e+07 + 16.05 29850.4 52.441 -830953 2.80993e+07 + 16.07 29842 52.4405 -830228 2.80654e+07 + 16.09 29834.1 52.4404 -829518 2.8032e+07 + 16.11 29826.9 52.4407 -828823 2.7999e+07 + 16.13 29820.2 52.4415 -828143 2.79664e+07 + 16.15 29814.2 52.4428 -827478 2.79343e+07 + 16.17 29808.7 52.4444 -826827 2.79025e+07 + 16.19 29803.7 52.4465 -826190 2.78712e+07 + 16.21 29799.3 52.4489 -825567 2.78403e+07 + 16.23 29795.4 52.4516 -824958 2.78098e+07 + 16.25 29792.1 52.4547 -824363 2.77797e+07 + 16.27 29789.3 52.4582 -823781 2.775e+07 + 16.29 29787 52.4619 -823213 2.77207e+07 + 16.31 29785.2 52.466 -822657 2.76917e+07 + 16.33 29783.9 52.4703 -822114 2.76631e+07 + 16.35 29783.1 52.4749 -821584 2.76349e+07 + 16.37 29782.8 52.4798 -821066 2.76071e+07 + 16.39 29783 52.4848 -820561 2.75796e+07 + 16.41 29783.6 52.4901 -820067 2.75524e+07 + 16.43 29784.6 52.4957 -819585 2.75256e+07 + 16.45 29786.1 52.5014 -819115 2.74992e+07 + 16.47 29788.1 52.5072 -818657 2.7473e+07 + 16.49 29790.5 52.5133 -818210 2.74473e+07 + 16.51 29793.3 52.5195 -817774 2.74218e+07 + 16.53 29796.5 52.5258 -817349 2.73966e+07 + 16.55 29800.1 52.5323 -816934 2.73718e+07 + 16.57 29804.1 52.5388 -816531 2.73473e+07 + 16.59 29808.5 52.5455 -816137 2.73231e+07 + 16.61 29813.3 52.5523 -815754 2.72992e+07 + 16.63 29818.5 52.5591 -815381 2.72755e+07 + 16.65 29824 52.566 -815018 2.72522e+07 + 16.67 29829.9 52.5729 -814665 2.72292e+07 + 16.69 29836.1 52.5798 -814321 2.72064e+07 + 16.71 29842.7 52.5868 -813987 2.71839e+07 + 16.73 29849.6 52.5938 -813662 2.71617e+07 + 16.75 29856.9 52.6008 -813346 2.71397e+07 + 16.77 29864.4 52.6078 -813039 2.71181e+07 + 16.79 29872.3 52.6147 -812740 2.70966e+07 + 16.81 29880.5 52.6216 -812450 2.70755e+07 + 16.83 29889 52.6285 -812169 2.70545e+07 + 16.85 29897.8 52.6353 -811896 2.70339e+07 + 16.87 29906.9 52.642 -811631 2.70134e+07 + 16.89 29916.3 52.6487 -811375 2.69932e+07 + 16.91 29925.9 52.6553 -811126 2.69733e+07 + 16.93 29935.8 52.6617 -810885 2.69535e+07 + 16.95 29946 52.6681 -810651 2.6934e+07 + 16.97 29956.4 52.6743 -810425 2.69147e+07 + 16.99 29967.1 52.6804 -810206 2.68957e+07 + 17.01 29978 52.6864 -809994 2.68768e+07 + 17.03 29989.2 52.6922 -809789 2.68582e+07 + 17.05 30000.6 52.6979 -809591 2.68397e+07 + 17.07 30012.2 52.7034 -809400 2.68215e+07 + 17.09 30024 52.7087 -809215 2.68034e+07 + 17.11 30036 52.7138 -809037 2.67856e+07 + 17.13 30048.3 52.7188 -808865 2.67679e+07 + 17.15 30060.7 52.7235 -808699 2.67504e+07 + 17.17 30073.4 52.728 -808540 2.67331e+07 + 17.19 30086.2 52.7323 -808386 2.6716e+07 + 17.21 30099.2 52.7364 -808238 2.66991e+07 + 17.23 30112.4 52.7403 -808096 2.66823e+07 + 17.25 30125.7 52.7439 -807960 2.66657e+07 + 17.27 30139.2 52.7472 -807829 2.66493e+07 + 17.29 30152.9 52.7503 -807703 2.6633e+07 + 17.31 30166.8 52.7531 -807582 2.66169e+07 + 17.33 30180.7 52.7557 -807467 2.6601e+07 + 17.35 30194.9 52.758 -807357 2.65852e+07 + 17.37 30209.1 52.76 -807251 2.65695e+07 + 17.39 30223.5 52.7617 -807150 2.6554e+07 + 17.41 30238 52.7631 -807054 2.65387e+07 + 17.43 30252.7 52.7642 -806963 2.65235e+07 + 17.45 30267.5 52.765 -806876 2.65084e+07 + 17.47 30282.3 52.7654 -806793 2.64935e+07 + 17.49 30297.3 52.7656 -806714 2.64787e+07 + 17.51 30312.4 52.7654 -806640 2.6464e+07 + 17.53 30327.6 52.7649 -806569 2.64494e+07 + 17.55 30342.9 52.764 -806503 2.6435e+07 + 17.57 30358.3 52.7628 -806440 2.64207e+07 + 17.59 30373.7 52.7613 -806382 2.64066e+07 + 17.61 30389.3 52.7594 -806326 2.63925e+07 + 17.63 30404.9 52.7571 -806275 2.63786e+07 + 17.65 30420.6 52.7545 -806226 2.63647e+07 + 17.67 30436.3 52.7515 -806181 2.6351e+07 + 17.69 30452.1 52.7481 -806140 2.63374e+07 + 17.71 30468 52.7444 -806101 2.63239e+07 + 17.73 30483.9 52.7403 -806066 2.63105e+07 + 17.75 30499.9 52.7357 -806034 2.62972e+07 + 17.77 30515.9 52.7308 -806004 2.6284e+07 + 17.79 30532 52.7255 -805978 2.62709e+07 + 17.81 30548.1 52.7198 -805954 2.62578e+07 + 17.83 30564.2 52.7137 -805933 2.62449e+07 + 17.85 30580.4 52.7072 -805914 2.62321e+07 + 17.87 30596.6 52.7003 -805898 2.62194e+07 + 17.89 30612.8 52.693 -805884 2.62067e+07 + 17.91 30629 52.6853 -805873 2.61941e+07 + 17.93 30645.3 52.6771 -805864 2.61816e+07 + 17.95 30661.6 52.6685 -805857 2.61692e+07 + 17.97 30677.8 52.6595 -805852 2.61569e+07 + 17.99 30694.1 52.6501 -805850 2.61447e+07 + 18.01 30710.4 52.6403 -805849 2.61325e+07 + 18.03 30726.6 52.63 -805851 2.61204e+07 + 18.05 30742.9 52.6193 -805854 2.61084e+07 + 18.07 30759.1 52.6082 -805859 2.60964e+07 + 18.09 30775.4 52.5966 -805865 2.60845e+07 + 18.11 30791.6 52.5846 -805874 2.60727e+07 + 18.13 30807.8 52.5721 -805883 2.60609e+07 + 18.15 30824 52.5592 -805895 2.60492e+07 + 18.17 30840.1 52.5459 -805908 2.60376e+07 + 18.19 30856.3 52.5321 -805922 2.60261e+07 + 18.21 30872.3 52.5179 -805938 2.60145e+07 + 18.23 30888.4 52.5033 -805954 2.60031e+07 + 18.25 30904.4 52.4882 -805973 2.59917e+07 + 18.27 30920.4 52.4726 -805992 2.59804e+07 + 18.29 30936.3 52.4566 -806012 2.59691e+07 + 18.31 30952.2 52.4402 -806034 2.59579e+07 + 18.33 30968 52.4233 -806056 2.59467e+07 + 18.35 30983.8 52.4059 -806079 2.59356e+07 + 18.37 30999.6 52.3882 -806104 2.59246e+07 + 18.39 31015.2 52.3699 -806129 2.59135e+07 + 18.41 31030.8 52.3512 -806155 2.59026e+07 + 18.43 31046.4 52.3321 -806181 2.58917e+07 + 18.45 31061.9 52.3125 -806209 2.58808e+07 + 18.47 31077.3 52.2925 -806237 2.587e+07 + 18.49 31092.6 52.272 -806265 2.58592e+07 + 18.51 31107.9 52.2511 -806294 2.58485e+07 + 18.53 31123.1 52.2297 -806324 2.58378e+07 + 18.55 31138.2 52.2079 -806354 2.58271e+07 + 18.57 31153.2 52.1856 -806384 2.58165e+07 + 18.59 31168.2 52.1629 -806415 2.58059e+07 + 18.61 31183 52.1398 -806446 2.57954e+07 + 18.63 31197.8 52.1162 -806477 2.57849e+07 + 18.65 31212.5 52.0922 -806509 2.57745e+07 + 18.67 31227.1 52.0677 -806541 2.57641e+07 + 18.69 31241.6 52.0428 -806573 2.57537e+07 + 18.71 31256 52.0175 -806605 2.57433e+07 + 18.73 31270.3 51.9917 -806637 2.5733e+07 + 18.75 31284.5 51.9655 -806669 2.57228e+07 + 18.77 31298.6 51.9388 -806702 2.57125e+07 + 18.79 31312.6 51.9118 -806734 2.57024e+07 + 18.81 31326.4 51.8843 -806766 2.56922e+07 + 18.83 31340.2 51.8564 -806798 2.56821e+07 + 18.85 31353.9 51.828 -806830 2.5672e+07 + 18.87 31367.4 51.7992 -806862 2.56619e+07 + 18.89 31380.9 51.7701 -806893 2.56519e+07 + 18.91 31394.2 51.7405 -806924 2.56419e+07 + 18.93 31407.4 51.7104 -806955 2.56319e+07 + 18.95 31420.5 51.68 -806986 2.56219e+07 + 18.97 31433.4 51.6492 -807016 2.5612e+07 + 18.99 31446.3 51.6179 -807046 2.56021e+07 + 19.01 31459 51.5863 -807076 2.55923e+07 + 19.03 31471.6 51.5542 -807105 2.55825e+07 + 19.05 31484 51.5218 -807133 2.55727e+07 + 19.07 31496.4 51.4889 -807161 2.55629e+07 + 19.09 31508.5 51.4557 -807189 2.55532e+07 + 19.11 31520.6 51.4221 -807216 2.55435e+07 + 19.13 31532.5 51.3881 -807242 2.55338e+07 + 19.15 31544.3 51.3537 -807268 2.55241e+07 + 19.17 31555.9 51.3189 -807294 2.55145e+07 + 19.19 31567.4 51.2837 -807318 2.55049e+07 + 19.21 31578.8 51.2482 -807342 2.54953e+07 + 19.23 31590 51.2123 -807365 2.54858e+07 + 19.25 31601.1 51.1761 -807388 2.54763e+07 + 19.27 31612 51.1395 -807409 2.54668e+07 + 19.29 31622.8 51.1025 -807430 2.54573e+07 + 19.31 31633.4 51.0651 -807450 2.54479e+07 + 19.33 31643.9 51.0275 -807469 2.54385e+07 + 19.35 31654.2 50.9894 -807488 2.54291e+07 + 19.37 31664.4 50.951 -807505 2.54197e+07 + 19.39 31674.4 50.9123 -807521 2.54104e+07 + 19.41 31684.3 50.8733 -807537 2.54011e+07 + 19.43 31694 50.8339 -807552 2.53918e+07 + 19.45 31703.5 50.7942 -807565 2.53825e+07 + 19.47 31712.9 50.7541 -807578 2.53733e+07 + 19.49 31722.1 50.7138 -807589 2.53641e+07 + 19.51 31731.2 50.6731 -807600 2.53549e+07 + 19.53 31740 50.6321 -807609 2.53457e+07 + 19.55 31748.8 50.5908 -807618 2.53366e+07 + 19.57 31757.3 50.5493 -807625 2.53275e+07 + 19.59 31765.7 50.5074 -807631 2.53184e+07 + 19.61 31773.9 50.4652 -807636 2.53093e+07 + 19.63 31782 50.4227 -807639 2.53003e+07 + 19.65 31789.8 50.3799 -807642 2.52913e+07 + 19.67 31797.5 50.3369 -807643 2.52823e+07 + 19.69 31805.1 50.2936 -807643 2.52734e+07 + 19.71 31812.4 50.25 -807642 2.52644e+07 + 19.73 31819.6 50.2061 -807639 2.52555e+07 + 19.75 31826.6 50.162 -807635 2.52466e+07 + 19.77 31833.4 50.1176 -807630 2.52378e+07 + 19.79 31840 50.073 -807624 2.52289e+07 + 19.81 31846.5 50.0281 -807616 2.52201e+07 + 19.83 31852.7 49.983 -807606 2.52114e+07 + 19.85 31858.8 49.9376 -807595 2.52026e+07 + 19.87 31864.7 49.892 -807583 2.51939e+07 + 19.89 31870.4 49.8462 -807570 2.51852e+07 + 19.91 31876 49.8001 -807554 2.51765e+07 + 19.93 31881.3 49.7538 -807538 2.51678e+07 + 19.95 31886.4 49.7073 -807519 2.51592e+07 + 19.97 31891.4 49.6606 -807500 2.51506e+07 + 19.99 31896.2 49.6137 -807478 2.5142e+07 diff --git a/tests/06_SDFT/CMakeLists.txt b/tests/06_SDFT/CMakeLists.txt index 3eef92c6b24..9e8953cb19c 100644 --- a/tests/06_SDFT/CMakeLists.txt +++ b/tests/06_SDFT/CMakeLists.txt @@ -10,7 +10,7 @@ if(ENABLE_ASAN) else() add_test( NAME 06_SDFT - COMMAND ${BASH} ../integrate/Autotest.sh -a ${ABACUS_BIN_PATH} -n 4 + COMMAND ${BASH} ../integrate/Autotest.sh -a ${ABACUS_BIN_PATH} -n 4 -j auto WORKING_DIRECTORY ${ABACUS_TEST_DIR}/06_SDFT ) endif() diff --git a/tests/07_OFDFT/24_OF_out_elf/refelftot.cube b/tests/07_OFDFT/24_OF_out_elf/elftot.cube.ref similarity index 100% rename from tests/07_OFDFT/24_OF_out_elf/refelftot.cube rename to tests/07_OFDFT/24_OF_out_elf/elftot.cube.ref diff --git a/tests/07_OFDFT/CMakeLists.txt b/tests/07_OFDFT/CMakeLists.txt index aadd6db00ba..663cb1e9c02 100644 --- a/tests/07_OFDFT/CMakeLists.txt +++ b/tests/07_OFDFT/CMakeLists.txt @@ -10,7 +10,7 @@ if(ENABLE_ASAN) else() add_test( NAME 07_OFDFT - COMMAND ${BASH} ../integrate/Autotest.sh -a ${ABACUS_BIN_PATH} -n 4 + COMMAND ${BASH} ../integrate/Autotest.sh -a ${ABACUS_BIN_PATH} -n 4 -j auto WORKING_DIRECTORY ${ABACUS_TEST_DIR}/07_OFDFT ) endif() diff --git a/tests/08_EXX/01_GO_S1_HSE/README b/tests/08_EXX/01_GO_S1_HSE/README deleted file mode 100644 index 8b82617266f..00000000000 --- a/tests/08_EXX/01_GO_S1_HSE/README +++ /dev/null @@ -1 +0,0 @@ -HSE calculation on H2O, gamma only, nspin=1, exx_seperate_loop=true \ No newline at end of file diff --git a/tests/08_EXX/01_GO_S1_HSE/result.ref b/tests/08_EXX/01_GO_S1_HSE/result.ref deleted file mode 100644 index cb0ae86be9d..00000000000 --- a/tests/08_EXX/01_GO_S1_HSE/result.ref +++ /dev/null @@ -1,5 +0,0 @@ -etotref -428.9942827105964511 -etotperatomref -142.9980942369 -totalforceref 30.937113 -totalstressref 256.779921 -totaltimeref 2.76 diff --git a/tests/08_EXX/02_GO_S2_HSE/README b/tests/08_EXX/02_GO_S2_HSE/README deleted file mode 100644 index e5669149d4d..00000000000 --- a/tests/08_EXX/02_GO_S2_HSE/README +++ /dev/null @@ -1 +0,0 @@ -HSE calculation on H2O, gamma only, nspin=2, exx_seperate_loop=true \ No newline at end of file diff --git a/tests/08_EXX/02_GO_S2_HSE/result.ref b/tests/08_EXX/02_GO_S2_HSE/result.ref deleted file mode 100644 index 7191dc0b2ba..00000000000 --- a/tests/08_EXX/02_GO_S2_HSE/result.ref +++ /dev/null @@ -1,5 +0,0 @@ -etotref -448.7146661094475348 -etotperatomref -149.5715553698 -totalforceref 33.613002 -totalstressref 825.537122 -totaltimeref 2.91 diff --git a/tests/08_EXX/03_KP_S4_HSE/README b/tests/08_EXX/03_KP_S4_HSE/README deleted file mode 100644 index 21b7dbc535e..00000000000 --- a/tests/08_EXX/03_KP_S4_HSE/README +++ /dev/null @@ -1 +0,0 @@ -HSE for H2O, multiple k-points, soc (nspin 4), exx_seperate_loop=true diff --git a/tests/08_EXX/03_KP_S4_HSE/result.ref b/tests/08_EXX/03_KP_S4_HSE/result.ref deleted file mode 100644 index e847652d45d..00000000000 --- a/tests/08_EXX/03_KP_S4_HSE/result.ref +++ /dev/null @@ -1,5 +0,0 @@ -etotref -448.3614595745778502 -etotperatomref -149.4538198582 -totalforceref 33.668068 -totalstressref 813.488522 -totaltimeref 3.18 diff --git a/tests/08_EXX/04_GO_S1_HSE_loop0/README b/tests/08_EXX/04_GO_S1_HSE_loop0/README deleted file mode 100644 index b8ee7f6a59e..00000000000 --- a/tests/08_EXX/04_GO_S1_HSE_loop0/README +++ /dev/null @@ -1 +0,0 @@ -HSE calculation on H2O, gamma only, nspin=1, exx_seperate_loop=false \ No newline at end of file diff --git a/tests/08_EXX/04_GO_S1_HSE_loop0/result.ref b/tests/08_EXX/04_GO_S1_HSE_loop0/result.ref deleted file mode 100644 index a084c91d716..00000000000 --- a/tests/08_EXX/04_GO_S1_HSE_loop0/result.ref +++ /dev/null @@ -1,3 +0,0 @@ -etotref -428.9946799664623 -etotperatomref -142.9982266555 -totaltimeref 2.02 diff --git a/tests/08_EXX/05_GO_RE_HSE/KPT b/tests/08_EXX/05_GO_RE_HSE/KPT deleted file mode 100644 index c289c0158aa..00000000000 --- a/tests/08_EXX/05_GO_RE_HSE/KPT +++ /dev/null @@ -1,4 +0,0 @@ -K_POINTS -0 -Gamma -1 1 1 0 0 0 diff --git a/tests/08_EXX/05_GO_RE_HSE/README b/tests/08_EXX/05_GO_RE_HSE/README deleted file mode 100644 index 8a8e4a13468..00000000000 --- a/tests/08_EXX/05_GO_RE_HSE/README +++ /dev/null @@ -1 +0,0 @@ -Relaxation using HSE for H2O, gamma only diff --git a/tests/08_EXX/05_GO_RE_HSE/result.ref b/tests/08_EXX/05_GO_RE_HSE/result.ref deleted file mode 100644 index baeeb2a7370..00000000000 --- a/tests/08_EXX/05_GO_RE_HSE/result.ref +++ /dev/null @@ -1,3 +0,0 @@ -etotref -461.2147366408772 -etotperatomref -153.7382455470 -totaltimeref 3.20 diff --git a/tests/08_EXX/06_KP_MD_HSE/README b/tests/08_EXX/06_KP_MD_HSE/README deleted file mode 100644 index 959455180f6..00000000000 --- a/tests/08_EXX/06_KP_MD_HSE/README +++ /dev/null @@ -1 +0,0 @@ -Molecular dynamics using HSE, multi-k points diff --git a/tests/08_EXX/06_KP_MD_HSE/result.ref b/tests/08_EXX/06_KP_MD_HSE/result.ref deleted file mode 100644 index 97b09a2cb47..00000000000 --- a/tests/08_EXX/06_KP_MD_HSE/result.ref +++ /dev/null @@ -1,3 +0,0 @@ -etotref -467.2676788578187 -etotperatomref -155.7558929526 -totaltimeref 3.40 diff --git a/tests/08_EXX/07_KP_CR_HSE/README b/tests/08_EXX/07_KP_CR_HSE/README deleted file mode 100644 index 6a4fb197652..00000000000 --- a/tests/08_EXX/07_KP_CR_HSE/README +++ /dev/null @@ -1 +0,0 @@ -Cell relaxation using HSE with multiple k-points, exx real number set to 1 diff --git a/tests/08_EXX/08_KP_HSE_symm/KPT b/tests/08_EXX/08_KP_HSE_symm/KPT deleted file mode 100644 index f5f7f4ec34c..00000000000 --- a/tests/08_EXX/08_KP_HSE_symm/KPT +++ /dev/null @@ -1,4 +0,0 @@ -K_POINTS -0 -Gamma -2 2 2 0 0 0 diff --git a/tests/08_EXX/08_KP_HSE_symm/README b/tests/08_EXX/08_KP_HSE_symm/README deleted file mode 100644 index 4dc65661511..00000000000 --- a/tests/08_EXX/08_KP_HSE_symm/README +++ /dev/null @@ -1,36 +0,0 @@ -HSE calculation on Si, multiple k-points, force and stress, real EXX, symmetry=1. - -The density at each irreducible k point must be averaged over its unitary -little group before expanding its star. A star stores one representative per -unique k point; it does not contain all operations fixing that point. On this -coarse finite grid, omitting that average leaves the EXX density outside the -space-group-invariant subspace, so rotating irreducible H(R) blocks can produce -a non-Hermitian Hamiltonian. - -In ABACUS's transposed density storage, the average is - P_k(D) = sum_{g in G_k} M_g^T D M_g^* / |G_k|, -where g k = k modulo a reciprocal lattice vector. P_k is a group projector: -it preserves Hermiticity and removes dependence on the chosen unitary star -representative. The existing star-size factor is applied only after averaging. -The unit tests in module_exx_symmetry/test/test_symm_rotation.cpp exercise this -contract with complex matrices and separate spin channels. - -The reference was regenerated after merging upstream develop 54aaffefb, -including its ELPA fallback correction, on 2026-09-14. It uses the unchanged -case with the default genelpa solver on one MPI rank. Its total energy is --189.4406826032011395 eV. ScaLAPACK runs with 1 and 4 MPI ranks and a full EXX -contraction run (exx_symmetry_realspace=0, with the same projected density) -agree within 1.4e-9 eV. All four report totalstressref=2143.792554 and -totalforceref=0.000000. INPUT, thresholds and the historical timing reference -are unchanged. This is a fixed-input regression reference, not a claim of -cutoff or outer-loop convergence. - -Verification used GCC, OpenMPI 5.0.10, ELPA 2026.02.001, LibXC 7.0.0 and the -repository-pinned LibRI on Sai DSPRHBM, with OMP_NUM_THREADS=1. The 12 EXX -symmetry tests and, on one rank, both complex ELPA upper-triangle tests and -all three native ELPA block-size tests passed. The complete LCAO solver test -still crashes inside the installed two-stage ELPA real routine; the four-rank -complex ELPA tests and HSE case also fail in that library after disabling -XPMEM to avoid a UCX communication error. These failures remain unresolved, -and four-rank ELPA is not included in the agreement claim. No solver-stage -or production INPUT default change is part of this merge. diff --git a/tests/08_EXX/09_KP_HSE_comp/README b/tests/08_EXX/09_KP_HSE_comp/README deleted file mode 100644 index b6f0ce2f9b4..00000000000 --- a/tests/08_EXX/09_KP_HSE_comp/README +++ /dev/null @@ -1 +0,0 @@ -HSE calculation on Si, multiple k-points, exx complex number \ No newline at end of file diff --git a/tests/08_EXX/10_KP_HF/README b/tests/08_EXX/10_KP_HF/README deleted file mode 100644 index 731657f9bb1..00000000000 --- a/tests/08_EXX/10_KP_HF/README +++ /dev/null @@ -1 +0,0 @@ -Hartree-Fock calculation on Si, multiple k-points \ No newline at end of file diff --git a/tests/08_EXX/11_KP_PBE0/README b/tests/08_EXX/11_KP_PBE0/README deleted file mode 100644 index 3e1f8a3a4d4..00000000000 --- a/tests/08_EXX/11_KP_PBE0/README +++ /dev/null @@ -1 +0,0 @@ -PBE0 calculation on Si, multiple k-points \ No newline at end of file diff --git a/tests/08_EXX/12_KP_OXC/README b/tests/08_EXX/12_KP_OXC/README deleted file mode 100644 index 5f49129742e..00000000000 --- a/tests/08_EXX/12_KP_OXC/README +++ /dev/null @@ -1 +0,0 @@ -test for output XC matrix with multipy k-points diff --git a/tests/08_EXX/13_NO_KP_CAMPBEH/README b/tests/08_EXX/13_NO_KP_CAMPBEH/README deleted file mode 100644 index 62f40af5438..00000000000 --- a/tests/08_EXX/13_NO_KP_CAMPBEH/README +++ /dev/null @@ -1 +0,0 @@ -CAM-PBEh calculation on Si, multiple k-points \ No newline at end of file diff --git a/tests/08_EXX/14_NO_TDDFT_PBE0/INPUT b/tests/08_EXX/14_NO_TDDFT_PBE0/INPUT deleted file mode 100644 index f21ac8e1756..00000000000 --- a/tests/08_EXX/14_NO_TDDFT_PBE0/INPUT +++ /dev/null @@ -1,50 +0,0 @@ -INPUT_PARAMETERS - - - -# general information -calculation md -esolver_type tddft -md_type nve -md_nstep 1 -estep_per_md 5 -td_dt 0.005 -md_tfirst 0 - - -# rt-TDDFT parameters -td_vext 1 # add time-dependent external potential -td_vext_dire 3 # direction along z -td_stype 2 # 2: hybrid gauge -td_ttype 3 # Heaviside type potential -td_tstart 1 # the step electric field starts -td_tend 10 # the step electric field ends -td_heavi_t0 3 -td_heavi_amp 0.05 - - -# print out current information -out_current 1 - - -suffix autotest -pseudo_dir ../../PP_ORB -orbital_dir ../../PP_ORB -basis_type lcao -gamma_only 0 - - -# electronic structure calculations -ecutwfc 20 -scf_nmax 50 -scf_thr 1e-6 -ks_solver scalapack_gvx - - -# charge mixing -mixing_type broyden -mixing_beta 0.7 -mixing_gg0 0.0 - -# functional -dft_functional pbe0 diff --git a/tests/08_EXX/14_NO_TDDFT_PBE0/STRU b/tests/08_EXX/14_NO_TDDFT_PBE0/STRU deleted file mode 100644 index 822d2c7cc68..00000000000 --- a/tests/08_EXX/14_NO_TDDFT_PBE0/STRU +++ /dev/null @@ -1,22 +0,0 @@ -ATOMIC_SPECIES -Si 28.085 Si_ONCV_PBE-1.0.upf - -NUMERICAL_ORBITAL -Si_gga_6au_100Ry_2s2p1d.orb - -LATTICE_CONSTANT -10.2 // add lattice constant - -LATTICE_VECTORS -0.0 0.5 0.5 -0.5 0.0 0.5 -0.5 0.5 0.0 - -ATOMIC_POSITIONS -Cartesian //Cartesian or Direct coordinate. - -Si // Element type -0.0 // magnetism -2 // number of atoms -0.00 0.00 0.00 0 0 0 -0.25 0.25 0.25 1 1 1 diff --git a/tests/08_EXX/14_NO_TDDFT_PBE0/current_tot.txt.ref b/tests/08_EXX/14_NO_TDDFT_PBE0/current_tot.txt.ref deleted file mode 100644 index 7a529b6b59f..00000000000 --- a/tests/08_EXX/14_NO_TDDFT_PBE0/current_tot.txt.ref +++ /dev/null @@ -1,6 +0,0 @@ -1 -2.4061105780900972e-19 2.4057746355074182e-19 2.4064137152788421e-19 -2 1.3424215192380810e-08 -1.3424215730033235e-08 2.4369257186711371e-06 -3 3.9266797908203167e-08 -3.9266803842385155e-08 7.3211720761214893e-06 -4 4.7558646233794483e-08 -4.7558654016584407e-08 9.7958584306165467e-06 -5 3.3345693326762254e-08 -3.3345695331858395e-08 9.8568251466587576e-06 -6 4.0507435957670880e-09 -4.0507458176112040e-09 9.9160372634246860e-06 diff --git a/tests/08_EXX/14_NO_TDDFT_PBE0/result.ref b/tests/08_EXX/14_NO_TDDFT_PBE0/result.ref deleted file mode 100644 index 9b443dac72b..00000000000 --- a/tests/08_EXX/14_NO_TDDFT_PBE0/result.ref +++ /dev/null @@ -1,4 +0,0 @@ -etotref -203.7710923478310 -etotperatomref -101.8855461739 -CompareCurrent_pass 0 -totaltimeref 16.38 diff --git a/tests/08_EXX/15_KP_HSE_SOC_symm/INPUT b/tests/08_EXX/15_KP_HSE_SOC_symm/INPUT deleted file mode 100644 index ecb818cc1cf..00000000000 --- a/tests/08_EXX/15_KP_HSE_SOC_symm/INPUT +++ /dev/null @@ -1,32 +0,0 @@ -INPUT_PARAMETERS -suffix autotest -pseudo_dir ../../PP_ORB -orbital_dir ../../PP_ORB - -calculation scf -basis_type lcao -gamma_only 0 - -ecutwfc 5 -scf_thr 1e-2 - -smearing_method gaussian -smearing_sigma 0.02 -mixing_type broyden -mixing_beta 0.15 - -symmetry 1 -symmetry_prec 1e-5 - -nspin 4 -lspinorb 1 -cal_force 1 -cal_stress 1 - -dft_functional hse -exx_separate_loop 0 -exx_hybrid_step 10 -exx_pca_threshold 1e-1 -exx_c_threshold 1e-1 -exx_v_threshold 1 -exx_dm_threshold 1e-2 diff --git a/tests/08_EXX/15_KP_HSE_SOC_symm/STRU b/tests/08_EXX/15_KP_HSE_SOC_symm/STRU deleted file mode 100644 index 41219f94a51..00000000000 --- a/tests/08_EXX/15_KP_HSE_SOC_symm/STRU +++ /dev/null @@ -1,22 +0,0 @@ -ATOMIC_SPECIES -Fe 55.845 Fe.upf - -NUMERICAL_ORBITAL -Fe_gga_6au_100Ry_4s2p2d1f.orb - -LATTICE_CONSTANT -1.8897259886 // 1 Angstrom in Bohr; vectors below in Angstrom (simple hexagonal, a=2.5, c=4.0, uniaxial C_6 axis along z) - -LATTICE_VECTORS - -1.4332500000 1.4332500000 1.4332500000 - 1.4332500000 -1.4332500000 1.4332500000 - 1.4332500000 1.4332500000 -1.4332500000 - -ATOMIC_POSITIONS -Direct - -Fe -0.0 -1 -0.0000000000 0.0000000000 0.0000000000 1 1 1 mag 0 0 2.2 - diff --git a/tests/08_EXX/15_KP_HSE_SOC_symm/result.ref b/tests/08_EXX/15_KP_HSE_SOC_symm/result.ref deleted file mode 100644 index 8b8c03167a3..00000000000 --- a/tests/08_EXX/15_KP_HSE_SOC_symm/result.ref +++ /dev/null @@ -1,9 +0,0 @@ -etotref -3419.2371787940001013 -etotperatomref -3419.2371787940 -totalforceref 0.000000 -totalstressref 20962.912099 -pointgroupref O_h -spacegroupref O_h -nksibzref 3 -magpointgroupref C_4h -totaltimeref 99.07 diff --git a/tests/08_EXX/15_KP_HSE_SOC_symm/threshold b/tests/08_EXX/15_KP_HSE_SOC_symm/threshold deleted file mode 100644 index a3a120cafb1..00000000000 --- a/tests/08_EXX/15_KP_HSE_SOC_symm/threshold +++ /dev/null @@ -1,15 +0,0 @@ -# Loosened tolerances for this deliberately-minimal metallic EXX+SOC case. -# The total energy is NOT bit-reproducible under MPI (np>1): the run-to-run -# seed is non-reproducible MPI floating-point reduction order in the EXX/LibRI -# sums, amplified by an ill-conditioned metallic SCF (ecutwfc=5, anisotropic -# 2x2x1 k-mesh, EXX updated every step) that never reaches a stable fixed point -# and stops mid-oscillation. Observed spread ~0.004 eV peak-to-peak; the -# stress sum (absolute values of 9 components) likewise wobbles run-to-run, -# observed deviation up to ~13 kbar against ref at np=4. The -# discrete symmetry/magnetic-group assertions (O_h / C_4h / nksibz=3), which are -# the real purpose of this test, remain exact. Mirrors sibling metallic HSE -# case 07_KP_CR_HSE. See also: np=1 is bit-identical, confirming the MPI seed. -threshold 0.006 -force_threshold 0.01 -stress_threshold 15 -fatal_threshold 20 diff --git a/tests/08_EXX/51_GO_LR/KPT b/tests/08_EXX/51_GO_LR/KPT deleted file mode 100644 index c289c0158aa..00000000000 --- a/tests/08_EXX/51_GO_LR/KPT +++ /dev/null @@ -1,4 +0,0 @@ -K_POINTS -0 -Gamma -1 1 1 0 0 0 diff --git a/tests/08_EXX/51_GO_LR/README b/tests/08_EXX/51_GO_LR/README deleted file mode 100644 index 59460d6983b..00000000000 --- a/tests/08_EXX/51_GO_LR/README +++ /dev/null @@ -1 +0,0 @@ -test for linear response TDDFT with gamma-only diff --git a/tests/08_EXX/52_GO_LR_PBE/KPT b/tests/08_EXX/52_GO_LR_PBE/KPT deleted file mode 100644 index c289c0158aa..00000000000 --- a/tests/08_EXX/52_GO_LR_PBE/KPT +++ /dev/null @@ -1,4 +0,0 @@ -K_POINTS -0 -Gamma -1 1 1 0 0 0 diff --git a/tests/08_EXX/52_GO_LR_PBE/README b/tests/08_EXX/52_GO_LR_PBE/README deleted file mode 100644 index 1707a78b1dd..00000000000 --- a/tests/08_EXX/52_GO_LR_PBE/README +++ /dev/null @@ -1 +0,0 @@ -test for linear response TDDFT with GGA XC kernel diff --git a/tests/08_EXX/53_GO_LR_HF/KPT b/tests/08_EXX/53_GO_LR_HF/KPT deleted file mode 100644 index c289c0158aa..00000000000 --- a/tests/08_EXX/53_GO_LR_HF/KPT +++ /dev/null @@ -1,4 +0,0 @@ -K_POINTS -0 -Gamma -1 1 1 0 0 0 diff --git a/tests/08_EXX/53_GO_LR_HF/README b/tests/08_EXX/53_GO_LR_HF/README deleted file mode 100644 index aa1d7b8dc2a..00000000000 --- a/tests/08_EXX/53_GO_LR_HF/README +++ /dev/null @@ -1 +0,0 @@ -test for linear response TDHF \ No newline at end of file diff --git a/tests/08_EXX/54_GO_ULR_HF/KPT b/tests/08_EXX/54_GO_ULR_HF/KPT deleted file mode 100644 index c289c0158aa..00000000000 --- a/tests/08_EXX/54_GO_ULR_HF/KPT +++ /dev/null @@ -1,4 +0,0 @@ -K_POINTS -0 -Gamma -1 1 1 0 0 0 diff --git a/tests/08_EXX/54_GO_ULR_HF/README b/tests/08_EXX/54_GO_ULR_HF/README deleted file mode 100644 index ef8a41364a8..00000000000 --- a/tests/08_EXX/54_GO_ULR_HF/README +++ /dev/null @@ -1 +0,0 @@ -test for linear response TDHF using open-shell solver \ No newline at end of file diff --git a/tests/08_EXX/55_KP_LR/README b/tests/08_EXX/55_KP_LR/README deleted file mode 100644 index 6c1689f0aba..00000000000 --- a/tests/08_EXX/55_KP_LR/README +++ /dev/null @@ -1 +0,0 @@ -test for linear response TDDFT with multiple k-points diff --git a/tests/08_EXX/56_GO_RPA_OUTPUT/KPT b/tests/08_EXX/56_GO_RPA_OUTPUT/KPT deleted file mode 100644 index c289c0158aa..00000000000 --- a/tests/08_EXX/56_GO_RPA_OUTPUT/KPT +++ /dev/null @@ -1,4 +0,0 @@ -K_POINTS -0 -Gamma -1 1 1 0 0 0 diff --git a/tests/08_EXX/56_GO_RPA_OUTPUT/README b/tests/08_EXX/56_GO_RPA_OUTPUT/README deleted file mode 100644 index b8f965905a6..00000000000 --- a/tests/08_EXX/56_GO_RPA_OUTPUT/README +++ /dev/null @@ -1 +0,0 @@ -Check molecular RPA output files (`Cs_*`, `coulomb_*`) for a gamma-only H2O case. diff --git a/tests/08_EXX/57_KP_RPA_SHRINK/README b/tests/08_EXX/57_KP_RPA_SHRINK/README deleted file mode 100644 index 651697f9034..00000000000 --- a/tests/08_EXX/57_KP_RPA_SHRINK/README +++ /dev/null @@ -1 +0,0 @@ -Check k-point shrink RPA output files (`shrink_sinvS_*`, `Cs_*`, `coulomb_*`) for a compact two-atom H `1x1x2` solid case. diff --git a/tests/08_EXX/58_KP_LR_BSE/KPT b/tests/08_EXX/58_KP_LR_BSE/KPT deleted file mode 100644 index f5f7f4ec34c..00000000000 --- a/tests/08_EXX/58_KP_LR_BSE/KPT +++ /dev/null @@ -1,4 +0,0 @@ -K_POINTS -0 -Gamma -2 2 2 0 0 0 diff --git a/tests/08_EXX/58_KP_LR_BSE/README b/tests/08_EXX/58_KP_LR_BSE/README deleted file mode 100644 index f029533426e..00000000000 --- a/tests/08_EXX/58_KP_LR_BSE/README +++ /dev/null @@ -1,3 +0,0 @@ -Test for Bethe-Salpeter Equation for two-atom H `2x2x2` solid case. -`energy_qp.txt` are generated by LibRPA with TZDP basis, and set `nbands 10`. -`Wc_xxx.mtx` are generated with `H_gga_6au_60Ry_1s.orb` for quick test. \ No newline at end of file diff --git a/tests/08_EXX/CASES_CPU.txt b/tests/08_EXX/CASES_CPU.txt deleted file mode 100644 index a0372490839..00000000000 --- a/tests/08_EXX/CASES_CPU.txt +++ /dev/null @@ -1,23 +0,0 @@ -01_GO_S1_HSE -02_GO_S2_HSE -03_KP_S4_HSE -04_GO_S1_HSE_loop0 -05_GO_RE_HSE -06_KP_MD_HSE -07_KP_CR_HSE -08_KP_HSE_symm -09_KP_HSE_comp -10_KP_HF -11_KP_PBE0 -12_KP_OXC -13_NO_KP_CAMPBEH -14_NO_TDDFT_PBE0 -#15_KP_HSE_SOC_symm -51_GO_LR -52_GO_LR_PBE -53_GO_LR_HF -54_GO_ULR_HF -55_KP_LR -56_GO_RPA_OUTPUT -57_KP_RPA_SHRINK -58_KP_LR_BSE diff --git a/tests/08_EXX/CMakeLists.txt b/tests/08_EXX/CMakeLists.txt deleted file mode 100644 index 19bc17ba551..00000000000 --- a/tests/08_EXX/CMakeLists.txt +++ /dev/null @@ -1,16 +0,0 @@ -enable_testing() - -find_program(BASH bash) -if(ENABLE_ASAN) - add_test( - NAME 08_EXX_test_with_asan - COMMAND ${BASH} ../integrate/Autotest.sh -a ${ABACUS_BIN_PATH} -n 2 -s true - WORKING_DIRECTORY ${ABACUS_TEST_DIR}/08_EXX - ) -else() - add_test( - NAME 08_EXX - COMMAND ${BASH} ../integrate/Autotest.sh -a ${ABACUS_BIN_PATH} -n 4 - WORKING_DIRECTORY ${ABACUS_TEST_DIR}/08_EXX - ) -endif() diff --git a/tests/08_RI/CASES_CPU.txt b/tests/08_RI/CASES_CPU.txt new file mode 100644 index 00000000000..d94befe08e8 --- /dev/null +++ b/tests/08_RI/CASES_CPU.txt @@ -0,0 +1,24 @@ +scf_hse_gamma +scf_hse_spin2_gamma +scf_hse_spin4_multik +scf_hse_loop0_gamma +relax_hse_gamma +md_hse_multik +relax_cell_hse_multik +scf_hse_symm_multik +scf_hse_complex_multik +scf_hf_multik +scf_pbe0_multik +scf_out_xc_multik +scf_campbeh_gamma +scf_hse_soc_symm_multik +lr_tddft_lda_gamma +lr_tddft_pbe_gamma +lr_tddft_hf_gamma +lr_tddft_hf_ulr_gamma +lr_tddft_multik +rpa_scf_gamma +rpa_scf_shrink_multik +rt_tddft_pbe0_gamma +bse_lr_multik +nscf_hse_multik diff --git a/tests/08_RI/CMakeLists.txt b/tests/08_RI/CMakeLists.txt new file mode 100644 index 00000000000..a16fc04f320 --- /dev/null +++ b/tests/08_RI/CMakeLists.txt @@ -0,0 +1,16 @@ +enable_testing() + +find_program(BASH bash) +if(ENABLE_ASAN) + add_test( + NAME 08_RI_test_with_asan + COMMAND ${BASH} ../integrate/Autotest.sh -a ${ABACUS_BIN_PATH} -n 2 -s true + WORKING_DIRECTORY ${ABACUS_TEST_DIR}/08_RI + ) +else() + add_test( + NAME 08_RI + COMMAND ${BASH} ../integrate/Autotest.sh -a ${ABACUS_BIN_PATH} -n 4 -j auto + WORKING_DIRECTORY ${ABACUS_TEST_DIR}/08_RI + ) +endif() diff --git a/tests/08_EXX/58_KP_LR_BSE/INPUT b/tests/08_RI/bse_lr_multik/INPUT similarity index 100% rename from tests/08_EXX/58_KP_LR_BSE/INPUT rename to tests/08_RI/bse_lr_multik/INPUT diff --git a/tests/01_PW/070_PW_CR_fix_c/KPT b/tests/08_RI/bse_lr_multik/KPT similarity index 100% rename from tests/01_PW/070_PW_CR_fix_c/KPT rename to tests/08_RI/bse_lr_multik/KPT diff --git a/tests/08_RI/bse_lr_multik/README b/tests/08_RI/bse_lr_multik/README new file mode 100644 index 00000000000..cf94bd29863 --- /dev/null +++ b/tests/08_RI/bse_lr_multik/README @@ -0,0 +1 @@ +BSE LR on H2O, multik, bse_spin_types=singlet triplet, bse_tda=tda. diff --git a/tests/08_EXX/57_KP_RPA_SHRINK/STRU b/tests/08_RI/bse_lr_multik/STRU similarity index 100% rename from tests/08_EXX/57_KP_RPA_SHRINK/STRU rename to tests/08_RI/bse_lr_multik/STRU diff --git a/tests/08_EXX/58_KP_LR_BSE/librpa.d/Cs_data_1.txt b/tests/08_RI/bse_lr_multik/librpa.d/Cs_data_1.txt similarity index 100% rename from tests/08_EXX/58_KP_LR_BSE/librpa.d/Cs_data_1.txt rename to tests/08_RI/bse_lr_multik/librpa.d/Cs_data_1.txt diff --git a/tests/08_EXX/58_KP_LR_BSE/librpa.d/Cs_shrinked_data_1.txt b/tests/08_RI/bse_lr_multik/librpa.d/Cs_shrinked_data_1.txt similarity index 100% rename from tests/08_EXX/58_KP_LR_BSE/librpa.d/Cs_shrinked_data_1.txt rename to tests/08_RI/bse_lr_multik/librpa.d/Cs_shrinked_data_1.txt diff --git a/tests/08_EXX/58_KP_LR_BSE/librpa.d/KS_eigenvector.txt b/tests/08_RI/bse_lr_multik/librpa.d/KS_eigenvector.txt similarity index 100% rename from tests/08_EXX/58_KP_LR_BSE/librpa.d/KS_eigenvector.txt rename to tests/08_RI/bse_lr_multik/librpa.d/KS_eigenvector.txt diff --git a/tests/08_EXX/58_KP_LR_BSE/librpa.d/Wc_iR_0_ifreq_0.mtx b/tests/08_RI/bse_lr_multik/librpa.d/Wc_iR_0_ifreq_0.mtx similarity index 100% rename from tests/08_EXX/58_KP_LR_BSE/librpa.d/Wc_iR_0_ifreq_0.mtx rename to tests/08_RI/bse_lr_multik/librpa.d/Wc_iR_0_ifreq_0.mtx diff --git a/tests/08_EXX/58_KP_LR_BSE/librpa.d/Wc_iR_1_ifreq_0.mtx b/tests/08_RI/bse_lr_multik/librpa.d/Wc_iR_1_ifreq_0.mtx similarity index 100% rename from tests/08_EXX/58_KP_LR_BSE/librpa.d/Wc_iR_1_ifreq_0.mtx rename to tests/08_RI/bse_lr_multik/librpa.d/Wc_iR_1_ifreq_0.mtx diff --git a/tests/08_EXX/58_KP_LR_BSE/librpa.d/Wc_iR_2_ifreq_0.mtx b/tests/08_RI/bse_lr_multik/librpa.d/Wc_iR_2_ifreq_0.mtx similarity index 100% rename from tests/08_EXX/58_KP_LR_BSE/librpa.d/Wc_iR_2_ifreq_0.mtx rename to tests/08_RI/bse_lr_multik/librpa.d/Wc_iR_2_ifreq_0.mtx diff --git a/tests/08_EXX/58_KP_LR_BSE/librpa.d/Wc_iR_3_ifreq_0.mtx b/tests/08_RI/bse_lr_multik/librpa.d/Wc_iR_3_ifreq_0.mtx similarity index 100% rename from tests/08_EXX/58_KP_LR_BSE/librpa.d/Wc_iR_3_ifreq_0.mtx rename to tests/08_RI/bse_lr_multik/librpa.d/Wc_iR_3_ifreq_0.mtx diff --git a/tests/08_EXX/58_KP_LR_BSE/librpa.d/Wc_iR_4_ifreq_0.mtx b/tests/08_RI/bse_lr_multik/librpa.d/Wc_iR_4_ifreq_0.mtx similarity index 100% rename from tests/08_EXX/58_KP_LR_BSE/librpa.d/Wc_iR_4_ifreq_0.mtx rename to tests/08_RI/bse_lr_multik/librpa.d/Wc_iR_4_ifreq_0.mtx diff --git a/tests/08_EXX/58_KP_LR_BSE/librpa.d/Wc_iR_5_ifreq_0.mtx b/tests/08_RI/bse_lr_multik/librpa.d/Wc_iR_5_ifreq_0.mtx similarity index 100% rename from tests/08_EXX/58_KP_LR_BSE/librpa.d/Wc_iR_5_ifreq_0.mtx rename to tests/08_RI/bse_lr_multik/librpa.d/Wc_iR_5_ifreq_0.mtx diff --git a/tests/08_EXX/58_KP_LR_BSE/librpa.d/Wc_iR_6_ifreq_0.mtx b/tests/08_RI/bse_lr_multik/librpa.d/Wc_iR_6_ifreq_0.mtx similarity index 100% rename from tests/08_EXX/58_KP_LR_BSE/librpa.d/Wc_iR_6_ifreq_0.mtx rename to tests/08_RI/bse_lr_multik/librpa.d/Wc_iR_6_ifreq_0.mtx diff --git a/tests/08_EXX/58_KP_LR_BSE/librpa.d/Wc_iR_7_ifreq_0.mtx b/tests/08_RI/bse_lr_multik/librpa.d/Wc_iR_7_ifreq_0.mtx similarity index 100% rename from tests/08_EXX/58_KP_LR_BSE/librpa.d/Wc_iR_7_ifreq_0.mtx rename to tests/08_RI/bse_lr_multik/librpa.d/Wc_iR_7_ifreq_0.mtx diff --git a/tests/08_EXX/58_KP_LR_BSE/librpa.d/band_out.txt b/tests/08_RI/bse_lr_multik/librpa.d/band_out.txt similarity index 100% rename from tests/08_EXX/58_KP_LR_BSE/librpa.d/band_out.txt rename to tests/08_RI/bse_lr_multik/librpa.d/band_out.txt diff --git a/tests/08_EXX/58_KP_LR_BSE/librpa.d/coulomb_cut_1.txt b/tests/08_RI/bse_lr_multik/librpa.d/coulomb_cut_1.txt similarity index 100% rename from tests/08_EXX/58_KP_LR_BSE/librpa.d/coulomb_cut_1.txt rename to tests/08_RI/bse_lr_multik/librpa.d/coulomb_cut_1.txt diff --git a/tests/08_EXX/58_KP_LR_BSE/librpa.d/coulomb_mat_1.txt b/tests/08_RI/bse_lr_multik/librpa.d/coulomb_mat_1.txt similarity index 100% rename from tests/08_EXX/58_KP_LR_BSE/librpa.d/coulomb_mat_1.txt rename to tests/08_RI/bse_lr_multik/librpa.d/coulomb_mat_1.txt diff --git a/tests/08_EXX/58_KP_LR_BSE/librpa.d/coulomb_unshrinked_cut_1.txt b/tests/08_RI/bse_lr_multik/librpa.d/coulomb_unshrinked_cut_1.txt similarity index 100% rename from tests/08_EXX/58_KP_LR_BSE/librpa.d/coulomb_unshrinked_cut_1.txt rename to tests/08_RI/bse_lr_multik/librpa.d/coulomb_unshrinked_cut_1.txt diff --git a/tests/08_EXX/58_KP_LR_BSE/librpa.d/energy_qp.txt b/tests/08_RI/bse_lr_multik/librpa.d/energy_qp.txt similarity index 100% rename from tests/08_EXX/58_KP_LR_BSE/librpa.d/energy_qp.txt rename to tests/08_RI/bse_lr_multik/librpa.d/energy_qp.txt diff --git a/tests/08_EXX/58_KP_LR_BSE/librpa.d/shrink_sinvS_0.txt b/tests/08_RI/bse_lr_multik/librpa.d/shrink_sinvS_0.txt similarity index 100% rename from tests/08_EXX/58_KP_LR_BSE/librpa.d/shrink_sinvS_0.txt rename to tests/08_RI/bse_lr_multik/librpa.d/shrink_sinvS_0.txt diff --git a/tests/08_EXX/58_KP_LR_BSE/librpa.d/stru_out.txt b/tests/08_RI/bse_lr_multik/librpa.d/stru_out.txt similarity index 100% rename from tests/08_EXX/58_KP_LR_BSE/librpa.d/stru_out.txt rename to tests/08_RI/bse_lr_multik/librpa.d/stru_out.txt diff --git a/tests/08_EXX/58_KP_LR_BSE/librpa.d/velocity_matrix.txt b/tests/08_RI/bse_lr_multik/librpa.d/velocity_matrix.txt similarity index 100% rename from tests/08_EXX/58_KP_LR_BSE/librpa.d/velocity_matrix.txt rename to tests/08_RI/bse_lr_multik/librpa.d/velocity_matrix.txt diff --git a/tests/08_EXX/58_KP_LR_BSE/librpa.d/vxc_out.txt b/tests/08_RI/bse_lr_multik/librpa.d/vxc_out.txt similarity index 100% rename from tests/08_EXX/58_KP_LR_BSE/librpa.d/vxc_out.txt rename to tests/08_RI/bse_lr_multik/librpa.d/vxc_out.txt diff --git a/tests/08_EXX/58_KP_LR_BSE/result.ref b/tests/08_RI/bse_lr_multik/result.ref similarity index 100% rename from tests/08_EXX/58_KP_LR_BSE/result.ref rename to tests/08_RI/bse_lr_multik/result.ref diff --git a/tests/08_EXX/53_GO_LR_HF/INPUT b/tests/08_RI/lr_tddft_hf_gamma/INPUT similarity index 100% rename from tests/08_EXX/53_GO_LR_HF/INPUT rename to tests/08_RI/lr_tddft_hf_gamma/INPUT diff --git a/tests/01_PW/802_PW_LT_fcc/KPT b/tests/08_RI/lr_tddft_hf_gamma/KPT similarity index 100% rename from tests/01_PW/802_PW_LT_fcc/KPT rename to tests/08_RI/lr_tddft_hf_gamma/KPT diff --git a/tests/08_RI/lr_tddft_hf_gamma/README b/tests/08_RI/lr_tddft_hf_gamma/README new file mode 100644 index 00000000000..4217b43320c --- /dev/null +++ b/tests/08_RI/lr_tddft_hf_gamma/README @@ -0,0 +1 @@ +LR-TDDFT HF on H2O, gamma_only, nspin=2, lr_nstates=2, lr_solver=dav. diff --git a/tests/08_EXX/53_GO_LR_HF/STRU b/tests/08_RI/lr_tddft_hf_gamma/STRU similarity index 100% rename from tests/08_EXX/53_GO_LR_HF/STRU rename to tests/08_RI/lr_tddft_hf_gamma/STRU diff --git a/tests/08_EXX/53_GO_LR_HF/result.ref b/tests/08_RI/lr_tddft_hf_gamma/result.ref similarity index 100% rename from tests/08_EXX/53_GO_LR_HF/result.ref rename to tests/08_RI/lr_tddft_hf_gamma/result.ref diff --git a/tests/08_EXX/54_GO_ULR_HF/INPUT b/tests/08_RI/lr_tddft_hf_ulr_gamma/INPUT similarity index 100% rename from tests/08_EXX/54_GO_ULR_HF/INPUT rename to tests/08_RI/lr_tddft_hf_ulr_gamma/INPUT diff --git a/tests/01_PW/803_PW_LT_bcc/KPT b/tests/08_RI/lr_tddft_hf_ulr_gamma/KPT similarity index 100% rename from tests/01_PW/803_PW_LT_bcc/KPT rename to tests/08_RI/lr_tddft_hf_ulr_gamma/KPT diff --git a/tests/08_RI/lr_tddft_hf_ulr_gamma/README b/tests/08_RI/lr_tddft_hf_ulr_gamma/README new file mode 100644 index 00000000000..6337e243c9c --- /dev/null +++ b/tests/08_RI/lr_tddft_hf_ulr_gamma/README @@ -0,0 +1 @@ +LR-TDDFT HF on H2O, gamma_only, nspin=2, lr_unrestricted=1, lr_nstates=3, lr_solver=lapack. diff --git a/tests/08_EXX/54_GO_ULR_HF/STRU b/tests/08_RI/lr_tddft_hf_ulr_gamma/STRU similarity index 100% rename from tests/08_EXX/54_GO_ULR_HF/STRU rename to tests/08_RI/lr_tddft_hf_ulr_gamma/STRU diff --git a/tests/08_EXX/54_GO_ULR_HF/result.ref b/tests/08_RI/lr_tddft_hf_ulr_gamma/result.ref similarity index 100% rename from tests/08_EXX/54_GO_ULR_HF/result.ref rename to tests/08_RI/lr_tddft_hf_ulr_gamma/result.ref diff --git a/tests/08_EXX/51_GO_LR/INPUT b/tests/08_RI/lr_tddft_lda_gamma/INPUT similarity index 100% rename from tests/08_EXX/51_GO_LR/INPUT rename to tests/08_RI/lr_tddft_lda_gamma/INPUT diff --git a/tests/01_PW/804_PW_LT_hex/KPT b/tests/08_RI/lr_tddft_lda_gamma/KPT similarity index 100% rename from tests/01_PW/804_PW_LT_hex/KPT rename to tests/08_RI/lr_tddft_lda_gamma/KPT diff --git a/tests/08_RI/lr_tddft_lda_gamma/README b/tests/08_RI/lr_tddft_lda_gamma/README new file mode 100644 index 00000000000..4d2ba97bb19 --- /dev/null +++ b/tests/08_RI/lr_tddft_lda_gamma/README @@ -0,0 +1 @@ +LR-TDDFT LDA on H2O, gamma_only, nspin=2, lr_nstates=2, lr_solver=dav. diff --git a/tests/08_EXX/51_GO_LR/STRU b/tests/08_RI/lr_tddft_lda_gamma/STRU similarity index 100% rename from tests/08_EXX/51_GO_LR/STRU rename to tests/08_RI/lr_tddft_lda_gamma/STRU diff --git a/tests/08_EXX/51_GO_LR/result.ref b/tests/08_RI/lr_tddft_lda_gamma/result.ref similarity index 100% rename from tests/08_EXX/51_GO_LR/result.ref rename to tests/08_RI/lr_tddft_lda_gamma/result.ref diff --git a/tests/08_EXX/55_KP_LR/INPUT b/tests/08_RI/lr_tddft_multik/INPUT similarity index 100% rename from tests/08_EXX/55_KP_LR/INPUT rename to tests/08_RI/lr_tddft_multik/INPUT diff --git a/tests/08_EXX/55_KP_LR/KPT b/tests/08_RI/lr_tddft_multik/KPT similarity index 100% rename from tests/08_EXX/55_KP_LR/KPT rename to tests/08_RI/lr_tddft_multik/KPT diff --git a/tests/08_RI/lr_tddft_multik/README b/tests/08_RI/lr_tddft_multik/README new file mode 100644 index 00000000000..b7038266094 --- /dev/null +++ b/tests/08_RI/lr_tddft_multik/README @@ -0,0 +1 @@ +LR-TDDFT on H2O, multik, nspin=2, lr_nstates=2, lr_solver=dav_subspace. diff --git a/tests/08_EXX/55_KP_LR/STRU b/tests/08_RI/lr_tddft_multik/STRU similarity index 100% rename from tests/08_EXX/55_KP_LR/STRU rename to tests/08_RI/lr_tddft_multik/STRU diff --git a/tests/08_EXX/55_KP_LR/result.ref b/tests/08_RI/lr_tddft_multik/result.ref similarity index 100% rename from tests/08_EXX/55_KP_LR/result.ref rename to tests/08_RI/lr_tddft_multik/result.ref diff --git a/tests/08_EXX/52_GO_LR_PBE/INPUT b/tests/08_RI/lr_tddft_pbe_gamma/INPUT similarity index 100% rename from tests/08_EXX/52_GO_LR_PBE/INPUT rename to tests/08_RI/lr_tddft_pbe_gamma/INPUT diff --git a/tests/01_PW/805_PW_LT_trigonal/KPT b/tests/08_RI/lr_tddft_pbe_gamma/KPT similarity index 100% rename from tests/01_PW/805_PW_LT_trigonal/KPT rename to tests/08_RI/lr_tddft_pbe_gamma/KPT diff --git a/tests/08_RI/lr_tddft_pbe_gamma/README b/tests/08_RI/lr_tddft_pbe_gamma/README new file mode 100644 index 00000000000..ed8d8ab79df --- /dev/null +++ b/tests/08_RI/lr_tddft_pbe_gamma/README @@ -0,0 +1 @@ +LR-TDDFT PBE on H2O, gamma_only, nspin=2, lr_nstates=2, lr_solver=dav. diff --git a/tests/08_EXX/52_GO_LR_PBE/STRU b/tests/08_RI/lr_tddft_pbe_gamma/STRU similarity index 100% rename from tests/08_EXX/52_GO_LR_PBE/STRU rename to tests/08_RI/lr_tddft_pbe_gamma/STRU diff --git a/tests/08_EXX/52_GO_LR_PBE/result.ref b/tests/08_RI/lr_tddft_pbe_gamma/result.ref similarity index 100% rename from tests/08_EXX/52_GO_LR_PBE/result.ref rename to tests/08_RI/lr_tddft_pbe_gamma/result.ref diff --git a/tests/08_EXX/06_KP_MD_HSE/INPUT b/tests/08_RI/md_hse_multik/INPUT similarity index 100% rename from tests/08_EXX/06_KP_MD_HSE/INPUT rename to tests/08_RI/md_hse_multik/INPUT diff --git a/tests/08_RI/md_hse_multik/KPT b/tests/08_RI/md_hse_multik/KPT new file mode 100644 index 00000000000..230094748bd --- /dev/null +++ b/tests/08_RI/md_hse_multik/KPT @@ -0,0 +1,4 @@ +K_POINTS +0 +Gamma +1 2 1 0 0 0 diff --git a/tests/08_RI/md_hse_multik/README b/tests/08_RI/md_hse_multik/README new file mode 100644 index 00000000000..73eb25d68c2 --- /dev/null +++ b/tests/08_RI/md_hse_multik/README @@ -0,0 +1 @@ +md HSE on H2O, multik, nspin=1, exx_separate_loop=1. diff --git a/tests/08_EXX/06_KP_MD_HSE/STRU b/tests/08_RI/md_hse_multik/STRU similarity index 100% rename from tests/08_EXX/06_KP_MD_HSE/STRU rename to tests/08_RI/md_hse_multik/STRU diff --git a/tests/08_RI/md_hse_multik/result.ref b/tests/08_RI/md_hse_multik/result.ref new file mode 100644 index 00000000000..ff5a2d05be3 --- /dev/null +++ b/tests/08_RI/md_hse_multik/result.ref @@ -0,0 +1,3 @@ +etotref -467.270790968887 +etotperatomref -155.7569303230 +totaltimeref 3.40 diff --git a/tests/08_RI/nscf_hse_multik/INPUT b/tests/08_RI/nscf_hse_multik/INPUT new file mode 100644 index 00000000000..a3f80dc0fb7 --- /dev/null +++ b/tests/08_RI/nscf_hse_multik/INPUT @@ -0,0 +1,38 @@ +INPUT_PARAMETERS +#Parameters (General) +suffix autotest +calculation nscf +#out_dmr 1 +init_chg dm +read_file_dir . + +symmetry 0 +nspin 1 +pseudo_dir ../../PP_ORB +orbital_dir ../../PP_ORB + +#Parameters (PW) +ecutwfc 5 +scf_thr 1e-7 + +#Parameters (Relaxation) +scf_nmax 100 + +#Parameters (LCAO) +basis_type lcao +gamma_only 0 + +#Parameters (Smearing) +smearing_method gaussian +smearing_sigma 0.02 + +#Parameters (Charge Mixing) +mixing_type broyden +mixing_beta 0.4 + +#Parameters (exx) +dft_functional hse +exx_pca_threshold 1e-1 +exx_ccp_rmesh_times 1 +exx_separate_loop 1 + diff --git a/tests/08_RI/nscf_hse_multik/KPT b/tests/08_RI/nscf_hse_multik/KPT new file mode 100644 index 00000000000..230094748bd --- /dev/null +++ b/tests/08_RI/nscf_hse_multik/KPT @@ -0,0 +1,4 @@ +K_POINTS +0 +Gamma +1 2 1 0 0 0 diff --git a/tests/08_RI/nscf_hse_multik/README b/tests/08_RI/nscf_hse_multik/README new file mode 100644 index 00000000000..28a8f46c7fc --- /dev/null +++ b/tests/08_RI/nscf_hse_multik/README @@ -0,0 +1,2 @@ +HSE NSCF on H2O, multik (gamma_only=0), nspin=1, init_chg=dm (reads dmrs1_nao.csr), exx_separate_loop=1. +Requires a preceding SCF run with out_dmr=1 to generate the DM file. diff --git a/tests/08_EXX/01_GO_S1_HSE/STRU b/tests/08_RI/nscf_hse_multik/STRU similarity index 100% rename from tests/08_EXX/01_GO_S1_HSE/STRU rename to tests/08_RI/nscf_hse_multik/STRU diff --git a/tests/08_RI/nscf_hse_multik/dmrs1_nao.csr b/tests/08_RI/nscf_hse_multik/dmrs1_nao.csr new file mode 100644 index 00000000000..92a5ce09e92 --- /dev/null +++ b/tests/08_RI/nscf_hse_multik/dmrs1_nao.csr @@ -0,0 +1,55 @@ + --- Ionic Step 1 --- + # print density matrix in real space DM(R) + 1 # number of spin directions + 1 # spin index + 6 # number of localized basis + 3 # number of Bravais lattice vector R + + user_defined_lattice + 1 + 10 0 0 + 0 10 0 + 0 0 10 + H O + 2 1 + Direct + 0.0979299999999057 0.00422000000071421 0 + 0.971340000000055 0.0937300000005052 0 + 0 0 0 + + #----------------------------------------------------------------------# + # CSR Format # + # The outer loop corresponds to the number of Bravais lattice vectors. # + # The first line contains the index of the Bravais lattice vector # + # (Rx, Ry, Rz), followed by the number of non-zero elements. # + # The subsequent lines consist of three blocks of data, which are # + # values, column indices, row pointers. # + #----------------------------------------------------------------------# + + -1 0 0 4 + # CSR values + -1.74704399e-01 -2.51410026e-01 1.24539629e-01 -7.44890946e-01 + # CSR column indices + 1 1 1 1 + # CSR row pointers + 0 1 1 2 2 3 4 + + 0 0 0 18 + # CSR values + 7.33227857e-01 -2.53104629e-01 -7.46329254e-01 -1.07367635e-01 7.23899259e-01 -2.53104629e-01 + 2.22673723e+00 3.30396323e-01 4.50272155e-01 2.00000000e+00 -7.46329254e-01 3.30396323e-01 + 7.65000705e-01 1.70944993e-01 -1.07367635e-01 4.50272155e-01 1.70944993e-01 8.87672833e-01 + # CSR column indices + 0 2 4 5 1 0 2 4 5 3 0 2 4 5 0 2 + 4 5 + # CSR row pointers + 0 4 5 9 10 14 18 + + 1 0 0 4 + # CSR values + -1.74704399e-01 -2.51410026e-01 1.24539629e-01 -7.44890946e-01 + # CSR column indices + 0 2 4 5 + # CSR row pointers + 0 0 4 4 4 4 4 + diff --git a/tests/08_RI/nscf_hse_multik/result.ref b/tests/08_RI/nscf_hse_multik/result.ref new file mode 100644 index 00000000000..7a4b4121c27 --- /dev/null +++ b/tests/08_RI/nscf_hse_multik/result.ref @@ -0,0 +1,3 @@ +etotref -429.6734298284445 +etotperatomref -143.2244766095 +totaltimeref 0.72 diff --git a/tests/08_EXX/07_KP_CR_HSE/INPUT b/tests/08_RI/relax_cell_hse_multik/INPUT similarity index 100% rename from tests/08_EXX/07_KP_CR_HSE/INPUT rename to tests/08_RI/relax_cell_hse_multik/INPUT diff --git a/tests/08_RI/relax_cell_hse_multik/KPT b/tests/08_RI/relax_cell_hse_multik/KPT new file mode 100644 index 00000000000..e769af76382 --- /dev/null +++ b/tests/08_RI/relax_cell_hse_multik/KPT @@ -0,0 +1,4 @@ +K_POINTS +0 +Gamma +2 1 1 0 0 0 diff --git a/tests/08_RI/relax_cell_hse_multik/README b/tests/08_RI/relax_cell_hse_multik/README new file mode 100644 index 00000000000..67cec84360c --- /dev/null +++ b/tests/08_RI/relax_cell_hse_multik/README @@ -0,0 +1 @@ +cell-relax HSE on H2O, multik, nspin=1, exx_separate_loop=1. diff --git a/tests/08_EXX/07_KP_CR_HSE/STRU b/tests/08_RI/relax_cell_hse_multik/STRU similarity index 100% rename from tests/08_EXX/07_KP_CR_HSE/STRU rename to tests/08_RI/relax_cell_hse_multik/STRU diff --git a/tests/08_EXX/07_KP_CR_HSE/result.ref b/tests/08_RI/relax_cell_hse_multik/result.ref similarity index 100% rename from tests/08_EXX/07_KP_CR_HSE/result.ref rename to tests/08_RI/relax_cell_hse_multik/result.ref diff --git a/tests/08_EXX/07_KP_CR_HSE/threshold b/tests/08_RI/relax_cell_hse_multik/threshold similarity index 100% rename from tests/08_EXX/07_KP_CR_HSE/threshold rename to tests/08_RI/relax_cell_hse_multik/threshold diff --git a/tests/08_EXX/05_GO_RE_HSE/INPUT b/tests/08_RI/relax_hse_gamma/INPUT similarity index 100% rename from tests/08_EXX/05_GO_RE_HSE/INPUT rename to tests/08_RI/relax_hse_gamma/INPUT diff --git a/tests/01_PW/811_PW_LT_bco/KPT b/tests/08_RI/relax_hse_gamma/KPT similarity index 100% rename from tests/01_PW/811_PW_LT_bco/KPT rename to tests/08_RI/relax_hse_gamma/KPT diff --git a/tests/08_RI/relax_hse_gamma/README b/tests/08_RI/relax_hse_gamma/README new file mode 100644 index 00000000000..0989e44a574 --- /dev/null +++ b/tests/08_RI/relax_hse_gamma/README @@ -0,0 +1 @@ +relax HSE on H2O, gamma_only, nspin=1, exx_separate_loop=1. diff --git a/tests/08_EXX/05_GO_RE_HSE/STRU b/tests/08_RI/relax_hse_gamma/STRU similarity index 100% rename from tests/08_EXX/05_GO_RE_HSE/STRU rename to tests/08_RI/relax_hse_gamma/STRU diff --git a/tests/08_RI/relax_hse_gamma/result.ref b/tests/08_RI/relax_hse_gamma/result.ref new file mode 100644 index 00000000000..925d8f9b61d --- /dev/null +++ b/tests/08_RI/relax_hse_gamma/result.ref @@ -0,0 +1,3 @@ +etotref -461.2147597707549 +etotperatomref -153.7382532569 +totaltimeref 3.20 diff --git a/tests/08_EXX/56_GO_RPA_OUTPUT/INPUT b/tests/08_RI/rpa_scf_gamma/INPUT similarity index 100% rename from tests/08_EXX/56_GO_RPA_OUTPUT/INPUT rename to tests/08_RI/rpa_scf_gamma/INPUT diff --git a/tests/02_NAO_Gamma/relax_bfgs2/KPT b/tests/08_RI/rpa_scf_gamma/KPT similarity index 100% rename from tests/02_NAO_Gamma/relax_bfgs2/KPT rename to tests/08_RI/rpa_scf_gamma/KPT diff --git a/tests/08_RI/rpa_scf_gamma/README b/tests/08_RI/rpa_scf_gamma/README new file mode 100644 index 00000000000..c1300036e35 --- /dev/null +++ b/tests/08_RI/rpa_scf_gamma/README @@ -0,0 +1 @@ +RPA scf on H2O, gamma_only, exx_real_number=1, exx_pca_threshold=1e-2. diff --git a/tests/08_EXX/56_GO_RPA_OUTPUT/STRU b/tests/08_RI/rpa_scf_gamma/STRU similarity index 100% rename from tests/08_EXX/56_GO_RPA_OUTPUT/STRU rename to tests/08_RI/rpa_scf_gamma/STRU diff --git a/tests/08_EXX/56_GO_RPA_OUTPUT/refCs_data_1.txt b/tests/08_RI/rpa_scf_gamma/refCs_data_1.txt similarity index 100% rename from tests/08_EXX/56_GO_RPA_OUTPUT/refCs_data_1.txt rename to tests/08_RI/rpa_scf_gamma/refCs_data_1.txt diff --git a/tests/08_EXX/56_GO_RPA_OUTPUT/refCs_data_2.txt b/tests/08_RI/rpa_scf_gamma/refCs_data_2.txt similarity index 100% rename from tests/08_EXX/56_GO_RPA_OUTPUT/refCs_data_2.txt rename to tests/08_RI/rpa_scf_gamma/refCs_data_2.txt diff --git a/tests/08_EXX/56_GO_RPA_OUTPUT/refCs_data_3.txt b/tests/08_RI/rpa_scf_gamma/refCs_data_3.txt similarity index 100% rename from tests/08_EXX/56_GO_RPA_OUTPUT/refCs_data_3.txt rename to tests/08_RI/rpa_scf_gamma/refCs_data_3.txt diff --git a/tests/08_EXX/56_GO_RPA_OUTPUT/refCs_data_4.txt b/tests/08_RI/rpa_scf_gamma/refCs_data_4.txt similarity index 100% rename from tests/08_EXX/56_GO_RPA_OUTPUT/refCs_data_4.txt rename to tests/08_RI/rpa_scf_gamma/refCs_data_4.txt diff --git a/tests/08_EXX/56_GO_RPA_OUTPUT/refcoulomb_cut_1.txt b/tests/08_RI/rpa_scf_gamma/refcoulomb_cut_1.txt similarity index 100% rename from tests/08_EXX/56_GO_RPA_OUTPUT/refcoulomb_cut_1.txt rename to tests/08_RI/rpa_scf_gamma/refcoulomb_cut_1.txt diff --git a/tests/08_EXX/56_GO_RPA_OUTPUT/refcoulomb_cut_2.txt b/tests/08_RI/rpa_scf_gamma/refcoulomb_cut_2.txt similarity index 100% rename from tests/08_EXX/56_GO_RPA_OUTPUT/refcoulomb_cut_2.txt rename to tests/08_RI/rpa_scf_gamma/refcoulomb_cut_2.txt diff --git a/tests/08_EXX/56_GO_RPA_OUTPUT/refcoulomb_cut_3.txt b/tests/08_RI/rpa_scf_gamma/refcoulomb_cut_3.txt similarity index 100% rename from tests/08_EXX/56_GO_RPA_OUTPUT/refcoulomb_cut_3.txt rename to tests/08_RI/rpa_scf_gamma/refcoulomb_cut_3.txt diff --git a/tests/08_EXX/56_GO_RPA_OUTPUT/refcoulomb_cut_4.txt b/tests/08_RI/rpa_scf_gamma/refcoulomb_cut_4.txt similarity index 100% rename from tests/08_EXX/56_GO_RPA_OUTPUT/refcoulomb_cut_4.txt rename to tests/08_RI/rpa_scf_gamma/refcoulomb_cut_4.txt diff --git a/tests/08_EXX/56_GO_RPA_OUTPUT/refcoulomb_mat_1.txt b/tests/08_RI/rpa_scf_gamma/refcoulomb_mat_1.txt similarity index 100% rename from tests/08_EXX/56_GO_RPA_OUTPUT/refcoulomb_mat_1.txt rename to tests/08_RI/rpa_scf_gamma/refcoulomb_mat_1.txt diff --git a/tests/08_EXX/56_GO_RPA_OUTPUT/refcoulomb_mat_2.txt b/tests/08_RI/rpa_scf_gamma/refcoulomb_mat_2.txt similarity index 100% rename from tests/08_EXX/56_GO_RPA_OUTPUT/refcoulomb_mat_2.txt rename to tests/08_RI/rpa_scf_gamma/refcoulomb_mat_2.txt diff --git a/tests/08_EXX/56_GO_RPA_OUTPUT/refcoulomb_mat_3.txt b/tests/08_RI/rpa_scf_gamma/refcoulomb_mat_3.txt similarity index 100% rename from tests/08_EXX/56_GO_RPA_OUTPUT/refcoulomb_mat_3.txt rename to tests/08_RI/rpa_scf_gamma/refcoulomb_mat_3.txt diff --git a/tests/08_EXX/56_GO_RPA_OUTPUT/refcoulomb_mat_4.txt b/tests/08_RI/rpa_scf_gamma/refcoulomb_mat_4.txt similarity index 100% rename from tests/08_EXX/56_GO_RPA_OUTPUT/refcoulomb_mat_4.txt rename to tests/08_RI/rpa_scf_gamma/refcoulomb_mat_4.txt diff --git a/tests/08_EXX/56_GO_RPA_OUTPUT/result.ref b/tests/08_RI/rpa_scf_gamma/result.ref similarity index 100% rename from tests/08_EXX/56_GO_RPA_OUTPUT/result.ref rename to tests/08_RI/rpa_scf_gamma/result.ref diff --git a/tests/08_EXX/57_KP_RPA_SHRINK/INPUT b/tests/08_RI/rpa_scf_shrink_multik/INPUT similarity index 100% rename from tests/08_EXX/57_KP_RPA_SHRINK/INPUT rename to tests/08_RI/rpa_scf_shrink_multik/INPUT diff --git a/tests/08_EXX/57_KP_RPA_SHRINK/KPT b/tests/08_RI/rpa_scf_shrink_multik/KPT similarity index 100% rename from tests/08_EXX/57_KP_RPA_SHRINK/KPT rename to tests/08_RI/rpa_scf_shrink_multik/KPT diff --git a/tests/08_RI/rpa_scf_shrink_multik/README b/tests/08_RI/rpa_scf_shrink_multik/README new file mode 100644 index 00000000000..648468c1856 --- /dev/null +++ b/tests/08_RI/rpa_scf_shrink_multik/README @@ -0,0 +1 @@ +RPA scf on H2O, multik, exx_real_number=1, exx_pca_threshold=1e-3. diff --git a/tests/08_EXX/58_KP_LR_BSE/STRU b/tests/08_RI/rpa_scf_shrink_multik/STRU similarity index 100% rename from tests/08_EXX/58_KP_LR_BSE/STRU rename to tests/08_RI/rpa_scf_shrink_multik/STRU diff --git a/tests/08_EXX/57_KP_RPA_SHRINK/refCs_data_1.txt b/tests/08_RI/rpa_scf_shrink_multik/refCs_data_1.txt similarity index 100% rename from tests/08_EXX/57_KP_RPA_SHRINK/refCs_data_1.txt rename to tests/08_RI/rpa_scf_shrink_multik/refCs_data_1.txt diff --git a/tests/08_EXX/57_KP_RPA_SHRINK/refCs_data_2.txt b/tests/08_RI/rpa_scf_shrink_multik/refCs_data_2.txt similarity index 100% rename from tests/08_EXX/57_KP_RPA_SHRINK/refCs_data_2.txt rename to tests/08_RI/rpa_scf_shrink_multik/refCs_data_2.txt diff --git a/tests/08_EXX/57_KP_RPA_SHRINK/refCs_data_3.txt b/tests/08_RI/rpa_scf_shrink_multik/refCs_data_3.txt similarity index 100% rename from tests/08_EXX/57_KP_RPA_SHRINK/refCs_data_3.txt rename to tests/08_RI/rpa_scf_shrink_multik/refCs_data_3.txt diff --git a/tests/08_EXX/57_KP_RPA_SHRINK/refCs_data_4.txt b/tests/08_RI/rpa_scf_shrink_multik/refCs_data_4.txt similarity index 100% rename from tests/08_EXX/57_KP_RPA_SHRINK/refCs_data_4.txt rename to tests/08_RI/rpa_scf_shrink_multik/refCs_data_4.txt diff --git a/tests/08_EXX/57_KP_RPA_SHRINK/refCs_shrinked_data_1.txt b/tests/08_RI/rpa_scf_shrink_multik/refCs_shrinked_data_1.txt similarity index 100% rename from tests/08_EXX/57_KP_RPA_SHRINK/refCs_shrinked_data_1.txt rename to tests/08_RI/rpa_scf_shrink_multik/refCs_shrinked_data_1.txt diff --git a/tests/08_EXX/57_KP_RPA_SHRINK/refCs_shrinked_data_2.txt b/tests/08_RI/rpa_scf_shrink_multik/refCs_shrinked_data_2.txt similarity index 100% rename from tests/08_EXX/57_KP_RPA_SHRINK/refCs_shrinked_data_2.txt rename to tests/08_RI/rpa_scf_shrink_multik/refCs_shrinked_data_2.txt diff --git a/tests/08_EXX/57_KP_RPA_SHRINK/refCs_shrinked_data_3.txt b/tests/08_RI/rpa_scf_shrink_multik/refCs_shrinked_data_3.txt similarity index 100% rename from tests/08_EXX/57_KP_RPA_SHRINK/refCs_shrinked_data_3.txt rename to tests/08_RI/rpa_scf_shrink_multik/refCs_shrinked_data_3.txt diff --git a/tests/08_EXX/57_KP_RPA_SHRINK/refCs_shrinked_data_4.txt b/tests/08_RI/rpa_scf_shrink_multik/refCs_shrinked_data_4.txt similarity index 100% rename from tests/08_EXX/57_KP_RPA_SHRINK/refCs_shrinked_data_4.txt rename to tests/08_RI/rpa_scf_shrink_multik/refCs_shrinked_data_4.txt diff --git a/tests/08_EXX/57_KP_RPA_SHRINK/refcoulomb_cut_1.txt b/tests/08_RI/rpa_scf_shrink_multik/refcoulomb_cut_1.txt similarity index 100% rename from tests/08_EXX/57_KP_RPA_SHRINK/refcoulomb_cut_1.txt rename to tests/08_RI/rpa_scf_shrink_multik/refcoulomb_cut_1.txt diff --git a/tests/08_EXX/57_KP_RPA_SHRINK/refcoulomb_cut_2.txt b/tests/08_RI/rpa_scf_shrink_multik/refcoulomb_cut_2.txt similarity index 100% rename from tests/08_EXX/57_KP_RPA_SHRINK/refcoulomb_cut_2.txt rename to tests/08_RI/rpa_scf_shrink_multik/refcoulomb_cut_2.txt diff --git a/tests/08_EXX/57_KP_RPA_SHRINK/refcoulomb_cut_3.txt b/tests/08_RI/rpa_scf_shrink_multik/refcoulomb_cut_3.txt similarity index 100% rename from tests/08_EXX/57_KP_RPA_SHRINK/refcoulomb_cut_3.txt rename to tests/08_RI/rpa_scf_shrink_multik/refcoulomb_cut_3.txt diff --git a/tests/08_EXX/57_KP_RPA_SHRINK/refcoulomb_cut_4.txt b/tests/08_RI/rpa_scf_shrink_multik/refcoulomb_cut_4.txt similarity index 100% rename from tests/08_EXX/57_KP_RPA_SHRINK/refcoulomb_cut_4.txt rename to tests/08_RI/rpa_scf_shrink_multik/refcoulomb_cut_4.txt diff --git a/tests/08_EXX/57_KP_RPA_SHRINK/refcoulomb_mat_1.txt b/tests/08_RI/rpa_scf_shrink_multik/refcoulomb_mat_1.txt similarity index 100% rename from tests/08_EXX/57_KP_RPA_SHRINK/refcoulomb_mat_1.txt rename to tests/08_RI/rpa_scf_shrink_multik/refcoulomb_mat_1.txt diff --git a/tests/08_EXX/57_KP_RPA_SHRINK/refcoulomb_mat_2.txt b/tests/08_RI/rpa_scf_shrink_multik/refcoulomb_mat_2.txt similarity index 100% rename from tests/08_EXX/57_KP_RPA_SHRINK/refcoulomb_mat_2.txt rename to tests/08_RI/rpa_scf_shrink_multik/refcoulomb_mat_2.txt diff --git a/tests/08_EXX/57_KP_RPA_SHRINK/refcoulomb_mat_3.txt b/tests/08_RI/rpa_scf_shrink_multik/refcoulomb_mat_3.txt similarity index 100% rename from tests/08_EXX/57_KP_RPA_SHRINK/refcoulomb_mat_3.txt rename to tests/08_RI/rpa_scf_shrink_multik/refcoulomb_mat_3.txt diff --git a/tests/08_EXX/57_KP_RPA_SHRINK/refcoulomb_mat_4.txt b/tests/08_RI/rpa_scf_shrink_multik/refcoulomb_mat_4.txt similarity index 100% rename from tests/08_EXX/57_KP_RPA_SHRINK/refcoulomb_mat_4.txt rename to tests/08_RI/rpa_scf_shrink_multik/refcoulomb_mat_4.txt diff --git a/tests/08_EXX/57_KP_RPA_SHRINK/refshrink_sinvS_1.txt b/tests/08_RI/rpa_scf_shrink_multik/refshrink_sinvS_1.txt similarity index 100% rename from tests/08_EXX/57_KP_RPA_SHRINK/refshrink_sinvS_1.txt rename to tests/08_RI/rpa_scf_shrink_multik/refshrink_sinvS_1.txt diff --git a/tests/08_EXX/57_KP_RPA_SHRINK/refshrink_sinvS_2.txt b/tests/08_RI/rpa_scf_shrink_multik/refshrink_sinvS_2.txt similarity index 100% rename from tests/08_EXX/57_KP_RPA_SHRINK/refshrink_sinvS_2.txt rename to tests/08_RI/rpa_scf_shrink_multik/refshrink_sinvS_2.txt diff --git a/tests/08_EXX/57_KP_RPA_SHRINK/refshrink_sinvS_3.txt b/tests/08_RI/rpa_scf_shrink_multik/refshrink_sinvS_3.txt similarity index 100% rename from tests/08_EXX/57_KP_RPA_SHRINK/refshrink_sinvS_3.txt rename to tests/08_RI/rpa_scf_shrink_multik/refshrink_sinvS_3.txt diff --git a/tests/08_EXX/57_KP_RPA_SHRINK/refshrink_sinvS_4.txt b/tests/08_RI/rpa_scf_shrink_multik/refshrink_sinvS_4.txt similarity index 100% rename from tests/08_EXX/57_KP_RPA_SHRINK/refshrink_sinvS_4.txt rename to tests/08_RI/rpa_scf_shrink_multik/refshrink_sinvS_4.txt diff --git a/tests/08_EXX/57_KP_RPA_SHRINK/result.ref b/tests/08_RI/rpa_scf_shrink_multik/result.ref similarity index 100% rename from tests/08_EXX/57_KP_RPA_SHRINK/result.ref rename to tests/08_RI/rpa_scf_shrink_multik/result.ref diff --git a/tests/08_RI/rt_tddft_pbe0_gamma/INPUT b/tests/08_RI/rt_tddft_pbe0_gamma/INPUT new file mode 100644 index 00000000000..b434a755ffe --- /dev/null +++ b/tests/08_RI/rt_tddft_pbe0_gamma/INPUT @@ -0,0 +1,47 @@ +INPUT_PARAMETERS +# general information +calculation md +esolver_type tddft +md_type nve +md_nstep 1 +estep_per_md 5 +td_dt 0.005 +md_tfirst 0 + + +# rt-TDDFT parameters +td_vext 1 # add time-dependent external potential +td_vext_dire 3 # direction along z +td_stype 2 # 2: hybrid gauge +td_ttype 3 # Heaviside type potential +td_tstart 1 # the step electric field starts +td_tend 10 # the step electric field ends +td_heavi_t0 3 +td_heavi_amp 0.05 + + +# print out current information +out_current 1 + + +suffix autotest +pseudo_dir ../../PP_ORB +orbital_dir ../../PP_ORB +basis_type lcao +gamma_only 0 + + +# electronic structure calculations +ecutwfc 5 +scf_nmax 50 +scf_thr 1e-6 +ks_solver scalapack_gvx + + +# charge mixing +mixing_type broyden +mixing_beta 0.7 +mixing_gg0 0.0 + +# functional +dft_functional pbe0 diff --git a/tests/08_RI/rt_tddft_pbe0_gamma/KPT b/tests/08_RI/rt_tddft_pbe0_gamma/KPT new file mode 100644 index 00000000000..e769af76382 --- /dev/null +++ b/tests/08_RI/rt_tddft_pbe0_gamma/KPT @@ -0,0 +1,4 @@ +K_POINTS +0 +Gamma +2 1 1 0 0 0 diff --git a/tests/08_RI/rt_tddft_pbe0_gamma/README b/tests/08_RI/rt_tddft_pbe0_gamma/README new file mode 100644 index 00000000000..af159ab7909 --- /dev/null +++ b/tests/08_RI/rt_tddft_pbe0_gamma/README @@ -0,0 +1 @@ +LR-TDDFT PBE0 on H2O, gamma_only, td_vext=1, out_current=1. diff --git a/tests/08_RI/rt_tddft_pbe0_gamma/STRU b/tests/08_RI/rt_tddft_pbe0_gamma/STRU new file mode 100644 index 00000000000..26d98c1effa --- /dev/null +++ b/tests/08_RI/rt_tddft_pbe0_gamma/STRU @@ -0,0 +1,22 @@ +ATOMIC_SPECIES +Si 28.085 Si_dojo_nsoc.upf + +NUMERICAL_ORBITAL +Si_dojo_6au_sz.orb + +LATTICE_CONSTANT +10.2 // add lattice constant + +LATTICE_VECTORS +0.0 0.5 0.5 +0.5 0.0 0.5 +0.5 0.5 0.0 + +ATOMIC_POSITIONS +Cartesian //Cartesian or Direct coordinate. + +Si // Element type +0.0 // magnetism +2 // number of atoms +0.00 0.00 0.00 0 0 0 +0.25 0.25 0.25 1 1 1 diff --git a/tests/08_RI/rt_tddft_pbe0_gamma/current_tot.txt.ref b/tests/08_RI/rt_tddft_pbe0_gamma/current_tot.txt.ref new file mode 100644 index 00000000000..7e4e7dc91da --- /dev/null +++ b/tests/08_RI/rt_tddft_pbe0_gamma/current_tot.txt.ref @@ -0,0 +1,6 @@ +1 -1.8473604992191180e-19 2.0801852529639785e-19 1.9848453033136227e-19 +2 2.2561065415232184e-07 -2.2561065578378673e-07 1.7131168034571031e-06 +3 6.7621981712630010e-07 -6.7621982930807363e-07 5.1323842206572918e-06 +4 8.9926990400359809e-07 -8.9926991573773239e-07 6.8177117779990930e-06 +5 8.9401239643724794e-07 -8.9401239614143905e-07 6.7627319534504960e-06 +6 8.8609994131450076e-07 -8.8609994098007574e-07 6.6814532248171548e-06 diff --git a/tests/08_RI/rt_tddft_pbe0_gamma/result.ref b/tests/08_RI/rt_tddft_pbe0_gamma/result.ref new file mode 100644 index 00000000000..01a5433056c --- /dev/null +++ b/tests/08_RI/rt_tddft_pbe0_gamma/result.ref @@ -0,0 +1,4 @@ +etotref -187.1839728992078 +etotperatomref -93.5919864496 +CompareCurrent_pass 0 +totaltimeref 2.40 diff --git a/tests/08_EXX/13_NO_KP_CAMPBEH/INPUT b/tests/08_RI/scf_campbeh_gamma/INPUT similarity index 100% rename from tests/08_EXX/13_NO_KP_CAMPBEH/INPUT rename to tests/08_RI/scf_campbeh_gamma/INPUT diff --git a/tests/08_EXX/10_KP_HF/KPT b/tests/08_RI/scf_campbeh_gamma/KPT similarity index 100% rename from tests/08_EXX/10_KP_HF/KPT rename to tests/08_RI/scf_campbeh_gamma/KPT diff --git a/tests/08_RI/scf_campbeh_gamma/README b/tests/08_RI/scf_campbeh_gamma/README new file mode 100644 index 00000000000..469d9e91aa5 --- /dev/null +++ b/tests/08_RI/scf_campbeh_gamma/README @@ -0,0 +1 @@ +scf CAM-PBEH on H2O, gamma_only, exx_hybrid_step=2. diff --git a/tests/08_EXX/08_KP_HSE_symm/STRU b/tests/08_RI/scf_campbeh_gamma/STRU similarity index 100% rename from tests/08_EXX/08_KP_HSE_symm/STRU rename to tests/08_RI/scf_campbeh_gamma/STRU diff --git a/tests/08_EXX/13_NO_KP_CAMPBEH/result.ref b/tests/08_RI/scf_campbeh_gamma/result.ref similarity index 100% rename from tests/08_EXX/13_NO_KP_CAMPBEH/result.ref rename to tests/08_RI/scf_campbeh_gamma/result.ref diff --git a/tests/08_EXX/10_KP_HF/INPUT b/tests/08_RI/scf_hf_multik/INPUT similarity index 100% rename from tests/08_EXX/10_KP_HF/INPUT rename to tests/08_RI/scf_hf_multik/INPUT diff --git a/tests/08_EXX/11_KP_PBE0/KPT b/tests/08_RI/scf_hf_multik/KPT similarity index 100% rename from tests/08_EXX/11_KP_PBE0/KPT rename to tests/08_RI/scf_hf_multik/KPT diff --git a/tests/08_RI/scf_hf_multik/README b/tests/08_RI/scf_hf_multik/README new file mode 100644 index 00000000000..038e3510cef --- /dev/null +++ b/tests/08_RI/scf_hf_multik/README @@ -0,0 +1 @@ +scf HF on H2O, multik, nspin=1, exx_separate_loop=1. diff --git a/tests/08_EXX/09_KP_HSE_comp/STRU b/tests/08_RI/scf_hf_multik/STRU similarity index 100% rename from tests/08_EXX/09_KP_HSE_comp/STRU rename to tests/08_RI/scf_hf_multik/STRU diff --git a/tests/08_EXX/10_KP_HF/result.ref b/tests/08_RI/scf_hf_multik/result.ref similarity index 100% rename from tests/08_EXX/10_KP_HF/result.ref rename to tests/08_RI/scf_hf_multik/result.ref diff --git a/tests/08_EXX/09_KP_HSE_comp/INPUT b/tests/08_RI/scf_hse_complex_multik/INPUT similarity index 100% rename from tests/08_EXX/09_KP_HSE_comp/INPUT rename to tests/08_RI/scf_hse_complex_multik/INPUT diff --git a/tests/08_EXX/13_NO_KP_CAMPBEH/KPT b/tests/08_RI/scf_hse_complex_multik/KPT similarity index 100% rename from tests/08_EXX/13_NO_KP_CAMPBEH/KPT rename to tests/08_RI/scf_hse_complex_multik/KPT diff --git a/tests/08_RI/scf_hse_complex_multik/README b/tests/08_RI/scf_hse_complex_multik/README new file mode 100644 index 00000000000..a7df327d69e --- /dev/null +++ b/tests/08_RI/scf_hse_complex_multik/README @@ -0,0 +1 @@ +scf HSE on H2O, multik, complex, exx_separate_loop=1. diff --git a/tests/08_EXX/10_KP_HF/STRU b/tests/08_RI/scf_hse_complex_multik/STRU similarity index 100% rename from tests/08_EXX/10_KP_HF/STRU rename to tests/08_RI/scf_hse_complex_multik/STRU diff --git a/tests/08_EXX/09_KP_HSE_comp/result.ref b/tests/08_RI/scf_hse_complex_multik/result.ref similarity index 100% rename from tests/08_EXX/09_KP_HSE_comp/result.ref rename to tests/08_RI/scf_hse_complex_multik/result.ref diff --git a/tests/08_EXX/01_GO_S1_HSE/INPUT b/tests/08_RI/scf_hse_gamma/INPUT similarity index 100% rename from tests/08_EXX/01_GO_S1_HSE/INPUT rename to tests/08_RI/scf_hse_gamma/INPUT diff --git a/tests/08_EXX/01_GO_S1_HSE/KPT b/tests/08_RI/scf_hse_gamma/KPT similarity index 100% rename from tests/08_EXX/01_GO_S1_HSE/KPT rename to tests/08_RI/scf_hse_gamma/KPT diff --git a/tests/08_RI/scf_hse_gamma/README b/tests/08_RI/scf_hse_gamma/README new file mode 100644 index 00000000000..5b70e697019 --- /dev/null +++ b/tests/08_RI/scf_hse_gamma/README @@ -0,0 +1 @@ +scf HSE on H2O, gamma_only, nspin=1, exx_separate_loop=1. diff --git a/tests/08_RI/scf_hse_gamma/STRU b/tests/08_RI/scf_hse_gamma/STRU new file mode 100644 index 00000000000..6d4b6d0cf4c --- /dev/null +++ b/tests/08_RI/scf_hse_gamma/STRU @@ -0,0 +1,32 @@ +ATOMIC_SPECIES +H 1.000 H_ONCV_PBE-1.0.upf +O 1.000 O_ONCV_PBE-1.0.upf + +NUMERICAL_ORBITAL +H_gga_6au_60Ry_1s.orb +O_gga_6au_60Ry_1s1p.orb + + +LATTICE_CONSTANT +1.8897259886 + +LATTICE_VECTORS +10 0 0 +0 10 0 +0 0 10 + +ATOMIC_POSITIONS +Cartesian # Cartesian(Unit is LATTICE_CONSTANT) + +H +0.0 +2 +0.9793 0.0422 0 1 1 1 +-0.2866 0.9373 0 1 1 1 + + +O +0.0 +1 +0 0 0 1 1 1 + diff --git a/tests/08_RI/scf_hse_gamma/result.ref b/tests/08_RI/scf_hse_gamma/result.ref new file mode 100644 index 00000000000..7083a7f4d94 --- /dev/null +++ b/tests/08_RI/scf_hse_gamma/result.ref @@ -0,0 +1,5 @@ +etotref -428.9943078090022368 +etotperatomref -142.9981026030 +totalforceref 30.937794 +totalstressref 256.784504 +totaltimeref 2.76 diff --git a/tests/08_EXX/04_GO_S1_HSE_loop0/INPUT b/tests/08_RI/scf_hse_loop0_gamma/INPUT similarity index 100% rename from tests/08_EXX/04_GO_S1_HSE_loop0/INPUT rename to tests/08_RI/scf_hse_loop0_gamma/INPUT diff --git a/tests/08_EXX/02_GO_S2_HSE/KPT b/tests/08_RI/scf_hse_loop0_gamma/KPT similarity index 100% rename from tests/08_EXX/02_GO_S2_HSE/KPT rename to tests/08_RI/scf_hse_loop0_gamma/KPT diff --git a/tests/08_RI/scf_hse_loop0_gamma/README b/tests/08_RI/scf_hse_loop0_gamma/README new file mode 100644 index 00000000000..d2ed0527836 --- /dev/null +++ b/tests/08_RI/scf_hse_loop0_gamma/README @@ -0,0 +1 @@ +scf HSE on H2O, gamma_only, exx_separate_loop=0. diff --git a/tests/08_EXX/04_GO_S1_HSE_loop0/STRU b/tests/08_RI/scf_hse_loop0_gamma/STRU similarity index 100% rename from tests/08_EXX/04_GO_S1_HSE_loop0/STRU rename to tests/08_RI/scf_hse_loop0_gamma/STRU diff --git a/tests/08_RI/scf_hse_loop0_gamma/result.ref b/tests/08_RI/scf_hse_loop0_gamma/result.ref new file mode 100644 index 00000000000..0be692dd436 --- /dev/null +++ b/tests/08_RI/scf_hse_loop0_gamma/result.ref @@ -0,0 +1,3 @@ +etotref -428.9947050431449 +etotperatomref -142.9982350144 +totaltimeref 2.02 diff --git a/tests/08_RI/scf_hse_soc_symm_multik/INPUT b/tests/08_RI/scf_hse_soc_symm_multik/INPUT new file mode 100644 index 00000000000..042748c8050 --- /dev/null +++ b/tests/08_RI/scf_hse_soc_symm_multik/INPUT @@ -0,0 +1,32 @@ +INPUT_PARAMETERS +suffix autotest +pseudo_dir ../../PP_ORB +orbital_dir ../../PP_ORB + +calculation scf +basis_type lcao +gamma_only 0 + +ecutwfc 5 +scf_thr 1e-7 + +smearing_method gaussian +smearing_sigma 0.02 +mixing_type broyden +mixing_beta 0.15 + +symmetry 1 +symmetry_prec 1e-5 + +nspin 4 +lspinorb 1 +cal_force 1 +cal_stress 1 + +dft_functional hse +exx_separate_loop 0 +exx_hybrid_step 10 +exx_pca_threshold 1e-1 +exx_c_threshold 1e-1 +exx_v_threshold 1 +exx_dm_threshold 1e-2 diff --git a/tests/08_EXX/15_KP_HSE_SOC_symm/KPT b/tests/08_RI/scf_hse_soc_symm_multik/KPT similarity index 100% rename from tests/08_EXX/15_KP_HSE_SOC_symm/KPT rename to tests/08_RI/scf_hse_soc_symm_multik/KPT diff --git a/tests/08_RI/scf_hse_soc_symm_multik/README b/tests/08_RI/scf_hse_soc_symm_multik/README new file mode 100644 index 00000000000..ba0fb41a02c --- /dev/null +++ b/tests/08_RI/scf_hse_soc_symm_multik/README @@ -0,0 +1 @@ +scf HSE on H2O, multik, SOC nspin=4, symmetry=1, exx_separate_loop=0. diff --git a/tests/08_RI/scf_hse_soc_symm_multik/STRU b/tests/08_RI/scf_hse_soc_symm_multik/STRU new file mode 100644 index 00000000000..636a8286873 --- /dev/null +++ b/tests/08_RI/scf_hse_soc_symm_multik/STRU @@ -0,0 +1,22 @@ +ATOMIC_SPECIES +Fe 55.845 Fe.upf + +NUMERICAL_ORBITAL +Fe_gga_6au_100Ry_2s1p1d.orb + +LATTICE_CONSTANT +2.5 // (simple hexagonal, a=2.5, c=4.0, uniaxial C_6 axis along z) + +LATTICE_VECTORS + -1.4332500000 1.4332500000 1.4332500000 + 1.4332500000 -1.4332500000 1.4332500000 + 1.4332500000 1.4332500000 -1.4332500000 + +ATOMIC_POSITIONS +Direct + +Fe +0.0 +1 +0.0000000000 0.0000000000 0.0000000000 1 1 1 mag 0 0 2.2 + diff --git a/tests/08_RI/scf_hse_soc_symm_multik/result.ref b/tests/08_RI/scf_hse_soc_symm_multik/result.ref new file mode 100644 index 00000000000..38a234055c5 --- /dev/null +++ b/tests/08_RI/scf_hse_soc_symm_multik/result.ref @@ -0,0 +1,9 @@ +etotref -2926.9134396250428836 +etotperatomref -2926.9134396250 +totalforceref 0.000000 +totalstressref 3851.624753 +pointgroupref O_h +spacegroupref O_h +nksibzref 3 +magpointgroupref C_4h +totaltimeref 5.20 diff --git a/tests/08_RI/scf_hse_soc_symm_multik/threshold b/tests/08_RI/scf_hse_soc_symm_multik/threshold new file mode 100644 index 00000000000..7e4dc2d7e73 --- /dev/null +++ b/tests/08_RI/scf_hse_soc_symm_multik/threshold @@ -0,0 +1,18 @@ +# Only the energy tolerance needs overriding; force/stress/fatal use the +# global defaults (1e-4 / 1e-3 / 1). +# +# This deliberately-minimal metallic EXX+SOC case (ecutwfc=5, anisotropic +# 2x2x1 k-mesh, EXX updated every step) never reaches a stable SCF fixed +# point and stops mid-oscillation, so the endpoint is sensitive to tiny +# numerical seeds. np=1 is bit-identical run-to-run; under MPI the EXX/LibRI +# reduction order shifts the final energy by up to ~9e-7 eV at np=4 +# (measured, three repeats against the np=1 reference). Stress stays within +# 2.1e-4 and the force is exactly zero by symmetry. The discrete +# symmetry/magnetic-group assertions (O_h / C_4h / nksibz=3), the real +# purpose of this test, remain exact. +# +# The reference was regenerated after merging upstream PR #8017, which +# applies LibXC vrho down to 1e-10 (QE convention): the shifted vxc seed +# moves this non-converged trajectory to a different endpoint (~0.11 eV), +# while converged HSE cases shift only by ~1e-5 eV. +threshold 0.00001 diff --git a/tests/08_EXX/02_GO_S2_HSE/INPUT b/tests/08_RI/scf_hse_spin2_gamma/INPUT similarity index 100% rename from tests/08_EXX/02_GO_S2_HSE/INPUT rename to tests/08_RI/scf_hse_spin2_gamma/INPUT diff --git a/tests/08_EXX/04_GO_S1_HSE_loop0/KPT b/tests/08_RI/scf_hse_spin2_gamma/KPT similarity index 100% rename from tests/08_EXX/04_GO_S1_HSE_loop0/KPT rename to tests/08_RI/scf_hse_spin2_gamma/KPT diff --git a/tests/08_RI/scf_hse_spin2_gamma/README b/tests/08_RI/scf_hse_spin2_gamma/README new file mode 100644 index 00000000000..a1bb98e788b --- /dev/null +++ b/tests/08_RI/scf_hse_spin2_gamma/README @@ -0,0 +1 @@ +scf HSE on H2O, gamma_only, nspin=2, exx_separate_loop=1. diff --git a/tests/08_EXX/02_GO_S2_HSE/STRU b/tests/08_RI/scf_hse_spin2_gamma/STRU similarity index 100% rename from tests/08_EXX/02_GO_S2_HSE/STRU rename to tests/08_RI/scf_hse_spin2_gamma/STRU diff --git a/tests/08_RI/scf_hse_spin2_gamma/result.ref b/tests/08_RI/scf_hse_spin2_gamma/result.ref new file mode 100644 index 00000000000..29aa06b0a78 --- /dev/null +++ b/tests/08_RI/scf_hse_spin2_gamma/result.ref @@ -0,0 +1,5 @@ +etotref -448.7146817184075758 +etotperatomref -149.5715605728 +totalforceref 33.612717 +totalstressref 825.550032 +totaltimeref 2.91 diff --git a/tests/08_EXX/03_KP_S4_HSE/INPUT b/tests/08_RI/scf_hse_spin4_multik/INPUT similarity index 100% rename from tests/08_EXX/03_KP_S4_HSE/INPUT rename to tests/08_RI/scf_hse_spin4_multik/INPUT diff --git a/tests/08_RI/scf_hse_spin4_multik/KPT b/tests/08_RI/scf_hse_spin4_multik/KPT new file mode 100644 index 00000000000..e769af76382 --- /dev/null +++ b/tests/08_RI/scf_hse_spin4_multik/KPT @@ -0,0 +1,4 @@ +K_POINTS +0 +Gamma +2 1 1 0 0 0 diff --git a/tests/08_RI/scf_hse_spin4_multik/README b/tests/08_RI/scf_hse_spin4_multik/README new file mode 100644 index 00000000000..6598f26d0b6 --- /dev/null +++ b/tests/08_RI/scf_hse_spin4_multik/README @@ -0,0 +1 @@ +scf HSE on H2O, multik, nspin=4, exx_separate_loop=1. diff --git a/tests/08_EXX/03_KP_S4_HSE/STRU b/tests/08_RI/scf_hse_spin4_multik/STRU similarity index 100% rename from tests/08_EXX/03_KP_S4_HSE/STRU rename to tests/08_RI/scf_hse_spin4_multik/STRU diff --git a/tests/08_RI/scf_hse_spin4_multik/result.ref b/tests/08_RI/scf_hse_spin4_multik/result.ref new file mode 100644 index 00000000000..43f6d7fdd86 --- /dev/null +++ b/tests/08_RI/scf_hse_spin4_multik/result.ref @@ -0,0 +1,5 @@ +etotref -448.3614743110510403 +etotperatomref -149.4538247704 +totalforceref 33.668875 +totalstressref 813.501361 +totaltimeref 3.18 diff --git a/tests/08_EXX/08_KP_HSE_symm/INPUT b/tests/08_RI/scf_hse_symm_multik/INPUT similarity index 100% rename from tests/08_EXX/08_KP_HSE_symm/INPUT rename to tests/08_RI/scf_hse_symm_multik/INPUT diff --git a/tests/01_PW/071_PW_CR_move/KPT b/tests/08_RI/scf_hse_symm_multik/KPT similarity index 100% rename from tests/01_PW/071_PW_CR_move/KPT rename to tests/08_RI/scf_hse_symm_multik/KPT diff --git a/tests/08_RI/scf_hse_symm_multik/README b/tests/08_RI/scf_hse_symm_multik/README new file mode 100644 index 00000000000..360e3808f51 --- /dev/null +++ b/tests/08_RI/scf_hse_symm_multik/README @@ -0,0 +1 @@ +scf HSE on H2O, multik, symmetry=1, exx_separate_loop=1. diff --git a/tests/08_EXX/11_KP_PBE0/STRU b/tests/08_RI/scf_hse_symm_multik/STRU similarity index 100% rename from tests/08_EXX/11_KP_PBE0/STRU rename to tests/08_RI/scf_hse_symm_multik/STRU diff --git a/tests/08_EXX/08_KP_HSE_symm/result.ref b/tests/08_RI/scf_hse_symm_multik/result.ref similarity index 100% rename from tests/08_EXX/08_KP_HSE_symm/result.ref rename to tests/08_RI/scf_hse_symm_multik/result.ref diff --git a/tests/08_EXX/08_KP_HSE_symm/threshold b/tests/08_RI/scf_hse_symm_multik/threshold similarity index 100% rename from tests/08_EXX/08_KP_HSE_symm/threshold rename to tests/08_RI/scf_hse_symm_multik/threshold diff --git a/tests/08_EXX/12_KP_OXC/INPUT b/tests/08_RI/scf_out_xc_multik/INPUT similarity index 100% rename from tests/08_EXX/12_KP_OXC/INPUT rename to tests/08_RI/scf_out_xc_multik/INPUT diff --git a/tests/08_RI/scf_out_xc_multik/KPT b/tests/08_RI/scf_out_xc_multik/KPT new file mode 100644 index 00000000000..e769af76382 --- /dev/null +++ b/tests/08_RI/scf_out_xc_multik/KPT @@ -0,0 +1,4 @@ +K_POINTS +0 +Gamma +2 1 1 0 0 0 diff --git a/tests/08_RI/scf_out_xc_multik/README b/tests/08_RI/scf_out_xc_multik/README new file mode 100644 index 00000000000..f11eb37f242 --- /dev/null +++ b/tests/08_RI/scf_out_xc_multik/README @@ -0,0 +1 @@ +scf HSE on H2O, multik, out_mat_xc=1, out_eband_terms=1. diff --git a/tests/08_EXX/12_KP_OXC/STRU b/tests/08_RI/scf_out_xc_multik/STRU similarity index 100% rename from tests/08_EXX/12_KP_OXC/STRU rename to tests/08_RI/scf_out_xc_multik/STRU diff --git a/tests/08_EXX/12_KP_OXC/kinetic_out.ref b/tests/08_RI/scf_out_xc_multik/kinetic_out.ref similarity index 100% rename from tests/08_EXX/12_KP_OXC/kinetic_out.ref rename to tests/08_RI/scf_out_xc_multik/kinetic_out.ref diff --git a/tests/08_EXX/12_KP_OXC/result.ref b/tests/08_RI/scf_out_xc_multik/result.ref similarity index 100% rename from tests/08_EXX/12_KP_OXC/result.ref rename to tests/08_RI/scf_out_xc_multik/result.ref diff --git a/tests/08_EXX/12_KP_OXC/vhartree_out.ref b/tests/08_RI/scf_out_xc_multik/vhartree_out.ref similarity index 100% rename from tests/08_EXX/12_KP_OXC/vhartree_out.ref rename to tests/08_RI/scf_out_xc_multik/vhartree_out.ref diff --git a/tests/08_EXX/12_KP_OXC/vpp_local_out.ref b/tests/08_RI/scf_out_xc_multik/vpp_local_out.ref similarity index 100% rename from tests/08_EXX/12_KP_OXC/vpp_local_out.ref rename to tests/08_RI/scf_out_xc_multik/vpp_local_out.ref diff --git a/tests/08_EXX/12_KP_OXC/vpp_nonlocal_out.ref b/tests/08_RI/scf_out_xc_multik/vpp_nonlocal_out.ref similarity index 100% rename from tests/08_EXX/12_KP_OXC/vpp_nonlocal_out.ref rename to tests/08_RI/scf_out_xc_multik/vpp_nonlocal_out.ref diff --git a/tests/08_EXX/12_KP_OXC/vxc_out.ref b/tests/08_RI/scf_out_xc_multik/vxc_out.ref similarity index 100% rename from tests/08_EXX/12_KP_OXC/vxc_out.ref rename to tests/08_RI/scf_out_xc_multik/vxc_out.ref diff --git a/tests/08_EXX/12_KP_OXC/vxck2_nao.txt.ref b/tests/08_RI/scf_out_xc_multik/vxck2_nao.txt.ref similarity index 100% rename from tests/08_EXX/12_KP_OXC/vxck2_nao.txt.ref rename to tests/08_RI/scf_out_xc_multik/vxck2_nao.txt.ref diff --git a/tests/08_EXX/11_KP_PBE0/INPUT b/tests/08_RI/scf_pbe0_multik/INPUT similarity index 100% rename from tests/08_EXX/11_KP_PBE0/INPUT rename to tests/08_RI/scf_pbe0_multik/INPUT diff --git a/tests/08_RI/scf_pbe0_multik/KPT b/tests/08_RI/scf_pbe0_multik/KPT new file mode 100644 index 00000000000..eb3bd8bfb96 --- /dev/null +++ b/tests/08_RI/scf_pbe0_multik/KPT @@ -0,0 +1,4 @@ +K_POINTS +0 +Gamma +3 1 1 0 0 0 diff --git a/tests/08_RI/scf_pbe0_multik/README b/tests/08_RI/scf_pbe0_multik/README new file mode 100644 index 00000000000..3a708e459a3 --- /dev/null +++ b/tests/08_RI/scf_pbe0_multik/README @@ -0,0 +1 @@ +scf PBE0 on H2O, multik, nspin=1, exx_separate_loop=1. diff --git a/tests/08_EXX/13_NO_KP_CAMPBEH/STRU b/tests/08_RI/scf_pbe0_multik/STRU similarity index 100% rename from tests/08_EXX/13_NO_KP_CAMPBEH/STRU rename to tests/08_RI/scf_pbe0_multik/STRU diff --git a/tests/08_EXX/11_KP_PBE0/result.ref b/tests/08_RI/scf_pbe0_multik/result.ref similarity index 100% rename from tests/08_EXX/11_KP_PBE0/result.ref rename to tests/08_RI/scf_pbe0_multik/result.ref diff --git a/tests/09_DeePKS/CMakeLists.txt b/tests/09_DeePKS/CMakeLists.txt index 11c9b0cd9df..54fc0f9dc7f 100644 --- a/tests/09_DeePKS/CMakeLists.txt +++ b/tests/09_DeePKS/CMakeLists.txt @@ -11,7 +11,7 @@ if(ENABLE_ASAN) else() add_test( NAME 09_DeePKS - COMMAND ${BASH} ../integrate/Autotest.sh -a ${ABACUS_BIN_PATH} -n 4 + COMMAND ${BASH} ../integrate/Autotest.sh -a ${ABACUS_BIN_PATH} -n 4 -j auto WORKING_DIRECTORY ${ABACUS_TEST_DIR}/09_DeePKS ) endif() diff --git a/tests/10_others/CMakeLists.txt b/tests/10_others/CMakeLists.txt index d0949e095f8..fe5905ba4b8 100644 --- a/tests/10_others/CMakeLists.txt +++ b/tests/10_others/CMakeLists.txt @@ -10,7 +10,7 @@ if(ENABLE_ASAN) else() add_test( NAME 10_others - COMMAND ${BASH} ../integrate/Autotest.sh -a ${ABACUS_BIN_PATH} -n 4 + COMMAND ${BASH} ../integrate/Autotest.sh -a ${ABACUS_BIN_PATH} -n 4 -j auto WORKING_DIRECTORY ${ABACUS_TEST_DIR}/10_others ) endif() diff --git a/tests/11_PW_GPU/CMakeLists.txt b/tests/11_PW_GPU/CMakeLists.txt index 1da426aca83..ff4d3b2c527 100644 --- a/tests/11_PW_GPU/CMakeLists.txt +++ b/tests/11_PW_GPU/CMakeLists.txt @@ -10,7 +10,7 @@ if(ENABLE_ASAN) else() add_test( NAME 11_PW_GPU - COMMAND ${BASH} ../integrate/Autotest.sh -f CASES_GPU.txt -a ${ABACUS_BIN_PATH} -n 4 + COMMAND ${BASH} ../integrate/Autotest.sh -f CASES_GPU.txt -a ${ABACUS_BIN_PATH} -n 4 -j auto WORKING_DIRECTORY ${ABACUS_TEST_DIR}/11_PW_GPU ) endif() diff --git a/tests/11_PW_GPU/scf_out_wf/result.ref b/tests/11_PW_GPU/scf_out_wf/result.ref index 641b31e2cff..dd4a5ac6920 100644 --- a/tests/11_PW_GPU/scf_out_wf/result.ref +++ b/tests/11_PW_GPU/scf_out_wf/result.ref @@ -1,11 +1,11 @@ etotref -211.8032873347885 etotperatomref -105.9016436674 -Max_wfc_1 0.6619 -Max_wfc_2 0.7295 -Max_wfc_3 0.2296 -Max_wfc_4 0.5042 -Max_wfc_5 0.3319 -Max_wfc_6 0.8765 +Max_wfc_1 0.9527 +Max_wfc_2 0.3675 +Max_wfc_3 0.3351 +Max_wfc_4 0.3825 +Max_wfc_5 0.4409 +Max_wfc_6 0.4752 pointgroupref T_d spacegroupref O_h nksibzref 3 diff --git a/tests/15_rtTDDFT_GPU/12_NO_re_TDDFT_GPU/restart/Restart_td.txt b/tests/15_rtTDDFT_GPU/12_NO_re_TDDFT_GPU/restart/Restart_td.txt index 4bcfc2a96fa..81128e41c60 100644 --- a/tests/15_rtTDDFT_GPU/12_NO_re_TDDFT_GPU/restart/Restart_td.txt +++ b/tests/15_rtTDDFT_GPU/12_NO_re_TDDFT_GPU/restart/Restart_td.txt @@ -1,3 +1,6 @@ -5 -0 0 0 -0 0 0 \ No newline at end of file +# Hartree atomic units; step indices start at 0. +# gauge: 0=length, 1=velocity, 2=hybrid; source: 0=field, 1=file +# step left_step(-1=none) gauge dt source +5 -1 0 2.0670686668280682 0 +0 0 0 # A_left: x y z +0 0 0 # A_right: x y z diff --git a/tests/15_rtTDDFT_GPU/19_NO_SUPERSINE_TDDFT_GPU/vector_pot.txt.ref b/tests/15_rtTDDFT_GPU/19_NO_SUPERSINE_TDDFT_GPU/vector_pot.txt.ref index 90a39b97af9..563b1e1dcd9 100644 --- a/tests/15_rtTDDFT_GPU/19_NO_SUPERSINE_TDDFT_GPU/vector_pot.txt.ref +++ b/tests/15_rtTDDFT_GPU/19_NO_SUPERSINE_TDDFT_GPU/vector_pot.txt.ref @@ -1,4 +1,4 @@ -#istep A_x A_y A_z +#istep A_x A_y A_z # propagation values, Hartree atomic units 1 0.000000e+00 0.000000e+00 0.000000e+00 2 4.096138e-03 -2.078722e-02 0.000000e+00 3 4.375154e-03 -5.368872e-02 0.000000e+00 diff --git a/tests/15_rtTDDFT_GPU/20_NO_MIXED_EFIELD_TDDFT_GPU/vector_pot.txt.ref b/tests/15_rtTDDFT_GPU/20_NO_MIXED_EFIELD_TDDFT_GPU/vector_pot.txt.ref index c863f6c0a6a..fe7ac60c596 100644 --- a/tests/15_rtTDDFT_GPU/20_NO_MIXED_EFIELD_TDDFT_GPU/vector_pot.txt.ref +++ b/tests/15_rtTDDFT_GPU/20_NO_MIXED_EFIELD_TDDFT_GPU/vector_pot.txt.ref @@ -1,4 +1,4 @@ -#istep A_x A_y A_z +#istep A_x A_y A_z # propagation values, Hartree atomic units 1 -4.438787e-04 -1.062812e-04 -3.472515e-05 2 -1.236916e-03 -5.069838e-05 -3.925135e-04 3 -3.258834e-03 -6.966863e-04 -9.687700e-04 diff --git a/tests/15_rtTDDFT_GPU/21_PW_len_TDDFT_GPU/INPUT b/tests/15_rtTDDFT_GPU/21_PW_len_TDDFT_GPU/INPUT new file mode 100644 index 00000000000..6c03adf451f --- /dev/null +++ b/tests/15_rtTDDFT_GPU/21_PW_len_TDDFT_GPU/INPUT @@ -0,0 +1,51 @@ +INPUT_PARAMETERS + +suffix autotest +pseudo_dir ../../PP_ORB +esolver_type tddft +basis_type pw +calculation md +device gpu +precision double +ecutwfc 20 +ecutrho 80 +nbands 6 +nspin 1 +kpar 2 +symmetry 0 +init_wfc atomic+random +pw_seed 1 +ks_solver cg +pw_diag_thr 1e-14 +scf_thr 1e-12 +scf_nmax 100 +mixing_type broyden +mixing_beta 0.5 +mixing_gg0 0 +smearing_method gauss +smearing_sigma 0.002 +md_type nve +init_vel 1 +md_dt 0.01 +md_nstep 9 +cal_force 0 +cal_stress 0 +td_stype 0 +td_vext 1 +td_vext_dire 1 +td_ttype 0 +td_tstart 0 +td_tend 100 +td_gauss_freq 2 +td_gauss_phase 0 +td_gauss_sigma 0.04 +td_gauss_t0 4 +td_gauss_amp 0.001 +lin_solver bicgstab +lin_precond kinetic +lin_thr 1e-14 +lin_maxiter 100 +td_cn_init 0 +lin_reconstruct 0 +out_current 1 +out_level ie diff --git a/tests/15_rtTDDFT_GPU/21_PW_len_TDDFT_GPU/KPT b/tests/15_rtTDDFT_GPU/21_PW_len_TDDFT_GPU/KPT new file mode 100644 index 00000000000..23f1466d0f8 --- /dev/null +++ b/tests/15_rtTDDFT_GPU/21_PW_len_TDDFT_GPU/KPT @@ -0,0 +1,5 @@ +K_POINTS +2 +Direct +0.17 0.11 0.07 0.5 +-0.17 -0.11 -0.07 0.5 diff --git a/tests/15_rtTDDFT_GPU/21_PW_len_TDDFT_GPU/README b/tests/15_rtTDDFT_GPU/21_PW_len_TDDFT_GPU/README new file mode 100644 index 00000000000..69b906ef37a --- /dev/null +++ b/tests/15_rtTDDFT_GPU/21_PW_len_TDDFT_GPU/README @@ -0,0 +1 @@ +Length-gauge PW RT-TDDFT: self-consistent propagation and nonlocal current. diff --git a/tests/15_rtTDDFT_GPU/21_PW_len_TDDFT_GPU/STRU b/tests/15_rtTDDFT_GPU/21_PW_len_TDDFT_GPU/STRU new file mode 100644 index 00000000000..e059e5d6062 --- /dev/null +++ b/tests/15_rtTDDFT_GPU/21_PW_len_TDDFT_GPU/STRU @@ -0,0 +1,18 @@ +ATOMIC_SPECIES +Si 28 Si.pz-n-nc.UPF upf201 + +LATTICE_CONSTANT +10.2 + +LATTICE_VECTORS +0.0 0.5 0.5 +0.5 0.0 0.5 +0.5 0.5 0.0 + +ATOMIC_POSITIONS +Direct +Si +0.0 +2 +0.00 0.00 0.00 0 0 0 v 0 0 0 +0.25 0.25 0.25 0 0 0 v 0 0 0 diff --git a/tests/15_rtTDDFT_GPU/21_PW_len_TDDFT_GPU/current_tot.txt.ref b/tests/15_rtTDDFT_GPU/21_PW_len_TDDFT_GPU/current_tot.txt.ref new file mode 100644 index 00000000000..26fef0c8591 --- /dev/null +++ b/tests/15_rtTDDFT_GPU/21_PW_len_TDDFT_GPU/current_tot.txt.ref @@ -0,0 +1,9 @@ +1 -8.7372533313726819e-13 1.7628266998490296e-14 -1.8749696788461911e-13 +2 -5.1131148800069390e-09 9.3768918085396477e-12 -6.5784959639751502e-12 +3 -1.9381978870963455e-08 3.3429211065289635e-11 -2.7056414902621067e-11 +4 -3.9203858190765775e-08 7.0456786011446578e-11 -5.0922105945962130e-11 +5 -5.9267906815224327e-08 1.2505092797541114e-10 -6.8173851992747819e-11 +6 -7.4450138263918426e-08 2.0667322060979868e-10 -7.4987113370199227e-11 +7 -8.1044871563319266e-08 3.1929273005902725e-10 -7.4715103247141183e-11 +8 -7.6863270621976298e-08 4.5319603618971026e-10 -7.9047871820757897e-11 +9 -6.1419514746217234e-08 5.8561397596435925e-10 -1.0917389591425692e-10 diff --git a/tests/15_rtTDDFT_GPU/21_PW_len_TDDFT_GPU/result.ref b/tests/15_rtTDDFT_GPU/21_PW_len_TDDFT_GPU/result.ref new file mode 100644 index 00000000000..1cf2fbe01ca --- /dev/null +++ b/tests/15_rtTDDFT_GPU/21_PW_len_TDDFT_GPU/result.ref @@ -0,0 +1,4 @@ +etotref -251.7584934168609 +etotperatomref -125.8792467084 +CompareCurrent_pass 0 +totaltimeref 0.96 diff --git a/tests/15_rtTDDFT_GPU/22_PW_vel_TDDFT_GPU/INPUT b/tests/15_rtTDDFT_GPU/22_PW_vel_TDDFT_GPU/INPUT new file mode 100644 index 00000000000..49a14074d5d --- /dev/null +++ b/tests/15_rtTDDFT_GPU/22_PW_vel_TDDFT_GPU/INPUT @@ -0,0 +1,51 @@ +INPUT_PARAMETERS + +suffix autotest +pseudo_dir ../../PP_ORB +esolver_type tddft +basis_type pw +calculation md +device gpu +precision double +ecutwfc 20 +ecutrho 80 +nbands 6 +nspin 1 +kpar 2 +symmetry 0 +init_wfc atomic+random +pw_seed 1 +ks_solver cg +pw_diag_thr 1e-14 +scf_thr 1e-12 +scf_nmax 100 +mixing_type broyden +mixing_beta 0.5 +mixing_gg0 0 +smearing_method gauss +smearing_sigma 0.002 +md_type nve +init_vel 1 +md_dt 0.01 +md_nstep 9 +cal_force 0 +cal_stress 0 +td_stype 1 +td_vext 1 +td_vext_dire 1 +td_ttype 0 +td_tstart 0 +td_tend 100 +td_gauss_freq 2 +td_gauss_phase 0 +td_gauss_sigma 0.04 +td_gauss_t0 4 +td_gauss_amp 0.001 +lin_solver bicgstab +lin_precond kinetic +lin_thr 1e-14 +lin_maxiter 100 +td_cn_init 0 +lin_reconstruct 0 +out_current 1 +out_level ie diff --git a/tests/15_rtTDDFT_GPU/22_PW_vel_TDDFT_GPU/KPT b/tests/15_rtTDDFT_GPU/22_PW_vel_TDDFT_GPU/KPT new file mode 100644 index 00000000000..23f1466d0f8 --- /dev/null +++ b/tests/15_rtTDDFT_GPU/22_PW_vel_TDDFT_GPU/KPT @@ -0,0 +1,5 @@ +K_POINTS +2 +Direct +0.17 0.11 0.07 0.5 +-0.17 -0.11 -0.07 0.5 diff --git a/tests/15_rtTDDFT_GPU/22_PW_vel_TDDFT_GPU/README b/tests/15_rtTDDFT_GPU/22_PW_vel_TDDFT_GPU/README new file mode 100644 index 00000000000..c5075f6fffa --- /dev/null +++ b/tests/15_rtTDDFT_GPU/22_PW_vel_TDDFT_GPU/README @@ -0,0 +1 @@ +Velocity-gauge PW RT-TDDFT: vector-potential coupling and nonlocal current. diff --git a/tests/15_rtTDDFT_GPU/22_PW_vel_TDDFT_GPU/STRU b/tests/15_rtTDDFT_GPU/22_PW_vel_TDDFT_GPU/STRU new file mode 100644 index 00000000000..e059e5d6062 --- /dev/null +++ b/tests/15_rtTDDFT_GPU/22_PW_vel_TDDFT_GPU/STRU @@ -0,0 +1,18 @@ +ATOMIC_SPECIES +Si 28 Si.pz-n-nc.UPF upf201 + +LATTICE_CONSTANT +10.2 + +LATTICE_VECTORS +0.0 0.5 0.5 +0.5 0.0 0.5 +0.5 0.5 0.0 + +ATOMIC_POSITIONS +Direct +Si +0.0 +2 +0.00 0.00 0.00 0 0 0 v 0 0 0 +0.25 0.25 0.25 0 0 0 v 0 0 0 diff --git a/tests/15_rtTDDFT_GPU/22_PW_vel_TDDFT_GPU/current_tot.txt.ref b/tests/15_rtTDDFT_GPU/22_PW_vel_TDDFT_GPU/current_tot.txt.ref new file mode 100644 index 00000000000..3da56c70b31 --- /dev/null +++ b/tests/15_rtTDDFT_GPU/22_PW_vel_TDDFT_GPU/current_tot.txt.ref @@ -0,0 +1,9 @@ +1 1.0605063946060377e-12 -4.2509106434665706e-13 1.5087168526571182e-12 +2 1.3751229920540302e-07 1.6621406925090305e-11 -6.2358083121770329e-12 +3 3.0372208642324491e-07 6.4089539038915301e-11 -2.3316792632040393e-12 +4 4.8810917648953070e-07 1.6195288859878264e-10 2.7263499438301492e-11 +5 6.7777087857869071e-07 3.2484685380946662e-10 9.8423686153289534e-11 +6 8.5563526468913026e-07 5.6892549437259944e-10 2.3151902321144712e-10 +7 1.0040847301693847e-06 9.1229061982594634e-10 4.4941574231629083e-10 +8 1.1100465588291771e-06 1.3739801542497154e-09 7.7301888872485590e-10 +9 1.1668032707079688e-06 1.9781422929124190e-09 1.2187167552518205e-09 diff --git a/tests/15_rtTDDFT_GPU/22_PW_vel_TDDFT_GPU/result.ref b/tests/15_rtTDDFT_GPU/22_PW_vel_TDDFT_GPU/result.ref new file mode 100644 index 00000000000..6fbfd086bad --- /dev/null +++ b/tests/15_rtTDDFT_GPU/22_PW_vel_TDDFT_GPU/result.ref @@ -0,0 +1,4 @@ +etotref -251.7684286794544 +etotperatomref -125.8842143397 +CompareCurrent_pass 0 +totaltimeref 0.95 diff --git a/tests/15_rtTDDFT_GPU/23_PW_spin_TDDFT_GPU/INPUT b/tests/15_rtTDDFT_GPU/23_PW_spin_TDDFT_GPU/INPUT new file mode 100644 index 00000000000..7abf5e4f25d --- /dev/null +++ b/tests/15_rtTDDFT_GPU/23_PW_spin_TDDFT_GPU/INPUT @@ -0,0 +1,52 @@ +INPUT_PARAMETERS + +suffix autotest +pseudo_dir ../../PP_ORB +esolver_type tddft +basis_type pw +calculation md +device gpu +precision double +ecutwfc 20 +ecutrho 80 +nbands 6 +nspin 2 +kpar 2 +symmetry 0 +init_wfc atomic+random +pw_seed 1 +ks_solver cg +pw_diag_thr 1e-14 +scf_thr 1e-12 +scf_nmax 100 +mixing_type broyden +mixing_beta 0.5 +mixing_gg0 0 +smearing_method gauss +smearing_sigma 0.002 +md_type nve +init_vel 1 +md_dt 0.01 +md_nstep 9 +cal_force 0 +cal_stress 0 +td_stype 1 +td_vext 1 +td_vext_dire 1 +td_ttype 0 +td_tstart 0 +td_tend 100 +td_gauss_freq 2 +td_gauss_phase 0 +td_gauss_sigma 0.04 +td_gauss_t0 4 +td_gauss_amp 0.001 +lin_solver bicgstab +lin_precond kinetic +lin_thr 1e-14 +lin_maxiter 100 +td_cn_init 0 +lin_reconstruct 0 +out_current 1 +out_level ie +nupdown 2 diff --git a/tests/15_rtTDDFT_GPU/23_PW_spin_TDDFT_GPU/KPT b/tests/15_rtTDDFT_GPU/23_PW_spin_TDDFT_GPU/KPT new file mode 100644 index 00000000000..23f1466d0f8 --- /dev/null +++ b/tests/15_rtTDDFT_GPU/23_PW_spin_TDDFT_GPU/KPT @@ -0,0 +1,5 @@ +K_POINTS +2 +Direct +0.17 0.11 0.07 0.5 +-0.17 -0.11 -0.07 0.5 diff --git a/tests/15_rtTDDFT_GPU/23_PW_spin_TDDFT_GPU/README b/tests/15_rtTDDFT_GPU/23_PW_spin_TDDFT_GPU/README new file mode 100644 index 00000000000..1149585cac2 --- /dev/null +++ b/tests/15_rtTDDFT_GPU/23_PW_spin_TDDFT_GPU/README @@ -0,0 +1 @@ +Collinear spin-polarized PW RT-TDDFT: energy and current in the velocity gauge. diff --git a/tests/15_rtTDDFT_GPU/23_PW_spin_TDDFT_GPU/STRU b/tests/15_rtTDDFT_GPU/23_PW_spin_TDDFT_GPU/STRU new file mode 100644 index 00000000000..e059e5d6062 --- /dev/null +++ b/tests/15_rtTDDFT_GPU/23_PW_spin_TDDFT_GPU/STRU @@ -0,0 +1,18 @@ +ATOMIC_SPECIES +Si 28 Si.pz-n-nc.UPF upf201 + +LATTICE_CONSTANT +10.2 + +LATTICE_VECTORS +0.0 0.5 0.5 +0.5 0.0 0.5 +0.5 0.5 0.0 + +ATOMIC_POSITIONS +Direct +Si +0.0 +2 +0.00 0.00 0.00 0 0 0 v 0 0 0 +0.25 0.25 0.25 0 0 0 v 0 0 0 diff --git a/tests/15_rtTDDFT_GPU/23_PW_spin_TDDFT_GPU/current_tot.txt.ref b/tests/15_rtTDDFT_GPU/23_PW_spin_TDDFT_GPU/current_tot.txt.ref new file mode 100644 index 00000000000..69dafa2d46a --- /dev/null +++ b/tests/15_rtTDDFT_GPU/23_PW_spin_TDDFT_GPU/current_tot.txt.ref @@ -0,0 +1,9 @@ +1 1.5025535501917469e-12 -8.8904842960136839e-13 1.1617423411051124e-12 +2 1.3797666509972984e-07 5.8049036859975030e-11 -1.6103222081221969e-11 +3 3.0486346906799782e-07 1.5631405417288439e-10 -6.1436137939717484e-11 +4 4.9018109890949085e-07 2.9946478049434098e-10 -1.4569551218743284e-10 +5 6.8117146315149763e-07 4.7250332772220869e-10 -2.6397355494475357e-10 +6 8.6108570707769328e-07 6.5023539402896511e-10 -4.0485208008971540e-10 +7 1.0125550888168468e-06 8.0823915159476320e-10 -5.6077585677627105e-10 +8 1.1224997542301526e-06 9.3482755759430696e-10 -7.3456197539074472e-10 +9 1.1841395591464872e-06 1.0386878203305518e-09 -9.3862300152470329e-10 diff --git a/tests/15_rtTDDFT_GPU/23_PW_spin_TDDFT_GPU/result.ref b/tests/15_rtTDDFT_GPU/23_PW_spin_TDDFT_GPU/result.ref new file mode 100644 index 00000000000..f3418be7773 --- /dev/null +++ b/tests/15_rtTDDFT_GPU/23_PW_spin_TDDFT_GPU/result.ref @@ -0,0 +1,4 @@ +etotref -249.7637335161314 +etotperatomref -124.8818667581 +CompareCurrent_pass 0 +totaltimeref 1.95 diff --git a/tests/15_rtTDDFT_GPU/24_PW_cgs_TDDFT_GPU/INPUT b/tests/15_rtTDDFT_GPU/24_PW_cgs_TDDFT_GPU/INPUT new file mode 100644 index 00000000000..8d767e853ce --- /dev/null +++ b/tests/15_rtTDDFT_GPU/24_PW_cgs_TDDFT_GPU/INPUT @@ -0,0 +1,52 @@ +INPUT_PARAMETERS + +suffix autotest +pseudo_dir ../../PP_ORB +esolver_type tddft +basis_type pw +calculation md +device gpu +precision double +ecutwfc 20 +ecutrho 80 +nbands 6 +nspin 1 +kpar 2 +symmetry 0 +init_wfc atomic+random +pw_seed 1 +ks_solver cg +pw_diag_thr 1e-14 +scf_thr 1e-12 +scf_nmax 100 +mixing_type broyden +mixing_beta 0.5 +mixing_gg0 0 +smearing_method gauss +smearing_sigma 0.002 +md_type nve +init_vel 1 +md_dt 0.01 +md_nstep 9 +cal_force 0 +cal_stress 0 +td_stype 1 +td_vext 1 +td_vext_dire 1 +td_ttype 0 +td_tstart 0 +td_tend 100 +td_gauss_freq 2 +td_gauss_phase 0 +td_gauss_sigma 0.04 +td_gauss_t0 4 +td_gauss_amp 0.001 +lin_solver cgs +td_orthonormal lowdin +lin_precond kinetic +lin_thr 1e-14 +lin_maxiter 100 +td_cn_init 0 +lin_reconstruct 0 +out_current 1 +out_level ie diff --git a/tests/15_rtTDDFT_GPU/24_PW_cgs_TDDFT_GPU/KPT b/tests/15_rtTDDFT_GPU/24_PW_cgs_TDDFT_GPU/KPT new file mode 100644 index 00000000000..23f1466d0f8 --- /dev/null +++ b/tests/15_rtTDDFT_GPU/24_PW_cgs_TDDFT_GPU/KPT @@ -0,0 +1,5 @@ +K_POINTS +2 +Direct +0.17 0.11 0.07 0.5 +-0.17 -0.11 -0.07 0.5 diff --git a/tests/15_rtTDDFT_GPU/24_PW_cgs_TDDFT_GPU/README b/tests/15_rtTDDFT_GPU/24_PW_cgs_TDDFT_GPU/README new file mode 100644 index 00000000000..52939c4e083 --- /dev/null +++ b/tests/15_rtTDDFT_GPU/24_PW_cgs_TDDFT_GPU/README @@ -0,0 +1 @@ +Velocity-gauge PW RT-TDDFT with CGS and kinetic preconditioning: energy and current. diff --git a/tests/15_rtTDDFT_GPU/24_PW_cgs_TDDFT_GPU/STRU b/tests/15_rtTDDFT_GPU/24_PW_cgs_TDDFT_GPU/STRU new file mode 100644 index 00000000000..e059e5d6062 --- /dev/null +++ b/tests/15_rtTDDFT_GPU/24_PW_cgs_TDDFT_GPU/STRU @@ -0,0 +1,18 @@ +ATOMIC_SPECIES +Si 28 Si.pz-n-nc.UPF upf201 + +LATTICE_CONSTANT +10.2 + +LATTICE_VECTORS +0.0 0.5 0.5 +0.5 0.0 0.5 +0.5 0.5 0.0 + +ATOMIC_POSITIONS +Direct +Si +0.0 +2 +0.00 0.00 0.00 0 0 0 v 0 0 0 +0.25 0.25 0.25 0 0 0 v 0 0 0 diff --git a/tests/15_rtTDDFT_GPU/24_PW_cgs_TDDFT_GPU/current_tot.txt.ref b/tests/15_rtTDDFT_GPU/24_PW_cgs_TDDFT_GPU/current_tot.txt.ref new file mode 100644 index 00000000000..7a893ef3a7c --- /dev/null +++ b/tests/15_rtTDDFT_GPU/24_PW_cgs_TDDFT_GPU/current_tot.txt.ref @@ -0,0 +1,9 @@ +1 1.0605063946060377e-12 -4.2509106434665706e-13 1.5087168526571182e-12 +2 1.3751229921042474e-07 1.6621411528317294e-11 -6.2358041274252230e-12 +3 3.0372208643266064e-07 6.4089539875865666e-11 -2.3316796816792203e-12 +4 4.8810917649507548e-07 1.6195288755259469e-10 2.7263502786102940e-11 +5 6.7777087859469741e-07 3.2484685255404112e-10 9.8423691593466895e-11 +6 8.5563526471643574e-07 5.6892549479107466e-10 2.3151902614077338e-10 +7 1.0040847302052688e-06 9.1229062129060941e-10 4.4941574399019156e-10 +8 1.1100465588729078e-06 1.3739801550866658e-09 7.7301888788790556e-10 +9 1.1668032707619521e-06 1.9781422910292807e-09 1.2187167552518205e-09 diff --git a/tests/15_rtTDDFT_GPU/24_PW_cgs_TDDFT_GPU/result.ref b/tests/15_rtTDDFT_GPU/24_PW_cgs_TDDFT_GPU/result.ref new file mode 100644 index 00000000000..72bef06f26f --- /dev/null +++ b/tests/15_rtTDDFT_GPU/24_PW_cgs_TDDFT_GPU/result.ref @@ -0,0 +1,4 @@ +etotref -251.7684286794546 +etotperatomref -125.8842143397 +CompareCurrent_pass 0 +totaltimeref 0.96 diff --git a/tests/15_rtTDDFT_GPU/25_PW_noprec_TDDFT_GPU/INPUT b/tests/15_rtTDDFT_GPU/25_PW_noprec_TDDFT_GPU/INPUT new file mode 100644 index 00000000000..3d2eb8ac337 --- /dev/null +++ b/tests/15_rtTDDFT_GPU/25_PW_noprec_TDDFT_GPU/INPUT @@ -0,0 +1,51 @@ +INPUT_PARAMETERS + +suffix autotest +pseudo_dir ../../PP_ORB +esolver_type tddft +basis_type pw +calculation md +device gpu +precision double +ecutwfc 20 +ecutrho 80 +nbands 6 +nspin 1 +kpar 2 +symmetry 0 +init_wfc atomic+random +pw_seed 1 +ks_solver cg +pw_diag_thr 1e-14 +scf_thr 1e-12 +scf_nmax 100 +mixing_type broyden +mixing_beta 0.5 +mixing_gg0 0 +smearing_method gauss +smearing_sigma 0.002 +md_type nve +init_vel 1 +md_dt 0.01 +md_nstep 9 +cal_force 0 +cal_stress 0 +td_stype 1 +td_vext 1 +td_vext_dire 1 +td_ttype 0 +td_tstart 0 +td_tend 100 +td_gauss_freq 2 +td_gauss_phase 0 +td_gauss_sigma 0.04 +td_gauss_t0 4 +td_gauss_amp 0.001 +lin_solver bicgstab +lin_precond none +lin_thr 1e-14 +lin_maxiter 100 +td_cn_init 0 +lin_reconstruct 0 +out_current 1 +out_level ie diff --git a/tests/15_rtTDDFT_GPU/25_PW_noprec_TDDFT_GPU/KPT b/tests/15_rtTDDFT_GPU/25_PW_noprec_TDDFT_GPU/KPT new file mode 100644 index 00000000000..23f1466d0f8 --- /dev/null +++ b/tests/15_rtTDDFT_GPU/25_PW_noprec_TDDFT_GPU/KPT @@ -0,0 +1,5 @@ +K_POINTS +2 +Direct +0.17 0.11 0.07 0.5 +-0.17 -0.11 -0.07 0.5 diff --git a/tests/15_rtTDDFT_GPU/25_PW_noprec_TDDFT_GPU/README b/tests/15_rtTDDFT_GPU/25_PW_noprec_TDDFT_GPU/README new file mode 100644 index 00000000000..6e973ef58ce --- /dev/null +++ b/tests/15_rtTDDFT_GPU/25_PW_noprec_TDDFT_GPU/README @@ -0,0 +1 @@ +Velocity-gauge PW RT-TDDFT with BiCGSTAB and no preconditioning: energy and current. diff --git a/tests/15_rtTDDFT_GPU/25_PW_noprec_TDDFT_GPU/STRU b/tests/15_rtTDDFT_GPU/25_PW_noprec_TDDFT_GPU/STRU new file mode 100644 index 00000000000..e059e5d6062 --- /dev/null +++ b/tests/15_rtTDDFT_GPU/25_PW_noprec_TDDFT_GPU/STRU @@ -0,0 +1,18 @@ +ATOMIC_SPECIES +Si 28 Si.pz-n-nc.UPF upf201 + +LATTICE_CONSTANT +10.2 + +LATTICE_VECTORS +0.0 0.5 0.5 +0.5 0.0 0.5 +0.5 0.5 0.0 + +ATOMIC_POSITIONS +Direct +Si +0.0 +2 +0.00 0.00 0.00 0 0 0 v 0 0 0 +0.25 0.25 0.25 0 0 0 v 0 0 0 diff --git a/tests/15_rtTDDFT_GPU/25_PW_noprec_TDDFT_GPU/current_tot.txt.ref b/tests/15_rtTDDFT_GPU/25_PW_noprec_TDDFT_GPU/current_tot.txt.ref new file mode 100644 index 00000000000..d43e525c342 --- /dev/null +++ b/tests/15_rtTDDFT_GPU/25_PW_noprec_TDDFT_GPU/current_tot.txt.ref @@ -0,0 +1,9 @@ +1 1.0605063946060377e-12 -4.2509106434665706e-13 1.5087168526571182e-12 +2 1.3751229918102684e-07 1.6621418223920190e-11 -6.2358032904748610e-12 +3 3.0372208635775354e-07 6.4089552430121092e-11 -2.3316704752252389e-12 +4 4.8810917641807606e-07 1.6195290303617638e-10 2.7263507389329932e-11 +5 6.7777087850033127e-07 3.2484685694803053e-10 9.8423689082615799e-11 +6 8.5563526460271511e-07 5.6892548244605687e-10 2.3151901358651796e-10 +7 1.0040847300863172e-06 9.1229059388048510e-10 4.4941572390338285e-10 +8 1.1100465587529101e-06 1.3739801153315235e-09 7.7301886277939466e-10 +9 1.1668032706442560e-06 1.9781422383014078e-09 1.2187167200999052e-09 diff --git a/tests/15_rtTDDFT_GPU/25_PW_noprec_TDDFT_GPU/result.ref b/tests/15_rtTDDFT_GPU/25_PW_noprec_TDDFT_GPU/result.ref new file mode 100644 index 00000000000..44d87fd6970 --- /dev/null +++ b/tests/15_rtTDDFT_GPU/25_PW_noprec_TDDFT_GPU/result.ref @@ -0,0 +1,4 @@ +etotref -251.7684286794546 +etotperatomref -125.8842143397 +CompareCurrent_pass 0 +totaltimeref 1.20 diff --git a/tests/15_rtTDDFT_GPU/26_PW_single_TDDFT_GPU/INPUT b/tests/15_rtTDDFT_GPU/26_PW_single_TDDFT_GPU/INPUT new file mode 100644 index 00000000000..104fdc223bd --- /dev/null +++ b/tests/15_rtTDDFT_GPU/26_PW_single_TDDFT_GPU/INPUT @@ -0,0 +1,53 @@ +INPUT_PARAMETERS + +suffix autotest +pseudo_dir ../../PP_ORB +esolver_type tddft +basis_type pw +calculation md +device gpu +precision single +ecutwfc 8 +ecutrho 32 +nbands 2 +nspin 1 +kpar 2 +symmetry 0 +init_wfc atomic +pw_seed 1 +ks_solver cg +pw_diag_thr 1e-8 +scf_thr 1e-8 +scf_nmax 100 +mixing_type broyden +mixing_beta 0.5 +mixing_gg0 0 +smearing_method gauss +smearing_sigma 0.002 +md_type nve +init_vel 1 +md_dt 0.01 +md_nstep 9 +cal_force 0 +cal_stress 0 +td_stype 1 +td_vext 1 +td_vext_dire 1 +td_ttype 0 +td_tstart 0 +td_tend 100 +td_gauss_freq 2 +td_gauss_phase 0 +td_gauss_sigma 0.04 +td_gauss_t0 4 +td_gauss_amp 0.1 +lin_solver bicgstab +td_orthonormal newton_schulz +td_out_stat 1 +lin_precond kinetic +lin_thr 1e-7 +lin_maxiter 100 +td_cn_init 0 +lin_reconstruct 0 +out_current 1 +out_level ie diff --git a/tests/15_rtTDDFT_GPU/26_PW_single_TDDFT_GPU/KPT b/tests/15_rtTDDFT_GPU/26_PW_single_TDDFT_GPU/KPT new file mode 100644 index 00000000000..e769af76382 --- /dev/null +++ b/tests/15_rtTDDFT_GPU/26_PW_single_TDDFT_GPU/KPT @@ -0,0 +1,4 @@ +K_POINTS +0 +Gamma +2 1 1 0 0 0 diff --git a/tests/15_rtTDDFT_GPU/26_PW_single_TDDFT_GPU/README b/tests/15_rtTDDFT_GPU/26_PW_single_TDDFT_GPU/README new file mode 100644 index 00000000000..140375e8bdc --- /dev/null +++ b/tests/15_rtTDDFT_GPU/26_PW_single_TDDFT_GPU/README @@ -0,0 +1 @@ +Single-precision PW RT-TDDFT for H2 under an electric field: energy and current with kpar=2. diff --git a/tests/15_rtTDDFT_GPU/26_PW_single_TDDFT_GPU/STRU b/tests/15_rtTDDFT_GPU/26_PW_single_TDDFT_GPU/STRU new file mode 100644 index 00000000000..a8ca5da26fc --- /dev/null +++ b/tests/15_rtTDDFT_GPU/26_PW_single_TDDFT_GPU/STRU @@ -0,0 +1,18 @@ +ATOMIC_SPECIES +H 1 H.pz-vbc.UPF + +LATTICE_CONSTANT +40 + +LATTICE_VECTORS +1 0 0 +0 1 0 +0 0 1 + +ATOMIC_POSITIONS +Cartesian_au +H +0.0 +2 +19.3 20 20 0 0 0 v 0 0 0 +20.7 20 20 0 0 0 v 0 0 0 diff --git a/tests/15_rtTDDFT_GPU/26_PW_single_TDDFT_GPU/current_tot.txt.ref b/tests/15_rtTDDFT_GPU/26_PW_single_TDDFT_GPU/current_tot.txt.ref new file mode 100644 index 00000000000..b768122dc76 --- /dev/null +++ b/tests/15_rtTDDFT_GPU/26_PW_single_TDDFT_GPU/current_tot.txt.ref @@ -0,0 +1,9 @@ +1 -5.6024873629212379e-13 -3.0559021979570391e-13 2.8529001383503785e-13 +2 1.5215795428957790e-08 2.4738255888223650e-13 3.4182434660579021e-13 +3 3.3523581805638968e-08 1.4115357771515847e-12 4.5536907578025425e-13 +4 5.2988645620644091e-08 8.7311491370201108e-13 4.9864823381540191e-13 +5 7.1118054620455948e-08 1.0040821507573129e-12 3.9423753150913400e-13 +6 8.5259031038731338e-08 4.3655745685100556e-14 4.2547299017314801e-13 +7 9.3154805654194207e-08 0.0000000000000000e+00 4.4994408199272583e-13 +8 9.3486822152044622e-08 4.6566128730773927e-13 3.6639136169469564e-13 +9 8.5802006651647392e-08 1.1350493878126144e-12 3.1320723792305215e-13 diff --git a/tests/15_rtTDDFT_GPU/26_PW_single_TDDFT_GPU/result.ref b/tests/15_rtTDDFT_GPU/26_PW_single_TDDFT_GPU/result.ref new file mode 100644 index 00000000000..bf589c77c5e --- /dev/null +++ b/tests/15_rtTDDFT_GPU/26_PW_single_TDDFT_GPU/result.ref @@ -0,0 +1,4 @@ +etotref -28.58908359094873 +etotperatomref -14.2945417955 +CompareCurrent_pass 0 +totaltimeref 4.05 diff --git a/tests/15_rtTDDFT_GPU/26_PW_single_TDDFT_GPU/threshold b/tests/15_rtTDDFT_GPU/26_PW_single_TDDFT_GPU/threshold new file mode 100644 index 00000000000..8fcfcebb260 --- /dev/null +++ b/tests/15_rtTDDFT_GPU/26_PW_single_TDDFT_GPU/threshold @@ -0,0 +1 @@ +threshold 0.0001 diff --git a/tests/15_rtTDDFT_GPU/27_PW_gmres_recycle_TDDFT_GPU/INPUT b/tests/15_rtTDDFT_GPU/27_PW_gmres_recycle_TDDFT_GPU/INPUT new file mode 100644 index 00000000000..fea2ea9c2a7 --- /dev/null +++ b/tests/15_rtTDDFT_GPU/27_PW_gmres_recycle_TDDFT_GPU/INPUT @@ -0,0 +1,52 @@ +INPUT_PARAMETERS + +suffix autotest +pseudo_dir ../../PP_ORB +esolver_type tddft +basis_type pw +calculation md +device gpu +precision double +ecutwfc 20 +ecutrho 80 +nbands 6 +nspin 1 +kpar 2 +symmetry 0 +init_wfc atomic +pw_seed 1 +ks_solver cg +pw_diag_thr 1e-15 +scf_thr 1e-13 +scf_nmax 100 +mixing_type broyden +mixing_beta 0.5 +mixing_gg0 0 +smearing_method gauss +smearing_sigma 0.002 +md_type nve +init_vel 1 +md_dt 0.01 +md_nstep 11 +cal_force 0 +cal_stress 0 +td_stype 1 +td_vext 1 +td_vext_dire 1 +td_ttype 0 +td_tstart 0 +td_tend 100 +td_gauss_freq 2 +td_gauss_phase 0 +td_gauss_sigma 0.04 +td_gauss_t0 4 +td_gauss_amp 0.01 +lin_solver gmres +lin_precond kinetic_recycle +lin_thr 1e-11 +lin_maxiter 100 +out_current 1 +out_level ie +lin_gmres_restart 20 +td_cn_init 1 +lin_reconstruct 1 diff --git a/tests/15_rtTDDFT_GPU/27_PW_gmres_recycle_TDDFT_GPU/KPT b/tests/15_rtTDDFT_GPU/27_PW_gmres_recycle_TDDFT_GPU/KPT new file mode 100644 index 00000000000..03dfec74177 --- /dev/null +++ b/tests/15_rtTDDFT_GPU/27_PW_gmres_recycle_TDDFT_GPU/KPT @@ -0,0 +1,6 @@ +K_POINTS +3 +Direct +0.0 0.0 0.0 0.5 +0.17 0.11 0.07 0.25 +-0.17 -0.11 -0.07 0.25 diff --git a/tests/15_rtTDDFT_GPU/27_PW_gmres_recycle_TDDFT_GPU/README b/tests/15_rtTDDFT_GPU/27_PW_gmres_recycle_TDDFT_GPU/README new file mode 100644 index 00000000000..448c8ab9a8a --- /dev/null +++ b/tests/15_rtTDDFT_GPU/27_PW_gmres_recycle_TDDFT_GPU/README @@ -0,0 +1 @@ +Double-precision PW RT-TDDFT with GMRES(20), history preconditioning, CN initialization and residual reconstruction. Tests energy and current with three k points distributed over two pools (kpar=2). diff --git a/tests/15_rtTDDFT_GPU/27_PW_gmres_recycle_TDDFT_GPU/STRU b/tests/15_rtTDDFT_GPU/27_PW_gmres_recycle_TDDFT_GPU/STRU new file mode 100644 index 00000000000..e059e5d6062 --- /dev/null +++ b/tests/15_rtTDDFT_GPU/27_PW_gmres_recycle_TDDFT_GPU/STRU @@ -0,0 +1,18 @@ +ATOMIC_SPECIES +Si 28 Si.pz-n-nc.UPF upf201 + +LATTICE_CONSTANT +10.2 + +LATTICE_VECTORS +0.0 0.5 0.5 +0.5 0.0 0.5 +0.5 0.5 0.0 + +ATOMIC_POSITIONS +Direct +Si +0.0 +2 +0.00 0.00 0.00 0 0 0 v 0 0 0 +0.25 0.25 0.25 0 0 0 v 0 0 0 diff --git a/tests/15_rtTDDFT_GPU/27_PW_gmres_recycle_TDDFT_GPU/current_tot.txt.ref b/tests/15_rtTDDFT_GPU/27_PW_gmres_recycle_TDDFT_GPU/current_tot.txt.ref new file mode 100644 index 00000000000..4f4e4f67551 --- /dev/null +++ b/tests/15_rtTDDFT_GPU/27_PW_gmres_recycle_TDDFT_GPU/current_tot.txt.ref @@ -0,0 +1,11 @@ +1 4.2936108047451097e-13 -2.4637098567017647e-13 1.1469044666133506e-13 +2 1.3684983720511565e-06 8.1369010321866002e-11 -3.9066490646652952e-11 +3 3.0190089257789310e-06 3.1699090575680301e-10 -1.7795759597298859e-11 +4 4.8444606284476297e-06 8.1008643537411144e-10 1.3715835422850556e-10 +5 6.7137769288199204e-06 1.6376879822670157e-09 5.0720485712728539e-10 +6 8.4530928015672038e-06 2.8846612733306468e-09 1.1949993113429527e-09 +7 9.8830552413079795e-06 4.6476825547231378e-09 2.3164995244103979e-09 +8 1.0870499501461829e-05 7.0273996237572805e-09 3.9779609187124104e-09 +9 1.1344957566042592e-05 1.0147693859726738e-08 6.2631648888650860e-09 +10 1.1298352661174024e-05 1.4154858985090144e-08 9.2238949751129842e-09 +11 1.0783503747898042e-05 1.9160638454416877e-08 1.2861428365591574e-08 diff --git a/tests/15_rtTDDFT_GPU/27_PW_gmres_recycle_TDDFT_GPU/result.ref b/tests/15_rtTDDFT_GPU/27_PW_gmres_recycle_TDDFT_GPU/result.ref new file mode 100644 index 00000000000..067e455a734 --- /dev/null +++ b/tests/15_rtTDDFT_GPU/27_PW_gmres_recycle_TDDFT_GPU/result.ref @@ -0,0 +1,4 @@ +etotref -249.0536998379458 +etotperatomref -124.5268499190 +CompareCurrent_pass 0 +totaltimeref 1.33 diff --git a/tests/15_rtTDDFT_GPU/28_PW_gmres_subspace_TDDFT_GPU/INPUT b/tests/15_rtTDDFT_GPU/28_PW_gmres_subspace_TDDFT_GPU/INPUT new file mode 100644 index 00000000000..beb7e3ec4be --- /dev/null +++ b/tests/15_rtTDDFT_GPU/28_PW_gmres_subspace_TDDFT_GPU/INPUT @@ -0,0 +1,52 @@ +INPUT_PARAMETERS + +suffix autotest +pseudo_dir ../../PP_ORB +esolver_type tddft +basis_type pw +calculation md +device gpu +precision single +ecutwfc 8 +ecutrho 32 +nbands 2 +nspin 1 +kpar 2 +symmetry 0 +init_wfc atomic +pw_seed 1 +ks_solver cg +pw_diag_thr 1e-8 +scf_thr 1e-8 +scf_nmax 100 +mixing_type broyden +mixing_beta 0.5 +mixing_gg0 0 +smearing_method gauss +smearing_sigma 0.002 +md_type nve +init_vel 1 +md_dt 0.01 +md_nstep 9 +cal_force 0 +cal_stress 0 +td_stype 1 +td_vext 1 +td_vext_dire 1 +td_ttype 0 +td_tstart 0 +td_tend 100 +td_gauss_freq 2 +td_gauss_phase 0 +td_gauss_sigma 0.04 +td_gauss_t0 4 +td_gauss_amp 0.1 +lin_solver gmres +lin_precond kinetic_subspace +lin_thr 1e-6 +lin_maxiter 100 +out_current 1 +out_level ie +lin_gmres_restart 20 +td_cn_init 1 +lin_reconstruct 1 diff --git a/tests/15_rtTDDFT_GPU/28_PW_gmres_subspace_TDDFT_GPU/KPT b/tests/15_rtTDDFT_GPU/28_PW_gmres_subspace_TDDFT_GPU/KPT new file mode 100644 index 00000000000..e769af76382 --- /dev/null +++ b/tests/15_rtTDDFT_GPU/28_PW_gmres_subspace_TDDFT_GPU/KPT @@ -0,0 +1,4 @@ +K_POINTS +0 +Gamma +2 1 1 0 0 0 diff --git a/tests/15_rtTDDFT_GPU/28_PW_gmres_subspace_TDDFT_GPU/README b/tests/15_rtTDDFT_GPU/28_PW_gmres_subspace_TDDFT_GPU/README new file mode 100644 index 00000000000..ea4ac160044 --- /dev/null +++ b/tests/15_rtTDDFT_GPU/28_PW_gmres_subspace_TDDFT_GPU/README @@ -0,0 +1 @@ +Single-precision PW RT-TDDFT with GMRES(20), coarse-space preconditioning and CN initialization. Tests energy and current with kpar=2 and residual reconstruction subject to independent residual checks. diff --git a/tests/15_rtTDDFT_GPU/28_PW_gmres_subspace_TDDFT_GPU/STRU b/tests/15_rtTDDFT_GPU/28_PW_gmres_subspace_TDDFT_GPU/STRU new file mode 100644 index 00000000000..a8ca5da26fc --- /dev/null +++ b/tests/15_rtTDDFT_GPU/28_PW_gmres_subspace_TDDFT_GPU/STRU @@ -0,0 +1,18 @@ +ATOMIC_SPECIES +H 1 H.pz-vbc.UPF + +LATTICE_CONSTANT +40 + +LATTICE_VECTORS +1 0 0 +0 1 0 +0 0 1 + +ATOMIC_POSITIONS +Cartesian_au +H +0.0 +2 +19.3 20 20 0 0 0 v 0 0 0 +20.7 20 20 0 0 0 v 0 0 0 diff --git a/tests/15_rtTDDFT_GPU/28_PW_gmres_subspace_TDDFT_GPU/current_tot.txt.ref b/tests/15_rtTDDFT_GPU/28_PW_gmres_subspace_TDDFT_GPU/current_tot.txt.ref new file mode 100644 index 00000000000..93a436a452f --- /dev/null +++ b/tests/15_rtTDDFT_GPU/28_PW_gmres_subspace_TDDFT_GPU/current_tot.txt.ref @@ -0,0 +1,9 @@ +1 -5.6024873629212379e-13 -3.0559021979570391e-13 2.8529001383503785e-13 +2 1.5216937754303218e-08 1.8917489796876906e-13 3.3454128356424919e-13 +3 3.3524207537993790e-08 1.0622898116707801e-12 4.0079584096019969e-13 +4 5.2989307732786984e-08 1.3242242857813835e-12 3.7205438729870367e-13 +5 7.1119036874733864e-08 7.5669959187507626e-13 3.7364245031312751e-13 +6 8.5259147454053163e-08 -1.3096723705530166e-13 4.5739056986349169e-13 +7 9.3145310529507696e-08 -1.6007106751203536e-13 3.9358027947855590e-13 +8 9.3477523478213694e-08 1.4551915228366852e-13 3.0836133646516830e-13 +9 8.5794068581890316e-08 1.3242242857813835e-12 2.4052937419583032e-13 diff --git a/tests/15_rtTDDFT_GPU/28_PW_gmres_subspace_TDDFT_GPU/result.ref b/tests/15_rtTDDFT_GPU/28_PW_gmres_subspace_TDDFT_GPU/result.ref new file mode 100644 index 00000000000..99001e33e8f --- /dev/null +++ b/tests/15_rtTDDFT_GPU/28_PW_gmres_subspace_TDDFT_GPU/result.ref @@ -0,0 +1,4 @@ +etotref -28.58910155231029 +etotperatomref -14.2945507762 +CompareCurrent_pass 0 +totaltimeref 3.27 diff --git a/tests/15_rtTDDFT_GPU/28_PW_gmres_subspace_TDDFT_GPU/threshold b/tests/15_rtTDDFT_GPU/28_PW_gmres_subspace_TDDFT_GPU/threshold new file mode 100644 index 00000000000..8fcfcebb260 --- /dev/null +++ b/tests/15_rtTDDFT_GPU/28_PW_gmres_subspace_TDDFT_GPU/threshold @@ -0,0 +1 @@ +threshold 0.0001 diff --git a/tests/15_rtTDDFT_GPU/CASES_GPU.txt b/tests/15_rtTDDFT_GPU/CASES_GPU.txt index 44b6b4e89e8..be6849d29d7 100644 --- a/tests/15_rtTDDFT_GPU/CASES_GPU.txt +++ b/tests/15_rtTDDFT_GPU/CASES_GPU.txt @@ -16,3 +16,11 @@ 18_NO_hyb_TDDFT_GPU 19_NO_SUPERSINE_TDDFT_GPU 20_NO_MIXED_EFIELD_TDDFT_GPU +21_PW_len_TDDFT_GPU +22_PW_vel_TDDFT_GPU +23_PW_spin_TDDFT_GPU +24_PW_cgs_TDDFT_GPU +25_PW_noprec_TDDFT_GPU +26_PW_single_TDDFT_GPU +27_PW_gmres_recycle_TDDFT_GPU +28_PW_gmres_subspace_TDDFT_GPU diff --git a/tests/16_SDFT_GPU/001_PW_KG_100_GPU/refOnsager.txt b/tests/16_SDFT_GPU/001_PW_KG_100_GPU/refOnsager.txt index 4f850b65d2c..d2098e33a71 100644 --- a/tests/16_SDFT_GPU/001_PW_KG_100_GPU/refOnsager.txt +++ b/tests/16_SDFT_GPU/001_PW_KG_100_GPU/refOnsager.txt @@ -1,11 +1,11 @@ ## w(eV) sigma(Sm^-1) kappa(W(mK)^-1) L12/e(Am^-1) L22/e^2(Wm^-1) - 0.5 356691 626.131 -7.13544e+06 2.02056e+08 - 1.5 348656 606.514 -7.03014e+06 1.99209e+08 - 2.5 333032 569.396 -6.81899e+06 1.93563e+08 - 3.5 310656 518.557 -6.50176e+06 1.85201e+08 - 4.5 282683 458.689 -6.08031e+06 1.74236e+08 - 5.5 250512 394.607 -5.56125e+06 1.60839e+08 - 6.5 215751 330.615 -4.95931e+06 1.45316e+08 - 7.5 180190 270.122 -4.3001e+06 1.28208e+08 - 8.5 145738 215.497 -3.62025e+06 1.10345e+08 - 9.5 114241 168.083 -2.96321e+06 9.27832e+07 + 0.5 437808 729.297 -8.87401e+06 2.48957e+08 + 1.5 428332 706.915 -8.74397e+06 2.45467e+08 + 2.5 409773 664.495 -8.4813e+06 2.38491e+08 + 3.5 382912 606.224 -8.08265e+06 2.28041e+08 + 4.5 348909 537.305 -7.54743e+06 2.14162e+08 + 5.5 309303 463.084 -6.88229e+06 1.97007e+08 + 6.5 266014 388.364 -6.10586e+06 1.76939e+08 + 7.5 221314 317.04 -5.25186e+06 1.54662e+08 + 8.5 177695 251.978 -4.36858e+06 1.31271e+08 + 9.5 137599 195.051 -3.5129e+06 1.08162e+08 diff --git a/tests/16_SDFT_GPU/002_PW_SKG_MALL_GPU/refOnsager.txt b/tests/16_SDFT_GPU/002_PW_SKG_MALL_GPU/refOnsager.txt index ed601a37fba..0a385fdd2b3 100644 --- a/tests/16_SDFT_GPU/002_PW_SKG_MALL_GPU/refOnsager.txt +++ b/tests/16_SDFT_GPU/002_PW_SKG_MALL_GPU/refOnsager.txt @@ -1,11 +1,11 @@ ## w(eV) sigma(Sm^-1) kappa(W(mK)^-1) L12/e(Am^-1) L22/e^2(Wm^-1) - 0.5 145953 177.613 -3.70776e+06 1.11017e+08 - 1.5 144420 175.68 -3.67128e+06 1.09969e+08 - 2.5 141416 171.896 -3.59971e+06 1.07914e+08 - 3.5 137057 166.414 -3.4958e+06 1.0493e+08 - 4.5 131513 159.458 -3.36351e+06 1.01129e+08 - 5.5 124997 151.301 -3.20781e+06 9.66556e+07 - 6.5 117754 142.252 -3.03447e+06 9.1673e+07 - 7.5 110048 132.636 -2.84971e+06 8.6359e+07 - 8.5 102146 122.778 -2.65987e+06 8.08943e+07 - 9.5 94302.4 112.981 -2.47103e+06 7.5452e+07 + 0.5 174510 210.224 -4.39362e+06 1.30532e+08 + 1.5 172468 207.753 -4.34347e+06 1.29068e+08 + 2.5 168460 202.913 -4.24504e+06 1.26194e+08 + 3.5 162638 195.9 -4.10197e+06 1.22016e+08 + 4.5 155221 186.995 -3.91955e+06 1.16689e+08 + 5.5 146482 176.545 -3.70439e+06 1.10405e+08 + 6.5 136736 164.941 -3.46414e+06 1.03387e+08 + 7.5 126322 152.599 -3.20706e+06 9.58771e+07 + 8.5 115583 139.929 -2.94162e+06 8.81211e+07 + 9.5 104850 127.32 -2.67598e+06 8.03576e+07 diff --git a/tests/17_DS_DFTU/03_LCAO_DFTU_S2_Z/INPUT b/tests/17_DS_DFTU/03_LCAO_DFTU_S2_Z/INPUT index 1eb50a84479..edd5eda7ef8 100644 --- a/tests/17_DS_DFTU/03_LCAO_DFTU_S2_Z/INPUT +++ b/tests/17_DS_DFTU/03_LCAO_DFTU_S2_Z/INPUT @@ -15,7 +15,8 @@ smearing_sigma 0.01 mixing_type broyden mixing_beta 0.4 ks_solver genelpa -symmetry 0 +symmetry 1 +kpar 4 # DFT+U parameters dft_plus_u 1 diff --git a/tests/17_DS_DFTU/03_LCAO_DFTU_S2_Z/result.ref b/tests/17_DS_DFTU/03_LCAO_DFTU_S2_Z/result.ref index bb1d7aa5ce8..2c12a092ecd 100644 --- a/tests/17_DS_DFTU/03_LCAO_DFTU_S2_Z/result.ref +++ b/tests/17_DS_DFTU/03_LCAO_DFTU_S2_Z/result.ref @@ -1,3 +1,6 @@ -etotref -6772.0999515218118177 -etotperatomref -3386.0499757609 +etotref -6771.6902262248995612 +etotperatomref -3385.8451131124 +pointgroupref C_3v +spacegroupref C_3v +nksibzref 2 totaltimeref 6.11 diff --git a/tests/17_DS_DFTU/04_LCAO_DFTU_S4_XY/INPUT b/tests/17_DS_DFTU/04_LCAO_DFTU_S4_XY/INPUT index a0de7786ca2..d1a025a67b7 100644 --- a/tests/17_DS_DFTU/04_LCAO_DFTU_S4_XY/INPUT +++ b/tests/17_DS_DFTU/04_LCAO_DFTU_S4_XY/INPUT @@ -16,7 +16,8 @@ smearing_sigma 0.01 mixing_type broyden mixing_beta 0.4 ks_solver genelpa -symmetry 0 +symmetry 1 +kpar 4 # DFT+U parameters dft_plus_u 1 diff --git a/tests/17_DS_DFTU/04_LCAO_DFTU_S4_XY/result.ref b/tests/17_DS_DFTU/04_LCAO_DFTU_S4_XY/result.ref index 1a258220cfb..3f7ce113b38 100644 --- a/tests/17_DS_DFTU/04_LCAO_DFTU_S4_XY/result.ref +++ b/tests/17_DS_DFTU/04_LCAO_DFTU_S4_XY/result.ref @@ -1,3 +1,7 @@ -etotref -6772.1004497577005168 -etotperatomref -3386.0500835053 +etotref -6771.5947932947956360 +etotperatomref -3385.7973966474 +pointgroupref C_3v +spacegroupref D_3d +nksibzref 4 +magpointgroupref C_2 totaltimeref 10.33 diff --git a/tests/17_DS_DFTU/12_PW_DS_S2_Z/result.ref b/tests/17_DS_DFTU/12_PW_DS_S2_Z/result.ref index 9dd83d8e706..dd48e508258 100644 --- a/tests/17_DS_DFTU/12_PW_DS_S2_Z/result.ref +++ b/tests/17_DS_DFTU/12_PW_DS_S2_Z/result.ref @@ -1,3 +1,3 @@ -etotref -6366.569118260046 -etotperatomref -3183.2845591300 -totaltimeref 1.97 +etotref -6369.198268154857 +etotperatomref -3184.5991340774 +totaltimeref 4.25 diff --git a/tests/17_DS_DFTU/18_PW_DFTU_DS_S2_Z/result.ref b/tests/17_DS_DFTU/18_PW_DFTU_DS_S2_Z/result.ref index 654cb15b3ee..72fb5c43b87 100644 --- a/tests/17_DS_DFTU/18_PW_DFTU_DS_S2_Z/result.ref +++ b/tests/17_DS_DFTU/18_PW_DFTU_DS_S2_Z/result.ref @@ -1,3 +1,3 @@ -etotref -6355.9855588350255857 -etotperatomref -3177.9927794175 -totaltimeref 2.89 +etotref -6360.5555597255606699 +etotperatomref -3180.2777798628 +totaltimeref 5.86 diff --git a/tests/17_DS_DFTU/19_PW_DFTU_DS_S4_XY/result.ref b/tests/17_DS_DFTU/19_PW_DFTU_DS_S4_XY/result.ref index e40be3f01bb..c3b83965fd3 100644 --- a/tests/17_DS_DFTU/19_PW_DFTU_DS_S4_XY/result.ref +++ b/tests/17_DS_DFTU/19_PW_DFTU_DS_S4_XY/result.ref @@ -1,3 +1,3 @@ -etotref -6360.5554588729937677 -etotperatomref -3180.277729436497 -totaltimeref 1.0 +etotref -6360.5555339529537378 +etotperatomref -3180.2777669765 +totaltimeref 9.56 diff --git a/tests/17_DS_DFTU/21_PW_DFTU_DS_S4_Z/result.ref b/tests/17_DS_DFTU/21_PW_DFTU_DS_S4_Z/result.ref index b06c7ee03a6..ae5a5a853dd 100644 --- a/tests/17_DS_DFTU/21_PW_DFTU_DS_S4_Z/result.ref +++ b/tests/17_DS_DFTU/21_PW_DFTU_DS_S4_Z/result.ref @@ -1,3 +1,3 @@ -etotref -6360.5554655414534864 -etotperatomref -3180.2777327707267 -totaltimeref 1.0 +etotref -6360.5554079455796455 +etotperatomref -3180.2777039728 +totaltimeref 6.65 diff --git a/tests/17_DS_DFTU/41_PW_DS_S4_Thr10_XY/result.ref b/tests/17_DS_DFTU/41_PW_DS_S4_Thr10_XY/result.ref index 8d127ea7298..eb51c776c6e 100644 --- a/tests/17_DS_DFTU/41_PW_DS_S4_Thr10_XY/result.ref +++ b/tests/17_DS_DFTU/41_PW_DS_S4_Thr10_XY/result.ref @@ -1,3 +1,3 @@ -etotref -6369.198254647004 -etotperatomref -3184.599127323502 -totaltimeref 1.0 +etotref -6369.198255055176 +etotperatomref -3184.5991275276 +totaltimeref 9.00 diff --git a/tests/17_DS_DFTU/CMakeLists.txt b/tests/17_DS_DFTU/CMakeLists.txt index 8eead45355c..4fb45bc4b38 100644 --- a/tests/17_DS_DFTU/CMakeLists.txt +++ b/tests/17_DS_DFTU/CMakeLists.txt @@ -15,7 +15,7 @@ if(ENABLE_ASAN) else() add_test( NAME 17_DS_DFTU - COMMAND ${BASH} ../integrate/Autotest.sh -a ${ABACUS_BIN_PATH} -n 4 + COMMAND ${BASH} ../integrate/Autotest.sh -a ${ABACUS_BIN_PATH} -n 4 -j auto WORKING_DIRECTORY ${ABACUS_TEST_DIR}/17_DS_DFTU ) endif() diff --git a/tests/CMakeLists.txt b/tests/CMakeLists.txt index 2347b02d6af..bcbc620a1a3 100644 --- a/tests/CMakeLists.txt +++ b/tests/CMakeLists.txt @@ -6,7 +6,7 @@ add_subdirectory(04_FF) add_subdirectory(05_rtTDDFT) add_subdirectory(06_SDFT) add_subdirectory(07_OFDFT) -add_subdirectory(08_EXX) +add_subdirectory(08_RI) add_subdirectory(10_others) add_subdirectory(11_PW_GPU) # TODO: Enable after DeltaSpin + DFT+U code is merged into develop diff --git a/tests/README b/tests/README index 4a9eb5d6b30..490cf91b182 100644 --- a/tests/README +++ b/tests/README @@ -16,7 +16,7 @@ 05_rtTDDFT: Real-time TDDFT tests. 06_SDFT: Stochastic DFT tests. 07_OFDFT: Orbital-free DFT tests. -08_EXX: Hybrid functional and LR-TDDFT tests. +08_RI: Resolution-of-identity (RI) based tests, including hybrid functionals, LR-TDDFT, RPA and BSE. 09_DeePKS: DeePKS tests. 10_others: Other tests such as LCAO in pw. 11_PW_GPU: KSDFT calculations in PW basis with multple k-point setting using GPU. @@ -47,6 +47,29 @@ cd 101_PW_OU and then type # 'ref': generate result.ref file (the answer you need) +-------------------------------------- + How to modify an existing test case +-------------------------------------- + +When merging or renaming a test case: + +1. Move/rename the directory: `git mv ` +2. Update the INPUT file with new parameters (keep the original style) +3. Update README to describe the new test purpose +4. Update `CASES_CPU.txt` and `CASES_GPU.txt`: remove old name, add new name +5. Remove the old directory: `git rm -rf ` +6. Regenerate `result.ref`: `OMP_NUM_THREADS=1 ../../integrate/Single_job.sh ref` +7. Commit the changes + +For smearing tests: insulators (e.g., H2) can accept any smearing method without changing the result. Metals require smearing for convergence. + +Marking unimplemented features: +- If a feature is not yet implemented (e.g., a certain code path, GPU backend, or functional lacks stress/force support), add the corresponding INPUT parameter but comment it out with an explanation, e.g.: + + `#cal_stress 1 # implicit solvation stress is not implemented yet` + + This documents the intended coverage, makes the gap visible to readers, and lets the test be enabled later by simply uncommenting once the feature lands. + -------------------------------------- Some comman names appear in the tests -------------------------------------- diff --git a/tests/integrate/Autotest.sh b/tests/integrate/Autotest.sh index ad00e05ec12..48b6d2308f7 100755 --- a/tests/integrate/Autotest.sh +++ b/tests/integrate/Autotest.sh @@ -5,6 +5,11 @@ abacus=abacus # number of MPI processes np=4 nt=$OMP_NUM_THREADS # number of OpenMP threads, default is $OMP_NUM_THREADS +# was the thread count pinned on the command line with -o? +nt_explicit=false +# number of test cases to run concurrently; "auto" derives it from the +# number of cores available to this process +njobs=1 # threshold with unit: eV threshold=0.0000001 force_threshold=0.0001 @@ -20,7 +25,7 @@ case='^[^#].*_.*$' # enable AddressSanitizer sanitize=false -threshold_file="threshold" +threshold_file="threshold" # can specify the threshold for each test case # threshold file example: # threshold 0.0000001 @@ -29,7 +34,7 @@ threshold_file="threshold" # fatal_threshold 1 -while getopts a:n:t:c:s:r:f:go: flag +while getopts a:n:t:c:s:r:f:go:j: flag do case "${flag}" in a) abacus=${OPTARG};; @@ -40,25 +45,75 @@ do r) case=${OPTARG};; f) cases_file=${OPTARG};; g) g=true;; #generate test reference - o) nt=${OPTARG};; # number of OpenMP threads + o) nt=${OPTARG}; nt_explicit=true;; # number of OpenMP threads + j) njobs=${OPTARG};; # number of concurrent test cases, or "auto" esac done +#---------------------------------------------------------- +# Core budget. +# +# `nproc` honours $OMP_NUM_THREADS, so it reports the per-case thread count +# rather than the machine size once a caller has exported that variable +# (the CI workflow does). Clear both OpenMP variables for this one call so +# that the count reflects the cores this process may actually use. +#---------------------------------------------------------- +ncores=$(OMP_NUM_THREADS= OMP_THREAD_LIMIT= nproc) + +#---------------------------------------------------------- +# Number of concurrent test cases. +# +# Each case takes $np MPI ranks, so "auto" fits as many whole cases as the +# core budget allows. A case is a separate ABACUS process: running several +# of them side by side uses the cores far better than giving one small case +# a wide OpenMP team, because these cases are only a few atoms each. +#---------------------------------------------------------- +if [ "$njobs" == "auto" ]; then + if [ "$np" -le 0 ] 2>/dev/null; then + njobs=$ncores + else + njobs=$(expr $ncores / $np) + fi +fi +if [ "$njobs" -lt 1 ] 2>/dev/null; then + njobs=1 +fi + +# Address Sanitizer appends every case to one shared report file, so those +# runs stay serial. +if [ "$sanitize" == true ] && [ "$njobs" -gt 1 ]; then + echo "Address Sanitizer run: forcing -j 1, the diagnostics report is shared." + njobs=1 +fi + +#---------------------------------------------------------- # number of OpenMP threads +# +# A concurrent run must not inherit an $OMP_NUM_THREADS that was sized for +# one case at a time, or the cases together oversubscribe the machine. +# Recompute it from the core budget unless -o pinned it. +#---------------------------------------------------------- +if [ "$njobs" -gt 1 ] && [ "$nt_explicit" == false ]; then + nt="" +fi if [[ -z "$nt" ]]; then if [ "$np" -le 0 ] 2>/dev/null; then # serial build (no MPI launcher): use all cores for OpenMP - nt=$(nproc) + nt=$(expr $ncores / $njobs) else - nt=$(expr `nproc` / ${np}) + nt=$(expr $ncores / ${np} / $njobs) fi fi +if [ "$nt" -lt 1 ] 2>/dev/null; then + nt=1 +fi export OMP_NUM_THREADS=${nt} echo "-----AUTO TESTS OF ABACUS ------" echo "ABACUS path: $abacus" echo "Number of processes: $np" echo "Number of threads: $nt" +echo "Concurrent test cases: $njobs" echo "Test accuracy totenergy: $threshold eV" echo "Test accuracy force: $force_threshold" echo "Test accuracy stress: $stress_threshold" @@ -100,7 +155,7 @@ check_out(){ properties=`awk '{print $1}' $outfile` #------------------------------------------------------ - # README + # README #------------------------------------------------------ if test -e "README"; then readme=`cat README` @@ -113,7 +168,7 @@ check_out(){ ifail=0 # if all properties have no warning. 0: no warning, 1: warning ifatal=0 # if all properties have no fatal error. 0: no fatal error, 1: fatal error for key in $properties; do - + if [ $key == "totaltimeref" ]; then # echo "time=$cal ref=$ref" break @@ -216,7 +271,7 @@ get_threshold() threshold_f=$1 threshold_name=$2 default_value=$3 - if [ -e $threshold_f ]; then + if [ -e $threshold_f ]; then threshold_value=$(awk -v tn="$threshold_name" '$1==tn {print $2}' "$threshold_f") if [ -n "$threshold_value" ]; then echo $threshold_value @@ -228,6 +283,107 @@ get_threshold() fi } +#--------------------------------------------- +# run_case(): run and check one test case. +# +# Console output goes to stdout. The counter deltas this case produced are +# also written to "$result_dir/$1.res", so the caller can add them up even +# when the case ran inside a subshell of the worker pool. +#--------------------------------------------- +run_case() +{ + dir=$1 + + # These are per-case deltas here; the caller sums them over all cases. + failed=0 + ok=0 + fatal=0 + failed_case_list="" + fatal_case_list="" + fatal_detail_list="" + case_status="" + + if [ ! -d $dir ];then + echo -e "\e[0;31m[ERROR ]\e[0m $dir is not a directory.\n" + let fatal++ + fatal_case_list+=$dir'\n' + else + cd $dir + echo -e "\e[0;32m[ RUN ]\e[0m $dir" + TIMEFORMAT='[----------] Time elapsed: %R seconds' + #parallel test + time { + if [ "$np" -le 0 ] 2>/dev/null; then + # serial build: run the binary directly, no MPI launcher. + # This lets a serial ABACUS (ENABLE_MPI=OFF, e.g. the native + # Windows build) reuse this harness unchanged. + $abacus > log.txt + elif [ "$case" = "282_NO_RPA" ]; then + mpirun -np 1 $abacus > log.txt + elif grep -qE '^[[:space:]]*of_ml_gene_data[[:space:]]+1([[:space:]]|$)' INPUT; then + # of_ml_gene_data supports single-rank only. + mpirun -np 1 $abacus > log.txt + else + mpirun -np $np $abacus > log.txt + fi + + # if ABACUS failed, print out the error message + if [ $? -ne 0 ]; then + echo -e "\e[0;31m[ERROR ]\e[0m $dir failed." + let failed++ + failed_case_list+=$dir'\n' + case_status+=$dir' 0\n' + cat log.txt + else + # check the output + test -d OUT.autotest || (echo "No 'OUT.autotest' dir presented. Some errors may happened in ABACUS." && exit 1) + if test -z $g + then + bash -e ../../integrate/tools/catch_properties.sh result.out + if [ $? -ne 0 ]; then + echo -e "\e[0;31m [ERROR ] Fatal Error in catch_properties.sh \e[0m" + let fatal++ + fatal_case_list+=$dir'\n' + else + my_threshold=$(get_threshold $threshold_file "threshold" $threshold) + my_force_threshold=$(get_threshold $threshold_file "force_threshold" $force_threshold) + my_stress_threshold=$(get_threshold $threshold_file "stress_threshold" $stress_threshold) + my_fatal_threshold=$(get_threshold $threshold_file "fatal_threshold" $fatal_threshold) + my_descriptor_threshold=$(get_threshold $threshold_file "descriptor_threshold" $descriptor_threshold) + check_out result.out $my_threshold $my_force_threshold $my_stress_threshold $my_fatal_threshold $my_descriptor_threshold + fi + else + bash -e ../../integrate/tools/catch_properties.sh result.ref + fi + fi + + if [ "$sanitize" == true ]; then + echo -e "## Test case ${dir}\n" >> ${report} + for diagnostic in asan.*; do + echo -e "### On process id ${diagnostic}\n" >> ${report} + echo -e "\`\`\`bash" >> ${report} + cat ${diagnostic} >> ${report} + echo -e "\`\`\`\n" >> ${report} + done + fi + } + echo "" + cd ../ + fi + + # The list variables hold literal "\n" escapes that printf %b expands at + # the very end, so each one still fits on a single line here. + { + echo "FAILED $failed" + echo "OK $ok" + echo "FATAL $fatal" + echo "FAILED_LIST $failed_case_list" + echo "FATAL_LIST $fatal_case_list" + echo "FATAL_DETAIL $fatal_detail_list" + echo "STATUS $case_status" + } > "$result_dir/$dir.res" +} + #--------------------------------------------- # the file name that contains all of the tests #--------------------------------------------- @@ -247,6 +403,9 @@ fatal_threshold=1 report="" repo="$(realpath ..)/" +result_dir=$(mktemp -d "${TMPDIR:-/tmp}/abacus-autotest.XXXXXX") +trap 'rm -rf "$result_dir"' EXIT + if [ "$sanitize" == true ]; then echo "Testing with Address Sanitizer..." mkdir ../html @@ -255,74 +414,59 @@ if [ "$sanitize" == true ]; then export ASAN_OPTIONS="log_path=asan" fi -for dir in $testdir; do - if [ ! -d $dir ];then - echo -e "\e[0;31m[ERROR ]\e[0m $dir is not a directory.\n" - let fatal++ - fatal_case_list+=$dir'\n' - continue - fi - cd $dir - echo -e "\e[0;32m[ RUN ]\e[0m $dir" - TIMEFORMAT='[----------] Time elapsed: %R seconds' - #parallel test - time { - if [ "$np" -le 0 ] 2>/dev/null; then - # serial build: run the binary directly, no MPI launcher. - # This lets a serial ABACUS (ENABLE_MPI=OFF, e.g. the native - # Windows build) reuse this harness unchanged. - $abacus > log.txt - elif [ "$case" = "282_NO_RPA" ]; then - mpirun -np 1 $abacus > log.txt - elif grep -qE '^[[:space:]]*of_ml_gene_data[[:space:]]+1([[:space:]]|$)' INPUT; then - # of_ml_gene_data supports single-rank only. - mpirun -np 1 $abacus > log.txt - else - mpirun -np $np $abacus > log.txt - fi +if [ "$njobs" -le 1 ]; then + # A subshell here too, so that a case's `cd` and its counters stay + # local exactly as they do in the worker pool below. + for dir in $testdir; do + ( run_case $dir ) + done +else + #----------------------------------------------------- + # Worker pool. Each case runs in its own subshell, so + # its `cd` and its counters stay local, and its console + # output is buffered to a file. + #----------------------------------------------------- + for dir in $testdir; do + ( run_case $dir ) > "$result_dir/$dir.out" 2>&1 & + while [ $(jobs -rp | wc -l) -ge $njobs ]; do + sleep 0.2 + done + done + wait - # if ABACUS failed, print out the error message - if [ $? -ne 0 ]; then - echo -e "\e[0;31m[ERROR ]\e[0m $dir failed." - let failed++ - failed_case_list+=$dir'\n' - case_status+=$dir' 0\n' - cat log.txt - else - # check the output - test -d OUT.autotest || (echo "No 'OUT.autotest' dir presented. Some errors may happened in ABACUS." && exit 1) - if test -z $g - then - bash -e ../../integrate/tools/catch_properties.sh result.out - if [ $? -ne 0 ]; then - echo -e "\e[0;31m [ERROR ] Fatal Error in catch_properties.sh \e[0m" - let fatal++ - fatal_case_list+=$dir'\n' - else - my_threshold=$(get_threshold $threshold_file "threshold" $threshold) - my_force_threshold=$(get_threshold $threshold_file "force_threshold" $force_threshold) - my_stress_threshold=$(get_threshold $threshold_file "stress_threshold" $stress_threshold) - my_fatal_threshold=$(get_threshold $threshold_file "fatal_threshold" $fatal_threshold) - my_descriptor_threshold=$(get_threshold $threshold_file "descriptor_threshold" $descriptor_threshold) - check_out result.out $my_threshold $my_force_threshold $my_stress_threshold $my_fatal_threshold $my_descriptor_threshold - fi - else - bash -e ../../integrate/tools/catch_properties.sh result.ref - fi - fi + #----------------------------------------------------- + # Replay the buffered logs in the order of the cases + # file, so a concurrent run reads like a serial one. + #----------------------------------------------------- + for dir in $testdir; do + test -f "$result_dir/$dir.out" && cat "$result_dir/$dir.out" + done +fi - if [ "$sanitize" == true ]; then - echo -e "## Test case ${dir}\n" >> ${report} - for diagnostic in asan.*; do - echo -e "### On process id ${diagnostic}\n" >> ${report} - echo -e "\`\`\`bash" >> ${report} - cat ${diagnostic} >> ${report} - echo -e "\`\`\`\n" >> ${report} - done - fi - } - echo "" - cd ../ +#----------------------------------------------------- +# Add up what the cases reported, in cases-file order. +# Start from zero: only the .res files are authoritative. +#----------------------------------------------------- +failed=0 +ok=0 +fatal=0 +failed_case_list="" +fatal_case_list="" +fatal_detail_list="" +case_status="" +for dir in $testdir; do + test -f "$result_dir/$dir.res" || continue + while read -r rkey rvalue; do + case "$rkey" in + FAILED) failed=$(expr $failed + $rvalue);; + OK) ok=$(expr $ok + $rvalue);; + FATAL) fatal=$(expr $fatal + $rvalue);; + FAILED_LIST) failed_case_list+=$rvalue;; + FATAL_LIST) fatal_case_list+=$rvalue;; + FATAL_DETAIL) fatal_detail_list+=$rvalue;; + STATUS) case_status+=$rvalue;; + esac + done < "$result_dir/$dir.res" done if [ "$sanitize" == true ]; then diff --git a/tests/integrate/CMakeLists.txt b/tests/integrate/CMakeLists.txt index c1db9c03d25..9f737798889 100644 --- a/tests/integrate/CMakeLists.txt +++ b/tests/integrate/CMakeLists.txt @@ -11,7 +11,7 @@ if(ENABLE_ASAN) else() add_test( NAME integrated_test - COMMAND ${BASH} Autotest.sh -a ${ABACUS_BIN_PATH} -n 4 + COMMAND ${BASH} Autotest.sh -a ${ABACUS_BIN_PATH} -n 4 -j auto WORKING_DIRECTORY ${ABACUS_TEST_DIR}/integrate ) endif() \ No newline at end of file diff --git a/tests/integrate/general_info b/tests/integrate/general_info index 7b41f3114e3..c0d397efdc5 100644 --- a/tests/integrate/general_info +++ b/tests/integrate/general_info @@ -1,3 +1,3 @@ -EXEC abacus +EXEC abacus CHECKACCURACY 2 NUMBEROFPROCESS 4 diff --git a/tests/integrate/tools/catch_properties.sh b/tests/integrate/tools/catch_properties.sh index d37daa7af6b..94b4f526119 100755 --- a/tests/integrate/tools/catch_properties.sh +++ b/tests/integrate/tools/catch_properties.sh @@ -152,6 +152,20 @@ if [ $calculation != "get_wf" ]\ echo "etotperatomref $etotperatom" >>$1 fi +# Opt-in collinear magnetic-state check. Comparing both moments distinguishes +# compensated AFM from NM; the reference tolerance is 1e-3 mu_B per cell. +if [ "$nspin" = "2" ] && [ -f magnetism.ref ]; then + awk ' + /Total magnetism \(Bohr mag\/cell\)/ {total = $NF; have_total = 1} + /Absolute magnetism \(Bohr mag\/cell\)/ {absolute = $NF; have_absolute = 1} + END { + if (!have_total || !have_absolute) exit 1 + print total, absolute + } + ' "$running_path" > magnetism.out + record_compare_result "$1" "CompareMagnetism_pass" "magnetism.ref" "magnetism.out" 3 +fi + #---------------------------- # force information # echo "hasforce:"$has_force @@ -233,7 +247,7 @@ fi # echo $out_elf #------------------------------- if ! test -z "$out_elf" && [ $out_elf == 1 ]; then - elf1ref=refelftot.cube + elf1ref=elftot.cube.ref elf1cal=OUT.autotest/elftot.cube python3 $COMPARE_SCRIPT $elf1ref $elf1cal 3 echo "ComparePot1_pass $?" >>$1 @@ -490,7 +504,7 @@ fi #----------------------------------- #echo $has_mat_r if ! test -z "$has_mat_r" && [ $has_mat_r == 1 ]; then - python3 $COMPARE_SCRIPT rr.csr.ref OUT.autotest/rr.csr 8 + python3 $COMPARE_SCRIPT rr_nao.txt.ref OUT.autotest/rr_nao.txt 8 echo "ComparerR_pass $?" >>$1 fi @@ -499,7 +513,7 @@ fi #----------------------------------- #echo $has_mat_t if ! test -z "$has_mat_t" && [ $has_mat_t == 1 ]; then - python3 $COMPARE_SCRIPT trs1_nao.csr.ref OUT.autotest/trs1_nao.csr 8 + python3 $COMPARE_SCRIPT tr_nao.csr.ref OUT.autotest/tr_nao.csr 8 echo "ComparerTR_pass $?" >>$1 fi diff --git a/tests/performance/catch_properties.sh b/tests/performance/catch_properties.sh index 9e815d54de3..d8b3794c41d 100644 --- a/tests/performance/catch_properties.sh +++ b/tests/performance/catch_properties.sh @@ -89,10 +89,10 @@ fi #echo $total_charge if ! test -z "$has_dos" && [ $has_dos -eq 1 ]; then - total_dos=`cat OUT.autotest/DOS1_smearing.dat | awk 'END {print}' | awk '{print $3}'` + total_dos=`cat OUT.autotest/doss1g1_nao.txt | awk 'END {print}' | awk '{print $3}'` echo "totaldosref $total_dos" >> $1 fi -# smearing_dos=`sum_file OUT.autotest/DOS1_smearing.dat` +# smearing_dos=`sum_file OUT.autotest/doss1g1_nao.txt` # echo "totaldossmearing $smearing_dos" >> $1 diff --git a/toolchain/README.md b/toolchain/README.md index 89dd8c5e9ab..5f81fedfc5b 100644 --- a/toolchain/README.md +++ b/toolchain/README.md @@ -127,9 +127,16 @@ Before running the toolchain, please make sure you have loaded the related envir - [LibComm](https://github.com/abacusmodeling/LibComm) - Communication library - [NEP](https://github.com/brucefan1983/NEP_CPU) - Neuroevolution Potential - [Cereal](https://github.com/USCiLab/cereal) - C++ serialization library - - [RapidJSON](https://github.com/Tencent/rapidjson) - Fast JSON parser/generator + - [nlohmann-json](https://github.com/nlohmann/json) - Fast JSON parser/generator - **Reference mirror:** [CP2K static downloads](https://www.cp2k.org/static/downloads) -- All package from GitHub will be downloaded by `wget` from `codeload.github.com`, which bypass the difficulty of CN Internet in some extent. +- Most GitHub source archives are downloaded from `codeload.github.com`. nlohmann-json uses the official `json.tar.xz` release asset. + +JSON output is enabled with `-DENABLE_JSON=ON`. `ENABLE_RAPIDJSON` is no longer +supported because the JSON backend has been replaced by nlohmann-json. The +toolchain package option is `--with-json=install|system|/prefix|no`. In `system` +mode, ABACUS uses CMake's normal package search; the installation must provide +`nlohmann_jsonConfig.cmake`, not just the headers. Release archives are cached as +`build/json-.tar.xz` for offline installation. ### Offline Installation @@ -200,7 +207,7 @@ Mix online and offline packages as needed - the toolchain automatically detects | ELPA | 2026.02.001 / 2024.05.001 | Eigenvalue solver | LGPL-3.0-only | Install | | **Advanced Features** ||||| | Cereal | pinned commit | C++ Serialization | BSD | Install | -| RapidJSON | pinned commit | JSON parsing | MIT | Install | +| nlohmann-json | 3.12.0 | JSON parsing | MIT | Install | | LibRI | pinned commit | EXX calculations | GPL-3.0 | Install | | LibComm | pinned commit | EXX calculations | GPL-3.0 | Install | | LibTorch | 2.1.2 / 1.12.1 | MLALGO support | BSD-3-Clause | Optional | diff --git a/toolchain/build_abacus_aocc-aocl.sh b/toolchain/build_abacus_aocc-aocl.sh index c78b86b1370..08bfab5fe43 100755 --- a/toolchain/build_abacus_aocc-aocl.sh +++ b/toolchain/build_abacus_aocc-aocl.sh @@ -71,7 +71,7 @@ cmake -B $BUILD_DIR -DCMAKE_INSTALL_PREFIX=$PREFIX \ -DENABLE_LIBXC=ON \ -DENABLE_OPENMP=ON \ -DENABLE_ELPA=ON \ - -DENABLE_RAPIDJSON=ON \ + -DENABLE_JSON=ON \ -DENABLE_LIBRI=ON \ -DLIBRI_DIR=$LIBRI \ -DLIBCOMM_DIR=$LIBCOMM \ diff --git a/toolchain/build_abacus_gcc-aocl.sh b/toolchain/build_abacus_gcc-aocl.sh index f2f96b1509d..153b898b21e 100755 --- a/toolchain/build_abacus_gcc-aocl.sh +++ b/toolchain/build_abacus_gcc-aocl.sh @@ -67,7 +67,7 @@ cmake -B $BUILD_DIR -DCMAKE_INSTALL_PREFIX=$PREFIX \ -DENABLE_LIBXC=ON \ -DENABLE_OPENMP=ON \ -DENABLE_ELPA=ON \ - -DENABLE_RAPIDJSON=ON \ + -DENABLE_JSON=ON \ -DENABLE_LIBRI=ON \ -DLIBRI_DIR=$LIBRI \ -DLIBCOMM_DIR=$LIBCOMM \ diff --git a/toolchain/build_abacus_gcc-mkl.sh b/toolchain/build_abacus_gcc-mkl.sh index db1a30e463f..dbce5db16c7 100755 --- a/toolchain/build_abacus_gcc-mkl.sh +++ b/toolchain/build_abacus_gcc-mkl.sh @@ -64,7 +64,7 @@ cmake -B $BUILD_DIR -DCMAKE_INSTALL_PREFIX=$PREFIX \ -DENABLE_OPENMP=ON \ -DENABLE_ELPA=ON \ -DENABLE_DFTD4=ON \ - -DENABLE_RAPIDJSON=ON \ + -DENABLE_JSON=ON \ -DENABLE_LIBRI=ON \ -DLIBRI_DIR=$LIBRI \ -DLIBCOMM_DIR=$LIBCOMM \ diff --git a/toolchain/build_abacus_gnu.sh b/toolchain/build_abacus_gnu.sh index 2a26780f3c9..505ae67268d 100755 --- a/toolchain/build_abacus_gnu.sh +++ b/toolchain/build_abacus_gnu.sh @@ -66,7 +66,7 @@ cmake -B $BUILD_DIR -DCMAKE_INSTALL_PREFIX=$PREFIX \ -DENABLE_OPENMP=ON \ -DENABLE_ELPA=ON \ -DENABLE_DFTD4=ON \ - -DENABLE_RAPIDJSON=ON \ + -DENABLE_JSON=ON \ -DENABLE_LIBRI=ON \ -DLIBRI_DIR=$LIBRI \ -DLIBCOMM_DIR=$LIBCOMM \ diff --git a/toolchain/build_abacus_intel.sh b/toolchain/build_abacus_intel.sh index da043fdc80c..100da97d583 100755 --- a/toolchain/build_abacus_intel.sh +++ b/toolchain/build_abacus_intel.sh @@ -65,7 +65,7 @@ cmake -B $BUILD_DIR -DCMAKE_INSTALL_PREFIX=$PREFIX \ -DENABLE_OPENMP=ON \ -DENABLE_ELPA=ON \ -DENABLE_DFTD4=ON \ - -DENABLE_RAPIDJSON=ON \ + -DENABLE_JSON=ON \ -DENABLE_LIBRI=ON \ -DLIBRI_DIR=$LIBRI \ -DLIBCOMM_DIR=$LIBCOMM \ diff --git a/toolchain/scripts/common_vars.sh b/toolchain/scripts/common_vars.sh index fd003555c3b..78a5833faa3 100755 --- a/toolchain/scripts/common_vars.sh +++ b/toolchain/scripts/common_vars.sh @@ -13,9 +13,23 @@ SETUPFILE=${SETUPFILE:-"${INSTALLDIR}/setup"} ARCH_FILE_TEMPLATE=${ARCH_FILE_TEMPLATE:-"${SCRIPTDIR}/arch_base.tmpl"} VERSION_FILE=${VERSION_FILE:-"${SCRIPTDIR}/VERSION"} -# system arch gotten from OpenBLAS prebuild -OPENBLAS_ARCH=${OPENBLAS_ARCH:-''} -OPENBLAS_LIBCORE=${OPENBLAS_LIBCORE:-''} +# host architecture +SYSTEM_ARCH=${SYSTEM_ARCH:-"$(uname -m)"} +case "${SYSTEM_ARCH}" in + x86_64 | amd64) + SYSTEM_ARCH="x86_64" + ;; + aarch64 | arm64) + SYSTEM_ARCH="arm64" + ;; + i?86) + SYSTEM_ARCH="i386" + ;; + *) + # Keep other architectures as reported by uname. + ;; +esac +export SYSTEM_ARCH # search paths SYS_INCLUDE_PATH=${SYS_INCLUDE_PATH:-'/usr/local/include:/usr/include'} diff --git a/toolchain/scripts/get_openblas_arch.sh b/toolchain/scripts/get_openblas_arch.sh deleted file mode 100755 index 79f40805e97..00000000000 --- a/toolchain/scripts/get_openblas_arch.sh +++ /dev/null @@ -1,76 +0,0 @@ -#!/bin/bash -e - -# TODO: Review and if possible fix shellcheck errors. -# shellcheck disable=all - -[ "${BASH_SOURCE[0]}" ] && SCRIPT_NAME="${BASH_SOURCE[0]}" || SCRIPT_NAME=$0 -SCRIPT_DIR="$(cd "$(dirname "$SCRIPT_NAME")" && pwd -P)" - -# Load centralized version management -source "${SCRIPT_DIR}"/package_versions.sh - -# Load OpenBLAS package variables with version suffix support -# Check for version configuration from environment or individual package setting -version_suffix="" -if [[ -n "${ABACUS_TOOLCHAIN_PACKAGE_VERSIONS}" ]]; then - # Check for individual package version override - if echo "${ABACUS_TOOLCHAIN_PACKAGE_VERSIONS}" | grep -q "openblas:alt"; then - version_suffix="alt" - elif echo "${ABACUS_TOOLCHAIN_PACKAGE_VERSIONS}" | grep -q "openblas:main"; then - version_suffix="main" - fi -fi -# Fall back to global version suffix if no individual setting -if [[ -z "$version_suffix" && -n "${ABACUS_TOOLCHAIN_VERSION_SUFFIX}" ]]; then - version_suffix="${ABACUS_TOOLCHAIN_VERSION_SUFFIX}" -fi -# Load package variables with appropriate version -load_package_vars "openblas" "$version_suffix" -openblas_pkg="OpenBLAS-${openblas_ver}.tar.gz" - -find_openblas_dir() { - local __dir='' - for __dir in *OpenBLAS*; do - if [ -d "$__dir" ]; then - echo "$__dir" - return - fi - done - echo '' -} - -source "${SCRIPT_DIR}"/common_vars.sh -source "${SCRIPT_DIR}"/tool_kit.sh -source "${SCRIPT_DIR}"/signal_trap.sh -source "${INSTALLDIR}"/toolchain.conf -source "${INSTALLDIR}"/toolchain.env - -! [ -d "${BUILDDIR}" ] && mkdir -p "${BUILDDIR}" -cd "${BUILDDIR}" - -echo "==================== Getting proc arch info using OpenBLAS tools ====================" -url="https://codeload.github.com/OpenMathLib/OpenBLAS/tar.gz/v${openblas_ver}" -retrieve_package "${openblas_sha256}" "${openblas_pkg}" "${url}" -# if toolchain run in pack-run mode, do exit -if [ "${PACK_RUN}" = "__TRUE__" ]; then - echo "--pack-run mode specified, skip arch detection" - exit 0 -fi -tar -xzf ${openblas_pkg} -openblas_dir="$(find_openblas_dir)" -openblas_conf="${openblas_dir}/Makefile.conf" -# try find Makefile.config, if not then generate one with make lapack_prebuild -if ! [ -f "$openblas_conf" ]; then - cd "$openblas_dir" - make lapack_prebuild - cd .. -fi -OPENBLAS_LIBCORE="$(grep 'LIBCORE=' $openblas_conf | cut -f2 -d=)" -OPENBLAS_ARCH="$(grep 'ARCH=' $openblas_conf | cut -f2 -d=)" -echo "OpenBLAS detected LIBCORE = $OPENBLAS_LIBCORE" -echo "OpenBLAS detected ARCH = $OPENBLAS_ARCH" -# output setup file -cat << EOF > "${BUILDDIR}/openblas_arch" -export OPENBLAS_LIBCORE="${OPENBLAS_LIBCORE}" -export OPENBLAS_ARCH="${OPENBLAS_ARCH}" -EOF diff --git a/toolchain/scripts/lib/config_manager.sh b/toolchain/scripts/lib/config_manager.sh index 80eacd51969..ce9345617c9 100644 --- a/toolchain/scripts/lib/config_manager.sh +++ b/toolchain/scripts/lib/config_manager.sh @@ -19,7 +19,7 @@ CONFIG_FILE_LOADED=false tool_list="gcc intel amd cmake" mpi_list="mpich openmpi intelmpi" math_list="mkl aocl openblas" -lib_list="fftw libxc scalapack elpa cereal rapidjson libtorch libnpy libri libcomm nep dftd4" +lib_list="fftw libxc scalapack elpa cereal json libtorch libnpy libri libcomm nep dftd4" package_list="${tool_list} ${mpi_list} ${math_list} ${lib_list}" # Configuration file paths for loading in advance @@ -367,7 +367,7 @@ config_set_defaults() { CONFIG_CACHE["with_scalapack"]="__INSTALL__" CONFIG_CACHE["with_elpa"]="__INSTALL__" CONFIG_CACHE["with_cereal"]="__INSTALL__" - CONFIG_CACHE["with_rapidjson"]="__INSTALL__" + CONFIG_CACHE["with_json"]="__INSTALL__" CONFIG_CACHE["with_libri"]="__INSTALL__" CONFIG_CACHE["with_libcomm"]="__INSTALL__" CONFIG_CACHE["with_nep"]="__DONTUSE__" diff --git a/toolchain/scripts/lib/package_manager.sh b/toolchain/scripts/lib/package_manager.sh index 71432578d81..0bdac523a06 100644 --- a/toolchain/scripts/lib/package_manager.sh +++ b/toolchain/scripts/lib/package_manager.sh @@ -62,7 +62,7 @@ package_manager_define_dependencies() { # Stage 4: Advanced Libraries PACKAGE_DEPENDENCIES["dftd4"]="cmake" PACKAGE_DEPENDENCIES["cereal"]="gcc cmake" - PACKAGE_DEPENDENCIES["rapidjson"]="gcc cmake" + PACKAGE_DEPENDENCIES["json"]="gcc cmake" PACKAGE_DEPENDENCIES["libtorch"]="gcc cmake" PACKAGE_DEPENDENCIES["libnpy"]="gcc cmake" PACKAGE_DEPENDENCIES["libri"]="gcc cmake" diff --git a/toolchain/scripts/lib/user_interface.sh b/toolchain/scripts/lib/user_interface.sh index 08a75f40450..3ef91c1731e 100644 --- a/toolchain/scripts/lib/user_interface.sh +++ b/toolchain/scripts/lib/user_interface.sh @@ -606,7 +606,7 @@ PACKAGE CONTROL OPTIONS: 📦 MPI Libraries: openmpi, mpich, intelmpi 📦 Math Libraries: mkl, aocl, openblas, scalapack 📦 Scientific: libxc, fftw, elpa - 📦 Advanced: cereal, rapidjson, libri, libcomm, libtorch, libnpy, nep + 📦 Advanced: cereal, json, libri, libcomm, libtorch, libnpy, nep ADVANCED OPTIONS: --enable-[=yes/no] Enable specific advanced features diff --git a/toolchain/scripts/package_versions.sh b/toolchain/scripts/package_versions.sh index 6cef74e63fd..0f19e9db4f3 100644 --- a/toolchain/scripts/package_versions.sh +++ b/toolchain/scripts/package_versions.sh @@ -85,8 +85,8 @@ scalapack_alt_sha256="a2f0c9180a210bf7ffe126c9cb81099cf337da1a7120ddb4cbe4894eb7 # ============================================================================= # DFT-D4 dispersion correction -dftd4_ver="4.2.0" -dftd4_sha256="467e024071510ad82b862c66c383c2ebc164fc1140e15dfc79f48d2f999fd184" +dftd4_ver="4.3.0" +dftd4_sha256="e94c5d021c0a4b4aa9b5587e36fda2f398271dc2dc1abfbe097fdb04022bef35" # LibTorch (supports dual versions) - main=2.1.2, alt=1.12.1 libtorch_main_ver="2.1.2" @@ -113,8 +113,8 @@ libcomm_sha256="d7b991465d98d7b715b484d86880bf3525b9bf0cc62c3e5d38b0f6d140f6b9d4 libri_ver="b0eff7a" libri_sha256="bee9df54e0d827942524640897c3a57c0c738d2db6b2d73a09ab6c054bef94ea" -rapidjson_ver="24b5e7a" -rapidjson_sha256="dcb57b11036cb8fc6b2a57a6aded68d52e9cfe543811bf4fa8941087f84e72d0" +json_ver="3.12.0" +json_sha256="42f6e95cad6ec532fd372391373363b62a14af6d771056dbfc86160e6dfff7aa" # NEP (Neural Evolution Potential) - CPU version nep_ver="629ec5d" @@ -139,24 +139,10 @@ load_package_vars() { fi ;; "cmake") - # Determine architecture for SHA256 selection - local arch_suffix="" - if [ "${OPENBLAS_ARCH}" = "arm64" ]; then - if [ "$(uname -s)" = "Darwin" ]; then - arch_suffix="_macos" - else - arch_suffix="_aarch64" - fi - else - arch_suffix="_x86_64" - fi - if [ "${version_suffix}" = "alt" ]; then cmake_ver="${cmake_alt_ver}" - eval "cmake_sha256=\${cmake_alt_sha256${arch_suffix}}" else cmake_ver="${cmake_main_ver}" - eval "cmake_sha256=\${cmake_main_sha256${arch_suffix}}" fi ;; "openmpi") @@ -260,9 +246,9 @@ load_package_vars() { libri_ver="${libri_ver}" libri_sha256="${libri_sha256}" ;; - "rapidjson") - rapidjson_ver="${rapidjson_ver}" - rapidjson_sha256="${rapidjson_sha256}" + "json") + json_ver="${json_ver}" + json_sha256="${json_sha256}" ;; "nep") nep_ver="${nep_ver}" diff --git a/toolchain/scripts/stage0/install_cmake.sh b/toolchain/scripts/stage0/install_cmake.sh index 1084e757f3e..55d82b82b2f 100755 --- a/toolchain/scripts/stage0/install_cmake.sh +++ b/toolchain/scripts/stage0/install_cmake.sh @@ -29,41 +29,12 @@ if [[ -z "$version_suffix" && -n "${ABACUS_TOOLCHAIN_VERSION_SUFFIX}" ]]; then version_suffix="${ABACUS_TOOLCHAIN_VERSION_SUFFIX}" fi -# Ensure OPENBLAS_ARCH is set before loading package variables -# This is needed for architecture-specific SHA256 selection -# In --pack-run mode, openblas_arch file may contain empty values, so we need fallback -if [ -f "${BUILDDIR}/openblas_arch" ]; then - source "${BUILDDIR}/openblas_arch" -fi - -if [ -z "${OPENBLAS_ARCH}" ]; then - case "$(uname -m)" in - x86_64|amd64) OPENBLAS_ARCH="x86_64" ;; - aarch64|arm64) OPENBLAS_ARCH="arm64" ;; - *) OPENBLAS_ARCH="x86_64" ;; # default fallback - esac - echo "OPENBLAS_ARCH not set, using fallback: ${OPENBLAS_ARCH}" -fi - -# Export OPENBLAS_ARCH to ensure it's available throughout the script -export OPENBLAS_ARCH - # Load package variables with appropriate version load_package_vars "cmake" "$version_suffix" source "${INSTALLDIR}"/toolchain.conf source "${INSTALLDIR}"/toolchain.env -# Re-apply architecture detection if OPENBLAS_ARCH is still empty after sourcing -if [ -z "${OPENBLAS_ARCH}" ]; then - case "$(uname -m)" in - x86_64|amd64) OPENBLAS_ARCH="x86_64" ;; - aarch64|arm64) OPENBLAS_ARCH="arm64" ;; - *) OPENBLAS_ARCH="x86_64" ;; # default fallback - esac - export OPENBLAS_ARCH -fi - [ -f "${BUILDDIR}/setup_cmake" ] && rm "${BUILDDIR}/setup_cmake" ! [ -d "${BUILDDIR}" ] && mkdir -p "${BUILDDIR}" @@ -71,22 +42,33 @@ cd "${BUILDDIR}" case "${with_cmake}" in __INSTALL__) echo "==================== Installing CMake ====================" - if [ "${OPENBLAS_ARCH}" = "arm64" ]; then - if [ "$(uname -s)" = "Darwin" ]; then + case "$(uname -s):${SYSTEM_ARCH}" in + Darwin:x86_64 | Darwin:arm64) cmake_arch="macos-universal" - elif [ "$(uname -s)" = "Linux" ]; then + cmake_checksum_arch="macos" + ;; + Linux:x86_64) + cmake_arch="linux-x86_64" + cmake_checksum_arch="x86_64" + ;; + Linux:arm64) cmake_arch="linux-aarch64" - else + cmake_checksum_arch="aarch64" + ;; + *) report_error ${LINENO} \ - "cmake installation for ARCH=${OPENBLAS_ARCH} under $(uname -s) is not supported. You can try to use the system installation using the flag --with-cmake=system instead." - fi - elif [ "${OPENBLAS_ARCH}" = "x86_64" ]; then - cmake_arch="linux-x86_64" + "cmake installation for ARCH=${SYSTEM_ARCH} under $(uname -s) is not supported. You can try to use the system installation using the flag \"--with-cmake=system\" instead." + exit 1 + ;; + esac + + if [ "${version_suffix}" = "alt" ]; then + cmake_checksum_var="cmake_alt_sha256_${cmake_checksum_arch}" else - report_error ${LINENO} \ - "cmake installation for ARCH=${OPENBLAS_ARCH} is not supported. You can try to use the system installation using the flag --with-cmake=system instead." - exit 1 + cmake_checksum_var="cmake_main_sha256_${cmake_checksum_arch}" fi + cmake_sha256="${!cmake_checksum_var}" + pkg_install_dir="${INSTALLDIR}/cmake-${cmake_ver}" #pkg_install_dir="${HOME}/apps/cmake/${cmake_ver}" install_lock_file="${pkg_install_dir}/install_successful" diff --git a/toolchain/scripts/stage0/setup_buildtools.sh b/toolchain/scripts/stage0/setup_buildtools.sh index 924282ed41e..095f5d677a3 100755 --- a/toolchain/scripts/stage0/setup_buildtools.sh +++ b/toolchain/scripts/stage0/setup_buildtools.sh @@ -61,10 +61,6 @@ else fi export LDFLAGS="${TSANFLAGS}" -# get system arch information using OpenBLAS prebuild -${SCRIPTDIR}/get_openblas_arch.sh -load "${BUILDDIR}/openblas_arch" - write_toolchain_env "${INSTALLDIR}" #EOF diff --git a/toolchain/scripts/stage1/install_openmpi.sh b/toolchain/scripts/stage1/install_openmpi.sh index e798cc7325e..f9700776bd4 100755 --- a/toolchain/scripts/stage1/install_openmpi.sh +++ b/toolchain/scripts/stage1/install_openmpi.sh @@ -61,7 +61,7 @@ case "${with_openmpi}" in [ -d openmpi-${openmpi_ver} ] && rm -rf openmpi-${openmpi_ver} tar -xjf ${openmpi_pkg} cd openmpi-${openmpi_ver} - if [ "${OPENBLAS_ARCH}" = "x86_64" ]; then + if [ "${SYSTEM_ARCH}" = "x86_64" ]; then # can have issue with older glibc libraries, in which case # we need to add the -fgnu89-inline to CFLAGS. We can check # the version of glibc using ldd --version, as ldd is part of diff --git a/toolchain/scripts/stage2/install_mkl.sh b/toolchain/scripts/stage2/install_mkl.sh index d0ff9f99239..62d9f0bd6ef 100755 --- a/toolchain/scripts/stage2/install_mkl.sh +++ b/toolchain/scripts/stage2/install_mkl.sh @@ -57,7 +57,7 @@ case "${with_mkl}" in ;; esac if [ "${with_mkl}" != "__DONTUSE__" ]; then - case ${OPENBLAS_ARCH} in + case "${SYSTEM_ARCH}" in x86_64) mkl_arch_dir="intel64" MKL_CFLAGS="-m64" @@ -67,7 +67,7 @@ if [ "${with_mkl}" != "__DONTUSE__" ]; then MKL_CFLAGS="-m32" ;; *) - report_error $LINENO "MKL only supports intel64 (x86_64) and ia32 (i386) at the moment, and your arch obtained from OpenBLAS prebuild is $OPENBLAS_ARCH" + report_error $LINENO "MKL only supports intel64 (x86_64) and ia32 (i386) at the moment, but the detected architecture is ${SYSTEM_ARCH}" exit 1 ;; esac diff --git a/toolchain/scripts/stage2/install_openblas.sh b/toolchain/scripts/stage2/install_openblas.sh index 6c6a8a40866..58cc4a9af5a 100755 --- a/toolchain/scripts/stage2/install_openblas.sh +++ b/toolchain/scripts/stage2/install_openblas.sh @@ -60,62 +60,21 @@ case "${with_openblas}" in [ -d OpenBLAS-${openblas_ver} ] && rm -rf OpenBLAS-${openblas_ver} tar -zxf ${openblas_pkg} cd OpenBLAS-${openblas_ver} - - # First attempt to make openblas using auto detected - # TARGET, if this fails, then make with forced - # TARGET=NEHALEM + # Build OpenBLAS with DYNAMIC_ARCH unless native architecture is requested. + # If the latter fails, then build with DYNAMIC_ARCH. # # wrt NUM_THREADS=64: this is what the most common Linux distros seem to choose atm # for a good compromise between memory usage and scalability # # Unfortunately, NO_SHARED=1 breaks ScaLAPACK build. - case "${TARGET_CPU}" in - "generic") - TARGET="NEHALEM" - ;; - "native") - TARGET=${OPENBLAS_LIBCORE} - ;; - "broadwell" | "skylake") - TARGET="HASWELL" - ;; - "skylake-avx512") - TARGET="SKYLAKEX" - ;; - *) - TARGET=${TARGET_CPU} - ;; - esac - TARGET=$(echo ${TARGET} | tr '[:lower:]' '[:upper:]') - echo "Installing OpenBLAS library for target ${TARGET}" - ( - make -j $(get_nprocs) \ - MAKE_NB_JOBS=0 \ - TARGET=${TARGET} \ - NUM_THREADS=64 \ - USE_THREAD=1 \ - USE_OPENMP=1 \ - NO_AFFINITY=1 \ - CC="${CC}" \ - FC="${FC}" \ - PREFIX="${pkg_install_dir}" \ - > make.log 2>&1 || tail -n ${LOG_LINES} make.log - ) || ( - make -j $(get_nprocs) \ - MAKE_NB_JOBS=0 \ - TARGET=NEHALEM \ - NUM_THREADS=64 \ - USE_THREAD=1 \ - USE_OPENMP=1 \ - NO_AFFINITY=1 \ - CC="${CC}" \ - FC="${FC}" \ - PREFIX="${pkg_install_dir}" \ - > make.nehalem.log 2>&1 || tail -n ${LOG_LINES} make.nehalem.log - ) - make -j $(get_nprocs) \ + BUILD_DYNAMIC=0 + if [ "native" != "${TARGET_CPU}" ]; then + BUILD_DYNAMIC=1 + fi + if [ "${BUILD_DYNAMIC}" = "0" ]; then + echo "Installing OpenBLAS library for native target" + if ! make -j "$(get_nprocs)" \ MAKE_NB_JOBS=0 \ - TARGET=${TARGET} \ NUM_THREADS=64 \ USE_THREAD=1 \ USE_OPENMP=1 \ @@ -123,8 +82,34 @@ case "${with_openblas}" in CC="${CC}" \ FC="${FC}" \ PREFIX="${pkg_install_dir}" \ - install > install.log 2>&1 || tail -n ${LOG_LINES} install.log - cd .. + > make.log 2>&1; then + tail -n "${LOG_LINES}" make.log + make clean + BUILD_DYNAMIC=1 + fi + fi + if [ "${BUILD_DYNAMIC}" != "0" ]; then + echo "Installing OpenBLAS library for dynamic target" + if ! make -j "$(get_nprocs)" \ + MAKE_NB_JOBS=0 \ + DYNAMIC_ARCH=1 \ + NUM_THREADS=64 \ + USE_THREAD=1 \ + USE_OPENMP=1 \ + NO_AFFINITY=1 \ + CC="${CC}" \ + FC="${FC}" \ + PREFIX="${pkg_install_dir}" \ + > make.log 2>&1; then + tail -n "${LOG_LINES}" make.log + exit 1 + fi + fi + if ! make MAKE_NB_JOBS=0 PREFIX="${pkg_install_dir}" install \ + > install.log 2>&1; then + tail -n "${LOG_LINES}" install.log + exit 1 + fi write_checksums "${install_lock_file}" "${SCRIPT_DIR}/stage2/$(basename ${SCRIPT_NAME})" fi OPENBLAS_CFLAGS="-I'${pkg_install_dir}/include'" diff --git a/toolchain/scripts/stage3/install_elpa.sh b/toolchain/scripts/stage3/install_elpa.sh index c0df99c2780..58b4b8bc2a1 100755 --- a/toolchain/scripts/stage3/install_elpa.sh +++ b/toolchain/scripts/stage3/install_elpa.sh @@ -93,7 +93,7 @@ case "$with_elpa" in SSE4_flag="" config_flags="--disable-avx-kernels --disable-avx2-kernels --disable-avx512-kernels --disable-sse-kernels --disable-sse-assembly-kernels" if [ "${TARGET_CPU}" = "native" ]; then - if [ -f /proc/cpuinfo ] && [ "${OPENBLAS_ARCH}" = "x86_64" ]; then + if [ -f /proc/cpuinfo ] && [ "${SYSTEM_ARCH}" = "x86_64" ]; then has_AVX=$(grep '\bavx\b' /proc/cpuinfo 1> /dev/null && echo 'yes' || echo 'no') [ "${has_AVX}" = "yes" ] && AVX_flag="-mavx" || AVX_flag="" has_AVX2=$(grep '\bavx2\b' /proc/cpuinfo 1> /dev/null && echo 'yes' || echo 'no') diff --git a/toolchain/scripts/stage4/install_json.sh b/toolchain/scripts/stage4/install_json.sh new file mode 100755 index 00000000000..384e807ac3d --- /dev/null +++ b/toolchain/scripts/stage4/install_json.sh @@ -0,0 +1,67 @@ +#!/bin/bash -e + +SCRIPT_NAME="${BASH_SOURCE[0]}" +SCRIPT_DIR="$(cd "$(dirname "${SCRIPT_NAME}")/.." && pwd -P)" + +source "${SCRIPT_DIR}/common_vars.sh" +source "${SCRIPT_DIR}/tool_kit.sh" +source "${SCRIPT_DIR}/signal_trap.sh" +source "${SCRIPT_DIR}/package_versions.sh" +load_package_vars "json" +source "${INSTALLDIR}/toolchain.conf" +source "${INSTALLDIR}/toolchain.env" + +rm -f "${BUILDDIR}/setup_json" +mkdir -p "${BUILDDIR}" +cd "${BUILDDIR}" +pkg_install_dir="" + +case "${with_json}" in + __INSTALL__) + echo "==================== Installing nlohmann-json ====================" + filename="json-${json_ver}.tar.xz" + url="https://github.com/nlohmann/json/releases/download/v${json_ver}/json.tar.xz" + pkg_install_dir="${INSTALLDIR}/json-${json_ver}" + install_lock_file="${pkg_install_dir}/install_successful" + if verify_checksums "${install_lock_file}"; then + echo "json-${json_ver} is already installed, skipping it." + else + retrieve_package "${json_sha256}" "${filename}" "${url}" + if [ "${PACK_RUN}" = "__TRUE__" ]; then + echo "--pack-run mode specified, skip installation" + exit 0 + fi + [ -d "json-${json_ver}" ] && rm -rf "json-${json_ver}" + tar -xJf "json-${json_ver}.tar.xz" + cd "json" + mkdir build && cd build + cmake .. \ + -DCMAKE_INSTALL_PREFIX="${pkg_install_dir}" \ + -DJSON_BuildTests=OFF \ + > cmake.log 2>&1 || tail -n "${LOG_LINES}" cmake.log + make install -j "$(get_nprocs)" > install.log 2>&1 || tail -n "${LOG_LINES}" install.log + write_checksums "${install_lock_file}" "${SCRIPT_DIR}/stage4/install_json.sh" + fi + ;; + __SYSTEM__) + # ABACUS resolves the installed configuration with CMake's find_package. + echo "Using system nlohmann-json (resolved by CMake)." + ;; + __DONTUSE__) ;; + *) + pkg_install_dir="${with_json}" + check_dir "${pkg_install_dir}" + ;; +esac + +if [ -n "${pkg_install_dir}" ]; then + cat > "${BUILDDIR}/setup_json" < cmake.log 2>&1 || tail -n ${LOG_LINES} cmake.log - make install -j $(get_nprocs) > make.log 2>&1 || tail -n ${LOG_LINES} make.log - write_checksums "${install_lock_file}" "${SCRIPT_DIR}/stage4/$(basename ${SCRIPT_NAME})" - fi - RAPIDJSON_CFLAGS="-I'${pkg_install_dir}/include'" - ;; - __SYSTEM__) - echo "==================== Finding RapidJSON from system paths ====================" - if [ "${PACK_RUN}" = "__TRUE__" ]; then - echo "--pack-run mode specified, skip system check" - exit 0 - fi - # Find rapidjson header file and derive package root directory - rapidjson_header_path="$(find_in_paths "rapidjson/rapidjson.h" $INCLUDE_PATHS)" - if [ "$rapidjson_header_path" != "__FALSE__" ]; then - # Derive pkg_install_dir from found header path - # rapidjson/rapidjson.h -> remove /rapidjson/rapidjson.h -> get include dir -> get parent dir - rapidjson_include_dir="$(dirname "$(dirname "$rapidjson_header_path")")" - pkg_install_dir="$(dirname "$rapidjson_include_dir")" - echo "Found rapidjson at: ${pkg_install_dir}" - RAPIDJSON_CFLAGS="-I'${rapidjson_include_dir}'" - else - report_error "Cannot find rapidjson/rapidjson.h in system paths" - exit 1 - fi - ;; - __DONTUSE__) ;; - - *) - echo "==================== Linking RapidJSON to user paths ====================" - pkg_install_dir="${with_rapidjson}" - check_dir "${pkg_install_dir}" - RAPIDJSON_CFLAGS="-I'${pkg_install_dir}/include'" - ;; -esac -if [ "$with_rapidjson" != "__DONTUSE__" ]; then - if [ "$with_rapidjson" != "__SYSTEM__" ]; then - cat << EOF > "${BUILDDIR}/setup_rapidjson" -prepend_path CMAKE_PREFIX_PATH "${pkg_install_dir}" -EOF - fi - cat << EOF >> "${BUILDDIR}/setup_rapidjson" -export RAPIDJSON_ROOT="${pkg_install_dir}" -EOF - filter_setup "${BUILDDIR}/setup_rapidjson" $SETUPFILE -fi - -load "${BUILDDIR}/setup_rapidjson" -write_toolchain_env "${INSTALLDIR}" - -cd "${ROOTDIR}" -report_timing "rapidjson" diff --git a/toolchain/scripts/stage4/install_stage4.sh b/toolchain/scripts/stage4/install_stage4.sh index 0980d809e5e..ffeefdcb761 100755 --- a/toolchain/scripts/stage4/install_stage4.sh +++ b/toolchain/scripts/stage4/install_stage4.sh @@ -5,7 +5,7 @@ ./scripts/stage4/install_dftd4.sh ./scripts/stage4/install_cereal.sh -./scripts/stage4/install_rapidjson.sh +./scripts/stage4/install_json.sh ./scripts/stage4/install_libtorch.sh ./scripts/stage4/install_libnpy.sh ./scripts/stage4/install_libri.sh diff --git a/toolchain/scripts/tool_kit.sh b/toolchain/scripts/tool_kit.sh index 361d915ea89..677bb07df64 100755 --- a/toolchain/scripts/tool_kit.sh +++ b/toolchain/scripts/tool_kit.sh @@ -126,7 +126,7 @@ Also. the --pack-run option can help in OFFLINE installation. You can manually install requirements packages via: 0. Download by 'wget $__url -O $__filename' manually 1. Download from www.cp2k.org/static/downloads (for OpenBLAS, OpenMPI and Others) -2. Download from github.com (especially for CEREAL, RapidJSON, libnpy, LibRI and other stage4 packages) +2. Download from github.com (especially for CEREAL, nlohmann-json, libnpy, LibRI and other stage4 packages) 3. for Intel-oneAPI and AMD AOCC/AOCL, please contact your server manager or visit their official website 4. For users in China, you can try Gitee mirror: git clone https://gitee.com/jamesmisaka/abacus_toolchain_build.git EOF diff --git a/toolchain/tests/test_rapidjson_cmake.sh b/toolchain/tests/test_rapidjson_cmake.sh deleted file mode 100755 index c7460257d82..00000000000 --- a/toolchain/tests/test_rapidjson_cmake.sh +++ /dev/null @@ -1,123 +0,0 @@ -#!/usr/bin/env bash -set -u - -REPO_ROOT="$(cd "$(dirname "${BASH_SOURCE[0]}")/../.." && pwd -P)" -FAILURES=0 - -fail() { - printf 'FAIL: %s\n' "$*" >&2 - FAILURES=$((FAILURES + 1)) -} - -write_target_package() { - local prefix="$1" - mkdir -p "${prefix}/include/rapidjson" "${prefix}/lib/cmake/RapidJSON" - printf '#pragma once\n' >"${prefix}/include/rapidjson/document.h" - cat >"${prefix}/lib/cmake/RapidJSON/RapidJSONConfig.cmake" <<'EOF' -message(STATUS "Loaded fake RapidJSON target package") -get_filename_component(RapidJSON_CMAKE_DIR "${CMAKE_CURRENT_LIST_FILE}" PATH) -get_filename_component(_rapidjson_prefix "${RapidJSON_CMAKE_DIR}/../../.." ABSOLUTE) -if(NOT TARGET RapidJSON) - add_library(RapidJSON INTERFACE IMPORTED) - set_property(TARGET RapidJSON PROPERTY INTERFACE_INCLUDE_DIRECTORIES "${_rapidjson_prefix}/include") -endif() -EOF -} - -write_fake_mkl() { - local prefix="$1" - mkdir -p "${prefix}/include" "${prefix}/lib" - printf '#pragma once\n' >"${prefix}/include/mkl_service.h" - : >"${prefix}/lib/libmkl_core.so" - : >"${prefix}/lib/libmkl_gf_lp64.so" - : >"${prefix}/lib/libmkl_gnu_thread.so" -} - -write_variable_only_package() { - local prefix="$1" - mkdir -p "${prefix}/include/rapidjson" "${prefix}/lib/cmake/RapidJSON" - printf '#pragma once\n' >"${prefix}/include/rapidjson/document.h" - cat >"${prefix}/lib/cmake/RapidJSON/RapidJSONConfig.cmake" <<'EOF' -get_filename_component(RapidJSON_CMAKE_DIR "${CMAKE_CURRENT_LIST_FILE}" PATH) -get_filename_component(_rapidjson_prefix "${RapidJSON_CMAKE_DIR}/../../.." ABSOLUTE) -set(RapidJSON_INCLUDE_DIR "${_rapidjson_prefix}/include") -set(RapidJSON_INCLUDE_DIRS "${RapidJSON_INCLUDE_DIR}") -EOF -} - -run_top_level_configure() { - local build_dir="$1" - local prefix="$2" - local mkl_root="$3" - - cmake -S "$REPO_ROOT" -B "$build_dir" \ - -DENABLE_RAPIDJSON=ON \ - -DENABLE_LCAO=OFF \ - -DENABLE_MPI=OFF \ - -DENABLE_OPENMP=OFF \ - -DMKLROOT="$mkl_root" \ - -DCMAKE_PREFIX_PATH="$prefix" \ - >"${build_dir}.log" 2>&1 -} - -test_top_level_accepts_rapidjson_target_package() { - local tmpdir prefix mkl_root build_dir status - tmpdir="$(mktemp -d)" - prefix="${tmpdir}/prefix" - mkl_root="${tmpdir}/mkl" - build_dir="${tmpdir}/target-build" - - write_target_package "$prefix" - write_fake_mkl "$mkl_root" - run_top_level_configure "$build_dir" "$prefix" "$mkl_root" - status=$? - - if ! grep -Fq "Loaded fake RapidJSON target package" "${build_dir}.log"; then - cat "${build_dir}.log" >&2 - fail "top-level CMake did not load the fake RapidJSON target package" - elif grep -Fq "RapidJSON was found, but target RapidJSON is missing" "${build_dir}.log"; then - cat "${build_dir}.log" >&2 - fail "top-level CMake rejected RapidJSON target package as target-missing" - elif grep -Fq 'Could not find a package configuration file provided by "RapidJSON"' "${build_dir}.log"; then - cat "${build_dir}.log" >&2 - fail "top-level CMake did not find the fake RapidJSON target package" - elif [[ "$status" -ne 0 ]]; then - cat "${build_dir}.log" >&2 - fail "top-level CMake failed with a RapidJSON target package" - fi - - rm -rf "$tmpdir" -} - -test_top_level_rejects_variable_only_package() { - local tmpdir prefix mkl_root build_dir status - tmpdir="$(mktemp -d)" - prefix="${tmpdir}/prefix" - mkl_root="${tmpdir}/mkl" - build_dir="${tmpdir}/variable-build" - - write_variable_only_package "$prefix" - write_fake_mkl "$mkl_root" - run_top_level_configure "$build_dir" "$prefix" "$mkl_root" - status=$? - - if [[ "$status" -eq 0 ]]; then - cat "${build_dir}.log" >&2 - fail "top-level CMake configured with variable-only RapidJSON package; expected target-missing failure" - elif ! grep -Fq "RapidJSON was found, but target RapidJSON is missing." "${build_dir}.log"; then - cat "${build_dir}.log" >&2 - fail "top-level CMake failed for the wrong reason with variable-only RapidJSON package" - fi - - rm -rf "$tmpdir" -} - -test_top_level_accepts_rapidjson_target_package -test_top_level_rejects_variable_only_package - -if [[ "$FAILURES" -ne 0 ]]; then - printf '%s RapidJSON CMake test(s) failed\n' "$FAILURES" >&2 - exit 1 -fi - -printf 'RapidJSON CMake tests passed\n' diff --git a/toolchain/toolchain_aocc-aocl.sh b/toolchain/toolchain_aocc-aocl.sh index 83ad0a1c841..17e2cd1cfec 100755 --- a/toolchain/toolchain_aocc-aocl.sh +++ b/toolchain/toolchain_aocc-aocl.sh @@ -36,7 +36,7 @@ WITH_ELPA="install" # Utility Libraries WITH_CEREAL="install" -WITH_RAPIDJSON="install" +WITH_JSON="install" # Advanced Features (EXX calculations) WITH_LIBRI="install" @@ -101,7 +101,7 @@ run_toolchain_with_log compile.log ./install_abacus_toolchain_new.sh \ --with-fftw="$WITH_FFTW" \ --with-elpa="$WITH_ELPA" \ --with-cereal="$WITH_CEREAL" \ - --with-rapidjson="$WITH_RAPIDJSON" \ + --with-json="$WITH_JSON" \ --with-libtorch="$WITH_LIBTORCH" \ --with-nep="$WITH_NEP" \ --with-libnpy="$WITH_LIBNPY" \ diff --git a/toolchain/toolchain_gcc-aocl.sh b/toolchain/toolchain_gcc-aocl.sh index 1238defd361..6ef8789665d 100755 --- a/toolchain/toolchain_gcc-aocl.sh +++ b/toolchain/toolchain_gcc-aocl.sh @@ -36,7 +36,7 @@ WITH_ELPA="install" # Utility Libraries WITH_CEREAL="install" -WITH_RAPIDJSON="install" +WITH_JSON="install" # Advanced Features (EXX calculations) WITH_LIBRI="install" @@ -98,7 +98,7 @@ run_toolchain_with_log compile.log ./install_abacus_toolchain_new.sh \ --with-fftw="$WITH_FFTW" \ --with-elpa="$WITH_ELPA" \ --with-cereal="$WITH_CEREAL" \ - --with-rapidjson="$WITH_RAPIDJSON" \ + --with-json="$WITH_JSON" \ --with-libtorch="$WITH_LIBTORCH" \ --with-nep="$WITH_NEP" \ --with-libnpy="$WITH_LIBNPY" \ diff --git a/toolchain/toolchain_gcc-mkl.sh b/toolchain/toolchain_gcc-mkl.sh index df0fc04d108..a44bd5d710a 100755 --- a/toolchain/toolchain_gcc-mkl.sh +++ b/toolchain/toolchain_gcc-mkl.sh @@ -36,7 +36,7 @@ WITH_ELPA="install" # Utility Libraries WITH_CEREAL="install" -WITH_RAPIDJSON="install" +WITH_JSON="install" # Advanced Features (EXX calculations) WITH_LIBRI="install" @@ -101,7 +101,7 @@ run_toolchain_with_log compile.log ./install_abacus_toolchain_new.sh \ --with-elpa="$WITH_ELPA" \ --with-dftd4="$WITH_DFTD4" \ --with-cereal="$WITH_CEREAL" \ - --with-rapidjson="$WITH_RAPIDJSON" \ + --with-json="$WITH_JSON" \ --with-libtorch="$WITH_LIBTORCH" \ --with-nep="$WITH_NEP" \ --with-libnpy="$WITH_LIBNPY" \ diff --git a/toolchain/toolchain_gnu.sh b/toolchain/toolchain_gnu.sh index e297c0e453b..e4b6359afba 100755 --- a/toolchain/toolchain_gnu.sh +++ b/toolchain/toolchain_gnu.sh @@ -37,7 +37,7 @@ WITH_ELPA="install" # Utility Libraries WITH_CEREAL="install" -WITH_RAPIDJSON="install" +WITH_JSON="install" # Advanced Features (EXX calculations) WITH_LIBRI="install" @@ -104,7 +104,7 @@ run_toolchain_with_log compile.log ./install_abacus_toolchain_new.sh \ --with-elpa="$WITH_ELPA" \ --with-dftd4="$WITH_DFTD4" \ --with-cereal="$WITH_CEREAL" \ - --with-rapidjson="$WITH_RAPIDJSON" \ + --with-json="$WITH_JSON" \ --with-libtorch="$WITH_LIBTORCH" \ --with-nep="$WITH_NEP" \ --with-libnpy="$WITH_LIBNPY" \ diff --git a/toolchain/toolchain_intel.sh b/toolchain/toolchain_intel.sh index 9f854a0b4a1..4ef5166b55f 100755 --- a/toolchain/toolchain_intel.sh +++ b/toolchain/toolchain_intel.sh @@ -37,7 +37,7 @@ WITH_ELPA="install" # Utility Libraries WITH_CEREAL="install" -WITH_RAPIDJSON="install" +WITH_JSON="install" # Advanced Features (EXX calculations) WITH_LIBRI="install" @@ -122,7 +122,7 @@ run_toolchain_with_log compile.log ./install_abacus_toolchain_new.sh \ --with-elpa="$WITH_ELPA" \ --with-dftd4="$WITH_DFTD4" \ --with-cereal="$WITH_CEREAL" \ - --with-rapidjson="$WITH_RAPIDJSON" \ + --with-json="$WITH_JSON" \ --with-libtorch="$WITH_LIBTORCH" \ --with-nep="$WITH_NEP" \ --with-libnpy="$WITH_LIBNPY" \ diff --git a/tools/02_postprocessing/plot-tools/README.md b/tools/02_postprocessing/plot-tools/README.md index aac183260b1..768db546486 100644 --- a/tools/02_postprocessing/plot-tools/README.md +++ b/tools/02_postprocessing/plot-tools/README.md @@ -15,7 +15,7 @@ - [NumPy](https://numpy.org/) - [Matplotlib](https://matplotlib.org/) -- [lxml](https://lxml.de/) +- [lxml](https://lxml.de/) (only required for parsing the XML projected-band output) - [setuptools](https://setuptools.pypa.io/en/latest/index.html) ## Installation @@ -100,7 +100,7 @@ First, prepare a file named 'config.json' in json format: ```json { - "pdosfile": "PDOS", + "pdosfile": "OUT.ABACUS/pdos", "efermi": 6.585653952007503, "energy_range": [-5, 7], "dos_range": [0, 5], @@ -119,7 +119,7 @@ If you only want to plot total DOS, you can modify `pdosfile` to `tdosfile` and | Property | Type | Note | | :------------: | :-------------------------------------------------------------------------: | :-------------------------------------------------------------------------------------------------------------------------------------------------------------------: | | _tdosfile_ | `str` | Total DOS data file output from ABACUS | -| _pdosfile_ | `str` | Partial DOS data file output from ABACUS in xml format | +| _pdosfile_ | `str` | Directory or path prefix containing the text PDOS files `pdoss1_*` (and `pdoss2_*` for spin-polarized runs) output from ABACUS | | _efermi_ | `float` | Fermi level in eV | | _energy_range_ | `list` | Range of energy in eV | | _shift_ | `bool` | If set `'true'`, it will evaluate band gap. Default: `'false'` | diff --git a/tools/02_postprocessing/plot-tools/abacus_plot/band.py b/tools/02_postprocessing/plot-tools/abacus_plot/band.py index c886da0619f..42eda2ba2e7 100644 --- a/tools/02_postprocessing/plot-tools/abacus_plot/band.py +++ b/tools/02_postprocessing/plot-tools/abacus_plot/band.py @@ -671,7 +671,7 @@ def _seg_plot(bandplot, lc, file_dir, name): for mag in wei[elem][ang].keys(): bandplot = BandPlot(fig, ax, **kwargs) m_index = int(mag) - bandplot._label = f"{elem}-{get_angular_momentum_name(l_index, m_index)}" + bandplot._label = f"{elem}-{get_angular_momentum_name(l_index, m_index, m_is_physical=False)}" for ib in range(self.nbands): points = np.array( (self.k_index, energy[0:, ib])).T.reshape(-1, 1, 2) @@ -836,7 +836,7 @@ def _plot_contributions(self, for mag in wei[elem][ang].keys(): m_index = int(mag) whole_label_parsed.append( - f"{elem}-{get_angular_momentum_name(l_index, m_index)}") + f"{elem}-{get_angular_momentum_name(l_index, m_index, m_is_physical=False)}") whole_data_parsed.append(wei[elem][ang][mag]) else: diff --git a/tools/02_postprocessing/plot-tools/abacus_plot/dos.py b/tools/02_postprocessing/plot-tools/abacus_plot/dos.py index a2d74904c51..3387b609029 100644 --- a/tools/02_postprocessing/plot-tools/abacus_plot/dos.py +++ b/tools/02_postprocessing/plot-tools/abacus_plot/dos.py @@ -6,6 +6,8 @@ ''' from collections import OrderedDict, namedtuple +import glob +import math import numpy as np from os import PathLike from pathlib import Path @@ -172,14 +174,42 @@ def _read(self) -> tuple: :params tdosfile: string of TDOS data file """ - - data = np.loadtxt(self.tdosfile) - self.nspin = data.shape[1]-1 - if self.nspin == 1: - self.energy, self.dos = np.split(data, self.nspin+1, axis=1) - elif self.nspin == 2: - self.energy, dos_up, dos_dw = np.split(data, self.nspin+1, axis=1) + rows = [] + with open(self.tdosfile) as f: + for line in f: + if line.startswith("#"): + continue + # strip inline comments, e.g. "1 # ionic step" + line = line.split("#")[0] + if not line.split(): + continue + rows.append([float(x) for x in line.split()]) + if not rows: + raise ValueError(f"Empty TDOS file: {self.tdosfile}") + # New format has two header lines ("ionic step" / "number of points") + # with a single column; data lines have 5 columns per spin channel. + # Old format (DOS1_smearing.dat) has only energy + dos columns. + ncols = len(rows[-1]) + data = np.asarray([r for r in rows if len(r) == ncols]) + if ncols == 2: + self.nspin = 1 + self.energy, self.dos = np.hsplit(data, 2) + elif ncols == 3: + self.nspin = 2 + self.energy, dos_up, dos_dw = np.hsplit(data, 3) + self.dos = np.hstack((dos_up, dos_dw)) + elif ncols == 5: + self.nspin = 1 + self.energy, self.dos, self.dos_int, self.dos_smear, self.dos_smear_int = np.hsplit(data, 5) + elif ncols == 9: + self.nspin = 2 + self.energy, dos_up, dos_dw, dos_int_up, dos_int_dw, smear_up, smear_dw, smear_int_up, smear_int_dw = np.hsplit(data, 9) self.dos = np.hstack((dos_up, dos_dw)) + self.dos_int = np.hstack((dos_int_up, dos_int_dw)) + self.dos_smear = np.hstack((smear_up, smear_dw)) + self.dos_smear_int = np.hstack((smear_int_up, smear_int_dw)) + else: + raise ValueError(f"Unrecognized TDOS format: {ncols} columns") def _shift_energy(self, efermi: float = 0, shift: bool = False, prec: float = 0.01): if shift: @@ -208,47 +238,174 @@ def plot(self, fig: Figure, ax: Union[axes.Axes, Sequence[axes.Axes]], efermi: f class PDOS(DOS): - """Parse partial DOS data""" + """Parse partial DOS data from the text PDOS files written by ABACUS. + + The text format (one file per spin channel) looks like:: + + # istep: 1 (optional) + # npoints: 2736 + # energy(eV) atom species pdos(1/eV), columns: s(m=0) p(m=0,+1,-1) ... + -55.607730 1 Fe 0.000000 0.000000 ... + + Each data row is one (energy, atom) pair. After the fixed leading columns + ``energy / atom_index / species`` come the PDOS columns of that atom, + expanded as ``s`` (1 column), ``p`` (3 columns, m=0,+1,-1), ``d`` (5), + ``f`` (7). The number of PDOS columns therefore depends on the highest + angular momentum of the atom's basis, i.e. ``(nwl+1)**2`` for a basis with + continuous ``l = 0..nwl``. Zeta components are already summed by ABACUS. + + ``pdosfile`` may point to a single spin-channel file + (``pdoss1_*.txt``), to the spin-up file of a spin-polarized pair, or to a + path/prefix from which the ``pdoss1_*`` and (optionally) ``pdoss2_*`` + files are discovered automatically. + """ + + # physical m value for each column index 0..2l of angular momentum l: + # column 0 -> m=0, 1 -> +1, 2 -> -1, 3 -> +2, 4 -> -2, ... + @staticmethod + def _col_to_m(l: int, col: int) -> int: + if col == 0: + return 0 + return (col + 1) // 2 if col % 2 == 1 else -(col // 2) def __init__(self, pdosfile: PathLike = None) -> None: self.pdosfile = pdosfile self._read() super().__init__(self.nspin) + def _find_spin_files(self) -> List[str]: + """Locate the per-spin text PDOS files. + + Returns a list with one (nspin=1) or two (nspin=2) file paths. + """ + path = Path(self.pdosfile) + if path.is_file(): + name = path.name + if name.startswith("pdoss"): + # A concrete spin-channel file e.g. pdoss1g1_nao.txt: use it + # directly as the spin-1 channel and look for the matching + # spin-2 file by replacing "s1" with "s2" in the filename. + spin1 = str(path) + name2 = name.replace("pdoss1", "pdoss2", 1) + spin2 = str(path.parent / name2) + if spin2 != spin1 and Path(spin2).is_file(): + return [spin1, spin2] + return [spin1] + else: + # a single arbitrary file; treat as the only spin channel + return [str(path)] + else: + # a directory (e.g. OUT.ABACUS) or a prefix such as ".../pdos" + prefix = str(path / 'pdos') if path.is_dir() else str(path) + + parent = Path(prefix).parent + base = Path(prefix).name + candidates = [] + for spin in (1, 2): + pattern = str(parent) + "/" + base + "s" + str(spin) + "*.txt" + matches = sorted(glob.glob(pattern)) + if matches: + candidates.append(matches[0]) + elif spin == 1: + raise FileNotFoundError( + f"No text PDOS file matching '{pattern}'") + return candidates + + def _read_one(self, fname: str): + """Parse one spin-channel text file. + + Returns (energy, orbitals) where energy is (npoints, 1) and orbitals is + a list of dicts with keys index/atom_index/species/l/m and a + (npoints, 1) ``data`` array. ``index`` is filled for API compatibility. + """ + rows = [] + with open(fname) as f: + for line in f: + if line.startswith("#") or not line.split(): + continue + rows.append(line.split()) + if not rows: + raise ValueError(f"Empty PDOS file: {fname}") + + energy_col = np.asarray([float(r[0]) for r in rows], dtype=float) + atom_col = np.asarray([int(r[1]) for r in rows], dtype=int) + species_col = [r[2] for r in rows] + pdos_cols = [np.asarray(r[3:], dtype=float) for r in rows] + + # rows are ordered as: outer loop over energy points, inner over atoms + nat = int(atom_col.max()) + npoints = len(rows) // nat + energy = energy_col[:npoints].reshape(-1, 1) + + # number of PDOS columns per atom (1-based atom index) + ncols = {} + for r in rows: + iat = int(r[1]) + ncols.setdefault(iat, len(r) - 3) + + # build one orbital entry per (atom, l, m) column + orbitals = [] + orb_index = 0 + for iat in range(1, nat + 1): + ncol = ncols[iat] + # continuous l = 0..nwl -> ncol = (nwl+1)**2 + nwl = int(round(math.sqrt(ncol))) - 1 + if (nwl + 1) ** 2 != ncol: + raise ValueError( + f"Cannot infer angular momenta from {ncol} PDOS columns " + f"for atom {iat} in {fname}") + + # slice this atom's data block: rows [ (iat-1) :: nat ] + block = np.asarray( + [pdos_cols[k] for k in range(len(rows)) if atom_col[k] == iat], + dtype=float) + species = next(sp for k, sp in enumerate(species_col) + if atom_col[k] == iat) + + col = 0 + for l in range(nwl + 1): + for mcol in range(2 * l + 1): + orb = OrderedDict() + # 1-based global orbital index, matching the legacy XML format + orb['index'] = orb_index + 1 + orb['atom_index'] = iat + orb['species'] = species + orb['l'] = l + orb['m'] = self._col_to_m(l, mcol) + orb['data'] = block[:, col].reshape(-1, 1) + orbitals.append(orb) + orb_index += 1 + col += 1 + + return energy, orbitals + def _read(self): - """Read partial DOS data file + """Read partial DOS data file(s) - :params pdosfile: string of PDOS data file + :params pdosfile: path of a text PDOS file, a spin-up file of a + spin-polarized pair, or a path/prefix used to discover the + ``pdoss1_*``/``pdoss2_*`` files. """ + spin_files = self._find_spin_files() + self.nspin = len(spin_files) - from lxml import etree - pdosdata = etree.parse(self.pdosfile) - root = pdosdata.getroot() - self.nspin = int(root.xpath('//nspin')[0].text.replace(' ', '')) - norbitals = int(root.xpath('//norbitals')[0].text.replace(' ', '')) - self.eunit = root.xpath('//energy_values/@units')[0].replace(' ', '') - e_list = root.xpath( - '//energy_values')[0].text.replace(' ', '').split('\n') - remove_empty(e_list) - self.orbitals = [] - for i in range(norbitals): - orb = OrderedDict() - orb['index'] = int(root.xpath( - '//orbital/@index')[i].replace(' ', '')) - orb['atom_index'] = int(root.xpath( - '//orbital/@atom_index')[i].replace(' ', '')) - orb['species'] = root.xpath( - '//orbital/@species')[i].replace(' ', '') - orb['l'] = int(root.xpath('//orbital/@l')[i].replace(' ', '')) - orb['m'] = int(root.xpath('//orbital/@m')[i].replace(' ', '')) - orb['z'] = int(root.xpath('//orbital/@z')[i].replace(' ', '')) - data = root.xpath('//data')[i].text.split('\n') - data = handle_data(data) - remove_empty(data) - orb['data'] = np.asarray(data, dtype=float) - self.orbitals.append(orb) - - self.energy = np.reshape(e_list, (-1, 1)).astype(float) + energy, orbitals_up = self._read_one(spin_files[0]) + self.energy = energy + + if self.nspin == 1: + self.orbitals = orbitals_up + else: + energy_dw, orbitals_dw = self._read_one(spin_files[1]) + if len(orbitals_up) != len(orbitals_dw): + raise ValueError( + "Spin-up and spin-down PDOS files have different orbitals") + # concatenate the two spin channels along axis=1 so that + # orb['data'] has shape (npoints, 2), matching the downstream + # split logic in DOS._plot / _write + self.orbitals = [] + for ou, od in zip(orbitals_up, orbitals_dw): + ou['data'] = np.hstack((ou['data'], od['data'])) + self.orbitals.append(ou) def _all_sum(self) -> Tuple[np.ndarray, int]: res = np.zeros_like(self.orbitals[0]["data"], dtype=float) @@ -281,7 +438,7 @@ def _write(self, species: Union[Sequence[Any], Dict[Any, List[int]], Dict[Any, D for orb in self.orbitals: if orb[keyname] == elem: header_list.append( - f"\tAdd data for index ={orb['index']:4d}, atom_index ={orb['atom_index']:4d}, element ={orb['species']:4s}, l,m,z={orb['l']:3d}, {orb['m']:3d}, {orb['z']:3d}") + f"\tAdd data for index ={orb['index']:4d}, atom_index ={orb['atom_index']:4d}, element ={orb['species']:4s}, l,m={orb['l']:3d}, {orb['m']:3d}") header_list.append('') header_list.append('\tEnergy'+10*' ' + 'spin 1'+8*' '+'spin 2') @@ -308,7 +465,7 @@ def _write(self, species: Union[Sequence[Any], Dict[Any, List[int]], Dict[Any, D for orb in self.orbitals: if orb[keyname] == elem and orb["l"] == l_index and orb["m"] == m_index: header_list.append( - f"\tAdd data for index ={orb['index']:4d}, atom_index ={orb['atom_index']:4d}, element ={orb['species']:4s}, l,m,z={orb['l']:3d}, {orb['m']:3d}, {orb['z']:3d}") + f"\tAdd data for index ={orb['index']:4d}, atom_index ={orb['atom_index']:4d}, element ={orb['species']:4s}, l,m={orb['l']:3d}, {orb['m']:3d}") header_list.append('') header_list.append( '\tEnergy'+10*' '+'spin 1'+8*' '+'spin 2') @@ -327,7 +484,7 @@ def _write(self, species: Union[Sequence[Any], Dict[Any, List[int]], Dict[Any, D for orb in self.orbitals: if orb[keyname] == elem and orb["l"] == l_index: header_list.append( - f"\tAdd data for index ={orb['index']:4d}, atom_index ={orb['atom_index']:4d}, element ={orb['species']:4s}, l,m,z={orb['l']:3d}, {orb['m']:3d}, {orb['z']:3d}") + f"\tAdd data for index ={orb['index']:4d}, atom_index ={orb['atom_index']:4d}, element ={orb['species']:4s}, l,m={orb['l']:3d}, {orb['m']:3d}") header_list.append('') header_list.append( '\tEnergy'+10*' '+'spin 1'+8*' '+'spin 2') @@ -406,7 +563,7 @@ def _parial_plot(self, energy_f, tdos = self._shift_energy(efermi, shift, prec) if not species: - dosplot = DOSPlot(fig, ax, self.nspin, **kwargs) + dosplot = DOSPlot(fig, ax[0], self.nspin, **kwargs) dosplot.ax = self._plot(dosplot, energy_f, tdos, "TDOS") if "notes" in dosplot.plot_params.keys(): dosplot._set_figure(energy_range, dos_range, @@ -417,7 +574,7 @@ def _parial_plot(self, return dosplot if isinstance(species, (list, tuple)): - dosplot = DOSPlot(fig, ax, self.nspin, **kwargs) + dosplot = DOSPlot(fig, ax[0], self.nspin, **kwargs) if "xlabel_params" in dosplot.plot_params.keys(): dosplot.ax.set_xlabel("Energy(eV)", ** dosplot.plot_params["xlabel_params"]) @@ -525,10 +682,11 @@ def plot(self, # energy_range=energy_range, dos_range=dos_range, notes={'s': '(a)'}) # fig.savefig("tdos.png") - pdosfile = r"../examples/Si/PDOS" + # directory or prefix containing the text PDOS files pdoss1_*[/pdoss2_*] + pdosfile = "./pdos" pdos = PDOS(pdosfile) #species = {"Ag": [2], "Cl": [1], "In": [0]} - atom_index = {1: {1: [0, 1]}} + atom_index = {1: [0, 1]} fig, ax = plt.subplots(1, 1, sharex=True) energy_range = [-5, 7] efermi = 6.585653952007503 diff --git a/tools/02_postprocessing/plot-tools/abacus_plot/utils.py b/tools/02_postprocessing/plot-tools/abacus_plot/utils.py index 36ba1315048..b42610c1bb0 100644 --- a/tools/02_postprocessing/plot-tools/abacus_plot/utils.py +++ b/tools/02_postprocessing/plot-tools/abacus_plot/utils.py @@ -253,21 +253,42 @@ def get_angular_momentum_label(l_index: int) -> str: ['$s$'], ['$p_z$', '$p_x$', '$p_y$'], ['$d_{z^2}$', '$d_{xz}$', '$d_{yz}$', '$d_{x^2-y^2}$', '$d_{xy}$'], - ['$f_{z^3}$', '$f_{xz^2}$', '$f_{yz^2}$', '$f_{z(x^2-y^2)}$', + ['$f_{z^3}$', '$f_{xz^2}$', '$f_{yz^2}$', '$f_{z(x^2-y^2)}$', '$f_{xyz}$', '$f_{x(x^2-3y^2)}$', '$f_{y(3x^2-y^2)}$'], ['$g_1$', '$g_2$', '$g_3$', '$g_4$', '$g_5$', '$g_6$', '$g_7$', '$g_8$', '$g_9$'] ] -def get_angular_momentum_name(l_index: int, m_index: int) -> str: - """Atomic orbital angular momentum name from l_index and m_index +def _m_to_col(m: int) -> int: + """Column index 0..2l of a physical magnetic quantum number m. - :params l_index: 0 or 1 or 2 or 3 or 4 - :params m_index: 0 ... 2*l_index + ABACUS stores real spherical harmonics in the order + (m=0), (m=+1), (m=-1), (m=+2), (m=-2), ..., so column 1 -> m=+1, + column 2 -> m=-1, column 3 -> m=+2, column 4 -> m=-2, etc. """ + if m == 0: + return 0 + return 2 * m - 1 if m > 0 else -2 * m + + +def get_angular_momentum_name(l_index: int, m: int, m_is_physical: bool = True) -> str: + """Atomic orbital angular momentum name from l_index and m. - return angular_momentum_name[l_index][m_index] + :params l_index: 0 or 1 or 2 or 3 or 4 + :params m: magnetic quantum number. When m_is_physical is True, m is the + physical value (-l_index <= m <= l_index); otherwise m is the column + index 0..2*l_index as stored in the legacy projected-band XML. + :params m_is_physical: whether m is a physical quantum number (True, + default) or a 0..2l column index (False, for the XML PBand format). + """ + if m_is_physical: + if abs(m) > l_index: + raise ValueError(f"|m|={abs(m)} exceeds l={l_index}") + col = _m_to_col(m) + else: + col = m + return angular_momentum_name[l_index][col] def parse_projected_data(orbitals, species: Union[Sequence[Any], Dict[Any, List[int]], Dict[Any, Dict[Any, List[int]]]], keyname=''): diff --git a/tools/02_postprocessing/plot-tools/examples/Si/PDOS b/tools/02_postprocessing/plot-tools/examples/Si/PDOS deleted file mode 100644 index 97abd9e6278..00000000000 --- a/tools/02_postprocessing/plot-tools/examples/Si/PDOS +++ /dev/null @@ -1,61447 +0,0 @@ - -1 -26 - - -5.50311 - -5.49311 - -5.48311 - -5.47311 - -5.46311 - -5.45311 - -5.44311 - -5.43311 - -5.42311 - -5.41311 - -5.40311 - -5.39311 - -5.38311 - -5.37311 - -5.36311 - -5.35311 - -5.34311 - -5.33311 - -5.32311 - -5.31311 - -5.30311 - -5.29311 - -5.28311 - -5.27311 - -5.26311 - -5.25311 - -5.24311 - -5.23311 - -5.22311 - -5.21311 - -5.20311 - -5.19311 - -5.18311 - -5.17311 - -5.16311 - -5.15311 - -5.14311 - -5.13311 - -5.12311 - -5.11311 - -5.10311 - -5.09311 - -5.08311 - -5.07311 - -5.06311 - -5.05311 - -5.04311 - -5.03311 - -5.02311 - -5.01311 - -5.00311 - -4.99311 - -4.98311 - -4.97311 - -4.96311 - -4.95311 - -4.94311 - -4.93311 - -4.92311 - -4.91311 - -4.90311 - -4.89311 - -4.88311 - -4.87311 - -4.86311 - -4.85311 - -4.84311 - -4.83311 - -4.82311 - -4.81311 - -4.80311 - -4.79311 - -4.78311 - -4.77311 - -4.76311 - -4.75311 - -4.74311 - -4.73311 - -4.72311 - -4.71311 - -4.70311 - -4.69311 - -4.68311 - -4.67311 - -4.66311 - -4.65311 - -4.64311 - -4.63311 - -4.62311 - -4.61311 - -4.60311 - -4.59311 - -4.58311 - -4.57311 - -4.56311 - -4.55311 - -4.54311 - -4.53311 - -4.52311 - -4.51311 - -4.50311 - -4.49311 - -4.48311 - -4.47311 - -4.46311 - -4.45311 - -4.44311 - -4.43311 - -4.42311 - -4.41311 - -4.40311 - -4.39311 - -4.38311 - -4.37311 - -4.36311 - -4.35311 - -4.34311 - -4.33311 - -4.32311 - -4.31311 - -4.30311 - -4.29311 - -4.28311 - -4.27311 - -4.26311 - -4.25311 - -4.24311 - -4.23311 - -4.22311 - -4.21311 - -4.20311 - -4.19311 - -4.18311 - -4.17311 - -4.16311 - -4.15311 - -4.14311 - -4.13311 - -4.12311 - -4.11311 - -4.10311 - -4.09311 - -4.08311 - -4.07311 - -4.06311 - -4.05311 - -4.04311 - -4.03311 - -4.02311 - -4.01311 - -4.00311 - -3.99311 - -3.98311 - -3.97311 - -3.96311 - -3.95311 - -3.94311 - -3.93311 - -3.92311 - -3.91311 - -3.90311 - -3.89311 - -3.88311 - -3.87311 - -3.86311 - -3.85311 - -3.84311 - -3.83311 - -3.82311 - -3.81311 - -3.80311 - -3.79311 - -3.78311 - -3.77311 - -3.76311 - -3.75311 - -3.74311 - -3.73311 - -3.72311 - -3.71311 - -3.70311 - -3.69311 - -3.68311 - -3.67311 - -3.66311 - -3.65311 - -3.64311 - -3.63311 - -3.62311 - -3.61311 - -3.60311 - -3.59311 - -3.58311 - -3.57311 - -3.56311 - -3.55311 - -3.54311 - -3.53311 - -3.52311 - -3.51311 - -3.50311 - -3.49311 - -3.48311 - -3.47311 - -3.46311 - -3.45311 - -3.44311 - -3.43311 - -3.42311 - -3.41311 - -3.40311 - -3.39311 - -3.38311 - -3.37311 - -3.36311 - -3.35311 - -3.34311 - -3.33311 - -3.32311 - -3.31311 - -3.30311 - -3.29311 - -3.28311 - -3.27311 - -3.26311 - -3.25311 - -3.24311 - -3.23311 - -3.22311 - -3.21311 - -3.20311 - -3.19311 - -3.18311 - -3.17311 - -3.16311 - -3.15311 - -3.14311 - -3.13311 - -3.12311 - -3.11311 - -3.10311 - -3.09311 - -3.08311 - -3.07311 - -3.06311 - -3.05311 - -3.04311 - -3.03311 - -3.02311 - -3.01311 - -3.00311 - -2.99311 - -2.98311 - -2.97311 - -2.96311 - -2.95311 - -2.94311 - -2.93311 - -2.92311 - -2.91311 - -2.90311 - -2.89311 - -2.88311 - -2.87311 - -2.86311 - -2.85311 - -2.84311 - -2.83311 - -2.82311 - -2.81311 - -2.80311 - -2.79311 - -2.78311 - -2.77311 - -2.76311 - -2.75311 - -2.74311 - -2.73311 - -2.72311 - -2.71311 - -2.70311 - -2.69311 - -2.68311 - -2.67311 - -2.66311 - -2.65311 - -2.64311 - -2.63311 - -2.62311 - -2.61311 - -2.60311 - -2.59311 - -2.58311 - -2.57311 - -2.56311 - -2.55311 - -2.54311 - -2.53311 - -2.52311 - -2.51311 - -2.50311 - -2.49311 - -2.48311 - -2.47311 - -2.46311 - -2.45311 - -2.44311 - -2.43311 - -2.42311 - -2.41311 - -2.40311 - -2.39311 - -2.38311 - -2.37311 - -2.36311 - -2.35311 - -2.34311 - -2.33311 - -2.32311 - -2.31311 - -2.30311 - -2.29311 - -2.28311 - -2.27311 - -2.26311 - -2.25311 - -2.24311 - -2.23311 - -2.22311 - -2.21311 - -2.20311 - -2.19311 - -2.18311 - -2.17311 - -2.16311 - -2.15311 - -2.14311 - -2.13311 - -2.12311 - -2.11311 - -2.10311 - -2.09311 - -2.08311 - -2.07311 - -2.06311 - -2.05311 - -2.04311 - -2.03311 - -2.02311 - -2.01311 - -2.00311 - -1.99311 - -1.98311 - -1.97311 - -1.96311 - -1.95311 - -1.94311 - -1.93311 - -1.92311 - -1.91311 - -1.90311 - -1.89311 - -1.88311 - -1.87311 - -1.86311 - -1.85311 - -1.84311 - -1.83311 - -1.82311 - -1.81311 - -1.80311 - -1.79311 - -1.78311 - -1.77311 - -1.76311 - -1.75311 - -1.74311 - -1.73311 - -1.72311 - -1.71311 - -1.70311 - -1.69311 - -1.68311 - -1.67311 - -1.66311 - -1.65311 - -1.64311 - -1.63311 - -1.62311 - -1.61311 - -1.60311 - -1.59311 - -1.58311 - -1.57311 - -1.56311 - -1.55311 - -1.54311 - -1.53311 - -1.52311 - -1.51311 - -1.50311 - -1.49311 - -1.48311 - -1.47311 - -1.46311 - -1.45311 - -1.44311 - -1.43311 - -1.42311 - -1.41311 - -1.40311 - -1.39311 - -1.38311 - -1.37311 - -1.36311 - -1.35311 - -1.34311 - -1.33311 - -1.32311 - -1.31311 - -1.30311 - -1.29311 - -1.28311 - -1.27311 - -1.26311 - -1.25311 - -1.24311 - -1.23311 - -1.22311 - -1.21311 - -1.20311 - -1.19311 - -1.18311 - -1.17311 - -1.16311 - -1.15311 - -1.14311 - -1.13311 - -1.12311 - -1.11311 - -1.10311 - -1.09311 - -1.08311 - -1.07311 - -1.06311 - -1.05311 - -1.04311 - -1.03311 - -1.02311 - -1.01311 - -1.00311 - -0.993109 - -0.983109 - -0.973109 - -0.963109 - -0.953109 - -0.943109 - -0.933109 - -0.923109 - -0.913109 - -0.903109 - -0.893109 - -0.883109 - -0.873109 - -0.863109 - -0.853109 - -0.843109 - -0.833109 - -0.823109 - -0.813109 - -0.803109 - -0.793109 - -0.783109 - -0.773109 - -0.763109 - -0.753109 - -0.743109 - -0.733109 - -0.723109 - -0.713109 - -0.703109 - -0.693109 - -0.683109 - -0.673109 - -0.663109 - -0.653109 - -0.643109 - -0.633109 - -0.623109 - -0.613109 - -0.603109 - -0.593109 - -0.583109 - -0.573109 - -0.563109 - -0.553109 - -0.543109 - -0.533109 - -0.523109 - -0.513109 - -0.503109 - -0.493109 - -0.483109 - -0.473109 - -0.463109 - -0.453109 - -0.443109 - -0.433109 - -0.423109 - -0.413109 - -0.403109 - -0.393109 - -0.383109 - -0.373109 - -0.363109 - -0.353109 - -0.343109 - -0.333109 - -0.323109 - -0.313109 - -0.303109 - -0.293109 - -0.283109 - -0.273109 - -0.263109 - -0.253109 - -0.243109 - -0.233109 - -0.223109 - -0.213109 - -0.203109 - -0.193109 - -0.183109 - -0.173109 - -0.163109 - -0.153109 - -0.143109 - -0.133109 - -0.123109 - -0.113109 - -0.103109 - -0.093109 - -0.083109 - -0.073109 - -0.063109 - -0.053109 - -0.043109 - -0.033109 - -0.023109 - -0.013109 - -0.00310898 - 0.00689102 - 0.016891 - 0.026891 - 0.036891 - 0.046891 - 0.056891 - 0.066891 - 0.076891 - 0.086891 - 0.096891 - 0.106891 - 0.116891 - 0.126891 - 0.136891 - 0.146891 - 0.156891 - 0.166891 - 0.176891 - 0.186891 - 0.196891 - 0.206891 - 0.216891 - 0.226891 - 0.236891 - 0.246891 - 0.256891 - 0.266891 - 0.276891 - 0.286891 - 0.296891 - 0.306891 - 0.316891 - 0.326891 - 0.336891 - 0.346891 - 0.356891 - 0.366891 - 0.376891 - 0.386891 - 0.396891 - 0.406891 - 0.416891 - 0.426891 - 0.436891 - 0.446891 - 0.456891 - 0.466891 - 0.476891 - 0.486891 - 0.496891 - 0.506891 - 0.516891 - 0.526891 - 0.536891 - 0.546891 - 0.556891 - 0.566891 - 0.576891 - 0.586891 - 0.596891 - 0.606891 - 0.616891 - 0.626891 - 0.636891 - 0.646891 - 0.656891 - 0.666891 - 0.676891 - 0.686891 - 0.696891 - 0.706891 - 0.716891 - 0.726891 - 0.736891 - 0.746891 - 0.756891 - 0.766891 - 0.776891 - 0.786891 - 0.796891 - 0.806891 - 0.816891 - 0.826891 - 0.836891 - 0.846891 - 0.856891 - 0.866891 - 0.876891 - 0.886891 - 0.896891 - 0.906891 - 0.916891 - 0.926891 - 0.936891 - 0.946891 - 0.956891 - 0.966891 - 0.976891 - 0.986891 - 0.996891 - 1.00689 - 1.01689 - 1.02689 - 1.03689 - 1.04689 - 1.05689 - 1.06689 - 1.07689 - 1.08689 - 1.09689 - 1.10689 - 1.11689 - 1.12689 - 1.13689 - 1.14689 - 1.15689 - 1.16689 - 1.17689 - 1.18689 - 1.19689 - 1.20689 - 1.21689 - 1.22689 - 1.23689 - 1.24689 - 1.25689 - 1.26689 - 1.27689 - 1.28689 - 1.29689 - 1.30689 - 1.31689 - 1.32689 - 1.33689 - 1.34689 - 1.35689 - 1.36689 - 1.37689 - 1.38689 - 1.39689 - 1.40689 - 1.41689 - 1.42689 - 1.43689 - 1.44689 - 1.45689 - 1.46689 - 1.47689 - 1.48689 - 1.49689 - 1.50689 - 1.51689 - 1.52689 - 1.53689 - 1.54689 - 1.55689 - 1.56689 - 1.57689 - 1.58689 - 1.59689 - 1.60689 - 1.61689 - 1.62689 - 1.63689 - 1.64689 - 1.65689 - 1.66689 - 1.67689 - 1.68689 - 1.69689 - 1.70689 - 1.71689 - 1.72689 - 1.73689 - 1.74689 - 1.75689 - 1.76689 - 1.77689 - 1.78689 - 1.79689 - 1.80689 - 1.81689 - 1.82689 - 1.83689 - 1.84689 - 1.85689 - 1.86689 - 1.87689 - 1.88689 - 1.89689 - 1.90689 - 1.91689 - 1.92689 - 1.93689 - 1.94689 - 1.95689 - 1.96689 - 1.97689 - 1.98689 - 1.99689 - 2.00689 - 2.01689 - 2.02689 - 2.03689 - 2.04689 - 2.05689 - 2.06689 - 2.07689 - 2.08689 - 2.09689 - 2.10689 - 2.11689 - 2.12689 - 2.13689 - 2.14689 - 2.15689 - 2.16689 - 2.17689 - 2.18689 - 2.19689 - 2.20689 - 2.21689 - 2.22689 - 2.23689 - 2.24689 - 2.25689 - 2.26689 - 2.27689 - 2.28689 - 2.29689 - 2.30689 - 2.31689 - 2.32689 - 2.33689 - 2.34689 - 2.35689 - 2.36689 - 2.37689 - 2.38689 - 2.39689 - 2.40689 - 2.41689 - 2.42689 - 2.43689 - 2.44689 - 2.45689 - 2.46689 - 2.47689 - 2.48689 - 2.49689 - 2.50689 - 2.51689 - 2.52689 - 2.53689 - 2.54689 - 2.55689 - 2.56689 - 2.57689 - 2.58689 - 2.59689 - 2.60689 - 2.61689 - 2.62689 - 2.63689 - 2.64689 - 2.65689 - 2.66689 - 2.67689 - 2.68689 - 2.69689 - 2.70689 - 2.71689 - 2.72689 - 2.73689 - 2.74689 - 2.75689 - 2.76689 - 2.77689 - 2.78689 - 2.79689 - 2.80689 - 2.81689 - 2.82689 - 2.83689 - 2.84689 - 2.85689 - 2.86689 - 2.87689 - 2.88689 - 2.89689 - 2.90689 - 2.91689 - 2.92689 - 2.93689 - 2.94689 - 2.95689 - 2.96689 - 2.97689 - 2.98689 - 2.99689 - 3.00689 - 3.01689 - 3.02689 - 3.03689 - 3.04689 - 3.05689 - 3.06689 - 3.07689 - 3.08689 - 3.09689 - 3.10689 - 3.11689 - 3.12689 - 3.13689 - 3.14689 - 3.15689 - 3.16689 - 3.17689 - 3.18689 - 3.19689 - 3.20689 - 3.21689 - 3.22689 - 3.23689 - 3.24689 - 3.25689 - 3.26689 - 3.27689 - 3.28689 - 3.29689 - 3.30689 - 3.31689 - 3.32689 - 3.33689 - 3.34689 - 3.35689 - 3.36689 - 3.37689 - 3.38689 - 3.39689 - 3.40689 - 3.41689 - 3.42689 - 3.43689 - 3.44689 - 3.45689 - 3.46689 - 3.47689 - 3.48689 - 3.49689 - 3.50689 - 3.51689 - 3.52689 - 3.53689 - 3.54689 - 3.55689 - 3.56689 - 3.57689 - 3.58689 - 3.59689 - 3.60689 - 3.61689 - 3.62689 - 3.63689 - 3.64689 - 3.65689 - 3.66689 - 3.67689 - 3.68689 - 3.69689 - 3.70689 - 3.71689 - 3.72689 - 3.73689 - 3.74689 - 3.75689 - 3.76689 - 3.77689 - 3.78689 - 3.79689 - 3.80689 - 3.81689 - 3.82689 - 3.83689 - 3.84689 - 3.85689 - 3.86689 - 3.87689 - 3.88689 - 3.89689 - 3.90689 - 3.91689 - 3.92689 - 3.93689 - 3.94689 - 3.95689 - 3.96689 - 3.97689 - 3.98689 - 3.99689 - 4.00689 - 4.01689 - 4.02689 - 4.03689 - 4.04689 - 4.05689 - 4.06689 - 4.07689 - 4.08689 - 4.09689 - 4.10689 - 4.11689 - 4.12689 - 4.13689 - 4.14689 - 4.15689 - 4.16689 - 4.17689 - 4.18689 - 4.19689 - 4.20689 - 4.21689 - 4.22689 - 4.23689 - 4.24689 - 4.25689 - 4.26689 - 4.27689 - 4.28689 - 4.29689 - 4.30689 - 4.31689 - 4.32689 - 4.33689 - 4.34689 - 4.35689 - 4.36689 - 4.37689 - 4.38689 - 4.39689 - 4.40689 - 4.41689 - 4.42689 - 4.43689 - 4.44689 - 4.45689 - 4.46689 - 4.47689 - 4.48689 - 4.49689 - 4.50689 - 4.51689 - 4.52689 - 4.53689 - 4.54689 - 4.55689 - 4.56689 - 4.57689 - 4.58689 - 4.59689 - 4.60689 - 4.61689 - 4.62689 - 4.63689 - 4.64689 - 4.65689 - 4.66689 - 4.67689 - 4.68689 - 4.69689 - 4.70689 - 4.71689 - 4.72689 - 4.73689 - 4.74689 - 4.75689 - 4.76689 - 4.77689 - 4.78689 - 4.79689 - 4.80689 - 4.81689 - 4.82689 - 4.83689 - 4.84689 - 4.85689 - 4.86689 - 4.87689 - 4.88689 - 4.89689 - 4.90689 - 4.91689 - 4.92689 - 4.93689 - 4.94689 - 4.95689 - 4.96689 - 4.97689 - 4.98689 - 4.99689 - 5.00689 - 5.01689 - 5.02689 - 5.03689 - 5.04689 - 5.05689 - 5.06689 - 5.07689 - 5.08689 - 5.09689 - 5.10689 - 5.11689 - 5.12689 - 5.13689 - 5.14689 - 5.15689 - 5.16689 - 5.17689 - 5.18689 - 5.19689 - 5.20689 - 5.21689 - 5.22689 - 5.23689 - 5.24689 - 5.25689 - 5.26689 - 5.27689 - 5.28689 - 5.29689 - 5.30689 - 5.31689 - 5.32689 - 5.33689 - 5.34689 - 5.35689 - 5.36689 - 5.37689 - 5.38689 - 5.39689 - 5.40689 - 5.41689 - 5.42689 - 5.43689 - 5.44689 - 5.45689 - 5.46689 - 5.47689 - 5.48689 - 5.49689 - 5.50689 - 5.51689 - 5.52689 - 5.53689 - 5.54689 - 5.55689 - 5.56689 - 5.57689 - 5.58689 - 5.59689 - 5.60689 - 5.61689 - 5.62689 - 5.63689 - 5.64689 - 5.65689 - 5.66689 - 5.67689 - 5.68689 - 5.69689 - 5.70689 - 5.71689 - 5.72689 - 5.73689 - 5.74689 - 5.75689 - 5.76689 - 5.77689 - 5.78689 - 5.79689 - 5.80689 - 5.81689 - 5.82689 - 5.83689 - 5.84689 - 5.85689 - 5.86689 - 5.87689 - 5.88689 - 5.89689 - 5.90689 - 5.91689 - 5.92689 - 5.93689 - 5.94689 - 5.95689 - 5.96689 - 5.97689 - 5.98689 - 5.99689 - 6.00689 - 6.01689 - 6.02689 - 6.03689 - 6.04689 - 6.05689 - 6.06689 - 6.07689 - 6.08689 - 6.09689 - 6.10689 - 6.11689 - 6.12689 - 6.13689 - 6.14689 - 6.15689 - 6.16689 - 6.17689 - 6.18689 - 6.19689 - 6.20689 - 6.21689 - 6.22689 - 6.23689 - 6.24689 - 6.25689 - 6.26689 - 6.27689 - 6.28689 - 6.29689 - 6.30689 - 6.31689 - 6.32689 - 6.33689 - 6.34689 - 6.35689 - 6.36689 - 6.37689 - 6.38689 - 6.39689 - 6.40689 - 6.41689 - 6.42689 - 6.43689 - 6.44689 - 6.45689 - 6.46689 - 6.47689 - 6.48689 - 6.49689 - 6.50689 - 6.51689 - 6.52689 - 6.53689 - 6.54689 - 6.55689 - 6.56689 - 6.57689 - 6.58689 - 6.59689 - 6.60689 - 6.61689 - 6.62689 - 6.63689 - 6.64689 - 6.65689 - 6.66689 - 6.67689 - 6.68689 - 6.69689 - 6.70689 - 6.71689 - 6.72689 - 6.73689 - 6.74689 - 6.75689 - 6.76689 - 6.77689 - 6.78689 - 6.79689 - 6.80689 - 6.81689 - 6.82689 - 6.83689 - 6.84689 - 6.85689 - 6.86689 - 6.87689 - 6.88689 - 6.89689 - 6.90689 - 6.91689 - 6.92689 - 6.93689 - 6.94689 - 6.95689 - 6.96689 - 6.97689 - 6.98689 - 6.99689 - 7.00689 - 7.01689 - 7.02689 - 7.03689 - 7.04689 - 7.05689 - 7.06689 - 7.07689 - 7.08689 - 7.09689 - 7.10689 - 7.11689 - 7.12689 - 7.13689 - 7.14689 - 7.15689 - 7.16689 - 7.17689 - 7.18689 - 7.19689 - 7.20689 - 7.21689 - 7.22689 - 7.23689 - 7.24689 - 7.25689 - 7.26689 - 7.27689 - 7.28689 - 7.29689 - 7.30689 - 7.31689 - 7.32689 - 7.33689 - 7.34689 - 7.35689 - 7.36689 - 7.37689 - 7.38689 - 7.39689 - 7.40689 - 7.41689 - 7.42689 - 7.43689 - 7.44689 - 7.45689 - 7.46689 - 7.47689 - 7.48689 - 7.49689 - 7.50689 - 7.51689 - 7.52689 - 7.53689 - 7.54689 - 7.55689 - 7.56689 - 7.57689 - 7.58689 - 7.59689 - 7.60689 - 7.61689 - 7.62689 - 7.63689 - 7.64689 - 7.65689 - 7.66689 - 7.67689 - 7.68689 - 7.69689 - 7.70689 - 7.71689 - 7.72689 - 7.73689 - 7.74689 - 7.75689 - 7.76689 - 7.77689 - 7.78689 - 7.79689 - 7.80689 - 7.81689 - 7.82689 - 7.83689 - 7.84689 - 7.85689 - 7.86689 - 7.87689 - 7.88689 - 7.89689 - 7.90689 - 7.91689 - 7.92689 - 7.93689 - 7.94689 - 7.95689 - 7.96689 - 7.97689 - 7.98689 - 7.99689 - 8.00689 - 8.01689 - 8.02689 - 8.03689 - 8.04689 - 8.05689 - 8.06689 - 8.07689 - 8.08689 - 8.09689 - 8.10689 - 8.11689 - 8.12689 - 8.13689 - 8.14689 - 8.15689 - 8.16689 - 8.17689 - 8.18689 - 8.19689 - 8.20689 - 8.21689 - 8.22689 - 8.23689 - 8.24689 - 8.25689 - 8.26689 - 8.27689 - 8.28689 - 8.29689 - 8.30689 - 8.31689 - 8.32689 - 8.33689 - 8.34689 - 8.35689 - 8.36689 - 8.37689 - 8.38689 - 8.39689 - 8.40689 - 8.41689 - 8.42689 - 8.43689 - 8.44689 - 8.45689 - 8.46689 - 8.47689 - 8.48689 - 8.49689 - 8.50689 - 8.51689 - 8.52689 - 8.53689 - 8.54689 - 8.55689 - 8.56689 - 8.57689 - 8.58689 - 8.59689 - 8.60689 - 8.61689 - 8.62689 - 8.63689 - 8.64689 - 8.65689 - 8.66689 - 8.67689 - 8.68689 - 8.69689 - 8.70689 - 8.71689 - 8.72689 - 8.73689 - 8.74689 - 8.75689 - 8.76689 - 8.77689 - 8.78689 - 8.79689 - 8.80689 - 8.81689 - 8.82689 - 8.83689 - 8.84689 - 8.85689 - 8.86689 - 8.87689 - 8.88689 - 8.89689 - 8.90689 - 8.91689 - 8.92689 - 8.93689 - 8.94689 - 8.95689 - 8.96689 - 8.97689 - 8.98689 - 8.99689 - 9.00689 - 9.01689 - 9.02689 - 9.03689 - 9.04689 - 9.05689 - 9.06689 - 9.07689 - 9.08689 - 9.09689 - 9.10689 - 9.11689 - 9.12689 - 9.13689 - 9.14689 - 9.15689 - 9.16689 - 9.17689 - 9.18689 - 9.19689 - 9.20689 - 9.21689 - 9.22689 - 9.23689 - 9.24689 - 9.25689 - 9.26689 - 9.27689 - 9.28689 - 9.29689 - 9.30689 - 9.31689 - 9.32689 - 9.33689 - 9.34689 - 9.35689 - 9.36689 - 9.37689 - 9.38689 - 9.39689 - 9.40689 - 9.41689 - 9.42689 - 9.43689 - 9.44689 - 9.45689 - 9.46689 - 9.47689 - 9.48689 - 9.49689 - 9.50689 - 9.51689 - 9.52689 - 9.53689 - 9.54689 - 9.55689 - 9.56689 - 9.57689 - 9.58689 - 9.59689 - 9.60689 - 9.61689 - 9.62689 - 9.63689 - 9.64689 - 9.65689 - 9.66689 - 9.67689 - 9.68689 - 9.69689 - 9.70689 - 9.71689 - 9.72689 - 9.73689 - 9.74689 - 9.75689 - 9.76689 - 9.77689 - 9.78689 - 9.79689 - 9.80689 - 9.81689 - 9.82689 - 9.83689 - 9.84689 - 9.85689 - 9.86689 - 9.87689 - 9.88689 - 9.89689 - 9.90689 - 9.91689 - 9.92689 - 9.93689 - 9.94689 - 9.95689 - 9.96689 - 9.97689 - 9.98689 - 9.99689 - 10.0069 - 10.0169 - 10.0269 - 10.0369 - 10.0469 - 10.0569 - 10.0669 - 10.0769 - 10.0869 - 10.0969 - 10.1069 - 10.1169 - 10.1269 - 10.1369 - 10.1469 - 10.1569 - 10.1669 - 10.1769 - 10.1869 - 10.1969 - 10.2069 - 10.2169 - 10.2269 - 10.2369 - 10.2469 - 10.2569 - 10.2669 - 10.2769 - 10.2869 - 10.2969 - 10.3069 - 10.3169 - 10.3269 - 10.3369 - 10.3469 - 10.3569 - 10.3669 - 10.3769 - 10.3869 - 10.3969 - 10.4069 - 10.4169 - 10.4269 - 10.4369 - 10.4469 - 10.4569 - 10.4669 - 10.4769 - 10.4869 - 10.4969 - 10.5069 - 10.5169 - 10.5269 - 10.5369 - 10.5469 - 10.5569 - 10.5669 - 10.5769 - 10.5869 - 10.5969 - 10.6069 - 10.6169 - 10.6269 - 10.6369 - 10.6469 - 10.6569 - 10.6669 - 10.6769 - 10.6869 - 10.6969 - 10.7069 - 10.7169 - 10.7269 - 10.7369 - 10.7469 - 10.7569 - 10.7669 - 10.7769 - 10.7869 - 10.7969 - 10.8069 - 10.8169 - 10.8269 - 10.8369 - 10.8469 - 10.8569 - 10.8669 - 10.8769 - 10.8869 - 10.8969 - 10.9069 - 10.9169 - 10.9269 - 10.9369 - 10.9469 - 10.9569 - 10.9669 - 10.9769 - 10.9869 - 10.9969 - 11.0069 - 11.0169 - 11.0269 - 11.0369 - 11.0469 - 11.0569 - 11.0669 - 11.0769 - 11.0869 - 11.0969 - 11.1069 - 11.1169 - 11.1269 - 11.1369 - 11.1469 - 11.1569 - 11.1669 - 11.1769 - 11.1869 - 11.1969 - 11.2069 - 11.2169 - 11.2269 - 11.2369 - 11.2469 - 11.2569 - 11.2669 - 11.2769 - 11.2869 - 11.2969 - 11.3069 - 11.3169 - 11.3269 - 11.3369 - 11.3469 - 11.3569 - 11.3669 - 11.3769 - 11.3869 - 11.3969 - 11.4069 - 11.4169 - 11.4269 - 11.4369 - 11.4469 - 11.4569 - 11.4669 - 11.4769 - 11.4869 - 11.4969 - 11.5069 - 11.5169 - 11.5269 - 11.5369 - 11.5469 - 11.5569 - 11.5669 - 11.5769 - 11.5869 - 11.5969 - 11.6069 - 11.6169 - 11.6269 - 11.6369 - 11.6469 - 11.6569 - 11.6669 - 11.6769 - 11.6869 - 11.6969 - 11.7069 - 11.7169 - 11.7269 - 11.7369 - 11.7469 - 11.7569 - 11.7669 - 11.7769 - 11.7869 - 11.7969 - 11.8069 - 11.8169 - 11.8269 - 11.8369 - 11.8469 - 11.8569 - 11.8669 - 11.8769 - 11.8869 - 11.8969 - 11.9069 - 11.9169 - 11.9269 - 11.9369 - 11.9469 - 11.9569 - 11.9669 - 11.9769 - 11.9869 - 11.9969 - 12.0069 - 12.0169 - 12.0269 - 12.0369 - 12.0469 - 12.0569 - 12.0669 - 12.0769 - 12.0869 - 12.0969 - 12.1069 - 12.1169 - 12.1269 - 12.1369 - 12.1469 - 12.1569 - 12.1669 - 12.1769 - 12.1869 - 12.1969 - 12.2069 - 12.2169 - 12.2269 - 12.2369 - 12.2469 - 12.2569 - 12.2669 - 12.2769 - 12.2869 - 12.2969 - 12.3069 - 12.3169 - 12.3269 - 12.3369 - 12.3469 - 12.3569 - 12.3669 - 12.3769 - 12.3869 - 12.3969 - 12.4069 - 12.4169 - 12.4269 - 12.4369 - 12.4469 - 12.4569 - 12.4669 - 12.4769 - 12.4869 - 12.4969 - 12.5069 - 12.5169 - 12.5269 - 12.5369 - 12.5469 - 12.5569 - 12.5669 - 12.5769 - 12.5869 - 12.5969 - 12.6069 - 12.6169 - 12.6269 - 12.6369 - 12.6469 - 12.6569 - 12.6669 - 12.6769 - 12.6869 - 12.6969 - 12.7069 - 12.7169 - 12.7269 - 12.7369 - 12.7469 - 12.7569 - 12.7669 - 12.7769 - 12.7869 - 12.7969 - 12.8069 - 12.8169 - 12.8269 - 12.8369 - 12.8469 - 12.8569 - 12.8669 - 12.8769 - 12.8869 - 12.8969 - 12.9069 - 12.9169 - 12.9269 - 12.9369 - 12.9469 - 12.9569 - 12.9669 - 12.9769 - 12.9869 - 12.9969 - 13.0069 - 13.0169 - 13.0269 - 13.0369 - 13.0469 - 13.0569 - 13.0669 - 13.0769 - 13.0869 - 13.0969 - 13.1069 - 13.1169 - 13.1269 - 13.1369 - 13.1469 - 13.1569 - 13.1669 - 13.1769 - 13.1869 - 13.1969 - 13.2069 - 13.2169 - 13.2269 - 13.2369 - 13.2469 - 13.2569 - 13.2669 - 13.2769 - 13.2869 - 13.2969 - 13.3069 - 13.3169 - 13.3269 - 13.3369 - 13.3469 - 13.3569 - 13.3669 - 13.3769 - 13.3869 - 13.3969 - 13.4069 - 13.4169 - 13.4269 - 13.4369 - 13.4469 - 13.4569 - 13.4669 - 13.4769 - 13.4869 - 13.4969 - 13.5069 - 13.5169 - 13.5269 - 13.5369 - 13.5469 - 13.5569 - 13.5669 - 13.5769 - 13.5869 - 13.5969 - 13.6069 - 13.6169 - 13.6269 - 13.6369 - 13.6469 - 13.6569 - 13.6669 - 13.6769 - 13.6869 - 13.6969 - 13.7069 - 13.7169 - 13.7269 - 13.7369 - 13.7469 - 13.7569 - 13.7669 - 13.7769 - 13.7869 - 13.7969 - 13.8069 - 13.8169 - 13.8269 - 13.8369 - 13.8469 - 13.8569 - 13.8669 - 13.8769 - 13.8869 - 13.8969 - 13.9069 - 13.9169 - 13.9269 - 13.9369 - 13.9469 - 13.9569 - 13.9669 - 13.9769 - 13.9869 - 13.9969 - 14.0069 - 14.0169 - 14.0269 - 14.0369 - 14.0469 - 14.0569 - 14.0669 - 14.0769 - 14.0869 - 14.0969 - 14.1069 - 14.1169 - 14.1269 - 14.1369 - 14.1469 - 14.1569 - 14.1669 - 14.1769 - 14.1869 - 14.1969 - 14.2069 - 14.2169 - 14.2269 - 14.2369 - 14.2469 - 14.2569 - 14.2669 - 14.2769 - 14.2869 - 14.2969 - 14.3069 - 14.3169 - 14.3269 - 14.3369 - 14.3469 - 14.3569 - 14.3669 - 14.3769 - 14.3869 - 14.3969 - 14.4069 - 14.4169 - 14.4269 - 14.4369 - 14.4469 - 14.4569 - 14.4669 - 14.4769 - 14.4869 - 14.4969 - 14.5069 - 14.5169 - 14.5269 - 14.5369 - 14.5469 - 14.5569 - 14.5669 - 14.5769 - 14.5869 - 14.5969 - 14.6069 - 14.6169 - 14.6269 - 14.6369 - 14.6469 - 14.6569 - 14.6669 - 14.6769 - 14.6869 - 14.6969 - 14.7069 - 14.7169 - 14.7269 - 14.7369 - 14.7469 - 14.7569 - 14.7669 - 14.7769 - 14.7869 - 14.7969 - 14.8069 - 14.8169 - 14.8269 - 14.8369 - 14.8469 - 14.8569 - 14.8669 - 14.8769 - 14.8869 - 14.8969 - 14.9069 - 14.9169 - 14.9269 - 14.9369 - 14.9469 - 14.9569 - 14.9669 - 14.9769 - 14.9869 - 14.9969 - 15.0069 - 15.0169 - 15.0269 - 15.0369 - 15.0469 - 15.0569 - 15.0669 - 15.0769 - 15.0869 - 15.0969 - 15.1069 - 15.1169 - 15.1269 - 15.1369 - 15.1469 - 15.1569 - 15.1669 - 15.1769 - 15.1869 - 15.1969 - 15.2069 - 15.2169 - 15.2269 - 15.2369 - 15.2469 - 15.2569 - 15.2669 - 15.2769 - 15.2869 - 15.2969 - 15.3069 - 15.3169 - 15.3269 - 15.3369 - 15.3469 - 15.3569 - 15.3669 - 15.3769 - 15.3869 - 15.3969 - 15.4069 - 15.4169 - 15.4269 - 15.4369 - 15.4469 - 15.4569 - 15.4669 - 15.4769 - 15.4869 - 15.4969 - 15.5069 - 15.5169 - 15.5269 - 15.5369 - 15.5469 - 15.5569 - 15.5669 - 15.5769 - 15.5869 - 15.5969 - 15.6069 - 15.6169 - 15.6269 - 15.6369 - 15.6469 - 15.6569 - 15.6669 - 15.6769 - 15.6869 - 15.6969 - 15.7069 - 15.7169 - 15.7269 - 15.7369 - 15.7469 - 15.7569 - 15.7669 - 15.7769 - 15.7869 - 15.7969 - 15.8069 - 15.8169 - 15.8269 - 15.8369 - 15.8469 - 15.8569 - 15.8669 - 15.8769 - 15.8869 - 15.8969 - 15.9069 - 15.9169 - 15.9269 - 15.9369 - 15.9469 - 15.9569 - 15.9669 - 15.9769 - 15.9869 - 15.9969 - 16.0069 - 16.0169 - 16.0269 - 16.0369 - 16.0469 - 16.0569 - 16.0669 - 16.0769 - 16.0869 - 16.0969 - 16.1069 - 16.1169 - 16.1269 - 16.1369 - 16.1469 - 16.1569 - 16.1669 - 16.1769 - 16.1869 - 16.1969 - 16.2069 - 16.2169 - 16.2269 - 16.2369 - 16.2469 - 16.2569 - 16.2669 - 16.2769 - 16.2869 - 16.2969 - 16.3069 - 16.3169 - 16.3269 - 16.3369 - 16.3469 - 16.3569 - 16.3669 - 16.3769 - 16.3869 - 16.3969 - 16.4069 - 16.4169 - 16.4269 - 16.4369 - 16.4469 - 16.4569 - 16.4669 - 16.4769 - 16.4869 - 16.4969 - 16.5069 - 16.5169 - 16.5269 - 16.5369 - 16.5469 - 16.5569 - 16.5669 - 16.5769 - 16.5869 - 16.5969 - 16.6069 - 16.6169 - 16.6269 - 16.6369 - 16.6469 - 16.6569 - 16.6669 - 16.6769 - 16.6869 - 16.6969 - 16.7069 - 16.7169 - 16.7269 - 16.7369 - 16.7469 - 16.7569 - 16.7669 - 16.7769 - 16.7869 - 16.7969 - 16.8069 - 16.8169 - 16.8269 - 16.8369 - 16.8469 - 16.8569 - 16.8669 - 16.8769 - 16.8869 - 16.8969 - 16.9069 - 16.9169 - 16.9269 - 16.9369 - 16.9469 - 16.9569 - 16.9669 - 16.9769 - 16.9869 - 16.9969 - 17.0069 - 17.0169 - 17.0269 - 17.0369 - 17.0469 - 17.0569 - 17.0669 - 17.0769 - 17.0869 - 17.0969 - 17.1069 - 17.1169 - 17.1269 - 17.1369 - - - - 0.0267289 - 0.0332634 - 0.0405593 - 0.0484563 - 0.0567214 - 0.065055 - 0.0731057 - 0.080493 - 0.0868365 - 0.0917875 - 0.0950608 - 0.096462 - 0.0959064 - 0.0934278 - 0.0891746 - 0.0833957 - 0.0764157 - 0.0686054 - 0.0603491 - 0.052014 - 0.0439245 - 0.0363438 - 0.0294639 - 0.0234038 - 0.0182146 - 0.0138896 - 0.0103776 - 0.00759702 - 0.00544909 - 0.0038295 - 0.00263692 - 0.00177905 - 0.00117603 - 0.000761697 - 0.000483376 - 0.000300555 - 0.000183105 - 0.000109298 - 6.39237e-05 - 3.6631e-05 - 2.0567e-05 - 1.13144e-05 - 6.09862e-06 - 3.2209e-06 - 1.66689e-06 - 8.45673e-07 - 4.21487e-07 - 2.08527e-07 - 1.07482e-07 - 6.89524e-08 - 7.43322e-08 - 1.29008e-07 - 2.66381e-07 - 5.65012e-07 - 1.1859e-06 - 2.44369e-06 - 4.93578e-06 - 9.76872e-06 - 1.89436e-05 - 3.59937e-05 - 6.70081e-05 - 0.000122226 - 0.000218444 - 0.000382517 - 0.000656295 - 0.00110328 - 0.00181721 - 0.00293268 - 0.00463725 - 0.00718444 - 0.0109059 - 0.0162207 - 0.023638 - 0.0337514 - 0.047218 - 0.0647234 - 0.0869264 - 0.114388 - 0.147483 - 0.186313 - 0.230612 - 0.279677 - 0.332329 - 0.386917 - 0.441372 - 0.49332 - 0.540247 - 0.579689 - 0.609454 - 0.627819 - 0.633699 - 0.626759 - 0.607453 - 0.576985 - 0.537197 - 0.49041 - 0.439226 - 0.386321 - 0.334265 - 0.285366 - 0.241567 - 0.204394 - 0.17494 - 0.153888 - 0.141548 - 0.137899 - 0.142625 - 0.15514 - 0.174596 - 0.199889 - 0.229665 - 0.262341 - 0.296142 - 0.329174 - 0.359519 - 0.385351 - 0.405065 - 0.417403 - 0.421551 - 0.417208 - 0.404606 - 0.384479 - 0.357983 - 0.326585 - 0.291924 - 0.255673 - 0.2194 - 0.18447 - 0.151968 - 0.122664 - 0.0970102 - 0.0751718 - 0.0570729 - 0.0424562 - 0.0309449 - 0.0220991 - 0.0154631 - 0.0106012 - 0.00712117 - 0.00468688 - 0.00302241 - 0.00190968 - 0.00118223 - 0.000717106 - 0.000426187 - 0.000248173 - 0.000141594 - 7.91538e-05 - 4.33547e-05 - 2.32668e-05 - 1.22342e-05 - 6.30304e-06 - 3.18172e-06 - 1.57366e-06 - 7.62598e-07 - 3.62091e-07 - 1.68452e-07 - 7.67844e-08 - 3.4293e-08 - 1.50064e-08 - 6.43402e-09 - 2.70292e-09 - 1.11268e-09 - 4.49155e-10 - 1.78704e-10 - 7.26516e-11 - 3.71681e-11 - 3.99254e-11 - 8.40618e-11 - 2.11363e-10 - 5.37488e-10 - 1.34516e-09 - 3.30051e-09 - 7.93528e-09 - 1.86932e-08 - 4.31463e-08 - 9.75755e-08 - 2.16209e-07 - 4.69402e-07 - 9.98509e-07 - 2.08112e-06 - 4.24988e-06 - 8.50344e-06 - 1.66705e-05 - 3.20215e-05 - 6.02656e-05 - 0.000111131 - 0.000200787 - 0.000355448 - 0.000616526 - 0.00104776 - 0.00174467 - 0.00284642 - 0.00455012 - 0.0071266 - 0.0109365 - 0.0164442 - 0.0242261 - 0.0349696 - 0.0494578 - 0.0685356 - 0.0930538 - 0.123791 - 0.161354 - 0.206067 - 0.257854 - 0.316138 - 0.379765 - 0.446983 - 0.515471 - 0.582443 - 0.644822 - 0.69946 - 0.743401 - 0.774142 - 0.789869 - 0.789636 - 0.773458 - 0.742311 - 0.698034 - 0.643151 - 0.580637 - 0.513651 - 0.445287 - 0.378343 - 0.315168 - 0.257556 - 0.206723 - 0.163341 - 0.127614 - 0.0993871 - 0.0782617 - 0.0637042 - 0.0551354 - 0.0519919 - 0.0537573 - 0.0599654 - 0.0701785 - 0.0839509 - 0.100784 - 0.120087 - 0.141144 - 0.163108 - 0.185011 - 0.2058 - 0.224405 - 0.239804 - 0.251113 - 0.257659 - 0.259043 - 0.25518 - 0.246301 - 0.232936 - 0.21586 - 0.196018 - 0.17445 - 0.152199 - 0.130247 - 0.109461 - 0.0905612 - 0.0741229 - 0.0605995 - 0.050366 - 0.0437782 - 0.0412372 - 0.0432522 - 0.0504919 - 0.0638147 - 0.0842689 - 0.113056 - 0.151449 - 0.200672 - 0.261733 - 0.33523 - 0.421147 - 0.518667 - 0.626021 - 0.740421 - 0.858084 - 0.974384 - 1.08411 - 1.18183 - 1.26234 - 1.3211 - 1.35466 - 1.36101 - 1.33977 - 1.29222 - 1.22118 - 1.13073 - 1.02583 - 0.911863 - 0.794182 - 0.677715 - 0.566645 - 0.464208 - 0.372606 - 0.293038 - 0.225806 - 0.170484 - 0.126116 - 0.0914095 - 0.0649157 - 0.0451695 - 0.0307949 - 0.0205706 - 0.0134634 - 0.0086337 - 0.00542471 - 0.00333959 - 0.00201441 - 0.00119052 - 0.00068939 - 0.000391137 - 0.000217435 - 0.000118432 - 6.32037e-05 - 3.30487e-05 - 1.69318e-05 - 8.49939e-06 - 4.18032e-06 - 2.01451e-06 - 9.51183e-07 - 4.40045e-07 - 1.99465e-07 - 8.85875e-08 - 3.85492e-08 - 1.6436e-08 - 6.86614e-09 - 2.81039e-09 - 1.12709e-09 - 4.42878e-10 - 1.70509e-10 - 6.43205e-11 - 2.37732e-11 - 8.60919e-12 - 3.05474e-12 - 1.06199e-12 - 3.61749e-13 - 1.20734e-13 - 3.94811e-14 - 1.26498e-14 - 3.97116e-15 - 1.22148e-15 - 3.68123e-16 - 1.08702e-16 - 3.14497e-17 - 8.91525e-18 - 2.47621e-18 - 6.73873e-19 - 1.79682e-19 - 4.6943e-20 - 1.20168e-20 - 3.01577e-21 - 7.49272e-22 - 2.14954e-22 - 1.93722e-22 - 6.24827e-22 - 2.44663e-21 - 9.5184e-21 - 3.63116e-20 - 1.35732e-19 - 4.97117e-19 - 1.7839e-18 - 6.27221e-18 - 2.16076e-17 - 7.2934e-17 - 2.41207e-16 - 7.81604e-16 - 2.48154e-15 - 7.71954e-15 - 2.35287e-14 - 7.02656e-14 - 2.056e-13 - 5.89443e-13 - 1.65576e-12 - 4.5571e-12 - 1.2289e-11 - 3.247e-11 - 8.4059e-11 - 2.13218e-10 - 5.29907e-10 - 1.29036e-09 - 3.07866e-09 - 7.19694e-09 - 1.64843e-08 - 3.6994e-08 - 8.13445e-08 - 1.75252e-07 - 3.69941e-07 - 7.65139e-07 - 1.55055e-06 - 3.07869e-06 - 5.98941e-06 - 1.14167e-05 - 2.13221e-05 - 3.90175e-05 - 6.99559e-05 - 0.000122893 - 0.000211527 - 0.000356732 - 0.000589461 - 0.000954344 - 0.00151388 - 0.00235297 - 0.00358324 - 0.00534656 - 0.00781646 - 0.0111965 - 0.0157143 - 0.0216095 - 0.0291162 - 0.0384384 - 0.0497211 - 0.063018 - 0.0782604 - 0.095232 - 0.113554 - 0.132684 - 0.15194 - 0.170533 - 0.187635 - 0.202445 - 0.214279 - 0.222647 - 0.227328 - 0.228424 - 0.226381 - 0.221989 - 0.216339 - 0.210757 - 0.206705 - 0.205663 - 0.209006 - 0.217869 - 0.233034 - 0.254818 - 0.283009 - 0.316827 - 0.354937 - 0.395514 - 0.43635 - 0.475018 - 0.509059 - 0.536184 - 0.554477 - 0.562561 - 0.559721 - 0.545954 - 0.52196 - 0.489056 - 0.44904 - 0.404009 - 0.356175 - 0.307674 - 0.260416 - 0.215969 - 0.175494 - 0.139728 - 0.10901 - 0.0833381 - 0.0624435 - 0.0458741 - 0.0330733 - 0.0234486 - 0.0164261 - 0.011488 - 0.00819575 - 0.00620026 - 0.00524187 - 0.00514359 - 0.00579949 - 0.00716005 - 0.00921593 - 0.0119807 - 0.0154733 - 0.0197007 - 0.024642 - 0.0302351 - 0.0363664 - 0.0428667 - 0.0495127 - 0.0560361 - 0.0621389 - 0.0675148 - 0.0718742 - 0.0749694 - 0.0766183 - 0.0767216 - 0.0752731 - 0.07236 - 0.0681545 - 0.0628966 - 0.0568718 - 0.0503852 - 0.0437368 - 0.0371986 - 0.0309987 - 0.0253103 - 0.0202483 - 0.0158714 - 0.0121893 - 0.00917237 - 0.00676269 - 0.00488534 - 0.00345785 - 0.00239803 - 0.00162945 - 0.00108483 - 0.000707655 - 0.000452291 - 0.000283237 - 0.000173788 - 0.000104478 - 6.15418e-05 - 3.55182e-05 - 2.0085e-05 - 1.11286e-05 - 6.04223e-06 - 3.21611e-06 - 1.68155e-06 - 8.715e-07 - 4.65702e-07 - 2.95409e-07 - 2.90595e-07 - 4.59571e-07 - 8.98517e-07 - 1.83681e-06 - 3.73474e-06 - 7.46497e-06 - 1.463e-05 - 2.80973e-05 - 5.28732e-05 - 9.74872e-05 - 0.000176115 - 0.000311734 - 0.000540638 - 0.000918686 - 0.00152955 - 0.00249516 - 0.00398812 - 0.00624563 - 0.00958342 - 0.0144079 - 0.0212236 - 0.0306319 - 0.0433178 - 0.0600198 - 0.0814817 - 0.108383 - 0.141254 - 0.180376 - 0.225679 - 0.276656 - 0.332297 - 0.391066 - 0.450931 - 0.509457 - 0.563951 - 0.611663 - 0.65001 - 0.676806 - 0.690471 - 0.690182 - 0.675957 - 0.64865 - 0.609873 - 0.56183 - 0.507116 - 0.448484 - 0.388618 - 0.329941 - 0.274464 - 0.223703 - 0.178647 - 0.139784 - 0.107165 - 0.0804984 - 0.0592459 - 0.0427234 - 0.0301863 - 0.0208974 - 0.0141746 - 0.00942031 - 0.00613419 - 0.00391369 - 0.00244654 - 0.00149849 - 0.000899277 - 0.000528773 - 0.000304637 - 0.000171962 - 9.51085e-05 - 5.15398e-05 - 2.73655e-05 - 1.42364e-05 - 7.25663e-06 - 3.62415e-06 - 1.77343e-06 - 8.50273e-07 - 3.99429e-07 - 1.83848e-07 - 8.29113e-08 - 3.66359e-08 - 1.58614e-08 - 6.72902e-09 - 2.79867e-09 - 1.14499e-09 - 4.71141e-10 - 2.22004e-10 - 1.83107e-10 - 3.17921e-10 - 7.37346e-10 - 1.77917e-09 - 4.24558e-09 - 9.94017e-09 - 2.28074e-08 - 5.12753e-08 - 1.12948e-07 - 2.43774e-07 - 5.15504e-07 - 1.0681e-06 - 2.16836e-06 - 4.31308e-06 - 8.40581e-06 - 1.60513e-05 - 3.00314e-05 - 5.50526e-05 - 9.8882e-05 - 0.000174018 - 0.000300059 - 0.000506941 - 0.000839159 - 0.00136103 - 0.00216286 - 0.00336765 - 0.00513761 - 0.00767949 - 0.0112471 - 0.0161393 - 0.0226917 - 0.0312598 - 0.0421931 - 0.0558 - 0.0723041 - 0.0917971 - 0.114191 - 0.139178 - 0.166207 - 0.194474 - 0.222952 - 0.250437 - 0.275628 - 0.297224 - 0.314038 - 0.3251 - 0.329752 - 0.327715 - 0.31911 - 0.304455 - 0.284604 - 0.260674 - 0.233932 - 0.205693 - 0.177209 - 0.149585 - 0.123717 - 0.100255 - 0.079601 - 0.0619254 - 0.0472015 - 0.0352517 - 0.0257953 - 0.0184943 - 0.0129919 - 0.00894221 - 0.0060305 - 0.00398473 - 0.00257977 - 0.00163644 - 0.00101708 - 0.000619366 - 0.000369553 - 0.000216045 - 0.00012375 - 6.94522e-05 - 3.81912e-05 - 2.05767e-05 - 1.08624e-05 - 5.61842e-06 - 2.84734e-06 - 1.41384e-06 - 6.87856e-07 - 3.27893e-07 - 1.53145e-07 - 7.0083e-08 - 3.14237e-08 - 1.38051e-08 - 5.94236e-09 - 2.50619e-09 - 1.03564e-09 - 4.19311e-10 - 1.66342e-10 - 6.46556e-11 - 2.46233e-11 - 9.18805e-12 - 3.35921e-12 - 1.20334e-12 - 4.22352e-13 - 1.45244e-13 - 4.89396e-14 - 1.61569e-14 - 5.22629e-15 - 1.6564e-15 - 5.14367e-16 - 1.56502e-16 - 4.66553e-17 - 1.36276e-17 - 3.9001e-18 - 1.09363e-18 - 3.00468e-19 - 8.08844e-20 - 2.13338e-20 - 5.51326e-21 - 1.396e-21 - 3.46337e-22 - 8.4189e-23 - 2.00563e-23 - 4.70346e-24 - 1.18097e-24 - 7.36257e-25 - 2.2269e-24 - 9.33913e-24 - 3.92109e-23 - 1.6148e-22 - 6.5162e-22 - 2.57637e-21 - 9.98064e-21 - 3.78832e-20 - 1.40888e-19 - 5.13378e-19 - 1.8329e-18 - 6.41178e-18 - 2.19763e-17 - 7.38023e-17 - 2.42841e-16 - 7.82911e-16 - 2.47309e-15 - 7.65429e-15 - 2.32118e-14 - 6.89682e-14 - 2.00783e-13 - 5.72721e-13 - 1.60065e-12 - 4.38317e-12 - 1.17603e-11 - 3.09161e-11 - 7.96322e-11 - 2.0097e-10 - 4.96948e-10 - 1.20401e-09 - 2.85814e-09 - 6.64778e-09 - 1.51498e-08 - 3.38279e-08 - 7.40085e-08 - 1.58645e-07 - 3.33202e-07 - 6.85688e-07 - 1.38256e-06 - 2.73135e-06 - 5.28701e-06 - 1.00272e-05 - 1.86332e-05 - 3.39261e-05 - 6.05225e-05 - 0.000105788 - 0.000181174 - 0.000304014 - 0.000499836 - 0.000805192 - 0.00127089 - 0.00196543 - 0.00297812 - 0.00442145 - 0.0064317 - 0.00916693 - 0.0128015 - 0.017516 - 0.0234826 - 0.0308458 - 0.0396993 - 0.050062 - 0.0618546 - 0.0748813 - 0.0888206 - 0.103227 - 0.117547 - 0.131151 - 0.143377 - 0.153579 - 0.161192 - 0.165778 - 0.167072 - 0.165011 - 0.159745 - 0.15162 - 0.141158 - 0.129004 - 0.115883 - 0.10254 - 0.0896927 - 0.0779926 - 0.0679908 - 0.060121 - 0.0546894 - 0.0518725 - 0.0517191 - 0.0541544 - 0.0589851 - 0.065905 - 0.0745031 - 0.0842757 - 0.094645 - 0.104985 - 0.114654 - 0.123038 - 0.129589 - 0.13387 - 0.135586 - 0.134612 - 0.130999 - 0.124968 - 0.11689 - 0.107254 - 0.0966231 - 0.0855928 - 0.0747527 - 0.064652 - 0.0557753 - 0.0485287 - 0.0432333 - 0.0401249 - 0.0393567 - 0.0410003 - 0.0450455 - 0.0513959 - 0.059862 - 0.0701542 - 0.081879 - 0.0945419 - 0.107559 - 0.120283 - 0.132037 - 0.142159 - 0.150053 - 0.155237 - 0.157386 - 0.156357 - 0.152207 - 0.145178 - 0.13568 - 0.124242 - 0.111471 - 0.0979927 - 0.0844041 - 0.0712312 - 0.0588999 - 0.0477194 - 0.0378803 - 0.0294624 - 0.0224522 - 0.0167644 - 0.0122646 - 0.00879136 - 0.00617441 - 0.00424885 - 0.00286473 - 0.00189249 - 0.00122496 - 0.000776862 - 0.00048273 - 0.000293901 - 0.000175321 - 0.000102472 - 5.86831e-05 - 3.29274e-05 - 1.81025e-05 - 9.75118e-06 - 5.1465e-06 - 2.66136e-06 - 1.34844e-06 - 6.69419e-07 - 3.25612e-07 - 1.55182e-07 - 7.2463e-08 - 3.31535e-08 - 1.48621e-08 - 6.52778e-09 - 2.80924e-09 - 1.18454e-09 - 4.8938e-10 - 1.98098e-10 - 7.8569e-11 - 3.05323e-11 - 1.16253e-11 - 4.33696e-12 - 1.58527e-12 - 5.67752e-13 - 1.99228e-13 - 6.84982e-14 - 2.30752e-14 - 7.61636e-15 - 2.46313e-15 - 7.80482e-16 - 2.42312e-16 - 7.37099e-17 - 2.19691e-17 - 6.41558e-18 - 1.83568e-18 - 5.14628e-19 - 1.4136e-19 - 3.80451e-20 - 1.00324e-20 - 2.5921e-21 - 6.56195e-22 - 1.62761e-22 - 3.95554e-23 - 9.41886e-24 - 2.19749e-24 - 5.02335e-25 - 1.12511e-25 - 2.46908e-26 - 5.30899e-27 - 1.11847e-27 - 2.30878e-28 - 4.67015e-29 - 9.267e-30 - 1.82231e-30 - 3.88417e-31 - 1.46568e-31 - 1.55901e-31 - 2.48189e-31 - 4.11727e-31 - 6.73756e-31 - 1.08104e-30 - 1.69961e-30 - 2.61817e-30 - 3.9517e-30 - 5.84394e-30 - 8.46769e-30 - 1.20216e-29 - 1.67222e-29 - 2.2791e-29 - 3.04348e-29 - 3.98212e-29 - 5.10498e-29 - 6.41227e-29 - 7.89161e-29 - 9.51605e-29 - 1.12431e-28 - 1.30151e-28 - 1.47622e-28 - 1.64055e-28 - 1.78634e-28 - 1.90579e-28 - 1.99216e-28 - 2.04038e-28 - 2.04755e-28 - 2.01324e-28 - 1.93952e-28 - 1.83077e-28 - 1.69322e-28 - 1.53442e-28 - 1.3625e-28 - 1.18554e-28 - 1.01095e-28 - 8.45045e-29 - 6.92741e-29 - 5.57456e-29 - 4.41176e-29 - 3.44649e-29 - 2.6766e-29 - 2.0934e-29 - 1.68469e-29 - 1.43733e-29 - 1.33907e-29 - 1.37959e-29 - 1.55069e-29 - 1.84575e-29 - 2.25848e-29 - 2.7814e-29 - 3.40412e-29 - 4.11181e-29 - 4.88408e-29 - 5.69458e-29 - 6.51134e-29 - 7.29812e-29 - 8.01651e-29 - 8.62869e-29 - 9.10048e-29 - 9.40444e-29 - 9.52235e-29 - 9.44702e-29 - 9.18299e-29 - 8.74602e-29 - 8.16158e-29 - 7.46235e-29 - 6.68518e-29 - 5.86797e-29 - 5.04661e-29 - 4.25254e-29 - 3.51102e-29 - 2.84024e-29 - 2.2512e-29 - 1.74828e-29 - 1.33028e-29 - 9.91773e-30 - 7.24467e-30 - 5.18516e-30 - 3.63616e-30 - 2.4984e-30 - 1.68198e-30 - 1.10952e-30 - 7.17189e-31 - 4.54406e-31 - 2.82474e-31 - 1.72835e-31 - 1.05198e-31 - 6.58503e-32 - 4.62923e-32 - 4.23403e-32 - 5.37657e-32 - 8.45634e-32 - 1.43958e-31 - 2.48283e-31 - 4.2389e-31 - 7.11257e-31 - 1.17038e-30 - 1.88743e-30 - 2.98249e-30 - 4.61765e-30 - 7.00468e-30 - 1.04105e-29 - 1.5159e-29 - 2.16256e-29 - 3.02247e-29 - 4.13843e-29 - 5.55101e-29 - 7.29374e-29 - 9.38732e-29 - 1.18334e-28 - 1.46085e-28 - 1.76589e-28 - 2.08981e-28 - 2.42059e-28 - 2.74326e-28 - 3.04057e-28 - 3.29405e-28 - 3.48541e-28 - 3.59803e-28 - 3.6185e-28 - 3.53791e-28 - 3.35298e-28 - 3.06655e-28 - 2.68768e-28 - 2.23113e-28 - 1.71642e-28 - 1.16641e-28 - 6.05666e-29 - 5.87455e-30 - -4.515e-29 - -9.05398e-29 - -1.28761e-28 - -1.58788e-28 - -1.80136e-28 - -1.92849e-28 - -1.97443e-28 - -1.94826e-28 - -1.8619e-28 - -1.72893e-28 - -1.56342e-28 - -1.37892e-28 - -1.18757e-28 - -9.99546e-29 - -8.22691e-29 - -6.62465e-29 - -5.22075e-29 - -4.0277e-29 - -3.04241e-29 - -2.25038e-29 - -1.62989e-29 - -1.15555e-29 - -8.01113e-30 - -5.41433e-30 - -3.53641e-30 - -2.17534e-30 - -1.1537e-30 - -3.11658e-31 - 5.03908e-31 - 1.4533e-30 - 2.72011e-30 - 4.52875e-30 - 7.16233e-30 - 1.09794e-29 - 1.64269e-29 - 2.4047e-29 - 3.44732e-29 - 4.84126e-29 - 6.66106e-29 - 8.97954e-29 - 1.18604e-28 - 1.53489e-28 - 1.94622e-28 - 2.41793e-28 - 2.94329e-28 - 3.51042e-28 - 4.10224e-28 - 4.697e-28 - 5.26935e-28 - 5.79202e-28 - 6.23793e-28 - 6.58244e-28 - 6.80567e-28 - 6.89431e-28 - 6.84303e-28 - 6.65492e-28 - 6.34123e-28 - 5.92027e-28 - 5.4156e-28 - 4.85387e-28 - 4.26252e-28 - 3.6676e-28 - 3.09196e-28 - 2.55401e-28 - 2.06704e-28 - 1.63912e-28 - 1.27353e-28 - 9.69498e-29 - 7.23136e-29 - 5.28482e-29 - 3.78423e-29 - 2.65498e-29 - 1.82508e-29 - 1.22925e-29 - 8.11209e-30 - 5.24523e-30 - 3.32301e-30 - 2.0627e-30 - 1.25452e-30 - 7.47579e-31 - 4.36488e-31 - 2.49704e-31 - 1.39963e-31 - 7.68669e-32 - 4.13621e-32 - 2.18073e-32 - 1.12652e-32 - 5.7018e-33 - 2.82763e-33 - 1.37395e-33 - 6.54133e-34 - 3.05176e-34 - 1.39593e-34 - 6.27992e-35 - 2.82667e-35 - 1.38791e-35 - 9.9475e-36 - 1.36478e-35 - 2.72717e-35 - 5.93916e-35 - 1.29385e-34 - 2.77206e-34 - 5.82302e-34 - 1.19862e-33 - 2.41748e-33 - 4.7773e-33 - 9.24992e-33 - 1.75481e-32 - 3.26183e-32 - 5.94056e-32 - 1.06006e-31 - 1.85341e-31 - 3.17503e-31 - 5.3292e-31 - 8.76421e-31 - 1.41221e-30 - 2.22959e-30 - 3.44895e-30 - 5.22738e-30 - 7.76282e-30 - 1.12951e-29 - 1.61028e-29 - 2.24929e-29 - 3.07842e-29 - 4.12807e-29 - 5.4238e-29 - 6.98227e-29 - 8.80697e-29 - 1.08841e-28 - 1.31794e-28 - 1.56364e-28 - 1.81767e-28 - 2.07028e-28 - 2.31036e-28 - 2.5262e-28 - 2.7064e-28 - 2.84091e-28 - 2.92198e-28 - 2.94527e-28 - 2.91212e-28 - 2.83857e-28 - 2.79971e-28 - 3.14646e-28 - 5.54256e-28 - 1.78173e-27 - 7.61874e-27 - 3.44876e-26 - 1.55331e-25 - 6.87462e-25 - 2.9827e-24 - 1.26809e-23 - 5.28244e-23 - 2.15605e-22 - 8.62222e-22 - 3.37844e-21 - 1.29703e-20 - 4.87891e-20 - 1.79817e-19 - 6.49345e-19 - 2.29751e-18 - 7.96482e-18 - 2.7054e-17 - 9.00377e-17 - 2.93598e-16 - 9.38036e-16 - 2.93645e-15 - 9.00663e-15 - 2.70669e-14 - 7.96989e-14 - 2.29934e-13 - 6.49965e-13 - 1.80017e-12 - 4.88512e-12 - 1.29889e-11 - 3.38382e-11 - 8.63731e-11 - 2.16016e-10 - 5.29337e-10 - 1.27091e-09 - 2.98973e-09 - 6.8911e-09 - 1.55626e-08 - 3.44358e-08 - 7.4658e-08 - 1.58591e-07 - 3.3008e-07 - 6.73126e-07 - 1.34496e-06 - 2.63306e-06 - 5.05065e-06 - 9.4923e-06 - 1.74796e-05 - 3.15377e-05 - 5.57526e-05 - 9.65687e-05 - 0.000163887 - 0.000272515 - 0.000443989 - 0.000708747 - 0.00110853 - 0.00169879 - 0.00255076 - 0.00375262 - 0.00540927 - 0.00763973 - 0.0105719 - 0.014334 - 0.0190423 - 0.024786 - 0.0316105 - 0.0394995 - 0.0483604 - 0.0580129 - 0.0681862 - 0.0785245 - 0.0886035 - 0.0979565 - 0.106109 - 0.112619 - 0.117113 - 0.119328 - 0.119129 - 0.116531 - 0.11169 - 0.104896 - 0.096536 - 0.0870678 - 0.0769742 - 0.0667276 - 0.0567567 - 0.0474229 - 0.0390066 - 0.0317027 - 0.0256258 - 0.0208203 - 0.0172751 - 0.0149371 - 0.0137248 - 0.0135373 - 0.0142605 - 0.0157693 - 0.0179273 - 0.0205862 - 0.0235843 - 0.0267481 - 0.0298956 - 0.032843 - 0.0354138 - 0.0374501 - 0.0388236 - 0.0394455 - 0.0392739 - 0.0383163 - 0.0366285 - 0.0343086 - 0.0314868 - 0.0283136 - 0.0249459 - 0.0215348 - 0.0182147 - 0.0150952 - 0.0122572 - 0.00975177 - 0.00760171 - 0.00580599 - 0.0043449 - 0.00318583 - 0.00228882 - 0.00161127 - 0.00111162 - 0.0007519 - 0.000499301 - 0.000326825 - 0.000213418 - 0.000143759 - 0.000108028 - 0.000101857 - 0.000126739 - 0.000191099 - 0.000312256 - 0.000519518 - 0.000858582 - 0.00139739 - 0.00223343 - 0.00350226 - 0.0053867 - 0.00812573 - 0.0120216 - 0.0174433 - 0.0248236 - 0.0346481 - 0.0474328 - 0.0636896 - 0.083879 - 0.108352 - 0.137285 - 0.170613 - 0.207972 - 0.248658 - 0.291611 - 0.335434 - 0.37845 - 0.418797 - 0.454557 - 0.483901 - 0.505245 - 0.517389 - 0.519629 - 0.511826 - 0.494418 - 0.468382 - 0.435139 - 0.396432 - 0.354169 - 0.310271 - 0.266533 - 0.224505 - 0.18542 - 0.150151 - 0.119216 - 0.0928035 - 0.0708289 - 0.0530002 - 0.0388861 - 0.0279804 - 0.0197564 - 0.0137097 - 0.00938729 - 0.00640552 - 0.00445878 - 0.00332166 - 0.00284708 - 0.00296262 - 0.0036664 - 0.00502305 - 0.00715947 - 0.0102594 - 0.0145556 - 0.0203168 - 0.0278303 - 0.0373755 - 0.0491924 - 0.0634435 - 0.0801731 - 0.099269 - 0.120431 - 0.143153 - 0.166724 - 0.190254 - 0.21272 - 0.233033 - 0.250129 - 0.263056 - 0.271062 - 0.273669 - 0.270719 - 0.262392 - 0.249183 - 0.231858 - 0.21138 - 0.188818 - 0.165256 - 0.141713 - 0.119069 - 0.0980222 - 0.0790655 - 0.0624865 - 0.0483863 - 0.036711 - 0.0272902 - 0.0198772 - 0.0141856 - 0.00991947 - 0.00679686 - 0.00456443 - 0.00300582 - 0.00194419 - 0.00124098 - 0.000792285 - 0.000524266 - 0.000388696 - 0.000359281 - 0.000429289 - 0.000610657 - 0.000934591 - 0.00145347 - 0.00224368 - 0.00340871 - 0.00508177 - 0.00742678 - 0.0106365 - 0.0149265 - 0.0205241 - 0.0276509 - 0.0365 - 0.0472078 - 0.0598234 - 0.0742789 - 0.0903642 - 0.107712 - 0.125797 - 0.14395 - 0.161395 - 0.177299 - 0.190835 - 0.201255 - 0.207957 - 0.21054 - 0.20885 - 0.202989 - 0.193306 - 0.180366 - 0.164893 - 0.147702 - 0.12963 - 0.111472 - 0.0939202 - 0.0775336 - 0.0627131 - 0.0497008 - 0.0385927 - 0.0293619 - 0.0218877 - 0.0159864 - 0.0114404 - 0.00802171 - 0.00551099 - 0.00370962 - 0.00244662 - 0.00158103 - 0.00100104 - 0.000621008 - 0.000377469 - 0.000224803 - 0.000131178 - 7.49988e-05 - 4.20132e-05 - 2.30597e-05 - 1.24011e-05 - 6.53433e-06 - 3.37349e-06 - 1.70646e-06 - 8.45761e-07 - 4.10712e-07 - 1.95417e-07 - 9.11016e-08 - 4.16127e-08 - 1.86236e-08 - 8.16673e-09 - 3.50932e-09 - 1.47874e-09 - 6.13789e-10 - 2.58271e-10 - 1.2874e-10 - 1.17134e-10 - 2.11594e-10 - 4.85849e-10 - 1.14897e-09 - 2.68349e-09 - 6.14836e-09 - 1.3805e-08 - 3.03712e-08 - 6.54678e-08 - 1.38271e-07 - 2.86134e-07 - 5.80156e-07 - 1.15254e-06 - 2.2434e-06 - 4.27852e-06 - 7.99498e-06 - 1.46379e-05 - 2.62588e-05 - 4.61539e-05 - 7.94839e-05 - 0.000134118 - 0.000221734 - 0.000359181 - 0.000570074 - 0.000886515 - 0.00135076 - 0.00201654 - 0.00294966 - 0.00422741 - 0.00593627 - 0.00816751 - 0.0110104 - 0.0145429 - 0.0188208 - 0.0238651 - 0.0296499 - 0.0360928 - 0.0430481 - 0.0503066 - 0.0576013 - 0.0646215 - 0.0710327 - 0.0765027 - 0.0807294 - 0.0834688 - 0.0845576 - 0.0839303 - 0.0816247 - 0.077779 - 0.0726175 - 0.0664294 - 0.0595419 - 0.0522922 - 0.0450009 - 0.0379502 - 0.0313689 - 0.025425 - 0.0202253 - 0.015822 - 0.0122232 - 0.00940706 - 0.00733663 - 0.0059741 - 0.00529347 - 0.00529026 - 0.00598776 - 0.00743939 - 0.00972643 - 0.0129513 - 0.0172261 - 0.0226564 - 0.0293225 - 0.0372574 - 0.0464256 - 0.0567052 - 0.0678749 - 0.0796106 - 0.0914928 - 0.103026 - 0.113671 - 0.122882 - 0.130157 - 0.135077 - 0.137352 - 0.136843 - 0.133582 - 0.127765 - 0.119732 - 0.109938 - 0.0989056 - 0.0871828 - 0.075297 - 0.0637179 - 0.0528302 - 0.042918 - 0.0341613 - 0.0266419 - 0.0203579 - 0.0152419 - 0.011181 - 0.00803634 - 0.00565944 - 0.00390504 - 0.00264007 - 0.0017488 - 0.00113502 - 0.000721777 - 0.000449717 - 0.000274545 - 0.000164219 - 9.62432e-05 - 5.52655e-05 - 3.10939e-05 - 1.71409e-05 - 9.25821e-06 - 4.89957e-06 - 2.54054e-06 - 1.29072e-06 - 6.425e-07 - 3.13366e-07 - 1.4975e-07 - 7.01164e-08 - 3.21669e-08 - 1.44589e-08 - 6.36792e-09 - 2.74788e-09 - 1.16181e-09 - 4.8129e-10 - 1.95352e-10 - 7.76899e-11 - 3.02725e-11 - 1.15577e-11 - 4.32374e-12 - 1.58565e-12 - 5.72373e-13 - 2.10546e-13 - 1.00464e-13 - 1.17138e-13 - 2.82074e-13 - 7.88586e-13 - 2.20757e-12 - 6.07018e-12 - 1.63587e-11 - 4.31963e-11 - 1.11759e-10 - 2.83306e-10 - 7.03665e-10 - 1.71243e-09 - 4.08317e-09 - 9.53933e-09 - 2.18361e-08 - 4.89745e-08 - 1.07622e-07 - 2.31723e-07 - 4.88848e-07 - 1.01045e-06 - 2.04642e-06 - 4.06079e-06 - 7.89519e-06 - 1.50401e-05 - 2.80723e-05 - 5.13382e-05 - 9.199e-05 - 0.000161502 - 0.000277812 - 0.000468232 - 0.000773229 - 0.0012511 - 0.00198341 - 0.00308085 - 0.00468883 - 0.00699192 - 0.0102156 - 0.0146241 - 0.0205122 - 0.0281897 - 0.0379582 - 0.0500793 - 0.0647363 - 0.0819924 - 0.101751 - 0.123719 - 0.147392 - 0.172047 - 0.196769 - 0.220498 - 0.242097 - 0.260441 - 0.274516 - 0.283507 - 0.286878 - 0.284426 - 0.2763 - 0.262989 - 0.245271 - 0.224139 - 0.200716 - 0.176155 - 0.151552 - 0.127879 - 0.105932 - 0.0863111 - 0.0694246 - 0.0555089 - 0.0446667 - 0.0369097 - 0.0322012 - 0.0304904 - 0.0317364 - 0.0359162 - 0.0430182 - 0.0530213 - 0.0658637 - 0.0814058 - 0.0993939 - 0.119431 - 0.14096 - 0.163268 - 0.185513 - 0.206765 - 0.226071 - 0.242535 - 0.255385 - 0.26405 - 0.268204 - 0.267793 - 0.263031 - 0.25437 - 0.242442 - 0.22799 - 0.211798 - 0.194613 - 0.177099 - 0.159795 - 0.143102 - 0.127283 - 0.112488 - 0.098775 - 0.0861443 - 0.0745635 - 0.0639902 - 0.0543848 - 0.0457162 - 0.0379612 - 0.0311 - 0.0251102 - 0.0199607 - 0.0156086 - 0.0119976 - 0.00905948 - 0.00671686 - 0.00488768 - 0.00348951 - 0.00244361 - 0.00167807 - 0.00112985 - 0.000745775 - 0.00048253 - 0.000306019 - 0.000190246 - 0.000115991 - 6.94814e-05 - 4.11611e-05 - 2.46614e-05 - 1.60011e-05 - 1.3041e-05 - 1.52032e-05 - 2.34787e-05 - 4.07624e-05 - 7.25905e-05 - 0.000128383 - 0.000223324 - 0.00038104 - 0.000637195 - 0.00104411 - 0.00167637 - 0.00263714 - 0.00406474 - 0.00613861 - 0.00908331 - 0.0131691 - 0.0187069 - 0.0260368 - 0.0355065 - 0.0474424 - 0.06211 - 0.0796698 - 0.10013 - 0.123301 - 0.148768 - 0.175869 - 0.203707 - 0.231185 - 0.257068 - 0.280076 - 0.298978 - 0.312708 - 0.320462 - 0.321774 - 0.316564 - 0.305147 - 0.2882 - 0.266695 - 0.24181 - 0.214817 - 0.186982 - 0.159467 - 0.133253 - 0.109098 - 0.087518 - 0.0687881 - 0.0529745 - 0.0399721 - 0.0295518 - 0.0214066 - 0.0151931 - 0.0105654 - 0.00719878 - 0.00480585 - 0.00314353 - 0.00201466 - 0.0012651 - 0.000778364 - 0.000469221 - 0.000277147 - 0.000160391 - 9.09462e-05 - 5.05275e-05 - 2.75047e-05 - 1.46699e-05 - 7.6664e-06 - 3.9259e-06 - 1.97083e-06 - 9.71818e-07 - 4.75235e-07 - 2.40786e-07 - 1.48546e-07 - 1.49402e-07 - 2.43803e-07 - 4.81945e-07 - 9.85453e-07 - 1.99755e-06 - 3.97698e-06 - 7.7618e-06 - 1.48441e-05 - 2.78157e-05 - 5.107e-05 - 9.1871e-05 - 0.00016193 - 0.00027965 - 0.000473193 - 0.00078451 - 0.00127437 - 0.00202829 - 0.00316301 - 0.0048329 - 0.00723523 - 0.0106129 - 0.0152529 - 0.0214786 - 0.0296346 - 0.0400615 - 0.0530631 - 0.0688644 - 0.0875658 - 0.109096 - 0.133175 - 0.159284 - 0.186664 - 0.21433 - 0.241126 - 0.265791 - 0.287062 - 0.303771 - 0.314959 - 0.319964 - 0.318482 - 0.310605 - 0.296807 - 0.2779 - 0.254954 - 0.2292 - 0.201925 - 0.174371 - 0.147651 - 0.122691 - 0.100198 - 0.0806612 - 0.06437 - 0.0514539 - 0.0419302 - 0.0357515 - 0.0328484 - 0.0331588 - 0.0366424 - 0.0432776 - 0.0530423 - 0.0658818 - 0.0816677 - 0.100155 - 0.120947 - 0.143467 - 0.166955 - 0.190486 - 0.213013 - 0.233429 - 0.250655 - 0.263726 - 0.271879 - 0.274625 - 0.271795 - 0.263562 - 0.250416 - 0.233118 - 0.212632 - 0.190028 - 0.166397 - 0.14276 - 0.120007 - 0.0988426 - 0.079766 - 0.0630708 - 0.0488625 - 0.0370902 - 0.0275855 - 0.020102 - 0.0143527 - 0.0100407 - 0.00688229 - 0.00462209 - 0.00304145 - 0.00196092 - 0.00123872 - 0.000766703 - 0.000464961 - 0.000276276 - 0.000160845 - 9.175e-05 - 5.12794e-05 - 2.80813e-05 - 1.5067e-05 - 7.9209e-06 - 4.07999e-06 - 2.05911e-06 - 1.01821e-06 - 4.93325e-07 - 2.34188e-07 - 1.08926e-07 - 4.96407e-08 - 2.21656e-08 - 9.69749e-09 - 4.15695e-09 - 1.74593e-09 - 7.18484e-10 - 2.89697e-10 - 1.14448e-10 - 4.43003e-11 - 1.68013e-11 - 6.24335e-12 - 2.27315e-12 - 8.10916e-13 - 2.83439e-13 - 9.70692e-14 - 3.25716e-14 - 1.07086e-14 - 3.44958e-15 - 1.08877e-15 - 3.36698e-16 - 1.02019e-16 - 3.02874e-17 - 8.81003e-18 - 2.5109e-18 - 7.01164e-19 - 1.91843e-19 - 5.14292e-20 - 1.35086e-20 - 3.47654e-21 - 8.76639e-22 - 2.16587e-22 - 5.24299e-23 - 1.24355e-23 - 2.88991e-24 - 6.58026e-25 - 1.46804e-25 - 3.209e-26 - 6.87287e-27 - 1.44226e-27 - 2.96542e-28 - 5.974e-29 - 1.17918e-29 - 2.28052e-30 - 4.3214e-31 - 8.02326e-32 - 1.45953e-32 - 2.60144e-33 - 4.54309e-34 - 7.77366e-35 - 1.30328e-35 - 2.14084e-36 - 3.44563e-37 - 5.43363e-38 - 8.39553e-39 - 1.27099e-39 - 1.88528e-40 - 2.73996e-41 - 3.90166e-42 - 5.44367e-43 - 7.44168e-44 - 9.96753e-45 - 1.3081e-45 - 1.68202e-46 - 2.11913e-47 - 2.6159e-48 - 3.16389e-49 - 3.74937e-50 - 4.35344e-51 - 4.95271e-52 - 5.52066e-53 - 6.02942e-54 - 6.45204e-55 - 6.76481e-56 - 6.94945e-57 - 6.99492e-58 - 6.89845e-59 - 6.66587e-60 - 6.31102e-61 - 5.85435e-62 - 5.32102e-63 - 4.73858e-64 - 4.13465e-65 - 3.5348e-66 - 2.96093e-67 - 2.43013e-68 - 1.95419e-69 - 1.53972e-70 - 1.18864e-71 - 8.99083e-73 - 6.66322e-74 - 4.83845e-75 - 3.44242e-76 - 2.39971e-77 - 1.63905e-78 - 1.09688e-79 - 7.19228e-81 - 4.62071e-82 - 2.90863e-83 - 1.79393e-84 - 1.08407e-85 - 6.41872e-87 - 3.7237e-88 - 2.1166e-89 - 1.1788e-90 - 6.43246e-92 - 3.43915e-93 - 1.80162e-94 - 9.2472e-96 - 4.65045e-97 - 2.29148e-98 - 1.10631e-99 - 5.23324e-101 - 2.42551e-102 - 1.10147e-103 - 4.90092e-105 - 2.13659e-106 - 9.12641e-108 - 3.81959e-109 - 1.56628e-110 - 6.29303e-112 - 2.4774e-113 - 9.68532e-115 - 3.77016e-115 - 8.76891e-114 - 2.20935e-112 - 5.45488e-111 - 1.3196e-109 - 3.12779e-108 - 7.2639e-107 - 1.65287e-105 - 3.68506e-104 - 8.04983e-103 - 1.72292e-101 - 3.61312e-100 - 7.42395e-99 - 1.4946e-97 - 2.94816e-96 - 5.6979e-95 - 1.07898e-93 - 2.00195e-92 - 3.63937e-91 - 6.48242e-90 - 1.13132e-88 - 1.9345e-87 - 3.24109e-86 - 5.32046e-85 - 8.55745e-84 - 1.34858e-82 - 2.08231e-81 - 3.15029e-80 - 4.66975e-79 - 6.78224e-78 - 9.65137e-77 - 1.34568e-75 - 1.83837e-74 - 2.4607e-73 - 3.22717e-72 - 4.14689e-71 - 5.22108e-70 - 6.44072e-69 - 7.78476e-68 - 9.21919e-67 - 1.06974e-65 - 1.21618e-64 - 1.35474e-63 - 1.47861e-62 - 1.58119e-61 - 1.65674e-60 - 1.70082e-59 - 1.71081e-58 - 1.68609e-57 - 1.62816e-56 - 1.54046e-55 - 1.42804e-54 - 1.29708e-53 - 1.15433e-52 - 1.00654e-51 - 8.59942e-51 - 7.19853e-50 - 5.90411e-49 - 4.74463e-48 - 3.73583e-47 - 2.8821e-46 - 2.17855e-45 - 1.61348e-44 - 1.17083e-43 - 8.32461e-43 - 5.79922e-42 - 3.95833e-41 - 2.64723e-40 - 1.73464e-39 - 1.11368e-38 - 7.0057e-38 - 4.31796e-37 - 2.60761e-36 - 1.54292e-35 - 8.94501e-35 - 5.08107e-34 - 2.82791e-33 - 1.54211e-32 - 8.23947e-32 - 4.31341e-31 - 2.21248e-30 - 1.11192e-29 - 5.47529e-29 - 2.64166e-28 - 1.24877e-27 - 5.78395e-27 - 2.62485e-26 - 1.16713e-25 - 5.08481e-25 - 2.17052e-24 - 9.07803e-24 - 3.72011e-23 - 1.49368e-22 - 5.87616e-22 - 2.265e-21 - 8.5542e-21 - 3.16539e-20 - 1.14766e-19 - 4.07694e-19 - 1.41903e-18 - 4.83937e-18 - 1.61704e-17 - 5.29408e-17 - 1.69823e-16 - 5.33751e-16 - 1.64369e-15 - 4.95947e-15 - 1.46619e-14 - 4.24697e-14 - 1.20533e-13 - 3.35174e-13 - 9.13211e-13 - 2.43786e-12 - 6.37651e-12 - 1.63416e-11 - 4.10339e-11 - 1.00955e-10 - 2.4336e-10 - 5.74789e-10 - 1.33016e-09 - 3.01604e-09 - 6.70048e-09 - 1.45852e-08 - 3.11068e-08 - 6.50032e-08 - 1.33092e-07 - 2.66996e-07 - 5.24802e-07 - 1.0107e-06 - 1.90716e-06 - 3.52605e-06 - 6.38742e-06 - 1.13371e-05 - 1.97157e-05 - 3.35939e-05 - 5.60848e-05 - 9.17418e-05 - 0.000147037 - 0.000230899 - 0.000355267 - 0.000535579 - 0.000791098 - 0.00114491 - 0.0016235 - 0.00225564 - 0.0030706 - 0.00409557 - 0.00535231 - 0.00685339 - 0.00859818 - 0.0105693 - 0.0127297 - 0.0150221 - 0.0173692 - 0.0196772 - 0.0218417 - 0.0237545 - 0.0253129 - 0.0264286 - 0.0270361 - 0.0270988 - 0.0266129 - 0.0256078 - 0.0241429 - 0.022302 - 0.0201853 - 0.0179004 - 0.0155534 - 0.0132412 - 0.011045 - - - - - -0.00206576 - -0.00257078 - -0.00313465 - -0.00374497 - -0.00438375 - -0.00502781 - -0.00565001 - -0.00622095 - -0.00671121 - -0.00709385 - -0.00734683 - -0.00745512 - -0.00741219 - -0.00722062 - -0.00689192 - -0.00644528 - -0.00590583 - -0.00530221 - -0.00466412 - -0.00401994 - -0.00339473 - -0.00280885 - -0.00227713 - -0.00180878 - -0.00140773 - -0.00107347 - -0.000802042 - -0.00058714 - -0.000421136 - -0.000295965 - -0.000203796 - -0.000137495 - -9.08899e-05 - -5.88682e-05 - -3.7358e-05 - -2.32286e-05 - -1.41514e-05 - -8.44717e-06 - -4.94038e-06 - -2.83105e-06 - -1.58954e-06 - -8.74443e-07 - -4.71337e-07 - -2.48932e-07 - -1.28834e-07 - -6.53783e-08 - -3.26243e-08 - -1.62362e-08 - -8.59254e-09 - -5.99475e-09 - -7.26475e-09 - -1.33706e-08 - -2.80399e-08 - -5.96719e-08 - -1.25329e-07 - -2.58291e-07 - -5.21712e-07 - -1.03256e-06 - -2.00235e-06 - -3.80456e-06 - -7.0828e-06 - -1.29194e-05 - -2.30897e-05 - -4.04323e-05 - -6.93708e-05 - -0.000116617 - -0.000192081 - -0.000309986 - -0.000490161 - -0.0007594 - -0.00115276 - -0.00171453 - -0.00249856 - -0.00356754 - -0.00499098 - -0.00684131 - -0.00918818 - -0.0120909 - -0.0155891 - -0.0196935 - -0.0243759 - -0.0295621 - -0.0351275 - -0.0408974 - -0.0466533 - -0.0521444 - -0.0571046 - -0.0612738 - -0.0644202 - -0.0663617 - -0.0669837 - -0.0662512 - -0.0642123 - -0.0609945 - -0.0567933 - -0.0518548 - -0.046455 - -0.0408787 - -0.0353993 - -0.0302635 - -0.0256799 - -0.0218134 - -0.0187838 - -0.0166674 - -0.0155012 - -0.0152867 - -0.0159929 - -0.0175585 - -0.0198923 - -0.0228734 - -0.0263515 - -0.0301487 - -0.0340648 - -0.0378847 - -0.0413897 - -0.0443713 - -0.0466459 - -0.0480693 - -0.0485484 - -0.0480492 - -0.0465983 - -0.0442805 - -0.0412291 - -0.037613 - -0.0336212 - -0.029446 - -0.0252685 - -0.0212456 - -0.0175023 - -0.0141273 - -0.0111728 - -0.00865761 - -0.00657314 - -0.00488972 - -0.00356396 - -0.00254518 - -0.0017809 - -0.00122095 - -0.000820152 - -0.000539793 - -0.000348094 - -0.000219939 - -0.000136159 - -8.25898e-05 - -4.90844e-05 - -2.85823e-05 - -1.63075e-05 - -9.11622e-06 - -4.9932e-06 - -2.67966e-06 - -1.40902e-06 - -7.25927e-07 - -3.66442e-07 - -1.8124e-07 - -8.78291e-08 - -4.17024e-08 - -1.94008e-08 - -8.84333e-09 - -3.94956e-09 - -1.7283e-09 - -7.41011e-10 - -3.11293e-10 - -1.28134e-10 - -5.16863e-11 - -2.04562e-11 - -8.01249e-12 - -3.29647e-12 - -1.92366e-12 - -2.56019e-12 - -5.76495e-12 - -1.44162e-11 - -3.59951e-11 - -8.82901e-11 - -2.12263e-10 - -5.00027e-10 - -1.15412e-09 - -2.61005e-09 - -5.78339e-09 - -1.25561e-08 - -2.67092e-08 - -5.56679e-08 - -1.1368e-07 - -2.27459e-07 - -4.45921e-07 - -8.56544e-07 - -1.61205e-06 - -2.97264e-06 - -5.37088e-06 - -9.50789e-06 - -1.64915e-05 - -2.80267e-05 - -4.66683e-05 - -7.61391e-05 - -0.000121711 - -0.00019063 - -0.000292542 - -0.000439867 - -0.000648025 - -0.000935404 - -0.00132295 - -0.00183326 - -0.0024891 - -0.00331129 - -0.00431607 - -0.0055121 - -0.00689736 - -0.00845639 - -0.0101584 - -0.0119564 - -0.0137884 - -0.0155798 - -0.0172484 - -0.0187099 - -0.0198853 - -0.0207075 - -0.0211281 - -0.0211218 - -0.0206889 - -0.0198553 - -0.0186703 - -0.0172011 - -0.0155269 - -0.0137317 - -0.0118973 - -0.0100974 - -0.00839278 - -0.00682883 - -0.00543428 - -0.00422203 - -0.00319098 - -0.00232873 - -0.00161469 - -0.00102309 - -0.000525842 - -9.49211e-05 - 0.000295706 - 0.00066859 - 0.0010417 - 0.00142767 - 0.00183334 - 0.0022596 - 0.00270153 - 0.003149 - 0.00358751 - 0.0039995 - 0.00436595 - 0.00466822 - 0.00488978 - 0.005018 - 0.00504537 - 0.00497037 - 0.00479765 - 0.00453761 - 0.00420546 - 0.00381981 - 0.00340117 - 0.00297039 - 0.00254737 - 0.00215021 - 0.00179488 - 0.00149551 - 0.00126519 - 0.00111731 - 0.00106721 - 0.00113403 - 0.00134235 - 0.0017236 - 0.00231665 - 0.00316746 - 0.00432738 - 0.00584992 - 0.00778586 - 0.0101769 - 0.0130483 - 0.0164007 - 0.0202032 - 0.0243877 - 0.028846 - 0.0334309 - 0.0379624 - 0.0422376 - 0.046045 - 0.0491818 - 0.051471 - 0.0527786 - 0.0530262 - 0.0521987 - 0.0503461 - 0.0475783 - 0.0440543 - 0.0399673 - 0.035527 - 0.030942 - 0.0264044 - 0.022077 - 0.0180859 - 0.0145171 - 0.011417 - 0.00879761 - 0.00664221 - 0.00491358 - 0.00356139 - 0.00252917 - 0.00175985 - 0.00119979 - 0.00080145 - 0.000524545 - 0.000336376 - 0.000211352 - 0.000130113 - 7.8483e-05 - 4.63838e-05 - 2.68592e-05 - 1.5239e-05 - 8.47147e-06 - 4.6142e-06 - 2.46247e-06 - 1.2876e-06 - 6.59676e-07 - 3.31143e-07 - 1.62869e-07 - 7.84869e-08 - 3.70589e-08 - 1.71445e-08 - 7.77132e-09 - 3.45145e-09 - 1.50191e-09 - 6.4036e-10 - 2.67511e-10 - 1.09495e-10 - 4.39122e-11 - 1.72549e-11 - 6.6432e-12 - 2.50598e-12 - 9.26224e-13 - 3.35422e-13 - 1.19015e-13 - 4.13762e-14 - 1.40941e-14 - 4.7039e-15 - 1.53822e-15 - 4.92848e-16 - 1.5472e-16 - 4.759e-17 - 1.43424e-17 - 4.23512e-18 - 1.22531e-18 - 3.47346e-19 - 9.64753e-20 - 2.62547e-20 - 7.00059e-21 - 1.82896e-21 - 4.68283e-22 - 1.17935e-22 - 3.10821e-23 - 1.6359e-23 - 4.05988e-23 - 1.58397e-22 - 6.28053e-22 - 2.44515e-21 - 9.32834e-21 - 3.48693e-20 - 1.27708e-19 - 4.5828e-19 - 1.61132e-18 - 5.55095e-18 - 1.87366e-17 - 6.19655e-17 - 2.00792e-16 - 6.37501e-16 - 1.98313e-15 - 6.04448e-15 - 1.80511e-14 - 5.28183e-14 - 1.51427e-13 - 4.2536e-13 - 1.17071e-12 - 3.15702e-12 - 8.34147e-12 - 2.15946e-11 - 5.47752e-11 - 1.36132e-10 - 3.31491e-10 - 7.909e-10 - 1.84888e-09 - 4.23479e-09 - 9.50367e-09 - 2.08972e-08 - 4.50217e-08 - 9.50371e-08 - 1.96562e-07 - 3.98332e-07 - 7.90908e-07 - 1.53867e-06 - 2.93291e-06 - 5.47761e-06 - 1.00235e-05 - 1.79715e-05 - 3.15709e-05 - 5.43408e-05 - 9.16436e-05 - 0.000151431 - 0.000245169 - 0.000388912 - 0.000604471 - 0.000920526 - 0.00137352 - 0.00200802 - 0.00287635 - 0.00403694 - 0.00555138 - 0.00747978 - 0.00987455 - 0.0127729 - 0.0161884 - 0.0201035 - 0.0244622 - 0.0291667 - 0.0340773 - 0.0390171 - 0.0437822 - 0.0481565 - 0.0519306 - 0.0549224 - 0.0569978 - 0.0580884 - 0.0582047 - 0.057442 - 0.0559787 - 0.0540666 - 0.0520136 - 0.0501604 - 0.048853 - 0.0484132 - 0.0491098 - 0.0511319 - 0.0545677 - 0.0593893 - 0.0654475 - 0.0724749 - 0.0801003 - 0.0878732 - 0.095297 - 0.101868 - 0.107117 - 0.11065 - 0.112183 - 0.111565 - 0.108787 - 0.103986 - 0.0974183 - 0.0894397 - 0.0804662 - 0.0709366 - 0.0612756 - 0.0518631 - 0.0430107 - 0.0349495 - 0.0278261 - 0.0217078 - 0.016594 - 0.0124308 - 0.00912769 - 0.00657296 - 0.0046476 - 0.00323574 - 0.00223212 - 0.00154656 - 0.00110583 - 0.000853632 - 0.000749224 - 0.00076522 - 0.000884946 - 0.00109963 - 0.00140561 - 0.00180157 - 0.00228607 - 0.0028553 - 0.00350116 - 0.00421001 - 0.00496194 - 0.00573094 - 0.00648586 - 0.00719215 - 0.00781434 - 0.00831889 - 0.00867714 - 0.00886798 - 0.00887994 - 0.00871228 - 0.00837512 - 0.00788836 - 0.0072798 - 0.00658247 - 0.00583171 - 0.0050622 - 0.00430545 - 0.00358786 - 0.00292948 - 0.00234358 - 0.001837 - 0.00141082 - 0.00106163 - 0.000782729 - 0.00056544 - 0.00040022 - 0.000277554 - 0.000188596 - 0.000125561 - 8.19057e-05 - 5.23492e-05 - 3.27825e-05 - 2.01146e-05 - 1.20926e-05 - 7.123e-06 - 4.11097e-06 - 2.32472e-06 - 1.28813e-06 - 6.99559e-07 - 3.72774e-07 - 1.95909e-07 - 1.03891e-07 - 6.08753e-08 - 4.99608e-08 - 6.85721e-08 - 1.29035e-07 - 2.6531e-07 - 5.4877e-07 - 1.11869e-06 - 2.23728e-06 - 4.38521e-06 - 8.42213e-06 - 1.58488e-05 - 2.92219e-05 - 5.27908e-05 - 9.34425e-05 - 0.000162057 - 0.000275377 - 0.000458485 - 0.000747926 - 0.00119544 - 0.00187213 - 0.00287264 - 0.00431879 - 0.00636181 - 0.00918196 - 0.0129846 - 0.017991 - 0.0244243 - 0.032488 - 0.0423412 - 0.0540678 - 0.0676475 - 0.082928 - 0.0996065 - 0.117223 - 0.135167 - 0.15271 - 0.169045 - 0.183347 - 0.194841 - 0.202874 - 0.20697 - 0.206883 - 0.202619 - 0.194434 - 0.18281 - 0.168409 - 0.152009 - 0.134434 - 0.116489 - 0.0989001 - 0.0822709 - 0.0670553 - 0.0535496 - 0.0419003 - 0.0321229 - 0.0241295 - 0.017759 - 0.0128064 - 0.00904839 - 0.00626401 - 0.00424885 - 0.00282375 - 0.00183873 - 0.00117313 - 0.000733352 - 0.000449174 - 0.000269559 - 0.0001585 - 9.13153e-05 - 5.15458e-05 - 2.85089e-05 - 1.54491e-05 - 8.20284e-06 - 4.26738e-06 - 2.17518e-06 - 1.08634e-06 - 5.31587e-07 - 2.5487e-07 - 1.19729e-07 - 5.51086e-08 - 2.48527e-08 - 1.09816e-08 - 4.75449e-09 - 2.01706e-09 - 8.38997e-10 - 3.43464e-10 - 1.41906e-10 - 6.83484e-11 - 5.95613e-11 - 1.07176e-10 - 2.50595e-10 - 6.05453e-10 - 1.44505e-09 - 3.3834e-09 - 7.76313e-09 - 1.7453e-08 - 3.8445e-08 - 8.29752e-08 - 1.75466e-07 - 3.63559e-07 - 7.38063e-07 - 1.46808e-06 - 2.86115e-06 - 5.46349e-06 - 1.0222e-05 - 1.87387e-05 - 3.36572e-05 - 5.92318e-05 - 0.000102134 - 0.000172551 - 0.000285631 - 0.000463265 - 0.000736191 - 0.00114627 - 0.00174873 - 0.00261393 - 0.00382826 - 0.00549347 - 0.00772375 - 0.0106401 - 0.0143616 - 0.0189931 - 0.0246107 - 0.0312457 - 0.0388681 - 0.0473732 - 0.056573 - 0.0661947 - 0.075888 - 0.0852433 - 0.0938176 - 0.101168 - 0.106891 - 0.110657 - 0.11224 - 0.111547 - 0.108618 - 0.10363 - 0.096873 - 0.0887275 - 0.0796253 - 0.0700132 - 0.0603179 - 0.0509154 - 0.0421104 - 0.0341245 - 0.0270944 - 0.021078 - 0.0160663 - 0.0119989 - 0.00878015 - 0.00629506 - 0.00442216 - 0.00304373 - 0.00205265 - 0.00135631 - 0.000878095 - 0.000557007 - 0.000346191 - 0.000210818 - 0.000125788 - 7.35367e-05 - 4.21219e-05 - 2.364e-05 - 1.29994e-05 - 7.00387e-06 - 3.69733e-06 - 1.91239e-06 - 9.6917e-07 - 4.81239e-07 - 2.34131e-07 - 1.11608e-07 - 5.21273e-08 - 2.38547e-08 - 1.06959e-08 - 4.69895e-09 - 2.02265e-09 - 8.53052e-10 - 3.52508e-10 - 1.42724e-10 - 5.66193e-11 - 2.20073e-11 - 8.38123e-12 - 3.12741e-12 - 1.1434e-12 - 4.09589e-13 - 1.43759e-13 - 4.9438e-14 - 1.6658e-14 - 5.49945e-15 - 1.77891e-15 - 5.63801e-16 - 1.75079e-16 - 5.32697e-17 - 1.58804e-17 - 4.63854e-18 - 1.32751e-18 - 3.72246e-19 - 1.02273e-19 - 2.75313e-20 - 7.26155e-21 - 1.87659e-21 - 4.75167e-22 - 1.17886e-22 - 2.86561e-23 - 6.82699e-24 - 1.60225e-24 - 4.07911e-25 - 2.77213e-25 - 8.74879e-25 - 3.68199e-24 - 1.54686e-23 - 6.37332e-23 - 2.573e-22 - 1.01778e-21 - 3.94461e-21 - 1.49794e-20 - 5.57341e-20 - 2.03183e-19 - 7.25755e-19 - 2.53999e-18 - 8.70986e-18 - 2.92637e-17 - 9.63352e-17 - 3.10727e-16 - 9.81998e-16 - 3.04075e-15 - 9.22547e-15 - 2.74243e-14 - 7.98765e-14 - 2.27951e-13 - 6.37385e-13 - 1.74622e-12 - 4.68745e-12 - 1.23285e-11 - 3.17705e-11 - 8.02186e-11 - 1.98456e-10 - 4.81051e-10 - 1.1425e-09 - 2.65864e-09 - 6.06178e-09 - 1.35419e-08 - 2.96413e-08 - 6.357e-08 - 1.33581e-07 - 2.75029e-07 - 5.54814e-07 - 1.09662e-06 - 2.12374e-06 - 4.02982e-06 - 7.49218e-06 - 1.3648e-05 - 2.43594e-05 - 4.25994e-05 - 7.29926e-05 - 0.000122544 - 0.000201578 - 0.000324887 - 0.000513049 - 0.000793824 - 0.00120345 - 0.0017876 - 0.00260165 - 0.00370994 - 0.0051835 - 0.00709606 - 0.00951809 - 0.0125089 - 0.0161075 - 0.0203225 - 0.0251225 - 0.0304289 - 0.0361118 - 0.0419907 - 0.0478406 - 0.0534051 - 0.0584136 - 0.0626032 - 0.0657415 - 0.0676484 - 0.0682146 - 0.067413 - 0.0653031 - 0.0620266 - 0.0577961 - 0.0528776 - 0.0475695 - 0.042181 - 0.0370117 - 0.0323346 - 0.0283836 - 0.0253448 - 0.0233519 - 0.0224848 - 0.0227691 - 0.0241766 - 0.0266268 - 0.0299878 - 0.0340796 - 0.0386788 - 0.0435267 - 0.048341 - 0.0528317 - 0.0567191 - 0.0597539 - 0.0617366 - 0.062533 - 0.0620863 - 0.0604203 - 0.0576377 - 0.0539097 - 0.0494607 - 0.04455 - 0.0394516 - 0.0344361 - 0.0297549 - 0.0256293 - 0.0222441 - 0.0197445 - 0.0182362 - 0.0177867 - 0.0184265 - 0.020149 - 0.0229088 - 0.0266194 - 0.0311498 - 0.0363235 - 0.0419191 - 0.0476766 - 0.0533075 - 0.0585107 - 0.0629926 - 0.0664884 - 0.0687844 - 0.0697358 - 0.0692798 - 0.0674405 - 0.0643261 - 0.0601173 - 0.0550495 - 0.0493909 - 0.0434189 - 0.037398 - 0.0315613 - 0.0260975 - 0.0211437 - 0.0167841 - 0.0130543 - 0.0099482 - 0.00742802 - 0.00543424 - 0.0038953 - 0.00273577 - 0.00188259 - 0.00126931 - 0.00083853 - 0.000542757 - 0.000344214 - 0.000213889 - 0.000130222 - 7.76819e-05 - 4.54036e-05 - 2.60015e-05 - 1.45896e-05 - 8.02091e-06 - 4.32058e-06 - 2.28033e-06 - 1.1792e-06 - 5.97472e-07 - 2.96608e-07 - 1.44273e-07 - 6.87582e-08 - 3.21071e-08 - 1.46897e-08 - 6.58512e-09 - 2.89235e-09 - 1.24473e-09 - 5.24849e-10 - 2.16836e-10 - 8.77739e-11 - 3.48126e-11 - 1.35283e-11 - 5.15096e-12 - 1.92163e-12 - 7.02406e-13 - 2.51561e-13 - 8.82745e-14 - 3.03504e-14 - 1.02242e-14 - 3.37468e-15 - 1.09137e-15 - 3.45818e-16 - 1.07364e-16 - 3.26596e-17 - 9.73413e-18 - 2.84263e-18 - 8.13357e-19 - 2.28023e-19 - 6.26344e-20 - 1.68571e-20 - 4.4452e-21 - 1.14851e-21 - 2.90749e-22 - 7.21167e-23 - 1.75263e-23 - 4.17333e-24 - 9.73671e-25 - 2.22576e-25 - 4.98517e-26 - 1.09401e-26 - 2.35232e-27 - 4.95575e-28 - 1.02297e-28 - 2.06918e-29 - 4.10483e-30 - 8.05296e-31 - 1.68276e-31 - 5.81688e-32 - 5.72945e-32 - 8.98722e-32 - 1.48841e-31 - 2.43522e-31 - 3.90723e-31 - 6.14294e-31 - 9.46291e-31 - 1.42827e-30 - 2.11219e-30 - 3.06049e-30 - 4.34498e-30 - 6.04395e-30 - 8.23741e-30 - 1.10001e-29 - 1.43926e-29 - 1.8451e-29 - 2.3176e-29 - 2.85228e-29 - 3.4394e-29 - 4.0636e-29 - 4.70409e-29 - 5.33552e-29 - 5.92946e-29 - 6.4564e-29 - 6.88814e-29 - 7.20031e-29 - 7.37458e-29 - 7.40049e-29 - 7.27648e-29 - 7.01004e-29 - 6.61697e-29 - 6.11986e-29 - 5.54591e-29 - 4.92454e-29 - 4.28495e-29 - 3.65397e-29 - 3.05439e-29 - 2.50401e-29 - 2.01518e-29 - 1.59512e-29 - 1.24656e-29 - 9.68759e-30 - 7.58636e-30 - 6.11849e-30 - 5.23719e-30 - 4.89906e-30 - 5.06773e-30 - 5.71447e-30 - 6.81605e-30 - 8.35032e-30 - 1.02904e-29 - 1.25986e-29 - 1.52203e-29 - 1.80805e-29 - 2.10818e-29 - 2.4106e-29 - 2.70191e-29 - 2.96789e-29 - 3.19454e-29 - 3.36922e-29 - 3.48175e-29 - 3.5254e-29 - 3.49752e-29 - 3.39976e-29 - 3.23799e-29 - 3.02162e-29 - 2.76274e-29 - 2.47502e-29 - 2.17246e-29 - 1.86838e-29 - 1.57439e-29 - 1.29986e-29 - 1.05153e-29 - 8.33449e-30 - 6.47254e-30 - 4.92502e-30 - 3.67178e-30 - 2.68215e-30 - 1.91967e-30 - 1.34619e-30 - 9.24949e-31 - 6.22668e-31 - 4.10674e-31 - 2.65312e-31 - 1.67783e-31 - 1.03634e-31 - 6.20412e-32 - 3.50175e-32 - 1.66065e-32 - 2.07333e-33 - -1.29335e-32 - -3.33254e-32 - -6.57346e-32 - -1.2004e-31 - -2.11439e-31 - -3.6325e-31 - -6.1063e-31 - -1.00534e-30 - -1.62152e-30 - -2.5624e-30 - -3.96726e-30 - -6.018e-30 - -8.94393e-30 - -1.30231e-29 - -1.85779e-29 - -2.59639e-29 - -3.55481e-29 - -4.76781e-29 - -6.26401e-29 - -8.06095e-29 - -1.01597e-28 - -1.25394e-28 - -1.51534e-28 - -1.79261e-28 - -2.07531e-28 - -2.35041e-28 - -2.60289e-28 - -2.81667e-28 - -2.97579e-28 - -3.06572e-28 - -3.0747e-28 - -2.99489e-28 - -2.82337e-28 - -2.56257e-28 - -2.22043e-28 - -1.80993e-28 - -1.34833e-28 - -8.559e-29 - -3.54556e-29 - 1.33727e-29 - 5.884e-29 - 9.91687e-29 - 1.32966e-28 - 1.59294e-28 - 1.77705e-28 - 1.88232e-28 - 1.91343e-28 - 1.87867e-28 - 1.78899e-28 - 1.65689e-28 - 1.49539e-28 - 1.31702e-28 - 1.13303e-28 - 9.52853e-29 - 7.83769e-29 - 6.30823e-29 - 4.96957e-29 - 3.83289e-29 - 2.89468e-29 - 2.14088e-29 - 1.55063e-29 - 1.09973e-29 - 7.63256e-30 - 5.17511e-30 - 3.41125e-30 - 2.15552e-30 - 1.25007e-30 - 5.59989e-31 - -3.4232e-32 - -6.46293e-31 - -1.39487e-30 - -2.415e-30 - -3.86943e-30 - -5.95888e-30 - -8.9302e-30 - -1.30806e-29 - -1.87561e-29 - -2.63424e-29 - -3.62453e-29 - -4.88615e-29 - -6.45375e-29 - -8.35203e-29 - -1.05903e-28 - -1.31571e-28 - -1.60158e-28 - -1.91018e-28 - -2.23222e-28 - -2.55586e-28 - -2.8673e-28 - -3.15171e-28 - -3.39434e-28 - -3.58181e-28 - -3.70328e-28 - -3.75152e-28 - -3.72361e-28 - -3.62125e-28 - -3.45056e-28 - -3.22149e-28 - -2.94688e-28 - -2.64121e-28 - -2.31943e-28 - -1.99571e-28 - -1.68248e-28 - -1.38976e-28 - -1.12477e-28 - -8.91922e-29 - -6.92989e-29 - -5.27549e-29 - -3.93492e-29 - -2.87572e-29 - -2.05917e-29 - -1.4447e-29 - -9.93109e-30 - -6.6889e-30 - -4.41417e-30 - -2.85417e-30 - -1.80821e-30 - -1.12241e-30 - -6.82644e-31 - -4.06792e-31 - -2.37513e-31 - -1.35875e-31 - -7.61605e-32 - -4.18269e-32 - -2.2507e-32 - -1.18664e-32 - -6.12991e-33 - -3.10261e-33 - -1.53864e-33 - -7.47626e-34 - -3.55933e-34 - -1.6603e-34 - -7.58797e-35 - -3.39721e-35 - -1.48868e-35 - -6.35395e-36 - -2.56741e-36 - -8.06046e-37 - 2.52037e-37 - 1.39081e-36 - 3.36508e-36 - 7.33774e-36 - 1.54615e-35 - 3.18438e-35 - 6.42314e-35 - 1.26933e-34 - 2.45771e-34 - 4.66256e-34 - 8.6667e-34 - 1.57841e-33 - 2.81659e-33 - 4.92452e-33 - 8.43609e-33 - 1.41597e-32 - 2.32866e-32 - 3.75227e-32 - 5.92404e-32 - 9.16388e-32 - 1.38892e-31 - 2.06259e-31 - 3.00113e-31 - 4.27852e-31 - 5.97639e-31 - 8.1794e-31 - 1.09683e-30 - 1.44111e-30 - 1.8552e-30 - 2.34002e-30 - 2.89192e-30 - 3.50179e-30 - 4.15461e-30 - 4.82956e-30 - 5.50074e-30 - 6.13864e-30 - 6.7121e-30 - 7.19077e-30 - 7.5474e-30 - 7.75857e-30 - 7.7976e-30 - 7.5884e-30 - 6.76448e-30 - 3.46619e-30 - -1.15254e-29 - -8.28018e-29 - -4.21322e-28 - -2.00418e-27 - -9.26322e-27 - -4.18846e-26 - -1.85506e-25 - -8.04966e-25 - -3.42238e-24 - -1.42566e-23 - -5.81889e-23 - -2.32702e-22 - -9.11798e-22 - -3.50052e-21 - -1.31675e-20 - -4.85302e-20 - -1.7525e-19 - -6.20068e-19 - -2.1496e-18 - -7.30153e-18 - -2.43e-17 - -7.92383e-17 - -2.53164e-16 - -7.92509e-16 - -2.43077e-15 - -7.30501e-15 - -2.15097e-14 - -6.20561e-14 - -1.75417e-13 - -4.85843e-13 - -1.31843e-12 - -3.50554e-12 - -9.13248e-12 - -2.3311e-11 - -5.83e-11 - -1.42861e-10 - -3.43001e-10 - -8.0689e-10 - -1.85982e-09 - -4.20013e-09 - -9.29377e-09 - -2.01492e-08 - -4.28017e-08 - -8.90843e-08 - -1.81668e-07 - -3.62987e-07 - -7.10627e-07 - -1.3631e-06 - -2.56185e-06 - -4.71753e-06 - -8.51162e-06 - -1.50469e-05 - -2.60626e-05 - -4.4231e-05 - -7.35482e-05 - -0.000119827 - -0.000191281 - -0.000299177 - -0.000458481 - -0.000688416 - -0.00101278 - -0.00145989 - -0.00206186 - -0.00285323 - -0.00386857 - -0.00513927 - -0.00668942 - -0.00853125 - -0.0106604 - -0.0130518 - -0.0156569 - -0.0184025 - -0.0211927 - -0.0239128 - -0.026437 - -0.0286371 - -0.0303935 - -0.0316057 - -0.0322019 - -0.0321457 - -0.0314398 - -0.0301256 - -0.0282785 - -0.0260006 - -0.0234103 - -0.020631 - -0.0177807 - -0.0149619 - -0.0122546 - -0.00971227 - -0.00735997 - -0.00519609 - -0.00319598 - -0.00131746 - 0.000492548 - 0.00229074 - 0.00413016 - 0.00605343 - 0.00808678 - 0.0102356 - 0.0124818 - 0.0147831 - 0.0170751 - 0.0192757 - 0.0212916 - 0.0230263 - 0.0243896 - 0.0253062 - 0.0257239 - 0.0256186 - 0.0249975 - 0.0238982 - 0.0223855 - 0.0205448 - 0.0184745 - 0.0162773 - 0.0140516 - 0.0118852 - 0.00984971 - 0.00799793 - 0.0063631 - 0.00496017 - 0.00378845 - 0.00283506 - 0.00207875 - 0.00149341 - 0.00105122 - 0.000725027 - 0.000489976 - 0.000324492 - 0.000210664 - 0.00013421 - 8.41738e-05 - 5.24698e-05 - 3.34056e-05 - 2.3238e-05 - 1.98238e-05 - 2.23914e-05 - 3.14497e-05 - 4.88383e-05 - 7.7918e-05 - 0.000123889 - 0.000194214 - 0.000299116 - 0.000452071 - 0.000670232 - 0.00097465 - 0.00139016 - 0.00194481 - 0.00266861 - 0.00359163 - 0.00474135 - 0.00613926 - 0.00779712 - 0.00971309 - 0.0118682 - 0.0142236 - 0.0167199 - 0.0192776 - 0.0218001 - 0.0241795 - 0.0263033 - 0.0280635 - 0.0293652 - 0.0301353 - 0.030329 - 0.0299345 - 0.0289739 - 0.0275012 - 0.0255973 - 0.0233628 - 0.0209089 - 0.0183485 - 0.0157878 - 0.0133194 - 0.0110173 - 0.00893483 - 0.00710404 - 0.0055377 - 0.00423219 - 0.00317136 - 0.00233065 - 0.00168094 - 0.00119189 - 0.000834605 - 0.000583648 - 0.00041835 - 0.000323717 - 0.000291017 - 0.000318222 - 0.000410421 - 0.000580238 - 0.000848215 - 0.00124303 - 0.00180133 - 0.00256695 - 0.00358917 - 0.00491979 - 0.0066089 - 0.0086993 - 0.0112199 - 0.0141788 - 0.017556 - 0.0212986 - 0.0253171 - 0.0294858 - 0.0336472 - 0.0376203 - 0.0412128 - 0.0442363 - 0.0465224 - 0.0479383 - 0.0483994 - 0.0478778 - 0.046405 - 0.044069 - 0.0410051 - 0.0373834 - 0.0333931 - 0.0292262 - 0.0250625 - 0.0210578 - 0.0173356 - 0.013983 - 0.011051 - 0.00855731 - 0.00649248 - 0.00482638 - 0.00351536 - 0.00250876 - 0.00175427 - 0.00120201 - 0.00080715 - 0.000531421 - 0.000343513 - 0.000218867 - 0.00013901 - 9.07312e-05 - 6.52515e-05 - 5.75322e-05 - 6.57899e-05 - 9.1257e-05 - 0.000138186 - 0.000214067 - 0.000329998 - 0.000501116 - 0.000746956 - 0.00109158 - 0.00156332 - 0.00219384 - 0.00301654 - 0.00406401 - 0.00536461 - 0.00693838 - 0.00879257 - 0.0109172 - 0.0132813 - 0.015831 - 0.018489 - 0.0211571 - 0.0237211 - 0.0260585 - 0.028048 - 0.0295795 - 0.0305645 - 0.0309443 - 0.0306958 - 0.0298343 - 0.0284112 - 0.0265094 - 0.0242352 - 0.0217085 - 0.0190525 - 0.0163836 - 0.013804 - 0.0113955 - 0.00921728 - 0.00730479 - 0.00567218 - 0.00431547 - 0.00321695 - 0.00234961 - 0.00168146 - 0.00117899 - 0.00080998 - 0.000545223 - 0.000359593 - 0.000232372 - 0.000147128 - 9.12729e-05 - 5.54786e-05 - 3.30405e-05 - 1.92799e-05 - 1.1023e-05 - 6.1749e-06 - 3.38921e-06 - 1.82265e-06 - 9.60385e-07 - 4.9582e-07 - 2.50807e-07 - 1.24306e-07 - 6.03645e-08 - 2.87215e-08 - 1.33897e-08 - 6.11605e-09 - 2.73721e-09 - 1.20029e-09 - 5.15726e-10 - 2.17179e-10 - 8.97827e-11 - 3.68248e-11 - 1.59826e-11 - 9.75677e-12 - 1.25514e-11 - 2.62185e-11 - 6.09961e-11 - 1.42097e-10 - 3.25442e-10 - 7.30675e-10 - 1.60748e-09 - 3.46506e-09 - 7.31836e-09 - 1.51444e-08 - 3.07064e-08 - 6.10016e-08 - 1.18738e-07 - 2.26453e-07 - 4.23156e-07 - 7.74749e-07 - 1.38982e-06 - 2.44282e-06 - 4.20691e-06 - 7.09857e-06 - 1.17359e-05 - 1.90106e-05 - 3.01727e-05 - 4.69212e-05 - 7.14926e-05 - 0.000106731 - 0.000156119 - 0.000223747 - 0.000314193 - 0.000432288 - 0.000582755 - 0.000769726 - 0.000996145 - 0.00126312 - 0.0015693 - 0.00191031 - 0.00227844 - 0.00266262 - 0.00304871 - 0.00342027 - 0.0037596 - 0.00404912 - 0.00427283 - 0.00441781 - 0.00447545 - 0.00444224 - 0.00432021 - 0.00411666 - 0.00384346 - 0.00351591 - 0.00315133 - 0.00276754 - 0.00238148 - 0.00200803 - 0.00165923 - 0.00134382 - 0.00106726 - 0.000831969 - 0.000637896 - 0.000483181 - 0.000364871 - 0.000279608 - 0.000224215 - 0.000196152 - 0.00019381 - 0.000216651 - 0.000265178 - 0.000340754 - 0.000445276 - 0.000580727 - 0.000748638 - 0.000949492 - 0.00118215 - 0.00144335 - 0.00172735 - 0.00202585 - 0.00232812 - 0.00262156 - 0.0028924 - 0.00312677 - 0.00331187 - 0.00343707 - 0.00349494 - 0.003482 - 0.00339903 - 0.003251 - 0.00304661 - 0.00279739 - 0.00251667 - 0.00221838 - 0.00191595 - 0.00162131 - 0.00134427 - 0.00109206 - 0.00086924 - 0.000677908 - 0.000518011 - 0.000387832 - 0.000284502 - 0.000204486 - 0.000144006 - 9.93646e-05 - 6.7177e-05 - 4.44986e-05 - 2.88808e-05 - 1.83658e-05 - 1.14431e-05 - 6.98584e-06 - 4.17858e-06 - 2.44893e-06 - 1.40624e-06 - 7.91189e-07 - 4.36152e-07 - 2.35577e-07 - 1.24671e-07 - 6.46445e-08 - 3.28425e-08 - 1.63485e-08 - 7.97365e-09 - 3.81042e-09 - 1.78412e-09 - 8.18492e-10 - 3.67909e-10 - 1.62033e-10 - 6.99202e-11 - 2.95624e-11 - 1.22465e-11 - 4.97076e-12 - 1.97683e-12 - 7.70284e-13 - 2.94065e-13 - 1.09939e-13 - 4.00913e-14 - 1.37519e-14 - 2.83217e-15 - -5.1357e-15 - -1.99766e-14 - -5.99553e-14 - -1.72282e-13 - -4.83806e-13 - -1.3308e-12 - -3.58656e-12 - -9.4706e-12 - -2.45027e-11 - -6.21137e-11 - -1.54276e-10 - -3.75444e-10 - -8.95217e-10 - -2.09146e-09 - -4.78748e-09 - -1.07375e-08 - -2.35957e-08 - -5.08043e-08 - -1.07178e-07 - -2.21538e-07 - -4.48669e-07 - -8.90311e-07 - -1.73099e-06 - -3.29749e-06 - -6.15473e-06 - -1.12557e-05 - -2.01684e-05 - -3.54086e-05 - -6.09091e-05 - -0.000102658 - -0.000169527 - -0.000274298 - -0.000434855 - -0.000675463 - -0.00102801 - -0.00153295 - -0.00223973 - -0.00320628 - -0.00449721 - -0.00618047 - -0.00832218 - -0.0109797 - -0.0141932 - -0.0179765 - -0.0223084 - -0.0271249 - -0.032315 - -0.0377206 - -0.0431408 - -0.0483433 - -0.0530787 - -0.0571007 - -0.0601865 - -0.0621577 - -0.0628967 - -0.062359 - -0.0605773 - -0.0576586 - -0.0537732 - -0.0491391 - -0.0440019 - -0.0386137 - -0.0332143 - -0.0280153 - -0.0231898 - -0.0188668 - -0.0151321 - -0.012033 - -0.00958594 - -0.00778576 - -0.00661503 - -0.00605167 - -0.0060743 - -0.00666458 - -0.00780636 - -0.00948195 - -0.0116662 - -0.0143193 - -0.0173792 - -0.020756 - -0.024328 - -0.0279419 - -0.0314171 - -0.0345546 - -0.0371497 - -0.0390077 - -0.0399608 - -0.0398843 - -0.0387107 - -0.0364381 - -0.0331337 - -0.0289313 - -0.0240213 - -0.0186365 - -0.0130343 - -0.00747615 - -0.00220866 - 0.00255415 - 0.00664412 - 0.00994752 - 0.0124083 - 0.0140264 - 0.0148508 - 0.0149705 - 0.0145022 - 0.0135779 - 0.0123329 - 0.0108956 - 0.00937953 - 0.00787832 - 0.00646298 - 0.00518198 - 0.00406317 - 0.00311692 - 0.00234001 - 0.0017197 - 0.00123741 - 0.00087191 - 0.000601693 - 0.000406692 - 0.000269261 - 0.000174633 - 0.000110952 - 6.90568e-05 - 4.21022e-05 - 2.51345e-05 - 1.46728e-05 - 8.33422e-06 - 4.52151e-06 - 2.17167e-06 - 5.58957e-07 - -8.62176e-07 - -2.59036e-06 - -5.20476e-06 - -9.50482e-06 - -1.66806e-05 - -2.85308e-05 - -4.77436e-05 - -7.82481e-05 - -0.000125638 - -0.000197646 - -0.000304643 - -0.000460074 - -0.000680773 - -0.000986991 - -0.00140204 - -0.0019514 - -0.00266113 - -0.0035557 - -0.004655 - -0.00597107 - -0.00750448 - -0.00924115 - -0.0111498 - -0.013181 - -0.0152674 - -0.0173267 - -0.0192667 - -0.020991 - -0.0224077 - -0.0234368 - -0.0240179 - -0.0241162 - -0.0237257 - -0.0228701 - -0.0215999 - -0.0199882 - -0.0181231 - -0.0161 - -0.0140139 - -0.0119517 - -0.00998697 - -0.00817667 - -0.00655927 - -0.00515551 - -0.00397031 - -0.00299581 - -0.00221484 - -0.00160437 - -0.00113869 - -0.00079185 - -0.000539532 - -0.000360187 - -0.0002356 - -0.000150994 - -9.48162e-05 - -5.83366e-05 - -3.51671e-05 - -2.07715e-05 - -1.20209e-05 - -6.81621e-06 - -3.78692e-06 - -2.06142e-06 - -1.09949e-06 - -5.74622e-07 - -2.94344e-07 - -1.47969e-07 - -7.34581e-08 - -3.70781e-08 - -2.14022e-08 - -1.86896e-08 - -2.78684e-08 - -5.43091e-08 - -1.12445e-07 - -2.32243e-07 - -4.71751e-07 - -9.39625e-07 - -1.83401e-06 - -3.50752e-06 - -6.57263e-06 - -1.20674e-05 - -2.17084e-05 - -3.82629e-05 - -6.60791e-05 - -0.000111812 - -0.000185373 - -0.000301123 - -0.000479268 - -0.000747393 - -0.00114198 - -0.00170963 - -0.00250774 - -0.00360413 - -0.00507523 - -0.00700241 - -0.00946622 - -0.0125384 - -0.0162721 - -0.0206911 - -0.0257786 - -0.0314682 - -0.0376376 - -0.0441072 - -0.0506445 - -0.0569761 - -0.0628044 - -0.0678304 - -0.0717786 - -0.0744223 - -0.0756046 - -0.0752542 - -0.0733923 - -0.0701309 - -0.0656613 - -0.0602356 - -0.0541439 - -0.0476884 - -0.0411598 - -0.0348168 - -0.0288722 - -0.0234843 - -0.0187563 - -0.0147398 - -0.0114436 - -0.00884406 - -0.00689638 - -0.00554479 - -0.00473098 - -0.00439975 - -0.00450194 - -0.00499464 - -0.00583918 - -0.00699773 - -0.00842913 - -0.0100849 - -0.0119061 - -0.0138222 - -0.0157513 - -0.0176033 - -0.0192845 - -0.0207045 - -0.0217825 - -0.022455 - -0.0226813 - -0.0224474 - -0.0217673 - -0.0206815 - -0.0192529 - -0.017561 - -0.0156942 - -0.0137425 - -0.0117904 - -0.00991122 - -0.00816327 - -0.00658776 - -0.00520892 - -0.00403548 - -0.00306323 - -0.00227824 - -0.00166019 - -0.00118537 - -0.000829248 - -0.000568398 - -0.000381731 - -0.000251189 - -0.000161949 - -0.000102304 - -6.33209e-05 - -3.84005e-05 - -2.28172e-05 - -1.32839e-05 - -7.5775e-06 - -4.23509e-06 - -2.31919e-06 - -1.24436e-06 - -6.54176e-07 - -3.3696e-07 - -1.70059e-07 - -8.40926e-08 - -4.0743e-08 - -1.93412e-08 - -8.99607e-09 - -4.09976e-09 - -1.83063e-09 - -8.00902e-10 - -3.43317e-10 - -1.44194e-10 - -5.93386e-11 - -2.39256e-11 - -9.45207e-12 - -3.6587e-12 - -1.3876e-12 - -5.15629e-13 - -1.87736e-13 - -6.69724e-14 - -2.34089e-14 - -8.01681e-15 - -2.69004e-15 - -8.84411e-16 - -2.84896e-16 - -8.99196e-17 - -2.78074e-17 - -8.42563e-18 - -2.50139e-18 - -7.27608e-19 - -2.07372e-19 - -5.79081e-20 - -1.5844e-20 - -4.24746e-21 - -1.11565e-21 - -2.87122e-22 - -7.24003e-23 - -1.78876e-23 - -4.33011e-24 - -1.02703e-24 - -2.38674e-25 - -5.43454e-26 - -1.21243e-26 - -2.65027e-27 - -5.67621e-28 - -1.19114e-28 - -2.4491e-29 - -4.93384e-30 - -9.73871e-31 - -1.88345e-31 - -3.56898e-32 - -6.62629e-33 - -1.20541e-33 - -2.14849e-34 - -3.75207e-35 - -6.42015e-36 - -1.07636e-36 - -1.76809e-37 - -2.8457e-38 - -4.48756e-39 - -6.93375e-40 - -1.0497e-40 - -1.55702e-41 - -2.26289e-42 - -3.22233e-43 - -4.49585e-44 - -6.14598e-45 - -8.23204e-46 - -1.08034e-46 - -1.38915e-47 - -1.75016e-48 - -2.16043e-49 - -2.61301e-50 - -3.09655e-51 - -3.59544e-52 - -4.09037e-53 - -4.55943e-54 - -4.97961e-55 - -5.32865e-56 - -5.58696e-57 - -5.73945e-58 - -5.777e-59 - -5.69733e-60 - -5.50525e-61 - -5.21218e-62 - -4.83503e-63 - -4.39456e-64 - -3.91353e-65 - -3.41474e-66 - -2.91934e-67 - -2.44539e-68 - -2.00701e-69 - -1.61394e-70 - -1.27163e-71 - -9.81684e-73 - -7.4254e-74 - -5.50306e-75 - -3.996e-76 - -2.84305e-77 - -1.98189e-78 - -1.35367e-79 - -9.05901e-81 - -5.94e-82 - -3.81618e-83 - -2.4022e-84 - -1.48158e-85 - -8.9532e-87 - -5.30113e-88 - -3.07535e-89 - -1.74807e-90 - -9.73553e-92 - -5.31247e-93 - -2.84034e-94 - -1.48793e-95 - -7.63713e-97 - -3.84074e-98 - -1.8925e-99 --9.13682e-101 --4.32206e-102 --2.00319e-103 --9.09687e-105 - -4.0476e-106 --1.76458e-107 --7.53737e-109 --3.15454e-110 --1.29357e-111 --5.19732e-113 --2.04539e-114 --6.24659e-116 - 4.28841e-115 - 1.11058e-113 - 2.79859e-112 - 6.90971e-111 - 1.67155e-109 - 3.96199e-108 - 9.20121e-107 - 2.0937e-105 - 4.66788e-104 - 1.01967e-102 - 2.18243e-101 - 4.57675e-100 - 9.40394e-99 - 1.89321e-97 - 3.73445e-96 - 7.21755e-95 - 1.36675e-93 - 2.53587e-92 - 4.61001e-91 - 8.21131e-90 - 1.43305e-88 - 2.45044e-87 - 4.1055e-86 - 6.73945e-85 - 1.08398e-83 - 1.70825e-82 - 2.63767e-81 - 3.99049e-80 - 5.91519e-79 - 8.59108e-78 - 1.22254e-76 - 1.70458e-75 - 2.32866e-74 - 3.11698e-73 - 4.08787e-72 - 5.25288e-71 - 6.61355e-70 - 8.15848e-69 - 9.86098e-68 - 1.1678e-66 - 1.35504e-65 - 1.54054e-64 - 1.71606e-63 - 1.87296e-62 - 2.0029e-61 - 2.09859e-60 - 2.15444e-59 - 2.16709e-58 - 2.13578e-57 - 2.0624e-56 - 1.95131e-55 - 1.80891e-54 - 1.64302e-53 - 1.4622e-52 - 1.27499e-51 - 1.08929e-50 - 9.1184e-50 - 7.47876e-49 - 6.01004e-48 - 4.73219e-47 - 3.65077e-46 - 2.75958e-45 - 2.0438e-44 - 1.4831e-43 - 1.05448e-42 - 7.34589e-42 - 5.01404e-41 - 3.35326e-40 - 2.19727e-39 - 1.41071e-38 - 8.87415e-38 - 5.46958e-37 - 3.30307e-36 - 1.95442e-35 - 1.13307e-34 - 6.4362e-34 - 3.58213e-33 - 1.95339e-32 - 1.0437e-31 - 5.46381e-31 - 2.80256e-30 - 1.40848e-29 - 6.93557e-29 - 3.34619e-28 - 1.58182e-27 - 7.32655e-27 - 3.32491e-26 - 1.47841e-25 - 6.44094e-25 - 2.74941e-24 - 1.14992e-23 - 4.71227e-23 - 1.89204e-22 - 7.44335e-22 - 2.86908e-21 - 1.08356e-20 - 4.00961e-20 - 1.45374e-19 - 5.16427e-19 - 1.7975e-18 - 6.13004e-18 - 2.04831e-17 - 6.70602e-17 - 2.15115e-16 - 6.76105e-16 - 2.08206e-15 - 6.28218e-15 - 1.85722e-14 - 5.37965e-14 - 1.5268e-13 - 4.24566e-13 - 1.15677e-12 - 3.08805e-12 - 8.07715e-12 - 2.07e-11 - 5.19778e-11 - 1.2788e-10 - 3.08265e-10 - 7.28087e-10 - 1.68492e-09 - 3.82042e-09 - 8.48752e-09 - 1.84751e-08 - 3.94031e-08 - 8.23398e-08 - 1.68588e-07 - 3.38205e-07 - 6.64769e-07 - 1.28026e-06 - 2.4158e-06 - 4.46645e-06 - 8.09097e-06 - 1.43607e-05 - 2.49739e-05 - 4.25535e-05 - 7.10428e-05 - 0.00011621 - 0.000186252 - 0.00029248 - 0.000450017 - 0.00067842 - 0.00100209 - 0.00145027 - 0.0020565 - 0.00285723 - 0.00388954 - 0.00518787 - 0.00677979 - 0.00868121 - 0.0108913 - 0.0133881 - 0.0161248 - 0.0190285 - 0.0220016 - 0.0249252 - 0.027667 - 0.0300899 - 0.0320639 - 0.0334772 - 0.0342467 - 0.0343261 - 0.0337107 - 0.0324375 - 0.0305819 - 0.02825 - 0.0255687 - 0.0226744 - 0.0197016 - 0.0167727 - 0.0139907 - - - - - -5.22724e-31 - -6.50517e-31 - -7.93196e-31 - -9.47624e-31 - -1.1092e-30 - -1.27183e-30 - -1.42734e-30 - -1.56123e-30 - -1.62865e-30 - -1.42829e-30 - 3.51589e-32 - 7.69927e-30 - 4.56532e-29 - 2.29074e-28 - 1.09734e-27 - 5.12432e-27 - 2.34221e-26 - 1.04873e-25 - 4.60069e-25 - 1.97749e-24 - 8.328e-24 - 3.43641e-23 - 1.38933e-22 - 5.50355e-22 - 2.13608e-21 - 8.12322e-21 - 3.02675e-20 - 1.105e-19 - 3.9526e-19 - 1.38529e-18 - 4.75704e-18 - 1.60055e-17 - 5.27641e-17 - 1.70429e-16 - 5.3937e-16 - 1.6725e-15 - 5.0814e-15 - 1.51264e-14 - 4.41191e-14 - 1.26082e-13 - 3.53035e-13 - 9.68542e-13 - 2.60349e-12 - 6.85694e-12 - 1.76946e-11 - 4.47394e-11 - 1.10834e-10 - 2.69027e-10 - 6.39816e-10 - 1.49091e-09 - 3.40395e-09 - 7.61469e-09 - 1.66901e-08 - 3.58427e-08 - 7.5419e-08 - 1.55488e-07 - 3.14088e-07 - 6.21643e-07 - 1.2055e-06 - 2.29051e-06 - 4.26416e-06 - 7.77805e-06 - 1.3901e-05 - 2.4342e-05 - 4.17643e-05 - 7.02086e-05 - 0.000115641 - 0.000186625 - 0.000295098 - 0.000457192 - 0.000694014 - 0.00103222 - 0.00150424 - 0.00214782 - 0.00300479 - 0.00411877 - 0.00553169 - 0.00727922 - 0.00938532 - 0.0118563 - 0.0146753 - 0.0177977 - 0.0211483 - 0.0246221 - 0.0280875 - 0.0313934 - 0.0343799 - 0.0368904 - 0.0387854 - 0.0399557 - 0.0403326 - 0.0398957 - 0.0386751 - 0.036749 - 0.0342377 - 0.0312926 - 0.0280847 - 0.0247919 - 0.0215871 - 0.0186297 - 0.0160581 - 0.013987 - 0.0125051 - 0.0116762 - 0.0115397 - 0.0121108 - 0.0133811 - 0.0153167 - 0.0178562 - 0.0209094 - 0.0243561 - 0.0280473 - 0.0318096 - 0.0354525 - 0.0387797 - 0.041602 - 0.0437517 - 0.0450968 - 0.045552 - 0.0450867 - 0.043727 - 0.041553 - 0.03869 - 0.0352968 - 0.031551 - 0.027633 - 0.0237127 - 0.0199375 - 0.0164247 - 0.0132575 - 0.0104848 - 0.00812456 - 0.00616843 - 0.00458866 - 0.00334453 - 0.00238847 - 0.00167125 - 0.00114578 - 0.000769656 - 0.000506558 - 0.000326662 - 0.000206398 - 0.000127776 - 7.75048e-05 - 4.60623e-05 - 2.68225e-05 - 1.53035e-05 - 8.55494e-06 - 4.68577e-06 - 2.51468e-06 - 1.32227e-06 - 6.81232e-07 - 3.4388e-07 - 1.70081e-07 - 8.24215e-08 - 3.91348e-08 - 1.82063e-08 - 8.29885e-09 - 3.70639e-09 - 1.62189e-09 - 6.95389e-10 - 2.92132e-10 - 1.2026e-10 - 4.85509e-11 - 1.93322e-11 - 7.90272e-12 - 4.15729e-12 - 4.69603e-12 - 1.00995e-11 - 2.54898e-11 - 6.48543e-11 - 1.62322e-10 - 3.9828e-10 - 9.57567e-10 - 2.25575e-09 - 5.20657e-09 - 1.17747e-08 - 2.60905e-08 - 5.66438e-08 - 1.20492e-07 - 2.51133e-07 - 5.12843e-07 - 1.02613e-06 - 2.01167e-06 - 3.8641e-06 - 7.27238e-06 - 1.34104e-05 - 2.42295e-05 - 4.28927e-05 - 7.43975e-05 - 0.000126436 - 0.000210533 - 0.000343484 - 0.000549073 - 0.000859983 - 0.00131974 - 0.00198436 - 0.00292342 - 0.00421986 - 0.00596819 - 0.00827035 - 0.011229 - 0.0149381 - 0.019471 - 0.0248666 - 0.0311158 - 0.0381491 - 0.0458271 - 0.0539385 - 0.062203 - 0.0702847 - 0.0778121 - 0.0844055 - 0.0897079 - 0.0934174 - 0.0953152 - 0.095287 - 0.0933346 - 0.0895758 - 0.0842323 - 0.0776088 - 0.0700637 - 0.0619781 - 0.0537245 - 0.0456401 - 0.0380067 - 0.0310392 - 0.0248817 - 0.0196118 - 0.0152499 - 0.0117717 - 0.00912236 - 0.00722982 - 0.00601571 - 0.00540334 - 0.0053222 - 0.00570923 - 0.00650744 - 0.00766253 - 0.00911876 - 0.0108148 - 0.0126804 - 0.0146355 - 0.0165902 - 0.0184485 - 0.020113 - 0.0214914 - 0.022504 - 0.0230902 - 0.0232141 - 0.022868 - 0.0220728 - 0.020876 - 0.0193474 - 0.0175725 - 0.0156453 - 0.0136615 - 0.0117119 - 0.00987887 - 0.00823454 - 0.00684165 - 0.00575724 - 0.00503837 - 0.00474935 - 0.00496969 - 0.00580156 - 0.00737575 - 0.00985451 - 0.0134302 - 0.0183181 - 0.0247429 - 0.0329181 - 0.0430191 - 0.0551515 - 0.0693182 - 0.0853881 - 0.103073 - 0.121914 - 0.141292 - 0.160443 - 0.178512 - 0.194604 - 0.207861 - 0.217536 - 0.223062 - 0.224109 - 0.220611 - 0.212781 - 0.201084 - 0.18619 - 0.168917 - 0.15015 - 0.130773 - 0.111595 - 0.0933057 - 0.0764379 - 0.0613545 - 0.0482527 - 0.037182 - 0.0280725 - 0.0207666 - 0.0150518 - 0.0106892 - 0.00743777 - 0.00507078 - 0.00338723 - 0.00221692 - 0.00142165 - 0.000893251 - 0.000549908 - 0.000331699 - 0.000196035 - 0.000113517 - 6.44059e-05 - 3.58036e-05 - 1.95013e-05 - 1.04073e-05 - 5.4419e-06 - 2.78804e-06 - 1.39954e-06 - 6.88345e-07 - 3.31715e-07 - 1.56625e-07 - 7.24592e-08 - 3.28445e-08 - 1.45871e-08 - 6.34764e-09 - 2.7064e-09 - 1.1306e-09 - 4.62768e-10 - 1.8559e-10 - 7.29258e-11 - 2.80766e-11 - 1.05912e-11 - 3.91457e-12 - 1.41762e-12 - 5.03003e-13 - 1.74872e-13 - 5.95668e-14 - 1.98805e-14 - 6.50108e-15 - 2.08296e-15 - 6.53904e-16 - 2.01133e-16 - 6.06164e-17 - 1.78992e-17 - 5.17861e-18 - 1.46801e-18 - 4.07741e-19 - 1.10962e-19 - 2.95871e-20 - 7.72981e-21 - 1.97881e-21 - 4.96994e-22 - 1.2513e-22 - 4.27973e-23 - 6.25412e-23 - 2.27169e-22 - 8.96772e-22 - 3.49045e-21 - 1.3316e-20 - 4.97752e-20 - 1.82301e-19 - 6.54186e-19 - 2.30012e-18 - 7.92386e-18 - 2.67461e-17 - 8.84545e-17 - 2.86627e-16 - 9.1002e-16 - 2.83088e-15 - 8.62837e-15 - 2.57675e-14 - 7.5397e-14 - 2.16158e-13 - 6.07193e-13 - 1.67116e-12 - 4.50658e-12 - 1.19073e-11 - 3.08258e-11 - 7.81904e-11 - 1.94325e-10 - 4.73197e-10 - 1.12899e-09 - 2.63923e-09 - 6.04507e-09 - 1.35663e-08 - 2.98304e-08 - 6.42676e-08 - 1.35663e-07 - 2.80589e-07 - 5.6861e-07 - 1.12901e-06 - 2.19641e-06 - 4.18667e-06 - 7.81917e-06 - 1.43083e-05 - 2.5654e-05 - 4.50668e-05 - 7.75704e-05 - 0.000130819 - 0.000216165 - 0.000349973 - 0.000555164 - 0.000862869 - 0.00131403 - 0.00196066 - 0.00286641 - 0.00410592 - 0.00576262 - 0.00792443 - 0.0106771 - 0.0140955 - 0.0182326 - 0.0231079 - 0.0286958 - 0.0349164 - 0.0416297 - 0.0486353 - 0.0556796 - 0.0624697 - 0.0686941 - 0.0740496 - 0.07827 - 0.0811551 - 0.0825952 - 0.0825891 - 0.0812525 - 0.0788147 - 0.0756052 - 0.0720296 - 0.068538 - 0.0655882 - 0.0636076 - 0.0629552 - 0.0638897 - 0.0665429 - 0.0709037 - 0.0768119 - 0.0839642 - 0.0919334 - 0.100197 - 0.108179 - 0.115294 - 0.120995 - 0.124823 - 0.126445 - 0.125678 - 0.122506 - 0.117071 - 0.10966 - 0.100669 - 0.0905633 - 0.0798348 - 0.0689607 - 0.0583676 - 0.0484063 - 0.0393372 - 0.0313261 - 0.0244505 - 0.0187122 - 0.0140553 - 0.0103846 - 0.00758494 - 0.00553752 - 0.00413347 - 0.00328399 - 0.00292721 - 0.00303184 - 0.00359822 - 0.00465668 - 0.00626355 - 0.00849453 - 0.0114355 - 0.0151707 - 0.019769 - 0.0252679 - 0.0316588 - 0.0388727 - 0.0467703 - 0.0551377 - 0.0636899 - 0.0720829 - 0.0799342 - 0.08685 - 0.092458 - 0.0964398 - 0.0985609 - 0.0986938 - 0.0968305 - 0.0930832 - 0.0876733 - 0.0809096 - 0.0731593 - 0.0648151 - 0.0562625 - 0.0478519 - 0.0398764 - 0.0325589 - 0.0260472 - 0.0204168 - 0.0156802 - 0.0117992 - 0.00869946 - 0.00628445 - 0.00444814 - 0.0030848 - 0.0020961 - 0.00139552 - 0.00091032 - 0.000581822 - 0.000364354 - 0.000223559 - 0.0001344 - 7.91667e-05 - 4.56902e-05 - 2.5837e-05 - 1.43154e-05 - 7.77184e-06 - 4.13516e-06 - 2.15827e-06 - 1.10966e-06 - 5.72705e-07 - 3.20383e-07 - 2.41707e-07 - 3.04402e-07 - 5.4587e-07 - 1.09169e-06 - 2.20877e-06 - 4.41012e-06 - 8.641e-06 - 1.65944e-05 - 3.12269e-05 - 5.75758e-05 - 0.000104013 - 0.000184109 - 0.0003193 - 0.000542574 - 0.00090335 - 0.00147364 - 0.00235538 - 0.00368866 - 0.00565995 - 0.0085093 - 0.0125346 - 0.0180912 - 0.0255834 - 0.0354476 - 0.048123 - 0.0640111 - 0.0834246 - 0.10653 - 0.133286 - 0.163393 - 0.196254 - 0.230963 - 0.266319 - 0.300885 - 0.333069 - 0.361247 - 0.383895 - 0.399721 - 0.407791 - 0.407621 - 0.399219 - 0.383092 - 0.36019 - 0.331816 - 0.299502 - 0.264874 - 0.229517 - 0.194862 - 0.162098 - 0.132119 - 0.105509 - 0.0825559 - 0.0632915 - 0.0475422 - 0.0349905 - 0.0252324 - 0.017828 - 0.012342 - 0.00837148 - 0.00556362 - 0.00362284 - 0.00231142 - 0.00144492 - 0.000885006 - 0.000531112 - 0.000312293 - 0.000179918 - 0.000101561 - 5.61709e-05 - 3.04393e-05 - 1.6162e-05 - 8.408e-06 - 4.28575e-06 - 2.14042e-06 - 1.04738e-06 - 5.0217e-07 - 2.35902e-07 - 1.0858e-07 - 4.89673e-08 - 2.16371e-08 - 9.36772e-09 - 3.97414e-09 - 1.65288e-09 - 6.76192e-10 - 2.78154e-10 - 1.30846e-10 - 1.07446e-10 - 1.85993e-10 - 4.31066e-10 - 1.04002e-09 - 2.48173e-09 - 5.81045e-09 - 1.33319e-08 - 2.99725e-08 - 6.60228e-08 - 1.42496e-07 - 3.01333e-07 - 6.24351e-07 - 1.2675e-06 - 2.52117e-06 - 4.91355e-06 - 9.38262e-06 - 1.75546e-05 - 3.21805e-05 - 5.78006e-05 - 0.000101721 - 0.000175397 - 0.000296328 - 0.000490523 - 0.00079558 - 0.00126428 - 0.00196853 - 0.00300315 - 0.00448898 - 0.00657439 - 0.00943411 - 0.0132642 - 0.0182726 - 0.0246636 - 0.0326174 - 0.0422648 - 0.0536592 - 0.0667494 - 0.0813555 - 0.0971547 - 0.113678 - 0.130325 - 0.146391 - 0.161116 - 0.17374 - 0.183568 - 0.190034 - 0.192754 - 0.191563 - 0.186533 - 0.177966 - 0.166363 - 0.152375 - 0.136743 - 0.120236 - 0.103586 - 0.0874387 - 0.0723175 - 0.058603 - 0.0465301 - 0.036198 - 0.0275912 - 0.0206061 - 0.0150784 - 0.0108107 - 0.00759432 - 0.00522709 - 0.00352508 - 0.00232924 - 0.00150798 - 0.000956565 - 0.000594525 - 0.000362045 - 0.000216019 - 0.000126287 - 7.23372e-05 - 4.05977e-05 - 2.23243e-05 - 1.2028e-05 - 6.34954e-06 - 3.2842e-06 - 1.66439e-06 - 8.26447e-07 - 4.0208e-07 - 1.91667e-07 - 8.95198e-08 - 4.09664e-08 - 1.83685e-08 - 8.06966e-09 - 3.47355e-09 - 1.46497e-09 - 6.05372e-10 - 2.45105e-10 - 9.7234e-11 - 3.77939e-11 - 1.43933e-11 - 5.37079e-12 - 1.9636e-12 - 7.03401e-13 - 2.46882e-13 - 8.49013e-14 - 2.86072e-14 - 9.44439e-15 - 3.05498e-15 - 9.68234e-16 - 3.00669e-16 - 9.14817e-17 - 2.7272e-17 - 7.96591e-18 - 2.27978e-18 - 6.39307e-19 - 1.7579e-19 - 4.79091e-20 - 1.49851e-20 - 1.3075e-20 - 3.86379e-20 - 1.42464e-19 - 5.24334e-19 - 1.89316e-18 - 6.6979e-18 - 2.32182e-17 - 7.886e-17 - 2.62435e-16 - 8.55703e-16 - 2.73377e-15 - 8.55729e-15 - 2.62451e-14 - 7.88674e-14 - 2.32211e-13 - 6.69893e-13 - 1.8935e-12 - 5.24398e-12 - 1.42297e-11 - 3.78324e-11 - 9.85533e-11 - 2.51544e-10 - 6.29065e-10 - 1.54139e-09 - 3.70056e-09 - 8.70479e-09 - 2.00626e-08 - 4.53056e-08 - 1.00243e-07 - 2.17317e-07 - 4.61603e-07 - 9.60688e-07 - 1.95899e-06 - 3.914e-06 - 7.66208e-06 - 1.46964e-05 - 2.76192e-05 - 5.08571e-05 - 9.17551e-05 - 0.000162199 - 0.000280936 - 0.00047677 - 0.000792781 - 0.00129164 - 0.00206196 - 0.0032253 - 0.00494329 - 0.00742376 - 0.0109245 - 0.015753 - 0.0222597 - 0.0308241 - 0.0418311 - 0.0556391 - 0.0725392 - 0.0927117 - 0.116181 - 0.142782 - 0.172134 - 0.203646 - 0.23654 - 0.269907 - 0.302784 - 0.334251 - 0.36353 - 0.390069 - 0.413606 - 0.434188 - 0.452139 - 0.467992 - 0.482371 - 0.495855 - 0.508845 - 0.521448 - 0.533411 - 0.54411 - 0.552604 - 0.557745 - 0.558334 - 0.553292 - 0.541826 - 0.523566 - 0.49865 - 0.46775 - 0.432034 - 0.393086 - 0.35278 - 0.313139 - 0.2762 - 0.243885 - 0.2179 - 0.199659 - 0.19023 - 0.190292 - 0.200109 - 0.219505 - 0.247844 - 0.284017 - 0.326452 - 0.373144 - 0.421726 - 0.469578 - 0.513978 - 0.552279 - 0.582113 - 0.601568 - 0.60936 - 0.604932 - 0.588498 - 0.561014 - 0.524074 - 0.479755 - 0.430422 - 0.378526 - 0.32641 - 0.276152 - 0.229449 - 0.187561 - 0.151296 - 0.121041 - 0.0968234 - 0.0783825 - 0.0652508 - 0.0568241 - 0.052421 - 0.0513267 - 0.0528233 - 0.0562096 - 0.0608144 - 0.0660069 - 0.0712085 - 0.0759058 - 0.0796663 - 0.0821531 - 0.083138 - 0.0825095 - 0.0802724 - 0.0765404 - 0.0715192 - 0.0654835 - 0.0587491 - 0.0516444 - 0.0444831 - 0.0375424 - 0.031047 - 0.0251611 - 0.0199871 - 0.0155706 - 0.0119103 - 0.00897034 - 0.00669384 - 0.00501675 - 0.00387995 - 0.00323985 - 0.00307697 - 0.00340228 - 0.00426134 - 0.00573584 - 0.00794229 - 0.011027 - 0.015157 - 0.0205062 - 0.0272364 - 0.0354751 - 0.0452892 - 0.0566592 - 0.0694565 - 0.0834268 - 0.0981838 - 0.113218 - 0.127916 - 0.141604 - 0.153589 - 0.163224 - 0.169959 - 0.173396 - 0.17333 - 0.169763 - 0.162911 - 0.153177 - 0.141116 - 0.127378 - 0.112654 - 0.0976202 - 0.0828835 - 0.0689498 - 0.0561998 - 0.0448822 - 0.0351196 - 0.0269254 - 0.0202261 - 0.0148867 - 0.0107355 - 0.00758544 - 0.00525143 - 0.00356214 - 0.00236745 - 0.00154166 - 0.000983633 - 0.000614914 - 0.000376646 - 0.000226043 - 0.000132922 - 7.65918e-05 - 4.32595e-05 - 2.39797e-05 - 1.31129e-05 - 7.21922e-06 - 4.30366e-06 - 3.33996e-06 - 4.01773e-06 - 6.68526e-06 - 1.24923e-05 - 2.37754e-05 - 4.47731e-05 - 8.28123e-05 - 0.000150164 - 0.000266831 - 0.000464579 - 0.000792543 - 0.00132472 - 0.00216951 - 0.00348126 - 0.0054733 - 0.00843136 - 0.0127257 - 0.0188194 - 0.0272687 - 0.0387134 - 0.053851 - 0.0733945 - 0.09801 - 0.128237 - 0.164397 - 0.206496 - 0.254135 - 0.306448 - 0.362063 - 0.41913 - 0.47539 - 0.528309 - 0.575259 - 0.613727 - 0.641541 - 0.657068 - 0.659377 - 0.648327 - 0.624588 - 0.589567 - 0.545274 - 0.494136 - 0.438774 - 0.381789 - 0.325571 - 0.272154 - 0.223122 - 0.179578 - 0.142167 - 0.111134 - 0.086417 - 0.0677452 - 0.0547365 - 0.04698 - 0.0440953 - 0.0457651 - 0.0517401 - 0.0618192 - 0.075809 - 0.0934719 - 0.114469 - 0.138309 - 0.164311 - 0.191592 - 0.21908 - 0.245557 - 0.269731 - 0.29033 - 0.306206 - 0.316433 - 0.320401 - 0.317866 - 0.308982 - 0.29428 - 0.274615 - 0.251088 - 0.224938 - 0.197441 - 0.169805 - 0.143086 - 0.118136 - 0.0955664 - 0.0757468 - 0.0588247 - 0.0447603 - 0.0333705 - 0.0243763 - 0.0174466 - 0.0122347 - 0.00840639 - 0.00565932 - 0.00373302 - 0.0024127 - 0.001528 - 0.000948448 - 0.000577417 - 0.000345638 - 0.0002051 - 0.000123843 - 8.18518e-05 - 6.81308e-05 - 7.89389e-05 - 0.00011715 - 0.000192719 - 0.000324255 - 0.000541737 - 0.000890371 - 0.00143553 - 0.00226859 - 0.00351311 - 0.00533082 - 0.00792609 - 0.0115475 - 0.0164852 - 0.0230612 - 0.0316131 - 0.0424685 - 0.0559121 - 0.0721461 - 0.0912487 - 0.113135 - 0.137526 - 0.163937 - 0.191679 - 0.219893 - 0.247604 - 0.273794 - 0.297493 - 0.317866 - 0.334291 - 0.346416 - 0.354179 - 0.357793 - 0.357701 - 0.354498 - 0.348835 - 0.34133 - 0.332489 - 0.322651 - 0.311975 - 0.30045 - 0.287942 - 0.274258 - 0.25922 - 0.242722 - 0.224784 - 0.205571 - 0.185391 - 0.164673 - 0.143916 - 0.123646 - 0.104359 - 0.0864807 - 0.0703312 - 0.0561129 - 0.0439079 - 0.0336895 - 0.0253425 - 0.0186881 - 0.0135096 - 0.00957556 - 0.00665938 - 0.00455333 - 0.00307793 - 0.00208716 - 0.00147068 - 0.00115451 - 0.00110122 - 0.00131073 - 0.0018224 - 0.00271851 - 0.00412898 - 0.00623645 - 0.00928047 - 0.0135588 - 0.0194236 - 0.0272705 - 0.0375177 - 0.0505746 - 0.0667991 - 0.0864465 - 0.109613 - 0.13618 - 0.165769 - 0.19771 - 0.231042 - 0.26454 - 0.296775 - 0.326212 - 0.351326 - 0.370729 - 0.383301 - 0.388294 - 0.385406 - 0.374811 - 0.357144 - 0.333435 - 0.305011 - 0.273374 - 0.240069 - 0.206562 - 0.174142 - 0.143844 - 0.116417 - 0.0923168 - 0.0717266 - 0.054603 - 0.0407277 - 0.0297646 - 0.0213131 - 0.0149531 - 0.010279 - 0.00692323 - 0.00456881 - 0.00295416 - 0.00187155 - 0.00116173 - 0.000706559 - 0.000421043 - 0.000245834 - 0.000140635 - 7.88285e-05 - 4.32922e-05 - 2.32955e-05 - 1.22821e-05 - 6.34466e-06 - 3.2113e-06 - 1.59255e-06 - 7.73819e-07 - 3.68407e-07 - 1.71859e-07 - 7.85727e-08 - 3.525e-08 - 1.56255e-08 - 7.1029e-09 - 3.90724e-09 - 3.74228e-09 - 6.36823e-09 - 1.3367e-08 - 2.89202e-08 - 6.18851e-08 - 1.29968e-07 - 2.67519e-07 - 5.39551e-07 - 1.06623e-06 - 2.06446e-06 - 3.91651e-06 - 7.27996e-06 - 1.32585e-05 - 2.36592e-05 - 4.13656e-05 - 7.08625e-05 - 0.000118941 - 0.000195606 - 0.000315187 - 0.000497615 - 0.000769759 - 0.00116668 - 0.00173256 - 0.00252093 - 0.00359392 - 0.00502012 - 0.00687063 - 0.00921332 - 0.0121052 - 0.0155835 - 0.019656 - 0.0242919 - 0.0294147 - 0.0348984 - 0.0405679 - 0.0462058 - 0.0515641 - 0.0563813 - 0.0604032 - 0.0634047 - 0.0652108 - 0.0657135 - 0.0648824 - 0.0627676 - 0.0594951 - 0.055254 - 0.0502786 - 0.044827 - 0.0391591 - 0.0335168 - 0.028108 - 0.0230958 - 0.018594 - 0.0146673 - 0.0113361 - 0.00858452 - 0.00636947 - 0.0046305 - 0.0032983 - 0.00230191 - 0.00157407 - 0.00105462 - 0.000692316 - 0.000445298 - 0.000280629 - 0.000173282 - 0.000104836 - 6.21444e-05 - 3.60937e-05 - 2.05399e-05 - 1.14526e-05 - 6.25667e-06 - 3.34904e-06 - 1.75645e-06 - 9.02581e-07 - 4.5444e-07 - 2.24189e-07 - 1.08378e-07 - 5.13693e-08 - 2.39435e-08 - 1.11485e-08 - 5.59702e-09 - 3.92679e-09 - 5.09955e-09 - 9.86095e-09 - 2.114e-08 - 4.55587e-08 - 9.66698e-08 - 2.0116e-07 - 4.10205e-07 - 8.19618e-07 - 1.60458e-06 - 3.07786e-06 - 5.78459e-06 - 1.06521e-05 - 1.9219e-05 - 3.39755e-05 - 5.88488e-05 - 9.98725e-05 - 0.00016607 - 0.000270566 - 0.000431909 - 0.000675535 - 0.00103524 - 0.00155443 - 0.00228684 - 0.0032964 - 0.00465564 - 0.00644253 - 0.00873514 - 0.0116043 - 0.0151046 - 0.0192634 - 0.024071 - 0.0294708 - 0.035353 - 0.0415526 - 0.0478528 - 0.053995 - 0.059695 - 0.0646638 - 0.0686316 - 0.0713724 - 0.0727252 - 0.0726101 - 0.0710369 - 0.068105 - 0.0639941 - 0.058949 - 0.0532582 - 0.0472318 - 0.0411794 - 0.035392 - 0.0301294 - 0.0256128 - 0.0220232 - 0.0195037 - 0.0181646 - 0.0180873 - 0.0193273 - 0.0219132 - 0.025842 - 0.0310708 - 0.0375062 - 0.0449944 - 0.0533132 - 0.0621703 - 0.0712086 - 0.0800204 - 0.0881704 - 0.0952259 - 0.10079 - 0.104537 - 0.106239 - 0.105792 - 0.10322 - 0.0986778 - 0.0924299 - 0.0848289 - 0.0762804 - 0.0672077 - 0.058018 - 0.049073 - 0.0406687 - 0.0330228 - 0.0262727 - 0.0204801 - 0.0156422 - 0.0117058 - 0.00858298 - 0.00616616 - 0.00434041 - 0.00299359 - 0.00202311 - 0.00133989 - 0.000870016 - 0.000554601 - 0.00034854 - 0.000218741 - 0.000142232 - 0.000104541 - 9.8566e-05 - 0.000124155 - 0.000188521 - 0.000307637 - 0.000508725 - 0.000833915 - 0.00134509 - 0.00212979 - 0.00330782 - 0.00503791 - 0.00752352 - 0.0110162 - 0.0158148 - 0.0222592 - 0.0307153 - 0.0415515 - 0.0551058 - 0.071643 - 0.0913074 - 0.114073 - 0.1397 - 0.167701 - 0.197328 - 0.227588 - 0.257282 - 0.285075 - 0.309593 - 0.329531 - 0.34377 - 0.35148 - 0.352196 - 0.345872 - 0.332879 - 0.313973 - 0.29022 - 0.262897 - 0.233379 - 0.203026 - 0.173082 - 0.144596 - 0.118376 - 0.0949657 - 0.0746562 - 0.057512 - 0.0434158 - 0.0321176 - 0.0232853 - 0.0165482 - 0.0115343 - 0.00789618 - 0.00532826 - 0.00357577 - 0.00243735 - 0.00176332 - 0.00145159 - 0.00144239 - 0.00171284 - 0.00227175 - 0.00315462 - 0.00441846 - 0.0061359 - 0.00838817 - 0.0112563 - 0.0148106 - 0.019099 - 0.0241341 - 0.0298819 - 0.0362517 - 0.0430913 - 0.0501867 - 0.0572696 - 0.064032 - 0.0701466 - 0.0752928 - 0.0791839 - 0.0815939 - 0.0823787 - 0.0814908 - 0.0789841 - 0.075008 - 0.069793 - 0.0636288 - 0.0568371 - 0.0497447 - 0.0426579 - 0.0358416 - 0.0295062 - 0.0238 - 0.0188094 - 0.0145651 - 0.0110507 - 0.00821493 - 0.00598368 - 0.00427072 - 0.00298721 - 0.00204851 - 0.00137888 - 0.000914107 - 0.0006025 - 0.000404956 - 0.000294455 - 0.000255581 - 0.000284413 - 0.000389119 - 0.00059143 - 0.000929092 - 0.0014592 - 0.00226219 - 0.00344601 - 0.00514967 - 0.00754534 - 0.0108375 - 0.0152582 - 0.0210566 - 0.0284828 - 0.0377642 - 0.0490773 - 0.0625148 - 0.0780522 - 0.0955182 - 0.114573 - 0.134702 - 0.155224 - 0.175321 - 0.194088 - 0.210595 - 0.223966 - 0.233452 - 0.238502 - 0.238817 - 0.234375 - 0.225439 - 0.212528 - 0.196368 - 0.177823 - 0.157822 - 0.13728 - 0.117031 - 0.0977797 - 0.0800658 - 0.0642531 - 0.0505343 - 0.0389512 - 0.0294236 - 0.0217826 - 0.0158036 - 0.0112366 - 0.00782971 - 0.00534666 - 0.00357803 - 0.00234654 - 0.00150811 - 0.000949846 - 0.000586259 - 0.0003546 - 0.000210184 - 0.000122087 - 6.94932e-05 - 3.87633e-05 - 2.11886e-05 - 1.13497e-05 - 5.95747e-06 - 3.06434e-06 - 1.54456e-06 - 7.62898e-07 - 3.69248e-07 - 1.75129e-07 - 8.13929e-08 - 3.7068e-08 - 1.65423e-08 - 7.23393e-09 - 3.0998e-09 - 1.30158e-09 - 5.35535e-10 - 2.15914e-10 - 8.52996e-11 - 3.30208e-11 - 1.25257e-11 - 4.65589e-12 - 1.69628e-12 - 6.07133e-13 - 2.17868e-13 - 9.17968e-14 - 8.26531e-14 - 1.80758e-13 - 5.04118e-13 - 1.4277e-12 - 3.9783e-12 - 1.0868e-11 - 2.90954e-11 - 7.63304e-11 - 1.96231e-10 - 4.94351e-10 - 1.2204e-09 - 2.95232e-09 - 6.9988e-09 - 1.62586e-08 - 3.70116e-08 - 8.25642e-08 - 1.80486e-07 - 3.86628e-07 - 8.11596e-07 - 1.66949e-06 - 3.36533e-06 - 6.64764e-06 - 1.28678e-05 - 2.44085e-05 - 4.53704e-05 - 8.26422e-05 - 0.000147512 - 0.000258019 - 0.000442253 - 0.000742825 - 0.00122264 - 0.001972 - 0.00311681 - 0.00482735 - 0.00732661 - 0.0108966 - 0.0158809 - 0.0226805 - 0.0317412 - 0.04353 - 0.0584987 - 0.0770365 - 0.0994122 - 0.125711 - 0.155777 - 0.189156 - 0.225077 - 0.262443 - 0.299868 - 0.335752 - 0.368386 - 0.396083 - 0.417326 - 0.430901 - 0.436023 - 0.432407 - 0.420307 - 0.400488 - 0.374157 - 0.34285 - 0.308297 - 0.27227 - 0.236449 - 0.202305 - 0.171016 - 0.143419 - 0.12 - 0.10091 - 0.0860138 - 0.0749435 - 0.0671671 - 0.0620535 - 0.0589346 - 0.0571569 - 0.0561247 - 0.0553308 - 0.0543762 - 0.052979 - 0.0509737 - 0.0483016 - 0.0449951 - 0.0411568 - 0.0369364 - 0.0325078 - 0.0280481 - 0.0237199 - 0.0196589 - 0.0159663 - 0.0127065 - 0.00990863 - 0.00757099 - 0.00566812 - 0.00415785 - 0.00298841 - 0.00210451 - 0.00145211 - 0.000981719 - 0.000650297 - 0.00042206 - 0.000268394 - 0.000167228 - 0.00010209 - 6.1065e-05 - 3.57882e-05 - 2.05505e-05 - 1.15623e-05 - 6.37384e-06 - 3.44267e-06 - 1.82192e-06 - 9.44718e-07 - 4.79996e-07 - 2.39019e-07 - 1.16785e-07 - 5.63189e-08 - 2.75876e-08 - 1.55066e-08 - 1.35065e-08 - 2.06346e-08 - 4.12348e-08 - 8.71701e-08 - 1.83491e-07 - 3.79656e-07 - 7.70155e-07 - 1.53093e-06 - 2.98182e-06 - 5.69044e-06 - 1.06401e-05 - 1.94933e-05 - 3.49914e-05 - 6.15423e-05 - 0.000106053 - 0.000179064 - 0.000296231 - 0.000480165 - 0.000762583 - 0.00118664 - 0.00180922 - 0.0027027 - 0.00395586 - 0.00567312 - 0.00797149 - 0.0109747 - 0.0148042 - 0.0195665 - 0.0253384 - 0.03215 - 0.0399688 - 0.0486854 - 0.0581051 - 0.0679468 - 0.0778511 - 0.0873986 - 0.0961378 - 0.10362 - 0.109438 - 0.113265 - 0.11489 - 0.114237 - 0.111386 - 0.106565 - 0.100144 - 0.092608 - 0.0845337 - 0.0765571 - 0.069347 - 0.0635806 - 0.0599254 - 0.0590243 - 0.0614823 - 0.0678499 - 0.0786025 - 0.0941094 - 0.114595 - 0.140095 - 0.170403 - 0.205033 - 0.243187 - 0.28375 - 0.325307 - 0.366205 - 0.404639 - 0.438768 - 0.466855 - 0.487399 - 0.499265 - 0.501787 - 0.494819 - 0.478757 - 0.454493 - 0.423343 - 0.386921 - 0.347013 - 0.305437 - 0.263913 - 0.223964 - 0.186845 - 0.15351 - 0.124609 - 0.100516 - 0.0813752 - 0.0671528 - 0.0576947 - 0.0527714 - 0.0521107 - 0.0554152 - 0.0623635 - 0.0726003 - 0.0857196 - 0.101248 - 0.118632 - 0.137242 - 0.156382 - 0.175324 - 0.193347 - 0.209787 - 0.224084 - 0.235824 - 0.244759 - 0.250819 - 0.25409 - 0.254781 - 0.253179 - 0.249598 - 0.24433 - 0.237611 - 0.229608 - 0.220414 - 0.210075 - 0.198611 - 0.186058 - 0.172493 - 0.15806 - 0.142974 - 0.127518 - 0.112024 - 0.0968447 - 0.0823232 - 0.0687639 - 0.056409 - 0.0454241 - 0.0358933 - 0.0278226 - 0.0211511 - 0.0157665 - 0.0115222 - 0.00825423 - 0.0057959 - 0.0039887 - 0.00269018 - 0.00177808 - 0.00115167 - 0.000731004 - 0.000454751 - 0.000277398 - 0.000166222 - 9.846e-05 - 5.88987e-05 - 3.79727e-05 - 3.04441e-05 - 3.46788e-05 - 5.25582e-05 - 9.00977e-05 - 0.000158898 - 0.000278618 - 0.000480699 - 0.000813605 - 0.00134977 - 0.00219438 - 0.00349573 - 0.00545668 - 0.00834609 - 0.0125084 - 0.0183688 - 0.0264315 - 0.0372673 - 0.0514867 - 0.0696987 - 0.0924523 - 0.120163 - 0.153035 - 0.190972 - 0.233514 - 0.27978 - 0.328461 - 0.377845 - 0.425897 - 0.470391 - 0.509067 - 0.539825 - 0.56091 - 0.571077 - 0.569717 - 0.55691 - 0.533424 - 0.500637 - 0.460399 - 0.414867 - 0.366308 - 0.31692 - 0.268674 - 0.223194 - 0.181698 - 0.144974 - 0.113407 - 0.0870373 - 0.0656418 - 0.048819 - 0.0360773 - 0.0269108 - 0.0208626 - 0.0175701 - 0.0167947 - 0.0184339 - 0.0225191 - 0.0291988 - 0.0387082 - 0.0513264 - 0.067323 - 0.0868961 - 0.110107 - 0.136818 - 0.166639 - 0.198893 - 0.232612 - 0.266561 - 0.299299 - 0.329271 - 0.354927 - 0.374855 - 0.387903 - 0.393297 - 0.390711 - 0.3803 - 0.36269 - 0.338907 - 0.310288 - 0.278347 - 0.244652 - 0.210696 - 0.177796 - 0.147015 - 0.119132 - 0.0946281 - 0.0737165 - 0.0563825 - 0.0424423 - 0.0316028 - 0.0235195 - 0.0178455 - 0.0142692 - 0.0125399 - 0.0124803 - 0.0139879 - 0.0170253 - 0.0216027 - 0.0277537 - 0.0355053 - 0.0448476 - 0.0557031 - 0.0679011 - 0.0811603 - 0.0950824 - 0.10916 - 0.1228 - 0.135357 - 0.146187 - 0.154696 - 0.160393 - 0.162941 - 0.162186 - 0.158174 - 0.151147 - 0.141516 - 0.129827 - 0.116706 - 0.102807 - 0.0887582 - 0.0751238 - 0.062369 - 0.0508463 - 0.0407927 - 0.0323401 - 0.0255341 - 0.0203573 - 0.0167544 - 0.0146531 - 0.0139808 - 0.0146745 - 0.0166817 - 0.019956 - 0.0244455 - 0.0300777 - 0.0367437 - 0.0442835 - 0.0524759 - 0.0610361 - 0.0696209 - 0.0778441 - 0.0852996 - 0.0915913 - 0.096366 - 0.0993442 - 0.100347 - 0.0993131 - 0.0963045 - 0.0915008 - 0.0851805 - 0.0776949 - 0.0694355 - 0.0608006 - 0.052164 - 0.0438501 - 0.0361166 - 0.0291461 - 0.0230458 - 0.0178541 - 0.0135526 - 0.0100796 - 0.00734517 - 0.00524441 - 0.00366883 - 0.00251476 - 0.00168889 - 0.00111133 - 0.00071651 - 0.000452625 - 0.00028015 - 0.000169895 - 0.00010095 - 5.87719e-05 - 3.35251e-05 - 1.87373e-05 - 1.02608e-05 - 5.50542e-06 - 2.89426e-06 - 1.49081e-06 - 7.52391e-07 - 3.7205e-07 - 1.80259e-07 - 8.55712e-08 - 3.98012e-08 - 1.81385e-08 - 8.09923e-09 - 3.54342e-09 - 1.51893e-09 - 6.37957e-10 - 2.62531e-10 - 1.05854e-10 - 4.18187e-11 - 1.61872e-11 - 6.13914e-12 - 2.28129e-12 - 8.306e-13 - 2.96305e-13 - 1.03568e-13 - 3.54687e-14 - 1.19015e-14 - 3.91289e-15 - 1.26046e-15 - 3.9783e-16 - 1.23028e-16 - 3.72774e-17 - 1.10669e-17 - 3.21915e-18 - 9.17473e-19 - 2.56202e-19 - 7.00986e-20 - 1.8792e-20 - 4.93598e-21 - 1.27031e-21 - 3.2032e-22 - 7.91398e-23 - 1.91577e-23 - 4.54388e-24 - 1.05596e-24 - 2.4044e-25 - 5.36415e-26 - 1.17255e-26 - 2.51132e-27 - 5.26996e-28 - 1.08355e-28 - 2.18287e-29 - 4.30869e-30 - 8.33293e-31 - 1.57902e-31 - 2.93166e-32 - 5.33307e-33 - 9.50556e-34 - 1.66003e-34 - 2.84046e-35 - 4.76211e-36 - 7.82253e-37 - 1.25902e-37 - 1.98542e-38 - 3.06769e-39 - 4.64415e-40 - 6.88872e-41 - 1.00117e-41 - 1.42565e-42 - 1.98909e-43 - 2.71916e-44 - 3.64209e-45 - 4.77973e-46 - 6.14602e-47 - 7.7432e-48 - 9.55837e-49 - 1.15607e-49 - 1.37e-50 - 1.59073e-51 - 1.8097e-52 - 2.01722e-53 - 2.20312e-54 - 2.35755e-55 - 2.47183e-56 - 2.5393e-57 - 2.55591e-58 - 2.52066e-59 - 2.43568e-60 - 2.30602e-61 - 2.13916e-62 - 1.94428e-63 - 1.73146e-64 - 1.51078e-65 - 1.2916e-66 - 1.08191e-67 - 8.87958e-69 - 7.14052e-70 - 5.62606e-71 - 4.34325e-72 - 3.28521e-73 - 2.43471e-74 - 1.76795e-75 - 1.25785e-76 - 8.76845e-78 - 5.98901e-79 - 4.00797e-80 - 2.62803e-81 - 1.68839e-82 - 1.0628e-83 - 6.55493e-85 - 3.96115e-86 - 2.34537e-87 - 1.36063e-88 - 7.73397e-90 - 4.30728e-91 - 2.35039e-92 - 1.25665e-93 - 6.58303e-95 - 3.37889e-96 - 1.69925e-97 - 8.37298e-99 - 4.04239e-100 - 1.9122e-101 - 8.8627e-103 - 4.02472e-104 - 1.79078e-105 - 7.807e-107 - 3.33476e-108 - 1.39764e-109 - 1.06571e-110 - 1.20963e-109 - 2.89465e-108 - 6.79983e-107 - 1.56509e-105 - 3.52952e-104 - 7.79884e-103 - 1.68842e-101 - 3.58153e-100 - 7.44378e-99 - 1.51585e-97 - 3.0245e-96 - 5.91274e-95 - 1.13256e-93 - 2.12554e-92 - 3.90855e-91 - 7.04204e-90 - 1.24313e-88 - 2.15018e-87 - 3.64391e-86 - 6.05059e-85 - 9.84384e-84 - 1.56916e-82 - 2.45081e-81 - 3.75048e-80 - 5.62342e-79 - 8.26136e-78 - 1.18916e-76 - 1.67712e-75 - 2.31753e-74 - 3.1378e-73 - 4.16256e-72 - 5.41044e-71 - 6.89035e-70 - 8.5978e-69 - 1.05116e-67 - 1.25919e-66 - 1.4779e-65 - 1.69957e-64 - 1.915e-63 - 2.11415e-62 - 2.28686e-61 - 2.42371e-60 - 2.51686e-59 - 2.56079e-58 - 2.55284e-57 - 2.49352e-56 - 2.38637e-55 - 2.23769e-54 - 2.05588e-53 - 1.85069e-52 - 1.63232e-51 - 1.41064e-50 - 1.19443e-49 - 9.90932e-49 - 8.05496e-48 - 6.41535e-47 - 5.00626e-46 - 3.82775e-45 - 2.86755e-44 - 2.10482e-43 - 1.51376e-42 - 1.06668e-41 - 7.36458e-41 - 4.98195e-40 - 3.30208e-39 - 2.14443e-38 - 1.3645e-37 - 8.50695e-37 - 5.19648e-36 - 3.11015e-35 - 1.82386e-34 - 1.04794e-33 - 5.89958e-33 - 3.25418e-32 - 1.75873e-31 - 9.31305e-31 - 4.83195e-30 - 2.45635e-29 - 1.22347e-28 - 5.97085e-28 - 2.85506e-27 - 1.33761e-26 - 6.1402e-26 - 2.76167e-25 - 1.21702e-24 - 5.25487e-24 - 2.22312e-23 - 9.21509e-23 - 3.74261e-22 - 1.48931e-21 - 5.80677e-21 - 2.2183e-20 - 8.30318e-20 - 3.04513e-19 - 1.09422e-18 - 3.85247e-18 - 1.32896e-17 - 4.49183e-17 - 1.48755e-16 - 4.82679e-16 - 1.53456e-15 - 4.7802e-15 - 1.45897e-14 - 4.36302e-14 - 1.2784e-13 - 3.67016e-13 - 1.03239e-12 - 2.84538e-12 - 7.68384e-12 - 2.03309e-11 - 5.27081e-11 - 1.33887e-10 - 3.33231e-10 - 8.12632e-10 - 1.94172e-09 - 4.54597e-09 - 1.04283e-08 - 2.34394e-08 - 5.16215e-08 - 1.11395e-07 - 2.35536e-07 - 4.87981e-07 - 9.90622e-07 - 1.97049e-06 - 3.84067e-06 - 7.33515e-06 - 1.37273e-05 - 2.51733e-05 - 4.52355e-05 - 7.96547e-05 - 0.000137449 - 0.000232424 - 0.000385157 - 0.000625493 - 0.000995518 - 0.00155285 - 0.00237402 - 0.00355735 - 0.00522491 - 0.00752254 - 0.0106172 - 0.0146907 - 0.0199294 - 0.0265096 - 0.0345787 - 0.0442341 - 0.0555004 - 0.0683094 - 0.0824842 - 0.0977307 - 0.11364 - 0.129703 - 0.145333 - 0.159905 - 0.172795 - 0.183427 - 0.191318 - 0.19611 - 0.197601 - 0.195756 - 0.190705 - 0.18273 - 0.172236 - 0.15972 - 0.145736 - 0.130848 - 0.115605 - 0.100507 - 0.0859807 - 0.0723703 - 0.0599273 - 0.0488129 - 0.0391035 - - - - - 4.94677e-30 - 6.15613e-30 - 7.50639e-30 - 8.96791e-30 - 1.04976e-29 - 1.20403e-29 - 1.35322e-29 - 1.49099e-29 - 1.61406e-29 - 1.7354e-29 - 1.94873e-29 - 2.74381e-29 - 6.52784e-29 - 2.48192e-28 - 1.11559e-27 - 5.14139e-27 - 2.34378e-26 - 1.04887e-25 - 4.60081e-25 - 1.9775e-24 - 8.32801e-24 - 3.43641e-23 - 1.38933e-22 - 5.50355e-22 - 2.13608e-21 - 8.12322e-21 - 3.02675e-20 - 1.105e-19 - 3.9526e-19 - 1.38529e-18 - 4.75704e-18 - 1.60055e-17 - 5.27641e-17 - 1.70429e-16 - 5.3937e-16 - 1.6725e-15 - 5.0814e-15 - 1.51264e-14 - 4.41191e-14 - 1.26082e-13 - 3.53035e-13 - 9.68542e-13 - 2.60349e-12 - 6.85694e-12 - 1.76946e-11 - 4.47394e-11 - 1.10834e-10 - 2.69027e-10 - 6.39816e-10 - 1.49091e-09 - 3.40395e-09 - 7.61469e-09 - 1.66901e-08 - 3.58427e-08 - 7.5419e-08 - 1.55488e-07 - 3.14088e-07 - 6.21643e-07 - 1.2055e-06 - 2.29051e-06 - 4.26416e-06 - 7.77805e-06 - 1.3901e-05 - 2.4342e-05 - 4.17643e-05 - 7.02086e-05 - 0.000115641 - 0.000186625 - 0.000295098 - 0.000457192 - 0.000694014 - 0.00103222 - 0.00150424 - 0.00214782 - 0.00300479 - 0.00411877 - 0.00553169 - 0.00727922 - 0.00938532 - 0.0118563 - 0.0146753 - 0.0177977 - 0.0211483 - 0.0246221 - 0.0280875 - 0.0313934 - 0.0343799 - 0.0368904 - 0.0387854 - 0.0399557 - 0.0403326 - 0.0398957 - 0.0386751 - 0.036749 - 0.0342377 - 0.0312926 - 0.0280847 - 0.0247919 - 0.0215871 - 0.0186297 - 0.0160581 - 0.013987 - 0.0125051 - 0.0116762 - 0.0115397 - 0.0121108 - 0.0133811 - 0.0153167 - 0.0178562 - 0.0209094 - 0.0243561 - 0.0280473 - 0.0318096 - 0.0354525 - 0.0387797 - 0.041602 - 0.0437517 - 0.0450968 - 0.045552 - 0.0450867 - 0.043727 - 0.041553 - 0.03869 - 0.0352968 - 0.031551 - 0.027633 - 0.0237127 - 0.0199375 - 0.0164247 - 0.0132575 - 0.0104848 - 0.00812456 - 0.00616843 - 0.00458866 - 0.00334453 - 0.00238847 - 0.00167125 - 0.00114578 - 0.000769656 - 0.000506558 - 0.000326662 - 0.000206398 - 0.000127776 - 7.75048e-05 - 4.60623e-05 - 2.68225e-05 - 1.53035e-05 - 8.55494e-06 - 4.68577e-06 - 2.51468e-06 - 1.32227e-06 - 6.81232e-07 - 3.4388e-07 - 1.70081e-07 - 8.24215e-08 - 3.91348e-08 - 1.82063e-08 - 8.29885e-09 - 3.70639e-09 - 1.62189e-09 - 6.95389e-10 - 2.92132e-10 - 1.2026e-10 - 4.85509e-11 - 1.93322e-11 - 7.90272e-12 - 4.15729e-12 - 4.69603e-12 - 1.00995e-11 - 2.54898e-11 - 6.48543e-11 - 1.62322e-10 - 3.9828e-10 - 9.57567e-10 - 2.25575e-09 - 5.20657e-09 - 1.17747e-08 - 2.60905e-08 - 5.66438e-08 - 1.20492e-07 - 2.51133e-07 - 5.12843e-07 - 1.02613e-06 - 2.01167e-06 - 3.8641e-06 - 7.27238e-06 - 1.34104e-05 - 2.42295e-05 - 4.28927e-05 - 7.43975e-05 - 0.000126436 - 0.000210533 - 0.000343484 - 0.000549073 - 0.000859983 - 0.00131974 - 0.00198436 - 0.00292342 - 0.00421986 - 0.00596819 - 0.00827035 - 0.011229 - 0.0149381 - 0.019471 - 0.0248666 - 0.0311158 - 0.0381491 - 0.0458271 - 0.0539385 - 0.062203 - 0.0702847 - 0.0778121 - 0.0844055 - 0.0897079 - 0.0934174 - 0.0953152 - 0.095287 - 0.0933346 - 0.0895758 - 0.0842323 - 0.0776088 - 0.0700637 - 0.0619781 - 0.0537245 - 0.0456401 - 0.0380067 - 0.0310392 - 0.0248817 - 0.0196118 - 0.0152499 - 0.0117717 - 0.00912236 - 0.00722982 - 0.00601571 - 0.00540334 - 0.0053222 - 0.00570923 - 0.00650744 - 0.00766253 - 0.00911876 - 0.0108148 - 0.0126804 - 0.0146355 - 0.0165902 - 0.0184485 - 0.020113 - 0.0214914 - 0.022504 - 0.0230902 - 0.0232141 - 0.022868 - 0.0220728 - 0.020876 - 0.0193474 - 0.0175725 - 0.0156453 - 0.0136615 - 0.0117119 - 0.00987887 - 0.00823454 - 0.00684165 - 0.00575724 - 0.00503837 - 0.00474935 - 0.00496969 - 0.00580156 - 0.00737575 - 0.00985451 - 0.0134302 - 0.0183181 - 0.0247429 - 0.0329181 - 0.0430191 - 0.0551515 - 0.0693182 - 0.0853881 - 0.103073 - 0.121914 - 0.141292 - 0.160443 - 0.178512 - 0.194604 - 0.207861 - 0.217536 - 0.223062 - 0.224109 - 0.220611 - 0.212781 - 0.201084 - 0.18619 - 0.168917 - 0.15015 - 0.130773 - 0.111595 - 0.0933057 - 0.0764379 - 0.0613545 - 0.0482527 - 0.037182 - 0.0280725 - 0.0207666 - 0.0150518 - 0.0106892 - 0.00743777 - 0.00507078 - 0.00338723 - 0.00221692 - 0.00142165 - 0.000893251 - 0.000549908 - 0.000331699 - 0.000196035 - 0.000113517 - 6.44059e-05 - 3.58036e-05 - 1.95013e-05 - 1.04073e-05 - 5.4419e-06 - 2.78804e-06 - 1.39954e-06 - 6.88345e-07 - 3.31715e-07 - 1.56625e-07 - 7.24592e-08 - 3.28445e-08 - 1.45871e-08 - 6.34764e-09 - 2.7064e-09 - 1.1306e-09 - 4.62768e-10 - 1.8559e-10 - 7.29258e-11 - 2.80766e-11 - 1.05912e-11 - 3.91457e-12 - 1.41762e-12 - 5.03003e-13 - 1.74872e-13 - 5.95668e-14 - 1.98805e-14 - 6.50108e-15 - 2.08296e-15 - 6.53904e-16 - 2.01133e-16 - 6.06164e-17 - 1.78992e-17 - 5.17861e-18 - 1.46801e-18 - 4.07741e-19 - 1.10962e-19 - 2.95871e-20 - 7.72981e-21 - 1.97881e-21 - 4.96994e-22 - 1.2513e-22 - 4.27973e-23 - 6.25412e-23 - 2.27169e-22 - 8.96772e-22 - 3.49045e-21 - 1.3316e-20 - 4.97752e-20 - 1.82301e-19 - 6.54186e-19 - 2.30012e-18 - 7.92386e-18 - 2.67461e-17 - 8.84545e-17 - 2.86627e-16 - 9.1002e-16 - 2.83088e-15 - 8.62837e-15 - 2.57675e-14 - 7.5397e-14 - 2.16158e-13 - 6.07193e-13 - 1.67116e-12 - 4.50658e-12 - 1.19073e-11 - 3.08258e-11 - 7.81904e-11 - 1.94325e-10 - 4.73197e-10 - 1.12899e-09 - 2.63923e-09 - 6.04507e-09 - 1.35663e-08 - 2.98304e-08 - 6.42676e-08 - 1.35663e-07 - 2.80589e-07 - 5.6861e-07 - 1.12901e-06 - 2.19641e-06 - 4.18667e-06 - 7.81917e-06 - 1.43083e-05 - 2.5654e-05 - 4.50668e-05 - 7.75704e-05 - 0.000130819 - 0.000216165 - 0.000349973 - 0.000555164 - 0.000862869 - 0.00131403 - 0.00196066 - 0.00286641 - 0.00410592 - 0.00576262 - 0.00792443 - 0.0106771 - 0.0140955 - 0.0182326 - 0.0231079 - 0.0286958 - 0.0349164 - 0.0416297 - 0.0486353 - 0.0556796 - 0.0624697 - 0.0686941 - 0.0740496 - 0.07827 - 0.0811551 - 0.0825952 - 0.0825891 - 0.0812525 - 0.0788147 - 0.0756052 - 0.0720296 - 0.068538 - 0.0655882 - 0.0636076 - 0.0629552 - 0.0638897 - 0.0665429 - 0.0709037 - 0.0768119 - 0.0839642 - 0.0919334 - 0.100197 - 0.108179 - 0.115294 - 0.120995 - 0.124823 - 0.126445 - 0.125678 - 0.122506 - 0.117071 - 0.10966 - 0.100669 - 0.0905633 - 0.0798348 - 0.0689607 - 0.0583676 - 0.0484063 - 0.0393372 - 0.0313261 - 0.0244505 - 0.0187122 - 0.0140553 - 0.0103846 - 0.00758494 - 0.00553752 - 0.00413347 - 0.00328399 - 0.00292721 - 0.00303184 - 0.00359822 - 0.00465668 - 0.00626355 - 0.00849453 - 0.0114355 - 0.0151707 - 0.019769 - 0.0252679 - 0.0316588 - 0.0388727 - 0.0467703 - 0.0551377 - 0.0636899 - 0.0720829 - 0.0799342 - 0.08685 - 0.092458 - 0.0964398 - 0.0985609 - 0.0986938 - 0.0968305 - 0.0930832 - 0.0876733 - 0.0809096 - 0.0731593 - 0.0648151 - 0.0562625 - 0.0478519 - 0.0398764 - 0.0325589 - 0.0260472 - 0.0204168 - 0.0156802 - 0.0117992 - 0.00869946 - 0.00628445 - 0.00444814 - 0.0030848 - 0.0020961 - 0.00139552 - 0.00091032 - 0.000581822 - 0.000364354 - 0.000223559 - 0.0001344 - 7.91667e-05 - 4.56902e-05 - 2.5837e-05 - 1.43154e-05 - 7.77184e-06 - 4.13516e-06 - 2.15827e-06 - 1.10966e-06 - 5.72705e-07 - 3.20383e-07 - 2.41707e-07 - 3.04402e-07 - 5.4587e-07 - 1.09169e-06 - 2.20877e-06 - 4.41012e-06 - 8.641e-06 - 1.65944e-05 - 3.12269e-05 - 5.75758e-05 - 0.000104013 - 0.000184109 - 0.0003193 - 0.000542574 - 0.00090335 - 0.00147364 - 0.00235538 - 0.00368866 - 0.00565995 - 0.0085093 - 0.0125346 - 0.0180912 - 0.0255834 - 0.0354476 - 0.048123 - 0.0640111 - 0.0834246 - 0.10653 - 0.133286 - 0.163393 - 0.196254 - 0.230963 - 0.266319 - 0.300885 - 0.333069 - 0.361247 - 0.383895 - 0.399721 - 0.407791 - 0.407621 - 0.399219 - 0.383092 - 0.36019 - 0.331816 - 0.299502 - 0.264874 - 0.229517 - 0.194862 - 0.162098 - 0.132119 - 0.105509 - 0.0825559 - 0.0632915 - 0.0475422 - 0.0349905 - 0.0252324 - 0.017828 - 0.012342 - 0.00837148 - 0.00556362 - 0.00362284 - 0.00231142 - 0.00144492 - 0.000885006 - 0.000531112 - 0.000312293 - 0.000179918 - 0.000101561 - 5.61709e-05 - 3.04393e-05 - 1.6162e-05 - 8.408e-06 - 4.28575e-06 - 2.14042e-06 - 1.04738e-06 - 5.0217e-07 - 2.35902e-07 - 1.0858e-07 - 4.89673e-08 - 2.16371e-08 - 9.36772e-09 - 3.97414e-09 - 1.65288e-09 - 6.76192e-10 - 2.78154e-10 - 1.30846e-10 - 1.07446e-10 - 1.85993e-10 - 4.31066e-10 - 1.04002e-09 - 2.48173e-09 - 5.81045e-09 - 1.33319e-08 - 2.99725e-08 - 6.60228e-08 - 1.42496e-07 - 3.01333e-07 - 6.24351e-07 - 1.2675e-06 - 2.52117e-06 - 4.91355e-06 - 9.38262e-06 - 1.75546e-05 - 3.21805e-05 - 5.78006e-05 - 0.000101721 - 0.000175397 - 0.000296328 - 0.000490523 - 0.00079558 - 0.00126428 - 0.00196853 - 0.00300315 - 0.00448898 - 0.00657439 - 0.00943411 - 0.0132642 - 0.0182726 - 0.0246636 - 0.0326174 - 0.0422648 - 0.0536592 - 0.0667494 - 0.0813555 - 0.0971547 - 0.113678 - 0.130325 - 0.146391 - 0.161116 - 0.17374 - 0.183568 - 0.190034 - 0.192754 - 0.191563 - 0.186533 - 0.177966 - 0.166363 - 0.152375 - 0.136743 - 0.120236 - 0.103586 - 0.0874387 - 0.0723175 - 0.058603 - 0.0465301 - 0.036198 - 0.0275912 - 0.0206061 - 0.0150784 - 0.0108107 - 0.00759432 - 0.00522709 - 0.00352508 - 0.00232924 - 0.00150798 - 0.000956565 - 0.000594525 - 0.000362045 - 0.000216019 - 0.000126287 - 7.23372e-05 - 4.05977e-05 - 2.23243e-05 - 1.2028e-05 - 6.34954e-06 - 3.2842e-06 - 1.66439e-06 - 8.26447e-07 - 4.0208e-07 - 1.91667e-07 - 8.95198e-08 - 4.09664e-08 - 1.83685e-08 - 8.06966e-09 - 3.47355e-09 - 1.46497e-09 - 6.05372e-10 - 2.45105e-10 - 9.7234e-11 - 3.77939e-11 - 1.43933e-11 - 5.37079e-12 - 1.9636e-12 - 7.03401e-13 - 2.46882e-13 - 8.49013e-14 - 2.86072e-14 - 9.44439e-15 - 3.05498e-15 - 9.68234e-16 - 3.00669e-16 - 9.14817e-17 - 2.7272e-17 - 7.96591e-18 - 2.27978e-18 - 6.39307e-19 - 1.7579e-19 - 4.79091e-20 - 1.49851e-20 - 1.3075e-20 - 3.86379e-20 - 1.42464e-19 - 5.24334e-19 - 1.89316e-18 - 6.6979e-18 - 2.32182e-17 - 7.886e-17 - 2.62435e-16 - 8.55703e-16 - 2.73377e-15 - 8.55729e-15 - 2.62451e-14 - 7.88674e-14 - 2.32211e-13 - 6.69893e-13 - 1.8935e-12 - 5.24398e-12 - 1.42297e-11 - 3.78324e-11 - 9.85533e-11 - 2.51544e-10 - 6.29065e-10 - 1.54139e-09 - 3.70056e-09 - 8.70479e-09 - 2.00626e-08 - 4.53056e-08 - 1.00243e-07 - 2.17317e-07 - 4.61603e-07 - 9.60688e-07 - 1.95899e-06 - 3.914e-06 - 7.66208e-06 - 1.46964e-05 - 2.76192e-05 - 5.08571e-05 - 9.17551e-05 - 0.000162199 - 0.000280936 - 0.00047677 - 0.000792781 - 0.00129164 - 0.00206196 - 0.0032253 - 0.00494329 - 0.00742376 - 0.0109245 - 0.015753 - 0.0222597 - 0.0308241 - 0.0418311 - 0.0556391 - 0.0725392 - 0.0927117 - 0.116181 - 0.142782 - 0.172134 - 0.203646 - 0.23654 - 0.269907 - 0.302784 - 0.334251 - 0.36353 - 0.390069 - 0.413606 - 0.434188 - 0.452139 - 0.467992 - 0.482371 - 0.495855 - 0.508845 - 0.521448 - 0.533411 - 0.54411 - 0.552604 - 0.557745 - 0.558334 - 0.553292 - 0.541826 - 0.523566 - 0.49865 - 0.46775 - 0.432034 - 0.393086 - 0.35278 - 0.313139 - 0.2762 - 0.243885 - 0.2179 - 0.199659 - 0.19023 - 0.190292 - 0.200109 - 0.219505 - 0.247844 - 0.284017 - 0.326452 - 0.373144 - 0.421726 - 0.469578 - 0.513978 - 0.552279 - 0.582113 - 0.601568 - 0.60936 - 0.604932 - 0.588498 - 0.561014 - 0.524074 - 0.479755 - 0.430422 - 0.378526 - 0.32641 - 0.276152 - 0.229449 - 0.187561 - 0.151296 - 0.121041 - 0.0968234 - 0.0783825 - 0.0652508 - 0.0568241 - 0.052421 - 0.0513267 - 0.0528233 - 0.0562096 - 0.0608144 - 0.0660069 - 0.0712085 - 0.0759058 - 0.0796663 - 0.0821531 - 0.083138 - 0.0825095 - 0.0802724 - 0.0765404 - 0.0715192 - 0.0654835 - 0.0587491 - 0.0516444 - 0.0444831 - 0.0375424 - 0.031047 - 0.0251611 - 0.0199871 - 0.0155706 - 0.0119103 - 0.00897034 - 0.00669384 - 0.00501675 - 0.00387995 - 0.00323985 - 0.00307697 - 0.00340228 - 0.00426134 - 0.00573584 - 0.00794229 - 0.011027 - 0.015157 - 0.0205062 - 0.0272364 - 0.0354751 - 0.0452892 - 0.0566592 - 0.0694565 - 0.0834268 - 0.0981838 - 0.113218 - 0.127916 - 0.141604 - 0.153589 - 0.163224 - 0.169959 - 0.173396 - 0.17333 - 0.169763 - 0.162911 - 0.153177 - 0.141116 - 0.127378 - 0.112654 - 0.0976202 - 0.0828835 - 0.0689498 - 0.0561998 - 0.0448822 - 0.0351196 - 0.0269254 - 0.0202261 - 0.0148867 - 0.0107355 - 0.00758544 - 0.00525143 - 0.00356214 - 0.00236745 - 0.00154166 - 0.000983633 - 0.000614914 - 0.000376646 - 0.000226043 - 0.000132922 - 7.65918e-05 - 4.32595e-05 - 2.39797e-05 - 1.31129e-05 - 7.21922e-06 - 4.30366e-06 - 3.33996e-06 - 4.01773e-06 - 6.68526e-06 - 1.24923e-05 - 2.37754e-05 - 4.47731e-05 - 8.28123e-05 - 0.000150164 - 0.000266831 - 0.000464579 - 0.000792543 - 0.00132472 - 0.00216951 - 0.00348126 - 0.0054733 - 0.00843136 - 0.0127257 - 0.0188194 - 0.0272687 - 0.0387134 - 0.053851 - 0.0733945 - 0.09801 - 0.128237 - 0.164397 - 0.206496 - 0.254135 - 0.306448 - 0.362063 - 0.41913 - 0.47539 - 0.528309 - 0.575259 - 0.613727 - 0.641541 - 0.657068 - 0.659377 - 0.648327 - 0.624588 - 0.589567 - 0.545274 - 0.494136 - 0.438774 - 0.381789 - 0.325571 - 0.272154 - 0.223122 - 0.179578 - 0.142167 - 0.111134 - 0.086417 - 0.0677452 - 0.0547365 - 0.04698 - 0.0440953 - 0.0457651 - 0.0517401 - 0.0618192 - 0.075809 - 0.0934719 - 0.114469 - 0.138309 - 0.164311 - 0.191592 - 0.21908 - 0.245557 - 0.269731 - 0.29033 - 0.306206 - 0.316433 - 0.320401 - 0.317866 - 0.308982 - 0.29428 - 0.274615 - 0.251088 - 0.224938 - 0.197441 - 0.169805 - 0.143086 - 0.118136 - 0.0955664 - 0.0757468 - 0.0588247 - 0.0447603 - 0.0333705 - 0.0243763 - 0.0174466 - 0.0122347 - 0.00840639 - 0.00565932 - 0.00373302 - 0.0024127 - 0.001528 - 0.000948448 - 0.000577417 - 0.000345638 - 0.0002051 - 0.000123843 - 8.18518e-05 - 6.81308e-05 - 7.89389e-05 - 0.00011715 - 0.000192719 - 0.000324255 - 0.000541737 - 0.000890371 - 0.00143553 - 0.00226859 - 0.00351311 - 0.00533082 - 0.00792609 - 0.0115475 - 0.0164852 - 0.0230612 - 0.0316131 - 0.0424685 - 0.0559121 - 0.0721461 - 0.0912487 - 0.113135 - 0.137526 - 0.163937 - 0.191679 - 0.219893 - 0.247604 - 0.273794 - 0.297493 - 0.317866 - 0.334291 - 0.346416 - 0.354179 - 0.357793 - 0.357701 - 0.354498 - 0.348835 - 0.34133 - 0.332489 - 0.322651 - 0.311975 - 0.30045 - 0.287942 - 0.274258 - 0.25922 - 0.242722 - 0.224784 - 0.205571 - 0.185391 - 0.164673 - 0.143916 - 0.123646 - 0.104359 - 0.0864807 - 0.0703312 - 0.0561129 - 0.0439079 - 0.0336895 - 0.0253425 - 0.0186881 - 0.0135096 - 0.00957556 - 0.00665938 - 0.00455333 - 0.00307793 - 0.00208716 - 0.00147068 - 0.00115451 - 0.00110122 - 0.00131073 - 0.0018224 - 0.00271851 - 0.00412898 - 0.00623645 - 0.00928047 - 0.0135588 - 0.0194236 - 0.0272705 - 0.0375177 - 0.0505746 - 0.0667991 - 0.0864465 - 0.109613 - 0.13618 - 0.165769 - 0.19771 - 0.231042 - 0.26454 - 0.296775 - 0.326212 - 0.351326 - 0.370729 - 0.383301 - 0.388294 - 0.385406 - 0.374811 - 0.357144 - 0.333435 - 0.305011 - 0.273374 - 0.240069 - 0.206562 - 0.174142 - 0.143844 - 0.116417 - 0.0923168 - 0.0717266 - 0.054603 - 0.0407277 - 0.0297646 - 0.0213131 - 0.0149531 - 0.010279 - 0.00692323 - 0.00456881 - 0.00295416 - 0.00187155 - 0.00116173 - 0.000706559 - 0.000421043 - 0.000245834 - 0.000140635 - 7.88285e-05 - 4.32922e-05 - 2.32955e-05 - 1.22821e-05 - 6.34466e-06 - 3.2113e-06 - 1.59255e-06 - 7.73819e-07 - 3.68407e-07 - 1.71859e-07 - 7.85727e-08 - 3.525e-08 - 1.56255e-08 - 7.1029e-09 - 3.90724e-09 - 3.74228e-09 - 6.36823e-09 - 1.3367e-08 - 2.89202e-08 - 6.18851e-08 - 1.29968e-07 - 2.67519e-07 - 5.39551e-07 - 1.06623e-06 - 2.06446e-06 - 3.91651e-06 - 7.27996e-06 - 1.32585e-05 - 2.36592e-05 - 4.13656e-05 - 7.08625e-05 - 0.000118941 - 0.000195606 - 0.000315187 - 0.000497615 - 0.000769759 - 0.00116668 - 0.00173256 - 0.00252093 - 0.00359392 - 0.00502012 - 0.00687063 - 0.00921332 - 0.0121052 - 0.0155835 - 0.019656 - 0.0242919 - 0.0294147 - 0.0348984 - 0.0405679 - 0.0462058 - 0.0515641 - 0.0563813 - 0.0604032 - 0.0634047 - 0.0652108 - 0.0657135 - 0.0648824 - 0.0627676 - 0.0594951 - 0.055254 - 0.0502786 - 0.044827 - 0.0391591 - 0.0335168 - 0.028108 - 0.0230958 - 0.018594 - 0.0146673 - 0.0113361 - 0.00858452 - 0.00636947 - 0.0046305 - 0.0032983 - 0.00230191 - 0.00157407 - 0.00105462 - 0.000692316 - 0.000445298 - 0.000280629 - 0.000173282 - 0.000104836 - 6.21444e-05 - 3.60937e-05 - 2.05399e-05 - 1.14526e-05 - 6.25667e-06 - 3.34904e-06 - 1.75645e-06 - 9.02581e-07 - 4.5444e-07 - 2.24189e-07 - 1.08378e-07 - 5.13693e-08 - 2.39435e-08 - 1.11485e-08 - 5.59702e-09 - 3.92679e-09 - 5.09955e-09 - 9.86095e-09 - 2.114e-08 - 4.55587e-08 - 9.66698e-08 - 2.0116e-07 - 4.10205e-07 - 8.19618e-07 - 1.60458e-06 - 3.07786e-06 - 5.78459e-06 - 1.06521e-05 - 1.9219e-05 - 3.39755e-05 - 5.88488e-05 - 9.98725e-05 - 0.00016607 - 0.000270566 - 0.000431909 - 0.000675535 - 0.00103524 - 0.00155443 - 0.00228684 - 0.0032964 - 0.00465564 - 0.00644253 - 0.00873514 - 0.0116043 - 0.0151046 - 0.0192634 - 0.024071 - 0.0294708 - 0.035353 - 0.0415526 - 0.0478528 - 0.053995 - 0.059695 - 0.0646638 - 0.0686316 - 0.0713724 - 0.0727252 - 0.0726101 - 0.0710369 - 0.068105 - 0.0639941 - 0.058949 - 0.0532582 - 0.0472318 - 0.0411794 - 0.035392 - 0.0301294 - 0.0256128 - 0.0220232 - 0.0195037 - 0.0181646 - 0.0180873 - 0.0193273 - 0.0219132 - 0.025842 - 0.0310708 - 0.0375062 - 0.0449944 - 0.0533132 - 0.0621703 - 0.0712086 - 0.0800204 - 0.0881704 - 0.0952259 - 0.10079 - 0.104537 - 0.106239 - 0.105792 - 0.10322 - 0.0986778 - 0.0924299 - 0.0848289 - 0.0762804 - 0.0672077 - 0.058018 - 0.049073 - 0.0406687 - 0.0330228 - 0.0262727 - 0.0204801 - 0.0156422 - 0.0117058 - 0.00858298 - 0.00616616 - 0.00434041 - 0.00299359 - 0.00202311 - 0.00133989 - 0.000870016 - 0.000554601 - 0.00034854 - 0.000218741 - 0.000142232 - 0.000104541 - 9.8566e-05 - 0.000124155 - 0.000188521 - 0.000307637 - 0.000508725 - 0.000833915 - 0.00134509 - 0.00212979 - 0.00330782 - 0.00503791 - 0.00752352 - 0.0110162 - 0.0158148 - 0.0222592 - 0.0307153 - 0.0415515 - 0.0551058 - 0.071643 - 0.0913074 - 0.114073 - 0.1397 - 0.167701 - 0.197328 - 0.227588 - 0.257282 - 0.285075 - 0.309593 - 0.329531 - 0.34377 - 0.35148 - 0.352196 - 0.345872 - 0.332879 - 0.313973 - 0.29022 - 0.262897 - 0.233379 - 0.203026 - 0.173082 - 0.144596 - 0.118376 - 0.0949657 - 0.0746562 - 0.057512 - 0.0434158 - 0.0321176 - 0.0232853 - 0.0165482 - 0.0115343 - 0.00789618 - 0.00532826 - 0.00357577 - 0.00243735 - 0.00176332 - 0.00145159 - 0.00144239 - 0.00171284 - 0.00227175 - 0.00315462 - 0.00441846 - 0.0061359 - 0.00838817 - 0.0112563 - 0.0148106 - 0.019099 - 0.0241341 - 0.0298819 - 0.0362517 - 0.0430913 - 0.0501867 - 0.0572696 - 0.064032 - 0.0701466 - 0.0752928 - 0.0791839 - 0.0815939 - 0.0823787 - 0.0814908 - 0.0789841 - 0.075008 - 0.069793 - 0.0636288 - 0.0568371 - 0.0497447 - 0.0426579 - 0.0358416 - 0.0295062 - 0.0238 - 0.0188094 - 0.0145651 - 0.0110507 - 0.00821493 - 0.00598368 - 0.00427072 - 0.00298721 - 0.00204851 - 0.00137888 - 0.000914107 - 0.0006025 - 0.000404956 - 0.000294455 - 0.000255581 - 0.000284413 - 0.000389119 - 0.00059143 - 0.000929092 - 0.0014592 - 0.00226219 - 0.00344601 - 0.00514967 - 0.00754534 - 0.0108375 - 0.0152582 - 0.0210566 - 0.0284828 - 0.0377642 - 0.0490773 - 0.0625148 - 0.0780522 - 0.0955182 - 0.114573 - 0.134702 - 0.155224 - 0.175321 - 0.194088 - 0.210595 - 0.223966 - 0.233452 - 0.238502 - 0.238817 - 0.234375 - 0.225439 - 0.212528 - 0.196368 - 0.177823 - 0.157822 - 0.13728 - 0.117031 - 0.0977797 - 0.0800658 - 0.0642531 - 0.0505343 - 0.0389512 - 0.0294236 - 0.0217826 - 0.0158036 - 0.0112366 - 0.00782971 - 0.00534666 - 0.00357803 - 0.00234654 - 0.00150811 - 0.000949846 - 0.000586259 - 0.0003546 - 0.000210184 - 0.000122087 - 6.94932e-05 - 3.87633e-05 - 2.11886e-05 - 1.13497e-05 - 5.95747e-06 - 3.06434e-06 - 1.54456e-06 - 7.62898e-07 - 3.69248e-07 - 1.75129e-07 - 8.13929e-08 - 3.7068e-08 - 1.65423e-08 - 7.23393e-09 - 3.0998e-09 - 1.30158e-09 - 5.35535e-10 - 2.15914e-10 - 8.52996e-11 - 3.30208e-11 - 1.25257e-11 - 4.65589e-12 - 1.69628e-12 - 6.07133e-13 - 2.17868e-13 - 9.17968e-14 - 8.26531e-14 - 1.80758e-13 - 5.04118e-13 - 1.4277e-12 - 3.9783e-12 - 1.0868e-11 - 2.90954e-11 - 7.63304e-11 - 1.96231e-10 - 4.94351e-10 - 1.2204e-09 - 2.95232e-09 - 6.9988e-09 - 1.62586e-08 - 3.70116e-08 - 8.25642e-08 - 1.80486e-07 - 3.86628e-07 - 8.11596e-07 - 1.66949e-06 - 3.36533e-06 - 6.64764e-06 - 1.28678e-05 - 2.44085e-05 - 4.53704e-05 - 8.26422e-05 - 0.000147512 - 0.000258019 - 0.000442253 - 0.000742825 - 0.00122264 - 0.001972 - 0.00311681 - 0.00482735 - 0.00732661 - 0.0108966 - 0.0158809 - 0.0226805 - 0.0317412 - 0.04353 - 0.0584987 - 0.0770365 - 0.0994122 - 0.125711 - 0.155777 - 0.189156 - 0.225077 - 0.262443 - 0.299868 - 0.335752 - 0.368386 - 0.396083 - 0.417326 - 0.430901 - 0.436023 - 0.432407 - 0.420307 - 0.400488 - 0.374157 - 0.34285 - 0.308297 - 0.27227 - 0.236449 - 0.202305 - 0.171016 - 0.143419 - 0.12 - 0.10091 - 0.0860138 - 0.0749435 - 0.0671671 - 0.0620535 - 0.0589346 - 0.0571569 - 0.0561247 - 0.0553308 - 0.0543762 - 0.052979 - 0.0509737 - 0.0483016 - 0.0449951 - 0.0411568 - 0.0369364 - 0.0325078 - 0.0280481 - 0.0237199 - 0.0196589 - 0.0159663 - 0.0127065 - 0.00990863 - 0.00757099 - 0.00566812 - 0.00415785 - 0.00298841 - 0.00210451 - 0.00145211 - 0.000981719 - 0.000650297 - 0.00042206 - 0.000268394 - 0.000167228 - 0.00010209 - 6.1065e-05 - 3.57882e-05 - 2.05505e-05 - 1.15623e-05 - 6.37384e-06 - 3.44267e-06 - 1.82192e-06 - 9.44718e-07 - 4.79996e-07 - 2.39019e-07 - 1.16785e-07 - 5.63189e-08 - 2.75876e-08 - 1.55066e-08 - 1.35065e-08 - 2.06346e-08 - 4.12348e-08 - 8.71701e-08 - 1.83491e-07 - 3.79656e-07 - 7.70155e-07 - 1.53093e-06 - 2.98182e-06 - 5.69044e-06 - 1.06401e-05 - 1.94933e-05 - 3.49914e-05 - 6.15423e-05 - 0.000106053 - 0.000179064 - 0.000296231 - 0.000480165 - 0.000762583 - 0.00118664 - 0.00180922 - 0.0027027 - 0.00395586 - 0.00567312 - 0.00797149 - 0.0109747 - 0.0148042 - 0.0195665 - 0.0253384 - 0.03215 - 0.0399688 - 0.0486854 - 0.0581051 - 0.0679468 - 0.0778511 - 0.0873986 - 0.0961378 - 0.10362 - 0.109438 - 0.113265 - 0.11489 - 0.114237 - 0.111386 - 0.106565 - 0.100144 - 0.092608 - 0.0845337 - 0.0765571 - 0.069347 - 0.0635806 - 0.0599254 - 0.0590243 - 0.0614823 - 0.0678499 - 0.0786025 - 0.0941094 - 0.114595 - 0.140095 - 0.170403 - 0.205033 - 0.243187 - 0.28375 - 0.325307 - 0.366205 - 0.404639 - 0.438768 - 0.466855 - 0.487399 - 0.499265 - 0.501787 - 0.494819 - 0.478757 - 0.454493 - 0.423343 - 0.386921 - 0.347013 - 0.305437 - 0.263913 - 0.223964 - 0.186845 - 0.15351 - 0.124609 - 0.100516 - 0.0813752 - 0.0671528 - 0.0576947 - 0.0527714 - 0.0521107 - 0.0554152 - 0.0623635 - 0.0726003 - 0.0857196 - 0.101248 - 0.118632 - 0.137242 - 0.156382 - 0.175324 - 0.193347 - 0.209787 - 0.224084 - 0.235824 - 0.244759 - 0.250819 - 0.25409 - 0.254781 - 0.253179 - 0.249598 - 0.24433 - 0.237611 - 0.229608 - 0.220414 - 0.210075 - 0.198611 - 0.186058 - 0.172493 - 0.15806 - 0.142974 - 0.127518 - 0.112024 - 0.0968447 - 0.0823232 - 0.0687639 - 0.056409 - 0.0454241 - 0.0358933 - 0.0278226 - 0.0211511 - 0.0157665 - 0.0115222 - 0.00825423 - 0.0057959 - 0.0039887 - 0.00269018 - 0.00177808 - 0.00115167 - 0.000731004 - 0.000454751 - 0.000277398 - 0.000166222 - 9.846e-05 - 5.88987e-05 - 3.79727e-05 - 3.04441e-05 - 3.46788e-05 - 5.25582e-05 - 9.00977e-05 - 0.000158898 - 0.000278618 - 0.000480699 - 0.000813605 - 0.00134977 - 0.00219438 - 0.00349573 - 0.00545668 - 0.00834609 - 0.0125084 - 0.0183688 - 0.0264315 - 0.0372673 - 0.0514867 - 0.0696987 - 0.0924523 - 0.120163 - 0.153035 - 0.190972 - 0.233514 - 0.27978 - 0.328461 - 0.377845 - 0.425897 - 0.470391 - 0.509067 - 0.539825 - 0.56091 - 0.571077 - 0.569717 - 0.55691 - 0.533424 - 0.500637 - 0.460399 - 0.414867 - 0.366308 - 0.31692 - 0.268674 - 0.223194 - 0.181698 - 0.144974 - 0.113407 - 0.0870373 - 0.0656418 - 0.048819 - 0.0360773 - 0.0269108 - 0.0208626 - 0.0175701 - 0.0167947 - 0.0184339 - 0.0225191 - 0.0291988 - 0.0387082 - 0.0513264 - 0.067323 - 0.0868961 - 0.110107 - 0.136818 - 0.166639 - 0.198893 - 0.232612 - 0.266561 - 0.299299 - 0.329271 - 0.354927 - 0.374855 - 0.387903 - 0.393297 - 0.390711 - 0.3803 - 0.36269 - 0.338907 - 0.310288 - 0.278347 - 0.244652 - 0.210696 - 0.177796 - 0.147015 - 0.119132 - 0.0946281 - 0.0737165 - 0.0563825 - 0.0424423 - 0.0316028 - 0.0235195 - 0.0178455 - 0.0142692 - 0.0125399 - 0.0124803 - 0.0139879 - 0.0170253 - 0.0216027 - 0.0277537 - 0.0355053 - 0.0448476 - 0.0557031 - 0.0679011 - 0.0811603 - 0.0950824 - 0.10916 - 0.1228 - 0.135357 - 0.146187 - 0.154696 - 0.160393 - 0.162941 - 0.162186 - 0.158174 - 0.151147 - 0.141516 - 0.129827 - 0.116706 - 0.102807 - 0.0887582 - 0.0751238 - 0.062369 - 0.0508463 - 0.0407927 - 0.0323401 - 0.0255341 - 0.0203573 - 0.0167544 - 0.0146531 - 0.0139808 - 0.0146745 - 0.0166817 - 0.019956 - 0.0244455 - 0.0300777 - 0.0367437 - 0.0442835 - 0.0524759 - 0.0610361 - 0.0696209 - 0.0778441 - 0.0852996 - 0.0915913 - 0.096366 - 0.0993442 - 0.100347 - 0.0993131 - 0.0963045 - 0.0915008 - 0.0851805 - 0.0776949 - 0.0694355 - 0.0608006 - 0.052164 - 0.0438501 - 0.0361166 - 0.0291461 - 0.0230458 - 0.0178541 - 0.0135526 - 0.0100796 - 0.00734517 - 0.00524441 - 0.00366883 - 0.00251476 - 0.00168889 - 0.00111133 - 0.00071651 - 0.000452625 - 0.00028015 - 0.000169895 - 0.00010095 - 5.87719e-05 - 3.35251e-05 - 1.87373e-05 - 1.02608e-05 - 5.50542e-06 - 2.89426e-06 - 1.49081e-06 - 7.52391e-07 - 3.7205e-07 - 1.80259e-07 - 8.55712e-08 - 3.98012e-08 - 1.81385e-08 - 8.09923e-09 - 3.54342e-09 - 1.51893e-09 - 6.37957e-10 - 2.62531e-10 - 1.05854e-10 - 4.18187e-11 - 1.61872e-11 - 6.13914e-12 - 2.28129e-12 - 8.306e-13 - 2.96305e-13 - 1.03568e-13 - 3.54687e-14 - 1.19015e-14 - 3.91289e-15 - 1.26046e-15 - 3.9783e-16 - 1.23028e-16 - 3.72774e-17 - 1.10669e-17 - 3.21915e-18 - 9.17473e-19 - 2.56202e-19 - 7.00986e-20 - 1.8792e-20 - 4.93598e-21 - 1.27031e-21 - 3.2032e-22 - 7.91398e-23 - 1.91577e-23 - 4.54388e-24 - 1.05596e-24 - 2.4044e-25 - 5.36415e-26 - 1.17255e-26 - 2.51132e-27 - 5.26996e-28 - 1.08355e-28 - 2.18287e-29 - 4.30869e-30 - 8.33293e-31 - 1.57902e-31 - 2.93166e-32 - 5.33307e-33 - 9.50556e-34 - 1.66003e-34 - 2.84046e-35 - 4.76211e-36 - 7.82253e-37 - 1.25902e-37 - 1.98542e-38 - 3.06769e-39 - 4.64415e-40 - 6.88872e-41 - 1.00117e-41 - 1.42565e-42 - 1.98909e-43 - 2.71916e-44 - 3.64209e-45 - 4.77973e-46 - 6.14602e-47 - 7.7432e-48 - 9.55837e-49 - 1.15607e-49 - 1.37e-50 - 1.59073e-51 - 1.8097e-52 - 2.01722e-53 - 2.20312e-54 - 2.35755e-55 - 2.47183e-56 - 2.5393e-57 - 2.55591e-58 - 2.52066e-59 - 2.43568e-60 - 2.30602e-61 - 2.13916e-62 - 1.94428e-63 - 1.73146e-64 - 1.51078e-65 - 1.2916e-66 - 1.08191e-67 - 8.87958e-69 - 7.14052e-70 - 5.62606e-71 - 4.34325e-72 - 3.28521e-73 - 2.43471e-74 - 1.76795e-75 - 1.25785e-76 - 8.76845e-78 - 5.98901e-79 - 4.00797e-80 - 2.62803e-81 - 1.68839e-82 - 1.0628e-83 - 6.55493e-85 - 3.96115e-86 - 2.34537e-87 - 1.36063e-88 - 7.73397e-90 - 4.30728e-91 - 2.35039e-92 - 1.25665e-93 - 6.58303e-95 - 3.37889e-96 - 1.69925e-97 - 8.37298e-99 - 4.04239e-100 - 1.9122e-101 - 8.8627e-103 - 4.02472e-104 - 1.79078e-105 - 7.807e-107 - 3.33476e-108 - 1.39764e-109 - 1.06571e-110 - 1.20963e-109 - 2.89465e-108 - 6.79983e-107 - 1.56509e-105 - 3.52952e-104 - 7.79884e-103 - 1.68842e-101 - 3.58153e-100 - 7.44378e-99 - 1.51585e-97 - 3.0245e-96 - 5.91274e-95 - 1.13256e-93 - 2.12554e-92 - 3.90855e-91 - 7.04204e-90 - 1.24313e-88 - 2.15018e-87 - 3.64391e-86 - 6.05059e-85 - 9.84384e-84 - 1.56916e-82 - 2.45081e-81 - 3.75048e-80 - 5.62342e-79 - 8.26136e-78 - 1.18916e-76 - 1.67712e-75 - 2.31753e-74 - 3.1378e-73 - 4.16256e-72 - 5.41044e-71 - 6.89035e-70 - 8.5978e-69 - 1.05116e-67 - 1.25919e-66 - 1.4779e-65 - 1.69957e-64 - 1.915e-63 - 2.11415e-62 - 2.28686e-61 - 2.42371e-60 - 2.51686e-59 - 2.56079e-58 - 2.55284e-57 - 2.49352e-56 - 2.38637e-55 - 2.23769e-54 - 2.05588e-53 - 1.85069e-52 - 1.63232e-51 - 1.41064e-50 - 1.19443e-49 - 9.90932e-49 - 8.05496e-48 - 6.41535e-47 - 5.00626e-46 - 3.82775e-45 - 2.86755e-44 - 2.10482e-43 - 1.51376e-42 - 1.06668e-41 - 7.36458e-41 - 4.98195e-40 - 3.30208e-39 - 2.14443e-38 - 1.3645e-37 - 8.50695e-37 - 5.19648e-36 - 3.11015e-35 - 1.82386e-34 - 1.04794e-33 - 5.89958e-33 - 3.25418e-32 - 1.75873e-31 - 9.31305e-31 - 4.83195e-30 - 2.45635e-29 - 1.22347e-28 - 5.97085e-28 - 2.85506e-27 - 1.33761e-26 - 6.1402e-26 - 2.76167e-25 - 1.21702e-24 - 5.25487e-24 - 2.22312e-23 - 9.21509e-23 - 3.74261e-22 - 1.48931e-21 - 5.80677e-21 - 2.2183e-20 - 8.30318e-20 - 3.04513e-19 - 1.09422e-18 - 3.85247e-18 - 1.32896e-17 - 4.49183e-17 - 1.48755e-16 - 4.82679e-16 - 1.53456e-15 - 4.7802e-15 - 1.45897e-14 - 4.36302e-14 - 1.2784e-13 - 3.67016e-13 - 1.03239e-12 - 2.84538e-12 - 7.68384e-12 - 2.03309e-11 - 5.27081e-11 - 1.33887e-10 - 3.33231e-10 - 8.12632e-10 - 1.94172e-09 - 4.54597e-09 - 1.04283e-08 - 2.34394e-08 - 5.16215e-08 - 1.11395e-07 - 2.35536e-07 - 4.87981e-07 - 9.90622e-07 - 1.97049e-06 - 3.84067e-06 - 7.33515e-06 - 1.37273e-05 - 2.51733e-05 - 4.52355e-05 - 7.96547e-05 - 0.000137449 - 0.000232424 - 0.000385157 - 0.000625493 - 0.000995518 - 0.00155285 - 0.00237402 - 0.00355735 - 0.00522491 - 0.00752254 - 0.0106172 - 0.0146907 - 0.0199294 - 0.0265096 - 0.0345787 - 0.0442341 - 0.0555004 - 0.0683094 - 0.0824842 - 0.0977307 - 0.11364 - 0.129703 - 0.145333 - 0.159905 - 0.172795 - 0.183427 - 0.191318 - 0.19611 - 0.197601 - 0.195756 - 0.190705 - 0.18273 - 0.172236 - 0.15972 - 0.145736 - 0.130848 - 0.115605 - 0.100507 - 0.0859807 - 0.0723703 - 0.0599273 - 0.0488129 - 0.0391035 - - - - - 4.23161e-30 - 5.26614e-30 - 6.42119e-30 - 7.67142e-30 - 8.97998e-30 - 1.02997e-29 - 1.15762e-29 - 1.27563e-29 - 1.38172e-29 - 1.48982e-29 - 1.69439e-29 - 2.48572e-29 - 6.27124e-29 - 2.45692e-28 - 1.1132e-27 - 5.13916e-27 - 2.34357e-26 - 1.04886e-25 - 4.6008e-25 - 1.9775e-24 - 8.32801e-24 - 3.43641e-23 - 1.38933e-22 - 5.50355e-22 - 2.13608e-21 - 8.12322e-21 - 3.02675e-20 - 1.105e-19 - 3.9526e-19 - 1.38529e-18 - 4.75704e-18 - 1.60055e-17 - 5.27641e-17 - 1.70429e-16 - 5.3937e-16 - 1.6725e-15 - 5.0814e-15 - 1.51264e-14 - 4.41191e-14 - 1.26082e-13 - 3.53035e-13 - 9.68542e-13 - 2.60349e-12 - 6.85694e-12 - 1.76946e-11 - 4.47394e-11 - 1.10834e-10 - 2.69027e-10 - 6.39816e-10 - 1.49091e-09 - 3.40395e-09 - 7.61469e-09 - 1.66901e-08 - 3.58427e-08 - 7.5419e-08 - 1.55488e-07 - 3.14088e-07 - 6.21643e-07 - 1.2055e-06 - 2.29051e-06 - 4.26416e-06 - 7.77805e-06 - 1.3901e-05 - 2.4342e-05 - 4.17643e-05 - 7.02086e-05 - 0.000115641 - 0.000186625 - 0.000295098 - 0.000457192 - 0.000694014 - 0.00103222 - 0.00150424 - 0.00214782 - 0.00300479 - 0.00411877 - 0.00553169 - 0.00727922 - 0.00938532 - 0.0118563 - 0.0146753 - 0.0177977 - 0.0211483 - 0.0246221 - 0.0280875 - 0.0313934 - 0.0343799 - 0.0368904 - 0.0387854 - 0.0399557 - 0.0403326 - 0.0398957 - 0.0386751 - 0.036749 - 0.0342377 - 0.0312926 - 0.0280847 - 0.0247919 - 0.0215871 - 0.0186297 - 0.0160581 - 0.013987 - 0.0125051 - 0.0116762 - 0.0115397 - 0.0121108 - 0.0133811 - 0.0153167 - 0.0178562 - 0.0209094 - 0.0243561 - 0.0280473 - 0.0318096 - 0.0354525 - 0.0387797 - 0.041602 - 0.0437517 - 0.0450968 - 0.045552 - 0.0450867 - 0.043727 - 0.041553 - 0.03869 - 0.0352968 - 0.031551 - 0.027633 - 0.0237127 - 0.0199375 - 0.0164247 - 0.0132575 - 0.0104848 - 0.00812456 - 0.00616843 - 0.00458866 - 0.00334453 - 0.00238847 - 0.00167125 - 0.00114578 - 0.000769656 - 0.000506558 - 0.000326662 - 0.000206398 - 0.000127776 - 7.75048e-05 - 4.60623e-05 - 2.68225e-05 - 1.53035e-05 - 8.55494e-06 - 4.68577e-06 - 2.51468e-06 - 1.32227e-06 - 6.81232e-07 - 3.4388e-07 - 1.70081e-07 - 8.24215e-08 - 3.91348e-08 - 1.82063e-08 - 8.29885e-09 - 3.70639e-09 - 1.62189e-09 - 6.95389e-10 - 2.92132e-10 - 1.2026e-10 - 4.85509e-11 - 1.93322e-11 - 7.90272e-12 - 4.15729e-12 - 4.69603e-12 - 1.00995e-11 - 2.54898e-11 - 6.48543e-11 - 1.62322e-10 - 3.9828e-10 - 9.57567e-10 - 2.25575e-09 - 5.20657e-09 - 1.17747e-08 - 2.60905e-08 - 5.66438e-08 - 1.20492e-07 - 2.51133e-07 - 5.12843e-07 - 1.02613e-06 - 2.01167e-06 - 3.8641e-06 - 7.27238e-06 - 1.34104e-05 - 2.42295e-05 - 4.28927e-05 - 7.43975e-05 - 0.000126436 - 0.000210533 - 0.000343484 - 0.000549073 - 0.000859983 - 0.00131974 - 0.00198436 - 0.00292342 - 0.00421986 - 0.00596819 - 0.00827035 - 0.011229 - 0.0149381 - 0.019471 - 0.0248666 - 0.0311158 - 0.0381491 - 0.0458271 - 0.0539385 - 0.062203 - 0.0702847 - 0.0778121 - 0.0844055 - 0.0897079 - 0.0934174 - 0.0953152 - 0.095287 - 0.0933346 - 0.0895758 - 0.0842323 - 0.0776088 - 0.0700637 - 0.0619781 - 0.0537245 - 0.0456401 - 0.0380067 - 0.0310392 - 0.0248817 - 0.0196118 - 0.0152499 - 0.0117717 - 0.00912236 - 0.00722982 - 0.00601571 - 0.00540334 - 0.0053222 - 0.00570923 - 0.00650744 - 0.00766253 - 0.00911876 - 0.0108148 - 0.0126804 - 0.0146355 - 0.0165902 - 0.0184485 - 0.020113 - 0.0214914 - 0.022504 - 0.0230902 - 0.0232141 - 0.022868 - 0.0220728 - 0.020876 - 0.0193474 - 0.0175725 - 0.0156453 - 0.0136615 - 0.0117119 - 0.00987887 - 0.00823454 - 0.00684165 - 0.00575724 - 0.00503837 - 0.00474935 - 0.00496969 - 0.00580156 - 0.00737575 - 0.00985451 - 0.0134302 - 0.0183181 - 0.0247429 - 0.0329181 - 0.0430191 - 0.0551515 - 0.0693182 - 0.0853881 - 0.103073 - 0.121914 - 0.141292 - 0.160443 - 0.178512 - 0.194604 - 0.207861 - 0.217536 - 0.223062 - 0.224109 - 0.220611 - 0.212781 - 0.201084 - 0.18619 - 0.168917 - 0.15015 - 0.130773 - 0.111595 - 0.0933057 - 0.0764379 - 0.0613545 - 0.0482527 - 0.037182 - 0.0280725 - 0.0207666 - 0.0150518 - 0.0106892 - 0.00743777 - 0.00507078 - 0.00338723 - 0.00221692 - 0.00142165 - 0.000893251 - 0.000549908 - 0.000331699 - 0.000196035 - 0.000113517 - 6.44059e-05 - 3.58036e-05 - 1.95013e-05 - 1.04073e-05 - 5.4419e-06 - 2.78804e-06 - 1.39954e-06 - 6.88345e-07 - 3.31715e-07 - 1.56625e-07 - 7.24592e-08 - 3.28445e-08 - 1.45871e-08 - 6.34764e-09 - 2.7064e-09 - 1.1306e-09 - 4.62768e-10 - 1.8559e-10 - 7.29258e-11 - 2.80766e-11 - 1.05912e-11 - 3.91457e-12 - 1.41762e-12 - 5.03003e-13 - 1.74872e-13 - 5.95668e-14 - 1.98805e-14 - 6.50108e-15 - 2.08296e-15 - 6.53904e-16 - 2.01133e-16 - 6.06164e-17 - 1.78992e-17 - 5.17861e-18 - 1.46801e-18 - 4.07741e-19 - 1.10962e-19 - 2.95871e-20 - 7.72981e-21 - 1.97881e-21 - 4.96994e-22 - 1.2513e-22 - 4.27973e-23 - 6.25412e-23 - 2.27169e-22 - 8.96772e-22 - 3.49045e-21 - 1.3316e-20 - 4.97752e-20 - 1.82301e-19 - 6.54186e-19 - 2.30012e-18 - 7.92386e-18 - 2.67461e-17 - 8.84545e-17 - 2.86627e-16 - 9.1002e-16 - 2.83088e-15 - 8.62837e-15 - 2.57675e-14 - 7.5397e-14 - 2.16158e-13 - 6.07193e-13 - 1.67116e-12 - 4.50658e-12 - 1.19073e-11 - 3.08258e-11 - 7.81904e-11 - 1.94325e-10 - 4.73197e-10 - 1.12899e-09 - 2.63923e-09 - 6.04507e-09 - 1.35663e-08 - 2.98304e-08 - 6.42676e-08 - 1.35663e-07 - 2.80589e-07 - 5.6861e-07 - 1.12901e-06 - 2.19641e-06 - 4.18667e-06 - 7.81917e-06 - 1.43083e-05 - 2.5654e-05 - 4.50668e-05 - 7.75704e-05 - 0.000130819 - 0.000216165 - 0.000349973 - 0.000555164 - 0.000862869 - 0.00131403 - 0.00196066 - 0.00286641 - 0.00410592 - 0.00576262 - 0.00792443 - 0.0106771 - 0.0140955 - 0.0182326 - 0.0231079 - 0.0286958 - 0.0349164 - 0.0416297 - 0.0486353 - 0.0556796 - 0.0624697 - 0.0686941 - 0.0740496 - 0.07827 - 0.0811551 - 0.0825952 - 0.0825891 - 0.0812525 - 0.0788147 - 0.0756052 - 0.0720296 - 0.068538 - 0.0655882 - 0.0636076 - 0.0629552 - 0.0638897 - 0.0665429 - 0.0709037 - 0.0768119 - 0.0839642 - 0.0919334 - 0.100197 - 0.108179 - 0.115294 - 0.120995 - 0.124823 - 0.126445 - 0.125678 - 0.122506 - 0.117071 - 0.10966 - 0.100669 - 0.0905633 - 0.0798348 - 0.0689607 - 0.0583676 - 0.0484063 - 0.0393372 - 0.0313261 - 0.0244505 - 0.0187122 - 0.0140553 - 0.0103846 - 0.00758494 - 0.00553752 - 0.00413347 - 0.00328399 - 0.00292721 - 0.00303184 - 0.00359822 - 0.00465668 - 0.00626355 - 0.00849453 - 0.0114355 - 0.0151707 - 0.019769 - 0.0252679 - 0.0316588 - 0.0388727 - 0.0467703 - 0.0551377 - 0.0636899 - 0.0720829 - 0.0799342 - 0.08685 - 0.092458 - 0.0964398 - 0.0985609 - 0.0986938 - 0.0968305 - 0.0930832 - 0.0876733 - 0.0809096 - 0.0731593 - 0.0648151 - 0.0562625 - 0.0478519 - 0.0398764 - 0.0325589 - 0.0260472 - 0.0204168 - 0.0156802 - 0.0117992 - 0.00869946 - 0.00628445 - 0.00444814 - 0.0030848 - 0.0020961 - 0.00139552 - 0.00091032 - 0.000581822 - 0.000364354 - 0.000223559 - 0.0001344 - 7.91667e-05 - 4.56902e-05 - 2.5837e-05 - 1.43154e-05 - 7.77184e-06 - 4.13516e-06 - 2.15827e-06 - 1.10966e-06 - 5.72705e-07 - 3.20383e-07 - 2.41707e-07 - 3.04402e-07 - 5.4587e-07 - 1.09169e-06 - 2.20877e-06 - 4.41012e-06 - 8.641e-06 - 1.65944e-05 - 3.12269e-05 - 5.75758e-05 - 0.000104013 - 0.000184109 - 0.0003193 - 0.000542574 - 0.00090335 - 0.00147364 - 0.00235538 - 0.00368866 - 0.00565995 - 0.0085093 - 0.0125346 - 0.0180912 - 0.0255834 - 0.0354476 - 0.048123 - 0.0640111 - 0.0834246 - 0.10653 - 0.133286 - 0.163393 - 0.196254 - 0.230963 - 0.266319 - 0.300885 - 0.333069 - 0.361247 - 0.383895 - 0.399721 - 0.407791 - 0.407621 - 0.399219 - 0.383092 - 0.36019 - 0.331816 - 0.299502 - 0.264874 - 0.229517 - 0.194862 - 0.162098 - 0.132119 - 0.105509 - 0.0825559 - 0.0632915 - 0.0475422 - 0.0349905 - 0.0252324 - 0.017828 - 0.012342 - 0.00837148 - 0.00556362 - 0.00362284 - 0.00231142 - 0.00144492 - 0.000885006 - 0.000531112 - 0.000312293 - 0.000179918 - 0.000101561 - 5.61709e-05 - 3.04393e-05 - 1.6162e-05 - 8.408e-06 - 4.28575e-06 - 2.14042e-06 - 1.04738e-06 - 5.0217e-07 - 2.35902e-07 - 1.0858e-07 - 4.89673e-08 - 2.16371e-08 - 9.36772e-09 - 3.97414e-09 - 1.65288e-09 - 6.76192e-10 - 2.78154e-10 - 1.30846e-10 - 1.07446e-10 - 1.85993e-10 - 4.31066e-10 - 1.04002e-09 - 2.48173e-09 - 5.81045e-09 - 1.33319e-08 - 2.99725e-08 - 6.60228e-08 - 1.42496e-07 - 3.01333e-07 - 6.24351e-07 - 1.2675e-06 - 2.52117e-06 - 4.91355e-06 - 9.38262e-06 - 1.75546e-05 - 3.21805e-05 - 5.78006e-05 - 0.000101721 - 0.000175397 - 0.000296328 - 0.000490523 - 0.00079558 - 0.00126428 - 0.00196853 - 0.00300315 - 0.00448898 - 0.00657439 - 0.00943411 - 0.0132642 - 0.0182726 - 0.0246636 - 0.0326174 - 0.0422648 - 0.0536592 - 0.0667494 - 0.0813555 - 0.0971547 - 0.113678 - 0.130325 - 0.146391 - 0.161116 - 0.17374 - 0.183568 - 0.190034 - 0.192754 - 0.191563 - 0.186533 - 0.177966 - 0.166363 - 0.152375 - 0.136743 - 0.120236 - 0.103586 - 0.0874387 - 0.0723175 - 0.058603 - 0.0465301 - 0.036198 - 0.0275912 - 0.0206061 - 0.0150784 - 0.0108107 - 0.00759432 - 0.00522709 - 0.00352508 - 0.00232924 - 0.00150798 - 0.000956565 - 0.000594525 - 0.000362045 - 0.000216019 - 0.000126287 - 7.23372e-05 - 4.05977e-05 - 2.23243e-05 - 1.2028e-05 - 6.34954e-06 - 3.2842e-06 - 1.66439e-06 - 8.26447e-07 - 4.0208e-07 - 1.91667e-07 - 8.95198e-08 - 4.09664e-08 - 1.83685e-08 - 8.06966e-09 - 3.47355e-09 - 1.46497e-09 - 6.05372e-10 - 2.45105e-10 - 9.7234e-11 - 3.77939e-11 - 1.43933e-11 - 5.37079e-12 - 1.9636e-12 - 7.03401e-13 - 2.46882e-13 - 8.49013e-14 - 2.86072e-14 - 9.44439e-15 - 3.05498e-15 - 9.68234e-16 - 3.00669e-16 - 9.14817e-17 - 2.7272e-17 - 7.96591e-18 - 2.27978e-18 - 6.39307e-19 - 1.7579e-19 - 4.79091e-20 - 1.49851e-20 - 1.3075e-20 - 3.86379e-20 - 1.42464e-19 - 5.24334e-19 - 1.89316e-18 - 6.6979e-18 - 2.32182e-17 - 7.886e-17 - 2.62435e-16 - 8.55703e-16 - 2.73377e-15 - 8.55729e-15 - 2.62451e-14 - 7.88674e-14 - 2.32211e-13 - 6.69893e-13 - 1.8935e-12 - 5.24398e-12 - 1.42297e-11 - 3.78324e-11 - 9.85533e-11 - 2.51544e-10 - 6.29065e-10 - 1.54139e-09 - 3.70056e-09 - 8.70479e-09 - 2.00626e-08 - 4.53056e-08 - 1.00243e-07 - 2.17317e-07 - 4.61603e-07 - 9.60688e-07 - 1.95899e-06 - 3.914e-06 - 7.66208e-06 - 1.46964e-05 - 2.76192e-05 - 5.08571e-05 - 9.17551e-05 - 0.000162199 - 0.000280936 - 0.00047677 - 0.000792781 - 0.00129164 - 0.00206196 - 0.0032253 - 0.00494329 - 0.00742376 - 0.0109245 - 0.015753 - 0.0222597 - 0.0308241 - 0.0418311 - 0.0556391 - 0.0725392 - 0.0927117 - 0.116181 - 0.142782 - 0.172134 - 0.203646 - 0.23654 - 0.269907 - 0.302784 - 0.334251 - 0.36353 - 0.390069 - 0.413606 - 0.434188 - 0.452139 - 0.467992 - 0.482371 - 0.495855 - 0.508845 - 0.521448 - 0.533411 - 0.54411 - 0.552604 - 0.557745 - 0.558334 - 0.553292 - 0.541826 - 0.523566 - 0.49865 - 0.46775 - 0.432034 - 0.393086 - 0.35278 - 0.313139 - 0.2762 - 0.243885 - 0.2179 - 0.199659 - 0.19023 - 0.190292 - 0.200109 - 0.219505 - 0.247844 - 0.284017 - 0.326452 - 0.373144 - 0.421726 - 0.469578 - 0.513978 - 0.552279 - 0.582113 - 0.601568 - 0.60936 - 0.604932 - 0.588498 - 0.561014 - 0.524074 - 0.479755 - 0.430422 - 0.378526 - 0.32641 - 0.276152 - 0.229449 - 0.187561 - 0.151296 - 0.121041 - 0.0968234 - 0.0783825 - 0.0652508 - 0.0568241 - 0.052421 - 0.0513267 - 0.0528233 - 0.0562096 - 0.0608144 - 0.0660069 - 0.0712085 - 0.0759058 - 0.0796663 - 0.0821531 - 0.083138 - 0.0825095 - 0.0802724 - 0.0765404 - 0.0715192 - 0.0654835 - 0.0587491 - 0.0516444 - 0.0444831 - 0.0375424 - 0.031047 - 0.0251611 - 0.0199871 - 0.0155706 - 0.0119103 - 0.00897034 - 0.00669384 - 0.00501675 - 0.00387995 - 0.00323985 - 0.00307697 - 0.00340228 - 0.00426134 - 0.00573584 - 0.00794229 - 0.011027 - 0.015157 - 0.0205062 - 0.0272364 - 0.0354751 - 0.0452892 - 0.0566592 - 0.0694565 - 0.0834268 - 0.0981838 - 0.113218 - 0.127916 - 0.141604 - 0.153589 - 0.163224 - 0.169959 - 0.173396 - 0.17333 - 0.169763 - 0.162911 - 0.153177 - 0.141116 - 0.127378 - 0.112654 - 0.0976202 - 0.0828835 - 0.0689498 - 0.0561998 - 0.0448822 - 0.0351196 - 0.0269254 - 0.0202261 - 0.0148867 - 0.0107355 - 0.00758544 - 0.00525143 - 0.00356214 - 0.00236745 - 0.00154166 - 0.000983633 - 0.000614914 - 0.000376646 - 0.000226043 - 0.000132922 - 7.65918e-05 - 4.32595e-05 - 2.39797e-05 - 1.31129e-05 - 7.21922e-06 - 4.30366e-06 - 3.33996e-06 - 4.01773e-06 - 6.68526e-06 - 1.24923e-05 - 2.37754e-05 - 4.47731e-05 - 8.28123e-05 - 0.000150164 - 0.000266831 - 0.000464579 - 0.000792543 - 0.00132472 - 0.00216951 - 0.00348126 - 0.0054733 - 0.00843136 - 0.0127257 - 0.0188194 - 0.0272687 - 0.0387134 - 0.053851 - 0.0733945 - 0.09801 - 0.128237 - 0.164397 - 0.206496 - 0.254135 - 0.306448 - 0.362063 - 0.41913 - 0.47539 - 0.528309 - 0.575259 - 0.613727 - 0.641541 - 0.657068 - 0.659377 - 0.648327 - 0.624588 - 0.589567 - 0.545274 - 0.494136 - 0.438774 - 0.381789 - 0.325571 - 0.272154 - 0.223122 - 0.179578 - 0.142167 - 0.111134 - 0.086417 - 0.0677452 - 0.0547365 - 0.04698 - 0.0440953 - 0.0457651 - 0.0517401 - 0.0618192 - 0.075809 - 0.0934719 - 0.114469 - 0.138309 - 0.164311 - 0.191592 - 0.21908 - 0.245557 - 0.269731 - 0.29033 - 0.306206 - 0.316433 - 0.320401 - 0.317866 - 0.308982 - 0.29428 - 0.274615 - 0.251088 - 0.224938 - 0.197441 - 0.169805 - 0.143086 - 0.118136 - 0.0955664 - 0.0757468 - 0.0588247 - 0.0447603 - 0.0333705 - 0.0243763 - 0.0174466 - 0.0122347 - 0.00840639 - 0.00565932 - 0.00373302 - 0.0024127 - 0.001528 - 0.000948448 - 0.000577417 - 0.000345638 - 0.0002051 - 0.000123843 - 8.18518e-05 - 6.81308e-05 - 7.89389e-05 - 0.00011715 - 0.000192719 - 0.000324255 - 0.000541737 - 0.000890371 - 0.00143553 - 0.00226859 - 0.00351311 - 0.00533082 - 0.00792609 - 0.0115475 - 0.0164852 - 0.0230612 - 0.0316131 - 0.0424685 - 0.0559121 - 0.0721461 - 0.0912487 - 0.113135 - 0.137526 - 0.163937 - 0.191679 - 0.219893 - 0.247604 - 0.273794 - 0.297493 - 0.317866 - 0.334291 - 0.346416 - 0.354179 - 0.357793 - 0.357701 - 0.354498 - 0.348835 - 0.34133 - 0.332489 - 0.322651 - 0.311975 - 0.30045 - 0.287942 - 0.274258 - 0.25922 - 0.242722 - 0.224784 - 0.205571 - 0.185391 - 0.164673 - 0.143916 - 0.123646 - 0.104359 - 0.0864807 - 0.0703312 - 0.0561129 - 0.0439079 - 0.0336895 - 0.0253425 - 0.0186881 - 0.0135096 - 0.00957556 - 0.00665938 - 0.00455333 - 0.00307793 - 0.00208716 - 0.00147068 - 0.00115451 - 0.00110122 - 0.00131073 - 0.0018224 - 0.00271851 - 0.00412898 - 0.00623645 - 0.00928047 - 0.0135588 - 0.0194236 - 0.0272705 - 0.0375177 - 0.0505746 - 0.0667991 - 0.0864465 - 0.109613 - 0.13618 - 0.165769 - 0.19771 - 0.231042 - 0.26454 - 0.296775 - 0.326212 - 0.351326 - 0.370729 - 0.383301 - 0.388294 - 0.385406 - 0.374811 - 0.357144 - 0.333435 - 0.305011 - 0.273374 - 0.240069 - 0.206562 - 0.174142 - 0.143844 - 0.116417 - 0.0923168 - 0.0717266 - 0.054603 - 0.0407277 - 0.0297646 - 0.0213131 - 0.0149531 - 0.010279 - 0.00692323 - 0.00456881 - 0.00295416 - 0.00187155 - 0.00116173 - 0.000706559 - 0.000421043 - 0.000245834 - 0.000140635 - 7.88285e-05 - 4.32922e-05 - 2.32955e-05 - 1.22821e-05 - 6.34466e-06 - 3.2113e-06 - 1.59255e-06 - 7.73819e-07 - 3.68407e-07 - 1.71859e-07 - 7.85727e-08 - 3.525e-08 - 1.56255e-08 - 7.1029e-09 - 3.90724e-09 - 3.74228e-09 - 6.36823e-09 - 1.3367e-08 - 2.89202e-08 - 6.18851e-08 - 1.29968e-07 - 2.67519e-07 - 5.39551e-07 - 1.06623e-06 - 2.06446e-06 - 3.91651e-06 - 7.27996e-06 - 1.32585e-05 - 2.36592e-05 - 4.13656e-05 - 7.08625e-05 - 0.000118941 - 0.000195606 - 0.000315187 - 0.000497615 - 0.000769759 - 0.00116668 - 0.00173256 - 0.00252093 - 0.00359392 - 0.00502012 - 0.00687063 - 0.00921332 - 0.0121052 - 0.0155835 - 0.019656 - 0.0242919 - 0.0294147 - 0.0348984 - 0.0405679 - 0.0462058 - 0.0515641 - 0.0563813 - 0.0604032 - 0.0634047 - 0.0652108 - 0.0657135 - 0.0648824 - 0.0627676 - 0.0594951 - 0.055254 - 0.0502786 - 0.044827 - 0.0391591 - 0.0335168 - 0.028108 - 0.0230958 - 0.018594 - 0.0146673 - 0.0113361 - 0.00858452 - 0.00636947 - 0.0046305 - 0.0032983 - 0.00230191 - 0.00157407 - 0.00105462 - 0.000692316 - 0.000445298 - 0.000280629 - 0.000173282 - 0.000104836 - 6.21444e-05 - 3.60937e-05 - 2.05399e-05 - 1.14526e-05 - 6.25667e-06 - 3.34904e-06 - 1.75645e-06 - 9.02581e-07 - 4.5444e-07 - 2.24189e-07 - 1.08378e-07 - 5.13693e-08 - 2.39435e-08 - 1.11485e-08 - 5.59702e-09 - 3.92679e-09 - 5.09955e-09 - 9.86095e-09 - 2.114e-08 - 4.55587e-08 - 9.66698e-08 - 2.0116e-07 - 4.10205e-07 - 8.19618e-07 - 1.60458e-06 - 3.07786e-06 - 5.78459e-06 - 1.06521e-05 - 1.9219e-05 - 3.39755e-05 - 5.88488e-05 - 9.98725e-05 - 0.00016607 - 0.000270566 - 0.000431909 - 0.000675535 - 0.00103524 - 0.00155443 - 0.00228684 - 0.0032964 - 0.00465564 - 0.00644253 - 0.00873514 - 0.0116043 - 0.0151046 - 0.0192634 - 0.024071 - 0.0294708 - 0.035353 - 0.0415526 - 0.0478528 - 0.053995 - 0.059695 - 0.0646638 - 0.0686316 - 0.0713724 - 0.0727252 - 0.0726101 - 0.0710369 - 0.068105 - 0.0639941 - 0.058949 - 0.0532582 - 0.0472318 - 0.0411794 - 0.035392 - 0.0301294 - 0.0256128 - 0.0220232 - 0.0195037 - 0.0181646 - 0.0180873 - 0.0193273 - 0.0219132 - 0.025842 - 0.0310708 - 0.0375062 - 0.0449944 - 0.0533132 - 0.0621703 - 0.0712086 - 0.0800204 - 0.0881704 - 0.0952259 - 0.10079 - 0.104537 - 0.106239 - 0.105792 - 0.10322 - 0.0986778 - 0.0924299 - 0.0848289 - 0.0762804 - 0.0672077 - 0.058018 - 0.049073 - 0.0406687 - 0.0330228 - 0.0262727 - 0.0204801 - 0.0156422 - 0.0117058 - 0.00858298 - 0.00616616 - 0.00434041 - 0.00299359 - 0.00202311 - 0.00133989 - 0.000870016 - 0.000554601 - 0.00034854 - 0.000218741 - 0.000142232 - 0.000104541 - 9.8566e-05 - 0.000124155 - 0.000188521 - 0.000307637 - 0.000508725 - 0.000833915 - 0.00134509 - 0.00212979 - 0.00330782 - 0.00503791 - 0.00752352 - 0.0110162 - 0.0158148 - 0.0222592 - 0.0307153 - 0.0415515 - 0.0551058 - 0.071643 - 0.0913074 - 0.114073 - 0.1397 - 0.167701 - 0.197328 - 0.227588 - 0.257282 - 0.285075 - 0.309593 - 0.329531 - 0.34377 - 0.35148 - 0.352196 - 0.345872 - 0.332879 - 0.313973 - 0.29022 - 0.262897 - 0.233379 - 0.203026 - 0.173082 - 0.144596 - 0.118376 - 0.0949657 - 0.0746562 - 0.057512 - 0.0434158 - 0.0321176 - 0.0232853 - 0.0165482 - 0.0115343 - 0.00789618 - 0.00532826 - 0.00357577 - 0.00243735 - 0.00176332 - 0.00145159 - 0.00144239 - 0.00171284 - 0.00227175 - 0.00315462 - 0.00441846 - 0.0061359 - 0.00838817 - 0.0112563 - 0.0148106 - 0.019099 - 0.0241341 - 0.0298819 - 0.0362517 - 0.0430913 - 0.0501867 - 0.0572696 - 0.064032 - 0.0701466 - 0.0752928 - 0.0791839 - 0.0815939 - 0.0823787 - 0.0814908 - 0.0789841 - 0.075008 - 0.069793 - 0.0636288 - 0.0568371 - 0.0497447 - 0.0426579 - 0.0358416 - 0.0295062 - 0.0238 - 0.0188094 - 0.0145651 - 0.0110507 - 0.00821493 - 0.00598368 - 0.00427072 - 0.00298721 - 0.00204851 - 0.00137888 - 0.000914107 - 0.0006025 - 0.000404956 - 0.000294455 - 0.000255581 - 0.000284413 - 0.000389119 - 0.00059143 - 0.000929092 - 0.0014592 - 0.00226219 - 0.00344601 - 0.00514967 - 0.00754534 - 0.0108375 - 0.0152582 - 0.0210566 - 0.0284828 - 0.0377642 - 0.0490773 - 0.0625148 - 0.0780522 - 0.0955182 - 0.114573 - 0.134702 - 0.155224 - 0.175321 - 0.194088 - 0.210595 - 0.223966 - 0.233452 - 0.238502 - 0.238817 - 0.234375 - 0.225439 - 0.212528 - 0.196368 - 0.177823 - 0.157822 - 0.13728 - 0.117031 - 0.0977797 - 0.0800658 - 0.0642531 - 0.0505343 - 0.0389512 - 0.0294236 - 0.0217826 - 0.0158036 - 0.0112366 - 0.00782971 - 0.00534666 - 0.00357803 - 0.00234654 - 0.00150811 - 0.000949846 - 0.000586259 - 0.0003546 - 0.000210184 - 0.000122087 - 6.94932e-05 - 3.87633e-05 - 2.11886e-05 - 1.13497e-05 - 5.95747e-06 - 3.06434e-06 - 1.54456e-06 - 7.62898e-07 - 3.69248e-07 - 1.75129e-07 - 8.13929e-08 - 3.7068e-08 - 1.65423e-08 - 7.23393e-09 - 3.0998e-09 - 1.30158e-09 - 5.35535e-10 - 2.15914e-10 - 8.52996e-11 - 3.30208e-11 - 1.25257e-11 - 4.65589e-12 - 1.69628e-12 - 6.07133e-13 - 2.17868e-13 - 9.17968e-14 - 8.26531e-14 - 1.80758e-13 - 5.04118e-13 - 1.4277e-12 - 3.9783e-12 - 1.0868e-11 - 2.90954e-11 - 7.63304e-11 - 1.96231e-10 - 4.94351e-10 - 1.2204e-09 - 2.95232e-09 - 6.9988e-09 - 1.62586e-08 - 3.70116e-08 - 8.25642e-08 - 1.80486e-07 - 3.86628e-07 - 8.11596e-07 - 1.66949e-06 - 3.36533e-06 - 6.64764e-06 - 1.28678e-05 - 2.44085e-05 - 4.53704e-05 - 8.26422e-05 - 0.000147512 - 0.000258019 - 0.000442253 - 0.000742825 - 0.00122264 - 0.001972 - 0.00311681 - 0.00482735 - 0.00732661 - 0.0108966 - 0.0158809 - 0.0226805 - 0.0317412 - 0.04353 - 0.0584987 - 0.0770365 - 0.0994122 - 0.125711 - 0.155777 - 0.189156 - 0.225077 - 0.262443 - 0.299868 - 0.335752 - 0.368386 - 0.396083 - 0.417326 - 0.430901 - 0.436023 - 0.432407 - 0.420307 - 0.400488 - 0.374157 - 0.34285 - 0.308297 - 0.27227 - 0.236449 - 0.202305 - 0.171016 - 0.143419 - 0.12 - 0.10091 - 0.0860138 - 0.0749435 - 0.0671671 - 0.0620535 - 0.0589346 - 0.0571569 - 0.0561247 - 0.0553308 - 0.0543762 - 0.052979 - 0.0509737 - 0.0483016 - 0.0449951 - 0.0411568 - 0.0369364 - 0.0325078 - 0.0280481 - 0.0237199 - 0.0196589 - 0.0159663 - 0.0127065 - 0.00990863 - 0.00757099 - 0.00566812 - 0.00415785 - 0.00298841 - 0.00210451 - 0.00145211 - 0.000981719 - 0.000650297 - 0.00042206 - 0.000268394 - 0.000167228 - 0.00010209 - 6.1065e-05 - 3.57882e-05 - 2.05505e-05 - 1.15623e-05 - 6.37384e-06 - 3.44267e-06 - 1.82192e-06 - 9.44718e-07 - 4.79996e-07 - 2.39019e-07 - 1.16785e-07 - 5.63189e-08 - 2.75876e-08 - 1.55066e-08 - 1.35065e-08 - 2.06346e-08 - 4.12348e-08 - 8.71701e-08 - 1.83491e-07 - 3.79656e-07 - 7.70155e-07 - 1.53093e-06 - 2.98182e-06 - 5.69044e-06 - 1.06401e-05 - 1.94933e-05 - 3.49914e-05 - 6.15423e-05 - 0.000106053 - 0.000179064 - 0.000296231 - 0.000480165 - 0.000762583 - 0.00118664 - 0.00180922 - 0.0027027 - 0.00395586 - 0.00567312 - 0.00797149 - 0.0109747 - 0.0148042 - 0.0195665 - 0.0253384 - 0.03215 - 0.0399688 - 0.0486854 - 0.0581051 - 0.0679468 - 0.0778511 - 0.0873986 - 0.0961378 - 0.10362 - 0.109438 - 0.113265 - 0.11489 - 0.114237 - 0.111386 - 0.106565 - 0.100144 - 0.092608 - 0.0845337 - 0.0765571 - 0.069347 - 0.0635806 - 0.0599254 - 0.0590243 - 0.0614823 - 0.0678499 - 0.0786025 - 0.0941094 - 0.114595 - 0.140095 - 0.170403 - 0.205033 - 0.243187 - 0.28375 - 0.325307 - 0.366205 - 0.404639 - 0.438768 - 0.466855 - 0.487399 - 0.499265 - 0.501787 - 0.494819 - 0.478757 - 0.454493 - 0.423343 - 0.386921 - 0.347013 - 0.305437 - 0.263913 - 0.223964 - 0.186845 - 0.15351 - 0.124609 - 0.100516 - 0.0813752 - 0.0671528 - 0.0576947 - 0.0527714 - 0.0521107 - 0.0554152 - 0.0623635 - 0.0726003 - 0.0857196 - 0.101248 - 0.118632 - 0.137242 - 0.156382 - 0.175324 - 0.193347 - 0.209787 - 0.224084 - 0.235824 - 0.244759 - 0.250819 - 0.25409 - 0.254781 - 0.253179 - 0.249598 - 0.24433 - 0.237611 - 0.229608 - 0.220414 - 0.210075 - 0.198611 - 0.186058 - 0.172493 - 0.15806 - 0.142974 - 0.127518 - 0.112024 - 0.0968447 - 0.0823232 - 0.0687639 - 0.056409 - 0.0454241 - 0.0358933 - 0.0278226 - 0.0211511 - 0.0157665 - 0.0115222 - 0.00825423 - 0.0057959 - 0.0039887 - 0.00269018 - 0.00177808 - 0.00115167 - 0.000731004 - 0.000454751 - 0.000277398 - 0.000166222 - 9.846e-05 - 5.88987e-05 - 3.79727e-05 - 3.04441e-05 - 3.46788e-05 - 5.25582e-05 - 9.00977e-05 - 0.000158898 - 0.000278618 - 0.000480699 - 0.000813605 - 0.00134977 - 0.00219438 - 0.00349573 - 0.00545668 - 0.00834609 - 0.0125084 - 0.0183688 - 0.0264315 - 0.0372673 - 0.0514867 - 0.0696987 - 0.0924523 - 0.120163 - 0.153035 - 0.190972 - 0.233514 - 0.27978 - 0.328461 - 0.377845 - 0.425897 - 0.470391 - 0.509067 - 0.539825 - 0.56091 - 0.571077 - 0.569717 - 0.55691 - 0.533424 - 0.500637 - 0.460399 - 0.414867 - 0.366308 - 0.31692 - 0.268674 - 0.223194 - 0.181698 - 0.144974 - 0.113407 - 0.0870373 - 0.0656418 - 0.048819 - 0.0360773 - 0.0269108 - 0.0208626 - 0.0175701 - 0.0167947 - 0.0184339 - 0.0225191 - 0.0291988 - 0.0387082 - 0.0513264 - 0.067323 - 0.0868961 - 0.110107 - 0.136818 - 0.166639 - 0.198893 - 0.232612 - 0.266561 - 0.299299 - 0.329271 - 0.354927 - 0.374855 - 0.387903 - 0.393297 - 0.390711 - 0.3803 - 0.36269 - 0.338907 - 0.310288 - 0.278347 - 0.244652 - 0.210696 - 0.177796 - 0.147015 - 0.119132 - 0.0946281 - 0.0737165 - 0.0563825 - 0.0424423 - 0.0316028 - 0.0235195 - 0.0178455 - 0.0142692 - 0.0125399 - 0.0124803 - 0.0139879 - 0.0170253 - 0.0216027 - 0.0277537 - 0.0355053 - 0.0448476 - 0.0557031 - 0.0679011 - 0.0811603 - 0.0950824 - 0.10916 - 0.1228 - 0.135357 - 0.146187 - 0.154696 - 0.160393 - 0.162941 - 0.162186 - 0.158174 - 0.151147 - 0.141516 - 0.129827 - 0.116706 - 0.102807 - 0.0887582 - 0.0751238 - 0.062369 - 0.0508463 - 0.0407927 - 0.0323401 - 0.0255341 - 0.0203573 - 0.0167544 - 0.0146531 - 0.0139808 - 0.0146745 - 0.0166817 - 0.019956 - 0.0244455 - 0.0300777 - 0.0367437 - 0.0442835 - 0.0524759 - 0.0610361 - 0.0696209 - 0.0778441 - 0.0852996 - 0.0915913 - 0.096366 - 0.0993442 - 0.100347 - 0.0993131 - 0.0963045 - 0.0915008 - 0.0851805 - 0.0776949 - 0.0694355 - 0.0608006 - 0.052164 - 0.0438501 - 0.0361166 - 0.0291461 - 0.0230458 - 0.0178541 - 0.0135526 - 0.0100796 - 0.00734517 - 0.00524441 - 0.00366883 - 0.00251476 - 0.00168889 - 0.00111133 - 0.00071651 - 0.000452625 - 0.00028015 - 0.000169895 - 0.00010095 - 5.87719e-05 - 3.35251e-05 - 1.87373e-05 - 1.02608e-05 - 5.50542e-06 - 2.89426e-06 - 1.49081e-06 - 7.52391e-07 - 3.7205e-07 - 1.80259e-07 - 8.55712e-08 - 3.98012e-08 - 1.81385e-08 - 8.09923e-09 - 3.54342e-09 - 1.51893e-09 - 6.37957e-10 - 2.62531e-10 - 1.05854e-10 - 4.18187e-11 - 1.61872e-11 - 6.13914e-12 - 2.28129e-12 - 8.306e-13 - 2.96305e-13 - 1.03568e-13 - 3.54687e-14 - 1.19015e-14 - 3.91289e-15 - 1.26046e-15 - 3.9783e-16 - 1.23028e-16 - 3.72774e-17 - 1.10669e-17 - 3.21915e-18 - 9.17473e-19 - 2.56202e-19 - 7.00986e-20 - 1.8792e-20 - 4.93598e-21 - 1.27031e-21 - 3.2032e-22 - 7.91398e-23 - 1.91577e-23 - 4.54388e-24 - 1.05596e-24 - 2.4044e-25 - 5.36415e-26 - 1.17255e-26 - 2.51132e-27 - 5.26996e-28 - 1.08355e-28 - 2.18287e-29 - 4.30869e-30 - 8.33293e-31 - 1.57902e-31 - 2.93166e-32 - 5.33307e-33 - 9.50556e-34 - 1.66003e-34 - 2.84046e-35 - 4.76211e-36 - 7.82253e-37 - 1.25902e-37 - 1.98542e-38 - 3.06769e-39 - 4.64415e-40 - 6.88872e-41 - 1.00117e-41 - 1.42565e-42 - 1.98909e-43 - 2.71916e-44 - 3.64209e-45 - 4.77973e-46 - 6.14602e-47 - 7.7432e-48 - 9.55837e-49 - 1.15607e-49 - 1.37e-50 - 1.59073e-51 - 1.8097e-52 - 2.01722e-53 - 2.20312e-54 - 2.35755e-55 - 2.47183e-56 - 2.5393e-57 - 2.55591e-58 - 2.52066e-59 - 2.43568e-60 - 2.30602e-61 - 2.13916e-62 - 1.94428e-63 - 1.73146e-64 - 1.51078e-65 - 1.2916e-66 - 1.08191e-67 - 8.87958e-69 - 7.14052e-70 - 5.62606e-71 - 4.34325e-72 - 3.28521e-73 - 2.43471e-74 - 1.76795e-75 - 1.25785e-76 - 8.76845e-78 - 5.98901e-79 - 4.00797e-80 - 2.62803e-81 - 1.68839e-82 - 1.0628e-83 - 6.55493e-85 - 3.96115e-86 - 2.34537e-87 - 1.36063e-88 - 7.73397e-90 - 4.30728e-91 - 2.35039e-92 - 1.25665e-93 - 6.58303e-95 - 3.37889e-96 - 1.69925e-97 - 8.37298e-99 - 4.04239e-100 - 1.9122e-101 - 8.8627e-103 - 4.02472e-104 - 1.79078e-105 - 7.807e-107 - 3.33476e-108 - 1.39764e-109 - 1.06571e-110 - 1.20963e-109 - 2.89465e-108 - 6.79983e-107 - 1.56509e-105 - 3.52952e-104 - 7.79884e-103 - 1.68842e-101 - 3.58153e-100 - 7.44378e-99 - 1.51585e-97 - 3.0245e-96 - 5.91274e-95 - 1.13256e-93 - 2.12554e-92 - 3.90855e-91 - 7.04204e-90 - 1.24313e-88 - 2.15018e-87 - 3.64391e-86 - 6.05059e-85 - 9.84384e-84 - 1.56916e-82 - 2.45081e-81 - 3.75048e-80 - 5.62342e-79 - 8.26136e-78 - 1.18916e-76 - 1.67712e-75 - 2.31753e-74 - 3.1378e-73 - 4.16256e-72 - 5.41044e-71 - 6.89035e-70 - 8.5978e-69 - 1.05116e-67 - 1.25919e-66 - 1.4779e-65 - 1.69957e-64 - 1.915e-63 - 2.11415e-62 - 2.28686e-61 - 2.42371e-60 - 2.51686e-59 - 2.56079e-58 - 2.55284e-57 - 2.49352e-56 - 2.38637e-55 - 2.23769e-54 - 2.05588e-53 - 1.85069e-52 - 1.63232e-51 - 1.41064e-50 - 1.19443e-49 - 9.90932e-49 - 8.05496e-48 - 6.41535e-47 - 5.00626e-46 - 3.82775e-45 - 2.86755e-44 - 2.10482e-43 - 1.51376e-42 - 1.06668e-41 - 7.36458e-41 - 4.98195e-40 - 3.30208e-39 - 2.14443e-38 - 1.3645e-37 - 8.50695e-37 - 5.19648e-36 - 3.11015e-35 - 1.82386e-34 - 1.04794e-33 - 5.89958e-33 - 3.25418e-32 - 1.75873e-31 - 9.31305e-31 - 4.83195e-30 - 2.45635e-29 - 1.22347e-28 - 5.97085e-28 - 2.85506e-27 - 1.33761e-26 - 6.1402e-26 - 2.76167e-25 - 1.21702e-24 - 5.25487e-24 - 2.22312e-23 - 9.21509e-23 - 3.74261e-22 - 1.48931e-21 - 5.80677e-21 - 2.2183e-20 - 8.30318e-20 - 3.04513e-19 - 1.09422e-18 - 3.85247e-18 - 1.32896e-17 - 4.49183e-17 - 1.48755e-16 - 4.82679e-16 - 1.53456e-15 - 4.7802e-15 - 1.45897e-14 - 4.36302e-14 - 1.2784e-13 - 3.67016e-13 - 1.03239e-12 - 2.84538e-12 - 7.68384e-12 - 2.03309e-11 - 5.27081e-11 - 1.33887e-10 - 3.33231e-10 - 8.12632e-10 - 1.94172e-09 - 4.54597e-09 - 1.04283e-08 - 2.34394e-08 - 5.16215e-08 - 1.11395e-07 - 2.35536e-07 - 4.87981e-07 - 9.90622e-07 - 1.97049e-06 - 3.84067e-06 - 7.33515e-06 - 1.37273e-05 - 2.51733e-05 - 4.52355e-05 - 7.96547e-05 - 0.000137449 - 0.000232424 - 0.000385157 - 0.000625493 - 0.000995518 - 0.00155285 - 0.00237402 - 0.00355735 - 0.00522491 - 0.00752254 - 0.0106172 - 0.0146907 - 0.0199294 - 0.0265096 - 0.0345787 - 0.0442341 - 0.0555004 - 0.0683094 - 0.0824842 - 0.0977307 - 0.11364 - 0.129703 - 0.145333 - 0.159905 - 0.172795 - 0.183427 - 0.191318 - 0.19611 - 0.197601 - 0.195756 - 0.190705 - 0.18273 - 0.172236 - 0.15972 - 0.145736 - 0.130848 - 0.115605 - 0.100507 - 0.0859807 - 0.0723703 - 0.0599273 - 0.0488129 - 0.0391035 - - - - - 1.07039e-30 - 1.33207e-30 - 1.62424e-30 - 1.94049e-30 - 2.27148e-30 - 2.60527e-30 - 2.92797e-30 - 3.22546e-30 - 3.48836e-30 - 3.73317e-30 - 4.10347e-30 - 5.36414e-30 - 1.12841e-29 - 3.99026e-29 - 1.75698e-28 - 8.0611e-28 - 3.67141e-27 - 1.64271e-26 - 7.20535e-26 - 3.09694e-25 - 1.30424e-24 - 5.38172e-24 - 2.17581e-23 - 8.61905e-23 - 3.34529e-22 - 1.27217e-21 - 4.74016e-21 - 1.73052e-20 - 6.19012e-20 - 2.16949e-19 - 7.44995e-19 - 2.5066e-18 - 8.26332e-18 - 2.66907e-17 - 8.44702e-17 - 2.61929e-16 - 7.95792e-16 - 2.36893e-15 - 6.90944e-15 - 1.97456e-14 - 5.52884e-14 - 1.51682e-13 - 4.0773e-13 - 1.07386e-12 - 2.77114e-12 - 7.00658e-12 - 1.73577e-11 - 4.21321e-11 - 1.00201e-10 - 2.33489e-10 - 5.33089e-10 - 1.19253e-09 - 2.61381e-09 - 5.61329e-09 - 1.18113e-08 - 2.43508e-08 - 4.91889e-08 - 9.73549e-08 - 1.88793e-07 - 3.58714e-07 - 6.67805e-07 - 1.21811e-06 - 2.17702e-06 - 3.81218e-06 - 6.54066e-06 - 1.09953e-05 - 1.81104e-05 - 2.92272e-05 - 4.6215e-05 - 7.16003e-05 - 0.000108689 - 0.000161656 - 0.000235577 - 0.000336367 - 0.000470576 - 0.000645036 - 0.000866312 - 0.00113999 - 0.00146982 - 0.00185681 - 0.00229829 - 0.00278727 - 0.00331201 - 0.00385604 - 0.00439875 - 0.00491649 - 0.0053842 - 0.00577737 - 0.00607415 - 0.00625743 - 0.00631647 - 0.00624807 - 0.00605694 - 0.00575536 - 0.00536215 - 0.00490107 - 0.0043989 - 0.00388353 - 0.00338212 - 0.00291963 - 0.00251784 - 0.00219476 - 0.0019644 - 0.00183683 - 0.00181827 - 0.00191123 - 0.00211441 - 0.0024225 - 0.0028259 - 0.00331037 - 0.00385693 - 0.00444204 - 0.00503828 - 0.00561553 - 0.0061427 - 0.00658983 - 0.00693041 - 0.0071435 - 0.00721563 - 0.00714193 - 0.00692656 - 0.00658218 - 0.00612867 - 0.00559119 - 0.00499782 - 0.0043772 - 0.0037562 - 0.00315819 - 0.00260175 - 0.00210005 - 0.00166085 - 0.00128697 - 0.000977109 - 0.000726866 - 0.000529789 - 0.000378345 - 0.000264734 - 0.000181497 - 0.000121917 - 8.02412e-05 - 5.17448e-05 - 3.26944e-05 - 2.02403e-05 - 1.22771e-05 - 7.29648e-06 - 4.24881e-06 - 2.42414e-06 - 1.35514e-06 - 7.42249e-07 - 3.98337e-07 - 2.09454e-07 - 1.0791e-07 - 5.44722e-08 - 2.69416e-08 - 1.3056e-08 - 6.19913e-09 - 2.88397e-09 - 1.31458e-09 - 5.87109e-10 - 2.56914e-10 - 1.10153e-10 - 4.62739e-11 - 1.90459e-11 - 7.67917e-12 - 3.02905e-12 - 1.15766e-12 - 3.97361e-13 - 3.41409e-14 - -2.89899e-13 - -8.92159e-13 - -2.32794e-12 - -5.84648e-12 - -1.43519e-11 - -3.45078e-11 - -8.12912e-11 - -1.87631e-10 - -4.24327e-10 - -9.40231e-10 - -2.04129e-09 - -4.34222e-09 - -9.05016e-09 - -1.84815e-08 - -3.6979e-08 - -7.24953e-08 - -1.39252e-07 - -2.62077e-07 - -4.83275e-07 - -8.73166e-07 - -1.54574e-06 - -2.68109e-06 - -4.55642e-06 - -7.58706e-06 - -1.23783e-05 - -1.97871e-05 - -3.09915e-05 - -4.75597e-05 - -7.15111e-05 - -0.000105352 - -0.000152073 - -0.000215078 - -0.000298041 - -0.000404664 - -0.00053833 - -0.000701682 - -0.000896126 - -0.00112133 - -0.00137479 - -0.00165149 - -0.0019438 - -0.00224163 - -0.00253287 - -0.00280414 - -0.00304174 - -0.00323283 - -0.0033665 - -0.00343488 - -0.00343384 - -0.00336344 - -0.0032279 - -0.00303518 - -0.00279621 - -0.00252384 - -0.00223166 - -0.00193291 - -0.00163942 - -0.00136091 - -0.00110451 - -0.000874518 - -0.000672571 - -0.00049787 - -0.000347626 - -0.000217553 - -0.000102387 - 3.62246e-06 - 0.000106247 - 0.000210868 - 0.000322098 - 0.000443445 - 0.000577031 - 0.000723357 - 0.000881165 - 0.0010474 - 0.00121734 - 0.00138482 - 0.0015427 - 0.00168342 - 0.00179962 - 0.00188484 - 0.00193416 - 0.00194463 - 0.00191564 - 0.00184894 - 0.00174849 - 0.00162005 - 0.00147064 - 0.00130789 - 0.00113938 - 0.000972049 - 0.000811701 - 0.000662676 - 0.000527676 - 0.000407728 - 0.000302261 - 0.000209254 - 0.000125426 - 4.64271e-05 - -3.29605e-05 - -0.000118629 - -0.000216928 - -0.0003344 - -0.000477448 - -0.000651899 - -0.000862487 - -0.00111225 - -0.00140188 - -0.00172921 - -0.00208869 - -0.00247127 - -0.00286449 - -0.003253 - -0.00361946 - -0.0039458 - -0.00421463 - -0.00441082 - -0.00452289 - -0.00454411 - -0.00447319 - -0.00431444 - -0.00407725 - -0.00377526 - -0.00342502 - -0.00304451 - -0.0026516 - -0.00226274 - -0.0018919 - -0.00154988 - -0.00124405 - -0.000978389 - -0.000753916 - -0.000569208 - -0.000421072 - -0.000305196 - -0.000216739 - -0.000150811 - -0.000102817 - -6.86807e-05 - -4.49512e-05 - -2.8826e-05 - -1.81119e-05 - -1.11501e-05 - -6.72565e-06 - -3.97489e-06 - -2.30172e-06 - -1.30592e-06 - -7.25967e-07 - -3.95417e-07 - -2.11023e-07 - -1.10342e-07 - -5.65314e-08 - -2.83775e-08 - -1.39572e-08 - -6.72598e-09 - -3.17579e-09 - -1.46921e-09 - -6.65968e-10 - -2.95774e-10 - -1.28707e-10 - -5.48761e-11 - -2.29245e-11 - -9.38325e-12 - -3.76308e-12 - -1.47867e-12 - -5.69293e-13 - -2.14752e-13 - -7.93733e-14 - -2.87441e-14 - -1.01991e-14 - -3.54576e-15 - -1.2078e-15 - -4.03104e-16 - -1.31818e-16 - -4.22349e-17 - -1.32588e-17 - -4.07825e-18 - -1.22908e-18 - -3.62931e-19 - -1.05004e-19 - -2.9766e-20 - -8.26751e-21 - -2.24991e-21 - -5.99919e-22 - -1.56732e-22 - -4.01224e-23 - -1.00739e-23 - -2.52276e-24 - -8.0687e-25 - -1.016e-24 - -3.58361e-24 - -1.41196e-23 - -5.4951e-23 - -2.09636e-22 - -7.83616e-22 - -2.86998e-21 - -1.02989e-20 - -3.62111e-20 - -1.24746e-19 - -4.21067e-19 - -1.39255e-18 - -4.5124e-18 - -1.43265e-17 - -4.45669e-17 - -1.35837e-16 - -4.05661e-16 - -1.18698e-15 - -3.40301e-15 - -9.55911e-15 - -2.63093e-14 - -7.09476e-14 - -1.87458e-13 - -4.85294e-13 - -1.23096e-12 - -3.05929e-12 - -7.4496e-12 - -1.77739e-11 - -4.15498e-11 - -9.51682e-11 - -2.13576e-10 - -4.69623e-10 - -1.01177e-09 - -2.13577e-09 - -4.41734e-09 - -8.9517e-09 - -1.77741e-08 - -3.45784e-08 - -6.59113e-08 - -1.23098e-07 - -2.25258e-07 - -4.03874e-07 - -7.09493e-07 - -1.2212e-06 - -2.05951e-06 - -3.40311e-06 - -5.50968e-06 - -8.74004e-06 - -1.35843e-05 - -2.06871e-05 - -3.08674e-05 - -4.51271e-05 - -6.46419e-05 - -9.07258e-05 - -0.000124764 - -0.00016811 - -0.000221944 - -0.000287112 - -0.000363934 - -0.000452037 - -0.000550211 - -0.000656332 - -0.000767384 - -0.000879594 - -0.000988694 - -0.00109031 - -0.00118047 - -0.00125612 - -0.00131572 - -0.00135978 - -0.00139119 - -0.00141553 - -0.00144105 - -0.00147842 - -0.00154024 - -0.00164022 - -0.00179215 - -0.0020087 - -0.00229996 - -0.00267211 - -0.00312609 - -0.00365653 - -0.00425123 - -0.00489099 - -0.00555038 - -0.00619899 - -0.00680353 - -0.00733032 - -0.00774812 - -0.00803088 - -0.00816018 - -0.00812687 - -0.00793199 - -0.0075865 - -0.00711016 - -0.00652951 - -0.00587539 - -0.00518015 - -0.00447502 - -0.00378787 - -0.0031416 - -0.00255317 - -0.00203343 - -0.00158747 - -0.0012155 - -0.000914005 - -0.000677014 - -0.000497327 - -0.000367638 - -0.00028146 - -0.000233825 - -0.000221778 - -0.000244658 - -0.000304193 - -0.000404393 - -0.000551247 - -0.0007522 - -0.00101539 - -0.00134866 - -0.00175834 - -0.00224794 - -0.00281677 - -0.00345876 - -0.00416153 - -0.00490608 - -0.00566706 - -0.00641387 - -0.00711248 - -0.00772785 - -0.00822684 - -0.00858114 - -0.00876987 - -0.0087817 - -0.0086159 - -0.00828247 - -0.0078011 - -0.00719927 - -0.00650966 - -0.00576719 - -0.0050062 - -0.00425783 - -0.00354817 - -0.00289707 - -0.00231766 - -0.00181667 - -0.00139522 - -0.00104989 - -0.000774071 - -0.000559185 - -0.000395792 - -0.000274483 - -0.00018651 - -0.000124172 - -8.09996e-05 - -5.17701e-05 - -3.24199e-05 - -1.98921e-05 - -1.19588e-05 - -7.04419e-06 - -4.06548e-06 - -2.29895e-06 - -1.27376e-06 - -6.91503e-07 - -3.67873e-07 - -1.91872e-07 - -9.83377e-08 - -5.00377e-08 - -2.64246e-08 - -1.68924e-08 - -1.70683e-08 - -2.72651e-08 - -5.27365e-08 - -1.05874e-07 - -2.1103e-07 - -4.13329e-07 - -7.93705e-07 - -1.49355e-06 - -2.75377e-06 - -4.97481e-06 - -8.80568e-06 - -1.52717e-05 - -2.59505e-05 - -4.32059e-05 - -7.04818e-05 - -0.000112654 - -0.000176423 - -0.000270707 - -0.000406987 - -0.000599513 - -0.000865274 - -0.00122362 - -0.00169541 - -0.00230165 - -0.00306155 - -0.00399008 - -0.00509515 - -0.00637485 - -0.00781483 - -0.00938655 - -0.0110466 - -0.0127377 - -0.0143909 - -0.0159302 - -0.0172779 - -0.0183611 - -0.0191181 - -0.0195041 - -0.0194959 - -0.0190941 - -0.0183227 - -0.0172274 - -0.0158703 - -0.0143247 - -0.0126685 - -0.0109775 - -0.00931998 - -0.00775291 - -0.00631905 - -0.00504632 - -0.00394853 - -0.00302714 - -0.00227387 - -0.00167355 - -0.00120683 - -0.000852687 - -0.000590298 - -0.000400396 - -0.0002661 - -0.000173275 - -0.000110552 - -6.91084e-05 - -4.23286e-05 - -2.54023e-05 - -1.49365e-05 - -8.60522e-06 - -4.85749e-06 - -2.68657e-06 - -1.45587e-06 - -7.73006e-07 - -4.02142e-07 - -2.04981e-07 - -1.02373e-07 - -5.00948e-08 - -2.4018e-08 - -1.12829e-08 - -5.19323e-09 - -2.34203e-09 - -1.03487e-09 - -4.48039e-10 - -1.90063e-10 - -7.90133e-11 - -3.2227e-11 - -1.2995e-11 - -5.44122e-12 - -3.02022e-12 - -3.51191e-12 - -7.21295e-12 - -1.70439e-11 - -4.05427e-11 - -9.4878e-11 - -2.17679e-10 - -4.89379e-10 - -1.07799e-09 - -2.32661e-09 - -4.92003e-09 - -1.01941e-08 - -2.06951e-08 - -4.11646e-08 - -8.02261e-08 - -1.53195e-07 - -2.86623e-07 - -5.25429e-07 - -9.43742e-07 - -1.66085e-06 - -2.8638e-06 - -4.83831e-06 - -8.00904e-06 - -1.29899e-05 - -2.06426e-05 - -3.21413e-05 - -4.9034e-05 - -7.3294e-05 - -0.000107344 - -0.000154036 - -0.000216573 - -0.000298347 - -0.000402696 - -0.000532562 - -0.00069008 - -0.000876123 - -0.00108985 - -0.00132834 - -0.0015863 - -0.00185609 - -0.00212789 - -0.00239021 - -0.00263063 - -0.00283674 - -0.00299722 - -0.00310279 - -0.0031472 - -0.00312775 - -0.00304563 - -0.00290576 - -0.0027163 - -0.0024879 - -0.00223268 - -0.00196316 - -0.0016913 - -0.00142766 - -0.00118077 - -0.000956844 - -0.000759722 - -0.000591024 - -0.000450497 - -0.000336447 - -0.000246194 - -0.000176512 - -0.000123997 - -8.53456e-05 - -5.75558e-05 - -3.80307e-05 - -2.46216e-05 - -1.56184e-05 - -9.70714e-06 - -5.91131e-06 - -3.52706e-06 - -2.06196e-06 - -1.18109e-06 - -6.62861e-07 - -3.64501e-07 - -1.96387e-07 - -1.03672e-07 - -5.36229e-08 - -2.71753e-08 - -1.34939e-08 - -6.56499e-09 - -3.12945e-09 - -1.46164e-09 - -6.6888e-10 - -2.99912e-10 - -1.31758e-10 - -5.67146e-11 - -2.39194e-11 - -9.88424e-12 - -4.00196e-12 - -1.58759e-12 - -6.17081e-13 - -2.35008e-13 - -8.76919e-14 - -3.20607e-14 - -1.14848e-14 - -4.03098e-15 - -1.38623e-15 - -4.67086e-16 - -1.54204e-16 - -4.98803e-17 - -1.58089e-17 - -4.90919e-18 - -1.49367e-18 - -4.45284e-19 - -1.30064e-19 - -3.72232e-20 - -1.04383e-20 - -2.87015e-21 - -7.81928e-22 - -2.43429e-22 - -2.08622e-22 - -6.12204e-22 - -2.2559e-21 - -8.30244e-21 - -2.99767e-20 - -1.06056e-19 - -3.67643e-19 - -1.24869e-18 - -4.15545e-18 - -1.35494e-17 - -4.32871e-17 - -1.35498e-16 - -4.15571e-16 - -1.2488e-15 - -3.67688e-15 - -1.06072e-14 - -2.99821e-14 - -8.30344e-14 - -2.25315e-13 - -5.99047e-13 - -1.56051e-12 - -3.98301e-12 - -9.96075e-12 - -2.44067e-11 - -5.85955e-11 - -1.37834e-10 - -3.17675e-10 - -7.1738e-10 - -1.58727e-09 - -3.44105e-09 - -7.30916e-09 - -1.52118e-08 - -3.10193e-08 - -6.19757e-08 - -1.21324e-07 - -2.32709e-07 - -4.37337e-07 - -8.05303e-07 - -1.45292e-06 - -2.56841e-06 - -4.44867e-06 - -7.54986e-06 - -1.25543e-05 - -2.04549e-05 - -3.26554e-05 - -5.10821e-05 - -7.82973e-05 - -0.000117598 - -0.000173075 - -0.000249615 - -0.000352798 - -0.000488684 - -0.000663453 - -0.000882913 - -0.00115189 - -0.00147356 - -0.00184879 - -0.00227563 - -0.00274905 - -0.00326099 - -0.00380083 - -0.00435636 - -0.00491499 - -0.00546527 - -0.00599832 - -0.00650907 - -0.00699689 - -0.00746558 - -0.00792243 - -0.00837658 - -0.00883672 - -0.00930857 - -0.00979246 - -0.0102816 - -0.0107611 - -0.0112087 - -0.011596 - -0.0118916 - -0.0120645 - -0.0120879 - -0.0119427 - -0.0116203 - -0.011124 - -0.0104692 - -0.00968215 - -0.00879775 - -0.00785624 - -0.00689989 - -0.00596968 - -0.00510249 - -0.00432903 - -0.00367258 - -0.00314848 - -0.00276429 - -0.00252032 - -0.00241049 - -0.00242322 - -0.00254229 - -0.00274769 - -0.00301656 - -0.00332402 - -0.00364428 - -0.00395183 - -0.00422275 - -0.00443609 - -0.00457514 - -0.00462847 - -0.0045906 - -0.00446213 - -0.00424935 - -0.00396339 - -0.0036188 - -0.00323204 - -0.00281971 - -0.00239708 - -0.0019767 - -0.00156754 - -0.00117457 - -0.000798753 - -0.000437587 - -8.59164e-05 - 0.000262944 - 0.000615954 - 0.000979073 - 0.00135609 - 0.00174768 - 0.00215078 - 0.00255837 - 0.00295977 - 0.00334128 - 0.00368735 - 0.00398195 - 0.00421019 - 0.00435982 - 0.00442255 - 0.00439499 - 0.00427906 - 0.00408186 - 0.003815 - 0.00349351 - 0.00313447 - 0.00275551 - 0.00237344 - 0.00200306 - 0.00165638 - 0.00134211 - 0.00106565 - 0.000829314 - 0.000632838 - 0.00047399 - 0.000349261 - 0.00025451 - 0.000185531 - 0.00013851 - 0.000110364 - 9.89575e-05 - 0.000103216 - 0.000123152 - 0.000159795 - 0.000215048 - 0.000291453 - 0.000391873 - 0.000519088 - 0.000675318 - 0.000861708 - 0.00107781 - 0.00132113 - 0.00158679 - 0.00186744 - 0.00215336 - 0.00243292 - 0.00269324 - 0.0029212 - 0.00310445 - 0.00323254 - 0.00329792 - 0.00329666 - 0.00322882 - 0.0030985 - 0.00291337 - 0.00268396 - 0.00242267 - 0.00214264 - 0.00185669 - 0.00157641 - 0.0013114 - 0.0010689 - 0.00085364 - 0.00066796 - 0.00051211 - 0.000384691 - 0.000283138 - 0.000204184 - 0.000144272 - 9.98799e-05 - 6.77504e-05 - 4.5028e-05 - 2.93217e-05 - 1.87083e-05 - 1.16954e-05 - 7.16366e-06 - 4.29928e-06 - 2.52821e-06 - 1.45697e-06 - 8.23301e-07 - 4.57272e-07 - 2.52047e-07 - 1.4307e-07 - 9.41611e-08 - 8.92735e-08 - 1.29198e-07 - 2.33164e-07 - 4.46224e-07 - 8.54461e-07 - 1.61153e-06 - 2.98178e-06 - 5.40736e-06 - 9.60873e-06 - 1.67298e-05 - 2.854e-05 - 4.77041e-05 - 7.81257e-05 - 0.000125363 - 0.000197097 - 0.000303619 - 0.000458263 - 0.0006777 - 0.000981967 - 0.0013941 - 0.00193921 - 0.00264299 - 0.00352941 - 0.00461791 - 0.00592006 - 0.00743607 - 0.0091516 - 0.0110354 - 0.0130381 - 0.0150932 - 0.0171191 - 0.0190248 - 0.0207155 - 0.0221008 - 0.0231024 - 0.0236615 - 0.0237446 - 0.0233467 - 0.0224919 - 0.0212307 - 0.0196357 - 0.0177941 - 0.0158004 - 0.0137482 - 0.0117235 - 0.00979954 - 0.00803325 - 0.00646427 - 0.00511562 - 0.00399594 - 0.00310269 - 0.00242575 - 0.00195092 - 0.00166287 - 0.00154732 - 0.00159223 - 0.00178801 - 0.00212689 - 0.00260153 - 0.00320325 - 0.00392002 - 0.00473471 - 0.00562383 - 0.006557 - 0.00749741 - 0.00840331 - 0.00923048 - 0.00993535 - 0.0104786 - 0.0108286 - 0.0109643 - 0.0108776 - 0.0105736 - 0.0100705 - 0.00939751 - 0.00859238 - 0.00769753 - 0.00675657 - 0.00581083 - 0.00489651 - 0.0040427 - 0.00327035 - 0.00259211 - 0.00201302 - 0.00153173 - 0.00114196 - 0.000834175 - 0.000597036 - 0.000418679 - 0.000287673 - 0.000193667 - 0.000127749 - 8.25696e-05 - 5.23006e-05 - 3.24804e-05 - 1.98088e-05 - 1.19273e-05 - 7.21506e-06 - 4.61863e-06 - 3.52388e-06 - 3.67649e-06 - 5.15343e-06 - 8.3891e-06 - 1.42612e-05 - 2.42428e-05 - 4.06281e-05 - 6.68359e-05 - 0.000107789 - 0.000170354 - 0.000263818 - 0.000400329 - 0.00059524 - 0.000867232 - 0.0012381 - 0.00173205 - 0.0023745 - 0.00319011 - 0.00420035 - 0.0054206 - 0.00685696 - 0.00850339 - 0.0103395 - 0.0123294 - 0.0144223 - 0.0165546 - 0.0186545 - 0.0206468 - 0.0224604 - 0.0240341 - 0.0253228 - 0.026301 - 0.0269648 - 0.0273304 - 0.0274301 - 0.0273064 - 0.0270053 - 0.0265691 - 0.0260305 - 0.0254095 - 0.0247121 - 0.0239319 - 0.023054 - 0.0220604 - 0.0209356 - 0.0196714 - 0.0182707 - 0.0167493 - 0.0151347 - 0.0134645 - 0.0117821 - 0.0101327 - 0.00855887 - 0.00709693 - 0.0057744 - 0.00460875 - 0.00360736 - 0.00276847 - 0.00208292 - 0.00153623 - 0.00111069 - 0.000787407 - 0.000547808 - 0.000374892 - 0.000253994 - 0.000173243 - 0.000123774 - 9.98233e-05 - 9.87971e-05 - 0.000121408 - 0.000171927 - 0.000258572 - 0.000393997 - 0.000595817 - 0.00088703 - 0.00129616 - 0.00185693 - 0.00260716 - 0.00358686 - 0.00483517 - 0.00638631 - 0.00826471 - 0.0104795 - 0.0130195 - 0.0158483 - 0.018902 - 0.0220888 - 0.0252913 - 0.0283731 - 0.0311875 - 0.0335885 - 0.0354435 - 0.0366455 - 0.0371228 - 0.0368467 - 0.0358338 - 0.0341447 - 0.031878 - 0.0291606 - 0.0261359 - 0.0229518 - 0.0197484 - 0.0166488 - 0.0137522 - 0.0111301 - 0.00882594 - 0.00685741 - 0.00522031 - 0.00389377 - 0.00284564 - 0.00203764 - 0.00142959 - 0.000982723 - 0.000661894 - 0.0004368 - 0.000282432 - 0.000178929 - 0.000111067 - 6.75505e-05 - 4.02538e-05 - 2.35029e-05 - 1.34454e-05 - 7.53639e-06 - 4.13894e-06 - 2.22716e-06 - 1.17423e-06 - 6.0658e-07 - 3.07016e-07 - 1.52255e-07 - 7.3981e-08 - 3.52217e-08 - 1.64312e-08 - 7.51334e-09 - 3.37362e-09 - 1.50265e-09 - 7.0035e-10 - 4.24077e-10 - 4.75335e-10 - 8.76813e-10 - 1.8765e-09 - 4.0748e-09 - 8.72522e-09 - 1.83265e-08 - 3.77229e-08 - 7.60823e-08 - 1.5035e-07 - 2.91111e-07 - 5.5227e-07 - 1.02655e-06 - 1.86959e-06 - 3.33619e-06 - 5.83299e-06 - 9.99237e-06 - 1.67719e-05 - 2.75825e-05 - 4.44448e-05 - 7.0169e-05 - 0.000108544 - 0.000164515 - 0.000244309 - 0.000355477 - 0.000506781 - 0.00070789 - 0.000968832 - 0.00129918 - 0.00170696 - 0.00219744 - 0.0027717 - 0.00342542 - 0.00414779 - 0.00492104 - 0.00572051 - 0.00651551 - 0.00727109 - 0.00795037 - 0.00851749 - 0.00894073 - 0.00919542 - 0.00926631 - 0.0091491 - 0.0088509 - 0.00838944 - 0.0077914 - 0.00708982 - 0.00632108 - 0.00552185 - 0.00472622 - 0.00396352 - 0.00325675 - 0.00262195 - 0.00206825 - 0.00159851 - 0.00121051 - 0.000898163 - 0.00065295 - 0.000465095 - 0.000324594 - 0.00022196 - 0.000148712 - 9.76239e-05 - 6.27917e-05 - 3.95717e-05 - 2.44345e-05 - 1.47829e-05 - 8.76302e-06 - 5.0896e-06 - 2.89635e-06 - 1.61493e-06 - 8.82256e-07 - 4.7225e-07 - 2.47678e-07 - 1.27274e-07 - 6.40809e-08 - 3.1613e-08 - 1.52825e-08 - 7.24349e-09 - 3.37597e-09 - 1.57127e-09 - 7.87334e-10 - 5.49235e-10 - 7.08757e-10 - 1.36716e-09 - 2.92932e-09 - 6.31231e-09 - 1.33936e-08 - 2.78707e-08 - 5.68338e-08 - 1.13558e-07 - 2.22314e-07 - 4.26437e-07 - 8.01454e-07 - 1.47584e-06 - 2.66279e-06 - 4.7073e-06 - 8.15349e-06 - 1.38373e-05 - 2.3009e-05 - 3.74869e-05 - 5.98409e-05 - 9.35953e-05 - 0.000143432 - 0.000215365 - 0.000316842 - 0.000456715 - 0.000645038 - 0.000892611 - 0.00121025 - 0.00160778 - 0.00209273 - 0.00266894 - 0.00333503 - 0.00408317 - 0.00489815 - 0.0057571 - 0.00662998 - 0.00748096 - 0.00827066 - 0.008959 - 0.00950859 - 0.00988805 - 0.010075 - 0.0100581 - 0.00983853 - 0.00942953 - 0.00885532 - 0.00814865 - 0.00734787 - 0.0064935 - 0.00562505 - 0.0047782 - 0.00398281 - 0.00326164 - 0.00263011 - 0.00209661 - 0.00166348 - 0.00132822 - 0.00108488 - 0.000925197 - 0.000839661 - 0.000818171 - 0.000850452 - 0.000926225 - 0.00103522 - 0.00116715 - 0.00131169 - 0.00145858 - 0.00159787 - 0.00172029 - 0.00181768 - 0.00188348 - 0.00191318 - 0.00190459 - 0.00185802 - 0.0017761 - 0.00166357 - 0.00152673 - 0.00137286 - 0.00120957 - 0.00104417 - 0.000883183 - 0.000731926 - 0.000594321 - 0.000472837 - 0.000368587 - 0.000281517 - 0.000210672 - 0.000154471 - 0.000110977 - 7.81216e-05 - 5.38892e-05 - 3.64365e-05 - 2.41667e-05 - 1.57607e-05 - 1.0179e-05 - 6.64527e-06 - 4.62443e-06 - 3.80367e-06 - 4.08572e-06 - 5.6009e-06 - 8.74399e-06 - 1.42407e-05 - 2.32471e-05 - 3.74834e-05 - 5.93981e-05 - 9.23544e-05 - 0.000140822 - 0.000210543 - 0.000308644 - 0.000443633 - 0.000625244 - 0.000864061 - 0.0011709 - 0.00155593 - 0.0020275 - 0.00259089 - 0.00324684 - 0.00399034 - 0.00480954 - 0.00568528 - 0.00659119 - 0.00749458 - 0.00835814 - 0.00914232 - 0.00980831 - 0.0103211 - 0.0106527 - 0.0107844 - 0.0107086 - 0.0104298 - 0.00996373 - 0.00933625 - 0.00858073 - 0.00773524 - 0.00683939 - 0.0059313 - 0.00504505 - 0.00420879 - 0.00344364 - 0.00276337 - 0.00217477 - 0.00167855 - 0.00127055 - 0.000943173 - 0.000686672 - 0.00049038 - 0.000343661 - 0.000236618 - 0.00016055 - 0.000108187 - 7.37676e-05 - 5.3005e-05 - 4.29873e-05 - 4.20568e-05 - 4.9689e-05 - 6.63831e-05 - 9.35609e-05 - 0.000133462 - 0.000189012 - 0.000263645 - 0.00036105 - 0.000484834 - 0.000638098 - 0.000822944 - 0.00103994 - 0.00128764 - 0.00156213 - 0.00185686 - 0.00216261 - 0.00246782 - 0.00275923 - 0.00302271 - 0.00324447 - 0.00341214 - 0.00351599 - 0.00354981 - 0.00351155 - 0.00340353 - 0.0032322 - 0.00300748 - 0.00274185 - 0.00244919 - 0.00214357 - 0.00183819 - 0.00154447 - 0.00127146 - 0.00102557 - 0.000810524 - 0.000627627 - 0.000476183 - 0.000353982 - 0.000257822 - 0.000183984 - 0.000128626 - 8.80794e-05 - 5.90389e-05 - 3.86628e-05 - 2.45949e-05 - 1.49291e-05 - 8.13403e-06 - 2.94857e-06 - -1.74116e-06 - -7.04209e-06 - -1.42113e-05 - -2.48328e-05 - -4.10146e-05 - -6.56039e-05 - -0.000102411 - -0.000156423 - -0.000233969 - -0.000342807 - -0.000492057 - -0.000691949 - -0.000953302 - -0.00128673 - -0.00170157 - -0.00220453 - -0.00279826 - -0.00347991 - -0.00423992 - -0.00506123 - -0.00591925 - -0.0067825 - -0.00761422 - -0.00837483 - -0.00902489 - -0.00952848 - -0.00985649 - -0.00998939 - -0.00991914 - -0.00965003 - -0.00919821 - -0.00859011 - -0.00785987 - -0.00704619 - -0.00618894 - -0.005326 - -0.00449065 - -0.00370973 - -0.00300262 - -0.00238113 - -0.00185009 - -0.0014084 - -0.00105048 - -0.000767675 - -0.000549659 - -0.0003856 - -0.000265039 - -0.000178489 - -0.000117771 - -7.61375e-05 - -4.82267e-05 - -2.99299e-05 - -1.81993e-05 - -1.08426e-05 - -6.32909e-06 - -3.61977e-06 - -2.0284e-06 - -1.11367e-06 - -5.99086e-07 - -3.15758e-07 - -1.63062e-07 - -8.25054e-08 - -4.0902e-08 - -1.98673e-08 - -9.45511e-09 - -4.40887e-09 - -2.01428e-09 - -9.01668e-10 - -3.95464e-10 - -1.69942e-10 - -7.15528e-11 - -2.9518e-11 - -1.19311e-11 - -4.72509e-12 - -1.83346e-12 - -6.97057e-13 - -2.59655e-13 - -9.47633e-14 - -3.3872e-14 - -1.18181e-14 - -3.90035e-15 - -8.29272e-16 - 1.18263e-15 - 4.68453e-15 - 1.39394e-14 - 3.96556e-14 - 1.10235e-13 - 3.00124e-13 - 8.00503e-13 - 2.0918e-12 - 5.35517e-12 - 1.34315e-11 - 3.30047e-11 - 7.94561e-11 - 1.87405e-10 - 4.33047e-10 - 9.80379e-10 - 2.17449e-09 - 4.72525e-09 - 1.006e-08 - 2.09837e-08 - 4.28816e-08 - 8.58558e-08 - 1.68414e-07 - 3.23665e-07 - 6.09432e-07 - 1.12426e-06 - 2.03198e-06 - 3.59822e-06 - 6.24264e-06 - 1.06112e-05 - 1.76715e-05 - 2.88336e-05 - 4.60937e-05 - 7.21939e-05 - 0.000110784 - 0.00016656 - 0.000245348 - 0.00035409 - 0.000500684 - 0.000693638 - 0.000941505 - 0.00125208 - 0.0016314 - 0.00208262 - 0.00260483 - 0.00319202 - 0.00383237 - 0.00450799 - 0.00519525 - 0.00586582 - 0.00648837 - 0.00703083 - 0.0074629 - 0.00775862 - 0.00789868 - 0.00787201 - 0.00767662 - 0.00731945 - 0.00681534 - 0.00618516 - 0.0054536 - 0.00464667 - 0.00378961 - 0.00290508 - 0.00201228 - 0.00112666 - 0.000260383 - -0.000576704 - -0.00137586 - -0.00212829 - -0.0028243 - -0.0034529 - -0.00400205 - -0.00445942 - -0.00481357 - -0.0050554 - -0.00517947 - -0.00518511 - -0.00507705 - -0.00486543 - -0.00456522 - -0.00419512 - -0.00377605 - -0.00332958 - -0.00287625 - -0.00243427 - -0.00201849 - -0.00163986 - -0.00130532 - -0.00101802 - -0.000777914 - -0.000582426 - -0.000427253 - -0.000307089 - -0.000216263 - -0.000149222 - -0.000100884 - -6.68267e-05 - -4.33723e-05 - -2.75812e-05 - -1.7185e-05 - -1.04911e-05 - -6.27527e-06 - -3.67773e-06 - -2.11185e-06 - -1.18819e-06 - -6.55e-07 - -3.53781e-07 - -1.87222e-07 - -9.70701e-08 - -4.92929e-08 - -2.44779e-08 - -1.17909e-08 - -5.27448e-09 - -1.60946e-09 - 1.27482e-09 - 5.18955e-09 - 1.26581e-08 - 2.82967e-08 - 6.12173e-08 - 1.29452e-07 - 2.68086e-07 - 5.43925e-07 - 1.08126e-06 - 2.10601e-06 - 4.01906e-06 - 7.51496e-06 - 1.37678e-05 - 2.47139e-05 - 4.34663e-05 - 7.49034e-05 - 0.00012647 - 0.000209223 - 0.000339133 - 0.0005386 - 0.000838107 - 0.00127782 - 0.00190887 - 0.00279396 - 0.00400683 - 0.00563014 - 0.00775128 - 0.010456 - 0.0138195 - 0.0178961 - 0.022707 - 0.0282292 - 0.0343855 - 0.0410382 - 0.0479888 - 0.0549831 - 0.0617247 - 0.0678938 - 0.0731722 - 0.0772704 - 0.0799544 - 0.0810684 - 0.0805512 - 0.0784438 - 0.0748868 - 0.0701092 - 0.0644092 - 0.0581302 - 0.0516357 - 0.0452844 - 0.0394097 - 0.0343041 - 0.0302089 - 0.0273084 - 0.0257283 - 0.0255355 - 0.0267383 - 0.0292872 - 0.0330737 - 0.0379307 - 0.0436337 - 0.0499051 - 0.0564229 - 0.0628345 - 0.0687764 - 0.073897 - 0.0778824 - 0.0804804 - 0.0815217 - 0.0809336 - 0.0787451 - 0.0750827 - 0.0701565 - 0.0642396 - 0.0576423 - 0.0506847 - 0.0436712 - 0.0368695 - 0.030495 - 0.024703 - 0.0195869 - 0.0151828 - 0.0114776 - 0.00841987 - 0.00593219 - 0.00392215 - 0.00229274 - 0.000950838 - -0.000186034 - -0.00118449 - -0.00209138 - -0.00293083 - -0.00370225 - -0.00437912 - -0.00490892 - -0.00521447 - -0.00519708 - -0.00474199 - -0.00372615 - -0.00202842 - 0.000458113 - 0.00381264 - 0.00807298 - 0.0132233 - 0.0191849 - 0.0258114 - 0.0328904 - 0.0401515 - 0.047281 - 0.0539439 - 0.059808 - 0.0645718 - 0.0679897 - 0.0698937 - 0.0702075 - 0.0689522 - 0.0662425 - 0.062274 - 0.0573033 - 0.0516237 - 0.0455395 - 0.0393419 - 0.0332886 - 0.0275894 - 0.0223987 - 0.017814 - 0.0138795 - 0.0105943 - 0.00792264 - 0.0058046 - 0.00416665 - 0.00293035 - 0.00201918 - 0.00136318 - 0.000901698 - 0.000584386 - 0.000371087 - 0.000230892 - 0.000140786 - 8.41686e-05 - 4.94262e-05 - 2.86893e-05 - 1.68175e-05 - 1.06308e-05 - 8.37418e-06 - 9.39916e-06 - 1.40463e-05 - 2.3726e-05 - 4.1207e-05 - 7.11357e-05 - 0.000120818 - 0.000201294 - 0.000328719 - 0.000526031 - 0.000824827 - 0.00126728 - 0.00190781 - 0.00281418 - 0.00406746 - 0.00576033 - 0.00799326 - 0.0108681 - 0.014479 - 0.0189005 - 0.0241747 - 0.0302971 - 0.0372043 - 0.044765 - 0.0527758 - 0.0609654 - 0.0690054 - 0.0765305 - 0.0831643 - 0.0885505 - 0.0923838 - 0.094439 - 0.0945928 - 0.0928357 - 0.0892735 - 0.0841163 - 0.0776584 - 0.0702501 - 0.0622666 - 0.0540773 - 0.0460179 - 0.0383702 - 0.0313489 - 0.0250975 - 0.0196902 - 0.0151414 - 0.0114171 - 0.00844929 - 0.00614968 - 0.00442187 - 0.00317137 - 0.00231276 - 0.00177424 - 0.00149973 - 0.00144902 - 0.00159637 - 0.00192796 - 0.00243858 - 0.00312782 - 0.00399592 - 0.00503982 - 0.00624939 - 0.00760438 - 0.00907248 - 0.0106086 - 0.0121559 - 0.0136483 - 0.0150148 - 0.0161846 - 0.0170932 - 0.0176882 - 0.0179342 - 0.0178162 - 0.0173415 - 0.0165385 - 0.0154541 - 0.0141492 - 0.012693 - 0.0111571 - 0.00960964 - 0.00811103 - 0.00671036 - 0.00544389 - 0.00433493 - 0.00339517 - 0.00262695 - 0.0020262 - 0.00158568 - 0.00129798 - 0.00115831 - 0.0011665 - 0.00132846 - 0.00165655 - 0.00216916 - 0.00288914 - 0.00384126 - 0.00504868 - 0.00652871 - 0.00828802 - 0.0103181 - 0.012591 - 0.015057 - 0.0176439 - 0.0202583 - 0.0227906 - 0.0251218 - 0.0271321 - 0.0287114 - 0.0297689 - 0.0302419 - 0.0301019 - 0.0293576 - 0.0280537 - 0.0262672 - 0.0240992 - 0.0216666 - 0.0190912 - 0.0164912 - 0.0139727 - 0.0116248 - 0.00951633 - 0.00769657 - 0.00619684 - 0.00503464 - 0.00421823 - 0.00375123 - 0.00363649 - 0.00387863 - 0.00448478 - 0.00546347 - 0.00682166 - 0.00856033 - 0.0106692 - 0.0131211 - 0.0158675 - 0.0188355 - 0.0219272 - 0.0250222 - 0.0279839 - 0.0306674 - 0.0329313 - 0.034649 - 0.0357203 - 0.0360812 - 0.0357095 - 0.0346278 - 0.0329006 - 0.030628 - 0.0279365 - 0.0249667 - 0.0218619 - 0.0187564 - 0.015767 - 0.0129863 - 0.01048 - 0.00828649 - 0.00641975 - 0.00487306 - 0.00362429 - 0.00264108 - 0.00188571 - 0.00131919 - 0.000904223 - 0.000607268 - 0.000399597 - 0.000257633 - 0.000162749 - 0.000100733 - 6.10885e-05 - 3.62982e-05 - 2.11324e-05 - 1.20545e-05 - 6.7373e-06 - 3.68943e-06 - 1.97957e-06 - 1.04068e-06 - 5.36045e-07 - 2.70534e-07 - 1.33777e-07 - 6.4815e-08 - 3.07685e-08 - 1.43112e-08 - 6.522e-09 - 2.91221e-09 - 1.2741e-09 - 5.46157e-10 - 2.29388e-10 - 9.43974e-11 - 3.80615e-11 - 1.50366e-11 - 5.82036e-12 - 2.20743e-12 - 8.20277e-13 - 2.98656e-13 - 1.06541e-13 - 3.72394e-14 - 1.27533e-14 - 4.27939e-15 - 1.40694e-15 - 4.5322e-16 - 1.43046e-16 - 4.42367e-17 - 1.34037e-17 - 3.97927e-18 - 1.1575e-18 - 3.29893e-19 - 9.21218e-20 - 2.52051e-20 - 6.75697e-21 - 1.77481e-21 - 4.56761e-22 - 1.15176e-22 - 2.8456e-23 - 6.88845e-24 - 1.63383e-24 - 3.79688e-25 - 8.64541e-26 - 1.92877e-26 - 4.21611e-27 - 9.02986e-28 - 1.8949e-28 - 3.89609e-29 - 7.84888e-30 - 1.54926e-30 - 2.99624e-31 - 5.67762e-32 - 1.05413e-32 - 1.91759e-33 - 3.41788e-34 - 5.9689e-35 - 1.02133e-35 - 1.7123e-36 - 2.81272e-37 - 4.52701e-38 - 7.13892e-39 - 1.10304e-39 - 1.66988e-40 - 2.47695e-41 - 3.59987e-42 - 5.12616e-43 - 7.15212e-44 - 9.77719e-45 - 1.30957e-45 - 1.71863e-46 - 2.2099e-47 - 2.78419e-48 - 3.43687e-49 - 4.15684e-50 - 4.92607e-51 - 5.71972e-52 - 6.50707e-53 - 7.25326e-54 - 7.92169e-55 - 8.47695e-56 - 8.88787e-57 - 9.13047e-58 - 9.1902e-59 - 9.06346e-60 - 8.75789e-61 - 8.29167e-62 - 7.69169e-63 - 6.99098e-64 - 6.22574e-65 - 5.43226e-66 - 4.64417e-67 - 3.89019e-68 - 3.1928e-69 - 2.56749e-70 - 2.02294e-71 - 1.56169e-72 - 1.18125e-73 - 8.75441e-75 - 6.35695e-76 - 4.5228e-77 - 3.15284e-78 - 2.15345e-79 - 1.44113e-80 - 9.44951e-82 - 6.07088e-83 - 3.82148e-84 - 2.35693e-85 - 1.4243e-86 - 8.43317e-88 - 4.89235e-89 - 2.78087e-90 - 1.54875e-91 - 8.45122e-93 - 4.51849e-94 - 2.36704e-95 - 1.21493e-96 - 6.10995e-98 - 3.01064e-99 - 1.45351e-100 - 6.87564e-102 - 3.18673e-103 - 1.44715e-104 - 6.43903e-106 - 2.80713e-107 - 1.19907e-108 - 5.01875e-110 - 2.16259e-111 - 2.64582e-111 - 6.14557e-110 - 1.44359e-108 - 3.32264e-107 - 7.49309e-106 - 1.65567e-104 - 3.58448e-103 - 7.6035e-102 - 1.5803e-100 - 3.2181e-99 - 6.42094e-98 - 1.25526e-96 - 2.4044e-95 - 4.51248e-94 - 8.29776e-93 - 1.49501e-91 - 2.63914e-90 - 4.56477e-89 - 7.73593e-88 - 1.28452e-86 - 2.08982e-85 - 3.3313e-84 - 5.203e-83 - 7.96217e-82 - 1.19384e-80 - 1.75387e-79 - 2.52455e-78 - 3.56049e-77 - 4.92007e-76 - 6.66146e-75 - 8.837e-74 - 1.14862e-72 - 1.4628e-71 - 1.82529e-70 - 2.23159e-69 - 2.67322e-68 - 3.13755e-67 - 3.60815e-66 - 4.06551e-65 - 4.4883e-64 - 4.85496e-63 - 5.14548e-62 - 5.34323e-61 - 5.43648e-60 - 5.41963e-59 - 5.29368e-58 - 5.0662e-57 - 4.75055e-56 - 4.36458e-55 - 3.92897e-54 - 3.46538e-53 - 2.99475e-52 - 2.53575e-51 - 2.10372e-50 - 1.71005e-49 - 1.36196e-48 - 1.06282e-47 - 8.12623e-47 - 6.08774e-46 - 4.46848e-45 - 3.21367e-44 - 2.26453e-43 - 1.56348e-42 - 1.05766e-41 - 7.01025e-41 - 4.55259e-40 - 2.89682e-39 - 1.80601e-38 - 1.1032e-37 - 6.60281e-37 - 3.87203e-36 - 2.22477e-35 - 1.25247e-34 - 6.90859e-34 - 3.73376e-33 - 1.97716e-32 - 1.02582e-31 - 5.21484e-31 - 2.59744e-30 - 1.26762e-29 - 6.06133e-29 - 2.83978e-28 - 1.30358e-27 - 5.86312e-27 - 2.58379e-26 - 1.11564e-25 - 4.71981e-25 - 1.95643e-24 - 7.94588e-24 - 3.16197e-23 - 1.23285e-22 - 4.70977e-22 - 1.7629e-21 - 6.46539e-21 - 2.32327e-20 - 8.17978e-20 - 2.82178e-19 - 9.53769e-19 - 3.15866e-18 - 1.02495e-17 - 3.25866e-17 - 1.01512e-16 - 3.0984e-16 - 9.26612e-16 - 2.71519e-15 - 7.79551e-15 - 2.19297e-14 - 6.04456e-14 - 1.63246e-13 - 4.31982e-13 - 1.12005e-12 - 2.84549e-12 - 7.0832e-12 - 1.72764e-11 - 4.12889e-11 - 9.66876e-11 - 2.21855e-10 - 4.98805e-10 - 1.09891e-09 - 2.37226e-09 - 5.01812e-09 - 1.04016e-08 - 2.11277e-08 - 4.2053e-08 - 8.20252e-08 - 1.56787e-07 - 2.93696e-07 - 5.39164e-07 - 9.70047e-07 - 1.71052e-06 - 2.95629e-06 - 5.00803e-06 - 8.3159e-06 - 1.35363e-05 - 2.16005e-05 - 3.37937e-05 - 5.18379e-05 - 7.7972e-05 - 0.000115015 - 0.000166394 - 0.000236125 - 0.000328721 - 0.00044901 - 0.000601858 - 0.000791802 - 0.00102259 - 0.00129668 - 0.00161472 - 0.00197509 - 0.00237354 - 0.00280296 - 0.00325346 - 0.00371255 - 0.00416568 - 0.00459692 - 0.00498986 - 0.00532857 - 0.00559865 - 0.00578819 - 0.00588854 - 0.00589499 - 0.00580705 - 0.00562858 - 0.00536746 - 0.00503512 - 0.00464575 - 0.0042153 - 0.00376048 - 0.0032977 - 0.00284209 - 0.00240674 - 0.00200212 - 0.0016358 - - - - - -3.49674e-30 - -4.35161e-30 - -5.30607e-30 - -6.33917e-30 - -7.42042e-30 - -8.51059e-30 - -9.5635e-30 - -1.05283e-29 - -1.13493e-29 - -1.19504e-29 - -1.21394e-29 - -1.11182e-29 - -5.10328e-30 - 2.39387e-29 - 1.60461e-28 - 7.9186e-28 - 3.65835e-27 - 1.64153e-26 - 7.20432e-26 - 3.09686e-25 - 1.30423e-24 - 5.38172e-24 - 2.17581e-23 - 8.61905e-23 - 3.34529e-22 - 1.27217e-21 - 4.74016e-21 - 1.73052e-20 - 6.19012e-20 - 2.16949e-19 - 7.44995e-19 - 2.5066e-18 - 8.26332e-18 - 2.66907e-17 - 8.44702e-17 - 2.61929e-16 - 7.95792e-16 - 2.36893e-15 - 6.90944e-15 - 1.97456e-14 - 5.52884e-14 - 1.51682e-13 - 4.0773e-13 - 1.07386e-12 - 2.77114e-12 - 7.00658e-12 - 1.73577e-11 - 4.21321e-11 - 1.00201e-10 - 2.33489e-10 - 5.33089e-10 - 1.19253e-09 - 2.61381e-09 - 5.61329e-09 - 1.18113e-08 - 2.43508e-08 - 4.91889e-08 - 9.73549e-08 - 1.88793e-07 - 3.58714e-07 - 6.67805e-07 - 1.21811e-06 - 2.17702e-06 - 3.81218e-06 - 6.54066e-06 - 1.09953e-05 - 1.81104e-05 - 2.92272e-05 - 4.6215e-05 - 7.16003e-05 - 0.000108689 - 0.000161656 - 0.000235577 - 0.000336367 - 0.000470576 - 0.000645036 - 0.000866312 - 0.00113999 - 0.00146982 - 0.00185681 - 0.00229829 - 0.00278727 - 0.00331201 - 0.00385604 - 0.00439875 - 0.00491649 - 0.0053842 - 0.00577737 - 0.00607415 - 0.00625743 - 0.00631647 - 0.00624807 - 0.00605694 - 0.00575536 - 0.00536215 - 0.00490107 - 0.0043989 - 0.00388353 - 0.00338212 - 0.00291963 - 0.00251784 - 0.00219476 - 0.0019644 - 0.00183683 - 0.00181827 - 0.00191123 - 0.00211441 - 0.0024225 - 0.0028259 - 0.00331037 - 0.00385693 - 0.00444204 - 0.00503828 - 0.00561553 - 0.0061427 - 0.00658983 - 0.00693041 - 0.0071435 - 0.00721563 - 0.00714193 - 0.00692656 - 0.00658218 - 0.00612867 - 0.00559119 - 0.00499782 - 0.0043772 - 0.0037562 - 0.00315819 - 0.00260175 - 0.00210005 - 0.00166085 - 0.00128697 - 0.000977109 - 0.000726866 - 0.000529789 - 0.000378345 - 0.000264734 - 0.000181497 - 0.000121917 - 8.02412e-05 - 5.17448e-05 - 3.26944e-05 - 2.02403e-05 - 1.22771e-05 - 7.29648e-06 - 4.24881e-06 - 2.42414e-06 - 1.35514e-06 - 7.42249e-07 - 3.98337e-07 - 2.09454e-07 - 1.0791e-07 - 5.44722e-08 - 2.69416e-08 - 1.3056e-08 - 6.19913e-09 - 2.88397e-09 - 1.31458e-09 - 5.87109e-10 - 2.56914e-10 - 1.10153e-10 - 4.62739e-11 - 1.90459e-11 - 7.67917e-12 - 3.02905e-12 - 1.15766e-12 - 3.97361e-13 - 3.41409e-14 - -2.89899e-13 - -8.92159e-13 - -2.32794e-12 - -5.84648e-12 - -1.43519e-11 - -3.45078e-11 - -8.12912e-11 - -1.87631e-10 - -4.24327e-10 - -9.40231e-10 - -2.04129e-09 - -4.34222e-09 - -9.05016e-09 - -1.84815e-08 - -3.6979e-08 - -7.24953e-08 - -1.39252e-07 - -2.62077e-07 - -4.83275e-07 - -8.73166e-07 - -1.54574e-06 - -2.68109e-06 - -4.55642e-06 - -7.58706e-06 - -1.23783e-05 - -1.97871e-05 - -3.09915e-05 - -4.75597e-05 - -7.15111e-05 - -0.000105352 - -0.000152073 - -0.000215078 - -0.000298041 - -0.000404664 - -0.00053833 - -0.000701682 - -0.000896126 - -0.00112133 - -0.00137479 - -0.00165149 - -0.0019438 - -0.00224163 - -0.00253287 - -0.00280414 - -0.00304174 - -0.00323283 - -0.0033665 - -0.00343488 - -0.00343384 - -0.00336344 - -0.0032279 - -0.00303518 - -0.00279621 - -0.00252384 - -0.00223166 - -0.00193291 - -0.00163942 - -0.00136091 - -0.00110451 - -0.000874518 - -0.000672571 - -0.00049787 - -0.000347626 - -0.000217553 - -0.000102387 - 3.62246e-06 - 0.000106247 - 0.000210868 - 0.000322098 - 0.000443445 - 0.000577031 - 0.000723357 - 0.000881165 - 0.0010474 - 0.00121734 - 0.00138482 - 0.0015427 - 0.00168342 - 0.00179962 - 0.00188484 - 0.00193416 - 0.00194463 - 0.00191564 - 0.00184894 - 0.00174849 - 0.00162005 - 0.00147064 - 0.00130789 - 0.00113938 - 0.000972049 - 0.000811701 - 0.000662676 - 0.000527676 - 0.000407728 - 0.000302261 - 0.000209254 - 0.000125426 - 4.64271e-05 - -3.29605e-05 - -0.000118629 - -0.000216928 - -0.0003344 - -0.000477448 - -0.000651899 - -0.000862487 - -0.00111225 - -0.00140188 - -0.00172921 - -0.00208869 - -0.00247127 - -0.00286449 - -0.003253 - -0.00361946 - -0.0039458 - -0.00421463 - -0.00441082 - -0.00452289 - -0.00454411 - -0.00447319 - -0.00431444 - -0.00407725 - -0.00377526 - -0.00342502 - -0.00304451 - -0.0026516 - -0.00226274 - -0.0018919 - -0.00154988 - -0.00124405 - -0.000978389 - -0.000753916 - -0.000569208 - -0.000421072 - -0.000305196 - -0.000216739 - -0.000150811 - -0.000102817 - -6.86807e-05 - -4.49512e-05 - -2.8826e-05 - -1.81119e-05 - -1.11501e-05 - -6.72565e-06 - -3.97489e-06 - -2.30172e-06 - -1.30592e-06 - -7.25967e-07 - -3.95417e-07 - -2.11023e-07 - -1.10342e-07 - -5.65314e-08 - -2.83775e-08 - -1.39572e-08 - -6.72598e-09 - -3.17579e-09 - -1.46921e-09 - -6.65968e-10 - -2.95774e-10 - -1.28707e-10 - -5.48761e-11 - -2.29245e-11 - -9.38325e-12 - -3.76308e-12 - -1.47867e-12 - -5.69293e-13 - -2.14752e-13 - -7.93733e-14 - -2.87441e-14 - -1.01991e-14 - -3.54576e-15 - -1.2078e-15 - -4.03104e-16 - -1.31818e-16 - -4.22349e-17 - -1.32588e-17 - -4.07825e-18 - -1.22908e-18 - -3.62931e-19 - -1.05004e-19 - -2.9766e-20 - -8.26751e-21 - -2.24991e-21 - -5.99919e-22 - -1.56732e-22 - -4.01224e-23 - -1.00739e-23 - -2.52276e-24 - -8.0687e-25 - -1.016e-24 - -3.58361e-24 - -1.41196e-23 - -5.4951e-23 - -2.09636e-22 - -7.83616e-22 - -2.86998e-21 - -1.02989e-20 - -3.62111e-20 - -1.24746e-19 - -4.21067e-19 - -1.39255e-18 - -4.5124e-18 - -1.43265e-17 - -4.45669e-17 - -1.35837e-16 - -4.05661e-16 - -1.18698e-15 - -3.40301e-15 - -9.55911e-15 - -2.63093e-14 - -7.09476e-14 - -1.87458e-13 - -4.85294e-13 - -1.23096e-12 - -3.05929e-12 - -7.4496e-12 - -1.77739e-11 - -4.15498e-11 - -9.51682e-11 - -2.13576e-10 - -4.69623e-10 - -1.01177e-09 - -2.13577e-09 - -4.41734e-09 - -8.9517e-09 - -1.77741e-08 - -3.45784e-08 - -6.59113e-08 - -1.23098e-07 - -2.25258e-07 - -4.03874e-07 - -7.09493e-07 - -1.2212e-06 - -2.05951e-06 - -3.40311e-06 - -5.50968e-06 - -8.74004e-06 - -1.35843e-05 - -2.06871e-05 - -3.08674e-05 - -4.51271e-05 - -6.46419e-05 - -9.07258e-05 - -0.000124764 - -0.00016811 - -0.000221944 - -0.000287112 - -0.000363934 - -0.000452037 - -0.000550211 - -0.000656332 - -0.000767384 - -0.000879594 - -0.000988694 - -0.00109031 - -0.00118047 - -0.00125612 - -0.00131572 - -0.00135978 - -0.00139119 - -0.00141553 - -0.00144105 - -0.00147842 - -0.00154024 - -0.00164022 - -0.00179215 - -0.0020087 - -0.00229996 - -0.00267211 - -0.00312609 - -0.00365653 - -0.00425123 - -0.00489099 - -0.00555038 - -0.00619899 - -0.00680353 - -0.00733032 - -0.00774812 - -0.00803088 - -0.00816018 - -0.00812687 - -0.00793199 - -0.0075865 - -0.00711016 - -0.00652951 - -0.00587539 - -0.00518015 - -0.00447502 - -0.00378787 - -0.0031416 - -0.00255317 - -0.00203343 - -0.00158747 - -0.0012155 - -0.000914005 - -0.000677014 - -0.000497327 - -0.000367638 - -0.00028146 - -0.000233825 - -0.000221778 - -0.000244658 - -0.000304193 - -0.000404393 - -0.000551247 - -0.0007522 - -0.00101539 - -0.00134866 - -0.00175834 - -0.00224794 - -0.00281677 - -0.00345876 - -0.00416153 - -0.00490608 - -0.00566706 - -0.00641387 - -0.00711248 - -0.00772785 - -0.00822684 - -0.00858114 - -0.00876987 - -0.0087817 - -0.0086159 - -0.00828247 - -0.0078011 - -0.00719927 - -0.00650966 - -0.00576719 - -0.0050062 - -0.00425783 - -0.00354817 - -0.00289707 - -0.00231766 - -0.00181667 - -0.00139522 - -0.00104989 - -0.000774071 - -0.000559185 - -0.000395792 - -0.000274483 - -0.00018651 - -0.000124172 - -8.09996e-05 - -5.17701e-05 - -3.24199e-05 - -1.98921e-05 - -1.19588e-05 - -7.04419e-06 - -4.06548e-06 - -2.29895e-06 - -1.27376e-06 - -6.91503e-07 - -3.67873e-07 - -1.91872e-07 - -9.83377e-08 - -5.00377e-08 - -2.64246e-08 - -1.68924e-08 - -1.70683e-08 - -2.72651e-08 - -5.27365e-08 - -1.05874e-07 - -2.1103e-07 - -4.13329e-07 - -7.93705e-07 - -1.49355e-06 - -2.75377e-06 - -4.97481e-06 - -8.80568e-06 - -1.52717e-05 - -2.59505e-05 - -4.32059e-05 - -7.04818e-05 - -0.000112654 - -0.000176423 - -0.000270707 - -0.000406987 - -0.000599513 - -0.000865274 - -0.00122362 - -0.00169541 - -0.00230165 - -0.00306155 - -0.00399008 - -0.00509515 - -0.00637485 - -0.00781483 - -0.00938655 - -0.0110466 - -0.0127377 - -0.0143909 - -0.0159302 - -0.0172779 - -0.0183611 - -0.0191181 - -0.0195041 - -0.0194959 - -0.0190941 - -0.0183227 - -0.0172274 - -0.0158703 - -0.0143247 - -0.0126685 - -0.0109775 - -0.00931998 - -0.00775291 - -0.00631905 - -0.00504632 - -0.00394853 - -0.00302714 - -0.00227387 - -0.00167355 - -0.00120683 - -0.000852687 - -0.000590298 - -0.000400396 - -0.0002661 - -0.000173275 - -0.000110552 - -6.91084e-05 - -4.23286e-05 - -2.54023e-05 - -1.49365e-05 - -8.60522e-06 - -4.85749e-06 - -2.68657e-06 - -1.45587e-06 - -7.73006e-07 - -4.02142e-07 - -2.04981e-07 - -1.02373e-07 - -5.00948e-08 - -2.4018e-08 - -1.12829e-08 - -5.19323e-09 - -2.34203e-09 - -1.03487e-09 - -4.48039e-10 - -1.90063e-10 - -7.90133e-11 - -3.2227e-11 - -1.2995e-11 - -5.44122e-12 - -3.02022e-12 - -3.51191e-12 - -7.21295e-12 - -1.70439e-11 - -4.05427e-11 - -9.4878e-11 - -2.17679e-10 - -4.89379e-10 - -1.07799e-09 - -2.32661e-09 - -4.92003e-09 - -1.01941e-08 - -2.06951e-08 - -4.11646e-08 - -8.02261e-08 - -1.53195e-07 - -2.86623e-07 - -5.25429e-07 - -9.43742e-07 - -1.66085e-06 - -2.8638e-06 - -4.83831e-06 - -8.00904e-06 - -1.29899e-05 - -2.06426e-05 - -3.21413e-05 - -4.9034e-05 - -7.3294e-05 - -0.000107344 - -0.000154036 - -0.000216573 - -0.000298347 - -0.000402696 - -0.000532562 - -0.00069008 - -0.000876123 - -0.00108985 - -0.00132834 - -0.0015863 - -0.00185609 - -0.00212789 - -0.00239021 - -0.00263063 - -0.00283674 - -0.00299722 - -0.00310279 - -0.0031472 - -0.00312775 - -0.00304563 - -0.00290576 - -0.0027163 - -0.0024879 - -0.00223268 - -0.00196316 - -0.0016913 - -0.00142766 - -0.00118077 - -0.000956844 - -0.000759722 - -0.000591024 - -0.000450497 - -0.000336447 - -0.000246194 - -0.000176512 - -0.000123997 - -8.53456e-05 - -5.75558e-05 - -3.80307e-05 - -2.46216e-05 - -1.56184e-05 - -9.70714e-06 - -5.91131e-06 - -3.52706e-06 - -2.06196e-06 - -1.18109e-06 - -6.62861e-07 - -3.64501e-07 - -1.96387e-07 - -1.03672e-07 - -5.36229e-08 - -2.71753e-08 - -1.34939e-08 - -6.56499e-09 - -3.12945e-09 - -1.46164e-09 - -6.6888e-10 - -2.99912e-10 - -1.31758e-10 - -5.67146e-11 - -2.39194e-11 - -9.88424e-12 - -4.00196e-12 - -1.58759e-12 - -6.17081e-13 - -2.35008e-13 - -8.76919e-14 - -3.20607e-14 - -1.14848e-14 - -4.03098e-15 - -1.38623e-15 - -4.67086e-16 - -1.54204e-16 - -4.98803e-17 - -1.58089e-17 - -4.90919e-18 - -1.49367e-18 - -4.45284e-19 - -1.30064e-19 - -3.72232e-20 - -1.04383e-20 - -2.87015e-21 - -7.81928e-22 - -2.43429e-22 - -2.08622e-22 - -6.12204e-22 - -2.2559e-21 - -8.30244e-21 - -2.99767e-20 - -1.06056e-19 - -3.67643e-19 - -1.24869e-18 - -4.15545e-18 - -1.35494e-17 - -4.32871e-17 - -1.35498e-16 - -4.15571e-16 - -1.2488e-15 - -3.67688e-15 - -1.06072e-14 - -2.99821e-14 - -8.30344e-14 - -2.25315e-13 - -5.99047e-13 - -1.56051e-12 - -3.98301e-12 - -9.96075e-12 - -2.44067e-11 - -5.85955e-11 - -1.37834e-10 - -3.17675e-10 - -7.1738e-10 - -1.58727e-09 - -3.44105e-09 - -7.30916e-09 - -1.52118e-08 - -3.10193e-08 - -6.19757e-08 - -1.21324e-07 - -2.32709e-07 - -4.37337e-07 - -8.05303e-07 - -1.45292e-06 - -2.56841e-06 - -4.44867e-06 - -7.54986e-06 - -1.25543e-05 - -2.04549e-05 - -3.26554e-05 - -5.10821e-05 - -7.82973e-05 - -0.000117598 - -0.000173075 - -0.000249615 - -0.000352798 - -0.000488684 - -0.000663453 - -0.000882913 - -0.00115189 - -0.00147356 - -0.00184879 - -0.00227563 - -0.00274905 - -0.00326099 - -0.00380083 - -0.00435636 - -0.00491499 - -0.00546527 - -0.00599832 - -0.00650907 - -0.00699689 - -0.00746558 - -0.00792243 - -0.00837658 - -0.00883672 - -0.00930857 - -0.00979246 - -0.0102816 - -0.0107611 - -0.0112087 - -0.011596 - -0.0118916 - -0.0120645 - -0.0120879 - -0.0119427 - -0.0116203 - -0.011124 - -0.0104692 - -0.00968215 - -0.00879775 - -0.00785624 - -0.00689989 - -0.00596968 - -0.00510249 - -0.00432903 - -0.00367258 - -0.00314848 - -0.00276429 - -0.00252032 - -0.00241049 - -0.00242322 - -0.00254229 - -0.00274769 - -0.00301656 - -0.00332402 - -0.00364428 - -0.00395183 - -0.00422275 - -0.00443609 - -0.00457514 - -0.00462847 - -0.0045906 - -0.00446213 - -0.00424935 - -0.00396339 - -0.0036188 - -0.00323204 - -0.00281971 - -0.00239708 - -0.0019767 - -0.00156754 - -0.00117457 - -0.000798753 - -0.000437587 - -8.59164e-05 - 0.000262944 - 0.000615954 - 0.000979073 - 0.00135609 - 0.00174768 - 0.00215078 - 0.00255837 - 0.00295977 - 0.00334128 - 0.00368735 - 0.00398195 - 0.00421019 - 0.00435982 - 0.00442255 - 0.00439499 - 0.00427906 - 0.00408186 - 0.003815 - 0.00349351 - 0.00313447 - 0.00275551 - 0.00237344 - 0.00200306 - 0.00165638 - 0.00134211 - 0.00106565 - 0.000829314 - 0.000632838 - 0.00047399 - 0.000349261 - 0.00025451 - 0.000185531 - 0.00013851 - 0.000110364 - 9.89575e-05 - 0.000103216 - 0.000123152 - 0.000159795 - 0.000215048 - 0.000291453 - 0.000391873 - 0.000519088 - 0.000675318 - 0.000861708 - 0.00107781 - 0.00132113 - 0.00158679 - 0.00186744 - 0.00215336 - 0.00243292 - 0.00269324 - 0.0029212 - 0.00310445 - 0.00323254 - 0.00329792 - 0.00329666 - 0.00322882 - 0.0030985 - 0.00291337 - 0.00268396 - 0.00242267 - 0.00214264 - 0.00185669 - 0.00157641 - 0.0013114 - 0.0010689 - 0.00085364 - 0.00066796 - 0.00051211 - 0.000384691 - 0.000283138 - 0.000204184 - 0.000144272 - 9.98799e-05 - 6.77504e-05 - 4.5028e-05 - 2.93217e-05 - 1.87083e-05 - 1.16954e-05 - 7.16366e-06 - 4.29928e-06 - 2.52821e-06 - 1.45697e-06 - 8.23301e-07 - 4.57272e-07 - 2.52047e-07 - 1.4307e-07 - 9.41611e-08 - 8.92735e-08 - 1.29198e-07 - 2.33164e-07 - 4.46224e-07 - 8.54461e-07 - 1.61153e-06 - 2.98178e-06 - 5.40736e-06 - 9.60873e-06 - 1.67298e-05 - 2.854e-05 - 4.77041e-05 - 7.81257e-05 - 0.000125363 - 0.000197097 - 0.000303619 - 0.000458263 - 0.0006777 - 0.000981967 - 0.0013941 - 0.00193921 - 0.00264299 - 0.00352941 - 0.00461791 - 0.00592006 - 0.00743607 - 0.0091516 - 0.0110354 - 0.0130381 - 0.0150932 - 0.0171191 - 0.0190248 - 0.0207155 - 0.0221008 - 0.0231024 - 0.0236615 - 0.0237446 - 0.0233467 - 0.0224919 - 0.0212307 - 0.0196357 - 0.0177941 - 0.0158004 - 0.0137482 - 0.0117235 - 0.00979954 - 0.00803325 - 0.00646427 - 0.00511562 - 0.00399594 - 0.00310269 - 0.00242575 - 0.00195092 - 0.00166287 - 0.00154732 - 0.00159223 - 0.00178801 - 0.00212689 - 0.00260153 - 0.00320325 - 0.00392002 - 0.00473471 - 0.00562383 - 0.006557 - 0.00749741 - 0.00840331 - 0.00923048 - 0.00993535 - 0.0104786 - 0.0108286 - 0.0109643 - 0.0108776 - 0.0105736 - 0.0100705 - 0.00939751 - 0.00859238 - 0.00769753 - 0.00675657 - 0.00581083 - 0.00489651 - 0.0040427 - 0.00327035 - 0.00259211 - 0.00201302 - 0.00153173 - 0.00114196 - 0.000834175 - 0.000597036 - 0.000418679 - 0.000287673 - 0.000193667 - 0.000127749 - 8.25696e-05 - 5.23006e-05 - 3.24804e-05 - 1.98088e-05 - 1.19273e-05 - 7.21506e-06 - 4.61863e-06 - 3.52388e-06 - 3.67649e-06 - 5.15343e-06 - 8.3891e-06 - 1.42612e-05 - 2.42428e-05 - 4.06281e-05 - 6.68359e-05 - 0.000107789 - 0.000170354 - 0.000263818 - 0.000400329 - 0.00059524 - 0.000867232 - 0.0012381 - 0.00173205 - 0.0023745 - 0.00319011 - 0.00420035 - 0.0054206 - 0.00685696 - 0.00850339 - 0.0103395 - 0.0123294 - 0.0144223 - 0.0165546 - 0.0186545 - 0.0206468 - 0.0224604 - 0.0240341 - 0.0253228 - 0.026301 - 0.0269648 - 0.0273304 - 0.0274301 - 0.0273064 - 0.0270053 - 0.0265691 - 0.0260305 - 0.0254095 - 0.0247121 - 0.0239319 - 0.023054 - 0.0220604 - 0.0209356 - 0.0196714 - 0.0182707 - 0.0167493 - 0.0151347 - 0.0134645 - 0.0117821 - 0.0101327 - 0.00855887 - 0.00709693 - 0.0057744 - 0.00460875 - 0.00360736 - 0.00276847 - 0.00208292 - 0.00153623 - 0.00111069 - 0.000787407 - 0.000547808 - 0.000374892 - 0.000253994 - 0.000173243 - 0.000123774 - 9.98233e-05 - 9.87971e-05 - 0.000121408 - 0.000171927 - 0.000258572 - 0.000393997 - 0.000595817 - 0.00088703 - 0.00129616 - 0.00185693 - 0.00260716 - 0.00358686 - 0.00483517 - 0.00638631 - 0.00826471 - 0.0104795 - 0.0130195 - 0.0158483 - 0.018902 - 0.0220888 - 0.0252913 - 0.0283731 - 0.0311875 - 0.0335885 - 0.0354435 - 0.0366455 - 0.0371228 - 0.0368467 - 0.0358338 - 0.0341447 - 0.031878 - 0.0291606 - 0.0261359 - 0.0229518 - 0.0197484 - 0.0166488 - 0.0137522 - 0.0111301 - 0.00882594 - 0.00685741 - 0.00522031 - 0.00389377 - 0.00284564 - 0.00203764 - 0.00142959 - 0.000982723 - 0.000661894 - 0.0004368 - 0.000282432 - 0.000178929 - 0.000111067 - 6.75505e-05 - 4.02538e-05 - 2.35029e-05 - 1.34454e-05 - 7.53639e-06 - 4.13894e-06 - 2.22716e-06 - 1.17423e-06 - 6.0658e-07 - 3.07016e-07 - 1.52255e-07 - 7.3981e-08 - 3.52217e-08 - 1.64312e-08 - 7.51334e-09 - 3.37362e-09 - 1.50265e-09 - 7.0035e-10 - 4.24077e-10 - 4.75335e-10 - 8.76813e-10 - 1.8765e-09 - 4.0748e-09 - 8.72522e-09 - 1.83265e-08 - 3.77229e-08 - 7.60823e-08 - 1.5035e-07 - 2.91111e-07 - 5.5227e-07 - 1.02655e-06 - 1.86959e-06 - 3.33619e-06 - 5.83299e-06 - 9.99237e-06 - 1.67719e-05 - 2.75825e-05 - 4.44448e-05 - 7.0169e-05 - 0.000108544 - 0.000164515 - 0.000244309 - 0.000355477 - 0.000506781 - 0.00070789 - 0.000968832 - 0.00129918 - 0.00170696 - 0.00219744 - 0.0027717 - 0.00342542 - 0.00414779 - 0.00492104 - 0.00572051 - 0.00651551 - 0.00727109 - 0.00795037 - 0.00851749 - 0.00894073 - 0.00919542 - 0.00926631 - 0.0091491 - 0.0088509 - 0.00838944 - 0.0077914 - 0.00708982 - 0.00632108 - 0.00552185 - 0.00472622 - 0.00396352 - 0.00325675 - 0.00262195 - 0.00206825 - 0.00159851 - 0.00121051 - 0.000898163 - 0.00065295 - 0.000465095 - 0.000324594 - 0.00022196 - 0.000148712 - 9.76239e-05 - 6.27917e-05 - 3.95717e-05 - 2.44345e-05 - 1.47829e-05 - 8.76302e-06 - 5.0896e-06 - 2.89635e-06 - 1.61493e-06 - 8.82256e-07 - 4.7225e-07 - 2.47678e-07 - 1.27274e-07 - 6.40809e-08 - 3.1613e-08 - 1.52825e-08 - 7.24349e-09 - 3.37597e-09 - 1.57127e-09 - 7.87334e-10 - 5.49235e-10 - 7.08757e-10 - 1.36716e-09 - 2.92932e-09 - 6.31231e-09 - 1.33936e-08 - 2.78707e-08 - 5.68338e-08 - 1.13558e-07 - 2.22314e-07 - 4.26437e-07 - 8.01454e-07 - 1.47584e-06 - 2.66279e-06 - 4.7073e-06 - 8.15349e-06 - 1.38373e-05 - 2.3009e-05 - 3.74869e-05 - 5.98409e-05 - 9.35953e-05 - 0.000143432 - 0.000215365 - 0.000316842 - 0.000456715 - 0.000645038 - 0.000892611 - 0.00121025 - 0.00160778 - 0.00209273 - 0.00266894 - 0.00333503 - 0.00408317 - 0.00489815 - 0.0057571 - 0.00662998 - 0.00748096 - 0.00827066 - 0.008959 - 0.00950859 - 0.00988805 - 0.010075 - 0.0100581 - 0.00983853 - 0.00942953 - 0.00885532 - 0.00814865 - 0.00734787 - 0.0064935 - 0.00562505 - 0.0047782 - 0.00398281 - 0.00326164 - 0.00263011 - 0.00209661 - 0.00166348 - 0.00132822 - 0.00108488 - 0.000925197 - 0.000839661 - 0.000818171 - 0.000850452 - 0.000926225 - 0.00103522 - 0.00116715 - 0.00131169 - 0.00145858 - 0.00159787 - 0.00172029 - 0.00181768 - 0.00188348 - 0.00191318 - 0.00190459 - 0.00185802 - 0.0017761 - 0.00166357 - 0.00152673 - 0.00137286 - 0.00120957 - 0.00104417 - 0.000883183 - 0.000731926 - 0.000594321 - 0.000472837 - 0.000368587 - 0.000281517 - 0.000210672 - 0.000154471 - 0.000110977 - 7.81216e-05 - 5.38892e-05 - 3.64365e-05 - 2.41667e-05 - 1.57607e-05 - 1.0179e-05 - 6.64527e-06 - 4.62443e-06 - 3.80367e-06 - 4.08572e-06 - 5.6009e-06 - 8.74399e-06 - 1.42407e-05 - 2.32471e-05 - 3.74834e-05 - 5.93981e-05 - 9.23544e-05 - 0.000140822 - 0.000210543 - 0.000308644 - 0.000443633 - 0.000625244 - 0.000864061 - 0.0011709 - 0.00155593 - 0.0020275 - 0.00259089 - 0.00324684 - 0.00399034 - 0.00480954 - 0.00568528 - 0.00659119 - 0.00749458 - 0.00835814 - 0.00914232 - 0.00980831 - 0.0103211 - 0.0106527 - 0.0107844 - 0.0107086 - 0.0104298 - 0.00996373 - 0.00933625 - 0.00858073 - 0.00773524 - 0.00683939 - 0.0059313 - 0.00504505 - 0.00420879 - 0.00344364 - 0.00276337 - 0.00217477 - 0.00167855 - 0.00127055 - 0.000943173 - 0.000686672 - 0.00049038 - 0.000343661 - 0.000236618 - 0.00016055 - 0.000108187 - 7.37676e-05 - 5.3005e-05 - 4.29873e-05 - 4.20568e-05 - 4.9689e-05 - 6.63831e-05 - 9.35609e-05 - 0.000133462 - 0.000189012 - 0.000263645 - 0.00036105 - 0.000484834 - 0.000638098 - 0.000822944 - 0.00103994 - 0.00128764 - 0.00156213 - 0.00185686 - 0.00216261 - 0.00246782 - 0.00275923 - 0.00302271 - 0.00324447 - 0.00341214 - 0.00351599 - 0.00354981 - 0.00351155 - 0.00340353 - 0.0032322 - 0.00300748 - 0.00274185 - 0.00244919 - 0.00214357 - 0.00183819 - 0.00154447 - 0.00127146 - 0.00102557 - 0.000810524 - 0.000627627 - 0.000476183 - 0.000353982 - 0.000257822 - 0.000183984 - 0.000128626 - 8.80794e-05 - 5.90389e-05 - 3.86628e-05 - 2.45949e-05 - 1.49291e-05 - 8.13403e-06 - 2.94857e-06 - -1.74116e-06 - -7.04209e-06 - -1.42113e-05 - -2.48328e-05 - -4.10146e-05 - -6.56039e-05 - -0.000102411 - -0.000156423 - -0.000233969 - -0.000342807 - -0.000492057 - -0.000691949 - -0.000953302 - -0.00128673 - -0.00170157 - -0.00220453 - -0.00279826 - -0.00347991 - -0.00423992 - -0.00506123 - -0.00591925 - -0.0067825 - -0.00761422 - -0.00837483 - -0.00902489 - -0.00952848 - -0.00985649 - -0.00998939 - -0.00991914 - -0.00965003 - -0.00919821 - -0.00859011 - -0.00785987 - -0.00704619 - -0.00618894 - -0.005326 - -0.00449065 - -0.00370973 - -0.00300262 - -0.00238113 - -0.00185009 - -0.0014084 - -0.00105048 - -0.000767675 - -0.000549659 - -0.0003856 - -0.000265039 - -0.000178489 - -0.000117771 - -7.61375e-05 - -4.82267e-05 - -2.99299e-05 - -1.81993e-05 - -1.08426e-05 - -6.32909e-06 - -3.61977e-06 - -2.0284e-06 - -1.11367e-06 - -5.99086e-07 - -3.15758e-07 - -1.63062e-07 - -8.25054e-08 - -4.0902e-08 - -1.98673e-08 - -9.45511e-09 - -4.40887e-09 - -2.01428e-09 - -9.01668e-10 - -3.95464e-10 - -1.69942e-10 - -7.15528e-11 - -2.9518e-11 - -1.19311e-11 - -4.72509e-12 - -1.83346e-12 - -6.97057e-13 - -2.59655e-13 - -9.47633e-14 - -3.3872e-14 - -1.18181e-14 - -3.90035e-15 - -8.29272e-16 - 1.18263e-15 - 4.68453e-15 - 1.39394e-14 - 3.96556e-14 - 1.10235e-13 - 3.00124e-13 - 8.00503e-13 - 2.0918e-12 - 5.35517e-12 - 1.34315e-11 - 3.30047e-11 - 7.94561e-11 - 1.87405e-10 - 4.33047e-10 - 9.80379e-10 - 2.17449e-09 - 4.72525e-09 - 1.006e-08 - 2.09837e-08 - 4.28816e-08 - 8.58558e-08 - 1.68414e-07 - 3.23665e-07 - 6.09432e-07 - 1.12426e-06 - 2.03198e-06 - 3.59822e-06 - 6.24264e-06 - 1.06112e-05 - 1.76715e-05 - 2.88336e-05 - 4.60937e-05 - 7.21939e-05 - 0.000110784 - 0.00016656 - 0.000245348 - 0.00035409 - 0.000500684 - 0.000693638 - 0.000941505 - 0.00125208 - 0.0016314 - 0.00208262 - 0.00260483 - 0.00319202 - 0.00383237 - 0.00450799 - 0.00519525 - 0.00586582 - 0.00648837 - 0.00703083 - 0.0074629 - 0.00775862 - 0.00789868 - 0.00787201 - 0.00767662 - 0.00731945 - 0.00681534 - 0.00618516 - 0.0054536 - 0.00464667 - 0.00378961 - 0.00290508 - 0.00201228 - 0.00112666 - 0.000260383 - -0.000576704 - -0.00137586 - -0.00212829 - -0.0028243 - -0.0034529 - -0.00400205 - -0.00445942 - -0.00481357 - -0.0050554 - -0.00517947 - -0.00518511 - -0.00507705 - -0.00486543 - -0.00456522 - -0.00419512 - -0.00377605 - -0.00332958 - -0.00287625 - -0.00243427 - -0.00201849 - -0.00163986 - -0.00130532 - -0.00101802 - -0.000777914 - -0.000582426 - -0.000427253 - -0.000307089 - -0.000216263 - -0.000149222 - -0.000100884 - -6.68267e-05 - -4.33723e-05 - -2.75812e-05 - -1.7185e-05 - -1.04911e-05 - -6.27527e-06 - -3.67773e-06 - -2.11185e-06 - -1.18819e-06 - -6.55e-07 - -3.53781e-07 - -1.87222e-07 - -9.70701e-08 - -4.92929e-08 - -2.44779e-08 - -1.17909e-08 - -5.27448e-09 - -1.60946e-09 - 1.27482e-09 - 5.18955e-09 - 1.26581e-08 - 2.82967e-08 - 6.12173e-08 - 1.29452e-07 - 2.68086e-07 - 5.43925e-07 - 1.08126e-06 - 2.10601e-06 - 4.01906e-06 - 7.51496e-06 - 1.37678e-05 - 2.47139e-05 - 4.34663e-05 - 7.49034e-05 - 0.00012647 - 0.000209223 - 0.000339133 - 0.0005386 - 0.000838107 - 0.00127782 - 0.00190887 - 0.00279396 - 0.00400683 - 0.00563014 - 0.00775128 - 0.010456 - 0.0138195 - 0.0178961 - 0.022707 - 0.0282292 - 0.0343855 - 0.0410382 - 0.0479888 - 0.0549831 - 0.0617247 - 0.0678938 - 0.0731722 - 0.0772704 - 0.0799544 - 0.0810684 - 0.0805512 - 0.0784438 - 0.0748868 - 0.0701092 - 0.0644092 - 0.0581302 - 0.0516357 - 0.0452844 - 0.0394097 - 0.0343041 - 0.0302089 - 0.0273084 - 0.0257283 - 0.0255355 - 0.0267383 - 0.0292872 - 0.0330737 - 0.0379307 - 0.0436337 - 0.0499051 - 0.0564229 - 0.0628345 - 0.0687764 - 0.073897 - 0.0778824 - 0.0804804 - 0.0815217 - 0.0809336 - 0.0787451 - 0.0750827 - 0.0701565 - 0.0642396 - 0.0576423 - 0.0506847 - 0.0436712 - 0.0368695 - 0.030495 - 0.024703 - 0.0195869 - 0.0151828 - 0.0114776 - 0.00841987 - 0.00593219 - 0.00392215 - 0.00229274 - 0.000950838 - -0.000186034 - -0.00118449 - -0.00209138 - -0.00293083 - -0.00370225 - -0.00437912 - -0.00490892 - -0.00521447 - -0.00519708 - -0.00474199 - -0.00372615 - -0.00202842 - 0.000458113 - 0.00381264 - 0.00807298 - 0.0132233 - 0.0191849 - 0.0258114 - 0.0328904 - 0.0401515 - 0.047281 - 0.0539439 - 0.059808 - 0.0645718 - 0.0679897 - 0.0698937 - 0.0702075 - 0.0689522 - 0.0662425 - 0.062274 - 0.0573033 - 0.0516237 - 0.0455395 - 0.0393419 - 0.0332886 - 0.0275894 - 0.0223987 - 0.017814 - 0.0138795 - 0.0105943 - 0.00792264 - 0.0058046 - 0.00416665 - 0.00293035 - 0.00201918 - 0.00136318 - 0.000901698 - 0.000584386 - 0.000371087 - 0.000230892 - 0.000140786 - 8.41686e-05 - 4.94262e-05 - 2.86893e-05 - 1.68175e-05 - 1.06308e-05 - 8.37418e-06 - 9.39916e-06 - 1.40463e-05 - 2.3726e-05 - 4.1207e-05 - 7.11357e-05 - 0.000120818 - 0.000201294 - 0.000328719 - 0.000526031 - 0.000824827 - 0.00126728 - 0.00190781 - 0.00281418 - 0.00406746 - 0.00576033 - 0.00799326 - 0.0108681 - 0.014479 - 0.0189005 - 0.0241747 - 0.0302971 - 0.0372043 - 0.044765 - 0.0527758 - 0.0609654 - 0.0690054 - 0.0765305 - 0.0831643 - 0.0885505 - 0.0923838 - 0.094439 - 0.0945928 - 0.0928357 - 0.0892735 - 0.0841163 - 0.0776584 - 0.0702501 - 0.0622666 - 0.0540773 - 0.0460179 - 0.0383702 - 0.0313489 - 0.0250975 - 0.0196902 - 0.0151414 - 0.0114171 - 0.00844929 - 0.00614968 - 0.00442187 - 0.00317137 - 0.00231276 - 0.00177424 - 0.00149973 - 0.00144902 - 0.00159637 - 0.00192796 - 0.00243858 - 0.00312782 - 0.00399592 - 0.00503982 - 0.00624939 - 0.00760438 - 0.00907248 - 0.0106086 - 0.0121559 - 0.0136483 - 0.0150148 - 0.0161846 - 0.0170932 - 0.0176882 - 0.0179342 - 0.0178162 - 0.0173415 - 0.0165385 - 0.0154541 - 0.0141492 - 0.012693 - 0.0111571 - 0.00960964 - 0.00811103 - 0.00671036 - 0.00544389 - 0.00433493 - 0.00339517 - 0.00262695 - 0.0020262 - 0.00158568 - 0.00129798 - 0.00115831 - 0.0011665 - 0.00132846 - 0.00165655 - 0.00216916 - 0.00288914 - 0.00384126 - 0.00504868 - 0.00652871 - 0.00828802 - 0.0103181 - 0.012591 - 0.015057 - 0.0176439 - 0.0202583 - 0.0227906 - 0.0251218 - 0.0271321 - 0.0287114 - 0.0297689 - 0.0302419 - 0.0301019 - 0.0293576 - 0.0280537 - 0.0262672 - 0.0240992 - 0.0216666 - 0.0190912 - 0.0164912 - 0.0139727 - 0.0116248 - 0.00951633 - 0.00769657 - 0.00619684 - 0.00503464 - 0.00421823 - 0.00375123 - 0.00363649 - 0.00387863 - 0.00448478 - 0.00546347 - 0.00682166 - 0.00856033 - 0.0106692 - 0.0131211 - 0.0158675 - 0.0188355 - 0.0219272 - 0.0250222 - 0.0279839 - 0.0306674 - 0.0329313 - 0.034649 - 0.0357203 - 0.0360812 - 0.0357095 - 0.0346278 - 0.0329006 - 0.030628 - 0.0279365 - 0.0249667 - 0.0218619 - 0.0187564 - 0.015767 - 0.0129863 - 0.01048 - 0.00828649 - 0.00641975 - 0.00487306 - 0.00362429 - 0.00264108 - 0.00188571 - 0.00131919 - 0.000904223 - 0.000607268 - 0.000399597 - 0.000257633 - 0.000162749 - 0.000100733 - 6.10885e-05 - 3.62982e-05 - 2.11324e-05 - 1.20545e-05 - 6.7373e-06 - 3.68943e-06 - 1.97957e-06 - 1.04068e-06 - 5.36045e-07 - 2.70534e-07 - 1.33777e-07 - 6.4815e-08 - 3.07685e-08 - 1.43112e-08 - 6.522e-09 - 2.91221e-09 - 1.2741e-09 - 5.46157e-10 - 2.29388e-10 - 9.43974e-11 - 3.80615e-11 - 1.50366e-11 - 5.82036e-12 - 2.20743e-12 - 8.20277e-13 - 2.98656e-13 - 1.06541e-13 - 3.72394e-14 - 1.27533e-14 - 4.27939e-15 - 1.40694e-15 - 4.5322e-16 - 1.43046e-16 - 4.42367e-17 - 1.34037e-17 - 3.97927e-18 - 1.1575e-18 - 3.29893e-19 - 9.21218e-20 - 2.52051e-20 - 6.75697e-21 - 1.77481e-21 - 4.56761e-22 - 1.15176e-22 - 2.8456e-23 - 6.88845e-24 - 1.63383e-24 - 3.79688e-25 - 8.64541e-26 - 1.92877e-26 - 4.21611e-27 - 9.02986e-28 - 1.8949e-28 - 3.89609e-29 - 7.84888e-30 - 1.54926e-30 - 2.99624e-31 - 5.67762e-32 - 1.05413e-32 - 1.91759e-33 - 3.41788e-34 - 5.9689e-35 - 1.02133e-35 - 1.7123e-36 - 2.81272e-37 - 4.52701e-38 - 7.13892e-39 - 1.10304e-39 - 1.66988e-40 - 2.47695e-41 - 3.59987e-42 - 5.12616e-43 - 7.15212e-44 - 9.77719e-45 - 1.30957e-45 - 1.71863e-46 - 2.2099e-47 - 2.78419e-48 - 3.43687e-49 - 4.15684e-50 - 4.92607e-51 - 5.71972e-52 - 6.50707e-53 - 7.25326e-54 - 7.92169e-55 - 8.47695e-56 - 8.88787e-57 - 9.13047e-58 - 9.1902e-59 - 9.06346e-60 - 8.75789e-61 - 8.29167e-62 - 7.69169e-63 - 6.99098e-64 - 6.22574e-65 - 5.43226e-66 - 4.64417e-67 - 3.89019e-68 - 3.1928e-69 - 2.56749e-70 - 2.02294e-71 - 1.56169e-72 - 1.18125e-73 - 8.75441e-75 - 6.35695e-76 - 4.5228e-77 - 3.15284e-78 - 2.15345e-79 - 1.44113e-80 - 9.44951e-82 - 6.07088e-83 - 3.82148e-84 - 2.35693e-85 - 1.4243e-86 - 8.43317e-88 - 4.89235e-89 - 2.78087e-90 - 1.54875e-91 - 8.45122e-93 - 4.51849e-94 - 2.36704e-95 - 1.21493e-96 - 6.10995e-98 - 3.01064e-99 - 1.45351e-100 - 6.87564e-102 - 3.18673e-103 - 1.44715e-104 - 6.43903e-106 - 2.80713e-107 - 1.19907e-108 - 5.01875e-110 - 2.16259e-111 - 2.64582e-111 - 6.14557e-110 - 1.44359e-108 - 3.32264e-107 - 7.49309e-106 - 1.65567e-104 - 3.58448e-103 - 7.6035e-102 - 1.5803e-100 - 3.2181e-99 - 6.42094e-98 - 1.25526e-96 - 2.4044e-95 - 4.51248e-94 - 8.29776e-93 - 1.49501e-91 - 2.63914e-90 - 4.56477e-89 - 7.73593e-88 - 1.28452e-86 - 2.08982e-85 - 3.3313e-84 - 5.203e-83 - 7.96217e-82 - 1.19384e-80 - 1.75387e-79 - 2.52455e-78 - 3.56049e-77 - 4.92007e-76 - 6.66146e-75 - 8.837e-74 - 1.14862e-72 - 1.4628e-71 - 1.82529e-70 - 2.23159e-69 - 2.67322e-68 - 3.13755e-67 - 3.60815e-66 - 4.06551e-65 - 4.4883e-64 - 4.85496e-63 - 5.14548e-62 - 5.34323e-61 - 5.43648e-60 - 5.41963e-59 - 5.29368e-58 - 5.0662e-57 - 4.75055e-56 - 4.36458e-55 - 3.92897e-54 - 3.46538e-53 - 2.99475e-52 - 2.53575e-51 - 2.10372e-50 - 1.71005e-49 - 1.36196e-48 - 1.06282e-47 - 8.12623e-47 - 6.08774e-46 - 4.46848e-45 - 3.21367e-44 - 2.26453e-43 - 1.56348e-42 - 1.05766e-41 - 7.01025e-41 - 4.55259e-40 - 2.89682e-39 - 1.80601e-38 - 1.1032e-37 - 6.60281e-37 - 3.87203e-36 - 2.22477e-35 - 1.25247e-34 - 6.90859e-34 - 3.73376e-33 - 1.97716e-32 - 1.02582e-31 - 5.21484e-31 - 2.59744e-30 - 1.26762e-29 - 6.06133e-29 - 2.83978e-28 - 1.30358e-27 - 5.86312e-27 - 2.58379e-26 - 1.11564e-25 - 4.71981e-25 - 1.95643e-24 - 7.94588e-24 - 3.16197e-23 - 1.23285e-22 - 4.70977e-22 - 1.7629e-21 - 6.46539e-21 - 2.32327e-20 - 8.17978e-20 - 2.82178e-19 - 9.53769e-19 - 3.15866e-18 - 1.02495e-17 - 3.25866e-17 - 1.01512e-16 - 3.0984e-16 - 9.26612e-16 - 2.71519e-15 - 7.79551e-15 - 2.19297e-14 - 6.04456e-14 - 1.63246e-13 - 4.31982e-13 - 1.12005e-12 - 2.84549e-12 - 7.0832e-12 - 1.72764e-11 - 4.12889e-11 - 9.66876e-11 - 2.21855e-10 - 4.98805e-10 - 1.09891e-09 - 2.37226e-09 - 5.01812e-09 - 1.04016e-08 - 2.11277e-08 - 4.2053e-08 - 8.20252e-08 - 1.56787e-07 - 2.93696e-07 - 5.39164e-07 - 9.70047e-07 - 1.71052e-06 - 2.95629e-06 - 5.00803e-06 - 8.3159e-06 - 1.35363e-05 - 2.16005e-05 - 3.37937e-05 - 5.18379e-05 - 7.7972e-05 - 0.000115015 - 0.000166394 - 0.000236125 - 0.000328721 - 0.00044901 - 0.000601858 - 0.000791802 - 0.00102259 - 0.00129668 - 0.00161472 - 0.00197509 - 0.00237354 - 0.00280296 - 0.00325346 - 0.00371255 - 0.00416568 - 0.00459692 - 0.00498986 - 0.00532857 - 0.00559865 - 0.00578819 - 0.00588854 - 0.00589499 - 0.00580705 - 0.00562858 - 0.00536746 - 0.00503512 - 0.00464575 - 0.0042153 - 0.00376048 - 0.0032977 - 0.00284209 - 0.00240674 - 0.00200212 - 0.0016358 - - - - - -3.27653e-30 - -4.07756e-30 - -4.97191e-30 - -5.93995e-30 - -6.95311e-30 - -7.97462e-30 - -8.9612e-30 - -9.86511e-30 - -1.06339e-29 - -1.11942e-29 - -1.13563e-29 - -1.03235e-29 - -4.31313e-30 - 2.47085e-29 - 1.61195e-28 - 7.92547e-28 - 3.65898e-27 - 1.64159e-26 - 7.20437e-26 - 3.09686e-25 - 1.30423e-24 - 5.38172e-24 - 2.17581e-23 - 8.61905e-23 - 3.34529e-22 - 1.27217e-21 - 4.74016e-21 - 1.73052e-20 - 6.19012e-20 - 2.16949e-19 - 7.44995e-19 - 2.5066e-18 - 8.26332e-18 - 2.66907e-17 - 8.44702e-17 - 2.61929e-16 - 7.95792e-16 - 2.36893e-15 - 6.90944e-15 - 1.97456e-14 - 5.52884e-14 - 1.51682e-13 - 4.0773e-13 - 1.07386e-12 - 2.77114e-12 - 7.00658e-12 - 1.73577e-11 - 4.21321e-11 - 1.00201e-10 - 2.33489e-10 - 5.33089e-10 - 1.19253e-09 - 2.61381e-09 - 5.61329e-09 - 1.18113e-08 - 2.43508e-08 - 4.91889e-08 - 9.73549e-08 - 1.88793e-07 - 3.58714e-07 - 6.67805e-07 - 1.21811e-06 - 2.17702e-06 - 3.81218e-06 - 6.54066e-06 - 1.09953e-05 - 1.81104e-05 - 2.92272e-05 - 4.6215e-05 - 7.16003e-05 - 0.000108689 - 0.000161656 - 0.000235577 - 0.000336367 - 0.000470576 - 0.000645036 - 0.000866312 - 0.00113999 - 0.00146982 - 0.00185681 - 0.00229829 - 0.00278727 - 0.00331201 - 0.00385604 - 0.00439875 - 0.00491649 - 0.0053842 - 0.00577737 - 0.00607415 - 0.00625743 - 0.00631647 - 0.00624807 - 0.00605694 - 0.00575536 - 0.00536215 - 0.00490107 - 0.0043989 - 0.00388353 - 0.00338212 - 0.00291963 - 0.00251784 - 0.00219476 - 0.0019644 - 0.00183683 - 0.00181827 - 0.00191123 - 0.00211441 - 0.0024225 - 0.0028259 - 0.00331037 - 0.00385693 - 0.00444204 - 0.00503828 - 0.00561553 - 0.0061427 - 0.00658983 - 0.00693041 - 0.0071435 - 0.00721563 - 0.00714193 - 0.00692656 - 0.00658218 - 0.00612867 - 0.00559119 - 0.00499782 - 0.0043772 - 0.0037562 - 0.00315819 - 0.00260175 - 0.00210005 - 0.00166085 - 0.00128697 - 0.000977109 - 0.000726866 - 0.000529789 - 0.000378345 - 0.000264734 - 0.000181497 - 0.000121917 - 8.02412e-05 - 5.17448e-05 - 3.26944e-05 - 2.02403e-05 - 1.22771e-05 - 7.29648e-06 - 4.24881e-06 - 2.42414e-06 - 1.35514e-06 - 7.42249e-07 - 3.98337e-07 - 2.09454e-07 - 1.0791e-07 - 5.44722e-08 - 2.69416e-08 - 1.3056e-08 - 6.19913e-09 - 2.88397e-09 - 1.31458e-09 - 5.87109e-10 - 2.56914e-10 - 1.10153e-10 - 4.62739e-11 - 1.90459e-11 - 7.67917e-12 - 3.02905e-12 - 1.15766e-12 - 3.97361e-13 - 3.41409e-14 - -2.89899e-13 - -8.92159e-13 - -2.32794e-12 - -5.84648e-12 - -1.43519e-11 - -3.45078e-11 - -8.12912e-11 - -1.87631e-10 - -4.24327e-10 - -9.40231e-10 - -2.04129e-09 - -4.34222e-09 - -9.05016e-09 - -1.84815e-08 - -3.6979e-08 - -7.24953e-08 - -1.39252e-07 - -2.62077e-07 - -4.83275e-07 - -8.73166e-07 - -1.54574e-06 - -2.68109e-06 - -4.55642e-06 - -7.58706e-06 - -1.23783e-05 - -1.97871e-05 - -3.09915e-05 - -4.75597e-05 - -7.15111e-05 - -0.000105352 - -0.000152073 - -0.000215078 - -0.000298041 - -0.000404664 - -0.00053833 - -0.000701682 - -0.000896126 - -0.00112133 - -0.00137479 - -0.00165149 - -0.0019438 - -0.00224163 - -0.00253287 - -0.00280414 - -0.00304174 - -0.00323283 - -0.0033665 - -0.00343488 - -0.00343384 - -0.00336344 - -0.0032279 - -0.00303518 - -0.00279621 - -0.00252384 - -0.00223166 - -0.00193291 - -0.00163942 - -0.00136091 - -0.00110451 - -0.000874518 - -0.000672571 - -0.00049787 - -0.000347626 - -0.000217553 - -0.000102387 - 3.62246e-06 - 0.000106247 - 0.000210868 - 0.000322098 - 0.000443445 - 0.000577031 - 0.000723357 - 0.000881165 - 0.0010474 - 0.00121734 - 0.00138482 - 0.0015427 - 0.00168342 - 0.00179962 - 0.00188484 - 0.00193416 - 0.00194463 - 0.00191564 - 0.00184894 - 0.00174849 - 0.00162005 - 0.00147064 - 0.00130789 - 0.00113938 - 0.000972049 - 0.000811701 - 0.000662676 - 0.000527676 - 0.000407728 - 0.000302261 - 0.000209254 - 0.000125426 - 4.64271e-05 - -3.29605e-05 - -0.000118629 - -0.000216928 - -0.0003344 - -0.000477448 - -0.000651899 - -0.000862487 - -0.00111225 - -0.00140188 - -0.00172921 - -0.00208869 - -0.00247127 - -0.00286449 - -0.003253 - -0.00361946 - -0.0039458 - -0.00421463 - -0.00441082 - -0.00452289 - -0.00454411 - -0.00447319 - -0.00431444 - -0.00407725 - -0.00377526 - -0.00342502 - -0.00304451 - -0.0026516 - -0.00226274 - -0.0018919 - -0.00154988 - -0.00124405 - -0.000978389 - -0.000753916 - -0.000569208 - -0.000421072 - -0.000305196 - -0.000216739 - -0.000150811 - -0.000102817 - -6.86807e-05 - -4.49512e-05 - -2.8826e-05 - -1.81119e-05 - -1.11501e-05 - -6.72565e-06 - -3.97489e-06 - -2.30172e-06 - -1.30592e-06 - -7.25967e-07 - -3.95417e-07 - -2.11023e-07 - -1.10342e-07 - -5.65314e-08 - -2.83775e-08 - -1.39572e-08 - -6.72598e-09 - -3.17579e-09 - -1.46921e-09 - -6.65968e-10 - -2.95774e-10 - -1.28707e-10 - -5.48761e-11 - -2.29245e-11 - -9.38325e-12 - -3.76308e-12 - -1.47867e-12 - -5.69293e-13 - -2.14752e-13 - -7.93733e-14 - -2.87441e-14 - -1.01991e-14 - -3.54576e-15 - -1.2078e-15 - -4.03104e-16 - -1.31818e-16 - -4.22349e-17 - -1.32588e-17 - -4.07825e-18 - -1.22908e-18 - -3.62931e-19 - -1.05004e-19 - -2.9766e-20 - -8.26751e-21 - -2.24991e-21 - -5.99919e-22 - -1.56732e-22 - -4.01224e-23 - -1.00739e-23 - -2.52276e-24 - -8.0687e-25 - -1.016e-24 - -3.58361e-24 - -1.41196e-23 - -5.4951e-23 - -2.09636e-22 - -7.83616e-22 - -2.86998e-21 - -1.02989e-20 - -3.62111e-20 - -1.24746e-19 - -4.21067e-19 - -1.39255e-18 - -4.5124e-18 - -1.43265e-17 - -4.45669e-17 - -1.35837e-16 - -4.05661e-16 - -1.18698e-15 - -3.40301e-15 - -9.55911e-15 - -2.63093e-14 - -7.09476e-14 - -1.87458e-13 - -4.85294e-13 - -1.23096e-12 - -3.05929e-12 - -7.4496e-12 - -1.77739e-11 - -4.15498e-11 - -9.51682e-11 - -2.13576e-10 - -4.69623e-10 - -1.01177e-09 - -2.13577e-09 - -4.41734e-09 - -8.9517e-09 - -1.77741e-08 - -3.45784e-08 - -6.59113e-08 - -1.23098e-07 - -2.25258e-07 - -4.03874e-07 - -7.09493e-07 - -1.2212e-06 - -2.05951e-06 - -3.40311e-06 - -5.50968e-06 - -8.74004e-06 - -1.35843e-05 - -2.06871e-05 - -3.08674e-05 - -4.51271e-05 - -6.46419e-05 - -9.07258e-05 - -0.000124764 - -0.00016811 - -0.000221944 - -0.000287112 - -0.000363934 - -0.000452037 - -0.000550211 - -0.000656332 - -0.000767384 - -0.000879594 - -0.000988694 - -0.00109031 - -0.00118047 - -0.00125612 - -0.00131572 - -0.00135978 - -0.00139119 - -0.00141553 - -0.00144105 - -0.00147842 - -0.00154024 - -0.00164022 - -0.00179215 - -0.0020087 - -0.00229996 - -0.00267211 - -0.00312609 - -0.00365653 - -0.00425123 - -0.00489099 - -0.00555038 - -0.00619899 - -0.00680353 - -0.00733032 - -0.00774812 - -0.00803088 - -0.00816018 - -0.00812687 - -0.00793199 - -0.0075865 - -0.00711016 - -0.00652951 - -0.00587539 - -0.00518015 - -0.00447502 - -0.00378787 - -0.0031416 - -0.00255317 - -0.00203343 - -0.00158747 - -0.0012155 - -0.000914005 - -0.000677014 - -0.000497327 - -0.000367638 - -0.00028146 - -0.000233825 - -0.000221778 - -0.000244658 - -0.000304193 - -0.000404393 - -0.000551247 - -0.0007522 - -0.00101539 - -0.00134866 - -0.00175834 - -0.00224794 - -0.00281677 - -0.00345876 - -0.00416153 - -0.00490608 - -0.00566706 - -0.00641387 - -0.00711248 - -0.00772785 - -0.00822684 - -0.00858114 - -0.00876987 - -0.0087817 - -0.0086159 - -0.00828247 - -0.0078011 - -0.00719927 - -0.00650966 - -0.00576719 - -0.0050062 - -0.00425783 - -0.00354817 - -0.00289707 - -0.00231766 - -0.00181667 - -0.00139522 - -0.00104989 - -0.000774071 - -0.000559185 - -0.000395792 - -0.000274483 - -0.00018651 - -0.000124172 - -8.09996e-05 - -5.17701e-05 - -3.24199e-05 - -1.98921e-05 - -1.19588e-05 - -7.04419e-06 - -4.06548e-06 - -2.29895e-06 - -1.27376e-06 - -6.91503e-07 - -3.67873e-07 - -1.91872e-07 - -9.83377e-08 - -5.00377e-08 - -2.64246e-08 - -1.68924e-08 - -1.70683e-08 - -2.72651e-08 - -5.27365e-08 - -1.05874e-07 - -2.1103e-07 - -4.13329e-07 - -7.93705e-07 - -1.49355e-06 - -2.75377e-06 - -4.97481e-06 - -8.80568e-06 - -1.52717e-05 - -2.59505e-05 - -4.32059e-05 - -7.04818e-05 - -0.000112654 - -0.000176423 - -0.000270707 - -0.000406987 - -0.000599513 - -0.000865274 - -0.00122362 - -0.00169541 - -0.00230165 - -0.00306155 - -0.00399008 - -0.00509515 - -0.00637485 - -0.00781483 - -0.00938655 - -0.0110466 - -0.0127377 - -0.0143909 - -0.0159302 - -0.0172779 - -0.0183611 - -0.0191181 - -0.0195041 - -0.0194959 - -0.0190941 - -0.0183227 - -0.0172274 - -0.0158703 - -0.0143247 - -0.0126685 - -0.0109775 - -0.00931998 - -0.00775291 - -0.00631905 - -0.00504632 - -0.00394853 - -0.00302714 - -0.00227387 - -0.00167355 - -0.00120683 - -0.000852687 - -0.000590298 - -0.000400396 - -0.0002661 - -0.000173275 - -0.000110552 - -6.91084e-05 - -4.23286e-05 - -2.54023e-05 - -1.49365e-05 - -8.60522e-06 - -4.85749e-06 - -2.68657e-06 - -1.45587e-06 - -7.73006e-07 - -4.02142e-07 - -2.04981e-07 - -1.02373e-07 - -5.00948e-08 - -2.4018e-08 - -1.12829e-08 - -5.19323e-09 - -2.34203e-09 - -1.03487e-09 - -4.48039e-10 - -1.90063e-10 - -7.90133e-11 - -3.2227e-11 - -1.2995e-11 - -5.44122e-12 - -3.02022e-12 - -3.51191e-12 - -7.21295e-12 - -1.70439e-11 - -4.05427e-11 - -9.4878e-11 - -2.17679e-10 - -4.89379e-10 - -1.07799e-09 - -2.32661e-09 - -4.92003e-09 - -1.01941e-08 - -2.06951e-08 - -4.11646e-08 - -8.02261e-08 - -1.53195e-07 - -2.86623e-07 - -5.25429e-07 - -9.43742e-07 - -1.66085e-06 - -2.8638e-06 - -4.83831e-06 - -8.00904e-06 - -1.29899e-05 - -2.06426e-05 - -3.21413e-05 - -4.9034e-05 - -7.3294e-05 - -0.000107344 - -0.000154036 - -0.000216573 - -0.000298347 - -0.000402696 - -0.000532562 - -0.00069008 - -0.000876123 - -0.00108985 - -0.00132834 - -0.0015863 - -0.00185609 - -0.00212789 - -0.00239021 - -0.00263063 - -0.00283674 - -0.00299722 - -0.00310279 - -0.0031472 - -0.00312775 - -0.00304563 - -0.00290576 - -0.0027163 - -0.0024879 - -0.00223268 - -0.00196316 - -0.0016913 - -0.00142766 - -0.00118077 - -0.000956844 - -0.000759722 - -0.000591024 - -0.000450497 - -0.000336447 - -0.000246194 - -0.000176512 - -0.000123997 - -8.53456e-05 - -5.75558e-05 - -3.80307e-05 - -2.46216e-05 - -1.56184e-05 - -9.70714e-06 - -5.91131e-06 - -3.52706e-06 - -2.06196e-06 - -1.18109e-06 - -6.62861e-07 - -3.64501e-07 - -1.96387e-07 - -1.03672e-07 - -5.36229e-08 - -2.71753e-08 - -1.34939e-08 - -6.56499e-09 - -3.12945e-09 - -1.46164e-09 - -6.6888e-10 - -2.99912e-10 - -1.31758e-10 - -5.67146e-11 - -2.39194e-11 - -9.88424e-12 - -4.00196e-12 - -1.58759e-12 - -6.17081e-13 - -2.35008e-13 - -8.76919e-14 - -3.20607e-14 - -1.14848e-14 - -4.03098e-15 - -1.38623e-15 - -4.67086e-16 - -1.54204e-16 - -4.98803e-17 - -1.58089e-17 - -4.90919e-18 - -1.49367e-18 - -4.45284e-19 - -1.30064e-19 - -3.72232e-20 - -1.04383e-20 - -2.87015e-21 - -7.81928e-22 - -2.43429e-22 - -2.08622e-22 - -6.12204e-22 - -2.2559e-21 - -8.30244e-21 - -2.99767e-20 - -1.06056e-19 - -3.67643e-19 - -1.24869e-18 - -4.15545e-18 - -1.35494e-17 - -4.32871e-17 - -1.35498e-16 - -4.15571e-16 - -1.2488e-15 - -3.67688e-15 - -1.06072e-14 - -2.99821e-14 - -8.30344e-14 - -2.25315e-13 - -5.99047e-13 - -1.56051e-12 - -3.98301e-12 - -9.96075e-12 - -2.44067e-11 - -5.85955e-11 - -1.37834e-10 - -3.17675e-10 - -7.1738e-10 - -1.58727e-09 - -3.44105e-09 - -7.30916e-09 - -1.52118e-08 - -3.10193e-08 - -6.19757e-08 - -1.21324e-07 - -2.32709e-07 - -4.37337e-07 - -8.05303e-07 - -1.45292e-06 - -2.56841e-06 - -4.44867e-06 - -7.54986e-06 - -1.25543e-05 - -2.04549e-05 - -3.26554e-05 - -5.10821e-05 - -7.82973e-05 - -0.000117598 - -0.000173075 - -0.000249615 - -0.000352798 - -0.000488684 - -0.000663453 - -0.000882913 - -0.00115189 - -0.00147356 - -0.00184879 - -0.00227563 - -0.00274905 - -0.00326099 - -0.00380083 - -0.00435636 - -0.00491499 - -0.00546527 - -0.00599832 - -0.00650907 - -0.00699689 - -0.00746558 - -0.00792243 - -0.00837658 - -0.00883672 - -0.00930857 - -0.00979246 - -0.0102816 - -0.0107611 - -0.0112087 - -0.011596 - -0.0118916 - -0.0120645 - -0.0120879 - -0.0119427 - -0.0116203 - -0.011124 - -0.0104692 - -0.00968215 - -0.00879775 - -0.00785624 - -0.00689989 - -0.00596968 - -0.00510249 - -0.00432903 - -0.00367258 - -0.00314848 - -0.00276429 - -0.00252032 - -0.00241049 - -0.00242322 - -0.00254229 - -0.00274769 - -0.00301656 - -0.00332402 - -0.00364428 - -0.00395183 - -0.00422275 - -0.00443609 - -0.00457514 - -0.00462847 - -0.0045906 - -0.00446213 - -0.00424935 - -0.00396339 - -0.0036188 - -0.00323204 - -0.00281971 - -0.00239708 - -0.0019767 - -0.00156754 - -0.00117457 - -0.000798753 - -0.000437587 - -8.59164e-05 - 0.000262944 - 0.000615954 - 0.000979073 - 0.00135609 - 0.00174768 - 0.00215078 - 0.00255837 - 0.00295977 - 0.00334128 - 0.00368735 - 0.00398195 - 0.00421019 - 0.00435982 - 0.00442255 - 0.00439499 - 0.00427906 - 0.00408186 - 0.003815 - 0.00349351 - 0.00313447 - 0.00275551 - 0.00237344 - 0.00200306 - 0.00165638 - 0.00134211 - 0.00106565 - 0.000829314 - 0.000632838 - 0.00047399 - 0.000349261 - 0.00025451 - 0.000185531 - 0.00013851 - 0.000110364 - 9.89575e-05 - 0.000103216 - 0.000123152 - 0.000159795 - 0.000215048 - 0.000291453 - 0.000391873 - 0.000519088 - 0.000675318 - 0.000861708 - 0.00107781 - 0.00132113 - 0.00158679 - 0.00186744 - 0.00215336 - 0.00243292 - 0.00269324 - 0.0029212 - 0.00310445 - 0.00323254 - 0.00329792 - 0.00329666 - 0.00322882 - 0.0030985 - 0.00291337 - 0.00268396 - 0.00242267 - 0.00214264 - 0.00185669 - 0.00157641 - 0.0013114 - 0.0010689 - 0.00085364 - 0.00066796 - 0.00051211 - 0.000384691 - 0.000283138 - 0.000204184 - 0.000144272 - 9.98799e-05 - 6.77504e-05 - 4.5028e-05 - 2.93217e-05 - 1.87083e-05 - 1.16954e-05 - 7.16366e-06 - 4.29928e-06 - 2.52821e-06 - 1.45697e-06 - 8.23301e-07 - 4.57272e-07 - 2.52047e-07 - 1.4307e-07 - 9.41611e-08 - 8.92735e-08 - 1.29198e-07 - 2.33164e-07 - 4.46224e-07 - 8.54461e-07 - 1.61153e-06 - 2.98178e-06 - 5.40736e-06 - 9.60873e-06 - 1.67298e-05 - 2.854e-05 - 4.77041e-05 - 7.81257e-05 - 0.000125363 - 0.000197097 - 0.000303619 - 0.000458263 - 0.0006777 - 0.000981967 - 0.0013941 - 0.00193921 - 0.00264299 - 0.00352941 - 0.00461791 - 0.00592006 - 0.00743607 - 0.0091516 - 0.0110354 - 0.0130381 - 0.0150932 - 0.0171191 - 0.0190248 - 0.0207155 - 0.0221008 - 0.0231024 - 0.0236615 - 0.0237446 - 0.0233467 - 0.0224919 - 0.0212307 - 0.0196357 - 0.0177941 - 0.0158004 - 0.0137482 - 0.0117235 - 0.00979954 - 0.00803325 - 0.00646427 - 0.00511562 - 0.00399594 - 0.00310269 - 0.00242575 - 0.00195092 - 0.00166287 - 0.00154732 - 0.00159223 - 0.00178801 - 0.00212689 - 0.00260153 - 0.00320325 - 0.00392002 - 0.00473471 - 0.00562383 - 0.006557 - 0.00749741 - 0.00840331 - 0.00923048 - 0.00993535 - 0.0104786 - 0.0108286 - 0.0109643 - 0.0108776 - 0.0105736 - 0.0100705 - 0.00939751 - 0.00859238 - 0.00769753 - 0.00675657 - 0.00581083 - 0.00489651 - 0.0040427 - 0.00327035 - 0.00259211 - 0.00201302 - 0.00153173 - 0.00114196 - 0.000834175 - 0.000597036 - 0.000418679 - 0.000287673 - 0.000193667 - 0.000127749 - 8.25696e-05 - 5.23006e-05 - 3.24804e-05 - 1.98088e-05 - 1.19273e-05 - 7.21506e-06 - 4.61863e-06 - 3.52388e-06 - 3.67649e-06 - 5.15343e-06 - 8.3891e-06 - 1.42612e-05 - 2.42428e-05 - 4.06281e-05 - 6.68359e-05 - 0.000107789 - 0.000170354 - 0.000263818 - 0.000400329 - 0.00059524 - 0.000867232 - 0.0012381 - 0.00173205 - 0.0023745 - 0.00319011 - 0.00420035 - 0.0054206 - 0.00685696 - 0.00850339 - 0.0103395 - 0.0123294 - 0.0144223 - 0.0165546 - 0.0186545 - 0.0206468 - 0.0224604 - 0.0240341 - 0.0253228 - 0.026301 - 0.0269648 - 0.0273304 - 0.0274301 - 0.0273064 - 0.0270053 - 0.0265691 - 0.0260305 - 0.0254095 - 0.0247121 - 0.0239319 - 0.023054 - 0.0220604 - 0.0209356 - 0.0196714 - 0.0182707 - 0.0167493 - 0.0151347 - 0.0134645 - 0.0117821 - 0.0101327 - 0.00855887 - 0.00709693 - 0.0057744 - 0.00460875 - 0.00360736 - 0.00276847 - 0.00208292 - 0.00153623 - 0.00111069 - 0.000787407 - 0.000547808 - 0.000374892 - 0.000253994 - 0.000173243 - 0.000123774 - 9.98233e-05 - 9.87971e-05 - 0.000121408 - 0.000171927 - 0.000258572 - 0.000393997 - 0.000595817 - 0.00088703 - 0.00129616 - 0.00185693 - 0.00260716 - 0.00358686 - 0.00483517 - 0.00638631 - 0.00826471 - 0.0104795 - 0.0130195 - 0.0158483 - 0.018902 - 0.0220888 - 0.0252913 - 0.0283731 - 0.0311875 - 0.0335885 - 0.0354435 - 0.0366455 - 0.0371228 - 0.0368467 - 0.0358338 - 0.0341447 - 0.031878 - 0.0291606 - 0.0261359 - 0.0229518 - 0.0197484 - 0.0166488 - 0.0137522 - 0.0111301 - 0.00882594 - 0.00685741 - 0.00522031 - 0.00389377 - 0.00284564 - 0.00203764 - 0.00142959 - 0.000982723 - 0.000661894 - 0.0004368 - 0.000282432 - 0.000178929 - 0.000111067 - 6.75505e-05 - 4.02538e-05 - 2.35029e-05 - 1.34454e-05 - 7.53639e-06 - 4.13894e-06 - 2.22716e-06 - 1.17423e-06 - 6.0658e-07 - 3.07016e-07 - 1.52255e-07 - 7.3981e-08 - 3.52217e-08 - 1.64312e-08 - 7.51334e-09 - 3.37362e-09 - 1.50265e-09 - 7.0035e-10 - 4.24077e-10 - 4.75335e-10 - 8.76813e-10 - 1.8765e-09 - 4.0748e-09 - 8.72522e-09 - 1.83265e-08 - 3.77229e-08 - 7.60823e-08 - 1.5035e-07 - 2.91111e-07 - 5.5227e-07 - 1.02655e-06 - 1.86959e-06 - 3.33619e-06 - 5.83299e-06 - 9.99237e-06 - 1.67719e-05 - 2.75825e-05 - 4.44448e-05 - 7.0169e-05 - 0.000108544 - 0.000164515 - 0.000244309 - 0.000355477 - 0.000506781 - 0.00070789 - 0.000968832 - 0.00129918 - 0.00170696 - 0.00219744 - 0.0027717 - 0.00342542 - 0.00414779 - 0.00492104 - 0.00572051 - 0.00651551 - 0.00727109 - 0.00795037 - 0.00851749 - 0.00894073 - 0.00919542 - 0.00926631 - 0.0091491 - 0.0088509 - 0.00838944 - 0.0077914 - 0.00708982 - 0.00632108 - 0.00552185 - 0.00472622 - 0.00396352 - 0.00325675 - 0.00262195 - 0.00206825 - 0.00159851 - 0.00121051 - 0.000898163 - 0.00065295 - 0.000465095 - 0.000324594 - 0.00022196 - 0.000148712 - 9.76239e-05 - 6.27917e-05 - 3.95717e-05 - 2.44345e-05 - 1.47829e-05 - 8.76302e-06 - 5.0896e-06 - 2.89635e-06 - 1.61493e-06 - 8.82256e-07 - 4.7225e-07 - 2.47678e-07 - 1.27274e-07 - 6.40809e-08 - 3.1613e-08 - 1.52825e-08 - 7.24349e-09 - 3.37597e-09 - 1.57127e-09 - 7.87334e-10 - 5.49235e-10 - 7.08757e-10 - 1.36716e-09 - 2.92932e-09 - 6.31231e-09 - 1.33936e-08 - 2.78707e-08 - 5.68338e-08 - 1.13558e-07 - 2.22314e-07 - 4.26437e-07 - 8.01454e-07 - 1.47584e-06 - 2.66279e-06 - 4.7073e-06 - 8.15349e-06 - 1.38373e-05 - 2.3009e-05 - 3.74869e-05 - 5.98409e-05 - 9.35953e-05 - 0.000143432 - 0.000215365 - 0.000316842 - 0.000456715 - 0.000645038 - 0.000892611 - 0.00121025 - 0.00160778 - 0.00209273 - 0.00266894 - 0.00333503 - 0.00408317 - 0.00489815 - 0.0057571 - 0.00662998 - 0.00748096 - 0.00827066 - 0.008959 - 0.00950859 - 0.00988805 - 0.010075 - 0.0100581 - 0.00983853 - 0.00942953 - 0.00885532 - 0.00814865 - 0.00734787 - 0.0064935 - 0.00562505 - 0.0047782 - 0.00398281 - 0.00326164 - 0.00263011 - 0.00209661 - 0.00166348 - 0.00132822 - 0.00108488 - 0.000925197 - 0.000839661 - 0.000818171 - 0.000850452 - 0.000926225 - 0.00103522 - 0.00116715 - 0.00131169 - 0.00145858 - 0.00159787 - 0.00172029 - 0.00181768 - 0.00188348 - 0.00191318 - 0.00190459 - 0.00185802 - 0.0017761 - 0.00166357 - 0.00152673 - 0.00137286 - 0.00120957 - 0.00104417 - 0.000883183 - 0.000731926 - 0.000594321 - 0.000472837 - 0.000368587 - 0.000281517 - 0.000210672 - 0.000154471 - 0.000110977 - 7.81216e-05 - 5.38892e-05 - 3.64365e-05 - 2.41667e-05 - 1.57607e-05 - 1.0179e-05 - 6.64527e-06 - 4.62443e-06 - 3.80367e-06 - 4.08572e-06 - 5.6009e-06 - 8.74399e-06 - 1.42407e-05 - 2.32471e-05 - 3.74834e-05 - 5.93981e-05 - 9.23544e-05 - 0.000140822 - 0.000210543 - 0.000308644 - 0.000443633 - 0.000625244 - 0.000864061 - 0.0011709 - 0.00155593 - 0.0020275 - 0.00259089 - 0.00324684 - 0.00399034 - 0.00480954 - 0.00568528 - 0.00659119 - 0.00749458 - 0.00835814 - 0.00914232 - 0.00980831 - 0.0103211 - 0.0106527 - 0.0107844 - 0.0107086 - 0.0104298 - 0.00996373 - 0.00933625 - 0.00858073 - 0.00773524 - 0.00683939 - 0.0059313 - 0.00504505 - 0.00420879 - 0.00344364 - 0.00276337 - 0.00217477 - 0.00167855 - 0.00127055 - 0.000943173 - 0.000686672 - 0.00049038 - 0.000343661 - 0.000236618 - 0.00016055 - 0.000108187 - 7.37676e-05 - 5.3005e-05 - 4.29873e-05 - 4.20568e-05 - 4.9689e-05 - 6.63831e-05 - 9.35609e-05 - 0.000133462 - 0.000189012 - 0.000263645 - 0.00036105 - 0.000484834 - 0.000638098 - 0.000822944 - 0.00103994 - 0.00128764 - 0.00156213 - 0.00185686 - 0.00216261 - 0.00246782 - 0.00275923 - 0.00302271 - 0.00324447 - 0.00341214 - 0.00351599 - 0.00354981 - 0.00351155 - 0.00340353 - 0.0032322 - 0.00300748 - 0.00274185 - 0.00244919 - 0.00214357 - 0.00183819 - 0.00154447 - 0.00127146 - 0.00102557 - 0.000810524 - 0.000627627 - 0.000476183 - 0.000353982 - 0.000257822 - 0.000183984 - 0.000128626 - 8.80794e-05 - 5.90389e-05 - 3.86628e-05 - 2.45949e-05 - 1.49291e-05 - 8.13403e-06 - 2.94857e-06 - -1.74116e-06 - -7.04209e-06 - -1.42113e-05 - -2.48328e-05 - -4.10146e-05 - -6.56039e-05 - -0.000102411 - -0.000156423 - -0.000233969 - -0.000342807 - -0.000492057 - -0.000691949 - -0.000953302 - -0.00128673 - -0.00170157 - -0.00220453 - -0.00279826 - -0.00347991 - -0.00423992 - -0.00506123 - -0.00591925 - -0.0067825 - -0.00761422 - -0.00837483 - -0.00902489 - -0.00952848 - -0.00985649 - -0.00998939 - -0.00991914 - -0.00965003 - -0.00919821 - -0.00859011 - -0.00785987 - -0.00704619 - -0.00618894 - -0.005326 - -0.00449065 - -0.00370973 - -0.00300262 - -0.00238113 - -0.00185009 - -0.0014084 - -0.00105048 - -0.000767675 - -0.000549659 - -0.0003856 - -0.000265039 - -0.000178489 - -0.000117771 - -7.61375e-05 - -4.82267e-05 - -2.99299e-05 - -1.81993e-05 - -1.08426e-05 - -6.32909e-06 - -3.61977e-06 - -2.0284e-06 - -1.11367e-06 - -5.99086e-07 - -3.15758e-07 - -1.63062e-07 - -8.25054e-08 - -4.0902e-08 - -1.98673e-08 - -9.45511e-09 - -4.40887e-09 - -2.01428e-09 - -9.01668e-10 - -3.95464e-10 - -1.69942e-10 - -7.15528e-11 - -2.9518e-11 - -1.19311e-11 - -4.72509e-12 - -1.83346e-12 - -6.97057e-13 - -2.59655e-13 - -9.47633e-14 - -3.3872e-14 - -1.18181e-14 - -3.90035e-15 - -8.29272e-16 - 1.18263e-15 - 4.68453e-15 - 1.39394e-14 - 3.96556e-14 - 1.10235e-13 - 3.00124e-13 - 8.00503e-13 - 2.0918e-12 - 5.35517e-12 - 1.34315e-11 - 3.30047e-11 - 7.94561e-11 - 1.87405e-10 - 4.33047e-10 - 9.80379e-10 - 2.17449e-09 - 4.72525e-09 - 1.006e-08 - 2.09837e-08 - 4.28816e-08 - 8.58558e-08 - 1.68414e-07 - 3.23665e-07 - 6.09432e-07 - 1.12426e-06 - 2.03198e-06 - 3.59822e-06 - 6.24264e-06 - 1.06112e-05 - 1.76715e-05 - 2.88336e-05 - 4.60937e-05 - 7.21939e-05 - 0.000110784 - 0.00016656 - 0.000245348 - 0.00035409 - 0.000500684 - 0.000693638 - 0.000941505 - 0.00125208 - 0.0016314 - 0.00208262 - 0.00260483 - 0.00319202 - 0.00383237 - 0.00450799 - 0.00519525 - 0.00586582 - 0.00648837 - 0.00703083 - 0.0074629 - 0.00775862 - 0.00789868 - 0.00787201 - 0.00767662 - 0.00731945 - 0.00681534 - 0.00618516 - 0.0054536 - 0.00464667 - 0.00378961 - 0.00290508 - 0.00201228 - 0.00112666 - 0.000260383 - -0.000576704 - -0.00137586 - -0.00212829 - -0.0028243 - -0.0034529 - -0.00400205 - -0.00445942 - -0.00481357 - -0.0050554 - -0.00517947 - -0.00518511 - -0.00507705 - -0.00486543 - -0.00456522 - -0.00419512 - -0.00377605 - -0.00332958 - -0.00287625 - -0.00243427 - -0.00201849 - -0.00163986 - -0.00130532 - -0.00101802 - -0.000777914 - -0.000582426 - -0.000427253 - -0.000307089 - -0.000216263 - -0.000149222 - -0.000100884 - -6.68267e-05 - -4.33723e-05 - -2.75812e-05 - -1.7185e-05 - -1.04911e-05 - -6.27527e-06 - -3.67773e-06 - -2.11185e-06 - -1.18819e-06 - -6.55e-07 - -3.53781e-07 - -1.87222e-07 - -9.70701e-08 - -4.92929e-08 - -2.44779e-08 - -1.17909e-08 - -5.27448e-09 - -1.60946e-09 - 1.27482e-09 - 5.18955e-09 - 1.26581e-08 - 2.82967e-08 - 6.12173e-08 - 1.29452e-07 - 2.68086e-07 - 5.43925e-07 - 1.08126e-06 - 2.10601e-06 - 4.01906e-06 - 7.51496e-06 - 1.37678e-05 - 2.47139e-05 - 4.34663e-05 - 7.49034e-05 - 0.00012647 - 0.000209223 - 0.000339133 - 0.0005386 - 0.000838107 - 0.00127782 - 0.00190887 - 0.00279396 - 0.00400683 - 0.00563014 - 0.00775128 - 0.010456 - 0.0138195 - 0.0178961 - 0.022707 - 0.0282292 - 0.0343855 - 0.0410382 - 0.0479888 - 0.0549831 - 0.0617247 - 0.0678938 - 0.0731722 - 0.0772704 - 0.0799544 - 0.0810684 - 0.0805512 - 0.0784438 - 0.0748868 - 0.0701092 - 0.0644092 - 0.0581302 - 0.0516357 - 0.0452844 - 0.0394097 - 0.0343041 - 0.0302089 - 0.0273084 - 0.0257283 - 0.0255355 - 0.0267383 - 0.0292872 - 0.0330737 - 0.0379307 - 0.0436337 - 0.0499051 - 0.0564229 - 0.0628345 - 0.0687764 - 0.073897 - 0.0778824 - 0.0804804 - 0.0815217 - 0.0809336 - 0.0787451 - 0.0750827 - 0.0701565 - 0.0642396 - 0.0576423 - 0.0506847 - 0.0436712 - 0.0368695 - 0.030495 - 0.024703 - 0.0195869 - 0.0151828 - 0.0114776 - 0.00841987 - 0.00593219 - 0.00392215 - 0.00229274 - 0.000950838 - -0.000186034 - -0.00118449 - -0.00209138 - -0.00293083 - -0.00370225 - -0.00437912 - -0.00490892 - -0.00521447 - -0.00519708 - -0.00474199 - -0.00372615 - -0.00202842 - 0.000458113 - 0.00381264 - 0.00807298 - 0.0132233 - 0.0191849 - 0.0258114 - 0.0328904 - 0.0401515 - 0.047281 - 0.0539439 - 0.059808 - 0.0645718 - 0.0679897 - 0.0698937 - 0.0702075 - 0.0689522 - 0.0662425 - 0.062274 - 0.0573033 - 0.0516237 - 0.0455395 - 0.0393419 - 0.0332886 - 0.0275894 - 0.0223987 - 0.017814 - 0.0138795 - 0.0105943 - 0.00792264 - 0.0058046 - 0.00416665 - 0.00293035 - 0.00201918 - 0.00136318 - 0.000901698 - 0.000584386 - 0.000371087 - 0.000230892 - 0.000140786 - 8.41686e-05 - 4.94262e-05 - 2.86893e-05 - 1.68175e-05 - 1.06308e-05 - 8.37418e-06 - 9.39916e-06 - 1.40463e-05 - 2.3726e-05 - 4.1207e-05 - 7.11357e-05 - 0.000120818 - 0.000201294 - 0.000328719 - 0.000526031 - 0.000824827 - 0.00126728 - 0.00190781 - 0.00281418 - 0.00406746 - 0.00576033 - 0.00799326 - 0.0108681 - 0.014479 - 0.0189005 - 0.0241747 - 0.0302971 - 0.0372043 - 0.044765 - 0.0527758 - 0.0609654 - 0.0690054 - 0.0765305 - 0.0831643 - 0.0885505 - 0.0923838 - 0.094439 - 0.0945928 - 0.0928357 - 0.0892735 - 0.0841163 - 0.0776584 - 0.0702501 - 0.0622666 - 0.0540773 - 0.0460179 - 0.0383702 - 0.0313489 - 0.0250975 - 0.0196902 - 0.0151414 - 0.0114171 - 0.00844929 - 0.00614968 - 0.00442187 - 0.00317137 - 0.00231276 - 0.00177424 - 0.00149973 - 0.00144902 - 0.00159637 - 0.00192796 - 0.00243858 - 0.00312782 - 0.00399592 - 0.00503982 - 0.00624939 - 0.00760438 - 0.00907248 - 0.0106086 - 0.0121559 - 0.0136483 - 0.0150148 - 0.0161846 - 0.0170932 - 0.0176882 - 0.0179342 - 0.0178162 - 0.0173415 - 0.0165385 - 0.0154541 - 0.0141492 - 0.012693 - 0.0111571 - 0.00960964 - 0.00811103 - 0.00671036 - 0.00544389 - 0.00433493 - 0.00339517 - 0.00262695 - 0.0020262 - 0.00158568 - 0.00129798 - 0.00115831 - 0.0011665 - 0.00132846 - 0.00165655 - 0.00216916 - 0.00288914 - 0.00384126 - 0.00504868 - 0.00652871 - 0.00828802 - 0.0103181 - 0.012591 - 0.015057 - 0.0176439 - 0.0202583 - 0.0227906 - 0.0251218 - 0.0271321 - 0.0287114 - 0.0297689 - 0.0302419 - 0.0301019 - 0.0293576 - 0.0280537 - 0.0262672 - 0.0240992 - 0.0216666 - 0.0190912 - 0.0164912 - 0.0139727 - 0.0116248 - 0.00951633 - 0.00769657 - 0.00619684 - 0.00503464 - 0.00421823 - 0.00375123 - 0.00363649 - 0.00387863 - 0.00448478 - 0.00546347 - 0.00682166 - 0.00856033 - 0.0106692 - 0.0131211 - 0.0158675 - 0.0188355 - 0.0219272 - 0.0250222 - 0.0279839 - 0.0306674 - 0.0329313 - 0.034649 - 0.0357203 - 0.0360812 - 0.0357095 - 0.0346278 - 0.0329006 - 0.030628 - 0.0279365 - 0.0249667 - 0.0218619 - 0.0187564 - 0.015767 - 0.0129863 - 0.01048 - 0.00828649 - 0.00641975 - 0.00487306 - 0.00362429 - 0.00264108 - 0.00188571 - 0.00131919 - 0.000904223 - 0.000607268 - 0.000399597 - 0.000257633 - 0.000162749 - 0.000100733 - 6.10885e-05 - 3.62982e-05 - 2.11324e-05 - 1.20545e-05 - 6.7373e-06 - 3.68943e-06 - 1.97957e-06 - 1.04068e-06 - 5.36045e-07 - 2.70534e-07 - 1.33777e-07 - 6.4815e-08 - 3.07685e-08 - 1.43112e-08 - 6.522e-09 - 2.91221e-09 - 1.2741e-09 - 5.46157e-10 - 2.29388e-10 - 9.43974e-11 - 3.80615e-11 - 1.50366e-11 - 5.82036e-12 - 2.20743e-12 - 8.20277e-13 - 2.98656e-13 - 1.06541e-13 - 3.72394e-14 - 1.27533e-14 - 4.27939e-15 - 1.40694e-15 - 4.5322e-16 - 1.43046e-16 - 4.42367e-17 - 1.34037e-17 - 3.97927e-18 - 1.1575e-18 - 3.29893e-19 - 9.21218e-20 - 2.52051e-20 - 6.75697e-21 - 1.77481e-21 - 4.56761e-22 - 1.15176e-22 - 2.8456e-23 - 6.88845e-24 - 1.63383e-24 - 3.79688e-25 - 8.64541e-26 - 1.92877e-26 - 4.21611e-27 - 9.02986e-28 - 1.8949e-28 - 3.89609e-29 - 7.84888e-30 - 1.54926e-30 - 2.99624e-31 - 5.67762e-32 - 1.05413e-32 - 1.91759e-33 - 3.41788e-34 - 5.9689e-35 - 1.02133e-35 - 1.7123e-36 - 2.81272e-37 - 4.52701e-38 - 7.13892e-39 - 1.10304e-39 - 1.66988e-40 - 2.47695e-41 - 3.59987e-42 - 5.12616e-43 - 7.15212e-44 - 9.77719e-45 - 1.30957e-45 - 1.71863e-46 - 2.2099e-47 - 2.78419e-48 - 3.43687e-49 - 4.15684e-50 - 4.92607e-51 - 5.71972e-52 - 6.50707e-53 - 7.25326e-54 - 7.92169e-55 - 8.47695e-56 - 8.88787e-57 - 9.13047e-58 - 9.1902e-59 - 9.06346e-60 - 8.75789e-61 - 8.29167e-62 - 7.69169e-63 - 6.99098e-64 - 6.22574e-65 - 5.43226e-66 - 4.64417e-67 - 3.89019e-68 - 3.1928e-69 - 2.56749e-70 - 2.02294e-71 - 1.56169e-72 - 1.18125e-73 - 8.75441e-75 - 6.35695e-76 - 4.5228e-77 - 3.15284e-78 - 2.15345e-79 - 1.44113e-80 - 9.44951e-82 - 6.07088e-83 - 3.82148e-84 - 2.35693e-85 - 1.4243e-86 - 8.43317e-88 - 4.89235e-89 - 2.78087e-90 - 1.54875e-91 - 8.45122e-93 - 4.51849e-94 - 2.36704e-95 - 1.21493e-96 - 6.10995e-98 - 3.01064e-99 - 1.45351e-100 - 6.87564e-102 - 3.18673e-103 - 1.44715e-104 - 6.43903e-106 - 2.80713e-107 - 1.19907e-108 - 5.01875e-110 - 2.16259e-111 - 2.64582e-111 - 6.14557e-110 - 1.44359e-108 - 3.32264e-107 - 7.49309e-106 - 1.65567e-104 - 3.58448e-103 - 7.6035e-102 - 1.5803e-100 - 3.2181e-99 - 6.42094e-98 - 1.25526e-96 - 2.4044e-95 - 4.51248e-94 - 8.29776e-93 - 1.49501e-91 - 2.63914e-90 - 4.56477e-89 - 7.73593e-88 - 1.28452e-86 - 2.08982e-85 - 3.3313e-84 - 5.203e-83 - 7.96217e-82 - 1.19384e-80 - 1.75387e-79 - 2.52455e-78 - 3.56049e-77 - 4.92007e-76 - 6.66146e-75 - 8.837e-74 - 1.14862e-72 - 1.4628e-71 - 1.82529e-70 - 2.23159e-69 - 2.67322e-68 - 3.13755e-67 - 3.60815e-66 - 4.06551e-65 - 4.4883e-64 - 4.85496e-63 - 5.14548e-62 - 5.34323e-61 - 5.43648e-60 - 5.41963e-59 - 5.29368e-58 - 5.0662e-57 - 4.75055e-56 - 4.36458e-55 - 3.92897e-54 - 3.46538e-53 - 2.99475e-52 - 2.53575e-51 - 2.10372e-50 - 1.71005e-49 - 1.36196e-48 - 1.06282e-47 - 8.12623e-47 - 6.08774e-46 - 4.46848e-45 - 3.21367e-44 - 2.26453e-43 - 1.56348e-42 - 1.05766e-41 - 7.01025e-41 - 4.55259e-40 - 2.89682e-39 - 1.80601e-38 - 1.1032e-37 - 6.60281e-37 - 3.87203e-36 - 2.22477e-35 - 1.25247e-34 - 6.90859e-34 - 3.73376e-33 - 1.97716e-32 - 1.02582e-31 - 5.21484e-31 - 2.59744e-30 - 1.26762e-29 - 6.06133e-29 - 2.83978e-28 - 1.30358e-27 - 5.86312e-27 - 2.58379e-26 - 1.11564e-25 - 4.71981e-25 - 1.95643e-24 - 7.94588e-24 - 3.16197e-23 - 1.23285e-22 - 4.70977e-22 - 1.7629e-21 - 6.46539e-21 - 2.32327e-20 - 8.17978e-20 - 2.82178e-19 - 9.53769e-19 - 3.15866e-18 - 1.02495e-17 - 3.25866e-17 - 1.01512e-16 - 3.0984e-16 - 9.26612e-16 - 2.71519e-15 - 7.79551e-15 - 2.19297e-14 - 6.04456e-14 - 1.63246e-13 - 4.31982e-13 - 1.12005e-12 - 2.84549e-12 - 7.0832e-12 - 1.72764e-11 - 4.12889e-11 - 9.66876e-11 - 2.21855e-10 - 4.98805e-10 - 1.09891e-09 - 2.37226e-09 - 5.01812e-09 - 1.04016e-08 - 2.11277e-08 - 4.2053e-08 - 8.20252e-08 - 1.56787e-07 - 2.93696e-07 - 5.39164e-07 - 9.70047e-07 - 1.71052e-06 - 2.95629e-06 - 5.00803e-06 - 8.3159e-06 - 1.35363e-05 - 2.16005e-05 - 3.37937e-05 - 5.18379e-05 - 7.7972e-05 - 0.000115015 - 0.000166394 - 0.000236125 - 0.000328721 - 0.00044901 - 0.000601858 - 0.000791802 - 0.00102259 - 0.00129668 - 0.00161472 - 0.00197509 - 0.00237354 - 0.00280296 - 0.00325346 - 0.00371255 - 0.00416568 - 0.00459692 - 0.00498986 - 0.00532857 - 0.00559865 - 0.00578819 - 0.00588854 - 0.00589499 - 0.00580705 - 0.00562858 - 0.00536746 - 0.00503512 - 0.00464575 - 0.0042153 - 0.00376048 - 0.0032977 - 0.00284209 - 0.00240674 - 0.00200212 - 0.0016358 - - - - - -1.59428e-33 - -1.98404e-33 - -2.41922e-33 - -2.89024e-33 - -3.38323e-33 - -3.8803e-33 - -4.36049e-33 - -4.80112e-33 - -5.17949e-33 - -5.4748e-33 - -5.67004e-33 - -5.75361e-33 - -5.72048e-33 - -5.57263e-33 - -5.31895e-33 - -4.97425e-33 - -4.55792e-33 - -4.09207e-33 - -3.59961e-33 - -3.10245e-33 - -2.61994e-33 - -2.16778e-33 - -1.75742e-33 - -1.39595e-33 - -1.08644e-33 - -8.28468e-34 - -6.18989e-34 - -4.53135e-34 - -3.25019e-34 - -2.28415e-34 - -1.57278e-34 - -1.06083e-34 - -6.9966e-35 - -4.44034e-35 - -2.30579e-35 - 1.38092e-35 - 1.59998e-34 - 8.95398e-34 - 4.65932e-33 - 2.36203e-32 - 1.17248e-31 - 5.70206e-31 - 2.717e-30 - 1.26849e-29 - 5.80254e-29 - 2.60068e-28 - 1.14207e-27 - 4.914e-27 - 2.07164e-26 - 8.55719e-26 - 3.46326e-25 - 1.37333e-24 - 5.33582e-24 - 2.03126e-23 - 7.57645e-23 - 2.76888e-22 - 9.91466e-22 - 3.47848e-21 - 1.19574e-20 - 4.02737e-20 - 1.32906e-19 - 4.29736e-19 - 1.36144e-18 - 4.226e-18 - 1.28528e-17 - 3.83005e-17 - 1.11827e-16 - 3.19909e-16 - 8.96693e-16 - 2.46262e-15 - 6.62654e-15 - 1.74709e-14 - 4.51313e-14 - 1.1423e-13 - 2.8328e-13 - 6.88321e-13 - 1.63871e-12 - 3.82252e-12 - 8.73645e-12 - 1.9564e-11 - 4.29255e-11 - 9.22807e-11 - 1.94376e-10 - 4.01155e-10 - 8.11183e-10 - 1.60717e-09 - 3.1199e-09 - 5.93414e-09 - 1.10589e-08 - 2.0193e-08 - 3.61267e-08 - 6.33275e-08 - 1.08766e-07 - 1.83034e-07 - 3.01791e-07 - 4.87548e-07 - 7.7173e-07 - 1.19688e-06 - 1.81875e-06 - 2.70789e-06 - 3.95027e-06 - 5.64623e-06 - 7.9073e-06 - 1.08501e-05 - 1.45874e-05 - 1.92157e-05 - 2.48013e-05 - 3.13637e-05 - 3.88613e-05 - 4.71786e-05 - 5.61189e-05 - 6.54049e-05 - 7.46875e-05 - 8.35646e-05 - 9.16081e-05 - 9.83971e-05 - 0.000103554 - 0.00010678 - 0.000107882 - 0.000106793 - 0.00010358 - 9.84341e-05 - 9.16541e-05 - 8.36171e-05 - 7.47438e-05 - 6.54624e-05 - 5.61753e-05 - 4.7232e-05 - 3.89102e-05 - 3.14071e-05 - 2.48387e-05 - 1.92471e-05 - 1.46131e-05 - 1.08706e-05 - 7.92321e-06 - 5.6583e-06 - 3.95921e-06 - 2.71436e-06 - 1.82332e-06 - 1.20004e-06 - 7.73865e-07 - 4.88958e-07 - 3.02702e-07 - 1.83609e-07 - 1.09122e-07 - 6.35427e-08 - 3.6254e-08 - 2.02667e-08 - 1.11006e-08 - 5.95729e-09 - 3.13247e-09 - 1.61384e-09 - 8.14654e-10 - 4.02923e-10 - 1.95257e-10 - 9.27106e-11 - 4.31309e-11 - 1.966e-11 - 8.78045e-12 - 3.84226e-12 - 1.64738e-12 - 6.92046e-13 - 2.84844e-13 - 1.14858e-13 - 4.53382e-14 - 1.74196e-14 - 6.23751e-15 - 1.31174e-15 - -2.20244e-15 - -7.77776e-15 - -2.0591e-14 - -5.18126e-14 - -1.27222e-13 - -3.05906e-13 - -7.20636e-13 - -1.66332e-12 - -3.76161e-12 - -8.33502e-12 - -1.80958e-11 - -3.84932e-11 - -8.02285e-11 - -1.63836e-10 - -3.27814e-10 - -6.42661e-10 - -1.23445e-09 - -2.32328e-09 - -4.28417e-09 - -7.7405e-09 - -1.37028e-08 - -2.37675e-08 - -4.03921e-08 - -6.72583e-08 - -1.09732e-07 - -1.7541e-07 - -2.74736e-07 - -4.21611e-07 - -6.33936e-07 - -9.33934e-07 - -1.3481e-06 - -1.90664e-06 - -2.6421e-06 - -3.58729e-06 - -4.77223e-06 - -6.22032e-06 - -7.94404e-06 - -9.94047e-06 - -1.21873e-05 - -1.46402e-05 - -1.72315e-05 - -1.98718e-05 - -2.24536e-05 - -2.48583e-05 - -2.69647e-05 - -2.86586e-05 - -2.98437e-05 - -3.04499e-05 - -3.04409e-05 - -2.9817e-05 - -2.8616e-05 - -2.69085e-05 - -2.47918e-05 - -2.23801e-05 - -1.97949e-05 - -1.71547e-05 - -1.45663e-05 - -1.21185e-05 - -9.87847e-06 - -7.88979e-06 - -6.17417e-06 - -4.734e-06 - -3.55643e-06 - -2.61781e-06 - -1.88798e-06 - -1.33412e-06 - -9.23695e-07 - -6.26613e-07 - -4.16492e-07 - -2.71238e-07 - -1.73074e-07 - -1.08205e-07 - -6.62806e-08 - -3.97758e-08 - -2.33778e-08 - -1.34388e-08 - -7.51457e-09 - -3.99226e-09 - -1.79761e-09 - -1.55833e-10 - 1.62647e-09 - 4.40583e-09 - 9.54109e-09 - 1.94299e-08 - 3.83984e-08 - 7.41803e-08 - 1.40332e-07 - 2.60078e-07 - 4.7225e-07 - 8.40183e-07 - 1.46457e-06 - 2.50142e-06 - 4.18598e-06 - 6.86349e-06 - 1.10263e-05 - 1.73561e-05 - 2.67677e-05 - 4.04488e-05 - 5.98878e-05 - 8.68775e-05 - 0.000123485 - 0.000171971 - 0.000234658 - 0.000313728 - 0.000410966 - 0.000527468 - 0.00066332 - 0.000817311 - 0.000986707 - 0.00116715 - 0.0013527 - 0.00153607 - 0.00170907 - 0.00186314 - 0.00199006 - 0.00208269 - 0.0021356 - 0.00214562 - 0.00211214 - 0.00203718 - 0.00192518 - 0.00178259 - 0.00161722 - 0.00143755 - 0.00125202 - 0.00106841 - 0.000893312 - 0.000731819 - 0.00058741 - 0.000461973 - 0.000355982 - 0.000268767 - 0.00019882 - 0.000144106 - 0.000102339 - 7.12094e-05 - 4.85478e-05 - 3.24294e-05 - 2.12249e-05 - 1.3611e-05 - 8.55202e-06 - 5.26484e-06 - 3.1757e-06 - 1.87685e-06 - 1.08682e-06 - 6.16624e-07 - 3.42785e-07 - 1.86707e-07 - 9.96401e-08 - 5.2101e-08 - 2.66928e-08 - 1.33992e-08 - 6.59024e-09 - 3.17585e-09 - 1.49953e-09 - 6.93727e-10 - 3.14455e-10 - 1.39658e-10 - 6.07726e-11 - 2.59112e-11 - 1.08244e-11 - 4.43055e-12 - 1.77684e-12 - 6.98194e-13 - 2.68807e-13 - 1.01401e-13 - 3.74782e-14 - 1.35723e-14 - 4.81577e-15 - 1.67423e-15 - 5.70295e-16 - 1.90336e-16 - 6.22416e-17 - 1.99424e-17 - 6.2605e-18 - 1.92566e-18 - 5.80343e-19 - 1.71367e-19 - 4.95802e-20 - 1.40548e-20 - 3.90372e-21 - 1.06236e-21 - 2.83268e-22 - 7.4006e-23 - 1.89476e-23 - 4.76861e-24 - 1.24276e-24 - 5.98879e-25 - 1.38172e-24 - 5.3505e-24 - 2.12055e-23 - 8.25555e-23 - 3.14952e-22 - 1.17729e-21 - 4.3118e-21 - 1.54729e-20 - 5.44027e-20 - 1.87416e-19 - 6.32602e-19 - 2.09214e-18 - 6.77934e-18 - 2.15239e-17 - 6.69563e-17 - 2.04079e-16 - 6.09457e-16 - 1.7833e-15 - 5.1126e-15 - 1.43614e-14 - 3.95265e-14 - 1.0659e-13 - 2.81632e-13 - 7.29096e-13 - 1.84937e-12 - 4.59621e-12 - 1.11921e-11 - 2.67031e-11 - 6.24235e-11 - 1.42979e-10 - 3.20872e-10 - 7.05551e-10 - 1.52007e-09 - 3.20873e-09 - 6.63652e-09 - 1.34488e-08 - 2.67034e-08 - 5.19498e-08 - 9.90237e-08 - 1.8494e-07 - 3.38423e-07 - 6.06771e-07 - 1.06593e-06 - 1.8347e-06 - 3.09416e-06 - 5.11276e-06 - 8.27761e-06 - 1.31308e-05 - 2.04087e-05 - 3.10796e-05 - 4.63738e-05 - 6.77965e-05 - 9.71133e-05 - 0.000136297 - 0.000187428 - 0.000252533 - 0.00033338 - 0.000431223 - 0.000546517 - 0.000678654 - 0.00082573 - 0.000984414 - 0.00114994 - 0.00131625 - 0.00147635 - 0.00162275 - 0.00174809 - 0.00184582 - 0.00191084 - 0.00194004 - 0.00193274 - 0.00189081 - 0.00181863 - 0.00172275 - 0.00161135 - 0.00149356 - 0.00137872 - 0.00127559 - 0.00119165 - 0.00113253 - 0.00110163 - 0.00109985 - 0.00112561 - 0.001175 - 0.00124218 - 0.00131983 - 0.00139982 - 0.00147384 - 0.00153416 - 0.00157416 - 0.00158894 - 0.00157562 - 0.00153351 - 0.00146403 - 0.00137047 - 0.00125758 - 0.00113103 - 0.000996861 - 0.000860974 - 0.000728651 - 0.000604241 - 0.000490968 - 0.000390879 - 0.000304913 - 0.000233051 - 0.000174528 - 0.000128061 - 9.20676e-05 - 6.48537e-05 - 4.47611e-05 - 3.02693e-05 - 2.00559e-05 - 1.30202e-05 - 8.28195e-06 - 5.16159e-06 - 3.15189e-06 - 1.8858e-06 - 1.10549e-06 - 6.34971e-07 - 3.57346e-07 - 1.97043e-07 - 1.06456e-07 - 5.63526e-08 - 2.92278e-08 - 1.4853e-08 - 7.39555e-09 - 3.60797e-09 - 1.72462e-09 - 8.07717e-10 - 3.70649e-10 - 1.66649e-10 - 7.34141e-11 - 3.16879e-11 - 1.34012e-11 - 5.55305e-12 - 2.25453e-12 - 8.96844e-13 - 3.49555e-13 - 1.3349e-13 - 4.99484e-14 - 1.83118e-14 - 6.57773e-15 - 2.31504e-15 - 7.98324e-16 - 2.69746e-16 - 8.9342e-17 - 2.9128e-17 - 9.76341e-18 - 4.72724e-18 - 6.95038e-18 - 1.96983e-17 - 6.09995e-17 - 1.87121e-16 - 5.62997e-16 - 1.65984e-15 - 4.79478e-15 - 1.35709e-14 - 3.76342e-14 - 1.02258e-13 - 2.72236e-13 - 7.1012e-13 - 1.81491e-12 - 4.5448e-12 - 1.11509e-11 - 2.68068e-11 - 6.31417e-11 - 1.45721e-10 - 3.29509e-10 - 7.30044e-10 - 1.58477e-09 - 3.37071e-09 - 7.02447e-09 - 1.43431e-08 - 2.86951e-08 - 5.62483e-08 - 1.08031e-07 - 2.03294e-07 - 3.74832e-07 - 6.77153e-07 - 1.1986e-06 - 2.07872e-06 - 3.53229e-06 - 5.88103e-06 - 9.59373e-06 - 1.53341e-05 - 2.40141e-05 - 3.68477e-05 - 5.53976e-05 - 8.16036e-05 - 0.000117778 - 0.000166554 - 0.000230773 - 0.000313293 - 0.000416728 - 0.000543115 - 0.000693534 - 0.000867722 - 0.00106373 - 0.00127766 - 0.00150363 - 0.0017338 - 0.00195883 - 0.00216836 - 0.00235181 - 0.00249925 - 0.00260228 - 0.00265482 - 0.00265371 - 0.00259901 - 0.00249402 - 0.00234493 - 0.0021602 - 0.00194983 - 0.00172439 - 0.00149421 - 0.0012686 - 0.0010553 - 0.000860126 - 0.000686887 - 0.000537459 - 0.000412043 - 0.000309512 - 0.000227797 - 0.000164269 - 0.000116065 - 8.03492e-05 - 5.45004e-05 - 3.62205e-05 - 2.35856e-05 - 1.50479e-05 - 9.40679e-06 - 5.76161e-06 - 3.45767e-06 - 2.0331e-06 - 1.17131e-06 - 6.61184e-07 - 3.65687e-07 - 1.98168e-07 - 1.05219e-07 - 5.47382e-08 - 2.79013e-08 - 1.39346e-08 - 6.81873e-09 - 3.26925e-09 - 1.53578e-09 - 7.06883e-10 - 3.18789e-10 - 1.40863e-10 - 6.09863e-11 - 2.58729e-11 - 1.07614e-11 - 4.4044e-12 - 1.81685e-12 - 8.67714e-13 - 7.40664e-13 - 1.31546e-12 - 3.06677e-12 - 7.40609e-12 - 1.76751e-11 - 4.13834e-11 - 9.49532e-11 - 2.13472e-10 - 4.70233e-10 - 1.01489e-09 - 2.14618e-09 - 4.4468e-09 - 9.02747e-09 - 1.79565e-08 - 3.49956e-08 - 6.68256e-08 - 1.25028e-07 - 2.29198e-07 - 4.11672e-07 - 7.24482e-07 - 1.24923e-06 - 2.11053e-06 - 3.49364e-06 - 5.66633e-06 - 9.00457e-06 - 1.40204e-05 - 2.13892e-05 - 3.19717e-05 - 4.68246e-05 - 6.71923e-05 - 9.44715e-05 - 0.000130143 - 0.000175661 - 0.00023231 - 0.000301021 - 0.000382176 - 0.000475407 - 0.000579436 - 0.000691962 - 0.000809647 - 0.000928209 - 0.00104264 - 0.00114751 - 0.00123742 - 0.00130742 - 0.00135347 - 0.00137284 - 0.00136436 - 0.00132854 - 0.00126753 - 0.00118488 - 0.00108525 - 0.00097392 - 0.000856353 - 0.000737767 - 0.000622762 - 0.000515065 - 0.000417387 - 0.0003314 - 0.000257812 - 0.000196512 - 0.000146762 - 0.000107393 - 7.69968e-05 - 5.40888e-05 - 3.72288e-05 - 2.51065e-05 - 1.65895e-05 - 1.07402e-05 - 6.81292e-06 - 4.23437e-06 - 2.57858e-06 - 1.53855e-06 - 8.9945e-07 - 5.15205e-07 - 2.89148e-07 - 1.59e-07 - 8.56664e-08 - 4.52232e-08 - 2.3391e-08 - 1.18542e-08 - 5.88618e-09 - 2.86373e-09 - 1.36511e-09 - 6.37584e-10 - 2.91774e-10 - 1.30825e-10 - 5.74743e-11 - 2.47396e-11 - 1.04339e-11 - 4.31162e-12 - 1.7457e-12 - 6.92527e-13 - 2.69178e-13 - 1.02513e-13 - 3.82523e-14 - 1.39853e-14 - 5.00981e-15 - 1.75836e-15 - 6.0469e-16 - 2.03748e-16 - 6.72655e-17 - 2.17584e-17 - 6.89602e-18 - 2.14145e-18 - 6.51557e-19 - 1.94238e-19 - 5.67355e-20 - 1.62376e-20 - 4.5551e-21 - 1.25945e-21 - 3.71542e-22 - 2.27974e-22 - 5.68172e-22 - 2.09885e-21 - 7.8741e-21 - 2.90139e-20 - 1.04766e-19 - 3.70658e-19 - 1.28488e-18 - 4.36407e-18 - 1.4523e-17 - 4.73541e-17 - 1.51285e-16 - 4.73556e-16 - 1.45239e-15 - 4.36447e-15 - 1.28504e-14 - 3.70715e-14 - 1.04785e-13 - 2.90199e-13 - 7.8746e-13 - 2.09362e-12 - 5.45388e-12 - 1.39203e-11 - 3.4812e-11 - 8.52996e-11 - 2.04786e-10 - 4.81717e-10 - 1.11025e-09 - 2.50718e-09 - 5.54737e-09 - 1.20261e-08 - 2.55447e-08 - 5.31635e-08 - 1.08408e-07 - 2.16596e-07 - 4.24008e-07 - 8.1327e-07 - 1.52839e-06 - 2.81429e-06 - 5.07741e-06 - 8.97539e-06 - 1.55454e-05 - 2.6381e-05 - 4.38651e-05 - 7.1464e-05 - 0.000114077 - 0.000178422 - 0.000273429 - 0.00041057 - 0.000604061 - 0.00087082 - 0.00123009 - 0.00170259 - 0.00230921 - 0.00306905 - 0.00399717 - 0.00510188 - 0.00638213 - 0.00782524 - 0.00940538 - 0.0110833 - 0.0128073 - 0.0145165 - 0.0161444 - 0.017625 - 0.0188984 - 0.0199167 - 0.0206486 - 0.0210826 - 0.0212274 - 0.0211103 - 0.0207729 - 0.0202659 - 0.0196419 - 0.0189493 - 0.0182273 - 0.0175019 - 0.0167852 - 0.0160761 - 0.015364 - 0.0146325 - 0.0138649 - 0.0130493 - 0.0121818 - 0.0112696 - 0.0103324 - 0.00940127 - 0.00851798 - 0.0077322 - 0.00709887 - 0.00667521 - 0.00651766 - 0.00667869 - 0.00720359 - 0.00812682 - 0.00946821 - 0.011229 - 0.0133879 - 0.0158984 - 0.0186868 - 0.0216531 - 0.0246735 - 0.0276071 - 0.0303043 - 0.0326177 - 0.0344137 - 0.0355844 - 0.0360574 - 0.0358021 - 0.0348328 - 0.033207 - 0.0310196 - 0.0283937 - 0.0254689 - 0.0223898 - 0.0192943 - 0.0163042 - 0.0135187 - 0.0110106 - 0.00882576 - 0.0069851 - 0.00548818 - 0.00431784 - 0.00344511 - 0.00283369 - 0.00244377 - 0.00223482 - 0.00216753 - 0.00220503 - 0.00231352 - 0.00246261 - 0.00262555 - 0.00277942 - 0.00290547 - 0.00298931 - 0.00302122 - 0.00299617 - 0.00291371 - 0.00277759 - 0.00259503 - 0.00237585 - 0.00213141 - 0.00187359 - 0.00161373 - 0.00136185 - 0.00112608 - 0.000912319 - 0.000724208 - 0.000563271 - 0.000429248 - 0.000320507 - 0.000234478 - 0.000168076 - 0.000118044 - 8.12306e-05 - 5.47687e-05 - 3.61811e-05 - 2.3419e-05 - 1.48522e-05 - 9.22895e-06 - 5.61887e-06 - 3.35184e-06 - 1.95909e-06 - 1.12192e-06 - 6.29515e-07 - 3.46089e-07 - 1.86426e-07 - 9.83922e-08 - 5.08806e-08 - 2.57799e-08 - 1.27981e-08 - 6.22514e-09 - 2.9668e-09 - 1.38537e-09 - 6.33837e-10 - 2.84137e-10 - 1.248e-10 - 5.37078e-11 - 2.26463e-11 - 9.3561e-12 - 3.78729e-12 - 1.5021e-12 - 5.83724e-13 - 2.22255e-13 - 8.29153e-14 - 3.03084e-14 - 1.08573e-14 - 3.81839e-15 - 1.34071e-15 - 5.41182e-16 - 4.60363e-16 - 1.01752e-15 - 2.94646e-15 - 8.68155e-15 - 2.5165e-14 - 7.1502e-14 - 1.99065e-13 - 5.43016e-13 - 1.45133e-12 - 3.80064e-12 - 9.75179e-12 - 2.4516e-11 - 6.03879e-11 - 1.45743e-10 - 3.44638e-10 - 7.985e-10 - 1.81269e-09 - 4.03189e-09 - 8.78681e-09 - 1.87625e-08 - 3.92542e-08 - 8.04673e-08 - 1.61618e-07 - 3.1805e-07 - 6.13252e-07 - 1.15856e-06 - 2.14455e-06 - 3.88947e-06 - 6.91164e-06 - 1.2034e-05 - 2.05292e-05 - 3.43142e-05 - 5.61969e-05 - 9.01753e-05 - 0.000141775 - 0.000218398 - 0.000329635 - 0.000487479 - 0.000706342 - 0.00100279 - 0.0013949 - 0.00190114 - 0.00253875 - 0.00332173 - 0.00425838 - 0.00534887 - 0.00658288 - 0.00793792 - 0.00937852 - 0.0108567 - 0.012314 - 0.0136848 - 0.0149009 - 0.0158974 - 0.0166178 - 0.01702 - 0.0170798 - 0.0167935 - 0.0161785 - 0.0152711 - 0.0141235 - 0.0127982 - 0.011363 - 0.00988489 - 0.00842537 - 0.00703628 - 0.0057575 - 0.00461596 - 0.00362599 - 0.0027908 - 0.00210459 - 0.00155504 - 0.00112578 - 0.000798551 - 0.000554994 - 0.00037793 - 0.000252157 - 0.000164842 - 0.000105584 - 6.62628e-05 - 4.07452e-05 - 2.45482e-05 - 1.44911e-05 - 8.38143e-06 - 4.74977e-06 - 2.63732e-06 - 1.4348e-06 - 7.64815e-07 - 3.99446e-07 - 2.04407e-07 - 1.02488e-07 - 5.03482e-08 - 2.42344e-08 - 1.14293e-08 - 5.28131e-09 - 2.39112e-09 - 1.06071e-09 - 4.6104e-10 - 1.96362e-10 - 8.19975e-11 - 3.37003e-11 - 1.39853e-11 - 6.7944e-12 - 6.04916e-12 - 1.11421e-11 - 2.65104e-11 - 6.50445e-11 - 1.57586e-10 - 3.74505e-10 - 8.72182e-10 - 1.99023e-09 - 4.44979e-09 - 9.74793e-09 - 2.09229e-08 - 4.40016e-08 - 9.06677e-08 - 1.83051e-07 - 3.62102e-07 - 7.01821e-07 - 1.33278e-06 - 2.47987e-06 - 4.52101e-06 - 8.07572e-06 - 1.4134e-05 - 2.42374e-05 - 4.07237e-05 - 6.70422e-05 - 0.000108141 - 0.000170912 - 0.000264666 - 0.000401576 - 0.000597016 - 0.000869669 - 0.0012413 - 0.00173603 - 0.00237906 - 0.00319468 - 0.00420373 - 0.00542055 - 0.00684968 - 0.00848282 - 0.0102964 - 0.0122502 - 0.0142879 - 0.016339 - 0.0183232 - 0.020156 - 0.0217557 - 0.0230507 - 0.0239861 - 0.0245286 - 0.0246698 - 0.0244259 - 0.0238353 - 0.0229533 - 0.0218461 - 0.020583 - 0.0192297 - 0.017843 - 0.0164669 - 0.0151311 - 0.0138521 - 0.0126351 - 0.0114777 - 0.0103735 - 0.00931555 - 0.00829889 - 0.00732211 - 0.00638771 - 0.00550152 - 0.00467147 - 0.00390611 - 0.00321307 - 0.0025979 - 0.00206328 - 0.00160877 - 0.00123095 - 0.000923967 - 0.00068019 - 0.000491014 - 0.000347568 - 0.000241316 - 0.000164498 - 0.000110414 - 7.35521e-05 - 4.96251e-05 - 3.55275e-05 - 2.92725e-05 - 2.99299e-05 - 3.75905e-05 - 5.33657e-05 - 7.9422e-05 - 0.000119033 - 0.000176624 - 0.000257776 - 0.000369135 - 0.000518189 - 0.000712868 - 0.000960941 - 0.00126921 - 0.00164251 - 0.00208268 - 0.00258746 - 0.00314965 - 0.00375654 - 0.00438986 - 0.00502632 - 0.0056388 - 0.00619812 - 0.00667528 - 0.00704395 - 0.00728283 - 0.00737769 - 0.00732281 - 0.00712151 - 0.00678583 - 0.00633536 - 0.0057953 - 0.00519418 - 0.00456137 - 0.00392474 - 0.00330874 - 0.00273308 - 0.00221196 - 0.00175404 - 0.00136282 - 0.00103747 - 0.000773837 - 0.000565535 - 0.000404955 - 0.000284112 - 0.000195304 - 0.000131543 - 8.68085e-05 - 5.61298e-05 - 3.556e-05 - 2.20733e-05 - 1.34248e-05 - 7.99994e-06 - 4.67091e-06 - 2.67211e-06 - 1.49776e-06 - 8.22563e-07 - 4.42621e-07 - 2.33363e-07 - 1.2055e-07 - 6.10156e-08 - 3.02588e-08 - 1.47027e-08 - 6.99974e-09 - 3.26515e-09 - 1.49231e-09 - 6.68274e-10 - 2.93215e-10 - 1.26053e-10 - 5.30957e-11 - 2.19129e-11 - 8.86093e-12 - 3.51071e-12 - 1.36285e-12 - 5.18366e-13 - 1.9318e-13 - 7.05383e-14 - 2.52363e-14 - 8.84629e-15 - 3.03833e-15 - 1.02246e-15 - 3.37126e-16 - 1.08912e-16 - 3.44744e-17 - 1.06919e-17 - 3.24899e-18 - 9.67342e-19 - 2.82194e-19 - 8.06591e-20 - 2.25889e-20 - 6.19832e-21 - 1.66644e-21 - 4.38977e-22 - 1.133e-22 - 2.86522e-23 - 7.09938e-24 - 1.72354e-24 - 4.09975e-25 - 9.55504e-26 - 2.18199e-26 - 4.88265e-27 - 1.07116e-27 - 2.31016e-28 - 4.97312e-29 - 1.15485e-29 - 3.80679e-30 - 2.39803e-30 - 2.26322e-30 - 2.35662e-30 - 2.46186e-30 - 2.53433e-30 - 2.57874e-30 - 2.6806e-30 - 3.2948e-30 - 6.70724e-30 - 2.41703e-29 - 1.09986e-28 - 5.20897e-28 - 2.44588e-27 - 1.12777e-26 - 5.09704e-26 - 2.25729e-25 - 9.79488e-25 - 4.16437e-24 - 1.73475e-23 - 7.08044e-23 - 2.83153e-22 - 1.10948e-21 - 4.25944e-21 - 1.60223e-20 - 5.90517e-20 - 2.13244e-19 - 7.545e-19 - 2.61564e-18 - 8.88451e-18 - 2.95683e-17 - 9.64174e-17 - 3.0805e-16 - 9.64327e-16 - 2.95777e-15 - 8.88875e-15 - 2.6173e-14 - 7.551e-14 - 2.13448e-13 - 5.91175e-13 - 1.60427e-12 - 4.26554e-12 - 1.11124e-11 - 2.83648e-11 - 7.09396e-11 - 1.73834e-10 - 4.17365e-10 - 9.81826e-10 - 2.26303e-09 - 5.11073e-09 - 1.13087e-08 - 2.45176e-08 - 5.20812e-08 - 1.08398e-07 - 2.21054e-07 - 4.41684e-07 - 8.64693e-07 - 1.65863e-06 - 3.11726e-06 - 5.7403e-06 - 1.0357e-05 - 1.83091e-05 - 3.17131e-05 - 5.38204e-05 - 8.94936e-05 - 0.000145806 - 0.000232752 - 0.00036404 - 0.000557881 - 0.000837666 - 0.00123236 - 0.0017764 - 0.00250888 - 0.00347182 - 0.00470729 - 0.00625347 - 0.0081397 - 0.0103808 - 0.0129716 - 0.0158815 - 0.0190514 - 0.0223923 - 0.0257874 - 0.0290973 - 0.0321688 - 0.0348462 - 0.0369839 - 0.03846 - 0.0391873 - 0.0391222 - 0.0382691 - 0.0366799 - 0.0344491 - 0.0317048 - 0.0285972 - 0.0252851 - 0.0219241 - 0.0186558 - 0.0155996 - 0.0128486 - 0.0104682 - 0.00849752 - 0.00695295 - 0.00583265 - 0.00512109 - 0.00479313 - 0.00481689 - 0.00515543 - 0.0057673 - 0.00660627 - 0.00762079 - 0.00875373 - 0.00994277 - 0.0111219 - 0.0122239 - 0.0131839 - 0.0139439 - 0.0144563 - 0.0146885 - 0.0146249 - 0.0142685 - 0.0136401 - 0.0127762 - 0.0117254 - 0.0105438 - 0.00928966 - 0.00801941 - 0.00678301 - 0.00562134 - 0.0045645 - 0.00363149 - 0.00283082 - 0.00216211 - 0.001618 - 0.00118636 - 0.000852305 - 0.00059995 - 0.000413799 - 0.000279676 - 0.000185285 - 0.000120433 - 7.70311e-05 - 4.89435e-05 - 3.17797e-05 - 2.27018e-05 - 2.03111e-05 - 2.46741e-05 - 3.75481e-05 - 6.28712e-05 - 0.000107589 - 0.000182895 - 0.000305951 - 0.00050214 - 0.000807822 - 0.0012735 - 0.00196715 - 0.00297727 - 0.00441506 - 0.00641494 - 0.00913241 - 0.0127384 - 0.0174093 - 0.0233123 - 0.0305863 - 0.0393191 - 0.0495243 - 0.061118 - 0.0739021 - 0.0875551 - 0.101635 - 0.115596 - 0.128818 - 0.140653 - 0.150473 - 0.157726 - 0.161989 - 0.163007 - 0.160717 - 0.155259 - 0.146955 - 0.136286 - 0.123839 - 0.110255 - 0.0961777 - 0.0822032 - 0.0688398 - 0.0564843 - 0.0454101 - 0.0357696 - 0.0276066 - 0.020876 - 0.0154675 - 0.0112287 - 0.00798685 - 0.00556619 - 0.00380082 - 0.00254293 - 0.00166697 - 0.00107067 - 0.00067379 - 0.000415459 - 0.000250997 - 0.000148575 - 8.6171e-05 - 4.8968e-05 - 2.72647e-05 - 1.48739e-05 - 7.95037e-06 - 4.16376e-06 - 2.13659e-06 - 1.07422e-06 - 5.29179e-07 - 2.55416e-07 - 1.2079e-07 - 5.59695e-08 - 2.54102e-08 - 1.13032e-08 - 4.92642e-09 - 2.10377e-09 - 8.80244e-10 - 3.60866e-10 - 1.44959e-10 - 5.70702e-11 - 2.20651e-11 - 8.50316e-12 - 3.61538e-12 - 2.59818e-12 - 4.39325e-12 - 1.06972e-11 - 2.73806e-11 - 6.93422e-11 - 1.72287e-10 - 4.19489e-10 - 1.00077e-09 - 2.3393e-09 - 5.35766e-09 - 1.20227e-08 - 2.6434e-08 - 5.6946e-08 - 1.20199e-07 - 2.48584e-07 - 5.03712e-07 - 1.00007e-06 - 1.94542e-06 - 3.70795e-06 - 6.92454e-06 - 1.26703e-05 - 2.27152e-05 - 3.99011e-05 - 6.86735e-05 - 0.000115806 - 0.000191342 - 0.000309759 - 0.000491334 - 0.0007636 - 0.00116276 - 0.00173482 - 0.00253603 - 0.00363237 - 0.00509757 - 0.00700928 - 0.00944323 - 0.0124653 - 0.0161222 - 0.0204307 - 0.0253675 - 0.0308609 - 0.0367855 - 0.0429617 - 0.0491612 - 0.055119 - 0.0605504 - 0.0651732 - 0.0687319 - 0.0710206 - 0.071903 - 0.0713258 - 0.0693239 - 0.0660171 - 0.061598 - 0.0563137 - 0.0504426 - 0.0442709 - 0.0380694 - 0.0320753 - 0.026479 - 0.0214176 - 0.0169736 - 0.01318 - 0.0100276 - 0.00747499 - 0.00545963 - 0.00390708 - 0.00273955 - 0.00188209 - 0.0012669 - 0.00083556 - 0.000539947 - 0.00034187 - 0.000212084 - 0.000128912 - 7.67739e-05 - 4.47993e-05 - 2.56133e-05 - 1.43482e-05 - 7.87527e-06 - 4.23517e-06 - 2.23158e-06 - 1.1521e-06 - 5.82783e-07 - 2.88841e-07 - 1.40265e-07 - 6.67383e-08 - 3.11127e-08 - 1.42114e-08 - 6.36025e-09 - 2.789e-09 - 1.19828e-09 - 5.04435e-10 - 2.08061e-10 - 8.40835e-11 - 3.32942e-11 - 1.29171e-11 - 4.91018e-12 - 1.8289e-12 - 6.6774e-13 - 2.39797e-13 - 8.72764e-14 - 3.99832e-14 - 4.35828e-14 - 1.03126e-13 - 2.89491e-13 - 8.16136e-13 - 2.26089e-12 - 6.1391e-12 - 1.6335e-11 - 4.25903e-11 - 1.08812e-10 - 2.72406e-10 - 6.68242e-10 - 1.6063e-09 - 3.78352e-09 - 8.7326e-09 - 1.97501e-08 - 4.37696e-08 - 9.50509e-08 - 2.02264e-07 - 4.21756e-07 - 8.61758e-07 - 1.7254e-06 - 3.38515e-06 - 6.50799e-06 - 1.22603e-05 - 2.26326e-05 - 4.09407e-05 - 7.25704e-05 - 0.000126052 - 0.000214548 - 0.000357837 - 0.000584835 - 0.000936631 - 0.00146991 - 0.00226049 - 0.00340645 - 0.00503026 - 0.00727894 - 0.0103214 - 0.0143415 - 0.0195274 - 0.0260547 - 0.0340659 - 0.0436462 - 0.0547983 - 0.067419 - 0.0812818 - 0.0960292 - 0.111177 - 0.126134 - 0.140238 - 0.1528 - 0.163165 - 0.170766 - 0.175185 - 0.17619 - 0.173768 - 0.168128 - 0.159688 - 0.149039 - 0.136897 - 0.124043 - 0.111262 - 0.0992842 - 0.0887352 - 0.0800981 - 0.073689 - 0.0696463 - 0.0679332 - 0.0683505 - 0.0705597 - 0.0741133 - 0.0784909 - 0.0831388 - 0.0875109 - 0.091108 - 0.0935133 - 0.09442 - 0.0936496 - 0.0911582 - 0.0870312 - 0.0814664 - 0.0747481 - 0.0672163 - 0.0592323 - 0.0511477 - 0.0432772 - 0.0358796 - 0.0291464 - 0.0231988 - 0.0180921 - 0.0138246 - 0.0103503 - 0.00759265 - 0.00545721 - 0.00384313 - 0.00265178 - 0.00179277 - 0.00118755 - 0.000770751 - 0.000490133 - 0.000305387 - 0.000186433 - 0.000111515 - 6.53553e-05 - 3.75288e-05 - 2.11147e-05 - 1.16397e-05 - 6.28691e-06 - 3.32712e-06 - 1.7252e-06 - 8.76506e-07 - 4.36367e-07 - 2.12967e-07 - 1.02108e-07 - 4.86122e-08 - 2.41809e-08 - 1.51788e-08 - 1.63679e-08 - 2.83794e-08 - 5.79776e-08 - 1.21189e-07 - 2.50402e-07 - 5.07817e-07 - 1.0094e-06 - 1.966e-06 - 3.75186e-06 - 7.01533e-06 - 1.28525e-05 - 2.30708e-05 - 4.05764e-05 - 6.99235e-05 - 0.000118062 - 0.000195313 - 0.000316586 - 0.000502791 - 0.000782386 - 0.00119286 - 0.00178196 - 0.00260821 - 0.00374044 - 0.00525582 - 0.00723593 - 0.00976081 - 0.0129007 - 0.0167063 - 0.0211974 - 0.0263525 - 0.0320995 - 0.0383101 - 0.044799 - 0.051329 - 0.0576237 - 0.0633852 - 0.0683175 - 0.0721517 - 0.0746721 - 0.0757375 - 0.0752976 - 0.0734009 - 0.0701941 - 0.0659136 - 0.0608694 - 0.0554259 - 0.0499808 - 0.0449442 - 0.0407215 - 0.0376986 - 0.0362307 - 0.0366324 - 0.0391679 - 0.0440395 - 0.0513723 - 0.0611977 - 0.0734333 - 0.0878657 - 0.104137 - 0.121742 - 0.140036 - 0.158258 - 0.175572 - 0.191121 - 0.204086 - 0.213755 - 0.219579 - 0.221225 - 0.218598 - 0.211856 - 0.201386 - 0.187772 - 0.171739 - 0.15409 - 0.13564 - 0.117162 - 0.0993331 - 0.0827063 - 0.0676928 - 0.0545624 - 0.043456 - 0.034407 - 0.0273684 - 0.022239 - 0.0188874 - 0.0171699 - 0.0169424 - 0.0180635 - 0.0203933 - 0.0237859 - 0.0280809 - 0.0330942 - 0.0386122 - 0.0443899 - 0.0501552 - 0.0556197 - 0.0604946 - 0.06451 - 0.0674365 - 0.069104 - 0.069417 - 0.0683625 - 0.0660108 - 0.0625073 - 0.0580577 - 0.0529082 - 0.0473225 - 0.0415602 - 0.0358572 - 0.0304111 - 0.0253728 - 0.0208431 - 0.0168751 - 0.0134805 - 0.0106382 - 0.00830404 - 0.00641967 - 0.00492092 - 0.00374385 - 0.00282905 - 0.0021241 - 0.00158457 - 0.00117404 - 0.000863316 - 0.000629412 - 0.000454431 - 0.000324503 - 0.000228896 - 0.000159295 - 0.000109251 - 7.37695e-05 - 4.89991e-05 - 3.19922e-05 - 2.05207e-05 - 1.2926e-05 - 7.99585e-06 - 4.86315e-06 - 2.92334e-06 - 1.76948e-06 - 1.1439e-06 - 9.06437e-07 - 1.02009e-06 - 1.55632e-06 - 2.72467e-06 - 4.93405e-06 - 8.89688e-06 - 1.57908e-05 - 2.74957e-05 - 4.69267e-05 - 7.84794e-05 - 0.0001286 - 0.000206474 - 0.00032481 - 0.000500644 - 0.000756078 - 0.00111877 - 0.001622 - 0.00230409 - 0.00320688 - 0.00437325 - 0.00584336 - 0.00764994 - 0.00981273 - 0.0123327 - 0.0151867 - 0.0183234 - 0.0216614 - 0.0250901 - 0.0284744 - 0.0316625 - 0.0344962 - 0.0368244 - 0.0385155 - 0.0394705 - 0.0396321 - 0.0389905 - 0.0375844 - 0.0354972 - 0.032849 - 0.0297846 - 0.0264615 - 0.0230361 - 0.0196523 - 0.0164334 - 0.0134759 - 0.0108485 - 0.00859408 - 0.00673434 - 0.00527724 - 0.00422554 - 0.0035862 - 0.00337974 - 0.00364906 - 0.0044669 - 0.00594138 - 0.00821867 - 0.0114817 - 0.0159437 - 0.0218363 - 0.0293904 - 0.0388107 - 0.0502452 - 0.0637519 - 0.0792647 - 0.0965666 - 0.115271 - 0.134821 - 0.154501 - 0.173478 - 0.190851 - 0.205722 - 0.217273 - 0.224836 - 0.227963 - 0.226463 - 0.220429 - 0.210222 - 0.196437 - 0.179848 - 0.161335 - 0.141805 - 0.122123 - 0.103052 - 0.0852089 - 0.0690443 - 0.0548367 - 0.0427079 - 0.0326479 - 0.0245476 - 0.0182336 - 0.0135014 - 0.0101435 - 0.00797117 - 0.00682901 - 0.00660242 - 0.0072185 - 0.00864139 - 0.0108629 - 0.0138899 - 0.0177288 - 0.0223696 - 0.0277703 - 0.0338438 - 0.0404482 - 0.0473843 - 0.0543987 - 0.0611951 - 0.0674527 - 0.0728495 - 0.0770893 - 0.0799283 - 0.0811978 - 0.0808212 - 0.0788213 - 0.075318 - 0.0705165 - 0.0646873 - 0.0581413 - 0.0512021 - 0.0441801 - 0.0373511 - 0.0309397 - 0.0251111 - 0.0199688 - 0.0155588 - 0.0118778 - 0.00888452 - 0.0065113 - 0.00467562 - 0.00328963 - 0.00226773 - 0.0015317 - 0.00101366 - 0.000657274 - 0.000417579 - 0.000259937 - 0.000158538 - 9.47405e-05 - 5.54722e-05 - 3.18238e-05 - 1.78881e-05 - 9.85178e-06 - 5.3162e-06 - 2.81077e-06 - 1.45608e-06 - 7.39066e-07 - 3.67551e-07 - 1.79097e-07 - 8.55059e-08 - 3.99982e-08 - 1.83325e-08 - 8.23268e-09 - 3.6224e-09 - 1.56167e-09 - 6.59656e-10 - 2.73013e-10 - 1.1071e-10 - 4.39871e-11 - 1.71239e-11 - 6.53153e-12 - 2.44099e-12 - 8.93824e-13 - 3.20683e-13 - 1.12729e-13 - 3.8827e-14 - 1.31029e-14 - 4.33251e-15 - 1.40361e-15 - 4.45546e-16 - 1.38572e-16 - 4.22272e-17 - 1.26081e-17 - 3.68842e-18 - 1.05723e-18 - 2.96917e-19 - 8.1703e-20 - 2.20281e-20 - 5.81908e-21 - 1.50615e-21 - 3.8196e-22 - 9.49084e-23 - 2.31062e-23 - 5.51175e-24 - 1.28821e-24 - 2.94999e-25 - 6.619e-26 - 1.45513e-26 - 3.13434e-27 - 6.61496e-28 - 1.36787e-28 - 2.77142e-29 - 5.50166e-30 - 1.0701e-30 - 2.03934e-31 - 3.80796e-32 - 6.96678e-33 - 1.24885e-33 - 2.19343e-34 - 3.77463e-35 - 6.36448e-36 - 1.05146e-36 - 1.702e-37 - 2.69944e-38 - 4.19508e-39 - 6.38815e-40 - 9.53255e-41 - 1.39414e-41 - 1.99888e-42 - 2.81123e-43 - 3.88255e-44 - 5.2771e-45 - 7.08851e-46 - 9.48461e-47 - 1.28189e-47 - 1.78939e-48 - 2.65579e-49 - 4.30178e-50 - 7.66461e-51 - 1.47835e-51 - 2.999e-52 - 6.22693e-53 - 1.29829e-53 - 2.68714e-54 - 5.48649e-55 - 1.10138e-55 - 2.17011e-56 - 4.19319e-57 - 7.94216e-58 - 1.47424e-58 - 2.68152e-59 - 4.77921e-60 - 8.34605e-61 - 1.42807e-61 - 2.39418e-62 - 3.93281e-63 - 6.32974e-64 - 9.98176e-65 - 1.54229e-65 - 2.33486e-66 - 3.46332e-67 - 5.03339e-68 - 7.16748e-69 - 1.00002e-69 - 1.36706e-70 - 1.83107e-71 - 2.40302e-72 - 3.08992e-73 - 3.8929e-74 - 4.80549e-75 - 5.81216e-76 - 6.88771e-77 - 7.9974e-78 - 9.09829e-79 - 1.01416e-79 - 1.10762e-80 - 1.18526e-81 - 1.24272e-82 - 1.27664e-83 - 1.28499e-84 - 1.26727e-85 - 1.22454e-86 - 1.15935e-87 - 1.07546e-88 - 9.77489e-90 - 8.70493e-91 - 7.59548e-92 - 6.49355e-93 - 5.43933e-94 - 4.46422e-95 - 3.58991e-96 - 2.82851e-97 - 2.18358e-98 - 1.65164e-99 - 1.22406e-100 - 8.88838e-102 - 6.32384e-103 - 4.40835e-104 - 3.01098e-105 - 2.01501e-106 - 1.32124e-107 - 8.4884e-109 - 5.34325e-110 - 3.2955e-111 - 1.99148e-112 - 1.17914e-113 - 6.84056e-115 - 3.88826e-116 - 2.16549e-117 - 1.18166e-118 - 6.31783e-120 - 3.30963e-121 - 1.69874e-122 - 8.54303e-124 - 4.20953e-125 - 2.03232e-126 - 9.61363e-128 - 4.45574e-129 - 2.02343e-130 - 9.00315e-132 - 3.92498e-133 - 1.67655e-134 - 7.01914e-136 - 3.4841e-137 - 1.49636e-136 - 3.55992e-135 - 8.36254e-134 - 1.92477e-132 - 4.34066e-131 - 9.59114e-130 - 2.07645e-128 - 4.40463e-127 - 9.15448e-126 - 1.86421e-124 - 3.71958e-123 - 7.27158e-122 - 1.39284e-120 - 2.61403e-119 - 4.8068e-118 - 8.66041e-117 - 1.52883e-115 - 2.64432e-114 - 4.48134e-113 - 7.44111e-112 - 1.21061e-110 - 1.92978e-109 - 3.01404e-108 - 4.61239e-107 - 6.91577e-106 - 1.016e-104 - 1.46245e-103 - 2.06255e-102 - 2.85014e-101 - 3.85891e-100 - 5.11918e-99 - 6.65384e-98 - 8.47386e-97 - 1.05737e-95 - 1.29274e-94 - 1.54857e-93 - 1.81755e-92 - 2.09016e-91 - 2.3551e-90 - 2.60002e-89 - 2.81242e-88 - 2.98072e-87 - 3.09527e-86 - 3.14929e-85 - 3.13953e-84 - 3.06657e-83 - 2.93479e-82 - 2.75194e-81 - 2.52835e-80 - 2.27601e-79 - 2.00746e-78 - 1.73482e-77 - 1.46893e-76 - 1.21866e-75 - 9.90612e-75 - 7.8897e-74 - 6.15678e-73 - 4.70743e-72 - 3.52656e-71 - 2.58854e-70 - 1.86164e-69 - 1.31182e-68 - 9.05709e-68 - 6.12689e-67 - 4.06096e-66 - 2.63727e-65 - 1.67809e-64 - 1.0462e-63 - 6.39074e-63 - 3.82493e-62 - 2.24302e-61 - 1.28878e-60 - 7.25545e-60 - 4.00207e-59 - 2.16293e-58 - 1.14535e-57 - 5.94248e-57 - 3.0209e-56 - 1.50467e-55 - 7.34318e-55 - 3.51127e-54 - 1.64505e-53 - 7.55152e-53 - 3.39645e-52 - 1.49676e-51 - 6.46277e-51 - 2.73414e-50 - 1.13334e-49 - 4.60298e-49 - 1.8317e-48 - 7.1418e-48 - 2.72834e-47 - 1.02124e-46 - 3.74536e-46 - 1.34586e-45 - 4.73851e-45 - 1.63465e-44 - 5.52516e-44 - 1.82981e-43 - 5.93751e-43 - 1.88775e-42 - 5.88063e-42 - 1.79492e-41 - 5.36793e-41 - 1.57294e-40 - 4.51605e-40 - 1.27042e-39 - 3.50174e-39 - 9.45725e-39 - 2.5026e-38 - 6.48885e-38 - 1.64852e-37 - 4.10365e-37 - 1.00093e-36 - 2.39215e-36 - 5.60189e-36 - 1.28541e-35 - 2.89012e-35 - 6.36733e-35 - 1.37459e-34 - 2.90782e-34 - 6.02764e-34 - 1.22439e-33 - 2.43718e-33 - 4.75406e-33 - 9.08781e-33 - 1.70248e-32 - 3.12568e-32 - 5.62421e-32 - 9.91858e-32 - 1.71446e-31 - 2.90477e-31 - 4.82424e-31 - 7.85422e-31 - 1.25361e-30 - 1.96175e-30 - 3.01008e-30 - 4.52904e-30 - 6.68305e-30 - 9.67234e-30 - 1.37319e-29 - 1.91264e-29 - 2.61398e-29 - 3.50596e-29 - 4.61557e-29 - 5.96537e-29 - 7.57053e-29 - 9.43589e-29 - 1.15531e-28 - 1.38985e-28 - 1.64318e-28 - 1.9096e-28 - 2.18189e-28 - 2.45155e-28 - 2.70921e-28 - 2.94517e-28 - 3.14991e-28 - 3.31475e-28 - 3.43242e-28 - 3.4975e-28 - 3.5069e-28 - 3.46002e-28 - 3.35883e-28 - 3.2078e-28 - 3.01352e-28 - 2.7843e-28 - 2.52962e-28 - 2.25946e-28 - 1.98368e-28 - 1.71145e-28 - 1.45073e-28 - 1.20793e-28 - 9.87744e-29 - - - - - 1.17565e-32 - 1.46307e-32 - 1.78398e-32 - 2.13137e-32 - 2.4952e-32 - 2.86344e-32 - 3.22697e-32 - 3.60346e-32 - 4.15858e-32 - 5.82494e-32 - 1.34146e-31 - 5.09688e-31 - 2.359e-30 - 1.12965e-29 - 5.36147e-29 - 2.49904e-28 - 1.14183e-27 - 5.11219e-27 - 2.24263e-26 - 9.63936e-26 - 4.05952e-25 - 1.67509e-24 - 6.77235e-24 - 2.68273e-23 - 1.04124e-22 - 3.9597e-22 - 1.4754e-21 - 5.38636e-21 - 1.92671e-20 - 6.75267e-20 - 2.31884e-19 - 7.80195e-19 - 2.57201e-18 - 8.30765e-18 - 2.62918e-17 - 8.15268e-17 - 2.47695e-16 - 7.37345e-16 - 2.15061e-15 - 6.14593e-15 - 1.72088e-14 - 4.7212e-14 - 1.26908e-13 - 3.34245e-13 - 8.62533e-13 - 2.18084e-12 - 5.40267e-12 - 1.31139e-11 - 3.11881e-11 - 7.26749e-11 - 1.65927e-10 - 3.71181e-10 - 8.13565e-10 - 1.74717e-09 - 3.67633e-09 - 7.57934e-09 - 1.53103e-08 - 3.03023e-08 - 5.87628e-08 - 1.11652e-07 - 2.07858e-07 - 3.79145e-07 - 6.77609e-07 - 1.18656e-06 - 2.03582e-06 - 3.42235e-06 - 5.63697e-06 - 9.09714e-06 - 1.43847e-05 - 2.2286e-05 - 3.383e-05 - 5.03162e-05 - 7.33248e-05 - 0.000104696 - 0.00014647 - 0.000200771 - 0.000269645 - 0.000354829 - 0.000457491 - 0.000577942 - 0.000715355 - 0.000867554 - 0.00103088 - 0.00120021 - 0.00136913 - 0.00153027 - 0.00167583 - 0.00179817 - 0.00189049 - 0.00194743 - 0.00196562 - 0.00194402 - 0.001884 - 0.00178928 - 0.00166553 - 0.00151987 - 0.00136029 - 0.00119498 - 0.00103175 - 0.000877556 - 0.000738187 - 0.000618089 - 0.000520341 - 0.000446731 - 0.000397902 - 0.000373517 - 0.000372412 - 0.000392709 - 0.000431897 - 0.000486877 - 0.000554005 - 0.000629151 - 0.000707798 - 0.000785206 - 0.00085663 - 0.000917596 - 0.000964193 - 0.000993359 - 0.00100311 - 0.000992716 - 0.000962696 - 0.000914787 - 0.000851735 - 0.000777026 - 0.000694558 - 0.000608305 - 0.000522003 - 0.000438896 - 0.000361567 - 0.000291845 - 0.000230809 - 0.000178851 - 0.000135789 - 0.000101013 - 7.3625e-05 - 5.25788e-05 - 3.67903e-05 - 2.52227e-05 - 1.69429e-05 - 1.11512e-05 - 7.19101e-06 - 4.54356e-06 - 2.8128e-06 - 1.70616e-06 - 1.014e-06 - 5.90459e-07 - 3.36884e-07 - 1.88325e-07 - 1.03151e-07 - 5.53571e-08 - 2.91079e-08 - 1.49964e-08 - 7.57003e-09 - 3.74409e-09 - 1.81439e-09 - 8.61497e-10 - 4.00786e-10 - 1.82687e-10 - 8.15908e-11 - 3.57036e-11 - 1.53081e-11 - 6.43124e-12 - 2.64844e-12 - 1.07197e-12 - 4.34795e-13 - 2.00078e-13 - 1.63936e-13 - 3.00174e-13 - 7.46316e-13 - 1.9281e-12 - 4.92178e-12 - 1.23241e-11 - 3.02408e-11 - 7.27073e-11 - 1.71278e-10 - 3.95331e-10 - 8.94041e-10 - 1.98103e-09 - 4.30092e-09 - 9.1489e-09 - 1.90683e-08 - 3.89398e-08 - 7.79133e-08 - 1.52745e-07 - 2.93398e-07 - 5.52187e-07 - 1.01824e-06 - 1.83973e-06 - 3.25681e-06 - 5.64895e-06 - 9.60021e-06 - 1.59856e-05 - 2.60805e-05 - 4.16907e-05 - 6.52979e-05 - 0.000100207 - 0.000150671 - 0.000221973 - 0.000320411 - 0.00045316 - 0.000627962 - 0.000852611 - 0.00113424 - 0.00147842 - 0.0018881 - 0.0023626 - 0.00289663 - 0.00347962 - 0.00409551 - 0.00472303 - 0.00533667 - 0.00590822 - 0.00640885 - 0.00681147 - 0.00709314 - 0.00723725 - 0.00723514 - 0.00708696 - 0.00680167 - 0.00639614 - 0.00589358 - 0.00532131 - 0.00470844 - 0.00408352 - 0.00347255 - 0.00289752 - 0.00237559 - 0.00191887 - 0.00153483 - 0.00122706 - 0.000996256 - 0.000841365 - 0.000760476 - 0.000751552 - 0.000812793 - 0.000942642 - 0.00113944 - 0.00140081 - 0.00172285 - 0.0020993 - 0.00252088 - 0.00297478 - 0.00344473 - 0.00391135 - 0.00435317 - 0.00474797 - 0.00507448 - 0.00531415 - 0.00545286 - 0.00548225 - 0.00540051 - 0.00521259 - 0.00492967 - 0.00456811 - 0.00414791 - 0.00369091 - 0.00321908 - 0.00275291 - 0.00231029 - 0.00190581 - 0.00155063 - 0.00125284 - 0.00101825 - 0.000851451 - 0.000756966 - 0.000740438 - 0.000809583 - 0.00097481 - 0.00124937 - 0.00164889 - 0.00219032 - 0.00289009 - 0.0037618 - 0.00481339 - 0.00604416 - 0.00744202 - 0.00898139 - 0.0106221 - 0.0123098 - 0.013978 - 0.015552 - 0.0169538 - 0.0181087 - 0.0189516 - 0.019433 - 0.0195241 - 0.0192195 - 0.0185373 - 0.0175182 - 0.0162207 - 0.0147159 - 0.013081 - 0.0113928 - 0.00972205 - 0.00812871 - 0.00665921 - 0.00534516 - 0.00420373 - 0.00323926 - 0.00244565 - 0.00180917 - 0.0013113 - 0.000931238 - 0.000647972 - 0.000441762 - 0.000295092 - 0.000193136 - 0.000123853 - 7.78193e-05 - 4.79075e-05 - 2.88973e-05 - 1.70784e-05 - 9.88952e-06 - 5.61099e-06 - 3.11918e-06 - 1.69894e-06 - 9.06678e-07 - 4.74094e-07 - 2.42892e-07 - 1.21926e-07 - 5.99681e-08 - 2.88987e-08 - 1.3645e-08 - 6.31259e-09 - 2.86139e-09 - 1.27082e-09 - 5.53001e-10 - 2.3578e-10 - 9.8497e-11 - 4.03159e-11 - 1.61684e-11 - 6.35323e-12 - 2.44601e-12 - 9.22699e-13 - 3.41034e-13 - 1.23502e-13 - 4.38212e-14 - 1.52347e-14 - 5.18941e-15 - 1.73197e-15 - 5.66369e-16 - 1.81466e-16 - 5.69676e-17 - 1.75226e-17 - 5.28085e-18 - 1.55936e-18 - 4.51156e-19 - 1.27892e-19 - 3.5522e-20 - 9.66693e-21 - 2.5776e-21 - 6.73413e-22 - 1.72385e-22 - 4.3267e-23 - 1.07691e-23 - 3.17029e-24 - 3.13795e-24 - 1.04192e-23 - 4.08832e-23 - 1.59072e-22 - 6.06845e-22 - 2.26838e-21 - 8.3079e-21 - 2.98129e-20 - 1.04822e-19 - 3.6111e-19 - 1.21889e-18 - 4.03109e-18 - 1.30623e-17 - 4.14718e-17 - 1.2901e-16 - 3.93216e-16 - 1.17429e-15 - 3.43603e-15 - 9.85087e-15 - 2.76713e-14 - 7.6159e-14 - 2.05376e-13 - 5.42644e-13 - 1.40481e-12 - 3.56333e-12 - 8.85588e-12 - 2.15648e-11 - 5.1451e-11 - 1.20276e-10 - 2.75489e-10 - 6.1825e-10 - 1.35944e-09 - 2.92883e-09 - 6.18252e-09 - 1.27871e-08 - 2.5913e-08 - 5.14516e-08 - 1.00096e-07 - 1.90797e-07 - 3.56339e-07 - 6.52066e-07 - 1.16912e-06 - 2.05381e-06 - 3.53508e-06 - 5.96177e-06 - 9.85117e-06 - 1.59492e-05 - 2.53002e-05 - 3.93232e-05 - 5.98839e-05 - 8.93528e-05 - 0.00013063 - 0.000187119 - 0.000262622 - 0.000361147 - 0.000486606 - 0.000642415 - 0.000830998 - 0.00105327 - 0.00130809 - 0.00159189 - 0.00189838 - 0.00221861 - 0.00254131 - 0.00285355 - 0.00314183 - 0.00339333 - 0.00359738 - 0.00374691 - 0.00383975 - 0.00387957 - 0.00387642 - 0.00384664 - 0.00381224 - 0.0037996 - 0.00383766 - 0.00395569 - 0.00418069 - 0.0045347 - 0.00503217 - 0.00567767 - 0.00646409 - 0.00737179 - 0.00836862 - 0.00941129 - 0.0104478 - 0.011421 - 0.012273 - 0.0129503 - 0.013408 - 0.0136139 - 0.013552 - 0.0132229 - 0.0126445 - 0.011849 - 0.0108804 - 0.00978988 - 0.00863109 - 0.00745595 - 0.00631082 - 0.00523372 - 0.00425281 - 0.00338595 - 0.00264134 - 0.00201887 - 0.00151194 - 0.00110944 - 0.000797693 - 0.000562024 - 0.000388087 - 0.000262733 - 0.000174531 - 0.00011398 - 7.34973e-05 - 4.72468e-05 - 3.0892e-05 - 2.1316e-05 - 1.63489e-05 - 1.45246e-05 - 1.48771e-05 - 1.6778e-05 - 1.98118e-05 - 2.36856e-05 - 2.81654e-05 - 3.30354e-05 - 3.80753e-05 - 4.30518e-05 - 4.77214e-05 - 5.18411e-05 - 5.51843e-05 - 5.75589e-05 - 5.88241e-05 - 5.89031e-05 - 5.77908e-05 - 5.55543e-05 - 5.23255e-05 - 4.82887e-05 - 4.36632e-05 - 3.86831e-05 - 3.35788e-05 - 2.85591e-05 - 2.37992e-05 - 1.94319e-05 - 1.55456e-05 - 1.21852e-05 - 9.35833e-06 - 7.04206e-06 - 5.19204e-06 - 3.7507e-06 - 2.65476e-06 - 1.84108e-06 - 1.251e-06 - 8.32877e-07 - 5.433e-07 - 3.47245e-07 - 2.17455e-07 - 1.33426e-07 - 8.02134e-08 - 4.72494e-08 - 2.72712e-08 - 1.54258e-08 - 8.55829e-09 - 4.67484e-09 - 2.55734e-09 - 1.50298e-09 - 1.16839e-09 - 1.50985e-09 - 2.83243e-09 - 5.99601e-09 - 1.28846e-08 - 2.73441e-08 - 5.6957e-08 - 1.16287e-07 - 2.32641e-07 - 4.56022e-07 - 8.75839e-07 - 1.64816e-06 - 3.03887e-06 - 5.48987e-06 - 9.71737e-06 - 1.68528e-05 - 2.86373e-05 - 4.76792e-05 - 7.77791e-05 - 0.000124318 - 0.000194689 - 0.000298735 - 0.000449125 - 0.000661584 - 0.00095486 - 0.0013503 - 0.00187094 - 0.00253995 - 0.00337853 - 0.00440319 - 0.00562268 - 0.00703487 - 0.00862394 - 0.0103584 - 0.0121903 - 0.0140565 - 0.0158808 - 0.0175795 - 0.0190668 - 0.0202621 - 0.0210974 - 0.0215234 - 0.0215144 - 0.021071 - 0.0202198 - 0.019011 - 0.0175134 - 0.0158078 - 0.0139802 - 0.012114 - 0.0102849 - 0.00855561 - 0.00697329 - 0.00556879 - 0.00435734 - 0.00334055 - 0.0025093 - 0.00184682 - 0.00133178 - 0.00094097 - 0.000651414 - 0.000441851 - 0.00029365 - 0.000191215 - 0.000121998 - 7.62635e-05 - 4.6711e-05 - 2.80323e-05 - 1.6483e-05 - 9.49616e-06 - 5.36041e-06 - 2.96473e-06 - 1.6066e-06 - 8.53039e-07 - 4.43778e-07 - 2.26204e-07 - 1.12972e-07 - 5.52814e-08 - 2.65047e-08 - 1.2451e-08 - 5.73091e-09 - 2.58451e-09 - 1.14202e-09 - 4.94439e-10 - 2.09776e-10 - 8.72944e-11 - 3.58409e-11 - 1.50894e-11 - 7.98683e-12 - 8.47379e-12 - 1.69387e-11 - 4.04834e-11 - 9.81476e-11 - 2.34372e-10 - 5.48791e-10 - 1.2592e-09 - 2.83093e-09 - 6.23592e-09 - 1.34589e-08 - 2.84612e-08 - 5.89705e-08 - 1.19716e-07 - 2.38127e-07 - 4.64089e-07 - 8.86197e-07 - 1.65804e-06 - 3.03948e-06 - 5.45932e-06 - 9.6076e-06 - 1.65664e-05 - 2.79884e-05 - 4.63304e-05 - 7.51432e-05 - 0.000119413 - 0.000185929 - 0.00028365 - 0.000423988 - 0.000620957 - 0.00089106 - 0.00125282 - 0.00172587 - 0.0023295 - 0.00308074 - 0.00399194 - 0.00506816 - 0.00630454 - 0.0076841 - 0.00917634 - 0.010737 - 0.0123093 - 0.0138268 - 0.0152175 - 0.0164099 - 0.0173382 - 0.0179489 - 0.0182058 - 0.0180933 - 0.0176182 - 0.0168091 - 0.0157131 - 0.0143919 - 0.0129155 - 0.0113564 - 0.00978378 - 0.00825866 - 0.00683045 - 0.00553511 - 0.0043948 - 0.00341893 - 0.00260602 - 0.00194626 - 0.00142417 - 0.00102108 - 0.00071729 - 0.000493703 - 0.000332947 - 0.000219998 - 0.00014243 - 9.03484e-05 - 5.61534e-05 - 3.41955e-05 - 2.04032e-05 - 1.19279e-05 - 6.83231e-06 - 3.83449e-06 - 2.10855e-06 - 1.13605e-06 - 5.9972e-07 - 3.10196e-07 - 1.57203e-07 - 7.80587e-08 - 3.79768e-08 - 1.81031e-08 - 8.45523e-09 - 3.86931e-09 - 1.73492e-09 - 7.62186e-10 - 3.2808e-10 - 1.38368e-10 - 5.71779e-11 - 2.31504e-11 - 9.18384e-12 - 3.56967e-12 - 1.35946e-12 - 5.07276e-13 - 1.85463e-13 - 6.64368e-14 - 2.33183e-14 - 8.01901e-15 - 2.70198e-15 - 8.9203e-16 - 2.88546e-16 - 9.14505e-17 - 2.83985e-17 - 8.64052e-18 - 2.57586e-18 - 7.52387e-19 - 2.15326e-19 - 6.03798e-20 - 1.65898e-20 - 4.46888e-21 - 1.1907e-21 - 3.54724e-22 - 2.70303e-22 - 7.45922e-22 - 2.68317e-21 - 9.67303e-21 - 3.42192e-20 - 1.1862e-19 - 4.02889e-19 - 1.34076e-18 - 4.37171e-18 - 1.39666e-17 - 4.37185e-17 - 1.34084e-16 - 4.02927e-16 - 1.18635e-15 - 3.42243e-15 - 9.67374e-15 - 2.67911e-14 - 7.26984e-14 - 1.93284e-13 - 5.03504e-13 - 1.28513e-12 - 3.21388e-12 - 7.87496e-12 - 1.89062e-11 - 4.44733e-11 - 1.02502e-10 - 2.31473e-10 - 5.12165e-10 - 1.11034e-09 - 2.35855e-09 - 4.90879e-09 - 1.00102e-08 - 2.00013e-08 - 3.91579e-08 - 7.51153e-08 - 1.41185e-07 - 2.60021e-07 - 4.69236e-07 - 8.29749e-07 - 1.43776e-06 - 2.44132e-06 - 4.06237e-06 - 6.62491e-06 - 1.0589e-05 - 1.65903e-05 - 2.54819e-05 - 3.83764e-05 - 5.66822e-05 - 8.21314e-05 - 0.000116792 - 0.00016307 - 0.000223695 - 0.000301717 - 0.000400522 - 0.000523905 - 0.000676234 - 0.000862732 - 0.00108992 - 0.00136617 - 0.00170242 - 0.00211279 - 0.00261515 - 0.00323128 - 0.00398644 - 0.0049082 - 0.00602429 - 0.00735944 - 0.00893147 - 0.0107468 - 0.0127958 - 0.0150493 - 0.0174561 - 0.0199423 - 0.0224143 - 0.0247631 - 0.0268725 - 0.0286279 - 0.0299271 - 0.0306897 - 0.0308656 - 0.0304406 - 0.0294379 - 0.027917 - 0.0259682 - 0.0237049 - 0.0212543 - 0.0187473 - 0.0163094 - 0.0140532 - 0.012073 - 0.0104412 - 0.00920708 - 0.00839714 - 0.00801617 - 0.00804901 - 0.00846236 - 0.00920671 - 0.0102183 - 0.0114214 - 0.0127314 - 0.0140583 - 0.0153113 - 0.0164042 - 0.0172608 - 0.0178203 - 0.0180418 - 0.0179068 - 0.017421 - 0.0166126 - 0.0155294 - 0.0142346 - 0.0128009 - 0.0113044 - 0.00981974 - 0.00841484 - 0.00714807 - 0.00606625 - 0.00520401 - 0.00458421 - 0.00421879 - 0.00410993 - 0.00425091 - 0.00462675 - 0.00521439 - 0.00598278 - 0.0068929 - 0.00789827 - 0.00894605 - 0.00997897 - 0.0109382 - 0.0117671 - 0.0124148 - 0.0128403 - 0.0130161 - 0.01293 - 0.0125862 - 0.0120047 - 0.011219 - 0.0102732 - 0.00921717 - 0.00810268 - 0.00697907 - 0.00588986 - 0.00487025 - 0.00394581 - 0.00313231 - 0.00243638 - 0.00185693 - 0.00138696 - 0.00101542 - 0.000729103 - 0.000514093 - 0.000357004 - 0.000245774 - 0.00017014 - 0.00012182 - 9.44836e-05 - 8.35732e-05 - 8.60424e-05 - 0.000100046 - 0.000124617 - 0.000159347 - 0.000204081 - 0.000258632 - 0.000322534 - 0.000394841 - 0.000473977 - 0.000557675 - 0.000642995 - 0.000726438 - 0.000804153 - 0.00087221 - 0.00092692 - 0.000965165 - 0.000984686 - 0.000984308 - 0.000964054 - 0.000925143 - 0.000869867 - 0.000801371 - 0.000723355 - 0.000639744 - 0.000554367 - 0.00047068 - 0.000391553 - 0.000319149 - 0.000254878 - 0.000199438 - 0.000152904 - 0.00011486 - 8.45387e-05 - 6.09647e-05 - 4.30763e-05 - 2.98219e-05 - 2.02287e-05 - 1.34443e-05 - 8.75481e-06 - 5.58588e-06 - 3.49199e-06 - 2.13892e-06 - 1.2837e-06 - 7.54953e-07 - 4.3522e-07 - 2.46288e-07 - 1.37595e-07 - 7.76206e-08 - 4.7872e-08 - 3.91206e-08 - 4.96818e-08 - 8.57782e-08 - 1.64423e-07 - 3.19802e-07 - 6.1486e-07 - 1.16078e-06 - 2.1483e-06 - 3.8961e-06 - 6.92335e-06 - 1.20543e-05 - 2.0564e-05 - 3.43723e-05 - 5.6292e-05 - 9.03278e-05 - 0.000142015 - 0.000218767 - 0.000330193 - 0.000488304 - 0.000707537 - 0.00100449 - 0.00139726 - 0.00190436 - 0.00254305 - 0.00332735 - 0.00426559 - 0.00535792 - 0.00659401 - 0.00795135 - 0.00939439 - 0.0108751 - 0.0123349 - 0.013708 - 0.0149262 - 0.0159243 - 0.016646 - 0.0170489 - 0.0171089 - 0.0168223 - 0.0162065 - 0.0152982 - 0.0141497 - 0.0128242 - 0.01139 - 0.00991533 - 0.00846323 - 0.00708799 - 0.00583302 - 0.00473027 - 0.00380108 - 0.00305813 - 0.00250808 - 0.00215444 - 0.00200017 - 0.00204972 - 0.00231011 - 0.00279091 - 0.003503 - 0.00445631 - 0.00565652 - 0.00710137 - 0.0087768 - 0.0106537 - 0.0126858 - 0.014809 - 0.0169432 - 0.0189962 - 0.0208692 - 0.0224645 - 0.0236937 - 0.0244855 - 0.0247927 - 0.0245967 - 0.0239093 - 0.0227716 - 0.0212499 - 0.0194294 - 0.0174059 - 0.0152782 - 0.0131396 - 0.0110721 - 0.00914149 - 0.00739501 - 0.00586135 - 0.00455191 - 0.00346359 - 0.00258224 - 0.00188626 - 0.00135004 - 0.000946729 - 0.000650493 - 0.000437923 - 0.000288862 - 0.000186692 - 0.000118227 - 7.33688e-05 - 4.46338e-05 - 2.66504e-05 - 1.56821e-05 - 9.21728e-06 - 5.63909e-06 - 3.97946e-06 - 3.75189e-06 - 4.85588e-06 - 7.54564e-06 - 1.24576e-05 - 2.06929e-05 - 3.39515e-05 - 5.47126e-05 - 8.64515e-05 - 0.000133876 - 0.000203149 - 0.000302064 - 0.000440104 - 0.00062834 - 0.000879078 - 0.00120523 - 0.00161937 - 0.00213248 - 0.00275244 - 0.00348253 - 0.00431992 - 0.00525458 - 0.00626873 - 0.00733715 - 0.00842831 - 0.00950652 - 0.0105347 - 0.0114778 - 0.0123057 - 0.0129966 - 0.0135384 - 0.0139296 - 0.0141784 - 0.0143005 - 0.0143162 - 0.0142468 - 0.0141109 - 0.0139217 - 0.0136857 - 0.0134019 - 0.0130632 - 0.012659 - 0.0121776 - 0.0116098 - 0.0109513 - 0.0102046 - 0.00937978 - 0.00849391 - 0.00756962 - 0.00663291 - 0.00571057 - 0.00482772 - 0.00400577 - 0.00326099 - 0.00260377 - 0.00203868 - 0.00156499 - 0.00117773 - 0.000868844 - 0.000628451 - 0.000445965 - 0.00031104 - 0.00021428 - 0.000147749 - 0.000105306 - 8.28571e-05 - 7.85795e-05 - 9.31949e-05 - 0.000130355 - 0.000197161 - 0.00030483 - 0.000469447 - 0.000712728 - 0.00106262 - 0.00155357 - 0.00222613 - 0.00312575 - 0.00430043 - 0.00579713 - 0.0076569 - 0.00990902 - 0.0125645 - 0.0156098 - 0.0190014 - 0.0226627 - 0.0264835 - 0.0303231 - 0.0340181 - 0.0373924 - 0.0402711 - 0.0424952 - 0.0439363 - 0.0445086 - 0.0441776 - 0.0429631 - 0.040938 - 0.0382204 - 0.0349623 - 0.0313358 - 0.0275182 - 0.0236774 - 0.0199612 - 0.0164883 - 0.0133445 - 0.0105819 - 0.00822174 - 0.00625893 - 0.00466846 - 0.0034118 - 0.00244304 - 0.00171401 - 0.00117824 - 0.000793582 - 0.000523705 - 0.000338624 - 0.000214529 - 0.000133165 - 8.09901e-05 - 4.82626e-05 - 2.8179e-05 - 1.61205e-05 - 9.03581e-06 - 4.96241e-06 - 2.67027e-06 - 1.40785e-06 - 7.27263e-07 - 3.68099e-07 - 1.82547e-07 - 8.87001e-08 - 4.22295e-08 - 1.97008e-08 - 9.00956e-09 - 4.04833e-09 - 1.81029e-09 - 8.60698e-10 - 5.58336e-10 - 6.85971e-10 - 1.31585e-09 - 2.8408e-09 - 6.1788e-09 - 1.32343e-08 - 2.77987e-08 - 5.7221e-08 - 1.15408e-07 - 2.28062e-07 - 4.4158e-07 - 8.37727e-07 - 1.55716e-06 - 2.83595e-06 - 5.0606e-06 - 8.84794e-06 - 1.51572e-05 - 2.5441e-05 - 4.18393e-05 - 6.74174e-05 - 0.000106438 - 0.000164648 - 0.000249549 - 0.000370588 - 0.000539216 - 0.000768726 - 0.00107378 - 0.0014696 - 0.00197069 - 0.00258926 - 0.00333325 - 0.00420434 - 0.00519594 - 0.0062917 - 0.00746463 - 0.00867732 - 0.00988325 - 0.0110294 - 0.0120597 - 0.01292 - 0.013562 - 0.0139483 - 0.0140559 - 0.0138781 - 0.0134257 - 0.0127258 - 0.0118186 - 0.0107544 - 0.00958832 - 0.00837598 - 0.00716911 - 0.00601218 - 0.0049401 - 0.00397719 - 0.00313728 - 0.00242475 - 0.00183619 - 0.00136241 - 0.000990447 - 0.000705493 - 0.00049237 - 0.000336687 - 0.000225579 - 0.000148084 - 9.52474e-05 - 6.00255e-05 - 3.70643e-05 - 2.24239e-05 - 1.32924e-05 - 7.72031e-06 - 4.39341e-06 - 2.44966e-06 - 1.33828e-06 - 7.16347e-07 - 3.75697e-07 - 1.93059e-07 - 9.7204e-08 - 4.79558e-08 - 2.31886e-08 - 1.10053e-08 - 5.16537e-09 - 2.49231e-09 - 1.45573e-09 - 1.44813e-09 - 2.49265e-09 - 5.27515e-09 - 1.15271e-08 - 2.49327e-08 - 5.29399e-08 - 1.10176e-07 - 2.24677e-07 - 4.48922e-07 - 8.78862e-07 - 1.68581e-06 - 3.16834e-06 - 5.83436e-06 - 1.05267e-05 - 1.86091e-05 - 3.22327e-05 - 5.47023e-05 - 9.09601e-05 - 0.000148195 - 0.000236566 - 0.000370005 - 0.000567022 - 0.000851392 - 0.00125255 - 0.00180551 - 0.00254999 - 0.00352871 - 0.00478442 - 0.00635594 - 0.00827308 - 0.010551 - 0.0131842 - 0.0161418 - 0.0193636 - 0.0227592 - 0.02621 - 0.0295742 - 0.0326961 - 0.0354175 - 0.0375905 - 0.0390913 - 0.0398314 - 0.0397669 - 0.0389028 - 0.0372926 - 0.0350338 - 0.0322585 - 0.0291224 - 0.0257915 - 0.02243 - 0.0191902 - 0.0162049 - 0.0135829 - 0.0114083 - 0.00974183 - 0.00862375 - 0.00807727 - 0.00811156 - 0.00872333 - 0.00989679 - 0.011602 - 0.0137919 - 0.0163994 - 0.0193348 - 0.0224847 - 0.0257144 - 0.0288725 - 0.0317988 - 0.0343349 - 0.0363363 - 0.0376843 - 0.0382965 - 0.0381344 - 0.037207 - 0.0355694 - 0.0333172 - 0.0305773 - 0.0274959 - 0.0242255 - 0.020913 - 0.0176887 - 0.0146593 - 0.0119033 - 0.00947019 - 0.00738221 - 0.00563834 - 0.00421942 - 0.0030938 - 0.00222264 - 0.00156455 - 0.0010791 - 0.000729329 - 0.000483146 - 0.000313949 - 0.000200581 - 0.000126914 - 8.12404e-05 - 5.5656e-05 - 4.55669e-05 - 4.94212e-05 - 6.87538e-05 - 0.000108612 - 0.000178427 - 0.000293379 - 0.000476297 - 0.000760061 - 0.00119043 - 0.00182912 - 0.00275668 - 0.00407487 - 0.00590754 - 0.00839955 - 0.0117125 - 0.0160171 - 0.0214803 - 0.0282497 - 0.0364326 - 0.0460744 - 0.0571362 - 0.0694755 - 0.0828344 - 0.0968356 - 0.110993 - 0.124731 - 0.137426 - 0.148443 - 0.157195 - 0.163191 - 0.16608 - 0.16569 - 0.162039 - 0.155338 - 0.145969 - 0.13445 - 0.121386 - 0.107417 - 0.0931679 - 0.0792032 - 0.0659922 - 0.0538902 - 0.0431307 - 0.0338313 - 0.0260076 - 0.0195944 - 0.0144686 - 0.0104718 - 0.00743065 - 0.00517275 - 0.00353869 - 0.0023892 - 0.00160904 - 0.00110793 - 0.000819713 - 0.000700487 - 0.000726235 - 0.000890471 - 0.00120196 - 0.00168246 - 0.00236436 - 0.00328777 - 0.00449696 - 0.00603582 - 0.00794235 - 0.0102424 - 0.0129428 - 0.0160253 - 0.0194414 - 0.0231094 - 0.0269146 - 0.0307131 - 0.0343397 - 0.0376189 - 0.0403787 - 0.0424655 - 0.0437579 - 0.0441788 - 0.0437027 - 0.0423583 - 0.040226 - 0.0374293 - 0.0341234 - 0.0304811 - 0.0266776 - 0.022877 - 0.0192215 - 0.0158239 - 0.0127637 - 0.0100873 - 0.00781109 - 0.00592634 - 0.00440555 - 0.00320891 - 0.0022902 - 0.00160171 - 0.00109801 - 0.000738346 - 0.000488074 - 0.000319114 - 0.000209886 - 0.000144899 - 0.000114287 - 0.00011345 - 0.000142967 - 0.000208865 - 0.000323272 - 0.000505441 - 0.000783052 - 0.00119362 - 0.00178573 - 0.00261982 - 0.00376788 - 0.00531191 - 0.00734032 - 0.0099422 - 0.0131993 - 0.0171757 - 0.0219067 - 0.0273861 - 0.0335566 - 0.0403008 - 0.0474392 - 0.0547326 - 0.0618925 - 0.0685978 - 0.074518 - 0.0793391 - 0.0827917 - 0.0846753 - 0.084878 - 0.0833871 - 0.0802907 - 0.0757691 - 0.0700771 - 0.0635207 - 0.0564296 - 0.0491301 - 0.0419213 - 0.0350564 - 0.0287303 - 0.0230756 - 0.0181636 - 0.0140114 - 0.0105924 - 0.00784763 - 0.00569779 - 0.00405413 - 0.0028269 - 0.00193171 - 0.00129357 - 0.000848892 - 0.000545916 - 0.00034404 - 0.000212471 - 0.000128587 - 7.62598e-05 - 4.43196e-05 - 2.52403e-05 - 1.40861e-05 - 7.70341e-06 - 4.12826e-06 - 2.16792e-06 - 1.1156e-06 - 5.62553e-07 - 2.77973e-07 - 1.34595e-07 - 6.38609e-08 - 2.96909e-08 - 1.35267e-08 - 6.03865e-09 - 2.64158e-09 - 1.13231e-09 - 4.75596e-10 - 1.95742e-10 - 7.89412e-11 - 3.11955e-11 - 1.20796e-11 - 4.58331e-12 - 1.70407e-12 - 6.20988e-13 - 2.22293e-13 - 7.9713e-14 - 3.33658e-14 - 2.94752e-14 - 6.38018e-14 - 1.77569e-13 - 5.02538e-13 - 1.39959e-12 - 3.82147e-12 - 1.02254e-11 - 2.6812e-11 - 6.88931e-11 - 1.73468e-10 - 4.28014e-10 - 1.03489e-09 - 2.45205e-09 - 5.69327e-09 - 1.29536e-08 - 2.88814e-08 - 6.31019e-08 - 1.35103e-07 - 2.83456e-07 - 5.82778e-07 - 1.17414e-06 - 2.3181e-06 - 4.48481e-06 - 8.50263e-06 - 1.57965e-05 - 2.87584e-05 - 5.13057e-05 - 8.96943e-05 - 0.00015366 - 0.000257961 - 0.00042437 - 0.000684119 - 0.00108072 - 0.001673 - 0.00253788 - 0.00377261 - 0.00549553 - 0.00784462 - 0.0109731 - 0.0150412 - 0.0202036 - 0.0265931 - 0.0343008 - 0.0433543 - 0.0536975 - 0.0651733 - 0.0775138 - 0.0903404 - 0.103176 - 0.115472 - 0.12664 - 0.136103 - 0.143344 - 0.147949 - 0.149655 - 0.14837 - 0.144186 - 0.137373 - 0.128353 - 0.117658 - 0.105889 - 0.093658 - 0.0815473 - 0.0700646 - 0.0596157 - 0.0504871 - 0.0428422 - 0.0367268 - 0.0320839 - 0.0287735 - 0.0265961 - 0.0253153 - 0.0246804 - 0.024446 - 0.0243882 - 0.0243168 - 0.0240838 - 0.0235875 - 0.0227725 - 0.021627 - 0.0201755 - 0.0184715 - 0.0165872 - 0.014604 - 0.0126035 - 0.0106603 - 0.00883602 - 0.00717679 - 0.00571177 - 0.00445418 - 0.00340341 - 0.00254803 - 0.00186912 - 0.00134342 - 0.000946069 - 0.000652789 - 0.000441327 - 0.000292338 - 0.000189735 - 0.000120655 - 7.51766e-05 - 4.5894e-05 - 2.74515e-05 - 1.60884e-05 - 9.2384e-06 - 5.19778e-06 - 2.86533e-06 - 1.54764e-06 - 8.19038e-07 - 4.24702e-07 - 2.15801e-07 - 1.07503e-07 - 5.26328e-08 - 2.56405e-08 - 1.31726e-08 - 8.77456e-09 - 1.02078e-08 - 1.85691e-08 - 3.89947e-08 - 8.33138e-08 - 1.75745e-07 - 3.63781e-07 - 7.38012e-07 - 1.46706e-06 - 2.85743e-06 - 5.45305e-06 - 1.01963e-05 - 1.86801e-05 - 3.35317e-05 - 5.89749e-05 - 0.000101629 - 0.000171594 - 0.000283873 - 0.000460134 - 0.00073077 - 0.00113714 - 0.00173374 - 0.00258995 - 0.00379084 - 0.00543645 - 0.00763894 - 0.0105169 - 0.0141866 - 0.0187503 - 0.0242813 - 0.0308088 - 0.0383013 - 0.0466542 - 0.0556807 - 0.0651114 - 0.0746018 - 0.0837495 - 0.0921213 - 0.0992861 - 0.104852 - 0.108503 - 0.110031 - 0.109357 - 0.106543 - 0.101791 - 0.0954271 - 0.087877 - 0.0796374 - 0.0712426 - 0.0632336 - 0.0561318 - 0.0504187 - 0.0465218 - 0.044805 - 0.045561 - 0.0490042 - 0.0552601 - 0.0643534 - 0.0761915 - 0.0905495 - 0.107057 - 0.125194 - 0.144296 - 0.163577 - 0.182163 - 0.199146 - 0.213648 - 0.224879 - 0.232211 - 0.235225 - 0.233747 - 0.227864 - 0.217912 - 0.204443 - 0.188177 - 0.169934 - 0.150572 - 0.130918 - 0.111717 - 0.0935927 - 0.0770224 - 0.0623326 - 0.0497065 - 0.0392039 - 0.0307866 - 0.0243474 - 0.0197367 - 0.0167861 - 0.0153248 - 0.0151908 - 0.0162352 - 0.0183228 - 0.0213286 - 0.0251345 - 0.0296256 - 0.0346886 - 0.0402135 - 0.0460961 - 0.052243 - 0.0585756 - 0.0650326 - 0.0715684 - 0.0781472 - 0.0847335 - 0.091278 - 0.0977038 - 0.103894 - 0.109684 - 0.114865 - 0.119189 - 0.122391 - 0.124217 - 0.124448 - 0.122933 - 0.119609 - 0.114518 - 0.107811 - 0.0997331 - 0.0906066 - 0.0808031 - 0.0707102 - 0.0607003 - 0.0511033 - 0.0421863 - 0.0341421 - 0.0270865 - 0.0210628 - 0.0160525 - 0.0119898 - 0.00877597 - 0.00629478 - 0.00442438 - 0.00304721 - 0.00205646 - 0.00135988 - 0.000881132 - 0.000559434 - 0.000348064 - 0.000212277 - 0.000127044 - 7.49014e-05 - 4.40862e-05 - 2.70444e-05 - 1.93388e-05 - 1.89574e-05 - 2.60199e-05 - 4.28955e-05 - 7.47722e-05 - 0.00013075 - 0.000225561 - 0.000382019 - 0.000634315 - 0.00103218 - 0.00164583 - 0.00257148 - 0.00393681 - 0.00590566 - 0.00868071 - 0.0125027 - 0.0176447 - 0.0243998 - 0.0330614 - 0.0438952 - 0.0571052 - 0.0727939 - 0.0909236 - 0.111281 - 0.133452 - 0.156816 - 0.180558 - 0.203706 - 0.225193 - 0.24393 - 0.258902 - 0.269258 - 0.274385 - 0.273977 - 0.268058 - 0.256982 - 0.241401 - 0.222195 - 0.200397 - 0.177096 - 0.153352 - 0.130119 - 0.108184 - 0.088143 - 0.0703811 - 0.0550906 - 0.0422952 - 0.0318886 - 0.0236757 - 0.0174137 - 0.0128489 - 0.00974575 - 0.00790899 - 0.00719695 - 0.00752776 - 0.00887847 - 0.0112782 - 0.0147954 - 0.0195211 - 0.0255467 - 0.03294 - 0.0417194 - 0.0518294 - 0.0631202 - 0.0753344 - 0.0881046 - 0.100962 - 0.113362 - 0.124714 - 0.134431 - 0.141978 - 0.146921 - 0.148964 - 0.147984 - 0.144041 - 0.137371 - 0.128363 - 0.117523 - 0.105425 - 0.0926619 - 0.0797996 - 0.0673356 - 0.0556729 - 0.0451043 - 0.0358108 - 0.0278695 - 0.0212705 - 0.0159377 - 0.0117507 - 0.0085666 - 0.0062372 - 0.00462324 - 0.00360352 - 0.00307992 - 0.00297862 - 0.00324833 - 0.00385612 - 0.00478184 - 0.00601151 - 0.00753051 - 0.0093171 - 0.011337 - 0.0135396 - 0.0158561 - 0.0182006 - 0.0204731 - 0.0225658 - 0.0243709 - 0.0257892 - 0.0267388 - 0.0271636 - 0.0270377 - 0.0263689 - 0.0251973 - 0.0235919 - 0.0216433 - 0.019456 - 0.0171389 - 0.014797 - 0.0125243 - 0.0103983 - 0.00847798 - 0.00680284 - 0.00539503 - 0.00426232 - 0.00340205 - 0.00280523 - 0.00246013 - 0.00235501 - 0.00247968 - 0.00282577 - 0.00338583 - 0.00415139 - 0.00511045 - 0.00624471 - 0.00752714 - 0.00892027 - 0.0103758 - 0.0118353 - 0.0132334 - 0.0145009 - 0.0155705 - 0.0163822 - 0.0168885 - 0.017059 - 0.0168832 - 0.0163718 - 0.0155551 - 0.0144807 - 0.0132081 - 0.011804 - 0.0103361 - 0.00886787 - 0.00745451 - 0.00613983 - 0.00495484 - 0.00391778 - 0.0030352 - 0.00230394 - 0.00171353 - 0.00124868 - 0.000891548 - 0.000623701 - 0.000427509 - 0.000287111 - 0.000188926 - 0.000121807 - 7.69461e-05 - 4.76255e-05 - 2.88821e-05 - 1.71615e-05 - 9.99121e-06 - 5.69926e-06 - 3.18533e-06 - 1.74433e-06 - 9.35921e-07 - 4.92024e-07 - 2.53438e-07 - 1.27906e-07 - 6.32485e-08 - 3.06439e-08 - 1.45471e-08 - 6.7662e-09 - 3.08354e-09 - 1.37687e-09 - 6.02381e-10 - 2.58218e-10 - 1.08453e-10 - 4.46303e-11 - 1.79952e-11 - 7.10917e-12 - 2.75181e-12 - 1.04365e-12 - 3.8782e-13 - 1.41202e-13 - 5.03719e-14 - 1.76065e-14 - 6.02967e-15 - 2.02326e-15 - 6.65191e-16 - 2.14278e-16 - 6.76311e-17 - 2.09147e-17 - 6.33715e-18 - 1.88137e-18 - 5.47254e-19 - 1.5597e-19 - 4.35544e-20 - 1.19168e-20 - 3.19464e-21 - 8.39116e-22 - 2.15953e-22 - 5.44544e-23 - 1.34538e-23 - 3.2568e-24 - 7.72459e-25 - 1.79513e-25 - 4.08747e-26 - 9.11906e-27 - 1.99334e-27 - 4.26924e-28 - 8.95892e-29 - 1.84204e-29 - 3.71088e-30 - 7.32476e-31 - 1.4166e-31 - 2.68433e-32 - 4.98382e-33 - 9.06622e-34 - 1.61594e-34 - 2.82204e-35 - 4.82878e-36 - 8.09559e-37 - 1.32983e-37 - 2.14033e-38 - 3.37522e-39 - 5.21507e-40 - 7.89506e-41 - 1.17108e-41 - 1.70199e-42 - 2.4236e-43 - 3.38146e-44 - 4.62257e-45 - 6.19155e-46 - 8.12554e-47 - 1.04482e-47 - 1.31634e-48 - 1.62492e-49 - 1.96532e-50 - 2.329e-51 - 2.70423e-52 - 3.07649e-53 - 3.42928e-54 - 3.74531e-55 - 4.00783e-56 - 4.20211e-57 - 4.31681e-58 - 4.34505e-59 - 4.28512e-60 - 4.14066e-61 - 3.92023e-62 - 3.63656e-63 - 3.30527e-64 - 2.94348e-65 - 2.56833e-66 - 2.19572e-67 - 1.83925e-68 - 1.50953e-69 - 1.21389e-70 - 9.56429e-72 - 7.38353e-73 - 5.58485e-74 - 4.13901e-75 - 3.00551e-76 - 2.13834e-77 - 1.49064e-78 - 1.01813e-79 - 6.81354e-81 - 4.46764e-82 - 2.87026e-83 - 1.80676e-84 - 1.11434e-85 - 6.73396e-87 - 3.98713e-88 - 2.31306e-89 - 1.31477e-90 - 7.32237e-92 - 3.99566e-93 - 2.1363e-94 - 1.11911e-95 - 5.7441e-97 - 2.88873e-98 - 1.42341e-99 - 6.87207e-101 - 3.25074e-102 - 1.50666e-103 - 6.84201e-105 - 3.04432e-106 - 1.32719e-107 - 5.66909e-109 - 2.37623e-110 - 1.87428e-111 - 2.20949e-110 - 5.28801e-109 - 1.24221e-107 - 2.85914e-106 - 6.44781e-105 - 1.42471e-103 - 3.08445e-102 - 6.54282e-101 - 1.35985e-99 - 2.76918e-98 - 5.52523e-97 - 1.08015e-95 - 2.06899e-94 - 3.88299e-93 - 7.14023e-92 - 1.28646e-90 - 2.27099e-89 - 3.92799e-88 - 6.65678e-87 - 1.10534e-85 - 1.79829e-84 - 2.86658e-83 - 4.47719e-82 - 6.85145e-81 - 1.0273e-79 - 1.5092e-78 - 2.17238e-77 - 3.0638e-76 - 4.23372e-75 - 5.7322e-74 - 7.60425e-73 - 9.88391e-72 - 1.25875e-70 - 1.57067e-69 - 1.92029e-68 - 2.30031e-67 - 2.69987e-66 - 3.10482e-65 - 3.49837e-64 - 3.86218e-63 - 4.1777e-62 - 4.42769e-61 - 4.59785e-60 - 4.6781e-59 - 4.66359e-58 - 4.55521e-57 - 4.35947e-56 - 4.08786e-55 - 3.75573e-54 - 3.38088e-53 - 2.98196e-52 - 2.57698e-51 - 2.18201e-50 - 1.81026e-49 - 1.4715e-48 - 1.17197e-47 - 9.14556e-47 - 6.99263e-46 - 5.23851e-45 - 3.84514e-44 - 2.76537e-43 - 1.94863e-42 - 1.34538e-41 - 9.10115e-41 - 6.03233e-40 - 3.91751e-39 - 2.49271e-38 - 1.55407e-37 - 9.49308e-37 - 5.68172e-36 - 3.33188e-35 - 1.91442e-34 - 1.07775e-33 - 5.94484e-33 - 3.2129e-32 - 1.70134e-31 - 8.8272e-31 - 4.48736e-30 - 2.2351e-29 - 1.09078e-28 - 5.21577e-28 - 2.44362e-27 - 1.12173e-26 - 5.0452e-26 - 2.22334e-25 - 9.6e-25 - 4.06138e-24 - 1.6835e-23 - 6.83739e-23 - 2.72085e-22 - 1.06086e-21 - 4.05272e-21 - 1.51696e-20 - 5.56339e-20 - 1.99914e-19 - 7.03857e-19 - 2.42809e-18 - 8.20698e-18 - 2.71795e-17 - 8.81938e-17 - 2.80398e-16 - 8.73476e-16 - 2.66605e-15 - 7.97306e-15 - 2.33627e-14 - 6.70756e-14 - 1.8869e-13 - 5.20086e-13 - 1.40458e-12 - 3.71676e-12 - 9.63669e-12 - 2.44816e-11 - 6.09399e-11 - 1.48633e-10 - 3.55208e-10 - 8.31775e-10 - 1.90848e-09 - 4.29072e-09 - 9.45234e-09 - 2.04041e-08 - 4.31586e-08 - 8.94536e-08 - 1.81682e-07 - 3.61591e-07 - 7.05214e-07 - 1.34782e-06 - 2.52441e-06 - 4.63355e-06 - 8.33507e-06 - 1.46946e-05 - 2.53909e-05 - 4.30015e-05 - 7.13836e-05 - 0.000116157 - 0.000185287 - 0.000289756 - 0.000444259 - 0.000667872 - 0.000984563 - 0.00142341 - 0.00201839 - 0.00280749 - 0.00383118 - 0.00512992 - 0.00674098 - 0.0086945 - 0.0110092 - 0.013688 - 0.0167144 - 0.020049 - 0.0236291 - 0.0273681 - 0.0311587 - 0.0348773 - 0.0383902 - 0.0415613 - 0.0442609 - 0.0463735 - 0.0478061 - 0.0484944 - 0.0484071 - 0.0475482 - 0.0459567 - 0.043704 - 0.0408888 - 0.0376304 - 0.0340605 - 0.0303155 - 0.0265272 - 0.022816 - 0.0192849 - 0.016015 - 0.013064 - - - - - 2.23487e-32 - 2.78124e-32 - 3.39127e-32 - 4.05161e-32 - 4.74298e-32 - 5.44146e-32 - 6.12403e-32 - 6.79326e-32 - 7.59977e-32 - 9.46233e-32 - 1.71817e-31 - 5.47915e-31 - 2.397e-30 - 1.13335e-29 - 5.365e-29 - 2.49937e-28 - 1.14186e-27 - 5.11222e-27 - 2.24264e-26 - 9.63936e-26 - 4.05952e-25 - 1.67509e-24 - 6.77235e-24 - 2.68273e-23 - 1.04124e-22 - 3.9597e-22 - 1.4754e-21 - 5.38636e-21 - 1.92671e-20 - 6.75267e-20 - 2.31884e-19 - 7.80195e-19 - 2.57201e-18 - 8.30765e-18 - 2.62918e-17 - 8.15268e-17 - 2.47695e-16 - 7.37345e-16 - 2.15061e-15 - 6.14593e-15 - 1.72088e-14 - 4.7212e-14 - 1.26908e-13 - 3.34245e-13 - 8.62533e-13 - 2.18084e-12 - 5.40267e-12 - 1.31139e-11 - 3.11881e-11 - 7.26749e-11 - 1.65927e-10 - 3.71181e-10 - 8.13565e-10 - 1.74717e-09 - 3.67633e-09 - 7.57934e-09 - 1.53103e-08 - 3.03023e-08 - 5.87628e-08 - 1.11652e-07 - 2.07858e-07 - 3.79145e-07 - 6.77609e-07 - 1.18656e-06 - 2.03582e-06 - 3.42235e-06 - 5.63697e-06 - 9.09714e-06 - 1.43847e-05 - 2.2286e-05 - 3.383e-05 - 5.03162e-05 - 7.33248e-05 - 0.000104696 - 0.00014647 - 0.000200771 - 0.000269645 - 0.000354829 - 0.000457491 - 0.000577942 - 0.000715355 - 0.000867554 - 0.00103088 - 0.00120021 - 0.00136913 - 0.00153027 - 0.00167583 - 0.00179817 - 0.00189049 - 0.00194743 - 0.00196562 - 0.00194402 - 0.001884 - 0.00178928 - 0.00166553 - 0.00151987 - 0.00136029 - 0.00119498 - 0.00103175 - 0.000877556 - 0.000738187 - 0.000618089 - 0.000520341 - 0.000446731 - 0.000397902 - 0.000373517 - 0.000372412 - 0.000392709 - 0.000431897 - 0.000486877 - 0.000554005 - 0.000629151 - 0.000707798 - 0.000785206 - 0.00085663 - 0.000917596 - 0.000964193 - 0.000993359 - 0.00100311 - 0.000992716 - 0.000962696 - 0.000914787 - 0.000851735 - 0.000777026 - 0.000694558 - 0.000608305 - 0.000522003 - 0.000438896 - 0.000361567 - 0.000291845 - 0.000230809 - 0.000178851 - 0.000135789 - 0.000101013 - 7.3625e-05 - 5.25788e-05 - 3.67903e-05 - 2.52227e-05 - 1.69429e-05 - 1.11512e-05 - 7.19101e-06 - 4.54356e-06 - 2.8128e-06 - 1.70616e-06 - 1.014e-06 - 5.90459e-07 - 3.36884e-07 - 1.88325e-07 - 1.03151e-07 - 5.53571e-08 - 2.91079e-08 - 1.49964e-08 - 7.57003e-09 - 3.74409e-09 - 1.81439e-09 - 8.61497e-10 - 4.00786e-10 - 1.82687e-10 - 8.15908e-11 - 3.57036e-11 - 1.53081e-11 - 6.43124e-12 - 2.64844e-12 - 1.07197e-12 - 4.34795e-13 - 2.00078e-13 - 1.63936e-13 - 3.00174e-13 - 7.46316e-13 - 1.9281e-12 - 4.92178e-12 - 1.23241e-11 - 3.02408e-11 - 7.27073e-11 - 1.71278e-10 - 3.95331e-10 - 8.94041e-10 - 1.98103e-09 - 4.30092e-09 - 9.1489e-09 - 1.90683e-08 - 3.89398e-08 - 7.79133e-08 - 1.52745e-07 - 2.93398e-07 - 5.52187e-07 - 1.01824e-06 - 1.83973e-06 - 3.25681e-06 - 5.64895e-06 - 9.60021e-06 - 1.59856e-05 - 2.60805e-05 - 4.16907e-05 - 6.52979e-05 - 0.000100207 - 0.000150671 - 0.000221973 - 0.000320411 - 0.00045316 - 0.000627962 - 0.000852611 - 0.00113424 - 0.00147842 - 0.0018881 - 0.0023626 - 0.00289663 - 0.00347962 - 0.00409551 - 0.00472303 - 0.00533667 - 0.00590822 - 0.00640885 - 0.00681147 - 0.00709314 - 0.00723725 - 0.00723514 - 0.00708696 - 0.00680167 - 0.00639614 - 0.00589358 - 0.00532131 - 0.00470844 - 0.00408352 - 0.00347255 - 0.00289752 - 0.00237559 - 0.00191887 - 0.00153483 - 0.00122706 - 0.000996256 - 0.000841365 - 0.000760476 - 0.000751552 - 0.000812793 - 0.000942642 - 0.00113944 - 0.00140081 - 0.00172285 - 0.0020993 - 0.00252088 - 0.00297478 - 0.00344473 - 0.00391135 - 0.00435317 - 0.00474797 - 0.00507448 - 0.00531415 - 0.00545286 - 0.00548225 - 0.00540051 - 0.00521259 - 0.00492967 - 0.00456811 - 0.00414791 - 0.00369091 - 0.00321908 - 0.00275291 - 0.00231029 - 0.00190581 - 0.00155063 - 0.00125284 - 0.00101825 - 0.000851451 - 0.000756966 - 0.000740438 - 0.000809583 - 0.00097481 - 0.00124937 - 0.00164889 - 0.00219032 - 0.00289009 - 0.0037618 - 0.00481339 - 0.00604416 - 0.00744202 - 0.00898139 - 0.0106221 - 0.0123098 - 0.013978 - 0.015552 - 0.0169538 - 0.0181087 - 0.0189516 - 0.019433 - 0.0195241 - 0.0192195 - 0.0185373 - 0.0175182 - 0.0162207 - 0.0147159 - 0.013081 - 0.0113928 - 0.00972205 - 0.00812871 - 0.00665921 - 0.00534516 - 0.00420373 - 0.00323926 - 0.00244565 - 0.00180917 - 0.0013113 - 0.000931238 - 0.000647972 - 0.000441762 - 0.000295092 - 0.000193136 - 0.000123853 - 7.78193e-05 - 4.79075e-05 - 2.88973e-05 - 1.70784e-05 - 9.88952e-06 - 5.61099e-06 - 3.11918e-06 - 1.69894e-06 - 9.06678e-07 - 4.74094e-07 - 2.42892e-07 - 1.21926e-07 - 5.99681e-08 - 2.88987e-08 - 1.3645e-08 - 6.31259e-09 - 2.86139e-09 - 1.27082e-09 - 5.53001e-10 - 2.3578e-10 - 9.8497e-11 - 4.03159e-11 - 1.61684e-11 - 6.35323e-12 - 2.44601e-12 - 9.22699e-13 - 3.41034e-13 - 1.23502e-13 - 4.38212e-14 - 1.52347e-14 - 5.18941e-15 - 1.73197e-15 - 5.66369e-16 - 1.81466e-16 - 5.69676e-17 - 1.75226e-17 - 5.28085e-18 - 1.55936e-18 - 4.51156e-19 - 1.27892e-19 - 3.5522e-20 - 9.66693e-21 - 2.5776e-21 - 6.73413e-22 - 1.72385e-22 - 4.3267e-23 - 1.07691e-23 - 3.17029e-24 - 3.13795e-24 - 1.04192e-23 - 4.08832e-23 - 1.59072e-22 - 6.06845e-22 - 2.26838e-21 - 8.3079e-21 - 2.98129e-20 - 1.04822e-19 - 3.6111e-19 - 1.21889e-18 - 4.03109e-18 - 1.30623e-17 - 4.14718e-17 - 1.2901e-16 - 3.93216e-16 - 1.17429e-15 - 3.43603e-15 - 9.85087e-15 - 2.76713e-14 - 7.6159e-14 - 2.05376e-13 - 5.42644e-13 - 1.40481e-12 - 3.56333e-12 - 8.85588e-12 - 2.15648e-11 - 5.1451e-11 - 1.20276e-10 - 2.75489e-10 - 6.1825e-10 - 1.35944e-09 - 2.92883e-09 - 6.18252e-09 - 1.27871e-08 - 2.5913e-08 - 5.14516e-08 - 1.00096e-07 - 1.90797e-07 - 3.56339e-07 - 6.52066e-07 - 1.16912e-06 - 2.05381e-06 - 3.53508e-06 - 5.96177e-06 - 9.85117e-06 - 1.59492e-05 - 2.53002e-05 - 3.93232e-05 - 5.98839e-05 - 8.93528e-05 - 0.00013063 - 0.000187119 - 0.000262622 - 0.000361147 - 0.000486606 - 0.000642415 - 0.000830998 - 0.00105327 - 0.00130809 - 0.00159189 - 0.00189838 - 0.00221861 - 0.00254131 - 0.00285355 - 0.00314183 - 0.00339333 - 0.00359738 - 0.00374691 - 0.00383975 - 0.00387957 - 0.00387642 - 0.00384664 - 0.00381224 - 0.0037996 - 0.00383766 - 0.00395569 - 0.00418069 - 0.0045347 - 0.00503217 - 0.00567767 - 0.00646409 - 0.00737179 - 0.00836862 - 0.00941129 - 0.0104478 - 0.011421 - 0.012273 - 0.0129503 - 0.013408 - 0.0136139 - 0.013552 - 0.0132229 - 0.0126445 - 0.011849 - 0.0108804 - 0.00978988 - 0.00863109 - 0.00745595 - 0.00631082 - 0.00523372 - 0.00425281 - 0.00338595 - 0.00264134 - 0.00201887 - 0.00151194 - 0.00110944 - 0.000797693 - 0.000562024 - 0.000388087 - 0.000262733 - 0.000174531 - 0.00011398 - 7.34973e-05 - 4.72468e-05 - 3.0892e-05 - 2.1316e-05 - 1.63489e-05 - 1.45246e-05 - 1.48771e-05 - 1.6778e-05 - 1.98118e-05 - 2.36856e-05 - 2.81654e-05 - 3.30354e-05 - 3.80753e-05 - 4.30518e-05 - 4.77214e-05 - 5.18411e-05 - 5.51843e-05 - 5.75589e-05 - 5.88241e-05 - 5.89031e-05 - 5.77908e-05 - 5.55543e-05 - 5.23255e-05 - 4.82887e-05 - 4.36632e-05 - 3.86831e-05 - 3.35788e-05 - 2.85591e-05 - 2.37992e-05 - 1.94319e-05 - 1.55456e-05 - 1.21852e-05 - 9.35833e-06 - 7.04206e-06 - 5.19204e-06 - 3.7507e-06 - 2.65476e-06 - 1.84108e-06 - 1.251e-06 - 8.32877e-07 - 5.433e-07 - 3.47245e-07 - 2.17455e-07 - 1.33426e-07 - 8.02134e-08 - 4.72494e-08 - 2.72712e-08 - 1.54258e-08 - 8.55829e-09 - 4.67484e-09 - 2.55734e-09 - 1.50298e-09 - 1.16839e-09 - 1.50985e-09 - 2.83243e-09 - 5.99601e-09 - 1.28846e-08 - 2.73441e-08 - 5.6957e-08 - 1.16287e-07 - 2.32641e-07 - 4.56022e-07 - 8.75839e-07 - 1.64816e-06 - 3.03887e-06 - 5.48987e-06 - 9.71737e-06 - 1.68528e-05 - 2.86373e-05 - 4.76792e-05 - 7.77791e-05 - 0.000124318 - 0.000194689 - 0.000298735 - 0.000449125 - 0.000661584 - 0.00095486 - 0.0013503 - 0.00187094 - 0.00253995 - 0.00337853 - 0.00440319 - 0.00562268 - 0.00703487 - 0.00862394 - 0.0103584 - 0.0121903 - 0.0140565 - 0.0158808 - 0.0175795 - 0.0190668 - 0.0202621 - 0.0210974 - 0.0215234 - 0.0215144 - 0.021071 - 0.0202198 - 0.019011 - 0.0175134 - 0.0158078 - 0.0139802 - 0.012114 - 0.0102849 - 0.00855561 - 0.00697329 - 0.00556879 - 0.00435734 - 0.00334055 - 0.0025093 - 0.00184682 - 0.00133178 - 0.00094097 - 0.000651414 - 0.000441851 - 0.00029365 - 0.000191215 - 0.000121998 - 7.62635e-05 - 4.6711e-05 - 2.80323e-05 - 1.6483e-05 - 9.49616e-06 - 5.36041e-06 - 2.96473e-06 - 1.6066e-06 - 8.53039e-07 - 4.43778e-07 - 2.26204e-07 - 1.12972e-07 - 5.52814e-08 - 2.65047e-08 - 1.2451e-08 - 5.73091e-09 - 2.58451e-09 - 1.14202e-09 - 4.94439e-10 - 2.09776e-10 - 8.72944e-11 - 3.58409e-11 - 1.50894e-11 - 7.98683e-12 - 8.47379e-12 - 1.69387e-11 - 4.04834e-11 - 9.81476e-11 - 2.34372e-10 - 5.48791e-10 - 1.2592e-09 - 2.83093e-09 - 6.23592e-09 - 1.34589e-08 - 2.84612e-08 - 5.89705e-08 - 1.19716e-07 - 2.38127e-07 - 4.64089e-07 - 8.86197e-07 - 1.65804e-06 - 3.03948e-06 - 5.45932e-06 - 9.6076e-06 - 1.65664e-05 - 2.79884e-05 - 4.63304e-05 - 7.51432e-05 - 0.000119413 - 0.000185929 - 0.00028365 - 0.000423988 - 0.000620957 - 0.00089106 - 0.00125282 - 0.00172587 - 0.0023295 - 0.00308074 - 0.00399194 - 0.00506816 - 0.00630454 - 0.0076841 - 0.00917634 - 0.010737 - 0.0123093 - 0.0138268 - 0.0152175 - 0.0164099 - 0.0173382 - 0.0179489 - 0.0182058 - 0.0180933 - 0.0176182 - 0.0168091 - 0.0157131 - 0.0143919 - 0.0129155 - 0.0113564 - 0.00978378 - 0.00825866 - 0.00683045 - 0.00553511 - 0.0043948 - 0.00341893 - 0.00260602 - 0.00194626 - 0.00142417 - 0.00102108 - 0.00071729 - 0.000493703 - 0.000332947 - 0.000219998 - 0.00014243 - 9.03484e-05 - 5.61534e-05 - 3.41955e-05 - 2.04032e-05 - 1.19279e-05 - 6.83231e-06 - 3.83449e-06 - 2.10855e-06 - 1.13605e-06 - 5.9972e-07 - 3.10196e-07 - 1.57203e-07 - 7.80587e-08 - 3.79768e-08 - 1.81031e-08 - 8.45523e-09 - 3.86931e-09 - 1.73492e-09 - 7.62186e-10 - 3.2808e-10 - 1.38368e-10 - 5.71779e-11 - 2.31504e-11 - 9.18384e-12 - 3.56967e-12 - 1.35946e-12 - 5.07276e-13 - 1.85463e-13 - 6.64368e-14 - 2.33183e-14 - 8.01901e-15 - 2.70198e-15 - 8.9203e-16 - 2.88546e-16 - 9.14505e-17 - 2.83985e-17 - 8.64052e-18 - 2.57586e-18 - 7.52387e-19 - 2.15326e-19 - 6.03798e-20 - 1.65898e-20 - 4.46888e-21 - 1.1907e-21 - 3.54724e-22 - 2.70303e-22 - 7.45922e-22 - 2.68317e-21 - 9.67303e-21 - 3.42192e-20 - 1.1862e-19 - 4.02889e-19 - 1.34076e-18 - 4.37171e-18 - 1.39666e-17 - 4.37185e-17 - 1.34084e-16 - 4.02927e-16 - 1.18635e-15 - 3.42243e-15 - 9.67374e-15 - 2.67911e-14 - 7.26984e-14 - 1.93284e-13 - 5.03504e-13 - 1.28513e-12 - 3.21388e-12 - 7.87496e-12 - 1.89062e-11 - 4.44733e-11 - 1.02502e-10 - 2.31473e-10 - 5.12165e-10 - 1.11034e-09 - 2.35855e-09 - 4.90879e-09 - 1.00102e-08 - 2.00013e-08 - 3.91579e-08 - 7.51153e-08 - 1.41185e-07 - 2.60021e-07 - 4.69236e-07 - 8.29749e-07 - 1.43776e-06 - 2.44132e-06 - 4.06237e-06 - 6.62491e-06 - 1.0589e-05 - 1.65903e-05 - 2.54819e-05 - 3.83764e-05 - 5.66822e-05 - 8.21314e-05 - 0.000116792 - 0.00016307 - 0.000223695 - 0.000301717 - 0.000400522 - 0.000523905 - 0.000676234 - 0.000862732 - 0.00108992 - 0.00136617 - 0.00170242 - 0.00211279 - 0.00261515 - 0.00323128 - 0.00398644 - 0.0049082 - 0.00602429 - 0.00735944 - 0.00893147 - 0.0107468 - 0.0127958 - 0.0150493 - 0.0174561 - 0.0199423 - 0.0224143 - 0.0247631 - 0.0268725 - 0.0286279 - 0.0299271 - 0.0306897 - 0.0308656 - 0.0304406 - 0.0294379 - 0.027917 - 0.0259682 - 0.0237049 - 0.0212543 - 0.0187473 - 0.0163094 - 0.0140532 - 0.012073 - 0.0104412 - 0.00920708 - 0.00839714 - 0.00801617 - 0.00804901 - 0.00846236 - 0.00920671 - 0.0102183 - 0.0114214 - 0.0127314 - 0.0140583 - 0.0153113 - 0.0164042 - 0.0172608 - 0.0178203 - 0.0180418 - 0.0179068 - 0.017421 - 0.0166126 - 0.0155294 - 0.0142346 - 0.0128009 - 0.0113044 - 0.00981974 - 0.00841484 - 0.00714807 - 0.00606625 - 0.00520401 - 0.00458421 - 0.00421879 - 0.00410993 - 0.00425091 - 0.00462675 - 0.00521439 - 0.00598278 - 0.0068929 - 0.00789827 - 0.00894605 - 0.00997897 - 0.0109382 - 0.0117671 - 0.0124148 - 0.0128403 - 0.0130161 - 0.01293 - 0.0125862 - 0.0120047 - 0.011219 - 0.0102732 - 0.00921717 - 0.00810268 - 0.00697907 - 0.00588986 - 0.00487025 - 0.00394581 - 0.00313231 - 0.00243638 - 0.00185693 - 0.00138696 - 0.00101542 - 0.000729103 - 0.000514093 - 0.000357004 - 0.000245774 - 0.00017014 - 0.00012182 - 9.44836e-05 - 8.35732e-05 - 8.60424e-05 - 0.000100046 - 0.000124617 - 0.000159347 - 0.000204081 - 0.000258632 - 0.000322534 - 0.000394841 - 0.000473977 - 0.000557675 - 0.000642995 - 0.000726438 - 0.000804153 - 0.00087221 - 0.00092692 - 0.000965165 - 0.000984686 - 0.000984308 - 0.000964054 - 0.000925143 - 0.000869867 - 0.000801371 - 0.000723355 - 0.000639744 - 0.000554367 - 0.00047068 - 0.000391553 - 0.000319149 - 0.000254878 - 0.000199438 - 0.000152904 - 0.00011486 - 8.45387e-05 - 6.09647e-05 - 4.30763e-05 - 2.98219e-05 - 2.02287e-05 - 1.34443e-05 - 8.75481e-06 - 5.58588e-06 - 3.49199e-06 - 2.13892e-06 - 1.2837e-06 - 7.54953e-07 - 4.3522e-07 - 2.46288e-07 - 1.37595e-07 - 7.76206e-08 - 4.7872e-08 - 3.91206e-08 - 4.96818e-08 - 8.57782e-08 - 1.64423e-07 - 3.19802e-07 - 6.1486e-07 - 1.16078e-06 - 2.1483e-06 - 3.8961e-06 - 6.92335e-06 - 1.20543e-05 - 2.0564e-05 - 3.43723e-05 - 5.6292e-05 - 9.03278e-05 - 0.000142015 - 0.000218767 - 0.000330193 - 0.000488304 - 0.000707537 - 0.00100449 - 0.00139726 - 0.00190436 - 0.00254305 - 0.00332735 - 0.00426559 - 0.00535792 - 0.00659401 - 0.00795135 - 0.00939439 - 0.0108751 - 0.0123349 - 0.013708 - 0.0149262 - 0.0159243 - 0.016646 - 0.0170489 - 0.0171089 - 0.0168223 - 0.0162065 - 0.0152982 - 0.0141497 - 0.0128242 - 0.01139 - 0.00991533 - 0.00846323 - 0.00708799 - 0.00583302 - 0.00473027 - 0.00380108 - 0.00305813 - 0.00250808 - 0.00215444 - 0.00200017 - 0.00204972 - 0.00231011 - 0.00279091 - 0.003503 - 0.00445631 - 0.00565652 - 0.00710137 - 0.0087768 - 0.0106537 - 0.0126858 - 0.014809 - 0.0169432 - 0.0189962 - 0.0208692 - 0.0224645 - 0.0236937 - 0.0244855 - 0.0247927 - 0.0245967 - 0.0239093 - 0.0227716 - 0.0212499 - 0.0194294 - 0.0174059 - 0.0152782 - 0.0131396 - 0.0110721 - 0.00914149 - 0.00739501 - 0.00586135 - 0.00455191 - 0.00346359 - 0.00258224 - 0.00188626 - 0.00135004 - 0.000946729 - 0.000650493 - 0.000437923 - 0.000288862 - 0.000186692 - 0.000118227 - 7.33688e-05 - 4.46338e-05 - 2.66504e-05 - 1.56821e-05 - 9.21728e-06 - 5.63909e-06 - 3.97946e-06 - 3.75189e-06 - 4.85588e-06 - 7.54564e-06 - 1.24576e-05 - 2.06929e-05 - 3.39515e-05 - 5.47126e-05 - 8.64515e-05 - 0.000133876 - 0.000203149 - 0.000302064 - 0.000440104 - 0.00062834 - 0.000879078 - 0.00120523 - 0.00161937 - 0.00213248 - 0.00275244 - 0.00348253 - 0.00431992 - 0.00525458 - 0.00626873 - 0.00733715 - 0.00842831 - 0.00950652 - 0.0105347 - 0.0114778 - 0.0123057 - 0.0129966 - 0.0135384 - 0.0139296 - 0.0141784 - 0.0143005 - 0.0143162 - 0.0142468 - 0.0141109 - 0.0139217 - 0.0136857 - 0.0134019 - 0.0130632 - 0.012659 - 0.0121776 - 0.0116098 - 0.0109513 - 0.0102046 - 0.00937978 - 0.00849391 - 0.00756962 - 0.00663291 - 0.00571057 - 0.00482772 - 0.00400577 - 0.00326099 - 0.00260377 - 0.00203868 - 0.00156499 - 0.00117773 - 0.000868844 - 0.000628451 - 0.000445965 - 0.00031104 - 0.00021428 - 0.000147749 - 0.000105306 - 8.28571e-05 - 7.85795e-05 - 9.31949e-05 - 0.000130355 - 0.000197161 - 0.00030483 - 0.000469447 - 0.000712728 - 0.00106262 - 0.00155357 - 0.00222613 - 0.00312575 - 0.00430043 - 0.00579713 - 0.0076569 - 0.00990902 - 0.0125645 - 0.0156098 - 0.0190014 - 0.0226627 - 0.0264835 - 0.0303231 - 0.0340181 - 0.0373924 - 0.0402711 - 0.0424952 - 0.0439363 - 0.0445086 - 0.0441776 - 0.0429631 - 0.040938 - 0.0382204 - 0.0349623 - 0.0313358 - 0.0275182 - 0.0236774 - 0.0199612 - 0.0164883 - 0.0133445 - 0.0105819 - 0.00822174 - 0.00625893 - 0.00466846 - 0.0034118 - 0.00244304 - 0.00171401 - 0.00117824 - 0.000793582 - 0.000523705 - 0.000338624 - 0.000214529 - 0.000133165 - 8.09901e-05 - 4.82626e-05 - 2.8179e-05 - 1.61205e-05 - 9.03581e-06 - 4.96241e-06 - 2.67027e-06 - 1.40785e-06 - 7.27263e-07 - 3.68099e-07 - 1.82547e-07 - 8.87001e-08 - 4.22295e-08 - 1.97008e-08 - 9.00956e-09 - 4.04833e-09 - 1.81029e-09 - 8.60698e-10 - 5.58336e-10 - 6.85971e-10 - 1.31585e-09 - 2.8408e-09 - 6.1788e-09 - 1.32343e-08 - 2.77987e-08 - 5.7221e-08 - 1.15408e-07 - 2.28062e-07 - 4.4158e-07 - 8.37727e-07 - 1.55716e-06 - 2.83595e-06 - 5.0606e-06 - 8.84794e-06 - 1.51572e-05 - 2.5441e-05 - 4.18393e-05 - 6.74174e-05 - 0.000106438 - 0.000164648 - 0.000249549 - 0.000370588 - 0.000539216 - 0.000768726 - 0.00107378 - 0.0014696 - 0.00197069 - 0.00258926 - 0.00333325 - 0.00420434 - 0.00519594 - 0.0062917 - 0.00746463 - 0.00867732 - 0.00988325 - 0.0110294 - 0.0120597 - 0.01292 - 0.013562 - 0.0139483 - 0.0140559 - 0.0138781 - 0.0134257 - 0.0127258 - 0.0118186 - 0.0107544 - 0.00958832 - 0.00837598 - 0.00716911 - 0.00601218 - 0.0049401 - 0.00397719 - 0.00313728 - 0.00242475 - 0.00183619 - 0.00136241 - 0.000990447 - 0.000705493 - 0.00049237 - 0.000336687 - 0.000225579 - 0.000148084 - 9.52474e-05 - 6.00255e-05 - 3.70643e-05 - 2.24239e-05 - 1.32924e-05 - 7.72031e-06 - 4.39341e-06 - 2.44966e-06 - 1.33828e-06 - 7.16347e-07 - 3.75697e-07 - 1.93059e-07 - 9.7204e-08 - 4.79558e-08 - 2.31886e-08 - 1.10053e-08 - 5.16537e-09 - 2.49231e-09 - 1.45573e-09 - 1.44813e-09 - 2.49265e-09 - 5.27515e-09 - 1.15271e-08 - 2.49327e-08 - 5.29399e-08 - 1.10176e-07 - 2.24677e-07 - 4.48922e-07 - 8.78862e-07 - 1.68581e-06 - 3.16834e-06 - 5.83436e-06 - 1.05267e-05 - 1.86091e-05 - 3.22327e-05 - 5.47023e-05 - 9.09601e-05 - 0.000148195 - 0.000236566 - 0.000370005 - 0.000567022 - 0.000851392 - 0.00125255 - 0.00180551 - 0.00254999 - 0.00352871 - 0.00478442 - 0.00635594 - 0.00827308 - 0.010551 - 0.0131842 - 0.0161418 - 0.0193636 - 0.0227592 - 0.02621 - 0.0295742 - 0.0326961 - 0.0354175 - 0.0375905 - 0.0390913 - 0.0398314 - 0.0397669 - 0.0389028 - 0.0372926 - 0.0350338 - 0.0322585 - 0.0291224 - 0.0257915 - 0.02243 - 0.0191902 - 0.0162049 - 0.0135829 - 0.0114083 - 0.00974183 - 0.00862375 - 0.00807727 - 0.00811156 - 0.00872333 - 0.00989679 - 0.011602 - 0.0137919 - 0.0163994 - 0.0193348 - 0.0224847 - 0.0257144 - 0.0288725 - 0.0317988 - 0.0343349 - 0.0363363 - 0.0376843 - 0.0382965 - 0.0381344 - 0.037207 - 0.0355694 - 0.0333172 - 0.0305773 - 0.0274959 - 0.0242255 - 0.020913 - 0.0176887 - 0.0146593 - 0.0119033 - 0.00947019 - 0.00738221 - 0.00563834 - 0.00421942 - 0.0030938 - 0.00222264 - 0.00156455 - 0.0010791 - 0.000729329 - 0.000483146 - 0.000313949 - 0.000200581 - 0.000126914 - 8.12404e-05 - 5.5656e-05 - 4.55669e-05 - 4.94212e-05 - 6.87538e-05 - 0.000108612 - 0.000178427 - 0.000293379 - 0.000476297 - 0.000760061 - 0.00119043 - 0.00182912 - 0.00275668 - 0.00407487 - 0.00590754 - 0.00839955 - 0.0117125 - 0.0160171 - 0.0214803 - 0.0282497 - 0.0364326 - 0.0460744 - 0.0571362 - 0.0694755 - 0.0828344 - 0.0968356 - 0.110993 - 0.124731 - 0.137426 - 0.148443 - 0.157195 - 0.163191 - 0.16608 - 0.16569 - 0.162039 - 0.155338 - 0.145969 - 0.13445 - 0.121386 - 0.107417 - 0.0931679 - 0.0792032 - 0.0659922 - 0.0538902 - 0.0431307 - 0.0338313 - 0.0260076 - 0.0195944 - 0.0144686 - 0.0104718 - 0.00743065 - 0.00517275 - 0.00353869 - 0.0023892 - 0.00160904 - 0.00110793 - 0.000819713 - 0.000700487 - 0.000726235 - 0.000890471 - 0.00120196 - 0.00168246 - 0.00236436 - 0.00328777 - 0.00449696 - 0.00603582 - 0.00794235 - 0.0102424 - 0.0129428 - 0.0160253 - 0.0194414 - 0.0231094 - 0.0269146 - 0.0307131 - 0.0343397 - 0.0376189 - 0.0403787 - 0.0424655 - 0.0437579 - 0.0441788 - 0.0437027 - 0.0423583 - 0.040226 - 0.0374293 - 0.0341234 - 0.0304811 - 0.0266776 - 0.022877 - 0.0192215 - 0.0158239 - 0.0127637 - 0.0100873 - 0.00781109 - 0.00592634 - 0.00440555 - 0.00320891 - 0.0022902 - 0.00160171 - 0.00109801 - 0.000738346 - 0.000488074 - 0.000319114 - 0.000209886 - 0.000144899 - 0.000114287 - 0.00011345 - 0.000142967 - 0.000208865 - 0.000323272 - 0.000505441 - 0.000783052 - 0.00119362 - 0.00178573 - 0.00261982 - 0.00376788 - 0.00531191 - 0.00734032 - 0.0099422 - 0.0131993 - 0.0171757 - 0.0219067 - 0.0273861 - 0.0335566 - 0.0403008 - 0.0474392 - 0.0547326 - 0.0618925 - 0.0685978 - 0.074518 - 0.0793391 - 0.0827917 - 0.0846753 - 0.084878 - 0.0833871 - 0.0802907 - 0.0757691 - 0.0700771 - 0.0635207 - 0.0564296 - 0.0491301 - 0.0419213 - 0.0350564 - 0.0287303 - 0.0230756 - 0.0181636 - 0.0140114 - 0.0105924 - 0.00784763 - 0.00569779 - 0.00405413 - 0.0028269 - 0.00193171 - 0.00129357 - 0.000848892 - 0.000545916 - 0.00034404 - 0.000212471 - 0.000128587 - 7.62598e-05 - 4.43196e-05 - 2.52403e-05 - 1.40861e-05 - 7.70341e-06 - 4.12826e-06 - 2.16792e-06 - 1.1156e-06 - 5.62553e-07 - 2.77973e-07 - 1.34595e-07 - 6.38609e-08 - 2.96909e-08 - 1.35267e-08 - 6.03865e-09 - 2.64158e-09 - 1.13231e-09 - 4.75596e-10 - 1.95742e-10 - 7.89412e-11 - 3.11955e-11 - 1.20796e-11 - 4.58331e-12 - 1.70407e-12 - 6.20988e-13 - 2.22293e-13 - 7.9713e-14 - 3.33658e-14 - 2.94752e-14 - 6.38018e-14 - 1.77569e-13 - 5.02538e-13 - 1.39959e-12 - 3.82147e-12 - 1.02254e-11 - 2.6812e-11 - 6.88931e-11 - 1.73468e-10 - 4.28014e-10 - 1.03489e-09 - 2.45205e-09 - 5.69327e-09 - 1.29536e-08 - 2.88814e-08 - 6.31019e-08 - 1.35103e-07 - 2.83456e-07 - 5.82778e-07 - 1.17414e-06 - 2.3181e-06 - 4.48481e-06 - 8.50263e-06 - 1.57965e-05 - 2.87584e-05 - 5.13057e-05 - 8.96943e-05 - 0.00015366 - 0.000257961 - 0.00042437 - 0.000684119 - 0.00108072 - 0.001673 - 0.00253788 - 0.00377261 - 0.00549553 - 0.00784462 - 0.0109731 - 0.0150412 - 0.0202036 - 0.0265931 - 0.0343008 - 0.0433543 - 0.0536975 - 0.0651733 - 0.0775138 - 0.0903404 - 0.103176 - 0.115472 - 0.12664 - 0.136103 - 0.143344 - 0.147949 - 0.149655 - 0.14837 - 0.144186 - 0.137373 - 0.128353 - 0.117658 - 0.105889 - 0.093658 - 0.0815473 - 0.0700646 - 0.0596157 - 0.0504871 - 0.0428422 - 0.0367268 - 0.0320839 - 0.0287735 - 0.0265961 - 0.0253153 - 0.0246804 - 0.024446 - 0.0243882 - 0.0243168 - 0.0240838 - 0.0235875 - 0.0227725 - 0.021627 - 0.0201755 - 0.0184715 - 0.0165872 - 0.014604 - 0.0126035 - 0.0106603 - 0.00883602 - 0.00717679 - 0.00571177 - 0.00445418 - 0.00340341 - 0.00254803 - 0.00186912 - 0.00134342 - 0.000946069 - 0.000652789 - 0.000441327 - 0.000292338 - 0.000189735 - 0.000120655 - 7.51766e-05 - 4.5894e-05 - 2.74515e-05 - 1.60884e-05 - 9.2384e-06 - 5.19778e-06 - 2.86533e-06 - 1.54764e-06 - 8.19038e-07 - 4.24702e-07 - 2.15801e-07 - 1.07503e-07 - 5.26328e-08 - 2.56405e-08 - 1.31726e-08 - 8.77456e-09 - 1.02078e-08 - 1.85691e-08 - 3.89947e-08 - 8.33138e-08 - 1.75745e-07 - 3.63781e-07 - 7.38012e-07 - 1.46706e-06 - 2.85743e-06 - 5.45305e-06 - 1.01963e-05 - 1.86801e-05 - 3.35317e-05 - 5.89749e-05 - 0.000101629 - 0.000171594 - 0.000283873 - 0.000460134 - 0.00073077 - 0.00113714 - 0.00173374 - 0.00258995 - 0.00379084 - 0.00543645 - 0.00763894 - 0.0105169 - 0.0141866 - 0.0187503 - 0.0242813 - 0.0308088 - 0.0383013 - 0.0466542 - 0.0556807 - 0.0651114 - 0.0746018 - 0.0837495 - 0.0921213 - 0.0992861 - 0.104852 - 0.108503 - 0.110031 - 0.109357 - 0.106543 - 0.101791 - 0.0954271 - 0.087877 - 0.0796374 - 0.0712426 - 0.0632336 - 0.0561318 - 0.0504187 - 0.0465218 - 0.044805 - 0.045561 - 0.0490042 - 0.0552601 - 0.0643534 - 0.0761915 - 0.0905495 - 0.107057 - 0.125194 - 0.144296 - 0.163577 - 0.182163 - 0.199146 - 0.213648 - 0.224879 - 0.232211 - 0.235225 - 0.233747 - 0.227864 - 0.217912 - 0.204443 - 0.188177 - 0.169934 - 0.150572 - 0.130918 - 0.111717 - 0.0935927 - 0.0770224 - 0.0623326 - 0.0497065 - 0.0392039 - 0.0307866 - 0.0243474 - 0.0197367 - 0.0167861 - 0.0153248 - 0.0151908 - 0.0162352 - 0.0183228 - 0.0213286 - 0.0251345 - 0.0296256 - 0.0346886 - 0.0402135 - 0.0460961 - 0.052243 - 0.0585756 - 0.0650326 - 0.0715684 - 0.0781472 - 0.0847335 - 0.091278 - 0.0977038 - 0.103894 - 0.109684 - 0.114865 - 0.119189 - 0.122391 - 0.124217 - 0.124448 - 0.122933 - 0.119609 - 0.114518 - 0.107811 - 0.0997331 - 0.0906066 - 0.0808031 - 0.0707102 - 0.0607003 - 0.0511033 - 0.0421863 - 0.0341421 - 0.0270865 - 0.0210628 - 0.0160525 - 0.0119898 - 0.00877597 - 0.00629478 - 0.00442438 - 0.00304721 - 0.00205646 - 0.00135988 - 0.000881132 - 0.000559434 - 0.000348064 - 0.000212277 - 0.000127044 - 7.49014e-05 - 4.40862e-05 - 2.70444e-05 - 1.93388e-05 - 1.89574e-05 - 2.60199e-05 - 4.28955e-05 - 7.47722e-05 - 0.00013075 - 0.000225561 - 0.000382019 - 0.000634315 - 0.00103218 - 0.00164583 - 0.00257148 - 0.00393681 - 0.00590566 - 0.00868071 - 0.0125027 - 0.0176447 - 0.0243998 - 0.0330614 - 0.0438952 - 0.0571052 - 0.0727939 - 0.0909236 - 0.111281 - 0.133452 - 0.156816 - 0.180558 - 0.203706 - 0.225193 - 0.24393 - 0.258902 - 0.269258 - 0.274385 - 0.273977 - 0.268058 - 0.256982 - 0.241401 - 0.222195 - 0.200397 - 0.177096 - 0.153352 - 0.130119 - 0.108184 - 0.088143 - 0.0703811 - 0.0550906 - 0.0422952 - 0.0318886 - 0.0236757 - 0.0174137 - 0.0128489 - 0.00974575 - 0.00790899 - 0.00719695 - 0.00752776 - 0.00887847 - 0.0112782 - 0.0147954 - 0.0195211 - 0.0255467 - 0.03294 - 0.0417194 - 0.0518294 - 0.0631202 - 0.0753344 - 0.0881046 - 0.100962 - 0.113362 - 0.124714 - 0.134431 - 0.141978 - 0.146921 - 0.148964 - 0.147984 - 0.144041 - 0.137371 - 0.128363 - 0.117523 - 0.105425 - 0.0926619 - 0.0797996 - 0.0673356 - 0.0556729 - 0.0451043 - 0.0358108 - 0.0278695 - 0.0212705 - 0.0159377 - 0.0117507 - 0.0085666 - 0.0062372 - 0.00462324 - 0.00360352 - 0.00307992 - 0.00297862 - 0.00324833 - 0.00385612 - 0.00478184 - 0.00601151 - 0.00753051 - 0.0093171 - 0.011337 - 0.0135396 - 0.0158561 - 0.0182006 - 0.0204731 - 0.0225658 - 0.0243709 - 0.0257892 - 0.0267388 - 0.0271636 - 0.0270377 - 0.0263689 - 0.0251973 - 0.0235919 - 0.0216433 - 0.019456 - 0.0171389 - 0.014797 - 0.0125243 - 0.0103983 - 0.00847798 - 0.00680284 - 0.00539503 - 0.00426232 - 0.00340205 - 0.00280523 - 0.00246013 - 0.00235501 - 0.00247968 - 0.00282577 - 0.00338583 - 0.00415139 - 0.00511045 - 0.00624471 - 0.00752714 - 0.00892027 - 0.0103758 - 0.0118353 - 0.0132334 - 0.0145009 - 0.0155705 - 0.0163822 - 0.0168885 - 0.017059 - 0.0168832 - 0.0163718 - 0.0155551 - 0.0144807 - 0.0132081 - 0.011804 - 0.0103361 - 0.00886787 - 0.00745451 - 0.00613983 - 0.00495484 - 0.00391778 - 0.0030352 - 0.00230394 - 0.00171353 - 0.00124868 - 0.000891548 - 0.000623701 - 0.000427509 - 0.000287111 - 0.000188926 - 0.000121807 - 7.69461e-05 - 4.76255e-05 - 2.88821e-05 - 1.71615e-05 - 9.99121e-06 - 5.69926e-06 - 3.18533e-06 - 1.74433e-06 - 9.35921e-07 - 4.92024e-07 - 2.53438e-07 - 1.27906e-07 - 6.32485e-08 - 3.06439e-08 - 1.45471e-08 - 6.7662e-09 - 3.08354e-09 - 1.37687e-09 - 6.02381e-10 - 2.58218e-10 - 1.08453e-10 - 4.46303e-11 - 1.79952e-11 - 7.10917e-12 - 2.75181e-12 - 1.04365e-12 - 3.8782e-13 - 1.41202e-13 - 5.03719e-14 - 1.76065e-14 - 6.02967e-15 - 2.02326e-15 - 6.65191e-16 - 2.14278e-16 - 6.76311e-17 - 2.09147e-17 - 6.33715e-18 - 1.88137e-18 - 5.47254e-19 - 1.5597e-19 - 4.35544e-20 - 1.19168e-20 - 3.19464e-21 - 8.39116e-22 - 2.15953e-22 - 5.44544e-23 - 1.34538e-23 - 3.2568e-24 - 7.72459e-25 - 1.79513e-25 - 4.08747e-26 - 9.11906e-27 - 1.99334e-27 - 4.26924e-28 - 8.95892e-29 - 1.84204e-29 - 3.71088e-30 - 7.32476e-31 - 1.4166e-31 - 2.68433e-32 - 4.98382e-33 - 9.06622e-34 - 1.61594e-34 - 2.82204e-35 - 4.82878e-36 - 8.09559e-37 - 1.32983e-37 - 2.14033e-38 - 3.37522e-39 - 5.21507e-40 - 7.89506e-41 - 1.17108e-41 - 1.70199e-42 - 2.4236e-43 - 3.38146e-44 - 4.62257e-45 - 6.19155e-46 - 8.12554e-47 - 1.04482e-47 - 1.31634e-48 - 1.62492e-49 - 1.96532e-50 - 2.329e-51 - 2.70423e-52 - 3.07649e-53 - 3.42928e-54 - 3.74531e-55 - 4.00783e-56 - 4.20211e-57 - 4.31681e-58 - 4.34505e-59 - 4.28512e-60 - 4.14066e-61 - 3.92023e-62 - 3.63656e-63 - 3.30527e-64 - 2.94348e-65 - 2.56833e-66 - 2.19572e-67 - 1.83925e-68 - 1.50953e-69 - 1.21389e-70 - 9.56429e-72 - 7.38353e-73 - 5.58485e-74 - 4.13901e-75 - 3.00551e-76 - 2.13834e-77 - 1.49064e-78 - 1.01813e-79 - 6.81354e-81 - 4.46764e-82 - 2.87026e-83 - 1.80676e-84 - 1.11434e-85 - 6.73396e-87 - 3.98713e-88 - 2.31306e-89 - 1.31477e-90 - 7.32237e-92 - 3.99566e-93 - 2.1363e-94 - 1.11911e-95 - 5.7441e-97 - 2.88873e-98 - 1.42341e-99 - 6.87207e-101 - 3.25074e-102 - 1.50666e-103 - 6.84201e-105 - 3.04432e-106 - 1.32719e-107 - 5.66909e-109 - 2.37623e-110 - 1.87428e-111 - 2.20949e-110 - 5.28801e-109 - 1.24221e-107 - 2.85914e-106 - 6.44781e-105 - 1.42471e-103 - 3.08445e-102 - 6.54282e-101 - 1.35985e-99 - 2.76918e-98 - 5.52523e-97 - 1.08015e-95 - 2.06899e-94 - 3.88299e-93 - 7.14023e-92 - 1.28646e-90 - 2.27099e-89 - 3.92799e-88 - 6.65678e-87 - 1.10534e-85 - 1.79829e-84 - 2.86658e-83 - 4.47719e-82 - 6.85145e-81 - 1.0273e-79 - 1.5092e-78 - 2.17238e-77 - 3.0638e-76 - 4.23372e-75 - 5.7322e-74 - 7.60425e-73 - 9.88391e-72 - 1.25875e-70 - 1.57067e-69 - 1.92029e-68 - 2.30031e-67 - 2.69987e-66 - 3.10482e-65 - 3.49837e-64 - 3.86218e-63 - 4.1777e-62 - 4.42769e-61 - 4.59785e-60 - 4.6781e-59 - 4.66359e-58 - 4.55521e-57 - 4.35947e-56 - 4.08786e-55 - 3.75573e-54 - 3.38088e-53 - 2.98196e-52 - 2.57698e-51 - 2.18201e-50 - 1.81026e-49 - 1.4715e-48 - 1.17197e-47 - 9.14556e-47 - 6.99263e-46 - 5.23851e-45 - 3.84514e-44 - 2.76537e-43 - 1.94863e-42 - 1.34538e-41 - 9.10115e-41 - 6.03233e-40 - 3.91751e-39 - 2.49271e-38 - 1.55407e-37 - 9.49308e-37 - 5.68172e-36 - 3.33188e-35 - 1.91442e-34 - 1.07775e-33 - 5.94484e-33 - 3.2129e-32 - 1.70134e-31 - 8.8272e-31 - 4.48736e-30 - 2.2351e-29 - 1.09078e-28 - 5.21577e-28 - 2.44362e-27 - 1.12173e-26 - 5.0452e-26 - 2.22334e-25 - 9.6e-25 - 4.06138e-24 - 1.6835e-23 - 6.83739e-23 - 2.72085e-22 - 1.06086e-21 - 4.05272e-21 - 1.51696e-20 - 5.56339e-20 - 1.99914e-19 - 7.03857e-19 - 2.42809e-18 - 8.20698e-18 - 2.71795e-17 - 8.81938e-17 - 2.80398e-16 - 8.73476e-16 - 2.66605e-15 - 7.97306e-15 - 2.33627e-14 - 6.70756e-14 - 1.8869e-13 - 5.20086e-13 - 1.40458e-12 - 3.71676e-12 - 9.63669e-12 - 2.44816e-11 - 6.09399e-11 - 1.48633e-10 - 3.55208e-10 - 8.31775e-10 - 1.90848e-09 - 4.29072e-09 - 9.45234e-09 - 2.04041e-08 - 4.31586e-08 - 8.94536e-08 - 1.81682e-07 - 3.61591e-07 - 7.05214e-07 - 1.34782e-06 - 2.52441e-06 - 4.63355e-06 - 8.33507e-06 - 1.46946e-05 - 2.53909e-05 - 4.30015e-05 - 7.13836e-05 - 0.000116157 - 0.000185287 - 0.000289756 - 0.000444259 - 0.000667872 - 0.000984563 - 0.00142341 - 0.00201839 - 0.00280749 - 0.00383118 - 0.00512992 - 0.00674098 - 0.0086945 - 0.0110092 - 0.013688 - 0.0167144 - 0.020049 - 0.0236291 - 0.0273681 - 0.0311587 - 0.0348773 - 0.0383902 - 0.0415613 - 0.0442609 - 0.0463735 - 0.0478061 - 0.0484944 - 0.0484071 - 0.0475482 - 0.0459567 - 0.043704 - 0.0408888 - 0.0376304 - 0.0340605 - 0.0303155 - 0.0265272 - 0.022816 - 0.0192849 - 0.016015 - 0.013064 - - - - - -4.95256e-34 - -6.16334e-34 - -7.51517e-34 - -8.9784e-34 - -1.05098e-33 - -1.2054e-33 - -1.35457e-33 - -1.49145e-33 - -1.60898e-33 - -1.70072e-33 - -1.76137e-33 - -1.78733e-33 - -1.77704e-33 - -1.73111e-33 - -1.65231e-33 - -1.54523e-33 - -1.4159e-33 - -1.27118e-33 - -1.1182e-33 - -9.63761e-34 - -8.13871e-34 - -6.73409e-34 - -5.45933e-34 - -4.33647e-34 - -3.37497e-34 - -2.5736e-34 - -1.92286e-34 - -1.40764e-34 - -1.00965e-34 - -7.09555e-35 - -4.8854e-35 - -3.29331e-35 - -2.16107e-35 - -1.30842e-35 - -3.1827e-36 - 2.61673e-35 - 1.67527e-34 - 8.99892e-34 - 4.66195e-33 - 2.36218e-32 - 1.17249e-31 - 5.70206e-31 - 2.717e-30 - 1.26849e-29 - 5.80254e-29 - 2.60068e-28 - 1.14207e-27 - 4.914e-27 - 2.07164e-26 - 8.55719e-26 - 3.46326e-25 - 1.37333e-24 - 5.33582e-24 - 2.03126e-23 - 7.57645e-23 - 2.76888e-22 - 9.91466e-22 - 3.47848e-21 - 1.19574e-20 - 4.02737e-20 - 1.32906e-19 - 4.29736e-19 - 1.36144e-18 - 4.226e-18 - 1.28528e-17 - 3.83005e-17 - 1.11827e-16 - 3.19909e-16 - 8.96693e-16 - 2.46262e-15 - 6.62654e-15 - 1.74709e-14 - 4.51313e-14 - 1.1423e-13 - 2.8328e-13 - 6.88321e-13 - 1.63871e-12 - 3.82252e-12 - 8.73645e-12 - 1.9564e-11 - 4.29255e-11 - 9.22807e-11 - 1.94376e-10 - 4.01155e-10 - 8.11183e-10 - 1.60717e-09 - 3.1199e-09 - 5.93414e-09 - 1.10589e-08 - 2.0193e-08 - 3.61267e-08 - 6.33275e-08 - 1.08766e-07 - 1.83034e-07 - 3.01791e-07 - 4.87548e-07 - 7.7173e-07 - 1.19688e-06 - 1.81875e-06 - 2.70789e-06 - 3.95027e-06 - 5.64623e-06 - 7.9073e-06 - 1.08501e-05 - 1.45874e-05 - 1.92157e-05 - 2.48013e-05 - 3.13637e-05 - 3.88613e-05 - 4.71786e-05 - 5.61189e-05 - 6.54049e-05 - 7.46875e-05 - 8.35646e-05 - 9.16081e-05 - 9.83971e-05 - 0.000103554 - 0.00010678 - 0.000107882 - 0.000106793 - 0.00010358 - 9.84341e-05 - 9.16541e-05 - 8.36171e-05 - 7.47438e-05 - 6.54624e-05 - 5.61753e-05 - 4.7232e-05 - 3.89102e-05 - 3.14071e-05 - 2.48387e-05 - 1.92471e-05 - 1.46131e-05 - 1.08706e-05 - 7.92321e-06 - 5.6583e-06 - 3.95921e-06 - 2.71436e-06 - 1.82332e-06 - 1.20004e-06 - 7.73865e-07 - 4.88958e-07 - 3.02702e-07 - 1.83609e-07 - 1.09122e-07 - 6.35427e-08 - 3.6254e-08 - 2.02667e-08 - 1.11006e-08 - 5.95729e-09 - 3.13247e-09 - 1.61384e-09 - 8.14654e-10 - 4.02923e-10 - 1.95257e-10 - 9.27106e-11 - 4.31309e-11 - 1.966e-11 - 8.78045e-12 - 3.84226e-12 - 1.64738e-12 - 6.92046e-13 - 2.84844e-13 - 1.14858e-13 - 4.53382e-14 - 1.74196e-14 - 6.23751e-15 - 1.31174e-15 - -2.20244e-15 - -7.77776e-15 - -2.0591e-14 - -5.18126e-14 - -1.27222e-13 - -3.05906e-13 - -7.20636e-13 - -1.66332e-12 - -3.76161e-12 - -8.33502e-12 - -1.80958e-11 - -3.84932e-11 - -8.02285e-11 - -1.63836e-10 - -3.27814e-10 - -6.42661e-10 - -1.23445e-09 - -2.32328e-09 - -4.28417e-09 - -7.7405e-09 - -1.37028e-08 - -2.37675e-08 - -4.03921e-08 - -6.72583e-08 - -1.09732e-07 - -1.7541e-07 - -2.74736e-07 - -4.21611e-07 - -6.33936e-07 - -9.33934e-07 - -1.3481e-06 - -1.90664e-06 - -2.6421e-06 - -3.58729e-06 - -4.77223e-06 - -6.22032e-06 - -7.94404e-06 - -9.94047e-06 - -1.21873e-05 - -1.46402e-05 - -1.72315e-05 - -1.98718e-05 - -2.24536e-05 - -2.48583e-05 - -2.69647e-05 - -2.86586e-05 - -2.98437e-05 - -3.04499e-05 - -3.04409e-05 - -2.9817e-05 - -2.8616e-05 - -2.69085e-05 - -2.47918e-05 - -2.23801e-05 - -1.97949e-05 - -1.71547e-05 - -1.45663e-05 - -1.21185e-05 - -9.87847e-06 - -7.88979e-06 - -6.17417e-06 - -4.734e-06 - -3.55643e-06 - -2.61781e-06 - -1.88798e-06 - -1.33412e-06 - -9.23695e-07 - -6.26613e-07 - -4.16492e-07 - -2.71238e-07 - -1.73074e-07 - -1.08205e-07 - -6.62806e-08 - -3.97758e-08 - -2.33778e-08 - -1.34388e-08 - -7.51457e-09 - -3.99226e-09 - -1.79761e-09 - -1.55833e-10 - 1.62647e-09 - 4.40583e-09 - 9.54109e-09 - 1.94299e-08 - 3.83984e-08 - 7.41803e-08 - 1.40332e-07 - 2.60078e-07 - 4.7225e-07 - 8.40183e-07 - 1.46457e-06 - 2.50142e-06 - 4.18598e-06 - 6.86349e-06 - 1.10263e-05 - 1.73561e-05 - 2.67677e-05 - 4.04488e-05 - 5.98878e-05 - 8.68775e-05 - 0.000123485 - 0.000171971 - 0.000234658 - 0.000313728 - 0.000410966 - 0.000527468 - 0.00066332 - 0.000817311 - 0.000986707 - 0.00116715 - 0.0013527 - 0.00153607 - 0.00170907 - 0.00186314 - 0.00199006 - 0.00208269 - 0.0021356 - 0.00214562 - 0.00211214 - 0.00203718 - 0.00192518 - 0.00178259 - 0.00161722 - 0.00143755 - 0.00125202 - 0.00106841 - 0.000893312 - 0.000731819 - 0.00058741 - 0.000461973 - 0.000355982 - 0.000268767 - 0.00019882 - 0.000144106 - 0.000102339 - 7.12094e-05 - 4.85478e-05 - 3.24294e-05 - 2.12249e-05 - 1.3611e-05 - 8.55202e-06 - 5.26484e-06 - 3.1757e-06 - 1.87685e-06 - 1.08682e-06 - 6.16624e-07 - 3.42785e-07 - 1.86707e-07 - 9.96401e-08 - 5.2101e-08 - 2.66928e-08 - 1.33992e-08 - 6.59024e-09 - 3.17585e-09 - 1.49953e-09 - 6.93727e-10 - 3.14455e-10 - 1.39658e-10 - 6.07726e-11 - 2.59112e-11 - 1.08244e-11 - 4.43055e-12 - 1.77684e-12 - 6.98194e-13 - 2.68807e-13 - 1.01401e-13 - 3.74782e-14 - 1.35723e-14 - 4.81577e-15 - 1.67423e-15 - 5.70295e-16 - 1.90336e-16 - 6.22416e-17 - 1.99424e-17 - 6.2605e-18 - 1.92566e-18 - 5.80343e-19 - 1.71367e-19 - 4.95802e-20 - 1.40548e-20 - 3.90372e-21 - 1.06236e-21 - 2.83268e-22 - 7.4006e-23 - 1.89476e-23 - 4.76861e-24 - 1.24276e-24 - 5.98879e-25 - 1.38172e-24 - 5.3505e-24 - 2.12055e-23 - 8.25555e-23 - 3.14952e-22 - 1.17729e-21 - 4.3118e-21 - 1.54729e-20 - 5.44027e-20 - 1.87416e-19 - 6.32602e-19 - 2.09214e-18 - 6.77934e-18 - 2.15239e-17 - 6.69563e-17 - 2.04079e-16 - 6.09457e-16 - 1.7833e-15 - 5.1126e-15 - 1.43614e-14 - 3.95265e-14 - 1.0659e-13 - 2.81632e-13 - 7.29096e-13 - 1.84937e-12 - 4.59621e-12 - 1.11921e-11 - 2.67031e-11 - 6.24235e-11 - 1.42979e-10 - 3.20872e-10 - 7.05551e-10 - 1.52007e-09 - 3.20873e-09 - 6.63652e-09 - 1.34488e-08 - 2.67034e-08 - 5.19498e-08 - 9.90237e-08 - 1.8494e-07 - 3.38423e-07 - 6.06771e-07 - 1.06593e-06 - 1.8347e-06 - 3.09416e-06 - 5.11276e-06 - 8.27761e-06 - 1.31308e-05 - 2.04087e-05 - 3.10796e-05 - 4.63738e-05 - 6.77965e-05 - 9.71133e-05 - 0.000136297 - 0.000187428 - 0.000252533 - 0.00033338 - 0.000431223 - 0.000546517 - 0.000678654 - 0.00082573 - 0.000984414 - 0.00114994 - 0.00131625 - 0.00147635 - 0.00162275 - 0.00174809 - 0.00184582 - 0.00191084 - 0.00194004 - 0.00193274 - 0.00189081 - 0.00181863 - 0.00172275 - 0.00161135 - 0.00149356 - 0.00137872 - 0.00127559 - 0.00119165 - 0.00113253 - 0.00110163 - 0.00109985 - 0.00112561 - 0.001175 - 0.00124218 - 0.00131983 - 0.00139982 - 0.00147384 - 0.00153416 - 0.00157416 - 0.00158894 - 0.00157562 - 0.00153351 - 0.00146403 - 0.00137047 - 0.00125758 - 0.00113103 - 0.000996861 - 0.000860974 - 0.000728651 - 0.000604241 - 0.000490968 - 0.000390879 - 0.000304913 - 0.000233051 - 0.000174528 - 0.000128061 - 9.20676e-05 - 6.48537e-05 - 4.47611e-05 - 3.02693e-05 - 2.00559e-05 - 1.30202e-05 - 8.28195e-06 - 5.16159e-06 - 3.15189e-06 - 1.8858e-06 - 1.10549e-06 - 6.34971e-07 - 3.57346e-07 - 1.97043e-07 - 1.06456e-07 - 5.63526e-08 - 2.92278e-08 - 1.4853e-08 - 7.39555e-09 - 3.60797e-09 - 1.72462e-09 - 8.07717e-10 - 3.70649e-10 - 1.66649e-10 - 7.34141e-11 - 3.16879e-11 - 1.34012e-11 - 5.55305e-12 - 2.25453e-12 - 8.96844e-13 - 3.49555e-13 - 1.3349e-13 - 4.99484e-14 - 1.83118e-14 - 6.57773e-15 - 2.31504e-15 - 7.98324e-16 - 2.69746e-16 - 8.9342e-17 - 2.9128e-17 - 9.76341e-18 - 4.72724e-18 - 6.95038e-18 - 1.96983e-17 - 6.09995e-17 - 1.87121e-16 - 5.62997e-16 - 1.65984e-15 - 4.79478e-15 - 1.35709e-14 - 3.76342e-14 - 1.02258e-13 - 2.72236e-13 - 7.1012e-13 - 1.81491e-12 - 4.5448e-12 - 1.11509e-11 - 2.68068e-11 - 6.31417e-11 - 1.45721e-10 - 3.29509e-10 - 7.30044e-10 - 1.58477e-09 - 3.37071e-09 - 7.02447e-09 - 1.43431e-08 - 2.86951e-08 - 5.62483e-08 - 1.08031e-07 - 2.03294e-07 - 3.74832e-07 - 6.77153e-07 - 1.1986e-06 - 2.07872e-06 - 3.53229e-06 - 5.88103e-06 - 9.59373e-06 - 1.53341e-05 - 2.40141e-05 - 3.68477e-05 - 5.53976e-05 - 8.16036e-05 - 0.000117778 - 0.000166554 - 0.000230773 - 0.000313293 - 0.000416728 - 0.000543115 - 0.000693534 - 0.000867722 - 0.00106373 - 0.00127766 - 0.00150363 - 0.0017338 - 0.00195883 - 0.00216836 - 0.00235181 - 0.00249925 - 0.00260228 - 0.00265482 - 0.00265371 - 0.00259901 - 0.00249402 - 0.00234493 - 0.0021602 - 0.00194983 - 0.00172439 - 0.00149421 - 0.0012686 - 0.0010553 - 0.000860126 - 0.000686887 - 0.000537459 - 0.000412043 - 0.000309512 - 0.000227797 - 0.000164269 - 0.000116065 - 8.03492e-05 - 5.45004e-05 - 3.62205e-05 - 2.35856e-05 - 1.50479e-05 - 9.40679e-06 - 5.76161e-06 - 3.45767e-06 - 2.0331e-06 - 1.17131e-06 - 6.61184e-07 - 3.65687e-07 - 1.98168e-07 - 1.05219e-07 - 5.47382e-08 - 2.79013e-08 - 1.39346e-08 - 6.81873e-09 - 3.26925e-09 - 1.53578e-09 - 7.06883e-10 - 3.18789e-10 - 1.40863e-10 - 6.09863e-11 - 2.58729e-11 - 1.07614e-11 - 4.4044e-12 - 1.81685e-12 - 8.67714e-13 - 7.40664e-13 - 1.31546e-12 - 3.06677e-12 - 7.40609e-12 - 1.76751e-11 - 4.13834e-11 - 9.49532e-11 - 2.13472e-10 - 4.70233e-10 - 1.01489e-09 - 2.14618e-09 - 4.4468e-09 - 9.02747e-09 - 1.79565e-08 - 3.49956e-08 - 6.68256e-08 - 1.25028e-07 - 2.29198e-07 - 4.11672e-07 - 7.24482e-07 - 1.24923e-06 - 2.11053e-06 - 3.49364e-06 - 5.66633e-06 - 9.00457e-06 - 1.40204e-05 - 2.13892e-05 - 3.19717e-05 - 4.68246e-05 - 6.71923e-05 - 9.44715e-05 - 0.000130143 - 0.000175661 - 0.00023231 - 0.000301021 - 0.000382176 - 0.000475407 - 0.000579436 - 0.000691962 - 0.000809647 - 0.000928209 - 0.00104264 - 0.00114751 - 0.00123742 - 0.00130742 - 0.00135347 - 0.00137284 - 0.00136436 - 0.00132854 - 0.00126753 - 0.00118488 - 0.00108525 - 0.00097392 - 0.000856353 - 0.000737767 - 0.000622762 - 0.000515065 - 0.000417387 - 0.0003314 - 0.000257812 - 0.000196512 - 0.000146762 - 0.000107393 - 7.69968e-05 - 5.40888e-05 - 3.72288e-05 - 2.51065e-05 - 1.65895e-05 - 1.07402e-05 - 6.81292e-06 - 4.23437e-06 - 2.57858e-06 - 1.53855e-06 - 8.9945e-07 - 5.15205e-07 - 2.89148e-07 - 1.59e-07 - 8.56664e-08 - 4.52232e-08 - 2.3391e-08 - 1.18542e-08 - 5.88618e-09 - 2.86373e-09 - 1.36511e-09 - 6.37584e-10 - 2.91774e-10 - 1.30825e-10 - 5.74743e-11 - 2.47396e-11 - 1.04339e-11 - 4.31162e-12 - 1.7457e-12 - 6.92527e-13 - 2.69178e-13 - 1.02513e-13 - 3.82523e-14 - 1.39853e-14 - 5.00981e-15 - 1.75836e-15 - 6.0469e-16 - 2.03748e-16 - 6.72655e-17 - 2.17584e-17 - 6.89602e-18 - 2.14145e-18 - 6.51557e-19 - 1.94238e-19 - 5.67355e-20 - 1.62376e-20 - 4.5551e-21 - 1.25945e-21 - 3.71542e-22 - 2.27974e-22 - 5.68172e-22 - 2.09885e-21 - 7.8741e-21 - 2.90139e-20 - 1.04766e-19 - 3.70658e-19 - 1.28488e-18 - 4.36407e-18 - 1.4523e-17 - 4.73541e-17 - 1.51285e-16 - 4.73556e-16 - 1.45239e-15 - 4.36447e-15 - 1.28504e-14 - 3.70715e-14 - 1.04785e-13 - 2.90199e-13 - 7.8746e-13 - 2.09362e-12 - 5.45388e-12 - 1.39203e-11 - 3.4812e-11 - 8.52996e-11 - 2.04786e-10 - 4.81717e-10 - 1.11025e-09 - 2.50718e-09 - 5.54737e-09 - 1.20261e-08 - 2.55447e-08 - 5.31635e-08 - 1.08408e-07 - 2.16596e-07 - 4.24008e-07 - 8.1327e-07 - 1.52839e-06 - 2.81429e-06 - 5.07741e-06 - 8.97539e-06 - 1.55454e-05 - 2.6381e-05 - 4.38651e-05 - 7.1464e-05 - 0.000114077 - 0.000178422 - 0.000273429 - 0.00041057 - 0.000604061 - 0.00087082 - 0.00123009 - 0.00170259 - 0.00230921 - 0.00306905 - 0.00399717 - 0.00510188 - 0.00638213 - 0.00782524 - 0.00940538 - 0.0110833 - 0.0128073 - 0.0145165 - 0.0161444 - 0.017625 - 0.0188984 - 0.0199167 - 0.0206486 - 0.0210826 - 0.0212274 - 0.0211103 - 0.0207729 - 0.0202659 - 0.0196419 - 0.0189493 - 0.0182273 - 0.0175019 - 0.0167852 - 0.0160761 - 0.015364 - 0.0146325 - 0.0138649 - 0.0130493 - 0.0121818 - 0.0112696 - 0.0103324 - 0.00940127 - 0.00851798 - 0.0077322 - 0.00709887 - 0.00667521 - 0.00651766 - 0.00667869 - 0.00720359 - 0.00812682 - 0.00946821 - 0.011229 - 0.0133879 - 0.0158984 - 0.0186868 - 0.0216531 - 0.0246735 - 0.0276071 - 0.0303043 - 0.0326177 - 0.0344137 - 0.0355844 - 0.0360574 - 0.0358021 - 0.0348328 - 0.033207 - 0.0310196 - 0.0283937 - 0.0254689 - 0.0223898 - 0.0192943 - 0.0163042 - 0.0135187 - 0.0110106 - 0.00882576 - 0.0069851 - 0.00548818 - 0.00431784 - 0.00344511 - 0.00283369 - 0.00244377 - 0.00223482 - 0.00216753 - 0.00220503 - 0.00231352 - 0.00246261 - 0.00262555 - 0.00277942 - 0.00290547 - 0.00298931 - 0.00302122 - 0.00299617 - 0.00291371 - 0.00277759 - 0.00259503 - 0.00237585 - 0.00213141 - 0.00187359 - 0.00161373 - 0.00136185 - 0.00112608 - 0.000912319 - 0.000724208 - 0.000563271 - 0.000429248 - 0.000320507 - 0.000234478 - 0.000168076 - 0.000118044 - 8.12306e-05 - 5.47687e-05 - 3.61811e-05 - 2.3419e-05 - 1.48522e-05 - 9.22895e-06 - 5.61887e-06 - 3.35184e-06 - 1.95909e-06 - 1.12192e-06 - 6.29515e-07 - 3.46089e-07 - 1.86426e-07 - 9.83922e-08 - 5.08806e-08 - 2.57799e-08 - 1.27981e-08 - 6.22514e-09 - 2.9668e-09 - 1.38537e-09 - 6.33837e-10 - 2.84137e-10 - 1.248e-10 - 5.37078e-11 - 2.26463e-11 - 9.3561e-12 - 3.78729e-12 - 1.5021e-12 - 5.83724e-13 - 2.22255e-13 - 8.29153e-14 - 3.03084e-14 - 1.08573e-14 - 3.81839e-15 - 1.34071e-15 - 5.41182e-16 - 4.60363e-16 - 1.01752e-15 - 2.94646e-15 - 8.68155e-15 - 2.5165e-14 - 7.1502e-14 - 1.99065e-13 - 5.43016e-13 - 1.45133e-12 - 3.80064e-12 - 9.75179e-12 - 2.4516e-11 - 6.03879e-11 - 1.45743e-10 - 3.44638e-10 - 7.985e-10 - 1.81269e-09 - 4.03189e-09 - 8.78681e-09 - 1.87625e-08 - 3.92542e-08 - 8.04673e-08 - 1.61618e-07 - 3.1805e-07 - 6.13252e-07 - 1.15856e-06 - 2.14455e-06 - 3.88947e-06 - 6.91164e-06 - 1.2034e-05 - 2.05292e-05 - 3.43142e-05 - 5.61969e-05 - 9.01753e-05 - 0.000141775 - 0.000218398 - 0.000329635 - 0.000487479 - 0.000706342 - 0.00100279 - 0.0013949 - 0.00190114 - 0.00253875 - 0.00332173 - 0.00425838 - 0.00534887 - 0.00658288 - 0.00793792 - 0.00937852 - 0.0108567 - 0.012314 - 0.0136848 - 0.0149009 - 0.0158974 - 0.0166178 - 0.01702 - 0.0170798 - 0.0167935 - 0.0161785 - 0.0152711 - 0.0141235 - 0.0127982 - 0.011363 - 0.00988489 - 0.00842537 - 0.00703628 - 0.0057575 - 0.00461596 - 0.00362599 - 0.0027908 - 0.00210459 - 0.00155504 - 0.00112578 - 0.000798551 - 0.000554994 - 0.00037793 - 0.000252157 - 0.000164842 - 0.000105584 - 6.62628e-05 - 4.07452e-05 - 2.45482e-05 - 1.44911e-05 - 8.38143e-06 - 4.74977e-06 - 2.63732e-06 - 1.4348e-06 - 7.64815e-07 - 3.99446e-07 - 2.04407e-07 - 1.02488e-07 - 5.03482e-08 - 2.42344e-08 - 1.14293e-08 - 5.28131e-09 - 2.39112e-09 - 1.06071e-09 - 4.6104e-10 - 1.96362e-10 - 8.19975e-11 - 3.37003e-11 - 1.39853e-11 - 6.7944e-12 - 6.04916e-12 - 1.11421e-11 - 2.65104e-11 - 6.50445e-11 - 1.57586e-10 - 3.74505e-10 - 8.72182e-10 - 1.99023e-09 - 4.44979e-09 - 9.74793e-09 - 2.09229e-08 - 4.40016e-08 - 9.06677e-08 - 1.83051e-07 - 3.62102e-07 - 7.01821e-07 - 1.33278e-06 - 2.47987e-06 - 4.52101e-06 - 8.07572e-06 - 1.4134e-05 - 2.42374e-05 - 4.07237e-05 - 6.70422e-05 - 0.000108141 - 0.000170912 - 0.000264666 - 0.000401576 - 0.000597016 - 0.000869669 - 0.0012413 - 0.00173603 - 0.00237906 - 0.00319468 - 0.00420373 - 0.00542055 - 0.00684968 - 0.00848282 - 0.0102964 - 0.0122502 - 0.0142879 - 0.016339 - 0.0183232 - 0.020156 - 0.0217557 - 0.0230507 - 0.0239861 - 0.0245286 - 0.0246698 - 0.0244259 - 0.0238353 - 0.0229533 - 0.0218461 - 0.020583 - 0.0192297 - 0.017843 - 0.0164669 - 0.0151311 - 0.0138521 - 0.0126351 - 0.0114777 - 0.0103735 - 0.00931555 - 0.00829889 - 0.00732211 - 0.00638771 - 0.00550152 - 0.00467147 - 0.00390611 - 0.00321307 - 0.0025979 - 0.00206328 - 0.00160877 - 0.00123095 - 0.000923967 - 0.00068019 - 0.000491014 - 0.000347568 - 0.000241316 - 0.000164498 - 0.000110414 - 7.35521e-05 - 4.96251e-05 - 3.55275e-05 - 2.92725e-05 - 2.99299e-05 - 3.75905e-05 - 5.33657e-05 - 7.9422e-05 - 0.000119033 - 0.000176624 - 0.000257776 - 0.000369135 - 0.000518189 - 0.000712868 - 0.000960941 - 0.00126921 - 0.00164251 - 0.00208268 - 0.00258746 - 0.00314965 - 0.00375654 - 0.00438986 - 0.00502632 - 0.0056388 - 0.00619812 - 0.00667528 - 0.00704395 - 0.00728283 - 0.00737769 - 0.00732281 - 0.00712151 - 0.00678583 - 0.00633536 - 0.0057953 - 0.00519418 - 0.00456137 - 0.00392474 - 0.00330874 - 0.00273308 - 0.00221196 - 0.00175404 - 0.00136282 - 0.00103747 - 0.000773837 - 0.000565535 - 0.000404955 - 0.000284112 - 0.000195304 - 0.000131543 - 8.68085e-05 - 5.61298e-05 - 3.556e-05 - 2.20733e-05 - 1.34248e-05 - 7.99994e-06 - 4.67091e-06 - 2.67211e-06 - 1.49776e-06 - 8.22563e-07 - 4.42621e-07 - 2.33363e-07 - 1.2055e-07 - 6.10156e-08 - 3.02588e-08 - 1.47027e-08 - 6.99974e-09 - 3.26515e-09 - 1.49231e-09 - 6.68274e-10 - 2.93215e-10 - 1.26053e-10 - 5.30957e-11 - 2.19129e-11 - 8.86093e-12 - 3.51071e-12 - 1.36285e-12 - 5.18366e-13 - 1.9318e-13 - 7.05383e-14 - 2.52363e-14 - 8.84629e-15 - 3.03833e-15 - 1.02246e-15 - 3.37126e-16 - 1.08912e-16 - 3.44744e-17 - 1.06919e-17 - 3.24899e-18 - 9.67342e-19 - 2.82194e-19 - 8.06591e-20 - 2.25889e-20 - 6.19832e-21 - 1.66644e-21 - 4.38977e-22 - 1.133e-22 - 2.86522e-23 - 7.09938e-24 - 1.72354e-24 - 4.09975e-25 - 9.55502e-26 - 2.18197e-26 - 4.88237e-27 - 1.07081e-27 - 2.30594e-28 - 4.92307e-29 - 1.09667e-29 - 3.14411e-30 - 1.6585e-30 - 1.45461e-30 - 1.49032e-30 - 1.55251e-30 - 1.59908e-30 - 1.63628e-30 - 1.75006e-30 - 2.39459e-30 - 5.85397e-30 - 2.33778e-29 - 1.09265e-28 - 5.20254e-28 - 2.44532e-27 - 1.12772e-26 - 5.097e-26 - 2.25729e-25 - 9.79488e-25 - 4.16437e-24 - 1.73475e-23 - 7.08044e-23 - 2.83153e-22 - 1.10948e-21 - 4.25944e-21 - 1.60223e-20 - 5.90517e-20 - 2.13244e-19 - 7.545e-19 - 2.61564e-18 - 8.88451e-18 - 2.95683e-17 - 9.64174e-17 - 3.0805e-16 - 9.64327e-16 - 2.95777e-15 - 8.88875e-15 - 2.6173e-14 - 7.551e-14 - 2.13448e-13 - 5.91175e-13 - 1.60427e-12 - 4.26554e-12 - 1.11124e-11 - 2.83648e-11 - 7.09396e-11 - 1.73834e-10 - 4.17365e-10 - 9.81826e-10 - 2.26303e-09 - 5.11073e-09 - 1.13087e-08 - 2.45176e-08 - 5.20812e-08 - 1.08398e-07 - 2.21054e-07 - 4.41684e-07 - 8.64693e-07 - 1.65863e-06 - 3.11726e-06 - 5.7403e-06 - 1.0357e-05 - 1.83091e-05 - 3.17131e-05 - 5.38204e-05 - 8.94936e-05 - 0.000145806 - 0.000232752 - 0.00036404 - 0.000557881 - 0.000837666 - 0.00123236 - 0.0017764 - 0.00250888 - 0.00347182 - 0.00470729 - 0.00625347 - 0.0081397 - 0.0103808 - 0.0129716 - 0.0158815 - 0.0190514 - 0.0223923 - 0.0257874 - 0.0290973 - 0.0321688 - 0.0348462 - 0.0369839 - 0.03846 - 0.0391873 - 0.0391222 - 0.0382691 - 0.0366799 - 0.0344491 - 0.0317048 - 0.0285972 - 0.0252851 - 0.0219241 - 0.0186558 - 0.0155996 - 0.0128486 - 0.0104682 - 0.00849752 - 0.00695295 - 0.00583265 - 0.00512109 - 0.00479313 - 0.00481689 - 0.00515543 - 0.0057673 - 0.00660627 - 0.00762079 - 0.00875373 - 0.00994277 - 0.0111219 - 0.0122239 - 0.0131839 - 0.0139439 - 0.0144563 - 0.0146885 - 0.0146249 - 0.0142685 - 0.0136401 - 0.0127762 - 0.0117254 - 0.0105438 - 0.00928966 - 0.00801941 - 0.00678301 - 0.00562134 - 0.0045645 - 0.00363149 - 0.00283082 - 0.00216211 - 0.001618 - 0.00118636 - 0.000852305 - 0.00059995 - 0.000413799 - 0.000279676 - 0.000185285 - 0.000120433 - 7.70311e-05 - 4.89435e-05 - 3.17797e-05 - 2.27018e-05 - 2.03111e-05 - 2.46741e-05 - 3.75481e-05 - 6.28712e-05 - 0.000107589 - 0.000182895 - 0.000305951 - 0.00050214 - 0.000807822 - 0.0012735 - 0.00196715 - 0.00297727 - 0.00441506 - 0.00641494 - 0.00913241 - 0.0127384 - 0.0174093 - 0.0233123 - 0.0305863 - 0.0393191 - 0.0495243 - 0.061118 - 0.0739021 - 0.0875551 - 0.101635 - 0.115596 - 0.128818 - 0.140653 - 0.150473 - 0.157726 - 0.161989 - 0.163007 - 0.160717 - 0.155259 - 0.146955 - 0.136286 - 0.123839 - 0.110255 - 0.0961777 - 0.0822032 - 0.0688398 - 0.0564843 - 0.0454101 - 0.0357696 - 0.0276066 - 0.020876 - 0.0154675 - 0.0112287 - 0.00798685 - 0.00556619 - 0.00380082 - 0.00254293 - 0.00166697 - 0.00107067 - 0.00067379 - 0.000415459 - 0.000250997 - 0.000148575 - 8.6171e-05 - 4.8968e-05 - 2.72647e-05 - 1.48739e-05 - 7.95037e-06 - 4.16376e-06 - 2.13659e-06 - 1.07422e-06 - 5.29179e-07 - 2.55416e-07 - 1.2079e-07 - 5.59695e-08 - 2.54102e-08 - 1.13032e-08 - 4.92642e-09 - 2.10377e-09 - 8.80244e-10 - 3.60866e-10 - 1.44959e-10 - 5.70702e-11 - 2.20651e-11 - 8.50316e-12 - 3.61538e-12 - 2.59818e-12 - 4.39325e-12 - 1.06972e-11 - 2.73806e-11 - 6.93422e-11 - 1.72287e-10 - 4.19489e-10 - 1.00077e-09 - 2.3393e-09 - 5.35766e-09 - 1.20227e-08 - 2.6434e-08 - 5.6946e-08 - 1.20199e-07 - 2.48584e-07 - 5.03712e-07 - 1.00007e-06 - 1.94542e-06 - 3.70795e-06 - 6.92454e-06 - 1.26703e-05 - 2.27152e-05 - 3.99011e-05 - 6.86735e-05 - 0.000115806 - 0.000191342 - 0.000309759 - 0.000491334 - 0.0007636 - 0.00116276 - 0.00173482 - 0.00253603 - 0.00363237 - 0.00509757 - 0.00700928 - 0.00944323 - 0.0124653 - 0.0161222 - 0.0204307 - 0.0253675 - 0.0308609 - 0.0367855 - 0.0429617 - 0.0491612 - 0.055119 - 0.0605504 - 0.0651732 - 0.0687319 - 0.0710206 - 0.071903 - 0.0713258 - 0.0693239 - 0.0660171 - 0.061598 - 0.0563137 - 0.0504426 - 0.0442709 - 0.0380694 - 0.0320753 - 0.026479 - 0.0214176 - 0.0169736 - 0.01318 - 0.0100276 - 0.00747499 - 0.00545963 - 0.00390708 - 0.00273955 - 0.00188209 - 0.0012669 - 0.00083556 - 0.000539947 - 0.00034187 - 0.000212084 - 0.000128912 - 7.67739e-05 - 4.47993e-05 - 2.56133e-05 - 1.43482e-05 - 7.87527e-06 - 4.23517e-06 - 2.23158e-06 - 1.1521e-06 - 5.82783e-07 - 2.88841e-07 - 1.40265e-07 - 6.67383e-08 - 3.11127e-08 - 1.42114e-08 - 6.36025e-09 - 2.789e-09 - 1.19828e-09 - 5.04435e-10 - 2.08061e-10 - 8.40835e-11 - 3.32942e-11 - 1.29171e-11 - 4.91018e-12 - 1.8289e-12 - 6.6774e-13 - 2.39797e-13 - 8.72764e-14 - 3.99832e-14 - 4.35828e-14 - 1.03126e-13 - 2.89491e-13 - 8.16136e-13 - 2.26089e-12 - 6.1391e-12 - 1.6335e-11 - 4.25903e-11 - 1.08812e-10 - 2.72406e-10 - 6.68242e-10 - 1.6063e-09 - 3.78352e-09 - 8.7326e-09 - 1.97501e-08 - 4.37696e-08 - 9.50509e-08 - 2.02264e-07 - 4.21756e-07 - 8.61758e-07 - 1.7254e-06 - 3.38515e-06 - 6.50799e-06 - 1.22603e-05 - 2.26326e-05 - 4.09407e-05 - 7.25704e-05 - 0.000126052 - 0.000214548 - 0.000357837 - 0.000584835 - 0.000936631 - 0.00146991 - 0.00226049 - 0.00340645 - 0.00503026 - 0.00727894 - 0.0103214 - 0.0143415 - 0.0195274 - 0.0260547 - 0.0340659 - 0.0436462 - 0.0547983 - 0.067419 - 0.0812818 - 0.0960292 - 0.111177 - 0.126134 - 0.140238 - 0.1528 - 0.163165 - 0.170766 - 0.175185 - 0.17619 - 0.173768 - 0.168128 - 0.159688 - 0.149039 - 0.136897 - 0.124043 - 0.111262 - 0.0992842 - 0.0887352 - 0.0800981 - 0.073689 - 0.0696463 - 0.0679332 - 0.0683505 - 0.0705597 - 0.0741133 - 0.0784909 - 0.0831388 - 0.0875109 - 0.091108 - 0.0935133 - 0.09442 - 0.0936496 - 0.0911582 - 0.0870312 - 0.0814664 - 0.0747481 - 0.0672163 - 0.0592323 - 0.0511477 - 0.0432772 - 0.0358796 - 0.0291464 - 0.0231988 - 0.0180921 - 0.0138246 - 0.0103503 - 0.00759265 - 0.00545721 - 0.00384313 - 0.00265178 - 0.00179277 - 0.00118755 - 0.000770751 - 0.000490133 - 0.000305387 - 0.000186433 - 0.000111515 - 6.53553e-05 - 3.75288e-05 - 2.11147e-05 - 1.16397e-05 - 6.28691e-06 - 3.32712e-06 - 1.7252e-06 - 8.76506e-07 - 4.36367e-07 - 2.12967e-07 - 1.02108e-07 - 4.86122e-08 - 2.41809e-08 - 1.51788e-08 - 1.63679e-08 - 2.83794e-08 - 5.79776e-08 - 1.21189e-07 - 2.50402e-07 - 5.07817e-07 - 1.0094e-06 - 1.966e-06 - 3.75186e-06 - 7.01533e-06 - 1.28525e-05 - 2.30708e-05 - 4.05764e-05 - 6.99235e-05 - 0.000118062 - 0.000195313 - 0.000316586 - 0.000502791 - 0.000782386 - 0.00119286 - 0.00178196 - 0.00260821 - 0.00374044 - 0.00525582 - 0.00723593 - 0.00976081 - 0.0129007 - 0.0167063 - 0.0211974 - 0.0263525 - 0.0320995 - 0.0383101 - 0.044799 - 0.051329 - 0.0576237 - 0.0633852 - 0.0683175 - 0.0721517 - 0.0746721 - 0.0757375 - 0.0752976 - 0.0734009 - 0.0701941 - 0.0659136 - 0.0608694 - 0.0554259 - 0.0499808 - 0.0449442 - 0.0407215 - 0.0376986 - 0.0362307 - 0.0366324 - 0.0391679 - 0.0440395 - 0.0513723 - 0.0611977 - 0.0734333 - 0.0878657 - 0.104137 - 0.121742 - 0.140036 - 0.158258 - 0.175572 - 0.191121 - 0.204086 - 0.213755 - 0.219579 - 0.221225 - 0.218598 - 0.211856 - 0.201386 - 0.187772 - 0.171739 - 0.15409 - 0.13564 - 0.117162 - 0.0993331 - 0.0827063 - 0.0676928 - 0.0545624 - 0.043456 - 0.034407 - 0.0273684 - 0.022239 - 0.0188874 - 0.0171699 - 0.0169424 - 0.0180635 - 0.0203933 - 0.0237859 - 0.0280809 - 0.0330942 - 0.0386122 - 0.0443899 - 0.0501552 - 0.0556197 - 0.0604946 - 0.06451 - 0.0674365 - 0.069104 - 0.069417 - 0.0683625 - 0.0660108 - 0.0625073 - 0.0580577 - 0.0529082 - 0.0473225 - 0.0415602 - 0.0358572 - 0.0304111 - 0.0253728 - 0.0208431 - 0.0168751 - 0.0134805 - 0.0106382 - 0.00830404 - 0.00641967 - 0.00492092 - 0.00374385 - 0.00282905 - 0.0021241 - 0.00158457 - 0.00117404 - 0.000863316 - 0.000629412 - 0.000454431 - 0.000324503 - 0.000228896 - 0.000159295 - 0.000109251 - 7.37695e-05 - 4.89991e-05 - 3.19922e-05 - 2.05207e-05 - 1.2926e-05 - 7.99585e-06 - 4.86315e-06 - 2.92334e-06 - 1.76948e-06 - 1.1439e-06 - 9.06437e-07 - 1.02009e-06 - 1.55632e-06 - 2.72467e-06 - 4.93405e-06 - 8.89688e-06 - 1.57908e-05 - 2.74957e-05 - 4.69267e-05 - 7.84794e-05 - 0.0001286 - 0.000206474 - 0.00032481 - 0.000500644 - 0.000756078 - 0.00111877 - 0.001622 - 0.00230409 - 0.00320688 - 0.00437325 - 0.00584336 - 0.00764994 - 0.00981273 - 0.0123327 - 0.0151867 - 0.0183234 - 0.0216614 - 0.0250901 - 0.0284744 - 0.0316625 - 0.0344962 - 0.0368244 - 0.0385155 - 0.0394705 - 0.0396321 - 0.0389905 - 0.0375844 - 0.0354972 - 0.032849 - 0.0297846 - 0.0264615 - 0.0230361 - 0.0196523 - 0.0164334 - 0.0134759 - 0.0108485 - 0.00859408 - 0.00673434 - 0.00527724 - 0.00422554 - 0.0035862 - 0.00337974 - 0.00364906 - 0.0044669 - 0.00594138 - 0.00821867 - 0.0114817 - 0.0159437 - 0.0218363 - 0.0293904 - 0.0388107 - 0.0502452 - 0.0637519 - 0.0792647 - 0.0965666 - 0.115271 - 0.134821 - 0.154501 - 0.173478 - 0.190851 - 0.205722 - 0.217273 - 0.224836 - 0.227963 - 0.226463 - 0.220429 - 0.210222 - 0.196437 - 0.179848 - 0.161335 - 0.141805 - 0.122123 - 0.103052 - 0.0852089 - 0.0690443 - 0.0548367 - 0.0427079 - 0.0326479 - 0.0245476 - 0.0182336 - 0.0135014 - 0.0101435 - 0.00797117 - 0.00682901 - 0.00660242 - 0.0072185 - 0.00864139 - 0.0108629 - 0.0138899 - 0.0177288 - 0.0223696 - 0.0277703 - 0.0338438 - 0.0404482 - 0.0473843 - 0.0543987 - 0.0611951 - 0.0674527 - 0.0728495 - 0.0770893 - 0.0799283 - 0.0811978 - 0.0808212 - 0.0788213 - 0.075318 - 0.0705165 - 0.0646873 - 0.0581413 - 0.0512021 - 0.0441801 - 0.0373511 - 0.0309397 - 0.0251111 - 0.0199688 - 0.0155588 - 0.0118778 - 0.00888452 - 0.0065113 - 0.00467562 - 0.00328963 - 0.00226773 - 0.0015317 - 0.00101366 - 0.000657274 - 0.000417579 - 0.000259937 - 0.000158538 - 9.47405e-05 - 5.54722e-05 - 3.18238e-05 - 1.78881e-05 - 9.85178e-06 - 5.3162e-06 - 2.81077e-06 - 1.45608e-06 - 7.39066e-07 - 3.67551e-07 - 1.79097e-07 - 8.55059e-08 - 3.99982e-08 - 1.83325e-08 - 8.23268e-09 - 3.6224e-09 - 1.56167e-09 - 6.59656e-10 - 2.73013e-10 - 1.1071e-10 - 4.39871e-11 - 1.71239e-11 - 6.53153e-12 - 2.44099e-12 - 8.93824e-13 - 3.20683e-13 - 1.12729e-13 - 3.8827e-14 - 1.31029e-14 - 4.33251e-15 - 1.40361e-15 - 4.45546e-16 - 1.38572e-16 - 4.22272e-17 - 1.26081e-17 - 3.68842e-18 - 1.05723e-18 - 2.96917e-19 - 8.1703e-20 - 2.20281e-20 - 5.81908e-21 - 1.50615e-21 - 3.8196e-22 - 9.49084e-23 - 2.31062e-23 - 5.51175e-24 - 1.28821e-24 - 2.94999e-25 - 6.619e-26 - 1.45513e-26 - 3.13434e-27 - 6.61496e-28 - 1.36787e-28 - 2.77142e-29 - 5.50166e-30 - 1.0701e-30 - 2.03934e-31 - 3.80796e-32 - 6.96678e-33 - 1.24885e-33 - 2.19343e-34 - 3.77464e-35 - 6.36451e-36 - 1.05147e-36 - 1.70203e-37 - 2.69954e-38 - 4.19537e-39 - 6.38903e-40 - 9.53519e-41 - 1.3949e-41 - 2.00106e-42 - 2.81732e-43 - 3.89922e-44 - 5.32179e-45 - 7.2059e-46 - 9.78671e-47 - 1.35807e-47 - 1.9776e-48 - 3.11139e-49 - 5.38237e-50 - 1.01758e-50 - 2.05015e-51 - 4.27466e-52 - 9.01541e-53 - 1.89551e-53 - 3.94041e-54 - 8.06331e-55 - 1.6205e-55 - 3.19476e-56 - 6.17487e-57 - 1.16973e-57 - 2.17142e-58 - 3.94979e-59 - 7.03975e-60 - 1.22938e-60 - 2.10356e-61 - 3.52667e-62 - 5.7931e-63 - 9.32385e-64 - 1.47033e-64 - 2.27182e-65 - 3.43929e-66 - 5.10154e-67 - 7.4143e-68 - 1.05579e-68 - 1.47305e-69 - 2.01371e-70 - 2.6972e-71 - 3.5397e-72 - 4.55152e-73 - 5.73433e-74 - 7.07859e-75 - 8.56144e-76 - 1.01457e-76 - 1.17804e-77 - 1.3402e-78 - 1.49388e-79 - 1.63155e-80 - 1.74591e-81 - 1.83055e-82 - 1.88051e-83 - 1.89282e-84 - 1.86671e-85 - 1.80378e-86 - 1.70776e-87 - 1.58418e-88 - 1.43986e-89 - 1.28225e-90 - 1.11883e-91 - 9.56514e-93 - 8.01225e-94 - 6.5759e-95 - 5.28801e-96 - 4.16646e-97 - 3.21646e-98 - 2.43291e-99 - 1.80306e-100 - 1.30928e-101 - 9.31516e-103 - 6.4936e-104 - 4.43525e-105 - 2.96816e-106 - 1.94622e-107 - 1.25036e-108 - 7.87072e-110 - 4.85435e-111 - 2.93349e-112 - 1.7369e-113 - 1.00763e-114 - 5.7275e-116 - 3.18981e-117 - 1.74062e-118 - 9.3063e-120 - 4.87515e-121 - 2.50228e-122 - 1.25841e-123 - 6.20073e-125 - 2.99365e-126 - 1.41611e-127 - 6.5634e-129 - 2.98056e-130 - 1.32618e-131 - 5.78158e-133 - 2.4696e-134 - 1.03398e-135 - 5.23961e-137 - 2.4671e-136 - 5.87423e-135 - 1.37991e-133 - 3.17607e-132 - 7.16255e-131 - 1.58264e-129 - 3.42636e-128 - 7.2681e-127 - 1.51059e-125 - 3.07615e-124 - 6.1377e-123 - 1.19989e-121 - 2.29833e-120 - 4.31342e-119 - 7.93173e-118 - 1.42906e-116 - 2.52272e-115 - 4.36341e-114 - 7.39468e-113 - 1.22786e-111 - 1.99764e-110 - 3.18435e-109 - 4.97348e-108 - 7.61094e-107 - 1.14118e-105 - 1.6765e-104 - 2.41319e-103 - 3.40343e-102 - 4.70303e-101 - 6.36761e-100 - 8.44718e-99 - 1.09795e-97 - 1.39828e-96 - 1.74477e-95 - 2.13315e-94 - 2.5553e-93 - 2.99915e-92 - 3.44898e-91 - 3.88617e-90 - 4.29031e-89 - 4.64079e-88 - 4.9185e-87 - 5.10753e-86 - 5.19667e-85 - 5.18055e-84 - 5.06016e-83 - 4.84272e-82 - 4.54099e-81 - 4.17205e-80 - 3.75565e-79 - 3.31251e-78 - 2.86264e-77 - 2.42389e-76 - 2.01092e-75 - 1.63461e-74 - 1.30188e-73 - 1.01593e-72 - 7.76775e-72 - 5.81919e-71 - 4.27136e-70 - 3.0719e-69 - 2.16464e-68 - 1.49451e-67 - 1.011e-66 - 6.70099e-66 - 4.35175e-65 - 2.76902e-64 - 1.72633e-63 - 1.05453e-62 - 6.31151e-62 - 3.7012e-61 - 2.12662e-60 - 1.19721e-59 - 6.60377e-59 - 3.56902e-58 - 1.88992e-57 - 9.80558e-57 - 4.98472e-56 - 2.48282e-55 - 1.21168e-54 - 5.79381e-54 - 2.71444e-53 - 1.24604e-52 - 5.60429e-52 - 2.46971e-51 - 1.06637e-50 - 4.51138e-50 - 1.87002e-49 - 7.59486e-49 - 3.02225e-48 - 1.17836e-47 - 4.50157e-47 - 1.68495e-46 - 6.17938e-46 - 2.22045e-45 - 7.81763e-45 - 2.69678e-44 - 9.11496e-44 - 3.01857e-43 - 9.79454e-43 - 3.1139e-42 - 9.69982e-42 - 2.96047e-41 - 8.85309e-41 - 2.59399e-40 - 7.44696e-40 - 2.09473e-39 - 5.77321e-39 - 1.55899e-38 - 4.12488e-38 - 1.06935e-37 - 2.71622e-37 - 6.76009e-37 - 1.64847e-36 - 3.93869e-36 - 9.22071e-36 - 2.11505e-35 - 4.75358e-35 - 1.0468e-34 - 2.25869e-34 - 4.77525e-34 - 9.89201e-34 - 2.00781e-33 - 3.99315e-33 - 7.78148e-33 - 1.48582e-32 - 2.77992e-32 - 5.09636e-32 - 9.15496e-32 - 1.61148e-31 - 2.77951e-31 - 4.6978e-31 - 7.78054e-31 - 1.26276e-30 - 2.00832e-30 - 3.13012e-30 - 4.78091e-30 - 7.15641e-30 - 1.04985e-29 - 1.50947e-29 - 2.12717e-29 - 2.93818e-29 - 3.9781e-29 - 5.27983e-29 - 6.8697e-29 - 8.76313e-29 - 1.09602e-28 - 1.34418e-28 - 1.61664e-28 - 1.90694e-28 - 2.20637e-28 - 2.50436e-28 - 2.78902e-28 - 3.04799e-28 - 3.2693e-28 - 3.44234e-28 - 3.55871e-28 - 3.61294e-28 - 3.60286e-28 - 3.52976e-28 - 3.39822e-28 - 3.21557e-28 - 2.99129e-28 - 2.73615e-28 - 2.46141e-28 - 2.17802e-28 - 1.896e-28 - 1.62392e-28 - 1.36859e-28 - 1.13498e-28 - 9.26219e-29 - 7.43767e-29 - 5.87669e-29 - - - - - -1.79304e-31 - -2.23139e-31 - -2.72081e-31 - -3.25055e-31 - -3.80497e-31 - -4.36383e-31 - -4.90295e-31 - -5.39336e-31 - -5.79128e-31 - -5.97855e-31 - -5.45356e-31 - -1.79829e-31 - 1.67345e-30 - 1.06286e-29 - 5.29773e-29 - 2.49307e-28 - 1.14128e-27 - 5.1117e-27 - 2.24259e-26 - 9.63932e-26 - 4.05952e-25 - 1.67509e-24 - 6.77235e-24 - 2.68273e-23 - 1.04124e-22 - 3.9597e-22 - 1.4754e-21 - 5.38636e-21 - 1.92671e-20 - 6.75267e-20 - 2.31884e-19 - 7.80195e-19 - 2.57201e-18 - 8.30765e-18 - 2.62918e-17 - 8.15268e-17 - 2.47695e-16 - 7.37345e-16 - 2.15061e-15 - 6.14593e-15 - 1.72088e-14 - 4.7212e-14 - 1.26908e-13 - 3.34245e-13 - 8.62533e-13 - 2.18084e-12 - 5.40267e-12 - 1.31139e-11 - 3.11881e-11 - 7.26749e-11 - 1.65927e-10 - 3.71181e-10 - 8.13565e-10 - 1.74717e-09 - 3.67633e-09 - 7.57934e-09 - 1.53103e-08 - 3.03023e-08 - 5.87628e-08 - 1.11652e-07 - 2.07858e-07 - 3.79145e-07 - 6.77609e-07 - 1.18656e-06 - 2.03582e-06 - 3.42235e-06 - 5.63697e-06 - 9.09714e-06 - 1.43847e-05 - 2.2286e-05 - 3.383e-05 - 5.03162e-05 - 7.33248e-05 - 0.000104696 - 0.00014647 - 0.000200771 - 0.000269645 - 0.000354829 - 0.000457491 - 0.000577942 - 0.000715355 - 0.000867554 - 0.00103088 - 0.00120021 - 0.00136913 - 0.00153027 - 0.00167583 - 0.00179817 - 0.00189049 - 0.00194743 - 0.00196562 - 0.00194402 - 0.001884 - 0.00178928 - 0.00166553 - 0.00151987 - 0.00136029 - 0.00119498 - 0.00103175 - 0.000877556 - 0.000738187 - 0.000618089 - 0.000520341 - 0.000446731 - 0.000397902 - 0.000373517 - 0.000372412 - 0.000392709 - 0.000431897 - 0.000486877 - 0.000554005 - 0.000629151 - 0.000707798 - 0.000785206 - 0.00085663 - 0.000917596 - 0.000964193 - 0.000993359 - 0.00100311 - 0.000992716 - 0.000962696 - 0.000914787 - 0.000851735 - 0.000777026 - 0.000694558 - 0.000608305 - 0.000522003 - 0.000438896 - 0.000361567 - 0.000291845 - 0.000230809 - 0.000178851 - 0.000135789 - 0.000101013 - 7.3625e-05 - 5.25788e-05 - 3.67903e-05 - 2.52227e-05 - 1.69429e-05 - 1.11512e-05 - 7.19101e-06 - 4.54356e-06 - 2.8128e-06 - 1.70616e-06 - 1.014e-06 - 5.90459e-07 - 3.36884e-07 - 1.88325e-07 - 1.03151e-07 - 5.53571e-08 - 2.91079e-08 - 1.49964e-08 - 7.57003e-09 - 3.74409e-09 - 1.81439e-09 - 8.61497e-10 - 4.00786e-10 - 1.82687e-10 - 8.15908e-11 - 3.57036e-11 - 1.53081e-11 - 6.43124e-12 - 2.64844e-12 - 1.07197e-12 - 4.34795e-13 - 2.00078e-13 - 1.63936e-13 - 3.00174e-13 - 7.46316e-13 - 1.9281e-12 - 4.92178e-12 - 1.23241e-11 - 3.02408e-11 - 7.27073e-11 - 1.71278e-10 - 3.95331e-10 - 8.94041e-10 - 1.98103e-09 - 4.30092e-09 - 9.1489e-09 - 1.90683e-08 - 3.89398e-08 - 7.79133e-08 - 1.52745e-07 - 2.93398e-07 - 5.52187e-07 - 1.01824e-06 - 1.83973e-06 - 3.25681e-06 - 5.64895e-06 - 9.60021e-06 - 1.59856e-05 - 2.60805e-05 - 4.16907e-05 - 6.52979e-05 - 0.000100207 - 0.000150671 - 0.000221973 - 0.000320411 - 0.00045316 - 0.000627962 - 0.000852611 - 0.00113424 - 0.00147842 - 0.0018881 - 0.0023626 - 0.00289663 - 0.00347962 - 0.00409551 - 0.00472303 - 0.00533667 - 0.00590822 - 0.00640885 - 0.00681147 - 0.00709314 - 0.00723725 - 0.00723514 - 0.00708696 - 0.00680167 - 0.00639614 - 0.00589358 - 0.00532131 - 0.00470844 - 0.00408352 - 0.00347255 - 0.00289752 - 0.00237559 - 0.00191887 - 0.00153483 - 0.00122706 - 0.000996256 - 0.000841365 - 0.000760476 - 0.000751552 - 0.000812793 - 0.000942642 - 0.00113944 - 0.00140081 - 0.00172285 - 0.0020993 - 0.00252088 - 0.00297478 - 0.00344473 - 0.00391135 - 0.00435317 - 0.00474797 - 0.00507448 - 0.00531415 - 0.00545286 - 0.00548225 - 0.00540051 - 0.00521259 - 0.00492967 - 0.00456811 - 0.00414791 - 0.00369091 - 0.00321908 - 0.00275291 - 0.00231029 - 0.00190581 - 0.00155063 - 0.00125284 - 0.00101825 - 0.000851451 - 0.000756966 - 0.000740438 - 0.000809583 - 0.00097481 - 0.00124937 - 0.00164889 - 0.00219032 - 0.00289009 - 0.0037618 - 0.00481339 - 0.00604416 - 0.00744202 - 0.00898139 - 0.0106221 - 0.0123098 - 0.013978 - 0.015552 - 0.0169538 - 0.0181087 - 0.0189516 - 0.019433 - 0.0195241 - 0.0192195 - 0.0185373 - 0.0175182 - 0.0162207 - 0.0147159 - 0.013081 - 0.0113928 - 0.00972205 - 0.00812871 - 0.00665921 - 0.00534516 - 0.00420373 - 0.00323926 - 0.00244565 - 0.00180917 - 0.0013113 - 0.000931238 - 0.000647972 - 0.000441762 - 0.000295092 - 0.000193136 - 0.000123853 - 7.78193e-05 - 4.79075e-05 - 2.88973e-05 - 1.70784e-05 - 9.88952e-06 - 5.61099e-06 - 3.11918e-06 - 1.69894e-06 - 9.06678e-07 - 4.74094e-07 - 2.42892e-07 - 1.21926e-07 - 5.99681e-08 - 2.88987e-08 - 1.3645e-08 - 6.31259e-09 - 2.86139e-09 - 1.27082e-09 - 5.53001e-10 - 2.3578e-10 - 9.8497e-11 - 4.03159e-11 - 1.61684e-11 - 6.35323e-12 - 2.44601e-12 - 9.22699e-13 - 3.41034e-13 - 1.23502e-13 - 4.38212e-14 - 1.52347e-14 - 5.18941e-15 - 1.73197e-15 - 5.66369e-16 - 1.81466e-16 - 5.69676e-17 - 1.75226e-17 - 5.28085e-18 - 1.55936e-18 - 4.51156e-19 - 1.27892e-19 - 3.5522e-20 - 9.66693e-21 - 2.5776e-21 - 6.73413e-22 - 1.72385e-22 - 4.3267e-23 - 1.07691e-23 - 3.17029e-24 - 3.13795e-24 - 1.04192e-23 - 4.08832e-23 - 1.59072e-22 - 6.06845e-22 - 2.26838e-21 - 8.3079e-21 - 2.98129e-20 - 1.04822e-19 - 3.6111e-19 - 1.21889e-18 - 4.03109e-18 - 1.30623e-17 - 4.14718e-17 - 1.2901e-16 - 3.93216e-16 - 1.17429e-15 - 3.43603e-15 - 9.85087e-15 - 2.76713e-14 - 7.6159e-14 - 2.05376e-13 - 5.42644e-13 - 1.40481e-12 - 3.56333e-12 - 8.85588e-12 - 2.15648e-11 - 5.1451e-11 - 1.20276e-10 - 2.75489e-10 - 6.1825e-10 - 1.35944e-09 - 2.92883e-09 - 6.18252e-09 - 1.27871e-08 - 2.5913e-08 - 5.14516e-08 - 1.00096e-07 - 1.90797e-07 - 3.56339e-07 - 6.52066e-07 - 1.16912e-06 - 2.05381e-06 - 3.53508e-06 - 5.96177e-06 - 9.85117e-06 - 1.59492e-05 - 2.53002e-05 - 3.93232e-05 - 5.98839e-05 - 8.93528e-05 - 0.00013063 - 0.000187119 - 0.000262622 - 0.000361147 - 0.000486606 - 0.000642415 - 0.000830998 - 0.00105327 - 0.00130809 - 0.00159189 - 0.00189838 - 0.00221861 - 0.00254131 - 0.00285355 - 0.00314183 - 0.00339333 - 0.00359738 - 0.00374691 - 0.00383975 - 0.00387957 - 0.00387642 - 0.00384664 - 0.00381224 - 0.0037996 - 0.00383766 - 0.00395569 - 0.00418069 - 0.0045347 - 0.00503217 - 0.00567767 - 0.00646409 - 0.00737179 - 0.00836862 - 0.00941129 - 0.0104478 - 0.011421 - 0.012273 - 0.0129503 - 0.013408 - 0.0136139 - 0.013552 - 0.0132229 - 0.0126445 - 0.011849 - 0.0108804 - 0.00978988 - 0.00863109 - 0.00745595 - 0.00631082 - 0.00523372 - 0.00425281 - 0.00338595 - 0.00264134 - 0.00201887 - 0.00151194 - 0.00110944 - 0.000797693 - 0.000562024 - 0.000388087 - 0.000262733 - 0.000174531 - 0.00011398 - 7.34973e-05 - 4.72468e-05 - 3.0892e-05 - 2.1316e-05 - 1.63489e-05 - 1.45246e-05 - 1.48771e-05 - 1.6778e-05 - 1.98118e-05 - 2.36856e-05 - 2.81654e-05 - 3.30354e-05 - 3.80753e-05 - 4.30518e-05 - 4.77214e-05 - 5.18411e-05 - 5.51843e-05 - 5.75589e-05 - 5.88241e-05 - 5.89031e-05 - 5.77908e-05 - 5.55543e-05 - 5.23255e-05 - 4.82887e-05 - 4.36632e-05 - 3.86831e-05 - 3.35788e-05 - 2.85591e-05 - 2.37992e-05 - 1.94319e-05 - 1.55456e-05 - 1.21852e-05 - 9.35833e-06 - 7.04206e-06 - 5.19204e-06 - 3.7507e-06 - 2.65476e-06 - 1.84108e-06 - 1.251e-06 - 8.32877e-07 - 5.433e-07 - 3.47245e-07 - 2.17455e-07 - 1.33426e-07 - 8.02134e-08 - 4.72494e-08 - 2.72712e-08 - 1.54258e-08 - 8.55829e-09 - 4.67484e-09 - 2.55734e-09 - 1.50298e-09 - 1.16839e-09 - 1.50985e-09 - 2.83243e-09 - 5.99601e-09 - 1.28846e-08 - 2.73441e-08 - 5.6957e-08 - 1.16287e-07 - 2.32641e-07 - 4.56022e-07 - 8.75839e-07 - 1.64816e-06 - 3.03887e-06 - 5.48987e-06 - 9.71737e-06 - 1.68528e-05 - 2.86373e-05 - 4.76792e-05 - 7.77791e-05 - 0.000124318 - 0.000194689 - 0.000298735 - 0.000449125 - 0.000661584 - 0.00095486 - 0.0013503 - 0.00187094 - 0.00253995 - 0.00337853 - 0.00440319 - 0.00562268 - 0.00703487 - 0.00862394 - 0.0103584 - 0.0121903 - 0.0140565 - 0.0158808 - 0.0175795 - 0.0190668 - 0.0202621 - 0.0210974 - 0.0215234 - 0.0215144 - 0.021071 - 0.0202198 - 0.019011 - 0.0175134 - 0.0158078 - 0.0139802 - 0.012114 - 0.0102849 - 0.00855561 - 0.00697329 - 0.00556879 - 0.00435734 - 0.00334055 - 0.0025093 - 0.00184682 - 0.00133178 - 0.00094097 - 0.000651414 - 0.000441851 - 0.00029365 - 0.000191215 - 0.000121998 - 7.62635e-05 - 4.6711e-05 - 2.80323e-05 - 1.6483e-05 - 9.49616e-06 - 5.36041e-06 - 2.96473e-06 - 1.6066e-06 - 8.53039e-07 - 4.43778e-07 - 2.26204e-07 - 1.12972e-07 - 5.52814e-08 - 2.65047e-08 - 1.2451e-08 - 5.73091e-09 - 2.58451e-09 - 1.14202e-09 - 4.94439e-10 - 2.09776e-10 - 8.72944e-11 - 3.58409e-11 - 1.50894e-11 - 7.98683e-12 - 8.47379e-12 - 1.69387e-11 - 4.04834e-11 - 9.81476e-11 - 2.34372e-10 - 5.48791e-10 - 1.2592e-09 - 2.83093e-09 - 6.23592e-09 - 1.34589e-08 - 2.84612e-08 - 5.89705e-08 - 1.19716e-07 - 2.38127e-07 - 4.64089e-07 - 8.86197e-07 - 1.65804e-06 - 3.03948e-06 - 5.45932e-06 - 9.6076e-06 - 1.65664e-05 - 2.79884e-05 - 4.63304e-05 - 7.51432e-05 - 0.000119413 - 0.000185929 - 0.00028365 - 0.000423988 - 0.000620957 - 0.00089106 - 0.00125282 - 0.00172587 - 0.0023295 - 0.00308074 - 0.00399194 - 0.00506816 - 0.00630454 - 0.0076841 - 0.00917634 - 0.010737 - 0.0123093 - 0.0138268 - 0.0152175 - 0.0164099 - 0.0173382 - 0.0179489 - 0.0182058 - 0.0180933 - 0.0176182 - 0.0168091 - 0.0157131 - 0.0143919 - 0.0129155 - 0.0113564 - 0.00978378 - 0.00825866 - 0.00683045 - 0.00553511 - 0.0043948 - 0.00341893 - 0.00260602 - 0.00194626 - 0.00142417 - 0.00102108 - 0.00071729 - 0.000493703 - 0.000332947 - 0.000219998 - 0.00014243 - 9.03484e-05 - 5.61534e-05 - 3.41955e-05 - 2.04032e-05 - 1.19279e-05 - 6.83231e-06 - 3.83449e-06 - 2.10855e-06 - 1.13605e-06 - 5.9972e-07 - 3.10196e-07 - 1.57203e-07 - 7.80587e-08 - 3.79768e-08 - 1.81031e-08 - 8.45523e-09 - 3.86931e-09 - 1.73492e-09 - 7.62186e-10 - 3.2808e-10 - 1.38368e-10 - 5.71779e-11 - 2.31504e-11 - 9.18384e-12 - 3.56967e-12 - 1.35946e-12 - 5.07276e-13 - 1.85463e-13 - 6.64368e-14 - 2.33183e-14 - 8.01901e-15 - 2.70198e-15 - 8.9203e-16 - 2.88546e-16 - 9.14505e-17 - 2.83985e-17 - 8.64052e-18 - 2.57586e-18 - 7.52387e-19 - 2.15326e-19 - 6.03798e-20 - 1.65898e-20 - 4.46888e-21 - 1.1907e-21 - 3.54724e-22 - 2.70303e-22 - 7.45922e-22 - 2.68317e-21 - 9.67303e-21 - 3.42192e-20 - 1.1862e-19 - 4.02889e-19 - 1.34076e-18 - 4.37171e-18 - 1.39666e-17 - 4.37185e-17 - 1.34084e-16 - 4.02927e-16 - 1.18635e-15 - 3.42243e-15 - 9.67374e-15 - 2.67911e-14 - 7.26984e-14 - 1.93284e-13 - 5.03504e-13 - 1.28513e-12 - 3.21388e-12 - 7.87496e-12 - 1.89062e-11 - 4.44733e-11 - 1.02502e-10 - 2.31473e-10 - 5.12165e-10 - 1.11034e-09 - 2.35855e-09 - 4.90879e-09 - 1.00102e-08 - 2.00013e-08 - 3.91579e-08 - 7.51153e-08 - 1.41185e-07 - 2.60021e-07 - 4.69236e-07 - 8.29749e-07 - 1.43776e-06 - 2.44132e-06 - 4.06237e-06 - 6.62491e-06 - 1.0589e-05 - 1.65903e-05 - 2.54819e-05 - 3.83764e-05 - 5.66822e-05 - 8.21314e-05 - 0.000116792 - 0.00016307 - 0.000223695 - 0.000301717 - 0.000400522 - 0.000523905 - 0.000676234 - 0.000862732 - 0.00108992 - 0.00136617 - 0.00170242 - 0.00211279 - 0.00261515 - 0.00323128 - 0.00398644 - 0.0049082 - 0.00602429 - 0.00735944 - 0.00893147 - 0.0107468 - 0.0127958 - 0.0150493 - 0.0174561 - 0.0199423 - 0.0224143 - 0.0247631 - 0.0268725 - 0.0286279 - 0.0299271 - 0.0306897 - 0.0308656 - 0.0304406 - 0.0294379 - 0.027917 - 0.0259682 - 0.0237049 - 0.0212543 - 0.0187473 - 0.0163094 - 0.0140532 - 0.012073 - 0.0104412 - 0.00920708 - 0.00839714 - 0.00801617 - 0.00804901 - 0.00846236 - 0.00920671 - 0.0102183 - 0.0114214 - 0.0127314 - 0.0140583 - 0.0153113 - 0.0164042 - 0.0172608 - 0.0178203 - 0.0180418 - 0.0179068 - 0.017421 - 0.0166126 - 0.0155294 - 0.0142346 - 0.0128009 - 0.0113044 - 0.00981974 - 0.00841484 - 0.00714807 - 0.00606625 - 0.00520401 - 0.00458421 - 0.00421879 - 0.00410993 - 0.00425091 - 0.00462675 - 0.00521439 - 0.00598278 - 0.0068929 - 0.00789827 - 0.00894605 - 0.00997897 - 0.0109382 - 0.0117671 - 0.0124148 - 0.0128403 - 0.0130161 - 0.01293 - 0.0125862 - 0.0120047 - 0.011219 - 0.0102732 - 0.00921717 - 0.00810268 - 0.00697907 - 0.00588986 - 0.00487025 - 0.00394581 - 0.00313231 - 0.00243638 - 0.00185693 - 0.00138696 - 0.00101542 - 0.000729103 - 0.000514093 - 0.000357004 - 0.000245774 - 0.00017014 - 0.00012182 - 9.44836e-05 - 8.35732e-05 - 8.60424e-05 - 0.000100046 - 0.000124617 - 0.000159347 - 0.000204081 - 0.000258632 - 0.000322534 - 0.000394841 - 0.000473977 - 0.000557675 - 0.000642995 - 0.000726438 - 0.000804153 - 0.00087221 - 0.00092692 - 0.000965165 - 0.000984686 - 0.000984308 - 0.000964054 - 0.000925143 - 0.000869867 - 0.000801371 - 0.000723355 - 0.000639744 - 0.000554367 - 0.00047068 - 0.000391553 - 0.000319149 - 0.000254878 - 0.000199438 - 0.000152904 - 0.00011486 - 8.45387e-05 - 6.09647e-05 - 4.30763e-05 - 2.98219e-05 - 2.02287e-05 - 1.34443e-05 - 8.75481e-06 - 5.58588e-06 - 3.49199e-06 - 2.13892e-06 - 1.2837e-06 - 7.54953e-07 - 4.3522e-07 - 2.46288e-07 - 1.37595e-07 - 7.76206e-08 - 4.7872e-08 - 3.91206e-08 - 4.96818e-08 - 8.57782e-08 - 1.64423e-07 - 3.19802e-07 - 6.1486e-07 - 1.16078e-06 - 2.1483e-06 - 3.8961e-06 - 6.92335e-06 - 1.20543e-05 - 2.0564e-05 - 3.43723e-05 - 5.6292e-05 - 9.03278e-05 - 0.000142015 - 0.000218767 - 0.000330193 - 0.000488304 - 0.000707537 - 0.00100449 - 0.00139726 - 0.00190436 - 0.00254305 - 0.00332735 - 0.00426559 - 0.00535792 - 0.00659401 - 0.00795135 - 0.00939439 - 0.0108751 - 0.0123349 - 0.013708 - 0.0149262 - 0.0159243 - 0.016646 - 0.0170489 - 0.0171089 - 0.0168223 - 0.0162065 - 0.0152982 - 0.0141497 - 0.0128242 - 0.01139 - 0.00991533 - 0.00846323 - 0.00708799 - 0.00583302 - 0.00473027 - 0.00380108 - 0.00305813 - 0.00250808 - 0.00215444 - 0.00200017 - 0.00204972 - 0.00231011 - 0.00279091 - 0.003503 - 0.00445631 - 0.00565652 - 0.00710137 - 0.0087768 - 0.0106537 - 0.0126858 - 0.014809 - 0.0169432 - 0.0189962 - 0.0208692 - 0.0224645 - 0.0236937 - 0.0244855 - 0.0247927 - 0.0245967 - 0.0239093 - 0.0227716 - 0.0212499 - 0.0194294 - 0.0174059 - 0.0152782 - 0.0131396 - 0.0110721 - 0.00914149 - 0.00739501 - 0.00586135 - 0.00455191 - 0.00346359 - 0.00258224 - 0.00188626 - 0.00135004 - 0.000946729 - 0.000650493 - 0.000437923 - 0.000288862 - 0.000186692 - 0.000118227 - 7.33688e-05 - 4.46338e-05 - 2.66504e-05 - 1.56821e-05 - 9.21728e-06 - 5.63909e-06 - 3.97946e-06 - 3.75189e-06 - 4.85588e-06 - 7.54564e-06 - 1.24576e-05 - 2.06929e-05 - 3.39515e-05 - 5.47126e-05 - 8.64515e-05 - 0.000133876 - 0.000203149 - 0.000302064 - 0.000440104 - 0.00062834 - 0.000879078 - 0.00120523 - 0.00161937 - 0.00213248 - 0.00275244 - 0.00348253 - 0.00431992 - 0.00525458 - 0.00626873 - 0.00733715 - 0.00842831 - 0.00950652 - 0.0105347 - 0.0114778 - 0.0123057 - 0.0129966 - 0.0135384 - 0.0139296 - 0.0141784 - 0.0143005 - 0.0143162 - 0.0142468 - 0.0141109 - 0.0139217 - 0.0136857 - 0.0134019 - 0.0130632 - 0.012659 - 0.0121776 - 0.0116098 - 0.0109513 - 0.0102046 - 0.00937978 - 0.00849391 - 0.00756962 - 0.00663291 - 0.00571057 - 0.00482772 - 0.00400577 - 0.00326099 - 0.00260377 - 0.00203868 - 0.00156499 - 0.00117773 - 0.000868844 - 0.000628451 - 0.000445965 - 0.00031104 - 0.00021428 - 0.000147749 - 0.000105306 - 8.28571e-05 - 7.85795e-05 - 9.31949e-05 - 0.000130355 - 0.000197161 - 0.00030483 - 0.000469447 - 0.000712728 - 0.00106262 - 0.00155357 - 0.00222613 - 0.00312575 - 0.00430043 - 0.00579713 - 0.0076569 - 0.00990902 - 0.0125645 - 0.0156098 - 0.0190014 - 0.0226627 - 0.0264835 - 0.0303231 - 0.0340181 - 0.0373924 - 0.0402711 - 0.0424952 - 0.0439363 - 0.0445086 - 0.0441776 - 0.0429631 - 0.040938 - 0.0382204 - 0.0349623 - 0.0313358 - 0.0275182 - 0.0236774 - 0.0199612 - 0.0164883 - 0.0133445 - 0.0105819 - 0.00822174 - 0.00625893 - 0.00466846 - 0.0034118 - 0.00244304 - 0.00171401 - 0.00117824 - 0.000793582 - 0.000523705 - 0.000338624 - 0.000214529 - 0.000133165 - 8.09901e-05 - 4.82626e-05 - 2.8179e-05 - 1.61205e-05 - 9.03581e-06 - 4.96241e-06 - 2.67027e-06 - 1.40785e-06 - 7.27263e-07 - 3.68099e-07 - 1.82547e-07 - 8.87001e-08 - 4.22295e-08 - 1.97008e-08 - 9.00956e-09 - 4.04833e-09 - 1.81029e-09 - 8.60698e-10 - 5.58336e-10 - 6.85971e-10 - 1.31585e-09 - 2.8408e-09 - 6.1788e-09 - 1.32343e-08 - 2.77987e-08 - 5.7221e-08 - 1.15408e-07 - 2.28062e-07 - 4.4158e-07 - 8.37727e-07 - 1.55716e-06 - 2.83595e-06 - 5.0606e-06 - 8.84794e-06 - 1.51572e-05 - 2.5441e-05 - 4.18393e-05 - 6.74174e-05 - 0.000106438 - 0.000164648 - 0.000249549 - 0.000370588 - 0.000539216 - 0.000768726 - 0.00107378 - 0.0014696 - 0.00197069 - 0.00258926 - 0.00333325 - 0.00420434 - 0.00519594 - 0.0062917 - 0.00746463 - 0.00867732 - 0.00988325 - 0.0110294 - 0.0120597 - 0.01292 - 0.013562 - 0.0139483 - 0.0140559 - 0.0138781 - 0.0134257 - 0.0127258 - 0.0118186 - 0.0107544 - 0.00958832 - 0.00837598 - 0.00716911 - 0.00601218 - 0.0049401 - 0.00397719 - 0.00313728 - 0.00242475 - 0.00183619 - 0.00136241 - 0.000990447 - 0.000705493 - 0.00049237 - 0.000336687 - 0.000225579 - 0.000148084 - 9.52474e-05 - 6.00255e-05 - 3.70643e-05 - 2.24239e-05 - 1.32924e-05 - 7.72031e-06 - 4.39341e-06 - 2.44966e-06 - 1.33828e-06 - 7.16347e-07 - 3.75697e-07 - 1.93059e-07 - 9.7204e-08 - 4.79558e-08 - 2.31886e-08 - 1.10053e-08 - 5.16537e-09 - 2.49231e-09 - 1.45573e-09 - 1.44813e-09 - 2.49265e-09 - 5.27515e-09 - 1.15271e-08 - 2.49327e-08 - 5.29399e-08 - 1.10176e-07 - 2.24677e-07 - 4.48922e-07 - 8.78862e-07 - 1.68581e-06 - 3.16834e-06 - 5.83436e-06 - 1.05267e-05 - 1.86091e-05 - 3.22327e-05 - 5.47023e-05 - 9.09601e-05 - 0.000148195 - 0.000236566 - 0.000370005 - 0.000567022 - 0.000851392 - 0.00125255 - 0.00180551 - 0.00254999 - 0.00352871 - 0.00478442 - 0.00635594 - 0.00827308 - 0.010551 - 0.0131842 - 0.0161418 - 0.0193636 - 0.0227592 - 0.02621 - 0.0295742 - 0.0326961 - 0.0354175 - 0.0375905 - 0.0390913 - 0.0398314 - 0.0397669 - 0.0389028 - 0.0372926 - 0.0350338 - 0.0322585 - 0.0291224 - 0.0257915 - 0.02243 - 0.0191902 - 0.0162049 - 0.0135829 - 0.0114083 - 0.00974183 - 0.00862375 - 0.00807727 - 0.00811156 - 0.00872333 - 0.00989679 - 0.011602 - 0.0137919 - 0.0163994 - 0.0193348 - 0.0224847 - 0.0257144 - 0.0288725 - 0.0317988 - 0.0343349 - 0.0363363 - 0.0376843 - 0.0382965 - 0.0381344 - 0.037207 - 0.0355694 - 0.0333172 - 0.0305773 - 0.0274959 - 0.0242255 - 0.020913 - 0.0176887 - 0.0146593 - 0.0119033 - 0.00947019 - 0.00738221 - 0.00563834 - 0.00421942 - 0.0030938 - 0.00222264 - 0.00156455 - 0.0010791 - 0.000729329 - 0.000483146 - 0.000313949 - 0.000200581 - 0.000126914 - 8.12404e-05 - 5.5656e-05 - 4.55669e-05 - 4.94212e-05 - 6.87538e-05 - 0.000108612 - 0.000178427 - 0.000293379 - 0.000476297 - 0.000760061 - 0.00119043 - 0.00182912 - 0.00275668 - 0.00407487 - 0.00590754 - 0.00839955 - 0.0117125 - 0.0160171 - 0.0214803 - 0.0282497 - 0.0364326 - 0.0460744 - 0.0571362 - 0.0694755 - 0.0828344 - 0.0968356 - 0.110993 - 0.124731 - 0.137426 - 0.148443 - 0.157195 - 0.163191 - 0.16608 - 0.16569 - 0.162039 - 0.155338 - 0.145969 - 0.13445 - 0.121386 - 0.107417 - 0.0931679 - 0.0792032 - 0.0659922 - 0.0538902 - 0.0431307 - 0.0338313 - 0.0260076 - 0.0195944 - 0.0144686 - 0.0104718 - 0.00743065 - 0.00517275 - 0.00353869 - 0.0023892 - 0.00160904 - 0.00110793 - 0.000819713 - 0.000700487 - 0.000726235 - 0.000890471 - 0.00120196 - 0.00168246 - 0.00236436 - 0.00328777 - 0.00449696 - 0.00603582 - 0.00794235 - 0.0102424 - 0.0129428 - 0.0160253 - 0.0194414 - 0.0231094 - 0.0269146 - 0.0307131 - 0.0343397 - 0.0376189 - 0.0403787 - 0.0424655 - 0.0437579 - 0.0441788 - 0.0437027 - 0.0423583 - 0.040226 - 0.0374293 - 0.0341234 - 0.0304811 - 0.0266776 - 0.022877 - 0.0192215 - 0.0158239 - 0.0127637 - 0.0100873 - 0.00781109 - 0.00592634 - 0.00440555 - 0.00320891 - 0.0022902 - 0.00160171 - 0.00109801 - 0.000738346 - 0.000488074 - 0.000319114 - 0.000209886 - 0.000144899 - 0.000114287 - 0.00011345 - 0.000142967 - 0.000208865 - 0.000323272 - 0.000505441 - 0.000783052 - 0.00119362 - 0.00178573 - 0.00261982 - 0.00376788 - 0.00531191 - 0.00734032 - 0.0099422 - 0.0131993 - 0.0171757 - 0.0219067 - 0.0273861 - 0.0335566 - 0.0403008 - 0.0474392 - 0.0547326 - 0.0618925 - 0.0685978 - 0.074518 - 0.0793391 - 0.0827917 - 0.0846753 - 0.084878 - 0.0833871 - 0.0802907 - 0.0757691 - 0.0700771 - 0.0635207 - 0.0564296 - 0.0491301 - 0.0419213 - 0.0350564 - 0.0287303 - 0.0230756 - 0.0181636 - 0.0140114 - 0.0105924 - 0.00784763 - 0.00569779 - 0.00405413 - 0.0028269 - 0.00193171 - 0.00129357 - 0.000848892 - 0.000545916 - 0.00034404 - 0.000212471 - 0.000128587 - 7.62598e-05 - 4.43196e-05 - 2.52403e-05 - 1.40861e-05 - 7.70341e-06 - 4.12826e-06 - 2.16792e-06 - 1.1156e-06 - 5.62553e-07 - 2.77973e-07 - 1.34595e-07 - 6.38609e-08 - 2.96909e-08 - 1.35267e-08 - 6.03865e-09 - 2.64158e-09 - 1.13231e-09 - 4.75596e-10 - 1.95742e-10 - 7.89412e-11 - 3.11955e-11 - 1.20796e-11 - 4.58331e-12 - 1.70407e-12 - 6.20988e-13 - 2.22293e-13 - 7.9713e-14 - 3.33658e-14 - 2.94752e-14 - 6.38018e-14 - 1.77569e-13 - 5.02538e-13 - 1.39959e-12 - 3.82147e-12 - 1.02254e-11 - 2.6812e-11 - 6.88931e-11 - 1.73468e-10 - 4.28014e-10 - 1.03489e-09 - 2.45205e-09 - 5.69327e-09 - 1.29536e-08 - 2.88814e-08 - 6.31019e-08 - 1.35103e-07 - 2.83456e-07 - 5.82778e-07 - 1.17414e-06 - 2.3181e-06 - 4.48481e-06 - 8.50263e-06 - 1.57965e-05 - 2.87584e-05 - 5.13057e-05 - 8.96943e-05 - 0.00015366 - 0.000257961 - 0.00042437 - 0.000684119 - 0.00108072 - 0.001673 - 0.00253788 - 0.00377261 - 0.00549553 - 0.00784462 - 0.0109731 - 0.0150412 - 0.0202036 - 0.0265931 - 0.0343008 - 0.0433543 - 0.0536975 - 0.0651733 - 0.0775138 - 0.0903404 - 0.103176 - 0.115472 - 0.12664 - 0.136103 - 0.143344 - 0.147949 - 0.149655 - 0.14837 - 0.144186 - 0.137373 - 0.128353 - 0.117658 - 0.105889 - 0.093658 - 0.0815473 - 0.0700646 - 0.0596157 - 0.0504871 - 0.0428422 - 0.0367268 - 0.0320839 - 0.0287735 - 0.0265961 - 0.0253153 - 0.0246804 - 0.024446 - 0.0243882 - 0.0243168 - 0.0240838 - 0.0235875 - 0.0227725 - 0.021627 - 0.0201755 - 0.0184715 - 0.0165872 - 0.014604 - 0.0126035 - 0.0106603 - 0.00883602 - 0.00717679 - 0.00571177 - 0.00445418 - 0.00340341 - 0.00254803 - 0.00186912 - 0.00134342 - 0.000946069 - 0.000652789 - 0.000441327 - 0.000292338 - 0.000189735 - 0.000120655 - 7.51766e-05 - 4.5894e-05 - 2.74515e-05 - 1.60884e-05 - 9.2384e-06 - 5.19778e-06 - 2.86533e-06 - 1.54764e-06 - 8.19038e-07 - 4.24702e-07 - 2.15801e-07 - 1.07503e-07 - 5.26328e-08 - 2.56405e-08 - 1.31726e-08 - 8.77456e-09 - 1.02078e-08 - 1.85691e-08 - 3.89947e-08 - 8.33138e-08 - 1.75745e-07 - 3.63781e-07 - 7.38012e-07 - 1.46706e-06 - 2.85743e-06 - 5.45305e-06 - 1.01963e-05 - 1.86801e-05 - 3.35317e-05 - 5.89749e-05 - 0.000101629 - 0.000171594 - 0.000283873 - 0.000460134 - 0.00073077 - 0.00113714 - 0.00173374 - 0.00258995 - 0.00379084 - 0.00543645 - 0.00763894 - 0.0105169 - 0.0141866 - 0.0187503 - 0.0242813 - 0.0308088 - 0.0383013 - 0.0466542 - 0.0556807 - 0.0651114 - 0.0746018 - 0.0837495 - 0.0921213 - 0.0992861 - 0.104852 - 0.108503 - 0.110031 - 0.109357 - 0.106543 - 0.101791 - 0.0954271 - 0.087877 - 0.0796374 - 0.0712426 - 0.0632336 - 0.0561318 - 0.0504187 - 0.0465218 - 0.044805 - 0.045561 - 0.0490042 - 0.0552601 - 0.0643534 - 0.0761915 - 0.0905495 - 0.107057 - 0.125194 - 0.144296 - 0.163577 - 0.182163 - 0.199146 - 0.213648 - 0.224879 - 0.232211 - 0.235225 - 0.233747 - 0.227864 - 0.217912 - 0.204443 - 0.188177 - 0.169934 - 0.150572 - 0.130918 - 0.111717 - 0.0935927 - 0.0770224 - 0.0623326 - 0.0497065 - 0.0392039 - 0.0307866 - 0.0243474 - 0.0197367 - 0.0167861 - 0.0153248 - 0.0151908 - 0.0162352 - 0.0183228 - 0.0213286 - 0.0251345 - 0.0296256 - 0.0346886 - 0.0402135 - 0.0460961 - 0.052243 - 0.0585756 - 0.0650326 - 0.0715684 - 0.0781472 - 0.0847335 - 0.091278 - 0.0977038 - 0.103894 - 0.109684 - 0.114865 - 0.119189 - 0.122391 - 0.124217 - 0.124448 - 0.122933 - 0.119609 - 0.114518 - 0.107811 - 0.0997331 - 0.0906066 - 0.0808031 - 0.0707102 - 0.0607003 - 0.0511033 - 0.0421863 - 0.0341421 - 0.0270865 - 0.0210628 - 0.0160525 - 0.0119898 - 0.00877597 - 0.00629478 - 0.00442438 - 0.00304721 - 0.00205646 - 0.00135988 - 0.000881132 - 0.000559434 - 0.000348064 - 0.000212277 - 0.000127044 - 7.49014e-05 - 4.40862e-05 - 2.70444e-05 - 1.93388e-05 - 1.89574e-05 - 2.60199e-05 - 4.28955e-05 - 7.47722e-05 - 0.00013075 - 0.000225561 - 0.000382019 - 0.000634315 - 0.00103218 - 0.00164583 - 0.00257148 - 0.00393681 - 0.00590566 - 0.00868071 - 0.0125027 - 0.0176447 - 0.0243998 - 0.0330614 - 0.0438952 - 0.0571052 - 0.0727939 - 0.0909236 - 0.111281 - 0.133452 - 0.156816 - 0.180558 - 0.203706 - 0.225193 - 0.24393 - 0.258902 - 0.269258 - 0.274385 - 0.273977 - 0.268058 - 0.256982 - 0.241401 - 0.222195 - 0.200397 - 0.177096 - 0.153352 - 0.130119 - 0.108184 - 0.088143 - 0.0703811 - 0.0550906 - 0.0422952 - 0.0318886 - 0.0236757 - 0.0174137 - 0.0128489 - 0.00974575 - 0.00790899 - 0.00719695 - 0.00752776 - 0.00887847 - 0.0112782 - 0.0147954 - 0.0195211 - 0.0255467 - 0.03294 - 0.0417194 - 0.0518294 - 0.0631202 - 0.0753344 - 0.0881046 - 0.100962 - 0.113362 - 0.124714 - 0.134431 - 0.141978 - 0.146921 - 0.148964 - 0.147984 - 0.144041 - 0.137371 - 0.128363 - 0.117523 - 0.105425 - 0.0926619 - 0.0797996 - 0.0673356 - 0.0556729 - 0.0451043 - 0.0358108 - 0.0278695 - 0.0212705 - 0.0159377 - 0.0117507 - 0.0085666 - 0.0062372 - 0.00462324 - 0.00360352 - 0.00307992 - 0.00297862 - 0.00324833 - 0.00385612 - 0.00478184 - 0.00601151 - 0.00753051 - 0.0093171 - 0.011337 - 0.0135396 - 0.0158561 - 0.0182006 - 0.0204731 - 0.0225658 - 0.0243709 - 0.0257892 - 0.0267388 - 0.0271636 - 0.0270377 - 0.0263689 - 0.0251973 - 0.0235919 - 0.0216433 - 0.019456 - 0.0171389 - 0.014797 - 0.0125243 - 0.0103983 - 0.00847798 - 0.00680284 - 0.00539503 - 0.00426232 - 0.00340205 - 0.00280523 - 0.00246013 - 0.00235501 - 0.00247968 - 0.00282577 - 0.00338583 - 0.00415139 - 0.00511045 - 0.00624471 - 0.00752714 - 0.00892027 - 0.0103758 - 0.0118353 - 0.0132334 - 0.0145009 - 0.0155705 - 0.0163822 - 0.0168885 - 0.017059 - 0.0168832 - 0.0163718 - 0.0155551 - 0.0144807 - 0.0132081 - 0.011804 - 0.0103361 - 0.00886787 - 0.00745451 - 0.00613983 - 0.00495484 - 0.00391778 - 0.0030352 - 0.00230394 - 0.00171353 - 0.00124868 - 0.000891548 - 0.000623701 - 0.000427509 - 0.000287111 - 0.000188926 - 0.000121807 - 7.69461e-05 - 4.76255e-05 - 2.88821e-05 - 1.71615e-05 - 9.99121e-06 - 5.69926e-06 - 3.18533e-06 - 1.74433e-06 - 9.35921e-07 - 4.92024e-07 - 2.53438e-07 - 1.27906e-07 - 6.32485e-08 - 3.06439e-08 - 1.45471e-08 - 6.7662e-09 - 3.08354e-09 - 1.37687e-09 - 6.02381e-10 - 2.58218e-10 - 1.08453e-10 - 4.46303e-11 - 1.79952e-11 - 7.10917e-12 - 2.75181e-12 - 1.04365e-12 - 3.8782e-13 - 1.41202e-13 - 5.03719e-14 - 1.76065e-14 - 6.02967e-15 - 2.02326e-15 - 6.65191e-16 - 2.14278e-16 - 6.76311e-17 - 2.09147e-17 - 6.33715e-18 - 1.88137e-18 - 5.47254e-19 - 1.5597e-19 - 4.35544e-20 - 1.19168e-20 - 3.19464e-21 - 8.39116e-22 - 2.15953e-22 - 5.44544e-23 - 1.34538e-23 - 3.2568e-24 - 7.72459e-25 - 1.79513e-25 - 4.08747e-26 - 9.11906e-27 - 1.99334e-27 - 4.26924e-28 - 8.95892e-29 - 1.84204e-29 - 3.71088e-30 - 7.32476e-31 - 1.4166e-31 - 2.68433e-32 - 4.98382e-33 - 9.06622e-34 - 1.61594e-34 - 2.82204e-35 - 4.82878e-36 - 8.09559e-37 - 1.32983e-37 - 2.14033e-38 - 3.37522e-39 - 5.21507e-40 - 7.89506e-41 - 1.17108e-41 - 1.70199e-42 - 2.4236e-43 - 3.38146e-44 - 4.62257e-45 - 6.19155e-46 - 8.12554e-47 - 1.04482e-47 - 1.31634e-48 - 1.62492e-49 - 1.96532e-50 - 2.329e-51 - 2.70423e-52 - 3.07649e-53 - 3.42928e-54 - 3.74531e-55 - 4.00783e-56 - 4.20211e-57 - 4.31681e-58 - 4.34505e-59 - 4.28512e-60 - 4.14066e-61 - 3.92023e-62 - 3.63656e-63 - 3.30527e-64 - 2.94348e-65 - 2.56833e-66 - 2.19572e-67 - 1.83925e-68 - 1.50953e-69 - 1.21389e-70 - 9.56429e-72 - 7.38353e-73 - 5.58485e-74 - 4.13901e-75 - 3.00551e-76 - 2.13834e-77 - 1.49064e-78 - 1.01813e-79 - 6.81354e-81 - 4.46764e-82 - 2.87026e-83 - 1.80676e-84 - 1.11434e-85 - 6.73396e-87 - 3.98713e-88 - 2.31306e-89 - 1.31477e-90 - 7.32237e-92 - 3.99566e-93 - 2.1363e-94 - 1.11911e-95 - 5.7441e-97 - 2.88873e-98 - 1.42341e-99 - 6.87207e-101 - 3.25074e-102 - 1.50666e-103 - 6.84201e-105 - 3.04432e-106 - 1.32719e-107 - 5.66909e-109 - 2.37623e-110 - 1.87428e-111 - 2.20949e-110 - 5.28801e-109 - 1.24221e-107 - 2.85914e-106 - 6.44781e-105 - 1.42471e-103 - 3.08445e-102 - 6.54282e-101 - 1.35985e-99 - 2.76918e-98 - 5.52523e-97 - 1.08015e-95 - 2.06899e-94 - 3.88299e-93 - 7.14023e-92 - 1.28646e-90 - 2.27099e-89 - 3.92799e-88 - 6.65678e-87 - 1.10534e-85 - 1.79829e-84 - 2.86658e-83 - 4.47719e-82 - 6.85145e-81 - 1.0273e-79 - 1.5092e-78 - 2.17238e-77 - 3.0638e-76 - 4.23372e-75 - 5.7322e-74 - 7.60425e-73 - 9.88391e-72 - 1.25875e-70 - 1.57067e-69 - 1.92029e-68 - 2.30031e-67 - 2.69987e-66 - 3.10482e-65 - 3.49837e-64 - 3.86218e-63 - 4.1777e-62 - 4.42769e-61 - 4.59785e-60 - 4.6781e-59 - 4.66359e-58 - 4.55521e-57 - 4.35947e-56 - 4.08786e-55 - 3.75573e-54 - 3.38088e-53 - 2.98196e-52 - 2.57698e-51 - 2.18201e-50 - 1.81026e-49 - 1.4715e-48 - 1.17197e-47 - 9.14556e-47 - 6.99263e-46 - 5.23851e-45 - 3.84514e-44 - 2.76537e-43 - 1.94863e-42 - 1.34538e-41 - 9.10115e-41 - 6.03233e-40 - 3.91751e-39 - 2.49271e-38 - 1.55407e-37 - 9.49308e-37 - 5.68172e-36 - 3.33188e-35 - 1.91442e-34 - 1.07775e-33 - 5.94484e-33 - 3.2129e-32 - 1.70134e-31 - 8.8272e-31 - 4.48736e-30 - 2.2351e-29 - 1.09078e-28 - 5.21577e-28 - 2.44362e-27 - 1.12173e-26 - 5.0452e-26 - 2.22334e-25 - 9.6e-25 - 4.06138e-24 - 1.6835e-23 - 6.83739e-23 - 2.72085e-22 - 1.06086e-21 - 4.05272e-21 - 1.51696e-20 - 5.56339e-20 - 1.99914e-19 - 7.03857e-19 - 2.42809e-18 - 8.20698e-18 - 2.71795e-17 - 8.81938e-17 - 2.80398e-16 - 8.73476e-16 - 2.66605e-15 - 7.97306e-15 - 2.33627e-14 - 6.70756e-14 - 1.8869e-13 - 5.20086e-13 - 1.40458e-12 - 3.71676e-12 - 9.63669e-12 - 2.44816e-11 - 6.09399e-11 - 1.48633e-10 - 3.55208e-10 - 8.31775e-10 - 1.90848e-09 - 4.29072e-09 - 9.45234e-09 - 2.04041e-08 - 4.31586e-08 - 8.94536e-08 - 1.81682e-07 - 3.61591e-07 - 7.05214e-07 - 1.34782e-06 - 2.52441e-06 - 4.63355e-06 - 8.33507e-06 - 1.46946e-05 - 2.53909e-05 - 4.30015e-05 - 7.13836e-05 - 0.000116157 - 0.000185287 - 0.000289756 - 0.000444259 - 0.000667872 - 0.000984563 - 0.00142341 - 0.00201839 - 0.00280749 - 0.00383118 - 0.00512992 - 0.00674098 - 0.0086945 - 0.0110092 - 0.013688 - 0.0167144 - 0.020049 - 0.0236291 - 0.0273681 - 0.0311587 - 0.0348773 - 0.0383902 - 0.0415613 - 0.0442609 - 0.0463735 - 0.0478061 - 0.0484944 - 0.0484071 - 0.0475482 - 0.0459567 - 0.043704 - 0.0408888 - 0.0376304 - 0.0340605 - 0.0303155 - 0.0265272 - 0.022816 - 0.0192849 - 0.016015 - 0.013064 - - - - - 0.0267289 - 0.0332634 - 0.0405593 - 0.0484563 - 0.0567214 - 0.065055 - 0.0731057 - 0.080493 - 0.0868365 - 0.0917875 - 0.0950608 - 0.096462 - 0.0959064 - 0.0934278 - 0.0891746 - 0.0833957 - 0.0764157 - 0.0686054 - 0.0603491 - 0.052014 - 0.0439245 - 0.0363438 - 0.0294639 - 0.0234038 - 0.0182146 - 0.0138896 - 0.0103776 - 0.00759702 - 0.00544909 - 0.0038295 - 0.00263692 - 0.00177905 - 0.00117603 - 0.000761697 - 0.000483376 - 0.000300555 - 0.000183105 - 0.000109298 - 6.39237e-05 - 3.6631e-05 - 2.0567e-05 - 1.13144e-05 - 6.09862e-06 - 3.2209e-06 - 1.66689e-06 - 8.45673e-07 - 4.21487e-07 - 2.08527e-07 - 1.07482e-07 - 6.89524e-08 - 7.43322e-08 - 1.29008e-07 - 2.66381e-07 - 5.65012e-07 - 1.1859e-06 - 2.44369e-06 - 4.93578e-06 - 9.76872e-06 - 1.89436e-05 - 3.59937e-05 - 6.70081e-05 - 0.000122226 - 0.000218444 - 0.000382517 - 0.000656295 - 0.00110328 - 0.00181721 - 0.00293268 - 0.00463725 - 0.00718444 - 0.0109059 - 0.0162207 - 0.023638 - 0.0337514 - 0.047218 - 0.0647234 - 0.0869264 - 0.114388 - 0.147483 - 0.186313 - 0.230612 - 0.279677 - 0.332329 - 0.386917 - 0.441372 - 0.49332 - 0.540247 - 0.579689 - 0.609454 - 0.627819 - 0.633699 - 0.626759 - 0.607453 - 0.576985 - 0.537197 - 0.49041 - 0.439226 - 0.386321 - 0.334265 - 0.285366 - 0.241567 - 0.204394 - 0.17494 - 0.153888 - 0.141548 - 0.137899 - 0.142625 - 0.15514 - 0.174596 - 0.199889 - 0.229665 - 0.262341 - 0.296142 - 0.329174 - 0.359519 - 0.385351 - 0.405065 - 0.417403 - 0.421551 - 0.417208 - 0.404606 - 0.384479 - 0.357983 - 0.326585 - 0.291924 - 0.255673 - 0.2194 - 0.18447 - 0.151968 - 0.122664 - 0.0970102 - 0.0751718 - 0.0570729 - 0.0424562 - 0.0309449 - 0.0220991 - 0.0154631 - 0.0106012 - 0.00712117 - 0.00468688 - 0.00302241 - 0.00190968 - 0.00118223 - 0.000717106 - 0.000426187 - 0.000248173 - 0.000141594 - 7.91538e-05 - 4.33547e-05 - 2.32668e-05 - 1.22342e-05 - 6.30304e-06 - 3.18172e-06 - 1.57366e-06 - 7.62598e-07 - 3.62091e-07 - 1.68452e-07 - 7.67844e-08 - 3.4293e-08 - 1.50064e-08 - 6.43402e-09 - 2.70292e-09 - 1.11268e-09 - 4.49155e-10 - 1.78704e-10 - 7.26516e-11 - 3.71681e-11 - 3.99254e-11 - 8.40618e-11 - 2.11363e-10 - 5.37488e-10 - 1.34516e-09 - 3.30051e-09 - 7.93528e-09 - 1.86932e-08 - 4.31463e-08 - 9.75755e-08 - 2.16209e-07 - 4.69402e-07 - 9.98509e-07 - 2.08112e-06 - 4.24988e-06 - 8.50344e-06 - 1.66705e-05 - 3.20215e-05 - 6.02656e-05 - 0.000111131 - 0.000200787 - 0.000355448 - 0.000616526 - 0.00104776 - 0.00174467 - 0.00284642 - 0.00455012 - 0.0071266 - 0.0109365 - 0.0164442 - 0.0242261 - 0.0349696 - 0.0494578 - 0.0685356 - 0.0930538 - 0.123791 - 0.161354 - 0.206067 - 0.257854 - 0.316138 - 0.379765 - 0.446983 - 0.515471 - 0.582443 - 0.644822 - 0.69946 - 0.743401 - 0.774142 - 0.789869 - 0.789636 - 0.773458 - 0.742311 - 0.698034 - 0.643151 - 0.580637 - 0.513651 - 0.445287 - 0.378343 - 0.315168 - 0.257556 - 0.206723 - 0.163341 - 0.127614 - 0.0993871 - 0.0782617 - 0.0637042 - 0.0551354 - 0.0519919 - 0.0537573 - 0.0599654 - 0.0701785 - 0.0839509 - 0.100784 - 0.120087 - 0.141144 - 0.163108 - 0.185011 - 0.2058 - 0.224405 - 0.239804 - 0.251113 - 0.257659 - 0.259043 - 0.25518 - 0.246301 - 0.232936 - 0.21586 - 0.196018 - 0.17445 - 0.152199 - 0.130247 - 0.109461 - 0.0905612 - 0.0741229 - 0.0605995 - 0.050366 - 0.0437782 - 0.0412372 - 0.0432522 - 0.0504919 - 0.0638147 - 0.0842689 - 0.113056 - 0.151449 - 0.200672 - 0.261733 - 0.33523 - 0.421147 - 0.518667 - 0.626021 - 0.740421 - 0.858084 - 0.974384 - 1.08411 - 1.18183 - 1.26234 - 1.3211 - 1.35466 - 1.36101 - 1.33977 - 1.29222 - 1.22118 - 1.13073 - 1.02583 - 0.911863 - 0.794182 - 0.677715 - 0.566645 - 0.464208 - 0.372606 - 0.293038 - 0.225806 - 0.170484 - 0.126116 - 0.0914095 - 0.0649157 - 0.0451695 - 0.0307949 - 0.0205706 - 0.0134634 - 0.0086337 - 0.00542471 - 0.00333959 - 0.00201441 - 0.00119052 - 0.00068939 - 0.000391137 - 0.000217435 - 0.000118432 - 6.32037e-05 - 3.30487e-05 - 1.69318e-05 - 8.49939e-06 - 4.18032e-06 - 2.01451e-06 - 9.51183e-07 - 4.40045e-07 - 1.99465e-07 - 8.85875e-08 - 3.85492e-08 - 1.6436e-08 - 6.86614e-09 - 2.81039e-09 - 1.12709e-09 - 4.42878e-10 - 1.70509e-10 - 6.43205e-11 - 2.37732e-11 - 8.60919e-12 - 3.05474e-12 - 1.06199e-12 - 3.61749e-13 - 1.20734e-13 - 3.94811e-14 - 1.26498e-14 - 3.97116e-15 - 1.22148e-15 - 3.68123e-16 - 1.08702e-16 - 3.14497e-17 - 8.91525e-18 - 2.47621e-18 - 6.73873e-19 - 1.79682e-19 - 4.6943e-20 - 1.20168e-20 - 3.01577e-21 - 7.49272e-22 - 2.14954e-22 - 1.93722e-22 - 6.24827e-22 - 2.44663e-21 - 9.5184e-21 - 3.63116e-20 - 1.35732e-19 - 4.97117e-19 - 1.7839e-18 - 6.27221e-18 - 2.16076e-17 - 7.2934e-17 - 2.41207e-16 - 7.81604e-16 - 2.48154e-15 - 7.71954e-15 - 2.35287e-14 - 7.02656e-14 - 2.056e-13 - 5.89443e-13 - 1.65576e-12 - 4.5571e-12 - 1.2289e-11 - 3.247e-11 - 8.4059e-11 - 2.13218e-10 - 5.29907e-10 - 1.29036e-09 - 3.07866e-09 - 7.19694e-09 - 1.64843e-08 - 3.6994e-08 - 8.13445e-08 - 1.75252e-07 - 3.69941e-07 - 7.65139e-07 - 1.55055e-06 - 3.07869e-06 - 5.98941e-06 - 1.14167e-05 - 2.13221e-05 - 3.90175e-05 - 6.99559e-05 - 0.000122893 - 0.000211527 - 0.000356732 - 0.000589461 - 0.000954344 - 0.00151388 - 0.00235297 - 0.00358324 - 0.00534656 - 0.00781646 - 0.0111965 - 0.0157143 - 0.0216095 - 0.0291162 - 0.0384384 - 0.0497211 - 0.063018 - 0.0782604 - 0.095232 - 0.113554 - 0.132684 - 0.15194 - 0.170533 - 0.187635 - 0.202445 - 0.214279 - 0.222647 - 0.227328 - 0.228424 - 0.226381 - 0.221989 - 0.216339 - 0.210757 - 0.206705 - 0.205663 - 0.209006 - 0.217869 - 0.233034 - 0.254818 - 0.283009 - 0.316827 - 0.354937 - 0.395514 - 0.43635 - 0.475018 - 0.509059 - 0.536184 - 0.554477 - 0.562561 - 0.559721 - 0.545954 - 0.52196 - 0.489056 - 0.44904 - 0.404009 - 0.356175 - 0.307674 - 0.260416 - 0.215969 - 0.175494 - 0.139728 - 0.10901 - 0.0833381 - 0.0624435 - 0.0458741 - 0.0330733 - 0.0234486 - 0.0164261 - 0.011488 - 0.00819575 - 0.00620026 - 0.00524187 - 0.00514359 - 0.00579949 - 0.00716005 - 0.00921593 - 0.0119807 - 0.0154733 - 0.0197007 - 0.024642 - 0.0302351 - 0.0363664 - 0.0428667 - 0.0495127 - 0.0560361 - 0.0621389 - 0.0675148 - 0.0718742 - 0.0749694 - 0.0766183 - 0.0767216 - 0.0752731 - 0.07236 - 0.0681545 - 0.0628966 - 0.0568718 - 0.0503852 - 0.0437368 - 0.0371986 - 0.0309987 - 0.0253103 - 0.0202483 - 0.0158714 - 0.0121893 - 0.00917237 - 0.00676269 - 0.00488534 - 0.00345785 - 0.00239803 - 0.00162945 - 0.00108483 - 0.000707655 - 0.000452291 - 0.000283237 - 0.000173788 - 0.000104478 - 6.15418e-05 - 3.55182e-05 - 2.0085e-05 - 1.11286e-05 - 6.04223e-06 - 3.21611e-06 - 1.68155e-06 - 8.715e-07 - 4.65702e-07 - 2.95409e-07 - 2.90595e-07 - 4.59571e-07 - 8.98517e-07 - 1.83681e-06 - 3.73474e-06 - 7.46497e-06 - 1.463e-05 - 2.80973e-05 - 5.28732e-05 - 9.74872e-05 - 0.000176115 - 0.000311734 - 0.000540638 - 0.000918686 - 0.00152955 - 0.00249516 - 0.00398812 - 0.00624563 - 0.00958342 - 0.0144079 - 0.0212236 - 0.0306319 - 0.0433178 - 0.0600198 - 0.0814817 - 0.108383 - 0.141254 - 0.180376 - 0.225679 - 0.276656 - 0.332297 - 0.391066 - 0.450931 - 0.509457 - 0.563951 - 0.611663 - 0.65001 - 0.676806 - 0.690471 - 0.690182 - 0.675957 - 0.64865 - 0.609873 - 0.56183 - 0.507116 - 0.448484 - 0.388618 - 0.329941 - 0.274464 - 0.223703 - 0.178647 - 0.139784 - 0.107165 - 0.0804984 - 0.0592459 - 0.0427234 - 0.0301863 - 0.0208974 - 0.0141746 - 0.00942031 - 0.00613419 - 0.00391369 - 0.00244654 - 0.00149849 - 0.000899277 - 0.000528773 - 0.000304637 - 0.000171962 - 9.51085e-05 - 5.15398e-05 - 2.73655e-05 - 1.42364e-05 - 7.25663e-06 - 3.62415e-06 - 1.77343e-06 - 8.50273e-07 - 3.99429e-07 - 1.83848e-07 - 8.29113e-08 - 3.66359e-08 - 1.58614e-08 - 6.72902e-09 - 2.79867e-09 - 1.14499e-09 - 4.71141e-10 - 2.22004e-10 - 1.83107e-10 - 3.17921e-10 - 7.37346e-10 - 1.77917e-09 - 4.24558e-09 - 9.94017e-09 - 2.28074e-08 - 5.12753e-08 - 1.12948e-07 - 2.43774e-07 - 5.15504e-07 - 1.0681e-06 - 2.16836e-06 - 4.31308e-06 - 8.40581e-06 - 1.60513e-05 - 3.00314e-05 - 5.50526e-05 - 9.8882e-05 - 0.000174018 - 0.000300059 - 0.000506941 - 0.000839159 - 0.00136103 - 0.00216286 - 0.00336765 - 0.00513761 - 0.00767949 - 0.0112471 - 0.0161393 - 0.0226917 - 0.0312598 - 0.0421931 - 0.0558 - 0.0723041 - 0.0917971 - 0.114191 - 0.139178 - 0.166207 - 0.194474 - 0.222952 - 0.250437 - 0.275628 - 0.297224 - 0.314038 - 0.3251 - 0.329752 - 0.327715 - 0.31911 - 0.304455 - 0.284604 - 0.260674 - 0.233932 - 0.205693 - 0.177209 - 0.149585 - 0.123717 - 0.100255 - 0.079601 - 0.0619254 - 0.0472015 - 0.0352517 - 0.0257953 - 0.0184943 - 0.0129919 - 0.00894221 - 0.0060305 - 0.00398473 - 0.00257977 - 0.00163644 - 0.00101708 - 0.000619366 - 0.000369553 - 0.000216045 - 0.00012375 - 6.94522e-05 - 3.81912e-05 - 2.05767e-05 - 1.08624e-05 - 5.61842e-06 - 2.84734e-06 - 1.41384e-06 - 6.87856e-07 - 3.27893e-07 - 1.53145e-07 - 7.0083e-08 - 3.14237e-08 - 1.38051e-08 - 5.94236e-09 - 2.50619e-09 - 1.03564e-09 - 4.19311e-10 - 1.66342e-10 - 6.46556e-11 - 2.46233e-11 - 9.18805e-12 - 3.35921e-12 - 1.20334e-12 - 4.22352e-13 - 1.45244e-13 - 4.89396e-14 - 1.61569e-14 - 5.22629e-15 - 1.6564e-15 - 5.14367e-16 - 1.56502e-16 - 4.66553e-17 - 1.36276e-17 - 3.9001e-18 - 1.09363e-18 - 3.00468e-19 - 8.08844e-20 - 2.13338e-20 - 5.51326e-21 - 1.396e-21 - 3.46337e-22 - 8.4189e-23 - 2.00563e-23 - 4.70346e-24 - 1.18097e-24 - 7.36257e-25 - 2.2269e-24 - 9.33913e-24 - 3.92109e-23 - 1.6148e-22 - 6.5162e-22 - 2.57637e-21 - 9.98064e-21 - 3.78832e-20 - 1.40888e-19 - 5.13378e-19 - 1.8329e-18 - 6.41178e-18 - 2.19763e-17 - 7.38023e-17 - 2.42841e-16 - 7.82911e-16 - 2.47309e-15 - 7.65429e-15 - 2.32118e-14 - 6.89682e-14 - 2.00783e-13 - 5.72721e-13 - 1.60065e-12 - 4.38317e-12 - 1.17603e-11 - 3.09161e-11 - 7.96322e-11 - 2.0097e-10 - 4.96948e-10 - 1.20401e-09 - 2.85814e-09 - 6.64778e-09 - 1.51498e-08 - 3.38279e-08 - 7.40085e-08 - 1.58645e-07 - 3.33202e-07 - 6.85688e-07 - 1.38256e-06 - 2.73135e-06 - 5.28701e-06 - 1.00272e-05 - 1.86332e-05 - 3.39261e-05 - 6.05225e-05 - 0.000105788 - 0.000181174 - 0.000304014 - 0.000499836 - 0.000805192 - 0.00127089 - 0.00196543 - 0.00297812 - 0.00442145 - 0.0064317 - 0.00916693 - 0.0128015 - 0.017516 - 0.0234826 - 0.0308458 - 0.0396993 - 0.050062 - 0.0618546 - 0.0748813 - 0.0888206 - 0.103227 - 0.117547 - 0.131151 - 0.143377 - 0.153579 - 0.161192 - 0.165778 - 0.167072 - 0.165011 - 0.159745 - 0.15162 - 0.141158 - 0.129004 - 0.115883 - 0.10254 - 0.0896927 - 0.0779926 - 0.0679908 - 0.060121 - 0.0546894 - 0.0518725 - 0.0517191 - 0.0541544 - 0.0589851 - 0.065905 - 0.0745031 - 0.0842757 - 0.094645 - 0.104985 - 0.114654 - 0.123038 - 0.129589 - 0.13387 - 0.135586 - 0.134612 - 0.130999 - 0.124968 - 0.11689 - 0.107254 - 0.0966231 - 0.0855928 - 0.0747527 - 0.064652 - 0.0557753 - 0.0485287 - 0.0432333 - 0.0401249 - 0.0393567 - 0.0410003 - 0.0450455 - 0.0513959 - 0.059862 - 0.0701542 - 0.081879 - 0.0945419 - 0.107559 - 0.120283 - 0.132037 - 0.142159 - 0.150053 - 0.155237 - 0.157386 - 0.156357 - 0.152207 - 0.145178 - 0.13568 - 0.124242 - 0.111471 - 0.0979927 - 0.0844041 - 0.0712312 - 0.0588999 - 0.0477194 - 0.0378803 - 0.0294624 - 0.0224522 - 0.0167644 - 0.0122646 - 0.00879136 - 0.00617441 - 0.00424885 - 0.00286473 - 0.00189249 - 0.00122496 - 0.000776862 - 0.00048273 - 0.000293901 - 0.000175321 - 0.000102472 - 5.86831e-05 - 3.29274e-05 - 1.81025e-05 - 9.75118e-06 - 5.1465e-06 - 2.66136e-06 - 1.34844e-06 - 6.69419e-07 - 3.25612e-07 - 1.55182e-07 - 7.2463e-08 - 3.31535e-08 - 1.48621e-08 - 6.52778e-09 - 2.80924e-09 - 1.18454e-09 - 4.8938e-10 - 1.98098e-10 - 7.8569e-11 - 3.05323e-11 - 1.16253e-11 - 4.33696e-12 - 1.58527e-12 - 5.67752e-13 - 1.99228e-13 - 6.84982e-14 - 2.30752e-14 - 7.61636e-15 - 2.46313e-15 - 7.80482e-16 - 2.42312e-16 - 7.37099e-17 - 2.19691e-17 - 6.41558e-18 - 1.83568e-18 - 5.14628e-19 - 1.4136e-19 - 3.80451e-20 - 1.00324e-20 - 2.5921e-21 - 6.56195e-22 - 1.62761e-22 - 3.95554e-23 - 9.41886e-24 - 2.19749e-24 - 5.02334e-25 - 1.12511e-25 - 2.46908e-26 - 5.30898e-27 - 1.11847e-27 - 2.30873e-28 - 4.66923e-29 - 9.24985e-30 - 1.79065e-30 - 3.31035e-31 - 4.46166e-32 - -2.16014e-32 - -5.46167e-32 - -9.44055e-32 - -1.55144e-31 - -2.49039e-31 - -3.91558e-31 - -6.0318e-31 - -9.104e-31 - -1.34634e-30 - -1.95081e-30 - -2.76956e-30 - -3.8525e-30 - -5.25065e-30 - -7.01164e-30 - -9.17409e-30 - -1.1761e-29 - -1.47727e-29 - -1.81809e-29 - -2.19233e-29 - -2.5902e-29 - -2.99846e-29 - -3.40094e-29 - -3.77953e-29 - -4.11541e-29 - -4.39062e-29 - -4.58961e-29 - -4.70071e-29 - -4.71727e-29 - -4.63831e-29 - -4.46864e-29 - -4.21839e-29 - -3.90207e-29 - -3.5372e-29 - -3.14284e-29 - -2.73808e-29 - -2.34077e-29 - -1.96662e-29 - -1.62865e-29 - -1.33721e-29 - -1.10036e-29 - -9.2453e-30 - -8.15412e-30 - -7.78881e-30 - -8.21758e-30 - -9.52312e-30 - -1.18035e-29 - -1.51679e-29 - -1.97275e-29 - -2.558e-29 - -3.27911e-29 - -4.13726e-29 - -5.12614e-29 - -6.23009e-29 - -7.42297e-29 - -8.66799e-29 - -9.91872e-29 - -1.11214e-28 - -1.22184e-28 - -1.31526e-28 - -1.38724e-28 - -1.43361e-28 - -1.4516e-28 - -1.44013e-28 - -1.39988e-28 - -1.33327e-28 - -1.24418e-28 - -1.13758e-28 - -1.01911e-28 - -8.94532e-29 - -7.6932e-29 - -6.4827e-29 - -5.35231e-29 - -4.32975e-29 - -3.4318e-29 - -2.66513e-29 - -2.02792e-29 - -1.51189e-29 - -1.1044e-29 - -7.90442e-30 - -5.54306e-30 - -3.80862e-30 - -2.56402e-30 - -1.69128e-30 - -1.09308e-30 - -6.92222e-31 - -4.29584e-31 - -2.61355e-31 - -1.56088e-31 - -9.19194e-32 - -5.41627e-32 - -3.33878e-32 - -2.39882e-32 - -2.3218e-32 - -3.0655e-32 - -4.80483e-32 - -7.95209e-32 - -1.32102e-31 - -2.16561e-31 - -3.48504e-31 - -5.4964e-31 - -8.4906e-31 - -1.28428e-30 - -1.90166e-30 - -2.75569e-30 - -3.90656e-30 - -5.41518e-30 - -7.33504e-30 - -9.70027e-30 - -1.25094e-29 - -1.57049e-29 - -1.91496e-29 - -2.26002e-29 - -2.56814e-29 - -2.78632e-29 - -2.84449e-29 - -2.65514e-29 - -2.11459e-29 - -1.1063e-29 - 4.9334e-30 - 2.80657e-29 - 5.9435e-29 - 9.98929e-29 - 1.49904e-28 - 2.09412e-28 - 2.77724e-28 - 3.53436e-28 - 4.34413e-28 - 5.17842e-28 - 6.00357e-28 - 6.78246e-28 - 7.47713e-28 - 8.05179e-28 - 8.47584e-28 - 8.72664e-28 - 8.79159e-28 - 8.66922e-28 - 8.3693e-28 - 7.91177e-28 - 7.32478e-28 - 6.64199e-28 - 5.89956e-28 - 5.13319e-28 - 4.37543e-28 - 3.65373e-28 - 2.98915e-28 - 2.39587e-28 - 1.88145e-28 - 1.44757e-28 - 1.09121e-28 - 8.05941e-29 - 5.83221e-29 - 4.13531e-29 - 2.87313e-29 - 1.95636e-29 - 1.30613e-29 - 8.5607e-30 - 5.52689e-30 - 3.54656e-30 - 2.31421e-30 - 1.61594e-30 - 1.31363e-30 - 1.3311e-30 - 1.64297e-30 - 2.26634e-30 - 3.2545e-30 - 4.69173e-30 - 6.68815e-30 - 9.37322e-30 - 1.28869e-29 - 1.73676e-29 - 2.29371e-29 - 2.96827e-29 - 3.76368e-29 - 4.67587e-29 - 5.69181e-29 - 6.78854e-29 - 7.93302e-29 - 9.08318e-29 - 1.019e-28 - 1.12008e-28 - 1.20631e-28 - 1.27293e-28 - 1.3161e-28 - 1.33324e-28 - 1.32332e-28 - 1.28694e-28 - 1.22628e-28 - 1.14488e-28 - 1.04728e-28 - 9.38652e-29 - 8.24296e-29 - 7.09249e-29 - 5.9793e-29 - 4.93901e-29 - 3.99729e-29 - 3.16977e-29 - 2.46279e-29 - 1.87484e-29 - 1.39842e-29 - 1.02199e-29 - 7.31803e-30 - 5.13426e-30 - 3.52938e-30 - 2.37715e-30 - 1.56874e-30 - 1.01433e-30 - 6.42612e-31 - 3.98891e-31 - 2.42603e-31 - 1.44569e-31 - 8.44091e-32 - 4.82883e-32 - 2.70664e-32 - 1.48647e-32 - 7.9987e-33 - 4.21715e-33 - 2.17849e-33 - 1.10263e-33 - 5.46814e-34 - 2.65699e-34 - 1.265e-34 - 5.90203e-35 - 2.70067e-35 - 1.21745e-35 - 5.54095e-36 - 2.86492e-36 - 2.35334e-36 - 3.63893e-36 - 7.55276e-36 - 1.65753e-35 - 3.616e-35 - 7.74919e-35 - 1.62788e-34 - 3.35089e-34 - 6.75835e-34 - 1.33555e-33 - 2.58592e-33 - 4.90578e-33 - 9.11881e-33 - 1.66075e-32 - 2.96352e-32 - 5.18142e-32 - 8.87617e-32 - 1.48984e-31 - 2.45014e-31 - 3.94801e-31 - 6.23308e-31 - 9.64193e-31 - 1.46138e-30 - 2.17019e-30 - 3.15769e-30 - 4.50171e-30 - 6.28816e-30 - 8.60609e-30 - 1.15405e-29 - 1.51629e-29 - 1.95197e-29 - 2.46209e-29 - 3.04278e-29 - 3.68446e-29 - 4.37134e-29 - 5.0815e-29 - 5.7877e-29 - 6.45887e-29 - 7.06228e-29 - 7.5661e-29 - 7.94232e-29 - 8.16998e-29 - 8.24067e-29 - 8.17741e-29 - 8.12451e-29 - 8.79235e-29 - 1.36288e-28 - 3.91958e-28 - 1.63703e-27 - 7.49233e-27 - 3.43794e-26 - 1.55241e-25 - 6.87388e-25 - 2.98264e-24 - 1.26808e-23 - 5.28244e-23 - 2.15605e-22 - 8.62222e-22 - 3.37844e-21 - 1.29703e-20 - 4.87891e-20 - 1.79817e-19 - 6.49345e-19 - 2.29751e-18 - 7.96482e-18 - 2.7054e-17 - 9.00377e-17 - 2.93598e-16 - 9.38036e-16 - 2.93645e-15 - 9.00663e-15 - 2.70669e-14 - 7.96989e-14 - 2.29934e-13 - 6.49965e-13 - 1.80017e-12 - 4.88512e-12 - 1.29889e-11 - 3.38382e-11 - 8.63731e-11 - 2.16016e-10 - 5.29337e-10 - 1.27091e-09 - 2.98973e-09 - 6.8911e-09 - 1.55626e-08 - 3.44358e-08 - 7.4658e-08 - 1.58591e-07 - 3.3008e-07 - 6.73126e-07 - 1.34496e-06 - 2.63306e-06 - 5.05065e-06 - 9.4923e-06 - 1.74796e-05 - 3.15377e-05 - 5.57526e-05 - 9.65687e-05 - 0.000163887 - 0.000272515 - 0.000443989 - 0.000708747 - 0.00110853 - 0.00169879 - 0.00255076 - 0.00375262 - 0.00540927 - 0.00763973 - 0.0105719 - 0.014334 - 0.0190423 - 0.024786 - 0.0316105 - 0.0394995 - 0.0483604 - 0.0580129 - 0.0681862 - 0.0785245 - 0.0886035 - 0.0979565 - 0.106109 - 0.112619 - 0.117113 - 0.119328 - 0.119129 - 0.116531 - 0.11169 - 0.104896 - 0.096536 - 0.0870678 - 0.0769742 - 0.0667276 - 0.0567567 - 0.0474229 - 0.0390066 - 0.0317027 - 0.0256258 - 0.0208203 - 0.0172751 - 0.0149371 - 0.0137248 - 0.0135373 - 0.0142605 - 0.0157693 - 0.0179273 - 0.0205862 - 0.0235843 - 0.0267481 - 0.0298956 - 0.032843 - 0.0354138 - 0.0374501 - 0.0388236 - 0.0394455 - 0.0392739 - 0.0383163 - 0.0366285 - 0.0343086 - 0.0314868 - 0.0283136 - 0.0249459 - 0.0215348 - 0.0182147 - 0.0150952 - 0.0122572 - 0.00975177 - 0.00760171 - 0.00580599 - 0.0043449 - 0.00318583 - 0.00228882 - 0.00161127 - 0.00111162 - 0.0007519 - 0.000499301 - 0.000326825 - 0.000213418 - 0.000143759 - 0.000108028 - 0.000101857 - 0.000126739 - 0.000191099 - 0.000312256 - 0.000519518 - 0.000858582 - 0.00139739 - 0.00223343 - 0.00350226 - 0.0053867 - 0.00812573 - 0.0120216 - 0.0174433 - 0.0248236 - 0.0346481 - 0.0474328 - 0.0636896 - 0.083879 - 0.108352 - 0.137285 - 0.170613 - 0.207972 - 0.248658 - 0.291611 - 0.335434 - 0.37845 - 0.418797 - 0.454557 - 0.483901 - 0.505245 - 0.517389 - 0.519629 - 0.511826 - 0.494418 - 0.468382 - 0.435139 - 0.396432 - 0.354169 - 0.310271 - 0.266533 - 0.224505 - 0.18542 - 0.150151 - 0.119216 - 0.0928035 - 0.0708289 - 0.0530002 - 0.0388861 - 0.0279804 - 0.0197564 - 0.0137097 - 0.00938729 - 0.00640552 - 0.00445878 - 0.00332166 - 0.00284708 - 0.00296262 - 0.0036664 - 0.00502305 - 0.00715947 - 0.0102594 - 0.0145556 - 0.0203168 - 0.0278303 - 0.0373755 - 0.0491924 - 0.0634435 - 0.0801731 - 0.099269 - 0.120431 - 0.143153 - 0.166724 - 0.190254 - 0.21272 - 0.233033 - 0.250129 - 0.263056 - 0.271062 - 0.273669 - 0.270719 - 0.262392 - 0.249183 - 0.231858 - 0.21138 - 0.188818 - 0.165256 - 0.141713 - 0.119069 - 0.0980222 - 0.0790655 - 0.0624865 - 0.0483863 - 0.036711 - 0.0272902 - 0.0198772 - 0.0141856 - 0.00991947 - 0.00679686 - 0.00456443 - 0.00300582 - 0.00194419 - 0.00124098 - 0.000792285 - 0.000524266 - 0.000388696 - 0.000359281 - 0.000429289 - 0.000610657 - 0.000934591 - 0.00145347 - 0.00224368 - 0.00340871 - 0.00508177 - 0.00742678 - 0.0106365 - 0.0149265 - 0.0205241 - 0.0276509 - 0.0365 - 0.0472078 - 0.0598234 - 0.0742789 - 0.0903642 - 0.107712 - 0.125797 - 0.14395 - 0.161395 - 0.177299 - 0.190835 - 0.201255 - 0.207957 - 0.21054 - 0.20885 - 0.202989 - 0.193306 - 0.180366 - 0.164893 - 0.147702 - 0.12963 - 0.111472 - 0.0939202 - 0.0775336 - 0.0627131 - 0.0497008 - 0.0385927 - 0.0293619 - 0.0218877 - 0.0159864 - 0.0114404 - 0.00802171 - 0.00551099 - 0.00370962 - 0.00244662 - 0.00158103 - 0.00100104 - 0.000621008 - 0.000377469 - 0.000224803 - 0.000131178 - 7.49988e-05 - 4.20132e-05 - 2.30597e-05 - 1.24011e-05 - 6.53433e-06 - 3.37349e-06 - 1.70646e-06 - 8.45761e-07 - 4.10712e-07 - 1.95417e-07 - 9.11016e-08 - 4.16127e-08 - 1.86236e-08 - 8.16673e-09 - 3.50932e-09 - 1.47874e-09 - 6.13789e-10 - 2.58271e-10 - 1.2874e-10 - 1.17134e-10 - 2.11594e-10 - 4.85849e-10 - 1.14897e-09 - 2.68349e-09 - 6.14836e-09 - 1.3805e-08 - 3.03712e-08 - 6.54678e-08 - 1.38271e-07 - 2.86134e-07 - 5.80156e-07 - 1.15254e-06 - 2.2434e-06 - 4.27852e-06 - 7.99498e-06 - 1.46379e-05 - 2.62588e-05 - 4.61539e-05 - 7.94839e-05 - 0.000134118 - 0.000221734 - 0.000359181 - 0.000570074 - 0.000886515 - 0.00135076 - 0.00201654 - 0.00294966 - 0.00422741 - 0.00593627 - 0.00816751 - 0.0110104 - 0.0145429 - 0.0188208 - 0.0238651 - 0.0296499 - 0.0360928 - 0.0430481 - 0.0503066 - 0.0576013 - 0.0646215 - 0.0710327 - 0.0765027 - 0.0807294 - 0.0834688 - 0.0845576 - 0.0839303 - 0.0816247 - 0.077779 - 0.0726175 - 0.0664294 - 0.0595419 - 0.0522922 - 0.0450009 - 0.0379502 - 0.0313689 - 0.025425 - 0.0202253 - 0.015822 - 0.0122232 - 0.00940706 - 0.00733663 - 0.0059741 - 0.00529347 - 0.00529026 - 0.00598776 - 0.00743939 - 0.00972643 - 0.0129513 - 0.0172261 - 0.0226564 - 0.0293225 - 0.0372574 - 0.0464256 - 0.0567052 - 0.0678749 - 0.0796106 - 0.0914928 - 0.103026 - 0.113671 - 0.122882 - 0.130157 - 0.135077 - 0.137352 - 0.136843 - 0.133582 - 0.127765 - 0.119732 - 0.109938 - 0.0989056 - 0.0871828 - 0.075297 - 0.0637179 - 0.0528302 - 0.042918 - 0.0341613 - 0.0266419 - 0.0203579 - 0.0152419 - 0.011181 - 0.00803634 - 0.00565944 - 0.00390504 - 0.00264007 - 0.0017488 - 0.00113502 - 0.000721777 - 0.000449717 - 0.000274545 - 0.000164219 - 9.62432e-05 - 5.52655e-05 - 3.10939e-05 - 1.71409e-05 - 9.25821e-06 - 4.89957e-06 - 2.54054e-06 - 1.29072e-06 - 6.425e-07 - 3.13366e-07 - 1.4975e-07 - 7.01164e-08 - 3.21669e-08 - 1.44589e-08 - 6.36792e-09 - 2.74788e-09 - 1.16181e-09 - 4.8129e-10 - 1.95352e-10 - 7.76899e-11 - 3.02725e-11 - 1.15577e-11 - 4.32374e-12 - 1.58565e-12 - 5.72373e-13 - 2.10546e-13 - 1.00464e-13 - 1.17138e-13 - 2.82074e-13 - 7.88586e-13 - 2.20757e-12 - 6.07018e-12 - 1.63587e-11 - 4.31963e-11 - 1.11759e-10 - 2.83306e-10 - 7.03665e-10 - 1.71243e-09 - 4.08317e-09 - 9.53933e-09 - 2.18361e-08 - 4.89745e-08 - 1.07622e-07 - 2.31723e-07 - 4.88848e-07 - 1.01045e-06 - 2.04642e-06 - 4.06079e-06 - 7.89519e-06 - 1.50401e-05 - 2.80723e-05 - 5.13382e-05 - 9.199e-05 - 0.000161502 - 0.000277812 - 0.000468232 - 0.000773229 - 0.0012511 - 0.00198341 - 0.00308085 - 0.00468883 - 0.00699192 - 0.0102156 - 0.0146241 - 0.0205122 - 0.0281897 - 0.0379582 - 0.0500793 - 0.0647363 - 0.0819924 - 0.101751 - 0.123719 - 0.147392 - 0.172047 - 0.196769 - 0.220498 - 0.242097 - 0.260441 - 0.274516 - 0.283507 - 0.286878 - 0.284426 - 0.2763 - 0.262989 - 0.245271 - 0.224139 - 0.200716 - 0.176155 - 0.151552 - 0.127879 - 0.105932 - 0.0863111 - 0.0694246 - 0.0555089 - 0.0446667 - 0.0369097 - 0.0322012 - 0.0304904 - 0.0317364 - 0.0359162 - 0.0430182 - 0.0530213 - 0.0658637 - 0.0814058 - 0.0993939 - 0.119431 - 0.14096 - 0.163268 - 0.185513 - 0.206765 - 0.226071 - 0.242535 - 0.255385 - 0.26405 - 0.268204 - 0.267793 - 0.263031 - 0.25437 - 0.242442 - 0.22799 - 0.211798 - 0.194613 - 0.177099 - 0.159795 - 0.143102 - 0.127283 - 0.112488 - 0.098775 - 0.0861443 - 0.0745635 - 0.0639902 - 0.0543848 - 0.0457162 - 0.0379612 - 0.0311 - 0.0251102 - 0.0199607 - 0.0156086 - 0.0119976 - 0.00905948 - 0.00671686 - 0.00488768 - 0.00348951 - 0.00244361 - 0.00167807 - 0.00112985 - 0.000745775 - 0.00048253 - 0.000306019 - 0.000190246 - 0.000115991 - 6.94814e-05 - 4.11611e-05 - 2.46614e-05 - 1.60011e-05 - 1.3041e-05 - 1.52032e-05 - 2.34787e-05 - 4.07624e-05 - 7.25905e-05 - 0.000128383 - 0.000223324 - 0.00038104 - 0.000637195 - 0.00104411 - 0.00167637 - 0.00263714 - 0.00406474 - 0.00613861 - 0.00908331 - 0.0131691 - 0.0187069 - 0.0260368 - 0.0355065 - 0.0474424 - 0.06211 - 0.0796698 - 0.10013 - 0.123301 - 0.148768 - 0.175869 - 0.203707 - 0.231185 - 0.257068 - 0.280076 - 0.298978 - 0.312708 - 0.320462 - 0.321774 - 0.316564 - 0.305147 - 0.2882 - 0.266695 - 0.24181 - 0.214817 - 0.186982 - 0.159467 - 0.133253 - 0.109098 - 0.087518 - 0.0687881 - 0.0529745 - 0.0399721 - 0.0295518 - 0.0214066 - 0.0151931 - 0.0105654 - 0.00719878 - 0.00480585 - 0.00314353 - 0.00201466 - 0.0012651 - 0.000778364 - 0.000469221 - 0.000277147 - 0.000160391 - 9.09462e-05 - 5.05275e-05 - 2.75047e-05 - 1.46699e-05 - 7.6664e-06 - 3.9259e-06 - 1.97083e-06 - 9.71818e-07 - 4.75235e-07 - 2.40786e-07 - 1.48546e-07 - 1.49402e-07 - 2.43803e-07 - 4.81945e-07 - 9.85453e-07 - 1.99755e-06 - 3.97698e-06 - 7.7618e-06 - 1.48441e-05 - 2.78157e-05 - 5.107e-05 - 9.1871e-05 - 0.00016193 - 0.00027965 - 0.000473193 - 0.00078451 - 0.00127437 - 0.00202829 - 0.00316301 - 0.0048329 - 0.00723523 - 0.0106129 - 0.0152529 - 0.0214786 - 0.0296346 - 0.0400615 - 0.0530631 - 0.0688644 - 0.0875658 - 0.109096 - 0.133175 - 0.159284 - 0.186664 - 0.21433 - 0.241126 - 0.265791 - 0.287062 - 0.303771 - 0.314959 - 0.319964 - 0.318482 - 0.310605 - 0.296807 - 0.2779 - 0.254954 - 0.2292 - 0.201925 - 0.174371 - 0.147651 - 0.122691 - 0.100198 - 0.0806612 - 0.06437 - 0.0514539 - 0.0419302 - 0.0357515 - 0.0328484 - 0.0331588 - 0.0366424 - 0.0432776 - 0.0530423 - 0.0658818 - 0.0816677 - 0.100155 - 0.120947 - 0.143467 - 0.166955 - 0.190486 - 0.213013 - 0.233429 - 0.250655 - 0.263726 - 0.271879 - 0.274625 - 0.271795 - 0.263562 - 0.250416 - 0.233118 - 0.212632 - 0.190028 - 0.166397 - 0.14276 - 0.120007 - 0.0988426 - 0.079766 - 0.0630708 - 0.0488625 - 0.0370902 - 0.0275855 - 0.020102 - 0.0143527 - 0.0100407 - 0.00688229 - 0.00462209 - 0.00304145 - 0.00196092 - 0.00123872 - 0.000766703 - 0.000464961 - 0.000276276 - 0.000160845 - 9.175e-05 - 5.12794e-05 - 2.80813e-05 - 1.5067e-05 - 7.9209e-06 - 4.07999e-06 - 2.05911e-06 - 1.01821e-06 - 4.93325e-07 - 2.34188e-07 - 1.08926e-07 - 4.96407e-08 - 2.21656e-08 - 9.69749e-09 - 4.15695e-09 - 1.74593e-09 - 7.18484e-10 - 2.89697e-10 - 1.14448e-10 - 4.43003e-11 - 1.68013e-11 - 6.24335e-12 - 2.27315e-12 - 8.10916e-13 - 2.83439e-13 - 9.70692e-14 - 3.25716e-14 - 1.07086e-14 - 3.44958e-15 - 1.08877e-15 - 3.36698e-16 - 1.02019e-16 - 3.02874e-17 - 8.81003e-18 - 2.5109e-18 - 7.01164e-19 - 1.91843e-19 - 5.14292e-20 - 1.35086e-20 - 3.47654e-21 - 8.76639e-22 - 2.16587e-22 - 5.24299e-23 - 1.24355e-23 - 2.88991e-24 - 6.58026e-25 - 1.46804e-25 - 3.209e-26 - 6.87287e-27 - 1.44226e-27 - 2.96542e-28 - 5.974e-29 - 1.17918e-29 - 2.28052e-30 - 4.3214e-31 - 8.02326e-32 - 1.45953e-32 - 2.60144e-33 - 4.54309e-34 - 7.77366e-35 - 1.30328e-35 - 2.14084e-36 - 3.44563e-37 - 5.43363e-38 - 8.39553e-39 - 1.27099e-39 - 1.88528e-40 - 2.73996e-41 - 3.90166e-42 - 5.44367e-43 - 7.44168e-44 - 9.96753e-45 - 1.3081e-45 - 1.68202e-46 - 2.11913e-47 - 2.6159e-48 - 3.16389e-49 - 3.74937e-50 - 4.35344e-51 - 4.95271e-52 - 5.52066e-53 - 6.02942e-54 - 6.45204e-55 - 6.76481e-56 - 6.94945e-57 - 6.99492e-58 - 6.89845e-59 - 6.66587e-60 - 6.31102e-61 - 5.85435e-62 - 5.32102e-63 - 4.73858e-64 - 4.13465e-65 - 3.5348e-66 - 2.96093e-67 - 2.43013e-68 - 1.95419e-69 - 1.53972e-70 - 1.18864e-71 - 8.99083e-73 - 6.66322e-74 - 4.83845e-75 - 3.44242e-76 - 2.39971e-77 - 1.63905e-78 - 1.09688e-79 - 7.19228e-81 - 4.62071e-82 - 2.90863e-83 - 1.79393e-84 - 1.08407e-85 - 6.41872e-87 - 3.7237e-88 - 2.1166e-89 - 1.1788e-90 - 6.43246e-92 - 3.43915e-93 - 1.80162e-94 - 9.2472e-96 - 4.65045e-97 - 2.29148e-98 - 1.10631e-99 - 5.23324e-101 - 2.42551e-102 - 1.10147e-103 - 4.90092e-105 - 2.13659e-106 - 9.12641e-108 - 3.81959e-109 - 1.56628e-110 - 6.29303e-112 - 2.4774e-113 - 9.68532e-115 - 3.77016e-115 - 8.76891e-114 - 2.20935e-112 - 5.45488e-111 - 1.3196e-109 - 3.12779e-108 - 7.2639e-107 - 1.65287e-105 - 3.68506e-104 - 8.04983e-103 - 1.72292e-101 - 3.61312e-100 - 7.42395e-99 - 1.4946e-97 - 2.94816e-96 - 5.6979e-95 - 1.07898e-93 - 2.00195e-92 - 3.63937e-91 - 6.48242e-90 - 1.13132e-88 - 1.9345e-87 - 3.24109e-86 - 5.32046e-85 - 8.55745e-84 - 1.34858e-82 - 2.08231e-81 - 3.15029e-80 - 4.66975e-79 - 6.78224e-78 - 9.65137e-77 - 1.34568e-75 - 1.83837e-74 - 2.4607e-73 - 3.22717e-72 - 4.14689e-71 - 5.22108e-70 - 6.44072e-69 - 7.78476e-68 - 9.21919e-67 - 1.06974e-65 - 1.21618e-64 - 1.35474e-63 - 1.47861e-62 - 1.58119e-61 - 1.65674e-60 - 1.70082e-59 - 1.71081e-58 - 1.68609e-57 - 1.62816e-56 - 1.54046e-55 - 1.42804e-54 - 1.29708e-53 - 1.15433e-52 - 1.00654e-51 - 8.59942e-51 - 7.19853e-50 - 5.90411e-49 - 4.74463e-48 - 3.73583e-47 - 2.8821e-46 - 2.17855e-45 - 1.61348e-44 - 1.17083e-43 - 8.32461e-43 - 5.79922e-42 - 3.95833e-41 - 2.64723e-40 - 1.73464e-39 - 1.11368e-38 - 7.0057e-38 - 4.31796e-37 - 2.60761e-36 - 1.54292e-35 - 8.94501e-35 - 5.08107e-34 - 2.82791e-33 - 1.54211e-32 - 8.23947e-32 - 4.31341e-31 - 2.21248e-30 - 1.11192e-29 - 5.47529e-29 - 2.64166e-28 - 1.24877e-27 - 5.78395e-27 - 2.62485e-26 - 1.16713e-25 - 5.08481e-25 - 2.17052e-24 - 9.07803e-24 - 3.72011e-23 - 1.49368e-22 - 5.87616e-22 - 2.265e-21 - 8.5542e-21 - 3.16539e-20 - 1.14766e-19 - 4.07694e-19 - 1.41903e-18 - 4.83937e-18 - 1.61704e-17 - 5.29408e-17 - 1.69823e-16 - 5.33751e-16 - 1.64369e-15 - 4.95947e-15 - 1.46619e-14 - 4.24697e-14 - 1.20533e-13 - 3.35174e-13 - 9.13211e-13 - 2.43786e-12 - 6.37651e-12 - 1.63416e-11 - 4.10339e-11 - 1.00955e-10 - 2.4336e-10 - 5.74789e-10 - 1.33016e-09 - 3.01604e-09 - 6.70048e-09 - 1.45852e-08 - 3.11068e-08 - 6.50032e-08 - 1.33092e-07 - 2.66996e-07 - 5.24802e-07 - 1.0107e-06 - 1.90716e-06 - 3.52605e-06 - 6.38742e-06 - 1.13371e-05 - 1.97157e-05 - 3.35939e-05 - 5.60848e-05 - 9.17418e-05 - 0.000147037 - 0.000230899 - 0.000355267 - 0.000535579 - 0.000791098 - 0.00114491 - 0.0016235 - 0.00225564 - 0.0030706 - 0.00409557 - 0.00535231 - 0.00685339 - 0.00859818 - 0.0105693 - 0.0127297 - 0.0150221 - 0.0173692 - 0.0196772 - 0.0218417 - 0.0237545 - 0.0253129 - 0.0264286 - 0.0270361 - 0.0270988 - 0.0266129 - 0.0256078 - 0.0241429 - 0.022302 - 0.0201853 - 0.0179004 - 0.0155534 - 0.0132412 - 0.011045 - - - - - -0.00206576 - -0.00257078 - -0.00313465 - -0.00374497 - -0.00438375 - -0.00502781 - -0.00565001 - -0.00622095 - -0.00671121 - -0.00709385 - -0.00734683 - -0.00745512 - -0.00741219 - -0.00722062 - -0.00689192 - -0.00644528 - -0.00590583 - -0.00530221 - -0.00466412 - -0.00401994 - -0.00339473 - -0.00280885 - -0.00227713 - -0.00180878 - -0.00140773 - -0.00107347 - -0.000802042 - -0.00058714 - -0.000421136 - -0.000295965 - -0.000203796 - -0.000137495 - -9.08899e-05 - -5.88682e-05 - -3.7358e-05 - -2.32286e-05 - -1.41514e-05 - -8.44717e-06 - -4.94038e-06 - -2.83105e-06 - -1.58954e-06 - -8.74443e-07 - -4.71337e-07 - -2.48932e-07 - -1.28834e-07 - -6.53783e-08 - -3.26243e-08 - -1.62362e-08 - -8.59254e-09 - -5.99475e-09 - -7.26475e-09 - -1.33706e-08 - -2.80399e-08 - -5.96719e-08 - -1.25329e-07 - -2.58291e-07 - -5.21712e-07 - -1.03256e-06 - -2.00235e-06 - -3.80456e-06 - -7.0828e-06 - -1.29194e-05 - -2.30897e-05 - -4.04323e-05 - -6.93708e-05 - -0.000116617 - -0.000192081 - -0.000309986 - -0.000490161 - -0.0007594 - -0.00115276 - -0.00171453 - -0.00249856 - -0.00356754 - -0.00499098 - -0.00684131 - -0.00918818 - -0.0120909 - -0.0155891 - -0.0196935 - -0.0243759 - -0.0295621 - -0.0351275 - -0.0408974 - -0.0466533 - -0.0521444 - -0.0571046 - -0.0612738 - -0.0644202 - -0.0663617 - -0.0669837 - -0.0662512 - -0.0642123 - -0.0609945 - -0.0567933 - -0.0518548 - -0.046455 - -0.0408787 - -0.0353993 - -0.0302635 - -0.0256799 - -0.0218134 - -0.0187838 - -0.0166674 - -0.0155012 - -0.0152867 - -0.0159929 - -0.0175585 - -0.0198923 - -0.0228734 - -0.0263515 - -0.0301487 - -0.0340648 - -0.0378847 - -0.0413897 - -0.0443713 - -0.0466459 - -0.0480693 - -0.0485484 - -0.0480492 - -0.0465983 - -0.0442805 - -0.0412291 - -0.037613 - -0.0336212 - -0.029446 - -0.0252685 - -0.0212456 - -0.0175023 - -0.0141273 - -0.0111728 - -0.00865761 - -0.00657314 - -0.00488972 - -0.00356396 - -0.00254518 - -0.0017809 - -0.00122095 - -0.000820152 - -0.000539793 - -0.000348094 - -0.000219939 - -0.000136159 - -8.25898e-05 - -4.90844e-05 - -2.85823e-05 - -1.63075e-05 - -9.11622e-06 - -4.9932e-06 - -2.67966e-06 - -1.40902e-06 - -7.25927e-07 - -3.66442e-07 - -1.8124e-07 - -8.78291e-08 - -4.17024e-08 - -1.94008e-08 - -8.84333e-09 - -3.94956e-09 - -1.7283e-09 - -7.41011e-10 - -3.11293e-10 - -1.28134e-10 - -5.16863e-11 - -2.04562e-11 - -8.01249e-12 - -3.29647e-12 - -1.92366e-12 - -2.56019e-12 - -5.76495e-12 - -1.44162e-11 - -3.59951e-11 - -8.82901e-11 - -2.12263e-10 - -5.00027e-10 - -1.15412e-09 - -2.61005e-09 - -5.78339e-09 - -1.25561e-08 - -2.67092e-08 - -5.56679e-08 - -1.1368e-07 - -2.27459e-07 - -4.45921e-07 - -8.56544e-07 - -1.61205e-06 - -2.97264e-06 - -5.37088e-06 - -9.50789e-06 - -1.64915e-05 - -2.80267e-05 - -4.66683e-05 - -7.61391e-05 - -0.000121711 - -0.00019063 - -0.000292542 - -0.000439867 - -0.000648025 - -0.000935404 - -0.00132295 - -0.00183326 - -0.0024891 - -0.00331129 - -0.00431607 - -0.0055121 - -0.00689736 - -0.00845639 - -0.0101584 - -0.0119564 - -0.0137884 - -0.0155798 - -0.0172484 - -0.0187099 - -0.0198853 - -0.0207075 - -0.0211281 - -0.0211218 - -0.0206889 - -0.0198553 - -0.0186703 - -0.0172011 - -0.0155269 - -0.0137317 - -0.0118973 - -0.0100974 - -0.00839278 - -0.00682883 - -0.00543428 - -0.00422203 - -0.00319098 - -0.00232873 - -0.00161469 - -0.00102309 - -0.000525842 - -9.49211e-05 - 0.000295706 - 0.00066859 - 0.0010417 - 0.00142767 - 0.00183334 - 0.0022596 - 0.00270153 - 0.003149 - 0.00358751 - 0.0039995 - 0.00436595 - 0.00466822 - 0.00488978 - 0.005018 - 0.00504537 - 0.00497037 - 0.00479765 - 0.00453761 - 0.00420546 - 0.00381981 - 0.00340117 - 0.00297039 - 0.00254737 - 0.00215021 - 0.00179488 - 0.00149551 - 0.00126519 - 0.00111731 - 0.00106721 - 0.00113403 - 0.00134235 - 0.0017236 - 0.00231665 - 0.00316746 - 0.00432738 - 0.00584992 - 0.00778586 - 0.0101769 - 0.0130483 - 0.0164007 - 0.0202032 - 0.0243877 - 0.028846 - 0.0334309 - 0.0379624 - 0.0422376 - 0.046045 - 0.0491818 - 0.051471 - 0.0527786 - 0.0530262 - 0.0521987 - 0.0503461 - 0.0475783 - 0.0440543 - 0.0399673 - 0.035527 - 0.030942 - 0.0264044 - 0.022077 - 0.0180859 - 0.0145171 - 0.011417 - 0.00879761 - 0.00664221 - 0.00491358 - 0.00356139 - 0.00252917 - 0.00175985 - 0.00119979 - 0.00080145 - 0.000524545 - 0.000336376 - 0.000211352 - 0.000130113 - 7.8483e-05 - 4.63838e-05 - 2.68592e-05 - 1.5239e-05 - 8.47147e-06 - 4.6142e-06 - 2.46247e-06 - 1.2876e-06 - 6.59676e-07 - 3.31143e-07 - 1.62869e-07 - 7.84869e-08 - 3.70589e-08 - 1.71445e-08 - 7.77132e-09 - 3.45145e-09 - 1.50191e-09 - 6.4036e-10 - 2.67511e-10 - 1.09495e-10 - 4.39122e-11 - 1.72549e-11 - 6.6432e-12 - 2.50598e-12 - 9.26224e-13 - 3.35422e-13 - 1.19015e-13 - 4.13762e-14 - 1.40941e-14 - 4.7039e-15 - 1.53822e-15 - 4.92848e-16 - 1.5472e-16 - 4.759e-17 - 1.43424e-17 - 4.23512e-18 - 1.22531e-18 - 3.47346e-19 - 9.64753e-20 - 2.62547e-20 - 7.00059e-21 - 1.82896e-21 - 4.68283e-22 - 1.17935e-22 - 3.10821e-23 - 1.6359e-23 - 4.05988e-23 - 1.58397e-22 - 6.28053e-22 - 2.44515e-21 - 9.32834e-21 - 3.48693e-20 - 1.27708e-19 - 4.5828e-19 - 1.61132e-18 - 5.55095e-18 - 1.87366e-17 - 6.19655e-17 - 2.00792e-16 - 6.37501e-16 - 1.98313e-15 - 6.04448e-15 - 1.80511e-14 - 5.28183e-14 - 1.51427e-13 - 4.2536e-13 - 1.17071e-12 - 3.15702e-12 - 8.34147e-12 - 2.15946e-11 - 5.47752e-11 - 1.36132e-10 - 3.31491e-10 - 7.909e-10 - 1.84888e-09 - 4.23479e-09 - 9.50367e-09 - 2.08972e-08 - 4.50217e-08 - 9.50371e-08 - 1.96562e-07 - 3.98332e-07 - 7.90908e-07 - 1.53867e-06 - 2.93291e-06 - 5.47761e-06 - 1.00235e-05 - 1.79715e-05 - 3.15709e-05 - 5.43408e-05 - 9.16436e-05 - 0.000151431 - 0.000245169 - 0.000388912 - 0.000604471 - 0.000920526 - 0.00137352 - 0.00200802 - 0.00287635 - 0.00403694 - 0.00555138 - 0.00747978 - 0.00987455 - 0.0127729 - 0.0161884 - 0.0201035 - 0.0244622 - 0.0291667 - 0.0340773 - 0.0390171 - 0.0437822 - 0.0481565 - 0.0519306 - 0.0549224 - 0.0569978 - 0.0580884 - 0.0582047 - 0.057442 - 0.0559787 - 0.0540666 - 0.0520136 - 0.0501604 - 0.048853 - 0.0484132 - 0.0491098 - 0.0511319 - 0.0545677 - 0.0593893 - 0.0654475 - 0.0724749 - 0.0801003 - 0.0878732 - 0.095297 - 0.101868 - 0.107117 - 0.11065 - 0.112183 - 0.111565 - 0.108787 - 0.103986 - 0.0974183 - 0.0894397 - 0.0804662 - 0.0709366 - 0.0612756 - 0.0518631 - 0.0430107 - 0.0349495 - 0.0278261 - 0.0217078 - 0.016594 - 0.0124308 - 0.00912769 - 0.00657296 - 0.0046476 - 0.00323574 - 0.00223212 - 0.00154656 - 0.00110583 - 0.000853632 - 0.000749224 - 0.00076522 - 0.000884946 - 0.00109963 - 0.00140561 - 0.00180157 - 0.00228607 - 0.0028553 - 0.00350116 - 0.00421001 - 0.00496194 - 0.00573094 - 0.00648586 - 0.00719215 - 0.00781434 - 0.00831889 - 0.00867714 - 0.00886798 - 0.00887994 - 0.00871228 - 0.00837512 - 0.00788836 - 0.0072798 - 0.00658247 - 0.00583171 - 0.0050622 - 0.00430545 - 0.00358786 - 0.00292948 - 0.00234358 - 0.001837 - 0.00141082 - 0.00106163 - 0.000782729 - 0.00056544 - 0.00040022 - 0.000277554 - 0.000188596 - 0.000125561 - 8.19057e-05 - 5.23492e-05 - 3.27825e-05 - 2.01146e-05 - 1.20926e-05 - 7.123e-06 - 4.11097e-06 - 2.32472e-06 - 1.28813e-06 - 6.99559e-07 - 3.72774e-07 - 1.95909e-07 - 1.03891e-07 - 6.08753e-08 - 4.99608e-08 - 6.85721e-08 - 1.29035e-07 - 2.6531e-07 - 5.4877e-07 - 1.11869e-06 - 2.23728e-06 - 4.38521e-06 - 8.42213e-06 - 1.58488e-05 - 2.92219e-05 - 5.27908e-05 - 9.34425e-05 - 0.000162057 - 0.000275377 - 0.000458485 - 0.000747926 - 0.00119544 - 0.00187213 - 0.00287264 - 0.00431879 - 0.00636181 - 0.00918196 - 0.0129846 - 0.017991 - 0.0244243 - 0.032488 - 0.0423412 - 0.0540678 - 0.0676475 - 0.082928 - 0.0996065 - 0.117223 - 0.135167 - 0.15271 - 0.169045 - 0.183347 - 0.194841 - 0.202874 - 0.20697 - 0.206883 - 0.202619 - 0.194434 - 0.18281 - 0.168409 - 0.152009 - 0.134434 - 0.116489 - 0.0989001 - 0.0822709 - 0.0670553 - 0.0535496 - 0.0419003 - 0.0321229 - 0.0241295 - 0.017759 - 0.0128064 - 0.00904839 - 0.00626401 - 0.00424885 - 0.00282375 - 0.00183873 - 0.00117313 - 0.000733352 - 0.000449174 - 0.000269559 - 0.0001585 - 9.13153e-05 - 5.15458e-05 - 2.85089e-05 - 1.54491e-05 - 8.20284e-06 - 4.26738e-06 - 2.17518e-06 - 1.08634e-06 - 5.31587e-07 - 2.5487e-07 - 1.19729e-07 - 5.51086e-08 - 2.48527e-08 - 1.09816e-08 - 4.75449e-09 - 2.01706e-09 - 8.38997e-10 - 3.43464e-10 - 1.41906e-10 - 6.83484e-11 - 5.95613e-11 - 1.07176e-10 - 2.50595e-10 - 6.05453e-10 - 1.44505e-09 - 3.3834e-09 - 7.76313e-09 - 1.7453e-08 - 3.8445e-08 - 8.29752e-08 - 1.75466e-07 - 3.63559e-07 - 7.38063e-07 - 1.46808e-06 - 2.86115e-06 - 5.46349e-06 - 1.0222e-05 - 1.87387e-05 - 3.36572e-05 - 5.92318e-05 - 0.000102134 - 0.000172551 - 0.000285631 - 0.000463265 - 0.000736191 - 0.00114627 - 0.00174873 - 0.00261393 - 0.00382826 - 0.00549347 - 0.00772375 - 0.0106401 - 0.0143616 - 0.0189931 - 0.0246107 - 0.0312457 - 0.0388681 - 0.0473732 - 0.056573 - 0.0661947 - 0.075888 - 0.0852433 - 0.0938176 - 0.101168 - 0.106891 - 0.110657 - 0.11224 - 0.111547 - 0.108618 - 0.10363 - 0.096873 - 0.0887275 - 0.0796253 - 0.0700132 - 0.0603179 - 0.0509154 - 0.0421104 - 0.0341245 - 0.0270944 - 0.021078 - 0.0160663 - 0.0119989 - 0.00878015 - 0.00629506 - 0.00442216 - 0.00304373 - 0.00205265 - 0.00135631 - 0.000878095 - 0.000557007 - 0.000346191 - 0.000210818 - 0.000125788 - 7.35367e-05 - 4.21219e-05 - 2.364e-05 - 1.29994e-05 - 7.00387e-06 - 3.69733e-06 - 1.91239e-06 - 9.6917e-07 - 4.81239e-07 - 2.34131e-07 - 1.11608e-07 - 5.21273e-08 - 2.38547e-08 - 1.06959e-08 - 4.69895e-09 - 2.02265e-09 - 8.53052e-10 - 3.52508e-10 - 1.42724e-10 - 5.66193e-11 - 2.20073e-11 - 8.38123e-12 - 3.12741e-12 - 1.1434e-12 - 4.09589e-13 - 1.43759e-13 - 4.9438e-14 - 1.6658e-14 - 5.49945e-15 - 1.77891e-15 - 5.63801e-16 - 1.75079e-16 - 5.32697e-17 - 1.58804e-17 - 4.63854e-18 - 1.32751e-18 - 3.72246e-19 - 1.02273e-19 - 2.75313e-20 - 7.26155e-21 - 1.87659e-21 - 4.75167e-22 - 1.17886e-22 - 2.86561e-23 - 6.82699e-24 - 1.60225e-24 - 4.07911e-25 - 2.77213e-25 - 8.74879e-25 - 3.68199e-24 - 1.54686e-23 - 6.37332e-23 - 2.573e-22 - 1.01778e-21 - 3.94461e-21 - 1.49794e-20 - 5.57341e-20 - 2.03183e-19 - 7.25755e-19 - 2.53999e-18 - 8.70986e-18 - 2.92637e-17 - 9.63352e-17 - 3.10727e-16 - 9.81998e-16 - 3.04075e-15 - 9.22547e-15 - 2.74243e-14 - 7.98765e-14 - 2.27951e-13 - 6.37385e-13 - 1.74622e-12 - 4.68745e-12 - 1.23285e-11 - 3.17705e-11 - 8.02186e-11 - 1.98456e-10 - 4.81051e-10 - 1.1425e-09 - 2.65864e-09 - 6.06178e-09 - 1.35419e-08 - 2.96413e-08 - 6.357e-08 - 1.33581e-07 - 2.75029e-07 - 5.54814e-07 - 1.09662e-06 - 2.12374e-06 - 4.02982e-06 - 7.49218e-06 - 1.3648e-05 - 2.43594e-05 - 4.25994e-05 - 7.29926e-05 - 0.000122544 - 0.000201578 - 0.000324887 - 0.000513049 - 0.000793824 - 0.00120345 - 0.0017876 - 0.00260165 - 0.00370994 - 0.0051835 - 0.00709606 - 0.00951809 - 0.0125089 - 0.0161075 - 0.0203225 - 0.0251225 - 0.0304289 - 0.0361118 - 0.0419907 - 0.0478406 - 0.0534051 - 0.0584136 - 0.0626032 - 0.0657415 - 0.0676484 - 0.0682146 - 0.067413 - 0.0653031 - 0.0620266 - 0.0577961 - 0.0528776 - 0.0475695 - 0.042181 - 0.0370117 - 0.0323346 - 0.0283836 - 0.0253448 - 0.0233519 - 0.0224848 - 0.0227691 - 0.0241766 - 0.0266268 - 0.0299878 - 0.0340796 - 0.0386788 - 0.0435267 - 0.048341 - 0.0528317 - 0.0567191 - 0.0597539 - 0.0617366 - 0.062533 - 0.0620863 - 0.0604203 - 0.0576377 - 0.0539097 - 0.0494607 - 0.04455 - 0.0394516 - 0.0344361 - 0.0297549 - 0.0256293 - 0.0222441 - 0.0197445 - 0.0182362 - 0.0177867 - 0.0184265 - 0.020149 - 0.0229088 - 0.0266194 - 0.0311498 - 0.0363235 - 0.0419191 - 0.0476766 - 0.0533075 - 0.0585107 - 0.0629926 - 0.0664884 - 0.0687844 - 0.0697358 - 0.0692798 - 0.0674405 - 0.0643261 - 0.0601173 - 0.0550495 - 0.0493909 - 0.0434189 - 0.037398 - 0.0315613 - 0.0260975 - 0.0211437 - 0.0167841 - 0.0130543 - 0.0099482 - 0.00742802 - 0.00543424 - 0.0038953 - 0.00273577 - 0.00188259 - 0.00126931 - 0.00083853 - 0.000542757 - 0.000344214 - 0.000213889 - 0.000130222 - 7.76819e-05 - 4.54036e-05 - 2.60015e-05 - 1.45896e-05 - 8.02091e-06 - 4.32058e-06 - 2.28033e-06 - 1.1792e-06 - 5.97472e-07 - 2.96608e-07 - 1.44273e-07 - 6.87582e-08 - 3.21071e-08 - 1.46897e-08 - 6.58512e-09 - 2.89235e-09 - 1.24473e-09 - 5.24849e-10 - 2.16836e-10 - 8.77739e-11 - 3.48126e-11 - 1.35283e-11 - 5.15096e-12 - 1.92163e-12 - 7.02406e-13 - 2.51561e-13 - 8.82745e-14 - 3.03504e-14 - 1.02242e-14 - 3.37468e-15 - 1.09137e-15 - 3.45818e-16 - 1.07364e-16 - 3.26596e-17 - 9.73413e-18 - 2.84263e-18 - 8.13357e-19 - 2.28023e-19 - 6.26344e-20 - 1.68571e-20 - 4.4452e-21 - 1.14851e-21 - 2.90749e-22 - 7.21167e-23 - 1.75263e-23 - 4.17333e-24 - 9.73671e-25 - 2.22576e-25 - 4.98517e-26 - 1.09401e-26 - 2.35232e-27 - 4.95574e-28 - 1.02295e-28 - 2.06876e-29 - 4.09682e-30 - 7.9044e-31 - 1.41317e-31 - 1.02563e-32 - -2.61295e-32 - -5.24454e-32 - -8.90409e-32 - -1.4606e-31 - -2.34413e-31 - -3.68554e-31 - -5.67741e-31 - -8.56911e-31 - -1.26724e-30 - -1.83619e-30 - -2.60683e-30 - -3.62616e-30 - -4.94215e-30 - -6.59968e-30 - -8.63508e-30 - -1.107e-29 - -1.39048e-29 - -1.71127e-29 - -2.06352e-29 - -2.43802e-29 - -2.82229e-29 - -3.20112e-29 - -3.55747e-29 - -3.87362e-29 - -4.13265e-29 - -4.31995e-29 - -4.42453e-29 - -4.44013e-29 - -4.36582e-29 - -4.20614e-29 - -3.97065e-29 - -3.673e-29 - -3.32973e-29 - -2.95883e-29 - -2.57833e-29 - -2.20518e-29 - -1.85434e-29 - -1.53836e-29 - -1.26739e-29 - -1.04958e-29 - -8.91768e-30 - -8.00335e-30 - -7.82109e-30 - -8.451e-30 - -9.9896e-30 - -1.255e-29 - -1.62571e-29 - -2.12368e-29 - -2.75998e-29 - -3.54212e-29 - -4.4717e-29 - -5.54211e-29 - -6.73661e-29 - -8.02704e-29 - -9.37371e-29 - -1.07265e-28 - -1.20272e-28 - -1.32136e-28 - -1.4224e-28 - -1.50024e-28 - -1.55038e-28 - -1.56984e-28 - -1.55743e-28 - -1.51391e-28 - -1.44187e-28 - -1.34552e-28 - -1.23024e-28 - -1.10212e-28 - -9.67395e-29 - -8.31985e-29 - -7.01074e-29 - -5.78828e-29 - -4.68243e-29 - -3.71134e-29 - -2.88221e-29 - -2.1931e-29 - -1.63504e-29 - -1.19436e-29 - -8.54827e-30 - -5.99458e-30 - -4.11885e-30 - -2.77289e-30 - -1.82909e-30 - -1.18222e-30 - -7.48838e-31 - -4.65061e-31 - -2.83643e-31 - -1.70821e-31 - -1.03399e-31 - -6.63129e-32 - -5.08043e-32 - -5.32915e-32 - -7.49773e-32 - -1.22247e-31 - -2.07936e-31 - -3.53558e-31 - -5.92612e-31 - -9.75029e-31 - -1.57275e-30 - -2.48621e-30 - -3.85138e-30 - -5.84646e-30 - -8.69723e-30 - -1.26794e-29 - -1.81166e-29 - -2.53715e-29 - -3.483e-29 - -4.6877e-29 - -6.1864e-29 - -8.0073e-29 - -1.01678e-28 - -1.26712e-28 - -1.55044e-28 - -1.86374e-28 - -2.20251e-28 - -2.5611e-28 - -2.93335e-28 - -3.3133e-28 - -3.69599e-28 - -4.07804e-28 - -4.458e-28 - -4.83634e-28 - -5.21493e-28 - -5.59609e-28 - -5.98136e-28 - -6.37011e-28 - -6.75832e-28 - -7.13768e-28 - -7.4954e-28 - -7.81467e-28 - -8.07596e-28 - -8.25889e-28 - -8.34452e-28 - -8.31767e-28 - -8.16899e-28 - -7.89635e-28 - -7.50553e-28 - -7.00984e-28 - -6.42901e-28 - -5.78734e-28 - -5.11146e-28 - -4.42805e-28 - -3.76164e-28 - -3.13298e-28 - -2.55795e-28 - -2.04706e-28 - -1.6056e-28 - -1.23419e-28 - -9.29723e-29 - -6.86348e-29 - -4.96575e-29 - -3.52194e-29 - -2.45044e-29 - -1.67587e-29 - -1.13269e-29 - -7.67384e-30 - -5.3933e-30 - -4.211e-30 - -3.98532e-30 - -4.71e-30 - -6.52106e-30 - -9.70896e-30 - -1.47351e-29 - -2.22496e-29 - -3.31062e-29 - -4.83662e-29 - -6.9286e-29 - -9.72763e-29 - -1.33829e-28 - -1.80403e-28 - -2.38277e-28 - -3.08361e-28 - -3.90998e-28 - -4.85765e-28 - -5.9131e-28 - -7.05247e-28 - -8.24145e-28 - -9.43633e-28 - -1.05862e-27 - -1.16362e-27 - -1.25321e-27 - -1.32242e-27 - -1.36727e-27 - -1.38507e-27 - -1.37477e-27 - -1.33698e-27 - -1.27396e-27 - -1.18939e-27 - -1.088e-27 - -9.75147e-28 - -8.56344e-28 - -7.36824e-28 - -6.21177e-28 - -5.13103e-28 - -4.1527e-28 - -3.29301e-28 - -2.55854e-28 - -1.94773e-28 - -1.45279e-28 - -1.06173e-28 - -7.60255e-29 - -5.33387e-29 - -3.6666e-29 - -2.46957e-29 - -1.62973e-29 - -1.05377e-29 - -6.67597e-30 - -4.14399e-30 - -2.52035e-30 - -1.50189e-30 - -8.76909e-31 - -5.01657e-31 - -2.81187e-31 - -1.54426e-31 - -8.30968e-32 - -4.38111e-32 - -2.26319e-32 - -1.1455e-32 - -5.68073e-33 - -2.76027e-33 - -1.31413e-33 - -6.13033e-34 - -2.80267e-34 - -1.25719e-34 - -5.56871e-35 - -2.52147e-35 - -1.3648e-35 - -1.26756e-35 - -2.11808e-35 - -4.42522e-35 - -9.56567e-35 - -2.04659e-34 - -4.29803e-34 - -8.84677e-34 - -1.78427e-33 - -3.52598e-33 - -6.82709e-33 - -1.29518e-32 - -2.40746e-32 - -4.38455e-32 - -7.82399e-32 - -1.36794e-31 - -2.3434e-31 - -3.93332e-31 - -6.4686e-31 - -1.04231e-30 - -1.64559e-30 - -2.54556e-30 - -3.85818e-30 - -5.72951e-30 - -8.3366e-30 - -1.1885e-29 - -1.66014e-29 - -2.27209e-29 - -3.04681e-29 - -4.00314e-29 - -5.1534e-29 - -6.50016e-29 - -8.03324e-29 - -9.72734e-29 - -1.15408e-28 - -1.34157e-28 - -1.52801e-28 - -1.70521e-28 - -1.86451e-28 - -1.99751e-28 - -2.09678e-28 - -2.15654e-28 - -2.17338e-28 - -2.14699e-28 - -2.08278e-28 - -2.00365e-28 - -2.00826e-28 - -2.55057e-28 - -5.749e-28 - -2.13834e-27 - -9.37805e-27 - -4.19809e-26 - -1.85586e-25 - -8.0503e-25 - -3.42244e-24 - -1.42566e-23 - -5.81889e-23 - -2.32702e-22 - -9.11798e-22 - -3.50052e-21 - -1.31675e-20 - -4.85302e-20 - -1.7525e-19 - -6.20068e-19 - -2.1496e-18 - -7.30153e-18 - -2.43e-17 - -7.92383e-17 - -2.53164e-16 - -7.92509e-16 - -2.43077e-15 - -7.30501e-15 - -2.15097e-14 - -6.20561e-14 - -1.75417e-13 - -4.85843e-13 - -1.31843e-12 - -3.50554e-12 - -9.13248e-12 - -2.3311e-11 - -5.83e-11 - -1.42861e-10 - -3.43001e-10 - -8.0689e-10 - -1.85982e-09 - -4.20013e-09 - -9.29377e-09 - -2.01492e-08 - -4.28017e-08 - -8.90843e-08 - -1.81668e-07 - -3.62987e-07 - -7.10627e-07 - -1.3631e-06 - -2.56185e-06 - -4.71753e-06 - -8.51162e-06 - -1.50469e-05 - -2.60626e-05 - -4.4231e-05 - -7.35482e-05 - -0.000119827 - -0.000191281 - -0.000299177 - -0.000458481 - -0.000688416 - -0.00101278 - -0.00145989 - -0.00206186 - -0.00285323 - -0.00386857 - -0.00513927 - -0.00668942 - -0.00853125 - -0.0106604 - -0.0130518 - -0.0156569 - -0.0184025 - -0.0211927 - -0.0239128 - -0.026437 - -0.0286371 - -0.0303935 - -0.0316057 - -0.0322019 - -0.0321457 - -0.0314398 - -0.0301256 - -0.0282785 - -0.0260006 - -0.0234103 - -0.020631 - -0.0177807 - -0.0149619 - -0.0122546 - -0.00971227 - -0.00735997 - -0.00519609 - -0.00319598 - -0.00131746 - 0.000492548 - 0.00229074 - 0.00413016 - 0.00605343 - 0.00808678 - 0.0102356 - 0.0124818 - 0.0147831 - 0.0170751 - 0.0192757 - 0.0212916 - 0.0230263 - 0.0243896 - 0.0253062 - 0.0257239 - 0.0256186 - 0.0249975 - 0.0238982 - 0.0223855 - 0.0205448 - 0.0184745 - 0.0162773 - 0.0140516 - 0.0118852 - 0.00984971 - 0.00799793 - 0.0063631 - 0.00496017 - 0.00378845 - 0.00283506 - 0.00207875 - 0.00149341 - 0.00105122 - 0.000725027 - 0.000489976 - 0.000324492 - 0.000210664 - 0.00013421 - 8.41738e-05 - 5.24698e-05 - 3.34056e-05 - 2.3238e-05 - 1.98238e-05 - 2.23914e-05 - 3.14497e-05 - 4.88383e-05 - 7.7918e-05 - 0.000123889 - 0.000194214 - 0.000299116 - 0.000452071 - 0.000670232 - 0.00097465 - 0.00139016 - 0.00194481 - 0.00266861 - 0.00359163 - 0.00474135 - 0.00613926 - 0.00779712 - 0.00971309 - 0.0118682 - 0.0142236 - 0.0167199 - 0.0192776 - 0.0218001 - 0.0241795 - 0.0263033 - 0.0280635 - 0.0293652 - 0.0301353 - 0.030329 - 0.0299345 - 0.0289739 - 0.0275012 - 0.0255973 - 0.0233628 - 0.0209089 - 0.0183485 - 0.0157878 - 0.0133194 - 0.0110173 - 0.00893483 - 0.00710404 - 0.0055377 - 0.00423219 - 0.00317136 - 0.00233065 - 0.00168094 - 0.00119189 - 0.000834605 - 0.000583648 - 0.00041835 - 0.000323717 - 0.000291017 - 0.000318222 - 0.000410421 - 0.000580238 - 0.000848215 - 0.00124303 - 0.00180133 - 0.00256695 - 0.00358917 - 0.00491979 - 0.0066089 - 0.0086993 - 0.0112199 - 0.0141788 - 0.017556 - 0.0212986 - 0.0253171 - 0.0294858 - 0.0336472 - 0.0376203 - 0.0412128 - 0.0442363 - 0.0465224 - 0.0479383 - 0.0483994 - 0.0478778 - 0.046405 - 0.044069 - 0.0410051 - 0.0373834 - 0.0333931 - 0.0292262 - 0.0250625 - 0.0210578 - 0.0173356 - 0.013983 - 0.011051 - 0.00855731 - 0.00649248 - 0.00482638 - 0.00351536 - 0.00250876 - 0.00175427 - 0.00120201 - 0.00080715 - 0.000531421 - 0.000343513 - 0.000218867 - 0.00013901 - 9.07312e-05 - 6.52515e-05 - 5.75322e-05 - 6.57899e-05 - 9.1257e-05 - 0.000138186 - 0.000214067 - 0.000329998 - 0.000501116 - 0.000746956 - 0.00109158 - 0.00156332 - 0.00219384 - 0.00301654 - 0.00406401 - 0.00536461 - 0.00693838 - 0.00879257 - 0.0109172 - 0.0132813 - 0.015831 - 0.018489 - 0.0211571 - 0.0237211 - 0.0260585 - 0.028048 - 0.0295795 - 0.0305645 - 0.0309443 - 0.0306958 - 0.0298343 - 0.0284112 - 0.0265094 - 0.0242352 - 0.0217085 - 0.0190525 - 0.0163836 - 0.013804 - 0.0113955 - 0.00921728 - 0.00730479 - 0.00567218 - 0.00431547 - 0.00321695 - 0.00234961 - 0.00168146 - 0.00117899 - 0.00080998 - 0.000545223 - 0.000359593 - 0.000232372 - 0.000147128 - 9.12729e-05 - 5.54786e-05 - 3.30405e-05 - 1.92799e-05 - 1.1023e-05 - 6.1749e-06 - 3.38921e-06 - 1.82265e-06 - 9.60385e-07 - 4.9582e-07 - 2.50807e-07 - 1.24306e-07 - 6.03645e-08 - 2.87215e-08 - 1.33897e-08 - 6.11605e-09 - 2.73721e-09 - 1.20029e-09 - 5.15726e-10 - 2.17179e-10 - 8.97827e-11 - 3.68248e-11 - 1.59826e-11 - 9.75677e-12 - 1.25514e-11 - 2.62185e-11 - 6.09961e-11 - 1.42097e-10 - 3.25442e-10 - 7.30675e-10 - 1.60748e-09 - 3.46506e-09 - 7.31836e-09 - 1.51444e-08 - 3.07064e-08 - 6.10016e-08 - 1.18738e-07 - 2.26453e-07 - 4.23156e-07 - 7.74749e-07 - 1.38982e-06 - 2.44282e-06 - 4.20691e-06 - 7.09857e-06 - 1.17359e-05 - 1.90106e-05 - 3.01727e-05 - 4.69212e-05 - 7.14926e-05 - 0.000106731 - 0.000156119 - 0.000223747 - 0.000314193 - 0.000432288 - 0.000582755 - 0.000769726 - 0.000996145 - 0.00126312 - 0.0015693 - 0.00191031 - 0.00227844 - 0.00266262 - 0.00304871 - 0.00342027 - 0.0037596 - 0.00404912 - 0.00427283 - 0.00441781 - 0.00447545 - 0.00444224 - 0.00432021 - 0.00411666 - 0.00384346 - 0.00351591 - 0.00315133 - 0.00276754 - 0.00238148 - 0.00200803 - 0.00165923 - 0.00134382 - 0.00106726 - 0.000831969 - 0.000637896 - 0.000483181 - 0.000364871 - 0.000279608 - 0.000224215 - 0.000196152 - 0.00019381 - 0.000216651 - 0.000265178 - 0.000340754 - 0.000445276 - 0.000580727 - 0.000748638 - 0.000949492 - 0.00118215 - 0.00144335 - 0.00172735 - 0.00202585 - 0.00232812 - 0.00262156 - 0.0028924 - 0.00312677 - 0.00331187 - 0.00343707 - 0.00349494 - 0.003482 - 0.00339903 - 0.003251 - 0.00304661 - 0.00279739 - 0.00251667 - 0.00221838 - 0.00191595 - 0.00162131 - 0.00134427 - 0.00109206 - 0.00086924 - 0.000677908 - 0.000518011 - 0.000387832 - 0.000284502 - 0.000204486 - 0.000144006 - 9.93646e-05 - 6.7177e-05 - 4.44986e-05 - 2.88808e-05 - 1.83658e-05 - 1.14431e-05 - 6.98584e-06 - 4.17858e-06 - 2.44893e-06 - 1.40624e-06 - 7.91189e-07 - 4.36152e-07 - 2.35577e-07 - 1.24671e-07 - 6.46445e-08 - 3.28425e-08 - 1.63485e-08 - 7.97365e-09 - 3.81042e-09 - 1.78412e-09 - 8.18492e-10 - 3.67909e-10 - 1.62033e-10 - 6.99202e-11 - 2.95624e-11 - 1.22465e-11 - 4.97076e-12 - 1.97683e-12 - 7.70284e-13 - 2.94065e-13 - 1.09939e-13 - 4.00913e-14 - 1.37519e-14 - 2.83217e-15 - -5.1357e-15 - -1.99766e-14 - -5.99553e-14 - -1.72282e-13 - -4.83806e-13 - -1.3308e-12 - -3.58656e-12 - -9.4706e-12 - -2.45027e-11 - -6.21137e-11 - -1.54276e-10 - -3.75444e-10 - -8.95217e-10 - -2.09146e-09 - -4.78748e-09 - -1.07375e-08 - -2.35957e-08 - -5.08043e-08 - -1.07178e-07 - -2.21538e-07 - -4.48669e-07 - -8.90311e-07 - -1.73099e-06 - -3.29749e-06 - -6.15473e-06 - -1.12557e-05 - -2.01684e-05 - -3.54086e-05 - -6.09091e-05 - -0.000102658 - -0.000169527 - -0.000274298 - -0.000434855 - -0.000675463 - -0.00102801 - -0.00153295 - -0.00223973 - -0.00320628 - -0.00449721 - -0.00618047 - -0.00832218 - -0.0109797 - -0.0141932 - -0.0179765 - -0.0223084 - -0.0271249 - -0.032315 - -0.0377206 - -0.0431408 - -0.0483433 - -0.0530787 - -0.0571007 - -0.0601865 - -0.0621577 - -0.0628967 - -0.062359 - -0.0605773 - -0.0576586 - -0.0537732 - -0.0491391 - -0.0440019 - -0.0386137 - -0.0332143 - -0.0280153 - -0.0231898 - -0.0188668 - -0.0151321 - -0.012033 - -0.00958594 - -0.00778576 - -0.00661503 - -0.00605167 - -0.0060743 - -0.00666458 - -0.00780636 - -0.00948195 - -0.0116662 - -0.0143193 - -0.0173792 - -0.020756 - -0.024328 - -0.0279419 - -0.0314171 - -0.0345546 - -0.0371497 - -0.0390077 - -0.0399608 - -0.0398843 - -0.0387107 - -0.0364381 - -0.0331337 - -0.0289313 - -0.0240213 - -0.0186365 - -0.0130343 - -0.00747615 - -0.00220866 - 0.00255415 - 0.00664412 - 0.00994752 - 0.0124083 - 0.0140264 - 0.0148508 - 0.0149705 - 0.0145022 - 0.0135779 - 0.0123329 - 0.0108956 - 0.00937953 - 0.00787832 - 0.00646298 - 0.00518198 - 0.00406317 - 0.00311692 - 0.00234001 - 0.0017197 - 0.00123741 - 0.00087191 - 0.000601693 - 0.000406692 - 0.000269261 - 0.000174633 - 0.000110952 - 6.90568e-05 - 4.21022e-05 - 2.51345e-05 - 1.46728e-05 - 8.33422e-06 - 4.52151e-06 - 2.17167e-06 - 5.58957e-07 - -8.62176e-07 - -2.59036e-06 - -5.20476e-06 - -9.50482e-06 - -1.66806e-05 - -2.85308e-05 - -4.77436e-05 - -7.82481e-05 - -0.000125638 - -0.000197646 - -0.000304643 - -0.000460074 - -0.000680773 - -0.000986991 - -0.00140204 - -0.0019514 - -0.00266113 - -0.0035557 - -0.004655 - -0.00597107 - -0.00750448 - -0.00924115 - -0.0111498 - -0.013181 - -0.0152674 - -0.0173267 - -0.0192667 - -0.020991 - -0.0224077 - -0.0234368 - -0.0240179 - -0.0241162 - -0.0237257 - -0.0228701 - -0.0215999 - -0.0199882 - -0.0181231 - -0.0161 - -0.0140139 - -0.0119517 - -0.00998697 - -0.00817667 - -0.00655927 - -0.00515551 - -0.00397031 - -0.00299581 - -0.00221484 - -0.00160437 - -0.00113869 - -0.00079185 - -0.000539532 - -0.000360187 - -0.0002356 - -0.000150994 - -9.48162e-05 - -5.83366e-05 - -3.51671e-05 - -2.07715e-05 - -1.20209e-05 - -6.81621e-06 - -3.78692e-06 - -2.06142e-06 - -1.09949e-06 - -5.74622e-07 - -2.94344e-07 - -1.47969e-07 - -7.34581e-08 - -3.70781e-08 - -2.14022e-08 - -1.86896e-08 - -2.78684e-08 - -5.43091e-08 - -1.12445e-07 - -2.32243e-07 - -4.71751e-07 - -9.39625e-07 - -1.83401e-06 - -3.50752e-06 - -6.57263e-06 - -1.20674e-05 - -2.17084e-05 - -3.82629e-05 - -6.60791e-05 - -0.000111812 - -0.000185373 - -0.000301123 - -0.000479268 - -0.000747393 - -0.00114198 - -0.00170963 - -0.00250774 - -0.00360413 - -0.00507523 - -0.00700241 - -0.00946622 - -0.0125384 - -0.0162721 - -0.0206911 - -0.0257786 - -0.0314682 - -0.0376376 - -0.0441072 - -0.0506445 - -0.0569761 - -0.0628044 - -0.0678304 - -0.0717786 - -0.0744223 - -0.0756046 - -0.0752542 - -0.0733923 - -0.0701309 - -0.0656613 - -0.0602356 - -0.0541439 - -0.0476884 - -0.0411598 - -0.0348168 - -0.0288722 - -0.0234843 - -0.0187563 - -0.0147398 - -0.0114436 - -0.00884406 - -0.00689638 - -0.00554479 - -0.00473098 - -0.00439975 - -0.00450194 - -0.00499464 - -0.00583918 - -0.00699773 - -0.00842913 - -0.0100849 - -0.0119061 - -0.0138222 - -0.0157513 - -0.0176033 - -0.0192845 - -0.0207045 - -0.0217825 - -0.022455 - -0.0226813 - -0.0224474 - -0.0217673 - -0.0206815 - -0.0192529 - -0.017561 - -0.0156942 - -0.0137425 - -0.0117904 - -0.00991122 - -0.00816327 - -0.00658776 - -0.00520892 - -0.00403548 - -0.00306323 - -0.00227824 - -0.00166019 - -0.00118537 - -0.000829248 - -0.000568398 - -0.000381731 - -0.000251189 - -0.000161949 - -0.000102304 - -6.33209e-05 - -3.84005e-05 - -2.28172e-05 - -1.32839e-05 - -7.5775e-06 - -4.23509e-06 - -2.31919e-06 - -1.24436e-06 - -6.54176e-07 - -3.3696e-07 - -1.70059e-07 - -8.40926e-08 - -4.0743e-08 - -1.93412e-08 - -8.99607e-09 - -4.09976e-09 - -1.83063e-09 - -8.00902e-10 - -3.43317e-10 - -1.44194e-10 - -5.93386e-11 - -2.39256e-11 - -9.45207e-12 - -3.6587e-12 - -1.3876e-12 - -5.15629e-13 - -1.87736e-13 - -6.69724e-14 - -2.34089e-14 - -8.01681e-15 - -2.69004e-15 - -8.84411e-16 - -2.84896e-16 - -8.99196e-17 - -2.78074e-17 - -8.42563e-18 - -2.50139e-18 - -7.27608e-19 - -2.07372e-19 - -5.79081e-20 - -1.5844e-20 - -4.24746e-21 - -1.11565e-21 - -2.87122e-22 - -7.24003e-23 - -1.78876e-23 - -4.33011e-24 - -1.02703e-24 - -2.38674e-25 - -5.43454e-26 - -1.21243e-26 - -2.65027e-27 - -5.67621e-28 - -1.19114e-28 - -2.4491e-29 - -4.93384e-30 - -9.73871e-31 - -1.88345e-31 - -3.56898e-32 - -6.62629e-33 - -1.20541e-33 - -2.14849e-34 - -3.75207e-35 - -6.42015e-36 - -1.07636e-36 - -1.76809e-37 - -2.8457e-38 - -4.48756e-39 - -6.93375e-40 - -1.0497e-40 - -1.55702e-41 - -2.26289e-42 - -3.22233e-43 - -4.49585e-44 - -6.14598e-45 - -8.23204e-46 - -1.08034e-46 - -1.38915e-47 - -1.75016e-48 - -2.16043e-49 - -2.61301e-50 - -3.09655e-51 - -3.59544e-52 - -4.09037e-53 - -4.55943e-54 - -4.97961e-55 - -5.32865e-56 - -5.58696e-57 - -5.73945e-58 - -5.777e-59 - -5.69733e-60 - -5.50525e-61 - -5.21218e-62 - -4.83503e-63 - -4.39456e-64 - -3.91353e-65 - -3.41474e-66 - -2.91934e-67 - -2.44539e-68 - -2.00701e-69 - -1.61394e-70 - -1.27163e-71 - -9.81684e-73 - -7.4254e-74 - -5.50306e-75 - -3.996e-76 - -2.84305e-77 - -1.98189e-78 - -1.35367e-79 - -9.05901e-81 - -5.94e-82 - -3.81618e-83 - -2.4022e-84 - -1.48158e-85 - -8.9532e-87 - -5.30113e-88 - -3.07535e-89 - -1.74807e-90 - -9.73553e-92 - -5.31247e-93 - -2.84034e-94 - -1.48793e-95 - -7.63713e-97 - -3.84074e-98 - -1.8925e-99 --9.13682e-101 --4.32206e-102 --2.00319e-103 --9.09687e-105 - -4.0476e-106 --1.76458e-107 --7.53737e-109 --3.15454e-110 --1.29357e-111 --5.19732e-113 --2.04539e-114 --6.24659e-116 - 4.28841e-115 - 1.11058e-113 - 2.79859e-112 - 6.90971e-111 - 1.67155e-109 - 3.96199e-108 - 9.20121e-107 - 2.0937e-105 - 4.66788e-104 - 1.01967e-102 - 2.18243e-101 - 4.57675e-100 - 9.40394e-99 - 1.89321e-97 - 3.73445e-96 - 7.21755e-95 - 1.36675e-93 - 2.53587e-92 - 4.61001e-91 - 8.21131e-90 - 1.43305e-88 - 2.45044e-87 - 4.1055e-86 - 6.73945e-85 - 1.08398e-83 - 1.70825e-82 - 2.63767e-81 - 3.99049e-80 - 5.91519e-79 - 8.59108e-78 - 1.22254e-76 - 1.70458e-75 - 2.32866e-74 - 3.11698e-73 - 4.08787e-72 - 5.25288e-71 - 6.61355e-70 - 8.15848e-69 - 9.86098e-68 - 1.1678e-66 - 1.35504e-65 - 1.54054e-64 - 1.71606e-63 - 1.87296e-62 - 2.0029e-61 - 2.09859e-60 - 2.15444e-59 - 2.16709e-58 - 2.13578e-57 - 2.0624e-56 - 1.95131e-55 - 1.80891e-54 - 1.64302e-53 - 1.4622e-52 - 1.27499e-51 - 1.08929e-50 - 9.1184e-50 - 7.47876e-49 - 6.01004e-48 - 4.73219e-47 - 3.65077e-46 - 2.75958e-45 - 2.0438e-44 - 1.4831e-43 - 1.05448e-42 - 7.34589e-42 - 5.01404e-41 - 3.35326e-40 - 2.19727e-39 - 1.41071e-38 - 8.87415e-38 - 5.46958e-37 - 3.30307e-36 - 1.95442e-35 - 1.13307e-34 - 6.4362e-34 - 3.58213e-33 - 1.95339e-32 - 1.0437e-31 - 5.46381e-31 - 2.80256e-30 - 1.40848e-29 - 6.93557e-29 - 3.34619e-28 - 1.58182e-27 - 7.32655e-27 - 3.32491e-26 - 1.47841e-25 - 6.44094e-25 - 2.74941e-24 - 1.14992e-23 - 4.71227e-23 - 1.89204e-22 - 7.44335e-22 - 2.86908e-21 - 1.08356e-20 - 4.00961e-20 - 1.45374e-19 - 5.16427e-19 - 1.7975e-18 - 6.13004e-18 - 2.04831e-17 - 6.70602e-17 - 2.15115e-16 - 6.76105e-16 - 2.08206e-15 - 6.28218e-15 - 1.85722e-14 - 5.37965e-14 - 1.5268e-13 - 4.24566e-13 - 1.15677e-12 - 3.08805e-12 - 8.07715e-12 - 2.07e-11 - 5.19778e-11 - 1.2788e-10 - 3.08265e-10 - 7.28087e-10 - 1.68492e-09 - 3.82042e-09 - 8.48752e-09 - 1.84751e-08 - 3.94031e-08 - 8.23398e-08 - 1.68588e-07 - 3.38205e-07 - 6.64769e-07 - 1.28026e-06 - 2.4158e-06 - 4.46645e-06 - 8.09097e-06 - 1.43607e-05 - 2.49739e-05 - 4.25535e-05 - 7.10428e-05 - 0.00011621 - 0.000186252 - 0.00029248 - 0.000450017 - 0.00067842 - 0.00100209 - 0.00145027 - 0.0020565 - 0.00285723 - 0.00388954 - 0.00518787 - 0.00677979 - 0.00868121 - 0.0108913 - 0.0133881 - 0.0161248 - 0.0190285 - 0.0220016 - 0.0249252 - 0.027667 - 0.0300899 - 0.0320639 - 0.0334772 - 0.0342467 - 0.0343261 - 0.0337107 - 0.0324375 - 0.0305819 - 0.02825 - 0.0255687 - 0.0226744 - 0.0197016 - 0.0167727 - 0.0139907 - - - - - 5.09678e-31 - 6.34282e-31 - 7.73404e-31 - 9.23999e-31 - 1.08166e-30 - 1.24092e-30 - 1.39637e-30 - 1.54781e-30 - 1.72541e-30 - 2.117e-30 - 3.70688e-30 - 1.14251e-29 - 4.93576e-29 - 2.32683e-28 - 1.10079e-27 - 5.12755e-27 - 2.34251e-26 - 1.04876e-25 - 4.60071e-25 - 1.97749e-24 - 8.32801e-24 - 3.43641e-23 - 1.38933e-22 - 5.50355e-22 - 2.13608e-21 - 8.12322e-21 - 3.02675e-20 - 1.105e-19 - 3.9526e-19 - 1.38529e-18 - 4.75704e-18 - 1.60055e-17 - 5.27641e-17 - 1.70429e-16 - 5.3937e-16 - 1.6725e-15 - 5.0814e-15 - 1.51264e-14 - 4.41191e-14 - 1.26082e-13 - 3.53035e-13 - 9.68542e-13 - 2.60349e-12 - 6.85694e-12 - 1.76946e-11 - 4.47394e-11 - 1.10834e-10 - 2.69027e-10 - 6.39816e-10 - 1.49091e-09 - 3.40395e-09 - 7.61469e-09 - 1.66901e-08 - 3.58427e-08 - 7.5419e-08 - 1.55488e-07 - 3.14088e-07 - 6.21643e-07 - 1.2055e-06 - 2.29051e-06 - 4.26416e-06 - 7.77805e-06 - 1.3901e-05 - 2.4342e-05 - 4.17643e-05 - 7.02086e-05 - 0.000115641 - 0.000186625 - 0.000295098 - 0.000457192 - 0.000694014 - 0.00103222 - 0.00150424 - 0.00214782 - 0.00300479 - 0.00411877 - 0.00553169 - 0.00727922 - 0.00938532 - 0.0118563 - 0.0146753 - 0.0177977 - 0.0211483 - 0.0246221 - 0.0280875 - 0.0313934 - 0.0343799 - 0.0368904 - 0.0387854 - 0.0399557 - 0.0403326 - 0.0398957 - 0.0386751 - 0.036749 - 0.0342377 - 0.0312926 - 0.0280847 - 0.0247919 - 0.0215871 - 0.0186297 - 0.0160581 - 0.013987 - 0.0125051 - 0.0116762 - 0.0115397 - 0.0121108 - 0.0133811 - 0.0153167 - 0.0178562 - 0.0209094 - 0.0243561 - 0.0280473 - 0.0318096 - 0.0354525 - 0.0387797 - 0.041602 - 0.0437517 - 0.0450968 - 0.045552 - 0.0450867 - 0.043727 - 0.041553 - 0.03869 - 0.0352968 - 0.031551 - 0.027633 - 0.0237127 - 0.0199375 - 0.0164247 - 0.0132575 - 0.0104848 - 0.00812456 - 0.00616843 - 0.00458866 - 0.00334453 - 0.00238847 - 0.00167125 - 0.00114578 - 0.000769656 - 0.000506558 - 0.000326662 - 0.000206398 - 0.000127776 - 7.75048e-05 - 4.60623e-05 - 2.68225e-05 - 1.53035e-05 - 8.55494e-06 - 4.68577e-06 - 2.51468e-06 - 1.32227e-06 - 6.81232e-07 - 3.4388e-07 - 1.70081e-07 - 8.24215e-08 - 3.91348e-08 - 1.82063e-08 - 8.29885e-09 - 3.70639e-09 - 1.62189e-09 - 6.95389e-10 - 2.92132e-10 - 1.2026e-10 - 4.85509e-11 - 1.93322e-11 - 7.90272e-12 - 4.15729e-12 - 4.69603e-12 - 1.00995e-11 - 2.54898e-11 - 6.48543e-11 - 1.62322e-10 - 3.9828e-10 - 9.57567e-10 - 2.25575e-09 - 5.20657e-09 - 1.17747e-08 - 2.60905e-08 - 5.66438e-08 - 1.20492e-07 - 2.51133e-07 - 5.12843e-07 - 1.02613e-06 - 2.01167e-06 - 3.8641e-06 - 7.27238e-06 - 1.34104e-05 - 2.42295e-05 - 4.28927e-05 - 7.43975e-05 - 0.000126436 - 0.000210533 - 0.000343484 - 0.000549073 - 0.000859983 - 0.00131974 - 0.00198436 - 0.00292342 - 0.00421986 - 0.00596819 - 0.00827035 - 0.011229 - 0.0149381 - 0.019471 - 0.0248666 - 0.0311158 - 0.0381491 - 0.0458271 - 0.0539385 - 0.062203 - 0.0702847 - 0.0778121 - 0.0844055 - 0.0897079 - 0.0934174 - 0.0953152 - 0.095287 - 0.0933346 - 0.0895758 - 0.0842323 - 0.0776088 - 0.0700637 - 0.0619781 - 0.0537245 - 0.0456401 - 0.0380067 - 0.0310392 - 0.0248817 - 0.0196118 - 0.0152499 - 0.0117717 - 0.00912236 - 0.00722982 - 0.00601571 - 0.00540334 - 0.0053222 - 0.00570923 - 0.00650744 - 0.00766253 - 0.00911876 - 0.0108148 - 0.0126804 - 0.0146355 - 0.0165902 - 0.0184485 - 0.020113 - 0.0214914 - 0.022504 - 0.0230902 - 0.0232141 - 0.022868 - 0.0220728 - 0.020876 - 0.0193474 - 0.0175725 - 0.0156453 - 0.0136615 - 0.0117119 - 0.00987887 - 0.00823454 - 0.00684165 - 0.00575724 - 0.00503837 - 0.00474935 - 0.00496969 - 0.00580156 - 0.00737575 - 0.00985451 - 0.0134302 - 0.0183181 - 0.0247429 - 0.0329181 - 0.0430191 - 0.0551515 - 0.0693182 - 0.0853881 - 0.103073 - 0.121914 - 0.141292 - 0.160443 - 0.178512 - 0.194604 - 0.207861 - 0.217536 - 0.223062 - 0.224109 - 0.220611 - 0.212781 - 0.201084 - 0.18619 - 0.168917 - 0.15015 - 0.130773 - 0.111595 - 0.0933057 - 0.0764379 - 0.0613545 - 0.0482527 - 0.037182 - 0.0280725 - 0.0207666 - 0.0150518 - 0.0106892 - 0.00743777 - 0.00507078 - 0.00338723 - 0.00221692 - 0.00142165 - 0.000893251 - 0.000549908 - 0.000331699 - 0.000196035 - 0.000113517 - 6.44059e-05 - 3.58036e-05 - 1.95013e-05 - 1.04073e-05 - 5.4419e-06 - 2.78804e-06 - 1.39954e-06 - 6.88345e-07 - 3.31715e-07 - 1.56625e-07 - 7.24592e-08 - 3.28445e-08 - 1.45871e-08 - 6.34764e-09 - 2.7064e-09 - 1.1306e-09 - 4.62768e-10 - 1.8559e-10 - 7.29258e-11 - 2.80766e-11 - 1.05912e-11 - 3.91457e-12 - 1.41762e-12 - 5.03003e-13 - 1.74872e-13 - 5.95668e-14 - 1.98805e-14 - 6.50108e-15 - 2.08296e-15 - 6.53904e-16 - 2.01133e-16 - 6.06164e-17 - 1.78992e-17 - 5.17861e-18 - 1.46801e-18 - 4.07741e-19 - 1.10962e-19 - 2.95871e-20 - 7.72981e-21 - 1.97881e-21 - 4.96994e-22 - 1.2513e-22 - 4.27973e-23 - 6.25412e-23 - 2.27169e-22 - 8.96772e-22 - 3.49045e-21 - 1.3316e-20 - 4.97752e-20 - 1.82301e-19 - 6.54186e-19 - 2.30012e-18 - 7.92386e-18 - 2.67461e-17 - 8.84545e-17 - 2.86627e-16 - 9.1002e-16 - 2.83088e-15 - 8.62837e-15 - 2.57675e-14 - 7.5397e-14 - 2.16158e-13 - 6.07193e-13 - 1.67116e-12 - 4.50658e-12 - 1.19073e-11 - 3.08258e-11 - 7.81904e-11 - 1.94325e-10 - 4.73197e-10 - 1.12899e-09 - 2.63923e-09 - 6.04507e-09 - 1.35663e-08 - 2.98304e-08 - 6.42676e-08 - 1.35663e-07 - 2.80589e-07 - 5.6861e-07 - 1.12901e-06 - 2.19641e-06 - 4.18667e-06 - 7.81917e-06 - 1.43083e-05 - 2.5654e-05 - 4.50668e-05 - 7.75704e-05 - 0.000130819 - 0.000216165 - 0.000349973 - 0.000555164 - 0.000862869 - 0.00131403 - 0.00196066 - 0.00286641 - 0.00410592 - 0.00576262 - 0.00792443 - 0.0106771 - 0.0140955 - 0.0182326 - 0.0231079 - 0.0286958 - 0.0349164 - 0.0416297 - 0.0486353 - 0.0556796 - 0.0624697 - 0.0686941 - 0.0740496 - 0.07827 - 0.0811551 - 0.0825952 - 0.0825891 - 0.0812525 - 0.0788147 - 0.0756052 - 0.0720296 - 0.068538 - 0.0655882 - 0.0636076 - 0.0629552 - 0.0638897 - 0.0665429 - 0.0709037 - 0.0768119 - 0.0839642 - 0.0919334 - 0.100197 - 0.108179 - 0.115294 - 0.120995 - 0.124823 - 0.126445 - 0.125678 - 0.122506 - 0.117071 - 0.10966 - 0.100669 - 0.0905633 - 0.0798348 - 0.0689607 - 0.0583676 - 0.0484063 - 0.0393372 - 0.0313261 - 0.0244505 - 0.0187122 - 0.0140553 - 0.0103846 - 0.00758494 - 0.00553752 - 0.00413347 - 0.00328399 - 0.00292721 - 0.00303184 - 0.00359822 - 0.00465668 - 0.00626355 - 0.00849453 - 0.0114355 - 0.0151707 - 0.019769 - 0.0252679 - 0.0316588 - 0.0388727 - 0.0467703 - 0.0551377 - 0.0636899 - 0.0720829 - 0.0799342 - 0.08685 - 0.092458 - 0.0964398 - 0.0985609 - 0.0986938 - 0.0968305 - 0.0930832 - 0.0876733 - 0.0809096 - 0.0731593 - 0.0648151 - 0.0562625 - 0.0478519 - 0.0398764 - 0.0325589 - 0.0260472 - 0.0204168 - 0.0156802 - 0.0117992 - 0.00869946 - 0.00628445 - 0.00444814 - 0.0030848 - 0.0020961 - 0.00139552 - 0.00091032 - 0.000581822 - 0.000364354 - 0.000223559 - 0.0001344 - 7.91667e-05 - 4.56902e-05 - 2.5837e-05 - 1.43154e-05 - 7.77184e-06 - 4.13516e-06 - 2.15827e-06 - 1.10966e-06 - 5.72705e-07 - 3.20383e-07 - 2.41707e-07 - 3.04402e-07 - 5.4587e-07 - 1.09169e-06 - 2.20877e-06 - 4.41012e-06 - 8.641e-06 - 1.65944e-05 - 3.12269e-05 - 5.75758e-05 - 0.000104013 - 0.000184109 - 0.0003193 - 0.000542574 - 0.00090335 - 0.00147364 - 0.00235538 - 0.00368866 - 0.00565995 - 0.0085093 - 0.0125346 - 0.0180912 - 0.0255834 - 0.0354476 - 0.048123 - 0.0640111 - 0.0834246 - 0.10653 - 0.133286 - 0.163393 - 0.196254 - 0.230963 - 0.266319 - 0.300885 - 0.333069 - 0.361247 - 0.383895 - 0.399721 - 0.407791 - 0.407621 - 0.399219 - 0.383092 - 0.36019 - 0.331816 - 0.299502 - 0.264874 - 0.229517 - 0.194862 - 0.162098 - 0.132119 - 0.105509 - 0.0825559 - 0.0632915 - 0.0475422 - 0.0349905 - 0.0252324 - 0.017828 - 0.012342 - 0.00837148 - 0.00556362 - 0.00362284 - 0.00231142 - 0.00144492 - 0.000885006 - 0.000531112 - 0.000312293 - 0.000179918 - 0.000101561 - 5.61709e-05 - 3.04393e-05 - 1.6162e-05 - 8.408e-06 - 4.28575e-06 - 2.14042e-06 - 1.04738e-06 - 5.0217e-07 - 2.35902e-07 - 1.0858e-07 - 4.89673e-08 - 2.16371e-08 - 9.36772e-09 - 3.97414e-09 - 1.65288e-09 - 6.76192e-10 - 2.78154e-10 - 1.30846e-10 - 1.07446e-10 - 1.85993e-10 - 4.31066e-10 - 1.04002e-09 - 2.48173e-09 - 5.81045e-09 - 1.33319e-08 - 2.99725e-08 - 6.60228e-08 - 1.42496e-07 - 3.01333e-07 - 6.24351e-07 - 1.2675e-06 - 2.52117e-06 - 4.91355e-06 - 9.38262e-06 - 1.75546e-05 - 3.21805e-05 - 5.78006e-05 - 0.000101721 - 0.000175397 - 0.000296328 - 0.000490523 - 0.00079558 - 0.00126428 - 0.00196853 - 0.00300315 - 0.00448898 - 0.00657439 - 0.00943411 - 0.0132642 - 0.0182726 - 0.0246636 - 0.0326174 - 0.0422648 - 0.0536592 - 0.0667494 - 0.0813555 - 0.0971547 - 0.113678 - 0.130325 - 0.146391 - 0.161116 - 0.17374 - 0.183568 - 0.190034 - 0.192754 - 0.191563 - 0.186533 - 0.177966 - 0.166363 - 0.152375 - 0.136743 - 0.120236 - 0.103586 - 0.0874387 - 0.0723175 - 0.058603 - 0.0465301 - 0.036198 - 0.0275912 - 0.0206061 - 0.0150784 - 0.0108107 - 0.00759432 - 0.00522709 - 0.00352508 - 0.00232924 - 0.00150798 - 0.000956565 - 0.000594525 - 0.000362045 - 0.000216019 - 0.000126287 - 7.23372e-05 - 4.05977e-05 - 2.23243e-05 - 1.2028e-05 - 6.34954e-06 - 3.2842e-06 - 1.66439e-06 - 8.26447e-07 - 4.0208e-07 - 1.91667e-07 - 8.95198e-08 - 4.09664e-08 - 1.83685e-08 - 8.06966e-09 - 3.47355e-09 - 1.46497e-09 - 6.05372e-10 - 2.45105e-10 - 9.7234e-11 - 3.77939e-11 - 1.43933e-11 - 5.37079e-12 - 1.9636e-12 - 7.03401e-13 - 2.46882e-13 - 8.49013e-14 - 2.86072e-14 - 9.44439e-15 - 3.05498e-15 - 9.68234e-16 - 3.00669e-16 - 9.14817e-17 - 2.7272e-17 - 7.96591e-18 - 2.27978e-18 - 6.39307e-19 - 1.7579e-19 - 4.79091e-20 - 1.49851e-20 - 1.3075e-20 - 3.86379e-20 - 1.42464e-19 - 5.24334e-19 - 1.89316e-18 - 6.6979e-18 - 2.32182e-17 - 7.886e-17 - 2.62435e-16 - 8.55703e-16 - 2.73377e-15 - 8.55729e-15 - 2.62451e-14 - 7.88674e-14 - 2.32211e-13 - 6.69893e-13 - 1.8935e-12 - 5.24398e-12 - 1.42297e-11 - 3.78324e-11 - 9.85533e-11 - 2.51544e-10 - 6.29065e-10 - 1.54139e-09 - 3.70056e-09 - 8.70479e-09 - 2.00626e-08 - 4.53056e-08 - 1.00243e-07 - 2.17317e-07 - 4.61603e-07 - 9.60688e-07 - 1.95899e-06 - 3.914e-06 - 7.66208e-06 - 1.46964e-05 - 2.76192e-05 - 5.08571e-05 - 9.17551e-05 - 0.000162199 - 0.000280936 - 0.00047677 - 0.000792781 - 0.00129164 - 0.00206196 - 0.0032253 - 0.00494329 - 0.00742376 - 0.0109245 - 0.015753 - 0.0222597 - 0.0308241 - 0.0418311 - 0.0556391 - 0.0725392 - 0.0927117 - 0.116181 - 0.142782 - 0.172134 - 0.203646 - 0.23654 - 0.269907 - 0.302784 - 0.334251 - 0.36353 - 0.390069 - 0.413606 - 0.434188 - 0.452139 - 0.467992 - 0.482371 - 0.495855 - 0.508845 - 0.521448 - 0.533411 - 0.54411 - 0.552604 - 0.557745 - 0.558334 - 0.553292 - 0.541826 - 0.523566 - 0.49865 - 0.46775 - 0.432034 - 0.393086 - 0.35278 - 0.313139 - 0.2762 - 0.243885 - 0.2179 - 0.199659 - 0.19023 - 0.190292 - 0.200109 - 0.219505 - 0.247844 - 0.284017 - 0.326452 - 0.373144 - 0.421726 - 0.469578 - 0.513978 - 0.552279 - 0.582113 - 0.601568 - 0.60936 - 0.604932 - 0.588498 - 0.561014 - 0.524074 - 0.479755 - 0.430422 - 0.378526 - 0.32641 - 0.276152 - 0.229449 - 0.187561 - 0.151296 - 0.121041 - 0.0968234 - 0.0783825 - 0.0652508 - 0.0568241 - 0.052421 - 0.0513267 - 0.0528233 - 0.0562096 - 0.0608144 - 0.0660069 - 0.0712085 - 0.0759058 - 0.0796663 - 0.0821531 - 0.083138 - 0.0825095 - 0.0802724 - 0.0765404 - 0.0715192 - 0.0654835 - 0.0587491 - 0.0516444 - 0.0444831 - 0.0375424 - 0.031047 - 0.0251611 - 0.0199871 - 0.0155706 - 0.0119103 - 0.00897034 - 0.00669384 - 0.00501675 - 0.00387995 - 0.00323985 - 0.00307697 - 0.00340228 - 0.00426134 - 0.00573584 - 0.00794229 - 0.011027 - 0.015157 - 0.0205062 - 0.0272364 - 0.0354751 - 0.0452892 - 0.0566592 - 0.0694565 - 0.0834268 - 0.0981838 - 0.113218 - 0.127916 - 0.141604 - 0.153589 - 0.163224 - 0.169959 - 0.173396 - 0.17333 - 0.169763 - 0.162911 - 0.153177 - 0.141116 - 0.127378 - 0.112654 - 0.0976202 - 0.0828835 - 0.0689498 - 0.0561998 - 0.0448822 - 0.0351196 - 0.0269254 - 0.0202261 - 0.0148867 - 0.0107355 - 0.00758544 - 0.00525143 - 0.00356214 - 0.00236745 - 0.00154166 - 0.000983633 - 0.000614914 - 0.000376646 - 0.000226043 - 0.000132922 - 7.65918e-05 - 4.32595e-05 - 2.39797e-05 - 1.31129e-05 - 7.21922e-06 - 4.30366e-06 - 3.33996e-06 - 4.01773e-06 - 6.68526e-06 - 1.24923e-05 - 2.37754e-05 - 4.47731e-05 - 8.28123e-05 - 0.000150164 - 0.000266831 - 0.000464579 - 0.000792543 - 0.00132472 - 0.00216951 - 0.00348126 - 0.0054733 - 0.00843136 - 0.0127257 - 0.0188194 - 0.0272687 - 0.0387134 - 0.053851 - 0.0733945 - 0.09801 - 0.128237 - 0.164397 - 0.206496 - 0.254135 - 0.306448 - 0.362063 - 0.41913 - 0.47539 - 0.528309 - 0.575259 - 0.613727 - 0.641541 - 0.657068 - 0.659377 - 0.648327 - 0.624588 - 0.589567 - 0.545274 - 0.494136 - 0.438774 - 0.381789 - 0.325571 - 0.272154 - 0.223122 - 0.179578 - 0.142167 - 0.111134 - 0.086417 - 0.0677452 - 0.0547365 - 0.04698 - 0.0440953 - 0.0457651 - 0.0517401 - 0.0618192 - 0.075809 - 0.0934719 - 0.114469 - 0.138309 - 0.164311 - 0.191592 - 0.21908 - 0.245557 - 0.269731 - 0.29033 - 0.306206 - 0.316433 - 0.320401 - 0.317866 - 0.308982 - 0.29428 - 0.274615 - 0.251088 - 0.224938 - 0.197441 - 0.169805 - 0.143086 - 0.118136 - 0.0955664 - 0.0757468 - 0.0588247 - 0.0447603 - 0.0333705 - 0.0243763 - 0.0174466 - 0.0122347 - 0.00840639 - 0.00565932 - 0.00373302 - 0.0024127 - 0.001528 - 0.000948448 - 0.000577417 - 0.000345638 - 0.0002051 - 0.000123843 - 8.18518e-05 - 6.81308e-05 - 7.89389e-05 - 0.00011715 - 0.000192719 - 0.000324255 - 0.000541737 - 0.000890371 - 0.00143553 - 0.00226859 - 0.00351311 - 0.00533082 - 0.00792609 - 0.0115475 - 0.0164852 - 0.0230612 - 0.0316131 - 0.0424685 - 0.0559121 - 0.0721461 - 0.0912487 - 0.113135 - 0.137526 - 0.163937 - 0.191679 - 0.219893 - 0.247604 - 0.273794 - 0.297493 - 0.317866 - 0.334291 - 0.346416 - 0.354179 - 0.357793 - 0.357701 - 0.354498 - 0.348835 - 0.34133 - 0.332489 - 0.322651 - 0.311975 - 0.30045 - 0.287942 - 0.274258 - 0.25922 - 0.242722 - 0.224784 - 0.205571 - 0.185391 - 0.164673 - 0.143916 - 0.123646 - 0.104359 - 0.0864807 - 0.0703312 - 0.0561129 - 0.0439079 - 0.0336895 - 0.0253425 - 0.0186881 - 0.0135096 - 0.00957556 - 0.00665938 - 0.00455333 - 0.00307793 - 0.00208716 - 0.00147068 - 0.00115451 - 0.00110122 - 0.00131073 - 0.0018224 - 0.00271851 - 0.00412898 - 0.00623645 - 0.00928047 - 0.0135588 - 0.0194236 - 0.0272705 - 0.0375177 - 0.0505746 - 0.0667991 - 0.0864465 - 0.109613 - 0.13618 - 0.165769 - 0.19771 - 0.231042 - 0.26454 - 0.296775 - 0.326212 - 0.351326 - 0.370729 - 0.383301 - 0.388294 - 0.385406 - 0.374811 - 0.357144 - 0.333435 - 0.305011 - 0.273374 - 0.240069 - 0.206562 - 0.174142 - 0.143844 - 0.116417 - 0.0923168 - 0.0717266 - 0.054603 - 0.0407277 - 0.0297646 - 0.0213131 - 0.0149531 - 0.010279 - 0.00692323 - 0.00456881 - 0.00295416 - 0.00187155 - 0.00116173 - 0.000706559 - 0.000421043 - 0.000245834 - 0.000140635 - 7.88285e-05 - 4.32922e-05 - 2.32955e-05 - 1.22821e-05 - 6.34466e-06 - 3.2113e-06 - 1.59255e-06 - 7.73819e-07 - 3.68407e-07 - 1.71859e-07 - 7.85727e-08 - 3.525e-08 - 1.56255e-08 - 7.1029e-09 - 3.90724e-09 - 3.74228e-09 - 6.36823e-09 - 1.3367e-08 - 2.89202e-08 - 6.18851e-08 - 1.29968e-07 - 2.67519e-07 - 5.39551e-07 - 1.06623e-06 - 2.06446e-06 - 3.91651e-06 - 7.27996e-06 - 1.32585e-05 - 2.36592e-05 - 4.13656e-05 - 7.08625e-05 - 0.000118941 - 0.000195606 - 0.000315187 - 0.000497615 - 0.000769759 - 0.00116668 - 0.00173256 - 0.00252093 - 0.00359392 - 0.00502012 - 0.00687063 - 0.00921332 - 0.0121052 - 0.0155835 - 0.019656 - 0.0242919 - 0.0294147 - 0.0348984 - 0.0405679 - 0.0462058 - 0.0515641 - 0.0563813 - 0.0604032 - 0.0634047 - 0.0652108 - 0.0657135 - 0.0648824 - 0.0627676 - 0.0594951 - 0.055254 - 0.0502786 - 0.044827 - 0.0391591 - 0.0335168 - 0.028108 - 0.0230958 - 0.018594 - 0.0146673 - 0.0113361 - 0.00858452 - 0.00636947 - 0.0046305 - 0.0032983 - 0.00230191 - 0.00157407 - 0.00105462 - 0.000692316 - 0.000445298 - 0.000280629 - 0.000173282 - 0.000104836 - 6.21444e-05 - 3.60937e-05 - 2.05399e-05 - 1.14526e-05 - 6.25667e-06 - 3.34904e-06 - 1.75645e-06 - 9.02581e-07 - 4.5444e-07 - 2.24189e-07 - 1.08378e-07 - 5.13693e-08 - 2.39435e-08 - 1.11485e-08 - 5.59702e-09 - 3.92679e-09 - 5.09955e-09 - 9.86095e-09 - 2.114e-08 - 4.55587e-08 - 9.66698e-08 - 2.0116e-07 - 4.10205e-07 - 8.19618e-07 - 1.60458e-06 - 3.07786e-06 - 5.78459e-06 - 1.06521e-05 - 1.9219e-05 - 3.39755e-05 - 5.88488e-05 - 9.98725e-05 - 0.00016607 - 0.000270566 - 0.000431909 - 0.000675535 - 0.00103524 - 0.00155443 - 0.00228684 - 0.0032964 - 0.00465564 - 0.00644253 - 0.00873514 - 0.0116043 - 0.0151046 - 0.0192634 - 0.024071 - 0.0294708 - 0.035353 - 0.0415526 - 0.0478528 - 0.053995 - 0.059695 - 0.0646638 - 0.0686316 - 0.0713724 - 0.0727252 - 0.0726101 - 0.0710369 - 0.068105 - 0.0639941 - 0.058949 - 0.0532582 - 0.0472318 - 0.0411794 - 0.035392 - 0.0301294 - 0.0256128 - 0.0220232 - 0.0195037 - 0.0181646 - 0.0180873 - 0.0193273 - 0.0219132 - 0.025842 - 0.0310708 - 0.0375062 - 0.0449944 - 0.0533132 - 0.0621703 - 0.0712086 - 0.0800204 - 0.0881704 - 0.0952259 - 0.10079 - 0.104537 - 0.106239 - 0.105792 - 0.10322 - 0.0986778 - 0.0924299 - 0.0848289 - 0.0762804 - 0.0672077 - 0.058018 - 0.049073 - 0.0406687 - 0.0330228 - 0.0262727 - 0.0204801 - 0.0156422 - 0.0117058 - 0.00858298 - 0.00616616 - 0.00434041 - 0.00299359 - 0.00202311 - 0.00133989 - 0.000870016 - 0.000554601 - 0.00034854 - 0.000218741 - 0.000142232 - 0.000104541 - 9.8566e-05 - 0.000124155 - 0.000188521 - 0.000307637 - 0.000508725 - 0.000833915 - 0.00134509 - 0.00212979 - 0.00330782 - 0.00503791 - 0.00752352 - 0.0110162 - 0.0158148 - 0.0222592 - 0.0307153 - 0.0415515 - 0.0551058 - 0.071643 - 0.0913074 - 0.114073 - 0.1397 - 0.167701 - 0.197328 - 0.227588 - 0.257282 - 0.285075 - 0.309593 - 0.329531 - 0.34377 - 0.35148 - 0.352196 - 0.345872 - 0.332879 - 0.313973 - 0.29022 - 0.262897 - 0.233379 - 0.203026 - 0.173082 - 0.144596 - 0.118376 - 0.0949657 - 0.0746562 - 0.057512 - 0.0434158 - 0.0321176 - 0.0232853 - 0.0165482 - 0.0115343 - 0.00789618 - 0.00532826 - 0.00357577 - 0.00243735 - 0.00176332 - 0.00145159 - 0.00144239 - 0.00171284 - 0.00227175 - 0.00315462 - 0.00441846 - 0.0061359 - 0.00838817 - 0.0112563 - 0.0148106 - 0.019099 - 0.0241341 - 0.0298819 - 0.0362517 - 0.0430913 - 0.0501867 - 0.0572696 - 0.064032 - 0.0701466 - 0.0752928 - 0.0791839 - 0.0815939 - 0.0823787 - 0.0814908 - 0.0789841 - 0.075008 - 0.069793 - 0.0636288 - 0.0568371 - 0.0497447 - 0.0426579 - 0.0358416 - 0.0295062 - 0.0238 - 0.0188094 - 0.0145651 - 0.0110507 - 0.00821493 - 0.00598368 - 0.00427072 - 0.00298721 - 0.00204851 - 0.00137888 - 0.000914107 - 0.0006025 - 0.000404956 - 0.000294455 - 0.000255581 - 0.000284413 - 0.000389119 - 0.00059143 - 0.000929092 - 0.0014592 - 0.00226219 - 0.00344601 - 0.00514967 - 0.00754534 - 0.0108375 - 0.0152582 - 0.0210566 - 0.0284828 - 0.0377642 - 0.0490773 - 0.0625148 - 0.0780522 - 0.0955182 - 0.114573 - 0.134702 - 0.155224 - 0.175321 - 0.194088 - 0.210595 - 0.223966 - 0.233452 - 0.238502 - 0.238817 - 0.234375 - 0.225439 - 0.212528 - 0.196368 - 0.177823 - 0.157822 - 0.13728 - 0.117031 - 0.0977797 - 0.0800658 - 0.0642531 - 0.0505343 - 0.0389512 - 0.0294236 - 0.0217826 - 0.0158036 - 0.0112366 - 0.00782971 - 0.00534666 - 0.00357803 - 0.00234654 - 0.00150811 - 0.000949846 - 0.000586259 - 0.0003546 - 0.000210184 - 0.000122087 - 6.94932e-05 - 3.87633e-05 - 2.11886e-05 - 1.13497e-05 - 5.95747e-06 - 3.06434e-06 - 1.54456e-06 - 7.62898e-07 - 3.69248e-07 - 1.75129e-07 - 8.13929e-08 - 3.7068e-08 - 1.65423e-08 - 7.23393e-09 - 3.0998e-09 - 1.30158e-09 - 5.35535e-10 - 2.15914e-10 - 8.52996e-11 - 3.30208e-11 - 1.25257e-11 - 4.65589e-12 - 1.69628e-12 - 6.07133e-13 - 2.17868e-13 - 9.17968e-14 - 8.26531e-14 - 1.80758e-13 - 5.04118e-13 - 1.4277e-12 - 3.9783e-12 - 1.0868e-11 - 2.90954e-11 - 7.63304e-11 - 1.96231e-10 - 4.94351e-10 - 1.2204e-09 - 2.95232e-09 - 6.9988e-09 - 1.62586e-08 - 3.70116e-08 - 8.25642e-08 - 1.80486e-07 - 3.86628e-07 - 8.11596e-07 - 1.66949e-06 - 3.36533e-06 - 6.64764e-06 - 1.28678e-05 - 2.44085e-05 - 4.53704e-05 - 8.26422e-05 - 0.000147512 - 0.000258019 - 0.000442253 - 0.000742825 - 0.00122264 - 0.001972 - 0.00311681 - 0.00482735 - 0.00732661 - 0.0108966 - 0.0158809 - 0.0226805 - 0.0317412 - 0.04353 - 0.0584987 - 0.0770365 - 0.0994122 - 0.125711 - 0.155777 - 0.189156 - 0.225077 - 0.262443 - 0.299868 - 0.335752 - 0.368386 - 0.396083 - 0.417326 - 0.430901 - 0.436023 - 0.432407 - 0.420307 - 0.400488 - 0.374157 - 0.34285 - 0.308297 - 0.27227 - 0.236449 - 0.202305 - 0.171016 - 0.143419 - 0.12 - 0.10091 - 0.0860138 - 0.0749435 - 0.0671671 - 0.0620535 - 0.0589346 - 0.0571569 - 0.0561247 - 0.0553308 - 0.0543762 - 0.052979 - 0.0509737 - 0.0483016 - 0.0449951 - 0.0411568 - 0.0369364 - 0.0325078 - 0.0280481 - 0.0237199 - 0.0196589 - 0.0159663 - 0.0127065 - 0.00990863 - 0.00757099 - 0.00566812 - 0.00415785 - 0.00298841 - 0.00210451 - 0.00145211 - 0.000981719 - 0.000650297 - 0.00042206 - 0.000268394 - 0.000167228 - 0.00010209 - 6.1065e-05 - 3.57882e-05 - 2.05505e-05 - 1.15623e-05 - 6.37384e-06 - 3.44267e-06 - 1.82192e-06 - 9.44718e-07 - 4.79996e-07 - 2.39019e-07 - 1.16785e-07 - 5.63189e-08 - 2.75876e-08 - 1.55066e-08 - 1.35065e-08 - 2.06346e-08 - 4.12348e-08 - 8.71701e-08 - 1.83491e-07 - 3.79656e-07 - 7.70155e-07 - 1.53093e-06 - 2.98182e-06 - 5.69044e-06 - 1.06401e-05 - 1.94933e-05 - 3.49914e-05 - 6.15423e-05 - 0.000106053 - 0.000179064 - 0.000296231 - 0.000480165 - 0.000762583 - 0.00118664 - 0.00180922 - 0.0027027 - 0.00395586 - 0.00567312 - 0.00797149 - 0.0109747 - 0.0148042 - 0.0195665 - 0.0253384 - 0.03215 - 0.0399688 - 0.0486854 - 0.0581051 - 0.0679468 - 0.0778511 - 0.0873986 - 0.0961378 - 0.10362 - 0.109438 - 0.113265 - 0.11489 - 0.114237 - 0.111386 - 0.106565 - 0.100144 - 0.092608 - 0.0845337 - 0.0765571 - 0.069347 - 0.0635806 - 0.0599254 - 0.0590243 - 0.0614823 - 0.0678499 - 0.0786025 - 0.0941094 - 0.114595 - 0.140095 - 0.170403 - 0.205033 - 0.243187 - 0.28375 - 0.325307 - 0.366205 - 0.404639 - 0.438768 - 0.466855 - 0.487399 - 0.499265 - 0.501787 - 0.494819 - 0.478757 - 0.454493 - 0.423343 - 0.386921 - 0.347013 - 0.305437 - 0.263913 - 0.223964 - 0.186845 - 0.15351 - 0.124609 - 0.100516 - 0.0813752 - 0.0671528 - 0.0576947 - 0.0527714 - 0.0521107 - 0.0554152 - 0.0623635 - 0.0726003 - 0.0857196 - 0.101248 - 0.118632 - 0.137242 - 0.156382 - 0.175324 - 0.193347 - 0.209787 - 0.224084 - 0.235824 - 0.244759 - 0.250819 - 0.25409 - 0.254781 - 0.253179 - 0.249598 - 0.24433 - 0.237611 - 0.229608 - 0.220414 - 0.210075 - 0.198611 - 0.186058 - 0.172493 - 0.15806 - 0.142974 - 0.127518 - 0.112024 - 0.0968447 - 0.0823232 - 0.0687639 - 0.056409 - 0.0454241 - 0.0358933 - 0.0278226 - 0.0211511 - 0.0157665 - 0.0115222 - 0.00825423 - 0.0057959 - 0.0039887 - 0.00269018 - 0.00177808 - 0.00115167 - 0.000731004 - 0.000454751 - 0.000277398 - 0.000166222 - 9.846e-05 - 5.88987e-05 - 3.79727e-05 - 3.04441e-05 - 3.46788e-05 - 5.25582e-05 - 9.00977e-05 - 0.000158898 - 0.000278618 - 0.000480699 - 0.000813605 - 0.00134977 - 0.00219438 - 0.00349573 - 0.00545668 - 0.00834609 - 0.0125084 - 0.0183688 - 0.0264315 - 0.0372673 - 0.0514867 - 0.0696987 - 0.0924523 - 0.120163 - 0.153035 - 0.190972 - 0.233514 - 0.27978 - 0.328461 - 0.377845 - 0.425897 - 0.470391 - 0.509067 - 0.539825 - 0.56091 - 0.571077 - 0.569717 - 0.55691 - 0.533424 - 0.500637 - 0.460399 - 0.414867 - 0.366308 - 0.31692 - 0.268674 - 0.223194 - 0.181698 - 0.144974 - 0.113407 - 0.0870373 - 0.0656418 - 0.048819 - 0.0360773 - 0.0269108 - 0.0208626 - 0.0175701 - 0.0167947 - 0.0184339 - 0.0225191 - 0.0291988 - 0.0387082 - 0.0513264 - 0.067323 - 0.0868961 - 0.110107 - 0.136818 - 0.166639 - 0.198893 - 0.232612 - 0.266561 - 0.299299 - 0.329271 - 0.354927 - 0.374855 - 0.387903 - 0.393297 - 0.390711 - 0.3803 - 0.36269 - 0.338907 - 0.310288 - 0.278347 - 0.244652 - 0.210696 - 0.177796 - 0.147015 - 0.119132 - 0.0946281 - 0.0737165 - 0.0563825 - 0.0424423 - 0.0316028 - 0.0235195 - 0.0178455 - 0.0142692 - 0.0125399 - 0.0124803 - 0.0139879 - 0.0170253 - 0.0216027 - 0.0277537 - 0.0355053 - 0.0448476 - 0.0557031 - 0.0679011 - 0.0811603 - 0.0950824 - 0.10916 - 0.1228 - 0.135357 - 0.146187 - 0.154696 - 0.160393 - 0.162941 - 0.162186 - 0.158174 - 0.151147 - 0.141516 - 0.129827 - 0.116706 - 0.102807 - 0.0887582 - 0.0751238 - 0.062369 - 0.0508463 - 0.0407927 - 0.0323401 - 0.0255341 - 0.0203573 - 0.0167544 - 0.0146531 - 0.0139808 - 0.0146745 - 0.0166817 - 0.019956 - 0.0244455 - 0.0300777 - 0.0367437 - 0.0442835 - 0.0524759 - 0.0610361 - 0.0696209 - 0.0778441 - 0.0852996 - 0.0915913 - 0.096366 - 0.0993442 - 0.100347 - 0.0993131 - 0.0963045 - 0.0915008 - 0.0851805 - 0.0776949 - 0.0694355 - 0.0608006 - 0.052164 - 0.0438501 - 0.0361166 - 0.0291461 - 0.0230458 - 0.0178541 - 0.0135526 - 0.0100796 - 0.00734517 - 0.00524441 - 0.00366883 - 0.00251476 - 0.00168889 - 0.00111133 - 0.00071651 - 0.000452625 - 0.00028015 - 0.000169895 - 0.00010095 - 5.87719e-05 - 3.35251e-05 - 1.87373e-05 - 1.02608e-05 - 5.50542e-06 - 2.89426e-06 - 1.49081e-06 - 7.52391e-07 - 3.7205e-07 - 1.80259e-07 - 8.55712e-08 - 3.98012e-08 - 1.81385e-08 - 8.09923e-09 - 3.54342e-09 - 1.51893e-09 - 6.37957e-10 - 2.62531e-10 - 1.05854e-10 - 4.18187e-11 - 1.61872e-11 - 6.13914e-12 - 2.28129e-12 - 8.306e-13 - 2.96305e-13 - 1.03568e-13 - 3.54687e-14 - 1.19015e-14 - 3.91289e-15 - 1.26046e-15 - 3.9783e-16 - 1.23028e-16 - 3.72774e-17 - 1.10669e-17 - 3.21915e-18 - 9.17473e-19 - 2.56202e-19 - 7.00986e-20 - 1.8792e-20 - 4.93598e-21 - 1.27031e-21 - 3.2032e-22 - 7.91398e-23 - 1.91577e-23 - 4.54388e-24 - 1.05596e-24 - 2.4044e-25 - 5.36415e-26 - 1.17255e-26 - 2.51132e-27 - 5.26996e-28 - 1.08355e-28 - 2.18287e-29 - 4.30869e-30 - 8.33293e-31 - 1.57902e-31 - 2.93166e-32 - 5.33307e-33 - 9.50556e-34 - 1.66003e-34 - 2.84046e-35 - 4.76211e-36 - 7.82253e-37 - 1.25902e-37 - 1.98542e-38 - 3.06769e-39 - 4.64415e-40 - 6.88872e-41 - 1.00117e-41 - 1.42565e-42 - 1.98909e-43 - 2.71916e-44 - 3.64209e-45 - 4.77973e-46 - 6.14602e-47 - 7.7432e-48 - 9.55837e-49 - 1.15607e-49 - 1.37e-50 - 1.59073e-51 - 1.8097e-52 - 2.01722e-53 - 2.20312e-54 - 2.35755e-55 - 2.47183e-56 - 2.5393e-57 - 2.55591e-58 - 2.52066e-59 - 2.43568e-60 - 2.30602e-61 - 2.13916e-62 - 1.94428e-63 - 1.73146e-64 - 1.51078e-65 - 1.2916e-66 - 1.08191e-67 - 8.87958e-69 - 7.14052e-70 - 5.62606e-71 - 4.34325e-72 - 3.28521e-73 - 2.43471e-74 - 1.76795e-75 - 1.25785e-76 - 8.76845e-78 - 5.98901e-79 - 4.00797e-80 - 2.62803e-81 - 1.68839e-82 - 1.0628e-83 - 6.55493e-85 - 3.96115e-86 - 2.34537e-87 - 1.36063e-88 - 7.73397e-90 - 4.30728e-91 - 2.35039e-92 - 1.25665e-93 - 6.58303e-95 - 3.37889e-96 - 1.69925e-97 - 8.37298e-99 - 4.04239e-100 - 1.9122e-101 - 8.8627e-103 - 4.02472e-104 - 1.79078e-105 - 7.807e-107 - 3.33476e-108 - 1.39764e-109 - 1.06571e-110 - 1.20963e-109 - 2.89465e-108 - 6.79983e-107 - 1.56509e-105 - 3.52952e-104 - 7.79884e-103 - 1.68842e-101 - 3.58153e-100 - 7.44378e-99 - 1.51585e-97 - 3.0245e-96 - 5.91274e-95 - 1.13256e-93 - 2.12554e-92 - 3.90855e-91 - 7.04204e-90 - 1.24313e-88 - 2.15018e-87 - 3.64391e-86 - 6.05059e-85 - 9.84384e-84 - 1.56916e-82 - 2.45081e-81 - 3.75048e-80 - 5.62342e-79 - 8.26136e-78 - 1.18916e-76 - 1.67712e-75 - 2.31753e-74 - 3.1378e-73 - 4.16256e-72 - 5.41044e-71 - 6.89035e-70 - 8.5978e-69 - 1.05116e-67 - 1.25919e-66 - 1.4779e-65 - 1.69957e-64 - 1.915e-63 - 2.11415e-62 - 2.28686e-61 - 2.42371e-60 - 2.51686e-59 - 2.56079e-58 - 2.55284e-57 - 2.49352e-56 - 2.38637e-55 - 2.23769e-54 - 2.05588e-53 - 1.85069e-52 - 1.63232e-51 - 1.41064e-50 - 1.19443e-49 - 9.90932e-49 - 8.05496e-48 - 6.41535e-47 - 5.00626e-46 - 3.82775e-45 - 2.86755e-44 - 2.10482e-43 - 1.51376e-42 - 1.06668e-41 - 7.36458e-41 - 4.98195e-40 - 3.30208e-39 - 2.14443e-38 - 1.3645e-37 - 8.50695e-37 - 5.19648e-36 - 3.11015e-35 - 1.82386e-34 - 1.04794e-33 - 5.89958e-33 - 3.25418e-32 - 1.75873e-31 - 9.31305e-31 - 4.83195e-30 - 2.45635e-29 - 1.22347e-28 - 5.97085e-28 - 2.85506e-27 - 1.33761e-26 - 6.1402e-26 - 2.76167e-25 - 1.21702e-24 - 5.25487e-24 - 2.22312e-23 - 9.21509e-23 - 3.74261e-22 - 1.48931e-21 - 5.80677e-21 - 2.2183e-20 - 8.30318e-20 - 3.04513e-19 - 1.09422e-18 - 3.85247e-18 - 1.32896e-17 - 4.49183e-17 - 1.48755e-16 - 4.82679e-16 - 1.53456e-15 - 4.7802e-15 - 1.45897e-14 - 4.36302e-14 - 1.2784e-13 - 3.67016e-13 - 1.03239e-12 - 2.84538e-12 - 7.68384e-12 - 2.03309e-11 - 5.27081e-11 - 1.33887e-10 - 3.33231e-10 - 8.12632e-10 - 1.94172e-09 - 4.54597e-09 - 1.04283e-08 - 2.34394e-08 - 5.16215e-08 - 1.11395e-07 - 2.35536e-07 - 4.87981e-07 - 9.90622e-07 - 1.97049e-06 - 3.84067e-06 - 7.33515e-06 - 1.37273e-05 - 2.51733e-05 - 4.52355e-05 - 7.96547e-05 - 0.000137449 - 0.000232424 - 0.000385157 - 0.000625493 - 0.000995518 - 0.00155285 - 0.00237402 - 0.00355735 - 0.00522491 - 0.00752254 - 0.0106172 - 0.0146907 - 0.0199294 - 0.0265096 - 0.0345787 - 0.0442341 - 0.0555004 - 0.0683094 - 0.0824842 - 0.0977307 - 0.11364 - 0.129703 - 0.145333 - 0.159905 - 0.172795 - 0.183427 - 0.191318 - 0.19611 - 0.197601 - 0.195756 - 0.190705 - 0.18273 - 0.172236 - 0.15972 - 0.145736 - 0.130848 - 0.115605 - 0.100507 - 0.0859807 - 0.0723703 - 0.0599273 - 0.0488129 - 0.0391035 - - - - - 1.7108e-30 - 2.12904e-30 - 2.59602e-30 - 3.10148e-30 - 3.63055e-30 - 4.1643e-30 - 4.68152e-30 - 5.16493e-30 - 5.62759e-30 - 6.24166e-30 - 7.97863e-30 - 1.57598e-29 - 5.36674e-29 - 2.36881e-28 - 1.10479e-27 - 5.13129e-27 - 2.34285e-26 - 1.04879e-25 - 4.60074e-25 - 1.97749e-24 - 8.32801e-24 - 3.43641e-23 - 1.38933e-22 - 5.50355e-22 - 2.13608e-21 - 8.12322e-21 - 3.02675e-20 - 1.105e-19 - 3.9526e-19 - 1.38529e-18 - 4.75704e-18 - 1.60055e-17 - 5.27641e-17 - 1.70429e-16 - 5.3937e-16 - 1.6725e-15 - 5.0814e-15 - 1.51264e-14 - 4.41191e-14 - 1.26082e-13 - 3.53035e-13 - 9.68542e-13 - 2.60349e-12 - 6.85694e-12 - 1.76946e-11 - 4.47394e-11 - 1.10834e-10 - 2.69027e-10 - 6.39816e-10 - 1.49091e-09 - 3.40395e-09 - 7.61469e-09 - 1.66901e-08 - 3.58427e-08 - 7.5419e-08 - 1.55488e-07 - 3.14088e-07 - 6.21643e-07 - 1.2055e-06 - 2.29051e-06 - 4.26416e-06 - 7.77805e-06 - 1.3901e-05 - 2.4342e-05 - 4.17643e-05 - 7.02086e-05 - 0.000115641 - 0.000186625 - 0.000295098 - 0.000457192 - 0.000694014 - 0.00103222 - 0.00150424 - 0.00214782 - 0.00300479 - 0.00411877 - 0.00553169 - 0.00727922 - 0.00938532 - 0.0118563 - 0.0146753 - 0.0177977 - 0.0211483 - 0.0246221 - 0.0280875 - 0.0313934 - 0.0343799 - 0.0368904 - 0.0387854 - 0.0399557 - 0.0403326 - 0.0398957 - 0.0386751 - 0.036749 - 0.0342377 - 0.0312926 - 0.0280847 - 0.0247919 - 0.0215871 - 0.0186297 - 0.0160581 - 0.013987 - 0.0125051 - 0.0116762 - 0.0115397 - 0.0121108 - 0.0133811 - 0.0153167 - 0.0178562 - 0.0209094 - 0.0243561 - 0.0280473 - 0.0318096 - 0.0354525 - 0.0387797 - 0.041602 - 0.0437517 - 0.0450968 - 0.045552 - 0.0450867 - 0.043727 - 0.041553 - 0.03869 - 0.0352968 - 0.031551 - 0.027633 - 0.0237127 - 0.0199375 - 0.0164247 - 0.0132575 - 0.0104848 - 0.00812456 - 0.00616843 - 0.00458866 - 0.00334453 - 0.00238847 - 0.00167125 - 0.00114578 - 0.000769656 - 0.000506558 - 0.000326662 - 0.000206398 - 0.000127776 - 7.75048e-05 - 4.60623e-05 - 2.68225e-05 - 1.53035e-05 - 8.55494e-06 - 4.68577e-06 - 2.51468e-06 - 1.32227e-06 - 6.81232e-07 - 3.4388e-07 - 1.70081e-07 - 8.24215e-08 - 3.91348e-08 - 1.82063e-08 - 8.29885e-09 - 3.70639e-09 - 1.62189e-09 - 6.95389e-10 - 2.92132e-10 - 1.2026e-10 - 4.85509e-11 - 1.93322e-11 - 7.90272e-12 - 4.15729e-12 - 4.69603e-12 - 1.00995e-11 - 2.54898e-11 - 6.48543e-11 - 1.62322e-10 - 3.9828e-10 - 9.57567e-10 - 2.25575e-09 - 5.20657e-09 - 1.17747e-08 - 2.60905e-08 - 5.66438e-08 - 1.20492e-07 - 2.51133e-07 - 5.12843e-07 - 1.02613e-06 - 2.01167e-06 - 3.8641e-06 - 7.27238e-06 - 1.34104e-05 - 2.42295e-05 - 4.28927e-05 - 7.43975e-05 - 0.000126436 - 0.000210533 - 0.000343484 - 0.000549073 - 0.000859983 - 0.00131974 - 0.00198436 - 0.00292342 - 0.00421986 - 0.00596819 - 0.00827035 - 0.011229 - 0.0149381 - 0.019471 - 0.0248666 - 0.0311158 - 0.0381491 - 0.0458271 - 0.0539385 - 0.062203 - 0.0702847 - 0.0778121 - 0.0844055 - 0.0897079 - 0.0934174 - 0.0953152 - 0.095287 - 0.0933346 - 0.0895758 - 0.0842323 - 0.0776088 - 0.0700637 - 0.0619781 - 0.0537245 - 0.0456401 - 0.0380067 - 0.0310392 - 0.0248817 - 0.0196118 - 0.0152499 - 0.0117717 - 0.00912236 - 0.00722982 - 0.00601571 - 0.00540334 - 0.0053222 - 0.00570923 - 0.00650744 - 0.00766253 - 0.00911876 - 0.0108148 - 0.0126804 - 0.0146355 - 0.0165902 - 0.0184485 - 0.020113 - 0.0214914 - 0.022504 - 0.0230902 - 0.0232141 - 0.022868 - 0.0220728 - 0.020876 - 0.0193474 - 0.0175725 - 0.0156453 - 0.0136615 - 0.0117119 - 0.00987887 - 0.00823454 - 0.00684165 - 0.00575724 - 0.00503837 - 0.00474935 - 0.00496969 - 0.00580156 - 0.00737575 - 0.00985451 - 0.0134302 - 0.0183181 - 0.0247429 - 0.0329181 - 0.0430191 - 0.0551515 - 0.0693182 - 0.0853881 - 0.103073 - 0.121914 - 0.141292 - 0.160443 - 0.178512 - 0.194604 - 0.207861 - 0.217536 - 0.223062 - 0.224109 - 0.220611 - 0.212781 - 0.201084 - 0.18619 - 0.168917 - 0.15015 - 0.130773 - 0.111595 - 0.0933057 - 0.0764379 - 0.0613545 - 0.0482527 - 0.037182 - 0.0280725 - 0.0207666 - 0.0150518 - 0.0106892 - 0.00743777 - 0.00507078 - 0.00338723 - 0.00221692 - 0.00142165 - 0.000893251 - 0.000549908 - 0.000331699 - 0.000196035 - 0.000113517 - 6.44059e-05 - 3.58036e-05 - 1.95013e-05 - 1.04073e-05 - 5.4419e-06 - 2.78804e-06 - 1.39954e-06 - 6.88345e-07 - 3.31715e-07 - 1.56625e-07 - 7.24592e-08 - 3.28445e-08 - 1.45871e-08 - 6.34764e-09 - 2.7064e-09 - 1.1306e-09 - 4.62768e-10 - 1.8559e-10 - 7.29258e-11 - 2.80766e-11 - 1.05912e-11 - 3.91457e-12 - 1.41762e-12 - 5.03003e-13 - 1.74872e-13 - 5.95668e-14 - 1.98805e-14 - 6.50108e-15 - 2.08296e-15 - 6.53904e-16 - 2.01133e-16 - 6.06164e-17 - 1.78992e-17 - 5.17861e-18 - 1.46801e-18 - 4.07741e-19 - 1.10962e-19 - 2.95871e-20 - 7.72981e-21 - 1.97881e-21 - 4.96994e-22 - 1.2513e-22 - 4.27973e-23 - 6.25412e-23 - 2.27169e-22 - 8.96772e-22 - 3.49045e-21 - 1.3316e-20 - 4.97752e-20 - 1.82301e-19 - 6.54186e-19 - 2.30012e-18 - 7.92386e-18 - 2.67461e-17 - 8.84545e-17 - 2.86627e-16 - 9.1002e-16 - 2.83088e-15 - 8.62837e-15 - 2.57675e-14 - 7.5397e-14 - 2.16158e-13 - 6.07193e-13 - 1.67116e-12 - 4.50658e-12 - 1.19073e-11 - 3.08258e-11 - 7.81904e-11 - 1.94325e-10 - 4.73197e-10 - 1.12899e-09 - 2.63923e-09 - 6.04507e-09 - 1.35663e-08 - 2.98304e-08 - 6.42676e-08 - 1.35663e-07 - 2.80589e-07 - 5.6861e-07 - 1.12901e-06 - 2.19641e-06 - 4.18667e-06 - 7.81917e-06 - 1.43083e-05 - 2.5654e-05 - 4.50668e-05 - 7.75704e-05 - 0.000130819 - 0.000216165 - 0.000349973 - 0.000555164 - 0.000862869 - 0.00131403 - 0.00196066 - 0.00286641 - 0.00410592 - 0.00576262 - 0.00792443 - 0.0106771 - 0.0140955 - 0.0182326 - 0.0231079 - 0.0286958 - 0.0349164 - 0.0416297 - 0.0486353 - 0.0556796 - 0.0624697 - 0.0686941 - 0.0740496 - 0.07827 - 0.0811551 - 0.0825952 - 0.0825891 - 0.0812525 - 0.0788147 - 0.0756052 - 0.0720296 - 0.068538 - 0.0655882 - 0.0636076 - 0.0629552 - 0.0638897 - 0.0665429 - 0.0709037 - 0.0768119 - 0.0839642 - 0.0919334 - 0.100197 - 0.108179 - 0.115294 - 0.120995 - 0.124823 - 0.126445 - 0.125678 - 0.122506 - 0.117071 - 0.10966 - 0.100669 - 0.0905633 - 0.0798348 - 0.0689607 - 0.0583676 - 0.0484063 - 0.0393372 - 0.0313261 - 0.0244505 - 0.0187122 - 0.0140553 - 0.0103846 - 0.00758494 - 0.00553752 - 0.00413347 - 0.00328399 - 0.00292721 - 0.00303184 - 0.00359822 - 0.00465668 - 0.00626355 - 0.00849453 - 0.0114355 - 0.0151707 - 0.019769 - 0.0252679 - 0.0316588 - 0.0388727 - 0.0467703 - 0.0551377 - 0.0636899 - 0.0720829 - 0.0799342 - 0.08685 - 0.092458 - 0.0964398 - 0.0985609 - 0.0986938 - 0.0968305 - 0.0930832 - 0.0876733 - 0.0809096 - 0.0731593 - 0.0648151 - 0.0562625 - 0.0478519 - 0.0398764 - 0.0325589 - 0.0260472 - 0.0204168 - 0.0156802 - 0.0117992 - 0.00869946 - 0.00628445 - 0.00444814 - 0.0030848 - 0.0020961 - 0.00139552 - 0.00091032 - 0.000581822 - 0.000364354 - 0.000223559 - 0.0001344 - 7.91667e-05 - 4.56902e-05 - 2.5837e-05 - 1.43154e-05 - 7.77184e-06 - 4.13516e-06 - 2.15827e-06 - 1.10966e-06 - 5.72705e-07 - 3.20383e-07 - 2.41707e-07 - 3.04402e-07 - 5.4587e-07 - 1.09169e-06 - 2.20877e-06 - 4.41012e-06 - 8.641e-06 - 1.65944e-05 - 3.12269e-05 - 5.75758e-05 - 0.000104013 - 0.000184109 - 0.0003193 - 0.000542574 - 0.00090335 - 0.00147364 - 0.00235538 - 0.00368866 - 0.00565995 - 0.0085093 - 0.0125346 - 0.0180912 - 0.0255834 - 0.0354476 - 0.048123 - 0.0640111 - 0.0834246 - 0.10653 - 0.133286 - 0.163393 - 0.196254 - 0.230963 - 0.266319 - 0.300885 - 0.333069 - 0.361247 - 0.383895 - 0.399721 - 0.407791 - 0.407621 - 0.399219 - 0.383092 - 0.36019 - 0.331816 - 0.299502 - 0.264874 - 0.229517 - 0.194862 - 0.162098 - 0.132119 - 0.105509 - 0.0825559 - 0.0632915 - 0.0475422 - 0.0349905 - 0.0252324 - 0.017828 - 0.012342 - 0.00837148 - 0.00556362 - 0.00362284 - 0.00231142 - 0.00144492 - 0.000885006 - 0.000531112 - 0.000312293 - 0.000179918 - 0.000101561 - 5.61709e-05 - 3.04393e-05 - 1.6162e-05 - 8.408e-06 - 4.28575e-06 - 2.14042e-06 - 1.04738e-06 - 5.0217e-07 - 2.35902e-07 - 1.0858e-07 - 4.89673e-08 - 2.16371e-08 - 9.36772e-09 - 3.97414e-09 - 1.65288e-09 - 6.76192e-10 - 2.78154e-10 - 1.30846e-10 - 1.07446e-10 - 1.85993e-10 - 4.31066e-10 - 1.04002e-09 - 2.48173e-09 - 5.81045e-09 - 1.33319e-08 - 2.99725e-08 - 6.60228e-08 - 1.42496e-07 - 3.01333e-07 - 6.24351e-07 - 1.2675e-06 - 2.52117e-06 - 4.91355e-06 - 9.38262e-06 - 1.75546e-05 - 3.21805e-05 - 5.78006e-05 - 0.000101721 - 0.000175397 - 0.000296328 - 0.000490523 - 0.00079558 - 0.00126428 - 0.00196853 - 0.00300315 - 0.00448898 - 0.00657439 - 0.00943411 - 0.0132642 - 0.0182726 - 0.0246636 - 0.0326174 - 0.0422648 - 0.0536592 - 0.0667494 - 0.0813555 - 0.0971547 - 0.113678 - 0.130325 - 0.146391 - 0.161116 - 0.17374 - 0.183568 - 0.190034 - 0.192754 - 0.191563 - 0.186533 - 0.177966 - 0.166363 - 0.152375 - 0.136743 - 0.120236 - 0.103586 - 0.0874387 - 0.0723175 - 0.058603 - 0.0465301 - 0.036198 - 0.0275912 - 0.0206061 - 0.0150784 - 0.0108107 - 0.00759432 - 0.00522709 - 0.00352508 - 0.00232924 - 0.00150798 - 0.000956565 - 0.000594525 - 0.000362045 - 0.000216019 - 0.000126287 - 7.23372e-05 - 4.05977e-05 - 2.23243e-05 - 1.2028e-05 - 6.34954e-06 - 3.2842e-06 - 1.66439e-06 - 8.26447e-07 - 4.0208e-07 - 1.91667e-07 - 8.95198e-08 - 4.09664e-08 - 1.83685e-08 - 8.06966e-09 - 3.47355e-09 - 1.46497e-09 - 6.05372e-10 - 2.45105e-10 - 9.7234e-11 - 3.77939e-11 - 1.43933e-11 - 5.37079e-12 - 1.9636e-12 - 7.03401e-13 - 2.46882e-13 - 8.49013e-14 - 2.86072e-14 - 9.44439e-15 - 3.05498e-15 - 9.68234e-16 - 3.00669e-16 - 9.14817e-17 - 2.7272e-17 - 7.96591e-18 - 2.27978e-18 - 6.39307e-19 - 1.7579e-19 - 4.79091e-20 - 1.49851e-20 - 1.3075e-20 - 3.86379e-20 - 1.42464e-19 - 5.24334e-19 - 1.89316e-18 - 6.6979e-18 - 2.32182e-17 - 7.886e-17 - 2.62435e-16 - 8.55703e-16 - 2.73377e-15 - 8.55729e-15 - 2.62451e-14 - 7.88674e-14 - 2.32211e-13 - 6.69893e-13 - 1.8935e-12 - 5.24398e-12 - 1.42297e-11 - 3.78324e-11 - 9.85533e-11 - 2.51544e-10 - 6.29065e-10 - 1.54139e-09 - 3.70056e-09 - 8.70479e-09 - 2.00626e-08 - 4.53056e-08 - 1.00243e-07 - 2.17317e-07 - 4.61603e-07 - 9.60688e-07 - 1.95899e-06 - 3.914e-06 - 7.66208e-06 - 1.46964e-05 - 2.76192e-05 - 5.08571e-05 - 9.17551e-05 - 0.000162199 - 0.000280936 - 0.00047677 - 0.000792781 - 0.00129164 - 0.00206196 - 0.0032253 - 0.00494329 - 0.00742376 - 0.0109245 - 0.015753 - 0.0222597 - 0.0308241 - 0.0418311 - 0.0556391 - 0.0725392 - 0.0927117 - 0.116181 - 0.142782 - 0.172134 - 0.203646 - 0.23654 - 0.269907 - 0.302784 - 0.334251 - 0.36353 - 0.390069 - 0.413606 - 0.434188 - 0.452139 - 0.467992 - 0.482371 - 0.495855 - 0.508845 - 0.521448 - 0.533411 - 0.54411 - 0.552604 - 0.557745 - 0.558334 - 0.553292 - 0.541826 - 0.523566 - 0.49865 - 0.46775 - 0.432034 - 0.393086 - 0.35278 - 0.313139 - 0.2762 - 0.243885 - 0.2179 - 0.199659 - 0.19023 - 0.190292 - 0.200109 - 0.219505 - 0.247844 - 0.284017 - 0.326452 - 0.373144 - 0.421726 - 0.469578 - 0.513978 - 0.552279 - 0.582113 - 0.601568 - 0.60936 - 0.604932 - 0.588498 - 0.561014 - 0.524074 - 0.479755 - 0.430422 - 0.378526 - 0.32641 - 0.276152 - 0.229449 - 0.187561 - 0.151296 - 0.121041 - 0.0968234 - 0.0783825 - 0.0652508 - 0.0568241 - 0.052421 - 0.0513267 - 0.0528233 - 0.0562096 - 0.0608144 - 0.0660069 - 0.0712085 - 0.0759058 - 0.0796663 - 0.0821531 - 0.083138 - 0.0825095 - 0.0802724 - 0.0765404 - 0.0715192 - 0.0654835 - 0.0587491 - 0.0516444 - 0.0444831 - 0.0375424 - 0.031047 - 0.0251611 - 0.0199871 - 0.0155706 - 0.0119103 - 0.00897034 - 0.00669384 - 0.00501675 - 0.00387995 - 0.00323985 - 0.00307697 - 0.00340228 - 0.00426134 - 0.00573584 - 0.00794229 - 0.011027 - 0.015157 - 0.0205062 - 0.0272364 - 0.0354751 - 0.0452892 - 0.0566592 - 0.0694565 - 0.0834268 - 0.0981838 - 0.113218 - 0.127916 - 0.141604 - 0.153589 - 0.163224 - 0.169959 - 0.173396 - 0.17333 - 0.169763 - 0.162911 - 0.153177 - 0.141116 - 0.127378 - 0.112654 - 0.0976202 - 0.0828835 - 0.0689498 - 0.0561998 - 0.0448822 - 0.0351196 - 0.0269254 - 0.0202261 - 0.0148867 - 0.0107355 - 0.00758544 - 0.00525143 - 0.00356214 - 0.00236745 - 0.00154166 - 0.000983633 - 0.000614914 - 0.000376646 - 0.000226043 - 0.000132922 - 7.65918e-05 - 4.32595e-05 - 2.39797e-05 - 1.31129e-05 - 7.21922e-06 - 4.30366e-06 - 3.33996e-06 - 4.01773e-06 - 6.68526e-06 - 1.24923e-05 - 2.37754e-05 - 4.47731e-05 - 8.28123e-05 - 0.000150164 - 0.000266831 - 0.000464579 - 0.000792543 - 0.00132472 - 0.00216951 - 0.00348126 - 0.0054733 - 0.00843136 - 0.0127257 - 0.0188194 - 0.0272687 - 0.0387134 - 0.053851 - 0.0733945 - 0.09801 - 0.128237 - 0.164397 - 0.206496 - 0.254135 - 0.306448 - 0.362063 - 0.41913 - 0.47539 - 0.528309 - 0.575259 - 0.613727 - 0.641541 - 0.657068 - 0.659377 - 0.648327 - 0.624588 - 0.589567 - 0.545274 - 0.494136 - 0.438774 - 0.381789 - 0.325571 - 0.272154 - 0.223122 - 0.179578 - 0.142167 - 0.111134 - 0.086417 - 0.0677452 - 0.0547365 - 0.04698 - 0.0440953 - 0.0457651 - 0.0517401 - 0.0618192 - 0.075809 - 0.0934719 - 0.114469 - 0.138309 - 0.164311 - 0.191592 - 0.21908 - 0.245557 - 0.269731 - 0.29033 - 0.306206 - 0.316433 - 0.320401 - 0.317866 - 0.308982 - 0.29428 - 0.274615 - 0.251088 - 0.224938 - 0.197441 - 0.169805 - 0.143086 - 0.118136 - 0.0955664 - 0.0757468 - 0.0588247 - 0.0447603 - 0.0333705 - 0.0243763 - 0.0174466 - 0.0122347 - 0.00840639 - 0.00565932 - 0.00373302 - 0.0024127 - 0.001528 - 0.000948448 - 0.000577417 - 0.000345638 - 0.0002051 - 0.000123843 - 8.18518e-05 - 6.81308e-05 - 7.89389e-05 - 0.00011715 - 0.000192719 - 0.000324255 - 0.000541737 - 0.000890371 - 0.00143553 - 0.00226859 - 0.00351311 - 0.00533082 - 0.00792609 - 0.0115475 - 0.0164852 - 0.0230612 - 0.0316131 - 0.0424685 - 0.0559121 - 0.0721461 - 0.0912487 - 0.113135 - 0.137526 - 0.163937 - 0.191679 - 0.219893 - 0.247604 - 0.273794 - 0.297493 - 0.317866 - 0.334291 - 0.346416 - 0.354179 - 0.357793 - 0.357701 - 0.354498 - 0.348835 - 0.34133 - 0.332489 - 0.322651 - 0.311975 - 0.30045 - 0.287942 - 0.274258 - 0.25922 - 0.242722 - 0.224784 - 0.205571 - 0.185391 - 0.164673 - 0.143916 - 0.123646 - 0.104359 - 0.0864807 - 0.0703312 - 0.0561129 - 0.0439079 - 0.0336895 - 0.0253425 - 0.0186881 - 0.0135096 - 0.00957556 - 0.00665938 - 0.00455333 - 0.00307793 - 0.00208716 - 0.00147068 - 0.00115451 - 0.00110122 - 0.00131073 - 0.0018224 - 0.00271851 - 0.00412898 - 0.00623645 - 0.00928047 - 0.0135588 - 0.0194236 - 0.0272705 - 0.0375177 - 0.0505746 - 0.0667991 - 0.0864465 - 0.109613 - 0.13618 - 0.165769 - 0.19771 - 0.231042 - 0.26454 - 0.296775 - 0.326212 - 0.351326 - 0.370729 - 0.383301 - 0.388294 - 0.385406 - 0.374811 - 0.357144 - 0.333435 - 0.305011 - 0.273374 - 0.240069 - 0.206562 - 0.174142 - 0.143844 - 0.116417 - 0.0923168 - 0.0717266 - 0.054603 - 0.0407277 - 0.0297646 - 0.0213131 - 0.0149531 - 0.010279 - 0.00692323 - 0.00456881 - 0.00295416 - 0.00187155 - 0.00116173 - 0.000706559 - 0.000421043 - 0.000245834 - 0.000140635 - 7.88285e-05 - 4.32922e-05 - 2.32955e-05 - 1.22821e-05 - 6.34466e-06 - 3.2113e-06 - 1.59255e-06 - 7.73819e-07 - 3.68407e-07 - 1.71859e-07 - 7.85727e-08 - 3.525e-08 - 1.56255e-08 - 7.1029e-09 - 3.90724e-09 - 3.74228e-09 - 6.36823e-09 - 1.3367e-08 - 2.89202e-08 - 6.18851e-08 - 1.29968e-07 - 2.67519e-07 - 5.39551e-07 - 1.06623e-06 - 2.06446e-06 - 3.91651e-06 - 7.27996e-06 - 1.32585e-05 - 2.36592e-05 - 4.13656e-05 - 7.08625e-05 - 0.000118941 - 0.000195606 - 0.000315187 - 0.000497615 - 0.000769759 - 0.00116668 - 0.00173256 - 0.00252093 - 0.00359392 - 0.00502012 - 0.00687063 - 0.00921332 - 0.0121052 - 0.0155835 - 0.019656 - 0.0242919 - 0.0294147 - 0.0348984 - 0.0405679 - 0.0462058 - 0.0515641 - 0.0563813 - 0.0604032 - 0.0634047 - 0.0652108 - 0.0657135 - 0.0648824 - 0.0627676 - 0.0594951 - 0.055254 - 0.0502786 - 0.044827 - 0.0391591 - 0.0335168 - 0.028108 - 0.0230958 - 0.018594 - 0.0146673 - 0.0113361 - 0.00858452 - 0.00636947 - 0.0046305 - 0.0032983 - 0.00230191 - 0.00157407 - 0.00105462 - 0.000692316 - 0.000445298 - 0.000280629 - 0.000173282 - 0.000104836 - 6.21444e-05 - 3.60937e-05 - 2.05399e-05 - 1.14526e-05 - 6.25667e-06 - 3.34904e-06 - 1.75645e-06 - 9.02581e-07 - 4.5444e-07 - 2.24189e-07 - 1.08378e-07 - 5.13693e-08 - 2.39435e-08 - 1.11485e-08 - 5.59702e-09 - 3.92679e-09 - 5.09955e-09 - 9.86095e-09 - 2.114e-08 - 4.55587e-08 - 9.66698e-08 - 2.0116e-07 - 4.10205e-07 - 8.19618e-07 - 1.60458e-06 - 3.07786e-06 - 5.78459e-06 - 1.06521e-05 - 1.9219e-05 - 3.39755e-05 - 5.88488e-05 - 9.98725e-05 - 0.00016607 - 0.000270566 - 0.000431909 - 0.000675535 - 0.00103524 - 0.00155443 - 0.00228684 - 0.0032964 - 0.00465564 - 0.00644253 - 0.00873514 - 0.0116043 - 0.0151046 - 0.0192634 - 0.024071 - 0.0294708 - 0.035353 - 0.0415526 - 0.0478528 - 0.053995 - 0.059695 - 0.0646638 - 0.0686316 - 0.0713724 - 0.0727252 - 0.0726101 - 0.0710369 - 0.068105 - 0.0639941 - 0.058949 - 0.0532582 - 0.0472318 - 0.0411794 - 0.035392 - 0.0301294 - 0.0256128 - 0.0220232 - 0.0195037 - 0.0181646 - 0.0180873 - 0.0193273 - 0.0219132 - 0.025842 - 0.0310708 - 0.0375062 - 0.0449944 - 0.0533132 - 0.0621703 - 0.0712086 - 0.0800204 - 0.0881704 - 0.0952259 - 0.10079 - 0.104537 - 0.106239 - 0.105792 - 0.10322 - 0.0986778 - 0.0924299 - 0.0848289 - 0.0762804 - 0.0672077 - 0.058018 - 0.049073 - 0.0406687 - 0.0330228 - 0.0262727 - 0.0204801 - 0.0156422 - 0.0117058 - 0.00858298 - 0.00616616 - 0.00434041 - 0.00299359 - 0.00202311 - 0.00133989 - 0.000870016 - 0.000554601 - 0.00034854 - 0.000218741 - 0.000142232 - 0.000104541 - 9.8566e-05 - 0.000124155 - 0.000188521 - 0.000307637 - 0.000508725 - 0.000833915 - 0.00134509 - 0.00212979 - 0.00330782 - 0.00503791 - 0.00752352 - 0.0110162 - 0.0158148 - 0.0222592 - 0.0307153 - 0.0415515 - 0.0551058 - 0.071643 - 0.0913074 - 0.114073 - 0.1397 - 0.167701 - 0.197328 - 0.227588 - 0.257282 - 0.285075 - 0.309593 - 0.329531 - 0.34377 - 0.35148 - 0.352196 - 0.345872 - 0.332879 - 0.313973 - 0.29022 - 0.262897 - 0.233379 - 0.203026 - 0.173082 - 0.144596 - 0.118376 - 0.0949657 - 0.0746562 - 0.057512 - 0.0434158 - 0.0321176 - 0.0232853 - 0.0165482 - 0.0115343 - 0.00789618 - 0.00532826 - 0.00357577 - 0.00243735 - 0.00176332 - 0.00145159 - 0.00144239 - 0.00171284 - 0.00227175 - 0.00315462 - 0.00441846 - 0.0061359 - 0.00838817 - 0.0112563 - 0.0148106 - 0.019099 - 0.0241341 - 0.0298819 - 0.0362517 - 0.0430913 - 0.0501867 - 0.0572696 - 0.064032 - 0.0701466 - 0.0752928 - 0.0791839 - 0.0815939 - 0.0823787 - 0.0814908 - 0.0789841 - 0.075008 - 0.069793 - 0.0636288 - 0.0568371 - 0.0497447 - 0.0426579 - 0.0358416 - 0.0295062 - 0.0238 - 0.0188094 - 0.0145651 - 0.0110507 - 0.00821493 - 0.00598368 - 0.00427072 - 0.00298721 - 0.00204851 - 0.00137888 - 0.000914107 - 0.0006025 - 0.000404956 - 0.000294455 - 0.000255581 - 0.000284413 - 0.000389119 - 0.00059143 - 0.000929092 - 0.0014592 - 0.00226219 - 0.00344601 - 0.00514967 - 0.00754534 - 0.0108375 - 0.0152582 - 0.0210566 - 0.0284828 - 0.0377642 - 0.0490773 - 0.0625148 - 0.0780522 - 0.0955182 - 0.114573 - 0.134702 - 0.155224 - 0.175321 - 0.194088 - 0.210595 - 0.223966 - 0.233452 - 0.238502 - 0.238817 - 0.234375 - 0.225439 - 0.212528 - 0.196368 - 0.177823 - 0.157822 - 0.13728 - 0.117031 - 0.0977797 - 0.0800658 - 0.0642531 - 0.0505343 - 0.0389512 - 0.0294236 - 0.0217826 - 0.0158036 - 0.0112366 - 0.00782971 - 0.00534666 - 0.00357803 - 0.00234654 - 0.00150811 - 0.000949846 - 0.000586259 - 0.0003546 - 0.000210184 - 0.000122087 - 6.94932e-05 - 3.87633e-05 - 2.11886e-05 - 1.13497e-05 - 5.95747e-06 - 3.06434e-06 - 1.54456e-06 - 7.62898e-07 - 3.69248e-07 - 1.75129e-07 - 8.13929e-08 - 3.7068e-08 - 1.65423e-08 - 7.23393e-09 - 3.0998e-09 - 1.30158e-09 - 5.35535e-10 - 2.15914e-10 - 8.52996e-11 - 3.30208e-11 - 1.25257e-11 - 4.65589e-12 - 1.69628e-12 - 6.07133e-13 - 2.17868e-13 - 9.17968e-14 - 8.26531e-14 - 1.80758e-13 - 5.04118e-13 - 1.4277e-12 - 3.9783e-12 - 1.0868e-11 - 2.90954e-11 - 7.63304e-11 - 1.96231e-10 - 4.94351e-10 - 1.2204e-09 - 2.95232e-09 - 6.9988e-09 - 1.62586e-08 - 3.70116e-08 - 8.25642e-08 - 1.80486e-07 - 3.86628e-07 - 8.11596e-07 - 1.66949e-06 - 3.36533e-06 - 6.64764e-06 - 1.28678e-05 - 2.44085e-05 - 4.53704e-05 - 8.26422e-05 - 0.000147512 - 0.000258019 - 0.000442253 - 0.000742825 - 0.00122264 - 0.001972 - 0.00311681 - 0.00482735 - 0.00732661 - 0.0108966 - 0.0158809 - 0.0226805 - 0.0317412 - 0.04353 - 0.0584987 - 0.0770365 - 0.0994122 - 0.125711 - 0.155777 - 0.189156 - 0.225077 - 0.262443 - 0.299868 - 0.335752 - 0.368386 - 0.396083 - 0.417326 - 0.430901 - 0.436023 - 0.432407 - 0.420307 - 0.400488 - 0.374157 - 0.34285 - 0.308297 - 0.27227 - 0.236449 - 0.202305 - 0.171016 - 0.143419 - 0.12 - 0.10091 - 0.0860138 - 0.0749435 - 0.0671671 - 0.0620535 - 0.0589346 - 0.0571569 - 0.0561247 - 0.0553308 - 0.0543762 - 0.052979 - 0.0509737 - 0.0483016 - 0.0449951 - 0.0411568 - 0.0369364 - 0.0325078 - 0.0280481 - 0.0237199 - 0.0196589 - 0.0159663 - 0.0127065 - 0.00990863 - 0.00757099 - 0.00566812 - 0.00415785 - 0.00298841 - 0.00210451 - 0.00145211 - 0.000981719 - 0.000650297 - 0.00042206 - 0.000268394 - 0.000167228 - 0.00010209 - 6.1065e-05 - 3.57882e-05 - 2.05505e-05 - 1.15623e-05 - 6.37384e-06 - 3.44267e-06 - 1.82192e-06 - 9.44718e-07 - 4.79996e-07 - 2.39019e-07 - 1.16785e-07 - 5.63189e-08 - 2.75876e-08 - 1.55066e-08 - 1.35065e-08 - 2.06346e-08 - 4.12348e-08 - 8.71701e-08 - 1.83491e-07 - 3.79656e-07 - 7.70155e-07 - 1.53093e-06 - 2.98182e-06 - 5.69044e-06 - 1.06401e-05 - 1.94933e-05 - 3.49914e-05 - 6.15423e-05 - 0.000106053 - 0.000179064 - 0.000296231 - 0.000480165 - 0.000762583 - 0.00118664 - 0.00180922 - 0.0027027 - 0.00395586 - 0.00567312 - 0.00797149 - 0.0109747 - 0.0148042 - 0.0195665 - 0.0253384 - 0.03215 - 0.0399688 - 0.0486854 - 0.0581051 - 0.0679468 - 0.0778511 - 0.0873986 - 0.0961378 - 0.10362 - 0.109438 - 0.113265 - 0.11489 - 0.114237 - 0.111386 - 0.106565 - 0.100144 - 0.092608 - 0.0845337 - 0.0765571 - 0.069347 - 0.0635806 - 0.0599254 - 0.0590243 - 0.0614823 - 0.0678499 - 0.0786025 - 0.0941094 - 0.114595 - 0.140095 - 0.170403 - 0.205033 - 0.243187 - 0.28375 - 0.325307 - 0.366205 - 0.404639 - 0.438768 - 0.466855 - 0.487399 - 0.499265 - 0.501787 - 0.494819 - 0.478757 - 0.454493 - 0.423343 - 0.386921 - 0.347013 - 0.305437 - 0.263913 - 0.223964 - 0.186845 - 0.15351 - 0.124609 - 0.100516 - 0.0813752 - 0.0671528 - 0.0576947 - 0.0527714 - 0.0521107 - 0.0554152 - 0.0623635 - 0.0726003 - 0.0857196 - 0.101248 - 0.118632 - 0.137242 - 0.156382 - 0.175324 - 0.193347 - 0.209787 - 0.224084 - 0.235824 - 0.244759 - 0.250819 - 0.25409 - 0.254781 - 0.253179 - 0.249598 - 0.24433 - 0.237611 - 0.229608 - 0.220414 - 0.210075 - 0.198611 - 0.186058 - 0.172493 - 0.15806 - 0.142974 - 0.127518 - 0.112024 - 0.0968447 - 0.0823232 - 0.0687639 - 0.056409 - 0.0454241 - 0.0358933 - 0.0278226 - 0.0211511 - 0.0157665 - 0.0115222 - 0.00825423 - 0.0057959 - 0.0039887 - 0.00269018 - 0.00177808 - 0.00115167 - 0.000731004 - 0.000454751 - 0.000277398 - 0.000166222 - 9.846e-05 - 5.88987e-05 - 3.79727e-05 - 3.04441e-05 - 3.46788e-05 - 5.25582e-05 - 9.00977e-05 - 0.000158898 - 0.000278618 - 0.000480699 - 0.000813605 - 0.00134977 - 0.00219438 - 0.00349573 - 0.00545668 - 0.00834609 - 0.0125084 - 0.0183688 - 0.0264315 - 0.0372673 - 0.0514867 - 0.0696987 - 0.0924523 - 0.120163 - 0.153035 - 0.190972 - 0.233514 - 0.27978 - 0.328461 - 0.377845 - 0.425897 - 0.470391 - 0.509067 - 0.539825 - 0.56091 - 0.571077 - 0.569717 - 0.55691 - 0.533424 - 0.500637 - 0.460399 - 0.414867 - 0.366308 - 0.31692 - 0.268674 - 0.223194 - 0.181698 - 0.144974 - 0.113407 - 0.0870373 - 0.0656418 - 0.048819 - 0.0360773 - 0.0269108 - 0.0208626 - 0.0175701 - 0.0167947 - 0.0184339 - 0.0225191 - 0.0291988 - 0.0387082 - 0.0513264 - 0.067323 - 0.0868961 - 0.110107 - 0.136818 - 0.166639 - 0.198893 - 0.232612 - 0.266561 - 0.299299 - 0.329271 - 0.354927 - 0.374855 - 0.387903 - 0.393297 - 0.390711 - 0.3803 - 0.36269 - 0.338907 - 0.310288 - 0.278347 - 0.244652 - 0.210696 - 0.177796 - 0.147015 - 0.119132 - 0.0946281 - 0.0737165 - 0.0563825 - 0.0424423 - 0.0316028 - 0.0235195 - 0.0178455 - 0.0142692 - 0.0125399 - 0.0124803 - 0.0139879 - 0.0170253 - 0.0216027 - 0.0277537 - 0.0355053 - 0.0448476 - 0.0557031 - 0.0679011 - 0.0811603 - 0.0950824 - 0.10916 - 0.1228 - 0.135357 - 0.146187 - 0.154696 - 0.160393 - 0.162941 - 0.162186 - 0.158174 - 0.151147 - 0.141516 - 0.129827 - 0.116706 - 0.102807 - 0.0887582 - 0.0751238 - 0.062369 - 0.0508463 - 0.0407927 - 0.0323401 - 0.0255341 - 0.0203573 - 0.0167544 - 0.0146531 - 0.0139808 - 0.0146745 - 0.0166817 - 0.019956 - 0.0244455 - 0.0300777 - 0.0367437 - 0.0442835 - 0.0524759 - 0.0610361 - 0.0696209 - 0.0778441 - 0.0852996 - 0.0915913 - 0.096366 - 0.0993442 - 0.100347 - 0.0993131 - 0.0963045 - 0.0915008 - 0.0851805 - 0.0776949 - 0.0694355 - 0.0608006 - 0.052164 - 0.0438501 - 0.0361166 - 0.0291461 - 0.0230458 - 0.0178541 - 0.0135526 - 0.0100796 - 0.00734517 - 0.00524441 - 0.00366883 - 0.00251476 - 0.00168889 - 0.00111133 - 0.00071651 - 0.000452625 - 0.00028015 - 0.000169895 - 0.00010095 - 5.87719e-05 - 3.35251e-05 - 1.87373e-05 - 1.02608e-05 - 5.50542e-06 - 2.89426e-06 - 1.49081e-06 - 7.52391e-07 - 3.7205e-07 - 1.80259e-07 - 8.55712e-08 - 3.98012e-08 - 1.81385e-08 - 8.09923e-09 - 3.54342e-09 - 1.51893e-09 - 6.37957e-10 - 2.62531e-10 - 1.05854e-10 - 4.18187e-11 - 1.61872e-11 - 6.13914e-12 - 2.28129e-12 - 8.306e-13 - 2.96305e-13 - 1.03568e-13 - 3.54687e-14 - 1.19015e-14 - 3.91289e-15 - 1.26046e-15 - 3.9783e-16 - 1.23028e-16 - 3.72774e-17 - 1.10669e-17 - 3.21915e-18 - 9.17473e-19 - 2.56202e-19 - 7.00986e-20 - 1.8792e-20 - 4.93598e-21 - 1.27031e-21 - 3.2032e-22 - 7.91398e-23 - 1.91577e-23 - 4.54388e-24 - 1.05596e-24 - 2.4044e-25 - 5.36415e-26 - 1.17255e-26 - 2.51132e-27 - 5.26996e-28 - 1.08355e-28 - 2.18287e-29 - 4.30869e-30 - 8.33293e-31 - 1.57902e-31 - 2.93166e-32 - 5.33307e-33 - 9.50556e-34 - 1.66003e-34 - 2.84046e-35 - 4.76211e-36 - 7.82253e-37 - 1.25902e-37 - 1.98542e-38 - 3.06769e-39 - 4.64415e-40 - 6.88872e-41 - 1.00117e-41 - 1.42565e-42 - 1.98909e-43 - 2.71916e-44 - 3.64209e-45 - 4.77973e-46 - 6.14602e-47 - 7.7432e-48 - 9.55837e-49 - 1.15607e-49 - 1.37e-50 - 1.59073e-51 - 1.8097e-52 - 2.01722e-53 - 2.20312e-54 - 2.35755e-55 - 2.47183e-56 - 2.5393e-57 - 2.55591e-58 - 2.52066e-59 - 2.43568e-60 - 2.30602e-61 - 2.13916e-62 - 1.94428e-63 - 1.73146e-64 - 1.51078e-65 - 1.2916e-66 - 1.08191e-67 - 8.87958e-69 - 7.14052e-70 - 5.62606e-71 - 4.34325e-72 - 3.28521e-73 - 2.43471e-74 - 1.76795e-75 - 1.25785e-76 - 8.76845e-78 - 5.98901e-79 - 4.00797e-80 - 2.62803e-81 - 1.68839e-82 - 1.0628e-83 - 6.55493e-85 - 3.96115e-86 - 2.34537e-87 - 1.36063e-88 - 7.73397e-90 - 4.30728e-91 - 2.35039e-92 - 1.25665e-93 - 6.58303e-95 - 3.37889e-96 - 1.69925e-97 - 8.37298e-99 - 4.04239e-100 - 1.9122e-101 - 8.8627e-103 - 4.02472e-104 - 1.79078e-105 - 7.807e-107 - 3.33476e-108 - 1.39764e-109 - 1.06571e-110 - 1.20963e-109 - 2.89465e-108 - 6.79983e-107 - 1.56509e-105 - 3.52952e-104 - 7.79884e-103 - 1.68842e-101 - 3.58153e-100 - 7.44378e-99 - 1.51585e-97 - 3.0245e-96 - 5.91274e-95 - 1.13256e-93 - 2.12554e-92 - 3.90855e-91 - 7.04204e-90 - 1.24313e-88 - 2.15018e-87 - 3.64391e-86 - 6.05059e-85 - 9.84384e-84 - 1.56916e-82 - 2.45081e-81 - 3.75048e-80 - 5.62342e-79 - 8.26136e-78 - 1.18916e-76 - 1.67712e-75 - 2.31753e-74 - 3.1378e-73 - 4.16256e-72 - 5.41044e-71 - 6.89035e-70 - 8.5978e-69 - 1.05116e-67 - 1.25919e-66 - 1.4779e-65 - 1.69957e-64 - 1.915e-63 - 2.11415e-62 - 2.28686e-61 - 2.42371e-60 - 2.51686e-59 - 2.56079e-58 - 2.55284e-57 - 2.49352e-56 - 2.38637e-55 - 2.23769e-54 - 2.05588e-53 - 1.85069e-52 - 1.63232e-51 - 1.41064e-50 - 1.19443e-49 - 9.90932e-49 - 8.05496e-48 - 6.41535e-47 - 5.00626e-46 - 3.82775e-45 - 2.86755e-44 - 2.10482e-43 - 1.51376e-42 - 1.06668e-41 - 7.36458e-41 - 4.98195e-40 - 3.30208e-39 - 2.14443e-38 - 1.3645e-37 - 8.50695e-37 - 5.19648e-36 - 3.11015e-35 - 1.82386e-34 - 1.04794e-33 - 5.89958e-33 - 3.25418e-32 - 1.75873e-31 - 9.31305e-31 - 4.83195e-30 - 2.45635e-29 - 1.22347e-28 - 5.97085e-28 - 2.85506e-27 - 1.33761e-26 - 6.1402e-26 - 2.76167e-25 - 1.21702e-24 - 5.25487e-24 - 2.22312e-23 - 9.21509e-23 - 3.74261e-22 - 1.48931e-21 - 5.80677e-21 - 2.2183e-20 - 8.30318e-20 - 3.04513e-19 - 1.09422e-18 - 3.85247e-18 - 1.32896e-17 - 4.49183e-17 - 1.48755e-16 - 4.82679e-16 - 1.53456e-15 - 4.7802e-15 - 1.45897e-14 - 4.36302e-14 - 1.2784e-13 - 3.67016e-13 - 1.03239e-12 - 2.84538e-12 - 7.68384e-12 - 2.03309e-11 - 5.27081e-11 - 1.33887e-10 - 3.33231e-10 - 8.12632e-10 - 1.94172e-09 - 4.54597e-09 - 1.04283e-08 - 2.34394e-08 - 5.16215e-08 - 1.11395e-07 - 2.35536e-07 - 4.87981e-07 - 9.90622e-07 - 1.97049e-06 - 3.84067e-06 - 7.33515e-06 - 1.37273e-05 - 2.51733e-05 - 4.52355e-05 - 7.96547e-05 - 0.000137449 - 0.000232424 - 0.000385157 - 0.000625493 - 0.000995518 - 0.00155285 - 0.00237402 - 0.00355735 - 0.00522491 - 0.00752254 - 0.0106172 - 0.0146907 - 0.0199294 - 0.0265096 - 0.0345787 - 0.0442341 - 0.0555004 - 0.0683094 - 0.0824842 - 0.0977307 - 0.11364 - 0.129703 - 0.145333 - 0.159905 - 0.172795 - 0.183427 - 0.191318 - 0.19611 - 0.197601 - 0.195756 - 0.190705 - 0.18273 - 0.172236 - 0.15972 - 0.145736 - 0.130848 - 0.115605 - 0.100507 - 0.0859807 - 0.0723703 - 0.0599273 - 0.0488129 - 0.0391035 - - - - - 1.67714e-31 - 2.08717e-31 - 2.54497e-31 - 3.04059e-31 - 3.5598e-31 - 4.08617e-31 - 4.61067e-31 - 5.17998e-31 - 6.14443e-31 - 9.42685e-31 - 2.49069e-30 - 1.0191e-29 - 4.81306e-29 - 2.31487e-28 - 1.09965e-27 - 5.12648e-27 - 2.34241e-26 - 1.04875e-25 - 4.6007e-25 - 1.97749e-24 - 8.32801e-24 - 3.43641e-23 - 1.38933e-22 - 5.50355e-22 - 2.13608e-21 - 8.12322e-21 - 3.02675e-20 - 1.105e-19 - 3.9526e-19 - 1.38529e-18 - 4.75704e-18 - 1.60055e-17 - 5.27641e-17 - 1.70429e-16 - 5.3937e-16 - 1.6725e-15 - 5.0814e-15 - 1.51264e-14 - 4.41191e-14 - 1.26082e-13 - 3.53035e-13 - 9.68542e-13 - 2.60349e-12 - 6.85694e-12 - 1.76946e-11 - 4.47394e-11 - 1.10834e-10 - 2.69027e-10 - 6.39816e-10 - 1.49091e-09 - 3.40395e-09 - 7.61469e-09 - 1.66901e-08 - 3.58427e-08 - 7.5419e-08 - 1.55488e-07 - 3.14088e-07 - 6.21643e-07 - 1.2055e-06 - 2.29051e-06 - 4.26416e-06 - 7.77805e-06 - 1.3901e-05 - 2.4342e-05 - 4.17643e-05 - 7.02086e-05 - 0.000115641 - 0.000186625 - 0.000295098 - 0.000457192 - 0.000694014 - 0.00103222 - 0.00150424 - 0.00214782 - 0.00300479 - 0.00411877 - 0.00553169 - 0.00727922 - 0.00938532 - 0.0118563 - 0.0146753 - 0.0177977 - 0.0211483 - 0.0246221 - 0.0280875 - 0.0313934 - 0.0343799 - 0.0368904 - 0.0387854 - 0.0399557 - 0.0403326 - 0.0398957 - 0.0386751 - 0.036749 - 0.0342377 - 0.0312926 - 0.0280847 - 0.0247919 - 0.0215871 - 0.0186297 - 0.0160581 - 0.013987 - 0.0125051 - 0.0116762 - 0.0115397 - 0.0121108 - 0.0133811 - 0.0153167 - 0.0178562 - 0.0209094 - 0.0243561 - 0.0280473 - 0.0318096 - 0.0354525 - 0.0387797 - 0.041602 - 0.0437517 - 0.0450968 - 0.045552 - 0.0450867 - 0.043727 - 0.041553 - 0.03869 - 0.0352968 - 0.031551 - 0.027633 - 0.0237127 - 0.0199375 - 0.0164247 - 0.0132575 - 0.0104848 - 0.00812456 - 0.00616843 - 0.00458866 - 0.00334453 - 0.00238847 - 0.00167125 - 0.00114578 - 0.000769656 - 0.000506558 - 0.000326662 - 0.000206398 - 0.000127776 - 7.75048e-05 - 4.60623e-05 - 2.68225e-05 - 1.53035e-05 - 8.55494e-06 - 4.68577e-06 - 2.51468e-06 - 1.32227e-06 - 6.81232e-07 - 3.4388e-07 - 1.70081e-07 - 8.24215e-08 - 3.91348e-08 - 1.82063e-08 - 8.29885e-09 - 3.70639e-09 - 1.62189e-09 - 6.95389e-10 - 2.92132e-10 - 1.2026e-10 - 4.85509e-11 - 1.93322e-11 - 7.90272e-12 - 4.15729e-12 - 4.69603e-12 - 1.00995e-11 - 2.54898e-11 - 6.48543e-11 - 1.62322e-10 - 3.9828e-10 - 9.57567e-10 - 2.25575e-09 - 5.20657e-09 - 1.17747e-08 - 2.60905e-08 - 5.66438e-08 - 1.20492e-07 - 2.51133e-07 - 5.12843e-07 - 1.02613e-06 - 2.01167e-06 - 3.8641e-06 - 7.27238e-06 - 1.34104e-05 - 2.42295e-05 - 4.28927e-05 - 7.43975e-05 - 0.000126436 - 0.000210533 - 0.000343484 - 0.000549073 - 0.000859983 - 0.00131974 - 0.00198436 - 0.00292342 - 0.00421986 - 0.00596819 - 0.00827035 - 0.011229 - 0.0149381 - 0.019471 - 0.0248666 - 0.0311158 - 0.0381491 - 0.0458271 - 0.0539385 - 0.062203 - 0.0702847 - 0.0778121 - 0.0844055 - 0.0897079 - 0.0934174 - 0.0953152 - 0.095287 - 0.0933346 - 0.0895758 - 0.0842323 - 0.0776088 - 0.0700637 - 0.0619781 - 0.0537245 - 0.0456401 - 0.0380067 - 0.0310392 - 0.0248817 - 0.0196118 - 0.0152499 - 0.0117717 - 0.00912236 - 0.00722982 - 0.00601571 - 0.00540334 - 0.0053222 - 0.00570923 - 0.00650744 - 0.00766253 - 0.00911876 - 0.0108148 - 0.0126804 - 0.0146355 - 0.0165902 - 0.0184485 - 0.020113 - 0.0214914 - 0.022504 - 0.0230902 - 0.0232141 - 0.022868 - 0.0220728 - 0.020876 - 0.0193474 - 0.0175725 - 0.0156453 - 0.0136615 - 0.0117119 - 0.00987887 - 0.00823454 - 0.00684165 - 0.00575724 - 0.00503837 - 0.00474935 - 0.00496969 - 0.00580156 - 0.00737575 - 0.00985451 - 0.0134302 - 0.0183181 - 0.0247429 - 0.0329181 - 0.0430191 - 0.0551515 - 0.0693182 - 0.0853881 - 0.103073 - 0.121914 - 0.141292 - 0.160443 - 0.178512 - 0.194604 - 0.207861 - 0.217536 - 0.223062 - 0.224109 - 0.220611 - 0.212781 - 0.201084 - 0.18619 - 0.168917 - 0.15015 - 0.130773 - 0.111595 - 0.0933057 - 0.0764379 - 0.0613545 - 0.0482527 - 0.037182 - 0.0280725 - 0.0207666 - 0.0150518 - 0.0106892 - 0.00743777 - 0.00507078 - 0.00338723 - 0.00221692 - 0.00142165 - 0.000893251 - 0.000549908 - 0.000331699 - 0.000196035 - 0.000113517 - 6.44059e-05 - 3.58036e-05 - 1.95013e-05 - 1.04073e-05 - 5.4419e-06 - 2.78804e-06 - 1.39954e-06 - 6.88345e-07 - 3.31715e-07 - 1.56625e-07 - 7.24592e-08 - 3.28445e-08 - 1.45871e-08 - 6.34764e-09 - 2.7064e-09 - 1.1306e-09 - 4.62768e-10 - 1.8559e-10 - 7.29258e-11 - 2.80766e-11 - 1.05912e-11 - 3.91457e-12 - 1.41762e-12 - 5.03003e-13 - 1.74872e-13 - 5.95668e-14 - 1.98805e-14 - 6.50108e-15 - 2.08296e-15 - 6.53904e-16 - 2.01133e-16 - 6.06164e-17 - 1.78992e-17 - 5.17861e-18 - 1.46801e-18 - 4.07741e-19 - 1.10962e-19 - 2.95871e-20 - 7.72981e-21 - 1.97881e-21 - 4.96994e-22 - 1.2513e-22 - 4.27973e-23 - 6.25412e-23 - 2.27169e-22 - 8.96772e-22 - 3.49045e-21 - 1.3316e-20 - 4.97752e-20 - 1.82301e-19 - 6.54186e-19 - 2.30012e-18 - 7.92386e-18 - 2.67461e-17 - 8.84545e-17 - 2.86627e-16 - 9.1002e-16 - 2.83088e-15 - 8.62837e-15 - 2.57675e-14 - 7.5397e-14 - 2.16158e-13 - 6.07193e-13 - 1.67116e-12 - 4.50658e-12 - 1.19073e-11 - 3.08258e-11 - 7.81904e-11 - 1.94325e-10 - 4.73197e-10 - 1.12899e-09 - 2.63923e-09 - 6.04507e-09 - 1.35663e-08 - 2.98304e-08 - 6.42676e-08 - 1.35663e-07 - 2.80589e-07 - 5.6861e-07 - 1.12901e-06 - 2.19641e-06 - 4.18667e-06 - 7.81917e-06 - 1.43083e-05 - 2.5654e-05 - 4.50668e-05 - 7.75704e-05 - 0.000130819 - 0.000216165 - 0.000349973 - 0.000555164 - 0.000862869 - 0.00131403 - 0.00196066 - 0.00286641 - 0.00410592 - 0.00576262 - 0.00792443 - 0.0106771 - 0.0140955 - 0.0182326 - 0.0231079 - 0.0286958 - 0.0349164 - 0.0416297 - 0.0486353 - 0.0556796 - 0.0624697 - 0.0686941 - 0.0740496 - 0.07827 - 0.0811551 - 0.0825952 - 0.0825891 - 0.0812525 - 0.0788147 - 0.0756052 - 0.0720296 - 0.068538 - 0.0655882 - 0.0636076 - 0.0629552 - 0.0638897 - 0.0665429 - 0.0709037 - 0.0768119 - 0.0839642 - 0.0919334 - 0.100197 - 0.108179 - 0.115294 - 0.120995 - 0.124823 - 0.126445 - 0.125678 - 0.122506 - 0.117071 - 0.10966 - 0.100669 - 0.0905633 - 0.0798348 - 0.0689607 - 0.0583676 - 0.0484063 - 0.0393372 - 0.0313261 - 0.0244505 - 0.0187122 - 0.0140553 - 0.0103846 - 0.00758494 - 0.00553752 - 0.00413347 - 0.00328399 - 0.00292721 - 0.00303184 - 0.00359822 - 0.00465668 - 0.00626355 - 0.00849453 - 0.0114355 - 0.0151707 - 0.019769 - 0.0252679 - 0.0316588 - 0.0388727 - 0.0467703 - 0.0551377 - 0.0636899 - 0.0720829 - 0.0799342 - 0.08685 - 0.092458 - 0.0964398 - 0.0985609 - 0.0986938 - 0.0968305 - 0.0930832 - 0.0876733 - 0.0809096 - 0.0731593 - 0.0648151 - 0.0562625 - 0.0478519 - 0.0398764 - 0.0325589 - 0.0260472 - 0.0204168 - 0.0156802 - 0.0117992 - 0.00869946 - 0.00628445 - 0.00444814 - 0.0030848 - 0.0020961 - 0.00139552 - 0.00091032 - 0.000581822 - 0.000364354 - 0.000223559 - 0.0001344 - 7.91667e-05 - 4.56902e-05 - 2.5837e-05 - 1.43154e-05 - 7.77184e-06 - 4.13516e-06 - 2.15827e-06 - 1.10966e-06 - 5.72705e-07 - 3.20383e-07 - 2.41707e-07 - 3.04402e-07 - 5.4587e-07 - 1.09169e-06 - 2.20877e-06 - 4.41012e-06 - 8.641e-06 - 1.65944e-05 - 3.12269e-05 - 5.75758e-05 - 0.000104013 - 0.000184109 - 0.0003193 - 0.000542574 - 0.00090335 - 0.00147364 - 0.00235538 - 0.00368866 - 0.00565995 - 0.0085093 - 0.0125346 - 0.0180912 - 0.0255834 - 0.0354476 - 0.048123 - 0.0640111 - 0.0834246 - 0.10653 - 0.133286 - 0.163393 - 0.196254 - 0.230963 - 0.266319 - 0.300885 - 0.333069 - 0.361247 - 0.383895 - 0.399721 - 0.407791 - 0.407621 - 0.399219 - 0.383092 - 0.36019 - 0.331816 - 0.299502 - 0.264874 - 0.229517 - 0.194862 - 0.162098 - 0.132119 - 0.105509 - 0.0825559 - 0.0632915 - 0.0475422 - 0.0349905 - 0.0252324 - 0.017828 - 0.012342 - 0.00837148 - 0.00556362 - 0.00362284 - 0.00231142 - 0.00144492 - 0.000885006 - 0.000531112 - 0.000312293 - 0.000179918 - 0.000101561 - 5.61709e-05 - 3.04393e-05 - 1.6162e-05 - 8.408e-06 - 4.28575e-06 - 2.14042e-06 - 1.04738e-06 - 5.0217e-07 - 2.35902e-07 - 1.0858e-07 - 4.89673e-08 - 2.16371e-08 - 9.36772e-09 - 3.97414e-09 - 1.65288e-09 - 6.76192e-10 - 2.78154e-10 - 1.30846e-10 - 1.07446e-10 - 1.85993e-10 - 4.31066e-10 - 1.04002e-09 - 2.48173e-09 - 5.81045e-09 - 1.33319e-08 - 2.99725e-08 - 6.60228e-08 - 1.42496e-07 - 3.01333e-07 - 6.24351e-07 - 1.2675e-06 - 2.52117e-06 - 4.91355e-06 - 9.38262e-06 - 1.75546e-05 - 3.21805e-05 - 5.78006e-05 - 0.000101721 - 0.000175397 - 0.000296328 - 0.000490523 - 0.00079558 - 0.00126428 - 0.00196853 - 0.00300315 - 0.00448898 - 0.00657439 - 0.00943411 - 0.0132642 - 0.0182726 - 0.0246636 - 0.0326174 - 0.0422648 - 0.0536592 - 0.0667494 - 0.0813555 - 0.0971547 - 0.113678 - 0.130325 - 0.146391 - 0.161116 - 0.17374 - 0.183568 - 0.190034 - 0.192754 - 0.191563 - 0.186533 - 0.177966 - 0.166363 - 0.152375 - 0.136743 - 0.120236 - 0.103586 - 0.0874387 - 0.0723175 - 0.058603 - 0.0465301 - 0.036198 - 0.0275912 - 0.0206061 - 0.0150784 - 0.0108107 - 0.00759432 - 0.00522709 - 0.00352508 - 0.00232924 - 0.00150798 - 0.000956565 - 0.000594525 - 0.000362045 - 0.000216019 - 0.000126287 - 7.23372e-05 - 4.05977e-05 - 2.23243e-05 - 1.2028e-05 - 6.34954e-06 - 3.2842e-06 - 1.66439e-06 - 8.26447e-07 - 4.0208e-07 - 1.91667e-07 - 8.95198e-08 - 4.09664e-08 - 1.83685e-08 - 8.06966e-09 - 3.47355e-09 - 1.46497e-09 - 6.05372e-10 - 2.45105e-10 - 9.7234e-11 - 3.77939e-11 - 1.43933e-11 - 5.37079e-12 - 1.9636e-12 - 7.03401e-13 - 2.46882e-13 - 8.49013e-14 - 2.86072e-14 - 9.44439e-15 - 3.05498e-15 - 9.68234e-16 - 3.00669e-16 - 9.14817e-17 - 2.7272e-17 - 7.96591e-18 - 2.27978e-18 - 6.39307e-19 - 1.7579e-19 - 4.79091e-20 - 1.49851e-20 - 1.3075e-20 - 3.86379e-20 - 1.42464e-19 - 5.24334e-19 - 1.89316e-18 - 6.6979e-18 - 2.32182e-17 - 7.886e-17 - 2.62435e-16 - 8.55703e-16 - 2.73377e-15 - 8.55729e-15 - 2.62451e-14 - 7.88674e-14 - 2.32211e-13 - 6.69893e-13 - 1.8935e-12 - 5.24398e-12 - 1.42297e-11 - 3.78324e-11 - 9.85533e-11 - 2.51544e-10 - 6.29065e-10 - 1.54139e-09 - 3.70056e-09 - 8.70479e-09 - 2.00626e-08 - 4.53056e-08 - 1.00243e-07 - 2.17317e-07 - 4.61603e-07 - 9.60688e-07 - 1.95899e-06 - 3.914e-06 - 7.66208e-06 - 1.46964e-05 - 2.76192e-05 - 5.08571e-05 - 9.17551e-05 - 0.000162199 - 0.000280936 - 0.00047677 - 0.000792781 - 0.00129164 - 0.00206196 - 0.0032253 - 0.00494329 - 0.00742376 - 0.0109245 - 0.015753 - 0.0222597 - 0.0308241 - 0.0418311 - 0.0556391 - 0.0725392 - 0.0927117 - 0.116181 - 0.142782 - 0.172134 - 0.203646 - 0.23654 - 0.269907 - 0.302784 - 0.334251 - 0.36353 - 0.390069 - 0.413606 - 0.434188 - 0.452139 - 0.467992 - 0.482371 - 0.495855 - 0.508845 - 0.521448 - 0.533411 - 0.54411 - 0.552604 - 0.557745 - 0.558334 - 0.553292 - 0.541826 - 0.523566 - 0.49865 - 0.46775 - 0.432034 - 0.393086 - 0.35278 - 0.313139 - 0.2762 - 0.243885 - 0.2179 - 0.199659 - 0.19023 - 0.190292 - 0.200109 - 0.219505 - 0.247844 - 0.284017 - 0.326452 - 0.373144 - 0.421726 - 0.469578 - 0.513978 - 0.552279 - 0.582113 - 0.601568 - 0.60936 - 0.604932 - 0.588498 - 0.561014 - 0.524074 - 0.479755 - 0.430422 - 0.378526 - 0.32641 - 0.276152 - 0.229449 - 0.187561 - 0.151296 - 0.121041 - 0.0968234 - 0.0783825 - 0.0652508 - 0.0568241 - 0.052421 - 0.0513267 - 0.0528233 - 0.0562096 - 0.0608144 - 0.0660069 - 0.0712085 - 0.0759058 - 0.0796663 - 0.0821531 - 0.083138 - 0.0825095 - 0.0802724 - 0.0765404 - 0.0715192 - 0.0654835 - 0.0587491 - 0.0516444 - 0.0444831 - 0.0375424 - 0.031047 - 0.0251611 - 0.0199871 - 0.0155706 - 0.0119103 - 0.00897034 - 0.00669384 - 0.00501675 - 0.00387995 - 0.00323985 - 0.00307697 - 0.00340228 - 0.00426134 - 0.00573584 - 0.00794229 - 0.011027 - 0.015157 - 0.0205062 - 0.0272364 - 0.0354751 - 0.0452892 - 0.0566592 - 0.0694565 - 0.0834268 - 0.0981838 - 0.113218 - 0.127916 - 0.141604 - 0.153589 - 0.163224 - 0.169959 - 0.173396 - 0.17333 - 0.169763 - 0.162911 - 0.153177 - 0.141116 - 0.127378 - 0.112654 - 0.0976202 - 0.0828835 - 0.0689498 - 0.0561998 - 0.0448822 - 0.0351196 - 0.0269254 - 0.0202261 - 0.0148867 - 0.0107355 - 0.00758544 - 0.00525143 - 0.00356214 - 0.00236745 - 0.00154166 - 0.000983633 - 0.000614914 - 0.000376646 - 0.000226043 - 0.000132922 - 7.65918e-05 - 4.32595e-05 - 2.39797e-05 - 1.31129e-05 - 7.21922e-06 - 4.30366e-06 - 3.33996e-06 - 4.01773e-06 - 6.68526e-06 - 1.24923e-05 - 2.37754e-05 - 4.47731e-05 - 8.28123e-05 - 0.000150164 - 0.000266831 - 0.000464579 - 0.000792543 - 0.00132472 - 0.00216951 - 0.00348126 - 0.0054733 - 0.00843136 - 0.0127257 - 0.0188194 - 0.0272687 - 0.0387134 - 0.053851 - 0.0733945 - 0.09801 - 0.128237 - 0.164397 - 0.206496 - 0.254135 - 0.306448 - 0.362063 - 0.41913 - 0.47539 - 0.528309 - 0.575259 - 0.613727 - 0.641541 - 0.657068 - 0.659377 - 0.648327 - 0.624588 - 0.589567 - 0.545274 - 0.494136 - 0.438774 - 0.381789 - 0.325571 - 0.272154 - 0.223122 - 0.179578 - 0.142167 - 0.111134 - 0.086417 - 0.0677452 - 0.0547365 - 0.04698 - 0.0440953 - 0.0457651 - 0.0517401 - 0.0618192 - 0.075809 - 0.0934719 - 0.114469 - 0.138309 - 0.164311 - 0.191592 - 0.21908 - 0.245557 - 0.269731 - 0.29033 - 0.306206 - 0.316433 - 0.320401 - 0.317866 - 0.308982 - 0.29428 - 0.274615 - 0.251088 - 0.224938 - 0.197441 - 0.169805 - 0.143086 - 0.118136 - 0.0955664 - 0.0757468 - 0.0588247 - 0.0447603 - 0.0333705 - 0.0243763 - 0.0174466 - 0.0122347 - 0.00840639 - 0.00565932 - 0.00373302 - 0.0024127 - 0.001528 - 0.000948448 - 0.000577417 - 0.000345638 - 0.0002051 - 0.000123843 - 8.18518e-05 - 6.81308e-05 - 7.89389e-05 - 0.00011715 - 0.000192719 - 0.000324255 - 0.000541737 - 0.000890371 - 0.00143553 - 0.00226859 - 0.00351311 - 0.00533082 - 0.00792609 - 0.0115475 - 0.0164852 - 0.0230612 - 0.0316131 - 0.0424685 - 0.0559121 - 0.0721461 - 0.0912487 - 0.113135 - 0.137526 - 0.163937 - 0.191679 - 0.219893 - 0.247604 - 0.273794 - 0.297493 - 0.317866 - 0.334291 - 0.346416 - 0.354179 - 0.357793 - 0.357701 - 0.354498 - 0.348835 - 0.34133 - 0.332489 - 0.322651 - 0.311975 - 0.30045 - 0.287942 - 0.274258 - 0.25922 - 0.242722 - 0.224784 - 0.205571 - 0.185391 - 0.164673 - 0.143916 - 0.123646 - 0.104359 - 0.0864807 - 0.0703312 - 0.0561129 - 0.0439079 - 0.0336895 - 0.0253425 - 0.0186881 - 0.0135096 - 0.00957556 - 0.00665938 - 0.00455333 - 0.00307793 - 0.00208716 - 0.00147068 - 0.00115451 - 0.00110122 - 0.00131073 - 0.0018224 - 0.00271851 - 0.00412898 - 0.00623645 - 0.00928047 - 0.0135588 - 0.0194236 - 0.0272705 - 0.0375177 - 0.0505746 - 0.0667991 - 0.0864465 - 0.109613 - 0.13618 - 0.165769 - 0.19771 - 0.231042 - 0.26454 - 0.296775 - 0.326212 - 0.351326 - 0.370729 - 0.383301 - 0.388294 - 0.385406 - 0.374811 - 0.357144 - 0.333435 - 0.305011 - 0.273374 - 0.240069 - 0.206562 - 0.174142 - 0.143844 - 0.116417 - 0.0923168 - 0.0717266 - 0.054603 - 0.0407277 - 0.0297646 - 0.0213131 - 0.0149531 - 0.010279 - 0.00692323 - 0.00456881 - 0.00295416 - 0.00187155 - 0.00116173 - 0.000706559 - 0.000421043 - 0.000245834 - 0.000140635 - 7.88285e-05 - 4.32922e-05 - 2.32955e-05 - 1.22821e-05 - 6.34466e-06 - 3.2113e-06 - 1.59255e-06 - 7.73819e-07 - 3.68407e-07 - 1.71859e-07 - 7.85727e-08 - 3.525e-08 - 1.56255e-08 - 7.1029e-09 - 3.90724e-09 - 3.74228e-09 - 6.36823e-09 - 1.3367e-08 - 2.89202e-08 - 6.18851e-08 - 1.29968e-07 - 2.67519e-07 - 5.39551e-07 - 1.06623e-06 - 2.06446e-06 - 3.91651e-06 - 7.27996e-06 - 1.32585e-05 - 2.36592e-05 - 4.13656e-05 - 7.08625e-05 - 0.000118941 - 0.000195606 - 0.000315187 - 0.000497615 - 0.000769759 - 0.00116668 - 0.00173256 - 0.00252093 - 0.00359392 - 0.00502012 - 0.00687063 - 0.00921332 - 0.0121052 - 0.0155835 - 0.019656 - 0.0242919 - 0.0294147 - 0.0348984 - 0.0405679 - 0.0462058 - 0.0515641 - 0.0563813 - 0.0604032 - 0.0634047 - 0.0652108 - 0.0657135 - 0.0648824 - 0.0627676 - 0.0594951 - 0.055254 - 0.0502786 - 0.044827 - 0.0391591 - 0.0335168 - 0.028108 - 0.0230958 - 0.018594 - 0.0146673 - 0.0113361 - 0.00858452 - 0.00636947 - 0.0046305 - 0.0032983 - 0.00230191 - 0.00157407 - 0.00105462 - 0.000692316 - 0.000445298 - 0.000280629 - 0.000173282 - 0.000104836 - 6.21444e-05 - 3.60937e-05 - 2.05399e-05 - 1.14526e-05 - 6.25667e-06 - 3.34904e-06 - 1.75645e-06 - 9.02581e-07 - 4.5444e-07 - 2.24189e-07 - 1.08378e-07 - 5.13693e-08 - 2.39435e-08 - 1.11485e-08 - 5.59702e-09 - 3.92679e-09 - 5.09955e-09 - 9.86095e-09 - 2.114e-08 - 4.55587e-08 - 9.66698e-08 - 2.0116e-07 - 4.10205e-07 - 8.19618e-07 - 1.60458e-06 - 3.07786e-06 - 5.78459e-06 - 1.06521e-05 - 1.9219e-05 - 3.39755e-05 - 5.88488e-05 - 9.98725e-05 - 0.00016607 - 0.000270566 - 0.000431909 - 0.000675535 - 0.00103524 - 0.00155443 - 0.00228684 - 0.0032964 - 0.00465564 - 0.00644253 - 0.00873514 - 0.0116043 - 0.0151046 - 0.0192634 - 0.024071 - 0.0294708 - 0.035353 - 0.0415526 - 0.0478528 - 0.053995 - 0.059695 - 0.0646638 - 0.0686316 - 0.0713724 - 0.0727252 - 0.0726101 - 0.0710369 - 0.068105 - 0.0639941 - 0.058949 - 0.0532582 - 0.0472318 - 0.0411794 - 0.035392 - 0.0301294 - 0.0256128 - 0.0220232 - 0.0195037 - 0.0181646 - 0.0180873 - 0.0193273 - 0.0219132 - 0.025842 - 0.0310708 - 0.0375062 - 0.0449944 - 0.0533132 - 0.0621703 - 0.0712086 - 0.0800204 - 0.0881704 - 0.0952259 - 0.10079 - 0.104537 - 0.106239 - 0.105792 - 0.10322 - 0.0986778 - 0.0924299 - 0.0848289 - 0.0762804 - 0.0672077 - 0.058018 - 0.049073 - 0.0406687 - 0.0330228 - 0.0262727 - 0.0204801 - 0.0156422 - 0.0117058 - 0.00858298 - 0.00616616 - 0.00434041 - 0.00299359 - 0.00202311 - 0.00133989 - 0.000870016 - 0.000554601 - 0.00034854 - 0.000218741 - 0.000142232 - 0.000104541 - 9.8566e-05 - 0.000124155 - 0.000188521 - 0.000307637 - 0.000508725 - 0.000833915 - 0.00134509 - 0.00212979 - 0.00330782 - 0.00503791 - 0.00752352 - 0.0110162 - 0.0158148 - 0.0222592 - 0.0307153 - 0.0415515 - 0.0551058 - 0.071643 - 0.0913074 - 0.114073 - 0.1397 - 0.167701 - 0.197328 - 0.227588 - 0.257282 - 0.285075 - 0.309593 - 0.329531 - 0.34377 - 0.35148 - 0.352196 - 0.345872 - 0.332879 - 0.313973 - 0.29022 - 0.262897 - 0.233379 - 0.203026 - 0.173082 - 0.144596 - 0.118376 - 0.0949657 - 0.0746562 - 0.057512 - 0.0434158 - 0.0321176 - 0.0232853 - 0.0165482 - 0.0115343 - 0.00789618 - 0.00532826 - 0.00357577 - 0.00243735 - 0.00176332 - 0.00145159 - 0.00144239 - 0.00171284 - 0.00227175 - 0.00315462 - 0.00441846 - 0.0061359 - 0.00838817 - 0.0112563 - 0.0148106 - 0.019099 - 0.0241341 - 0.0298819 - 0.0362517 - 0.0430913 - 0.0501867 - 0.0572696 - 0.064032 - 0.0701466 - 0.0752928 - 0.0791839 - 0.0815939 - 0.0823787 - 0.0814908 - 0.0789841 - 0.075008 - 0.069793 - 0.0636288 - 0.0568371 - 0.0497447 - 0.0426579 - 0.0358416 - 0.0295062 - 0.0238 - 0.0188094 - 0.0145651 - 0.0110507 - 0.00821493 - 0.00598368 - 0.00427072 - 0.00298721 - 0.00204851 - 0.00137888 - 0.000914107 - 0.0006025 - 0.000404956 - 0.000294455 - 0.000255581 - 0.000284413 - 0.000389119 - 0.00059143 - 0.000929092 - 0.0014592 - 0.00226219 - 0.00344601 - 0.00514967 - 0.00754534 - 0.0108375 - 0.0152582 - 0.0210566 - 0.0284828 - 0.0377642 - 0.0490773 - 0.0625148 - 0.0780522 - 0.0955182 - 0.114573 - 0.134702 - 0.155224 - 0.175321 - 0.194088 - 0.210595 - 0.223966 - 0.233452 - 0.238502 - 0.238817 - 0.234375 - 0.225439 - 0.212528 - 0.196368 - 0.177823 - 0.157822 - 0.13728 - 0.117031 - 0.0977797 - 0.0800658 - 0.0642531 - 0.0505343 - 0.0389512 - 0.0294236 - 0.0217826 - 0.0158036 - 0.0112366 - 0.00782971 - 0.00534666 - 0.00357803 - 0.00234654 - 0.00150811 - 0.000949846 - 0.000586259 - 0.0003546 - 0.000210184 - 0.000122087 - 6.94932e-05 - 3.87633e-05 - 2.11886e-05 - 1.13497e-05 - 5.95747e-06 - 3.06434e-06 - 1.54456e-06 - 7.62898e-07 - 3.69248e-07 - 1.75129e-07 - 8.13929e-08 - 3.7068e-08 - 1.65423e-08 - 7.23393e-09 - 3.0998e-09 - 1.30158e-09 - 5.35535e-10 - 2.15914e-10 - 8.52996e-11 - 3.30208e-11 - 1.25257e-11 - 4.65589e-12 - 1.69628e-12 - 6.07133e-13 - 2.17868e-13 - 9.17968e-14 - 8.26531e-14 - 1.80758e-13 - 5.04118e-13 - 1.4277e-12 - 3.9783e-12 - 1.0868e-11 - 2.90954e-11 - 7.63304e-11 - 1.96231e-10 - 4.94351e-10 - 1.2204e-09 - 2.95232e-09 - 6.9988e-09 - 1.62586e-08 - 3.70116e-08 - 8.25642e-08 - 1.80486e-07 - 3.86628e-07 - 8.11596e-07 - 1.66949e-06 - 3.36533e-06 - 6.64764e-06 - 1.28678e-05 - 2.44085e-05 - 4.53704e-05 - 8.26422e-05 - 0.000147512 - 0.000258019 - 0.000442253 - 0.000742825 - 0.00122264 - 0.001972 - 0.00311681 - 0.00482735 - 0.00732661 - 0.0108966 - 0.0158809 - 0.0226805 - 0.0317412 - 0.04353 - 0.0584987 - 0.0770365 - 0.0994122 - 0.125711 - 0.155777 - 0.189156 - 0.225077 - 0.262443 - 0.299868 - 0.335752 - 0.368386 - 0.396083 - 0.417326 - 0.430901 - 0.436023 - 0.432407 - 0.420307 - 0.400488 - 0.374157 - 0.34285 - 0.308297 - 0.27227 - 0.236449 - 0.202305 - 0.171016 - 0.143419 - 0.12 - 0.10091 - 0.0860138 - 0.0749435 - 0.0671671 - 0.0620535 - 0.0589346 - 0.0571569 - 0.0561247 - 0.0553308 - 0.0543762 - 0.052979 - 0.0509737 - 0.0483016 - 0.0449951 - 0.0411568 - 0.0369364 - 0.0325078 - 0.0280481 - 0.0237199 - 0.0196589 - 0.0159663 - 0.0127065 - 0.00990863 - 0.00757099 - 0.00566812 - 0.00415785 - 0.00298841 - 0.00210451 - 0.00145211 - 0.000981719 - 0.000650297 - 0.00042206 - 0.000268394 - 0.000167228 - 0.00010209 - 6.1065e-05 - 3.57882e-05 - 2.05505e-05 - 1.15623e-05 - 6.37384e-06 - 3.44267e-06 - 1.82192e-06 - 9.44718e-07 - 4.79996e-07 - 2.39019e-07 - 1.16785e-07 - 5.63189e-08 - 2.75876e-08 - 1.55066e-08 - 1.35065e-08 - 2.06346e-08 - 4.12348e-08 - 8.71701e-08 - 1.83491e-07 - 3.79656e-07 - 7.70155e-07 - 1.53093e-06 - 2.98182e-06 - 5.69044e-06 - 1.06401e-05 - 1.94933e-05 - 3.49914e-05 - 6.15423e-05 - 0.000106053 - 0.000179064 - 0.000296231 - 0.000480165 - 0.000762583 - 0.00118664 - 0.00180922 - 0.0027027 - 0.00395586 - 0.00567312 - 0.00797149 - 0.0109747 - 0.0148042 - 0.0195665 - 0.0253384 - 0.03215 - 0.0399688 - 0.0486854 - 0.0581051 - 0.0679468 - 0.0778511 - 0.0873986 - 0.0961378 - 0.10362 - 0.109438 - 0.113265 - 0.11489 - 0.114237 - 0.111386 - 0.106565 - 0.100144 - 0.092608 - 0.0845337 - 0.0765571 - 0.069347 - 0.0635806 - 0.0599254 - 0.0590243 - 0.0614823 - 0.0678499 - 0.0786025 - 0.0941094 - 0.114595 - 0.140095 - 0.170403 - 0.205033 - 0.243187 - 0.28375 - 0.325307 - 0.366205 - 0.404639 - 0.438768 - 0.466855 - 0.487399 - 0.499265 - 0.501787 - 0.494819 - 0.478757 - 0.454493 - 0.423343 - 0.386921 - 0.347013 - 0.305437 - 0.263913 - 0.223964 - 0.186845 - 0.15351 - 0.124609 - 0.100516 - 0.0813752 - 0.0671528 - 0.0576947 - 0.0527714 - 0.0521107 - 0.0554152 - 0.0623635 - 0.0726003 - 0.0857196 - 0.101248 - 0.118632 - 0.137242 - 0.156382 - 0.175324 - 0.193347 - 0.209787 - 0.224084 - 0.235824 - 0.244759 - 0.250819 - 0.25409 - 0.254781 - 0.253179 - 0.249598 - 0.24433 - 0.237611 - 0.229608 - 0.220414 - 0.210075 - 0.198611 - 0.186058 - 0.172493 - 0.15806 - 0.142974 - 0.127518 - 0.112024 - 0.0968447 - 0.0823232 - 0.0687639 - 0.056409 - 0.0454241 - 0.0358933 - 0.0278226 - 0.0211511 - 0.0157665 - 0.0115222 - 0.00825423 - 0.0057959 - 0.0039887 - 0.00269018 - 0.00177808 - 0.00115167 - 0.000731004 - 0.000454751 - 0.000277398 - 0.000166222 - 9.846e-05 - 5.88987e-05 - 3.79727e-05 - 3.04441e-05 - 3.46788e-05 - 5.25582e-05 - 9.00977e-05 - 0.000158898 - 0.000278618 - 0.000480699 - 0.000813605 - 0.00134977 - 0.00219438 - 0.00349573 - 0.00545668 - 0.00834609 - 0.0125084 - 0.0183688 - 0.0264315 - 0.0372673 - 0.0514867 - 0.0696987 - 0.0924523 - 0.120163 - 0.153035 - 0.190972 - 0.233514 - 0.27978 - 0.328461 - 0.377845 - 0.425897 - 0.470391 - 0.509067 - 0.539825 - 0.56091 - 0.571077 - 0.569717 - 0.55691 - 0.533424 - 0.500637 - 0.460399 - 0.414867 - 0.366308 - 0.31692 - 0.268674 - 0.223194 - 0.181698 - 0.144974 - 0.113407 - 0.0870373 - 0.0656418 - 0.048819 - 0.0360773 - 0.0269108 - 0.0208626 - 0.0175701 - 0.0167947 - 0.0184339 - 0.0225191 - 0.0291988 - 0.0387082 - 0.0513264 - 0.067323 - 0.0868961 - 0.110107 - 0.136818 - 0.166639 - 0.198893 - 0.232612 - 0.266561 - 0.299299 - 0.329271 - 0.354927 - 0.374855 - 0.387903 - 0.393297 - 0.390711 - 0.3803 - 0.36269 - 0.338907 - 0.310288 - 0.278347 - 0.244652 - 0.210696 - 0.177796 - 0.147015 - 0.119132 - 0.0946281 - 0.0737165 - 0.0563825 - 0.0424423 - 0.0316028 - 0.0235195 - 0.0178455 - 0.0142692 - 0.0125399 - 0.0124803 - 0.0139879 - 0.0170253 - 0.0216027 - 0.0277537 - 0.0355053 - 0.0448476 - 0.0557031 - 0.0679011 - 0.0811603 - 0.0950824 - 0.10916 - 0.1228 - 0.135357 - 0.146187 - 0.154696 - 0.160393 - 0.162941 - 0.162186 - 0.158174 - 0.151147 - 0.141516 - 0.129827 - 0.116706 - 0.102807 - 0.0887582 - 0.0751238 - 0.062369 - 0.0508463 - 0.0407927 - 0.0323401 - 0.0255341 - 0.0203573 - 0.0167544 - 0.0146531 - 0.0139808 - 0.0146745 - 0.0166817 - 0.019956 - 0.0244455 - 0.0300777 - 0.0367437 - 0.0442835 - 0.0524759 - 0.0610361 - 0.0696209 - 0.0778441 - 0.0852996 - 0.0915913 - 0.096366 - 0.0993442 - 0.100347 - 0.0993131 - 0.0963045 - 0.0915008 - 0.0851805 - 0.0776949 - 0.0694355 - 0.0608006 - 0.052164 - 0.0438501 - 0.0361166 - 0.0291461 - 0.0230458 - 0.0178541 - 0.0135526 - 0.0100796 - 0.00734517 - 0.00524441 - 0.00366883 - 0.00251476 - 0.00168889 - 0.00111133 - 0.00071651 - 0.000452625 - 0.00028015 - 0.000169895 - 0.00010095 - 5.87719e-05 - 3.35251e-05 - 1.87373e-05 - 1.02608e-05 - 5.50542e-06 - 2.89426e-06 - 1.49081e-06 - 7.52391e-07 - 3.7205e-07 - 1.80259e-07 - 8.55712e-08 - 3.98012e-08 - 1.81385e-08 - 8.09923e-09 - 3.54342e-09 - 1.51893e-09 - 6.37957e-10 - 2.62531e-10 - 1.05854e-10 - 4.18187e-11 - 1.61872e-11 - 6.13914e-12 - 2.28129e-12 - 8.306e-13 - 2.96305e-13 - 1.03568e-13 - 3.54687e-14 - 1.19015e-14 - 3.91289e-15 - 1.26046e-15 - 3.9783e-16 - 1.23028e-16 - 3.72774e-17 - 1.10669e-17 - 3.21915e-18 - 9.17473e-19 - 2.56202e-19 - 7.00986e-20 - 1.8792e-20 - 4.93598e-21 - 1.27031e-21 - 3.2032e-22 - 7.91398e-23 - 1.91577e-23 - 4.54388e-24 - 1.05596e-24 - 2.4044e-25 - 5.36415e-26 - 1.17255e-26 - 2.51132e-27 - 5.26996e-28 - 1.08355e-28 - 2.18287e-29 - 4.30869e-30 - 8.33293e-31 - 1.57902e-31 - 2.93166e-32 - 5.33307e-33 - 9.50556e-34 - 1.66003e-34 - 2.84046e-35 - 4.76211e-36 - 7.82253e-37 - 1.25902e-37 - 1.98542e-38 - 3.06769e-39 - 4.64415e-40 - 6.88872e-41 - 1.00117e-41 - 1.42565e-42 - 1.98909e-43 - 2.71916e-44 - 3.64209e-45 - 4.77973e-46 - 6.14602e-47 - 7.7432e-48 - 9.55837e-49 - 1.15607e-49 - 1.37e-50 - 1.59073e-51 - 1.8097e-52 - 2.01722e-53 - 2.20312e-54 - 2.35755e-55 - 2.47183e-56 - 2.5393e-57 - 2.55591e-58 - 2.52066e-59 - 2.43568e-60 - 2.30602e-61 - 2.13916e-62 - 1.94428e-63 - 1.73146e-64 - 1.51078e-65 - 1.2916e-66 - 1.08191e-67 - 8.87958e-69 - 7.14052e-70 - 5.62606e-71 - 4.34325e-72 - 3.28521e-73 - 2.43471e-74 - 1.76795e-75 - 1.25785e-76 - 8.76845e-78 - 5.98901e-79 - 4.00797e-80 - 2.62803e-81 - 1.68839e-82 - 1.0628e-83 - 6.55493e-85 - 3.96115e-86 - 2.34537e-87 - 1.36063e-88 - 7.73397e-90 - 4.30728e-91 - 2.35039e-92 - 1.25665e-93 - 6.58303e-95 - 3.37889e-96 - 1.69925e-97 - 8.37298e-99 - 4.04239e-100 - 1.9122e-101 - 8.8627e-103 - 4.02472e-104 - 1.79078e-105 - 7.807e-107 - 3.33476e-108 - 1.39764e-109 - 1.06571e-110 - 1.20963e-109 - 2.89465e-108 - 6.79983e-107 - 1.56509e-105 - 3.52952e-104 - 7.79884e-103 - 1.68842e-101 - 3.58153e-100 - 7.44378e-99 - 1.51585e-97 - 3.0245e-96 - 5.91274e-95 - 1.13256e-93 - 2.12554e-92 - 3.90855e-91 - 7.04204e-90 - 1.24313e-88 - 2.15018e-87 - 3.64391e-86 - 6.05059e-85 - 9.84384e-84 - 1.56916e-82 - 2.45081e-81 - 3.75048e-80 - 5.62342e-79 - 8.26136e-78 - 1.18916e-76 - 1.67712e-75 - 2.31753e-74 - 3.1378e-73 - 4.16256e-72 - 5.41044e-71 - 6.89035e-70 - 8.5978e-69 - 1.05116e-67 - 1.25919e-66 - 1.4779e-65 - 1.69957e-64 - 1.915e-63 - 2.11415e-62 - 2.28686e-61 - 2.42371e-60 - 2.51686e-59 - 2.56079e-58 - 2.55284e-57 - 2.49352e-56 - 2.38637e-55 - 2.23769e-54 - 2.05588e-53 - 1.85069e-52 - 1.63232e-51 - 1.41064e-50 - 1.19443e-49 - 9.90932e-49 - 8.05496e-48 - 6.41535e-47 - 5.00626e-46 - 3.82775e-45 - 2.86755e-44 - 2.10482e-43 - 1.51376e-42 - 1.06668e-41 - 7.36458e-41 - 4.98195e-40 - 3.30208e-39 - 2.14443e-38 - 1.3645e-37 - 8.50695e-37 - 5.19648e-36 - 3.11015e-35 - 1.82386e-34 - 1.04794e-33 - 5.89958e-33 - 3.25418e-32 - 1.75873e-31 - 9.31305e-31 - 4.83195e-30 - 2.45635e-29 - 1.22347e-28 - 5.97085e-28 - 2.85506e-27 - 1.33761e-26 - 6.1402e-26 - 2.76167e-25 - 1.21702e-24 - 5.25487e-24 - 2.22312e-23 - 9.21509e-23 - 3.74261e-22 - 1.48931e-21 - 5.80677e-21 - 2.2183e-20 - 8.30318e-20 - 3.04513e-19 - 1.09422e-18 - 3.85247e-18 - 1.32896e-17 - 4.49183e-17 - 1.48755e-16 - 4.82679e-16 - 1.53456e-15 - 4.7802e-15 - 1.45897e-14 - 4.36302e-14 - 1.2784e-13 - 3.67016e-13 - 1.03239e-12 - 2.84538e-12 - 7.68384e-12 - 2.03309e-11 - 5.27081e-11 - 1.33887e-10 - 3.33231e-10 - 8.12632e-10 - 1.94172e-09 - 4.54597e-09 - 1.04283e-08 - 2.34394e-08 - 5.16215e-08 - 1.11395e-07 - 2.35536e-07 - 4.87981e-07 - 9.90622e-07 - 1.97049e-06 - 3.84067e-06 - 7.33515e-06 - 1.37273e-05 - 2.51733e-05 - 4.52355e-05 - 7.96547e-05 - 0.000137449 - 0.000232424 - 0.000385157 - 0.000625493 - 0.000995518 - 0.00155285 - 0.00237402 - 0.00355735 - 0.00522491 - 0.00752254 - 0.0106172 - 0.0146907 - 0.0199294 - 0.0265096 - 0.0345787 - 0.0442341 - 0.0555004 - 0.0683094 - 0.0824842 - 0.0977307 - 0.11364 - 0.129703 - 0.145333 - 0.159905 - 0.172795 - 0.183427 - 0.191318 - 0.19611 - 0.197601 - 0.195756 - 0.190705 - 0.18273 - 0.172236 - 0.15972 - 0.145736 - 0.130848 - 0.115605 - 0.100507 - 0.0859807 - 0.0723703 - 0.0599273 - 0.0488129 - 0.0391035 - - - - - -2.53423e-32 - -3.15378e-32 - -3.84549e-32 - -4.59406e-32 - -5.37674e-32 - -6.16144e-32 - -6.89444e-32 - -7.42922e-32 - -7.14359e-32 - -2.95895e-32 - 2.06522e-31 - 1.40975e-30 - 7.35251e-30 - 3.60726e-29 - 1.72042e-28 - 8.02691e-28 - 3.66828e-27 - 1.64242e-26 - 7.2051e-26 - 3.09692e-25 - 1.30424e-24 - 5.38172e-24 - 2.17581e-23 - 8.61905e-23 - 3.34529e-22 - 1.27217e-21 - 4.74016e-21 - 1.73052e-20 - 6.19012e-20 - 2.16949e-19 - 7.44995e-19 - 2.5066e-18 - 8.26332e-18 - 2.66907e-17 - 8.44702e-17 - 2.61929e-16 - 7.95792e-16 - 2.36893e-15 - 6.90944e-15 - 1.97456e-14 - 5.52884e-14 - 1.51682e-13 - 4.0773e-13 - 1.07386e-12 - 2.77114e-12 - 7.00658e-12 - 1.73577e-11 - 4.21321e-11 - 1.00201e-10 - 2.33489e-10 - 5.33089e-10 - 1.19253e-09 - 2.61381e-09 - 5.61329e-09 - 1.18113e-08 - 2.43508e-08 - 4.91889e-08 - 9.73549e-08 - 1.88793e-07 - 3.58714e-07 - 6.67805e-07 - 1.21811e-06 - 2.17702e-06 - 3.81218e-06 - 6.54066e-06 - 1.09953e-05 - 1.81104e-05 - 2.92272e-05 - 4.6215e-05 - 7.16003e-05 - 0.000108689 - 0.000161656 - 0.000235577 - 0.000336367 - 0.000470576 - 0.000645036 - 0.000866312 - 0.00113999 - 0.00146982 - 0.00185681 - 0.00229829 - 0.00278727 - 0.00331201 - 0.00385604 - 0.00439875 - 0.00491649 - 0.0053842 - 0.00577737 - 0.00607415 - 0.00625743 - 0.00631647 - 0.00624807 - 0.00605694 - 0.00575536 - 0.00536215 - 0.00490107 - 0.0043989 - 0.00388353 - 0.00338212 - 0.00291963 - 0.00251784 - 0.00219476 - 0.0019644 - 0.00183683 - 0.00181827 - 0.00191123 - 0.00211441 - 0.0024225 - 0.0028259 - 0.00331037 - 0.00385693 - 0.00444204 - 0.00503828 - 0.00561553 - 0.0061427 - 0.00658983 - 0.00693041 - 0.0071435 - 0.00721563 - 0.00714193 - 0.00692656 - 0.00658218 - 0.00612867 - 0.00559119 - 0.00499782 - 0.0043772 - 0.0037562 - 0.00315819 - 0.00260175 - 0.00210005 - 0.00166085 - 0.00128697 - 0.000977109 - 0.000726866 - 0.000529789 - 0.000378345 - 0.000264734 - 0.000181497 - 0.000121917 - 8.02412e-05 - 5.17448e-05 - 3.26944e-05 - 2.02403e-05 - 1.22771e-05 - 7.29648e-06 - 4.24881e-06 - 2.42414e-06 - 1.35514e-06 - 7.42249e-07 - 3.98337e-07 - 2.09454e-07 - 1.0791e-07 - 5.44722e-08 - 2.69416e-08 - 1.3056e-08 - 6.19913e-09 - 2.88397e-09 - 1.31458e-09 - 5.87109e-10 - 2.56914e-10 - 1.10153e-10 - 4.62739e-11 - 1.90459e-11 - 7.67917e-12 - 3.02905e-12 - 1.15766e-12 - 3.97361e-13 - 3.41409e-14 - -2.89899e-13 - -8.92159e-13 - -2.32794e-12 - -5.84648e-12 - -1.43519e-11 - -3.45078e-11 - -8.12912e-11 - -1.87631e-10 - -4.24327e-10 - -9.40231e-10 - -2.04129e-09 - -4.34222e-09 - -9.05016e-09 - -1.84815e-08 - -3.6979e-08 - -7.24953e-08 - -1.39252e-07 - -2.62077e-07 - -4.83275e-07 - -8.73166e-07 - -1.54574e-06 - -2.68109e-06 - -4.55642e-06 - -7.58706e-06 - -1.23783e-05 - -1.97871e-05 - -3.09915e-05 - -4.75597e-05 - -7.15111e-05 - -0.000105352 - -0.000152073 - -0.000215078 - -0.000298041 - -0.000404664 - -0.00053833 - -0.000701682 - -0.000896126 - -0.00112133 - -0.00137479 - -0.00165149 - -0.0019438 - -0.00224163 - -0.00253287 - -0.00280414 - -0.00304174 - -0.00323283 - -0.0033665 - -0.00343488 - -0.00343384 - -0.00336344 - -0.0032279 - -0.00303518 - -0.00279621 - -0.00252384 - -0.00223166 - -0.00193291 - -0.00163942 - -0.00136091 - -0.00110451 - -0.000874518 - -0.000672571 - -0.00049787 - -0.000347626 - -0.000217553 - -0.000102387 - 3.62246e-06 - 0.000106247 - 0.000210868 - 0.000322098 - 0.000443445 - 0.000577031 - 0.000723357 - 0.000881165 - 0.0010474 - 0.00121734 - 0.00138482 - 0.0015427 - 0.00168342 - 0.00179962 - 0.00188484 - 0.00193416 - 0.00194463 - 0.00191564 - 0.00184894 - 0.00174849 - 0.00162005 - 0.00147064 - 0.00130789 - 0.00113938 - 0.000972049 - 0.000811701 - 0.000662676 - 0.000527676 - 0.000407728 - 0.000302261 - 0.000209254 - 0.000125426 - 4.64271e-05 - -3.29605e-05 - -0.000118629 - -0.000216928 - -0.0003344 - -0.000477448 - -0.000651899 - -0.000862487 - -0.00111225 - -0.00140188 - -0.00172921 - -0.00208869 - -0.00247127 - -0.00286449 - -0.003253 - -0.00361946 - -0.0039458 - -0.00421463 - -0.00441082 - -0.00452289 - -0.00454411 - -0.00447319 - -0.00431444 - -0.00407725 - -0.00377526 - -0.00342502 - -0.00304451 - -0.0026516 - -0.00226274 - -0.0018919 - -0.00154988 - -0.00124405 - -0.000978389 - -0.000753916 - -0.000569208 - -0.000421072 - -0.000305196 - -0.000216739 - -0.000150811 - -0.000102817 - -6.86807e-05 - -4.49512e-05 - -2.8826e-05 - -1.81119e-05 - -1.11501e-05 - -6.72565e-06 - -3.97489e-06 - -2.30172e-06 - -1.30592e-06 - -7.25967e-07 - -3.95417e-07 - -2.11023e-07 - -1.10342e-07 - -5.65314e-08 - -2.83775e-08 - -1.39572e-08 - -6.72598e-09 - -3.17579e-09 - -1.46921e-09 - -6.65968e-10 - -2.95774e-10 - -1.28707e-10 - -5.48761e-11 - -2.29245e-11 - -9.38325e-12 - -3.76308e-12 - -1.47867e-12 - -5.69293e-13 - -2.14752e-13 - -7.93733e-14 - -2.87441e-14 - -1.01991e-14 - -3.54576e-15 - -1.2078e-15 - -4.03104e-16 - -1.31818e-16 - -4.22349e-17 - -1.32588e-17 - -4.07825e-18 - -1.22908e-18 - -3.62931e-19 - -1.05004e-19 - -2.9766e-20 - -8.26751e-21 - -2.24991e-21 - -5.99919e-22 - -1.56732e-22 - -4.01224e-23 - -1.00739e-23 - -2.52276e-24 - -8.0687e-25 - -1.016e-24 - -3.58361e-24 - -1.41196e-23 - -5.4951e-23 - -2.09636e-22 - -7.83616e-22 - -2.86998e-21 - -1.02989e-20 - -3.62111e-20 - -1.24746e-19 - -4.21067e-19 - -1.39255e-18 - -4.5124e-18 - -1.43265e-17 - -4.45669e-17 - -1.35837e-16 - -4.05661e-16 - -1.18698e-15 - -3.40301e-15 - -9.55911e-15 - -2.63093e-14 - -7.09476e-14 - -1.87458e-13 - -4.85294e-13 - -1.23096e-12 - -3.05929e-12 - -7.4496e-12 - -1.77739e-11 - -4.15498e-11 - -9.51682e-11 - -2.13576e-10 - -4.69623e-10 - -1.01177e-09 - -2.13577e-09 - -4.41734e-09 - -8.9517e-09 - -1.77741e-08 - -3.45784e-08 - -6.59113e-08 - -1.23098e-07 - -2.25258e-07 - -4.03874e-07 - -7.09493e-07 - -1.2212e-06 - -2.05951e-06 - -3.40311e-06 - -5.50968e-06 - -8.74004e-06 - -1.35843e-05 - -2.06871e-05 - -3.08674e-05 - -4.51271e-05 - -6.46419e-05 - -9.07258e-05 - -0.000124764 - -0.00016811 - -0.000221944 - -0.000287112 - -0.000363934 - -0.000452037 - -0.000550211 - -0.000656332 - -0.000767384 - -0.000879594 - -0.000988694 - -0.00109031 - -0.00118047 - -0.00125612 - -0.00131572 - -0.00135978 - -0.00139119 - -0.00141553 - -0.00144105 - -0.00147842 - -0.00154024 - -0.00164022 - -0.00179215 - -0.0020087 - -0.00229996 - -0.00267211 - -0.00312609 - -0.00365653 - -0.00425123 - -0.00489099 - -0.00555038 - -0.00619899 - -0.00680353 - -0.00733032 - -0.00774812 - -0.00803088 - -0.00816018 - -0.00812687 - -0.00793199 - -0.0075865 - -0.00711016 - -0.00652951 - -0.00587539 - -0.00518015 - -0.00447502 - -0.00378787 - -0.0031416 - -0.00255317 - -0.00203343 - -0.00158747 - -0.0012155 - -0.000914005 - -0.000677014 - -0.000497327 - -0.000367638 - -0.00028146 - -0.000233825 - -0.000221778 - -0.000244658 - -0.000304193 - -0.000404393 - -0.000551247 - -0.0007522 - -0.00101539 - -0.00134866 - -0.00175834 - -0.00224794 - -0.00281677 - -0.00345876 - -0.00416153 - -0.00490608 - -0.00566706 - -0.00641387 - -0.00711248 - -0.00772785 - -0.00822684 - -0.00858114 - -0.00876987 - -0.0087817 - -0.0086159 - -0.00828247 - -0.0078011 - -0.00719927 - -0.00650966 - -0.00576719 - -0.0050062 - -0.00425783 - -0.00354817 - -0.00289707 - -0.00231766 - -0.00181667 - -0.00139522 - -0.00104989 - -0.000774071 - -0.000559185 - -0.000395792 - -0.000274483 - -0.00018651 - -0.000124172 - -8.09996e-05 - -5.17701e-05 - -3.24199e-05 - -1.98921e-05 - -1.19588e-05 - -7.04419e-06 - -4.06548e-06 - -2.29895e-06 - -1.27376e-06 - -6.91503e-07 - -3.67873e-07 - -1.91872e-07 - -9.83377e-08 - -5.00377e-08 - -2.64246e-08 - -1.68924e-08 - -1.70683e-08 - -2.72651e-08 - -5.27365e-08 - -1.05874e-07 - -2.1103e-07 - -4.13329e-07 - -7.93705e-07 - -1.49355e-06 - -2.75377e-06 - -4.97481e-06 - -8.80568e-06 - -1.52717e-05 - -2.59505e-05 - -4.32059e-05 - -7.04818e-05 - -0.000112654 - -0.000176423 - -0.000270707 - -0.000406987 - -0.000599513 - -0.000865274 - -0.00122362 - -0.00169541 - -0.00230165 - -0.00306155 - -0.00399008 - -0.00509515 - -0.00637485 - -0.00781483 - -0.00938655 - -0.0110466 - -0.0127377 - -0.0143909 - -0.0159302 - -0.0172779 - -0.0183611 - -0.0191181 - -0.0195041 - -0.0194959 - -0.0190941 - -0.0183227 - -0.0172274 - -0.0158703 - -0.0143247 - -0.0126685 - -0.0109775 - -0.00931998 - -0.00775291 - -0.00631905 - -0.00504632 - -0.00394853 - -0.00302714 - -0.00227387 - -0.00167355 - -0.00120683 - -0.000852687 - -0.000590298 - -0.000400396 - -0.0002661 - -0.000173275 - -0.000110552 - -6.91084e-05 - -4.23286e-05 - -2.54023e-05 - -1.49365e-05 - -8.60522e-06 - -4.85749e-06 - -2.68657e-06 - -1.45587e-06 - -7.73006e-07 - -4.02142e-07 - -2.04981e-07 - -1.02373e-07 - -5.00948e-08 - -2.4018e-08 - -1.12829e-08 - -5.19323e-09 - -2.34203e-09 - -1.03487e-09 - -4.48039e-10 - -1.90063e-10 - -7.90133e-11 - -3.2227e-11 - -1.2995e-11 - -5.44122e-12 - -3.02022e-12 - -3.51191e-12 - -7.21295e-12 - -1.70439e-11 - -4.05427e-11 - -9.4878e-11 - -2.17679e-10 - -4.89379e-10 - -1.07799e-09 - -2.32661e-09 - -4.92003e-09 - -1.01941e-08 - -2.06951e-08 - -4.11646e-08 - -8.02261e-08 - -1.53195e-07 - -2.86623e-07 - -5.25429e-07 - -9.43742e-07 - -1.66085e-06 - -2.8638e-06 - -4.83831e-06 - -8.00904e-06 - -1.29899e-05 - -2.06426e-05 - -3.21413e-05 - -4.9034e-05 - -7.3294e-05 - -0.000107344 - -0.000154036 - -0.000216573 - -0.000298347 - -0.000402696 - -0.000532562 - -0.00069008 - -0.000876123 - -0.00108985 - -0.00132834 - -0.0015863 - -0.00185609 - -0.00212789 - -0.00239021 - -0.00263063 - -0.00283674 - -0.00299722 - -0.00310279 - -0.0031472 - -0.00312775 - -0.00304563 - -0.00290576 - -0.0027163 - -0.0024879 - -0.00223268 - -0.00196316 - -0.0016913 - -0.00142766 - -0.00118077 - -0.000956844 - -0.000759722 - -0.000591024 - -0.000450497 - -0.000336447 - -0.000246194 - -0.000176512 - -0.000123997 - -8.53456e-05 - -5.75558e-05 - -3.80307e-05 - -2.46216e-05 - -1.56184e-05 - -9.70714e-06 - -5.91131e-06 - -3.52706e-06 - -2.06196e-06 - -1.18109e-06 - -6.62861e-07 - -3.64501e-07 - -1.96387e-07 - -1.03672e-07 - -5.36229e-08 - -2.71753e-08 - -1.34939e-08 - -6.56499e-09 - -3.12945e-09 - -1.46164e-09 - -6.6888e-10 - -2.99912e-10 - -1.31758e-10 - -5.67146e-11 - -2.39194e-11 - -9.88424e-12 - -4.00196e-12 - -1.58759e-12 - -6.17081e-13 - -2.35008e-13 - -8.76919e-14 - -3.20607e-14 - -1.14848e-14 - -4.03098e-15 - -1.38623e-15 - -4.67086e-16 - -1.54204e-16 - -4.98803e-17 - -1.58089e-17 - -4.90919e-18 - -1.49367e-18 - -4.45284e-19 - -1.30064e-19 - -3.72232e-20 - -1.04383e-20 - -2.87015e-21 - -7.81928e-22 - -2.43429e-22 - -2.08622e-22 - -6.12204e-22 - -2.2559e-21 - -8.30244e-21 - -2.99767e-20 - -1.06056e-19 - -3.67643e-19 - -1.24869e-18 - -4.15545e-18 - -1.35494e-17 - -4.32871e-17 - -1.35498e-16 - -4.15571e-16 - -1.2488e-15 - -3.67688e-15 - -1.06072e-14 - -2.99821e-14 - -8.30344e-14 - -2.25315e-13 - -5.99047e-13 - -1.56051e-12 - -3.98301e-12 - -9.96075e-12 - -2.44067e-11 - -5.85955e-11 - -1.37834e-10 - -3.17675e-10 - -7.1738e-10 - -1.58727e-09 - -3.44105e-09 - -7.30916e-09 - -1.52118e-08 - -3.10193e-08 - -6.19757e-08 - -1.21324e-07 - -2.32709e-07 - -4.37337e-07 - -8.05303e-07 - -1.45292e-06 - -2.56841e-06 - -4.44867e-06 - -7.54986e-06 - -1.25543e-05 - -2.04549e-05 - -3.26554e-05 - -5.10821e-05 - -7.82973e-05 - -0.000117598 - -0.000173075 - -0.000249615 - -0.000352798 - -0.000488684 - -0.000663453 - -0.000882913 - -0.00115189 - -0.00147356 - -0.00184879 - -0.00227563 - -0.00274905 - -0.00326099 - -0.00380083 - -0.00435636 - -0.00491499 - -0.00546527 - -0.00599832 - -0.00650907 - -0.00699689 - -0.00746558 - -0.00792243 - -0.00837658 - -0.00883672 - -0.00930857 - -0.00979246 - -0.0102816 - -0.0107611 - -0.0112087 - -0.011596 - -0.0118916 - -0.0120645 - -0.0120879 - -0.0119427 - -0.0116203 - -0.011124 - -0.0104692 - -0.00968215 - -0.00879775 - -0.00785624 - -0.00689989 - -0.00596968 - -0.00510249 - -0.00432903 - -0.00367258 - -0.00314848 - -0.00276429 - -0.00252032 - -0.00241049 - -0.00242322 - -0.00254229 - -0.00274769 - -0.00301656 - -0.00332402 - -0.00364428 - -0.00395183 - -0.00422275 - -0.00443609 - -0.00457514 - -0.00462847 - -0.0045906 - -0.00446213 - -0.00424935 - -0.00396339 - -0.0036188 - -0.00323204 - -0.00281971 - -0.00239708 - -0.0019767 - -0.00156754 - -0.00117457 - -0.000798753 - -0.000437587 - -8.59164e-05 - 0.000262944 - 0.000615954 - 0.000979073 - 0.00135609 - 0.00174768 - 0.00215078 - 0.00255837 - 0.00295977 - 0.00334128 - 0.00368735 - 0.00398195 - 0.00421019 - 0.00435982 - 0.00442255 - 0.00439499 - 0.00427906 - 0.00408186 - 0.003815 - 0.00349351 - 0.00313447 - 0.00275551 - 0.00237344 - 0.00200306 - 0.00165638 - 0.00134211 - 0.00106565 - 0.000829314 - 0.000632838 - 0.00047399 - 0.000349261 - 0.00025451 - 0.000185531 - 0.00013851 - 0.000110364 - 9.89575e-05 - 0.000103216 - 0.000123152 - 0.000159795 - 0.000215048 - 0.000291453 - 0.000391873 - 0.000519088 - 0.000675318 - 0.000861708 - 0.00107781 - 0.00132113 - 0.00158679 - 0.00186744 - 0.00215336 - 0.00243292 - 0.00269324 - 0.0029212 - 0.00310445 - 0.00323254 - 0.00329792 - 0.00329666 - 0.00322882 - 0.0030985 - 0.00291337 - 0.00268396 - 0.00242267 - 0.00214264 - 0.00185669 - 0.00157641 - 0.0013114 - 0.0010689 - 0.00085364 - 0.00066796 - 0.00051211 - 0.000384691 - 0.000283138 - 0.000204184 - 0.000144272 - 9.98799e-05 - 6.77504e-05 - 4.5028e-05 - 2.93217e-05 - 1.87083e-05 - 1.16954e-05 - 7.16366e-06 - 4.29928e-06 - 2.52821e-06 - 1.45697e-06 - 8.23301e-07 - 4.57272e-07 - 2.52047e-07 - 1.4307e-07 - 9.41611e-08 - 8.92735e-08 - 1.29198e-07 - 2.33164e-07 - 4.46224e-07 - 8.54461e-07 - 1.61153e-06 - 2.98178e-06 - 5.40736e-06 - 9.60873e-06 - 1.67298e-05 - 2.854e-05 - 4.77041e-05 - 7.81257e-05 - 0.000125363 - 0.000197097 - 0.000303619 - 0.000458263 - 0.0006777 - 0.000981967 - 0.0013941 - 0.00193921 - 0.00264299 - 0.00352941 - 0.00461791 - 0.00592006 - 0.00743607 - 0.0091516 - 0.0110354 - 0.0130381 - 0.0150932 - 0.0171191 - 0.0190248 - 0.0207155 - 0.0221008 - 0.0231024 - 0.0236615 - 0.0237446 - 0.0233467 - 0.0224919 - 0.0212307 - 0.0196357 - 0.0177941 - 0.0158004 - 0.0137482 - 0.0117235 - 0.00979954 - 0.00803325 - 0.00646427 - 0.00511562 - 0.00399594 - 0.00310269 - 0.00242575 - 0.00195092 - 0.00166287 - 0.00154732 - 0.00159223 - 0.00178801 - 0.00212689 - 0.00260153 - 0.00320325 - 0.00392002 - 0.00473471 - 0.00562383 - 0.006557 - 0.00749741 - 0.00840331 - 0.00923048 - 0.00993535 - 0.0104786 - 0.0108286 - 0.0109643 - 0.0108776 - 0.0105736 - 0.0100705 - 0.00939751 - 0.00859238 - 0.00769753 - 0.00675657 - 0.00581083 - 0.00489651 - 0.0040427 - 0.00327035 - 0.00259211 - 0.00201302 - 0.00153173 - 0.00114196 - 0.000834175 - 0.000597036 - 0.000418679 - 0.000287673 - 0.000193667 - 0.000127749 - 8.25696e-05 - 5.23006e-05 - 3.24804e-05 - 1.98088e-05 - 1.19273e-05 - 7.21506e-06 - 4.61863e-06 - 3.52388e-06 - 3.67649e-06 - 5.15343e-06 - 8.3891e-06 - 1.42612e-05 - 2.42428e-05 - 4.06281e-05 - 6.68359e-05 - 0.000107789 - 0.000170354 - 0.000263818 - 0.000400329 - 0.00059524 - 0.000867232 - 0.0012381 - 0.00173205 - 0.0023745 - 0.00319011 - 0.00420035 - 0.0054206 - 0.00685696 - 0.00850339 - 0.0103395 - 0.0123294 - 0.0144223 - 0.0165546 - 0.0186545 - 0.0206468 - 0.0224604 - 0.0240341 - 0.0253228 - 0.026301 - 0.0269648 - 0.0273304 - 0.0274301 - 0.0273064 - 0.0270053 - 0.0265691 - 0.0260305 - 0.0254095 - 0.0247121 - 0.0239319 - 0.023054 - 0.0220604 - 0.0209356 - 0.0196714 - 0.0182707 - 0.0167493 - 0.0151347 - 0.0134645 - 0.0117821 - 0.0101327 - 0.00855887 - 0.00709693 - 0.0057744 - 0.00460875 - 0.00360736 - 0.00276847 - 0.00208292 - 0.00153623 - 0.00111069 - 0.000787407 - 0.000547808 - 0.000374892 - 0.000253994 - 0.000173243 - 0.000123774 - 9.98233e-05 - 9.87971e-05 - 0.000121408 - 0.000171927 - 0.000258572 - 0.000393997 - 0.000595817 - 0.00088703 - 0.00129616 - 0.00185693 - 0.00260716 - 0.00358686 - 0.00483517 - 0.00638631 - 0.00826471 - 0.0104795 - 0.0130195 - 0.0158483 - 0.018902 - 0.0220888 - 0.0252913 - 0.0283731 - 0.0311875 - 0.0335885 - 0.0354435 - 0.0366455 - 0.0371228 - 0.0368467 - 0.0358338 - 0.0341447 - 0.031878 - 0.0291606 - 0.0261359 - 0.0229518 - 0.0197484 - 0.0166488 - 0.0137522 - 0.0111301 - 0.00882594 - 0.00685741 - 0.00522031 - 0.00389377 - 0.00284564 - 0.00203764 - 0.00142959 - 0.000982723 - 0.000661894 - 0.0004368 - 0.000282432 - 0.000178929 - 0.000111067 - 6.75505e-05 - 4.02538e-05 - 2.35029e-05 - 1.34454e-05 - 7.53639e-06 - 4.13894e-06 - 2.22716e-06 - 1.17423e-06 - 6.0658e-07 - 3.07016e-07 - 1.52255e-07 - 7.3981e-08 - 3.52217e-08 - 1.64312e-08 - 7.51334e-09 - 3.37362e-09 - 1.50265e-09 - 7.0035e-10 - 4.24077e-10 - 4.75335e-10 - 8.76813e-10 - 1.8765e-09 - 4.0748e-09 - 8.72522e-09 - 1.83265e-08 - 3.77229e-08 - 7.60823e-08 - 1.5035e-07 - 2.91111e-07 - 5.5227e-07 - 1.02655e-06 - 1.86959e-06 - 3.33619e-06 - 5.83299e-06 - 9.99237e-06 - 1.67719e-05 - 2.75825e-05 - 4.44448e-05 - 7.0169e-05 - 0.000108544 - 0.000164515 - 0.000244309 - 0.000355477 - 0.000506781 - 0.00070789 - 0.000968832 - 0.00129918 - 0.00170696 - 0.00219744 - 0.0027717 - 0.00342542 - 0.00414779 - 0.00492104 - 0.00572051 - 0.00651551 - 0.00727109 - 0.00795037 - 0.00851749 - 0.00894073 - 0.00919542 - 0.00926631 - 0.0091491 - 0.0088509 - 0.00838944 - 0.0077914 - 0.00708982 - 0.00632108 - 0.00552185 - 0.00472622 - 0.00396352 - 0.00325675 - 0.00262195 - 0.00206825 - 0.00159851 - 0.00121051 - 0.000898163 - 0.00065295 - 0.000465095 - 0.000324594 - 0.00022196 - 0.000148712 - 9.76239e-05 - 6.27917e-05 - 3.95717e-05 - 2.44345e-05 - 1.47829e-05 - 8.76302e-06 - 5.0896e-06 - 2.89635e-06 - 1.61493e-06 - 8.82256e-07 - 4.7225e-07 - 2.47678e-07 - 1.27274e-07 - 6.40809e-08 - 3.1613e-08 - 1.52825e-08 - 7.24349e-09 - 3.37597e-09 - 1.57127e-09 - 7.87334e-10 - 5.49235e-10 - 7.08757e-10 - 1.36716e-09 - 2.92932e-09 - 6.31231e-09 - 1.33936e-08 - 2.78707e-08 - 5.68338e-08 - 1.13558e-07 - 2.22314e-07 - 4.26437e-07 - 8.01454e-07 - 1.47584e-06 - 2.66279e-06 - 4.7073e-06 - 8.15349e-06 - 1.38373e-05 - 2.3009e-05 - 3.74869e-05 - 5.98409e-05 - 9.35953e-05 - 0.000143432 - 0.000215365 - 0.000316842 - 0.000456715 - 0.000645038 - 0.000892611 - 0.00121025 - 0.00160778 - 0.00209273 - 0.00266894 - 0.00333503 - 0.00408317 - 0.00489815 - 0.0057571 - 0.00662998 - 0.00748096 - 0.00827066 - 0.008959 - 0.00950859 - 0.00988805 - 0.010075 - 0.0100581 - 0.00983853 - 0.00942953 - 0.00885532 - 0.00814865 - 0.00734787 - 0.0064935 - 0.00562505 - 0.0047782 - 0.00398281 - 0.00326164 - 0.00263011 - 0.00209661 - 0.00166348 - 0.00132822 - 0.00108488 - 0.000925197 - 0.000839661 - 0.000818171 - 0.000850452 - 0.000926225 - 0.00103522 - 0.00116715 - 0.00131169 - 0.00145858 - 0.00159787 - 0.00172029 - 0.00181768 - 0.00188348 - 0.00191318 - 0.00190459 - 0.00185802 - 0.0017761 - 0.00166357 - 0.00152673 - 0.00137286 - 0.00120957 - 0.00104417 - 0.000883183 - 0.000731926 - 0.000594321 - 0.000472837 - 0.000368587 - 0.000281517 - 0.000210672 - 0.000154471 - 0.000110977 - 7.81216e-05 - 5.38892e-05 - 3.64365e-05 - 2.41667e-05 - 1.57607e-05 - 1.0179e-05 - 6.64527e-06 - 4.62443e-06 - 3.80367e-06 - 4.08572e-06 - 5.6009e-06 - 8.74399e-06 - 1.42407e-05 - 2.32471e-05 - 3.74834e-05 - 5.93981e-05 - 9.23544e-05 - 0.000140822 - 0.000210543 - 0.000308644 - 0.000443633 - 0.000625244 - 0.000864061 - 0.0011709 - 0.00155593 - 0.0020275 - 0.00259089 - 0.00324684 - 0.00399034 - 0.00480954 - 0.00568528 - 0.00659119 - 0.00749458 - 0.00835814 - 0.00914232 - 0.00980831 - 0.0103211 - 0.0106527 - 0.0107844 - 0.0107086 - 0.0104298 - 0.00996373 - 0.00933625 - 0.00858073 - 0.00773524 - 0.00683939 - 0.0059313 - 0.00504505 - 0.00420879 - 0.00344364 - 0.00276337 - 0.00217477 - 0.00167855 - 0.00127055 - 0.000943173 - 0.000686672 - 0.00049038 - 0.000343661 - 0.000236618 - 0.00016055 - 0.000108187 - 7.37676e-05 - 5.3005e-05 - 4.29873e-05 - 4.20568e-05 - 4.9689e-05 - 6.63831e-05 - 9.35609e-05 - 0.000133462 - 0.000189012 - 0.000263645 - 0.00036105 - 0.000484834 - 0.000638098 - 0.000822944 - 0.00103994 - 0.00128764 - 0.00156213 - 0.00185686 - 0.00216261 - 0.00246782 - 0.00275923 - 0.00302271 - 0.00324447 - 0.00341214 - 0.00351599 - 0.00354981 - 0.00351155 - 0.00340353 - 0.0032322 - 0.00300748 - 0.00274185 - 0.00244919 - 0.00214357 - 0.00183819 - 0.00154447 - 0.00127146 - 0.00102557 - 0.000810524 - 0.000627627 - 0.000476183 - 0.000353982 - 0.000257822 - 0.000183984 - 0.000128626 - 8.80794e-05 - 5.90389e-05 - 3.86628e-05 - 2.45949e-05 - 1.49291e-05 - 8.13403e-06 - 2.94857e-06 - -1.74116e-06 - -7.04209e-06 - -1.42113e-05 - -2.48328e-05 - -4.10146e-05 - -6.56039e-05 - -0.000102411 - -0.000156423 - -0.000233969 - -0.000342807 - -0.000492057 - -0.000691949 - -0.000953302 - -0.00128673 - -0.00170157 - -0.00220453 - -0.00279826 - -0.00347991 - -0.00423992 - -0.00506123 - -0.00591925 - -0.0067825 - -0.00761422 - -0.00837483 - -0.00902489 - -0.00952848 - -0.00985649 - -0.00998939 - -0.00991914 - -0.00965003 - -0.00919821 - -0.00859011 - -0.00785987 - -0.00704619 - -0.00618894 - -0.005326 - -0.00449065 - -0.00370973 - -0.00300262 - -0.00238113 - -0.00185009 - -0.0014084 - -0.00105048 - -0.000767675 - -0.000549659 - -0.0003856 - -0.000265039 - -0.000178489 - -0.000117771 - -7.61375e-05 - -4.82267e-05 - -2.99299e-05 - -1.81993e-05 - -1.08426e-05 - -6.32909e-06 - -3.61977e-06 - -2.0284e-06 - -1.11367e-06 - -5.99086e-07 - -3.15758e-07 - -1.63062e-07 - -8.25054e-08 - -4.0902e-08 - -1.98673e-08 - -9.45511e-09 - -4.40887e-09 - -2.01428e-09 - -9.01668e-10 - -3.95464e-10 - -1.69942e-10 - -7.15528e-11 - -2.9518e-11 - -1.19311e-11 - -4.72509e-12 - -1.83346e-12 - -6.97057e-13 - -2.59655e-13 - -9.47633e-14 - -3.3872e-14 - -1.18181e-14 - -3.90035e-15 - -8.29272e-16 - 1.18263e-15 - 4.68453e-15 - 1.39394e-14 - 3.96556e-14 - 1.10235e-13 - 3.00124e-13 - 8.00503e-13 - 2.0918e-12 - 5.35517e-12 - 1.34315e-11 - 3.30047e-11 - 7.94561e-11 - 1.87405e-10 - 4.33047e-10 - 9.80379e-10 - 2.17449e-09 - 4.72525e-09 - 1.006e-08 - 2.09837e-08 - 4.28816e-08 - 8.58558e-08 - 1.68414e-07 - 3.23665e-07 - 6.09432e-07 - 1.12426e-06 - 2.03198e-06 - 3.59822e-06 - 6.24264e-06 - 1.06112e-05 - 1.76715e-05 - 2.88336e-05 - 4.60937e-05 - 7.21939e-05 - 0.000110784 - 0.00016656 - 0.000245348 - 0.00035409 - 0.000500684 - 0.000693638 - 0.000941505 - 0.00125208 - 0.0016314 - 0.00208262 - 0.00260483 - 0.00319202 - 0.00383237 - 0.00450799 - 0.00519525 - 0.00586582 - 0.00648837 - 0.00703083 - 0.0074629 - 0.00775862 - 0.00789868 - 0.00787201 - 0.00767662 - 0.00731945 - 0.00681534 - 0.00618516 - 0.0054536 - 0.00464667 - 0.00378961 - 0.00290508 - 0.00201228 - 0.00112666 - 0.000260383 - -0.000576704 - -0.00137586 - -0.00212829 - -0.0028243 - -0.0034529 - -0.00400205 - -0.00445942 - -0.00481357 - -0.0050554 - -0.00517947 - -0.00518511 - -0.00507705 - -0.00486543 - -0.00456522 - -0.00419512 - -0.00377605 - -0.00332958 - -0.00287625 - -0.00243427 - -0.00201849 - -0.00163986 - -0.00130532 - -0.00101802 - -0.000777914 - -0.000582426 - -0.000427253 - -0.000307089 - -0.000216263 - -0.000149222 - -0.000100884 - -6.68267e-05 - -4.33723e-05 - -2.75812e-05 - -1.7185e-05 - -1.04911e-05 - -6.27527e-06 - -3.67773e-06 - -2.11185e-06 - -1.18819e-06 - -6.55e-07 - -3.53781e-07 - -1.87222e-07 - -9.70701e-08 - -4.92929e-08 - -2.44779e-08 - -1.17909e-08 - -5.27448e-09 - -1.60946e-09 - 1.27482e-09 - 5.18955e-09 - 1.26581e-08 - 2.82967e-08 - 6.12173e-08 - 1.29452e-07 - 2.68086e-07 - 5.43925e-07 - 1.08126e-06 - 2.10601e-06 - 4.01906e-06 - 7.51496e-06 - 1.37678e-05 - 2.47139e-05 - 4.34663e-05 - 7.49034e-05 - 0.00012647 - 0.000209223 - 0.000339133 - 0.0005386 - 0.000838107 - 0.00127782 - 0.00190887 - 0.00279396 - 0.00400683 - 0.00563014 - 0.00775128 - 0.010456 - 0.0138195 - 0.0178961 - 0.022707 - 0.0282292 - 0.0343855 - 0.0410382 - 0.0479888 - 0.0549831 - 0.0617247 - 0.0678938 - 0.0731722 - 0.0772704 - 0.0799544 - 0.0810684 - 0.0805512 - 0.0784438 - 0.0748868 - 0.0701092 - 0.0644092 - 0.0581302 - 0.0516357 - 0.0452844 - 0.0394097 - 0.0343041 - 0.0302089 - 0.0273084 - 0.0257283 - 0.0255355 - 0.0267383 - 0.0292872 - 0.0330737 - 0.0379307 - 0.0436337 - 0.0499051 - 0.0564229 - 0.0628345 - 0.0687764 - 0.073897 - 0.0778824 - 0.0804804 - 0.0815217 - 0.0809336 - 0.0787451 - 0.0750827 - 0.0701565 - 0.0642396 - 0.0576423 - 0.0506847 - 0.0436712 - 0.0368695 - 0.030495 - 0.024703 - 0.0195869 - 0.0151828 - 0.0114776 - 0.00841987 - 0.00593219 - 0.00392215 - 0.00229274 - 0.000950838 - -0.000186034 - -0.00118449 - -0.00209138 - -0.00293083 - -0.00370225 - -0.00437912 - -0.00490892 - -0.00521447 - -0.00519708 - -0.00474199 - -0.00372615 - -0.00202842 - 0.000458113 - 0.00381264 - 0.00807298 - 0.0132233 - 0.0191849 - 0.0258114 - 0.0328904 - 0.0401515 - 0.047281 - 0.0539439 - 0.059808 - 0.0645718 - 0.0679897 - 0.0698937 - 0.0702075 - 0.0689522 - 0.0662425 - 0.062274 - 0.0573033 - 0.0516237 - 0.0455395 - 0.0393419 - 0.0332886 - 0.0275894 - 0.0223987 - 0.017814 - 0.0138795 - 0.0105943 - 0.00792264 - 0.0058046 - 0.00416665 - 0.00293035 - 0.00201918 - 0.00136318 - 0.000901698 - 0.000584386 - 0.000371087 - 0.000230892 - 0.000140786 - 8.41686e-05 - 4.94262e-05 - 2.86893e-05 - 1.68175e-05 - 1.06308e-05 - 8.37418e-06 - 9.39916e-06 - 1.40463e-05 - 2.3726e-05 - 4.1207e-05 - 7.11357e-05 - 0.000120818 - 0.000201294 - 0.000328719 - 0.000526031 - 0.000824827 - 0.00126728 - 0.00190781 - 0.00281418 - 0.00406746 - 0.00576033 - 0.00799326 - 0.0108681 - 0.014479 - 0.0189005 - 0.0241747 - 0.0302971 - 0.0372043 - 0.044765 - 0.0527758 - 0.0609654 - 0.0690054 - 0.0765305 - 0.0831643 - 0.0885505 - 0.0923838 - 0.094439 - 0.0945928 - 0.0928357 - 0.0892735 - 0.0841163 - 0.0776584 - 0.0702501 - 0.0622666 - 0.0540773 - 0.0460179 - 0.0383702 - 0.0313489 - 0.0250975 - 0.0196902 - 0.0151414 - 0.0114171 - 0.00844929 - 0.00614968 - 0.00442187 - 0.00317137 - 0.00231276 - 0.00177424 - 0.00149973 - 0.00144902 - 0.00159637 - 0.00192796 - 0.00243858 - 0.00312782 - 0.00399592 - 0.00503982 - 0.00624939 - 0.00760438 - 0.00907248 - 0.0106086 - 0.0121559 - 0.0136483 - 0.0150148 - 0.0161846 - 0.0170932 - 0.0176882 - 0.0179342 - 0.0178162 - 0.0173415 - 0.0165385 - 0.0154541 - 0.0141492 - 0.012693 - 0.0111571 - 0.00960964 - 0.00811103 - 0.00671036 - 0.00544389 - 0.00433493 - 0.00339517 - 0.00262695 - 0.0020262 - 0.00158568 - 0.00129798 - 0.00115831 - 0.0011665 - 0.00132846 - 0.00165655 - 0.00216916 - 0.00288914 - 0.00384126 - 0.00504868 - 0.00652871 - 0.00828802 - 0.0103181 - 0.012591 - 0.015057 - 0.0176439 - 0.0202583 - 0.0227906 - 0.0251218 - 0.0271321 - 0.0287114 - 0.0297689 - 0.0302419 - 0.0301019 - 0.0293576 - 0.0280537 - 0.0262672 - 0.0240992 - 0.0216666 - 0.0190912 - 0.0164912 - 0.0139727 - 0.0116248 - 0.00951633 - 0.00769657 - 0.00619684 - 0.00503464 - 0.00421823 - 0.00375123 - 0.00363649 - 0.00387863 - 0.00448478 - 0.00546347 - 0.00682166 - 0.00856033 - 0.0106692 - 0.0131211 - 0.0158675 - 0.0188355 - 0.0219272 - 0.0250222 - 0.0279839 - 0.0306674 - 0.0329313 - 0.034649 - 0.0357203 - 0.0360812 - 0.0357095 - 0.0346278 - 0.0329006 - 0.030628 - 0.0279365 - 0.0249667 - 0.0218619 - 0.0187564 - 0.015767 - 0.0129863 - 0.01048 - 0.00828649 - 0.00641975 - 0.00487306 - 0.00362429 - 0.00264108 - 0.00188571 - 0.00131919 - 0.000904223 - 0.000607268 - 0.000399597 - 0.000257633 - 0.000162749 - 0.000100733 - 6.10885e-05 - 3.62982e-05 - 2.11324e-05 - 1.20545e-05 - 6.7373e-06 - 3.68943e-06 - 1.97957e-06 - 1.04068e-06 - 5.36045e-07 - 2.70534e-07 - 1.33777e-07 - 6.4815e-08 - 3.07685e-08 - 1.43112e-08 - 6.522e-09 - 2.91221e-09 - 1.2741e-09 - 5.46157e-10 - 2.29388e-10 - 9.43974e-11 - 3.80615e-11 - 1.50366e-11 - 5.82036e-12 - 2.20743e-12 - 8.20277e-13 - 2.98656e-13 - 1.06541e-13 - 3.72394e-14 - 1.27533e-14 - 4.27939e-15 - 1.40694e-15 - 4.5322e-16 - 1.43046e-16 - 4.42367e-17 - 1.34037e-17 - 3.97927e-18 - 1.1575e-18 - 3.29893e-19 - 9.21218e-20 - 2.52051e-20 - 6.75697e-21 - 1.77481e-21 - 4.56761e-22 - 1.15176e-22 - 2.8456e-23 - 6.88845e-24 - 1.63383e-24 - 3.79688e-25 - 8.64541e-26 - 1.92877e-26 - 4.21611e-27 - 9.02986e-28 - 1.8949e-28 - 3.89609e-29 - 7.84888e-30 - 1.54926e-30 - 2.99624e-31 - 5.67762e-32 - 1.05413e-32 - 1.91759e-33 - 3.41788e-34 - 5.9689e-35 - 1.02133e-35 - 1.7123e-36 - 2.81272e-37 - 4.52701e-38 - 7.13892e-39 - 1.10304e-39 - 1.66988e-40 - 2.47695e-41 - 3.59987e-42 - 5.12616e-43 - 7.15212e-44 - 9.77719e-45 - 1.30957e-45 - 1.71863e-46 - 2.2099e-47 - 2.78419e-48 - 3.43687e-49 - 4.15684e-50 - 4.92607e-51 - 5.71972e-52 - 6.50707e-53 - 7.25326e-54 - 7.92169e-55 - 8.47695e-56 - 8.88787e-57 - 9.13047e-58 - 9.1902e-59 - 9.06346e-60 - 8.75789e-61 - 8.29167e-62 - 7.69169e-63 - 6.99098e-64 - 6.22574e-65 - 5.43226e-66 - 4.64417e-67 - 3.89019e-68 - 3.1928e-69 - 2.56749e-70 - 2.02294e-71 - 1.56169e-72 - 1.18125e-73 - 8.75441e-75 - 6.35695e-76 - 4.5228e-77 - 3.15284e-78 - 2.15345e-79 - 1.44113e-80 - 9.44951e-82 - 6.07088e-83 - 3.82148e-84 - 2.35693e-85 - 1.4243e-86 - 8.43317e-88 - 4.89235e-89 - 2.78087e-90 - 1.54875e-91 - 8.45122e-93 - 4.51849e-94 - 2.36704e-95 - 1.21493e-96 - 6.10995e-98 - 3.01064e-99 - 1.45351e-100 - 6.87564e-102 - 3.18673e-103 - 1.44715e-104 - 6.43903e-106 - 2.80713e-107 - 1.19907e-108 - 5.01875e-110 - 2.16259e-111 - 2.64582e-111 - 6.14557e-110 - 1.44359e-108 - 3.32264e-107 - 7.49309e-106 - 1.65567e-104 - 3.58448e-103 - 7.6035e-102 - 1.5803e-100 - 3.2181e-99 - 6.42094e-98 - 1.25526e-96 - 2.4044e-95 - 4.51248e-94 - 8.29776e-93 - 1.49501e-91 - 2.63914e-90 - 4.56477e-89 - 7.73593e-88 - 1.28452e-86 - 2.08982e-85 - 3.3313e-84 - 5.203e-83 - 7.96217e-82 - 1.19384e-80 - 1.75387e-79 - 2.52455e-78 - 3.56049e-77 - 4.92007e-76 - 6.66146e-75 - 8.837e-74 - 1.14862e-72 - 1.4628e-71 - 1.82529e-70 - 2.23159e-69 - 2.67322e-68 - 3.13755e-67 - 3.60815e-66 - 4.06551e-65 - 4.4883e-64 - 4.85496e-63 - 5.14548e-62 - 5.34323e-61 - 5.43648e-60 - 5.41963e-59 - 5.29368e-58 - 5.0662e-57 - 4.75055e-56 - 4.36458e-55 - 3.92897e-54 - 3.46538e-53 - 2.99475e-52 - 2.53575e-51 - 2.10372e-50 - 1.71005e-49 - 1.36196e-48 - 1.06282e-47 - 8.12623e-47 - 6.08774e-46 - 4.46848e-45 - 3.21367e-44 - 2.26453e-43 - 1.56348e-42 - 1.05766e-41 - 7.01025e-41 - 4.55259e-40 - 2.89682e-39 - 1.80601e-38 - 1.1032e-37 - 6.60281e-37 - 3.87203e-36 - 2.22477e-35 - 1.25247e-34 - 6.90859e-34 - 3.73376e-33 - 1.97716e-32 - 1.02582e-31 - 5.21484e-31 - 2.59744e-30 - 1.26762e-29 - 6.06133e-29 - 2.83978e-28 - 1.30358e-27 - 5.86312e-27 - 2.58379e-26 - 1.11564e-25 - 4.71981e-25 - 1.95643e-24 - 7.94588e-24 - 3.16197e-23 - 1.23285e-22 - 4.70977e-22 - 1.7629e-21 - 6.46539e-21 - 2.32327e-20 - 8.17978e-20 - 2.82178e-19 - 9.53769e-19 - 3.15866e-18 - 1.02495e-17 - 3.25866e-17 - 1.01512e-16 - 3.0984e-16 - 9.26612e-16 - 2.71519e-15 - 7.79551e-15 - 2.19297e-14 - 6.04456e-14 - 1.63246e-13 - 4.31982e-13 - 1.12005e-12 - 2.84549e-12 - 7.0832e-12 - 1.72764e-11 - 4.12889e-11 - 9.66876e-11 - 2.21855e-10 - 4.98805e-10 - 1.09891e-09 - 2.37226e-09 - 5.01812e-09 - 1.04016e-08 - 2.11277e-08 - 4.2053e-08 - 8.20252e-08 - 1.56787e-07 - 2.93696e-07 - 5.39164e-07 - 9.70047e-07 - 1.71052e-06 - 2.95629e-06 - 5.00803e-06 - 8.3159e-06 - 1.35363e-05 - 2.16005e-05 - 3.37937e-05 - 5.18379e-05 - 7.7972e-05 - 0.000115015 - 0.000166394 - 0.000236125 - 0.000328721 - 0.00044901 - 0.000601858 - 0.000791802 - 0.00102259 - 0.00129668 - 0.00161472 - 0.00197509 - 0.00237354 - 0.00280296 - 0.00325346 - 0.00371255 - 0.00416568 - 0.00459692 - 0.00498986 - 0.00532857 - 0.00559865 - 0.00578819 - 0.00588854 - 0.00589499 - 0.00580705 - 0.00562858 - 0.00536746 - 0.00503512 - 0.00464575 - 0.0042153 - 0.00376048 - 0.0032977 - 0.00284209 - 0.00240674 - 0.00200212 - 0.0016358 - - - - - -3.18479e-31 - -3.96339e-31 - -4.8327e-31 - -5.77362e-31 - -6.75833e-31 - -7.75075e-31 - -8.70697e-31 - -9.57062e-31 - -1.02378e-30 - -1.03623e-30 - -8.36013e-31 - 3.5185e-31 - 6.3007e-30 - 3.5048e-29 - 1.71064e-28 - 8.01776e-28 - 3.66744e-27 - 1.64235e-26 - 7.20503e-26 - 3.09692e-25 - 1.30424e-24 - 5.38172e-24 - 2.17581e-23 - 8.61905e-23 - 3.34529e-22 - 1.27217e-21 - 4.74016e-21 - 1.73052e-20 - 6.19012e-20 - 2.16949e-19 - 7.44995e-19 - 2.5066e-18 - 8.26332e-18 - 2.66907e-17 - 8.44702e-17 - 2.61929e-16 - 7.95792e-16 - 2.36893e-15 - 6.90944e-15 - 1.97456e-14 - 5.52884e-14 - 1.51682e-13 - 4.0773e-13 - 1.07386e-12 - 2.77114e-12 - 7.00658e-12 - 1.73577e-11 - 4.21321e-11 - 1.00201e-10 - 2.33489e-10 - 5.33089e-10 - 1.19253e-09 - 2.61381e-09 - 5.61329e-09 - 1.18113e-08 - 2.43508e-08 - 4.91889e-08 - 9.73549e-08 - 1.88793e-07 - 3.58714e-07 - 6.67805e-07 - 1.21811e-06 - 2.17702e-06 - 3.81218e-06 - 6.54066e-06 - 1.09953e-05 - 1.81104e-05 - 2.92272e-05 - 4.6215e-05 - 7.16003e-05 - 0.000108689 - 0.000161656 - 0.000235577 - 0.000336367 - 0.000470576 - 0.000645036 - 0.000866312 - 0.00113999 - 0.00146982 - 0.00185681 - 0.00229829 - 0.00278727 - 0.00331201 - 0.00385604 - 0.00439875 - 0.00491649 - 0.0053842 - 0.00577737 - 0.00607415 - 0.00625743 - 0.00631647 - 0.00624807 - 0.00605694 - 0.00575536 - 0.00536215 - 0.00490107 - 0.0043989 - 0.00388353 - 0.00338212 - 0.00291963 - 0.00251784 - 0.00219476 - 0.0019644 - 0.00183683 - 0.00181827 - 0.00191123 - 0.00211441 - 0.0024225 - 0.0028259 - 0.00331037 - 0.00385693 - 0.00444204 - 0.00503828 - 0.00561553 - 0.0061427 - 0.00658983 - 0.00693041 - 0.0071435 - 0.00721563 - 0.00714193 - 0.00692656 - 0.00658218 - 0.00612867 - 0.00559119 - 0.00499782 - 0.0043772 - 0.0037562 - 0.00315819 - 0.00260175 - 0.00210005 - 0.00166085 - 0.00128697 - 0.000977109 - 0.000726866 - 0.000529789 - 0.000378345 - 0.000264734 - 0.000181497 - 0.000121917 - 8.02412e-05 - 5.17448e-05 - 3.26944e-05 - 2.02403e-05 - 1.22771e-05 - 7.29648e-06 - 4.24881e-06 - 2.42414e-06 - 1.35514e-06 - 7.42249e-07 - 3.98337e-07 - 2.09454e-07 - 1.0791e-07 - 5.44722e-08 - 2.69416e-08 - 1.3056e-08 - 6.19913e-09 - 2.88397e-09 - 1.31458e-09 - 5.87109e-10 - 2.56914e-10 - 1.10153e-10 - 4.62739e-11 - 1.90459e-11 - 7.67917e-12 - 3.02905e-12 - 1.15766e-12 - 3.97361e-13 - 3.41409e-14 - -2.89899e-13 - -8.92159e-13 - -2.32794e-12 - -5.84648e-12 - -1.43519e-11 - -3.45078e-11 - -8.12912e-11 - -1.87631e-10 - -4.24327e-10 - -9.40231e-10 - -2.04129e-09 - -4.34222e-09 - -9.05016e-09 - -1.84815e-08 - -3.6979e-08 - -7.24953e-08 - -1.39252e-07 - -2.62077e-07 - -4.83275e-07 - -8.73166e-07 - -1.54574e-06 - -2.68109e-06 - -4.55642e-06 - -7.58706e-06 - -1.23783e-05 - -1.97871e-05 - -3.09915e-05 - -4.75597e-05 - -7.15111e-05 - -0.000105352 - -0.000152073 - -0.000215078 - -0.000298041 - -0.000404664 - -0.00053833 - -0.000701682 - -0.000896126 - -0.00112133 - -0.00137479 - -0.00165149 - -0.0019438 - -0.00224163 - -0.00253287 - -0.00280414 - -0.00304174 - -0.00323283 - -0.0033665 - -0.00343488 - -0.00343384 - -0.00336344 - -0.0032279 - -0.00303518 - -0.00279621 - -0.00252384 - -0.00223166 - -0.00193291 - -0.00163942 - -0.00136091 - -0.00110451 - -0.000874518 - -0.000672571 - -0.00049787 - -0.000347626 - -0.000217553 - -0.000102387 - 3.62246e-06 - 0.000106247 - 0.000210868 - 0.000322098 - 0.000443445 - 0.000577031 - 0.000723357 - 0.000881165 - 0.0010474 - 0.00121734 - 0.00138482 - 0.0015427 - 0.00168342 - 0.00179962 - 0.00188484 - 0.00193416 - 0.00194463 - 0.00191564 - 0.00184894 - 0.00174849 - 0.00162005 - 0.00147064 - 0.00130789 - 0.00113938 - 0.000972049 - 0.000811701 - 0.000662676 - 0.000527676 - 0.000407728 - 0.000302261 - 0.000209254 - 0.000125426 - 4.64271e-05 - -3.29605e-05 - -0.000118629 - -0.000216928 - -0.0003344 - -0.000477448 - -0.000651899 - -0.000862487 - -0.00111225 - -0.00140188 - -0.00172921 - -0.00208869 - -0.00247127 - -0.00286449 - -0.003253 - -0.00361946 - -0.0039458 - -0.00421463 - -0.00441082 - -0.00452289 - -0.00454411 - -0.00447319 - -0.00431444 - -0.00407725 - -0.00377526 - -0.00342502 - -0.00304451 - -0.0026516 - -0.00226274 - -0.0018919 - -0.00154988 - -0.00124405 - -0.000978389 - -0.000753916 - -0.000569208 - -0.000421072 - -0.000305196 - -0.000216739 - -0.000150811 - -0.000102817 - -6.86807e-05 - -4.49512e-05 - -2.8826e-05 - -1.81119e-05 - -1.11501e-05 - -6.72565e-06 - -3.97489e-06 - -2.30172e-06 - -1.30592e-06 - -7.25967e-07 - -3.95417e-07 - -2.11023e-07 - -1.10342e-07 - -5.65314e-08 - -2.83775e-08 - -1.39572e-08 - -6.72598e-09 - -3.17579e-09 - -1.46921e-09 - -6.65968e-10 - -2.95774e-10 - -1.28707e-10 - -5.48761e-11 - -2.29245e-11 - -9.38325e-12 - -3.76308e-12 - -1.47867e-12 - -5.69293e-13 - -2.14752e-13 - -7.93733e-14 - -2.87441e-14 - -1.01991e-14 - -3.54576e-15 - -1.2078e-15 - -4.03104e-16 - -1.31818e-16 - -4.22349e-17 - -1.32588e-17 - -4.07825e-18 - -1.22908e-18 - -3.62931e-19 - -1.05004e-19 - -2.9766e-20 - -8.26751e-21 - -2.24991e-21 - -5.99919e-22 - -1.56732e-22 - -4.01224e-23 - -1.00739e-23 - -2.52276e-24 - -8.0687e-25 - -1.016e-24 - -3.58361e-24 - -1.41196e-23 - -5.4951e-23 - -2.09636e-22 - -7.83616e-22 - -2.86998e-21 - -1.02989e-20 - -3.62111e-20 - -1.24746e-19 - -4.21067e-19 - -1.39255e-18 - -4.5124e-18 - -1.43265e-17 - -4.45669e-17 - -1.35837e-16 - -4.05661e-16 - -1.18698e-15 - -3.40301e-15 - -9.55911e-15 - -2.63093e-14 - -7.09476e-14 - -1.87458e-13 - -4.85294e-13 - -1.23096e-12 - -3.05929e-12 - -7.4496e-12 - -1.77739e-11 - -4.15498e-11 - -9.51682e-11 - -2.13576e-10 - -4.69623e-10 - -1.01177e-09 - -2.13577e-09 - -4.41734e-09 - -8.9517e-09 - -1.77741e-08 - -3.45784e-08 - -6.59113e-08 - -1.23098e-07 - -2.25258e-07 - -4.03874e-07 - -7.09493e-07 - -1.2212e-06 - -2.05951e-06 - -3.40311e-06 - -5.50968e-06 - -8.74004e-06 - -1.35843e-05 - -2.06871e-05 - -3.08674e-05 - -4.51271e-05 - -6.46419e-05 - -9.07258e-05 - -0.000124764 - -0.00016811 - -0.000221944 - -0.000287112 - -0.000363934 - -0.000452037 - -0.000550211 - -0.000656332 - -0.000767384 - -0.000879594 - -0.000988694 - -0.00109031 - -0.00118047 - -0.00125612 - -0.00131572 - -0.00135978 - -0.00139119 - -0.00141553 - -0.00144105 - -0.00147842 - -0.00154024 - -0.00164022 - -0.00179215 - -0.0020087 - -0.00229996 - -0.00267211 - -0.00312609 - -0.00365653 - -0.00425123 - -0.00489099 - -0.00555038 - -0.00619899 - -0.00680353 - -0.00733032 - -0.00774812 - -0.00803088 - -0.00816018 - -0.00812687 - -0.00793199 - -0.0075865 - -0.00711016 - -0.00652951 - -0.00587539 - -0.00518015 - -0.00447502 - -0.00378787 - -0.0031416 - -0.00255317 - -0.00203343 - -0.00158747 - -0.0012155 - -0.000914005 - -0.000677014 - -0.000497327 - -0.000367638 - -0.00028146 - -0.000233825 - -0.000221778 - -0.000244658 - -0.000304193 - -0.000404393 - -0.000551247 - -0.0007522 - -0.00101539 - -0.00134866 - -0.00175834 - -0.00224794 - -0.00281677 - -0.00345876 - -0.00416153 - -0.00490608 - -0.00566706 - -0.00641387 - -0.00711248 - -0.00772785 - -0.00822684 - -0.00858114 - -0.00876987 - -0.0087817 - -0.0086159 - -0.00828247 - -0.0078011 - -0.00719927 - -0.00650966 - -0.00576719 - -0.0050062 - -0.00425783 - -0.00354817 - -0.00289707 - -0.00231766 - -0.00181667 - -0.00139522 - -0.00104989 - -0.000774071 - -0.000559185 - -0.000395792 - -0.000274483 - -0.00018651 - -0.000124172 - -8.09996e-05 - -5.17701e-05 - -3.24199e-05 - -1.98921e-05 - -1.19588e-05 - -7.04419e-06 - -4.06548e-06 - -2.29895e-06 - -1.27376e-06 - -6.91503e-07 - -3.67873e-07 - -1.91872e-07 - -9.83377e-08 - -5.00377e-08 - -2.64246e-08 - -1.68924e-08 - -1.70683e-08 - -2.72651e-08 - -5.27365e-08 - -1.05874e-07 - -2.1103e-07 - -4.13329e-07 - -7.93705e-07 - -1.49355e-06 - -2.75377e-06 - -4.97481e-06 - -8.80568e-06 - -1.52717e-05 - -2.59505e-05 - -4.32059e-05 - -7.04818e-05 - -0.000112654 - -0.000176423 - -0.000270707 - -0.000406987 - -0.000599513 - -0.000865274 - -0.00122362 - -0.00169541 - -0.00230165 - -0.00306155 - -0.00399008 - -0.00509515 - -0.00637485 - -0.00781483 - -0.00938655 - -0.0110466 - -0.0127377 - -0.0143909 - -0.0159302 - -0.0172779 - -0.0183611 - -0.0191181 - -0.0195041 - -0.0194959 - -0.0190941 - -0.0183227 - -0.0172274 - -0.0158703 - -0.0143247 - -0.0126685 - -0.0109775 - -0.00931998 - -0.00775291 - -0.00631905 - -0.00504632 - -0.00394853 - -0.00302714 - -0.00227387 - -0.00167355 - -0.00120683 - -0.000852687 - -0.000590298 - -0.000400396 - -0.0002661 - -0.000173275 - -0.000110552 - -6.91084e-05 - -4.23286e-05 - -2.54023e-05 - -1.49365e-05 - -8.60522e-06 - -4.85749e-06 - -2.68657e-06 - -1.45587e-06 - -7.73006e-07 - -4.02142e-07 - -2.04981e-07 - -1.02373e-07 - -5.00948e-08 - -2.4018e-08 - -1.12829e-08 - -5.19323e-09 - -2.34203e-09 - -1.03487e-09 - -4.48039e-10 - -1.90063e-10 - -7.90133e-11 - -3.2227e-11 - -1.2995e-11 - -5.44122e-12 - -3.02022e-12 - -3.51191e-12 - -7.21295e-12 - -1.70439e-11 - -4.05427e-11 - -9.4878e-11 - -2.17679e-10 - -4.89379e-10 - -1.07799e-09 - -2.32661e-09 - -4.92003e-09 - -1.01941e-08 - -2.06951e-08 - -4.11646e-08 - -8.02261e-08 - -1.53195e-07 - -2.86623e-07 - -5.25429e-07 - -9.43742e-07 - -1.66085e-06 - -2.8638e-06 - -4.83831e-06 - -8.00904e-06 - -1.29899e-05 - -2.06426e-05 - -3.21413e-05 - -4.9034e-05 - -7.3294e-05 - -0.000107344 - -0.000154036 - -0.000216573 - -0.000298347 - -0.000402696 - -0.000532562 - -0.00069008 - -0.000876123 - -0.00108985 - -0.00132834 - -0.0015863 - -0.00185609 - -0.00212789 - -0.00239021 - -0.00263063 - -0.00283674 - -0.00299722 - -0.00310279 - -0.0031472 - -0.00312775 - -0.00304563 - -0.00290576 - -0.0027163 - -0.0024879 - -0.00223268 - -0.00196316 - -0.0016913 - -0.00142766 - -0.00118077 - -0.000956844 - -0.000759722 - -0.000591024 - -0.000450497 - -0.000336447 - -0.000246194 - -0.000176512 - -0.000123997 - -8.53456e-05 - -5.75558e-05 - -3.80307e-05 - -2.46216e-05 - -1.56184e-05 - -9.70714e-06 - -5.91131e-06 - -3.52706e-06 - -2.06196e-06 - -1.18109e-06 - -6.62861e-07 - -3.64501e-07 - -1.96387e-07 - -1.03672e-07 - -5.36229e-08 - -2.71753e-08 - -1.34939e-08 - -6.56499e-09 - -3.12945e-09 - -1.46164e-09 - -6.6888e-10 - -2.99912e-10 - -1.31758e-10 - -5.67146e-11 - -2.39194e-11 - -9.88424e-12 - -4.00196e-12 - -1.58759e-12 - -6.17081e-13 - -2.35008e-13 - -8.76919e-14 - -3.20607e-14 - -1.14848e-14 - -4.03098e-15 - -1.38623e-15 - -4.67086e-16 - -1.54204e-16 - -4.98803e-17 - -1.58089e-17 - -4.90919e-18 - -1.49367e-18 - -4.45284e-19 - -1.30064e-19 - -3.72232e-20 - -1.04383e-20 - -2.87015e-21 - -7.81928e-22 - -2.43429e-22 - -2.08622e-22 - -6.12204e-22 - -2.2559e-21 - -8.30244e-21 - -2.99767e-20 - -1.06056e-19 - -3.67643e-19 - -1.24869e-18 - -4.15545e-18 - -1.35494e-17 - -4.32871e-17 - -1.35498e-16 - -4.15571e-16 - -1.2488e-15 - -3.67688e-15 - -1.06072e-14 - -2.99821e-14 - -8.30344e-14 - -2.25315e-13 - -5.99047e-13 - -1.56051e-12 - -3.98301e-12 - -9.96075e-12 - -2.44067e-11 - -5.85955e-11 - -1.37834e-10 - -3.17675e-10 - -7.1738e-10 - -1.58727e-09 - -3.44105e-09 - -7.30916e-09 - -1.52118e-08 - -3.10193e-08 - -6.19757e-08 - -1.21324e-07 - -2.32709e-07 - -4.37337e-07 - -8.05303e-07 - -1.45292e-06 - -2.56841e-06 - -4.44867e-06 - -7.54986e-06 - -1.25543e-05 - -2.04549e-05 - -3.26554e-05 - -5.10821e-05 - -7.82973e-05 - -0.000117598 - -0.000173075 - -0.000249615 - -0.000352798 - -0.000488684 - -0.000663453 - -0.000882913 - -0.00115189 - -0.00147356 - -0.00184879 - -0.00227563 - -0.00274905 - -0.00326099 - -0.00380083 - -0.00435636 - -0.00491499 - -0.00546527 - -0.00599832 - -0.00650907 - -0.00699689 - -0.00746558 - -0.00792243 - -0.00837658 - -0.00883672 - -0.00930857 - -0.00979246 - -0.0102816 - -0.0107611 - -0.0112087 - -0.011596 - -0.0118916 - -0.0120645 - -0.0120879 - -0.0119427 - -0.0116203 - -0.011124 - -0.0104692 - -0.00968215 - -0.00879775 - -0.00785624 - -0.00689989 - -0.00596968 - -0.00510249 - -0.00432903 - -0.00367258 - -0.00314848 - -0.00276429 - -0.00252032 - -0.00241049 - -0.00242322 - -0.00254229 - -0.00274769 - -0.00301656 - -0.00332402 - -0.00364428 - -0.00395183 - -0.00422275 - -0.00443609 - -0.00457514 - -0.00462847 - -0.0045906 - -0.00446213 - -0.00424935 - -0.00396339 - -0.0036188 - -0.00323204 - -0.00281971 - -0.00239708 - -0.0019767 - -0.00156754 - -0.00117457 - -0.000798753 - -0.000437587 - -8.59164e-05 - 0.000262944 - 0.000615954 - 0.000979073 - 0.00135609 - 0.00174768 - 0.00215078 - 0.00255837 - 0.00295977 - 0.00334128 - 0.00368735 - 0.00398195 - 0.00421019 - 0.00435982 - 0.00442255 - 0.00439499 - 0.00427906 - 0.00408186 - 0.003815 - 0.00349351 - 0.00313447 - 0.00275551 - 0.00237344 - 0.00200306 - 0.00165638 - 0.00134211 - 0.00106565 - 0.000829314 - 0.000632838 - 0.00047399 - 0.000349261 - 0.00025451 - 0.000185531 - 0.00013851 - 0.000110364 - 9.89575e-05 - 0.000103216 - 0.000123152 - 0.000159795 - 0.000215048 - 0.000291453 - 0.000391873 - 0.000519088 - 0.000675318 - 0.000861708 - 0.00107781 - 0.00132113 - 0.00158679 - 0.00186744 - 0.00215336 - 0.00243292 - 0.00269324 - 0.0029212 - 0.00310445 - 0.00323254 - 0.00329792 - 0.00329666 - 0.00322882 - 0.0030985 - 0.00291337 - 0.00268396 - 0.00242267 - 0.00214264 - 0.00185669 - 0.00157641 - 0.0013114 - 0.0010689 - 0.00085364 - 0.00066796 - 0.00051211 - 0.000384691 - 0.000283138 - 0.000204184 - 0.000144272 - 9.98799e-05 - 6.77504e-05 - 4.5028e-05 - 2.93217e-05 - 1.87083e-05 - 1.16954e-05 - 7.16366e-06 - 4.29928e-06 - 2.52821e-06 - 1.45697e-06 - 8.23301e-07 - 4.57272e-07 - 2.52047e-07 - 1.4307e-07 - 9.41611e-08 - 8.92735e-08 - 1.29198e-07 - 2.33164e-07 - 4.46224e-07 - 8.54461e-07 - 1.61153e-06 - 2.98178e-06 - 5.40736e-06 - 9.60873e-06 - 1.67298e-05 - 2.854e-05 - 4.77041e-05 - 7.81257e-05 - 0.000125363 - 0.000197097 - 0.000303619 - 0.000458263 - 0.0006777 - 0.000981967 - 0.0013941 - 0.00193921 - 0.00264299 - 0.00352941 - 0.00461791 - 0.00592006 - 0.00743607 - 0.0091516 - 0.0110354 - 0.0130381 - 0.0150932 - 0.0171191 - 0.0190248 - 0.0207155 - 0.0221008 - 0.0231024 - 0.0236615 - 0.0237446 - 0.0233467 - 0.0224919 - 0.0212307 - 0.0196357 - 0.0177941 - 0.0158004 - 0.0137482 - 0.0117235 - 0.00979954 - 0.00803325 - 0.00646427 - 0.00511562 - 0.00399594 - 0.00310269 - 0.00242575 - 0.00195092 - 0.00166287 - 0.00154732 - 0.00159223 - 0.00178801 - 0.00212689 - 0.00260153 - 0.00320325 - 0.00392002 - 0.00473471 - 0.00562383 - 0.006557 - 0.00749741 - 0.00840331 - 0.00923048 - 0.00993535 - 0.0104786 - 0.0108286 - 0.0109643 - 0.0108776 - 0.0105736 - 0.0100705 - 0.00939751 - 0.00859238 - 0.00769753 - 0.00675657 - 0.00581083 - 0.00489651 - 0.0040427 - 0.00327035 - 0.00259211 - 0.00201302 - 0.00153173 - 0.00114196 - 0.000834175 - 0.000597036 - 0.000418679 - 0.000287673 - 0.000193667 - 0.000127749 - 8.25696e-05 - 5.23006e-05 - 3.24804e-05 - 1.98088e-05 - 1.19273e-05 - 7.21506e-06 - 4.61863e-06 - 3.52388e-06 - 3.67649e-06 - 5.15343e-06 - 8.3891e-06 - 1.42612e-05 - 2.42428e-05 - 4.06281e-05 - 6.68359e-05 - 0.000107789 - 0.000170354 - 0.000263818 - 0.000400329 - 0.00059524 - 0.000867232 - 0.0012381 - 0.00173205 - 0.0023745 - 0.00319011 - 0.00420035 - 0.0054206 - 0.00685696 - 0.00850339 - 0.0103395 - 0.0123294 - 0.0144223 - 0.0165546 - 0.0186545 - 0.0206468 - 0.0224604 - 0.0240341 - 0.0253228 - 0.026301 - 0.0269648 - 0.0273304 - 0.0274301 - 0.0273064 - 0.0270053 - 0.0265691 - 0.0260305 - 0.0254095 - 0.0247121 - 0.0239319 - 0.023054 - 0.0220604 - 0.0209356 - 0.0196714 - 0.0182707 - 0.0167493 - 0.0151347 - 0.0134645 - 0.0117821 - 0.0101327 - 0.00855887 - 0.00709693 - 0.0057744 - 0.00460875 - 0.00360736 - 0.00276847 - 0.00208292 - 0.00153623 - 0.00111069 - 0.000787407 - 0.000547808 - 0.000374892 - 0.000253994 - 0.000173243 - 0.000123774 - 9.98233e-05 - 9.87971e-05 - 0.000121408 - 0.000171927 - 0.000258572 - 0.000393997 - 0.000595817 - 0.00088703 - 0.00129616 - 0.00185693 - 0.00260716 - 0.00358686 - 0.00483517 - 0.00638631 - 0.00826471 - 0.0104795 - 0.0130195 - 0.0158483 - 0.018902 - 0.0220888 - 0.0252913 - 0.0283731 - 0.0311875 - 0.0335885 - 0.0354435 - 0.0366455 - 0.0371228 - 0.0368467 - 0.0358338 - 0.0341447 - 0.031878 - 0.0291606 - 0.0261359 - 0.0229518 - 0.0197484 - 0.0166488 - 0.0137522 - 0.0111301 - 0.00882594 - 0.00685741 - 0.00522031 - 0.00389377 - 0.00284564 - 0.00203764 - 0.00142959 - 0.000982723 - 0.000661894 - 0.0004368 - 0.000282432 - 0.000178929 - 0.000111067 - 6.75505e-05 - 4.02538e-05 - 2.35029e-05 - 1.34454e-05 - 7.53639e-06 - 4.13894e-06 - 2.22716e-06 - 1.17423e-06 - 6.0658e-07 - 3.07016e-07 - 1.52255e-07 - 7.3981e-08 - 3.52217e-08 - 1.64312e-08 - 7.51334e-09 - 3.37362e-09 - 1.50265e-09 - 7.0035e-10 - 4.24077e-10 - 4.75335e-10 - 8.76813e-10 - 1.8765e-09 - 4.0748e-09 - 8.72522e-09 - 1.83265e-08 - 3.77229e-08 - 7.60823e-08 - 1.5035e-07 - 2.91111e-07 - 5.5227e-07 - 1.02655e-06 - 1.86959e-06 - 3.33619e-06 - 5.83299e-06 - 9.99237e-06 - 1.67719e-05 - 2.75825e-05 - 4.44448e-05 - 7.0169e-05 - 0.000108544 - 0.000164515 - 0.000244309 - 0.000355477 - 0.000506781 - 0.00070789 - 0.000968832 - 0.00129918 - 0.00170696 - 0.00219744 - 0.0027717 - 0.00342542 - 0.00414779 - 0.00492104 - 0.00572051 - 0.00651551 - 0.00727109 - 0.00795037 - 0.00851749 - 0.00894073 - 0.00919542 - 0.00926631 - 0.0091491 - 0.0088509 - 0.00838944 - 0.0077914 - 0.00708982 - 0.00632108 - 0.00552185 - 0.00472622 - 0.00396352 - 0.00325675 - 0.00262195 - 0.00206825 - 0.00159851 - 0.00121051 - 0.000898163 - 0.00065295 - 0.000465095 - 0.000324594 - 0.00022196 - 0.000148712 - 9.76239e-05 - 6.27917e-05 - 3.95717e-05 - 2.44345e-05 - 1.47829e-05 - 8.76302e-06 - 5.0896e-06 - 2.89635e-06 - 1.61493e-06 - 8.82256e-07 - 4.7225e-07 - 2.47678e-07 - 1.27274e-07 - 6.40809e-08 - 3.1613e-08 - 1.52825e-08 - 7.24349e-09 - 3.37597e-09 - 1.57127e-09 - 7.87334e-10 - 5.49235e-10 - 7.08757e-10 - 1.36716e-09 - 2.92932e-09 - 6.31231e-09 - 1.33936e-08 - 2.78707e-08 - 5.68338e-08 - 1.13558e-07 - 2.22314e-07 - 4.26437e-07 - 8.01454e-07 - 1.47584e-06 - 2.66279e-06 - 4.7073e-06 - 8.15349e-06 - 1.38373e-05 - 2.3009e-05 - 3.74869e-05 - 5.98409e-05 - 9.35953e-05 - 0.000143432 - 0.000215365 - 0.000316842 - 0.000456715 - 0.000645038 - 0.000892611 - 0.00121025 - 0.00160778 - 0.00209273 - 0.00266894 - 0.00333503 - 0.00408317 - 0.00489815 - 0.0057571 - 0.00662998 - 0.00748096 - 0.00827066 - 0.008959 - 0.00950859 - 0.00988805 - 0.010075 - 0.0100581 - 0.00983853 - 0.00942953 - 0.00885532 - 0.00814865 - 0.00734787 - 0.0064935 - 0.00562505 - 0.0047782 - 0.00398281 - 0.00326164 - 0.00263011 - 0.00209661 - 0.00166348 - 0.00132822 - 0.00108488 - 0.000925197 - 0.000839661 - 0.000818171 - 0.000850452 - 0.000926225 - 0.00103522 - 0.00116715 - 0.00131169 - 0.00145858 - 0.00159787 - 0.00172029 - 0.00181768 - 0.00188348 - 0.00191318 - 0.00190459 - 0.00185802 - 0.0017761 - 0.00166357 - 0.00152673 - 0.00137286 - 0.00120957 - 0.00104417 - 0.000883183 - 0.000731926 - 0.000594321 - 0.000472837 - 0.000368587 - 0.000281517 - 0.000210672 - 0.000154471 - 0.000110977 - 7.81216e-05 - 5.38892e-05 - 3.64365e-05 - 2.41667e-05 - 1.57607e-05 - 1.0179e-05 - 6.64527e-06 - 4.62443e-06 - 3.80367e-06 - 4.08572e-06 - 5.6009e-06 - 8.74399e-06 - 1.42407e-05 - 2.32471e-05 - 3.74834e-05 - 5.93981e-05 - 9.23544e-05 - 0.000140822 - 0.000210543 - 0.000308644 - 0.000443633 - 0.000625244 - 0.000864061 - 0.0011709 - 0.00155593 - 0.0020275 - 0.00259089 - 0.00324684 - 0.00399034 - 0.00480954 - 0.00568528 - 0.00659119 - 0.00749458 - 0.00835814 - 0.00914232 - 0.00980831 - 0.0103211 - 0.0106527 - 0.0107844 - 0.0107086 - 0.0104298 - 0.00996373 - 0.00933625 - 0.00858073 - 0.00773524 - 0.00683939 - 0.0059313 - 0.00504505 - 0.00420879 - 0.00344364 - 0.00276337 - 0.00217477 - 0.00167855 - 0.00127055 - 0.000943173 - 0.000686672 - 0.00049038 - 0.000343661 - 0.000236618 - 0.00016055 - 0.000108187 - 7.37676e-05 - 5.3005e-05 - 4.29873e-05 - 4.20568e-05 - 4.9689e-05 - 6.63831e-05 - 9.35609e-05 - 0.000133462 - 0.000189012 - 0.000263645 - 0.00036105 - 0.000484834 - 0.000638098 - 0.000822944 - 0.00103994 - 0.00128764 - 0.00156213 - 0.00185686 - 0.00216261 - 0.00246782 - 0.00275923 - 0.00302271 - 0.00324447 - 0.00341214 - 0.00351599 - 0.00354981 - 0.00351155 - 0.00340353 - 0.0032322 - 0.00300748 - 0.00274185 - 0.00244919 - 0.00214357 - 0.00183819 - 0.00154447 - 0.00127146 - 0.00102557 - 0.000810524 - 0.000627627 - 0.000476183 - 0.000353982 - 0.000257822 - 0.000183984 - 0.000128626 - 8.80794e-05 - 5.90389e-05 - 3.86628e-05 - 2.45949e-05 - 1.49291e-05 - 8.13403e-06 - 2.94857e-06 - -1.74116e-06 - -7.04209e-06 - -1.42113e-05 - -2.48328e-05 - -4.10146e-05 - -6.56039e-05 - -0.000102411 - -0.000156423 - -0.000233969 - -0.000342807 - -0.000492057 - -0.000691949 - -0.000953302 - -0.00128673 - -0.00170157 - -0.00220453 - -0.00279826 - -0.00347991 - -0.00423992 - -0.00506123 - -0.00591925 - -0.0067825 - -0.00761422 - -0.00837483 - -0.00902489 - -0.00952848 - -0.00985649 - -0.00998939 - -0.00991914 - -0.00965003 - -0.00919821 - -0.00859011 - -0.00785987 - -0.00704619 - -0.00618894 - -0.005326 - -0.00449065 - -0.00370973 - -0.00300262 - -0.00238113 - -0.00185009 - -0.0014084 - -0.00105048 - -0.000767675 - -0.000549659 - -0.0003856 - -0.000265039 - -0.000178489 - -0.000117771 - -7.61375e-05 - -4.82267e-05 - -2.99299e-05 - -1.81993e-05 - -1.08426e-05 - -6.32909e-06 - -3.61977e-06 - -2.0284e-06 - -1.11367e-06 - -5.99086e-07 - -3.15758e-07 - -1.63062e-07 - -8.25054e-08 - -4.0902e-08 - -1.98673e-08 - -9.45511e-09 - -4.40887e-09 - -2.01428e-09 - -9.01668e-10 - -3.95464e-10 - -1.69942e-10 - -7.15528e-11 - -2.9518e-11 - -1.19311e-11 - -4.72509e-12 - -1.83346e-12 - -6.97057e-13 - -2.59655e-13 - -9.47633e-14 - -3.3872e-14 - -1.18181e-14 - -3.90035e-15 - -8.29272e-16 - 1.18263e-15 - 4.68453e-15 - 1.39394e-14 - 3.96556e-14 - 1.10235e-13 - 3.00124e-13 - 8.00503e-13 - 2.0918e-12 - 5.35517e-12 - 1.34315e-11 - 3.30047e-11 - 7.94561e-11 - 1.87405e-10 - 4.33047e-10 - 9.80379e-10 - 2.17449e-09 - 4.72525e-09 - 1.006e-08 - 2.09837e-08 - 4.28816e-08 - 8.58558e-08 - 1.68414e-07 - 3.23665e-07 - 6.09432e-07 - 1.12426e-06 - 2.03198e-06 - 3.59822e-06 - 6.24264e-06 - 1.06112e-05 - 1.76715e-05 - 2.88336e-05 - 4.60937e-05 - 7.21939e-05 - 0.000110784 - 0.00016656 - 0.000245348 - 0.00035409 - 0.000500684 - 0.000693638 - 0.000941505 - 0.00125208 - 0.0016314 - 0.00208262 - 0.00260483 - 0.00319202 - 0.00383237 - 0.00450799 - 0.00519525 - 0.00586582 - 0.00648837 - 0.00703083 - 0.0074629 - 0.00775862 - 0.00789868 - 0.00787201 - 0.00767662 - 0.00731945 - 0.00681534 - 0.00618516 - 0.0054536 - 0.00464667 - 0.00378961 - 0.00290508 - 0.00201228 - 0.00112666 - 0.000260383 - -0.000576704 - -0.00137586 - -0.00212829 - -0.0028243 - -0.0034529 - -0.00400205 - -0.00445942 - -0.00481357 - -0.0050554 - -0.00517947 - -0.00518511 - -0.00507705 - -0.00486543 - -0.00456522 - -0.00419512 - -0.00377605 - -0.00332958 - -0.00287625 - -0.00243427 - -0.00201849 - -0.00163986 - -0.00130532 - -0.00101802 - -0.000777914 - -0.000582426 - -0.000427253 - -0.000307089 - -0.000216263 - -0.000149222 - -0.000100884 - -6.68267e-05 - -4.33723e-05 - -2.75812e-05 - -1.7185e-05 - -1.04911e-05 - -6.27527e-06 - -3.67773e-06 - -2.11185e-06 - -1.18819e-06 - -6.55e-07 - -3.53781e-07 - -1.87222e-07 - -9.70701e-08 - -4.92929e-08 - -2.44779e-08 - -1.17909e-08 - -5.27448e-09 - -1.60946e-09 - 1.27482e-09 - 5.18955e-09 - 1.26581e-08 - 2.82967e-08 - 6.12173e-08 - 1.29452e-07 - 2.68086e-07 - 5.43925e-07 - 1.08126e-06 - 2.10601e-06 - 4.01906e-06 - 7.51496e-06 - 1.37678e-05 - 2.47139e-05 - 4.34663e-05 - 7.49034e-05 - 0.00012647 - 0.000209223 - 0.000339133 - 0.0005386 - 0.000838107 - 0.00127782 - 0.00190887 - 0.00279396 - 0.00400683 - 0.00563014 - 0.00775128 - 0.010456 - 0.0138195 - 0.0178961 - 0.022707 - 0.0282292 - 0.0343855 - 0.0410382 - 0.0479888 - 0.0549831 - 0.0617247 - 0.0678938 - 0.0731722 - 0.0772704 - 0.0799544 - 0.0810684 - 0.0805512 - 0.0784438 - 0.0748868 - 0.0701092 - 0.0644092 - 0.0581302 - 0.0516357 - 0.0452844 - 0.0394097 - 0.0343041 - 0.0302089 - 0.0273084 - 0.0257283 - 0.0255355 - 0.0267383 - 0.0292872 - 0.0330737 - 0.0379307 - 0.0436337 - 0.0499051 - 0.0564229 - 0.0628345 - 0.0687764 - 0.073897 - 0.0778824 - 0.0804804 - 0.0815217 - 0.0809336 - 0.0787451 - 0.0750827 - 0.0701565 - 0.0642396 - 0.0576423 - 0.0506847 - 0.0436712 - 0.0368695 - 0.030495 - 0.024703 - 0.0195869 - 0.0151828 - 0.0114776 - 0.00841987 - 0.00593219 - 0.00392215 - 0.00229274 - 0.000950838 - -0.000186034 - -0.00118449 - -0.00209138 - -0.00293083 - -0.00370225 - -0.00437912 - -0.00490892 - -0.00521447 - -0.00519708 - -0.00474199 - -0.00372615 - -0.00202842 - 0.000458113 - 0.00381264 - 0.00807298 - 0.0132233 - 0.0191849 - 0.0258114 - 0.0328904 - 0.0401515 - 0.047281 - 0.0539439 - 0.059808 - 0.0645718 - 0.0679897 - 0.0698937 - 0.0702075 - 0.0689522 - 0.0662425 - 0.062274 - 0.0573033 - 0.0516237 - 0.0455395 - 0.0393419 - 0.0332886 - 0.0275894 - 0.0223987 - 0.017814 - 0.0138795 - 0.0105943 - 0.00792264 - 0.0058046 - 0.00416665 - 0.00293035 - 0.00201918 - 0.00136318 - 0.000901698 - 0.000584386 - 0.000371087 - 0.000230892 - 0.000140786 - 8.41686e-05 - 4.94262e-05 - 2.86893e-05 - 1.68175e-05 - 1.06308e-05 - 8.37418e-06 - 9.39916e-06 - 1.40463e-05 - 2.3726e-05 - 4.1207e-05 - 7.11357e-05 - 0.000120818 - 0.000201294 - 0.000328719 - 0.000526031 - 0.000824827 - 0.00126728 - 0.00190781 - 0.00281418 - 0.00406746 - 0.00576033 - 0.00799326 - 0.0108681 - 0.014479 - 0.0189005 - 0.0241747 - 0.0302971 - 0.0372043 - 0.044765 - 0.0527758 - 0.0609654 - 0.0690054 - 0.0765305 - 0.0831643 - 0.0885505 - 0.0923838 - 0.094439 - 0.0945928 - 0.0928357 - 0.0892735 - 0.0841163 - 0.0776584 - 0.0702501 - 0.0622666 - 0.0540773 - 0.0460179 - 0.0383702 - 0.0313489 - 0.0250975 - 0.0196902 - 0.0151414 - 0.0114171 - 0.00844929 - 0.00614968 - 0.00442187 - 0.00317137 - 0.00231276 - 0.00177424 - 0.00149973 - 0.00144902 - 0.00159637 - 0.00192796 - 0.00243858 - 0.00312782 - 0.00399592 - 0.00503982 - 0.00624939 - 0.00760438 - 0.00907248 - 0.0106086 - 0.0121559 - 0.0136483 - 0.0150148 - 0.0161846 - 0.0170932 - 0.0176882 - 0.0179342 - 0.0178162 - 0.0173415 - 0.0165385 - 0.0154541 - 0.0141492 - 0.012693 - 0.0111571 - 0.00960964 - 0.00811103 - 0.00671036 - 0.00544389 - 0.00433493 - 0.00339517 - 0.00262695 - 0.0020262 - 0.00158568 - 0.00129798 - 0.00115831 - 0.0011665 - 0.00132846 - 0.00165655 - 0.00216916 - 0.00288914 - 0.00384126 - 0.00504868 - 0.00652871 - 0.00828802 - 0.0103181 - 0.012591 - 0.015057 - 0.0176439 - 0.0202583 - 0.0227906 - 0.0251218 - 0.0271321 - 0.0287114 - 0.0297689 - 0.0302419 - 0.0301019 - 0.0293576 - 0.0280537 - 0.0262672 - 0.0240992 - 0.0216666 - 0.0190912 - 0.0164912 - 0.0139727 - 0.0116248 - 0.00951633 - 0.00769657 - 0.00619684 - 0.00503464 - 0.00421823 - 0.00375123 - 0.00363649 - 0.00387863 - 0.00448478 - 0.00546347 - 0.00682166 - 0.00856033 - 0.0106692 - 0.0131211 - 0.0158675 - 0.0188355 - 0.0219272 - 0.0250222 - 0.0279839 - 0.0306674 - 0.0329313 - 0.034649 - 0.0357203 - 0.0360812 - 0.0357095 - 0.0346278 - 0.0329006 - 0.030628 - 0.0279365 - 0.0249667 - 0.0218619 - 0.0187564 - 0.015767 - 0.0129863 - 0.01048 - 0.00828649 - 0.00641975 - 0.00487306 - 0.00362429 - 0.00264108 - 0.00188571 - 0.00131919 - 0.000904223 - 0.000607268 - 0.000399597 - 0.000257633 - 0.000162749 - 0.000100733 - 6.10885e-05 - 3.62982e-05 - 2.11324e-05 - 1.20545e-05 - 6.7373e-06 - 3.68943e-06 - 1.97957e-06 - 1.04068e-06 - 5.36045e-07 - 2.70534e-07 - 1.33777e-07 - 6.4815e-08 - 3.07685e-08 - 1.43112e-08 - 6.522e-09 - 2.91221e-09 - 1.2741e-09 - 5.46157e-10 - 2.29388e-10 - 9.43974e-11 - 3.80615e-11 - 1.50366e-11 - 5.82036e-12 - 2.20743e-12 - 8.20277e-13 - 2.98656e-13 - 1.06541e-13 - 3.72394e-14 - 1.27533e-14 - 4.27939e-15 - 1.40694e-15 - 4.5322e-16 - 1.43046e-16 - 4.42367e-17 - 1.34037e-17 - 3.97927e-18 - 1.1575e-18 - 3.29893e-19 - 9.21218e-20 - 2.52051e-20 - 6.75697e-21 - 1.77481e-21 - 4.56761e-22 - 1.15176e-22 - 2.8456e-23 - 6.88845e-24 - 1.63383e-24 - 3.79688e-25 - 8.64541e-26 - 1.92877e-26 - 4.21611e-27 - 9.02986e-28 - 1.8949e-28 - 3.89609e-29 - 7.84888e-30 - 1.54926e-30 - 2.99624e-31 - 5.67762e-32 - 1.05413e-32 - 1.91759e-33 - 3.41788e-34 - 5.9689e-35 - 1.02133e-35 - 1.7123e-36 - 2.81272e-37 - 4.52701e-38 - 7.13892e-39 - 1.10304e-39 - 1.66988e-40 - 2.47695e-41 - 3.59987e-42 - 5.12616e-43 - 7.15212e-44 - 9.77719e-45 - 1.30957e-45 - 1.71863e-46 - 2.2099e-47 - 2.78419e-48 - 3.43687e-49 - 4.15684e-50 - 4.92607e-51 - 5.71972e-52 - 6.50707e-53 - 7.25326e-54 - 7.92169e-55 - 8.47695e-56 - 8.88787e-57 - 9.13047e-58 - 9.1902e-59 - 9.06346e-60 - 8.75789e-61 - 8.29167e-62 - 7.69169e-63 - 6.99098e-64 - 6.22574e-65 - 5.43226e-66 - 4.64417e-67 - 3.89019e-68 - 3.1928e-69 - 2.56749e-70 - 2.02294e-71 - 1.56169e-72 - 1.18125e-73 - 8.75441e-75 - 6.35695e-76 - 4.5228e-77 - 3.15284e-78 - 2.15345e-79 - 1.44113e-80 - 9.44951e-82 - 6.07088e-83 - 3.82148e-84 - 2.35693e-85 - 1.4243e-86 - 8.43317e-88 - 4.89235e-89 - 2.78087e-90 - 1.54875e-91 - 8.45122e-93 - 4.51849e-94 - 2.36704e-95 - 1.21493e-96 - 6.10995e-98 - 3.01064e-99 - 1.45351e-100 - 6.87564e-102 - 3.18673e-103 - 1.44715e-104 - 6.43903e-106 - 2.80713e-107 - 1.19907e-108 - 5.01875e-110 - 2.16259e-111 - 2.64582e-111 - 6.14557e-110 - 1.44359e-108 - 3.32264e-107 - 7.49309e-106 - 1.65567e-104 - 3.58448e-103 - 7.6035e-102 - 1.5803e-100 - 3.2181e-99 - 6.42094e-98 - 1.25526e-96 - 2.4044e-95 - 4.51248e-94 - 8.29776e-93 - 1.49501e-91 - 2.63914e-90 - 4.56477e-89 - 7.73593e-88 - 1.28452e-86 - 2.08982e-85 - 3.3313e-84 - 5.203e-83 - 7.96217e-82 - 1.19384e-80 - 1.75387e-79 - 2.52455e-78 - 3.56049e-77 - 4.92007e-76 - 6.66146e-75 - 8.837e-74 - 1.14862e-72 - 1.4628e-71 - 1.82529e-70 - 2.23159e-69 - 2.67322e-68 - 3.13755e-67 - 3.60815e-66 - 4.06551e-65 - 4.4883e-64 - 4.85496e-63 - 5.14548e-62 - 5.34323e-61 - 5.43648e-60 - 5.41963e-59 - 5.29368e-58 - 5.0662e-57 - 4.75055e-56 - 4.36458e-55 - 3.92897e-54 - 3.46538e-53 - 2.99475e-52 - 2.53575e-51 - 2.10372e-50 - 1.71005e-49 - 1.36196e-48 - 1.06282e-47 - 8.12623e-47 - 6.08774e-46 - 4.46848e-45 - 3.21367e-44 - 2.26453e-43 - 1.56348e-42 - 1.05766e-41 - 7.01025e-41 - 4.55259e-40 - 2.89682e-39 - 1.80601e-38 - 1.1032e-37 - 6.60281e-37 - 3.87203e-36 - 2.22477e-35 - 1.25247e-34 - 6.90859e-34 - 3.73376e-33 - 1.97716e-32 - 1.02582e-31 - 5.21484e-31 - 2.59744e-30 - 1.26762e-29 - 6.06133e-29 - 2.83978e-28 - 1.30358e-27 - 5.86312e-27 - 2.58379e-26 - 1.11564e-25 - 4.71981e-25 - 1.95643e-24 - 7.94588e-24 - 3.16197e-23 - 1.23285e-22 - 4.70977e-22 - 1.7629e-21 - 6.46539e-21 - 2.32327e-20 - 8.17978e-20 - 2.82178e-19 - 9.53769e-19 - 3.15866e-18 - 1.02495e-17 - 3.25866e-17 - 1.01512e-16 - 3.0984e-16 - 9.26612e-16 - 2.71519e-15 - 7.79551e-15 - 2.19297e-14 - 6.04456e-14 - 1.63246e-13 - 4.31982e-13 - 1.12005e-12 - 2.84549e-12 - 7.0832e-12 - 1.72764e-11 - 4.12889e-11 - 9.66876e-11 - 2.21855e-10 - 4.98805e-10 - 1.09891e-09 - 2.37226e-09 - 5.01812e-09 - 1.04016e-08 - 2.11277e-08 - 4.2053e-08 - 8.20252e-08 - 1.56787e-07 - 2.93696e-07 - 5.39164e-07 - 9.70047e-07 - 1.71052e-06 - 2.95629e-06 - 5.00803e-06 - 8.3159e-06 - 1.35363e-05 - 2.16005e-05 - 3.37937e-05 - 5.18379e-05 - 7.7972e-05 - 0.000115015 - 0.000166394 - 0.000236125 - 0.000328721 - 0.00044901 - 0.000601858 - 0.000791802 - 0.00102259 - 0.00129668 - 0.00161472 - 0.00197509 - 0.00237354 - 0.00280296 - 0.00325346 - 0.00371255 - 0.00416568 - 0.00459692 - 0.00498986 - 0.00532857 - 0.00559865 - 0.00578819 - 0.00588854 - 0.00589499 - 0.00580705 - 0.00562858 - 0.00536746 - 0.00503512 - 0.00464575 - 0.0042153 - 0.00376048 - 0.0032977 - 0.00284209 - 0.00240674 - 0.00200212 - 0.0016358 - - - - - 1.59192e-30 - 1.9811e-30 - 2.41562e-30 - 2.88595e-30 - 3.37822e-30 - 3.8746e-30 - 4.35439e-30 - 4.79602e-30 - 5.18269e-30 - 5.5241e-30 - 5.95827e-30 - 7.24628e-30 - 1.31554e-29 - 4.17255e-29 - 1.77438e-28 - 8.07737e-28 - 3.6729e-27 - 1.64284e-26 - 7.20546e-26 - 3.09695e-25 - 1.30424e-24 - 5.38172e-24 - 2.17581e-23 - 8.61905e-23 - 3.34529e-22 - 1.27217e-21 - 4.74016e-21 - 1.73052e-20 - 6.19012e-20 - 2.16949e-19 - 7.44995e-19 - 2.5066e-18 - 8.26332e-18 - 2.66907e-17 - 8.44702e-17 - 2.61929e-16 - 7.95792e-16 - 2.36893e-15 - 6.90944e-15 - 1.97456e-14 - 5.52884e-14 - 1.51682e-13 - 4.0773e-13 - 1.07386e-12 - 2.77114e-12 - 7.00658e-12 - 1.73577e-11 - 4.21321e-11 - 1.00201e-10 - 2.33489e-10 - 5.33089e-10 - 1.19253e-09 - 2.61381e-09 - 5.61329e-09 - 1.18113e-08 - 2.43508e-08 - 4.91889e-08 - 9.73549e-08 - 1.88793e-07 - 3.58714e-07 - 6.67805e-07 - 1.21811e-06 - 2.17702e-06 - 3.81218e-06 - 6.54066e-06 - 1.09953e-05 - 1.81104e-05 - 2.92272e-05 - 4.6215e-05 - 7.16003e-05 - 0.000108689 - 0.000161656 - 0.000235577 - 0.000336367 - 0.000470576 - 0.000645036 - 0.000866312 - 0.00113999 - 0.00146982 - 0.00185681 - 0.00229829 - 0.00278727 - 0.00331201 - 0.00385604 - 0.00439875 - 0.00491649 - 0.0053842 - 0.00577737 - 0.00607415 - 0.00625743 - 0.00631647 - 0.00624807 - 0.00605694 - 0.00575536 - 0.00536215 - 0.00490107 - 0.0043989 - 0.00388353 - 0.00338212 - 0.00291963 - 0.00251784 - 0.00219476 - 0.0019644 - 0.00183683 - 0.00181827 - 0.00191123 - 0.00211441 - 0.0024225 - 0.0028259 - 0.00331037 - 0.00385693 - 0.00444204 - 0.00503828 - 0.00561553 - 0.0061427 - 0.00658983 - 0.00693041 - 0.0071435 - 0.00721563 - 0.00714193 - 0.00692656 - 0.00658218 - 0.00612867 - 0.00559119 - 0.00499782 - 0.0043772 - 0.0037562 - 0.00315819 - 0.00260175 - 0.00210005 - 0.00166085 - 0.00128697 - 0.000977109 - 0.000726866 - 0.000529789 - 0.000378345 - 0.000264734 - 0.000181497 - 0.000121917 - 8.02412e-05 - 5.17448e-05 - 3.26944e-05 - 2.02403e-05 - 1.22771e-05 - 7.29648e-06 - 4.24881e-06 - 2.42414e-06 - 1.35514e-06 - 7.42249e-07 - 3.98337e-07 - 2.09454e-07 - 1.0791e-07 - 5.44722e-08 - 2.69416e-08 - 1.3056e-08 - 6.19913e-09 - 2.88397e-09 - 1.31458e-09 - 5.87109e-10 - 2.56914e-10 - 1.10153e-10 - 4.62739e-11 - 1.90459e-11 - 7.67917e-12 - 3.02905e-12 - 1.15766e-12 - 3.97361e-13 - 3.41409e-14 - -2.89899e-13 - -8.92159e-13 - -2.32794e-12 - -5.84648e-12 - -1.43519e-11 - -3.45078e-11 - -8.12912e-11 - -1.87631e-10 - -4.24327e-10 - -9.40231e-10 - -2.04129e-09 - -4.34222e-09 - -9.05016e-09 - -1.84815e-08 - -3.6979e-08 - -7.24953e-08 - -1.39252e-07 - -2.62077e-07 - -4.83275e-07 - -8.73166e-07 - -1.54574e-06 - -2.68109e-06 - -4.55642e-06 - -7.58706e-06 - -1.23783e-05 - -1.97871e-05 - -3.09915e-05 - -4.75597e-05 - -7.15111e-05 - -0.000105352 - -0.000152073 - -0.000215078 - -0.000298041 - -0.000404664 - -0.00053833 - -0.000701682 - -0.000896126 - -0.00112133 - -0.00137479 - -0.00165149 - -0.0019438 - -0.00224163 - -0.00253287 - -0.00280414 - -0.00304174 - -0.00323283 - -0.0033665 - -0.00343488 - -0.00343384 - -0.00336344 - -0.0032279 - -0.00303518 - -0.00279621 - -0.00252384 - -0.00223166 - -0.00193291 - -0.00163942 - -0.00136091 - -0.00110451 - -0.000874518 - -0.000672571 - -0.00049787 - -0.000347626 - -0.000217553 - -0.000102387 - 3.62246e-06 - 0.000106247 - 0.000210868 - 0.000322098 - 0.000443445 - 0.000577031 - 0.000723357 - 0.000881165 - 0.0010474 - 0.00121734 - 0.00138482 - 0.0015427 - 0.00168342 - 0.00179962 - 0.00188484 - 0.00193416 - 0.00194463 - 0.00191564 - 0.00184894 - 0.00174849 - 0.00162005 - 0.00147064 - 0.00130789 - 0.00113938 - 0.000972049 - 0.000811701 - 0.000662676 - 0.000527676 - 0.000407728 - 0.000302261 - 0.000209254 - 0.000125426 - 4.64271e-05 - -3.29605e-05 - -0.000118629 - -0.000216928 - -0.0003344 - -0.000477448 - -0.000651899 - -0.000862487 - -0.00111225 - -0.00140188 - -0.00172921 - -0.00208869 - -0.00247127 - -0.00286449 - -0.003253 - -0.00361946 - -0.0039458 - -0.00421463 - -0.00441082 - -0.00452289 - -0.00454411 - -0.00447319 - -0.00431444 - -0.00407725 - -0.00377526 - -0.00342502 - -0.00304451 - -0.0026516 - -0.00226274 - -0.0018919 - -0.00154988 - -0.00124405 - -0.000978389 - -0.000753916 - -0.000569208 - -0.000421072 - -0.000305196 - -0.000216739 - -0.000150811 - -0.000102817 - -6.86807e-05 - -4.49512e-05 - -2.8826e-05 - -1.81119e-05 - -1.11501e-05 - -6.72565e-06 - -3.97489e-06 - -2.30172e-06 - -1.30592e-06 - -7.25967e-07 - -3.95417e-07 - -2.11023e-07 - -1.10342e-07 - -5.65314e-08 - -2.83775e-08 - -1.39572e-08 - -6.72598e-09 - -3.17579e-09 - -1.46921e-09 - -6.65968e-10 - -2.95774e-10 - -1.28707e-10 - -5.48761e-11 - -2.29245e-11 - -9.38325e-12 - -3.76308e-12 - -1.47867e-12 - -5.69293e-13 - -2.14752e-13 - -7.93733e-14 - -2.87441e-14 - -1.01991e-14 - -3.54576e-15 - -1.2078e-15 - -4.03104e-16 - -1.31818e-16 - -4.22349e-17 - -1.32588e-17 - -4.07825e-18 - -1.22908e-18 - -3.62931e-19 - -1.05004e-19 - -2.9766e-20 - -8.26751e-21 - -2.24991e-21 - -5.99919e-22 - -1.56732e-22 - -4.01224e-23 - -1.00739e-23 - -2.52276e-24 - -8.0687e-25 - -1.016e-24 - -3.58361e-24 - -1.41196e-23 - -5.4951e-23 - -2.09636e-22 - -7.83616e-22 - -2.86998e-21 - -1.02989e-20 - -3.62111e-20 - -1.24746e-19 - -4.21067e-19 - -1.39255e-18 - -4.5124e-18 - -1.43265e-17 - -4.45669e-17 - -1.35837e-16 - -4.05661e-16 - -1.18698e-15 - -3.40301e-15 - -9.55911e-15 - -2.63093e-14 - -7.09476e-14 - -1.87458e-13 - -4.85294e-13 - -1.23096e-12 - -3.05929e-12 - -7.4496e-12 - -1.77739e-11 - -4.15498e-11 - -9.51682e-11 - -2.13576e-10 - -4.69623e-10 - -1.01177e-09 - -2.13577e-09 - -4.41734e-09 - -8.9517e-09 - -1.77741e-08 - -3.45784e-08 - -6.59113e-08 - -1.23098e-07 - -2.25258e-07 - -4.03874e-07 - -7.09493e-07 - -1.2212e-06 - -2.05951e-06 - -3.40311e-06 - -5.50968e-06 - -8.74004e-06 - -1.35843e-05 - -2.06871e-05 - -3.08674e-05 - -4.51271e-05 - -6.46419e-05 - -9.07258e-05 - -0.000124764 - -0.00016811 - -0.000221944 - -0.000287112 - -0.000363934 - -0.000452037 - -0.000550211 - -0.000656332 - -0.000767384 - -0.000879594 - -0.000988694 - -0.00109031 - -0.00118047 - -0.00125612 - -0.00131572 - -0.00135978 - -0.00139119 - -0.00141553 - -0.00144105 - -0.00147842 - -0.00154024 - -0.00164022 - -0.00179215 - -0.0020087 - -0.00229996 - -0.00267211 - -0.00312609 - -0.00365653 - -0.00425123 - -0.00489099 - -0.00555038 - -0.00619899 - -0.00680353 - -0.00733032 - -0.00774812 - -0.00803088 - -0.00816018 - -0.00812687 - -0.00793199 - -0.0075865 - -0.00711016 - -0.00652951 - -0.00587539 - -0.00518015 - -0.00447502 - -0.00378787 - -0.0031416 - -0.00255317 - -0.00203343 - -0.00158747 - -0.0012155 - -0.000914005 - -0.000677014 - -0.000497327 - -0.000367638 - -0.00028146 - -0.000233825 - -0.000221778 - -0.000244658 - -0.000304193 - -0.000404393 - -0.000551247 - -0.0007522 - -0.00101539 - -0.00134866 - -0.00175834 - -0.00224794 - -0.00281677 - -0.00345876 - -0.00416153 - -0.00490608 - -0.00566706 - -0.00641387 - -0.00711248 - -0.00772785 - -0.00822684 - -0.00858114 - -0.00876987 - -0.0087817 - -0.0086159 - -0.00828247 - -0.0078011 - -0.00719927 - -0.00650966 - -0.00576719 - -0.0050062 - -0.00425783 - -0.00354817 - -0.00289707 - -0.00231766 - -0.00181667 - -0.00139522 - -0.00104989 - -0.000774071 - -0.000559185 - -0.000395792 - -0.000274483 - -0.00018651 - -0.000124172 - -8.09996e-05 - -5.17701e-05 - -3.24199e-05 - -1.98921e-05 - -1.19588e-05 - -7.04419e-06 - -4.06548e-06 - -2.29895e-06 - -1.27376e-06 - -6.91503e-07 - -3.67873e-07 - -1.91872e-07 - -9.83377e-08 - -5.00377e-08 - -2.64246e-08 - -1.68924e-08 - -1.70683e-08 - -2.72651e-08 - -5.27365e-08 - -1.05874e-07 - -2.1103e-07 - -4.13329e-07 - -7.93705e-07 - -1.49355e-06 - -2.75377e-06 - -4.97481e-06 - -8.80568e-06 - -1.52717e-05 - -2.59505e-05 - -4.32059e-05 - -7.04818e-05 - -0.000112654 - -0.000176423 - -0.000270707 - -0.000406987 - -0.000599513 - -0.000865274 - -0.00122362 - -0.00169541 - -0.00230165 - -0.00306155 - -0.00399008 - -0.00509515 - -0.00637485 - -0.00781483 - -0.00938655 - -0.0110466 - -0.0127377 - -0.0143909 - -0.0159302 - -0.0172779 - -0.0183611 - -0.0191181 - -0.0195041 - -0.0194959 - -0.0190941 - -0.0183227 - -0.0172274 - -0.0158703 - -0.0143247 - -0.0126685 - -0.0109775 - -0.00931998 - -0.00775291 - -0.00631905 - -0.00504632 - -0.00394853 - -0.00302714 - -0.00227387 - -0.00167355 - -0.00120683 - -0.000852687 - -0.000590298 - -0.000400396 - -0.0002661 - -0.000173275 - -0.000110552 - -6.91084e-05 - -4.23286e-05 - -2.54023e-05 - -1.49365e-05 - -8.60522e-06 - -4.85749e-06 - -2.68657e-06 - -1.45587e-06 - -7.73006e-07 - -4.02142e-07 - -2.04981e-07 - -1.02373e-07 - -5.00948e-08 - -2.4018e-08 - -1.12829e-08 - -5.19323e-09 - -2.34203e-09 - -1.03487e-09 - -4.48039e-10 - -1.90063e-10 - -7.90133e-11 - -3.2227e-11 - -1.2995e-11 - -5.44122e-12 - -3.02022e-12 - -3.51191e-12 - -7.21295e-12 - -1.70439e-11 - -4.05427e-11 - -9.4878e-11 - -2.17679e-10 - -4.89379e-10 - -1.07799e-09 - -2.32661e-09 - -4.92003e-09 - -1.01941e-08 - -2.06951e-08 - -4.11646e-08 - -8.02261e-08 - -1.53195e-07 - -2.86623e-07 - -5.25429e-07 - -9.43742e-07 - -1.66085e-06 - -2.8638e-06 - -4.83831e-06 - -8.00904e-06 - -1.29899e-05 - -2.06426e-05 - -3.21413e-05 - -4.9034e-05 - -7.3294e-05 - -0.000107344 - -0.000154036 - -0.000216573 - -0.000298347 - -0.000402696 - -0.000532562 - -0.00069008 - -0.000876123 - -0.00108985 - -0.00132834 - -0.0015863 - -0.00185609 - -0.00212789 - -0.00239021 - -0.00263063 - -0.00283674 - -0.00299722 - -0.00310279 - -0.0031472 - -0.00312775 - -0.00304563 - -0.00290576 - -0.0027163 - -0.0024879 - -0.00223268 - -0.00196316 - -0.0016913 - -0.00142766 - -0.00118077 - -0.000956844 - -0.000759722 - -0.000591024 - -0.000450497 - -0.000336447 - -0.000246194 - -0.000176512 - -0.000123997 - -8.53456e-05 - -5.75558e-05 - -3.80307e-05 - -2.46216e-05 - -1.56184e-05 - -9.70714e-06 - -5.91131e-06 - -3.52706e-06 - -2.06196e-06 - -1.18109e-06 - -6.62861e-07 - -3.64501e-07 - -1.96387e-07 - -1.03672e-07 - -5.36229e-08 - -2.71753e-08 - -1.34939e-08 - -6.56499e-09 - -3.12945e-09 - -1.46164e-09 - -6.6888e-10 - -2.99912e-10 - -1.31758e-10 - -5.67146e-11 - -2.39194e-11 - -9.88424e-12 - -4.00196e-12 - -1.58759e-12 - -6.17081e-13 - -2.35008e-13 - -8.76919e-14 - -3.20607e-14 - -1.14848e-14 - -4.03098e-15 - -1.38623e-15 - -4.67086e-16 - -1.54204e-16 - -4.98803e-17 - -1.58089e-17 - -4.90919e-18 - -1.49367e-18 - -4.45284e-19 - -1.30064e-19 - -3.72232e-20 - -1.04383e-20 - -2.87015e-21 - -7.81928e-22 - -2.43429e-22 - -2.08622e-22 - -6.12204e-22 - -2.2559e-21 - -8.30244e-21 - -2.99767e-20 - -1.06056e-19 - -3.67643e-19 - -1.24869e-18 - -4.15545e-18 - -1.35494e-17 - -4.32871e-17 - -1.35498e-16 - -4.15571e-16 - -1.2488e-15 - -3.67688e-15 - -1.06072e-14 - -2.99821e-14 - -8.30344e-14 - -2.25315e-13 - -5.99047e-13 - -1.56051e-12 - -3.98301e-12 - -9.96075e-12 - -2.44067e-11 - -5.85955e-11 - -1.37834e-10 - -3.17675e-10 - -7.1738e-10 - -1.58727e-09 - -3.44105e-09 - -7.30916e-09 - -1.52118e-08 - -3.10193e-08 - -6.19757e-08 - -1.21324e-07 - -2.32709e-07 - -4.37337e-07 - -8.05303e-07 - -1.45292e-06 - -2.56841e-06 - -4.44867e-06 - -7.54986e-06 - -1.25543e-05 - -2.04549e-05 - -3.26554e-05 - -5.10821e-05 - -7.82973e-05 - -0.000117598 - -0.000173075 - -0.000249615 - -0.000352798 - -0.000488684 - -0.000663453 - -0.000882913 - -0.00115189 - -0.00147356 - -0.00184879 - -0.00227563 - -0.00274905 - -0.00326099 - -0.00380083 - -0.00435636 - -0.00491499 - -0.00546527 - -0.00599832 - -0.00650907 - -0.00699689 - -0.00746558 - -0.00792243 - -0.00837658 - -0.00883672 - -0.00930857 - -0.00979246 - -0.0102816 - -0.0107611 - -0.0112087 - -0.011596 - -0.0118916 - -0.0120645 - -0.0120879 - -0.0119427 - -0.0116203 - -0.011124 - -0.0104692 - -0.00968215 - -0.00879775 - -0.00785624 - -0.00689989 - -0.00596968 - -0.00510249 - -0.00432903 - -0.00367258 - -0.00314848 - -0.00276429 - -0.00252032 - -0.00241049 - -0.00242322 - -0.00254229 - -0.00274769 - -0.00301656 - -0.00332402 - -0.00364428 - -0.00395183 - -0.00422275 - -0.00443609 - -0.00457514 - -0.00462847 - -0.0045906 - -0.00446213 - -0.00424935 - -0.00396339 - -0.0036188 - -0.00323204 - -0.00281971 - -0.00239708 - -0.0019767 - -0.00156754 - -0.00117457 - -0.000798753 - -0.000437587 - -8.59164e-05 - 0.000262944 - 0.000615954 - 0.000979073 - 0.00135609 - 0.00174768 - 0.00215078 - 0.00255837 - 0.00295977 - 0.00334128 - 0.00368735 - 0.00398195 - 0.00421019 - 0.00435982 - 0.00442255 - 0.00439499 - 0.00427906 - 0.00408186 - 0.003815 - 0.00349351 - 0.00313447 - 0.00275551 - 0.00237344 - 0.00200306 - 0.00165638 - 0.00134211 - 0.00106565 - 0.000829314 - 0.000632838 - 0.00047399 - 0.000349261 - 0.00025451 - 0.000185531 - 0.00013851 - 0.000110364 - 9.89575e-05 - 0.000103216 - 0.000123152 - 0.000159795 - 0.000215048 - 0.000291453 - 0.000391873 - 0.000519088 - 0.000675318 - 0.000861708 - 0.00107781 - 0.00132113 - 0.00158679 - 0.00186744 - 0.00215336 - 0.00243292 - 0.00269324 - 0.0029212 - 0.00310445 - 0.00323254 - 0.00329792 - 0.00329666 - 0.00322882 - 0.0030985 - 0.00291337 - 0.00268396 - 0.00242267 - 0.00214264 - 0.00185669 - 0.00157641 - 0.0013114 - 0.0010689 - 0.00085364 - 0.00066796 - 0.00051211 - 0.000384691 - 0.000283138 - 0.000204184 - 0.000144272 - 9.98799e-05 - 6.77504e-05 - 4.5028e-05 - 2.93217e-05 - 1.87083e-05 - 1.16954e-05 - 7.16366e-06 - 4.29928e-06 - 2.52821e-06 - 1.45697e-06 - 8.23301e-07 - 4.57272e-07 - 2.52047e-07 - 1.4307e-07 - 9.41611e-08 - 8.92735e-08 - 1.29198e-07 - 2.33164e-07 - 4.46224e-07 - 8.54461e-07 - 1.61153e-06 - 2.98178e-06 - 5.40736e-06 - 9.60873e-06 - 1.67298e-05 - 2.854e-05 - 4.77041e-05 - 7.81257e-05 - 0.000125363 - 0.000197097 - 0.000303619 - 0.000458263 - 0.0006777 - 0.000981967 - 0.0013941 - 0.00193921 - 0.00264299 - 0.00352941 - 0.00461791 - 0.00592006 - 0.00743607 - 0.0091516 - 0.0110354 - 0.0130381 - 0.0150932 - 0.0171191 - 0.0190248 - 0.0207155 - 0.0221008 - 0.0231024 - 0.0236615 - 0.0237446 - 0.0233467 - 0.0224919 - 0.0212307 - 0.0196357 - 0.0177941 - 0.0158004 - 0.0137482 - 0.0117235 - 0.00979954 - 0.00803325 - 0.00646427 - 0.00511562 - 0.00399594 - 0.00310269 - 0.00242575 - 0.00195092 - 0.00166287 - 0.00154732 - 0.00159223 - 0.00178801 - 0.00212689 - 0.00260153 - 0.00320325 - 0.00392002 - 0.00473471 - 0.00562383 - 0.006557 - 0.00749741 - 0.00840331 - 0.00923048 - 0.00993535 - 0.0104786 - 0.0108286 - 0.0109643 - 0.0108776 - 0.0105736 - 0.0100705 - 0.00939751 - 0.00859238 - 0.00769753 - 0.00675657 - 0.00581083 - 0.00489651 - 0.0040427 - 0.00327035 - 0.00259211 - 0.00201302 - 0.00153173 - 0.00114196 - 0.000834175 - 0.000597036 - 0.000418679 - 0.000287673 - 0.000193667 - 0.000127749 - 8.25696e-05 - 5.23006e-05 - 3.24804e-05 - 1.98088e-05 - 1.19273e-05 - 7.21506e-06 - 4.61863e-06 - 3.52388e-06 - 3.67649e-06 - 5.15343e-06 - 8.3891e-06 - 1.42612e-05 - 2.42428e-05 - 4.06281e-05 - 6.68359e-05 - 0.000107789 - 0.000170354 - 0.000263818 - 0.000400329 - 0.00059524 - 0.000867232 - 0.0012381 - 0.00173205 - 0.0023745 - 0.00319011 - 0.00420035 - 0.0054206 - 0.00685696 - 0.00850339 - 0.0103395 - 0.0123294 - 0.0144223 - 0.0165546 - 0.0186545 - 0.0206468 - 0.0224604 - 0.0240341 - 0.0253228 - 0.026301 - 0.0269648 - 0.0273304 - 0.0274301 - 0.0273064 - 0.0270053 - 0.0265691 - 0.0260305 - 0.0254095 - 0.0247121 - 0.0239319 - 0.023054 - 0.0220604 - 0.0209356 - 0.0196714 - 0.0182707 - 0.0167493 - 0.0151347 - 0.0134645 - 0.0117821 - 0.0101327 - 0.00855887 - 0.00709693 - 0.0057744 - 0.00460875 - 0.00360736 - 0.00276847 - 0.00208292 - 0.00153623 - 0.00111069 - 0.000787407 - 0.000547808 - 0.000374892 - 0.000253994 - 0.000173243 - 0.000123774 - 9.98233e-05 - 9.87971e-05 - 0.000121408 - 0.000171927 - 0.000258572 - 0.000393997 - 0.000595817 - 0.00088703 - 0.00129616 - 0.00185693 - 0.00260716 - 0.00358686 - 0.00483517 - 0.00638631 - 0.00826471 - 0.0104795 - 0.0130195 - 0.0158483 - 0.018902 - 0.0220888 - 0.0252913 - 0.0283731 - 0.0311875 - 0.0335885 - 0.0354435 - 0.0366455 - 0.0371228 - 0.0368467 - 0.0358338 - 0.0341447 - 0.031878 - 0.0291606 - 0.0261359 - 0.0229518 - 0.0197484 - 0.0166488 - 0.0137522 - 0.0111301 - 0.00882594 - 0.00685741 - 0.00522031 - 0.00389377 - 0.00284564 - 0.00203764 - 0.00142959 - 0.000982723 - 0.000661894 - 0.0004368 - 0.000282432 - 0.000178929 - 0.000111067 - 6.75505e-05 - 4.02538e-05 - 2.35029e-05 - 1.34454e-05 - 7.53639e-06 - 4.13894e-06 - 2.22716e-06 - 1.17423e-06 - 6.0658e-07 - 3.07016e-07 - 1.52255e-07 - 7.3981e-08 - 3.52217e-08 - 1.64312e-08 - 7.51334e-09 - 3.37362e-09 - 1.50265e-09 - 7.0035e-10 - 4.24077e-10 - 4.75335e-10 - 8.76813e-10 - 1.8765e-09 - 4.0748e-09 - 8.72522e-09 - 1.83265e-08 - 3.77229e-08 - 7.60823e-08 - 1.5035e-07 - 2.91111e-07 - 5.5227e-07 - 1.02655e-06 - 1.86959e-06 - 3.33619e-06 - 5.83299e-06 - 9.99237e-06 - 1.67719e-05 - 2.75825e-05 - 4.44448e-05 - 7.0169e-05 - 0.000108544 - 0.000164515 - 0.000244309 - 0.000355477 - 0.000506781 - 0.00070789 - 0.000968832 - 0.00129918 - 0.00170696 - 0.00219744 - 0.0027717 - 0.00342542 - 0.00414779 - 0.00492104 - 0.00572051 - 0.00651551 - 0.00727109 - 0.00795037 - 0.00851749 - 0.00894073 - 0.00919542 - 0.00926631 - 0.0091491 - 0.0088509 - 0.00838944 - 0.0077914 - 0.00708982 - 0.00632108 - 0.00552185 - 0.00472622 - 0.00396352 - 0.00325675 - 0.00262195 - 0.00206825 - 0.00159851 - 0.00121051 - 0.000898163 - 0.00065295 - 0.000465095 - 0.000324594 - 0.00022196 - 0.000148712 - 9.76239e-05 - 6.27917e-05 - 3.95717e-05 - 2.44345e-05 - 1.47829e-05 - 8.76302e-06 - 5.0896e-06 - 2.89635e-06 - 1.61493e-06 - 8.82256e-07 - 4.7225e-07 - 2.47678e-07 - 1.27274e-07 - 6.40809e-08 - 3.1613e-08 - 1.52825e-08 - 7.24349e-09 - 3.37597e-09 - 1.57127e-09 - 7.87334e-10 - 5.49235e-10 - 7.08757e-10 - 1.36716e-09 - 2.92932e-09 - 6.31231e-09 - 1.33936e-08 - 2.78707e-08 - 5.68338e-08 - 1.13558e-07 - 2.22314e-07 - 4.26437e-07 - 8.01454e-07 - 1.47584e-06 - 2.66279e-06 - 4.7073e-06 - 8.15349e-06 - 1.38373e-05 - 2.3009e-05 - 3.74869e-05 - 5.98409e-05 - 9.35953e-05 - 0.000143432 - 0.000215365 - 0.000316842 - 0.000456715 - 0.000645038 - 0.000892611 - 0.00121025 - 0.00160778 - 0.00209273 - 0.00266894 - 0.00333503 - 0.00408317 - 0.00489815 - 0.0057571 - 0.00662998 - 0.00748096 - 0.00827066 - 0.008959 - 0.00950859 - 0.00988805 - 0.010075 - 0.0100581 - 0.00983853 - 0.00942953 - 0.00885532 - 0.00814865 - 0.00734787 - 0.0064935 - 0.00562505 - 0.0047782 - 0.00398281 - 0.00326164 - 0.00263011 - 0.00209661 - 0.00166348 - 0.00132822 - 0.00108488 - 0.000925197 - 0.000839661 - 0.000818171 - 0.000850452 - 0.000926225 - 0.00103522 - 0.00116715 - 0.00131169 - 0.00145858 - 0.00159787 - 0.00172029 - 0.00181768 - 0.00188348 - 0.00191318 - 0.00190459 - 0.00185802 - 0.0017761 - 0.00166357 - 0.00152673 - 0.00137286 - 0.00120957 - 0.00104417 - 0.000883183 - 0.000731926 - 0.000594321 - 0.000472837 - 0.000368587 - 0.000281517 - 0.000210672 - 0.000154471 - 0.000110977 - 7.81216e-05 - 5.38892e-05 - 3.64365e-05 - 2.41667e-05 - 1.57607e-05 - 1.0179e-05 - 6.64527e-06 - 4.62443e-06 - 3.80367e-06 - 4.08572e-06 - 5.6009e-06 - 8.74399e-06 - 1.42407e-05 - 2.32471e-05 - 3.74834e-05 - 5.93981e-05 - 9.23544e-05 - 0.000140822 - 0.000210543 - 0.000308644 - 0.000443633 - 0.000625244 - 0.000864061 - 0.0011709 - 0.00155593 - 0.0020275 - 0.00259089 - 0.00324684 - 0.00399034 - 0.00480954 - 0.00568528 - 0.00659119 - 0.00749458 - 0.00835814 - 0.00914232 - 0.00980831 - 0.0103211 - 0.0106527 - 0.0107844 - 0.0107086 - 0.0104298 - 0.00996373 - 0.00933625 - 0.00858073 - 0.00773524 - 0.00683939 - 0.0059313 - 0.00504505 - 0.00420879 - 0.00344364 - 0.00276337 - 0.00217477 - 0.00167855 - 0.00127055 - 0.000943173 - 0.000686672 - 0.00049038 - 0.000343661 - 0.000236618 - 0.00016055 - 0.000108187 - 7.37676e-05 - 5.3005e-05 - 4.29873e-05 - 4.20568e-05 - 4.9689e-05 - 6.63831e-05 - 9.35609e-05 - 0.000133462 - 0.000189012 - 0.000263645 - 0.00036105 - 0.000484834 - 0.000638098 - 0.000822944 - 0.00103994 - 0.00128764 - 0.00156213 - 0.00185686 - 0.00216261 - 0.00246782 - 0.00275923 - 0.00302271 - 0.00324447 - 0.00341214 - 0.00351599 - 0.00354981 - 0.00351155 - 0.00340353 - 0.0032322 - 0.00300748 - 0.00274185 - 0.00244919 - 0.00214357 - 0.00183819 - 0.00154447 - 0.00127146 - 0.00102557 - 0.000810524 - 0.000627627 - 0.000476183 - 0.000353982 - 0.000257822 - 0.000183984 - 0.000128626 - 8.80794e-05 - 5.90389e-05 - 3.86628e-05 - 2.45949e-05 - 1.49291e-05 - 8.13403e-06 - 2.94857e-06 - -1.74116e-06 - -7.04209e-06 - -1.42113e-05 - -2.48328e-05 - -4.10146e-05 - -6.56039e-05 - -0.000102411 - -0.000156423 - -0.000233969 - -0.000342807 - -0.000492057 - -0.000691949 - -0.000953302 - -0.00128673 - -0.00170157 - -0.00220453 - -0.00279826 - -0.00347991 - -0.00423992 - -0.00506123 - -0.00591925 - -0.0067825 - -0.00761422 - -0.00837483 - -0.00902489 - -0.00952848 - -0.00985649 - -0.00998939 - -0.00991914 - -0.00965003 - -0.00919821 - -0.00859011 - -0.00785987 - -0.00704619 - -0.00618894 - -0.005326 - -0.00449065 - -0.00370973 - -0.00300262 - -0.00238113 - -0.00185009 - -0.0014084 - -0.00105048 - -0.000767675 - -0.000549659 - -0.0003856 - -0.000265039 - -0.000178489 - -0.000117771 - -7.61375e-05 - -4.82267e-05 - -2.99299e-05 - -1.81993e-05 - -1.08426e-05 - -6.32909e-06 - -3.61977e-06 - -2.0284e-06 - -1.11367e-06 - -5.99086e-07 - -3.15758e-07 - -1.63062e-07 - -8.25054e-08 - -4.0902e-08 - -1.98673e-08 - -9.45511e-09 - -4.40887e-09 - -2.01428e-09 - -9.01668e-10 - -3.95464e-10 - -1.69942e-10 - -7.15528e-11 - -2.9518e-11 - -1.19311e-11 - -4.72509e-12 - -1.83346e-12 - -6.97057e-13 - -2.59655e-13 - -9.47633e-14 - -3.3872e-14 - -1.18181e-14 - -3.90035e-15 - -8.29272e-16 - 1.18263e-15 - 4.68453e-15 - 1.39394e-14 - 3.96556e-14 - 1.10235e-13 - 3.00124e-13 - 8.00503e-13 - 2.0918e-12 - 5.35517e-12 - 1.34315e-11 - 3.30047e-11 - 7.94561e-11 - 1.87405e-10 - 4.33047e-10 - 9.80379e-10 - 2.17449e-09 - 4.72525e-09 - 1.006e-08 - 2.09837e-08 - 4.28816e-08 - 8.58558e-08 - 1.68414e-07 - 3.23665e-07 - 6.09432e-07 - 1.12426e-06 - 2.03198e-06 - 3.59822e-06 - 6.24264e-06 - 1.06112e-05 - 1.76715e-05 - 2.88336e-05 - 4.60937e-05 - 7.21939e-05 - 0.000110784 - 0.00016656 - 0.000245348 - 0.00035409 - 0.000500684 - 0.000693638 - 0.000941505 - 0.00125208 - 0.0016314 - 0.00208262 - 0.00260483 - 0.00319202 - 0.00383237 - 0.00450799 - 0.00519525 - 0.00586582 - 0.00648837 - 0.00703083 - 0.0074629 - 0.00775862 - 0.00789868 - 0.00787201 - 0.00767662 - 0.00731945 - 0.00681534 - 0.00618516 - 0.0054536 - 0.00464667 - 0.00378961 - 0.00290508 - 0.00201228 - 0.00112666 - 0.000260383 - -0.000576704 - -0.00137586 - -0.00212829 - -0.0028243 - -0.0034529 - -0.00400205 - -0.00445942 - -0.00481357 - -0.0050554 - -0.00517947 - -0.00518511 - -0.00507705 - -0.00486543 - -0.00456522 - -0.00419512 - -0.00377605 - -0.00332958 - -0.00287625 - -0.00243427 - -0.00201849 - -0.00163986 - -0.00130532 - -0.00101802 - -0.000777914 - -0.000582426 - -0.000427253 - -0.000307089 - -0.000216263 - -0.000149222 - -0.000100884 - -6.68267e-05 - -4.33723e-05 - -2.75812e-05 - -1.7185e-05 - -1.04911e-05 - -6.27527e-06 - -3.67773e-06 - -2.11185e-06 - -1.18819e-06 - -6.55e-07 - -3.53781e-07 - -1.87222e-07 - -9.70701e-08 - -4.92929e-08 - -2.44779e-08 - -1.17909e-08 - -5.27448e-09 - -1.60946e-09 - 1.27482e-09 - 5.18955e-09 - 1.26581e-08 - 2.82967e-08 - 6.12173e-08 - 1.29452e-07 - 2.68086e-07 - 5.43925e-07 - 1.08126e-06 - 2.10601e-06 - 4.01906e-06 - 7.51496e-06 - 1.37678e-05 - 2.47139e-05 - 4.34663e-05 - 7.49034e-05 - 0.00012647 - 0.000209223 - 0.000339133 - 0.0005386 - 0.000838107 - 0.00127782 - 0.00190887 - 0.00279396 - 0.00400683 - 0.00563014 - 0.00775128 - 0.010456 - 0.0138195 - 0.0178961 - 0.022707 - 0.0282292 - 0.0343855 - 0.0410382 - 0.0479888 - 0.0549831 - 0.0617247 - 0.0678938 - 0.0731722 - 0.0772704 - 0.0799544 - 0.0810684 - 0.0805512 - 0.0784438 - 0.0748868 - 0.0701092 - 0.0644092 - 0.0581302 - 0.0516357 - 0.0452844 - 0.0394097 - 0.0343041 - 0.0302089 - 0.0273084 - 0.0257283 - 0.0255355 - 0.0267383 - 0.0292872 - 0.0330737 - 0.0379307 - 0.0436337 - 0.0499051 - 0.0564229 - 0.0628345 - 0.0687764 - 0.073897 - 0.0778824 - 0.0804804 - 0.0815217 - 0.0809336 - 0.0787451 - 0.0750827 - 0.0701565 - 0.0642396 - 0.0576423 - 0.0506847 - 0.0436712 - 0.0368695 - 0.030495 - 0.024703 - 0.0195869 - 0.0151828 - 0.0114776 - 0.00841987 - 0.00593219 - 0.00392215 - 0.00229274 - 0.000950838 - -0.000186034 - -0.00118449 - -0.00209138 - -0.00293083 - -0.00370225 - -0.00437912 - -0.00490892 - -0.00521447 - -0.00519708 - -0.00474199 - -0.00372615 - -0.00202842 - 0.000458113 - 0.00381264 - 0.00807298 - 0.0132233 - 0.0191849 - 0.0258114 - 0.0328904 - 0.0401515 - 0.047281 - 0.0539439 - 0.059808 - 0.0645718 - 0.0679897 - 0.0698937 - 0.0702075 - 0.0689522 - 0.0662425 - 0.062274 - 0.0573033 - 0.0516237 - 0.0455395 - 0.0393419 - 0.0332886 - 0.0275894 - 0.0223987 - 0.017814 - 0.0138795 - 0.0105943 - 0.00792264 - 0.0058046 - 0.00416665 - 0.00293035 - 0.00201918 - 0.00136318 - 0.000901698 - 0.000584386 - 0.000371087 - 0.000230892 - 0.000140786 - 8.41686e-05 - 4.94262e-05 - 2.86893e-05 - 1.68175e-05 - 1.06308e-05 - 8.37418e-06 - 9.39916e-06 - 1.40463e-05 - 2.3726e-05 - 4.1207e-05 - 7.11357e-05 - 0.000120818 - 0.000201294 - 0.000328719 - 0.000526031 - 0.000824827 - 0.00126728 - 0.00190781 - 0.00281418 - 0.00406746 - 0.00576033 - 0.00799326 - 0.0108681 - 0.014479 - 0.0189005 - 0.0241747 - 0.0302971 - 0.0372043 - 0.044765 - 0.0527758 - 0.0609654 - 0.0690054 - 0.0765305 - 0.0831643 - 0.0885505 - 0.0923838 - 0.094439 - 0.0945928 - 0.0928357 - 0.0892735 - 0.0841163 - 0.0776584 - 0.0702501 - 0.0622666 - 0.0540773 - 0.0460179 - 0.0383702 - 0.0313489 - 0.0250975 - 0.0196902 - 0.0151414 - 0.0114171 - 0.00844929 - 0.00614968 - 0.00442187 - 0.00317137 - 0.00231276 - 0.00177424 - 0.00149973 - 0.00144902 - 0.00159637 - 0.00192796 - 0.00243858 - 0.00312782 - 0.00399592 - 0.00503982 - 0.00624939 - 0.00760438 - 0.00907248 - 0.0106086 - 0.0121559 - 0.0136483 - 0.0150148 - 0.0161846 - 0.0170932 - 0.0176882 - 0.0179342 - 0.0178162 - 0.0173415 - 0.0165385 - 0.0154541 - 0.0141492 - 0.012693 - 0.0111571 - 0.00960964 - 0.00811103 - 0.00671036 - 0.00544389 - 0.00433493 - 0.00339517 - 0.00262695 - 0.0020262 - 0.00158568 - 0.00129798 - 0.00115831 - 0.0011665 - 0.00132846 - 0.00165655 - 0.00216916 - 0.00288914 - 0.00384126 - 0.00504868 - 0.00652871 - 0.00828802 - 0.0103181 - 0.012591 - 0.015057 - 0.0176439 - 0.0202583 - 0.0227906 - 0.0251218 - 0.0271321 - 0.0287114 - 0.0297689 - 0.0302419 - 0.0301019 - 0.0293576 - 0.0280537 - 0.0262672 - 0.0240992 - 0.0216666 - 0.0190912 - 0.0164912 - 0.0139727 - 0.0116248 - 0.00951633 - 0.00769657 - 0.00619684 - 0.00503464 - 0.00421823 - 0.00375123 - 0.00363649 - 0.00387863 - 0.00448478 - 0.00546347 - 0.00682166 - 0.00856033 - 0.0106692 - 0.0131211 - 0.0158675 - 0.0188355 - 0.0219272 - 0.0250222 - 0.0279839 - 0.0306674 - 0.0329313 - 0.034649 - 0.0357203 - 0.0360812 - 0.0357095 - 0.0346278 - 0.0329006 - 0.030628 - 0.0279365 - 0.0249667 - 0.0218619 - 0.0187564 - 0.015767 - 0.0129863 - 0.01048 - 0.00828649 - 0.00641975 - 0.00487306 - 0.00362429 - 0.00264108 - 0.00188571 - 0.00131919 - 0.000904223 - 0.000607268 - 0.000399597 - 0.000257633 - 0.000162749 - 0.000100733 - 6.10885e-05 - 3.62982e-05 - 2.11324e-05 - 1.20545e-05 - 6.7373e-06 - 3.68943e-06 - 1.97957e-06 - 1.04068e-06 - 5.36045e-07 - 2.70534e-07 - 1.33777e-07 - 6.4815e-08 - 3.07685e-08 - 1.43112e-08 - 6.522e-09 - 2.91221e-09 - 1.2741e-09 - 5.46157e-10 - 2.29388e-10 - 9.43974e-11 - 3.80615e-11 - 1.50366e-11 - 5.82036e-12 - 2.20743e-12 - 8.20277e-13 - 2.98656e-13 - 1.06541e-13 - 3.72394e-14 - 1.27533e-14 - 4.27939e-15 - 1.40694e-15 - 4.5322e-16 - 1.43046e-16 - 4.42367e-17 - 1.34037e-17 - 3.97927e-18 - 1.1575e-18 - 3.29893e-19 - 9.21218e-20 - 2.52051e-20 - 6.75697e-21 - 1.77481e-21 - 4.56761e-22 - 1.15176e-22 - 2.8456e-23 - 6.88845e-24 - 1.63383e-24 - 3.79688e-25 - 8.64541e-26 - 1.92877e-26 - 4.21611e-27 - 9.02986e-28 - 1.8949e-28 - 3.89609e-29 - 7.84888e-30 - 1.54926e-30 - 2.99624e-31 - 5.67762e-32 - 1.05413e-32 - 1.91759e-33 - 3.41788e-34 - 5.9689e-35 - 1.02133e-35 - 1.7123e-36 - 2.81272e-37 - 4.52701e-38 - 7.13892e-39 - 1.10304e-39 - 1.66988e-40 - 2.47695e-41 - 3.59987e-42 - 5.12616e-43 - 7.15212e-44 - 9.77719e-45 - 1.30957e-45 - 1.71863e-46 - 2.2099e-47 - 2.78419e-48 - 3.43687e-49 - 4.15684e-50 - 4.92607e-51 - 5.71972e-52 - 6.50707e-53 - 7.25326e-54 - 7.92169e-55 - 8.47695e-56 - 8.88787e-57 - 9.13047e-58 - 9.1902e-59 - 9.06346e-60 - 8.75789e-61 - 8.29167e-62 - 7.69169e-63 - 6.99098e-64 - 6.22574e-65 - 5.43226e-66 - 4.64417e-67 - 3.89019e-68 - 3.1928e-69 - 2.56749e-70 - 2.02294e-71 - 1.56169e-72 - 1.18125e-73 - 8.75441e-75 - 6.35695e-76 - 4.5228e-77 - 3.15284e-78 - 2.15345e-79 - 1.44113e-80 - 9.44951e-82 - 6.07088e-83 - 3.82148e-84 - 2.35693e-85 - 1.4243e-86 - 8.43317e-88 - 4.89235e-89 - 2.78087e-90 - 1.54875e-91 - 8.45122e-93 - 4.51849e-94 - 2.36704e-95 - 1.21493e-96 - 6.10995e-98 - 3.01064e-99 - 1.45351e-100 - 6.87564e-102 - 3.18673e-103 - 1.44715e-104 - 6.43903e-106 - 2.80713e-107 - 1.19907e-108 - 5.01875e-110 - 2.16259e-111 - 2.64582e-111 - 6.14557e-110 - 1.44359e-108 - 3.32264e-107 - 7.49309e-106 - 1.65567e-104 - 3.58448e-103 - 7.6035e-102 - 1.5803e-100 - 3.2181e-99 - 6.42094e-98 - 1.25526e-96 - 2.4044e-95 - 4.51248e-94 - 8.29776e-93 - 1.49501e-91 - 2.63914e-90 - 4.56477e-89 - 7.73593e-88 - 1.28452e-86 - 2.08982e-85 - 3.3313e-84 - 5.203e-83 - 7.96217e-82 - 1.19384e-80 - 1.75387e-79 - 2.52455e-78 - 3.56049e-77 - 4.92007e-76 - 6.66146e-75 - 8.837e-74 - 1.14862e-72 - 1.4628e-71 - 1.82529e-70 - 2.23159e-69 - 2.67322e-68 - 3.13755e-67 - 3.60815e-66 - 4.06551e-65 - 4.4883e-64 - 4.85496e-63 - 5.14548e-62 - 5.34323e-61 - 5.43648e-60 - 5.41963e-59 - 5.29368e-58 - 5.0662e-57 - 4.75055e-56 - 4.36458e-55 - 3.92897e-54 - 3.46538e-53 - 2.99475e-52 - 2.53575e-51 - 2.10372e-50 - 1.71005e-49 - 1.36196e-48 - 1.06282e-47 - 8.12623e-47 - 6.08774e-46 - 4.46848e-45 - 3.21367e-44 - 2.26453e-43 - 1.56348e-42 - 1.05766e-41 - 7.01025e-41 - 4.55259e-40 - 2.89682e-39 - 1.80601e-38 - 1.1032e-37 - 6.60281e-37 - 3.87203e-36 - 2.22477e-35 - 1.25247e-34 - 6.90859e-34 - 3.73376e-33 - 1.97716e-32 - 1.02582e-31 - 5.21484e-31 - 2.59744e-30 - 1.26762e-29 - 6.06133e-29 - 2.83978e-28 - 1.30358e-27 - 5.86312e-27 - 2.58379e-26 - 1.11564e-25 - 4.71981e-25 - 1.95643e-24 - 7.94588e-24 - 3.16197e-23 - 1.23285e-22 - 4.70977e-22 - 1.7629e-21 - 6.46539e-21 - 2.32327e-20 - 8.17978e-20 - 2.82178e-19 - 9.53769e-19 - 3.15866e-18 - 1.02495e-17 - 3.25866e-17 - 1.01512e-16 - 3.0984e-16 - 9.26612e-16 - 2.71519e-15 - 7.79551e-15 - 2.19297e-14 - 6.04456e-14 - 1.63246e-13 - 4.31982e-13 - 1.12005e-12 - 2.84549e-12 - 7.0832e-12 - 1.72764e-11 - 4.12889e-11 - 9.66876e-11 - 2.21855e-10 - 4.98805e-10 - 1.09891e-09 - 2.37226e-09 - 5.01812e-09 - 1.04016e-08 - 2.11277e-08 - 4.2053e-08 - 8.20252e-08 - 1.56787e-07 - 2.93696e-07 - 5.39164e-07 - 9.70047e-07 - 1.71052e-06 - 2.95629e-06 - 5.00803e-06 - 8.3159e-06 - 1.35363e-05 - 2.16005e-05 - 3.37937e-05 - 5.18379e-05 - 7.7972e-05 - 0.000115015 - 0.000166394 - 0.000236125 - 0.000328721 - 0.00044901 - 0.000601858 - 0.000791802 - 0.00102259 - 0.00129668 - 0.00161472 - 0.00197509 - 0.00237354 - 0.00280296 - 0.00325346 - 0.00371255 - 0.00416568 - 0.00459692 - 0.00498986 - 0.00532857 - 0.00559865 - 0.00578819 - 0.00588854 - 0.00589499 - 0.00580705 - 0.00562858 - 0.00536746 - 0.00503512 - 0.00464575 - 0.0042153 - 0.00376048 - 0.0032977 - 0.00284209 - 0.00240674 - 0.00200212 - 0.0016358 - - - - - 1.1695e-32 - 1.45541e-32 - 1.77463e-32 - 2.12016e-32 - 2.48179e-32 - 2.84642e-32 - 3.19867e-32 - 3.5219e-32 - 3.79945e-32 - 4.01608e-32 - 4.1593e-32 - 4.22061e-32 - 4.1963e-32 - 4.08785e-32 - 3.90176e-32 - 3.6489e-32 - 3.3435e-32 - 3.00177e-32 - 2.64052e-32 - 2.27583e-32 - 1.92188e-32 - 1.59019e-32 - 1.28917e-32 - 1.02401e-32 - 7.96965e-33 - 6.07729e-33 - 4.54064e-33 - 3.32401e-33 - 2.3842e-33 - 1.67556e-33 - 1.15377e-33 - 7.78439e-34 - 5.14739e-34 - 3.34303e-34 - 2.1727e-34 - 1.63241e-34 - 2.51036e-34 - 9.49739e-34 - 4.6911e-33 - 2.36385e-32 - 1.17258e-31 - 5.70211e-31 - 2.71701e-30 - 1.26849e-29 - 5.80254e-29 - 2.60068e-28 - 1.14207e-27 - 4.914e-27 - 2.07164e-26 - 8.55719e-26 - 3.46326e-25 - 1.37333e-24 - 5.33582e-24 - 2.03126e-23 - 7.57645e-23 - 2.76888e-22 - 9.91466e-22 - 3.47848e-21 - 1.19574e-20 - 4.02737e-20 - 1.32906e-19 - 4.29736e-19 - 1.36144e-18 - 4.226e-18 - 1.28528e-17 - 3.83005e-17 - 1.11827e-16 - 3.19909e-16 - 8.96693e-16 - 2.46262e-15 - 6.62654e-15 - 1.74709e-14 - 4.51313e-14 - 1.1423e-13 - 2.8328e-13 - 6.88321e-13 - 1.63871e-12 - 3.82252e-12 - 8.73645e-12 - 1.9564e-11 - 4.29255e-11 - 9.22807e-11 - 1.94376e-10 - 4.01155e-10 - 8.11183e-10 - 1.60717e-09 - 3.1199e-09 - 5.93414e-09 - 1.10589e-08 - 2.0193e-08 - 3.61267e-08 - 6.33275e-08 - 1.08766e-07 - 1.83034e-07 - 3.01791e-07 - 4.87548e-07 - 7.7173e-07 - 1.19688e-06 - 1.81875e-06 - 2.70789e-06 - 3.95027e-06 - 5.64623e-06 - 7.9073e-06 - 1.08501e-05 - 1.45874e-05 - 1.92157e-05 - 2.48013e-05 - 3.13637e-05 - 3.88613e-05 - 4.71786e-05 - 5.61189e-05 - 6.54049e-05 - 7.46875e-05 - 8.35646e-05 - 9.16081e-05 - 9.83971e-05 - 0.000103554 - 0.00010678 - 0.000107882 - 0.000106793 - 0.00010358 - 9.84341e-05 - 9.16541e-05 - 8.36171e-05 - 7.47438e-05 - 6.54624e-05 - 5.61753e-05 - 4.7232e-05 - 3.89102e-05 - 3.14071e-05 - 2.48387e-05 - 1.92471e-05 - 1.46131e-05 - 1.08706e-05 - 7.92321e-06 - 5.6583e-06 - 3.95921e-06 - 2.71436e-06 - 1.82332e-06 - 1.20004e-06 - 7.73865e-07 - 4.88958e-07 - 3.02702e-07 - 1.83609e-07 - 1.09122e-07 - 6.35427e-08 - 3.6254e-08 - 2.02667e-08 - 1.11006e-08 - 5.95729e-09 - 3.13247e-09 - 1.61384e-09 - 8.14654e-10 - 4.02923e-10 - 1.95257e-10 - 9.27106e-11 - 4.31309e-11 - 1.966e-11 - 8.78045e-12 - 3.84226e-12 - 1.64738e-12 - 6.92046e-13 - 2.84844e-13 - 1.14858e-13 - 4.53382e-14 - 1.74196e-14 - 6.23751e-15 - 1.31174e-15 - -2.20244e-15 - -7.77776e-15 - -2.0591e-14 - -5.18126e-14 - -1.27222e-13 - -3.05906e-13 - -7.20636e-13 - -1.66332e-12 - -3.76161e-12 - -8.33502e-12 - -1.80958e-11 - -3.84932e-11 - -8.02285e-11 - -1.63836e-10 - -3.27814e-10 - -6.42661e-10 - -1.23445e-09 - -2.32328e-09 - -4.28417e-09 - -7.7405e-09 - -1.37028e-08 - -2.37675e-08 - -4.03921e-08 - -6.72583e-08 - -1.09732e-07 - -1.7541e-07 - -2.74736e-07 - -4.21611e-07 - -6.33936e-07 - -9.33934e-07 - -1.3481e-06 - -1.90664e-06 - -2.6421e-06 - -3.58729e-06 - -4.77223e-06 - -6.22032e-06 - -7.94404e-06 - -9.94047e-06 - -1.21873e-05 - -1.46402e-05 - -1.72315e-05 - -1.98718e-05 - -2.24536e-05 - -2.48583e-05 - -2.69647e-05 - -2.86586e-05 - -2.98437e-05 - -3.04499e-05 - -3.04409e-05 - -2.9817e-05 - -2.8616e-05 - -2.69085e-05 - -2.47918e-05 - -2.23801e-05 - -1.97949e-05 - -1.71547e-05 - -1.45663e-05 - -1.21185e-05 - -9.87847e-06 - -7.88979e-06 - -6.17417e-06 - -4.734e-06 - -3.55643e-06 - -2.61781e-06 - -1.88798e-06 - -1.33412e-06 - -9.23695e-07 - -6.26613e-07 - -4.16492e-07 - -2.71238e-07 - -1.73074e-07 - -1.08205e-07 - -6.62806e-08 - -3.97758e-08 - -2.33778e-08 - -1.34388e-08 - -7.51457e-09 - -3.99226e-09 - -1.79761e-09 - -1.55833e-10 - 1.62647e-09 - 4.40583e-09 - 9.54109e-09 - 1.94299e-08 - 3.83984e-08 - 7.41803e-08 - 1.40332e-07 - 2.60078e-07 - 4.7225e-07 - 8.40183e-07 - 1.46457e-06 - 2.50142e-06 - 4.18598e-06 - 6.86349e-06 - 1.10263e-05 - 1.73561e-05 - 2.67677e-05 - 4.04488e-05 - 5.98878e-05 - 8.68775e-05 - 0.000123485 - 0.000171971 - 0.000234658 - 0.000313728 - 0.000410966 - 0.000527468 - 0.00066332 - 0.000817311 - 0.000986707 - 0.00116715 - 0.0013527 - 0.00153607 - 0.00170907 - 0.00186314 - 0.00199006 - 0.00208269 - 0.0021356 - 0.00214562 - 0.00211214 - 0.00203718 - 0.00192518 - 0.00178259 - 0.00161722 - 0.00143755 - 0.00125202 - 0.00106841 - 0.000893312 - 0.000731819 - 0.00058741 - 0.000461973 - 0.000355982 - 0.000268767 - 0.00019882 - 0.000144106 - 0.000102339 - 7.12094e-05 - 4.85478e-05 - 3.24294e-05 - 2.12249e-05 - 1.3611e-05 - 8.55202e-06 - 5.26484e-06 - 3.1757e-06 - 1.87685e-06 - 1.08682e-06 - 6.16624e-07 - 3.42785e-07 - 1.86707e-07 - 9.96401e-08 - 5.2101e-08 - 2.66928e-08 - 1.33992e-08 - 6.59024e-09 - 3.17585e-09 - 1.49953e-09 - 6.93727e-10 - 3.14455e-10 - 1.39658e-10 - 6.07726e-11 - 2.59112e-11 - 1.08244e-11 - 4.43055e-12 - 1.77684e-12 - 6.98194e-13 - 2.68807e-13 - 1.01401e-13 - 3.74782e-14 - 1.35723e-14 - 4.81577e-15 - 1.67423e-15 - 5.70295e-16 - 1.90336e-16 - 6.22416e-17 - 1.99424e-17 - 6.2605e-18 - 1.92566e-18 - 5.80343e-19 - 1.71367e-19 - 4.95802e-20 - 1.40548e-20 - 3.90372e-21 - 1.06236e-21 - 2.83268e-22 - 7.4006e-23 - 1.89476e-23 - 4.76861e-24 - 1.24276e-24 - 5.98879e-25 - 1.38172e-24 - 5.3505e-24 - 2.12055e-23 - 8.25555e-23 - 3.14952e-22 - 1.17729e-21 - 4.3118e-21 - 1.54729e-20 - 5.44027e-20 - 1.87416e-19 - 6.32602e-19 - 2.09214e-18 - 6.77934e-18 - 2.15239e-17 - 6.69563e-17 - 2.04079e-16 - 6.09457e-16 - 1.7833e-15 - 5.1126e-15 - 1.43614e-14 - 3.95265e-14 - 1.0659e-13 - 2.81632e-13 - 7.29096e-13 - 1.84937e-12 - 4.59621e-12 - 1.11921e-11 - 2.67031e-11 - 6.24235e-11 - 1.42979e-10 - 3.20872e-10 - 7.05551e-10 - 1.52007e-09 - 3.20873e-09 - 6.63652e-09 - 1.34488e-08 - 2.67034e-08 - 5.19498e-08 - 9.90237e-08 - 1.8494e-07 - 3.38423e-07 - 6.06771e-07 - 1.06593e-06 - 1.8347e-06 - 3.09416e-06 - 5.11276e-06 - 8.27761e-06 - 1.31308e-05 - 2.04087e-05 - 3.10796e-05 - 4.63738e-05 - 6.77965e-05 - 9.71133e-05 - 0.000136297 - 0.000187428 - 0.000252533 - 0.00033338 - 0.000431223 - 0.000546517 - 0.000678654 - 0.00082573 - 0.000984414 - 0.00114994 - 0.00131625 - 0.00147635 - 0.00162275 - 0.00174809 - 0.00184582 - 0.00191084 - 0.00194004 - 0.00193274 - 0.00189081 - 0.00181863 - 0.00172275 - 0.00161135 - 0.00149356 - 0.00137872 - 0.00127559 - 0.00119165 - 0.00113253 - 0.00110163 - 0.00109985 - 0.00112561 - 0.001175 - 0.00124218 - 0.00131983 - 0.00139982 - 0.00147384 - 0.00153416 - 0.00157416 - 0.00158894 - 0.00157562 - 0.00153351 - 0.00146403 - 0.00137047 - 0.00125758 - 0.00113103 - 0.000996861 - 0.000860974 - 0.000728651 - 0.000604241 - 0.000490968 - 0.000390879 - 0.000304913 - 0.000233051 - 0.000174528 - 0.000128061 - 9.20676e-05 - 6.48537e-05 - 4.47611e-05 - 3.02693e-05 - 2.00559e-05 - 1.30202e-05 - 8.28195e-06 - 5.16159e-06 - 3.15189e-06 - 1.8858e-06 - 1.10549e-06 - 6.34971e-07 - 3.57346e-07 - 1.97043e-07 - 1.06456e-07 - 5.63526e-08 - 2.92278e-08 - 1.4853e-08 - 7.39555e-09 - 3.60797e-09 - 1.72462e-09 - 8.07717e-10 - 3.70649e-10 - 1.66649e-10 - 7.34141e-11 - 3.16879e-11 - 1.34012e-11 - 5.55305e-12 - 2.25453e-12 - 8.96844e-13 - 3.49555e-13 - 1.3349e-13 - 4.99484e-14 - 1.83118e-14 - 6.57773e-15 - 2.31504e-15 - 7.98324e-16 - 2.69746e-16 - 8.9342e-17 - 2.9128e-17 - 9.76341e-18 - 4.72724e-18 - 6.95038e-18 - 1.96983e-17 - 6.09995e-17 - 1.87121e-16 - 5.62997e-16 - 1.65984e-15 - 4.79478e-15 - 1.35709e-14 - 3.76342e-14 - 1.02258e-13 - 2.72236e-13 - 7.1012e-13 - 1.81491e-12 - 4.5448e-12 - 1.11509e-11 - 2.68068e-11 - 6.31417e-11 - 1.45721e-10 - 3.29509e-10 - 7.30044e-10 - 1.58477e-09 - 3.37071e-09 - 7.02447e-09 - 1.43431e-08 - 2.86951e-08 - 5.62483e-08 - 1.08031e-07 - 2.03294e-07 - 3.74832e-07 - 6.77153e-07 - 1.1986e-06 - 2.07872e-06 - 3.53229e-06 - 5.88103e-06 - 9.59373e-06 - 1.53341e-05 - 2.40141e-05 - 3.68477e-05 - 5.53976e-05 - 8.16036e-05 - 0.000117778 - 0.000166554 - 0.000230773 - 0.000313293 - 0.000416728 - 0.000543115 - 0.000693534 - 0.000867722 - 0.00106373 - 0.00127766 - 0.00150363 - 0.0017338 - 0.00195883 - 0.00216836 - 0.00235181 - 0.00249925 - 0.00260228 - 0.00265482 - 0.00265371 - 0.00259901 - 0.00249402 - 0.00234493 - 0.0021602 - 0.00194983 - 0.00172439 - 0.00149421 - 0.0012686 - 0.0010553 - 0.000860126 - 0.000686887 - 0.000537459 - 0.000412043 - 0.000309512 - 0.000227797 - 0.000164269 - 0.000116065 - 8.03492e-05 - 5.45004e-05 - 3.62205e-05 - 2.35856e-05 - 1.50479e-05 - 9.40679e-06 - 5.76161e-06 - 3.45767e-06 - 2.0331e-06 - 1.17131e-06 - 6.61184e-07 - 3.65687e-07 - 1.98168e-07 - 1.05219e-07 - 5.47382e-08 - 2.79013e-08 - 1.39346e-08 - 6.81873e-09 - 3.26925e-09 - 1.53578e-09 - 7.06883e-10 - 3.18789e-10 - 1.40863e-10 - 6.09863e-11 - 2.58729e-11 - 1.07614e-11 - 4.4044e-12 - 1.81685e-12 - 8.67714e-13 - 7.40664e-13 - 1.31546e-12 - 3.06677e-12 - 7.40609e-12 - 1.76751e-11 - 4.13834e-11 - 9.49532e-11 - 2.13472e-10 - 4.70233e-10 - 1.01489e-09 - 2.14618e-09 - 4.4468e-09 - 9.02747e-09 - 1.79565e-08 - 3.49956e-08 - 6.68256e-08 - 1.25028e-07 - 2.29198e-07 - 4.11672e-07 - 7.24482e-07 - 1.24923e-06 - 2.11053e-06 - 3.49364e-06 - 5.66633e-06 - 9.00457e-06 - 1.40204e-05 - 2.13892e-05 - 3.19717e-05 - 4.68246e-05 - 6.71923e-05 - 9.44715e-05 - 0.000130143 - 0.000175661 - 0.00023231 - 0.000301021 - 0.000382176 - 0.000475407 - 0.000579436 - 0.000691962 - 0.000809647 - 0.000928209 - 0.00104264 - 0.00114751 - 0.00123742 - 0.00130742 - 0.00135347 - 0.00137284 - 0.00136436 - 0.00132854 - 0.00126753 - 0.00118488 - 0.00108525 - 0.00097392 - 0.000856353 - 0.000737767 - 0.000622762 - 0.000515065 - 0.000417387 - 0.0003314 - 0.000257812 - 0.000196512 - 0.000146762 - 0.000107393 - 7.69968e-05 - 5.40888e-05 - 3.72288e-05 - 2.51065e-05 - 1.65895e-05 - 1.07402e-05 - 6.81292e-06 - 4.23437e-06 - 2.57858e-06 - 1.53855e-06 - 8.9945e-07 - 5.15205e-07 - 2.89148e-07 - 1.59e-07 - 8.56664e-08 - 4.52232e-08 - 2.3391e-08 - 1.18542e-08 - 5.88618e-09 - 2.86373e-09 - 1.36511e-09 - 6.37584e-10 - 2.91774e-10 - 1.30825e-10 - 5.74743e-11 - 2.47396e-11 - 1.04339e-11 - 4.31162e-12 - 1.7457e-12 - 6.92527e-13 - 2.69178e-13 - 1.02513e-13 - 3.82523e-14 - 1.39853e-14 - 5.00981e-15 - 1.75836e-15 - 6.0469e-16 - 2.03748e-16 - 6.72655e-17 - 2.17584e-17 - 6.89602e-18 - 2.14145e-18 - 6.51557e-19 - 1.94238e-19 - 5.67355e-20 - 1.62376e-20 - 4.5551e-21 - 1.25945e-21 - 3.71542e-22 - 2.27974e-22 - 5.68172e-22 - 2.09885e-21 - 7.8741e-21 - 2.90139e-20 - 1.04766e-19 - 3.70658e-19 - 1.28488e-18 - 4.36407e-18 - 1.4523e-17 - 4.73541e-17 - 1.51285e-16 - 4.73556e-16 - 1.45239e-15 - 4.36447e-15 - 1.28504e-14 - 3.70715e-14 - 1.04785e-13 - 2.90199e-13 - 7.8746e-13 - 2.09362e-12 - 5.45388e-12 - 1.39203e-11 - 3.4812e-11 - 8.52996e-11 - 2.04786e-10 - 4.81717e-10 - 1.11025e-09 - 2.50718e-09 - 5.54737e-09 - 1.20261e-08 - 2.55447e-08 - 5.31635e-08 - 1.08408e-07 - 2.16596e-07 - 4.24008e-07 - 8.1327e-07 - 1.52839e-06 - 2.81429e-06 - 5.07741e-06 - 8.97539e-06 - 1.55454e-05 - 2.6381e-05 - 4.38651e-05 - 7.1464e-05 - 0.000114077 - 0.000178422 - 0.000273429 - 0.00041057 - 0.000604061 - 0.00087082 - 0.00123009 - 0.00170259 - 0.00230921 - 0.00306905 - 0.00399717 - 0.00510188 - 0.00638213 - 0.00782524 - 0.00940538 - 0.0110833 - 0.0128073 - 0.0145165 - 0.0161444 - 0.017625 - 0.0188984 - 0.0199167 - 0.0206486 - 0.0210826 - 0.0212274 - 0.0211103 - 0.0207729 - 0.0202659 - 0.0196419 - 0.0189493 - 0.0182273 - 0.0175019 - 0.0167852 - 0.0160761 - 0.015364 - 0.0146325 - 0.0138649 - 0.0130493 - 0.0121818 - 0.0112696 - 0.0103324 - 0.00940127 - 0.00851798 - 0.0077322 - 0.00709887 - 0.00667521 - 0.00651766 - 0.00667869 - 0.00720359 - 0.00812682 - 0.00946821 - 0.011229 - 0.0133879 - 0.0158984 - 0.0186868 - 0.0216531 - 0.0246735 - 0.0276071 - 0.0303043 - 0.0326177 - 0.0344137 - 0.0355844 - 0.0360574 - 0.0358021 - 0.0348328 - 0.033207 - 0.0310196 - 0.0283937 - 0.0254689 - 0.0223898 - 0.0192943 - 0.0163042 - 0.0135187 - 0.0110106 - 0.00882576 - 0.0069851 - 0.00548818 - 0.00431784 - 0.00344511 - 0.00283369 - 0.00244377 - 0.00223482 - 0.00216753 - 0.00220503 - 0.00231352 - 0.00246261 - 0.00262555 - 0.00277942 - 0.00290547 - 0.00298931 - 0.00302122 - 0.00299617 - 0.00291371 - 0.00277759 - 0.00259503 - 0.00237585 - 0.00213141 - 0.00187359 - 0.00161373 - 0.00136185 - 0.00112608 - 0.000912319 - 0.000724208 - 0.000563271 - 0.000429248 - 0.000320507 - 0.000234478 - 0.000168076 - 0.000118044 - 8.12306e-05 - 5.47687e-05 - 3.61811e-05 - 2.3419e-05 - 1.48522e-05 - 9.22895e-06 - 5.61887e-06 - 3.35184e-06 - 1.95909e-06 - 1.12192e-06 - 6.29515e-07 - 3.46089e-07 - 1.86426e-07 - 9.83922e-08 - 5.08806e-08 - 2.57799e-08 - 1.27981e-08 - 6.22514e-09 - 2.9668e-09 - 1.38537e-09 - 6.33837e-10 - 2.84137e-10 - 1.248e-10 - 5.37078e-11 - 2.26463e-11 - 9.3561e-12 - 3.78729e-12 - 1.5021e-12 - 5.83724e-13 - 2.22255e-13 - 8.29153e-14 - 3.03084e-14 - 1.08573e-14 - 3.81839e-15 - 1.34071e-15 - 5.41182e-16 - 4.60363e-16 - 1.01752e-15 - 2.94646e-15 - 8.68155e-15 - 2.5165e-14 - 7.1502e-14 - 1.99065e-13 - 5.43016e-13 - 1.45133e-12 - 3.80064e-12 - 9.75179e-12 - 2.4516e-11 - 6.03879e-11 - 1.45743e-10 - 3.44638e-10 - 7.985e-10 - 1.81269e-09 - 4.03189e-09 - 8.78681e-09 - 1.87625e-08 - 3.92542e-08 - 8.04673e-08 - 1.61618e-07 - 3.1805e-07 - 6.13252e-07 - 1.15856e-06 - 2.14455e-06 - 3.88947e-06 - 6.91164e-06 - 1.2034e-05 - 2.05292e-05 - 3.43142e-05 - 5.61969e-05 - 9.01753e-05 - 0.000141775 - 0.000218398 - 0.000329635 - 0.000487479 - 0.000706342 - 0.00100279 - 0.0013949 - 0.00190114 - 0.00253875 - 0.00332173 - 0.00425838 - 0.00534887 - 0.00658288 - 0.00793792 - 0.00937852 - 0.0108567 - 0.012314 - 0.0136848 - 0.0149009 - 0.0158974 - 0.0166178 - 0.01702 - 0.0170798 - 0.0167935 - 0.0161785 - 0.0152711 - 0.0141235 - 0.0127982 - 0.011363 - 0.00988489 - 0.00842537 - 0.00703628 - 0.0057575 - 0.00461596 - 0.00362599 - 0.0027908 - 0.00210459 - 0.00155504 - 0.00112578 - 0.000798551 - 0.000554994 - 0.00037793 - 0.000252157 - 0.000164842 - 0.000105584 - 6.62628e-05 - 4.07452e-05 - 2.45482e-05 - 1.44911e-05 - 8.38143e-06 - 4.74977e-06 - 2.63732e-06 - 1.4348e-06 - 7.64815e-07 - 3.99446e-07 - 2.04407e-07 - 1.02488e-07 - 5.03482e-08 - 2.42344e-08 - 1.14293e-08 - 5.28131e-09 - 2.39112e-09 - 1.06071e-09 - 4.6104e-10 - 1.96362e-10 - 8.19975e-11 - 3.37003e-11 - 1.39853e-11 - 6.7944e-12 - 6.04916e-12 - 1.11421e-11 - 2.65104e-11 - 6.50445e-11 - 1.57586e-10 - 3.74505e-10 - 8.72182e-10 - 1.99023e-09 - 4.44979e-09 - 9.74793e-09 - 2.09229e-08 - 4.40016e-08 - 9.06677e-08 - 1.83051e-07 - 3.62102e-07 - 7.01821e-07 - 1.33278e-06 - 2.47987e-06 - 4.52101e-06 - 8.07572e-06 - 1.4134e-05 - 2.42374e-05 - 4.07237e-05 - 6.70422e-05 - 0.000108141 - 0.000170912 - 0.000264666 - 0.000401576 - 0.000597016 - 0.000869669 - 0.0012413 - 0.00173603 - 0.00237906 - 0.00319468 - 0.00420373 - 0.00542055 - 0.00684968 - 0.00848282 - 0.0102964 - 0.0122502 - 0.0142879 - 0.016339 - 0.0183232 - 0.020156 - 0.0217557 - 0.0230507 - 0.0239861 - 0.0245286 - 0.0246698 - 0.0244259 - 0.0238353 - 0.0229533 - 0.0218461 - 0.020583 - 0.0192297 - 0.017843 - 0.0164669 - 0.0151311 - 0.0138521 - 0.0126351 - 0.0114777 - 0.0103735 - 0.00931555 - 0.00829889 - 0.00732211 - 0.00638771 - 0.00550152 - 0.00467147 - 0.00390611 - 0.00321307 - 0.0025979 - 0.00206328 - 0.00160877 - 0.00123095 - 0.000923967 - 0.00068019 - 0.000491014 - 0.000347568 - 0.000241316 - 0.000164498 - 0.000110414 - 7.35521e-05 - 4.96251e-05 - 3.55275e-05 - 2.92725e-05 - 2.99299e-05 - 3.75905e-05 - 5.33657e-05 - 7.9422e-05 - 0.000119033 - 0.000176624 - 0.000257776 - 0.000369135 - 0.000518189 - 0.000712868 - 0.000960941 - 0.00126921 - 0.00164251 - 0.00208268 - 0.00258746 - 0.00314965 - 0.00375654 - 0.00438986 - 0.00502632 - 0.0056388 - 0.00619812 - 0.00667528 - 0.00704395 - 0.00728283 - 0.00737769 - 0.00732281 - 0.00712151 - 0.00678583 - 0.00633536 - 0.0057953 - 0.00519418 - 0.00456137 - 0.00392474 - 0.00330874 - 0.00273308 - 0.00221196 - 0.00175404 - 0.00136282 - 0.00103747 - 0.000773837 - 0.000565535 - 0.000404955 - 0.000284112 - 0.000195304 - 0.000131543 - 8.68085e-05 - 5.61298e-05 - 3.556e-05 - 2.20733e-05 - 1.34248e-05 - 7.99994e-06 - 4.67091e-06 - 2.67211e-06 - 1.49776e-06 - 8.22563e-07 - 4.42621e-07 - 2.33363e-07 - 1.2055e-07 - 6.10156e-08 - 3.02588e-08 - 1.47027e-08 - 6.99974e-09 - 3.26515e-09 - 1.49231e-09 - 6.68274e-10 - 2.93215e-10 - 1.26053e-10 - 5.30957e-11 - 2.19129e-11 - 8.86093e-12 - 3.51071e-12 - 1.36285e-12 - 5.18366e-13 - 1.9318e-13 - 7.05383e-14 - 2.52363e-14 - 8.84629e-15 - 3.03833e-15 - 1.02246e-15 - 3.37126e-16 - 1.08912e-16 - 3.44744e-17 - 1.06919e-17 - 3.24899e-18 - 9.67342e-19 - 2.82194e-19 - 8.06591e-20 - 2.25889e-20 - 6.19832e-21 - 1.66644e-21 - 4.38977e-22 - 1.133e-22 - 2.86522e-23 - 7.09938e-24 - 1.72354e-24 - 4.09974e-25 - 9.555e-26 - 2.18194e-26 - 4.88199e-27 - 1.07034e-27 - 2.30029e-28 - 4.85607e-29 - 1.01878e-29 - 2.25705e-30 - 6.68573e-31 - 3.72194e-31 - 3.30695e-31 - 3.35269e-31 - 3.47158e-31 - 3.74704e-31 - 5.04446e-31 - 1.18957e-30 - 4.71178e-30 - 2.23171e-29 - 1.083e-28 - 5.19393e-28 - 2.44457e-27 - 1.12765e-26 - 5.09695e-26 - 2.25728e-25 - 9.79487e-25 - 4.16437e-24 - 1.73475e-23 - 7.08044e-23 - 2.83153e-22 - 1.10948e-21 - 4.25944e-21 - 1.60223e-20 - 5.90517e-20 - 2.13244e-19 - 7.545e-19 - 2.61564e-18 - 8.88451e-18 - 2.95683e-17 - 9.64174e-17 - 3.0805e-16 - 9.64327e-16 - 2.95777e-15 - 8.88875e-15 - 2.6173e-14 - 7.551e-14 - 2.13448e-13 - 5.91175e-13 - 1.60427e-12 - 4.26554e-12 - 1.11124e-11 - 2.83648e-11 - 7.09396e-11 - 1.73834e-10 - 4.17365e-10 - 9.81826e-10 - 2.26303e-09 - 5.11073e-09 - 1.13087e-08 - 2.45176e-08 - 5.20812e-08 - 1.08398e-07 - 2.21054e-07 - 4.41684e-07 - 8.64693e-07 - 1.65863e-06 - 3.11726e-06 - 5.7403e-06 - 1.0357e-05 - 1.83091e-05 - 3.17131e-05 - 5.38204e-05 - 8.94936e-05 - 0.000145806 - 0.000232752 - 0.00036404 - 0.000557881 - 0.000837666 - 0.00123236 - 0.0017764 - 0.00250888 - 0.00347182 - 0.00470729 - 0.00625347 - 0.0081397 - 0.0103808 - 0.0129716 - 0.0158815 - 0.0190514 - 0.0223923 - 0.0257874 - 0.0290973 - 0.0321688 - 0.0348462 - 0.0369839 - 0.03846 - 0.0391873 - 0.0391222 - 0.0382691 - 0.0366799 - 0.0344491 - 0.0317048 - 0.0285972 - 0.0252851 - 0.0219241 - 0.0186558 - 0.0155996 - 0.0128486 - 0.0104682 - 0.00849752 - 0.00695295 - 0.00583265 - 0.00512109 - 0.00479313 - 0.00481689 - 0.00515543 - 0.0057673 - 0.00660627 - 0.00762079 - 0.00875373 - 0.00994277 - 0.0111219 - 0.0122239 - 0.0131839 - 0.0139439 - 0.0144563 - 0.0146885 - 0.0146249 - 0.0142685 - 0.0136401 - 0.0127762 - 0.0117254 - 0.0105438 - 0.00928966 - 0.00801941 - 0.00678301 - 0.00562134 - 0.0045645 - 0.00363149 - 0.00283082 - 0.00216211 - 0.001618 - 0.00118636 - 0.000852305 - 0.00059995 - 0.000413799 - 0.000279676 - 0.000185285 - 0.000120433 - 7.70311e-05 - 4.89435e-05 - 3.17797e-05 - 2.27018e-05 - 2.03111e-05 - 2.46741e-05 - 3.75481e-05 - 6.28712e-05 - 0.000107589 - 0.000182895 - 0.000305951 - 0.00050214 - 0.000807822 - 0.0012735 - 0.00196715 - 0.00297727 - 0.00441506 - 0.00641494 - 0.00913241 - 0.0127384 - 0.0174093 - 0.0233123 - 0.0305863 - 0.0393191 - 0.0495243 - 0.061118 - 0.0739021 - 0.0875551 - 0.101635 - 0.115596 - 0.128818 - 0.140653 - 0.150473 - 0.157726 - 0.161989 - 0.163007 - 0.160717 - 0.155259 - 0.146955 - 0.136286 - 0.123839 - 0.110255 - 0.0961777 - 0.0822032 - 0.0688398 - 0.0564843 - 0.0454101 - 0.0357696 - 0.0276066 - 0.020876 - 0.0154675 - 0.0112287 - 0.00798685 - 0.00556619 - 0.00380082 - 0.00254293 - 0.00166697 - 0.00107067 - 0.00067379 - 0.000415459 - 0.000250997 - 0.000148575 - 8.6171e-05 - 4.8968e-05 - 2.72647e-05 - 1.48739e-05 - 7.95037e-06 - 4.16376e-06 - 2.13659e-06 - 1.07422e-06 - 5.29179e-07 - 2.55416e-07 - 1.2079e-07 - 5.59695e-08 - 2.54102e-08 - 1.13032e-08 - 4.92642e-09 - 2.10377e-09 - 8.80244e-10 - 3.60866e-10 - 1.44959e-10 - 5.70702e-11 - 2.20651e-11 - 8.50316e-12 - 3.61538e-12 - 2.59818e-12 - 4.39325e-12 - 1.06972e-11 - 2.73806e-11 - 6.93422e-11 - 1.72287e-10 - 4.19489e-10 - 1.00077e-09 - 2.3393e-09 - 5.35766e-09 - 1.20227e-08 - 2.6434e-08 - 5.6946e-08 - 1.20199e-07 - 2.48584e-07 - 5.03712e-07 - 1.00007e-06 - 1.94542e-06 - 3.70795e-06 - 6.92454e-06 - 1.26703e-05 - 2.27152e-05 - 3.99011e-05 - 6.86735e-05 - 0.000115806 - 0.000191342 - 0.000309759 - 0.000491334 - 0.0007636 - 0.00116276 - 0.00173482 - 0.00253603 - 0.00363237 - 0.00509757 - 0.00700928 - 0.00944323 - 0.0124653 - 0.0161222 - 0.0204307 - 0.0253675 - 0.0308609 - 0.0367855 - 0.0429617 - 0.0491612 - 0.055119 - 0.0605504 - 0.0651732 - 0.0687319 - 0.0710206 - 0.071903 - 0.0713258 - 0.0693239 - 0.0660171 - 0.061598 - 0.0563137 - 0.0504426 - 0.0442709 - 0.0380694 - 0.0320753 - 0.026479 - 0.0214176 - 0.0169736 - 0.01318 - 0.0100276 - 0.00747499 - 0.00545963 - 0.00390708 - 0.00273955 - 0.00188209 - 0.0012669 - 0.00083556 - 0.000539947 - 0.00034187 - 0.000212084 - 0.000128912 - 7.67739e-05 - 4.47993e-05 - 2.56133e-05 - 1.43482e-05 - 7.87527e-06 - 4.23517e-06 - 2.23158e-06 - 1.1521e-06 - 5.82783e-07 - 2.88841e-07 - 1.40265e-07 - 6.67383e-08 - 3.11127e-08 - 1.42114e-08 - 6.36025e-09 - 2.789e-09 - 1.19828e-09 - 5.04435e-10 - 2.08061e-10 - 8.40835e-11 - 3.32942e-11 - 1.29171e-11 - 4.91018e-12 - 1.8289e-12 - 6.6774e-13 - 2.39797e-13 - 8.72764e-14 - 3.99832e-14 - 4.35828e-14 - 1.03126e-13 - 2.89491e-13 - 8.16136e-13 - 2.26089e-12 - 6.1391e-12 - 1.6335e-11 - 4.25903e-11 - 1.08812e-10 - 2.72406e-10 - 6.68242e-10 - 1.6063e-09 - 3.78352e-09 - 8.7326e-09 - 1.97501e-08 - 4.37696e-08 - 9.50509e-08 - 2.02264e-07 - 4.21756e-07 - 8.61758e-07 - 1.7254e-06 - 3.38515e-06 - 6.50799e-06 - 1.22603e-05 - 2.26326e-05 - 4.09407e-05 - 7.25704e-05 - 0.000126052 - 0.000214548 - 0.000357837 - 0.000584835 - 0.000936631 - 0.00146991 - 0.00226049 - 0.00340645 - 0.00503026 - 0.00727894 - 0.0103214 - 0.0143415 - 0.0195274 - 0.0260547 - 0.0340659 - 0.0436462 - 0.0547983 - 0.067419 - 0.0812818 - 0.0960292 - 0.111177 - 0.126134 - 0.140238 - 0.1528 - 0.163165 - 0.170766 - 0.175185 - 0.17619 - 0.173768 - 0.168128 - 0.159688 - 0.149039 - 0.136897 - 0.124043 - 0.111262 - 0.0992842 - 0.0887352 - 0.0800981 - 0.073689 - 0.0696463 - 0.0679332 - 0.0683505 - 0.0705597 - 0.0741133 - 0.0784909 - 0.0831388 - 0.0875109 - 0.091108 - 0.0935133 - 0.09442 - 0.0936496 - 0.0911582 - 0.0870312 - 0.0814664 - 0.0747481 - 0.0672163 - 0.0592323 - 0.0511477 - 0.0432772 - 0.0358796 - 0.0291464 - 0.0231988 - 0.0180921 - 0.0138246 - 0.0103503 - 0.00759265 - 0.00545721 - 0.00384313 - 0.00265178 - 0.00179277 - 0.00118755 - 0.000770751 - 0.000490133 - 0.000305387 - 0.000186433 - 0.000111515 - 6.53553e-05 - 3.75288e-05 - 2.11147e-05 - 1.16397e-05 - 6.28691e-06 - 3.32712e-06 - 1.7252e-06 - 8.76506e-07 - 4.36367e-07 - 2.12967e-07 - 1.02108e-07 - 4.86122e-08 - 2.41809e-08 - 1.51788e-08 - 1.63679e-08 - 2.83794e-08 - 5.79776e-08 - 1.21189e-07 - 2.50402e-07 - 5.07817e-07 - 1.0094e-06 - 1.966e-06 - 3.75186e-06 - 7.01533e-06 - 1.28525e-05 - 2.30708e-05 - 4.05764e-05 - 6.99235e-05 - 0.000118062 - 0.000195313 - 0.000316586 - 0.000502791 - 0.000782386 - 0.00119286 - 0.00178196 - 0.00260821 - 0.00374044 - 0.00525582 - 0.00723593 - 0.00976081 - 0.0129007 - 0.0167063 - 0.0211974 - 0.0263525 - 0.0320995 - 0.0383101 - 0.044799 - 0.051329 - 0.0576237 - 0.0633852 - 0.0683175 - 0.0721517 - 0.0746721 - 0.0757375 - 0.0752976 - 0.0734009 - 0.0701941 - 0.0659136 - 0.0608694 - 0.0554259 - 0.0499808 - 0.0449442 - 0.0407215 - 0.0376986 - 0.0362307 - 0.0366324 - 0.0391679 - 0.0440395 - 0.0513723 - 0.0611977 - 0.0734333 - 0.0878657 - 0.104137 - 0.121742 - 0.140036 - 0.158258 - 0.175572 - 0.191121 - 0.204086 - 0.213755 - 0.219579 - 0.221225 - 0.218598 - 0.211856 - 0.201386 - 0.187772 - 0.171739 - 0.15409 - 0.13564 - 0.117162 - 0.0993331 - 0.0827063 - 0.0676928 - 0.0545624 - 0.043456 - 0.034407 - 0.0273684 - 0.022239 - 0.0188874 - 0.0171699 - 0.0169424 - 0.0180635 - 0.0203933 - 0.0237859 - 0.0280809 - 0.0330942 - 0.0386122 - 0.0443899 - 0.0501552 - 0.0556197 - 0.0604946 - 0.06451 - 0.0674365 - 0.069104 - 0.069417 - 0.0683625 - 0.0660108 - 0.0625073 - 0.0580577 - 0.0529082 - 0.0473225 - 0.0415602 - 0.0358572 - 0.0304111 - 0.0253728 - 0.0208431 - 0.0168751 - 0.0134805 - 0.0106382 - 0.00830404 - 0.00641967 - 0.00492092 - 0.00374385 - 0.00282905 - 0.0021241 - 0.00158457 - 0.00117404 - 0.000863316 - 0.000629412 - 0.000454431 - 0.000324503 - 0.000228896 - 0.000159295 - 0.000109251 - 7.37695e-05 - 4.89991e-05 - 3.19922e-05 - 2.05207e-05 - 1.2926e-05 - 7.99585e-06 - 4.86315e-06 - 2.92334e-06 - 1.76948e-06 - 1.1439e-06 - 9.06437e-07 - 1.02009e-06 - 1.55632e-06 - 2.72467e-06 - 4.93405e-06 - 8.89688e-06 - 1.57908e-05 - 2.74957e-05 - 4.69267e-05 - 7.84794e-05 - 0.0001286 - 0.000206474 - 0.00032481 - 0.000500644 - 0.000756078 - 0.00111877 - 0.001622 - 0.00230409 - 0.00320688 - 0.00437325 - 0.00584336 - 0.00764994 - 0.00981273 - 0.0123327 - 0.0151867 - 0.0183234 - 0.0216614 - 0.0250901 - 0.0284744 - 0.0316625 - 0.0344962 - 0.0368244 - 0.0385155 - 0.0394705 - 0.0396321 - 0.0389905 - 0.0375844 - 0.0354972 - 0.032849 - 0.0297846 - 0.0264615 - 0.0230361 - 0.0196523 - 0.0164334 - 0.0134759 - 0.0108485 - 0.00859408 - 0.00673434 - 0.00527724 - 0.00422554 - 0.0035862 - 0.00337974 - 0.00364906 - 0.0044669 - 0.00594138 - 0.00821867 - 0.0114817 - 0.0159437 - 0.0218363 - 0.0293904 - 0.0388107 - 0.0502452 - 0.0637519 - 0.0792647 - 0.0965666 - 0.115271 - 0.134821 - 0.154501 - 0.173478 - 0.190851 - 0.205722 - 0.217273 - 0.224836 - 0.227963 - 0.226463 - 0.220429 - 0.210222 - 0.196437 - 0.179848 - 0.161335 - 0.141805 - 0.122123 - 0.103052 - 0.0852089 - 0.0690443 - 0.0548367 - 0.0427079 - 0.0326479 - 0.0245476 - 0.0182336 - 0.0135014 - 0.0101435 - 0.00797117 - 0.00682901 - 0.00660242 - 0.0072185 - 0.00864139 - 0.0108629 - 0.0138899 - 0.0177288 - 0.0223696 - 0.0277703 - 0.0338438 - 0.0404482 - 0.0473843 - 0.0543987 - 0.0611951 - 0.0674527 - 0.0728495 - 0.0770893 - 0.0799283 - 0.0811978 - 0.0808212 - 0.0788213 - 0.075318 - 0.0705165 - 0.0646873 - 0.0581413 - 0.0512021 - 0.0441801 - 0.0373511 - 0.0309397 - 0.0251111 - 0.0199688 - 0.0155588 - 0.0118778 - 0.00888452 - 0.0065113 - 0.00467562 - 0.00328963 - 0.00226773 - 0.0015317 - 0.00101366 - 0.000657274 - 0.000417579 - 0.000259937 - 0.000158538 - 9.47405e-05 - 5.54722e-05 - 3.18238e-05 - 1.78881e-05 - 9.85178e-06 - 5.3162e-06 - 2.81077e-06 - 1.45608e-06 - 7.39066e-07 - 3.67551e-07 - 1.79097e-07 - 8.55059e-08 - 3.99982e-08 - 1.83325e-08 - 8.23268e-09 - 3.6224e-09 - 1.56167e-09 - 6.59656e-10 - 2.73013e-10 - 1.1071e-10 - 4.39871e-11 - 1.71239e-11 - 6.53153e-12 - 2.44099e-12 - 8.93824e-13 - 3.20683e-13 - 1.12729e-13 - 3.8827e-14 - 1.31029e-14 - 4.33251e-15 - 1.40361e-15 - 4.45546e-16 - 1.38572e-16 - 4.22272e-17 - 1.26081e-17 - 3.68842e-18 - 1.05723e-18 - 2.96917e-19 - 8.1703e-20 - 2.20281e-20 - 5.81908e-21 - 1.50615e-21 - 3.8196e-22 - 9.49084e-23 - 2.31062e-23 - 5.51175e-24 - 1.28821e-24 - 2.94999e-25 - 6.619e-26 - 1.45513e-26 - 3.13434e-27 - 6.61496e-28 - 1.36787e-28 - 2.77142e-29 - 5.50166e-30 - 1.0701e-30 - 2.03934e-31 - 3.80796e-32 - 6.96678e-33 - 1.24885e-33 - 2.19343e-34 - 3.77464e-35 - 6.36453e-36 - 1.05147e-36 - 1.70205e-37 - 2.69958e-38 - 4.1955e-39 - 6.38942e-40 - 9.53634e-41 - 1.39524e-41 - 2.00202e-42 - 2.82e-43 - 3.90655e-44 - 5.34142e-45 - 7.25747e-46 - 9.91944e-47 - 1.39154e-47 - 2.06029e-48 - 3.31157e-49 - 5.85717e-50 - 1.12792e-50 - 2.30138e-51 - 4.83517e-52 - 1.02406e-52 - 2.15792e-53 - 4.49107e-54 - 9.19553e-55 - 1.84859e-55 - 3.64499e-56 - 7.04559e-57 - 1.33472e-57 - 2.47776e-58 - 4.50705e-59 - 8.033e-60 - 1.40284e-60 - 2.40037e-61 - 4.02427e-62 - 6.61049e-63 - 1.06394e-63 - 1.67779e-64 - 2.59237e-65 - 3.92457e-66 - 5.82135e-67 - 8.46043e-68 - 1.20475e-68 - 1.6809e-69 - 2.29784e-70 - 3.07777e-71 - 4.03914e-72 - 5.19372e-73 - 6.54343e-74 - 8.07735e-75 - 9.76944e-76 - 1.15773e-76 - 1.34425e-77 - 1.5293e-78 - 1.70467e-79 - 1.86176e-80 - 1.99226e-81 - 2.08883e-82 - 2.14585e-83 - 2.15989e-84 - 2.1301e-85 - 2.05829e-86 - 1.94871e-87 - 1.80771e-88 - 1.64302e-89 - 1.46318e-90 - 1.27669e-91 - 1.09147e-92 - 9.14276e-94 - 7.50374e-95 - 6.03414e-96 - 4.75433e-97 - 3.67029e-98 - 2.77618e-99 - 2.05747e-100 - 1.49401e-101 - 1.06295e-102 - 7.40983e-104 - 5.06105e-105 - 3.38695e-106 - 2.22083e-107 - 1.42678e-108 - 8.98126e-110 - 5.53928e-111 - 3.34739e-112 - 1.98197e-113 - 1.1498e-114 - 6.53563e-116 - 3.63989e-117 - 1.98621e-118 - 1.06194e-119 - 5.56302e-121 - 2.85535e-122 - 1.43596e-123 - 7.07564e-125 - 3.41605e-126 - 1.61592e-127 - 7.48948e-129 - 3.40111e-130 - 1.51331e-131 - 6.59735e-133 - 2.81805e-134 - 1.17951e-135 - 5.09391e-137 - 6.49642e-137 - 1.51104e-135 - 3.54941e-134 - 8.16953e-133 - 1.84236e-131 - 4.07088e-130 - 8.81332e-129 - 1.86951e-127 - 3.88555e-126 - 7.9125e-125 - 1.57875e-123 - 3.08636e-122 - 5.9118e-121 - 1.1095e-119 - 2.04021e-118 - 3.67584e-117 - 6.48898e-116 - 1.12236e-114 - 1.90207e-113 - 3.15832e-112 - 5.13834e-111 - 8.19081e-110 - 1.27928e-108 - 1.95769e-107 - 2.93534e-106 - 4.31231e-105 - 6.20723e-104 - 8.75433e-103 - 1.20972e-101 - 1.63788e-100 - 2.17279e-99 - 2.82417e-98 - 3.59666e-97 - 4.48793e-96 - 5.48692e-95 - 6.57277e-94 - 7.71445e-93 - 8.87152e-92 - 9.99605e-91 - 1.10356e-89 - 1.19371e-88 - 1.26514e-87 - 1.31376e-86 - 1.33669e-85 - 1.33255e-84 - 1.30158e-83 - 1.24565e-82 - 1.16804e-81 - 1.07314e-80 - 9.66035e-80 - 8.5205e-79 - 7.36334e-78 - 6.23478e-77 - 5.17254e-76 - 4.20459e-75 - 3.34874e-74 - 2.61321e-73 - 1.99805e-72 - 1.49683e-71 - 1.0987e-70 - 7.90167e-70 - 5.56798e-69 - 3.84426e-68 - 2.60055e-67 - 1.72367e-66 - 1.11939e-65 - 7.12269e-65 - 4.44063e-64 - 2.71258e-63 - 1.62352e-62 - 9.52068e-62 - 5.47037e-61 - 3.07966e-60 - 1.69874e-59 - 9.18094e-59 - 4.86167e-58 - 2.52244e-57 - 1.28231e-56 - 6.38713e-56 - 3.11713e-55 - 1.49053e-54 - 6.98339e-54 - 3.20575e-53 - 1.44189e-52 - 6.35436e-52 - 2.74379e-51 - 1.16084e-50 - 4.81204e-50 - 1.95446e-49 - 7.77798e-49 - 3.03282e-48 - 1.15869e-47 - 4.33743e-47 - 1.59089e-46 - 5.7173e-46 - 2.0132e-45 - 6.9459e-45 - 2.34811e-44 - 7.77778e-44 - 2.52432e-43 - 8.02758e-43 - 2.50138e-42 - 7.63716e-42 - 2.28478e-41 - 6.6976e-41 - 1.9238e-40 - 5.41468e-40 - 1.49335e-39 - 4.0358e-39 - 1.06877e-38 - 2.77354e-38 - 7.05319e-38 - 1.75771e-37 - 4.29269e-37 - 1.02741e-36 - 2.40993e-36 - 5.54019e-36 - 1.24831e-35 - 2.75685e-35 - 5.9679e-35 - 1.2664e-34 - 2.63443e-34 - 5.37279e-34 - 1.07435e-33 - 2.10647e-33 - 4.05021e-33 - 7.63758e-33 - 1.41267e-32 - 2.56323e-32 - 4.5631e-32 - 7.97119e-32 - 1.36662e-31 - 2.29993e-31 - 3.80017e-31 - 6.16594e-31 - 9.82641e-31 - 1.53845e-30 - 2.36677e-30 - 3.57858e-30 - 5.31911e-30 - 7.77373e-30 - 1.11729e-29 - 1.57951e-29 - 2.19666e-29 - 3.00568e-29 - 4.0467e-29 - 5.36119e-29 - 6.98925e-29 - 8.9661e-29 - 1.13176e-28 - 1.40556e-28 - 1.71724e-28 - 2.06365e-28 - 2.43888e-28 - 2.83408e-28 - 3.23751e-28 - 3.63494e-28 - 4.01027e-28 - 4.34657e-28 - 4.62726e-28 - 4.83743e-28 - 4.96514e-28 - 5.00257e-28 - 4.94676e-28 - 4.80002e-28 - 4.56976e-28 - 4.26785e-28 - 3.90962e-28 - 3.51252e-28 - 3.09467e-28 - 2.6735e-28 - 2.26454e-28 - 1.88052e-28 - - - - - -3.636e-31 - -4.52491e-31 - -5.51739e-31 - -6.59163e-31 - -7.71593e-31 - -8.8494e-31 - -9.94361e-31 - -1.09434e-30 - -1.17787e-30 - -1.23073e-30 - -1.2008e-30 - -8.44938e-31 - 1.01217e-30 - 9.98446e-30 - 5.23624e-29 - 2.48732e-28 - 1.14076e-27 - 5.11123e-27 - 2.24255e-26 - 9.63929e-26 - 4.05952e-25 - 1.67509e-24 - 6.77235e-24 - 2.68273e-23 - 1.04124e-22 - 3.9597e-22 - 1.4754e-21 - 5.38636e-21 - 1.92671e-20 - 6.75267e-20 - 2.31884e-19 - 7.80195e-19 - 2.57201e-18 - 8.30765e-18 - 2.62918e-17 - 8.15268e-17 - 2.47695e-16 - 7.37345e-16 - 2.15061e-15 - 6.14593e-15 - 1.72088e-14 - 4.7212e-14 - 1.26908e-13 - 3.34245e-13 - 8.62533e-13 - 2.18084e-12 - 5.40267e-12 - 1.31139e-11 - 3.11881e-11 - 7.26749e-11 - 1.65927e-10 - 3.71181e-10 - 8.13565e-10 - 1.74717e-09 - 3.67633e-09 - 7.57934e-09 - 1.53103e-08 - 3.03023e-08 - 5.87628e-08 - 1.11652e-07 - 2.07858e-07 - 3.79145e-07 - 6.77609e-07 - 1.18656e-06 - 2.03582e-06 - 3.42235e-06 - 5.63697e-06 - 9.09714e-06 - 1.43847e-05 - 2.2286e-05 - 3.383e-05 - 5.03162e-05 - 7.33248e-05 - 0.000104696 - 0.00014647 - 0.000200771 - 0.000269645 - 0.000354829 - 0.000457491 - 0.000577942 - 0.000715355 - 0.000867554 - 0.00103088 - 0.00120021 - 0.00136913 - 0.00153027 - 0.00167583 - 0.00179817 - 0.00189049 - 0.00194743 - 0.00196562 - 0.00194402 - 0.001884 - 0.00178928 - 0.00166553 - 0.00151987 - 0.00136029 - 0.00119498 - 0.00103175 - 0.000877556 - 0.000738187 - 0.000618089 - 0.000520341 - 0.000446731 - 0.000397902 - 0.000373517 - 0.000372412 - 0.000392709 - 0.000431897 - 0.000486877 - 0.000554005 - 0.000629151 - 0.000707798 - 0.000785206 - 0.00085663 - 0.000917596 - 0.000964193 - 0.000993359 - 0.00100311 - 0.000992716 - 0.000962696 - 0.000914787 - 0.000851735 - 0.000777026 - 0.000694558 - 0.000608305 - 0.000522003 - 0.000438896 - 0.000361567 - 0.000291845 - 0.000230809 - 0.000178851 - 0.000135789 - 0.000101013 - 7.3625e-05 - 5.25788e-05 - 3.67903e-05 - 2.52227e-05 - 1.69429e-05 - 1.11512e-05 - 7.19101e-06 - 4.54356e-06 - 2.8128e-06 - 1.70616e-06 - 1.014e-06 - 5.90459e-07 - 3.36884e-07 - 1.88325e-07 - 1.03151e-07 - 5.53571e-08 - 2.91079e-08 - 1.49964e-08 - 7.57003e-09 - 3.74409e-09 - 1.81439e-09 - 8.61497e-10 - 4.00786e-10 - 1.82687e-10 - 8.15908e-11 - 3.57036e-11 - 1.53081e-11 - 6.43124e-12 - 2.64844e-12 - 1.07197e-12 - 4.34795e-13 - 2.00078e-13 - 1.63936e-13 - 3.00174e-13 - 7.46316e-13 - 1.9281e-12 - 4.92178e-12 - 1.23241e-11 - 3.02408e-11 - 7.27073e-11 - 1.71278e-10 - 3.95331e-10 - 8.94041e-10 - 1.98103e-09 - 4.30092e-09 - 9.1489e-09 - 1.90683e-08 - 3.89398e-08 - 7.79133e-08 - 1.52745e-07 - 2.93398e-07 - 5.52187e-07 - 1.01824e-06 - 1.83973e-06 - 3.25681e-06 - 5.64895e-06 - 9.60021e-06 - 1.59856e-05 - 2.60805e-05 - 4.16907e-05 - 6.52979e-05 - 0.000100207 - 0.000150671 - 0.000221973 - 0.000320411 - 0.00045316 - 0.000627962 - 0.000852611 - 0.00113424 - 0.00147842 - 0.0018881 - 0.0023626 - 0.00289663 - 0.00347962 - 0.00409551 - 0.00472303 - 0.00533667 - 0.00590822 - 0.00640885 - 0.00681147 - 0.00709314 - 0.00723725 - 0.00723514 - 0.00708696 - 0.00680167 - 0.00639614 - 0.00589358 - 0.00532131 - 0.00470844 - 0.00408352 - 0.00347255 - 0.00289752 - 0.00237559 - 0.00191887 - 0.00153483 - 0.00122706 - 0.000996256 - 0.000841365 - 0.000760476 - 0.000751552 - 0.000812793 - 0.000942642 - 0.00113944 - 0.00140081 - 0.00172285 - 0.0020993 - 0.00252088 - 0.00297478 - 0.00344473 - 0.00391135 - 0.00435317 - 0.00474797 - 0.00507448 - 0.00531415 - 0.00545286 - 0.00548225 - 0.00540051 - 0.00521259 - 0.00492967 - 0.00456811 - 0.00414791 - 0.00369091 - 0.00321908 - 0.00275291 - 0.00231029 - 0.00190581 - 0.00155063 - 0.00125284 - 0.00101825 - 0.000851451 - 0.000756966 - 0.000740438 - 0.000809583 - 0.00097481 - 0.00124937 - 0.00164889 - 0.00219032 - 0.00289009 - 0.0037618 - 0.00481339 - 0.00604416 - 0.00744202 - 0.00898139 - 0.0106221 - 0.0123098 - 0.013978 - 0.015552 - 0.0169538 - 0.0181087 - 0.0189516 - 0.019433 - 0.0195241 - 0.0192195 - 0.0185373 - 0.0175182 - 0.0162207 - 0.0147159 - 0.013081 - 0.0113928 - 0.00972205 - 0.00812871 - 0.00665921 - 0.00534516 - 0.00420373 - 0.00323926 - 0.00244565 - 0.00180917 - 0.0013113 - 0.000931238 - 0.000647972 - 0.000441762 - 0.000295092 - 0.000193136 - 0.000123853 - 7.78193e-05 - 4.79075e-05 - 2.88973e-05 - 1.70784e-05 - 9.88952e-06 - 5.61099e-06 - 3.11918e-06 - 1.69894e-06 - 9.06678e-07 - 4.74094e-07 - 2.42892e-07 - 1.21926e-07 - 5.99681e-08 - 2.88987e-08 - 1.3645e-08 - 6.31259e-09 - 2.86139e-09 - 1.27082e-09 - 5.53001e-10 - 2.3578e-10 - 9.8497e-11 - 4.03159e-11 - 1.61684e-11 - 6.35323e-12 - 2.44601e-12 - 9.22699e-13 - 3.41034e-13 - 1.23502e-13 - 4.38212e-14 - 1.52347e-14 - 5.18941e-15 - 1.73197e-15 - 5.66369e-16 - 1.81466e-16 - 5.69676e-17 - 1.75226e-17 - 5.28085e-18 - 1.55936e-18 - 4.51156e-19 - 1.27892e-19 - 3.5522e-20 - 9.66693e-21 - 2.5776e-21 - 6.73413e-22 - 1.72385e-22 - 4.3267e-23 - 1.07691e-23 - 3.17029e-24 - 3.13795e-24 - 1.04192e-23 - 4.08832e-23 - 1.59072e-22 - 6.06845e-22 - 2.26838e-21 - 8.3079e-21 - 2.98129e-20 - 1.04822e-19 - 3.6111e-19 - 1.21889e-18 - 4.03109e-18 - 1.30623e-17 - 4.14718e-17 - 1.2901e-16 - 3.93216e-16 - 1.17429e-15 - 3.43603e-15 - 9.85087e-15 - 2.76713e-14 - 7.6159e-14 - 2.05376e-13 - 5.42644e-13 - 1.40481e-12 - 3.56333e-12 - 8.85588e-12 - 2.15648e-11 - 5.1451e-11 - 1.20276e-10 - 2.75489e-10 - 6.1825e-10 - 1.35944e-09 - 2.92883e-09 - 6.18252e-09 - 1.27871e-08 - 2.5913e-08 - 5.14516e-08 - 1.00096e-07 - 1.90797e-07 - 3.56339e-07 - 6.52066e-07 - 1.16912e-06 - 2.05381e-06 - 3.53508e-06 - 5.96177e-06 - 9.85117e-06 - 1.59492e-05 - 2.53002e-05 - 3.93232e-05 - 5.98839e-05 - 8.93528e-05 - 0.00013063 - 0.000187119 - 0.000262622 - 0.000361147 - 0.000486606 - 0.000642415 - 0.000830998 - 0.00105327 - 0.00130809 - 0.00159189 - 0.00189838 - 0.00221861 - 0.00254131 - 0.00285355 - 0.00314183 - 0.00339333 - 0.00359738 - 0.00374691 - 0.00383975 - 0.00387957 - 0.00387642 - 0.00384664 - 0.00381224 - 0.0037996 - 0.00383766 - 0.00395569 - 0.00418069 - 0.0045347 - 0.00503217 - 0.00567767 - 0.00646409 - 0.00737179 - 0.00836862 - 0.00941129 - 0.0104478 - 0.011421 - 0.012273 - 0.0129503 - 0.013408 - 0.0136139 - 0.013552 - 0.0132229 - 0.0126445 - 0.011849 - 0.0108804 - 0.00978988 - 0.00863109 - 0.00745595 - 0.00631082 - 0.00523372 - 0.00425281 - 0.00338595 - 0.00264134 - 0.00201887 - 0.00151194 - 0.00110944 - 0.000797693 - 0.000562024 - 0.000388087 - 0.000262733 - 0.000174531 - 0.00011398 - 7.34973e-05 - 4.72468e-05 - 3.0892e-05 - 2.1316e-05 - 1.63489e-05 - 1.45246e-05 - 1.48771e-05 - 1.6778e-05 - 1.98118e-05 - 2.36856e-05 - 2.81654e-05 - 3.30354e-05 - 3.80753e-05 - 4.30518e-05 - 4.77214e-05 - 5.18411e-05 - 5.51843e-05 - 5.75589e-05 - 5.88241e-05 - 5.89031e-05 - 5.77908e-05 - 5.55543e-05 - 5.23255e-05 - 4.82887e-05 - 4.36632e-05 - 3.86831e-05 - 3.35788e-05 - 2.85591e-05 - 2.37992e-05 - 1.94319e-05 - 1.55456e-05 - 1.21852e-05 - 9.35833e-06 - 7.04206e-06 - 5.19204e-06 - 3.7507e-06 - 2.65476e-06 - 1.84108e-06 - 1.251e-06 - 8.32877e-07 - 5.433e-07 - 3.47245e-07 - 2.17455e-07 - 1.33426e-07 - 8.02134e-08 - 4.72494e-08 - 2.72712e-08 - 1.54258e-08 - 8.55829e-09 - 4.67484e-09 - 2.55734e-09 - 1.50298e-09 - 1.16839e-09 - 1.50985e-09 - 2.83243e-09 - 5.99601e-09 - 1.28846e-08 - 2.73441e-08 - 5.6957e-08 - 1.16287e-07 - 2.32641e-07 - 4.56022e-07 - 8.75839e-07 - 1.64816e-06 - 3.03887e-06 - 5.48987e-06 - 9.71737e-06 - 1.68528e-05 - 2.86373e-05 - 4.76792e-05 - 7.77791e-05 - 0.000124318 - 0.000194689 - 0.000298735 - 0.000449125 - 0.000661584 - 0.00095486 - 0.0013503 - 0.00187094 - 0.00253995 - 0.00337853 - 0.00440319 - 0.00562268 - 0.00703487 - 0.00862394 - 0.0103584 - 0.0121903 - 0.0140565 - 0.0158808 - 0.0175795 - 0.0190668 - 0.0202621 - 0.0210974 - 0.0215234 - 0.0215144 - 0.021071 - 0.0202198 - 0.019011 - 0.0175134 - 0.0158078 - 0.0139802 - 0.012114 - 0.0102849 - 0.00855561 - 0.00697329 - 0.00556879 - 0.00435734 - 0.00334055 - 0.0025093 - 0.00184682 - 0.00133178 - 0.00094097 - 0.000651414 - 0.000441851 - 0.00029365 - 0.000191215 - 0.000121998 - 7.62635e-05 - 4.6711e-05 - 2.80323e-05 - 1.6483e-05 - 9.49616e-06 - 5.36041e-06 - 2.96473e-06 - 1.6066e-06 - 8.53039e-07 - 4.43778e-07 - 2.26204e-07 - 1.12972e-07 - 5.52814e-08 - 2.65047e-08 - 1.2451e-08 - 5.73091e-09 - 2.58451e-09 - 1.14202e-09 - 4.94439e-10 - 2.09776e-10 - 8.72944e-11 - 3.58409e-11 - 1.50894e-11 - 7.98683e-12 - 8.47379e-12 - 1.69387e-11 - 4.04834e-11 - 9.81476e-11 - 2.34372e-10 - 5.48791e-10 - 1.2592e-09 - 2.83093e-09 - 6.23592e-09 - 1.34589e-08 - 2.84612e-08 - 5.89705e-08 - 1.19716e-07 - 2.38127e-07 - 4.64089e-07 - 8.86197e-07 - 1.65804e-06 - 3.03948e-06 - 5.45932e-06 - 9.6076e-06 - 1.65664e-05 - 2.79884e-05 - 4.63304e-05 - 7.51432e-05 - 0.000119413 - 0.000185929 - 0.00028365 - 0.000423988 - 0.000620957 - 0.00089106 - 0.00125282 - 0.00172587 - 0.0023295 - 0.00308074 - 0.00399194 - 0.00506816 - 0.00630454 - 0.0076841 - 0.00917634 - 0.010737 - 0.0123093 - 0.0138268 - 0.0152175 - 0.0164099 - 0.0173382 - 0.0179489 - 0.0182058 - 0.0180933 - 0.0176182 - 0.0168091 - 0.0157131 - 0.0143919 - 0.0129155 - 0.0113564 - 0.00978378 - 0.00825866 - 0.00683045 - 0.00553511 - 0.0043948 - 0.00341893 - 0.00260602 - 0.00194626 - 0.00142417 - 0.00102108 - 0.00071729 - 0.000493703 - 0.000332947 - 0.000219998 - 0.00014243 - 9.03484e-05 - 5.61534e-05 - 3.41955e-05 - 2.04032e-05 - 1.19279e-05 - 6.83231e-06 - 3.83449e-06 - 2.10855e-06 - 1.13605e-06 - 5.9972e-07 - 3.10196e-07 - 1.57203e-07 - 7.80587e-08 - 3.79768e-08 - 1.81031e-08 - 8.45523e-09 - 3.86931e-09 - 1.73492e-09 - 7.62186e-10 - 3.2808e-10 - 1.38368e-10 - 5.71779e-11 - 2.31504e-11 - 9.18384e-12 - 3.56967e-12 - 1.35946e-12 - 5.07276e-13 - 1.85463e-13 - 6.64368e-14 - 2.33183e-14 - 8.01901e-15 - 2.70198e-15 - 8.9203e-16 - 2.88546e-16 - 9.14505e-17 - 2.83985e-17 - 8.64052e-18 - 2.57586e-18 - 7.52387e-19 - 2.15326e-19 - 6.03798e-20 - 1.65898e-20 - 4.46888e-21 - 1.1907e-21 - 3.54724e-22 - 2.70303e-22 - 7.45922e-22 - 2.68317e-21 - 9.67303e-21 - 3.42192e-20 - 1.1862e-19 - 4.02889e-19 - 1.34076e-18 - 4.37171e-18 - 1.39666e-17 - 4.37185e-17 - 1.34084e-16 - 4.02927e-16 - 1.18635e-15 - 3.42243e-15 - 9.67374e-15 - 2.67911e-14 - 7.26984e-14 - 1.93284e-13 - 5.03504e-13 - 1.28513e-12 - 3.21388e-12 - 7.87496e-12 - 1.89062e-11 - 4.44733e-11 - 1.02502e-10 - 2.31473e-10 - 5.12165e-10 - 1.11034e-09 - 2.35855e-09 - 4.90879e-09 - 1.00102e-08 - 2.00013e-08 - 3.91579e-08 - 7.51153e-08 - 1.41185e-07 - 2.60021e-07 - 4.69236e-07 - 8.29749e-07 - 1.43776e-06 - 2.44132e-06 - 4.06237e-06 - 6.62491e-06 - 1.0589e-05 - 1.65903e-05 - 2.54819e-05 - 3.83764e-05 - 5.66822e-05 - 8.21314e-05 - 0.000116792 - 0.00016307 - 0.000223695 - 0.000301717 - 0.000400522 - 0.000523905 - 0.000676234 - 0.000862732 - 0.00108992 - 0.00136617 - 0.00170242 - 0.00211279 - 0.00261515 - 0.00323128 - 0.00398644 - 0.0049082 - 0.00602429 - 0.00735944 - 0.00893147 - 0.0107468 - 0.0127958 - 0.0150493 - 0.0174561 - 0.0199423 - 0.0224143 - 0.0247631 - 0.0268725 - 0.0286279 - 0.0299271 - 0.0306897 - 0.0308656 - 0.0304406 - 0.0294379 - 0.027917 - 0.0259682 - 0.0237049 - 0.0212543 - 0.0187473 - 0.0163094 - 0.0140532 - 0.012073 - 0.0104412 - 0.00920708 - 0.00839714 - 0.00801617 - 0.00804901 - 0.00846236 - 0.00920671 - 0.0102183 - 0.0114214 - 0.0127314 - 0.0140583 - 0.0153113 - 0.0164042 - 0.0172608 - 0.0178203 - 0.0180418 - 0.0179068 - 0.017421 - 0.0166126 - 0.0155294 - 0.0142346 - 0.0128009 - 0.0113044 - 0.00981974 - 0.00841484 - 0.00714807 - 0.00606625 - 0.00520401 - 0.00458421 - 0.00421879 - 0.00410993 - 0.00425091 - 0.00462675 - 0.00521439 - 0.00598278 - 0.0068929 - 0.00789827 - 0.00894605 - 0.00997897 - 0.0109382 - 0.0117671 - 0.0124148 - 0.0128403 - 0.0130161 - 0.01293 - 0.0125862 - 0.0120047 - 0.011219 - 0.0102732 - 0.00921717 - 0.00810268 - 0.00697907 - 0.00588986 - 0.00487025 - 0.00394581 - 0.00313231 - 0.00243638 - 0.00185693 - 0.00138696 - 0.00101542 - 0.000729103 - 0.000514093 - 0.000357004 - 0.000245774 - 0.00017014 - 0.00012182 - 9.44836e-05 - 8.35732e-05 - 8.60424e-05 - 0.000100046 - 0.000124617 - 0.000159347 - 0.000204081 - 0.000258632 - 0.000322534 - 0.000394841 - 0.000473977 - 0.000557675 - 0.000642995 - 0.000726438 - 0.000804153 - 0.00087221 - 0.00092692 - 0.000965165 - 0.000984686 - 0.000984308 - 0.000964054 - 0.000925143 - 0.000869867 - 0.000801371 - 0.000723355 - 0.000639744 - 0.000554367 - 0.00047068 - 0.000391553 - 0.000319149 - 0.000254878 - 0.000199438 - 0.000152904 - 0.00011486 - 8.45387e-05 - 6.09647e-05 - 4.30763e-05 - 2.98219e-05 - 2.02287e-05 - 1.34443e-05 - 8.75481e-06 - 5.58588e-06 - 3.49199e-06 - 2.13892e-06 - 1.2837e-06 - 7.54953e-07 - 4.3522e-07 - 2.46288e-07 - 1.37595e-07 - 7.76206e-08 - 4.7872e-08 - 3.91206e-08 - 4.96818e-08 - 8.57782e-08 - 1.64423e-07 - 3.19802e-07 - 6.1486e-07 - 1.16078e-06 - 2.1483e-06 - 3.8961e-06 - 6.92335e-06 - 1.20543e-05 - 2.0564e-05 - 3.43723e-05 - 5.6292e-05 - 9.03278e-05 - 0.000142015 - 0.000218767 - 0.000330193 - 0.000488304 - 0.000707537 - 0.00100449 - 0.00139726 - 0.00190436 - 0.00254305 - 0.00332735 - 0.00426559 - 0.00535792 - 0.00659401 - 0.00795135 - 0.00939439 - 0.0108751 - 0.0123349 - 0.013708 - 0.0149262 - 0.0159243 - 0.016646 - 0.0170489 - 0.0171089 - 0.0168223 - 0.0162065 - 0.0152982 - 0.0141497 - 0.0128242 - 0.01139 - 0.00991533 - 0.00846323 - 0.00708799 - 0.00583302 - 0.00473027 - 0.00380108 - 0.00305813 - 0.00250808 - 0.00215444 - 0.00200017 - 0.00204972 - 0.00231011 - 0.00279091 - 0.003503 - 0.00445631 - 0.00565652 - 0.00710137 - 0.0087768 - 0.0106537 - 0.0126858 - 0.014809 - 0.0169432 - 0.0189962 - 0.0208692 - 0.0224645 - 0.0236937 - 0.0244855 - 0.0247927 - 0.0245967 - 0.0239093 - 0.0227716 - 0.0212499 - 0.0194294 - 0.0174059 - 0.0152782 - 0.0131396 - 0.0110721 - 0.00914149 - 0.00739501 - 0.00586135 - 0.00455191 - 0.00346359 - 0.00258224 - 0.00188626 - 0.00135004 - 0.000946729 - 0.000650493 - 0.000437923 - 0.000288862 - 0.000186692 - 0.000118227 - 7.33688e-05 - 4.46338e-05 - 2.66504e-05 - 1.56821e-05 - 9.21728e-06 - 5.63909e-06 - 3.97946e-06 - 3.75189e-06 - 4.85588e-06 - 7.54564e-06 - 1.24576e-05 - 2.06929e-05 - 3.39515e-05 - 5.47126e-05 - 8.64515e-05 - 0.000133876 - 0.000203149 - 0.000302064 - 0.000440104 - 0.00062834 - 0.000879078 - 0.00120523 - 0.00161937 - 0.00213248 - 0.00275244 - 0.00348253 - 0.00431992 - 0.00525458 - 0.00626873 - 0.00733715 - 0.00842831 - 0.00950652 - 0.0105347 - 0.0114778 - 0.0123057 - 0.0129966 - 0.0135384 - 0.0139296 - 0.0141784 - 0.0143005 - 0.0143162 - 0.0142468 - 0.0141109 - 0.0139217 - 0.0136857 - 0.0134019 - 0.0130632 - 0.012659 - 0.0121776 - 0.0116098 - 0.0109513 - 0.0102046 - 0.00937978 - 0.00849391 - 0.00756962 - 0.00663291 - 0.00571057 - 0.00482772 - 0.00400577 - 0.00326099 - 0.00260377 - 0.00203868 - 0.00156499 - 0.00117773 - 0.000868844 - 0.000628451 - 0.000445965 - 0.00031104 - 0.00021428 - 0.000147749 - 0.000105306 - 8.28571e-05 - 7.85795e-05 - 9.31949e-05 - 0.000130355 - 0.000197161 - 0.00030483 - 0.000469447 - 0.000712728 - 0.00106262 - 0.00155357 - 0.00222613 - 0.00312575 - 0.00430043 - 0.00579713 - 0.0076569 - 0.00990902 - 0.0125645 - 0.0156098 - 0.0190014 - 0.0226627 - 0.0264835 - 0.0303231 - 0.0340181 - 0.0373924 - 0.0402711 - 0.0424952 - 0.0439363 - 0.0445086 - 0.0441776 - 0.0429631 - 0.040938 - 0.0382204 - 0.0349623 - 0.0313358 - 0.0275182 - 0.0236774 - 0.0199612 - 0.0164883 - 0.0133445 - 0.0105819 - 0.00822174 - 0.00625893 - 0.00466846 - 0.0034118 - 0.00244304 - 0.00171401 - 0.00117824 - 0.000793582 - 0.000523705 - 0.000338624 - 0.000214529 - 0.000133165 - 8.09901e-05 - 4.82626e-05 - 2.8179e-05 - 1.61205e-05 - 9.03581e-06 - 4.96241e-06 - 2.67027e-06 - 1.40785e-06 - 7.27263e-07 - 3.68099e-07 - 1.82547e-07 - 8.87001e-08 - 4.22295e-08 - 1.97008e-08 - 9.00956e-09 - 4.04833e-09 - 1.81029e-09 - 8.60698e-10 - 5.58336e-10 - 6.85971e-10 - 1.31585e-09 - 2.8408e-09 - 6.1788e-09 - 1.32343e-08 - 2.77987e-08 - 5.7221e-08 - 1.15408e-07 - 2.28062e-07 - 4.4158e-07 - 8.37727e-07 - 1.55716e-06 - 2.83595e-06 - 5.0606e-06 - 8.84794e-06 - 1.51572e-05 - 2.5441e-05 - 4.18393e-05 - 6.74174e-05 - 0.000106438 - 0.000164648 - 0.000249549 - 0.000370588 - 0.000539216 - 0.000768726 - 0.00107378 - 0.0014696 - 0.00197069 - 0.00258926 - 0.00333325 - 0.00420434 - 0.00519594 - 0.0062917 - 0.00746463 - 0.00867732 - 0.00988325 - 0.0110294 - 0.0120597 - 0.01292 - 0.013562 - 0.0139483 - 0.0140559 - 0.0138781 - 0.0134257 - 0.0127258 - 0.0118186 - 0.0107544 - 0.00958832 - 0.00837598 - 0.00716911 - 0.00601218 - 0.0049401 - 0.00397719 - 0.00313728 - 0.00242475 - 0.00183619 - 0.00136241 - 0.000990447 - 0.000705493 - 0.00049237 - 0.000336687 - 0.000225579 - 0.000148084 - 9.52474e-05 - 6.00255e-05 - 3.70643e-05 - 2.24239e-05 - 1.32924e-05 - 7.72031e-06 - 4.39341e-06 - 2.44966e-06 - 1.33828e-06 - 7.16347e-07 - 3.75697e-07 - 1.93059e-07 - 9.7204e-08 - 4.79558e-08 - 2.31886e-08 - 1.10053e-08 - 5.16537e-09 - 2.49231e-09 - 1.45573e-09 - 1.44813e-09 - 2.49265e-09 - 5.27515e-09 - 1.15271e-08 - 2.49327e-08 - 5.29399e-08 - 1.10176e-07 - 2.24677e-07 - 4.48922e-07 - 8.78862e-07 - 1.68581e-06 - 3.16834e-06 - 5.83436e-06 - 1.05267e-05 - 1.86091e-05 - 3.22327e-05 - 5.47023e-05 - 9.09601e-05 - 0.000148195 - 0.000236566 - 0.000370005 - 0.000567022 - 0.000851392 - 0.00125255 - 0.00180551 - 0.00254999 - 0.00352871 - 0.00478442 - 0.00635594 - 0.00827308 - 0.010551 - 0.0131842 - 0.0161418 - 0.0193636 - 0.0227592 - 0.02621 - 0.0295742 - 0.0326961 - 0.0354175 - 0.0375905 - 0.0390913 - 0.0398314 - 0.0397669 - 0.0389028 - 0.0372926 - 0.0350338 - 0.0322585 - 0.0291224 - 0.0257915 - 0.02243 - 0.0191902 - 0.0162049 - 0.0135829 - 0.0114083 - 0.00974183 - 0.00862375 - 0.00807727 - 0.00811156 - 0.00872333 - 0.00989679 - 0.011602 - 0.0137919 - 0.0163994 - 0.0193348 - 0.0224847 - 0.0257144 - 0.0288725 - 0.0317988 - 0.0343349 - 0.0363363 - 0.0376843 - 0.0382965 - 0.0381344 - 0.037207 - 0.0355694 - 0.0333172 - 0.0305773 - 0.0274959 - 0.0242255 - 0.020913 - 0.0176887 - 0.0146593 - 0.0119033 - 0.00947019 - 0.00738221 - 0.00563834 - 0.00421942 - 0.0030938 - 0.00222264 - 0.00156455 - 0.0010791 - 0.000729329 - 0.000483146 - 0.000313949 - 0.000200581 - 0.000126914 - 8.12404e-05 - 5.5656e-05 - 4.55669e-05 - 4.94212e-05 - 6.87538e-05 - 0.000108612 - 0.000178427 - 0.000293379 - 0.000476297 - 0.000760061 - 0.00119043 - 0.00182912 - 0.00275668 - 0.00407487 - 0.00590754 - 0.00839955 - 0.0117125 - 0.0160171 - 0.0214803 - 0.0282497 - 0.0364326 - 0.0460744 - 0.0571362 - 0.0694755 - 0.0828344 - 0.0968356 - 0.110993 - 0.124731 - 0.137426 - 0.148443 - 0.157195 - 0.163191 - 0.16608 - 0.16569 - 0.162039 - 0.155338 - 0.145969 - 0.13445 - 0.121386 - 0.107417 - 0.0931679 - 0.0792032 - 0.0659922 - 0.0538902 - 0.0431307 - 0.0338313 - 0.0260076 - 0.0195944 - 0.0144686 - 0.0104718 - 0.00743065 - 0.00517275 - 0.00353869 - 0.0023892 - 0.00160904 - 0.00110793 - 0.000819713 - 0.000700487 - 0.000726235 - 0.000890471 - 0.00120196 - 0.00168246 - 0.00236436 - 0.00328777 - 0.00449696 - 0.00603582 - 0.00794235 - 0.0102424 - 0.0129428 - 0.0160253 - 0.0194414 - 0.0231094 - 0.0269146 - 0.0307131 - 0.0343397 - 0.0376189 - 0.0403787 - 0.0424655 - 0.0437579 - 0.0441788 - 0.0437027 - 0.0423583 - 0.040226 - 0.0374293 - 0.0341234 - 0.0304811 - 0.0266776 - 0.022877 - 0.0192215 - 0.0158239 - 0.0127637 - 0.0100873 - 0.00781109 - 0.00592634 - 0.00440555 - 0.00320891 - 0.0022902 - 0.00160171 - 0.00109801 - 0.000738346 - 0.000488074 - 0.000319114 - 0.000209886 - 0.000144899 - 0.000114287 - 0.00011345 - 0.000142967 - 0.000208865 - 0.000323272 - 0.000505441 - 0.000783052 - 0.00119362 - 0.00178573 - 0.00261982 - 0.00376788 - 0.00531191 - 0.00734032 - 0.0099422 - 0.0131993 - 0.0171757 - 0.0219067 - 0.0273861 - 0.0335566 - 0.0403008 - 0.0474392 - 0.0547326 - 0.0618925 - 0.0685978 - 0.074518 - 0.0793391 - 0.0827917 - 0.0846753 - 0.084878 - 0.0833871 - 0.0802907 - 0.0757691 - 0.0700771 - 0.0635207 - 0.0564296 - 0.0491301 - 0.0419213 - 0.0350564 - 0.0287303 - 0.0230756 - 0.0181636 - 0.0140114 - 0.0105924 - 0.00784763 - 0.00569779 - 0.00405413 - 0.0028269 - 0.00193171 - 0.00129357 - 0.000848892 - 0.000545916 - 0.00034404 - 0.000212471 - 0.000128587 - 7.62598e-05 - 4.43196e-05 - 2.52403e-05 - 1.40861e-05 - 7.70341e-06 - 4.12826e-06 - 2.16792e-06 - 1.1156e-06 - 5.62553e-07 - 2.77973e-07 - 1.34595e-07 - 6.38609e-08 - 2.96909e-08 - 1.35267e-08 - 6.03865e-09 - 2.64158e-09 - 1.13231e-09 - 4.75596e-10 - 1.95742e-10 - 7.89412e-11 - 3.11955e-11 - 1.20796e-11 - 4.58331e-12 - 1.70407e-12 - 6.20988e-13 - 2.22293e-13 - 7.9713e-14 - 3.33658e-14 - 2.94752e-14 - 6.38018e-14 - 1.77569e-13 - 5.02538e-13 - 1.39959e-12 - 3.82147e-12 - 1.02254e-11 - 2.6812e-11 - 6.88931e-11 - 1.73468e-10 - 4.28014e-10 - 1.03489e-09 - 2.45205e-09 - 5.69327e-09 - 1.29536e-08 - 2.88814e-08 - 6.31019e-08 - 1.35103e-07 - 2.83456e-07 - 5.82778e-07 - 1.17414e-06 - 2.3181e-06 - 4.48481e-06 - 8.50263e-06 - 1.57965e-05 - 2.87584e-05 - 5.13057e-05 - 8.96943e-05 - 0.00015366 - 0.000257961 - 0.00042437 - 0.000684119 - 0.00108072 - 0.001673 - 0.00253788 - 0.00377261 - 0.00549553 - 0.00784462 - 0.0109731 - 0.0150412 - 0.0202036 - 0.0265931 - 0.0343008 - 0.0433543 - 0.0536975 - 0.0651733 - 0.0775138 - 0.0903404 - 0.103176 - 0.115472 - 0.12664 - 0.136103 - 0.143344 - 0.147949 - 0.149655 - 0.14837 - 0.144186 - 0.137373 - 0.128353 - 0.117658 - 0.105889 - 0.093658 - 0.0815473 - 0.0700646 - 0.0596157 - 0.0504871 - 0.0428422 - 0.0367268 - 0.0320839 - 0.0287735 - 0.0265961 - 0.0253153 - 0.0246804 - 0.024446 - 0.0243882 - 0.0243168 - 0.0240838 - 0.0235875 - 0.0227725 - 0.021627 - 0.0201755 - 0.0184715 - 0.0165872 - 0.014604 - 0.0126035 - 0.0106603 - 0.00883602 - 0.00717679 - 0.00571177 - 0.00445418 - 0.00340341 - 0.00254803 - 0.00186912 - 0.00134342 - 0.000946069 - 0.000652789 - 0.000441327 - 0.000292338 - 0.000189735 - 0.000120655 - 7.51766e-05 - 4.5894e-05 - 2.74515e-05 - 1.60884e-05 - 9.2384e-06 - 5.19778e-06 - 2.86533e-06 - 1.54764e-06 - 8.19038e-07 - 4.24702e-07 - 2.15801e-07 - 1.07503e-07 - 5.26328e-08 - 2.56405e-08 - 1.31726e-08 - 8.77456e-09 - 1.02078e-08 - 1.85691e-08 - 3.89947e-08 - 8.33138e-08 - 1.75745e-07 - 3.63781e-07 - 7.38012e-07 - 1.46706e-06 - 2.85743e-06 - 5.45305e-06 - 1.01963e-05 - 1.86801e-05 - 3.35317e-05 - 5.89749e-05 - 0.000101629 - 0.000171594 - 0.000283873 - 0.000460134 - 0.00073077 - 0.00113714 - 0.00173374 - 0.00258995 - 0.00379084 - 0.00543645 - 0.00763894 - 0.0105169 - 0.0141866 - 0.0187503 - 0.0242813 - 0.0308088 - 0.0383013 - 0.0466542 - 0.0556807 - 0.0651114 - 0.0746018 - 0.0837495 - 0.0921213 - 0.0992861 - 0.104852 - 0.108503 - 0.110031 - 0.109357 - 0.106543 - 0.101791 - 0.0954271 - 0.087877 - 0.0796374 - 0.0712426 - 0.0632336 - 0.0561318 - 0.0504187 - 0.0465218 - 0.044805 - 0.045561 - 0.0490042 - 0.0552601 - 0.0643534 - 0.0761915 - 0.0905495 - 0.107057 - 0.125194 - 0.144296 - 0.163577 - 0.182163 - 0.199146 - 0.213648 - 0.224879 - 0.232211 - 0.235225 - 0.233747 - 0.227864 - 0.217912 - 0.204443 - 0.188177 - 0.169934 - 0.150572 - 0.130918 - 0.111717 - 0.0935927 - 0.0770224 - 0.0623326 - 0.0497065 - 0.0392039 - 0.0307866 - 0.0243474 - 0.0197367 - 0.0167861 - 0.0153248 - 0.0151908 - 0.0162352 - 0.0183228 - 0.0213286 - 0.0251345 - 0.0296256 - 0.0346886 - 0.0402135 - 0.0460961 - 0.052243 - 0.0585756 - 0.0650326 - 0.0715684 - 0.0781472 - 0.0847335 - 0.091278 - 0.0977038 - 0.103894 - 0.109684 - 0.114865 - 0.119189 - 0.122391 - 0.124217 - 0.124448 - 0.122933 - 0.119609 - 0.114518 - 0.107811 - 0.0997331 - 0.0906066 - 0.0808031 - 0.0707102 - 0.0607003 - 0.0511033 - 0.0421863 - 0.0341421 - 0.0270865 - 0.0210628 - 0.0160525 - 0.0119898 - 0.00877597 - 0.00629478 - 0.00442438 - 0.00304721 - 0.00205646 - 0.00135988 - 0.000881132 - 0.000559434 - 0.000348064 - 0.000212277 - 0.000127044 - 7.49014e-05 - 4.40862e-05 - 2.70444e-05 - 1.93388e-05 - 1.89574e-05 - 2.60199e-05 - 4.28955e-05 - 7.47722e-05 - 0.00013075 - 0.000225561 - 0.000382019 - 0.000634315 - 0.00103218 - 0.00164583 - 0.00257148 - 0.00393681 - 0.00590566 - 0.00868071 - 0.0125027 - 0.0176447 - 0.0243998 - 0.0330614 - 0.0438952 - 0.0571052 - 0.0727939 - 0.0909236 - 0.111281 - 0.133452 - 0.156816 - 0.180558 - 0.203706 - 0.225193 - 0.24393 - 0.258902 - 0.269258 - 0.274385 - 0.273977 - 0.268058 - 0.256982 - 0.241401 - 0.222195 - 0.200397 - 0.177096 - 0.153352 - 0.130119 - 0.108184 - 0.088143 - 0.0703811 - 0.0550906 - 0.0422952 - 0.0318886 - 0.0236757 - 0.0174137 - 0.0128489 - 0.00974575 - 0.00790899 - 0.00719695 - 0.00752776 - 0.00887847 - 0.0112782 - 0.0147954 - 0.0195211 - 0.0255467 - 0.03294 - 0.0417194 - 0.0518294 - 0.0631202 - 0.0753344 - 0.0881046 - 0.100962 - 0.113362 - 0.124714 - 0.134431 - 0.141978 - 0.146921 - 0.148964 - 0.147984 - 0.144041 - 0.137371 - 0.128363 - 0.117523 - 0.105425 - 0.0926619 - 0.0797996 - 0.0673356 - 0.0556729 - 0.0451043 - 0.0358108 - 0.0278695 - 0.0212705 - 0.0159377 - 0.0117507 - 0.0085666 - 0.0062372 - 0.00462324 - 0.00360352 - 0.00307992 - 0.00297862 - 0.00324833 - 0.00385612 - 0.00478184 - 0.00601151 - 0.00753051 - 0.0093171 - 0.011337 - 0.0135396 - 0.0158561 - 0.0182006 - 0.0204731 - 0.0225658 - 0.0243709 - 0.0257892 - 0.0267388 - 0.0271636 - 0.0270377 - 0.0263689 - 0.0251973 - 0.0235919 - 0.0216433 - 0.019456 - 0.0171389 - 0.014797 - 0.0125243 - 0.0103983 - 0.00847798 - 0.00680284 - 0.00539503 - 0.00426232 - 0.00340205 - 0.00280523 - 0.00246013 - 0.00235501 - 0.00247968 - 0.00282577 - 0.00338583 - 0.00415139 - 0.00511045 - 0.00624471 - 0.00752714 - 0.00892027 - 0.0103758 - 0.0118353 - 0.0132334 - 0.0145009 - 0.0155705 - 0.0163822 - 0.0168885 - 0.017059 - 0.0168832 - 0.0163718 - 0.0155551 - 0.0144807 - 0.0132081 - 0.011804 - 0.0103361 - 0.00886787 - 0.00745451 - 0.00613983 - 0.00495484 - 0.00391778 - 0.0030352 - 0.00230394 - 0.00171353 - 0.00124868 - 0.000891548 - 0.000623701 - 0.000427509 - 0.000287111 - 0.000188926 - 0.000121807 - 7.69461e-05 - 4.76255e-05 - 2.88821e-05 - 1.71615e-05 - 9.99121e-06 - 5.69926e-06 - 3.18533e-06 - 1.74433e-06 - 9.35921e-07 - 4.92024e-07 - 2.53438e-07 - 1.27906e-07 - 6.32485e-08 - 3.06439e-08 - 1.45471e-08 - 6.7662e-09 - 3.08354e-09 - 1.37687e-09 - 6.02381e-10 - 2.58218e-10 - 1.08453e-10 - 4.46303e-11 - 1.79952e-11 - 7.10917e-12 - 2.75181e-12 - 1.04365e-12 - 3.8782e-13 - 1.41202e-13 - 5.03719e-14 - 1.76065e-14 - 6.02967e-15 - 2.02326e-15 - 6.65191e-16 - 2.14278e-16 - 6.76311e-17 - 2.09147e-17 - 6.33715e-18 - 1.88137e-18 - 5.47254e-19 - 1.5597e-19 - 4.35544e-20 - 1.19168e-20 - 3.19464e-21 - 8.39116e-22 - 2.15953e-22 - 5.44544e-23 - 1.34538e-23 - 3.2568e-24 - 7.72459e-25 - 1.79513e-25 - 4.08747e-26 - 9.11906e-27 - 1.99334e-27 - 4.26924e-28 - 8.95892e-29 - 1.84204e-29 - 3.71088e-30 - 7.32476e-31 - 1.4166e-31 - 2.68433e-32 - 4.98382e-33 - 9.06622e-34 - 1.61594e-34 - 2.82204e-35 - 4.82878e-36 - 8.09559e-37 - 1.32983e-37 - 2.14033e-38 - 3.37522e-39 - 5.21507e-40 - 7.89506e-41 - 1.17108e-41 - 1.70199e-42 - 2.4236e-43 - 3.38146e-44 - 4.62257e-45 - 6.19155e-46 - 8.12554e-47 - 1.04482e-47 - 1.31634e-48 - 1.62492e-49 - 1.96532e-50 - 2.329e-51 - 2.70423e-52 - 3.07649e-53 - 3.42928e-54 - 3.74531e-55 - 4.00783e-56 - 4.20211e-57 - 4.31681e-58 - 4.34505e-59 - 4.28512e-60 - 4.14066e-61 - 3.92023e-62 - 3.63656e-63 - 3.30527e-64 - 2.94348e-65 - 2.56833e-66 - 2.19572e-67 - 1.83925e-68 - 1.50953e-69 - 1.21389e-70 - 9.56429e-72 - 7.38353e-73 - 5.58485e-74 - 4.13901e-75 - 3.00551e-76 - 2.13834e-77 - 1.49064e-78 - 1.01813e-79 - 6.81354e-81 - 4.46764e-82 - 2.87026e-83 - 1.80676e-84 - 1.11434e-85 - 6.73396e-87 - 3.98713e-88 - 2.31306e-89 - 1.31477e-90 - 7.32237e-92 - 3.99566e-93 - 2.1363e-94 - 1.11911e-95 - 5.7441e-97 - 2.88873e-98 - 1.42341e-99 - 6.87207e-101 - 3.25074e-102 - 1.50666e-103 - 6.84201e-105 - 3.04432e-106 - 1.32719e-107 - 5.66909e-109 - 2.37623e-110 - 1.87428e-111 - 2.20949e-110 - 5.28801e-109 - 1.24221e-107 - 2.85914e-106 - 6.44781e-105 - 1.42471e-103 - 3.08445e-102 - 6.54282e-101 - 1.35985e-99 - 2.76918e-98 - 5.52523e-97 - 1.08015e-95 - 2.06899e-94 - 3.88299e-93 - 7.14023e-92 - 1.28646e-90 - 2.27099e-89 - 3.92799e-88 - 6.65678e-87 - 1.10534e-85 - 1.79829e-84 - 2.86658e-83 - 4.47719e-82 - 6.85145e-81 - 1.0273e-79 - 1.5092e-78 - 2.17238e-77 - 3.0638e-76 - 4.23372e-75 - 5.7322e-74 - 7.60425e-73 - 9.88391e-72 - 1.25875e-70 - 1.57067e-69 - 1.92029e-68 - 2.30031e-67 - 2.69987e-66 - 3.10482e-65 - 3.49837e-64 - 3.86218e-63 - 4.1777e-62 - 4.42769e-61 - 4.59785e-60 - 4.6781e-59 - 4.66359e-58 - 4.55521e-57 - 4.35947e-56 - 4.08786e-55 - 3.75573e-54 - 3.38088e-53 - 2.98196e-52 - 2.57698e-51 - 2.18201e-50 - 1.81026e-49 - 1.4715e-48 - 1.17197e-47 - 9.14556e-47 - 6.99263e-46 - 5.23851e-45 - 3.84514e-44 - 2.76537e-43 - 1.94863e-42 - 1.34538e-41 - 9.10115e-41 - 6.03233e-40 - 3.91751e-39 - 2.49271e-38 - 1.55407e-37 - 9.49308e-37 - 5.68172e-36 - 3.33188e-35 - 1.91442e-34 - 1.07775e-33 - 5.94484e-33 - 3.2129e-32 - 1.70134e-31 - 8.8272e-31 - 4.48736e-30 - 2.2351e-29 - 1.09078e-28 - 5.21577e-28 - 2.44362e-27 - 1.12173e-26 - 5.0452e-26 - 2.22334e-25 - 9.6e-25 - 4.06138e-24 - 1.6835e-23 - 6.83739e-23 - 2.72085e-22 - 1.06086e-21 - 4.05272e-21 - 1.51696e-20 - 5.56339e-20 - 1.99914e-19 - 7.03857e-19 - 2.42809e-18 - 8.20698e-18 - 2.71795e-17 - 8.81938e-17 - 2.80398e-16 - 8.73476e-16 - 2.66605e-15 - 7.97306e-15 - 2.33627e-14 - 6.70756e-14 - 1.8869e-13 - 5.20086e-13 - 1.40458e-12 - 3.71676e-12 - 9.63669e-12 - 2.44816e-11 - 6.09399e-11 - 1.48633e-10 - 3.55208e-10 - 8.31775e-10 - 1.90848e-09 - 4.29072e-09 - 9.45234e-09 - 2.04041e-08 - 4.31586e-08 - 8.94536e-08 - 1.81682e-07 - 3.61591e-07 - 7.05214e-07 - 1.34782e-06 - 2.52441e-06 - 4.63355e-06 - 8.33507e-06 - 1.46946e-05 - 2.53909e-05 - 4.30015e-05 - 7.13836e-05 - 0.000116157 - 0.000185287 - 0.000289756 - 0.000444259 - 0.000667872 - 0.000984563 - 0.00142341 - 0.00201839 - 0.00280749 - 0.00383118 - 0.00512992 - 0.00674098 - 0.0086945 - 0.0110092 - 0.013688 - 0.0167144 - 0.020049 - 0.0236291 - 0.0273681 - 0.0311587 - 0.0348773 - 0.0383902 - 0.0415613 - 0.0442609 - 0.0463735 - 0.0478061 - 0.0484944 - 0.0484071 - 0.0475482 - 0.0459567 - 0.043704 - 0.0408888 - 0.0376304 - 0.0340605 - 0.0303155 - 0.0265272 - 0.022816 - 0.0192849 - 0.016015 - 0.013064 - - - - - -3.54863e-31 - -4.41618e-31 - -5.38481e-31 - -6.43324e-31 - -7.53052e-31 - -8.63675e-31 - -9.70465e-31 - -1.06803e-30 - -1.14948e-30 - -1.20073e-30 - -1.16973e-30 - -8.13407e-31 - 1.04352e-30 - 1.0015e-29 - 5.23916e-29 - 2.4876e-28 - 1.14078e-27 - 5.11125e-27 - 2.24255e-26 - 9.63929e-26 - 4.05952e-25 - 1.67509e-24 - 6.77235e-24 - 2.68273e-23 - 1.04124e-22 - 3.9597e-22 - 1.4754e-21 - 5.38636e-21 - 1.92671e-20 - 6.75267e-20 - 2.31884e-19 - 7.80195e-19 - 2.57201e-18 - 8.30765e-18 - 2.62918e-17 - 8.15268e-17 - 2.47695e-16 - 7.37345e-16 - 2.15061e-15 - 6.14593e-15 - 1.72088e-14 - 4.7212e-14 - 1.26908e-13 - 3.34245e-13 - 8.62533e-13 - 2.18084e-12 - 5.40267e-12 - 1.31139e-11 - 3.11881e-11 - 7.26749e-11 - 1.65927e-10 - 3.71181e-10 - 8.13565e-10 - 1.74717e-09 - 3.67633e-09 - 7.57934e-09 - 1.53103e-08 - 3.03023e-08 - 5.87628e-08 - 1.11652e-07 - 2.07858e-07 - 3.79145e-07 - 6.77609e-07 - 1.18656e-06 - 2.03582e-06 - 3.42235e-06 - 5.63697e-06 - 9.09714e-06 - 1.43847e-05 - 2.2286e-05 - 3.383e-05 - 5.03162e-05 - 7.33248e-05 - 0.000104696 - 0.00014647 - 0.000200771 - 0.000269645 - 0.000354829 - 0.000457491 - 0.000577942 - 0.000715355 - 0.000867554 - 0.00103088 - 0.00120021 - 0.00136913 - 0.00153027 - 0.00167583 - 0.00179817 - 0.00189049 - 0.00194743 - 0.00196562 - 0.00194402 - 0.001884 - 0.00178928 - 0.00166553 - 0.00151987 - 0.00136029 - 0.00119498 - 0.00103175 - 0.000877556 - 0.000738187 - 0.000618089 - 0.000520341 - 0.000446731 - 0.000397902 - 0.000373517 - 0.000372412 - 0.000392709 - 0.000431897 - 0.000486877 - 0.000554005 - 0.000629151 - 0.000707798 - 0.000785206 - 0.00085663 - 0.000917596 - 0.000964193 - 0.000993359 - 0.00100311 - 0.000992716 - 0.000962696 - 0.000914787 - 0.000851735 - 0.000777026 - 0.000694558 - 0.000608305 - 0.000522003 - 0.000438896 - 0.000361567 - 0.000291845 - 0.000230809 - 0.000178851 - 0.000135789 - 0.000101013 - 7.3625e-05 - 5.25788e-05 - 3.67903e-05 - 2.52227e-05 - 1.69429e-05 - 1.11512e-05 - 7.19101e-06 - 4.54356e-06 - 2.8128e-06 - 1.70616e-06 - 1.014e-06 - 5.90459e-07 - 3.36884e-07 - 1.88325e-07 - 1.03151e-07 - 5.53571e-08 - 2.91079e-08 - 1.49964e-08 - 7.57003e-09 - 3.74409e-09 - 1.81439e-09 - 8.61497e-10 - 4.00786e-10 - 1.82687e-10 - 8.15908e-11 - 3.57036e-11 - 1.53081e-11 - 6.43124e-12 - 2.64844e-12 - 1.07197e-12 - 4.34795e-13 - 2.00078e-13 - 1.63936e-13 - 3.00174e-13 - 7.46316e-13 - 1.9281e-12 - 4.92178e-12 - 1.23241e-11 - 3.02408e-11 - 7.27073e-11 - 1.71278e-10 - 3.95331e-10 - 8.94041e-10 - 1.98103e-09 - 4.30092e-09 - 9.1489e-09 - 1.90683e-08 - 3.89398e-08 - 7.79133e-08 - 1.52745e-07 - 2.93398e-07 - 5.52187e-07 - 1.01824e-06 - 1.83973e-06 - 3.25681e-06 - 5.64895e-06 - 9.60021e-06 - 1.59856e-05 - 2.60805e-05 - 4.16907e-05 - 6.52979e-05 - 0.000100207 - 0.000150671 - 0.000221973 - 0.000320411 - 0.00045316 - 0.000627962 - 0.000852611 - 0.00113424 - 0.00147842 - 0.0018881 - 0.0023626 - 0.00289663 - 0.00347962 - 0.00409551 - 0.00472303 - 0.00533667 - 0.00590822 - 0.00640885 - 0.00681147 - 0.00709314 - 0.00723725 - 0.00723514 - 0.00708696 - 0.00680167 - 0.00639614 - 0.00589358 - 0.00532131 - 0.00470844 - 0.00408352 - 0.00347255 - 0.00289752 - 0.00237559 - 0.00191887 - 0.00153483 - 0.00122706 - 0.000996256 - 0.000841365 - 0.000760476 - 0.000751552 - 0.000812793 - 0.000942642 - 0.00113944 - 0.00140081 - 0.00172285 - 0.0020993 - 0.00252088 - 0.00297478 - 0.00344473 - 0.00391135 - 0.00435317 - 0.00474797 - 0.00507448 - 0.00531415 - 0.00545286 - 0.00548225 - 0.00540051 - 0.00521259 - 0.00492967 - 0.00456811 - 0.00414791 - 0.00369091 - 0.00321908 - 0.00275291 - 0.00231029 - 0.00190581 - 0.00155063 - 0.00125284 - 0.00101825 - 0.000851451 - 0.000756966 - 0.000740438 - 0.000809583 - 0.00097481 - 0.00124937 - 0.00164889 - 0.00219032 - 0.00289009 - 0.0037618 - 0.00481339 - 0.00604416 - 0.00744202 - 0.00898139 - 0.0106221 - 0.0123098 - 0.013978 - 0.015552 - 0.0169538 - 0.0181087 - 0.0189516 - 0.019433 - 0.0195241 - 0.0192195 - 0.0185373 - 0.0175182 - 0.0162207 - 0.0147159 - 0.013081 - 0.0113928 - 0.00972205 - 0.00812871 - 0.00665921 - 0.00534516 - 0.00420373 - 0.00323926 - 0.00244565 - 0.00180917 - 0.0013113 - 0.000931238 - 0.000647972 - 0.000441762 - 0.000295092 - 0.000193136 - 0.000123853 - 7.78193e-05 - 4.79075e-05 - 2.88973e-05 - 1.70784e-05 - 9.88952e-06 - 5.61099e-06 - 3.11918e-06 - 1.69894e-06 - 9.06678e-07 - 4.74094e-07 - 2.42892e-07 - 1.21926e-07 - 5.99681e-08 - 2.88987e-08 - 1.3645e-08 - 6.31259e-09 - 2.86139e-09 - 1.27082e-09 - 5.53001e-10 - 2.3578e-10 - 9.8497e-11 - 4.03159e-11 - 1.61684e-11 - 6.35323e-12 - 2.44601e-12 - 9.22699e-13 - 3.41034e-13 - 1.23502e-13 - 4.38212e-14 - 1.52347e-14 - 5.18941e-15 - 1.73197e-15 - 5.66369e-16 - 1.81466e-16 - 5.69676e-17 - 1.75226e-17 - 5.28085e-18 - 1.55936e-18 - 4.51156e-19 - 1.27892e-19 - 3.5522e-20 - 9.66693e-21 - 2.5776e-21 - 6.73413e-22 - 1.72385e-22 - 4.3267e-23 - 1.07691e-23 - 3.17029e-24 - 3.13795e-24 - 1.04192e-23 - 4.08832e-23 - 1.59072e-22 - 6.06845e-22 - 2.26838e-21 - 8.3079e-21 - 2.98129e-20 - 1.04822e-19 - 3.6111e-19 - 1.21889e-18 - 4.03109e-18 - 1.30623e-17 - 4.14718e-17 - 1.2901e-16 - 3.93216e-16 - 1.17429e-15 - 3.43603e-15 - 9.85087e-15 - 2.76713e-14 - 7.6159e-14 - 2.05376e-13 - 5.42644e-13 - 1.40481e-12 - 3.56333e-12 - 8.85588e-12 - 2.15648e-11 - 5.1451e-11 - 1.20276e-10 - 2.75489e-10 - 6.1825e-10 - 1.35944e-09 - 2.92883e-09 - 6.18252e-09 - 1.27871e-08 - 2.5913e-08 - 5.14516e-08 - 1.00096e-07 - 1.90797e-07 - 3.56339e-07 - 6.52066e-07 - 1.16912e-06 - 2.05381e-06 - 3.53508e-06 - 5.96177e-06 - 9.85117e-06 - 1.59492e-05 - 2.53002e-05 - 3.93232e-05 - 5.98839e-05 - 8.93528e-05 - 0.00013063 - 0.000187119 - 0.000262622 - 0.000361147 - 0.000486606 - 0.000642415 - 0.000830998 - 0.00105327 - 0.00130809 - 0.00159189 - 0.00189838 - 0.00221861 - 0.00254131 - 0.00285355 - 0.00314183 - 0.00339333 - 0.00359738 - 0.00374691 - 0.00383975 - 0.00387957 - 0.00387642 - 0.00384664 - 0.00381224 - 0.0037996 - 0.00383766 - 0.00395569 - 0.00418069 - 0.0045347 - 0.00503217 - 0.00567767 - 0.00646409 - 0.00737179 - 0.00836862 - 0.00941129 - 0.0104478 - 0.011421 - 0.012273 - 0.0129503 - 0.013408 - 0.0136139 - 0.013552 - 0.0132229 - 0.0126445 - 0.011849 - 0.0108804 - 0.00978988 - 0.00863109 - 0.00745595 - 0.00631082 - 0.00523372 - 0.00425281 - 0.00338595 - 0.00264134 - 0.00201887 - 0.00151194 - 0.00110944 - 0.000797693 - 0.000562024 - 0.000388087 - 0.000262733 - 0.000174531 - 0.00011398 - 7.34973e-05 - 4.72468e-05 - 3.0892e-05 - 2.1316e-05 - 1.63489e-05 - 1.45246e-05 - 1.48771e-05 - 1.6778e-05 - 1.98118e-05 - 2.36856e-05 - 2.81654e-05 - 3.30354e-05 - 3.80753e-05 - 4.30518e-05 - 4.77214e-05 - 5.18411e-05 - 5.51843e-05 - 5.75589e-05 - 5.88241e-05 - 5.89031e-05 - 5.77908e-05 - 5.55543e-05 - 5.23255e-05 - 4.82887e-05 - 4.36632e-05 - 3.86831e-05 - 3.35788e-05 - 2.85591e-05 - 2.37992e-05 - 1.94319e-05 - 1.55456e-05 - 1.21852e-05 - 9.35833e-06 - 7.04206e-06 - 5.19204e-06 - 3.7507e-06 - 2.65476e-06 - 1.84108e-06 - 1.251e-06 - 8.32877e-07 - 5.433e-07 - 3.47245e-07 - 2.17455e-07 - 1.33426e-07 - 8.02134e-08 - 4.72494e-08 - 2.72712e-08 - 1.54258e-08 - 8.55829e-09 - 4.67484e-09 - 2.55734e-09 - 1.50298e-09 - 1.16839e-09 - 1.50985e-09 - 2.83243e-09 - 5.99601e-09 - 1.28846e-08 - 2.73441e-08 - 5.6957e-08 - 1.16287e-07 - 2.32641e-07 - 4.56022e-07 - 8.75839e-07 - 1.64816e-06 - 3.03887e-06 - 5.48987e-06 - 9.71737e-06 - 1.68528e-05 - 2.86373e-05 - 4.76792e-05 - 7.77791e-05 - 0.000124318 - 0.000194689 - 0.000298735 - 0.000449125 - 0.000661584 - 0.00095486 - 0.0013503 - 0.00187094 - 0.00253995 - 0.00337853 - 0.00440319 - 0.00562268 - 0.00703487 - 0.00862394 - 0.0103584 - 0.0121903 - 0.0140565 - 0.0158808 - 0.0175795 - 0.0190668 - 0.0202621 - 0.0210974 - 0.0215234 - 0.0215144 - 0.021071 - 0.0202198 - 0.019011 - 0.0175134 - 0.0158078 - 0.0139802 - 0.012114 - 0.0102849 - 0.00855561 - 0.00697329 - 0.00556879 - 0.00435734 - 0.00334055 - 0.0025093 - 0.00184682 - 0.00133178 - 0.00094097 - 0.000651414 - 0.000441851 - 0.00029365 - 0.000191215 - 0.000121998 - 7.62635e-05 - 4.6711e-05 - 2.80323e-05 - 1.6483e-05 - 9.49616e-06 - 5.36041e-06 - 2.96473e-06 - 1.6066e-06 - 8.53039e-07 - 4.43778e-07 - 2.26204e-07 - 1.12972e-07 - 5.52814e-08 - 2.65047e-08 - 1.2451e-08 - 5.73091e-09 - 2.58451e-09 - 1.14202e-09 - 4.94439e-10 - 2.09776e-10 - 8.72944e-11 - 3.58409e-11 - 1.50894e-11 - 7.98683e-12 - 8.47379e-12 - 1.69387e-11 - 4.04834e-11 - 9.81476e-11 - 2.34372e-10 - 5.48791e-10 - 1.2592e-09 - 2.83093e-09 - 6.23592e-09 - 1.34589e-08 - 2.84612e-08 - 5.89705e-08 - 1.19716e-07 - 2.38127e-07 - 4.64089e-07 - 8.86197e-07 - 1.65804e-06 - 3.03948e-06 - 5.45932e-06 - 9.6076e-06 - 1.65664e-05 - 2.79884e-05 - 4.63304e-05 - 7.51432e-05 - 0.000119413 - 0.000185929 - 0.00028365 - 0.000423988 - 0.000620957 - 0.00089106 - 0.00125282 - 0.00172587 - 0.0023295 - 0.00308074 - 0.00399194 - 0.00506816 - 0.00630454 - 0.0076841 - 0.00917634 - 0.010737 - 0.0123093 - 0.0138268 - 0.0152175 - 0.0164099 - 0.0173382 - 0.0179489 - 0.0182058 - 0.0180933 - 0.0176182 - 0.0168091 - 0.0157131 - 0.0143919 - 0.0129155 - 0.0113564 - 0.00978378 - 0.00825866 - 0.00683045 - 0.00553511 - 0.0043948 - 0.00341893 - 0.00260602 - 0.00194626 - 0.00142417 - 0.00102108 - 0.00071729 - 0.000493703 - 0.000332947 - 0.000219998 - 0.00014243 - 9.03484e-05 - 5.61534e-05 - 3.41955e-05 - 2.04032e-05 - 1.19279e-05 - 6.83231e-06 - 3.83449e-06 - 2.10855e-06 - 1.13605e-06 - 5.9972e-07 - 3.10196e-07 - 1.57203e-07 - 7.80587e-08 - 3.79768e-08 - 1.81031e-08 - 8.45523e-09 - 3.86931e-09 - 1.73492e-09 - 7.62186e-10 - 3.2808e-10 - 1.38368e-10 - 5.71779e-11 - 2.31504e-11 - 9.18384e-12 - 3.56967e-12 - 1.35946e-12 - 5.07276e-13 - 1.85463e-13 - 6.64368e-14 - 2.33183e-14 - 8.01901e-15 - 2.70198e-15 - 8.9203e-16 - 2.88546e-16 - 9.14505e-17 - 2.83985e-17 - 8.64052e-18 - 2.57586e-18 - 7.52387e-19 - 2.15326e-19 - 6.03798e-20 - 1.65898e-20 - 4.46888e-21 - 1.1907e-21 - 3.54724e-22 - 2.70303e-22 - 7.45922e-22 - 2.68317e-21 - 9.67303e-21 - 3.42192e-20 - 1.1862e-19 - 4.02889e-19 - 1.34076e-18 - 4.37171e-18 - 1.39666e-17 - 4.37185e-17 - 1.34084e-16 - 4.02927e-16 - 1.18635e-15 - 3.42243e-15 - 9.67374e-15 - 2.67911e-14 - 7.26984e-14 - 1.93284e-13 - 5.03504e-13 - 1.28513e-12 - 3.21388e-12 - 7.87496e-12 - 1.89062e-11 - 4.44733e-11 - 1.02502e-10 - 2.31473e-10 - 5.12165e-10 - 1.11034e-09 - 2.35855e-09 - 4.90879e-09 - 1.00102e-08 - 2.00013e-08 - 3.91579e-08 - 7.51153e-08 - 1.41185e-07 - 2.60021e-07 - 4.69236e-07 - 8.29749e-07 - 1.43776e-06 - 2.44132e-06 - 4.06237e-06 - 6.62491e-06 - 1.0589e-05 - 1.65903e-05 - 2.54819e-05 - 3.83764e-05 - 5.66822e-05 - 8.21314e-05 - 0.000116792 - 0.00016307 - 0.000223695 - 0.000301717 - 0.000400522 - 0.000523905 - 0.000676234 - 0.000862732 - 0.00108992 - 0.00136617 - 0.00170242 - 0.00211279 - 0.00261515 - 0.00323128 - 0.00398644 - 0.0049082 - 0.00602429 - 0.00735944 - 0.00893147 - 0.0107468 - 0.0127958 - 0.0150493 - 0.0174561 - 0.0199423 - 0.0224143 - 0.0247631 - 0.0268725 - 0.0286279 - 0.0299271 - 0.0306897 - 0.0308656 - 0.0304406 - 0.0294379 - 0.027917 - 0.0259682 - 0.0237049 - 0.0212543 - 0.0187473 - 0.0163094 - 0.0140532 - 0.012073 - 0.0104412 - 0.00920708 - 0.00839714 - 0.00801617 - 0.00804901 - 0.00846236 - 0.00920671 - 0.0102183 - 0.0114214 - 0.0127314 - 0.0140583 - 0.0153113 - 0.0164042 - 0.0172608 - 0.0178203 - 0.0180418 - 0.0179068 - 0.017421 - 0.0166126 - 0.0155294 - 0.0142346 - 0.0128009 - 0.0113044 - 0.00981974 - 0.00841484 - 0.00714807 - 0.00606625 - 0.00520401 - 0.00458421 - 0.00421879 - 0.00410993 - 0.00425091 - 0.00462675 - 0.00521439 - 0.00598278 - 0.0068929 - 0.00789827 - 0.00894605 - 0.00997897 - 0.0109382 - 0.0117671 - 0.0124148 - 0.0128403 - 0.0130161 - 0.01293 - 0.0125862 - 0.0120047 - 0.011219 - 0.0102732 - 0.00921717 - 0.00810268 - 0.00697907 - 0.00588986 - 0.00487025 - 0.00394581 - 0.00313231 - 0.00243638 - 0.00185693 - 0.00138696 - 0.00101542 - 0.000729103 - 0.000514093 - 0.000357004 - 0.000245774 - 0.00017014 - 0.00012182 - 9.44836e-05 - 8.35732e-05 - 8.60424e-05 - 0.000100046 - 0.000124617 - 0.000159347 - 0.000204081 - 0.000258632 - 0.000322534 - 0.000394841 - 0.000473977 - 0.000557675 - 0.000642995 - 0.000726438 - 0.000804153 - 0.00087221 - 0.00092692 - 0.000965165 - 0.000984686 - 0.000984308 - 0.000964054 - 0.000925143 - 0.000869867 - 0.000801371 - 0.000723355 - 0.000639744 - 0.000554367 - 0.00047068 - 0.000391553 - 0.000319149 - 0.000254878 - 0.000199438 - 0.000152904 - 0.00011486 - 8.45387e-05 - 6.09647e-05 - 4.30763e-05 - 2.98219e-05 - 2.02287e-05 - 1.34443e-05 - 8.75481e-06 - 5.58588e-06 - 3.49199e-06 - 2.13892e-06 - 1.2837e-06 - 7.54953e-07 - 4.3522e-07 - 2.46288e-07 - 1.37595e-07 - 7.76206e-08 - 4.7872e-08 - 3.91206e-08 - 4.96818e-08 - 8.57782e-08 - 1.64423e-07 - 3.19802e-07 - 6.1486e-07 - 1.16078e-06 - 2.1483e-06 - 3.8961e-06 - 6.92335e-06 - 1.20543e-05 - 2.0564e-05 - 3.43723e-05 - 5.6292e-05 - 9.03278e-05 - 0.000142015 - 0.000218767 - 0.000330193 - 0.000488304 - 0.000707537 - 0.00100449 - 0.00139726 - 0.00190436 - 0.00254305 - 0.00332735 - 0.00426559 - 0.00535792 - 0.00659401 - 0.00795135 - 0.00939439 - 0.0108751 - 0.0123349 - 0.013708 - 0.0149262 - 0.0159243 - 0.016646 - 0.0170489 - 0.0171089 - 0.0168223 - 0.0162065 - 0.0152982 - 0.0141497 - 0.0128242 - 0.01139 - 0.00991533 - 0.00846323 - 0.00708799 - 0.00583302 - 0.00473027 - 0.00380108 - 0.00305813 - 0.00250808 - 0.00215444 - 0.00200017 - 0.00204972 - 0.00231011 - 0.00279091 - 0.003503 - 0.00445631 - 0.00565652 - 0.00710137 - 0.0087768 - 0.0106537 - 0.0126858 - 0.014809 - 0.0169432 - 0.0189962 - 0.0208692 - 0.0224645 - 0.0236937 - 0.0244855 - 0.0247927 - 0.0245967 - 0.0239093 - 0.0227716 - 0.0212499 - 0.0194294 - 0.0174059 - 0.0152782 - 0.0131396 - 0.0110721 - 0.00914149 - 0.00739501 - 0.00586135 - 0.00455191 - 0.00346359 - 0.00258224 - 0.00188626 - 0.00135004 - 0.000946729 - 0.000650493 - 0.000437923 - 0.000288862 - 0.000186692 - 0.000118227 - 7.33688e-05 - 4.46338e-05 - 2.66504e-05 - 1.56821e-05 - 9.21728e-06 - 5.63909e-06 - 3.97946e-06 - 3.75189e-06 - 4.85588e-06 - 7.54564e-06 - 1.24576e-05 - 2.06929e-05 - 3.39515e-05 - 5.47126e-05 - 8.64515e-05 - 0.000133876 - 0.000203149 - 0.000302064 - 0.000440104 - 0.00062834 - 0.000879078 - 0.00120523 - 0.00161937 - 0.00213248 - 0.00275244 - 0.00348253 - 0.00431992 - 0.00525458 - 0.00626873 - 0.00733715 - 0.00842831 - 0.00950652 - 0.0105347 - 0.0114778 - 0.0123057 - 0.0129966 - 0.0135384 - 0.0139296 - 0.0141784 - 0.0143005 - 0.0143162 - 0.0142468 - 0.0141109 - 0.0139217 - 0.0136857 - 0.0134019 - 0.0130632 - 0.012659 - 0.0121776 - 0.0116098 - 0.0109513 - 0.0102046 - 0.00937978 - 0.00849391 - 0.00756962 - 0.00663291 - 0.00571057 - 0.00482772 - 0.00400577 - 0.00326099 - 0.00260377 - 0.00203868 - 0.00156499 - 0.00117773 - 0.000868844 - 0.000628451 - 0.000445965 - 0.00031104 - 0.00021428 - 0.000147749 - 0.000105306 - 8.28571e-05 - 7.85795e-05 - 9.31949e-05 - 0.000130355 - 0.000197161 - 0.00030483 - 0.000469447 - 0.000712728 - 0.00106262 - 0.00155357 - 0.00222613 - 0.00312575 - 0.00430043 - 0.00579713 - 0.0076569 - 0.00990902 - 0.0125645 - 0.0156098 - 0.0190014 - 0.0226627 - 0.0264835 - 0.0303231 - 0.0340181 - 0.0373924 - 0.0402711 - 0.0424952 - 0.0439363 - 0.0445086 - 0.0441776 - 0.0429631 - 0.040938 - 0.0382204 - 0.0349623 - 0.0313358 - 0.0275182 - 0.0236774 - 0.0199612 - 0.0164883 - 0.0133445 - 0.0105819 - 0.00822174 - 0.00625893 - 0.00466846 - 0.0034118 - 0.00244304 - 0.00171401 - 0.00117824 - 0.000793582 - 0.000523705 - 0.000338624 - 0.000214529 - 0.000133165 - 8.09901e-05 - 4.82626e-05 - 2.8179e-05 - 1.61205e-05 - 9.03581e-06 - 4.96241e-06 - 2.67027e-06 - 1.40785e-06 - 7.27263e-07 - 3.68099e-07 - 1.82547e-07 - 8.87001e-08 - 4.22295e-08 - 1.97008e-08 - 9.00956e-09 - 4.04833e-09 - 1.81029e-09 - 8.60698e-10 - 5.58336e-10 - 6.85971e-10 - 1.31585e-09 - 2.8408e-09 - 6.1788e-09 - 1.32343e-08 - 2.77987e-08 - 5.7221e-08 - 1.15408e-07 - 2.28062e-07 - 4.4158e-07 - 8.37727e-07 - 1.55716e-06 - 2.83595e-06 - 5.0606e-06 - 8.84794e-06 - 1.51572e-05 - 2.5441e-05 - 4.18393e-05 - 6.74174e-05 - 0.000106438 - 0.000164648 - 0.000249549 - 0.000370588 - 0.000539216 - 0.000768726 - 0.00107378 - 0.0014696 - 0.00197069 - 0.00258926 - 0.00333325 - 0.00420434 - 0.00519594 - 0.0062917 - 0.00746463 - 0.00867732 - 0.00988325 - 0.0110294 - 0.0120597 - 0.01292 - 0.013562 - 0.0139483 - 0.0140559 - 0.0138781 - 0.0134257 - 0.0127258 - 0.0118186 - 0.0107544 - 0.00958832 - 0.00837598 - 0.00716911 - 0.00601218 - 0.0049401 - 0.00397719 - 0.00313728 - 0.00242475 - 0.00183619 - 0.00136241 - 0.000990447 - 0.000705493 - 0.00049237 - 0.000336687 - 0.000225579 - 0.000148084 - 9.52474e-05 - 6.00255e-05 - 3.70643e-05 - 2.24239e-05 - 1.32924e-05 - 7.72031e-06 - 4.39341e-06 - 2.44966e-06 - 1.33828e-06 - 7.16347e-07 - 3.75697e-07 - 1.93059e-07 - 9.7204e-08 - 4.79558e-08 - 2.31886e-08 - 1.10053e-08 - 5.16537e-09 - 2.49231e-09 - 1.45573e-09 - 1.44813e-09 - 2.49265e-09 - 5.27515e-09 - 1.15271e-08 - 2.49327e-08 - 5.29399e-08 - 1.10176e-07 - 2.24677e-07 - 4.48922e-07 - 8.78862e-07 - 1.68581e-06 - 3.16834e-06 - 5.83436e-06 - 1.05267e-05 - 1.86091e-05 - 3.22327e-05 - 5.47023e-05 - 9.09601e-05 - 0.000148195 - 0.000236566 - 0.000370005 - 0.000567022 - 0.000851392 - 0.00125255 - 0.00180551 - 0.00254999 - 0.00352871 - 0.00478442 - 0.00635594 - 0.00827308 - 0.010551 - 0.0131842 - 0.0161418 - 0.0193636 - 0.0227592 - 0.02621 - 0.0295742 - 0.0326961 - 0.0354175 - 0.0375905 - 0.0390913 - 0.0398314 - 0.0397669 - 0.0389028 - 0.0372926 - 0.0350338 - 0.0322585 - 0.0291224 - 0.0257915 - 0.02243 - 0.0191902 - 0.0162049 - 0.0135829 - 0.0114083 - 0.00974183 - 0.00862375 - 0.00807727 - 0.00811156 - 0.00872333 - 0.00989679 - 0.011602 - 0.0137919 - 0.0163994 - 0.0193348 - 0.0224847 - 0.0257144 - 0.0288725 - 0.0317988 - 0.0343349 - 0.0363363 - 0.0376843 - 0.0382965 - 0.0381344 - 0.037207 - 0.0355694 - 0.0333172 - 0.0305773 - 0.0274959 - 0.0242255 - 0.020913 - 0.0176887 - 0.0146593 - 0.0119033 - 0.00947019 - 0.00738221 - 0.00563834 - 0.00421942 - 0.0030938 - 0.00222264 - 0.00156455 - 0.0010791 - 0.000729329 - 0.000483146 - 0.000313949 - 0.000200581 - 0.000126914 - 8.12404e-05 - 5.5656e-05 - 4.55669e-05 - 4.94212e-05 - 6.87538e-05 - 0.000108612 - 0.000178427 - 0.000293379 - 0.000476297 - 0.000760061 - 0.00119043 - 0.00182912 - 0.00275668 - 0.00407487 - 0.00590754 - 0.00839955 - 0.0117125 - 0.0160171 - 0.0214803 - 0.0282497 - 0.0364326 - 0.0460744 - 0.0571362 - 0.0694755 - 0.0828344 - 0.0968356 - 0.110993 - 0.124731 - 0.137426 - 0.148443 - 0.157195 - 0.163191 - 0.16608 - 0.16569 - 0.162039 - 0.155338 - 0.145969 - 0.13445 - 0.121386 - 0.107417 - 0.0931679 - 0.0792032 - 0.0659922 - 0.0538902 - 0.0431307 - 0.0338313 - 0.0260076 - 0.0195944 - 0.0144686 - 0.0104718 - 0.00743065 - 0.00517275 - 0.00353869 - 0.0023892 - 0.00160904 - 0.00110793 - 0.000819713 - 0.000700487 - 0.000726235 - 0.000890471 - 0.00120196 - 0.00168246 - 0.00236436 - 0.00328777 - 0.00449696 - 0.00603582 - 0.00794235 - 0.0102424 - 0.0129428 - 0.0160253 - 0.0194414 - 0.0231094 - 0.0269146 - 0.0307131 - 0.0343397 - 0.0376189 - 0.0403787 - 0.0424655 - 0.0437579 - 0.0441788 - 0.0437027 - 0.0423583 - 0.040226 - 0.0374293 - 0.0341234 - 0.0304811 - 0.0266776 - 0.022877 - 0.0192215 - 0.0158239 - 0.0127637 - 0.0100873 - 0.00781109 - 0.00592634 - 0.00440555 - 0.00320891 - 0.0022902 - 0.00160171 - 0.00109801 - 0.000738346 - 0.000488074 - 0.000319114 - 0.000209886 - 0.000144899 - 0.000114287 - 0.00011345 - 0.000142967 - 0.000208865 - 0.000323272 - 0.000505441 - 0.000783052 - 0.00119362 - 0.00178573 - 0.00261982 - 0.00376788 - 0.00531191 - 0.00734032 - 0.0099422 - 0.0131993 - 0.0171757 - 0.0219067 - 0.0273861 - 0.0335566 - 0.0403008 - 0.0474392 - 0.0547326 - 0.0618925 - 0.0685978 - 0.074518 - 0.0793391 - 0.0827917 - 0.0846753 - 0.084878 - 0.0833871 - 0.0802907 - 0.0757691 - 0.0700771 - 0.0635207 - 0.0564296 - 0.0491301 - 0.0419213 - 0.0350564 - 0.0287303 - 0.0230756 - 0.0181636 - 0.0140114 - 0.0105924 - 0.00784763 - 0.00569779 - 0.00405413 - 0.0028269 - 0.00193171 - 0.00129357 - 0.000848892 - 0.000545916 - 0.00034404 - 0.000212471 - 0.000128587 - 7.62598e-05 - 4.43196e-05 - 2.52403e-05 - 1.40861e-05 - 7.70341e-06 - 4.12826e-06 - 2.16792e-06 - 1.1156e-06 - 5.62553e-07 - 2.77973e-07 - 1.34595e-07 - 6.38609e-08 - 2.96909e-08 - 1.35267e-08 - 6.03865e-09 - 2.64158e-09 - 1.13231e-09 - 4.75596e-10 - 1.95742e-10 - 7.89412e-11 - 3.11955e-11 - 1.20796e-11 - 4.58331e-12 - 1.70407e-12 - 6.20988e-13 - 2.22293e-13 - 7.9713e-14 - 3.33658e-14 - 2.94752e-14 - 6.38018e-14 - 1.77569e-13 - 5.02538e-13 - 1.39959e-12 - 3.82147e-12 - 1.02254e-11 - 2.6812e-11 - 6.88931e-11 - 1.73468e-10 - 4.28014e-10 - 1.03489e-09 - 2.45205e-09 - 5.69327e-09 - 1.29536e-08 - 2.88814e-08 - 6.31019e-08 - 1.35103e-07 - 2.83456e-07 - 5.82778e-07 - 1.17414e-06 - 2.3181e-06 - 4.48481e-06 - 8.50263e-06 - 1.57965e-05 - 2.87584e-05 - 5.13057e-05 - 8.96943e-05 - 0.00015366 - 0.000257961 - 0.00042437 - 0.000684119 - 0.00108072 - 0.001673 - 0.00253788 - 0.00377261 - 0.00549553 - 0.00784462 - 0.0109731 - 0.0150412 - 0.0202036 - 0.0265931 - 0.0343008 - 0.0433543 - 0.0536975 - 0.0651733 - 0.0775138 - 0.0903404 - 0.103176 - 0.115472 - 0.12664 - 0.136103 - 0.143344 - 0.147949 - 0.149655 - 0.14837 - 0.144186 - 0.137373 - 0.128353 - 0.117658 - 0.105889 - 0.093658 - 0.0815473 - 0.0700646 - 0.0596157 - 0.0504871 - 0.0428422 - 0.0367268 - 0.0320839 - 0.0287735 - 0.0265961 - 0.0253153 - 0.0246804 - 0.024446 - 0.0243882 - 0.0243168 - 0.0240838 - 0.0235875 - 0.0227725 - 0.021627 - 0.0201755 - 0.0184715 - 0.0165872 - 0.014604 - 0.0126035 - 0.0106603 - 0.00883602 - 0.00717679 - 0.00571177 - 0.00445418 - 0.00340341 - 0.00254803 - 0.00186912 - 0.00134342 - 0.000946069 - 0.000652789 - 0.000441327 - 0.000292338 - 0.000189735 - 0.000120655 - 7.51766e-05 - 4.5894e-05 - 2.74515e-05 - 1.60884e-05 - 9.2384e-06 - 5.19778e-06 - 2.86533e-06 - 1.54764e-06 - 8.19038e-07 - 4.24702e-07 - 2.15801e-07 - 1.07503e-07 - 5.26328e-08 - 2.56405e-08 - 1.31726e-08 - 8.77456e-09 - 1.02078e-08 - 1.85691e-08 - 3.89947e-08 - 8.33138e-08 - 1.75745e-07 - 3.63781e-07 - 7.38012e-07 - 1.46706e-06 - 2.85743e-06 - 5.45305e-06 - 1.01963e-05 - 1.86801e-05 - 3.35317e-05 - 5.89749e-05 - 0.000101629 - 0.000171594 - 0.000283873 - 0.000460134 - 0.00073077 - 0.00113714 - 0.00173374 - 0.00258995 - 0.00379084 - 0.00543645 - 0.00763894 - 0.0105169 - 0.0141866 - 0.0187503 - 0.0242813 - 0.0308088 - 0.0383013 - 0.0466542 - 0.0556807 - 0.0651114 - 0.0746018 - 0.0837495 - 0.0921213 - 0.0992861 - 0.104852 - 0.108503 - 0.110031 - 0.109357 - 0.106543 - 0.101791 - 0.0954271 - 0.087877 - 0.0796374 - 0.0712426 - 0.0632336 - 0.0561318 - 0.0504187 - 0.0465218 - 0.044805 - 0.045561 - 0.0490042 - 0.0552601 - 0.0643534 - 0.0761915 - 0.0905495 - 0.107057 - 0.125194 - 0.144296 - 0.163577 - 0.182163 - 0.199146 - 0.213648 - 0.224879 - 0.232211 - 0.235225 - 0.233747 - 0.227864 - 0.217912 - 0.204443 - 0.188177 - 0.169934 - 0.150572 - 0.130918 - 0.111717 - 0.0935927 - 0.0770224 - 0.0623326 - 0.0497065 - 0.0392039 - 0.0307866 - 0.0243474 - 0.0197367 - 0.0167861 - 0.0153248 - 0.0151908 - 0.0162352 - 0.0183228 - 0.0213286 - 0.0251345 - 0.0296256 - 0.0346886 - 0.0402135 - 0.0460961 - 0.052243 - 0.0585756 - 0.0650326 - 0.0715684 - 0.0781472 - 0.0847335 - 0.091278 - 0.0977038 - 0.103894 - 0.109684 - 0.114865 - 0.119189 - 0.122391 - 0.124217 - 0.124448 - 0.122933 - 0.119609 - 0.114518 - 0.107811 - 0.0997331 - 0.0906066 - 0.0808031 - 0.0707102 - 0.0607003 - 0.0511033 - 0.0421863 - 0.0341421 - 0.0270865 - 0.0210628 - 0.0160525 - 0.0119898 - 0.00877597 - 0.00629478 - 0.00442438 - 0.00304721 - 0.00205646 - 0.00135988 - 0.000881132 - 0.000559434 - 0.000348064 - 0.000212277 - 0.000127044 - 7.49014e-05 - 4.40862e-05 - 2.70444e-05 - 1.93388e-05 - 1.89574e-05 - 2.60199e-05 - 4.28955e-05 - 7.47722e-05 - 0.00013075 - 0.000225561 - 0.000382019 - 0.000634315 - 0.00103218 - 0.00164583 - 0.00257148 - 0.00393681 - 0.00590566 - 0.00868071 - 0.0125027 - 0.0176447 - 0.0243998 - 0.0330614 - 0.0438952 - 0.0571052 - 0.0727939 - 0.0909236 - 0.111281 - 0.133452 - 0.156816 - 0.180558 - 0.203706 - 0.225193 - 0.24393 - 0.258902 - 0.269258 - 0.274385 - 0.273977 - 0.268058 - 0.256982 - 0.241401 - 0.222195 - 0.200397 - 0.177096 - 0.153352 - 0.130119 - 0.108184 - 0.088143 - 0.0703811 - 0.0550906 - 0.0422952 - 0.0318886 - 0.0236757 - 0.0174137 - 0.0128489 - 0.00974575 - 0.00790899 - 0.00719695 - 0.00752776 - 0.00887847 - 0.0112782 - 0.0147954 - 0.0195211 - 0.0255467 - 0.03294 - 0.0417194 - 0.0518294 - 0.0631202 - 0.0753344 - 0.0881046 - 0.100962 - 0.113362 - 0.124714 - 0.134431 - 0.141978 - 0.146921 - 0.148964 - 0.147984 - 0.144041 - 0.137371 - 0.128363 - 0.117523 - 0.105425 - 0.0926619 - 0.0797996 - 0.0673356 - 0.0556729 - 0.0451043 - 0.0358108 - 0.0278695 - 0.0212705 - 0.0159377 - 0.0117507 - 0.0085666 - 0.0062372 - 0.00462324 - 0.00360352 - 0.00307992 - 0.00297862 - 0.00324833 - 0.00385612 - 0.00478184 - 0.00601151 - 0.00753051 - 0.0093171 - 0.011337 - 0.0135396 - 0.0158561 - 0.0182006 - 0.0204731 - 0.0225658 - 0.0243709 - 0.0257892 - 0.0267388 - 0.0271636 - 0.0270377 - 0.0263689 - 0.0251973 - 0.0235919 - 0.0216433 - 0.019456 - 0.0171389 - 0.014797 - 0.0125243 - 0.0103983 - 0.00847798 - 0.00680284 - 0.00539503 - 0.00426232 - 0.00340205 - 0.00280523 - 0.00246013 - 0.00235501 - 0.00247968 - 0.00282577 - 0.00338583 - 0.00415139 - 0.00511045 - 0.00624471 - 0.00752714 - 0.00892027 - 0.0103758 - 0.0118353 - 0.0132334 - 0.0145009 - 0.0155705 - 0.0163822 - 0.0168885 - 0.017059 - 0.0168832 - 0.0163718 - 0.0155551 - 0.0144807 - 0.0132081 - 0.011804 - 0.0103361 - 0.00886787 - 0.00745451 - 0.00613983 - 0.00495484 - 0.00391778 - 0.0030352 - 0.00230394 - 0.00171353 - 0.00124868 - 0.000891548 - 0.000623701 - 0.000427509 - 0.000287111 - 0.000188926 - 0.000121807 - 7.69461e-05 - 4.76255e-05 - 2.88821e-05 - 1.71615e-05 - 9.99121e-06 - 5.69926e-06 - 3.18533e-06 - 1.74433e-06 - 9.35921e-07 - 4.92024e-07 - 2.53438e-07 - 1.27906e-07 - 6.32485e-08 - 3.06439e-08 - 1.45471e-08 - 6.7662e-09 - 3.08354e-09 - 1.37687e-09 - 6.02381e-10 - 2.58218e-10 - 1.08453e-10 - 4.46303e-11 - 1.79952e-11 - 7.10917e-12 - 2.75181e-12 - 1.04365e-12 - 3.8782e-13 - 1.41202e-13 - 5.03719e-14 - 1.76065e-14 - 6.02967e-15 - 2.02326e-15 - 6.65191e-16 - 2.14278e-16 - 6.76311e-17 - 2.09147e-17 - 6.33715e-18 - 1.88137e-18 - 5.47254e-19 - 1.5597e-19 - 4.35544e-20 - 1.19168e-20 - 3.19464e-21 - 8.39116e-22 - 2.15953e-22 - 5.44544e-23 - 1.34538e-23 - 3.2568e-24 - 7.72459e-25 - 1.79513e-25 - 4.08747e-26 - 9.11906e-27 - 1.99334e-27 - 4.26924e-28 - 8.95892e-29 - 1.84204e-29 - 3.71088e-30 - 7.32476e-31 - 1.4166e-31 - 2.68433e-32 - 4.98382e-33 - 9.06622e-34 - 1.61594e-34 - 2.82204e-35 - 4.82878e-36 - 8.09559e-37 - 1.32983e-37 - 2.14033e-38 - 3.37522e-39 - 5.21507e-40 - 7.89506e-41 - 1.17108e-41 - 1.70199e-42 - 2.4236e-43 - 3.38146e-44 - 4.62257e-45 - 6.19155e-46 - 8.12554e-47 - 1.04482e-47 - 1.31634e-48 - 1.62492e-49 - 1.96532e-50 - 2.329e-51 - 2.70423e-52 - 3.07649e-53 - 3.42928e-54 - 3.74531e-55 - 4.00783e-56 - 4.20211e-57 - 4.31681e-58 - 4.34505e-59 - 4.28512e-60 - 4.14066e-61 - 3.92023e-62 - 3.63656e-63 - 3.30527e-64 - 2.94348e-65 - 2.56833e-66 - 2.19572e-67 - 1.83925e-68 - 1.50953e-69 - 1.21389e-70 - 9.56429e-72 - 7.38353e-73 - 5.58485e-74 - 4.13901e-75 - 3.00551e-76 - 2.13834e-77 - 1.49064e-78 - 1.01813e-79 - 6.81354e-81 - 4.46764e-82 - 2.87026e-83 - 1.80676e-84 - 1.11434e-85 - 6.73396e-87 - 3.98713e-88 - 2.31306e-89 - 1.31477e-90 - 7.32237e-92 - 3.99566e-93 - 2.1363e-94 - 1.11911e-95 - 5.7441e-97 - 2.88873e-98 - 1.42341e-99 - 6.87207e-101 - 3.25074e-102 - 1.50666e-103 - 6.84201e-105 - 3.04432e-106 - 1.32719e-107 - 5.66909e-109 - 2.37623e-110 - 1.87428e-111 - 2.20949e-110 - 5.28801e-109 - 1.24221e-107 - 2.85914e-106 - 6.44781e-105 - 1.42471e-103 - 3.08445e-102 - 6.54282e-101 - 1.35985e-99 - 2.76918e-98 - 5.52523e-97 - 1.08015e-95 - 2.06899e-94 - 3.88299e-93 - 7.14023e-92 - 1.28646e-90 - 2.27099e-89 - 3.92799e-88 - 6.65678e-87 - 1.10534e-85 - 1.79829e-84 - 2.86658e-83 - 4.47719e-82 - 6.85145e-81 - 1.0273e-79 - 1.5092e-78 - 2.17238e-77 - 3.0638e-76 - 4.23372e-75 - 5.7322e-74 - 7.60425e-73 - 9.88391e-72 - 1.25875e-70 - 1.57067e-69 - 1.92029e-68 - 2.30031e-67 - 2.69987e-66 - 3.10482e-65 - 3.49837e-64 - 3.86218e-63 - 4.1777e-62 - 4.42769e-61 - 4.59785e-60 - 4.6781e-59 - 4.66359e-58 - 4.55521e-57 - 4.35947e-56 - 4.08786e-55 - 3.75573e-54 - 3.38088e-53 - 2.98196e-52 - 2.57698e-51 - 2.18201e-50 - 1.81026e-49 - 1.4715e-48 - 1.17197e-47 - 9.14556e-47 - 6.99263e-46 - 5.23851e-45 - 3.84514e-44 - 2.76537e-43 - 1.94863e-42 - 1.34538e-41 - 9.10115e-41 - 6.03233e-40 - 3.91751e-39 - 2.49271e-38 - 1.55407e-37 - 9.49308e-37 - 5.68172e-36 - 3.33188e-35 - 1.91442e-34 - 1.07775e-33 - 5.94484e-33 - 3.2129e-32 - 1.70134e-31 - 8.8272e-31 - 4.48736e-30 - 2.2351e-29 - 1.09078e-28 - 5.21577e-28 - 2.44362e-27 - 1.12173e-26 - 5.0452e-26 - 2.22334e-25 - 9.6e-25 - 4.06138e-24 - 1.6835e-23 - 6.83739e-23 - 2.72085e-22 - 1.06086e-21 - 4.05272e-21 - 1.51696e-20 - 5.56339e-20 - 1.99914e-19 - 7.03857e-19 - 2.42809e-18 - 8.20698e-18 - 2.71795e-17 - 8.81938e-17 - 2.80398e-16 - 8.73476e-16 - 2.66605e-15 - 7.97306e-15 - 2.33627e-14 - 6.70756e-14 - 1.8869e-13 - 5.20086e-13 - 1.40458e-12 - 3.71676e-12 - 9.63669e-12 - 2.44816e-11 - 6.09399e-11 - 1.48633e-10 - 3.55208e-10 - 8.31775e-10 - 1.90848e-09 - 4.29072e-09 - 9.45234e-09 - 2.04041e-08 - 4.31586e-08 - 8.94536e-08 - 1.81682e-07 - 3.61591e-07 - 7.05214e-07 - 1.34782e-06 - 2.52441e-06 - 4.63355e-06 - 8.33507e-06 - 1.46946e-05 - 2.53909e-05 - 4.30015e-05 - 7.13836e-05 - 0.000116157 - 0.000185287 - 0.000289756 - 0.000444259 - 0.000667872 - 0.000984563 - 0.00142341 - 0.00201839 - 0.00280749 - 0.00383118 - 0.00512992 - 0.00674098 - 0.0086945 - 0.0110092 - 0.013688 - 0.0167144 - 0.020049 - 0.0236291 - 0.0273681 - 0.0311587 - 0.0348773 - 0.0383902 - 0.0415613 - 0.0442609 - 0.0463735 - 0.0478061 - 0.0484944 - 0.0484071 - 0.0475482 - 0.0459567 - 0.043704 - 0.0408888 - 0.0376304 - 0.0340605 - 0.0303155 - 0.0265272 - 0.022816 - 0.0192849 - 0.016015 - 0.013064 - - - - - -5.10468e-35 - -6.35265e-35 - -7.74601e-35 - -9.25418e-35 - -1.08327e-34 - -1.24242e-34 - -1.39617e-34 - -1.53726e-34 - -1.6584e-34 - -1.75296e-34 - -1.81547e-34 - -1.84223e-34 - -1.83162e-34 - -1.78428e-34 - -1.70306e-34 - -1.59269e-34 - -1.45939e-34 - -1.31023e-34 - -1.15255e-34 - -9.93364e-35 - -8.3887e-35 - -6.94094e-35 - -5.62701e-35 - -4.46966e-35 - -3.47863e-35 - -2.65265e-35 - -1.98192e-35 - -1.45088e-35 - -1.04065e-35 - -7.31273e-36 - -5.03083e-36 - -3.36688e-36 - -2.06623e-36 - -4.25511e-37 - 4.85055e-36 - 3.11622e-35 - 1.7057e-34 - 9.01708e-34 - 4.66301e-33 - 2.36224e-32 - 1.17249e-31 - 5.70206e-31 - 2.717e-30 - 1.26849e-29 - 5.80254e-29 - 2.60068e-28 - 1.14207e-27 - 4.914e-27 - 2.07164e-26 - 8.55719e-26 - 3.46326e-25 - 1.37333e-24 - 5.33582e-24 - 2.03126e-23 - 7.57645e-23 - 2.76888e-22 - 9.91466e-22 - 3.47848e-21 - 1.19574e-20 - 4.02737e-20 - 1.32906e-19 - 4.29736e-19 - 1.36144e-18 - 4.226e-18 - 1.28528e-17 - 3.83005e-17 - 1.11827e-16 - 3.19909e-16 - 8.96693e-16 - 2.46262e-15 - 6.62654e-15 - 1.74709e-14 - 4.51313e-14 - 1.1423e-13 - 2.8328e-13 - 6.88321e-13 - 1.63871e-12 - 3.82252e-12 - 8.73645e-12 - 1.9564e-11 - 4.29255e-11 - 9.22807e-11 - 1.94376e-10 - 4.01155e-10 - 8.11183e-10 - 1.60717e-09 - 3.1199e-09 - 5.93414e-09 - 1.10589e-08 - 2.0193e-08 - 3.61267e-08 - 6.33275e-08 - 1.08766e-07 - 1.83034e-07 - 3.01791e-07 - 4.87548e-07 - 7.7173e-07 - 1.19688e-06 - 1.81875e-06 - 2.70789e-06 - 3.95027e-06 - 5.64623e-06 - 7.9073e-06 - 1.08501e-05 - 1.45874e-05 - 1.92157e-05 - 2.48013e-05 - 3.13637e-05 - 3.88613e-05 - 4.71786e-05 - 5.61189e-05 - 6.54049e-05 - 7.46875e-05 - 8.35646e-05 - 9.16081e-05 - 9.83971e-05 - 0.000103554 - 0.00010678 - 0.000107882 - 0.000106793 - 0.00010358 - 9.84341e-05 - 9.16541e-05 - 8.36171e-05 - 7.47438e-05 - 6.54624e-05 - 5.61753e-05 - 4.7232e-05 - 3.89102e-05 - 3.14071e-05 - 2.48387e-05 - 1.92471e-05 - 1.46131e-05 - 1.08706e-05 - 7.92321e-06 - 5.6583e-06 - 3.95921e-06 - 2.71436e-06 - 1.82332e-06 - 1.20004e-06 - 7.73865e-07 - 4.88958e-07 - 3.02702e-07 - 1.83609e-07 - 1.09122e-07 - 6.35427e-08 - 3.6254e-08 - 2.02667e-08 - 1.11006e-08 - 5.95729e-09 - 3.13247e-09 - 1.61384e-09 - 8.14654e-10 - 4.02923e-10 - 1.95257e-10 - 9.27106e-11 - 4.31309e-11 - 1.966e-11 - 8.78045e-12 - 3.84226e-12 - 1.64738e-12 - 6.92046e-13 - 2.84844e-13 - 1.14858e-13 - 4.53382e-14 - 1.74196e-14 - 6.23751e-15 - 1.31174e-15 - -2.20244e-15 - -7.77776e-15 - -2.0591e-14 - -5.18126e-14 - -1.27222e-13 - -3.05906e-13 - -7.20636e-13 - -1.66332e-12 - -3.76161e-12 - -8.33502e-12 - -1.80958e-11 - -3.84932e-11 - -8.02285e-11 - -1.63836e-10 - -3.27814e-10 - -6.42661e-10 - -1.23445e-09 - -2.32328e-09 - -4.28417e-09 - -7.7405e-09 - -1.37028e-08 - -2.37675e-08 - -4.03921e-08 - -6.72583e-08 - -1.09732e-07 - -1.7541e-07 - -2.74736e-07 - -4.21611e-07 - -6.33936e-07 - -9.33934e-07 - -1.3481e-06 - -1.90664e-06 - -2.6421e-06 - -3.58729e-06 - -4.77223e-06 - -6.22032e-06 - -7.94404e-06 - -9.94047e-06 - -1.21873e-05 - -1.46402e-05 - -1.72315e-05 - -1.98718e-05 - -2.24536e-05 - -2.48583e-05 - -2.69647e-05 - -2.86586e-05 - -2.98437e-05 - -3.04499e-05 - -3.04409e-05 - -2.9817e-05 - -2.8616e-05 - -2.69085e-05 - -2.47918e-05 - -2.23801e-05 - -1.97949e-05 - -1.71547e-05 - -1.45663e-05 - -1.21185e-05 - -9.87847e-06 - -7.88979e-06 - -6.17417e-06 - -4.734e-06 - -3.55643e-06 - -2.61781e-06 - -1.88798e-06 - -1.33412e-06 - -9.23695e-07 - -6.26613e-07 - -4.16492e-07 - -2.71238e-07 - -1.73074e-07 - -1.08205e-07 - -6.62806e-08 - -3.97758e-08 - -2.33778e-08 - -1.34388e-08 - -7.51457e-09 - -3.99226e-09 - -1.79761e-09 - -1.55833e-10 - 1.62647e-09 - 4.40583e-09 - 9.54109e-09 - 1.94299e-08 - 3.83984e-08 - 7.41803e-08 - 1.40332e-07 - 2.60078e-07 - 4.7225e-07 - 8.40183e-07 - 1.46457e-06 - 2.50142e-06 - 4.18598e-06 - 6.86349e-06 - 1.10263e-05 - 1.73561e-05 - 2.67677e-05 - 4.04488e-05 - 5.98878e-05 - 8.68775e-05 - 0.000123485 - 0.000171971 - 0.000234658 - 0.000313728 - 0.000410966 - 0.000527468 - 0.00066332 - 0.000817311 - 0.000986707 - 0.00116715 - 0.0013527 - 0.00153607 - 0.00170907 - 0.00186314 - 0.00199006 - 0.00208269 - 0.0021356 - 0.00214562 - 0.00211214 - 0.00203718 - 0.00192518 - 0.00178259 - 0.00161722 - 0.00143755 - 0.00125202 - 0.00106841 - 0.000893312 - 0.000731819 - 0.00058741 - 0.000461973 - 0.000355982 - 0.000268767 - 0.00019882 - 0.000144106 - 0.000102339 - 7.12094e-05 - 4.85478e-05 - 3.24294e-05 - 2.12249e-05 - 1.3611e-05 - 8.55202e-06 - 5.26484e-06 - 3.1757e-06 - 1.87685e-06 - 1.08682e-06 - 6.16624e-07 - 3.42785e-07 - 1.86707e-07 - 9.96401e-08 - 5.2101e-08 - 2.66928e-08 - 1.33992e-08 - 6.59024e-09 - 3.17585e-09 - 1.49953e-09 - 6.93727e-10 - 3.14455e-10 - 1.39658e-10 - 6.07726e-11 - 2.59112e-11 - 1.08244e-11 - 4.43055e-12 - 1.77684e-12 - 6.98194e-13 - 2.68807e-13 - 1.01401e-13 - 3.74782e-14 - 1.35723e-14 - 4.81577e-15 - 1.67423e-15 - 5.70295e-16 - 1.90336e-16 - 6.22416e-17 - 1.99424e-17 - 6.2605e-18 - 1.92566e-18 - 5.80343e-19 - 1.71367e-19 - 4.95802e-20 - 1.40548e-20 - 3.90372e-21 - 1.06236e-21 - 2.83268e-22 - 7.4006e-23 - 1.89476e-23 - 4.76861e-24 - 1.24276e-24 - 5.98879e-25 - 1.38172e-24 - 5.3505e-24 - 2.12055e-23 - 8.25555e-23 - 3.14952e-22 - 1.17729e-21 - 4.3118e-21 - 1.54729e-20 - 5.44027e-20 - 1.87416e-19 - 6.32602e-19 - 2.09214e-18 - 6.77934e-18 - 2.15239e-17 - 6.69563e-17 - 2.04079e-16 - 6.09457e-16 - 1.7833e-15 - 5.1126e-15 - 1.43614e-14 - 3.95265e-14 - 1.0659e-13 - 2.81632e-13 - 7.29096e-13 - 1.84937e-12 - 4.59621e-12 - 1.11921e-11 - 2.67031e-11 - 6.24235e-11 - 1.42979e-10 - 3.20872e-10 - 7.05551e-10 - 1.52007e-09 - 3.20873e-09 - 6.63652e-09 - 1.34488e-08 - 2.67034e-08 - 5.19498e-08 - 9.90237e-08 - 1.8494e-07 - 3.38423e-07 - 6.06771e-07 - 1.06593e-06 - 1.8347e-06 - 3.09416e-06 - 5.11276e-06 - 8.27761e-06 - 1.31308e-05 - 2.04087e-05 - 3.10796e-05 - 4.63738e-05 - 6.77965e-05 - 9.71133e-05 - 0.000136297 - 0.000187428 - 0.000252533 - 0.00033338 - 0.000431223 - 0.000546517 - 0.000678654 - 0.00082573 - 0.000984414 - 0.00114994 - 0.00131625 - 0.00147635 - 0.00162275 - 0.00174809 - 0.00184582 - 0.00191084 - 0.00194004 - 0.00193274 - 0.00189081 - 0.00181863 - 0.00172275 - 0.00161135 - 0.00149356 - 0.00137872 - 0.00127559 - 0.00119165 - 0.00113253 - 0.00110163 - 0.00109985 - 0.00112561 - 0.001175 - 0.00124218 - 0.00131983 - 0.00139982 - 0.00147384 - 0.00153416 - 0.00157416 - 0.00158894 - 0.00157562 - 0.00153351 - 0.00146403 - 0.00137047 - 0.00125758 - 0.00113103 - 0.000996861 - 0.000860974 - 0.000728651 - 0.000604241 - 0.000490968 - 0.000390879 - 0.000304913 - 0.000233051 - 0.000174528 - 0.000128061 - 9.20676e-05 - 6.48537e-05 - 4.47611e-05 - 3.02693e-05 - 2.00559e-05 - 1.30202e-05 - 8.28195e-06 - 5.16159e-06 - 3.15189e-06 - 1.8858e-06 - 1.10549e-06 - 6.34971e-07 - 3.57346e-07 - 1.97043e-07 - 1.06456e-07 - 5.63526e-08 - 2.92278e-08 - 1.4853e-08 - 7.39555e-09 - 3.60797e-09 - 1.72462e-09 - 8.07717e-10 - 3.70649e-10 - 1.66649e-10 - 7.34141e-11 - 3.16879e-11 - 1.34012e-11 - 5.55305e-12 - 2.25453e-12 - 8.96844e-13 - 3.49555e-13 - 1.3349e-13 - 4.99484e-14 - 1.83118e-14 - 6.57773e-15 - 2.31504e-15 - 7.98324e-16 - 2.69746e-16 - 8.9342e-17 - 2.9128e-17 - 9.76341e-18 - 4.72724e-18 - 6.95038e-18 - 1.96983e-17 - 6.09995e-17 - 1.87121e-16 - 5.62997e-16 - 1.65984e-15 - 4.79478e-15 - 1.35709e-14 - 3.76342e-14 - 1.02258e-13 - 2.72236e-13 - 7.1012e-13 - 1.81491e-12 - 4.5448e-12 - 1.11509e-11 - 2.68068e-11 - 6.31417e-11 - 1.45721e-10 - 3.29509e-10 - 7.30044e-10 - 1.58477e-09 - 3.37071e-09 - 7.02447e-09 - 1.43431e-08 - 2.86951e-08 - 5.62483e-08 - 1.08031e-07 - 2.03294e-07 - 3.74832e-07 - 6.77153e-07 - 1.1986e-06 - 2.07872e-06 - 3.53229e-06 - 5.88103e-06 - 9.59373e-06 - 1.53341e-05 - 2.40141e-05 - 3.68477e-05 - 5.53976e-05 - 8.16036e-05 - 0.000117778 - 0.000166554 - 0.000230773 - 0.000313293 - 0.000416728 - 0.000543115 - 0.000693534 - 0.000867722 - 0.00106373 - 0.00127766 - 0.00150363 - 0.0017338 - 0.00195883 - 0.00216836 - 0.00235181 - 0.00249925 - 0.00260228 - 0.00265482 - 0.00265371 - 0.00259901 - 0.00249402 - 0.00234493 - 0.0021602 - 0.00194983 - 0.00172439 - 0.00149421 - 0.0012686 - 0.0010553 - 0.000860126 - 0.000686887 - 0.000537459 - 0.000412043 - 0.000309512 - 0.000227797 - 0.000164269 - 0.000116065 - 8.03492e-05 - 5.45004e-05 - 3.62205e-05 - 2.35856e-05 - 1.50479e-05 - 9.40679e-06 - 5.76161e-06 - 3.45767e-06 - 2.0331e-06 - 1.17131e-06 - 6.61184e-07 - 3.65687e-07 - 1.98168e-07 - 1.05219e-07 - 5.47382e-08 - 2.79013e-08 - 1.39346e-08 - 6.81873e-09 - 3.26925e-09 - 1.53578e-09 - 7.06883e-10 - 3.18789e-10 - 1.40863e-10 - 6.09863e-11 - 2.58729e-11 - 1.07614e-11 - 4.4044e-12 - 1.81685e-12 - 8.67714e-13 - 7.40664e-13 - 1.31546e-12 - 3.06677e-12 - 7.40609e-12 - 1.76751e-11 - 4.13834e-11 - 9.49532e-11 - 2.13472e-10 - 4.70233e-10 - 1.01489e-09 - 2.14618e-09 - 4.4468e-09 - 9.02747e-09 - 1.79565e-08 - 3.49956e-08 - 6.68256e-08 - 1.25028e-07 - 2.29198e-07 - 4.11672e-07 - 7.24482e-07 - 1.24923e-06 - 2.11053e-06 - 3.49364e-06 - 5.66633e-06 - 9.00457e-06 - 1.40204e-05 - 2.13892e-05 - 3.19717e-05 - 4.68246e-05 - 6.71923e-05 - 9.44715e-05 - 0.000130143 - 0.000175661 - 0.00023231 - 0.000301021 - 0.000382176 - 0.000475407 - 0.000579436 - 0.000691962 - 0.000809647 - 0.000928209 - 0.00104264 - 0.00114751 - 0.00123742 - 0.00130742 - 0.00135347 - 0.00137284 - 0.00136436 - 0.00132854 - 0.00126753 - 0.00118488 - 0.00108525 - 0.00097392 - 0.000856353 - 0.000737767 - 0.000622762 - 0.000515065 - 0.000417387 - 0.0003314 - 0.000257812 - 0.000196512 - 0.000146762 - 0.000107393 - 7.69968e-05 - 5.40888e-05 - 3.72288e-05 - 2.51065e-05 - 1.65895e-05 - 1.07402e-05 - 6.81292e-06 - 4.23437e-06 - 2.57858e-06 - 1.53855e-06 - 8.9945e-07 - 5.15205e-07 - 2.89148e-07 - 1.59e-07 - 8.56664e-08 - 4.52232e-08 - 2.3391e-08 - 1.18542e-08 - 5.88618e-09 - 2.86373e-09 - 1.36511e-09 - 6.37584e-10 - 2.91774e-10 - 1.30825e-10 - 5.74743e-11 - 2.47396e-11 - 1.04339e-11 - 4.31162e-12 - 1.7457e-12 - 6.92527e-13 - 2.69178e-13 - 1.02513e-13 - 3.82523e-14 - 1.39853e-14 - 5.00981e-15 - 1.75836e-15 - 6.0469e-16 - 2.03748e-16 - 6.72655e-17 - 2.17584e-17 - 6.89602e-18 - 2.14145e-18 - 6.51557e-19 - 1.94238e-19 - 5.67355e-20 - 1.62376e-20 - 4.5551e-21 - 1.25945e-21 - 3.71542e-22 - 2.27974e-22 - 5.68172e-22 - 2.09885e-21 - 7.8741e-21 - 2.90139e-20 - 1.04766e-19 - 3.70658e-19 - 1.28488e-18 - 4.36407e-18 - 1.4523e-17 - 4.73541e-17 - 1.51285e-16 - 4.73556e-16 - 1.45239e-15 - 4.36447e-15 - 1.28504e-14 - 3.70715e-14 - 1.04785e-13 - 2.90199e-13 - 7.8746e-13 - 2.09362e-12 - 5.45388e-12 - 1.39203e-11 - 3.4812e-11 - 8.52996e-11 - 2.04786e-10 - 4.81717e-10 - 1.11025e-09 - 2.50718e-09 - 5.54737e-09 - 1.20261e-08 - 2.55447e-08 - 5.31635e-08 - 1.08408e-07 - 2.16596e-07 - 4.24008e-07 - 8.1327e-07 - 1.52839e-06 - 2.81429e-06 - 5.07741e-06 - 8.97539e-06 - 1.55454e-05 - 2.6381e-05 - 4.38651e-05 - 7.1464e-05 - 0.000114077 - 0.000178422 - 0.000273429 - 0.00041057 - 0.000604061 - 0.00087082 - 0.00123009 - 0.00170259 - 0.00230921 - 0.00306905 - 0.00399717 - 0.00510188 - 0.00638213 - 0.00782524 - 0.00940538 - 0.0110833 - 0.0128073 - 0.0145165 - 0.0161444 - 0.017625 - 0.0188984 - 0.0199167 - 0.0206486 - 0.0210826 - 0.0212274 - 0.0211103 - 0.0207729 - 0.0202659 - 0.0196419 - 0.0189493 - 0.0182273 - 0.0175019 - 0.0167852 - 0.0160761 - 0.015364 - 0.0146325 - 0.0138649 - 0.0130493 - 0.0121818 - 0.0112696 - 0.0103324 - 0.00940127 - 0.00851798 - 0.0077322 - 0.00709887 - 0.00667521 - 0.00651766 - 0.00667869 - 0.00720359 - 0.00812682 - 0.00946821 - 0.011229 - 0.0133879 - 0.0158984 - 0.0186868 - 0.0216531 - 0.0246735 - 0.0276071 - 0.0303043 - 0.0326177 - 0.0344137 - 0.0355844 - 0.0360574 - 0.0358021 - 0.0348328 - 0.033207 - 0.0310196 - 0.0283937 - 0.0254689 - 0.0223898 - 0.0192943 - 0.0163042 - 0.0135187 - 0.0110106 - 0.00882576 - 0.0069851 - 0.00548818 - 0.00431784 - 0.00344511 - 0.00283369 - 0.00244377 - 0.00223482 - 0.00216753 - 0.00220503 - 0.00231352 - 0.00246261 - 0.00262555 - 0.00277942 - 0.00290547 - 0.00298931 - 0.00302122 - 0.00299617 - 0.00291371 - 0.00277759 - 0.00259503 - 0.00237585 - 0.00213141 - 0.00187359 - 0.00161373 - 0.00136185 - 0.00112608 - 0.000912319 - 0.000724208 - 0.000563271 - 0.000429248 - 0.000320507 - 0.000234478 - 0.000168076 - 0.000118044 - 8.12306e-05 - 5.47687e-05 - 3.61811e-05 - 2.3419e-05 - 1.48522e-05 - 9.22895e-06 - 5.61887e-06 - 3.35184e-06 - 1.95909e-06 - 1.12192e-06 - 6.29515e-07 - 3.46089e-07 - 1.86426e-07 - 9.83922e-08 - 5.08806e-08 - 2.57799e-08 - 1.27981e-08 - 6.22514e-09 - 2.9668e-09 - 1.38537e-09 - 6.33837e-10 - 2.84137e-10 - 1.248e-10 - 5.37078e-11 - 2.26463e-11 - 9.3561e-12 - 3.78729e-12 - 1.5021e-12 - 5.83724e-13 - 2.22255e-13 - 8.29153e-14 - 3.03084e-14 - 1.08573e-14 - 3.81839e-15 - 1.34071e-15 - 5.41182e-16 - 4.60363e-16 - 1.01752e-15 - 2.94646e-15 - 8.68155e-15 - 2.5165e-14 - 7.1502e-14 - 1.99065e-13 - 5.43016e-13 - 1.45133e-12 - 3.80064e-12 - 9.75179e-12 - 2.4516e-11 - 6.03879e-11 - 1.45743e-10 - 3.44638e-10 - 7.985e-10 - 1.81269e-09 - 4.03189e-09 - 8.78681e-09 - 1.87625e-08 - 3.92542e-08 - 8.04673e-08 - 1.61618e-07 - 3.1805e-07 - 6.13252e-07 - 1.15856e-06 - 2.14455e-06 - 3.88947e-06 - 6.91164e-06 - 1.2034e-05 - 2.05292e-05 - 3.43142e-05 - 5.61969e-05 - 9.01753e-05 - 0.000141775 - 0.000218398 - 0.000329635 - 0.000487479 - 0.000706342 - 0.00100279 - 0.0013949 - 0.00190114 - 0.00253875 - 0.00332173 - 0.00425838 - 0.00534887 - 0.00658288 - 0.00793792 - 0.00937852 - 0.0108567 - 0.012314 - 0.0136848 - 0.0149009 - 0.0158974 - 0.0166178 - 0.01702 - 0.0170798 - 0.0167935 - 0.0161785 - 0.0152711 - 0.0141235 - 0.0127982 - 0.011363 - 0.00988489 - 0.00842537 - 0.00703628 - 0.0057575 - 0.00461596 - 0.00362599 - 0.0027908 - 0.00210459 - 0.00155504 - 0.00112578 - 0.000798551 - 0.000554994 - 0.00037793 - 0.000252157 - 0.000164842 - 0.000105584 - 6.62628e-05 - 4.07452e-05 - 2.45482e-05 - 1.44911e-05 - 8.38143e-06 - 4.74977e-06 - 2.63732e-06 - 1.4348e-06 - 7.64815e-07 - 3.99446e-07 - 2.04407e-07 - 1.02488e-07 - 5.03482e-08 - 2.42344e-08 - 1.14293e-08 - 5.28131e-09 - 2.39112e-09 - 1.06071e-09 - 4.6104e-10 - 1.96362e-10 - 8.19975e-11 - 3.37003e-11 - 1.39853e-11 - 6.7944e-12 - 6.04916e-12 - 1.11421e-11 - 2.65104e-11 - 6.50445e-11 - 1.57586e-10 - 3.74505e-10 - 8.72182e-10 - 1.99023e-09 - 4.44979e-09 - 9.74793e-09 - 2.09229e-08 - 4.40016e-08 - 9.06677e-08 - 1.83051e-07 - 3.62102e-07 - 7.01821e-07 - 1.33278e-06 - 2.47987e-06 - 4.52101e-06 - 8.07572e-06 - 1.4134e-05 - 2.42374e-05 - 4.07237e-05 - 6.70422e-05 - 0.000108141 - 0.000170912 - 0.000264666 - 0.000401576 - 0.000597016 - 0.000869669 - 0.0012413 - 0.00173603 - 0.00237906 - 0.00319468 - 0.00420373 - 0.00542055 - 0.00684968 - 0.00848282 - 0.0102964 - 0.0122502 - 0.0142879 - 0.016339 - 0.0183232 - 0.020156 - 0.0217557 - 0.0230507 - 0.0239861 - 0.0245286 - 0.0246698 - 0.0244259 - 0.0238353 - 0.0229533 - 0.0218461 - 0.020583 - 0.0192297 - 0.017843 - 0.0164669 - 0.0151311 - 0.0138521 - 0.0126351 - 0.0114777 - 0.0103735 - 0.00931555 - 0.00829889 - 0.00732211 - 0.00638771 - 0.00550152 - 0.00467147 - 0.00390611 - 0.00321307 - 0.0025979 - 0.00206328 - 0.00160877 - 0.00123095 - 0.000923967 - 0.00068019 - 0.000491014 - 0.000347568 - 0.000241316 - 0.000164498 - 0.000110414 - 7.35521e-05 - 4.96251e-05 - 3.55275e-05 - 2.92725e-05 - 2.99299e-05 - 3.75905e-05 - 5.33657e-05 - 7.9422e-05 - 0.000119033 - 0.000176624 - 0.000257776 - 0.000369135 - 0.000518189 - 0.000712868 - 0.000960941 - 0.00126921 - 0.00164251 - 0.00208268 - 0.00258746 - 0.00314965 - 0.00375654 - 0.00438986 - 0.00502632 - 0.0056388 - 0.00619812 - 0.00667528 - 0.00704395 - 0.00728283 - 0.00737769 - 0.00732281 - 0.00712151 - 0.00678583 - 0.00633536 - 0.0057953 - 0.00519418 - 0.00456137 - 0.00392474 - 0.00330874 - 0.00273308 - 0.00221196 - 0.00175404 - 0.00136282 - 0.00103747 - 0.000773837 - 0.000565535 - 0.000404955 - 0.000284112 - 0.000195304 - 0.000131543 - 8.68085e-05 - 5.61298e-05 - 3.556e-05 - 2.20733e-05 - 1.34248e-05 - 7.99994e-06 - 4.67091e-06 - 2.67211e-06 - 1.49776e-06 - 8.22563e-07 - 4.42621e-07 - 2.33363e-07 - 1.2055e-07 - 6.10156e-08 - 3.02588e-08 - 1.47027e-08 - 6.99974e-09 - 3.26515e-09 - 1.49231e-09 - 6.68274e-10 - 2.93215e-10 - 1.26053e-10 - 5.30957e-11 - 2.19129e-11 - 8.86093e-12 - 3.51071e-12 - 1.36285e-12 - 5.18366e-13 - 1.9318e-13 - 7.05383e-14 - 2.52363e-14 - 8.84629e-15 - 3.03833e-15 - 1.02246e-15 - 3.37126e-16 - 1.08912e-16 - 3.44744e-17 - 1.06919e-17 - 3.24899e-18 - 9.67342e-19 - 2.82194e-19 - 8.06591e-20 - 2.25889e-20 - 6.19832e-21 - 1.66644e-21 - 4.38977e-22 - 1.133e-22 - 2.86522e-23 - 7.09938e-24 - 1.72354e-24 - 4.09975e-25 - 9.555e-26 - 2.18194e-26 - 4.88209e-27 - 1.07047e-27 - 2.30181e-28 - 4.87407e-29 - 1.0397e-29 - 2.49533e-30 - 9.34488e-31 - 6.62952e-31 - 6.42193e-31 - 6.62246e-31 - 6.83449e-31 - 7.13588e-31 - 8.39043e-31 - 1.51326e-30 - 5.01859e-30 - 2.2602e-29 - 1.08559e-28 - 5.19624e-28 - 2.44477e-27 - 1.12767e-26 - 5.09696e-26 - 2.25728e-25 - 9.79488e-25 - 4.16437e-24 - 1.73475e-23 - 7.08044e-23 - 2.83153e-22 - 1.10948e-21 - 4.25944e-21 - 1.60223e-20 - 5.90517e-20 - 2.13244e-19 - 7.545e-19 - 2.61564e-18 - 8.88451e-18 - 2.95683e-17 - 9.64174e-17 - 3.0805e-16 - 9.64327e-16 - 2.95777e-15 - 8.88875e-15 - 2.6173e-14 - 7.551e-14 - 2.13448e-13 - 5.91175e-13 - 1.60427e-12 - 4.26554e-12 - 1.11124e-11 - 2.83648e-11 - 7.09396e-11 - 1.73834e-10 - 4.17365e-10 - 9.81826e-10 - 2.26303e-09 - 5.11073e-09 - 1.13087e-08 - 2.45176e-08 - 5.20812e-08 - 1.08398e-07 - 2.21054e-07 - 4.41684e-07 - 8.64693e-07 - 1.65863e-06 - 3.11726e-06 - 5.7403e-06 - 1.0357e-05 - 1.83091e-05 - 3.17131e-05 - 5.38204e-05 - 8.94936e-05 - 0.000145806 - 0.000232752 - 0.00036404 - 0.000557881 - 0.000837666 - 0.00123236 - 0.0017764 - 0.00250888 - 0.00347182 - 0.00470729 - 0.00625347 - 0.0081397 - 0.0103808 - 0.0129716 - 0.0158815 - 0.0190514 - 0.0223923 - 0.0257874 - 0.0290973 - 0.0321688 - 0.0348462 - 0.0369839 - 0.03846 - 0.0391873 - 0.0391222 - 0.0382691 - 0.0366799 - 0.0344491 - 0.0317048 - 0.0285972 - 0.0252851 - 0.0219241 - 0.0186558 - 0.0155996 - 0.0128486 - 0.0104682 - 0.00849752 - 0.00695295 - 0.00583265 - 0.00512109 - 0.00479313 - 0.00481689 - 0.00515543 - 0.0057673 - 0.00660627 - 0.00762079 - 0.00875373 - 0.00994277 - 0.0111219 - 0.0122239 - 0.0131839 - 0.0139439 - 0.0144563 - 0.0146885 - 0.0146249 - 0.0142685 - 0.0136401 - 0.0127762 - 0.0117254 - 0.0105438 - 0.00928966 - 0.00801941 - 0.00678301 - 0.00562134 - 0.0045645 - 0.00363149 - 0.00283082 - 0.00216211 - 0.001618 - 0.00118636 - 0.000852305 - 0.00059995 - 0.000413799 - 0.000279676 - 0.000185285 - 0.000120433 - 7.70311e-05 - 4.89435e-05 - 3.17797e-05 - 2.27018e-05 - 2.03111e-05 - 2.46741e-05 - 3.75481e-05 - 6.28712e-05 - 0.000107589 - 0.000182895 - 0.000305951 - 0.00050214 - 0.000807822 - 0.0012735 - 0.00196715 - 0.00297727 - 0.00441506 - 0.00641494 - 0.00913241 - 0.0127384 - 0.0174093 - 0.0233123 - 0.0305863 - 0.0393191 - 0.0495243 - 0.061118 - 0.0739021 - 0.0875551 - 0.101635 - 0.115596 - 0.128818 - 0.140653 - 0.150473 - 0.157726 - 0.161989 - 0.163007 - 0.160717 - 0.155259 - 0.146955 - 0.136286 - 0.123839 - 0.110255 - 0.0961777 - 0.0822032 - 0.0688398 - 0.0564843 - 0.0454101 - 0.0357696 - 0.0276066 - 0.020876 - 0.0154675 - 0.0112287 - 0.00798685 - 0.00556619 - 0.00380082 - 0.00254293 - 0.00166697 - 0.00107067 - 0.00067379 - 0.000415459 - 0.000250997 - 0.000148575 - 8.6171e-05 - 4.8968e-05 - 2.72647e-05 - 1.48739e-05 - 7.95037e-06 - 4.16376e-06 - 2.13659e-06 - 1.07422e-06 - 5.29179e-07 - 2.55416e-07 - 1.2079e-07 - 5.59695e-08 - 2.54102e-08 - 1.13032e-08 - 4.92642e-09 - 2.10377e-09 - 8.80244e-10 - 3.60866e-10 - 1.44959e-10 - 5.70702e-11 - 2.20651e-11 - 8.50316e-12 - 3.61538e-12 - 2.59818e-12 - 4.39325e-12 - 1.06972e-11 - 2.73806e-11 - 6.93422e-11 - 1.72287e-10 - 4.19489e-10 - 1.00077e-09 - 2.3393e-09 - 5.35766e-09 - 1.20227e-08 - 2.6434e-08 - 5.6946e-08 - 1.20199e-07 - 2.48584e-07 - 5.03712e-07 - 1.00007e-06 - 1.94542e-06 - 3.70795e-06 - 6.92454e-06 - 1.26703e-05 - 2.27152e-05 - 3.99011e-05 - 6.86735e-05 - 0.000115806 - 0.000191342 - 0.000309759 - 0.000491334 - 0.0007636 - 0.00116276 - 0.00173482 - 0.00253603 - 0.00363237 - 0.00509757 - 0.00700928 - 0.00944323 - 0.0124653 - 0.0161222 - 0.0204307 - 0.0253675 - 0.0308609 - 0.0367855 - 0.0429617 - 0.0491612 - 0.055119 - 0.0605504 - 0.0651732 - 0.0687319 - 0.0710206 - 0.071903 - 0.0713258 - 0.0693239 - 0.0660171 - 0.061598 - 0.0563137 - 0.0504426 - 0.0442709 - 0.0380694 - 0.0320753 - 0.026479 - 0.0214176 - 0.0169736 - 0.01318 - 0.0100276 - 0.00747499 - 0.00545963 - 0.00390708 - 0.00273955 - 0.00188209 - 0.0012669 - 0.00083556 - 0.000539947 - 0.00034187 - 0.000212084 - 0.000128912 - 7.67739e-05 - 4.47993e-05 - 2.56133e-05 - 1.43482e-05 - 7.87527e-06 - 4.23517e-06 - 2.23158e-06 - 1.1521e-06 - 5.82783e-07 - 2.88841e-07 - 1.40265e-07 - 6.67383e-08 - 3.11127e-08 - 1.42114e-08 - 6.36025e-09 - 2.789e-09 - 1.19828e-09 - 5.04435e-10 - 2.08061e-10 - 8.40835e-11 - 3.32942e-11 - 1.29171e-11 - 4.91018e-12 - 1.8289e-12 - 6.6774e-13 - 2.39797e-13 - 8.72764e-14 - 3.99832e-14 - 4.35828e-14 - 1.03126e-13 - 2.89491e-13 - 8.16136e-13 - 2.26089e-12 - 6.1391e-12 - 1.6335e-11 - 4.25903e-11 - 1.08812e-10 - 2.72406e-10 - 6.68242e-10 - 1.6063e-09 - 3.78352e-09 - 8.7326e-09 - 1.97501e-08 - 4.37696e-08 - 9.50509e-08 - 2.02264e-07 - 4.21756e-07 - 8.61758e-07 - 1.7254e-06 - 3.38515e-06 - 6.50799e-06 - 1.22603e-05 - 2.26326e-05 - 4.09407e-05 - 7.25704e-05 - 0.000126052 - 0.000214548 - 0.000357837 - 0.000584835 - 0.000936631 - 0.00146991 - 0.00226049 - 0.00340645 - 0.00503026 - 0.00727894 - 0.0103214 - 0.0143415 - 0.0195274 - 0.0260547 - 0.0340659 - 0.0436462 - 0.0547983 - 0.067419 - 0.0812818 - 0.0960292 - 0.111177 - 0.126134 - 0.140238 - 0.1528 - 0.163165 - 0.170766 - 0.175185 - 0.17619 - 0.173768 - 0.168128 - 0.159688 - 0.149039 - 0.136897 - 0.124043 - 0.111262 - 0.0992842 - 0.0887352 - 0.0800981 - 0.073689 - 0.0696463 - 0.0679332 - 0.0683505 - 0.0705597 - 0.0741133 - 0.0784909 - 0.0831388 - 0.0875109 - 0.091108 - 0.0935133 - 0.09442 - 0.0936496 - 0.0911582 - 0.0870312 - 0.0814664 - 0.0747481 - 0.0672163 - 0.0592323 - 0.0511477 - 0.0432772 - 0.0358796 - 0.0291464 - 0.0231988 - 0.0180921 - 0.0138246 - 0.0103503 - 0.00759265 - 0.00545721 - 0.00384313 - 0.00265178 - 0.00179277 - 0.00118755 - 0.000770751 - 0.000490133 - 0.000305387 - 0.000186433 - 0.000111515 - 6.53553e-05 - 3.75288e-05 - 2.11147e-05 - 1.16397e-05 - 6.28691e-06 - 3.32712e-06 - 1.7252e-06 - 8.76506e-07 - 4.36367e-07 - 2.12967e-07 - 1.02108e-07 - 4.86122e-08 - 2.41809e-08 - 1.51788e-08 - 1.63679e-08 - 2.83794e-08 - 5.79776e-08 - 1.21189e-07 - 2.50402e-07 - 5.07817e-07 - 1.0094e-06 - 1.966e-06 - 3.75186e-06 - 7.01533e-06 - 1.28525e-05 - 2.30708e-05 - 4.05764e-05 - 6.99235e-05 - 0.000118062 - 0.000195313 - 0.000316586 - 0.000502791 - 0.000782386 - 0.00119286 - 0.00178196 - 0.00260821 - 0.00374044 - 0.00525582 - 0.00723593 - 0.00976081 - 0.0129007 - 0.0167063 - 0.0211974 - 0.0263525 - 0.0320995 - 0.0383101 - 0.044799 - 0.051329 - 0.0576237 - 0.0633852 - 0.0683175 - 0.0721517 - 0.0746721 - 0.0757375 - 0.0752976 - 0.0734009 - 0.0701941 - 0.0659136 - 0.0608694 - 0.0554259 - 0.0499808 - 0.0449442 - 0.0407215 - 0.0376986 - 0.0362307 - 0.0366324 - 0.0391679 - 0.0440395 - 0.0513723 - 0.0611977 - 0.0734333 - 0.0878657 - 0.104137 - 0.121742 - 0.140036 - 0.158258 - 0.175572 - 0.191121 - 0.204086 - 0.213755 - 0.219579 - 0.221225 - 0.218598 - 0.211856 - 0.201386 - 0.187772 - 0.171739 - 0.15409 - 0.13564 - 0.117162 - 0.0993331 - 0.0827063 - 0.0676928 - 0.0545624 - 0.043456 - 0.034407 - 0.0273684 - 0.022239 - 0.0188874 - 0.0171699 - 0.0169424 - 0.0180635 - 0.0203933 - 0.0237859 - 0.0280809 - 0.0330942 - 0.0386122 - 0.0443899 - 0.0501552 - 0.0556197 - 0.0604946 - 0.06451 - 0.0674365 - 0.069104 - 0.069417 - 0.0683625 - 0.0660108 - 0.0625073 - 0.0580577 - 0.0529082 - 0.0473225 - 0.0415602 - 0.0358572 - 0.0304111 - 0.0253728 - 0.0208431 - 0.0168751 - 0.0134805 - 0.0106382 - 0.00830404 - 0.00641967 - 0.00492092 - 0.00374385 - 0.00282905 - 0.0021241 - 0.00158457 - 0.00117404 - 0.000863316 - 0.000629412 - 0.000454431 - 0.000324503 - 0.000228896 - 0.000159295 - 0.000109251 - 7.37695e-05 - 4.89991e-05 - 3.19922e-05 - 2.05207e-05 - 1.2926e-05 - 7.99585e-06 - 4.86315e-06 - 2.92334e-06 - 1.76948e-06 - 1.1439e-06 - 9.06437e-07 - 1.02009e-06 - 1.55632e-06 - 2.72467e-06 - 4.93405e-06 - 8.89688e-06 - 1.57908e-05 - 2.74957e-05 - 4.69267e-05 - 7.84794e-05 - 0.0001286 - 0.000206474 - 0.00032481 - 0.000500644 - 0.000756078 - 0.00111877 - 0.001622 - 0.00230409 - 0.00320688 - 0.00437325 - 0.00584336 - 0.00764994 - 0.00981273 - 0.0123327 - 0.0151867 - 0.0183234 - 0.0216614 - 0.0250901 - 0.0284744 - 0.0316625 - 0.0344962 - 0.0368244 - 0.0385155 - 0.0394705 - 0.0396321 - 0.0389905 - 0.0375844 - 0.0354972 - 0.032849 - 0.0297846 - 0.0264615 - 0.0230361 - 0.0196523 - 0.0164334 - 0.0134759 - 0.0108485 - 0.00859408 - 0.00673434 - 0.00527724 - 0.00422554 - 0.0035862 - 0.00337974 - 0.00364906 - 0.0044669 - 0.00594138 - 0.00821867 - 0.0114817 - 0.0159437 - 0.0218363 - 0.0293904 - 0.0388107 - 0.0502452 - 0.0637519 - 0.0792647 - 0.0965666 - 0.115271 - 0.134821 - 0.154501 - 0.173478 - 0.190851 - 0.205722 - 0.217273 - 0.224836 - 0.227963 - 0.226463 - 0.220429 - 0.210222 - 0.196437 - 0.179848 - 0.161335 - 0.141805 - 0.122123 - 0.103052 - 0.0852089 - 0.0690443 - 0.0548367 - 0.0427079 - 0.0326479 - 0.0245476 - 0.0182336 - 0.0135014 - 0.0101435 - 0.00797117 - 0.00682901 - 0.00660242 - 0.0072185 - 0.00864139 - 0.0108629 - 0.0138899 - 0.0177288 - 0.0223696 - 0.0277703 - 0.0338438 - 0.0404482 - 0.0473843 - 0.0543987 - 0.0611951 - 0.0674527 - 0.0728495 - 0.0770893 - 0.0799283 - 0.0811978 - 0.0808212 - 0.0788213 - 0.075318 - 0.0705165 - 0.0646873 - 0.0581413 - 0.0512021 - 0.0441801 - 0.0373511 - 0.0309397 - 0.0251111 - 0.0199688 - 0.0155588 - 0.0118778 - 0.00888452 - 0.0065113 - 0.00467562 - 0.00328963 - 0.00226773 - 0.0015317 - 0.00101366 - 0.000657274 - 0.000417579 - 0.000259937 - 0.000158538 - 9.47405e-05 - 5.54722e-05 - 3.18238e-05 - 1.78881e-05 - 9.85178e-06 - 5.3162e-06 - 2.81077e-06 - 1.45608e-06 - 7.39066e-07 - 3.67551e-07 - 1.79097e-07 - 8.55059e-08 - 3.99982e-08 - 1.83325e-08 - 8.23268e-09 - 3.6224e-09 - 1.56167e-09 - 6.59656e-10 - 2.73013e-10 - 1.1071e-10 - 4.39871e-11 - 1.71239e-11 - 6.53153e-12 - 2.44099e-12 - 8.93824e-13 - 3.20683e-13 - 1.12729e-13 - 3.8827e-14 - 1.31029e-14 - 4.33251e-15 - 1.40361e-15 - 4.45546e-16 - 1.38572e-16 - 4.22272e-17 - 1.26081e-17 - 3.68842e-18 - 1.05723e-18 - 2.96917e-19 - 8.1703e-20 - 2.20281e-20 - 5.81908e-21 - 1.50615e-21 - 3.8196e-22 - 9.49084e-23 - 2.31062e-23 - 5.51175e-24 - 1.28821e-24 - 2.94999e-25 - 6.619e-26 - 1.45513e-26 - 3.13434e-27 - 6.61496e-28 - 1.36787e-28 - 2.77142e-29 - 5.50166e-30 - 1.0701e-30 - 2.03934e-31 - 3.80796e-32 - 6.96678e-33 - 1.24885e-33 - 2.19343e-34 - 3.77465e-35 - 6.36455e-36 - 1.05148e-36 - 1.70207e-37 - 2.69967e-38 - 4.19579e-39 - 6.3903e-40 - 9.53894e-41 - 1.39599e-41 - 2.00417e-42 - 2.826e-43 - 3.92298e-44 - 5.38547e-45 - 7.37316e-46 - 1.02172e-46 - 1.46661e-47 - 2.24577e-48 - 3.76058e-49 - 6.92214e-50 - 1.37541e-50 - 2.86492e-51 - 6.09239e-52 - 1.29888e-52 - 2.74651e-53 - 5.72622e-54 - 1.17351e-54 - 2.3602e-55 - 4.65483e-56 - 8.99862e-57 - 1.7048e-57 - 3.16486e-58 - 5.75699e-59 - 1.02609e-59 - 1.79191e-60 - 3.0661e-61 - 5.14039e-62 - 8.4439e-63 - 1.35902e-63 - 2.14313e-64 - 3.31136e-65 - 5.01304e-66 - 7.4359e-67 - 1.08069e-67 - 1.53889e-68 - 2.14709e-69 - 2.93514e-70 - 3.93138e-71 - 5.15939e-72 - 6.6342e-73 - 8.35824e-74 - 1.03176e-74 - 1.2479e-75 - 1.47882e-76 - 1.71708e-77 - 1.95345e-78 - 2.17745e-79 - 2.37812e-80 - 2.54481e-81 - 2.66817e-82 - 2.741e-83 - 2.75893e-84 - 2.72088e-85 - 2.62915e-86 - 2.48919e-87 - 2.30907e-88 - 2.09871e-89 - 1.86899e-90 - 1.63078e-91 - 1.39419e-92 - 1.16785e-93 - 9.58489e-95 - 7.7077e-96 - 6.07294e-97 - 4.68824e-98 - 3.54616e-99 - 2.6281e-100 - 1.90838e-101 - 1.35776e-102 - 9.46494e-104 - 6.46472e-105 - 4.32632e-106 - 2.83677e-107 - 1.8225e-108 - 1.14722e-109 - 7.0756e-111 - 4.27579e-112 - 2.53167e-113 - 1.4687e-114 - 8.34828e-116 - 4.64941e-117 - 2.53708e-118 - 1.35647e-119 - 7.10592e-121 - 3.64727e-122 - 1.83423e-123 - 9.03805e-125 - 4.36348e-126 - 2.06409e-127 - 9.56667e-129 - 4.3444e-130 - 1.93302e-131 - 8.42711e-133 - 3.59963e-134 - 1.5068e-135 - 6.8759e-137 - 1.73326e-136 - 4.09615e-135 - 9.62208e-134 - 2.21467e-132 - 4.99444e-131 - 1.10357e-129 - 2.3892e-128 - 5.06804e-127 - 1.05333e-125 - 2.14499e-124 - 4.27981e-123 - 8.36681e-122 - 1.60263e-120 - 3.00774e-119 - 5.53078e-118 - 9.96482e-117 - 1.75909e-115 - 3.0426e-114 - 5.1563e-113 - 8.56187e-112 - 1.39295e-110 - 2.22044e-109 - 3.46801e-108 - 5.3071e-107 - 7.95741e-106 - 1.16902e-104 - 1.68272e-103 - 2.37321e-102 - 3.27942e-101 - 4.44013e-100 - 5.89021e-99 - 7.65602e-98 - 9.75017e-97 - 1.21663e-95 - 1.48745e-94 - 1.78181e-93 - 2.0913e-92 - 2.40497e-91 - 2.70982e-90 - 2.99163e-89 - 3.23602e-88 - 3.42967e-87 - 3.56147e-86 - 3.62363e-85 - 3.61239e-84 - 3.52844e-83 - 3.37682e-82 - 3.16643e-81 - 2.90917e-80 - 2.61881e-79 - 2.30981e-78 - 1.99612e-77 - 1.69017e-76 - 1.40221e-75 - 1.13981e-74 - 9.07801e-74 - 7.08409e-73 - 5.41645e-72 - 4.05772e-71 - 2.97842e-70 - 2.14203e-69 - 1.5094e-68 - 1.04212e-67 - 7.04969e-67 - 4.6726e-66 - 3.03447e-65 - 1.93084e-64 - 1.20377e-63 - 7.35325e-63 - 4.40101e-62 - 2.58084e-61 - 1.48289e-60 - 8.34816e-60 - 4.6048e-59 - 2.48867e-58 - 1.31784e-57 - 6.83741e-57 - 3.47584e-56 - 1.73127e-55 - 8.44899e-55 - 4.04001e-54 - 1.89277e-53 - 8.6886e-53 - 3.90786e-52 - 1.72212e-51 - 7.43579e-51 - 3.14577e-50 - 1.30396e-49 - 5.29586e-49 - 2.1074e-48 - 8.21665e-48 - 3.13892e-47 - 1.1749e-46 - 4.30884e-46 - 1.5483e-45 - 5.45116e-45 - 1.88044e-44 - 6.35575e-44 - 2.1048e-43 - 6.82958e-43 - 2.17127e-42 - 6.76349e-42 - 2.06427e-41 - 6.17304e-41 - 1.80871e-40 - 5.19252e-40 - 1.46058e-39 - 4.02541e-39 - 1.08701e-38 - 2.87605e-38 - 7.45588e-38 - 1.89383e-37 - 4.71327e-37 - 1.14933e-36 - 2.74603e-36 - 6.4285e-36 - 1.47454e-35 - 3.31393e-35 - 7.29752e-35 - 1.57453e-34 - 3.32869e-34 - 6.89512e-34 - 1.39945e-33 - 2.78307e-33 - 5.42303e-33 - 1.03541e-32 - 1.93704e-32 - 3.55078e-32 - 6.37779e-32 - 1.12249e-31 - 1.93581e-31 - 3.27126e-31 - 5.41684e-31 - 8.78944e-31 - 1.39755e-30 - 2.17756e-30 - 3.32491e-30 - 4.97514e-30 - 7.29554e-30 - 1.04845e-29 - 1.4767e-29 - 2.03847e-29 - 2.75805e-29 - 3.65769e-29 - 4.75488e-29 - 6.05936e-29 - 7.57e-29 - 9.27214e-29 - 1.11356e-28 - 1.3114e-28 - 1.51458e-28 - 1.71566e-28 - 1.90636e-28 - 2.07813e-28 - 2.22279e-28 - 2.33321e-28 - 2.40387e-28 - 2.43136e-28 - 2.41466e-28 - 2.35514e-28 - 2.25645e-28 - 2.12411e-28 - 1.965e-28 - 1.78681e-28 - 1.59741e-28 - 1.40429e-28 - 1.21419e-28 - 1.03269e-28 - 8.64112e-29 - 7.11427e-29 - 5.76345e-29 - 4.59455e-29 - 3.60422e-29 - - - - - 1.08149e-32 - 1.34589e-32 - 1.6411e-32 - 1.96068e-32 - 2.2954e-32 - 2.63428e-32 - 2.96945e-32 - 3.31992e-32 - 3.85269e-32 - 5.50162e-32 - 1.30798e-31 - 5.06291e-31 - 2.35562e-30 - 1.12932e-29 - 5.36116e-29 - 2.49901e-28 - 1.14183e-27 - 5.11219e-27 - 2.24263e-26 - 9.63936e-26 - 4.05952e-25 - 1.67509e-24 - 6.77235e-24 - 2.68273e-23 - 1.04124e-22 - 3.9597e-22 - 1.4754e-21 - 5.38636e-21 - 1.92671e-20 - 6.75267e-20 - 2.31884e-19 - 7.80195e-19 - 2.57201e-18 - 8.30765e-18 - 2.62918e-17 - 8.15268e-17 - 2.47695e-16 - 7.37345e-16 - 2.15061e-15 - 6.14593e-15 - 1.72088e-14 - 4.7212e-14 - 1.26908e-13 - 3.34245e-13 - 8.62533e-13 - 2.18084e-12 - 5.40267e-12 - 1.31139e-11 - 3.11881e-11 - 7.26749e-11 - 1.65927e-10 - 3.71181e-10 - 8.13565e-10 - 1.74717e-09 - 3.67633e-09 - 7.57934e-09 - 1.53103e-08 - 3.03023e-08 - 5.87628e-08 - 1.11652e-07 - 2.07858e-07 - 3.79145e-07 - 6.77609e-07 - 1.18656e-06 - 2.03582e-06 - 3.42235e-06 - 5.63697e-06 - 9.09714e-06 - 1.43847e-05 - 2.2286e-05 - 3.383e-05 - 5.03162e-05 - 7.33248e-05 - 0.000104696 - 0.00014647 - 0.000200771 - 0.000269645 - 0.000354829 - 0.000457491 - 0.000577942 - 0.000715355 - 0.000867554 - 0.00103088 - 0.00120021 - 0.00136913 - 0.00153027 - 0.00167583 - 0.00179817 - 0.00189049 - 0.00194743 - 0.00196562 - 0.00194402 - 0.001884 - 0.00178928 - 0.00166553 - 0.00151987 - 0.00136029 - 0.00119498 - 0.00103175 - 0.000877556 - 0.000738187 - 0.000618089 - 0.000520341 - 0.000446731 - 0.000397902 - 0.000373517 - 0.000372412 - 0.000392709 - 0.000431897 - 0.000486877 - 0.000554005 - 0.000629151 - 0.000707798 - 0.000785206 - 0.00085663 - 0.000917596 - 0.000964193 - 0.000993359 - 0.00100311 - 0.000992716 - 0.000962696 - 0.000914787 - 0.000851735 - 0.000777026 - 0.000694558 - 0.000608305 - 0.000522003 - 0.000438896 - 0.000361567 - 0.000291845 - 0.000230809 - 0.000178851 - 0.000135789 - 0.000101013 - 7.3625e-05 - 5.25788e-05 - 3.67903e-05 - 2.52227e-05 - 1.69429e-05 - 1.11512e-05 - 7.19101e-06 - 4.54356e-06 - 2.8128e-06 - 1.70616e-06 - 1.014e-06 - 5.90459e-07 - 3.36884e-07 - 1.88325e-07 - 1.03151e-07 - 5.53571e-08 - 2.91079e-08 - 1.49964e-08 - 7.57003e-09 - 3.74409e-09 - 1.81439e-09 - 8.61497e-10 - 4.00786e-10 - 1.82687e-10 - 8.15908e-11 - 3.57036e-11 - 1.53081e-11 - 6.43124e-12 - 2.64844e-12 - 1.07197e-12 - 4.34795e-13 - 2.00078e-13 - 1.63936e-13 - 3.00174e-13 - 7.46316e-13 - 1.9281e-12 - 4.92178e-12 - 1.23241e-11 - 3.02408e-11 - 7.27073e-11 - 1.71278e-10 - 3.95331e-10 - 8.94041e-10 - 1.98103e-09 - 4.30092e-09 - 9.1489e-09 - 1.90683e-08 - 3.89398e-08 - 7.79133e-08 - 1.52745e-07 - 2.93398e-07 - 5.52187e-07 - 1.01824e-06 - 1.83973e-06 - 3.25681e-06 - 5.64895e-06 - 9.60021e-06 - 1.59856e-05 - 2.60805e-05 - 4.16907e-05 - 6.52979e-05 - 0.000100207 - 0.000150671 - 0.000221973 - 0.000320411 - 0.00045316 - 0.000627962 - 0.000852611 - 0.00113424 - 0.00147842 - 0.0018881 - 0.0023626 - 0.00289663 - 0.00347962 - 0.00409551 - 0.00472303 - 0.00533667 - 0.00590822 - 0.00640885 - 0.00681147 - 0.00709314 - 0.00723725 - 0.00723514 - 0.00708696 - 0.00680167 - 0.00639614 - 0.00589358 - 0.00532131 - 0.00470844 - 0.00408352 - 0.00347255 - 0.00289752 - 0.00237559 - 0.00191887 - 0.00153483 - 0.00122706 - 0.000996256 - 0.000841365 - 0.000760476 - 0.000751552 - 0.000812793 - 0.000942642 - 0.00113944 - 0.00140081 - 0.00172285 - 0.0020993 - 0.00252088 - 0.00297478 - 0.00344473 - 0.00391135 - 0.00435317 - 0.00474797 - 0.00507448 - 0.00531415 - 0.00545286 - 0.00548225 - 0.00540051 - 0.00521259 - 0.00492967 - 0.00456811 - 0.00414791 - 0.00369091 - 0.00321908 - 0.00275291 - 0.00231029 - 0.00190581 - 0.00155063 - 0.00125284 - 0.00101825 - 0.000851451 - 0.000756966 - 0.000740438 - 0.000809583 - 0.00097481 - 0.00124937 - 0.00164889 - 0.00219032 - 0.00289009 - 0.0037618 - 0.00481339 - 0.00604416 - 0.00744202 - 0.00898139 - 0.0106221 - 0.0123098 - 0.013978 - 0.015552 - 0.0169538 - 0.0181087 - 0.0189516 - 0.019433 - 0.0195241 - 0.0192195 - 0.0185373 - 0.0175182 - 0.0162207 - 0.0147159 - 0.013081 - 0.0113928 - 0.00972205 - 0.00812871 - 0.00665921 - 0.00534516 - 0.00420373 - 0.00323926 - 0.00244565 - 0.00180917 - 0.0013113 - 0.000931238 - 0.000647972 - 0.000441762 - 0.000295092 - 0.000193136 - 0.000123853 - 7.78193e-05 - 4.79075e-05 - 2.88973e-05 - 1.70784e-05 - 9.88952e-06 - 5.61099e-06 - 3.11918e-06 - 1.69894e-06 - 9.06678e-07 - 4.74094e-07 - 2.42892e-07 - 1.21926e-07 - 5.99681e-08 - 2.88987e-08 - 1.3645e-08 - 6.31259e-09 - 2.86139e-09 - 1.27082e-09 - 5.53001e-10 - 2.3578e-10 - 9.8497e-11 - 4.03159e-11 - 1.61684e-11 - 6.35323e-12 - 2.44601e-12 - 9.22699e-13 - 3.41034e-13 - 1.23502e-13 - 4.38212e-14 - 1.52347e-14 - 5.18941e-15 - 1.73197e-15 - 5.66369e-16 - 1.81466e-16 - 5.69676e-17 - 1.75226e-17 - 5.28085e-18 - 1.55936e-18 - 4.51156e-19 - 1.27892e-19 - 3.5522e-20 - 9.66693e-21 - 2.5776e-21 - 6.73413e-22 - 1.72385e-22 - 4.3267e-23 - 1.07691e-23 - 3.17029e-24 - 3.13795e-24 - 1.04192e-23 - 4.08832e-23 - 1.59072e-22 - 6.06845e-22 - 2.26838e-21 - 8.3079e-21 - 2.98129e-20 - 1.04822e-19 - 3.6111e-19 - 1.21889e-18 - 4.03109e-18 - 1.30623e-17 - 4.14718e-17 - 1.2901e-16 - 3.93216e-16 - 1.17429e-15 - 3.43603e-15 - 9.85087e-15 - 2.76713e-14 - 7.6159e-14 - 2.05376e-13 - 5.42644e-13 - 1.40481e-12 - 3.56333e-12 - 8.85588e-12 - 2.15648e-11 - 5.1451e-11 - 1.20276e-10 - 2.75489e-10 - 6.1825e-10 - 1.35944e-09 - 2.92883e-09 - 6.18252e-09 - 1.27871e-08 - 2.5913e-08 - 5.14516e-08 - 1.00096e-07 - 1.90797e-07 - 3.56339e-07 - 6.52066e-07 - 1.16912e-06 - 2.05381e-06 - 3.53508e-06 - 5.96177e-06 - 9.85117e-06 - 1.59492e-05 - 2.53002e-05 - 3.93232e-05 - 5.98839e-05 - 8.93528e-05 - 0.00013063 - 0.000187119 - 0.000262622 - 0.000361147 - 0.000486606 - 0.000642415 - 0.000830998 - 0.00105327 - 0.00130809 - 0.00159189 - 0.00189838 - 0.00221861 - 0.00254131 - 0.00285355 - 0.00314183 - 0.00339333 - 0.00359738 - 0.00374691 - 0.00383975 - 0.00387957 - 0.00387642 - 0.00384664 - 0.00381224 - 0.0037996 - 0.00383766 - 0.00395569 - 0.00418069 - 0.0045347 - 0.00503217 - 0.00567767 - 0.00646409 - 0.00737179 - 0.00836862 - 0.00941129 - 0.0104478 - 0.011421 - 0.012273 - 0.0129503 - 0.013408 - 0.0136139 - 0.013552 - 0.0132229 - 0.0126445 - 0.011849 - 0.0108804 - 0.00978988 - 0.00863109 - 0.00745595 - 0.00631082 - 0.00523372 - 0.00425281 - 0.00338595 - 0.00264134 - 0.00201887 - 0.00151194 - 0.00110944 - 0.000797693 - 0.000562024 - 0.000388087 - 0.000262733 - 0.000174531 - 0.00011398 - 7.34973e-05 - 4.72468e-05 - 3.0892e-05 - 2.1316e-05 - 1.63489e-05 - 1.45246e-05 - 1.48771e-05 - 1.6778e-05 - 1.98118e-05 - 2.36856e-05 - 2.81654e-05 - 3.30354e-05 - 3.80753e-05 - 4.30518e-05 - 4.77214e-05 - 5.18411e-05 - 5.51843e-05 - 5.75589e-05 - 5.88241e-05 - 5.89031e-05 - 5.77908e-05 - 5.55543e-05 - 5.23255e-05 - 4.82887e-05 - 4.36632e-05 - 3.86831e-05 - 3.35788e-05 - 2.85591e-05 - 2.37992e-05 - 1.94319e-05 - 1.55456e-05 - 1.21852e-05 - 9.35833e-06 - 7.04206e-06 - 5.19204e-06 - 3.7507e-06 - 2.65476e-06 - 1.84108e-06 - 1.251e-06 - 8.32877e-07 - 5.433e-07 - 3.47245e-07 - 2.17455e-07 - 1.33426e-07 - 8.02134e-08 - 4.72494e-08 - 2.72712e-08 - 1.54258e-08 - 8.55829e-09 - 4.67484e-09 - 2.55734e-09 - 1.50298e-09 - 1.16839e-09 - 1.50985e-09 - 2.83243e-09 - 5.99601e-09 - 1.28846e-08 - 2.73441e-08 - 5.6957e-08 - 1.16287e-07 - 2.32641e-07 - 4.56022e-07 - 8.75839e-07 - 1.64816e-06 - 3.03887e-06 - 5.48987e-06 - 9.71737e-06 - 1.68528e-05 - 2.86373e-05 - 4.76792e-05 - 7.77791e-05 - 0.000124318 - 0.000194689 - 0.000298735 - 0.000449125 - 0.000661584 - 0.00095486 - 0.0013503 - 0.00187094 - 0.00253995 - 0.00337853 - 0.00440319 - 0.00562268 - 0.00703487 - 0.00862394 - 0.0103584 - 0.0121903 - 0.0140565 - 0.0158808 - 0.0175795 - 0.0190668 - 0.0202621 - 0.0210974 - 0.0215234 - 0.0215144 - 0.021071 - 0.0202198 - 0.019011 - 0.0175134 - 0.0158078 - 0.0139802 - 0.012114 - 0.0102849 - 0.00855561 - 0.00697329 - 0.00556879 - 0.00435734 - 0.00334055 - 0.0025093 - 0.00184682 - 0.00133178 - 0.00094097 - 0.000651414 - 0.000441851 - 0.00029365 - 0.000191215 - 0.000121998 - 7.62635e-05 - 4.6711e-05 - 2.80323e-05 - 1.6483e-05 - 9.49616e-06 - 5.36041e-06 - 2.96473e-06 - 1.6066e-06 - 8.53039e-07 - 4.43778e-07 - 2.26204e-07 - 1.12972e-07 - 5.52814e-08 - 2.65047e-08 - 1.2451e-08 - 5.73091e-09 - 2.58451e-09 - 1.14202e-09 - 4.94439e-10 - 2.09776e-10 - 8.72944e-11 - 3.58409e-11 - 1.50894e-11 - 7.98683e-12 - 8.47379e-12 - 1.69387e-11 - 4.04834e-11 - 9.81476e-11 - 2.34372e-10 - 5.48791e-10 - 1.2592e-09 - 2.83093e-09 - 6.23592e-09 - 1.34589e-08 - 2.84612e-08 - 5.89705e-08 - 1.19716e-07 - 2.38127e-07 - 4.64089e-07 - 8.86197e-07 - 1.65804e-06 - 3.03948e-06 - 5.45932e-06 - 9.6076e-06 - 1.65664e-05 - 2.79884e-05 - 4.63304e-05 - 7.51432e-05 - 0.000119413 - 0.000185929 - 0.00028365 - 0.000423988 - 0.000620957 - 0.00089106 - 0.00125282 - 0.00172587 - 0.0023295 - 0.00308074 - 0.00399194 - 0.00506816 - 0.00630454 - 0.0076841 - 0.00917634 - 0.010737 - 0.0123093 - 0.0138268 - 0.0152175 - 0.0164099 - 0.0173382 - 0.0179489 - 0.0182058 - 0.0180933 - 0.0176182 - 0.0168091 - 0.0157131 - 0.0143919 - 0.0129155 - 0.0113564 - 0.00978378 - 0.00825866 - 0.00683045 - 0.00553511 - 0.0043948 - 0.00341893 - 0.00260602 - 0.00194626 - 0.00142417 - 0.00102108 - 0.00071729 - 0.000493703 - 0.000332947 - 0.000219998 - 0.00014243 - 9.03484e-05 - 5.61534e-05 - 3.41955e-05 - 2.04032e-05 - 1.19279e-05 - 6.83231e-06 - 3.83449e-06 - 2.10855e-06 - 1.13605e-06 - 5.9972e-07 - 3.10196e-07 - 1.57203e-07 - 7.80587e-08 - 3.79768e-08 - 1.81031e-08 - 8.45523e-09 - 3.86931e-09 - 1.73492e-09 - 7.62186e-10 - 3.2808e-10 - 1.38368e-10 - 5.71779e-11 - 2.31504e-11 - 9.18384e-12 - 3.56967e-12 - 1.35946e-12 - 5.07276e-13 - 1.85463e-13 - 6.64368e-14 - 2.33183e-14 - 8.01901e-15 - 2.70198e-15 - 8.9203e-16 - 2.88546e-16 - 9.14505e-17 - 2.83985e-17 - 8.64052e-18 - 2.57586e-18 - 7.52387e-19 - 2.15326e-19 - 6.03798e-20 - 1.65898e-20 - 4.46888e-21 - 1.1907e-21 - 3.54724e-22 - 2.70303e-22 - 7.45922e-22 - 2.68317e-21 - 9.67303e-21 - 3.42192e-20 - 1.1862e-19 - 4.02889e-19 - 1.34076e-18 - 4.37171e-18 - 1.39666e-17 - 4.37185e-17 - 1.34084e-16 - 4.02927e-16 - 1.18635e-15 - 3.42243e-15 - 9.67374e-15 - 2.67911e-14 - 7.26984e-14 - 1.93284e-13 - 5.03504e-13 - 1.28513e-12 - 3.21388e-12 - 7.87496e-12 - 1.89062e-11 - 4.44733e-11 - 1.02502e-10 - 2.31473e-10 - 5.12165e-10 - 1.11034e-09 - 2.35855e-09 - 4.90879e-09 - 1.00102e-08 - 2.00013e-08 - 3.91579e-08 - 7.51153e-08 - 1.41185e-07 - 2.60021e-07 - 4.69236e-07 - 8.29749e-07 - 1.43776e-06 - 2.44132e-06 - 4.06237e-06 - 6.62491e-06 - 1.0589e-05 - 1.65903e-05 - 2.54819e-05 - 3.83764e-05 - 5.66822e-05 - 8.21314e-05 - 0.000116792 - 0.00016307 - 0.000223695 - 0.000301717 - 0.000400522 - 0.000523905 - 0.000676234 - 0.000862732 - 0.00108992 - 0.00136617 - 0.00170242 - 0.00211279 - 0.00261515 - 0.00323128 - 0.00398644 - 0.0049082 - 0.00602429 - 0.00735944 - 0.00893147 - 0.0107468 - 0.0127958 - 0.0150493 - 0.0174561 - 0.0199423 - 0.0224143 - 0.0247631 - 0.0268725 - 0.0286279 - 0.0299271 - 0.0306897 - 0.0308656 - 0.0304406 - 0.0294379 - 0.027917 - 0.0259682 - 0.0237049 - 0.0212543 - 0.0187473 - 0.0163094 - 0.0140532 - 0.012073 - 0.0104412 - 0.00920708 - 0.00839714 - 0.00801617 - 0.00804901 - 0.00846236 - 0.00920671 - 0.0102183 - 0.0114214 - 0.0127314 - 0.0140583 - 0.0153113 - 0.0164042 - 0.0172608 - 0.0178203 - 0.0180418 - 0.0179068 - 0.017421 - 0.0166126 - 0.0155294 - 0.0142346 - 0.0128009 - 0.0113044 - 0.00981974 - 0.00841484 - 0.00714807 - 0.00606625 - 0.00520401 - 0.00458421 - 0.00421879 - 0.00410993 - 0.00425091 - 0.00462675 - 0.00521439 - 0.00598278 - 0.0068929 - 0.00789827 - 0.00894605 - 0.00997897 - 0.0109382 - 0.0117671 - 0.0124148 - 0.0128403 - 0.0130161 - 0.01293 - 0.0125862 - 0.0120047 - 0.011219 - 0.0102732 - 0.00921717 - 0.00810268 - 0.00697907 - 0.00588986 - 0.00487025 - 0.00394581 - 0.00313231 - 0.00243638 - 0.00185693 - 0.00138696 - 0.00101542 - 0.000729103 - 0.000514093 - 0.000357004 - 0.000245774 - 0.00017014 - 0.00012182 - 9.44836e-05 - 8.35732e-05 - 8.60424e-05 - 0.000100046 - 0.000124617 - 0.000159347 - 0.000204081 - 0.000258632 - 0.000322534 - 0.000394841 - 0.000473977 - 0.000557675 - 0.000642995 - 0.000726438 - 0.000804153 - 0.00087221 - 0.00092692 - 0.000965165 - 0.000984686 - 0.000984308 - 0.000964054 - 0.000925143 - 0.000869867 - 0.000801371 - 0.000723355 - 0.000639744 - 0.000554367 - 0.00047068 - 0.000391553 - 0.000319149 - 0.000254878 - 0.000199438 - 0.000152904 - 0.00011486 - 8.45387e-05 - 6.09647e-05 - 4.30763e-05 - 2.98219e-05 - 2.02287e-05 - 1.34443e-05 - 8.75481e-06 - 5.58588e-06 - 3.49199e-06 - 2.13892e-06 - 1.2837e-06 - 7.54953e-07 - 4.3522e-07 - 2.46288e-07 - 1.37595e-07 - 7.76206e-08 - 4.7872e-08 - 3.91206e-08 - 4.96818e-08 - 8.57782e-08 - 1.64423e-07 - 3.19802e-07 - 6.1486e-07 - 1.16078e-06 - 2.1483e-06 - 3.8961e-06 - 6.92335e-06 - 1.20543e-05 - 2.0564e-05 - 3.43723e-05 - 5.6292e-05 - 9.03278e-05 - 0.000142015 - 0.000218767 - 0.000330193 - 0.000488304 - 0.000707537 - 0.00100449 - 0.00139726 - 0.00190436 - 0.00254305 - 0.00332735 - 0.00426559 - 0.00535792 - 0.00659401 - 0.00795135 - 0.00939439 - 0.0108751 - 0.0123349 - 0.013708 - 0.0149262 - 0.0159243 - 0.016646 - 0.0170489 - 0.0171089 - 0.0168223 - 0.0162065 - 0.0152982 - 0.0141497 - 0.0128242 - 0.01139 - 0.00991533 - 0.00846323 - 0.00708799 - 0.00583302 - 0.00473027 - 0.00380108 - 0.00305813 - 0.00250808 - 0.00215444 - 0.00200017 - 0.00204972 - 0.00231011 - 0.00279091 - 0.003503 - 0.00445631 - 0.00565652 - 0.00710137 - 0.0087768 - 0.0106537 - 0.0126858 - 0.014809 - 0.0169432 - 0.0189962 - 0.0208692 - 0.0224645 - 0.0236937 - 0.0244855 - 0.0247927 - 0.0245967 - 0.0239093 - 0.0227716 - 0.0212499 - 0.0194294 - 0.0174059 - 0.0152782 - 0.0131396 - 0.0110721 - 0.00914149 - 0.00739501 - 0.00586135 - 0.00455191 - 0.00346359 - 0.00258224 - 0.00188626 - 0.00135004 - 0.000946729 - 0.000650493 - 0.000437923 - 0.000288862 - 0.000186692 - 0.000118227 - 7.33688e-05 - 4.46338e-05 - 2.66504e-05 - 1.56821e-05 - 9.21728e-06 - 5.63909e-06 - 3.97946e-06 - 3.75189e-06 - 4.85588e-06 - 7.54564e-06 - 1.24576e-05 - 2.06929e-05 - 3.39515e-05 - 5.47126e-05 - 8.64515e-05 - 0.000133876 - 0.000203149 - 0.000302064 - 0.000440104 - 0.00062834 - 0.000879078 - 0.00120523 - 0.00161937 - 0.00213248 - 0.00275244 - 0.00348253 - 0.00431992 - 0.00525458 - 0.00626873 - 0.00733715 - 0.00842831 - 0.00950652 - 0.0105347 - 0.0114778 - 0.0123057 - 0.0129966 - 0.0135384 - 0.0139296 - 0.0141784 - 0.0143005 - 0.0143162 - 0.0142468 - 0.0141109 - 0.0139217 - 0.0136857 - 0.0134019 - 0.0130632 - 0.012659 - 0.0121776 - 0.0116098 - 0.0109513 - 0.0102046 - 0.00937978 - 0.00849391 - 0.00756962 - 0.00663291 - 0.00571057 - 0.00482772 - 0.00400577 - 0.00326099 - 0.00260377 - 0.00203868 - 0.00156499 - 0.00117773 - 0.000868844 - 0.000628451 - 0.000445965 - 0.00031104 - 0.00021428 - 0.000147749 - 0.000105306 - 8.28571e-05 - 7.85795e-05 - 9.31949e-05 - 0.000130355 - 0.000197161 - 0.00030483 - 0.000469447 - 0.000712728 - 0.00106262 - 0.00155357 - 0.00222613 - 0.00312575 - 0.00430043 - 0.00579713 - 0.0076569 - 0.00990902 - 0.0125645 - 0.0156098 - 0.0190014 - 0.0226627 - 0.0264835 - 0.0303231 - 0.0340181 - 0.0373924 - 0.0402711 - 0.0424952 - 0.0439363 - 0.0445086 - 0.0441776 - 0.0429631 - 0.040938 - 0.0382204 - 0.0349623 - 0.0313358 - 0.0275182 - 0.0236774 - 0.0199612 - 0.0164883 - 0.0133445 - 0.0105819 - 0.00822174 - 0.00625893 - 0.00466846 - 0.0034118 - 0.00244304 - 0.00171401 - 0.00117824 - 0.000793582 - 0.000523705 - 0.000338624 - 0.000214529 - 0.000133165 - 8.09901e-05 - 4.82626e-05 - 2.8179e-05 - 1.61205e-05 - 9.03581e-06 - 4.96241e-06 - 2.67027e-06 - 1.40785e-06 - 7.27263e-07 - 3.68099e-07 - 1.82547e-07 - 8.87001e-08 - 4.22295e-08 - 1.97008e-08 - 9.00956e-09 - 4.04833e-09 - 1.81029e-09 - 8.60698e-10 - 5.58336e-10 - 6.85971e-10 - 1.31585e-09 - 2.8408e-09 - 6.1788e-09 - 1.32343e-08 - 2.77987e-08 - 5.7221e-08 - 1.15408e-07 - 2.28062e-07 - 4.4158e-07 - 8.37727e-07 - 1.55716e-06 - 2.83595e-06 - 5.0606e-06 - 8.84794e-06 - 1.51572e-05 - 2.5441e-05 - 4.18393e-05 - 6.74174e-05 - 0.000106438 - 0.000164648 - 0.000249549 - 0.000370588 - 0.000539216 - 0.000768726 - 0.00107378 - 0.0014696 - 0.00197069 - 0.00258926 - 0.00333325 - 0.00420434 - 0.00519594 - 0.0062917 - 0.00746463 - 0.00867732 - 0.00988325 - 0.0110294 - 0.0120597 - 0.01292 - 0.013562 - 0.0139483 - 0.0140559 - 0.0138781 - 0.0134257 - 0.0127258 - 0.0118186 - 0.0107544 - 0.00958832 - 0.00837598 - 0.00716911 - 0.00601218 - 0.0049401 - 0.00397719 - 0.00313728 - 0.00242475 - 0.00183619 - 0.00136241 - 0.000990447 - 0.000705493 - 0.00049237 - 0.000336687 - 0.000225579 - 0.000148084 - 9.52474e-05 - 6.00255e-05 - 3.70643e-05 - 2.24239e-05 - 1.32924e-05 - 7.72031e-06 - 4.39341e-06 - 2.44966e-06 - 1.33828e-06 - 7.16347e-07 - 3.75697e-07 - 1.93059e-07 - 9.7204e-08 - 4.79558e-08 - 2.31886e-08 - 1.10053e-08 - 5.16537e-09 - 2.49231e-09 - 1.45573e-09 - 1.44813e-09 - 2.49265e-09 - 5.27515e-09 - 1.15271e-08 - 2.49327e-08 - 5.29399e-08 - 1.10176e-07 - 2.24677e-07 - 4.48922e-07 - 8.78862e-07 - 1.68581e-06 - 3.16834e-06 - 5.83436e-06 - 1.05267e-05 - 1.86091e-05 - 3.22327e-05 - 5.47023e-05 - 9.09601e-05 - 0.000148195 - 0.000236566 - 0.000370005 - 0.000567022 - 0.000851392 - 0.00125255 - 0.00180551 - 0.00254999 - 0.00352871 - 0.00478442 - 0.00635594 - 0.00827308 - 0.010551 - 0.0131842 - 0.0161418 - 0.0193636 - 0.0227592 - 0.02621 - 0.0295742 - 0.0326961 - 0.0354175 - 0.0375905 - 0.0390913 - 0.0398314 - 0.0397669 - 0.0389028 - 0.0372926 - 0.0350338 - 0.0322585 - 0.0291224 - 0.0257915 - 0.02243 - 0.0191902 - 0.0162049 - 0.0135829 - 0.0114083 - 0.00974183 - 0.00862375 - 0.00807727 - 0.00811156 - 0.00872333 - 0.00989679 - 0.011602 - 0.0137919 - 0.0163994 - 0.0193348 - 0.0224847 - 0.0257144 - 0.0288725 - 0.0317988 - 0.0343349 - 0.0363363 - 0.0376843 - 0.0382965 - 0.0381344 - 0.037207 - 0.0355694 - 0.0333172 - 0.0305773 - 0.0274959 - 0.0242255 - 0.020913 - 0.0176887 - 0.0146593 - 0.0119033 - 0.00947019 - 0.00738221 - 0.00563834 - 0.00421942 - 0.0030938 - 0.00222264 - 0.00156455 - 0.0010791 - 0.000729329 - 0.000483146 - 0.000313949 - 0.000200581 - 0.000126914 - 8.12404e-05 - 5.5656e-05 - 4.55669e-05 - 4.94212e-05 - 6.87538e-05 - 0.000108612 - 0.000178427 - 0.000293379 - 0.000476297 - 0.000760061 - 0.00119043 - 0.00182912 - 0.00275668 - 0.00407487 - 0.00590754 - 0.00839955 - 0.0117125 - 0.0160171 - 0.0214803 - 0.0282497 - 0.0364326 - 0.0460744 - 0.0571362 - 0.0694755 - 0.0828344 - 0.0968356 - 0.110993 - 0.124731 - 0.137426 - 0.148443 - 0.157195 - 0.163191 - 0.16608 - 0.16569 - 0.162039 - 0.155338 - 0.145969 - 0.13445 - 0.121386 - 0.107417 - 0.0931679 - 0.0792032 - 0.0659922 - 0.0538902 - 0.0431307 - 0.0338313 - 0.0260076 - 0.0195944 - 0.0144686 - 0.0104718 - 0.00743065 - 0.00517275 - 0.00353869 - 0.0023892 - 0.00160904 - 0.00110793 - 0.000819713 - 0.000700487 - 0.000726235 - 0.000890471 - 0.00120196 - 0.00168246 - 0.00236436 - 0.00328777 - 0.00449696 - 0.00603582 - 0.00794235 - 0.0102424 - 0.0129428 - 0.0160253 - 0.0194414 - 0.0231094 - 0.0269146 - 0.0307131 - 0.0343397 - 0.0376189 - 0.0403787 - 0.0424655 - 0.0437579 - 0.0441788 - 0.0437027 - 0.0423583 - 0.040226 - 0.0374293 - 0.0341234 - 0.0304811 - 0.0266776 - 0.022877 - 0.0192215 - 0.0158239 - 0.0127637 - 0.0100873 - 0.00781109 - 0.00592634 - 0.00440555 - 0.00320891 - 0.0022902 - 0.00160171 - 0.00109801 - 0.000738346 - 0.000488074 - 0.000319114 - 0.000209886 - 0.000144899 - 0.000114287 - 0.00011345 - 0.000142967 - 0.000208865 - 0.000323272 - 0.000505441 - 0.000783052 - 0.00119362 - 0.00178573 - 0.00261982 - 0.00376788 - 0.00531191 - 0.00734032 - 0.0099422 - 0.0131993 - 0.0171757 - 0.0219067 - 0.0273861 - 0.0335566 - 0.0403008 - 0.0474392 - 0.0547326 - 0.0618925 - 0.0685978 - 0.074518 - 0.0793391 - 0.0827917 - 0.0846753 - 0.084878 - 0.0833871 - 0.0802907 - 0.0757691 - 0.0700771 - 0.0635207 - 0.0564296 - 0.0491301 - 0.0419213 - 0.0350564 - 0.0287303 - 0.0230756 - 0.0181636 - 0.0140114 - 0.0105924 - 0.00784763 - 0.00569779 - 0.00405413 - 0.0028269 - 0.00193171 - 0.00129357 - 0.000848892 - 0.000545916 - 0.00034404 - 0.000212471 - 0.000128587 - 7.62598e-05 - 4.43196e-05 - 2.52403e-05 - 1.40861e-05 - 7.70341e-06 - 4.12826e-06 - 2.16792e-06 - 1.1156e-06 - 5.62553e-07 - 2.77973e-07 - 1.34595e-07 - 6.38609e-08 - 2.96909e-08 - 1.35267e-08 - 6.03865e-09 - 2.64158e-09 - 1.13231e-09 - 4.75596e-10 - 1.95742e-10 - 7.89412e-11 - 3.11955e-11 - 1.20796e-11 - 4.58331e-12 - 1.70407e-12 - 6.20988e-13 - 2.22293e-13 - 7.9713e-14 - 3.33658e-14 - 2.94752e-14 - 6.38018e-14 - 1.77569e-13 - 5.02538e-13 - 1.39959e-12 - 3.82147e-12 - 1.02254e-11 - 2.6812e-11 - 6.88931e-11 - 1.73468e-10 - 4.28014e-10 - 1.03489e-09 - 2.45205e-09 - 5.69327e-09 - 1.29536e-08 - 2.88814e-08 - 6.31019e-08 - 1.35103e-07 - 2.83456e-07 - 5.82778e-07 - 1.17414e-06 - 2.3181e-06 - 4.48481e-06 - 8.50263e-06 - 1.57965e-05 - 2.87584e-05 - 5.13057e-05 - 8.96943e-05 - 0.00015366 - 0.000257961 - 0.00042437 - 0.000684119 - 0.00108072 - 0.001673 - 0.00253788 - 0.00377261 - 0.00549553 - 0.00784462 - 0.0109731 - 0.0150412 - 0.0202036 - 0.0265931 - 0.0343008 - 0.0433543 - 0.0536975 - 0.0651733 - 0.0775138 - 0.0903404 - 0.103176 - 0.115472 - 0.12664 - 0.136103 - 0.143344 - 0.147949 - 0.149655 - 0.14837 - 0.144186 - 0.137373 - 0.128353 - 0.117658 - 0.105889 - 0.093658 - 0.0815473 - 0.0700646 - 0.0596157 - 0.0504871 - 0.0428422 - 0.0367268 - 0.0320839 - 0.0287735 - 0.0265961 - 0.0253153 - 0.0246804 - 0.024446 - 0.0243882 - 0.0243168 - 0.0240838 - 0.0235875 - 0.0227725 - 0.021627 - 0.0201755 - 0.0184715 - 0.0165872 - 0.014604 - 0.0126035 - 0.0106603 - 0.00883602 - 0.00717679 - 0.00571177 - 0.00445418 - 0.00340341 - 0.00254803 - 0.00186912 - 0.00134342 - 0.000946069 - 0.000652789 - 0.000441327 - 0.000292338 - 0.000189735 - 0.000120655 - 7.51766e-05 - 4.5894e-05 - 2.74515e-05 - 1.60884e-05 - 9.2384e-06 - 5.19778e-06 - 2.86533e-06 - 1.54764e-06 - 8.19038e-07 - 4.24702e-07 - 2.15801e-07 - 1.07503e-07 - 5.26328e-08 - 2.56405e-08 - 1.31726e-08 - 8.77456e-09 - 1.02078e-08 - 1.85691e-08 - 3.89947e-08 - 8.33138e-08 - 1.75745e-07 - 3.63781e-07 - 7.38012e-07 - 1.46706e-06 - 2.85743e-06 - 5.45305e-06 - 1.01963e-05 - 1.86801e-05 - 3.35317e-05 - 5.89749e-05 - 0.000101629 - 0.000171594 - 0.000283873 - 0.000460134 - 0.00073077 - 0.00113714 - 0.00173374 - 0.00258995 - 0.00379084 - 0.00543645 - 0.00763894 - 0.0105169 - 0.0141866 - 0.0187503 - 0.0242813 - 0.0308088 - 0.0383013 - 0.0466542 - 0.0556807 - 0.0651114 - 0.0746018 - 0.0837495 - 0.0921213 - 0.0992861 - 0.104852 - 0.108503 - 0.110031 - 0.109357 - 0.106543 - 0.101791 - 0.0954271 - 0.087877 - 0.0796374 - 0.0712426 - 0.0632336 - 0.0561318 - 0.0504187 - 0.0465218 - 0.044805 - 0.045561 - 0.0490042 - 0.0552601 - 0.0643534 - 0.0761915 - 0.0905495 - 0.107057 - 0.125194 - 0.144296 - 0.163577 - 0.182163 - 0.199146 - 0.213648 - 0.224879 - 0.232211 - 0.235225 - 0.233747 - 0.227864 - 0.217912 - 0.204443 - 0.188177 - 0.169934 - 0.150572 - 0.130918 - 0.111717 - 0.0935927 - 0.0770224 - 0.0623326 - 0.0497065 - 0.0392039 - 0.0307866 - 0.0243474 - 0.0197367 - 0.0167861 - 0.0153248 - 0.0151908 - 0.0162352 - 0.0183228 - 0.0213286 - 0.0251345 - 0.0296256 - 0.0346886 - 0.0402135 - 0.0460961 - 0.052243 - 0.0585756 - 0.0650326 - 0.0715684 - 0.0781472 - 0.0847335 - 0.091278 - 0.0977038 - 0.103894 - 0.109684 - 0.114865 - 0.119189 - 0.122391 - 0.124217 - 0.124448 - 0.122933 - 0.119609 - 0.114518 - 0.107811 - 0.0997331 - 0.0906066 - 0.0808031 - 0.0707102 - 0.0607003 - 0.0511033 - 0.0421863 - 0.0341421 - 0.0270865 - 0.0210628 - 0.0160525 - 0.0119898 - 0.00877597 - 0.00629478 - 0.00442438 - 0.00304721 - 0.00205646 - 0.00135988 - 0.000881132 - 0.000559434 - 0.000348064 - 0.000212277 - 0.000127044 - 7.49014e-05 - 4.40862e-05 - 2.70444e-05 - 1.93388e-05 - 1.89574e-05 - 2.60199e-05 - 4.28955e-05 - 7.47722e-05 - 0.00013075 - 0.000225561 - 0.000382019 - 0.000634315 - 0.00103218 - 0.00164583 - 0.00257148 - 0.00393681 - 0.00590566 - 0.00868071 - 0.0125027 - 0.0176447 - 0.0243998 - 0.0330614 - 0.0438952 - 0.0571052 - 0.0727939 - 0.0909236 - 0.111281 - 0.133452 - 0.156816 - 0.180558 - 0.203706 - 0.225193 - 0.24393 - 0.258902 - 0.269258 - 0.274385 - 0.273977 - 0.268058 - 0.256982 - 0.241401 - 0.222195 - 0.200397 - 0.177096 - 0.153352 - 0.130119 - 0.108184 - 0.088143 - 0.0703811 - 0.0550906 - 0.0422952 - 0.0318886 - 0.0236757 - 0.0174137 - 0.0128489 - 0.00974575 - 0.00790899 - 0.00719695 - 0.00752776 - 0.00887847 - 0.0112782 - 0.0147954 - 0.0195211 - 0.0255467 - 0.03294 - 0.0417194 - 0.0518294 - 0.0631202 - 0.0753344 - 0.0881046 - 0.100962 - 0.113362 - 0.124714 - 0.134431 - 0.141978 - 0.146921 - 0.148964 - 0.147984 - 0.144041 - 0.137371 - 0.128363 - 0.117523 - 0.105425 - 0.0926619 - 0.0797996 - 0.0673356 - 0.0556729 - 0.0451043 - 0.0358108 - 0.0278695 - 0.0212705 - 0.0159377 - 0.0117507 - 0.0085666 - 0.0062372 - 0.00462324 - 0.00360352 - 0.00307992 - 0.00297862 - 0.00324833 - 0.00385612 - 0.00478184 - 0.00601151 - 0.00753051 - 0.0093171 - 0.011337 - 0.0135396 - 0.0158561 - 0.0182006 - 0.0204731 - 0.0225658 - 0.0243709 - 0.0257892 - 0.0267388 - 0.0271636 - 0.0270377 - 0.0263689 - 0.0251973 - 0.0235919 - 0.0216433 - 0.019456 - 0.0171389 - 0.014797 - 0.0125243 - 0.0103983 - 0.00847798 - 0.00680284 - 0.00539503 - 0.00426232 - 0.00340205 - 0.00280523 - 0.00246013 - 0.00235501 - 0.00247968 - 0.00282577 - 0.00338583 - 0.00415139 - 0.00511045 - 0.00624471 - 0.00752714 - 0.00892027 - 0.0103758 - 0.0118353 - 0.0132334 - 0.0145009 - 0.0155705 - 0.0163822 - 0.0168885 - 0.017059 - 0.0168832 - 0.0163718 - 0.0155551 - 0.0144807 - 0.0132081 - 0.011804 - 0.0103361 - 0.00886787 - 0.00745451 - 0.00613983 - 0.00495484 - 0.00391778 - 0.0030352 - 0.00230394 - 0.00171353 - 0.00124868 - 0.000891548 - 0.000623701 - 0.000427509 - 0.000287111 - 0.000188926 - 0.000121807 - 7.69461e-05 - 4.76255e-05 - 2.88821e-05 - 1.71615e-05 - 9.99121e-06 - 5.69926e-06 - 3.18533e-06 - 1.74433e-06 - 9.35921e-07 - 4.92024e-07 - 2.53438e-07 - 1.27906e-07 - 6.32485e-08 - 3.06439e-08 - 1.45471e-08 - 6.7662e-09 - 3.08354e-09 - 1.37687e-09 - 6.02381e-10 - 2.58218e-10 - 1.08453e-10 - 4.46303e-11 - 1.79952e-11 - 7.10917e-12 - 2.75181e-12 - 1.04365e-12 - 3.8782e-13 - 1.41202e-13 - 5.03719e-14 - 1.76065e-14 - 6.02967e-15 - 2.02326e-15 - 6.65191e-16 - 2.14278e-16 - 6.76311e-17 - 2.09147e-17 - 6.33715e-18 - 1.88137e-18 - 5.47254e-19 - 1.5597e-19 - 4.35544e-20 - 1.19168e-20 - 3.19464e-21 - 8.39116e-22 - 2.15953e-22 - 5.44544e-23 - 1.34538e-23 - 3.2568e-24 - 7.72459e-25 - 1.79513e-25 - 4.08747e-26 - 9.11906e-27 - 1.99334e-27 - 4.26924e-28 - 8.95892e-29 - 1.84204e-29 - 3.71088e-30 - 7.32476e-31 - 1.4166e-31 - 2.68433e-32 - 4.98382e-33 - 9.06622e-34 - 1.61594e-34 - 2.82204e-35 - 4.82878e-36 - 8.09559e-37 - 1.32983e-37 - 2.14033e-38 - 3.37522e-39 - 5.21507e-40 - 7.89506e-41 - 1.17108e-41 - 1.70199e-42 - 2.4236e-43 - 3.38146e-44 - 4.62257e-45 - 6.19155e-46 - 8.12554e-47 - 1.04482e-47 - 1.31634e-48 - 1.62492e-49 - 1.96532e-50 - 2.329e-51 - 2.70423e-52 - 3.07649e-53 - 3.42928e-54 - 3.74531e-55 - 4.00783e-56 - 4.20211e-57 - 4.31681e-58 - 4.34505e-59 - 4.28512e-60 - 4.14066e-61 - 3.92023e-62 - 3.63656e-63 - 3.30527e-64 - 2.94348e-65 - 2.56833e-66 - 2.19572e-67 - 1.83925e-68 - 1.50953e-69 - 1.21389e-70 - 9.56429e-72 - 7.38353e-73 - 5.58485e-74 - 4.13901e-75 - 3.00551e-76 - 2.13834e-77 - 1.49064e-78 - 1.01813e-79 - 6.81354e-81 - 4.46764e-82 - 2.87026e-83 - 1.80676e-84 - 1.11434e-85 - 6.73396e-87 - 3.98713e-88 - 2.31306e-89 - 1.31477e-90 - 7.32237e-92 - 3.99566e-93 - 2.1363e-94 - 1.11911e-95 - 5.7441e-97 - 2.88873e-98 - 1.42341e-99 - 6.87207e-101 - 3.25074e-102 - 1.50666e-103 - 6.84201e-105 - 3.04432e-106 - 1.32719e-107 - 5.66909e-109 - 2.37623e-110 - 1.87428e-111 - 2.20949e-110 - 5.28801e-109 - 1.24221e-107 - 2.85914e-106 - 6.44781e-105 - 1.42471e-103 - 3.08445e-102 - 6.54282e-101 - 1.35985e-99 - 2.76918e-98 - 5.52523e-97 - 1.08015e-95 - 2.06899e-94 - 3.88299e-93 - 7.14023e-92 - 1.28646e-90 - 2.27099e-89 - 3.92799e-88 - 6.65678e-87 - 1.10534e-85 - 1.79829e-84 - 2.86658e-83 - 4.47719e-82 - 6.85145e-81 - 1.0273e-79 - 1.5092e-78 - 2.17238e-77 - 3.0638e-76 - 4.23372e-75 - 5.7322e-74 - 7.60425e-73 - 9.88391e-72 - 1.25875e-70 - 1.57067e-69 - 1.92029e-68 - 2.30031e-67 - 2.69987e-66 - 3.10482e-65 - 3.49837e-64 - 3.86218e-63 - 4.1777e-62 - 4.42769e-61 - 4.59785e-60 - 4.6781e-59 - 4.66359e-58 - 4.55521e-57 - 4.35947e-56 - 4.08786e-55 - 3.75573e-54 - 3.38088e-53 - 2.98196e-52 - 2.57698e-51 - 2.18201e-50 - 1.81026e-49 - 1.4715e-48 - 1.17197e-47 - 9.14556e-47 - 6.99263e-46 - 5.23851e-45 - 3.84514e-44 - 2.76537e-43 - 1.94863e-42 - 1.34538e-41 - 9.10115e-41 - 6.03233e-40 - 3.91751e-39 - 2.49271e-38 - 1.55407e-37 - 9.49308e-37 - 5.68172e-36 - 3.33188e-35 - 1.91442e-34 - 1.07775e-33 - 5.94484e-33 - 3.2129e-32 - 1.70134e-31 - 8.8272e-31 - 4.48736e-30 - 2.2351e-29 - 1.09078e-28 - 5.21577e-28 - 2.44362e-27 - 1.12173e-26 - 5.0452e-26 - 2.22334e-25 - 9.6e-25 - 4.06138e-24 - 1.6835e-23 - 6.83739e-23 - 2.72085e-22 - 1.06086e-21 - 4.05272e-21 - 1.51696e-20 - 5.56339e-20 - 1.99914e-19 - 7.03857e-19 - 2.42809e-18 - 8.20698e-18 - 2.71795e-17 - 8.81938e-17 - 2.80398e-16 - 8.73476e-16 - 2.66605e-15 - 7.97306e-15 - 2.33627e-14 - 6.70756e-14 - 1.8869e-13 - 5.20086e-13 - 1.40458e-12 - 3.71676e-12 - 9.63669e-12 - 2.44816e-11 - 6.09399e-11 - 1.48633e-10 - 3.55208e-10 - 8.31775e-10 - 1.90848e-09 - 4.29072e-09 - 9.45234e-09 - 2.04041e-08 - 4.31586e-08 - 8.94536e-08 - 1.81682e-07 - 3.61591e-07 - 7.05214e-07 - 1.34782e-06 - 2.52441e-06 - 4.63355e-06 - 8.33507e-06 - 1.46946e-05 - 2.53909e-05 - 4.30015e-05 - 7.13836e-05 - 0.000116157 - 0.000185287 - 0.000289756 - 0.000444259 - 0.000667872 - 0.000984563 - 0.00142341 - 0.00201839 - 0.00280749 - 0.00383118 - 0.00512992 - 0.00674098 - 0.0086945 - 0.0110092 - 0.013688 - 0.0167144 - 0.020049 - 0.0236291 - 0.0273681 - 0.0311587 - 0.0348773 - 0.0383902 - 0.0415613 - 0.0442609 - 0.0463735 - 0.0478061 - 0.0484944 - 0.0484071 - 0.0475482 - 0.0459567 - 0.043704 - 0.0408888 - 0.0376304 - 0.0340605 - 0.0303155 - 0.0265272 - 0.022816 - 0.0192849 - 0.016015 - 0.013064 - - - diff --git a/tools/02_postprocessing/plot-tools/examples/Si/README b/tools/02_postprocessing/plot-tools/examples/Si/README index 889eef6c74d..24b1038ee08 100644 --- a/tools/02_postprocessing/plot-tools/examples/Si/README +++ b/tools/02_postprocessing/plot-tools/examples/Si/README @@ -11,9 +11,13 @@ NiteL (1) The input files of this example is chosen from ABACUS/examples/02(c-d). Extra parameter "out_proj_band" for PBANDS calculation is needed in 02c. -(2) It is recommended that one use a script to parse PDOS and PBANDS, or you can do it by `python band.py` and `python dos.py` directly. +(2) The PDOS is now read from the text files written by ABACUS (`pdoss1_*.txt`, and `pdoss2_*.txt` for +spin-polarized calculations) instead of the old XML file. Set `pdosfile` in config.json to the directory or +prefix containing those files (e.g. the `OUT.*` directory, or the path prefix ending in `pdos`). -(3) There is a bug when use command-line to parse PBANDS: +(3) It is recommended that one use a script to parse PDOS and PBANDS, or you can do it by `python band.py` and `python dos.py` directly. + +(4) There is a bug when use command-line to parse PBANDS: ``` Traceback (most recent call last): File "F:\software\Anaconda\Scripts\abacus-plot-script.py", line 33, in diff --git a/tools/02_postprocessing/plot-tools/examples/Si/config.json b/tools/02_postprocessing/plot-tools/examples/Si/config.json index 7a175e88812..73fae7b4333 100644 --- a/tools/02_postprocessing/plot-tools/examples/Si/config.json +++ b/tools/02_postprocessing/plot-tools/examples/Si/config.json @@ -1,6 +1,6 @@ { "bandfile": "PBANDS_1", - "pdosfile": "PDOS", + "pdosfile": "pdos", "efermi": 6.585653952007503, "energy_range": [ -5, diff --git a/tools/02_postprocessing/rt-tddft-tools/examples/ground-state-projection-Si/INPUT b/tools/02_postprocessing/rt-tddft-tools/examples/ground-state-projection-Si/INPUT index 3e39ff78aae..34fe0d1040c 100644 --- a/tools/02_postprocessing/rt-tddft-tools/examples/ground-state-projection-Si/INPUT +++ b/tools/02_postprocessing/rt-tddft-tools/examples/ground-state-projection-Si/INPUT @@ -34,4 +34,4 @@ out_current_k 1 out_wfc_lcao 1 out_hsk 1 out_app_flag 0 -out_interval 25 +out_freq_ion 25 diff --git a/tools/02_postprocessing/rt-tddft-tools/overlap_generator.py b/tools/02_postprocessing/rt-tddft-tools/overlap_generator.py deleted file mode 100644 index d86f1af9405..00000000000 --- a/tools/02_postprocessing/rt-tddft-tools/overlap_generator.py +++ /dev/null @@ -1,241 +0,0 @@ -from pyatb.easy_use import stru_analyzer -from pyabacus import ModuleNAO as nao -from pyabacus import ModuleBase as base -import numpy as np -from collections import Counter -import pyatb -from pyatb.easy_use import stru_analyzer -import re -Bohr=0.5291772105638411 -class overlap_R: - def __init__(self, orb_file_dir, atoms0, atomst): - # get orbital information from the structure - orb_file_list = list(atoms0.info['basis'].values()) - orb_file_list = [orb_file_dir + orbfile for orbfile in orb_file_list] - orb_file_num = len(orb_file_list) - - symbols = atoms0.get_chemical_symbols() - element_counts = Counter(symbols) - unique_elements = set(symbols) - assert orb_file_num == len(unique_elements), f"The number of orbitals{orb_file_num}does not match the number of unique elements{len(unique_elements)}!" - - # Build orbital collection - self.orb = nao.RadialCollection() - self.orb.build(orb_file_num, orb_file_list, 'o') - - # Standardize the orbital grid - rmax = self.orb.rcut_max * 2.0 - dr = 0.01 - nr = int(rmax/dr) + 1 - self.orb.set_uniform_grid(True, nr, rmax, 'i', True) - - # Print basic orbital information - ntype = self.orb.ntype - lmax = self.orb.lmax - - # Initialize the integrator - self.S_intor = nao.TwoCenterIntegrator() - self.S_intor.tabulate(self.orb, self.orb, 'S', nr, rmax) - - # Retrieve orbital index sorting from the structure - # Determine the correspondence between the basis set indices and orbital information - self.lattice_vector = atoms0.get_cell()[:] / Bohr - self.atom_positions_c = atoms0.get_positions() / Bohr - self.atom_positions_ct = atomst.get_positions() / Bohr - self.cal_R_direct_coor() - - self.iw2it = dict() - self.iw2positions_c = dict() - self.iw2positions_ct = dict() - self.iw2iL = dict() - self.iw2iN = dict() - self.iw2im = dict() - count = 0 - count_atom = 0 - for it, element in enumerate(unique_elements): - for ia in range(element_counts[element]): - for iL in range(self.orb.lmax_(it)+1): - for iN in range(self.orb.nzeta(it, iL)): - for im in range(2*iL+1): - self.iw2it[count] = it - self.iw2positions_c[count] = self.atom_positions_c[count_atom] - self.iw2positions_ct[count] = self.atom_positions_ct[count_atom] - self.iw2iL[count] = iL - self.iw2iN[count] = iN - if im%2 == 0: - self.iw2im[count] = ((im+1) // 2) * -1 - else: - self.iw2im[count] = ((im+1) // 2) * 1 - - count = count + 1 - count_atom = count_atom + 1 - - def cal_R_direct_coor(self): - rcut=self.orb.rcut_max*np.ones(len(self.atom_positions_c),dtype=float) - print(rcut) - Ncell = NeighbourCell(self.lattice_vector, self.atom_positions_c, self.atom_positions_ct, rcut) - self.R_direct_coor = np.array(Ncell.check_interaction_neighbours()) - - def cal_overlap(self): - basis_num = len(self.iw2it) - self.SR = np.zeros((self.R_direct_coor.shape[0], basis_num, basis_num), dtype=float) - for iR, R_coor in enumerate(self.R_direct_coor): - for row in range(basis_num): - for col in range(basis_num): - result = self.__cal_S(row, col, R_coor) - self.SR[iR, row, col] = result[0] - return self.SR - - def __cal_S(self, iw1, iw2, dR): - dR = dR @ self.lattice_vector - dtau = self.iw2positions_c[iw2] - self.iw2positions_ct[iw1] - R = dR + dtau - - result = self.S_intor.calculate( - self.iw2it[iw1], - self.iw2iL[iw1], - self.iw2iN[iw1], - self.iw2im[iw1], - self.iw2it[iw2], - self.iw2iL[iw2], - self.iw2iN[iw2], - self.iw2im[iw2], - R, True) - - return result - def cal_Sk(self,kpoint_direct_coor): - basis_num = len(self.iw2it) - Sk = np.zeros([basis_num, basis_num], dtype=complex) - for iR in range(self.R_direct_coor.shape[0]): - arg = np.inner(kpoint_direct_coor, self.R_direct_coor[iR]) * 2 * np.pi - phase = complex(np.cos(arg), np.sin(arg)) - Sk = Sk + phase * self.SR[iR] - return Sk -def read_kpoints(file='.'): - file=file+'/kpoints' - with open(file,'r') as f: - read_flag=0 - for line in f: - if('nkstot now' in line): - nks_tot=int(line.split()[3]) - kpoints=np.zeros([nks_tot,3],dtype=float) - if(read_flag==1): - tmp=line.split() - kpoints[ik]=np.array([float(tmp[1]),float(tmp[2]),float(tmp[3])]) - ik+=1 - if(ik==nks_tot): - break - if('WEIGHT' in line): - read_flag=1 - ik=0 - return kpoints - -class NeighbourCell: - def __init__(self, lattice_vectors, atom_positions0, atom_positionst, cutoff_radii): - """ - Initialize lattice vectors, atom positions, and cutoff radii for each atom. - - lattice_vectors: Lattice vectors - - atom_positions: Atom positions - - cutoff_radii: Cutoff radii for each atom - """ - if len(lattice_vectors) != 3 or len(atom_positions0) == 0 or len(cutoff_radii) != len(atom_positions0): - raise ValueError("Invalid lattice vectors, atom positions or cutoff radii") - - self.lattice_vectors = np.array(lattice_vectors) - self.atom_positions0 = np.array(atom_positions0) - self.atom_positionst = np.array(atom_positionst) - self.cutoff_radii = np.array(cutoff_radii) - self.natoms = len(atom_positions0) - - def calculate_distance_to_neighbour(self, atom_index1, atom_index2, translation_vector): - """ - Calculate the distance between any two atoms - """ - if atom_index1 < 0 or atom_index1 >= self.natoms or \ - atom_index2 < 0 or atom_index2 >= self.natoms: - raise IndexError("Atom index out of range") - - displacement = self.atom_positionst[atom_index2] - self.atom_positions0[atom_index1] + (self.lattice_vectors@np.array(translation_vector).T).T - - # Return the Euclidean distance - return np.linalg.norm(displacement) - - def check_interaction_neighbours(self): - """ - Check for interactions between atoms in the unit cell and atoms in neighbouring unit cells. - atom_index: Index of the atom to check - translation_vector: The translation vector for the neighbouring unit cell - Returns a list of translation vectors for all neighbouring cells with interactions. - """ - interactions = [] # List to store translation vectors of neighbouring cells with interactions - # Loop over translation vectors - for t in range(-5, 5): - for u in range(-5, 5): - for v in range(-5, 5): - # Create a translation vector - neighbour_translation = np.array([t, u, v]) - # Loop through each atom in the unit cell and check interactions with atoms in neighbouring cells - if_find = False - for i in range(self.natoms): - for j in range(self.natoms): - distance = self.calculate_distance_to_neighbour(i, j, neighbour_translation) - # if overlap exists, add the translation vector to the list - if distance <= (self.cutoff_radii[i] + self.cutoff_radii[j]): - interactions.append(neighbour_translation) - if_find = True - break - if if_find: - break - return interactions - -def save_upper_triangle_as_matrix(matrix, filename): - # Open the file for writing - with open(filename, 'w') as f: - f.write(f'{matrix.shape[0]} ') - # Iterate over each row - for i in range(matrix.shape[0]): - # Extract the upper triangular part of the row - row = matrix[i, i:] - row_str = ' '.join(f'({x.real:.8e},{x.imag:.8e})' for x in row) - f.write(row_str + '\n') - -def overlap_gen(stepref: int, klist, steps, stru_dir='./OUT.ABACUS/STRU', orb_file_dir='',kpoits_dir='./OUT.ABACUS'): - """Generate overlap files for moved atoms - :params stepref: the ground state step - :params klist: the list for needed kpoints - :params steps: the list for needed steps - :params stru_dir: the directory for STRU files - :params orb_file_dir: the directory for orb files, same as 'orbital_dir' in the INPUT - :params kpoits_dir: the directory for kpoints files - """ - #read in ref STRU - with open(stru_dir+'/STRU_MD_'+str(stepref), 'r') as fd: - atoms0 = stru_analyzer.read_abacus_stru(fd, verbose=True) - #read in kpoints - kpt=read_kpoints(kpoits_dir) - #loop over steps - for n, nstep in enumerate(steps): - #read in time t STRU - with open(stru_dir+'/STRU_MD_'+str(nstep), 'r') as fd: - atomst = stru_analyzer.read_abacus_stru(fd, verbose=True) - SR_tmp = overlap_R(orb_file_dir, atoms0, atomst) - SR_tmp.cal_overlap() - #loop over kpoints needed - for ik in klist: - Sk_ik = SR_tmp.cal_Sk(kpt[ik]) - save_upper_triangle_as_matrix(Sk_ik, f'{nstep}_data-{ik}-S') - - -#warning : pyatb and pyabacus are required, install them first -if __name__ == "__main__": - #the kpoints you need, check kpoints file to get the index - klist=[0] - #the steps you need, check STRU_MD file to get the index - start_step = 0 - end_step = 20 - out_interval = 5 - steps = range(start_step, end_step, out_interval) - #the ground state step - stepref=0 - #generate overlap files for moved atoms - overlap_gen(stepref, klist, steps) diff --git a/tools/03_code_analysis/agent_governance_check.py b/tools/03_code_analysis/agent_governance_check.py index 561773872c7..7992d729b8b 100644 --- a/tools/03_code_analysis/agent_governance_check.py +++ b/tools/03_code_analysis/agent_governance_check.py @@ -247,6 +247,8 @@ def check_line_endings( # `#define private public` / `#define protected public`. See AGENTS.md rule 10. ACCESS_HACK_RE = re.compile(r"^\s*#\s*define\s+(?:private|protected)\s+public\b") +# `#pragma once`. See AGENTS.md rule 15. +PRAGMA_ONCE_RE = re.compile(r"^\s*#\s*pragma\s+once\b") def is_global_dependency_check_path(path: str) -> bool: @@ -362,6 +364,29 @@ def check_access_hacks( ) +def check_pragma_once( + findings: List[Finding], + added_lines: Iterable[DiffLine], +) -> None: + """Block new `#pragma once` in headers (AGENTS.md rule 15). + + Use traditional `#ifndef`/`#define`/`#endif` include guards instead. + """ + for line in added_lines: + if Path(line.path).suffix.lower() not in HEADER_EXTENSIONS | {".cuh"}: + continue + if PRAGMA_ONCE_RE.search(line.content): + add_finding( + findings, + "No #pragma once", + BLOCK, + line.path, + line.line, + "Adds `#pragma once` to a header file.", + "Use traditional `#ifndef`/`#define`/`#endif` include guards instead.", + allow_exception=False, + ) + def _has_default_arg_in_parens(stripped: str) -> bool: paren_depth = 0 @@ -799,6 +824,7 @@ def collect_findings(root: Path, args: argparse.Namespace) -> List[Finding]: check_line_endings(findings, root, changed, statuses, args) check_global_dependencies(findings, lines, removed_lines) check_access_hacks(findings, lines, removed_lines) + check_pragma_once(findings, lines) check_default_parameters(findings, lines) check_hpp_warnings(findings, statuses, lines) check_header_include_warnings(findings, lines) diff --git a/tools/03_code_analysis/code_quality_score.py b/tools/03_code_analysis/code_quality_score.py old mode 100644 new mode 100755 index 77fbb2b97dd..97f25bd2d7d --- a/tools/03_code_analysis/code_quality_score.py +++ b/tools/03_code_analysis/code_quality_score.py @@ -13,7 +13,8 @@ - All caps listed below are per-file limits on the cumulative deduction for that rule within a single file. - Files under any of these directories are skipped entirely: test/, - tests/, test_serial/, test_parallel/, test_gpu/, unit_test/, unittest/. + tests/, test_serial/, test_parallel/, test_gpu/, unit_test/, unittest/, + unittests/. - Pass threshold: score >= 60. Output is sorted by score ascending (worst files first); ties are broken by `real_score` ascending (more negative = more total deduction = ranked first). Text and @@ -119,7 +120,7 @@ ".git", "build", "__pycache__", "node_modules", ".cache", "third_party", "thirdparty", ".vscode", ".idea", ".trae-cn", "Dependencies", - "test", "tests", "test_serial", "test_parallel", "test_gpu", "unit_test", "unittest", + "test", "tests", "test_serial", "test_parallel", "test_gpu", "unit_test", "unittest", "unittests", } CAPS = { diff --git a/tools/03_code_analysis/test_agent_governance_check.py b/tools/03_code_analysis/test_agent_governance_check.py index 092da9faf9c..36bb58d7e4b 100644 --- a/tools/03_code_analysis/test_agent_governance_check.py +++ b/tools/03_code_analysis/test_agent_governance_check.py @@ -643,6 +643,47 @@ def test_rejects_staged_with_base_head(self): self.assertNotEqual(result.returncode, 0, result.stdout + result.stderr) self.assertIn("--staged cannot be combined with --base/--head", result.stderr) + def test_blocks_new_pragma_once_in_header(self): + self.write("source/source_base/new_header.h", "#pragma once\n\nint foo();\n") + head = self.commit_change() + + result = self.run_checker("--base", self.base, "--head", head) + + self.assert_blocked_by(result, "No #pragma once") + + def test_blocks_new_pragma_once_in_cuh(self): + self.write("source/source_base/kernels/new_header.cuh", "#pragma once\n\nint foo();\n") + head = self.commit_change() + + result = self.run_checker("--base", self.base, "--head", head) + + self.assert_blocked_by(result, "No #pragma once") + + def test_allows_ifndef_include_guard(self): + self.write( + "source/source_base/new_header.h", + "#ifndef NEW_HEADER_H\n#define NEW_HEADER_H\n\nint foo();\n\n#endif\n", + ) + head = self.commit_change() + + result = self.run_checker("--base", self.base, "--head", head) + + self.assertNotIn("No #pragma once", result.stdout) + + def test_allows_existing_pragma_once_on_unchanged_lines(self): + self.write("source/source_base/old_header.h", "#pragma once\n\nint foo();\n") + self.write("source/source_base/CMakeLists.txt", "add_library(old old_header.h)\n") + self.git("add", ".") + self.git("commit", "-m", "add header with pragma once") + base2 = self.git("rev-parse", "HEAD").stdout.strip() + # Touch only an unrelated line; the #pragma once line is not in the diff. + self.write("source/source_base/old_header.h", "#pragma once\n\nint foo();\nint bar();\n") + head = self.commit_change() + + result = self.run_checker("--base", base2, "--head", head) + + self.assertNotIn("No #pragma once", result.stdout) + if __name__ == "__main__": unittest.main()