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
|
#pragma once
#include "config.h"
#include "pcmengin.h"
#include "oma.h"
#include "aea.h"
#include "atrac/atrac3.h"
#include "atrac/atrac3_qmf.h"
#include "transient_detector.h"
#include "atrac/atrac3_bitstream.h"
#include "atrac/atrac_scale.h"
#include "mdct/mdct.h"
#include "gain_processor.h"
#include <functional>
#include <array>
namespace NAtracDEnc {
class TAtrac3MDCT : public virtual NAtrac3::TAtrac3Data {
NMDCT::TMDCT<512> Mdct512;
NMDCT::TMIDCT<512> Midct512;
public:
typedef TGainProcessor<TAtrac3Data> TAtrac3GainProcessor;
TAtrac3GainProcessor GainProcessor;
TAtrac3MDCT()
: Mdct512(2)
{}
public:
using TGainModulator = TAtrac3GainProcessor::TGainModulator;
using TGainDemodulator = TAtrac3GainProcessor::TGainDemodulator;
typedef std::array<TGainDemodulator, 4> TGainDemodulatorArray;
typedef std::array<TGainModulator, 4> TGainModulatorArray;
void Mdct(TFloat specs[1024],
TFloat* bands[4],
TFloat maxLevels[4],
TGainModulatorArray gainModulators = TGainModulatorArray());
void Midct(TFloat specs[1024],
TFloat* bands[4],
TGainDemodulatorArray gainDemodulators = TGainDemodulatorArray());
protected:
TAtrac3MDCT::TGainModulatorArray MakeGainModulatorArray(const TAtrac3Data::SubbandInfo& si);
};
//returns threshhold
typedef std::function<float(const TFloat* p, uint16_t len)> TTonalDetector;
class TAtrac3Processor : public IProcessor<TFloat>, public TAtrac3MDCT, public virtual NAtrac3::TAtrac3Data {
TCompressedIOPtr Oma;
const NAtrac3::TAtrac3EncoderSettings Params;
TFloat PcmBuffer[2][4][256 + 256]; //2 channel, 4 band, 256 sample + 256 for overlap buffer
TFloat LastLevels[2][4]; //2 channel, 4 band - level of last subblock, used to create curve
TFloat LastHPLevels[2][4]; //2 channel, 4 band - level of last HP filtered subblock, used for transient detection
TFloat PrevPeak[2][4]; //2 channel, 4 band - peak level (after windowing), used to check overflow during scalling
Atrac3SplitFilterBank<TFloat> SplitFilterBank[2];
TScaler<TAtrac3Data> Scaler;
std::vector<TTransientDetector> TransientDetectors;
typedef std::array<uint8_t, NumSpecs> TonalComponentMask;
#ifdef ATRAC_UT_PUBLIC
public:
#endif
uint32_t CheckLevelOverflow(TFloat max, uint32_t levelIdx);
std::vector<SubbandInfo::TGainPoint> FilterCurve(const std::vector<SubbandInfo::TGainPoint>& curve,
const int threshold);
TFloat LimitRel(TFloat x);
std::vector<TFloat> CalcBaseLevel(TFloat prev, const std::vector<TFloat>& gain);
TAtrac3Data::SubbandInfo CreateSubbandInfo(TFloat* in[4], uint32_t channel, TTransientDetector* transientDetector);
TonalComponentMask AnalyzeTonalComponent(TFloat* specs);
TTonalComponents ExtractTonalComponents(TFloat* specs, TTonalDetector fn);
std::vector<NAtrac3::TTonalComponent> MapTonalComponents(const TTonalComponents& tonalComponents);
public:
TAtrac3Processor(TCompressedIOPtr&& oma, NAtrac3::TAtrac3EncoderSettings&& encoderSettings);
~TAtrac3Processor();
TPCMEngine<TFloat>::TProcessLambda GetDecodeLambda() override;
TPCMEngine<TFloat>::TProcessLambda GetEncodeLambda() override;
};
}
|
