1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45
|
/* This file is part of the Spring engine (GPL v2 or later), see LICENSE.html */
#ifndef _NANO_PIECE_CACHE_H
#define _NANO_PIECE_CACHE_H
#include <vector>
#include "Sim/Misc/GlobalConstants.h"
#include "System/bitops.h"
#include "System/creg/creg_cond.h"
class CUnitScript;
struct NanoPieceCache
{
CR_DECLARE_STRUCT(NanoPieceCache)
public:
NanoPieceCache(): lastNanoPieceCnt(0), curBuildPowerMask(0) {
}
void Update() { curBuildPowerMask >>= 1; }
float GetBuildPower() const { return (count_bits_set(curBuildPowerMask) / float(UNIT_SLOWUPDATE_RATE)); }
/// returns modelPiece (NOT scriptModelPiece)
int GetNanoPiece(CUnitScript* ownerScript);
const std::vector<int>& GetNanoPieces() const { return nanoPieces; }
std::vector<int>& GetNanoPieces() { return nanoPieces; }
void StopPolling() { lastNanoPieceCnt = MAX_QUERYNANOPIECE_CALLS * 2; }
private:
// model-piece indices
std::vector<int> nanoPieces;
static const int MAX_QUERYNANOPIECE_CALLS = 30;
int lastNanoPieceCnt;
int curBuildPowerMask;
};
#endif
|