scpsection6.cs

来自「ecg tool kit for medical image retrieval」· CS 代码 · 共 528 行 · 第 1/2 页

										temp[time1++] = leads[loper][time2];
									}

									if ((time2 + ((endNonProtected - (time1 + time1Offset)) / rate)) >= leads[loper].Length)
									{
										endNonProtected -= ((time2 + ((endNonProtected - (time1 + time1Offset)) / rate)) - leads[loper].Length) * rate;
									}

									endNonProtected -= rate + (rate >> 1);

									while ((time1 + time1Offset) < endNonProtected)
									{
										for (int i=0;(i < rate) && (time1 < temp.Length);i++)
										{
											temp[time1++] = (short) (((leads[loper][time2+1] - leads[loper][time2]) / rate) * i + leads[loper][time2]);
										}
										time2++;
									}

									endNonProtected += rate + (rate >> 1);

									for (int end=0;end < (rate >> 1);end++)
									{
										temp[time1++] = leads[loper][time2];
									}

									time2++;

									beginNonProtected = (zone == qrsLocations.getNrProtectedZones() ? temp.Length : qrsLocations.getProtectedEnd(zone));
								}
								else
								{
									// This should never happen!!
									if (zone == qrsLocations.getNrProtectedZones())
									{
										break;
									}

									while (((time1 + time1Offset) < qrsLocations.getProtectedEnd(zone))
										&& (time1 < temp.Length)
										&& (time2 < leads[loper].Length))
									{
										temp[time1++] = leads[loper][time2++];
									}
									zone++;
									endNonProtected = (zone == qrsLocations.getNrProtectedZones() ? temp.Length : qrsLocations.getProtectedStart(zone));
								}
							}

							leads[loper] = temp;
						}
						else
						{
							ECGTool.ResampleLead(leads[loper], localFreq, medianFreq, out leads[loper]);
						}
					}
				}
				return leads;
			}
			return null;
		}
		/// <summary>
		/// Function to encode data in this section.
		/// </summary>
		/// <param name="data">Rhythm data to encode</param>
		/// <param name="tables">Huffman table to use during enconding</param>
		/// <param name="leadDefinition">Lead Definitions to use for encoding</param>
		/// <param name="difference">difference to use durring decoding</param>
		/// <returns>0 on succes</returns>
		public int EncodeData(short[][] data, SCPSection2 tables, SCPSection3 leadDefinition, SCPSection4 qrsLocations, int medianFreq, byte difference)
		{
			int localFreq = getSamplesPerSecond();

			if ((data != null)
			&&	(tables != null)
			&&	(leadDefinition != null)
			&&  (localFreq > 0))
			{
				if ((medianFreq <= 0)
				||  (medianFreq == localFreq))
				{
					medianFreq = 1;
					localFreq = 1;
				}

				if ((_Bimodal == 0x1)
				&&  (qrsLocations == null))
				{
					return 2;
				}

				ushort nrleads = leadDefinition.getNrLeads();
				_Data = new byte[nrleads][];
				_DataLength = new ushort[nrleads];
				for (int loper=0;loper < nrleads;loper++)
				{
					if (data[loper] == null)
					{
						return 4;
					}

					short[] temp = data[loper];

					int time = (leadDefinition.getLeadLength(loper) * localFreq) / medianFreq;
					if (localFreq != medianFreq)
					{
						int rate = (medianFreq / localFreq);
						// Bimodal part might be buggy unable to test.
						if ((_Bimodal == 0x1)
						&&	((medianFreq % localFreq) == 0)
						&&	(rate > 0)
						&&	(rate < 5))
						{
							// Calculate nr of samples stored in section.
							time = 0;

							int nrzones = qrsLocations.getNrProtectedZones();
							for (int zone=0;zone < nrzones;zone++)
							{
								int begin = (qrsLocations.getProtectedStart(zone) >= leadDefinition.getLeadStart(loper) ? qrsLocations.getProtectedStart(zone) : leadDefinition.getLeadStart(loper));
								int end = (qrsLocations.getProtectedEnd(zone) <= leadDefinition.getLeadEnd(loper) ? qrsLocations.getProtectedEnd(zone) : leadDefinition.getLeadEnd(loper));
							
								begin = (end > begin ? end - begin + 1 : 0);

								time += begin + (rate - (begin % rate));
							}

							time += ((leadDefinition.getLeadLength(loper) - time) * localFreq) / medianFreq;

							int leadLength = leadDefinition.getLeadLength(loper);

							time += ((leadLength - time) * localFreq) / medianFreq;

							// Restructure bimodal data to length set in Section3.
							temp = new short[time];

							int time2Offset = leadDefinition.getLeadStart(loper);
							int time1=0;
							int time2=0;

							while ((time1 < temp.Length)
								&& (time2 <= leadLength)
								&& (time2 < data[loper].Length))
							{
								int zone=0;
								int end = qrsLocations.getNrProtectedZones();
								for (;zone < end;zone++)
								{
									if ((qrsLocations.getProtectedLength(zone) > 0)
									&&	(time2 + time2Offset >= qrsLocations.getProtectedStart(zone))
									&&  (time2 + time2Offset <= qrsLocations.getProtectedEnd(zone)))
									{
										break;
									}
								}

								if (zone < end)
								{
									temp[time1] = data[loper][time2++];
								}
								else
								{
									int Sum = 0;
									for (int sumLoper=0;sumLoper < rate;sumLoper++)
									{
										Sum += data[loper][time2 + sumLoper];
									}
									temp[time1] = (short) (Sum / rate);
									time2 += rate;
								}

								time1++;
							}
						}
						else
						{
							_Bimodal = 0;
							ECGTool.ResampleLead(temp, medianFreq, localFreq, out temp);
						}
					}

					_Difference = difference;
					_Data[loper] = tables.Encode(temp, time, 0, _Difference);
					if (_Data[loper] == null)
					{
						_Data = null;
						_DataLength = null;
						return 8;
					}
					_DataLength[loper] = (ushort) _Data[loper].Length;
					if ((_DataLength[loper] & 0x1) == 0x1)
					{
						_DataLength[loper]++;
					}
				}
				return 0;
			}
			return 1;
		}
		/// <summary>
		/// Function to get AVM.
		/// </summary>
		/// <returns>AVM in uV</returns>
		public double getAVM()
		{
			if (_AVM > 0)
			{
				return (((double)_AVM) / 1000.0);
			}
			return -1;
		}
		/// <summary>
		/// Function to set AVM.
		/// </summary>
		/// <param name="avm">AVM in uV</param>
		public void setAVM(double avm)
		{
			if (avm > 0)
			{
				_AVM  = (ushort) (avm * 1000);
			}
		}
		/// <summary>
		/// Function to get samples per second used in data.
		/// </summary>
		/// <returns>samples per second</returns>
		public int getSamplesPerSecond()
		{
			if (_TimeInterval > 0)
			{
				return (1000000 / _TimeInterval);
			}
			return -1;
		}
		/// <summary>
		/// Function to set samples per second used in data.
		/// </summary>
		/// <param name="sps">samples per second</param>
		public void setSamplesPerSecond(int sps)
		{
			if (sps > 0)
			{
				_TimeInterval = (ushort) (1000000 / sps);
			}
		}
		/// <summary>
		/// Function to get bimodal settings.
		/// </summary>
		/// <returns>true if bimodal used</returns>
		public bool getBimodal()
		{
			return _Bimodal == 1;
		}
		/// <summary>
		/// Function to set bimodal settings.
		/// </summary>
		/// <param name="bimodal">true if bimodal used</param>
		public void setBimodal(bool bimodal)
		{
			_Bimodal = (byte) (bimodal ? 1 : 0);
		}
	}
}