local sScriptPath = getScriptPath()
package.path = sScriptPath..'\\?.lua;'..package.path
require('mod_lua')
package.cpath = sScriptPath..'\\?.dll;'..package.cpath
require('mod_cpp')

function main()
  -- Засекаем время:
  local nClock = os.clock()
  -- Считываем данные из файла и заносим их в массив в памяти в виде свечей
  local sError = mod_lua.CreateCandles('R:\\data.csv')
  -- Если функция завершилась с ошибкой, выводим сообщение об ошибке и выходим:
  if sError then message(sError, 3) error(sError, 0) end
  -- Выводим время работы функции:
  message(string.format('Чтение данных (Lua): %0.1f сек', os.clock()-nClock), 2)

  nClock = os.clock()
  sError = mod_cpp.CreateCandles('R:\\data.csv')
  if sError then message(sError, 3) error(sError, 0) end
  message(string.format('Чтение данных (C++): %0.1f сек', os.clock()-nClock), 2)

  nClock = os.clock()
  -- Расчитываем индикатор и заносим значения в массив
  -- Расчёт будет вестись юпо параметру Weighted Close ('W')
  -- Дополнительно передаём стартовое время и ID таблицы для вывода хода расчёта
  mod_lua.CalcSMA(50, 'W')
  -- Выводим время работы функции:
  message(string.format('Расчёт индикатора (Lua): %0.1f сек', os.clock()-nClock), 2)

  nClock = os.clock()
  mod_cpp.CalcSMA(50, 'W')
  message(string.format('Расчёт индикатора (C++): %0.1f сек', os.clock()-nClock), 2)
end

-- Заголовок файла с данными свечей:
local sHeader = '<TICKER>,<PER>,<DATE>,<TIME>,<OPEN>,<HIGH>,<LOW>,<CLOSE>,<VOL>'
local io_open, os_time, os_clock, pcall = io.open, os.time, os.clock, pcall
local string_format, string_match = string.format, string.match
local SetCell = SetCell

module('mod_lua')

-- Обявляем массивы для хранения значений индикатора и данных свечей:
local tSMA, tCandles = {}, {}

function CreateCandles(sFileName)
-- Функция чтения файла данных
  local hFile, sError = io_open(sFileName, 'r')
  if not hFile then return false, sError end
  local sLine = hFile:read()
  if sLine ~= sHeader then
    -- Если заголовок файла неверный - выходим с ошибкой:
    sError = 'Неверный заголовок в файле "'..sFileName..'": "'..(sLine or 'nil')..'"'
  else
    local tData, tPrice, nNum = {}, {}
    for sLine in hFile:lines() do
        -- Парсим считанную строку:
        local Year, Month, Day, Hour, Min, Sec, Open, High, Low, Close, Volume = string_match(sLine, '^[^,]+,%d+,(%d%d%d%d)(%d%d)(%d%d),(%d?%d)(%d%d)(%d%d),(%d+%.?%d-),(%d+%.?%d-),(%d+%.?%d-),(%d+%.?%d-),(%d+)')
        nNum, Close = (nNum or 0) + 1, Close+0
        -- Заносим в массив в виде свечей:
        tCandles[nNum] = {Open+0, High+0, Low+0, Close, Volume+0}
    end
  end
  hFile:close() hFile = nil
  return sError
end
local function _GetValue(data, sParam)
-- Функция получения значения заданного параметра из массива данных
-- (Close, Open, High, Low, Volume, Median Price, Typical Price, Weighted Close)
  if not sParam or sParam == 'C' then return data[4]
  elseif sParam == 'O' then return data[1]
  elseif sParam == 'H' then return data[2]
  elseif sParam == 'L' then return data[3]
  elseif sParam == 'V' then return data[5]
  elseif sParam == 'M' then return (data[2] + data[3]) / 2
  elseif sParam == 'T' then return (data[2] + data[3] + data[4]) / 3
  elseif sParam == 'W' then return (data[2] + data[3] + 2*data[4]) / 4
  else return data[4] end
end
local function SMA(nIndex, nPeriod, tData, sParam)
-- Расчёт среднего значения по заданному периоду
  if nIndex < nPeriod then return nil end
  local nSum = 0
  for i = nIndex-nPeriod+1, nIndex do
    nSum = nSum + _GetValue(tData[i], sParam)
  end
  return nSum / nPeriod
end
function CalcSMA(nPeriod, sParam, nClock, hID)
-- Расчёт простой скользящей средней
  for nIndex = 1, #tCandles do
    -- Заносим полученное значение в массив:
    tSMA[nIndex] = SMA(nIndex, nPeriod, tCandles, sParam)
  end
end

#include <windows.h>
#include <fstream>
#include <vector>

// в случае вызова функций из LUA-кода во внешней DLL
// необходимо определить эти константы до подключения заголовочных файлов LUA
#define LUA_LIB
#define LUA_BUILD_AS_DLL

// заголовочные файлы LUA из дистрибутива LUA
extern "C" {
  #include "../contrib/lauxlib.h"
  #include "../contrib/lua.h"
}

// стандартная точка входа для DLL
BOOL APIENTRY DllMain(HANDLE hmod_cppule, DWORD  ul_reason_for_call, LPVOID lpReserved) {
  return TRUE;
}

using namespace std;

vector<vector<double> > aData;
vector<double> aSMA;

double _GetValue(vector<double> &data, const char *sParam) {
// Функция получения значения заданного параметра из массива данных
// (Close, Open, High, Low, Volume, Median Price, Typical Price, Weighted Close)
  switch (*sParam) {
    case NULL:
      return data[3];
    case 'C':
      return data[3];
    case 'O':
      return data[0];
    case 'H':
      return data[1];
    case 'L':
      return data[2];
    case 'V':
      return data[4];
    case 'M':
      return (data[1] + data[2]) / 2;
    case 'T':
      return (data[1] + data[2] + data[3]) / 3;
    case 'W':
      return (data[1] + data[2] + 2*data[3]) / 4;
    default:
      return data[3];
  }
}

double _SMA(const unsigned int &iIndex, const unsigned short int &iPeriod, vector<vector<double> > &aData, const char *sParam) {
// Расчёт среднего значения по заданному периоду
  if (iIndex < iPeriod-1) return NULL;
  double nSum = 0;
  for ( size_t i = iIndex-iPeriod+1; i<=iIndex; i++ ) {
    nSum = nSum + _GetValue(aData[i], sParam);
  }
  return nSum / iPeriod;
}

// реализация функций, вызываемых из LUA

static int lua_CalcSMA(lua_State *L) {
// Расчёт простой скользящей средней
  const unsigned short int iPeriod = luaL_checknumber(L, 1);
  const char *sParam = lua_tostring(L, 2);
  for ( size_t i=0; i<aData.size(); i++ ) {
    // Заносим полученное значение в массив:
    aSMA.push_back(_SMA(i, iPeriod, aData, sParam));
  }
  lua_pushnil(L);
  return(1);
}

static int lua_CreateCandles(lua_State *L) {
  const char *sFileName = lua_tostring(L, 1);
  ifstream hFile (sFileName);
  if (!hFile.is_open()) lua_pushstring(L, "Не удалось открыть файл");  // Как добавить сюда имя файла?
  else {
    char sLine[100];
    hFile >> sLine;
    if (strcmp(sLine, "<TICKER>,<PER>,<DATE>,<TIME>,<OPEN>,<HIGH>,<LOW>,<CLOSE>,<VOL>") != 0) {
      lua_pushstring(L, "Неверный заголовок в файле");  // Как добавить сюда имя файла и считанную строку?
    } else {
      char s;
      double n;
      vector<double> v;
      while (!hFile.eof()) {
        if ( !hFile.get(sLine, 99, ',') ) break;
        for ( size_t i=0; i<=2; i++ ) hFile >> s >> n;
        for ( size_t i=3; i<=7; i++ ) {
          hFile >> s >> n;
          v.push_back(n);
        }
        hFile.get();
         aData.push_back(v);
        v.clear();
      }
      lua_pushnil(L);
    }
    hFile.close();
  }
  return(1);
}

// регистрация реализованных в dll функций, чтобы они стали "видимы" для LUA

static struct luaL_reg lib_functions[] = {
    {"CreateCandles", lua_CreateCandles},
    {"CalcSMA", lua_CalcSMA},
    {NULL, NULL}
};

extern "C" LUALIB_API int luaopen_mod_cpp(lua_State *L) {
    luaL_openlib(L, "mod_cpp", lib_functions, 0);
    return 0;
}

#include <fstream>
#include <vector>

using namespace std;

vector<vector<double> > aData;

static int lua_CreateCandles(lua_State *L) {
  const char *sFileName = lua_tostring(L, 1);
  ifstream hFile (sFileName);
  if (!hFile.is_open()) lua_pushstring(L, "Не удалось открыть файл");  // Как добавить сюда имя файла?
  else {
    char sLine[100];
    hFile >> sLine;
    if (strcmp(sLine, "<TICKER>,<PER>,<DATE>,<TIME>,<OPEN>,<HIGH>,<LOW>,<CLOSE>,<VOL>") != 0) {
      lua_pushstring(L, "Неверный заголовок в файле");  // Как добавить сюда имя файла и считанную строку?
    } else {
      char s;
      double d;
      vector<double> v;
      while (!hFile.eof()) {
        if ( !hFile.get(sLine, 99, ',') ) break;
        for ( size_t i=0; i<=2; i++ ) hFile >> s >> d;
        for ( size_t i=3; i<=7; i++ ) {
          hFile >> s >> d;
          v.push_back(d);
        }
        hFile.get();
        aData.push_back(v);
        v.clear();
      }
      lua_pushnil(L);
    }
    hFile.close();
  }
  return(1);
}

#include "stdafx.h"
using namespace std;
void read_fopen ( const char* f_name )
{
    FILE* fp = NULL;
    fopen_s ( &fp, f_name, "rt" );
    if ( fp )
    {
        char line[1024];
        while ( !feof ( fp ) )
        {
            fgets ( line, 1023, fp );
        }
        fclose ( fp );
    }
    else
        return;
}
void read_ifstream ( const char* fname )
{
    ifstream hFile ( fname ); //Открываем файл для считывания информации
    if ( hFile.is_open() ) {
        char sLine[1024];  //В sLine будут считываться строки
        while ( !hFile.eof() ) { //Будем читать информацию пока не дойдем до конца файла
            hFile >> sLine;  //Построчное считывание информации в sLine
        }
        hFile.close();  // Закрываем файл
    }
    else
        return;
}
int _tmain ( int argc, _TCHAR* argv[] )
{
    const char f_name[MAX_PATH] = "D:\\data.txt";
    int MAX_N = 100;
    long start = GetTickCount();
    for ( int i = 0; i < MAX_N; i++ )
    {
        read_fopen ( f_name );
    }
    long end = GetTickCount();
    printf ( "read_fopen : %d ticks\n", end - start );

    start = GetTickCount();
    for ( int i = 0; i < MAX_N; i++ )
    {
        read_ifstream ( f_name );
    }
    end = GetTickCount();
    printf ( "read_ifstream : %d ticks\n", end - start );

}

static int lua_CreateCandles(lua_State *L) {
  const char * sFileName = lua_tostring(L, 1);
  ifstream hFile (sFileName);  //Открываем файл для считывания информации
  if (hFile.is_open()) {
    char sLine[100];  //В sLine будут считываться строки
    while (!hFile.eof()) {  //Будем читать информацию пока не дойдем до конца файла
      hFile >> sLine;  //Построчное считывание информации в sLine
    }
    hFile.close();  // Закрываем файл
    lua_pushstring(L, NULL);
  }
  else {
    lua_pushstring(L, "Не удалось открыть файл");  // Как сюда добавить имя файла?
  }
  return(1);
}