/*******************************************************************************
 *  Copyright 2017-2018 Centre for Development of Advanced Computing(C-DAC), Pune
 *  
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *  
 *   http://www.apache.org/licenses/LICENSE-2.0
 *  
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *******************************************************************************/
package cdac.medinfo.sdk.dcm30_04.services.cstore.commons;

import java.io.IOException;
import java.util.Locale;

import cdac.medinfo.sdk.dcm30_04.commoninterface.IDataSet;
import cdac.medinfo.sdk.dcm30_04.enums.EnumCommandField;
import cdac.medinfo.sdk.dcm30_04.exceptions.MandatoryMissingException;
import cdac.medinfo.sdk.dcm30_04.services.dicomnetcommons.EnumPriority;
import cdac.medinfo.sdk.dcm30_04.services.dicomnetcommons.ICommandElement;
import cdac.medinfo.sdk.dcm30_04.services.dicomnetcommons.ICommandSet;
import cdac.medinfo.sdk.dcm30_04.services.dicomnetcommons.InvalidCommandSetException;
import cdac.medinfo.sdk.dcm30_04.services.dicomnetcommons.InvalidDICOMMessageException;
import cdac.medinfo.sdk.dcm30_04.services.dimsecommons.CommandElement;
import cdac.medinfo.sdk.dcm30_04.services.dimsecommons.DIMSERequest;
import cdac.medinfo.sdk.dcm30_04.services.dimsecommons.DicomMessage;
import cdac.medinfo.sdk.dcm30_04.utils.DicomConfig;
import cdac.medinfo.sdk.dcm30_04.utils.DicomConstants;
import cdac.medinfo.sdk.dcm30_04.utils.DicomLogger;
import cdac.medinfo.sdk.dcm30_04.utils.ICollectionIterator;
import cdac.medinfo.sdk.dcm30_04.utils.ISingleCollection;
import cdac.medinfo.sdk.dcm30_04.utils.ResourceBundleWrapper;
import cdac.medinfo.sdk.dcm30_04.utils.SingleCollection;

// TODO: Auto-generated Javadoc
/**
 * The C-STORE service is invoked by a DIMSE-service-user to request the storage of Composite SOP Instance information by a peer DIMSE-service-user.
 * This class provides elements necessary to construct C-Store Request to perform C-Store operation.
 * The getter and setter methods of all attributes required for C-Store request is provided.
 */
public class CStoreRQ extends DIMSERequest 
{
	/**
	 * Instantiates a new CStore RQ. Creates default Dicom Message for CStore RQ.
	 */
	public CStoreRQ()
	{
		
	}

	/**
	 * Instantiates a new CStore RQ when Dicom Message is given by user.
	 * Retrieves Command Set and Data Set from the DicomMessage and sets in CStore Request.
	 * 
	 * @param objDicomMessage Object of DicomMessage from which C-Store request is initialized.
	 * 
	 * @throws IOException Signals that an I/O exception has occurred.
	 */
	public CStoreRQ(DicomMessage objDicomMessage) throws IOException
	{
		objICommandSet = objDicomMessage.getCommandSet();
		objIDataSet = objDicomMessage.getDataSet();
	}

	/**
	 * Sets Priority
	 * The priority shall be set to one of the following values:
	 * LOW = 0002H
	 * MEDIUM = 0000H
	 * HIGH = 0001H.
	 * 
	 * @param enumPriority enum value of Priority
	 */
	public void setPriority(EnumPriority enumPriority)
	{
		ICommandElement objICommandElement = null;
		objICommandElement = new CommandElement(0x0000, 0x0700,enumPriority.getValue());
		objICommandSet.addCommandElement(objICommandElement);
	}

	/**
	 * Retrieves Priority .
	 * 
	 * @return short - value of Priority
	 */
	public int getPriority()
	{
		Object objValue = getValue(0x0000,0x0700);
		if(objValue != null)
			return Short.parseShort(objValue.toString());
		return DicomConstants.DCM_NO_VALUE;
	}
	
	/**
	 * Sets Affected SOP Class UID. . 
	 * SOP Class UID. associated with the operation.
	 * @param strAffectedSOPClassUID value of affected SOP Class UID..
	 */
	public void setAffectedSOPClassUID(String strAffectedSOPClassUID)
	{
		ICommandElement objICommandElement = null;
		objICommandElement = new CommandElement(0x0000, 0x0002,strAffectedSOPClassUID);
		objICommandSet.addCommandElement(objICommandElement);
	}

	/**
	 * Retrieves Affected SOP Class UID. . 
	 * SOP Class UID. associated with the operation.
	 * @return String - value of Affected SOP Class UID. .
	 */
	public String getAffectedSOPClassUID()
	{
		Object objValue = getValue(0x0000,0x0002);
		if(objValue != null)
			return objValue.toString();
		return null;
	}

	/**
	 * Sets the UID of the SOP Instance to be stored.
	 * @param strAffectedSOPInstanceUID String value of Affected SOP Instance UID.
	 */
	public void setAffectedSOPInstanceUID(String strAffectedSOPInstanceUID)
	{
		ICommandElement objICommandElement = null;
		objICommandElement = new CommandElement(0x0000, 0x1000,strAffectedSOPInstanceUID);
		objICommandSet.addCommandElement(objICommandElement);
	}

	/**
	 * Retrieves the UID of the SOP Instance to be stored.
	 * @return String - value of Priority
	 */
	public String getAffectedSOPInstanceUID()
	{
		Object objValue = getValue(0x0000,0x1000);
		if(objValue != null)
			return objValue.toString();
		return null;
	}

	/**
	 * Sets the DICOM AE Title of the DICOM AE which invoked the C-MOVE operation from which this C-STORE sub-operation is being performed.
	 * @param strMoveOriginatorApplicationEntityTitle String value of Move Originator Application Entity Title
	 */
	public void setMoveOriginatorApplicationEntityTitle(String strMoveOriginatorApplicationEntityTitle)
	{
		ICommandElement objICommandElement = null;
		objICommandElement = new CommandElement(0x0000, 0x1030,strMoveOriginatorApplicationEntityTitle);
		objICommandSet.addCommandElement(objICommandElement);
	}

	/**
	 * Retrieves Move Originator Application Entity Title.
	 * @return String - value of Move Originator Application Entity Title
	 */
	public String getMoveOriginatorApplicationEntityTitle()
	{
		Object objValue = getValue(0x0000,0x1030);
		if(objValue != null)
			return objValue.toString();
		return null;
	}

	/**
	 * Sets the Message ID (0000,0110) of the C CMOVE-RQ Message from which this C-STORE sub-operations is being performed.
	 * @param shMoveOriginatorMessageID short value of Move Originator Message ID.
	 */
	public void setMoveOriginatorMessageID(short shMoveOriginatorMessageID)
	{
		ICommandElement objICommandElement = null;
		objICommandElement = new CommandElement(0x0000, 0x1031,shMoveOriginatorMessageID);
		objICommandSet.addCommandElement(objICommandElement);
	}

	/**
	 * Retrieves Move Originator Message ID.
	 * @return short - value of Move Originator Message ID.
	 */
	public short getMoveOriginatorMessageID()
	{
		Object objValue = getValue(0x0000,0x1031);
		if(objValue != null)
			return (Short.parseShort(objValue.toString()));
		return DicomConstants.DCM_NO_VALUE;
	}

	/**
	 * This method makes CStore RQ's according to the given DataSet Collection.
	 * @param objIDataSetColl Collection of DataSet.
	 * @return ISingleCollection<CStoreRQ> Collection of CStore RQ.
	 */
	public static ISingleCollection<CStoreRQ> getDefaultRQ(ISingleCollection<IDataSet> objIDataSetColl)
	{
		ICollectionIterator<IDataSet> objDataSetColIterator = objIDataSetColl.iterator();
		ISingleCollection<CStoreRQ> objCollecCStoreRQ = new SingleCollection<CStoreRQ>();
		CStoreRQ objCStoreRQ = null;

		while(objDataSetColIterator.hasNext())
		{
			IDataSet objIDataSet = objDataSetColIterator.next();
			
			objCStoreRQ = new CStoreRQ();
			ICommandSet objICommandSet = objCStoreRQ.getCommandSet();
			
			objCStoreRQ.setCommandField(EnumCommandField.C_STORE_RQ);
			objCStoreRQ.setPriority(EnumPriority.HIGH);
			if(null != objIDataSet && !objIDataSet.isEmpty())
			{
				objCStoreRQ.setCommandDataSetType(DicomConstants.DCM_DATASET_PRESENT);
				String strSOPUID = objIDataSet.getDataForTag(0x0008,0x0016).getDicomValue().getValue().toString();
				String strSOPInstanceUID = objIDataSet.getDataForTag(0x0008,0x0018).getDicomValue().getValue().toString();
				
				objCStoreRQ.setAffectedSOPClassUID(strSOPUID);
				objCStoreRQ.setAffectedSOPInstanceUID(strSOPInstanceUID);
			}
			else
				objCStoreRQ.setCommandDataSetType(DicomConstants.DCM_DATASET_NOT_PRESENT);
			objCStoreRQ.setMessageID(shMsgID++);
			objCStoreRQ.setDataSet(objIDataSet);
			objCStoreRQ.setCommandGroupLength(objICommandSet.getCommandSetLength());
			
			// Validating RQ.
			objCStoreRQ.validate();
			
			objCollecCStoreRQ.add(objCStoreRQ);
		}
		return objCollecCStoreRQ;
	}

	/**
	 * This method makes CStore RQ's according to the given DataSet Collection.
	 * @param objIDataSet Collection of DataSet.
	 * @return CStoreRQ Collection of CStore RQ.
	 */
	public static CStoreRQ getDefaultRQ(IDataSet objIDataSet)
	{
		CStoreRQ objCStoreRQ = new CStoreRQ();
		
		ICommandSet objICommandSet = objCStoreRQ.getCommandSet();
		objCStoreRQ.setCommandField(EnumCommandField.C_STORE_RQ);
		objCStoreRQ.setPriority(EnumPriority.HIGH);
		if(null != objIDataSet && !objIDataSet.isEmpty())
		{
			objCStoreRQ.setCommandDataSetType(DicomConstants.DCM_DATASET_PRESENT);
			String strSOPUID = objIDataSet.getDataForTag(0x0008,0x0016).getDicomValue().getValue().toString();
			String strSOPInstanceUID = objIDataSet.getDataForTag(0x0008,0x0018).getDicomValue().getValue().toString();
			
			objCStoreRQ.setAffectedSOPClassUID(strSOPUID);
			objCStoreRQ.setAffectedSOPInstanceUID(strSOPInstanceUID);
		}
		else
			objCStoreRQ.setCommandDataSetType(DicomConstants.DCM_DATASET_NOT_PRESENT);
		objCStoreRQ.setMessageID(shMsgID++);
		objCStoreRQ.setDataSet(objIDataSet);
		objCStoreRQ.setCommandGroupLength(objICommandSet.getCommandSetLength());
		
		// Validating RQ.
		objCStoreRQ.validate();
		return objCStoreRQ;
	}
	
	/**
	 * Validates the request for their mandatory attributes only.
	 * @return true, if validation is successful.
	 * @throws InvalidDICOMMessageException If validation of mandatory fields fails.
	 */
	public boolean validate() throws InvalidDICOMMessageException
	{
		super.validate();
		try
		{	
			//Checks for service type.
			if(getCommandType() != EnumCommandField.C_STORE_RQ)
				throw new InvalidDICOMMessageException(ResourceBundleWrapper.getValue("exceptions","RQNotMatched",new Locale(DicomConfig.createInstance().getLocale())) + getCommandType().name());
			
			if(!objICommandSet.isPresent(0x0000,0x0002))			// Affected SOP Class UID.
				throw new MandatoryMissingException("0000,0002");
			else
			{
				if(objICommandSet.getValue(0x0000,0x0002) == null)
					throw new InvalidDICOMMessageException(ResourceBundleWrapper.getValue("exceptions", "NullValue",new Locale(DicomConfig.createInstance().getLocale())) + 0x0000 + "," + 0x0002);
			}
			
			if(!objICommandSet.isPresent(0x0000,0x1000))		// Affected SOP Instance UID
				throw new MandatoryMissingException("0000,1000");
			else
			{
				if(objICommandSet.getValue(0x0000,0x1000) == null)
					throw new InvalidDICOMMessageException(ResourceBundleWrapper.getValue("exceptions", "NullValue",new Locale(DicomConfig.createInstance().getLocale())) + 0x0000 + "," + 0x1000);
			}
			
			if(!objICommandSet.isPresent(0x0000,0x0700))		// Priority
				throw new MandatoryMissingException("0000,0700");
			else
			{
				if(objICommandSet.getValue(0x0000,0x0700) == null)
					throw new InvalidDICOMMessageException(ResourceBundleWrapper.getValue("exceptions", "NullValue",new Locale(DicomConfig.createInstance().getLocale())) + 0x0000 + "," + 0x0700);
			}
			
			//	For Identifier or Data Set
			if(objIDataSet == null || objIDataSet.isEmpty())
			{
				if(DicomConfig.createInstance().isLoggingOn())
					DicomLogger.getLogger().log(ResourceBundleWrapper.getValue("exceptions", "EmptyDataSet", new Locale(DicomConfig.createInstance().getLocale())));
				
				throw new InvalidDICOMMessageException(ResourceBundleWrapper.getValue("exceptions", "EmptyDataSet",new Locale(DicomConfig.createInstance().getLocale())));
			}
		}
		catch(MandatoryMissingException mme)
		{
			InvalidCommandSetException objCommandSetException = new InvalidCommandSetException(mme);
			if(DicomConfig.createInstance().isLoggingOn())
				DicomLogger.getLogger().log(mme.getMessage());

			throw new InvalidDICOMMessageException(this.getClass().getSimpleName(),objCommandSetException);
		}	
		return true;
	}
}
