aboutsummaryrefslogtreecommitdiffstats
path: root/src/atracdenc.h
blob: 30dbd20b8ca65ef5ed74490de8a207beba5b4ab5 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
#pragma once
#include "pcmengin.h"
#include "aea.h"
#include "atrac_encode_settings.h"
#include "transient_detector.h"
#include "atrac/atrac1.h"
#include "atrac/atrac1_qmf.h"
#include "atrac/atrac_scale.h"
#include "mdct/mdct.h"

#include <assert.h>
#include <vector>

namespace NAtracDEnc {

enum EMode {
    E_ENCODE = 1,
    E_DECODE = 2
};

class TAtrac1MDCT : public virtual TAtrac1Data {
    NMDCT::TMDCT<512> Mdct512;
    NMDCT::TMDCT<256> Mdct256;
    NMDCT::TMDCT<64> Mdct64;
    NMDCT::TMIDCT<512> Midct512;
    NMDCT::TMIDCT<256> Midct256;
public:
    void IMdct(double specs[512], const TBlockSize& mode, double* low, double* mid, double* hi);
    void Mdct(double specs[512], double* low, double* mid, double* hi, const TBlockSize& blockSize);
    TAtrac1MDCT()
        : Mdct512(2)
        , Mdct256(1)
    {}
};

class TAtrac1Processor : public IProcessor<double>, public TAtrac1MDCT, public virtual TAtrac1Data {
    TAeaPtr Aea;
    const TAtrac1EncodeSettings Settings;

    double PcmBufLow[2][256 + 16];
    double PcmBufMid[2][256 + 16];
    double PcmBufHi[2][512 + 16];

    int32_t PcmValueMax = 32767;
    int32_t PcmValueMin = -32767;

    Atrac1SynthesisFilterBank<double> SynthesisFilterBank[2];
    Atrac1SplitFilterBank<double> SplitFilterBank[2];

    class TTransientDetectors {
        std::vector<TTransientDetector> transientDetectorLow;
        std::vector<TTransientDetector> transientDetectorMid;
        std::vector<TTransientDetector> transientDetectorHi;
    public:
        TTransientDetectors()
            : transientDetectorLow(2, TTransientDetector(16, 128))
            , transientDetectorMid(2, TTransientDetector(16, 128))
            , transientDetectorHi(2, TTransientDetector(16, 256))
        {}
        TTransientDetector& GetDetector(uint32_t channel, uint32_t band) {
            switch (band) {
                case 0:
                    return transientDetectorLow[channel];
                break;
                case 1:
                    return transientDetectorMid[channel];
                break;
                case 2:
                    return transientDetectorHi[channel];
                break;
                default:
                    assert(false);
                    return transientDetectorLow[channel];
            }
        }
    };
    TAtrac1Processor::TTransientDetectors TransientDetectors;
 
    TScaler<TAtrac1Data> Scaler;

public:
    TAtrac1Processor(TAeaPtr&& aea, TAtrac1EncodeSettings&& settings);
    TPCMEngine<double>::TProcessLambda GetDecodeLambda() override;

    TPCMEngine<double>::TProcessLambda GetEncodeLambda() override;
};
}