en/master/_unit_tests___d_m_r_g_block___set_disk_storage_8cpp_source.html

 static char help[] =
     "Test code for DMRGBlock module on saving and retrieving blocks from disk \n";

 #include <slepceps.h>
 #include "DMRGBlock.hpp"
 #include "linalg_tools.hpp"
 #include <iostream>

 PETSC_EXTERN PetscErrorCode SetRow(const Mat& A, const PetscInt& row, const std::vector<PetscInt>& idxn);
 PETSC_EXTERN PetscErrorCode Makedir(const std::string& dir_name);
 PETSC_EXTERN PetscErrorCode SetSz0(const Mat& Sz);
 PETSC_EXTERN PetscErrorCode SetSp0(const Mat& Sp);
 PETSC_EXTERN PetscErrorCode SetSz1(const Mat& Sz);
 PETSC_EXTERN PetscErrorCode SetSp1(const Mat& Sp);
 PETSC_EXTERN PetscErrorCode Makedir(const std::string& dir_name);

 PetscErrorCode Test()
 {
     PetscErrorCode ierr;
     PetscMPIInt     nprocs, rank;
     ierr = MPI_Comm_size(PETSC_COMM_WORLD, &nprocs); CHKERRQ(ierr);
     ierr = MPI_Comm_rank(PETSC_COMM_WORLD, &rank); CHKERRQ(ierr);

     Block::SpinBase blk;
     ierr = blk.Initialize(PETSC_COMM_WORLD, 2, {1.5,0.5,-0.5,-1.5}, {2,3,2,1});CHKERRQ(ierr);
     ierr = blk.CheckSectors(); CHKERRQ(ierr);
     ierr = Makedir("trash_block_test_save01"); CHKERRQ(ierr);
     ierr = Makedir("trash_block_test_save02"); CHKERRQ(ierr);
     ierr = Makedir("trash_block_test_save03"); CHKERRQ(ierr);

     ierr = blk.SetDiskStorage("trash_block_test_save01","trash_block_test_save01"); CHKERRQ(ierr);

     /*  Set the entries of the operators */
     ierr = SetSz0(blk.Sz(0)); CHKERRQ(ierr);
     ierr = SetSp0(blk.Sp(0)); CHKERRQ(ierr);
     ierr = SetSz1(blk.Sz(1)); CHKERRQ(ierr);
     ierr = SetSp1(blk.Sp(1)); CHKERRQ(ierr);
     ierr = blk.AssembleOperators(); CHKERRQ(ierr);

     ierr = PetscPrintf(PETSC_COMM_WORLD,"Writing data to:   %s\n", blk.SaveDir().c_str()); CHKERRQ(ierr);
     ierr = blk.EnsureSaved(); CHKERRQ(ierr);

     ierr = PetscPrintf(PETSC_COMM_WORLD,"\nSetting read_dir as: \"%s\" and write_dir as: \"%s\"\n",
         "trash_block_test_save01","trash_block_test_save02"); CHKERRQ(ierr);
     ierr = blk.SetDiskStorage("trash_block_test_save01","trash_block_test_save02"); CHKERRQ(ierr);
     ierr = PetscPrintf(PETSC_COMM_WORLD,"Reading data from: %s\n", blk.SaveDir().c_str()); CHKERRQ(ierr);
     ierr = blk.EnsureRetrieved(); CHKERRQ(ierr);
     ierr = PetscPrintf(PETSC_COMM_WORLD,"Writing data to:   %s\n", blk.SaveDir().c_str()); CHKERRQ(ierr);
     ierr = blk.EnsureSaved(); CHKERRQ(ierr);
     ierr = PetscPrintf(PETSC_COMM_WORLD,"Reading data from: %s\n", blk.SaveDir().c_str()); CHKERRQ(ierr);
     ierr = blk.EnsureRetrieved(); CHKERRQ(ierr);
     ierr = PetscPrintf(PETSC_COMM_WORLD,"Writing data to:   %s\n", blk.SaveDir().c_str()); CHKERRQ(ierr);
     ierr = blk.EnsureSaved(); CHKERRQ(ierr);

     ierr = PetscPrintf(PETSC_COMM_WORLD,"\nSetting read_dir as: \"%s\" and write_dir as: \"%s\"\n",
         "trash_block_test_save02","trash_block_test_save03"); CHKERRQ(ierr);
     ierr = blk.SetDiskStorage("trash_block_test_save02","trash_block_test_save03"); CHKERRQ(ierr);
     ierr = PetscPrintf(PETSC_COMM_WORLD,"Reading data from: %s\n", blk.SaveDir().c_str()); CHKERRQ(ierr);
     ierr = blk.EnsureRetrieved(); CHKERRQ(ierr);
     ierr = PetscPrintf(PETSC_COMM_WORLD,"Writing data to:   %s\n", blk.SaveDir().c_str()); CHKERRQ(ierr);
     ierr = blk.EnsureSaved(); CHKERRQ(ierr);
     ierr = PetscPrintf(PETSC_COMM_WORLD,"Reading data from: %s\n", blk.SaveDir().c_str()); CHKERRQ(ierr);
     ierr = blk.EnsureRetrieved(); CHKERRQ(ierr);
     ierr = PetscPrintf(PETSC_COMM_WORLD,"Writing data to:   %s\n", blk.SaveDir().c_str()); CHKERRQ(ierr);
     ierr = blk.EnsureSaved(); CHKERRQ(ierr);

     ierr = blk.Destroy(); CHKERRQ(ierr);
     return(0);
 }

 int main(int argc, char **argv)
 {
     PetscErrorCode  ierr = 0;
     PetscMPIInt     nprocs, rank;
     ierr = SlepcInitialize(&argc, &argv, (char*)0, help); CHKERRQ(ierr);
     ierr = MPI_Comm_size(PETSC_COMM_WORLD, &nprocs); CHKERRQ(ierr);
     ierr = MPI_Comm_rank(PETSC_COMM_WORLD, &rank); CHKERRQ(ierr);

     ierr = Test(); CHKERRQ(ierr);

     ierr = SlepcFinalize(); CHKERRQ(ierr);
     return ierr;
 }
Block::SpinBase::EnsureRetrieved
PetscErrorCode EnsureRetrieved()
Ensures that the block matrices have been retrieved if the block is initialized, otherwise does nothi...
Definition: DMRGBlock.cpp:1098

Makedir
PETSC_EXTERN PetscErrorCode Makedir(const std::string &dir_name)
Creates a directory named dir_name.
Definition: MiscTools.cpp:306

Block::SpinBase::SaveDir
std::string SaveDir() const
Returns the value of save_dir where the block data will be read from/written.
Definition: DMRGBlock.hpp:287

Block::SpinBase
Base class for the implementation of a block of spin sites.
Definition: DMRGBlock.hpp:79

Block::SpinBase::AssembleOperators
PetscErrorCode AssembleOperators()
Ensures that all operators are assembled.
Definition: DMRGBlock.cpp:825

Block::SpinBase::EnsureSaved
PetscErrorCode EnsureSaved()
Ensures that the block matrices have been saved if the block is initialized, otherwise does nothing...
Definition: DMRGBlock.cpp:1090

Block::SpinBase::SetDiskStorage
PetscErrorCode SetDiskStorage(const std::string &read_dir_in, const std::string &write_dir_in)
Tells where to read from and save the operators and data about the block.
Definition: DMRGBlock.cpp:857

Block::SpinBase::CheckSectors
PetscErrorCode CheckSectors() const
Checks whether sector indexing was done properly.
Definition: DMRGBlock.cpp:429

Block::SpinBase::Sz
Mat Sz(const PetscInt &Isite) const
Returns the matrix pointer to the  operator at site Isite.
Definition: DMRGBlock.hpp:353

Block::SpinBase::Destroy
PetscErrorCode Destroy()
Destroys all operator matrices and frees memory.
Definition: DMRGBlock.cpp:655

Block::SpinBase::Initialize
PetscErrorCode Initialize(const MPI_Comm &comm_in)
Initializes block object&#39;s MPI attributes.
Definition: DMRGBlock.cpp:31

Block::SpinBase::Sp
Mat Sp(const PetscInt &Isite) const
Returns the matrix pointer to the  operator at site Isite.
Definition: DMRGBlock.hpp:359