using System;
using System.Threading;
namespace Preparation.Utility
{
//理论上结构体最好不可变,这里采用了可变结构。
//其对应属性不应当有set访问器,避免不安全的=赋值
public struct AtomicInt
{
private int v;
public AtomicInt(int x)
{
v = x;
}
public override string ToString() => Interlocked.CompareExchange(ref v, -1, -1).ToString();
public int Get() => Interlocked.CompareExchange(ref v, -1, -1);
public static implicit operator int(AtomicInt aint) => Interlocked.CompareExchange(ref aint.v, -1, -1);
public int Set(int value) => Interlocked.Exchange(ref v, value);
public int Add(int x) => Interlocked.Add(ref v, x);
public int Sub(int x) => Interlocked.Add(ref v, -x);
public int Inc() => Interlocked.Increment(ref v);
public int Dec() => Interlocked.Decrement(ref v);
public void CompareExchange(int newV, int compareTo) => Interlocked.CompareExchange(ref v, newV, compareTo);
/// 返回操作前的值
public int CompareExReturnOri(int newV, int compareTo) => Interlocked.CompareExchange(ref v, newV, compareTo);
}
public struct AtomicBool
{
private int v;//v==0为false,v!=0(v==1或v==-1)为true
public AtomicBool(bool x)
{
v = x ? 1 : 0;
}
public override string ToString() => (Interlocked.CompareExchange(ref v, -2, -2) == 0) ? "false" : "true";
public bool Get() => (Interlocked.CompareExchange(ref v, -1, -1) != 0);
public static implicit operator bool(AtomicBool abool) => (Interlocked.CompareExchange(ref abool.v, -1, -1) != 0);
public bool Set(bool value) => (Interlocked.Exchange(ref v, value ? 1 : 0) != 0);
/// 赋值前的值是否与将赋予的值不相同
public bool TrySet(bool value)
{
return (Interlocked.CompareExchange(ref v, value ? 1 : 0, value ? 0 : 1) ^ (value ? 1 : 0)) != 0;
}
public bool Invert() => Interlocked.Add(ref v, -1) != 0;
public bool And(bool x) => Interlocked.And(ref v, x ? 1 : 0) != 0;
public bool Or(bool x) => Interlocked.Or(ref v, x ? 1 : 0) != 0;
}
///
/// 一个能记录Start后完成多少进度的进度条(int),
/// 只允许Start时修改needTime(请确保大于0);
/// 支持TrySet0使未完成的进度条终止清零;支持Set0使进度条强制终止清零;
/// 不支持暂停
///
public struct IntProgressContinuously
{
private long endT = long.MaxValue;
private long needT;
public IntProgressContinuously(long needTime)
{
if (needTime <= 0) Debugger.Output("Bug:IntProgressContinuously.needTime (" + needTime.ToString() + ") is less than 0.");
this.needT = needTime;
}
public long GetEndTime() => Interlocked.CompareExchange(ref endT, -2, -2);
public long GetNeedTime() => Interlocked.CompareExchange(ref needT, -2, -2);
public override string ToString() => "EndTime:" + Interlocked.CompareExchange(ref endT, -2, -2).ToString() + " ms, NeedTime:" + Interlocked.CompareExchange(ref needT, -2, -2).ToString() + " ms";
public bool IsFinished()
{
return Interlocked.CompareExchange(ref endT, -2, -2) <= Environment.TickCount64;
}
///
/// GetProgress<0则表明未开始
///
public long GetProgress()
{
long cutime = Interlocked.CompareExchange(ref endT, -2, -2) - Environment.TickCount64;
if (cutime <= 0) return Interlocked.CompareExchange(ref needT, -2, -2);
return Interlocked.CompareExchange(ref needT, -2, -2) - cutime;
}
///
/// GetProgressDouble<0则表明未开始
///
public double GetProgressDouble()
{
long cutime = Interlocked.CompareExchange(ref endT, -2, -2) - Environment.TickCount64;
if (cutime <= 0) return 1;
return 1.0 - ((double)cutime / Interlocked.CompareExchange(ref needT, -2, -2));
}
public bool Start(long needTime)
{
if (needTime <= 0)
{
Debugger.Output("Warning:Start IntProgressContinuously with the needTime (" + needTime.ToString() + ") which is less than 0.");
return false;
}
//规定只有Start可以修改needT,且需要先访问endTime,从而避免锁(某种程度上endTime可以认为是needTime的锁)
if (Interlocked.CompareExchange(ref endT, Environment.TickCount64 + needTime, long.MaxValue) != long.MaxValue) return false;
if (needTime <= 2) Debugger.Output("Warning:the field of IntProgressContinuously is " + needTime.ToString() + ",which is too small.");
Interlocked.Exchange(ref this.needT, needTime);
return true;
}
public bool Start()
{
long needTime = Interlocked.CompareExchange(ref needT, -2, -2);
if (Interlocked.CompareExchange(ref endT, Environment.TickCount64 + needTime, long.MaxValue) != long.MaxValue) return false;
return true;
}
public void Set0() => Interlocked.Exchange(ref endT, long.MaxValue);
public bool TrySet0()
{
if (Environment.TickCount64 < Interlocked.CompareExchange(ref endT, -2, -2))
{
Interlocked.Exchange(ref endT, long.MaxValue);
return true;
}
return false;
}
//增加其他新的写操作可能导致不安全
}
///
/// 一个保证在[0,maxValue]的可变int,支持可变的maxValue(请确保大于0)
///
public struct IntWithVariableRange
{
private int v;
private int maxV;
private readonly object vLock = new();
public IntWithVariableRange(int value, int maxValue)
{
if (maxValue < 0)
{
Debugger.Output("Warning:Try to set IntWithVariableRange.maxValue to " + maxValue.ToString() + ".");
maxValue = 0;
}
v = value < maxValue ? value : maxValue;
this.maxV = maxValue;
}
public override string ToString()
{
lock (vLock)
{
return "value:" + v.ToString() + " ,maxValue:" + maxV.ToString();
}
}
public int GetValue() { lock (vLock) return v; }
public int GetMaxV() { lock (vLock) return maxV; }
///
/// 若maxValue<=0则maxValue设为0并返回False
///
public bool SetMaxV(int maxValue)
{
if (maxValue < 0) maxValue = 0;
lock (vLock)
{
maxV = maxValue;
if (v > maxValue) v = maxValue;
}
return maxValue > 0;
}
///
/// 应当保证该maxValue>=0
///
public void SetPositiveMaxV(int maxValue)
{
lock (vLock)
{
maxV = maxValue;
if (v > maxValue) v = maxValue;
}
}
///
/// 应当保证该value>=0
///
public int SetPositiveV(int value)
{
lock (vLock)
{
return v = (value > maxV) ? maxV : value;
}
}
public int SetV(int value)
{
if (value < 0) value = 0;
lock (vLock)
{
return v = (value > maxV) ? maxV : value;
}
}
public int AddV(int addV)
{
lock (vLock)
{
v += addV;
if (v < 0) return v = 0;
if (v > maxV) return v = maxV;
return v;
}
}
///
/// 应当保证该增加值大于0
///
public int AddPositiveV(int addPositiveV)
{
lock (vLock)
{
v += addPositiveV;
if (v > maxV) return v = maxV;
return v;
}
}
///
/// 应当保证该减少值大于0
///
public int SubPositiveV(int subPositiveV)
{
lock (vLock)
{
v += subPositiveV;
if (v < 0) return v = 0;
return v;
}
}
}
///
/// 一个保证在[0,maxNum],每CDms自动更新的可变int,支持可变的CD、maxNum(请确保大于0)
///
public struct IntNumUpdateByCD
{
private int num;
private int maxNum;
private int cd;
private long updateTime = 0;
private object numLock = new();
public IntNumUpdateByCD(int num, int maxNum, int cd)
{
if (num < 0) Debugger.Output("Bug:IntNumUpdateByCD.num (" + num.ToString() + ") is less than 0.");
if (maxNum < 0) Debugger.Output("Bug:IntNumUpdateByCD.maxNum (" + maxNum.ToString() + ") is less than 0.");
if (cd <= 0) Debugger.Output("Bug:IntNumUpdateByCD.cd (" + cd.ToString() + ") is less than 0.");
this.num = num;
this.maxNum = maxNum;
this.cd = cd;
this.updateTime = Environment.TickCount64;
}
public IntNumUpdateByCD(int maxNum, int cd)
{
if (maxNum < 0) Debugger.Output("Bug:IntNumUpdateByCD.maxNum (" + maxNum.ToString() + ") is less than 0.");
if (cd <= 0) Debugger.Output("Bug:IntNumUpdateByCD.cd (" + cd.ToString() + ") is less than 0.");
this.num = this.maxNum = maxNum;
this.cd = cd;
}
public int GetMaxNum() { lock (numLock) return maxNum; }
public int GetCD() { lock (numLock) return cd; }
public int GetNum(long time)
{
lock (numLock)
{
if (num < maxNum && time - updateTime >= cd)
{
int add = (int)Math.Min(maxNum - num, (time - updateTime) / cd);
updateTime += add * cd;
return (num += add);
}
return num;
}
}
///
/// 应当保证该subV>=0
///
public int TrySub(int subV)
{
if (subV < 0) Debugger.Output("Bug:IntNumUpdateByCD Try to sub " + subV.ToString() + ", which is less than 0.");
long time = Environment.TickCount64;
lock (numLock)
{
if (num < maxNum && time - updateTime >= cd)
{
int add = (int)Math.Min(maxNum - num, (time - updateTime) / cd);
updateTime += add * cd;
num += add;
}
if (num == maxNum) updateTime = time;
num -= subV = Math.Min(subV, num);
}
return subV;
}
///
/// 应当保证该addV>=0
///
public void TryAdd(int addV)
{
if (addV < 0) Debugger.Output("Bug:IntNumUpdateByCD Try to add " + addV.ToString() + ", which is less than 0.");
lock (numLock)
{
num += Math.Min(addV, maxNum - num);
}
}
///
/// 若maxNum<=0则maxNum及Num设为0并返回False
///
public bool SetMaxNumAndNum(int maxNum)
{
if (maxNum < 0) maxNum = 0;
lock (numLock)
{
this.num = this.maxNum = maxNum;
}
return maxNum > 0;
}
///
/// 应当保证该maxnum>=0
///
public void SetPositiveMaxNumAndNum(int maxNum)
{
lock (numLock)
{
this.num = this.maxNum = maxNum;
}
}
///
/// 应当保证该maxnum>=0
///
public void SetPositiveMaxNum(int maxNum)
{
lock (numLock)
{
if ((this.maxNum = maxNum) < num)
num = maxNum;
}
}
///
/// 若maxNum<=0则maxNum及Num设为0并返回False
///
public bool SetMaxNum(int maxNum)
{
if (maxNum < 0) maxNum = 0;
lock (numLock)
{
if ((this.maxNum = maxNum) < num)
num = maxNum;
}
return maxNum > 0;
}
///
/// 若num<0则num设为0并返回False
///
public bool SetNum(int num)
{
lock (numLock)
{
if (num < 0) { this.num = 0; return false; }
this.num = num;
return true;
}
}
///
/// 应当保证该num>=0
///
public void SetPositiveNum(int num)
{
lock (numLock)
{
this.num = num;
}
}
public void SetCD(int cd)
{
lock (numLock)
{
if (cd <= 0) Debugger.Output("Bug:Set IntNumUpdateByCD.cd to " + cd.ToString() + ".");
this.cd = cd;
}
}
}
}