mirror of
https://github.com/mihakralj/QuanTAlib.git
synced 2026-08-12 15:48:05 +00:00
278 lines
6.7 KiB
Plaintext
278 lines
6.7 KiB
Plaintext
#!meta
|
|
|
|
{"kernelInfo":{"defaultKernelName":"csharp","items":[{"aliases":[],"name":"csharp"}]}}
|
|
|
|
#!csharp
|
|
|
|
#r ".\bin\Debug\calculations.dll"
|
|
using QuanTAlib;
|
|
|
|
#!csharp
|
|
|
|
public class CircularBuffer
|
|
{
|
|
private double[] _buffer;
|
|
private int _start;
|
|
private int _size;
|
|
|
|
public CircularBuffer(int capacity)
|
|
{
|
|
_buffer = new double[capacity];
|
|
_start = 0;
|
|
_size = 0;
|
|
}
|
|
|
|
public int Capacity => _buffer.Length;
|
|
public int Count => _size;
|
|
|
|
public void Add(double item, bool isNew)
|
|
{
|
|
if (!isNew)
|
|
{
|
|
// Add new item
|
|
if (_size < Capacity)
|
|
{
|
|
_buffer[(_start + _size) % Capacity] = item;
|
|
_size++;
|
|
}
|
|
else
|
|
{
|
|
_start = (_start + 1) % Capacity;
|
|
_buffer[(_start + _size - 1) % Capacity] = item;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
// Update the last item
|
|
if (_size > 0)
|
|
{
|
|
_buffer[(_start + _size - 1) % Capacity] = item;
|
|
}
|
|
else
|
|
{
|
|
// If buffer is empty, add the item even if isNew is true
|
|
_buffer[0] = item;
|
|
_size = 1;
|
|
}
|
|
}
|
|
}
|
|
|
|
public double this[int index]
|
|
{
|
|
get
|
|
{
|
|
if (index < 0 || index >= _size)
|
|
throw new IndexOutOfRangeException();
|
|
return _buffer[(_start + index) % Capacity];
|
|
}
|
|
set
|
|
{
|
|
if (index < 0 || index >= _size)
|
|
throw new IndexOutOfRangeException();
|
|
_buffer[(_start + index) % Capacity] = value;
|
|
}
|
|
}
|
|
}
|
|
|
|
#!csharp
|
|
|
|
public class SMA1
|
|
{
|
|
private CircularBuffer buffer;
|
|
private int period;
|
|
private double sum;
|
|
public TValue Value { get; private set; }
|
|
public bool IsHot => buffer.Count >= period;
|
|
|
|
public SMA1(int period)
|
|
{
|
|
Init(period);
|
|
}
|
|
|
|
public void Init(int period)
|
|
{
|
|
this.period = period;
|
|
this.buffer = new CircularBuffer(period);
|
|
this.sum = 0;
|
|
this.Value = default;
|
|
}
|
|
|
|
public TValue Update(TValue input, bool isNew = true)
|
|
{
|
|
if (!isNew)
|
|
{
|
|
if (buffer.Count == period)
|
|
{
|
|
sum -= buffer[0];
|
|
}
|
|
sum += input.Value;
|
|
buffer.Add(input.Value, isNew);
|
|
}
|
|
else
|
|
{
|
|
if (buffer.Count > 0)
|
|
{
|
|
sum -= buffer[buffer.Count - 1];
|
|
sum += input.Value;
|
|
buffer.Add(input.Value, isNew);
|
|
}
|
|
else
|
|
{
|
|
// If buffer is empty, add the item even if isNew is true
|
|
sum += input.Value;
|
|
buffer.Add(input.Value, false);
|
|
}
|
|
}
|
|
|
|
double sma = buffer.Count > 0 ? sum / buffer.Count : double.NaN;
|
|
Value = new TValue(input.Time, sma, isNew, IsHot);
|
|
return Value;
|
|
}
|
|
}
|
|
|
|
#!csharp
|
|
|
|
GBM_Feed feed = new(initialPrice: 100, mu: 0.1, sigma: 0.9);
|
|
int i=10;
|
|
SMA1 ma = new(i);
|
|
Console.WriteLine($"{"Close",10} {"MA(" + i + ")",10}");
|
|
for (int i = 0; i < 20; i++)
|
|
{
|
|
TValue c =(double)feed.Generate().Close;
|
|
ma.Update(1000,false);
|
|
ma.Update(-10000,false);
|
|
|
|
ma.Update(c,true);
|
|
|
|
Console.WriteLine($"{i+1} {(double)c,10:F2} {(double)ma.Value,10:F2}");
|
|
}
|
|
|
|
#!csharp
|
|
|
|
public class Emitter {
|
|
private Random random = new Random();
|
|
public event EventHandler<EventArg<TValue>> Pub;
|
|
public void Emit() {
|
|
DateTime now = DateTime.Now;
|
|
double randomValue = random.NextDouble() * 100; // Generates a random number between 0 and 100
|
|
TValue value = new TValue(now, randomValue);
|
|
|
|
EventArg<TValue> eventArg = new EventArg<TValue>(value, true, true);
|
|
OnValuePub(eventArg);
|
|
}
|
|
protected virtual void OnValuePub(EventArg<TValue> eventArg) {
|
|
Pub?.Invoke(this, eventArg);
|
|
}
|
|
}
|
|
|
|
public class BarEmitter
|
|
{
|
|
private Random random = new Random();
|
|
public event EventHandler<EventArg<TBar>> Pub;
|
|
private double lastClose = 100.0; // Starting price
|
|
|
|
public void Emit()
|
|
{
|
|
double open = lastClose;
|
|
double close = open * (1 + (random.NextDouble() - 0.5) * 0.02); // +/- 1% change
|
|
double high = Math.Max(open, close) * (1 + random.NextDouble() * 0.005); // Up to 0.5% higher
|
|
double low = Math.Min(open, close) * (1 - random.NextDouble() * 0.005); // Up to 0.5% lower
|
|
double volume = random.NextDouble() * 1000000; // Random volume between 0 and 1,000,000
|
|
|
|
TBar bar = new TBar(DateTime.Now, open, high, low, close, volume);
|
|
lastClose = close;
|
|
|
|
EventArg<TBar> eventArg = new EventArg<TBar>(bar, true, true);
|
|
OnBarPub(eventArg);
|
|
}
|
|
|
|
protected virtual void OnBarPub(EventArg<TBar> eventArg)
|
|
{
|
|
Pub?.Invoke(this, eventArg);
|
|
}
|
|
}
|
|
|
|
|
|
public class Listener
|
|
{
|
|
public void Sub(object sender, EventArgs e)
|
|
{
|
|
if (e is EventArg<TValue> tValueArg) {
|
|
Console.WriteLine($"TValue: {tValueArg.Data.Value:F2}");
|
|
} else if (e is EventArg<TBar> tBarArg) {
|
|
Console.WriteLine($"TBar: o={tBarArg.Data.Open:F2}, v={tBarArg.Data.Volume:F2}");
|
|
} else {
|
|
Console.WriteLine($"Unknown type: {e.GetType().Name}");
|
|
}
|
|
}
|
|
}
|
|
|
|
#!csharp
|
|
|
|
Emitter em1 = new();
|
|
BarEmitter em2 = new();
|
|
Listener list = new();
|
|
|
|
em1.Pub += list.Sub;
|
|
em2.Pub += list.Sub;
|
|
|
|
// Emit 5 random values
|
|
for (int i = 0; i < 3; i++) {
|
|
em1.Emit();
|
|
em2.Emit();
|
|
}
|
|
|
|
#!csharp
|
|
|
|
public abstract class Indicator {
|
|
protected Indicator() {
|
|
Init(); }
|
|
public virtual void Init() {}
|
|
public virtual TValue Calc(TValue input, bool isNew=true, bool isHot=true) {
|
|
return new TValue();
|
|
}
|
|
}
|
|
|
|
public class EMA : Indicator
|
|
{
|
|
private double lastEma, lastEmaCandidate, k;
|
|
private int period, i;
|
|
|
|
public EMA(int period) {
|
|
Init(period);
|
|
}
|
|
|
|
public void Init(int period)
|
|
{
|
|
this.period = period;
|
|
this.k = 2.0 / (period + 1);
|
|
this.lastEma = this.lastEmaCandidate = double.NaN;
|
|
this.i = 0;
|
|
}
|
|
|
|
public override TValue Calc(TValue input, bool isNew = true, bool isHot = true) {
|
|
double ema;
|
|
|
|
if (double.IsNaN(lastEma)) { lastEma = lastEmaCandidate = input.Value; }
|
|
|
|
if (isNew) {
|
|
lastEma = lastEmaCandidate;
|
|
i++;
|
|
}
|
|
|
|
double kk = (i>=period)?k:(2.0/(i+1));
|
|
ema = lastEma + kk * (input.Value - lastEma);
|
|
lastEmaCandidate = ema;
|
|
|
|
return new TValue(input.Timestamp, ema);
|
|
}
|
|
}
|
|
|
|
#!csharp
|
|
|
|
EMA ema = new(3);
|
|
display(ema.Calc(100));
|
|
display(ema.Calc(0,false));
|
|
display(ema.Calc(100,false));
|
|
display(ema.Calc(0));
|