TimeStretchPitchScaleReader.h

Go to the documentation of this file.

/*******************************************************************************
 * Copyright 2009-2025 Jörg Müller
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 ******************************************************************************/

#pragma once

#include "TimeStretchPitchScale.h"

#include "fx/EffectReader.h"
#include "rubberband/RubberBandStretcher.h"
#include "util/Buffer.h"

using namespace RubberBand;

AUD_NAMESPACE_BEGIN

class AUD_API TimeStretchPitchScaleReader : public EffectReader
{
private:
    Buffer m_buffer;

    std::vector<Buffer> m_deinterleaved;

    std::vector<sample_t*> m_channelData;

    int m_samplesToDrop;

    // delete copy constructor and operator=
    TimeStretchPitchScaleReader(const TimeStretchPitchScaleReader&) = delete;
    TimeStretchPitchScaleReader& operator=(const TimeStretchPitchScaleReader&) = delete;

protected:
    void reset();

    std::unique_ptr<RubberBandStretcher> m_stretcher;

    int m_position;

    bool m_finishedReader;

public:
    TimeStretchPitchScaleReader(std::shared_ptr<IReader> reader, double timeRatio, double pitchScale, StretcherQuality quality, bool preserveFormant);

    virtual void read(int& length, bool& eos, sample_t* buffer);

    virtual void seek(int position);
    virtual int getLength() const;
    virtual int getPosition() const;

    double getTimeRatio() const;

    void setTimeRatio(double timeRatio);

    double getPitchScale() const;

    void setPitchScale(double pitchScale);
};

AUD_NAMESPACE_END