Method convert

Lib/Model/Notes.cpp:14–234

}

std::vector<Notes::Event> Notes::convert(const std::vector<std::vector<float>>& inNotesPG,
                                         const std::vector<std::vector<float>>& inOnsetsPG,
                                         const std::vector<std::vector<float>>& inContoursPG,
                                         const ConvertParams& inParams,
                                         bool inNewAudio)
{
    std::vector<Event> events;
    events.reserve(1024);

    const auto n_frames = static_cast<int>(inNotesPG.size());
    if (n_frames == 0) {
        return events;
    }

    const auto n_notes = static_cast<int>(inNotesPG[0].size());
    assert(n_frames == inOnsetsPG.size());
    assert(n_frames == inContoursPG.size());
    assert(n_notes == inOnsetsPG[0].size());
    assert(n_notes == NUM_FREQ_OUT);

    std::vector<std::vector<float>> inferred_onsets;
    auto onsets_ptr = &inOnsetsPG;
    if (inParams.inferOnsets) {
        inferred_onsets = _inferredOnsets<float>(inOnsetsPG, inNotesPG);
        onsets_ptr = &inferred_onsets;
    }
    auto& onsets = *onsets_ptr;

    if (inParams.melodiaTrick) {
        if (inNewAudio) {
            // Copy
            mRemainingEnergy = inNotesPG;

            // Fill mRemainingEnergyIndex
            mRemainingEnergyIndex.clear();
            mRemainingEnergyIndex.reserve(static_cast<size_t>(n_frames) * static_cast<size_t>(NUM_FREQ_OUT));

            for (int frame_idx = 0; frame_idx < n_frames; frame_idx++) {
                for (int freq_idx = 0; freq_idx < NUM_FREQ_OUT; freq_idx++) {
                    mRemainingEnergyIndex.push_back(
                        {&mRemainingEnergy[static_cast<size_t>(frame_idx)][static_cast<size_t>(freq_idx)],
                         frame_idx,
                         freq_idx});
                }
            }

            mRemainingEnergyIndex.shrink_to_fit();
        } else {
            // Copy without changing the location of the original data
            assert(mRemainingEnergy.size() == n_frames);
            for (size_t f = 0; f < n_frames; f++) {
                assert(inNotesPG[f].size() == NUM_FREQ_OUT);
                assert(mRemainingEnergy[f].size() == NUM_FREQ_OUT);

                std::copy(inNotesPG[f].begin(), inNotesPG[f].end(), mRemainingEnergy[f].begin());
            }
        }
    }