emtra.cpp File Reference

#include <string.h>
#include <iostream>
#include <TEnv.h>
#include "EdbLog.h"
#include "EdbScanProc.h"
#include "EdbScanTracking.h"

Include dependency graph for emtra.cpp:

Functions
int	main (int argc, char *argv[])
void	MakeEraseFiles (EdbID id, TEnv &env)
void	print_help_message ()
void	set_default (TEnv &cenv)

Function Documentation

main()

int main	(	int	argc,
		char *	argv[]
	)

{
  if (argc < 2)   { print_help_message();  return 0; }
  
  TEnv cenv("trackenv");
  set_default(cenv);
  gEDBDEBUGLEVEL        = cenv.GetValue("emtra.EdbDebugLevel" , 1);
  const char *env       = cenv.GetValue("emtra.env"            , "track.rootrc");
  const char *outdir    = cenv.GetValue("emtra.outdir"         , "..");
 
  bool      do_set              = false;
  bool      do_new              = false;
  bool      do_pred             = false;
  bool      do_make_erase_file  = false;
  bool      do_VSB              = false;
  Int_t     pred_plate  = 0, to_plate=0;
  Int_t     brick=0, plate=0, major=0, minor=0;
  Int_t     ix = -1, iy = -1; //cell numbers
 
  for(int i=1; i<argc; i++ ) {
    char *key  = argv[i];
 
    if(!strncmp(key,"-set=",5))
      {
    if(strlen(key)>5)   sscanf(key+5,"%d.%d.%d.%d",&brick,&plate,&major,&minor);
    do_set=true;
      }
    else if(!strncmp(key,"-o=",3)) 
      {
    if(strlen(key)>3)   outdir=key+3;
      }
    else if(!strncmp(key,"-new",4)) 
      {
    do_new=true;
      }
    else if(!strncmp(key,"-pred=",6))
      {
    if(strlen(key)>6)   {
      pred_plate = atoi(key+6);
      do_pred=true;
    }
      }
    else if(!strncmp(key,"-makeerasefile",14))
      {
    do_make_erase_file=true;
      }
     else if(!strncmp(key,"-VSB=",5))
      {
    if(strlen(key)>5)  {
      do_VSB=true;
      to_plate = atoi(key+5);
    }
      }
    else if(!strncmp(key,"-v=",3))
      {
    if(strlen(key)>3)   gEDBDEBUGLEVEL = atoi(key+3);
      }
    else if(!strncmp(key,"-ix=",4))
      {
    if(strlen(key)>4)   ix = atoi(key+4);
      }
    else if(!strncmp(key,"-iy=",4))
      {
    if(strlen(key)>4)   iy = atoi(key+4);
      }
  }
 
  if(!do_set)   { print_help_message(); return 0; }
 
  cenv.SetValue("emtra.env"                  , env);
  cenv.ReadFile( cenv.GetValue("emtra.env"   , "track.rootrc") ,kEnvLocal);
  cenv.SetValue("emtra.outdir"               , outdir);
  cenv.WriteFile("track.save.rootrc");
 
  if(do_set&&(!do_VSB)) {
      EdbScanProc sproc;
      sproc.eProcDirClient=outdir;
      printf("\n----------------------------------------------------------------------------\n");
      printf("tracking set %d.%d.%d.%d\n", brick,plate, major,minor);
      printf("----------------------------------------------------------------------------\n\n");
 
      EdbID id(brick,plate,major,minor);
      EdbScanSet *ss = sproc.ReadScanSet(id);
      ss->Brick().SetID(brick);
      if(do_new) 
      {
          EdbScanTracking est;
          est.eSproc=&sproc;
          est.TrackSetBT(id,cenv,ix,iy);
      }
      else if(do_make_erase_file){
        MakeEraseFiles(id, cenv);
      }
      else {
    //ss->MakePIDList();
    //sproc.AssembleScanSet(*ss);
 
//    TCut c = gEnv->GetValue("emtra.cpcut" ,"s.eW>13&&eCHI2P<2.5&&s1.eFlag>=0&&s2.eFlag>=0&&eN1==1&&eN2==1");
//    EdbScanCond cond;
//    cond.Print();
//    sproc.TrackSetBT(*ss,cond,c);
    
          sproc.TrackSetBT(*ss,cenv);
      }
  }
  if(do_VSB) {
    EdbScanProc sproc;
    sproc.eProcDirClient=outdir;
    printf("\n----------------------------------------------------------------------------\n");
    printf("do virtual scanback in %d.%d.%d.%d  for the plate %d\n", brick,plate, major,minor, pred_plate);
    printf("----------------------------------------------------------------------------\n\n");
 
    EdbID id(brick,plate,major,minor);
    EdbScanSet *ss = sproc.ReadScanSet(id);
    //ss->Brick().SetID(brick);
    //sproc.AssembleScanSet(*ss);
 
 
    EdbID predid(id);
    predid.ePlate = pred_plate;
 
    //sproc.FindPredictionsRaw(predid,predid);
 
    sproc.FindPredictionsRawSet(predid, *ss, to_plate);
    EdbPVRec ali;
    sproc.ReadFoundTracks(*ss, ali);
    sproc.WriteSBTracks(*(ali.eTracks), id);
    
 
    /*
    TCut c = gEnv->GetValue("emtra.trcut" ,"nseg>2");
 
 
    EdbLayer *layer  = ss->GetPlate(pred_plate);
 
    EdbPattern pred, fnd;
    sproc.ReadPred(pred, predid);
    EdbRunAccess ra;
    sproc.InitRunAccess(ra, predid);
 
    EdbScanCond condBT, condMT;
    sproc.FindPredictionsRaw( pred, fnd, ra, condBT, condMT );
    
    */
 
  }
  cenv.WriteFile("track.save.rootrc");
 
  return 1;
}

MakeEraseFiles()

void MakeEraseFiles	(	EdbID	id,
		TEnv &	env
	)

{
  EdbScanProc  sproc;
  sproc.eProcDirClient = cenv.GetValue("emtra.outdir"         , "..");
  EdbPVRec     ali;
  sproc.ReadTracksTree(id, ali, "1");
 
  for(int i=0; i<60; i++)
    if( ali.GetPattern(i) ) {
      EdbPattern *pat = ali.GetPattern(i);
      //pat->SetPID( pat->GetSegment(0)->Vid(0));
      pat->SetPID( pat->GetSegment(0)->Plate() );
      printf(" %d PID=%d ID=%d nseg=%d\n", i, pat->PID(),  pat->ID(), pat->N() );
      
      EdbID idp(id); idp.ePlate=pat->PID();
      sproc.MakeEraseFile(idp, *pat);
    }
 
}

print_help_message()

void print_help_message

(

)

{
  cout<< "\nUsage: \n\t  emtra -set=ID [ -o=DATA_DIRECTORY -v=DEBUG] \n\n";
  cout<< "\t\t  ID              - id of the dataset formed as BRICK.PLATE.MAJOR.MINOR \n";
  cout<< "\t\t  DEBUG           - verbosity level: 0-print nothing, 1-errors only, 2-normal, 3-print all messages\n";
  cout<< "\t\t  -m              - make the affine files starting from EdbScanSet\n";
  cout<< "\t\t  -new            - apply new tracking algorithm\n";
  cout<< "\t\t  -VSB=plate         - \n";
  cout<< "\t\t  -makeerasefile  - to create files *.er.root with the list of basetracks from the trk.root to be excluded from the next tracking\n";
  cout<< "\t\t  -o              - the data directory\n";
  cout<< "\nExample: \n";
  cout<< "\t  emtra -id=4554.10.1.0 -o=/scratch/BRICKS \n";
  cout<< "\n If the data location directory if not explicitly defined\n";
  cout<< " the current directory will be assumed to be the brick directory \n";
  cout<<endl;
}

set_default()

void set_default

(

TEnv &

cenv

)

{
  // default parameters for tracking
  cenv.SetValue("fedra.readCPcut"       , "eCHI2P<2.5&&s.eW>13&&eN1==1&&eN2==1&&s1.eFlag>=0&&s2.eFlag>=0");
  cenv.SetValue("fedra.track.TrZmap", "2400 0 120000   2000 0 100000   30" );
  cenv.SetValue("fedra.track.npass"     ,    1 );
  cenv.SetValue("fedra.track.minPlate"  ,-999 );
  cenv.SetValue("fedra.track.maxPlate"  , 999 );
  cenv.SetValue("fedra.track.refPlate"  , 999 );
  cenv.SetValue("fedra.track.nsegmin"   , 2 );
  cenv.SetValue("fedra.track.ngapmax"   , 4 );
  cenv.SetValue("fedra.track.DZGapMax"  , 5000. );
  cenv.SetValue("fedra.track.DRmax"     , 45. );
  cenv.SetValue("fedra.track.DTmax"     , 0.07 );
  
  cenv.SetValue("fedra.track.PredictionCutXY"   ,  "0 0. 0. 0. 0.");   // doit x0 y0 dx dy
  cenv.SetValue("fedra.track.PredictionCutZ"    ,  "0 0. 0. 0.");      // doit z0 zmin zmax
  cenv.SetValue("fedra.track.PredictionCutAng"  ,  "0 0. 0. 0. 0.");   // doit tx ty dtx dty
  cenv.SetValue("fedra.track.Sigma0" , "3 3 0.005 0.005");
  cenv.SetValue("fedra.track.PulsRamp0"      , "15 20");
  cenv.SetValue("fedra.track.PulsRamp04"     , "15 20");
  cenv.SetValue("fedra.track.Degrad"         ,  4   );
 
  cenv.SetValue("fedra.track.probmin"   , 0.001 );
  cenv.SetValue("fedra.track.momentum"  , 2 );
  cenv.SetValue("fedra.track.mass"      , 0.14 );
  cenv.SetValue("fedra.track.do_use_mcs", 0 );
  cenv.SetValue("fedra.track.RadX0"     , 5810.);
 
  cenv.SetValue("fedra.track.do_track"       , true  );
  cenv.SetValue("fedra.track.do_shtag"       , false );
  cenv.SetValue("fedra.track.erase"          , false );
  cenv.SetValue("fedra.track.do_realign"     , 0 );
  cenv.SetValue("fedra.track.do_misalign"    , false );
  cenv.SetValue("fedra.track.do_misalign"    , false );
  cenv.SetValue("fedra.track.misalign_offset", 500.);
  cenv.SetValue("fedra.track.do_local_corr"  ,    1 );
  cenv.SetValue("fedra.track.do_comb"        ,    0 );
  cenv.SetValue("fedra.track.NsegMin"        ,    2 );
  cenv.SetValue("emtra.outdir"          , "..");
  cenv.SetValue("emtra.env"             , "track.rootrc");
  cenv.SetValue("emtra.EdbDebugLevel"   , 1);
}