config.c

视频压缩编解码标准MPEG4商业级别的VC代码实现标准

/* ===================================================================================== */

/* initialise pages */
static void adv_init(HWND hDlg, int idd, CONFIG * config)
{
	unsigned int i;

	switch(idd) {
	case IDD_PROFILE :
		for (i=0; i<sizeof(profiles)/sizeof(profile_t); i++)
			SendDlgItemMessage(hDlg, IDC_PROFILE_PROFILE, CB_ADDSTRING, 0, (LPARAM)profiles[i].name);
		SendDlgItemMessage(hDlg, IDC_QUANTTYPE, CB_ADDSTRING, 0, (LPARAM)"H.263");
		SendDlgItemMessage(hDlg, IDC_QUANTTYPE, CB_ADDSTRING, 0, (LPARAM)"MPEG");
		SendDlgItemMessage(hDlg, IDC_QUANTTYPE, CB_ADDSTRING, 0, (LPARAM)"MPEG-Custom");
		break;

	case IDD_AR:
		SendDlgItemMessage(hDlg, IDC_ASPECT_RATIO, CB_ADDSTRING, 0, (LPARAM)"Square (default)");
		SendDlgItemMessage(hDlg, IDC_ASPECT_RATIO, CB_ADDSTRING, 0, (LPARAM)"4:3 PAL");
		SendDlgItemMessage(hDlg, IDC_ASPECT_RATIO, CB_ADDSTRING, 0, (LPARAM)"4:3 NTSC");
		SendDlgItemMessage(hDlg, IDC_ASPECT_RATIO, CB_ADDSTRING, 0, (LPARAM)"16:9 PAL");
		SendDlgItemMessage(hDlg, IDC_ASPECT_RATIO, CB_ADDSTRING, 0, (LPARAM)"16:9 NTSC");
		SendDlgItemMessage(hDlg, IDC_ASPECT_RATIO, CB_ADDSTRING, 0, (LPARAM)"Custom...");
		break;

	case IDD_LEVEL :
		for (i=0; i<sizeof(profiles)/sizeof(profile_t); i++)
			SendDlgItemMessage(hDlg, IDC_LEVEL_PROFILE, CB_ADDSTRING, 0, (LPARAM)profiles[i].name);
		break;

	case IDD_BITRATE :
		SendDlgItemMessage(hDlg, IDC_BITRATE_CFORMAT, CB_ADDSTRING, 0, (LPARAM)"AVI-Legacy");
		SendDlgItemMessage(hDlg, IDC_BITRATE_CFORMAT, CB_ADDSTRING, 0, (LPARAM)"AVI-OpenDML");
		SendDlgItemMessage(hDlg, IDC_BITRATE_CFORMAT, CB_ADDSTRING, 0, (LPARAM)"Matroska");
		SendDlgItemMessage(hDlg, IDC_BITRATE_CFORMAT, CB_ADDSTRING, 0, (LPARAM)"OGM");
		SendDlgItemMessage(hDlg, IDC_BITRATE_CFORMAT, CB_ADDSTRING, 0, (LPARAM)"(None)");

		SendDlgItemMessage(hDlg, IDC_BITRATE_TSIZE, CB_ADDSTRING, 0, (LPARAM)"665600");
		SendDlgItemMessage(hDlg, IDC_BITRATE_TSIZE, CB_ADDSTRING, 0, (LPARAM)"716800");
		SendDlgItemMessage(hDlg, IDC_BITRATE_TSIZE, CB_ADDSTRING, 0, (LPARAM)"1331200");
		SendDlgItemMessage(hDlg, IDC_BITRATE_TSIZE, CB_ADDSTRING, 0, (LPARAM)"1433600");

		for (i=0; i<sizeof(video_fps_list)/sizeof(named_float_t); i++)
			SendDlgItemMessage(hDlg, IDC_BITRATE_FPS, CB_ADDSTRING, 0, (LPARAM)video_fps_list[i].name);

		for (i=0; i<sizeof(audio_type_list)/sizeof(named_int_t); i++)
			SendDlgItemMessage(hDlg, IDC_BITRATE_AFORMAT, CB_ADDSTRING, 0, (LPARAM)audio_type_list[i].name);

		SendDlgItemMessage(hDlg, IDC_BITRATE_ARATE, CB_ADDSTRING, 0, (LPARAM)"32");
		SendDlgItemMessage(hDlg, IDC_BITRATE_ARATE, CB_ADDSTRING, 0, (LPARAM)"56");
		SendDlgItemMessage(hDlg, IDC_BITRATE_ARATE, CB_ADDSTRING, 0, (LPARAM)"64");
		SendDlgItemMessage(hDlg, IDC_BITRATE_ARATE, CB_ADDSTRING, 0, (LPARAM)"96");
		SendDlgItemMessage(hDlg, IDC_BITRATE_ARATE, CB_ADDSTRING, 0, (LPARAM)"112");
		SendDlgItemMessage(hDlg, IDC_BITRATE_ARATE, CB_ADDSTRING, 0, (LPARAM)"128");
		SendDlgItemMessage(hDlg, IDC_BITRATE_ARATE, CB_ADDSTRING, 0, (LPARAM)"160");
		SendDlgItemMessage(hDlg, IDC_BITRATE_ARATE, CB_ADDSTRING, 0, (LPARAM)"192");
		SendDlgItemMessage(hDlg, IDC_BITRATE_ARATE, CB_ADDSTRING, 0, (LPARAM)"224");
		SendDlgItemMessage(hDlg, IDC_BITRATE_ARATE, CB_ADDSTRING, 0, (LPARAM)"256");
		SendDlgItemMessage(hDlg, IDC_BITRATE_ARATE, CB_ADDSTRING, 0, (LPARAM)"384");
		SendDlgItemMessage(hDlg, IDC_BITRATE_ARATE, CB_ADDSTRING, 0, (LPARAM)"448");
		SendDlgItemMessage(hDlg, IDC_BITRATE_ARATE, CB_ADDSTRING, 0, (LPARAM)"512");
		break;

	case IDD_ZONE :
		EnableDlgWindow(hDlg, IDC_ZONE_FETCH, config->ci_valid);
		break;

	case IDD_MOTION :
		SendDlgItemMessage(hDlg, IDC_MOTION, CB_ADDSTRING, 0, (LPARAM)"0 - None");
		SendDlgItemMessage(hDlg, IDC_MOTION, CB_ADDSTRING, 0, (LPARAM)"1 - Very Low");
		SendDlgItemMessage(hDlg, IDC_MOTION, CB_ADDSTRING, 0, (LPARAM)"2 - Low");
		SendDlgItemMessage(hDlg, IDC_MOTION, CB_ADDSTRING, 0, (LPARAM)"3 - Medium");
		SendDlgItemMessage(hDlg, IDC_MOTION, CB_ADDSTRING, 0, (LPARAM)"4 - High");
		SendDlgItemMessage(hDlg, IDC_MOTION, CB_ADDSTRING, 0, (LPARAM)"5 - Very High");
		SendDlgItemMessage(hDlg, IDC_MOTION, CB_ADDSTRING, 0, (LPARAM)"6 - Ultra High");

		SendDlgItemMessage(hDlg, IDC_VHQ, CB_ADDSTRING, 0, (LPARAM)"0 - Off");
		SendDlgItemMessage(hDlg, IDC_VHQ, CB_ADDSTRING, 0, (LPARAM)"1 - Mode Decision");
		SendDlgItemMessage(hDlg, IDC_VHQ, CB_ADDSTRING, 0, (LPARAM)"2 - Limited Search");
		SendDlgItemMessage(hDlg, IDC_VHQ, CB_ADDSTRING, 0, (LPARAM)"3 - Medium Search");
		SendDlgItemMessage(hDlg, IDC_VHQ, CB_ADDSTRING, 0, (LPARAM)"4 - Wide Search");
		break;

	case IDD_DEBUG :
		/* force threads disabled */
		EnableWindow(GetDlgItem(hDlg, IDC_NUMTHREADS_STATIC), FALSE);
		EnableWindow(GetDlgItem(hDlg, IDC_NUMTHREADS), FALSE);

		SendDlgItemMessage(hDlg, IDC_FOURCC, CB_ADDSTRING, 0, (LPARAM)"XVID");
		SendDlgItemMessage(hDlg, IDC_FOURCC, CB_ADDSTRING, 0, (LPARAM)"DIVX");
		SendDlgItemMessage(hDlg, IDC_FOURCC, CB_ADDSTRING, 0, (LPARAM)"DX50");
		break;

	case IDD_DEC :
		SendDlgItemMessage(hDlg, IDC_DEC_BRIGHTNESS, TBM_SETRANGE, (WPARAM)TRUE, (LPARAM)MAKELONG(-96, 96));
		SendDlgItemMessage(hDlg, IDC_DEC_BRIGHTNESS, TBM_SETTICFREQ, (WPARAM)16, (LPARAM)0);
		break;
	}
}


/* enable/disable controls based on encoder-mode or user selection */

static void adv_mode(HWND hDlg, int idd, CONFIG * config)
{
	int profile;
	int weight_en, quant_en;
	int cpu_force;
	int custom_quant, bvops;
	int ar_mode, ar_par;

	switch(idd) {
	case IDD_PROFILE :
		profile = SendDlgItemMessage(hDlg, IDC_PROFILE_PROFILE, CB_GETCURSEL, 0, 0);
		EnableDlgWindow(hDlg, IDC_BVOP, profiles[profile].flags&PROFILE_BVOP);

		EnableDlgWindow(hDlg, IDC_QUANTTYPE_S, profiles[profile].flags&PROFILE_MPEGQUANT);
		EnableDlgWindow(hDlg, IDC_QUANTTYPE_S, profiles[profile].flags&PROFILE_MPEGQUANT);
		EnableDlgWindow(hDlg, IDC_QUANTTYPE, profiles[profile].flags&PROFILE_MPEGQUANT);
		custom_quant = (profiles[profile].flags&PROFILE_MPEGQUANT) && SendDlgItemMessage(hDlg, IDC_QUANTTYPE, CB_GETCURSEL, 0, 0)==QUANT_MODE_CUSTOM;
		EnableDlgWindow(hDlg, IDC_QUANTMATRIX, custom_quant);
		EnableDlgWindow(hDlg, IDC_LUMMASK, profiles[profile].flags&PROFILE_ADAPTQUANT);
		EnableDlgWindow(hDlg, IDC_INTERLACING, profiles[profile].flags&PROFILE_INTERLACE);
		EnableDlgWindow(hDlg, IDC_TFF, IsDlgChecked(hDlg, IDC_INTERLACING));
		EnableDlgWindow(hDlg, IDC_QPEL, profiles[profile].flags&PROFILE_QPEL);
		EnableDlgWindow(hDlg, IDC_GMC, profiles[profile].flags&PROFILE_GMC);

		bvops = (profiles[profile].flags&PROFILE_BVOP) && IsDlgChecked(hDlg, IDC_BVOP);
		EnableDlgWindow(hDlg, IDC_MAXBFRAMES,	   bvops);
		EnableDlgWindow(hDlg, IDC_BQUANTRATIO,	  bvops);
		EnableDlgWindow(hDlg, IDC_BQUANTOFFSET,	 bvops);
		EnableDlgWindow(hDlg, IDC_MAXBFRAMES_S,	 bvops);
		EnableDlgWindow(hDlg, IDC_BQUANTRATIO_S,	bvops);
		EnableDlgWindow(hDlg, IDC_BQUANTOFFSET_S,   bvops);
		EnableDlgWindow(hDlg, IDC_PACKED,		   bvops);
		break;

	case IDD_AR:
		ar_mode = IsDlgChecked(hDlg, IDC_PAR);
		EnableDlgWindow(hDlg, IDC_ASPECT_RATIO, ar_mode);

		ar_par = SendDlgItemMessage(hDlg, IDC_ASPECT_RATIO, CB_GETCURSEL, 0, 0);
		if (ar_par == 5) { /* custom par */
			SetDlgItemInt(hDlg, IDC_PARY, config->par_y, FALSE);
			SetDlgItemInt(hDlg, IDC_PARX, config->par_x, FALSE);

			EnableDlgWindow(hDlg, IDC_PARX, ar_mode);
			EnableDlgWindow(hDlg, IDC_PARY, ar_mode);
		} else {
			SetDlgItemInt(hDlg, IDC_PARX, PARS[ar_par][0], FALSE);
			SetDlgItemInt(hDlg, IDC_PARY, PARS[ar_par][1], FALSE);
			EnableDlgWindow(hDlg, IDC_PARX, FALSE);
			EnableDlgWindow(hDlg, IDC_PARY, FALSE);
		}

		ar_mode = IsDlgChecked(hDlg, IDC_AR);

		config->ar_x = config_get_uint(hDlg, IDC_ARX, config->ar_x);
		config->ar_y = config_get_uint(hDlg, IDC_ARY, config->ar_y);

		EnableDlgWindow(hDlg, IDC_ARX, ar_mode);
		EnableDlgWindow(hDlg, IDC_ARY, ar_mode);
		break;

	case IDD_LEVEL :
		profile = SendDlgItemMessage(hDlg, IDC_LEVEL_PROFILE, CB_GETCURSEL, 0, 0);
		SetDlgItemInt(hDlg, IDC_LEVEL_WIDTH, profiles[profile].width, FALSE);
		SetDlgItemInt(hDlg, IDC_LEVEL_HEIGHT, profiles[profile].height, FALSE);
		SetDlgItemInt(hDlg, IDC_LEVEL_FPS, profiles[profile].fps, FALSE);
		SetDlgItemInt(hDlg, IDC_LEVEL_VMV, profiles[profile].max_vmv_buffer_sz, FALSE);
		SetDlgItemInt(hDlg, IDC_LEVEL_VCV, profiles[profile].vcv_decoder_rate, FALSE);
		SetDlgItemInt(hDlg, IDC_LEVEL_VBV, profiles[profile].max_vbv_size, FALSE);
    set_dlgitem_float1000(hDlg, IDC_LEVEL_BITRATE, profiles[profile].max_bitrate);
    SetDlgItemInt(hDlg, IDC_LEVEL_PEAKRATE, profiles[profile].vbv_peakrate, FALSE);

    {
      int en_dim = profiles[profile].width && profiles[profile].height;
      int en_vmv = profiles[profile].max_vmv_buffer_sz;
      int en_vcv = profiles[profile].vcv_decoder_rate;
      EnableDlgWindow(hDlg, IDC_LEVEL_LEVEL_G, en_dim || en_vmv || en_vcv);
      EnableDlgWindow(hDlg, IDC_LEVEL_DIM_S, en_dim);
      EnableDlgWindow(hDlg, IDC_LEVEL_WIDTH, en_dim);
      EnableDlgWindow(hDlg, IDC_LEVEL_HEIGHT,en_dim);
      EnableDlgWindow(hDlg, IDC_LEVEL_FPS,   en_dim);
      EnableDlgWindow(hDlg, IDC_LEVEL_VMV_S, en_vmv);
      EnableDlgWindow(hDlg, IDC_LEVEL_VMV,   en_vmv);
      EnableDlgWindow(hDlg, IDC_LEVEL_VCV_S, en_vcv);
      EnableDlgWindow(hDlg, IDC_LEVEL_VCV,   en_vcv);
    }
    {
      int en_vbv = profiles[profile].max_vbv_size;
      int en_br = profiles[profile].max_bitrate;
      int en_pr = profiles[profile].vbv_peakrate;

      EnableDlgWindow(hDlg, IDC_LEVEL_VBV_G,      en_vbv || en_br || en_pr);
      EnableDlgWindow(hDlg, IDC_LEVEL_VBV_S,      en_vbv);
      EnableDlgWindow(hDlg, IDC_LEVEL_VBV,        en_vbv);
      EnableDlgWindow(hDlg, IDC_LEVEL_BITRATE_S,  en_br);
      EnableDlgWindow(hDlg, IDC_LEVEL_BITRATE,    en_br);
      EnableDlgWindow(hDlg, IDC_LEVEL_PEAKRATE_S, en_pr);
      EnableDlgWindow(hDlg, IDC_LEVEL_PEAKRATE,   en_pr);
    }
		break;

	case IDD_BITRATE :
		{
			int ctype = SendDlgItemMessage(hDlg, IDC_BITRATE_CFORMAT, CB_GETCURSEL, 0, 0);
			int target_size = config_get_cbuint(hDlg, IDC_BITRATE_TSIZE, 0);
			int subtitle_size = config_get_uint(hDlg, IDC_BITRATE_SSIZE, 0);
			int fps = SendDlgItemMessage(hDlg, IDC_BITRATE_FPS, CB_GETCURSEL, 0, 0);

			int duration = 
				3600 * config_get_uint(hDlg, IDC_BITRATE_HOURS, 0) +
				60 * config_get_uint(hDlg, IDC_BITRATE_MINUTES, 0) +
				config_get_uint(hDlg, IDC_BITRATE_SECONDS, 0);

			int audio_type = SendDlgItemMessage(hDlg, IDC_BITRATE_AFORMAT, CB_GETCURSEL, 0, 0);
			int audio_mode = IsDlgChecked(hDlg, IDC_BITRATE_AMODE_SIZE);
			int audio_rate = config_get_cbuint(hDlg, IDC_BITRATE_ARATE, 0);
			int audio_size = config_get_uint(hDlg, IDC_BITRATE_ASIZE, 0);

			int frames;
			int overhead;
			int vsize;

			if (duration == 0) 
				break;

			if (fps < 0 || fps >= sizeof(video_fps_list)/sizeof(named_float_t)) {
				fps = 0;
			}
			if (audio_type < 0 || audio_type >= sizeof(audio_type_list)/sizeof(named_int_t)) {
				audio_type = 0;
			}

			EnableDlgWindow(hDlg, IDC_BITRATE_AMODE_RATE, audio_type!=NO_AUDIO);
			EnableDlgWindow(hDlg, IDC_BITRATE_AMODE_SIZE, audio_type!=NO_AUDIO);
			EnableDlgWindow(hDlg, IDC_BITRATE_ARATE, audio_type!=NO_AUDIO && !audio_mode);
			EnableDlgWindow(hDlg, IDC_BITRATE_ASIZE, audio_type!=NO_AUDIO && audio_mode);
			EnableDlgWindow(hDlg, IDC_BITRATE_ASELECT, audio_type!=NO_AUDIO && audio_mode);

			/* step 1: calculate number of frames */
			frames = (int)(duration * video_fps_list[fps].value);
			
			/* step 2: calculate audio_size (kbytes)*/
			if (audio_type!=NO_AUDIO) {
				if (audio_mode==0) {
					int new_audio_size = (int)( (1000.0 * duration * audio_rate) / (8.0*1024) );
					
					/* this check is needed to avoid a loop */
					if (new_audio_size!=audio_size) {
						audio_size = new_audio_size;
						SetDlgItemInt(hDlg, IDC_BITRATE_ASIZE, new_audio_size, TRUE);
					}
				}else{
					int tmp_rate = (int)( (audio_size * 8.0 * 1024) / (1000.0 * duration) );
					SetDlgItemInt(hDlg, IDC_BITRATE_ARATE, tmp_rate, TRUE);
				}
			}else{
				audio_size = 0;
			}

			/* step 3: calculate container overhead */

			switch(ctype) {
			case 0 :	/* AVI */
			case 1 :	/* AVI-OpenDML */

				overhead = frames;

				if (audio_type!=NO_AUDIO) {
					overhead += (duration * 1000) / audio_type_list[audio_type].avi_interval;
				}

				overhead *= (ctype==0) ? 24 : 16;

				overhead /= 1024;
				break;

			case 2 :	/* Matroska: gknot formula */

				/* common overhead */
				overhead = 40 + 12 + 8+ 16*duration + 200 + 100*1/*one audio stream*/ + 11*duration;

				/* video overhead */
				overhead += frames*8 + (int)(frames * 4 * 0.94);

				/* cue tables and menu seek entries (300k default) */
				overhead += 300 * 1024;

				/* audio */
				overhead += (int)(duration * audio_type_list[audio_type].mkv_multiplier);

				overhead /= 1024;
				break;

			case 3 :	/* alexnoe formula */
				overhead = (int)( (target_size - subtitle_size) * (28.0/4224.0 + (1.0/255.0)) );
				break;

			default	:	/* (none) */
				overhead = 0;
				break;
			}

			SetDlgItemInt(hDlg, IDC_BITRATE_COVERHEAD, overhead, TRUE);

			/* final video bitstream size */
			vsize = target_size - subtitle_size - audio_size - overhead;
			if (vsize > 0) {
				SetDlgItemInt(hDlg, IDC_BITRATE_VSIZE, vsize, TRUE);
				/* convert from kbytes to kbits-per-second */
				SetDlgItemInt(hDlg, IDC_BITRATE_VRATE, (int)(((__int64)vsize * 8 * 128) / (duration * 125)), TRUE);
			}else{
				SetDlgItemText(hDlg, IDC_BITRATE_VSIZE, "Overflow");
				SetDlgItemText(hDlg, IDC_BITRATE_VRATE, "Overflow");
			}

		}
		break;

	case IDD_ZONE :
		weight_en = IsDlgChecked(hDlg, IDC_ZONE_MODE_WEIGHT);
		quant_en =   IsDlgChecked(hDlg, IDC_ZONE_MODE_QUANT);
		EnableDlgWindow(hDlg, IDC_ZONE_WEIGHT, weight_en);
		EnableDlgWindow(hDlg, IDC_ZONE_QUANT, quant_en);
		EnableDlgWindow(hDlg, IDC_ZONE_SLIDER, weight_en|quant_en);

		if (weight_en) {
			SendDlgItemMessage(hDlg, IDC_ZONE_SLIDER, TBM_SETRANGE, TRUE, MAKELONG(001,200));
			SendDlgItemMessage(hDlg, IDC_ZONE_SLIDER, TBM_SETPOS, TRUE, get_dlgitem_float(hDlg, IDC_ZONE_WEIGHT, 100));
			SetDlgItemText(hDlg, IDC_ZONE_MIN, "0.01");
			SetDlgItemText(hDlg, IDC_ZONE_MAX, "2.00");
		}else if (quant_en) {
			SendDlgItemMessage(hDlg, IDC_ZONE_SLIDER, TBM_SETRANGE, TRUE, MAKELONG(100,3100));
			SendDlgItemMessage(hDlg, IDC_ZONE_SLIDER, TBM_SETPOS, TRUE, get_dlgitem_float(hDlg, IDC_ZONE_QUANT, 100));
			SetDlgItemText(hDlg, IDC_ZONE_MIN, "1");
			SetDlgItemText(hDlg, IDC_ZONE_MAX, "31");
		}

		bvops = (profiles[config->profile].flags&PROFILE_BVOP) && config->use_bvop;
		EnableDlgWindow(hDlg, IDC_ZONE_BVOPTHRESHOLD_S, bvops);
		EnableDlgWindow(hDlg, IDC_ZONE_BVOPTHRESHOLD, bvops);
		break;

	case IDD_DEBUG :
		cpu_force			= IsDlgChecked(hDlg, IDC_CPU_FORCE);
		EnableDlgWindow(hDlg, IDC_CPU_MMX,		cpu_force);
		EnableDlgWindow(hDlg, IDC_CPU_MMXEXT,	cpu_force);
		EnableDlgWindow(hDlg, IDC_CPU_SSE,		cpu_force);
		EnableDlgWindow(hDlg, IDC_CPU_SSE2,		cpu_force);
		EnableDlgWindow(hDlg, IDC_CPU_3DNOW,	cpu_force);
		EnableDlgWindow(hDlg, IDC_CPU_3DNOWEXT,	cpu_force);
		break;
	}
}


/* upload config data into dialog */
static void adv_upload(HWND hDlg, int idd, CONFIG * config)
{
	switch (idd)
	{
	case IDD_PROFILE :
		SendDlgItemMessage(hDlg, IDC_PROFILE_PROFILE, CB_SETCURSEL, config->profile, 0);

		SendDlgItemMessage(hDlg, IDC_QUANTTYPE, CB_SETCURSEL, config->quant_type, 0);