Module: TALibFFI

Extended by:
Fiddle::Importer
Defined in:
lib/ta_lib_ffi.rb

Overview

Ruby FFI wrapper for TA-Lib (Technical Analysis Library)

Defined Under Namespace

Classes: Inflector, TALibError

Constant Summary collapse

VERSION = 
"0.2.0"
TA_SUCCESS = 
0
TA_LIB_NOT_INITIALIZE = 
1
TA_BAD_PARAM = 
2
TA_ALLOC_ERR = 
3
TA_GROUP_NOT_FOUND = 
4
TA_FUNC_NOT_FOUND = 
5
TA_INVALID_HANDLE = 
6
TA_INVALID_PARAM_HOLDER = 
7
TA_INVALID_PARAM_HOLDER_TYPE = 
8
TA_INVALID_PARAM_FUNCTION = 
9
TA_INPUT_NOT_ALL_INITIALIZE = 
10
TA_OUTPUT_NOT_ALL_INITIALIZE = 
11
TA_OUT_OF_RANGE_START_INDEX = 
12
TA_OUT_OF_RANGE_END_INDEX = 
13
TA_INVALID_LIST_TYPE = 
14
TA_BAD_OBJECT = 
15
TA_NOT_SUPPORTED = 
16
TA_INTERNAL_ERROR = 
5000
TA_UNKNOWN_ERR = 
0xFFFF
TA_StringTable = 
struct [
  "unsigned int size",
  "const char **string",
  "void *hiddenData"
]
TA_FuncInfo = 
struct [
  "const char *name",
  "const char *group",
  "const char *hint",
  "const char *camelCaseName",
  "TA_FuncFlags flags",
  "unsigned int nbInput",
  "unsigned int nbOptInput",
  "unsigned int nbOutput",
  "const TA_FuncHandle *handle"
]
TA_ParamHolder = 
struct [
  "void *hiddenData"
]
TA_InputParameterInfo = 
struct [
  "TA_InputParameterType type",
  "const char *paramName",
  "TA_InputFlags flags"
]
TA_OptInputParameterInfo = 
struct [
  "TA_OptInputParameterType type",
  "const char *paramName",
  "TA_OptInputFlags flags",
  "const char *displayName",
  "const void *dataSet",
  "TA_Real defaultValue",
  "const char *hint",
  "const char *helpFile"
]
TA_OutputParameterInfo = 
struct [
  "TA_OutputParameterType type",
  "const char *paramName",
  "TA_OutputFlags flags"
]
TA_PARAM_TYPE = 
{
  TA_Input_Price: 0,
  TA_Input_Real: 1,
  TA_Input_Integer: 2,
  TA_OptInput_RealRange: 0,
  TA_OptInput_RealList: 1,
  TA_OptInput_IntegerRange: 2,
  TA_OptInput_IntegerList: 3,
  TA_Output_Real: 0,
  TA_Output_Integer: 1
}.freeze
TA_FLAGS = 
{
  TA_InputFlags: {
    TA_IN_PRICE_OPEN: 0x00000001,
    TA_IN_PRICE_HIGH: 0x00000002,
    TA_IN_PRICE_LOW: 0x00000004,
    TA_IN_PRICE_CLOSE: 0x00000008,
    TA_IN_PRICE_VOLUME: 0x00000010,
    TA_IN_PRICE_OPENINTEREST: 0x00000020,
    TA_IN_PRICE_TIMESTAMP: 0x00000040
  },
  TA_OptInputFlags: {
    TA_OPTIN_IS_PERCENT: 0x00100000,
    TA_OPTIN_IS_DEGREE: 0x00200000,
    TA_OPTIN_IS_CURRENCY: 0x00400000,
    TA_OPTIN_ADVANCED: 0x01000000
  },
  TA_OutputFlags: {
    TA_OUT_LINE: 0x00000001,
    TA_OUT_DOT_LINE: 0x00000002,
    TA_OUT_DASH_LINE: 0x00000004,
    TA_OUT_DOT: 0x00000008,
    TA_OUT_HISTO: 0x00000010,
    TA_OUT_PATTERN_BOOL: 0x00000020,
    TA_OUT_PATTERN_BULL_BEAR: 0x00000040,
    TA_OUT_PATTERN_STRENGTH: 0x00000080,
    TA_OUT_POSITIVE: 0x00000100,
    TA_OUT_NEGATIVE: 0x00000200,
    TA_OUT_ZERO: 0x00000400,
    TA_OUT_UPPER_LIMIT: 0x00000800,
    TA_OUT_LOWER_LIMIT: 0x00001000
  }
}.freeze

Class Method Summary collapse

Class Method Details

.build_function_info_map ⇒ Object 



525
526
527
528
529
530
531
532
533
534
535
536
 
# File 'lib/ta_lib_ffi.rb', line 525

def build_function_info_map
  info_map = {}
  each_function do |func_info|
    info_map[func_info["name"].to_s] = {
      info: func_info,
      inputs: collect_input_info(func_info),
      outputs: collect_output_info(func_info),
      opt_inputs: collect_opt_input_info(func_info)
    }
  end
  info_map
end

.calculate_lookback(params_ptr) ⇒ Object

rubocop:enable Metrics/MethodLength



347
348
349
350
351
352
 
# File 'lib/ta_lib_ffi.rb', line 347

def calculate_lookback(params_ptr)
  lookback_ptr = Fiddle::Pointer.malloc(Fiddle::SIZEOF_INT)
  ret_code = TA_GetLookback(params_ptr, lookback_ptr)
  check_ta_return_code(ret_code)
  lookback_ptr[0, Fiddle::SIZEOF_INT].unpack1("l")
end

.calculate_results(params_ptr, input_size, func_name) ⇒ Object

rubocop:disable Metrics/MethodLength



444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
 
# File 'lib/ta_lib_ffi.rb', line 444

def calculate_results(params_ptr, input_size, func_name)
  out_begin = Fiddle::Pointer.malloc(Fiddle::SIZEOF_INT)
  out_size = Fiddle::Pointer.malloc(Fiddle::SIZEOF_INT)
  output_arrays = setup_output_buffers(params_ptr, input_size, func_name)

  begin
    ret_code = TA_CallFunc(params_ptr, 0, input_size - 1, out_begin, out_size)
    check_ta_return_code(ret_code)

    actual_size = out_size[0, Fiddle::SIZEOF_INT].unpack1("l")
    format_output_results(output_arrays, actual_size, func_name)
  ensure
    out_begin.free
    out_size.free
    output_arrays.each(&:free)
  end
end

.call_func(func_name, args) ⇒ Object

rubocop:disable Metrics/MethodLength



327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
 
# File 'lib/ta_lib_ffi.rb', line 327

def call_func(func_name, args)
  options = args.last.is_a?(Hash) ? args.pop : {}
  input_arrays = args

  validate_inputs!(input_arrays)

  handle_ptr = get_function_handle(func_name)
  params_ptr = create_parameter_holder(handle_ptr)

  begin
    setup_input_parameters(params_ptr, input_arrays, func_name)
    setup_optional_parameters(params_ptr, options, func_name)
    _lookback = calculate_lookback(params_ptr)
    calculate_results(params_ptr, input_arrays.first.length, func_name)
  ensure
    TA_ParamHolderFree(params_ptr)
  end
end

.check_ta_return_code(code) ⇒ Object

rubocop:disable Metrics/MethodLength rubocop:disable Metrics/AbcSize rubocop:disable Metrics/CyclomaticComplexity

Raises:



580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
 
# File 'lib/ta_lib_ffi.rb', line 580

def check_ta_return_code(code)
  return if code == TA_SUCCESS

  error_message = case code
                  when TA_LIB_NOT_INITIALIZE
                    "TA-Lib not initialized, please call TA_Initialize first"
                  when TA_BAD_PARAM
                    "Bad parameter, please check input parameters"
                  when TA_ALLOC_ERR
                    "Memory allocation error, possibly insufficient memory"
                  when TA_GROUP_NOT_FOUND
                    "Function group not found"
                  when TA_FUNC_NOT_FOUND
                    "Function not found"
                  when TA_INVALID_HANDLE
                    "Invalid handle"
                  when TA_INVALID_PARAM_HOLDER
                    "Invalid parameter holder"
                  when TA_INVALID_PARAM_HOLDER_TYPE
                    "Invalid parameter holder type"
                  when TA_INVALID_PARAM_FUNCTION
                    "Invalid parameter function"
                  when TA_INPUT_NOT_ALL_INITIALIZE
                    "Input parameters not fully initialized"
                  when TA_OUTPUT_NOT_ALL_INITIALIZE
                    "Output parameters not fully initialized"
                  when TA_OUT_OF_RANGE_START_INDEX
                    "Start index out of range"
                  when TA_OUT_OF_RANGE_END_INDEX
                    "End index out of range"
                  when TA_INVALID_LIST_TYPE
                    "Invalid list type"
                  when TA_BAD_OBJECT
                    "Invalid object"
                  when TA_NOT_SUPPORTED
                    "Operation not supported"
                  when TA_INTERNAL_ERROR
                    "TA-Lib internal error"
                  when TA_UNKNOWN_ERR
                    "Unknown error"
                  else
                    "Undefined TA-Lib error (Error code: #{code})"
                  end

  raise TALibError, error_message
end

.collect_input_info(func_info) ⇒ Object 



538
539
540
541
542
543
544
 
# File 'lib/ta_lib_ffi.rb', line 538

def collect_input_info(func_info)
  func_info["nbInput"].times.map do |i|
    param_info_ptr = Fiddle::Pointer.malloc(Fiddle::SIZEOF_VOIDP)
    TA_GetInputParameterInfo(func_info["handle"], i, param_info_ptr.ref)
    TA_InputParameterInfo.new(param_info_ptr)
  end
end

.collect_opt_input_info(func_info) ⇒ Object 



546
547
548
549
550
551
552
 
# File 'lib/ta_lib_ffi.rb', line 546

def collect_opt_input_info(func_info)
  func_info["nbOptInput"].times.map do |i|
    param_info_ptr = Fiddle::Pointer.malloc(Fiddle::SIZEOF_VOIDP)
    TA_GetOptInputParameterInfo(func_info["handle"], i, param_info_ptr.ref)
    TA_OptInputParameterInfo.new(param_info_ptr)
  end
end

.collect_output_info(func_info) ⇒ Object 



554
555
556
557
558
559
560
 
# File 'lib/ta_lib_ffi.rb', line 554

def collect_output_info(func_info)
  func_info["nbOutput"].times.map do |i|
    param_info_ptr = Fiddle::Pointer.malloc(Fiddle::SIZEOF_VOIDP)
    TA_GetOutputParameterInfo(func_info["handle"], i, param_info_ptr.ref)
    TA_OutputParameterInfo.new(param_info_ptr)
  end
end

.create_parameter_holder(handle_ptr) ⇒ Object 



380
381
382
383
384
385
 
# File 'lib/ta_lib_ffi.rb', line 380

def create_parameter_holder(handle_ptr)
  params_ptr = Fiddle::Pointer.malloc(Fiddle::SIZEOF_VOIDP)
  ret_code = TA_ParamHolderAlloc(handle_ptr, params_ptr.ref)
  check_ta_return_code(ret_code)
  params_ptr
end

.define_ta_function(method_name, func_name) ⇒ Object 



639
640
641
642
643
 
# File 'lib/ta_lib_ffi.rb', line 639

def define_ta_function(method_name, func_name)
  define_singleton_method(method_name) do |*args|
    call_func(func_name, args)
  end
end

.each_function(&block) ⇒ Object 



258
259
260
261
262
263
264
265
266
267
268
269
 
# File 'lib/ta_lib_ffi.rb', line 258

def each_function(&block)
  callback = Fiddle::Closure::BlockCaller.new(
    Fiddle::TYPE_VOID,
    [Fiddle::TYPE_VOIDP, Fiddle::TYPE_VOIDP],
    Fiddle::Function::DEFAULT
  ) do |func_info_ptr, _|
    block.call TA_FuncInfo.new(func_info_ptr)
  end

  ret_code = TA_ForEachFunc(callback, nil)
  check_ta_return_code(ret_code)
end

.extract_flags(value, type) ⇒ Object 



213
214
215
216
217
218
219
 
# File 'lib/ta_lib_ffi.rb', line 213

def extract_flags(value, type)
  flags_set = []
  TA_FLAGS[type].each do |k, v|
    flags_set << k if (value & v) != 0
  end
  flags_set
end

.format_output_results(output_ptrs, size, func_name) ⇒ Object

rubocop:disable Metrics/MethodLength rubocop:disable Metrics/AbcSize



496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
 
# File 'lib/ta_lib_ffi.rb', line 496

def format_output_results(output_ptrs, size, func_name)
  func_info = function_info_map[func_name]
  results = output_ptrs.zip(func_info[:outputs]).map do |ptr, output|
    case output["type"]
    when TA_PARAM_TYPE[:TA_Output_Real]
      ptr[0, Fiddle::SIZEOF_DOUBLE * size].unpack("d#{size}")
    when TA_PARAM_TYPE[:TA_Output_Integer]
      ptr[0, Fiddle::SIZEOF_INT * size].unpack("l#{size}")
    end
  end

  return results.first if results.length == 1

  output_names = func_info[:outputs].map do |output|
    normalize_parameter_name(output["paramName"].to_s).to_sym
  end
  output_names.zip(results).to_h
end

.function_description_xml ⇒ Object

rubocop:enable Metrics/AbcSize rubocop:enable Metrics/MethodLength



517
518
519
 
# File 'lib/ta_lib_ffi.rb', line 517

def function_description_xml
  TA_FunctionDescriptionXML().to_s
end

.function_info(name) ⇒ Object 



246
247
248
249
250
251
252
253
254
255
256
 
# File 'lib/ta_lib_ffi.rb', line 246

def function_info(name)
  handle_ptr = Fiddle::Pointer.malloc(Fiddle::SIZEOF_VOIDP)
  ret_code = TA_GetFuncHandle(name, handle_ptr.ref)
  check_ta_return_code(ret_code)

  info_ptr = Fiddle::Pointer.malloc(Fiddle::SIZEOF_VOIDP)
  ret_code = TA_GetFuncInfo(handle_ptr, info_ptr.ref)
  check_ta_return_code(ret_code)

  TA_FuncInfo.new(info_ptr)
end

.function_info_map ⇒ Object 



521
522
523
 
# File 'lib/ta_lib_ffi.rb', line 521

def function_info_map
  @function_info_map ||= build_function_info_map
end

.function_table(group) ⇒ Object 



233
234
235
236
237
238
239
240
241
242
243
244
 
# File 'lib/ta_lib_ffi.rb', line 233

def function_table(group)
  string_table_ptr = Fiddle::Pointer.malloc(Fiddle::SIZEOF_VOIDP)
  ret_code = TA_FuncTableAlloc(group, string_table_ptr.ref)
  check_ta_return_code(ret_code)

  string_table = TA_StringTable.new(string_table_ptr)
  func_names = Fiddle::Pointer.new(string_table["string"])[0, Fiddle::SIZEOF_VOIDP * string_table["size"]].unpack("Q*").collect { |ptr| Fiddle::Pointer.new(ptr).to_s }

  TA_FuncTableFree(string_table)

  func_names
end

.generate_ta_functions ⇒ Object 



562
563
564
565
566
 
# File 'lib/ta_lib_ffi.rb', line 562

def generate_ta_functions
  each_function do |func_info|
    define_ta_function(func_info["name"].to_s.downcase, func_info["name"].to_s)
  end
end

.get_function_handle(func_name) ⇒ Object

rubocop:enable Metrics/CyclomaticComplexity rubocop:enable Metrics/PerceivedComplexity



373
374
375
376
377
378
 
# File 'lib/ta_lib_ffi.rb', line 373

def get_function_handle(func_name)
  handle_ptr = Fiddle::Pointer.malloc(Fiddle::SIZEOF_VOIDP)
  ret_code = TA_GetFuncHandle(func_name, handle_ptr.ref)
  check_ta_return_code(ret_code)
  handle_ptr
end

.group_table ⇒ Object 



221
222
223
224
225
226
227
228
229
230
231
 
# File 'lib/ta_lib_ffi.rb', line 221

def group_table
  string_table_ptr = Fiddle::Pointer.malloc(Fiddle::SIZEOF_VOIDP)
  ret_code = TA_GroupTableAlloc(string_table_ptr.ref)
  check_ta_return_code(ret_code)

  string_table = TA_StringTable.new(string_table_ptr)
  group_names = Fiddle::Pointer.new(string_table["string"])[0, Fiddle::SIZEOF_VOIDP * string_table["size"]].unpack("Q*").collect { |ptr| Fiddle::Pointer.new(ptr).to_s }
  TA_GroupTableFree(string_table_ptr)

  group_names
end

.initialize_ta_lib ⇒ Object

rubocop:enable Metrics/CyclomaticComplexity rubocop:enable Metrics/MethodLength rubocop:enable Metrics/AbcSize



630
631
632
633
634
635
636
637
 
# File 'lib/ta_lib_ffi.rb', line 630

def initialize_ta_lib
  return if @initialized

  ret_code = TA_Initialize()
  check_ta_return_code(ret_code)
  at_exit { TA_Shutdown() }
  @initialized = true
end

.normalize_parameter_name(name) ⇒ Object 



568
569
570
571
572
573
574
575
 
# File 'lib/ta_lib_ffi.rb', line 568

def normalize_parameter_name(name)
  name.sub(/^(optIn|outReal|outInteger|out)/, "")
      .gsub(/::/, "/")
      .gsub(/([A-Z]+)([A-Z][a-z])/, '\1_\2')
      .gsub(/([a-z\d])([A-Z])/, '\1_\2')
      .tr("-", "_")
      .downcase
end

.prepare_double_array(array) ⇒ Object 



409
410
411
412
413
414
415
 
# File 'lib/ta_lib_ffi.rb', line 409

def prepare_double_array(array)
  array_ptr = Fiddle::Pointer.malloc(Fiddle::SIZEOF_DOUBLE * array.length)
  array.each_with_index do |value, i|
    array_ptr[i * Fiddle::SIZEOF_DOUBLE, Fiddle::SIZEOF_DOUBLE] = [value.to_f].pack("d")
  end
  array_ptr
end

.prepare_integer_array(array) ⇒ Object 



417
418
419
420
421
422
423
 
# File 'lib/ta_lib_ffi.rb', line 417

def prepare_integer_array(array)
  array_ptr = Fiddle::Pointer.malloc(Fiddle::SIZEOF_INT * array.length)
  array.each_with_index do |value, i|
    array_ptr[i * Fiddle::SIZEOF_INT, Fiddle::SIZEOF_INT] = [value.to_i].pack("l")
  end
  array_ptr
end

.print_function_info(func_info) ⇒ Object

rubocop:disable Metrics/MethodLength rubocop:disable Metrics/AbcSize



273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
 
# File 'lib/ta_lib_ffi.rb', line 273

def print_function_info(func_info)
  puts "Function Name: #{func_info["name"]}"
  puts "Function Group: #{func_info["group"]}"
  puts "Function Hint: #{func_info["hint"]}"
  puts "Camel Case Name: #{func_info["camelCaseName"]}"
  puts "Flags: #{func_info["flags"]}"
  puts "Number of Inputs: #{func_info["nbInput"]}"
  puts "Number of Optional Inputs: #{func_info["nbOptInput"]}"
  puts "Number of Outputs: #{func_info["nbOutput"]}"
  puts "Function Handle: #{func_info["handle"].to_i}"

  puts "\nInput Parameter Info:"
  func_info["nbInput"].times do |i|
    param_info_ptr = Fiddle::Pointer.malloc(Fiddle::SIZEOF_VOIDP)
    ret_code = TA_GetInputParameterInfo(func_info["handle"], i, param_info_ptr.ref)
    check_ta_return_code(ret_code)
    param_info = TA_InputParameterInfo.new(param_info_ptr)
    puts "  Parameter #{i + 1}:"
    puts "    Name: #{param_info["paramName"]}"
    puts "    Type: #{param_info["type"]}"
    puts "    Flags: #{extract_flags(param_info["flags"], :TA_InputFlags)}"
  end

  puts "\nOptional Input Parameter Info:"
  func_info["nbOptInput"].times do |i|
    param_info_ptr = Fiddle::Pointer.malloc(Fiddle::SIZEOF_VOIDP)
    ret_code = TA_GetOptInputParameterInfo(func_info["handle"], i, param_info_ptr.ref)
    check_ta_return_code(ret_code)
    param_info = TA_OptInputParameterInfo.new(param_info_ptr)
    puts "  Parameter #{i + 1}:"
    puts "    Name: #{param_info["paramName"]}"
    puts "    Type: #{param_info["type"]}"
    puts "    Flags: #{extract_flags(param_info["flags"], :TA_OptInputFlags)}"
    puts "    Display Name: #{param_info["displayName"]}"
    puts "    Default Value: #{param_info["defaultValue"]}"
    puts "    Hint: #{param_info["hint"]}"
  end

  puts "\nOutput Parameter Info:"
  func_info["nbOutput"].times do |i|
    param_info_ptr = Fiddle::Pointer.malloc(Fiddle::SIZEOF_VOIDP)
    ret_code = TA_GetOutputParameterInfo(func_info["handle"], i, param_info_ptr.ref)
    check_ta_return_code(ret_code)
    param_info = TA_OutputParameterInfo.new(param_info_ptr)
    puts "  Parameter #{i + 1}:"
    puts "    Name: #{param_info["paramName"]}"
    puts "    Type: #{param_info["type"]}"
    puts "    Flags: #{extract_flags(param_info["flags"], :TA_OutputFlags)}"
  end
end

.set_input_parameter(params_ptr, index, array, input_info) ⇒ Object 



396
397
398
399
400
401
402
403
404
405
406
407
 
# File 'lib/ta_lib_ffi.rb', line 396

def set_input_parameter(params_ptr, index, array, input_info)
  case input_info["type"]
  when TA_PARAM_TYPE[:TA_Input_Real]
    input_ptr = prepare_double_array(array)
    TA_SetInputParamRealPtr(params_ptr, index, input_ptr)
  when TA_PARAM_TYPE[:TA_Input_Integer]
    input_ptr = prepare_integer_array(array)
    TA_SetInputParamIntegerPtr(params_ptr, index, input_ptr)
  when TA_PARAM_TYPE[:TA_Input_Price]
    setup_price_inputs(params_ptr, index, array, input_info["flags"])
  end
end

.set_optional_parameter(params_ptr, index, value, type) ⇒ Object 



433
434
435
436
437
438
439
440
441
 
# File 'lib/ta_lib_ffi.rb', line 433

def set_optional_parameter(params_ptr, index, value, type)
  case type
  when TA_PARAM_TYPE[:TA_OptInput_RealRange], TA_PARAM_TYPE[:TA_OptInput_RealList]
    ret_code = TA_SetOptInputParamReal(params_ptr, index, value)
  when TA_PARAM_TYPE[:TA_OptInput_IntegerRange], TA_PARAM_TYPE[:TA_OptInput_IntegerList]
    ret_code = TA_SetOptInputParamInteger(params_ptr, index, value)
  end
  check_ta_return_code(ret_code)
end

.setup_input_parameters(params_ptr, input_arrays, func_name) ⇒ Object 



387
388
389
390
391
392
393
394
 
# File 'lib/ta_lib_ffi.rb', line 387

def setup_input_parameters(params_ptr, input_arrays, func_name)
  func_info = function_info_map[func_name]
  input_arrays.each_with_index do |array, index|
    input_info = func_info[:inputs][index]
    ret_code = set_input_parameter(params_ptr, index, array, input_info)
    check_ta_return_code(ret_code)
  end
end

.setup_optional_parameters(params_ptr, options, func_name) ⇒ Object 



425
426
427
428
429
430
431
 
# File 'lib/ta_lib_ffi.rb', line 425

def setup_optional_parameters(params_ptr, options, func_name)
  func_info = function_info_map[func_name]
  func_info[:opt_inputs]&.each_with_index do |opt_input, index|
    param_name = normalize_parameter_name(opt_input["paramName"].to_s)
    set_optional_parameter(params_ptr, index, options[param_name.to_sym], opt_input["type"]) if options.key?(param_name.to_sym)
  end
end

.setup_output_buffers(params_ptr, size, func_name) ⇒ Object

rubocop:disable Metrics/MethodLength rubocop:disable Metrics/AbcSize



465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
 
# File 'lib/ta_lib_ffi.rb', line 465

def setup_output_buffers(params_ptr, size, func_name)
  func_info = function_info_map[func_name]
  output_ptrs = []

  func_info[:outputs].each_with_index do |output, index|
    ptr = case output["type"]
          when TA_PARAM_TYPE[:TA_Output_Real]
            Fiddle::Pointer.malloc(Fiddle::SIZEOF_DOUBLE * size)
          when TA_PARAM_TYPE[:TA_Output_Integer]
            Fiddle::Pointer.malloc(Fiddle::SIZEOF_INT * size)
          end

    output_ptrs << ptr

    ret_code =  case output["type"]
                when TA_PARAM_TYPE[:TA_Output_Real]
                  TA_SetOutputParamRealPtr(params_ptr, index, ptr)
                when TA_PARAM_TYPE[:TA_Output_Integer]
                  TA_SetOutputParamIntegerPtr(params_ptr, index, ptr)
                end

    check_ta_return_code(ret_code)
  end

  output_ptrs
end

.setup_price_inputs(params_ptr, index, price_data, flags) ⇒ Object 



645
646
647
648
649
650
651
652
653
654
655
656
657
 
# File 'lib/ta_lib_ffi.rb', line 645

def setup_price_inputs(params_ptr, index, price_data, flags)
  required_flags = extract_flags(flags, :TA_InputFlags)
  data_pointers = Array.new(6) { nil }
  TA_FLAGS[:TA_InputFlags].keys[0..5].each_with_index do |flag, i|
    data_pointers[i] = if required_flags.include?(flag)
                         prepare_double_array(price_data[required_flags.index(flag)])
                       else
                         Fiddle::Pointer.malloc(0)
                       end
  end

  TA_SetInputParamPricePtr(params_ptr, index, *data_pointers)
end

.validate_inputs!(arrays) ⇒ Object

rubocop:disable Metrics/CyclomaticComplexity rubocop:disable Metrics/PerceivedComplexity

Raises:



356
357
358
359
360
361
362
363
364
365
366
367
368
369
 
# File 'lib/ta_lib_ffi.rb', line 356

def validate_inputs!(arrays)
  raise TALibError, "Input arrays cannot be empty" if arrays.empty?

  arrays.each do |arr|
    raise TALibError, "Input must be arrays" unless arr.is_a?(Array)
  end

  sizes = arrays.map(&:length)
  raise TALibError, "Input arrays cannot be empty" if sizes.any?(&:zero?)

  arrays.each do |arr|
    raise TALibError, "Input arrays must contain only numbers" unless arr.flatten.all? { |x| x.is_a?(Numeric) }
  end
end