#pragma once

#include <map>
#include<vector>
using namespace std;

#include "ElectricEqpmt_Telesignalisation.h"
#include "BaseDefine.h"

class StationEqpmt_Telesignalisation
{
public:
	StationEqpmt_Telesignalisation();
	~StationEqpmt_Telesignalisation();

	void AddStationEqpmtTelesignalisation( ElectricEqpmt_Telesignalisation& telesignalisation);

	StationEqpmt_Telesignalisation& operator=( const StationEqpmt_Telesignalisation& telesignalisation);

	void TelesignalisationInit( CommandQueue* pSendList, int nTrainID );

	void DealTelesignalisationMsg( CommandQueue* pSendList, int nTrainID, int nMark );

	int GetMark()
	{
		return m_nMark;
	}

	void SetMark( int nMark )
	{
		m_nMark = nMark;
	}

public:
	vector<ElectricEqpmt_Telesignalisation> m_vecTelesignalisation;
	int m_nMark;
};

class ElectricEqpmt_TelesignalisationMng
{
public:
	ElectricEqpmt_TelesignalisationMng();
	~ElectricEqpmt_TelesignalisationMng();

	void AddStationEqpmtTelesignalisation( ElectricEqpmt_Telesignalisation& telesignalisation, int nMark = 0 );

	string FormatKey( int nProtectID, const string& strEqpmtStrID );

	void DealTelesignalisationMsg( CommandQueue* pSendList, int nTrainID, int nProtectID, const string& strEqpmtStrID, int nMark );

	void TelesignalisationInit( CommandQueue* pSendList, int nTrainID );

public:
	map<string, StationEqpmt_Telesignalisation> m_mapStationEqpmtTelesignalisation;
};