Function exportSimilaritySamplesCSV

src/aliceVision/depthMap/volumeIO.cpp:28–86

namespace depthMap {

void exportSimilaritySamplesCSV(const CudaHostMemoryHeap<TSim, 3>& in_volumeSim_hmh,
                                const std::vector<float>& in_depths,
                                const std::string& name,
                                const depthMapCommon::SgmParams& sgmParams,
                                const std::string& filepath,
                                const ROI& roi)
{
    const ROI downscaledRoi = downscaleROI(roi, sgmParams.scale * sgmParams.stepXY);

    const size_t spitch = in_volumeSim_hmh.getBytesPaddedUpToDim(1);
    const size_t pitch = in_volumeSim_hmh.getBytesPaddedUpToDim(0);

    const int sampleSize = 3;

    const int xOffset = std::floor(downscaledRoi.width() / (sampleSize + 1.0f));
    const int yOffset = std::floor(downscaledRoi.height() / (sampleSize + 1.0f));

    std::vector<Point2d> ptsCoords(sampleSize * sampleSize);
    std::vector<std::vector<float>> ptsDepths(sampleSize * sampleSize);

    for (int iy = 0; iy < sampleSize; ++iy)
    {
        for (int ix = 0; ix < sampleSize; ++ix)
        {
            const int ptIdx = iy * sampleSize + ix;
            const int x = (ix + 1) * xOffset;
            const int y = (iy + 1) * yOffset;

            ptsCoords.at(ptIdx) = {x, y};
            std::vector<float>& pDepths = ptsDepths.at(ptIdx);

            pDepths.reserve(in_depths.size());

            for (int iz = 0; iz < in_depths.size(); ++iz)
            {
                const float simValue = float(*get3DBufferAt_h<TSim>(in_volumeSim_hmh.getBuffer(), spitch, pitch, x, y, iz));
                pDepths.push_back(simValue);
            }
        }
    }

    std::stringstream ss;

    ss << name << "\n";

    for (int i = 0; i < ptsCoords.size(); ++i)
    {
        const Point2d& coord = ptsCoords.at(i);
        ss << "p" << (i + 1) << " (x: " << coord.x << ", y: " << coord.y << ");";
        for (const float depth : ptsDepths.at(i))
            ss << depth << ";";
        ss << "\n";
    }

    std::ofstream file;
    file.open(filepath, std::ios_base::app);
    if (file.is_open())
        file << ss.str();