Files » sharedDataClass.py
1 |
import cv2 as cv |
---|---|
2 |
import numpy as np |
3 |
import tkinter as tk |
4 |
from tkinter import filedialog, messagebox |
5 |
import os |
6 |
from PIL import Image, ImageTk |
7 |
import configparser |
8 |
|
9 |
|
10 |
class SharedData: |
11 |
def __init__(self): |
12 |
self.left_image = None |
13 |
self.right_image = None |
14 |
self.overlap_region = 0 |
15 |
|
16 |
def save_to_ini(self, filename): |
17 |
"""Save the current data to an .ini file."""
|
18 |
config = configparser.ConfigParser() |
19 |
|
20 |
# Create a section and add attributes
|
21 |
config['Images'] = { |
22 |
'left_image': self.left_image, |
23 |
'right_image': self.right_image, |
24 |
'overlap_region': str(self.overlap_region) |
25 |
}
|
26 |
# Write to the .ini file
|
27 |
with open(filename, 'w') as configfile: |
28 |
config.write(configfile) |
29 |
|
30 |
def load_from_ini(self, filename): |
31 |
"""Load data from an .ini file."""
|
32 |
config = configparser.ConfigParser() |
33 |
config.read(filename) |
34 |
|
35 |
# Read attributes from the section
|
36 |
if 'Images' in config: |
37 |
self.left_image = config['Images'].get('left_image') |
38 |
self.right_image = config['Images'].get('right_image') |
39 |
self.overlap_region = config['Images'].getint('overlap_region') |
40 |
|