AudioWaveformer.h

Go to the documentation of this file.

 
// Copyright (c) 2008-2022 OpenShot Studios, LLC
//
// SPDX-License-Identifier: LGPL-3.0-or-later
 
#ifndef OPENSHOT_WAVEFORMER_H
#define OPENSHOT_WAVEFORMER_H
 
#include "ReaderBase.h"
#include "Frame.h"
#include <vector>
 
 
namespace openshot {
 
    struct AudioWaveformData
    {
        std::vector<float> max_samples;
        std::vector<float> rms_samples;
 
        void resize(int total_samples) {
            max_samples.resize(total_samples);
            rms_samples.resize(total_samples);
        }
 
        void zero(int total_samples) {
            std::fill(max_samples.begin(), max_samples.end(), 0);
            std::fill(rms_samples.begin(), rms_samples.end(), 0);
        }
 
        void scale(int total_samples, float factor) {
            for (auto s = 0; s < total_samples; s++) {
                max_samples[s] *= factor;
                rms_samples[s] *= factor;
            }
        }
 
        void clear() {
            max_samples.clear();
            max_samples.shrink_to_fit();
            rms_samples.clear();
            rms_samples.shrink_to_fit();
        }
 
        std::vector<std::vector<float>> vectors() {
            std::vector<std::vector<float>> output;
            output.push_back(max_samples);
            output.push_back(rms_samples);
            return output;
        }
    };
 
    class AudioWaveformer {
    private:
        ReaderBase* reader;
 
    public:
        AudioWaveformer(ReaderBase* reader);
 
        AudioWaveformData ExtractSamples(int channel, int num_per_second, bool normalize);
 
        ~AudioWaveformer();
    };
 
}
 
#endif