src/subtitles/libssf/Split.cpp

   1 /*
   2  *      Copyright (C) 2003-2006 Gabest
   3  *      http://www.gabest.org
   4  *
   5  *  This Program is free software; you can redistribute it and/or modify
   6  *  it under the terms of the GNU General Public License as published by
   7  *  the Free Software Foundation; either version 2, or (at your option)
   8  *  any later version.
   9  *
  10  *  This Program is distributed in the hope that it will be useful,
  11  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  13  *  GNU General Public License for more details.
  14  *
  15  *  You should have received a copy of the GNU General Public License
  16  *  along with GNU Make; see the file COPYING.  If not, write to
  17  *  the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA.
  18  *  http://www.gnu.org/copyleft/gpl.html
  19  *
  20  */
  21
  22 #include "stdafx.h"
  23 #include "Split.h"
  24 #include "Exception.h"
  25
  26 namespace ssf
  27 {
  28         Split::Split(LPCWSTR sep, CStringW str, size_t limit, SplitType type)
  29         {
  30                 DoSplit(sep, str, limit, type);
  31         }
  32
  33         Split::Split(WCHAR sep, CStringW str, size_t limit, SplitType type)
  34         {
  35                 DoSplit(CStringW(sep), str, limit, type);
  36         }
  37
  38         void Split::DoSplit(LPCWSTR sep, CStringW str, size_t limit, SplitType type)
  39         {
  40                 RemoveAll();
  41
  42                 if(size_t seplen = wcslen(sep))
  43                 {
  44                         for(int i = 0, j = 0, len = str.GetLength();
  45                                 i <= len && (limit == 0 || GetCount() < limit);
  46                                 i = j + (int)seplen)
  47                         {
  48                                 j = str.Find(sep, i);
  49                                 if(j < 0) j = len;
  50
  51                                 CStringW s = i < j ? str.Mid(i, j - i) : L"";
  52
  53                                 switch(type)
  54                                 {
  55                                 case Min: s.Trim(); // fall through
  56                                 case Def: if(s.IsEmpty()) break; // else fall through
  57                                 case Max: Add(s); break;
  58                                 }
  59                         }
  60                 }
  61         }
  62
  63         int Split::GetAtInt(size_t i)
  64         {
  65                 if(i >= GetCount()) throw Exception(_T("Index out of bounds"));
  66                 return _wtoi(GetAt(i));
  67         }
  68
  69         float Split::GetAtFloat(size_t i)
  70         {
  71                 if(i >= GetCount()) throw Exception(_T("Index out of bounds"));
  72                 return (float)_wtof(GetAt(i));
  73         }
  74 }