/*
 *
 *  Copyright (C) 2016, Open Connections GmbH
 *  All rights reserved.  See COPYRIGHT file for details.
 *
 *  This software and supporting documentation are maintained by
 *
 *    OFFIS e.V.
 *    R&D Division Health
 *    Escherweg 2
 *    D-26121 Oldenburg, Germany
 *
 *
 *  Module:  dcmfg
 *
 *  Author:  Michael Onken
 *
 *  Purpose: Class for managing Derivation Image Functional Group
 *
 */


#include "dcmtk/config/osconfig.h"
#include "dcmtk/dcmdata/dcitem.h"
#include "dcmtk/dcmdata/dcdatutl.h"
#include "dcmtk/dcmfg/fgderimg.h"
#include "dcmtk/dcmiod/iodutil.h"


FGDerivationImage::FGDerivationImage() :
  FGBase(DcmFGTypes::EFG_DERIVATIONIMAGE),
  m_DerivationImageItems()
{
}


FGDerivationImage::~FGDerivationImage()
{
  DcmIODUtil::freeContainer(m_DerivationImageItems);
}


FGBase* FGDerivationImage::clone() const
{
  FGDerivationImage* copy = new FGDerivationImage();
  if (copy)
  {
    OFVector<DerivationImageItem*>::const_iterator it = m_DerivationImageItems.begin();
    while ( it != m_DerivationImageItems.end() )
    {
      DerivationImageItem* itemCopy = new DerivationImageItem();
      if (!itemCopy)
      {
        DCMFG_FATAL("Cannot clone FGDerivationImage: Memory exhausted");
        return NULL;
      }
      *itemCopy = **it;
      copy->getDerivationImageItems().push_back(itemCopy);
      it++;
    }
  }
  return copy;
}


FGDerivationImage* FGDerivationImage::createMinimal(const OFVector<ImageSOPInstanceReferenceMacro>& derivationImages,
                                                    const OFString& derivationDescription,
                                                    const CodeSequenceMacro& derivationCode,
                                                    const CodeSequenceMacro& purposeOfReference)
{
  OFCondition result;
  FGDerivationImage* group = new FGDerivationImage();
  if (group)
  {
    DerivationImageItem* derivItem = NULL;
    result = group->addDerivationImageItem(derivationCode, derivationDescription, derivItem);
    if (result.good())
    {
      OFVector<SourceImageItem*> srcImages = derivItem->getSourceImageItems();
      OFVector<ImageSOPInstanceReferenceMacro>::const_iterator imageRef = derivationImages.begin();
      while (imageRef != derivationImages.end())
      {
        SourceImageItem* item = new SourceImageItem();
        if (item)
        {
          item->getPurposeOfReferenceCode() = purposeOfReference;
          item->getImageSOPInstanceReference() = *imageRef;
          derivItem->getSourceImageItems().push_back(item);
        }
        else
        {
          DCMFG_ERROR("Cannot create derivation description: Out of memory");
          result = EC_MemoryExhausted;
        }
        imageRef++;
      }
    }
    else
    {
      DCMFG_ERROR("Cannot create derivation description: " << result.text());
    }
  }
  else
  {
    result = EC_MemoryExhausted;
  }

  if (result.bad())
  {
    delete group;
    group = NULL;
  }

  return group;
}


int FGDerivationImage::compare(const FGBase& rhs) const
{
  int result = FGBase::compare(rhs);
  if (result != 0)
    return result;

  const FGDerivationImage* myRhs = OFstatic_cast(const FGDerivationImage*, &rhs);

  size_t thisSize = m_DerivationImageItems.size();
  size_t rhsSize = myRhs->m_DerivationImageItems.size();
  size_t maxSize = (thisSize > rhsSize) ? thisSize : rhsSize;
  for (size_t count = 0; count < maxSize; count++)
  {
    int error = (*(m_DerivationImageItems[count])).compare( (*(myRhs->m_DerivationImageItems[count])) );
    if (error != 0)
    {
      return error;
    }
  }

  if (thisSize < rhsSize)
  {
    return 1;
  }
  else if (thisSize > rhsSize)
  {
    return -1;
  }

  return 0;

}


void FGDerivationImage::clearData()
{
  DcmIODUtil::freeContainer(m_DerivationImageItems);
}


OFCondition FGDerivationImage::check() const
{
  OFCondition result;
  OFVector<DerivationImageItem*>::const_iterator it = m_DerivationImageItems.begin();
  while ( (it != m_DerivationImageItems.end()) && result.good() )
  {
    result = (*it)->check();
    it++;
  }

  return result;
}

DerivationImageItem& DerivationImageItem::operator=(const DerivationImageItem& rhs)
{
  m_DerivationDescription = rhs.m_DerivationDescription;
  // Copy nested structures: Derivation Codes
  OFVector<CodeSequenceMacro*>::const_iterator code = rhs.m_DerivationCodeItems.begin();
  while ( code != rhs.m_DerivationCodeItems.end() )
  {
    m_DerivationCodeItems.push_back(new CodeSequenceMacro(**code));
    code++;
  }
  // Copy nested structures: Derivation Codes
  OFVector<SourceImageItem*>::const_iterator source = rhs.m_SourceImageItems.begin();
  while ( source != rhs.m_SourceImageItems.end() )
  {
    m_SourceImageItems.push_back(new SourceImageItem(**source));
    source++;
  }
  return *this;
}


OFCondition FGDerivationImage::addDerivationImageItem(const CodeSequenceMacro& derivationCode,
                                                      const OFString& derivationDescription,
                                                      DerivationImageItem*& item)
{
  OFCondition result;
  item = new DerivationImageItem();
  if (item == NULL)
  {
    return EC_MemoryExhausted;
  }

  // Initialize with given data
  if (result.good())
  {
    result = item->setDerivationDescription(derivationDescription);
  }
  if (result.good())
  {
    item->getDerivationCodeItems().push_back(new CodeSequenceMacro(derivationCode));
    m_DerivationImageItems.push_back(item);
  }
  else
  {
    delete item;
    item = NULL;
    result = FG_EC_InvalidData;
  }
  return result;
}


OFVector<DerivationImageItem*>& FGDerivationImage::getDerivationImageItems()
{
  return m_DerivationImageItems;
}


OFCondition FGDerivationImage::read(DcmItem& item)
{
  /* re-initialize object */
  clearData();

  DcmSequenceOfItems *derivationImageSequence = NULL;
  if (item.findAndGetSequence(DCM_DerivationImageSequence, derivationImageSequence).bad())
    return FG_EC_NoSuchGroup;

  /* Read Derivation Image Sequence */
  DcmIODUtil::readSubSequence<OFVector<DerivationImageItem*> >
  ( derivationImageSequence,
    DCM_DerivationImageSequence,
    m_DerivationImageItems,
    "0-n",
    "2",
    "DerivationImageMacro");

  return EC_Normal;
}


OFCondition FGDerivationImage::write(DcmItem& item)
{
  OFCondition result = check();
  DcmIODUtil::writeSubSequence<OFVector<DerivationImageItem*> >
  ( result,
    DCM_DerivationImageSequence,
    m_DerivationImageItems,
    item,
    "0-n",
    "2",
    "DerivationImageMacro" );
  return result;
}


/* -- Class SourceImage Implementation -- */

SourceImageItem::SourceImageItem()
  : m_PurposeOfReferenceCode(),
    m_ImageSOPInstanceReference()
{
}


SourceImageItem::~SourceImageItem()
{
  clearData();
}


SourceImageItem& SourceImageItem::operator=(const SourceImageItem& rhs)
{
  m_PurposeOfReferenceCode = rhs.m_PurposeOfReferenceCode;
  m_ImageSOPInstanceReference = rhs.m_ImageSOPInstanceReference;
  return *this;
}


void SourceImageItem::clearData()
{
  m_PurposeOfReferenceCode.clearData();
  m_ImageSOPInstanceReference.clear();
}


OFCondition SourceImageItem::check() const
{
  OFCondition result = OFconst_cast(ImageSOPInstanceReferenceMacro*, &m_ImageSOPInstanceReference)->check();
  if (result.bad())
  {
    return result;
  }

  return OFconst_cast(CodeSequenceMacro*, &m_PurposeOfReferenceCode)->check();
}


int SourceImageItem::compare(const SourceImageItem& rhs) const
{
  int result = this->m_PurposeOfReferenceCode.compare(rhs.m_PurposeOfReferenceCode);
  if (result != 0)
  {
    result = this->m_ImageSOPInstanceReference.compare(rhs.m_ImageSOPInstanceReference);
  }

  return result;
}


CodeSequenceMacro& SourceImageItem::getPurposeOfReferenceCode()
{
  return m_PurposeOfReferenceCode;
}


ImageSOPInstanceReferenceMacro& SourceImageItem::getImageSOPInstanceReference()
{
  return m_ImageSOPInstanceReference;
}


OFCondition SourceImageItem::read(DcmItem& itemOfSourceImageSequence,
                                  const OFBool clearOldData)
{
  /* Re-initialize object */
  if (clearOldData)
    clearData();

  /* Read Purpose of Reference Code Sequence */
  DcmIODUtil::readSingleItem<CodeSequenceMacro>(itemOfSourceImageSequence, DCM_PurposeOfReferenceCodeSequence, m_PurposeOfReferenceCode, "1", "DerivationImageMacro");

  /* Read Source Image Sequence */
  m_ImageSOPInstanceReference.read(itemOfSourceImageSequence);

  return EC_Normal;
}


OFCondition SourceImageItem::write(DcmItem& itemOfSourceImageSequence)
{
  OFCondition result = EC_Normal;

  /* Write Purpose of Reference Code Sequence */
  DcmIODUtil::writeSingleItem<CodeSequenceMacro>(result, DCM_PurposeOfReferenceCodeSequence, m_PurposeOfReferenceCode, itemOfSourceImageSequence, "1", "DerivationImageMacro");

  /* Write Image SOP Instance Reference Macro */
  if ( result.good() )
  {
    m_ImageSOPInstanceReference.write(itemOfSourceImageSequence);
  }

  return result;
}


/* -- Class DerivationImageItem Implementation -- */

DerivationImageItem::DerivationImageItem()
  : m_DerivationDescription(DCM_DerivationDescription),
    m_DerivationCodeItems(),
    m_SourceImageItems()
{
}


DerivationImageItem::DerivationImageItem(const DerivationImageItem& rhs)
 : m_DerivationDescription(rhs.m_DerivationDescription),
   m_DerivationCodeItems(),
   m_SourceImageItems()
{
  OFVector<CodeSequenceMacro*>::const_iterator it = rhs.m_DerivationCodeItems.begin();
  while (it != rhs.m_DerivationCodeItems.end())
  {
    CodeSequenceMacro* copy = new CodeSequenceMacro(**it);
    if (!copy)
    {
      DCMFG_ERROR("Could not create copy of DerivationImageItem: Memory exhausted");
      this->clearData();
      return;
    }
    m_DerivationCodeItems.push_back(copy);
    it++;
  }

  OFVector<SourceImageItem*>::const_iterator src = rhs.m_SourceImageItems.begin();
  while (src != rhs.m_SourceImageItems.end())
  {
    SourceImageItem* copy = new SourceImageItem(**src);
    if (!copy)
    {
      DCMFG_ERROR("Could not create copy of DerivationImageItem: Memory exhausted");
      this->clearData();
      return;
    }
    m_SourceImageItems.push_back(copy);
    it++;
  }
}


void DerivationImageItem::clearData()
{
  m_DerivationDescription.clear();
  DcmIODUtil::freeContainer(m_DerivationCodeItems);
  DcmIODUtil::freeContainer(m_SourceImageItems);
}


int DerivationImageItem::compare(const DerivationImageItem& rhs) const
{
  if (this == &rhs)
    return OFTrue;

  int result = m_DerivationDescription.compare(rhs.m_DerivationDescription);
  if (result != 0) return result;

  size_t numThis = m_DerivationCodeItems.size();
  size_t numRhs = rhs.m_DerivationCodeItems.size();
  size_t maxSize = (numThis > numRhs) ? numThis : numRhs;
  for (size_t count = 0; count < maxSize; count++)
  {
    int error = (*m_DerivationCodeItems[count]).compare(*(rhs.m_DerivationCodeItems[count]));
    if (error != 0)
    {
      return error;
    }
  }
  if (numThis < numRhs)
  {
    return 1;
  }
  else if (numThis > numRhs)
  {
    return -1;
  }

  numThis = m_SourceImageItems.size();
  numRhs = rhs.m_SourceImageItems.size();
  maxSize = (numThis > numRhs) ? numThis : numRhs;
  for (size_t count = 0; count < maxSize; count++)
  {
    int error = (*m_SourceImageItems[count]).compare(*(rhs.m_SourceImageItems[count]));
    if (error != 0)
    {
      return error;
    }
  }
  if (numThis < numRhs)
  {
    return 1;
  }
  else if (numThis > numRhs)
  {
    return -1;
  }

  // If we get here, the object values must be identical
  return 0;
}


OFCondition DerivationImageItem::check() const
{
  if (m_DerivationCodeItems.size() == 0)
  {
    DCMFG_ERROR("Derivation Code Sequence in Derivation Image Functional Group Macro must have one or more items");
    return FG_EC_InvalidData;
  }

  OFCondition result;
  OFVector<CodeSequenceMacro*>::const_iterator it = m_DerivationCodeItems.begin();
  while (it != m_DerivationCodeItems.end() && result.good())
  {
    result = (*it)->check();
    it++;
  }

  OFVector<SourceImageItem*>::const_iterator ref = m_SourceImageItems.begin();
  while ((ref != m_SourceImageItems.end()) && result.good())
  {
    result = (*ref)->check();
    ref++;
  }
  return result;
}


DerivationImageItem::~DerivationImageItem()
{
  clearData();
}


OFCondition DerivationImageItem::addSourceImageItem(const OFString& file,
                                                    const CodeSequenceMacro& purposeOfReference,
                                                    SourceImageItem*& resultSourceImageItem)
{
  DcmFileFormat dcmff;
  DcmDataset *dataset = NULL;
  OFCondition result = dcmff.loadFile(file.c_str());
  if (result.bad())
  {
    DCMFG_ERROR("Could not load file " << file << ": " << result.text());
    return result;
  }
  dataset = dcmff.getDataset();
  return addSourceImageItem(dataset, purposeOfReference, resultSourceImageItem);
}


OFCondition DerivationImageItem::addSourceImageItem(DcmDataset *dataset,
                                                    const CodeSequenceMacro& purposeOfReference,
                                                    SourceImageItem*& resultSourceImageItem)
{
  // Create new source image item
  SourceImageItem *item = new SourceImageItem();
  if (item == NULL)
  {
    return EC_MemoryExhausted;
  }

  // check code validity
  if ( OFconst_cast(CodeSequenceMacro*, &purposeOfReference)->check().bad() )
  {
    DCMFG_ERROR("Purpose of Reference code within item of Derivation Image Sequence is invalid");
    return FG_EC_InvalidData;
  }

  item->getPurposeOfReferenceCode() = purposeOfReference;
  resultSourceImageItem = NULL;
  OFString sopClass, sopInstance, ts;
  OFCondition result = DcmDataUtil::getSOPInstanceFromDataset(dataset, EXS_Unknown, sopClass, sopInstance, ts /*ignored*/);
  {
    if ( result.good() ) result = item->getImageSOPInstanceReference().setReferencedSOPClassUID(sopClass);
    if ( result.good() ) result = item->getImageSOPInstanceReference().setReferencedSOPInstanceUID(sopInstance);
  }
  if (result.good())
  {
    m_SourceImageItems.push_back(item);
    resultSourceImageItem = item;
  }
  else
  {
    delete item;
  }

  return result;
}


OFCondition DerivationImageItem::addSourceImageItems(const OFVector< OFString >& files,
                                                     const CodeSequenceMacro& purposeOfReference,
                                                     OFVector<SourceImageItem*>& resultSourceImageItems,
                                                     const OFBool skipErrors)
{
  DcmFileFormat dcmff;
  OFCondition result;
  DcmDataset *dataset = NULL;
  OFVector<DcmDataset*> datasets;
  OFVector<OFString>::const_iterator it = files.begin();
  while (it != files.end())
  {
    result = dcmff.loadFile((*it).c_str());
    if (result.bad())
    {
      DCMFG_ERROR("Could not load file " << (*it) << ": " << result.text());
      return result;
    }
    dataset = dcmff.getDataset();
    datasets.push_back(dataset);
    it++;
  }
  return addSourceImageItems(datasets, purposeOfReference, resultSourceImageItems, skipErrors);
}


OFCondition DerivationImageItem::addSourceImageItems(const OFVector<DcmDataset*>& datasets,
                                                     const CodeSequenceMacro& purposeOfReference,
                                                     OFVector<SourceImageItem*>& resultSourceImageItems,
                                                     const OFBool skipErrors)
  {
  OFCondition result;
  OFVector<DcmDataset*>::const_iterator it = datasets.begin();
  while (it != datasets.end())
  {
    SourceImageItem* resultItem = NULL;
    result = addSourceImageItem((*it), purposeOfReference, resultItem);
    if (result.good())
    {
      resultSourceImageItems.push_back(resultItem);
    }
    else if (result.bad() && !skipErrors)
    {
      DCMFG_ERROR("Could not add Source Image Sequence item for file " << *it << ": " << result.text());
      break;
    }
    else if (result.bad())
    {
      DCMFG_WARN("Could not add Source Image Sequence item for file " << *it << ": " << result.text());
    }
    it++;
  }

  return result;
}


OFVector<CodeSequenceMacro*>& DerivationImageItem::getDerivationCodeItems()
{
  return m_DerivationCodeItems;
}


OFCondition DerivationImageItem::getDerivationDescription(OFString& value,
                                                          const long signed int pos) const
{
  return DcmIODUtil::getStringValueFromElement(m_DerivationDescription, value, pos);
}


OFCondition DerivationImageItem::setDerivationDescription(const OFString& value,
                                                          const OFBool checkValue)
{
  (void)checkValue;
  return m_DerivationDescription.putOFStringArray(value);
}


OFVector<SourceImageItem*>& DerivationImageItem::getSourceImageItems()
{
  return m_SourceImageItems;
}


OFCondition DerivationImageItem::read(DcmItem& itemOfDerivationImageSequence,
                                      const OFBool clearOldData)
{
  OFCondition result = EC_Normal;

  /* re-initialize object */
  if (clearOldData)
    clearData();

  /* Derivation Description */
  DcmIODUtil::getAndCheckElementFromDataset(itemOfDerivationImageSequence, m_DerivationDescription, "1" /* vm */, "3" /* type */, "DerivationImageMacro");

  /* Derivation Code Sequence */
  DcmIODUtil::readSubSequence<OFVector<CodeSequenceMacro*> >
  (  itemOfDerivationImageSequence,
     DCM_DerivationCodeSequence,
     m_DerivationCodeItems,
     "1" /* vm */,
     "3" /* type */,
     "DerivationImageMacro" );


  /* Source Image Sequence */
  DcmIODUtil::readSubSequence<OFVector<SourceImageItem*> >
  ( itemOfDerivationImageSequence,
    DCM_SourceImageSequence,
    m_SourceImageItems,
    "0-n" /* vm */,
    "2" /* type */,
    "DerivationImageMacro" );

    return result;
}


OFCondition DerivationImageItem::write(DcmItem& itemOfDerivationImageSequence)
{
  OFCondition result = EC_Normal;

  /* Write Derivation Description */
  DcmIODUtil::copyElementToDataset(result, itemOfDerivationImageSequence, m_DerivationDescription, "1", "3", "DerivationImageMacro");

  /* Write Derivation Code Sequence */
  DcmIODUtil::writeSubSequence<OFVector<CodeSequenceMacro*> >
  ( result,
    DCM_DerivationCodeSequence,
    m_DerivationCodeItems,
    itemOfDerivationImageSequence,
    "1-n",
    "1",
    "DerivationImageMacro" );

  /* Write Source Image Sequence */
  DcmIODUtil::writeSubSequence<OFVector<SourceImageItem*> >
  ( result,
    DCM_SourceImageSequence,
    m_SourceImageItems,
    itemOfDerivationImageSequence,
    "0-n",
    "2",
    "DerivationImageMacro" );

  return result;
}

