Lua модуль для работы с комплексными числами

Опубликовано в QLua

Lua модуль для работы с комплексными числами. Например, потребуется для расчета FFT -  Fast Fourier Transform


 

  1. -------------------------------------------------------------
  2. -- complex.lua — модуль комплексных чисел (OOP + метаметоды)
  3.  
  4. complex = {
  5. -- Сложение: a + b
  6. __add = function(a, b)
  7.    return complex(a.re + b.re, a.im + b.im)
  8. end,
  9. -- Вычитание: a - b
  10. __sub = function(a, b)
  11.    return complex(a.re - b.re, a.im - b.im)
  12. end,
  13. -- Умножение: a * b
  14. __mul = function(a, b)
  15.    return complex(a.re * b.re - a.im * b.im, a.re * b.im + a.im * b.re)
  16. end,
  17. -- Деление: a / b
  18. __div = function(a, b)
  19. local denom = b.re * b.re + b.im * b.im
  20.    return complex((a.re * b.re + a.im * b.im) / denom, (a.im * b.re - a.re * b.im) / denom)
  21. end,
  22. -- Унарный минус: -a
  23. __unm = function(a)
  24.    return complex(-a.re, -a.im)
  25. end,
  26. -- Оператор длины (#a): модуль числа
  27. __len = function(a)
  28.    return math.sqrt(a.re ^ 2 + a.im ^ 2)
  29. end,
  30. -- Преобразование в строку: tostring(a)
  31. __tostring = function(a)
  32.    return string.format("%.4f %+ .4fi", a.re, a.im)
  33. end,
  34. ---------------------------------------------------------
  35. -- Методы экземпляра
  36. ---------------------------------------------------------
  37.  
  38. -- Модуль (амплитуда)
  39. abs = function(self)
  40.    return math.sqrt(self.re * self.re + self.im * self.im)
  41. end,
  42. -- Аргумент (фаза)
  43. arg = function(self)
  44.    return math.atan2(self.im, self.re)
  45. end,
  46. -- Сопряжённое число
  47. conj = function(self)
  48.    return complex(self.re, -self.im)
  49. end,
  50. -- Экспонента e^{i?}
  51. exp = function(theta)
  52.    return complex(math.cos(theta), math.sin(theta))
  53. end,
  54. -- Копия числа
  55. clone = function(self)
  56.    return complex(self.re, self.im)
  57. end
  58. }
  59.  
  60. -------------------------------------------------------------
  61. -- Метатаблица: конструктор complex(re, im)
  62. -------------------------------------------------------------
  63. complex.__index = complex
  64. local complex =
  65. setmetatable(
  66. complex,
  67. {
  68. __call = function(_, re, im)
  69. return setmetatable(
  70. {
  71. re = re or 0.0,
  72. im = im or 0.0
  73. },
  74. complex
  75. )
  76. end
  77. }
  78. )
  79.  
  80. -------------------------------------------------------------
  81.  
  82. -- local function log(label, value)
  83. -- print(string.format("%-18s = %s", label, tostring(value)))
  84. -- end
  85.  
  86. -----------------------------------------------------------
  87. -- local a = complex(2, 3)
  88. -- local b = complex(1, -1)
  89.  
  90. -- print("=== Complex Numbers Test ===\n")
  91.  
  92. -- log("a", a)
  93. -- log("b", b)
  94.  
  95. -----------------------------------------------------------
  96. -- print("\n=== Arithmetic Operations ===")
  97.  
  98. -- log("a + b", a + b)
  99. -- log("a - b", a - b)
  100. -- log("a * b", a * b)
  101. -- log("a / b", a / b)
  102. -- log("-a", -a)
  103.  
  104. -----------------------------------------------------------
  105. -- print("\n=== Metamethods ===")
  106.  
  107. -- log("#a (magnitude)", #a)
  108. -- log("tostring(a)", tostring(a))
  109.  
  110. -----------------------------------------------------------
  111. -- print("\n=== Instance Methods ===")
  112.  
  113. -- log("a:abs()", a:abs())
  114. -- log("a:arg()", a:arg())
  115. -- log("a:conj()", a:conj())
  116. -- log("a:clone()", a:clone())
  117.  
  118. -----------------------------------------------------------
  119. -- print("\n=== Complex Exponential ===")
  120.  
  121. -- local theta = math.pi / 4
  122. -- log("exp(i?/4)", complex.exp(theta))
  123.  

Как узнать, что окно QUIK вынесено?

Опубликовано в QLua

В терминале QUIK можно вынести окно скрипта за пределы главного окна терминала. Делается это мышкой при зажатой клавише Ctrl. Поведение таких окон отличается от поведения обычных, например в вопросе определения их координат.


Чтобы узнать, что окно вынесено, можно использовать такую библиотечную С++ функцию:

static int b4s_IsWindowDetached(lua_State* L)
{
   HWND hwnd = (HWND) luaL_checkinteger(L, 1);

   LONG style = GetWindowLong(hwnd, GWL_STYLE);

   lua_pushboolean(L, (style & WS_CHILD) == 0);
   return 1;
}

 

Она получает на входе хендл окна и анализирует его стиль. Если маска WS_CHILD установлена, значит, окно не вынесено и наоборот.

Узнать хэндл окна можно поиском окна по его имени посредством WinAPI метода FindWindowEx

Кодек UTF8

Опубликовано в QLua

Кодек ANSI - UTF-8 для Lua

 

utf8 = {
    _ansi_decode = {
        [128] = "\208\130",
        [129] = "\208\131",
        [130] = "\226\128\154",
        [131] = "\209\147",
        [132] = "\226\128\158",
        [133] = "\226\128\166",
        [134] = "\226\128\160",
        [135] = "\226\128\161",
        [136] = "\226\130\172",
        [137] = "\226\128\176",
        [138] = "\208\137",
        [139] = "\226\128\185",
        [140] = "\208\138",
        [141] = "\208\140",
        [142] = "\208\139",
        [143] = "\208\143",
        [144] = "\209\146",
        [145] = "\226\128\152",
        [146] = "\226\128\153",
        [147] = "\226\128\156",
        [148] = "\226\128\157",
        [149] = "\226\128\162",
        [150] = "\226\128\147",
        [151] = "\226\128\148",
        [152] = "\194\152",
        [153] = "\226\132\162",
        [154] = "\209\153",
        [155] = "\226\128\186",
        [156] = "\209\154",
        [157] = "\209\156",
        [158] = "\209\155",
        [159] = "\209\159",
        [160] = "\194\160",
        [161] = "\209\142",
        [162] = "\209\158",
        [163] = "\208\136",
        [164] = "\194\164",
        [165] = "\210\144",
        [166] = "\194\166",
        [167] = "\194\167",
        [168] = "\208\129",
        [169] = "\194\169",
        [170] = "\208\132",
        [171] = "\194\171",
        [172] = "\194\172",
        [173] = "\194\173",
        [174] = "\194\174",
        [175] = "\208\135",
        [176] = "\194\176",
        [177] = "\194\177",
        [178] = "\208\134",
        [179] = "\209\150",
        [180] = "\210\145",
        [181] = "\194\181",
        [182] = "\194\182",
        [183] = "\194\183",
        [184] = "\209\145",
        [185] = "\226\132\150",
        [186] = "\209\148",
        [187] = "\194\187",
        [188] = "\209\152",
        [189] = "\208\133",
        [190] = "\209\149",
        [191] = "\209\151"
    },
    _utf8_decode = {
        [128] = {
            [147] = "\150",
            [148] = "\151",
            [152] = "\145",
            [153] = "\146",
            [154] = "\130",
            [156] = "\147",
            [157] = "\148",
            [158] = "\132",
            [160] = "\134",
            [161] = "\135",
            [162] = "\149",
            [166] = "\133",
            [176] = "\137",
            [185] = "\139",
            [186] = "\155"
        },
        [130] = {[172] = "\136"},
        [132] = {[150] = "\185", [162] = "\153"},
        [194] = {
            [152] = "\152",
            [160] = "\160",
            [164] = "\164",
            [166] = "\166",
            [167] = "\167",
            [169] = "\169",
            [171] = "\171",
            [172] = "\172",
            [173] = "\173",
            [174] = "\174",
            [176] = "\176",
            [177] = "\177",
            [181] = "\181",
            [182] = "\182",
            [183] = "\183",
            [187] = "\187"
        },
        [208] = {
            [129] = "\168",
            [130] = "\128",
            [131] = "\129",
            [132] = "\170",
            [133] = "\189",
            [134] = "\178",
            [135] = "\175",
            [136] = "\163",
            [137] = "\138",
            [138] = "\140",
            [139] = "\142",
            [140] = "\141",
            [143] = "\143",
            [144] = "\192",
            [145] = "\193",
            [146] = "\194",
            [147] = "\195",
            [148] = "\196",
            [149] = "\197",
            [150] = "\198",
            [151] = "\199",
            [152] = "\200",
            [153] = "\201",
            [154] = "\202",
            [155] = "\203",
            [156] = "\204",
            [157] = "\205",
            [158] = "\206",
            [159] = "\207",
            [160] = "\208",
            [161] = "\209",
            [162] = "\210",
            [163] = "\211",
            [164] = "\212",
            [165] = "\213",
            [166] = "\214",
            [167] = "\215",
            [168] = "\216",
            [169] = "\217",
            [170] = "\218",
            [171] = "\219",
            [172] = "\220",
            [173] = "\221",
            [174] = "\222",
            [175] = "\223",
            [176] = "\224",
            [177] = "\225",
            [178] = "\226",
            [179] = "\227",
            [180] = "\228",
            [181] = "\229",
            [182] = "\230",
            [183] = "\231",
            [184] = "\232",
            [185] = "\233",
            [186] = "\234",
            [187] = "\235",
            [188] = "\236",
            [189] = "\237",
            [190] = "\238",
            [191] = "\239"
        },
        [209] = {
            [128] = "\240",
            [129] = "\241",
            [130] = "\242",
            [131] = "\243",
            [132] = "\244",
            [133] = "\245",
            [134] = "\246",
            [135] = "\247",
            [136] = "\248",
            [137] = "\249",
            [138] = "\250",
            [139] = "\251",
            [140] = "\252",
            [141] = "\253",
            [142] = "\254",
            [143] = "\255",
            [144] = "\161",
            [145] = "\184",
            [146] = "\144",
            [147] = "\131",
            [148] = "\186",
            [149] = "\190",
            [150] = "\179",
            [151] = "\191",
            [152] = "\188",
            [153] = "\154",
            [154] = "\156",
            [155] = "\158",
            [156] = "\157",
            [158] = "\162",
            [159] = "\159"
        },
        [210] = {[144] = "\165", [145] = "\180"}
    },
    _nmdc = {
        [36] = "$",
        [124] = "|"
    },
    toUTF8 = function(s)
    --  m4_assert(type(s) == "string")

        return (s:gsub(
            ".",
            function(c)
                local b = string.byte(c)
                return b < 128 and c or b > 239 and "\209" .. string.char(b - 112) or
                    b > 191 and "\208" .. string.char(b - 48) or
                    utf8._ansi_decode[b] or
                    "_"
            end
        ))
    end,
    toANSI = function(s)
    --  m4_assert(type(s) == "string")

        local a, j, r, b = 0, 0, ""
        for i = 1, string.len(s) do
            b = s:byte(i)
            if b < 128 then
                r = r .. (utf8._nmdc[b] or string.char(b))
            elseif a == 2 then
                a, j = a - 1, b
            elseif a == 1 then
                a, r = a - 1, r .. utf8._utf8_decode[j][b]
            elseif b == 226 then
                a = 2
            elseif b == 194 or b == 208 or b == 209 or b == 210 then
                j, a = b, 1
            else
                r = r .. "_"
            end
        end
        return r
    end
}

local utf8 = utf8

-- local testStringsANSI =
-- {  "Privet, kak dela?",          -- только ASCII
-- "\192\176\192\182\192\184",	-- кириллица АБВ в ANSI
-- "Test $| symbols",            -- с символами из _nmdc
-- "\200\201\202",                -- случайные ANSI символы
-- "Строка с русскими символами"
-- }

-- for _, s in ipairs(testStringsANSI) do
-- local utf = utf8.toUTF8(s)
-- local ansi = utf8.toANSI(utf)
-- print("ANSI:        ", s)
-- print("UTF-8:       ", utf)
-- print("Back to ANSI:", ansi)
-- print("-----------")
-- end

Форматирование строк Lua в Python стиле

Опубликовано в QLua

В Lua имеется функция string.format, позволяющая формировать строку с использованием значений переменных или вычисляемых значений. Можно ли создать механизм формирования строк, аналогичный тому, что используется в Python версии 3?

Архив QLua