#ifndef KRATTA_GEO_H
#define KRATTA_GEO_H

#include "FairGeoSet.h"

class  KrattaGeo : public FairGeoSet {
protected:
  char modName[20];  // name of module
  char eleName[20];  // substring for elements in module
public:
  KrattaGeo();
  ~KrattaGeo() {}
  const char* getModuleName(Int_t);
  const char* getEleName(Int_t);
  inline Int_t getModNumInMod(const TString&);
  ClassDef(KrattaGeo,0) // Class for STS
};

inline Int_t KrattaGeo::getModNumInMod(const TString& name) {
  // returns the module index from module name
  return (Int_t)(name[3]-'0')-1;
}

#endif  /* !KRATTA_GEO_H */