//+>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>+
//| From IPFTest |
//+>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>+
bool PFTestSplitBySource::Proceed(IObject* pCont,
PreciseTimeValue timeStart,
PreciseTimeValue& timeEnd,
Object* pSystem,
INode* pNode,
INode* actionNode,
IPFIntegrator* integrator,
BitArray& testResult,
Tab<float>& testTime)
{
if (pNode == NULL) return false;
bool exactStep = IsExactIntegrationStep(timeEnd, pSystem);
int conditionType = pblock()->GetInt(kSplitBySource_conditionType, timeStart);
// acquire absolutely necessary particle channels
IParticleChannelAmountR* chAmount = GetParticleChannelAmountRInterface(pCont);
if (chAmount == NULL) return false; // can't find number of particles in the container
int count = chAmount->Count();
bool isSelectedSource = false;
int i, systemHandle = pNode->GetHandle();
for(i=0; i<pblock()->Count(kSplitBySource_sources); i++) {
if (systemHandle == pblock()->GetInt(kSplitBySource_sources, 0, i)) {
isSelectedSource = true; break;
}
}
// test all particles
testResult.SetSize(count);
testResult.ClearAll();
testTime.SetCount(count);
if (exactStep) {
if ((isSelectedSource && (conditionType == kSplitBySource_conditionType_selected))
|| (!isSelectedSource && (conditionType == kSplitBySource_conditionType_notSelected))) {
testResult.SetAll();
for(i=0; i<count; i++) testTime[i] = 0.0f;
}
}
return true;
}
//+>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>+
//| From IPFTest |
//+>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>+
bool PFTestGoToNextEvent::Proceed(IObject* pCont,
PreciseTimeValue timeStart,
PreciseTimeValue& timeEnd,
Object* pSystem,
INode* pNode,
INode* actionNode,
IPFIntegrator* integrator,
BitArray& testResult,
Tab<float>& testTime)
{
int conditionType = pblock()->GetInt(kGoToNextEvent_conditionType, timeStart);
bool exactStep = IsExactIntegrationStep(timeEnd, pSystem);
// get channel container interface
IChannelContainer* chCont;
chCont = GetChannelContainerInterface(pCont);
if (chCont == NULL) return false;
// acquire absolutely necessary particle channels
IParticleChannelAmountR* chAmount = GetParticleChannelAmountRInterface(pCont);
if (chAmount == NULL) return false; // can't find number of particles in the container
IParticleChannelPTVR* chTime = GetParticleChannelTimeRInterface(pCont);
if (chTime == NULL) return false; // can't read timing info for a particle
int count = chAmount->Count();
if (count <= 0) return true;
testResult.SetSize(count);
testTime.SetCount(count);
if((conditionType == kGoToNextEvent_conditionType_all) && exactStep) {
testResult.SetAll();
for(int i=0; i<count; i++) testTime[i] = 0.0f;
} else {
testResult.ClearAll();
}
return true;
}
//+>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>+
//| From IPFTest |
//+>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>+
bool PFTestSplitSelected::Proceed(IObject* pCont,
PreciseTimeValue timeStart,
PreciseTimeValue& timeEnd,
Object* pSystem,
INode* pNode,
INode* actionNode,
IPFIntegrator* integrator,
BitArray& testResult,
Tab<float>& testTime)
{
bool exactStep = IsExactIntegrationStep(timeEnd, pSystem);
int conditionType = pblock()->GetInt(kSplitSelected_conditionType, timeStart);
// acquire absolutely necessary particle channels
IParticleChannelAmountR* chAmount = GetParticleChannelAmountRInterface(pCont);
if (chAmount == NULL) return false; // can't find number of particles in the container
int count = chAmount->Count();
IParticleChannelPTVR* chTime = GetParticleChannelTimeRInterface(pCont);
if (chTime == NULL) return false; // can't read timing info for a particle
IParticleChannelBoolR* chSelect = GetParticleChannelSelectionRInterface(pCont);
// test all particles
testResult.SetSize(count);
testResult.ClearAll();
testTime.SetCount(count);
for(int i=0; i<count; i++)
{
bool selected = (chSelect != NULL) ? chSelect->GetValue(i) : false;
bool sendOut = ((selected && (conditionType == kSplitSelected_conditionType_selected))
|| (!selected && (conditionType == kSplitSelected_conditionType_notSelected)));
if (sendOut && exactStep) {
testResult.Set(i);
testTime[i] = 0.0f;
}
}
return true;
}
//+>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>+
//| From IPFTest |
//+>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>+
bool PFTestSpeed::Proceed(IObject* pCont,
PreciseTimeValue timeStart,
PreciseTimeValue& timeEnd,
Object* pSystem,
INode* pNode,
INode* actionNode,
IPFIntegrator* integrator,
BitArray& testResult,
Tab<float>& testTime)
{
bool exactStep = IsExactIntegrationStep(timeEnd, pSystem);
// get the constant properties of the test
int testType = pblock()->GetInt(kSpeedTest_testType, timeEnd);
int condType = pblock()->GetInt(kSpeedTest_conditionType, timeEnd);
int syncType = pblock()->GetInt(kSpeedTest_sync, timeEnd);
ParamID varParamID = (testType == kSpeedTest_testType_steering) ? kSpeedTest_angleVariation : kSpeedTest_unitVariation;
bool hasTestVariation = (pblock()->GetFloat(varParamID, 0) != 0.0f);
if (!hasTestVariation) {
Control* ctrl = pblock()->GetControllerByID(varParamID);
if (ctrl != NULL)
hasTestVariation = (ctrl->IsAnimated() != 0);
}
if (testType >= kSpeedTest_testType_whenAccels) {
hasTestVariation = false;
syncType = kSpeedTest_sync_time;
}
bool needPrevValue = (testType >= kSpeedTest_testType_accel);
// get channel container interface
IChannelContainer* chCont;
chCont = GetChannelContainerInterface(pCont);
if (chCont == NULL) return false;
// acquire absolutely necessary particle channels
IParticleChannelAmountR* chAmount = GetParticleChannelAmountRInterface(pCont);
if (chAmount == NULL) return false; // can't find number of particles in the container
int count = chAmount->Count();
if (count == 0) return true; // no particles to test
IParticleChannelPTVR* chTime = GetParticleChannelTimeRInterface(pCont);
if (chTime == NULL) return false; // can't read timing info for a particle
IParticleChannelNewR* chNew = GetParticleChannelNewRInterface(pCont);
if (chNew == NULL) return false; // can't find newly entered particles for duration calculation
IParticleChannelPoint3R* chSpeed = GetParticleChannelSpeedRInterface(pCont);
if (chSpeed == NULL) return false; // can't read speed values
// acquire more particle channels
IParticleChannelPTVR* chBirthTime = NULL;
if (syncType == kSpeedTest_sync_age && (testType < kSpeedTest_testType_whenAccels))
{
chBirthTime = GetParticleChannelBirthTimeRInterface(pCont);
if (chBirthTime == NULL) return false; // can't read particle age
}
IParticleChannelPTVR* chEventStartR = NULL;
IParticleChannelPTVW* chEventStartW = NULL;
bool initEventStart = false;
if (syncType == kSpeedTest_sync_event && (testType < kSpeedTest_testType_whenAccels)) {
chEventStartR = (IParticleChannelPTVR*)chCont->EnsureInterface(PARTICLECHANNELEVENTSTARTR_INTERFACE,
ParticleChannelPTV_Class_ID,
true, PARTICLECHANNELEVENTSTARTR_INTERFACE,
PARTICLECHANNELEVENTSTARTW_INTERFACE, false,
actionNode, NULL, &initEventStart);
if (chEventStartR == NULL) return false; // can't read event start time
if (initEventStart) {
chEventStartW = GetParticleChannelEventStartWInterface(pCont);
if (chEventStartW == NULL) return false; // can't write event start time
}
}
IParticleChannelFloatR* chRandFloatR = NULL;
IParticleChannelFloatW* chRandFloatW = NULL;
bool initRandFloat = false;
if (hasTestVariation) {
chRandFloatW = (IParticleChannelFloatW*)chCont->EnsureInterface(PARTICLECHANNELRANDFLOATW_INTERFACE,
ParticleChannelFloat_Class_ID,
true, PARTICLECHANNELRANDFLOATR_INTERFACE,
PARTICLECHANNELRANDFLOATW_INTERFACE, true,
actionNode, (Object*)this, &initRandFloat);
chRandFloatR = (IParticleChannelFloatR*)chCont->GetPrivateInterface(PARTICLECHANNELRANDFLOATR_INTERFACE, (Object*)this);
if ((chRandFloatR == NULL) || (chRandFloatW == NULL)) return false; // can't set rand float value for newly entered particles
}
IParticleChannelPoint3R* chPrevSpeedR = NULL;
IParticleChannelPoint3W* chPrevSpeedW = NULL;
IParticleChannelPTVR* chPrevTimeR = NULL;
IParticleChannelPTVW* chPrevTimeW = NULL;
bool initPrevValue = false;
if (needPrevValue) {
chPrevSpeedW = (IParticleChannelPoint3W*)chCont->EnsureInterface(PARTICLECHANNELPREVSPEEDW_INTERFACE,
ParticleChannelPoint3_Class_ID,
true, PARTICLECHANNELPREVSPEEDR_INTERFACE,
PARTICLECHANNELPREVSPEEDW_INTERFACE, true,
actionNode, (Object*)this, &initPrevValue);
chPrevSpeedR = (IParticleChannelPoint3R*)chCont->GetPrivateInterface(PARTICLECHANNELPREVSPEEDR_INTERFACE, (Object*)this);
chPrevTimeW = (IParticleChannelPTVW*)chCont->EnsureInterface(PARTICLECHANNELPREVTIMEW_INTERFACE,
ParticleChannelPTV_Class_ID,
true, PARTICLECHANNELPREVTIMER_INTERFACE,
PARTICLECHANNELPREVTIMEW_INTERFACE, true,
actionNode, (Object*)this, &initPrevValue);
chPrevTimeR = (IParticleChannelPTVR*)chCont->GetPrivateInterface(PARTICLECHANNELPREVTIMER_INTERFACE, (Object*)this);
if ((chPrevSpeedR == NULL) || (chPrevSpeedW == NULL) || (chPrevTimeR == NULL) || (chPrevTimeW == NULL)) return false;
}
// grab the rand generator for test variation
RandGenerator* randGen = randLinker().GetRandGenerator(pCont);
if (randGen == NULL) return false;
// check all particles
testResult.SetSize(count);
testResult.ClearAll();
testTime.SetCount(count);
for(int i=0; i<count; i++)
{
if (chNew->IsNew(i)) { // initialize some channels
if (initEventStart)
chEventStartW->SetValue(i, chTime->GetValue(i));
if (initRandFloat)
chRandFloatW->SetValue(i, randGen->Rand11());
}
PreciseTimeValue prevTime;
Point3 prevSpeed;
Point3 currentSpeed = chSpeed->GetValue(i);
PreciseTimeValue currentTime = chTime->GetValue(i);
if (needPrevValue) {
prevTime = chPrevTimeR->GetValue(i);
prevSpeed = chPrevSpeedR->GetValue(i);
chPrevTimeW->SetValue(i, currentTime);
chPrevSpeedW->SetValue(i, currentSpeed);
if (initPrevValue && chNew->IsNew(i))
continue; // particle just came into the event and doesn't have previous value
}
PreciseTimeValue syncTime = currentTime;
switch(syncType) {
case kSpeedTest_sync_age:
syncTime -= chBirthTime->GetValue(i);
break;
case kSpeedTest_sync_event:
syncTime -= chEventStartR->GetValue(i);
break;
}
TimeValue syncTimeTV = TimeValue(syncTime);
float testValue = 0.0f;
if (testType < kSpeedTest_testType_whenAccels) {
if (testType == kSpeedTest_testType_steering) {
testValue = GetPFFloat(pblock(), kSpeedTest_angleValue, syncTimeTV);
if (hasTestVariation)
testValue += chRandFloatR->GetValue(i)*GetPFFloat(pblock(), kSpeedTest_angleVariation, syncTimeTV);
} else {
testValue = GetPFFloat(pblock(), kSpeedTest_unitValue, syncTimeTV);
if (hasTestVariation)
testValue += chRandFloatR->GetValue(i)*GetPFFloat(pblock(), kSpeedTest_unitVariation, syncTimeTV);
}
testValue /= TIME_TICKSPERSEC;
}
float currentValue = 0.0f;
bool testSatisfied = false;
if (testType < kSpeedTest_testType_whenAccels) {
if (testType < kSpeedTest_testType_accel) {
switch(testType) {
case kSpeedTest_testType_speed:
currentValue = Length(currentSpeed);
break;
case kSpeedTest_testType_speedX:
currentValue = currentSpeed.x;
break;
case kSpeedTest_testType_speedY:
currentValue = currentSpeed.y;
break;
case kSpeedTest_testType_speedZ:
currentValue = currentSpeed.z;
break;
}
} else if (testType == kSpeedTest_testType_steering) {
float timeDif = float(currentTime - prevTime);
if (timeDif <= 0.0f) continue; // no time difference
float normFactor = Length(currentSpeed)*Length(prevSpeed);
if (normFactor <= 0.0f) continue; // steering rate is not calculatable
float vv = DotProd(currentSpeed,prevSpeed)/normFactor;
float uu = 0.0;
if (vv >= 1.0f) uu = 0.0;
else if (vv <= -1.0f) uu = PI;
else uu = acos(vv);
currentValue = uu/timeDif;
} else { // acceleration
float timeDif = float(currentTime - prevTime);
if (timeDif <= 0.0f) continue; // no time difference
Point3 curAccel;
switch(testType) {
case kSpeedTest_testType_accel:
currentValue = Length((currentSpeed - prevSpeed)/timeDif);
break;
case kSpeedTest_testType_accelX:
currentValue = (currentSpeed.x - prevSpeed.x)/timeDif;
break;
case kSpeedTest_testType_accelY:
currentValue = (currentSpeed.y - prevSpeed.y)/timeDif;
break;
case kSpeedTest_testType_accelZ:
currentValue = (currentSpeed.z - prevSpeed.z)/timeDif;
break;
}
testValue /= TIME_TICKSPERSEC; // acceleration is per second squared
}
testSatisfied = (condType == kSpeedTest_conditionType_less) ?
(currentValue < testValue) : (currentValue > testValue);
} else {
if (testType == kSpeedTest_testType_whenAccels) {
testSatisfied = (Length(currentSpeed) > Length(prevSpeed));
} else {
testSatisfied = (Length(currentSpeed) < Length(prevSpeed));
}
}
if (testSatisfied && exactStep) {
testResult.Set(i);
testTime[i] = 0.0f;
}
}
return true;
}
//+>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>+
//| From IPFTest |
//+>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>+
bool PFTestScale::Proceed(IObject* pCont,
PreciseTimeValue timeStart,
PreciseTimeValue& timeEnd,
Object* pSystem,
INode* pNode,
INode* actionNode,
IPFIntegrator* integrator,
BitArray& testResult,
Tab<float>& testTime)
{
bool exactStep = IsExactIntegrationStep(timeEnd, pSystem);
// get the constant properties of the test
int testType = pblock()->GetInt(kScaleTest_testType, timeEnd);
int axisType = pblock()->GetInt(kScaleTest_axisType, timeEnd);
int condType = pblock()->GetInt(kScaleTest_conditionType, timeEnd);
int syncType = pblock()->GetInt(kScaleTest_sync, timeEnd);
ParamID varParamID = (testType == kScaleTest_testType_scale) ? kScaleTest_scaleVariation : kScaleTest_sizeVariation;
bool hasTestVariation = (pblock()->GetFloat(varParamID, 0) != 0.0f);
if (!hasTestVariation) {
Control* ctrl = pblock()->GetControllerByID(varParamID);
if (ctrl != NULL)
hasTestVariation = (ctrl->IsAnimated() != 0);
}
// get channel container interface
IChannelContainer* chCont;
chCont = GetChannelContainerInterface(pCont);
if (chCont == NULL) return false;
// acquire absolutely necessary particle channels
IParticleChannelAmountR* chAmount = GetParticleChannelAmountRInterface(pCont);
if (chAmount == NULL) return false; // can't find number of particles in the container
int count = chAmount->Count();
if (count == 0) return true; // no particles to test
IParticleChannelPTVR* chTime = GetParticleChannelTimeRInterface(pCont);
if (chTime == NULL) return false; // can't read timing info for a particle
IParticleChannelNewR* chNew = GetParticleChannelNewRInterface(pCont);
if (chNew == NULL) return false; // can't find newly entered particles for duration calculation
// acquire more particle channels
IParticleChannelPTVR* chBirthTime = NULL;
if (syncType == kScaleTest_sync_age)
{
chBirthTime = GetParticleChannelBirthTimeRInterface(pCont);
if (chBirthTime == NULL) return false; // can't read particle age
}
IParticleChannelPTVR* chEventStartR = NULL;
IParticleChannelPTVW* chEventStartW = NULL;
bool initEventStart = false;
if (syncType == kScaleTest_sync_event) {
chEventStartR = (IParticleChannelPTVR*)chCont->EnsureInterface(PARTICLECHANNELEVENTSTARTR_INTERFACE,
ParticleChannelPTV_Class_ID,
true, PARTICLECHANNELEVENTSTARTR_INTERFACE,
PARTICLECHANNELEVENTSTARTW_INTERFACE, false,
actionNode, NULL, &initEventStart);
if (chEventStartR == NULL) return false; // can't read event start time
if (initEventStart) {
chEventStartW = GetParticleChannelEventStartWInterface(pCont);
if (chEventStartW == NULL) return false; // can't write event start time
}
}
IParticleChannelMeshR* chShape = NULL;
if (testType != kScaleTest_testType_scale) {
chShape = GetParticleChannelShapeRInterface(pCont);
if (chShape == NULL) return false; // can't read particle shape to find bounding box
}
IParticleChannelPoint3R* chScale = NULL;
if (testType != kScaleTest_testType_preSize) {
chScale = GetParticleChannelScaleRInterface(pCont);
if (chScale == NULL) return false; // can't read particle scale
}
IParticleChannelFloatR* chRandFloatR = NULL;
IParticleChannelFloatW* chRandFloatW = NULL;
bool initRandFloat = false;
if (hasTestVariation) {
chRandFloatW = (IParticleChannelFloatW*)chCont->EnsureInterface(PARTICLECHANNELRANDFLOATW_INTERFACE,
ParticleChannelFloat_Class_ID,
true, PARTICLECHANNELRANDFLOATR_INTERFACE,
PARTICLECHANNELRANDFLOATW_INTERFACE, true,
actionNode, (Object*)this, &initRandFloat);
chRandFloatR = (IParticleChannelFloatR*)chCont->GetPrivateInterface(PARTICLECHANNELRANDFLOATR_INTERFACE, (Object*)this);
if ((chRandFloatR == NULL) || (chRandFloatW == NULL)) return false; // can't set rand float value for newly entered particles
}
// grab the rand generator for test variation
RandGenerator* randGen = randLinker().GetRandGenerator(pCont);
if (randGen == NULL) return false;
// check all particles
testResult.SetSize(count);
testResult.ClearAll();
testTime.SetCount(count);
for(int i=0; i<count; i++)
{
if (chNew->IsNew(i)) { // initialize some channels
if (initEventStart)
chEventStartW->SetValue(i, chTime->GetValue(i));
if (initRandFloat)
chRandFloatW->SetValue(i, randGen->Rand11());
}
PreciseTimeValue syncTime = chTime->GetValue(i);
switch(syncType) {
case kScaleTest_sync_age:
syncTime -= chBirthTime->GetValue(i);
break;
case kScaleTest_sync_event:
syncTime -= chEventStartR->GetValue(i);
break;
}
TimeValue syncTimeTV = TimeValue(syncTime);
float testValue = 0.0f;
if (testType == kScaleTest_testType_scale) {
testValue = GetPFFloat(pblock(), kScaleTest_scaleValue, syncTimeTV);
if (hasTestVariation)
testValue += chRandFloatR->GetValue(i)*GetPFFloat(pblock(), kScaleTest_scaleVariation, syncTimeTV);
} else {
testValue = GetPFFloat(pblock(), kScaleTest_sizeValue, syncTimeTV);
if (hasTestVariation)
testValue += chRandFloatR->GetValue(i)*GetPFFloat(pblock(), kScaleTest_sizeVariation, syncTimeTV);
}
Point3 cur3DValue;
if (testType == kScaleTest_testType_scale) {
cur3DValue = chScale->GetValue(i);
} else {
Mesh* curMesh = const_cast <Mesh*>(chShape->GetValue(i));
if (curMesh == NULL) continue;
Box3 curBox = curMesh->getBoundingBox();
cur3DValue = curBox.pmax - curBox.pmin;
if (testType == kScaleTest_testType_postSize)
cur3DValue *= chScale->GetValue(i);
}
float currentValue = 0.0f;
switch(axisType) {
case kScaleTest_axisType_average:
currentValue = (cur3DValue.x + cur3DValue.y + cur3DValue.z)/3.0f;
break;
case kScaleTest_axisType_minimum:
currentValue = min(cur3DValue.x, min(cur3DValue.y, cur3DValue.z));
break;
case kScaleTest_axisType_median:
currentValue = max(min(cur3DValue.x, cur3DValue.y),
max(min(cur3DValue.x, cur3DValue.z), min(cur3DValue.y, cur3DValue.z)));
break;
case kScaleTest_axisType_maximum:
currentValue = max(cur3DValue.x, max(cur3DValue.y, cur3DValue.z));
break;
case kScaleTest_axisType_x:
currentValue = cur3DValue.x;
break;
case kScaleTest_axisType_y:
currentValue = cur3DValue.y;
break;
case kScaleTest_axisType_z:
currentValue = cur3DValue.z;
break;
}
bool testSatisfied = (condType == kScaleTest_conditionType_less) ?
(currentValue < testValue) : (currentValue > testValue);
if (testSatisfied && exactStep) {
testResult.Set(i);
testTime[i] = 0.0f;
}
}
return true;
}
//+>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>+
//| From IPFTest |
//+>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>+
bool PFTestSplitByAmount::Proceed(IObject* pCont,
PreciseTimeValue timeStart,
PreciseTimeValue& timeEnd,
Object* pSystem,
INode* pNode,
INode* actionNode,
IPFIntegrator* integrator,
BitArray& testResult,
Tab<float>& testTime)
{
int contIndex;
if (!hasParticleContainer(pCont, contIndex)) return false;
_lastUpdate(contIndex) = timeEnd.TimeValue();
bool exactStep = IsExactIntegrationStep(timeEnd, pSystem);
// update all other systems to the current time; everybody should be in sync
// for proper accumulation amounts
int i;
for(i=0; i<allParticleContainers().Count(); i++) {
if (allParticleContainer(i) == pCont) continue;
if (allSystemNode(i) == pNode) continue;
if (lastUpdate(i) == timeEnd.TimeValue()) continue;
TimeValue timeToUpdateTo = timeEnd.TimeValue();
allSystemNode(i)->NotifyDependents(FOREVER, PartID(&timeToUpdateTo), kPFMSG_UpdateToTime, NOTIFY_ALL, TRUE );
}
// get channel container interface
IChannelContainer* chCont;
chCont = GetChannelContainerInterface(pCont);
if (chCont == NULL) return false;
// acquire absolutely necessary particle channels
IParticleChannelAmountR* chAmount = GetParticleChannelAmountRInterface(pCont);
if (chAmount == NULL) return false; // can't find number of particles in the container
IParticleChannelNewR* chNew = GetParticleChannelNewRInterface(pCont);
if (chNew == NULL) return false; // can't find "new" property of particles in the container
// acquire TestSplitByAmount private particle channel; if not present then create it
IParticleChannelBoolW* chTestW = (IParticleChannelBoolW*)chCont->EnsureInterface(PARTICLECHANNELTESTSPLITBYAMOUNTW_INTERFACE,
ParticleChannelBool_Class_ID,
true, PARTICLECHANNELTESTSPLITBYAMOUNTR_INTERFACE,
PARTICLECHANNELTESTSPLITBYAMOUNTW_INTERFACE, false,
actionNode, (Object*)this);
IParticleChannelBoolR* chTestR = (IParticleChannelBoolR*)chCont->GetPrivateInterface(PARTICLECHANNELTESTSPLITBYAMOUNTR_INTERFACE, (Object*)this);
if ((chTestR == NULL) || (chTestW == NULL)) return false; // can't set test value for newly entered particles
int count = chAmount->Count();
// check if all particles are "old". If some particles are "new" then we
// have to calculate test values for those.
if (!chNew->IsAllOld())
{
RandGenerator* randGen = randLinker().GetRandGenerator(pCont);
if (randGen == NULL) return false;
int testType = pblock()->GetInt(kSplitByAmount_testType, timeStart);
float fraction = GetPFFloat(pblock(), kSplitByAmount_fraction, timeStart);
int everyN = GetPFInt(pblock(), kSplitByAmount_everyN, timeStart);
int firstN = pblock()->GetInt(kSplitByAmount_firstN, timeStart);
bool perSource = (pblock()->GetInt(kSplitByAmount_perSource, timeStart) != 0);
int curWentThru = perSource ? wentThruTotal(pNode) : wentThruTotal();
// number of "first N" particles is adjusted by multiplier coefficient
// of the master particle system. This is done to make "first N"
// parameter to be consistent to "total" number of particles acclaimed
// by a birth operator
IPFSystem* pfSys = PFSystemInterface(pSystem);
if (pfSys == NULL) return false; // no handle for PFSystem interface
firstN *= pfSys->GetMultiplier(timeStart);
for(i=0; i<count; i++) {
if (chNew->IsNew(i)) { // calculate test value only for new particles
bool sendOut = false;
switch(testType) {
case kSplitByAmount_testType_fraction:
sendOut = (randGen->Rand01() <= fraction);
break;
case kSplitByAmount_testType_everyN:
_wentThruAccum(contIndex) += 1;
if (wentThruAccum(contIndex) >= everyN) {
sendOut = true;
_wentThruAccum(contIndex) = 0;
}
break;
case kSplitByAmount_testType_firstN:
_wentThruTotal(contIndex) += 1;
if (curWentThru++ < firstN) sendOut = true;
break;
case kSplitByAmount_testType_afterFirstN:
_wentThruTotal(contIndex) += 1;
if (curWentThru++ >= firstN) sendOut = true;
break;
}
chTestW->SetValue(i, sendOut);
}
}
}
// check all particles by predefined test channel
testResult.SetSize(count);
testResult.ClearAll();
testTime.SetCount(count);
if (exactStep) {
for(i=0; i<count; i++)
{
if (chTestR->GetValue(i)) {
testResult.Set(i);
testTime[i] = 0.0f;
}
}
}
return true;
}
