dbs_annotator.models.session_data.SessionData

class dbs_annotator.models.session_data.SessionData(file_path=None)[source]

Bases: object

Manages all data for a clinical DBS programming session.

This class handles: - TSV file creation and writing - Block ID tracking - Clinical and session scales management - Stimulation parameters tracking

Methods

close_file()

Close the TSV file if it's open.

initialize_simple_file(filepath)

Initialize a simple TSV file for annotations-only mode.

is_file_open()

Check if a TSV file is currently open.

open_file(file_path)

Open a TSV file for writing and initialize the CSV writer.

open_file_append(file_path[, start_block_id])

Open an existing TSV file in append mode and continue block numbering.

open_simple_file_append(filepath)

Open an existing annotations-only TSV file in append mode (or create it if missing).

write_clinical_scales(scales, stimulation[, ...])

Write clinical scales data to the TSV file.

write_session_scales(scales, stimulation[, ...])

Write session scales data to the TSV file with current timestamp.

write_simple_annotation(annotation)

Write a simple annotation with timestamp.