New class - the factory for building mapping segmentations

[u/mrichter/AliRoot.git] / MUON / mapping / AliMpDEIterator.cxx

/**************************************************************************
 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 *                                                                        *
 * Author: The ALICE Off-line Project.                                    *
 * Contributors are mentioned in the code where appropriate.              *
 *                                                                        *
 * Permission to use, copy, modify and distribute this software and its   *
 * documentation strictly for non-commercial purposes is hereby granted   *
 * without fee, provided that the above copyright notice appears in all   *
 * copies and that both the copyright notice and this permission notice   *
 * appear in the supporting documentation. The authors make no claims     *
 * about the suitability of this software for any purpose. It is          *
 * provided "as is" without express or implied warranty.                  *
 **************************************************************************/

/* $Id$ */

#include "AliMpDEIterator.h"
#include "AliMpDEManager.h"
#include "AliMpFiles.h"

#include "AliLog.h"

#include <Riostream.h>
#include <TSystem.h>

ClassImp(AliMpDEIterator)

const  Int_t  AliMpDEIterator::fgkMaxNofDetElements = 250;
TArrayI  AliMpDEIterator::fgDetElemIds(fgkMaxNofDetElements);
Int_t    AliMpDEIterator::fgNofDetElemIds = 0;	

//
// static private methods
//

//______________________________________________________________________________
Bool_t AliMpDEIterator::ReadDEIds(AliMpStationType station)
{ 
/// Read det element ids from the file specified by name
/// and fill the map (the deNames are ignored)
/// Return true if the data were read ok

  // Open file
  TString filePath = AliMpFiles::DENamesFilePath(station);
  std::ifstream in(filePath);
  if (!in.good()) {
    AliErrorClassStream() << "Cannot open file " << filePath << endl;;
    return false;
  }
  
  // Skip plane types per cathods + empty lines
  //
  char line[80];
  in.getline(line, 80);
  in.getline(line, 80);
  in.getline(line, 80);
    
  // Read DE Ids
  //
  Int_t detElemId;
  TString word;
  in >> word;
  while ( ! in.eof() ) {
    if ( word[0] == '#' ) {
      in.getline(line, 80);
    }  
    else {  
      detElemId = word.Atoi();
      in.getline(line, 80);
      AliDebugClassStream(1) 
        << "Adding  " << fgNofDetElemIds << "  "  << detElemId << endl;
      fgDetElemIds.AddAt(detElemId, fgNofDetElemIds++);
    } 
    in >> word;
  }

  // Close file
  in.close();
  
  return true;
}

//______________________________________________________________________________
void AliMpDEIterator::ReadData()
{
/// Fill DE Ids array from DE names files
/// Return true if all data were read ok

  Bool_t result1 = ReadDEIds(kStation1);
  Bool_t result2 = ReadDEIds(kStation2);
  Bool_t result3 = ReadDEIds(kStation345);
  Bool_t result4 = ReadDEIds(kStationTrigger);
  
  Bool_t result = result1 && result2 && result3 && result4;
  if ( ! result ) {
    AliErrorClassStream() << "Error in reading DE names files" << endl;
  }  
}

//
// constructors, destructor
//

//______________________________________________________________________________
AliMpDEIterator::AliMpDEIterator()
    : TObject(),
      fIndex(-1),
      fModuleId(-1)
{  
/// Standard and default constructor

  if (! fgNofDetElemIds ) ReadData();
}

//______________________________________________________________________________
AliMpDEIterator::AliMpDEIterator(const AliMpDEIterator& rhs)
 : TObject(rhs),
   fIndex(rhs.fIndex),
   fModuleId(rhs.fModuleId)
{
/// Copy constructor
}

//______________________________________________________________________________

AliMpDEIterator::~AliMpDEIterator()
{
/// Destructor
}

//______________________________________________________________________________
AliMpDEIterator&  AliMpDEIterator::operator=(const AliMpDEIterator& rhs)
{
/// Assignement operator

  // check assignment to self
  if (this == &rhs) return *this;

  // base class assignment
  TObject::operator=(rhs);

  fIndex    = rhs.fIndex;
  fModuleId = rhs.fModuleId;

  return *this;
} 

//
// public methods
//

//______________________________________________________________________________
void AliMpDEIterator::First()
{
/// Set iterator to the first DE Id defined 

  fIndex = 0;
  fModuleId = -1;
}  

//______________________________________________________________________________
void AliMpDEIterator::First(Int_t moduleId)
{
 
  fModuleId = -1;
  fIndex = -1;  
  if ( ! AliMpDEManager::IsValidModuleId(moduleId) ) {
    AliErrorStream() << "Invalid module Id " << moduleId << endl;
    return;
  }    

  Int_t i=0;
  while ( i < fgNofDetElemIds && fModuleId < 0 ) {
    Int_t detElemId = fgDetElemIds.At(i);
    if ( AliMpDEManager::GetModuleId(detElemId) == moduleId ) {
      fModuleId = moduleId;
      fIndex = i;
    } 
    i++; 
  }

  if ( fModuleId < 0 ) {
    AliErrorStream() 
      << "No DEs of Module Id " << moduleId << " found" << cout;
    return;
  }    

}

//______________________________________________________________________________
void AliMpDEIterator::Next()
{
  fIndex++;

  // Invalidate if at the end
  if ( ( fIndex == fgNofDetElemIds ) ||
       ( fModuleId >= 0 &&    
         AliMpDEManager::GetModuleId(CurrentDE()) != fModuleId ) ) {
    fIndex = -1;
  }   
}

//______________________________________________________________________________
Bool_t AliMpDEIterator::IsDone() const
{
  return ( fIndex < 0 );
}   

//______________________________________________________________________________
Int_t AliMpDEIterator::CurrentDE() const
{
  if ( ! IsDone() )
    return fgDetElemIds.At(fIndex);
  else {   
    AliErrorStream()
      << "Not in valid position - returning invalid DE." << endl;
    return 0;
  }  
}