ref: 2b5daa544020703f6a284dbcaf870d591e3c970c
dir: /plugins/cooledit/Faac.cpp/
/* FAAC - codec plugin for Cooledit Copyright (C) 2002 Antonio Foranna 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. 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., 675 Mass Ave, Cambridge, MA 02139, USA. The author can be contacted at: kreel@tiscali.it */ #include <windows.h> #include <stdio.h> // FILE * #include "resource.h" #include "filters.h" // CoolEdit #include <faac.h> #include <faad.h> // FAAD2 version #include <win32_ver.h> // mpeg4ip version #include "CRegistry.h" #include "Defines.h" // my defines #include "Structs.h" // my structs // ********************************************************************************************* extern HINSTANCE hInst; // ----------------------------------------------------------------------------------------------- extern BOOL DialogMsgProcDecoder(HWND hWndDlg, UINT Message, WPARAM wParam, LPARAM lParam); extern void ReadCfgDec(MY_DEC_CFG *cfg); extern void WriteCfgDec(MY_DEC_CFG *cfg); // ********************************************************************************************* #define RAW 0 #define ADTS 1 // ********************************************************************************************* typedef struct output_tag // any special vars associated with output file { FILE *aacFile; long Samprate; WORD BitsPerSample; WORD Channels; //DWORD src_size; //char *dst_name; // name of compressed file faacEncHandle hEncoder; unsigned char *bitbuf; DWORD maxBytesOutput; long samplesInput; } MYOUTPUT; // ********************************************************************************************* void ReadCfgEnc(MY_ENC_CFG *cfg) { CRegistry reg; if(reg.openCreateReg(HKEY_LOCAL_MACHINE,REGISTRY_PROGRAM_NAME "\\FAAC")) { cfg->AutoCfg=reg.getSetRegBool("Auto",true); cfg->EncCfg.mpegVersion=reg.getSetRegDword("MPEG version",MPEG2); cfg->EncCfg.aacObjectType=reg.getSetRegDword("Profile",LOW); cfg->EncCfg.allowMidside=reg.getSetRegDword("MidSide",true); cfg->EncCfg.useTns=reg.getSetRegDword("TNS",true); cfg->EncCfg.useLfe=reg.getSetRegDword("LFE",false); cfg->EncCfg.bitRate=reg.getSetRegDword("BitRate",0); cfg->EncCfg.bandWidth=reg.getSetRegDword("BandWidth",0); cfg->EncCfg.outputFormat=reg.getSetRegDword("Header",ADTS); } else MessageBox(0,"Can't open registry!",0,MB_OK|MB_ICONSTOP); } // ----------------------------------------------------------------------------------------------- void WriteCfgEnc(MY_ENC_CFG *cfg) { CRegistry reg; if(reg.openCreateReg(HKEY_LOCAL_MACHINE,REGISTRY_PROGRAM_NAME "\\FAAC")) { reg.setRegBool("Auto",cfg->AutoCfg); reg.setRegDword("MPEG version",cfg->EncCfg.mpegVersion); reg.setRegDword("Profile",cfg->EncCfg.aacObjectType); reg.setRegDword("MidSide",cfg->EncCfg.allowMidside); reg.setRegDword("TNS",cfg->EncCfg.useTns); reg.setRegDword("LFE",cfg->EncCfg.useLfe); reg.setRegDword("BitRate",cfg->EncCfg.bitRate); reg.setRegDword("BandWidth",cfg->EncCfg.bandWidth); reg.setRegDword("Header",cfg->EncCfg.outputFormat); } else MessageBox(0,"Can't open registry!",0,MB_OK|MB_ICONSTOP); } // ----------------------------------------------------------------------------------------------- #define INIT_CB(hWnd,nID,list,IdSelected) \ { \ for(int i=0; list[i]; i++) \ SendMessage(GetDlgItem(hWnd, nID), CB_ADDSTRING, 0, (LPARAM)list[i]); \ SendMessage(GetDlgItem(hWnd, nID), CB_SETCURSEL, IdSelected, 0); \ } // ----------------------------------------------------------------------------------------------- #define DISABLE_LTP \ { \ if(IsDlgButtonChecked(hWndDlg,IDC_RADIO_MPEG2) && \ IsDlgButtonChecked(hWndDlg,IDC_RADIO_LTP)) \ { \ CheckDlgButton(hWndDlg,IDC_RADIO_LTP,FALSE); \ CheckDlgButton(hWndDlg,IDC_RADIO_MAIN,TRUE); \ } \ EnableWindow(GetDlgItem(hWndDlg, IDC_RADIO_LTP), FALSE); \ } // ----------------------------------------------------------------------------------------------- // EnableWindow(GetDlgItem(hWndDlg, IDC_RADIO_SSR), Enabled); // EnableWindow(GetDlgItem(hWndDlg, IDC_CHK_USELFE), Enabled); #define DISABLE_CTRL(Enabled) \ { \ CheckDlgButton(hWndDlg,IDC_CHK_AUTOCFG, !Enabled); \ EnableWindow(GetDlgItem(hWndDlg, IDC_RADIO_MPEG4), Enabled); \ EnableWindow(GetDlgItem(hWndDlg, IDC_RADIO_MPEG2), Enabled); \ EnableWindow(GetDlgItem(hWndDlg, IDC_RADIO_RAW), Enabled); \ EnableWindow(GetDlgItem(hWndDlg, IDC_RADIO_ADTS), Enabled); \ EnableWindow(GetDlgItem(hWndDlg, IDC_CHK_ALLOWMIDSIDE), Enabled); \ EnableWindow(GetDlgItem(hWndDlg, IDC_CHK_USETNS), Enabled); \ EnableWindow(GetDlgItem(hWndDlg, IDC_CB_BITRATE), Enabled); \ EnableWindow(GetDlgItem(hWndDlg, IDC_CB_BANDWIDTH), Enabled); \ EnableWindow(GetDlgItem(hWndDlg, IDC_RADIO_MAIN), Enabled); \ EnableWindow(GetDlgItem(hWndDlg, IDC_RADIO_LOW), Enabled); \ EnableWindow(GetDlgItem(hWndDlg, IDC_RADIO_LTP), Enabled); \ if(IsDlgButtonChecked(hWndDlg,IDC_RADIO_MPEG4)) \ EnableWindow(GetDlgItem(hWndDlg, IDC_RADIO_LTP), Enabled); \ else \ DISABLE_LTP \ } // ----------------------------------------------------------------------------------------------- BOOL DialogMsgProcAbout(HWND hWndDlg, UINT Message, WPARAM wParam, LPARAM lParam) { switch(Message) { case WM_INITDIALOG: { char buf[512]; sprintf(buf, APP_NAME " plugin " APP_VER " by Antonio Foranna\n\n" "Engines used:\n" "\tFAAC v%g\n" "\tFAAD2 v" FAAD2_VERSION "\n" "\t" PACKAGE " v" VERSION "\n\n" "This code is given with FAAC package and does not contain executables.\n" "This program is free software and can be distributed/modifyed under the terms of the GNU General Public License.\n" "This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY.\n\n" "Compiled on %s\n", FAACENC_VERSION, __DATE__ ); SetDlgItemText(hWndDlg, IDC_L_ABOUT, buf); } break; case WM_COMMAND: switch(LOWORD(wParam)) { case IDOK: EndDialog(hWndDlg, 0); break; case IDC_AUDIOCODING: ShellExecute(hWndDlg, NULL, "http://www.audiocoding.com", NULL, NULL, SW_SHOW); break; case IDC_MPEG4IP: ShellExecute(hWndDlg, NULL, "http://www.mpeg4ip.net", NULL, NULL, SW_SHOW); break; case IDC_EMAIL: ShellExecute(hWndDlg, NULL, "mailto:kreel@tiscali.it", NULL, NULL, SW_SHOW); break; } break; default: return FALSE; } return TRUE; } // ----------------------------------------------------------------------------------------------- BOOL DIALOGMsgProc(HWND hWndDlg, UINT Message, WPARAM wParam, LPARAM lParam) { switch(Message) { case WM_INITDIALOG: { char buf[50]; char *BitRate[]={"Auto","8","18","20","24","32","40","48","56","64","96","112","128","160","192","256",0}; char *BandWidth[]={"Auto","Full","4000","8000","16000","22050","24000","48000",0}; MY_ENC_CFG cfg; ReadCfgEnc(&cfg); INIT_CB(hWndDlg,IDC_CB_BITRATE,BitRate,0); INIT_CB(hWndDlg,IDC_CB_BANDWIDTH,BandWidth,0); if(cfg.EncCfg.mpegVersion==MPEG4) CheckDlgButton(hWndDlg,IDC_RADIO_MPEG4,TRUE); else CheckDlgButton(hWndDlg,IDC_RADIO_MPEG2,TRUE); switch(cfg.EncCfg.aacObjectType) { case MAIN: CheckDlgButton(hWndDlg,IDC_RADIO_MAIN,TRUE); break; case LOW: CheckDlgButton(hWndDlg,IDC_RADIO_LOW,TRUE); break; case SSR: CheckDlgButton(hWndDlg,IDC_RADIO_SSR,TRUE); break; case LTP: CheckDlgButton(hWndDlg,IDC_RADIO_LTP,TRUE); DISABLE_LTP break; } switch(cfg.EncCfg.outputFormat) { case RAW: CheckDlgButton(hWndDlg,IDC_RADIO_RAW,TRUE); break; case ADTS: CheckDlgButton(hWndDlg,IDC_RADIO_ADTS,TRUE); break; } CheckDlgButton(hWndDlg, IDC_CHK_ALLOWMIDSIDE, cfg.EncCfg.allowMidside); CheckDlgButton(hWndDlg, IDC_CHK_USETNS, cfg.EncCfg.useTns); CheckDlgButton(hWndDlg, IDC_CHK_USELFE, cfg.EncCfg.useLfe); switch(cfg.EncCfg.bitRate) { case 0: SendMessage(GetDlgItem(hWndDlg, IDC_CB_BITRATE), CB_SETCURSEL, 0, 0); break; default: sprintf(buf,"%lu",cfg.EncCfg.bitRate); SetDlgItemText(hWndDlg, IDC_CB_BITRATE, buf); break; } switch(cfg.EncCfg.bandWidth) { case 0: SendMessage(GetDlgItem(hWndDlg, IDC_CB_BANDWIDTH), CB_SETCURSEL, 0, 0); break; case 0xffffffff: SendMessage(GetDlgItem(hWndDlg, IDC_CB_BANDWIDTH), CB_SETCURSEL, 1, 0); break; default: sprintf(buf,"%lu",cfg.EncCfg.bandWidth); SetDlgItemText(hWndDlg, IDC_CB_BANDWIDTH, buf); break; } CheckDlgButton(hWndDlg,IDC_CHK_AUTOCFG, cfg.AutoCfg); DISABLE_CTRL(!cfg.AutoCfg); } break; // End of WM_INITDIALOG case WM_CLOSE: // Closing the Dialog behaves the same as Cancel PostMessage(hWndDlg, WM_COMMAND, IDCANCEL, 0); break; // End of WM_CLOSE case WM_COMMAND: switch(LOWORD(wParam)) { case IDC_CHK_AUTOCFG: { char Enabled=!IsDlgButtonChecked(hWndDlg,IDC_CHK_AUTOCFG); DISABLE_CTRL(Enabled); } break; case IDOK: { char buf[50]; MY_ENC_CFG cfg; cfg.AutoCfg=IsDlgButtonChecked(hWndDlg,IDC_CHK_AUTOCFG) ? TRUE : FALSE; cfg.EncCfg.mpegVersion=IsDlgButtonChecked(hWndDlg,IDC_RADIO_MPEG4) ? MPEG4 : MPEG2; if(IsDlgButtonChecked(hWndDlg,IDC_RADIO_MAIN)) cfg.EncCfg.aacObjectType=MAIN; if(IsDlgButtonChecked(hWndDlg,IDC_RADIO_LOW)) cfg.EncCfg.aacObjectType=LOW; if(IsDlgButtonChecked(hWndDlg,IDC_RADIO_SSR)) cfg.EncCfg.aacObjectType=SSR; if(IsDlgButtonChecked(hWndDlg,IDC_RADIO_LTP)) cfg.EncCfg.aacObjectType=LTP; cfg.EncCfg.allowMidside=IsDlgButtonChecked(hWndDlg, IDC_CHK_ALLOWMIDSIDE); cfg.EncCfg.useTns=IsDlgButtonChecked(hWndDlg, IDC_CHK_USETNS); cfg.EncCfg.useLfe=IsDlgButtonChecked(hWndDlg, IDC_CHK_USELFE); GetDlgItemText(hWndDlg, IDC_CB_BITRATE, buf, 50); switch(*buf) { case 'A': // Auto cfg.EncCfg.bitRate=0; break; default: cfg.EncCfg.bitRate=GetDlgItemInt(hWndDlg, IDC_CB_BITRATE, 0, FALSE); } GetDlgItemText(hWndDlg, IDC_CB_BANDWIDTH, buf, 50); switch(*buf) { case 'A': // Auto cfg.EncCfg.bandWidth=0; break; case 'F': // Full cfg.EncCfg.bandWidth=0xffffffff; break; default: cfg.EncCfg.bandWidth=GetDlgItemInt(hWndDlg, IDC_CB_BANDWIDTH, 0, FALSE); } cfg.EncCfg.outputFormat=IsDlgButtonChecked(hWndDlg,IDC_RADIO_RAW) ? RAW : ADTS; WriteCfgEnc(&cfg); EndDialog(hWndDlg, 1); } break; case IDCANCEL: // Ignore data values entered into the controls // and dismiss the dialog window returning FALSE EndDialog(hWndDlg, FALSE); break; case IDC_BTN_ABOUT: DialogBox((HINSTANCE)hInst,(LPCSTR)MAKEINTRESOURCE(IDD_ABOUT), (HWND)hWndDlg, (DLGPROC)DialogMsgProcAbout); break; case IDC_RADIO_MPEG4: EnableWindow(GetDlgItem(hWndDlg, IDC_RADIO_LTP), !IsDlgButtonChecked(hWndDlg,IDC_CHK_AUTOCFG)); break; case IDC_RADIO_MPEG2: EnableWindow(GetDlgItem(hWndDlg, IDC_RADIO_LTP), FALSE); DISABLE_LTP break; } break; // End of WM_COMMAND default: return FALSE; } return TRUE; } // ********************************************************************************************* // ********************************************************************************************* // ********************************************************************************************* #define ERROR_FGO(msg) \ { \ if(msg) \ MessageBox(hWnd, msg, APP_NAME " plugin", MB_OK|MB_ICONSTOP); \ return 0; \ } // ----------------------------------------------------------------------------------------------- DWORD FAR PASCAL FilterGetOptions(HWND hWnd, HINSTANCE hInst, long lSamprate, WORD wChannels, WORD wBitsPerSample, DWORD dwOptions) { CRegistry reg; long retVal; BOOL OpenDialog=FALSE; if(!reg.openCreateReg(HKEY_LOCAL_MACHINE,REGISTRY_PROGRAM_NAME "\\FAAD")) ERROR_FGO("Can't open registry!") else if(OpenDialog=reg.getSetRegBool("OpenDialog",FALSE)) reg.setRegBool("OpenDialog",FALSE); if(OpenDialog) { MY_DEC_CFG *Cfg; if(!(Cfg=(MY_DEC_CFG *)malloc(sizeof(MY_DEC_CFG)))) ERROR_FGO("Memory allocation error"); ReadCfgDec(Cfg); Cfg->DecCfg.defSampleRate=lSamprate; switch(wBitsPerSample) { case 16: Cfg->DecCfg.outputFormat=FAAD_FMT_16BIT; break; case 24: Cfg->DecCfg.outputFormat=FAAD_FMT_24BIT; break; case 32: Cfg->DecCfg.outputFormat=FAAD_FMT_32BIT; break; default: ERROR_FGO("Invalid Bps"); } Cfg->Channels=(BYTE)wChannels; retVal=DialogBoxParam((HINSTANCE)hInst,(LPCSTR)MAKEINTRESOURCE(IDD_DECODER), (HWND)hWnd, (DLGPROC)DialogMsgProcDecoder, (DWORD)Cfg); WriteCfgDec(Cfg); FREE(Cfg); } else retVal=DialogBoxParam((HINSTANCE)hInst,(LPCSTR)MAKEINTRESOURCE(IDD_ENCODER), (HWND)hWnd, (DLGPROC)DIALOGMsgProc, dwOptions); if(retVal==-1) ERROR_FGO("DialogBoxParam"); return retVal; } // ********************************************************************************************* // GetSuggestedSampleType() is called if OpenFilterOutput() returns NULL void FAR PASCAL GetSuggestedSampleType(LONG *lplSamprate, WORD *lpwBitsPerSample, WORD *wChannels) { *lplSamprate=0; // don't care *lpwBitsPerSample= *lpwBitsPerSample==16 ? 0: 16; *wChannels= *wChannels<49 ? 0 : 48; } // ********************************************************************************************* void FAR PASCAL CloseFilterOutput(HANDLE hOutput) { if(!hOutput) return; MYOUTPUT *mo; GLOBALLOCK(mo,hOutput,MYOUTPUT,return); if(mo->aacFile) { fclose(mo->aacFile); mo->aacFile=0; } if(mo->hEncoder) faacEncClose(mo->hEncoder); if(mo->bitbuf) { free(mo->bitbuf); mo->bitbuf=0; } // FREE(mi->dst_name); GlobalUnlock(hOutput); GlobalFree(hOutput); } // ********************************************************************************************* #define ERROR_OFO(msg) \ { \ if(msg) \ MessageBox(0, msg, APP_NAME " plugin", MB_OK|MB_ICONSTOP); \ if(hOutput) \ { \ GlobalUnlock(hOutput); \ CloseFilterOutput(hOutput); \ } \ return 0; \ } // ----------------------------------------------------------------------------------------------- HANDLE FAR PASCAL OpenFilterOutput(LPSTR lpstrFilename,long lSamprate,WORD wBitsPerSample,WORD wChannels,long lSize, long far *lpChunkSize, DWORD dwOptions) { HANDLE hOutput; MYOUTPUT *mo; MY_ENC_CFG cfg; DWORD samplesInput, maxBytesOutput; if(wBitsPerSample!=16) // FAAC supports 16 bit audio only! return 0; if(!(hOutput=GlobalAlloc(GMEM_MOVEABLE|GMEM_SHARE|GMEM_ZEROINIT,sizeof(MYOUTPUT)))) ERROR_OFO("Memory allocation error: hOutput"); if(!(mo=(MYOUTPUT *)GlobalLock(hOutput))) ERROR_OFO("GlobalLock(hOutput)"); // open the aac output file if(!(mo->aacFile=fopen(lpstrFilename, "wb"))) ERROR_OFO("Can't create file"); // use bufferized stream setvbuf(mo->aacFile,NULL,_IOFBF,32767); // open the encoder library if(!(mo->hEncoder=faacEncOpen(lSamprate, wChannels, &samplesInput, &maxBytesOutput))) ERROR_OFO("Can't open library"); if(!(mo->bitbuf=(unsigned char*)malloc(maxBytesOutput*sizeof(unsigned char)))) ERROR_OFO("Memory allocation error: output buffer"); ReadCfgEnc(&cfg); if(!cfg.AutoCfg) { faacEncConfigurationPtr myFormat=&cfg.EncCfg; faacEncConfigurationPtr CurFormat=faacEncGetCurrentConfiguration(mo->hEncoder); if(!myFormat->bitRate) myFormat->bitRate=CurFormat->bitRate; else myFormat->bitRate*=1000; switch(myFormat->bandWidth) { case 0: myFormat->bandWidth=CurFormat->bandWidth; break; case 0xffffffff: myFormat->bandWidth=lSamprate/2; break; default: break; } if(!faacEncSetConfiguration(mo->hEncoder, myFormat)) ERROR_OFO("Unsupported parameters"); } *lpChunkSize=samplesInput*2; // mo->src_size=lSize; mo->Samprate=lSamprate; mo->BitsPerSample=wBitsPerSample; mo->Channels=wChannels; mo->samplesInput=samplesInput; mo->maxBytesOutput=maxBytesOutput; // mi->dst_name=strdup(lpstrFilename); GlobalUnlock(hOutput); return hOutput; } // ********************************************************************************************* #define ERROR_WFO(msg) \ { \ if(msg) \ MessageBox(0, msg, APP_NAME " plugin", MB_OK|MB_ICONSTOP); \ if(hOutput) \ GlobalUnlock(hOutput); \ return 0; \ } // ----------------------------------------------------------------------------------------------- DWORD FAR PASCAL WriteFilterOutput(HANDLE hOutput, unsigned char far *buf, long lBytes) { if(!hOutput) return 0; int bytesWritten; int bytesEncoded; MYOUTPUT far *mo; GLOBALLOCK(mo,hOutput,MYOUTPUT,return 0); // call the actual encoding routine bytesEncoded=faacEncEncode(mo->hEncoder, (short *)buf, mo->samplesInput, mo->bitbuf, mo->maxBytesOutput); if(bytesEncoded<1) // end of flushing process { if(bytesEncoded<0) ERROR_WFO("faacEncEncode"); bytesWritten=lBytes ? 1 : 0; // bytesWritten==0 stops CoolEdit. 1 is used when intializing flushing process } else { // write bitstream to aac file bytesWritten=fwrite(mo->bitbuf, 1, bytesEncoded, mo->aacFile); if(bytesWritten!=bytesEncoded) ERROR_WFO("fwrite"); } GlobalUnlock(hOutput); return bytesWritten; }