int number_of_observations() const override { return dat().size(); }
bool CScriptFunctionData::_readData(int stack,const int* expectedArguments,int requiredArgumentCount,const char* functionName,const char* argumentText1,const char* argumentText2,std::vector<CScriptFunctionDataItem>& inOutData)
{ // use this when reading data from a script from inside of a custom script function call
inOutData.clear();
int argCnt=simGetStackSize(stack);
if (argCnt<requiredArgumentCount)
{
std::ostringstream str;
str << "Not enough " << argumentText1;
simSetLastError(functionName,str.str().c_str());
return(false);
}
for (int i=0;i<argCnt;i++)
{
if (i>=expectedArguments[0])
break;
bool done=false;
simMoveStackItemToTop(stack,0);
if (simIsStackValueNull(stack)==1)
{
// is nil explicitely allowed?
if (expectedArguments[1+i*2+0]&SIM_SCRIPT_ARG_NULL_ALLOWED)
{ // yes. This is for an argument that can optionally also be nil.
CScriptFunctionDataItem dat;
inOutData.push_back(dat);
done=true;
}
else
{ // no
if (int(inOutData.size())<requiredArgumentCount)
{
std::ostringstream str;
str << argumentText2 << i+1 << " is not correct.";
simSetLastError(functionName,str.str().c_str());
return(false);
}
break; // this argument is nil, so it is like inexistant. But we also won't explore any more arguments, we have enough.
}
}
if (!done)
{
int tableSize=simGetStackTableInfo(stack,0);
bool expectingATable=(((expectedArguments[1+i*2+0]|SIM_SCRIPT_ARG_NULL_ALLOWED)-SIM_SCRIPT_ARG_NULL_ALLOWED)&sim_script_arg_table)!=0;
if ( (tableSize>=0)!=expectingATable )
{
std::ostringstream str;
str << argumentText2 << i+1 << " is not correct.";
simSetLastError(functionName,str.str().c_str());
return(false);
}
int expectingType=(((expectedArguments[1+i*2+0]|SIM_SCRIPT_ARG_NULL_ALLOWED)-SIM_SCRIPT_ARG_NULL_ALLOWED)|sim_script_arg_table)-sim_script_arg_table;
if (expectingATable)
{ // we have a table
int infoType=0;
if (expectingType==sim_script_arg_null)
infoType=1;
if (expectingType==sim_script_arg_bool)
infoType=3;
if (expectingType==sim_script_arg_float)
infoType=2;
if (expectingType==sim_script_arg_int32)
infoType=2;
if (expectingType==sim_script_arg_string)
infoType=4;
if (expectingType==sim_script_arg_charbuff)
infoType=4;
if (expectingType==sim_script_arg_double)
infoType=2;
if (simGetStackTableInfo(stack,infoType)<1)
{ // table content cannot be converted
std::ostringstream str;
str << argumentText2 << i+1 << " is not correct.";
simSetLastError(functionName,str.str().c_str());
return(false);
}
if ( (tableSize<expectedArguments[1+i*2+1])&&(expectedArguments[1+i*2+1]!=0) )
{
std::ostringstream str;
str << argumentText2 << i+1 << " is not correct (wrong table size).";
simSetLastError(functionName,str.str().c_str());
return(false);
}
else
{
int t=expectingType;
int itemCnt=tableSize;
if (t==sim_script_arg_null)
{
CScriptFunctionDataItem* a=new CScriptFunctionDataItem();
a->setNilTable(itemCnt);
CScriptFunctionDataItem dat;
dat.setNilTable(itemCnt);
inOutData.push_back(dat);
}
if (t==sim_script_arg_bool)
{
std::vector<bool> vect;
simUnfoldStackTable(stack); // this removes the table and exposes the inside
for (int j=0;j<itemCnt;j++)
{
simBool val;
simGetStackBoolValue(stack,&val);
vect.insert(vect.begin(),val!=0);
simPopStackItem(stack,2);
}
simPushTableOntoStack(stack); // empty table, will be removed at the end of the loop
CScriptFunctionDataItem dat(vect);
inOutData.push_back(dat);
}
if (t==sim_script_arg_int32)
{
std::vector<int> vect;
if (itemCnt>0)
{
vect.resize(itemCnt);
simGetStackInt32Table(stack,&vect[0],itemCnt);
}
CScriptFunctionDataItem dat(vect);
inOutData.push_back(dat);
}
if (t==sim_script_arg_float)
{
std::vector<float> vect;
if (itemCnt>0)
{
vect.resize(itemCnt);
simGetStackFloatTable(stack,&vect[0],itemCnt);
}
CScriptFunctionDataItem dat(vect);
inOutData.push_back(dat);
}
if (t==sim_script_arg_double)
{
std::vector<double> vect;
if (itemCnt>0)
{
vect.resize(itemCnt);
simGetStackDoubleTable(stack,&vect[0],itemCnt);
}
CScriptFunctionDataItem dat(vect);
inOutData.push_back(dat);
}
if (t==sim_script_arg_string)
{
std::vector<std::string> vect;
simUnfoldStackTable(stack); // this removes the table and exposes the inside
for (int j=0;j<itemCnt;j++)
{
int l;
char* str=simGetStackStringValue(stack,&l);
std::string str2(str); // treat it as a char string, not buffer
simReleaseBuffer(str);
vect.insert(vect.begin(),str2);
simPopStackItem(stack,2);
}
simPushTableOntoStack(stack); // empty table, will be removed at the end of the loop
CScriptFunctionDataItem dat(vect);
inOutData.push_back(dat);
}
if (t==sim_script_arg_charbuff)
{
std::ostringstream str;
str << argumentText2 << i+1 << " cannot be a table.";
simSetLastError(functionName,str.str().c_str());
return(false);
}
}
}
else
{ // we do not have a table
int t=expectingType;
bool failedMsgAndLeave=false;
if (t==sim_script_arg_null)
{
if (simIsStackValueNull(stack)>0)
{
CScriptFunctionDataItem dat;
inOutData.push_back(dat);
}
else
failedMsgAndLeave=true;
}
if (t==sim_script_arg_bool)
{
simBool val=0;
if (simGetStackBoolValue(stack,&val)==1)
{
CScriptFunctionDataItem dat(val!=0);
inOutData.push_back(dat);
}
else
failedMsgAndLeave=true;
}
if (t==sim_script_arg_int32)
{
int val=0;
if (simGetStackInt32Value(stack,&val)==1)
{
CScriptFunctionDataItem dat(val);
inOutData.push_back(dat);
}
else
failedMsgAndLeave=true;
}
if (t==sim_script_arg_float)
{
float val=0.0;
if (simGetStackFloatValue(stack,&val)==1)
{
CScriptFunctionDataItem dat(val);
inOutData.push_back(dat);
}
else
failedMsgAndLeave=true;
}
if (t==sim_script_arg_double)
{
double val=0.0;
if (simGetStackDoubleValue(stack,&val)==1)
{
CScriptFunctionDataItem dat(val);
inOutData.push_back(dat);
}
else
failedMsgAndLeave=true;
}
if (t==sim_script_arg_string)
{
int l;
char* str=simGetStackStringValue(stack,&l);
if (str!=NULL)
{
std::string str2(str);
simReleaseBuffer(str);
CScriptFunctionDataItem dat(str2); // treat it as a char string, not buffer
inOutData.push_back(dat);
}
else
failedMsgAndLeave=true;
}
if (t==sim_script_arg_charbuff)
{
int l;
char* str=simGetStackStringValue(stack,&l);
if (str!=NULL)
{
if ( (l<expectedArguments[1+i*2+1])&&(expectedArguments[1+i*2+1]!=0) )
{
simReleaseBuffer(str);
std::ostringstream str;
str << argumentText2 << i+1 << " is not correct (wrong buffer size).";
simSetLastError(functionName,str.str().c_str());
return(false);
}
else
{
CScriptFunctionDataItem dat(str,l);
inOutData.push_back(dat);
simReleaseBuffer(str);
}
}
else
failedMsgAndLeave=true;
}
if (failedMsgAndLeave)
{
std::ostringstream str;
str << argumentText2 << i+1 << " is not correct.";
simSetLastError(functionName,str.str().c_str());
return(false);
}
}
}
simPopStackItem(stack,1);
}
return(true);
}
//----------------------------------------------------------------------
void AggregatedRegressionModel::distribute_group_totals(){
for(int i = 0; i < dat().size(); ++i){
dat()[i]->distribute_total(model_->Beta(), model_->sigma());
}
refresh_suf();
}
