.nmr files recognize all data types; closes #71

This commit is contained in:
Dominik Demuth 2023-05-19 16:53:54 +02:00
parent 8673e5acdb
commit 84d136dd4c
2 changed files with 7 additions and 1 deletions

View File

@ -10,6 +10,7 @@ from nmreval.data.points import Points
from nmreval.data.signals import Signal
from nmreval.utils.text import convert
from nmreval.data.bds import BDS
from nmreval.data.dsc import DSC
from nmreval.lib.colors import BaseColor, TUColors
from nmreval.lib.lines import LineStyle
from nmreval.lib.symbols import SymbolStyle, symbolcycle
@ -488,6 +489,9 @@ class PointContainer(ExperimentContainer):
self.mode = 'pts'
self._init_plot(**kwargs)
if isinstance(self._data, DSC):
self.mode = 'dsc'
def _init_plot(self, **kwargs):
self.plot_imag = None

View File

@ -5,6 +5,8 @@ from collections import OrderedDict
from .read_old_nmr import HAS_BSDDB3, _read_file_v1
from ..data.nmr import FID, Spectrum
from ..data.bds import BDS
from ..data.dsc import DSC
from ..data.points import Points
from ..fit.result import FitResult, FitResultCreator
from ..lib.colors import Colors
@ -51,7 +53,7 @@ class NMRReader:
states = pickle.load(fp)
datalist = OrderedDict()
_dtypes = {'pts': Points, 'fit': FitResult, 'fid': FID}
_dtypes = {'pts': Points, 'fit': FitResult, 'fid': FID, 'bds': BDS, 'dsc': DSC}
for s in states['sets']:
set_id = s.pop('id')