ShipGeoConfig Namespace Reference

Classes
class	AttrDict
class	Config

Functions
Config	load_from_root_file (root_file, str key="ShipGeo")

Function Documentation

load_from_root_file()

Config ShipGeoConfig.load_from_root_file	(		root_file,
		str	key = "ShipGeo"
	)

Load configuration from ROOT file.

Automatically detects and handles both formats:
- New format: JSON string (stored as std::string or TObjString)
- Old format: Pickled Python object

Args:
    root_file: Either a ROOT.TFile object or a string path to ROOT file
    key: The key name for the stored config (default: 'ShipGeo')

Returns:
    Config object with the loaded configuration

Definition at line 95 of file ShipGeoConfig.py.

def load_from_root_file(root_file, key: str = "ShipGeo") -> Config:
    """
    Load configuration from ROOT file.
 
    Automatically detects and handles both formats:
    - New format: JSON string (stored as std::string or TObjString)
    - Old format: Pickled Python object
 
    Args:
        root_file: Either a ROOT.TFile object or a string path to ROOT file
        key: The key name for the stored config (default: 'ShipGeo')
 
    Returns:
        Config object with the loaded configuration
    """
    import ROOT
 
    own_file = False
    if isinstance(root_file, str):
        root_file = ROOT.TFile.Open(root_file)
        own_file = True
 
    try:
        # Get the object (could be std::string or TObjString)
        config_obj = root_file.Get(key)
        if not config_obj:
            raise ValueError(f"No object with key '{key}' found in ROOT file")
 
        # Convert to Python string
        content_str = str(config_obj)
 
        # Auto-detect format by checking first character
        if content_str.startswith("{"):
            # JSON format - parse it
            config = Config()
            config.loads_json(content_str)
        else:
            # Assume pickle format - unpickle it
            # Convert to bytes for pickle (using latin-1 encoding)
            pickle_bytes = content_str.encode("latin-1")
            config = pickle.loads(pickle_bytes)
 
            # Ensure it's a Config object (might be if it was pickled as Config)
            if not isinstance(config, Config):
                # Wrap in Config if needed
                c = Config()
                c.update(config)
                config = c
 
        return config
 
    finally:
        if own_file:
            root_file.Close()