RooWorkspace.cxx

Go to the documentation of this file.

/*****************************************************************************
 * Project: RooFit                                                           *
 * Package: RooFitCore                                                       *
 * @(#)root/roofitcore:$Id$
 * Authors:                                                                  *
 *   WV, Wouter Verkerke, UC Santa Barbara, verkerke@slac.stanford.edu       *
 *                                                                           *
 * Copyright (c) 2000-2005, Regents of the University of California          *
 *                          and Stanford University. All rights reserved.    *
 *                                                                           *
 * Redistribution and use in source and binary forms,                        *
 * with or without modification, are permitted according to the terms        *
 * listed in LICENSE (http://roofit.sourceforge.net/license.txt)             *
 *****************************************************************************/
 
/**
\file RooWorkspace.cxx
\class RooWorkspace
\ingroup Roofitcore
 
Persistable container for RooFit projects. A workspace
can contain and own variables, p.d.f.s, functions and datasets. All objects
that live in the workspace are owned by the workspace. The `import()` method
enforces consistency of objects upon insertion into the workspace (e.g. no
duplicate object with the same name are allowed) and makes sure all objects
in the workspace are connected to each other. Easy accessor methods like
`pdf()`, `var()` and `data()` allow to refer to the contents of the workspace by
object name. The entire RooWorkspace can be saved into a ROOT TFile and organises
the consistent streaming of its contents without duplication.
If a RooWorkspace contains custom classes, i.e. classes not in the
ROOT distribution, portability of workspaces can be enhanced by
storing the source code of those classes in the workspace as well.
This process is also organized by the workspace through the
`importClassCode()` method.
 
### Seemingly random crashes when reading large workspaces
When reading or loading workspaces with deeply nested PDFs, one can encounter
ouf-of-memory errors if the stack size is too small. This manifests in crashes
at seemingly random locations, or in the process silently ending.
Unfortunately, ROOT neither recover from this situation, nor warn or give useful
instructions. When suspecting to have run out of stack memory, check
```
ulimit -s
```
and try reading again.
**/
 
#include <RooWorkspace.h>
 
#include <RooAbsData.h>
#include <RooAbsPdf.h>
#include <RooAbsStudy.h>
#include <RooCategory.h>
#include <RooCmdConfig.h>
#include <RooConstVar.h>
#include <RooFactoryWSTool.h>
#include <RooLinkedListIter.h>
#include <RooMsgService.h>
#include <RooPlot.h>
#include <RooRandom.h>
#include <RooRealVar.h>
#include <RooResolutionModel.h>
#include <RooTObjWrap.h>
#include <RooWorkspaceHandle.h>
 
#include "TBuffer.h"
#include "TInterpreter.h"
#include "TClassTable.h"
#include "TBaseClass.h"
#include "TSystem.h"
#include "TRegexp.h"
#include "TROOT.h"
#include "TFile.h"
#include "TH1.h"
#include "TClass.h"
#include "strlcpy.h"
 
#ifdef ROOFIT_LEGACY_EVAL_BACKEND
#include "RooAbsOptTestStatistic.h"
#endif
 
#include "ROOT/StringUtils.hxx"
 
#include <map>
#include <sstream>
#include <string>
#include <iostream>
#include <fstream>
#include <cstring>
 
namespace {
 
// Infer from a RooArgSet name whether this set is used internally by
// RooWorkspace to cache things.
bool isCacheSet(std::string const& setName) {
   // Check if the setName starts with CACHE_.
   return setName.rfind("CACHE_", 0) == 0;
}
 
} // namespace
 
using std::string, std::list, std::map, std::vector, std::ifstream, std::ofstream, std::fstream, std::make_unique;
 
 
////////////////////////////////////////////////////////////////////////////////
 
 
////////////////////////////////////////////////////////////////////////////////
 
 
list<string> RooWorkspace::_classDeclDirList ;
list<string> RooWorkspace::_classImplDirList ;
string RooWorkspace::_classFileExportDir = ".wscode.%s.%s" ;
bool RooWorkspace::_autoClass = false ;
 
 
////////////////////////////////////////////////////////////////////////////////
/// Add `dir` to search path for class declaration (header) files. This is needed
/// to find class headers custom classes are imported into the workspace.
void RooWorkspace::addClassDeclImportDir(const char* dir)
{
  _classDeclDirList.push_back(dir) ;
}
 
 
////////////////////////////////////////////////////////////////////////////////
/// Add `dir` to search path for class implementation (.cxx) files. This is needed
/// to find class headers custom classes are imported into the workspace.
void RooWorkspace::addClassImplImportDir(const char* dir)
{
  _classImplDirList.push_back(dir) ;
}
 
 
////////////////////////////////////////////////////////////////////////////////
/// Specify the name of the directory in which embedded source
/// code is unpacked and compiled. The specified string may contain
/// one '%s' token which will be substituted by the workspace name
 
void RooWorkspace::setClassFileExportDir(const char* dir)
{
  if (dir) {
    _classFileExportDir = dir ;
  } else {
    _classFileExportDir = ".wscode.%s.%s" ;
  }
}
 
 
////////////////////////////////////////////////////////////////////////////////
/// If flag is true, source code of classes not the ROOT distribution
/// is automatically imported if on object of such a class is imported
/// in the workspace
 
void RooWorkspace::autoImportClassCode(bool flag)
{
  _autoClass = flag ;
}
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Default constructor
 
RooWorkspace::RooWorkspace() : _classes(this)
{
}
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Construct empty workspace with given name and title
 
RooWorkspace::RooWorkspace(const char* name, const char* title) :
  TNamed(name,title?title:name), _classes(this)
{
}
 
////////////////////////////////////////////////////////////////////////////////
/// Construct empty workspace with given name and option to export reference to
/// all workspace contents to a CINT namespace with the same name.
 
RooWorkspace::RooWorkspace(const char* name, bool /*doCINTExport*/)  :
  TNamed(name,name), _classes(this)
{
}
 
 
////////////////////////////////////////////////////////////////////////////////
/// Workspace copy constructor
 
RooWorkspace::RooWorkspace(const RooWorkspace& other) :
  TNamed(other), _uuid(other._uuid), _classes(other._classes,this)
{
  // Copy owned nodes
  other._allOwnedNodes.snapshot(_allOwnedNodes,true) ;
 
  // Copy datasets
  for(TObject *data2 : other._dataList) _dataList.Add(data2->Clone());
 
  // Copy snapshots
  for(auto * snap : static_range_cast<RooArgSet*>(other._snapshots)) {
    auto snapClone = new RooArgSet;
    snap->snapshot(*snapClone);
    snapClone->setName(snap->GetName()) ;
    _snapshots.Add(snapClone) ;
  }
 
  // Copy named sets
  for (map<string,RooArgSet>::const_iterator iter3 = other._namedSets.begin() ; iter3 != other._namedSets.end() ; ++iter3) {
    // Make RooArgSet with equivalent content of this workspace
    _namedSets[iter3->first].add(*std::unique_ptr<RooArgSet>{_allOwnedNodes.selectCommon(iter3->second)});
  }
 
  // Copy generic objects
  for(TObject * gobj : other._genObjects) {
    TObject *theClone = gobj->Clone();
 
    auto handle = dynamic_cast<RooWorkspaceHandle*>(theClone);
    if (handle) {
      handle->ReplaceWS(this);
    }
 
    _genObjects.Add(theClone);
  }
}
 
 
/// TObject::Clone() needs to be overridden.
TObject *RooWorkspace::Clone(const char *newname) const
{
   auto out = new RooWorkspace{*this};
   if(newname && std::string(newname) != GetName()) {
      out->SetName(newname);
   }
   return out;
}
 
 
////////////////////////////////////////////////////////////////////////////////
/// Workspace destructor
 
RooWorkspace::~RooWorkspace()
{
  // Delete contents
  _dataList.Delete() ;
  if (_dir) {
    delete _dir ;
  }
  _snapshots.Delete() ;
 
  // WVE named sets too?
 
  _genObjects.Delete() ;
 
   _embeddedDataList.Delete();
   _views.Delete();
   _studyMods.Delete();
 
}
 
 
////////////////////////////////////////////////////////////////////////////////
/// Import a RooAbsArg or RooAbsData set from a workspace in a file. Filespec should be constructed as "filename:wspacename:objectname"
/// The arguments will be passed to the relevant import() or import(RooAbsData&, ...) import calls
/// \note From python, use `Import()`, since `import` is a reserved keyword.
bool RooWorkspace::import(const char* fileSpec,
             const RooCmdArg& arg1, const RooCmdArg& arg2, const RooCmdArg& arg3,
             const RooCmdArg& arg4, const RooCmdArg& arg5, const RooCmdArg& arg6,
             const RooCmdArg& arg7, const RooCmdArg& arg8, const RooCmdArg& arg9)
{
  // Parse file/workspace/objectname specification
  std::vector<std::string> tokens = ROOT::Split(fileSpec, ":");
 
  // Check that parsing was successful
  if (tokens.size() != 3) {
    std::ostringstream stream;
    for (const auto& token : tokens) {
      stream << "\n\t" << token;
    }
    coutE(InputArguments) << "RooWorkspace(" << GetName() << ") ERROR in file specification, expecting 'filename:wsname:objname', but '" << fileSpec << "' given."
        << "\nTokens read are:" << stream.str() << std::endl;
    return true ;
  }
 
  const std::string& filename = tokens[0];
  const std::string& wsname = tokens[1];
  const std::string& objname = tokens[2];
 
  // Check that file can be opened
  std::unique_ptr<TFile> f{TFile::Open(filename.c_str())};
  if (f==nullptr) {
    coutE(InputArguments) << "RooWorkspace(" << GetName() << ") ERROR opening file " << filename << std::endl ;
    return false;
  }
 
  // That that file contains workspace
  RooWorkspace* w = dynamic_cast<RooWorkspace*>(f->Get(wsname.c_str())) ;
  if (w==nullptr) {
    coutE(InputArguments) << "RooWorkspace(" << GetName() << ") ERROR: No object named " << wsname << " in file " << filename
           << " or object is not a RooWorkspace" << std::endl ;
    return false;
  }
 
  // Check that workspace contains object and forward to appropriate import method
  RooAbsArg* warg = w->arg(objname.c_str()) ;
  if (warg) {
    bool ret = import(*warg,arg1,arg2,arg3,arg4,arg5,arg6,arg7,arg8,arg9) ;
    return ret ;
  }
  RooAbsData* wdata = w->data(objname.c_str()) ;
  if (wdata) {
    bool ret = import(*wdata,arg1,arg2,arg3,arg4,arg5,arg6,arg7,arg8,arg9) ;
    return ret ;
  }
 
  coutE(InputArguments) << "RooWorkspace(" << GetName() << ") ERROR: No RooAbsArg or RooAbsData object named " << objname
         << " in workspace " << wsname << " in file " << filename << std::endl ;
  return true ;
}
 
 
////////////////////////////////////////////////////////////////////////////////
/// Import multiple RooAbsArg objects into workspace. For details on arguments see documentation
/// of import() method for single RooAbsArg
/// \note From python, use `Import()`, since `import` is a reserved keyword.
bool RooWorkspace::import(const RooArgSet& args,
             const RooCmdArg& arg1, const RooCmdArg& arg2, const RooCmdArg& arg3,
             const RooCmdArg& arg4, const RooCmdArg& arg5, const RooCmdArg& arg6,
             const RooCmdArg& arg7, const RooCmdArg& arg8, const RooCmdArg& arg9)
{
  bool ret(false) ;
  for(RooAbsArg * oneArg : args) {
    ret |= import(*oneArg,arg1,arg2,arg3,arg4,arg5,arg6,arg7,arg8,arg9) ;
  }
  return ret ;
}
 
 
 
////////////////////////////////////////////////////////////////////////////////
///  Import a RooAbsArg object, e.g. function, p.d.f or variable into the workspace. This import function clones the input argument and will
///  own the clone. If a composite object is offered for import, e.g. a p.d.f with parameters and observables, the
///  complete tree of objects is imported. If any of the _variables_ of a composite object (parameters/observables) are already
///  in the workspace the imported p.d.f. is connected to the already existing variables. If any of the _function_ objects (p.d.f, formulas)
///  to be imported already exists in the workspace an error message is printed and the import of the entire tree of objects is cancelled.
///  Several optional arguments can be provided to modify the import procedure.
///
///  <table>
///  <tr><th> Accepted arguments
///  <tr><td> `RenameConflictNodes(const char* suffix)`   <td>  Add suffix to branch node name if name conflicts with existing node in workspace
///  <tr><td> `RenameAllNodes(const char* suffix)`    <td>  Add suffix to all branch node names including top level node.
///  <tr><td> `RenameAllVariables(const char* suffix)`    <td>  Add suffix to all variables of objects being imported.
///  <tr><td> `RenameAllVariablesExcept(const char* suffix, const char* exceptionList)`   <td>  Add suffix to all variables names, except ones listed
///  <tr><td> `RenameVariable(const char* inputName, const char* outputName)` <td>  Rename a single variable as specified upon import.
///  <tr><td> `RecycleConflictNodes()`    <td>  If any of the function objects to be imported already exist in the name space, connect the
///                            imported expression to the already existing nodes.
///                            \attention Use with care! If function definitions do not match, this alters the definition of your function upon import
///
///  <tr><td> `Silence()` <td>  Do not issue any info message
///  </table>
///
///  The RenameConflictNodes, RenameNodes and RecycleConflictNodes arguments are mutually exclusive. The RenameVariable argument can be repeated
///  as often as necessary to rename multiple variables. Alternatively, a single RenameVariable argument can be given with
///  two comma separated lists.
/// \note From python, use `Import()`, since `import` is a reserved keyword.
bool RooWorkspace::import(const RooAbsArg& inArg,
    const RooCmdArg& arg1, const RooCmdArg& arg2, const RooCmdArg& arg3,
    const RooCmdArg& arg4, const RooCmdArg& arg5, const RooCmdArg& arg6,
    const RooCmdArg& arg7, const RooCmdArg& arg8, const RooCmdArg& arg9)
{
  RooLinkedList args ;
  args.Add((TObject*)&arg1) ;
  args.Add((TObject*)&arg2) ;
  args.Add((TObject*)&arg3) ;
  args.Add((TObject*)&arg4) ;
  args.Add((TObject*)&arg5) ;
  args.Add((TObject*)&arg6) ;
  args.Add((TObject*)&arg7) ;
  args.Add((TObject*)&arg8) ;
  args.Add((TObject*)&arg9) ;
 
  // Select the pdf-specific commands
  RooCmdConfig pc("RooWorkspace::import(" + std::string(GetName()) + ")");
 
  pc.defineString("conflictSuffix","RenameConflictNodes",0) ;
  pc.defineInt("renameConflictOrig","RenameConflictNodes",0,0) ;
  pc.defineString("allSuffix","RenameAllNodes",0) ;
  pc.defineString("allVarsSuffix","RenameAllVariables",0) ;
  pc.defineString("allVarsExcept","RenameAllVariables",1) ;
  pc.defineString("varChangeIn","RenameVar",0,"",true) ;
  pc.defineString("varChangeOut","RenameVar",1,"",true) ;
  pc.defineString("factoryTag","FactoryTag",0) ;
  pc.defineInt("useExistingNodes","RecycleConflictNodes",0,0) ;
  pc.defineInt("silence","Silence",0,0) ;
  pc.defineInt("noRecursion","NoRecursion",0,0) ;
  pc.defineMutex("RenameConflictNodes","RenameAllNodes") ;
  pc.defineMutex("RenameConflictNodes","RecycleConflictNodes") ;
  pc.defineMutex("RenameAllNodes","RecycleConflictNodes") ;
  pc.defineMutex("RenameVariable","RenameAllVariables") ;
 
  // Process and check varargs
  pc.process(args) ;
  if (!pc.ok(true)) {
    return true ;
  }
 
  // Decode renaming logic into suffix string and boolean for conflictOnly mode
  const char* suffixC = pc.getString("conflictSuffix") ;
  const char* suffixA = pc.getString("allSuffix") ;
  const char* suffixV = pc.getString("allVarsSuffix") ;
  const char* exceptVars = pc.getString("allVarsExcept") ;
  const char* varChangeIn = pc.getString("varChangeIn") ;
  const char* varChangeOut = pc.getString("varChangeOut") ;
  bool renameConflictOrig = pc.getInt("renameConflictOrig") ;
  Int_t useExistingNodes = pc.getInt("useExistingNodes") ;
  Int_t silence = pc.getInt("silence") ;
  Int_t noRecursion = pc.getInt("noRecursion") ;
 
 
  // Turn zero length strings into null pointers
  if (suffixC && strlen(suffixC)==0) suffixC = nullptr ;
  if (suffixA && strlen(suffixA)==0) suffixA = nullptr ;
 
  bool conflictOnly = suffixA ? false : true ;
  const char* suffix = suffixA ? suffixA : suffixC ;
 
  // Process any change in variable names
  std::map<string,string> varMap ;
  if (strlen(varChangeIn)>0) {
 
    // Parse comma separated lists into map<string,string>
    const std::vector<std::string> tokIn = ROOT::Split(varChangeIn, ", ", /*skipEmpty= */ true);
    const std::vector<std::string> tokOut = ROOT::Split(varChangeOut, ", ", /*skipEmpty= */ true);
    for (unsigned int i=0; i < tokIn.size(); ++i) {
      varMap.insert(std::make_pair(tokIn[i], tokOut[i]));
    }
 
    assert(tokIn.size() == tokOut.size());
  }
 
  // Process RenameAllVariables argument if specified
  // First convert exception list if provided
  std::set<string> exceptVarNames ;
  if (exceptVars && strlen(exceptVars)) {
    const std::vector<std::string> toks = ROOT::Split(exceptVars, ", ", /*skipEmpty= */ true);
    exceptVarNames.insert(toks.begin(), toks.end());
  }
 
  if (suffixV != nullptr && strlen(suffixV)>0) {
    std::unique_ptr<RooArgSet> vars{inArg.getVariables()};
    for (const auto v : *vars) {
      if (exceptVarNames.find(v->GetName())==exceptVarNames.end()) {
        varMap[v->GetName()] = Form("%s_%s",v->GetName(),suffixV) ;
      }
    }
  }
 
  // Scan for overlaps with current contents
  RooAbsArg* wsarg = _allOwnedNodes.find(inArg.GetName()) ;
 
  // Check for factory specification match
  const char* tagIn = inArg.getStringAttribute("factory_tag") ;
  const char* tagWs = wsarg ? wsarg->getStringAttribute("factory_tag") : nullptr ;
  bool factoryMatch = (tagIn && tagWs && !strcmp(tagIn,tagWs)) ;
  if (factoryMatch) {
    ((RooAbsArg&)inArg).setAttribute("RooWorkspace::Recycle") ;
  }
 
  if (!suffix && wsarg && !useExistingNodes && !(inArg.isFundamental() && !varMap[inArg.GetName()].empty())) {
    if (!factoryMatch) {
      if (wsarg!=&inArg) {
        coutE(ObjectHandling) << "RooWorkSpace::import(" << GetName() << ") ERROR importing object named " << inArg.GetName()
                   << ": another instance with same name already in the workspace and no conflict resolution protocol specified" << std::endl ;
        return true ;
      } else {
        if (!silence) {
          coutI(ObjectHandling) << "RooWorkSpace::import(" << GetName() << ") Object " << inArg.GetName() << " is already in workspace!" << std::endl ;
        }
        return true ;
      }
    } else {
      if(!silence) {
        coutI(ObjectHandling) << "RooWorkSpace::import(" << GetName() << ") Recycling existing object " << inArg.GetName() << " created with identical factory specification" << std::endl ;
      }
    }
  }
 
  // Make list of conflicting nodes
  RooArgSet conflictNodes ;
  RooArgSet branchSet ;
  if (noRecursion) {
    branchSet.add(inArg) ;
  } else {
    inArg.branchNodeServerList(&branchSet) ;
  }
 
  for (const auto branch : branchSet) {
    RooAbsArg* wsbranch = _allOwnedNodes.find(branch->GetName()) ;
    if (wsbranch && wsbranch!=branch && !branch->getAttribute("RooWorkspace::Recycle") && !useExistingNodes) {
      conflictNodes.add(*branch) ;
    }
  }
 
  // Terminate here if there are conflicts and no resolution protocol
  if (!conflictNodes.empty() && !suffix && !useExistingNodes) {
    coutE(ObjectHandling) << "RooWorkSpace::import(" << GetName() << ") ERROR object named " << inArg.GetName() << ": component(s) "
        << conflictNodes << " already in the workspace and no conflict resolution protocol specified" << std::endl ;
    return true ;
  }
 
  // Now create a working copy of the incoming object tree
  RooArgSet cloneSet;
  cloneSet.useHashMapForFind(true); // Accelerate finding
  RooArgSet(inArg).snapshot(cloneSet, !noRecursion);
  RooAbsArg* cloneTop = cloneSet.find(inArg.GetName()) ;
 
  // Mark all nodes for renaming if we are not in conflictOnly mode
  if (!conflictOnly) {
    conflictNodes.removeAll() ;
    conflictNodes.add(branchSet) ;
  }
 
  // Mark nodes that are to be renamed with special attribute
  string topName2 = cloneTop->GetName() ;
  if (!renameConflictOrig) {
    // Mark all nodes to be imported for renaming following conflict resolution protocol
    for (const auto cnode : conflictNodes) {
      RooAbsArg* cnode2 = cloneSet.find(cnode->GetName()) ;
      string origName = cnode2->GetName() ;
      cnode2->SetName(Form("%s_%s",cnode2->GetName(),suffix)) ;
      cnode2->SetTitle(Form("%s (%s)",cnode2->GetTitle(),suffix)) ;
      string tag = Form("ORIGNAME:%s",origName.c_str()) ;
      cnode2->setAttribute(tag.c_str()) ;
      if (!cnode2->getStringAttribute("origName")) {
        cnode2->setStringAttribute("origName",origName.c_str());
      }
 
      // Save name of new top level node for later use
      if (cnode2==cloneTop) {
        topName2 = cnode2->GetName() ;
      }
 
      if (!silence) {
        coutI(ObjectHandling) << "RooWorkspace::import(" << GetName()
                   << ") Resolving name conflict in workspace by changing name of imported node  "
                   << origName << " to " << cnode2->GetName() << std::endl ;
      }
    }
  } else {
 
    // Rename all nodes already in the workspace to 'clear the way' for the imported nodes
    for (const auto cnode : conflictNodes) {
 
      string origName = cnode->GetName() ;
      RooAbsArg* wsnode = _allOwnedNodes.find(origName.c_str()) ;
      if (wsnode) {
 
        if (!wsnode->getStringAttribute("origName")) {
          wsnode->setStringAttribute("origName",wsnode->GetName()) ;
        }
 
        if (!_allOwnedNodes.find(Form("%s_%s",cnode->GetName(),suffix))) {
          wsnode->SetName(Form("%s_%s",cnode->GetName(),suffix)) ;
          wsnode->SetTitle(Form("%s (%s)",cnode->GetTitle(),suffix)) ;
        } else {
          // Name with suffix already taken, add additional suffix
          for (unsigned int n=1; true; ++n) {
            string newname = Form("%s_%s_%d",cnode->GetName(),suffix,n) ;
            if (!_allOwnedNodes.find(newname.c_str())) {
              wsnode->SetName(newname.c_str()) ;
              wsnode->SetTitle(Form("%s (%s %d)",cnode->GetTitle(),suffix,n)) ;
              break ;
            }
          }
        }
        if (!silence) {
          coutI(ObjectHandling) << "RooWorkspace::import(" << GetName()
                << ") Resolving name conflict in workspace by changing name of original node "
                << origName << " to " << wsnode->GetName() << std::endl ;
        }
      } else {
        coutW(ObjectHandling) << "RooWorkspace::import(" << GetName() << ") Internal error: expected to find existing node "
            << origName << " to be renamed, but didn't find it..." << std::endl ;
      }
 
    }
  }
 
  // Process any change in variable names
  if (strlen(varChangeIn)>0 || (suffixV && strlen(suffixV)>0)) {
 
    // Process all changes in variable names
    for (const auto cnode : cloneSet) {
 
      if (varMap.find(cnode->GetName())!=varMap.end()) {
        string origName = cnode->GetName() ;
        cnode->SetName(varMap[cnode->GetName()].c_str()) ;
        string tag = Form("ORIGNAME:%s",origName.c_str()) ;
        cnode->setAttribute(tag.c_str()) ;
        if (!cnode->getStringAttribute("origName")) {
          cnode->setStringAttribute("origName",origName.c_str()) ;
        }
 
        if (!silence) {
          coutI(ObjectHandling) << "RooWorkspace::import(" << GetName() << ") Changing name of variable "
              << origName << " to " << cnode->GetName() << " on request" << std::endl ;
        }
 
        if (cnode==cloneTop) {
          topName2 = cnode->GetName() ;
        }
 
      }
    }
  }
 
  // Now clone again with renaming effective
  RooArgSet cloneSet2;
  cloneSet2.useHashMapForFind(true); // Faster finding
  RooArgSet(*cloneTop).snapshot(cloneSet2, !noRecursion);
  RooAbsArg* cloneTop2 = cloneSet2.find(topName2.c_str()) ;
 
  // Make final check list of conflicting nodes
  RooArgSet conflictNodes2 ;
  RooArgSet branchSet2 ;
  for (const auto branch2 : branchSet2) {
    if (_allOwnedNodes.find(branch2->GetName())) {
      conflictNodes2.add(*branch2) ;
    }
  }
 
  // Terminate here if there are conflicts and no resolution protocol
  if (!conflictNodes2.empty()) {
    coutE(ObjectHandling) << "RooWorkSpace::import(" << GetName() << ") ERROR object named " << inArg.GetName() << ": component(s) "
        << conflictNodes2 << " cause naming conflict after conflict resolution protocol was executed" << std::endl ;
    return true ;
  }
 
  // Perform any auxiliary imports at this point
  for (const auto node : cloneSet2) {
    if (node->importWorkspaceHook(*this)) {
      coutE(ObjectHandling) << "RooWorkSpace::import(" << GetName() << ") ERROR object named " << node->GetName()
                 << " has an error in importing in one or more of its auxiliary objects, aborting" << std::endl ;
      return true ;
    }
  }
 
  RooArgSet recycledNodes ;
  RooArgSet nodesToBeDeleted ;
  for (const auto node : cloneSet2) {
    if (_autoClass) {
      if (!_classes.autoImportClass(node->IsA())) {
        coutW(ObjectHandling) << "RooWorkspace::import(" << GetName() << ") WARNING: problems import class code of object "
            << node->ClassName() << "::" << node->GetName() << ", reading of workspace will require external definition of class" << std::endl ;
      }
    }
 
    // Point expensiveObjectCache to copy in this workspace
    RooExpensiveObjectCache& oldCache = node->expensiveObjectCache() ;
    node->setExpensiveObjectCache(_eocache) ;
    _eocache.importCacheObjects(oldCache,node->GetName(),true) ;
 
    // Check if node is already in workspace (can only happen for variables or identical instances, unless RecycleConflictNodes is specified)
    RooAbsArg* wsnode = _allOwnedNodes.find(node->GetName()) ;
 
    if (wsnode) {
      // Do not import node, add not to list of nodes that require reconnection
      if (!silence && useExistingNodes) {
        coutI(ObjectHandling) << "RooWorkspace::import(" << GetName() << ") using existing copy of " << node->ClassName()
                   << "::" << node->GetName() << " for import of " << cloneTop2->ClassName() << "::"
                   << cloneTop2->GetName() << std::endl ;
      }
      recycledNodes.add(*_allOwnedNodes.find(node->GetName())) ;
 
      // Delete clone of incoming node
      nodesToBeDeleted.addOwned(std::unique_ptr<RooAbsArg>{node});
 
      //cout << "WV: recycling existing node " << existingNode << " = " << existingNode->GetName() << " for imported node " << node << std::endl ;
 
    } else {
      // Import node
      if (!silence) {
        coutI(ObjectHandling) << "RooWorkspace::import(" << GetName() << ") importing " << node->ClassName() << "::"
            << node->GetName() << std::endl ;
      }
      _allOwnedNodes.addOwned(std::unique_ptr<RooAbsArg>{node});
      node->setWorkspace(*this);
      if (_openTrans) {
        _sandboxNodes.add(*node) ;
      } else {
        if (_dir && node->IsA() != RooConstVar::Class()) {
          _dir->InternalAppend(node) ;
        }
      }
    }
  }
 
  // Reconnect any nodes that need to be
  if (!recycledNodes.empty()) {
    for (const auto node : cloneSet2) {
      node->redirectServers(recycledNodes) ;
    }
  }
 
  cloneSet2.releaseOwnership() ;
 
  return false ;
}
 
 
 
////////////////////////////////////////////////////////////////////////////////
///  Import a dataset (RooDataSet or RooDataHist) into the workspace. The workspace will contain a copy of the data.
///  The dataset and its variables can be renamed upon insertion with the options below
///
///  <table>
/// <tr><th> Accepted arguments
/// <tr><td> `Rename(const char* suffix)` <td> Rename dataset upon insertion
/// <tr><td> `RenameVariable(const char* inputName, const char* outputName)` <td> Change names of observables in dataset upon insertion
/// <tr><td> `Silence` <td> Be quiet, except in case of errors
/// \note From python, use `Import()`, since `import` is a reserved keyword.
bool RooWorkspace::import(RooAbsData const& inData,
             const RooCmdArg& arg1, const RooCmdArg& arg2, const RooCmdArg& arg3,
             const RooCmdArg& arg4, const RooCmdArg& arg5, const RooCmdArg& arg6,
             const RooCmdArg& arg7, const RooCmdArg& arg8, const RooCmdArg& arg9)
 
{
 
  RooLinkedList args ;
  args.Add((TObject*)&arg1) ;
  args.Add((TObject*)&arg2) ;
  args.Add((TObject*)&arg3) ;
  args.Add((TObject*)&arg4) ;
  args.Add((TObject*)&arg5) ;
  args.Add((TObject*)&arg6) ;
  args.Add((TObject*)&arg7) ;
  args.Add((TObject*)&arg8) ;
  args.Add((TObject*)&arg9) ;
 
  // Select the pdf-specific commands
  RooCmdConfig pc(Form("RooWorkspace::import(%s)",GetName())) ;
 
  pc.defineString("dsetName","Rename",0,"") ;
  pc.defineString("varChangeIn","RenameVar",0,"",true) ;
  pc.defineString("varChangeOut","RenameVar",1,"",true) ;
  pc.defineInt("embedded","Embedded",0,0) ;
  pc.defineInt("silence","Silence",0,0) ;
 
  // Process and check varargs
  pc.process(args) ;
  if (!pc.ok(true)) {
    return true ;
  }
 
  // Decode renaming logic into suffix string and boolean for conflictOnly mode
  const char* dsetName = pc.getString("dsetName") ;
  const char* varChangeIn = pc.getString("varChangeIn") ;
  const char* varChangeOut = pc.getString("varChangeOut") ;
  bool embedded = pc.getInt("embedded") ;
  Int_t silence = pc.getInt("silence") ;
 
  if (!silence)
    coutI(ObjectHandling) << "RooWorkspace::import(" << GetName() << ") importing dataset " << inData.GetName() << std::endl ;
 
  // Transform empty string into null pointer
  if (dsetName && strlen(dsetName)==0) {
    dsetName=nullptr ;
  }
 
  RooLinkedList& dataList = embedded ? _embeddedDataList : _dataList ;
  if (dataList.size() > 50 && dataList.getHashTableSize() == 0) {
    // When the workspaces get larger, traversing the linked list becomes a bottleneck:
    dataList.setHashTableSize(200);
  }
 
  // Check that no dataset with target name already exists
  if (dsetName && dataList.FindObject(dsetName)) {
    coutE(ObjectHandling) << "RooWorkspace::import(" << GetName() << ") ERROR dataset with name " << dsetName << " already exists in workspace, import aborted" << std::endl ;
    return true ;
  }
  if (!dsetName && dataList.FindObject(inData.GetName())) {
    coutE(ObjectHandling) << "RooWorkspace::import(" << GetName() << ") ERROR dataset with name " << inData.GetName() << " already exists in workspace, import aborted" << std::endl ;
    return true ;
  }
 
  // Rename dataset if required
  RooAbsData* clone ;
  if (dsetName) {
    if (!silence)
      coutI(ObjectHandling) << "RooWorkSpace::import(" << GetName() << ") changing name of dataset from  " << inData.GetName() << " to " << dsetName << std::endl ;
    clone = static_cast<RooAbsData*>(inData.Clone(dsetName)) ;
  } else {
    clone = static_cast<RooAbsData*>(inData.Clone(inData.GetName())) ;
  }
 
 
  // Process any change in variable names
  if (strlen(varChangeIn)>0) {
    // Parse comma separated lists of variable name changes
    const std::vector<std::string> tokIn  = ROOT::Split(varChangeIn, ",");
    const std::vector<std::string> tokOut = ROOT::Split(varChangeOut, ",");
    for (unsigned int i=0; i < tokIn.size(); ++i) {
      if (!silence)
        coutI(ObjectHandling) << "RooWorkSpace::import(" << GetName() << ") changing name of dataset observable " << tokIn[i] << " to " << tokOut[i] << std::endl ;
      clone->changeObservableName(tokIn[i].c_str(), tokOut[i].c_str());
    }
  }
 
  // Now import the dataset observables, unless dataset is embedded
  if (!embedded) {
    for(RooAbsArg* carg : *clone->get()) {
      if (!arg(carg->GetName())) {
   import(*carg) ;
      }
    }
  }
 
  dataList.Add(clone) ;
  if (_dir) {
    _dir->InternalAppend(clone) ;
  }
 
  // Set expensive object cache of dataset internal buffers to that of workspace
  for(RooAbsArg* carg : *clone->get()) {
    carg->setExpensiveObjectCache(expensiveObjectCache()) ;
  }
 
 
  return false ;
}
 
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Define a named RooArgSet with given constituents. If importMissing is true, any constituents
/// of aset that are not in the workspace will be imported, otherwise an error is returned
/// for missing components
 
bool RooWorkspace::defineSet(const char* name, const RooArgSet& aset, bool importMissing)
{
  // Check if set was previously defined, if so print warning
  map<string,RooArgSet>::iterator i = _namedSets.find(name) ;
  if (i!=_namedSets.end()) {
    coutW(InputArguments) << "RooWorkspace::defineSet(" << GetName() << ") WARNING redefining previously defined named set " << name << std::endl ;
  }
 
  RooArgSet wsargs ;
 
  // Check all constituents of provided set
  for (RooAbsArg* sarg : aset) {
    // If missing, either import or report error
    if (!arg(sarg->GetName())) {
      if (importMissing) {
   import(*sarg) ;
      } else {
   coutE(InputArguments) << "RooWorkspace::defineSet(" << GetName() << ") ERROR set constituent \"" << sarg->GetName()
               << "\" is not in workspace and importMissing option is disabled" << std::endl ;
   return true ;
      }
    }
    wsargs.add(*arg(sarg->GetName())) ;
  }
 
 
  // Install named set
  _namedSets[name].removeAll() ;
  _namedSets[name].add(wsargs) ;
 
  return false ;
}
 
//_____________________________________________________________________________
bool RooWorkspace::defineSetInternal(const char *name, const RooArgSet &aset)
{
   // Define a named RooArgSet with given constituents. If importMissing is true, any constituents
   // of aset that are not in the workspace will be imported, otherwise an error is returned
   // for missing components
 
   // Check if set was previously defined, if so print warning
   map<string, RooArgSet>::iterator i = _namedSets.find(name);
   if (i != _namedSets.end()) {
      coutW(InputArguments) << "RooWorkspace::defineSet(" << GetName()
                            << ") WARNING redefining previously defined named set " << name << std::endl;
   }
 
   // Install named set
   _namedSets[name].removeAll();
   _namedSets[name].add(aset);
 
   return false;
}
 
////////////////////////////////////////////////////////////////////////////////
/// Define a named set in the workspace through a comma separated list of
/// names of objects already in the workspace
 
bool RooWorkspace::defineSet(const char* name, const char* contentList)
{
  // Check if set was previously defined, if so print warning
  map<string,RooArgSet>::iterator i = _namedSets.find(name) ;
  if (i!=_namedSets.end()) {
    coutW(InputArguments) << "RooWorkspace::defineSet(" << GetName() << ") WARNING redefining previously defined named set " << name << std::endl ;
  }
 
  RooArgSet wsargs ;
 
  // Check all constituents of provided set
  for (const std::string& token : ROOT::Split(contentList, ",")) {
    // If missing, either import or report error
    if (!arg(token.c_str())) {
      coutE(InputArguments) << "RooWorkspace::defineSet(" << GetName() << ") ERROR proposed set constituent \"" << token
             << "\" is not in workspace" << std::endl ;
      return true ;
    }
    wsargs.add(*arg(token.c_str())) ;
  }
 
  // Install named set
  _namedSets[name].removeAll() ;
  _namedSets[name].add(wsargs) ;
 
  return false ;
}
 
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Define a named set in the workspace through a comma separated list of
/// names of objects already in the workspace
 
bool RooWorkspace::extendSet(const char* name, const char* newContents)
{
  RooArgSet wsargs ;
 
  // Check all constituents of provided set
  for (const std::string& token : ROOT::Split(newContents, ",")) {
    // If missing, either import or report error
    if (!arg(token.c_str())) {
      coutE(InputArguments) << "RooWorkspace::defineSet(" << GetName() << ") ERROR proposed set constituent \"" << token
             << "\" is not in workspace" << std::endl ;
      return true ;
    }
    wsargs.add(*arg(token.c_str())) ;
  }
 
  // Extend named set
  _namedSets[name].add(wsargs,true) ;
 
  return false ;
}
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Return pointer to previously defined named set with given nmame
/// If no such set is found a null pointer is returned
 
const RooArgSet* RooWorkspace::set(RooStringView name)
{
  std::map<string,RooArgSet>::iterator i = _namedSets.find(name.c_str());
  return (i!=_namedSets.end()) ? &(i->second) : nullptr;
}
 
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Rename set to a new name
 
bool RooWorkspace::renameSet(const char* name, const char* newName)
{
  // First check if set exists
  if (!set(name)) {
    coutE(InputArguments) << "RooWorkspace::renameSet(" << GetName() << ") ERROR a set with name " << name
           << " does not exist" << std::endl ;
    return true ;
  }
 
  // Check if no set exists with new name
  if (set(newName)) {
    coutE(InputArguments) << "RooWorkspace::renameSet(" << GetName() << ") ERROR a set with name " << newName
           << " already exists" << std::endl ;
    return true ;
  }
 
  // Copy entry under 'name' to 'newName'
  _namedSets[newName].add(_namedSets[name]) ;
 
  // Remove entry under old name
  _namedSets.erase(name) ;
 
  return false ;
}
 
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Remove a named set from the workspace
 
bool RooWorkspace::removeSet(const char* name)
{
  // First check if set exists
  if (!set(name)) {
    coutE(InputArguments) << "RooWorkspace::removeSet(" << GetName() << ") ERROR a set with name " << name
           << " does not exist" << std::endl ;
    return true ;
  }
 
  // Remove set with given name
  _namedSets.erase(name) ;
 
  return false ;
}
 
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Open an import transaction operations. Returns true if successful, false
/// if there is already an ongoing transaction
 
bool RooWorkspace::startTransaction()
{
  // Check that there was no ongoing transaction
  if (_openTrans) {
    return false ;
  }
 
  // Open transaction
  _openTrans = true ;
  return true ;
}
 
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Cancel an ongoing import transaction. All objects imported since startTransaction()
/// will be removed and the transaction will be terminated. Return true if cancel operation
/// succeeds, return false if there was no open transaction
 
bool RooWorkspace::cancelTransaction()
{
  // Check that there is an ongoing transaction
  if (!_openTrans) {
    return false ;
  }
 
  // Delete all objects in the sandbox
  for(RooAbsArg * tmpArg : _sandboxNodes) {
    _allOwnedNodes.remove(*tmpArg) ;
  }
  _sandboxNodes.removeAll() ;
 
  // Mark transaction as finished
  _openTrans = false ;
 
  return true ;
}
 
bool RooWorkspace::commitTransaction()
{
  // Commit an ongoing import transaction. Returns true if commit succeeded,
  // return false if there was no ongoing transaction
 
  // Check that there is an ongoing transaction
  if (!_openTrans) {
    return false ;
  }
 
  // Publish sandbox nodes in directory and/or CINT if requested
  for(RooAbsArg* sarg : _sandboxNodes) {
    if (_dir && sarg->IsA() != RooConstVar::Class()) {
      _dir->InternalAppend(sarg) ;
    }
  }
 
  // Remove all committed objects from the sandbox
  _sandboxNodes.removeAll() ;
 
  // Mark transaction as finished
  _openTrans = false ;
 
  return true ;
}
 
 
 
 
////////////////////////////////////////////////////////////////////////////////
 
bool RooWorkspace::importClassCode(TClass* theClass, bool doReplace)
{
  return _classes.autoImportClass(theClass,doReplace) ;
}
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Import code of all classes in the workspace that have a class name
/// that matches pattern 'pat' and which are not found to be part of
/// the standard ROOT distribution. If doReplace is true any existing
/// class code saved in the workspace is replaced
 
bool RooWorkspace::importClassCode(const char* pat, bool doReplace)
{
  bool ret(true) ;
 
  TRegexp re(pat,true) ;
  for (RooAbsArg * carg : _allOwnedNodes) {
    TString className = carg->ClassName() ;
    if (className.Index(re)>=0 && !_classes.autoImportClass(carg->IsA(),doReplace)) {
      coutW(ObjectHandling) << "RooWorkspace::import(" << GetName() << ") WARNING: problems import class code of object "
             << carg->ClassName() << "::" << carg->GetName() << ", reading of workspace will require external definition of class" << std::endl ;
      ret = false ;
    }
  }
 
  return ret ;
}
 
 
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Save snapshot of values and attributes (including "Constant") of given parameters.
/// \param[in] name Name of the snapshot.
/// \param[in] paramNames Comma-separated list of parameter names to be snapshot.
bool RooWorkspace::saveSnapshot(RooStringView name, const char* paramNames)
{
  return saveSnapshot(name,argSet(paramNames),false) ;
}
 
 
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Save snapshot of values and attributes (including "Constant") of parameters 'params'.
/// If importValues is FALSE, the present values from the object in the workspace are
/// saved. If importValues is TRUE, the values of the objects passed in the 'params'
/// argument are saved
 
bool RooWorkspace::saveSnapshot(RooStringView name, const RooArgSet& params, bool importValues)
{
  RooArgSet actualParams;
  _allOwnedNodes.selectCommon(params, actualParams);
  auto snapshot = new RooArgSet;
  actualParams.snapshot(*snapshot);
 
  snapshot->setName(name.c_str()) ;
 
  if (importValues) {
    snapshot->assign(params) ;
  }
 
  if (std::unique_ptr<RooArgSet> oldSnap{static_cast<RooArgSet*>(_snapshots.FindObject(name.c_str()))}) {
    coutI(ObjectHandling) << "RooWorkspace::saveSnapshot(" << GetName() << ") replacing previous snapshot with name " << name << std::endl ;
    _snapshots.Remove(oldSnap.get()) ;
  }
 
  _snapshots.Add(snapshot) ;
 
  return true ;
}
 
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Load the values and attributes of the parameters in the snapshot saved with
/// the given name
 
bool RooWorkspace::loadSnapshot(const char* name)
{
  RooArgSet* snap = static_cast<RooArgSet*>(_snapshots.find(name)) ;
  if (!snap) {
    coutE(ObjectHandling) << "RooWorkspace::loadSnapshot(" << GetName() << ") no snapshot with name " << name << " is available" << std::endl ;
    return false ;
  }
 
  RooArgSet actualParams;
  _allOwnedNodes.selectCommon(*snap, actualParams);
  actualParams.assign(*snap) ;
 
  return true ;
}
 
 
////////////////////////////////////////////////////////////////////////////////
/// Return the RooArgSet containing a snapshot of variables contained in the workspace
///
/// Note that the variables of the objects in the snapshots are **copies** of the
/// variables in the workspace. To load the values of a snapshot in the workspace
/// variables, use loadSnapshot() instead.
 
const RooArgSet* RooWorkspace::getSnapshot(const char* name) const
{
  return static_cast<RooArgSet*>(_snapshots.find(name));
}
 
 
////////////////////////////////////////////////////////////////////////////////
/// Retrieve p.d.f (RooAbsPdf) with given name. A null pointer is returned if not found
 
RooAbsPdf* RooWorkspace::pdf(RooStringView name) const
{
  return dynamic_cast<RooAbsPdf*>(_allOwnedNodes.find(name.c_str())) ;
}
 
 
////////////////////////////////////////////////////////////////////////////////
/// Retrieve function (RooAbsReal) with given name. Note that all RooAbsPdfs are also RooAbsReals. A null pointer is returned if not found.
 
RooAbsReal* RooWorkspace::function(RooStringView name) const
{
  return dynamic_cast<RooAbsReal*>(_allOwnedNodes.find(name.c_str())) ;
}
 
 
////////////////////////////////////////////////////////////////////////////////
/// Retrieve real-valued variable (RooRealVar) with given name. A null pointer is returned if not found
 
RooRealVar* RooWorkspace::var(RooStringView name) const
{
  return dynamic_cast<RooRealVar*>(_allOwnedNodes.find(name.c_str())) ;
}
 
 
////////////////////////////////////////////////////////////////////////////////
/// Retrieve discrete variable (RooCategory) with given name. A null pointer is returned if not found
 
RooCategory* RooWorkspace::cat(RooStringView name) const
{
  return dynamic_cast<RooCategory*>(_allOwnedNodes.find(name.c_str())) ;
}
 
 
////////////////////////////////////////////////////////////////////////////////
/// Retrieve discrete function (RooAbsCategory) with given name. A null pointer is returned if not found
 
RooAbsCategory* RooWorkspace::catfunc(RooStringView name) const
{
  return dynamic_cast<RooAbsCategory*>(_allOwnedNodes.find(name.c_str())) ;
}
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Return RooAbsArg with given name. A null pointer is returned if none is found.
 
RooAbsArg* RooWorkspace::arg(RooStringView name) const
{
  return _allOwnedNodes.find(name.c_str()) ;
}
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Return set of RooAbsArgs matching to given list of names
 
RooArgSet RooWorkspace::argSet(RooStringView nameList) const
{
  RooArgSet ret ;
 
  for (const std::string& token : ROOT::Split(nameList, ",")) {
    RooAbsArg* oneArg = arg(token.c_str()) ;
    if (oneArg) {
      ret.add(*oneArg) ;
    } else {
      std::stringstream ss;
      ss << " RooWorkspace::argSet(" << GetName() << ") no RooAbsArg named \"" << token << "\" in workspace" ;
      const std::string errorMsg = ss.str();
      coutE(InputArguments) << errorMsg << std::endl;
      throw std::runtime_error(errorMsg);
    }
  }
  return ret ;
}
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Return fundamental (i.e. non-derived) RooAbsArg with given name. Fundamental types
/// are e.g. RooRealVar, RooCategory. A null pointer is returned if none is found.
 
RooAbsArg* RooWorkspace::fundArg(RooStringView name) const
{
  RooAbsArg* tmp = arg(name) ;
  if (!tmp) {
    return nullptr;
  }
  return tmp->isFundamental() ? tmp : nullptr;
}
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Retrieve dataset (binned or unbinned) with given name. A null pointer is returned if not found
 
RooAbsData* RooWorkspace::data(RooStringView name) const
{
  return static_cast<RooAbsData*>(_dataList.FindObject(name.c_str())) ;
}
 
 
////////////////////////////////////////////////////////////////////////////////
/// Retrieve dataset (binned or unbinned) with given name. A null pointer is returned if not found
 
RooAbsData* RooWorkspace::embeddedData(RooStringView name) const
{
  return static_cast<RooAbsData*>(_embeddedDataList.FindObject(name.c_str())) ;
}
 
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Return set with all variable objects
 
RooArgSet RooWorkspace::allVars() const
{
  RooArgSet ret ;
 
  // Split list of components in pdfs, functions and variables
  for(RooAbsArg* parg : _allOwnedNodes) {
    if (parg->IsA()->InheritsFrom(RooRealVar::Class())) {
      ret.add(*parg) ;
    }
  }
 
  return ret ;
}
 
 
////////////////////////////////////////////////////////////////////////////////
/// Return set with all category objects
 
RooArgSet RooWorkspace::allCats() const
{
  RooArgSet ret ;
 
  // Split list of components in pdfs, functions and variables
  for(RooAbsArg* parg : _allOwnedNodes) {
    if (parg->IsA()->InheritsFrom(RooCategory::Class())) {
      ret.add(*parg) ;
    }
  }
 
  return ret ;
}
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Return set with all function objects
 
RooArgSet RooWorkspace::allFunctions() const
{
  RooArgSet ret ;
 
  // Split list of components in pdfs, functions and variables
  for(RooAbsArg* parg : _allOwnedNodes) {
    if (parg->IsA()->InheritsFrom(RooAbsReal::Class()) &&
   !parg->IsA()->InheritsFrom(RooAbsPdf::Class()) &&
   !parg->IsA()->InheritsFrom(RooConstVar::Class()) &&
   !parg->IsA()->InheritsFrom(RooRealVar::Class())) {
      ret.add(*parg) ;
    }
  }
 
  return ret ;
}
 
 
////////////////////////////////////////////////////////////////////////////////
/// Return set with all category function objects
 
RooArgSet RooWorkspace::allCatFunctions() const
{
  RooArgSet ret ;
 
  // Split list of components in pdfs, functions and variables
  for(RooAbsArg* parg : _allOwnedNodes) {
    if (parg->IsA()->InheritsFrom(RooAbsCategory::Class()) &&
   !parg->IsA()->InheritsFrom(RooCategory::Class())) {
      ret.add(*parg) ;
    }
  }
  return ret ;
}
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Return set with all resolution model objects
 
RooArgSet RooWorkspace::allResolutionModels() const
{
  RooArgSet ret ;
 
  // Split list of components in pdfs, functions and variables
  for(RooAbsArg* parg : _allOwnedNodes) {
    if (parg->IsA()->InheritsFrom(RooResolutionModel::Class())) {
      if (!(static_cast<RooResolutionModel*>(parg))->isConvolved()) {
   ret.add(*parg) ;
      }
    }
  }
  return ret ;
}
 
 
////////////////////////////////////////////////////////////////////////////////
/// Return set with all probability density function objects
 
RooArgSet RooWorkspace::allPdfs() const
{
  RooArgSet ret ;
 
  // Split list of components in pdfs, functions and variables
  for(RooAbsArg* parg : _allOwnedNodes) {
    if (parg->IsA()->InheritsFrom(RooAbsPdf::Class()) &&
   !parg->IsA()->InheritsFrom(RooResolutionModel::Class())) {
      ret.add(*parg) ;
    }
  }
  return ret ;
}
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Return list of all dataset in the workspace
 
std::list<RooAbsData*> RooWorkspace::allData() const
{
  std::list<RooAbsData*> ret ;
  for(auto * dat : static_range_cast<RooAbsData*>(_dataList)) {
    ret.push_back(dat) ;
  }
  return ret ;
}
 
 
////////////////////////////////////////////////////////////////////////////////
/// Return list of all dataset in the workspace
 
std::list<RooAbsData*> RooWorkspace::allEmbeddedData() const
{
  std::list<RooAbsData*> ret ;
  for(auto * dat : static_range_cast<RooAbsData*>(_embeddedDataList)) {
    ret.push_back(dat) ;
  }
  return ret ;
}
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Return list of all generic objects in the workspace
 
std::list<TObject*> RooWorkspace::allGenericObjects() const
{
  std::list<TObject*> ret ;
  for(TObject * gobj : _genObjects) {
 
    // If found object is wrapper, return payload
    if (gobj->IsA()==RooTObjWrap::Class()) {
      ret.push_back((static_cast<RooTObjWrap*>(gobj))->obj()) ;
    } else {
      ret.push_back(gobj) ;
    }
  }
  return ret ;
}
 
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Import code of class 'tc' into the repository. If code is already in repository it is only imported
/// again if doReplace is false. The names and location of the source files is determined from the information
/// in TClass. If no location is found in the TClass information, the files are searched in the workspace
/// search path, defined by addClassDeclImportDir() and addClassImplImportDir() for declaration and implementation
/// files respectively. If files cannot be found, abort with error status, otherwise update the internal
/// class-to-file map and import the contents of the files, if they are not imported yet.
 
bool RooWorkspace::CodeRepo::autoImportClass(TClass* tc, bool doReplace)
{
 
  oocxcoutD(_wspace,ObjectHandling) << "RooWorkspace::CodeRepo(" << _wspace->GetName() << ") request to import code of class " << tc->GetName() << std::endl ;
 
  // *** PHASE 1 *** Check if file needs to be imported, or is in ROOT distribution, and check if it can be persisted
 
  // Check if we already have the class (i.e. it is in the classToFile map)
  if (!doReplace && _c2fmap.find(tc->GetName())!=_c2fmap.end()) {
    oocxcoutD(_wspace,ObjectHandling) << "RooWorkspace::CodeRepo(" << _wspace->GetName() << ") code of class " << tc->GetName() << " already imported, skipping" << std::endl ;
    return true ;
  }
 
  // Check if class is listed in a ROOTMAP file - if so we can skip it because it is in the root distribution
  const char* mapEntry = gInterpreter->GetClassSharedLibs(tc->GetName()) ;
  if (mapEntry && strlen(mapEntry)>0) {
    oocxcoutD(_wspace,ObjectHandling) << "RooWorkspace::CodeRepo(" << _wspace->GetName() << ") code of class " << tc->GetName() << " is in ROOT distribution, skipping " << std::endl ;
    return true ;
  }
 
  // Retrieve file names through ROOT TClass interface
  string implfile = tc->GetImplFileName() ;
  string declfile = tc->GetDeclFileName() ;
 
  // Check that file names are not empty
  if (implfile.empty() || declfile.empty()) {
    oocoutE(_wspace,ObjectHandling) << "RooWorkspace::CodeRepo(" << _wspace->GetName() << ") ERROR: cannot retrieve code file names for class "
               << tc->GetName() << " through ROOT TClass interface, unable to import code" << std::endl ;
    return false ;
  }
 
  // Check if header filename is found in ROOT distribution, if so, do not import class
  TString rootsys = gSystem->Getenv("ROOTSYS") ;
  if (TString(implfile.c_str()).Index(rootsys)>=0) {
    oocxcoutD(_wspace,ObjectHandling) << "RooWorkspace::CodeRepo(" << _wspace->GetName() << ") code of class " << tc->GetName() << " is in ROOT distribution, skipping " << std::endl ;
    return true ;
  }
 
  // Require that class meets technical criteria to be persistable (i.e it has a default constructor)
  // (We also need a default constructor of abstract classes, but cannot check that through is interface
  //  as TClass::HasDefaultCtor only returns true for callable default constructors)
  if (!(tc->Property() & kIsAbstract) && !tc->HasDefaultConstructor()) {
    oocoutW(_wspace,ObjectHandling) << "RooWorkspace::autoImportClass(" << _wspace->GetName() << ") WARNING cannot import class "
                << tc->GetName() << " : it cannot be persisted because it doesn't have a default constructor. Please fix " << std::endl ;
    return false ;
  }
 
 
  // *** PHASE 2 *** Check if declaration and implementation files can be located
 
  std::string declpath;
  std::string implpath;
 
  // Check if header file can be found in specified location
  // If not, scan through list of 'class declaration' paths in RooWorkspace
  if (gSystem->AccessPathName(declfile.c_str())) {
 
    // Check list of additional declaration paths
    list<string>::iterator diter = RooWorkspace::_classDeclDirList.begin() ;
 
    while(diter!= RooWorkspace::_classDeclDirList.end()) {
 
      declpath = gSystem->ConcatFileName(diter->c_str(),declfile.c_str()) ;
      if (!gSystem->AccessPathName(declpath.c_str())) {
   // found declaration file
   break ;
      }
      // cleanup and continue ;
      declpath.clear();
 
      ++diter ;
    }
 
    // Header file cannot be found anywhere, warn user and abort operation
    if (declpath.empty()) {
      oocoutW(_wspace,ObjectHandling) << "RooWorkspace::autoImportClass(" << _wspace->GetName() << ") WARNING Cannot access code of class "
                  << tc->GetName() << " because header file " << declfile << " is not found in current directory nor in $ROOTSYS" ;
      if (!_classDeclDirList.empty()) {
   ooccoutW(_wspace,ObjectHandling) << ", nor in the search path " ;
   diter = RooWorkspace::_classDeclDirList.begin() ;
 
   while(diter!= RooWorkspace::_classDeclDirList.end()) {
 
     if (diter!=RooWorkspace::_classDeclDirList.begin()) {
       ooccoutW(_wspace,ObjectHandling) << "," ;
     }
     ooccoutW(_wspace,ObjectHandling) << diter->c_str() ;
     ++diter ;
   }
      }
      ooccoutW(_wspace,ObjectHandling) << ". To fix this problem, add the required directory to the search "
                   << "path using RooWorkspace::addClassDeclImportDir(const char* dir)" << std::endl ;
 
      return false ;
    }
  }
 
 
  // Check if implementation file can be found in specified location
  // If not, scan through list of 'class implementation' paths in RooWorkspace
  if (gSystem->AccessPathName(implfile.c_str())) {
 
    // Check list of additional declaration paths
    list<string>::iterator iiter = RooWorkspace::_classImplDirList.begin() ;
 
    while(iiter!= RooWorkspace::_classImplDirList.end()) {
 
      implpath = gSystem->ConcatFileName(iiter->c_str(),implfile.c_str()) ;
      if (!gSystem->AccessPathName(implpath.c_str())) {
   // found implementation file
   break ;
      }
      // cleanup and continue ;
      implpath.clear();
 
      ++iiter ;
    }
 
    // Implementation file cannot be found anywhere, warn user and abort operation
    if (implpath.empty()) {
      oocoutW(_wspace,ObjectHandling) << "RooWorkspace::autoImportClass(" << _wspace->GetName() << ") WARNING Cannot access code of class "
                  << tc->GetName() << " because implementation file " << implfile << " is not found in current directory nor in $ROOTSYS" ;
      if (!_classDeclDirList.empty()) {
   ooccoutW(_wspace,ObjectHandling) << ", nor in the search path " ;
   iiter = RooWorkspace::_classImplDirList.begin() ;
 
   while(iiter!= RooWorkspace::_classImplDirList.end()) {
 
     if (iiter!=RooWorkspace::_classImplDirList.begin()) {
       ooccoutW(_wspace,ObjectHandling) << "," ;
     }
     ooccoutW(_wspace,ObjectHandling) << iiter->c_str() ;
     ++iiter ;
   }
      }
      ooccoutW(_wspace,ObjectHandling) << ". To fix this problem add the required directory to the search "
                   << "path using RooWorkspace::addClassImplImportDir(const char* dir)" << std::endl;
      return false;
    }
  }
 
  char buf[64000];
 
  // *** Phase 3 *** Prepare to import code from files into STL string buffer
  //
  // Code storage is organized in two linked maps
  //
  // _fmap contains stl strings with code, indexed on declaration file name
  //
  // _c2fmap contains list of declaration file names and list of base classes
  //                  and is indexed on class name
  //
  // Phase 3 is skipped if fmap already contains an entry with given filebasename
 
  const std::string declfilename = !declpath.empty() ? gSystem->BaseName(declpath.c_str())
                                                     : gSystem->BaseName(declfile.c_str());
 
  // Split in base and extension
  int dotpos2 = strrchr(declfilename.c_str(),'.') - declfilename.c_str() ;
  string declfilebase = declfilename.substr(0,dotpos2) ;
  string declfileext = declfilename.substr(dotpos2+1) ;
 
  list<string> extraHeaders ;
 
  // If file has not been stored yet, enter stl strings with implementation and declaration in file map
  if (_fmap.find(declfilebase) == _fmap.end()) {
 
    // Open declaration file
    std::fstream fdecl(!declpath.empty() ? declpath.c_str() : declfile.c_str());
 
    // Abort import if declaration file cannot be opened
    if (!fdecl) {
      oocoutE(_wspace,ObjectHandling) << "RooWorkspace::autoImportClass(" << _wspace->GetName()
                  << ") ERROR opening declaration file " <<  declfile << std::endl ;
      return false ;
    }
 
    oocoutI(_wspace,ObjectHandling) << "RooWorkspace::autoImportClass(" << _wspace->GetName()
                << ") importing code of class " << tc->GetName()
                << " from " << (!implpath.empty() ? implpath.c_str() : implfile.c_str())
                << " and " << (!declpath.empty() ? declpath.c_str() : declfile.c_str()) << std::endl ;
 
 
    // Read entire file into an stl string
    string decl ;
    while(fdecl.getline(buf,1023)) {
 
      // Look for include state of self
      bool processedInclude = false ;
      char* extincfile = nullptr ;
 
      // Look for include of declaration file corresponding to this implementation file
      if (strstr(buf,"#include")) {
   // Process #include statements here
   char tmp[64000];
   strlcpy(tmp, buf, 64000);
   bool stdinclude = strchr(buf, '<');
   strtok(tmp, " <\"");
   char *incfile = strtok(nullptr, " <>\"");
 
   if (!stdinclude) {
      // check if it lives in $ROOTSYS/include
      TString hpath = gSystem->Getenv("ROOTSYS");
      hpath += "/include/";
      hpath += incfile;
      if (gSystem->AccessPathName(hpath.Data())) {
         oocoutI(_wspace, ObjectHandling) << "RooWorkspace::autoImportClass(" << _wspace->GetName()
                                          << ") scheduling include file " << incfile << " for import" << std::endl;
         extraHeaders.push_back(incfile);
         extincfile = incfile;
         processedInclude = true;
      }
   }
      }
 
      if (processedInclude) {
   decl += "// external include file below retrieved from workspace code storage\n" ;
   decl += Form("#include \"%s\"\n",extincfile) ;
      } else {
   decl += buf ;
   decl += '\n' ;
      }
    }
 
    // Open implementation file
    fstream fimpl(!implpath.empty() ? implpath.c_str() : implfile.c_str()) ;
 
    // Abort import if implementation file cannot be opened
    if (!fimpl) {
      oocoutE(_wspace,ObjectHandling) << "RooWorkspace::autoImportClass(" << _wspace->GetName()
                  << ") ERROR opening implementation file " <<  implfile << std::endl ;
      return false ;
    }
 
 
    // Import entire implementation file into stl string
    string impl ;
    while(fimpl.getline(buf,1023)) {
      // Process #include statements here
 
      // Look for include state of self
      bool foundSelfInclude=false ;
      bool processedInclude = false ;
      char* extincfile = nullptr ;
 
      // Look for include of declaration file corresponding to this implementation file
      if (strstr(buf,"#include")) {
   // Process #include statements here
   char tmp[64000];
   strlcpy(tmp, buf, 64000);
   bool stdinclude = strchr(buf, '<');
   strtok(tmp, " <\"");
   char *incfile = strtok(nullptr, " <>\"");
 
   if (strstr(incfile, declfilename.c_str())) {
      foundSelfInclude = true;
   }
 
   if (!stdinclude && !foundSelfInclude) {
      // check if it lives in $ROOTSYS/include
      TString hpath = gSystem->Getenv("ROOTSYS");
      hpath += "/include/";
      hpath += incfile;
 
      if (gSystem->AccessPathName(hpath.Data())) {
         oocoutI(_wspace, ObjectHandling) << "RooWorkspace::autoImportClass(" << _wspace->GetName()
                                          << ") scheduling include file " << incfile << " for import" << std::endl;
         extraHeaders.push_back(incfile);
         extincfile = incfile;
         processedInclude = true;
      }
   }
      }
 
      // Explicitly rewrite include of own declaration file to string
      // any directory prefixes, copy all other lines verbatim in stl string
      if (foundSelfInclude) {
   // If include of self is found, substitute original include
   // which may have directory structure with a plain include
   impl += "// class declaration include file below retrieved from workspace code storage\n" ;
   impl += Form("#include \"%s.%s\"\n",declfilebase.c_str(),declfileext.c_str()) ;
      } else if (processedInclude) {
   impl += "// external include file below retrieved from workspace code storage\n" ;
   impl += Form("#include \"%s\"\n",extincfile) ;
      } else {
   impl += buf ;
   impl += '\n' ;
      }
    }
 
    // Create entry in file map
    _fmap[declfilebase]._hfile = decl ;
    _fmap[declfilebase]._cxxfile = impl ;
    _fmap[declfilebase]._hext = declfileext ;
 
    // Process extra includes now
    for (list<string>::iterator ehiter = extraHeaders.begin() ; ehiter != extraHeaders.end() ; ++ehiter ) {
      if (_ehmap.find(*ehiter) == _ehmap.end()) {
 
   ExtraHeader eh ;
   eh._hname = ehiter->c_str() ;
   fstream fehdr(ehiter->c_str()) ;
   string ehimpl ;
   char buf2[1024] ;
   while(fehdr.getline(buf2,1023)) {
 
     // Look for include of declaration file corresponding to this implementation file
     if (strstr(buf2,"#include")) {
       // Process #include statements here
       char tmp[64000];
       strlcpy(tmp, buf2, 64000);
       bool stdinclude = strchr(buf, '<');
       strtok(tmp, " <\"");
       char *incfile = strtok(nullptr, " <>\"");
 
       if (!stdinclude) {
          // check if it lives in $ROOTSYS/include
          TString hpath = gSystem->Getenv("ROOTSYS");
          hpath += "/include/";
          hpath += incfile;
          if (gSystem->AccessPathName(hpath.Data())) {
             oocoutI(_wspace, ObjectHandling) << "RooWorkspace::autoImportClass(" << _wspace->GetName()
                                              << ") scheduling recursive include file " << incfile << " for import"
                                              << std::endl;
             extraHeaders.push_back(incfile);
          }
       }
     }
 
     ehimpl += buf2;
     ehimpl += '\n';
   }
   eh._hfile = ehimpl.c_str();
 
   _ehmap[ehiter->c_str()] = eh;
      }
    }
 
  } else {
 
    // Inform that existing file entry is being recycled because it already contained class code
    oocoutI(_wspace,ObjectHandling) << "RooWorkspace::autoImportClass(" << _wspace->GetName()
                << ") code of class " << tc->GetName()
                << " was already imported from " << (!implpath.empty() ? implpath : implfile)
                << " and " << (!declpath.empty() ? declpath.c_str() : declfile.c_str()) << std::endl;
 
  }
 
 
  // *** PHASE 4 *** Import stl strings with code into workspace
  //
  // If multiple classes are declared in a single code unit, there will be
  // multiple _c2fmap entries all pointing to the same _fmap entry.
 
  // Make list of all immediate base classes of this class
  TString baseNameList ;
  TList* bl = tc->GetListOfBases() ;
  std::list<TClass*> bases ;
  for(auto * base : static_range_cast<TBaseClass*>(*bl)) {
    if (baseNameList.Length()>0) {
      baseNameList += "," ;
    }
    baseNameList += base->GetClassPointer()->GetName() ;
    bases.push_back(base->GetClassPointer()) ;
  }
 
  // Map class name to above _fmap entries, along with list of base classes
  // in _c2fmap
  _c2fmap[tc->GetName()]._baseName = baseNameList ;
  _c2fmap[tc->GetName()]._fileBase = declfilebase ;
 
  // Recursive store all base classes.
  for(TClass* bclass : bases) {
    autoImportClass(bclass,doReplace) ;
  }
 
  return true ;
}
 
 
////////////////////////////////////////////////////////////////////////////////
/// Create transient TDirectory representation of this workspace. This directory
/// will appear as a subdirectory of the directory that contains the workspace
/// and will have the name of the workspace suffixed with "Dir". The TDirectory
/// interface is read-only. Any attempt to insert objects into the workspace
/// directory representation will result in an error message. Note that some
/// ROOT object like TH1 automatically insert themselves into the current directory
/// when constructed. This will give error messages when done in a workspace
/// directory.
 
bool RooWorkspace::makeDir()
{
  if (_dir) return true ;
 
  std::string title= "TDirectory representation of RooWorkspace " + std::string(GetName());
  _dir = new WSDir(GetName(),title.c_str(),this) ;
 
  for (RooAbsArg * darg : _allOwnedNodes) {
    if (darg->IsA() != RooConstVar::Class()) {
      _dir->InternalAppend(darg) ;
    }
  }
 
  return true ;
}
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Import a clone of a generic TObject into workspace generic object container. Imported
/// object can be retrieved by name through the obj() method. The object is cloned upon
/// importation and the input argument does not need to live beyond the import call
///
/// Returns true if an error has occurred.
 
bool RooWorkspace::import(TObject const& object, bool replaceExisting)
{
  // First check if object with given name already exists
  std::unique_ptr<TObject> oldObj{_genObjects.FindObject(object.GetName())};
  if (oldObj && !replaceExisting) {
    coutE(InputArguments) << "RooWorkspace::import(" << GetName() << ") generic object with name "
           << object.GetName() << " is already in workspace and replaceExisting flag is set to false" << std::endl ;
    return true ;
  }
 
  // Grab the current state of the directory Auto-Add
  ROOT::DirAutoAdd_t func = object.IsA()->GetDirectoryAutoAdd();
  object.IsA()->SetDirectoryAutoAdd(nullptr);
  bool tmp = RooPlot::setAddDirectoryStatus(false) ;
 
  if (oldObj) {
    _genObjects.Replace(oldObj.get(),object.Clone()) ;
  } else {
    _genObjects.Add(object.Clone()) ;
  }
 
  // Reset the state of the directory Auto-Add
  object.IsA()->SetDirectoryAutoAdd(func);
  RooPlot::setAddDirectoryStatus(tmp) ;
 
  return false ;
}
 
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Import a clone of a generic TObject into workspace generic object container.
/// The imported object will be stored under the given alias name rather than its
/// own name. Imported object can be retrieved its alias name through the obj() method.
/// The object is cloned upon importation and the input argument does not need to live beyond the import call
/// This method is mostly useful for importing objects that do not have a settable name such as TMatrix
///
/// Returns true if an error has occurred.
 
bool RooWorkspace::import(TObject const& object, const char* aliasName, bool replaceExisting)
{
  // First check if object with given name already exists
  std::unique_ptr<TObject> oldObj{_genObjects.FindObject(aliasName)};
  if (oldObj && !replaceExisting) {
    coutE(InputArguments) << "RooWorkspace::import(" << GetName() << ") generic object with name "
           << aliasName << " is already in workspace and replaceExisting flag is set to false" << std::endl ;
    return true ;
  }
 
  TH1::AddDirectory(false) ;
  auto wrapper = new RooTObjWrap(object.Clone()) ;
  TH1::AddDirectory(true) ;
  wrapper->setOwning(true) ;
  wrapper->SetName(aliasName) ;
  wrapper->SetTitle(aliasName) ;
 
  if (oldObj) {
    _genObjects.Replace(oldObj.get(),wrapper) ;
  } else {
    _genObjects.Add(wrapper) ;
  }
  return false ;
}
 
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Insert RooStudyManager module
 
bool RooWorkspace::addStudy(RooAbsStudy& study)
{
  RooAbsStudy* clone = static_cast<RooAbsStudy*>(study.Clone()) ;
  _studyMods.Add(clone) ;
  return false ;
}
 
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Remove all RooStudyManager modules
 
void RooWorkspace::clearStudies()
{
  _studyMods.Delete() ;
}
 
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Return any type of object (RooAbsArg, RooAbsData or generic object) with given name)
 
TObject* RooWorkspace::obj(RooStringView name) const
{
  // Try RooAbsArg first
  TObject* ret = arg(name) ;
  if (ret) return ret ;
 
  // Then try RooAbsData
  ret = data(name) ;
  if (ret) return ret ;
 
  // Finally try generic object store
  return genobj(name) ;
}
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Return generic object with given name
 
TObject* RooWorkspace::genobj(RooStringView name)  const
{
  // Find object by name
  TObject* gobj = _genObjects.FindObject(name.c_str()) ;
 
  // Exit here if not found
  if (!gobj) return nullptr;
 
  // If found object is wrapper, return payload
  if (gobj->IsA()==RooTObjWrap::Class()) return (static_cast<RooTObjWrap*>(gobj))->obj() ;
 
  return gobj ;
}
 
 
 
////////////////////////////////////////////////////////////////////////////////
 
bool RooWorkspace::cd(const char* path)
{
  makeDir() ;
  return _dir->cd(path) ;
}
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Save this current workspace into given file
 
bool RooWorkspace::writeToFile(const char* fileName, bool recreate)
{
  TFile f(fileName,recreate?"RECREATE":"UPDATE") ;
  Write() ;
  return false ;
}
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Return instance to factory tool
 
RooFactoryWSTool& RooWorkspace::factory()
{
  if (_factory) {
    return *_factory;
  }
  cxcoutD(ObjectHandling) << "INFO: Creating RooFactoryWSTool associated with this workspace" << std::endl ;
  _factory = make_unique<RooFactoryWSTool>(*this);
  return *_factory;
}
 
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Short-hand function for `factory()->process(expr);`
///
/// \copydoc RooFactoryWSTool::process(const char*)
RooAbsArg* RooWorkspace::factory(RooStringView expr)
{
  return factory().process(expr.c_str()) ;
}
 
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Print contents of the workspace
 
void RooWorkspace::Print(Option_t* opts) const
{
  bool treeMode(false) ;
  bool verbose(false);
  if (TString(opts).Contains("t")) {
    treeMode=true ;
  }
  if (TString(opts).Contains("v")) {
     verbose = true;
  }
 
  std::cout << std::endl << "RooWorkspace(" << GetName() << ") " << GetTitle() << " contents" << std::endl << std::endl  ;
 
  RooArgSet pdfSet ;
  RooArgSet funcSet ;
  RooArgSet varSet ;
  RooArgSet catfuncSet ;
  RooArgSet convResoSet ;
  RooArgSet resoSet ;
 
 
  // Split list of components in pdfs, functions and variables
  for(RooAbsArg* parg : _allOwnedNodes) {
 
    //---------------
 
    if (treeMode) {
 
      // In tree mode, only add nodes with no clients to the print lists
 
      if (parg->IsA()->InheritsFrom(RooAbsPdf::Class())) {
   if (!parg->hasClients()) {
     pdfSet.add(*parg) ;
   }
      }
 
      if (parg->IsA()->InheritsFrom(RooAbsReal::Class()) &&
     !parg->IsA()->InheritsFrom(RooAbsPdf::Class()) &&
     !parg->IsA()->InheritsFrom(RooConstVar::Class()) &&
     !parg->IsA()->InheritsFrom(RooRealVar::Class())) {
   if (!parg->hasClients()) {
     funcSet.add(*parg) ;
   }
      }
 
 
      if (parg->IsA()->InheritsFrom(RooAbsCategory::Class()) &&
     !parg->IsA()->InheritsFrom(RooCategory::Class())) {
   if (!parg->hasClients()) {
     catfuncSet.add(*parg) ;
   }
      }
 
    } else {
 
      if (parg->IsA()->InheritsFrom(RooResolutionModel::Class())) {
   if ((static_cast<RooResolutionModel*>(parg))->isConvolved()) {
     convResoSet.add(*parg) ;
   } else {
     resoSet.add(*parg) ;
   }
      }
 
      if (parg->IsA()->InheritsFrom(RooAbsPdf::Class()) &&
     !parg->IsA()->InheritsFrom(RooResolutionModel::Class())) {
   pdfSet.add(*parg) ;
      }
 
      if (parg->IsA()->InheritsFrom(RooAbsReal::Class()) &&
     !parg->IsA()->InheritsFrom(RooAbsPdf::Class()) &&
     !parg->IsA()->InheritsFrom(RooConstVar::Class()) &&
     !parg->IsA()->InheritsFrom(RooRealVar::Class())) {
   funcSet.add(*parg) ;
      }
 
      if (parg->IsA()->InheritsFrom(RooAbsCategory::Class()) &&
     !parg->IsA()->InheritsFrom(RooCategory::Class())) {
   catfuncSet.add(*parg) ;
      }
    }
 
    if (parg->IsA()->InheritsFrom(RooRealVar::Class())) {
      varSet.add(*parg) ;
    }
 
    if (parg->IsA()->InheritsFrom(RooCategory::Class())) {
      varSet.add(*parg) ;
    }
 
  }
 
 
  RooFit::MsgLevel oldLevel = RooMsgService::instance().globalKillBelow() ;
  RooMsgService::instance().setGlobalKillBelow(RooFit::WARNING) ;
 
  if (!varSet.empty()) {
    varSet.sort() ;
    std::cout << "variables" << std::endl ;
    std::cout << "---------" << std::endl ;
    std::cout << varSet << std::endl ;
    std::cout << std::endl ;
  }
 
  if (!pdfSet.empty()) {
    std::cout << "p.d.f.s" << std::endl ;
    std::cout << "-------" << std::endl ;
    pdfSet.sort() ;
    for(RooAbsArg* parg : pdfSet) {
      if (treeMode) {
   parg->printComponentTree() ;
      } else {
   parg->Print() ;
      }
    }
    std::cout << std::endl ;
  }
 
  if (!treeMode) {
    if (!resoSet.empty()) {
      std::cout << "analytical resolution models" << std::endl ;
      std::cout << "----------------------------" << std::endl ;
      resoSet.sort() ;
      for(RooAbsArg* parg : resoSet) {
   parg->Print() ;
      }
      std::cout << std::endl ;
    }
  }
 
  if (!funcSet.empty()) {
    std::cout << "functions" << std::endl ;
    std::cout << "--------" << std::endl ;
    funcSet.sort() ;
    for(RooAbsArg * parg : funcSet) {
      if (treeMode) {
   parg->printComponentTree() ;
      } else {
   parg->Print() ;
      }
    }
    std::cout << std::endl ;
  }
 
  if (!catfuncSet.empty()) {
    std::cout << "category functions" << std::endl ;
    std::cout << "------------------" << std::endl ;
    catfuncSet.sort() ;
    for(RooAbsArg* parg : catfuncSet) {
      if (treeMode) {
   parg->printComponentTree() ;
      } else {
   parg->Print() ;
      }
    }
    std::cout << std::endl ;
  }
 
  if (!_dataList.empty()) {
    std::cout << "datasets" << std::endl ;
    std::cout << "--------" << std::endl ;
    for(auto * data2 : static_range_cast<RooAbsData*>(_dataList)) {
      std::cout << data2->ClassName() << "::" << data2->GetName() << *data2->get() << std::endl;
    }
    std::cout << std::endl ;
  }
 
  if (!_embeddedDataList.empty()) {
    std::cout << "embedded datasets (in pdfs and functions)" << std::endl ;
    std::cout << "-----------------------------------------" << std::endl ;
    for(auto * data2 : static_range_cast<RooAbsData*>(_embeddedDataList)) {
      std::cout << data2->ClassName() << "::" << data2->GetName() << *data2->get() << std::endl ;
    }
    std::cout << std::endl ;
  }
 
  if (!_snapshots.empty()) {
    std::cout << "parameter snapshots" << std::endl ;
    std::cout << "-------------------" << std::endl ;
    for(auto * snap : static_range_cast<RooArgSet*>(_snapshots)) {
      std::cout << snap->GetName() << " = (" ;
      bool first(true) ;
      for(RooAbsArg* a : *snap) {
   if (first) { first=false ; } else { std::cout << "," ; }
   std::cout << a->GetName() << "=" ;
   a->printValue(std::cout) ;
   if (a->isConstant()) {
     std::cout << "[C]" ;
   }
      }
      std::cout << ")" << std::endl ;
    }
    std::cout << std::endl ;
  }
 
 
  if (!_namedSets.empty()) {
    std::cout << "named sets" << std::endl ;
    std::cout << "----------" << std::endl ;
    for (map<string,RooArgSet>::const_iterator it = _namedSets.begin() ; it != _namedSets.end() ; ++it) {
       if (verbose || !isCacheSet(it->first)) {
          std::cout << it->first << ":" << it->second << std::endl;
       }
    }
 
    std::cout << std::endl ;
  }
 
 
  if (!_genObjects.empty()) {
    std::cout << "generic objects" << std::endl ;
    std::cout << "---------------" << std::endl ;
    for(TObject* gobj : _genObjects) {
      if (gobj->IsA()==RooTObjWrap::Class()) {
   std::cout << (static_cast<RooTObjWrap*>(gobj))->obj()->ClassName() << "::" << gobj->GetName() << std::endl ;
      } else {
   std::cout << gobj->ClassName() << "::" << gobj->GetName() << std::endl ;
      }
    }
    std::cout << std::endl ;
 
  }
 
  if (!_studyMods.empty()) {
    std::cout << "study modules" << std::endl ;
    std::cout << "-------------" << std::endl ;
    for(TObject* smobj : _studyMods) {
      std::cout << smobj->ClassName() << "::" << smobj->GetName() << std::endl ;
    }
    std::cout << std::endl ;
 
  }
 
  if (!_classes.listOfClassNames().empty()) {
    std::cout << "embedded class code" << std::endl ;
    std::cout << "-------------------" << std::endl ;
    std::cout << _classes.listOfClassNames() << std::endl ;
    std::cout << std::endl ;
  }
 
  if (!_eocache.empty()) {
    std::cout << "embedded precalculated expensive components" << std::endl ;
    std::cout << "-------------------------------------------" << std::endl ;
    _eocache.print() ;
  }
 
  RooMsgService::instance().setGlobalKillBelow(oldLevel) ;
 
  return ;
}
 
 
////////////////////////////////////////////////////////////////////////////////
/// Custom streamer for the workspace. Stream contents of workspace
/// and code repository. When reading, read code repository first
/// and compile missing classes before proceeding with streaming
/// of workspace contents to avoid errors.
 
void RooWorkspace::CodeRepo::Streamer(TBuffer &R__b)
{
   typedef ::RooWorkspace::CodeRepo thisClass;
 
   // Stream an object of class RooWorkspace::CodeRepo.
   if (R__b.IsReading()) {
 
      UInt_t R__s;
      UInt_t R__c;
      Version_t R__v = R__b.ReadVersion(&R__s, &R__c);
 
      // Stream contents of ClassFiles map
      Int_t count(0);
      R__b >> count;
      while (count--) {
         TString name;
         name.Streamer(R__b);
         _fmap[name]._hext.Streamer(R__b);
         _fmap[name]._hfile.Streamer(R__b);
         _fmap[name]._cxxfile.Streamer(R__b);
      }
 
      // Stream contents of ClassRelInfo map
      count = 0;
      R__b >> count;
      while (count--) {
         TString name;
         name.Streamer(R__b);
         _c2fmap[name]._baseName.Streamer(R__b);
         _c2fmap[name]._fileBase.Streamer(R__b);
      }
 
      if (R__v == 2) {
 
         count = 0;
         R__b >> count;
         while (count--) {
            TString name;
            name.Streamer(R__b);
            _ehmap[name]._hname.Streamer(R__b);
            _ehmap[name]._hfile.Streamer(R__b);
         }
      }
 
      R__b.CheckByteCount(R__s, R__c, thisClass::IsA());
 
      // Instantiate any classes that are not defined in current session
      _compiledOK = !compileClasses();
 
   } else {
 
      UInt_t R__c;
      R__c = R__b.WriteVersion(thisClass::IsA(), true);
 
      // Stream contents of ClassFiles map
      UInt_t count = _fmap.size();
      R__b << count;
      map<TString, ClassFiles>::iterator iter = _fmap.begin();
      while (iter != _fmap.end()) {
         TString key_copy(iter->first);
         key_copy.Streamer(R__b);
         iter->second._hext.Streamer(R__b);
         iter->second._hfile.Streamer(R__b);
         iter->second._cxxfile.Streamer(R__b);
 
         ++iter;
      }
 
      // Stream contents of ClassRelInfo map
      count = _c2fmap.size();
      R__b << count;
      map<TString, ClassRelInfo>::iterator iter2 = _c2fmap.begin();
      while (iter2 != _c2fmap.end()) {
         TString key_copy(iter2->first);
         key_copy.Streamer(R__b);
         iter2->second._baseName.Streamer(R__b);
         iter2->second._fileBase.Streamer(R__b);
         ++iter2;
      }
 
      // Stream contents of ExtraHeader map
      count = _ehmap.size();
      R__b << count;
      map<TString, ExtraHeader>::iterator iter3 = _ehmap.begin();
      while (iter3 != _ehmap.end()) {
         TString key_copy(iter3->first);
         key_copy.Streamer(R__b);
         iter3->second._hname.Streamer(R__b);
         iter3->second._hfile.Streamer(R__b);
         ++iter3;
      }
 
      R__b.SetByteCount(R__c, true);
   }
}
 
 
////////////////////////////////////////////////////////////////////////////////
/// Stream an object of class RooWorkspace. This is a standard ROOT streamer for the
/// I/O part. This custom function exists to detach all external client links
/// from the payload prior to writing the payload so that these client links
/// are not persisted. (Client links occur if external function objects use
/// objects contained in the workspace as input)
/// After the actual writing, these client links are restored.
 
void RooWorkspace::Streamer(TBuffer &R__b)
{
   if (R__b.IsReading()) {
 
      R__b.ReadClassBuffer(RooWorkspace::Class(), this);
 
      // Perform any pass-2 schema evolution here
      for (RooAbsArg *node : _allOwnedNodes) {
         node->ioStreamerPass2();
      }
      RooAbsArg::ioStreamerPass2Finalize();
 
      // Make expensive object cache of all objects point to intermal copy.
      // Somehow this doesn't work OK automatically
      for (RooAbsArg *node : _allOwnedNodes) {
         node->setExpensiveObjectCache(_eocache);
         node->setWorkspace(*this);
#ifdef ROOFIT_LEGACY_EVAL_BACKEND
         if (dynamic_cast<RooAbsOptTestStatistic *>(node)) {
            RooAbsOptTestStatistic *tmp = static_cast<RooAbsOptTestStatistic *>(node);
            if (tmp->isSealed() && tmp->sealNotice() && strlen(tmp->sealNotice()) > 0) {
               std::cout << "RooWorkspace::Streamer(" << GetName() << ") " << node->ClassName() << "::" << node->GetName()
                    << " : " << tmp->sealNotice() << std::endl;
            }
         }
#endif
      }
 
   } else {
 
      // Make lists of external clients of WS objects, and remove those links temporarily
 
      map<RooAbsArg *, vector<RooAbsArg *>> extClients;
      map<RooAbsArg *, vector<RooAbsArg *>> extValueClients;
      map<RooAbsArg *, vector<RooAbsArg *>> extShapeClients;
 
      for (RooAbsArg *tmparg : _allOwnedNodes) {
 
         // Loop over client list of this arg
         std::vector<RooAbsArg *> clientsTmp{tmparg->_clientList.begin(), tmparg->_clientList.end()};
         for (auto client : clientsTmp) {
            if (!_allOwnedNodes.containsInstance(*client)) {
 
               const auto refCount = tmparg->_clientList.refCount(client);
               auto &bufferVec = extClients[tmparg];
 
               bufferVec.insert(bufferVec.end(), refCount, client);
               tmparg->_clientList.Remove(client, true);
            }
         }
 
         // Loop over value client list of this arg
         clientsTmp.assign(tmparg->_clientListValue.begin(), tmparg->_clientListValue.end());
         for (auto vclient : clientsTmp) {
            if (!_allOwnedNodes.containsInstance(*vclient)) {
               cxcoutD(ObjectHandling) << "RooWorkspace::Streamer(" << GetName() << ") element " << tmparg->GetName()
                                       << " has external value client link to " << vclient << " (" << vclient->GetName()
                                       << ") with ref count " << tmparg->_clientListValue.refCount(vclient) << std::endl;
 
               const auto refCount = tmparg->_clientListValue.refCount(vclient);
               auto &bufferVec = extValueClients[tmparg];
 
               bufferVec.insert(bufferVec.end(), refCount, vclient);
               tmparg->_clientListValue.Remove(vclient, true);
            }
         }
 
         // Loop over shape client list of this arg
         clientsTmp.assign(tmparg->_clientListShape.begin(), tmparg->_clientListShape.end());
         for (auto sclient : clientsTmp) {
            if (!_allOwnedNodes.containsInstance(*sclient)) {
               cxcoutD(ObjectHandling) << "RooWorkspace::Streamer(" << GetName() << ") element " << tmparg->GetName()
                                       << " has external shape client link to " << sclient << " (" << sclient->GetName()
                                       << ") with ref count " << tmparg->_clientListShape.refCount(sclient) << std::endl;
 
               const auto refCount = tmparg->_clientListShape.refCount(sclient);
               auto &bufferVec = extShapeClients[tmparg];
 
               bufferVec.insert(bufferVec.end(), refCount, sclient);
               tmparg->_clientListShape.Remove(sclient, true);
            }
         }
      }
 
      R__b.WriteClassBuffer(RooWorkspace::Class(), this);
 
      // Reinstate clients here
 
      for (auto &iterx : extClients) {
         for (auto client : iterx.second) {
            iterx.first->_clientList.Add(client);
         }
      }
 
      for (auto &iterx : extValueClients) {
         for (auto client : iterx.second) {
            iterx.first->_clientListValue.Add(client);
         }
      }
 
      for (auto &iterx : extShapeClients) {
         for (auto client : iterx.second) {
            iterx.first->_clientListShape.Add(client);
         }
      }
   }
}
 
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Return STL string with last of class names contained in the code repository
 
std::string RooWorkspace::CodeRepo::listOfClassNames() const
{
  string ret ;
  map<TString,ClassRelInfo>::const_iterator iter = _c2fmap.begin() ;
  while(iter!=_c2fmap.end()) {
    if (!ret.empty()) {
      ret += ", " ;
    }
    ret += iter->first ;
    ++iter ;
  }
 
  return ret ;
}
 
namespace {
UInt_t crc32(const char* data, ULong_t sz, UInt_t crc)
{
  // update CRC32 with new data
 
  // use precomputed table, rather than computing it on the fly
  static const UInt_t crctab[256] = { 0x00000000,
    0x04c11db7, 0x09823b6e, 0x0d4326d9, 0x130476dc, 0x17c56b6b,
    0x1a864db2, 0x1e475005, 0x2608edb8, 0x22c9f00f, 0x2f8ad6d6,
    0x2b4bcb61, 0x350c9b64, 0x31cd86d3, 0x3c8ea00a, 0x384fbdbd,
    0x4c11db70, 0x48d0c6c7, 0x4593e01e, 0x4152fda9, 0x5f15adac,
    0x5bd4b01b, 0x569796c2, 0x52568b75, 0x6a1936c8, 0x6ed82b7f,
    0x639b0da6, 0x675a1011, 0x791d4014, 0x7ddc5da3, 0x709f7b7a,
    0x745e66cd, 0x9823b6e0, 0x9ce2ab57, 0x91a18d8e, 0x95609039,
    0x8b27c03c, 0x8fe6dd8b, 0x82a5fb52, 0x8664e6e5, 0xbe2b5b58,
    0xbaea46ef, 0xb7a96036, 0xb3687d81, 0xad2f2d84, 0xa9ee3033,
    0xa4ad16ea, 0xa06c0b5d, 0xd4326d90, 0xd0f37027, 0xddb056fe,
    0xd9714b49, 0xc7361b4c, 0xc3f706fb, 0xceb42022, 0xca753d95,
    0xf23a8028, 0xf6fb9d9f, 0xfbb8bb46, 0xff79a6f1, 0xe13ef6f4,
    0xe5ffeb43, 0xe8bccd9a, 0xec7dd02d, 0x34867077, 0x30476dc0,
    0x3d044b19, 0x39c556ae, 0x278206ab, 0x23431b1c, 0x2e003dc5,
    0x2ac12072, 0x128e9dcf, 0x164f8078, 0x1b0ca6a1, 0x1fcdbb16,
    0x018aeb13, 0x054bf6a4, 0x0808d07d, 0x0cc9cdca, 0x7897ab07,
    0x7c56b6b0, 0x71159069, 0x75d48dde, 0x6b93dddb, 0x6f52c06c,
    0x6211e6b5, 0x66d0fb02, 0x5e9f46bf, 0x5a5e5b08, 0x571d7dd1,
    0x53dc6066, 0x4d9b3063, 0x495a2dd4, 0x44190b0d, 0x40d816ba,
    0xaca5c697, 0xa864db20, 0xa527fdf9, 0xa1e6e04e, 0xbfa1b04b,
    0xbb60adfc, 0xb6238b25, 0xb2e29692, 0x8aad2b2f, 0x8e6c3698,
    0x832f1041, 0x87ee0df6, 0x99a95df3, 0x9d684044, 0x902b669d,
    0x94ea7b2a, 0xe0b41de7, 0xe4750050, 0xe9362689, 0xedf73b3e,
    0xf3b06b3b, 0xf771768c, 0xfa325055, 0xfef34de2, 0xc6bcf05f,
    0xc27dede8, 0xcf3ecb31, 0xcbffd686, 0xd5b88683, 0xd1799b34,
    0xdc3abded, 0xd8fba05a, 0x690ce0ee, 0x6dcdfd59, 0x608edb80,
    0x644fc637, 0x7a089632, 0x7ec98b85, 0x738aad5c, 0x774bb0eb,
    0x4f040d56, 0x4bc510e1, 0x46863638, 0x42472b8f, 0x5c007b8a,
    0x58c1663d, 0x558240e4, 0x51435d53, 0x251d3b9e, 0x21dc2629,
    0x2c9f00f0, 0x285e1d47, 0x36194d42, 0x32d850f5, 0x3f9b762c,
    0x3b5a6b9b, 0x0315d626, 0x07d4cb91, 0x0a97ed48, 0x0e56f0ff,
    0x1011a0fa, 0x14d0bd4d, 0x19939b94, 0x1d528623, 0xf12f560e,
    0xf5ee4bb9, 0xf8ad6d60, 0xfc6c70d7, 0xe22b20d2, 0xe6ea3d65,
    0xeba91bbc, 0xef68060b, 0xd727bbb6, 0xd3e6a601, 0xdea580d8,
    0xda649d6f, 0xc423cd6a, 0xc0e2d0dd, 0xcda1f604, 0xc960ebb3,
    0xbd3e8d7e, 0xb9ff90c9, 0xb4bcb610, 0xb07daba7, 0xae3afba2,
    0xaafbe615, 0xa7b8c0cc, 0xa379dd7b, 0x9b3660c6, 0x9ff77d71,
    0x92b45ba8, 0x9675461f, 0x8832161a, 0x8cf30bad, 0x81b02d74,
    0x857130c3, 0x5d8a9099, 0x594b8d2e, 0x5408abf7, 0x50c9b640,
    0x4e8ee645, 0x4a4ffbf2, 0x470cdd2b, 0x43cdc09c, 0x7b827d21,
    0x7f436096, 0x7200464f, 0x76c15bf8, 0x68860bfd, 0x6c47164a,
    0x61043093, 0x65c52d24, 0x119b4be9, 0x155a565e, 0x18197087,
    0x1cd86d30, 0x029f3d35, 0x065e2082, 0x0b1d065b, 0x0fdc1bec,
    0x3793a651, 0x3352bbe6, 0x3e119d3f, 0x3ad08088, 0x2497d08d,
    0x2056cd3a, 0x2d15ebe3, 0x29d4f654, 0xc5a92679, 0xc1683bce,
    0xcc2b1d17, 0xc8ea00a0, 0xd6ad50a5, 0xd26c4d12, 0xdf2f6bcb,
    0xdbee767c, 0xe3a1cbc1, 0xe760d676, 0xea23f0af, 0xeee2ed18,
    0xf0a5bd1d, 0xf464a0aa, 0xf9278673, 0xfde69bc4, 0x89b8fd09,
    0x8d79e0be, 0x803ac667, 0x84fbdbd0, 0x9abc8bd5, 0x9e7d9662,
    0x933eb0bb, 0x97ffad0c, 0xafb010b1, 0xab710d06, 0xa6322bdf,
    0xa2f33668, 0xbcb4666d, 0xb8757bda, 0xb5365d03, 0xb1f740b4
  };
 
  crc = ~crc;
  while (sz--) crc = (crc << 8) ^ UInt_t(*data++) ^ crctab[crc >> 24];
 
  return ~crc;
}
 
UInt_t crc32(const char* data)
{
  // Calculate crc32 checksum on given string
  unsigned long sz = strlen(data);
  switch (strlen(data)) {
    case 0:
      return 0;
    case 1:
      return data[0];
    case 2:
      return (data[0] << 8) | data[1];
    case 3:
      return (data[0] << 16) | (data[1] << 8) | data[2];
    case 4:
      return (data[0] << 24) | (data[1] << 16) | (data[2] << 8) | data[3];
    default:
      return crc32(data + 4, sz - 4, (data[0] << 24) | (data[1] << 16) |
      (data[2] << 8) | data[3]);
  }
}
 
}
 
////////////////////////////////////////////////////////////////////////////////
/// For all classes in the workspace for which no class definition is
/// found in the ROOT class table extract source code stored in code
/// repository into temporary directory set by
/// setClassFileExportDir(), compile classes and link them with
/// current ROOT session. If a compilation error occurs print
/// instructions for user how to fix errors and recover workspace and
/// abort import procedure.
 
bool RooWorkspace::CodeRepo::compileClasses()
{
  bool haveDir=false ;
 
  // Retrieve name of directory in which to export code files
  string dirName = Form(_classFileExportDir.c_str(),_wspace->uuid().AsString(),_wspace->GetName()) ;
 
  bool writeExtraHeaders(false) ;
 
  // Process all class entries in repository
  map<TString,ClassRelInfo>::iterator iter = _c2fmap.begin() ;
  while(iter!=_c2fmap.end()) {
 
    oocxcoutD(_wspace,ObjectHandling) << "RooWorkspace::CodeRepo::compileClasses() now processing class " << iter->first.Data() << std::endl ;
 
    // If class is already known, don't load
    if (gClassTable->GetDict(iter->first.Data())) {
      oocoutI(_wspace,ObjectHandling) << "RooWorkspace::CodeRepo::compileClasses() Embedded class "
                  << iter->first << " already in ROOT class table, skipping" << std::endl ;
      ++iter ;
      continue ;
    }
 
    // Check that export directory exists
    if (!haveDir) {
 
      // If not, make local directory to extract files
      if (!gSystem->AccessPathName(dirName.c_str())) {
   oocoutI(_wspace,ObjectHandling) << "RooWorkspace::CodeRepo::compileClasses() reusing code export directory " << dirName.c_str()
               << " to extract coded embedded in workspace" << std::endl ;
      } else {
   if (gSystem->MakeDirectory(dirName.c_str())==0) {
     oocoutI(_wspace,ObjectHandling) << "RooWorkspace::CodeRepo::compileClasses() creating code export directory " << dirName.c_str()
                 << " to extract coded embedded in workspace" << std::endl ;
   } else {
     oocoutE(_wspace,ObjectHandling) << "RooWorkspace::CodeRepo::compileClasses() ERROR creating code export directory " << dirName.c_str()
                 << " to extract coded embedded in workspace" << std::endl ;
     return false ;
   }
      }
      haveDir=true ;
 
    }
 
    // First write any extra header files
    if (!writeExtraHeaders) {
      writeExtraHeaders = true ;
 
      map<TString,ExtraHeader>::iterator extraIter = _ehmap.begin() ;
      while(extraIter!=_ehmap.end()) {
 
   // Check if identical declaration file (header) is already written
   bool needEHWrite=true ;
   string fdname = Form("%s/%s",dirName.c_str(),extraIter->second._hname.Data()) ;
   ifstream ifdecl(fdname.c_str()) ;
   if (ifdecl) {
     TString contents ;
     char buf[64000];
     while (ifdecl.getline(buf, 64000)) {
        contents += buf;
        contents += "\n";
     }
     UInt_t crcFile = crc32(contents.Data());
     UInt_t crcWS = crc32(extraIter->second._hfile.Data());
     needEHWrite = (crcFile != crcWS);
   }
 
   // Write declaration file if required
   if (needEHWrite) {
      oocoutI(_wspace, ObjectHandling) << "RooWorkspace::CodeRepo::compileClasses() Extracting extra header file "
                                       << fdname << std::endl;
 
      // Extra headers may contain non-existing path - create first to be sure
      gSystem->MakeDirectory(gSystem->GetDirName(fdname.c_str()));
 
      ofstream fdecl(fdname.c_str());
      if (!fdecl) {
         oocoutE(_wspace, ObjectHandling) << "RooWorkspace::CodeRepo::compileClasses() ERROR opening file " << fdname
                                          << " for writing" << std::endl;
         return false;
      }
      fdecl << extraIter->second._hfile.Data();
      fdecl.close();
   }
   ++extraIter;
      }
    }
 
 
    // Navigate from class to file
    ClassFiles& cfinfo = _fmap[iter->second._fileBase] ;
 
    oocxcoutD(_wspace,ObjectHandling) << "RooWorkspace::CodeRepo::compileClasses() now processing file with base " << iter->second._fileBase << std::endl ;
 
    // If file is already processed, skip to next class
    if (cfinfo._extracted) {
      oocxcoutD(_wspace,ObjectHandling) << "RooWorkspace::CodeRepo::compileClasses() file with base name " << iter->second._fileBase
                << " has already been extracted, skipping to next class" << std::endl ;
      continue ;
    }
 
    // Check if identical declaration file (header) is already written
    bool needDeclWrite=true ;
    string fdname = Form("%s/%s.%s",dirName.c_str(),iter->second._fileBase.Data(),cfinfo._hext.Data()) ;
    ifstream ifdecl(fdname.c_str()) ;
    if (ifdecl) {
      TString contents ;
      char buf[64000];
      while (ifdecl.getline(buf, 64000)) {
         contents += buf;
         contents += "\n";
      }
      UInt_t crcFile = crc32(contents.Data()) ;
      UInt_t crcWS   = crc32(cfinfo._hfile.Data()) ;
      needDeclWrite = (crcFile!=crcWS) ;
    }
 
    // Write declaration file if required
    if (needDeclWrite) {
      oocoutI(_wspace,ObjectHandling) << "RooWorkspace::CodeRepo::compileClasses() Extracting declaration code of class " << iter->first << ", file " << fdname << std::endl ;
      ofstream fdecl(fdname.c_str()) ;
      if (!fdecl) {
   oocoutE(_wspace,ObjectHandling) << "RooWorkspace::CodeRepo::compileClasses() ERROR opening file "
               << fdname << " for writing" << std::endl ;
   return false ;
      }
      fdecl << cfinfo._hfile ;
      fdecl.close() ;
    }
 
    // Check if identical implementation file is already written
    bool needImplWrite=true ;
    string finame = Form("%s/%s.cxx",dirName.c_str(),iter->second._fileBase.Data()) ;
    ifstream ifimpl(finame.c_str()) ;
    if (ifimpl) {
      TString contents ;
      char buf[64000];
      while (ifimpl.getline(buf, 64000)) {
         contents += buf;
         contents += "\n";
      }
      UInt_t crcFile = crc32(contents.Data()) ;
      UInt_t crcWS   = crc32(cfinfo._cxxfile.Data()) ;
      needImplWrite = (crcFile!=crcWS) ;
    }
 
    // Write implementation file if required
    if (needImplWrite) {
      oocoutI(_wspace,ObjectHandling) << "RooWorkspace::CodeRepo::compileClasses() Extracting implementation code of class " << iter->first << ", file " << finame << std::endl ;
      ofstream fimpl(finame.c_str()) ;
      if (!fimpl) {
   oocoutE(_wspace,ObjectHandling) << "RooWorkspace::CodeRepo::compileClasses() ERROR opening file"
               << finame << " for writing" << std::endl ;
   return false ;
      }
      fimpl << cfinfo._cxxfile ;
      fimpl.close() ;
    }
 
    // Mark this file as extracted
    cfinfo._extracted = true ;
    oocxcoutD(_wspace,ObjectHandling) << "RooWorkspace::CodeRepo::compileClasses() marking code unit  " << iter->second._fileBase << " as extracted" << std::endl ;
 
    // Compile class
    oocoutI(_wspace,ObjectHandling) << "RooWorkspace::CodeRepo::compileClasses() Compiling code unit " << iter->second._fileBase.Data() << " to define class " << iter->first << std::endl ;
    bool ok = gSystem->CompileMacro(finame.c_str(),"k") ;
 
    if (!ok) {
      oocoutE(_wspace,ObjectHandling) << "RooWorkspace::CodeRepo::compileClasses() ERROR compiling class " << iter->first.Data() << ", to fix this you can do the following: " << std::endl
                  << "  1) Fix extracted source code files in directory " << dirName.c_str() << "/" << std::endl
                  << "  2) In clean ROOT session compiled fixed classes by hand using '.x " << dirName.c_str() << "/ClassName.cxx+'" << std::endl
                  << "  3) Reopen file with RooWorkspace with broken source code in UPDATE mode. Access RooWorkspace to force loading of class" << std::endl
                  << "     Broken instances in workspace will _not_ be compiled, instead precompiled fixed instances will be used." << std::endl
                  << "  4) Reimport fixed code in workspace using 'RooWorkspace::importClassCode(\"*\",true)' method, Write() updated workspace to file and close file" << std::endl
                  << "  5) Reopen file in clean ROOT session to confirm that problems are fixed" << std::endl ;
   return false ;
    }
 
    ++iter ;
  }
 
  return true ;
}
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Internal access to TDirectory append method
 
void RooWorkspace::WSDir::InternalAppend(TObject* obj)
{
  TDirectory::Append(obj,false) ;
}
 
 
////////////////////////////////////////////////////////////////////////////////
/// Overload TDirectory interface method to prohibit insertion of objects in read-only directory workspace representation
 
void RooWorkspace::WSDir::Add(TObject* obj,bool)
{
  if (dynamic_cast<RooAbsArg*>(obj) || dynamic_cast<RooAbsData*>(obj)) {
    coutE(ObjectHandling) << "RooWorkspace::WSDir::Add(" << GetName() << ") ERROR: Directory is read-only representation of a RooWorkspace, use RooWorkspace::import() to add objects" << std::endl ;
  } else {
    InternalAppend(obj) ;
  }
}
 
 
////////////////////////////////////////////////////////////////////////////////
/// Overload TDirectory interface method to prohibit insertion of objects in read-only directory workspace representation
 
void RooWorkspace::WSDir::Append(TObject* obj,bool)
{
  if (dynamic_cast<RooAbsArg*>(obj) || dynamic_cast<RooAbsData*>(obj)) {
    coutE(ObjectHandling) << "RooWorkspace::WSDir::Add(" << GetName() << ") ERROR: Directory is read-only representation of a RooWorkspace, use RooWorkspace::import() to add objects" << std::endl ;
  } else {
    InternalAppend(obj) ;
  }
}
 
 
////////////////////////////////////////////////////////////////////////////////
/// If one of the TObject we have a referenced to is deleted, remove the
/// reference.
 
void RooWorkspace::RecursiveRemove(TObject *removedObj)
{
   _dataList.RecursiveRemove(removedObj);
   if (removedObj == _dir) _dir = nullptr;
 
   _allOwnedNodes.RecursiveRemove(removedObj); // RooArgSet
 
   _dataList.RecursiveRemove(removedObj);
   _embeddedDataList.RecursiveRemove(removedObj);
   _views.RecursiveRemove(removedObj);
   _snapshots.RecursiveRemove(removedObj);
   _genObjects.RecursiveRemove(removedObj);
   _studyMods.RecursiveRemove(removedObj);
 
   std::vector<std::string> invalidSets;
 
   for(auto &c : _namedSets) {
      auto const& setName = c.first;
      auto& set = c.second;
      std::size_t oldSize = set.size();
      set.RecursiveRemove(removedObj);
      // If the set is used internally by RooFit to cache parameters or
      // constraints, it is invalidated by object removal. We will keep track
      // of its name to remove the cache set later.
      if(set.size() < oldSize && isCacheSet(setName)) {
         invalidSets.emplace_back(setName);
      }
   }
 
   // Remove the sets that got invalidated by the object removal
   for(std::string const& setName : invalidSets) {
      removeSet(setName.c_str());
   }
 
   _eocache.RecursiveRemove(removedObj); // RooExpensiveObjectCache
}