92 lines
2.5 KiB
C++
92 lines
2.5 KiB
C++
// Copyright (C) 2022 Then Try This
|
|
//
|
|
// This program is free software; you can redistribute it and/or modify
|
|
// it under the terms of the GNU General Public License as published by
|
|
// the Free Software Foundation; either version 2 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU General Public License
|
|
// along with this program; if not, write to the Free Software
|
|
// Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
|
|
|
|
#include <string>
|
|
#include <vector>
|
|
#include "spiralcore/sample.h"
|
|
#include "spiralcore/types.h"
|
|
#include "fft.h"
|
|
#include "mfcc.h"
|
|
#include "search_params.h"
|
|
#include "window.h"
|
|
#include "spiralcore/stream.h"
|
|
|
|
#ifndef BLOCK
|
|
#define BLOCK
|
|
|
|
namespace spiralcore {
|
|
|
|
class block {
|
|
public:
|
|
// runs analysis on pcm
|
|
block(u64 id, const std::string &filename, const sample &pcm, u32 rate, const window &w, bool ditchpcm=false);
|
|
block() {}
|
|
|
|
// returns distance based on ratio of fft-mfcc values
|
|
double compare(const block &other, const search_params ¶ms) const;
|
|
|
|
static void init_fft(u32 block_size);
|
|
static bool unit_test();
|
|
|
|
const sample &get_pcm() const { return m_pcm; }
|
|
const sample &get_n_pcm() const { return m_n_pcm; }
|
|
|
|
std::vector<u32> &get_synapse() { return m_synapse; }
|
|
const std::vector<u32> &get_synapse_const() const { return m_synapse; }
|
|
float &get_usage() { return m_usage; }
|
|
float get_freq() const { return m_dominant_freq; }
|
|
float get_n_freq() const { return m_n_dominant_freq; }
|
|
|
|
private:
|
|
|
|
void process(const sample &pcm, sample &fft, sample &mfcc, float &freq);
|
|
|
|
double _compare(const sample &fft_a, const sample &mfcc_a,
|
|
const sample &fft_b, const sample &mfcc_b,
|
|
const search_params ¶ms) const;
|
|
|
|
u64 m_id;
|
|
|
|
sample m_pcm;
|
|
sample m_fft;
|
|
sample m_mfcc;
|
|
|
|
sample m_n_pcm;
|
|
sample m_n_fft;
|
|
sample m_n_mfcc;
|
|
|
|
u32 m_block_size;
|
|
u32 m_rate;
|
|
std::string m_orig_filename;
|
|
static FFT *m_fftw;
|
|
static Aquila::Mfcc *m_mfcc_proc;
|
|
|
|
std::vector<u32> m_synapse;
|
|
float m_usage;
|
|
|
|
float m_dominant_freq;
|
|
float m_n_dominant_freq;
|
|
|
|
friend ios &operator||(ios &s, block &b);
|
|
|
|
};
|
|
|
|
ios &operator||(ios &s, block &b);
|
|
|
|
}
|
|
|
|
#endif
|