Method tdfLinspace

src/python/handler/py_GDMR.cpp:197–268

}

py::UniqueObj GDMRModelObject::tdfLinspace(PyObject* numericMetadataStart, PyObject* numericMetadataEnd, PyObject* numericMetadataNum, const std::string& metadata, PyObject* multiMetadata, bool endpoint, bool normalize) const
{
	auto* inst = getInst<tomoto::IGDMRModel>();

	vector<tomoto::Float> start, stop;
	if(!py::toCpp(numericMetadataStart, start))
	{
		throw py::ValueError{
			"`metadata_start` must be an iterable of float (but given " + py::repr(numericMetadataStart) + ")"
		};
	}
	if (!py::toCpp(numericMetadataEnd, stop))
	{
		throw py::ValueError{
			"`metadata_stop` must be an iterable of float (but given " + py::repr(numericMetadataEnd) + ")"
		};
	}

	vector<std::ptrdiff_t> num;
	if (!py::toCpp(numericMetadataNum, num))
	{
		throw py::ValueError{
			"`num` must be an iterable of int (but given " + py::repr(numericMetadataNum) + ")"
		};
	}

	if (start.size() != inst->getFs().size()) throw py::ValueError{ "`len(metadata_start)` must be equal to `len(degree).`" };
	if (stop.size() != inst->getFs().size()) throw py::ValueError{ "`len(metadata_stop)` must be equal to `len(degree).`" };
	if (num.size() != inst->getFs().size()) throw py::ValueError{ "`len(num)` must be equal to `len(degree).`" };

	std::ptrdiff_t tot = 1;
	for (auto& v : num)
	{
		if (v <= 0) v = 1;
		tot *= v;
	}

	Eigen::MatrixXf mds{ (Eigen::Index)num.size(), (Eigen::Index)tot };
	vector<std::ptrdiff_t> idcs(num.size());
	for (size_t i = 0; i < tot; ++i)
	{
		for (size_t j = 0; j < num.size(); ++j)
		{
			mds(j, i) = start[j] + (stop[j] - start[j]) * idcs[j] / (endpoint ? max(num[j] - 1, (std::ptrdiff_t)1) : num[j]);
		}

		idcs.back()++;
		for (int j = idcs.size() - 1; j >= 0; --j)
		{
			if (idcs[j] >= num[j])
			{
				idcs[j] = 0;
				if (j) idcs[j - 1]++;
			}
			else break;
		}
	}