ref: 949671fbac526e9c6ed31b12cbc19ca24c8e3887
dir: /src/pt2_mod2wav.c/
// for finding memory leaks in debug mode with Visual Studio #if defined _DEBUG && defined _MSC_VER #include <crtdbg.h> #endif #include <stdio.h> #include <stdint.h> #include <stdbool.h> #include <sys/stat.h> // stat() #include "pt2_audio.h" #include "pt2_mouse.h" #include "pt2_textout.h" #include "pt2_visuals.h" #include "pt2_mod2wav.h" #include "pt2_downsample2x.h" #include "pt2_config.h" #include "pt2_askbox.h" #include "pt2_replayer.h" #define FADEOUT_CHUNK_SAMPLES 16384 #define TICKS_PER_RENDER_CHUNK 64 static int16_t *mod2WavBuffer, fadeOutBuffer[FADEOUT_CHUNK_SAMPLES * 2]; static char lastFilename[PATH_MAX + 1]; static void calcMod2WavTotalRows(void); void mod2WavDrawFadeoutToggle(void) { fillRect(143, 50, FONT_CHAR_W, FONT_CHAR_H, video.palette[PAL_GENBKG]); if (editor.mod2WavFadeOut) charOut(143, 50, 'X', video.palette[PAL_GENTXT]); } void mod2WavDrawFadeoutSeconds(void) { fillRect(259, 61, FONT_CHAR_W*2, FONT_CHAR_H+1, video.palette[PAL_GENBKG]); charOut(259, 62, '0' + ((editor.mod2WavFadeOutSeconds / 10) % 10), video.palette[PAL_GENTXT]); charOut(267, 62, '0' + ( editor.mod2WavFadeOutSeconds % 10), video.palette[PAL_GENTXT]); } void mod2WavDrawLoopCount(void) { fillRect(259, 72, FONT_CHAR_W*2, FONT_CHAR_H+1, video.palette[PAL_GENBKG]); charOut(259, 73, '0' + ((editor.mod2WavNumLoops / 10) % 10), video.palette[PAL_GENTXT]); charOut(267, 73, '0' + ( editor.mod2WavNumLoops % 10), video.palette[PAL_GENTXT]); } void toggleMod2WavFadeout(void) { editor.mod2WavFadeOut ^= 1; mod2WavDrawFadeoutToggle(); } void mod2WavFadeoutUp(void) { if (editor.mod2WavFadeOutSeconds < 60) { editor.mod2WavFadeOutSeconds++; mod2WavDrawFadeoutSeconds(); } } void mod2WavFadeoutDown(void) { if (editor.mod2WavFadeOutSeconds > 1) { editor.mod2WavFadeOutSeconds--; mod2WavDrawFadeoutSeconds(); } } void mod2WavLoopCountUp(void) { if (editor.mod2WavNumLoops < 50) { editor.mod2WavNumLoops++; mod2WavDrawLoopCount(); } } void mod2WavLoopCountDown(void) { if (editor.mod2WavNumLoops > 0) { editor.mod2WavNumLoops--; mod2WavDrawLoopCount(); } } void drawMod2WavProgressDialog(void) { drawFramework3(120, 44, 200, 55); textOut2(150, 53, "- RENDERING MODULE -"); const int32_t buttonW = (MOD2WAV_CANCEL_BTN_X2 - MOD2WAV_CANCEL_BTN_X1)+1; const int32_t buttonH = (MOD2WAV_CANCEL_BTN_Y2 - MOD2WAV_CANCEL_BTN_Y1)+1; drawButton1(MOD2WAV_CANCEL_BTN_X1, MOD2WAV_CANCEL_BTN_Y1, buttonW, buttonH, "CANCEL"); } static void showMod2WavProgress(void) { char percText[16]; if (song->rowsInTotal == 0) return; // render progress bar int32_t percent = (song->rowsCounter * 100) / song->rowsInTotal; if (percent > 100) percent = 100; const int32_t x = 130; const int32_t y = 66; const int32_t w = 180; const int32_t h = 11; // foreground (progress) const int32_t progressBarWidth = (percent * w) / 100; if (progressBarWidth > 0) fillRect(x, y, progressBarWidth, h, video.palette[PAL_GENBKG2]); // foreground (progress) // background int32_t bgWidth = w - progressBarWidth; if (bgWidth > 0) fillRect(x+progressBarWidth, y, bgWidth, h, video.palette[PAL_BORDER]); // draw percentage text sprintf(percText, "%d%%", percent); const int32_t percTextW = (int32_t)strlen(percText) * (FONT_CHAR_W-1); textOutTight(x + ((w - percTextW) / 2), y + ((h - FONT_CHAR_H) / 2), percText, video.palette[PAL_GENTXT]); } static void resetAudio(void) { audio.oversamplingFlag = (audio.outputRate < 96000); const int32_t paulaMixFrequency = audio.oversamplingFlag ? audio.outputRate*2 : audio.outputRate; paulaSetup(paulaMixFrequency, audio.amigaModel); generateBpmTable(audio.outputRate, editor.timingMode == TEMPO_MODE_VBLANK); clearMixerDownsamplerStates(); modSetTempo(song->currBPM, true); // update BPM (samples per tick) with the tracker's audio frequency } static void handleMod2WavEnd(void) { pointerSetMode(POINTER_MODE_IDLE, DO_CARRY); if (editor.abortMod2Wav) { displayErrorMsg("MOD2WAV ABORTED!"); } else { displayMsg("MOD RENDERED!"); setMsgPointer(); } removeAskBox(); resetAudio(); editor.mod2WavOngoing = false; // must be set before calling resetSong() ! resetSong(); } void updateMod2WavDialog(void) { if (ui.updateMod2WavDialog) { ui.updateMod2WavDialog = false; if (editor.mod2WavOngoing) { if (ui.mod2WavFinished) { ui.mod2WavFinished = false; handleMod2WavEnd(); } else { showMod2WavProgress(); } } } } static int32_t SDLCALL mod2WavThreadFunc(void *ptr) { wavHeader_t wavHeader; FILE *f = (FILE *)ptr; assert(mod2WavBuffer != NULL && f != NULL); // skip wav header place, render data first fseek(f, sizeof (wavHeader_t), SEEK_SET); uint32_t sampleCounter = 0; uint64_t samplesToMixFrac = 0; int8_t numLoops = editor.mod2WavNumLoops; bool renderDone = false; while (!renderDone) { uint32_t samplesInChunk = 0; // render several ticks at once to prevent frequent disk I/O (speeds up the process) int16_t *ptr16 = mod2WavBuffer; for (uint32_t i = 0; i < TICKS_PER_RENDER_CHUNK; i++) { if (!editor.mod2WavOngoing || renderDone || editor.abortMod2Wav) { renderDone = true; break; } /* PT replayer ticker (also sets audio.samplesPerTickInt and audio.samplesPerTickFrac). ** Returns false on end of song. */ if (!intMusic()) { if (--numLoops < 0) { renderDone = true; // this tick is the last tick } else { // clear the "last visisted rows" table and let the song continue playing (loop) memset(editor.rowVisitTable, 0, MOD_ORDERS * MOD_ROWS); } } uint32_t samplesToMix = audio.samplesPerTickInt; samplesToMixFrac += audio.samplesPerTickFrac; if (samplesToMixFrac >= BPM_FRAC_SCALE) { samplesToMixFrac &= BPM_FRAC_MASK; samplesToMix++; } outputAudio(ptr16, samplesToMix); ptr16 += samplesToMix * 2; // *2 for stereo samplesInChunk += samplesToMix; sampleCounter += samplesToMix; ui.updateMod2WavDialog = true; } // write buffer to disk if (samplesInChunk > 0) fwrite(mod2WavBuffer, sizeof (int16_t), samplesInChunk * 2, f); } ui.updateMod2WavDialog = true; uint32_t endOfDataOffset = ftell(f); free(mod2WavBuffer); if (sampleCounter & 1) fputc(0, f); // pad align byte uint32_t totalRiffChunkLen = (uint32_t)ftell(f) - 8; // go back and fill in WAV header rewind(f); wavHeader.chunkID = 0x46464952; // "RIFF" wavHeader.chunkSize = totalRiffChunkLen; wavHeader.format = 0x45564157; // "WAVE" wavHeader.subchunk1ID = 0x20746D66; // "fmt " wavHeader.subchunk1Size = 16; wavHeader.audioFormat = 1; wavHeader.numChannels = 2; wavHeader.sampleRate = config.mod2WavOutputFreq; wavHeader.bitsPerSample = 16; wavHeader.byteRate = (wavHeader.sampleRate * wavHeader.numChannels * wavHeader.bitsPerSample) / 8; wavHeader.blockAlign = (wavHeader.numChannels * wavHeader.bitsPerSample) / 8; wavHeader.subchunk2ID = 0x61746164; // "data" wavHeader.subchunk2Size = sampleCounter * sizeof (int16_t) * 2; // write main header fwrite(&wavHeader, sizeof (wavHeader_t), 1, f); fclose(f); // apply fadeout (if enabled) if (editor.mod2WavFadeOut) { uint32_t numFadeOutSamples = config.mod2WavOutputFreq * editor.mod2WavFadeOutSeconds; if (numFadeOutSamples > sampleCounter) numFadeOutSamples = sampleCounter; f = fopen(lastFilename, "r+b"); const double dFadeOutDelta = 1.0 / numFadeOutSamples; double dFadeOutVal = 1.0; fseek(f, endOfDataOffset - (numFadeOutSamples * sizeof (int16_t) * 2), SEEK_SET); uint32_t samplesLeft = numFadeOutSamples; while (samplesLeft > 0) { uint32_t samplesTodo = FADEOUT_CHUNK_SAMPLES; if (samplesTodo > samplesLeft) samplesTodo = samplesLeft; fread(fadeOutBuffer, sizeof (int16_t), samplesTodo * 2, f); fseek(f, 0 - (samplesTodo * sizeof (int16_t) * 2), SEEK_CUR); // apply fadeout for (uint32_t i = 0; i < samplesTodo; i++) { fadeOutBuffer[(i*2)+0] = (int16_t)(fadeOutBuffer[(i*2)+0] * dFadeOutVal); // L fadeOutBuffer[(i*2)+1] = (int16_t)(fadeOutBuffer[(i*2)+1] * dFadeOutVal); // R dFadeOutVal -= dFadeOutDelta; } fwrite(fadeOutBuffer, sizeof (int16_t), samplesTodo * 2, f); samplesLeft -= samplesTodo; } fclose(f); } ui.mod2WavFinished = true; ui.updateMod2WavDialog = true; return true; } bool mod2WavRender(char *filename) { struct stat statBuffer; lastFilename[0] = '\0'; if (stat(filename, &statBuffer) == 0) { if (!askBox(ASKBOX_YES_NO, "OVERWRITE FILE?")) return false; } FILE *fOut = fopen(filename, "wb"); if (fOut == NULL) { displayErrorMsg("FILE I/O ERROR"); return false; } strncpy(lastFilename, filename, PATH_MAX-1); const int32_t paulaMixFrequency = config.mod2WavOutputFreq * 2; // *2 for oversampling (we always do oversampling in MOD2WAV) int32_t maxSamplesPerTick = (int32_t)ceil(paulaMixFrequency / (MIN_BPM / 2.5)) + 1; mod2WavBuffer = (int16_t *)malloc((TICKS_PER_RENDER_CHUNK * maxSamplesPerTick) * sizeof (int16_t) * 2); if (mod2WavBuffer == NULL) { fclose(fOut); statusOutOfMemory(); return false; } editor.mod2WavOngoing = true; // set this first // do some prep work audio.oversamplingFlag = true; generateBpmTable(config.mod2WavOutputFreq, editor.timingMode == TEMPO_MODE_VBLANK); paulaSetup(paulaMixFrequency, audio.amigaModel); storeTempVariables(); calcMod2WavTotalRows(); restartSong(); // this also updates BPM (samples per tick) with the MOD2WAV audio output rate clearMixerDownsamplerStates(); drawMod2WavProgressDialog(); editor.abortMod2Wav = false; pointerSetMode(POINTER_MODE_MSG2, NO_CARRY); setStatusMessage("RENDERING MOD...", NO_CARRY); editor.mod2WavThread = SDL_CreateThread(mod2WavThreadFunc, NULL, fOut); if (editor.mod2WavThread == NULL) { fclose(fOut); free(mod2WavBuffer); doStopIt(true); editor.mod2WavOngoing = false; // must be set before calling resetAudio() resetAudio(); removeAskBox(); pointerSetMode(POINTER_MODE_IDLE, DO_CARRY); displayErrorMsg("THREAD ERROR !"); return false; } return true; } #define CALC__END_OF_SONG \ if (--numLoops < 0) \ { \ calcingRows = false; \ break; \ } \ else \ { \ memset(editor.rowVisitTable, 0, MOD_ORDERS * MOD_ROWS); \ } // ONLY used for a visual percentage counter, so accuracy is not very important static void calcMod2WavTotalRows(void) { int8_t n_pattpos[PAULA_VOICES], n_loopcount[PAULA_VOICES]; // for pattern loop memset(n_pattpos, 0, sizeof (n_pattpos)); memset(n_loopcount, 0, sizeof (n_loopcount)); song->rowsCounter = song->rowsInTotal = 0; uint8_t modRow = 0; int16_t modOrder = 0; uint16_t modPattern = song->header.order[0]; uint8_t pBreakPosition = 0; bool posJumpAssert = false; bool pBreakFlag = false; memset(editor.rowVisitTable, 0, MOD_ORDERS * MOD_ROWS); int8_t numLoops = editor.mod2WavNumLoops; // make a copy bool calcingRows = true; while (calcingRows) { editor.rowVisitTable[(modOrder * MOD_ROWS) + modRow] = true; for (int32_t ch = 0; ch < PAULA_VOICES; ch++) { note_t *note = &song->patterns[modPattern][(modRow * PAULA_VOICES) + ch]; if (note->command == 0x0B) // Bxx - Position Jump { modOrder = note->param - 1; pBreakPosition = 0; posJumpAssert = true; } else if (note->command == 0x0D) // Dxx - Pattern Break { pBreakPosition = (((note->param >> 4) * 10) + (note->param & 0x0F)); if (pBreakPosition > 63) pBreakPosition = 0; posJumpAssert = true; } else if (note->command == 0x0F && note->param == 0) // F00 - Set Speed 0 (stop) { CALC__END_OF_SONG } else if (note->command == 0x0E && (note->param >> 4) == 0x06) // E6x - Pattern Loop { uint8_t pos = note->param & 0x0F; if (pos == 0) { n_pattpos[ch] = modRow; } else if (n_loopcount[ch] == 0) { n_loopcount[ch] = pos; pBreakPosition = n_pattpos[ch]; pBreakFlag = true; for (pos = pBreakPosition; pos <= modRow; pos++) editor.rowVisitTable[(modOrder * MOD_ROWS) + pos] = false; } else if (--n_loopcount[ch]) { pBreakPosition = n_pattpos[ch]; pBreakFlag = true; for (pos = pBreakPosition; pos <= modRow; pos++) editor.rowVisitTable[(modOrder * MOD_ROWS) + pos] = false; } } } modRow++; song->rowsInTotal++; if (pBreakFlag) { modRow = pBreakPosition; pBreakPosition = 0; pBreakFlag = false; } if (modRow >= MOD_ROWS || posJumpAssert) { modRow = pBreakPosition; pBreakPosition = 0; posJumpAssert = false; modOrder = (modOrder + 1) & 127; if (modOrder >= song->header.numOrders) { modOrder = 0; CALC__END_OF_SONG } modPattern = song->header.order[modOrder]; if (modPattern > MAX_PATTERNS-1) modPattern = MAX_PATTERNS-1; } if (calcingRows && editor.rowVisitTable[(modOrder * MOD_ROWS) + modRow]) { // row has been visited before, we're now done! CALC__END_OF_SONG } } }