src/subtitles/libssf/SubtitleFile.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 "SubtitleFile.h"
  24
  25 namespace ssf
  26 {
  27         SubtitleFile::SubtitleFile()
  28         {
  29         }
  30
  31         SubtitleFile::~SubtitleFile()
  32         {
  33         }
  34
  35         void SubtitleFile::Parse(InputStream& s)
  36         {
  37                 m_segments.RemoveAll();
  38
  39                 __super::Parse(s, s_predef);
  40
  41                 // TODO: check file.format == "ssf" and file.version == 1
  42
  43                 CAtlList<Definition*> defs;
  44                 GetRootRef()->GetChildDefs(defs, L"subtitle");
  45
  46                 StringMapW<float> offset;
  47
  48                 POSITION pos = defs.GetHeadPosition();
  49                 while(pos)
  50                 {
  51                         Definition* pDef = defs.GetNext(pos);
  52
  53                         try
  54                         {
  55                                 Definition::Time time;
  56
  57                                 if(pDef->GetAsTime(time, offset) && (*pDef)[L"@"].IsValue())
  58                                 {
  59                                         m_segments.Insert(time.start.value, time.stop.value, pDef);
  60                                 }
  61                         }
  62                         catch(Exception&)
  63                         {
  64                         }
  65                 }
  66         }
  67
  68         void SubtitleFile::Append(InputStream& s, float start, float stop, bool fSetTime)
  69         {
  70                 Reference* pRootRef = GetRootRef();
  71
  72                 ParseDefs(s, pRootRef);
  73
  74                 CAtlList<Definition*> defs;
  75                 GetNewDefs(defs);
  76
  77                 POSITION pos = defs.GetHeadPosition();
  78                 while(pos)
  79                 {
  80                         Definition* pDef = defs.GetNext(pos);
  81
  82                         if(pDef->m_parent == pRootRef && pDef->m_type == L"subtitle" && (*pDef)[L"@"].IsValue())
  83                         {
  84                                 m_segments.Insert(start, stop, pDef);
  85
  86                                 if(fSetTime)
  87                                 {
  88                                         try
  89                                         {
  90                                                 Definition::Time time;
  91                                                 StringMapW<float> offset;
  92                                                 pDef->GetAsTime(time, offset);
  93                                                 if(time.start.value == start && time.stop.value == stop)
  94                                                         continue;
  95                                         }
  96                                         catch(Exception&)
  97                                         {
  98                                         }
  99
 100                                         CStringW str;
 101                                         str.Format(L"%.3f", start);
 102                                         pDef->SetChildAsNumber(L"time.start", str, L"s");
 103                                         str.Format(L"%.3f", stop);
 104                                         pDef->SetChildAsNumber(L"time.stop", str, L"s");
 105                                 }
 106                         }
 107                 }
 108
 109                 Commit();
 110         }
 111
 112         bool SubtitleFile::Lookup(float at, CAutoPtrList<Subtitle>& subs)
 113         {
 114                 if(!subs.IsEmpty()) {ASSERT(0); return false;}
 115
 116                 CAtlList<SegmentItem> sis;
 117                 m_segments.Lookup(at, sis);
 118
 119                 POSITION pos = sis.GetHeadPosition();
 120                 while(pos)
 121                 {
 122                         SegmentItem& si = sis.GetNext(pos);
 123
 124                         CAutoPtr<Subtitle> s(new Subtitle(this));
 125
 126                         if(s->Parse(si.pDef, si.start, si.stop, at))
 127                         {
 128                                 for(POSITION pos = subs.GetHeadPosition(); pos; subs.GetNext(pos))
 129                                 {
 130                                         if(s->m_layer < subs.GetAt(pos)->m_layer)
 131                                         {
 132                                                 subs.InsertBefore(pos, s);
 133                                                 break;
 134                                         }
 135                                 }
 136
 137                                 if(s)
 138                                 {
 139                                         subs.AddTail(s);
 140                                 }
 141                         }
 142                 }
 143
 144                 return !subs.IsEmpty();
 145         }
 146
 147         //
 148
 149         SubtitleFile::Segment::Segment(float start, float stop, const SegmentItem* si)
 150         {
 151                 m_start = start;
 152                 m_stop = stop;
 153                 if(si) AddTail(*si);
 154         }
 155
 156         SubtitleFile::Segment::Segment(const Segment& s)
 157         {
 158                 *this = s;
 159         }
 160
 161         void SubtitleFile::Segment::operator = (const Segment& s)
 162         {
 163                 m_start = s.m_start;
 164                 m_stop = s.m_stop;
 165                 RemoveAll();
 166                 AddTailList(&s);
 167         }
 168
 169         //
 170
 171         void SubtitleFile::SegmentList::RemoveAll()
 172         {
 173                 __super::RemoveAll();
 174                 m_index.RemoveAll();
 175         }
 176
 177         void SubtitleFile::SegmentList::Insert(float start, float stop, Definition* pDef)
 178         {
 179                 if(start >= stop) {ASSERT(0); return;}
 180
 181                 m_index.RemoveAll();
 182
 183                 SegmentItem si = {pDef, start, stop};
 184
 185                 if(IsEmpty())
 186                 {
 187                         AddTail(Segment(start, stop, &si));
 188                         return;
 189                 }
 190
 191                 Segment& head = GetHead();
 192                 Segment& tail = GetTail();
 193
 194                 if(start >= tail.m_stop && stop > tail.m_stop)
 195                 {
 196                         if(start > tail.m_stop) AddTail(Segment(tail.m_stop, start));
 197                         AddTail(Segment(start, stop, &si));
 198                 }
 199                 else if(start < head.m_start && stop <= head.m_start)
 200                 {
 201                         if(stop < head.m_start) AddHead(Segment(stop, head.m_start));
 202                         AddHead(Segment(start, stop, &si));
 203                 }
 204                 else
 205                 {
 206                         if(start < head.m_start)
 207                         {
 208                                 AddHead(Segment(start, head.m_start, &si));
 209                                 start = head.m_start;
 210                         }
 211
 212                         if(stop > tail.m_stop)
 213                         {
 214                                 AddTail(Segment(tail.m_stop, stop, &si));
 215                                 stop = tail.m_stop;
 216                         }
 217
 218                         for(POSITION pos = GetHeadPosition(); pos; GetNext(pos))
 219                         {
 220                                 Segment& s = GetAt(pos);
 221
 222                                 if(start >= s.m_stop) continue;
 223                                 if(stop <= s.m_start) break;
 224
 225                                 if(s.m_start < start && start < s.m_stop)
 226                                 {
 227                                         Segment s2 = s;
 228                                         s2.m_start = start;
 229                                         InsertAfter(pos, s2);
 230                                         s.m_stop = start;
 231                                 }
 232                                 else if(s.m_start == start)
 233                                 {
 234                                         if(stop > s.m_stop)
 235                                         {
 236                                                 start = s.m_stop;
 237                                         }
 238                                         else if(stop < s.m_stop)
 239                                         {
 240                                                 Segment s2 = s;
 241                                                 s2.m_start = stop;
 242                                                 InsertAfter(pos, s2);
 243                                                 s.m_stop = stop;
 244                                         }
 245
 246                                         s.AddTail(si);
 247                                 }
 248                         }
 249                 }
 250         }
 251
 252         size_t SubtitleFile::SegmentList::Index(bool fForce)
 253         {
 254                 if(m_index.IsEmpty() || fForce)
 255                 {
 256                         m_index.RemoveAll();
 257                         POSITION pos = GetHeadPosition();
 258                         while(pos) m_index.Add(&GetNext(pos));
 259                 }
 260
 261                 return m_index.GetCount();
 262         }
 263
 264         void SubtitleFile::SegmentList::Lookup(float at, CAtlList<SegmentItem>& sis)
 265         {
 266                 sis.RemoveAll();
 267
 268                 size_t k;
 269                 if(Lookup(at, k))
 270                 {
 271                         sis.AddTailList(GetSegment(k));
 272                 }
 273         }
 274
 275         bool SubtitleFile::SegmentList::Lookup(float at, size_t& k)
 276         {
 277                 if(!Index()) return false;
 278
 279                 size_t i = 0, j = m_index.GetCount()-1;
 280
 281                 if(m_index[i]->m_start <= at && at < m_index[j]->m_stop)
 282                 do
 283                 {
 284                         k = (i+j)/2;
 285                         if(m_index[k]->m_start <= at && at < m_index[k]->m_stop) {return true;}
 286                         else if(at < m_index[k]->m_start) {if(j == k) k--; j = k;}
 287                         else if(at >= m_index[k]->m_stop) {if(i == k) k++; i = k;}
 288                 }
 289                 while(i <= j);
 290
 291                 return false;
 292         }
 293
 294         const SubtitleFile::Segment* SubtitleFile::SegmentList::GetSegment(size_t k)
 295         {
 296                 return 0 <= k && k < m_index.GetCount() ? m_index[k] : NULL;
 297         }
 298
 299         // TODO: this should be overridable from outside
 300
 301         LPCWSTR SubtitleFile::s_predef =
 302                 L"color#white {a: 255; r: 255; g: 255; b: 255;}; \n"
 303                 L"color#black {a: 255; r: 0; g: 0; b: 0;}; \n"
 304                 L"color#gray {a: 255; r: 128; g: 128; b: 128;}; \n"
 305                 L"color#red {a: 255; r: 255; g: 0; b: 0;}; \n"
 306                 L"color#green {a: 255; r: 0; g: 255; b: 0;}; \n"
 307                 L"color#blue {a: 255; r: 0; g: 0; b: 255;}; \n"
 308                 L"color#cyan {a: 255; r: 0; g: 255; b: 255;}; \n"
 309                 L"color#yellow {a: 255; r: 255; g: 255; b: 0;}; \n"
 310                 L"color#magenta {a: 255; r: 255; g: 0; b: 255;}; \n"
 311                 L" \n"
 312                 L"align#topleft {v: \"top\"; h: \"left\";}; \n"
 313                 L"align#topcenter {v: \"top\"; h: \"center\";}; \n"
 314                 L"align#topright {v: \"top\"; h: \"right\";}; \n"
 315                 L"align#middleleft {v: \"middle\"; h: \"left\";}; \n"
 316                 L"align#middlecenter {v: \"middle\"; h: \"center\";}; \n"
 317                 L"align#middleright {v: \"middle\"; h: \"right\";}; \n"
 318                 L"align#bottomleft {v: \"bottom\"; h: \"left\";}; \n"
 319                 L"align#bottomcenter {v: \"bottom\"; h: \"center\";}; \n"
 320                 L"align#bottomright {v: \"bottom\"; h: \"right\";}; \n"
 321                 L" \n"
 322                 L"time#time {scale: 1;}; \n"
 323                 L"time#startstop {start: \"start\"; stop: \"stop\";}; \n"
 324                 L" \n"
 325                 L"#b {font.weight: \"bold\"}; \n"
 326                 L"#i {font.italic: \"true\"}; \n"
 327                 L"#u {font.underline: \"true\"}; \n"
 328                 L"#s {font.strikethrough: \"true\"}; \n"
 329                 L" \n"
 330                 L"#nobr {linebreak: \"none\"}; \n"
 331                 L" \n"
 332                 L"subtitle#subtitle \n"
 333                 L"{ \n"
 334                 L"      frame \n"
 335                 L"      { \n"
 336                 L"              reference: \"video\"; \n"
 337                 L"              resolution: {cx: 640; cy: 480;}; \n"
 338                 L"      }; \n"
 339                 L" \n"
 340                 L"      direction \n"
 341                 L"      { \n"
 342                 L"              primary: \"right\"; \n"
 343                 L"              secondary: \"down\"; \n"
 344                 L"      }; \n"
 345                 L" \n"
 346                 L"      wrap: \"normal\"; \n"
 347                 L" \n"
 348                 L"      layer: 0; \n"
 349                 L" \n"
 350                 L"      style \n"
 351                 L"      { \n"
 352                 L"              linebreak: \"word\"; \n"
 353                 L" \n"
 354                 L"              placement \n"
 355                 L"              { \n"
 356                 L"                      clip: \"none\"; \n"
 357                 L"                      margin: {t: 0; r: 0; b: 0; l: 0;}; \n"
 358                 L"                      align: bottomcenter; \n"
 359                 L"                      pos: \"auto\" \n"
 360                 L"                      offset: {x: 0; y: 0;}; \n"
 361                 L"                      angle: {x: 0; y: 0; z: 0;}; \n"
 362                 L"                      org: \"auto\" \n"
 363                 L"                      path: \"\"; \n"
 364                 L"              }; \n"
 365                 L" \n"
 366                 L"              font \n"
 367                 L"              { \n"
 368                 L"                      face: \"Arial\"; \n"
 369                 L"                      size: 20; \n"
 370                 L"                      weight: \"bold\"; \n"
 371                 L"                      color: white; \n"
 372                 L"                      underline: \"false\"; \n"
 373                 L"                      strikethrough: \"false\"; \n"
 374                 L"                      italic: \"false\"; \n"
 375                 L"                      spacing: 0; \n"
 376                 L"                      scale: {cx: 1; cy: 1;}; \n"
 377                 L"                      kerning: \"true\"; \n"
 378                 L"              }; \n"
 379                 L" \n"
 380                 L"              background \n"
 381                 L"              { \n"
 382                 L"                      color: black; \n"
 383                 L"                      size: 2; \n"
 384                 L"                      type: \"outline\"; \n"
 385                 L"                      blur: 0; \n"
 386                 L"              }; \n"
 387                 L" \n"
 388                 L"              shadow \n"
 389                 L"              { \n"
 390                 L"                      color: black {a: 128;}; \n"
 391                 L"                      depth: 2; \n"
 392                 L"                      angle: -45; \n"
 393                 L"                      blur: 0; \n"
 394                 L"              }; \n"
 395                 L" \n"
 396                 L"              fill \n"
 397                 L"              { \n"
 398                 L"                      color: yellow; \n"
 399                 L"                      width: 0; \n"
 400                 L"              }; \n"
 401                 L"      }; \n"
 402                 L"}; \n";
 403 }