#include "MvdDetIdSelector.h"

ClassImp(MvdDetIdSelector);

void MvdDetIdSelector::find(TString selString, bool newSel)
{
	TObjArray* tempArray;
	TObjArray result;
	if (newSel){
		tempArray = detNameArray;
	}
	else tempArray = &actSelection;
	
	for (int i = 0; i < tempArray->GetEntries(); i++){
		MvdDetIdPair* pair = (MvdDetIdPair*)(tempArray->At(i));
		TString name = pair->getName();
		if (name.Contains(selString.Data()))
			result.Add(pair);
	}
	actSelection = result;
}

void MvdDetIdSelector::exclude(TString selString, bool newSel)
{
	TObjArray* tempArray;
	TObjArray result;
	if (newSel){
		tempArray = detNameArray;
	}
	else tempArray = &actSelection;
	
	for (int i = 0; i < tempArray->GetEntries(); i++){
		MvdDetIdPair* pair = (MvdDetIdPair*)(tempArray->At(i));
		TString name = pair->getName();
		if (!name.Contains(selString.Data()))
			result.Add(pair);
	}
	actSelection = result;
}

std::vector<int> MvdDetIdSelector::GetSelectedIds()
{
	std::vector<int> result;
	for (int i = 0; i < actSelection.GetEntries(); i++){
		MvdDetIdPair* pair = (MvdDetIdPair*)(actSelection.At(i));
		result.push_back(pair->getId());
	}
	return result;
}