Методы контекстного моделирования

struct ContextModel{
		int	esc, TotFr;
		int	count[256];
	};

	ContextModel cm[257];

ContextModel *stack[2];
	int	SP, context [1]; //контекст вырождается в 1 символ

void init_model (void){
		/*Так как cm является глобальной переменной, то значения
		всех полей равны 0. Нам требуется только распределить
		кодовое пространство в КМ(0) так, чтобы все символы, 
		включая символ ухода, всегда бы имели ненулевые оценки. 
		Пусть также символы будут равновероятными
		*/
		
		for ( int j = 0; j < 256; j++ )
			cm[256].count[j] = 1;
	
		cm[256].TotFr = 256;
	
		/*Явно запишем, что в начале моделирования мы считаем 
		контекст равным 0. Число не имеет значения, лишь бы 
		кодер и декодер точно следовали принятым 
		соглашениям. Обратите на это внимание
		*/
		
		context [0] = 0;
		SP = 0; 
	}

const int MAX_TotFr = 0x3fff;
	
	void rescale (ContextModel *CM){
		CM->TotFr = 0;
	
		for (int i = 0; i < 256; i++){
			/*обеспечим отличие от нуля значения
			счетчика после масштабирования
			*/
			CM->count[i] -= CM->count[i] >> 1;
			CM->TotFr += CM->count[i];
		}
	}

	void update_model (int c){
		while (SP) {
			SP--;
			
			if ((stack[SP]->TotFr + stack[SP]->esc) >= MAX_TotFr) 
				rescale (stack[SP]);
			
			if (!stack[SP]->count[c])
				/*в этом контексте это новый символ, увеличим 
				счетчик уходов
				*/
			stack[SP]->esc += 1;
		
			stack[SP]->count[c] += 1;
			stack[SP]->TotFr += 1;
		}
	}

int encode_sym (ContextModel *CM, int c){
		// КМ потребует инкремента счетчиков, запомним ее
		stack [SP++] = CM;	 

		if (CM->count[c]){
			/*счетчик сжимаемого символа не равен нулю, тогда
			его можно оценить в текущей КМ; найдем 
			накопленную частоту предыдущего в массиве count 
			символа
			*/	
			
			int CumFreqUnder = 0;
			for (int i = 0; i < c; i++)
				CumFreqUnder += CM->count[i];
			
			/*передадим описание кодового пространства, 
			занимаемого символом c, арифметическому кодеру 
			*/
		
			AC.encode (CumFreqUnder, CM->count[c], CM->TotFr + CM->esc);
			
			return 1;	// возвращаемся в encode с победой
		}else{
			/*нужно уходить на КМ(0);
			если текущий контекст 1-го порядка встретился первый
			раз, то заранее известно, что его КМ пуста (все
			счетчики равны нулю), и кодировать уход не только не 
			имеет смысла, но и нельзя, т.к. TotFr+esc = 0
			*/
		
			if (CM->esc)
				AC.encode (CM->TotFr, CM->esc, CM->TotFr + CM->esc);
			
			return 0; // закодировать символ не удалось
		}
	}

	void encode (void){
		int	c,	// текущий символ
		success;	// успешность кодирования символа в КМ
		
		init_model ();
		AC.StartEncode ();		// проинициализируем арифм. кодер
		
		while (( c = DataFile.ReadSymbol() ) != EOF) {
			// попробуем закодировать в КМ(1)
			success = encode_sym (&cm[context[0]], c);
		
			if (!success)
				/*уходим на КМ(0), где любой символ получит 
				ненулевую оценку и будет закодирован
				*/
				encode_sym (&cm[256], c);
			
			update_model (c);
			context [0] = c; // сдвинем контекст
		}
		
		// закодируем знак конца файла символом ухода с КМ(0)	
		AC.encode (cm[context[0]].TotFr, cm[context[0]].esc, 
		cm[context[0]].TotFr + cm[context[0]].esc);
	
		AC.encode (cm[256].TotFr, cm[256].esc, 
		cm[256].TotFr + cm[256].esc);
		
		// завершим работу арифметического кодера
		AC.FinishEncode();
	}

int decode_sym (ContextModel *CM, int *c){
		stack [SP++] = CM;
		if (!CM->esc) return 0;

		int cum_freq = AC.get_freq (CM->TotFr + CM->esc);
		if (cum_freq < CM->TotFr){
			/*символ был закодирован в этой КМ; найдем символ и 
			его точное кодовое пространство
			*/
			
			int CumFreqUnder = 0;
			int i = 0;		
			
			for (;;){
				if ( (CumFreqUnder + CM->count[i]) <= cum_freq)
					CumFreqUnder += CM->count[i];
				else break;
			
				i++;
			}
			
			/*обновим  состояние арифметического кодера на 
			основании точной накопленной частоты символа
			*/
			
			AC.decode_update (CumFreqUnder, CM->count[i], CM->TotFr + CM->esc);
			
			*c = i;
			return 1;
		}else{
			/*обновим  состояние арифметического кодера на 
			основании точной накопленной частоты символа, 
			оказавшегося символом ухода
			*/
		
			AC.decode_update (CM->TotFr, CM->esc, CM->TotFr + CM->esc);
			return 0;
		}
	}

	void decode (void){
		int	c, success;
		
		init_model ();
		AC.StartDecode ();
		
		for (;;){
			success = decode_sym (&cm[context[0]], &c);
			if (!success){
				success = decode_sym (&cm[256], &c);
				
				if (!success) break; //признак конца файла
			}
			
			update_model (c);
			context [0] = c;
			DataFile.WriteSymbol (c);
		}
	}