PHGeomIOTGeo.cc

Go to the documentation of this file.

// $Id: $                                                                                             
 
#include "PHGeomIOTGeo.h"
 
#include <TGeoManager.h>
#include <TGeoVolume.h>
#include <TMemFile.h>
 
#include <cassert>
#include <sstream>
#include <iostream>
 
using namespace std;
 
PHGeomIOTGeo::PHGeomIOTGeo() :
    Data(0)
{
  SplitLevel(0); // this class is packed binary stream, no need to split
}
 
PHGeomIOTGeo::PHGeomIOTGeo(const PHGeomIOTGeo& geom) :
    Data(geom.Data)
{
}
 
PHGeomIOTGeo::~PHGeomIOTGeo()
{
  Reset();
}
 
PHObject*
PHGeomIOTGeo::clone() const
{
  PHGeomIOTGeo * geo = new PHGeomIOTGeo(*this);
  return geo;
}
 
void
PHGeomIOTGeo::SetGeometry(const TGeoVolume * g)
{
  if (!g)
    {
      cout << __PRETTY_FUNCTION__ << " - Error - Invalid input" << endl;
      return;
    }
 
  // Stream TGeoVolume into binary stream with its streamer using TFIle utility
  TMemFile f1("mem","CREATE");
  g->Write("TOP");
  f1.Close();
 
  const Long64_t n = f1.GetSize();
 
  Data.resize(n);
  Long64_t n1 = f1.CopyTo(Data.data(), n);
  assert(n1 == n);
 
}
 
TGeoVolume *
PHGeomIOTGeo::GetGeometryCopy()
{
  if (not isValid()) return NULL;
 
  TMemFile f2("mem2", Data.data(), Data.size(), "READ");
  TGeoVolume * vol = dynamic_cast<TGeoVolume *>(f2.Get("TOP"));
  assert(vol);
  f2.Close();
 
  return vol;
}
 
TGeoManager *
PHGeomIOTGeo::
ConstructTGeoManager()
{
  if (not isValid()) return NULL;
 
  // build new TGeoManager
  TGeoManager * tgeo = new TGeoManager("PHGeometry", "");
  assert(tgeo);
 
  TGeoVolume * vol = GetGeometryCopy();
  vol->RegisterYourself();
 
  tgeo->SetTopVolume(vol);
//  tgeo->CloseGeometry();
 
  stringstream stitle;
  stitle
      << "TGeoManager built by PHGeomUtility::LoadFromIONode based on RUN/GEOMETRY_IO node with name ("
      << vol->GetName() << ") and title ("
      << vol->GetTitle() << ")";
 
  tgeo->SetTitle(stitle.str().c_str());
 
  return tgeo;
}
 
void
PHGeomIOTGeo::identify(std::ostream& os) const
{
  os << "PHGeomIOTGeo - ";
  if (isValid())
    os << " with geometry data " << Data.size()<<"Byte";
  else
    os << "Empty";
  os << endl;
}
 
void
PHGeomIOTGeo::Reset()
{
  Data.resize(0);
}
 
int
PHGeomIOTGeo::isValid() const
{
  return Data.size();
}