32 #ifndef INTERNAL_UTIL_H_ 33 #define INTERNAL_UTIL_H_ 44 idx* result) noexcept {
50 for (
idx i = 0; i < numdims; ++i)
56 for (
idx i = 0; i < numdims; ++i) {
57 result[numdims - i - 1] = n % (dims[numdims - i - 1]);
58 n /= (dims[numdims - i - 1]);
64 #if (__GNUC__ && !__clang__) 65 #pragma GCC diagnostic push 66 #pragma GCC diagnostic ignored "-Warray-bounds" 67 #pragma GCC diagnostic ignored "-Wmaybe-uninitialized" 72 const idx*
const dims) noexcept {
85 part_prod[numdims - 1] = 1;
86 for (
idx i = 1; i < numdims; ++i) {
87 part_prod[numdims - i - 1] = part_prod[numdims - i] * dims[numdims - i];
88 result += midx[numdims - i - 1] * part_prod[numdims - i - 1];
91 return result + midx[numdims - 1];
93 #if (__GNUC__ && !__clang__) 94 #pragma GCC diagnostic pop 98 template <
typename Derived>
100 return A.rows() == A.cols();
104 template <
typename Derived>
106 return A.rows() == 1 || A.cols() == 1;
110 template <
typename Derived>
112 return A.rows() == 1;
116 template <
typename Derived>
118 return A.cols() == 1;
122 template <
typename T>
124 return x.size() != 0;
128 template <
typename T1,
typename T2>
130 return lhs.size() == rhs.size();
135 if (dims.size() == 0)
138 return std::find_if(std::begin(dims), std::end(dims),
139 [dims](
idx i) ->
bool {
144 }) == std::end(dims);
149 template <
typename Derived>
151 const Eigen::MatrixBase<Derived>& A) {
154 assert(dims.size() > 0);
155 assert(A.rows() == A.cols());
157 idx proddim = std::accumulate(std::begin(dims), std::end(dims),
158 static_cast<idx>(1), std::multiplies<idx>());
160 return proddim ==
static_cast<idx>(A.rows());
164 template <
typename Derived>
166 const Eigen::MatrixBase<Derived>& A) {
169 assert(dims.size() > 0);
170 assert(A.rows() > 0);
171 assert(A.cols() == 1);
173 idx proddim = std::accumulate(std::begin(dims), std::end(dims),
174 static_cast<idx>(1), std::multiplies<idx>());
176 return proddim ==
static_cast<idx>(A.rows());
180 template <
typename Derived>
182 const Eigen::MatrixBase<Derived>& A) {
185 assert(dims.size() > 0);
186 assert(A.cols() > 0);
187 assert(A.rows() == 1);
189 idx proddim = std::accumulate(std::begin(dims), std::end(dims),
190 static_cast<idx>(1), std::multiplies<idx>());
193 return proddim ==
static_cast<idx>(A.cols());
200 assert(dims.size() > 0);
211 const std::vector<idx>& dims) {
215 if (subsys.size() > dims.size())
219 std::vector<idx> subsyssort = subsys;
220 std::sort(std::begin(subsyssort), std::end(subsyssort));
223 if (std::unique(std::begin(subsyssort), std::end(subsyssort)) !=
224 std::end(subsyssort))
228 return std::find_if(std::begin(subsyssort), std::end(subsyssort),
229 [dims](
idx i) ->
bool {
230 return i > dims.size() - 1;
231 }) == std::end(subsyssort);
235 template <
typename Derived>
237 return A.rows() == 2 && A.cols() == 2;
241 template <
typename Derived>
243 return A.rows() == 2 && A.cols() == 1;
247 template <
typename Derived>
249 return A.rows() == 1 && A.cols() == 2;
253 template <
typename Derived>
255 return (A.rows() == 1 && A.cols() == 2) || (A.rows() == 2 && A.cols() == 1);
260 if (perm.size() == 0)
263 std::vector<idx> ordered(perm.size());
264 std::iota(std::begin(ordered), std::end(ordered), 0);
266 return std::is_permutation(std::begin(ordered), std::end(ordered),
272 template <
typename Derived1,
typename Derived2>
274 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_ 306 for (
idx j = 0; j < Acols; ++j)
307 for (
idx i = 0; i < Arows; ++i)
308 result.block(i * Brows, j * Bcols, Brows, Bcols) = rA(i, j) * rB;
315 template <
typename Derived1,
typename Derived2>
318 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());
346 result.block(0, 0, Arows, Acols) = rA;
347 result.block(Arows, Acols, Brows, Bcols) = rB;
353 template <
typename T>
358 template <
typename T,
typename First,
typename... Args>
360 v.emplace_back(std::forward<First>(first));
372 return static_cast<idx>(std::llround(std::log2(sz) / std::log2(d)));
385 return static_cast<idx>(std::llround(std::sqrt(sz)));
387 return static_cast<idx>(std::llround(std::pow(sz, 1. / N)));
392 template <
typename T>
396 std::ostringstream ostr;
399 std::vector<std::string> vstr;
402 for (
idx i = 0; i < static_cast<idx>(A.rows()); ++i) {
403 for (
idx j = 0; j < static_cast<idx>(A.cols()); ++j) {
406 ostr.str(std::string{});
409 double re =
static_cast<cplx>(A(i, j)).real();
410 double im =
static_cast<cplx>(A(i, j)).imag();
412 if (std::abs(re) <
chop && std::abs(im) <
chop) {
416 vstr.push_back(ostr.str());
417 }
else if (std::abs(re) <
chop) {
419 vstr.push_back(ostr.str() +
"i");
420 }
else if (std::abs(im) <
chop) {
422 vstr.push_back(ostr.str() +
" ");
427 strA += (im > 0 ?
" + " :
" - ");
429 ostr.str(std::string());
430 ostr << std::abs(im);
433 vstr.push_back(strA);
439 std::vector<idx> maxlengthcols(A.cols(), 0);
441 for (
idx i = 0; i < static_cast<idx>(A.rows()); ++i)
442 for (
idx j = 0; j < static_cast<idx>(A.cols()); ++j)
443 if (vstr[i * A.cols() + j].size() > maxlengthcols[j])
444 maxlengthcols[j] = vstr[i * A.cols() + j].size();
447 for (
idx i = 0; i < static_cast<idx>(A.rows()); ++i) {
448 os << std::setw(static_cast<int>(maxlengthcols[0])) << std::right
449 << vstr[i * A.cols()];
451 for (
idx j = 1; j < static_cast<idx>(A.cols()); ++j)
452 os << std::setw(static_cast<int>(maxlengthcols[j] + 2))
453 << std::right << vstr[i * A.cols() + j];
455 if (i < static_cast<idx>(A.rows()) - 1)
bool check_nonzero_size(const T &x) noexcept
Definition: util.h:123
constexpr double chop
Used in qpp::disp() for setting to zero numbers that have their absolute value smaller than qpp::chop...
Definition: constants.h:58
constexpr idx maxn
Maximum number of allowed qubits/qudits (subsystems)
Definition: constants.h:73
bool check_subsys_match_dims(const std::vector< idx > &subsys, const std::vector< idx > &dims)
Definition: util.h:210
dyn_mat< typename Derived1::Scalar > kron2(const Eigen::MatrixBase< Derived1 > &A, const Eigen::MatrixBase< Derived2 > &B)
Definition: util.h:273
bool check_cvector(const Eigen::MatrixBase< Derived > &A)
Definition: util.h:117
std::ostream & display_impl_(const T &A, std::ostream &os, double chop=qpp::chop) const
Definition: util.h:394
Eigen::Matrix< Scalar, Eigen::Dynamic, Eigen::Dynamic > dyn_mat
Dynamic Eigen matrix over the field specified by Scalar.
Definition: types.h:81
bool check_dims_match_mat(const std::vector< idx > &dims, const Eigen::MatrixBase< Derived > &A)
Definition: util.h:150
idx multiidx2n(const idx *const midx, idx numdims, const idx *const dims) noexcept
Definition: util.h:71
void n2multiidx(idx n, idx numdims, const idx *const dims, idx *result) noexcept
Definition: util.h:43
Quantum++ main namespace.
Definition: codes.h:35
bool check_perm(const std::vector< idx > &perm)
Definition: util.h:259
idx get_num_subsys(idx sz, idx d)
Definition: util.h:366
bool check_rvector(const Eigen::MatrixBase< Derived > &A)
Definition: util.h:111
bool check_eq_dims(const std::vector< idx > &dims, idx dim) noexcept
Definition: util.h:197
bool check_vector(const Eigen::MatrixBase< Derived > &A)
Definition: util.h:105
bool check_dims_match_cvect(const std::vector< idx > &dims, const Eigen::MatrixBase< Derived > &A)
Definition: util.h:165
bool check_dims_match_rvect(const std::vector< idx > &dims, const Eigen::MatrixBase< Derived > &A)
Definition: util.h:181
bool check_matching_sizes(const T1 &lhs, const T2 &rhs) noexcept
Definition: util.h:129
std::complex< double > cplx
Complex number in double precision.
Definition: types.h:49
void variadic_vector_emplace(std::vector< T > &)
Definition: util.h:355
Type mismatch exception.
Definition: exception.h:528
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:317
bool check_qubit_rvector(const Eigen::MatrixBase< Derived > &A) noexcept
Definition: util.h:248
bool check_dims(const std::vector< idx > &dims)
Definition: util.h:134
bool check_qubit_vector(const Eigen::MatrixBase< Derived > &A) noexcept
Definition: util.h:254
std::size_t idx
Non-negative integer index.
Definition: types.h:39
bool check_qubit_cvector(const Eigen::MatrixBase< Derived > &A) noexcept
Definition: util.h:242
Object has zero size exception.
Definition: exception.h:132
bool check_qubit_matrix(const Eigen::MatrixBase< Derived > &A) noexcept
Definition: util.h:236