#pragma comment(lib, "winmm")
#define _USE_MATH_DEFINES
#include <Windows.h>
#include <tchar.h>
#include <math.h>
#include <vector>
using namespace std;
typedef struct {
int pos;
int wl;
int noteNum;
double vol;
} Wave;
typedef vector<Wave> VWave;
// note number
#define A1 33
#define A4 69
#define A5 81
#define SAFE_FREE(p) if (p) { free(p); p = NULL; }
#define APP_NAME TEXT("WaveGraph")
// 関数プロトタイプ宣言
void Trace(LPCTSTR format, ...);
BOOL Load(LPTSTR pszFileName);
BOOL ReadWaveFile(LPTSTR pszFileName);
BOOL AnalyzeWave(void);
double LeastSquares(int pos, double freq, double waveLen);
double Volume(int pos, int wl);
LRESULT CALLBACK WndProc(HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lParam);
void OnCreate(HWND hWnd);
void OnSize(HWND hWnd, WPARAM wParam, LPARAM lParam);
void OnHScroll(HWND hWnd, WPARAM wParam);
void OnPaint(HWND hWnd);
LPCTSTR GetScale(int noteNum);
void OnChar(HWND hWnd, WPARAM wParam, LPARAM lParam);
void Play(HWND hWnd);
void OnWomOpen(void);
void OnWomDone(HWND hWnd);
// 音階
LPCTSTR scale[] = {L"C",L"C#",L"D",L"D#",L"E",L"F",L"F#",L"G",L"G#",L"A",L"A#",L"B"};
// 外部変数
SCROLLINFO siHorz;
DWORD sampleLength;
PBYTE waveformData = NULL;
WAVEFORMATEX wfx;
VWave vWave;
HWAVEOUT hwo;
WAVEHDR wh;
HCURSOR hCursor = NULL;
DWORD sampleLength2;
PBYTE waveformData2 = NULL;
//==============================================================================
int WINAPI WinMain(HINSTANCE hInstance, HINSTANCE, LPSTR, int nCmdShow)
{
// プログラム引数
LPTSTR lpCmdLine = GetCommandLine();
int argc;
LPTSTR *argv = CommandLineToArgvW(lpCmdLine, &argc);
if (2 <= argc) {
Load(argv[1]);
}
// ウィンドウクラスの登録
WNDCLASSEX wcx;
ZeroMemory(&wcx, sizeof wcx);
wcx.cbSize = sizeof wcx;
wcx.style = CS_HREDRAW | CS_VREDRAW;
wcx.lpfnWndProc = WndProc;
wcx.hInstance = hInstance;
wcx.hCursor = LoadCursor(NULL, IDC_ARROW);
wcx.hbrBackground = (HBRUSH)(COLOR_WINDOW + 1);
wcx.lpszClassName = APP_NAME;
if (RegisterClassEx(&wcx) == 0) {
return 0;
}
// ウィンドウの作成
HWND hWnd = CreateWindow(
APP_NAME, APP_NAME,
WS_OVERLAPPEDWINDOW | WS_HSCROLL,
CW_USEDEFAULT, 0,
CW_USEDEFAULT, 0,
NULL, NULL, hInstance, NULL);
if (hWnd == NULL) {
return 0;
}
ShowWindow(hWnd, nCmdShow);
UpdateWindow(hWnd);
// メッセージループ
MSG msg;
while (GetMessage(&msg, NULL, 0, 0)) {
TranslateMessage(&msg);
DispatchMessage(&msg);
}
return msg.wParam;
}
void Trace(LPCTSTR format, ...)
{
va_list arg_ptr;
TCHAR buffer[256];
int size;
va_start(arg_ptr, format);
size = _vsntprintf_s(buffer, _countof(buffer), _TRUNCATE, format, arg_ptr);
va_end(arg_ptr);
OutputDebugString(buffer);
if (size < 0) {
OutputDebugString(_T("...\n"));
}
}
BOOL Load(LPTSTR pszFileName)
{
if (ReadWaveFile(pszFileName) == FALSE) {
return FALSE;
}
AnalyzeWave();
return TRUE;
}
BOOL ReadWaveFile(LPTSTR pszFileName)
{
MMRESULT mmr;
sampleLength = 0;
SAFE_FREE(waveformData)
// Open
HMMIO hmmio = mmioOpen(pszFileName, NULL, MMIO_READ);
if (hmmio == NULL) {
return FALSE;
}
// RIFFチャンク
MMCKINFO ckParent;
ckParent.fccType = mmioFOURCC('W','A','V','E');
mmr = mmioDescend(hmmio, &ckParent, NULL, MMIO_FINDRIFF);
if (mmr != MMSYSERR_NOERROR) {
return FALSE;
}
// fmtチャンク
MMCKINFO ckSub;
ckSub.ckid = mmioFOURCC('f','m','t',' ');
mmr = mmioDescend(hmmio, &ckSub, &ckParent, MMIO_FINDCHUNK);
if (mmr != MMSYSERR_NOERROR) {
return FALSE;
}
LONG read = mmioRead(hmmio, (HPSTR)&wfx, 16);
if (read != 16) {
return FALSE;
}
mmioAscend(hmmio, &ckSub, 0);
// dataチャンク
ckSub.ckid = mmioFOURCC('d','a','t','a');
mmr = mmioDescend(hmmio, &ckSub, &ckParent, MMIO_FINDCHUNK);
if (mmr != MMSYSERR_NOERROR) {
return FALSE;
}
sampleLength = ckSub.cksize;
waveformData = (PBYTE)malloc(ckSub.cksize);
if (waveformData == NULL) {
return FALSE;
}
read = mmioRead(hmmio, (HPSTR)waveformData, ckSub.cksize);
if (read != ckSub.cksize) {
return FALSE;
}
mmioAscend(hmmio, &ckSub, 0);
// RIFFチャンク
mmioAscend(hmmio, &ckParent, 0);
// Close
mmioClose(hmmio, 0);
return TRUE;
}
BOOL AnalyzeWave(void)
{
for (int pos = 0; pos <= (int)sampleLength - 201; ) { // 11,025Hz / 55Hz = 200.45
int noteNum = -1;
double jMin = 4.0;
int wl;
for (int n = A1; n <= A5; n++) {
double freq = 440.0 * pow(pow(2.0, n - A4), 1.0 / 12);
double waveLen = wfx.nSamplesPerSec / freq;
double j = LeastSquares(pos, freq, waveLen);
if (j < jMin) {
noteNum = n;
jMin = j;
wl = (int)ceil(waveLen);
}
}
Wave wave;
wave.pos = pos;
wave.wl = wl;
wave.noteNum = noteNum;
wave.vol = Volume(pos, wl);
vWave.push_back(wave);
pos += wl;
}
sampleLength2 = 0;
for (VWave::iterator it = vWave.begin(); it != vWave.end(); it++) {
sampleLength2 += it->wl;
}
waveformData2 = (PBYTE)malloc(sampleLength2);
int pos = 0;
for (VWave::iterator it = vWave.begin(); it != vWave.end(); it++) {
for (int i = 0; i < it->wl; i++) {
double t = i / (double)it->wl;
double y = t < 0.5 ? 1 : -1;
waveformData2[pos + i] = BYTE(128 + 127 * it->vol * y);
}
pos += it->wl;
}
return TRUE;
}
// 最小二乗法
double LeastSquares(int pos, double freq, double waveLen)
{
double sum = 0;
int wl = (int)ceil(waveLen);
for (int i = 0; i < wl; i++) {
double t = i * freq / wfx.nSamplesPerSec;
double y = sin(2 * M_PI * t);
// double y = t < 0.5 ? t * 2 : (t - 1) * 2;
double d = (waveformData[pos + i] - 128.0) / 128 - y;
sum += d * d;
}
return sum / waveLen;
}
// 音量
double Volume(int pos, int wl)
{
double sum = 0;
for (int i = 0; i < wl; i++) {
sum += abs((waveformData[pos + i] - 128.0) / 128);
}
return sum / wl;
}
//------------------------------------------------------------------------------
LRESULT CALLBACK WndProc(HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
{
switch (uMsg) {
case MM_WOM_OPEN:
OnWomOpen();
return 0;
case MM_WOM_DONE:
OnWomDone(hWnd);
return 0;
case WM_PAINT:
OnPaint(hWnd);
return 0;
case WM_HSCROLL:
OnHScroll(hWnd, wParam);
return 0;
case WM_SIZE:
OnSize(hWnd, wParam, lParam);
return 0;
case WM_CHAR:
OnChar(hWnd, wParam, lParam);
return 0;
case WM_CREATE:
OnCreate(hWnd);
return 0;
case WM_DESTROY:
SAFE_FREE(waveformData)
SAFE_FREE(waveformData2)
PostQuitMessage(0);
return 0;
}
return DefWindowProc(hWnd, uMsg, wParam, lParam);
}
void OnCreate(HWND hWnd)
{
RECT rc;
GetClientRect(hWnd, &rc);
siHorz.cbSize = sizeof siHorz;
siHorz.fMask = SIF_RANGE | SIF_PAGE | SIF_POS | SIF_DISABLENOSCROLL;
siHorz.nMin = 0;
siHorz.nMax = sampleLength - 1;
siHorz.nPage = rc.right;
siHorz.nPos = 0;
SetScrollInfo(hWnd, SB_HORZ, &siHorz, FALSE);
}
void OnSize(HWND hWnd, WPARAM wParam, LPARAM lParam)
{
if (wParam == SIZE_MINIMIZED) return;
siHorz.nPage = LOWORD(lParam);
int nPosMax = max(siHorz.nMax - (int)siHorz.nPage + 1, 0);
siHorz.nPos = min(siHorz.nPos, nPosMax);
SetScrollInfo(hWnd, SB_HORZ, &siHorz, TRUE);
}
void OnHScroll(HWND hWnd, WPARAM wParam)
{
int nPos = siHorz.nPos;
switch (LOWORD(wParam)) {
case SB_LINEUP:
nPos -= 10;
break;
case SB_LINEDOWN:
nPos += 10;
break;
case SB_PAGEUP:
nPos -= siHorz.nPage;
break;
case SB_PAGEDOWN:
nPos += siHorz.nPage;
break;
case SB_THUMBTRACK:
SCROLLINFO si;
si.cbSize = sizeof si;
si.fMask = SIF_TRACKPOS;
if (GetScrollInfo(hWnd, SB_HORZ, &si) != 0) {
nPos = si.nTrackPos;
}
break;
}
int nPosMax = max(siHorz.nMax - (int)siHorz.nPage + 1, 0);
nPos = min(nPos, nPosMax);
nPos = max(nPos, 0);
if (nPos == siHorz.nPos) return;
ScrollWindowEx(hWnd, siHorz.nPos - nPos, 0,
NULL, NULL, NULL, NULL, SW_INVALIDATE | SW_ERASE);
siHorz.nPos = nPos;
SetScrollInfo(hWnd, SB_HORZ, &siHorz, TRUE);
UpdateWindow(hWnd);
}
void OnPaint(HWND hWnd)
{
PAINTSTRUCT ps;
HDC hdc = BeginPaint(hWnd, &ps);
RECT rc;
GetClientRect(hWnd, &rc);
// Trace(_T("OnPaint %d %d\n"), ps.rcPaint.left, ps.rcPaint.right);
// waveform
HPEN pen = CreatePen(PS_SOLID, 0, RGB(0,0,255));
HGDIOBJ penOld = SelectObject(hdc, pen);
for (int x = ps.rcPaint.left; x < ps.rcPaint.right; x++) {
size_t i = siHorz.nPos + x;
if (sampleLength <= i) break;
MoveToEx(hdc, x, rc.bottom / 2, NULL);
LineTo(hdc, x, rc.bottom * (255 - waveformData[i]) / 256);
}
SelectObject(hdc, penOld);
DeleteObject(pen);
// waveform2
pen = CreatePen(PS_SOLID, 0, RGB(0,255,0));
penOld = SelectObject(hdc, pen);
for (int x = ps.rcPaint.left; x < ps.rcPaint.right; x++) {
size_t i = siHorz.nPos + x;
if (sampleLength2 <= i) break;
int y = rc.bottom * (255 - waveformData2[i]) / 256;
MoveToEx(hdc, x, y, NULL);
LineTo(hdc, x, y + 1);
}
SelectObject(hdc, penOld);
DeleteObject(pen);
// waveLen
pen = CreatePen(PS_SOLID, 0, RGB(255,0,0));
penOld = SelectObject(hdc, pen);
for (VWave::iterator it = vWave.begin(); it != vWave.end(); it++) {
int x = it->pos - siHorz.nPos;
if (x + it->wl < 0 || rc.right <= x) continue;
MoveToEx(hdc, x, 0, NULL);
LineTo(hdc, x, rc.bottom);
LPCTSTR scale = GetScale(it->noteNum);
TextOut(hdc, x, 0, scale, _tcslen(scale));
}
SelectObject(hdc, penOld);
DeleteObject(pen);
EndPaint(hWnd, &ps);
}
LPCTSTR GetScale(int noteNum)
{
static TCHAR buf[4+1];
_stprintf_s(buf, _T("%s%d"), scale[noteNum % 12], noteNum / 12 - 1);
return buf;
}
void OnChar(HWND hWnd, WPARAM wParam, LPARAM lParam)
{
Trace(_T("OnChar %x %x\n"), wParam, lParam);
switch (wParam) {
case VK_ESCAPE:
DestroyWindow(hWnd);
break;
case 'p':
Play(hWnd);
break;
}
}
void Play(HWND hWnd)
{
if (hCursor) return;
WAVEFORMATEX wf;
wf.wFormatTag = WAVE_FORMAT_PCM;
wf.wBitsPerSample = 8;
wf.nChannels = 1;
wf.nSamplesPerSec = wfx.nSamplesPerSec;
wf.nBlockAlign = (wf.wBitsPerSample / 8) * wf.nChannels;
wf.nAvgBytesPerSec = wf.nSamplesPerSec * wf.nBlockAlign;
wf.cbSize = 0;
MMRESULT mmr = waveOutOpen(&hwo, WAVE_MAPPER, &wf, (DWORD_PTR)hWnd, 0, CALLBACK_WINDOW);
if (mmr != MMSYSERR_NOERROR) {
Trace(_T("waveOutOpen\n"));
return;
}
HCURSOR hWait = LoadCursor(NULL, IDC_WAIT);
hCursor = SetCursor(hWait);
SetClassLong(hWnd, GCL_HCURSOR, (LONG)hWait);
}
void OnWomOpen(void)
{
wh.lpData = (LPSTR)waveformData2;
wh.dwBufferLength = sampleLength2;
wh.dwBytesRecorded = 0;
wh.dwUser = 0;
wh.dwFlags = 0;
wh.dwLoops = 0;
wh.lpNext = NULL;
wh.reserved = 0;
MMRESULT mmr = waveOutPrepareHeader(hwo, &wh, sizeof wh);
if (mmr != MMSYSERR_NOERROR) {
Trace(_T("waveOutPrepareHeader\n"));
return;
}
wh.dwFlags |= WHDR_BEGINLOOP | WHDR_ENDLOOP;
wh.dwLoops = 1;
mmr = waveOutWrite(hwo, &wh, sizeof wh);
if (mmr != MMSYSERR_NOERROR) {
Trace(_T("waveOutWrite\n"));
}
}
void OnWomDone(HWND hWnd)
{
Trace(_T("OnWomDone\n"));
waveOutReset(hwo);
waveOutUnprepareHeader(hwo, &wh, sizeof wh);
waveOutClose(hwo);
SetCursor(hCursor);
SetClassLong(hWnd, GCL_HCURSOR, (LONG)hCursor);
hCursor = NULL;
}