samplebrain/brain/src/block.h

// 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 &params) const;

    static void init_fft(u32 block_size, u32 rate);
    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 &params) 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