VariableRearrangeTransform.cxx

Go to the documentation of this file.

// @(#)root/tmva $Id$
// Author: Andreas Hoecker, Joerg Stelzer, Helge Voss, Peter Speckmayer
 
/**********************************************************************************
 * Project: TMVA - a Root-integrated toolkit for multivariate data analysis       *
 * Package: TMVA                                                                  *
 * Class  : VariableRearrangeTransform                                            *
 *                                             *
 *                                                                                *
 * Description:                                                                   *
 *      Implementation (see header for description)                               *
 *                                                                                *
 * Authors (alphabetical):                                                        *
 *      Peter Speckmayer <Peter.Speckmayer@cern.ch> - CERN, Switzerland           *
 *                                                                                *
 * Copyright (c) 2005:                                                            *
 *      CERN, Switzerland                                                         *
 *      MPI-K Heidelberg, Germany                                                 *
 *                                                                                *
 * Redistribution and use in source and binary forms, with or without             *
 * modification, are permitted according to the terms listed in LICENSE           *
 * (see tmva/doc/LICENSE)                                          *
 **********************************************************************************/
 
/*! \class TMVA::VariableRearrangeTransform
\ingroup TMVA
Rearrangement of input variables
*/
 
#include "TMVA/VariableRearrangeTransform.h"
 
#include "TMVA/DataSet.h"
#include "TMVA/Event.h"
#include "TMVA/MsgLogger.h"
#include "TMVA/Tools.h"
#include "TMVA/Types.h"
 
#include <iostream>
#include <stdexcept>
 
 
////////////////////////////////////////////////////////////////////////////////
/// constructor
 
TMVA::VariableRearrangeTransform::VariableRearrangeTransform( DataSetInfo& dsi )
:  VariableTransformBase( dsi, Types::kRearranged, "Rearrange" )
{
}
 
////////////////////////////////////////////////////////////////////////////////
 
TMVA::VariableRearrangeTransform::~VariableRearrangeTransform() {
}
 
////////////////////////////////////////////////////////////////////////////////
/// initialization of the rearrangement transformation
/// (nothing to do)
 
void TMVA::VariableRearrangeTransform::Initialize()
{
}
 
////////////////////////////////////////////////////////////////////////////////
/// prepare transformation --> (nothing to do)
 
Bool_t TMVA::VariableRearrangeTransform::PrepareTransformation (const std::vector<Event*>& /*events*/)
{
   if (!IsEnabled() || IsCreated()) return kTRUE;
 
   UInt_t nvars = 0, ntgts = 0, nspcts = 0;
   CountVariableTypes( nvars, ntgts, nspcts );
   if (ntgts>0) Log() << kFATAL << "Targets used in Rearrange-transformation." << Endl;
 
   SetCreated( kTRUE );
   return kTRUE;
}
 
////////////////////////////////////////////////////////////////////////////////
 
const TMVA::Event* TMVA::VariableRearrangeTransform::Transform( const TMVA::Event* const ev, Int_t /*cls*/ ) const
{
   if (!IsEnabled()) return ev;
 
   // apply the normalization transformation
   if (!IsCreated()) Log() << kFATAL << "Transformation not yet created" << Endl;
 
   if (fTransformedEvent==0) fTransformedEvent = new Event();
 
   FloatVector input; // will be filled with the selected variables, (targets)
   std::vector<Char_t> mask; // masked variables
   GetInput( ev, input, mask );
   SetOutput( fTransformedEvent, input, mask, ev );
 
   return fTransformedEvent;
}
 
////////////////////////////////////////////////////////////////////////////////
 
const TMVA::Event* TMVA::VariableRearrangeTransform::InverseTransform( const TMVA::Event* const ev, Int_t /*cls*/ ) const
{
   if (!IsEnabled()) return ev;
 
   // apply the inverse transformation
   if (!IsCreated()) Log() << kFATAL << "Transformation not yet created" << Endl;
 
   if (fBackTransformedEvent==0) fBackTransformedEvent = new Event( *ev );
 
   FloatVector input;  // will be filled with the selected variables, targets, (spectators)
   std::vector<Char_t> mask; // masked variables
   GetInput( ev, input, mask, kTRUE );
   SetOutput( fBackTransformedEvent, input, mask, ev, kTRUE );
 
   return fBackTransformedEvent;
}
 
 
////////////////////////////////////////////////////////////////////////////////
/// creates string with variable transformations applied
 
std::vector<TString>* TMVA::VariableRearrangeTransform::GetTransformationStrings( Int_t /*cls*/ ) const
{
   const UInt_t size = fGet.size();
   std::vector<TString>* strVec = new std::vector<TString>(size);
 
   return strVec;
}
 
////////////////////////////////////////////////////////////////////////////////
/// create XML description of Rearrange transformation
 
void TMVA::VariableRearrangeTransform::AttachXMLTo(void* parent)
{
   void* trfxml = gTools().AddChild(parent, "Transform");
   gTools().AddAttr(trfxml, "Name", "Rearrange");
 
   VariableTransformBase::AttachXMLTo( trfxml );
}
 
////////////////////////////////////////////////////////////////////////////////
/// Read the transformation matrices from the xml node
 
void TMVA::VariableRearrangeTransform::ReadFromXML( void* trfnode )
{
 
   void* inpnode = NULL;
 
   inpnode = gTools().GetChild(trfnode, "Selection"); // new xml format
   if(inpnode == NULL)
      Log() << kFATAL << "Unknown weight file format for transformations. (tried to read in 'rearrange' transform)" << Endl;
 
   VariableTransformBase::ReadFromXML( inpnode );
 
   SetCreated();
}
 
////////////////////////////////////////////////////////////////////////////////
/// prints the transformation ranges
 
void TMVA::VariableRearrangeTransform::PrintTransformation( std::ostream& )
{
}
 
////////////////////////////////////////////////////////////////////////////////
/// creates a normalizing function
 
void TMVA::VariableRearrangeTransform::MakeFunction( std::ostream& /*fout*/, const TString& /*fcncName*/,
                                                     Int_t /*part*/, UInt_t /*trCounter*/, Int_t )
{
}