45template <
class IT,
class NT>
49template <
class IT,
class NT>
54template <
class IT,
class NT>
55SpDCCols<IT,NT>::SpDCCols(
IT size,
IT nRow,
IT nCol,
IT nzc)
56:m(nRow), n(nCol), nnz(
size), splits(0)
59 dcsc =
new Dcsc<IT,NT>(nnz, nzc);
64template <
class IT,
class NT>
65SpDCCols<IT,NT>::~SpDCCols()
73 for(
int i=0; i<splits; ++i)
87template <
class IT,
class NT>
88SpDCCols<IT,NT>::SpDCCols(
const SpDCCols<IT,NT> & rhs)
89: m(rhs.m), n(rhs.n), nnz(rhs.nnz), splits(rhs.splits)
93 for(
int i=0; i<splits; ++i)
94 CopyDcsc(rhs.dcscarr[i]);
108template <
class IT,
class NT>
109SpDCCols<IT,NT>::SpDCCols(
const SpTuples<IT, NT> & rhs,
bool transpose)
110: m(rhs.m), n(rhs.n), nnz(rhs.nnz), splits(0)
115 if(transpose) std::swap(m,n);
124 for(
IT i=1; i< rhs.nnz; ++i)
126 if(rhs.rowindex(i) != rhs.rowindex(i-1))
131 dcsc =
new Dcsc<IT,NT>(rhs.nnz,localnzc);
132 dcsc->jc[0] = rhs.rowindex(0);
135 for(
IT i=0; i<rhs.nnz; ++i)
137 dcsc->ir[i] = rhs.colindex(i);
138 dcsc->numx[i] = rhs.numvalue(i);
142 for(
IT i=1; i<rhs.nnz; ++i)
144 if(rhs.rowindex(i) != dcsc->jc[jspos-1])
146 dcsc->jc[jspos] = rhs.rowindex(i);
147 dcsc->cp[jspos++] = i;
150 dcsc->cp[jspos] = rhs.nnz;
155 for(
IT i=1; i<rhs.nnz; ++i)
157 if(rhs.colindex(i) != rhs.colindex(i-1))
162 dcsc =
new Dcsc<IT,NT>(rhs.nnz,localnzc);
163 dcsc->jc[0] = rhs.colindex(0);
166 for(
IT i=0; i<rhs.nnz; ++i)
168 dcsc->ir[i] = rhs.rowindex(i);
169 dcsc->numx[i] = rhs.numvalue(i);
173 for(
IT i=1; i<rhs.nnz; ++i)
175 if(rhs.colindex(i) != dcsc->jc[jspos-1])
177 dcsc->jc[jspos] = rhs.colindex(i);
178 dcsc->cp[jspos++] = i;
181 dcsc->cp[jspos] = rhs.nnz;
197template <
class IT,
class NT>
198SpDCCols<IT,NT>::SpDCCols(
IT nRow,
IT nCol,
IT nTuples,
const std::tuple<IT, IT, NT>* tuples,
bool transpose)
199: m(nRow), n(nCol), nnz(nTuples), splits(0)
212 totThreads = omp_get_num_threads();
216 std::vector <IT> tstart(totThreads);
217 std::vector <IT> tend(totThreads);
218 std::vector <IT> tdisp(totThreads+1);
221 IT* temp_jc =
new IT[nTuples];
222 IT* temp_cp =
new IT[nTuples];
230 threadID = omp_get_thread_num();
232 IT start = threadID * (nTuples / totThreads);
233 IT end = (threadID + 1) * (nTuples / totThreads);
234 if(threadID == (totThreads-1)) end = nTuples;
238 temp_jc[start] = std::get<1>(tuples[start]);
239 temp_cp[start] = start;
240 for (
IT i = start+1; i < end; ++i)
242 if(std::get<1>(tuples[i]) != temp_jc[curpos] )
244 temp_jc[++curpos] = std::get<1>(tuples[i]);
250 tstart[threadID] = start;
251 if(end>start) tend[threadID] = curpos+1;
252 else tend[threadID] = end;
257 for(
int t=totThreads-1; t>0; --t)
259 if(tend[t] > tstart[t] && tend[t-1] > tstart[t-1])
261 if(temp_jc[tstart[t]] == temp_jc[tend[t-1]-1])
269 for(
int t=0; t<totThreads; ++t)
271 tdisp[t+1] = tdisp[t] + tend[t] - tstart[t];
274 IT localnzc = tdisp[totThreads];
275 dcsc =
new Dcsc<IT,NT>(nTuples,localnzc);
283 threadID = omp_get_thread_num();
285 std::copy(temp_jc + tstart[threadID], temp_jc + tend[threadID], dcsc->jc + tdisp[threadID]);
286 std::copy(temp_cp + tstart[threadID], temp_cp + tend[threadID], dcsc->cp + tdisp[threadID]);
288 dcsc->cp[localnzc] = nTuples;
294#pragma omp parallel for schedule (static)
296 for(
IT i=0; i<nTuples; ++i)
298 dcsc->ir[i] = std::get<0>(tuples[i]);
299 dcsc->numx[i] = std::get<2>(tuples[i]);
303 if(transpose) Transpose();
364template <
class IT,
class NT>
365SpDCCols<IT,NT> & SpDCCols<IT,NT>::operator=(
const SpDCCols<IT,NT> & rhs)
371 if(dcsc != NULL && nnz > 0)
377 dcsc =
new Dcsc<IT,NT>(*(rhs.dcsc));
393template <
class IT,
class NT>
394SpDCCols<IT,NT> & SpDCCols<IT,NT>::operator+= (
const SpDCCols<IT,NT> & rhs)
400 if(m == rhs.m && n == rhs.n)
408 dcsc =
new Dcsc<IT,NT>(*(rhs.dcsc));
413 (*dcsc) += (*(rhs.dcsc));
419 std::cout<<
"Not addable: " << m <<
"!=" << rhs.m <<
" or " << n <<
"!=" << rhs.n <<std::endl;
424 std::cout<<
"Missing feature (A+A): Use multiply with 2 instead !"<<std::endl;
429template <
class IT,
class NT>
430template <
typename _UnaryOperation,
typename GlobalIT>
431SpDCCols<IT,NT>* SpDCCols<IT,NT>::PruneI(_UnaryOperation __unary_op,
bool inPlace, GlobalIT rowOffset, GlobalIT colOffset)
435 Dcsc<IT,NT>* ret = dcsc->PruneI (__unary_op, inPlace, rowOffset, colOffset);
450 SpDCCols<IT,NT>* retcols =
new SpDCCols<IT, NT>();
452 retcols->nnz = retcols->dcsc->nz;
466 SpDCCols<IT,NT>* retcols =
new SpDCCols<IT, NT>();
467 retcols->dcsc = NULL;
476template <
class IT,
class NT>
477template <
typename _UnaryOperation>
478SpDCCols<IT,NT>* SpDCCols<IT,NT>::Prune(_UnaryOperation __unary_op,
bool inPlace)
482 Dcsc<IT,NT>* ret = dcsc->Prune (__unary_op, inPlace);
497 SpDCCols<IT,NT>* retcols =
new SpDCCols<IT, NT>();
499 retcols->nnz = retcols->dcsc->nz;
513 SpDCCols<IT,NT>* retcols =
new SpDCCols<IT, NT>();
514 retcols->dcsc = NULL;
525template <
class IT,
class NT>
526template <
typename _BinaryOperation>
527SpDCCols<IT,NT>* SpDCCols<IT,NT>::PruneColumn(
NT* pvals, _BinaryOperation __binary_op,
bool inPlace)
531 Dcsc<IT,NT>* ret = dcsc->PruneColumn (pvals, __binary_op, inPlace);
546 SpDCCols<IT,NT>* retcols =
new SpDCCols<IT, NT>();
548 retcols->nnz = retcols->dcsc->nz;
562 SpDCCols<IT,NT>* retcols =
new SpDCCols<IT, NT>();
563 retcols->dcsc = NULL;
572template <
class IT,
class NT>
573void SpDCCols<IT,NT>::PruneColumnByIndex(
const std::vector<IT>& ci)
577 dcsc->PruneColumnByIndex(ci);
582template <
class IT,
class NT>
583template <
typename _BinaryOperation>
584SpDCCols<IT,NT>* SpDCCols<IT,NT>::PruneColumn(
IT* pinds,
NT* pvals, _BinaryOperation __binary_op,
bool inPlace)
588 Dcsc<IT,NT>* ret = dcsc->PruneColumn (pinds, pvals, __binary_op, inPlace);
603 SpDCCols<IT,NT>* retcols =
new SpDCCols<IT, NT>();
605 retcols->nnz = retcols->dcsc->nz;
619 SpDCCols<IT,NT>* retcols =
new SpDCCols<IT, NT>();
620 retcols->dcsc = NULL;
630template <
class IT,
class NT>
631void SpDCCols<IT,NT>::SetDifference (
const SpDCCols<IT,NT> & rhs)
635 if(m == rhs.m && n == rhs.n)
641 else if (rhs.nnz != 0 && nnz != 0)
643 dcsc->SetDifference (*(rhs.dcsc));
651 std::cout<<
"Matrices do not conform for A - B !"<<std::endl;
656 std::cout<<
"Missing feature (A .* A): Use Square_EWise() instead !"<<std::endl;
661template <
class IT,
class NT>
662void SpDCCols<IT,NT>::EWiseMult (
const SpDCCols<IT,NT> & rhs,
bool exclude)
666 if(m == rhs.m && n == rhs.n)
676 *
this = SpDCCols<IT,NT>(0,m,n,0);
679 else if (rhs.nnz != 0 && nnz != 0)
681 dcsc->EWiseMult (*(rhs.dcsc), exclude);
689 std::cout<<
"Matrices do not conform for A .* op(B) !"<<std::endl;
694 std::cout<<
"Missing feature (A .* A): Use Square_EWise() instead !"<<std::endl;
701template <
class IT,
class NT>
702void SpDCCols<IT,NT>::EWiseScale(
NT ** scaler,
IT m_scaler,
IT n_scaler)
704 if(m == m_scaler && n == n_scaler)
707 dcsc->EWiseScale (scaler);
711 std::cout<<
"Matrices do not conform for EWiseScale !"<<std::endl;
720template <
class IT,
class NT>
721void SpDCCols<IT,NT>::CreateImpl(
IT * _cp,
IT * _jc,
IT * _ir,
NT * _numx,
IT _nz,
IT _nzc,
IT _m,
IT _n)
728 dcsc =
new Dcsc<IT,NT>(_cp, _jc, _ir, _numx, _nz, _nzc,
false);
733template <
class IT,
class NT>
734void SpDCCols<IT,NT>::CreateImpl(
const std::vector<IT> & essentials)
736 assert(essentials.size() == esscount);
742 dcsc =
new Dcsc<IT,NT>(nnz,essentials[3]);
747template <
class IT,
class NT>
748void SpDCCols<IT,NT>::CreateImpl(
IT size,
IT nRow,
IT nCol, std::tuple<IT, IT, NT> * mytuples)
750 SpTuples<IT,NT> tuples(
size, nRow, nCol, mytuples);
751 tuples.SortColBased();
754 std::pair<IT,IT> rlim = tuples.RowLimits();
755 std::pair<IT,IT> clim = tuples.ColLimits();
758 std::stringstream ss;
761 MPI_Comm_rank(MPI_COMM_WORLD, &myrank);
764 std::string ofilename =
"Read";
766 oput.open(ofilename.c_str(), std::ios_base::app );
767 oput <<
"Creating of dimensions " << nRow <<
"-by-" << nCol <<
" of size: " <<
size <<
768 " with row range (" << rlim.first <<
"," << rlim.second <<
") and column range (" << clim.first <<
"," << clim.second <<
")" << std::endl;
769 if(tuples.getnnz() > 0)
771 IT minfr = joker::get<0>(tuples.front());
772 IT minto = joker::get<1>(tuples.front());
773 IT maxfr = joker::get<0>(tuples.back());
774 IT maxto = joker::get<1>(tuples.back());
776 oput <<
"Min: " << minfr <<
", " << minto <<
"; Max: " << maxfr <<
", " << maxto << std::endl;
781 SpDCCols<IT,NT> object(tuples,
false);
786template <
class IT,
class NT>
787std::vector<IT> SpDCCols<IT,NT>::GetEssentials()
const
789 std::vector<IT> essentials(esscount);
793 essentials[3] = (nnz > 0) ? dcsc->nzc : 0;
797template <
class IT,
class NT>
798template <
typename NNT>
799SpDCCols<IT,NT>::operator SpDCCols<IT,NNT> ()
const
803 convert =
new Dcsc<IT,NNT>(*dcsc);
807 return SpDCCols<IT,NNT>(m, n,
convert);
811template <
class IT,
class NT>
812template <
typename NIT,
typename NNT>
813SpDCCols<IT,NT>::operator SpDCCols<NIT,NNT> ()
const
817 convert =
new Dcsc<NIT,NNT>(*dcsc);
821 return SpDCCols<NIT,NNT>(m, n,
convert);
825template <
class IT,
class NT>
826Arr<IT,NT> SpDCCols<IT,NT>::GetArrays()
const
832 arr.indarrs[0] = LocArr<IT,IT>(dcsc->cp, dcsc->nzc+1);
833 arr.indarrs[1] = LocArr<IT,IT>(dcsc->jc, dcsc->nzc);
834 arr.indarrs[2] = LocArr<IT,IT>(dcsc->ir, dcsc->nz);
835 arr.numarrs[0] = LocArr<NT,IT>(dcsc->numx, dcsc->nz);
839 arr.indarrs[0] = LocArr<IT,IT>(NULL, 0);
840 arr.indarrs[1] = LocArr<IT,IT>(NULL, 0);
841 arr.indarrs[2] = LocArr<IT,IT>(NULL, 0);
842 arr.numarrs[0] = LocArr<NT,IT>(NULL, 0);
853template <
class IT,
class NT>
854void SpDCCols<IT,NT>::Transpose()
858 SpTuples<IT,NT> Atuples(*
this);
859 Atuples.SortRowBased();
862 *
this = SpDCCols<IT,NT>(Atuples,
true);
866 *
this = SpDCCols<IT,NT>(0, n, m, 0);
876template <
class IT,
class NT>
877SpDCCols<IT,NT> SpDCCols<IT,NT>::TransposeConst()
const
879 SpTuples<IT,NT> Atuples(*
this);
880 Atuples.SortRowBased();
882 return SpDCCols<IT,NT>(Atuples,
true);
890template <
class IT,
class NT>
891SpDCCols<IT,NT> * SpDCCols<IT,NT>::TransposeConstPtr()
const
893 SpTuples<IT,NT> Atuples(*
this);
894 Atuples.SortRowBased();
896 return new SpDCCols<IT,NT>(Atuples,
true);
905template <
class IT,
class NT>
906void SpDCCols<IT,NT>::Split(SpDCCols<IT,NT> & partA, SpDCCols<IT,NT> & partB)
911 std::cout<<
"Matrix is too small to be splitted" << std::endl;
915 Dcsc<IT,NT> *Adcsc = NULL;
916 Dcsc<IT,NT> *Bdcsc = NULL;
920 dcsc->Split(Adcsc, Bdcsc, cut);
923 partA = SpDCCols<IT,NT> (m, cut, Adcsc);
924 partB = SpDCCols<IT,NT> (m, n-cut, Bdcsc);
927 *
this = SpDCCols<IT, NT>();
935template <
class IT,
class NT>
936void SpDCCols<IT,NT>::ColSplit(
int parts, std::vector< SpDCCols<IT,NT> > & matrices)
940 matrices.emplace_back(*
this);
944 std::vector<IT> cuts(parts-1);
945 for(
int i=0; i< (parts-1); ++i)
947 cuts[i] = (i+1) * (n/parts);
951 std::cout<<
"Matrix is too small to be splitted" << std::endl;
954 std::vector< Dcsc<IT,NT> * > dcscs(parts, NULL);
958 dcsc->ColSplit(dcscs, cuts);
961 for(
int i=0; i< (parts-1); ++i)
963 SpDCCols<IT,NT> matrix = SpDCCols<IT,NT>(m, (n/parts), dcscs[i]);
964 matrices.emplace_back(matrix);
966 SpDCCols<IT,NT> matrix = SpDCCols<IT,NT>(m, n-cuts[parts-2], dcscs[parts-1]);
967 matrices.emplace_back(matrix);
969 *
this = SpDCCols<IT, NT>();
977template <
class IT,
class NT>
978void SpDCCols<IT,NT>::ColSplit(
int parts, std::vector< SpDCCols<IT,NT>* > & matrices)
982 matrices.emplace_back(
new SpDCCols<IT,NT>(*
this));
986 std::vector<IT> cuts(parts-1);
987 for(
int i=0; i< (parts-1); ++i)
989 cuts[i] = (i+1) * (n/parts);
993 std::cout<<
"Matrix is too small to be splitted" << std::endl;
996 std::vector< Dcsc<IT,NT> * > dcscs(parts, NULL);
1000 dcsc->ColSplit(dcscs, cuts);
1003 for(
int i=0; i< (parts-1); ++i)
1005 SpDCCols<IT,NT>* matrix =
new SpDCCols<IT,NT>(m, (n/parts), dcscs[i]);
1006 matrices.emplace_back(matrix);
1008 SpDCCols<IT,NT>* matrix =
new SpDCCols<IT,NT>(m, n-cuts[parts-2], dcscs[parts-1]);
1009 matrices.emplace_back(matrix);
1011 *
this = SpDCCols<IT, NT>();
1019template <
class IT,
class NT>
1020void SpDCCols<IT,NT>::ColSplit(std::vector<IT> & cutSizes, std::vector< SpDCCols<IT,NT> > & matrices)
1023 int parts = cutSizes.size();
1024 for(
int i = 0; i < parts; i++) totn += cutSizes[i];
1026 matrices.emplace_back(*
this);
1029 std::cout <<
"Cut sizes are not appropriate" << std::endl;
1033 std::vector<IT> cuts(parts-1);
1034 cuts[0] = cutSizes[0];
1035 for(
int i = 1; i < parts-1; i++){
1036 cuts[i] = cuts[i-1] + cutSizes[i];
1038 std::vector< Dcsc<IT,NT> * > dcscs(parts, NULL);
1041 dcsc->ColSplit(dcscs, cuts);
1044 for(
int i=0; i< parts; ++i){
1045 SpDCCols<IT,NT> matrix = SpDCCols<IT,NT>(m, cutSizes[i], dcscs[i]);
1046 matrices.emplace_back(matrix);
1049 *
this = SpDCCols<IT, NT>();
1058template <
class IT,
class NT>
1059void SpDCCols<IT,NT>::ColSplit(std::vector<IT> & cutSizes, std::vector< SpDCCols<IT,NT>* > & matrices)
1062 int parts = cutSizes.size();
1063 for(
int i = 0; i < parts; i++) totn += cutSizes[i];
1065 matrices.emplace_back(
new SpDCCols<IT,NT>(*
this));
1068 std::cout <<
"Cut sizes are not appropriate" << std::endl;
1072 std::vector<IT> cuts(parts-1);
1073 cuts[0] = cutSizes[0];
1074 for(
int i = 1; i < parts-1; i++){
1075 cuts[i] = cuts[i-1] + cutSizes[i];
1077 std::vector< Dcsc<IT,NT> * > dcscs(parts, NULL);
1080 dcsc->ColSplit(dcscs, cuts);
1083 for(
int i=0; i< parts; ++i){
1084 SpDCCols<IT,NT>* matrix =
new SpDCCols<IT,NT>(m, cutSizes[i], dcscs[i]);
1085 matrices.emplace_back(matrix);
1088 *
this = SpDCCols<IT, NT>();
1095template <
class IT,
class NT>
1096void SpDCCols<IT,NT>::ColConcatenate(std::vector< SpDCCols<IT,NT> > & matrices)
1098 std::vector< SpDCCols<IT,NT> * > nonempties;
1099 std::vector< Dcsc<IT,NT> * > dcscs;
1100 std::vector< IT > offsets;
1101 IT runningoffset = 0;
1103 for(
size_t i=0; i< matrices.size(); ++i)
1105 if(matrices[i].nnz != 0)
1107 nonempties.push_back(&(matrices[i]));
1108 dcscs.push_back(matrices[i].dcsc);
1109 offsets.push_back(runningoffset);
1111 runningoffset += matrices[i].n;
1114 if(nonempties.size() < 1)
1117 std::cout <<
"Nothing to ColConcatenate" << std::endl;
1128 Dcsc<IT,NT> * Cdcsc =
new Dcsc<IT,NT>();
1129 Cdcsc->ColConcatenate(dcscs, offsets);
1130 *
this = SpDCCols<IT,NT> (nonempties[0]->m, runningoffset, Cdcsc);
1134 for(
size_t i=0; i< matrices.size(); ++i)
1136 matrices[i] = SpDCCols<IT,NT>();
1144template <
class IT,
class NT>
1145void SpDCCols<IT,NT>::ColConcatenate(std::vector< SpDCCols<IT,NT>* > & matrices)
1147 std::vector< SpDCCols<IT,NT> * > nonempties;
1148 std::vector< Dcsc<IT,NT> * > dcscs;
1149 std::vector< IT > offsets;
1150 IT runningoffset = 0;
1152 for(
size_t i=0; i< matrices.size(); ++i)
1154 if(matrices[i]->nnz != 0)
1156 nonempties.push_back(matrices[i]);
1157 dcscs.push_back(matrices[i]->dcsc);
1158 offsets.push_back(runningoffset);
1160 runningoffset += matrices[i]->n;
1163 if(nonempties.size() < 1)
1166 std::cout <<
"Nothing to ColConcatenate" << std::endl;
1177 Dcsc<IT,NT> * Cdcsc =
new Dcsc<IT,NT>();
1178 Cdcsc->ColConcatenate(dcscs, offsets);
1179 *
this = SpDCCols<IT,NT> (nonempties[0]->m, runningoffset, Cdcsc);
1183 for(
size_t i=0; i< matrices.size(); ++i)
1194template <
class IT,
class NT>
1195void SpDCCols<IT,NT>::Merge(SpDCCols<IT,NT> & partA, SpDCCols<IT,NT> & partB)
1197 assert( partA.m == partB.m );
1199 Dcsc<IT,NT> * Cdcsc =
new Dcsc<IT,NT>();
1201 if(partA.nnz == 0 && partB.nnz == 0)
1205 else if(partA.nnz == 0)
1207 Cdcsc =
new Dcsc<IT,NT>(*(partB.dcsc));
1208 std::transform(Cdcsc->jc, Cdcsc->jc + Cdcsc->nzc, Cdcsc->jc, std::bind2nd(std::plus<IT>(), partA.n));
1210 else if(partB.nnz == 0)
1212 Cdcsc =
new Dcsc<IT,NT>(*(partA.dcsc));
1216 Cdcsc->Merge(partA.dcsc, partB.dcsc, partA.n);
1218 *
this = SpDCCols<IT,NT> (partA.m, partA.n + partB.n, Cdcsc);
1220 partA = SpDCCols<IT, NT>();
1221 partB = SpDCCols<IT, NT>();
1230template <
class IT,
class NT>
1232int SpDCCols<IT,NT>::PlusEq_AnXBt(
const SpDCCols<IT,NT> &
A,
const SpDCCols<IT,NT> &
B)
1234 if(
A.isZero() ||
B.isZero())
1238 Isect<IT> *isect1, *isect2, *itr1, *itr2, *cols, *rows;
1239 SpHelper::SpIntersect(*(
A.dcsc), *(
B.dcsc), cols, rows, isect1, isect2, itr1, itr2);
1241 IT kisect =
static_cast<IT>(itr1-isect1);
1248 StackEntry< NT, std::pair<IT,IT> > * multstack;
1249 IT cnz = SpHelper::SpCartesian< SR > (*(
A.dcsc), *(
B.dcsc), kisect, isect1, isect2, multstack);
1256 dcsc =
new Dcsc<IT,NT>(multstack, mdim, ndim, cnz);
1260 dcsc->AddAndAssign(multstack, mdim, ndim, cnz);
1264 delete [] multstack;
1274template <
class IT,
class NT>
1275template <
typename SR>
1276int SpDCCols<IT,NT>::PlusEq_AnXBn(
const SpDCCols<IT,NT> &
A,
const SpDCCols<IT,NT> &
B)
1278 if(
A.isZero() ||
B.isZero())
1282 StackEntry< NT, std::pair<IT,IT> > * multstack;
1283 int cnz = SpHelper::SpColByCol< SR > (*(
A.dcsc), *(
B.dcsc),
A.n, multstack);
1289 dcsc =
new Dcsc<IT,NT>(multstack, mdim, ndim, cnz);
1293 dcsc->AddAndAssign(multstack, mdim, ndim, cnz);
1297 delete [] multstack;
1302template <
class IT,
class NT>
1303template <
typename SR>
1304int SpDCCols<IT,NT>::PlusEq_AtXBn(
const SpDCCols<IT,NT> &
A,
const SpDCCols<IT,NT> &
B)
1306 std::cout <<
"PlusEq_AtXBn function has not been implemented yet !" << std::endl;
1310template <
class IT,
class NT>
1311template <
typename SR>
1312int SpDCCols<IT,NT>::PlusEq_AtXBt(
const SpDCCols<IT,NT> &
A,
const SpDCCols<IT,NT> &
B)
1314 std::cout <<
"PlusEq_AtXBt function has not been implemented yet !" << std::endl;
1319template <
class IT,
class NT>
1320SpDCCols<IT,NT> SpDCCols<IT,NT>::operator() (
IT ri,
IT ci)
const
1322 IT * itr = std::find(dcsc->jc, dcsc->jc + dcsc->nzc, ci);
1323 if(itr != dcsc->jc + dcsc->nzc)
1325 IT irbeg = dcsc->cp[itr - dcsc->jc];
1326 IT irend = dcsc->cp[itr - dcsc->jc + 1];
1328 IT * ele = std::find(dcsc->ir + irbeg, dcsc->ir + irend, ri);
1329 if(ele != dcsc->ir + irend)
1331 SpDCCols<IT,NT> SingEleMat(1, 1, 1, 1);
1332 *(SingEleMat.dcsc->numx) = dcsc->numx[ele - dcsc->ir];
1333 *(SingEleMat.dcsc->ir) = *ele;
1334 *(SingEleMat.dcsc->jc) = *itr;
1335 (SingEleMat.dcsc->cp)[0] = 0;
1336 (SingEleMat.dcsc->cp)[1] = 1;
1341 return SpDCCols<IT,NT>();
1346 return SpDCCols<IT,NT>();
1354template <
class IT,
class NT>
1355SpDCCols<IT,NT> SpDCCols<IT,NT>::operator() (
const std::vector<IT> & ri,
const std::vector<IT> & ci)
const
1359 IT rsize = ri.size();
1360 IT csize = ci.size();
1362 if(rsize == 0 && csize == 0)
1366 return SpDCCols<IT,NT> (0, m, n, 0);
1370 return ColIndex(ci);
1374 SpDCCols<IT,NT> LeftMatrix(rsize, rsize, this->m, ri,
true);
1375 return LeftMatrix.OrdColByCol< PT >(*this);
1379 SpDCCols<IT,NT> LeftMatrix(rsize, rsize, this->m, ri,
true);
1380 SpDCCols<IT,NT> RightMatrix(csize, this->n, csize, ci,
false);
1381 return LeftMatrix.OrdColByCol< PT >( OrdColByCol< PT >(RightMatrix) );
1385template <
class IT,
class NT>
1386std::ofstream & SpDCCols<IT,NT>::put(std::ofstream & outfile)
const
1390 outfile <<
"Matrix doesn't have any nonzeros" <<std::endl;
1393 SpTuples<IT,NT> tuples(*
this);
1394 outfile << tuples << std::endl;
1399template <
class IT,
class NT>
1400std::ifstream & SpDCCols<IT,NT>::get(std::ifstream & infile)
1402 std::cout <<
"Getting... SpDCCols" << std::endl;
1404 infile >> m >> n >> nnz;
1405 SpTuples<IT,NT> tuples(nnz, m, n);
1407 tuples.SortColBased();
1409 SpDCCols<IT,NT> object(tuples,
false);
1415template<
class IT,
class NT>
1416void SpDCCols<IT,NT>::PrintInfo(std::ofstream & out)
const
1419 out <<
", n: " << n ;
1420 out <<
", nnz: "<< nnz ;
1424 out <<
", local splits: " << splits << std::endl;
1430 out <<
", nzc: "<< dcsc->nzc << std::endl;
1434 out <<
", nzc: "<< 0 << std::endl;
1439template<
class IT,
class NT>
1440void SpDCCols<IT,NT>::PrintInfo()
const
1442 std::cout <<
"m: " << m ;
1443 std::cout <<
", n: " << n ;
1444 std::cout <<
", nnz: "<< nnz ;
1448 std::cout <<
", local splits: " << splits << std::endl;
1454 std::cout <<
", nzc: "<< dcsc->nzc << std::endl;
1458 std::cout <<
", nzc: "<< 0 << std::endl;
1463 std::string **
A = SpHelper::allocate2D<std::string>(m,n);
1464 for(
IT i=0; i< m; ++i)
1465 for(
IT j=0; j<n; ++j)
1469 for(
IT i=0; i< dcsc->nzc; ++i)
1471 for(
IT j = dcsc->cp[i]; j<dcsc->cp[i+1]; ++j)
1473 IT colid = dcsc->jc[i];
1474 IT rowid = dcsc->ir[j];
1475 A[rowid][colid] = std::to_string(dcsc->numx[j]);
1479 for(
IT i=0; i< m; ++i)
1481 for(
IT j=0; j<n; ++j)
1483 std::cout <<
A[i][j];
1486 std::cout << std::endl;
1488 SpHelper::deallocate2D(
A,m);
1499template <
class IT,
class NT>
1500SpDCCols<IT,NT>::SpDCCols(
IT nRow,
IT nCol, Dcsc<IT,NT> * mydcsc)
1501:dcsc(mydcsc), m(nRow), n(nCol), splits(0)
1510template <
class IT,
class NT>
1511SpDCCols<IT,NT>::SpDCCols (
IT size,
IT nRow,
IT nCol,
const std::vector<IT> & indices,
bool isRow)
1512:m(nRow), n(nCol), nnz(
size), splits(0)
1515 dcsc =
new Dcsc<IT,NT>(
size,indices,isRow);
1525template <
class IT,
class NT>
1526inline void SpDCCols<IT,NT>::CopyDcsc(Dcsc<IT,NT> * source)
1530 dcsc =
new Dcsc<IT,NT>(*source);
1541template <
class IT,
class NT>
1542SpDCCols<IT,NT> SpDCCols<IT,NT>::ColIndex(
const std::vector<IT> & ci)
const
1544 IT csize = ci.size();
1547 return SpDCCols<IT,NT>(0, m, csize, 0);
1551 return SpDCCols<IT,NT>(0, m,0, 0);
1557 for(
IT i=0, j=0; i< dcsc->nzc && j < csize;)
1559 if((dcsc->jc)[i] < ci[j])
1563 else if ((dcsc->jc)[i] > ci[j])
1569 estsize += (dcsc->cp)[i+1] - (dcsc->cp)[i];
1576 SpDCCols<IT,NT> SubA(estsize, m, csize, estnzc);
1581 SubA.dcsc->cp[0] = 0;
1584 for(
IT i=0, j=0; i < dcsc->nzc && j < csize;)
1586 if((dcsc->jc)[i] < ci[j])
1590 else if ((dcsc->jc)[i] > ci[j])
1596 IT columncount = (dcsc->cp)[i+1] - (dcsc->cp)[i];
1597 SubA.dcsc->jc[cnzc++] = j;
1598 SubA.dcsc->cp[cnzc] = SubA.dcsc->cp[cnzc-1] + columncount;
1599 std::copy(dcsc->ir + dcsc->cp[i], dcsc->ir + dcsc->cp[i+1], SubA.dcsc->ir + cnz);
1600 std::copy(dcsc->numx + dcsc->cp[i], dcsc->numx + dcsc->cp[i+1], SubA.dcsc->numx + cnz);
1609template <
class IT,
class NT>
1610template <
typename SR,
typename NTR>
1611SpDCCols< IT, typename promote_trait<NT,NTR>::T_promote > SpDCCols<IT,NT>::OrdOutProdMult(
const SpDCCols<IT,NTR> & rhs)
const
1615 if(isZero() || rhs.isZero())
1617 return SpDCCols< IT, T_promote > (0, m, rhs.n, 0);
1619 SpDCCols<IT,NTR> Btrans = rhs.TransposeConst();
1621 Isect<IT> *isect1, *isect2, *itr1, *itr2, *cols, *rows;
1622 SpHelper::SpIntersect(*dcsc, *(Btrans.dcsc), cols, rows, isect1, isect2, itr1, itr2);
1624 IT kisect =
static_cast<IT>(itr1-isect1);
1628 return SpDCCols< IT, T_promote > (0, m, rhs.n, 0);
1630 StackEntry< T_promote, std::pair<IT,IT> > * multstack;
1631 IT cnz = SpHelper::SpCartesian< SR > (*dcsc, *(Btrans.dcsc), kisect, isect1, isect2, multstack);
1634 Dcsc<IT, T_promote> * mydcsc = NULL;
1637 mydcsc =
new Dcsc< IT,T_promote > (multstack, m, rhs.n, cnz);
1638 delete [] multstack;
1640 return SpDCCols< IT,T_promote > (m, rhs.n, mydcsc);
1644template <
class IT,
class NT>
1645template <
typename SR,
typename NTR>
1646SpDCCols< IT, typename promote_trait<NT,NTR>::T_promote > SpDCCols<IT,NT>::OrdColByCol(
const SpDCCols<IT,NTR> & rhs)
const
1650 if(isZero() || rhs.isZero())
1652 return SpDCCols<IT, T_promote> (0, m, rhs.n, 0);
1654 StackEntry< T_promote, std::pair<IT,IT> > * multstack;
1655 IT cnz = SpHelper::SpColByCol< SR > (*dcsc, *(rhs.dcsc), n, multstack);
1657 Dcsc<IT,T_promote > * mydcsc = NULL;
1660 mydcsc =
new Dcsc< IT,T_promote > (multstack, m, rhs.n, cnz);
1661 delete [] multstack;
1663 return SpDCCols< IT,T_promote > (m, rhs.n, mydcsc);
void convert(string ifname, string ofname, string sort="revsize")