Upload Audio_Effects_SDK/samples/utils/wave_reader/wave.hpp with huggingface_hub

Audio_Effects_SDK/samples/utils/wave_reader/wave.hpp

@@ -0,0 +1,142 @@
+/*
+* Copyright (c) 2019, NVIDIA Corporation.  All rights reserved.
+*
+* NVIDIA Corporation and its licensors retain all intellectual property
+* and proprietary rights in and to this software, related documentation
+* and any modifications thereto.  Any use, reproduction, disclosure or
+* distribution of this software and related documentation without an express
+* license agreement from NVIDIA Corporation is strictly prohibited.
+*/
+
+#pragma once
+
+#include <string>
+
+#ifndef WAVE_FORMAT_PCM
+#define WAVE_FORMAT_PCM        0x0001
+#endif
+#define WAVE_FORMAT_IEEE_FLOAT 0x0003
+#define WAVE_FORMAT_ALAW       0x0006
+#define WAVE_FORMAT_MULAW      0x0007
+#define WAVE_FORMAT_EXTENSIBLE 0xFFFE
+
+#define WAVE_WAVE   0x45564157  // 'EVAW' (little endian WAVE)
+#define WAVE_RIFF   0x46464952  // 'FFIR' (little endian RIFF)
+#define WAVE_FORMAT 0x20746D66  // ' tmf' (little endian fmt )
+#define WAVE_DATA   0x61746164  // 'atad' (little endian data)
+
+#define DEBUG 1
+
+#if DEBUG
+#define PRINTCONTROL(x) x;
+#define WAVE_ZERO_ON_ALLOCATE 1
+#else
+#define PRINTCONTROL(x)
+#define WAVE_ZERO_ON_ALLOCATE 0
+#endif
+
+#define MAX_CHANNELS 64
+
+typedef struct {
+  int ckId;
+  int cksize;
+  int waveId;
+  int waveChunks;
+} WaveHeader;
+
+typedef struct {
+  uint16_t   formatTag;         /* format type */
+  uint16_t   nChannels;         /* number of channels (i.e. mono, stereo...) */
+  unsigned int     nSamplesPerSec;    /* sample rate */
+  unsigned int     nAvgBytesPerSec;   /* for buffer estimation */
+  uint16_t   nBlockAlign;       /* block size of data */
+} waveFormat_basic_nopcm;
+
+typedef struct {
+  uint16_t   formatTag;         /* format type */
+  uint16_t   nChannels;         /* number of channels (i.e. mono, stereo...) */
+  unsigned int     nSamplesPerSec;    /* sample rate */
+  unsigned int     nAvgBytesPerSec;   /* for buffer estimation */
+  uint16_t   nBlockAlign;       /* block size of data */
+  uint16_t   wBitsPerSample;    /* Number of bits per sample of mono data */
+} waveFormat_basic;
+
+typedef struct {
+  uint16_t   wFormatTag;         /* format type */
+  uint16_t   nChannels;         /* number of channels (i.e. mono, stereo...) */
+  unsigned int     nSamplesPerSec;    /* sample rate */
+  unsigned int     nAvgBytesPerSec;   /* for buffer estimation */
+  uint16_t   nBlockAlign;       /* block size of data */
+  uint16_t   wBitsPerSample;    /* Number of bits per sample of mono data */
+  uint16_t   cbSize;      /* the count in bytes of the size of */
+  /* extra information (after cbSize) */
+} waveFormat_ext;
+
+class Wave {
+ public:
+  Wave();
+  explicit Wave(std::string filename);
+  Wave(const Wave & argB);
+  ~Wave();
+
+  /* Tools for creating waves in software */
+  Wave(unsigned int nSamples, unsigned int nChannels, unsigned int sampleRate, unsigned int bitDepth);
+
+  // Special Constructor used for construction by extractChannel
+  Wave(unsigned int nSamples, unsigned int sampleRate, unsigned int bitDepth, float * singleChannel);
+
+  /* Tools for allowing waves to interact */
+  Wave & operator -=(const Wave & argB);
+  Wave & operator *=(float scale);
+  void operator >> (int shiftSamples);
+  void operator >> (float shiftSeconds);
+  void operator <<(int shiftSamples);
+  void operator <<(float shiftSeconds);
+
+  float& operator[](int sampleIndex);
+  float* getDataPtr(int channelIndex, int sampleIndex);
+
+  Wave extractChannel(int channelIndex);
+  void setChannel(int channelIndex, Wave* argB);
+
+  unsigned int getNumSamples();
+  unsigned int getNumChannels();
+  unsigned int getBitDepth();
+  unsigned int getSampleRate();
+
+  void         writeFile(std::string filename);
+  int          maxInt(int channelId);
+  int          minInt(int channelId);
+  float        maxFloat(int channelId);
+  float        minFloat(int channelId);
+  void         append(int numChannels, int numAppendedSamples, float ** buffers);
+
+  void         normalize();
+
+ private:
+  unsigned int m_numChannels;
+  unsigned int m_sampleRate;
+  unsigned int m_bitDepth;
+  unsigned int m_numSamples;
+  unsigned int m_bufferAllocation;
+
+  int  ** m_intData;
+  float ** m_floatData;
+
+  bool    m_PCMIntValid;
+  bool    m_PCMFloatValid;
+
+  bool allocateInt();
+  bool allocateFloat();
+  void reallocate();
+
+  void freeInt();
+  void freeFloat();
+
+  void fillFloatFromInt();
+  void fillIntFromFloat();
+  void validFloat();
+
+  void readFromFile(std::string filename);
+};
+