C++ Patron::getTable方法代码示例

void Restaurant::updateSeatingForDpp(void) {

	for (int idx=0; idx<numPatrons; idx++)
		{
		// get a pointer to the element and its current table
		Patron *p = patrons[idx];
		Table *tablePtr = p->getTable();

		// remove the element from the table
		tablePtr->removePatronFromTable(p);
		if (tablePtr->getNumPatrons() == 0)
			deleteTable(tablePtr);
		
		// calculate the likelihood when the element is seated at all possible tables
		vector<double> probs;
		for (int i=0; i<getNumTables(); i++)
			{
			Table *tp = getTable(i);
			probs.push_back( tp->lnLikelihood(p) + log(tp->getNumPatrons()) );
			}
		probs.push_back( p->getLnLikelihoodalone() + log(concentrationParm) );
		normalizeVector( probs );

		// pick a table for reassignment
		int whichTable = multinomialRv(probs);
			
		// add the element to that table
		if (whichTable < getNumTables())
			{
			tablePtr = getTable(whichTable);
			tablePtr->addPatronToTable(p);
			}
		else
			{
			tablePtr = new Table(this, ranPtr, observationsPtr, observationsPtr->getNumUniqueAlleles(), lnFactorial);
			addTableToRestaurant( tablePtr );
			tablePtr->addPatronToTable(p);
			}
		}
}

void Restaurant::updateSeatingForFixedTables(void) {

	// allocate memory for reassignment probabilities
	vector<double> probs;
	
	// loop over all patrons, attempting to reseat each
	for (int idx=0; idx<numPatrons; idx++)
		{
		// get a pointer to the element and its current table
		Patron *p = patrons[idx];
		Table *tablePtr = p->getTable();

		// remove the element from the table
		tablePtr->removePatronFromTable(p);
		if (assumingAdmixture == true)
			popCounts[ p->getIndividual() ][ getIndexForTable(tablePtr) ]--;
		
		// calculate the likelihood when the element is seated at all possible tables
		for (int i=0; i<getNumTables(); i++)
			{
			Table *tp = getTable(i);
			double lnC = 0.0;
			if (assumingAdmixture == true)
				lnC = log( popProbs[ p->getIndividual() ][ i ] );
			probs.push_back( tp->lnLikelihood(p) + lnC );
			}
		normalizeVector( probs );

		// pick a table for reassignment
		tablePtr = getTable( multinomialRv(probs) );
			
		// add the element to that table
		tablePtr->addPatronToTable(p);
		if (assumingAdmixture == true)
			popCounts[ p->getIndividual() ][ getIndexForTable(tablePtr) ]++;
			
		// clear the probs vector
		probs.clear();
		}
}

double Restaurant::lnPriorProbability(void) {

	if (assumingAdmixture == false && assumingDpp == false)
		{
		// fixed number of populations with no admixture
		return -numPatrons * log(getNumTables());
		}
	else if (assumingAdmixture == true && assumingDpp == false)
		{
		// fixed number of poulations with admixture
		double lnP = 0.0;
		for (int idx=0; idx<numPatrons; idx++)
			{
			Patron *p = patrons[idx];
			Table *tablePtr = p->getTable();
			lnP += log(popProbs[ p->getIndividual() ][ getIndexForTable(tablePtr) ]);
			}
		return lnP;
		}
	else if (assumingAdmixture == false && assumingDpp == true)
		{
		// number of populations is a random variable, but no admixture
		/* f({\mathbf z}, k | \alpha, c) = \alpha^k {\prod_{i=1}^k(\eta_i - 1)! \over \prod_{i=1}^c(\alpha + i - 1)} */
		double lnSum1 = 0.0;
		for (vector<Table *>::iterator p=tables.begin(); p != tables.end(); p++)
			{
			lnSum1 += lnFactorial[ (*p)->getNumPatrons() - 1 ];
			}
		double lnSum2 = 0.0;
		for (int i=1; i<=numPatrons; i++)
			lnSum2 += log(concentrationParm + i - 1.0);
		double lnP = getNumTables() * log(concentrationParm) + lnSum1 - lnSum2;
		return lnP;
		}
	return 0.0;
}

void Franchise::updateSeating(void) {

	// loop over the restaurants
	for (vector<Restaurant *>::iterator r=restaurants.begin(); r != restaurants.end(); r++)
		{
		// loop over the patrons in a restaurant
		for (int idx=0; idx<(*r)->getNumPatronsInRestaurant(); idx++)
			{
			// get a pointer to the patron and its current table
			Patron *p = (*r)->getRestaurantPatron(idx);
			Table *tablePtr = p->getTable();
			Menu *menuItemPtr = tablePtr->getMenuItem();

			// remove the element from the table
			tablePtr->removePatronFromTable(p);
			if (tablePtr->getNumPatrons() == 0)
				{
				menuItemPtr->deleteTableFromMenu(tablePtr);
				if (menuItemPtr->getNumTables() == 0)
					deleteMenuItem(menuItemPtr);
				(*r)->deleteTable(tablePtr);
				}
				
			// calculate the probability of reseating patron at each of the tables
			int n = (*r)->getNumTables() + 1;
			vector<double> probs;
			for (int i=0; i<(*r)->getNumTables(); i++)
				{
				Table *t = (*r)->getTable(i);
				Menu *m = t->getMenuItem();
				int n_jt = t->getNumPatrons();
				probs.push_back( m->lnLikelihood(p) + log(n_jt) );
				}
			vector<double> lnProbsNew;
			probs.push_back( log(restaurantAlpha) + lnLikelihoodNewTable(p, lnProbsNew) );
			(*r)->normalizeVector(probs);
			
			// pick a table
			int whichTable = (*r)->multinomialRv( probs );
			Table *newTablePtr = NULL;
			if  (whichTable < n-1)
				newTablePtr = (*r)->getTable(whichTable);
			
			// seat the patron
			if (newTablePtr == NULL)
				{
				// we are seating the patron at a new table, which may have an already used menu item, or a brand new menu item
				Table *newRestaurantTablePtr = new Table( (*r), ranPtr, observationsPtr, observationsPtr->getNumUniqueAlleles(), (*r)->getFactorialPtr() );
				newRestaurantTablePtr->addPatronToTable( p );
				(*r)->addTableToRestaurant( newRestaurantTablePtr );
				(*r)->normalizeVector(lnProbsNew);
				whichTable = (*r)->multinomialRv( lnProbsNew );
				Menu *newMenuItemPtr = NULL;
				if (whichTable < lnProbsNew.size()-1)
					{
					newMenuItemPtr = menus[whichTable];
					}
				else
					{
					newMenuItemPtr = new Menu( this, ranPtr, observationsPtr, observationsPtr->getNumUniqueAlleles(), (*r)->getFactorialPtr() );
					menus.push_back( newMenuItemPtr );
					}
				newMenuItemPtr->addTableToMenu( newRestaurantTablePtr );
				newRestaurantTablePtr->setMenuItem(newMenuItemPtr);
				}
			else
				{
				// otherwise, we are seating the element at one of the pre-existing tables
				newTablePtr->addPatronToTable( p );
				}
			
		//print();
		//getchar();
			}
		}
}