27 #ifndef INTERNAL_UTIL_H_
28 #define INTERNAL_UTIL_H_
32 #if (__GNUC__ && !__clang__)
33 #pragma GCC diagnostic ignored "-Warray-bounds"
34 #pragma GCC diagnostic ignored "-Wmaybe-uninitialized"
56 for (
idx i = 0; i < numdims; ++i)
62 for (
idx i = 0; i < numdims; ++i)
64 result[numdims - i - 1] = n % (dims[numdims - i - 1]);
65 n /= (dims[numdims - i - 1]);
86 part_prod[numdims - 1] = 1;
87 for (
idx i = 1; i < numdims; ++i)
89 part_prod[numdims - i - 1] =
90 part_prod[numdims - i] * dims[numdims - i];
91 result += midx[numdims - i - 1] * part_prod[numdims - i - 1];
94 return result + midx[numdims - 1];
98 template<
typename Derived>
101 return A.rows() == A.cols();
105 template<
typename Derived>
108 return A.rows() == 1 || A.cols() == 1;
112 template<
typename Derived>
115 return A.rows() == 1;
119 template<
typename Derived>
122 return A.cols() == 1;
129 return x.size() != 0;
133 template<
typename T1,
typename T2>
136 return lhs.size() == rhs.size();
142 if (dims.size() == 0)
145 return std::find_if(std::begin(dims), std::end(dims),
146 [dims](
idx i) ->
bool
148 if (i == 0)
return true;
150 }) == std::end(dims);
155 template<
typename Derived>
157 const Eigen::MatrixBase<Derived>& A)
159 idx proddim = std::accumulate(std::begin(dims), std::end(dims),
160 static_cast<idx>(1), std::multiplies<idx>());
162 return proddim ==
static_cast<idx>(A.rows());
166 template<
typename Derived>
168 const Eigen::MatrixBase<Derived>& A)
170 idx proddim = std::accumulate(std::begin(dims), std::end(dims),
171 static_cast<idx>(1), std::multiplies<idx>());
173 return proddim ==
static_cast<idx>(A.rows());
177 template<
typename Derived>
179 const Eigen::MatrixBase<Derived>& A)
181 idx proddim = std::accumulate(std::begin(dims), std::end(dims),
182 static_cast<idx>(1), std::multiplies<idx>());;
184 return proddim ==
static_cast<idx>(A.cols());
199 const std::vector<idx>& dims)
206 if (subsys.size() > dims.size())
210 std::vector<idx> subsyssort = subsys;
211 std::sort(std::begin(subsyssort), std::end(subsyssort));
214 if (std::unique(std::begin(subsyssort), std::end(subsyssort))
215 != std::end(subsyssort))
219 return std::find_if(std::begin(subsyssort), std::end(subsyssort),
220 [dims](
idx i) ->
bool
222 return i > dims.size() - 1;
223 }) == std::end(subsyssort);
227 template<
typename Derived>
230 return A.rows() == 2 && A.cols() == 2;
234 template<
typename Derived>
237 return A.rows() == 2 && A.cols() == 1;
241 template<
typename Derived>
244 return A.rows() == 1 && A.cols() == 2;
248 template<
typename Derived>
251 return (A.rows() == 1 && A.cols() == 2) ||
252 (A.rows() == 2 && A.cols() == 1);
259 if (perm.size() == 0)
262 std::vector<idx> ordered(perm.size());
263 std::iota(std::begin(ordered), std::end(ordered), 0);
265 return std::is_permutation(std::begin(ordered), std::end(ordered),
271 template<
typename Derived1,
typename Derived2>
273 const Eigen::MatrixBase<Derived2>& B)
281 if (!std::is_same<
typename Derived1::Scalar,
282 typename Derived2::Scalar>::value)
294 idx Acols =
static_cast<idx>(rA.cols());
295 idx Arows =
static_cast<idx>(rA.rows());
296 idx Bcols =
static_cast<idx>(rB.cols());
297 idx Brows =
static_cast<idx>(rB.rows());
300 result.resize(Arows * Brows, Acols * Bcols);
303 #pragma omp parallel for collapse(2)
304 #endif // WITH_OPENMP_
305 for (
idx j = 0; j < Acols; ++j)
306 for (
idx i = 0; i < Arows; ++i)
307 result.block(i * Brows, j * Bcols, Brows, Bcols) = rA(i, j) * rB;
314 template<
typename Derived1,
typename Derived2>
316 const Eigen::MatrixBase<Derived1>& A,
317 const Eigen::MatrixBase<Derived2>& B)
325 if (!std::is_same<
typename Derived1::Scalar,
326 typename Derived2::Scalar>::value)
338 idx Acols =
static_cast<idx>(rA.cols());
339 idx Arows =
static_cast<idx>(rA.rows());
340 idx Bcols =
static_cast<idx>(rB.cols());
341 idx Brows =
static_cast<idx>(rB.rows());
347 result.block(0, 0, Arows, Acols) = rA;
348 result.block(Arows, Acols, Brows, Bcols) = rB;
361 template<
typename T,
typename First,
typename ... Args>
364 v.emplace_back(std::forward<First>(first));
372 return static_cast<idx>(std::llround(std::log2(sz) / std::log2(d)));
381 return static_cast<idx>(std::llround(std::sqrt(sz)));
383 return static_cast<idx>(std::llround(std::pow(sz, 1. / N)));
395 std::ostringstream ostr;
398 std::vector<std::string> vstr;
401 for (
idx i = 0; i < static_cast<idx>(A.rows()); ++i)
404 j < static_cast<idx>(A.cols()); ++j)
408 ostr.str(std::string {});
411 double re =
static_cast<cplx>(A(i, j)).real();
412 double im =
static_cast<cplx>(A(i, j)).imag();
414 if (std::abs(re) <
chop && std::abs(im) <
chop)
419 vstr.push_back(ostr.str());
420 }
else if (std::abs(re) <
chop)
423 vstr.push_back(ostr.str() +
"i");
424 }
else if (std::abs(im) <
chop)
427 vstr.push_back(ostr.str() +
" ");
433 strA += (im > 0 ?
" + " :
" - ");
435 ostr.str(std::string());
436 ostr << std::abs(im);
439 vstr.push_back(strA);
445 std::vector<idx> maxlengthcols(A.cols(), 0);
447 for (
idx i = 0; i < static_cast<idx>(A.rows());
450 j < static_cast<idx>(A.cols()); ++j)
451 if (vstr[i * A.cols() + j].size() > maxlengthcols[j])
452 maxlengthcols[j] = vstr[i * A.cols() + j].size();
455 for (
idx i = 0; i < static_cast<idx>(A.rows()); ++i)
457 os << std::setw(static_cast<int>(maxlengthcols[0])) << std::right
458 << vstr[i * A.cols()];
461 j < static_cast<idx>(A.cols()); ++j)
462 os << std::setw(static_cast<int>(maxlengthcols[j] + 2))
463 << std::right << vstr[i * A.cols() + j];
465 if (i < static_cast<idx>(A.rows()) - 1)
bool check_nonzero_size(const T &x) noexcept
Definition: util.h:127
constexpr double chop
Used in qpp::disp() for setting to zero numbers that have their absolute value smaller than qpp::chop...
Definition: constants.h:59
constexpr idx maxn
Maximum number of allowed qubits/qudits (subsystems)
Definition: constants.h:74
bool check_subsys_match_dims(const std::vector< idx > &subsys, const std::vector< idx > &dims)
Definition: util.h:198
dyn_mat< typename Derived1::Scalar > kron2(const Eigen::MatrixBase< Derived1 > &A, const Eigen::MatrixBase< Derived2 > &B)
Definition: util.h:272
bool check_cvector(const Eigen::MatrixBase< Derived > &A)
Definition: util.h:120
Eigen::Matrix< Scalar, Eigen::Dynamic, Eigen::Dynamic > dyn_mat
Dynamic Eigen matrix over the field specified by Scalar.
Definition: types.h:78
bool check_dims_match_mat(const std::vector< idx > &dims, const Eigen::MatrixBase< Derived > &A)
Definition: util.h:156
idx multiidx2n(const idx *const midx, idx numdims, const idx *const dims) noexcept
Definition: util.h:71
std::ostream & display_impl_(const T &A, std::ostream &os, double chop=qpp::chop) const
Definition: util.h:391
void n2multiidx(idx n, idx numdims, const idx *const dims, idx *result) noexcept
Definition: util.h:48
Quantum++ main namespace.
Definition: codes.h:30
bool check_perm(const std::vector< idx > &perm)
Definition: util.h:257
idx get_num_subsys(idx sz, idx d)
Definition: util.h:370
bool check_rvector(const Eigen::MatrixBase< Derived > &A)
Definition: util.h:113
bool check_eq_dims(const std::vector< idx > &dims, idx dim) noexcept
Definition: util.h:188
bool check_vector(const Eigen::MatrixBase< Derived > &A)
Definition: util.h:106
bool check_dims_match_cvect(const std::vector< idx > &dims, const Eigen::MatrixBase< Derived > &A)
Definition: util.h:167
bool check_dims_match_rvect(const std::vector< idx > &dims, const Eigen::MatrixBase< Derived > &A)
Definition: util.h:178
bool check_matching_sizes(const T1 &lhs, const T2 &rhs) noexcept
Definition: util.h:134
std::complex< double > cplx
Complex number in double precision.
Definition: types.h:46
Generates custom exceptions, used when validating function parameters.
Definition: exception.h:39
void variadic_vector_emplace(std::vector< T > &)
Definition: util.h:356
idx get_dim_subsys(idx sz, idx N)
Definition: util.h:378
bool check_square_mat(const Eigen::MatrixBase< Derived > &A)
Definition: util.h:99
dyn_mat< typename Derived1::Scalar > dirsum2(const Eigen::MatrixBase< Derived1 > &A, const Eigen::MatrixBase< Derived2 > &B)
Definition: util.h:315
bool check_qubit_rvector(const Eigen::MatrixBase< Derived > &A) noexcept
Definition: util.h:242
bool check_dims(const std::vector< idx > &dims)
Definition: util.h:140
bool check_qubit_vector(const Eigen::MatrixBase< Derived > &A) noexcept
Definition: util.h:249
std::size_t idx
Non-negative integer index.
Definition: types.h:36
bool check_qubit_cvector(const Eigen::MatrixBase< Derived > &A) noexcept
Definition: util.h:235
bool check_qubit_matrix(const Eigen::MatrixBase< Derived > &A) noexcept
Definition: util.h:228