/**** Local functions definitions. ****/
static TA_RetCode rangeTestFunction( TA_Integer startIdx,
TA_Integer endIdx,
TA_Real *outputBuffer,
TA_Integer *outputBufferInt,
TA_Integer *outBegIdx,
TA_Integer *outNbElement,
TA_Integer *lookback,
void *opaqueData,
unsigned int outputNb,
unsigned int *isOutputInteger )
{
TA_RetCode retCode;
TA_RangeTestParam *testParam;
(void)outputNb;
(void)outputBufferInt;
*isOutputInteger = 0;
testParam = (TA_RangeTestParam *)opaqueData;
if( testParam->test->doAverage )
{
retCode = TA_ATR(
startIdx,
endIdx,
testParam->high,
testParam->low,
testParam->close,
testParam->test->optInTimePeriod,
outBegIdx,
outNbElement,
outputBuffer );
*lookback = TA_ATR_Lookback( testParam->test->optInTimePeriod );
}
else
{
retCode = TA_TRANGE(
startIdx,
endIdx,
testParam->high,
testParam->low,
testParam->close,
outBegIdx,
outNbElement,
outputBuffer );
*lookback = TA_TRANGE_Lookback();
}
return retCode;
}
static ErrorNumber do_test( const TA_History *history,
const TA_Test *test )
{
TA_RetCode retCode;
ErrorNumber errNb;
TA_Integer outBegIdx;
TA_Integer outNbElement;
TA_RangeTestParam testParam;
/* Set to NAN all the elements of the gBuffers. */
clearAllBuffers();
/* Build the input. */
setInputBuffer( 0, history->high, history->nbBars );
setInputBuffer( 1, history->low, history->nbBars );
setInputBuffer( 2, history->close, history->nbBars );
if( test->doAverage )
{
TA_SetUnstablePeriod( TA_FUNC_UNST_ATR, test->unstablePeriod );
retCode = TA_ATR( test->startIdx,
test->endIdx,
gBuffer[0].in,
gBuffer[1].in,
gBuffer[2].in,
test->optInTimePeriod_0,
&outBegIdx,
&outNbElement,
gBuffer[0].out0 );
}
else
{
retCode = TA_TRANGE( test->startIdx,
test->endIdx,
gBuffer[0].in,
gBuffer[1].in,
gBuffer[2].in,
&outBegIdx,
&outNbElement,
gBuffer[0].out0 );
}
errNb = checkDataSame( gBuffer[0].in, history->high,history->nbBars );
if( errNb != TA_TEST_PASS )
return errNb;
errNb = checkDataSame( gBuffer[1].in, history->low, history->nbBars );
if( errNb != TA_TEST_PASS )
return errNb;
errNb = checkDataSame( gBuffer[2].in, history->close, history->nbBars );
if( errNb != TA_TEST_PASS )
return errNb;
errNb = checkExpectedValue( gBuffer[0].out0,
retCode, test->expectedRetCode,
outBegIdx, test->expectedBegIdx,
outNbElement, test->expectedNbElement,
test->oneOfTheExpectedOutReal,
test->oneOfTheExpectedOutRealIndex );
if( errNb != TA_TEST_PASS )
return errNb;
outBegIdx = outNbElement = 0;
/* Make another call where the input and the output are the
* same buffer.
*/
if( test->doAverage )
{
TA_SetUnstablePeriod( TA_FUNC_UNST_ATR, test->unstablePeriod );
retCode = TA_ATR( test->startIdx,
test->endIdx,
gBuffer[0].in,
gBuffer[1].in,
gBuffer[2].in,
test->optInTimePeriod_0,
&outBegIdx,
&outNbElement,
gBuffer[0].in );
}
else
{
retCode = TA_TRANGE( test->startIdx,
test->endIdx,
gBuffer[0].in,
gBuffer[1].in,
gBuffer[2].in,
&outBegIdx,
&outNbElement,
gBuffer[0].in );
}
/* The previous call to TA_MA should have the same output
* as this call.
*
* checkSameContent verify that all value different than NAN in
* the first parameter is identical in the second parameter.
*/
errNb = checkSameContent( gBuffer[0].out0, gBuffer[0].in );
if( errNb != TA_TEST_PASS )
return errNb;
errNb = checkExpectedValue( gBuffer[0].in,
retCode, test->expectedRetCode,
outBegIdx, test->expectedBegIdx,
outNbElement, test->expectedNbElement,
test->oneOfTheExpectedOutReal,
test->oneOfTheExpectedOutRealIndex );
if( errNb != TA_TEST_PASS )
return errNb;
/* Do a systematic test of most of the
* possible startIdx/endIdx range.
*/
testParam.test = test;
testParam.high = history->high;
testParam.low = history->low;
testParam.close = history->close;
if( test->doRangeTestFlag )
{
errNb = doRangeTest( rangeTestFunction,
TA_FUNC_UNST_ATR,
(void *)&testParam, 1, 0 );
if( errNb != TA_TEST_PASS )
return errNb;
}
return TA_TEST_PASS;
}
/* The gateway routine */
void mexFunction(int nlhs, mxArray *plhs[],
int nrhs, const mxArray *prhs[])
{
/* ----------------- Variables ----------------- */
/* input variables */
/* mandatory input */
int startIdx;
int endIdx;
double * high;
double * low;
double * close;
/* optional input */
/* output variables */
int outBegIdx;
int outNbElement;
double* outReal;
/* input dimentions */
int inSeriesRows;
int inSeriesCols;
/* error handling */
TA_RetCode retCode;
/* ----------------- input/output count ----------------- */
/* Check for proper number of arguments. */
if (nrhs < 3 || nrhs > 3) mexErrMsgTxt("#3 inputs possible #0 optional.");
if (nlhs != 1) mexErrMsgTxt("#1 output required.");
/* ----------------- INPUT ----------------- */
/* Create a pointer to the input matrix high. */
high = mxGetPr(prhs[0]);
/* Get the dimensions of the matrix input high. */
inSeriesCols = mxGetN(prhs[0]);
if (inSeriesCols != 1) mexErrMsgTxt("high only vector alowed.");
/* Create a pointer to the input matrix low. */
low = mxGetPr(prhs[1]);
/* Get the dimensions of the matrix input low. */
inSeriesCols = mxGetN(prhs[1]);
if (inSeriesCols != 1) mexErrMsgTxt("low only vector alowed.");
/* Create a pointer to the input matrix close. */
close = mxGetPr(prhs[2]);
/* Get the dimensions of the matrix input close. */
inSeriesCols = mxGetN(prhs[2]);
if (inSeriesCols != 1) mexErrMsgTxt("close only vector alowed.");
inSeriesRows = mxGetM(prhs[2]);
endIdx = inSeriesRows - 1;
startIdx = 0;
/* Process optional arguments */
/* ----------------- OUTPUT ----------------- */
outReal = mxCalloc(inSeriesRows, sizeof(double));
/* -------------- Invocation ---------------- */
retCode = TA_TRANGE(
startIdx, endIdx,
high,
low,
close,
&outBegIdx, &outNbElement,
outReal);
/* -------------- Errors ---------------- */
if (retCode) {
mxFree(outReal);
mexPrintf("%s%i","Return code=",retCode);
mexErrMsgTxt(" Error!");
}
// Populate Output
plhs[0] = mxCreateDoubleMatrix(outBegIdx+outNbElement,1, mxREAL);
memcpy(((double *) mxGetData(plhs[0]))+outBegIdx, outReal, outNbElement*mxGetElementSize(plhs[0]));
mxFree(outReal);
} /* END mexFunction */
